diff --git a/.gitattributes b/.gitattributes
index 7951c7c135148f7d2ec7ac68774eec1d510c312c..a120fc76048a89c057f53b8c865e53efc41cae0b 100644
--- a/.gitattributes
+++ b/.gitattributes
@@ -4932,3 +4932,892 @@ wall-prob-tree-13x13/wp_0.9_alpha_1/files/media/videos/train-replay/rollouts_gif
 wall-prob-tree-13x13/wp_0.9_alpha_1/files/media/videos/train-replay/rollouts_gif_2016_3550de9877c0b4ada353.gif filter=lfs diff=lfs merge=lfs -text
 wall-prob-tree-13x13/wp_0.9_alpha_1/files/media/videos/train-replay/rollouts_gif_5088_c5313d6438b86d0c34d6.gif filter=lfs diff=lfs merge=lfs -text
 wall-prob-tree-13x13/wp_0.9_alpha_1/run-ponuksup.wandb filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/14065f5f7e608cd567d4b276f5b9cebf filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/aa9f6b4f8103559231db4a9d6e075637 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/c3fa4628eb352e5e16eb72654572e80f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/9a4dffa36085bbf7e523a43f926ff88c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/a6a3ae1d6fc2453df7f3c981b1dd58ce filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/19897cc70c3e579a4701bc6bedd3cd30 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/f73893b3800ac13b8528776462c2ae26 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/2118a11b11e7606bf962398b0a72c5d8 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/a576c7d2d5bc2e11831bb056632d7208 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/d7642041ab8645de8ee4f92df130161f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/24e38a1b5d97cf9d6d9970a4d3d99472 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/2f4cb6327def353fc2355370b2bc687c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/2be25adc9d7f28355836cbfe8376291e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/ec31cddb61102f429a7582b45d6b5deb filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/9a6f6f80b65a3214d6c599318bd00504 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/a98b594633221a2d39dca7eb1306f774 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/da1988eabc9e008acbbcc87ad9e80424 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/e35a878e67066209be052fa81262353d filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/93049ac0151d5e051fd486717a5b92d4 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/f72f06833f3b9eda1d0779c2ea631fb7 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/4c9348416f9a37bcbec31d7d386aed6d filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/dd01a8c95a6ac8f306f723df55a566fb filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/21c41450a859fb1c4fc6787da008c8f8 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/31ec8309a3f1d4764f203899f20bb374 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/c4385e1ec115fd1980cd71d66008d970 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/d505e8c0e66c95e6ebc3ce3b032dbfa7 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/6a35290584552b8d01d141201c7ca47d filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/c3b57d3ab4fb2f8500f5878a5a5fb849 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/93f7c0a0013c727c9fbc80056c37d404 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/9d539ba4a88bac57cff042ea9f066b95 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/135f477c9fba1ce746558f2e49938118 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/d12b1d56dcda152a29e89b1e149cccdd filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/ad4357335f1a234b0c9287a0ed692c7f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/bfe08001f4f4f106703a1464e0dada7f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/115cb89b1e23935a36665a44caef850a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/e12f6a3db68a90ffa95374d0f339ffcb filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/510e83f18e413538e91f66729178752f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/7d9919fb39f2f7eb21d7a7baec26808c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/71680f44f2fd187bc9624f82f79c96d7 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/f3c9773476d82585a2cf7e3e82759bdb filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/89a264fa5ca6139df168b9e7960ea60a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/b44a7962f3411d8c8c6cb3654c159cbd filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/21f622c1d71e934137518ee35b0a5ab5 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/b6c6addfbd2073713b8db261bdef76a4 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/66c8e6428f49043f7137102f8fcf83c5 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/c07ecb7d151e2ac360dd65cdab891a7a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/46a1759e0bd7a84059ec598089ae6e84 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/a7c3cce902b66cf42d9ffb3d3cff8a43 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/44ff4e8f12e8c5fb8969cf4800d7dae0 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/e41eb6b2053ba81412c0715cccbb81ea filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/855c65aa943fa6d0a93f05193908c0f8 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/e9db3f54734f35b52ae729cfbd446ead filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/614433fa59a2480fde06276deb96932b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/944e606be351d41f7dc5b1bf37efcccd filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/3592c1883d9c392d8d571ef602c6ffb7 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/a0f43e574e44af6506d367ac2d42ba75 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/398d655902ce2df11f1f3949f6830417 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/52f68df65b5ba985e762db1a385b1843 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/80c7594b024b466555aee6c0127bec56 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/c0ce8fa8d312846737795a459433b2d2 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/6f9af543e9cc92e9f7c73e2f9f9132ec filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/b3e7266483b095547a1d10df58bd9a7a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/527e2eb739ba13dab5647eb0c4e3e0f4 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/22b956c6107b539c20584cc44c6ee6b4 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/6803b43051ac4bec22e32c523f21b585 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/a6d1067f7ff17500850e303d5d1adadc filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/c7471136f185c6eeab8e5f8c525b3e34 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/2af1b58309b1805503e472c1de79fc77 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/70829fdc66affb6d17cd7b8dca68643c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/d094d661476d7ba6cf74d6a674473ed2 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/d1ed51ac61beb0b1d04f6b9e7272be58 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/5944fcb123ca1ad039f9c550a14303a6 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/ad28a0293b5e54125e025f606d7eb075 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/12fbd7b5e42344862b16ed466987d842 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/14d3bf358fcb3badf34c5c37ba1be1ff filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/0ce40dcdbf58dd2f61c2bfd4bc814f09 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/843d9d8c5f42e4114457e189c0c788f8 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/53bbbb4aa7e0d38a9a10218a7eec55bf filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/73cd9ed172392c93ffb36640c6073fd4 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/1528c63698ccb626e299692796656645 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/43b20833bd98982a019415f77cb361fe filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/2035e50ba15fa1c30cc18dd8555de790 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/8ece03f93432f1cd8235675a39f8b09a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/a3462313446182bad856f93d82425ecf filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/b7b09c15a83f58216fe3f736a8782b71 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/2fc25c971b42cc5e15814d2e8ad32d3e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/38aad302f5ad0f4c5a175f2e48062cb1 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/c26c2636f332e4a9c6f09ab5876bb86d filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/f80b2ca144369e9ee4da8cb6f1ceb22f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/f7a03761716f03aace07a1a022468c54 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/6de3bda0d9cc9da54179fc363882cd97 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/f75393ba9140e39f96ab07e137f4ee09 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/16ce0534d90b5dbd1e8e20db3cf8ed09 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/d69c602879cf589e15423a7239afbe24 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/39d924bd2ce7d26f0e30aefae8196467 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/83ae15e97ecb299d6fd8fe957b314981 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/46bbe50bccc393a4faa144b4fdf50c98 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/dc4fc5c3c7b133695d6683aa6c74bfb9 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/c90480b2adca42150bfe86ed1b3e69b1 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/e328073556e5fd67479fb6ca29eee754 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/42fd4a2e07b8479166e8cb0defaf249c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/8772788a55861c3631b014c293e5328a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/9b0c60f87d1217a33ee26a22b2dc41a1 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/b8da4fcad32d95f3344bebe84feeb33f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/31f56790053af52df0935d815098f420 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/40d714629c3cf2eb595ec74a5dd0bfdd filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/46f5998c0a54ca741be483e8ec6ae2d3 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/8571532fdda3cb005f95a5d6f83c5f77 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/c8bea4ba6d9763f1e306c52fad1894a3 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/f714b2a410d6a8400d81689823b07574 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/7cb6f65f197e9e5da0fed888eaac8a55 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/b0d44ea8e7585fe60d156183266ba17d filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/2633d9e55530680f7f586641d9173b11 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/375817f533476f2bd532b9f8805bb46b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/2e56c1184a259abf1c122855c9c2fe19 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/f53a454a4da99bb7ab185d5f2b76677d filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/5991ee7228b3d71632b644b5dbd88db3 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/78ecc53c64c1a303749cd6c112d69400 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/15a47401043129c383ae5016e686f665 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/cc3199f51eb916ace4cec44a24a03e34 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/bbc543d5cd785634fb24b3ea9bc5ae8e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/db5d2994037403cd317ccc8cd6efdaf4 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/64f25661ea1c834b93b34a7bb0c34744 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/fb0c4686ec479f92a27dcde0e7bfb54b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/828e19f6f7bd8b771fc28459d14e1cbf filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/eac4193ee16d9bfaa41865e13c3b6fdc filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/397eeeeaeac47b359117e5ca491b1c02 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/71f35085b5a9115b2ed9ff7a49533279 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/a6d84a43e1bbae4882e860f0c4548355 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/c6c3c8b3d2b5f201aab22df943704572 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/509073e7b788797e8d3bc577d1f0f9bc filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/ba4d0445f50a8876a55b52b752b7a9ee filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/38894fbd397dfb5d2d839706f70eb7eb filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/8b3ca8312cef9af65d4150291994305b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/1e03d0173aad08ff56af3b32943a69fc filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/bfe9124c0d87e82a9a1b6778313dbc5f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/1e3a5f7b03fb48de5c95b6619b0c55c4 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/b16406f1dd98ef168d96cde8f9721e73 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/49b765118cb23e604fded968cc92323b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/fdda237eb7ce577d14039e28520184a1 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/5b29a585ccb9e5436b503814559f7067 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/a6469f100009ec813dc89697ba9cd763 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/368b831b402b4e44632b8f16425f14ba filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/d6b2dff90e7fe0fbd7184f2551d1c61b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/208fed29bd0a506ca816a54acdfa1e56 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/8af4364c5ad5c47b5425d39f68af054f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/1991ed1e10326388e3929434d80bb932 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/dd8bee54ed91e91ab2227eefb0b64e7a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/49d33d076910ced05911075a401f527c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/9495287317b56964ade04f87e03fb646 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/b74be78bac76ff58da5a9da71fef321a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/c12f49dfcd319f64ae928c7d21d8f3dc filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/0328838b6df83a5d443446ee4dce0f32 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/304115fa3fdffc8daf39d7b9a63c40e2 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/19387ffb5598d72dcfc1af15b81a4288 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/4abf08ad9ff543fc451aaddb6571ee79 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/7b430006749f0a454a60f78b1a22cd29 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/bba18a4656339fa15a4a90f9c3f27503 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/75c2ab2733c57ef8ac39f8e6678cbc97 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/7e595f0558392eddb3039ffdc0c75ee7 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/120f98e39fc7a0d88e132ff440a0d2e7 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/4481bd7afea1202541fe7b337bd0481d filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/1df0299962a5e93df60f242edad4ae11 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/211756e4fc3286b3a4d895ee3c1763cc filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/1b7f6d12e7f57f8ab227191a7f3fbcb7 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/eb37db0748dc518562c7911c2ff72837 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/09d9ec642b647a62e93313eabebdaab4 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/8b7823472ad2c653d79e4d748c5873fb filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/235b3e7f5b8732da8ea15fecded19fb2 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/40d0bbb189cbc2a9e278ade801deece0 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/0509468c166fee8019c6ae42d7b0224f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/d0eca04fd9a29ad360dba533b4cf01a6 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/65cec42acf5abb1fa2499276234b6cda filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/fd52a2813009e105900375a96f457911 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/051efa471a7bcc7ad392fa3826fccb67 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/5ffc964b427ca5df75a2cfef73e7d352 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/965f86928986376022220e0bcaad4995 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/9929bd4c3b86a7a8b30329186746ee25 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/126055cd3d00a897517fba8fe31ba9fb filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/c00d0760616c5ab4bf83c9e64d3f4034 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/3e8e8fed443c80e2f3e186a314ddb756 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/910b919247ffaa26751e9b5cf9a76e35 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/c131f8a71412437c593df286c02a89ec filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/e4b44ddff0c1064ffbf2596de65fc022 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/2e34a99175f69fb81eabb32a9e3fbb28 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/615c9986c9365f8f29c7b01638d24983 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/28734af768bf608b05c4b7610f12e283 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/4913843e712a8c4e42db40b0662047d0 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/4acc4afb3fe1857adfcc0dd50a534af0 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/7eddf021947f8bcfe9a58e03a88345bf filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/2846cc8d40dc92f16bab4f8b8d0b2b7a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/fc0ebd7aee925c3353be40990679c917 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_0_f9a2c7b12fae2212b0df.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_2016_00d269aa1bc9116f8bd7.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_3040_725b9a6669dd81146e9c.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_4064_94f43fd1734d82639b65.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_5088_b3c2b0820fb0978fe942.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_992_8d84766ff56b911dc3c5.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_0_4d343630d22822dd23f9.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_2016_43487c06f7c8cf0ce4b3.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_3040_9c0d784d9b58feb05295.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_4064_46e3a9cb8d942efee5a7.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_5088_6c6bfa49d6645a4ec744.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_992_ea0b419521e17e14c4f8.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_0_bc8a36e71dc82e452ec5.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_2016_94d3910b42650d354441.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_3040_d0223c2c92823109e63d.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_4064_871827662648763adffe.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_5088_d89e30114bcc244cca07.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_992_100ea78a33a1369d62ea.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_0_92fcce4df8a2aebf7a46.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-all/rollouts_gif_0_aaa4468a8c6a23e11cd1.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-all/rollouts_gif_2016_e428956eff714d710b89.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-all/rollouts_gif_3040_a0da2a31a518817b6e32.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-all/rollouts_gif_4064_9f611d045168c7ee20d2.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-all/rollouts_gif_5088_16ca120b4aa0033a7e82.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-all/rollouts_gif_992_287901f7a7d2c26d0a85.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-generate/rollouts_gif_3040_a0da2a31a518817b6e32.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-generate/rollouts_gif_4064_9f611d045168c7ee20d2.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-generate/rollouts_gif_992_287901f7a7d2c26d0a85.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-replay/rollouts_gif_0_aaa4468a8c6a23e11cd1.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-replay/rollouts_gif_2016_e428956eff714d710b89.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-replay/rollouts_gif_5088_16ca120b4aa0033a7e82.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/run-wp_0.0_alpha_1.wandb filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/4ab02c394b802aced363252db83c1dee filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/a6c6be437ba217836fe168fa1012e2ba filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/b1041ad8104208025272b9e5d0c95389 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/f1dd3874cdec8cae74c9ffbe8f7524e1 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/08b1c66999e4746acdd31d332fb5fa6c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/43db39653722f748fd24578a5858b4da filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/92409fdc57cd4b54702da8cc6321b9a4 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/c9f242ed2d2810c4ed3917cbdff88bef filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/28c61ada2b511ed98b59f8af2e29cee2 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/6963f1550847deaeae29c600557818a8 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/0ff559359842a8d04f6b789f6d25e07c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/6a547ee5b403358f0473b049b5ca3427 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/27ba5a225d0ec2513782819d0c81bbcf filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/521ac19beb59256acd2bb3c326a28be6 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/8dcff388050a83cc67f8a69f83d938bd filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/a69df5845a474dc9e0b92293e8fd7613 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/9e0ab5da9f821773ce5c431a1ef5f70e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/b36d04854e981974598284690f6e7d35 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/96bc19425282f706850acfef7c036cb0 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/be921a4bed3020b2fd76a62bae3d80f4 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/a65a484bb6ebd2c7fe174980f2b408bf filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/ac136a93168bdae194f3733209fdaf32 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/072f5ae4c2056e2511e5e84fca0795d8 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/f95508e89f5dcf2d89ee4bdd9b78ad09 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/b79f6eb9c727341a583d420f43252c9e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/db0ff1bd938599d1fba0749c16397dfd filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/2faf607f0ff722e9e24da0f7ac43be4e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/46ec0b825d2ab81ca9c73a87067cd399 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/09103f6d933d8e646b0533f5a1d6b56e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/ddc6170fa1025762215884e722d282f5 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/e97e592102f2f73360be48f1fecd06c5 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/0006571684894e34b313ef657038e210 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/a7859ed79c1028d22bdc6501b9733c06 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/a326187bc642e0f4da8b87c22915beec filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/af99c8d9a321dec4daeeaa563d84e97c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/1a3cfcd6fa0a1f8b601573ef0406374c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/5e5f2f3e8e356e0fb29bf82e5fc586e4 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/981cf88349801bea79ff21a9ee502320 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/9f619b3036e38de15d2dba2ca21fe104 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/4b98720177d384cbd2ccda50ccbb172e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/c45e677fd02878c10b3ba75aaca46961 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/93a12c0def55eba04d111034da58bdc3 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/a2810388c8fbe9ee3cf069ced4855508 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/b19fb6ba48fbe2bf18de2a0533a9715b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/c8daee29bbb44342edd48c44f16bf60e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/1bc3bf01910de50fb3a0c8665460fee8 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/9b5005af93430771332d85d7fe14169a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/7c16d39b29f494ed0c7963ce920d6939 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/ddb81cd408c2cae8b4695ed8b29e3a01 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/1be8dba722c4740d0d32bdfeac1bb952 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/a89cbff66011ebc668cdccc94b0e5824 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/ecc852ecb448308a5820ff56809dfbab filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/fa994bd50a06db9db0670eb20ddce333 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/13b62b197acd654122b44245b59ff7c3 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/0c86bb3880b0aa0fb0daa9a89bd91e23 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/fbd878740d6fedada801f081db8dc3a1 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/8bb85d1461a16a590bbab46ae1f98d14 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/b5e44b88aa6e9804b148436affd5ecf5 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/b3b2ff0bbdbb1fff35e6674ff00c4ce0 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/eca0aa7edd75e21b35d6d3806c7224c0 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/407e8c652f327b4a5324711eeb83fa02 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/6b8f9277eab502f3ad0994f51599fffd filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/8531ce33fc56db52c28513eb6b762784 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/87807d9ce709b91cdcb059b858ce4d9f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/11364cc9854dac3e37c7513087c74350 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/8edc120237220bbeda09bb365c6ecd86 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/788e3100e6764596a94ed8126f63a76f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/4435c7c639b69dc8fb7972ae695f2e50 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/69414957e851803c85b68aced0bec3d4 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/040f2e79748b040023dcedbe0a6e18d1 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/146b90889bcef8073e60f7c72ae34116 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/2cfd1579a45dea476113c94d9265d101 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/dfbb8e68a79b8f6fce16ddaa206092cf filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/680b88edfb00978daa90f1787c9ab2d6 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/907dfde75583be2cdfec419eb1319969 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/65eef76c240e89beb264fc811f1dc085 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/dfd28b1e061576dd27309906776c6bb0 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/16672c3c651f10e1f392ce4a418bdb2d filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/388d9d4a94469c61e821ced62eb00b2f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/12e5713a3842ea89aae80faa5a4db522 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/49198107e3fff2faab54082b464a6c90 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/e2e49bdc567282e08d7e1d3fc2ada7ec filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/0ac01023d23c553503c7301a1129d7b6 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/fdb427fe39a700b8a1b559d8fac68184 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/4081a4ef9d4d1762c9ccf509a01dd7cb filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/9aec37f0dd8ddfc10c4dbd7718ee426e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/abd79adc576a29bf9198ba0cc0458cde filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/d64dd38e10433aed24d1914b60b6b775 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/2cb4f6839f965a868fd4b59948c394e6 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/c49e63a00315a6d9a0e4c356892fe828 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/4a6271c6fab55f2fc30a37a895fec244 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/f84a6c6eec17a8cb8b1432f9b63e16a8 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/a7cc308a2147fd0655564ee505070829 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/a99179ba74cb0f79d1b494aac81ec409 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/93d392ec1ace0cecafe2f1e392048ebc filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/bac894a308a52ee006788efa5755352a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/6a61973c4ddd830c40b27e193c8de3fe filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/d998aa4f18f0e24578b11d3fcd2b605b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/a76ff3dc5e07ccfa25fa1ac4adab25d4 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/f04e1eced8eb19291002fb8930ddb501 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/636457bfbb2d1d45f4e51fb47b1ea215 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/c406a5fecf5113e1a39defb19ea4302b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/46706b9e932be3c4f91d3577162f5f62 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/d63ac5b40bb0b72ea2da9b87848d1fd2 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/652862d98e52aeaecea9210ef71ccd60 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/8732dc0381ea72ca9b08fea6c87deef9 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/097bc235f9a9ebfb22bc2b5769f1aa86 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/ff9b49e2a363b175588a4d48e7161a82 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/1c82e6652e12cc388b7b12f34d1a7847 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/a1572db1892abdb444cd7c2b46040be8 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/458623a32cca2228650e9df682a4fe9d filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/ab78cd7ca6e2cc8c7b7a4e63f323de74 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/503b1a91b9581b4396c5af7f2795bc0f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/e5866a66eed10b448f57bce830da269d filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/7daab2913609df8f1bfae5a267638a9c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/d4244c8c0d4e96986db5793be0180ca8 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/ee8a88812e9ccf0da70c34e35e43f5c0 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/f1fcaa851415acb992c001f01d7862b6 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/0f55e80d8efc9419d83a37fa8b15e470 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/a8058069fc7156926e6890d80e04e6bc filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/1a804c7301a1d69949ab2a7faa66c2f1 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/e6f142347fa91252634df254b6f88618 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/1981728470f88259ed1f8998654a0172 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/49592ffd5d4633fa256ae0e9920831df filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/41da96eb31f5857df2f8e7c0da41b7d9 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/d3dc726d64ae4282d262021589860d86 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/52da67b4da2e58684eadaad3ab1cfa7c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/f36b83f2a27b9181d64037c2a6397bcb filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/b552def1252864be8e4388aab6d4afa8 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/fbdb7843441508891cdb58112fabb7c3 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/d39d76002a2a916c9a2d05df9751a8a1 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/eea40b3e4e75a43b4bdaef8a372d732c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/2bd4db35f96c7dc81cfd5d4b12107d9f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/bfd5b06d62565e26c6edf851315fd9b9 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/57ccc73c9b47c46d323670d6cd160d59 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/b6653240ad51193c77eb0b482af4599b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/5fe80ee08b12b88e2be8eea7eb8027d2 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/e2d9e24de945e232fda14a4dc94e16a8 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/5d4c0ded748ae39e97271d8129581a6b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/e2be54beffdc507d0d9cb6d51d1839ca filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/b57c489da78ae08fe0f8554d2c2a3e74 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/f193c781721df09e038d554bdbb950be filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/4271f2f02b270ba264ba9cf862b7473f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/e7d790bff641c7afc384b7a3538e1d98 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/a3015629bc80c29e21a3b7413ea35831 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/d954cd8d10e4e1057626733a9b4832c0 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/9423cf31f5304ca39d0cad1e2c15c6a2 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/d5387e9d3a53f20c031c68e36c6024de filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/74014576438054bbb2a2eb5ab213c1e9 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/2d0f8b9246216c14e2b81b5b04aebc67 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/41aaf88151cdc1c2adc5f512b41916e4 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/644aaecd70e0bfa033b32ff1fd2b174b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/85f8b5a3f16481751987f67751ba2218 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/92c4e1152b153ddfdf5f65f6fbce3278 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/e7c102265c760dbc35b9abbd1d7d02b8 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/2bc057f7fd66833f70f187691a559169 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/b12b1e534f759c429a2866672b8a2906 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/1d1d676ea87126cce63c552aa083bf38 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/36a495cae1ee407e0d8559b696ab4f80 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/a573635f8eee854be47b38c025ea531b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/ae65567848b96d14fb6ab436a9faee26 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/616adc2ca59c2d93b47c74f924247ab9 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/b7c9e9e74ae5db09471d9d7a0e94366b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/ac6723d7d822c0d5a5dcc65bdfb061a8 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/c6d1ea88b20107b68b1ddd4766f1bd9e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/218613bfe0df1c90e8a9e46d5be70328 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/c23ec5639e6bafbd899eec00733810cd filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/08135e35d6fee1d325515158dd12c855 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/482fa089b0fce3e31c2e7aa4355feeca filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/69996d07f2c99529519db2fea22cae96 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/7fe01396b3777d1c649f82e5cf2fb267 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/9218150f805196f26074893cc94c7d19 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/a6ddd6eb8118dc9b305a4eabec7a56ea filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/0961b9ab53cff549978be614a32e5eae filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/537f3dc8c6bb07667700d00da9141df5 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/5a31b52bb1efcf6c32cda498aae4a70d filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/ed557e6ea73f6e4df18c7c0a027b07a9 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/0f759a2ccc6311f5e38d2cc3e4bc38af filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/b2f6385b500e14b5bbfee8b149509ecf filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/6190d5a6e8107e3b9c10d0c98df5cf3c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/cd957f850e8afe0b9029f4a60ebf83be filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/2fd69da09a4a9e2532118d516e42c19b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/b2d85c6cea1b06e62babccfc19e1e47a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/11f2f18926f9d238004c5d099dc64be3 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/e6447a432bdb659b243a5ef5308b2f57 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/788d690ddd3f698ccad42befa041015a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/c57b5e88251d9485b28167e64cf33f6e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/197098672dfbd2d4278a296e2ce4063e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/80b6e48e813b6e65d76ff22b3e865c07 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_0_d2d9756f4ea98d9be959.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_2016_f2f66af7b1bf030e09df.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_3040_c7fc4d034a9a17f8c668.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_4064_e62d975a47004a0b87b4.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_5088_1a999dd354e9cacb552d.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_992_824355b2d3a467bb3a18.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_0_1f596720145f47629a86.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_0_487c9da232a04dd895aa.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_4064_08f09aceb06e4c59172f.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_5088_e00a0d045117b56c0b6c.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_992_f7294f691ff5dbc9e887.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-all/rollouts_gif_0_8bdbad86b104ab61e521.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-all/rollouts_gif_2016_d617283381c3ba4ffa74.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-all/rollouts_gif_3040_1deab0e6e9ee2422ffbc.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-all/rollouts_gif_4064_7bef146b11b7b5f9c1e2.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-all/rollouts_gif_5088_016bb4c85f0bf1211686.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-all/rollouts_gif_992_c68b64c3490ca4eb840c.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-generate/rollouts_gif_3040_1deab0e6e9ee2422ffbc.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-generate/rollouts_gif_4064_7bef146b11b7b5f9c1e2.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-generate/rollouts_gif_992_c68b64c3490ca4eb840c.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-replay/rollouts_gif_0_8bdbad86b104ab61e521.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-replay/rollouts_gif_2016_d617283381c3ba4ffa74.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-replay/rollouts_gif_5088_016bb4c85f0bf1211686.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/run-wp_0.0_alpha_0.wandb filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/36e91efea5a545989c7295460e157a8a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/431c17a3173241437f5d1d3c8844ac83 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/84e4299faa8f0c1738101d7c5d42528e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/89694b1e03245a9852a7fb339b5502e1 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/249d560b1af9e58a3ace5e526b565135 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/6ed1e9a6205df4b37473c74a2b2695e7 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/9b5ffacd56962a808779a7954865fdac filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/bf470c099961fe85aa8022ef3aa4a978 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/c3eecb32ef1539eeed2e09be830a6db6 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/30e59021206dd15612b60febffa04dd9 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/d55be9d4f5454610a1c8287d079a9bb9 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/6a7bf0005a5bb6752b5dcb19e9dce832 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/af9d357c6e8b8bf2c43d6646410956f9 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/06b12b6fdef59dbafd40ed197948b3d9 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/ac4e3bee1263dd7ab41f2fc828f85e13 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/be8bbbd819a7ee901ddbcf6fdb3a7142 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/cd89905e4e62e2b595e40ab86c32abc2 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/64f61f98cd6eca248e82768de1347ffe filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/67105c903f92bbdd84019df2832f15f0 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/04a858d6423c615365176d6b9573745f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/0f9bf86a57551992814b4526bf82fc8f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/9f3cefaf47e0993e6a01bf03a3457a45 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/204aafaa9e97f5a4f366073245a9dd74 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/f81d7b092b75728d27c71051fa16aaca filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/2653eaea8f71ebe235cff97b786e2dc0 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/7822a51da8e96b79354d7188e90b628c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/a367c55d7ce563112448d8edfff098f6 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/38e8f740ad6ed2f527f4eb9df8575081 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/5f2719ad6efa52043edd48048b18eb2e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/003ebe1c7d45f3d1eef095cb6fc4f747 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/410b76a203d615d731fb6279f9a10320 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/994696f7d15fef7b1480f5a55598fef5 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/edf5414e53554dec7d974b4daa220128 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/09050c8f4d4e2f9afb50135a1082711e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/de0c65931eba3faefeb3256bd8cf2b6e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/289c8f1b35ff5f9265dece44c44d5c60 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/bec03153b9de8b9a95e12b35cf515a98 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/a2a9172d083ebb30ab3f895855244eb2 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/aa2482ce0148df1123ea491e7eeeb3d2 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/4899e287f7a797f7fd02f8f5ab7886a6 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/53b23ebc4deb0c536120db9a8729edd0 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/250a965233d96599e9ea36b52fe8e4a9 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/58a0c3b17ee88fa9644744456b024015 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/a4ae84a1cbf83ad469477f76b3d4bbec filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/e592dac11b4e7d2361e4e06e3be53537 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/340c50da00adbf2b78f937717f6fed0e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/49eab01a75f387ac61902ebd64aaa085 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/83010ee6c93ef0d0495abd698ae9400a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/015bf18b668a13d9da9e649a9d9475db filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/adc12c61446171b0baf87e0e947072ef filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/ac7781cffe229f70f16b5c6fc48c3d8b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/b26893938da1b55e0f92dc9bed57c9ee filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/13c163c8b3bf5318245383c10b92298a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/d6e340db8f5e265b01623e12218a7143 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/971b720f884e1d8c0b72bd24bceb7e3b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/a7d64ad23e2e1c5a942544213acb72f7 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/537ff4ddb21a26186a336f5bb972e7db filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/e01ebfc8c72e60bea6f99c806b3c3671 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/692e06b628420588956601f4448b1580 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/c5c5c7a69ef431e4ee08ced12044a186 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/1a148382f42bcbcfefaa3139c6346996 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/2cea1f25bf36a33d8bc4a903680e822f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/6658d7dc1cd4d33f540a951a51ac5ae5 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/9db22c6b0638985859f444c3131a0416 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/c4e5d43bb05102bf6410167206b13566 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/e495772951c60236bebdf289f626160f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/d186f03faa35a9d784e262c8b8eae788 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/dfb0b805e2d76d2d1a0458c1e5dd5ef4 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/62dcfb9edba6daaa3acd36fa3d169745 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/dc3036e9077e98af8bebe856e9a9da92 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/431a2655b38a53b864bbab8e1bb87f1c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/5f65206b43930e4f0acac362a0dc9e1c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/459bf5d9eb4e4c9898c743827fd3bb38 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/e6f0905e86042a644bf97008cbbefc45 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/516d66f2ec20b5a9ecfb0d71d07c0cf6 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/fa25a72facaa7ee80550d244371a2a12 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/040c78899a049a65b06772a3c0ef0678 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/4b79dd02099b17942c8ea936fad18c07 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/a16a3c5dcd32e61b96495e5ea2f9a0a4 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/dc69a934a4f5788a928a344ff348f894 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/3940704a6ae6f9ec0f7910f9d9ca4ac5 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/738c35cea843469e4c123757c57981f5 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/5b95c56ae399c0f1257fe7586b342e12 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/dbbd2fc7d31537fb0ad780129a5e6fc6 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/4d34ddbaa2dd92756599b928e8d5add9 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/6884d6bf55801e033ef60c9efe7da122 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/6556f1a65c0e94a7aff2d8e523a64770 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/e852a29e92f0a60b3e5eb6015ba4e898 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/64ad4999040c3b26269a81211c1abd1c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/a84b1a74483b261d32e663dd053d417d filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/c15a3667b2f8b5add6f6f07043e5131f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/e6718bd9c5ca1c404f2c2afb3cfc162a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/5ceeba811ebbf1df4745c297b7b9ddb3 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/a7c3c160605b604b2d6e742ff5083617 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/cb45c8bb9ce20ebe548105c7f05359f4 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/db85454e735f2f28ccdd655fd369dc0d filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/3f28f47f17f1dc3c1ee748b367f26a74 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/58c1f1c8e0e982a09911188cb01d36cd filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/5d4a322d3d50aa4bb46f36a2ab41bd88 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/030d57ecf8a4da650a6ca20493b5de7b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/aa76b7fd856f6f243f9c342bb3c9fcf4 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/c3b67c68d9e3dbd4751faa7918284982 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/27d6be176dc519c7c7acd6395400193b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/7f80a83d19a74041c4dde99b46224a85 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/125e2361ace8c1cefc8431aa0cba59ee filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/88900d8673a5e97ed96139e58cf0a864 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/0c0cbb2a561255db54845314176fe8ac filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/5eefd6ee8d09fcbcda9896b099de3e0b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/125329bc054444ffa7caac0d5731b0b6 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/5b53ff438a977b7ad94d488142663bbd filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/77cd6df2513836ebf054a845a4aef8c5 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/ef2b4e1176c7e77420471209362990d6 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/2d74f39016acc212ba2fd5406065e032 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/8d864d198b1afd23001e9335be631f12 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/9245be6933980adcaf20ab85bd895536 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/e28d870952ac9e4f675633dcbcd8906b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/ba097af6f13358310d424325ff148e5d filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/f8ab6ead08c6abaae2bd3f01959b32ef filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/b6f0fed6f828f72febb2fd91b3d04ccc filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/c545edc0e85b9ceb48a0bbc0b48f3cde filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/3b2fc8d604e8553dec6d35e2678830f7 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/6ba685a70abb0c723997f4e971261742 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/25b47c9b9c8a78ba2091270a1264b33f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/35a4a5d59f509a7480190c96c5c89398 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/403bbf03935d01b542bfdb296ae2de4f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/9dad92436ad02733ea9fc741f3d73fc3 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/34109772b8581063be7a856e7a968d74 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/f1c64234b4f8bea9d89ef53e3d31bc9c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/3d76a645524883881aef5f13da0318b3 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/e7a8073aadff11c2cacd6ca5265cd363 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/a0aaaf2ab8b5d7511315ea05b76d8a14 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/b8398679fad91f9a95d8a71d6d69f16a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/390cf6ab50c7a9c56d0ac69420c1badb filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/7b81f23b96ebc55b9b8257dfb1c2df9a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/0ffea5a10eea79ffc62ae36790b53de7 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/2084d415460d422beabeb4d438e8d540 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/1952e2717c4e9fd9020df41b058137e5 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/e17cb904d17cdd98c5dedb4bbf0ca5f5 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/77e4c0b8b873d524e558dc8284973c4a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/967b785c4c49cd206b5f6f5c7e3540bf filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/503f942bd3c63100bba26a2d9d13ef99 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/dacf0f3f369f7e51ca9f13422fcdf425 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/4e333a8b96275982005cf5b0407854d5 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/c0303a2dcdcec50dd2aa94ffe15bae5b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/c190bb8e94d1bde67939583932fa4838 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/700c88a987e7c6af14760133ae5768f7 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/cc5aa590fd4b02b04acf77e999ac630d filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/09b0b7c9d701be6bc3ec3ed124ee7311 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/c2ec1a0ba842721e37f93c69fb75fe08 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/6e8415c7d3cd7e03a169f3a89df636f8 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/a02758296894c7567bce7aab8e54f89e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/44c0faaad68a46b1c606f34bf4027230 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/d706746ce429d8219a4547500af59f0c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/1c943c1912a6c228c5d1e1ac5bfd10e0 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/e0eaa98f18d08152871a6c67389b2ebb filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/3a616a2b5470b9c260f691504c537355 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/9f6bbb7f8d8e05578b030bf0c0536ffa filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/9528ebe562d9f256e014979a364879f3 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/a8425bed9234f5ab21a2963fa488a466 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/bc1230e68b718e048fdaedbc3e670089 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/870c41d4157c8fffab929ae9ce7635fa filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/e6f4f029f75b20306db29669dd41d2ea filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/d5696c803c808058975b3620646d9171 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/ea8cab2b2507dfaa380bac159650f3ba filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/acf674ac86865613f008f36802f7565e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/e01070495badaee32af83d69566bca3e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/99d618b5efb2fc7d3f4d6d6fae0baf58 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/b5f78f20996eaa79677ef6a75a2a6035 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/626c1d2f14378ef122933744a0a1938d filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/ae35325174df8db1e3c57116163a1595 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/4ba87b317bf720ff421c862ed24891d0 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/f734e89f2dde04d6460abf31736681dd filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/a6b8ff5fbe051b06ad9aa06b1d60fe73 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/dd34b9a56539a392e544aa5f12cf6e2f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/200bc68bd7777ef18492b7eb370bd25f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/478cb4fb32a3c1cc459179f087898239 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/100f5abb20c0d73347ee7ebc5ddefbe6 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/e3907512383630308ea8e1b1352de058 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/0a4413e51d4d366c5a74ad8a12a7045c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/4c86a4d6171932a8334c175ffddb8f95 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/6398c0c9b7040735420dedf11da1035d filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/ca1be08eb54daa6d8661ae846c2c603f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/3057f0149e43236cefe8611ec9d87db5 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/4c5ff490a7a87ef58ba8c50791a5cf67 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/9ee7be16b6bea157b038e277af1d7b58 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/f1d7ea393a631451032884386e1127b5 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/1d6c4015094c3bc53a3b70e469a9b841 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/a9ed4c6dcadd66afc18f8c7c5f94a0e9 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/de738fdeb23cd0022048bcd682161835 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/5985bd0cc590d125034dafb8ee8c09a8 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/a5df94198b83ebbf634c2805318aa2ed filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/095495c0d9c6bea06272c5696a26e1d1 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/eb36eb551729ca955014039eef56323a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/69b4da8bdd92066ae7383e5e81e7ae6b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/8ba3b3c6070083e4cf213cf6f831b498 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/807b746632f698bcee6590649cfebc07 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/a4ac6c28bb430e7d7d43d868cd98894e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/767212143f180cbdfabb302861c9f1ff filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/bb155c9aec4120dfedb634f3c2d60c6d filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_0_e9b0c2fbc187ce8be111.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_2016_d316c3b96ceead9ecc37.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_3040_474a966bc96cd0bdea63.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_4064_f561be9801993e334890.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_5088_39b0cce15a654761e9eb.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_992_b9240eb49ff83fe19258.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_0_0ef543c2e6a82ce9a719.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_2016_ad6e3184ad5a5974f18c.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_3040_a44991b76ff59434c6b9.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_4064_4416679b4f405fb2718f.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_5088_b1ee6f21586850948d88.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_992_665485c6d61f2dd34382.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_0_4bf97f21068441f21f0d.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_2016_c3d54e7eb89803deb72a.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_3040_7ca026a606a168bd4191.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_4064_0af9f52337a02c1fdaa9.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_5088_e51653d7a53328cbe862.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_992_38c7f3069078799c586d.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_0_0145cd3ed1d03ba4c52e.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_2016_bb9f77472c43ed2b5ce4.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_3040_c771511fda654606efa3.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_4064_97bc6a39ac75aa3a1e40.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_5088_3b8df0ac5041fd8b5630.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_992_150795aab7f0ae84791d.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-all/rollouts_gif_0_a6cb2e3f92072df15fbd.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-all/rollouts_gif_2016_a1610c3bb8d0ce3d345e.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-all/rollouts_gif_3040_193304d5f265da614c6b.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-all/rollouts_gif_4064_3da39e8f9a0fc99e101f.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-all/rollouts_gif_5088_5ac4920811574afecb73.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-all/rollouts_gif_992_16f1011b05a687326ee7.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-generate/rollouts_gif_3040_193304d5f265da614c6b.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-generate/rollouts_gif_4064_3da39e8f9a0fc99e101f.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-generate/rollouts_gif_992_16f1011b05a687326ee7.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-replay/rollouts_gif_0_a6cb2e3f92072df15fbd.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-replay/rollouts_gif_2016_a1610c3bb8d0ce3d345e.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-replay/rollouts_gif_5088_5ac4920811574afecb73.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/run-wp_1.0_alpha_1.wandb filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/db549f8304ac8952e56f96867d5995cf filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/e1643b359b36c89cbd956409fabc8d48 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/7ab8e0777196979d1e647f6fa7d9dc4c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/c25f85d05f6c3bbff803a65c40a67991 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/22e99b5c963c5f0a282339076c9ed0f0 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/3d0205761aa0c1cbfaaab19551f23acc filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/b47013dc050122f9f34545cd0d1c935c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/cefcd948f8aae6188309ac9e076aa3a0 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/3c81674da9c4c5bf8880727815ba41ff filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/d8698f42a300d264cf30bd7638b5de88 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/0cee8893a2f32e528130dba213ff3b13 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/62ab1667550e5eb666614f539c24649e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/4629dd57c1238bbcff5bd6ba81feceaa filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/6362103176fe4b833c72d74a5f16e311 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/29339e28c112d7e1853527ac452307f0 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/947b9639682d2daea6de16f498020362 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/1ca5f176cc8d89674d827e563b445bf8 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/a9c0929fbe9c29fb0274188cec70f6d6 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/54fa2fba89060229fa9f6084ad41aef9 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/e12aa63b987cde9221d1a86b13093bad filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/36aa694255f6dc0f61db0ce39bb4a4ef filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/dd6a6de760398b8d6a23113adf77258c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/1e90ad460bc7ae39a8623be2c46d6a20 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/a3c01001f0605feb13e4321cf29bbd6b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/527a4b9a7e65ad511b03e1a7b9a90e4d filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/58637e1a82a5d4c9070618cff8a060e2 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/444a94924327002ee0d7a6db0ba672d7 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/6e13acdafa3a952576ed801cb34ec2d6 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/72079c59a1eff30cfb1cb076f6332bec filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/d225e269b915f80ee6b566ddb16ace28 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/352faf90c4c126e6505fce64ce48de50 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/543897bd695f0504a303f4ac75693959 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/5b4dbe2828600cf4feb89cc31141e7eb filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/972f45bd2148cd7f8f8b058342f802a2 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/02ab3d5664c9f40eca95961c0dda2566 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/7e3073aa787478da5053d538e427dedb filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/a9eb2ec0432b8292f7681063d22f33f1 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/fc97a770d0c6ae0635d8ab0fe021a397 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/243db48a89e8995a002611b124369c2f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/299c1014f14a90e131fe5e087423b318 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/2e0db4fa97e3aad33851f45b98837acc filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/96e2db1aead93baac03e8e9990ec842f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/87fb7974add0746c27915f27c278bfc0 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/d1f45fc2b4c1b8845ea48f7cd9bc1571 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/ce7ddde8ac3da3ce60e1fdcef1a29219 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/e5f11efceb463ac6827d91267a9d4260 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/c90bee4582842a8cd8d67087509cc38e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/e00b39be644cb3c8cde260e6de14c159 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/19cb35c025c6ace0c071dc2d1d5bafc9 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/dde4376df3dc35b123a1bcad5fbbb3b9 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/594f091b9a32730f561344b43c1e3e33 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/9044cddcbff97e56cfcdce777d7c7327 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/51e2693c41cf7532f6247b2c948d9274 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/e741eb34f789546b32088a8f483dd982 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/144da8943944216ad4a02d911aea1db3 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/b1c053491931cdf44b967228b207ff72 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/688df1de52824ff4a9199f0e446aeb99 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/7486eba87e1576b7c8856f618a2e8eb9 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/00e167d3dc1caa71c22c4290637e58c7 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/902f1da199c36a20b80712c0edbca031 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/a108376e4c3e520a2faae195c6261a26 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/b973851850686ecd6ad808c5757854ae filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/020cbf2833c57f1e9f6bddb700eb97c8 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/ec846de8788d90981aece9aba9c42abf filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/36b3341a1087d26070ca7eb26bb637b4 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/538348898686c502c18b3a2e24ec8109 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/6d5e2d655cc7aca05788e437e5296e74 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/16bcc28610c142ec3a608c32d976afcd filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/4813a9c426f6563e914a267c2b1e9225 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/23e284ade8be27338355f16726454f7d filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/2c019cd394164b08f0af7aec35ec84e7 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/6df958cb57f18ada934f181cf410e9ea filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/fbe88ee3fbcb2f07a2a4ad7f399eab96 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/008b0660bdadd1e71877fd66882b427b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/0716ad50979bf94c9e6030e733b40122 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/42db52df956b8890faad93a38b7f0fd1 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/6eb3111c52ea9eeb7119b140cbcf039b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/2603ab82e0b715d9ec39a03fcec25471 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/c38056009818c69d4d080ad0a9758d76 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/0f56ea346ffd7f60e4feb0d268a41ea6 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/3910ca9f0439b6594f0e10c00404bad6 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/3e61c82b3acc0037b26e9781fb87d7c3 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/46fab3e374bb48acde8b4bd6fa4178aa filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/be912c84ee305dd7bcda5bdc0298c332 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/ee0429b5bf304a2d8c0ac75786eac2e9 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/0528e2cdc1e56b2c8b32d097cac2489b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/1afed5203ef18cb17509f62372f9c357 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/c6f352e4d61860259c1f147f4793e94d filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/74171d0253936dca17d49ebad2249880 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/96e041eedb2463cc4a1806b2ab07f747 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/bb4ff882e58e173106dd378aa9154e39 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/018626e588efb60dded9660bb929cb67 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/b9ce01469555dc3e0e559af11c8ed9a3 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/5726137db3bb1c04ef4f998d0d24cedd filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/8652d5e6b532d41f97e651b0b2662ea0 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/22eefd72c8be679b224acd5cceb4eb0a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/9436a7d098136283940d90d4f09a7dc8 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/ba66f1abb492f032809466238a7fa8aa filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/e3d6457ed82238e7ff61153abb23df4b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/5f1287abc3d6ed2dab2a6bd8f9fa98c5 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/79a99cdb494d1c9cdc798a01581c2b5e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/7f2df274782b6861bee8175167efdd76 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/96ed3b0f0f3f680bd9ea22a36121ad37 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/d299cb6111f79b80df20c55061e508c4 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/32b980de126bb7648c7ad53e756e694c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/d46befa50b74038fb404c5ce826b124b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/0aa14bc830365186e0a1df4a118645d0 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/3d579eed0ed6643a4e5082039cd23521 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/51e53460ef81401e8b75056d861685f4 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/7a25ab5e90d5243b7105ad17e31f7c13 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/7d427dc5b2ca2c9df100a2cb920f7e44 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/b4285486ca8399f139309695e526ab74 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/3ccd1c4698560be2c6e8f8ca6d914fdc filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/4ee6e40ab01392089555a042b99894ae filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/9d53a7affb45e9975657dd2f540f8964 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/c279d830c44e48ed37f1e393137418e4 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/3c95a5a2a09e5395d74a755930861275 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/4a6e0dc2cbed23275d476ad14679e270 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/6014935216cf6cfd35ef9f3c3759602d filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/39875efee752abfd8e50082231226c27 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/c109446946521f6e8e5a9a8079d79517 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/79dff3627a27fb19abafccd358fad496 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/fc1772e31cd66a3bbd167429554d8079 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/2e24e5509071c36ede8a0dc7f19d5fc7 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/0cb061330dd19ab823306623ea6fdf83 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/567a268f842da4a82bd80b5e2861123c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/65464fdeae2a520df3aad2c2518d53a6 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/799e6dfc1cac6f911b88d8a872bf5971 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/eafe136ca741896b388dac0c8b635b85 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/adac7c6c762224cd4b26832da5f6bfab filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/298145a7a01972f925ed3500c4991bb8 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/413940acacce0d24ea8e5a3ad34f4375 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/1fdbaf290fa767a41b25c1a1defcdd4a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/ce49e86dbd371a81b06cb0e0d4513f3e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/035dbbb6b9ba5e8e390953ef84a7dbbd filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/50c850b63be8cc3626fa9458d2d54caa filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/136cbfbb520725abaa6e22650950ca7e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/77c8e1e953dbef2c18104d8cf83f4d30 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/952154972d51f63f6d28a7482033c55a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/c9bb6b91cd7153c6dcb8bce5fc7f6c3e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/8b254d17a83d04d047a00d2472b14106 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/b84e8c64ad1aa000624eecee725967da filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/5e4e84f08ee12292b4936095df2d0335 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/f9cd12020691dd6da865d71d10c051ee filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/6ef8bd80a24b277bb9c93d361b5bc0c6 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/e2f1132ae4dd70ae673d0c819170fe2e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/7fd6c03e550759cab1e958f47c134c65 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/c74c1d8609f11d655f8b48ff55f19932 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/57daa6b736cf4e3f44bc3301a66c422f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/63cbdacd1f822a430be99f53f8b379cc filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/37a18c0c74ab3278491165c31380c1bd filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/3f0528384d0e2d5c631fa0d8cb07fedd filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/42e6be7be6fb77d162af4b54acef4936 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/62242f3be5a8a0f977b70eafdbbcc671 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/01834ef899ccafd3fd95b8cc60d04de8 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/af84fe9183b46daf53b3f8e19f093bfb filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/3ee739ddb2a1b29b4e45583605e91d4c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/f3d280984005216109679f3cb670008e filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/aeba78c6859cb9d4f58a90b278081fcf filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/d263679d2b9e4b6d1504cfbceedb35ec filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/23d23a1e895550dca61547d08b9ec5ee filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/512e0d2480713d4df17bb3bbc9a4fe51 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/8faf62cb7f4a628acf1a4831c813f9ae filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/79adb5e32f7628bb98a3c81b96316d0b filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/82596a3412d7f37e3123c1a5a389111c filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/189079a229bd2f7af98c739bdb8a69ac filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/54e87626e3dc373781e4576985cda789 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/76154ac859f612b63547dd250d0cd7f5 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/985925908c958e4e0f5403f382451022 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/81e033a3623f932f114fd9cc23446837 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/d00cb274a39eb47104f57d9ef8862dfa filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/ae6475a569be62c236a65d8abb006884 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/bcbcb51634b513b8adb23639fc13e398 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/10647960aef0ed0acb77d17e9c2f854f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/ff944bd4fb5b0801374030f18f84dfd0 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/a3600783e5970dd07039414f0dea70d7 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/bc732652adfa7fe4b0b16b05a7ecf5b1 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/294492af37123877e8cd291154459b9a filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/eb74b21e5801c5630ce3a406e02734c7 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/a15bd7fda036ecd345ceecbf441af41f filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/f2eda4c1918af53d7177debcc6450042 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/88c00eaa68751a1db4883553ede59f82 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/eb771b477137704a3b2600511364ef0d filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/a83194b17cc55f752f0f4efb2ac3f072 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/fcfd2cd5c23f83fc280d013563819bc7 filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_0_2b3fc2f7ee28cc608858.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_2016_950008b781fcaf7b9f05.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_3040_6a9f6ba0ad5bbaedc7e1.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_4064_60b8de8fbb57e41ea5ee.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_5088_949a784c4acec1cebdc6.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_992_ed151ca7a2d60364e9ff.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_0_70c9cfc8d51724e65056.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_2016_5f538c3d498acae44865.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_3040_f223b4590f6d97f13770.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_4064_232b38e52a22d511bbf4.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_5088_6995668c199776ef24f5.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_992_f775e15a53e01e0be2ed.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_0_b75f456da959a3e97098.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_2016_fbb8a8a9b4216197273a.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_3040_d689abf40b9f27332b3e.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_4064_42ab2ea3b07646e983b3.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_5088_2401e16bcdff1cb057e2.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_992_79f267376e951d66e220.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-all/rollouts_gif_0_de0ca76414e4221f684c.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-all/rollouts_gif_2016_27833df9519347170912.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-all/rollouts_gif_3040_6d081750b2a0636a7633.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-all/rollouts_gif_4064_14950c5dc884e09b130d.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-all/rollouts_gif_5088_b3d811352e422cf93cc6.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-all/rollouts_gif_992_4b91e1cbe9b6c9e185f4.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-generate/rollouts_gif_3040_6d081750b2a0636a7633.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-generate/rollouts_gif_4064_14950c5dc884e09b130d.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-generate/rollouts_gif_992_4b91e1cbe9b6c9e185f4.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-replay/rollouts_gif_0_de0ca76414e4221f684c.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-replay/rollouts_gif_2016_27833df9519347170912.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-replay/rollouts_gif_5088_b3d811352e422cf93cc6.gif filter=lfs diff=lfs merge=lfs -text
+wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/run-wp_1.0_alpha_0.wandb filter=lfs diff=lfs merge=lfs -text
diff --git a/wall-prob-tree-13x13/debug-internal.log b/wall-prob-tree-13x13/debug-internal.log
new file mode 100644
index 0000000000000000000000000000000000000000..f08029548b0f038a9cb5e4bd7e8536b90024409d
--- /dev/null
+++ b/wall-prob-tree-13x13/debug-internal.log
@@ -0,0 +1,1167 @@
+{"time":"2025-08-07T18:02:11.662606973Z","level":"INFO","msg":"stream: starting","core version":"0.21.0"}
+{"time":"2025-08-07T18:02:12.019688846Z","level":"INFO","msg":"stream: created new stream","id":"wp_1.0_alpha_0"}
+{"time":"2025-08-07T18:02:12.019782955Z","level":"INFO","msg":"stream: started","id":"wp_1.0_alpha_0"}
+{"time":"2025-08-07T18:02:12.019879857Z","level":"INFO","msg":"writer: Do: started","stream_id":"wp_1.0_alpha_0"}
+{"time":"2025-08-07T18:02:12.019886042Z","level":"INFO","msg":"sender: started","stream_id":"wp_1.0_alpha_0"}
+{"time":"2025-08-07T18:02:12.019948055Z","level":"INFO","msg":"handler: started","stream_id":"wp_1.0_alpha_0"}
+{"time":"2025-08-07T18:58:50.262815671Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0, Name: checkpoints/1920/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1920/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=119fcb55d98110d4a2000064f38c13badb037e72676efb1f4974b6c091cf613b2e42e2ed1a8acb75eab518cc0d56b2709d3577bb8558a562c6b3e732bbf061c1b6c9baa341a9b098e832e6577c734ef66dc6c101622a42825c5a754ae0a55de0ea131ce8e8b3f525db9ca55cd1df6b7298f74fecc850c3abd13ccf999c045b958c180daf17ccdf72b930ae7e8934136f489fc2490c1358c30c3e81f0b0272540e2cba7e1db1edb3f0c15aa99fb8ca1075450703072a5dc4979c0eef11794727102fa6f64c9127d95b52f70c7544d1d751714fc45ca4b07ce1725c10106b053dcf1e3b51b639b2577859abd9b2aadebb37d23da7b41364346b5a40e77e736ecf6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.262913168Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/array_metadatas, Name: checkpoints/1024/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1024/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=67519dc32bf9e1708d7d36b5fee60e300c4de9d75d40bfa83bf954e81d8ebf1c7f7085fa152456e7b47ead4620099b80c3b4da4b6f2e39b0b259446b9bfeccb30e4438668e07cb0bfd85a044eb117682277cf225dd00e8460c79e81779c641688c86e22849f04bb1ee0c16a78026ced16b36de5be4d542b7fcedb06b5b4da6710b572d94be0b871af47b8890ae182f8df81c1dbac3aa095ecf1809f08aba0312d6226b6023c4997a451da814e4cbf8e1325d604e544662fcd8147d8e5175429f3d931be9fd1364638fa7eeaed52e702c2178308eb9fba0870dfaeb0452b64bbf9092ea3b94ca385351b6d18cb1109ae7dc4d6f3099353ecd72eca45f6375ad83&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.262974561Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/array_metadatas, Name: checkpoints/4416/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4416/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=07d5f714800fd6ee96d9e0631a8b0d6be760cb8e1c06b01bb4f7ef253c3c9d89c32ee7a1435a1596ca09df0f12eb03430d9f33f53b776a6911bc9ac5367238a367fa636131055e891d34dab64405c57f0eeafd658cc527be71cf2a2f81c8a4b6d8af551bdd58225e8e6ef218bd1345d6395188013670c1ba5abd0fd3b5aa30b3af08fdcf5a76dd0a7cfa16ebbb244f066e53f2a4701ebc4d03d46bde820acb43c589f402f000b909b148eace171553151142717deae48b436533fc959ad784242c44fe1f06bbefd203b5e2c5d1fd02fc259a3a3220af09ffd136b64e014c88385cddaae287abe9dc42d6891df20ae748a6219f9267e3bdfe6340fa5014085897&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263025934Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d, Name: checkpoints/2880/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2880/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0d60cb41a3878377cd4ba90fb6cd74394cb000f8e24287670cfb4d5bfd8346f4a2fab1ccd9e577662c1da03ef54525f1934d304334fd98953968edd1c23d478149c02ea8182c154e806ae3136fc8146d0f4306e75aa676b0b8a3583aae1b3d084178921dbff627238a36c5eacd610e9a78cce9f8b15ad067628cde4c713f4e859f9d5b72c88ae24268ea830261de4e80eeb1fc39a72bb6d5302e58cb2a10e87c888235055c08ed0161373c5f9b795b2e776df2ac20dc58bf2f26cd68bce9cb9400b24d186f5c110482490757b3b6004bb164227fda6f91007afccb56fa717a140935f027b7ada7ffebc4f4529b9d215d28c4eb8d44d5f1ad34a3840d5f0d0d90&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263049073Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224, Name: checkpoints/4224, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4224?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b8f7530ef87ae4faf61ccbdd9920008b5b8677c947e21790ec42e81e88df114005eb44af489b3a5430f8f733ec7b8feffb830c8b7a5f9a899fecbb7532fc8704952ae3b110727ce0a08f811237fbec0a26cb45bf1e1c8871c10f4a257038f7ca3299fb533fb2f5e83b2c9f34da268c97822e0922ca6559eee2839463b2cf65790a389732fac1b7889470da620f37c51f75c8fba6542a3e450b236bb9243f9c05fd852b0193b47e48af9ac51aad3c963b8269a88701ae990e9e772967834b29060f14c81b2fe120baa8fd6d9118ae859b1ea47f19f9992e37e316d473e16f9e0311afbfff0737a78fd1b17f1b17a2e7ce0c2cec5aec035b8c5091731ad21ca201&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263066941Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/array_metadatas, Name: checkpoints/4672/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4672/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a2db75ada3f33505396dc35ff427c1f5c835fe56e58f0d5f6150c91bad734c0664c9cae681fd157c4db520d4368a7192e507005823e18102c2c8e157a29763da6c0cee04f548002389c25f38a21c06a535203fe3119d70dd7b4d752fd23fc32e0456f70ba8ba1a1b8f66e764dbdd566bebadf8ae03cec872d510da11c05888d18462ebf906ed1c25b621da18306a09b5963c64ae419b77f5752d5671e8c3fee0db3dd31308ce797b67cd7aa19acc706287c989a24b53cab116ef48cf320668e9e539d20524b63e31180c0213df37bbfc841f9a4aed69d5ee4885024101cb5cb9181ac9b5123c527bff2b92e80174b7ba8ce677cb7cd49a6cc67cd0a717044258&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263108838Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/array_metadatas, Name: checkpoints/384/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/384/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=70b406f72d1257ba03ba053f811b2f977606d5962a7814c3e9e6779a1232d1c8c6123c79355e0fcd271d62fcb4ffabb004237b842a7b1f8e9ee50505827418c2b8c5a25dba3b45bc59b53f2cb3d3cd7b8b0f67491159522c58378b7d66ae8d57257cd59b9dbee835145502268a40d4f3d4f84b2e7171ccc910294c977fe3c5a0f80ce800aaf59095b593831becc716a288714bb006862bc8f15cc8540d7eb883aff25c7578ac49f2cbb8abab6f1a1a51a95e5b3cc967af223d2dc6c845b64dbcb81a45c02f4b62d0c153cf0bc47b4cb42b5c8ef2cdf790289ce9d97359aacaf4b8dcf13f11d6c5ed9bb143a130c3d2f2461ba9f4a91434a9f7c99d9c2b086b9e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263144365Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0, Name: checkpoints/2624/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2624/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7f7c7ea4eaa6a997d2598c0b7ed1047053b35f827b4e554a746be712c75e511a26ff3cc480a89914ddc8b62d805edc7de1b65f0482209b833f818b3298472498c5d1a5333ee2b4c4e462d2561d69653272493ea74c8454449198b7b5e64f1fa26c985f9426a2ae99f8991f362081c66ec51e4b4b865ec91a9f53a6d35eb99ccb3dd1d46d08ef97a2984cfeb4f2187489955141c1d75689647420e95b37622e851413f22b9839920c2d3b6a5bfdaf282eaa50ca193ad0c97c8b66e7c13de36106b49d8611217cf3449d2e7219764ed0fcd8956feeffff1f965320412a2ee63570e8c641eb97d1e7de4bb61bb7251bae2bd17d0fb1db2d8823ca5004c83117f384&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263164804Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640, Name: checkpoints/640, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/640?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=413a64c23345e874fdb78a8bcf878d616c25d42d5d2ac070bc481c8fab6be103767f7e8b3c72e28c834d2e548c8f83d524fbcea9ffe580c7936cd40fc02e9ecb6ebfb5b9ffd725a50c69e43d267f4d7808bdd16247c4187b0b435edc494f8e48821ca7700d871c3c14206307414eadc76612a210644316ebf3a9745eafe5a749564a343b33e21bcf23ef1ed85ce7f357cdd99384ef3518acbffe300816b6d5c135451f326fc53c19706c8c47c02e951cb8536c797a3a2d811a74dc4ad7d667c485c609607fb490f695d5c576870314f20e72f644a3959933b524dbe22d72259bb2803108920c0948bb0beec8f0e2d65b35ec4a8ab8862796dc72039b0640cebe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263219037Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default, Name: checkpoints/1600/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1600/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=437c69df83f5581b62ae20a552b8c376e14c5304453037a35487e5d35f58a4d0aece2acbac2eb1154e23cb03a82a5f15a7667393ab951f62660216f53a809838aac1c3695ddcd540d84e43291a886c7002e2b1c9567281377e3979c7c0e1a328d3530e215cdb81d37db85ba67a692eeaa1a67c02363ad55e6564049ee4f2845a3f1ab73d7016a582a12ed79876a64a01de561ed44e21d04fe787bc68e2d3ac00d4e68b3067194189890fc062c32ac800b0ccf4c081d84d4edcb977b792e9b1e144f1f638f7a963eb669b82a2b4e8481620231e05644fa464f01d75725450d23a187ebe0d091b659cfbc5be6f3333a5666fa349a8119d2cde6ab90f9b698f19e2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263271509Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0, Name: checkpoints/192/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/192/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=330f233f5804a3f1f69dc7507117040d01b10455bed76d052c691e8d27c8460a4f090b86455fa3e9a4c256a3cc7bf5b53ac1fb04679ed433f3bddee1e214a6cb8ea88a4e62f6ed0d67a06c713753c5961f3cf57d133b3024c684154910da0fe08148666c3d136859d75b3dcfe8fd568ecd1903d37b35833c25b7724551635a671a2980c239068bc8cff17381d3f361f0f101a11347420f14d84592b18b5f5b4efb2efcb06939f57ea8923fc2a9f5479005b9430c252ede985832770c41d53c1ffaebda0fe7abb01b05a39b534a89120c7ba225eb46684454594146048418d61dbbea3e520c3ac0e647ae8167741880a2603103688497eccc419e9411766ea7fa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263348993Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/d, Name: checkpoints/6016/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6016/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2dddb7560fa5d4169ef788683afe43cdf8df1a3f3c49f474e474a0b6561b3b718ea7e4a79fbce88fecac074cb22d1ecfc6e6ad650ddff5ceb09057148c3c0c4472350342e9349ab69ac9e6283474b410cde5c5d69ab90c71dbe8698442fcf7f985068bf6507a4341c0d82235be344ebca0bb41b98e77de3d97fb0b0bb2540395d8af61ec8816003552c67424f771089cbdf340635315c80d60f974937c5452199d31b000a37a1aad80c8ccd441792cf974950059f0d62b4bbd58a8ac5328261d58b535be869606b32400ae5c74f251e8f489ae9ee6c81e91581e341fbab809febd5a0373c2075cfd345fab2a0582ec367b6043d263f5d062e4314689e00b2b4b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263232879Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default, Name: checkpoints/1984/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1984/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4bdc361490792f603c3fc6b636d9fdd233811e76dc85bf7e45c0361659e44f77496d81e7b2451cd3cbd8183b1c5c5e3c54eb5e58155543f74be6af85304fac9b396f8ad997c5b48134302107235d8d0c1c841bab331c41628d72e5e158733acf01e36919f87a5f9ee0a300855d4c5c7cf5cf59681c6d52ca006f61e664102fb19f58d8987d7231f44bd913b46e6a590d299268e9f5de554a46b74c28c1a26ddac123cbc67109ac874bb24f33ad4cf558b25c78dbcbc2df2f25389e70b3e8617bcfd34e38f942be8a0686c20d73ae844f7906c8f261acc0614c8e829e709abd9f3a4372718d7914052da0b9c3d3d8e41af43750a00aa117a7c495b062da5b6f8a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.26298745Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default, Name: checkpoints/5312/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5312/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=403a77b68396130a3e7a0a2155f7d06de5385d1e79bfefe0fdd50ce7ddf64f478c08f1e562c356850629a3c514c8cbc16eb2d85247057770bce9511f07198aff6a80717c3751061959c7db1f882d1159eeddd5af65bf313e3cf0d6a46c713172f75ab56b1dbb2828af0c9e20981916dcc1928b65e1cebd4f2568a6fde3fb48aa0b685112f5ac9690d472a8487d4dc92b70d9cbfbe742486bcabb2bd0935898538c75bf497b82b9f012fae6ddf06c62c37ee13ef8de88be5d7623e17f88511aa7d640884374cc6f192ed45a76412b6159bbb9c619c3db8aaa02c407522b81c2aa57199ef8601f18b8dbb9b8138905051149c098680868dc965c90b47696ccbb08&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263253495Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/d, Name: checkpoints/2304/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2304/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a5006bdcd2da0add1c596fd2676bddbf6ff60c645c8a89d02a9744641026c725e89ca310af5a80508dff265491305c872a5f755ca33608b5be13bcdf7f09c8cf274724bcfa40eefbd994f63a795a7a96959cbfdc347044c53caba5470770eaeb525aed5f7746126c4fbe2faefced6614a8c3a999dd0bc1e357e881346605c1da408ff4165853db3e40119176c95b05bbce7ff57ce89336ab6ae0027b8645baf8023bafe4cce52a249de74bd2b0d511081095fb6d8de368f288e37b7314eefc55e24fd86f731f6738f6e311d7c3ab2267dbbbd8d201491fc540fc2585988cd6fe3aae244c92c3b078b4f1594508525892547a6d7ca2d519f39189cbee35c1d434&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263019769Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default, Name: checkpoints/5120/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5120/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b94ec295aa5279fc71f9aab943df38e344b94abcf9f8f3babb689399fa4c7d7d3db8872d9e3a9bf1d7c0cb8e01f1c710fa8e7645647c12390fb8710eed5a584f2a49aca9b1abe0577c86200670763600c42ec493f740ff134040abcb800724f5d17b7f7169ea2c04b2a55f1897a0315c9e6a2cf6467d05dfe9c12af6262f1dabac8edcf5473a6bc7931976d6146b9959b6e7f34f94f24046e88a57baa587de144a56ad978ede59e9c0258559f4f9c3da7c1d054c12444732392efae747059aa0b39517fd017b949033692aec9d965754e296df391cdc29ec8db02cdd497f0cd98733e2caee4ef996a5aeba2ec3ddcab743454361793ff6ea180a62bacfcdf7fb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263525591Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0, Name: checkpoints/2048/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2048/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2bf9bd04beee3deef9eb3c0ccc0a0961736981c1e51c2a08fa81f5aaf606201cdd86dac27358f1a9c9bda7e281253550b472cc2d8cd89707196cb93646369e49fc26db66c847a5d5373ca5a7789cb3fa367acea3eb449c55592b7ce8f54ba3b7de2f99ad9a3eef86e40e9370066fdeacea79ca877e2337d871d93e387c70faf6659107e1d9f70af9013165e326598841c2c3610822110249a8d7da180bbe1e2dd3abea95682280467e467141f7673c1e6aaeb26afc05f6a71962fa83246af6dacaa809d1ca8593c776840bb79de2287c80b017f8177e350d676c10a2bda5b794338ac8c3dba83c73fa9ddf1df51b418784bb106472cfcd9eac14cfe518dd27a7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264242626Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0, Name: checkpoints/4480/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4480/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=32e9785417d25b540fafcc4f795cb1d9fae3ba3119693ae3a9e2b27d4247bed9c5d5260d8cdd536ca192e3ee4933eaa7556655657bb1ef20e311377cab1c8d52079b49502c765766c1a09fe48bf719f57afd3def2573528cb065cd44a37d851ff64759f8b1c79e99bf2bb25f7ce5e234d663d9167ae0e5265164266f90a75d069caa88968b190f7c233b4cdfaa844d4ed19e0c0bc01071caf73b0166958a6700ca90e307519ec7c999bc814e06b52674eea27349e51a08b340bb43412798d1ea6568472aefcc5afda8df04f7f9f838c721df2b1be82be7634b04dc7431ea123cdacb74607558dc41674e4a706b72a487c161f9ca8f536d1d8ab74db9621dc09a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264031477Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default, Name: checkpoints/5504/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5504/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a7f5835fa83b55cdf2ecb37716f0655047463c40ab87789d8ea5d8da732ee7bbc238340330c5774ff9a942507da978eb9d02aec54e02e043066b75ea7eb26ecca884038bc7a313c946b2aaba4688ce6740dfd7933b54c0e227d54382b33a1e8c02535e047fe7099f1f1dd635803a5324c881350a96ca2571bda99b722d1ab376d283892efa556e1aaaddb28b9b3eeabaf420717f7b640031ece1fe9d1a39e84a4dc4e95fdd514c25210ea379042a1cd943c415f3fd96b94eaf5840596ee00f60c99d9e6685f8d713b521892bb5ba86b746a880c7437a0d92f4167557c71b9d325c277c91fb592a9361a12034acf5a1c53b99be50c4dbccdec90bf346e11c709c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263104433Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/array_metadatas, Name: checkpoints/5120/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5120/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4225a990863d4d9ff203364cee1b90e6ece2661fc5c44e40a43f32501f5ff7e1c83fb80f5b789f31aeb2f84ff65ad44a390fed704c168184ed020da3748079e1675a4078f811db1a70ad083ad024a8c3b612c7e67c145106f872840c8d82c8a284132855de9c29debc543a8911247ef97a1eb7d0386cb62bd7221bf4f606e980efce2313d37304b4fb56258fc45de4d5f92c22150789aeb52525b0604d1375598cde14721c6671d6cbd11642576b2635a57ca29548a0d58468fe706f4649050f31af30ec1f6262c61ac1d6b0ef91e1e110ad42b39707ae340ab3c0ed425ef909c95b132c40ad68cf8c636708f0dc278b98599a2d78c9e61ac52245fea6eb7efc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.262942267Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d, Name: checkpoints/5632/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5632/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7536ee84f76023fbfe12885f64b1eca45fcbde0584c65f8bd463552ebf75b92f9a290a25eee45e26004233096e57a9505086a4355eccffc29f5561d0e4b669c9543897a991cb1d27e101a97de4db86b61f836eb10596810ba0fa0c85e095c71cfc46f128ddd851a41ab150406131515b783f5a38bf13893892109e3f9e6ec1697adb3e925b0a055b67791af3003b933c3d859db2dd2e0da557a184bb3500b9dbac37b9cd305f33600cc73cd9226777bcc95275e57bffdb75faf49308b9ea1e9e2859b84b0f5c7426d623606affc8ca0d7f21f25085c5f0f5a4d4cd6d500f3aeca0861a38163c2e000bb8d2dbe45a3fe32eeded07ea7754f3acbe0eb91a15e9a4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.262880144Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/array_metadatas, Name: checkpoints/4224/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4224/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=52943b33c4b9bba292632c10a7cc56707827a61308249b3201ead44aa1befdce63bef882335ebcbe86114e1838657aa383aeb52a436c6e249ea58cb4cc0f16ba5d1ec7c2f0c86e1d45d02c6843968475a82e7df07151a45e3a61cbd5880767f3ff062774fa4f28476b12d5c95d60cef6026ec33c63e5d34f047c30181182b4ef42d35f8479f02af5598ce01459737f99a953a07d30f78157867d059b0d002be19c6a0a7172adcb5a78a7f37e72527d170c846cb36d584e84a077520913d98451a90d4c4ef82444e6cbddb672ce34fc3a9d44291e9acbd64ea2741dacc0a8a3984a223f35a6f21fb8231447857266eeaf95ec1ec05af348aacbab833acdb37c25&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263527655Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/array_metadatas, Name: checkpoints/64/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/64/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3d761e76b5d584fb27b36fe87bc0a5f3ab1d232b878017d08123fcb6ddd0e1380b0b54ff3fe3ae42ee6de547dcadc6081751833e840d0aa64a2086457955ecc2c378988cadcbca152dd801e818ba31e929f416a54fcab485af8a302634677b380c390e7ea8d91de948813067dfb2e2a76c92261f0190e518b9be7db15e81248673447189c6cab563478e62d35d876e1b8e9e85960a67c0b632e318887801defce6546837fcb4890226e2fc1f698614406d452d9696313d03dd9b21864ee52dc09d4201c98f11b060cd998563da1e67c408a40369a70e1b8c5ed5a45d3f1f39f10d49c631f80a6fec158027fe48d44f7db2ca43bd1a47fc58c69902ce7e852c6d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264399372Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/d, Name: checkpoints/1728/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1728/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=162555472d845311893529c31c80b7c6d837552275251b94be44956d29fdb123f45ebb87c9fab01852d2b5b91d4bc79f3bfad1b50335396370ce12012786344df15a09fc7678cc8d810153a2c412fbc0b9627f90840de9538aa821476c30b414799170ff34c0eb8875cf765dc9c73a4dbc7d649a367bb2aab32169eeb282d896e8b858d76ed48e555177763a53409ca195e5f0880c4b36b5914cdaeaee34fcc66fcbbe48f5224dd68d268b052910ff8194807d20817653cdad4dfd706850e8472569f87e35a7a8a31117b9ff017c5464a5e52e5c06748622012c4ab13e771bfbab0605fb9e807e190f04257cc168c0bc84f433d9a2a4bd2ceb6c5d0dcc7320bb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.26295168Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928, Name: checkpoints/4928, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4928?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=261a18b071c73a60f6471b233ddd5970a2e72d40d286b9a72e3a07698003a36329756474da5be488cd2ea91ae517676be875ebd6e6d4cc8e1203e2523b314bd345e73944ea53156f8d7f804220f5b8ebb12570433895b8d354a246975ead219466f7f5ee8a5e820939b65a06b537c0bf8c974e8e27c2d88deab2c2d840d68f6c8c9e96ae847b3f12b9a604956d4214c0e20fbf1ff9db29caec3438ebbb3b040ecaed7bf7aa93e1f6f087a837a9d7cb717fcb2dae054051f39bc45ce5ad5ffe50fd1170b90ea8bbdc505244fc6e8c0096e47285fc3274dac8adb3c7bf7f89b339563e9933fb263d211988e4314c172d51282f4d93c897af845ff557505f5084a4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.262965575Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d, Name: checkpoints/448/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/448/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=26a5de71a18f14e55c33bd8175739b0133745f965e667d027fb16ec6ec47fae605d6e7fc7ed895ba67b1c7967dd6e7738d93cb6f134d264aff21eaeb313b48fb272542cbba215758044cdc2503f687e28609662fc368d62f740109930159afa99dde4838d7e0868a190012bd98880b9e311385d3fb33a41c25700daac1e35c274d882bdff575f478d0ae88be99309a5b2cbeff5cdab0d14f7a84bfbf1e4b41b7b21c2ab01c071bd80330d6791f8833fab689797f23c01fa8e936cb89deb87a2110fd26f93a2d337407dd601a701b5ed8b1ea98b935d56db64b2827786a9571340da13d5998a2e2e6fb86c97d56933ef3f96546784ace24a9d91b43f2dc9fb8fa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.262996598Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d, Name: checkpoints/5120/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5120/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=105b5c414aa7530a679354ef6a10b80f29978489f31ec453740988ad791850da9e0f131e720ed22faa9befa04923304b80b278dde4eb87e847e9e22f69e82aa7402311a68856035e06c4ec0e3cc85419cb8919c3550b65a7276c1ef2d28d734f44f0bf756cbf126501614583c24b03f03efc7b6d2097015ea2f8d80be9326835e7dcc8b9560234b3d02f6b4c284417a4a63e5b392ece9a92db67c19dc3c70d8d32058e53d8d9a6a7e10bb2002ff451fd741a622da8a4cefee110c329a49a36c3c0fb78a6866e830bef3439bbc7f3ba17b6bf9e67daa11ca922f9827169cf8604f9a537cb96f69f0292e86061528775673392090b5d4e2965a4abfd4360ebbca0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263110885Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/array_metadatas, Name: checkpoints/2880/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2880/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=07a62995ae81bfccaec5795379f94849356b6b484976eb8ccda2e6993900b9d9fa83fe57c6af9937d71681f98bd5d52f2ef039495322d889fc9576932e3f98340664f889a014a5ab1d96b187f5a1ad5678a2e7a4b4ce8034b56a3c403c3e8ce94edef0fa5b4be1248a958d9d5ca504230b08e94f078bd8687dd3cc8583c84875aaeeb0bfc77f1bdcff22854cc08430d98d29bcf8e1121ef419f4b49497322198e5533e38afdec59f6f8fdcd47ec3556da6a25fe7b1fd93e74369ce7d5f2ef4d1d0919ec6db7166e3b47f9b44611bcd6da3a2499ee0a41e157daf53c7477f1bb829f89b8f06cf774dd13407e97d1878a2f370a32cacc5870527c9c82e84876fbf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263121072Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default, Name: checkpoints/832/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/832/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8746c02ccfd9075413c13f46183a4140334b3e72a874a5a2cc40b3abdf19cea0677a527d0cb50cda7105fde146e1c557b40e823f92c3a4838b81466d8a16a96471a9b2132ea639bb8fd99c5ffa202295fc81dfebd61ef17183ea03f9c413b2aeff98194449f8a23b24cb4f0eec0438a94b52cdf49764612438e9b7e37fc6757a9a970a6fc9c913ecebab0cc777e37d1b6676bb64dea458ce9e169b826a61e60d8c761373fd6c57263c5927f1ee5c2404de31cf6cc22559abdcde358b52869e4b48b6b8f9086c494440df1097e3dfd6b70b77d1dc2ec9ac65665ab0300e0d9942e7934eb39dc0de8baed9cd722f8cb5408ca058e04b25c7aefed5545d8f5fcdaf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263203413Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/array_metadatas, Name: checkpoints/4800/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4800/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5cb560943c4dc130db38d1bb9ffff9b264c57fddea53b7df92451ed478155bc2a84f009804b5f6500ed7f4a846729e43a3da3f73b8793caa84923ecac30a389461bf5743be2997120eec565a6ace4bffeadf6d73bff2f95a4a595819cf6c022ae4b5a3441f9173ca02130fda06eb5ea11c0ba6264a27cbcf2dbd2e678931b83c41166c371c737f4019a7ee3515f750cd383b8c5c0e770e60b17c1fbe5df848d440742f65c1293e4c1779a34b91fbcdd453a0524cb5729277d9d2f305a6c582a9194ece30fd77b6b13d67e7850d9cfd580d8777b36325b654976bca2f616e0677685b2f1301620566814b3750a0d0d82aba0c8e73acabbc94bd4ce92189291bc1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263283542Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0, Name: checkpoints/1472/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1472/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=05df0e7d5d500020785d7b66a7b84dcde8523adb54c8658432a03c6a81801ec97456a88743730a686314533e97d43a13bb2b1f2b98d732f0c9d29ff45b36801d2863462b7c6fd8ec48db23e117fbdd63fe64844cafdc7620b51c96057ff7e27361a0cae06f73f9fdcfe8c9f46482c09fec78e6a4cf5ca5a3fe32d63b5001b110152d2a1c7bcf9252844e6f6c41e9ee5b80ad00664a82b01153eaf63679ca7e30e6a1f799b24e3ac50f3c81436c3974fa05c962d27e77ba3f0ef77f8ebf81f2d5395318cad28e53177761d27ad10775f1954f177702199b06c8d796d115889d87377e77d7c9183738d63050c9af7d7fc70be6eec132d6e9dd6bed452ea587ff97&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263272058Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/array_metadatas, Name: checkpoints/3136/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3136/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=01f18f11c4e7f4a0babdb0cd87305306720ca9cacefa8579b78c82d368e125db21b7a115c8bc96eee6132f2fbcdd68da68e7511c450fae0e104e5da75cb1f77f54b6ceefc650a24e30f1e27173d15abf1557632b6e74eab8f1663f8e970dd7d0a50b0b0e0aa7e8af8081718d367e5c020ee4b9d5785e8af5cb4e9fd91979385e58c729cd66a0b6f8b3a16414656056f82aad9608068febd6f229e72c8d522ea302959b71ae8e36fe540c73fce65df734d154ef031c4ca9b53aa5db8c7795a003d07ea334ccc9288a895110df1238c7ad2e789c16af644be84a84bf79ea60bf6cb3d3a1943dccdf3f9e97f6ac5ed604d7bc46d539c6f6031ab5e3089f9627c5e7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263285118Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d, Name: checkpoints/3968/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3968/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=aff02b488190ff36df37a27cf794c5494d8789e7017b202dfb6ecb8a5c9aec7a96cfbf19329959d1e7aa5a69a3de3e8a07548a945975572d7394adf47ea53aaa914c4b336301537c302b3bb0dce8b637d3d406c42058d0ff1be3bca503897acde96c8722a3d850357126b952f55ec30293379a8f5305a31b6c6a8189e47bce036e8266bad1b2942612b31b6d38365965e8831e4d1e3446ca94c07e406226801216a17318e54d02772e39378d0bd7d355733dcf7cbe6e47c8d9fa1ad14a9a9f7a62d1d5f02424ffc58987c5b1e55ec30d1923dd4cd88ef47e401f137257baf93f5fc592131aa8a6478cc3b674c65ef2c38ad7cbcfd736cd852c09e26e618b7bb7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263368175Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0, Name: checkpoints/4416/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4416/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=bd456fc3e6c9d2bd7d60eb889753598400b1f65125a1772c8bd2e03b930100d0ebb86b9141a6d0efabbe11126bebc30ffee4ce7d0fdcec5781ac52f48ec14bc6500f14cd1a737728fd2ae57ec35dd09e9635465654123d30687a474add8bc1a868024e64fe49a8137e759073108c50f31e978bcbebccdd40b0145648c64fe739bbf48886ec8c85ba6b0aee7e4911ab102547c1873e867356a375a3ad6d81b4c7b4d06305e498599bf87644286c407e669c6496abd84a5cf5369e9f102f9ca129dc66d1999bc4ddfe01920f35060de9aa66359152b3d9a97018a9a95be6a0733822ef1647dae548df7f357d891070b84b98263817a1c0ad101651672b17daabd4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263386798Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d, Name: checkpoints/5824/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5824/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4e67ba7c8ac8a53fbfd5326e9ebab571a2ae69ed35793da0064fece7d6f026e2438e2224a0d2aa6d5e7d6f96d7a064c3093ef9ab13ab2b0e426f33476e3c7dd71d2771c73d17eaa8036043ecfa62162e59323ee14f4db171adb15236563519d4f9f3aeabedf538d1f83d4a780a02ac02a26f818703dff87518062782cd0f07382342be60e2889c60ec5c030e337067924ec3087be954482a9d3d8c47e8d4322e7fc4ab37825829be99d9e9219af09d854287581598319aa5244ad7791ee9b5340143b29955aadc5b0ea7cfd773bb40e91fbcd2988615f3cbace137a0353a8c3310474b8cf4181150b926382c68550a2c9b48eace4a941579a80036fe8c4b4313&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264359699Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d, Name: checkpoints/4032/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4032/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6f7c34ee0d386e3b780af2564b2dd09f267b580f5ff64a2b477fc027cef70df60ebe0a8bc23e4aa53980fe2fcb5acdbb25a3791d1b84fc905f7aa812a14a865216bb96e127a99024f287a510d069b24722b016da16ab847f976cb138991a27417615d6e30e48fa738d36ee6cfb1c31d6d7a4d8101f738bce26a19c8f2fd85e0a7ab7f8d2536a54ebf0c5cbcf55449741862bfbd521bc75f544b6e579bf5685b121c7fe169ce0f38b1334616c56f001782bf188bc4411c45e7b00eb0d4e54c7bb712514fe212f75b3a56dafc90bdc7adc1472f5ba7d447d9a9718c7cdb85bda56918ab7bda23e6536b47e08108753a2fe7ebc66d4f99ef1c0ddb568d4ab980ec4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263202164Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/d, Name: checkpoints/3712/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3712/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=82be31787381b4831010bb271785bed785b119f937981077b20c7bb64e56ec09df784ad5f04cc652547e9d7ec7b947f6b0a595dae14ebd8c2b7a6362e6a6f28fb22b60e1fefac911cf01ee86f4c9a472b95e7d4c2fd1a5c13a0526ec987569a9c8874797da84a133c5c1a38a67f47e9c0a95eb081e96a8424eb2e36dd1343dae1cc170cf4053c43aaa62aaa3e355bdf915dabe088fb64ead7cc4a413b22ec97ac353623c4230bf61026907dd9cca436c5d19ede07f97523d31ff36b1278895bd2182938d4f3d71c0b8b32d8f4f96f89b6bc945424fb96d2af7c0ccb026171b5ff0046a343aa55f66e87254a72018f5c767b1174d79df2e391b8c47da243a5665&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264612749Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default, Name: checkpoints/3520/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3520/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7f4341682fec51d723fdb2c43bd2f44617bf623a093fbba0c33f096b19ecc38ba3844a7a1ea35b8ada91780fc566139d22a91dd7ea91db0baadcea2750694b12fb6e571a0dd3c00fc0318b69ffd48a4fa18208c420534f0743739918ec7be8b517cfc12d67419d6b2be006e647a790fc15533f1d2656c4aa8ac963d781a2866cfdc0e30f4af03199f0536f5ce1c77bf5c5fac65e137102e746d034db6d22e30b28e188defdd838ee8c3dd1e7d89ed1880918642b96141b3eda37e61e2fb766b4a2238cd3385034640599821f5429e8ef224fde74e602f8d91e188bbd8587e0c9113f32f34d086e6de9cd49eedc173b99381ff4a85bc78471272a152b8faac919&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264629128Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0, Name: checkpoints/256/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/256/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6435e458066d921218f0057a1f10e6841cdfa23d8af2359812fc9c2ba7a5cda8991b78adaa853b68f828473df20978221da8541a1340b5272c7a7168cf7427082ec1bb8b06335284656ca162c88390215dd3823c782d50aabd6454d3974c1386a13e9c56d1259d7808d73ca7966756f75f2b4928d8da256b604ce78901514badf7b3066577581199397be039bb9647f80f00559cddb17fa93be3c20be84261011b334bbe98afd9c54f53371f036355aaba50f531c0d7edc4fb65f3bb13dae546a65b684bf91315ab10519e3bee56f2b3abc75dbc617ea8bd74795decbd1c8de0e2dc7f9bdf1cb5f94d18076afc0e3d13d07922060ff539f7fb091f84591fa1f0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264677976Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default, Name: checkpoints/128/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/128/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1afd627f998e30a0fcdb00c1b8a304ad6e8968d87f75dff563066189a8f2b65e20a6ec1a08d3e6604b3601ad5c8edaa98689f8451ef9a371d09b38a2ccc0f9d3e6f598e884bb7a815942fe9ccba6a67cad53204e0a4c7172d2140c2b60d6a79f9b38240384cb253bbc949d22b5a70f283d16d9e3a2cbf9993dadb65222a5bea2247f81692db01d4bb06b107cb2eff2736a3fdc086a145a0d6cc2a443bdd0fa39a8a811cdaa3cbd845792acb9e124286de87e62cd12d0b70ee099ef1313ca92a2d1fd679f1befb9aca0a780020fc8f59ee9b90648615a41e0337c166b5ab34dca61b52fb6ba3ded5201fc64a74526c888c4d3967ca75ae217f59e72c4fe7609fb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.26353414Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default, Name: checkpoints/5440/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5440/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=24c7fa6aa51fe7f566c9e0414889180029779ad8b5a0838a3feeda1f40610a0343a3660b969845d9535b86d263144d145b2bd7250de84268859bb4dbc16bef1537a23e28582f009bfba4f413070360532a249f44f806660ff46eec86d0fc2388ebce59441c7ba2e529fc5fbb4fd2a8d64d1afe8725078145b9e34c3c7c63605dba9ebc329f24c59e9187c33a74d5a0908ddbc705bf5442d9ee18c0694b3996c33f3f29d5b7e638c0b66b51139ab1d33171636832a83bd0f12c7f4f39aaaa00cae24e31e01c4968076dbab499becd623534d55e197071d8d364737ea3984fa0d9962baba19d6c63a7142eea8c4758f4343e64556dbbbb5a7819f780b7944f835c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264452635Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/d, Name: checkpoints/4736/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4736/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=844e0ff1e4eb6b4bcccec243bd6943d6cdae2375a760e313040992b79295d0b2fbdd4b29e35f646222164af849b0124dfbd8236ce0d75e59f07d3676e08709eaed18f081a07ebf9d9a0e212527c9a12f0f4cd34bb0b2d2628ddc890e2f15b1c5d99f21db1507d449315bfe44a505961f532f4a943aad0cdd55d29b0ee4c87afd5802fc1609af1a04b24f0a81af98b99a6e50e8d17ecd7dd4a35c28a29c806f5947413a82d56f793ac6a39bf3dd0a1f3db7a307bfda85923d53dff36463f4e4b76ee78e54e3137f79f6ea4ee163ab768fc8bd79767393058a9647328fa32cbb49abcd847ae6e2d263c5981dd1d299938e971c0071779b3756edcc7400d46b46af&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264424338Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d, Name: checkpoints/2688/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2688/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=371e7585a5d9f00d6a85b08235fec1bdffc285d4e2d7079afc69c66709666ca14e9959fcd29eeeed317bd11694042d1a6031beaafdaba385b13674c67a82cc41720b7ce6495dec87a8ff86f543df69e04859261c9718b3fccdecb0406340256c4ee767a2e6a256d5149d45db4fcd8e0657b6d80cc453c1dc26770519ee13b04b03a22f46ee04733b3cc5df8126c0d54c61199c5114fb8b670d26fc02d9f48bdf332238ff39eea352d10281652679b5b4645e8747542bfdefcb2e29fb502fe2aff2f1047901afb75d1313db4e1eb0350cffb1ccc0ad1819bc817fb4d7801ee1fa7f704da520be41b9e87e0efcd234a1e085f78d274143963136d387f3db351c73&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264808293Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0, Name: checkpoints/2240/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2240/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1604bc69985428c742c242e3f972d73379cb5368bb0ab71638d57acb0cd0f2eccf42f9739ab4fe1455b62051d005560d0d85faa51ee13e474bafebe4ed08fbe279c256e0815e200d3a41a9c4b8549a48e5172a33f794d6e3566fe721432b1742f74efbc11bd9a4bdd98fa2bc23568a882d78eef06c730e3dacdbb68ab997b503a964161b728dc82936fd11a8be3179c80d8179ab973f20122b6f3f157deec567354f888cfa7f5f969aa0350f817974e7183d1ccb8acad7f188980b4839185f5c8756682f931512831b085689791192338088c6cb0ab869a5de1c9914e5c872b6241edb4b9f7dd3421a7e80fd4fb1db85d401a4d28f22af36f0ae917f6fb31fae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264815454Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/array_metadatas, Name: checkpoints/2432/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2432/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9e4ac22895d68de091f28953476430b3ce3ecc4c1cff2efa653daeffa25d234dfb0a8a396bcda0e0f89d7f5263e4886af50912b93989d0f124ab532d92968583e1eeebb9ec438a999edee74b79a44d4f8242844a78c545a441e8ee0215c81aeb2f220026e277bac82a91200dc8eb0595bd934ec563bb435b133edfec8d66b0ccaf626aec09dd9ac3d2550d89f6860538a5a5fb89375527d4dd6d4139a0dbf8231f14dacc07d6261f93700156077e6236fb5c2588e0e20cdd4589b5acd0cc2dd605b49139eb57a338f586b95ed3ee134afb5101ae6197c2b5a3ff01693ad73eee5e981be6f2701f9ddec51909abd090f6d8b219cb0d46d50946a4929d2f64743f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.26485181Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0, Name: checkpoints/4032/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4032/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=459737702d77f5bcfcbbf9176ef847ce353121a26cfc690873ce9d672123f56e2b312d84fb356a88c158ed40bfb2f46940a6931e5fe63c578131e984ccf1dce585ca8e2e066bcfe23e5370bb435fd26f9e738cb25810426f29e08a79e3e4c608295d511c444df08b785a33b82926a5bd29dfa697a741c29714b9db36080f9b87315d113dcd419d49adeb78205cb28d07bab9ce598b3465aed39c2d636896e2c2cfdb9bd72a4ae0ed4abca90373dd2bf41da3585443239fc71ac200afdc90e109b4480c0b86156a8831a6f7dbf92d4fdd780a8d69f9ef4603608374af1be8a6b0f8a217b35166eac72bc52af023e8990e553253815bbb96e8fb91661a0ff97cb3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.26488158Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default, Name: checkpoints/5952/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5952/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=84c342b4914e69a72aa393f0a3d15f9b85d326caa1a2d2a8c96ab80e802cca308ab54598087bb78859ac2b72d149ad59348431a8225061174d58ad3bed5e8b64f9f59dc5492b145e4746843b7a7f15b187e59210d2a92b84bbe59687f76e788b5756236f539905b0266effa3435212c328bfbf39daa04d8f0d3c75bd98f9e368237237a73b2f13796c18c94cb8a9c846e8c6f07746abffa5de06a8a66f688b5a9abd0e12472bd93e594c4bb232f4b3c2c46c0e95c427775a08c2152b8db736d9d7774e94da4f7489417eda0a4e78b7e4e01d2ffac5ea61a04ca3454ae00d1ae1f2c87f1bcc20733cdd85c5bd3b12c32a3eb400527be0148a6e8a308a0aa56d24&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264951944Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d, Name: checkpoints/2496/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2496/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=36949477d54c64f296fadc880823ed1cb2c54bb844bf99419af0bad0d99de290004e854d1b2dface7c7929b6a9482427d2895fccba153f3a15a5dc0fb1d37fd279966f7551b19e8f246cb71e4b10d76407a744863921cec7e86d298cb0c537ac1ed4e5be6871b2349d4f01691b5dbb8e0df647dc82aa0b7702c92ad15218ec8848b339ec48f9c8693b90a72f364b18d9da1828f3aaf85da59e7de6c0192306a317e557dc37164103b56ac15fade73f416876f1b24f4e2000464dce650515bac7d94a0b92a0e2851a15cde8425b4c10e95264f7c4d357136e9159bc5e2208cb63d614dd10b2b5fab4fd43324c9f4bd788d26e7f58fb75b16889a52690d01eb9b7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264430305Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0, Name: checkpoints/4288/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4288/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5d54f593b79019d104723613fa3ebb3b2bb17a60b375b63c3187903da6d12a04fadbbb729f305b0782cfa07a7f841e8b220de1470a374c15d83c633f82a6881acf5783fdb3b2fa1c398eb567fd2a223e8eb70489d5e696dcc622520c683440637af830945d23ca15faa165194a43628a8e95816d2a0d1d69a7aa17e7d1c771d7bd75444037b629ebaf8b0b9d5c6fd41adc91001638ee7c582c81be54c74bc06b01a6b7f6d9b781c7d34b5501fe5a141cefc809898adac0a2291635f5a1cd8e3693a12ea8f0ee09f6e4fda7ba36dc79525934ad8fd924180da32713ce42d06cb00df054034f8621a2197aaf1ca4f6dd489561b99b6ba22b8b13a4f2b512494773&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264912019Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0, Name: checkpoints/5120/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5120/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=42105ede82e926108fd60abfb7e991417f6881bc669a7acbf0623f1f949e9acfdc2a9d1df8e9fd6ef821d019eea90f0e9fe845ebee0d1467b26f31f6fe67c4433d7b2611a52fc679afe90ff2262fa4e69ec1e0e81f9e84b6a8926e8bfc8529b9ce430ac0885732b88bcb10d2f777f897c9c7b1d24796ae3b4484ec4d2da33d973bbe0a43111363ca6ed695fe2d3228f50bcfd7fa698a674aacdbb58e3279109ea20d5648d6db7f27061ab1b76d64df75fd5ce5de900b51031f79694e0de16c72116ceeb18dc4e364ef87aef52e7ffbd701389aaeaec85d8c2e0c5b2ab1d6fc32840bd9dfd9dbf6a70c25b46df6ffdafb57e15c2f982ac65e35fc12498e652de6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.2649816Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0, Name: checkpoints/4672/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4672/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=96e4d58cb1dd7072a848a7b06acad1a37c6cc09318cb180cc2b7e1e14c539fd5e2b9726d5b8900253dc2b8fbdbb8418b2f8e81197c1e6689e97a99f9e1fbb6a2367cd1cdc69a96c35e5b4b228e86ef114b7d555efbf13243ca5733cb81f3f0b754618287cddb3c601893295b36384885830d1cd702469de48c3e04b1ac2bce199a368b32bd17581bcc8dfb7a44e576c4510a91582760e9a1248cb8dec41f062f0731943a92d7bc91b9bb1770327efae303091a31abfbd5b941680aff498b1aca2f38beaddd0fdaba05c665a943b265fca61a92706b4f321e8658c00414b37cf570590dac36e725dbc1da160405d44b6d1710e8074f6409fb703d5c520199ed8e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264478827Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default, Name: checkpoints/192/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/192/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1d601c90179bb0c09758314675ad9493e05c31a4fb427f93d56d31c6ba2fa9e426ac30ef0f784f3b22720a4a1e03a47a21c39c969626a16823e75213d55e000a0ee4dbeb5256a254b9f092fe6a8252cac8b2d62b1242454f985d48aa0737de603dd2c0d9c901b078d062b6cb86f4ee52654f1b6662c951b9abf6eac49ddd24e643e5386d45c8e8efa41a90c1e73218e17fede8085b4ce0e0669b24287c42fe427266940560171636169bb7809dd9671afc50f252e73d310c411b854023d238c2592a128f7448127e491627a7453eb68f9e7e9c68dacae578a6d29e9777f4fb90f5efc729f26aa5aac4b5fff9be4d1528f9092d409f4f53d96cbcc99a063098fd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263434826Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default, Name: checkpoints/960/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/960/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1be081c5e5d9f699570fc5781fde6ef35241228401716e533a56d5fccc51e42a5fa6fab9cebec0a40d80d77beb86c94785b723ef6285c5d1d713ac49089bee36f3fce1a1406b3d7280fbc50c9940d7d4822aa22239cef490c40b26d79c4eeebcac866a55b39ba011d5f9ecfd78164b755c5490101710fdc4bf754712cd4e48fb97884f4a49834480c3de623f1aa04bc6e8260cc536154ab7fdaad0285cb71b035e1b8814942bbd3832b9922418c523517eb5f21a3e60d4fdec59ca6171ea1bcac7b1056196aa0789f2f5f340103c5ab2acbff1962da449caf7bde974972d26c063f077df53ad5bae7edac56a0d749dafbbf8275d2a34acfc098ab00f04eb2fde&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.26504276Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/d, Name: checkpoints/4992/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4992/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6ea10f4e050c0021f9d3ae0236d3274da4f4df7d23739e3fc3ab2dbf7c1951d57245cafddf58ed42f36795bd6eba59a157ee543b813c27914857a3a46169ed3be55d0b623675ff8416a91dab6bfc5a96b89492b84690836425d9f1bbf7e927166d3712c469a6be41abc3563bc4c7d98e5d6410e49a2fac3bb771883b9598ccb1d8d762c09af8d5aa5d2ba1d9404a3747cd3a4ea871d427a23e73f78c04b4608c6c8406b61067556ae9267f357aa162e5cd0c29efff7d6e906d3b43a8bc294371cb93201415303f3140698803c1359ec984d855b30435f5936297682ddbda4e33d35dca38d782cd6a948ade2598f8dfd98413645f324204abcff37db690a34c34&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265066086Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/d, Name: checkpoints/3840/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3840/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=073194bc776248b4e42b723442126e21eea5e5d91c981a22269ce2c49adfe709faf38fa84b0a6845e679a0d46c5364d3fda2ddc7689563eeb3e0df52302c9c6737d68f980c25c688fe020d08375f78e43dfe3246986ffab357485b77d1cd89e00c06cb6a33729caa47de71ca9d524e143549c36ac8b8ad55e261fe7fdc920202b19d6b0245c350917a40d89aceccc6f1c76609b20af32e4a5200a281dd2c2970adad2cd41bac2b302cb0d202499cd877fa406b39868edcea0b77ce75942f3be61a30d979e44710197c8f4aab1625437ecdae12cc8b7b574db9c66270b76b2b8ca145a5dd4c57fe202b20f59713d390966d30759326b1f362ab47ca59bfa40446&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265108649Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0, Name: checkpoints/2880/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2880/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=789f0fb2dd96be5823309f2cba55fdf40f77aa5f52e7b5eac7cce82f2922d19803ff2191c77691230be0db06ef6f927a39248eeae69bca83e124e515d3af980af5f3adaea9489e9046cf96f5871d4b4f4afa848358ccc470132ecee5a847f1d91e523f9f249c5523b5f7ef2f50381bcc1959557393fceb09853231889e418603c4c50ad640f643959def2b74eb41783c34d8cfca2d08505ac8ac89d76708d9d07c392002af84c8917304c4d6494390813e1d13c6c360d1593ed80d1b92cceb4b76c237578da15c0729d15892708def42c7159d12eb71b80e436dbfda4e98a8e3eb1cde019c8bbceb28aa77c03b4811a7209c4efadea0f2002cc4ff919c74162f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265048281Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d, Name: checkpoints/2624/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2624/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=64e4a9d4e7b3b83f3e1f178f3b95be1c77aa1f81285e68193f71d97bbf8a6392440220046a1bab434693bb0e6bc4debddaec5ab1918542da7ec4da680275abf07de4bccf161c17bc8b9c60dd14a30225ba324e0b2dfd1e9f57affa4a3f089da290bbeda7576b7564a793ad3cde469f0d9f1a82666b4465773c590f5a76ec7ec9ea8cfb7df7d3d8338e7116c316554417ba6b9b4f4e1ae2f36b1bdb1b0dd68f2633ee7ee5b070370ab0dfaff496415f0e96c0c0d753a6e87e152cf4c71fc51d3621d49c39983382297a7745dab118548d456d48884876a939ef6999a7fa995d44976f04b93508834a905c044dbe82559291a4ec04fc0bf0e97ffe1f0f8e51f8bc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265080432Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default, Name: checkpoints/448/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/448/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=03c5271e68795593141dafab2e078808075ee85a143f7bcdc18a292b9337c0a3a6a4d7a9a9074eba838a775f683f726e26e6f5371132b51603ec54e42c406c258d94067063afe3dba06f41b6591de13ef57868e569e47324b7ac17ddb53666d91e1d19cd6b30f5c2cc0a7282a0376f8015d864de5faf7df8511eb25fd854b06a5da3a0672f8f2852e44b73ed09368bdbb45f8dbdb8b640b08e68741ffcea66d194d3719067ec21f33f8908836f87250977a4fda2c4e7be98acef3ab2c32ade274de69dfab7c161d0281022f9eefa5abd60abcf19a781e2de57596cf85104a802892c9e4fa3b0af357b615d45e62dd0eabda7ac8ac8bb385ac3ccc9cb3c53102b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265162544Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0, Name: checkpoints/2496/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2496/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=67ceab3028e12b1474fe8c3a1d3f35156064351407c2eda21ae64c874e167b04a6878c0b6a7d80dd01e21db0f7d862377024364f2cf868c6302fdeb151f6fe3a7131d8aeb3498310eb97a946848966067b5b28b3d8e40435625dd1134455805b5effb014d330ac32f498878f33c6cb2882f68999fc4a0a9c9195cc17f1ec80434338d96ad3fe3336082d83ad587313b5347572ee8fd0c0a519916b498dface62d7f09b07c7929973c3912f836eb3005ca3b0971f211675135456d2fe40679e45f6709430730ba960a6a7cfb079dc6511ab2ec6d1c03c30692bb13367bffe4c27be0abd7c2fc3be3c07a26f34800ade7d293f362cfd836e0be133dc5fddc10da6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263527067Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/d, Name: checkpoints/896/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/896/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=18ad7607e3c93c4b81ef8660f7c795e306862071ab974bb65bc47c721ad9a653c02e0c53858f94d22228ee7bc0e12709e3c46c5b625e4fbab8ee22a7992919c97239a62adf0476c9a3aa1b846b77ee96374d19a47bbbf08e7c38ad863d6721b9c66554f43eaec1741f883bd2b3b5c243bf95077a8f8509ffffd9ac821f4e142c90ce5c7781f4ce082be16a5efaae0f011b9682f4e8c8df455eda16fdded40f884547ae2eb6daf82284a0213361e669b5fd9129c0a8e7bc747d7ccd0acadc966013bc1ce58ac32526326f69047d88c3ff27943e99f1e56219a4c4162da9fb5a0bf37b7a02ac1ecdd62764447ff6502c25229aabd95869cba3975c107d3c829933&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.26472177Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default, Name: checkpoints/2240/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2240/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=42b5f4b22cc9d50e3eef788b175d7968d827e6b729399810860058c413d5c7a5599223c4e8cfb2c6808bb68c67454c216a0cbc55a974c9495b974cad88f04bebf645dba4d318d96aa2d115fd0be0a2a2b88fe22c9b4f659dd67d292bc78646c0ce046b079f00973e42132ac3c0697dceb159c06898e085879fe2e58c7cbcd9b7672c53b3e7a86185cd8c2e6fc82bc2af52345ee3140c83ac8e888d327217e2380a59652a69ced00428d91d508cd4f7fae073a1b66e36f70ffb9f09a6c7acf32bfd07b8f28450e84a6759ee735dbb7f44bb860e08c3ff33d83aca2c8544fec4bc731a9ed382095f8777ea7a90a52478525c2bca2f82775b1d4e8373f62b369909&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265316808Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default, Name: checkpoints/3840/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3840/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=913692e75b793f0480252fba111772689e6fb4ca7975ef347d50baae6b6709769e41d9f21e6ac4b8ac1d03f3d7331dfd712cef8b8fb072299f601181c669b30e36777fa0bd02446f8d5692fa25f3c1515f4ac9faa995604620cd5b5a1bf01eabf18d1042df8902be8b88b58868c17cf596edd859a6641d59c330076f83d9a376ae1be9566e6e7cdab46ca2b9a3e12386b719398cf8d14c06283919724c7a865af1ea961efde9d94d047069520159045485b6402e6121e0a887733431bf140c36b99f698b8bb0957adf52a3aefbe66005760a1f4339d401cecce739109aa7fd13ca6055a87618c39e1e9a2c499bccb1306daf45b3a9c64a4bada67ddd67ebdb84&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265332779Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/array_metadatas, Name: checkpoints/4288/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4288/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0853cb966d62d9996a69a92818723d53533813fcbffaf6b925fa8340bdb9db9be54b7375d24e25cb92e1a15a4455441457c82624faf92982bb6b641bef29071690fafe57ccf13d75eca8969c0352b483b3bab39b47bde6f426a4a2d797cd2ff259eeed5045b3505ef6240b0073a9a6185db239efa8b8e3e8c4b6ccfcd31133bde8948494f4d4403f66f6e09c5a8bb54d7b40714a6b61d16a87fab820ef66eb116a5bd5b15b6ec816a35c827eda03aca16fb676649e56ba0f02edc141a26478a169860992e12ca1dda82c90521ad3a074ecbc68b1fae0170fcd0c227a1021b36e95df60fe091c31cdee4fc0ef3f6211b188fc0c729d6b514fdd36d8c7fb7aab5c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264792888Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0, Name: checkpoints/1024/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1024/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c20c30a04489b475806c00976d3bb5168f735ac3fb23e3d51d64ff5910fe8235ce85fe6d5c7183aac5bf5dcd0ab74705edc5b6b256acf62d9df68837fd09c12d1f58f8f4cc17fc8dae54fa491a946c1606d2790774d6601e6e92d426cc02b6e9bae83d58349afd43dfaa60fadfdca7c645a1954339946a4205da2f9ea2a89c27b7cc5f8d7f1170b55443d79d91f9155df582af88d56132567529802a46a15342399b47952f06da33169474fe16ea122657cbc72a3ebadbf4e3dea4cc4f4570bde80087276b1654814877e09c15b36778eda766d50e487766c10fb13971646f8d51cfc23fe30f550147f09f8276c9a91a4ad80dd3f717a12b5ac3d2ba95218418&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265452567Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0, Name: checkpoints/4224/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4224/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=013f8340f97b7a7ba009c3d0760a13d36f4206f216fdf59cb340f2c549e84b7fb7d2aee7a3f6d44accfb3f16371ef25fe70f9da9282fe785b458abdbf19190d2e91cfdd4ea595a3d127560414d9fac71dc03ebbc2e006f739a1fe667aec5afdfe9a178b79f9fcb4b773eddd4b69467fe291a1fc0f64332c40e10d3b2556ebb02354bb8993944f8965371e871e7566138c50c7c8d5f1017a22e31928ee859b109d7c7603faec822d28984688e5c347946c298d34bf61257fa7a3be258f3a3bd4b7ec2db438200103ef60250dfe07b95f88f0eb6ce6759cab750fede5952215e42756ade9d50d2e4a298b239ccf4cc9238a9e3c007aa68bc462f02b95c9a0b137b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265468827Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/d, Name: checkpoints/320/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/320/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7b7dc911d4b0cf0d985f2ff108f2b352d0ae553685c6dbc7b4b744546d4950dd8ec383f8951a645c31e7fa4655777ae3b72ac41dd33ba1c5dae1774dee09004933a18b0ddac2726ad691fbf7f057cfc68af44d976db53d8d9948b5e9507d4bc01bb87dec69429a418c35020ded1656204f534a42a6cda322157f9eb69e3d2006659b0053c0464ca47816c0dcd0861498d45c26ca3116d0c1a7e74fe7748e1dc0c5af4dd7d2f5f28adaa10c98957c045083d540e513886121dbebfbc88ff8d61c2b3125c5bf9d5819c8bb194240e339d93714c155827d22bbddee529c9029bdb8b51786d2ea9c25daf53abce52dd4fec7741a6a8b8315da62a20abc5da9451dc6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265505273Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/d, Name: checkpoints/1536/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1536/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=729025fa1e94a0852b6e7a9ed778f9b5067352e7580ebe0e9a8cd62e5dd7491dad03b821a0f1856726849794db96e9dfbbb825edaf011575bdc230249df2de61217f99e093398c1d1aae1680fdbcd907ccdfbedb4e56d91faae41b668ebe020ecb642836689533f72c00231ca20add32612fe3a9f034c241132195175f190e5862d17415f23dbbf8383d1ea31df890d4c11c534c05ab3d44efdbe91c2523584fc57e0233ad3c12322c545cdaf47d2f22d76e03d0cea0e4b5147ab94910de58a75998f7d6beb0c93c5f28807fb8a5660e0c1dbcad1c5fefa772648c0b89f0b400eba40c11976f19a96e552e65883370a85cf468df776d96dd4f6fd1d2d1b7e33a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265542427Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/d, Name: checkpoints/4480/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4480/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1e97007efe7c152d98dc966f4ec36067d42901278189f2a59549b5d275c600701d9fbf84f3312e8fa162a06122ae2a27373af0be7efa69c9800c2dfcab5aa0a06f437a99868cc64a174f96cbabdb9086c10a54db3894db94aa0f9a28cef59f419488639fac1f957c231d31f0a2a9ec8d3022728093a1161f6a5ca7da0dca4027a45e9be3ba294abb8bfcbcb32d473c2ef20b3a8e46ea9a156a84ec282e61437d8e9f174c858363e421ae72c96f101676c014501e09f53678c07508908d5877086ae6d29c13449e556de8ce28182e10e9a7da851343621db25eca2dc4a5f04282f17ff625087ff41d97ba54f19c0c167592dad8b2caf01d9548dce6c9de118488&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264845858Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032, Name: checkpoints/4032, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4032?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c06c467efb1db031b1609ec8bf69195e8c9649ac894b798faa3ba7fe9c7b9e5dc6fa0fd8aaa81574a38201d01dd3d38d2864710c7f1a85047b608a35ed6382c893f6f092302e3a0efdfe443a00673598d75abdf2d9e663f6c2f9c66b10ce5d93f2c7a4d53d050c2095663e91f7e0d746a2ab7680dc5bf2d605e21f591280dc3740a8f11dbdd291e77935ec7fdcf13b973de7fa6b8201e85104fe2bce174cbc7424eaf7de333235639fec983035a742a3a9c6533665a6015e7e0818e6eebc1a7707e2c7e9205250d6c224f9555f6791135a10b47065a21c5d923a6eede04e9d4c82b513ee59683ab6a6017511b8bfeb66bdd2bc0a8c83be5672c28bc5ff469367&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265635858Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64, Name: checkpoints/64, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/64?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=49db45559a062d1757d9e4af9b5e009d9a3f790d431d26573ad7610937cb37d546d7b590dbd9fc88c2e3e86dd6308e59c4ee1a10abe616a8329051351adbef91b9f811872c9d42fe5f93d4d66a67ed85ebd63b3b9718086bd05fd34d681e150967be291edc02834a28b3bfdada94a36bb8ffa3e93b4df79889943a1dc28a50b453fea266b3c9fb23bae7d16022bd833eb4e7f3504ea8ab2046d336736afaa5798d4afc1047562c67af1aebb8eafde3f2390dc9d551660f8be4c33f1e5033c623f7fb4e37cbcf2259ffb81c4d8d64c48d59fd1935caec83cbc500baba007cc2eb5e3995ba784f8bf3483c50c0ed9dccb5e4c915f6dd05861ce60b48d55956d967&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265616929Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/d, Name: checkpoints/3008/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3008/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5a67cd68f0f5afa98e477d291c4a97e192657a6b07e0920038da0d0dc65d4560bb3aa204da351729de4554585f2fd320f5bb2017123fbf14ccfde3e40f75f589b8e77ee9d47642da5e4c9259783de415bb69bf3a99aba4cdaf94221f3bca39430cff47946696afcba04d0483e002e6baaae17dffe449a6a3adc90be5fbd2c75676ba55e1fdca2993f0c520d888c33554c6b035836065fb0775a0336a1edf3498e5a85cf0c5da69b92a6ec67ea69621b401e495f49089de774b9f9a6289daba5a557c8333479f6198ecd12037d7f3684a495239828fce1eedb80654659b1d809e1ec502f4be186e0ed8fee1c65327df0bee72078f4d740d54aaf59a5bb96ae939&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265652106Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536, Name: checkpoints/1536, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1536?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=006f17fdcd5d4a1c80946b7472ab08344bdabd7ec6895435a87642f52a6c1aa9a2414c09a27f52bc6601409783ecf5f0a17c44c7e8cb1dd91e9cdab6bb02fcd35358d788aac17521584685bac00973422cdb02e3cfc5d4d1547dd40472c178645f15122dcfc815ee4086a0740331a72769db740a95cd063e2cf976757ef1171d97890892b932dda83a8c4a1a6b32f48472be574010a98f6c134c3d2b70a49f05617149f1e00e00cc4942d654d1f05d642361076b76461698cce487fe5b8c440e537e066d7b9cf6f3c0b8f1672f7c0cfcf111d2083d4130d0e9f7501455bce8f021e26607dbb615308909eb5376d1d1722e2a24e6d1571e142f81b0934c18d6cd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263364433Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0, Name: checkpoints/512/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/512/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b01376bf4a1a60c1dcc7539f9db959018e0ec6257e4340957e03095e385791592928c3c3c1127408ab72af83fd75818dd650a504515b601731cdec21a5d3245c4fa9eec5797b969ff7ba4924b835e26953923effe2f04930d0f9d3e5789b460a4c626d38dd805b1ee95715881e010516dfed891b3cbbf540494224ee1f33118464e033058f03f617c2a28f28d2b20b82519a107e2c50a90a0db8848d407b7d2aa27a5f7ce1b3d4504ac97d923a7c7ea5ec7c78f5021c3ef4eca84f8cc43646c95d23bdb089e8b50669d83f14737462a3c31b03aee8927e72e04d2ebfbe14d55a65528a54e248449a3436a0545353e7667973500ba1524953e26944c765ce0dd5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263402917Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/d, Name: checkpoints/4160/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4160/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7d57cee07eccc8597b4e24cc1c2e0562e9d0d5bfac85b931c3954d60ea3448dc611523174c5b3123aee2be70236a71d5682f62e009790c978ecb4bdb699dd7f1c4d3dd185b1a765892beebff23e679c3d59a5074d12130a2533df9748f4ac88d54fd3389ab52abe1912b671ba81dc1b902049ef5958a6130b8ce9921566a736c8371f11080735f3cd2ddaaa00aea9d194daf83b56cadf06c415b2e46258fc29b588f7584cda1a1d20404840262038dc97506deacca8e23f840137bccbcee79f8606bd4e52723736a809d2ae652c5d7b04b3e8889387bc4a3828cf1dc76ea0d85cbb57bb64eb7f509db6c1abe7b6bc174f570b1e32be01aee84efe45ca66c143e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264817559Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/array_metadatas, Name: checkpoints/2816/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2816/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=81eea785c2b15221ae26c6956a6a12f9fb8fa3701d71d4301c159171f4241a6bcaa5638548ed8fb6fd9b3cd3aeb20029d923113d3882633190755268eb439e84ee00f115b03d67f80e5c45c69243cd2ffc8091ea853bb5cd2f5a2f53a08f9ca053ed3fdf9da6a587d2378407bfa1cb4393954e4de5cfbfc1e5c9a501544714e6f6796b349553327325abb39be6553ed418d56ea5ea097033934d6806573258f71197f37a929db1704a942434c04c63ac5de301cebcf1210f4dcda4cdb531a7c89069999000c0fa88ce90008ec5876afd8f293bd1ee4d5173c36cbd2ca5f5863e898986086efb1a362d2f706ba3982a357f50bc60999fdf94452776e09b78539c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265723117Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/d, Name: checkpoints/2368/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2368/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=079510b7af2dea2c3b33930ad2bf3710d7515a4b831d0833fee794e2190fcfcf2523c284a5a7400cd0b95723698c01f329b7cf1075a4b333b6312b82be3bfd90bf4ff15b0b89f39d3482ba3433e1133e24734d9d5b7263fc2097870aaed05f0a5d3e12a10013cde4e42e581bebfa448ab4a1451b46f516e69f04263adecca5d6dc5eaf269e9f1e6fc788179a6787647584aaacb856037c17e71ad3ad1d3430843c79402045a9b578ce7f2f041c2cdf5a9e268f5996ce57b8cdfc527c978113f165716d701b38b7ec1d3037054af4d49c36eb061c6e4141bb2602242f5c47edbbb9f53a3804945b582b9f40a72aadfeed17f25810bcba5baa6fa938c6aceac013&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263158408Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/d, Name: checkpoints/3584/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3584/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=67137c5e95e496c8b8984f30ee7daf9df0f9731b7e7f2c05a288a8e2dcae549918dacfa79dd8af33a299cfb8645109dd888ecf6eb9e7e27bbced12f2fa7b786a02016889ce08c5f9af27c5fbe4ced253b55cfae50c77c36aaa154f8dbd4050a5bdba71d7c074b62649df0397b8c12f63cccac78e1ff7d4ae12ceddf1eca19f2b979d184cf5509eea5e9deec64ae2102e95112da5be303b791fc81a67521820c2063bd6de4fd28d32d38c5f2a2a1e881dd1603750ae093bda3a7c792793895d63497af0cc2ecf4beea8ac8cc19463fbda5ed8dd2628b841995923d7896e709ed8a3d5c198f61d6b21fdb54b68c6dc533bce19b97a014120c438c32569b49a9aef&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264959755Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0, Name: checkpoints/6080/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6080/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4b6fd4aab497de33f512e48a39a7dd2ecb406cd717ad796cbc450dbd2ba46c7972ae3686699f22f5a6cba3a3718443792410e078a78b19fc1f42d29d0b7d615bf790540526c49d54c89db568bd0079b880d7957d7f7ce8e4ce69d4fe0c5266b50a53c249c15c74e6c255e4402fd86b1f382edc399493cc2e1a6cf343af688d779df755db776c66ebdc2b1a0e8416a85cbaadf24f6d006ae9bc54bf01356dd1e6989683788a8553d9c5925d067ba039a83e469667608bee0118a36fec5b78437745a4abd7e02e45f659aef72f520c4a1b255582c7ecc5b11cf05d39144db4bc8a98c8a89c2b5a1a278d6cbb74e1bb0ef265b3d90eebe7bb63fc304bbebeecf2d3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265471924Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0, Name: checkpoints/2176/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2176/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8506c68880863562aacaf53a94f4a4c4c15d456c8dab942c75d2e64a0ef71ec3d6cde58980bb6d7416ff46fc05f679472fab668e71c21f428c011e4ddc412e383d2a800427be144eb3681369c5293a979d308159501c7895d56f4c6ff26310e24988cead6abb7f62d662c1c2a55ba7bba3e1f06e44f1cdcec47ffb0d3974053ff04a530369adc8074f676466ae1ef6654abc5affe9dab10ae7b8b9610375e53a12d7a25958f31b472cc9b003ccb354c872b4c82c67492b3e367b41ab0e457f7fb95affb5c182c052e9d36f0753aa84469aea2e5ce66f271e74191f8be7c752c721c541483daffdb0805e1931be71695c25a424687f31fe688526693afbf3d8a2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.262809039Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/d, Name: checkpoints/1088/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1088/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6f3ff8e5842bf723043dd1014f989bd74d6dec7a1690a9f16881c8d7cc160136891c0337d13505381d9f438c0315460e5dc07f0b1f9ce845dbee44643f5c547cc2fb099bf740b1e2f8b07bebf4c9df20898f9adf09dc28b19b5727d8d2a9f57b68223649fafc7e6baf2e55fd4050be43ca82bc3b17387d2dcb13bbb649473e81036e8d1461004747622e21d3fa9faf92c6aaf67580da0dcc2dc28b4f74abbaca40bd9cea48a510efae926944b3f82da9b0f732492c94645863679308d8f63f86de9df5c983716c7624ec20cc465799404f05271d9af68321d4e0b73ec128b6529c1958069130d4a2d26367feb57d3bab8f434b3570b29a6e0075f3bc7de495bd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265135821Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d, Name: checkpoints/6016/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6016/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=331be26adacb9d017ad09cc60ded9d845fc984218100a80faed05088a8de113c320aaa14c6c945ab27b654073fd5e5990d67880ef7b07ea5c2272c1c91849127a9bcbb04960f5c4090b3948341011f56aeed46c6906dc8a31ab616b61e5b467899de40ef0d979d33e712b9d572120b5149346e7933928422476dccfad0a611a9bb738bf4040cb59195edd4a0f8f088014ca9710e8e0d8ac84d6a87511cbbb0fc80de332cf5de70a0cdb8859c6766b5bba3b78759ba75e911e511bc40b68a1c5fdf2c946c9022ddaefe67c2d5bbb6c03d00419431853a468ceb807c0a1b5c2417e626851cf132959350ec5cbe4dfa6ccb35b5f6977680f688820cc531bb68bff3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265371279Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d, Name: checkpoints/5504/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5504/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1f9f903a85db1b21b29fd0dc418acff5a4ef9f4fda8e6ae2ff2d2d35d0a1e535382cd81ea8395fde0e8a461321ab74d692fbcee24596ce07f4929a3137947c1ca888da1ecce3f00e0f6ccc7a308d033ec0d75071aea82667e5d7ebb943788c8ea0b061b88c54883a9190b8c37df57d594d341cd18ca0d03ffd9bbb23137d67a14014de26bdac5b4d20996a5d71bc662162ad6595edf742d5de67f73f12d824230f4ba857ea135ba402fca2b15abba1ff6cf4ec0a1211d5ab2a2a2ff38060ee88fa7c487f85f36b53dca5ff7195ee61c98b4a322f3e7a38352a6a29728a80f14505139b1637e8bbf751a39f135ea0bb3538214ef56abdcc528fde5a2801d26eaa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265195419Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d, Name: checkpoints/1856/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1856/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=ae46f70698bea4a18d69e6883f7ef9d184328d88c05262eb5183643cb211e3109ae54cd1bac8601a02dcb2620c9ba36a7e1404f9f37cf4e366400cb44042d2ac99dae7fe5b5a8f740a4a6c1ffd8d4a20c26d9b9ecbaef5b7b8586c2e9492a5310614be93b3ae0490fa0c78ec21f87405ccc02b6cebfeadee937e746c31bf83b98eceb10894e3126bad86785795028adbdde1d89823b5b37be3ca9b1b88015facb5e99833b1e6bbf578c3e721c545c1264c5a3cf138211dbfb3f2ad6abbd73c77dbddb099c1fae5d5a85c8c3fdea820afb9b5d5fa9c62faeb6081960750a4b1782037b71937b0a364767fd208f5b10b7d62b6902b7b7b23d3d5edefe56671967a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263446529Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0, Name: checkpoints/3840/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3840/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3a8936201759001f7c5a8ced8574dc5caeb3372c7c91a75b9aad42c788ac4544adfd68543730fdf79f918343155ac741a657ca1408ee184fd056dc7961399f008140648d94c3f5aaf3e24fe59657409e42725d7265fd36b38979992bc65aa6ed51701bb6a4c3211b7dad0feaece834de11785ae1341ef61b6f1c71842530f2b07019a96db4174a6c0e007b90bfe25f0ec36ac1d73b64cd834022f3fe3f45c1eadeda836f0673a51432430f141b2a89bca634f00d23df9d6fb91eb5650176a8c7d34ca3f8637e8d6d7c29b08d0eed71c18ca85377474c916945339d0e1436089693be8ccd60e887722771635a159287acd9b0e057d064fa1a4f505bbab219c52c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.266041756Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/array_metadatas, Name: checkpoints/1920/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1920/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=37471f74e51f169879c5eb824e4b1b2b4523c3e2d2264e302dd013e52c8c971e0a96996769fe9a6268a5d9f4ea8092203485f8331c29f94e124265037abfa3d1564499088e1064a2d4385a227389b6f190ee65829810f51c059d1c96af7cbe58d2d82d07b0452d2b318cbcfe420a7d8acbcf48878f730082a8c10737e47cc70eb6833e67fbc7ad477f16b5ca1813537c83afdf424a981eb57b3dff2e678a0afee275365341e2f8c6c2e5971a2633307e2f2199648034812250e6b7cfea8e8ac0efef9d4f4228a0c33f352054eae7bc3b9c2118a0ea83fc66fef097b32424e47db59a9dd196faa43766c00bd34904b31c76c28edadf58f142add39c9c3dd288b8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265485223Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0, Name: checkpoints/1600/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1600/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c2248996c6e1088e6b94a50af3fe90bf567e3efde0da712ce7f827790978af80403fd3108cf19ced692eb084010886d2a93044c2e00bf74116d549eb9fdf71f3937e287131a330b9fc38ba1e0d39a17f575b71153968540b3ac3c243c57b2ee98e986ea07ec26f60bf06ec6a3ff6428e536a3255fd64a99cec0f0431af2efb9e9df2e70ee08e2a9b1c22a7c2f1a959e94b362a9303f4f0b6d430c839dcbd1036d610876585b3a0125d9129ff41b578ecb4b1617880cb59b704a212154a3833599742b03403a4bc5504ccd842fea17980dc23c423d5074a21cbd997182e378fb304f8e722b6c140c209d340fa1a55196d690fa8cedf1eb4c23fd98c2573c65d6d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.26609021Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0, Name: checkpoints/2304/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2304/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=73caf2c3a144d663c8a3f7ad32c5ea9ee70036016994e2bd72346d3a2c5cc808a40ad5599757a76425c88de22c8f9d75f47496dada14a6c2d527f5aa0b9c9df4ab957cfe3698559edbaa6309e59675b636a56602ccce1183930d28609fab779ae3d3ba03e0227db19f6e72ae8210ce88de9c83ac693abac53b5e2fb6ea5cbaf79ca7170834836ba444a3e6dd3a6718fb16a07d33550c5c812ce8771cb09ae5d4faf393169fdaa758396547289c9330085525f31ebdf4083cff4083547e24bc2bfcbbc709c27d37e6117611a0c8341e9497f094706763cbd59ba39f2ce3ccf06b10d9f2f960789f7e97230a4eb3b8e4773dfc750e02b58aaf757b0397f7a06b18&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265053383Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default, Name: checkpoints/1408/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1408/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1c84d92dbfc594ec6b80898678a472ccc602403da6b4c753c719e0c131d55714e3a19008f03979d9e7d179e434cb4964eff9dde88f5b84b15bec35aae18eb831b2ddda4f645a32c0fc7fc3c39cbe4587c330ba0ccc9cfdea5b3eeddefe1aee57d61e1ad67917f6f74c68f101604c583939479825b168e218672d8ebd2e9e9210e30d21251ef8e8f9ee322450c719f9866e44fd062aa77923b57ac1cf28fca50bced00055e33724f998edba64f43861cd024e0eafe743637b6d9cda90e75e8a8751e4ab44d33feff634a3c988c805bfa1d9337d43d73deb1cc65c729ac1f1e30bd2b4b0f5d2b4b711928350e7565992c913ce39d91d1363eff29cf2c153517ffd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265628607Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/array_metadatas, Name: checkpoints/4160/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4160/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=86bb09d9a35e75527266d171a7f52db95cb449f1a8fde57bfff8df6c5b86ef932c1b84fb0e541748415ad639da93dfa93f0bccea9d2bae4f62231e34991ba0bca92063b23e5e43aee0a9c667a7155ab5bac5d3e534991a039b8a8038a2beb4b5c8aa3b9cdf36ed41bc11d41f104afaea1916d92cc990d910a0a1132f7d8687c3736da7e5c22dfbe96f34775eca36704d35a681172d578f61a91b6c52fa23bea05c092921da186f322bc9437e6b70c41bfa0a505acb0957f3c2f28c7a6672ddef329156b7b2239c2ab0c387d852585918ba0c86da8de87618499e6042567a44337813f545e803093c46bda5ba599d0a38e1a8d1eebe396900f3d99286bb21ca8f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265657824Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d, Name: checkpoints/704/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/704/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8e5d18c7ba679be60faca853b816ec3bcc93fbb4b0a2d3ea0f197111ac15a8ba43aa97df8ede92ad74bbe4045dd563b7b46211ec82447b29190780ef0f887e83887a60bee2a61bc0d18b946cda73fe10dbe91bcf3022dad1dd3add2029760aca25b18a077920df4e81639e84b5fe7e8419efc470f0822dd6265396537a04f2a613c1f613af68b8ed1537322a9311cb4cdb7c4c757593db75f929954822446334ddc10bdf2ccea02ffb06d9c5d6f6dbe9f1a862f92fc410070ebd98b8bec7162534f5205297526568ec4e285b4a8cf7a67bfcdd0191309eb9457dfec32cfcc105bbcf8919992b9b9e34ec44e2f93386e2c2abf190d000221698922c574a444253&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265460492Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/array_metadatas, Name: checkpoints/1472/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1472/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=603d268386a4f0303a9abdd5d656bba3e592335b8a0211b60ca5385f4126740f593ec55f29d616c699f0f08b5b95876e6b217dec384088d84b900694d64ac99128a30f1b1fab6656f752a67c5edc058b406931cd11d42832389c5741df6d0a3370266ed087a09cd48bc5ae14a6313e29344afccf3c943fd615c0011907b952cf1dc74c3f5cfec0d036bce191e4afaa9cc85dc3cf1130aecc7c698bca2e1391f5659fc529aa8545de8d4fb50647dc79cbbdc28131a7613a29a618b2f2dffac860e05412b1d16b7cfc200d0b9cf312b7cd8c117f86dc7ff819d385c6e773cc41d4e5bc01f002441624e5a5bdc8316a39d32c9540326f23fb5f3c83231c2fd153ca&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265670892Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d, Name: checkpoints/5376/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5376/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4066775666ebee4308a3667476fd435e405ad9c64e5672ebbf63fbad99905c265fa37f6cb038192e12b2226a668e5972a4d02b214e69a482bdbbd6c8a2cc02bc853fe602fcb38923f4023227314727e45e1a0e854f2ead83200a0f9b6027206ab6d5b7197496a923e5ced98a9172fb227e13622a8f35db69e6b8e3e6bf9d26cb7fa2f46c8db5346d2b226cb580834360a21d0b653fc146a28b6bfee4715f07e105b61cbd9d8b4c19453720f8c37ab12cb62e6dc3cf20560f87fc7c7c76ebb08165a8b404cd06bc3627d4ace6b95f63c5a02ad1392f242ba84ee00bfbdde76fd53b71ca53c5d6f421b2788579d0e81d710547322c89cbe9262b034356d847ca7d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265468546Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664, Name: checkpoints/1664, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1664?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=67a0c0720ce7c82222f46a002835d59d4b74b86adfddeeeb5e0ece30135ae4615fb62c20d89714b86ac7cb9726b4c4a89bb613de0dc9d8677abcf661ad324ffa0ac12cbf7efd0b908978babae755dfa0e8c87f3daee5f1a87e5bd3e8e266fd0810a7c8e66896cca28e3250dd3cdc4ee4dd0f9b3c615fdee6bf7247ba7b54c0ba1f6b78e94050b243b51ad5cbad8fbe16185a95473554d1045afdcc34473cfc52699b0e2f83522f0069129f29585c0c25013a3fe89541e768108ccab6d7e5236887fc1e2e88c1928d2da496e4017a485788525aca0b3d62890003bd6ec307f9930b8b7e1f90c0d1ac27961d06149525e06870a9c750d86823a2f2988767e9cd74&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264885697Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/array_metadatas, Name: checkpoints/2624/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2624/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a7bf54c16cd48a48943cce4221576e334f44c649802b941bddd1ef41d447bcdfc43ccc472a20abdbbe3a373c313f4086f15c85371d12343509b18db2b81c5c52ca9d57666ac0945419cfb128ae649a058a35a55ce2c83191b6a50458be98ddc771470ca5122e1f9c4e58c5d8a540dc69e05d4943b1a8ba0eab5cf4e505ce9a4abcb92b9578ba40aceef029833b012a51cc2842e073759727aaae59be03810f526c9398bcc38e50e32c1423366b9727f2e3f6a61644474be4e628a8963dce971e739227ae6a274ca2cde6525137c7a1cd5a339abaceb2b4d89e4246cdf9ab950b79dca6bda848de653eba6d1125cea2e05db5571695e0ee6cc1095cb1df90c350&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265682935Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088, Name: checkpoints/1088, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1088?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=affda7720c181126fe058999277de049c1815d3bd1868eddf292937236adb0f5e1c0e3ffd29eb9feca457f05a521a70e6d85722ee2f0f3aa38fc2baa049635c98f71487dd041b67866c86694b987f12d017d0eb277f294de365c6a3c3a16f60703820cce8c099a85f184ff568fe6a147356fa3c4265f708921510e7eb92f8b0d55343a843a71729582e1578e2c88b24d04f3bd1d529623a4154504eabcb0a33f10579d55fbab93cb1b477ffb95f062b88461334a84db93d56b2565edff7ad79159d6af754132ac03a367fec9bde59e1e9b520ec190b5512190d6a5d9c78866fd526a7102e4868441e552b71a9deaa7178e28a07d2f6ccff068c99e19a7fe2651&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268025861Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768, Name: checkpoints/768, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/768?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8d5a57c99e06f3d31a45d48e9ed25e63701ae61e8887ae80dfa8272f06d2aa7b2b26d5d0f943c4bdafcfa5d612ea6031d1962e6f0d2a4aa453cc57ad1cca8cb32b247d4ae18c825203cce11fd2826fb35b7978c8fb49a66e800954f457341f60a91139829bd2fc359ddf56773e97272fae553228ceaea72c417b825011acfa5c0350ba80ed437f17891bbdbe5c073ac3844be56a7ce4f3ee87b14ae99cb981e139c1d89a7155860da68761845bd0b5e5488caef36b9a5480f13e97f3958d6bd6f50293d359faf44d0ac0319213e02eed9b019308799822fdea02d5aafcfed70f3089d41cd9eb6981f3a500b724f55cdf2c2a1d263258774d6b836a58f11cadf1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268077127Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default, Name: checkpoints/4800/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4800/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5351d8aebd5ec243fae30d365f2c4378da51da06577a270694eae587694f7ba02e3189018e1a6f49488698f58d06de03eb5096714a0a1f89391fb0bc38877ff427f42e8c31e41ef2ee599f8c39652733b72986a4164837bff041e9392e2d3acaa57e25b6df5655d7512da5005b79a55993c194b0b7d1c8eef05b0fd5e002a176a29a2f6ffc56e6830ab524745272120c530ac67a4662efb655db6305c3bef4d18f13ee5630079fba37cccbc61eb41e301e41a9979a88054d1ab2a4b145cfd2b6863c61876d8fbecd1e8fd13c544c75874483417b2433efd53fe8e1cd952212bc9341e327453a577e3af74c02208bbefebacdd0d96c5d73af237ae674a4d15025&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268180586Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264, Name: checkpoints/3264, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3264?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b7af4176cdbf6506369f10b9ea674cfbebbfd28fdfa8e565cf72872c945c8b2b753ff76fbff3cacd2ed9aa69e9f56fe26d0827af63454c864ae851b76b4cbaaef5b86499d1662df0c463ef8eadd12627dbdb472f79d917d390d96212d8f3740c1896ddcf1688d41c2e717ba126133964291e718d3f8828aa93ae5f86200cf7508de17f15a23fa26798248fa4be67bde6d28a728ea7428d66d72124ae04ac09e0173c065c69472115f8451b8a7c2292a04e2353232dd824208ead887a4e5bc2086d7704ad83707cbab56fed2b49b47c12e3aeb539bec8b9166b5a19f8e161399caf5c758243abd7cd85f47e93650b10f64e47b5b04a90958d86bef98a295ed9cc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268192382Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default, Name: checkpoints/2176/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2176/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6766d3b3766938ee5752007a9ce630cc010059549da475e47aa44bb599da49e399d19bcd3d2927ade376a3bc432bc7f3b932577f72c494e273da7c4caf0cf7f231b7b42b27602caf3dff280f09f9706853cf57c666e2fefd83f853d4d37f2095d0b98ca2e5c0f6667290f2c5560266b4a87fc9a7a8ef43eae57110307e98e8c7b8464a343fc8a1ff04118b37f24d5b4ddd921e4062fadacee04916c9827e8aab849854fc94430cb18fc0b3b7ea35375f9c0e5b4f28e0a3e973a710e560b87c714426699bf44f6cc699abc4f8eb39171b2e27286ba4e1b8c649fe59bdeb845c107c0b36efc4b85a9c843f54cb24558a987afbdc398144bab2f94aedbda19d7939&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265890682Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d, Name: checkpoints/4544/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4544/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=37d0141a96f7165a563469dc19c620f8192e217ac5fd7c52d1e671f363a250803702da0f7c45ea54a3ba9e2ed47068c602741db78ec2747b7f07f0adfea8461e630ae0ca9cc953799e24fc4ef4ffabb498f64657b80f80d5a947a4bc158fe373c4cbd8464bdb6783adeea8ade5ff3d9b794ed8a6eafa734fc782352c73f881e0da9a7358184572a4bb39d8b77a9074ac65710bc6de4018308e0a8b693803b886260a1b13bd55d0e869d005440654b97fea93d8325b85f1418d6f1459ec23bb264adab424275179c2fbf20944e51304349327110eb48abe002058821c0091c4b8c0e39950a6022135d5be4da018de987599906d6cdead853f14c4068ef6e82445&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268275348Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624, Name: checkpoints/2624, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2624?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=655b894637e302a59d688251d8b4a23ce4a8c698e62101a59cfe731a1f6d0d9d44f902b7b46e55b38b438463230009c91d6e0e38e1f401e6049d3f2b1419063c87998c2e859fdb12a4926baff8a4eb4d57231c36bb6cedcb17956eaddce69807356c4f2792a766259046cf1570a00d75a1fdb4e89a348f392c4e07649970d83972e533ddb028fa92c455569824df24297ff9e1bf820ec8eb04fa16cd80f3ca7bcd350842c74daad328046bf4b3c46350d515a7411ccca71dcb3171ca3372c4b22d8264fe42a8225bf4893f8f3a755f5c95b93e81006171846304d3936a7b64d1747e0d7648693ac68c5bbb85b6fb913a8afca88693d5d06b7f2da1a1ed932c7b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268332061Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default, Name: checkpoints/640/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/640/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0b5a773b46399db99bb1c91f5b20e2a18a58fd5a8dd2781bf9b8a64c712411bbf68e9787b33db6ee4b6951c6677f5a538138bf589b00cd0e5b6f96b80ccf548ed055a2fe41aed763fea981da5453c836d1025c1f0ef7221dca9f96b1963e5aef31583c69e22df60277c4ab62556c225ef0687be3191f48d2eea793deb1ab16b776884ff2eeb25a47cb1d40643c460a4b5b2fd7024cc3dda9375e4e72509975246ef0197c8ecd2bebae669324218265564c72b2ccb440573114565ecd3278f782ac6ef30b11b422ab9d659ffce18337fb20b73f6992269d20c31bf0167b7107e182a3e3482f3a7fda08a4c4955ab61a63e112c6dc91b15d8268ca49d1b32668fd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268364341Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776, Name: checkpoints/3776, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3776?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7465d9e43951e65b8002c725adb1c3d423bf4eff1a3a23bfc2d3f35bbdd0bd700c4cb6ff12d7d9e0f07ad8c1a8fbe260bbd29b9cdd525df0726477ccf11494c7828ff9f50be1450758df589dc23ef2f8b7dedfde5d4ed92e51dad41a49db5403f234c14a18125410c84022c51e31a53f1d1b008d5d350f4368759355ec989539c21834c447ab4d1052abcea2cde823f355a48bb952d8b530e55c9bf6b0a38761796d171068ab7f1e6fef2e954aef7cdae584b950a1f3222f7945c972528821488b0f8d16b093c29dfe4378380ee87575fdfc94ff2f2ce83a97321ae028af9a251e61153c827a4fba64f2be3f001e478a45f3d89c662ad97a4b83d15d471c75c3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268420567Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/d, Name: checkpoints/5376/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5376/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=81971af18190c9b5d789debc70baf722fcbc992ccc39920845e40dce86c1b179a4c763770c58d3e374843a958f57223258f67d224422f272ddac9bf36c95c662ed3869f76c33beb713d7d08bc9a479bd90a5474999b7cbbc0934583e9726ee216f233e2d90cfd164c8fef2fd340c1a3f63c9c139fe0e98ead69e4a5c592950d24e688e7fcfd79d9555ede2276dc88ee4d8742a7769c6e379f6c898b6a078d8bc280a6be659d622c4cc1ce17dce80af08b616c1ac48defa937195c356432723d4b588b20d52032312e8d744bc62a2806ec7c90f30c9c3459964aefbdf5ed15aac0c0d2106d98a82404a585f1b2091400a4c976b726d8fc9fbf638d89485c51a2f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268491511Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d, Name: checkpoints/5888/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5888/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6606d59dc95824d8b38c1c0b80f22d988af8e0ebafb56ce3843b5e34d7141ecb161ed1b19c45b0c6d399634f71687834600bd877d323ce6aa6bf31d760767cb7d5498c6a78cb89edfa86646b5eaea5f5ded0b3e630e9df8b296287b06e4be5cfd96ac358fbf9ebdfbc037311b123856a7fa2c947cd4fb24ffee5fb212fc55dcc2d1f8a284e241e2ae3444538e57d131ac59dd3de60a8670d13ddccf5c11dd0a5e96e3d8b8a6e68e755f2205373edbbf9ecadd766eacbdcd3c3b9fbd79a2407c84f58bfa6c48b28758d1be156dc41112b1608183050aa03ae010da7833a4b541db954bd941222098a463f0f2c5973c19db10e60f71af2cad0e046d98f2babd765&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.26826792Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default, Name: checkpoints/2880/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2880/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4edc378ef069b015fe6411eaa0c13ef1e073a83d822e611f2714c12ff8820df12133557ef69bfc57755082a799e006a09fb389a97470eea5c7dd15caed1c84f6ed4833500f24df5cfb6839c0b462ce9164b9863d6fba03d66183c457d24c41cbafb40f6ba4d65d3c142a27e19149936eec30434e7d218c4eb5ebba397107b7529db0ba027dc46d13d4eca43d7a325dc3628d54f8beb78f286bad203de388c03c9d228a841ed150726cfba41d2df594eafe2fdf630069541ef0058ca6999ea6ef0871f1116e39fce6bb84b78546d928d191c4f065b6ce36caecc80965d5238d7d55101d03de20cd8d724c70476ddbcb863b2b7e841e4796316f9a25887619c90a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268852729Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d, Name: checkpoints/2752/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2752/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6fa0973a5a29d0fb3f92d53946341b3a2d9cb5d76f0938c989cc249a904cd6b507dec16f74852d921424744d6039a4e5f0fd367d27181c0268812f467505c256012ba0f07b88be551a269b324469f74d7e7db6d4f1de423a369af20908c6f9a2935de803bcf991376318189cd242dc5ed00ed1d40737b03ba0ef8de3315403597d5ec2c4359081eee600a9d21ffe58ae7fd71efd37971a077679a1cd2b4ea881f9e7ce80b8b7b78bfa5fcd0130d551518ef7d711b8d81cd203b571632b4b8e73bf01dac55c5f181df6f8fca27cf083ed296c04d2004985b73912dee8a3b4afa42b447be97e1e729006aacf720bb5baad8956fdd0951fc89557b27f34dc8e9091&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.269081458Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/d, Name: checkpoints/1664/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1664/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=abb3eb902dbb5339906bf43143c91703596fb5bb173797a30fe49312485e8b202cc945602a46ef150d187d6fbe99c6e2468395958a523ffb5b8faecfc74941aea65b36266539ba4562dcfe80ac19bc6edbd20db97b80d5775a8fa1dca1d80f6166b3a281f9876d2d4d4e4944dd2e14d68dfa0e9b351642bf67164d8e2a1bb383fe89123dfeecc0675850967c6eb9f73464dc79ced0e43b201ead1db8d144ec3b075a4f01ba0f060a410eb6079fa069290c91bac4cb6f748189203e42bfb8a21dfd8c66ad7e846e520d18400fec5da2e98e9ebc2c4640177424d1c244463e68a4ad6998f7ba03c703f182581024857d5ed1a4f74fe4017ae4a099c95961850297&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.269124425Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default, Name: checkpoints/2304/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2304/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=15fc6ada76ef9c6b07305bf9fe1537489fdbaff7cede89c6f05707b4c74eb8cdf7c2e1c0d26658d52826a7affc2d017c21a4efda876f9626635d5f514743a66e5704297dd6fb56f63a0c0fd5ea442b659175a6ffa62004bbe279d92fbba7a7e6f353da6164a50525931fd7e54065e595b380490096cd14965636457dee5c3d78a05ad6ff5a961fa653430e01f8029febc39b897ba2d2cdb8bfcbbc04554e0be4f5ddcc35ae4665eda1d4aeb6e93151008c1c5f4e8bd1795fd22f22e7d6785da680e584c955ad6e8ef20470cedac563c7a476a6166af2da0f04c2dd22762cccbb8ed0809a4aff35f87ae38b7004437c819682e0214dbb9d185674efde9fab5771&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.269155728Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0, Name: checkpoints/0/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/0/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1d6112b441cdcc8f151e2214e8766e35a01d80826cc7c2bfc6a645e99dbfed196eddeb8f110a44d3c08f76e858e78dca76a63ce218b9c420c8acccb8a3413eaad9b36cda74f21b4045d8cba8e543a1d79bcaf2100aa020127f144fa04b9f84dde31c77436cfe3cfdc3652b106ff8fbd9699bbfe133d14f028c085867256b632fc1ff524e4f6d66f9540acfea78ebc640723541321978c9c69c5ad978bbf7afa33aecda447c204c5b222e9577737444183faf4c7617ad185af76224034c68522003588f187ef719859f3ec7aac9552fd4a7be539b72db5c3856e98085b13fdd4abc70a68088cf656a391466dea6b9b8e2791ec6c3918eb25bcae72f1c4004c9ce&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.269201777Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/array_metadatas, Name: checkpoints/1664/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1664/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=76a34bf893ac6b84f069e68d616c860c6c1b841921abf54981531fd9e5c013e655af6856fd5aa274d2b5267b6b3b47757cbd9f17e53bb5348bbe7556104ad19edf7c59d035c78b232ed33521d0d427673984fd316ff1c9fa8eaf5d5f11b2bd70263baa42f39e7729c5f4f2e97e0d6e86e4eda7675c085eb7ac477980f3df3b2f94f5f79cdcd432d0d9aa16447277fd66db137505a377f2efd5ccb06533e8b316bae550c038b4edf4cbab76d0eca31dd5870f0c40160653a996f551f489e33a8582b6d7daa755d3d8912a9a3bcd87b2eb90bdcf2f8df210359c96db76384a962ef650dc64357c2ddbe20b462013be45f54405e192480630522613bf5c8e763039&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.269261715Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176, Name: checkpoints/2176, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2176?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=047feaec07b3afe590cee5e5500e51412975d0bffd3b73e070b25a1b5f56061059f78be4dcba882500ae4f572081404ee75afc6bf0e5395a786b7470d3bd17c6996921f12239b9f377a14232020e36a03b5b004ca5d3e125a0de224fc9ddbc8f4bf4e8f848c90ae3bbd7ab875ffb8dbc8b3f05abdfa673987a9e8eb7f1926fd92d423f850569f3340534eaa7d89c4c19f6fc58cd1845514cb34793eb50e84a5e03aa79888b3b0ddb363e322f5bb0ef7347323f1e3c067464f35046f303afbee5837cf72bf430e6255e7260f7ff5ff51166e098b473e91124d05b9679455e110eb2330205d58490d2d643bc39e94aac1eb20cd1f305be413129b6a7e49fc5eb5f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268386898Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default, Name: checkpoints/4992/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4992/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=813ec7932436623ff77e86465d9a015451ee1d6fda6ecb21706c3b8dc2154dd1e2663b52b7eb01c7981760cc9e567807952e9f2473ed3bf628527c53d596563ef57c7933c298572dbc0e355dc77ca076cf79926c46facb06553add2935410145d740e39f423e903e4505a1f7110095da3baf82e8e776481e51925c661ed0383587c27b1220655a13c812aae535834f7c120d4b157c89cf12f60b88e718edfc36a4465f75c596a2bdc557c66bc8d090912ec69cb105d62a80a73ad80b25a8408130eb28929ce0fd293edc526838fab1a808f0d143633ac31a9f091386845eb9888e9fa53f3b453e5775708e70fe380a9b85ffdd945259a70ecfda01ea3fc31583&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.269336137Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/d, Name: checkpoints/5824/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5824/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1a3e6f822a7fc3edc5d90a0334f0257e900c7e248983f23b97c128e31c0ebc59d2d73e08cb0f9b03993d3aae2416bb4925f4338843a6e09765edd4ff70557a22068083b51b6b27e5835efed14e7f57a198d57f49b268f071fdcc4ae50ba0659c25687d0389f4f331958875bb766d2bf86239e53d905153c19bdbf61e8a472d16e06149b8e16b9cc1a3e0ef2ace892b024af2a2fd65a30e3d9ef4eb7fddc3dffe083378ac051c20500f023096c328d1b1071bbf17912278c142471e8369b8770bb5b62179851faab2272f90f469a69b7877fe4a796b2ccadec72310828c54b06c862ae8e7585fa48919822b79350085ac3fe98433b3831e7f83472eedeaa916ce&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265806744Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888, Name: checkpoints/5888, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5888?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3c7c1ec60ad406fff3c4569e4ef25920affbd2135fb4cebe36466d484b49c8aca1f946e890d1dc9a50716ab99ab813578c7f2598473b2f286e94b3706a8db961ed781311180008cd5c835101b3019307e6a49d75e81e7ffad17537fd08941a376f7605e52e5976a94a457d580725e7f1233f427f279903a830e738b3a35bb0c0f98f700b13c44a26f54f98a04263fd318dc4b6e04f924252ea79a680a475df072c63be83a815c539b9462f6dce25331a35de45d49c9e493793729708c7d3d0c1485d9c2c3cbb20226d00df56d1fe7203b412f864ca26914aed19cff2daacf1e9d5acf85280e0a152dba6973457f583064d83f5f274c789f3a5c3fd53835c672d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268368297Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096, Name: checkpoints/4096, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4096?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c4ccb0f6533b1f2742bfb8a0db7b8449d772964090b9860c51b7b00026e729e87826adea020e83579ca4bb61d56c04c39637b819cf64c5e2e4ec287b11967aa256bda3e248c339738fb377b0f650b34c09f38d29d08c218b745520375a4acae02bb44015ef8919e35df80acab34c24030752480511c6fe137989bc0d06a6b0314ca138c48a012ca97c368e80f797c3d67113325321ef2a4dc319e3fd87dca221997c2ff394cb0d09d0ccbf27b6a084e3f446c6a7fbdc93d7163860bf9b099cb9e8d271e2b93f5fd689c9c3bd0736f8c21672b6f74a0aa17e4d9db679fd9943a4d8b600ffb3b3cdb5413726587d477251207277c02db628765c3e2806c0e9ee6f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268173691Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/d, Name: checkpoints/5632/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5632/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2717c6c98278f889af2c601918fc73c2e1512429e2a7120b3f1352b90e84f2f13f09a4d8e1b72e00732e0c78ac7cb7306bacd4a04aea02551c96508b73f517e1d84d5131559e4fd42a16306dd66f6dc8ae3926874d92153bbdbe99299abdf0592988b73cf876b99b7d339073df0cb695545137170a62bf9612bb377e9bc8d781260f7b26d1441d5233bbe1be747ec20521ca144826e62f722d6d04e41a1ddc68e996a3a3ff076ad79f4a6dd16b7e0d217a866370d671b6e593b8d2fbfd02fcfd1d7e7b03e40dce8682c96560507d33507138680bbbb6ad55d89a85bc4963da2d2c511252bc5066f457a9f5e29f3710a83e8f961d3a690c9ec09cf468a51f16b5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.269033838Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0, Name: checkpoints/4992/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4992/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b9ccf202b5e9485144200b203a683e6f622faf8cacfd147aec909b5fb201e22114781af8b83d458631ca2a5d5cd9554305b859e8bc6d369673341c230a09c0d15dc1e62c9153a2b5f8a504a04d0ce24a48887288d05f403895ad46b245f60f32f35330a774b308ad9fcf2964c005671c41fc51388119dd15fc2d27c4498e32958e995c415e2f9c5187b62bdc52c17d67760952ecec91b6808c4eef61a33ce5a5fce1f7756b550f4a65efde1d0a2ebc19ba2ff33dcd4dc2a1b50a83373704659d2b0a4675b610296b392827aa555e57378c355915e0a9ad740b493f41fa6ef1cd9e6967e01989cfeda016c09f8712db51b5b466d84d6022da4de2c972231313c9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.2690886Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d, Name: checkpoints/4672/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4672/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8a394fd30c320a0d02eea3683997e3bb152de0f1256bc46789029c722007292568bb4f18914e420d3810cc0be15fd7ab19b2aa407fb63fe6cf810929c0812831bac8bfa8bec4e9caf754423aab6931f840c0191600abef01f40267a8eb32790f599537b7eb94fb4e11b35747b0db44f6219ecd1a1700d01a419c7f347eda027f659b5dac087da4a409b01d476331a209901772ded6c0ca122f09a95d9e90ea82a135a37697ca51fbaae9b66026a05154f10b087332958b12c729d306e4637adeeedd3ef7cd34dc765e78a114ae46e2f9431e8b5d997e4b88882d59de41bea13d7e738100beb1638fe4a3761b5096249cabbde22e1f17bbb2a70bb69e40b2aaa6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.269355096Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d, Name: checkpoints/5312/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5312/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=896093065e11a4f907ae281a9698366763b2f002dc88bb5de939bba089a4b25e0533eda9300b18a1cf2e96f923d726a3b63aa4088df32ce7c20cc9612e297265bec0ba44c06f35de2831cddefaf407875c30e5c5c62812fd4ba2aad437d88bfae9cb93d26bae6b480c211f12e976b3a8c1411e4817de9116f7b33da51011d6dc32dda4f8c38810ac9bc1f5390d0c1bb5b84000eebc605da52fea8702a664f7d967016d7a8d902b2cf7633427095c038d4ed45784c8e5354ab426011e1986a26d677ce899b97f1c17db070ff0676a3c0a3694e175c31beff7a5073b737523da9eaf8c15dbfdf35a8f30ad45d4894cb37c791c24ef62fe9a5562a4fa183c547e6c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.26940621Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/array_metadatas, Name: checkpoints/5952/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5952/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b4ddc31c4701dcb92f689c6eb3f0c422a80f55d61a6cd75f69c710219abf8d462b479a6667c03dc3aadba3ca279940c14ca51a71d40783c3917e551dbf79bdb218fabe26b7ac93bd93b0457bd9730931c88b61a3659b93e90dd1cd85b2f9917650ab13b816a32be7c37c6e8afce5ba9e6bad5335ca12059206dd4a1c4d65d20bab515d24f99a0e9f0201187b5bd8b5c1bad8e1633e3746c7629ad59b18c1b0d6da4d4899996006a71221edae8fcebcc22b569401ab49991283d36da49f83791e74501f374f63114d5acbc6257fcdd9e4527b521de71e5655a079f91b88fadfd7a9b3cf841f95f8a196b332f3d8f9c8ef3c277b2d33884270f553fa58f9bb6390&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.270058121Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/array_metadatas, Name: checkpoints/4352/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4352/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=507caff6ea512fc174a3871a44a369b6d8953e71b1480f88cec63dea13f006a743bc0a58000eca44becca26b0e81d5fc0a2b000c8ea94713e68f48e5b62d84df4fa0188028b2e71047081e3294d023f5f0f045bd1d4da6e8cacc4f7d933d77bfa6bc4002df0a9e1902f49307cd14157ff598552c7b8bb237c6ffc31ca20814223ca919cddbba429d524b9c4de06a6824fee7bb9c1da7f9b3cf0251490fe8dd40a9f9b0583409f31579562a554ac40df1b8c607957d354bcf2a13a01b89ab90d443e60c8f1429cd6ac3b868f6c641927c0f2d252eee69d50c91669f9f65fc4412a2514243747a9890c98ec168e5a0ac5e16964fe0a69b4910dd27d103b3c60c09&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.270145161Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0, Name: checkpoints/3072/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3072/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=57521c02a13f01914ee3061d39037b47fb79f2545d2df79ff4cedf59daa9ca2bf1bbfb8dad6f7c09a857b6b1ed88b3e12435d64971ee2b773a618e95583b6847b485c1e0639ba7fdf432834f95d54b9c3d44351ca356d091f2e3c93306e634ff0917c8b3f978efcf4e940643351bc3a68b80bcafdaadd397e51ff09d4990385e1d2efc000f52143d852b5c24938a59709437ae28e6ff6a880317831fc307b2629656f1177ac1f12845e43e42931e9575fbc9eb1fbec190487a35b65f883c9cd145e622fb7cddd1e84d8a2be8e76c2470c2c52e7352b166750602823779dae750403dadfb37c1755529e95a807c2abc8ffe30b13c4d05bb40d0ee5982ce4811a0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.270291559Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/array_metadatas, Name: checkpoints/0/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/0/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=85530a682f3574f067ac94c7903283e9a5eb1278475c31891e1b182af061b5e2a12bd72609d225a39b859eabd2715ba67f0a2534f2f80244e000a0b58e55d90dad52219e6f732736dfb019184b7a260887edc25eb08492d3b4bf9a58c461d9c35b8b87e7c55b69b6374a917535083c1c62f9cf419f073e9f744ba1f945324520e756c57b7f374427069f4b928df307f33b411a32f23b5df87561759a2d471a25912625f1ca030a5bd4e2ae82eea4ca213b955689b5ddd52033a783bab980a05b235738ff1b39b1cfe98a0afd212cd453f2a748dff2a55e950441342892abeb949bb39b20501c84512ff9656c8aaa18f88babf5bfe19da07cecd04bd35aaf0e96&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.272419897Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/d, Name: checkpoints/4800/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4800/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8a97c7e80de6d74519d77bab21c764584870cb2dfcd35a1c7fbff4cc563325345482254c0c93069bca94494aae9796ad698fcf01c867ab59038c6892fae594413cc7046525151acbb5bab1dd1d7e67330931107306349a06731f4635b90bee40539ccb03aa90095289b0b3aada18348c22adc56f2338a62a6b53bc87c8448b1e1c702723ea92d8131ca290f4f4f3ed6d301366e4f631808e87f6aeeece86a79b80ad06b3e5a4fb02ccbd5cb7ecf418b9c910ac9623642c67ff3bbd1d109a6d628bde69b104f505199bbad3be5b5ac75285579348251d9c8b745ebe98cbb42f44e660fca2570a86316f49a696cb2e37606d87847d862bbe1f0644bcd27153e350&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274056463Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/d, Name: checkpoints/4096/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4096/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7037be5a00f45ae1b8316eec97c249b8d71fdb7553274b2c45b7aef6b6dd6aa29dc9f31b740efa263de864372e112afc687a339ae4c3d1ee4e8593b5efa7b2ce023eb62c06947e3b5cfe08fc77d7b2caa80d0f8bbc576ed881650d5ac199cb84100ede0526d5e8cf98c82ea6557b44be32d89404f86eb1e08b1daf0d480ec703f84c938ffd355c3420a1589da93602065654b76fe0fb1bfde4098094ec1d4a4cc176fbb93e1a65da6b4acdf4411a0ccafb85e3b170bf7f9e7a3df42a5877a84407d163477059475c357f3087bcac8175825982ed8b02aecc55baa7b07e2c5072f213dbe236192c49399256510c8510a126c75efd9585181e37fd630b3e86b136&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274073525Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default, Name: checkpoints/3968/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3968/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3c5770f02b6489cfac947e102c469e0a130e6f8b7b2eb5182be276eaf403556888f475be43732c3d70400b4160c7e3c3e4c7fd0e0b5beb9bcd7235fdbac5ac04d8c0560cbd22f32e40f629efbed3f1d7cd2fe6f2b18ff45f62196dc30b317fed1cff9e8ccbb50f267da7c940044bf757d7e3a10e6bb93e088e9b0e92ebac0036e5d66da38cf0ce979f5f356a17a5af1c1c72c55add67ee920d5530b3335cd59a1d70295299c3636fa8b2e44d9ae679511f67c9836bb2da59c38e8a68b1b5c98dd37896950ecc364476a1893aa32c2ce57c674f25cfb9989ac3357373397b3e28dc3ba6b214c16d9cdaab0c730641bb28ddcd1ace70495256ff4b73a1c61a0312&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274079127Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/array_metadatas, Name: checkpoints/5888/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5888/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5e070ce453261a343c59365c5827ba3d0cc66ffd7720177b10bdea895b7dbeb2c394ec10ebaa27bf441607a6562f80a73916e749f100950bed87e44597c76df72a0cbcd9e360086c4725767ec06fad6cfc9d06784a27640f0c919ae5e3524448e352d47ecdf618cda4e52210a8bb4fe2dff532199d8274574cd4f145351d614695dc16a877371af1fdca10044c1b2a267529bc2d2711f7f87a2087e4831554f847eca7b868e14af880c9cd5ac661e4cab33914dfe2c4d632c46ba66f41818b59fba95cb8eb69614b92e71bd52c3542808fbd909b75b939591016508b4d28fdf67ecd228ca3a0f2cb03f32eefaa791d9f28878305587f75b7b9577dace6ba27af&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274071552Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016, Name: checkpoints/6016, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6016?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=302c230b26b66a76117a36264112258ffc233a1e6216d008d1ce6546f63febd658677dd70a1466ee890645b6aa2507ebff69f2b175d046fbc0bb869fce10d541a0db3a5f419c83f63473e913ec8797030f96aef2d08cc415107995d63c3027f561543696afaf97076a6a9e11542be8577ee21d7a0a1d58e3d4bebb78f0874d96e98c77e4caa61f84395532cd4cfef93790e78cd0a0e4727403de9e81c33740644421ab32f0eb15e4ff8aabaefa0d8a73f8249f4726a7e9d936f2e357b4eb38aa70f13d67523285d0599069443245b0ef0623d6a53ad4695ce4d6756264ff2ce6154f940e553305ad467a121b82e8081c35abac35015a3a05df7a8c32efe27474&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274070465Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d, Name: checkpoints/3840/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3840/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1cfc69b8245821209823e6e0932cd18d55ee6678290579bc2b5aea67444df74f2d9f3f0d915a8ce7bca9cfc1e5b7738e980661a178658c1a2284816e32f4d8aab2c740f09b2640cc8f5e501f946f45899d1df6300a655e56d7867a34153c5b32fa62fb73af232fb31340bac23ba34439cb25e6034ee0c926c2f3b896360c299b8f8675d2d984d297915655b3e530571b4b36049bd13ca4ac8059dc70a480bce68c6eb3ff70fe8beef67043895556d731a89a9534d3a4665f90d3e287c5b978f99bbb5878dcaaf10aa8314333b077bcba72c47a27e5b774289b0bab6c271e31ae45a373db5fa2edf930f81cb9114060947123d4127d21da438df275347fdf1f43&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274168738Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384, Name: checkpoints/384, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/384?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=705e767e4b9d0ee72007e4609e4c2eda14d9afd8db9270ea717f278c3ad39c30d7ef884bc90968411327f93cb164e54ecc5e561d7ba61bc0adc60724e33963cd7a265189b6096ba6c8a3ae00bafe80236f2e5ecf698adc179f264f3cec1636d5cc63ea636efdbb4ed7d04d9cb509d4f3eeacb352fda7e97f7376b734bddfb010577bdb4df44d8b67f79029a9697876fc6e3ae9510c055dbea1cc0652534dcf5c20c7e833cf73cb4d2c51bdf5b7075d797b3ff2ea1f01d653a368697aba0fc65a00aeabed83207e6fd4bca070775990a88a56ba37f39a9bfc0f0f536399dcf0a13a88efe1033207d45a5af7767a3f401581444295e617bb1d92b51df9194ebd35&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274058669Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/array_metadatas, Name: checkpoints/4928/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4928/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1be6da55ae781c34721c6b1bdf2941b669fc6ae2910224d2503ca938cdf05a94173fbbc39f5df77709419b6dbaea971149f18f46a03086f76f4b8481770f4eca2eb7f8fe910ea530279a8462923567503245cebfda4f71123622d7e1b59045f709f8734d74dc0a46af81d7ca9793833b2aa6348b613fa51ea8591b8d40da79408dec52f79775461a83c1c09fbee4b16f46500ccfb986288b50c219ed4a5b708597c35513e7a79c18bf621f7f1cb0cbabb0f2d7b11cda9c3ffa661b7263febc320762ab2ff62f33a49cf36a221dd94a180fdf5e1516f80cdb30fc1bed643cc3c013381602c39d84d4c4c3bec34236b980f67e99a83fccb7be9c22ba73921375da&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274200356Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default, Name: checkpoints/4544/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4544/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=052395f154b121961d6e6dc794d894bff84381ebcf2af1b517e84632e1519d7a5fcf8e927d4c7eabd3b2bcc909ba7cecef72ed2ec373635e5c042aa1875138248b557d1c497bf5b99a5b15bfc6d8bd508464dc68e304cef2618786dcd15c61fbbd11d32f120823e6f44992cd00622c97196428e8de6a09d59c397d7fe303820113a320266e1a5c1fa4448407afcc5dcebdcf57b93227e2e49bedd3819a80d43fb7b81815c4f42f170f49b56f3b524c3dfa0e2d79a2c169914a541419279f09542fcc31220e2a05bc6335b3027337d44610606f3d0c08fce1ef238ea2a8c2fbe2db9c3603187ab00ae65386e4ea14249427b8839c9494382e985389b163a992c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274248464Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d, Name: checkpoints/4096/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4096/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0acece29107f77bbf68ad5cfbd88edcfc25f25db081a8bae3c9fccca54d4b5e2a1e9d282da0662ab3106ec7fe88959931add5671cacf8b6c350fc038af11b2f7f25e6cd27eea3fcb6e9956f22ee989259a09764aefc1ff8e62230481964686bc5a7ef94f113e9e81f7b529b2580ed4371640447f08de39902a7e17bbbe74aeb42c147b4984f5a84dc3e83f7160bbeea347f5d83a7b99735e38fdaaf2f07bea829066f99528e4735acfb4dd6451b5dfd64547a632395272e18370a9cddd15b8e670272ff974533b274238d3c7c97bfa8b499efb3d154bcf3596eeab8ee77853c8326bfbe654b2f4ccc836128070ff5edbbd4eeed1574bb30fb14431ff6a80adbd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274205156Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d, Name: checkpoints/2816/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2816/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=507f398126db769da92514497a5dff35ada1f1f491c68cb10b569a42b965cabdd7e6437e55c4599a8ffbba54fa07790b6db0e58351e329aa3c51dd4f313ad2b532a4c4fc56757fd7724bdc8575fec877b644157476ffac55a60270fcbbbcd19e162b015e95fc1fd4622ac8d70879ac07f2ed152117a7a6a932608c4db840f04f5be2d060ed39328f687a5c7c06f20c2fcd2335ce80a7daf0bcde543e6aa36d8f0cca2c699ffe54f67bdb5a5810ac524f2667844c2c10697a0e27ba22bacf513a14d6f341b192917362217603c09e9c3ce98cf5fcb1dddc060cc28945962eb83ae993d6a75da244aa1f4031ac5769ba48337dcb2acb997da7343562be940319e7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274257009Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/array_metadatas, Name: checkpoints/256/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/256/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4e6e1d0a24493fed07c51c72bf68f4d46b8b8a1a8e668153477701e19b6b3736c9ddc2cc323a8773c432eb464c7dd1d7c190df426fddd376e20ae8b75d9f71243fd85cf13cfdb1dc9c687974f835f914f154df829b4a7ecf77bc82a533dbaba3a5fec28679660ad3c12eb69a11861296d0dd87cc44cc6f8bf784b8a2f7e6d6add06214e2a8968b24553652015b5856565dcdd209376ccb7752574d68868532408cf42f4b8996cfb5bdda4fe402fee1b1b39dbed91cbcb87abd566e64b93d9f5fad88448a4abb856d4cc0b97ad2bffff17c2adaf4d95dc42519f2904392a44b9705a9b0886af44014e3410aea1c5d1bebec3edd0da1c74af3961d9f31f0d52bda&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27426698Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d, Name: checkpoints/2112/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2112/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4151809913b0fd21524710548ac55463d7ebd156c6c723a002ca02c575a01a9e8e265010a27b2a3baebf28e78267024cf076552dfc67ab81657c2bb34bb3cea7be5e279ee2c7c6382056293572a6373abd58d858cdb2fce0ec72ab2717b663dcd934525c78f76dac3ecb6c855da4e43f5634edd29648362e7b2dd0d9305f31ed7a6bb48e308052860fb984832f55c36b3d9a1eada2256275886601da426febc830da55b958ba5320fa6bf3f835e31adc952cb69aee7b021966ed5f03d3757c0599e290453440474dfe0e4ac6f58090b7dcaf1ff23af2aabd569d9bcbe9d1ac92cc2b056781c2a1e2ac3928d586dc1444b0002175d9e05b81b987a0d549f3de1d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274349093Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default, Name: checkpoints/3264/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3264/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3a3a223f748a0fb130b55970640767f60bcce17e4cc071e7ce7f2f86cb804af82a2743ed25cec59dab38ebbd730d41044e929b6f80a36b47c3c040f62a3f3b6a1a268d8a3881fd9c0a0ec6f48694f6b61c17e900da521ee8a435ebc50fcd635b113243f1ef7ec8b9bd5c4a4d136b50e1a50dcf15dbe0c15a878637a849b10647316404063b6aba0ef9ed366b3142118f6f58d3aacd18b94aba987f3a0996ec18df5362c2e823ffd9da0cc4347361a7289b05af86d4432ac4d8fa2a3accbec6b4159519ec5518492b71b380dbce3c92b49c099e1c58c2f322a7aff5582fc5a3e698d6640ca8d4f0916094fc91a5fefd5b4af09e8a8ad2ac82c0a1243a2bb739fb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274831702Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d, Name: checkpoints/5248/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5248/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=51966ce2c015903a98014da593ecf2c2bbb4ede833d2863066a805d82edabceed53810e38badd1ca139b202080ba8eb0c3b0ae19ffcebbfe5831c3d0a4236fc8878b3c4f5baafd77edb21834303c9ad9b38610b19f7597f711212e8a03e22970ca4ca064679b068efedbcb95dcbd26c259d1c4bebd0db9de6ed2382af1832bbc2efdc1fc03be089a7aed0255d33d7ae81624a230acbf9c13edf75231838c5e7bcd69d0ceabc8b01484aea853029b3f0a4d3c38cf7d773c867f9aaad0e0eea27d54cfa0ce72ba10bbaf809f7e1747a634c420ecda7bda684f42a998e5e7e270c51baf946ae8688e5e1feecc5f2e5f0b30307519172b1b297a8a4d4ea604c66848&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274862565Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/array_metadatas, Name: checkpoints/5248/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5248/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=bd3905b65aa60ca8001b01cf81b9096d4152950fa85337ad0492ab1e048106970cf58597062b6785e77714015876ed4a38a17ab5129a0e834d20acd0e237ca4c2a42e167ed5701503a13f0652ea4c0e956a2c4d0eb007cfde1d02d1776910d61d6174f03f225fb6f8d95cd9b9dabe51b4eb521484b98ad647a80a59ab3ad970f3fd268b1cb008d69b9016559feef2c7143066abb086f95cceb6699c010ad7dbe19b3954f2b89781dab10aba70884054980f2bfc2df8daf87c90b21be4bc7ba2cc5187f6ad6687b43191420be7000440b086be8df0585c832cd0fa87ec6e5095d87040ce4aa221840d7c970f8d5a179f4616ca658629d8647e489d0b6929fcf46&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274910256Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d, Name: checkpoints/3200/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3200/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=89f8f7c98aa9d3702bdd193c9054091ab6682edf6cf4a5bc942f865dc974151269accaece7a788ded75819fd5c4a7ac88a00a84c86dc210834641f6f40ab231542562371cd1a5cd44f61d3fcdab8496f5b66ae318741d812199909e6ac611b63d0d81e0bb1129f588e45e145251894004856f135ba73f8fe2180a93a310f3872a02d4fd6ebee27afef3d668f309bea823d626dd6f48c47a51b1b9884208ed512d534d09ef40b36d08168a074f578e282cb263e1643459ab0a61af06d12a51dba81e9f52d62746a5e4abdf36cfa60d1b522c4d3a00dab402d94615c7d7b538d9c82a949a411603b4ecf7664a69e0bbb7b240229398914862b54c6129acf5c925d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274964489Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200, Name: checkpoints/3200, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3200?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7738e1792fe3f54402fc7c553e5d85f909ad347de91886029b2d2f358b0f097ccd6d086e2c4d4fc5b3bbaedf7918a51197866d3e466c51e1f464e60868dfb199e08c45bdecea0c0bbd0fadb42bab4e84f144cbec61c8ec2c32a778eee4014cfff379b28831b879c803de53132e1804abd422c89af2c2f208b70244623aacf6bfeb83c39310b2fbb47633f6711fb4519a51a68ba85a5730aab936c9571faf528acfd7010db24a94d2822090087f9fdfcb0af3ce49dbea7583472dacefc9ff3c38cf8a27d7034ad493cb22353b213bd57b525196bbba919e0367dfb609dc0f1b149e93fdb0dd1dc625106fd3edec3381b8b6861bf75203a6d8e86881aa3764d02e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274985094Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0, Name: checkpoints/2944/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2944/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7ff54fb3316e07e7dfb781836496b17ff70c84ac725fb4f98a0528aa88e4f72c3639f36b399a8a0218d882ad16027cf0b13281f6833a4fefe0d1711087e6ec40daa47c670bef6ca4ff08da9e0b9f3e6dfffa48c9a8cb0dc19701b37eb5b2a0bfe3e4352cac1ab8747da556a53b866c8a6062a7553cfb1132bef8fce1b912351b6ce046c722f06d390a180c74937aa79463bb303649b8e7bf7392e8f05989156e377dcd6cab92817d61148d384da1ff97d33388dd508e13454254e5d2e2b404fe1ad9720d62d1c88e7cefe9cac77808fec8abb0c07783266d980f8349e6a78d0c0d10111c0c3b18e900525c1b38928a484bce52f1637e80dfb05da116bbf5fa95&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275015278Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/d, Name: checkpoints/2752/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2752/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a51d8820a0c9a59637db5ca47ce8cd447fdacd5fe06288eb8e8e04710e86cdd69774e28d902149cd54e7f8c24702d8540be7bff2bedc069489260eb5e4d86185d75b21a16e61df7713c45de5a567bef02080eb106ed34b98aa34fb3d00a80bd2bcb232b2fecd3e3cb12511dddced73ecf76178ec276fc06875b367e1cf575994ba89d0d63fbbc5669a2b9f7308a2fc2455dfd8ce92a7fceb487f00af5e763abd3ae430ea7fe9926bbebf9f667db66d98f004b8b1455e97627398170da28be908b32a6caa216652cced88b076588d4f8e8f840e539c5e2bd22c8614474ab0a2e858d2c2dcbf66510ab2a039770bd1d078fb498b044c2e664224236f47eed2b9e6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275036899Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default, Name: checkpoints/3072/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3072/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=880249221879bf5af60d0845f8c480e76f751c6acaf7e0a6ede83b1c5718449f1422f1d2b61a607ed9be241be28307d999b44f0d64aa586741724b18f4f78442fba53ca482ec4dbc46f700941c82eabd8039a34dde1a1def3e5fc24fc84521d8cfe265576086dbe757414d6029c72ba827f8fe61832a56a6aa70668f907a3a9176e92427a28e569f0b4c5a7177817aae226d83eea29935cbecf5abc6294df539f8e14acd9c887897734515587ed8f6123204731c0c6797ab8ff70f5f542827eacd01353ccc259a631040ef8c53b7d4fac5638a4885ddfc784b6292a790ff29229c58c4455b43212104716c546ae4c4c753d6a8121901885f746fe8560263734f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275039762Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0, Name: checkpoints/5376/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5376/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8ed6edabafa296469c39d641d07b0c759ace372ab5b8bc7190a622bfd667cc97aa369f6118f18d5b0ee59418704b201ba7ca7c9fca6e521f861ecd2c68eccff13672660c02bbef55342c6c2ff6412f4caaf2652e840e910657ad5066dadadb706f850a3c6633b6a4840a86ce73fcbe582224a7ed0e7a6d5304d32daaf189f8543870dd19dc66415a54780d2cc0c579b0a8784ab3e7fcb5732591e0e83b0e1f6d15fb1228177e1b4725c9f31256260b2d5200386a3c02bfc8651be17a7524f9bb3c05e103ecd3f38abf4f086da3ce42fa498357de443df565fbdc1c1a32c13bad4580bf8dc3c09ec74a23b7136a49c3ab4eebf92cea375bb028b12351a62a1738&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275071221Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default, Name: checkpoints/1088/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1088/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2706ff5f2adf8e3fc77902cb3e5ab5dea3c40e8b460bdbd37a24e55117edd6fb8f3889afca221c18b39e6cdbb15741f7b9082d16ccc2a6fdbb6b2d7c5ad84577ee6270c7b3607a4b6eca40ab25463cd5a48a45f9782fad217d1ef9497bb00c4c0f4f636e30559834f028d6141483e2da28bfb40e05ad563c63bd7bad207888e39fcb7c3a636bb28c3131a3e2787749fb5d1df4f08bba84596129e1e585cda1c7d714af95046d50e19999326ea17ea710e423674dd0087ed232c3d400d40b96376c72794365bc0c01ba5c43084ec3a53e2556213a26c728839199b10010a048f80b00192206a184255eff2a6440f0cfc9d0ad149099b6bb9a75cd94c4c26f13b6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275092411Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0, Name: checkpoints/5504/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5504/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=019b9396d90ef3738be21bbd86c16e69d35c64b74bcf2b88866bdb3d27c7952cd9de09b22742740a1c60826061936caa01ce842cf105abb37cfeddc0a121b1c4053b309b8ce99101819e8cfa83453d1fbb38403629f44ff9ede9be82eb79e3304c0ea0165fec5fa8cc3266f4d7dd581722145884ce9839634129bb62db75d943f862f1e41d3aecd4e59669e506c28f07c0692961acd354704eafe5679ac71990d8fe21c8f173c80cf0e1db589db9470a651daf418f45e3b9700d613d386a25d7918842f90276a1bd8832a0a01ce4fbc020c71c051ddf9477565d2008b4220f615eb535d76d448560aaba184e8c6e5b056354caea0f2820c29db8fee0a800f54f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275108191Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints, Name: checkpoints, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5d231a7f4d4e53e28922913af8892dbbb19dce8c26c4f3e147641da16904c5171451623c894a297dc709e98fee1e78e8cd20c06745e7c4f65ccbb54ef66e3c959a8245ef8df693fca9fca8f87791c6a2510e6dd454ac73668e6958ec8e2a0df38893b9e4a09fda3e9f999d60ef5cce042b5bbb5412b0dcee7fa16da720e5c279e07c6fca5efd9177ba8e98db38eca3af0ceec7ca9dfa7fea2585841db1090f0f0acaf461e907e683e76a228c3516c864636f130c7a056d3371416bba3a2ead12d6eda38cc32baf19d050f37969bd53df21d97ca75ac56f8d49065a8668a361cb959dce4aaa172fb937a78f16ecfd2e5b41392114f75c11f5a4915e089703503c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275140319Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d, Name: checkpoints/1280/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1280/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1fc5bbae754fbf8c568f0d6bfe202657466e3cf393aaa04bca68d30bec01a08f27217df55ad73e47fde24e1b49f37709bddc558148c349f40574aaf58e25da28dfb3e2c8678e14d845f87665ec5a1a9eb2e68e706fef77899a333227de122580874e896b9914be47eeffb846c6e8c59c6b6b74a9a0ebc93e704271513932cb5f4655386a981317d4d949c98c856ee478cf57b082be89b367811140fd263ee564d818f06d2e7c6e1a06bb66e16cd8a4ec582bab2f8e06afc241ab8131d6c666032c504f22870541283b7fa9af97bd7d5822fcb91ec2cd461b3a6102fc6d1d13a8a2794a93f447696c432e4f34bdc10c23ad938c6dfe91505fdfa2bfd92ef7e8c7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275148989Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default, Name: checkpoints/6016/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6016/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=244764c5bfa533d1a360e6d4db221854c67b4359c0df7f887c9a7e9d73a69d0e5e23d8b5e04b97e8c7f9a1ce2591ee73de1c258d8c686713483be59499d87e88e6ef6db15418b758049327c5514053b3a5a20e39ccd1d249bdc9e7a4fa69586b661cbefb9d97bd98243c35d9395046d3cd76b6b13fd70b3d633cd7085ac93f912e43d7d07a8be79ad30894a7e2b3b2d0c0599049b6b76c0d4f24a80077a2c381dc85653b9b9a058e1318ad612d6342cb18369dce6f27feab27ed9fbee06e0a28ba65edcb38fc27ca0b21e241ae32a9213d6beaac6cb13425b94bca903fbd31e9472334341d8d8e345fc0a1697a893dd0160870f579e174d3f417d52adfc7b8ff&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275164615Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d, Name: checkpoints/3776/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3776/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=175d9f043ded4f8d9970877522c75b1410f57a28817dd02f768d97ac646b9a129a7693cccf0e4cfb4b039ff700e47fed44449961947dca308aeea7ab465befe46066dcbe543c227291d91bd498df12052269f36ca619c1a4f4a023128bc0572c5e66ebfe5155b668ece242af0ade2878b90a5be97f357b79aec41dbe049535d7b00f2e0560572f57364eb1f8e5b4666b61f194283bb31452e1d49545d5f29062df1eb93e0fd9ecbc018ea9946d7fdde56f277400d3f4785652514fbd9299ed02fac2fe37a588993e905d4ba8381dfd17b33f9029afcbce71d661025cb2a6a226d9159687d1dcae778e20f77f5f5bdce74b5e8f2350cabc1174afde1f28cb8674&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275168747Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136, Name: checkpoints/3136, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3136?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=39d2199705862478bb5ff9283ea8f4479141917265f96c2fe41624c4bd94abcfffedf68b526525e585a93435f2441542233e14886a5abf39aa9c5b5f2df83beccaab5405d15592ce3c75747f11f5cb20b7f04695e0a91ac2c414b08589f6db940c9e5ef0722cdceec47dc359e740400189c090b14c37ae4ea35353656b0584113f58509713f5a678429e151a0c3ae453bf7636b3abbfb2c01af5a27fc5ee8d2fc5d47fffd86cb6928722dc3c1505eb84950978f3dc694f4c4370be9cea8caafb95c5eb0347eabca10d3ce8a7a00859d0c11dbb116dd55487c4bacd1e9508403c0c28d74371b9116ae2d11f34acbe12fe56598a02c6c47d3a95570a00691f2e53&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275207946Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248, Name: checkpoints/5248, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5248?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=622cffaac1833f58cbba7d1bf730347fb24d82e95956b905446ff20a41343196dce3461553b72631801208e55e5c3bbd5184101337246713e1b5f4a0ef856f4d7676c0fd02c0f2906944e178abc6650e770ea84a9dcb98b093e53735c1caa1356075b38e2b9678fc9d06368fa12e76182848ce35f4e5f9aa78df98742beb223b5edce6a99a7da1feaad71c05fbfcf6616fb38c95dbe8f122ec8d4885745eb91d4d24fca4596d24bfbef732e5eb522d0a6644b03e75543143e080edab01fc7ffaec882ce13f7e5023fbd176a8b4d92f58d1c140efd7a0834cc704689c2ab7de5ce33a3b35789aac7b4b4b31c5c6232d243d6f4c530ce9ecb5b81ab7b90ff5cdd9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275223088Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/array_metadatas, Name: checkpoints/4608/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4608/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=214eb9097f5cfa77f5925934c41385d9fe2f0ab7da8af5a5844e7c8113805bf4cbed9f00998e9f0b6447174f6caa947cb9fd1085846a63ce341fe1092e21aa92ea39c141cd585d151e19ca50157349b4f33ecedb05ff20c893ca3025f966d7d1457a85e25af3fa0dcd86628642c95bf9ecf271699e57a3921c1d3fc6a462e96cdd80ac8444e22f72866da1c7e726555dcc569d6084bb24640929911fa1181eb41ebedefaba6ec5a9e5cb00ac03e3c3a94ef1ac1513e22af41d65ac2565d965c866c781bd704bb1283eeef157edf8a83c0e3c19a85de2e8aa3ed2ae8e01f30a02d27abbf8f5fb37650758919b36c6baded52eb17484e2a2878580c95c028a3ab6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275237075Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d, Name: checkpoints/4480/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4480/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=77814e02f9709766bcba9069e4776d8a0a42a76df75b1a895d8cf2656a752924324364413267562949b17baae9827288fe7859b8d80af13844a65c6988337c3cff745ac8533b0dbfa68f3954e4a70641abe90e173cbdbcce14e5ce8ce2630813ec3ccef5796c82925e72539a52c6cb0d46c8926a01c765f56b0b81837807b39360adbbdfd34ec8bb2e91d340b666a4bd273c6c1d72cbe007eac9493a7615e84b7e60ba948e9280984601230189311452549065700a4858d2080d3e5ff3ae2f28845f182f53c4c6558d0f9a830c545e3bcc49ec5067d948e8c9320c1e4b53be648e660673df06a41c20d9ebef1a622f8bfa024ff1e4839dcf425429f49352fc12&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275280504Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/array_metadatas, Name: checkpoints/5568/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5568/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7cbe8209693f89423ed17f7f9a8976365da3ee183409a1054495196a1faeb1c15e829f59befa6367fd09274911bdac9c69d2c84ec26026eb29e6e43f96c064b6f2eb4a8fc8f4e1d7ee89de7cad5c0102e2a82dd7eddd1b5586ae76dab57cd489498817a58d0610b613bf1d9ba8f57bac6f45efc19e3b751627ded257365113095b365642b15e0868f36245845b52ef6040f43d9006de25136024bf0607bca24d49c13369f2808a8959358dcf6c8fa9241459f1ec492353aee88f54c73a4c356636b75126265dcd954f0586730f5623c5e9ce5327f2a7d077fa379c394a708ee7b92c9946c3ef6bacd26dc3825a0574b60aa76426a9072877a0e9dc656a59669d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275292318Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d, Name: checkpoints/896/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/896/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=674d05f61b8f256697ef99467d6bbff99b33c729763f2dda2d0b0b0c2e2095624d09e20a47b591921dbf9d1200c8f4f3a3c2441465db0ac48d5df549f53685048f6e4e57a376a16a783dd1bec895cbeedd2f9450ff8bcdfce90786feeb29b4cc959f428a85983214a6fa52b05963c03e77221cc858e742e406ab8839e53c6ab662b307d9d3f3b3b50c32771aa1cfcb42ccfff072286b53701784809f2bbf3c5ba3025b4526271745fe3f5600deeecc7eea775ff722daece16a4e8793253e04fd99711be9a46b42e3ab7b17c1019f97aea5a005434411d11fa1eba2f8562abd63f9bbe7ec185d8a01d404edeb4ea9c9b79251b8cb534007fd6f6cb61b66a35b94&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275362296Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default, Name: checkpoints/1728/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1728/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a620975d967458a9f572bc4f7c52cdfeadbb96a0b148a2ada75e5f60e234c6fae1757739416b5d5cae3b078e8467bb85fd0cdc52b67b152827d75d524ec5213674e934e87d4b7ef572687efedf65cb0a6a594b5b041486fca6e9d49e1e837aa8a5b51e813d1d33f613cdbfa1b44fc7b2292aef431817a5119d2a744b141f7ab553d563297b1010358fd44c03de891c5e8fa89126d19e9a4d2543ea19288f2cff0f8c16855dc12d28becfb5edc7a161f78baf343aa1857934d3482fab519e6523b34acad5d3e7546b7f1cb36f149b6269da5ece64c045d8e793b7c0bf32a97ce3c6af2a776cfcc3e5ba30f10c22193d273d7db4ea5790db00ea7a467cbedae7f9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275370585Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default, Name: checkpoints/0/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/0/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=70c364e67b2a6966a2fe323149084d179e8ef8106e5720c0c7e055e1aa69263f4a6cfdcad2f324dfc53f0655c823a29969c73811cd7cecb7cfd6fcc1b580519ed9165a3d3a654262320652f199fd10afe738559fc836a816f41b543262652b514bf22a9aff60e0fdc0c17b88d1b2583f947e449619e6ebe44811e94fa37a7d146d7bbb66d76e8c9f705f165305b26b35d2143386d94ae5d067bb93450cf1923aaa5639baf189b0f7b5b6bc883e6ad37573abf332d1fc241afbee430ca413a58a46fede0e1c4d93840da58b57f092f689626600a24daf4f07fc0a1cc2b73be124eca6f8dc00cbee2f3648c51064663603945e40ab136f868bf218cb84fe33f1a0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275386434Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/array_metadatas, Name: checkpoints/2496/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2496/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=702a13846f6edf768ca0058969c40e5930b99c11dcd542456bebc94e432b123f2c044cf0a42e9292f1b64d634bc046e8c9a6630de364b7f2f80a0985824fa8663f6addbae1beac33ca07b47af17e751adf86c604e9d718ea5c7808a1894798ba023b8e8cae5b5db198815c3e9b739a7651436fc3cef525bb63a5aef7a539c08fbcc57f667a461c7236c05405123b978a9f8697d052579ea01f5844337078980d88444bf89e18fc368d341596a04282ed67d7f12152c0c9d67d338c5827a5d1bb384168d41388e6bb4d6eb482f9b5137c8579d9a9f1279123e7bcec0344ce5347b9754660fb177478ef1553597e8f49e6286b2ba6c36d1d9d9d8b45de75ca4b2d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275454668Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368, Name: checkpoints/2368, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2368?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8a79e76c3e88e6dff27f48a27aafccc1796483bab2b77c9be6b2992fee84d6575958f3f2ae8c0cbb074728fc60bb1eb3d26d35f06bcd18ca1c919c38c3c42077cec17005ba05e1aa2de7f0e782bb6fc35e8ba54a0dbfa617490fd970a44ba07e755aef781d4e532bb4cd7a697c75156bd4f70576e2f089bd63bd9fc9b5d69cedb5f05ed58e034fea159650a0b07c545e5a21dc2b9e66c4c84cc6f5b05fc8c998d4041cca0783ea066501eaadfb5f738bf96977e80158ef9d6b41a4d2ffb4c09fc1e28f160405705d467d2c43855f185fb7321b9e382ee860be94de83ab07d5f0183505614687a65be9d8a2638f70d66f9a1bcf65688e22aac3ad223d8dda8ff3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275445697Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/d, Name: checkpoints/1024/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1024/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=78bfdd46f73253aba37554615e88e223f2e2ac083dd324b2665cb1840b67e69b4fbfe2d3ee6b49356d00bcd4e6adc3fc32789864f17ac003b6958d1acbd7f2bd0e246f0ac5f6941b2860ac00a113eb742d9b9df7305bc41e3fddfc160e5ba34455f01ab3c268b4f5bb9a97ecee9b13a1d55eb8d66430b581869394ac386c5648ffa4257d530c751f847eb4fded2fed8efd904f1985025c7c3d729073ba4f1d46dabba3db2637c718928c2623d48c29e22d4b89b5b40deff05cb9b84851ef040a0e80805c4cdd892846c07ad3a39b59222674c64e49700490732e8edd69101bd84ea9f39a8fda531190bf162182151efdf0988970d243c339d89fb4eb8fc8fdc3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275501928Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d, Name: checkpoints/2304/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2304/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a403b354fc64c935ebbd3d4c31a0a88871ab24dff81c4505b745d9b714bae64633a1b52f71271aa36453d7789870364e9e8b600f5acd1b5f3cd54c00b7de8bd9efea42a90f26650d450d67ea283bec760ed7cdc065de46f311611506c3a9a7fbb29e58aa0de4239a67cc1470a4039efa265d01ec93faac5876df9b8edff12109b9697949f034b42891259e61b5af094d2739d2371352351c9dd5ba984f79cf7c09fd6ff82d558828c0bb5ec9b347c45aca2ab83433ac688b1f7facfb4cde6bcb0d4f29e7bbc1cf897de331a59b3e3ed380d1360e420a2158edf1efb6632679e6154b2a182b96ddacf35ac7aa99b38b2409719d047eb2fb1592468486b707ec28&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275536461Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d, Name: checkpoints/576/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/576/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2f69b8975ef523b66ba8bfd850baf1a73c72bed763ad4f957e24fa0479169e5f25fda130094b00f161fea385f89bea283f76f517f933bab94a4ae4b4fc4454c22895350c9f9640b4e7df7276b681e9c3c9ea545ac9fb70666a332e6cf4607764aee5af03064cf52bf314a70f41f1dfe9dbdad86fd3e3fd7a364eb03f4dca4c166508ba69cc9d807bec548d39c6df6394f3859a4a8165afeaebb60028c9b323658327444b1ab1b4ddfaa4594de4b604218914d26194031c58cea02688d51444a1f5d78acf4577be01f588481a882ebdac24e3d8b82001fbf71b77a438280eeee9871c27e0d6b676f223507061068daa19e7ca4c0fdfff590c3b99bb91eb1a0ca4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275539196Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/d, Name: checkpoints/1408/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1408/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=785542b21d971f978fb68e03ef6d003fb14f7c50fde7f7eb22a4535f8bd4259e1d6ef6225de8c919131d7e5b7fcbdc90fed31da7625cab005f5691aa4d7d35a59140e249096697fe068a3db403d04b10603994e356f70cbce2917dba80dc6bce8b254fd2ae512fe8296501aa089a4fc93a05903dc4278730165507619c3bb08972bf7ce69e52167ed887180ac755abd003f0a25b41dd84d65315e60b375cdba85f6f24217724461ae6cb369db6e50aaefd36907e16bc85b0cbcbb72543d0e6848b4b21af357c2ae6ba0f28a7d4710e7d89a1a2ff053216bf417b459c91f4863517ff6d275c6b603269f86e51a20b7ef53bccebf4f7899c5cd2620dda6fc0c1ae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275535279Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/array_metadatas, Name: checkpoints/4992/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4992/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4c00298c0d7928c003eb417cb23b94a7cef0721e36c8abc26c67f5bfb672f3dbd7f945903ced474a89d3fcfb119a00c448caa14e2f544aa95b01dd452ca85eb44515877d8b1a2e716893f94d0f56ff6d18939ed98ee52770636883a74804f445861ee0e06a129dad71a42b9def0067e28807a0d06e58a6942fe87ee97e56db41b3d1e4ea42087eaef060ea8bf75e5449a2e4cddcb995ca4b55e4cbedb3baba4700859bf2617cbb030f68d2138660cd547f143ec843871b715e98bc87d1b3a8da58e5339360dbac626ad8f43484874b9f11564e58163b6699d7c547f755d9a999850593b9365cf41b92c2ff9a965b75cbbd8d055589156774bd5915049994fd21&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275402335Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/array_metadatas, Name: checkpoints/3072/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3072/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a1a503f28b2ee6a706ad3c821a57b47fa7c81fe60ef5d8a8f8b480c0ecc47eddceaab99f5a1cc9604cadaa1c107e481fa5d3f62261525beebe1005a7dece155246979fe14f7689ed052f9c9d4bd0e531753b7e72a7e94672565503b95af0baf074c71d4a06a7d4c13b9d650c1220c3c9aa3faf2da00b06359cdb3420357fb0a88848e8897bc7df2b13ed99c013815ee2fec58bad3e36f2af98ce5a610c8fb30a9c9239832ee04b62c4d2b3b0c7e0992c973427b830d2649c3024c9fb63620a114a365d05349a5e2773be99a3b27a15a3dccb2bb4e61205dc8fb116dcce44c3390d39e432e1331813df9d5ca2360ccba1ef96dc702162279f01acf8a42c5d9723&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275224196Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d, Name: checkpoints/832/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/832/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3e2f1820c3deddd284116ac791ab85bdda5e65f3f8e4ec79ba62de385c05940ea069eceb12a4a38c38bf19ab8803d8a761aa013357697398cc29540816635baad9c5c7edfcce35dc1cff28710cc937bed023371d2b143bbd1f93771345aad316848211fc4f27c8b03a452c7cff1d89219537750db44b17b5c9da47e5595093aa0fc2470e1aff09c0861b54952629fad6f9ef2bc0c39f4d89fbd11d5348ee7393efac508dd65050a1a92af64a4db31c45a2c70a5a15f5fe292499ceab5a13da3be3eaacd48955b6d980c9b9b1686ca80d63d942082840f96fb95704104cd5635729b8a1ff48f45ded020731df8c1b74bd1a146131b95b153ea4de234b4c6e5e84&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275687825Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0, Name: checkpoints/4928/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4928/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=97bf34b3ab436d5ab9017b783da951d29cee80f8ab312076efa7f0e3efae2035f0d7e6a8542d71641e8b999a4f78df7fbb904f727188fb91aae523b2072fd95a9de877515982887386f6b8f36b32ab83645e7aa7e1bf47441ee23e4f0fee8e07ba3cb71fdf2046d98bad573d9860e5e55a0c82791538ff6016544895f26ea455dd1414e41cf83b9bc315671a9559356680013806f19c5f159f5a007f4f4db2f1efe610b13ab4a90098304c0e494ff0c0749db620a91b19b6bf1cffee9cc420866db6c0dd71fddd2325363fadea2b204d5d146e1b0e4b6ca7ee3b2ce1a8bd1c550f161db1a48b9acdf07887cddd491ec8f5c68d557aa9ee08e9367bcfe4217bea&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275067942Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/d, Name: checkpoints/1984/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1984/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=243ac2180075eb0e706138208295d6281dcbbd82455784ac2fee0e1d3b9775780230364ae11ba32380f5bd8ed10092d99432608eb61f126a979cac3fdcd24dbb71132c26c8f9933940b447d409d643e65abd5bcff2f214cbb1358d3ee269a048923b08f01ece336910ab0bdd0985a0dad22a1177018e9b733a8d818e4e5815d3d28675def51015977c991c14124b3e49230c26aaebd3c8725c2c039892ef9fe7316361b86b91aa25c5be9a6f02163198d53b127bde95959a8d5a30b1e7dfc55bc5c1be96edc515e48a011e96b2f422f34641a92e098e79e38e928262613f28f1125586709ff8efb0f44fa4caa61ad2a446d2e0a39fc204ee61d9d79d7c843ec6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274970436Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/array_metadatas, Name: checkpoints/1984/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1984/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=401e06ff75837d877c9efe835d2816071ff360a1f7277b52729af5eaefffa89acad59308bde32ba4093b686a81ba5eca77e0c3554270406940061e84f7c1a3b26cbc68b66625a4538e7e48c057a63c98df7e28e70f1ff6738b5caa7c4c97ff9b5fba58a9537eb9a9783bf633afcbf56a8764fb04db0cf61c0a8160b5476e512d5c034f7bd5f74b6cfe62e3d8ad5922b5b9c18a630c608b15f8c0c03312c64037fb9a0d46e924be11011a107e7c7bb306c4a073bda80d4f3d749e39e2d10fb456327cdb1ab66fe8f6368091d62b975695dd729fda3466567c01c0d6d0bf5dcb8c26cf7b65e70a0728dde2258d55c2ce88b25587fc2decc625d47b4e4f19d9374c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275112609Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/array_metadatas, Name: checkpoints/1408/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1408/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=056644fdb6aaa7f7aff823188527196c27ff906858e9a67124fa20e4a0e50a70788ed95f893ec9158be540f8f3b9e319ca6e383e2395b42f9e17e4f53c912335cb721cd38a2f950c47f072e594c37b7836babf5a9fb3e5a9ae885abbd3d5758023e535d34bd4278ee4930a6c0e7921daf291d2b3e4081c6a05bdaa9306794e4c2409bcabe146debc56412b0f44eeb3628cebdefa184118868766b1be906c44b4efacf82fa16089154311fed0905c13f52e27ebef2f8b65109ce904141c3a31792da78664ff69dc3b324538d47c9d377479b880b555f9156b2be040943ea01a0b35c77a3fcacc2dd7aa49663dbfb64f64a4502e5cb1bf61e20f982c2bf9a3b2eb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276599448Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default, Name: checkpoints/4928/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4928/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9166a33ecc90c74996b1f6a9d85db5a4bc5bbc8d4487013ac8bcb8235e60c69fdaf902489291b4f7092728490175c6057055f7c4c992e7e3a71c7c7ee67aea313f81e771394b27edf8dc428eba5640e0b12c82e9fc3b7cd0088c543172b5abc0c0a2bd3d4cecb17a378a96897eae2c91748da813e81e0f195b64712302838e907251a6fa19a7eef0dbbe04af1b3970bc60994bbe0932f7dfd003b1c755ae3915b0ab1a032053d5276bf05b895ac749e38cf2e898fb548e802e8c0b3fe95852a1153f5a31a1ef265216f37a02285a7bf64b34b6ac90f207214064bf927dceea3d4e8059fbfedf2641ee49ef443ff00ff7733c3a4f5d1152c3ca01e89688c4c055&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275078345Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0, Name: checkpoints/5824/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5824/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=97022a88f331e364b4388f487924138ba05e2cc0e5007549b8c3893cdc586450e6ef66d5ddd09bcf97c71efb0f9c03e688de82aee894f7db0430f34c076aec59d098b18df2bc9f57bd3679c2af549d4eb4d54c36477b232b1d2f52a33d96b2cb6dfbae3d1a8f39d9ff2bd0f7ca1b0b03e71fd832baa4ef17a3d6215f331fc325ef7c429b43c565cdf656435b288143cb956d9f89ab643612c57cfd04bbd527e5d093655c3d59fa264f304f9268d54cc7baa34a4226a96632d30619fd2d44f5c56a165b70f3fa2f5b6d6bd53ed0f46b7f1e473400a559ec56dd45982fddc01527e4472ebca4f64a24d74bc22a939ba145384742cd2b830619abcac0802ada2f3d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275713863Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/d, Name: checkpoints/2560/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2560/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0dae24d7dcf30dc06ff02e5c970ad9c21de925207c9fd2bccff0847de6d776da27a5aa1c61d352ec988125fca03a3fb59e6b150335d83555f9f8bef258ec63ef33908f7627b6741d19e0c62ce510fb61724153fe2d99cd553a45f40a9a520153f00b82c21f3698bbe7a14bfb69404cf35e90e40d6b2f301eae0c699177018124a6f0b747e700401ae350e608a5801e36bcb21db85da3fc175633a54c99d08522a37b23e85c173d9204ee9e29333e6ece0512da32725e475c47fab99e508649b250e26721f78e0872d07c610a9a7b10bf38d6b0e4640956eb20dd39aa15396d817438524a5b677c2ccfeab15c4ba2d86456370a3a8824e1fb3864457d1e3d9372&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276602742Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/array_metadatas, Name: checkpoints/1600/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1600/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=541c48aee4e3d06245b9138271b28b3232e3ff8caa4748b7a9c926c72c4451a60c2321fe2939f54a6e0453db0c7feb6ade330fc809a3db37f217621e828d9640206dcd1326d2c7e8deea1eae5325be23f39679a04950b2b389046245ab5960ad191c274d3f4fcc85e10bbc3187a8ba71cfbb7f08f48b6f30c5eb7723be9fa13b215077d9c3b8fe78c8796a977e5d8fde702b9b2fd3439a00a19e6a6e21a4fad726d514e3fa026c7d5ff67dfd8989844752199566359dea820c34faf60b3eb85d945a4d5ee173dbcf7932caa642359b61eb0b487e057ef56d1c67505c567aa429fdb8012503b06f7e360dae21d55af98f92b35919c501cb3df426f2d5a3dd11fd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275209453Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0, Name: checkpoints/128/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/128/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3d7249060bf496e948978f728a208fa06821df592ed5b6233ce5c991baa44fc5ad807a586839ff5847ba2738a6a30eee649438e56160a1f28c61076884c36fbfa42298b0da8153a4d79bb074b4fe9d7ce0ecac9b636777c3942f6b5fff035f8bc225f579f5df17118538d9d53fed983d1dec053fe575fcb90efaad30bce0984924df037b24036214d053ea41cf2f7b1d0d99b7bdeddf667df2ecbd595383b4e5b353fe00f2f2cdc99b8ea02389ef6451585d882c36deef277c47110749578950e894db0b34f34536b161af3b14d57294480da5c5bc101078da824421b19e90e0018ab4a471d1925482322c5457eb25abdc62067a64d6cf9e079a3dc44a77659e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275533681Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408, Name: checkpoints/1408, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1408?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9c5e3aee5d6c4a95b875e36a4d71569312f5e274db06a4a7b39070c4037937ef99931ef091e1340f988bf3d3247706e61472da39f4d12f7c7fdb24a4b6db4598a9060dbcc43e144a473495586185958d16b140d365c44b7d69f5733e2db98f1f49361d7900b553944aee953610f08203fc220010286311980ec73349aaa5269a4d20d58629ad4b2c3d987a08f905edfc336d05f6c77ff0ead936808ccddc578b991e6a55788275aa25dd2e66beadb387141a81e7f84712686ed6243d4967240f22c6ea9544767f455438f903a976c2ee70f4825d974679d90d346665d35bb8ff00ad31df013528c9408841442d404faa049d4406119a901a023c80a60383cae2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276697269Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/d, Name: checkpoints/2624/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2624/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=086fdb8b9294c7b1532a9691d88b6a7a90029f79299107a25e26bb81ad61160b7c71e8e621eebe86949b15eacc75a674e0bd2d20f0bd1bfdc34d800a9ea572ec2bdef2df53583235e55c8535fa47534bef88d989c298ebb0f3f16094673056737f8673586d19e9a51aca9759056184fe6771248911f39e981a8fada150af32a644feba828b5a0c0302169442036a20e93d021de333cf08ddcf02c019df2cd403bd647634964dd2d4b77ca79e6c3a78166202a3fdc7b0e131ae8b0058da10940bcbb8c02d949c302ef2c8f736abab62c8588bc9222310ede4fcb29b0dd57a3047b0d7256ce8f434cecc2340ee444c31a3b7317a70b67b10cd5747960544c18221&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276708981Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/d, Name: checkpoints/768/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/768/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a951525086203db2b4a6d7001bf8c11418f699268c03b94358c3e5a46377b5635671fd546dcfd3a59a624ff690425b9148815524cfec941577bde6d169d453266df03f6021ea639f7876aa434724e98daf3b260a2e2b6a9e3272fe150c501a4be2248a7f5b1f3506a237deac53eeb2ca1aa4044dfb7551ea7cbec1914189cdf4eb8ddb862592a3ca7d9ab195be6a815620b4d9318d866396439415d21e33d98f2192b376be17ec9cf23ab97454c53b342dcae75b273a8ab29a54bd243b8e7f6217454f3dbbe0b04f1f82b41566927f8dbf325548b7ccc2f6ff149a8267c3e08c8ccf56f5117377b80532dc795b37ddefa24004599c51af8135935455835e741b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276690952Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0, Name: checkpoints/5248/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5248/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a3527cd6748516df34acc9ccddd91b3602338e549b18eb3fc7f634426ac7fc96d01742f37b57cff3380469fed141cd7cf43d48267006df027532cc4a4c064e72815eba5b7dd439cc77dadf321081572311cc6dadb3cbc4bba7fdc22a2d5b991c58c9468c95f61d2903534b546b3b018280153702597cc80d4b17e1b41add9cde2024793a62938ca647f3149a6b3eb2be671fb8100781ddc41d55f140cb7a71dcdc91eee95d650496b21914d93165b86c07856a1111cac3cb9c7f367f0e9e9a42c321924a250b816f2f91760ccccf0ba04838f36b65fce57e9fee605a90e66f94f3b45c6c14d5071b9eb6a44b7c80882bc875745ba8004b098626c99394c957ce&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276754854Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default, Name: checkpoints/2368/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2368/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=984db9b99c7c7c75546f5b037abdf4ccc19497fbb422f0d93aabeb95a807c519037da77bb5d342b582824e08e0c1af1cfa1aa254a2c9836764b8e218653b8aa013336292aea6ca8fce856d32d4facf41e96ca8170738e23f101725a9661e836792eef39f4b7d8aaa7b66559c25fd59be05f62958f9d07d2241052af169a2122803ca4d0481cb28a76065da984ca7e0c0f16dceccd28ce80b341508936f1c782de22543e2922f7b0ba6d6eb262e33224df192242d0bc23ac88bc2a2d9237d9cccbe2f83595bff8ea00df4e65f47075275cb03a8032506b25c3e219b78804e4f13dd0b75c22ad05788a7a41941e4e7a5d7b139c57d1781dd480480b3851b5cb40a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27503091Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/d, Name: checkpoints/3456/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3456/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=08019263496789041313d79db6a29c1d085948c0bedbdbcc5183245f8e7a425104e7c770ea011b9937668a6e5d2c92ba3e2cddf94a5615c7814214e63d917fc02308f4a14509ca8fde1a404ca2e0a6d4f15cb613459c7e107137a330a7a5d283e6e6f434805cb315186dd9a2a87920d6e20b1c3811de45ee25d38d41124bfb6981bb5f1748392a3964381b9ddf61681b81c5bd66302e42cc10929423622950e3f3cb4e1fd4bcef4438a8868e8129873c63e083b144348a7527984d68ec5a7553095357773939b3d0268e6e7b215c0f93309c81498687e9eec3c317f977bd0e2ade31a004677dea1c0c4e61809ed15804af6d715f03c155e6175c47b96a31eaa7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276809833Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/d, Name: checkpoints/384/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/384/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=ba57372d428e0c06776196256ee06c344913d2b64879dd8c65eedc48b3320c015a6bc74e7adcf336f4ead594296a9af7a41aad4bc1e84b5defb5b2231115d899128a19b3d7f12ccc3721c4140adfd762211d94359983af8674cc60b09d5b4147121391f44556297122fc9d0c201c3c2c07dc648ae47536e369d7c6433c6c98a55cfcb9b9fa336ba9191b25fdedd443bfbcb474f2217a445501f4c470d9e932e98d07f78b9aa69cac0b1ca7bf48644a793b64fbd09c8afb9d9b56fa6420a48fd0f2dc70edf3a0ad3da376708213e7f67b390d7131b3a3cc72d793abebaae9aa926d55be2428fa443a1d23c7b0eaa2852cd410621724c3df8ca2e04297d3a12de3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276840457Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/array_metadatas, Name: checkpoints/5504/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5504/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a9454fe463378c90588f03601a447bc631337e3a1c910aed023d979dc64c6e2fee28af91df3ab289b9ce286920b72eef6a3b1805da7bfa1fef93b86362d258245ac0b76f8fa0a0d6d71762425e36a35c0490b5905cca8ef63df889c07e6901f7a64d845d161ceb7bc379f4aef3e95eb3a1731ca549af9ee56218e95707e7b9ffce95bcc07c9b54fe2bc41a4f8a7f524976ff91b9fdd06aaa6f7afb85d4587e687a8d83dd866af70098022bd25eb8a61999a9c420c772d759f0e2840653cafd97c8c4fe4f65bed9096752fee1027ca2543f286c88caff5371913799de4c6074cd8627be33b84321143fbe978544cc1df97b78ee8c51d6387fa67eb39f081aeb7f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276738246Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256, Name: checkpoints/256, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/256?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9a753eccd55e914ee4961744c124c6d6cd47b87a5c4079461e083bd5925d9b97332b4a4d7be8d6356098f33e851f2bf51470680642c66cc45c82adf0ba7694f2bd52686246ed1cd63666664ee4bda69d02277a1189cc7dd0ea5d28ce8ce8c845d450b4fe5ccb949a1479099bc5765e7098710e5c75d122c22107b673a54191bb5af2bafe3243856aedb4f2efd9573a07aa107becbf4a4f6a67139d25f27dc4a8663842ea17a755ba0aba9098bd0f60762d1a2185f03d4601bc999c1b0c90ea12fc8f366ad36acf510248249477211fa95e564bca014cef5bebbef74dd676521398aa153cffe2856d82eb0b30a603281883bb97cd30d0d95f97392b621dccab4c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276890954Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0, Name: checkpoints/5888/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5888/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0a6c808a86fa9262815337bd6aa579019f07d46b5f784b6ba61cf04c9fd71561d06752efd0a486b11723faf1995049782297b0bdcf28dce8f388a17646c076b4544de90106c9df8e40f9ae5f4d1dbca1d1b4d1a93bf13b1ebd5407716a2280e3f92c25267b190cbaf065b9e3f94f85da2a5c2bba779f001cbc179f6d647e9293cc6092740c71475ea15e9e848b6c298fecd689b3144ca2d732db282a8b58a70bfadea1ec0c32239c6cf4e704da2bc7d1cd364f46b52f050efec0bae53ef09dfa73d72375af6aa9312b686475fb35d47b69b959581643e880de19f936cb0c4852efca0058431da3a6adfbe33def6c94baec9e9e2aa461a171c86e9dc092f93c8c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276908372Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0, Name: checkpoints/384/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/384/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0b17021f00248773708b9c4e76c411a38297e21fd008609dce452741ea0d12a90e8ed8744d87039213053a410b3c9e286afd16d6df4c953124d98924d25ee8b35d85aaf3339b48095d36a832434ac6b144b8b122d969d50c17140c701a1ecf183137f5eab2c3605864810a0ff7c7843a811e957e8d55afa99e99bef9d33315393586e0121df48df2717525de3a9003ccbbc1308a2ee61a85242b0a6309148be014e41ce9413863388cf8d2a7e95f51aa8386e4b6ccacd2c41b50e1b812283701064c5f4fbc15481145fdf36da2672db9455b5408c0b2c11be15113fa2608ea8cf21107f672e0071fcb593ab956b5ea64fedecb45f7fee2645e22963ace27e505&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276920072Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968, Name: checkpoints/3968, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3968?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=62973a1063f2d04d44928770801bb71c5af554792c54fd52a4572f1765c5db588ff5f6e13447ea34c7d645b01e9f88bd9aad15499458cc2d45669473fe6eba82514c4e0dce8e42dec57b489e93278d577d0cadad6702e6080363176151b37ce05c026806355465829e0f99044fda456b2425fc6af8300a00b07af5f7d723563164d7177b9105613ae8c9710e5519c75a2132a1fb5e6d022649cd63959149d340240868e50e938fd539476e77ad1880eca1136d2170c8268be23886dccdc92f5639a9ab465889a38e6b03399ef262431b048f9e696db60bb57ba42cf83434350a27ae64fc59d2e9a4789646a7e5713a338863c407a67130fef1396c1145313d1b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27692435Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480, Name: checkpoints/4480, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4480?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a3cf4be2de63555378abdd465845250dd903ba2cfa5a34aaac263e2c627d491668768b1f47af8a5bc5a1acc574a55cba69419696c84126c5394dbf4ae5afd5cd2a9c193b289a3976aa57bb49bfb553974feb3a06c708938e99f414c35c51350ad5d744fc325a8ad23b10524622d97299ecd67430cf9719f01e98b827d32bb333d65c217df5a1204a5b203b10668c8efc8375c35c4a4a95d6a4b800bb4eeddc70095ab956da2a1f9e49b0de7f9da3c73ebbcb63c283785a01fed97c8a268d11d664e8d967eb41fdf9a708c009a9922a9d4e8db8e4a4173553259ffc7eaa4983bcc905199379ddc9866f5034dda70c4011c0cf6d8a521bf28ddff8d18c215605e4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276928121Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d, Name: checkpoints/1920/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1920/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=31a34d9ac2624f2775aa0098255c1e06e14f65c4b4679b8f3dda37e4aad78e968a6c72df4bcfb3b7e6c028131c0a921571f85b3e8e2e5eb804d81f0300d4c8c8daefbf1e4715a6fca917eb1ba44bee1e69e46f2397702bd1e2d82a5a467e649b29c1d1daba9a429182b1eb426564362bd767309176dd588c6813772400bd93b0537b32053101d2a78becb221e48a194241280e3c092d02e576429d96aac48043cd690515b736fa11196d275b25927f080ac99cba11a9e1e498dd45fe8bcdc72d6b90969636821d22be058722e46160bcf898707d1f9e88b6a280c858c76553735985983f00980683134d2eec97abf4b546d36bd357b24c4e7ac69eeb410a8ed1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276812899Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default, Name: checkpoints/5632/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5632/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7a61f39562dfe8dd44efea3ba429a4ec3ba6cd1ab5cdb37a7a76df5b5a4ec34229be092e3bdef8560dd427bae50d80a173e99a3f78d2aa73c84336bf9398d8c3f6d225cfe4d12a184da35ae6222a22e7ed0015ca77d103077d55b30ce7c407bef93e6ced8e8e8908fec606194745308b7b624d8e6e1a66e17971869e57eadbdfcf0fe5add161e63f54f9e0d1d764495c111e82519fedb4db23fa345c5ce7c5729782214ad388bede6d7b3e1109085087ac7015ffb500c4185dc70cfaa5349da2e531bb57aa49bc19b1ef7ad803b33b92a79613423be148e7fe0f85ccba6b8f5c3ad2ce6104863456415f31a85f60c9dbe9123fea4a25299d69bef3302e2f9167&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277003773Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024, Name: checkpoints/1024, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1024?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=245893cf0c02e448bf5c066d798a2ab66bd3ec17736286d10008e3e37117967d9f9e7fa93030c755f66c0ce352383d3b483b4f23641d200226cf881a3374f7f1a605c38d08346276259f9f012937bcd01ac7928b9c83b43be945f6157dc54b3f5de80d5a85a8d30ebae668f059a406d21154d2d170eef28fff17963d22f58a3c0dc1b6a5354b4099d02b31b3a21ded572039494bc6097b19d0535ce5a12b9c3f01e21f308f793aba431857a8a3f3186332814e89dad5987904e1058d1cf8575d6933acb321960b85dfc759455bb636557f4cf543060dc2657c726ef6c3924d984f540cb894fb1c4889e6021caddb6a0fd15963d6297c936e125c888047d5dd45&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276842896Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560, Name: checkpoints/2560, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2560?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=87ddab72ce9073d466d0bae0e069c63ed294cd101cbdcdbebd642850ab8def0b3593c48fae7b4a930c11524a32932fa6ff70bb6ac9987ccd4d15721c510812d4a8e548b5c810d33a23da05ae25dc27b77966061d89f4906fca1a6e9200778f5cadf001710b1f9f0043eea06c30343010bdbd12480770cf2f8ed5479ef869b276af541e49a2c8ac4dc88c31e9b92ab909aa88c7eda5d6fa8be20090d76b98f2d01849c2192ddec05554bedb7b0ac5f120f7d09daae8ae14ce25d1333b821bfd85f5ea88e08bfa36e79d7fa38a4b5d4ab8d1abb2ebec1d8c31a4e59dcb1c0de32cd12a8d3d2ad69146469b6422a6cab10189581111511a1a1abf15e1120d63f583&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277032424Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216, Name: checkpoints/1216, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1216?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=771220f6ace9b20d761846387c01090cd82be1b54bcbea013166e7a1f898a18cfad1e55c8b3649d61cac345222768d7121559a8dc296d05a83a613f70f5a4d18144fa9af66b0b981f5c79abefb6690e946a37a24dc9b7b7b48ce5c49c70764c549db6310a760ed168b28425a054aa1c55a3d0b963d498d8e60b344b8720e1c636392762f231a1dc883d541b07c403dcd1500a032ba219acf2b122e2aeb3f044752bd8f72a999d8a9ef14bf35f77e4b4554b672dfdd68359b3b76991d0bbf7df5658e2004735490b99c03a98c374a0a7dddeb44c0f0f94b96ea74102cf7c94bfe72c125810fab0051039c8082dce58487f90b29ae8f83f90dffff41fe494e6d9b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277049765Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d, Name: checkpoints/2432/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2432/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3a5d24c6d1cf42c7744a5697f32c1b46304f239f83731216c91153804df8a304be5e62875fd8688dcc0f62988ba69e4de46e92deca18e901b83d34b675831e560cbc833fd1fca66539e49bb8d0c5c362b37eac5720d6a01eb27cb872815233b548f133ac98bcc92a73a592449330bd0d2ca325f750d054e98f5f268433e62c643113233cad0f1f189f2382dd2851b9ec5018b7ac78a939dda75ec679565e10868ab6b22f4b4f5c76cb8a4042dd5acbbd3fb04ffe7d55ac4d16ac5ace0c37764a8ec287d0227605cacb992772019837776e30b0e580f5e37ebbfd18bf06c5ea685b538938ad0f4fb65508388706ed4620974fba41512603d84e39fbed7d98057a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277053038Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d, Name: checkpoints/3072/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3072/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1a7ce69eefee70256097f58c0817c8dea8049311c2ffa9518e6181d263f63861631654fe4c1c17acfff93119d0627ed41d4b9ed8beee220e6332308600a856ffa7898ab033da9c550f2f5ce7da872c3a1b26b4398ab601df09769dccd9a5936d0f88ae8ff59f9b306517bdde161534a1842270b62a03f78faa09ff034d0008ceb54d76d1be3316dc803db3721e7ed02e77fa2cc13d85dae0cde486d4dd08b3e6fdb4ddac5341bf8996e46d72d0432c081e3671a16ffe978dbcee594c4eb9ab53e4a29ef6d2d53df66294c595eb59982b106f819c94a489412bfdcbfe6c56385ac2150648639a5e3190c0c20fe5d7940526c7c7a73c49f54ab92d3636fad383d8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275136834Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/array_metadatas, Name: checkpoints/704/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/704/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=54034b4d83016e23bc0698978b660e21ab14fc51334c4b7d9f53f78e33bde22b7de456283fdac6aad8258b4a31c54e5041cdb0bebd02e9e2882c0b7f2946d8a79f628373a3e0c74a4c40e7d1baa4533465418d81ca7eea5e86f08da23a84f46cf3c495f72c70df364ea6a5a9fa94ca5a8fbb1d51290fddbf96fca23768b2b6f0cbdc54456be9a634f8af9b359ea15dcdf5fd4d9472050831ce7b924a9bd1a740c275aa6b471c8d55916a7a8973ccdf95e1cb4b6b73fc33b2a25fc805589338925a0b111dc8ca682dfe153578df31d6feda34c1c7a85e7feafcb925e6265f15dabe182da15ece38a0c199c09b5c9709b4447e473e66b5e5c3fe68ae557b55cec0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277052755Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d, Name: checkpoints/1344/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1344/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8d446286df1cfc3e13d5ff2e6a8368eea054452ec2d0891f46f986e9838bb03d69a8920dba9da4da1fac068101a596f6a71ce1220bcc8d660fca3625d90855fb9173ee10f95f3f9c0968dbc4c5eda171a366ef6ec10f0e6364440780b1290519b3b6859b518f93163e4666b56eeffc23a9beb5acb5444e817fba50a763fc07293f1ea813241b7f537d9121b231ecbb6b9263f9fc500b094c6eaf95de7944e0bf6dc8fc26fe4f01fe9b6da32b547b7b0c9135a9f59af8c3de1baea0170b9bf5a2a055a950d6755c624bb6eba94982eaf40bbdc6d04f9205b1cd2dc513648c44d4114d9f9a07b7aeec5c6da6f0e4d967447c21c7deee91d683c9a1606aa72a0ec4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277139527Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default, Name: checkpoints/896/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/896/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=14eed89e99b936d6de8e10ed038073a9f35b1567698079994a373c7a108c73b975d527c07ca9ffdfb39ade94611303cd70513a985aac06bf0833e50bff8100fb993cd78822c094b6ec0872f198e1b97909ae4eaa28039e3168151de27205faa123e44d72bff368a4a19b869edfc7663369c9e845e2eaccb59f2298374e83a8d6d7148c3649377346422248b8f22200e417d18a2d762baa2dbcb13bf3c7afba81e1966d08e536decc2e9c6c9414f7d8b949f718c7ae1c47f5302d03a0bd1a6bf18c9d3cbbb3e82acb6d8a69406e1f0b6c2a80d9cc45341d02c81813b5f9f85da670e4e2aedcfc2564903be35b2d97ce92a50668dbd2ec1bfca2612d9b9e2f56ea&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277020467Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/d, Name: checkpoints/704/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/704/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=33003ff6e9f1fccbc3c3ab0dd3622908caff4b819abe8d6a184e18cc116b7bc5ea4550193b96f1efd87b6af1d72ec510d174d2e966f05a3c3fa895ae8501519b363ee593171d921d38e73b89896fdf283feab09091226c2ac11a55f717618c432635f03e5ffa1ee9b06e645ab467104b86121736ec9aa07c8f27f0ed8423ac0b9fecddf40bb276b8636edf79d7084447eb5ad33cfb26a878ee2ebf24e1a321a6b23ac03d3884a54dfb8320c49698a3ce3859ccea2b1673dfc7156a161740d980b9f66b7a30ebdeb8c0b0bdbf5a443bba28685fc02f67f1f9ad78d84c8d734066c1c760adddf28ee542f24558c9704d20f4ef6179e6dbab45010cc6d4b8e58cce&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277147851Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/array_metadatas, Name: checkpoints/512/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/512/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=07a0c69374dcbec1b434ce10d484922e4305263970637b3f2af0e11230ea1d508360ae34abee58c71098d9c5f592f3e9a7ac5b0a6bc1d63b0f281d4ddb3eb829d8bdd02c935d1824154a7615e23ea3c6328f009cdde8eb6acec80ac8c5f5bcc8ff1f39f3c9d5ea67bccf6c22e24f3c495ab072b5e83901a69d0b6c911bf4f766e756431070833965ac7a0f0cba50aa1be97db4fbf8e660af3b01e06cf06646b6aa7dea0806a3994f2a81c23e859e35ca70b5f513a9fdff0c5b499b7534e4fe11ce1b92487f2dbbe6790d72b8a7067b5b8b7f854c28508bb9dd2e2421c24914ceca5dbedbe2b9209430c6d8535fdedfb3f6980d316a22eb9b44c94f4aecd74644&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277182817Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512, Name: checkpoints/512, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/512?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=222694d515c6fad904d0d934b2e3c321dc0446082bc6c0e7c4adbbdc74a2fdecfb430af96436e84e342e4b36c30c899c0dc14b95022180c1e4284332f17f72bd7b27046cdc386a9ddb57e4abdce226eb1491cc04cece0b862dd977b09ca691bd9f598dc230a1e9a66b47976f2c9dbdfc6202ac11e3dacd23b2a354b77f5012090c9a17a53d0e793c35cadbbd286cb882e6fc7b42ba6988e9a479d9306d791da05ee3ab385392c9aab8739c01ef35d8534245719a7e05b2dfebd14277b39cd306a372fa65c2f034d478e436cd8b341ce4af02407f5cd2a60aa8a1009abce8701a67e4831cd8afba46730fcbd834b0edc0f8baab601ec5e381db08008a6bb66ae5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276664853Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0, Name: checkpoints/3776/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3776/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=794afa0fb46b6ececce91f068e25ed0528b75ee6c2f5218c64967e60794cd7ded3d6496d2d6c8fbf5d79604f611392ea14c5d697ad4b9f78a9b53e2ca785c82e980337544d3610a40ae17f466357901f06e79838644befb2f6144f60fd734c46165788ca154072f3752e5ca3ea38abf86b19f28e8886e5ba215aada313766b1c4b82fe2e5a0d42608088ab4570193c60f8b9b9b68a34c456c8ee9b192f12f48088b3ffaae5edcd9b9d6276954b7ac4c6900ff3af51e5ae85d80601463c0c34dfba0e2ba82027cfb3e25619b45fe3ac0c02143ce9f87666bb68b0fd74e02036b4bd16fb8bb2e96b0614ada3bd081ed8cec4637ffa3461edfdeb74ee9f41eb1bf7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275118696Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default, Name: checkpoints/1536/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1536/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2baed1202144af81ccfe7da24cd689102e349ec506d3724b6ca98e18fb71278be28c2b2fbcb2aaa2b575cd36a939b71eea56a0a47c7b2bef38273020c6bd82964bf296619269dd5c5a290da93b5ec277a5abcfedf1399365905b31d78fe0831357b89d2d7a6e102aadf13ceda145d194466be0a7ca13614d266567dfb700a4b497714a88bd29948cf5bbfa6ddaeae8beaf6bf5a5ff2d67263e8440f6ab2eabd2d446d87eabb23ed641200f00cb35adcb8297b4f6ecc880547eaca7dc94e91efa4345f9746f2108360cb8628c8cc970933eb29f9f0d5e719b1466cefbb44e4c8a32013b30e7de385115e190b6e932c7c466caf692f4a9bef402b7fdb36abfd0e3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275112242Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default, Name: checkpoints/3328/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3328/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=11a0c2fe4007b1b999743153ea3858f9d3c2160e97dbd819b1907ae200674ce2c2390ea3ef8f7d89c108f47a45d90b3da423b0c8e100912c90d37f4d715f6c7074e3ddeba43508a92588e25b7d109d175eabd8856e2ef1b0c3f43fab3387358d3a2d622dfa4c0f093a5698fd1775ca6da0da8469fa043ca40fdc2a8dc008150c673a4986eb26bd60283b447c57fcb00ba30b9026dc9af1bfea92887908946cca2ff2161f7340fb90ff5c9e146e29032c760a42a0f0831ea1a561991398379cea8b98cdf1a03cdeee03af1827f1ec6af9f9f1766b63079336cc37bed3b691a60f2ac7262506098989cbd6bf29489b6d9d87f93dea3ac4c3df5b066e35edf14729&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275468473Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0, Name: checkpoints/5952/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5952/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=91e689f2958b63450ff217fe5ab22003bd7eebb0bf7bd7aa726eced3297bea1e9a0979eac7381c567f36f258aa952960d2cac554298c8fed9ddef4049ad109384beebb73a82f774e7bd399b4870a9514cd5efe5db2095d509eb2053ddafbc2d57f9c116d823cbc5c002305d68698848b4c9c344b7eff5e8b3fe2c09cbe5cb4eef3a96fb96933ea1368e28485352678b4f7ca18d26d5f966b05c35faff757e882366ad2f0d8f781fdecca2713ac27df41e5673e806ba81a9665bcab31eed38edba33c7351aba729224404199ee0e50a6ca5e1e6610e89c8660f5f93a27ce45e95f007904fcbc31199f03ff141b855ab38145127e92675162a3f6c25b895106f36&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277213894Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default, Name: checkpoints/4736/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4736/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0c8529ffcf9baeb30745fa710d261412991426ef9001814861a5951be0fc8ce52238026d5188b5da90f06339b578a4e4429924c94e168f66b340004fce90854f0d99f00a766956ba0f621d1f58d1d45efafc692c13c935c16465a2e16221e35106160ac14aee2ea909149079810689a7877db23e0c2fc62a884feb51f79444aacdb1fee06af38d0fc2c81a86cbf03aadeb85985739414f415af2154f594166bc4c0626780b32d3d05a687610a5769e8b91ee70df1a89a477552ac603c13c8419ef468175c0a572c796c002ad9b174ee386293faf0989eb83d556bc25ba8de91e759431f8510175275fe40528d9df81cad4c9dfeeb9ff906509e919de91d474ae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27515004Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0, Name: checkpoints/640/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/640/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a2ed1b2787571ae2af1e409b569ed04493c2da26425c69729371ff0f8aede3c5468c07c9cdfa884ec720e669178e42be71a79376e8b9dd89092688edb478c20be11f76f0cdbe72597bdf7852b2f6a0acdc92b5c87f6cf71f615f5887ccfe8cf3622b75e5eb13483fbf8b667f97c25d451405ebc75b892567ed2b1d930395a93804b1071691c269da8cd1c807839245f610c0a5ce7802ce87631495b43e74284e7cae8bec2538145ee2e28ddad2728153daf23cab828f22a822afcf6a9418af97d5c130bd0414f895e7a1bbbd501af47edd168f00f7bcd9d9b3b0df79207e0575b333f30746c2811ae43033ea17ba53a49fe8a59148ee632ba2136a8142775a74&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275012718Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default, Name: checkpoints/3712/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3712/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3334b1f16c8c6443044d498f1ff9d96276ed4fcd94e5907cce320807fe4b2514513a3bb89c01cd5d83d126401a7e9fbaa6847c39eb4bb16fdeb906bbd17efba9ea82c49ccd23f99e2433f329c7307496646e766f2d4cea6d767bd5ea057c1e13cf1cd18eac54c1dd798dac25b2da4190a71a60489ede8f3c51a6d1f5fd541c730d8ffd4315d8d57bdc32539fb63d4da246566617afa5167d6f6021bfd2189b4a61cb9403ee8c36f2aae7651b12c2c3e4edb9dda5db7988842ad13f14fa071dba1060b50046c1319f347905da5d0ed496fd97a5b81fe4a40b43f965170e799ddd533bf05daff0ed7e46686ae57c776d74ab3f0fc21c4b863627865c7f7b0c07e6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275209926Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/array_metadatas, Name: checkpoints/3776/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3776/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=79112074e2292173da61f8fbf94ceb584b3cca2e3fa363ca05a170bc227be0b53e9538d09a15dd58f006a9e4d5bb842696819b317c08c239f3aec65069680ed12f2c75930776a204283513b28fbfc3ddd711d781b3ed63d59845ca9901f219c60a188782405e1fd336521a1dd8b4861a47f39326baad9404ceaff60e397722529ac77df30f6f5d3bb30bcb58e081356e74f97e9a3c0e759b01f086cb125995d0d2207377b34fd01500535f4c49cee9c6f93fe046772eb9af8cc743eb0278931b8a476b32d274c29d80abdb8b97a660a2bc25b0165c210fd084080f775c7700688caf59d402ff8c0427f18e59610448f482d6d518660b8f5e49d1fa883904d4c4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275293768Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/array_metadatas, Name: checkpoints/2176/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2176/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3409be322ad241e7df613daac97113a8dcf262407004bb00878aefd76c0980a2e1d8197a21e63d5566dd917babf4da032391409a8ca1451a19a9ffe692295f8c61548f4c1829735dcc7409e33931bf03f93242edf61791f24f3e37f0cd2c32886108498efa1e0a9094b4b803710d19aa227333544ffd0b43894f93b3dced1160a3aecd386b596346d2bab1bbec4599d82ee99277b50c20f5cf6ca9d328669355ac44c29edea4883d126eb1a7ad6c6522377dad3ea2a2b0f6db583e721bccaadab0526fc283607f73b451fc0ab3903c57290086b66986ecbce16d815470df277e11037f356eb901869157b12bc4d3d2183604b66a4161ffc48b08e78dd4406229&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275319126Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default, Name: checkpoints/6080/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6080/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6651d7417a9d57801581e869c25bb005eb070b6999f7502f759a232c27e6cbe50cbfa90a94c77072e3381d17c3ed42dcf2f3a0d762f238d74cf3f722d82dc77933fc8cd16c77793c2c3bfb37ee6517148143214a623a98e656afa640ab26a5a6c748e408383ece01c090472bd64fb567bf6787ae9360316b37a59cc44e7aace62442bb34a5a76d7399b61d770c35439f30144502c0a492f92b0552fe5afea7cf7fbc49e86f495f168130827c5b38b5efbeff00f8d08b21b5b693041e8dde52528ae9bfff3529d450e7698d6af48c51ad075d167b04ed1047e19b2bf13beb4f31f42b00fd49b3b815e43700214f4962baea542ef50febb388bd0503c44eb49951&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275401628Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/d, Name: checkpoints/5248/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5248/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=02ab19b4efbd6560d73d307121df81a661334c20104c922f2aa79a646cf5d141fb75e5ed80eb75e3f795d599f61e47bf0f8a9d83ecf2aa695fda24df74403232d1e3df32a705fcca5c8d05a1af016851aad52343294ce95bf1a0b3fc4c2ac51d5b2e079914f298ddd968308508fa6df1d03f741331adab3efed946d27cb45795171a522d9de3b7b28882b043c752dfbae3c1d3c199a5453eede7407aaf29b0fa047a99fdc3f567d3a1ce477120aa47705a15351321c851f90302cf4e86990c4ff1cde8b77bd2077e7b07dd836438aadc357384e4c23cacd1e4cc2f78a34fc8d3ef60aa1f564fc086ba0f95036c225d85c327360434c7d286a14563a571e2e18e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27567293Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/d, Name: checkpoints/2176/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2176/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7847bc2cd9c123931ac01e04bdfed5cc1d51c61e2163b60eb1b7ee8dd4e21b8afbb37873ff0e62ec8a6b7caf7097fe9a011969c568c85b362b03d4957d56f05e6d3ff2502c3df3af0498a75c3bf701982993acf0ed317a55037dcfa31a5de9a48b096fa53c88fde92f14d5f5f4fa255437e1d02687ad4b8ca8ff9c99f843ef336f55c5fa62cd470d4a104a79b48c0bca889526fdf5183527441183f2d347f0014a4511b55655187066283813b8a279caf43fb0f4eab2ec45e4015468a41fcde96caa343046df2a4206ea9033872096b7b2d4b82ae204c9dcfa68f0c78f16659ddb640dc3cebc9a8490234e517c87e20bd2f408d0c5cbb615c587a96d01c40dce&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276631813Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/array_metadatas, Name: checkpoints/3456/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3456/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3446611bab83e5517700a3979d558ffc41001aa59f145bf41c67f964b7e6fe487f9e66592eb1049d2bcfe5988d637bc6cc0ed15772b1ed44915c531f810f085a38a5fffb21a421f39cb323e6c5d37373d53c478f58ffebef8264cedc57753de906829cb3049e47ecab7f26844c310b3c219551767c9ec7d17564c49814f41114b64cbe3553247da42efefa439d975f938005f523cbd5540743788734816053967e63d37d41a9ce4c633fc29597cf227f4fbc27f95d1a69d4fd43a14050fc1c9df89555e3de6947b512da8522ab6162a1357e4551ca5070487f5ca4714a057526b3668380f6a39fdf93575825da81d020c19eda25e71df5697455863b7b29dc02&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276626431Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d, Name: checkpoints/2240/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2240/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1621b842bf6036b171e21de82421291ee9afd553ba85aa4a59f5f77e5b5019c9dadc020b48881e7de60eb38057d6ea4060e2c8b53e7487a781b8d22a1655c337362509e64d2ae20fcc3ca8b85268e0861da7de68e67a921fbbf6ab0dbe274e4d68f84ad85429e943e9fb7ebbc05c2e6ad7bd7e73839af2258d0d659ccbc542fd4de5228cc96cc8d49381e4aaefff42484b4c2ee4e689e11bb6c7efef1ea26e7baf5c737d92292655f571cd69bbf4d9d30e8f7969296c05c4e976f391266d0b879a412f3c45913ad6a18be979efaa6f32d046d2d0eff69d9b88e224b5b78a8c680a779bb3e4dfb602d0785eb7594c8679c3415281fa77f7cc23f27334d36a08be&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277361517Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/array_metadatas, Name: checkpoints/1536/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1536/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3977291699776975e7f6beafe46fa05dafa108f601148de5bae2c790f3e75f681e241fccff8ca1739bbfaf661f72f8f44b828eb10c5e2f142f15e05f61329aae17f132d8bf65dfaa1fa9f0818700e80d16742d8c8dbb3d7af12784a3a4baa72ff53f7abef18ccd55ea0534406a04428370a48412036bd4b3b21334433de4831a956a0c99c34bfe570bd2be65a4b671b263da3b4018ea7b21caed164c8d42e04a8dba8cf1f3784fd55cf358dce0e23e11751544dd4148384480cd4d94259139edee3b31422bc77ab61a4e2014f18715aabcf46e66ebbd689850ed023b74f12ed92e4b254609e18f3a2faca7cb775f84c5704701d4bb4f7641093c1234568af127&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27660575Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d, Name: checkpoints/1088/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1088/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=74225ad9e3901973b93c9e2b85d6716523048153493a64f1b03c54d8f323be12be0852e1958ac0ad77a86906ea8ffd38515b2b51e00a8eb46a028f71e78da25c16076801b97c948e65bdcc4c46909047ef271f2a918e7c32e8c2c354b0a0482b02e9c58a1216bb13f0c0309fe5c8924fce9fc777b7ab2c5267fbdf11de16089f806e9287423ac805544ca9906f83e73339743a14c289e32b257f6727ced18004a77394e169fcbcee1779db56c63976489775b09ef5c8021fccf5a18fe5f5dd3ebc45cd0846bb0f5867fedd4e3e0651e7a5ed159b92461c1d21c7ecc760e9948eb177156aeab304a1d994e8114680aeee7219093bd220e19fceba7163bd283ed1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277357515Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/array_metadatas, Name: checkpoints/1856/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1856/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=329c796a34904c7b8c3c9a3104dc39a97fbb5ef2f03d706a1bc6d5713eb7fecccaaf240f287cfaa4aeb62577d32fdade680434f1af0fb61172b1dd4671508ba00d24a65b2d9417568f252c6c0004bbdda06588d15bfbabb78a7683d53cae67ae71fb12cf8f9a62984d6922284d78d5f20e554ef367473761e8213bd0155d5210e2ff493c39bbb806669ddc5b106f5e46408439129c15de220f6ae63fa561a105e3019ff5905203cf5e739e6dab23e584dc1258aaab105366c451fb8c92f6149441067d79f5837161c11a62aacb85be021603672e6b9675f2af9382172a27f90dd01f273d299c805b3d176743d073230546c5c1b2ea9408d7afb66165e52684d6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275005626Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0, Name: checkpoints/3456/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3456/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1711f32326b771042b572520e4a1a4de8a8bfaa8839d67d53baf77b8dc545abc4f878c3a1ab2cd20a91d7a3dc90f97b82c6d3e43eddd3d685dd30923614ce16ed35449dbb2584bfb96d67827d24a92dbb5b84c9df76cac5e30911a2fdb5462f21c837c315011bdd4a168d4f6c08cf293e11511574e57bb083540c514b5dda7fada2fa514addd28c4fec0fa895adbe3e141e47916f88531f4ff0869b5c584907d41ce395d4db1a1e9776973ac20d255fadd89da71791578c5165365c02ef2d155dc0b2984af53aa30aa4bb7f26cba917d798919709daf891d76a3c6a82982a6b6eb2a923c28273b23635a0d443469850e621e94efecbbc359a84eca29c9d2b7ff&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27665742Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688, Name: checkpoints/2688, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2688?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1341bdc1cfb34c55213352c0dfed8993dd723845ee5a265983f69198085fa7bea47ae20064b978e8c630524277e400e90e9e56e98472d11eca666f07ef78b7723277812c9c565ba1d5e7da1b035ad4d9a2bc80dfcb5dda3b16b424c1d9c6da8eb11c33467f4776e375b5b6434d73a638a06060b77bf8afe9d93d41431ecec747f64ba208e7d787d04387117695ecaa147fc84d67f36e5d46c9f86a10a84e299933c70c821ce24655b8e38d1a810ac7b1c57f6fea0d65269763952732be4adbdd93a309f5288d9a4501cf0292f999e818e6ba7ea1e7b56981b10587d372b046d90dcd65cb819f85e2cd6287201f1571b0f42dd7ebe69ff656ca9a8c366a089ca8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277399578Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/array_metadatas, Name: checkpoints/1344/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1344/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6e9ed9a41cb6462ea68c787777a1f4ac7c0bcc7e50be82b8ce61139821d4e4dcf5316db2e87c546d24a716b41313ce2bb20216eb73abaa963339910df48ceb3d0057e1f38f8071eddb178d370c9c8df34127608787fdbf10434a6ccf734b924594c73acf91a4dabe9714b57b447d2d301b3ae8d676d004004faa04034027b0476eb5e4648cb048d951fd4eb84ce32058a8da3789c93b034dbc35bdf7fc349c854578d22a1704874d8d6148340eb82911f6a1eaafaa6f85791bd1411c1203d19f6a076a79b74f39f21229283d5b5a540d42e46dfb15ebed457b5e68a068fc40e60917ffe24f2ad03d42214738c298fedf011ddd17ac049f99bb8095299a3e8702&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275312328Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0, Name: checkpoints/1664/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1664/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=48dd1b96a52b77db1c56cb17aa95c6c021de3378e238125f81f076acae557efd97b89e7f6075df4db145a60cc7f84629df73ed77a037ad4ee3d238570e5d50a440fe4cf8623bb4128153f1c3e1726e8f8720c6519d12e01443478a434619710a0621a4621bfe8c69eaf6183fd95b6ac073df7bd3c8bfab0000e779b984d7e1922162373b40f13e6d3854160dc27a0a13ed67ab464b2d8528111b519d655926a23df22cd0e245ef0aaef45d530ffaffa332e4e4919f241857747d2cca03fa2c0490b0e164461ef94516919eeb0537172e1de9747511b0928fb76dd976a1bbfa95b02b95f1c01eb2da1996acb46a5c301a884ab3dcadaa693946399d9140fca4a3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276656689Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240, Name: checkpoints/2240, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2240?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=658d73cdef2eccbb75de1e26fabec16e001d3f8743519cf21eba19668c996ad56c67753f05fb94c298d688ad4317b11c0de86368796075191cacf85c3352c6fc752ecc7fe2fee9ea2ab9a67798a282c74b0b4d56312ee79fa337a12d074b3c52ae859db6779a3085750a5b52aec33a703fe872512f0615251ba6ce8f527f22dc43bde1a9e9432d010b2ca9cac9fdb76a8430ac632f48f9c94390e898b834130e4ecf2541b97c97975a04fcf78d08fb2157c08676967eba89c05a348f324b04c30f9dd4321416a9c87c1cc861830f8a9bb69261e8b1efc4c36ababf2acf036bc39155cfa456927f18ac9cc463aec2322764478e37afde1e9197b180867015bb11&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275076717Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default, Name: checkpoints/2816/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2816/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3e33e1c49da095843dcd75b4bb267f5ac48adcee7dda19038bafbc908cf952854be60afb20d8bb011afd09076b11e74736ff39ef1b4a07c6d557f83433af0fea9148086ac07732d9ab4dc9434f9061b43bd561f6424fbfad766f337d3d22518dd3798a0ebc3eb44794d0bea37c443539b7546b1804d7b704d6d3ff0acecc94493941f1a23bd083de68e0279d6c1c55561186335f15034a61e8eeb6dd57d5280117ed858824fc9905a5ddcb6e12e8a1ff3386c12e170b935d6cccaca1ce250f0c399dff97aecdff9dc8173e8fe913979a9fdf76e9b0582ed2ce5a2b07440e3b5085e5fa9bf0d4d705f83037369ba1afbec33d1ea9418c4abf67902fb1dd7812ad&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277502254Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/d, Name: checkpoints/3904/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3904/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=442577d6b4c415cf6fa55d704939520f2afba771b2dc00113da9fa881893a790a8010fb19a909fe9227b96324ec7ec1eb80aac6f84669e92c97a4554a4a948465ceba6bd6b99d659136af8280f7b028067a17f5c57e7b317623094fff6e299cb05565085c8b4151e4632d435708c91026c88801eda8cc09a75f21380122562de5e71d04711e0dd622043f52147ea7ec1ef4f363f2b00e7e59cf98298b26e3e504fb246b9c28e9c04d51e2c7e4e5c8181e3bd918f64ce143293c4c31fe385979bba22bee62cfd230ca3929c739891ff8ed4889b61ab7bf2a8bf23eaf7c34eb139b56e8c77f6b91dd5d7069aeaff24be4563660597f5bc24a9ef23796a52858290&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27529099Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128, Name: checkpoints/128, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/128?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4f0d0d15e8e726237f0ef698f2fa74cb6ec0273d4feaa34184d7b06c124818d65eb9ee67698895b98c7435284585f750c66e5d1d8c51fd02cf000878daa91bcef6357d31d51e6a4d3c2e5279d3ca49805fdd7737f2e9795385c1a494897498c870a41cd0070000b8dc47477b11d4311b04d774d8030c6c117f6565dc0786899d1d3fbc9f375561fe11a4eb347e39beac21de3823e25b142fd1024f8049afbc9d9d4afaa8d843acb8a42f694f85dc9030629d26349064941ebaa182837746949d01c8a1ff9299cc2b7ef868256673a960fe391ba7b2bac1147875cd5971a89b9041c4ad2c1ebbd68235010a7fa205bc85b296f65886790f1fae623b0fb512e970&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27665992Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/array_metadatas, Name: checkpoints/2944/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2944/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7e812d12f7a1abe090167ee7414a652090cf98c7744a2e7833cfbb5e5509572f43d18e881ea75e387fd8e0b0a2b4d8c757c12aade39878bd3f6d35cdd5407b1f7b4bddac0a8b4f1a4ce919779c3ca8d79d201ef6c651be2ba35af3a1b33f3a9a72fda86bc82f5be1b663e82389127546c8fecfa7bbe1906290facd4914680c9e78d5187ed0aac6747659efff11d1cebbd7eaf5ff381d9e34d82533af44c2a150bfc54c6674f23b4824d320d98c9249f3d7447e83f25dfe8eef56bc400220c6d41307e265051c8f379f4d42ea338ee9848956bf821a946564b5db7a60c8692850aa40a915acc8d63fdd79cfaeaf85cd191cb5adb17c63b2034d1c85c4afaf70c8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275605258Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d, Name: checkpoints/5440/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5440/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=86ab8f53038f38cd135fe0bf8c160078a6f0d3430a156979a9d17c8df3f9b3a3b9bf1af352b091241d1e0ffffb7eda3d3d8ab16e8547f037ad7eac9cc1f531da18cb1500534ebae994e34c993d62af178af5e6312481a51d5a8a60e3e1b2c548f2dc4f9c0591894a574174ee0487b74e2a51937fbbc0827aa839e43a277fcee4dadbf76a7f3806b91e0ef151561a67fdb74b254cd6155edc2f26196f809bd02d3ae320c7cea647901bc22b82460fb5faf1c92ea1fbeb671fab65fc59795fa628c633fad5ffe6655f11d788dc35f83b4325cb97090fca1a6964b09a37e231affe4a56567f1a3ba588ae075dd46e589fb9a3e6816f7793d9815390c91dfb2b9839&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276732016Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d, Name: checkpoints/3584/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3584/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=baae405f778fff7ad56f795fdb2b87a0ed176c11ce1384eb138914d408828a040ea1b14e33226c12104d5d4296b8bb67a9471578cebb49f86d33a61ae7813d39fca5afbe9228c480a74e016d1c66dd248e25ff7baffffc393877509f11d09614c971166a03fd0a5a5f279144209148ec7ca802b81c8be3aa74e9954fa7ac858dadaefe53792e9e9e9008befab8c18c9ee8d43fd862dbc11a4403f60b79ccca91f95b52c22fc14b05ed903c7db8ee41d0a513784e57d1e88a01cbd3bb3106d0c08a7d75987832fc8e293523d2c21a33ef55a24d5c3b1eac451f2ce87e88ae38807d733a492511b1a64978301bbba6c34b1b3e8e6629d090aab98a0c2f3f8deb43&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277624414Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/array_metadatas, Name: checkpoints/3200/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3200/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=916170304f1fa5c28138a3996d2aa9ece1884162315cf3b2efddebd78b19586fadb6ac42aad897221499de8c0baeaab726c56f393eb2f70b3711aef1fba9e0b7cca3fd30f3090143ff0c6e4c9dcbe546cea0f4cbe88c89680ad2aa615be5b98daad2579015aa0bbc1b9482cda83c1e046641f682a12b2b039ec9868d6c9f1a57e064fffa38e59c53331119c4916dcdb5f5fba8cefbeec50c178fd98e9c6186babd18757f69861c3dccac818d7f325925c7c3becbc8a1d5a12f4aa67660ca7403ef73fdff16926fb8e4f317e475fad6ac87b8c5cae48a3eef3b91855ee1f559712d25c60d5ee6f2b86e6374fe763d103179484028d0ac8dc92cbcb198f7353420&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276944876Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/array_metadatas, Name: checkpoints/5824/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5824/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=494d84d8088ab8b2763f2f9a2f982556d2535bd4b7a46f48cf04969254f7dc0b1c86ef1dad486450770c0c30e1827e6d97bcc1b2d07e6eb03e6fb4544a2c50d61400bb0a217595cbab5c53446bac0df035c0af90ad8098b9c9bf789e2ee2596ded3700215ddc91f4164bbe4ad869a4c8b966deda00371fc4daedea7d90c4b90fa8bd9552366c90d26c2063e859cfc0c68de6f545100e6b7f16cb99125b7479f3b7978a4a52af0ae45816c4a54c3fbc4eb0759744095b550d28385e78a5c562f4908c5d6344e2e45226d16543e61d95e8a99611c1e1551dbdfcc41e7e0e234237c733e71a6dbfc5d00f8b60bd8b7e6877d5f4e0a1c4d2141bb8e730f8aeaabdd0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27694067Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/d, Name: checkpoints/4672/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4672/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1fc4d74eb4d209b67b46324857570ef1be26a79ab03e9086debdc3d78dc5c8dbc312ea0d4b97f82950df3e52793d3f05953ac9f1385c86c76733729b8b2474ebcfa11bfec593a71fbe27b75898230896631bcf33c797da8e0c2f9b292d4ff3966315c08896bae0b3feb63011a3593c1f765aed2f2e66f94d4b5fb9cfc7bf19bee12fde48ec58a18ddcb2409e6672786e431cce46f2be935911f01a0c8e3987283fce77bcede0c5f9911ab4e42f326ec5ef620bb670423a1d9ce1336efc77e5fd4b5ae2e8d3663a114bf7d7bd82f85af2fb794607ded4cd20a04f259034df78b1a0e26aa1a1a6d80a5df3f4b0e7587906de55d8f078731e8b1eded35d5f1c9758&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276925824Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default, Name: checkpoints/3136/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3136/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=aa40f06a9503354e062d047bb653fbaa970040607e00b172bd62708bed0dbfce4d06b67649b1438127ece7e003b917f4fb49470aeb3100cf4a1f576c2e1519d13627b90c4e7036be3ec818e5c204b1d39efb5ab01c912aa0963fe86859d0a88dcb0ac19273b487910b000583f876d456fcbcbd801401fdc07da7fdcba11404a4d2115885c9d940a4cc2141387e2ab8caff4bc7f0425d8b51c09d554fe6bdf04cc632fe8944bea57a4e66ae54453c22236207b7965a17bb9e99ff62b533b9f1831dfafdce7c751287c52d4817b7047164519a0b1b24be55a99c56025c90849baccec87c18a849598b222985107a8e3bfc567df166333fd8b53ee04feec78af2c2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277687843Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/array_metadatas, Name: checkpoints/3904/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3904/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5d78749a139224f5ddc9cbfde0f571d11aad9d84b7383ab3de06eb98a718d37b432da98744e33457eb902a2a2ca4817a9e1eecbc101d5aa87458fa0d8bb1ad346aee2a53d15591fa9a880579b8c1da5f1be30c260f06dbfde6c560190e58882ccb196b6f31edf2e3465e430d0fe795baf9abbea544ad4f90249105a224f079ad351cdc838e2eb0c14b982397ff7364a2230e7eef9ca34e568b54efe762860a60dd343ca9cc3b29385e54d8cdfaf57658f6158ac40027df040403dddb8ee9aebfbef21876391ebdfec8ef771574be1af70a780d193e9af22e6f0cdd3f80d4648d9b6474d5d5af3f157611bad7845539912b5cfc1a16101a9f3e26cebad9c54cd7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277697588Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/array_metadatas, Name: checkpoints/1216/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1216/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9710612ca98c1dcf51eacd12abec57e54b9dc42557952e6da64dfc138241f149652e6e10cf42cdf50537254c6d2396f2c4f069d2c8398ea098f976f52fc852faf502bca77b3cf506ac0726ba77bffc062b657b4bf94b98ea0718d8da84caa4742d47f2b825db8932b521352c5feff0d92f8eca240638899669d66adca0e9f334a5c3ffdcd35d1b787361652029e84cbb83bb70391d89e84bdcb47adfc48276328e0e3b6400ef27d53d8d03352ec146f9a09e83d776f1809add6209b944f94ae870f240f790cdc925e07e70a8df59206e1675ade3996314cecbf6cb26aff1664b0ac9ec49d54bd40c94472bbd8c613aadef6d2a150b3ddef6607f6b449344e85b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277730394Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704, Name: checkpoints/704, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/704?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2bdbee9391c7ad22a0b29d3c6aee61fc9ce5a72437e0a6d87fcd2a8be1f3fd84e80fc18d0fd300dd0c828a11d6e8460e70d0d3701e40d7a41a9c3c3e517139a51542995ee234648ceba68258bd832b5ba638927c9017c17d91ed92886f1404b000330fe638aa221583d028555f8377a90a42c21cf5a893c8a8c25d7352baadc2fa99c382333ae57d09fa2fbf3b0a60f19c7fac5367d0866627f932d18939bf7d8f36cd491ce672a4b6a9e7e49229c673917a6a4c8d17bb4492fdb3572f55c6e3accbfdeb31484a092293fbe01e50a615648ec7358af1502ea56a4995b406f704ef77f66fefd47c64376a933e607bf24a14f6c9098ef02000d7d4ffe2612d838a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277743365Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952, Name: checkpoints/5952, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5952?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=31ca06210739ebe120b5e96e20a998ea4a291dd19f6c46b5f87e16e5f303b3e8b4acad57e03f27e23ee41cfa08f8d1638a3bb12553902a3dfef083cc93aa3dd6a03393095e018ee5d3cda433e36395ae031b4c75224241d46f7ce3972cb39eeed81563629602be6ccc668b0b7cc1ae4663ffc844a1506f87c6af44cb1d9a31b8c2a579dc30d50591540c69ef1127f7fe430623921a899f9d14793c7b16b415ad352b799157d2c0c3b5ec3b79fb36530b66179ba466cfcbf0eb0d3b8714a5aff019444c31f66fc14de2d4b3bb46f1160ff3178bb92f78647ead494ea4880c356c438f4282615688045de47c67d7b073c0f0dea97717f3905eb6f5d7e3061f0196&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276846103Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d, Name: checkpoints/5952/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5952/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0c881bd07199c16ed49bc8913ef822fc22b61eef990fe4f7a0585cdf820baa5fa53761d139fa0fda8c2483f4b385ad16a244bb69b313d3d609f56af5b8c7b97b00251cd2978b13f04c1d7f28df71a6b0273b593fb0913253782d7f976bcaf494aee064b8a86ed3dfc6ddcf24cbb0d8c559c7f516c6993ac56d2b00cea6907ad024267326b65efe956ca075bcf61436ebe8f211136d98a837ec69b2ebbcfe23c1080ff37bb590aca5c6964648c4056da09c97325c399d9085148a71eb1ec9055d15eb2045d3a37fadab243ee68d751375abdb944a8ddaf677a61f00312ed04098546f9a8b326d3513ee02d3d6f2b6e94e5c69665ae07acd1ff1613e6b5a8133dc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27697767Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960, Name: checkpoints/960, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/960?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5959937cb2e2f19eca4097cfa8d3ce0ca270b21242be466d9dd955b6a407d263f666294703444ac75fb46e2257ef7242d272e88b7eafa22d6c438d2c6371853d7e9d760b1e1689520494f0f43f27f4d9a1096a7a51c57612684b78dc6e463c8922c26c495dc3ac7e4ec87b6df50dfa4269b7544a3309aafefabba07621fcc91e70afcd473dae5c0a6724e82a9b99a79382c189668628a575788f303508464fdc5bb443c7958f77f851c1f3d6c9b2fb32a1aa90effbc52b2850207fd60462e8574510e8035f1e0bf693b54cc6cbbc4478e6c58772e89362662323f43b34190327b16da737568852696c7a9674c03503d57146b8ad19389fff6e0bd18362004782&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276956494Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default, Name: checkpoints/2944/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2944/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c11a0d930c28707292956b6082d960599041a7a95a8475d34ff531de20b30ccf0014c29004d4aa352b9b0d2ec96124219fe7451fbbbb4a05f9aa9fbcdeabf1f29dfd0ea3a266e9f9ba7eb1be6a1d1ab2b8eddcb3a8d2291ad0998db824097d2a68fa7e1efaad556280fd2e14c8b44d9366822f72868b01b7ab5e41d3dca6102f299bfeaec0d46163cd3437540dbe7a5bf340e730e89ec8e7d114a64da900d06924e2168734b07ce0f4a34eb1622ff40def841c742cef7f0491d21bce9395a1b3c082ac06d4009c66ea2b16969b00bd8328b0b8fa51510beba1960af26b3faa9a490d6072df9460fb70a1b12f0d8fc1b80c7219cc9f3b896d114cee8f408a7216&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277021083Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/array_metadatas, Name: checkpoints/5376/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5376/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=795d12806ac06bbd6f8ecad82d3ab3e8019e7ff2c9629eb4a7f4a6e7149088235dd25a0e4b0396a83b20133cd3f0396effc1e14cc76e3c5611828071693d39ad8057702c70e0b1cf75fa5dd73be24a714f15e0e23ed6d0878d988034c43c8098813b5272139673a7e8046459c9b00bc1dc547557b6463de5f093e79696549b5362c3f6e90a02b49de7b1819660e22edb77ef2da91140918c88ad2d081cc4504b7326c5354c69a98076fda2194283cbdb0a7ab6d61cb40e0e826636d3dc65fce9366271e3cc49aa149aad28be10f748264632895c16c8e89fafdcf955cadbb2d1da44f9df744f9bba4251bc48bc5db84476e8965c3a265fbcf6525e8acb35f3a5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277052797Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/d, Name: checkpoints/1856/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1856/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6abb85c76c02f175e67f503dbdc9f4437bf5ef4d0df42308363c631b2a1d935ff5a2a1ee2efad1acddcc2bfe02d981e49241d4ce59f7df7ca09fb4c603b2068f7234c92ff8f838cc67c7e26314a7f5ac58caea273e93e39b2772eb1b2fb918f4a75a06016582d682d14f3ae05b916765a573bd662ee904190cff9103e5c2a38ce7e1b98c80203767d1631a3b9f6b3028d35415a70d4c8f63ae558b8aebedc4426839227e64d8db919ab3192dbb0390773436fa50343014f7c27844c5f83367d163da134b57edd87f8621bc212b0d480c247529ecb21547c81f8b735847245783c025b2ef28e41f4d85ef03aaa842194bddc6416c0d040a3275ea1aa93f0e5ee5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277794674Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0, Name: checkpoints/448/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/448/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=940b1640fea8176d024fda41ef250f856c69ee92df4592a0def037e81a28671f6ca093895645e8d36692108c9446e1c51c64172b01e7ff41a2e0c9a712d5f034fd9ad03b0c8202fbd8823a5b318c60ed8100e10bb966537927d9c82e802a6f29ec3d3291d2801cdeff81b0e9f59e4cda64e433eb4905fb3b6b6c0b20d33fa648b7aa034c411ddf81bbf7213e14d08618edb1a80ba488c94282f3781db9d623cf63af0cbb4e9bb81b3e5457cb2a4080337eec43b4cb179995934a27b872de5f5bc25a0389c91041ecc53728c98e5af19ab3b01f97d306eb5bbf7837d69d1508a1062d209886488f8e4affd7e9412ba3d73dcc1ee63dce623910a742075f31119a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277036395Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608, Name: checkpoints/4608, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4608?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1693920b420c8fda83b765522e9826fcc979356d99a089ee5d11498e3ea97211d044c95140152e46a6d06f59aaf718d1c3abfb92d958cc6a45ae5a3fe08d6466297433266ac49cf72f87330ff6362dfd9e0cfbf02d6e11a29798cc783cf26abf3ec44eb92059151ec759feb7db92f51af228516d859efb79fa9f67d00e9653b921797d678bc665247710b5fcb5778d741ea19dc43fe149fe07c491f1dfea441e6ef80c91b19e0514bf83bd904f2d41fee442203cca389bd95b166e4770918cf21bb90c5331f569cf8c2154c47eea6a720accd9f4a64a41ff39f4150b3626f946a57e3c433c447651608791a2b8fdd68375cba7c69bfd78b621537d37826e1f55&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277034774Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d, Name: checkpoints/1408/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1408/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=402f85640722e6b2f7eccad4f419b8a674fcaed009332c96a038317842a789bf5ccd8bf7389a1e7bffd4e15aaadda4b3d3041b3cd3a05014c03b112354233e51a6b22e6cbeb967d921fcc657eeada4da05e007670798fcfb506c9526e784e4a8fa632690ef8f913675fb0ccb89679780c736454477f64fa6b806b18d2e2f1731743818bd6d168c88cd4ceb41c4025b8c923fc67abebcc368841930ec449433cf9f3ac1c38c7589e150689245af382942e1ddddf34f1d178f8156a907850f17c65583c8f738ff652795f2e423bddfa57d5cd2b2ae3904b68b04eb0f151c7f1b4cc28a869c1468a88e687df566a44e91daa5dd46ae104fb0a56ab3ddefebe1409e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277831525Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/array_metadatas, Name: checkpoints/3968/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3968/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=03c8e9f8cf445dad8a43798cee0e5047a2ecac47a8239c0a708a8e536f9fa9454cb902727b9a419bce6b66bb246bae5b0104e7cd000ca83449ad0df95ac4502c3d6ff050fc86a25af67db1016bb90565ca8eee685fc1408de1eda787d3b43d052a767b9fff13c638e1bf9d904633158fd186e6bce1e33fd22b95c7d2a35cc85dfec8e918ca84780f53741fc3a4370a07dbb0d86efcbb0ff86e6917512772d887e6bb01f7b4d28442d29e89fd3aa901de3e709c00432927c83f4d0c7f42ff134fc562a5a93a70fc412a00029001a33bf2bd27e17ecccff080e76dfad68201f4318327d0506a80350e7cf15b3d3e8d7d90043856857fa32d0cfa20cfd6c3211d48&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277062921Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/d, Name: checkpoints/5888/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5888/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4b917fc9e1e0f32eea1e4c2c267f63bfd0b19bc5d9e20b5e73fba4fd0ccf49f75725d867c0cf7da0d6e5bcc6543f45117b3cdd8ff986fe1b193c6a59fc930f908050d534cd04a56d097447ff63828f2b0f2e1a3795ba93c7fa52a1e4866a6d51f2c77bea7649d176b6b84cc16fcba7dc6c841660e9a4efddc1d0718258c531a336562177761ce816912d876a694052529dac93879a31254feb6adfcd26370b83efd2bfbf082d787a94e7ce09608427d668bee89bdc72e906f5632b1feb05554894bd6498d43301092a688a105867410b12965f53129b97aeeccfff65129201d76a24851bc4f2418ba219880456665e494638e817c1c3df6f1d214dcd43170bbd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277860236Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d, Name: checkpoints/3520/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3520/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=007deadb4725609014469ac68d56916d5554dc2672e3f23ceb4b40ee62bc0aff37f072e757ecfc1702fc3f2944644db3107fcdddf74a4dfe0904771775b297f7cfbb4f2572a67a791dd90b7d2f6b937b99955897dcd8a639e76a9131a8a8dd2255efe8ff682987ddc790bb57f70ab9fa7e6f94c928dab326578060a0e9bb113ad2485bfcc51d002a75b78d1c8916ae6cae7b4aa2fe5c8989e5098ad1dfe2d29aebb4c3d535d43e074a0976c508f0d99ac7a28324979b27488452d8b5c960d846dc4a343b9f0d11b0ddbfc3be2dbe7815e8b2deb6e5ea10d94ee48da9e18860b5440ec297df52aa398fea21b993e0048e07d98c9e0c67b2a6dcd21eb442049b38&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277887121Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/d, Name: checkpoints/5568/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5568/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0881e15932fcc8ab09f382f3439f876bea3c9d62df33ddba1358326a4c27f75c2696edc91a62806b9d2697d9582c736d3234186f8ca15372da49257fd4212488f8d6911f69abe6c720f6ca9cd04fe654220c1dfba53125dbcf63c57152f5163ccea39ca45bfb688299f9bebcd3d0c4a74e505f81325a7a8e910a771797fe48d7cd0019096cbf369dc8dd34534c7eff8629a7c2a056359ef940f84c07ac6652eec598ec64572aaa5c88236c21eb78c7cf10b2d09c4f06a0499a771cbe3a9a06d6d340e726a7771372caa0eb6c0ba8730b0e9734ce616a47857f504ce13b6f59f774dc930b65521c40bb02bb4ba5316046993e5013e40677c1493c12c5c273a92e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277983711Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d, Name: checkpoints/1664/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1664/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=034e4bb8dc96886a7342f64e24d87e20a8a970b43e1dac0158b3b0c1baef30ab8556e2b050954c36982b7a081deabf4285b1f27ee97b6cffbf7376506976f965238f83be2c355122861d00bd0d7640708175c1376b35d5ec132e84b74d81591115e4fed874535bbd41d566b9acd8a0bf6e88760fc764999b21dde0fd395d44088d04194b8d96e5dcab55c4f1235640b077f865cfd0393d0a14add6ce136869a466d17f8b559f2ec6ad395059d0056d2e68a717dc7d0a3ef86cba6671dd117bb205d43144046f5409bc040166be5df6396496cfd95663b7cc88fa63e831a1bd41375bb27b4ebc8c72eff8508c87cd07362477b759ceb55e9a51f482352581f6a2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276553626Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0, Name: checkpoints/1984/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1984/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5d7fe2c084c8fde8c12012e6fb48e211614a6eaf7356756849129f44a49421ba27e3a1e5aa39ee1cad26b60694836dae76f29d51ff70e3c2c9fc815de1888e5c83b0ea577395d478a1f238f1b2dda4032ceff490d0343597f1a44849847908c5228600077c6872123ae8d076acc120613ef22cca53e7f8a40e27b89ae974b7244218c25c3995096f1ad67c2b963768b05d6bde764e013dd92be6f509c1402d1d06c757bad9e9e1041bcf6e09decaf239ae7c1e21b69323a0e23a08ea26a27991d5e2adbf67da3d3f183ca44e30f5d3fe6e5d25fc322fb2668068798c91c366ac67c81f670fe9e7bd8c86abba6e11fdf0245b238a393fadf6cc3f6063bc5ab747&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277119662Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default, Name: checkpoints/3776/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3776/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=34420e64926682234d5beac20df608e339e1bafb39e7a80fd6db573cbd46863a3fed01f054e578b42df6ce92c1d6c20b009864f0924132396ff79735ccc689ec08a33671f013ce872ddce1ef6b3ea72699cf0bac52fa80bb8610370c9d222898ccf8c5b971bbcbb7307db8d41d9a88216dcd45d0e20070369569a90d3383c2c8a6b8ce76bc98a946b2fb8a2315c66c9c5b65cacee9e4a74158012393830ec01e2de9cba5704576a7e006f7a088331d4a11c8fc24bb05b6c2eecbaa7969bc075eb23d84d3b133068d8ff706e5dd2413af6ccae73bf95fd9bd16be0eedaddceb89c47ddc3e3b8e206b4b996fa174197b45de5f7eb2f8107bdead6e7f03ee0d55fa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277108779Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d, Name: checkpoints/4224/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4224/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=206ccc462a76354674a3b9b8286031636f3f144fc13e4da532a027fcc4f1752879ea3ba4863039d0dbe229eaaaa3565866a50e2856b4f0f91f4a0a708bf4bf86ad94a97f2ca87649985678e8ba04f8d63971143ff01658da2e0cdce7745eddf206acda9050c099c8af20397478537caac1c300bb00cf73a96fb7d9afc921d7dadf51dd441f7921b374728e8e042b0f09ef2f9c3454d5691903ce60bf1e8df79d700818180e7dd405dfa9053febb49493b8a0a9dc5e243c76279063bcd55fb1a4cae55635db4a7502f3e26694247413b6676f36dfd864f65ee2c1221c14e581df1a38c89b8bda8f4d63fb8ade91069c09ec738659b738684aa1842deb101b757b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277147113Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d, Name: checkpoints/4736/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4736/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4df5cf3a1980fb521a2559d74ccfb94dc564632c6e5c61fd0e4959c02057eab04c868f1ba02f64e4db7c943e263046ddb13580bc987fff546ff9ea15b28eec4e9c74f96c0aef7f8576fa0f5b9f6b838f114aaef46cdd4e956fa8cb7bc9f8d18686ebbb8c7aab4661029b2469b272c430b173b6f6eecf4165547b5b4a00708652b5d61b1779397730e975f371d5f89a7317e52f4043055f77bee6fdf7c494d6a5ffea991e111548bad00aeb0c201afa9fa5f349e36aa1e04630e077b3adba236d7a5c060f00e70d3ca7173d2f587eab63b7ccca05830990126d88f189fca113807c20179d00fcbd4777db2eda57f56cc359e6ff9705cd905973d166e52f271a48&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277142595Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d, Name: checkpoints/4608/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4608/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6302f17f6592ad0aad72445c4bfbe6c92c0c4c729b42021d104a0445e65b69cc47728949d7b888e799a87fd3651a1fad18c4f358659f41e96d31442e5a9fb062f04eed44d90e8a5f3ee6557dc353e4d5733e5591e043db398cef3b02e8789d76741bdbed40f454aed8e58dbd0371beabc05ccad4067e9c57f2560afa3d3146f740ac184ec29902d17ee6961143804e97163180e77f8cdc254d8c9c0ef442b75e80c2de05b2f7d5b85621bcf81e4b2d6ef7b3341c4a0c6b561ebcfcdaba1045fb3f1591c289685b73f8a9273f308552c39f5c751fd81ad45e0b9c0ce2b1ba15933f3c53de0eca51c6201977b41abe09468cb2dbc71afffc184848f21112789013&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277174689Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0, Name: checkpoints/4608/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4608/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0338acb819faba4a8e5be0afafc33681584bbf21ae020ba2b9cc23ca4674ba1c40f4a3c25164053457322a5a1cfadd91adeff59976a66fb47724202e9104d95a19084bd70cbb96dd65d8ca2d36ad8b982a2de07169b2696814cb740fdc2924327f733a1e64a625823a470dc4442a4515ceacca3923838414377e4497285aabaf0d891efe6991499110310b9348cf8a3f61ae1aaec134db83fbd92f963340095b722144bec2505b0de34be6d0f8269234cba696298cf911ecac86233e0b420a35b52cbf579747b9274e3a0d4840e78a44a39042c8fb880da7f5516b850067955f51d6ffe8ebbbf381b70a2911ce2b03dc49bb1966af49807d61cce692549f9d26&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277180413Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default, Name: checkpoints/1024/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1024/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=77fc8a406bdb8473d945d81bcea64e8ae2b9db669aa6f0fd274a0a4596402d3038eb7f642c29cdddeb84cf83ad36fe7f8da1aec66402d131c48ebbe17bac98c6e9734ec140c1bc53c74413ae9654c08885d12ba61e924e4adc869fa626a5de968b055bd37f374b6e8d0ba8f4da9945d6558dbe658f075da703be570ee549c016b786703ad4b427fa286ed74d529facd467627df1a9171dea9327c58c65cc4954d276cde214af822f3b51cbeb70e806bd7f750d247b49ef1053d104511ca6aa8715743f0ff9117eb929cf9fdfa290fc55f30253b91a8df105d0722eb2737f2200d02c7cc1c1f96420be8db39e18a18ea84e2577743b903882d42ece491ffc1e8b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27722837Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0, Name: checkpoints/4544/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4544/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1db502f94ed86b05cbe62dbe0a4cf9fe12da6c6c6d77aeeda787a67836289be168826976e201a223f8231eb1b99dac34053319f4e4ec51e28abc89cc2d3cb2fd5ec04feb70b05975807b89ff253a2708d9a60ccbd07fe10a5f8351229b71fbff107985e137ebc7353c96e3e6c6374c712afb477978dfc43ece33ea3f8fc06bfd62e2c0beab060289a9f0551d876b6591a563ce160b5d7a1d0a397d53d787915355068a6ebcafdf72040be1c5214d7282217e00310fa87d5b0662798a6ed7374178b9286a255a1650cbdced0ad8892ee3b5bbc576bb43d6fdf2ea478be7c00131267c776d90989435c7562bbf163a1c7c4555ede6429261cdbf18837382ad2249&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277187203Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/array_metadatas, Name: checkpoints/192/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/192/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=99a635336ceac6b3b6426c683a1db1f4b392847af004bce6c6ce3ffd0c71aadaa3fa61cc2c66ca0e9cc7cbba8b02fd45c55eb5ca8f446e5ae0aa006ba6d309a47bcf447e9feb9f2b560c0b4c8e63830fc8d43184822bf511fe9c2870d4a87b5f4ebb57e1e38efc74e94500bdc788086e3e4202da88d880886a99a6514ce62c1c4f171fe092a1055001930330c216484d16d733df4ea35b712e8bc77d51c32ee25af2b225b752e74081b62d1f94c57bce8cdb07877d75c3f65c38ec8574e889659a5552872d3951eae10e899b3d32362c244b36471225e7fd22800351963d0675200ff413e24cdb033da5050f7c33feac3988425055023ea6038fb3df6bf2adef&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277245761Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0, Name: checkpoints/1728/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1728/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=337424009198d556e0762821d1bb1bcdc400a54a5b0c159f64b3e2465d64288d15498510c1cae82294c81976ff9f6a221a048a29adc9695e1d8ad2030a22c844af58995082ac2544928a44fb380228e026b6e132bebaa12b1e61500c5e44e3c3e25ddf090a1d75825109d0e70bb879c23a7c0a1ff37eaa759136e76a98c1e483c37a13be587c3bc252a0f12e1f5519b54545f5f6d9a0df568100b6e2722a4c7054856b2e6ea96e2c7817585485a8bf896c919109832c82024e534bfd7aeafa1dda51c9e0e9ebc3c2ebeebac146b10deaf6b87f4b6709278dffaaf353684e01dc5dac057fe80efd0e938485843d3c5b888ee0b547431e86fde38971cb70015318&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277218407Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d, Name: checkpoints/4352/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4352/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=52a49bb03be65c469c290f53380f0779bf9c4d47cf849b7c63bacb82c9137b532d08509581644dfbe7c444046d1c1cc76fb591b5dd1d488745b6444cb33dda5b3dc52b1211e052e3bf2aa2746cce6f3c0f0f91dd5582c5ce7ea6bf0e3875d5140be8cd36e46025aab448ed9a4b8d668866d900bc699958e93ec6ef3b7249507818a467dfa318b7c6a9e5787179401361fdfe0d5e0b640e918329c7e6a31ea46d38d6893bb3fc065eea5855e3ac4f51240806560bb0ea7365121b4e21e4da28f43cdc69bccbe2a9e7255532cd683c0268806ef1d244c165e57b29a6021246439f5851af85d32f30c7a141b46ea8bc413b30c0e68fbd2c7b269146bc235e22f963&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275131461Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default, Name: checkpoints/2624/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2624/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=54a13cfc56e0e3a67aae7d05eb6d5ea17502ab0dd0d48f6a94aa1640bd29081986bad39806e4861db908afa79de366aa2c872dd3bed437b74020f6114f267d7f4bca42bcdd048bf82c7d68988e15f52525a8bdf1876e8cd0065271918d13dfa863f9c59ec40fa877a63c36aa00d96f0dfbbe2bb3ed3579418777d145f11b5ee9003b3f70cd8d569f134fb1d5f3866db12459238fcbcee005961e85bc3808b49b6dddaf5d04b80025e37b8878050b702d73a4a6fd560d81f7bef76c70e3ca520e55fe3bfa0ab0cff7eab0ff2ad760a3437193264978374b4ea2a9060dee542482f168fd9de2c50ca2c82f0918ed2394f444da07072b7fe9708465655696fd2f67&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277290872Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default, Name: checkpoints/384/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/384/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1832c39c94b4bec9c5688aa773eadc84d81efa6fc40dc1b916c2cc0f5e0e65b3653d9d6a584613f1f4d9755934eacec8ef0e775afbf94d3867198465e5ce60753cbd14be3aecfa4a026030bfcb9089093e79f532b8c7007adfb6f7f317a47361b14f780715bfedc868dee4222f51c3cf834e100a79b510ca044929688a478608905502b901f547e0f16fc65b21a846557a0f9c9112e4b0517004fc5372c7c6defae0906aced637245741170274ec32752240e89e6abd119879d69a752ae4b75a65444919eb80d947994155aa98b04c58e1f7d08c75b5da5b1c89643caa740d584f15274a991587e828b0f38ba34098a1defe16c719ea114dad4be9a8264b1475&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276556338Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d, Name: checkpoints/384/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/384/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=290543e4e4703451b5dac9f5899d3111de14e158c1fe1c3ba366b13f91dcd77ec52ad87abd9758f17eb41f13f3624f7984253c6f81cb9cf8093cd9b4a25de6d0146f957aaa21f1c8516bc7f0f8ce84d45edc02a37e287189ce2c7eb043fc86cbc93a446693887d2ca09ccb77ccb04a1d061c02a87a632cdb35dc5a745e263e0a933763911e4f3c60a4250f713816f078f9638a02fd7114d6789d6ef38a90c22eb0debc17b00c82949b482185f8d9fdc37ea5122339fdb5abcb97590f5e9304cd971e6e08445b96921f1f689237974797a675845d26257563cca47876481151d7b427a25ff9f36727cb58afe8b0184acd8fafea0206bd9d2bd16393a5ca1b71e0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277313305Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992, Name: checkpoints/4992, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4992?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=bef3183c8847e5dfce9d06e33852caf040833b54825f6bbe146afe2ddf310e8352e999e1b0a7ed9402e7716201fa72aec618529b46acdc4a4bde92d7d25a861c9934f7c8547ca37d0716a78986872b0e12dcf1defebece23f8493656c9e78ac89d403609438c1e21db49e5718e9c51185fc2644130855eaa3f186b564b436a236c35a6e735601b225a7f63c828a2ae498bfe652e7a397f827881eb84d714120d6df3b3cbc3ca9eb65d9a4985f4319f86e8308d52faa927fd019ad68cab57fa29e6ff4196d124710591bfcb52e47c742a9a39df3ee5b251d8ac510053567abac71ee4f900ef4608b814c989c932fa9626b962fce454f285f56616bcbfafc70e66&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277283102Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/d, Name: checkpoints/448/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/448/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2a2716be0bb2bd60749cd34ae2de5a2f69a1a0a745d644c47714590c2a157b0ab3bc2c0816cfb51b64160148184301820bf94947fb7db863547cf94fa807a328d94f013aa9acefae1faea7d4cfe6972601f70d58a363a350dd9e141e6e0e1fc55662761c9108030be56c100958ab44bf508da108187cf66c58d69b2a7193cf7b90caf81832a4d05c9febff3d607793e37bfb439d7594bd2b076d4cb80e6f8f5fdb09a757b35d4478ad43eb034d0029b93d04fb2c664864f8bdd5cd043e385b2494d086beaacae8a1bc3680038761084a374ebbba680c60c3e59483a945eea7a41e8c5058db845e6a35da4bf445513092ab60b067735abba26986030b0ea97563&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277299499Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d, Name: checkpoints/2048/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2048/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9aa3bf5739fc19fd64d28b36bab7702da140bc3cc5e38db05b3cdc2482b12d8775751182721c1c1ebd38b63796977eca1290889ca611aad655c6d7b6ad8bd9f055e9c9ca5aec5124c8fba602e8b5918f529c11e16fc135156461a545ca228c9ed09a5ca63a77b579f643f2b7080afff2e56f777ab857e04a39907edb1eb208e4097c99f5e08004baa956def2eaea4f2a93ec3b517ebf9b5844c527d4ee539d91bbd16c04213a431ce5a90075fb649382bfd0f8e394e9bcbbf24d7b11cf4cda302160dcc257aa2148bee8d1459dcc5a47e7fc7b113807c38622a4a7b03bc45e3a69ce8522357f5223889e86460d3821ea7ff7fd41ae450f0fd82daf5b963f4cc5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277358907Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0, Name: checkpoints/3712/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3712/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7e5562ba233e4ac1b7059788de64d8dabbfaf6ff6d23336636a2a1f7197154bca962e2003a1e9c83bbd65d3e554a5aa903e96ee437c188d7f1d3879be323fb6217a025ff656685f9bf6171729261e624d907344dfd41a4e4bb44d6289e7616912edc0e97560e51e801a800edfdc205d4e39d4e131dad9b8a0d40346d2b610fef2c68732e9f2608e94f2a08cebc50a59bcc608fa6e3c759b0e6e45c6a6d64c07c53184e50e3f1c0b81a461dc11b91f0ceffce77a7020c48f57afc252f94986ea5dec978525ee51204072f19b2cf5d25096c77cea052721f6227ca758467c5cf2353641268b1f9ab7d52943693d8bca974833eaa4c5c6c0f80f34956082b51b7b0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27752023Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072, Name: checkpoints/3072, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3072?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b65b3f3d45b938591147d41db36c7d640a5bab266b3418aebe17b1bf1130a6ba3d919f3764c5aff81ad4921dcaa427a9ce31d0833220ba383f2d806a7120e9f3293d411fdc63bad5d2c960347fc7bf73f110f0b8aaca13b62b93cb79882798f3ca14b3c1479d1726445aa6cb05b2cddb5683de3ffaa21a97b538c3ea183a261e8938a2405ce00e9f9ab8801463578f79f34bf6b3704593232d5595d7d196c234acb4b6e7a5d0c34530ebaa138a651f79f0f12a273c0ed8b48c9ae2746b9fcd6fc0cc950a7c3917df0f6ecc864804372628161838b37d2a06571fa1508a2c690c8eebb5cf5292e89b7bf1beb4b6aff030b1099c1ed694106d0ac798791fd3be2c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277532542Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default, Name: checkpoints/4864/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4864/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=14b14dfcc9a7369518180f3d1b9d7bdacd7d5f5a591d9aab593d55c71bd4db4a453759ec69166341897c1695c3bc3bcefab1532319e2fdf353b291cc3e4fd27135865b51194f420207c40937cec40937cd95c1366906c5a1132b3aa10078d6981316cdf0d9cbc0f14cd9c75b5e84e709d3b12cdbfed7e6e279af5e854d60083e875453be027711a86df3b310c66a3e4c80f0272c8aafca795bc281ea5652f8d2d03d961e3475ef891293902e9a11b5a0fffb465758d1144347bb383b47eae8dca4270ab1b14564a6e8f7bc277c1f76178e5cd6fc240edada005cc19fb8a103511ff5d138da77c9e797ddf7d5bb5aef527a4b2f3b28df42ad27e76dd1698bffd8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277593458Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0, Name: checkpoints/3584/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3584/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6e1b6779a1890b1171e1fff701025e96a880da95b6ac2cdebf312579fbd4ec15feb55e78f7873e02d1a78a913b5c25ad692748c1617636b27873a28b51ed9d04e763fc6cf5033c4b41b23e4b504bd846d2bee0718133e5b25565adda90983215cb6dd946ab783ea0ba2060115c857273f039f800fdf0460d0d0be61ed00e227ed0f560992a1245033d556246d57dcd4e1e6c047c6f08bffad437f9633c5a7e4366c39a4ae75fcb708f5bdbce248ff7065c4ad658aab218e54466e4c379f818ca327d7de164990ac23c6d213322558eeeaf483ef08cd19d51c0cf0fe1ce54267abba323ab807b7411ba5b9047785a7ce5ad3a9806aa981fe7a3a5b8b3bea2b3ba&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277566372Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d, Name: checkpoints/1216/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1216/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=65b195ac3491e5d76dcc012df1802dd5acb08f871e7b6596d284eb9fbfd3cb486792e043f4c1254a22d644b31ae866c54d24066ac1022d0fa32dd07952317efd21dca7abb5313344f82846f9f36a79a6b5d07f2b39953a98ce04d8c239deea471efd4af8b24673cf4c54f97d843b7128569c5b1ad5f1e84fc01862a36ad4e1db72729b6060588588b3ab878f2cb7f7374e156df8545c595526f0641b40b24f9982916d03c08148f7665e379c4b00984212beac856e78ed1a990a30b00e63c5143208a31da4eea71a5a85aed02030b8b0584bc7cded8e7f10d3ec58dd1c81e7425723c3e5d4b462c214819ce723429500c4ed11b8e0e2a263c15b6f9c25a3e685&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277628868Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0, Name: checkpoints/2688/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2688/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c30be1409e64718a5143014b6ddb7e30a0865b1c639602c89371057115545e4ff770f3108fe32198cbc5f47c16083ef0b87d9ccd98af74c9552364ba355d83d4b4e6400ee8d984941878ca674b24c42468302a244d9f94af91296eacdf9939fac7f297ceb2f45e1cf4e0677c15c9c92ef13e594138060c6104fbb83280b7c3341f0938971d7e40afaeb49f335c4d27dba833d5c7086f3ae8df6592d9522931c801cfd02d9006a560d5b84946a0b1e67be1129859edebb1062ff4d5bd451bc8d3edab465a0f79ebd03e38be059d50fc0a26549b076cef5d695390fc22bbabac53fefef6a1cb7a5684009191f98e0ee5a358007405c41616866111fc0d60a63079&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277672454Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600, Name: checkpoints/1600, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1600?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5f3cf2d3cfbc0c3f0d8b507f5e8a4cce7b29f5b288d5ab26f0143180a283d501090d1a03e5eba0732a42a5e713a1842cb3a7a5c9a08065842daa926a9e0adf9828224b3213a7203744fd755f7f3cead64a91c11647749c658ff5172287dbea7bf181835f0ef29f2505bee1dd82f6c795997dda19333a21c47a49b2bc7b49b3fa26a53e4a65f17a631f224bb5aa62d9a841df42b0c571e1e96feb91c2f6645f0f7f96539b0c953b925b75a76b0e53534a8ff63165d1b580d7d87f651039f20b1ee288370916bdf487dd17372e4ad124cbf073dee001a709cb667cb0d31e4e87664cd1c8cf2922f1deb62c8b2343193e17cda51e26929e1a726a431a9c9ea4ac9f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277766971Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default, Name: checkpoints/3456/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3456/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8ee15074aca09a1d06ac648af6b24c2ac100d6c604b053571d7229eed5840d025e8efe17ad797ea3895e88e3c194412692debabe770cd793a68680d920d31cc32ae9fb6a54389171648c9746dd6fea424a70ef92f6e34daced3a4761f65c605224a1cb5a1e334fc3644a964fa4fc4f46193c62191da35fc4e4bc1aa5a79c8bcd06e102f06e1e38609d566637952043d6ad8262ad2f842955c3e8de423b8c7ecf2bf303d6df5aedc75613bdb6e57bd320ff76e8fe11a8142c7a220a5f52780050a249d091e153d854937cbbe8a8ce02665d98c5556bb880affc77a4118448efb0e4444f4cdf650d759d54fca655667738f35387239552e04ecaf997ba94eedbcb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277031131Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904, Name: checkpoints/3904, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3904?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=bf5ba3bb247359bfdbc67b920d2bb10c64a358709fe1e6bc81ef53a16351f0a20a7e5ad1fd5d770bf323fb312b8a46eb93830553495f4d675c43663943b1fae68fdbfd67dc33e9a9a32b9f4e1d66878569b5ebbd1dcabf100f5bcd0a1dfbdaa616acc0aad326e9339fc0f1582a6631ae8366cc536cf612da8f9f992df57bcfcbf5d67be7fba128f85b58cdd842233ce60ca8f054de305bcde94406cfc1e7466f162ad9673e5c83d23ae51fb8b16b5e1553b4188b49933403348663d73a670ac46a4b4b2865c3d9ccdd34e09036cb6fecc549bae4f75c820b6831e0e760ab8d2dcb80e4f9b0c8c0936d3d639b12a100461057f9e3c05266d96da55764e7c8f7b0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27778348Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584, Name: checkpoints/3584, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3584?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=731df541b997669a2a534c893ba3a58eb658cd3bc864f0d4615037c06a288dd28cc462750cf90d47dad983b2b93b575f7c6b9bf9f7360a43c284011190bbdf4a087174a8ef8f2df0d94ddb137f5d89377041761249d6e8cf3474574632af8e939d53affd40780fc3e43f72af9c20144aac174ab8ac5a23203cac47861150f723591d798beb70314d8eb33bc99473de87501964ca7ae5c64f3dccf80460e8e157a93af3819b547f44d20d0063021524a5ba74987116892ae5eaff38bcb141f3563bb1a5cf1aab851921cf4a03bbe8651653f4cfd42e3e4eab385eefbddcd35ecfd28f6a6926b4707d0ff5e43211c923ec9d262ae6d6ff0802382b23eeb3e94542&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277844752Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default, Name: checkpoints/1792/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1792/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1844681299b30d4c7fbbd349da3bf8ece91e193becc83d8719be114f32eb3c96b28d2396972131487e8c5b4b8d37a251da219f66423fb93f6ffdd0745d4343f255e5bd3905a5257c2fe3731495d5ac8db7506421721cfb8a48eaa6fdbbeed0d67b41063928fb7b51a38f1dd0bdb5dc7b8f1b8bf3e388a761c070616948d674bd8f5f724e839d003f1b23f7b13d490fb9ded47ead1ebb4dc21b65e086ded5b3b54c2d0377ea20fdddb67eb6679762e3545caaf225afc92858713f449522bb8bc807899d17edd461c7cd57a947ad2993870a1475fb5e0b60743b8dc01e3240b403d1c9bbd8c25356c1ea57ba8f92b3157c29c66939e9193d0fa65af376d2ea92bc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277098738Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d, Name: checkpoints/4800/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4800/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7feef6cc65ecdd1460024344543cd09078adda9fcc16bf85067c434f3e1232e69cee6417d96bb56ec4d12f079dadb674d73c95dbe2f55c9b028ac94eaeb49978b29f8836d7e58b20d39801f7ffe4c8dfeeea1871db9fb7ce3accd3dcc138a4b3b91cd02ffb61fc6a25f5f8ea7e601c24db6c3ef20fda7b5aad06f008bc129d85af2a3ea8062d8b705a7b1d5af6c0382c420015b14d208506c07d8cde3481d2baf85a9919945742d1f544371a1520c4d7009945055388460d3cc4722364ebdc0bae473809a6b1c24a938cac08b7b87a5bf9bc2a64a332310506ed761391e088b139f282a611d93b338ac6d2858a7e4b51c8a3ed7ea3baa06d2acf010ad2395211&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278457124Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/d, Name: checkpoints/832/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/832/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=78bdc60e5a9ab3d6a30d9b5e55c8cfbcdb2f628eaf206c20ef02846ba9281792f41e1644fb5433303bdc20fbf497d30faf6f24620a34c54bdc2bcd2b732b561c83133bd0b73f3aeab2ad5f24fb53db579ad88115312af6472931a88da48e3589e03865ffbf6150f559161ece4eb0487500b95e12765f497d2e0dffc2ff1b453f3f74bcaacf1d748fd3ef0367f058d8132a271ab996bf58cefebfc055b4b21ef30af051955c7f360233bd5ac2c9324d5a1ea3ba196e20886ede3d6dfcb069726d729e297b6ee729475c3ed1e611393e83ef418ededdc79441e40f6ad137ac40b02258fe5a4c085599b8629819cd5c46f617022803208ea8d2e9770bf3c4ed48e9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27846881Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d, Name: checkpoints/4992/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4992/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=bd07efd29c7049ae224fa43fc6629d9af1efc7e11ca52de0700ce0ea57aa7b00bbea7e65e7ae03b6f622b88270d3aaedf88191c6a5ab4350d8dccb0a28a9582b3c0b33ee9d6d49f60536572c6275aea80df6c3eda97091738f7af8cad7102f04155eae57148ba77808a17c95011b06330f396be1d466b01e63d8a8a8a4fd872e23e479c60bce662ca0f33b0113d81c88eb958c1f0bd1b35dc673bfff5770743f9a5f8ceebc300851392f9a093f856e7b468ed92c32fb87a5b8efb542f8d9b17b5b73152308d3f5032297fb8168f6372fc566968c9ab4f250a4596baaef0ad215d07273a3b98c4603007c22de6c34d776cd5855a8b30f41d1200f231639f285b8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278482874Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default, Name: checkpoints/2752/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2752/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=819747b74a1c1448502bce85f2434624cba80bda06651445811e78a89d6db5b29c0b2daf0d2282c26fb2975af379ddb5dbc664bc5b3f84b26da1fa4ef46a0d8ef25acdf50d4baf693a26703a3389212c06a78ecdb0293560d5b437b96e009843f081beec6fefb25e291cf8a0d08e7dab815a4946fe5bf3500d656df20c79d3aff474a8861a4665d0231ccff64438f72339bba226b9325f151a808cafaaf4b6313fc878a848e6ac861dfedfb20af61faf473b64fe3110a698543f6919a5755069b1d50d4664eaec7bd9f1f9c4f2c7a88cc897c26df0246c4cc76df2c3d1ded180b144fa41ea3c33ac5880901b5315459bdadc3b3beb577f79a58729d21c646775&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278516326Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default, Name: checkpoints/5568/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5568/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=84d7b2d3abd92b43e6a2348c869568c396a54e80445ab6b29b05c790dbca1b9555a907d171a0afe7f649cddcb5a12c08656ed06ff6ff06a25ea09054bdf69cd6c292ab73d13b7760f98f55f7cf96b19048c0e3638d68cc8379b8416b48e6b9f9764246dd6b756cee51387cd6c3c49f0a99dbae6cf56422b3e5253f56e66751fe996cc286d978ec293a79bf430e1f3856e2d943d4255a6321c7ca6a5a1aabd172e14977d88f56890d30168d90a5172a4a2bbb4f71f70379182750fb9971d037ce5e3a425225b6dc9a3665d3ae146646af8c637affdfa9c1d8fe89956594e2daee84dff84d3bbb1c5b2342d233f8236755db1bdb4466c0d851d7cacee9f894dfcc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278488653Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/array_metadatas, Name: checkpoints/3712/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3712/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=533f72eb156d6dc92edc1e5ff2606ab60ad45c3769cb7c497b12c6571aad04572bb1ca1107204c7849c77b0f8450c96c1ff86250afa640d4a3eb48684328077d790c961ba175450f07a03b69544a2f3bb963a4f1aae677b7e8841d151e07d3b73c35c2368ee2eece2d10ecca59732e6e8d7a3f273be2b8b542e4ede3c53c95a051ad36e7496cbdffa3294b2ad723302e66f612b0dddf33efc6d0e54eab754657613df528d71b3937cc39663bcb9097ee7ec14420c1e4389cf242762657f94caa9181e9fff7f2099c4701c05bd683890545b4d8c9a8de0e704b89aa11cc3a9dca9a71d1de56779c635520f87ba0888d6480c22e4d44ada9aa58433fc59daedbd5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278641691Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default, Name: checkpoints/1920/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1920/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=50ece88151b4591c31a0128ba1789e7ef3b312b8a45444cfc1aefb0e8d3e4a06e02832cb3170770f604a0832a2cd22fba7703113d55afd8300326c1beed0381da93cc6524e52b4f8cd5dd159c8f8874afa525e6ca9f700c1dd9b7df048efb155efed969d634ca78986d512842980f4eb780ccc0580b95ba2353a4aed4ca4c14f8c8aa2adcec6ea3ba6a34410789db1fad46835037b0d0ef1800d8f8e733655ccb30b6e554a7f6a4e0366966e021f9e8f5ad589ab183e22fe141ec1259e4b56454d2e6a6ff091844e0dbb1fcf7fc779a1352d4c70e2473ec6818f9a47bbd7d0bb995ccd66f80a6164062c63eee579897554eb6c8f14fa2116e283b18312d1975b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278643564Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/array_metadatas, Name: checkpoints/1728/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1728/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=20bff5db31d6f9472ad665e0d585dce1cdfe07c237260004d79de731e6ac7bc3016afc8d8d2aca6adb198572080d7b41eef581bab720c8f1e4d7e21f034cfae95e40bae426fa206c92b83586e3e9b6ae50bc22be8ff6124e21004b89a785763f8d1628eb118808860518ec2762271eaa3aeeccee62d27d3ac6df533eb2c33b293af78a7570676a4d7b1fa9dd475536b0b2725536d429781ac9d0b8dbf226496b8e933e08a448464fd0c2251fdeccdab73420235c5c304e43876b36b5e09461e4b05c83492ffca94991a0bc37264ae2455e03ddc97ab4ecb0e7fe10d93bfaf88c4fc9841d1abb20bff516ccffb7edc97ac015981f6500153a1bdc51d830773039&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278684297Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d, Name: checkpoints/128/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/128/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=20b48674b2effc92ee40356b0afbdeba8814c5eaed438c0bf35c4cab371015e607405d70ec083a4b1f12f3ddacca1f4e6f2d61fff3431701535473dfad3f9d65cc41dab0b95eefffbf3580085bd813159a023d82cfa859b16bb1cf9e6800df970e89f0f21edaea3a50e6f09f5c8b78331e35a782e09b6a8a5948fc13a9cca48166c7723beef75a0e58843b62534064c73d9730a237ea3c77eac6aab024b058de233c4b5a6689d615caf11b500ab03a3650711b1a5585ceb76bb2620632d4af8cc7c8558f8d1163adeb6fed571faf5c094f6d0ba7121d250e23842eb95382c205eb414f8bfb72a624f6f8543b01d804bb020812ecf7b2e0bf627a8b6bd473f52b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278715722Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d, Name: checkpoints/512/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/512/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=264cf3bce619c615ba3245c49bb73d6d848349eb44f17bbb4f9ebdbdd069d9b683ac4768dc5b3e9bddb9ace35a72c3995c7f48bbc947d00a6f10a04a8950f9ae4929d3cc48178903aa2183a4ded9e140d6a0c572227218647784374389fc41c71c5b763da169011e3a608dd95a4668733205cb52262792c75515247e01bc0048253a69fc2df8375e83132152faae509075f66d6875ba208a6f86e7ac455cbf807e6ae74bf141549301b797ce12eb81174784245ade25976c1ed6a400ef8176197c3ea286e908a036c85fdf959e5bab8f4ff0181080f3273ec526b64148d4db451ffce57c6f52ff12ff02529c5d33d8fd10038ace35e9d42e148e497ea5eb351d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27877198Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/array_metadatas, Name: checkpoints/4544/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4544/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5f9879c87d9bfc4962736dca71efa2f86a4dede0c1dd55d01e26eb109253836f89a208ae154d4f2f27c6d7273eb25f958593e91af1d0bb132ea9f0856f669ffd26f36735161c425562d7b14c25661096f5c503ba606ebe104efe0829a08786481979457031663576eff769ca40231b156dc12d1bd549bf6b3494f51c54b78a738ef9261d73c952388cab7baa6f2fb711c7f430772a392e783bf1b9de681334ab151282b9647b49c3e945b5bd62cc195fc5ac79b974c81c32e0e2f1435404fb8518c324f5aa6b8a3cad1cc3cef705866e2d8dd304c5c433c50bf3e299efaacd2c87213b15a39bbfa775368fc01cdd2cd1cc10626659671f5e74240bdf2f37d53d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278772707Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0, Name: checkpoints/1088/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1088/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=29ea2c82391de40b7886ab38403633d0d20a7f430a22f967068a936bd65ede2206ca8c95ed6fa023711f4796e95189a8675d941d6ced5f10a37d230beb18356511c138eb39f78f8bd3d7463d09f5aa5e3ec561538e480d78c161206b44fb334eab9df8dadd74defa8c69279c6efd1876797ea338fa2a2c3ef4828d09dede7edf654b9c776e6564479d1b93d3e99b9901162f2d33b433cf6f01c31058b4830d43106cf72d21e141ac68fe63e172cddc9fdde1d65b8c3db8d6cde6dcb5f954a29778f44a0dbf6f73b39c3e0848d80e6caa10f00de4239a57cf5fd9394e043d374f343d451ac057d5ce69b6c9637361e74569f9fcc50ce214756e5ddb57a8ce1c65&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278771976Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/array_metadatas, Name: checkpoints/320/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/320/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3158f2b4f687ec8506efde373181b16ee7f98b0c36bfa8102fdf780805bfcd5d75801c904284ea5793fe6f2aaa59c4654cf69e5f4c9de9103562fa112f59a51acf22cb61ab428df657ee9e3fcbde74a7510304e165ccfcf5167638b6c880a3b02b23028eb2efa53c6082a616ce2d61aac282c008d4a55d296215c3de84437c700995edd685694e005c17eea14304bbb89dc3104c88fcb749e75bfeddbf7582815f3a16be0f7fa3952c08837bdb3fc303ccb0adc07b4032da95d0fefe6506f693d1c1c439c1f1713f8ec80c65bf1ecd3e4f4d1372d14806f9bf64535a5a8b1c84e1f26e387920b494904a387e95c452f75c310ae9658a585b7e06f876c58f07b8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278840905Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/d, Name: checkpoints/192/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/192/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4793df6f31889dccbc4ef76bd2775af0286c785dba8a0876629866070eb88bb2183ef6012845ed123ca4945a7ba1089518fcb15ebef27bd841252002460f9488695378fee2b016e1f9a89e83cd8e4263adccd4c66aa1c723b951fdafe5d8d1bbe364ff1194547f0adfd2387a7dfb05e45d85077249d3bceafe2aa5cb43f7f7b15d3caeba96b8a78005ffa373767f3fea27bca7a34f2473df5e7f6d8b0fb4d671ba782a9ca40f538e43aef322d68c9eaecca271349c8358b0add08fbace100bcda74a5ea0a52982f8902c4623c1a5ffbe973a28cabbe339ff8c232b49d4e39c32f42aacfb7ca063c78d9e5f0ede18684abe21fc41fe5680adf2aa3363e56c1d2b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278873979Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d, Name: checkpoints/4160/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4160/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=214f79141ba28b195b4f750521d16bd7a3353f2a336528f4cc38e4884addd273b3d7f15ef7a813ed7de29593b4f707fc0972446a2720115db4df27e59c52f3f6e0b199b141e3dd1b745fd8329c606a937e887ac735215a1d03dafa242fed73a5380392fa4905a02aea0855b1978979521f92dbfb46b038b6116c66d0fb85ff382b69a21ed06ce3741dc28ff57c0eb6e042c8c88c5fbe3265532b0332c93c3ffe12084ef4ad952e937959d1c1a5aa9407398ab059124fd83088726820a50d119ed3d6528b8e0f5c42d03f42fadb6dfddf8f3de908ada3288cb9a77451a69f1949dcab450933b1e11bb80271f7eab0c765456c59c966a643c5b7815c3c52a10c5c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278873496Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760, Name: checkpoints/5760, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5760?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=aed115c675af71d9dd51f5151fc91768b3ebc11d9275d581f97f8b1aa4bd95bf346b3106557206c8fd9949a8bbf0872377d717e0b3477de6eea7e7a50c27260cef7e349a4627470ccf85b9865b9aad6abddb08e166087b934a692f8e19a6ae32ff6c356f55e26568afb3983f2136dda7b5364ae603d91a884eafbbfc03921203379961b2dc5cd1728bc2e53a69150eceacb733b0f2b2fa137e63537db28c61015fda8c5ee2492fafe172f0abbc3d3c26b432e3f75bae78b9fb6c271bba734a310318400eb020941af6a053e5b6301b79fb91703d11f4b26f14acf45dd2f28a2e671ecf3995f3449eda0729fb4126dcffe6f8fc6e749754fdad26b0f1a17c07ee&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278925267Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0, Name: checkpoints/1216/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1216/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6251ae8921760f1c384afeb482870c78c7a453c01b4585815ced79b257fc34a9e809f181ce0fb70b88df74754dca9b1066a80a660541fd4bf52d5faf5311c8f7cd7db2a3f14c9e539a7fae1f1035d06cbade8f8dd40c67ade5948061b30e6f8c5ab310812ac23de6401f49bfa078e553ec352d1dda9d9fa58f0f262cae08e33690f09a13221d9b2149bfdbe6eaab395b5b7f78bc35db63303470a91d14541e367877c2b7219cee26eedba643267663bf3a6806356d1aa3bb8d87aa76cc4bb62d88aa54e5fc7cfc67bad19bb679eb4bd6ee7d4591a153eb66446dc11fa5c025f7bbe5e2999ae52de4b42372eafff65c757b4152cf4c56ce56af621cff462d8d52&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277853115Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default, Name: checkpoints/1664/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1664/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8d23609b196ff0af9d930ede7d9933c8deb33ae2ff9e05b5edbfea95d77a77beb4f1971a38cc1de2bef6bc73f0e1edec883c50610128d60a563076126c711c98dd8ab95659854f7c61ca0c54d4ba5ab590db62d7bc460ff6169b53b38992c6dfcfc45c510a0dde02624bce30fa05fd3a56bd611630096e93ee7d1825f9b0813bd9dcfd8e4ee46b7c3861cd006beac3b34ed9f35be13c283db9f33188ccae86db3ec9d99b9636ad78ca65ad078c66376c916353128559947e83834dbbb34bd72c9ebba098a5ffbaa68552298be7d48d6aaff2228a834707c83873a442a882b5869734a10cd6424d47d71af6fde04ad1fb0de735f578ca5f3451787726a387750f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.316002736Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192, Name: checkpoints/192, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/192?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=be72e358c2231a427755add729c544d1c745993983302f614ec4bba212000b473976b80a84939be3cca61f10946c0a041af45783bf4fc257f6dcc4368a76651ee17bccb6cf7ba4b6e74274562c71faac2d9cf5908ac0450875c7b802a3b0f8b093b61c958467f5a69f27bf107e6c78c7d140c9ad4d19890e99a700ca48a036f0603e3dd77b00f7b8711605789a14f3e8ccfd080bf74868ebf149d2083881ee3d7a411c192cc1392bfb81000eb70e735fe0a79b0dcce0833ee9f12847646de1c8f881fcd5bfdd1f1014f5b2fa996c4e38c6b75c7ab19ccab58d601ac956f7b50ee2c5f43beab4c38471a222c4e4471e2cf934003ec7ab95889c4f0925c9566993&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.316045381Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0, Name: checkpoints/5568/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5568/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4f74f6b6934843b9faa8efe5c5c0fff6495d808e647a20aa98d91cb3df6b27036ae21d80d93252a683b6cfb6d3d23e20a4890ad6d8f0f16763f808273633e48f85f1e983f9ae238ee72bdbfd6368fe0dd196d66e693a680108920629400f29e3a8ce646f76f1870fbe62f0eae35d6961d1824ade3397c70eb4dd75cc5f27a53de09fddaa01bf4813437d0c23b48d6daab7712960bc4520227dcde13aae4b682b9d6d3ddfdd2daf14dd07cd3a7da8457413f5b5303695ca5883aec7c9d350323be4803ac447767a53e3559a0cd74db9d3c7bbae477ae2cb17df79d8d5d80f53a576dc8b447ccdcecfd0b44a2d5c0c800c099d237aa444d33ff3c25d42692ce352&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.316082901Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0, Name: checkpoints/1152/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1152/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b7f3b21d2513b2b4c473128cb5a93fc8a34b2fd5a83abd1be766071d461d58faa74f272c04795f672d5281092fe220fded6cf4343d61ec4f2df21775e009cd710d0c4ff47e78dfe6d0523fdc203f1cdcdda67cf785f53ae9703756ea9280839a7b35bc5d41734e975668b2df163ba0d051a1c9e1002be32e3c069240d20c616dfd090a032682fb01020abe2fcde16d2205c34a9af74a3d2c3e11226a3f45394a038de8ab1b62073f419fe87eeb613fa10cd42692ab4ba86bbd000c511a405363e0c5449dd86f50a95c1898563e5d2dd4002c4a0539a90557b680d5fd2426452b0d9a5ec7d0af0ec6896117f91fb1314f8b461c0d145d2fb2a155a654fcee1a37&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.316141594Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/d, Name: checkpoints/3072/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3072/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7f867b5136694c79de577dbb16c3ee8a0225f41ffbc9b98f13d75d142372187029e95d46d2acb7359afb012c5fed87d6baa2bf3fffd915a7e1d95b59c8617a852bb221ae0c1d3d2b10a67a8be64dce26771a8c1f92b6164502debf300af03994e0240be06b9683626ae84d6f6c920afd74cb08c5f20eac6b5a8696a0cf2ee6b1d9f8efa7a0e597118a239c8f7f770e02b52feaaf86d7376cf28197e6f2c107248192635fd0412c0a39e9fcdf7e7437a52ba404eb5b093ca499d1f5f10894da5da4233cb8bcf9305d8ef780cc8dbf62f8e5a39332149eb10224118e9c7ade20fb4a6fa34b75902184d4529d0ef23bcd4c9b7535928407d31b13245821fc8d44f3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.315979373Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/d, Name: checkpoints/3648/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3648/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=95fa5345191421aa7235fd20dfc90790b40478ded3464c5606a10184a0d6d17e246197be173d34b5c74b06529ca071c40ddcb94b829c8e530d41bc816be6f94fd46bfcaa7085cc8b4d36c9d95dad1cdf1edfae1c62bdec10569782c88000aa781e9205549c78be8247c22f7c99ce50931ae9fdc5d2da3abf8ac5209a46448b3164c5bf87bfbfde2b12e014981df94d4fe2b3164794f89cbb283133126b04ab4f5b512903b8a14667a5469aa11905d569a7bedc67f3672bb617ab73207be76d03877fbb01891772285447773d4a3851c5cd7b7346b212e733015a7061b6d33be9d42ff73182b1055016f3c4b56dace0f598dcb975d59e8bdf78aec86014591c61&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.316017194Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0, Name: checkpoints/2368/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2368/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=008e74947d7e5d10cc57a22e0d6d2ab131a0877db4785e6f7c6b304e70eedc3f27d535baba98549c602215be1bbbebf6e79ea12de219c5c17e301ef4e2088e142f8efda329e2ddd4530d9b5497407a64f7edeef94e7fdbaa40ce7fd172e11a70f5ae747154f23bcbe4b1f9c41cc34e507845321ebf7347c6a2279cb3382a37f1f8d97def4932343559d94c2211566dc29cb5decf761887f778c2a2cfaa0af55a22281bd331080c45f47935d42e67506b9006a3390172d992c2a0f8b4b6cc08eaa52bdecfa64cf6485713bb45c3bb5537b0a5547d743bc5f50f2fa57bbb0821b6efaa4b936f21a7e82aace27b5ecf571628792bd3e037a8b336eee792259b742c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.31616046Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672, Name: checkpoints/4672, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4672?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4061faa2f8ded8689744d7257879affff1787318f5e708f374786f1081711a3cf56b044310fad2d23ff3b72f8c096098e9eee4fcd04c29759bf8e262a6a87ee6935d617af40a8cee5c045854f17c939fa9ee69652ab74e9e5124a685bf0961ead6c2c46a67c0e245f6b7b498575a5dac44891a792703158bb504ba93f43d2e279b8ac022b5bd04b15a349ac79207133065d8242ddf4a8741ebb0340fa323396c86f628836f57b128b8ef452961baa8bac704c19a1b3929f743d6b3990214d484df5b427d1c9338d723dc8e58391b4b8a28beb40a85479adc405bd3cf7aea917d223cce5140c63fd84558b867f9919b718aa9dc64270f4fc037ac510cbbbb8339&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.316163539Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d, Name: checkpoints/4928/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4928/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7971857ccfbd9d69cad91b44dc978517c4beee942180a0a9cecdb60ef8a2188287be388f54d75077ea5f9195490a0ff7373efbfea027476741225727b097947453d31b92ec17c13dcaefcc1ba19419b10537c5b97506506d2fc0bd777c1e120758d78a6a6135e207dad0417c3d67e7b8ffdc2710640ac6e119ddcf212495fa7edd90452cdcb03687a609083aa2f69943d63f112d2b147580d7acb55a3891f24f2d40f73aa5608ea5f4a31b16e728940e95c722b713057fdcb2e50fb0710036a10ba1dd164610b9f37ea82f47ac3af9ebc13ef3769234d0c55bc2ab153219fe9dcc13cdc69b2e78afb38302bf3a54959b4ce027e31ef33a478a76fe3649af40ba&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.316555032Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/array_metadatas, Name: checkpoints/576/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/576/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5e7d4eda1c38e3b04f9d3052b35eba4b32c71a85b5cf18b8eb60682421c3f141c0a822d848f089f434ef18c58a196d06eec73aa3c41f9ede28b1d460ecbe2193043e6dc5b8f0f56bb5693fd8f32df3b545fdb5bc35b5e483badfa4b7c52d1138b69829bf3449329139006d59f5dabec1b9a0c0a3d30f0fae674e05a49e2a94b2964ee4868b10a80eed8bae870c06ad7116ea8d7b065ea232ebd26d86795c3906b4b793893e1e980898d883fbfbb95f250ce404bfa25679060f84d7227d0dc5ece068bd2877b73c975f63b56c6192b01dd81baa786cd519da2cde348c7697973366b15a00aa3add2ed325a1760de61e15b61cd290b63afaf05d84b0bdb147cb19&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.31615544Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0, Name: checkpoints/832/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/832/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=bfbdfe4897d4bc582c9afa1f2f17e5711b46584490eed3f6e9d8e67410525ea22247ae96ef53edd76bbcbe3dc5b197c3aa44db55a5735bea379cb7110b2305b245bd56abc846403b4b7fe6e012febe2f1e758798436bb9555ac7b615098e82f0295a71b60606280eed6207f08bf00525808478b9257d065eea626e0e49de41442b4186ad1329f42e7414b56d53967d17c8cf63a432562eec6e062376d3e59cae50fae7dff4e3e03834966f3fcfe3a6ff2fc7ce825768088f41d82f106a34e0fcca253d8aa87ba3eb4a8094be367b9a7fe047c970f8c7aa840148344cabbeb3b09dd25153c79c258f5ffc6fa9fc91dc5e567cf4a88f39486e9f35b9fdcf1e87dc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.316138103Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d, Name: checkpoints/5696/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5696/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=55e15d05aa24b648dd8995d81a7e7a12be7c9c9b0e24ebab00fd3e2f3d15eaf4b141957071686a4f93b22f543d4ee4047f52f50f96637d8d81b25830a216e56836e2cd82b9243f802fd24091c22eaba0bb21f633161f931f85f3bb93fd76cf13866ea3a0659279d5526ee97d91a1d28daecf024e19da588a7a55b2cf4766eaea7aedab58c52ab89f2007a1a4567a8409e272ff37a4dc58024d8186527c2e56e6ae69178842ca431b8d866270cbb8bd231699d4f7b8e40e68149df0b92f967b75e2c35195d06af6c650016a727d4987c8565e7a391bbca5e87dac0a46572b74e2fe80fe6b7d8b7b02b7174fedeeac2d37f9717921e1eae3a546bd849dff7d195f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.315992036Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0, Name: checkpoints/3008/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3008/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b68c6e807d391d84a8695d248ad745fde6011cd2a6f719b585bc7b10ea8409c58af0c7978e9f0f2f4246fca2b74fd5f7b0d3f5e170d5c2de01f1fa0ab5de7548efa25bbd84597dbab0a5a3d9ea87ff487a36663b687c231ab2c7281a3aae9492ef91990d264d464f9b35ec77abcc13a30860ee662c965c25e531f826003d93a95d260fd8c2346544fd88e4eced25ac3a1db2676b5d614c878c09d746765414df45ca827b9c7408722ce28d63837011eafa2c5db686708adf82b1a9d1c10f48e591d5af35c595d7bf610343d19ff948b03783ed4f03d05261f7e8d3705da5f47de3e3e3d7177ca6e903b5534e9b7181a6ad8f23a945839cd277333673df3415d5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.31832123Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/d, Name: checkpoints/5440/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5440/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b6b8c342083c629c8394ea38c848126ed7573c32b3cac1e1a62fcda5a171c9d6da632bff117e0fbe636be506c621844ce030776034e69592395e29c340e7559ad1d2c32814cbcbba50df6c34439300653938669930686712c59b31b7e49da46257a42a72e6e30c579042592ffcd3e52bc1bf0820a10d052bfe307f02211860e11429f5e3b61f8c78a6220bd4a6efe1ebc772f2beada38a991a6641afe795b9de8b5b0ab2246af594d46944b374f51c38b5b068b2bb7be0a006271cef09c3377927112f935529b851d78212272e806a57b9d48ae19259d2d2f3d7c74a5cc913a51159e121fcf6a4ae1cfae92321774d4e23e01c4abe9cbf5a5f117ff15b05df68&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.319430563Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520, Name: checkpoints/3520, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3520?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=428f39c8118caefeee90a8522ef0176dc4d150be7e8e63eca7e5b352c85f8d13ec179965a0987bf58c54ece81fc86d82c9f45511fb457efa58cce1fc5ec3a741202b263527a6369be1caf225fb440275a1821c2002ca5e8b0b7771a1e293e14aeb2b1dceedb27989d3c6ea2d10c7c6bb698167c6dae4494da4170bace3f9c57b154b389a91ec3b6637465e66b62f0fb2a1459707d45b15e1d1381b288e1159fb6eb4ae41ff7dae0cbadf3b7030cee74d30a9a7fbe645026956bf246c735f16dc406838640deb5bd5eb8f5ad6e0ae1e98413e706150beb904b35aea2894e1be23feaf24aaf23042df88597aaa41f1fa6da1beb3f3e39b29d071f76f8461cc64f7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.319528499Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default, Name: checkpoints/256/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/256/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2a4c9e6845688c93bc96597a3c8c967f8474b6cbfdea77f138e3cca33460695d0aa1e20d89bd473f0619460aab12ce3a35e14796d26379fc84fee1f37714e1e59217d0734fabecd41e66fe99cab52740edbfbce70e12b00368b84ee9ab622328bb6f927bfa0ae773cee561b3a7852a511f6b31fd603d184d630d8617175cbd922fc93499d503dd1f51f8345d7938490e95e01e9fd2df94ff3d167b00a32c37db6e567dbb7736ba5152c750a86a434407bef11361ccb809a1fd701e00956440a63b1fcb0c86f586ba0441071481442b83c99d713ab6ad16f05d5cd7cd1dbb4e2db86ac7c7aebeb511dc7ba610f28865c980946c579484af0dd9a75bc373e7698c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.319551566Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default, Name: checkpoints/4672/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4672/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0c5195f8c05fd2ad7dd63cc6072e15dd02d08984f78758a15a9637c8cb8dc5225d9e676de2abd3d8eb89468e70aad3352d546b0d2e964cbcf0f5b19a29c3620c1a7ce1f21a932bbda532bf6a4e79384083df8f0185542f20e8011fb971b4c5fb61a6fd11d8b24cd9e18563733a057b75b1545d82534289fad416121d119bcffe7230cd466afdf891834127e726f1934e8e63f5064fa45cc4120234b97dffd8a054b8b07a8e5542ab57b42fe4d87c41fafab80f27fc7b767f861fdfa39aed92373003154fcd5d649fdff48b57deceadb934e2beaf42ebb9dedf38fb1303885c24677c4ab7e574bfa85920275c7fd09032624547128fb754b2611865a1089b229f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.319557868Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0, Name: checkpoints/960/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/960/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=69c435767f0737305ef5642694cf1cb332add2a0ae8e411aac50e31b6752e77f797ebe63c8c5d9eb0476a4fdfd40819ff4158fce84267ee3830b29922aad624c2fd875a4991565e4ab6dad5a55bf3efeb66af08c93cb6233bdae979891dbdedfbf2d2f22baaee7df6bb7caf9903700e109243cab4dd0d17d4e5601b5e86ed9a9aae0b6cf51133bd9dec8578a5b95e79c3970c85136ab2b710230e5fd5e88340d61be5dce88c1ef3f7da6c107deab3d00506122487fec69e8dbf6cb70aa9a7ba08c209a5b1ef4bb9cdcb9b50dff820a9b483e071617208ad52ebf15f5562ea44e5a758e26c5204ba239bb1c93e874f4ce3948319120687074c2a0c6e55ee56030&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.319581067Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/array_metadatas, Name: checkpoints/1088/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1088/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=ae0b557f140362fac39cf28c097e234374a9e70a356b97acd14e44cbb9cdd534ed94386ce6c72bb70bd39a578efd40d878199099a1b24888cace73934f5b74ff73d4160d91c5efb37fb30b1da2c962233f31eb75fef36a45d2cd4fa225a8cf2123176c48ff63b3cc7e44906b631e5ce26c0ab15e8e01fdcbd49ab793687db9272df10ec351f60776bdaaf70d3fb623325b642bb392a5a037a424483b9894797992c751de2cdafe20c768ff36464101e3f1941096fd7c50b7e35b28dcff873d282ef82bcd70823d2af6e48e4faee3392274344f5e4657ff87aaace57a5e6bbcf1f1abc9ca09c072a32a5e4b63eb85ac4646885fdcf07cd74b13475a2c0d0e657a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.319649838Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0, Name: checkpoints/1280/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1280/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4bb4f050ba40877ff82d371591630c790c911a28cbf57a94df73e854b0634309d5a84e59eccdbdde7f7d850f664b8dd79c794bb29599a0cb8f5f1d06cbf26fe0e79498a8309714935e4a53e44b47aa60e6d949c1e6fb350735814a4ba61289f925adaacaab84937394cf7f0e4f2bc78c10134ab7b1f2285cfba0dd3aa4e6cb450e779cb07b97c14afd946c0313cf67be94476d8bf3fecff3f40bffaa4638abfcf63042b515680b41ed3ad3d1bc57781be144af3c707b223c4e7498f9442eca9f1ded0c39a28b2d339b7af78b9432c99c7cc9666cea6fa22e892bb61d443bf7e3d233bfcb10e4558ce99425e80bb3196d0ce2db66226b5986d5d4f2bc98f26f69&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.319729901Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d, Name: checkpoints/6080/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6080/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b61828ed7c4f7ee7974d8d4cf1c597b3b6175501ffa542cfeef7069aacafc8029c8b3e5148b9d24bca0ed391f2ca40fec34f7c59bc7c3c4d85f83cb69f818478a7d5cccee48bd5d593a31e66a5bdaf210376e773285fb0fc84cb68da33b05b241c0caa50a0061c7848d85230f02215468f2b6e3985d7bb3199c503904867c7c935a2c624c457babf61881779f820fcf966451c558b8c2bdb280aedfb40a55e238129d608a67f13bb05dde9ccb331f0b38f402e621bf03b5ecf82cc356db01c97c055e145e6b62f49866d99420481d71823ffdb71b49b4c664d54b0617ab28343647e6091fe0847871c2bafefec2628d2a92bb3ece5fe02ceeca755150c074cf6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.319782421Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0, Name: checkpoints/4352/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4352/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=79bd5942c15e58d3bfa9a969588c2c3ace723046f6b68804a5a1343531e42dc562acc7940de25b675b91dc312c622ddbb430c2a08783368c455384012ec35d2188a283617ce691ccb1f0b6367cb41949f6537f907fc3da7f40b2c9e8d0f7b53e30f59408a2482997dcfbba03d27d5b50b72f0519f5786ec61e6ac106acf63cf8152c4b3cba537c505cf8196c85ef1edd06bbfcb57d2e13f3e0871cbb49f3b61a95a3feef45fffb7b548f60d0428c74543c639b872a9fa569cfab2f502cccdd2f4fe2d489f0f7558b3611751ed6a55f168d3310d11c77223c0d199ad7eb7a7321c112902175469709d832a22534724207abb115c89e5d80e1c78f3f00d2ebc60f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.319799131Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default, Name: checkpoints/320/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/320/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1533aa84aec195ffbf24b6f25e531ed6daf2e0d21fc963d7f1a6319249a529500f0236a3daf176012825e5a7c7a633c046c413d0064409af6cee6097121ef2be16d972465e306abb659f868f54e555148a9249c1a2390780e2cefabf1fb81c0fe7d3473015df369cce3a5af0ec95d083f506c4f898b713749c7a6e2408f588cc0a7113c4ff3f10159ab3a8223e04e505c1cf6823555261a91cf7da66af536451e5c1090eafb994ebbe8fad7f55f9e13b7ebd36e0642910981399828d77e56b156753ca0ed30bed91ff46f1b2e9fc29c70c642fa12163941109b95910c690b54ab153e821bdd358149f247d0962399d971962f2227e254c538ec7dacf70ce7282&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.31983686Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default, Name: checkpoints/2496/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2496/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2e7dea6c1db9d7af85f2a084865dea6ee3e513393c8a2c21f88d4f01d2f24b9591da86751ac23408e58d99be9eb15edc2a5d771a25c9aa29f8bb8c2450c7bc89ee713c090faffd9f6c4d3ac219fc2796eaec0460264e04abe6c7da8f577c3f90db7c011d6a848e18b767da18110822fada05e7adf21c9f8f76593a26212d2cc93367644d74729bf8ef0e956c030941de8df2eb6c7e2e319cf29cd06a4e64c8087ed53c77885b89bf9756694c28cb129c270dc20ce89495cd65285e968f24ed0855d8371b12893ec43cccd3033d0b05a13929c10d8112bed04ae8f56b7ff53f7dffdd37e8b15656d664c4a5a0c74fd3700b426d47cc86f01868021526a831734f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.319897026Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/d, Name: checkpoints/4928/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4928/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9090f039b53794cd5f3a08f182ecd4da9708028e87025c4ab33955b72d40a75ec4fdcdd346f3a8bea280f04df6bf785cca41673e7d445117ceb3913bdc19fc59916185afbbe4e594c6702725cec3abbc35922dc3d3f6569e4d6b18905b7ffc4635c2c5f1c1965320d2c040a5cc2c1d261cdd91a80a9981dded69d6659374fa895964b78185fc29612ecb901c9b2a73b9050dc15d1f98029477bbad349aba04a40d82bb2947a88e03c45c625246ef6c8d950cb1bbdc8ca8ee86de1e6059e775ec1aef369d360ca159593fc0253dbb50a3f4c1fd17e7ea054f2f0d9f3f8a04b69c56327dd803ae925cfc932af8c78d0a08264828293a2096d10a839c460f2eaae4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.319969392Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/d, Name: checkpoints/64/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/64/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=78834d364204b5f938d03284ff2774ade8644a537d4cc3733e15ca50ce50518538111842b4ca97cdc75710b3ea8e91114a2f6b740cfb2952e170f19f567c371cf807522ebc7de1c6ece50503bc183a410a9c7ef635f77ffb9b5cfad8b2f34dcd0e0ac24d8ca23294a4cd3115b64190f69b0c29100db44c0b619c2f0039bd113a52d654c01c865af613d376d8bec4800d47e95193a155d41341f8d568f80815a63ef65adac05eb7588f268c5b259c3e150f426ac31689fd98b5f3cabc35d86f2bd537f95494b3bea46f712f22a690a5b16810d6dba9c231cd72d2d88fc93e50d126150cce6144756558c5a37e9eada76571f621ba5f8704693f05d19fec375263&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.320108023Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/d, Name: checkpoints/1792/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1792/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=87a7f33a6cffb2234589068ae1e7a3fbac02acbd61296303b563ad0cb5b76b0f7fa427e65496a908c363ae7c5f5541d6a5bf8d2b51686dac14967a62637a49592c95db7460658ab0bf3ae87e2ff048ee9c9441c4b44f85a09408d0800ef7fbc197e9835ac66e0868738a3114c5f3e980eed364689c61c2716ba8120269bfdb4df103058833bbc2ce6442c8ec473f7179ba7fd93dd2f3ccdee5948f45dcce0dc93682a66adcf71c27a89e8fca91b5071fba3b80412ff29662cc2393344aa3b22c5c3ea77a2d96279d188110bb3f2e8cc0f707fc0c92c0b0f058fd6b1648703918b5012e5bb06c703d00568e2b349257c765e2eb89ad2fa78fcb7c08eb5834590c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.320178047Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default, Name: checkpoints/704/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/704/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=58bde679d46e8a3e90982102dea7ab7077533e6a5ec574f9dd266227790f52c876f8916f0f05cb712aa049ea58a2310462f9b3ff820371533bfe2f39db102ccfc0b9ddec924103014bf8bc2dd3bad7b8e17af032fe5a1064d0e966a7ae3f9825460bcbd1ee5e52c37c566ca7f38a0f713dd93502de0f062b344654217ce120923cb109e957db0cbc849ed6df24f967ed76807a48e30ce80efbefd981210c12444eb174587bce5a834b1edab8da6ad0ea9ac544f2b427ad9c3e4053450b1efe217bf6ed3d33ce3f39efc1a8fa5228ba2502b57b9cd306ee42f4524d2c338258bad7da8118439cb0028680363b50aa5a6699f360e7111c19fd6afdf3663baa0701&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.320256316Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792, Name: checkpoints/1792, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1792?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=906580b34625802e4b51a92f833bae3c0dd864e6c7d819ff68ba6a31799b3205b2a3c42a1779c38b6a0e80a3fe7acbedb068ea506ac6b82c04bac94437133201df810997a6dbf1b9bf0bc14d12d7a9063b2a2702b2f16936652a8e577bb868ade10cfd96408ea70a7d277ce54a781aa710fd59ba590b38173958d48e1aceb26b8b1c67e3a15a09339d6edb612c9c0ad2b51c1a8e1c5b4daaf6d5557ec84b15b3217a1f4dda78bf99669c86bde5d7ef251b28f56e4fd27fb104beea8afc3dc1cf7d68a0760594b4ddb50605bf9275f97be31fae67ced138faff6107387c42e5a8145d93d7fcf68b4d3bc03b47c5ceaebe2d98680f0ec37874920ea6a4c5477d57&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.320389892Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/array_metadatas, Name: checkpoints/5184/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5184/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b9157e641777a9a5afa9bc5443a4c9c9f191fe23b15a1ea705c403cf12fe9f496a643bccef30d06a2b89816539a5ec69786aa18073f306d8388dbd3f4173a099cf35e5aa4b1a126f80d79bcf07319ed34da94497c85f40262eacd3663ddd5c8eda778724295f268a6bb5d4346820fc72ff41b3c9df06f312c77a0687fb9c306b15c3478e024844533208923ba623623f4f4c1e5d58a2864c25f54679414150ba91b197d77211f12c0255c97ed282893610d09645ed33c1fe78e91f1e5f7f9bcd36fca7577f99b6719b4627b089c7ee23dc84de4726b79b85b745366439f9dafa65d9b920a2db216f0cef3f4800c7aa76c3d58529eec691dae485df0e2f48e8a2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.320440181Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0, Name: checkpoints/1408/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1408/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6aa4c19811a21242a941a9ac6c063b55112f79c07f9737d7213341e5e86daf93a4d9eb012a8c3ff8e3c60442b642a2ae251880d49659b66cac62a9dec0ba46b638f663f844430eb50f98469d715988045ae0d5123d3195af089bc5787a7e865a803c0b40f1c7c59e8229b7cdb37e6923b236e086f80a7d8dff6a9ae586866c4bac50891b538ed40325dd75c9031a0227512e8858a48274386d295c7285df01d3e563dc710a169b140ca9aac3a93be557db7aae74ed142403ff5f2cb853211fd5e32fbd91e73b9d90c5c2d23d6f0b07350c938b557bf297243b1606c506685068088679359038a20c7e8ae933fccb5e591e5992873836772945be9c2e54e5389c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.320681229Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376, Name: checkpoints/5376, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5376?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8a47ef81590765d64fd1133ce3a531ca03af071b63e517d01ea2bf9791b7510142a65d6dfecc4740d2b4402a3f7da92de836924825f0eb423ad0ff703fe52b21e3db1e281b92ea33ad39e312d2ab3bb588455a62134c1b006a118a32303d021de9900ec574b11cdfcba88de59f4818e03ee9141afce7d2d86c3edb897528bf7b7baee695232c9bcf1127b7bfbfe339e34bd14197a7bc373475137eb70a13064a68e477635b7b38717f6a4b056d32811e815e3c1575a65f5c5781e8c17ad54125c41552eb991940acc1384db71a5d796c4db16dc74baa2b211eecd80e8ce8731c81a108225eb4885ad4fe2832e1299f716a02e2fc50bd9b0da4ca78aef94af724&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.320685095Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280, Name: checkpoints/1280, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1280?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=ba9769666558dae26088f88349b17791b57049b4330c250251e110f5174526bf270c5284893b6c5357e519f2ef316ad04b60ddafd6438cc70d51320a37574e83b6b62d7e257c55296a40d50bf7793af4817609cf0dd294d80fc41182ee519920bf865d4552cc3f291dd1bf50b1b8a9950f9ad87b531ec426104bbfa2259fb12056fd1fb14a4436e5a81b40f17105e509e8b64093241daca376305eebcf2008583eddb2fe56bf67db4e5d48aa7e646d306c5ff7dc253bf6846d70c671fc56cd3d01601aeb433d115c7f926182d0d5d17548e9d65ae6cea4eec02572e829d907fa738f75c25429e0a280c5c0f615d7a5b14e071b49dae9c88c2c9f4ce849b6c473&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.320710816Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696, Name: checkpoints/5696, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5696?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4f57c276cc1243edbd4ddd36137d08019f5347d2757e1ae00688949920aa2bab293e73ec711c9351a74a518e80d175b3506fd1caae06e5963ef57063ed3211f94ff995494246d34e363bd43ac8ab885b8fbe0272623119b4844f6f75ad672c6e02677c4d98f4c2ec3baff783260b9a971ed33759dabfd52a8ff3874d72d0f11cb368f153f5eab63218b03fb323a4b5644f52ea80e934a51fc69a9fdcd8f12f8755d4f2411675759daebaad8886b3d01258498cbf47279c249e6c8f4b4585a4d40f6224ce548e5bd0f8c423509b0a5e921cc8fdbc2fdf0d8f9985918b8aedb1bbc2299b98b1a8ef174bdcc9bd60647b338ffa759325ddcae4932cd4adb9c379c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32069223Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default, Name: checkpoints/5888/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5888/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6c0d14c871cb578218fe716239b7e8b6f41e56095edc7fb2563f6feb88e8180892ebdf65f0e2b0a2d97d3d38e0cb3e16c0f44d96d6bd059c9e55d56a53efe1eb8416456b52a8717efeb09b8224cd9ea161b995eff1433114213c84d64c6e39876408a0a4ade62fd7234426c8adfd29200baa696d66b78b36c226dce4ed76864e6011f59cf9a41e14ce3dd58b9551ec30abace52e7dcf3c694e1ffa2aa08d773710408a881f2ca32af9b1c92ad319fddeb9740d744eca5df6fe0b25639b27e4e6b63d1cde9e98664be11ce21b79b42f72c21cd7ad49867d16b69e6f9dc3d96e8c1b8cb647984a88898ecd854b09bce30e83b79faa13e6013b9e9d999ef1dc79cc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.320754448Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/d, Name: checkpoints/3776/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3776/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8bba2bdf7aba2ac850160ca3beed4b0c3b74b7a01e4e258ea26788a989cc05e25784c7629b851ed12c6627c2a383cae3e2b144249224320a971ee8ea3c2203a2c0f2446e492657f2567957cff899670fba6e50b3aceb1a799f2f838309ab26b334bf0a8203df91eabfdd8b0530949b61bae6b439df17934f59a656b8405cbc26801449f7f93aaa7b061f491bd8993a7c28058ec30cc82d9f5f450d043525b87d09a8ca3d4239ffb881f93e0a0f1d84e27627c8ef8361e0c98f1da6aaab6fb33658460235d2d8f9ba2f7c5706ab50b152dc5a4f4423856b2f06efd40f37d2e94b8a525b3e8688bfc03179cba30ae499dd96e8c4535053777c8d985067d57735a6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.320753607Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default, Name: checkpoints/2560/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2560/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=78dcd3670effdc997541c0bed7742c6b24747d5687bfe17c6e94dd780d54fbf661acf0ec963798f30cc0c1a7493f8647afa1d1f8b91d982ce38aaec3fb135c46ed9086b7eb84f26aa4d98c291e19c0ddfc2c1b06fd3d6f01abc3fcfc72d666b86f434b431473b71b019a878bd0068b83e61ba407ae2da169c9754809a19ed53e7f2fb23cc7f20f54a37023bc77bcbb7ca81889f770fa43414ce55a9bf53b61a271430261b5d7eed9578f2d315e3396a6d57b67367fe5a7c6550875ca99c2d5f598e5aad13f49fc98fed53995e703485f1bd0638d02102dc52089d4d4c5d618d5d7f237da1c1fe3b1f62643d8d82d1a83381a893fad76393d4078fcbb32cc051a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.320970934Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d, Name: checkpoints/64/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/64/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=79533ccf414730bd00e07182b324251d131f9d90efcae92e8f7d1814e5c36493d143412c688cbe3343f971e1126ffa14a1ee7a699e3e3b7140dd25d774da4444d83e403e3e7aca186256e29cea9c735eff17ad8de60ee028463b226659c316ef35d45834bfdca7ad9377873d605f75555f3dfd80ff2c0cdecae08020a9d0ff28e2733b8d9a6924adb66bc99d749ff2bd047b8f95b34656f068ed4fe5e16aad929e924f20d4d1e40070fd0380c4252c4d150b0dd4e49609b49c8121d5f4c0f40cd858f97e0e89e66f183d2668e68df1679da25bc362c317217c93d9c9ef1cbe36ec79a1986dd2b2db2c0d01c25d68836ee2910ee1bb50cdf0ed749f7a52e8ee41&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.321200107Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392, Name: checkpoints/3392, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3392?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=01d5adeda194e9802d289b6c8ecfc18ac50412f7513a10df1d1176fdb2b26c6065fdcc6b60b9c4d6c92934d983e93094307c77cf1a5b68ec44b41cb1cdc3f5581026271278105d9c30852f16004d52d507ec384af7343e055fc5b305a63d7db06a812cdfe56a28809ea9f4e68ae6a0fd19ef0ed72e20e7f642a6a457b3a3e849474317eb4ce28645342ff0689c5de063fb313e0bf029c04b57fddae3f23a2a2dbe97204fee4c8c72a7331270eb83ce75fd46794cdf1c4cb4ce983898219d3033c773e8333eae31d094c001cde2e10944050d8efb2ae26ff2e20662aae95a1412fa64a31e79719fbfbf2913cfe3b8c0a682a90387e81f289c0970d3082bedebfc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.321480941Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0, Name: checkpoints/768/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/768/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5dea9480583a7e9ba949ae74e499a1d3151262118e690cbfa4e7ae2e1f0ae39177ab0499e07973f0fe862ff41b46bbab566ae46f1e5312dbaf78943dce59600da64a4ef8d7ec6c7fa5479df4686ccf850026749be36daedc544b85526544f3260a6a5660c242324bc25f6dbeeef3c089ee53dc444eca255948be32603e4cda7fea7bc23a141fe4fc79f488b997ca5dd26db7693d2220fad76ba993d66b3e9a741b8fc72392579bc6a3f900cd8e4c57c7066dda8d0a04a24816f8f4c954359fd8c6db735a5f2fff9c42bd8c5911d8c45096913e493edbd508a8f399f9230f9336ffbfcf90099e4c02135591acff627002fe6398d4da741a39ac95e149d31a5349&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.321515082Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default, Name: checkpoints/3904/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3904/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4101a0826f5e14454acae7548dbdfeaf08bef130990338211d6321c8c1514fcc95f54b60c217acf6702a66d83672f3116057aeace0de149ef1bc5cc037819890c34b67b0d079543ffebb0d1cdbe1378d2f42507286772d37027cff9162cb61bccdd291ec71b2d4065a9fd2504a054aa117e4b68399f72345edf842cb78d2fa1ac1ab8fd6524d9d54db6687c315c0967935024bd655e58eb6e0ce98e2bc8693e06e0ef50bd07a066c3122be5b0cf7c66204c6352fed415f24f4bdb3b881a997c03503b974ccece3b50491ce7ac39b9d544fc035e11032524b090177f642ac6eeea3c48cbaef9fb8d3cb156e9f7ca5e49db4ab9a35703ea27536e308853df46d15&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.321522468Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896, Name: checkpoints/896, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/896?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9f89a1f5075b4aaf4d2170163ccf5780bb95462811d405fbc7a026a2a87fe4bfb36c99a065edab6c861a1ae39f6f20c02333cd5b3cb036c90bc15d2cd351d49b3398bd49bf9205fc7c87f0af5ce1116b30f046f841b2236916015c1e018ae31c9ffcad2a04a5d3899a4ada09bd623314179bf46224be6d8bbcedac81cf4505f98ba0e20a3f2961b94ad664f94260e349f8f88fcff3a2e5af751a7673ac7436b192f2b51cc43a3169e1539962f2379d499e88a2bdc4e483eb31c7cd3119e3b77bc0f6d26a45d3e084cea8dcde1a679fd5691ab629e444271135c4ed82165bac078191d1eb34144e0f014b36513ee56c2b6892645a211be698ff4cf25af3b09708&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.321605933Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d, Name: checkpoints/640/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/640/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3f7a5e2880e2f6b17972b7e8d6e18c3afe8230d90450082dfb8030258c9bff024b0fb22be7348df64131a01d503cb3d5f13c0299e6d214ff415c36c6fe076a038cf283897e945a30c9bcd7844d3c7163e15e8d888ebe5fc905254adde335087e22f6b55ef8c099092cdf0072d0508b38286cd8d45c6da65faeca94ea28c71d393f13845b2e65de8dd79f1b5444bba78dd58506f27b2480466d3c3945058328d054a55a17c306178f4328f7a5797266cc1640c10b31f283cf2f74bef874407b1a0e5a30f5a291d96710538741d763076376faf08e8d93e95eb0ab17a06f782c84ff5946152e0a9377640532842dfbcbdf78b448a92a1799ddd55856d2abe47363&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.321653217Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d, Name: checkpoints/768/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/768/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8b35e6f020ae0114f192a254d4c7695f79ac07b16ca980ecc7b5c2f71e89dbff73a631ad919b6588d53924ffc06b0da495dde61b32fe4c60a4f77612dcdc1b08e20301ad9815c41a9398163c02ddd7aeea8bfe494c18051744d7bfded3d88f96d685c28bef51f2022a928d1016540e942fb4faac80fe0039ff40443e18297c6d67f5b479e067757caa77da1e5013b2f0fe8c03f272aaa918e5d6d99e8b50a62a308511ce7a371e2937aab04b7be43158c4a85da3ffb988574fa488712b639970ef1cce779dcea4a63f484d205aa4afacd2986290544807b50a200b5912e33f6ff0c919e547cbaa2795ecedd45e7071c01922a598bbebd16524186792eb515d5a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.321675372Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0, Name: checkpoints/1856/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1856/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9b84d731fbdc2bdfca34be9194981ccb1ec981280b7f2126fe33985f7fc2a640d767fdf592d2a11a60933bdbb3ad53b4dbaf51f15d8d26fccfe0d9b4442f46799896f8581d303d6b7a21fddc368e9b561080a7b40fe91d699fcdad5b76f9d820cb4d2eea6fc55e26b8bd5e721149619b4d6446bdaf3ea80f28ecf0acebc80a60b13e09953634b773bc482a32b709b62789d89e3f33f2ad0f7c7241a4d68295888a9cd866d31d29c5daeb9575b2477cb5c22eed7815f6f56c0e06643069a0bffc6063628fd518d0d9e5ff0cc0e956a8b0a461d9526b0791561f103d8f2e28aadc749947bf3753b0c09042e7bdc825ad67f900d72c1f203ea4f31d31ca54eb5040&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.321847029Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416, Name: checkpoints/4416, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4416?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=38c33365d560969c6d75456e8df67d3f8227e73c5646af2820db500cebc56f0d370e6d527d2cffcfae87aeda2ca1053e35734b7e8087f927e98dd8093f9d967d353c8e35b34dc2bdd39a5c31e97abddafcdd4d0d724c7f3630f8577d4b6fd38e95e217aaa2ebbcb0ed96043981afb871efe440ec32a8e03f582b7ad2a6ed050553295655769a87f45a52aff96e691a3f3da01f4b739048a06a1f0b9325ced7590fd1a0684ab3668330dce288d7411677bbebb35aa3117268f93c46bccbda4c3c2e75ef7474fdf1b89bf2a762f0a11a922540d519288a5faafb15fb64854695a0c2566626b013d75fb6161dab080eb9dd71d68a6f0ef33030d4d864ca0e8b27f9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.321684142Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default, Name: checkpoints/1280/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1280/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0f9abb1e0113ad20e035e3e0711b6c5d1ee935e779ee412d9a8b72fdfef5e2bba23412fdce9d315a7ccd1b3138a679b514fcdc550a79f2c561860b58414866be40290ea6e7a92a53f3cb2224ab02aba72b470f9560fb6a85e823ec7fb1a4e13e688fcf916cc6e67bb3b54af14e604c1fb23daddc9fd1038774ef3a9a93fa60372ce5de889ddd7458fe564f84cf7b7be73ae609ecace8d1daddcfb5764c40421a2b337fe24a8ab1f29c32995b12b6dc6a2dd5a945ee8e4eb5ffb1e4562ee520833a1c78b6eb1269866cc1367c00300490d27e0e11fddae341eebf3d14824bb07de56c2100da92f60d7dc45e76b9a67af74185709718cf503710570f6030f84017&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.322094989Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0, Name: checkpoints/1536/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1536/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=23de6ce038b7ceb1711c56b22d34eb49901a8e7847a7a303226321cd2409f3b3e833f195440f8992f097853bc8e836bfb7c4ebd94c25ef2500b7211050fb7807ed1935dfbd9874ac29e63df536178b488d277875be0a6d93d3ea70cde8e1597ef62ea6aedf23a66d4ea2daf5400487929c7fc9c457d84c0fcff44e5edd92fd6a6624fdf870ebe6b95d2c2acf3496add2abd7699b76341cbc8e9108b87fad17876e292f2d27964a4a8ba352880da218968b1bfadcdea203cd9fff028834724cd7155afafa9cea4fa112daebb159d3ee7a2ebc187e47f8517ac11edff69548ebbfacf40f62a151d5ea775507f9b77adda9b6871147502dc58ecc90f2191f072fb9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.322266825Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0, Name: checkpoints/1344/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1344/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=09e06c529fc984096d7e28b592f01b11c073b4b02eb3df50ba432ac90d775e9ef8f3d80c6bddd9e25d99101f16844c7930aa698300c005a50d375c99daae78e83c83f2e572d190f802a87d4502df04a19956976f726dfcf5c9a2b06a6a1af07bcabc3353756e3b5decd2a1118a388d928c6842aa51c80216c7cfdae01b2aec5c7029a0fd2f03b6b9675f6aead8e4e09dba0001a9bf574267697f4ebfe622d6cb880ef6a922837851f23c098f76eae6187438d39d3de312157c294aa7e4004adce50df2a2cac79db58001a5db81c6062021afcef0c8a58164ab718275d03efa48927f3b7383c3ba4ac2c640952eec841a0530a5d9da7bea250c21e6e7e0373871&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.322376597Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default, Name: checkpoints/1216/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1216/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=96e9ca14dfe417d5700a8ea2075f2b45903f815a2c7b4dba4e884824501c89ec5ca1d60aa764b2a05ff53af8f0466ba4d41489d5f171597b400c8f5336d1ec8c33f1f77a87c92d77840bb5a1690203cf5567bf7b26c05081f34da0a83644dbd24e7fff041b3aeaf1a52ba86a68146494ca9cd4ab1bfa97e6dfaf03eb67b2e8e448e146e511a909b22461770cdf439b6dc020af1ed671476fa7ae596319659e0f345b986a299ad8112b2b92ca291288aa876e6d24e0d47afb0753e3f315d50b28f0c6fc0abf9bf541ead7958df07899a72e59a1f28a70aea3f40ba7455af3997519e3c5f10125debd94209b58770ce73af88659cc5ad987eb135e20294f234459&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.322410209Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default, Name: checkpoints/5824/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5824/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6bf56dbf8d0b764dd8b7f7861da01f3b72ee215176149e485196680f5162d258e47f07175d3d7ecd138eac752ba827bf4deca6c8d59f27f368cdc90d17bf31081919bba63542fbcd658ac52b75a87211b9ffc90c9b9dc4c5db4befd721dfb08aa8edf16fe1f94ed0cbc7860d8efa6bb1a0504d22a6dd5ff2a5fcc7aed6ce72c4918d0dd21996e211806d1163588bc1e590615fb5777ba45b92f5140a6e9d8d098c650f8e84b4e56da9906f294d32695d3245e8c1d8a1a06a8d1e98f8e31b6cf3d07e2b537ff2e3664619bd4ea8a843ab9c92a3b5e01e49751e2a0d15c6dadc2ae66aa0ac4ab139ec1f418413c026b2f97ce8731ea92e0abb7951bfd11a60d30f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.322302703Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/d, Name: checkpoints/2880/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2880/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=550f0cbaefebce759dbab700c3d91813d6675c44018868d82a96cde0f6da903db34d74f73828109f9d47f31fb18fd49b21c9fec423b4517a152fc8cb32819c17a2e85b386c9d902187c4ff455b302040efbe2123cac265c1a9da8724766a84f60707585a3d8a8434bed570acf9c4fed3cc4ab8903c448d65f6bc9ef9dad43c26b8de9fc792be73d271af272bd64cf4dcb1f8a2f53543bb5482da1f612db051e8635458700eae8da3d4238dbd60f7442595064e30a590853fe2985fb18b6bcd1f70a7541979b501ecd6e2b2fed23db6ca267c471d992d95773b9dab38107d86a288796d8acdb040b6f8d01475c507b54ab3ffbe4b60efcde88ae13ea3b79e6293&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.322572467Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0, Name: checkpoints/3968/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3968/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8509bbd825014a5c4af4764d719ecbbf16747cfe44b0e621fdedd9007a387c0cee657a3fb5513f60c5bce6bfaaeb8bd2dd297580a962cd7ae14ac03c9a50a29ec27dab889509fd70f70856ef998d089957d26d73ec451f661746f7771c8fc2959ee944634ede0f8ff30106066d2c89bb946b5ba6537d4b67e2b22a66e9986f77a3deebb0c81acc1e34a1dbefcd192187e857128b330c40fe7b446bc35de6a69225a2c112be7cb1409633dd6508fbf9eec07acca2c577fa7116815447ef54aa07938c74c1f7ba425303aba8ef4c0c4efbfc3a0fce28197fa5e2bbdea5cbd380267062282feeb247da4556fef2540533b3c7d2396499a25727922971360f855888&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.322635623Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320, Name: checkpoints/320, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/320?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=82db80d458a53ce9248e60309884eb65c32bf94a4ed1dfc337fb37936be87a9332c5af562d21aca9e5867ce969da0b7df189c1b5472735b9a6ad4abb7fdcbe638c4591a144165fd7b5271f9d8cc0e35f09ed22134440b277fd732c7328767dba0924930433e3d72c054647fb13d3a78ac3211f5f7bb66b88e8623bd238da3c0c45872911bb5ee0530ac6fb01453316bf0a21bb40945d30b2d2c189240f0c3cb472e7489c6ad50243e9f48c496012f558b45deb846a3aacfb80164d07bfafc73b392322663019fbce17a66bbfc36c43ad1bcaf8c896dc6fa2450e5d68354b64dfd7a8291878b7dd741e15204464ce45d3292ec37bcc389f7ebc8ebde4fa8424fc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.322909501Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/array_metadatas, Name: checkpoints/1152/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1152/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=73e0354dd6a1083add7d33857bb935ef8434e3d03e0a64994ae1ade4212afa1f2760f75eebe77034e61b2f27519ec8424b4910d0478738d5d79f278bfc0246c02ff12fe5f401e45b6300863cac616de8de648b8b930c16ebbc634852ceca765ba0bdbfdbe694783f294ae2300242a4311d9db36ddb3d55a5e2131dd6fd8905982c267e8b53e9f7b28566709d28b0020e0c1eb76fda24e4f69b9eb9bb283391a6fd64911b5495f41ba75bccb85c8b846a7d75d24a17901ae972d2086594040c6af07874740a69226f5bb2d4ead9461fd742ff83b58b9446dfc1925cdcadad198ebb9c08db1ededb4bc4155389ff9113800395c6ecebe7cb7d898c92c28b7d3d5a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.323044826Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/array_metadatas, Name: checkpoints/2368/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2368/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3bd59ea94107027a8aa0ce8e34445b9b6b39fc7ae752ed7630de74ac4c12beb06e120476509c7eeeccb764af96761f01999dd69fe39f3c493c35ecab8356cd95fb500dbe6979ac7a28ea79ceca155ed5ff828ce58dbfeef6f206c9bf7423b3f0588599da592e62c6af5c52cd1fe496cd74c7f8c67519a04a4958f93dfe06dcfd3da8386aaecfd345145550519c791b84708bd36fc06716ffc1d2ea5372f2d9c3ac8389cc5a89207a44aa6050224739d4787541f358b49448779116977bf490efdc894d1ee003490d362a1292bab22a97c884cf8bbef78cec8f340a1e1fd4f3194f83ec2099121e8c38f79f297b2d25a88327048f91b02e0cf071dfdac98057e2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.323067043Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/d, Name: checkpoints/2048/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2048/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6ac4a5546bd71ad3f9071322fa4ac245f047cf6ac94556639caa5d8d4bf0f344023eb09a65931af0d0f4a44910c3b899e87e806731c90fe96aca770aa41e3e609583a81a933101385217dce1fbf267bcbbf39fce232e96a909a7188af85469a56eaf641fdb94fb2fe69f701f94dc6ea6c0bbe2ff7522c55818bc6e7cda9843faca20585026410012816ef471d3d2962a4579843cd2ad9b6b6213d1c152c7b844d8296c2aef090e8087bee8d7bad8cec992658ffa64b2a278bce4edc1a1f72e72467749552101107ec2f72b746eb1fde0a95293eb3f602d6c42cc73ffa75ceec99841e5206b88778dd701ed0bb314ac463a637806770919044d8e1d3d81986872&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.322953731Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0, Name: checkpoints/3200/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3200/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=34f542511dc1736d6af711e0ab5e18e83883c9a6e601f5e2639b3dc7077c7e62a50d7034bc8ad24b6ccae46877ac72e34017a8a757065b3566dbc8dfa0c04d76d15be85c403c64fd84baae13d08b34707a12aabd083fd61a711f41d4780a1b6ddb3ecce9522c6d1303fdae156283e8a59a581235a14292d153e7005b27bc71a4a4358063cb953f69f0419d88f9d7997bdad3474bc95930f3566c62764d16aa58658e9f6a55167e9bf4952599aeadef711e35fab087ae701619f5102d1f9ab9441ef713b1dbce338b7253eb5509ca53147523354a961a84165fdd8a0bb95d646ba4eb82b33ff00340e25db88b6068c23152ebb4f8753dd899562a5608a94fb0f1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.323237509Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/d, Name: checkpoints/4352/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4352/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2d2812c5101ae2b4811f0d979bfd254230db77ad2d4f9c252ea11565e9ce0bd6ae24e05e7bde1a56feb1d6d89c018d252749f6f8c96ee600247aba32b28709c67c149fe1bc4fe7afd9034f122b2a02c87b3b7141e8dcf43093aadf377668864f2e6e770f670d0ae87c0c90382ec830c3e82753b42240d7bbb0ccd29f4ea921854567ee455e00d7aae6efa9b405d9857c8ddb381098c1b5483b0471c7891a07ea777378d6ea810da38403a03002247461783f49f0f2a5f4d60f7bd36359a991e6fafa508ff70d6b4dfd652e43d2bde751ba8c9daaacb59e1b313baef1fc9bd5901ad49d90847ed2d5aba113fe2bfcaefe62c0eb100822042f4f3485349bec318d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.323280674Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856, Name: checkpoints/1856, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1856?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=190ce3b6616f181f603e7e443e3de78ee37b67f4762c02a2a4cb0c93774cec2a9203e564eff2bbcaa5a426b972545cb6fb15869bb419ed92585ed3d8bf0b07544ea10092ee68610a175ce430dc30b8e8e4d737cf1adb8f0d61674a5c38275baa7776f70e6cdb83e45955d8cc054394e8f217d7e36128ea3a03b2dbb4f6826d51731ddda0ad9d28a657482c8917fc002c44a06fee948371ac018fe8445c2f3c42132fd09c758aec1c4de604c234843670e55e6821f678a7284675944fe7fb69f3e5f16aba999a076244082fcdd6d3de1f2a700169d77fd87073376f9c67831d677486a4c0676a8417a6514963c3113c1c84c3bfab6307e86751a73d474d81c242&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.323249876Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056, Name: checkpoints/5056, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5056?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=85645a917db3ef0460d1cadce36cfcadaa18920eff6517d8d6e942067b3e704a26bc30d5ea1b8fcdd1e76fe61990f967fddd59119d09005219ff8bb73a31fe984b639d85c1deccb633ede8e57a9caa753d6024df692310a827ac5999c50e526cc2a7436db10a9e1168f7bea74b2294b9d4dc2e78e2e86e30446cf310c1d5a48b59a0c442556d290c3594c74fb2a26db3b1f981564aa707983804e2e2310b4c9947e41dbbc44898fd9911fb039b4fdec412ce48e7f56a2dcfed628bb7a5661bed0c37520953683a3cf6e796f5f3c51666cf0fc47d2164a4f1b2f45f4f8793a8873437389858a38b3f3d3c83f4d14149174f14a136f92245766c21ef205e657cc3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.323414976Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/d, Name: checkpoints/2496/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2496/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=99a85d39a4b43f8a66f87d100ab0f72218ac1d0a87fe40772d47e2e3e50120fae347aaafdffbd3b4854072b42272101d3132bd3855a087d590b7e8306e653f96474281ddf3c8bd8c99abd1b00ce6382a3c69f8fafd97edec86c2f0df6fbfe544b09b7a36a558112cd1fb54e4842364f0d3f912a5331dabf9adb39eef17200e8e5a96a73f3f5c4c88a1107645b5bdd05d3f9cec0fb6c0e8cde13d086d5af80c47a51dc1a556ffe6a70a7e0d2cf7f2b8682b8624c4aa965408808c0bed16891d1b74621bf06de22ad82aa33bd0492043414fd9c8514618e16bb8de493c48455e1768b88d09fb7392275d57127f120b56482d21f1bb6cd8c36d455618b8ca538aeb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.323298437Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/array_metadatas, Name: checkpoints/3008/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3008/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9c5cf9109fedab92c96611fd4187cfc79634ce3a183e729e48764362269a2415a390f69a16d63b6eef9172e19256913552e8dbfa9c5402f13bc774ced6b449b5c867946948bb699a8b7beef6ac7834eb31c9bbab887b8d506b02d9a10c2e64d3d421327f2a62d721f67cf50b6dfba008344f4cb601286c5eac381087e0ffd7962270aa9f39448d6cee08135c2cc57ab8ba193f6a639f5a768ad790800c545dfea01e468534100d4a1c87b8dfd8b035a92f37f6aaf6d8d18782295298af60d46b2e7812f4d72fef5b21c44b6b3e3bc5eb742b43b11514bae00e5adc33ea8d9083857e8f25f75bf34d741b902c073ea556b9d591ad2557840f45ffd011f2c4bd56&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.323341113Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0, Name: checkpoints/3904/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3904/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1d3960b7c9edbb063cabc33a814214906e87d548cd025da1882d87d881fc9c9fab17e91139f196cef95630e275ec9c9fdb847bab7567f85ecd32e65b2fa2c6e1ac95bc557b53049b0cb1f8630e7c64e84fac0f7d1f713cd9f319cedcb32c0c094d7b8b858db30e53088bca5280ef38838de35388c637acf5ef300f415799a12f00c672c74f9c5a79d74ad4948d0516fb4c659c52aaa9450f8b448ec4deecb1986b09f89be942ed16175726cced90d52f9e7f67a3adb302fd85fc2640c30fcb768012b4481ada80c3f32c7ede3c4e5f618720779a7c95639ad846496a4b72138ceb5cc2f3244f97e9f1893999eedec4e8bfa1b5f1fa3f87a7770743f48fd5a0d5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.323737616Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/array_metadatas, Name: checkpoints/3584/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3584/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=64501640fb8e8cb525c6d356fd0a10dd0c68de7ac51eaf54b39be1deb86630a136f23f48a31ce0fa9ca591a59c559e601d20a2bd8049ba8f8887d860d2e2bd2506ba3e762b67b96fcd980ba1b68ed727a5c1605f404e1922354256863985544604f9bebbf32468412481f9688da3d895b9a777a2d647bef88a7a689ba5ae2fc65cf0132404269a686ff072f19dc64c703c61138ffa3f96260e7e80142d54cd2f49e20818c7366abaea9037b150f953e4035a3792db9b2aabadfe1edbe043a38922a9dcf124de98da639000ee86097ed87823c4f8229abe32379345af4a46b60483b74a9d1529d0b682c5e6be1fdc7f3c3add43d71af0f7097066764ec31c3a41&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.323893581Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/d, Name: checkpoints/960/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/960/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=09abb006b681f2692ccac04c16e359ced815c9d11679960859903f690f0caa8f711e860ef229e41ede9d32f95fac54cfa33a12ba0019eda1be1f38d69c9fcc84e4b74daed449e143678ee8bbcfea7a55a36bffd0c6339d9626bb020c07e3f11f54dee5fff3d54d01c960f52a39d341293131d65696bf77e81de54eed900c8c4e15080f5244d678ed52513932856c378619710f64d05296188d135d5b66bac4f8aaeef0bd5f300728ab48a82ac3dbcd0bbfa2432f397b1e723d8ffe3903583b1dafe72d75a924679336aeba0a0a004820eb6540820deb36121ec8574564af3157c647c3e76a99b5d034db91f130246e3e04c3998a6e589c20b65e533e653a4a29&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.323937584Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/array_metadatas, Name: checkpoints/2688/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2688/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c142bd36fe2768a6c6c6b478ca804b1e776b262ffa387809b86bb091e28a34b0990152b4ea454ff193e1c1c6d3fad2fb648dda61b9e4cda42830a7a11aeecf0ff59c70eed3983ba5b43583cea149ec3d77209508ab016465f9aa84a1eaa41bfa6478800f51949f2c13cbc5942ab07d51c686890534a574632b9756a660594eccef3e7202e8a29b96fb54b9f380108d33566e2114c9ca9569415c08dbccea20a63a5ab53385888cfe7b16e483dd4f9354faaf538014a186ac238ec93b9132f1aabb822dd3918f02283545eac0c2f212048cb2a9ab2ae57038469d25d0fe1dc7cd1c3368866c240dcb489d475e1cea3c295e481c78b9390120a8e30f7de0906556&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324040852Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/d, Name: checkpoints/2944/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2944/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0a8e8c5faf169443ebc90569dc0dc72bf04584d0e8a2ea78dc0c461689eab768d391e275599e9f712a7fe8cd7e26c9c5769c6ff7f348fb2574a43872644fec227c1dbec0948bd13c8c23e5e4dcc83b91cfa1c88d8eb2b77b67050ee9d7ca96acc51be06e7acf7f94f42ae09db73775823352ea21c7da083ea8415fabeb2d2efc8a0d4e07131cb6a46912de14dfbe27cbe213dfb26cdf60ba1c2f1cc6329d28e816a93c78d08c1d35ccfb74765f4bc5dd63c12b833a759443ff0e487bb8ba9936917871c7b36af5c272fa6c2528e3fd41c816bb601c543b28851baacf3a2ac933fbfb774f0ed9945e56ec12bdfbf4aef5fe7f88fb41e8d57c7812b4ad72357e37&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324084264Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default, Name: checkpoints/4416/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4416/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=41fe165715fddab59cd802cfa057e5c6c46c3408248bd235c4968c136c2482b42811ea71bdffdb8df1c48eec82ad198bfec1fbc08e8b6762e9bfce394c79d5126f1ef6cb1afe171f35c7b06f79d88822185042b205d9989725157fae5982a894e1574072b51970a8b0b4688eac62985c532515ccf3461225a9a5bfdddd7635a25af7d5ca31e2ab35931b3306295fc601006a4f84ccb5a2933dd05735a3b148317ee94fb67bfcc7166d0b095ffc535d652879f218c5673968c0e44e54bcb49e2338a6e8eee8b119f523692c13626326d43b60e602f8050a1f82e98afbc14b4ce015c35917e5a5a384bd385c40d80a70567f0553105d155f82f010f8da0ca69c43&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324107874Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/d, Name: checkpoints/5760/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5760/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=023e4cc0724588b912675017170d8f3b3a9743e1095f59a5d3a7608f24b785cb3ed9ad760d768e5a89caf420f409d96a11c0990dd452b7dc00ef0507429fdae29cde446f92f3d3b2d44fb321e7ce8faffb942cd200fae4d06eeddab406e4eabe9f2046ddb092e2f89a3dc376d3254f5c8febf1ba9a824878312734f912bc5de25a90388195ed80ac7cd3f6fd874b224e99b46be79a954261a29ca517ab1801bdc85f89a3bab16fd339d283b55558568bb5491a3ec45893670d186bf0fa6fadb514dc755eefef114befd8ddab5eeb94fa222d21b8b48b04f22abadc4868108b655a78a0dead2a481e5dcad6236cbb051e89338bfd12ee55e9db60ad79c5ab31c7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324152646Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824, Name: checkpoints/5824, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5824?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=34ed2dc286d7ba919a195e86e7b384ce8b154da5d9dd7fc05d9ebfff223f164665c02c4041b359963f34994cf9721eb5e69a0d9d38f3eb7f546f3035e4cdc05e79815618a8d550bbe008d9917c17eab0b4e03b7ceee34c3e3ae3a5690882a6ffa1604bb28b24131107d41143e8da924685fc07fa45909fe5756ffacc1ac77342093a0bacd73462bd22959513c169c342c8036392eaae643f6801e5c06c0e669b4e91153b3dd7ff85ab214559018abbef678ba492e79346d6143ac38d1107f1fa30d328ca92cb6edbe0fee6dad8a0c060d932e5caace257671a51fadb14a8549d89487b12ddc0f95f7586a9e4b4183eb6ef3ab74bea376117469b9f0466e9ae43&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324298981Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default, Name: checkpoints/2432/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2432/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=36cdc2802512f2cac35b588500f2297b1740b763bff7627cc6f129b55ed758a94ae77bb1ce8d8d97271a6d3477817bd96c052fc1e7078c33dc8bf376bbef66c0bf777d8ba4f4d776efb09d44c3ad82a8fa1be3c0d3a4c65ce0e7bec70ff13acfe849ffa4ab427ae660447e25efd6020d63061bbc5c43736d9fd60185d114c7a78763c6b5a27c2485c9df6725c3c21b738d6492510370e1878f948cc00accfc533770de7b9b38bee7faa37fc30ddbc3f441419eedca00ce88af31a530bed96302de14b9ef5b47067754e3643baee5dd3d35162bfe1547ce118882e427d82702762607c7cc826c96e2623b11e0e46eb892529283fa6c828655b9f47a15b8602b7e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324623955Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0, Name: checkpoints/64/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/64/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7551223621822c8f476c02eb47444f28be42379d4bed2a7631ff7952f352b7de52f556b1cdcfd27629c6a621ced6e462e3bcd4b2ef739409bb3452fcc22e34f65b6d653602dcf6c90e1e772fac6f98f1a5c8446dbbbcd0e54dc876b98369a7534520659bf1326dea42e12d7db433291bb9249b9221c700b0c5d705dad568a7b78fc3b9f2b47830b423329949f9212e49cd1cd118aaead7c720ac559b3f4a2eb681f04c783e0b05953eafe26bd22cd1f12a4d7e66e3a241a20bb2e1b5d461e0b98f35e81865f98adbadbc163bee09dec526cb77621119ed5e88e75538503120f06e9cedb889a8ec288bf818a636fe146fd2eefa5931232e2dac4628cf2dd0f9fc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32465732Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0, Name: checkpoints/4096/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4096/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1ace0d927b4f34847c4a9f56565a11ee22bdf78261c87e6085836a452290457171595d919f6cd314dd51c2af95b2e15960be6dce6232127c7c2cb3f25469da9d58513bc99d3274304271e547e945dc326caac8976c4adaf9a500280a9eea68a23a2a79805d1b9e968af7ad837270165494e70197885d359b9154a2e336fe55d3cd213fb5480bc89f085d98d4d449ed7c6cb38ff0cc33c81866ab59e3cbf34bf2dcfe8089e7f8afc648ffac6a82a58ef0ac31b394cf5d6b1e52739353b440f04286ab68e686c388fd7bf06f213f95f339a94995e62526d73752561ec3e96cde2ecbab86c0edf50b99ca050a588584c8fed9932dbf8d1a17c711db5511c98b6363&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324693203Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304, Name: checkpoints/2304, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2304?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a7b9471607ad0b5dac647026c924bc0721fd6ba849969607ac9e4616e80e94d96d938ba559b1403f9a45a10d8724972889d9270e4e3efd88b719b9cdee948958b7f904658dd5666df75d38baad2da974f27b55aeddfaeeb82774bc010fd4a950945f4adbcf1faebaf82ea2f739b3657fc170254fabe5b80882ef5d07fd5b99e0449f5cd8dbd9fbdb6486c792f0fe6cd056ec4850182d8a8c5c05ea65234554bf20d6a46d42aea721b22045284318e2a8d9feb6a32b936cca93203a322064f97e78dd5aa89ba81b7633eda3d99d7cd4d0252a965e5105c5554f35d85417c52eda00cee8d3bd03a0016a13372cbe65366b627a5f5dc7592fa982d345b49931a1ae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324712972Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/d, Name: checkpoints/5696/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5696/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=36cd9949404bae941025d57dc65bde64d1555288d9885b7eeabd2842d59818d43b81fd66c59b15709d180d3f3382bfeca3be591de5456245435c6cce66798dc1311ce2af51f5f9e16cbbca04326d9643b8552318df06886570bdafb3726382c007b0f155b3cd19ae3c5e883b125229ba2cff2852aace57841f95344c1a0adf525eff9463159354301bb250869c32f4dd6d5586b50c6143d8643cf36364d3dab7d6236dd044ca94d56e04e668e6edf58c2f21ed807f6cffcb03682398c51d078bf529dda7c037202bcbb94db11ac729e44e812a1757925f72945d2022c7b252fc06b8b2d4e097a285a666098496753329376e6f482acbe93711c4932d74d5af3e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32473087Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d, Name: checkpoints/3136/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3136/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c3d0e2bd6bf63c821acefcc63ab620c24f6626b0ad870d24ca7b37391b4fc5ca6f26e248c27c1e6a37586c87a22c59d5661e972579c6f2b9309194ebd01631696bb44158baa2f0cdb98abdf7b80419d8a511f7a16b5139ca15cc846c0ad18066933be9eb8a25f7041db2db940d0a38adc6d9bfc2cdc91e9277dc78b033a301f022695ec9090750c72b8c302bf31da9229ef8a6fe80e659820dc8541b50dbbabc73482ac8e3cd203d97f696f56c535054e1235b5b0bce0777a6001ecf3ce1942979f13aaf38cf4ca1ef9f8fa3604609571582ef3a53219d389e57bf5062d9ccd3d69e2c25fb50f1efed6be7aaa1e01bd446efb94a629e9d4261b7dbc52ec395f1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324758954Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/array_metadatas, Name: checkpoints/6016/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6016/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6e4686d3cac14db457c11fbca9f4b4a95b93dab05591762cf96d0410b59a3332946721a9b1b00e944cb8aa9b71932cceac4172153d1dd8d0d057bc31f4e8a2da6aa0496cfe3baf7ac169702d3f6164d19138124f9fefb1a65d366712641fbd41b8c8bc334df7bbe45b49d20d64cfcef5b13fec1d741919c671950b64b1e448681804ccac761f1c789412f882e711ee17c71edb99e7e11534a7f63a759e1b856882e1f1e79823bc58e73a23691cdf7c0c27b044d8aed759ca80b346e0eec077cc8aa33fe72c109b2763de160f52d66203bd40c71d4f9c25d717e79c88f9db7f4cb10edeb84b632f055cbb2e1d32f49191fa9fd9bdca76b1db5de4ef6f1f43fb75&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324780828Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/d, Name: checkpoints/128/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/128/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a5337d86d53b0443d06eb5e330a107b1ddab351ae3227c7af2069ec442ae5900f244cce12f71b551dccfeed30b54332938b532132f9baf945cde58ab39a81a39832b44adc25d0c46b2d0d6413fca234f7d62194e166ee52c7afc6a0a9410678263df43d7d6eb632d4f1cfe8c2c7c4720fba1f3277f4b2aa0134bc3ddced5aa878e750552d2b12eea1d82dafd28c825b96d8076a77c3ee7e9c1809fced281c38a69c71ee0a6ee874bddfe96ccda3f24c0d94542adfad9a061e4167cb17b6e0c6e74701cf7bd1a533f5b38f1d8ef685a6af19c799317d328d87612e317cdf4eafdfa23cef4d8f2f6b957c6f4fdfa5ef4db8dd56ecec16e3d9a0c53dd9c1a24d3fb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324813683Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/array_metadatas, Name: checkpoints/5696/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5696/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=202685df0c6255d5b8db4fb8570faa52b01bf9b88c3812b25250b9cc6753acb38f8fb267355d82bfeea10f03e7d4872e976d6c364a606e747fd61638f730d5c4faf507377902b749375e2e4a05ac3a7ddbb1ba18647266a3c35c5d93631f47a5ab96d3846db5c019b20e7e7219ec4c8cca410ba0fec0d61ef869192ed9e2743d37e1b908561067184462043dc9e5e0016f09b7b63b507481fd78f653d9b6a9f43a602886c61afc127b71662577a894f2fec5b1e9675812d7ed713fc4eb323fb55ffd2d7d7d2f20b0115c8eefb1221f886f4c87ceedc89b8f721ec1dfc8cd56a228f2e4c64330c7d96dcd677f7b6a660de07edb2d5e27c9cc159bedcd5878bdd6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324845932Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default, Name: checkpoints/3648/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3648/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=57e0a2bf4646d79ac3c42d760b58eeef417bb83d99745d2321d17009dd32208db39bf207ff65ecb778fa843aff5330cfe9dc0256d21b7f9c0fffcc07a17766acabf8333eaff661a426de0fd7300ff4e7a0875de07f95791c9011d64b82777ceff31bca1b31a7357984e42c9b3d373b01382f79b5eb6cf983deec2adc5d73e7aef24d2b57c61d2ee05b4973d9daf01205cb168a5d92ab8c09edc22ac4253aed87fcea918ec1fe579cbd3b1081c44ba2c0ddbc04d93e821589141d21d89f7b39c4d01fe77cbbe52f6879e0cacb2250aa533be7502d73bbed5b5a763a887e6bc678c7fb1297f54333a4679623f547e2073715368bea6996b8609fac71fcc93def39&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324866275Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/d, Name: checkpoints/3392/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3392/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=83f5e2febaa23e84a55952da2c36a3b4c704bd19ab64cb2ad3852059bc1d2cc16840ec8f88a3a0b31b3e5ee6ec77ae16723702ad411eb3ea0af6d233c24843ca8a15b2698cafd1763771845b70a34171d14291e4509db68e3f31ba0ada4637c9312c2eedff604b0f760fe386891218b5a4a1e5e829835cddc05b87df36d47d87547e188988f2cad259514563602cacdf8804683942864030e988ea88c15ff60d774113bba40e0a25f07a257d749c43040ddae8a147581b32cc97efd28ff70d873cb1852e3b6036992581f36dd04b736ebc8c62536fa424a947c2efc86dfac2fcce1cc8cbb62901dcab70d2663ee2b0948733130ee93680e2deb4692f472769fa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324889117Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504, Name: checkpoints/5504, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5504?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c12312bb8cc2beba088fa89848383f9507d55beb729ad4afe75e491409fcfb087fbe6bf1853809afb6dfcbb3bdb21502c69f0d1937cd898398495b3b68b06d3770fe10e81d148dc0f2b3b0cb6739b1a3a89d4655af73376e1500f063ff23a4e8d426ca89d38fb4fed65ebb849596f1b71f4d376c7fe0c9c4611c9ac0f6c98a23d3d2fe0292c28037be13a75b85ad0a57672bdde39f6d542453a296472becf3cc27e540cbe69982cd8b966ca74a90df307770f66d1057b9964c2831d5d0c13ee9f09e895d18fb7e251d28ef0bd0270bdb093feb65004b2a854b16e5f8583acb689b2d03de64db5a41db281ae7b6ad1500aaa1cafb37900de88ab7794b6ab82561&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324951466Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/array_metadatas, Name: checkpoints/1280/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1280/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7bea07ac44627cba20376092c2ed061d8cb73ae722372a9054d4224d55575be83c7656464d9cff43a5720ea0930e717a43aed70780730ff07298a7e14ef140e87e9e8f6a14b921c474409ea66a2e966b621a61de6d392372affcb0db574ba2dd89bdf11e3d514ce1818bb8bc8c6d043607cb656d81d5f39e02ba73fefa73db95c964171bf2eaa4ca87d2a3cec39424860738d56aaee7224d962dfc726be41baef36a2bf1e22008c253ca7a4e639cd967028ca385530cdc5bd9eb32cdf35dc5dd38c507c3caa57ae0e8e2c19bc2fabc1d0fae7f0d89fc6f8b089a148e84d53555cb62b1b6bfeab8a9e8ba6af7ec6e93db215b78994df900360a4e5619cee6d4c5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324964014Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/d, Name: checkpoints/4608/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4608/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2e98ce6d431fd7e8d5cc8ce5974e40478e58bd4006913b2de5a87632a08e486f68f66429f819609eb18007bec053e69bc6a88adac23dec794ab26e43da14935233c375902410504608f041d0972347f9625b3c8ede3bb7656d5245bd427c48b4b81cfb325743f9a2d5e24d0cda48a79a275010d6fa86fbd65ace5c233c7543ec43d77ddeaa74c29868ca8620393f8212492b56a5f30d7f97c317e3a72646c1cd184e70fdcb718dda2338b611d79aadad2042727fdb5a3fc5c3febcca0596612f7fd2392dad6b1a2e0564f2d31e6db4adb3cd236954ad386b7d8d4760d3025d12f1a43d7444c3aa0a944b43b74f6a93eb160d495a00a5b915c174b50c3fbadf81&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32501531Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d, Name: checkpoints/1792/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1792/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=36c9827548c619d79d2346cc21f8f6ef101a38cec98ce4f0f44f6c0fc320598d7d80be6a189cce886627b04aa058ac46c06f0e17083ba77045a65ce7219460cf351d471a4fed965b75407d94fca5184b81bc2d58e6750b79b7c3f0898669459bea1ed99992387c86cdb8d355cb69b27bcc7ea264823e861d3397fca3a0a291d934f635026b7cb86b22c74d2acaf1f019acaa0b07b9f159c1805a62d2a040c36c33523856a35b08584d74feab0d6745d6f858979a8d1eabbcb8886018511ef522c62ddc4e2a7e98190445044113707e9d029f7d5f86e78d665979413ed4f16989ed1afc3d6bbb47676efc2a6bc44991f0bb7dae51543541d0778318af0bbb22e1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.325032469Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0, Name: checkpoints/576/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/576/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=92d6e3b7fb9d8d4fc22a5459f05bd9a4f8642ffb7a64e0cba9d81983c5ef1a8e5024fbef6451c91853fd2ecc4e40c71b2fc431989812a624fdb9ee9c167e6c86c7c03737dcaee368d8921f8f62e499dedae2a4aeebd786de8211dee08f3cbbea25a16a8850db4bf1afd3e1d0ee66ccc00351b62ff05773b7f198d36c1f236ddae142a8f9f0aaa43286045ae7011f349be041f11d1dcfe1a383fcf8952dd1248481f5e7d232b9a03baaa7d194a0dc6eb9c020684f4cc9811cd21deb131a8eda5fa296bc9534f20df4ad3cd7a59319a46d2073091dc15010d134154d88c67391e260599da63a1cbe6d8c3697d34020d4c2f5a178c4cd940b8c98c77b6864027b9e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.325061805Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/d, Name: checkpoints/3520/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3520/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8dcb05d3ae42fc77c914932568b2f278bcb8f8384a8576425217f57f285ce8fb7da64bffaeeeabc6c34699e51d1e80d6cb714d27dbfb553da9e13426db40061986ea8ca50c816638c203dd8983939522911c2298c92909c117cc1302d52e4087af634981d2a2f83fc5101f4a151551c532c9a5f12b03486cb1d60b9cf91a60c463b0051bdba2e370d27f093da1b6d27465ed5e3d9d1bef6dd439cf56ec4624c025ddef908cd3d22a4113b80c9c6ca572f16d58c11f2f02efe4b3ef8a5e0af5c704440d852f5905c12570fc52e27ad77a74b7ec180a86dded60feede45f3c39d5b2b0ea467a65d927977db0947111638bc06a1a1c97553fdf4c4cb53e82774f7e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32510078Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/array_metadatas, Name: checkpoints/448/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/448/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8e757fcf3123087778c28529527f99becd338271597baeda92d5d607a329e1eb42c2ffd1a013109ad678fa52e1a0d68718d4ed3062f7bc996c77157be8936659cf70a747a2b1d421f1aa9f47d4e4eab42cfd44335f8049518f0a1e6a51681dcec28d5159ccbfae5ec02e3761d59be2980c9d4b5e4a115b208975ad044c4c3943b0f0ff8b0d9e7bcd382b4e3a0d6d0adfcba4ae54ba46d5519deb3e999e9b32fc2999ff4edc1f91cfb9f8c851bc9326f78b3f7a02d533855b32ca92128586b47308e141a74fbb746a63ce6ab63fc6b04b108e6bb0e264ea4c98b5b9f24d7eb148ec37f1e64219c9b2285081d271335ae102e7348504940ec45a1eac9ba686a955&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.325140658Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d, Name: checkpoints/3392/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3392/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c1e218f77df8dc2d583b370349e3a972b39b92eab9ef3f32a41661eb2b46ed58bd473ea2c0fc059cc2767da8160aaf03d23273119279aad0b6511d6a61856a31748830fdc98aaa111fe381b415f2251563c13549c88b9def72dc58514688d9718b77da9744d136e3c2b94543670acb872ab910f76149b06d9cb31de8f2b98b9995abcc71541730ca221f11704ea9647fd5c86dc710aff131357a948b2c41ba28317895e53e7110afe8a0f3f66cd59c678f99f6a52e7051d211d4910ab00d9482c15c00985e316b978b3efdc54f7ea4d382674e4d34efdc132744dbc248de99c20e3df2f01ed8147992f159c65cde65b513bce6ac08dae4f3435c4c229d61c63b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.325185022Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0, Name: checkpoints/0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a06b3770b651e7f859d69d3b5815c9072d1029d62d881a60b3f780d5f1cad43202b431a4a3eb560259492047732cc8b902bacfee735f2685ce4b21172ef213d4bb6e4e08c6ceade96cf4c3999b4ace32e89fe0a9e9881cc3794f7209fabecf990801c951d8e420ea4ffbf9ca066272c71f731c01e91230fe1eaf275b6d9e671d642ceb6f24f6b53544c8b8ad1486adff358ca8289a2488557e6ac51dfa896ab28cf0cb0e87b673878d7d462f9e24afb1bd5c501e67bcb206793bc0a9fa2b70361117479a63401dc5f8de36684e1716d9e8cf1b1ddfbc51836bcaf3e4c65a0accb85d554c5cd7599bc14422524c142d138d6f705d45c3224faf4f86dc8325871f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324729821Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/d, Name: checkpoints/4032/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4032/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7faf1a1d65c5b516d7adceb19fa52c3190addf628edbaa1de0bb6707d36818a34269e2799033af8f36858e65b328fccd5c3305e83fa8c505548afe1bdd1b12197c48e99beb156b3fc0438cffded2495bec5ae7dfd6bcdb2ec60041c9a7c171716d670e815eee7e935b0e92c8fff9a78eb39966ce090c9c80767476070e4c0da7e007af969201a81669012cdb236b1e8f3b1a5d53a375d6a55b20282f8d1541c9bf6b0b733d9db2dfc56f4c1fc889c1d1d69afd706c5b01922ae9902f6db28e2538f79eec45785d5f9bde9ca21c6643ebbda526cb6a65585630863184558e218bd66d0737e972a635545dbc5ea514eca20d2196607fd63edca59b82a75632f193&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.325049651Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default, Name: checkpoints/512/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/512/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9f643d432c091859d06cdfa46a9deb8601506e2bb38ea93383d265d33e66de8b79893b6448d4c1fd3651d82af300829777b860e42594d89aecf806e070ce46e1e3bb7e5d543cd3870d5f5b7426d0205451d54709d4b5b3267e7170dbb2bc9f38512ef5f572cd82c594274fbb09008cf5c68a4fb6292e50448ef90c576b02daf3c2e3e150e510e66f8e9cae20195381b7bee9abe6f6021346af3b0b16bd5c2951ea037e95fe377b211b31f4ae7915dedd3892a3012a7c7747bf19c10e673b2f219b65367a9660eb351cfefa36689129c1dcb27c596abb1a2c1799888c32c415100580864a6a38b08b40966d85e6ff87a62a9cdaa0d3b0274ba1a0b72ee5e12927&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324875077Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default, Name: checkpoints/1344/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1344/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=83658d7ae528d8cea8672b97c617ec29c7790d586ce0cfb266cb39b54e9abc7321b12eafb2ce4364b05fa10391ee7ead4a207777ead98d9f5ba05160fb69bee4208414698e45382cf6ed8eef5252972e8fded42af798000415c1eed891843cbf3bea16eceb41a5b59eb7f43bd33fae4d5dc25f99b9b416973a16439dc58f1f02f9d35e826dddac59f53bdcc3b28a2d052f2127d19eaf1b64897f1d42fa718ecee9a3bc4cb0b612721caecb1a1663c5e1c9ae9f0c512227430af221e698dba388aff728ca071df6527c41e5215cf550f3cc060bf42e9b5d2b7c5f5fbf7d9bb95d16cf01132dc26dcd54856d34b3ce89757f740db11db50c50b6117c94db4e8a81&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324979471Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048, Name: checkpoints/2048, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2048?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=32aaeeda41042c5fd058eb4067bed7ffced8e056a6f22778708bec624fd01329ee7d1035c53b55cb2ea13b34db4510298547551f0194d7b5e6c98b004b50f124f8a639e99ce20e68f61cde2bef2e4fd866ae555f0ddcab01b051e8924c5e3fd2265594dd28013a28d14cf9615111bb8ab4768e7bc3f59f2f5758d4bc783ea902f4c8c3c05925d1ad27e33df99127b584190a042cd2bf7bc60f55cd19a02a82a19987c8f74ad1fd7af2af081ee945023b22a367dd8f1361c03ac2361f5977cbfd68006f5d13ed95529628393e2763f8dc66dc99bf7258f0d704a665d4cd9841ff7e77cf0810ca32f0963f1781076e03931da51df85c2afb7702c270a4b81cadf0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.325314177Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840, Name: checkpoints/3840, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3840?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=28c2530238d76f718308c82128bd0bd0908e9bfb93c4f68d41601b3fa3c1e964533a94acddfa87713b4b1ff0dcf7252c8666630c25db658ca98f1b6873610093b7d3fb8eb6478acfa07d90ad4d5c49fa4e4e3bb1e3c4b94f2f7cbf4fb27ad36da839f448de6fd2ffc2aefc7639b5a37fecfc6d3af1f996ff38ada10e72b79ff75eea09ace192a687edbb34b417943bcd93dd9df707b72f1844e4c199d95acf85fde8e147b5aabf7e1144999781e1af714177d6f7b0aa209e8d2606a3e35d40b7cbbfb0d76cb012f36047ba43540b93ca4d06042be34f8aeecdba49a877002044a793fab899ce59733585fede765c684b926c83b4230928d36bbc512434c80578&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32596923Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/d, Name: checkpoints/1152/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1152/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0fa9692524b6158d5a17b3e398420673d7301aed22b1d8e1fb0930db9eb80ced967ec1abf1e01f2047039b8244459542f40461a8e44271db7ced6a54a05e20ee6f9e4528d62e8183c6d2a1c8965bdf57057a9dbe7bebe240336490c9c320280a2775bfc0d75e5822883e08e07f10587932b45cbcd342483baf0901201d4cb73ce5cc7378e720ad479835cb0b0aa06363ce007523f65b37c9e79440e1fc838d4cf629bda0a738368424396a3df38444849d9f48ec23cc11ec3ae26c5b9beaf979116e36ac95e645d02d87e8c9226265ae2fcb3e0f92a02851aa71add6c2f48dff42ad891976b129ac033dd848813dbc9524ad62de5164829dfc3271a64bdd9ccb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324664371Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d, Name: checkpoints/1536/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1536/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=78ae6faf074fa497ac7f902eafc55b95476caf57bcfd272b5c90247976854e1a8022c032bb95d5f71fea1fddc6e91a9d929dde29f5fc385dfdfa8f047e14bf775b7e70431d3ad4a602e4fb1d44399918a0b505b764f14fee681ae40069e11782f73315c08ea0e8670a5b22e79b229fa9b715365584513c43be32fba178532a0eefe27bc0d60a46e76432d803d8d263a52469821287fd64e4eb5e5bb16d362ac00e0377c121a66ad857ae8c14f4cf13b707eada1238794323eb45b4a0c93b196728f999211aecc9a8b8b6a13aaf91508c86385fbe118dcb185f218706f9ce5b390eb4c2180b2eb834139715be574aee9ec10185b5446fbbb6cc844f3bcf67c666&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324759355Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0, Name: checkpoints/5760/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5760/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5a6f6429963a526fb236b06d0f426667ad4aee77f197054f6c015055b6717bff7821366142667b878ac66500190edc776dba6b32f1fa8e46d5ceeb31e016782ea88a270c8b1ae37777674e879890fcc0b2a75bc466e21fe10cfbe301dd752efe7d73d78ce1b99e05ead22f4097317fdd5d533b636b5df1a39558e0043f05268e38bd38f927c6eec2aeafc17f2d094dc2276dfd1f01212ecf6218842285a1bb4ec223825e4bead014f1909111f77f70714a9574703f91ddfc7e49bf0b8999e693381fc04c05cf6e08f23e0aea80d88c5b2d84c42ef12ed2853aa1bbcae11716ddee3ceaa91e951e995d399d6b3349e87ebd0855277c53fca824c8f7a557a1852a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324747263Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184, Name: checkpoints/5184, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5184?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=64afe75c363bab7ef65cb58b343bc0966876c79ff852e319e85096145a081760dd6f2bb57e13f5e940c979bce269e38b60d2f0d030c2f25e9e4715fed55dc0b104d705e3d37b462beabe806b7a75ddd1a0d1ee20fa8ef8196a13db9f7dacd013b94781bde485cd49973fab0651aa82c26d5f39eba2d2ad480858d6d9998174ddf7224b62ad261d2e452e9ccd149f3e8edcd2d9d6840f1d683e5d8423e560ad65eb4aad3bf4470e299c1b46bfaaa9479bb1d6ed6f2087a4be5ba26fcbd3d07d5a7dd4f83d6828bd7b5b05eec5e65b7c5505357eae8131780f4b899c0da22312dfb8eea921b6a90c0080d25e29770dd8e73b33c484dc922d0ceb848fdb51cf46c5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324785763Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632, Name: checkpoints/5632, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5632?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4a9a86a6d4f948228740fbc5d25e06c129e2760befdeb21915986d890fc0906404b2b15ea7e5b3e2d891086b4ced3edc28a73e34a7013d2d90a77db91c04f2cc5f2e46fc643bc071cef8ac27abfb79ffdd895e0a0aa804393c005f164f94b202e2eeab22c6bf72efd3433bda01b8787ecec835bd8cd0a325196dc0d453cf194c6ff449f3c6ade156646d4417def22bd9e4232337a582a824924f4da145e04c50a741cb65d0e629f6626927864ed6c89941b9e4aa85d47003260837f90d77bd325027991fb97d54331f6454855f9bb0fcec66db625b167be49ae574190abceed503981a1fb70bc83dd254aa1145c00ddc4d69552629db9c68d482b6d4ccc633b0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324781173Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120, Name: checkpoints/5120, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5120?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3a9ca849cf1c06761c0fa63c2af71c20f489964336cf38ca2b5ae53895c860e07380189467f188a6077a46788a324e14bd37574d27df7a68417a63107fe93789523344e962caadf104fdd1e930b309511288fc32fce4570ada2be90860005fcc78496cbd7954b26ad4cc9da2efbf50357779998bcc54afc5c3710ae205464742ca8b4410b785778859c266e75ea23980a962459700476ab1f5bd773c508d047aadfdcde3551149866884a5f4dc917b0749bc531fed8d6605d7820c6546c829dc937e9c0266c9e01fb1dfa9d46dd38a9cbc1499990071850c9822acd19126c66f56a7afd807725aa2f7995a15af03da7f78d8306db1741225b0e77173e5ea0081&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324802469Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0, Name: checkpoints/3648/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3648/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6f14ca5d49a9404dae8092c4e5dc1ce78007aa33153f4b44378332349954073ff90f726f182a40a4dc0cf304e5c6a6f09548c640ddce243ce32bc9110ac7fbfb163f813a26e951c92a23ef5ef0a0d613ed5f52c45c56fb8d02b6a8e1391936d85f4ced5fb463ce573e2a336430d08ad7e335882650d3d9143e3fc721a38b8b806b004db08a4345c64785465512b5d3585110cc3beded7cd1f5cb45b6d2413e66c74c448b201b500adf3191a754c7a041795d9eb7e1435e9a7e7339440ff2fb285a1ee6119fb0cd09fc9eb69545cab590643f65505fb29e0bf82e79fe453f19aa5aaa11b2b15d041fd73e0e0b77698405f9cef7e5396ffa51fe8dacadbb9135e0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324838137Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/d, Name: checkpoints/2688/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2688/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b3c9f8b1e37f505bd82a2c6058df6a8de99aea731483d55ae1801af6832044e5c96243390d599343f29a01e45877665cd71fbcf0ded5c1d9da691c46063046c3318a16fd36bab9872f9f8b6fe1f4ba0a784195a73e4f134296f573b5468c1ee1f68ce76205b3fe8b7c49262c34d671ef603faab1466b42aa778ea1635b82e0830bd8d7b5ab63bff9d355018751dce6ea9a07a551d2822f5a77c38097d46e5defa7214f1c4f4544097c6a80c4385684c14cf5e7af8ff885af679641316c161dc2f4927a06a5695e391bf8e205056621cc7b3973fc65c0c17e4082013f34cc6735c2bb96ae2e82386779ee807cfde867898c87848b030c6bd4dcc9f1882cf7b1b8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324866368Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544, Name: checkpoints/4544, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4544?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0cc0609f91a6584ef1dd5c3f7ad02ccb494c1068dbdacc8f19c42e81f315ad43c6d5516f280ec1ef27b0e01ef4c54ec7e02892e17239fd1493604ca28e703fd57defb4f88d6bbbefa651a5cc4268a5ee9da48730c6ae3e37ef1593f814d55c3d6d2002d981a1a860211e04978d1fc8626c636bd13a9be1179786e5b413d2ded057e3f22d33ba01e47737b9516f2b3775fcc0d7fc26c6909bbbe864738fc71569fa53ab4bccd8e00b15a3413ac70fece31cf8b26a9ef1c5a1ab898cb5e0990b2173e42d4c793e593e4a50c937e5b63399ffe46dbb69023d91bb9fdedb73b6bd6e7feb87c7abfc5092b619ef4c1cb1af6cf3c5dd07890c0b752a157e4e7a0e468f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324856597Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112, Name: checkpoints/2112, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2112?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2c1439aba59cffd14e1f405ca3150c055f2390eaaab58802b73fc2d2d038555596c853a88e413c7ebccab7014f3ac071805bc882cce4985c5a31ff0ae1bb797e052fc683720981fcb70976409e2bbd922330650009103aeb32695e946f2510de08aa44f67b78a79f6dd0e3d2f85b9667763eee816faac71b6a6ad40bf073de20675cb0f03d7963ef08221dcc388c9263aa438051938b2c20e1bd61440f0ab69859516a3c2ba4e241111e11286a8195dd15f897c5b890f2ea9dd78673365fc7fd589a7fc4cde0beb979afb860d0129a51f2e1313b7a33f550f66fcb1cdc5a2606758a9faf0b028886419e4908776770691b39afdffad6cc755229f1d2de905e5f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324887625Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/d, Name: checkpoints/2816/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2816/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=82698ce5fe30ca475e1c263d1b64d485a4e5c47a9c5b9cf4f4a00724857fc4ddb5b3879e18237d6ea3cf414ef094324f412af0172a514cfa616d371ef7388e6b975ba00979582860d18579f9ac792fbc11770dd913ff93f07c5290b9dbcb87fac6a7c65463af1c3fbcfaad370b09e2bcbf7ef108520711e33d738c05f7374f209f88a6e7f74fa1c0dae347389937c5c369f43018f45c9ab5bb0788f0f37646edd2a5740752d83a09405d5f98dc361d72665c1dbd3d57accf7c00de06f5b893b0bdf559742f7c300ad8f96aaf7fdae160ac0cea9b5934147a3f1f22559c158990199358f7297e943881277c59b95010b2238627ba2cb444efdf5a1ee00606eb4a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324928222Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/d, Name: checkpoints/3328/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3328/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0112f4a8ef41a6be6a1606c5330b3609bba026be24eb0151e02414e717ceca9f81eacf5cfc7b96e7823900b1970f33fd008b84cd7af6389a45215fc1b3c152b67aa1c24ce6e3ca2cca534da2cf34e079b0a9cdc1db0df86ee7e2089c370204a9b3d1aa9e6a9d6a59301071e39280d3d49635df5fa875e1c04f03637f5434b3dcdd2f5f6b3a9abb173d7f0c9962871a7103e2f79121c60f9110754e287da93bfdae48308ea6c4291eae16471511b4078ee173eab1a3598f71188419f67a2894ce9c80f9ee1cf514e747f0e0374fb5b5925a10b8fa4eaad1f58f8b27929937dda0844c03a7e72d9af5bf5dd9b24471361fc7ae0cb3939a6f642c53416c236431d8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.325188597Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344, Name: checkpoints/1344, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1344?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a0cd6a172a0cb37ec26dcfd9e45ff21483fe520a44a5bdeceb31c5814a5534955c07fb35ac1b76685d99648ba4766bffd7fb58c110d3ca4142aea7c039426cc040feedff4aa77eda372edb105d51a6b119579b72e9eb0bc9b89276ad8553993de9d51ee4558ca8c3bfb86d69884767e559b601f1fde0ae7a4b4ddb02b9e14022a8af6fe762e26ab1e42024b6528a109540c7dbc47778ec6be51d8a21e8bef9683db105a3624326dfedd2e60462977bf7e86922115bd1b235d21f5e37830a3f01190d7ab2f732dacc819d160629031968843420c8206b07feb0c3b2b7038db271a902db9479964be727e3cb72395242b5fcb6f9d4436c8e1f84b999c593c1dfc5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324909098Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/array_metadatas, Name: checkpoints/960/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/960/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9c0d452739858beefeb011b45d2f356dfe8d1a96560ec7497f8a5705072f061ff83a1b73a453e8e04185d70878f70cbff563730752164a6a7828aef50a4c97dc877263ce9afcaeaae72c79c942321f0e0b53ad731f68f59078029e0256aa306c2aec056a4f106999c833c2ce313fadcee60a0cb06bb15d8c3ea53527ba5a0358c71270372ce2bc16ef12c628c8bdaec01a674bf73506d8ae4bd24624fd51dd630dcb32be53538992a88875d752a4baca9608b8af4379d161d25fb5a53df708247ea30fa4a8c9c469e7242e9cd2f1435f97064e3b1cd711f651aaf41dd80aec44650d7f69c992e06c6193e883a71f7ddef8dc595213035bc9fddc6720d6f173b7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324966121Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/array_metadatas, Name: checkpoints/4864/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4864/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a46198dfc970a8ee13eb03ab69d329dd8faa1da0ec0356bfe9ad43c393ada99ce99c3b695a46ff35f129d01b87a0bba5536ee32223f2da63e78eda1714e00ff7fe24699d52095e0cc04aea820aab48459a7874e88754875711cbc2acc166858e28fa6eeffcb06731145e97e1b4100b7e48812345d6b2879866f01750f357a040b8fed55e924d3e7c6f17aa0ea5845d8e759520e04c28c8d83b9aeb835412939b799d1767d86164eaadfbbbae1d24fcf894694431153caac2ed5f1e329ceef7ca57755da8a56b986fbb72cd09c3b503ea33d53ff699acfda270c0efde8c30ee10e65478639fe6974000497d419555971469f0d44a979896292929e73e991cd393&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326236788Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864, Name: checkpoints/4864, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4864?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=699cab229d7181f2ac5cd29ae3381c39ac86057d51dad32ee0a5d9da4fe20c6cd184259c91346315c3e7fdd6181c410e9bf4b6bed52a8e50b66e609b637fdd978094b388ce5bd5898c6bad1db7efddddb911a47916394de5097cdc699ccf2b92cf2f5de992930d8737e94e8594c9fc4f21b18efe254a10e1a1a536e8382d9ea304e1e17bac1b514d0380440dea95debf5705305079c3924a96e3d6b595b1048c33ec618b5e7eefa354ff41f6335d1bca2ddc3b0d0ee3ea6765a3b6ff5c423ee044f40c82e0c5924fe394052d3f6ce5cc5289224ebffef4b50ecb8a53ecced0deb3571e0a57b805753ce4738dae303c68ff4ad8db5f43e916fb48ea8f05d04188&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.325131559Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d, Name: checkpoints/1152/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1152/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0b3672b48cccd2ac38792e8b73f52c72632b55c4c2d2572446da0b29c22cdbeb6fea739d00220afaaca0185fbba4ce7acf2e49924372d8ce031801cafa52237da88648ff83dc0dab2657b4405376270d55ffb11fd2e519b25c8716574ffff40d5a1b53d78d6c0ca0c3022e114a7f881ed1b5e4c927b12ea1865a897aec398153c8e40676a116330d9c889813ad42bc83f2e2bc4e400d13a9405b24741c354ed444e73c7b3129a126b2e3de8d5392303f2ab299d1a6474a63d754568093a28bc06f6eb4b709b273633dfbe92cfefaf331dfe214560010e7fa2552960f9fc5031fbc162a1338e0583ffeaa20d2a8f857855bfd21b3608dbe50dd16611f944a9a3f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326323707Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default, Name: checkpoints/4288/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4288/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b03f4594b58f238935ab020d7eb7e781f3bb0b2581431bd713ff82b90f7c62128df1486989389d0ce001ab71e3bd4d48ba29c8774845be658533aeaf1a186842dafddac5aab0dd60a1a7ad882ff1112a4fa93e08fa415db73d9e5026e53e4d110db16492d781a394fcb06e238ab0dfbc6df067a9392ecd0de45234cc7cb8b39fc60ac9e36f21971c6398d28cf42287ca6480652e6c691f6c55fbc55222cb2e7b7e00fbe06fc1c7624676142371af05920225f8f792f74206d279f6553ff8950c05b9a95502418cb7ab0655dfcbdb6713038f25a76808c399ae05181d1ae25ed46e43d108ca40bed58e75519d5c3f8b5111c6b7937b9da60856bd20a5f345f853&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326335578Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0, Name: checkpoints/3264/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3264/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9ce38d025a11faf702299bb968b5c1712d48a1ca2047f6f6f22494655c841e7c3afc209526a9b77998753baa75d50bb650a71382163b6f9cf5a35ac0aa5766706d5fe8a988c6dfde3864c16fa39a2840c87f24b60016f4d51a9cdfadf04986f6e8837f6eba4ed7a3b94e61b4c950a2ebfc90e181ffb69b7fc8cb0935e08f4ee4099d203e08e70392a96749c94b888a481cd699689dd53262b7019175666d2dfb2466964ef054b73b7c2ba2492e5cfbd96251cef8fc3327329c9b048b139e77f8b690caed96bb838b9de55fa7bce87696e7a4a7b83713e5b2a1a2701ff41f6412a7b5b95203e92353b45f5daed4611f70e8708f5afd86c5259e04ead6c23c45b7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326430823Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default, Name: checkpoints/2048/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2048/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=ae79df3f6b3068bd76ee4537b9347230ee9153794ef855c290bf71d30613e075bd301aac4d1bce04278c8b6de3b8bccc55a8fa07f0f9d2f6e1efe5a4d6f34154fba9b8dc3c490db91d54b7a4e9a6350f9f9732b84b4101d1d258febd3a8e4b8198397a347a208055690329470863ac04dd813962bd3dc70ee0c6599fd65bee0c6f01e7686d82595b0b3646e0bfe69bbddba12ebde88e95a35dc8867cff41fa41ce5f158c31351780c603f1933738d32378743a363e1865424b5ccb6454fa52471f3b4c0a10cb457f81200743381958851ab714d13ea198449966c0b5bcb02e8659de603195d4c5d2b41ecdf83b589f5fa243fd00a0154af76efafdbcea24e515&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32529695Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d, Name: checkpoints/0/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/0/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5943b68ebee66a7340dec3606813836a7e21bb5b12654386f0ee23af9ed44d381a6069e29f14d3760a49c396aa33efa77e89bcf649c71983547ed8b717ef108d06ff4ba7ebc9bf041fbfb5b5381aa5f48bdcb71bdfa67aa9069f53265c8891d677f3d455d9f39516fc7c1cebba452c5d4be61d3e0e0b3e55e021496f53db499b381910bf9daa94dc4ae8b2604cfdacb1c1568a877a97e38eda3d2ac312b65fb3ace61458e7b1cafdbc6cfc5c8706a591ca88d8fdc937a14b1021539c3e381f622c2895b2e8b1dd219a6e71b3605afb7e204c28bb07640058dc9232c6463bf0626afbd387741cadcb4c08adb4c1fdfb9ce4850e7d40d421875d7065ff23e285cc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326456664Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default, Name: checkpoints/4352/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4352/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=38c5a592313253ca1face7e890bed513da9cf3cc14b0f41009aa39981d29c6a07a8bf71f1d7b6920df76ba15191bd5d0b00dfb51dab1151895eda3da6623fe89d2ecca1e3c2ec5508c07570356cb46d03752d96e7bc9ee8685f425442cc6e8ff76a25bbea9ffbcce4e14025e74c9c4646403076db2c204553d21434765617c7b9b8095e202315a457048885b89a7847762de26f0ce831988c9cf0662b21fb3e20202c56099287cfd522335b52b9d6d7fba665851ed0a64741edec464bed03a17a659d2332f3f9eb414ef2f01c0a722818a68716013da350a2e5b4f8ef94e28f3871f7391e4cf169b34276ebcb4161b19cdac453c93e82350a8a7b0deecbc7357&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326450289Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/array_metadatas, Name: checkpoints/5312/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5312/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8db784fdfa6c378ba84a9828d58b43c27f03ef69c29a5a16bd37b3f9dd12d07807d0cb58dec149bd4f7c5eb3dcd5e25d657a9af3fc3684ce355373593b51da59266beab96765a7f423abe378b0a9e57771c90e2addab72006d200e88f36e722821ab2091cccd6f01f3a12cc1db61c812cd291220275a0225614b81c227cae879d7d3e6033b828cab9ba33df3d88222d8d3c3c8e7eb80af2192fe13557d7979b0bd78751e812ece8276e3a5f6179aaabac4661c0ba425824a47672e74998cf9aca6bda2990bf7010a727fb5040aaec1b9de554985668efb16c1c36ada4fec7627e2307e5610144189866296278aa5132b7ec610277d094d1de9dfd3d8b504d5a4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326524307Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d, Name: checkpoints/3456/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3456/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6f1875c79cf3535886e0567ff46390633e12fc3793b05393d3a7b604dbb7960cf92629db74b0230371893844d5d54f695815b62fc81600b38a1766bafac1967747a50289c73c370ad7fa8f585900b1bf8bbd761ab14cff3c958c87372d1b4dcd1a33617a64aaf4f28a7be008bfaf0456b8775a5c60b8c9123c7ab1f4c0b2f915fdd7e319d677aa2adfa25d240c6d25307eaa821a8e07366d3fb1b32bb9b748db5bf330db4c6fdb50ab2d5f007d3a3d67cf11b7c13f1dd1dffe02a1c8a56e92da0626fac55de8fdf4d383423da428b9c18e57e6d7e3ec30be5b07211267eea44d04a531a3327e0891544eb0dd275e52335a5b6354ba35b40842135f0f6f889c2f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326609292Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/d, Name: checkpoints/3968/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3968/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=41c69823ba004ef47d23471ae7962f6eec1b2bfb8b020ee24cd22175b08e4266f42559f5fc771d0e884b881415daaf8c994545443092004ac842afe8258a150c3bd2c76ab2c45501e5ed1cd00ded674fa9f7848824785f2825494bf542a5a28ce33758224570748c1c01d4b52e4b33c68b5f151df6946169c85eada7543e33fac73048790903bdc14a3037e1db5e8e6a2a9ee04f14d5258bea8857fbecc61fbcf0dc4904f877e73aa5244101128c15e4f4f5e45235de3cf352e7e0f4607f30da135dd50906da89b53244ddfc4b3b0c02d1d77e40cf208710c0161a9a5ced4649367f54f0510f39dab843f64a8cff7788a0321f9bd51e8062d6b423a0c8f426b4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326661342Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default, Name: checkpoints/64/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/64/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=33be788730c18ca0bbe0ef72bd7d6e60cedf231ab937e9db008fcbf3ac6b825f44553a7a3602d68137d5509eb790359a0dba5d13ecdaa47e698c65200543c6e30d0f070d0c17f107a74035c59d25db0f17951933eb7f2599bee54ac8fc402e716782457aae11eee9515f462e9e44da937999700ef5136d44c878cfcdea4dbbb9cd3c0c84c62bcc7ec8faa8b34d065e638c45ba049c6241ddf3b65b4a1b3ca397c569cdad65f9a377bdad2c17d9fb17a40efaaa35f4352a8f47e506f2eb376b510a579e3b517edbc201820b885788ccaf22f45f9c3282df6632148d6fcc3fdc3b0daf0629f813e21ff888181017e7c04d2af865c4b11d98168d9dca4bfafe9b78&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326737033Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default, Name: checkpoints/5184/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5184/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=69772fad4acd599df7a3c96f121ac89d2cca26b7605f4e319a622107c93a4c20f1c82e0d137656b46d45b0f1c03a719b9ccb187085b22681b9ae785241dda4f65301c07eb88e717151ce3485e14329f5fb9074a5da64ebda019063b61ffedddb6134e3dcb1ad11229eba49378dabc05265947f31d6da6d8dc150355c7c9dbb63b853744d6ff195a1c97562ddaf4e6c4d86815ba1a4e95397f17367aa3ec0e423982c803c027ee6982bde856da9a4e21eb339aba63346b7b0cc205a33cf4d3ff06a1ecfb68a2b29523d4af41cb2a5165455ac9c52b109e84a375316682d8cc967a94ad9aa77b431123d002075333b972ea2eadf67928c605f3889ae5ffbaedd84&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326775667Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d, Name: checkpoints/5184/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5184/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=18de6ebb435333b1e59d293c928b55719ee994caa23e2f401021d0af4d72647d720160afd05af084f86f7a1ca903b443b3597befab7913b8b74970d156226fae109e7299ba4fd3bb8bd456715b99d9b367d2593a40474671f27869368d19838c758a71146eefcbed6109686d3cd356e0b956116b2caed37aa92ba45ab9f5b0864c01da8286d91a74786171840c24acd61ec15d4ff208e41acb45897dee7e0af4588a15cfd2436acf1f4681f7f7b37b8bd339412de5ebdd950ac18125994baf98f420f34a0c61af9713f4d239b38d620d65c1ade91924c8beb3ba955537a1a1855e58501b505d374ef59b34137e1c6ed37755dfc12c0f62cfe3f541dec43f9c10&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326881186Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default, Name: checkpoints/4224/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4224/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1d9aaa736c254e5bb4653208e0399debe53bf295bb3528aba43aa1a173503e5c729a7f84b7b1f0345688f723d0d593f6301cec097f4f6bce456140d85106c216e15f27896cd33dede7b4142739fd8350b00c185453a9f62a59ecda78bb4ebeb6c20da070ffd568b4636c6ab8e8a2d06419f4dc4d2081979a23466b7ead7274e50f07e4206a43eb67d1ed4a4c6268b1cb9634aa788123f5a7737ad6098edae531d458bfae81bfcb63828734c1c481bffcf0386c7f7b9233a7b99dfc97d293b69b0610a973667edbbc8fb006f438c8ee91a04b05adf20b3a3d9514466be1fc42fc6b43b5f4c46bacf6bbce3fe600abce7b4364fcf88d4628cd0916f0d18089f8f5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326905661Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/array_metadatas, Name: checkpoints/4736/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4736/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7086994998ecb0bf9fb816cceb72e5d09cb62767e9797ed141f0b49920a6b3068dc42c83ccf8f0cbc4afc1ad470e8cd39a11ef780b7c9573ff9d1469afcb5d9eb849d224d09990b24134ec6f20ad9b1c69c8914320d69f9ada77296fb07d4451a399272f33e60cd45daf87632654c08f6685ed9126da5a1f7b7970b1a3f9164cecb3c75d1e3bda2f41e774f69a2617e67e59ac5d08e60d7c8cad8f36bbe417fbda7e4b42fcfc7efcd4970e78fc8dd9e9c1ddb9b33ea0458c598f65521955b357810556985994cca58f1a84688f68dd93dc455b90f2ea38420af1177e978097f46595e069085aa20de4eda115cabc5ddd387594fd5e2c93bc60a5a92f9e607575&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.327030576Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d, Name: checkpoints/5760/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5760/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2c6321dfcaa190f9b5df19a12cdfc48c8f41d8fc0ad2fe97e18a016fd64843ff3926af7c281ba402881678512153b0a56e73a720cb04b6fcf9a02d1decafd6ac17d68b4a5d1d4cefaa7d148cf387dd45e53a5c2cd2631c1cd1e9a11a6178fa0723aef494400106b0cbfa061385f18d5ecf8d50be0ce636fa003872a08d3dcee4495b549496a61fbe930eba319decd9b95a01dd9f5efc318629fdaea48e3eafe889364c632b805d99e8a4a627c0a6cda4067e77e1189d2932b9766a51890e9196753a9348c522e664153420830bb2c25e9ea92812f9fcbf5422778b9072d409cbfbc42a920bd98514f9fba272666f4ffc00aada7899a5930e5bc6b3e70fdbe911&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326195337Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/array_metadatas, Name: checkpoints/3328/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3328/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3441451a56127a9401de026bdd502581d193dc1c1bc70aa2e65a859219156c8b470bdbf1174dc4fd3d9b271bffc7173e798ff5c37311d91e30b8ad18360457e19ea28b3c5811ff7e0288dd365a88fb69fe271aa35b0db6cce39f2f6384f6093030f52387521a559a40644530414b04bd0b1746b15ba4be85fb2df4ae48d6bb51b7112a887f0672c13b1a5e7547dd66e08bccdf623e76ee5a0f2d497c1499efbcf9f8ad69a50042c626b4b5cbd2dc853c4e885f47b0acde294dd7fa44a07fc1eb70ad4843b3a62caff0ee3d09ee9816775be1c8969d29b6b7b6dffe05050c07f9db1aaf380c103410627f4af1deffcdf1a439b80ff09f5862352952dcfbcf9e1d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.327228646Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008, Name: checkpoints/3008, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3008?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3cc01cde5293088e72f1f7808f904fb6cb140b2a927a56e8e0a89a9b42a33549ca48720420510e8a1ae6893c5e6a4940f31e1571754108b16918746b9e253feb74da233d3fbd22a0eb10a33b117676504368ae2154319a878276a386cf9b6ce8c532f60f8406ed354128263b49b1c4b9808e6e1eb8baa7b16f2330782e7262a041a83295422c1e4b9de61329a0f1069db922ade2183bb73f250e6f70c4b1f2e530192e724c331adeb02c759b959700c8720533f1066824c4f6e9e5ae8398abc00c4965ff1ad6422e0ce53b0f10ee4c8f3c0782828e06cb0128c9cd23fef1ff51d9e3da9e97bbda802b66e629aea789f69cdf260223a13c16e933f0571e64cde0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.3246444Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/array_metadatas, Name: checkpoints/5632/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5632/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=96fa898e03892024d3c14446a1a8df2df1c93dc582277d9bd329a29afa9fd711cfe8965032041b2232af8daa6e3e765767b025f7bd63932adf8f118858c9089ccd9df8fa646b1279daf2cdf01a9eb191445c79a54b15990728dc66d46d60cd162391db03957a35274bbd6c2c11a22a70a802d9ca83ac89622e4bc3b0bc3a9abbd525954a9ca99b1aadf9d08f8ebc0af982fc1e9ed5626482d77d307e6e549d5dbd2d41337c1f22b72f61bf8fb33b16a1b78227cfd0b59d304b86459c16c9f9b155bcc67c52d892997c226a684fd546cd996676c6cbe860c7242fccf50ed7604fcd8b742cf526ec2f891e79bc368ed000b55fad3135b26341bab0312e2ce96209&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324779826Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d, Name: checkpoints/2560/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2560/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1eb1d4319516cb443bd26fde14b251ae24af33a5d32623de0fbbd5b7d04a0431bbe5d049c43ed3f4114259c520ad99e40daa39e3dc40f5ef6ae138dd1e6c1958bdd3b5da1e793553bbfd92ccb1ca3a1f8217e65797c762ef8bdac7ad628bb7c94584405b32927a44608aee4a2761b6c68c85525703c9d8118e564f62699230442e1065747ef399b8250cbec5f542b38d608aafb1d7752b0ffcc7357c412f7c0c393ffe0180345d15d635741c61abf52b87c56343eb3cdd19b03c6e66efd3bec938eb93adc564fb4150c3390feebf40d6a68912830b3987e1543fffcc4efb1b809f37fe6c6c817481eb232642d5a20834bfdd2d35584b4f55e4934261167e4341&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326082113Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default, Name: checkpoints/1856/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1856/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=93928661b8d66eaec308efa99ca84ac8c4a9b00471c78d5b4a262592e212ae34e7ec11f833ac8e6d2a2d3b05ab76012a5b12b1294c9e60565fa59d12c7438ff091f1ff16f87c6bcfbcdab899056d36809dc31e232912f5769639727a6dc7220fa22670cc469b31cda5f73ae5a9fa549da9864d58e95812e1b9c08b646814e2e8445f6da041efb02976991480e158bc0d04ac5d08d5889505bf88f4a034a5bae83f1a54205e38d5306d2058319783b2b46d8a02895614a4ea9c56879022e5e9dfa0fa319461e89e1ecf6afe1f75ff9e9732d166773b758db0c66fe36a93ced7f65a20d8f8855ea6a3a778ea8fe3c4506443008e2699c67f521fd74d39b5cc5643&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.327636Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0, Name: checkpoints/3392/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3392/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8272698e173d4df43f7625e3dc2edd6fae2e358f76cabd0ed68e4e6e7789848d7e70f3c13dc478114583a61774758ed3c06ba6442e1074a8ab62ea104a52d90e3e4b37906884f22608a286f34be340f102944877ba8cda3ad7a283b4f1a1ffb51d131c1d2a786ad468c59a2190ead25e06ce4742a85c0db0fe3082e90a00681cfa5d1ae92f9fed2b6fb84199b363a67f128f279c566ad70c0673e6f8978370c2b6274204e9be2ba57800daafa307c975403706e16c9ee6aa161e17273420504ee95ebdd7f3f19fb287f5204c825a7b9fa61fe5b1575a97840d51117d478b3f0c909a1dab861a0ce61c12634808221ae63f96f4a9fab1cc92b61d0559606f43be&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32680964Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d, Name: checkpoints/1600/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1600/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6bed78f59b89813c9d55edd430f226f950ae94ae5a396ee918de8ad48425f6bc3406c9a13b17db71154d9e66e993c4280767857d4bbce8c4ef760a121b0813f9ca86b4bfb7dec17c1657f3ce738289d3a16a3cf3a66d918801bbb4e838fef5df8c13ba00810009c4f0855b2d3abe416d5f88c8ec17f34e1a6d116a3eb556d1957688a1a303ecf239cf4ccad17b5adcd47b89138830c288767a6e089faa52a50c6910e9343aa4cd829fb40bcd9e0501ae81bdacacf5bf25647be42a8d4ad6af0489ec1b4b149def666550908ed9afa0f3e623b5bfde297407a52361a20da625fa013661b8ab158c0e2fc0ec3f4454422e655b2ba934cdd5930b4a24e27584b100&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.327747995Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312, Name: checkpoints/5312, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5312?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=56789a4912e6a59f713494cfc7cee0e617e9ab0c26a1d2bc6d14906dd43231860307ee5914e3dc89014dbc46e807bb9b92ece7671fa33fbdb4585e68222eade355c43fc505125a0feb266c7e090aa0f72e8498ee011d0fc07621a5d32eba83b10b9da686adac604005bee7597f8294671b1a948dc0fe5ae88c338dabac355b8d3c538185ddddb48e98a9fde13dce9fa06997bf602b32c4b5de56e999be2ed16484f514801c87c101b4c91efc5de3dc0723c70f4e1d3eada8bb2a858c23abc4ee58b5e453400f84297af7b5db960d10914f95b85b927a00d33615e218fe839439eb3a6c39fb174cda64aff66a1bffbabf57cecf1571b8a218727bd1256a07e3e3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.327635303Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/array_metadatas, Name: checkpoints/768/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/768/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7e9a2c9fdb4df27f7bac6d3441f625f56e58b292a46a229749d3a240a00e59273eb23f91944d05e4e67eb4c5a90b897231887a90ead453cb566df01d5bd1629fa79c456651e5d809a6aafe29775138e64b274e1836e2c8c232f1a3ea0ef2ceafdac4d6635b6b433ef9da254d78c153db66462a9c6497473cf98f1727188e1380030471683f2749c3446fbc6b292f95d14c483d15f5191a6098ef02bbeada7f707887ac6c1e8cd329bc2d17b6fd944297b6a1a1ebf8eb9e5a133848db3dfe53688e7809ceca3321aea01601f64ca5a715eaf38f2a28a6bbb7d0f016e5a9a85a7f264c6260984734843d22efaa040d50c7e653485d12a5075950774678feee86cb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328297656Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default, Name: checkpoints/1472/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1472/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6fe809fa468fd6eb54508866da8c77303fc4f601e70aa73f45ed36405527521749cacabeba3900500595ecb814319c4ad6da920f4aa34f609ab20f2d24b0611e500b15fd6959f00ac112203bd392cc14a2316fa8d4ba6a5cc1a9e62ca40e739d7e3dab2b27d41c12b93d5f644e72c2136b01d2c260ac4915553f274747b1fa9753c395290ae2ffda5404ab0814923004cc5604b2a1cfa99030b0535a7cf0f362b39b90753cd40414f7de1fc1e696b11bce9757ac01186865f0db0e2a5e6353738eb6ce8bd505b55566e0b364b96adfd81f9efe70cf28c3c58a86459d5438c42943374278811d5f6ad3814709bf47776ab0ae0f8b59fc5672d8191cd88d9d8600&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326641084Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352, Name: checkpoints/4352, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4352?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=291f0b9e79f729c3b6fd404826c27fa4485fe0aa3a71d420f3bbf2c7ad21d54cc26f3820b906f85e9fc43f8ff7891f6a008981bf6c152138f5b79c6fb4f251c79c64ac9f39fb355ca810912b12b3581b9739bf3e2e68d2bcc0dc0c6c329b4a13a9be055ba1a863928a2e8ed6aaa8152dcf1f46a54f65265575e47f5946209af46d11f2db367a9f49ce2acb6eaef24da3e8c8c9a42352bd7b2f2fc697326576bf9b49bc97bdad87d3078d6b655d594a072f13c8e2b4da20ab59ccfbc6af3ce7b338d8bd759a9122c923d68a531cd0b1a3f76940ff45416d9a2ea0ed201903a25c5f410c3fc92fbaff435be5728a47d3749424093b1a826eb5e04a84b0c7b356a0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32844769Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/array_metadatas, Name: checkpoints/640/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/640/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3d121fedd5200300d083f7dfe29099810e4c0090b72e23a0de27548f829c895088c93293717b98b9546ba8f981f73b5456536fdeeee39deb60b9ab5c7b01114b5967966fdb1789cc998decd6e06872d7bceb3358f2d30a3741733ffc4e2b0515034867f4728b55f2bce05ff0ef858dc2506ea2469ae80cff869e07650d52962d7c8a92ec0519fb1b76e2885f81a52704246ea6d779e4c09700a8f5344eadbc3dfee7824b303e0668b6e4435f909f41291021e7336cad2d6064c6f5ac0c1151b25f8e34f61a31bfa43ba8429fe5d6947d6025ca9c7f3249cc34b01971ada96d906abec56ea23536c589464a20c4667a7b9ca4eddc55be16877d567f2c5a250c4d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32824404Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0, Name: checkpoints/2752/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2752/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=632587c802ac1f7c4c1a9fa392332db0fe405202004d7814d98018e377e19acd5bc3380c1e0533547713224b9728f23ffe2fbcd19a41d1765a7c1fc0870b7e1985e045d03aef1f481f1e0fbd38e5f580703097aa6816007144f20f4e18d0e2b6e54913ba4eb3cb46a346d15c4ea8ca9705479bb09e364d8d7932cfc80642d5555e2c050410da4e8beb6afe2eb17532d133946335c07d34de550c8ff4fe94ee216b7463ee1b7f8d71440e819dfb5d31774f056cc5f7871f72b6633b5603c2d2ba39ebfbe8512c8093e35a8585d96f864400c853a4ea6a6b3be1521a8c961f5ac2b0886b35ff1eaa0a9f1bc72c934088eb35ffd9a08497ec9b6a4f9cdd03b62091&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324841152Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/array_metadatas, Name: checkpoints/5056/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5056/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=227eeac17e81dbb11cdaec7f2ed747def41e9fa46b6213029382525511f33ef7293416a6999f2710988120bba2fdf1bae60ba37ac1f7bcdc9b8887e21e61e4ad836071e732ebde38363f9a18def0ecac5f3f9dcd16911a31aad39219b09659f8d85f28366f8d3bf5314f1eb90689f3852376c373ec6adf0750eb3687077dd69a462b021af6c32a1fce828ae8a2299196b0462e2b2f5ec3572db12d0b2b35b766a59e7d3f34da05e9c509a0aee14acd39141044d9ef1d60c374dc8499cae6c307b057d84789798fd82315e01675a926014b80d5ed427b291ffb11dd5815c2da606000a22bc62efe22715976e13558af09194d2bea348a3d72793dc44a12c94e2d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324954506Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/array_metadatas, Name: checkpoints/1792/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1792/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9db66ea88f2ae0270395288ed5a0644496a035fb1fbac833d7d814022eafba6521b56a1afb62fffd7ad4e9548c3afe635a6bcd0ca06f8a20b26204f93ba1152128983d7e50250c4970c07246fec6fb240aaed79ddb75d4f5217afbd5ab1b3f5b0bc325a0a1c836cf6f2fbe387e5fc2a91f8b455b4fe26022cfc337d8f4f17eafe0b6f4dea142bf45f39ae22333ef20e97cd9b1fbcbfa7ddb13203ecef2d9e04c203ecdf2fc485be2a12b1e5a92eedf79331b7e6d78016bcf8e5bd013bc3144752d57fa5e414405fa56dfae4444a0e42c1e3f46f7f0836cf649d8555fb29de80c81512cf7021a5cde18159c54e23b696117f525120ad9eab247065376dc804f1e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328733245Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816, Name: checkpoints/2816, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2816?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1e49a10806de53bee3077bf11b67076dfd183f3b8e5944302adec8771167ecb888c1899cece2304e6bf454428f7775905baaf96a55d3455f8373bf4b22fee244bb62b84f2b91054dcd8584c807603f9db7a2e4adb48269d326336b1c4384c9ccf90bcbe9483b94bca3b52df5f34a6a6969217ac76a5742b6a3b0868cccc5eb6ade09742e49d4124df1bd51276526177729596a5bc71b9f78e49dc65c257850674da47cec19e3512cb1e0ff5cc632ac670b25260293877d6967f0247a2786dd2ff1e7bcbb9935f4f4de96db0f3d0214aa0096ff70717f65c83ce53c8c1a94f6acfd18caade3f7b3070be97e419f349d1eb6bf81949a8d65a79d8acf292a5f3716&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326920764Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d, Name: checkpoints/1984/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1984/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=125270df3f57184965e3a7ebe37923116526f9513062bc806b93ee468d2baca7bb7f36de90f1ac82ad08b786e26d67fed423377fd58a30b58e13396f314ac78bfce2942e1ccb9c5251e966e4ccdf1aa5b7a93b919bbe842607784eb8b97096380a1e03a22289d43f3af662764994d1506fdf68bbc49dca4dc446976b2cf069a4dec95e7c956c427c55b9cd3f168f3cddba50f4b99e524e997614a6195d8140c2bb10d94e2b5b62c78450ea36deb52868e274f2ee02228edb64bf0554276c2b407396fa062891655550587a7adbfa678730169908d14849286af7a063cc16d1cd0e9c0d35e814a3551df87aaabb4ff6af3065637b5f7ad9ce0fc4c8b6bfd760e6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328746087Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576, Name: checkpoints/576, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/576?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=45e8253959c3e4d574d5b61290751488cdf5409e449c66837d5feb10281140947544bca5be971366de84f777d6a12d7673137a7dd747aca1579a1d9765a9971eef289a713551e78127a0c86093efab1f4686963e8ee17442d324d756be899dfde873bc4050bcf22f59bab873b568c0c47eff6a80e7d456b8687e9ebae1b9c346aa911b3f80decae1e1e0885abc317587ff606aea62b8972e0057250da4634ac63494ad3217a9f7833d35211bac3e010bf597dc3b2a4ecf1d116e6c4f7fafcadaec565cb2c98076f1dbd5d947e5017f6547d95beef1ea88cf16ca064007d2068d500b1c0344e52c7336f42e806286faed9ead7bb875ef2429d29fc46392f97b4c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32876813Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472, Name: checkpoints/1472, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1472?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3fe1bd7375724bee6577d2e562089306249e7387cea60a4296926f3507936f55698e08ac84c2d66820222a4ce8ddc9c1fc69d36887b24878d94db246bac66924b3102b823bea0aa0b71e88dad51ddd43ba871eae26d89e8daf0553d277ddf148008fc9654287f2a46420f4f1f4f1b5faded132f9c0c98613ee75bb516cb5192913837418fa2e124bf62c0e6dafe4f9d1722d9bf52662dd1e066acb9e740c9ff13505718998c8b87dcfc96f43261686a29f444e482ad00e5183081c17ef3d2fa5a8c637964036d3f9b4900ec62af5ec5c49145dd4aa0b931d525f9248a55a4f6f1f2e7102672474169d6d2451f4fbca04d2f9670f1798a2d880d305fc4e0b23de&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328799844Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default, Name: checkpoints/4608/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4608/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=77852c3594b43bce5a0ca39b0a7e8b69f30c4655e717c74fd32e1c441906e5f92c585038d84d00c1ae789efeeeb386511b115d6147a4279ff4cf7b6166643158060dfb2a84bad7d38764a1561854238c8db9750021f27bb73274c0e2f90da5f23d6b863cfdc0c53e3da5b86c6cc17e8329aeda48f533338d031cf4f7abf8020c265409f2e9ff099f333a8821f3338ba4ace14f69850f90a18c76be9590283e7ae349fe920100f9125dda2341d05ae9756f2efce8037ca966a12c73fd9d460eb36132edc269699bfbcf5a7e6c2492066bc7f5c2508f50ffc5803cc7c81823fa963ace7daa1846a07223ef05db1b2da7fd8c773aa687cf883b89e06a2566162ecb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328871862Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0, Name: checkpoints/5632/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5632/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5e1a8f16c17c107650a86b45376ea6e89cfd5e1bbed4e46ae9a5b121a3ab899f25d2b86eda25d35582bf266cfa537f8b0ae4c24107c3d71ac88468e4c474bac3fa72cc5c8683261d3d393e228b162b81451361fedbf6aa5d63de46250ec636abdff080aad72beacf5299c0a9e3aa5e05896288da12e6f6f2373c0d2a53c23fc6991a14e8901067adda84d1b400af7d6bebbbccb19b23d5685ae3985af50100d1b55fc3fa4154cdfd7e4fb31c56ef7cb7bbdde805e92b6eaf75449ebb6c2f0757be53b5f35cc6a19689b9e1f717204711f7574e409298724e37d88c6f29a3c3062434c3d3382a57f16cfd6f79faa94019c57f55c86c35ce99051594979f440554&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328920107Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d, Name: checkpoints/1024/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1024/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1bc2e23bcf0112d0694825cd9f39d187206d71675f8e87e1fe0cc86334f130563a726ebb63c3297649886b13ee1e0621da5e950bb15a695a9ed282664a657b0f01734b76b891a5844620f6dd095f9b429687136655bee38ea9226dfd530c9db0f9303c7b74fe84a3037902a0eab103cda65be6c2bcc878fffefcd13fc9882e47df5021de259b8d8575bd73e337995d36a0d6325687ad728017e1fb765e1bff42ba2c78e32b1d59013939985c30140b40615c64117f6b14cad0bc384f12a8001201b1c0636df7bd86e2794470d8f2a0e9ab5c0da8a3eab1a448553c0b70a882d9ae9fe6a2141ca2063ce7ff94466730425edf82e61fdbabd32acbd3fd899430a0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328330072Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648, Name: checkpoints/3648, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3648?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7b42f1962a86ae5dd1b67b0d7ab475ea837e8ef76e124c34ebb35dde1236467563c7f685f68675fafbabe8c367f8364bffabbf410812c1e6b32bbaa41039791ad61d9a7bb601eb6d765afb203fe1cb8bc2f6d642fd021f5342da6a5e9fbcd726f11a50687185086fcbb8e68ad133864afd37e3f2ee2694354d7e9f5c46e2d16eda8d8b1c2b6e92cedee4aa72d218c86ccce578f740fa0eecfd21c4e1ee6aaf92032b37f608816745f82defa1c7b259a722a753bab4c8eb0607d5bafeca5b2e7a4a157c9efed618af071f5b306e191da17cf0fb7e667c51541c78e94751c4b9155086d6e90af37ec4d58a77da7b62e379c11a27f19182f9e4462c052ae62a7d30&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.325087387Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712, Name: checkpoints/3712, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3712?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2100b80ac44e8fafc7e344535c1af6b08ce25322383a37bae97b1aae2b9a400b7543f0bd62b50d881d222466e6ed71a7980e2ebd9449be14deced7b7f18a8a5058117468b4c1eeeef70c06df031fdbaccb4418559c48c724535d2f833a350d7390280b762a9fa1685d489c0afeacbe560341f0774d6b74890cf8ffcf87276971d3987b1e936e6a341ed3161d09a77d676dcac08775a847d1388fd48b6021231ce967bc9a6c519438d43dcd6d6f017bce4d7e93cf6967a74c3975d9b7e1f5e1d0cacc90c8f1bac9e2d8c0f085ffc7b27d9dc42138872bcd76ef9e11965f47c8678bba7292de5acd1670d4af02860d61e55dc79c8a7cd74f2ce0ae62c7c70a708d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328970223Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456, Name: checkpoints/3456, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3456?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8bcacb09aa498ac5992a9ea1248fab51c31e22bbd905aeae78fa11e28862b917b4bdc5f3d0619bd7a8cc6a8fbb0e1424436faef8bf61d9e66d4597bfdb6ec48104287f1dcb6978339e5eabbad4e26ccd2857378d9ab679e36f241f806450b30709bb4b541a2165ae8f02b51c8bf75b1a478cfa281c9858ecd4beec96b5625ab793dfcbf851cc8c1d0e9cadf6d78698ecf7e8dee92ecf88924dcd579eb2bddaa6a035510183dd5953c301f58148336aeb8fb0f0abc86de2476234b7bba4f3bdfd9243bee891ecc9e704cd9385bc8c0270bf104cfc7595195b005cd096552b69973b8284c8878e5f66eb34571852cfaa901cbf379a5ee8cdb33ad2636eac359067&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32876716Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d, Name: checkpoints/3008/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3008/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=870f3b1e46771b62d6b1273f975fca46b5908a9a6a4b4348faec2043b3930812d663b958150d707fc189fb58eab8e6239b78eb544c10d65782a58b8cf2a15982dec8c0b84f90fc8c8eb982d59fe007db394b98160c5239686cf90cee906ef182f98b7c210474b287c0d03dd9399387ff235cb7e1f1271eb5ea2effe0db8ce205c8c81dfa08d37af80b2c144cb561a182eee24aa9f505190046e910a05bd994b982f25c7bd6e39eb864dbbae5c3145560861f7ef7f5b4a19d90ca17f1ade07d73e8aa3ef698b774e1362ddfc2f4ecf9ca70c720e7663344c99400663a0b74e0e8f8da80028714155155a2d998f1334448b8f93170e584e29b01e2cb1549d6e1b5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32902559Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/array_metadatas, Name: checkpoints/4096/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4096/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7c711dff97765f22836f9e7e42192df98f2fdc48aafb8a020b36265fa39a16647ff70d9621d142ac5d703ec0832e85b467286f53717888ddc8dfb1d3c9c1fd9b8a18020d1afc321998d002c1a9962123b769eff8129aaf981707491bb4a3a266fe798563947b690e2d9027deedbfacbb53ecd02857e32e8f3407576fc5a06bfe886878f44221f27a26f2fc6b05828534a320f125b9f98fcb5c002ddf3d68cd562c4cc31068b9a09270455e6d2cd12e41669ef4908720cc61e9f68d6b637a844c97d24ae562cbace276e0021f76d381b57d99f17a82838e0af2a40d22ada40ed3e5c94e9decf26268bb636686641db685c6d70724ba87edcc5a08f4fdc8e5d7cc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328893375Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0, Name: checkpoints/6016/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6016/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=137e661ee2bd39a2cdb29f22ea7c54655554376491c901e5f6e39c852f9afe9b678c556f7837557ef2667ff5fb9728353f3b3d263985a1763a719925584aff53e2175374d7dbaae4993a8415496db6ffe09ec71972063e175c7d42574ea0a472e91141df71249cfc88335d5505e217d78d617ea1b7a8a826da2faea4b9e4b5efe4ae5d8d31e8d3deb073b1abda7cd3c36879618e5fc98f0079b84f78d6966ebb02277778303dabb484485527dab2580279679150988ea3333759fba62e7b20406071ab2d7e0d10431ea230095eda06ef285cdad7bfe527c3f85d66d652df2bb97071640b2119640ef180669da92f5f8d748221bd8be6773150447f5dfd423c91&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324712245Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/d, Name: checkpoints/5120/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5120/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7221dd7fb98f9aa9aa790fc585b0cb5d5d0ef8def6de0c794aa0787f81bd36c893f87c469b35905c2c92dc44f3cafd401dae9edd51694647c64ba1b1ea9fc91907dc4cd6f95f728d23bd043dabd97666669a276badc90fc9175153840e9937342fe8e3f8f38f5bdaccf638d43ae51a79544fbdaecc3c883e1cbffe133b78d9c725d104e64d33f11d2f5e3a1fc10912e95dbfdd98d397438fbcbb9d5cb40aca780a998cd67b8c46165ed96785b10591a73ba6973a91613c39d421de4c6a1350910525bb39a8a29066eaf95d097158cb15f9a0a3fc0d6cecfffa678d1987c25c8b9bbd3014e37258d615eded0778ffa0d2700332230d26644ade3033b2703b9ba9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.329054719Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0, Name: checkpoints/5696/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5696/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3166e074ae0517f1ba563a7b13118d2ed8dd0a8d3b8157abd3515ab8b499776d37a34ee344ba9ddba07847d7db5a821577aa97a3d9beea54a031b9a840a2e118143a9e9e35727766b46d0c3ef99438fa870a2faf73e1122dce04683667ab519c9eb6ab903be1bb16694fc835b0f1cb3fe47f177e6bdae825ed2ce0df1c38942a5653cf4531efac90d749617e989dccc2fcb39d77429feebb345e5dbef3686c651e0d8f45b6b66fff1e5f8cee9bde1e17465840b4b15b7726bbe53dd0b1de246189a6b64b39d33b7af6a85b53446b1dc89d3a317f7f4a244432bf8ecd64009008c07159d9c7094317b18e533ecc5952c2a19a3ac219d90ec0abf0961de626d52e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326228443Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/array_metadatas, Name: checkpoints/2560/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2560/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=09b4fdf3f1b91fe4e8eeba4948e70064c8c825cfa97a36a2a3c8601557a41af4c2847a297b73ad1e780269eb2c04a4ff1cede51817de2673f0782ba31ad436ca2a6d5ccd985c39593d31073f1122cf8bd0cc109e0f06f8bc39bbf3e5f4739c402ee62a49828a9dd4c720ae7c855d72d99882cc7879c13126dfc8c92ebe81fbed92431bc4122ee94ab211e267decf2bf24153265f8bca187b1e8bb79176cf9b4987894e4439dffcd61d6427b6f2cafdbd86e1d6c76644285bc19898258700e37a2e39087f21ab964660fb094e0a545a4c3bfde9254962c4cae127181661e7e335409685cae99c31647e14be521272a4498a9ec9cd139060860cfc0b316017ef93&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326242406Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/d, Name: checkpoints/2432/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2432/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9e89268b7bcb22cb338bc470482e1e513f646f911b8243441cd0a4589ef01f6a055e50fc269bab6c7d8e41f2e118141f39b310331c08e89d7915c94c78976a2b94e722db85e436dd1cf0232d9f2595be72338a7b7c0e72723fa0340675a8b25489416c4760d606b22a2ca89e105468b5530c013c45fb2f29451815961feecf4a8bbf33527b859aabf7fb1e7810cfe0efeb3b37ed7f86fae7c94c18f9c0b424bfbeec0d8b3c6f2c8a9098c47f7dd418f6f90e1bc9022b5d68302315260a394c5871b04a4cebf5974acbb865442c8c8458237593ab15bbb96703788583a55ea6afc16058b4eae5bef12aeb9714973d0b611dcb05ddab8432ddec6641b8e1a9cd72&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.325099046Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/d, Name: checkpoints/5184/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5184/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=771a8415a8cfb148101c2ea96dc43bf86b6fb3ebb5747b3efd089e1cfe855efb000a28c42ee640027a963c7633938dbb70558931cb111b12f1a7020e8bcf1e714cc03414f3952cdc27a8ba8c2dc128bedd9d4a649ca29b7791642b27f89d03bf3b16329f2460d75875ee2f8b04e3355c51dc2c9e3e51d5ed699108bede2bc1e73412b310c309bdb2e3abaec34cfb0d2e5e217c32e2cd416410423727462e5587b7f9c4d71b70b92bcb8934fe44b645e57febae4287ffb2e5195a8fa1e4b13354e1793e6a4b5fe09652bb9247c81daa2fb8ac437f011b74dfad1d3a3af8bd5acc1bdba9ee4e850ddd67dcdad37c4e70a432a02fa0effd7e612ebd0899e9c30710&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324710135Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752, Name: checkpoints/2752, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2752?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7a20430b766ab01075d3894e211c93f3bd70e41f17c8250f0ff85e98de78ca3655289d8de0ed8500a692c1787392f124f4afe824fd83127e4a3265f006c726837ac8cce57a6c869e849c631d8dd7cd481a92904b01a1c77d4eff35e4f71d0a7f6534e9fd4a7d248502903e5a27cdbf5c68c946405ca445a41526968ed4a19f490ae6c8c521145c2c0e6ffc90a597a31c30eb7c2232a92d5c387b531bbf6ede6ba33dd80fc053b5927d3873421659c3b44cd0d2dcabd6ecbfc707d19bb056afc7aa359ad47d6ab15fcc7094375a466d1d7344857937b1b82cca6357008832a9ee30d49a0cd007da0841fe71aa9669dc9b798190d05c7a2f620654d06f4edc65ae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326217104Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/d, Name: checkpoints/1472/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1472/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0858f096674ed609f095af82a550cb30a14ca4d83f6c5986701cbb39955d61cd54ff92a2c9a6ada731ac134d44eef2035ee1ce3ac8a85f6f6f0f4dd0e44bef23bc3ed5c97bd99ef77fedfad789583a0c624bfaa20d850ee1516644ea27bfb816d2e094862950f22d913c7380fd6d6ec2194a409f4cd4abc8f71656a9f7206a8a04210a990bf1d1066a67d2ec4adbe0d83c48b85add97227f6fefc1efa896ce9c3778cc3cc4b5e30f0c2dc34950829d2e76fbf666b8f1729813e1496d892c39d0e5e764883849f5ba73ca3a038b44d9bfa783381022f893de0c19524adbabaad83f44778b9e82012501f92d9dbd6a8a23fcd9f40a47bbeb30c7e6ba079db8577d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326109482Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d, Name: checkpoints/960/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/960/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c1eb26730e095ed822c2936054e08224f361b4d875e4779e8202ad5c64f8c061f368fd6f26842af572b3f6d8baefd4c4a8fa4b8b72f44bb5de9ce4ea197d869d1885e343ff777f09525f6588c61fa9f54e851d6340afa3de056cbe961099686abe07a9de383dfe317a72cfec281e6c2961532222c515f1f11404e066b35ffea746b5c2bc177c52842fb385c17a14ad5c056517c19bd870180535bd1085a30376fc52f36d98a39204897cf9ae309c6594365e1d5c1f2bdf0747be0fa09aecbe70e8bd4a94d5aec33a7647bdeb8071a4063f195a0b5eea1a4e1abd2e5931dbbde57bac41b176f4e864bf40ff50e4b8e4a92b57c5c6c8e8e2a7a36ef42691be0fea&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326242183Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/d, Name: checkpoints/3264/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3264/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1b2a0ccb957633512223bd60ada691c85cedf2bbdb73c9835233d2f50bb6ec356e0d6f6b68117c2eb1978b601e5dc88f7d467c1d5ef357ea50e68477b2a2bb4f50666077ca7884863808401e038b8097db207dec6dd44e81ae9aa270d4c3e6ddf008036f8faf63fb63b4c181985f5d48f4d3cb7cda232d7b91091138551b8050357c05cb3b37724a6da7b871857d4fc15d3f65ba5ccfed98951127614ed56c741d2e25551cf7b56561abd2a9c934b707802e428a1222f68dcbf34020df1023596f8252f2b672cb35b484522116396a330ee5d8401f5d35bb3e4a1b54f5eb3c120363c20eed1151a06e450a658fcf4d2c80737dd970a7e9b22733ef84e9d0e175&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32499398Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/d, Name: checkpoints/6080/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6080/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=58262a302619b98f9aa0c4fdadb0d54de90abf7f612177ef4f3ceff82f1307b0904f6b114f1e4dd36e3f3d05a189e0b94b332c983ebcf3fa8cca426dfc5e45100cc29947aadd1cba6ee53d7bfdf3147bcbda06c41361516292682926ceb90e19e9e1cd76ef44a3e99732bf888b694128907f7d42fed7d6fc3afbcbc3b50aaeecb6209c9c5442a5c1f5d7654b0ea7b0837c40d7b07ed3b85361249db180ce51ca1b9ed899376affa213399bd048b4bebb98647d528fcbb01b320cd5cb761ab9bdf4ef1a7bc0ffa625a91cdc24beff36f81179ded4e25b19edeb2707b45221ff6a6c9b99555e11458e45920ce5de3ece3bae78f1ed8c841d740590452ee87064f8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32868801Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default, Name: checkpoints/1152/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1152/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=29b313f24456ed04e7889ed9a1a804a6aa97060b6747f429b71affb05f950fcaeb64d7569f09125e20a7b2337e5285c7d10d5e99e5a0205290bd38e2c115ddb1833e9d60efd0cfe442b505c885545b46f583823ce9d25327817502141e2215323c88ab626a40124cac71910ee742831bbc586eaa02bd445d099d141aa973acf9ae6b71c9ab47fd7275ea29e6daa1db455188b13269d20c3e4b5506090d2005f3d01b43a3a0da4f92f0896dd52a4e3fb87ca5758bd2ea9a39a6a518b31e0c8626bd5e20c54ba0f40cb290e955f1c7619b50f634100ad538048d3bb98ad96c0d44b3bfa812d4d7c0a526f6d8aaf8175d6debe581e84f3524c4efaf0826758af4f2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328700222Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d, Name: checkpoints/1472/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1472/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a8832f6f19f7b24f6e11e5add58d7fdc11c935950cfa154be7e110d92ccd14e7969c327c39748e0e83c8b4f7163942468089a4dc18f992239c9a2ea6447e5f34deac5d1dde426f9873c98e812bdf8b83b3c9cbdfc5a09a66b7c340dc97ea54bbc0fd4632d0dd2d2a1e03ad8631046c8f82d3af5bac7d8b05db132282ed34ace00c57ce2a53a7837528827ce06ae583a47854b6de04795b7b571836774a8bc9018f8733e809554f46b20ce496cb7cabf9dd808bc1246341145bef2ce3ca21f8553ee6920e3b2612cc9545f8a0b237319c58e98acaf2216f2d5d2b7c9a0eeff591d4e7bd86e721d4de7884f06bf1b166c1af90c2d929c36baff9197e414dafa1a1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328768835Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/d, Name: checkpoints/512/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/512/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3b24d5e40494d6f7f6e77a18acfb9c7925800976ec27eb014c501c61ba31e117381e93e09f042a5d0b1776e404b19e7c000e6f460114e3f8ebea199e1463ce65e468bf9a0702c252b53086c323331ac85c35726b3feb6fa75eadfd427a8d968c7518d7a8891084c110a780d7d0c1e6651c0727bcd7f2f0dfff5889d2cd0fee143d8ef5da50b5a56f9c7840b0ab10cc5bd645819cb8ab5f939553e3c9b9207eb48d42fb7d397456f857bc46796cb020e1eaa5cc4cc1d4181ea05672d4b3960c119159862706aa6804aec593216d81cc8b12e6d6bd7ddd1b46e582145777f19f98c42b49aaa260d4588cca7c404fcea1c6b0b541a12ea839126d7f9e50f79d549b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.329499644Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800, Name: checkpoints/4800, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4800?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9e4deedac55b1ce7754d1ce9192fb22ae1d5299c3c7470e664754707a0b8cb82643dc76333ae4d62f754abcf4bfa564c8d97225eac19e0f4b589d8cb4b2c4c7058d8e686531c97c9d64cf80623077e2578fead7c75b6df52223775079ff63677ac1d1baaa64816e736dc5ea6aa162615c5321ef640e94d2eb25e5179c72f4ec5ae7c99f447d45c85906b829986cc08d1a6ec8d4762e732087fd4602484a5275d749612f408246b356da3e30719950e1fc5684ce3263a9e3bd24ff8f9bc75d10ef3f8d92ed49a13f97db101de33555a030516508ba165b55eae04f639f283b75ee3e9a540c12920556a3b958976950e6b194b30694b7e817ce39dc4101e83f5a7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328708791Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default, Name: checkpoints/3200/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3200/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=40a746f5d7f783656299ae760ffdbf9c2af9e88852723ad0ed66d17e8833a374366bdb88a2edd977afa29ba27408922734fe2b80756f07ee724cd8ec1758e60d9e647ac184040bacf75e619167b6b791b4fc449371709df1b0064ae5af5a84586af41ce56348901393de3a29c4c6bfa0655f6e0f78d959b0326102953ee2414a61d1f126634309c0da304b1d671722a88af1a898d53f9c70b496d3ca4094b3191e63f602a62b2900c0f8316da19368e7923193487353c7cedb6c4265dc9500df9ebc8ecc94a4d479e773d546d300e3ceb516c65c5d7126cd0734972d5f7ccd0a74e4e53cc7a15a6334c4aa9fb989c4b2f50f56dd87389df381916a1b84328eb2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32898167Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d, Name: checkpoints/3904/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3904/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7e2d9be553f29f354648b1a6ad35a8f1f4e329108427b53b4ffdd7e49acd5923db102fb2663fcba94785431016ef359e694929d5c20f78b29384d50d5ff6d1bbe62ee6953b4b4c548eb18b09a2947a538639b80bf60604ef4df3a22c3fb17e8834c0f31970ff02fb96a0abfd56af0bad6ccb923e3ea1b0b573de5556f22b507b96617f66d3f0c3cb183e74f8edb6db88518e9892476ea99e9f93316921e75c55fcfe6b1e886cd191d83c8baa6e1a48e8b150f773bbdb0bcee75e7af64f3ec460b306e02f33f9c992eb89833008187b39f17aa3d0d3c456e019c4c38f50633c2ba48c89f50334d54e0701c1f2c97e001dacb7811176539801f203fb0164ea6944&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32896746Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832, Name: checkpoints/832, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/832?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=69c3a6e7a57ad2ee0ebb7ae2b3c05b2b1bc5a21118cac5d2bdfbbc174071d9e964c09ca2be9c423f28026c4e59fa8f4942a8cb587b7484271de873c0154d7369b298856b7742e437675e50136b03ad34fd5ec94dac0eb7ad32a0a0a3dbe455c9a388b2f56bdb2c4c1e1fac20f8ec0b681c95f1ab2588d0f6246e46bc690b19c28f9dda7ef09ae2d0e9c15c96382b186381e2ed8348fbe900d84067fb6f8ef10e9119dad5af0bd6281bd89f2c1761cb50540f20f4d230272902e64107630911b073798216ee06039a085fdd0360442f635ec42da84e4fbb69597c28181dfdadd15c5b07ab6128b888f3751dc536cf01e5d3040c248e898a37cb9bc6ec16dd540f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328925541Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default, Name: checkpoints/3008/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3008/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b68c10865830fef3a5b9c16c77502e1a2e57e03af246973da079fabc04cdac1fdd788791a458a040c049db9360ae5d1b3c817be7302a02d85a11200c7fa0270c8c2c44b3bfd589562082151ecbd3bc5ae13053673bf40452731b75a82dd4b8059c8b2ff67b68302187da19ece37f29a521790b58c5288aad01e72eed8713d438cc36357b7cd17b5965d4e4b744763c76d184fb978824cfec0dd0b206b95a01e9bdf4f8463fbf6004c7bbc10db0b72177ad3c11ee50a5278fddc795898c84de15237c19f207785e2a9cb201c06c83b935f81bf770dec71238c12e46803cbe70b37cb721f8356c822b3506c7e22cdb082626f726d644835f37ff3cdcf01734158a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328971579Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/array_metadatas, Name: checkpoints/5760/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5760/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5a5d7dc8b89c4b456e447cd0ab7c770107e588c522427034ffb38a31ded6bd99ca19647496453172fe1893142ae0482380854c8fb28ae12d0ca948a94ebe989c409104c447694fc7a178429f438031ccda097f7d1ec5dac6c8bc317aa60d8c0a011c4c1ef1213aaba6f8893cacd4e0dd5a8767dc08c50dd52c861fc6de58ed0f70b6b13e63b06990f4a8aaefabaee4914b796b5bf8ffc0900f0c95d8d1bb6571f7d671a5c9728e842dfe33624b8bd2bfc10a8ce5b567227934eec1d91aae394ea563c4884277ec2fb3a13ca6ff69edd3fce1c592514dd39b55ff83515104c848102ceb6af9daab1e61fec78d9278efdba307f6ebd3b9896efb291ca0c915a6d0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.3290257Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d, Name: checkpoints/5568/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5568/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a7ec784d67a15c9ffce2d80efd00f4235f84db9772308eef6c08c3b5af72a15e455aece3ac5d31ad79b404e032b12c7d1e13575a7f9f06f8086f8b12e510982579db1d28925768414539a57c623da7049c9456185a80891b7b5732a91b9cd0bae58699f2bc70791c80c07f0e268e4e4bb345bcbf4d9d1b9db9fe3fec9182abb6961a0618b56c41aba123cf2b82b094eee79ecd84e3e5a5c26f9472e4611d4d3da394b17de847957bdd4b57616e7fbc84034ff8ac2bd8c8aa9ec88d450ef2b5f1942d7bc11a08e5f47b3751644621168aa19d8c7ff27a6c368772ff19fca20b447e1adbc158768e5c6af9896f6c20b24aa3614dda6a96c9ce0979fde0789a5a49&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.329189988Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944, Name: checkpoints/2944, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2944?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8f5899870d161931414300d5c3de4e9d274bcd429e07343a8614f8ae784dd6588ea87f7ae4bd24d35defef70cd12d5da8364e2d5c113fd9c210608d5b59b360584371e67f7ee47f61f18d9903beba1bc1c3cdb7319e7564af7a1c955516f75bb9d6564d331be05a0496177c3b8ad11d1d46b842daf48635580b785b7a1b10b79b78b11da5d42174101107a32fdbd2a845173498167113615f031ab534715897d1794df8da6128ae9f16b7a9452a44f281aa149aad5c879a0fd03e5706c022277e431a1e446e6c94c7cb267f37c617f7ed07d2afd93a38835963947df58ae6700e5bb126bd505aa8748d506a6132e64076adc96f02dd37bbd90352be418328a50&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331540473Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432, Name: checkpoints/2432, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2432?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=07bd6e09689e1c56e51fce18563084a43579b4735180d5763067b7082271a089a57e37c8158cc0c875375e16b4a09f0360aa025890d65392d496fc52361a6c458bc36d0a66098572c69eb56be8ebcd911800e8bed5dd773bb37e640c7c04b8fd58fa924b0d5df83b18efd8c4feb1ccf3370b3766dcd586ad2f2408700cd85100f6d39d3e6a9896f8e861076540c7945766d02ecb24792501ae0b3f0486efe28242d377342bedcc2bf75e19faa849d7fa48e6bef8dcd39b226c546656be239fff4a361433774e5774d3e4aa0d90f701be6386fd1e0384108808fb390cd8f3e240127d1fd1c6f78037327e9aa99acb8b5ba4e3913891cb6e371720f817c5be58f5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331580099Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/array_metadatas, Name: checkpoints/2048/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2048/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=597ab9011095a667c6207ffbaed68a8ab7c4d3cb5ad310bb1719b25475f3c0e84a1811f976dd0c8c3330af32769be3916b4dd30a8132f87bf891fcd1dfdc4fdd97a7a46445218474717643247e4dab4822474a43718886f92c64fd5012358916d5d82e844a1ffcadcf176f9118f29ce3e009aca30220834057f7957def8d3c9321fb0a585ab3707ca596e208261be796438afe82d3c9df83ffbaa3a333baa777240b2b6f5389a984bb8f76f271c2e49f35e0d2b37838defff9ae2a7907e5cfec1496b73e8e9e14369ad4c67c999f495511d3774762e7e2acd0e6411bce182d67dded87a25d7820048be24a566a083a768300ad8f6c4afc4db8d954e69599ac8f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331511094Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/d, Name: checkpoints/5504/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5504/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=210584c5765f905b1ca46a285b4583f064f2688d7db74a1b660836748b0fecf8f9a8f3f80c9436b25cf9a9e817c47f0ef24b2c4d84c659d1082c2e064946f43886d55b59fd4fd6d11edc97556203ba0d3037ea6e1c6325c9ec975e44f2ca36f9b249929467b9ec8772e3ee60c30e2a72ec0caea4a832c1813a82d4ae487917ebad9305b28636e5a0502ab5bfb2c1d34631290508a5872c341af75b5806ca6417d497181bde01514f2c61639939576be5e0cf8e728b0cdb96ce425f92e3dcf183f564ce636d5f6c63bc60b0ee2bdda9d1619244fb372bdf81cbb96e3a823a76ee4702379a8b3b31ed657877b34f844fd736a038154f2fd1460974d23bf81de033&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331381187Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/d, Name: checkpoints/1280/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1280/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b3bc09223d1afd43a928216bd1a3acf341ca0180eea6254027e99d3a480159b4e3303cff3baecbeeee451300d600fbef919e3a87752444ad6fd5487ddc8938d0fed5019d088d7ca80aa77ec0e16292bea98c36b1ac8c751681293ebe0057faf7d1461e4f8fd4ee6d57d369de6bb90a48ae6360afd16357ef516929b6bbf44119edf832e6a07ce374cd8a98a89dff48f5bf773a7f13a12b3ff0b90645b99238ea8810717971f4c0f60b27fde1e6bb5e52f42b7f01c60b98df4add05f17d0fd7b29f3dc432994965efbdce7e0fbe7f113a294638c070cc5ad1ca64c92547d6ae208a6ae57475ad11f50ac41e6a46db9e0dc2bd28be3ed5c0e7922e7989b291d2d4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331624501Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/array_metadatas, Name: checkpoints/896/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/896/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=35453be86cc0d9e2b835ce03a5e1a6925669ab592dbc0d1cdc53f90206f36f1d605569a0dfafff0c291b7850f0ac9d6cb35336bf1fa41600f80f9f5ccc9f9b0f4f2e706cf2022237b06e196a17510801cf53f1b5cc558e51e6d1fd96dc2467f5b6efbaff344d21ce9684b5447457b562e0ae444eacd2f0f5c47f6a398aeb5af2c8000992adb49cd8e4672d418808f570fc67cbc38f1bac20203cf7b069857189464281c19ca8452064102a6efae325a05931540a2f9bbd480fedd33fd64dce73803a3b114408cac295820262ffd11b53d23a294f9b5b643f7bd8312ba9fa5b6d90440e3f621a929d0b96f5db3d3a2bd46c4860108614ecbbb05ecbb09e5bee10&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.33163045Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default, Name: checkpoints/3584/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3584/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=87443a955bf9602ad7a110c653e15d847fb01cbf7a85c19225faa1fa04a607490844798d3461df1e594a65bb630333ef7e4040c1fe9d13bda3f40e537efa3efbe7c957323273a0684d402062c5e83bf38b240a0176b1e821c30b7846ae2a02608fc528594377a366ffd72c1371b847d63a265cee2cb45c899b1512dc235de3b3a79b81ce4dfac4fd0c0bc19bb88ee1dd76e6d7ade9bcbc6cc7151d05a3d1311d26d044aa4dd6d7b7d1ec5617b6af2e51900f2722a785c7cc62bfadd42b99d745a43d28ba2dd781738a15288c538ddee792498e69256963e9269aa9b41f23492a5b01702a444faab58c1fe4d35f87b19d6debea69f9f89312eb4d1cccbe1e7674&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331678168Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default, Name: checkpoints/4032/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4032/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=060495e7c5eef41db487d9e599bd0b445cce33b9ec006f08d08f3bd9ba64c90bcaefcd1cce8f3da8bce1556cacc3efdd7b431f2ca643cd8739b05082c814d7892e1534ca3be55b77e8108cc6a9e4e284a49a705dc203e2731b698c24a670057beb6dfa682e3af05e88eaa8a850e38be176eca560bb8b2c30d97a26354bd7941ee4524d770a153bf0d0ecd06035a63eb4b25ed6aec63a0e53abbf6aa6bee24620e80437a8a2443126ded95d71c6f67db4411a5a216cd92bb809ad78bc0a55304508e29f42535fed7d2d404f2362d6563264fe1ee6d2cbe141f1e2d20e34310b127585858f899bf1d9ca541f33335251284a14a98f652b92332e8aca5184297e92&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331689835Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d, Name: checkpoints/3648/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3648/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=904ee75e93991580b237d37ddf2d2fc3948107ce5e1848a4290b6fdb374661a812551152ddbb4732274e24702058144e0998697e164536ac7daf2539e4367936f4c046d6e9c3ce9b954f3bbbe5e122912551112e913fd58ccee446723232b47bf11f8a09103df07f35b2fb367ed9d8755c88ea9689765cba374c8f0892b1e0a0986b84af2a21cf59b149f91ebcecf8024cde494c0f5ed0c5e3dd62882d8f91e6199255c1993e5b612642561e5cba0e37f9cd933444621a8d72bd8295038da35f847ee6d97ae5f06dc7dca710bba7f442a1fde5fc5be4b40c26c5a0c771fc47ef2228e7914cd53120317c72213f86f7cd6fe8ad405e5c62d1271102dbca37c17c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331701028Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default, Name: checkpoints/4096/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4096/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6b0353bb70fcf89bf980b50bfbb03431dff29a2cea9c4856e194e66b8a7cb49ccf314630c00ed74cabc5c9e585fc2ea674c614d4c5f4982db4262538bbd625ce24df5971c0c86d34737cebc0dfce5f150f05cb9bf52a59c264cf9d60f011552acad18f1d4ec230ee4c9e7f77d4e6167ea96e7efc6d2eaefb653233b81afaa7a6d1f0567d077d7211586ce880b40b684fde17baa5f34b4fee84cb3ddc68a401284655fa194bdb2efbf6b049186ec09f9c62a0f5c0e098ac49764bfc2f395727e3a8525deb78b3281d504645785322bac73b69ce62874efd317ec9896c6ecd7668ea1a43c2e8f1e0db1e68942cd265ec1d3a7ee41bad469ab9a2a558f98acc1b14&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331810827Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/d, Name: checkpoints/2112/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2112/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2531a9d22af6a409871828d25ad524378b6943a17ed23e362b7a3a507fcf0c7f9dd14155e5290616594cbc24af772843f261bc3d32b6b90f425d82e579d2bac23b4e88cbb5bff030fe44b922646b16f462be5b0ec6db5873a274358a48f97574bcd6a65f19033b81168516242063c8c07e7db141f05a627bc7eba3ae1213c9c6f25ca19b75dda339ce1e6a6486b98ba2f387228180b7b8efe8c42db188505cc5227e05b3321f75462ed2db32881d422ce0a0c004d1cf5d2043507580a60404d7024376e4554780ea488e4c87a40b9fd5c663406c8b23f9f8949ed9e9ad18bbf517207b20626bd797b350f257ba91b4991fd6e535542e611f8c2212cf12c5aaac&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331833535Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/d, Name: checkpoints/1216/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1216/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7eebd1de5c1f7341928204e9789f3c7c95fb27cfa7b78742248cd1d2205a0badf88c944cc96626108b871bf85e9dec934d89028cb1f33d1604b0fadb56831ee46950ecabc033cdbd968a058d8dfa98c6460c1cc16e3fcdb796d5ba24fbea56d7644a66f9f7069ad96725a81f3fd3496cac9c8c47d5e61a1a66fbad5480cb8782811958a48522add7e3588d70781bf38b1590685a81cddd0f2e8c70be7ae48cc1d3f319037316fd55a22b12efaab1ce01a5c7070982e13097e2aa979778cb5d607b544e8139d471baf4df5c56ea0e70646b6b3f555ead9400b262304e25f3c29dc0cd6320b88e5a8de463cb8499498c7d5383d750ca44f9ef1ceab703c9b8e135&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.33181634Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d, Name: checkpoints/256/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/256/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=306462fc3048c4e96f85a11f374c7a51ea940813c0de9e09d751c7d31b8f8fa5ca433fd03ada09f4345b7a8bc235c25f40fd5849c8d5657c3960d78b4a98943205109e70b55689077cccd8d29def3cdcd50764ed84f0cd17a98b958c1f09fdec90a0ff666052e7f1b2bdbc927a20cd5e1f3e274ed2e5dc9c1fe9bdb8386edf4a3cd4b84e6bcdd8d12ed72c67870cd6dc4b4c09ed694f306bfbaa18da67941375a9011021822dcba387dfbc78e738ff44b79257259fc8fd5fb08a1aa5635cacc7b40ccde4da2f37ab87f5405b654baf42a37a2b242bdab60b4a5747163dc8a63364e823bc9e4423a0d69a01b1c83079a4301d31cb214f559f05fd358d451b3c3c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.33186299Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/array_metadatas, Name: checkpoints/2240/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2240/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=18ee1cb0b9f60908e3f65ca49e129a08ec7091e19ef6055681801c5d9e207c4badff3ff3cff8e6576b862eb4de625b51bc2ee40a36090c567c71bddfecda6935de772eecf5359241bc43685a5873c81c9296df7cad18a4c89390af274b6709921c7dbc63af57380911805d86d614fc365ccf851680eaae161f3aa4a5c57eb75efaf12fee0e232aaee05498fab4448e0af9f9d46800058ecb3715f37323b1da525c1901cadddaad63304508c8e885a42aa73428c20047fa76dba177ecf4402cbd81874484f60b0723bf4fbb3370071d11353a1704ed181c38e6fc2be1cc092be1b8436313c6351ba4568974603319ed9d685f22f4b88414fda267df085ddd1013&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.33189555Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/array_metadatas, Name: checkpoints/3264/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3264/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=39d64e4d759b1dc29d30846f1ba7038f6dd2719c8b453d844b70e0940320f046d37bc045a2e8847cfcec235a221c687c16111d31aed14c123b122635794c69dce2cc435bd9f81b3cb7cf902b16cb218f7a96a2c5384e6fe71a514464744aac87cef91fec9e4dc3f47fd6282d5b7473be61de780470f1986a75b7f9da2f1b721bd08523739ba01394312232f24a01c86d445b2dc54cd992b83d4a359ff2c8a51dbe028d97b2bc25471dc99119ccc19ad738ed421a58e5a0aefa9d3b0f0f532d55c4377679276b71fd2190dacfa94699ace1148c5a8df36d005a233f5efb4f494f5e0cc15155c45762aafad396223b87c14e8add21021f0c8d2055a5d00c98e177&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331904054Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0, Name: checkpoints/320/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/320/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0e575db05e5440cb108744bed5ee90c11572c4bba4432998af68ccf306437cf8587609ace641be2c5397df3acbf513b05cba5ea2b07b16c67b5e33123df47bd1c696ffb9ac6128e5a1ace3d58e933b31dca92dd3ef7402be16905448b63d7aaa68efc711f079f2305eeb7326a8fe8d0efb6e101984cb77129a76f9937612e47ceba027b4bd533355b6ce359ead8a9cc325021412f7e2d69d52b9cb0ee6a70ab86b4be589234f02d72cc94c2c412cb22c88dd16d0e630cd2c289044cc95aa9fe9b2cd62d64cd978bd524551b1c0a7645bce05174a00a21cb8a27b577872b1a8524b4dcd846ab36c3cd5ce5bc118bcff5a99ace9da50685ed8b16d2471c202d9b9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331914823Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0, Name: checkpoints/2816/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2816/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=466974256784f38c84707e3ecc0f89bde5e9e231588e6d9d627619cfd4827bbb4dddcc045d7f4a1b3d84728f2a3826b7adc505bf5a108cb20ea68cf5c51051f5745a1a88e41061f83f3b0d9df0ef217a29232e987561130a5d6451a152cdc8539f55bb9c655039038c46e2d8148242c15f7b2a7f4658b12cf4252802ffe2ba923bd1c78a88a463ab3631f04e070dfc747fecb69afba2bbe952930a3be48738ba37550e223133176a9aef9d52cf102ade2f68145810dcf423fdfabe12e1a6796b886b894fe6d1923e4f46c32a3997c6568607929f0b15bdcd83af4223b8c30742b5ed60d306332cfdf76864c3c7142b17ba489ab5421f7f68b71ea4957c8bb583&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331933817Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/d, Name: checkpoints/640/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/640/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=911e58887ed036f656c34264bc21597f2f60a0e5c3df24446a693e20be452297728914c8282d616bfba8c99f20cdeed54a00615f1e29f244a15c2a90a54d5ddc40107d62ac2a3c36183fd7fc2cdf3d5fa1d9c6339a57f0f176c5772cbd149f172771797dce611204b39d4a87266ebc523fc0a4a2c279d85dfb8a9584131779bffa8fe4d057062f1953632ae515650bdd1096771e53c04189e51ef668bed8f43c268e13c2076e5b4c5f50277e6acb459bd7f5d5a5638e4124e01c0ef205f16e266c93d30676f033cefbde6d0f14cdc6b2b727fbc1a742644dc4befe85e1f9f203ea8a34b05707e0b9d64dc72e29827bb04cb67349634483fde649bf7725a52f5a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331955999Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/d, Name: checkpoints/1344/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1344/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2c0a7b21d3efafce50715d74050422f2e5a3aacff1658912ac24fab0b50b79a6f6fe7e86bf34da366a23c62ce3baae3c866c4a707046143830a4f102b4ac90214f93c54a045d3d98c5aa3e14a50bd9e61023a522d64c7b2dfd5ed8cafc30587dba43f9cdb7d758fcb1463913b2cb34100c6bd66558b972c8967201e42aa8680951025714a4ef0bc011e204d89dcf2fd650e0e0f2e4ea5dcb99a6cc5a024124dbb1eba36b67348fc96034dd2c045908744fb9221669e7ca79c94d7dbd59a17a0121731f25fb63f6d3d02a215139bd7b72659293d06f2831d1242ce0ecdf2214f070e9c7dfa3c98dfd759de04983b928b41c538ab87bb2c6c8cf9af54a29601292&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331967394Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880, Name: checkpoints/2880, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2880?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=abe325f6fab0cc94524c2b1d5fab0463e8dd59ee2892b9af6f5da4290645196168c930d56e7d553831b8af340e78183994f8ea6a6373090c86f930f48f2e006e55d68054c7bc5a84f8f4fc78754b970331ef461bdc547159014db27d246591003faf653af9b02dfe3730a0c44c136eac4688724048d14d53540a073455ec562657531f5c8b26932eb1308698c652136e712418346edba6240c713785a52221d5d57e915ccb4a03f5f0d5c3ad0e17b13fd679219f4d129d35279f4974a15378ca9fe29980b9645280927d37cc17a1cc9b8a54d1eacd3b661919c0997c1ca0a5959834e6d214470e1658a4f27517a36c08d6cde12e0fc040505537e13c43ca6d66&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.332000833Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/array_metadatas, Name: checkpoints/3392/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3392/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1e1fafa07fb1bf90ba5c8745359a4ec095b154880c8740c49375ad7bcd3527965deae2db45220e0b73f3402289d2c63aac45f1ba8ff6c7166ae830eb1fb977f33bcef196d8d3fc0ae0d7aefb15b339a6fec13a60370a09a899a5e16f61245aebe9415fde29697f302575fa9dc588e83ce106b706853bae638ca3316424ba429566d044a972d91fa29eda6fb37426cc8ab7f6907ae9941a6396852c518f5eab063ef5c57e393deefa489437ed9f559165c647806b08dc9ace9cbbc32c3b2815822cc3d9131c2818987eea56da349775e920a1f41ab72a907c797c4044015ae6a39f8e4b1fce2ca0be8c3a8aa35ae5be2c0eb78ab215d8594281948a50598cd1c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331979638Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d, Name: checkpoints/4864/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4864/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b8e030fb8be096e837187550bc71f064ec39e0186e7226d5cbd36bd915aa5de578320a596dd2132aa124dfc2c782ac5bba277276db66d24c268aee66abdb0f520a754b7c4bb324100f350e67d9f7a0d1c10d6ec58da35788d48448b253c704539af78891c3122c5cbee1370ffaabca387cf339a403d39331ca2cddb559b6f1cc21ce47ceea42201f256d3ad049ad588d0978a8602624ed83e5f7907830642672cc568a1066a3834353183b73c98b69eba4a5b05e0ccfa796655843a27ffcd1652b2d5c747e633921cc8cfd0c75cda994fecc4db5cf41cc1f16a1e1175bac37beb52df03e4edc410540adb23cdb56f17555b39328ddacd8288fa884d04d414fb8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331972272Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328, Name: checkpoints/3328, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3328?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=65a9a86b86eb094284e1aa992f286491c1fab8b146fd9f5078ed38b1a5439a4efccc6ad599c56861d52f933ea7ef768b30632cd9f92e243acc7b4422f5e1db8947555915ef46f26fc8e7f767b41af26204063c42435200f3f02553699a4dfbd6fe8c35dce99cc6249618d6b3d246000569461375f7e243a6ae3af5ea3f35f1e5f2de525bde1e823509c48cfddaeb071ce9f22f092ac6d1f33829a0720003721288c8a17191c1cd1cf5d71707562e761955eae7429e0b10983ff5d343a8edf33e76dfb433998ef1252b1618aef7692d7fe025dff9dfac8c9d0ee2b1476ddefbc2d3ea2f21c1ffadb820638d90b34801a4ba76210eec93f74ecb797da913fe5794&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331962649Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d, Name: checkpoints/3264/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3264/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=56e59c5528a2c1859f488a7ca4185f218ef4c3482b8e15da2d47b9d17d7f032b0fe37eb4687449f0ab5a4cbaa09014bec363efc8e5a87e9a4d9e20a7ded3c83c83cbe130b1e9f34ac2130b0f902ae97a07e5e219361c157549563b8e055800b4b5e15fa1efbd7682424c83434d17ebd4d41378b8fd94bb41f33c5410fe2a0420d698548d5bfb75eec3f5040d203b9d2c7c5736812cb8d114d6d0ff89cb391844226bdc2c2f65f73d3258d6cab759280950f15ec1257eefeeba13c4eb293cbdb82d276c89845b6332838f444aa247acf96c36c0cd257ff0dd1f3ef304af5fdf27d49c0cfa80755eef350bf4a0058795762b19764eddbee5498540bff036a65dfa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331912066Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0, Name: checkpoints/3328/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3328/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=badd62fc77463ada7313c59ca8112f4e903e1113562714bc094e45e9f14890bd44593b991f13954df0655bd11b5e878be1177d3fdd464e1c2ad131dadcc9574724444bf4b1df8556fc64bcd83367c7deedc446b532b0389936954940302e4f513ee0cebe2d95cf55fce8e77a637c3d19ac31810dfd59bb4a6d08ef893428ff033f2a41abe84cc1db323ac32f54947a21e241598bbe749e3a1887000a25cc52a101a83b10d2536e37dc7efbbd66ce63089323bea9bc40b91fb2c8d88731994237db39c02904cf43852aa3504214d87e14f7ad19b0db4827d7a1a7ba2ee06e9bdba5d9ef99c5e00f896ad32d854f58294c80e615c4e63dbe1224f5bb92694cc882&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331982254Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/array_metadatas, Name: checkpoints/832/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/832/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4b6b5a786dd9126a087ac8781d0b9a4a2cbf0176a2a9ef680d8ec52f3203fda507af2686384d01c2c6de07458e5403995f6ad72c335d13aeef0bd613097c6427b9537a81dd25fcf2916104deb8d9b3e20bd4c63468ed37fcf3157e34b8ecc972e4412b577fee8683bb59e959758eb3c5b488b735b985d9403a1223f237c5633c40fc4d6ddbba6756db20a96eba3dfdda9da056843e6692a1d456c05c5f63088cb591759e6d2612c5a0c0724bfae48787f4be5d21af41cbcfe0267649f0c7099c578f4293ac11290aaf9372403d91c380d0ee6a16cf75cc5ec4313a69ca752dac4b6a69a715fc5cc05c2114725076f1ac089b931c0a346d5e93daf74abb52f9ab&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.3322219Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920, Name: checkpoints/1920, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1920?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=becc5fe079bcaf22c00eefcf46eee648ac18594e5ec6d2d3d02c0e00c13be7549fd7af2e8813d2b09335a46f7a0042adf78f8b5cc54ef6d16ec07e683e5d87af2c0ee07da53a5a4bee86b0c27cecfc37d423c72c5fb276a61a4bd204bc5e1c0326163757a887c7552c62336cef3e98795422855df11ec2c6989a458a7d37c877e501994302c49c442268e814505932c632effe86b4fcd9cd647b0e5d70821d8c4f645e4f88312ddfb7dcbdabeb18a2fd9b4245cc80d76d6b3294a50fd23a1d7cb37adcb106247f740481e85cc5a69f4bc26701527e91d886e2db9c8666dc83d6a944c74b7ca203d9ea18002c78217cfc7f6514032fd5f646bdf4b259650472a2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.332254496Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d, Name: checkpoints/1728/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1728/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4b26826583cb50cd9e6b752f27702fe84c9c6c1c6a3cb793f4af6f8c096926e83dcce1a8f57f2fddb831ad7b5f298adbf064bf20e6b559b8da7d41fe77fef804ea8e6749e1d77147135e5bea35fbfdd4d6730a4c70e8f40171adab3bd6512968dce49fac11e5627b8b6b16ef67923808a50a72562f986f0a7312d98528ab8b9910aa425b6692bcc79f23881ad28ecb56e2000994aef2751919cf177496bd0a9777ab228f1a9a317379b5f5f9667399252ea3b09d8036b8485136a10362fd9615e962684682a28bf445423f0d2fedb8d13d6c838beb92eeb09a36a7e4d59e8dc53f7a76be36c68bb6d290060ed8e6c317edc3528fd80a340e9233d5fb83775276&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331954845Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0, Name: checkpoints/3136/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3136/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4fcf6540bd1d59803f8a391529d091ab08dfb22297288f0a7586ae8a2becfebe0cc0cc4a481388dbdcc9d7b8194e54d28a502d20096662a35b2bd8af209932faba5ed15998db1e4ed3b2cecbbe8ca302c681eb17acfe41a05244c8d881d7595b07932c5ec0ab89a910cd09083007ad8d5b9c5850623316541436924f106315677b5dded3cddcaa50f2031fc5679013460e8960b77b3820db5edd2525ac3d28179b938eacd2055ff70f5eb2cecaeb48af5f130f18cc066559d3f8175f965322fa5c54f52cd40bfab2d4073e7425e675125ce4091701be1d5d04e686707ff18012e4da22138e3a7bfd8fb6dc68e70f53573eaa05de21bf2ff6103dceb6cad7341e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.332296474Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288, Name: checkpoints/4288, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4288?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=bbe53118447cf38f0aed906c633cb6d942bbcaa366110a8866158f348573ee2aad45304826863b1798e48b2e7f1d5e77e422fbcb10b520bc78c5c08217b73f289c11bfe2445c957e3c1437e85c7de3e7a6ad7f85151a6e38cbec6cebd65471b7b6d736e4ea637721725050fa149386d1ecfaa0ace6427e41bc6e0d366ad325182005dae6a6bf9004d346f31a934b39f67929c7297236c5a85d944dd960da1461d8f3d6bac915409e87f8d75486196ccb137553add4eddf196462ed6f7ac262fde8cd07f7e3c8a948195b470735f398b1b2996df5fd506de65906761f995d6d0d6b15f5ad5183f3b4d5ea89cd588601c0bd7eed4695e4349542442e3e70102e1f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331919Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984, Name: checkpoints/1984, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1984?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9a968f329457ed2552ad0312aa32cdf612e663166d7f02e7447809395ece40f7c49fa7ce958cb78e0bfeb07bfaa65dff7856c338473cbd2656293d213be0383b7093ce00fc4c4aa54d6e57f641494b7e0be11e6e06f68d6c93ee236ff2ea20b1823ab335a19ff1de0e514e5efad0785b71cfed4bf43f32a0801c0a2e56fa3ecea815cb450ae42c806392c5f96a817dc5b3baba2d1e58189d73b1926fcde313063bdd4682cca531e3a5383b54ee5b0cbc8bb53068d09c26679617a2e20eb5b50933a69bd185390f0290f6f2d62c763f95e48f68b3f70c078f398160ca7922af9f9fcccf3a32af6fcd6b593cda7ee35cf8d652348c7bfe5f628050103d4306f07c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.332444218Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440, Name: checkpoints/5440, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5440?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=372dc1a0dd090578ad310be080d5bb19063bee6955bb0a11f432a59576c28a02be629134f788201a5418b93459a57c2b33e47b1020f6459e3f21065300dcec25a814f984e7a6473331511f3b0952eaeb8e4396419b8c9424e06e01608d9d68b3b09d0a825ca73b9bb0d9300b72aa5548f17911969c6ba3aa5195d63b3f376b9c4a770a1119ef5ad3c6e151ba702f026f4595cf374fd3dc1e47b554cca1150d8a5e31d3dd74ce576dcf9cfc312b26117d65c49057c237de3765a70db8c9cc9d8875af07736e705c3e98e8adc8ecdc55ec43e034aa07a8c84ce15ade606238ac37fc01aff7395e95f161ae02a7af2b0f111cbecbe9034754b56b164654848f98d4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.332517546Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default, Name: checkpoints/5376/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5376/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=81ec02d9f827f4affd952bd17df04f535d4755be5271fc7607001115bb8f224debe198c36861df1d347d482f87b013e76eb0f9b115a47e7bdece328cb838f2914a419ea89b1bf7c0a2b1ace402cc19f4f5ad309ff65889fbb36ba8e1a4154af18e12a498b31767dd9a579fb8bcd4896c105805a8217e8fa4fcd361e926573bd5ea06e376ce4e333c69cd79e655e12db4ccf0cd1ddf4aa900765ce107f348d4a79d58ac6bb7c1910f33dedac93b6ae3878b4df8029123f9e93e4387d825be0853813dbab9c3603b4643055726d68810da6e3d51f96a466f68cf912fed43f0a5f73036e39e471dccab93784dc1f26c39882cdce6e48d2bcf9acac9cfc18cc35147&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.332529419Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/d, Name: checkpoints/576/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/576/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=890898fc828ec49952c96c2b22f0b20436c82673f30ecfa00d2b4f5e8d122650cf21d241c02d1087d4cd4087c96e8ca31399942476d046d38b82ca2a09bafda99a0b8f514ab0b3e208bf375f705c4738e0d9fa7514b3ecd2d1fcb0b6bb55214b7e4821af07146ee82035087d5a57333d2dee20a59933fcb6f7eb93419835837e3b935c926a110763f193ab81f1e762135a4b0b8f3bc37a76c0bb77b6424446720a9a3912ed37de6f3ae2368c324e49bc26c0940a6180679c4436a6d2f53cfe6a86fa4fed2cfa0914e8f85e66cf464129fb574adc536a55ca4431149da062509b1b34f5dd0d0b416ce3804988f89139f22d69c4ff055a3cd441858d1a62e9bc97&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.332583709Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0, Name: checkpoints/5184/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5184/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=85fafe39648e858f79cd832c52b24f973a90d18f39f9548f765b705a0e0037a9fbd7543ac7d61382248a732db74485e59eeeefa731bacb41e7b39fb5307cc6cb4102430e89894d82192f4a9d554d4fad44a096ffcf47f73da5564381a8017c837418f4202e66cca2ea6ba495ec2bb4915d3cb41f25999a87e2d661ce87414125bd5903f9a937c7e80ee17770f9e2e14e83e3004b06efaa0e6c278872b7b3112ec1bff022b2a943ce102b204a27b299e86666ce5eb384f00ce5ad1e2208f1ca42597fd3c4518281f256a83ddaf22af19554f60ba3f1b6d0eda72a30d42b9f6dfcfee7317619ba76f6e9e689b85dc11778e28bada5d843fec9bcb2ff054a4e9c54&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331894457Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0, Name: checkpoints/2112/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2112/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=256dc350eb206eebe417c0301f1a6e1dac2f797392208892b4d219daf7a851276d4e1cbe8bcef0908e7c1f9970e3e8c03b17c2f8d25dba951cc7d1df03ff8c0931916dc1952492241ac073dbb172952e75d3cb7d73dbabbc5f2a9a3b0bf6de9d1ad2b59fe1d5127a8e1c4629927432e42c4271b7ac680dc1213bf16a13cb1c938ce18e38fc2d98068a6faeb9f458019d7afbedf09db08a16e5526317b58e27f03ca83f00670d102a0d41bc0fb0cf6296ebb731ca34daa5d3e5fac8b301e51cecf0c21e7ddddd2bcbb60f2a89f584201343c8f5ea3391a78e7f4cda02df0e484a0c8094ee0b163ebfa13afb3b5194aee96fec0233aea3ec8352bcb78cd891cb6b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.33276125Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d, Name: checkpoints/2176/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2176/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2ea8003e206e34cd5d8688da5c87211983b36edef29685dc8319725449fcb6886b473b6476d6a8dfc642d1f2bc6625d9e320cc70da16990cfe5747e8db07f963c00d8fc4cce882a9582f28f4b95b419f0b7fe7bc551f8a3c162cf341cbe890d917fa1e593361640a7585cd4d76ca34562af5031ed06a1c6594baefeab422bf052d2e34b6fecd0f0e76e9e563baae37af68033674bc19d332a5b89f824ce079667097611a12518b7fb014b50399537c182ccb870bfe2a397739f6131d2ef57c89e7f56b52b78ece029e168954f9e37a0ca7838a25c966223a0f7acbeab1fec458abcaaf832887aa5c6e872641e068319fa9037a365a03ab1f403f32536b540481&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331720614Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/array_metadatas, Name: checkpoints/2112/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2112/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=97757db709b9e9d9e3c31b9486c0ccf019dc10b5a1ee6316bf7793acba8e3d053e114dc3b4dfe089d7d4fea5963f36f982779a173538258bc4df3764f2b82376b1b2c1ce939fbf989df23b8d42cfd67b406b5d53d475e8a31c64abdd576b2fed8a374846e7d6099592010c680ad0f50643e3f18201e28268071712be7a8659914a4ce4857b5af13cd98aef623309b033aab0c6ce40b55eb603cae5ce514a1e39c170aee84a997d72e06a6181bdf922566690292f28826fe2c000f14ce4a4fe024cabd1b574f21ee96743ea4a1167e4ab60886536a579b2e0e6e82e30819e877573f56e40599ad4797a1113e0b9c818d040f40b50ce55cf9d16c42d86e49fc938&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331717797Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default, Name: checkpoints/3392/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3392/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0aee75e9d56a13983cd917933e556762878d6aa1216b903ce6b17eac84433aeef6a6ea929081ff2848d45cd17dde6a24e907735ee959717aca4a477c47a40c469e045ab7b4016daa62cddc82700d229cba97d0d02f0ed17d16acc1b7972ee067c788400922f7d128ea83e470370cff1680d5ea358f0f0004f3e50a2cfdb2606eed3780301aaa26b96de6549a67d9b6b5f98200a68965633ffcdf61bf911bd8ff21d2801c7918eceb6d37f7aa8e5a2b8a008b251ae0f3879c3ff7f8df2f96162a41f65dc74113794fca206799b9b8dfaac5b00e943d990ca5d4b9dca8e634156f982eec13675e6c2a00cff35f1c589e7cf50bb73675ca3491522555318fe5933c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.333010121Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d, Name: checkpoints/320/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/320/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=74a5a048b79b41faf470b322821e8a7c2c9a9b7e774bb50eb549647432ec03b398d7428107a1782c8242cc0d2d52cb57c8ddb8b75a5678600f34c9246edeecd2bfbafead20e05953477308c23c75d3304fdd0ba41aec8240ab08542ec5f0400814129fc76919012f821410b7b5449b9bac26da4390f32c1758cc01eede599d3d9110cf99254f090a046771643fe8c93e03e530a8ee090050eee5c469bf9b745b5913d6e1b6177b7ca095d2ef791082c4fffd50603e683890c41ff34209f43a34d76b68b37f8645531e3a4f0fe99469b5bad4c6bc42476b98593f53da10c69c4208a7e60e3f129c9be7d22188b9831a389763a2081d922a6f06704348470fa498&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.333050376Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default, Name: checkpoints/5760/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5760/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=48c9faea056648f3bbf1f3f1f0b320585f18cc56d001443c987fcdc1a13fd940d75a44b451ce9ee779fdf9cb9ad5c1e4e8ee750a91036887aba358362154d0021ba16f168c9f302d3d6705a9b748841c350b4ccd42ac1c411b2c46b6edcb7f4373333b62692526ccff6bb404f6ed8a5e085c80104a4e4b151caac48e1baf4b22e79849a1973b5303c892d3c93ebc78e8f5507dd42757bb613f011efee59758d93651114770ebda57819c62ccd06090a9f966e39b0f6a53925356448fd6abebd5499e17287f9a4171858f3c0b4cbc41ff243ab03a21b8e5bfff1bffe839436ae3c7e58301ef2caa9f38efe9149c91c7b8d315f360c571952b0810535fe98b087f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331740257Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/array_metadatas, Name: checkpoints/4480/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4480/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a7b4caadcd2c1b77fc0f06549cdc76287ccf62381a791ff4e86a9a9e081384265031eb510435a279fc61a226146ce440cf9445dbcb34335ec89537238a7e34fd315959a38775569383954672954fe62c10f5ea07e56e6ade29ce5a3f21d94e8b658b320d6e715c2a6862a728dbedf2cce0633ef7e9f7eece808e2f04bc6cfe32465a638fc9bc98d34e2d6ca4a7f3e3b63b334e080256aa1bce541704cfbdf199e75442256d61e2f8ce3709714f63ce112a5e2c10c30de6de79deddea276f253ae8acb011a4918db09420405325abe97a1bc0363e3246481adfb133c3efc8e6cfae5c81abbbc7d8218508e8e51eb422dd8b4058ed5e811c692fd9022a2c08bc66&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331797939Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d, Name: checkpoints/3712/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3712/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b4f5db4450dfdbdda1807b8137127b568ed5a1f1e4155c4644f03a0ae49654b926fd989fc0510c005b5982572fdd68f553ebf55887d2c725eb68f3bc580566fcc11c93382cbf75fa735688e458219e1b36c7e3d57e677c7f8812b7c0672ea992bf776944527efd57d10344a681c7235e3f74778ceaec8ba2ada0908f7e60b78eadf3dc5f6a1af6b80a3323c67c7d7e56159830ee5cb135b807321eb05536a69623818d885a441477727dd717cfcf8ea96472ca7538ba7f5f90540e691ec7e628435bd591f55f60aa21c7aba776b2dab729e3870a5eda3274eab2c2aac39f8da8d0334df5ca7f39788c09a3514aa4d5cc8794db9e0c5d9e21095b24d7feb91e7d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331783001Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0, Name: checkpoints/4864/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4864/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=bd44db02f7f2332f0eef1a7bcbf034acbaa879be77d93a53e4984447056eb7f1bdf10ebd26996e4d5a03c03258ba304bb87af6bc92dc5a747b1390c7f19f31a03659a59b6f78f9773018a1de4780467bba7717d9de39b6a9cee3b16adc77aa16004e730c6ddc78396170221db831916e2232079edf1b29ebca7ee295a44dcc59e86298ad862226eab7292699809033bf0559f2c7be0ac2694ecfb2a0cd456a7b53e95714871a2f4db1686689174edca739fa8caa0deb3ecb1e18532acc6c134b8d3741c9bb1bc93a4a0feceaaae9240ecfbffeec4021e41a86b228c62cf84f2f614a8bf4437046702dc3181aca646651402b5fe28d8f297ff3ca436b92f19fa4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331813437Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/d, Name: checkpoints/5952/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5952/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=24943a0e08e3c3d5fc2a10b700514822b2b23aa014854caa947b5d36a3b899a5b5bab06d38b35b439f983216f3ac66faf8e6d4411d9e7e158ea9766ae510dfd4b566364e239513935a7da557ed7f3ea09f284d71c569163555c159a770c327ba3dc3eeefbab91e8f79627b0a331080e45bc6c9fb22105a2513904081bb66f3d4f10f40286915008079c0c98a3e0d833f2e728bb129fba0e860b89202e911fbeb5cd3acfc5040677f97db71b6ae332e5dd3348113a2eace7aed1ea451d7284aec48affd1c34309b5ff55dca541df6d71dbc613fb3872afe5e751f31beecc2a2f11bf9a75afa0a03008315a8d1cdf7247da79fa96ebd5849c744b783f61c0688dc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331680481Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0, Name: checkpoints/3520/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3520/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8058ebf85e48bdb133ff9a2df94478702b1185d9acd4abc0936d85b2d6381686e90da15806409594370734f48343b12a0e329f68f6b6154afbe87ec449e389f938ad0f7f2650551049a44d765508785469bb2c9f2a7bac03a495463f294edd8a2b44063240ee80b6b5da1ce41b5113439e7b4d52e9985085c884cd595ebb0aca59b3a83094d4e51acbe7bea91ffef5401535960e405d0da46cf289a50db4ae6c2122b73716a836e63c6dab1f5d31e713c7592fca21115cc28bdb68c54fd4a827427d06f733de0c1b1b9f125bb934857dd53d5ab2b193452f6607654b12c2baf821a976511722ce02980f37ecae498997c6b5812ca47eea3562df954092851468&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331806604Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/array_metadatas, Name: checkpoints/4032/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4032/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=840aa6bebe71300307e73dc768147b6bfd73037071c2ce081d02b9f780e243ac447d49408288391d27861ee2dbc85e675754fd87a81dec63fb1aaf223decc98a6e7a2846bd2d26d41471dbd16cf99dc0b3f03f6cc9112055a991065b4ca276e04f46591ca89eb853265384156bf5f6aa4518b7a8c70b61cff9f59c8b9e622bfdf7b4d10716d7efc5cc1457a68838c132e40d1037d64bb2b098f35441d7887b5f401d6b547f31361280e93944224432d63495d180ab5fcd40b4f56c49c4be3bdbb5045cb47e553c4179c341df0867399bdf377390d0433fddf5af65cfc1f0e19130c1605bf1504764fa3fe3cfc238945d8f4bdaf0b92a413657d33a4ada58c8c7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331831996Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default, Name: checkpoints/576/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/576/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b4d6684a68f81a7e3a180c35df86262cc8cdda8b54e802c2fcae1aad0f1a96cb0820ea1a94c059235aaf20f0d2ad4571b34f5ae145a74c3bf2400723b353ca468e14a4b8207def027a5318ec1126aee01b5eb24470c5319beeac5840e7b9489c63f213fa1ce994a029e29d040a463f6a9ea2a43eaf04f8a429137a5a6295fdab88f642170cee188dd28538d338e5dc393311f632c9820d4c77165d3bf5f31579caa704dbd628bc57c41715a29c99e2a201d1b95f5d08e814d7d7ae1afbbeb9ad9c9687e81ba8c7fdb7848e162393a407681e6b483c6c3fa4df78887a8b9e1b3bf19480a59d763527a22688e6cfc4da0205ac077bfc031ac0f603ed13d57332e3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331815185Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0, Name: checkpoints/2560/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2560/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=56707fad2c9a44bd7fd1af2bf358b6f8e12c56976fb8ef1b788eda41249ae8dc87b0da9ef31d02ea8650e5a8182bc6fb852574d6252ab9343fd1a8eca88b279f53ee3782b84f4ff93891979a13307ba9974544387c42784d58262d8dc99561a7da70552c03371547262f2277e40fb8852b8ff682792aee85f3fd73dbff9fd2f40d50f7f649d03a61fe7eac96128a8afe045d58f382457a9b35468d48644a1a2e69f69c6e057a49156bc7dd6330815333b5e0e2b2926738f86699f31d0d315cebae1ce6eb6090af18adbff8816df8018deaf265d5eb1eb7399a27b5be5de0cc86d90d3e653c3135bd1b0b40017d3282b41fdaad8ed6978a30fbeb2f38f95f9a55&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331836659Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/array_metadatas, Name: checkpoints/3648/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3648/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=320556db64a10c606650af39217fe34036bc794a011c48f14357cf20d1735149bb3726ffcddd92a9fbf2fa952767ad128310078947ae203b2edbb6960e6a04a65990c64a75b10d0889f1b45771212d9cbeeead8df6cfab63758c6343a1afc02bd139c983c5418d6c0fe88c0770adf892f0bf9b1ae752ac8de3ce567ec293399e7a96d1e0664fc75a86dd5ec4e4924324309637c4876dfed2da1e5cc8d8acccdd278356c1a5631a799ef087402b1a213cab60ba649a712d24e6e2f384ba0f0fe2f85094f8178c50832b4aeb96096322db831d7cfea18140770d75c2345461fbe3f0f59c8da4c97d022e14f8fe6e67208ee1d86d49997d2cae1b73c2eb8e01cff3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331850145Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default, Name: checkpoints/4160/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4160/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c234802c3567c877720c345adea419aed7bb5dea8d47031709c39b056c106416284cc90880620804181c2008186cef0662d29c231a211812411988344147e5935b5eff9ebe6e3e3ed502f92e49219c27056e018099c0fb74e74666682fc6eac2da9175a6fcab7257f5d1630fa7c43b5bfbf5240288bd0f885665191ee6a14d1977fef3e812316f7a465f3d388c12ab453becd5ab5d59316f1a47a17f83fa7e9c748e16d30909c6588169baa81a9e21b0bab024c2fbd73cb60fbacf90eb5032c04b679da92fce1d539c31b949b58f65979478ccd751ea09a6cdab4f80a0af9349bb4564a3a643823b55fd875f3e925b02a34e21dac6afffb4665fe6b761080a88&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331688399Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448, Name: checkpoints/448, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/448?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=18739c59906efb57353876647ce4dba7024937398584492b8060451f9b5a2f571c178241c445a6f6e4a7ea636514ffdb5fa5028796578d5a2be6c212666d66922a18f9ffbdb27a489704da0a19b0ef67d58c604a5e683e290bc7b0ad88970aac73e332c6d8660616b0ff57dca55259b27e0435ecbb5aecc771bc4867d86338ca46aff52dad59464ab9d91b7b9b06def89d6e8506c4b43ca940150b34cddda56dabf1f877ed772eed2037b5bbd1f5c6b5a4e6e5387b53cebe60df2a5fd524b50da244a4efc60013ab09db17bd7bcb586d7e5b7578256c90fb401211b2eeebfe4bb8b1c4210099613f4f924bdd75228c8aebf01296c31d7066edb07d20cdf5f966&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331855299Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/array_metadatas, Name: checkpoints/5440/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5440/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=54da16a699b92e6cc8ec72fa4e1b988024ee6fbda8056381511880d257a2e342619fc644829ee0d96879997c021b482c9a36849abcc16acb2725800950f841af458f9b43418edf886eed7c0c7c7be2717ef1e00cfb6e47f61e5c43dd6b0f3c36c17affabd61d5e686fd2314645c903cb37768393b2c414a5a38fce08f51483dc22b52195347b4698f52261a890ad151b4592f70f4b77e281083af6de90216a4f53f256dfd28a7255b5dc5b356991991854032628d026331c0ecd58bcc91537afdc919c92dac6939ea2dcccb9782d6235edde0101a36eda3ab781d5e3b9ce5a452e5f0c2b4585da3aa5f541109306df7b64efb546483321d28b24636ae379e574&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331824266Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/array_metadatas, Name: checkpoints/2304/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2304/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a0ab3ad6ed3996db32820bb9aa3cfe1eafe95f5dd36432c65c1a0251f3f5d83d32d4e20c9e411ccb834f49fc8b66023eadf7773944ff44aa65e4227412fb485317daac2edfa91635c419016036bcdb1874d924cbf78885bca4aa38bceca932517d248642de7d2cfab6714bafe8460d44c24c4cd49b58c610b9afafb754e15afb26bc19f28a0a7ec6c4b151d1154ff7658e817f4586046eb48c1c995bdd0ffa5f769e6a0c7b7c4e21f197cd73d88bf04ed38bf2be70f9e463f37516e262613f3272c0209e3e027cfcef69fd6615b935ecd258b5ea435a357f406fef6839f513e60dde26831ee086d0da820a129cefa9c5c7ee6d9c3d10ce1cdea4459c543b54f1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.3318994Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/d, Name: checkpoints/3200/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3200/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=217de988f4010436f94d784052538d3194a39e33dc36902e3a35a9794d47c5a52f7aa942212e686291f291f3e119794fc642168d47566db5fa7e7f237fc319819b08ecce6b8fc1f06d834639cc57f16f0165486fd06a817c2bce77ce503c22af0317b18590cd3b40044ce5ceff5a47ca7dbeb9b23adb5d50cac4a906657efca1fa487e25dbdce80a88ca26ee49556f1c965dd0a4180f10aea25e1b15eae8b57730f2a6632fb1170d48c2662371d63f1191ad407f3d1c27540cb57d35b810f3b429ae660c4d2dd8009630b333f2612fd3a65714711db1bbdf11856bebd91a9aa4386bf4d098ca8f4c0bc1ba0dc270a3e1584e773ac4a88befa5b08a68a3882925&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331701987Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0, Name: checkpoints/1792/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1792/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7a3efd46e4a4204256979df7049c75b642164a380a2e3109d3d457229bf0271439c2fe78fa8327c1568fd5b5efcf75687a082a9f754c5b623aae92f7471a7a35abb6d3a37a10784fd6df4f3ca3d74605abaa14904a8b2a152d67e2abc47b3768273f952f84be2e3f7a22103bae3663ee516087a25de785c26b2b02e961aede859db7a64f679bd20c5f69800e503eeaefbdef46dcd43615ddc8c871a308e1f7f38d534bd3a29210485b2e229278724aab5fc3323518d96fd4ccf1a3524c7807a58745b809e878f503055a6e3e4b399abbd0d54390f3ce1282349e9798ffc4fc407be2445a6c213bc9a16ec5ec6c3aab03b23cabd10802579f12bc592c195ff4ec&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331956622Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default, Name: checkpoints/5056/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5056/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=49f35915e90d4162b52ddc1775f46837299922b7b718bca14ddf1c95fd6f6e34868a8dc006ee550b64fa385e8258e4977afcac67b4666d6a40f9860805c937e5f1e789caeace347493739a28b4fe9c8a7bfbb05eefea51badc66498ebb4c8e3ead63a9e51ff626d9f469dfbe91675588f56b90477bd9cda6e25a44b65110ab447ceb19054aaaf4fe9e4f17b426286a6dbe0a72fe793a1c8f892c2b42aa378e1fb978195364de3d714c3ea635a45800f63c075589518a6bfc01433d2f426448c78bbaef62ab1e3b37c1bfa737c85746b0f411d025d2cb9d52ec02f8ca28dd224960f3f252b81ea39fb8a3312f2f35287363ba0da335d283608e7dfe12a9879a13&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.332013793Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d, Name: checkpoints/192/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/192/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1d6d59ac3d3fda7ed57627bd6d3af09383a1af97b95702c03db11c3d07c088e0cbf986d68cfdfe4aeac055127356a87d10eb5958f357c33894871ec50bbe5f15bdc017ece4073976efd3d4a63883bf673a43c4386b62e49c032640c1ff58cd1ad95625f0a9d18307d6f502b7a7c4af9c0829daa7c00abe5d059b35ca12fe28392aed5e9c6e039719d410865d6f26d6b1d655724e35d98946fbb15c4a0c9e06af877afd524cbbaaa0bf7a1c8a9a8b22099ba0ac4eb8bc5c18bcbc031f7af26b0c048f303d4fcb0a3710b20551fdd27d4bdd81a8d9d31f2e1a25586e71dfacf693bb718b2b7ff74247603876676bf2df03817c091dc2116a760ea2d4415168ca02&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331950928Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0, Name: checkpoints/4736/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4736/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=77dfbd611210757af6642f32e7c46575082c6ec6aa48a4f94be301cdc9a90b9f4c77987d0b12f3c3ea122dd357bc178f0324584c9e80d18440cc26c33a3996af52402f63acc35159b2b859411ab1ac3556ed6f297221b7beab925f7ee46715ae8d023371f930001e149de19f6d69a605c244906d1539b922981e4d45dc575036f6851755668ab1e41d94fb08882977c81fbd94d68950240f2c4492028d516d1d0d00221f867ed854376abfe6ad5e1dab06d743ccb3dae29b1ab98da64043965e61ba6a042f5be53bea315ba7b87561cff1eacda78dbcc0a467d7580f59d44e632223ed17621feda99c8dddaf1f9781d20cb2666eba1cdcd79be114d89dd180d4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.33193547Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/d, Name: checkpoints/4224/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4224/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2ada1746b4dc9ef962a2c8f59b52a57b12bf3fcdfa2b17b98a834e11318889e173771ac4724481d6f7da4484ff4e6a1ca290b55acf6f9a968f3820ca9e02899599832ed0e2543e70fa45d8cc8f8b8f764b76323953446c6609849c4beb86e1867e858311527262ef2a220f780ba80c2319659b2d6c806b81bf99a6e9cb4c40ef49feae2e2005ded8eef9971164d14e669dc33135200f13fd4a81ccdbe7ff98b0673a99df9a1e2affdf7b522c9e7d9b6ebf7b24c8c77c149b30979af80b58e190fdac85fd2fef174421860295a3a8be6065894f11d468a71d9f1865bfe3811b272dae9b51aa763c1ccee6c0084e3d34e23aea524643492e1a171a2878b4eedf50&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.332066178Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d, Name: checkpoints/2944/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2944/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5ed526c7123108bf5a8c7be88c8c3d7fe59de724db8c515a9a4dac1eb2a3d7144bebc44d944026c7fbbc99d9e4cc4db1c49517e5cb258f07b51abb498715b1d29a0967d3249e665cdf0b3bd8f0897c9c25e556ace3a7832bcb257e81b28ca3df6650e0e18912c1580c0818e49ecfe844950113764738d274dd73283a5c3b5c9bd2540efbc6bc072b46395e78b304b34a58eea7ca8bdd23ddf29465a93eb4ff35531eb192cdd43663ffe2e16a8ac6dc4b34a73d153c4120dec6279a508e2db2f99631845bbf6f4b696b73fb2ac26951363fc7415be4fda4ca1d19b29a9a344c83ff8b614705e237b7ebf1c23a468e47f419f5309a79c9848ae7c2268a7e8461b9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331966604Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/d, Name: checkpoints/5312/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5312/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6c66235f15b2d200b41d971ddf7ad9e9841138b534b65315485e8b7b0d274f61d083f219319e25b7b9b36319509177334c433a704b4d172089d0621fa349e078730c3adb8fb055b10b7303c23e7af2eb7e33b36baf8f0d66cd8c4ecf6a626e87a003311d0c590245880a2220113aa6a5e78559fe615417dabde7a5d778bfba7bd5c08c8c92ec6a3ee93ff60ac8d83d1771c125f5cc35c2179049f2efe6a9da257208fa9334df318907500bf3ed39ad8ec5fe2feab7fbe395d4aa4f79d2589ba561396bebdc4198f78de7be7f90c43bc589d18da1f797a5844f7d9a8baaac3fa9a2e0005a4d688c5d2297fa7cf95a52617c26c3f0d4c41e47aba9ad43451428aa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.332066361Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default, Name: checkpoints/5696/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5696/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=184e3a1f0213578de7b2286283eb6e6db561dbaf001103680436b1c7dddba7e562a2a2a87ddc102fe9ea5172d78c2e55ba751370a56606fafdeea18dc288fd1c6d4efafa4e233d4bd8620a2fe641580a7fbf7d69271965496b6b214b68e9b6b7c11f39e591158fa76ae75f961cf71ba25af7102f597239bce1558b7791e10294bd54a90a471841f35d0a30fa4dccf7b095d79c733cb50ddde590db8d3d754e11dedc8e95c7a498eb34793205d5ea78f007ce72a2e1ccecf011b48ff897384863995e16cd5b3394e0f564c19d240433bd8201d67e48b751ee6e41ffb627322c224c375a23b1141162af3e38d4e64b2cbadceae99ca970fc1c1384cf5618e75062&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331705599Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/d, Name: checkpoints/1920/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1920/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=28bb03a0a1ffe71ff7edf02b0222f3654f0365950d2b94f6ce260d084f6460359d212a5d51d1975996f6fd3dccc6a3bd16c0d6e95fafd7c698754e1c561cddf1266df952ce60ad663087310898fc83ac31a005199b0fb690b6b9678416b7d9a479a5c7d901ede815ef372cc2af419c95ff04b06551ed8ddaf8026ba0ece42065fbcee2532ca1541ce7113e74c202a95e9751345d2e88fa99613e2ec80a1110b3e6424096e8408cfb7ceac237a410aba17f802171159721773c26e6f7adc14af3d641199613b5eb11c9e2517a14a4a362a427514e32ea934952916008b56fe91094cf2106a1bcad4663a2cfd095ea115c0c7ee3fbd40c992929160e70f448436d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.334246374Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728, Name: checkpoints/1728, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1728?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1e851dad19a32762bd482f7f8f22e094b5b73b8a72b3e22d18038cdf25ca3ceaadbd8e67b3f6983e46c140663167492bc3d65efaed523b0a1509e671f175e3688449792c54b9eefe3b3805d861ebc4387b2ebb7882fbf8b353e85b46a841d5e5ce41baf7c6b51cf5f22d43cd25e76df803a670a01d33540d74747d13a317920e412cd9e6275d85fb63acb6fa9b7712c9f55663e2f1e4dfbc012e7a31198d37e08ecacc3539a68441b8771a0c394e12ef4057732e4bab5e4944c6aa83a6eae924d1a7dace76b2c62b8e42aa3e1bbb6c03fd34d8a83b94af201a9d32a5af7fe755ca20613009b08b58a9b68dc5a63ac81688dc6aea5d28a2b357f2fb761766b82c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.334328972Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/array_metadatas, Name: checkpoints/3520/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3520/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4ea213980db6236991b444589cd589f6f98f85da20639713ea131d92aab1e563c2b7dd0d496bee93bf5e6ad7e259214a58df234938500c0c70f56dd4f2fade98a5aeef23f0cecb4c122e76f1ae8c067f9c6118f73e5c9d8dd5ffedf8da5971e6cb2f1e7e4abe1531af1e8fa4e56e6e45d370a52c677c8c4f1459f285e10f40a8f3f584f80adca552028e3306b58e198862b79dcc65898b3df6b6d244994f7e9967c9389e7540f6cb21b8770bedbdfc4cbdff0111eb84b1cb0d48bf9b48564f4edf18a17af9d8c18e70593c5e22987ad92c5e13a864d2e2935898a74457fed825e33cbb1cf556d86e451e22bb70ef3335d0f8079ad904770b015bd8f7ecb74c0a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.334417326Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160, Name: checkpoints/4160, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4160?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8b9dcec27fbfd21d4ae0a14797eb7ae444de2f4c01dd3f1aebf2b80f78ad79f76a073cc9a8be2b7c9504b802372a3e1dd13b791cae07707612847456915bc805a6073cb6f7710aa04b38343d591eb19b6f9c73504af47ef2727380f8a0fa6a417333955aeab1811a5b0ca24c8731ccb89c7b2a5ecc42e4e0b22ceedc1398f35021d5d3fce523de8c68fb6f79b9cc0d1602ced0ff5489db61da190cc69866fa2cbc08657537a25d60de8f208d71dd99ca96e6a1962cd3b304a5fd62efe244e0fae137f3d6746b944ba12339ac69e1707abd9e0fe5cd5839baf53789d036ff2a2999988fe6f8a10230889460d57e4fc459efc1d758932e33153bef139717f6d6da&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.334603779Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0, Name: checkpoints/5312/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5312/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8d98b5ffb6cfcc4b7f8d29d7c97489255ea7b371d03c36c32aaac55599cfc1579787e678e65f1b755952369d323941a8c27afb03873afe25585362106edc338de427e76b914bfdcf624e4e8bac938486cabaafc317433e9afca8e578de92259eb0865def28d3fb321c426fe4de7efe591b73faebb3018e6e505ba8740d9954a7daf469e3b18e409d41ad01f2667b59c88fb317f819574dddb09687fe79ad338b5e09829960dd2b142b423e0dbe13d879df63bebbafa1311d4607fba5b811ca1fad3dd66f7a607040707be999609f8c6a79c8413ab5ae1ab4b8f14dcf1228f29a20fad6d86604a3785b0015b4fee789cb7c82620ba7d51f3ffa4da9270a5b0a6d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.334624939Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/array_metadatas, Name: checkpoints/2752/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2752/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=779d0553ef691000569528d1b092c5afdba3a9a4651a238032d15e839643325c74cb572a60e7ad860f8f1cc73afb355b62135066adf547199e342d85484ffc5bcad3f288b72d2a73c1873a0202b955864da97d0a0254774d72839bfe75bf18dfc1bfad8361c00763882595cac4a9f9a220216b8c781805fb1258a0fc03940e7763523fac9f2fa1e3272b17289b2fd53e82527220191f150c029e0e35d06b3270c0e5fd12f59634c2527c339e72730febf3e05f615326e5218751ca92914416cde001a73f71f534ca3e7c7c6653482b0de5bc74156e55a98e222e87d315c2aca07c980ed963224dbd271fd81f37f156a8bb8a0efa439d6be62672a8677eea414a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.335434126Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/d, Name: checkpoints/4416/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4416/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=25b8cf5f85b2414e4cc2ecf0b6c4e087ca051465bd93c635f72dbf2ecd8a006f8eea4d21667122d3b1be1ee1bcce2b7868be766d578e3b610f8236935edb1b5328320b718c3d0989afc97df22008e2a38f21a0ff8fda5659155539311e96e3e6705b6e47e15091445a574583c599cd089d60978fd9524bc2f79da4a40a34bcbc7ee31ac37b816413a0128e4f8c5ca80d5b84f7ffceb3b4349b7670dd228461195d7eca8f1f8158ca03cc4d190b005c8e9f040974313814ada3ce234657d629c417e60bd624139feea4dc9d77fb8fd2e3888edc932ac8025c1f7d2797509219d80537978a319aab13a5eb6fc57b00bd8225b4d916dde6f1ac6bb4209813bc6134&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.335598795Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0, Name: checkpoints/5056/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5056/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=95347907bb75ef77d00a958e9a3ee8b35cf6c2bf7ac230b989192db297c5056d658b367f49d4fcbad63897580b4638ff00d541a230cae05672708712222ef88d281a30371d59fed61e9507cb26ff23daf8014e016c8900eefceab090d49c5521e5a39799d0bb4dfff8d54587101c9b8825237e63e5d4ad5b11f2a99537c99bbd2f64ec1773ddc035ff1c454b0ea0d72173ccf7665a3b6bb42c3007746750f69d9b5b9229c2865dc7c7fcd8dda2be10c3e5ba7b98884aaebc4f85cbf0ea3f33bb96139b9751347375230c35480a6a2ac0752447ea6671e5655d491624473d61dedd62ddb5d07724f253ae73d939276f22390f72fbff8aa0156a65c0d75154832a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.336447846Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d, Name: checkpoints/3328/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3328/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=685fd0c07a211f6453541bbe7afd7449f019dbf5094cef261fc8b8d109b295d31849840837d5f7749e5a394b71e6a8ecc9766a3fa2207214c427b7c610c2bef39e3f17e6517af91963b9b16bf612095b7fdbe60bb3246f334e582ddcc369f331c3fe6f6302e90c08342553c86be1a5dceb909cf181e3c5944e079cf3787b4f857f0f0b61ecfd7d3ed4f21defc866c6c7bec86334cee31491b63fd58e89dcb1617f89a1b7fb6a74bf3645151faa0834b704466fff5620ef7cc6435c4c115cc5d7b79e9b4fb46988b15cbd7eae7e50213027aafd98efffb4bf22261711ba3c2add79b68695efe31fb86a3eccc589e3f98c7ddf41a19175036adb3093f82486d181&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.33657108Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568, Name: checkpoints/5568, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5568?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=253d59544d08e626fcebe29ec02c916902fd808d94fc5c6ea0cf787bc7389eff885fbfef8a217e9f77e88bf8246960cd7fff0ea8db9e02de940ff47fc224740a5caf579867c279f1c64c807edb046671a722cfae14d0ad029f59268a6b66b24a80a68d8661d7a83d73f01559f68997c557debf3e0a6593a1ce396d92dc85d4c93299c3ab069105d472c80f56aa7bd4f0f6130a1921e536d2701829703dd676e93f966ac8da3a3c563d0bfee005d26c2c3dd98547940589465caf408d67e2bea5a96c7e6913b9fb3c4a17c9db696d2747716ff89fff5713ebf66f14f6a071be7f0b40dc0a09b39e9792b2db0c0e0476f2aacc6d67a68afcd48e013da74626db63&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.336810161Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0, Name: checkpoints/4800/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4800/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0cccd4891a49bb284d607a37ef06c09cf4d343ec49cc109c7c2cbd3722a7a9be16bd41969c15bfa3776b4e146722b74367edd7729afa08989678f8644e8bb62152c710fb9cf9704fc591e6260d13ca66f7aa12a8dffcbf80e0a3d82a3002258fd0b4462f93ed57863061ac8131f1229234ee976b6b1a6b9c9270e535ff33863536d4f2d61b4583c28e15851d6eaa17d863bfb69eda00e332e280925bfa66d1b539fd417a6896b195af5ecf0dfb19b2bc36feb653ae931faa34db92d17fec7953a92b32f1d189b0995c4b9bc32eeb73239b85a0e6995db1fd66fa0b539c3e860605dccf161683e510e9ca371283d33fba39f6928514ba38a645d2b8febda8e1d0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.336858092Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/d, Name: checkpoints/3136/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3136/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=502f63d91b9d61a5aaee1a0c67997c5d8d93b7c336507cb38e1dcaad26a679e1044169e5987c8aa5765b61e3a298b1eccb012fedb52e694ec04f4f3012ed46e16cf8ee9f9d5e8048fd6dcf1df0f017856d9b7edcbec2b1b6fe33a1169f3ec33b4ab7f21931ff0143bddfaaa0b8f203dd034a78143918fa7c0f466f53e2cd02d5081adeaf13253ec8e83b3eb5d242d74b6684ee061b75b4c0a6276b34c051f4e2272458175652ce7ba733aed06c90cab0a07a03ea1d7220233b79db2d1b9f61595aec7b58f969a3f6b361c3cf94cdce04b9429dda4fb2286474ad50c6042b63d47667157222c482830cd19e86c33ed0491bffc11862cfab2dfae561782a6b7164&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.336956061Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/array_metadatas, Name: checkpoints/128/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/128/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0ac8b2521f7324649f56a2d8a970448a8a1d5bb563371f8056d414449a4186b4bb5f0eff3f87803766ee4b4777168c29793b0936035ae92e407cf83617c080b9444882889c1cffe8d8340506548a63afc6867d4c10068657e5c4e43c2bd8734498b313ebacffe721efc6585fefa6562286cf8bce05c2a437ce71a77b6ca4ad108b9c1204f6577ec93f52f6acb3c7673460a82fcf9c782302ad99c7e8f6c335376c164e233b51a58f269f091bf6685e8b816b77f22e569a72c61c9f12ede64f128158f0b5147dbef2f4bbc70944082a447b413efa6ae747c1f455d72838e37ca8073c886952df8a1a2867e25ac89760686f2d37cc7836e7e2a841caececfb50b4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.336462513Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d, Name: checkpoints/4288/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4288/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=247b96763ce724263ccded0c82cb14cb63e9717707d176784ff0997f279c70ea1a913ec1acafbc1abb07797b8ae4abe3bfa989521aab218b679a1f55b40f7a5784c2e0461d2c3c81d0f600cee5b45c9708a6c4404f164da8ce6a62fc4041f982f19f93f90a799b923c6934be630454e8de9d26cb713d98a3d0fb6d6a88e9e298b65ea19b83a008d76e26158465d3c78bac8663680fe7ba97576865c09633140373b23f9a8c0ff9ca99fa048b1392489f67248fd3ddfbffaddf9276192ccdbfdb51e21e88293859ce49a39c0faf75a30618291a973d5f08c02cad149231ee87adad39ac913dd6efcd59f214d9ede3f8b833fa983b9e2800497de25f60aad8d83f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.337659913Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/d, Name: checkpoints/0/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/0/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=27d532a568257b02cd63cfbe5df58e4f096ebcf20d71300605bae9753ed295e32b0789f396f8d1d24d92b48abe84ebea44b6788d0f35b05499f8de0905bb04670a380e2af954914c6be22aee00bb2ab7849f50f536e2c75980761bf15d684dbd1a9ded9ef2e2ebdc4996fb5e3d8f2fbc8a0b5f47635c3a6ea5b05cb0e766758046395f30412a4d910f044d986aea6b909ee8de52cb16ec14d4df101d4fcaef51ea8c54a814db86b9c045295e02a013b24babae27d3cec1ad7454e83eacca7e511ec9cc55612065e085ea42eac7f86c674009d5f3c0fb8ead7fa412f4c591291a70efd0d55de601bee9a29095c518c486c08ce351813dd1fdfe88610f21d402bf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.337891719Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152, Name: checkpoints/1152, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1152?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=93029a67ad5e911cd671c6770153ef0cc91b587e83b6b1f7192bd79cd5dd356872054ca0cab9b978dc04db0b3aa41eb2f87e935d2aaf78609c1087c4599e4ace4216ed5b35afd464badce052d0d9e81922af54b214c9ec87df54e87bb81954a53aef0232a823b7d26905ac7e61e3ea91746475e9eca737c5af7c5aeb907581b3009d64cda358fc12972aa7660b95639896fd546211ef0528fd6b3c4f91aaaad8d4a82d218c37acdde68aca0751ac66e3ee02ae679e95a411b33e7a651d5dbc8172aca79955b1a4ccfac23c621fa9582e01eaf8f250fa328a5a8d8a180b54ad696679fe4b798af5a7269960d8c154cb4917150a2aff2944914cf1090a5e2bf9e0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.339517065Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496, Name: checkpoints/2496, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2496?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a6be33abd389a34268d2aa9a731beffc557face4b8c1aedf2f9b909d7d46356ebe1447a1f04704f218e7d724b542ac3f61e80ef984edcee36ef5841cfd7ae17c6809b2756d8df98185178aefe9fe38ff229996e28f1bc096245804c5e3f6ad7da58fa4633290fa9c6320b66bc0e3766856d0cc09b55c44822ee900797888842155b6616ee9f23d1b70d9ed2958c599ed491954ccafe9567ff1f2c384ada05a042c7ef3fc47547e7ec0cd1e02be6f40e4c82e928e69358e9f60413635557da5da1a41e46a338328194f5b15170f6e976804885796aea2bfe75a668a337d6c8fb62628a860665ef0cfb61768173824371f657a0e5342f7b795cae3fee7f60aea4d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.340833756Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/array_metadatas, Name: checkpoints/6080/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6080/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a80ad10dd642e762c2f19df18b3cd4e384022aeea8190bacce41e5c11be8f2cd97cca9144ef9bf3a8294e3bc74a965bfac0004684b43c0fa442d10631e9cdf3e609ff5c31ec7077a3d5689f4c7f08503267f7094a34e944cc793ee074c3074d89e6d815410e1f2a604f1e69792ec5519b9a668ae87ad21c09836eb827d41f63de2ea3599bc19bf1aadf6094392bae1937aa087ef61c1d710c0260bccfabcd0ce22d7cf828ace6e136e3348388ee52b5dd2c8535934af4f89ced26af733c451424aa54a9434c0dfcddc1e05e5894d10d65243122b41ad1d3d23ede411d4ad9462376e1ec479847223baf1bc22152e51e734fa1800f57976e60c5a9d74b879dbc7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.340844013Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/d, Name: checkpoints/2240/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2240/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7bcdf221cc3e1761665acb1bc572523d5bf141166a81fc7ba882814e36ecc8ee8b94f544c1a8e91f397992823f3c216f902b8d8b6bbf1241ccdefc77623a858f739a5c9fd2c89e9a6789bfd975a773e2e0a98efb968aab8d83fa96165049d25024f65084e5a05709d8844fdd51f8aaa8e231d000cd3d1b0b49bab06aae62a9ba6b69a518df33887e9a14ffc602df4aea12f35615368d9ba0b441a465a59fb5934915143dc9f8532fea2fcb7acf186eb22bf574f30057ed6e014c6df9e7469e8ddf041c5db53e9e3afd2ecc529959e70cdac3f57c808f38e2d7acd74b5af16df2b8aafa4095f5b0d1025b22f487e2be8e2c8fbee6e1363a273e16c7814f6f53ed&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.341100354Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0, Name: checkpoints/2432/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2432/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=90f58d987e1fc1790f70f7a8720adb32922a46d5063f8d8aaf1d25321b5b69c531856a1cf750850fe4c4b44813809b7b9da095acd3646d90dcd60aa23fb7996dc6176d9e9f6dfb9939a85dd427222b1c8bf1ab82e4b124f36da0c65172d2a66881773cd394ab528d3866b675630a7e3e38729d093813a0e2e73fca9101750fdf31b5269d83c14691d6ffcda2a3aa0a8346caa798cc62003a7295ec6cff507a0be409f01f3f7f5dab6becca95cc91b4baa2299af086c3a9d0b4daac5315d847280e6df3dafab2fbe9d26028980bb135624740d0ee417f67aa80720e576df9e61b7ec4d680f2df1f55f650c8d97871c399b6c54df50f50bcab8aaa03eb9eb1b3f6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.341226162Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0, Name: checkpoints/4160/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4160/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7c4d19cde93ac20ec72015b894f8ef734c5a11dcced21e1602c220ac0fd921aae4c760a4e82e6cc7d362c170ff3634023e88efbacb93cb02c0c4d85b277382ac34f312ac4920823efc28bd5a61ec1f4553b025a652bb7241c8e0bc2920ae3d1c40afc0ead944c8ab082b94157e2e18eae6cf289c8a01a3a383cbf7a8fc9a2707d3ca56aa18dc0b49fa0a33cdd49d9233664c1bad5e75edd712fd4bf9f68c69edd4f582a26d43c9a6fb8ae979d0f26b24dc280d6ca9d65e454044552eda4eb2139195b3f1babeab84623cb2653ecfa38031d11a1a8006b8e884b533a91d68eb29502a7dee26eb5ea9b6054d80f981a89056f7d167644b0c4b390f4307778d3ddd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.341251041Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default, Name: checkpoints/2688/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2688/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9d0790efbc1f5b3f355a6968f8f36d39554827259b8889614ba5a2c8fe8fd6b5cd514f3a64e5974125c29e7eebc1b5ee0042bc91a8a815b92a6244b62f23afa53fc84c32655a0d8ffe66aaadb0e65bc782b5c97a3737f68324d7e1af6e5347b37a7c90f291b5efe83522869babaa10befdb3e39a8cf17b12013b801a64fcf6b2f703322bdcc2868247d5941eb5ddc266fa78df234707baf90b573f46182e4662a2e80569924f8fa40693b96195e26ad07abfc601ba79c0792302ba7cb3ef13eb14bab5adcc19d863ed7f2e34b7926883c5cf606dd651fb0a4ee9eca6a27d99de804aacb5d2d76ec03ce31eb0701645c4b2c050c9df9f3a82297b7cb026a1c0a4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.341234624Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/d, Name: checkpoints/4864/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4864/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6b3cb7eda719d201af09db9e06fb67309d60760ac607f37b7a3ba17856f1f3b4f8da5e5e2681e0848c562c9d3f298f0be74b8b4ac0d2338799541191366254e663685380ca38160b8a99c9a3d1f75cc00eeb31206d257e8c1eb9e5e95cf24844807de6aebe208da168a4d4723c65ab13ef7114d3cb9ce74afc77079f56113890cee833d2419f3a9f2b8af6ead03c9ac7af8bc8e7c2a3d7109be8bbe01f00e9579d6341355e66e556a4bd09e671f3e783e03283926d436fd97543ea33e8ae1d504b69e5458f5437547f70efb084d0c0cdb0c1c01045039bf9a9240254205f3aec1763f13b1bbcb078319dc0a275ac45d3a5fa5c80a5e196a7c2303b6898d4b164&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.341315529Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0, Name: checkpoints/704/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/704/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0c849a04b78a29119e822b6a87e2c36e962309bb00403f873b37476f50a4ea9fe5f43ec1039bff87ae04f88b32f9b0a63959df5a00c1670e4e179b0ec71c6f7fb580820bbebbc136d59635938f07ad9b5286601e491969bef2eb2660ded544144ef4900a1d06b01711fd224da0420ce2d5444f5c1c9da14e71f8bb37195b3d4b097b4c6e0b3c4217b123ca8ac04c5dbdc96c6723f8e253cb3cc5d2385480396d19799074d10c56f624a9b1e7241a50135c713669686baf8a3193338bf9889302217e6a0cad1c73181fb95a688677d8f237fbca6257788faa414ff72933023ca17f1805c2fa55e3d795c0979472ad55bc146b4aa9b8820ccc05849aa7107e66e5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.341379343Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default, Name: checkpoints/768/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/768/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8fba896ca1f0f5c720a8e187cf6f98f67d90d1634e3b1571e995c851cc3418c1b2fd30cc0df6cb7b864d3bf7340ce37897d98b9449cbd0974e67e08791a39dd6d6e668d18a863dda7367f819b1c047960016a8c3f2c0381361b1d8db0c89929276f11b80d4dbc120318a635cf61411e7c5e6f7e6fd1d3099d4c16572dcf6a3bb33a9ce44c1a52b04202671e347ac68e1f395ac9942ff75a767bb857861cfb3b188175e22fd59697289cefdf354095890c01dd4bfe2c5fa5b17f3288d6e70526c9fa7d68787a394ea932ae7f7044dfc104c48f12834c33f869708690c48cc15ee44c4ba5a0deadfe9044e9bcb899515b1cf49743f57998898bf776797b310f9bc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.341254437Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/d, Name: checkpoints/256/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/256/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4a22ede53bfda16a5f41119274d9952c4517c05088ddab9a45ba40b5391f7ae4f1f392e8b730e24f07a3024241858a139fc9ca8f5c212b71e3af1aa0c210875ce0f064de897c66cdcf6c25200dec263109e24dc6fe8a8fa7ee479a881ee77f6a53cb28243a996be36dad924569f58509b3937be1f030d78a08eb98d45a45fa056cada6f323d199df3676a0786694fefb3736627968f8d051fc2cd525f6d51299ffbf3068fd64b426f0b7d2be1629b770be220a11bf35cce27971ec9e2bd478b7027513b53b0171e249fa68cf2ddc68a73f896c8b789b90cd0c462102226bd4fa615c31f31d0bac758e1837a9efa4e3218289dc1b76feecc1846bd3a3e6283883&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.341494433Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080, Name: checkpoints/6080, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6080?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a8014d8cdd3f329e0120b766e98cb7cf6a5b15c372d4586adc03c3daaea0980986a5ec675b3989f359059d1fe722cd87266e67d98e64d4376bcd8e0d2854681837aa5f6dd8b2258e3306f9764fd0c6edf69ac60740f67a2c385d179ac5a2d7cff4c83a069a75392c67858788a9528f61ae9cd4eafb7ba37ecf39cc8375f8c96552dd00ef6c01bbff65ad5ad6b5d991599a77f5c2e8db4b5bca8876f82b7eb336187f833bc6fafae3a7cc92135443fa5d80179d763d63c104be306580fdeed9f26563b86f0e7c11e343b1b0e8150db63cd563bfc747587f6f3f873629c8a4e16ef39985d50e5b56b4382eb8c67d21adfb137e4bc24169238c5d952ad53c818b02&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.341717348Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d, Name: checkpoints/4416/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4416/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b90cc9a41b96e471717776209e02efc099e8d68c28e5b84bc50b98d574380f88db73b05f226d0cb6b919ca15c752d99d826ea4e6f230c0086cb3438799dff5fae9c2c3b1e877e0d1ffffa4f16da3f27eff905826f5d7883bb026b2675944ff15c3e67f38b2c3da1512c7de5fbd519c1732ba61178b47fd4ba79e42f57d06acb1221d4b7acd56dc3cc9712ccdb8d5fc4477951f4b490153836395399f528cb29129f5bf576b0aea6ee177990b24e2757775b80d971d6cdac20a2a7cf8ef8e7e1ae9775c12edf16066569def9cfebe0afd2a7f203d6fdb617ce7bf6f41b8775b46d636cbf434f985e8a0a2976233ee66f1159602e93ba6823d95888feb86ce23df&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.341954949Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d, Name: checkpoints/2368/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2368/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3740f6977bb83e4b194a0ea9367d9b2333b9ec5a553e96caff9d0981fafe5e00b4f5104d3305bf97f24b88e07d6cf1f8f40c68f089393c9c72adf5734027740602f81da1b149ce87d8e3fbbddd6a82b579c5dec5bbb3fe020c0b3aedabcf466490297c4341c09db83aed14fe180036b864ae93c655b45d6c4de83079e0c36865a97de1227a8544403b12149df176c0c9a905fff3077310e6034712957732aefd106191a3112584b30f58a0fd0e2e9a871b56476753c02d5b4f4eae00bdd20174eca9f405bd378e6255322a08a85c2fc19ecad397814b2634f3fc00822236aa36a8db751f097385954fc6b82bce055018d8b116ea2a20c18419887f310c3bd2b2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.342177528Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/d, Name: checkpoints/5056/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5056/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7629b2d864c71bac359cc77fbb9df72efac00b6f372c564aa6eea7f59e5aea836394b8ee23c069887413024031e17a1708b71436546708709ee967fe1433de461552ee3bd7ec8926fbc86039bf90578b420a82bc0f63ba10c5c3dc3a8910f5ca6d45d847677fa70b7f68900774dabdcb3244a209fb670deebd97068183e56b7cae57035adc903abae5d91241cf6d2fe192b33f6129adfd7afc9080a70c0ff38b146d9761384b399b79b3eab22107d57109cf4e7c08bb5f8c88211dbf3d219bfde279b83c745a0d5c0f48722df7aea42c2de1bf7429d53431abdfa24cf49a00bf82f91e2bc12eac976ffb01c020c92b35d7021a66dc22b17b310cba68f21ccc63&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.342197316Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736, Name: checkpoints/4736, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4736?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=ad82e2ae8bc6ea5133a3edbd7c57d87c4332fae8328537f29626f33b8af5804f2be4d07c4bfb78a276b566fe8bbf359b1a661e6a52457f6975f29690176f4b74813682671120cdb1a05866806096755d44ecd4bf39bf634a91a3c1ec67555406d231b906cdb0e3b0b7c747ee175414bb00d41e26c67929258c16c7ba31341c390c6ad713bb8ae7f707cf836052ad9c646540f2e1ca0ffd61d81f49b84dffbde113ba57698e31879222b7fb741a90081c40824fdc865ffa10d3564c8583821ca05e2b5b65b5129c58461b92f5a7ce0fec3954ccc40350fbf7f26d5efdd1a8cf96858f37f6e759373abd8e55cc099a1d3fb061f155f4352d3bb67d965ed7d57753&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.342275305Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default, Name: checkpoints/5248/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5248/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=11158c9fa6268073b9aad939bdb7de709aa7006ec4570de89d2536e50b096a48f03d179441f37f3d3fbac8fe1a64605a7e6b006bbd3678fa6505765268fa488bbafdb540b6acf7ce359f20ae549908070eacafbcd9be3d972e9f2c66577541969ce4efc5442d1284a69d16a4f883dad1cf14b7af198523e1f2b2e7c681ade88e3bf8181bf3aa84783262ec05c11b88164eb8639bc70edf51213975c7fa6cc38c45f4c3bfed0f5500cec62c9152a03e2659fbb588180567f034ce9dfb2e83a2df7e7c182a5a30ca5b8cf4722fdd1cb583682ec7e74179091974f31bd8678e52658cbd4ddc73ec1af7cd8ce2d49cb321a394c89a839b2ac5152fd81f8e45a0943f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.342188754Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0, Name: checkpoints/896/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/896/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=be4b00f4975a02867a99f81c68a3df3dcf2b36caf028916d041cddd3d44b9b042da2aaa27a90e3e8444925df35c06499d44edb2058e8191fff00297eee2612c76de9d1d36868dd5e26c2c1e11d94bb3d1ce91d05f5141e8b61c5f4ad0231f36c2552262e8d65c4d9f37f9b5eb794982894acba9d8f4e10ee54dfcbf2bf25c7389d5bd41c3a040ad25d323d5f5ff0fe52a01345bd309dc553788c688f7f3d6fa3aa25dc90477cfcbb697677522af3ae9dc4fdcf4c9c10da531aa569a0ed75eb3c0fc45cc74633ee948024faaa7fadb8677e0058e052a4a6f837628f0d82cce622ecfa6196f7a15ecce9fbdbc731dea4d1bd780b50d58b4369d12b3d2ee528ad61&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.342370689Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default, Name: checkpoints/4480/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4480/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5ee1ebddac1fb5440813c426c76ff44764b6ac03c42b6b50159334cb10d9b50d171f4cad99245ad03fc956729faba75463fb5535f1b3acf1ed1a1c27cd4b56d7c4e1c77026a5711e3ec572d19e24ee9c03f63d3025e125b7cf71ebb932c115b1f191a018862eaecd277c81f9d82e8c1c383eecc0d7f419852c6a4f5a072dc598ac915a3b1381c49b9101d256e7069b78f4996f582a77f4f73bdeeadbb3cbfb1750a935929cab7f3b67727405bdf7d53d3e2c412f94a211b6feec2350d703d60934f8cd47c6935eae1b9cd76257c099eea247bb3141011fdbb5db742b17eac530fd11abe4f86006913f80a5bb2d20dd5fcc6be6a013027bdbbc7670c31bf74dda&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.3424235Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/d, Name: checkpoints/4544/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4544/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a9e0c6a0fe66ee5eb3ad39bf73dd68b65b6efc14092e06241eea95b9527cca9dca243f43a448b3c9f53b1a42fd9a2c40dd1bd484297016a94d3255e73f757953cdf975ac9637fe59201ece887f386cf73d6e8b438cf716c30be9c4b55b45fbce4aa2cac52a4c5652933adb4387579c0eb1cb2168d45666c5e2790930ff4cb6c3bda8b8a2fc827446887b8a73905addf9f3462b65e5b2261e89d9de9336696e56c568f0fb002445e48ddda8a009886faa38551248894e034ee52abad1d326bbc9e12f4ffabaa3f415180d38d7a748c2d5648ab3439c44c22a20ccf577dffa91faa91c709c0b18f08242698ed3f13bbb643c5d061d502de06287745d1275a8f0a0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.342289832Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default, Name: checkpoints/2112/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2112/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=99895f8d3e449be070f6dc262cfba61ac2026f58e0e13c1bdc32795b053d63a0e760da4f187c67187cf735cc4e04519a09d4d06fd9d9d3e6c3e3a728851637b8c231766f9e17bf1b1fd490806af44994c644175b37cc2beb332049d796933c1ef0c3f8551d37371c1d6e124b4773d23369286a855ebf4b8dd1801e0e6b4d8a9276e46759a81ecd0450f1cfea7c9c79bcb829919286537dbe13983f930e616174f7e47cf90dc6f17551691c2c0d4f7fd9e138f9209f1cc3ca70d26eb151aad8ef96baaedbfd4d9798b000fda3277b79b53a237fa1327a3617e56ef2b633442f91fd77167430c8f5bc55191ac3982030e9cc04e56cf3bff51103713b4c72d93746&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.342512618Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d, Name: checkpoints/5056/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5056/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=79fbdf7ce87abc8d3631b920c2636ec3d36d560cd59713d0dd626efda487f16a5b3bf67edd60527cd1d4cce6d71499a0eb5e6f558b4c61cec0556f3fbded18c18cc65c447492fdfea39da383f7572e8c967fb058095641e0fb7f1ca6e1757e09c3ce03bee2abc0cfd6dd6c15407f57793a6253f86fd2537275430c4ad5f723d33ecccbe9ef39f960914fa9fe0f72294b2a84b29d8cc5856a4ebb45101ca3bc7be21c297031c39a87dbafe6661e7f51fa3679da87246710ec040d3738038818f9f63846a1d3efec75f493419c436375cc7e3d3f851dd4c7a0a3823133378daad896ae035590f1f26ac704098fa8e2dc8e4b92dcef659c7a4729999eda482f9b6f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.343526785Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0, Name: checkpoints/5440/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5440/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=59c3b8b57ee98911807dd9ec44755abd98ed56c6ea6f5f2ca2a211ad8f7f1281ebff6b250294556e1eb2a4a72653f5035b1d26c91c4a7698723dc9e8b828b5cad3f061aab9072ba311e6e19232e4fc10612023cab8ebc04869ac96d11eae17d32e60de9e375d03ca6fac283790b4ad22ae69dc1ddfe5595c45fb72711c6632d52c93d0da8db7442c7d0e212ddb38ce8542a8433f2ebf9b51a5e512c8ffb867f4ebc7f9eb2f5372d654e5d946d9127995f70640102cc9afbaa2027ed5893aa2ef612f539f2435f3ed332b0438db1bfeb4ebae10644df699c0dc7a1298e851914d0fc54d1d96b319cb07bdb7e112bc41a0c1fe916610e0bb0109e21e81f6eea81c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.345051191Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/d, Name: checkpoints/1600/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1600/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9248c8cdd1a32c3a47040d051e122a3cb52883bfdc3f222839eebb0c09af52c0588d4c33ba257b8e93cc77ea5d7d4c44cc443db2aba9557968684e0d96adee0307523eb44eff8f123b00c5362099a1940c9cab2908f5edc4f6466c2481c180025a98a9c1c815e9bde7d9f619b993dfe9162840d8c1672d7451a465488da298b3c03367285591a2238149b6d425892efd200d800222e97e004eb492b6cf8eda8222fb3a171a039dce210887a69fb3049823ed6308cf1ae50d7ce40c499f1b5f2ea9ef4194041efd5906bf0d92c4ebac39d6d0491faee97af62158c22058a0689f2a12672f44a60788fdea89dffa1058be3f9931c14685f3d22cc85a9d95d27748&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.345417272Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/array_metadatas, Name: checkpoints/3840/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3840/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c01c37f591d031428d2f710a96461fa97a9c7436d2aaf2dd624a35f21bd74816b8dec58c8a73b12def9dd1d095f7ca6a972266fc3d5e6e46051f2cf9bb81578107c63bff574f93d54e8f2e03978c607840a9bbf4c61ed0521cd5984effd983c42408a124bda5ad46598c37e94144c844f1134aa0a28ba4a2f8227502ea8f912d04f273b383c305aabaf5dabd7c2af8c1c2b1e56c7a3e210850b6a89299bde240381965b2e96622c353bbd77f4b0ea8f2c369be69bf553d003b06972b9f87d2832f475090f8e4169029e7e5025940d2fb97aa96eaa644df6fab71f5e96f25acb538356be6d4d046020d0ac0f932780f3dda20602cb8f7b34c6ebaafbf980d9a62&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.345428935Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/d, Name: checkpoints/4288/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4288/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6ac31db83cd29773f9a6edf7cbe8cdb3ea4f4aec3f34de79d09f97ebe032c3b686f6b179d88a072b8a373799d4d0d85b7517351432accb55b5fd30f1f3a8eb98e6a47f6e788ff260c7f7ac9745cdab199b582c5fa3a3051df12927b6afe9fb2f8d6ab26f1d8aad816c883b940e1b261cfb3ebc11bd14efd9c9c75fb6c116f126a6dd8cfd55625bd80735e8bd3709409314c95d616736edd1a13cee88c094ac24c29c5c1b331fbefb8a4f5f7e98aeefb122bbbd5834c8ea0db0d1ee0e94f82f3ab28693be23173bfd7e5b50e4186f321b2914812a65c7c5f27f49f2f9b7f9c810566e61f5e8433cdd405006c7e0a90518195f3a6617709e3a23a148bca6efb6bb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.481604646Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default, Name: checkpoints/64/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/64/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5be05365330b57ea8f4807f6d0adc54e622cf36440416a28ad6d8d755bf29bcd7dc44a2414b0eedcf605868e6d99a533116b383424ee27d7321e91556de4359401f7556b2199c9851e0f6983655df591b7ae8ae9a5efd39e89185c0a1993694a785f508dca8f421c87559e50ee0baad0200fae6436205df0124940fd3bd56800b430f5fedb2c5c2db85fe712e95b6f3de4b729b0e5e0f03d996f3fbdd25c55dfe64aeb74d1883a316f874e742497ffd7625dcb74afc3121edb9e603490a6dd90ab3ac2e108f16bd71660b94ebebd594acf0ab49b042acf55bfa608242c712d6183ddf658707428eb8c6251a53e8c3018d71e2045863732d4ab59b93f7ae3dda8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.481672396Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default, Name: checkpoints/960/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/960/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6d97f5872ffe26e3633020931fef0cfe3347b88a83797a740d710f42c779c0a1b82e9e8d205c977be3a0f02ac7980b594ac00a5bd2a78d5ec08e0dd658cd0f34d110d83b1d1370a64ada642a2330e6570935a8a28f26acaaa1ab3aacdb73ffe50782d5635b14c2d6fbaf35c8f6c8c8a42bea2b936133ffa44e359cc7666b069b390a6554e1fdd0626095234423d6386b6550cf2a7e60c5ff7892a49d93812e4e3e287098dd890838702fcbaa825f36d6cc97f76a1d5d21741a124f29dc293a14a78dc72dbbfa02d65cc658a2eeb47f6001a420aefd909e34df55f37d85828535e3c5c75a0fb6639a87aa79e87e6f5b628767634141a10ab16b86309b1f2e7b23&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48173268Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128, Name: checkpoints/128, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/128?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c4b5c03bf5ca62de80eb7acef9df6fcef3fa39be22b681ccdcac82fa31e8956ecd91bd83ce3b5b42b692c109b8217bb9dc0117b3ec8abf2196ea833c73e84109bc7df19a38ed057cbad8b8a7a0caa90af19f1b74a094b2ba1918ee960f74cda34982e20999ac2dfc783d01dbaccd5cbee849b010963b4a5901c94ab02d03338f5542639a134d68e14917165c8b179766b175dad557479c6308a34b383d23a0304d8d132f65ca5f2477d434a130fea88d86066bb91994e3a8a291671086c630c6d4412f4c4c0feb832eb2914b5fad81d274ea8bef67b0c756dafe1689aec09046e0151870a3acbf1b9a9ba5dc7816fc25703fefc0dba2f0ce2e445b7839da6273&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.481763479Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/d, Name: checkpoints/5440/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5440/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=007592fafea18e86306b279720106244a70074b35798168000adc31f91e08d308b6ad7ba7ae56b604ce1a4a56280baf2571bf26a8bff2973b1ab786dfbab75e6ff5f3935d91f8cb001bcfb06d2c6fc71adfb78d35aff169011af47bd4bd1973f252f6437d94a5926a2c842802e7e826c1b7c8ec6d247f0824b973c23bee98d9018f0a7e50530f72fab101118c13582975124935c5d82c7968372531c59633b0445b28cbe9f2699850fd66d82abe78eb5a8681c960617912ccddb5f64620f845869304ca857f4af7cb676c1576b916b41cddd163a812e2db9ac1def6bcbb979efa1422f4b67862dcb9770f16fe38ecea99d2b8cee3c176b5ed6378d5c9e87144a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.481847213Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default, Name: checkpoints/3520/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3520/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3db7a10d3fb91c5e6667508de1d74a57a5cfdc301fcd6c23bf22c6d1df2b3800e019afad444893036b9ff49eff83ecffda416cf4d08ee8b42c978d3f0b9e562a046322cdfdf9e8e31ff8e9702908094eb8dc544a1b910e4f1383ddf0b52dd25b1209d4041cfbb57d17bc40ac2798696c6953683213ed1434152e8c5bc8c0f52004c7e48dfb8c37c37901fb66a84cead464ab45caf01d569e398da654268284990d43ae9d4e9be4ed87fb2aa5a16a5878d2bac01a9b3ce5935f9403b30717b8e9d4df56641ff2bba4b46b7e8eda2c91a8620992e56c5ab293de088e142742fd22789b6ea7a5299c4365de52248a7db5117a9aa1fbe45c0761fa77209575194273&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.481947869Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752, Name: checkpoints/2752, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2752?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a42fb064cbcd43ba4c3dbcc914f564a39bc851e18234f966819f730fe4099af24d66c0f7668608bbf62a41e9da22033851a4ec4be97bf1bc69523f989277f37d48c94354ff4d94aee2b18c9bf85da3698689056d6b632c0fcd9afaa8f3641e60f7fa9842c1e854b511e7ca3634c9ec4988c2091dc8306a38a64352924c7cc0ed2fd571e507aca40bb3fdb3949c5dcd1e4d11b38afe8aefce2537f3cd1276b15aa4c818eb3079afe27fef390d090af62b9df04adaeb6b24c2ab07ad8cecfff8e10125d6912b1481fc9b531b3113d64cfe9607f1abd7b2a6c124363c703a7488049e98db1803a6149ce5b5ac884e0aefac9b99ec9ec1cace7cdb7f4606967c9248&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48199567Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d, Name: checkpoints/2752/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2752/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6318a4a59156905bc76ec750e4de2f96cbc1e5f4349afc582436397cd646031040a8f6861bb62eff805ca7e80eebad7adf25819e1256ad63c2612f4613a3ebfe1ed7f582abd2b0b637a1b645ab1493c096e0e8aa00b6c49124d990a7359b0e76fd477998ab50e94837a6a5cbbe6d23426fd6a94cec71afb67b8895d00aa96af087d081e618ba51d16cbe0e0361100dcb7f7440697574f9bc543633f478f34085260e93c2ce5a999b2da9542d77b1db62f4559ec5ac48f21429ddedf72d2e8b7e5cae5ead8fe2831625e7d2629fd39340e6901190c75f74f2bbeb0c1a2cbe7e4f3bc1d2812463453c67394057e17dd1060b8d51cfc611eb8cab614e6e709154ab&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.481855921Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0, Name: checkpoints/5696/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5696/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4c504a2c1a476a0a6152bf6d8a0da166ca0cd9a0ee33e8f6d39b649f75997e20d308b5011de3e76a3686286f098325594784814e36cc932bfe3b6a7ee86b970b1d362506d6d49a0fe4562272e7c7de7f37407f4616d3ec42526caee066716ea70fba5c0ec5033e1a270e9af57125a6c94aba47fc6364543af24ab4ada795037555086f710b6f32b26f9d6c03521b0d28277106ad5d8d26923da34702c4ebd61f7e3653c19f52262c797771198368b5aa2c1fc6092f5ce586384a277a9e0abadb18a4d9a534a4c73d7098d5b65e76938a7d01ba025d2e3a1ce480d52b70e12003e3e1d52d3440df6de7e4fcc96dc877aa9e24ebb87d5ec3c140784b5b2f791b39&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.481854494Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0, Name: checkpoints/896/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/896/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=50d7fb6010e7eef69b67cfe47091f116c4f03c23d5e9c92864ed90f85c29b282cf62c774f89e6db2c7372f6e4cb28ea002bfcf4de08a7f51a97bd3c849834ec252c2bbaa93523a39900cace2da1081bb4a1bacec60e3600ac55152f9e5b0bc847706aa6218d8ecfba6486f37ae0b1edeee07add9debaa052dd4b71872900457766a2ea792fc21bb01264677f26f0eb11dc9f38da5117834d5c155dc2fdaba24ab7346bad98c519c7a2882a9184f0e5430bec97f10862c6d4afd45f2c4cf4159737b64b99f0f920505f652971ccd77ea0099000a72bdfe265398a23e997a3e03803151208c6335b9cf0bb0907782d2eab22268fbef51767284a6f03f7bff59dd4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482066905Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/array_metadatas, Name: checkpoints/1984/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1984/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=702bf8281cd007c598f26424d3da9619656d82b59db9b917e13b239b165d13b0c9630619a494f233a0a85b2bfb348f12a18d38d824313e21be8ac9127aec5b959c8835e1ec47053fdfa11cd5306df61c01024178d8818aafb772de6c826da685600f63b753383ae4da8f0cdde60fd3c20e727157f5f97fcaef347e5ef258179a5056d6c2d8213ae3c2455ef9a7c2ed4dfd05cd1b5b5a39431cdea28d72da87a322e4fa7ec7fdc8f82840a356a9b60f04dfcfab22a0d73d5a48a6d7a3fb5a223a0238d6773011aedd35df81abe3b9081a7414c1dd3b0ded5dcd3ed30b868137d4d1b72b9ffa070a2a184a3764bffe85f4b4811a7533fad5ee7b024cbe2377e272&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482165186Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0, Name: checkpoints/5504/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5504/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=810dcadde89ecdd070c7718244c9ab52d61f17f4c570205a093326dc2abd1ba463083cc26eb906d7085ba17ead319ffcbd1039c6d385d4c5115584c26c46b5f9b884791c55dcee1e48113d38d4d028e700aa264e286aaaed758a5d32d95ab398e4b42137bf27c2bf50f80a82dee1d501db8539979c0a59b2bb60baf5e8dcc5e1bb910449edb7bfef76897055eba6f91cfe60f969b65bb91c7b03240db1579ca9bdf9581de4b21e5ca26b2bbd612f2b8ce86139d5a2f7b42d54c67a603809cf6218c5df4c8e7309d2727955266013733604105682274005dc566a803905a8a2dbda5e5a2f67ee2688543bd54c64de06470c29edbb353722b4205700fbff2ec9a9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482116179Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default, Name: checkpoints/320/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/320/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=16e68f85653a059eebfca3b7fc186c03911a1a4b1bef79c487c2b06ecf847b2a257be7e6c71f7e3f22a08ca17ddbc4c05f8c7d780d2566b0aec526cf4bdfdd5ebac5e512b633bd0d79fe48556afa16237d2f4f1be51272b04749f6c9466ac9ab6b5a59dd504c6fbf1a1e6393c03b851a76675f8dfcadabc359806bfc7ccaa7a6a477a8ca621da9b3e504cdb7635159808f5a53074fe277df60346cb149d8401311de6acebac1fa8c0c01babcecb696bb0220e9bb4d56bd64541e408307f6037345e556b84c717a6d1809a85a188a9f3b0c7b930e83b6c076299b1574a9f23aab9a2fd2a74cc2a54a75a078546118515fda657a5f54f9660e493cc20bb595e282&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482176174Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608, Name: checkpoints/4608, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4608?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=be982a54425f67affe281bbf3052eb3071c7d172373d6b2672eec5b23bb27254308b2c65a9cdf7130e1fd2178b889933df4da8dcdf405d6a0debc94012efcbabbf91892b6d5b630c019f7bf0373aa678e9b6cbc651e5907fa70d8589d12c1f3ec6a2c767d0df6cea240eb794c732eab6da7acf98a1bcef0ff5019535db05ec9f91637a58d4b232fb02e82d1413bd0f216a83ce4ce63f259cae5b25f176b35bfafd0f199a4e947c7554fbb15cfe218bd11e23c3cf699c2c7bdedd25956aa522311f56c4e2a5585db9eca1645105402556ac19418e502a16df505c731749e1194c0fdb8c9e930930ce81e76f9d1c81b05f8b52e4897ff9a5bc9627c4259ae79fe6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482198904Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/array_metadatas, Name: checkpoints/3776/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3776/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=235bf0058f3a5dcbb48db05a1620420ddd62729249abf4be3604bed17628a7d80f2dab80a3f8a86539ff926b5c62d2bfa35d61b3aa7cedd0e30c2d7572eac409caf0d2114d2849b6773dd5d5bbfad18ce9669a7282e24239f34c2a86d5fde6c1ce6b9ffbed3fd8fa9c2dd4721866adf1b8249f9f8b63d26e8fe853bea2ec3bb88826920f43bd6777b5385c4a2c680b7955d8cee627b7249131086f2d31d5e721709c7538b5972770d6d60f948e4297ac632ab6f98c48f80664bd62bee30d6a9a6946a8d4c669ca99e54ee5b9e07498e7037083af4f5610140cedd9110e81d56d1c226ee256e58468ab31d5eb9aa4fae08a2a3db99f289c2f76935d1a2e2e1d8c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482280474Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d, Name: checkpoints/1024/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1024/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=601320704c649b6d547f6f295fd898ad16f360778ba50c2fd2e50031c8990e4470d2a6d75a53b70b037c334e2ddb61f753c2c178025de8ee638c0f63e5fff70e83ce3a0dc76972a512ff8323bd19131dee91404cc53e279f923a53fdf30a4502a31f2a7defdcd0f5290ad7c7c65496ee269a95aafde7e3a666bff0f7b5292e56dd76f9199f23cc723dcaddccf40433f662aed91c393bb2d0dbbaa5aba0c093b556db1f0d48ff0c17c0b9ce4c60f114b7626e759300a01bd6b45e3b2d2e634739a448ad958e7ea320265faa6d091ec5ec25ece65b605577c4afcf778b62c97668940348a2d1d218cc12d4678f3e2d7b5de99c29994dd17a8dc69d94be0f2b3ab7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482321753Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/array_metadatas, Name: checkpoints/3520/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3520/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=81989e4ac35abfc7c47747275fd464012c6c0fb8eb8a933588321c9a679ad260714806cd0531661b065980f8af681afaf560da32e046ff673192800e23bb768279bfc0e3993ae42a31e6cab10c5b2bbea126db36e2ce67b61223fe56bc88140ac0616eef7d1d6f18e7f0dca7755751852949d76b6da953a004e4b5774cc02b5aa7538a1c30fea8d7adb808821bbf61c663cf7770a2008bf5e9b9c1485a021042a3948155d5f7157a1aae216dec74a2cba6d4b0615d8a2f207abb73078c3a142d6667bad9082f0f193a1257d84dc4383b65204f4ad8da6b927ebbb19b5e43b192020a7f755ea1e836fdd3486472d37c4c6fc48883137aa9f36e078993617e47b3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482028081Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default, Name: checkpoints/1600/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1600/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3b381b98e5d7fb66cbf7f926bc1378491d045a962a9594bf63a02de7096a826e60e5b116e23d756a9302fba411d055dcb09651464c881549e38f274dee241599671327195d33f93b247a42190a2bccdc538d7d0198e2242da40dd4cfa0de7783028b0017259d3f556542459d535260eba19c6a32d9db7e00022f41e127d7339f7faedda5e7cb2107d5b437fd0381b6004d3cccafd11c2aa9ec092a182e979acecafec4b25ad50209ea3b7225eb8dd8e98c8ad131145fc1c051cf8fe36f730a1eb9e18770949e7e3f4a12723a370b7417f6c58219a2eb409e2289d9dcd842c153c6faec687b28fb0eb24c30c81f1116f0439a7921edb041ced73404b3a2c4dcbc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482176162Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/array_metadatas, Name: checkpoints/5248/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5248/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=63b00a6aa009049ad5a78744b95fefacfd1cd4abcb7b36d300e5c1320a9c3aa975a67ec45740ac33379d79d1b6c8ae3e41c9ca6c13e4f92a2f15d3f58dc41ec1e49327c008da123e124c5b089ddfbea0cfdbd70d94f649482bbe7720038423ed5f80f812d2321a9bcdc4a959ffabcb047e8ddd0337ae96f28cff9442c6e8a2af25ff54dd7a1ca9c29d14fbea5ce8ea86460da1a592fee0080467c4e7198f4c3386bd893a7f2bb3fc115bf41778b2278f4c63c2e4d064e1aee86586c38da69d98f879bd2b9149a44ca81f6b7a513e83362ec982cf8ec135f43404a3068d8dae053203d46eca5d01efe4f4d1118237f4cbb800e9a08d49319d89adf5da91852a90&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482307966Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d, Name: checkpoints/2688/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2688/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a57a957a73079c479656acca981f5514cf3c8b5b7574eb807d8883ae8d9c4ff776f0fe484d5037b403a8f829e829aeceb3786c3559c6248ecffd80aba23e6a1850dc7af8f4aecc6c2d1b4a8b1efccc71f52cfd10a746769d7cff9f720365b4bafb9f4bc80c61be8fb4f52cc94f52e412c8427ab608f35f8257619934bf6b2cb6750a4edc83cc81d802cc8c8e9c897c752453a5ad11b05e2fd48d25a4561b3695bc603147248cd26a3c5cd169117413737c350ed693d8ce7b62278444af0eaaa419585aa03432ffbd0f73c3074a0b772c048b9512632688a9670c90d16c93b344e8ed98129c57a28d1eb8c15fdfe58e9f59fe00e1e2a47be636e8aeae27d77367&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482381986Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d, Name: checkpoints/3136/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3136/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=310ef023322389b44d680e0fddecd06e908a9b916c66d9a5543411219b228999b8a7dffe8c0eb3dd88baf696c87a7269de2f3d2f30013da20fc62b33d7c1109820545fc4004cec83cff2c5b49a2a71a7a33a0464e8852a8271648048c989f0b817a108ce3edf79895ca04a42973a4fe43657f1933fc365f9352252083e4fc0a7677fa8daef972da864fc9e27310f34c622286225d2f8cbbed81b3a1efb3d37a24b46ffe382d0d1db1c87622875d14acb1205cd39f8a24c2649252972ab66248b442bd3d7584c0a281dbe28751b07783c9ba149a251d2d8606faabcd93adbf9a6878ade2a08a6bef2eb7009b686a239ffcd8c3dfa1def0a1a3b9635f9a4a47cc0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482411025Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/d, Name: checkpoints/4032/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4032/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=66d7b46b743054a82bcbe199ca297e206d8166c0e045a1338fab2d85168d1498ff830c8bf72308b633472da4e105983f72ff8c917dd584ee232c0b9b135552bab785bee1dbadeae6dbe9e5b33f4ff14e1bfe90b5ee980830fa46a7dd7cb8fad0ed8039dc781c17001f6db39c7fe0756ce1719ace0ecddb291b728431aa95cb26598433df62b304bd5ed5aa22c617d8b38fb663dc3b5d1a5e5d24d7ec9c4ea19638d6d2c554178abab1b0f255bd5d65977d1d61982354a206f1f030f5132c5a9b41ca08a6550a519a75585c1d9d301c74950fbbee6bf46a532dcbdd193e3ee94cb7497c922074dbf6f999e72450411e828555dadac603d9bd6a2ec41afddd3ee9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482479744Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d, Name: checkpoints/5504/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5504/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b5ac2437bf9e8e6b8cb8079ef78b2efb79628912b1507a5270bb4a3e1df2f6f75fd7d500499a4d5bcf8f5f114bf1ef51bae70231e6923801f07cde2bebe8b59d606842784e8399c24e8d4013834cd905dc46710a8433069e28269e93371f72c5ed223b49c36fb01e1537e98fb9876eaf30bdb5185d2ff4918ecc2b14fd97df3c5b82d2377e47e1b7612b6bc77dfd1f25b85281a9dd2ce7d3546b097c4f13403b637b5558c58a2fb676f724e0fa3260d4b287f47eea007c39b9b0a748dd64e76e8b17877715483b1cd2eb8679d3b75bb6afeb455ebc4b133dc55380560d365f19331e29934ff38585076325cba31f50ab5ddff37f074255b1ec7becf3b643798b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482545576Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184, Name: checkpoints/5184, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5184?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7d232e9372564048e5ed26645157187f5e675aad48390a503f6007d5580fb59c08a4cb329fec945564cfcfae3811aa2f95ec975e102cfeeb750cefb4d0721e7e20f984dc7aeaf5d8405fb3cc9111317f2b9d0c3a32e364201c9fd59cb56aea87d7a4886fbb4834496a380af7680f1fabd759e4c98a6b4820ecb7d0fdf786c0cf34cec0ad84861f3b3ff2377df2d36b585c7fb9b931d631d37d1f876fb5676afb407e9fe3167f56fc341004d064b2bd42f81ae0924142991c43be34a69e4dca3a11295754ae4c55268923ba39231daf6f1b856922fe3e3f68e17707c2b31a7a190c0d4c37b1ed2a539e535fa79a1ce4577f06378f5ddce3e5972ee3ce3a43eda0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48225971Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0, Name: checkpoints/3008/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3008/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bf8a34997e49d004e9c38f6a841a88de400c9cce496a0b7fc67f9a877d497eda7408dee5e8d01ad6cdad125098f7134683d761db5e19739057bef9aafdbeecbb8b94bff76c5d4716d7b40646c100dc682bc44d5d16056a121e5c81918933de597b9764fa2ebc6d6e2c4e1a1d94ebcf6e4dd9f090a22c44f904331429aa798824ba1b58af44f0d9ebc21b106dfbc564623e08842c1ca6e43e9ee71cfb4861e87b18876a25dbba2b5b43523f758b00e9d626f5b95cf42415b5a57466079e62d877daf5b38bbfc587f20a3e3f7038f16c0c6fde2d49db101dd7641bdccc1dfaf4130fe3ba41a6c9b577870c33f86cde88e86caf7f720a612dbdbbce95af959acdfc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482591491Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0, Name: checkpoints/3264/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3264/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=95c81b27afe62e2d21e97b93fa0e7141816e4237d3c61225620a866ac7778e78a75a16bb806727849d2b53edd3d0800d753c315ac80d9c0ba26dd28a356f082604ee97f6cd86b5c297294d37cb69c4dc5a296419153e785f9c9840ae7631163799598072a18a1fac9f4c6f9bb8a4df0e2d2227d846440b2bc0c3d3c341d8302c7ba8e289e2c8e89f0cc5646ce80b9fd20d62c8196bc00207dd5f39af98b864a050c0c0eab4db45600dce5eded087b94fba6857d2c3e231c56d0f65ea2d14d1bf1d1051265bdbf6707a2b2908cd44fdd28eebd450eedbd5e34ad0e7dae9e6e32314cfdf5ed3983c378dcf818aa18c5283e86248349d2290f50eb1ec5a02302547&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482623017Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0, Name: checkpoints/3584/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3584/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8c9b07672d57c6ee9f67f64f2e37c89d15d81454c72d79ae41e055d764469b382485c27b8603e729f18df0e7f3cf1f5396b6ab074608c1c93e74e7cb54cbc4fc202791ca636257065e29849b517fa0422f24d4d2a23300700cd8f4876eda1c8e6a9bec9eb79daaacf7797099d6c71c3b8ef7af3b41f100b114ba5a043489bd7c413acc410d842914b2e679ff00285b86636f4273d02f8e6edabb0c54927fe8366d094161581ea7e12dd4d1e079867c75cfd38ed41846c267d2c393be92fdb9a594d0b66d9251f4aea04191efb619d6279f5e50af5f3c7848f7acb924a1f70710260f232b8ec1865dc2633817c2e475fb8cb0e86182824476d3363f5f2f401198&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482635555Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072, Name: checkpoints/3072, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3072?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6e6e16bc4c2282d4ce399e29510522ab9f568ba888af37e73b7fafea7f42db3e7d00c3e7ba90e268bd5409e29775e9681d520da9b6777ad2a983f6d2540504a78c5c74e6902c0315e92a0c6e9393ba1ab779b27a937fa4aa2189bf931b150e209b9b83989f875efb060de735b2269ce718d1a771ff1d70ea4def990d70d18e7f4341a19b43677f69ca884651619b1a08ed73440712684e0aa5cd83de55be6bb4316991da1a54ec43a0b2d53cca7d4764d2524d20d3e6d3c5f0da474a0c17a23d955ce15f0811cb7f15368ca4b66c8b4029b921bb36e305c8e9267b552279600cf75861a25a3ef18ed6fe2222596b4b33a76a76d4100de9d56eff35478c1bdc9b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482478466Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d, Name: checkpoints/4352/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4352/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8815229af51d1b1fdb05234c49b4afb3cbaeec8198829dfb5a21499414a9118ac52e35899a9e8b2d6784796a8fb92fbc3a693846d32fd4462b97fbee1f868ba59952a6ee2373d020e190cf7b3953d7df2b3b0d8aaf7c8d4752c7826ce1815ce8c7357d13dd18fa08feccbca944ea38ca547ae491b74927a12131cda79a7e74882f5e46229faa297ca723097ada5c264ce0d09356db140506fa79de58b453e06fdd4ad72baa302774500f0ef40980c05b9ebef9454faaaa9dc6a64b222a08e20f8cb64e658b59e4892effc28010dfd6f6dec7d4eddea0bf5bfd8df9fb7ced03343a1205ba8863fd0e739fcddd0a6b54a0d0df7663cee67a38814ae4421efe25cf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482682655Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default, Name: checkpoints/5248/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5248/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=49abf39e614fde5c131a496ebf1cb607f3b803956cb0950dbf76c4f59fee335d2eea4be662e7b314847fae9d76c8ae996fe2d1e3b90825733bbc4cfc10099df7fd43bdfead5fcc6f3388a0068618dc2d9a63b01aa534eddd6ce29bf08927a8d5ced7273ee136d70412d869bf0504c0e364617fbb9b22f27e4f63b8f33e21060e3a04b2067eb4b900dbd1d2fe2bf11c9924ecc1c1364db0816b548d1a11b896856fa29102e90310de3522e6a855e8b7cf3eeeb67b25621d98029c14b21040079757b4c9111c1c5689c5ce4069caa89f3b4f6a20fb3abe64eb28cfdace1a5cc536044eaa4e4c4391ae588d60fc55a30aff82c31858faa19e84bc5ae7010bb2d1e6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48268036Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/d, Name: checkpoints/3712/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3712/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=50b2a7a994efbaca5fa81afa85a9502db49fce680e6392c0639d37b3aa1cc6ca97dc424679bf5f07fd3a87975649813d12f2e654fabae63ccecfb5c93a716414cb2c8630e61889edf5b2405d7077a604fb0c64c91d50af68bb810422cf1075d757bfde2b74e2177154d324b386882ad02315da41c9dbfacf9fbd09f390fe2b4c65b5cfd6bb9f5a271a42b842c67ce7c4765be28f72f67384be621e8a74f17e67e486e1353b4b967c55f54a84650b3b64ce446a711bbc47567d1b5e3026f056abe2e7d19198fca63aa61f70cf314bf72ff9a6b9fc72732775cf184b2d7d7a633578acbbc637d7dbdd1b5ab22ce15b60d837d4c86586bbf2093e11579f80ea2400&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482705994Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0, Name: checkpoints/4224/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4224/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c3573b59d113aad393e8a01c077f3dc7595c9d1bebc272186c92d332f044fcbed84d584c3044709b4aaedf44dad040985c77368be5790c306c2329288c2a4e7d6bc124868aa313f4432948a4a86d2ffb93c3182d5d3d0302dac3cd1936e9fc90e8aa5c08816d41deb979eb53f30ffffa0fc69ac2b2914aa22fd48b9c7ae85b9382ef4f06911c0802431dfb0e213a9d7589737c08a7875e5728041478358bab07267ed0369a0e31f64a1f163ad5e9eecf3b8654f054f64172c9b0379b46a955b24df6983087e1192cabed458a14ea9c4b8b8dc1947d167f6e04654f509b4848b9e4765b7f44432c50403f3c495a60aea8ecea1b1365a699709deba07a6c612c07&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482766793Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d, Name: checkpoints/4864/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4864/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=040dd851456304354cb7695f685ce83bcd6737f567232f156e45c1e6ce7636922187d2674c289b847389d86fbd44c4ba22672abfa36a0f164b762f3a0933c66ef03caf256f9b58ee889cc51f47415e292ebd996c06a698a698d5d6b6f221f9f6df8d1b936688eb80012c18a3fe128866a4547e83a753eec50c6515b0e9dbcb120fcc24d014abec0e4af934c7e1d198f964bbdbc8b858da5598783537de9f53371ebb8abe1f760528437931f10159f9a412f95ed8639882dc6a3b9e491c3c8d5f36c8769ed296593b0f0e5f2815ccce2c312c6b0d823901adf8eb18fd8503164c65fd61674986c3f9e64b6838d80fe40c1e274dab7fdc5d0e97b1657113ae79b8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482791817Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0, Name: checkpoints/768/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/768/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=415e9ca0db5c8b4997e93dd924a9bdc0ccb7bc559a6a56018553c166430080e8e6f9b8aeeb3b18733becc175a6f149f2a18b3fd435e9ca275bad6077272a80eda900fb91bbfe8d53a826dc84a82120b311a4c0c94e1003085e610eea501be919b941b8fdf28aafa36c88a1c088fff609c2ff9d0239e08b467467c55b2be63cf2140c9503f1f902cac208f958f7d020467104fe0f04f2f1f24b67e13b7f06d20bcea40b28a14929a8daae8fab84c5f593825d050314d79c3bd30586e3d6f3031e841dadc2809e7be961ea92251af276a100a5f374b0bc8b170681876d867f3f7ba2678309100ff31b4a00e57406ccc2e0bc1bfe9d3f2c87eef23becbf9bfe2d32&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482750676Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0, Name: checkpoints/4928/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4928/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1fed027fc10835058fe41e7b2d243b87fe76128efcc7bd525663fd47ffcbebe22f52f21894178c1903b25cc220ace53035e2f5ac954f3a645f703b4f6a1a4b4e111bfce4efdcec30dab448d89d12eb4b0eed0bcbf919ee81fe3611e88612e5cc2ee6f334ffbb7c7b9cb62576dc1d74fe58f433d19bcfedee17a0701fd1ff66d9b59b15f6ef73c37fd9f1aa4190eb17d7416ff4b32fcfdc10607495e8061063380d97a81889138532a5e6a818a7e3c497308dd3a0cd3ad5e9df741eccb4dc65c4504e2a690750bdd82e8a95bf65bbffc3f1dc7d64b18e3faaf5c4082c18bbf5a255ad31e69763cf733da6542b5bdb7f0ba0fc3e5747f10a2c1e4a302d8c9a5b83&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482819712Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default, Name: checkpoints/896/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/896/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0477a806a86c7c32ba1205f92a8d25849c67dd7cc170828e16f6a0b30610d05391f5620d4370b28bcc5104a8208ac61398b26363b13a868029a4461b5dc1d0bb1be055bf0e06845033b98b98e06d7d9a844a25eccc202fd1a3dd09227d436c72a16feac585c7941c13ec92fde72a28beaab5e6c8b73cc120ecd161c6cca3e4cc7e6d6912f434a7dde5d3b91ba56ee88debac5ec3be074db836f07d6173eab9c05ed6147f4935234ca0c9addb3ff612b1f3ad928e6da6125f8a4a702575daef0b9115037a326a5ba9c6512fe5d030f33673ba21d094513f79c6b0429a2ef053437be908f951a14019161a802aef9a5046ff7c6765824bc811c4742532fac799ee&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.481642758Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688, Name: checkpoints/2688, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2688?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b573c967a236875a962855ea5317c1ce64f66f4aaef4b512d4c5be26b9866251add123dcb44ac694f963762d0ff8635ab488b847d4600853147e1ec2c7f4d24994606b769a30cfef2fef0fbe0dfc637f5cbebbbe48d16ab65c46b74a36206c58078cd4a55d66177cb3518f735c1df5fc778e7ce41ac9172b8bc05077fc715031de908bdd2439fffd74012b0ae6daacb655ca6aca386d23d431de0a8a8e3c3f62bbd8b27fbcfcf266a30ffe86d28b6bc1c179d3ae73926592527a0cec55c127aeb1df605d6984995f387c5844b8a3789c29b33953b7d725362c87df75a32579013b7f0ad4bd495b83ac1649a8c2b649e19a025f0129d8467091e4350f70623513&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48285509Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/array_metadatas, Name: checkpoints/1920/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1920/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=68e5abfb81378ea6dccdc00c18c65b840edae3c0d39b1e0a54bccb5a4009b1cdf202faebe8d7dea51b258a94731a9338821497a36616e740c0edf8cbb12af6c34fe016314f88928d0be996fa2a8c3a33e145787948905cb32be98177c44447c25662ee3a3f9390fd7fb8cc23fea17ac370162b02b3fd759d852ef029bc7ef7deb028b2f61b8442bf647df1da086bbc3105434b8bc255fee38245564d6b710a6070ecc0f50beb3d73a73ff4246150543ac4a1c5c8ce666886773060f71bd47a1f761010cb9af43026717aee3da0393a1a2fe2fd0a7c6a7814a2f7ebec4b99bbb7defc95603d8ca0bb672803d3ea2d9671fdd759d2dc1ac81496d69b8f6d21fabb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482745609Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d, Name: checkpoints/1152/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1152/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6bfba25803ffcb5482a33fd4269307a5cd24021ad2e9aa5c51c730dcb6a795d65a9b6fef908a22e8e439934715514529147de784057087812619b2905cc74180f19316337f8da7c90e573dd137789945cb10f72b8892623f4ab9bb5dc16bb7e82e1cb3263b4cfac52a50319edacc55a7d35c2a4a5f61924be25697106e2604084cda45bf6c35ee88c60b501b58c2b37ca0ea281cbf9705868584d47778698367b0b28b91e54ef5e56c74f02b77f7e5d3faa963304dd13c23849ba6267aa9b937fa727bb1eb779843c1c3ba84197a61437afd2c812b94cb1b905fd09746d4920e56192af70e85d47fa31c3b3c6d8e6ee2497db0dde22473a0f558882fa49b3c70&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482957163Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/d, Name: checkpoints/768/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/768/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a504a18b697139afcb5a3bd28d1234b965ac7fb85be5de23603d18f1b0700299f5f0c26bc1a9335ea90b779a2f96efb1f7da749033866198ac4aada03bf5459a240c5bf65be65313d2ae73ce8ab578ee8077381f2c98cf5ea6c9ee1cf4332170c6846adf867c0465ea5aeb0184d6ca7481d38b8f09c95d7d94a21e3eb762f7b491c442e7ca457876e7fd985215bf572dd655d3f3fde97a959df25f12d5086d7e72496d8b9c92b0d40615e4dd15bebfd416be82c1eeeedde13972530c51712e2d20f98d255d838e8cf6eb746e16a2acfabd717711c3fbc31df8c21e56eb922fb27ba76a8d6ea8d778ecd7bf75d6938199f40325ec74746cd1690cb957c98951c5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48295985Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d, Name: checkpoints/3328/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3328/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c34728af47674fc1361e8eed11f9efc19fef461ec31d92d1dd5a846803b30efb13863c201e0f8d981b00f1b5d83cbafac329f9ea982afbd91a565c2b0e07eebbe7642b2939bc21efa49fb79328e2681d1902f8e0b58977908b09063f86404dd093b311542f85615dc13cf6852162a1d8fb32396df365c2a77e9d3ac67caea67943630dd5b5462310cec27ec10e20e3aadeaf0b685c9be12a18d7f96d6a17afcff7fb177e6ec553e038a2c276707eb6eb36f1ee94cb8b028da2d1ef34550b5dfef16f66eb639b0b0da78e330aa0b198c4b4f08358bcf9be8fe6b9841a5929b89ccd88b14c97a6b110591c71ed90f9a1df4b3c2e5d16c93f65678bcb53c726a8c5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483014657Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default, Name: checkpoints/3456/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3456/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0542c9fa519e8915d840d6d36091a78bf1ab22dc1d38e941d6ed9e8c0fcb33d204b7585bb4ac10248b2e030c74d9164bb1fae07b23f2d21188661d9c02c7aff55d0aee065707d94a970b857bae57cb5a11f1f827f9abbfc9fc2f2f68a912c3b9989e46b934eb80e4df41c54bb36e9c4a8fdeaf718b15f1766f6332dcb899258bbcf60a1f52fd1fbd565ac717521531bc9566dec70a5d3a6b3ce66a990b9ac2bae9a258dae074b2ea949601dea1023e48a4e30482a38a7849bc8801c549f61afb43707df69069dda9a5c083b80862902224757f008d7963aea037edba9629755721de710f075f62631a8272416bf391c36e13c13cf0280c26698e977a3249e757&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482893014Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default, Name: checkpoints/2752/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2752/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=647e97eb05bb715492c6c8d2c2b45f67ce137007a2b5f14b02b3b5366226220d3cef583f7e966ba987d169e6aee980b5ca15cadb6aabd074420a6e069147156c9092a5e3abc2ecd22d2680881d14291093e4460f185d56bb9481b99cbdfb5e21494eff4209ef340a7918941c01d4ff1b6664946a5f8ab993c6cf125667dfe65ae993483826aeaca0dcd7101f1000b1c0d37b3e74b9392a3cecc3cee4f39097022acb6a7f7806f767ab6e02bdeb01111bbeacfefc515f90dd6801d17da570b11d9f33ec9b6653b56b22649e6e13f7a4045dbcd569fefb36e522ad52ac0930cc90020a466d3b851b3610fb5c0b890b13dd40c22d220b9a787d5feba4bbbbe72e74&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483062222Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0, Name: checkpoints/4416/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4416/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=70aa7ec0bfcf60aeb7b11238c658fbda034923cd14c88b5a72778bf7b65947995451d84a7d1db6fdbe6ae5d9d2d8f46ce7a6995f47d9177eda1b5e7c4776d10a2ac62d4315f1fcfdf874a8c271503abfb0f1b5a751e423f3d3bf10ce7781fcc6345d812b6ae6604ddb77c078d8dac3cee57abb823c5f3de11b201ea32914acefb0e90482f0051e03366e25b058072b18aba5d710f3213e362057d49e55809e32bd9c1e8462a21c7277cb8e20e255d76bc8dd59a9f9c7fe23cc70c9dfbc67111c23ab841d7f70cd66896d1fa988234053caf95357d60d7a6f8f60362803d62086112590b902f6bd0d432eacca8a544bafea56458a26b140c28cba8d5f362e76a2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483101068Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/d, Name: checkpoints/2240/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2240/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a7f76f98b9f7a61068a66a5cbc59ba9c035246e2ed9c8b97f5cd295cbdcdb76a8866748d9e4dfe5119d1bcc8bfc3f509c62819b7e36ada5dbf52677533cff9ce7e174e174302ea3e172d471109d6ccbeb68d5a499623a20bcf7d38b9e2c532d63575c154a09fd7c5bb54112e15f92dcc1b50249b7b0edf654d2a9dcd50afcbef4fd610e194981676dc4e98375d7557d3882c2a12a741795a5c6e682e63d1610182b9f86fcc2eabb7a367b1d62c813f1ff1627c660e817522e096805396c0f8dfbb59c2ea067a804e81e5a8e8b90633e95dbae0abefcd0bb211a9345c5bb24871ac9b2dc20bb799dfc7f721ba1d5c3dec2e7b3980cf128e5dd9dd68da340a852f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483375514Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/d, Name: checkpoints/3904/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3904/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=46c75289d791d5a48710e7d2e9b1142853f36b71d94707d69b8ca05275ecc676113f6cd4afa2738c78509a6ac84dff1bfd8ea93bf933bedd4db151fde2a574fa80755c4e422237c06e4a7f34fd3e5b40a78f5a00d5d0664c54f4bc27559bab103d62b1fd8b52893b939d17b29e9fa976c7d67c9ed51ce7f8254c52127b17441ca9843cf3dca4c3c3822b412e75b3f603bf27b7b23f36e32f60263a08a7f459473b39e0efd2d1eb10ef466f78cd5490e9664d4b13e2cce4ac7c021dabeccd9b96b1f100cdec433a463d5dedadd3249f5f83513c0970266d65966eb29eea617dcd74e067ed285a848d00b154e4d53d35c8d596a636f4d4d8a39af7410a02940bdd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483456175Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0, Name: checkpoints/1408/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1408/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3a65d6fc14c57c15e5c3c0ff17a38fdb3bc66d81e10d39e3998eb9ff74e7f5dfb3bccf918178d9cb79cb60b86de4f94ebd97d2e7a3fa662210eac1e7860282a37396beeab227f2bd703ba8aa3c5dc1ddedd4aab66e2c9758f3c3df9560ca3237d2f3483ed590aff9f1e0a5f3ee5caa08df1e300301172c5d211d84e564c859354ce2943c44931ff6db47959896578171aefaeb33cb0a7f83ecd215c16b8159494fcda003f9e13b06b52b808e1b7f3f47ce4cccff385929121961984c6a94578ebb9c8dff9ac48cff175e14e40841234b51c14e0ddab59db2580b77524fc9254ac04bff4ee016c1cadd2aad29fc3f39dd15a9f1fccc1fe24cfe7bb8a4b109bce1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483438768Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/d, Name: checkpoints/2048/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2048/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3c64bfbbfc6bc0be40113de1152516c21d4b8ea1b61fb31df1f79d36061c6204be4f5201c74eb1fb211b2313b0f46d90c99286ef799204d2594278b7d3a6e672df071bbd70165739be274639a6e4752c995278907aba4f2651db5ec5e49c5199c9841990c0c3245c0fbbf35009303cad2419ae4a3cb1f9bc22c7e5ce12bb112c1780742f0776da21e15fe2f8710927f666d09506e472a269e1290e7fee681bca2dc11d232675c6d5ccfcd1e6bfd1970bd6dfc747adb2432495c094952c764f82867abbaf52c1194c68369c09d3ccc5a364d7db8272c5d64603b03f12337d86b12ecba181458eb7316c9a8f842b0bc564c2af17c7bca115fa23d23b4cda561804&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483518754Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default, Name: checkpoints/5824/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5824/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0c5772b4cf6f47d3fa1c657f463adbf86ae697d8b76c2ec031a1202a2026b11f11ad17aa82feb26869b93112963baac3ff2c599fdebd30581afcc1a27e9d723273d2496bb60935db30cf760121fff4e084465c0d58ca498aded167f48f57ecd7baf276668f6c25382d2105b19680a1f4324ae2a2a865ea3e8d828892cdf1af8812b10b472a9eabf53d0188bf737aaa20a3c239205a5e28944dcb117e56141653abd9a44c8af352ea3b7474f5f3efd488e074fafa85187f600a8fdf5bfe41e662e55ac13d95a7c34e42f67a0dfb1a614f9c3bc0e7755e00aee598aa88f89eb2f08f624acd24590f63077d82f015f0efa76e899335556b64d0204a1aee96d17c11&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482307099Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0, Name: checkpoints/3648/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3648/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=44cbe7fdd4e58b7278eb7bffe65a1f1d9cd5a9c3b3872da1195993d33d00dd3e30e6de8d443538ea4ddbe3d7d51e8c08ba6c3e48b5c0ffec3380b8f261dbd31ef9c4b510da2308e76aba8a0f97492e6a7c7f005ec845dadea4191c3e74d6a8e7cecd93c39bc918098e08d3bed6dee1b0075697b333039b16f8d87a39b23cbb10e7de175c1fa9991c4e530045a4fcab0cf9d50a9400d3939dd89298d7b62b66f49a807ed66f04986d182c4997ce667cb154ae3e33082f5e896b01b46561977b67ef4da7c3a5a4bc866aa790a0e1daf90e27e4d33189c941341f1ee0ea1811311039dc3cd860ebb7ba0cc7622a966208f9da2b4cc76f57ac726ba92af241a686ee&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483533902Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/d, Name: checkpoints/832/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/832/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=ae992135c502a80744cf9b2fa17995411d04afa5b322a24f04e9267619abe48d96075a09c2c8f0c8d99123d62ec6b3a19ced9a76b5467e9a75549325e7fc0873323f81ce16f6ae46187fab368bd88ccc829e2436816ab3c28110ce98f8dcb4e651c48a0b84b13dc5d61b622ee34160628e34e2a18bc30f1af94665811946653052813db52b7fcf3bad594383cdb531335f1eaf192adb15ca0720347e773d72348d141a2affed050f6ac2b4a3bf594b9b53419944ed23ec25b2f2917042671e1318834a4f2bc6a3cd4706387835c49b9a6095395848bcdf399b8b7ed3340490a9ab14a3e2dd2ccf15f3fd2b29bd352be7cb791c865dda7b8aed16cfad2f65be90&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483480596Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0, Name: checkpoints/5312/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5312/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c3f02231380b1aa20952878b7477dc0b63bc3bf4cd190be132e4dd56c1ce88c976abdb78a561a4c1d31758bf8ab511135862dbb0e2227dea9fac9cc563a342ed5d595a170e13bd27d4eb7d2bdf0c84c4197278af68ee4895e3ca6d3a85ed47e51a287e943fc164c91367868a8a7875a54834eb787f37d6d3c7a082acb29d5a777e34f161369a685fb2b8fc98c993552b889e3490995539ac67c99aae5b89f501bad41736fd39433b2eaa47bc4eee09e7ba7451c4e419378fa86ed6f9487ac7909f613d2389cdd91742e915288724ddb0ff0792a9da9ae373f730fd11f2422ad23afe2a7a059d708b117a8705cc2742c4c48c800570de3072bd3258d45375a454&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483357967Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/d, Name: checkpoints/1344/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1344/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1274f57cb61806446db838fc5c2886bdd25af4e64aea01a1c5f221d97c88da896ad888159705ed60e92d8debed14e66ea3641db39e4c288bd6a615df6ec296a478b7b599dc851cdebeb950eb9b21c2391382582fdfd3e5a7ba5b2a09c15582b20993d0bc98e889e5af8476fc8a324343f983624540d1e3e29bef3f9f1d2d1a28e9e3d560f4614fac86bbc0d094bdbfd59344d0d1c1de0c5687b2d1c4ae8a25f73d978f5730499695cbb1615c64b1711b8c7f87851c09d0f62b59ae305dc206dc362dc17b26c8f84c89f751537a9b7c05c40c94736c964c8b1baebbe68130d9cdd13e9b097b5cd8c82e3e9d3cd899f569c79896f670440f1d04593679962bc7a1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483624088Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0, Name: checkpoints/1856/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1856/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=621ceb26d82995cc888e25aff7f854cae85732963b5c58584340df2ad9746688da56bb39b8bc79b121c39b07ab02aaf71f1a741169b2ae01c6ed1888acc633116acf2c2e510d1aecaa76fe7ba367359adef3aa2a07d373d7fe23076462af51622e2d60ef6ef42e6a6a8c37a4bf2df0d3b8989b419f3856e651d8af15c9c818d7324d1700b4aa274b9a090b0419ae53c3062d73579fad4a9187b0be680ccd0155b37b6121186e1ce295b1611df20d8aeb37ba6cb12db01252ad1b8fd32bd0074f71e1f28135f4a0403f5ca9144e9b3419016600929e5af8f75a15c5827add79bae84e75db26715c37337ca5b98e35e65134951fd11d01b277df0c90bb92ebf536&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482316154Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/array_metadatas, Name: checkpoints/2304/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2304/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9b25a0d6b1dcbc62eac23986be31167f89510a030ab8ae3b6f0c34cc0e5b506bdc091e5ab39738ae173a440cf14980b173f307206d9aa1e7c9e5043dab79159435504ac314d2f9bcfcf190154f7c015f7986ca91d57bd2a169a3855d7a4424d9aa39aae31af00770bdce826a42d0d5c7215af9566da13486d976ef42bc1db78da0ccc12d6c14c6fd93a39cc76aab7988e4fbbc5d4883f844dec4e8a0afd971812b2aef2999a95338e57e16eccc3660a1140fffa5a1836d2a6e67eecf86c2557bc9a542508f35e584067c818c61b0c637a444429b3ecf11fe3ab8dc21e707058e8ff2be610dcce2497fb2e5596fe2f90139ed795bf391cef8a2414986220bff96&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483458083Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/array_metadatas, Name: checkpoints/4544/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4544/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=77004d39c91a2caaabadb76dd68887bfd8c37f80e55f4f0eafeb17aad1c7bc498ff0a8c1027f992e77190414ad946ef4b55fb146f247f5d59929e78b26eb517e63e0b5d817ac5abbdd252700e2c7e2b8cdadf28aa3e04400c911dd73f386a492476d306aa57b6ad2fe3c40b992083793edb76d280e110ded3b112d2f0c77c595af561471e92c9d402d07def3244cd89c141a855e9f0a7af9ce3162114821124b673684f31c0b6815b5ad3bb4ed6addad587d601e1643f92eeca6fe60637bdb9982d782f4309769db08d617d4c222c97a3ca95e8fe26b48e379dda2fa5031dcb00baf6f602d1915243c25a6fee70e812a71b17d5f49c7b8da2757a287e76131a5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482739544Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d, Name: checkpoints/1792/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1792/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=107a590e9607b0588874336ec75ff15fd4e80a8766c18e70d2225c8f37687534780be0ab37f150996d0c10b0c7e3f3a028a46c74bc98ce4244a7b9703b28a41afd0ff5f494ca057e0d44d8d48c7dff281c7c7ef920728f72401abeb18627b0c25a9824be45b1957ca3a3ce4d552fcd4b487433d5869b83ff5899a335df4012e471b0b48d9d215b0fdd7a1b9a29eb1e2aa1dc4749ea308b156679e4e3e4676f8bc6cf681f6361c52fa6582ba9b8ac9b4e2cede5bcab5223afd4cb21307e25773b96c79eff8f4fbc3ef54d3f9b05ba2f02f80510adfc2e927ddf9b68b97dc5f04c8d53c453f403bac52e1c3fa6bff015250457a9633e59515b5e69e7f962733c57&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482392423Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704, Name: checkpoints/704, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/704?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6eb9adfa61a51a0abe53adefb5c9d74b1e69d9df935ffe444ce09f587ca8eb9f0ed21fba307ff8e2b3e2b6bd93b5b159e03a2e906b891e12eba2ae5e6e1eb7ce6d8b582647ce747b862e5073cc3bcb1ad0ce5ad87abee7b6d4a3f07046dbffa7a796cf1beb4ff7986be6164246a220bae50662b17cbd72bbd3d5108681a54a576e0b5943e7841b32fe4fa259e164f98d904b10e72aa7fad1e6e7c24310dd049f5d01204219219ac831af5a8802a1d15de6862d80ad261a84d204d43961e081f5ffb9822531a1269a8c66fadec4628de4da1ca1c301585c6e0a7b80e516faf9158b598a8f04c8407addd76cec8bc92857dbeb0ee24522ae0e89b5468f3fd850ac&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483511273Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d, Name: checkpoints/4288/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4288/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6706f3a8f9dcd23edde962e8c893ab427745f76823a7c2e6237539ac2b742f6580dc108e5b0bf316069c26504ec2dd3511979e0db7c7dbb740a7b39a6a4d7670e711fa84f095fd5cc0caed0342713355be9668b3804571333b155351a174cdad798cfaffec0b9c74f6d2beb6b996986d559015110e467df43bd5b2af0500815fb9a6b7296d4518d9675b455024b392f03eec3a8666d2cc6ed84358f7214507085ae8fa088f4914b2c0a388de0867888d12baa70fa408e3c7b4a746a602ba7d3baa12a3c37f2d3bcb2b5dec1d2e57e20a94672766e0872828e6ed631dc8de58c0d0a746f767896e15fb8bf78ff27a7adf7a2a1e97cbb3273ac34bdd6140a469a7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48248263Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0, Name: checkpoints/1024/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1024/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=831c716a3d6b4484e261b3fc71563c4b8bd689fc727c7618174c72d4898095a2e37e69876f3bd0248fcd1f2275ef28253f4900656179223876ef0d817391669f16b8b7a5830f9d28d2553f29ed90391cf485045deda395a71e0d960dfa56b64d439d50e6333238d8152a88d12f074473b6a39d397636ce8aa942f9607fa473c8889c66508ab607ff8a3102c7880928bafbf9249dfcd6b39d6aec564cd1dd1268989ba01bc5aaa1989263208d4553e62b3bb58ab690cff20cf97e778df95312ee77c57a45a79a50dd00ef3c2b28364740cc169f3ca8a91927f2798cb767003f0655dc58e9adf6c408362aae7f7b72c184fb5d4d8b539d53db08db855418f9365f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483512204Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/array_metadatas, Name: checkpoints/4864/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4864/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7920bf4d83508ab5d16aefe5ce6b4689580fa20bef7eaf5f63b1aba8b34fa15a0508a4d78c80bc24b0a822a476339692c92fe9a22c535aa12133ceddfb65a8f57c86a19a0326899947b3cf406114aee63bb2dbe958812c1a2fe5c279c27ceffc8d7897849e83cfd547a9235df81c74ae66964965c38bc9482184d042907f67221eec8918819241cc75f4e1fb84d7bd01599c17639937c48def9e8a7bfc0b6969eb3d0568ed647e40d07d7da7863e15351412d70c63b9c466c171a18edd8d0a4b439e41bc1f25d8f1681702b01aaddc121bfabe789b50eb5c804f21ce30e2a409669be7608d1dadec980e28af49daf7e7d065932d205934a90a73ca23a6475518&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482958664Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/array_metadatas, Name: checkpoints/3840/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3840/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c072a9b62c2931316a55cdcd961ce65f60e71c2e58c7bb79abe22a9b9e0c24c15997183e6a75226d6ab15ba5753458f1c7e8436e64539aa38528bbb7f7b8b05a29ca0266a8c2990617902ddea5745e6ad0f66a135590960c9f31f5bcf5658858c556c9ee26ccddee5ec3f421a94b8eec39960fbe80105fa20b1fecc0a202f7cd0da4826fe76c233d994c4126b9e4ab4e359c98211a837ba35a08b6c441f2f59a5654c3b2edcb8b221822626c232ccbabb7c1e21110ff14832738274a0e9b3e4fcf051ecfe39258993e57ded8c2bbb64d85a42e59a9d8e14b694e3b7b925faca248f1bc9432edc24b5b77e694b1935402d949f5ba89fa72f10475a0ae08589846&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483032331Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/array_metadatas, Name: checkpoints/3072/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3072/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8777e4da708525808af7a644f64d7d81f475f82e64772f8a8936bb451d7532b26fbe9bf4c64788ddbb97a244ca14104291ec50cd7d463625d9b993dbc92d343e0b2859d3e18a33298529ab89bf4d90958dde09c22cf51872d5a82fd738fad72fd45dcda42c17749e41c02ec6ff44415924debd972d2512da03552c1caa74efc1f71b721c7fbd6f884e29e099e168cd696e2fdc41e76e505e411746a0a4c5ab916529284ee0dc5c015385b618d846830ae8f9558757c24c5fefc1dd51111318b5e6f17c2b4a9bfb7cab6a2aac4862efee14948be41f410de2a545f368fdf3b02f618a301d80bfe2e1616cc061886cf1bd29c9e7ee50e9c3aff4f2f1ec1e7108c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483950743Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0, Name: checkpoints/4032/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4032/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8e1a1a2480537d7e20597f44d52dae2dd52fef2eb9458b707db091201fe226eb5d396682135bee12b6a94641efab4c70745d31a5c169b2fcf48f519c80611e5ae201ba77258e8bcdbc2931d9ea388ee609c52402bd3c0e8a383d7893221e87e34b909f527bf30e4151915316967565d169298e8a3613d210881066d442b9c775dc8f8f265b93eb7aaefe7b7953c92c027831b5462f7c547dca9a8a8e3cee4a4c8137b70a997b1c522c6642fa1dbe961c42a499a595c8d656ae1257e48c2343acf5ca243510975a037a1bd06e695a455fa1c0bfa093d9fc22644bfe06ac1194e1d7a757b9bdd9e1bf9c8eb7c57067eaf8eed7db72a68374765c87a75e6fef2ef0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48344972Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/array_metadatas, Name: checkpoints/896/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/896/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b9823271e10ca387ad289815a76c0307aa07f6729185c511a2497da0b5d720f329f908c1211f4dcb71563fab466535780aeb4557d0cb56fb1406c8f3fe3ca7216f87fe496a198068b7585be4008bb1102fe0550814cfbad14b95959861d60d70c44d14ce5fb2e685543033f900e0834dfb98b3a02de57d86e4eee60f6de8c2cb19663510048fbbd79e5215a0b18842fb4549d5a7d47833f08c8f186f96e547f2931353938b207373671f02f6c7237be3b3c6930e70cc3a12fb9b9b46e3ee7e94ff0b26942a500e48f790a14b4cb92cc5b506ce7943a957afd2da4e55c28a8103ac9b089bf18fd46149a9be0af674a2a5dda984f35e5d2d1757b90463ed7ce554&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.484015049Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/d, Name: checkpoints/2368/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2368/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6838207b65e31b0a1b5d2ba5e14b3d9ee7f0d25b31910d9f1acd67cb07abe6b640183b4f8179a2e8d9f98fc0228f3721b25da2c0a362a4e689e58db17b1840ff0f5e0c2e635e533f26f8b40dd283f71b22662931ef3f6dac614389583ebb3df489ed2613943a7ed874e86646df4139b26fce144d88fe2c22fa0499d16f52df72c6f40361c5898c4a1e49f7b9258a6987e9ff5fadc90f059b4e8e7c4fa9f0fb0eadc5fdf156f21ed7effadb65d713b39f87ec42bf2f7d23c06497ad6e67687e70375481f15295438979798cb134713c14f6dfb29dca56d89d6af06264e4a4aaa2a96768dfb345103855fb8ee99de3b559942e3e7b73293a149387bf6eab87c821&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48213258Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/array_metadatas, Name: checkpoints/1856/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1856/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b9bb52eacdc0ee9f57231c1b0f1fd8a6718a53e82e928f9bee62d4a6a55d24e0ca366b7637eb87450672543ee296a327b7b6d39d9261020f37160d5d7ded0ee0ce72196c4788ff1983692847e28c7d9e676fa83e7ab698025f307f5bf72dfa53058827cd1c8a55256873c6bbcdb7120975dae12f97f1366aaf5ee6c858368814b678252f9f49268c691014fa3a9d8c392769e408b692889fe7ceb6267c228b5a91884691c858c4ff95e5d8579569cc60f46f061226ab33fe7db65eed5d105506fa32384df07731a97d01f2794fb14bd5a985b93556168e433e7bcfbc690fc4cf441602644aae271aba2d51df96cffe59c47c0e6ed2a5641981646df54a7115fa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.484856776Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d, Name: checkpoints/3520/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3520/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9552a636abf8a141a0df42aa68e3bdf5c814cb6ed5bf8ecb5d8264277f94ca7c13705f5a2255fc7406817a7eea5897ddfb9c7f33a5563feee84707f00b29e01fdbc82adf21fea236a9dbe5446433ad0f1f6af9f4f356e0e66a75c683370a19ac5d1c1d1cbd3b2fd75e4085d24c9ddab0a9efc55b93d0add8d7b56ac68d56e6c98e9c8cfe1f083e67f1c9400ca4b60c5c9e3b875553ca81df4f3ab23f935e454a81be3311ec4bd991ba6c5da5d6d11999af530f516fca27c83322a4783f4104fd5a50c89e754cb8ec3114e9c656edb9619190e44a56ea69397b128886ef38be884dafc9a9e3d840f7f7a2798d7e141f2cad95e4552bdae7538de2e12a45d88460&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483492281Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d, Name: checkpoints/2560/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2560/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6e1f02853d7a61f1e0b90c0cbb21a5f0f78f958f4dcc37a9b55e3b7c71fbc5dbcb46c1c7e70456f2c1e9ba6431e5d9309c40542b4b4967c6e0921570f0da4d68e9233ea4898367bdc2a4ca259ee9beb65c193c9666928c8236fff3a394e3755d37618583f6f5c24de4c6e2c6feb36dd56d1e39525772e35be2224b57abba5b552c8a90818355c22d879d0fe3b3de806ed8c7e89ae3d97b26dae6ee4420c33633befa7a066da64dcf486a3f3262cca533f01b5956b563c024a33b2b02a47954922d93e8fbde78242808b89c8958564113035bd00ace0491915358dbd7086939f682b34dff25b15ad642e28ac69920c8c7e96ef3f308172449253f1d468d869b40&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.484013555Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384, Name: checkpoints/384, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/384?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a00d620f97c1e2e11bdf2bb19053b3d9896831549de1b9d185a0e2ee813d93e287c69bcbd8c8c14bffd35e31eab59a2c779ab3ade41458e4d15faeb8aa28dff8f747c7abd6654e88f7429b1cdbb36b16dd40005c7742411db4dc046591632cec9af9293056852adcb24394bb004ab30dbd11482a3470b4c495759fd828bb254c72fa7498eb991505b4483b9f730a082c1d6bd680bb72b3a65472c802be224497092a1eb97134e7e42cad42082a0d332d3728f6545c36c93979980c44e2be1b16e8f624a79f92f02daa89bdb6180eb193b20dce68d881e875f8b76e6e7d13f0799976df4e8e310b4bcf4dca2b35fcbe632693f1825d70137b984e70525d77e3f8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482816132Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/array_metadatas, Name: checkpoints/4288/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4288/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=465d5b2ce73dca61e851b62a2adc205f550e9fe57f77d9187618ff36fc9f7ad3bbbc12e967dd8f90d804ed8f00b51c3a8c31f54178e2643411e69b60a24d022d9763e18a2c647f5752b555e5aa197806c743d754e3fd4d22967331e26117521353903b4a82ea31cff4509a94612feb81ac8bdfdc460b4bbf024ae7711180963b1ec11663c3e57aa6af4de90a2df84f679087a0556994ce5cefac82ba1cf2eb11d5c82d2d2fbb4ee38a5e79a0cd9fb1fb85781bb75fb7c60f651a2703cdba47513e7b68b9fbeb481f4e160ebfeb931d9d9125c29265a487f9e37303cc8e575703eef7d7e590306da83c131c317a9617d716ae35843bd4546b66f8a97259ae5ec0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485127025Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568, Name: checkpoints/5568, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5568?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=ba8f393182018f87e104030e1bad89c948305cc0beacec5e61fe1a55f8741527f9768196c7db9962ae2b8f4e66e4839ee936eabd8c2935d96cd87469ee6f8606bdaff043011173838d9aebb71c2af6870357097d1b7eb5ff33cb726e5c0c84a9b844af62ff7e1ad4dea7aed9c9fb15790ce349b0f385c77487d25da0f4fe2bf56df949f2067a3e0bf10cd4955e8b66530a795e4eb99f3057b3595b29f7990f29b64c3e7e0b6ee0d380ffeac116cbf4aa20ee81377a8e3fe131a0ea93cff0fad6dc8d6e8929465d7489b09cad2b785492a009e0be595cc1c8019a10f18cf12f1f528b1c257c262b9f0f68626c4bff2ba97dfa5e9f7716e2510478e0e8583081d6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482534399Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/array_metadatas, Name: checkpoints/320/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/320/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3d832725741a2f0ff829d8431c5d2dbeedd63daff8d9d0e5501d9c0223eebc33d0c26827709cc88d7c9e5b115be240d11bf66cf3fd05cad40e1ecc9299c048ad8e82d4034d4caf1faf243d9f674b925555540cd57865bb5c7fa4c05e17a58e6d1670f424ac4d6f078873fec1111ecbe0fea87c300d03f431ba9f4dfbe048bc56b4bdfd50592894b4a4fdd93bb1a2239a7c5d4061c81cd226076fe26bcd8bbf4d5fd76f294e83352eae14dc93926c599b1baf39ff3ea60ccffc624c565a650db04e1233d7470d9cb850835c19e6636da21e146923e05c1025cbbcfd7de8e041a0af29b5ffc2001ea092bead7b8d11ce6d6e8833cc04983e2f4d16255d777a0732&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482702216Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default, Name: checkpoints/5632/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5632/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=52218e3d439986ea961b48cfa222703e086a8e4ba22429a88ac10ff8bccc9d896e5379e1ec43967b8c6e8ffb086dfdf84ce5b3745cc774bde3fca60899130fbed13ece81ba0aac59ca6b35a1d083f7442424d0faec9b82b643d352922e854facb923dfafa51a93832a5223d9d9cd872ae44cedad240c994918a51f81ff7dcc3774144e077fee3c6babc435298faabad0f9c406a95b917e6cce3e8edc5a6e34327e596151ef0c0e3d20193b45a3253e99b292481efe40d2e7f4cf82eb921080ded0d9b0ac9beca2d3b47b890c7b362389c6548ba412c210ff803091092d8d4db38d490d3c2e7546a6ad5f21a040f70f1eb315cfee9f3358ee428acb7c75493370&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482642889Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/array_metadatas, Name: checkpoints/3904/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3904/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1b86ac488844ced758f7cce00911ba42e4db0639a98340c4536a87ce6c62a58fd5226b101e896c81ef2139b5e5faad5a991afb93670685b8c0fe7c82c6ac98ed2419c86c00bf0aa931d83d2425a2abeea88490fa32359c9c315884b82ef7bdc9856bf5904716a1de57502c1f25dacfee5df4ca6f160246af20d6b62478308fac98d3fcbf548f3b4527c9a14b0b826fcfc77c16d8799bdfcf21c553ccb27ff74c415fa7885dbfd6af37f105cdeb43fa73dba286e45960046ce4a264f942b1143f1446f43cd1857fc9038249bd1ba219460b91533aba2bea83c3d8bfc2ca7cf973e2c863b65c1800822c5abcde5f12776dd99a17f4a652107c7728f776782fd65e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482217406Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432, Name: checkpoints/2432, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2432?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b9bd72284fd084fa1fd2bd1de538cb6e7d78cbefd04d0b57dfa89e71f0b86c72fad56a4df5a807845830e1bbf7db64a8e10dd6fab320e5c827fc7bab0e38d41a82bf0e6aaa8c3889ccf07c1d2eb5889b88a102dd89a9578c070de5ac9f465780539c1619b4ab7c5897710740d5c6d0e350080b549a65c456450c1b9830aeb51820d9d65d2188801505ba0153a72369d75ff2a5f7da003d0b45a41315ab15deda0046277b34d262949a5ba5fc2a3e80f7d55d98930d7181618b2d0801233a97a5cc5566c350317612c85406f60240c5c690096544c44b6fd6f3928e25e76db7b4b4734495abed8eec025cd16e6705da90c341431098040625d6d2bddf7172163f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48215218Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280, Name: checkpoints/1280, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1280?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2db2c1a251ae9b2b2740385f7ff246e1ac0b6aaa0f5c9f156719a46ccbe4216d3ab1569ba0cbb32b935f7915727a7f7fa7a5f41f96976ddbcf01cc2e5d4da3a408aea30f72e23f0e7509f7778596e913fd474500dd2180e6037f879913429f7289cca367e102814a6e4152685f492b7d8fc410d18fe08854b8520e7e261be79a4eac6298b12fd7feb34a19bd85c64097dd11d38e016bd74d639ed0e4de355f70007c47c35ae545d39d9a7e1e3f6edd2374e0aa218362e129b9f8f6a402f824b47b68ad8b267b1ea0986b1426fb64f29793eeeb5c40e0dea81cb1500938631425297fd87eb481e956df6fd7533ba5fcf48f2e8ca04b211d79d073263c6be8121a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482090447Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/array_metadatas, Name: checkpoints/1344/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1344/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c24af34cf4db77c2307614532634560cb92bb5f7a9aafa3119574018896ea01e46edb995d9aaf1c5da1adcc7b8da322bd4cd3cce2973024f01470c5cecc533fe3f8a6bcecf027973b968898d3ea7806df8a99f1df23933158ebe07dba9756bf06fc11bff198484e530f5af27e12a5dec1331ab33de0f3751512423384739ddaabfa7cf1fe756dcf6751f18719536e8d2b8a0574c2dd4d0ac7fb1644b26a94ba38b90b2cfc5a9d3cc172d763ce80d26ce48ad714b4e8b549530a16cd419ad1898a48ced96c3b6fc88126c31967a2159c8c3715c467041ff29792727b63408b00838c977dce6fd79c0728ece4d0e092b4ec369ea106e8d7ae9ec089d09a140896c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482603566Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824, Name: checkpoints/5824, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5824?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b98f25a9407ad233276d6970865e643ecc0f0bd08c7325596fe7d45428c2d3ba97a1b994be75f788c1eae813ac972c6d48db859af83da0d44becd3335ce2282f76dde5ab0ad65599a78670142dca0606ca5b3e3c80f0b6d4f780fc568b1e64bdded430ce203ce13abd650e130fac2fdca0093207aaad8c87300cae8f2c3cb8885f34eca7af539ac17d3dc7fd89bfc22ced56ddb64756a974ad369b4e4bcafb1bd164ed14cfd242ed8e7b3efe064b93ef2167f1f0b64f226babf7b978af344382de6efafac1d1f08cc57aee729888c683232b232251143a2ea4fd84c8fd2b94cd3edcc964d37d4af50dd0988fd5f24a1b3c2a6b77cdf6b4176ac1d1f8cdd888f7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48209881Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0, Name: checkpoints/1152/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1152/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b7533cb144084785870c3ffcb01c05fb32cc8df31269a3a1a410261b650501bfdcd2ac934f6c573dd3f368d6c8cb630177cf934e33fb5ec0ba0f739a50e117d7e79e9e9e2c597a863c689bbed87ff9b6c0c0e85172455dae044abe56e346ff42e0ec3b393a09e826dfa8b8d2cba7170f1c2e60d3b4f6963bd15b73168ff804aeb2b1ee19a3602d411aa50c441c014056a49364af2a34e0e8905f7d27a2d0f0a3df4e0cd3313756549d3a5927636cd22c4992035f74c309a99df4cab7f982607a1c91d49753eaa60b27138550103db6cf564f7897cd0c63c401bd68f2aa4137e744cec52884b6694aeb206893a583fa9ce67757fd793aaba5386d78b398415f16&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482257068Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default, Name: checkpoints/3264/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3264/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=156612bf9fefefdda6b25f11f89540728522acf09f39cb0fd394efde2fbb2ee3c7ffd09903156cd7102112fe83733b9c5f6fece099637a903f8432a3cf9b5b82c0a05672117c94ee33ce0d4b6fb7da75757a0b4e5f1952d28bca2ab6c60b7c257d6c45fa875a75051b07ef9e889ef54d24fd13ab93ff2d86495dbbe3b57a8aaec41f916992e624b9d7e7fe9bb7a3e6d09454d7df1f02fe23f5723315bfb6c48a43d55ff2a58938d24f40455635a6f1e5d800f05ebcc065e3bdefbd17efce89475a448f1812c363587161f9d993ec52b4ec106172a4395011439a49239e23306a6c0c4565fba6c97930fcadc7a4c42cc6ac00e3b655e86512583e682593202ec5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482309538Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/d, Name: checkpoints/4864/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4864/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=23110d9e3fced21ed675a210805029a001956e797cceaaa74c7722033379486f430c071c1dac85f3c0926f266d062753722f274439f7ad4e01fe368737a28e232d5db02689ee2931b3176441f48f51c263d8e7d64938f0d61286a52c4587dc1b1fca85d20745eeeefc690a68cd374009c437004ae26be750a10aea2e49e2fa4fff0ad4533d4195a94bdc7286c5c8d12cafc964fc8a42c5321aed06bfcc9235fae95559d4f130d7f7cb457cbb0d3b8dd71ef3aa5cba65cb04d25f3c26c173716b624a17c9dc10eaae2bd0be655f1cb86bee5908c1290dfd29c02b97cd542d2b06777ae25544ff0dd9298350331e9d32c303ec596d2c4a6590dce4775b0b8720d8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48242772Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d, Name: checkpoints/4224/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4224/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b0a9b2aeafee09de6967653a7cfb4f20bc8ff8a3789a295bd184fbb90955bcd3509406766c6cc80a701033dd2456ece18761cd085f30f2c9be96e89bc9f175d3ed3c1a254b10cc416e805e8a19692d61aee824968b3b65c2fa86ca2c2aa7dce9df5264146831c3fc0d56be35c604761a55f2e897b5aacab873115d5537f4cc0759eb0ab6284de1508b91b2873e134cd8f2239fc69f46c32e159fc7684834ba44d0c2695cc6ba251a4890d8ecae90f8cca2efed07c91e09481e5e753b5a337d709ce4a3ddce7d1d09d1215734814aa3f1287ea1c49cbeedd312ec4bbf98e36b456ec0546100e39534df05f5f4ae6a71d7a49d77ac65744cdb1593418b2a827100&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482170209Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/array_metadatas, Name: checkpoints/2496/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2496/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b6dc086a97cf941587d202a2c8232375957d7e0e78802194b33d40a5cf2e35ab11490bcefeb38a4032b0c5056ec260b1f40a8edfa42f6fbf3169b3cc2e3db7200a60ea90ff44a8fba77776f2a34e2e4dc160bc1fccc614481dc5cf9e85cc790929b529a8d626d8b911eaafcbc4f22c8560cf72e68b6e1dfd355c52e5ca7d17b03e092412caae5282a82ebb8cb2892f6d7602be0bcd1d16b2b90a56bf086806d2cb01895a32d33d4642e2ab88b5925df353baf184dc13a447ccfb5ea779815c5e6da5ffd4d2444e71927a0043ba78e4e46d17ce6b44f35d3333afa11612da571cff9cf22113761adab21248715c2c46d9df3abf4407966a17198ce07200da830a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482426324Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0, Name: checkpoints/5760/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5760/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=118d8d13e129b330b6159f2551f3f607b9d5fe6c7f663cd5bba65e33115fd98cee1844501d687b547c9e132920811228036967022ed75deb022f59dc66f4b25a3eee560a741e62e9d0ccb3fc9b1ae6e56d2466310b6b3b1e00955e8dcdc91169c4fd5519db15c31a3d342007052342021e1a25e6942f8b195fb69317f49a3cdc2e7cf371002e3ac9986a3fab575d94e4a2bcde3f34f90998967d3ce51570b36f8d48d1a4411c6898e48b2d4922939811e62298440264241df7f0bd9eda008a45a463a5ad7fabc59b297c453fcb5a663cbfb7e44a8255725c04c511bdc9373370da2ad0f10081b061729548dd5cc011deaa7109a4a9bbf6fe23ccdd51cc135e32&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482422632Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0, Name: checkpoints/192/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/192/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=334d6dfa59f6111ad4c25e3c3d13d1dd67bdd1a48cc42668fcffc3ca0f6718ad027c6d344658ec6f15d161cf4bb4ced24fbeff7672c991fdad2ea4d02ff425d40ebb3b643aeb53736e9cdd4aa94111c6f75b1f8df2debbbb3545b5a9c2ef8fbb95136397198a87ea871f3cfa3d07c3d143a6aaec4d8da1aede9a767c8d20acbc62daae3d80345747eb8902e63d119f3aa0242610ff08bf4796ccb35ae37232a8be55496d443eefe411825c70767f3acb5fef0c9c0255d8b9028d5acfd0a2c1e524f5fc2371ac7ea642f6b41b3070f7463d4a10c0c7c0b77e83160999befa9d9baf87d967ac9a51d82300290be812c347c51faff2bf1f1d45a3a6ae0422ac9857&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482409317Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d, Name: checkpoints/1728/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1728/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9a23276861fef55482b781095f96d1ed8241ff71ed7f24eca13c26af4e0f043c0a46d7cdda91ca37db20c22d30032c3b51b3c827a712a0d0a7ebc4cae8525afd5f4c0b933495dfddbd890d9819516ed2794da5b1a7b2e44bd0eee9a0ede86f5726b1c999d844a57287c635aad6cec44e1f343b45a766340e5bf0dd3a0f116253a3cd4d3678149336c72fd2e7fe40eed62dd80e3a625287043d1a0d0413c93017a081dd9c0e1fb4c92f484d096cc9ef74477b382305bc5422fcc9e5c319d0ee37a7290d867411cb6396eed4d1c5e68812b54b0bc9c7dabc8aa56b891bd83597fe3deef47af20b68112c032232914625257c0c67fa5d31d8c82a5dbe23103707ee&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482129046Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d, Name: checkpoints/3200/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3200/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=33c9bd796e298ba3d825f5fbec40b708ade190d8783caf6e212fdc1b2731fe01bde21a77a40cc972b9a0ff13038f856eeeb291010c3a9a461a14fe8622e714d042dfdc192f0d6bee428dfc72a22341db78c0abf8fab1a5ce5ffb34d3e9594cafabc9514839443a04146c249f6adf19615b70b0b16bbd779bdbc0a1fc3b141cc6a933cbe8ec4f2c2c7322defc48ba33ebdac95c399ffb3a4d82ab4265e19fa761428c0c7b83449ed356613d6232b071c6774011c5fead46ccea938ae9036fe6acf937d52e330562b465d202a9d2e5ef3af81942b49f596e745f164ca6be84aab2d5eba8c6ee65cb0165be824438facb5676ee9306ec3809d6df9c008efe5d7282&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482555208Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320, Name: checkpoints/320, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/320?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a7aadc4d8dde7cb6e61f2c513d6e69d372dd5cb1e59abe5ddc214e182337264312c16647e82b9f3887244cdca7a8bf5c8b83bf046d1455747c2862026d52e50f960b963e7d25f545df63eed387963641afe5d50950a7f4b07bbe75ea59e4df17a0acc024c4729ad4ec686200b60701a557bb99f39cce67587fdd0ea94fa39da2cbb81568a4003dd518f9607543452f5a82f89ab500c16bcb21e3288bdd11749066aceda338c0e9144032572e08beb67a9244b40c383942f74e2fcedc9f6762b6fbcefc6924f05a4592dd4e9aaaf44f87a0d714391fce024de944af2c74aa1690ce4310c6b138eab44b7097a63f521b75fa6fcd35a4c45b115416a7fedae9727d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482570895Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/array_metadatas, Name: checkpoints/5632/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5632/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b228221234d7fa1464ea0746b9961cd51ea837726d02b75cd1f81b02cde18bba086ce6c37ff86d61fb2ed7d2e32e702705142d117667b089f9ac7bf579ebf24df1242fff93e863aca8a98e56491c3cad66a88004a5d7f41cb246726ca74662ae6a75423a75b5f6bf82ac00bbc6effa5cf560abce092c11fc3c7f72e257551d3f909d8092aef038788378363866b55da6847620b33794b130a2b19658ead6a5344077590762afd4a50adf29e38d53a3b3358e15855d6f568a96a641e07d5a281fa98abecdce89cbd39b383bae6b81895093c0951168c838a80507881c1cf8800dd7c69702520060377ba45f93d3e4d16074258dff9db2411e2ad27331742d6c81&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485379709Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/d, Name: checkpoints/1152/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1152/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=72b8b8b8653dc3e0b936a1dc5b0315a59a4cf7bc2cbc9e02ec951f71fe5d49c5eaee67beeedee7306444bccb4d58db3b3747f4ba74e759fba81a08b971fa45b1223eb9ef445211a243e99f1a3b32c92701143a98ff913892a26885ea0e5834f672cf2316c20a5b214a5e885158d7e2e90d6f90d8552083031d9ea1d2d975774d73ba905357af6f8cfd37a3b31df5652b6a52ed345d5a53806a43882c3a26de66feda9801cdd128d84b788516ce8e0cd1842b575495637df65ad9c020bdbf14ceecdd626e078fc47f771ec0fb4c23cf63970118eaee7f58819fa16ace04d3021136a2f2e0dec9e0af3267a63bbd997c29941c64b25ed23475e5313eef30d8b9a6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485395489Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/array_metadatas, Name: checkpoints/2880/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2880/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bcf5a94243aad1f0f448792eb998eaa278459cdbc8c9f47aa98872f63546a75b1e3ca1a720415fdcc5958403010c63474a167db306c67c4bd6a59d2ad26de1fb807bd6646cd5f6a0b41ada113ea1876470b6fc4edf124b9737cc658f6a25f7d5776ff75d87f3072292a3da1c36d046366471725a27c5e0e700821038d9760796174cfbd81b21100b096d710e304743dc616e3db49b3ac682c37d30743f0d6282c41d544e9d784f1a662df360e2d7355042048087144cf3a84b8edcd51dfb718e328e021144e5d25dab74c6d17c05892f89f7cc27b4e242ee44b7ed70b645a9520e5bf9b4f1db11b84d90fc4d829f5767de4632bff8f70d9ee06d9a7061e667dc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485405058Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d, Name: checkpoints/5888/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5888/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6dd26ddb5014ad2eeb48b3839a129595f63d80e21cb6ebcb52def8ba351d50fe3f14335d3dd64cd0ec4a521f90aa52cf7154ff1e5e604618e1a8cb76f556da5d8cc57384cc5392d5ad69f11ad69bbe935c7149cecb0de6b79758c13cc669ca2da05b6c4bafd04addb277e91dcb07f84a74e0caa532b16277fa91fd99de5dfc41a023d5c711a066e4fa142c97eddb0f2637e138b9e8aab2e504fd2108a0b75357ac3cace38d94e4d01f530211f2e28b39b0e12db1a55f2791fae87d86aa70618108736294a20d1df204f2021d97a902ee1406451d686cc6f93c3570d37db80a2fe19b396e8f45396a6f62ee201966b8e22b69f7a8d9d50d89f46b073bfc2554bc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485450081Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default, Name: checkpoints/768/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/768/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7b1582cb512ee196abc3eb24a3dcc5b27eb5f0405f7d8372c66722cdc706fa3400594595ad7f1dea58bc6fb25cec0d60a8a0ab78d1d06569252a6bcaf97ff3b868699f1b6f4534643411fd82dd941a4d25634d4c3f8a4ad685bbb27d1819e042c7cff7eedf6ca130ff0fc3ee72ff9f937b4a6fc1d30ff045357091d2d04a03abcbe28b44592e41effaa66a219f0276791dc31764a59857868094a716a4287ebc71e54391222864a85ca96298f5ebd03652e129e7108793511229dbb3afb43aa66bc3f37d2a0c0f0943c34d99d469b9ced05ddff2ea8ede1c1e453891761225db7a7ba8633262a6b5012e9abee2e88de3aec735d4fa5524b2f85fece54d7c88c6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485464949Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/d, Name: checkpoints/1280/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1280/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=39dd0f3caea529697d37454e3d60556720d08c8d1eae7e13bde98cf353a66d57b36db2eb34af22d31fbee131c2c385b7ef4ea2342ca0b70fa8358a4f8c064631d9e35fb5961352b89328e5902e50469aee2752bb3c243c76be4e4f3fa0c912d7bbfbe84a17c42f73720864b64b5b00570429c4a3b2eb78ded3e0f9d2638851c67520d739866ead82bec2e0edc66904ae9116670f1df760560ca9dbed410ce294f7b806e298ab8d9e4474801e109430394fcb7999fa459beaddeda0deb0265171decc133fc67592ca88d210c36714bad857657044edea762eb3fded49631e04525d104ce5cdb8b679f06b2f463f55fe76f287e293001f886f90cf60a72aed4c12&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485502285Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/array_metadatas, Name: checkpoints/2688/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2688/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=302713d527d37b387dd4000917ab73eb24d67a2e2f2e4f94d5902283719dbb2248523408b1a6a005e10969b4a9152edcf5bc5cde1c565ee33620fa69d1b49b44cd67149a4da3f408cb6a600cf8b29b38eacd0b3084e6091177870880c14328b90b7b35a08f0acc332ff160fd3a3f21b6255aeaee0217da21c8122ee166a86b2411b687010778cef36d3d2271b4e66617312a074f120b4c67e42cc6536f3d09e8336f4b0eb303132931d6685d459661e815929f0212ba98f368bb534de22131435a8cc639ef6135278f24fe5654d304e132d134e585f9254871f2b911d91c98ef17459b9703646227d6dba909bed28a2bd8918d94836c9b53b947c444927d2118&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485544576Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/array_metadatas, Name: checkpoints/5120/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5120/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4d62bfdb66934cd52b8c5204b82b276ae273f0c891218358702ce980f708c86595c951b7a7a879504f36470e053368c08fdc09d17af951810f75d86c3caaaf331645ec9728cc45572e2429722ca7694cce8e257a8ee710eb8ed314ffa73074a440dcfdfeb459383a964c279c20129c4d47ad169213c5298ffa5ab19d7ae701bda8cf542a3ab553889b5f778e07a220439a285617a765db5b859654fea4579126237917b0def97dc5bcf6289d7f03c7c3133549eaf7da55f040c5074f35df00de2ba89240832576d7ab0b1b80c62f91a2ccf45635de80f5819a58b2c34fb5231f3c98f70d31114bd79603c29ac30ca28a808e52942e079daa1bfcb8df8aac049d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482584844Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/d, Name: checkpoints/2880/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2880/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bd50130b512b6b20873bdf607ec2a446d7f518234cf4df230371df7ce1dd450cae802536594f93e60f5b8e14eb435b2e75e92807f611bdc82bb6a352efa2288e53a69c4b3481bff3466e03b622cd75ee68ac6425c44bd59836dc3ab6b655e34130e61834bc7ea533dfb31f89801cf808b151f6ac2ed96f6adaa95e6f96180175b418713678577cfc0be7f8ca76f83b2eada1452a263d5d2808071dd383345d3fb881f649d4e69cf9f2d3481dd312fd3d336c9578799b5f1ab559d20b9f2f7ee82958e83ad7dd5f739f8ddd901832772d4cc3b667e94b7ebfd590e248b1924046fce84947fdfa6186aa0674f9dcfc57fecfcc1f2a99167f948c8aa0c9a2c72d23&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482663744Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d, Name: checkpoints/5568/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5568/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=ba6072fbedded4bb50f236c00a9c6d5e2fa7d7c28cdd17f81559fa4fba2ca28f6eee5a831dd2d40754ece50fd44238fa6e6486d9f9c18cf1bc35ee84c35b17786b5b3da7b24445548d5ae865f6042ee88e317190fdc13570e5039bc53ce050627ab8a63ba75f5a77bc8ec376806e262f035d08812aa95754d1c80d060ad87aceac44ab7e44d294151bcd525c30e34624ff51f6f85b39e63003378755fe46e95175fb01b7d62d3a8a79dfe4c7311f9b362abd517e77969bda8d74814b0af99925939930763ce4838c05786fe86245e78bd46f30f4724fd83e8c4c7b8c83d1bbd4849e18eb71cbcde6c75af3daa3874110f9a137ac52cb93330d21b35a30b4ce9c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482630483Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/array_metadatas, Name: checkpoints/4160/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4160/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=599a11fbdcd0d18901013d98ef93efe3c2de800a2b14f9dbdf04e695e0a41463eb57bc6253cac9dd597a4c49d55bad21040df15813b65711c7904a17ec836b00295269ed956dee55bfffeb81205435f709080a71421f38a286286e13b66ee24ee01507ebcecb825d4c98032d23177cd97cd3214e7204ac94dd4181e8acf35c33e82b4ffc4e6d444181ac4998476c239ee1b086fa27c8bb4ed331f1543d92e194bf85ca9b55720c484019ac2ded8ea3f0de450b46a46d712eb700dcfd286de674690274067697c75b39a9eb4ad74e76cb0bd4236990f275d3e6c388636b5a18d53bcfc75d9ee1f3ba95c31654e0927f0e295ee7d7ea52a1fb0b178883b23d706a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482703519Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456, Name: checkpoints/3456, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3456?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a037dc42c3fc78c45cabdb121bae90be86819dabd84a3068a26c05d49fd3e002b4e1cd694a4af6720f2e25acd523a213b0c3b2c362f3b12072aa7af430260bcbe89fb049757f55cc2a3c074a7f18aa3e709f4cf1a0bac7e3829628aef33a4ee4cefe9f65e2ee3159ff0e2ad41da5de2733f90abd32e239b132b2a6385edaeafae1fcdcd908f054f616bce4c0f24ed8274a550b8bb319ce88de2bec51bc90127a7b6036b8c5be8a44f572f002e50d81044cca83f87cc6e49eb4f14d947a7d603174fc0354d71663d2ee6da428489586238df05aaf477b6cedbb17051100ef0805dca24e4376e10ec4387abbc4febfc3dc35e785f53490f0ed9b9066244382604f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482732076Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d, Name: checkpoints/3648/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3648/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b84ec846d8737493dfa14b59ee7eeee98e78bdb57ef8c970b877441fcf36a511f4bea2404696071c66add7bc78177d6e23ba7541254bb483cddccd28b8d2c48c19fa3c14b9cffdcfb9bed2907ee6d7335ae75e04164f81e21d51cebc26b3c8c8fbf06f3a685d1ed2743361a8448660988fc0076d59b76ffe424b8bca9917d8c75da1a3f298a15268d34c2f07e61df1bb6ed32c5ef00c1596ce4269ce9983d6f7d10712283d41413817ca011d757f90175e96fb7943856c2e17313cd3d228a0e246252732ba4a638d05b17c78701ff0ed5082489f7d0788e87d6339a266b203d7db0e2c8e1eb8112f4f0e33118e8873175f861266396cafb67cc7f8654c3ba438&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482740058Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/array_metadatas, Name: checkpoints/4032/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4032/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4c9431c160ef70b044e07940e10fb50944c3852d74857a15dbbff6811e81f5699e518d6fb5751918f53eb6bd0ac03197762a7388ced5daa203506f9a643b2750f1add92a228183f8da8f982325e1145b000dfc73bcf171fdfd852456cc5e5f74aa5814b704c3a922ca19478dbe943b31e1663ba921087caa4dfc020bceae75c1c0d773cca7ec861a727e420a281435bee387330e54d55b549f6ca72a90872d9137f488c89e981d0bd3b693bedc9d808ab0ae4ab2c0259bd6fd07102e4d32ce4e8dd633f6a5baa4d591fe093158dd004a4fc76995853d018417ddb1630cb9429573fb6cac37ce304bcd89d17189729fca1a1d76c0e4896c9c7d0d55dbf343b4e0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482691099Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default, Name: checkpoints/2560/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2560/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=50d36a0dae8cc24b2d82c09820c09eb2eb28a2bd4e4ef0273aec6a6c2058360a9f11b55018274bfc8c10eb8c98da4992f4d5c720b161364dcbfdfb031995fbb2657ff5a6000ca05ecbf7ca17ff7b0b211b871853836ff0980fd9f0a169b38dde5d6e3b35ca51c842256047d44b68b0fc672716ff3b8a19a4dccecf7942267fadba8032b25fd3e2c1321be531118ad82cbdd0cc12d683f69d5801d3c4e2f13c03c37ccf1184ab8b8b34dac3ae843cc9d15c31f49d7275e6c6a8a46dc12cfd58a62f3713bdf959156327473b7ebc6a6b14ac6dc6e16a2002e6ab9905b4d171f14c3d54fa3b725a54583f530090c39a8fc8bae72945c3836073b2a02986645f4ed2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48257025Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472, Name: checkpoints/1472, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1472?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a5000fe95c8770141970b78175d54a6d3e6a91468c56fab8fcb22ac80999080076ff8c5194639a5c69677ef3c24b65cf4891df0a7e47479b2c60a800388d286e14cf6da34b524a58a080fd74271eab027b73f33c9e953830cb5765df635f70d0237b086adc178788003280fc6f576eef820e38666c55e6d57acf9e4e79c9cf20e56ec6d0ccdc64ab1ab7fae3b419d4fe5560eaff7df0b385e99bcf5567ac019f166762104f13794cf0dce2d0fc746c5f2df9c1ce466a580251d65a5c34209dfc5354ebc25f9a9630d2c44a113d65635e78cb51eb4b9759973b6680e8ffc1416a2c8692bcd8b7b836a8682341fa6b8147b6ab7678d8bafdd7b01789a45f568733&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485377239Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d, Name: checkpoints/2240/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2240/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=385fd4d981971f57a1e598ae1a4f9d03a249b4017e459f84e930b2141bb792b33c6dde801444aee1ca955053672bae383dde27dfddbdc7f160ffb93ca1ead7cc0c95bc4a0ad1a58472af902041d20f8feee217775bbec17068c00a656ff5736f7ef4e5e20536b5d71819241fddfe335f5d47144b79f9d06910cebb6cf40bc7dd2bd19ba72d3c52b82dc32c431d3af3b6eedc2f758ebd175470c98cb38aab0f4e4a700d177644842a760e23b2000c8a7369bddf40deb85678a6aca6ba75d166bc1ed73f796ce41c8b0e7510e19118d72e970d0b5eb4c36b2a1d70df578e772d45be5f0522e6979408d50094f7ecf369e60f3a62c47b93e6a4982341dcc7e7ec05&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485644305Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/d, Name: checkpoints/1920/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1920/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=633b36010e5c3d182991beb037697ea179ff4b8ffb5c8027898a9f1fb00d6fc6190316ddae6b56a4678bc765b47f9fbdcd4631f3e8dadff8f79fc102c5095efab3c9995b89249ee2c636b7ea67224c995ec96bcd575d8534804ee9b2d96effbb0a12a76930cebeef6b71c0032c58fdf0c9356e895a2576cbde7654cba1e7c7966725ca60af95672d3f45542aa96de5fe3d55ba1347749461a88b326810cec78f7e359a6851a06fd14f2ce02a07fc54ad7ee45d7b24f25f5cbddbb6850f7120e7d2ad147bddc08e2bb56055af76a6a599d1a97e9ca65304b031aa6f1cadca22137bcf96f531aee265520183378646b9951e6214b99afe7da8623eeb9b2b77bc3b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482658032Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/d, Name: checkpoints/64/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/64/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a666418fc627dbe53057da02034a53e650bea66926449e7edbfd33dee1853477bd088b2ee67326cac6b814cd391056ad26bb70fca349b9ae30725e5914bd63e8cc2cba985a32e6dc39d195b57f8f460739adaa56cd2c7e57eece74e8ae2a7b83fe3f2b1b25af2a4c6ed4fe6b1f0ddbacb7e41272bb3fe23c10790fd874b9b61fa9aaa57ca0b0cfbbe3ccacaaf29826efa23caf34344f6e6b94112292a4a5d7d78d45fb3c6365d5a3d6582cb52fd31d0019894424415f988c31d166821d14065b742714f48015dbfd8a3bf8c083f4954a01b53443dbe7c936c1d01bbbe25a5df1cdc9f310e8655b7f0ed6fab01d4ab0f13ed046f006951e5a1ed89910c578d8f3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482786367Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d, Name: checkpoints/5440/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5440/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=97e696d671bcf650f7b2c05861c7dd3193d053652dd597249d90c69168248340a44236984b3178787968d678d6a093f6c16d884b881c3caf14bf2348b2b678943becd7430b2979819b9a4cf72624af52508635aeccdc8c9f17f028ce5c1d7683cc71511e160205e85cf379428633ef30be25e56edecb076b72904a4c533db139bd78528004a271f212d8a716d56dcd8da5188c29ac8765322ccbffec7c889ee5fcd147f26529aa147edcb15b2f439d2db67619816339ccf83a98c833bd62cdd79c677e20e439519aefc12d7ac28ad54e5c4bb79d4084a4f04b06fb081bf94eaed415b96419aaae4914f19bc21743b3d99e9ff9913feec06bf9ce10f07565bc33&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48279962Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/d, Name: checkpoints/5888/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5888/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6778762b60bedf384559dd9998f10d1df3548f871c5b4aa541367d99390558e97db6054f7a1f0a84103a7856c7e1a6496bed6d0862e4f9d3a60c4c714f2b45357a2c378dc28a26301d4d583bd401cf5e2a19d7fd2ee3da169c0fe768d50d62fdfc06301cc8dab919055fe496c38e40f9d8693e9418a734d139502dfc8bbc5e916d75bca29f9e7f959f52bf3972c960f90afed37fd1b2cf03a34181a1e5929be6c9dd1623401e6576341f9bb64fdb3cc62ea66d1b81f64b4699d6f83de733a2456148052d87a8f20fa59382b94e72337dc0b72418a939125dd19e429f1d701765c2383bb9bbfbef61d8007ffa93c42e2c701bccdab991518fd437125e58295219&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482495401Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d, Name: checkpoints/5632/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5632/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=19db1543bd8f8ab0765c428f62566d03fbe57a39427811ff7ee0c1df261826cbe2d9bdd7a462d743425951fdce5035e6d7baf5799d016f79da233d59d3df7bd660cf851afeee549ddfe78bbd586c98497c42c09e201514fc142074149862f8a64d7db715829ae92dc5332ef522e163c8ee60b78dd8c49b92653ce8d2e10532b68adc4eb2a8ca1f86897e15a5673b7f74791ae69af16cd62495731f5862f75e0fa3d65b0a98afb4c414e7c368fbcb412d887b7745026de0c8d332710a4c8e71c6f33c73874029c524c46c32fdca2446d2181f74adb63d47e0df4c08c9ac50c39a6bf3998427a3ea57642dafa0574728932268f0a381933099adb45f233c4ca806&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482837754Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default, Name: checkpoints/4032/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4032/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=73b82cbc09f91b6b23c3f6b08453b513d45499c88a4a7c278d2e7bf624f8c51c7d8584f49720c9d87f984ad0fb1613981d6329adcaf416b008edc196bd3f153a779365d262d451818d57ebea00e34bf0bee2dcde4dedc0a393c2180b36c4de6ede92e286f511c44ebfec89f35b87181ebff15cf79bd5b2dac069914c2cf4a214a801222235462747aa88cf1e81d2582147fe5828fff8d087d4e197b969a99cb2d6c06f81b8c5f98796cdba74704fdc7795e79d83fafafb685998f391b9bf08a5352013ece4e90f1c820345ad3034cf5c0e7f206a8f0abf67fdd268e872ea8a44807b77dba441bee5ae7d5b35d29218cd5c0836b4648387061911880209ea007d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482825229Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default, Name: checkpoints/1152/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1152/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=186e63efa2054b7d34d71bcb97b671089799f48a355c588749fa261124e1793de4c3da775b131f73ddf3bf620e0dcf5974b478543a19db0b1d631dc885272424379d2dbc007e61667f30e2e9f0db8f7d40a7ef0b4439d880df5fadb64e0f9030917bd4485e42af4a01b1bf55ef16025de822ae61888594682ebf973ace8c6603b8e436dcbeff88df9f578b8ded5ff5c220997b553e319161379d4395eda750579702c4edd1f77ddff5434ac0cec4a4a36224607cf85f1d26964b3ab97eda85e58d2799ba54b67d6d0f2e73ae2165569e60f7b8058822bb26d04a7ae69189198d2df916f26c6cee81f0b29919590d11342e2340dc9ec7ef4465e1380d132315fc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482759918Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default, Name: checkpoints/2176/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2176/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6d4cbc9820a6bbf9217240a9471df132fd22130d6ebeee2eb9c22ff541f732f072fba95e61c3702ddd2f85e4d26f5c3a14dcd226c8c2ea71db0417330209ebc4b72b639b34903d7fa397ad97c4e0ff4d992eef7a306cae2a918968f8b05538d6003dbdb66ba98537e6b5fa6bf4c01de164b84f714ffdb37a32cbd2b8d37162f13267fe017353830a51cabd401da37be97bdede12f88c8b1853305b108b46b6fbfd802294fd00e9509b004b7f898b30d2eb565ec36e8c0b0a4bda939c1202118f111b6b08035f99ff697321b0fc815b5577e05c21ffbef30444d6bc2587a37bd37da4067bbb2e2960fd5fc33553284d129db2c38d53ccd84eeb3a2c85268c2db5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485751743Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/array_metadatas, Name: checkpoints/6016/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6016/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5eda6c061b79841c3c6103127a6c493ee61c373790911a69406acde0b3604de6971a02773501e94383d503a35f1990b635049f99e2245d1cb615eb105d752ef8e011dbfd798b0119c7a99f2c404e662d88ff707d85eddf7e536ebc30a91db9cd08a69e7ace82470cd6b7a50a0fafb3b0878fc417c0808d252af67ac37e098ba10dc0885c9fcb81d6b903cca7f4bcafe63e2c1a29a8a42e51eb7b07923b20ef5258b2bf30cdce778c9285532d8429b241d0ce938697400874bfbc6f3b08402a0ffc3ff8fd5c2c914a5f35bcecae43d942a6f556d9f5dc0397e339849ca7a82cf3b79b2af29f11072cf822255b9765324de070890afeb417ecfce9459111b552f2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485777762Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0, Name: checkpoints/0/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/0/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7a490ef312c28ce8309d36ecb06aac5e0f4c146d9c347325dc105748ed829a7fdf80e2519e2764949bfe3176a535aaca8699c6402ce039cdb83d4cef76b7421ca5dc74424b790a7d0fc63c7f5b7f4f024b68ff0df69b30a96fddf7f5f14d47dfae4a174470d27dae86d16e902daca05dfa12ea6abfd4630afeac00af8e2e0d0592c8b71bb2a8180df450c5f484f8d0868ebe2cd2fa6fc8106a123cc48e2174d33f510b92f53f0666c33bf45b59072c3f5c4b1099c49a263c7d14bd8145cce4b9ef59382a2c69c1626a15bf69fccd3b4865e2fcecd41772c87b1bdec29360bd14b6bf439d65a30fcbb40aef3cd0282c17ed9f6fa0ebe50a97c8882384ad7b5255&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485784604Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/d, Name: checkpoints/4352/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4352/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8b048b18c8aba31c413eafe122d408f2e41883b977dc433b9c568a44479cbca7234cd6fcfe78bff893b60835cf4f29321e9c85d5d41d259c0f3b231c52651147835631133647194291dbd9bbbc8a9b821c86e50db311fe1711a9ecd6a92b27325cd15d426de05ded97a60959e32c54c307653a3454978b2deb8c9d527aaaacac1042a53859b03b999adc7cbf9123149b81c20eca6c8733e2d3aa438d3f271b4eb80f5eedba78b0a43d7e17260a0ac1e278bf160b63ca28ba2df8e5b04678b1f9376acacac293bb86fcd0bce634f660b57d34f8b45931b13fcb30956e7b7a2e9cf2900f3afee3e576129c90bbe8cbc1be0e23b4b914b61b4256edb9a4567f614e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485801273Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216, Name: checkpoints/1216, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1216?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=63e364acb22efaf0a1915c6e645e3881e32af8cf724cfaa5faf021e87d3f1c0d39b23a123e73f37dfd06f0fe858ab0b013821f9eaf582062034f1c68b2b6fcba25db4ec15a87105fb94784bc12bf663fb253452ee3b9827086c1db8855a3e5250299f3f27a396ea88dbbbf603b9668cd463f775a55d29a921513e14fb39647aff6f6c698e18a3adb90ae2a6b00372f99d450a8f543e6fbf5d52df9847ff0ddb8baaacb8cf67c3dce42cb7963c9885fd7b6d6faed95ecac879f0f721d4c0b10fb91ebe1e54904085041029bbdfe4e8234a408cf3be9dde78598f3beb7e6de37065ae9e8d7f22af0327c1b3e155f7c5ba1fc18acaaa954f92190f7be1a778db638&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482945714Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d, Name: checkpoints/5824/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5824/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=621aff808282347e2cbd6ce9954eca19c9e35678e7563f6b0ff4c3f76bc86f330ca2bd653fdb2847a002c2447da29c7df287f2c0e22f541daba70ec068f8bd8ce1bf490ca1409a26e836ab56579d24e7198ebeed45eb79f6d483da660161f9e456d6c960916e22fc588ee5e5a9129d786dfb2c64b2e244504cee2971adf1d5530a6038b1309b904bd1ffb9a9a98966116691d2bb8f8b175b9c00880acae4b6f1cb6bf054ab82ea1797d738a6be01fecda8c87a294876aaf0425f59f47414c7674ea30aee827ea68fc619bf224e74e7ac71f0c541a35dfd8b75bda56a3d8e5c67e50934bb9ff270b1d08ac1b57d7ad8f1b5b0bb9bcad64b63397dc404337281f1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48587814Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default, Name: checkpoints/1856/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1856/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=91b708ba89d82127191c762445769fc80fc7af4d39a505e66bb9a3bfe3b4df158a16c9ada856d0ad24ff1af1cc6ada6fab47fb07db6d9068c9632af4836b051d9534f4a3a9cc6425f014502b747c43f3967c9043c743dd74c6228e686f83ff3ca71196dbd4f0e46ba42f0ec3d34a85efd80f14e15e8312c995f02f92b6327c14f40597452c69054bfe9732b59bbaa53562baf51813fb1c6653bfa82d958daad561ce0b3744aa956ac45d829915991b74f0bdcfeb6cda4f2e63bd60906b4a1a34698bd968bf686f8fff5de014b68cc779a74a76c1e88b37702141a8762842e5c652bad07046d75885d1c0d01e6974c8e8807159a28067f6b68998c51b21b82201&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485850499Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256, Name: checkpoints/256, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/256?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3ba6505e17267433eb1b2780d60f31288cbebfad0366cb51971b3c717d1978f030d9830f2127620542ab9f3eff69d13237b309657a8f3394058aaba2e402c663afd0cb7f3afb27151f6340b250bca50794a25714c9ba946f1ec9dca82d81719b57f208bc864510d2f1f151feefc3a3f2b0a9c533a153485611164af04340b18cb14fa77738086545d89fd38dd4fed3947ec35b907469e8cedae2a0b3b79057e73c15328b1fbeca3824cf8caeb220d86a6e99b220257f24f389233909158bd38611111d0e7ce5aac20ef2a7759f78a6ea8ec950812042f5f3db84e0e93fcf9c6425326fb69afdb5e51cbc92c213c17b03af0dccca10e7b046b506578993ad9f88&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485936167Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d, Name: checkpoints/5376/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5376/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=99c21eae9d2ad0e4e028d90d9518923ab072dc7ba2cc60ca36a2ece22c90c2df8609faf529363249c335c1da786f2fb9a769f7460f9df0ce92b1ae31893736b3d94deb1bbf8ad94cadb8cf1f4188cfb7e800b354ecd7d86b934d41e279ce1df99a0096b940d892dfe6b4d9dad620fe5df24cec985591702e66070d3d124baee670229fcf33106bce38dbcbda89070cecdc0c4965d0885963864d99e51a2c286b076c4e4dcc37df2559dceaa186df1a2c99d791fb14102dc4e1ff8166cc6d328691d3a44700ce9eedf4665b02cbef9b3dca106e00ab384c08fbed07c0de71804aa3d58cb88bddfb90ebba91a5a6142fb1a4296bbc44079c0ee8e8f2443d750b72&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485942032Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096, Name: checkpoints/4096, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4096?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5ae73d2cd67cab0525f82a48f8c53d8550273e4fb8a4121c79b07e26ac71342d50ca5de9197f75d10cef4fe10ab7790ff8c450e177b20b66588a276001c051d6d0350e4a34551eb29603c85b844501823d17de8efe18f6a8bed180a63af5b540341e578e6f5a584e528d30a2b96dfc15139e7fbe1fdade13b108eb36a0aca886d670d04e98879842aa694dbcf19e4709013909593be945535a2c987550fbb3f8d754f8ce3619e5ef67f72a93beae510142a756cdb3d3a835c2da2314baed87772c0d71445737ab0165a12a69cd5382ef5d1bca0e8401158233df14fd2668cac99391572e8fa0359e3a6c786ee29f1bcbe8c477b9463f7ea06afc7aa9dceff8d5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485957302Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default, Name: checkpoints/4160/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4160/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bf5e271f7721abbc6e3cd8aabc8abba2acff81c2e449956159d2a6c1489cc21c21598d053aa60046aaa6c65bb92db683c89a873bb421b4237a1b13081be72a6765f4ac1e5dc2f34fd7a237a3931b454137bc7a65b5966a18aa9c3121b31e3a9176f88891b6307c66243c6e1c5c6052cc6bc8fbb4528a29330281e5f2fdce215f362be2e60683412732e924e62e7868a6075c98de6f49552070cadfd081477b4841f2731c4fe324b92f6a28ee0e2578822371bcbfe81cc2b50d44d20fb7bddd1817d1c521c0b6673f732771a781bcf8296804e8d04217dacc408cfb2fccb21646f42a284d9d630a311749637c8ed25a25f7bd93b9f7184d9be7d11ef0fea0b246&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482946371Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0, Name: checkpoints/1216/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1216/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b333681c56d9cd772cbd2da37a9b4e029a1fbb014e3982368610f2c57acdb28993f481625bfa651bb153b5280690e5f22e0927b3b49cefe7e3775806dfcc3e5e512fb486a9c195e5879dfca7e7f779c7e3096b76f8ed56c04e2030a7397d330675a11c02f3b789825ab4dfc4ad6924b4b62c0f06189b29481681c891f59c443196d127072b227461369adfcee2bbfe299c8c27dc3d8e62cb08e830eb8f487b2f1ec87f23d58faa634b283916b2e6342ef7995a8fec87a413ebd0436d223d8632b1272fed681b967c013595025b34e960d6c43167e6f23379a40efbe2e73f074110be0d380c9abf1f56dd6f5bcd5613227434b258c17177b0b9f431b0a71a5063&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486024935Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0, Name: checkpoints/5184/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5184/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=49dd5a7df46e32df408fd8ac8ccbbfb49518ff40d4bc906b711ab07f2081213c775213bb7605206f54f2530671f6456a4098f83d0ef555d9f1adc891874d1e9b4dcd8b5aec2afc79a6674eaca7af730f153425e8fa787f6eaece4fb61ea3125617cb3d77981b9e06b2f46330129041400ff0e6e915cfe80a4f5ef04f6806a62c698643324eada89de131b837376dae1852f1ef5737964c8cb0d6c91a51e24ec3dff62784254f57ec4d4eab05260e4d17fb7927d1d740be3831642ea8dd6b696ec6587659977d23fa2894c9a9e388016783f2618f8dfecf5fab8a71ebdce4c75a7a3a8a830b09ec2e419ac72d15f3eddf375d1e97f2a0edd39f8f598ddcf8d88f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486041371Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d, Name: checkpoints/5248/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5248/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c0c802805dfd84fd1e0c7369fa1e169a38208771b14fa122e9badd0ffe5abc95d239048fa21b25e65ba5213c68b39be01ae6f3ff3d91203923765456933526cce4a42e64ce608ee6b7d31d36e89c66d35b24e3349d938e63f8bce82218976efe7770fc04f1d1bc953fb000e1b95fa591674eb6ebdd49214c733f66f4d47f76505c6c174710664781343fef4f565620d44a52bc3db554d1909ece8323cb3bfc825d636659eb40598182249dd20bf56213a1c33f2181ecc1d8c8cd1a047c6f131bebf00144a9d34e20748ddd152320ab252471b1b73369e373da92b6e0201b9b0835d08b0ca44d1f18cac1b370020a5df865f374056574f2696744cc4793197fe3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482948944Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/d, Name: checkpoints/384/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/384/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=731c66a3613dc4aa4c5887397ad2398096c78703c470fcda148e1673ebb5efc1bea7895e726a5fac317dca8542c9c74f57dd12eac9a4e7d6d5f3473ee1276b864a5a8b3a80c60992750e1ec156c67d5d83ec2a9a10e0a4f22f205a27debe90d37642a0f13e681ef6d9e7d6daee685bc62f1af8fbf322f1bc31f1e0af098f149c87e56b418579aa5f2d338d65843d7de30918e10a03af1ba98315b6c33e69929a9df88b4d973806f9183f84d8b371eb6085c701fff3e487daaa502cf125ab474a4ebc09d3863c606181005ee0b6e83c7c4eab172fb1c572403091d0077360fcd41d1a3157a0fff379ca2628d867091440410d94966f9302573874a553027a2639&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486109136Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504, Name: checkpoints/5504, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5504?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b41733a329307d90fede2642db409e8e15bef0892874b23bc687703d97800b5aba4ef620e8e891c6565212e89257714690dc144991f4baeeef9d3e5f685aa2cd131c7148f09cfd870d1e8666af94715a628869b99858c18217e6efe1891d38251ce9a9aa0dbb2ab0a365663c5d87bf0db922dc818430926050862f1aafe44ce2c86e39afa6e365976ce1c7aef658a020c52a8b2fc93fe6a1f40cfc83def8b4e7e5b624fc48baab631b8be76d326fbbfd5770922bc14bc651f40c6eecc810a752c8bad3a37e1c439e9aaa8566952e335a41cfe2ea661ba0c62fa57a8aa40226f3f7bed61223132e0c412169fb1e1d61c5d1e95efb952c37b194d4edf149db7003&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486124947Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d, Name: checkpoints/704/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/704/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5ccce54655e0a20a1192756244a616419434855c20167f51ed324d4b5e9e3a38218fe6520842d3b5436f2e8068b5e64e1a969d0867a38bdd3b671895c4a150e866ea4e00d8e7081044a4907f9f5db6a6eb847ad4fd89e2518fbdaf867bbe3a83274dc23e16f1675315271ec3c6b1d5dd47f455398fbb6eaf0965f04660dcdc348237940b067c269b00c65a885ea9daddafb523aed9e2293d6d057d170113af3d8b3998ec41414142df7975a3ea6b36b0833a250cf147b981f4ff673509956941253adacf0f7de4191e8e4e262ef57c7a8b5822090248076a1ae73fe298e20da51e912381baff7001c76522c2600ec9e35464ab3f9b83b52fdbb6840890a652f7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486154761Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0, Name: checkpoints/3136/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3136/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a71ff76cf6f6eae008ee90804bdadb16583bd117660542f5be2b4dbf1733f758142c9e34afaf0e9eb07e7be3ab8a6731b87376f2eedbb87b3519567d68f84b32733d22916de031d71f195a277de54d80c64c4a59cfda54f225921059f2246843e2514b8bedd77e95c44cf33c03ec1e0b9dd0767b127061082167f7d6d9fbd439905c3b7e98563971802fa9c6e3692b78d118a3a2eac534ada345b5ff7d54397586b55893b6c2ae23067f353fea79bb7263c082a3560875e42f90bc90b9c421862734f218b73291f3508f0d4e00506bbc980a2f0bb081daed89bede375931041f04148efe04db3c700abcba9147f77231b858d1768a79c6d7b3887a9d16a69322&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486225099Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default, Name: checkpoints/2048/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2048/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8020e4adf1d43905623353cde02fdc33dad9c65fc248e017275229d018a5e57cf5b4a07fca210f9a470516b148f88c6d1cd4aac5852f5fcedeb97302a8cf7771d4ee970d6a5e66d3a2b7ceb236b5aa786f97490472c124f9e11571a552be320d2ef55823767a75e7093663384f018a1da901c4801166780dd8345e63b7459acc44b43c8f8ff517b801c577a67be371e48be98e783f4be5c71809fd6d583f803f49f1dee1cbb14af2b4ecd82d83b319f9d7154d7e280a25c77fba8526246bb05fcbd72e2be14afa4d74f100f93a27e97165451cdaf231f6b4b0951c56941a4f370fe68a95f57f72fdac34ae6a48bbb278c8a2600bd379d4c0e2a91bc35067afc1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486219791Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0, Name: checkpoints/4800/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4800/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5256f8ba8e1a26907915f59b33aef4b7f79c4c3ab01684f13e791cf357a26bdc1786fabfc2f949c0f785b83aa56f2fb19570b0a38a7d4485141445d6db72182919971e23f201fd4ad497d6f5b2441048dc5052a35add47b5dada669f7987e0c6202f9283f84f33851f8fc0ac54f56a79b9ed051ad009cf35c635b9044d8102262f9f9b70d82678b55e69c54a937e8f5978d3bf24c912ff3a28b4e70b7a5c5726aa20c2237e647bb2af7662a4295180d4895c48f9f28d3bacfbd7326c63d3111a5aca063f806153ca1e82cccfd6c3f08cdcc74be93b7cab20846a18e69aac39e82ddf9483e21652bf2a4edc9506f1b168fe3af832418484b4d620365b1efd94c8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486260968Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520, Name: checkpoints/3520, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3520?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4c31cce9a7f040fab5faeb59cad3881e11d0bddf9de0a7a2c36cfdfadc38780dafc06c41aace2fe58b610e01e709f11875d4fef9e3a0d96608c11b108ff3883e08b73ceeb32b64e37f311a05e009ec8daef7c80c07a9be0363477cec8874d6beb5a31288587ac677de084e6f5e0b201fff583c150238d50e78a69ecebe775d785e6d145ca1b2c4c755e7b1b6618870d0de8d7cf468cd397b75530ae69a354e90bda7f1760404fea3cf8c9abf250053004a3e72e2075e6f1f67d5a47d62a5730d15c9fa7926f1227e6c68be5d7a82114d45f60cde3d561e95b0ae65e751e2e2c4d3971d4d0ce0d6122104417bc85b60ce19d18b1075e93c8d65fdde8a470882d3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486276005Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/d, Name: checkpoints/5184/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5184/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c144c15751330c111b0409a1e8c69262bccb867cdbe8b73fed4d9ba2fd4dcdde3d8f0d2ac48b009f6ecd546ffd0a094a23a04a93977a8d094d29c4ab34b4cc4aac68c72e70828a69c20b7ffa9bd4fbec77f8ae93325a01b2b4e451c6572074d807c5d695adbc1479b3df183e556bb38d77cb65852f551d7dd173e0e2999f688afd3f860808e293488a67ddf0b21252fe5ede16061febd3e217500509c50403c1b17087ceb70380a3bf57738dd3a9a73d5480df38b79e6dd80a6f2cda4c665e46cb6919b61727902bbaae915427a3844f418e6a3c4a046ba5ab3c5a451c967d503f79db875494a4515a3b3f644e1ebae0b93ce01a61e233dc3f6771d828e31651&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486299419Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default, Name: checkpoints/3904/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3904/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=69fab2c198927ee0c74eb4f0faf39c73701a40af4cacc9f48254cb90b6f84105fcd0f8ca1d88d571e0df1cf66604229e72fa764d3f9e3f93f6f1fb51b2367ee23f28c492b18b91c54f87d355e2000596ef6cf96cf6bf333ec1d80e8cd828996adf3719056187746ccb82a32b7f353b277e464d1832e492fc3daf1e597f2ddbd02e453b5dc9db48c2cd239d55cd7f165d227298a1daa162effceac3a86f0e5d40e9cb97309e5a121bd8e19470801d6ea6232f4c51d5720c77ea5dd43bc2ed066ad5657f96ee84eb58c5f3e2e5ed5818048eb1cbd846193847b6ef26024cba4b7b71888a1bdc7cb8c324ab83f55a810000f51e3d8d077602267489740b5e2f9a47&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48632261Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800, Name: checkpoints/4800, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4800?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=97d1b6a446819233130a48aa9e307f6c282df972178c9995c3a1c089eb4f35bd1056d526da1e2f87506c03a8de0116d2cd24501e18e3c2e8caeaeeb9557aa561a6f31cce0c763f97cc3973e7ab8096bbfc2e113a630baa82d200b50d08d14f8bda85b2419a067354f5d769b28149e3b73629cbf0ceea5c64d5be45800e631068b99a206d2629018f2cb3dbf2734487bb6c2938ac80275de12dd19c1f332862f2e31ee48bac3618f5149f05322bc57e950056a7575557bd793f2d6861d3d1e3e0115282f16a3961f40e206b06f8736734dfacc6e5567c3121455441b0510c0f18143d16ce4ad76d606089c82e46e219ae82da95c6eee99a3cce46265006382921&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486370566Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0, Name: checkpoints/1472/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1472/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bab4559bc39e5b9dea606e3b3a90acb4d93dbaa5ed36a0db1c670aed4a3a16f875ffb096cfbd4f1b5cf263db430ecf8e71448a2bd494d3661680e85ae657687488b6561deed85f6d11576034531835ca53a99444d9b4754a6ce9f22297e56df71f0b0ad64158a8e960ea8499cf17532dd8be8d3aca4f09892d2b280bf0833052415c86e7a1586ae012354cf4dfe494969eca70d7f07573dd221448af0bbf8d43eb58f41fb351253034caa274cdae4a26c1616bd2c39e1e6b46e980000a32d146d71524963f9153201ff2da184b1aee9dc1d296b57f4bd103128ea9d52e77de89eff378138d2a52573f16d05e93554ae51be85ada3d54e06187330f9554932e23&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486396177Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768, Name: checkpoints/768, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/768?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=66c415c8a561da28eb9036e5cfe33ee79704d1c13ea0c8e8bddcd9925a219223c1484c7196ee9e5095ee1429623bad83886d2b0b8ee5a46cd992d6b48a841b52ac13eeb392ab22ccbad6df681e176c488fd10fbb6e6ac33f6daefa6cbc786f4fc6daf904eeb90a894537df9db4a26ffb7651e5f331d87417bce4eb3abc14491f47619e29c6051dbb9536faa74788924135bc0f6d46f8527abf60d984706da2b86297e099fd1bdfb8cc12e263432a622f6eeae71236e4e921d558f8f854fa99c2efe53e7bbb3972c21fd76544cf88f82f32cc929fcff74651d50fcee704197059e2a7820521ba6c377865ddba7406e8471e94d37e2739d4699436f435a3bc5699&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486391336Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/d, Name: checkpoints/2496/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2496/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=45be1d98c3d37f621f03f47a99692d62e9a2e3e8167db9aad510b5893a4d23be181d5a89816d86a09f360170a0008a29d15e8bfbeb284c6e942c3aa87b47704f83428176d5f920dcaf40476339f138549e0ece4ed87ca09df12766f1c5ed81550191a7b1e1625fb136982127d629ef6023974ba2eca10ee0b7f700a57098dd74d443ceda6a3f1ab8aa6f796728fc76de90cab07f68e5a4cdfa10fa6b69c0b0e4df9d27bd28d53e49f2aa99530a4f92ced4368340c7ad8bb6eef1b7c4ca633bd06b50fef28298b3e11038d36137abc2bd3842eb57559d6a8f8127eb282effdc751a898f2bf60e26e030cbab0bebb597c5535ed0d6d10cb74dfeb9ff66341feff1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486434822Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default, Name: checkpoints/448/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/448/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b9c165d3e9b88405c9d5f69ec5f9c9aeeca0be5778310de41ccc90abb2d656cf8d331337ed9ee059d0c5bb47aa6b5fdbf5cff823b4d8a5203270f589764ce0ccd6d0961a254bf7ea08c8656f99aa7701e4aba8e51a95cb8ffe4da0d87198255dc37681feba1358cfddd5088d02fda74df1fcb79216b57ac50a11ec197bd03b68c1cc597d9b3bcd9a878e90610dfdf09365a79f92b654de3e3edc5c5993467e6eaed8d8052139be63271723c7846e20015690e3b72ddcc22d5922c0c8597adf13767e69ab9fb984e38479b1725ec45e649ef751de425005b15b584a7d4c7d06c84624856052733227c02c2f6d542604863aea2935490b68b49e5578596fea1a49&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486482052Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default, Name: checkpoints/192/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/192/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=15031f6b49479a8a384ac87d9395d0bb4dab5517880a5d57afa2b6dbc78d84f82292d341cab29ce81dc66fbe4a7b85e52e28ed7374065809caf1eb4a79663eca549934986f7318caca647e78003b3444a40080eece3bfa39314cc80fb221bbafb9c4949439a3bab3d95c6c2a5e16669dcc3b50450af51ffabbf991445a01432578ed42de12f526ba01d14a03827ce869017092fc824d078091a33fad443eff66b8dea615f0d4ab5294955dabe9ee40ea52a26994ef0d68230ec60596023cc646c30844c1269ee6cb5159097a8519fbaa0b1f6fba64ca0d9e389564cbe8a4175c0b192ce538f89c78909e9b421f0bfbbc318841fe5294d29575ffd6f4b8fb2e15&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486526079Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/array_metadatas, Name: checkpoints/2432/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2432/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b74286ed5c2ee0dfd019976e40804b27947479eb6ee2de6c8b3f6a542a2f2aa0e0d35759787334b30f02c636ec96b73a130f0ae5a5b3a64ab8f63e8cacd97289c450bf75200f37bbc39cb4670ea5b91744f2a6c8ba69532e533019806251ef412b122dd45b1b3b4df62aa944892d4a083abde3cd534fcd6dc236c6523c9f69eb873dc99e64653dda98b35434091b7482515fd94fb121d7651027da54fd7a088abfbf13fc39eb28febb90f33e73bdab1502543ed7a6538a7d623b02d178c24c42ddc481811d43d4afecaa472420ac984deb61797cc68647abcf0df0568b49d688b0e8be707494df5cb6f51c6a23021da328f53b053e2d9685dff9dbe8ff8647a0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486542146Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0, Name: checkpoints/4992/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4992/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=924ef8a69ab99fa1579bccf70e44fc3443bfbf03ad59b71d85324dfee6c423a24dfb5e95c5f8a985f6d3d30fa90d63f891db4042590e0f1e76b540af6fcd4532369183c8065189c16405e5338935976b2b7bddef670085f3d9dd9de255ac6a776e5544a440eb5c4f8d33ea32d76851a107ad3eb52442ae6079a671c6d3429b6c3cef720e09ca46564dc8902fa5c63cf44bc2b9d5313d531f03cec5e761100b8a73e5fde82efaa97d981da1a5af617fb41bbfe50c7926d4a576238894dab270f3cd5a6302a94c8744940ccefdb26ebd96c15abc52ae4b227fbdf1cddb4daa8e7aad708ecf7a3809c6999a19c2046f70d248e486924fba06de6976dd4cd1ec7ec0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483000205Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/array_metadatas, Name: checkpoints/256/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/256/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6a61f0f85260c6617e8ab6c402d5771811b9eb0e8017cba46101ba778cfe7b22c3e1363cdcbbace364cad798d04b18030f3aa3e6da6f09696dd4cce4cbfc21dc8afb7ebc4d67f70c84690bd94611cb017e47a823e01aaff8e4301a53deb87c9bfbdb9bccb26d6b7db86c2efb3f731814d7f5daa0ba4140a7530c66bc427cf330c164842f8de9e09720cac6d28aeaf79b532093146c1c67e878b0fdb38eb20618011bab807ce0bd0a54198ba3c5429f8902859c8e54fd52b1d4e282c0da3ed701890b734f86a93527f0c5b43a8808a58a8bed2771cc13df0cf6402698b4ed4c9da6ce83c7f305046a5eac6f9f3c5b7fc93c0ac6e768b78dd475a0bebc1d08bdb1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486370782Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/array_metadatas, Name: checkpoints/2624/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2624/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9de26c81b2248aaa328a81b49eae47365d1e11bd1fd7947e81c1ddd16b5eb9cd1375d2d4778966adf3553e5f72720c4481a9208a31b78193186690dc940ccd640a762f865fab6dee728add00c235cafe7d95f8ebd88760a2dcc2b0fe286acde8248d1bb1a7107fd3f5cb75a0c9c0adf4ee96c2cb023c554a7da17678b7a2d535de76815e663c87e269b11f49b8dc9e025ef5484a3273e10b3e07896a0b016f41048314be4c83ad1271c15bfc4e2e76aa24b217ac5583653b65ca3c33271da1b26bc839fe2119be04a951f623ef3b99f52eecb35b2c61764e4bd77fa012007ec8984b89495fafaf6ace2ba8f059ca32b669c5df6c84362d0df30d9eee473dbccd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48655264Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default, Name: checkpoints/2944/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2944/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8eaf6cb1a9ef5e57ba1ac0294e48bb33530d0a984a526c020348fc0cc73d446b9a478c284269d21d61fe1ccf93a1ddae6b99f4963840ea4f4771907daa0736760b9f79d7049fe42e4683edfb8bd5ec43b2edb1c432c982541cd419689a71063acb179a4641ef3b28008e48a4fb948d578fe266f0eb79dcec01ce64a7502be04503577965eab3301ad87c0a9dbbbb94491f25d4679bc05501ed304ab744cd0e0fab644e4d744fe448341a2957ba846d1d712fe58608500ee90452260f60728c0505a5091a262e19667613bb4fe0bc7af815fcb1d8ed87894660567cdca8fec92b91346e431d6ac0d96ab826e34c5465a27b585d987d0ae4c5a2af799d5d2a20a0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486552637Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0, Name: checkpoints/4352/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4352/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2d17e5bd965107d1e27c45cf81a89f5a501ae13bf7972ce78f7fffbe506e34d86e18650c077ea4a41c6689a0800881ee1c0b82e2d51e9e658ea56017039166f067b5dc0efdd3e1efe9fd2baf493136978493e9ab88437c6717546529322183bf16b99385a8dce1edfacca5397cbef3eca32ecf6d31aacd9ea76fafcba202485f702eabc76003ba97cc63d29738651c58b1d915d67a970b1fa8c7b6d95311ad37c40616fde620bbc0cc2115a8d0f757dacc6373a77b17aa40e9b61a89c9f5d5c15b46f8f9296dc72a35d1038474579e559d511221c6722b47dd587e4612353fabc30f12c16cd24aa85a3171c726b9bd8a18483a021d405c722f4297086ab77d5d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486700615Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/array_metadatas, Name: checkpoints/2240/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2240/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=10ee6aa16a557cf092b93484dabc6b428a30716c3729f75cfcdd673c17241720bff76b0230a0f6967a8fc04cbf272942cd1d31914728fcebbaf26c6f35388ea5e832170fab2f3ac5de61fb4cabe34128da393efcb2f13c160a3e190b5760b1be3399195d7f20c6a89e4c05130994296c3144343dcee5a035e60efbd5cdb3020033ea4e7ae3cc982e199a33cd983c828b16c0ddcf5e71cd52b3ba4d597301071ba50a72fbbfbe268d50c19b100619a27a37bbe24130f240c0ca48bb793dfc89b5bd43de5540152f00471978516f4e60fb61af5e19a1ba7f04f60680e69fa234ecf46ca99a2831f9d3beb2780e45ef27208d570a1db314dde072ab68924bc4d762&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48671644Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/array_metadatas, Name: checkpoints/4736/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4736/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8eb72a7f01ed04f9e0e26355fe27d41bc009d6369da7dc62b110b393f828222c6c3819711207d44020e83d15d08b74bd85546fcf7f8098e1ab71d501b6ba9c76b9ab95a452917130a4a23a3e8084c140c7497ede48c64645c75df60f18229f7a185c74f2627fb9c91d795f097deb1e5f4c482921ca8f10d15e2e61c0d4600dd8cc5ba6befe25daf45e254beaec5b9e590879fd33f318b9dc72c09bde0466507bb1d4b4b0e3aa5294497d701c5ea4e7de525b99233fbc702947b1383150c3341d8979dd9a9ecaadddbb120c3865b245110ea57058a7694009266b3416fd2d29517725371a3005e11924fd23bd9493d3e39c60b60a61f3ebd56c2c66cac5e3486c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486728324Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/array_metadatas, Name: checkpoints/832/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/832/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3b4221edc117fe629cb651908d8a10a2addadd0ecc37c0b11b030e014e99d50e65bdd4f779e44ade36bf9cf3cb075c558b47780c881230746e6af924ae07f4ffb865a4aad46762db5d0fa235fe26632ac4d9ee8f64e8a72cb5d8913db687e21c35be67e7c5109583b92254472d460f387a67cba79434311c81fac55990b06fd7a25b92fa9e3b0f7fcf8068f069461f5cfa2873cb1252e6477d55de868cd833d3138182aeb0250ea8e295f594f98147223367722b25ba10b11c4a6fa73f4da8d361d826395eb7623427baca0a59ee873240c45b6102f183bd7d168bce03c71a7d33bc497d9ab34145fb3115f0f07f9024c4a15c49aa53641fe26808c81037f57f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486770726Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0, Name: checkpoints/64/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/64/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=51b0c94fd01ea34f0eb884937bc835a7ba060b799469decedd5feb1162f29c80b3d79183e83fede2eee2956952913126f6b0b56973bb382410c15b21e37f71d347cdfb4a03cdd45a52a7e0a2dc02efa9690005f4488e3935e0d3f517158db1caf08e1aabb4c445388d7dff4b866c9528c77b4c7fe97dade898c187dd96a40498397ddc4a0a661a232d440071338ff3793d69a9ff2976b2dbf8cf412b153453d965317537933992598638605238ccfe01916b6826b0fe4322b22d422de54fa69650462003e3b7335137db46187eee0dea524a8634e28337cad6915d481b3cf073da5a1023e64468740b4cf38b263c4dc6dafc7d6362282ae38357accbae9c20be&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51590501Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376, Name: checkpoints/5376, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5376?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c30275ffb845251b3adbd1cc6c22ddc81d867d2bfa0f639930a52f7794037afa084e316aa065b4e8e4744119ce91764932edff40041341182d044f2334da59d4c0558193edf98f1c2e9a7b99ebe02a1be43f0c76e2b33b58b69994f8105de0cbcd40816507b0b05f9a0667a5e2045b3a21ece08ff61f01b495aa6b5ddc618c4b43aaeee862ec4a0f6c475d7993391ec1579f5e61c8976404fc5746884dc555e45b2434301fc8da6910077934e38972a94657de4dfea3d39983274dc2d01273150a8c3da446c1944f211573e592d8d77947c75e69f930f8a29566c67d4d3ff27ee7ebca8d11525f1369db3d91430b335b4d5ec0af781393181a5f9b84a93073fc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.515928364Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d, Name: checkpoints/6016/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6016/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=94ba89947e051320a155cc3086fe945057ed28ed49e31cbaa5bb1dea1a8a93a8e80106e4b160e514b877882a0927d857d6aea1654ddff92c3df2323d30356570749c854b329db05e8a0d053639d6c3b6459e059602daf63a75574b133798c9d8cd13674440ccfd45117bc98773ad59c65b607daa4cbb09e869e8518977c720634c8d9288e2974eb91f0463f25dc3e6c86e3f6e35c41551d3e72dc78f6203d79572f08bb0d81c1e5964bd70efc40cfe13a41af73fe3b20ed5093153d95c4786d99b5dc4a3073d58e47197b499ba4d7ea120a372db1fdcfaed3f41130b8cb88d1463b724ceddbfec565cd26aece8ec6bc6cbb4535dff3c3e687073c7d8b9e2519e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486717068Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0, Name: checkpoints/3968/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3968/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2f19bd402d46a6c3c85736e1da0c5a6b799a2d19b2a35733f6bc65314885903abc3e5ab51da88bce35e202d0559ea558dd90a328bdcf6143752ab5de312304116df7133d506f8639952f5a209c49ad9ef7cbd1e3f95c5cdb2daea13168e53d2efb1a3b24acbc70cca60203dd044d53d9b1608f9e04dad8846c222e93bde522cbcaba16c51fb081d2637e287fa78055063fab565ddf2cc46a3bce3ed4ac58e50f4493aa465806051c7490d91bda47f67e4f9c5ea528563e3945409ddef60a6505f4370252e8740bce167a1f1fba98bcb5d54f181daa3311fe734cf7121178857bcb404d775291d60c0ae1e32b0e1a5ccdb464ed6a00437cd10881cde61f937a16&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516114933Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default, Name: checkpoints/3968/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3968/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1e07b76cb07ac8dfc8ad7fcca72645ef7625055358a7e1ec6cffb745f5f6bb2153da30a63523e8e64305242d53d33436ef965f63027b0d586eba076075068fd44bfeb0b4cce3416583527ed5d7682c9386cd0d5276a5fd49d6b41ae894eef8a42a4562f6c1f8584913b99ce956af5b65416645dc52c3fbb4fc6b8fc139379722d4338c6e79bc82eb0d87c742aefa15b0db4dc0195419f5e3fe93bda2a320407d658e8dbbdf0bc3a6570648649447a52da4d42b7ce4ca6349eb68e9b490068571b242e78f5826c1eb2c93b607415aaaa9c9a13ae5b7386a6e744cef8c321c31236d2b01c29ee941f6b0322e815cbe91ddca3c9266b4aadb4e75faf6ae742a91d2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486760921Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728, Name: checkpoints/1728, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1728?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b9243b9cf5205a1e9b2af4110859aa3c8aad94243dce4356d868db468642efb3ff9bb823592b9a5573a03fdc1d0261088b1d9af82fba2d01d0546790c90c541136b08a64506e8c2faceb8663a5a922f9b28a1e4bbb17ebed5cb40fa5ed3ecd0b6e333aa7a4a70db8c323c94ed3e78b9aee4ebbd403e5c48fd1f5699d4c81b06b28042a0ba3c0f292d6debc6d8094f84fc777250e4ab2dad30ad624b274bd763e49f0da851e6a41d43966052e29d22f2567e2198b6730d9b45fcd0127d18f10e82a02083e1b2f2fc65f6bd19f1b4833cddcf00c5d3f3d6cc92927bc2c3396df34b3fb4a11cf7beaa74220965ff5dfa4c48c88d81000d4158a752c50354d24f560&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.484041045Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d, Name: checkpoints/1088/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1088/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4df33941bc6a458d28475dde6913841dd2193c7de652c5819b3580e2b6fc40688dba9ab0e835ee5d744acf31029f1a03e706dc1cac25fda60e47357c096a29d3406f1c5dfd58b9df2678132862bc49915a4c445bdbfb3661b51c4fa6298425a51513e48828a8e2e3ef62b084311d4b7535b20062e8b6cc670e0ae29f0fa6e532eeeda8c2a2d9da197a020027e38d906699bf87eccec7e8eb02ea1d0f5f2959e672e1f59607573f1b6b13f08dd1f263ec2d31bacfaf1cd098a7715edebef18531c8603b24ba3f7cb3ccddfec735a75e7155976af8c01a1ad1dca01ead807b20a085d05e3f5ea9dcb7cd8e3441d6066716efe2d7974295bfc5b3994e5fda773c17&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486300587Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0, Name: checkpoints/5568/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5568/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=17959f04218def508f3406ec2a14859ad999aa6ca4fa63feafbf479dcd003701f500647568978301bc896aaf1ca2968502717a6d8cc373d4bbecb59e9e54c543ecd22e668752e62cfdb26323a8abbd195b9627f2dcabe81f0c584322b0f1f8fc41234ca48eb6ef151aa744fb7e18a390d2e10c4a3b52c007dd4b5e06da400b7bfc5d4853ab538cdf18c405aed31699421c25a4a6df03c5bd1090eb3c4651e090a86e3f32ff4c1d29c82f4bbda96a21516f9a7b12115698b572d1dea81ed5befaed9d735f8d07b17844ceb8f01fc002364533935f14ea003a3cb6204c4d908ad253ce097b56ae979aadaeec6b9de6e5057807f5d9093856eff3439185dbc021f4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486696422Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/array_metadatas, Name: checkpoints/1280/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1280/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9513af39cc0a22d446dedce010d1b165623e05f4364e818c020ed99de578e6e4d9107c4029c6e0b87259c3ff658b432b7cf61b7ca86252ce44b2562652df6f9cfefcfdb15f983258a591bda6a186a0f4967465d33889dfd84ba6ccec035b1b628b4a6c159e2e3f6a575787d31ba26d29b2f834791067364ef7e99a0224ee36ac8fbcb149f2655e3724d81647d6775a86eb150b91e92524651fbdc250ff4b8e8a0fdce06482a589e7c46255a3673628f907f8ce73d0544b0f4884390583293bab24e1abbbe2cef5620faa95e2c7dba19b980a65a9d06119da66b8e214f9eadf3b5576ebc1f973d5f07822b1b914e92efb887bc1f5c9d552e5844d6f41accface4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483021963Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default, Name: checkpoints/4800/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4800/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2f1e695460c29b2dcb3c48e024d4a05394e305ba1855323171b856f0136e9377891cec9f3f27ed68a8a8976488e32839f2c6234785e85ed7a590be3acb5ea2e17ef962d8ed7d5a2b80ba43f09054dc6807ad14301bc2516ab5a37c4f271254faa418cbf1fa036e67d1a004b62d6c25567833ce8bcc711236aa462f7259e172655bf81bfcc512c1f8d38722e36684a002be6a96402c71a8c74562a807c13430050140df306ef2af57bc7d6474fcfbf84467508743725c1686f80a677b4de519185f81bffe30449cd871766d1a5fcbe3135f6e9e75ddfffc6a252d760da1c0586155667f97be8c1b6440c92e0cd5fc2dcf098d713304ebfe0920c075e73433feaa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482875339Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240, Name: checkpoints/2240, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2240?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b4ad006d7c2d8ad2179ad28997af0b6dddbbbdb749a85917c4ac88274dc24ce95591d5f576c761ce4d62eb10a03ea6fcc1237dc96a9294e9d089713a9fd293926b6a4b407fdb208dbf22f94553db7737c381031c98195d07d6c628401861d28f4e28e70eb5e0ab0d48bb59a10b83dfd60d41195e5943d0159af7baaaa647d2b83f73958a8ebcaa5a2e19f2e88bb308cc2c9d5c972c50c96296a244deba9474e776fff1bb489e759fd794522afea4102ac2e06964e7a0e32972a1ab0c40a0a7d624224f1451d657f8cf0a614aad674203dc8d8897e45a2ff39de628089cbe5b0574a7d0afe537133acbc55774e07716963b7ec7be9c8224315fca8684fce69ff2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.4830336Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/d, Name: checkpoints/1216/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1216/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=39fc44a8dbf48711fc447dde21ab82e8736d90e61edb5d630cfe4f52d9342883478c23131eaade4a8685609237ac0edc045e5137e2fbeae773c0433b41e2fc636b15affc1510d322dac567bb70e7a47702b7a13ede0b0e10b8d55d45a76f7477e0e0c6ea913aa53b4ff79d72086d7880b845ccea3cc6939152ef538654406186c34fa511fc3a99b2b5da5214890da0e527a2acf1ae7e35073c37cc58fde0ca3d2e12954fdce0fce7ca4386c07e8477f6ccb32e4f42d345ed72637a1594b3a1dcbd403ba462053f4c929612f9d1f2e60b36d16b3d948acafba2a471485853f5ed0fe9e0894e5d56a376fade07dd3e3c6af408c8eb1681edfde74b881e8424adf5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483013447Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d, Name: checkpoints/4544/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4544/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a5e63ff0a0805af7cd3a557749ca1fb5844ceb4d9a4df31d2afc0822f85c1b34fe9277845f4ff60c54fd0570d0cf98b061c3464de41adb2c21bb559000102c445e78de0568db08ba4675db9ef69febb4921fe2be3baa8a264bc30cee4869e364fd1bcfc11bf0fe8726613196b013eb06931ed66c7e41c5915df37a70172eec5f42521a745d5d52a24ca3097009ef82e826e52e2b7cbce60c1dfb8f537f72299c0f355e1591ec24b3df01dcdaf0abbf756404e23fa9251c24e7cf0ee1d0de70ab35a3cb430decf086559e1cbe1f61d45c7428ad8292a53dcdb4221522ca3f9c84388a7a86b8302169ce2074ed2409eab1b8a2828bdc218163e15b3000b5dabf8c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48303618Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/array_metadatas, Name: checkpoints/576/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/576/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5189e8580cc722d8715d3db63ab0d4d321f5fb1c89f26d3d67569ea69ac45498107f573bb8f56097f63381f5f9d85e6b5a9a50c637348de521226daf5d5fee805d7e12a7fc82613bd1c5e32619f937ba81f6d8e1d8ce88e7e9c8e4a54b020c66e9497709f68b6c0162b5cec28fe54ff0a85629035cb706f5f79856bcfffc07e5d4a5a56a43373dc0180af756425a36c88f9cca9b8fdead3e1b541406afa05c18657a51533610a0f7b80bb0173497e137aee2d8a547a0c56467791eb2b1d777b8db48115017e9c2888c7b9617b650c1636b7f03ace2e842235524cb944cadbb5323e2faf86f352bb40cf43771a4be99b164ceaae31a245ab074147bb1b2b48002&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51652616Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d, Name: checkpoints/3392/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3392/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b64071164fc2b4c1141c58d058c93dbc8c8677ec436c5675a70e3b18e246bb173bf8d373d9e781f889c680f3b0b6e9e8070278d4b4795bc25caab7b134a5b67252593c50702d52fd9cf44518e03c007dd0f1facec9e11e727cfb8698c5ade3878e91c086f8ad9f47c2fb397a17266059d8d9a0af11f5b0100cefe126ba1427010de3dabe223ff613aaa1b46d042b6a8da88369dc66614859b224944556b2a8a8ebd6cf3315b19617dfcfef291f6fb3c1fb69fc11e9c9b318bb5f99848d4bd41fa81ea2a04b3da0e5ed4d9e966507e85ebe41961ad58cb85c9f9c708935d6c28765c90fd924ed3d4f404b72c4c2deb742b5cb5396a737e383e02ace382421e88d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516566537Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0, Name: checkpoints/5440/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5440/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5b956654e9dca43dead6f7d4752bc2f526b986bb6cc3dbc75aa7546946a4892b8d7d2658ebbcc2699fd2f0dde864dfd0c0b43c6c8a95b1bd60a6a5aa34aa9dca635ad12cc73fff5293d772aaf68f88b921f679ec38abdda6f29e821702df195d9cff89476efa7ebb06a6e37798ed425af8aa980f1453bdbb73042deca289457eca959d0a1707b097721915234c9c1e19057fbac3885c8767740a6d9795d37ed7687dac436cadcc210969bab232800d9e6b5621a1c1c1007e01e40e3930f713a02d86bf15290912958fcf58280e30b664b228a4d55d08ba9c08d9b6f4e99275158a29a8f3b7247a38d8434d1cc8213b968f4fca331dd0e5f8ef8e027b8f6a398e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516583188Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/array_metadatas, Name: checkpoints/5312/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5312/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=390cef52fbd826e4624d2bb31ed11e2020340ebf0e00e333813efef48af5288a0eefa8d7e86feece970c7eff7395b331e6406f510375240a9234e6aeabdcf9547e529a2a36f8ab42b037dbd5e802f77ec4d0d54afd9fa1edc508511439ae0035b8c7f6a39a13a979a20155e1b99f66055db64f59b1c5322e192a17f771e35f0088a78c9d111e29990c64565539209fbe555cbc20ecf64cdce2c1072ae7bb4af99bb46cec7284f898827380042e1bb33c76ae8a295bf28f8127f15dbe8d2e58660ae10afdbec0131a7a95a2e56f49bcdcd6cf7478038bbf334fb37bea8fa9eacfc45c65ce451537bcee598695d27f4ed2eec180490dd9912fb7c999d0f0dba9fd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516546159Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d, Name: checkpoints/2176/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2176/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3ddbf06519fd149ef5fb7c3a123b720ef5514f01ff20b3ed403fcf15006fcfceee94dae4073e21e84b6d9ef8a286236059b3b0e9e5a1ae48e339447de577a0674bb821c59e200de8a3f41c2171305cc3892b219fc00f59ee1d8686665e9d84d7b9b805478f8b5da7bb9bfc79d0de0f197d5a8f1b12d4bd7f64968c313cb297e9d77ef6f061494823011bd71a8d72ae424377c18e50ae3e8877545510d6987931be73e13566463ad16d6d74879763da0087318538f488848c4fb8645989a49f2f9a480b49d66a00286ece45ef40268912f242fce6224a830136874469679df97cfb9b633ab867f48e431a28c1752e5da4baa770130225d9f464f2a48ae05a0f85&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516652001Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d, Name: checkpoints/1664/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1664/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=21994318be51283315ab958b78c40afd725240105e3bb3f97dae81d072f211da772561d054fddaed130b1d231a6d55b79a940e1e6a13cb847d6843be5b0c468e420de944d37451330866db6374b95d027f7943ee59ca142e8d550a9e026446083c97a38328e66b4b2b2e171f973732669a6505510c7f7a970ceee4ce934e702a58a89bd5b19620de487524bcf6f1657b8429a3b3b07a4680fcc8233a949a5b7d2ee8c1ad46a63ce8988b150a80258bc70935e6efb6728a970ae54c27a5188cb7a003b8a534797089b42203ee3d67e87fc33256f676ceb4ea923c033b4952f9c2b10f346263a46af11fa21517936b4bbf85ab6878ef4fb0e07a98447709f438e1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51667696Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984, Name: checkpoints/1984, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1984?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=230eddc18f62e7ab562e7a061e72a98293c57ccae5add8a528f7ad05cc38854922661ca1355664faecb112faba58a64f9136fb116c3a0b5ed0894d07afbb35e8873bb2bf73f50cdc2ad815ccd7bc07bbbe3daacc97bff0ab3444601ced250b02b7cc828d7753f4f4fe1434e9f1bb74df7dc8dea7f5a0950f010f04404e1f87d6120c8f198cb7980444908182541edf366c77bc2913ede47ceba7a52f5b275bf0d0b442849cb37b3fcdc3a40e64058a5f647a539eb0fc85779f8b658eb8923725a6cfd4c54094307befaaa5684ff8cac4e69c929ae2cbf362a3e88c9eb6bf1c9254504b4c0474b689b4a1626534930c3d52858fc3ad09f75f45254b344514f0a8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516752555Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/d, Name: checkpoints/5952/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5952/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1b76a53e7bacb7a796ece60c1c25365d86dd0328f7a23068d2028f8352e55a382164c7c93bc3d6e3162057e197fbb6b290dde1e0697c994ce96e0554570cc2d8983d0bcf2f0a2406beb1cedd641d62898be32a19a9f8e17966e0199292d8f6b030977e260600f799bdbfc685f0b34e6d513bc105dbc0320bb53b6e6e1fb66b57a0a0e99fc99d346b9bf00fd4a4f36bc967b9fa4ef180668a6fdc8998d352a83cb05559f879e1cc6fb826930f381b154378504a1ef84a2ab26b103d21878f88a0dc9ed0533fe41ea1b7622f4890c40a53b9601f710af0a554b506e715ef92fffb15162aca00589b105277976bb48d9163fe2b9f29e2973ecaf50aa1f966ef4d20&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516908476Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/d, Name: checkpoints/4288/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4288/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4a467f9e9ecd0f347c51934729f1837a75c64ba5c4bc47694b91558dec96af9cbec4a5a26407d3aaa853df71449397ca542caac23bfc2fdfb004c3169434d3dd33c464083ce2534b6b41a87a83fd226d6495f9457ce02168be8b5aac01a8284c31fd4db14ee229c171f718d409f7d47b97ee15e9c339dc71adbd963422362c01f3baffe8d6bfa2d22c12e548e99e0ed64b81fc89075a8c01707a2f4451884ad6c67899742e9aaa12d11b832350d15dd07ac89a24a2ae101408b1f972a553a429a08ba44cdbb7dfcebc83cc2958a4ed06d58a6cb7956622174e6ee12230941f69c4b23ed2cfa960681923dd53b7212321a6bf7aee26cc3d860f5339a84e4e9592&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517210967Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/array_metadatas, Name: checkpoints/448/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/448/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=196d94e86e7afec98e88dbe077ca59e5ffce5f7b4c1bfa503d40b6689c1364f9639e62b26f7987c4866576c785ee72b3a3b926fdd2a42b4046bf72ae3189360f2b885fd34b4a80bd3f69563e4ed11319496274b2316073a6d5604bde94abc0a45053e50aa72b1120281cac26284be0f71f0f75b2cbbcf4b74cb829a04f53f50d4d87298c8620ce409e925f660c23190fb2e6f79d6f3d25c8f3e499e149f4faff2599d64de44b70ea2d0195febc72ca305b7050b1b3d43ebce923f4572f25ac57df04aafeb3b6fd0c4d10cba0bec1c46f62ee28bc82e51766eed878a8f062f5cd531e2271aef957db802160b76fa945414442173e9a257fda96583865a0a075fd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517209778Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default, Name: checkpoints/704/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/704/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=648253e663a8e3973c461b1a921fb23b69443da5c4feae22834cbd51d69a80f9a151487411edbd116c899689bc66ad646daf43c08aed545a42ea35f89b215e3f3f798b787b728bd6b15d9a84fab694770721e571327f2dd3333abf378a437db8a095789b6e6d91a4e3ca2745a44709851f43549955ea529dd7963a990fed0e075051691730b31b99c566443bb1be18bd7b941b1960b9174cf7bf5df7ca3878a3ac859cebf2273564d110afea4e78d66aed9ca644b1dc01b4b6140f78a6f16f243595253a21408ca19b6982383de16c94722cab90854289ebfce4bbf208c07333927bdf5c9fd3634cda879ce923fcb6bf3d399e1a5537b6a294d051e38540e562&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517216379Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d, Name: checkpoints/4160/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4160/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=63b6ec8d3b5c4c6bba77c0760af9e4ca87255c935c12a8119b77902b0a83846fbde86e8e6f9eafe2e0cb18485feeb61427b34f4856d3b8dfdf7b753ee7f834c5c661d6971808f549f98cfb6b9b1e98970ac5cf3c03991c8ac2a307dceb7b581e812d626cadbacc24b6f6c95cf2124ee8d8637adc0062793d2d49553eb7b87e912ef608573d85b65abb8d00c81b35f12d746f44ccb78716a652d7cfdbd4f031ea9faa1f8af8dbb40789d601aad33987da8156168609341ed6f8862906cb5f62b587d662ab45dadcb24b15148b22f4c7294ed26498aab3d4bc393cc6ed6826a944cac7f5ae5d302a1fe9032238e1e4abfc906295c293843fe778e1658ee777b183&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517251643Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008, Name: checkpoints/3008, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3008?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=68726ae305f10a02b8cb9fd08dc3060b16ac51a12fe48bfeb1a20b7843e516c933f7c9283f074ba3b403d4631770a49457bd40304eb26efe61c18ea99c924221dba3f80c4619676d7f8986eeffb6ae9e9a22ad1481c95ebde4951c53fbe1866b9ebe91fbaa07521780dd568ed60aaa737defdc7dce62a8648cd5570b7b0d332994c62be224c5f72d0a8016117615a34c30a03bb7004a7318a429d81cb79838b9ea8b33cd5bb4113870e4a317ba6cbe5da9675a107511fedd13fcdcb054ca94fc886b6d85ec154fecaf6b621f01b3a944b23c85f81b04358e50fd980f9f99eea253312fade9b3f2cb78c204750a27fcd3fb28f8753fa11a33124980787d69fd1b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517238425Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default, Name: checkpoints/4608/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4608/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2249bf81011a484da9e6fb322b343e0a38712d6b76c219193c3926b0fd7916a58974ef864d6a6dcedf790dc9747761dbcbe0b34b3a097c7c8d0dff3a4d8d0d5c5a5a615903f903f06be39ea366971c3ed15bfae34289a433d7943a761679167dc952bce57a9b361226ba47d503cefefedff6669fa67e2d6754e186c8bec596a790f1abcbdd8203c858a88281107fe7ea8bac0a4f2da07efd85510c0b195a33cbcb11e6a84045f3c6c717da2275af3cfec2647a383047e364d7736255966b8beedb0ee907cb31ff0dc6ec09f0498841a03860dbbca0c82bb03739d6928faf6c9446562ea31bdf696b79e7bd4dc8cf11a39f992cc4485f030d3031213f0f0a4132&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517347438Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0, Name: checkpoints/3072/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3072/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bc255dbde6ed3d781bac2b4eb2eb995474a032ae43b9541f8d3bc3336fb82764fd651ab9ffef11690a5bc146087e585e0e21e615b9af6f6c30949ed647b164be3b9e6f52aa43d22662308abfe6399075a6c5b1b4148b9e366f4c73278bb61023b592b02058fa05e378f50f323dbd8091352df1dff99e1929a0347d4a01fcf99ae191d416f85db900e34b5f6aaf8d04ed2738adec990b763f08a122bf8f5267b9c6ce94c5e5355d723b430bdfa443591ea9d59166f8afa0c6f7f65500cde8d3dc3373a672133df8d8585ac47ced12349a4c3902310c2b820f61891652d18b31ca3891f9a19011f8e3000bc85be6b039bc27cbb0f749c77d3ac3f332b5dd7169e7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517355033Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default, Name: checkpoints/4352/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4352/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4d3549df679c1508ceeb6b06b69d5b6e1ddfa155ba8c59260e4888e849a05c0240605d36917eb9c2ff178981801536fb9b3aaffde39c0f4521668955bb6df8b7e8f7f6f6e7ac9b033896b185e8c42a802887ce964d87fd3a1d2077fb3ac998c97d3b4db6eafae193b0b969dce48c6caf8f650dc275d8a27b1e676d877cccb9d056b7387f4538b52a7049059f28550720b0a0886fa11b987a029be95ef4cf631f6bba7f5bc3c08eb5a59803fbe5670c133cbcc126bf49e78dba2872dbaa21e4543db0672f47370b0c34be2b144301defbe8e15dfa6ad1d159be079b872ef8e8d888371b7a5e466e9d73ce7722f7aef1a52d175c0b1653276824b5a1bf9c6937a5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517291838Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d, Name: checkpoints/3840/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3840/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1eea23b53fffd0204c599e950dbf74791724e65fde6f32b03f408d7ea4010ff15f23f463580877ce0fbbee8b3085f40dd6e155efbd996f422edd4a21a7aaadccca7e7baf34c76a3784a3be4c91ea1eff57341c35533d8b75ba3bb263d55cb661f8c65cb66e403cdb889938e0d6b6debbb2ecf6f8742be2ac62050e7124317f583e251ecf90e918ae8cac26c32114db0cc2b049ab16a2ef25193835b961d6cc17b381d2e31aadf7c78c275e937635078460db8f94c458df2c785a583d797c7cb0b371f503f6bb50ba349e4830d1d3e19afad1c97d704ffa1118641d48c59d3de4097343e5d78b0dc4034fd7bdc74405a0defe401abc921189654fc0858f286580&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517421161Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d, Name: checkpoints/0/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/0/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bd82c7936341b75563bb1f1b04bbad074ad1b685af41cf13d8b4f199d34565795f735ff37c1ca703b6e980de55356f2e103c2a621d70a00e59b083d6186be728c1a980cd28bcfa9b3596f8649e72ef2560015db9d2b09d65103e7a41b4ca3758529edf6137bab35a395c3bd02408a483659fe1b69a7cfa0be7de52e859fc168c867835ec764a82e6483ad2808fa0f0f62684444b1c56dd8776c9f6733ffb06bf43e45d13481904b7a1fbead49ebd70aabe474c08134a5b521384d482493026268872fba15539168980b6806294a04e481adb6bec1fdf52db2ed14802d7273269b88612f07d0a99e9faec08a0b7ffd9021b6e03fe415376d6fe8bec0a3c520f24&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517299726Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0, Name: checkpoints/3200/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3200/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2be87a8c4d47524eee12ae3fa9700af46fe3f6dc6b240f610d41efa57c8eed540dacd44a2fc330aae07a4295a4b2860d36b1534dad5030e89e92a7d1ce6e7b2ca5381fa05728a1be25a22d6208d574a5ab5de9ba7614c64563ad26db1024922de4ca8370c780ee0fa88ab9f4e9af6d7eca4f914170e35030df4bfe0c010a7629a6c65135d1b8d17e1500cd00d6604423189c13a0ffb81126a1feb2563234688c71197ae80424782dfa12e0fda52c771c9cc5c94dab73eb0737dd4dc390c6ac307763208691ed172dff69468f6f8de595c921bcd97b13fae950c9b137338aa0d14d1b08ed4db53cf009e9f099848fabef2e1850ab0ab950e46f51591bf83426f9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517441595Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0, Name: checkpoints/4288/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4288/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2cb2351d0cf1b6a011766ca3b06adb69086ccb626c2b376892332d9f9596090bc19b9b0cd2c43bcec3c5bc0551d59b1e47e2102f483b8411e1d1a2a3e7fa0eb4f27f3de2d5e83579eaf5f2ee1dbc28cfdee562709e1d68af68b33db80e6a16f58488e827eafd4860d4126845148d8fa1baa8147fccde854c9f13cfe0029e3fefdbd3844f96dd8ca52d5409dd43fcacf46f284b4c66a779167aa3a10dd694d902fa076273cc1158bbb0e17cdb57b0a7127211603e54955e6bd8e35e389bc0d7535a68d76b22994bd1242ed443946c383241d09ba89eced710e45b00f76fad2bf5a3439029a804b2bba48d4ed5991fa355bd20c2afa98f54e5c9bfb3ca9503af0f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51746464Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/d, Name: checkpoints/192/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/192/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=07a8aefcae0b783a5370d3864eccae2fe4e7d6919c18a10c7632c3ff0f553e01b4069a91bf25625fd554b72be964f3fd036d2158e6f942b300347033703905bfe8748f0b6995dc39eb5ec8e8c3f5e43ebe688da19da3c8ba6001528bdd51fcb83c3c26362234806c30ad03c6e6c828a572469e6875be550fa0a92df74fb188dbcb9a277aa8b7a354d1a80517c9df0d3e052549db0928d0d617f7864664469d87afd6e240799fb89316cc02fcbf9f2a1dc2407e943039fa3ad05fb6787003a976b945ccf45d56c0fe807d12266608ebcdd056ae3ab76f8bf028eee45fc2e31c2f1dae24bd35bb9996727aeba1ac474dfa5c72817ff094007c4f5598f128acf7b6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517479151Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default, Name: checkpoints/2240/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2240/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=109aa3a935322231170382e1c3446058e097ecdc8861771c06cadd8ca8afdd4981455ad24fef5dea308d573a77ded784577dcf6fc17d4278cd1aaaf134cf564bbaa4e727ce0cf72d7ab0624f9b61f758c69542ece4f474ec8b6dd43d259292e3aa7821e9dc780ec921468a288eaa8c83373f73dfa8d39a8fd448eea9f57045bf64e6b861d280da18a425675da6363122151c0a32549f1fa35f610440a941009b45a0e4af62e5bee9566598ca44380d1dc528388ff275fbe6a1d54bfbe3a887179e3e75a109cd2239e002602851a2adcb695e26be2f4f3346da96b1c63aa85274aa62546101a145796692f7a2fd40e8ec3d73b6a37d1142359f0445d4c32b91cb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517540889Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/d, Name: checkpoints/4928/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4928/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4616c4a7e267facf37d10efeadd3320efe9f05c2f1dc775ae015157bf86f9ad849adfc94506b3110899912c31f7c8a768064269c89eb19f255f3152d4ceed01876f056466fbb72d7cb8031e49ee86db14882ba07a91849cf95bb4cd9b32fbeb0c4c9cf0be0c35f2fbe603ffca47db40c805a588aba3d340461ced4674b7b85292d64f6cc284fd7a4099685754f8d149e7978fa78d3e6d0e39da393049502c2fc9e4f035aa261a2960acbdc3f68a0781909ed1908d7dadc9dc7a93a4617ed724c5c6a06a73fba3af3ef827f3fbb5e35c92637ffe6c5932162683953fe7f234f6e7f453cbcc4b4a61619e25a81773a23fef7dc8bebf6b1d3e77c9707f7372898e8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51718866Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d, Name: checkpoints/2048/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2048/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2110118428f5b72df8cb87b967b8de093ab1effc5deb002b7a68ed8edcfc2facf324a8b5c081f3957ceeb4f7a09b6386506f9b9608d796dfa967963dafd16a0df79a42f7e14628429571f223c7a1c1213f0d9b94c586784889fc5429a1efb4915edfc6a445d2f78244966fcf83c565cf23d501c157acfdeb30021ca8d57e6d01adab4a912a8b557ad2d1e84d36e84d7ee6b2ea45e20ad02cfeb0bb84429f2bde06159f160c88232b0b12f65eae3a319142311c0687ed80a8bf0ad8c0205815718776829df9f140975ad33243b8ea64a5b655f79cc6f168f146591ee4ccbef0473678fe85f607b9749f0b755a2a4d6cef0e84c1aef2bb5cf87437b56599802481&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517466336Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0, Name: checkpoints/5888/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5888/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=754a95a49316451888edd1594dbf8e6b202a70c0f14c195408490f8439d0afb12f9a2623c4d4aa79e46a551bbacc60fd7f6d954c87b1d3af90576b26b223523b653ae30ce97cb53e53aead1ae07cd366d8ad6aea79d3007f768792c7fd8c138923ac87c49374d6663755be61bfff874a4b688fae0deff511b66ad514dad13a356b0f5377280c41aca6ed04eb1cfdd88f26d0f89a11070509c4ad2d129c1324e3914309990fc853f03d769d41f2e79d9dff0f770ca0253748357a71dcdc278568eafdc918aaa34c30c787cfaa9c254fbe22cb978b0153867a31a792ac0720bc91bf2510ec10d17c7d3c5740efa90dd4834c672a3ff38d9f49cab1b7085f81daf4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517595999Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880, Name: checkpoints/2880, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2880?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bc93cd965dc49f51207e9ba42ce214b5d467d49f17cb87fbbdd2def59424e029a792d0b26e6679b5301793c6a198d2abbdfb4b01c45c6e6dcab9063b1203534900e14cc26c3e6f76c7ed377080c25c4be50e75d498ad38eeea9afcc5d160b24df905d4caf3c80019337a407a8b1d651723901e3d62f108212eda7481f8828dddd2dbadf695dccb94b2ed5869d54b7b459345d70538e6c5356eaa3c5dab5d01c839f1e39ecf5902005fa66d87705634f0e5888dd6a2bb49a503e9c93e9423baca0b38c167de41307078e909622608433fba0d368b4cd9aee0ae7c3f97bd862fc46c1c6ca2fc4fca72b9d1ea513ecb92a878fa979bd60546cd1a941805144554d2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517541397Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576, Name: checkpoints/576, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/576?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b5cf30d08477bc74383e3870f7b844b039eabc06999e8aba5ba7b3cb85d687a72c2d64b0f5a83ca5bae62d920c770deeb52f64e6edf0ff97f9566005a34abe7a341db9509924060b7829a08b836006c062919453551a3fecd12ce222fc319fd250a3ad4f2cda831cabe3e5fd73fad056a5dcabf72c17543bfa58b7f3a9783e7678f55c588483b8ea3ee50672742e64fe8dd45a3fe4e65c19579d60530bb82c3608b8395c127652761379f2e1560c12dba57bdca00ba3d161fecb79b38b47a5a2f443d2a37a205e933f2647ce16ef962ac91c7b2fbfb0a2f61ee0efb3d9ed2cca7bb75a6c41c6d13e45c61ed2d8a189dd3f38cd4e85f274bb1b8eb2314e77a162&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517698006Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0, Name: checkpoints/1920/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1920/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=35c2c41d6d57444f17c694af644798f10212737ff87769753cfc85bb5e22530ca6a5ead32804e05c4a1cf841ea2b09567bde28e33935559f812d89b76443850945eab01a0e12357dc9bc051354931f9768f5470c1cd431bb1924572acd4ee546cf8fba58ee95cf68ef4feb08f6fb0454bd8b55ed63dd66ee0ac17aa67edf5c0bbd08d7b16ef8330dbd2049f08a8aca138d7d2312e3fc52b7055c4e9ee9a94c21e71d3a08c21e3176de7942511a077caa4623112dcc553acb73aedfac09862beef750e845a8e0a29c27527816c2f2ab14d310205ac74b5d4a321b138a2d6c338cbe3ebc99e8307775d9f0453dfd595a3f9dcea56ae3c58ee193de2f0548afedb9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517748258Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/d, Name: checkpoints/3072/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3072/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7d2cbb5128dad500bc006d455c3c312544a798e3651901c6e1c7d5fcc2f337bec8534fd051026351627c976dfcb517ef539a505cfe7df7fb5f170e26d4b472718cea745481cf997ce6d05c2c9f38b4809cd56ff0aa2b3a6f0d1ede8fbee9049cc4ec1eb5dcd366e540a6776cbf98e25417336af3655417d8988a27620b6f07315f107a2f1ae4c1ba24b692c23d5253b8a6fb0e5715f676064bfe29640a756db06e2aee0bd6150d079a0be249552715f8a4596c64a906b8850c93dd26a70cb0f77d6188f21c04e9bb22ffd8ea210a75801fdce507a31bd1a87a6dde4e2a5a4bd06bbe1d1ec1db19a24a8c7f3887df4cd1e2dbdf5a9cbe49bb908165c44fcfdb1d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519186074Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/array_metadatas, Name: checkpoints/2112/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2112/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=14a5d20a4acce812555a39f9404a7b655aa097f54c302faafa550beb7dc16e651ed56337c798a915f7e696f080dcffc465125fd0b744f428a0f57eaf6977598f63a377295e5009eb11d71bd259f8ce3630bb6019ac8344643f490e5153a23c2dbaf8a25ec9f0dec805596903914211da9e157ae32a8578d58543d837bcf0ab2669f224bfc863e097d64f8b6136b7f35537b4aeba1ab6457e2c3e05768b8c8f9a2684437b4831d99cd5918bf564c9858849e5b92a067bb2e5d9ba871d7ee22e7cda7604d540f46bb2b6d99bd1e74da3d87c989740a83cc6ff93fffd382cf4511a33eb7f3918ff59e87917e6be763656b2b71950fd0d7299dc914ce5fa91dbd453&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519227723Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/array_metadatas, Name: checkpoints/4480/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4480/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8e2889c2a4da9085f3ad55bf61af8261b536447199ce351c89e6c55cc85f032fe79d405a1ed937b66db68968c9763bc6af8ac64ad43732ab2e3f828e2470482e532dc49e5c616e441b1901970f339892935dd383ec5617ab6088d8e4a0de72cbdf74065d1b5ee908120afdce1d40d90f90e364ef1a0e9562c11e671a4feb826930601acf54b857687b64f10cfe1c93a56d5265191545adcecaac130ecf4b0e20b98e6b534b65041dc89d4371b67ac86f75792705b9f121683a1bd1e89eecbf26c1a8bb35070070109dea00fdea5e13e2c539410433f58d66f57ddf4b95845c94ed9a75c0a3d262a7db713bcd07c438899dfaef39b28829000e5f9a2074c7e961&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51927176Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248, Name: checkpoints/5248, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5248?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=181ac7cc3b3b9513f589dbd080a4c4c86f7ea46e0a60fae813fac4dc215f91d6a0f144dfe5e8942557c08f9db1851de8a19ddd64fa2869d6ddf33239c667806ccfb11ff01091900407d811012729bf9369036dfc3b5dbaf6312d407f283660ad58c7dae440f502ef47f29e1d6c67e043e3dc90f6487c60671d540b953688328379ec78b66236dcbd77e53c6c93cb8eba416af968f1caf62f4aa384e256ee390b33f2c41dbb2c6d9864feebf084786484a570974649c2eea3d5e84b20c7c3a5403d58169dd1f42e4d17d897407bdfa8c1807c059084bba40a36bb539f1dd89ffe8c9c7b7ef9cf712ef30497ceed84ec7073138a17614624bae471ac7f132528f2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517673181Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0, Name: checkpoints/3840/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3840/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=415a94931b3065e4c725a2d8f58d6f20ac0dc4c52ce3123f115cffd0ca6db195e7483a09b2b445d4e8b518232612e3a4af6a58ed18e9da5567b3d2ce12797c424a1207098266f4a9f5711bcad6ffe79190b981e226de83527a7f8d903046fdcdb0ab3837366820eb5555635642094511ed67b13ef6bad3214c078b59f828b1085cb4f8e2835d7d27b9dbbdaf430fcb5ea63c94f63d41f9a61f1fed1383af8653918b46af76d5852f7d2140b7b4a3e1fd713fd31d5035ddd6a0336ad9192aa16e14286f2b43157faf49e4db2229901707d466fa67e3f02bb6fabe059c8c72bac393c899ed7cae8c1a1ec63d9dcec58ede813cb4189c9b8cc0e399cfb353f364a6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519192321Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d, Name: checkpoints/2368/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2368/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8de1d22a041feec15efae24f961a5009caf2f5ca9db7d0aaf6a62729d2175009456f64077d46a05cd62cef3a4e9790c2ab351249e0f08ba97e7689f9bb9db03c45c15c98524355e343d20ffd1a902857561bb941b8edfd77b68ce75f9a71a03a5174a086669db1774a81c8d5edb8b6ab13e30d8838f7c675cadb23c793375888e29611b970faafb7acab1cf028649fb6571e9bb95ab579cfba72d7626d8f8dc18c07187ded89565ef625b84d6b35f8e5b5b392958d3beb33607a6c09030f6088f55b13efeba3f49036f856aad089dce95209d7bad3f1ba0f8ed6db9b9ffe37f930a943185d51b71db898665f0946679d67a21efc03705a4e6098890c7a3dfb01&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519338904Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/d, Name: checkpoints/5376/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5376/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3650f2e52508ed685a47ced77ca2fd0c5f22377c300d7baabfd71687baa29a50d1c87c614c9206c47274719649fb69a28a3b7f90ffcfb23b15ce54adf7ab7b922586eb8cd6114a44eda60ad40600f027fda1cc76bea9afed795a378a3867cad36ff61aadbd8f26f09bd92b1d2b65b03b2fdc17e226a57cc4d994b99a762c4b6f9c6aeb773d9d942661da65f9ae71bd2970a139242aea067c56e7974a60e787d74790e87d4b40cbb56c8f197752bab8db5ade41b5c64114ed09b200d55af2b0fc96604d0213c7e18bef3cad9bfdff779dd80559e561d2d090b13db1fd55f338197e8467fa4ea8c71fbe1e0d3793a50bdfcce2cdf7d0ca3b23487786aacc3da7b8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519345159Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/d, Name: checkpoints/5760/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5760/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0ddb1ef435ef192a2939d55b26c4f711b7223d6d88a3e171f505a9e79707fa8e73a45e3b9853a213a962b92f651fd39827ec09ab1186383dd3c0e4ad4d8ec5850d9c00eefebcc0fc6da333de556de1e99ae9c8ffcbbd6e06f312bd68116468a08b56756531d9cf69a864edb12b64f3a73fc035de039c41ff656cc99b246f68c2ad890f88c6bca58da1d6c09fed0f6d114ce9f69e477ff8198b81d6ea6dc58abed398c75729458d1b7034f3309f5f269f300176ed20086b0969b29c903d2fc36100af6f1458e3c1089bca2401cb8a387c65538b007beb9801236436eb57a2cb004892a888de74bbe981dc16e2d26f91a22720db136baee95d648d2493dac392d6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519394573Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0, Name: checkpoints/832/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/832/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=68aca98e00fde03cd43fcba884f03c805faed9499ac25877e72360847d4e07f434c1caa5df14d7fe439cd21135da7de4913bebc83a36a53eef42d04b8eab757136e59da991b67020821698eadac0baa4d35aed42dea7972b2ce2a8baec5eafe398b6b5b7073100e622d033e74d44acfb5f989fca925420a10cafea92e402b7e1ec24cacb4ab0a32c196601a09c2bf193aec8d8e83b46b591b618f6523953625f28512c4b9270029f386b0bdaed733b31c370a62d14dede1e77b503adccdc9ea01bf31ae0b96f692a1414d447774671f778d11b81416d8ff417584c8a1a890ce3c24249590f224fbf10d26ae28fb9ce7d143dffc76a258718562ba4385ca3f04d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519407434Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/d, Name: checkpoints/5248/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5248/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c25ec3179f974b9b07d8efbb6f38377935f8b28340dbf1578083d9b665e4c1764157eb9d44d2358b22a0017ad0bc8a769bdc2f9abbe9908541c6b3e11ccc7f5811c4a555448f34e9cb11f33c0690616f248f4c04c59ce80938c31288c6e7df853996a9bd59b7f4930263fea50a51262d6da2279d993dd3438cc0165b847ee757bc481d9aaf9f8d18d7644bc5373e8b31e3e71dee767936bd865fc99ffd476f64ca7d39020dc3201baa1189503634176861d681fe87541659c5459c2f4552a09ee00847dd7e5e852d3c2421b389747ad9f5cd6f16b01d50aa368750e295a0b037ee7be3951b19c8c68945b53aa0ca6ceee2354ef5276862483b503fc9df145248&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519441552Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default, Name: checkpoints/6016/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6016/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=16a5cbd839a50723142a7f977b99a312edfbb1953e337534baa6dbe5f5fe1298470e3190a863c33b93b11a66e468c158df3ef3c3111ddbb5d1be655f7c378ddb0e59444b963468cf395254742bfd37e568636998f16aec43e616a07177a5c2cea7daef342f1fa91b0d833d90b9d4a75cf494203826275166a531df0000b8fc6690f5755c95e76cf8dc82da81238e0b855485217dac47f0f3b165a859e389efaa50de9c55cd32ffebd4bd8642d6d59753baaf0e059912d9493e29d166d5f8b050ae8a4ba3c2f5a5e65fe558d876ef77817273dbae0ad0cfbea0137df07e2a88325372dfab5301b054eb0aaf6fdf4229a128fc7da1a447380260034d272f38a29b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519426098Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0, Name: checkpoints/0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=90236595e586a23b6d831a73143b6b83623525230f05f1ad42fe46967931421b636763eed38fb9ad76376fb6f8b5cff2d5528efe642c421eb9d42ee49385a2dd00990c44485d5859c5fcdcba5d203e80555b97b0e439927bb6dcf163a852284846c121a8159eb74bd9928916bfc5c6506e484cdadfab79f8a6a8f2d40fc6997f0fe01471201f94d3fd2a066cda61bea49fdc534e46ae7cdd7aa85340b5556515ade6e131edfd56e78f6037b740dbf44817aa6bfe65d172d016edf8ed378c349172dea49299b4789be4c33a10c0203208d3198b7c2e867ab943dfacba27e773bfd981713ab34b4739ea1e8d543fa1d86a932015170f81e36565d9b1982a3b97a5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519231341Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/array_metadatas, Name: checkpoints/4928/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4928/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=46c6d12c978b7ca618f42e1c6294bdd5fb2d3d1798162f7401e71bd97ed4ee663f877bae51fbdbaf1162857634915fc0b3eb3ffe17703fc94194c1c805a37c106d26106db36542e555213840014d630a9b36693875576b8b6dea39205cc98c920efa53280b4285f63447e651f674018faab74a24400a8b28875172869da2cdf0bea4c8220a663802e126392752a80f981c0494f86f6b09f8c08b9d9c3f293bd316cf226c9068451390ab496ef860bfd3cd66fa7d29c5ad3e8f1b963ba2dd939221d286e6987c690e453f94fc0c36976bf452ec1e9ac2cd10f46cdaf707942096a80fec750ee7a062a09101cb73dece5a40941f4d67440799e8ec3c1bef38c93c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519477864Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/array_metadatas, Name: checkpoints/3008/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3008/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=80b0a396645c178188e689d5c398939aa43cd049accb9c37012509373712223dd88c648891d658fa857e37edbcf05579d379e70e2084daab5c2606cb93fa29e0f5b91a03f3222c85a1283ecb95f51f8345f299143b2113b228aab25a32b6c7fb6e7d7ea25a8fb35acb51af49e018618818196de1049177cbe8b6be3297269cbefce9562f38627f657667ffcf755d1b9141ed5ba10b49d07858d0e43e60fb803733493cbb17a45168146a4d504ec3c9186bd57ab93ac0191d6e7cbebd2ba631d9e738f736028347c383dcae0ef505c2f508f18b54f343e8a47b3c670b9fac4d957ee25ab16299850398f7f51aa3997fd7967d23a7d321bb5e40d2a46932ba022f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519515061Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0, Name: checkpoints/1344/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1344/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2a38f04f47b69c9de04595fe7c72dc61ae9e3700d4d61897cada4ef7341181e85b2154d782a80f71e9c29657f39d838b9430df4bc4201eda565af7b06848def582b1ae52389f1cd90065676a1d5017189ca41a3fb294a066427287d5599fdeecded2652fbc2ff995f4fdb0591551d38cbddb38ccd727f8bd1edf09ec349fcd3e4344d2bf2599ff35324fdd53b9f678a4dc9676863b0c048896cbadf4f633cd35f71f19c26890e675ae9f16522248eae4ceca6dee5e8e735e3a3c213beda239c6bf7f8b5897b98654a77de4c0b6e1ec1083e995ec3646d0db8e08f1fdcb60a07a25b29ce08049664787faae9df9587bc4e1bb3703e8fadac0d399e2531cf3b31b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519537336Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d, Name: checkpoints/4800/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4800/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5052318c5c59b16df2d521e7646857418fc65d2f748929fc73b4f7e2f817092866ef8fc45e7b8024038b868ee7d389cfb6011c5bd7f4dfeeedd3c84f5f2442af25d242154c490c374e9de04785e1131464040b8b4f33d13814e7198aa1c6a67b04db999104d4c9447ce195435d92fcd66e037325da75b6629c5074687c0aa346901d39b487138432c000f15d5387a7242ca99c23813c10fca33d57f354a72b7d5758fc88b2fb2f36e54ba49df3674a0fe4b1fdb014f26549064d69d5d0afffdfc37c0ba824c876f8008db30e25e7f037ae30b44b04a6c4bbc68d662a2b190f750a5152829e8b44fed56b27138a6837ac664e913b61c3f6354bcdb7721ef027de&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519418939Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864, Name: checkpoints/4864, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4864?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=96550e8dc2ee1ec574821eab792f150c031a12667381ece7e1504a9f73936d1e3865071a102c5276e415b5e4134036dfa832f2967220a693328b7cd0039a5a0e21c1cddb19e1e459bbbb0bb078b1042aca3a023000158de4157a661202289566af58def3cdb4d4fe87d7625b82ce611b39a7c7bf5accec5f6bdbd11a1d0a9ad337a39f66fefda70e1c5ff3e911d677ac33082112bb4f03ea893305be865b9b3bf4dda080a9d53346dd472265dd256534ca61b937f2bfe19a39d2aede10bb43573a718f393c57d5300616b10ddecd212b8378ffa0449a06c12338497abd5c941021a3b62b4207fb5e7f5259fd39833918cd5b92bdd1cf786412150c64473a245a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519578337Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0, Name: checkpoints/5248/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5248/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1aaf435a550dcf3d5eeccc39e9023908d1234a0f6f1883844ae4c3e0faee7dae160fcf664ddd375ce2acceb6754951dfdc5e5882de4a88ce88e8d2e9b96a51fc137f923a54dc7ea92061fe9155ab3e2f7b6dcc11d61e067a2614451fb590ae920683ca46616b8601a6b690ad89e117e686920aaffd0e06ca7d7a1f38bdf78d65263d340def913507ec97ad3d2d3e0cd0c16494e0b710e2e99ca8edf97784a665c4c3ff8021d172dccb7af08c79e4566c068925711b7dba30891f5dc9599cf1c7d3dd18bed4db547ea30829bf4b77f556c77abe7ba67943c2d4a11caa903714ed501bb62aeab35ece04133a3edcf104f268001380ee4c729ec85f655cf5e90a19&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519616198Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d, Name: checkpoints/896/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/896/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=14669fde866c735e7f28c9026149fa1dd975bbf633b436adb660b1341a4eff6f975da76d5aec7e3c0a86dfca1fd16f62dd6cec9c12a6db2d4acc788516a5ebea8770ab8c9f0da29fd70ad637e8a9b78638c551a964905e6055b920fe99fd2085357cdea3343f39423b51578df429874aba8cce24710306efa30d4986e25ddaeef6b6548931edaefeaeb4ede68d13b4a5fb299b65f1ff7d730af7f87c24277ae24c38990f61b8e267434bff5a3b49a240d849734c300c118e9459f2e83da2b133c8b7ae9e99c3a65f01247128edbf112dbf885a02772972574e169919ab9f9aa73d253b208e7b2016b579f48f3abcb39ab7ad65887fa7665d180dce6889ecdf21&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517274383Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0, Name: checkpoints/3456/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3456/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=53aaae1bd026d22ae22468ea04bd40cd21d558e9faa2fa1d19593689ece4553a3e2dd276de1e014d27aa9e0e2a114ebff696f51a3e30234f0570a1e50a98bc087a0fb7c4839a9feec759819609a3eb9b6e8a889b84e0918504c824d220d74242bf7030974c8fab60b5c800340310f288e1c8298b74e5ab2036007915d490240919aa5277236d1c8576ddf55e05266e5b21cc2ebb4d2da0da36d3fcd38ceeed9682a34785cd1b35b7b87c94c33b4777095071a7ba644cad4fd72053975447e057fdc6b3a0ffa3a895ab8bd5ddf9e6963b074f4365f2f5d859e9b71360410ce27054fb50429ca377908935ea0f05c803225549f7c65f69e9bb01df9deea5b78ab7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51958904Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d, Name: checkpoints/2880/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2880/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=12cf4d0662e06bc9f180646e54ca5bcb0c640e2abbd32093c256d14d609f09b68e478ff9cb977bcca36e712f276e8e66227fbcbcf0fad85eec36e9593466ad2586b9b321399e8de37af11b43e8535605b5ffee958bcd0f76a2de9f0fd1c85d5beed570e0694c79230e3b3cf71e0a977f80554ad90afc431616bc9cb926aa9a5e4e8ff8ba7d193e8cb8fb8ccaa7eab1a45f91ffd2c2e1b063e8c80dcdc571bac72eadc0dcccb3118a35a5cd237037fa16bd887bb01d7ffa956f6b36ef853a5b91e743df71979db0c338f2782dfaa66eee2a41f3548586b1817632cd980151bc5c305e1537801799bcb14d9cbbbedda771f9185aa3289177615de3699198a52757&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519664624Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d, Name: checkpoints/448/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/448/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b0b5a2a144caf4dce24cb9b233f91c021aa5be2d7ac0fbd00b96dbde7bfe16bee2ef1de3e1b296a644cc18c410f39aa05fd913bca5387d03931e6a810b39f3b98c6abd30aa4c38f78e1589f14ab677feac9403459b0dfe1c1ad360a249a4e587925fcd000c7cb8a2997894eddfb3cecad8c1795efddd7219a00b2286ada8ca746ab110ab126644bc012a5d1539ae52a47ea7c0213cc1c2ec2c12987f47cee9c1acb667777e1f039e2012b207c06969849f0f955476363da968e4737c6e5f44603aa9daa672a1c8a00e4d1e1bbb144b26e1b1f1e1aae163a4fbf5256a9e455d5328a730b134b9817fc36e0062edb6ecbace7f255cbec7362b726843d75f657e8c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519707029Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default, Name: checkpoints/2432/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2432/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=67dc895917429aefe0fac0b7811858d68647a9f7e26efc12dfbe81eebb2f97e9bcc8acc454b591dacf926dea6ab93605ba84b33694192edaac80b47a871cb4bfe0358eed66d426cb208970ef5f06e6000cff7797b6247cea4412c01d075d75f6baea76de6d1019786901a619fb0b5295464386cbb230ed9cc590b3f022c89c3b3242a5a3b6e10e3aaf240645593ac162cd5fef30c3f79ac45a6f3b5c342bc00fa4490ebba1fdc62ba842291adcbc608383cc0a4a0519b2ff2e2211a1fedda068a90f8298cc929eac64f1fb7f4fddca1473cf4cd43a67469699527d7ca66912864b8037d0271ca2a3619d8c05c326f6b7f86161267664f854ea866995620de115&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519725848Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/array_metadatas, Name: checkpoints/3648/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3648/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8dc9dc5dd5e5b697137a1710c333e4ae76a6db60d13e193fcfebe5695923d7d55e4dd2cfc58d4b809d440e5226bc4893b13dbc049bde2828a798bd551af2e52bb610b3289d1d2fd26b829f3714ec29d709131617c7d409b176b08be57dc301dc689b1bfbfd49b0f62b10235eaaa10f825909d8212cfc515ac3623c886fa0adf4d48aef2469b248211f4dc5b818bebb909d0268f9d9b555947e6698c283eb178478f2addead86658159054caea19d3d09099d787f04a125d025e474b36240462b3a5b98c41366e013faa3f7c6e828c8f9591aff2aa8bd2eccacb4a85326ad147ef6fae2eda360afc0197062a03d8cbbbaa1a5a107d54a88a53b16331b51ebbe9f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519742043Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d, Name: checkpoints/4672/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4672/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=12b6c94ab0567c028ea30030f52470ee7a096ae5da30a7fb43f2355e4fd2e80c5587f97b87dcfe31ab80a4e41adcd65b7ea6b71f242fb213f162ff5ab069ca8b86d21dd268ad247b5d78f733baac21a7711cff799be0db8a0cf68c38d02e1be2fc4d1266c6936d2f3ade50853b412a561f189234db0f3d046d9909b2990cc38ee29b359ad90ad42e58222cf7744bd94a61a6d9b5c752c1a7e979dc02c3fea99faa7c675cc42300b4ea623d8fe4824b9e30683ddfea74f66bf402be3cf54d6c7df4635dfe4f16da3cf677e08430eb2edf97d8e9d901e92ee1779b4c2dd8aa46ae5d315ef19ebe6bccb2fce2068106d6f63320765b8701fb3b1ed333ab240dc646&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519814413Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/d, Name: checkpoints/1472/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1472/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5eebaa670ceb42f9dbced56ab5b5974f682bd835ca80ee75b4eb13470092da31c9d4b4f5a6a681ae6b0b8b0fd1efc9120303345217bfb55844160c5ce902301deaccb5ca3918ddcf369286531b8da9302ef81f4d2ebce9f325aeedcbe0e0b989567369766ecd8b3f11cf7b147f11b963dc0e5f31bfb1eafa2b6a59c30219405f5c613fc56a0c7a5633af6fa78ee354e165d925f9eee8d6a09f1309f057b80d3c83eca38d8de14516775d1f68b0fb0c7dcc0c5b95f0361ad5424c3a715d6ac474315f84381a7ebccf5ef9e231f25c3d6709213778f3a32a4e2f02f0e40faa9e4980b50b1878fc0b8bde231a41214fec91417fb4af392ebe5794f2589acdcdce1b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51983182Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304, Name: checkpoints/2304, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2304?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4676fc5ffe163021b84ff0926887b1ebc6bff68b4527fb43c85292c5965b8c85ad29ff015829c4ef9629b6d77315f6dd36e45b7b561f665f03b017ae9aadedcad73da18c3a47991e43b3a4a73e806efd6e4230979064ef355fd25bdfbc5f63e6540001bdf27885179e29a4765b16fef23147553becc1dbdf97df46eccd1664713d13ff8dee621f15c26adaca22954980955d2113b1092c5c6763a54fe5c8cab664ee1bd75f88aa5f078392fbab32241404e6f5676d323b3af587a50b8db81fb86975f8318c0aa21d5cf2ffaafd7c8ae81208b83d57e640fe7e2eae812395318b3fa45398818dc3274e8215f00e1dc744894c27c81564d37ac1bf2e471f6d99cd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519831849Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/d, Name: checkpoints/4480/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4480/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=44b9d343eb09e4b2371118ef13737b31dbaa14d29e0532c93458f13dcc816df0f27c1e8506e5748810313b4c0939369c24d180ad8d7ba1ae50b4407f123c65193ccb61ab77908a177952498512d509777b9c27b9eed77157e0d494f4075fe40b9624701c8fcc63c3c94e4d9735bff4f89ab15b6bb4c8f77a77813477bfe2fce2ebe301114ca1038161abc31c9bc2b4ce93bf33525da6bc3b48e5124bc03dda74dc188913baa1391873b7cf6644c738d2b62b6383f73b25003b952e9a50808784a49702f9ce79e2f4e8d829427049e352b8d7e86ea022f81884bb40860132f077b0526a4a974e0fda9b1ddf296181a511424ad6004d6bc241b5285b678d23e795&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519895853Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/array_metadatas, Name: checkpoints/640/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/640/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4353ef946ae08d47134d7c1eff49110c881c829541855a7148a3fcb7fa59baee01a40818ba7806f0fa940bc8bda3099501be5e5f0bd78949af460428037de3c02236036b8e9f337619c6da327faf283d65fc43198bdafca16f79b06387a3c88e10255244e677c03cdb52ec01b9198d66ccfc56a2634f60d96cfda1065cfa5bdd605b2964198f6d397e2fb893d2565f9c480d50f27ea8938dedae56a58a4ca80abe2dd196c4c9719401c1cbe65d0451f16293b63854b4377337dd53d026432bda1e479d834d7fffd8ab4b0885b33cf4e3d81687032018bf4ecfa99217dba5ec49c8dd2ace53b2be9b0d729144605ea10b5732da03b0c9b75c141e0eb556af40d8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519937796Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088, Name: checkpoints/1088, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1088?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0f960bc6c38e69810beaa19d5684da73cc634284a84c430d127c91d431daa571f3a999be8552b7c4d15775e7c41f0f7753d113877697fef2ae0cdbed37babe54e7d31104c2f35e89399abafe53384e84df349e4410f514c784b96ccc2764398dd865ade4ecda4a4593cfb068662e28815c81e6c6c1f2147a22c2d9ddb01e5609d75495171970fefe6d711115adce19b59a74d4c5dfcc6c2bce36dfe27b579df75187b2d2c9c1ee1868b337201ca04bf499b1730d52a20a846e436d3ff3dd72ea17d9377dfa7b53d20dec986e9325d68276512f1dafceddc2391fd5eb780ef7495b10fec53427ae2595f6594ce524beeb9401b9f555e644797d21ca4bf19f939f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519960743Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default, Name: checkpoints/5120/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5120/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=305ae533e454147fabfe58f54cbf96ad25623c4bd592e307b3201588cdeb333cbd7865f979598d6b711a94562254466ff8ee38a3fb7a803f003a2e9237792c5a6b6f323922d3fb0114182f9d88b58ceb56a8f1388dab7d04040199bd4ca8633af70a0ba010d31b1e8fd2385da0a29be9a15570d2d60d5a2f3d76f6b814f3463cbe4c0246416bba50ece1693c87641bfaf008bb2de0a7ee891c6280755e5ab7cc14ca7fd4cbce15a535f34fc76fa0d7cffc138c9a89a4f3e9bbbcec1ea5bfc0a0c04730e5775f98cbb68cd835b31ec8d664863cb7d20a91c30524bb217bf58fb25cb2ef7743c4f46d977824527da34005d10de30495fe4ae9570d88a4c4cf4fc8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519830403Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/array_metadatas, Name: checkpoints/3264/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3264/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0c615005413f77b4436c86d4557c74400fb8db6e24f8ec886055a93262fd91cb0491a223382a838f8ad75ed542d257109bea9a6aba8628f050b2241545def8147a55e4aa6c1330e94121b8950777803069aea87985d77b2af7f7c6b4c7ea9c08265bb561f8945bdecdaa0d5330061835df4ebfbdd523d3dcf55c4f0bfff220458540d616c9306141e8b13edd088d11433acda91e1a83a133958fb5604e9485a4b23cf7489332ff4e58390488c5f7b0f438316edaf1f79d72c734cef642114c26e40896705fea5de163bcfd6318562b2f65f0b8333a3817722cb7fc8c449a64a7b1a0aa8727ccc9c707728a101abeb96059bf46837202f9e36581b45f30f74637&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520018573Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264, Name: checkpoints/3264, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3264?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2ff4a970c1a5dc9f727f81a8cc5f66cee251d6b02d5763b56d6c8daf42ce226544e5d4477babe74c3d6b28a962fcd053f851e90b06034c30b8baae6ca2171c3a122feedaf39ed31c7eabbe310e7c7241eebca260d36333ce5af28223b9aad5cade158335ba558c014cb5581fba3e51be8f311d3c2c82368d4e31131076947551e58b69db68a623b5799330866f13ae77d0f3acad61315d14cc8144a147ff549eba589b2455d4b818fba693269969278adfdd1277b9b2d044355c597724e623a130aba5c6e3d038cbb1428358c2ea70e8a6e877ecede91bff4f81c1df2ee3c025082116fe819d27f8145e132a04de00c5b5bc18d5c3384398edd17d619614221f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520055106Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default, Name: checkpoints/4544/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4544/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=15193fd79781faf32f37ef05cd42268e10c43a2e385e853387f1c580da017138ac9fa32de967910235c074c7c4eee890de2e2e7fc3226af28508c6ab9125d22b3d0ddbc6938e20f77ee304ae35aab0bbcae14681e73fe9f40a81fad7c3006b7dcff044d7f16d39e6f7169aacc9798e33a529dafad187a890dc39a5932e1869d0a9dcf2a7e6fb2a6fab627047e33a45661b476d4bf55e6f4d8f7b00e77613727511e3bcbaab08f968561b9cf0460d0a73271cad21378ccb63abbac8d825d4f0a2b379c59aef705c1e14858bfdb28442073e87183badaa195a4431e1291797930286781ad48ff2662c7b3e2a42f4f0f6578a04aa9cee806326616da9b5fea53cc1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520083536Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/d, Name: checkpoints/3008/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3008/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9e8f47dcf159cd756a24bbee80c2ca0138fc92aa0124248b38e38aa4f71a2ec3a9f5ebb1e77914671d688b0e3f7f722205342c44234f6e9d72b5f058ab561d31775f6c950978e397bd3a6038cdcd99ca29ae97fa3f50ba6353c4c1c844e5e73d61c9c1f3283d14cb67f3fafe2485d9ed3e3c8201ba5055fe71e972e85921228df1edeb52e7dcabb8f2ad2204f3d9fc05b09c2a169a80e775c5eb5e9036f83c94315c40bdd9c6b34a3dc05ab394c49ecb5ffd4978a685461c2514e0ab4fbf55894068cab3f44247d4f803ed714783977078c5763f07c9702adb6d6f9354417b9751ffc0cb877a494ac7154009b2b6bbf84cfe4572e73bd38f0e9ddf65b40baa4c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520083503Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584, Name: checkpoints/3584, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3584?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0c512688c9aa66d30d930195a070e6b5e0f72bdab1e8270961d110d52a696aa4cfe1473ff2ff384a1b06653c523faa5c67ac65b5136fcfce6cfe06592f882d7cef39e73a65f5638dd53019d846451a0b62f0ead886d6e811626f323ac52cbb787a4a8f688890dc7f6a25f2ec4d70164af88d3479f3f7470adf469979d14508a3e48af65b4f0d8746b99562c8e6ac59de98fb0757dbd80130b40dc47a2769763dbde8c7e964dff4297c55f9c568e2d71e0b0e36e974f83f1e712a462014fde03875d4287707e2a1b7ac10806f253b572fb4b8f004868465b37d1965cbd1dc55f3a32c3167c9643371e00f36044e163b18dda6569ef391a00df770cfc3b0731d9e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520083566Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0, Name: checkpoints/4736/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4736/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c4057306c83db7f0120921ef4f212dca2e3fe4769621bd17f9f0521c27cb3d0b66a7fde78e921b37b23321bd2282212368d756f42fc3db47e4d5f089d80c933bbd9d769da79a7a16ad6a82842d56f94bb37bd1d353e53b8f4831fded37f1a2dfb7db50d15e0e43e14ddf7560ed314c877aed6b72b5d4a7c8fdf7885b1983f66bb291716205aaeeb7d6f2b1cdcbc89fa4880a500e97b17d564592063af8b2aea7833b8ff3adbd9837dea47bb1c7afe112b9bf362d331830e8606949ec6e95a12fab332df033cda66a25aa895dc3069ecd22616e4bb4ee2008d5186f3fe48bb1a10c4d97619a12ee1d4bcd2006ca288dd33c2c0bc5c4c31d318786c12ab9253b7d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519877121Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0, Name: checkpoints/2496/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2496/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=097fb898d990abc1a282ae9e673ee1bd80e09329b6144d0a6cf833dcd8af6a5a023279419b14cd03865e0a852f63c7f913b924cec4211493ea2c6227000ee042634bb57e830dc0200238a5569707a03bf892118000ddbc211d6a6133d728d1ee09de491e058c1a7b483d0a3640bd5698f01d35a9f390bbdc9842180d44c2e94651dc2f5054e3aecaf6a14d2906d11fc46dce317c09d290a2232514eea8a6abf11f8c3f81a2193ac41925c1f619c595ba4b528a55b566195443c5a131402013deda81d143f8011a9ee0b8343f7d82cc98ace66704b6442ffe3a8cfe66ec2abbac30dc293485658bbcb73c85c3194dc058ebddee0da4395107af3b88833be4eee8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519768763Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default, Name: checkpoints/576/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/576/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2660fb7a8c9bce0d47c6dc0199036465e578ec9db9a48a6ea8eca006fb53c9d599d049ed923cc16bbbacda24d2db6344bd4c435229b08b9deefd53c59cd9b293a3845ca65a09b2fbf57544f7b09a3f9a4de481616157d0b9c9b0076c4cc1e634eb2f52debe2efc449e389ac6a2f47f28799ba0528b9884b0d9e441697c9a13e6f4aa0b9f205de1b6387a13310e21addf56bd31a6ab1ddbab9b8350c356c018dd333bcf48162a5babb0d8db6cec1182a66f2394f0c17a08a5a7c07a27090aa21b21f780d184debb49f8326e1004fd9ff8e96ab8586d93a245ab02f75fdc94d9c99b242a0b6b462152d651e589ce82018fb5fbe4def2c459d74378d02cbf132e9d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519969293Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default, Name: checkpoints/1280/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1280/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=ab11e2b26e8693f91a4d75765b17a73bd53c429221daebe354fc93eb05726356cd93269fae70faabf45e3035dd69577f907b01b68fee3f1d875ffd0083e2cb8a96b696d0b74d5fc542b45b8365c1c78a65ffba549b0ed459ed2fedaecb0b981bc144f7f9a0e649f27e8c5f65ff30adba3af99730b1d79b8e2384669b9cb51197faacf283503480440eba4b6a898f0f12a8220dbb7c35dfef79266f1b711ee84b1b60647da23589f5288bbf5f93bdbfc3c97cef174ef8537f7ab0cad1d996693412b6853077a126d19c0e13460f52e4b2aad771f413edb4417a11f27e702d48300b4f1c48ba16d64926f8783aff3c3a29e81cb02857ed26aa85ac24003f08622f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520219283Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0, Name: checkpoints/4608/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4608/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=30cb1aba890851694eb2cabd8acdfe255bb38f2b2998b19cb8cff9a5625d39b8e78dbfa804f170562ab41011f38424582fedc223c3656ded1448f32ea5f7ec6a78d8f6f8c6c99de1b8814ec13cb1d524a327e3b308aff9719e9f15faef1eb5eb687a9ab3e56123da61c24143e864dca8a2e6acdd42f0f79812d0fdb07c2d01762a4f9adfc1da1a7c3152d1f7912acbc0cf3d102ed62ed0d7c90b37022e2dd6c0f72613a0c38f714c11cf6eecfa937b6a03ad154869879561c3e591875e821487278c074223fa07c158ea237a975bace671fd3370c7c3e153b81f323d310e89e1dcd0bd5016a1d45452f292ba2a83b1acbf7d1b886fc600af349fca2c37247366&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520228027Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0, Name: checkpoints/3392/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3392/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6c10e853a2889affd4410d0f9ec2336248d06cf179388404476696a2f5c2c4a30dbc992343d571bb9e9af32fb9ad28a09aae66db4eb21c53936727fd8e304cc0be9abad4742f31b66963738df7d139dd0a044d9a8533c1ab91735b443410b5670f4d38d9b8ab642ef63369f494184d2bdd4391d50e428f553bd9c5d45546b68961441f178fff5e9bc5e950491de111156bcfef0565e893f4d034a0d2516b994e5c481929e74addd032c4120832c6af8673524bfe75bd406965f7a2e36ec8487322cbec0380b131883e02e1352e8dc31ce36c12c8cb0f3009dbb5fa06f1a54977f981dcef68db2d562d7e52cdd7d54eaada0b9fa7f1ed98bf467dec3f0105cfee&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520117176Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default, Name: checkpoints/1728/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1728/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=ac082f43966cd5b00aa06637a5ec814ff70a4a71e4dcee241ab5413f08cf6412c89d6e4d94ff78d126842f6000c352695131338e719fdb95e71688ac99e74353099054eaec0412438fd60b46b4f4f9dc3d92bef00c3e068536ea68c4227502270733462c1695178ed912a371eb812e0742e4c993bc7204313793a12a889f9baade2f8e0597ffe271cabe3cd91272a9ea7dd964cb65818491250d694b34f0803cb704ebe392b5d286c0726d8de84fb98fbb9736d28f634533a2b5efc7a95d1e6a8def0d0f8dc8aeb7e92b8cba655e0b5f6f9cdd38929c5284de2c3bc2751b1fc1f40422d5740d74122d3c3595a568009d2c1beab8f982218e494474bdcfa0cad0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519345223Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default, Name: checkpoints/4992/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4992/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=45e446c298d986c964da8b9ef521afd62942351802540972a70e6f72b490b73436dd5531161fcb918250d8576218097876a9a603d65b0df43698a52ce3fac0fea63f2196d8b1dc7e276259290feb58d9b13450bddd2a55acaebd19fd1863b5411f47d5efce806230324e498ec67ea258dfb8092959911ea9673873da91dc512254bf95430eeaff740a055e8bff0cdab46ff75f2a868ef2220ed71abe530102b4fa2f3bdf63d12058916230fef0833df2aeea6e96947f28e3b6eb33a3918f7660cb7f60a6dc51761ba7ddebdf67aca1f4136caa3be42a9d6e2fc01682860426b7f0efb6ab57ee4f60ce5b1e81f92bf961f0721c744d83cb24d326dac9016d2e13&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520300994Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default, Name: checkpoints/4736/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4736/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=78597cdb4f4b0a08df54cdb82240ba1bdb2d79c410c7b7d6d74908af59eeac2f2831ad29985c37805f7256ae4251403b1753b666377a66e93dafa25b83fe35d9ac2bf572b82abf58b83e0146e1230b4b7a4e6ab25b54dfdf9d9fb5044873622bfbd8adb5f5f19c4d67830c15c9ba72764e96f305608231889c24883d2edc7f4b85a6301f511e2c53355973ad8c29514a3e94edfd96d143433bd80a4af5fbf7ed05c1737ecf54da8bcbe2be46caacacfa0006b614d5936aa40d4d4bee4dcbd354c5b2956f40c818a63854d634d30fde70bf261ec0ef13fdb710f19080ab95b3e2d369a4719efa551491282fab0411dba34e216a08c3aea7de1d62df4f9011113b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51954855Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0, Name: checkpoints/3776/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3776/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8d7307a95fe572b2d029c0c90a48fcad3b5e80f00da72ec692aedacf529fed480105292da0da6952e67e63fae47d6814e95b9f245b5df34cd021e9a2f66a8f363b2145cd1107c6f37c16aa00fc70e02f1677b89fb9bda6bf3349614a5efc843f87e332c4238f9b3e56982b70a828d7c35c9968a24cf2b6bb384e9de764249d3ccfcf898aea443feb06603f38912433f5a592ab3800f5ba10c3d7a8a8b4ba3fb878ca7ac398d4a1ee2d1ac2c64e43010c07cb3b61472d34d63d79a9a6886cd949efdce48a4a8bd95e5f5cfd023d9a101ba36f4d1663ad2be2ab422d4ed13f0da9a7562aaaff3cf16a04b54895647ed966dbc6371c1c4ecb88829947720b044b1f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520336359Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640, Name: checkpoints/640, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/640?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b3fea5e286e3d5525767395d5bc902ae4b08e3e2f83e04bf90ca64d9fd32d981e5790656bdd7a636091f91fcef6ae7b0b3cfc039dc5e0e1e920d05243376217df352e3d48648c3265c10c05723ef2c2558e368dc769c12eebc63ee37ea171d29f2ff584e21b8ff4a573466622fa2cac68f96e8f8cce701f52b8e38361739fce832f8cb6ac170b429dcc87320250d591326bf7692caeba5a0a5e9760b7914aefbe90b6abcc98b6e3bd0eee9acb142d0d3b308dbb5f951981a8e6efbd6deed15844365321bf5374b2066f86dbedb1cd012af01e60f43eeb5ebf92a015bf1d41941345ec55149e98a6d2696df8d034d792779c3bac03ff2432f57403272ac6a7d03&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520090212Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default, Name: checkpoints/3776/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3776/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7da4f809ffc255c92d0cb1d3c50209ba6d18c24930883b6dafcc28b5ffcf9c6b18ab897a3d285647ea0bc56b4eaf73d0a13d0a684b7efc2c52dc239a6cfefa6ffda190651c73b67e06d0dba47b58df7325731e9713fde3b36f4466146236d2873cdd82786fb295bbae31dd3b0854d1a4dd5b0f2226fa33aa3438d975a88ff64930baaec87f48232b086075273a9075b0d672bda7200ca38cc99f8080c6ca2c0bb41edaec3149d7cb8a10b4244b1de8b095251f35d80eb26856693af26ab377581f02c091948dc9c5e2b3fce00230604fb1eaa9e16d60c7e0dcdf1b5677f17db8819c90b2cb87850332931a3ad07d8d929d416e786e03cdd27b7ef009c174c042&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520415967Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368, Name: checkpoints/2368, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2368?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6ac0ec5cb1631936d465e2f3267ffb2f613a85303ae73ff15062bb9bb22aceed621ddd9da6c29dde93ae188bad935c675ba3c131c10e63c5e20a557f0b6f6687f8025487c268e6bbf6af8a46dbac8114c2c5d3023e4b27d0b37b29ef9dbce6547cf36c58bab5154dd83b342689df54aa195469ad847bba8c9fe63c9657d86028f4dc67e4ede009888580996a98c284d05a36410b0c9f0226beae269eba3555e7931b38706aa784c7385fe037d8c1c789a63bd54e1a6a7f06606ca6e24fc22510cf1d7dee2453cec2ff33641b748ff1d7152e3326e30bd0988f745f24abe9a0b1e553cb14dbfc803dcd46ddf865404bba00fe74b3d2585a74530566b60dbde84c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520226116Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d, Name: checkpoints/5760/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5760/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5f39894508feb26051af0f3df48c3f446a6019d6a150f465d82994571dc8fa9bc69df98e3ecd5ecdfc2957320448357001fa3edb9a8374711dfaea8f9d96767203670a37c65aa2e4e940ad1387b42a783146d004c224f7dac680b81b2ebaa27d99343c857021e2de289dc44b45a8d61d3890cddbbaa7d7dca7788f1d179ce097a9b9ec26520caa3bd7d9c73e7a528401c3875544b472ddbde6e67ccacc887ec68e14670a4ae59f5f5e937e7724e4a0c7f874c374ec948e13be4985865c524362c5e43ca4663fcd489c6066e8c2158f5296f47333cb1acd55ce24a3f2977f54a2b4b952f9bdad0d1fef2041009ab91290bced48041e15454b0ce3b95e01d908e6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520485066Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default, Name: checkpoints/2368/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2368/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=04ec5df1dca082e169b7b929f6c381c5da719336494c6c466275c5f024fa962efe491928e994a3fed627fdc536adad018096791b4c8d90342ad9addc367258a97cf410c365c42f096779b54d090cb5e6d68dc9a86e2ce48192e3b2a937df6be89f55527c1ec9d7106b317c490072718681fa6be75216d4b76e6924868640b68312d151478a2902035cf577a39ab549fd6e9d2edf9434c92acbe464152734cd63aa1a6150d105930d0c043d380646d222cac31f4eabdf56f52241fdee7be347014f7390334f7281ae565bc809cee4d131aabaf9970b99a8ad84175fe3bb0f8b27abd7c8e65359b94450c54a266cff67eac2bac85b5548129bc8af1f1eb4f2d30e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520495131Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d, Name: checkpoints/1984/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1984/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0ce957a1e7ef1cbaa80abaa5e497ff06eb1e4544f094484fe848e1c6fa4ce0e4ae527ab667f70d7721bf5542ecb9cb9ad21b137987562662219054eb7fc183c2814ceba244b2b98108d6fce0dac981963b22d9f861d455da50d122e6e982efa81212e6943ea1fb80db0f09fa19baddf656159af339e04a7bb00cb57dd77aa70fef1b6bee63c1fb50fdbe8218fa036f437e2ce05ba5f93dd9e554d98b9c7c441488e198de24fb43953d5139051d854df965466b7e78891926c5bacf4e45a0e9df58a3703abcef337c15087f8b20ac46f5670be1536298e4df651e194931c77fd4c469495dc8c6f908046f983bf4b010f10d12e11cf0e838dcf22463fb051eee90&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520542095Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d, Name: checkpoints/3456/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3456/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c47f59159ffdafb09f3d6a8efa0a6d8b63d398d841d166b55d4cffac8cb997b52d7d107578b16b2e100ae361314daccabcf873a9402446c715ae7275d7d68e3438c44e1cdc2ec06f007bf43b88ac45465a31107f8c377f6d8cf1e4170d484f36b9769b96dcca58ccf488d3358cbbe76d59628316fb698d1aa141eba3ea9921b31e7b4c99b75a8a9fc2a93f9a84780214b630c1888ec19712d8d7e28d2592c11a56ed59510844f7fe88bb8c6ba51b699f0bdcf76ff5fc06bd1f3c87f2559fb20c76ea26cdabaff45f7494f2164791fe379af14f8cd7e143489de735275b8134f5625ff20ad9be2d1dfcf659f8601566071d60573677749ec2abd71e9ed46abc05&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520562009Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/array_metadatas, Name: checkpoints/768/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/768/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=07f0580ddb832a5b4cf61ec5e75958acda465a052dfc235da4b0cc7e740140fbe383f14c91074f28b40657e77a70b9ab1955b394665a14a923551d7192927bd606aed3510501c38bce87124a9c880c06afb5905b5c28ec607577e001236b819a2fa229d8222781f12cb9aa4237f8df574f118ba30185b697ff9ef712dbf3ef47ed135ebee3c65ebb306c1071213cd9084576a77ec361082b64ed47bcb31ab022696f5a0f688d95797ade33f5fb80e63a6ae65d22939558d89472e614dc399d989ce71285538242aeaca0b61229d619eb74322c328162e7794110f0490523ec5138420633e5a98c0e5508109cc6993d10dc480d2af3f9d5f1d07a418d301945c2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520573021Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856, Name: checkpoints/1856, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1856?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3624ef657129a12df116bb73fa93dbaf90737c0ed64456ebbfc620021709e64026af1837a7e853b9a1e47c70139fabf881ece71d463c417399d491dca727f9c5d4fd51a40f2ac3b2a46a0a5b7f2bf3ec6d5ba12bae9a157d10aa016dbb04adca1507a4daed9b10642d6aa355c0b30e43856afb962d3cca3563617c27a661f7610f3f2a51fae17d81ddcb9768bdd9fa56056b04b00a94186e7d04110e5bd83a6dfe11cf6f30641f7006b597aefe120d9b919f4dc1dee85c5f9e392fa084d136c8bcff0fff64efc3a989278f9df4e06fb6c5a6efd1593437acb68e7c24dbbd2516e331cde6359c4969bcc3c6a9b2c03b227a02e3a58ed1edaab6184d6167f5dbcb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520630714Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0, Name: checkpoints/2624/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2624/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=be7ff124391e49feecd3db640f3b35385f9709117dca6719d9c55aa5d94fbadf4acad28f1574054968b596648c0b8d885ada572dcb2e3a7a17103738fa0035aa1c47e552fb6b000b7bb854ea46fcf8cf0c5545990b21674cf0ab3b679e49c7e3ab653dd434a5d97a2191f1993378c97987b471fa9d7b08d01a3939dccf46f79538357bc5d28ac72cabab9f3e211e1af147340e8ce51d921be0c8d5aed24e4a8b10c2e30ad7354d1a121d6cb9aa5c81757c84106973f5a6aaefd2e793a74edc9ba5e8cc656fa24ec91e99b59939d216a2c873ca56730438d1ea10f1c7459c47931b2400901e7bf61b96b05e6725d99b922eb584aeb1a8275a8964f0fb77d47ccc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520317375Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0, Name: checkpoints/384/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/384/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4630ab26e5bbd6ddd1a6d5747e9fe5ba631b429734c1d96920ff8f3032de9055fc5be5792ceda8984350ff15ec7be773b5a27cb378e42de36e05f71e06e649bc18a081b132a624c8cd8fb825ef544cdfc26842f5b011f4ee83f1098ad8a56260c7318f98ea9daec7130c24e4461ab26d51e7e3a3ff9ae13c7cc2b34fda00a06ae2bfeceead848594e2231aa752b377e832b0fbbe7db35657520b8206a801ceaeda01c8744f3be8cf8972e4b026c9891d53739bece981ddc8846a062977747020d13482ad62d0a4555d47b28d28372f38c0385dbc0d4f9ffd1528e8070d21689e1bd168dfa0fb00faea55a262f9a337ce657323a52182114125f5fda4d0d18d64&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520651531Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/array_metadatas, Name: checkpoints/5568/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5568/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=973ecc69e055f2d49f9f34ed6a1cfcd0df787c9e22e99642e733b3760fb21c75eebd3b8afeccda4c96ee5aab126ed780daf1552589bcd2ed2a0579eca21f2f7222f95e32ca2b3abbf7dfbfe23c1af0d5f692877e456530d57d3a7d65e9450a264d42ad6a1fd8f1fb8d40ca4627a52d3edade30d1f41287df5276386c04a9acbf6e2a00a0b78d67fc03d89df3c4597fd24b9ab1f030ea9ecff7b9c71b03d591a2dee1f12da8cf89a052154ef84c633f568e48c167d497af405ac90b301a40d42de71109fa205c2f10ce9a4ee75a3fff3c0ececd1475a9da9cacac91d07c77230f5cbade66aaf748de59f5bb5a9dead894b8d5de1e150826a87a64f2214447292e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520678244Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/d, Name: checkpoints/4224/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4224/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=10cfebf668a7949e8e424dedc755e8b505cfba69d605e0cff7b3cb1bcd9fea532d5c7169da498a5ef208f263c555d70652248aabb46b036e6ad2f161eeb87d08be8810e549554c4094bf8f940f0f400491c7d2f0d9b00c641922fa9f8baad97d4fc40a30e83e77a63ff0f7ed7848c5d4b638b813b8ee890c342a355cf115d5d5c1c0b81453946af4e0e1183b67ee28303c85a49d3f94e17c6ddf464c36f0ab4bfce09ad8829a50a232926a48dd6c2bd1752e5facb64ac806abfdef73d2c0da1340469bba417614dbbbda20e6fb3de1f7e7f1bde314924a8b3f0634025b457465050e572e74b8540543167a629bfb0e7abacc70fe117f0d23d47e16d98ee63190&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520561855Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default, Name: checkpoints/4928/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4928/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=83530307168c1292fb1e1cdd720c1afdede3031e8e55e05c9926e5a3e0158ed5f5f7cd9827b8fb6aa919588ebf6c56edbe9fba6cd1ac7689bbb4b18447e945e7d3e7cd62b7e81d4bda94191d10e300c6388d3e62e789e62d5e45466028d8f22270f509fb6c78379bf85dc7035a94e39d44534d30327a3a98409d6b5090b26996e330b0cb7c565c425709e1647bf5c75f89de191a1e2f656624a2e4082251a1d46d86ab302b6d58c77af36390121b434155cbc444a70f3b53f2c58b3dc97bc837ae889c71ce9483e1e14e39da031e092c2a5aa1f058f563603b5ec4e8ffd4961171497ede830549e66746dad67cdc8ddf7d6f557564e143b57135d119f58c0e72&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520729404Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0, Name: checkpoints/2880/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2880/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3bdaabe435dbba6746014cbd0b598b6629da801d9df4c3f81e4a5c3ec650b498ec3ec749f5bdf559786df3037aa4d1d1a974cd3ab1174a02aacd9aee0d68310aa6477ac026942b8a6d8f115f2d49e9b753e6ffc688a0fa542d7c709ebb913ef67f1ef615fccf71aa05f705d586b2016bd6cc66171670df7d5cb11af292c5714d2c7218d9e4a375f2c48de1540967d42d2a985ed5e02824ec296a209160c8d7c8d9e771c32663a42ed4366b7fbc2b55559922dd8d212e618bce945be3bb0e64e471e7fceb7a1f8a08bba9a3a1afd70c21ab4819749fc99dc0b5dcb80301384fbac77ce349b096492ebc78b80c839f119ab73c297d5c1d466da69db196358a0fab&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520743816Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/array_metadatas, Name: checkpoints/1472/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1472/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7c7b08bf5a45473dca887cedd5607bea6231c3b275ad797d777b07ca98765aabfc55e590d1481d81ec2c9b9d973ec4cad47132be8cf337eaf2ea6cc1b21df32bd88b983aa344ea214da87c0b33f393fb90eb74e23f3d5c954943dbc5f42b1ced5c42cb1e9ebfeb83d9649584d37ca044a451bb6993bcf0caf9727e1058aea8e89434c59486cdcea8c52bdefe50df1245d8b7376582ac64ef327b422fdad7f0dc056a9fe6e51dbeb3d4e0271f059010445feaac5e1109d7b786eec8d574bfc65530ea1b398642c63f33956e4a4ffa725432db547bcbd67fc1bf1fcd306afdcd9a9c0c255d2f328b62d371583b0c7b5556668abd054f56513654ff3ec9e1c04c58&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520752815Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d, Name: checkpoints/2624/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2624/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2d0909b6e345ab1b442e14d951db480c36acde647041c6fbb4754a0f213380f1aea30ed7db952e00ba7c83a12936ea8be9494152b1d69b2765dcc7cdcf0cf78a28058ce2623d017b534ac7a5f980bd22508dd6126b6430016b979ea22fe744bd0a664766b182ef7addd9b1aff0344dbcedbb786cb1abe9bc8fe2cfbcb394c570f580dfc0e8c01952ba82aa6fb0b134685b432a198c255f0b00f455d5112e8888a0d913f77b54f7e83fe798572f40b53995e46c37585f1e5943c2b58018a03dd17d29f0cfd27071893dad884cbdb4c928fb61416168617e1466b5231d59ab3e03b9da0130f4a95a9f1e5f2e4a7a9d4a1589f77db6787e6b04421e57fd725b7a32&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.52078932Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120, Name: checkpoints/5120, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5120?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0f8d5f8cb26dfa7e57ed7fb2b05071481f2d205f16f19275a8a8a32a8822eb68a246047ce52dda9a87994ba3c1cb3f9bde0edbd5df0c64c6f20e8023cbfa877b7929126ac84303c02127a0d6a8129cbf87e4d6976521eb7394ee518049758e51c2e5839e559963605fab0c79f03c7421e115ae83cc28350533061a3ee1f5201683467fc5eb886dd27b68667c0bb9ac07774a9db829ae8f8cca10b757dd35ba1c6f9d0dfc5025f1d3173d781bf75b4d325eebcb118e112be33cda07b7663a2f8fce9cb11ac1e6e23bbe67b99171f8ce0149787ff167e5869c80729587fc5277dbd45bf5b01ce78c4e842fd09100f0ca5b62437751f3da8b5f0f83aa89e281b464&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520758813Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default, Name: checkpoints/5696/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5696/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1afe1937520203a84ddacce4241b6ed9de4e8aab5f8c18e4cdf778e2535b087e8b8bd84adc1529462293cd57e5fabbb20c23308a3a918ee5c41770f7a83c5323bc3af67214eab7edc1a68e78021d044191cded335c3d0b0e161fd5061d86f1684a43fb79c578b27b6b8bde434e956dd6b6d2afd50704a3f5d703b05c06302551020e7ac0c7bcf57c0546c35d0214e569aac506be757864d1df5f5ddb6200893e11bc42378b010b7a1a31ac64e342308cb96b17779931cea6ef281b65a3d17213dd918061eec6a260cb2dc85a1aeed44c214a0ad9626c080daed155c05a51aa6af040d882b08366f7694b45a82eac6ee812cc8c5a63d3d695d2ebd858601974bf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.52083672Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/d, Name: checkpoints/448/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/448/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=abdb68a0c4f323bcd12e6d6b66bd2de94febe9507842d213cf4c575a58ffb951f135395ed46d515b9b8ff90bdbe7a2a3ff63525935fe249a891d3dd373c51546c6729009920f986bbb78ed942516cd60185061a5ce97de453a41cb3114f9dd69262d1f1530059dabbd424ae3501e2ab8432594009864bb8b33ca4ce0d5ff4db1f645ebe61129e68d99f54c72324aeec22e08a970ebe7ce819574094c012d6cc7d5e6756c4d433152f607dcb3b59d0847460b8a4e14ee122c94f327a9aee03076b19c1748c83e407873dacb6f19b5a346595ebd3720c93d04b7f3a017e9763abd2804de8e03e414303f7d44682e582de4bbe940f01b7927c619647d96cae71c6d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519602765Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/d, Name: checkpoints/6080/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6080/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=79d650e06d1587e373f9123e852b964ec97d982a0078a9150f6aa7f236bef5c16deac153e7e580c92eeacd266a22dc274339d1350d58b0e560dcb7e93badb6afa640ef8e3a219de504f6f1eb3ba402beb4a4cddf1d77c55f4d679c3b3abb9252d6461862e1af766bb78a54fc984e8877047189c90f3f52417828f9600a6b9a2fc58d31b0f2d829c91844bd52a698ddfc6c7cccaca2c7769a1dac4b2f174d914c8d751cffde62b3590cdad2a591678e5b75ede3791d39ff3b232720bd7ac3da550bd0c2cc69935ef8ab4192cc57e764ab53623a574c6a769ddba43987eeabd06bf4415191b9dfd5564be1c3881df874b588cdb85d5f2064188175b48563e7962e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520754416Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0, Name: checkpoints/1664/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1664/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4056412d6285ff9640496d084b91866ec5dc0063ef2e0b0f733d0ace927c730c40ee65019434ad4cf70c65b4506aeb7f6769dfc053aaa37887b260a8bdab3d84316519c64bd6b77f48ddc1b7a8b55541ee586c911d6679011f8664ad5de28a25e71bacd6f1bb2abd493da4e4e6e16dc2a0e949ab8722dc60fa240156794657af1e8dfc81bf3eb275603a78fb0adb09c9ee20fe5861888b6da52f1bb1ae41b65f0a64e9fdcb2828f41af1d54d6e7ae3f4658d534a59f4ea94ba43aea6296bd9c2ea9dc7209794897e9f15f90d63a7aec482912fcb5c4a7ee65863157a4786dd2e64ba53cdce577684778080dfde019afc033715451172d4ebd264b452243b2c80&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520863413Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d, Name: checkpoints/384/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/384/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=30f76d08c6278e93cbf0dc37714b711832eb1fb201c55f16917107f0c2168e94af40b8310d2ae576aeddad419859f3fe5c0d4ce3d104ca7f56da0681c7dc58a4df603de7555c560e0cd32ecb24aef0efaadc20bb0c0250564b4cc76e1b4d1c0a5d218e7ab1f73f5f3442789ad19ecfcd7ba1283346ac8d8bc8970f8810c27551bf27c9e59a2663af158960dab38a7d4b479a14e1011a588b7378ff665d96520e82f77b0c9f18448087c6c519a99a27e78c0d4813c00ca5f07f5b1377d9c448be4338a1b712847daa7f6a1a55e53dfc649bdf2bba1ae2f6eb670f403713434a13312b50150c397754cf9ddb82f508d1f6afe613274165a6283a3be6fad5f60349&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519712692Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192, Name: checkpoints/192, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/192?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=344fbf3980c7ffba6e758d5397f3af2eb14c0cd7c39fed21000615de658df6daebed65c01d82bc18dcf57a90ae1263498797ca35be1cec6daf736606f7d5fb06fb2db92b01aa630ce9f78ad2ce59736e63a37ecdee4ad8c93bc3dab5b58c30905974e639881dbabed11e7d665face94cac07e96e9e437f473c1be3c6346f1b4d138728e7d5e26aa63c7e2789b82714adffd17c67f12c1c6cf7f084e55cc0e5e0d57a366b96ea35649f8b77f259261fbfffea15f0487bfa0e91971c545eeb2f1a7aa37bff7fc07fe3348bf072812651ef2b7f3e915b058aa4472bce0066cb416382d7676bbae1ab184b87f90bb536bf2e07c8fb5b82c668b12b285885db7df567&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520649895Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/d, Name: checkpoints/2688/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2688/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=43e189213f8a66641fc8ccf5a518c36c8b49b27126feed7c17e4df3c82fa309398c1a0b2965fffa971b2218f90fe077cd0d6def901e5edeb5746f65b2f3129ab95bdf552be735e3c7f0de9f5467523f77b1f93d28b4207fc865ab4272473432af78b3e6e5d7eed7e49976af55a4ffa9c1485fde5cf9c4f9cea92ad326f4093e34d0d7c8e8ed6aa37457d51f36848a3887eb05c9ac327d51c048f79e5b9f739ee40c63ce75dc3a5991963d59d0d65d1290144bb26feb49469ca5a4bacf08f64dc4874c894c3036fec3e776a55e4456594ed8d60298b260a3316d257c9c81d45c5757bc90a1a55c36aa2317f708bfb1dc521e52eec8e642e084c8ecf5d098fd5de&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520924202Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d, Name: checkpoints/3776/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3776/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a9e48a2b24aae0882dde0a384125241adc3a864b71236dfee6372723f48019c3599271fee840f96772365fb793462e01fadf2e9280f54366ca05d40f40f21aec7fef6eebf47474ef0674b2c2e2f494b0bd8c0148860bef39a25ac7f2e7b13b2d3ca2854f930f07b386641631e37f1f3f7f5220f3d3f724c607e85744b9f61b13777310b1d964fa9d13489f03a6d2c08dffd1a67f970b6db17dc72ca2a68157aa15aa5d73d0f65a79a4b8760be156b4c33272407664d28f49ccd6cd73050db4e90982b42351917abe17a95b87d372f41e2cc866e850cdbbdd7d7b574d1d17ec42f8a93ad660d3375a10d4e18461f053a69e407711405694dd81f813b2801ba6a9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520944672Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/d, Name: checkpoints/704/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/704/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a45b34e8fc13afefcd56a3516e912538b5de310c9b02bf140dc2762214bf3503cc44c3cea2029737e525c44c12acbc82b485808db2604f412c2f5c1e6df2ee9ae0c27722dd18c2d81867523f26cde97f3c281420de1617e50e72472086cabd7d3efd46ee497bf52c114e906fc15f7376f5bcb068c6d4257b618319b416a09f323c39956d9511064bb9d6aa90d16e648e709842f72313b6af6a4f2284f53dc17a70f96f00d4b39fe9be666df5e5bbf0e8674e9ee4673b4ad385864ec67568029f9f4ed0fbeb3dead530d8581eab7b233829f073398ff627cd38683b8336ce6aec2242673bd88f77f2d834d49b9b3768dd17db234fd38d727525ea772d6175c0ae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520970189Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/array_metadatas, Name: checkpoints/2560/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2560/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4c7cc6e86b0915f09a5d4cdb310626d2c6d8633c2fcfd36630e4ae1879a67d6e748f1fe778d68b4fe0443ad82372b920c1a8905734c056378fa49323674ae95a80b137d5c04130848ae467522c2d5d3028d5182d2a15c0023e3d864401fd97f9f3bf43d5e735368a52aee60103ddc75297adf5d0e31b7dd3b7cae2d81d2564c79d5c4ec34c28430ec72372897bddeb5820abd56fb6eb10a5853cdbb01b26e34607f353cbb83916e1e42ee2a549cbedeaabea9739704301afbdbf7e23ad2782c74c76f92e3f54a524e6072c902663f009fd99c19b4cfded8a7ee6b941c55c3c6b8c94afdebff75d37ea51a62b1a48c7f2ac25bc063037351993ffcdb6da65b78e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521041524Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/d, Name: checkpoints/1728/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1728/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bc66a37c88673e1eb50e25763ef9fec9077ad3eac436358f473f3d72f9d09ffacfedb6dac44ed2f4b16f66552e6c27176ae86c1f3094238a1091f6b4b9a3decc032cbfe6a6ab617d87169b52ce66a95cdd12a588660c5161a926e28c4cc2a40711a2458e6df46dece120be6e96c0e380283f87728b3e2e4faf17a2c7ee1cb65580cef5b6a6ddbe9947409b8ca6fdb912186a1eaa1aeaf97d8d60dd7c644006f5e3e13802428a1c9842fffb5f7427e0c1be1bfec53be9d16b84b63a4c525cbe109f58b9eee1676a544f5f48679d534e3233af97e0cce8c03c802fbb726588d289f29041fbc9d60c5bca032b9e7cd7f2b19406d00c9de2fd9ff1909f66cb10c106&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519505432Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0, Name: checkpoints/3520/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3520/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=aa4d6d7cb2434d3cf97528a59c4569b05f46e81f20bff6d10fd3660141f82533a6adbefa2fc826ea12e975b2396367ed8d37b224f7c9fbcd537f54cfd9f236bfd9797d51a85ee5c990f1a89e4d51cad467bbe9dfb4cb3142cfde6f5e5962c1d0ff5193d5c0783f24588007537046c9d1d165e0a2bbe6b56037497cc2ce47b4a84e53fb566ad2d31fb7d79dc75005616ec85a286eb9c4f07481c0b26bf323fa9a73945c4310845f4dd7094a65b8525584975936d1cb7e698630dfb92683a3eddb88e34a01151497a7c31be080c73edd433ce56fbaea8bc2648932dc8b58a535792d851ece7bcb3fba1968b1a3925ef08936520a20456bc19d01cd527c605488a5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.52108671Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672, Name: checkpoints/4672, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4672?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3320b0b7d2224d859eba33ae87885c28229a52a27d87401546ca1b7994229a9d7f6499384bb0535e275bb174d2c7b16d4f27bc3548a2c84b26ec08493dbf4ecf0bb54cb2bd9144764b51484ea8ccfdbc1d09dbf2952a9c1ad729d211d268e6843e7e215a0f027ff0f4b14e9d3e4dd62f1b4eaf588a9d0e198b8eadf99f87f33bbc4e3881e3bca98898e9a7e253807903b72319a6c351b4484173119e42e65d8c4b4ad874a7e58b3a27f1063fe781189160c5d43b9beda92e8bdd5e91198b229540f76d5e1daedbb1eaab64e0bd1e0494aa4f78bac76be82afda8c67fe310ef4dd2841a65f13ede066594b6cb4dd6a887252a6a2fc7ab54866cb962d5b5c3e7a9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517712838Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/array_metadatas, Name: checkpoints/3584/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3584/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=000662c4e009700e3c06b08bed8646682accc94f95259be92bde40a94076bcb2b9c09a5c5a86f0810e71c0ea30a1cbcfaa98dfb2f6861c3f826350a313c8e2ebf0d4031ba3b8f43ab32da683c5b6d3123a99ec004e8ce8f1e35d114235fd4963511b4ba5646e285cd2153b13db1435f617f6afb8b1bdd224b5ca82e22d5725182e4dab10ff4fe5ff16dbd87fac0853719779735009b86947ff970655ce7f2ef4380532e66a6e2f2a2f5e0dacaf767e2f2067b4ed7f1dfda67ef6adc34a61e34a1523beccb87e91f419b84657fb003fb9cf04670f10daf360ae03bc84933c470c4d39939a57033f6d407c1864be882ed5f94c3957c51c18b90ab9b57806e7ea14&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521066964Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/d, Name: checkpoints/256/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/256/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a8ff70c3a2a59557343f88569cb7a9ecb7b5dce53549dc36b04254af8779f9f4c0c362fb8c6890a72560b87ba4ae968bbda6adfdd6573a124168e1f521ce3e33ee3c0485b496f974e1b376b3b3d474aaa377fa5e4e61d3193e67fff8775db5c289a56d07a6fd54a9d5a8df737a881aa927ce5d9b2ec09b73982d4dbda8cae1bb7756e315b98be684e495d7aebf0eb269e03ebfc59fa2a6aaf4f7f94fb203d14d29c1f16d1d03ac537699eeccd6efb3f0b31c701d3d5c179d09934206cd7029ec51f798d69c2a2d098f06e63f635c1f8dd5b6d5a540df59e083e288d4ad41f7a7a2db0aac1ebee43d24612e1f6d53fd48c038932489af25fe701d36c6c53a85eb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521131374Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/array_metadatas, Name: checkpoints/5952/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5952/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=ac51d4e5f60bb1ce6c29806200cda3a94ef68175b50b42b4bd01d3266409ff92428071ec66917fc7d1ce221139707ec3fcdcf05ea320b76f4d731956c71198772b3cf42b1757e81f7b9e515db717bf1ea41ecf1fdeff76ad4329bd82b10acb609c709de1f3f631dc2681cf5eb8ced5ce93ba0a63d2f4f68fd33920f6b6c8edaf4191157f4245e902f98b3d82570594117f69655ccc71b21ef16d59b4c1b59c7fc5ed453eb8fbc2bf6d41f19e28fdd855ecdfa7273dfbdcd208c605a49e6f2c7059bfdd83f39e53803528fb702776477f35c3bf7c6af1dd0c25a1cda1c6f01dfa698d0ceb2a3f49b6205d917d72623897201f7660b056f0d36ba7eed2c562ccc7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520962454Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d, Name: checkpoints/1408/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1408/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3b4640eded7a64d314555875d7a978e5b636d2ffebd15340fa66a38863ee69a95aede65f781240b41b38e0a40f26decbacda832abaf11fc3bbac3d2dd75e9a02b96982bdb7e682ad15ecd01508b411a8cf69c7b2926663370e4435247233f5bd5ac19f0c129dc6f8b2c24c6ab1436d82a9771280ffe5a7ba96700824dbd3e790544b70fbcf9fedacd3588e1e6460a0e9333c4aabe80e0dc6c4882d8e71411f4465760e2d5a4c5409ff927327d69c7a67aa7cce6fe202caea9b4458b1f6557e738919e82228e5c27412556f86be6172a97e8d531c5da7e3cc772f04bd2885bb11da0a7b75d8ebcab411de161d108734a3e70978e4750ea84110c0fddb985a8367&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521197679Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392, Name: checkpoints/3392, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3392?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7cf847444ab3dde1aaa780808cc2a68a459c74ed69e001a632ac886e608e2829ac4c4753d120121a6cf5c429333721116e577e68e605750b66e632c0665ae7c0d2ad04e37b4d07f6ad35304a9ba27fd9b292598c5b659ae1fc391d04f7b9bca8991117001986a72cbaaf7254e5c6516d675c9ddd2ee85039555155b0b5fb9ac30d1a52cbbaadae75973fd167974ff8df3368972ee2ae05c168f341d995532ea7b7d80a47645154a760a2dd43322c32d36ad5d8209159c8c55f77b1121ad5dbe9a5a179f83e5749845d78cd0a62cdb9353d507abe39be0db44ac3f7d18fd55cd119af21d2fb827995c2cdfdc2dc3938f9b6cd6357849ce6985a2ae55a1bc5a265&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.5172366Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632, Name: checkpoints/5632, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5632?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7633198d2be3cc7fb6d62e44406bb5e02803c9034f2becc03d41a45312caaddc1833a05a6aee58f8e2c6048f3bfa8d595eaba18bda125a3dbe1fe21098e0f8c068137800d3905f493183b0d45abd33bdb5c499ad3cd670f52fef1cbb29ff70179aa1b7be617d2618c693129628e6c95f0c00cd6d4770a91e9b83551a3f0265fa437d4dbc2a442c67e6d2e3108a264cc17a25582d849f94636304a31f88a5b2695a8c0a0f4c56c2a343f8c716fcfb6174ec62fef90b5d726cf37fc291216f0da7c7e47ff9ff4608aacb3f4f01c5f15c6f7a5f1ea3a735029b94417480a7b1030a6444ce0a559cf6fbda93942ead08aca3986c1855c50c8e261c64974a429b5d1d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51729018Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default, Name: checkpoints/4288/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4288/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=855cd6f016cd580c1074eaf5668a7cfcf4c9e9932faf067a1bd4b839062a03000bd00d8a8b6254a176de1dbe9cbea5f818613d98f4686dd4963d689996a744be765685fbaaf380a86c73501bc0ff2a17edcb3902c02a2d1329ad03ecfa739444df4570eed7294a76284d29ea4499ec3182a4924f7be5e678b7d9fe116c0174f50c867c505e6e7216fed0dded57a4709eb51d41c4ba96f5ce3da7d8322e30dee062fa974fb7dde0af3cd9c5146f6e1b299c11759e115ed3febcdf76d73ffc3cad051c062eebbe81b680cd705a39ab8b4076627aad9238239ea199aa1456694d3c1f8c437b43b15eb5f3e3f8130b6d6d544bf1dc74b5fa564e5dd809083d951718&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516725941Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/d, Name: checkpoints/3328/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3328/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9378a071420e1a01dcb7f6c9a4782620559faaa2ea74e1edb0c58dff3a873030c52ee05d2ddc2e4883f2e93e187366f28be0d1a4abdf16bd93b801e353f836593fd90757b43f889407cc2accfd6d7e428ed31f490e9ae21870f407d490df2f9b508a838f69a1ab24ae263f4eb408150d19f7d4ef78dbbdfde9f7e89a077d8f00dff48a88bc1e8aa750480377e144e3f10ed8678f86cbb7eed0ddf4f39adc5d8d0412ef8cc87b9b00aab902c257bbb42c93462243da934ded1f753fa321b63199e3eb823149fb09347569ddb722fd26aa79caee25caac22ee5b41948b8b1edacee4446e370fd50008d563a2683453434f00704f388f5a261c74919d63039ddf0d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516933191Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344, Name: checkpoints/1344, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1344?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=21b9dc8c12373c3c76da01dea2a1c69bf461bd307fbc8842470eb79051df4ff8e0aaaa4aaebd53f4514257d2a9e21a46b3973636276b47157958f07fbf6d9fbf529841146d09fc16bb8577808b55233b449f01335651988c857e1e29d20f520f06602313ebf73958435742e12f5df7290e88a22d44707b232c77b8d0ddf0fc8d95fab2507076ff4d950941f7213429773e1206fc29e79642b05c89d5dcb97cb63725f2309a99fa1130412c9a63ff0edd447f04b9efdc913f3deab1e5c44d6f528f57c7a515907d53437f566d281dc92ef08f0b972c6dab7776f589237659c1c27755bd2b97c0b67e39a199357825c50b43017ebd595c090b113db5b0afd0c39f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516582846Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0, Name: checkpoints/1536/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1536/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=ba416798ca084a521b523c85a684cce12b79d1f28179f0ee51e0310eac0e36e0fb93fec515fb9c7a5ed12e15f97b95c2d9604e2163d7f85bce66de5750f5519bf118306fcb7cb22ececd2760097b9a4130eeed704aa8a62db488a16ff3c83df3c0ed2bca689259a02429abffb382a168f918f1dae92092b42466e8780b9677283dc4f82372468d8797a8a17c7d9c391f6d48b0c6ae108d40db9dc35158a32426a90041481579ae9685c8ce9b5359bce35cb5dd667c6720672e83bc765b40c26439b4176666b82587f186c486c339ab171b8d9e92911178b8e8750ff478921b3601adfe2bcd834ddc3ffc911bca4af160b5aef3ab3842418f1980435306cfd9e7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521144881Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d, Name: checkpoints/2304/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2304/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4afd74564f24ea63d10f249e4d241171a3f6315c29b5976874bae83202e9bf0e2716739d62767f4dc4424df29c145e04c1ed78e45d8789fcca28f601dc941f1b72e6c1c9b8c787c4ef61bc24805ef6d314e820c7fc81f9231db0866a342bd39040519313db75838feb54bf2a15f58207098fd977d19fe723d886e557fae6f9e0d69bd793e332a91464bd17bbefd56f003b02ec8528d2cc7351724f0b27ae824678b44c6c215385d484896612ee6073aee2096460ecd549387d166ff6267fff9e8655d26b496d2a815896fa5373b4f0b61223d4e77a59eabc136cdab7196b73970819ca77e1dde579240e2781e9f3245864c56975e1b474b192b22f54c17d7aa4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521257784Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d, Name: checkpoints/3968/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3968/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0c2ee8876aad2db9c3bf3267265150b3a7c763e4421286a151b45dca4cfed31dc65f5234a691ae3d73ec8f0033e2e248fe93d465a7623fc03f12cf8eb260a57459b1136a0196925844457bafed6ec3a14f3cf10da2a2fa7565cd012cc6f639a1debf689b0180fb8bdbea9bceb886b5aa0b03d41c858bef29836010a9b246bf1be1168e390f06c869b0a868e68583430fa8831b95eab1536c3f18bead3d748fe6b78c6e57347dc5d9f3f6b39ae9531a68aad5c174fd240cd797ba43ddbea38d010992f4229c9e7fe4f6514c120765747999a9135a801f260e5d7d0a7956a83001cc5201cff848e24021d787d1a8bd8a8c1a3c893fbd9fa5ba7d7223eb420a0c67&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517638185Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0, Name: checkpoints/512/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/512/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=49efc3b1cb8de0a13cb1bdcdfbe9b289f620b73cd575244106c57938a834aa25e039b2479874ff01d615e6f7fdc3590fc87a95f38a7bb5ea81cab4f4c8d34a7b68dbca34c26551e8cc190722df21c24b2fdc78c41349f44d8de83b333f65e0412592bb655804329650a1f91f0e8f2aec32322dfb6df32dab193cd0bfd5adcc328efe0cd987c6f883b36ad8f005c80f614e04d21548eafa8ae6ba100efde1e3809c66d83d755cd357b7f3bac1ec9a0a871e3ef70224eb4d5626d4bb9689a58713389e8b03d9e73b4c5cda27d4b3cf5efa3a01a5a5be0d677c9b5cf1508934d81ef2b92a31809dda788b6110408a73e05b103c59990376638a8c2b416e51e5a231&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483105408Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/array_metadatas, Name: checkpoints/5888/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5888/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=70bde3099cd389efd197a3886ab7c26f3db732d568f8d8f0fc057cc80d4e267c2312c067043e6f646b5c0ac40ee867b5d2a34715e8a432eb9d83a9006dc81be6207dd14b48c8b800dc2f1dd45e7db3d10388004c0945cbf4bdad30e62f99ae371c801bc105c4de54111f8a059682cf06edb41f8f0a62db5e3fb77f80ce9329b5954b24d6fae8e69c49d0dda813923f00957f3aeabe595f79ac5b09fedcc3cee07017b00493ab1f06ca299c2476903afeebedf799edd653284d87f616f1f17449a9ba0943d95b68b774ccebf75cf45e4d975822bf6b11632ea109768638746640b1288988468b95d2737b17e4205be772ef293a17e01d092a3fd0e130dbd8d154&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483103665Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792, Name: checkpoints/1792, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1792?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3f665273230e81fdaefd719bb89ce3512dc3223b9b028f87d16ac98b21e8b9438c539156ae169664252553382e50172a8064f7bbed129f0639022497f5a9f86bbddaeeec4b0591b74358cbc64e9922d198a4eaf99bbda3e8d78db9ad35dbd57e287199a2a0e7f61d0a4856181e6a95b5ed75ba5c107d1f58e363b375dfcebe84f16ead35039ccaaea05d84377082b898c17fac46d60fecee33d00f5f779b15b1141ca1af4f40c6616b6aa865928be38bc6008649bd151c020f9115dc514d052953bf2cd98df0c0decbbefa4711f7d7888327c5c2141904d7e06974139f0c772c4401b3a9c06dc6af534c9737752d5d135ef946c4f4c026f5c6080fa1fe5af661&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483101761Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/array_metadatas, Name: checkpoints/1536/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1536/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c257ada61e2769f05263b14bdac2b656c9bad4c216ee9dbf13895e89d9a9d97217a56cf8e9e1732503770a9b69e539882539809f4220e403f84dc59718184f6eaef45bfcadde657e571a55de0e4d6b0b45c53f84b61894b74f95761540c65d27908e0c659639bb5d22e5920eb6eb6a26ec606823d3ad239d6d2c9851243465a084022a62d5cfc3d354f538e91ab12f1b94ac36566146d6c5243a9f830d4a5f049a22630bfc8b770bae1cc8616e52f5c7998a18cadb1c1253f3fce97e6d9f5b885cc7d099136081f4a18afbd9360e9da6cb53d9164af59f4c232b122ac3538239b29fe83f6fb1a7b9e9f1c68e2c88c313a248cf6820c76a51ddbb307073e09998&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483090633Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/d, Name: checkpoints/2624/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2624/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8e4fb8431bb7e79a3d1aa90a007b29d02d7c5bef755403830e6682c8e51c86f99c0c96c74c4090c4da2cccc9b75b481361289bd17cb8f8aba89b739e46639e5ad2575e99efc0d9acacb61840206b252ac734aafa7458a4d630a5c8511e62a3a923d8ea67af6d49973ffae7b79df76ac3e198d39b4a1f7730df9673444261a79ffa9056ff5a5a026254e53857814ee22a0bd577bbe5ff18db0fb1a0f5bff44784ed467caa2c6c8a0e83d26d0cbd348fd04f21f844e4b01c90304831b87271088ba119b9290bc7803c23387bf8752dcc1e63c8b9289b9a7e1d36d00f139d37286323fa2dca65c236c76bc045f4caddd1085cc73d93d5a2053e3824030da927b08b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483117274Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0, Name: checkpoints/5376/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5376/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b73ffd9cd33744e4e9c6efff5a573d5a695e53ed9379adaa1a8969d7f4b22a26999c2e507fb0ee2141c60b3f91e989ffab33b4fa46c3377a2e466a6335e953895e5c93c5796fe472973e82658f861bdc720de6abde04961c05e0767a46880b21378ac094a57cfa149cc45847d9b7d3231602d8a5295adba9c8be44f8bbc4df9025faec8f93108c669ae6028ceedad3791439343fa43f46a13f3f0956f6a73c7bb6eb124e237a076bc566a85a4ab9f15a92e77a78922e9c24d706b3873969e8c766fae19a0ccaedabcfdfd406023d3211aca5de6cfc9927491fae159e602e7da00f05e42964efc13a855d73b111df88f7330d8fd92bb9a97a26b6f50e64d8f891&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483141424Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default, Name: checkpoints/5568/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5568/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5e92789f86b74f5ae333813f536278ee70d375788ce5a72a45cbb9b84d863c307bc35c9d269b21e1b860c69f231d664360d08177611431b36febb4fb24fb2919755e155db0e068be9a7236b00d9fa304be3d802fbb2024d99ff770e7cfcafb15da3827cf071c08501ed9456b2abb518ef93865d0d4ecaab265b1c3e888dadf99b96c6b256da0cc0631f5d65928c6fe3e8514af72e871bff241381f510940d557d5995e09ad44bc95934cce29e80847d6ac0a41c0c6efec7a3dc2799faf6d6b5b14df6166f8e77123e2e6da1d8c34566047b3ea9dec4b445e44b3a800e2a342586587c47baae6838da16740f6844d722a52213dce125c84ed93f0c20bcfec6544&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483111977Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/d, Name: checkpoints/5312/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5312/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=96e1aaa568d2aadb925b40484c64612c0a6519d85daca8a63297d9ca040487885b377d653a8d561ef343b70eacfc0ade02b90daa73092cf67d413f459522b124f93229e36104bb4fb6cef2f1113756d361d77c4443c23f0587425bb7f9e9e5782a1d728039cc201a3328981c814bd08ab9233709c9bde758f562713502464db038f7ec4d3f33f9fa1f901f5a4ba4ce352d74fb472e05dcb4d841d7ac80c957174465da0ac81d8d7bc5d654bb8e1f681af239cd2361f7111c0b84cef6afcbbd6909a601c006ebc53d9ef348ede4fabff0f5342ecba7ad292177cacbcd7b3354be7bad1db39b80a42a5cf6e4a00e4ea868f136a6b0501b1c43f5de3deb30768869&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483086609Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/d, Name: checkpoints/3968/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3968/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=167f5679df8300efd566e667b6e713425d866023dc44eada30c9e6f5ea16ee76723f9b4a659a00b1df4fbe0b86e16fc5a872953aac96edf9bfd266607b329eedeb950e003fed4c49caedee3bb6a0fdc5734f1f34bb50eb8590bd5c839cba15e7a15c683c56ea80a812432ab0cf52f06f16e2929f4de70e71564c0cd62405d7c4b3e62b3c7945752bd4d9c39a428d43599c5dc3e0dc38cc055af2979203f3b118f1566233257e8766263a5ed4f50fd4cae3021abc84227dbf4bda6e0a16d2654c84ff5977b341cbc0e952c5d4fa5c5c1d64091969d959986f21850da514960cb15843bd7e6cb8a0658191488ef1fb89f17e25f149ca67629e85e031f5dd9b4cbe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48311376Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d, Name: checkpoints/3584/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3584/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=29df74273d4a4f689e7e7216615eb0a16dfdad41b0bea97f3ea8277fb28cc3ecad52b250d304a7d221c0e6ef2816733522a0076509cbd8a04bfac3b4964dcbce219edca7c67e7747d76bb6be72526e6d06b26446f448708b362481bbe9a658bc899a960d03877ec267ee9303b25f43bdeeeb219644fa186adef7ea3b1226b4fa92036cb8abaaf99aa21c418a8974700927e119b448687d0aebb7478338ee8bd01e91261eddd88e6eafde771ec3e45681af783a9533cf0ac25ef8e5e2e2dbd24e8a2576ffbb2066d64d3001eba6df8f26b3ed2233fe09af1501915b59d40fc7797cb86a8f93a1649bd85f31be6d085d4c7055778f3fe9697ecc6bc29bc4fedef7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483150917Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/array_metadatas, Name: checkpoints/4672/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4672/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=94c7df99dd578ba2f20c9f4f79db83cb7c305a257c40b9e8827507b80403f3208ef1ffa0e37bc76582d1d0471c47d1a3c9eb0171996d489a1e3bbbbda17dcda4a8a83d8c497b7fa3dd156479866dbe3580ef757ad0c0f4ad89920adba87bf01fb68206b1f7fdfe633caceee6c14da28083a8d4f7dbe6340de43c1d933444ecb1db80ece9a12f198ddd8de6350a9e4d8abadb2601e753cdf91ff1833aeef3d7a91ef0b2d5fd8ad571a61d28f98f76d399b31a1ed285d6e0819a25069ff7821621fdac9e21b66086f5c1e382074054ab92cb47c582377e1655a7ecdaeebe13f45bff0ace9ad522a9196cbd9a274a6828d7dfd100716131681ebb6fcc5769f102c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483164485Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/array_metadatas, Name: checkpoints/3456/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3456/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=93c3e9c1a2a92632485388735f50eeb592c883fdf498e91ab57800365ac48d151974f3b04cb0488a784ba03f8181b06c72f61e766580f2ab6441ab00283e3de9019e15fb447cacd661f530fda5a208808b2a4a0783558440dcb5d4415756e646001e7f27cd27cfe80448f2c9317b26160ddef7c3b4fa33c153ebf46f8453121c10b6b90891149611ae3e4ee0ecdf9e4460fba88d6f62faa54c416dc0ec485754bf7c41ce484deeccf8b84bf832fd677b2e822b040e22029a1d0e2cf1173cc13ee953269c54e6e4c9a308ebe380b5144c66713c12aa714c95a5e4b19fc9e2a4e2f551a0668f02f07a91fc9d7145aee72d05d04168c3a15a73be54bb1f6a7802c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483141549Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d, Name: checkpoints/5120/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5120/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2228682ef1d9beed775c6c15e9d053632cfd89ab8095d603d9e96d5a921db16fd8db1f2fde01427773dc891cb0d84f4fd3859072e9a810a02efae5da515fccca75a4802576b6d5839fc1db5bc5419cfbe4f9fef2116db3c55a0bbbe13c82957663cbcb262ed389776d995495e2ac38563749d78387e316cb40148b9dc60f67a14333342cda44cb4499dcf57ef8235069cecb96fbc6fe4472811d9d0847a37afa76f284c3fa7347a1c1b7dd8a0e730d84611696e88500ed2a6c5169d1e25e62c98147a660483062d6539889ae2ce00bb9c19d9a36de1c586026512e281397e0d7011f0fbbb355effa0cab4f971f63b8c97d328db9ce84dbc7f26e36f9afb564bc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483148649Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0, Name: checkpoints/960/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/960/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a38b72da5c7b6f90d72306471d34ad42069eb05ae6dd475be7f3dedfa631319e08bf06eb50f7cc00c740119e7ba8c35ce7c8c74551789c008c435fdafbef6a10b8deb6d9053184a3da310e049c608ce9ca4bc56eac8734ef2ff1b739a9449064c8805df9490ae1413c600db9b0218f77db1449b262bd7124665ff9c67dae635bebbec3fa6ceeea2d62d74fdbdff3d158cad0e52100db3219e21964d0ba8b732fc647b5e88c6b14b847f5f6525f7a72deda9c06bf159b2195635973339c8804636192455b57e1341b9812f8d7947fe8bc6a796f9c2042dd3c0e3633b21e1115eb79d741e425e7b5e54ee5bb30d16b01da250baf8017bee2b7967d274ad7c39bd6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483169381Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default, Name: checkpoints/2880/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2880/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=26e0d5f98141a7fc49426e0fd8fed0401b5148842ef34c4e2c18edd7d440636c602e0374d8b25ee4f4f8163d65ed8dce2a85cfca9066b8f26eeb8463a4af8b31b8678efb6d96d525b87d77f678a04a86356e9d4196b2b4ce9711d0e1b7757fe3cae3c502c7f02b115d3d816f23445b90ca0984af731e30b12e38c423df476034872c742bf48edf6a083438603193bf68ccb7c50096fcabc011ae9051392c0b41f5ba184110d1c812abc50c1879c457deeb02e8d98ddb3a3889c45e388a312cb51a434d263a6ab7b9bada9f9cec0d3b028f36236d03297e7d349eeb8d2c754cbb9d22cf250d3350f478437cc3d1ff686e1e28a1245400bd38b8cbac0999554c1d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483148861Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d, Name: checkpoints/2112/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2112/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5bb62ad8fd8a853621c24104b78eca5018961dda07b9a4332eebd095b6d6f4d621d3784db3e2ac2a538ef35f03342faec752d4dc55418fbe96115ab2b0cd76a37dafbdb236427747d8f8dacd518453e90e2c620192d4926f9a1380f09e7a2b6eb1ff519f34090724b27b106cb429497abeec5fa2addd260db0270728232aac361011daa370064ff9de8f14de77f0c6d9ea874e8518e487e7198ba0ce3dee18b9322440526337177013ef51a8108cd3ffae58b87dad75b5d3449770f790f16e9d633044d192f3f8dd557ffa028b70472932b39187cc4b865cef788af0fc2e7a0f52f2a2978f39332d39372fb0f8d3117b8364f506616ae16028002bb856db3872&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483199537Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/d, Name: checkpoints/1600/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1600/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=614f44f6cc4c4e37a3906669d56c57a2c34c234d788bbad24f9743bb12eff6f66abf661cb5674fbb09d84c5a3d91dbb48b827e115b26e9a69379cfa63bcc57da0cd36002fe482dda0988e81a10506c6830ec6e595c43f87220cdce2fa141b1878a5b0143fa7ce49de40b171e53413b2ed253ed61771d00f8c9e07ebf1418f6e5fc4f248fc09c180975b2dbb73093a4616f052c452130bcef106c28d004f5d4a2e62485e6fd022f2fba8c94aef35b8855694659e6e87a5d7470f7985739c7d85de0a81fa6966c6d4ff74e86e60255c178fd7b4ad1049efdd84583ba5208d2067aaa88f026b697d465139d0e50c789e618e09e5f00063bc5ad01ff0b38a90a45d2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483218763Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/array_metadatas, Name: checkpoints/128/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/128/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b23c0fcfe620759937374180f1566796ebecfc2474272f58e79e8062362cf3aac7fd32dd29ad49aa78ee148f28e2340e155f7e9e3ab14d1204b3e95534a5d8f2b5b915494cd0e69a870dfa142be31cdbfab31c5d2b9444608b566898c615f05911dec82d4821df08021ad80754c261d7bd83a7683a5730e34b322a9f2562e4a2268bf47d182b0f60f034d14d52f44396175f34f2974eb33a4c351584100e3d5939f70e458e853b061bee188a4f38fdd545dcdf9979551386d03a954405348b55aa5597dd768b11b60ca864aee0c2a88752ee2257cbc558e6e38355783d13ace826f9f183c9303538ec51b824e2b70d38f8fb2c306150e4a63935bf4774ad5006&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521468127Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default, Name: checkpoints/4672/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4672/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9359315263890c372ef4efc7781b6621fd201eac32250511a4f051b04dd83a6ce33468af5169fb069893aa8f8ccda0923bfdc827878bab38eed36f24d876c9d4aa44e333919d14223436b14928c1cd57a6b3be6f8ec1b44569698a6e4e533680ca79144b1b1ea4139c63c54e2901fa0cb33632b4d2a24b2bd8590798d5f50afc621fa4d1a06585ac2caeae27440cc56abd566a37ffe3780b2203d4c2450afe35617612087f6390e8fc7aa2a37ac1b29fac767efec0e5f1bab80a280c856047b9f7a8dbad48c9b4f58b1741e08efacb5136b168379e408dfd3dbc80f4387d8c1311893e0e00dffa9cc69080da296a38717b28fe6fdf2c9a5cf4332da64e2b12e3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483178373Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/d, Name: checkpoints/960/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/960/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=01012804b9edffb1f9efc0b2256d170c44860e8916d8647d267fdf115216f98cbd45e50d8fc3ab73a924238e2f18a1221a1d8c2a6732c91b536a4bbb0c8f5425df7eae0b895cd0ad68d7c867260fa510065abb5e839946b3789a76e9e3284579b83b885afd88b766bca468cd68460fe60daa6b094fae6c742f59f2000604d986e829dac1fdf0e1f9aaf4774a61ea24c2c3d417df32fdf3ff1fc760dd2c14455d74f3b5f9d3a97fb57e41aa4f028d7ee716a60fea6a919f2d02f56a148f5e824f06cf4549e6b630dea822b1f011eaf3d29e7b7721d6a4b71bfea16f9c3dced4de93d89977cf3a43929a52ba9676a498e1a786de66b4f38356869276c10f082988&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483226322Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/d, Name: checkpoints/3392/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3392/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=892e20a60784148154643a9f37ac278aa7b43ecebc89b37f5398bd64341eb0447c24e3b42bc218991de6bcb1c31b0a8939cd4e25f420bd1dfb400ccdcd29c7cd264280806c4743a39b39a26af67490e61d81c49dd05895750dd74c8d31fbb3091a72becbb0dc33d0bae1683f3037d797c15027313f0cc52efb040b8f30ea089c72388059caab13d19e1c3b3ab0f1edaa0f2f918e28bfe93f32a333e13e473c46a6ca131e337e001f8cc97fa59aa7c7f7e16641dad20c6d16cc3a2ee201c8fbc4dab863443f7e52ca8c1978c8626c9c45dc98875c688e225d9fd38f80c30fe71eec30289e2cfd974f8570072e4f120277889b7889d3210ef760c2fd71de9e458d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483268635Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/d, Name: checkpoints/1792/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1792/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9b9c6e52f514657f81005d2a0adedb27bacaaf7463bad132992bd81db8d5e0ead58ea6c96133dbb4abeb92401a86a52b98554d32bf017e16a5cee06734029a1ead5fc43e6a4f03f12d59549722020e26efa1700ffb09ff19a3068bc7646f36fc9078f9b35e9249e1928b13724f815c948f2bb035c84f244de076e398615dbd3f72240fb46c24350c742a560f461cac4360a8b5d476e001340f356d769a8956b3aab858647e340b67c1d8095735c4dd294525405c57513a988588241f7c75e44cf6f8776f19f4b0815ba52ee629f12f46fbcbe52a297cf8d3284b9968d722237fd8c8fdb10875d1d2a7551bc04f5a9d130ab08c58ae53c9dcae9aed0b5a39a9b5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483265625Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/d, Name: checkpoints/1536/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1536/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0706f9472c36a911c3f21a6cb4aec504b62753db9dd1a1bca370663d94cb895918e55f0f03a319cd128208391ca83051600b5f8ec6e71730d98392a1f8f6d2f5b141b1e938f1b4a963b0bb57d2be4fde57175f9c069d2b9ef9207a82dae9fe2f31b9a6fcf00dcf6b82ed188ade2050976547e2cde2117e97a5b72abf00f8d6dd34a15625874988daa3106b62de51213a7117b3914aad283483557b84d34340846ab22eb8918ca3a7b4ffc1e9d5cf545081a5437118f4bbcfa0202bfd23d75b00304523c696714cf9773ba62e5af69ad9752ecf708ba7d8959860676df9b3a84e69810d9dd717ad3ed3443d4a18ea8d9f143fe6eca13a81cbd7ab1e76c8462c53&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483257847Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/d, Name: checkpoints/3136/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3136/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a56b08e382dd2a371dbf2983ede3fd768747a76cd18ae4706b95cf84a0300e3097f15feac0bd7b22f2cd2b69d1cfe83650519c259c05a901a65fde76255c018358f6f4f70097a1113390ecd26ef2f25703f3aa981a3f25f6a9bb63189aea37e6d7fca0ca87978f9fab5295bdca3476c1e925995edd7a83673a13cdd82e62b4c5117e0d3755e0460c5265fe870b766ba5ade85ca11557cd7168a3ff7f5ffac3c1d8a6d3a9a7395fe288e2e32f7e1f1989f288765ed6993eb338ab539da6e3fe3b1e42c82acc433552c2fcb82bc2077d5292369a3062620a76e6b8ac2a987d4a57ad1557e7d1236159b8cdf8e119ced0d0ed6b5cbc76f899cbff68e767ab8e5c8e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483222291Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/d, Name: checkpoints/4544/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4544/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7aabaaad8ac3c95540055cfe980225709f0a93cc224ee0157dbee7221bcf0710d122a227180d257760865ca08bb6a6f19d263f2c52fc804a4e09a4969c3c00f101d4a8efdee8c07e4e95a4fc6240db7b6cc6c119ebd7b6c214ede0b073a85cdc8c35f10fc9848c4438786f0f6760e18474f6ddc69434083d41eade0eae995fc148faad0a0623c4febb379951abf08f191b5c1f7a644675c2378abd37d819018d80d68d7ab1ed99b2d78b94e31eaab359f44190b6252641e031dcdc3d522de1d8aeeb0b51b57700742d480273427f157b1453d98dca17bcedbe802e38a2337157e0999c50cc25e381c21ac36a8599fbb5df9e06e57964cbc758bcf47f6363e754&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483275544Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/d, Name: checkpoints/3648/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3648/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=786fbc51480845d99d3dc32a8765381a0f5188767819c7a2c2f7a261e4af574069594fc41dcc3721984d55eca6011e085d473343bc5e978bb3f5e92631c3c52cfc7c813698f1c4a3eaf41782307acf9d96938e4f0036df87ae50c74c3363dfeccfe8b1c077ade6cb5f7c9822db948c3457b5fa5532b25884b8466a1ab1b3ba19a305dc65c38d023a33332f1b03a50ca37e84fa5a7fba066dbc80781e1fb436f856f9b2ca0cd5ba134b7f2d81d2ede39067cc17b81e79a9cc5762141b4f0d1574ce8ebde3967894e00344feabea9a25257da82d583fec4c918c8beb474bd349de06d8b3e59fdfe276d3043e7d72819f5dbd36de1882fca095e2511180840d195c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483277118Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/d, Name: checkpoints/4672/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4672/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=af43ab7c202287786a0a322d451cca62d4659b7218ad1850876033e6d1b18cba3c7faecc3fed1a05f4d50ca1c435ec4d9e99b7f2b8ef025d6ce6d9cf59051a909b947d4853466f3714518ddc92866bfff1f7aeaec39e1dfc87b70fb4bad72f667fb750190851ebb346c7d39c7a85b20acb9a95532eb3c5c1aaf7976b6ce5de468866f98f79b39c93e37609cd6de73df60c85f8965b14d663f86ef5a2010d83cbe35edbe6037426f370bb0e279a5347e0abe6779c38a9dae6d014c7ebd1471a32e6ff12dffdbaa86fb02f0944bef7e3377e079c63b013328d4a144d651fc369cf8980eafe2e8bb858fce9c8f49a7bfaacc2f3eaf1a7a85ddfb1cd4e27766c0087&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483283942Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440, Name: checkpoints/5440, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5440?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bedacf337bc7f90d46c5ed1520df55290e5cc9df41ecfccd25b148be8f75a21a36788475c124021e8d82221ec322347eace0ae9d64e42a6aeb952aa7974b25d21d674fd09539e83206b296df8150eba3dae8bf1964df4627333a3dffbee01c1b4d082093ab2e3961c6eb81c93d90522f2df6dc61a4db31000af2dddd6f92fb0577b7446c738380e7e6e526730b5bf68d845ee2ae82da471696045ded41e82693909d6d1532f7481142d25d2060d929454bcebf952f0a6e8faa9c27976e966a8d2d46fc8dc42cca03c8dfe45216d8164c2c63fb1fb6f6f626782915254095a7a09099f43c0162b8765e09b33c911b036b9b205cfce92f0e972913e45da5418cb7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482879189Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d, Name: checkpoints/6080/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6080/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2e6e6c9f26d26f0a4d1ad8bc37d0bd9c43664c90c093028040aa5fcfead3fbde5a882ec0e47db4cca657731e22db13d89036eb261fe394523750b69b1a9db713e743b755cb2b427fd017109b8f425d44674c8d319c5b9512561e73f1f1f7f6762100386fb1bd689ffeb09337bcf2af68fb60b250f9fac1bfe301f7a7c423c26d5a1169a2f1aa10e547692afdf437f8a12d0435065aa96446161705642f3538eb92f5ddff42dfa7b5cd1d87079c152c86221af1cf61fe6024f12e926583f5539c6df65a9fe11eb59c583a37605caffc7db201785457dcbc752897f9f4179813bfc1a49f5a90c567dd56a2fedc83e056eae80ca81c5712e51a89f94dc945892ccc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483274895Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/array_metadatas, Name: checkpoints/3200/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3200/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a37dc4d4fdd5f341493a3df62b43ca5e1aaf698d83299caa7476032062d1597e34589dfef8714dc700e03055720fca956dde4c816f98dc774218822d909b67a38985ad01b447c957637c3dd1c1c14a2bc9c305ac746d88eb40a881ea652df4e4cbb5f891ba8bcb6ee8dde3ae0896727f6de916992f5b8e2cd1375e123d70e435edfd3395f4285d3aad71f52e9a102765e8160697f508c9a8d8cace198e568f440d3ab7e885c6fe5571c49e97c7a3b14aa1e3fb6c22642952c8253b3545facd26e9a6ce7783dc35c5e84b7716d9c93fe2c117b0b88259443f64cfbf45ff0381d1cfd9ca4dc06435e51e0d2171e49a8e10eb13a2805aa1314669e8e5b84f182fd0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483304334Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/array_metadatas, Name: checkpoints/1728/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1728/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1b3c5629a4dfe8f7bf3526a42a5bcc99e1dccbc54d2bf2885607186e6f47aab2549e128049ffbfd834b3bf509ad6ea8550067dc9cab04d029e491c156d92bdfb06152fe769723f8da726da7ce172b3c3e6c41db913aa9248a443730a23de5472aa7deff1e20e8d83022bfbf41dbc5b3e89aebed1f056a266e6729b6d0c6fb815950fc289db5968df6b1602a95cc854a10cb14731ae959e67394c1b183fe6b224904fc5d2dd1d892360dd7ed9a5bccc3139d3ac6d6576f1084aa913a2e323b0bcd85aa681821d68b3684bfa22213cc7e700ee5279bfeded89b099b47ff45a44604ee9cb81c731adcd2b5cdd0e93e761120d6e6ee9aa4815465cca639ce57154aa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483320386Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600, Name: checkpoints/1600, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1600?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=196a9edb7a5d5088ab446602efd9c73dd5634aa4c322e4c86c6ad3fe7dea442df1b2a81ee1f8be722587e78c4861108365bf78904b0584cb6d86555fc62240dfc1f43e25ab2647ac1c668711c15e84ff2ef64d1d060adc5a32beac13390043958a36e13b0c6c1c894a545433d9b2643d9e40daaf96923d8e7e7d8146b74d1f4a821fcfa47716b42db402eb371648666f7c68962f60e7a5a87b570ce5f690e0bf2c840a632e630e7f7bfe8cf7f999ca110a7d3f3980bc01bc6830a2369599b29a3f3d2658e59b1fcd6a3a15bf872c41e3d9fe6d62cac4f0afe86be0710e715dd6545e461877cd158497477087b76769ff0c42f5228a117767a94dabcba695b096&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483315449Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default, Name: checkpoints/5376/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5376/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=432c68932473501256f6812a91dc495a05c92b0d0775b1ae587f1245bd6f109141a715aa22af3468153e8b4876d70b58d2f8c8d1bdc751a1370de711218f706816f9ea76590fecc81d7977ffa46a381b2393f9036941686410f7e3aa77c3b7dfe3689937b3b166af55b752bb539eb635ff3aa77c81b7603700997330b94d6f0ec5b88784f182cc0d4bfb7dce58293b3af3531208b3ecd7c8b59d4a2d15dc33622ee4983cd2e1df66b06f1f2da047a6e90d39d6e49aef963ec340b34a0bfa911ba1c8ba58fe11db007f8fa1f521be74b544324bac2d037908e130b808673be136694db14ec80bbe8c3ec0e0b64701f5e5be8e472fc475a8817116f103746a510b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483280495Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default, Name: checkpoints/128/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/128/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=48b26fa95a46d68ea1d7ed084485b31baa16c7562ad2959246cc4f06ea1d42c5d3a8653a89eaa11d1a1e7244a6976e65262346e80f96fedc948f50e540af5dce46054e7e75edf87a4620fbccc705032f3893742cf49289e5517a956b894ef2db09458d49f09e9f5537ebe05299f0fdb00cec7a6c13e19692c43ca081ba80727d53dc30d6e41a11316ff1d895bc7f443d18c999eead0e656318869f2c56c689a0e6fe28379fb7196aba57fa335c23b926daa96e286b74adf8f2513a818e39d172b06661d5546740a44046ffa2fc2e5739fe46dd1de6cdecd3f910bb8effed0b75b170eca0e658687f82ea81a094e691be87a388d5016927bc4c7228d627d86921&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483308831Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0, Name: checkpoints/320/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/320/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=00d69675a5a4ade19d677ea7dc55215c60b70d58d6fab19d1071373a88f96d20bf380d164e1ce2d429f6d036bd4b58c3bfefca2119946bac9b65136582da831c2b3192c9836f2965bd16dd8fd861a7c293bf17d27d2e980f239f8caffb7ec587868864f39dd03f998593ce26ab8fee683d69fa53b02ef48c1721a143884e75f2b2adfe04d40a46367d80ec2598febf0b77f4996193ea360d5dc1cc5d28e4fb2b2daaa6e58dd669ae41ce9b4258ebceb0b0dd9d2ef6fcabf08fca87adc94a5c9f942ef65f34c42fc1076da5473392532e05309ff9bff075ef7f35c50f8d62742f2bf441e8f6ffac6a51fa3a02c626dd9cd7c842c9ddebee4812b5e57604c94634&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48328299Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/array_metadatas, Name: checkpoints/384/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/384/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8f19d10f372ef6df22751c281f1fef67dd2d8639349af7edcf7d8e3664459735d121683c3e53c2733c20d9a33d0d34a551358181d27d0471881a3264102bd524cea800f52e76f5269af723d9c754355b78f412ea5922d4c0a2d5151918d246ce103cc42117593d92714e1f7886056c40c5c7bb9e9a31743b4ac8bd661f41d7d92e772bbc8aa51729d8241f15285cb5f84666ec7fe7fdcef29d601054dfdfbe880a7fbbfa4ae768f36abc9aaf656925f9b8dee9c7909ad4a32e723a173fec2e587bdd49ebf31056240de0cc57322aa901f9fb29ad84b445fa5715d99297ffac9b8ba9f05ba80792b76fa6b90dbe7c3c1d2b65ad91e287cc33b8af797ad2af9184&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483351825Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/d, Name: checkpoints/3584/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3584/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=20d6328fe9499cf84a32cfefe09898ecdbe34f8744fc864606d3d75b07fbeb459521915c8a513c8396933150c32fd44e011119d0546403ce5667e952a45d2a502d1a56bffcd9fbee272fa7fcf5c3c049d49dc51e4f299bd1d31cd0be4c9951d851404d781dbba30f143756872f8a2673c8cc9a51ef511590d0b5ceab9a64a11077e59721195d3d4ebe681715ef1174b7f23c3f531803cbcd693bbb0e4112e2be9e1ed152f15eb9a5bb8502e00d4b19b82c5e13e56bfa9e24a527202155dc7e3a88393c75a56b18f1e1bf562b008b220ae42f8f63b49d38676808c42cfc50a67c216a21245f77dc0068a75f932bb169a7acfebadfa9e3990f770bb5dfe307ba99&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483236769Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default, Name: checkpoints/4096/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4096/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9a5d2b1595f472f5d0c9812c382a65bd37388d0bc2192c893ad8d8522dd071a816ed5b5a5dcc89824f657a73269d042793538193f14219e02fa613cb7847f3a8430f704dd9aa2855a2ec376b2f87bd42ca5aa2dc9e1fc2ee5b80ee979e38afc04dcad95c35123372b1acf0dc76e9f6909b39cb2d87cf59623e2935cee1e7a1914511e1141cf92df3cd7ca3ab7c09bf00d4deebe7eb814e65b497762755d0c088d83152a5e8f505747e830769235979d37585dbbb8d63c8c4b8e2f2e29bac6f5a36f315219cdd9135381f822ea3e33d469999939aec6c174f58ab883ad64a5fe0972991b977714610478c2e1ddde47ba7a3b0d9d3e3979a64db73d435b5bfc4c5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483307905Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352, Name: checkpoints/4352, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4352?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=00f11d793d0b255d1ae20b9b6198bc5021bfdf47b046bdf9c591142d6caee18ab2be7c5b2d7eabab02b7a656585a1adeaad00ad1324e30397bc84f1baaaede6a2768f4764095d7cc578a1fb11e21b517edc5f469ea32e375cdff93deb7aa909221dd6b1b4e3b3324e7521ac7f67865436fff0272f1a1b561046affa3dac16a3ade1bf6a0b287f8393e2fcb7fc8e953581687aad98bf926b9886fdd52838e8fa8bbab0d40cd83252365626d66379b9ee716ccacd39255f986d1a30228509311f1522acb56d611cb4c2d331f8992302efb740d32d776361253bd26e065c8e6b26a8038a619d2f918970afff92c76dad81076291570d4e04318e8ee236de7b7abe5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483383076Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840, Name: checkpoints/3840, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3840?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=aa0ca352ad727779306f8d48f4793afa998599fc62218e20bd98c80db5530b243217dfaa1bc52d732c1078acff32e03d1e55e8b00d922348a6a452bf370ecc029796360a43b33a99850287209203a180aacde9c47624c0ce456e01baedc7f371b054ae7d631cd7311da3d73b3c3a2b947c0014aeecbb1c4b75ce6971247cef47f108643f3289f25cd518debce25af92b4c11668eb9feadf2c4c894711b8c250785d6e9769db3c85c8921bd66e1b18fb1674e26fdb40bb93f7d9ca03d79a55debdc66f27413589028564aab57e714d756f4714119af6df2ed3c0eb4644153988d8fa75bb1cc1736236565fa43b1fc50bb27fe4f092cbbad277c53e101298cab6a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483327989Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default, Name: checkpoints/1536/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1536/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8eca7b170ed5402a0cd687808d6d3fe26801796dc93a6bb640cb99f2791f343d5b195d1a4995f8a82754ac83e8c95d187761b010aa082938be02352578f818d70fb83a64d99b861ff902bb19310311ee4c9a50c289454bc5f8377b57a3c5f2ecd9840c2479bdfa8afefa3747a49f5416f89acc27ad2e497e35be857a11da9e767b20ea3bd2fd08f7fa12d70aa70577e7e4017b302b48f3089b85ebe00f4699332cb942d3ab7130befeb20b6b481d25a4666e2bf21da339f43e4f3ca3f88246d84c461e29ee9a7a5f2422b0bd3cf0926333068e4e0cb43d0e974b23a1f984bdf6f93ce004e1c1a49f2dcf7381ad1a9bd2d6236c9941fee7a0e919b1e97d532a73&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483386717Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/array_metadatas, Name: checkpoints/4096/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4096/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=99f46a59b501576e1a495c1863556718455c9ff46c8f06371aa15bc7987a80f3736fa10d6e01ad26a09cc786ebaa944dd48563c57736f1ffc1f9c6cd7b6cc9519aa6979831ce91ab9a1c04b0f0fdd09d8ae8f34aef1b70e055141e9ce7c97271a0e791b72e46b31b338298fd4c90ebcb97882a110c0446564fcdfea8b0c91722b888257bfdb4519668735ce1fbb26e0078adaa770f7e65e892970fe84360ec0e5521e4d76fce51cc0b7b123935c56adcfb8136326ad4b297be78704b2198cfeb08705830ac15223dace417370fedf30f3629e9b9eaaacb74ffb6120c9ed4266d61894765fc49179df86186780836b9712e4635ec3af0efafb9edfc3262857077&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483344625Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/array_metadatas, Name: checkpoints/6080/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6080/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=924ab0332f9910840260dad3159b273146257d6b5f23d1f95df5603300df31a96a7633b82efc57eed8d3be74cb2d805df929a6f94f0cb5dec2ebe0bca1ffa7b6974418ac3106faba87d3b62a6aca0b18a71f10fbb29bd23a1cef53227cb5c22911a2ba15f59c06af5555a08f74dd713de17a0a8dc4b4c7d1db69ebac0028f69f21aebf9e75d8b47c0e30e73a7dd3f2a9191898be9226bebc40c0a4a165ca7a978b82c04d078ee382b1cd9525ca29a8918d65142f76d4030d5542da44090f1525e2910a433ca3f2a9a129e11b64f4562dd2bd1018ea67a1c3f7580fa9c34ae8d47a7a4ce5d83decf80e486d364ecf25d3745e77b7a7d017be1b8512d2455fe0a4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483375007Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/array_metadatas, Name: checkpoints/5440/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5440/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=77b4d8ea9b5bcd237df37619ae80e17357b51f3da368e0b4271b07ad136880356c8fc808477ab23f6abe1655049c542cdc154c973bc8778b9e5b1ea61eee2458f8bf7cba857fc6387b5911e3fd1690a256e9995b85812318c6c63283ef2fc6b63322c4eecd4e0c362149afc98eb60b2258be8ecc913efd28e72fc351f9629af3bc0151030a9c215646f3b2cf753c3f54a6ab1e5074d7e6fa58d5c32933750802e2bc71711bbd9e3aba6af869ba822ba55a3cccb39cd17dd121298be25d1564fc9c87b1c713815c11572a762ce44441a1fd909ce081f83a767e5b95cf54f70e8555860735a0144b3ac390d8fb4023fdac6388e2d89c3fb5a8f8fbb0987d3dd0f2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483362691Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0, Name: checkpoints/2112/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2112/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=25419c5ac634b314a6af57426735a7281a6b29e573a78fcbb28c2bd280d1f1e2fe52dea0e566740cd20db536280b835beda8468460a622701194754a0df9873c9eb97389c66c56f5427422b8a2f88bab200d5c2eea4f9a6f173499a5c62927e983f6de3744c1e53dd5dbc738789a34b169849759f8e8fb30ad29aed326deb8f99e878c32677917e3523c2986e55fabccf6f85d8a219e074f4d4b77e692d7a59ac43256277f0f59c4103c6b4dc351af959d273f6930683e74b2d4e56e3fc2707cbc44d7209f012eff52caa342d15f1d83f44fbd33bb7fadfe7c364c94aed8a4f97c782573513216061983a3ce817002ff8a01da2d4254678e11f95f29a94049ed&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48335492Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default, Name: checkpoints/3136/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3136/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2b9b47dbc1b22f46e05fbed0071500b2f5e830d1129a70c49229c07a9ce81ecd7d3f8ff06b43d63995ab11c0aa9166f3a367a1f7b6f6928d5d567616c8c56a8c8832b8e6f408b527a3dd25d68d187a53c45b99fc65c980d1b5331f00082ced08dfc1523a9b2e3c5fa9e15f38850f5d1c5dfa95e4f597513facf1f00015fd2b9118378599aae2168bc329d95ddf310fe469b7e435fc75a264ceef09aec2ecbd3d206b21c1a5564dd29c91ba7bee3a82b01ca7804bb7b1df4a4488eb2373ae4155fc4d9e70b057e9086c5bb84e5abcb869089f8542bc749f897b5cc4dece2503f2ff10024f740f4a4171ec09c1618db7372a47f1990e419faf6d22fce4158cf055&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483431704Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/d, Name: checkpoints/5056/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5056/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=be3713480598647f98464c7731b13ba4fccb04cf90eaf0023950101711eb656f5550a3f8d01ab6405fec3fb3413371d6bc3d97506cf36bc05ff64bb9d7d54e65757b6933cdf392dcae99b69dd10a6c953e6f5caa22c47a15cd7f8bbae3e6e5fbc02a7b4ec9e9ee1f95e917d00f00b822197a9c2b9cb916564c1e648336fbe3ff595f3ab9d9e4f5938f0293507023567b955e1f55cb4cded6b44b8892a934165bbc8ceae63ba1573f1d9f0beaa6af47b0ede8eba5fb1794438c6673180791df66996ae14512923b14b95a17814039137e30b7ede33afb6b391803f8fd4401280626c1433ff0a41519ed15587128a70937adae799827b5aae58f879e8d25477b92&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483419996Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d, Name: checkpoints/5056/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5056/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3afd2281fb05e0c4fcc1bc59d24c841662d46b3e8cc20d50cb5ba894f61ace468713989ca9c60cfea0f620033b5a51c6e8d47fe51e0d1d31e4c34c5e3746c81ec97be4a0995da85e22e349fb32d70df8b47b0f42907969ac57cdcb99e3a1f044c2a1c76ba9eecc3a9692b65b2793687edfe750e2fd6b24fdef42ce89cca46d28f698ccf43b3b56e5c0ee52e2ec596cc43b55d2852d0b163f582da034ddbf143ef8e9b231cce96af7f993a88795a1abe2086e2f2b932403dc37c099149bf7031e0ffcd4ddc067c54a9bd77acad648f0e601364e306acf3544e923eac8ba50e8307a93a8e43197648ef598e23a68832547ae63b3532492a22497b1d169e5bb419d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483424968Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048, Name: checkpoints/2048, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2048?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=40cd7aa7372f8c97f795d92d5b5658e7e8fb72f2c61f6ccb169bedd237e2587a26a906cb33fa071aae9dbc0593b8383e94e5a7982c1bf4c41b077db6a53e66ade7b46de4842b8b0f25da341b2949652014169d0c787111ae731a799cda935cdf05a44b2ca9ce9b74261f7477922adc3d0e963540be6bb46b7fa96b7a8bacfd67e57e6f46c18e25f8b23702ddf030d015e1c4fc3ba6753f80831078da903aa44211b53effbf142218aa4edca1f0673779caac11e8dcc7aee1548e0b9cd96e284252cf1f2b279d5e64b4a8a318d1a5e29a2e9fc6d7946f632326770ed78c5c9d8f97a29124d65a89a12bedd7167e674b3a3286004af5ca94cf61065f2aeb29166a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483476652Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968, Name: checkpoints/3968, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3968?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0a60e178e79ba0ce84b2f4cc2b0ff8dd2e4987e2ecc450e38645179823761d461ebd4c7394808f5d978e28eb770c5fd5e446fa55c202c4a04e1ab0d864c4290c71372a28684c4e86cc6c0c67ccc5940f60ad93a4cb28deda55e1d7562a58b80a79ff6e58ddf4362ddb150826dd4c662252a9ff059a046148448fed9135f15d7c4632327fe545ddfbf356819e1e880518adba434b6893716c6108a5995100383eb32b1e655202183a51437a0dbaf8f4390e74dc95a73ea30ba0de7f8a823487ff1d311ca2ae7f5355f72e835f1da13bcf50ca357abae15947fb5921f25a6902763af5d34587a31be5a22215427340857af12f15f55c2e03391ebf7f6457c30253&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483870122Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960, Name: checkpoints/960, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/960?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=574f978a23f683f22b620bd1e8fae62c7003aeeed1b9f8caf4505d5e436268fa900817cdf9786a95c1ca38400c0ede9be294b602911fbab317f17c008a49fb69cc62e53a9e2ed752fe97b7831f641074cbd0024d4d5f32c6977ceaebe80aac429e1433af6afa7812256387fc98915c9e4aef9954c622a8600a974ce42f8e03ffbf4a3f1a4a2cf37a23a957c352db52169b8a88c0c9f8ea6387eea29af4ba2dd437e28641bf4ea34c2b083d9b9b9a40810d07a2bb8488708c9904f7f5063ba3b1da95c759141d765d834511238950fc95c400b5cc07373cd69f78b018feb044e06741f09e72f5fc745c0fe14c5b93664890158ff94c7e66a1a58976a3035a92b0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483442698Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0, Name: checkpoints/2048/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2048/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4943aeca4554e256f95b05daac1a3b72dd0bcd6fe13fc75aecf2bbbb10c878bd12c3c995b37906340d8ccc40260baf00959e6dba6a4ee66a3d16f93f9cded88e9c1c90aed79358ddb9aa313a1d0e1ded4b1c33d9fb7c1349b1899b77cbfeddd314da397b994c746b7571e9a77dc479f12627282673f6497e9e07ec35f4fee3698c13807da65fd616a31dc21794b7e07db16d55fa0e9d1a30e29874ab3599a6c540b67b2267f55999cebfa5828a607455ce8867e710af5da8478edef51355ef879c520405878c763c33b468cbc64d14928f68479b31a1bd773096f25d2f9a1a189a36cc9e1d36a5a638bf4dbd6cffef60fd79d54c7d41bed3aa6515f80a6f3e29&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483976115Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/d, Name: checkpoints/4992/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4992/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4ba8005acd8a9f4a9cfef2f44e281daacc22f35868e3ca41fbc0061d785f01d769633f7414da9742462245abccd3d9eb16326b3e38c4c6096238629f86cad8421dbb532727d21fdcd6f0b898e3220b9e586ba0feb4aef1b10db031373af2ccacf6ecd48d5e03b36af141744fbde5bb1e19a13685f9fac5bf80cd7a9b2ae26b98ccebd24e5f45e5d87ffd64a2aeb65756c3b8b829451e927ff49ad6ffec5774e0d6dfbdeacdc11de992dc3ff0b3bb45e5938756be1694a321236c5999bd9d85d359bbd2b995a402c8d46dcf53b97d8940701d1c999afb362f41448c3e2e18876af03c63debc84a57d641b5cf3f0fce589a17fe9c6e1d1ce53ddadebe0c9862402&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48330434Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0, Name: checkpoints/1600/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1600/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4c20c96f5d1544038b3c957d80e6762e1cc5089bac79e13116ec008e5137ba3df927e3803962d8f1ea5e7e13eb75bebf7051de025d1225d529b6be4ffba84dc4a68dc980e7f42f14da381fdd9200df08143df857fb9816e0fe14d20c937d86e0e5471825536813a7a85799ec7bec6e0005a65df3518bcb76f76eed37d1b3e9aef3ff80cad946385d9245fdfc2cee94fcf93c714bab9428f41a8dd068e3151ce87e005203b09a7b294774d6de762c796c5da3da0dabecd17d5b86107461a6d9326054e02c4ca86187b0d26c3e699331052d34fe06edb94ff9edb0a00a038871cba88b849d10db4d1280ab0162389ab8c9cbf8ed0640120a6f0c97d24f01110201&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48402179Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default, Name: checkpoints/6080/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6080/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=796a24b2544fbb1b2f6dbdbc3c11ad1f025da62bd0d1703147877a1dd3d236d3140d98c028d71e2fe3726542cae24f767fb223c536686d1b6bbf0d35d24db4e3a903a23599ef87982f62f122a9bb5863eee55b6dedf5f4817e1f64416101a8c0dad67313af7f5cc01b9e8ed3cfdd44badeb26ea46ab39f185f552cc67c6686ea9165851775dc6fda61d07cf34fbd78fdf444133243a8e3f805a362f68834649c1111e17fd5264cf5b753deb652cd188322707e07f2432dc1017c3ca70d51197f0553c1d4460c4b3c1fad2af8b97c66819023ad99e8454f024496cd58aa1d00f4d74af8a0f363a685a60340e407cd1640aabc8f5ca52397d408162c8896c62efa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48400449Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024, Name: checkpoints/1024, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1024?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=76aeb05ebc141a97d0250664bed751a5a2f2a338dad1663fbf7493f4039d3514f1d633db588c72f330ae72cd5d9409f3da5715f02d2aa8c29e37b82eab8997fec9d64af35316e3f03781b8bc99beb204343113a50ab56e66941c55fd73d4175496dade3e5d9a4a8daa4bfc8a6787a46a197d5b809cf666aabe1ad7d181f85d787608edda75eb74ff5c22f37f28ab41ef1a54f764b73de9758d931480c16233cefc6e466572306fe5e38fb81527e60512331b62833102ebc43215655ab41b0b1bacac1544d7a55001de2403b79669f67ae60ac0e8ac3bf1a0722bb927e69f6f02b02bd05ffc71776ed5688fb19d5314eb26e6ce8415185cd7a64176f6558bcef6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.484042452Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d, Name: checkpoints/4928/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4928/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7926d34ab66d9ec49ca0327be76f0e20f5882eec229936fe8ebd5e5e22d59a59a245360063576d5c982faf3f5c2baeb8abf8b2724ee1deb47e28b676a3297316d90036a8d97614cccf7380d12125bcd46da8d2bffd922be5e4e37e6ea548c8e0222265b114373752a96fd6d532ce74bfded8b9ad34232a190ea65a5e1327e919c4efb86ca8cf8f7b575572ce85a9f07052aae75aac8aff4ce54d8c6d8ddd35546ccc5b58cc209ce86a3ea7832bcb535915961887fefe073fc124d13f2ad166095c6daf355b82fe5bbb591136641ab034faf6876070a7c0c15ab00fdffece854631b1f0d644705c4564c91804310d36b3b738e709ee81c34c5b282c805ebb9e74&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482968001Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0, Name: checkpoints/4480/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4480/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=14553e4468d3bf51980b284b8355fcb04840e7a744132500dc4f1dd21bae2524ce2db7e302c28c18f7fcba9c63a6db462a37fdf3e85d8bd7daacdaf001538f1a4525925e73be97d0b34c59fee07c9e19c721d611854c523b9f0fe337b4181cdd9b17d51307a8bfa35aacc4711a673fc70905023920e2b41f3ec3e98340c0d497d50382b8562d65359026faaacadbdbe0692c8ea5c5ab3195e0dc26a41f645ff493ea8f3487c20c147f127311d02e6b319d4542b55ba996f9a1dab0575969a64016263faa0d84d66ac31486522908318801c638c3e5e41911add8be2156970751d7b6131087fef24eb66581e7bc63b27cb6533d02cfa3196ce94a1548a2ab030f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48335293Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default, Name: checkpoints/5440/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5440/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=782ad297871d3ca8b6cd93a9c58160674002802b1264b2fe5f603b7362ea06d72d1291aca91ff7aade5517c0107678700cd17d4baca1baba4b61a674491457ff2a3b597f84f621e406894883cf7188c0d6e94a7193a042824591a9e58d65cf1558ce06b37722c3b9f98490a018821797f6c89995c617bdce25578aee18e05eabee2b9725cdc602034a4fecd4e58615b3090a2f607eb3be44f09aecd1248d96b365abd52e5a1bad9d006cbd7f29f28334fc89f46d77a3866533e2c5eaea70ef90b544167d8b7db5f4a771d8c73bc07884871d900daebba4195418f5662e7891baa713e721c7973ae0d5ce8806c47dcf4fc69ee98a1a09975af367c02314a4f673&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483300664Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/array_metadatas, Name: checkpoints/2368/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2368/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b10fa4b7d4b985606cc1640639c0c0f70949f5b8379f729798a68b0da9c282c6e391955f77ade7e2aebd3c28daab3d9e3eac90429ff6d2f3f762ba3c97f142184db239e71aeffd4a45079af4474db3e021bed6f81246606095cffa07080d84e73a098659a6553e08986efcde9ac57c54ed13213ce5e08ac37a0b217ba562d3e3f8bfed9a1e370a0845ebf6d75a2dd4afb32b6fd4b3baf3a0201a5f7e8ddba4ba1b29794d7f9a0f7b734a3b33d10d08471812fd6117d62c248481db16fcebc73a86e247076d1f30c7a7abb4efe9d44285f98c3adb5ea9f88bedca812778a91dceef2d6e6ce0a904d65d70b2720e72e84530f25ea90e45acd8b6245f489f9d062f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482865658Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default, Name: checkpoints/3328/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3328/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5f4d80bfa463b9915dcc7eb74782f8b659ab4b741856bd03a86b180bed4e1e274503a55b9efe70b0d78f340f77fe26ec23222a93518012d6acd7995bed70da59d81719d243601e35432ba8296638b985efe9396250ba5f1d1ffb41b18b0446d513864e11c5d3ca7d194114457067c8707c549c938bfd58a4dc44e2038a0e8745adce16ede1296b4e2a47b7963ec60c11e195d79910e7fa060eed7f37569988900d2226b450d67383fe65113f64ed8c7e00fafcc52b8430568252a5c0f3e5ca491248160c4c58f65f5e0172199611009755e60f5c4bffa1f4e54a8f528479c753cf28a8137504cbabb0bd8af049c05648b047820270cecc972280abed6b410d9c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483213281Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d, Name: checkpoints/4480/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4480/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=ac5f0a4d9026b735c72111491584f9d06cf6ea18faa448e8f675534abe74bdf7e153b8d262b23fe4b684e8d1a3613b2f71a2bfdfeb2a042049bdcff6e987789dc0669bf94d63a8f14afeb279188482addf9393157aaae77312ec580041a486ff6420f14b4d1f4789fd0b1e13888a8448ec630bf86bbc668618c390f5e60d092afbb3a8d44814d495c0511b293c6267d833b29d57d872ba2748580472343f0288065a06430f7270b00ec3db55ebee1f1b440764faa576ac521ae9b7a1b3bc8b516225db56e8f5eb298d0e79fa63cb6670f27b836b41ced403b1a921016a8ac83f0140ae8b23ea4260cf1569e92862c36fa150fc0a4a26bd42c9234c2cd8ec83d0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.484021939Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/d, Name: checkpoints/0/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/0/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=71120fde60a8bb609c44709a5db275a40b338ba23f0a1fac86453d9d545bfd97f1c5258416a623c96337b0fc8d1f2d0b5db16968f18be562f68c6e37f0cb753198e45ba77369fb6bbc0b85ad61c31c6e1b6f3ca6a3337f139b8d985a4992963e7b15e1b252fd0c03a96b117820fdbfdd2052da604bd884e9ba738ce0562faa90601f4fab27a42889fed66277f6ef9e5df7451f0d3edcfe6be567d1adf2a22d65549829d2843048bf2779d3d9a7f69254602ee6bdba88a37966c096f2e521ac3d1ae45c9ee3dadd06f198cb6b3364a3f787e46240de790ca2152c7175fa7c317505a7492a155d632bcc4690e7038405b37472877e5d85942ccd95eed00e82557e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483396655Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448, Name: checkpoints/448, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/448?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0675da3b19d3e736f748f6aed3dfd65079a75d8071839fd8783cc52b8145dab9e5a8f4551d2a1766ed308ea60be3277300318aad9e4bf42add024b9ea0fd67f8b5df33e5a651780a0ad3bc01ba2632b29471ac78051f308a7835422333a7c6416004bdd6cb793623acdc8232b83faed80a217f6c21529fb5f5964d389e6e103a4e0ce2439c5e4176ec87f5b99f77ef7fd1a04f26400e4a1ccfb2838d56c23bda4e44e8896d026442d8283b4b1f078dbeae01133c365fdb7ffb7bff0c3ffe8129a9770712c4f9b92992288e22e04dbe78d1d2ca59a91a87080f24320d9851bb6e6d4a4f430f6fff52817a1bd6bf1f863c7eaeab6cd196392787a22840d2606df9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48523231Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/array_metadatas, Name: checkpoints/704/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/704/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0ad57020463b42f41a930960a9adeac6eabde2330a6086806fa2a11942e3ff3527a32a1c419cf185bdcc0359faa350202c42ebe34276b8c31679dcb1ce057ec31f9a90b09f9123533e22a9f79aa1305483f6fa6c62eee93e080d841522783d41ce6e6198df72606ea4f11bc0e0a4fb1a1cbe514281361dca684314a14c9a7bd9569c1393d67701d970e504b4ab10265ed263dd5b78514decfe56a91ad5d7c0b7817680fee37d75c2f33df72da607640bbb376c5590e178572180cfbd9cb81b3f5936b8818365d555fab736bdc4b5f891442d5b3e2e7f217a5e6adb737f34f52204d46061dd6344aea622c5b20a6f5fc65bd70a2363e46124749bd9a6283bd683&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485247459Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default, Name: checkpoints/1472/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1472/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=22871ad0e02d82a0da9f5997f6e171e6ff9af3578cd053748dcbd91295e23c3a0bd598ff4e1f4dcd1cb5ba56497340022888bb9b2e75b726f4f70a35a4713e11d86012da9f2781b23af6de0007f3b1d088e1962b24c19ff38ad09349d9b9163b206612c71b1bf926b9cb0285c8ebb6f367bd33dc511ddcfa384d7b46413a9435be7ed38e73429ac088cd7482058ff6ea707c98a71e7fcd012a37a70ebd303a66c488b71768d29042c6aec9ec112094ff25bbfe4bf977db019d1f4efca0bf4a7271ba573b28ca47b9902cf56ba742137d3eea1121e76d08c419a2789827b122cf143fbf33625b4c4b29c4463ac661dd10505d1daa7c2de6daab607c42e57e40ea&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485259543Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default, Name: checkpoints/4416/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4416/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3ea04d0b983355e353820b6dfdf87c5df8458f2dfb288fee1dc6c096606ac6a275f1a7caaed05c3e2679a2aca0134bc348cb8ffbdd21c6e9bec0d45ee51dcaf4e6396f55e2deb553b1cf2ba13bd9b5a8e49dca0a2085bd0b3e2154ea2aad5db79928e5b99d044371ac228f83efdcaaec22a9e9096f19d38710db89c87a64b6841b492e8430eac4dc26f77e49967d10dfbe0a21806e98e25b53e34299980c771c7ae4bdc0fbb39c37dee0acb86c5733cac450cb4cab4879d98ef09b2afa453992552db4b08fc05ee1094389a064223a74dc19d3193fefde79521c3d357590f9709f810ddea1e8f75c4ea8939d6a7754d412b9cb1f0f6d72dba9008f47ff2b80d3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485275225Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/array_metadatas, Name: checkpoints/5824/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5824/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2138cc58fdd74d8daf6568e5e269e2c2dee56857bf2895518ffe28a35339771e3a29c126dc81e650436b85d667027a9a1f8edb9b1773cc504cd09b538aac8a5746ae5b355eb6d2be5be387a35f3dfbe08aae1105dcb37e02636ccf6d291a7be3e08788b69b86b60a462fdc95ba13c73c98d29ec9f9a665f06b6ed47b5e8d25c3d03b42c022c25e3a149cb36270ffa95a1efb129a8858523e033ebe669f4731400b03f766120811eac576316a111531f4daa759c771a50b01b2572b8a0c1a6195feece8ce5d7af957206aabd99fd5797eaa7a56268e246d786e240dcb14012c4f827e9898e312229a7351515fe827f38847af4f267335d048f5ff0443744dad36&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485319175Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d, Name: checkpoints/768/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/768/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0d957da01b9157444953990c0a210cdb6130f86e200f00b4a27eb533b7ed6c1b3697c6884797cc04196e102ddcf391ac16e13c5e419a277eb8b82aae7684b127141d7f57165ff87c12ab77ebfdb5dfa61179a121dcef391057a3cb7998073859248dc6e408be7f0885e6ae655903915c95e30a15de705fea639dcc5812595bdab20a68c0e985407a56f4dcb9894c3de9558d33c36170cbaa49e220cb2007c7379fe70ac88c89082390dccdf8b1f1253d9db50b1299bd2c0501f1540ef9ea2eb17e499ad679a554937300d5baf30b358292f6fee67a13715aabd0442e80f301edd45286860db4b2758dcc8ab7dae9d10df19ab03a5d6a8cc1f7447834d0574e81&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485320411Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712, Name: checkpoints/3712, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3712?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5bc1deea65794ecda5ecd5a135f18b0e569ecf15d63036f20f994bc877b8889a237a75680f92a9b2f77c8ea1e3b35ad364699b6381e3a4fd214c40a9b993131f36b6a26b67e63fae29e38d2720d11aa4bd211b617d890ff343078bc3296fd33c437a68c4300d914135e95265edd4bb534d595b05d512baf5a798022b033187928c747c17eb485fc183138655cb0cb213eef292a90974658a8be6e04e7972f56018c6754de161fd4988fc373e2b22d9fe61b878a272d306daafb4d58fa51bcc2bae5379571e6d2bff7849bc98aa41ca85584c9fa24867e33ddcf6dbcf71ba64fc35969e70399a393c5106a108aec7716eaf43272c069b2c3537b414ff9dda44f1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485318741Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288, Name: checkpoints/4288, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4288?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4b37a28fb4ddee4ed57dd5238796bc63e825f6fab0fe93f660238197eceb864470aa7c56754d346e721b53fd3d475b49e90ff2de5ddc4dbc55107698023603a8a15a13d0174ca0a8ae5f2750aab6e0281bd39df33c8d308350452617e7ece70473d456e44f1b393e34578951cbd4445c98accb04d518da8afbe09e5de39e586ef8b8a09b325b36071357238fb4e38fcaa2c6535b47e401a59f3a9db1c5429afa859b81bf7954faa6bd8178414edebb9a8c739221aca6052d4629ccc9af08cb051e5703fad7d29e4937eca76d531d06b6eae5c0844713f88c06368bd94a94d1c205a116c3278635480c02fb9a4cd717afe4a116943f3b2f9e5fc31ad8b84360e3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483486877Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default, Name: checkpoints/1984/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1984/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=beb6f27e8e2726c50133bf18a41a1165241bdc70de868372c0ae5051deeeacf4cece0d7e3a2977fe597664c76a6efb4c60f31ab52e5f1238b2f52f8842be5d97b98d653f69c82e3e86fece58b91adba3b189ea728619667dde0ce0b10a953f9364d3e74101b480a73f2747a9e44b68c933e64109281ebdceb13db03dfdb4613c74de851a64d97a9872be52d8dd00f5a2b68a3a164731537430e6b78212994f4bb938fbad3213cbc9c828aa9c62cff6c80c8b72e24172a6ae6b92b366735268254f8eb4567a776bce42afb2f78183f2317d9d39e5008024c6ce763ac5cecbb7202c56bcb3e771580c4a9ccf8b9927c771a4f1d727b8595104b57fd83ac77a14e8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485348387Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/array_metadatas, Name: checkpoints/1664/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1664/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bfea166bb13d651e5db1571a10e042821853b090254a3ebf42709d911b8564ca9172d278002163c6cfc40655f49f3a940c197e46d8db7f5a8a88796d90836a31f96f52645d95ab6890c39ecdc4691639bdb127cc40dec8243631d87daac073b71c66c075397f1d0151fcf0ccfbdf08fa4da6d84c89d3522722316218cc31708c432327721800b8a8d7f98064caf0d3358e9707089614a43b682c5ac35c35fb5e741eae7e5474a9c3508d1d2c400910de866f64a70856a54ab250f1bc74e389724bc9f4e1f74fbe4a62071141ab0bc5c3f18a44f4d786003d4d8afadf80837d073b784a59b58f8d7bbfff363e96d3428229dff321964733469b8c14957d9febf2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485378728Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152, Name: checkpoints/1152, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1152?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=930d82dad8053e34dbf70c3116890be7255637ec61ae010e7a578ce32391b81fbc2b861342a83c784331e7882e050d026408f14986160ccbb503d2f8cde27e1ef87487f5e397773b22c0da94c2abea0561303c6669f9109849db336a9b4592faa53b54b1f25e2f982acef3c1a4cb62e53e897b9a93380c250b025b2cb799273e58cec793424b20fed723beb6f214970da3bbb4542162dc0cb75293bfc323d62bf7a5111889359f30031624876a2ec03a92efd1bbaade16fee0d24e584edc004222e896f6e493203f634784033489db4b980537e133aae5bf8b3b75a1127c84a2401aa21fc93670c461943a14c478f772083bab8c2384e0847369c9b2e6cd27e2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482565118Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default, Name: checkpoints/3648/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3648/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4abb5808c94cbbdc08b0d2ce99c93cffee154b1a5e33aac84eea81ca2f1eb129b69d7cff635ea0973d547c2312f82b5fdd788ea86f7c975ac81a45b098ee26fd47c8c2a4790d6d955617dd3c5a0f85961b014311971b1734eb6fa8c387185e62171af05252e7bf758fe11bb4074b7e9fd2bf7031b4e0e86504630248e10436cf98df3ce3b0a3542ccd8550729bc2037afd16e081592d5432c3ee38793041f7e56f3d7420b7f4c17b611e33e602514da2138df5de2325f44a703880b0d520874ea4e7f254162372091211d51dc37aad2cbdcebdc4b61a4a606ff896482adff4b94102913cd54049f30a91842ff35cc394e44f5cc5d61611bcaca3979b88178036&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485425757Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/d, Name: checkpoints/2560/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2560/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=06b692a5df9a8e2ebe713eaabb536d39bb9e5390f56948e8679a72864912584c36ae6aad8ae3f96b4f76da24544702e80d1866869810557c31456984b2ce4cd085b1c4294a276b32cafa4fb2366a90d85084133cdefe338be1e7dac57163700410fa7c27afa4965411c087cdc845a6bca3ab0fa9c898cc686d4bc8778bbd7d0dc694b2c6f1968c5d471b57a975b3b0ef98dfa8e9791133dc355d909fb63cab7ba49ff3d278572dc71b40ab6e0ac5c4f52f5b97dd2c9a9d5ea69eeee123525010fbdc0f86010d666c8e6fe97c0f00fb71148252bd9a143260530a2bcd3c9ecf2e7f6ce034069e2235c441b6d0d684e219b715ccec1fa3858709f91fad965d32ff&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485466917Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/array_metadatas, Name: checkpoints/1792/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1792/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=be8bd0c2bd2bb1dba24bce573d0296abfd9dfe2a72c69cc01ef4ce15fba2ea32966be1efdc9302ebc8199f168bed3a51b2fd1c7ef8d12c7060b01b993381850e76160d021f77cb0c6a7eabf96944fb59acd34235a243306f9c15997c443f092a0efe9d375c1e8febf60dbf2773dd12caefea3f22a5c959ddd8550dc4e62b5aa5c1471a9e0631e3161b437482c973a2f7e1e2fed9241fc4db7d78aaa498d0ec5b5772e43738a8cfc9e825b970bbb7c7fa8488cb1169d5850dcfccc0fffc73630f63a192be9d55318bbb32225adce083bba8b75f164873c2ffb308a083f29a3565727626caf2f176134ffca515b95cdd971d6dd72a779751aa5daa36c1af2654a3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485556635Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544, Name: checkpoints/4544, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4544?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7b3619888752c0beb82c6b8c4da881f1c2c85d6da779b3ba745aead52539b53a7c01b9b3f448d322cd4c5fb8f71d20e1c4ca39fa710e9b7d934aab33c204e5eef93a7015601ff271ac9fdfa5d1e80605a5f75ca7e16536d3ea66f91b86c1a148669be8c56ed2d082b01e4f682aa4938b0977291da213131b39bf37434bfee349bffe25b02ba6c583f78877ddea5078ace2b93fd9dac5ef588a152ae73682cd489700bf31b41187dea322fc83eecae821ca347c0ee768b06a4d5ac9ec53566f7b901f493a8531cf6a687a98a398caa9c233801215e84211f7f3af48461e0fabbf700f656563a661b045edb05306424af66123ae1ba23f91a8d3d0dc6f2715ab9f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485567528Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d, Name: checkpoints/4992/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4992/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a1a8828a98abab611d9c6aab15f0dd1342f3db1f9b509c6167516d42ab2b5cb768ddc2cac4263edcf7452f32eb9035c38098aca3a5455a87bce55ee6b78dcf7fb30aab7180bd330759195978bb5789f7ca6da87048164e3aec5e5aef2a010747dfa2defc4f905a6d66f0d0128c8adbbdfe64808ac6a2948f77b742c46bf78aa2eb5db8cfe7ad2fb77de1e463d097a3a0c826d7cb0070eda624a4e2a642d933ebbfb5f42de1655756b5ef86b9aed12bcd3a536f28a30ec5aa7eb18b048cca0c6805a6abdff751ba9493bfd9ebd6a598e94702f98c97f26eb114cc0f8f98863a3858389be170fc1ac6ca5da4ece7c15d7234098551b91ff0322c163eb43bd0a0c6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485575964Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/d, Name: checkpoints/2944/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2944/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=83b16213bbfea83310c270b2c0b05d07e845c0bf0e9bfd3478c285f7f121bafd4560d0175bcb8342166d2169b873400346e3c36ba0fac8ee62d3131456fcdf501eb0d8c6ed6e20e17cab9e422d328ed0929141edf13ed3d2a40bd57a3431f14b0989e1634c140f42862327357f762039b83cb21c8ba025899af0962b6032156ab3f83dfe11d1567444189cac56c6851717d9298f6a1a8b52e2af63410b7529036b72a7e49d9e95c960d5b0b7b837e36572588054e9a85c570dca2e72218dee5935575aa989aecc0e4c016b737961d8423d8deb290b2ed9f55cc3d16c5bc1b8036e2af596175d6884b41771e25d0699e8c3e302b23c3bdc5e9ad39e295f7ec2fe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485625753Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d, Name: checkpoints/5952/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5952/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7a6f160afc1a9efec258449bbd77b4b101b4ccade8e58b81541d84ec91e8614b1d198cb52f47438ad4cbf210d2ff14772d1cf9eaf3f8a33e08492dd87be8b86d80d1ca8a473b3ef0a5a2007a073014fa421b4715c55f52b67be88b38b449685681b268662b0faafce984969474cdca27e9dd7fb7d057f4f794adc447d964fe4363d658b34668e1c80f6335fdc8285cc80233e165bf98254f11807c06bcb73b266438ccea81d1111d59318ed231e9ac504c4c2173b6c544dde7a54f6bf2e18701fd59b0fc7baf7a5a974190ece92737e283e14b28d5033facd442ed98c39ad649fb04dd34b5b889e127cdfa689e1512a30b10f6024cce915872e958ed08c625a2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485656875Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0, Name: checkpoints/2560/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2560/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1735ef89a4486598d4284eed795632fa6a2d8117ab24852ce284836eea3a9f19f35e736046ffd369a72211f605684a04de17b535385dac4aa49545839fe18ce14c986b3e03fb83cfeb7627ca9e371d5288751c3f7b09f0d3487d03e7f7c33b797183e51a78d6eefa3dc1e26bad6e637dd79e019b96b531920b3a12a225ac6ada01ec91c668ca3808d65f29a9ba5ba3eb168106481e2a5100016d3b0a8127c057c8d1d4d7f67a448e3ed3c70dd728efcd2971a933ac6e52c26b90458e7747fd338888592ec00921f5029aa2f89aca314ea77fbda96997db0b9423fe2e474a2d627cdf2a455d77f1393fe48c5c0a37edd061572f753f3ac5467fa7cdb8a80dfe0d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485655523Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176, Name: checkpoints/2176, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2176?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8e33fb24f2b8cab60b85e193b40e9bd8799e53459f16769e1d276618fc186aaf5005b65073cb53c905a988f935f8dc69a8861e99287f85f1f559122821d0d15e1f2b660ab4b122f27e5f28c692ad1ae2892bf0a6ee0eecd5a24643e3900000777263f025f821559f0652f92c09cda5ebd623835cfe659d235153c0282979f7673675a29a9d7525e57b63c42c0a6aec6a9b66b7a7f02307b38c2363c9e2839127e25b2cbdaa47e8a19bf47023a312c348ec53b8994a61c0b751f6b24dc688aae139f89741be6156c3dfa3e0dff27885d79214f45eecee494857383da7bfb7a04fe2d53584f6b1a366df27fe9d3286de79d07870cb54a55ecd0bc3dd37e408903d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485666881Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d, Name: checkpoints/128/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/128/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=80adefe975c070ee83790ea9709fe0490629e220d7ffbdea7b66b5d7514584308b62c09b9f85fb494bec368c698a904b703545530f54276a9a944314865502bdd3b66ca8229ce74f38434c680ef8921fd9c13633b53f52e78c79c6974980626c18e320ef3726ff5f04c632cefe7e816a3723e8242661ea42619626c3e83479d1a6bb87af924ae3b7005594409cd6f89d0c15880bd7123e4ffd8c7bdcbe0e83a501bf79c654afac4738d732a418ad44e255ae2c23dc7832f852a38ab437037bb9bec10b519a0215b969571ca63b68473a389a491f60b82a50e4c8898751bbb452837d59f18168c37b48b6ec3293c71a9d0f36827bbb0e334c44860410d013cff9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485667149Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0, Name: checkpoints/448/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/448/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7217a4dcbc7dfca9a04b65414c5598ee05a82e09cfeabac1b4696bbece751c1e9393dbf8c5fca7b1517c143036a06af2d9698dc719ed3192046faa11769f0748bba6a13ac3c1d39a7ce46f5a2eeaa03b10a86157bd998a548d0012e1199ed7856f13a9f7546717db225f135f97606fb53f215ede4bd44a033c046e81640b4add5d2433e7aeb49c0e4c1ee846db206ac39564462242a1cd96290ace4100cb4433fa2a1223db781bec9fdbb93b89e4b17bcaa9c9daa564dc512330e9071549d833d9072048883e4011b850ed587923b5509f3d9b72db4fd3ede8cc1f311ba79674e050f59f65229d1a87a18c247e7cecdc71841d5f7695a5f2a8cb1472a4b06db4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485697916Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d, Name: checkpoints/64/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/64/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=49e25c7be8380e10d1d3f87a1950cfa1f4ea3fba32ba8e6f4895ad6df2f608d09b1f36f1da999790de750f8cbb2ca3ee04d3680bb88e430b58e970c57a79a3c3e587e72ce6ee5daa29af46f5a35a36c6351661f4ab78f51318a69464aa9bba915103d87366596bbf46a1a6b8e5b7b416ba18f11aa30216ce5d156858b5d66375c84e5f54b28684206625b75af989ff8b905c916e959a72bbd4e0ccfd8569a00deec0e282f900ca760771048c5918b8755b9082a94408b4eb6f14f4eaf4e51c5b2eb3d5eb49b36d45c3d99ee51fb8e930aeb5e1358cd34e834c46110691ada5fbc32b9c11520ba98e336574ce794d7f9470d61c0a4489a91abe11d03fa105ede8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485724901Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0, Name: checkpoints/5952/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5952/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2742a5e075d810eddd1b2d492503a0c091d917ed6e6c5d50ea0e60d1f09f6a9050eab3705882179908ca1ab7c5bba6aa5b1de469591a30898cf4efd30e5255f33504eaeba1623d9c0dc503fc8cc0d57cc9fd0818f8019925a425cd526a710433d0192e033ffd4441d9f6211302abf447bd089808ab1b79cdd936396fb9c895a8e02e2916f682a323dbe73618b77ca6d91c60a59265e90e889d1ade3cd9b5f2cbf56460cc178e3923a0cae737c6b9f922a51a54bdac2c7c7e0c85b0c4c68c46ac84b115009213fe55dc634d8932c24eee56fb3a1be5c72f8f7f07d152ec5d1b0eecbf3485f54936633cc0cec60e7fadfbfc2eea7a7e5aefbbfb813db8a2d61403&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485744556Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/array_metadatas, Name: checkpoints/2176/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2176/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=53811c96aff89d43f793f52f2571c0415553bdeb01eb1371629c480c4a8dce732dc493aa8df5cba8d5771b2b14f5e7fdb326c34c53593c16c0c4d3fd0d9ee6636b78327a19452b9d35ec9c563f4f99cc00e0f4844def01b9dac09dc0b4aaacc946a273ddc7f09122298469e450aa2a82a5d9ba9f7917a7846b2ac8cf20a0312e7dd7b7df0050ce719729ba624329e1bea519978b0438b008c1c4b0d3e1e4af442ea16183f8bcaa46c3ae01dcc1d3fd850728669ce211723a53457d92e9a199690aa17303de4121a158f2f635a26824e87920c699fec9feef01a0c3f22df698d585e13fd8be54085356f4455afd6dc934e25307dd1ced975943a74bc296f8f748&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485767742Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0, Name: checkpoints/1088/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1088/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9289fbfdda8024049407f583fcfd436fc0bb03527353eea5f9e02fe93e38f732bd338601e007206b94daed48eeb4f4af63532c7edf90b0fc1c24193adddaeabc88ff1b378d0290a96eba3e474f60b1ce286638c2ee2c96ab18cae9e19ec88a89cf4ec80aa9e8bbb996025efcbc76acbdc2764e6b7018f882e6e1f04665dcc1be95b2ccf0d6a206e0554cc3fb7fc18c514f31b2096b6e59141d0af735b4ea9ed3c74fdbac4bd48d6f3ffd4e4fe3dea26a20ec5cbaeffc2189625276af5f2a2eef707e9f0ee082cf86b4e753df4715a2d90895a51efc0bb74d1a407d446b4fbe2558a28e0f69aa4f1483908604c089c67e441f30d4261220931c622933405bc903&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483928516Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0, Name: checkpoints/3712/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3712/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6559d6b6f0895efe58f985534910c8ed828dd84a433ff3a191f1cf49b41291e66955f3d25e7f184592097b1d00bd7e23dd7907b25eb3e5cb05cf49ef3e3e1b9e7220e4a3dc4013694a9d196c57303d860ba452a91ff664406f2dbb73774be71fba8093b241836ecc62329e11d62f8a56e6133f04c36d4b0257b282f5b03d5ce27a7c3dca8f0dda61cbd26797c576df7b25c7d6f3fa5d5435d4df3367e4406aa7e429152d0a566c61132b897a5b741a055454751eaac40261ff310f41d4ea6c58908a1dfaf6125ef36b92c21c3525484b0bd17d85e34b6ce8b55c9b52742aa4cd5868272596576a0d366f1dbaef589254add2b179cb1f7d1aea1bb3fcb595d985&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482939839Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/d, Name: checkpoints/4096/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4096/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=113390b08d447051b6f1b05c7267b4c6bb4a7b0fe8459d97003ac5d86e975add59a3adee0f6339ffa92c52f6dc3ad00307ad5088570fc760c57f4db9e83a60290c651a21558957d45815b349e41e1475352a7d1b1b13b431aabef5dc29ed0898af8cec95e002b84e0a45fae707b96547a29052ea0728cbf1088b3981c03f27fc28b9c9076cbff6b4913784170f614a2578d7a47a7156530eee58c6533d6978832d11e2bd76ea9e8fe29db1786e29b318b9312b7e40c82678b0974be2ca31e813381f9aa0b58f32eab6ec4ac788285a3da9fafa7661b71850ffd3059e6061710daeb165b932903458fceddd8949431a7023bb02bdda799e711f524d11062c08d2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485821594Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d, Name: checkpoints/960/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/960/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5d5fb2327f362bf7cbc9a09d75b45b680ec2a280e667257d6287974bcfc8cb5067657e623a8528568716fa520b558ef34814a5ea0de0b36fb232c06162640a6842b9c57338fbd3a991fbd5ef06d0557bc94cd41a2a4d04688ddd2e952fb8f1ce82d99bdbd9730a0e6465c3e8692285a2d03aeffb6b49749afead8fca5312fb0079405e57e2d9edf2ccb5ad426136f0282d329c3f8ab521099e4f5a0ce499a37d44296e9d8323b047f72bdf5de6541035a0ab839232a66acabc5cc20c026e3a713910d94d25d07bf13cee0d89700c12aabe6d922dbd1a7bb226bf8d95c893d0f286581d4a42083d7f4bd432ad93657423044b46ecc4389c66dedb3e94cb562c8b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485971458Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0, Name: checkpoints/2816/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2816/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4ebf9ff81785350ccb5811a6320a16bf27d674af05b26a05a24d35a417b6e14692bf04f6955931fbba41945fb3584a49b2281332bd22aeca921130894166f326474682f9c0bbf1d64ca8ac51420eb0641a89ecbdb0e4ef056af4a05a25951122682bcb9b6f007974fddd5cb03f8df7bbfab8d797d6187749bcb19b83719f4e2965ca50835a58ce6e02cb78f911f1e4d03a2172c4356a63c1a524f9ef7f2e8aabc876bffa84d671069b37a222788666b38b394107da908c00da6550563f554af90e245bbe8aaf92820be5cb90043d213c4df9aa5632cde4a8b64bfc1ec55ec820291bb4445244b347acdbc99a7864d6f716c2c3d2471f51496b9c9ad5a4bb08de&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485977556Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896, Name: checkpoints/896, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/896?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8a69cbe80abce82d43268de09b4ecbef54fa20c65927ef48bbc97df4e1a56cbbaac2a49fd4f4c9b08eb6a80fa45c5410c10dcb5b6a6ae33eed485d8bb059a5bfd883ee2908fb3640a3516b4614ad9999d7438a09cb25c88a2a186cefad082c46a0818d0cbe1b34e816f66c7dc9dc2a21577525d02bfdd5c258c65140c6dd32f4f5c844b1a2facd02e30bb1bfec67a146bc186964af8d046955687daf60f65a02a6e4eb56a681f7ce18e9d5ce61bc4fb9008e7137eb5c80e2d956d6a851be21441e1a21cbb8df57df9b69582cac699173195cd2d1a8c43ec03a5d7befbea60db396d6345f42a565c1e8609fe478ca2c824df6388a5eaf49acb48c64558c678ea6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485979661Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d, Name: checkpoints/832/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/832/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b045dcd91d12e9c314d4ee5e22d9da03f1c1cacb1b3b3a1e17122385f64ba5fc41d67d07f4a9709edd8c4f1fe1ab50a7f1687f001d81daadd948905acae5f606b595a3356b726e207f66d866f1a0303266058140781ca728e6037588663ef9bda89acd2a97467255c833146eee4a63fd7b58e8573e9ab34b4a83c7bda8c5eb209c57132ad0a39dc257e88657e85b05063e1fec27693a177c1a50653ce8989617277173e4ff44b02a4925c76cfdd89d351a088d6fdc323bffea3d6e6269f2e59d74c0da4a70c73db5bc79da66d7886bbb899756ae995c81452f34856b7e4e2a4e6edceabc94a823e67f804233329ed3c2cf551dd4fd158abf158842570f86010d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485985186Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64, Name: checkpoints/64, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/64?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9d3c16de689401cdd01fb559fee6812a0f31cbdcb7445b1729437eee2fc11ae36cfd638d1868bf4643080fa4c45789f2f525469df5de719bd51e324512f0b6912e49b2bf557e50b23d26198c8aabbcf0445bca4efdf6d5bc3e4c6bbd879788b26ca72086ed8fd3073c6a5bd659ffe81669db3b47f264696cd7abd69dce32d08881cdef39e401516870f4efd29af3c19b87d06c7a67021c8d37f5f1dd6ac00db227e2be48641df11e9835d5f3a5ae3635bb0478bb94c272175b4ce91285d850ee57ae0b4fd754a63879c995e2f6a07aacdc4712e95e9541ab2c61485aa1cffe4d58329903f22fd1eb0f7eadc93b4b482239a370e7b338e37ef550c04e465d97df&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48599076Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/array_metadatas, Name: checkpoints/512/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/512/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bd62c1217167fd6e66e39092de80f2ab9752ec07850380ab8da4b38e82fe31ce2d620aa47eb6f10830b158c7dcd832de6bb72e893751879b8bd17396986175eff78c852474d283e2b22dc0654802ec327c248ef6a644a3f6af7da00c9f2286d00d90b9c0c8f117bd645af12ead78dd74912c3aded3086282883e1b7a445d9a5b7754209383aade0d9e078079c908bf1ef1b6986bafa2278b6dde6839dc716a7d471347490f245bc7b9692180b1ac381e6f2179a99e1dc84004780da23ba3c67db0214efbffedbcaf356095b5cb2b77836406876bf3858f807ab16277f131a0da82578d299029a592f37f694b3f86ec01b06c9d9e3147f9bd84442fd31e19ec34&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486019104Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776, Name: checkpoints/3776, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3776?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=80a93c587a519ca3187a0d36720fb519c4cef75cbf5d0490b259806212f5ba479b62674f1dadddc2663a8d4d87951c14b2c127ec3798caf2b2336532788e52c38be8771d556ea9bfea56d4271058960b73af58664d9eeb7064adfff5e78f90a36eb1a88659916460720cb2a58d4c6c57479a09ac4a7e8bc9042d9ddf1b900f57fb8f65b3bf241c7a378d5209875417f7195a9c72317ce9e77c428efbfbe6a33c12ac85f2bbb9a90d461e1d476f7d63abe15754b040c59cc7fc7e856c145de3c023b242beb14b6027b8702ee90a228e9a2676b3b2c2a55a651de12b5b943d1df84e134bce943ab5fa3b972ba4a79acc49dfa59e01e3e497d20f046a295ef47ade&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486067277Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default, Name: checkpoints/3840/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3840/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=765238539f914ca94cc4694295e7620495940ddb4040f2d87eb7aaf1ed8a33e4466430039b31ac73e9e71a171b8bce1032a8afe8637d84b939091d649b20627fb378f2d3ec32dde8c3a1669c429acffdf302f611d2339175b5bf3b95a910553fd32438469603a63743c76543295cedd002ef870e7c6b9b0344d0355451921b3f467d53d86234342777801ca29829d9986508825d6e271ce1ddc143fcf5d9895cb6d633c8f7e938b4c8a135b5b259475f4acd47cb945093197bd269d6fdfc94aed28e1e409cad913d0d296f22169d01087cf5c5c8ae158306f53c3f51fa3eaa74e442e473d5b0da472d14b6fedebf869b716bf0c5c7d4e879d95aed09c1c34b7d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486073149Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default, Name: checkpoints/832/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/832/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0358494cf2a39c87208d042f02430ce5b5b969b99bb92791ed74203a6738c3e4fc212e70917146e4c4d6eb26f6dd8570baa5641f409ed827c9063f7f8bf4e545628ab35d29f9b8ca70e170f4a48f6e877573a5cab3a5769f69104bcadfc0ccf6f9166dcf885c073f0a027654726b9bfd30fed72882b8e1673e0a0a1bece7f1c1f74b664c995a2231eac6f3ff97db86e8db47163423b93af574c0a6efecd7c998ee8c4f5a3257964cd23549663af504436332d582067605fa1e2536d5d0445c51dcd39e6d3ca5e2115c580b040c757b3070d5753f52eca44fa5e07a9c99fb4baf4330720a71d11a87310c552d5a372972f127673d60cb8a8d1e13e1a8bcc1a91e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482962525Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d, Name: checkpoints/2944/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2944/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=276a8500e6a81443bac80bd68ab6a1da6ee28d85707a70a06c9900bc25cd9e34344e4db137ab1891770b36dcc6ae2377e34a1ebf5971579d365488a7b8956fcfc68f096d2ecd58f6eaac8ff37d93c93d269e956da416e25f5556516af067cd6898b132438863cd7bd16e655c5a8c7503993dfebc5b725e0fd48b47f8632530f1639f4526a6ec0f1ca4b2562e35cccce70d6eb1e40c1ffde8d96a08184bfe8fd13dc521bff7da0d911a6f1ee333b6378ec66524860caaf1384a3033331538aabdfa05cf1734d7cee72738711e3cec23315416a83dff4871c8aac61587dc5bdf97f1186178e01dab30c3018bbce7ae9fd1ca2edbee16300a0056df0caabd7a274b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486169943Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0, Name: checkpoints/5056/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5056/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=78e32bc1276d066d6bc28a49d5396ee39baa0475c70564604d4b86c120674793bd1560d5b1c28441e1bde775e7d0773d54d7537cc3b65da5fb98c257b8a17cebc24175ada11ca53e28edf8c03d9ca648f4a3335250b7fe48ef2e001637ecae7eb3fb0572663445b87bf2ad669420826983216143b321313cf3388053425181c9aa79e1b7c56cfb78322bdbc67c4d73d5abc9a9c56221295d82e7179a1e8e9f8d0a39881483e434d9de4a76fc92bb48f7c93aaff69c2e51236de694f63e256c5efa1bd84cfd85c0bab18394d8d0306b6174fd3380de7e3960f7bbd204c3de065cdc01b2e3c043b75ce6828b8658b07c8b4892805a232db1a707eb96536607870b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486182629Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/array_metadatas, Name: checkpoints/2752/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2752/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=52d8e00fd91304c3b8ebcc28b36894f1ee2d8c83d018d350f2dad41e9340a11b43029c3387f5bd17d5a006d614f16a41383acfea06d477e8d5d003c7c4f1375d86a05643554cc32f7bfd71cef69b519619b79fc04b1f3ba1134c70607fe86ddcec3fe27db23ae9b23c7bb0328641537d3c97e07fda23971c03bf9615d9f87de888f1cea98bc8c453e6ba1aaa27d8876b531fc8e686f560b701b50d1b73351bd5c6bbbe5c4dd0915e384870a5afb148045daf000043d85415202fa3736a149470e8abe74a27ed92a166b3120088f08f554935af2ceb3e1f21d5a32755661bb9432a09dc876a740335139e5de8dc06021f435ce20e0a49340041c236ec5e17c57c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482991094Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default, Name: checkpoints/0/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/0/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3f7ffb60f7f68af26204c6a2a45ac7bcef611754bda5c2d717ed2134e0bf96216b89a538eea7cfa5191feb9520bcbb512db328a4bcaafd50165cb51b78846912a78a787f1b2f86f4c20f49afa2ad4a9c811b1f9b4af3f711550930e564aebaf0f55e6f51fe6840e4a7771a8ba0885a3f0292f93fbca985c4f927d39139d62ff68c3cddf016feb688e84920b7570783483e21412ac0941b1ccd9eb9eeb0fb257cece5fa7256834e0c46dc4c654d76659f3daf6468942ad53935d23a88a772b67f4bca6481d5315400383156cea85de423715ea313a5014aebd14292960d24a7b6aece6a6a203540b86bac8a55672ec2b5ced373599b2a58d0608cc8cd0cccc97e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486300713Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/d, Name: checkpoints/3264/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3264/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=97e800b24cc99a63876fe67e1d90093f43138092c80a5acee7161f47c4f14a4ea9d0e57af950d76958be994c70cd8f39ac9246942b0e59cfbc7ac9f52b8d60054066815e7043518a6d394b6b296b48ff2ba106d6255a08ae19724ec1f89d90a77f2bc5418bacf8075252286ae75d7bb2a1e71f5b1c047952301ff292c4538162691efa93501900e8ceb24581548a0c1f68b1ab6b858eb70e5e14736f1bd9a754c7da2df48bebb37589efc043258433e0d25ebbd9e90daade7cb9489ceba5f0b77d56fffa01c84f6b9e9a1fd5adb89bd4706449698746d84e0c5e32cbf57669faa1842e587919a4131edc4755d27b2ef0be31dc5a1215efece652e83605ac3450&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486321329Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408, Name: checkpoints/1408, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1408?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9a333747c66a177acf81c6a0c3cc7412767dab4efbb2b7fda6987f6ac68ab8d22c5215154f9c81d9955c227043ca29ac917a71fdfe097b393ce6f9c53384465fd386aad2be8153fb7760b60b9f8798c84520066c1eba1ff8f660d18300f8782d69d7e03c0fef734677f4df60dc35dab927b2f8a4c5cb41783d1209a67d417245dd862dad9c0b5f99ac8c94b6c85f1b892d3ab58d86db6b5c00a797c15343743f13e2150f09f952686ef5454894cf2a59d2fc4bfeed3419187b2540aae5a934485d4c3207967aaddcd787a780d86e1b6b2d8019c6c227fda8cc42465005ffcaadc20c46f3fe7bf04abced80e7b50df09d41b4932e989b051949246898b1f650b8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486307401Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/d, Name: checkpoints/4800/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4800/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6fe49838bb4a6bfcf81b05d846c5b67e1fd313881db17940929a4e435d3741dabc51655f98b944969090697ae2b0bf2ad95639f8e558893c56b7d5e1cddc0c103bee615b19c1de45842ee6888e4aa79d46895812c3d1689be5f122b5b6784ed73a35812b54e11c7a5936f5d2977e054f97352303913fb6403030a3b30bb1e2a1d9a3cae42791ae542d6d18938d821deb37aad96a2d0747345fa5f2fce574ba1ecf54a7fe1bcc021ccdb1c4e992d8463d482a4e2d202ee3652718c84d2cfe341dc677d72f45911da07a4f5adeb30779e91c37ea5dc94db56330b1b5fc0b70a7a64d0e7be7beeba199f0e05b3772e184b07c28fc428264325a1f083667a32a8b50&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48639596Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200, Name: checkpoints/3200, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3200?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=465cc9601f1ceabae5e9569f5025456c7a303ff7a3a4a8952254c73b5e7494474a019f10fc620156ac7c55196827d8e50a18f403ecc7617f6fde5d6bf609f60cb87a47757f5b0451cabd3d23b67ec1009cb7fd6a3297f4ce2adef3a109ddce1348245783837c2aef8a216afad530a2400d15c3975534790b7ea7372d3fddc63fc31005d39f6330145c945cbe56fee1678c75dd259d6ae8d49a02b622c32e12e75ebfee691419cc3c27d36b8e1d495969757a0aec1338511fb89334e8bfdc3c458483ad47541b00f9b34450ca48bdaa6ed19596536f4147a9bc829d9a406150026a2b858201b02946f41255cf4e85d6b1461d836d01b24b278d0632b1093d65b9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486463812Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/d, Name: checkpoints/1088/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1088/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3715f5eeb5178fa67e954b5cac5fd8f3a5de755880854706b4dfd4693aa49694e9014495cae5f233c37899a2abb4dbe01e0a450b24e2c18ad4358ebb115e74a4fde652fefa70067e5999bd5055230abe24136f9cbdf93f5d66d586d4e1d6c50ea15ab6b159d2d9eb2d67c572340e7dc438be41764139d5e93a3b19c57a4458283e34a2c4dbcaea9d5d9d757068ea7d4d29c3fc50fb593685dca82605198905dfee0436a18b2c384f5c03f94ae365289daf83e36c1396117875cf45f927181101563307098efc7a7e704017d08b2f1a596abffdb1f419ced8a4216bfc0d0621719bdc37f8a7ec4f51dd5aec89c29c18b032f1ab59f5c4def160f6f780eb507598&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486603606Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d, Name: checkpoints/1216/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1216/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=84231a2150e0eda503c236c9a1ceec9f6519ee3bb5e2b00c0b26b69e48902992b7b0c212214c999bc79f14f5376e966d06e4fa4fea4fc1aa604b434e926b050f216b787335edca5a8620859a855635c156421b5b7216a99b507cfab04885f28c51a5c82800b30313613c5e713639ed6515a9ccd8ae46ecdfdea447f9f5af4e1945f1032724ab1da921d1f514cb1b04185a79a6fe825e03c2b7bb458e63845a67f8fcdf8591dd08d59c70903ebaf2f9c23f9c000b05603a7e66f8690f06120fdc61b67ffa29eaf1f59270398c8b06c98d8e4e2a783c41dedd33bd18d285793a65d7db2ac35b0bbfacc46e077e6454a8a61576cde4994f0de4267a46d7ea7b9f34&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486648451Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default, Name: checkpoints/1664/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1664/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6303b4c158f2c51ea7218e1e497b34578d4b72f4b2c503db5a03fc888652e64de50fd70150060bf6af6a6273b74427f7281b8c8e3e15837c2f6acfbcb9123826430f5244a3262b120d3c88481716bfb9c4c9a6f08218ea00b2e29b30e7b8e85dd4e4f90d03a356d89c544bc21d6120bc7c9d7cfda7e6f9ffb5f143af22a6a809a69b3cff039bed2f711de6050c1e5ba6f2542633c93d4f3a75a9c42ca70de4401183e5fec3b2e6811b035e0367163c2b054ea5799889035854d7c4a13ab1ba1e7e5739a20d0875960f21b1bd1212dfb42719a79701c3e61b5044876362746163a4c820934455a9968f2314d8c6132e3d5c946bc17b6c5a9cf8dd5922a145af3c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487120623Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/d, Name: checkpoints/128/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/128/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0be6a2192b22c40f44c754b2c3349b178484acc1c0d74f57e3a2f0e07bbea5f0524f0987e6634bf127e5a50df6cd09745adf37d904429e4e64b15f874d145dd1904b4f93d815408ec93c2ef116c779ae4132c8d622a81a72a8b52163ee4d5f5a9d0cc63df0be2a3d2e73276da66e07262da4ad6a788175a33a916ebdc49b5105c6be6f0e2b48c029224f76a164599eafdff17c2693db174827174c96275e6c618d4cd966fd93bcd17ae104617cf1955a2ca5443861d2164edcc72f122bd59499b95e317e648ec6e503c906e365fb44c1160286a250b90c61437cd0ba509268ddc6f1d294d225ffdb8c3f16d0e071fb853a92b6c4d95f7daa8e947f8096b85ca4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48713864Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736, Name: checkpoints/4736, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4736?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6b6491c05e932af0e213ad4f5bcef1e6d25db51237a3ba7365ac3d1b3c8ae8cee0d8ec6cd6af33e5f6cae5c2b4299dc214bf27b642e39dbc3fc7c570fa50db7d7918b61d35cdc348579f7d8374b9402b68bae625a5dc4fd94c5adbeede4369f9c0fc60c99e347f15828b87512c27242b1d50776660ed8fd2635d0e3d10dc07f8f271542c23dfa8226fd7cb317f75ef7a6d2aea872d34005350c0c217cdfbb4b2548ee077402d469b93b74987e1f2f65147cdfb69e8d175d689d125ec544efc85c577ae05c775a69cf2ee11c17c4ebc7aa4af6e0e2f671c843172c8f97ceca451df5b450b8402c9d9bfd63d4344cfd99044bcddc75f908b1b21a5f2289685f616&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487143426Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080, Name: checkpoints/6080, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6080?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6ee8fa58e3d3582aa1b144e8fc5d00729e0933e22ccf6ee2e022b5a3c0cdd69c8f9c315d788331bf9bd566ceaef5ecbfa70eb5fd8926e5b4633773a19ab34ebbd42a5de280792eb1cbfc656adc9be47f901e5426f6a83a17383b5a196707942f165d2c37d4874533183373e0682d4e8ea234106ec01cf9a649b2af760d6896baa0c44dfcc341bf3068a223181554e1246ab4a919d11740443d4dad9c71d2f6f0c5d36622ed4f96063a34c998239fc7e44f89666ff0f49690d34ee761d7483d8c103c92ee96589dd9f0db4fc81895e7a3b39463d674fd5f09f840d0ee04ccebd1589694ae8e9436f0a5e38ebd339a64f237b5c032b3e746106f4903e5759d0028&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487179262Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default, Name: checkpoints/1920/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1920/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=90a1711099990656147823c9569e0a91093af9954ea03a907557500869fe1c03ec35559dc9f3b6192582ca333adb151ac4549c10987ab122e44264a5d4ff08cd07d5b994a740588e758189d75c489d71f884da1f9f33d476e4f8b0b1e2914501a299e7a54c4749c5fd6ab26e84e32c748e64b9002e21099534fa675f7d7f099ad65c6141dc7efa3090a00baeaba9b1f5332d370645727a2d4037b8e8015e04ee5237e8af134356f8a05d4a981ff132e755243ebc4468ad6abbdd110276382138e2a095061afe52767c7ece8d0d9f46ab6d1c6e2b70cda93441219b3edf80ea66e662ddef4da8ff3f28d87ec184377d4e2a8c54fc96ec110d2bd960b480224166&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487303265Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default, Name: checkpoints/2496/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2496/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=20fa7e8345a9201b46f931096c5d49c22e41b79ff853b1c6d12e5206e382aa3a07478a90d3cd17b2adbab5050280790d5de159eb3241b5fbcd57dd80cf44038e30aaa66be82f7dbe13e8f4be7963d2621b585b197b7f21553939bdf49bd0b2e9b1f3ec17e8afe65e1c65e553c9b4466edc378b520098927033a2a8e5463b612161f94f267bf1d58787a372a1b4d79297c34a898157184e330c6eeb61da7e18dbc3602ca21560b4a2732713cb28220feb998a744812836d07cc07923e43db6be3ae5309f64d101baa462522cc4b901348c9730be70e13b2b4c24155bc864dadd050608d373f6403a11fcf9e9957facd2e46ee19a3685313d5b055a1ab0c43ff9f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487324741Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/array_metadatas, Name: checkpoints/4608/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4608/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9c1e0c755618630df49a5f7fad08ae06122e10ed53e952d1be3aa9e27434c84cb80f82cca6d75635f8c60a96b7d731d86a678671905e3ffd88598178ebdda5d3316b64042c3500d34e2eab0cd9c411d0fd181762b9cfc5e1d9003aa78c38413ee1d0c3810fb63760e0edab04caab90412cf478107c452898137bce30a9ed762afb7433f4a83f13b35fdc89b80d58f693b2c0dc5c9bd5fb22cb0e7f53e699b7cfd262ae26b81f36bcd2a72114d2aa154e98502dedee63fe5ea81212719026d303afa117d7588d0d4cef1176d32b514573448128a772d65fc68d1eb17c037eef4619284a565fa4c8a95354d512df67065552ceeb9ed156928d8e82132a80d10034&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487219341Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/array_metadatas, Name: checkpoints/3392/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3392/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a638a0be14615a9490df87fd196af54bdd3703e1145592b17b60281503a80a8a91cdc3eba732d0652fdd95e89984306a2c63d3b4bc39629115936d131883c01a3d5a290caf179592c0090d96337871669c6886f7d6b71b60a6f034d48f45e91032cdbdd0009c38051111123bdf3e67b6a7e938f2fe185441c6023f5860975e2d0a5a2fd59e0aa5fe031a005a60a6f2d2b7681d5d9ca3042ab24a79f7d5a0aeb3ccbe315e03555ac2bc93f127bc5eb0d591f3cdbd7c8e8d20e0f01c904a8629d1df2b762e5b5b86e11645c1d23d876f3bc8d418d37c76928d05eacd7769f1391593a44f41df41d9d96adab0eab076e19eb452221320e08eeff70332e42580782c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487441868Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d, Name: checkpoints/1472/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1472/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=750cece16eed143424bfe4195fc076fb1721b4811e53907bacf15b51e87761dc3dc2ab2b1bb0017898133ad9318414dcb5cb3029e234cbac3e44475a319603c70b1cc8d544a17ff1f38c7a5be478759d836101d819044bcaf7035d137c0408204ddff2cc4deb2657b1076c500c09244ffe42edc35ded3cf4385970176f1443ec259908025b23ce3c8ec3b5fb415afaa85e96b0b77d6e757812d0d16b13c725a2c41c87399962791b2d153fce78b8ae6439c373c6d4528deee3ff71052153e97e784729768a43d4dd276e57079376a8d8f1412021f20967b2f2e95811181f048538c00b743063d4b235f90f605656a3018bd6d348d0adcf528c935346b7d849fe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48748667Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0, Name: checkpoints/1728/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1728/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=160bdd6e52aae114643acc7bd11e251c16e42fc6ed5f4bf49f4c25bf7e2daa2c970bfce71d495be894c377ea658fece0fc30a65f5f3d48e296ae4150f732e1959ecb741384eb61d4fb8cb0d4f9415543a3d0d4bfc870b1f2eb0dc888a7a70b78813dd3bc05adb1b8d9ce97cdb476df8b8e88f8ceb7401a4dc893fe40d37f22dc1af1c7e99f2436b7cc53eaa42869f6438b9e07844d65ba2b4f564423690170ea53ab127d58650b4c0f2f9321852edb9ca4b01d4bd3d236199aef8f0633f319eb09ef5bc07db7181dcf451a914294f3a19cfd1af2bfbbd8784f9c6b5890e25a304a512252efaa871442aa588b5cb8e591733bfd118c43e24925aa2f49cfddd710&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487562828Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d, Name: checkpoints/5696/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5696/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a3c305c0ff91706027277de23b452f5d52d00e6c226a75ad31fce73d4ebdf9db5135697dfec180b8b7dd9971d52e1fca8ff6804e901e870a7162795dec853f08e87b497d7ccc0b2a92925750c9254900843c0c967dd70cbcc810b1690f333e76798f9a8c5ebbdd64ae034821cb42e40277b9df2914c0ad938dd1902c1cacd0122dbfd5a0b898e3faf9cc05e297e502dd09ebc3a00504cb7942303b0a584a6b40fc7195e0051eda782fa650fa3980ae7bdd27b9d9f1b45ebd5f8e1a2cb9992eda2b0c78d408f932c3b5ae7937ad91195ff641d57c29574c508ecc06232351db5740bec09871e4c9793750a052268b24f7f1fa7b15029e91306a2a85e22da70538&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487634324Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/d, Name: checkpoints/640/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/640/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8896350142d5280074dac14f8f1d2ef62c7fcbfdd1819bb95d8e8ceebd05fd7536374aa597dada0100ff3d01e93ce002573eb64ae3499f0d51a205c3aede83ccff4ce404d834a189f3d877258f6d78b5e4bb7ce6e4c21ff7b4f60f43d7cb88f3ee7ed73e3992ea423a9e2cd64c5e9715b12436ab1369e4dd15bf98db98ad573ca86b74388e646fbab8e31397727e59abea571188f17dad3ca3211c8de941b938ca596233d82226b0daaf16b97b2f86f6e9c04431bea13fc54c2f5fe4349c6b1374d4ccf10b3ed229f721f39edd7320658c31db21b7c9ada6ea56d3e2d65617ddfa1a0a565a41391ab447e346272d98c3ac777e65e8cab63dab9bb937108a5c41&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487687792Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/array_metadatas, Name: checkpoints/1600/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1600/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=19f7c9f8d146ed99cce2f526cae684c6126e3c2d517cb5058dadcdbd2baf4bea80fee8e50ddff3905635c69b58643b5c9e0af3df749e83f5417182de22a675958d0e1df878018af44ed0651e3bc6ac0d54f9b17ac097f22a8ea8753ebe87f60cb81a6106a74cab1a2beb9b7156e3d532b9e405e4d79db5affa9c87afb5a470b79ec6f9f38eb6a0ede3e7f86ad29270de3fa320aceae1db41683b613dc741f9e6f1811b70616b47337585e7d913f2cbcf386ab9c2a1f0d0486b629d99fd6e32e72dedd282f8d8859a7ba485154afa40afa6455aae9642b05c7e1d0cc03b39d90e128bee115899dbadcc10665bcc82dad82741f97199e3c0a6f5018786c463680a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487820382Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default, Name: checkpoints/3072/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3072/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b5a7427b2426a10288e8b8db87c2c069e69e7b2dafbd559e0f90730d2910da9979b8c1f5bdfb3b78e8a40e334b631d2d5a8bac4d85c72b6d1d99d01f8204bd2df2629296f83eea2faa921a33dcd68ccf6286d4cc0d7c38699514e647437b9d0cb829585f2348ee91cf0b1905ea72851c4c82af053f083703765482e1400cf798d0eb2e8a84039f971aa6254d44bca5a2b368be932ec33fb563aa0d67171301bfeaeb6750f7b6b4ccd4de17af99ea43f154b6bb28a70292600f3cc36396422e6fdec0aae5d3aca6b22d45e55c5b74a598144f3b8e13339b4955ffd49697f07b519061e47e5e7a98e90d8a1cc343a378ba2486c094e057aa5c0cb79f4769451eb4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487724672Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d, Name: checkpoints/2496/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2496/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c51b8bce4c4f33145d37b242f330daaa2d747d1c6fbb59ecbd4b701f8a4a3d165400408c2ae45d9f54839c81fd527fbb0c9732447a0f653b80d5f9ab82e94ccac786697335e1a50189ae970b447fc96e627925e8ed9612d9d1c524d63cc38ec3c0a4997159da037c7e2ff49fee11a69be5ffe5fe2a03c590b5d8ba5ca7f3ca8099428a67f9f12bc192271ff3791dcf273ecb28985f210f94bdb7e2eea230b76bc4a8b10466dfeaa507723cd304b876333f2d8c0e01dbda87ec4d041b1944c811ace31b7241c103a432fdf60db2a3db9fdfc584f7901f2db76a28c928a25ffae3039cfb4fed0963838a97802164d4a7a856ecffbbe4535626961ada24c942cbba&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487943304Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default, Name: checkpoints/3712/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3712/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=697efe4d7c16246f23e14ea878a5e7b98b972fb75d8ac3501980b97b3bf8dfd247457d940e28a54ffb8b8658f02b650e83c783c6c13993866839f481f13e7c5205d5c2e66582cabd45544963006f2428bad75c4e4746de9941010a545b087f6b4cb2c7b3826ca3b3e9b167c54948d4f84788a508b7c5938f984439938b8f30164e58937aa61539a98ac0abd745c5474fe15d6c1dab2c48259b856dd749d19f6d2e632a54552468331bc34a38584d3665bd4586ed59475442b600d22ffb9731107787cd1b972f5d81cea8e3684196ee4af380ffbc1fa5d1f4d3c37948608fee22a2c40fdf3e7a1a3bc7d4466854cbee5b731254a788fd3bfcebd8416d027f9755&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.488027667Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952, Name: checkpoints/5952, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5952?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6153ce726fda4471657c46d4e1df4af2c6ddcade60fd20b41f52ac5ca0528905e10f1cbd4c09a1c54b19fd0129cc5f2e5594a343ce9515ac64997ff259ec323d47600d619499ac3df0a27ddbd737553684812aca44644b9876a93ec532c4d3453a4ca5592835d2a47ec381d8401b0c6ce5560e0390c3092374218f6d7d2693f07ac8b1218b0a5b0896013ccae94f09a268e77df5f945eaff42a442770f382b3a3d8c62a72018769ae30d1ee1e2264b1753ad676d68ee8feb5027efa88ee5ffc4b5dc4fea96d8b3165e70f0048db063c12ce1bebeb12ff1de5448b8eb5bf26794b50abb41e6e673e1fd3ad7bce18f9902c8859f82c5f68beffbe0fdcfa2521551&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.488118968Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224, Name: checkpoints/4224, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4224?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5cef0666599d004719e11cce57b2066d8d9520b2d7f3343de09ae8d36b0a54af5f53b895dea8cf50689dbe48a91990f0abbc4a8b47d7fdd1abe1f0b36dc729ab20efba4ec116c6c4b3cea041ab1448acd218030a661e062e055f32357292f9aa7f0a88124d0da0bc3430e1bb1d6ea371908e51082c2aefb6a7d2d5e9e7755f4167c6273b66815e29f25af75703b733fa79a93d354ec3b16b3d35e4f9718c5fd1d72369f4389207e2d3a2d59c150e2eae554c862a2751ef5065ae51b65c9e8c1c2ee2090646b126014c467688a9fb219fb82aad58be958abb33c81c4b8e84169b8a3d7dc616591e6fff327877e7288dbabca5c18a1df4741a2421a6cc9a5de25d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.488211173Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/d, Name: checkpoints/3520/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3520/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0711d127e75443bebf7d43b72cd456930738cc73942df1deab50eb92b73176e55b54268bcb27d464a960f9efaccdfb95fbb9ce346d30235b231890e7c81c479e69827e89de58377c0e9a9b9f21ecc0dde1deea253e8ca0a6607759f7e89a8dccd805952656c0e3e8cfb4b381ffc20a33e968f995c104d0438739d96080d502728172a7064d33a23c90d3b8aa92ffbd72ef8fdc7dd7a4b180ce20c7d62a6ff5d422f4f1695e995715f6fc6fd83e38ca1685bf509d85e59f599237c59a8d7b49fe21c3a70c58418ee75e8ea1e6e9f75e431a7fac2062f62011ac865bdab9bce71a9074e3c2a55e7ec0c7158c9a82a097a6bee3e037dfd79c7322aa6627fd1bc49a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.488304457Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0, Name: checkpoints/3328/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3328/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1973bc6b354e24182e4a43bc03864de9dcced6a8b4c05ec598297103b462b28a605a23249526efbf9b92311aec5be4266519d73777f9e39f90dfb242bbe386e800fb8648af96342d8b30b45bb6f3dec7f0915e3a2ee269e460d31a9b9424aaefdf5afd6469f44ca81c9745ef751b5a2972d89df6539b1b5a830a2adf9b42c62f2510b82f96e1e99e623a06cb7fd872ecf93af9f7804734a27f8d10c4028b0b8cf041f8d6784cf9d4d23b0d8cdf306253bb49f722806444f91eea5f2010669a07d51bbc9fde09d76a78e7a7a9c0b8decd8f2d096915cfd5af17380466cdc7282a2b56ed77956e3b61c386dc0d28f46ae5126d5dd354c3a05e7b02a013c194333e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.488382126Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d, Name: checkpoints/2432/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2432/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bbac2608dce3c4a9a227efa7afb1973a6137cedd5513f478d96181a843005aaa3de76fdb603d7f447da48831f6c4a7f33ee5cb6a8a405e4e8d9a790fefdabae908709339875708ea0811a309c606799f19d41e334b428171a563294acd66dc394650a1039832b6a89b043a3bdf979265fe45a9c16f3682275804e33803da1bee881349e82b078bcb14df362a4a1896e496040aa47a7ef1b48aa28c52f07549cb810b60f82508527a6bd43a749f93d3e7f56eee8a50d20f986ba6ee73a9d63054393f17b9dc00f10aaad8fb4ae995912d18beafb605874d07f70b15ee61b228d132ff9fd7bfeae3ec3ed9b354ec258aa96a91e06471a8c8a7267d0f890a37c4df&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.488501227Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/d, Name: checkpoints/2176/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2176/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a5eadba7c26fccbd11753b3f117a0f7eac3a5994cb3077645933887262ae039bd965b247ad08565e3d3bbaab8c621f0820c4f85d68e6a1d6216d289cfa784b61809d9735c830d484d101735801a8691d6397705477f4f6ac91f9d11b0a6c57905ce8c94fbf2f3d26145bd10280f455c8952acc1d3fffc025f321515eb38027f9f41c176bb922a71450ced3106dac45e2f800819142c26c349f5c3c2dd8d107d5fc0dc21856311a19f2fa580dfae8d125aa95f1488fe16502c88b8ac248cf633a9faf440b4e97d8fbd74be2c1a4a5389d02311545b564e021f9cca722a5530f4cfacaa1a894dac4cd255e291b4c24aa7aab6ad49b9f17b311b354b8b2d9150b0f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.488623882Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default, Name: checkpoints/2688/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2688/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0cb255014e1b4f8fc3b1edbc2ec2fc3f1053b1b9a08f5132a281c2945a8e3de18d71d55b0349941c4cfa2aed655fa650be539e452380ceb7299efcb0537911746563bd7835d898d34227d2f37052698504d1b1ff970aabe70a3316dd0ea89121e5dcb130fb89c65a096de8bed6bf909182b2f4ff7e9c35d611eb3715701c579ea8e598936cb8bf7e3492404c9c12e8bded827bfb9c105ae2380dd899bfbb30efab0070831f370477a302bbdbc142ddf1e61313f14038d63f8119f4eacfcc1b3d42fd29d6f981f0ca65c6b87bf598fb3528baab8c6b47963d9e2785b9019bbb91b648a4d6427214647085d8ef7544a7aaa40be9272bb7515cd80f2a70ce7825f9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.488677567Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d, Name: checkpoints/3264/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3264/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a12dc44afd7cabedcc557c493c93430bf4fbaa2bb16739e828cf0f3f35e988887e438273447e129b929d4b71fa8a1ed2830c3f65fdef4c3f0f7696c2f3d03159bdde2b73032d11fa27681b572ba703598a94903be7b63845c226d05dab6fb273805b180fbfe777cd3f60ee7e9e8e4dbba2293ab3fbd4e2cd2c82a9c35c4889dde7548ace77b3502c3ff0fddab97d8174200b6a3065bba06644bead1a1b4f790c739f95f7f5c9aeb0f276d747b21ccce1f0e933bbb0749fa43dd3ba4bad46556fdd9aded3d051617e4cfcfcbc546828008ac7ef48ce5edca46264efa050516415045f1019cf9872f6f6a7aee772701f7ae1c8b260bffb43668f88922d7511e1b6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486559256Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056, Name: checkpoints/5056, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5056?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1a1f026adc5c07229b26344b7b51e40c389e6abb1755aa42d8f5a894b588b7a2ca1d6b3d97056af4243f5ce665ca67a7f304b108f50a0fcc5a2baf91e76c3242a6aed18b41702659c4e1a7e49982ebf0c12ce0836ce3f348ab83f4e76ab9e4ffad60e729f445a1a45564eb8a384074561eecefcd66ed33718ebfde4c6ccb8f938cbcccb0c1482d1b4143964e5b24cd9368886e1f7e1727fca89a8228c7864a878d8df1e39c1918258636416938b8a0f5bda0fbae3fc19962a724c9c97fb69424430272135f47e7d87e5b88fdf2cf3db94baa3054c28b865c9d400092bfaed2edf00ff3d6b3c8eb207d7e3fb965e7a57a0edb8b81eb70e7af4087f61568c3145c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.515895756Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/array_metadatas, Name: checkpoints/2944/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2944/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=290423198a3747254c03d86ea6f8b89a724bc6582b04bd9f46725a340c44c3ab65e25eaeb7ca6ee54cd73eaac988dccd7c8c3de6eda40e46b7ad0de623d13ca7460aaff514bfc93437cd76ca126f0379f24ce77e87ec5fd1e1be20d46a409c0916cf4ffe7e39509f4b8a5c9c2e8d58e28977f314fa6188ff67b51862019fda6b898b958a8e57e08dac5cc78f1f774408744b6b191cbb34265d834edc1b39d289428e58af759f30fecd79422f0595df06653f5cea53cfce53b76406e3dfa06f0f8d3a5d98342300cfd3589599386890926ccc24351833ab90ff5ff8131fd834422194abeb7936f0742b2a5035a54998f69a32c59e6ccfea2c3eb8047af74bee3f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.515924621Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888, Name: checkpoints/5888, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5888?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=96c1ea32e4bfc3b274bbfd6f5573fb41a1460167948e7094589594f71188d2191232e583daaebdc686a6cc37a9046283315cd5d2e96576fd8a6e7c51fa634b610a96c79963eaad67e62c1e3bd973e3412a137aa37e71924a37eda8b5259beb55d43cd6c7c9df1b34287383064adcd6f0658c440878627cee856b55aba6c61c2fc576cae7fcb4c93f16815dfc8c65871ee671325c21b8bb6eae4d4dac18b2a3172b864ba5abf78cd5448a997a8c0a7cb1ec2575d16fb3fc58abf4e429737b3b7e07dd0e41316b4ae7915db4559cd96010f3a94ef2270ae00e676ba7244a2992eb5f34c0f341f0ddcccc6538c56024d33786c19fa54db801316a41adb2a718926f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.515984406Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/d, Name: checkpoints/5120/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5120/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2f83cffd470256fab0ae61da8e1ffa685554a28443b35fad0222f9cbc2532a9c2a19799c005445a17c472e2369298952efb25aac36d458a06d64c4a079ef6f5ffac7a3e09de82b3b903b171e17cc3b2da468ef17e247c8b5a912fa18134b263de9a7f2d2088a4819c2856cd1023412f28415d216e7feb946fc7acfb96cea9c76ff233e89445160bd9b7d0cf9580f8cba0165f1ae5cb30bb5de1002e4848e72901ccafc3917918a74cfa047296ae8e49ee3ca3ab5a944050002cf2099f8c09cdd05f7732288c141e408d233eab9eb65aae50c3a3dc58fc03039eb93fbe0841f8417c01f9281d5b681e2c228c78fa56c1ee63a456fe81f18ceeb40c7f9b9917e0b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516061577Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0, Name: checkpoints/640/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/640/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=64f0fdda50558a704c56f62f5108e18c729e96a5a2438a54ec04f5113355010d1f049baa0c10a36e3af4da9bb651f7a0e9b299a4d3db11722a5a9820b995ebde23b1e32dc1c80311b5fbfedad91a53d3e3d5f03fac1ce256ef2cc2e7cb22ef9d0ecedfc3bab1f8c02e159f209ad479eb1d44a4f4c706bc8d02301f63d79bfd90f4329ee7b332180854c9bb04ec25d03995246c98522c2072f369431f5c4dfaadd5e93c7447701b7e8e5b231bb2efcb0c53670d255fb24a160207c2b1ba2715e793e469c1e552c0ea38aeb724009a1b497813c78b95c1d4fd945006523f46045eb7dff5f42ff3124cf5a2a9b24b67151423acc5ed8a472202c00e867069930e99&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516166979Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928, Name: checkpoints/4928, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4928?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a72d3ac941153a451819f3ab37d45160f19c40e51af50e348075fed56c9ed983e93ac3f9f4f150dfff3db633404a4d681183cb30647af0f089fb84d8947e16c46b4055ea64539cc23de8e537fad643d75902ac7811049a2374328a39e3bf2bc6eb0e5b4b7b9f31ae94c5d32bfd6a451baad1eeb2a8e58c04cecf89bd2810a35e3bd942c103cd43722a1d75ff72d9a4086d2e5f49f15acd94277563ef6cb3bcc8246eb7ed8c21e954117e56fcc7a46b413f636b00343797869e985982be819572cda3c6dc7144bc811533ff5c21eca3ca73e00dcc8aa4062db3edf36c9b8c070e03fe9c459dcecb1934acb437716bc6b6dd473872ccfc04c0bb6ef94a0f4ecff2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516276299Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d, Name: checkpoints/4416/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4416/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=15467994ed3b236bdcffdf942525c44d0f4ca0b40a54aa37eef25f5cf716660c5c9595b4e16d282f761ed3c141a3d17b33e2aee72fbf9bb543eb109182331080a9dc82167e4d1eae984383faf59ad4c1cc1ecea53f8326dcd14856fbcdccb8289beaf515eafcd9a65277392b2ccedf90917273c01987c7b62c3033303f3990335fdb5e7fba2cbf9ce2ee531c9714e5b53fff7a05eec2a4cf3288b3035668814d2c74373d7cd979a89bc833aeff00f403692d18829fcd80042821cb7ed923a309b6b1b2b133b7f3cd383345ac8674ad5dc44b797ac447805868ae329af3d2f1ff9eb68321086863e38975a0a6ac6d7cdb72cb2b4423f6d5cca31aed463e9756f3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51649456Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default, Name: checkpoints/5184/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5184/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=185fc34c34a451f46aaa231b36674c7edb064f2f88f6b6e7f96d845c4f627c25859d9a4baecc1a2e6d5dc4f2adfd25c5bb77657d6d65c40f2afef4aae96b891e6aa0b623b5f615008a7f09cb02fc81a1e28fca674da8adf2e5786f990b0bcefe9728c53f3bcf1a9bd7924928391be00f9788188757cf95687027ee594ec25c382a6d3fbfcf3c3128f64f8a641b42991edc83b1e04dc49094b709340ffa07f5a5bf3372716418f4b856ae518cd38631dfced1ec49d43ceeb0b8bb71e65e2f745e61319c2230039a017346974eaa7e6a657815c86f8944ce8a6bae8d4a45f68dca3596164ef97895a1e001b09d3e917c343e59082ec185dd7d9cf1507df0cf4994&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516511897Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/array_metadatas, Name: checkpoints/64/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/64/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3f2f5a4ef4ebdba4986ad6e8b4e127d670c6fad0c1df174dae61e4f16fdb76993fe1f408dada4cc8cd767567d2499a54f03310be5ac31fadb41027eeaa76def269ccefd7558df985933fa0b129dcd89bbd11a560385fa2ca0f53ab87037f46ac202662589a3a9e1a2cf3070eb05cbede29da6725667d1398d57caa9804e67c79e0c6390c54e80e7c678fdc15f8a2715b2b639a042364be6fc377fabbd14be7eb636e109a1fedf75f6016863a629851ee8957279c7a9a6487130ca84863b7e9ac67c1050d89f1b38aec8b08687422046e17da400e5dd98bf9bc3b30c1f37326a5514f05376bcc2d6c0ba26c6258a3fe9af9b845fa44478b46fbe2784e703eea47&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516550498Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default, Name: checkpoints/5952/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5952/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=41f0c667d71cc8723986d34c8116a593ef19a3842b27c135259ae845a3f04110cb7e2a2bfe485542a3dc8015fae4bc6c4cd7fc4f50f4b8ffa105b62f5940d987623e1887d4c374efb7c22815442bea51fb5d732a820f2b029f8001952890f8bb0f194d64a7a63fee7b85bcf92953b9c6902d25ad10746657de34909da3dd4f0b613fcd61131569ee4d260e8d9a13bfaa25cfe24502d85e936b5a4c4829b1f08d81e49adc28f6c15a5bda1629f5fd6b4ec956146347a57ae055d4663ce028ae36bba37e175db7b7891184d4c27252935d0c71dc36eb2f340d217a86d1387cfc53eb73dc89deb3be982df0c84099236074cba6d0e6218b4fd0c674bce4a323e1aa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483083138Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/d, Name: checkpoints/3840/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3840/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=450c4e9c33e97a42aefe8775c8395cae7d81e00eb534ec2cffaae0a9debc6ebe06175b3afead29df40c967035ccae40febc0bb16893069b82b83f8d7cbb0fdab791dcd1e2416b85f5fcf30525661207e545410e602c7c98a33ecf6ab030986f8b06d903e8527bc82e6627fe80846ad7930d4a48f985a0ac01e9a86d40c13cebaf26527132b665788adc5238397dfb5f03ceaf5e6ee6b8c0bc601070d7b2ed0e1132a7ca0857f1301bf540a4a262253d51947326ffe1f63f19ad6bc2d713524ce8df15a776022ff7b870a47d25ac2376e1de8e560ef9b3556fa1a37fc458852d6a2e0395c77f01e15ff7c3310ff8e3a2bbfecd85f21d04cd7fb9c5e4cebb22c40&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516554349Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default, Name: checkpoints/2112/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2112/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=524bfc7930f8b525daf76e18406261b4c4195e703ec81961b368cc45c678e56be2b7c6061908d295b508af5b046f30e0e70d479bc4fe321d015a0889607fd6ce17b6f0f482a021de5582d7f0631848bda9033b288d3e735790a81ba4776ebeda979396e631f68401cfb548ebb542ab809d25d76551080b2a38632ffc95ad3a79322122c7bb75edd513972b73ef757b32f33ec13950ded8e4c4adfc4560e716d844d8266b1db31dea2afd80d16c703f0069a6d47fa9bf1acfd7133f05cf92ca204219dc69fe54d71e5973bc3d39b0106657901bc05b759e135264e06116b14c07f32017862f652f4ee92c081e3a9453c404fa5ebc78e20c2a9d4172dea700433b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516550277Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d, Name: checkpoints/2816/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2816/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2a68cae55990f7a65dd63c0594e41e796e91482989b9c3187f4b0addc97894b11302401f36f4eaf717bdd52b06ef8ba15d94017336d69180dc1cc183ab4f581ba178527cb55553e65d482ba5eff3b8471d0905d9b83f0a774ddabf8f494821297affc77cf0369da6df6d6c90795dc569c95355f349ac30ef3c7180dcc7462d3e29ca1a0da3469e394a3f8bdf36f20140e4070f43bf3b91d3ba75b8ae695b71acf2515ca692e76f4ceca7abc6e8163b244170514cd71f62b6ecf64ba3940cd46039f359e117b0b4319fd413fc6eb30678e627e6769d427b180910417335afb1d6232bc943ebead8c4a0ed26d264251cff6d7b6bf834eca9ce7c5fd70e68dd16e2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516580545Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default, Name: checkpoints/5888/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5888/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=704d9d10e7ee65f067b7234328214ebde1ce4426dedff57543b236b7da3c4835a3809b146cef6fcd6548668ddd84bee50a2049b16e35168164ce2168476978a6c8b14a04a504cc74060682d1c9582677e0e3507a2902983bf4da7b8290a440bad48ee3300be4a8abd44adee7f9656c31ce24daf116e66a08b35a7f1a9850e4d2315ddd3fbbcb65ccc36044c86170accb987aa63c8c24481d287a2873c474dbd7cd02f03c53857395999e9e615e2fac33f33fc3d3581618ee34842d7b698d08011175a0d73e42d12f63cd1808817159ec390f7e6407eb86e2584168c3b242ec9e07efdbdd7fd8e1e83b6274453dd20d17687adf7994c3832a018c174e5915ea13&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516616925Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0, Name: checkpoints/4096/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4096/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=08dd0eae94603a0cacd4d22912d0b3c7c1cb052d7d9ab5d4e9e4361d64a49c102698bbba446a9d31f198bd75bdc7b4d77803eeb5f7f7d1f19a0c74e67155529f41c2e0a277ae343ce27bd629562c59b692ee8a6b72c9fd0ae7298ef20d7ac4daae8d2fc06291dc3fe10973d15ae649bbbd88a6f0b3b83d6333144ec58d95c33002cdd7130f7615f30ad80302da620eaaf8889b2511dcd837a0811915cf856f5f104eac63f8d036941cbb8ce80ba42f0aa03f68b5065f37ebced24f8c0901e7a86dca455a229cb2d7405a6c91096bdf400a4bd37884f46b7cf2a550dbe17213fbb122ef1115537721c894ab07a804d408bc57791306870a1522d10ea7024d3adf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516573064Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0, Name: checkpoints/704/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/704/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0b55c6b5354b071a622a4b810a0e70b2c3e92d362cac3d1ee42326dce42d49caee632884eb65190e70e13f082539ea401669b1a24d3e223a4ed22bcd14a1d1f65402b15e8d72999b983e521363eacbdbfc3d2a042773f858fb4da63642194e2b3c267ad44245daa2b634b8a20c47c8e33aee339ba9810c02dee834cd11a54bdc8d3f3d3c9b428c103ce3d66784f27cb49e7aaa4bf16ccffb938196be29019abafa15c649b4ce4ee3f02072861a6b1bd11048f131557f05bbb14234764c424cfa69f80c3c9701412bbdcfa112e89e555fdcbad156e41282d1d7a87ff2f54feec8161663064b78643e08aac59f6af1bfec0df50bd28e962b53fdb673cb12d18cd2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516616052Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d, Name: checkpoints/5184/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5184/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=113fb9fcba8e12e0180dca861b284186f95e3f5a14971f83940edc4aac8d885c7ceb061faef53371c845506f37161c2d2cb7999e7e58f2128546c54629346fb39757ff204892706e6e213e612afd9dc25ebbb9fb0f69d928dcfaeffb419f38ea988163a36a70d05cac82c48670856331de6ec6331173e49b3e42b6f9d658ada43d875b9e86a03a1d91506ab17b3c8272307b268f4c1d1c0e8832fea04dae18eb049bd31e7b3714aaada1c96317f6fd657a6d3fbcca22912f7c0ad45e6dec0e1b5d93a91d309700331bdf18468e02f0a0d223c55aa9d0e0d5f7dd9e0b61c1ce08a9c778faf307fba00fd285fb1d83fafb2a36b5692698b6cb7489cd934ad197a1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516612567Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/d, Name: checkpoints/896/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/896/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0418561c774508f2809b5883a92f6b19e81ea6e6ea8021fdd82e6bf7d3762a9a2e0b249c25fa2c50a042c5b1b115079af12a6ceec0722ad036b5e4ab6aceca7e492429dfe2d3aed2e4ba79e63890b5fcf97a346bd56170125d7657efc2b2a28f3eea50dbed9ba242d5d82c2a9a79f2766b97ff6ff61110ccd87c2e017d612c9315f48337e18888d72a7fab6b06e50d018d5ebea38bb6db311296e9868da736ba8945c759c758c89593e5619f3d3a8bf94749e4ebe21e44033e7c416f48fe21aa7295320b169ae2ab70b11b6743e9e340f13d815d46bb04250fabba4c782d73566221f63c7ff07103ac789a47176b6ae20693f7401892694f4c659ac77e880d33&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51664635Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0, Name: checkpoints/2368/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2368/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=00868dc8e3e3646b133d077f8a45cf32e433e1938551f98f7406412b511df6bbf14fb19e3afadf5e314fcff42327f5f0d0fbec95e38e279ad884c20b032a4436fce41c8fb03feec9f60f9dd96fc9458188c877ac51cf073a89ea4f1ecfd251277c4895080768ac22f6783e10947da499217a3b9ce4acee2582e0e2985bd94161a3f06d798795ed6b7428f196b1de8859eb943d8f0f09eafbca2d5bc7785b2b978da770a82630f68f5d0e02dc0090e376fcef1cc193913b78dffc9be3e3efcab8bdc94f3b2414aee88c310db2cd44ca7d623cb5e90c9d3244f12cf5385e13111c54e4fc54d7191756eb8b827ecd94eaeba2f189fcc8eb647d343d3d91ae4b999e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51653014Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d, Name: checkpoints/1536/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1536/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c485455ae6cca8f37d5c6c4c4913ddeda3765fda0433e057763e901603d3f59e96814a33b98589f78d59d3407461fa7c0d8f01497a3d2c080ec695668b1daa7a20e2bb092671fa911aa9726538635ab42e59f88418ab1d5ea6555c6c0fbe03166b1197c303b081b1246147cd876359d37a96601764aaa62086f1bf84a3d3a17a6d489ef49bb2b3a1b80fff297c00fe09e0ec78a7efcb30e5c73d469db238267a268199bd827f557033b0541b1db2699590a85f61d3e4a4f0b9eb191f727e2fda5efc3ec5b9f25174792864e20d1ef3d6c876d841590caa025b68a0ea9a380ad1571ffe5638ca5575f0d3161e1a1d080b7f1c205c715238733460d3a4ed0bef81&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516622043Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560, Name: checkpoints/2560, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2560?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7c6a168eee8bf5e953f32d715c5722fb60ae9e5f12218697724b21d401dd3f260c4ccdcc3c34e3b8ea99dd86a73e828d964d4012b12cc0e12b7894d8ac3b25bf34dbadd632ea56345bd641b37ea3a5bf62f655c711fa7eef42c0e152457e433f490d598eb68d0105b9ebc51657ef3a2f2e85473b4fb0c56d2d70db96ecf51f3d17391d299ea75a54f0d2a9c6f1351c9a3192185be7dd7d457e70e582014b7f0569950da7c43ebcfd046b53f0e966efb9f9e2a5f1342fcef3af3bb5b9c3b4cdeb61c58ce4ecabd0241741f88b124a0d5fdfabd40a8c766fae91160837e07077594b20f08fbe5e0707460401d81baa8380d8cb2224ea931254caf7f5f299b30b4b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516642481Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0, Name: checkpoints/4672/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4672/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9c5cd4e18b5ea508ccbaa01a573a8dcac013f96cc1555f47847c7e3c58bb9ddda054741c872487c55435da37baf8c2322f470d7e34f7d89c92e22e708ede63a2595f892eedf995293d6f2159d16f61a4d6163503b594952db7ebc01b57ebf73d4bb0d513bb0685f91345d8166ae71c339584f4a547d004007bd1d2352cc9f131d679f9911ae738050087d7a6ace815fd1328810ad901e2d37b452be14e6adc53ea5e670c55e31618b162bade769f82b16652a867e5c6b768ec1c6b13295075edac5ddd66f387cbb1aa6fb9d1b1fc2ccd00e2817f018027c7ab45917f3cc000f8c630faf6948912fc172dd6a01dec1480cdb536489790e0994080ad891cca5922&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516658207Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default, Name: checkpoints/3392/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3392/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4f4e35b33dd8dfa6112ad7cce1649aa0dd9f6ab841fa1d7c04cddd993e1b15a3aa37d490cf4ed21921e6e319eb8702f7e43ee82c5a32a6b7313d78b690111010c8212022aecf06b679fed77f7d85cee0fd7375f55d571e3564a4982a378cdef0036d950bffe63571d3e7ce2ac9a94da877ad8981480ca056389679efa3ef1a577a6818332a9676187aa8d80b2c73aa9d016d3e24a90efd86a3fbbc4e2efc855292ddec6edc8357da35e9a506fd46e1b0a449c03c2fdd0e8fa6a4696b1372f0ca540696085c63070ef6348efcb3e28f40e1f1e27255d7d0e294eb624f8512ec2b88d738ae69bec67b7b30fa0cdf70d155cf0859866889c41b9f6a71123444c0a2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516693355Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/array_metadatas, Name: checkpoints/4416/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4416/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bc5603d3288c77411a50cf356c797f6ffeeac4b26270f0c701e3e20aab097cf8269af0b8f7c71f2d379df8d2e252bae6c6f2c1c5e7677190b01d50442928687400757d3aa21e19fb4f5da3f1c21f3ce82482b2728c3c049ce169b9d982d3e81286adc3b238fbcb4d1b7776796660928c344ec330ae8f3e1c5bb7eac66a696cf1c5ee2f574725a2d119aceb91b2362e5d4b7d9c0038648d8bed25f3d845cee2e1d03daa76a6633d1664f6eb2e9983f140d44106296c5bc529eede0db6c7bf433d140d98256cc37b030a3237a7c8d522f986b39c86e8e85a0f43ed9dd53fe19f4c553879e0245762826c0c6baf45758a5696d5758d0906762580a6881dd756878f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516764546Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312, Name: checkpoints/5312, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5312?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c2f1899eeeef52582cb50858de7f169c20766d612290347bbcac703afcb14765a1d5f448fc26c9dc4d3456c10635e6ac1bd736ac6350d43a4d1a69de61d916eae06015b77f550084f36dc4de27ec1871db83f256cefcc9561b9254d9771b12b7daadbb3be0df23c7d2335106114501b359ecc4e036e95def3e90bafec17b3f521112434735e70d85ba9e85d0d616f8ad3170136e4a2f50d0cac09cccb3441fef480f0234a4bb136e8970b1634bef00ef636030007e30a72522c6dd5f416c6dfce96ca83f8f7a70d287b0af42b430b2b1b511ea84713ea7405f52084dd53ab6783b59a3c21da196e4a0bf19d799e6ff2e488903045b1c9c6dd0691b13023dea90&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516949911Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/d, Name: checkpoints/1408/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1408/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2fbbb024c2266313f36c3509d684a8607981be64d6cc15f8e813f449126f0f43ab611e5c244940a3f2e93976c5137d664a584e31219afc30a3edec957f233eb994216f3f72d7644cbb9b4ee162ee95ac3cba433d71ee7c65cdcfa637b4a802917b1fa4883279569fbe2e3a954b6ce4372d2a4231e1f0c2bbb69fa7dc26fd5725fd5d0113a8875dc32e27065246a4e6a981b18154e089d42a65dcc4948794efae2b70f71ef90ad2fd0b5ffb77b73e12b2b172b73b5589adcb08bbdf10466789656d2fd59a8152ba65fff1f026b0b714794c0d3cac6973036e42e4b09655e95e7efc5fb244f8c680556b1bb43823ac222f4c67a870f6d83b02d72e6da4d11aad02&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517139628Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/d, Name: checkpoints/2752/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2752/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=381785b55cd90182dc4c7457bcdf3d6ec815d574fd97b633b556ae4f385846c9e5cac0433ffb7efc1fddb493ff8d63169d95d4dcb81aeff1ae081863cd605c763b4c439f52380fe2419377039d96cdf8759975ce1f78aa83c04c661dbdbf9f70ecbfcd67feee80d2232286ca9e13c2370dac54491b14851feb83b58a2c400a36f984c5d395f7c580ffe519df45274c9bfbc08ebf075f9ac6fb5f6684278ad57105032a631d95092a98fdd3135c7b68d806b5db019e4cf07574a65fa53a9ed5e62386ba92f078e6da524ed1dc561b94662213772dfcced8c0d8a8d9a3dd4271f5cfe036226d2c9448ddaedfd6ea096047b967ef00d25431d90ea73239bb10611c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517158081Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d, Name: checkpoints/192/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/192/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=86c37dcd2e3b4bf8e4bf4679b56814c0f0a41a077bf37c4e6edcc20c772f2eaa5b576465b68bf30f800c1e18327f9e143d6453748623516584b81fa1eda6454616e4ca4886d958b930443aef4f2ff6d45a36c27af96490cfd70a2d6e698e63eaef732864170061137898db7c704074d421e7f0ee85df1ffb0221055c731fae6fd73c9bd065fe54b2949f7450e83fb9e49d9e44f0c995b926c467df76fc83b0ada425a552904fa9651b1043cb80c28c7172366955426139dc4421d6cf0d3a7aceb5b631a3902de34c0bb28820bbea94aa8eaa854d70b5a9d93f420e994ef0f917e1b9d81e6eab4d63dfadb84758b5314533038ef0cb9d4acc7742d955aba8c2f3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516958032Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/d, Name: checkpoints/3776/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3776/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=392c5b2ca11f40b67b40b4f475bdc3c63ecc56996a8d1f1127cad01c31b8608cf5b85ebcb48e0de986b05da54121a59ef7a33055ab61558aabe51081f41b31b91fe92720f5b08240913017715083dc7fbf81951325a52fcfcc69720e89b52b907bd6393fd35a287020013ba287e1cb8f269e8da91c829d84c6646a726549331511a65e041ccd1555057a13f7555b4adb973cdeb3b0ade12bf52d0c4a643148fe8e659b9240e77f5fff1b5ba40b4c074b5748ff06633136876a20c2fbbf071777cc7282ba46f81f361181b29493ed7b537a445b754c10636242b4879ef95aeb32b5e64c9f67db8fd1badd6b84a9ee312b83ff751fca722915e3319d6d9c28ffae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517196622Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/array_metadatas, Name: checkpoints/5376/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5376/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c1c5f0f7caf61e3f084196dcde204588aeeea54c7c74e40d6f3c67d11e240ff83ced4b68ef9ad262c07316311afc37a3c8e18254854ae4f202783ef823cf92ce3927c00773bb6c68578ba1a087177c12b4ae144a2044263659c593bec811391071c93ba757d57c80b10e243098d13254d85ea3a5bdfb050aebb50022c577ae4ec8fb2ab24e747420180d1df3951f3ae10b27b17eb91419260c98baf3632087fdfc98a2edcd4fb69382a8dda882fa22cc71204d81984302a28bee59df23629018a2e0bf066de714186af12e8c112e61c741a062788c812cef42ec41a83acf622199e1129dfd389a30d1cefdddf5886ae863d35e58cfa01cddd5f016632bf84bd0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517238194Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944, Name: checkpoints/2944, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2944?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=be5e95145e30e51d1149f35f2706dd7a8fd33f9be179d97576e9b41f2e408896a621e3745827cccf709187d54bbd12216a9f4553a740c58c85f7f6d7665e68934c2824f017d023f15d96339de4525f7886c574546085ea8795559c4180e5103af95fa9ec0b974f0498e31f2d391d0592516b8e6da29a5d45c9f83792c7d783b486017151f0cb7b4c915724b076286095980b076054b3adfdb772ce88adf535bb57c579ca6f0ca5910073e43d576a00671323716b88d807977e4343d2725fb389886c9d5c69a412f4503c9226ea72d43fbf2735e2f1bc944bc6d4464288d884defbe504716b5f0a98848c13644388356609d1d704f5f31f6109404822c28c521a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517377187Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d, Name: checkpoints/1920/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1920/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=61a69c55a7fb772a7d5e88266553e3870479033110fdc67abe537b430b4c21333c6c2fc69372f18574e468169ef140c8f5038a005c9f5f4dff187b2687cb6c048a830997661080c7678051672586977550668d1e65adb7aba546ef76a5b0f736bb6f27f6956f16ca10e70c650d4c85c12c829a59664cf00f0b137f0575d59bc42fa2628f6a6c9336ab57c1265670295e8b1480dcf79ed6768a257dfc2b58df63e4a5526210f020066d54b36886f015a4da7fdcd7dbac60b1130e3094baad6dae09de4eecc5a3a4744e238b77c3b3743e72a16dea4852a3c059b7e1b84aa988f7e72f0801247a13b95ef21b22c963dee9e255cb2867bf3b99901b25d5bf91a44e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517407975Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/array_metadatas, Name: checkpoints/1216/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1216/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1c955be9801daa043b5dff0c289ae2835711649c43aacbb6b11a2ff6761d6d54744ee5424177c5ce0f5f0e068a88e74ebd4cf6eba3cac1182092179b6c0d500299e42444ad2b89264acee0e337da156c0a71eb0db2f64cfd24522cee51f3dfc5f09c61faf5ff3b0e3babd76aac8087299cf3a656444b2424e76121b9548a92fe610d2f0c3e2788889b5c3e87c2f618dbf9c2eb661b459d91636163397b4f3efed7e10a1c9273ff94f7c9ad33b687cc58a341b257f21d2259b9aafe1d02526101f3db81e6484372c4301de4d24ec2a89834a6dfb44f7d69563fc1a70c8eda5c4c6329fc132ef828e8ba85749ecae947911b1a185b9d21a58fc2feca1db9dab6b7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517409194Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760, Name: checkpoints/5760, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5760?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=843215770273c9b216e7bf2efc036d98a4d585a27a0010312016e88b777bb0ba8e99d53e4dcc5a77793a6a5159d291adcc8598672d37bdefcb744582d36954b8fce2296db4e448885885e35348c11b42708835c2e91fb450f36e5dac78aef7e25a4c97410f3e90ff3de9cad123d9e6c3999556af856e1fa2f4407b70db233c14ba0ddccf89f3c4f04152744d1c6a5b7bc77635449f0f8f0f1cbe6540c0ac31a59e6d601939dc5f07d00a0958aaf0d0f0001762319f494bbb31319819a7580877925a0b92094f1b18a5c9865307c587c7e358d8ff534e4d898ca8fcc3e0354c9c776651a40d05e3071b65aafd764a28c4cfa3ec36ac2d39b7e645db505827972a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517546308Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0, Name: checkpoints/6080/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6080/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5985a282b87bc531237a5c3500de4fe5517a111ab145959aec2c530b277780b4619094034725e2803c8f27dc3ef0d9e9a81ac2b730b229bffbbc97f7f8d42662d6f2cb16a571387850316c9a7b051ec8bab2dd25541a1a0a3a56f99b3339e7dca704a02b41cd514b4336e0d1f514740fbc89a3e45b32e4918e3b6b0196c47bd3e1c68461810c54859465305867bac40122ae46f6dcaa99e9933f8602464c80a788bef3ca6f211e4e3413e43b6e41b04d1f528158d97f119a94181c6729ed076fc7648f4bfcf7d763872ff8f725f1f815eb931c506a224380d0e07daaf387403ae71412872efea2f026bb3e8ea4b3ab43ffed66da2681cd571d0bf8a39e021f30&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517571035Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664, Name: checkpoints/1664, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1664?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2b964d8c58733f2cf22a08fbf7086eaf2ff49ad2b90cff378eb45cae541c0149c0abfc540f7ce3478d979356c6c505062d48f198774a8e08694d51f31a31e6499f7cc48a7b87c52edcad4c53ea490710f976b0132f72adffe9d5805454087bdefbc791ce57f1a2066a5a04c629a9242366f40d2febb1a20a49e4be75f68c50146e26858fe214fed06ee453977bd8ce18782e721040c8bed04b89383d1c3203a149303dd9cd16d1218668feeab2301f3d656c46dff09d3041ed46d05a5bcffd3c033614d423fb6ed3173aefcc92658e6070c6c3fea34c296b060efa91a15694ada69ed504dc16a351ad321ab21f0b6abb3c0cf2a286db230b3fddbb0f75873f6d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517600799Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/d, Name: checkpoints/2112/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2112/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b01aac19e0993672426350e8e124bff9a451b2e7760787e66481daabe71bbcf8b0d80c1eea4ce6df274b971c37bbd503303fac5921c7b07c0845b007c537fca6de11a80000393415e89fbb33964bf16e4d074f3d20ca0ec88c8885530d348424402055bce6b63b948e1fa132f03766eb279e9729c0bd28266c460b16d104b9c1289ae6c82ef9a2d1eccec9ea4ee57f166d6d0faf8f47d504a0a2d2fc8fcf893c0267c45fe2dc9f36c53312a09b41092a4d416553d76ea060f95ab65cd3e91ae74e53d32451d46451dce33f9ce3ec286a26593d12afe3dc7b0f891e45bac254f5b2d4cde5291e42a0e6c6db7ea89c803a7d8500c59e5f5fe36dcf8feb1fedfc36&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517641785Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0, Name: checkpoints/2176/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2176/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=938a976706ecb6cbf96de4c4d3fefc980d3093d292e6d0d1f8eb001401dd96fa669d2df72b0beeae7940001c362414782f3def9cb2e604c088c831f053649b1b4693e9f114cb6f265a6c538349c4d27602cf6dd89e54029353859b6dd104f4138bd5c68835316b4e7ed19acd1e230d89c5962d76132ea8f832b6f54b30672d6b1e337e03e51e820c4bf238dc488e98c2e36d3458c33a31179d6516906412c8665c54a33d5b1c89cef56dfccd359b77e8768d9c7edc191decd73a5c35e1f57fb9c2171fe8ec6f044009399947952b8606df41ca767155a3bc431f8d7f0b2372f38c64d54c7eafbc738a6b6f4c77bf37ec98132f30468e760a26b31f15636ae460&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517709905Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d, Name: checkpoints/1856/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1856/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b491c6e2ca62bcfab4230de8dd3d2c44a24f8084145b4075a1bc3160b044daf40a2791a9c6f996749a7bf177ee3e7dbbdb9df8f833e7c71a7fc22b029637b179859063bf7e44cbde3e6da4c48dc68ce1b5d62392e8850ac3b3e6ed85e95862beaebecc4cbd2921bd925d70d86f8561a3e79ba6a596eba8bc6ee8efc3a7d08db50240139d028cae7ed72aa9814a26488d212702b47155fc2af7dee55020bc7ceaa4edd86b11ea0825d3f5a895211e207f390d0e7b24b43f139b75ccf5777ae855ba211b629e6399d387f93a68eae0a6d8099038dfc2a2329a5ddd5c7f62638d9888d88cbdb559e586202dc6657c2991628951efdf1f89ac9909f5869095591869&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517742304Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0, Name: checkpoints/4160/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4160/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=095b9c6d41d0687a64c3e2c985258df640f5ed988399df5a2d22e8ce03f11e438ae72012aa900c5cb79a077e6d11df85931f463c0f20b590c4c4d56d218913b3c562ebaf82b6c9017c1a0b4878feb90e9efe388f1f6206499350fac656abbac5fa5e8b24948f3990e1168599117b940311434347c20b06549bab963f3990ee0eb29892fdfe880e01aa03c6e8189d6904a878b1e92d7e1553f57131f00d8df24055c55e6a2de399353b3b8ec02b57187326a2d2385879cb9c8978cec7e01f93f3a7f9e75d9250285fbb26250aed8a723999ce81984981cc35367cca29ca27f60aaeffdc1f20a81b4b01b3f9d3cf572030700a61b23fbd148db6e2305dbde21667&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519210522Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/array_metadatas, Name: checkpoints/1024/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1024/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=92df37ee08454d099089ed04390ae255b12f910bf9c92d2a2a2b25fa28130914ca063914cba0c3479dcfdcdb61f7c62dbd2727159da8a4f81ef6f9bd799d2ae204c075f3a1f90b12d122376caa0120d95fda54caa8b0fb17120d29c105083f43e15ed5c654ea2e081914d2b85d4333617a8d4156c09fe3ff9d3fdbc5d9d9d8e07ccd54f3646275aab95640a86daaea535d16a1199aedd37ce50916ee583dbf051380af534c6f71518fbde284cb6c7920a3e7b3a699e0308b4d5a4cea1bb6c75c5ac745dba43ddfec387fbc70ae617d0111482d7ae71f552f405887b2bb323d6eb7ea77d6952425c6c6a54658b2e2f75ac4c33bc5d75e5ec22d8a373945e9d1b8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519234516Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496, Name: checkpoints/2496, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2496?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bd78d45640fbb9459f883785e835d684ffc135ddf14089803795c7fac384607551135e7f52715d3d37460eb9359c2b921e7b00b7d75c72c7b232f2ac602dd22ce074df9091d9dbfcf35115d00c208c8862bb9e8c0ca285f11485e84c7adf3873b457476c255fb532bf59f4526907f763a4517df83bc38450e99e51182eb415a6f9747c8d781a63172e9a878e8fa517b38e829ef575246cba5c71b5ed0904fe0063806003b214236869ccc7edc5ada269710871fc103aaea59efd806e7b03ca0007d82c0dfec400850b1b9c12904d21f6d9b1e047837d96702e54b3704e23233b30a443f39488bd980e0008e854acc8121ca32e08110a3805d852b91015f4d625&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519288913Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/d, Name: checkpoints/4608/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4608/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bd6bc6f85f1461e92f61f4510367e317e92abdc996e64a801fae012a74828c9a35685d7b33dc899747cbb1a312972c7f708115bf014452df6165c29fa8a0cf77d3033ce60febb834bf3b83eab84f5c5d58c1fdd81856290df6b6ee5a9dc1e137237af56ad7a19d51c1a1bfd5889a0452ac48825c3735730e240b8033467c6513ecba12e831caf64bfbdf97a810497c711dbe9e6eb100a09c6a096e86991e73bafd185461cb21faa36ce9a005109f5e1e0b384284f6f3873acd96143a8d6e13fdbf1f46563eb9b9e8bf1112e25f671fbc77d7a264c7693d5afeb3dbdb5bbcabd7869a2519ed3fdcf43eaaaad1b1ae596ca28cf5ef2da54e2df3df6ab96e1dc639&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519293588Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d, Name: checkpoints/3072/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3072/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=94d232cf5064cfac1a7a254a6d24f06188e2955e51c4c2ce5098aa46adbde5e79476cb1e54a0fdaacd216afe3ade7628410d33af92f5f6d33569f9929ac572eff47cdb5ecd45e24adb0ec61888ab2e48976df995a6d455d200090d9b7b0cfc261ec65040b877f8dbab835bb91e2c9b8bd5e590634df60d209aea52a6644a3eaa6f1e83044a9c4ce848181963a118e3c8c32a9d03de51df80f4e215479c8a21b2bf1147e2c2ae63b7a69de0eef763df614902a37133e2c8a74b5bf590fc596a787666bb581c752266d815e4af40be8fe77d60e686164b9d87f2bb85a24cf65b8fdc3f3ba25b6b3ae2e2880edeed4e8dbe1aa415a28c0ecee22ae3648d0c757bba&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519298062Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0, Name: checkpoints/5632/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5632/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3bf97277e3abcee9ea8aafd3d4849e688f9698bd9e6226c09b7b5430e210c44d3115c656ca3fabf43eeed0b3c2bed90949e64d28c737421868c539c2063adba5e28d2206b6f89e0ec8afd356cc406335563e290b25567d222d96c8b1d78f50b57f6c1258bf556e98ed98a666242af095e315744bb1955687f2c77425ad6764daebb8c746d8bff82e64c8952a8991fdde0e8ea70a7577ddf40316b593d95c00d26e649a3b2e45133951e66f3906993d01feab29dca391ae5fe6fb31b80f9bc99a8d2d9eef1e9506938b48ffe3b09e5d979e11b203ef9fa5271c26a942365b9a95da21ddf08a8a6ae9650cc5c2fb34e74b4976d807424e2ad7ad42dd5ffce93c3c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519362649Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/d, Name: checkpoints/1856/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1856/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=58fea280c9473759327c8dfd20985f61286e69ae96ab3dc3ec98d21bd093876be00d4ebdd3e8f0d8cc3ebda5d927e41d63a267c237cda87d93c907f34ccd185c797051762fbbee41dee1a82da209e76a56c3b174a098ab753474909e77e63612ce919eedddee01b06f1aa1640a24c4aac2a4bb3c079d137d454d6ea9af4d6fb7cc1b92135605911355932e7b569f308cd1e8638b50d966571f56364efb5c767b1da37d826e41839b2be23054eb77fa48710927b79008bcc0274bb7b968054c4a9f2383bf13a1115932cbb736ab2b62120cbd866aec1ece4e0adda43f7980b43a55315887ce4dc2b58379deef28d3562da3ccc92e40bb62f1f3bfee52da139193&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519355691Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/array_metadatas, Name: checkpoints/5504/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5504/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b40abc17ee3b230e8660f533dee49aa2c82f7ee051879b27975d092d49a3ae315111320633bac0eca280f7fe24373abca7642418418ad7611b253f89000a85bcc989e93fc7b3049d569529c5eec4b8d0a90f2bf922372d3a5fb13dd0bc7a767c8965584dd1c1cebc3168ac04d3a0879cc173afd12ab29e2afc7c29bf31f514b2cf18f35aab05d3a8eed93e3e528f8051a2877145e4c48a3a2ce3867947e679bf9bd3d2d936056ffd1f4dcc9142ba542a0ebc713ac7923b74ce73734a0a53ee96d99f06301cedd3436c12224fb57a3a2d018407a3e827c2e41d5a3672719bcead2963e80ea51e3143d4eab6532e9db55a679d7b8c1b06a0eb3ef5d16fa94b9497&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519345441Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d, Name: checkpoints/576/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/576/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8a17e8e2a351f21bc5b24ac91f0a78ceaa0c1ce382550de35eb20ed57f40d3031cd3bf7fbaaac0e2a393c59b78d609c05dd283027d8f0b9a39ee7f5773575d7bd32339c28f518db36a249739e4e51f9df3e186d7312185ca687e21d9c804bad4376b1c992f6468a2f76f7b870be5c52ba3e848a5626db2c89caf348620b423806398227de5a7454a65ec61af6d22f3df7a268c5a7c45bdf191cb44c1c4b6e2f89e8617a28fd60250dbdaecd9e892f659ef59c41944da44cd9ab0360960266161a9f01294b79f0098656b82698e9a67998fa7828de978c6264e4a399ec264360e7b4edb7a2ae912d64dd311716d119d29af88a13351f632a29844e1d3cefa2bc0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519363611Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/d, Name: checkpoints/1664/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1664/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=adc2bed409a40c606e1b4f21db8b8f8743c4548061fe621b7b1561d86def21ffeb84ed1b792b8d3460b4f8c51bbcdf03c5e97e917c1d58c7208242c7b04b73aaa6f2dbc0a133406404ad32f1114053435c76348da688903cc14a9351219d68a7b002ea84468835fb56ed557681b7d4dad8d07bab04aadcee93c4c7d154c957ca836c84e94d03a7b481e791a49d0a9afefa799da5cc8ac48b707629513ef2b7d4e280feee5243b9602f93b9c852928be26e70624b7b1d51f27391b7f57d637c40ab8376a16cdabbd51e2b985afcc8c9b898d9346fa89d2349bdcadb5bb8e5bd60134e7d7f0348dd6e605c0fbe2a293cca0397c624345b1ae025f2a61e72ba3e6f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519367957Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default, Name: checkpoints/384/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/384/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5e8951b1a64be51573652a907375b0fc92f4dc1c8e4e636d9898614ccf589dd0f9e19047c1fd5c1b1658e08d44ab377c2ba760b4f0c311e86de3a2bcbb8abeb1715dd0d90f247ce1dcbfd51817c8ae6a6bf07dec3bf30b374c3bb2f566202b2a1f234fdec0744eda892cd60ed611dcac36b7fdf39f94454e98a8ebd69b4bfd9a05ed3a6f1eb5390115cc3363f97e426299f0934ee80c86ba6d63a6d0dd94d65986a8fb25762965915d1d11ba4522fce729dfabcb06c09f39a2e304f7ce2bca1924c9121f041f7785115cddb06f88156850d16506f2582b4f33842c840a0c504d59746669666e5c5961f4b157ad36a4b5e69221858dcfc36b8957f8b63b07d088&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519402783Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016, Name: checkpoints/6016, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6016?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5b3b75aeef2ab0502a5d653f0f49dae4da0cae0b183acf2fe4e937c78407a92bc39ecc576433a4d8df571c62ca3b8bd7956b7d1769dd14fd5e04492b182b15431111598e8885f4ab001386791636e608ca3dd989cccee2b0968bfd7f44da6b319345b8b81ad69232681150f5bf5b6f5e05ab964fc7bf1244bacb756b288329b0072518a04813a0f546a4f4e1e670e1f60e6986e0c8503b2c69913b316de4975ebe9cb4da447f80c85efa522dbcbbacea4677f5d4f662d19fd7a3ff1c41171d75a7d734b91319d07356afa9e0646b10d51b9a26effb39077771e43dcf16fd5d089806c3faad892f9fdf197ba37a54a27e6532520b1db6976cead2ee9c1e3f11e1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519420454Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d, Name: checkpoints/640/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/640/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3486b2c5fcbece5de3645ad3c3935de1fa87b5795a03eb3618681c8b30a810e864b93c004567ae0a941ffb462b4f9ea4b5a036c49eae54288861c6995108c6174284521ec3881498b7e22da63b49fbf64d1805fe09b313551a965f818c4d2280b321f174f8f7e62e36230c66b25c505fef80ae535f6edc95c0f09e3efbec9c82998c6c2f631657d4455384a9622f23171e0d1b1d8c95097695ef18a0e8ec67a726a4faa127a001f02292c3e9d813dab51ea1de0dcea9137068862113676ef9d61ae19f1e4490c83718b3445c716ebd4941afb4fe992ea0f94778af500316239b79bcf9cf5fb33bbb434c09881e37c03bf1d3b3652cb4c1f4d709ff91bbf4290c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519443291Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/array_metadatas, Name: checkpoints/1152/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1152/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=59e744f47bc0c3659f97bf58ef4169645338d899e911aca2344d2deb834609eb98dd57b7d5b2e22eda8041a8e3a5e769b28ab0e3b0f07f0fd11a2eb941ee96e78950ebd3628a94aea21a53be43d3352aa744d37197ad9b467244f22a93c8495c4a6c1e792c123337d2131c1efc55d477984f67b946ede273d056c304b7d5d5371452a70cc36e40f5fc10a5f5ab2af70dc681139dd064057aed2629c3a67ae80a2bd7e4e57eab095a49b6ddaf10bb86e5bd4535377cd536b1ccdb819d456d3c7b3232b323ff9fd22670ee066f831201f4cc5732636ad8dd68b8c20be464011c95e86fd31334ae917ef4822f181ef48c6a926f40289096fee627e7e517686d309e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519464762Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0, Name: checkpoints/2304/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2304/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5aa1c1a7027bb7bff9a2c6e41ba9512b37cdb67f1605202f49d5dec5fb32480ce6c6489a44e1a10603cb4530cdb7cd05bc1928749c63f47254fffb581436ad1595706c60181a4da2eef08a8ab505e5c5b09557a8a62ef8542febbded5edd5b32896a5fe971ca5e603fc16e93069526b03767c54304ae49236d9768fb3b633d3d2a52e4b96235c9358074a653bd39df07c74529b0ed70440f5bb3cda15ec20d783a9572d73c463b91e89ba9eb7be33c0395999d3b6eb08a3a2f64998c1435333258b11e668c6b52adaab6011cfc468efa37bc853077b0614e6f2211f385ccb3c8d7fd062f1fa22bb5adeb4564a5a8f992bc271eef9fb13a97251d9c48a99ca4db&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51948447Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992, Name: checkpoints/4992, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4992?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2f92624fdd73453ab173676b07e776823927f6ec6f3262f493eb0fd955653254ed03637fd3f941bccf301f5248c162c96cbb11721f5ee0f4cc5955819d08f28614176bed5eceed39b8de6d4248dee61b977a6f0026dc0ab66c76b3423790b4310b80620ba6a030d09d997cc93d4ebe7a784b212cee00fe714db6497022d910734b42021bed62f352c805566ddde4931d380879177f316539412966ebf1e1e15682f603d6c4bb46cf51e0974ebf304aab9badac4f11c3291dd38ba003c715a5b5fbf33279d5767fd7b1f49af493c547423460b65d281bffba4c3ff9856ea30f22f528b2967c654634164cda9b54a09add067182336e9b0c2c28c4bed983a5a1a8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519470606Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default, Name: checkpoints/3584/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3584/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=15e8ca757c894e9496518940f53885a85597441831a330c3547506dbbb2082fb5ed9c4517891ec1cab40b918598bf47d23dce7f31ff0e106d1a5436e95e1941ce98be336091c6585f46343fb15b8b4ae8c756ff6cb6d3635799d2209f9e4104f8f0577c76160c079468a7db07b41994a31fd02068352371d218073fd5335dc079082cca212527b67e97de0828d819ebeef4b281c814214f931ca8089c2d45fd85b95ef81913d8ec9f92eb244efb4273df4a6709a708b35fea47e339328c3853d789dc69ea524d8ed90e7291279b3dcfbdbcf741556b507ab8a6208602da2993217e541ed537b7fdacb00a22bcc20b04bc3aeb15347405e3b46e21a6cbb2a1960&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519473376Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/array_metadatas, Name: checkpoints/3712/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3712/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b9b602af34404de1915a5afe936760f6b6d84368a35c2cf2785340f4ac7b452eb54abe18340a078eef429e1c76df4b48ab0dc9a0d1edd0be70d0a7cf5e24784977c7b85cf1cb65801b1332ee6536b443e83e122d469a440bb242b3587335d16ad9e5bae5d269bab46b2c17dbcda4a0a232dab6048266a850b25edaa2ac54bb07a6b0267b78c3ac0e6844b146b81f26c634623b902c0ccf7deebd64b317af5e7d7c2d96b294622343351abe5351c8cf4a1db56b5cf6b7574f044b4c48e5ccb9b774a88878983e5786907b181dae9deed7381e8861886473bbebfb0d480951dbc1004b47a1e4517de41e09311d5eee13b279f7ebee726930955d4316aa67fc34fb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519479276Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816, Name: checkpoints/2816, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2816?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3f28e04ce01bb38e2a2852308e9f517758950c6b1dd832fc77257711dabfabb60c688122576f3edeaa62f7ae7352552d8a6940c6496fdc8ecd2f58c3277ae0b4eee24de7612396bd6fe737baadba28766f13b4fd7c6e7417782bdd5e848c7e66616fc55171e6f925f6bbb4b3cf72f18c1ade02ca37df1d40827036a5ced5d039e1e1b06797ad3d19da88bd683cf6c023a3a1b58da4a0a99f54bab2d9972ace395433154f09e7175e6062ae26b435531101b2366f6be6569b84a820ad4c5b0705338a2a282f85ef28312b6a8f56797e9cc8a533601c9cb68193f24a3bc37bfb4f22e875b42c575d20f54716d2b2334ef445b5998cbe3d061f745acfbb849bf69d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519524013Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0, Name: checkpoints/1984/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1984/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1b7b3001162ab4d123d208506f9546c87b80bf71cea78de134608974e9ca9be17e9ccaa66a5cd2851b7a639b1cc07210e32a63735eef446abcbb4161cf240bdec0fd9d8fd4c477624b47d08c756fbc3b13164da1e00227ee10f99d199cc1e366ac8c691523181ef5f7963449e1d3f14635aeb6dba7a92c9f95c0446ee9df56ab0dcaa9052f13155a6646c6385e998df6ffca0b337e9b364468bb180cf02ecaa66ba6931eef0210fe017786ec9b1475b0a51f999ff9e81ae71ddb54a111d31f7da2ad497f68c4db8ee1afe2bfa830a15b8a1db7720434ea4ae638d660fe65047fda973e0705f2709f987d7fa8700e3a410e2e39eb6f6aae8cb792a3d1ef3a62d2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519521638Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d, Name: checkpoints/3712/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3712/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a1e45edce35f90ee7cdd20934cee6c0757ca8b55c59e4665fff3f9343c11655213b0f1213403a4f58cf01a4699269d772ec4e314bca9e17fbf8f0e5a42a07901d527306073f027a73481feb13ff95060dc898939f7987d6c3786b65b79a5a55f9d6553fb136bb6ec2f941855e979cd6106598670c91550d2bb2e9bf160beaac7497e66c9de3ea5117e31fa6f19c908d6ceea8f9b69d5b0aed77ca3b64e347e21742b6527a450a6bb64dfbaf6e0e080162e399ecf23d92ec1777f6f170b58e4a1a2c504b64b6eb26cb85b90da755f80583e1d7d44f24080a5fbf01a30df4a07a0fdf6e70abc0029e3f8594ddab53b47e1addaf898fc2fd29a689f6788e83c58dd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51951561Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default, Name: checkpoints/1344/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1344/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=84a990b43acd0f128a44d05ca9a958ee3fff13563ef321c44a738ee50c26a8b8abe91ec8f191aa81caede8daf790f01a951f78d5b09b75fd7d436aed32984da8f0341ed4f3b364cf0ceeb8ac8c026245530a16118210d02007d7e789edf60d975070c99b917999307b7d832648eece4483117fe73cb719855c3faa8038b038d6ac81bae928ec702e2fa3d463906f3b4f69132f55dde14e70944d7322af1c6a6c1b85b45b5413814185e2f5b441d0abb2c2400ba4e1bf27efd78301e25a37407b043283d5a4f8e9d1ca535c693dd5e1a0e8ff5649bea09c7aae9d9a63aeed9d71fb980f6fd3043fe536c0947cab20cec6059fd4ab560f37f29ad2d736a95103cd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519538242Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default, Name: checkpoints/5312/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5312/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7934008b8f0a142ff227f482d17d5aa800831c20b316d95b9f3aa9a2f722bccb7c175f979d60885ad9ba1fa66e71a3aafbc544a18ed16f0396d5fd85980dcd78df56a3fc0bcaca3568ae2f7e5523323f8e6d3e27c83bf8d618c3aeba43e43aa6bcfef30b9f6b0320aec3c2b1e6b1610607e06918bd645a550160233b04c877dd7b7262669008f61e69f22017bf69c0cafaa6b1cceebd91c8f88d5efa1c580a6c503bc79a629f58f4e5a5721b05ccf199a58ac7928eac7f7c2832baf578cc11c81c0efc6e97377b93263874d7f99a617375e705f05283499d0be4476738018c07b67cae14573085022b61e22b13f5fa044a4ca6ad1580d26135a383a5b1d5dfa9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519541543Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/d, Name: checkpoints/3200/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3200/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bf1dcc9b133ab40d09770bf7e94aeb20d183552de4a7a904c1b621afd0640e3b5142a583e06353cadc2752c2511a08006af6293470bfb56aae81118950478b012e4e9c858ece631e28416d3d21c31560efec1d40785df436d7b49b43092d1d67139ca587c00ddd88dd4860f73f05f1d89260e1daa527eff55145cccb65157b7ca7d1d7c8091c7257d31137aa4098a2365b3cda9d14de82803724438919f1c1be373129037e8aa8db60a276243547e2f8721e1fe9d043a34650a1c77e7db9605dd1bcb4ddf852b1cbe5506c04ff4291b1fd2ddaf7346cf8b324e4bc6df5218c07549de8daa70f270ad7578f8a52b1a8cd5baa3a666099fa42584c21843c05ffb4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519537875Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/array_metadatas, Name: checkpoints/4224/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4224/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=31fe3f2da634b3d548768c854487e4aa026b5085cc445835de00f6c79065619ff4406ce53aaac057987ef7d996087db79b864bf1851ef3fa1b78ef432222d9d11fad6b6ab121a87f945905115671914ca7f3bd2d023be8ed614104d8cac69c53ae67e8f62d305de0309ec208761a82c4cb320f0704621e83a59d360fbfd19b8c407280ea5b022679f7825274b40625e06f9ad88d8d2e612335899d02c3a296829a8d2be63823d2ccd81a921363b7baa607255672b4cadca4126614262d6264b4b21534187f3b71ea733cd9b9beca978cc73833cb84109950b7c2de4acb4341cc4cbfcc1ba3735c356647eaeca7b16f56ce857206bd7fe2578adad8b5a438fc18&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519588502Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/d, Name: checkpoints/512/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/512/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=591ee0527b7481008b8557dfff2d8d360ada70cda909c096e6501eba5566a06f50e8099ef973bfb5a82a94b9067d6d0f09a039820ce7627e8f3116b254cd13315dca3efcaca46a68d00166244e1971213edbb4fdda085063e00857eeb3d5f854b4e08a6cc49e9a54a43d718819c86f04581e00e17fa9d6bd693487e798b14aa8cbb00bf0c884d926fcb6f14425dd476dcdcb1a41d5e0c8ad6c78c9a87c78d82bab715bdf3c87b5714fb66013f6af4ea37addedc9813e8b5aaff732329e8385983e470294a56d637c9ac9181176e3aa73ae12ecdf7cb0e74b1536ef0db5dc1a0730cc2aa9aabb27108d06737bca883d51e2234d60cff35a232f515134b3292ad1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519606345Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default, Name: checkpoints/3008/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3008/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6168c81f466ff11adbb3bfe702fbdd06ef3e6ad43245d60fc4856dd17a61c2803681044e542bac84f68a825d6aed78754bc3728efe71306279003891cb90e52ce422f5082ad6608228d7eb2d59388af1e22a5b96d7315bc84848dfa3b58fc7cb49094a803387b4a4c53c17d2a1979f14320521eb3cb98166838b3592564907417f23170e7c742c96e8e459d6fa593ad7d624417083fb4e2c3ae9314fa3207332cb6854f7881d1616e0dfffd965b5128dcb3d8e437860e2bca38376c20d04b26dac02865b825e31d5c4b6dc5290a1952ee518718dd21072cf29463f4ebeb7eddf487c88283b620b039a70f315a51b4eb1db0c5fdabe86ec64b7c9bb1dcfa7a4a7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519629491Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/array_metadatas, Name: checkpoints/5760/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5760/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c164855580eeb445e7d2ff5ed5e80dd132736518aa789b3386da6766597d8d499d9b1288ec506174316d8a0395b0efc529ec2780051341b6bacaaaac77cdc65c55820eb77f1dac155e6c26e85cbe0367aa37974378e6c06428ac1317653ed7cdef272146908f3c448fc6fe27df50d566646d911863c01c0e92e844f50d6c59814a06c4c36281312b73dbc1d79f56972ff16bb7e5a058cc50896f737246a8976bbbf72c358335361b8d1ccecc8c7021f5b8bfc11bf5aa983dfb57a62fa211ade6ef947bf2d98aa5a39063e5e1e188b4d17e077f2df2429d2acafd18b0282c5f67fd5f7d1633ad237b5819d23a24ac29d0bf91ca321c23fac40cb02fcb28b69f2f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519713479Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/array_metadatas, Name: checkpoints/5696/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5696/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=398c7a60f3f8e4c418512a85935488c3a01ad9395c9de5ab794906f00da290b4836d21c0e74c4186abd699860ca0522d7d9cbcecf9b1b09c2ccfc5414db59019f55fa5f36b86131192ce8d0e73081bd5d49b0e326b94e2c2bdada4ac3a6a7b30fa3f3cf7bb0c9dceaa6a4c15ba8122d2ccbb25b38cffa1955f34c6f56ca9957ce4fbb8a6191d4657f064d4c267fbf8a0c5fe3aa309f7c2850e49ac8fd2d0e0d463e3af3f42cd556bc51666f93be653257d1ab21ec8ee90e42955ea4220d962a640c25806e1999caba510f81d64621a8b9cb5143d75a0ef1c6b4945e24d31d0e215b262e662745e7d3716a67250415a8bf73165e452b2983f802e87663dfd1be9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519716149Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512, Name: checkpoints/512, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/512?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8bec80a5ace7ae38aa29f03195e1f9b85aef7a3d62ca1b6c09d751fe7ffa7250ebc895d0af0fdbc5ae31fece480eed1fa22c5708983460714d5f4a70d964b9a395192b84a1e778c695167630558b8849f6bec1bd8a06385463dc05af299bdf965cd513b7167d244eb62e9cf9b9641df3d15a5834636c6297aa26ca403f85db45ddd3e1342da9d5961250c9ce3f6fe6e404ca73aeb80804ba8f1723e36a023b1d197d4603bed3263abf9914f86db583d5a32a539b9c5f9d3da43e99523e4f63f8c73e1a0bbb890c764926dc01568296074262c7aa324f3d269ab9379b96305c7abdc436966c8b46f95cd16e117926431ff0911007fc3bc93bd71372c3b93ddc4b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519796874Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/array_metadatas, Name: checkpoints/0/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/0/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=28dbe58fd5c270ee35713b548551bdbaaa532cd26e70f76145de179446edaafba89b51de151d937739557ee91a3b3e3494e8af4b32365dc2f38ac24cd7e62b603c705045ca2c6c4ef6b0434c494d9f4d4d4fae1f72617219ff3588146e68bc75a6a5ff8afb66f4333fd1a88f1d579dc223e95a4b85b80586e3aadb7d2ee68e5b6a33f88464339ae459ec8f99b292b517125e327bfa18cdf60fdb0b70293f0288ec970da077f6ae4c1b07960b8936dcf6350f05fef0b340b941577b15e701fd9e6340b6a47230938a3f4caf5bb76729581d467979d8c3ef01a2d65d97abd87c10e0ad04ad48c04d29902fe5dd131cbb81b2ebd036d22ca0d11f3efe01cec89c1f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519796982Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d, Name: checkpoints/4096/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4096/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=baeec32a0adfb8ca3f96fd3190491f5bc804fac843d684d6459f776e246f7f0b6867fc99ba847ca0128caf1fa28391d995095747d15aa7bd01edd7e559eb18bdca401af5050baacb596f13d8f28122d277c805f1a201396c4db8cfe562b495a800a31c0405afba43eed4650f2d44a025572ebe4dfc3fd20f3dcf8c255facf46746ac67fa7bfc20f1e3a02dee77a2c4b71a09b27b2ccd91c2c2163c7d923827cdd12c99b7c0350ac1d39c55f69c26a9392a9d77ab5cd16561f9ac4c9c9c7fdb4bbef82bed71ebafc5740d55fef9bd901a84a71b6ed4c6d52b98d44cae5244f13c36922f8ea88b76afd911f820ec168b97a3097969fb30ee3647a51f8022be3d06&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519804787Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d, Name: checkpoints/4736/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4736/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=68d250b2d5ac4d91ad92526504b22ca57f874c48afb2cb9392e661ea852730757899e97dc0c9d7a2c9c45e3d0783c859e68250c8122b2f8c08a8bf1c7e5bf6ef164897c441024064f63ead3bd4ae9355dfb2fc59ef5744e1759cc6e81533d419864bbab443fda2cc4dcff44a047a847a86796c024355c2cfe6aeba5cd9e71c64c988f705e04bed4c48cfe88d650f467e2aab3e525d6be0ddd2571a29f4992a4c387793e0e5f2896dd418691fa3d74672546fb23b41d67396170df1c159bc9f4245862a2c61886c794dcae15fec00b15fa61ef34de168b31780423bb6bd7e0f154ee3dc7aed145c592a0b9d71afb379aa60882fea6f76988aaf28e35fdc004f21&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51982243Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136, Name: checkpoints/3136, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3136?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5231d81ef1baaa81a5a0fa89d66af913504870fa17e5604feb7051841b42ecc3d9676d2548f9ca8605e5597dd24f992cb50ec171e8449dab417a3a3f1eef67f2399fe9564d38f5dc2449c65e507502a7cfe7acb909d4e7f5ce394a7caea15008271e20b7054c3d693f0fb68763cfbdf7b305141beea4589123c2a2ad81a18158b05f1b606407187fafadd82d158b10abe73153c24e657b4adc87601e8252ff838e6390128dee97deba44f2f8e213844b8395601f4ed044829ccd513c0be3dea44ffc160fd6737ef57b134298356f6941b1139def68e2267d23ae819c4635d115b26acfa9afc8ab20ce124571e8a7b88d72dbf269ab35e505c0221d689517cb96&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519852611Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0, Name: checkpoints/3904/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3904/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5d925c734a5a39dda769b46fdae9327e1c073c817c67836ce83eccee6ee9824e0cbd8e63727cab8c978158ea23267299009e436093ab6bf8c74ffa1947a8e6e768c8ae7c7934895336612782e5c484388dac0a5f3fce575e47c30fccc2b14721469826d00602af6bd50ae9057c215c3f678d93d14162d30089a21ba488e9a9d431c45d3f54a53ed5455764a81a8a041476cf50f085d5aeafb23ba1dfac497f0d156f7b1f6e863e68509acfca3fc780377c9a50386897c0ed3cb10e51f42f3a2b2d612caee71d32294f42b9d5e0f7d36db16e74b24fe29cbaf59ab103024a702bd5cb12d2caa581f52f7e65ec33ebf024c30d7c3bfd3d4df3d8b55bd9d19c5659&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519885473Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832, Name: checkpoints/832, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/832?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=15237e6e30e2212e82b2deae7a1fda0d2d82c561c35add0b5c3021f50e749494b6abddb1b6dc6146f8b341fee46ac3febe3e062979a9a1e4d96610e3aa35edc71f09af8ee66064325aabe1c9352a2d04a8d59863dfb2f57132bf2de09fab21495f3700f2f84e6e12c07355a8241601a5194e4b91ac71997e648c880b5e347c340bb8e2faa00a62c92b8eddfb9df7d84c36a11fb605b83f4ceb63bedd7f05484e2a05f8581444b0089db3637c5780f20de38c0c4324927a143fe2bbde7a9215e5d453d181c033370222e58336890b77a5dc2cf65f7f24c1c62dc55957d87c32e26827be9aee122d0d3f1c266cca76b7aaf530cb48b5eb17e26a7b00b931418b96&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519958314Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default, Name: checkpoints/4224/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4224/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=875c66456f023f0da00ff09cbb1f83d5e181a6ab9ab27d0d3d3351d26bd7b38f32fefddcbcc7ba2377cae9e0c01ed129c8eb6a77f2fc9dc41f7c8b4e6e20c3fb1fa4181ffbb79a9375ea03fa0f674e736a90967d4d73d8f9caa35ae440620fffc1e5ccc0173ecba174871a163f20bbf51f8d408fd3e47c6174670a0b5c200cbf4a6d062a4e5d834477e03f55266a55e9ec8683730233e4863df9871ee275f84547f04bbc8065cc0253521bbb01d5d683c8fff1dc3551d3f87e7f214558855f0d9ce02b6783cb3024cf14924bef84ab932feef92d170e475d429e128dfea960122d8b60a15e45e42dd3dd077b9ef07cb6db9a3a0b57b577aeb93edda94cb23448&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519881975Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0, Name: checkpoints/2240/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2240/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=877ff4af1fdf495519c9215890d392043cd06e5aaaa304b7f83d9ced35ecbafacc5f1ce65a80f9d3629993a8cf0eb91ff81f9b98674a6e4e15cd67cb548be489c9866ba9cbf8e734a8ad80c3f5363f9891409e5d8d02d72e8a4509c6fcf6123625867f7a755959a1410236dc70eec05a31e78bca3e1472259b686c15a9a9e336f840a6cf600d3104249a7e7ae47a3dc9cb248a0d1403fea9bdf5cc4b47aa9aae8825d8534bb10d46166cb207f7d80c7c32e95ced1fa83763d9f1bd7e8448bc35ac66d6100ba6a3c81e8ebf8374fa2efd968fa24ea8e11b960010438793cabff4fae3db6a2a126a3cb70098eccd207a1b9c50632a9050aea358fa5b94a074f1dd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.52005056Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0, Name: checkpoints/2944/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2944/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4ecb66866266623f7ac65e8351e92ea892817bb6566eb7f16f5c4fefa59fe067781d6485a8cc8057f940feb489b0b0c7ad343df33bf887126b20055771c61e3c074618a0aadb7985e89248490dc4645e81ca45c1eb573179c408eed9d57cfae8307eea819ed0e70340b04dd2edea68c402db59c833aa251c83251a2d5e5ca0a7ae1f30bc41ef359ed844fd5565f2b5592492d6deb72de654283de5c1a90c3db6de5e2c1a6e91760c6373eb610b38ff952e27246e7dfa98b2ec367caa21fa9dc7e23ff43d48ff1056f7dad13cccf3ae44ef81c62d1c846957cb46af40513cda3b78787aae786d7b92c29f43ba747a1e01a2cc34967bef02a920f86fc061d5e284&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520133278Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/array_metadatas, Name: checkpoints/3968/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3968/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9d3a6de0cf11d2b4328de1069071b2330ecf159232a137afdc32a0d9aa080db66d49b49a9eeeccab0bdc45a55a307d8d53e76218698f992e7b33d7ea8010d46224e4a44fa98b477ec68760ebf27b04983d9aff9102a0a8e57f74066a9614d3b7f4a3ffd537b8871a002e5edf41e3458ee89006eaac4dda41dc0a93fca289056acffa43977a09eed2724a690e0d3dc51e93765dfb08babbe1e86c6aa5f04f5d9c7d830915eee06c67cb26dd346b619d88a9e8d015befe9b7f6b016c0aa518cb9ec0731931bbee754499da879f7ccd672f42408680c5e77a48cc50c47d5ce0d22d92f95b347a6a43745c9648f4558a022f21d947405f904eddf504e62797ed94f0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520119114Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/d, Name: checkpoints/2432/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2432/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4a9b557fc7469ef574e720e50245ad4cec29bf5f4f3729fa7dcfebc45e886908bf19c4e9020abf2ddc55ba0e2abca7b1f5beaf11d9435a709e955c54a4b65cdc05d364d7671e1f65efeccd5e9d360316271b2d0f0f4e90f6d0c7725a2d9240b039b82e588e3f6c62486c6754534585743bccc3b48e54ba63b60811ef05fe5e7a982c28bce50ff2fd52c66e17f4ad8c15eeeb2829a06f1633ad9d01a852d741df78dfdacf2076a3668412fbaf5cc078410d41131f93769e52a05729545816c120a141848e8c54b8d17a61fe3b1a2aff02b00a6682db062195e1be9c9d30ecfdd4ab17e9db56a49cbf4eff11893324e79e2ba550e40cdf71b9a967cab36b9490a0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520163876Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d, Name: checkpoints/256/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/256/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=93c84ff06c20eb546a0826766664c8c3e7c5071e83dd677015760d5c8610337e621ee3cee5f065818141c988010c30f47d7b530acf4f4b7427f2ad7f42aa41a8f48862f0f2f008e12f5519340b6a6635082c2207b2f666b2009a061b58458fb26bc7efb2f63e263685bde130ed6055372a04ef293600b112cfa0fb79135ed121c88cb434945b869543f2b4483239b0d1b094f0dc63389fddff1818c221d57c599cecc58bb3aa9cff515fc45694023263ff275b83596bd172fcd3a3cfe7c38504dfc84a7a4c584dc944660f25e9e8e8810089f97c706d7edc5ab75f9a75690d6b84d5e4d95b20b65f122eb02eeccbe700a4f9cdaa1d242178a94c704e887e0695&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520158299Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default, Name: checkpoints/3200/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3200/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2901466392ae120ef533543b1e1178cc30c524fd23a19f5668a4cb7c91ea0c3a4e5fc6b2a74c6113230f722eb27db371b99046426c761eeb01c9ceba1d184e24f88c998b45402a34dbf73c57fc3c72d6f239f2ed797a38807bb38772740ad9e4820e5c14b259bdf485e65e7ccb343a56a1b3690e11eb6b2c5a4b614453dbf44be0111c8a44def4a137a01c04081c6bdb8dcfe8a4577fb9d4e233a984c6c60dc651949cb8521de07a23f0fa9ea7a002dc743a706df0544d1f000285dab5e28b35252c20d60870f70f778c257f698c93f592b5512f38b71c86c8df0637257eb82d26286bc1770fc3462af6cf205345436b778cac1ddd727bf6dadcdb2722e8ea2c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520050896Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/d, Name: checkpoints/6016/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6016/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3beca96c1d1f4539331c3d2983f00f025e5eba9e6db1414ddf785823d78fb09ecb73711444445945e7fd6f937f5b10a705153dc661992132ecd4fcf850eaff12b140f1cfa2c915ecd85806afe082bdd40d9e14f6069b7620fd8e6735cf9c1a4ef843cc2c05ca7c5e66b4e48161b98d2810a3ba63e490bbda476cc29e27f41ec2bca2f79dbe80b6c1fbfb57c36677a5c8fa5a18e1902add884bf2aaa79a5deb6deba1734e5fb9feb5ac7e19b70b089569b9bfd683d3ec67a3465c5b8893eb2d811d58058aeb67700e09c45f6aecc55afcd8c9765ea1ed217ab972865a968b2bc0ce3cb4094fdd3cff62103fdfd4a9077f1553866ab7e7428a74443403b6cfdd22&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520159558Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0, Name: checkpoints/2432/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2432/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0debaa329b2d282bd410dce04b7a40956011973c2473aeee64cce7d5d8ad240b3b6c6acf127fe36ff8d8e24719077fca07e07c1552fdfd2201d823b72b3f94803aeaf88cccbeee5a174c07d992255971f100de2346a60847ec86c5d75e25471695b1f5ac56f5b2bf9023e06c6df5bb118d8dcce09035d594b9607a9335f7188142b651f27a78402f6e4b2ebfd13c1eebc44dbeed0e176d72073b9eeb28ff2c47a501f6b5f16b87bd3aa81933c9778588c0889b3707b490b283f8eee1d789a95c2ef5cb0140c553ee42c15e06b713ed1561c78126bd2b3e07f8f2bd8fcebb7a1f52641be6f159febd8faf9df2dcb90f8bb70e71562678bd815037124ee67e48f2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520147381Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default, Name: checkpoints/640/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/640/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c25402294b84e8dc255183975194c166ee426bf7f5f7274c2c79fb2515b355a03aa1f3b899487efd1d91e30b0e99b77cbeaa63de8c44555e600ed59ce89374cdc22628a311d703d062ce885cfcb8d17fdc98456d4142cacdd5b75e6f88af273f9a21a2491c29b698f3bc190e9b9a58ee7ab365e71cb022dff6f36894ef5f88acd5643ceaace962ddfdf92b252e16495d317e373336f204b1861de08683a39efb41323ee5621328796b632ca9315ceb901fd6c84f7cd67f22bafc353198be83ccdea0ae8cb460f55b86e7c5ce68e736760d774529525a5d047440806f7ab87d8690e7c085454d84f29d94f820e2d75022e412aafc1626fbf0cf9f4fd30d493976&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520150304Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/array_metadatas, Name: checkpoints/4352/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4352/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0057d5e4a2e67071483ed7a0318f0f76f2200c6361853ea57378f5c91ff6917f8690d916bb1f6f70040fcd1657aae61d7f526a28e8836c67e9f6437f1b779811211b09d3ab9a05173d4d70a3b8f9784bc55746f14870b0f7d285f88cbb5d81397f60c174db2aca8fdecead94b25dd46aa2f98218ee65b4d292a5a9dcdc8e7d4ad8943b4a42cca7345242f5b546fdd1f5aa9cdf087d5f87284e7920cc78905eb4907628485607387b445322dc6e25e7c7ea9124ee7283bfed54569692576eea17f09d89638c78a1e054105a346913b32be687b44e0ba1bf0dc634cd3994510ba0f154a0871e84f01ba05a5ef7b235a281f4104405485311c90de1622ee4b32f56&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520191257Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0, Name: checkpoints/1280/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1280/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=822b4cd72c294a66f582d7bb88ae7fdc04f7ccab8c15f9aa64094d2a18f6b6d41ce856113122ed2da4ec319d96ca7d981b80da26e767b695ca6bc4dd19df02c2a11be7b2c8b46e479f22ad547c015240bc0737b14991aec6773b238ba18772cc1d43e7cdb3f98fcd2e8dfbe1f3160c87db24a87a5f4f988ed72cd4d1067d6d6e27d0c1bd7535afd64c669b8ac116973216c39fdd1ba2be4ddd7d946e921be480f3045e12e802287c683e30e5ebff571f2acf7c9f89344adc480aca2280a05bc895142b305a73d9e2e9b5f4ca8e0678d12fe37dc02bfaad62a954e8964276d2bc50f0e4ce6d744ef3cf8b39fc1b02b82be4224b266600e1fe3b53a980d97e06b8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.5202377Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/d, Name: checkpoints/1024/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1024/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=66e7533a18c7fa0d1eef6bd2b2fc7fbee81c000be8cf197a4884b938ba01c2040030a3784ece2f4e1bff9df85f21a38cb482fab4e2f905aeae17bc3c2ed5c6ff993713930d56294bb84ba8dd54f82887d6fc1dadec6e1a870ae28574c75431e1dbf4ac31540ae28fcd198e187607a500bf1d2038bf159c16d8a45a062c6acb858d0c782de121c58e1ec60fc7561bdb9e9662c0376b9196c1828a7e718d84e3949baf0a1186e6c73314a7674914d191a58c386d370b8b69b82303409b52a52312fc76d7d2e8abcde4ea747ea5f04619ad1d3cfbcde4ef34a42824e3babc517b1fe6fd003ff5472ebb5c027c0a4a99dcc450048272a742290e8e5a386da75fd78e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520245284Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/array_metadatas, Name: checkpoints/192/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/192/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3cb00fe35a19ea5984b33dd4d68ee3cbddb243c3afd352aa50c51d552d89e38c2483070d5cf1966887afa0b973cb201cdede646814396cf928154d1ad0f4662a9fe4bf7f73e0941673abb2bb047b0ad99d46f695ac1b62f11d0e9171f79a00a1e6ec8cde12dc6d369f486604dc0fb04acbdf36696b7ef4995c9c94876fcc28d755b15765fd68d0b1fb6de79bb4c02693c90d501e552da39d63bd7b5129253481b5fb0ed571a51b4724a51963f38732fa04b179c10416f665677a7eec858db4ef07a8789fdc27f9e1f0f642f258085c56a72325930da731a2f0d4d21e72aec5ab940e214a880099483342de060d442c42fe0553da1575209fe7fe2dceca710c46&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520242361Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default, Name: checkpoints/1024/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1024/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=60474308343f24297faa896dc6e2c9cfaa34961c3ab635ec6290005bd960d4723a6704e38330ff3dbf25e656c13578a745ac3f12434ae046d05c0828a304601d42e295a785e31892c822c341fdb536bb94a75c960a80b63e2be21de7187bb85779ad6250f998194405ccc43e5914b3f29030ed3796732e42d2d56b874b3072d38f1084d5fc9277f2a083aaf6a75b8263cabd56da5deea5abe1421c292a2ebad297d1a1d6a0f9fc4c115df89af401427b16e44969cfbaf0ee48d4d6d875b65891ef8c99fe7081507119b736d3ae7f7ac89986bc5e70e344d3efc2121b87ebce7629455113b14a1471b3fa0264178c12b19bc03af52bdf9e0017433b4054ed7ed9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520244369Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default, Name: checkpoints/2816/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2816/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0ed3fd666e51b75640c3921390b1d42b386f041c2846cd9bd6c86d6a2656be797f1428a6703e155cb2766dbf23215260a7d2aa35f35a7231c06ece9e001476c9b2828a6407d0f0394bcbfe7db4e61da3ca11ede5094186cffa125d9b98ef856e28a36325fb80cef10e79345af908366c7736919a2604795c2c39720bbb1b1eb3c5798c628f8e200b27be4dba2e17b141c897438f1e2ebe938aaaacccce56f2e59f4b0cc0b802cb213c3a23f760ea5a4efc6d917de0a7c38e58b25ce3f636c40a7d2645b6ce8af65aef590bae29bd24a4106ed6e433f889c808e8552243d37d5662378c980a6873b836927fe6c7f9a2d6710ade1b48167b5a21f4c612755687f5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.52025733Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/d, Name: checkpoints/5824/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5824/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=32dddde2b9daa763342fa72f30398adecb89d22ec18f8d8d8d31de8dbab900b2fe043539e40a94a39032cfe0ebd6ba5d8623aae9aef86ff789d61eefa682bcbaf342b84da99e6fef9d3f2b5f440a82e312c889b9f9f8e3e1d2e525749b294d775dad4380ec553d54022fc1a15d07f664a5b2ff8339e4a85534afe9c2aa64a54f72485d654045d0b4e573e2e77216aa97aa4d43890909595353230a5698c350bd63f9e149d8a9e9dd13cd7edc258ab2a624d8e3b34507c76dbf0add39361e0c9408ad9955fa26c9da45bc6508a48116b694e98d740c47af2970e2a1e3e0b28e36be29dd584806e94f65872fc558fbf82181bbab6a6bc43e0d93b95d7987265bf1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520244467Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d, Name: checkpoints/1600/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1600/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8583625501433c6133bca3d5eba7d813ccc9becded114f0e7ce939b3c6ed455c833ac1c6aba49450a02c6a18d695cf20b91c65f2c470faf5a6593eef0164db0dbed77ba03e1a036574541d4fe9a02b1babc4ca7ace375be2d93c0daae8b145dd9210c2d6c378f1d2bf0e39bd2711d771ee28165d597ecb09c5931226be5452b772c1a291b1f4d2162306c7502802f9918933af20122c5f88003fbaefea6010a879980a1672b6433dfea7ae287f40794722727d57bd464fcd527b7dd1d67d3e5c57a24c807cb97ba5caf05ebe61aa98f8a03eb068fdc3ef8aab1647f2be766719ef4d9a9f964cd2c7aed8f3ad5e4cd8f27aa1f924f52766a16f72bbe0759476db&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520326369Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/d, Name: checkpoints/5696/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5696/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2560ed2d3e598c1990b9fc175eca6a05cc2231d019c395668aa646f9ccd2eefb5a46164dc39f0baa8c40bd0deca6d108c7cbd6b74fe4c2dfe541a63b608c0b322f9051931817064140d50c3918ea0cb044565f0a33e407bffcfa96a01ec5770ae019181e0ab861a81ac16ad92b31bbda08c4000fd4aaaade42db6bf93267e3b1e9e3b23bae02c5e773ef186e9c196345a3040dbc0c79b04591b09f68756e97b897d27dbf7815aeee6d95e0a286b2d9cda8742dcfa5f4415899f15a1338feaecb920cddd38d9e856d166da40e37bf3c440db893fd7fe7e3e13030da9ba8b57b11b6103e0b996f358410739033b2b334f336f6b672a66e767ac2b0cca19edd3ffa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520320332Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d, Name: checkpoints/512/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/512/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3bec6b3355d329cad43391552d72e4c188a51a72873da4c79d5a8b46903b095edba035906e87269618608982b7a2dd17dd3820dc114de07829465b01c591d5023f5ac05f5808c8fc56c1b7b8966474810cf788cea1b8c89884c3a2d31798170ca1fc8b3388af774aa791a927ef5980b8f6df5f92de46bd79ec0ccb52df7553d1043913185ab17f6125e3c8f8ff2089d628bddedac343239e3f753a00099584f58bd0698760b6a68138942e20fcf207a71ec3dd9dcbe5d6249c463532e4e79e6a5c2a9771d7970493e5b1017f0df94db8e12f9d58dc950c4becbc22b53d7b0f0824adadca62217f5652af9410420d364a9803ecf00e9dc14cdd2cdd2871bacecd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520332634Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/array_metadatas, Name: checkpoints/5056/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5056/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3be6931bce055edec50b34ec98a0ffa4ac63aa606fd0844001259d52ef08f7f940aaf143e936b4f8356f715ccce7a6c704ff12aefb9fbe064aae141870476cc492c02b0ad042d25404614bf93c9e50aa9d271675a2f988545b8b1d0609788071cc559d836bc8bc4d3dc15046f5dda2c119294336d303fc079f0012a1791959f7da220ef3e39132494af446c61185b2ebd5055097ce12f8760d463b7a9180791e2ffc405ee0c21b5d16bb62db3fd12c801b0408382e662c95c241a626491badd0ac856c18079b49f2fae83ddb7657dd3b70bd01442a012b623906463b22e052122cca76adfbf594ee8cbb66f1ba29c52b87b55b3fd26fb3f4931f107a23007356&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520338753Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/d, Name: checkpoints/5568/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5568/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=be5d751cda836fac681af13b35026041003ea8e3bda9abfb1c617472eacdaa3fdd019cdca9701719f7d2b86a99503c07ba8e202a12ec1dd9d6ed2757ca9dc67c25634b98244a135ed7dd0bf49e5e6aa2b1d41d752ecd454a92148a9da796f4f11c32987106ff5e294291c96cd031e55993a7a61ce401953be6ad082a92c2bc47a9aae1f1b269116e10699228bf6ce3e101d9d15a5b0e1606c1af9a1aecc775ba7f4cb16d9c37a415a07f6050e14adf69b6d65ad1237a47e308127b52ed701b4d00f3b20514b61ed58d10db38ca4d293317edff0e1496483af6b8624c9574cf32fecd0d4e125525cd4db747eaf09179e79ed52107a8e6939e3341a7f35565f422&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520309967Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160, Name: checkpoints/4160, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4160?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=58b3d145c1245f5b6ef5e5a3396ecdf15f6affeb5ec6121261d11fcbdfbc903fd7fb22c849ebaaafab6c1c0bb589b56aa159aff612740f8198b5c5279c2daf1c22978f6ceaf16bda04897fd76aa3d1a831a58cd1652881f190efa6548415ca48045a440b536950b4064d6a7a58101fce44e470ac25c05de418c976fbe61d58df00dc1d7813e581bd2aba0f43913434ae050bf3aafa2f5cb58ac6b1f2964ead1079d9e74bac98be485b43ae32e302c0bff95f13326af03187c810bd214717c9fe4a2a20212395691af902d1b1ea65cbfe6d586379c0a808a8e2cad4f7faf08bcceb401051a6daa0589b679db3bb93432c1c5efac5b3669013d480dc026d420ccd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520343824Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/array_metadatas, Name: checkpoints/3136/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3136/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5a0af2e6d32f145ebbf3bb8ae56a73f386c15a7eb73c987af7d9ffb5bcf36127bf1c5614690cb760c0d824b1e5ed2d8d676b4ed400ceb95968d8b9fcf791c6aa511af2bd2e668b6b3003bb9289ae00494995ee3da0c8ae371f349c8acf235c24c94660d7e2aeb5e0c5bada1853a4ca5e7e36b62652fccf5cced91e811eb6436b8c2bfe178bd011f41598b369b616f2cd19c6723d88fd1d61303fa79ae22e81225c782110598edf4310c1e9af34c1527c7c13d6f365ef820bd63e359fd472129638aca1e60c81b8b442da4e7f45bf1fa9dcf55887b97e6e7129f014637e4a7e3e22c413e799733db6fb1b8bb6349766c08de70c6c42ec153b02025051e6d5c456&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520365167Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/array_metadatas, Name: checkpoints/4800/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4800/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9248a928722f2240b16b48e38f33fc0e3d1e2bd10115d5b1c8608aa08c8141016ae6fdcdc03dbbf03d2836806a84f1e80539a262c38775d40adb0ed030953bc7f2155f8e638e812bb113bcb0b9d95f519677debed81a1bce2566f3b40dce936276b80ac8355c4bf9b7bd7d695a13841274d3968582b9d3089436838f4dcb45091da1c5469a387f4b37c03138b1f6aa196b76c24decbdff4888e364a833f6b6d69be89c7ad391536075772b2acf3a207fbcf9e4a7e4d46faf98fe3414553f6174f4feea8b05489c391d204cf543d7f0a9ae2f5db2bb4ad7d89a475109d45f1dfff132ccb03660ecf2b2fd56955f62f52183e60d02dfbf43ac3b7cf75b500fef1d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520383838Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default, Name: checkpoints/1088/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1088/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2329f77c6ce9bcbc8a7be4ec9f25fb2d82605c5c72e3ef083d38dc5d036315b19583fc208a9fc42b2465d6e4bd298ff5c9e4a8ff62e4148fcb9b184708913882a438fac9d5e48ec6bb212b34f1dfe897a76bce9f3007ca8c39133769c16973b0abcc307211419f601a0522ecf652dbb8b55ddb32e666549ffc715dd6f9322b37336349c3d31fa8e97d3933225c2cd21ef6e559c665d314426d994693c864fae458dc0f1a0540fc9bd4ce8074580f31199469f86aad7d05f9ba3579b2136fc8944960c422b32b979d4e01c14e76afc57fcbfcf6b83d8939dec6f5a72414c59e26a86566f0f890c22fe9c4d19f4181860b5f089eb44e00e52f7c439e9f365d6bb4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520395539Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default, Name: checkpoints/256/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/256/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=72e6f79367279f213a0e338ee4124ca2ceb61801c3b9f93812b6527ac46b8caaaae731c8423b1d678cb6bd20c29dee5dd2d3dfaaf5d768a6e295865e5371d5e46946f84613a5fc6a50eee880066e166e3ec1b670fa4dc91b3b84e19b309eaad029500c33c2b0f319a4a21b597adca63739f9fa78f466e92201568875f7358ac36099fd94be163ef4a8a3ac21a8571a90c5942a1a97745a5c6830336441d9996a5726c0668cc840522d6f4f9a3294de34ca6727d1c968f7bf78b8a714a3da09cdbb4dcb310bb49588df5c15293358e74267a54dcf0e1e3ebc4625bc01c4ea84eda70bd9594d594f1e89dc1576e5c57c90194ac4d41313d04476d0fe19133e5548&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.5204106Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0, Name: checkpoints/2752/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2752/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a228235ce1d16ba74c0d84a46f8e4bc4d9517daf7ee81809f61f1f2ec889e55573828bcfb48d18017be84ed16756e9ed7524ec38b720891f1f3aa8bd9961a762fd769cfda5322d2266ddd4d0ca040b66ab3eaac06d5f8bd310ba2668664bb12c04282fe4d623fd34fcce825d33337877573374fc6c19eab4c16faf87ac087b788beb391891b68353bc1fbda11b2fede5afd85a35ffc910510f79c5a06316ecdb38d574b7315eb19d3f1ed200da0a60884e6694e3a05e9889345bd3d43a11dc27bcbe4b8d1a8560b1daede88c01d1e5da143be5b62b59c57d77e334a8f65d370ff7a08703638b9456d3639151ab8c0b2d1c98e4fa066aeb05e06b67b1ac541d88&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520475745Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/array_metadatas, Name: checkpoints/5184/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5184/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6bd59baafdabe4db37580684c9f27c512f0811affc9697a5f1fc36af8aa8ec01dc0ecd218c647396621442064d8bb25d14d8937986c8e84b31bd671dea9c76688a6d91b9c0ee8b37f385b0ebd6e8efabd1fefe3a3bd7ef9c8ff5beccfd8c2dc34597f5e0708d3559ac70409a73938deb8b5ec87c7dc3af638851e64e57f098b1ae5e4f97c95264ba9ca9711c051ac2c3b9b4a371bd05f2987c77a4aec655610ce5932e569530d4d8350534ef85829c1b80301f5b8607c7b6af07a756d93062d886bcb9252e04957cab6cbcf07dd350e010cd29fa6fed8d2d04765e30f7535dc01ed9079a27b13a1e404d20b6d46162dcd7a3d014e18887ee5faf24523291942b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520441638Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/d, Name: checkpoints/576/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/576/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7e01f479897ce5a83f8bdca723c5aed9c98c11c4a44c01b74a5a5cba2dbcf5eeb3a6fd145c16e38f9a5b9b8ba69a4548421725fad34beafc70d46bff8b4042227246510bb1f6a4da085aa2f157008a34c91c315b6553414baf60b3a473f6a3ea02036d37bb872b8f10c1c3f122eaaae935641e990cc0aafd029f3caefcfd84658ad17f19e5e3ae2ff9f81458b18d782204106b9ca2ee23ac3c7f85f5986587896ebd7de2e15fe7927fc2906837dc9906164b74e9b15a5c83ee59b8c3780c20e836e1c8a3d95d14551b5a8e4fc1fa8457d3339953c0094a755ed2f34c790a41d17482941f90f864fc99c2d0743d4c941f710b21b5774eebd14b974179075509a0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520484469Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default, Name: checkpoints/2624/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2624/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7ca77243e07f479ebbf5379ae72a7b5c5efb7848d392f8131b69d6ef065bd886a112972d13f695050c4f44386f9d6b0546ce1c063895eb7ffbf89aa0bdb05551bf03220ca8f88630912355931b0bbfcc6e1addc20379310fa0689d53a992dbdbd3e8bcdd6ec117368aa4068ce4508085a3dddc4e7e8df3860b26fbf9b7177165c75f76f97b74ce95ea14d352d79f6c7bd939de201fcfdfb67228fa6ae5c9d4b1e92363ffc390625094706d8978a8676aafe5a94ca2190f1f318a961924968f6bf23ed48ea761bc1d9d4f8199aaf4ffcc0359ddecf3ee25e7aca15eb2f14cb40fb37c520bfd5a99accce83680af89c88064a46dcc0b02df10f63699effcf31ec0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520475037Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default, Name: checkpoints/5504/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5504/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=058408dcb9d24b0d38cc5619398fc280fe2c2bfc8203c3e573d4a60816b0e3ad1b21c8664eebabf913f7b5b7ee4a10b9318c905fa26d505bc6cdfcc83cb929c46bff15be6a5c2dd53ad6c1bb6dfafaf3614bf7d8d0dca7f3cbac0b1a3d7472569ee5f2a320771447a9ed9e04332de8e245a26ab42e0265c6330a1ab9d00ab1552f6568163f602134f424fc3bc6ecaef50a012b6b30d346d3c63a40ffca6756b47671cbd7ac6aa89667a4b58ebdaa1bca63557f6c752a07eba31641e5e21b140c5208029758a94f229ca71c0abc46a83e5de813c6493834947f05c2ea8df6c136feb1afcd1e760a765a313dbcbf3bfce449df9a67a2f5cc7f126702418ca36981&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520524913Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920, Name: checkpoints/1920, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1920?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=27b801cf6285bb307b518f804688882af6e55b99647d6f6763b03842371fb4de786da82ddadb2efbb3d9be527ac4374b5f4c59de384a542a668ae69821d76bde1c257c4cbdce2124d1a287b6f7d4d5f624843c1ef8863e29908226918e1552d76649a5f930e57196a285ccd0cfbe3faba7918413d1041ec79a4dbe1bc3d35855430ea26c20053279458d7c36f5077b130e95be7946ab0fa654c38d60e0a751159327c8ea4bf9b4b306885669fa626d7b66f034b400716da46902f73ac2ca7804a2de32961023acc41493c8042cee2af37c115684e8b93bde4b69b6bb1ad9f71433b91f87bb6ca9208de5df7df12687fd032923d8e5f46c2391cb9f7a0fda1884&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520566402Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default, Name: checkpoints/1216/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1216/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=579a5ef8c99f385b49efd559fbe91d466a53822036add7525ff12a0eaecb95dbc45bf430f174fc012a7586ea141c58455a818b9f02948c5357533059882f87c72e2735cecd27e66b5a01e1b801d9e0ab51e6c7609b8f8ed129b22db45ad5a488f728bd8cc210a76bf5bdcac0bcc614e6bf1bcb3804b67f17d722a21b76d72c66e3fe171cf70b6b302af890b4aa7a62b550abe57752fb197463c16abce634adb059881fde34c002e90cc1bc84292df6264a209756dc53cc3d963a7a4c2e79d84a6ca7946758f36948b508ea952f44deac00572924d36d651c4c3550815ab1a44222ab5ca38b204f398023f59aec83b4c9aabbe30ce508c9ecdf72fb75b11cf4c3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520602184Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/d, Name: checkpoints/4736/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4736/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=262aeaace4c8072ac724817c1eff75deab76eea02328d21e3afa9ceaec718a1772330774542ab1b98112d418a9c26eb289538375ee640d62fbdccd1c600d54ab89a8b740bd0c865d260d072fd4f88c878b6ae5d20b7cd7f86ae69f1af99be3cbee73ea6fe8dad28c5e1a3f4b04e6674b2ab285e9d5c88b69c2c36d17d8068401a4097670f6efe98e729ab6eff257dc8408f123814bb2fc615f4581c3d96ef97b22158636f0ac51a075be4a64c066148cec7aed38a395900b8f2f1559a218e252ac388afe7b4c4fb9d47fbc67d0498830935fa8de9fa1daea784ad7902c2cd14cec651b84c5c16906e135f163f2aa958caf205be3195fa17352f4b7142c0c72af&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520614145Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d, Name: checkpoints/1280/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1280/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7a13b169ac1326b59c600d5acb3531c94bd047986e698470366366c650b42f9b563427045ef27e3fa2bd8b9e54b34312e96525e75a79254791c6965a12b24c58ad2857f08c76cb1633dcad740f9abdceabef380d7905281936e097e8d74f9b8156a6f0a3cdae8ea0d022d52963ef57e69bf188f04b02ca297351d1b2d407ef4cb08fbf5012fbb364e58d601b0428d2c662d78a7a2d534598c1619eaf0f87734e2ff62c3072815959f813094e965817b064dab78f471e5b7e0f52fabcaad2b74c855358c51eff358a7276ce91f7f6b8aa25d20b3e9f4b15c14d04ea021f6b079d6751b92bd6d576c7cd6004e755f9fc709678dfdbca3dfcf743145d22806a0e7b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520625559Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default, Name: checkpoints/5760/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5760/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=972a95744988ae3cf8d1d41c3f1fce77f9431d67f7d34f118a2d53fa9a553bbc26419d2c71537c39f100277103cca1f97a78851d947778b57eef94eac4ed1399b7a97037f5a1f5d5253ed22f24230e2d1f0ed697f5764df90e432e753f574f4a4316a3f67ef494928b45202a96831cbcda41a6912af1cd18ae7c34c591f554267e20b7c76df84f2aef7f7970898d06f190343797ccfdac79774614c333f1baae357fa5c0eda4b1704ca4d16d401afbe2ebdd75a3d808db1aa06d8ca1eabb62f503acbc06d7880bdde77483613891e40500cea19c593ee0e3e0806100b5cdd1c7278bb569ba882f9c72f4c29d714ba22afd942d5da959bf15d12539ebdb059980&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520590507Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/array_metadatas, Name: checkpoints/4992/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4992/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a37714c234e8f410880ed81f72a972346fd301a51523dd0f76e84b47045fc4db70a63aa8eef9938f2a62dc28a4569d2875f1fd6fbbe26c00970a5728621b73cf9bd26fc61db0915eb1b71db61d853137bffbb66d238fbca9656b09658c51ac0ff37f7f27c0db56e7ef637cbcb001ab54a76437a2b81833906c1981036ca5e9e2f695e900bd5081ff34ce7147760c9a66fd40d79312763f2c531c90403b1e31fb682b7398d2c420c0e09777d25298aa47a7292584473940cbc64d31264c60d8bc252c783927078bb5722ab0cb2e5ca34671f88716a092f16dc00b4f6b751387532cbcf7df7231dd012c9df94c25b8769ecaf1a21b8e45d1805d4e1c4167d98fef&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520614964Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default, Name: checkpoints/512/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/512/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=ada14c2696840b850e236c37ef46939bf09001b01602b7e2c726548cf4cc3fafdf9e2c967b9e1a9c9c2a6d76d1fb907b5405cdcecfce314806e6fd7411ad5f8fb90e97979703c9b1281c559185442f51a53bcd83b15a9cafe3a043b321339cc58dd74f0def504ce38b100143002d174ab4c7a49f5f663a518dfe838af46b98e25200caf78cccd6d47258ba7cc09cdc7b140c2a906a153b90cc92760658c3f15d1e2cea6184ad91bb72b8e17518414fda32dfd9491991ac8795a9d41123d4ff18118637dcbf4b3c73dc1fc8ac3d7e500b52da2967e050fcc3b0a776b874536e4bd6bf5b759c9e35c759b006da2f3f8a121c56f04ed56e327a7dea3395e76d0e51&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520682223Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/d, Name: checkpoints/5504/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5504/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=33adba019eeead41e5a704d0e79187591759f9f71a5a054b0a5aa22c93846d4dfb2ed49db1a7c1bcd602fab97ca945d362d14067a7646db312677f286d7ef81b53a77a53764b5df2a3392c6936d8fd2c0cfb12b3a8212ba09732d896f95a8771ec37905cd3835bd61563a1decd8c0908aec40a39e4c2cfb848db989f05c04871f22976f1beb06846dac6770852d9f25d4f7bd3ead7a87faf599c333ea0d744aa5c50515475410d3299ba60d266dfda86c99099425ba6b58b37ceea2fdd445b6915603fdd65d741a9cf3f85a620abaf52a2d4ad22396852a7de4950ef8a200af686e6318dbd6706cc8bbeb884eff1399073c6f0077b45ced518944ad3278e7794&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520678614Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/d, Name: checkpoints/2304/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2304/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8db0647c4d11000ef810969f5ca7876c63f3031092fed4bb20d36d047d3ebb5ffb013a199636215427bd6ea771710b544f1a38a4c9cbf536737e8c5eefa763d141312a5a038bc50555a81b4dde6847e2c7faefbcd15a964e2d7cc9f98d28c2304daab477ca90f544121c1f6e9888c065098e0353c700fdd6db70b479d9e0041cdad57e5d43655e6d091c312e4968b7fb57287db8024bd45d2a0053ca1c73c7ca8d709ecf5612084a1a2e27203747e5d9b13d410d8d12a9eeeb278dc9501825fd8172f714ab69f8014e1bf5255d86781ca7e56a4abb56166dce41e1fd06580e44af249d915ad48b0d3aa89cc8cab10826ff1a1518d55b8251b9acc67c70a65844&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520720366Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032, Name: checkpoints/4032, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4032?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c2435fab7036b125b6e7fe0e0e1d701fe4e1f3b94b2b4251e06f89a063757bf53aa1b1cc21678f13281b40a732c9bc01e36e0c559b5f2ec65d9ad04ff04975360001f85507200c8028f132da22ea2ca84fec37aeafe38f62afca78b20eabaa39cd30bdbb723355586bffda85777408a995b6570bbe11681e3e1a6e6d77d72212fabeda84c86dde919baa405737cbf2ca8db07f499dd6ce200d1644c72ede862b228072fe249aaeb4a8a6bcc54320263a05eb578e62ca9bff3f62bba911b6d1944bf8a40656c745653665349881896e8a07d884960e683a4e7b46087818226f829899ae3a17eac6ea44f8f9181e28dda22f04458cfc7f5940474a904c5d073976&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520731442Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416, Name: checkpoints/4416, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4416?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3b564b4e91cc3ecf7631f765201bfe5c9c97cca7867b7fc5104d6eb6bfee8ed1a0f2d538489ef08a858c24161e8ad9958f577cdc9dbe5fce782251a64313c3f1a6af14eaa47050f8b8abe2f522daf284d0c2b2bf9d17b55a48a948fbc2366f6f529562925e725c97f8d076af5d0b575e27e30a0ca6007e9629d1b099f25959372698b1049111ce697585afc43e33df76a20039ecaa50aa5b1af8a51008a4cb8f47e7a005cfc4118c2d1caaa09755d8e1b038d98e7b71d8ad5fa7f7d87b39c3c23f9210a10cd841ca9255490240c93ea3611caf44077e5fc8e520cffcbce7b011e06a90e8cd62fedb3f8f088ddf4b2499b33d341f310818d1921e5a858159a8e1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520737098Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/d, Name: checkpoints/2816/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2816/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8bc6978147274f9de72d29016d72de5de9420f65bfe5cdad5a00ac65d0292ae5eb3da6d05fc01b4590bd3f663ae3d11e8eaac0851ae375540fab302ba2251a369ec44f0c92195a6869be70a2bde1c1c7c50c64e04c6257a3d331ccd0f100c495e5acd6d69e09fb092d2e87fdac105c0ba59cc92cf8f069cfa14349578f005fb76f5a3bc0fb350bb016faa65557899b438dbcc5ecb743646936edc849feec45806dae20ab71e53b9b222383779dd994f2d864d2ce66c18faf8715cfd060db281813512588f21035aa6010fcd6807c0ceb65b0d34f9726866600e7c8eb452a8ddeb9ad9bb97d6d7b9c23177678d95b6ca834cb89648dae490aa03cc8bc0910b3bb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.52077998Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/array_metadatas, Name: checkpoints/2816/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2816/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=ad703a022ad0e8b5bb591dad8a97e9d9682de395d1e975ce519d8df8b79735f889c386db94f9a24dd69189c44afde4c28c2c6425f55c065bff1de79c58f5421f57086fd1ad41005a2bf84b7d160bf43d76500c9f0fe860ebb756d3f3168cef639ad4a45a972218d4690a9aeec81ca9ce5f3e8a4f842b8bdf1b47217d70339cf9cc57b4387e8ca9e1f796ac0fc792734fd99c99a5554bdcf6dab149f01bbd6f2d82b0edf8fa6ea27909410e68f0840e325759b575f32c8ae63989cb9bc71f4cb295220413b58efafb5847acb8e174c5e3137bc4c7a1a3a9699417c5fc3e312cf5a2d5d642c32c2f73fe38295f5269e91b7cfedb04876f53c8db375c139a5bbb38&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520795958Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default, Name: checkpoints/2304/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2304/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b29bcd992712c5f8cbcbc347ccfcf9ef8c78b3d0936a1b41dc06c98dc3615369928637da9bf014258fc96711085e00b1654e9040a4e11b5cfb11697b04d02f26900b5b7890f5a9f753c6bb0f05430c4ea53fb4a3c4ef4f1de19e859dcac6f956b8bd71b32e338e64999770811b9b5ac6a9cf0599b8a3c224313e61bd2f625744f3f905d157e662db60ea60d20cac8f132d99871ed99fcdd2b15f6a445d5f2eb81a45e23a3250bb0866f8826f46ffa8221ff951b48ca850906973977b37abbead008776b204a5119d4b4d59dd77707dd82c388d9acb18d51a0c164f5e18cd1fffc62a0e3a3179bd528b6414ce38c72ef618c6e20f2a9ecf31daff8b7c9650804d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520747226Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648, Name: checkpoints/3648, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3648?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7f7368919955a9b714d77f70480daadbac2e1e4215856669553b8278ae10ba797b8163de298ebb6e35c9cb241a73d297f559b05bb2ebb76add88f13950918840c542754eb08a65ddf6e4e00f8416492f0babf9f79b5632dcccad70af0401e2f4380ee32486382ae96279c0bd3e1ee85bbee27160dc204d612cf1c5f92f43312cbb90da42819c9c666897c9812c03110abf40d57f56ba48b8bb483b68969dcdfb608d66b9f8ca63aed44e26df4402839e7e78b66c361de611cc4e33bac87085a8a6b6f1c8ed81d708f20cee15ad42e11752aa7639d94cf9a94afff074e5622a55bff13a88ffb678f9ee10d07ede77249ea6cd018f013ff9f17b9f7bff9a0a3a16&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520808134Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d, Name: checkpoints/4032/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4032/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4d5615e40d2490d9df8e8cf33c4e31b8eb2efc8d853d77e486b9fefa4bc55a0f9d645d043e35d6700ab83a27129bef04023923daa4e391697f37138304f877349408d6ce68c7018c6eb1b7fa5e84f6a04d91479b823aeb2d2a25547afb1bd1d43f18516a5dbc9d1d3e0afb86c21b823d3adfd2175aa3363d4014fe8df2aee020cdd566ece060cafac8570a8ccf1ea46c85cb243b694496f250f70b76cae293fb1865cecf158028b1000098d13de3f15b4c9e5f3e1f756222aa9a8e4eb30425a1122a1ec8b1d3e2c9995ca0b6976f252baea2340a2bae0d9b60ea13d5b3d6b04d8bd7eb349d772ce299c61666469dc126c37cc678bfbceb69f2fe99276a9a7f8e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520845718Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480, Name: checkpoints/4480, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4480?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8adbd5ae61a81090584326fdd0f778da4487829224943f9daa734aabd53d59de7fd7f674e46abac6a076a4b1c2aa0ea3ed0811c8a64b6c5bcc91bb9fed5a5d5285ecdb50db9a375e417789a8db8269843383c73e778dfda077a940014d5a4390438c33f834d821b170b8821dcd09eb8bc3347a24131c57b733f96e5c94fa989a8c47b8d7df9e1f325fe037e99efc8c0e19f5000c3a93e27bb02f7298be7c1aeedadeb154fcc8e954965847096ecc547fa17eee051f91c1a737ca9dc409df385895cf7464133adc2b9d7e2991081588f12c20ff0fc17f79841754cc4749c2bd4beefb9068bbdd6295eacdc01c3ca21d5c0acf1b20608f2d25c41ba1f1e32f414e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520841812Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0, Name: checkpoints/576/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/576/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8cd065f0675b901c4dad8092c2fe9aaa3b9fdf0d6cd80ba7d4d075fae4f70c6c480cee3af96730ab1dc8cabe6102155acd1dd57a954a97d1af0d9218eb8ff43e4494b2528a67371757082ac66300d722838d02e05daf3a78dcdf7599403a6f31423f767faec56fddb1fe76f58812c4c87447f423e87d47702e07ef094165d5714c3729b2ea04993a95415e4835ef615d54e1b3aa82eca5ff4934246c51df412fb345f34f758c1f2532e5485d75c637189400958fd9fda328b6811a894460c4216e970166ebd070a9767b8ecb3dd9c9750c604e2f27efb5e47c5c6fd0604d07b874c42857d0c79a32de1f31ff3614e1a11d539b1cb5e81c8e2920af51256920a7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520841799Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d, Name: checkpoints/3904/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3904/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5daaf01dc34c41933ce73aced3ef86d87c8317109218879162fe8b34cbc26405b7dae0f9ccf706067577e646196e35b7e38dafdd730c0a13bf99515fe8febcea4b4b4024d85fc185a184ae6d7f87b66e834401d6eca597f716761ee2650dbad7f7086ec7b68471ddfe98db506557f34500b92d9c35a202e0ec3836074de4ed0c6d4604152fbc9c83f6d242fa01cd8aaf304fdc4393a498c7563f6750fa9ec9669220928c12b8286a3e046d2822b86e31d166fc21a14acc798f9be7ed97706b0a9bf4ac0dbf966d3dcbc51e2a7a9a7e24eec70a70581706faf1cb49a03d2ed68c35f220322e7102ab727ae46740d70a0adaa3cd58acdea63d0683784ac9b38adc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520864924Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0, Name: checkpoints/5824/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5824/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0c122c380c4e323532014f04457aa0c7d9aa9642dfa7112de61d7516793dcf039e336948faf64011ea0468b4493d65f41379a229968278f66eace366f80aaa9cd61136e24a23d2653eb9115728e4245a01d25662d9102420b5c1a81ee6540f79e705175d636f562c9e2989e2f9e54ca1d98e5a72558d8c39f45e98911a245a14b5251a09a6e9a04b86d901259c3e18630de0a4cd0389b678fe3e99912d356b343d59968eb96863a75f18e479ce5a49a262bcc6517f271992eb09110caf62b8c58f2be49dd9b1232454e6ea49bb36729f4533e0dba34dfa4e35ec026b52b9cff15d1573d9f6459045b2951959c47cebaeade09e0d8a844a1511669b82e02ba235&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520855613Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624, Name: checkpoints/2624, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2624?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=932dd37a26ffb9f817200770bf0817f89b85123331255278b8f736b4b67d5ffd7dcd5f50aac176bfd510f49f403cdec463df913082aee9c0d76fcddc9d5e1e5e5e0415daf054ea9c09613a7796d2321e52fefc75fb5cdeef38b859d5e027859e1562d8d6488322a69a395f7673b320751e7654a1c0c37ceae689b30ae98afa21e11f66f8b42992870835afc9791b71587a0d11678af60100617c3b6d2ee24dd4483d4f04e5c8aad3e18d7139ed5f69b0780a374fb950afc3dbfcf60d080c3bc6dff9444fd91c31b8d7c1081a73f353c290445497c77d47e431c62f4abec5071f6be7ab8d1037c6a79504fb19d655e94764b69a79066b074917e9afb16fa46df9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520889589Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/d, Name: checkpoints/3456/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3456/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=184b430b83ef989dd7ee624fd0419fe65ccb99deed9a34e1bc773b63e3a624299e23486393e69abdc2fa8ac31b90ffb350481fe9247d68a6fe880d043e96c2c409a546c4171a30a2dd7ea49462cc925aeb32e5bc7e3764238228e48efac9504e18e9b0db6a9ea55dff80f512744707cfaab6b4ed9acabc41a45bf7168798998ce145dd45741f9cd09cb45a493c6ac6c2d4c185db7fbb69637cb9a824cda18308b47dce9fe240962c8146a70c74ec730d98a408a9e41a31ac7e74ecdbcb1430616544f2e3ae6e8ad8e5d176448184a92c5b68129d28e96c4c06b02bb9a38bd9df1c49ce415d26f0757c6f70cc0a8cfe7716f8a4cc52e36dd6e7aacf3f89aeb34e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.52091135Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/d, Name: checkpoints/320/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/320/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4b77194bd8adc5fa622df90ecacbd7fe1650e6636381e9e2e8d1412c6d1cd2083eed86d1fb52c8406428863c203acd02f19d82dc58341f2df2ce45efb69dafb06bb3f45a82e449c3051ac014da6680659ba6e97ddc0fa4ddb6e1aa7c004677fdf9d6cbbb7cd6a2be39014e6806d319d3e6296059f35b100bb2dc6509d10983a49d13a1ae00abe9d50e4884df34a4666bb5e0fc6dddbfb49567c82626a44e7fa30b09144b9f918606d9d45ffbfd65ba1f4a8087d58d37a59e5dab20b8735d637b3b1b98dd01a444f46dabfd30f7b588ea14a3c1929754305fd988a42cd13cefd9c6214c0f328da9b5ebdcd0aa354ade211bb2724bc9f391e95022676907247ec3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.52090865Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d, Name: checkpoints/320/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/320/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c192a3bf2ad10f504fe8ab60c4f0845e2853ef8c1e0318123496d0b21ba511e6bb439e96d1cc3c5e52d94bac07469530b19e2c195817c910a1bc49d17faeea961c711e5a24a10534f2b39d89f37330685b9e57e7ec5ee1ae58fa0bc54badf4bfd0ade47fa5562dbf14e01409b8b7ee37a9217dab6197f3c069f684ceb9408be0a4ab244f6772dade2a8c6c2f6e7edcaedfa5db6b007fd6ccef46c20a249957a8f7dacfb6123ca726dfe25d6c569bb29525e40aff27b4cc2846f43a20bcee69d8cb6e8309ca0a9567052e469e07d16926b5d2296733e6e2ca38d796e43ddfc0598f2365495c50c3107b90afba99ece465b92068097d9a36ca05f6c9e72a5364da&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521040385Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/array_metadatas, Name: checkpoints/1088/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1088/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8e02cd47a20fa64eed94d6179f9bac39033e0b0294a93d42f3de273bab7c3e9d58b6fc131c55f336139f89d7e8132f622e2cd5d541d6f0aac713955200fcf88eddc5d8c1867f81e52a7a2121d16074090b0a63d2081700b31bda7ac6b8917f88e5397d1625326b3c7a482042c15cca8612497b60d9540eb666184215c2ca599e0a1c1e83d8aeea4e6c3400510599c31bce2d8a777990ef8a2392f68b5f31ff66c3c849fdf782ee5493d04bc5fd01a26742b0fbd9e805da8b86fa27e3ae87569156f6036e4a5582a577663579c0e1e10c427510cbc92f4689a88a7fcca8f1dcacd34687e7812a2b71e934487b0e5eb18d3419f900dc84d91dd317ba6acd495146&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.52105705Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/d, Name: checkpoints/1984/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1984/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=64911880bf91f48120e117bd10a0b713f64785a58f4856ee8b8d37df1e5648df9f9528b168e61f64103dc977f49ad1a5dddad1e517740553c5dd167e13a0012a254d2f920edb7e55d9a60bd9717f2491a7d716ce9755a0070382d11a0e8d909197c3fd5d1089dc752d03eab668c9cc0d25edde1009e7355612ff3aa6d01bf3f216bea41e417e9b8524e525bc96310f745e6e672b83ddad2c25b01525fbd9a8421e8563b82c7293d53723753de21e0395e0d65dfab1f02738331560ebed3a7e772109ac6d100e39abe50660cb66dcb7b1da7aee9641a279d3ee8f37cdced62be53c7fcf260141a400a8bbfd9db1e3f9fd6a9e9957bca4aeeb0f4c702cbd55531f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521057157Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default, Name: checkpoints/5056/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5056/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9000967cc1f75287e09d02a6a68bb9fee3145a3666a6bc49b4fa3e34c6be87c202367ea0219f3747fe7bc02602a0f005e0121a78107d05b100c9c108e06b8edcb025f77ec32d9cf8b14d0f8c1ee456ef5cc8f9e2c33b0efa33dc885bd96d270c40c0f71a65141c66c8dddeb3ff1b0308bf0927c5210101def84eab47eeddadd915d6cbb2d9a375762447440e48f50075b4c70d1efe40b6ccf78592fd37300d5be6f14794470b30e41990923e29e2707a40358eb3844b376bd2afaec12247ae403ec8795e45de61bfb016e5f271dbee7eb151c1a56c1309523121c6b647ba089c37d8edf5910a032317a75e34d2f28f5d9592a1e7c6e0790506afabb144c6a163&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.52106235Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints, Name: checkpoints, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=342a68c7f47f8c89058706a38bbac5c4adcfa27bff728c659df2a3c812915e14e392dbc348f9db06d52da274099230d5577cb988b6e38b5bb2c89fd82c576cef91772e87d3c702fec773540b55119f08750c775798e2ab64c2708d77e58202655515c906270d5debd5287561fe571a46d8278635fca96224e005b761dc17fc6c8b84bea60bb5add0b6fc5bccde9837aba98d6f83ff8b76086fe53a0c28fad3e291e933b37fea356885b78509bfbac191a72a023f7eab02b567b6dfac8df813a29b3fb1c1d484c09c745e04b2de02536c3c06cea15d41bb97bac71b055cb16cc2b85faff3abb191b8f522f0d3662c5aca2c1e661654873ea65f07ed8cdac6b2f4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521099695Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0, Name: checkpoints/256/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/256/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b212ae5c219d033c1c0c34332d88f36308dd0b96da56e73ac6b929f7c6d27706af4516cd8c79540d9db2f386e7031691d58dc9e39bb767c60a06d121f8ff9be3503526e08304fdd2969f48ae222f08d8b30f4ecb0d7bc7d8502a8be7dc81e879db99ac65b1a6e4de463d3677317c5201b0cd7ceb81ccf369799f44ed2a862ad04ea45158e94bcef7471229154891806a0c17b4abf968969558405a07d16ab37e110da5a96314d881cbd4510fa81a1f4764935ea95eeeb29af6a0b4adadbf4d47b4e2cdacef067e7de0aae4614cc749b395e132054461baf3e2fd897a771b6367ca314357dff816cfa7866514fb54e6c57b7dde1117f9574cda02ad817cb593a2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.52109653Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0, Name: checkpoints/1792/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1792/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=15ddf6cb77e8f06e684132114305c3432ce62879b2fc95b94a8ceea810cff52bb0356ccd047f4595500ad9fb29ffdc8d572d37975bdd778c306de974db9d68c2043c460905e759c077f39f914827b695c99db7c21180f219330c4101a6ebb331b5fcafd7c8f6749482485f9066641e0079a52f8086d1fd41bb9289f62da659c2469585870db51a10ceda0e6ac694023140ecb6605d8abb9daf0f5139fc515c90b2dcbea813fc3fbb733da2b0a9fc29102c81eb92dd015324eacd4ea400b055901a12bc5bc68feb9fc4810cecb4caf02dcde464e8e2eab46deca66a31e93562bbe4ffc819fbe582e9dfc9a6cad3b422f4b234a6d8b52aa964d4f8c2ecb3918e19&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521153963Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536, Name: checkpoints/1536, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1536?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6fcd8097b9d34d562ff292c23387e3c8c122de839d171484024e2bb6a8aa782bbae8c02b62d3c833c2bbb740eba3bac219fb648c1d389081bee3ca44939bf3ac6293c22e87cb61dd2c951ff6e57b3e296ece96482d6ec6a3e4cbef7b720c018422d6d48a9120058b849916fc79b2390ddd837022970940b195eb67225abffdd64471a664f70a8a38f6ef1e446ff7e2f238080cdd4615a6239bdee35c5578d6d69a3cfcc526c4aaaacce2f01e3c9a13fe01af5269b7bf1dc944716440e3556dda62cd415b2dafea3891060f99783caffb465b11da30b4e6097de181f3a3762c7bf5ca3b3330a8f35306e59ecc1b221ae793b6c2a5f2a68b8032562de22647c58f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521160568Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0, Name: checkpoints/5120/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5120/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b82c25ff053fe564b0857936767fc358bd34a5729b4a762123962d57bbdf49e55557de3c69135a37074480c9e844ac7d15a3ea2833032250fa1cbb0bda9bb60bfa6f088ba8fd7f5f2de76ead08e0315fe5ea1a7f587d11f44c7f4e8e8e3df0cff059969579277e7933df4aff37b7722e2a14daeb1dfb60a3a55fd0300797848b43940197a6442e85494546c06eff19f98179d38fdce6b51bba3cfa15c71262d685364ed11c1a64c67d8fdabc1b60829b01a2aeaa0d282b932ebdba50e88ba567191c69414ce1c9c94a79cb1eef7688fa564704e4e3988d1ef5d744af69e1df446cfb43097ea07514cebaba86474a797a03ae7c6f1ab0900395516183c89c06d5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521202302Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0, Name: checkpoints/6016/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6016/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=08b5cbc9df8a475db556376b7e164ff430ece8afcb90172cfbf760e34d24b9181494cefc0b1937c964c01dbb1a9ac81d61f19f0613ad0880703aa43e6d7b35128d873644bb1ea664ba79a4f54122c8413d8f5c1ca997eedb05516c98a5dd13e2fc2b4128870f9ecfc3152e91cbcb0c8529f98a312a1752204b472817bf946e26b013f3f9a5835bb1c2e4aa54ddf7c5c3236553aed2a036b5f35dfd21516fe824a82f219ed25ecdf8f71962c2cf242e5f3c8da9c212eb20a59f16fe0f2e44779e553b64a23ed54c64f6eebe9f08657ee877cf69f8fa2ff2b7e8f38bb7fac48567111a367000620c3309df206deb3626aff2b5b355e33e956d3880e0f366e91c68&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521223126Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default, Name: checkpoints/4864/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4864/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=392a7f86eba30841d06bc5ebde146323f9a6b0a819a8ff0a3510f30646161e133bdb0274f2cf8c7311b370851bb341b9bc5c2b25702aa5d5f9ff022b96343a88fcc53e5d8f963521fab9f39c44f008657b244a92f8da0dd2868dd0b93dc6c19528371850713ea197dd11a61e20a2ceb6fe925e2fa587a39beefbc15757d6343a1a453f654577979336c4536c0c1a94d1e20674980564df6ddf7b5b004db5cc4d035ba432c249e15ff645b4c9dcc14ae5e04ca0320a03ad94fe42522809b71d4160da88367793d9f9d486039c33f7027e2df8be6ebbb2772f7ebb13287fdab501fef453abcf15cccb043b86ffb5eb3b0d77f7baccdb29329e06a98ced3e0e17f3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521232665Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0, Name: checkpoints/2688/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2688/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9b4ad49d1add7fe22b73f2178ba3b9f54530f30fa0de084ce308fae5b8efea3727db21c8026d32cbe9a8651135fb029dbeca17032b88f64a0fbc207687c859ea6e50fa14aa4b9a7f29c9e79ed92d5d09fecb94f828a5cb55fdec4f65dd7c4f69e58878fb3fbdf586ee0dcd80bc05d2cbe6281da065f97935c4ce295566377348e5ae355d825799a4457d3679582546b997aabd88e75760693de81c269e356f7ef1c5872139815f1b3c99f1b42ff011a3ebdfbf5a845433e2456774646086777ff0e3db97d8fd021116aa0e84fc8dfc2bb4eb153c4e2c34301804b9920b88ba81235c246d84e116f1d4b3c715ac2bf660da03b2d5a76e1316b748c70019fb8619&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521241516Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0, Name: checkpoints/4864/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4864/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=37ef823afaf29730d0a2d327e4036a78440eb3852736d03446e136899214543858c394799137cba1d8f5b229fb274be18412548374720287eee12c21386bc94017f1c8b3509a4bea9c72ff4d2e1bef1e29aea85ba912b254f2449bd2ae7e007c05c054a8c547ca453af152eb94f7011902ddcd0b74f39269250c587982bfe75e61f6718a2fcae54ab1712c4f81ccb5650076fb6918557bfa828227966a7269026acc195cd36f2804625690f5faa95222dbcb97496f747d28a60a0920e95ae8e4462622d4c9155d83323fb171201dd26c6f12e866fef8eb9ee154f9eef393750e4aa36a6f2ee3607dcce70f5db155daf6cd9478977b7424827f89f142456da030&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521239124Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d, Name: checkpoints/4608/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4608/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0348a41918a261b167e3d22d99826b7736d5a686d81b062a97f52b6cb1fd70199f8b9ae27b85a23ebf9fe326c9370438d156f7399a856d72d4db3af86ea9df95df54f71f5a1684735606cc2c7f63a86ac0c8554dc12e50ca1889effa68424dcdc4dd5a502aebb641c126c76e6ca4fb73905f546309b17eeb7abe3ddd8fb2570bff8b2069d27af3794d914ba1c3656b3d610532d6514a3ac336f6dc8db0b376a0a0055166831614b23a5af9c1f5eb764498db12dd1de972dc7adde74b7703e32c213b3e6c8de0570a79f53a4a89638def91c74e606f06784cfb7d87602dedb9c0aefc8e29cf55438ec68b72808bcced263585bc8c5b2c0d828cba757fee711ba2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521279125Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/d, Name: checkpoints/5632/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5632/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a383513219a0e3ef9d76c37b71ff2f4d06b2446b4b1af424efe614dd88e6dfd2c9bf2cb92cfc972b2663a044f7ba696657d83b378371a561b4f1f712558d6c45b388355b49fa0c02153837e370162a86273bd7ce10d8eb14a0af09972f4a04a2ee0d93689335abd0b5777df085e2209c72cfd9b6e0494e58576257961eb6f45a32c979a91e090f9bc92d054c31513ff36c2b372e9d6d9002b53fab7bd4f0f9ed39b6cf735392a2d3e4d88dbcf1d93975914c784004df0e82938c9cda4d048d553784ec9bbb059eb18192f93ce6e2767c6642cad2711f264c63492d21c64a3e6b53559fd8ff187f1db3959a498ffd23dacbcba97b4fba2d08a74a60c36990b59c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521289639Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/array_metadatas, Name: checkpoints/1408/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1408/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6b1827b797b487e327860c3f984d23e85803d5dbc86d185f38f47387b7f2e53f9b6a1d69eb1bf6aaa8889736e050bf867adf684fe58b06b565b3ff390914dcbad6066888b0098b52fbd0dbf338d616aa1163b235aaae65b44ca7fc3735d0c76d05b84ee2af978a7d66be65da00219fdfef0e1469adf84ce63cc36c566dd870617babc0f2e569429062fe04f5196d3ac49943fd0fef5f562fc71c00af05e70000626908d61a55dcea6278bd5b40eaecbd70c72dd9ccae0eed74bda9f30d9eab2e35b6e76f81b35ea6bf4979192ef76551edb4e861e0b2bc942f78f26c579b5a5bc1d3c81bba3581890e9a04c96c9c4e97bd098f2b4ffef06959a4925fc9e2a57a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521299212Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112, Name: checkpoints/2112, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2112?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=accdace88deccc966dae7fbb8ba6b47c34666c62e2d6c8aacca06c32cb7baa98f13c6b5bc7afda514ca6e16c48ee3a191a30cdb5208a9ae5d194b81a11620811d0f83f43dae223415afbd319ee4382b4f839f94e30ebe2dcf156bc7886c21ef8b925f25f228d07470332f5c5cf9d7c75991ed26d4b588a5aad7817365c781f602aa846276a1cb4b208d509d369d2505c793b964836782c38c60646fe1d0461dd2d2ab611e697590c6b15c776c68c9090e15cf5ef67356be8645791691b6f906173584af11c1536b604f2ebd4fb9586a0e7675f056e0496763dc4e53ba75dc239942777bb6c81d9740c6a3f818fab4ebb84fc320993de90558397320505bce59b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521315633Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/d, Name: checkpoints/4416/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4416/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=232020313867d8f6676c851bf0b9824b5ac4a2f59ddd0b36f0d0b8e8a3cd30d8a02be5d41de9e666161de6351801fa70e5ef8ae129202753a590cd649d98922ea916e1c5afb64657e0b33af9c129fde4cc7e5bf4244a974d7b753992d199b482f86b95c051cd6e075e1f3129e118b66643ba36d9ca336f3649860487b59c08a6888e16fa0c1019dafa606076f282a7aeb7672e702e5b901ffec1ebe1610fc843543d35ea4a5ea655914020dc7f68cb3800b5c26271a2daa36b38dfb99e8dff05840d22e6d5fb896ed9deb005f4bf05869d0f0c59a7bd4a2c332ce1fa41a25259dd6f1c7e98b6e1845d7619c1bd444a279f7e746ef10a8ba4fd37dd41feb747b0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521333232Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d, Name: checkpoints/3008/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3008/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8e124e5fafef4ad49a87fdf46fb048dfe8c3fe8c8d60a27b2cac843f21d61864df1c9ea19d1cef75afdfc43dafd4b96d4825802575755fd0daf52fcda4f968edfbfe661638ff6ebcdd4cb82ff78dd9fa551b6136c058a46dde02a354f107e1ea5dbdc79c917cf458266d4b3485671cad695d4e54364047213e62cac134dd73a50ef717150248336872c12803007f5e5592ebdd852095943683c59dad5ebcc011b4361c38f77e9614149388bc19eacfcb4ab7bfb4e969c63ddd53236bd57d45b09a1994709536e02e240c0777bec38669a0ad21aa74b055f4db58f500f17dcd49a4b1e5d79dc0d2193bc9dad1ca32818591db3eeda6535192aa0c6d6ec8dbcdec&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521320125Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default, Name: checkpoints/1792/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1792/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=52424820a3ff8185aa35ddd5af5ab483840fcc8d7ef690820bdeffa12525a6d1a174b1475b2b7671b2e6ff372b9009a531d7c662cf692f67135726734bcd527c00df49626f165f934fb5e521e18b080f29c0919b487879df6a77d8fb1b6cbc7b7cd4f733aaefb3e864383b0aa6ea937a767b6f687b17cac6132412414316cbd7dfceafbbcf92a8dc652c2cc91d8aa8c9b47bcccd4d7f6cbe0a9f6515c7bedc28cb269e2400f98813efc9c5b141969799ecacedcf96d3f0931bc65d146854c22cdc8e1d77041ecfb0d5910f0047c067bfd2307c6a66f2943150d0df22559b5873b5342d99149a6e8129dff799aaf94018a3cf389aad66105563a68b425b86d11f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521342586Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328, Name: checkpoints/3328, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3328?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4bb5a54112c0cc49049ddbf04af36de0596fac9f45325a629e3d001ca0759c8fe91f9ecb2dd9ddd26c4be9b68f46fbabeeeabb27ac8a50c70fbc00b97e2dc737b5834354aba2b9ccf0e468e966db15bfc59457031bbed990affc30c0ebf8b0dea8c52dae6f743d4e19764840cf7cffd8054b05729229c74a104f946a34bf863051ff5cdaf4020d3ac0a7dbce7aba53d41fc2d500795ba59f6ed11d200fdb1d9f8ce9b785ea7bb9ccc3a60637735ccb10cb87f23f2454e7e8166054103c04ad9508994c0a461a3d8623975ee1ceac1f9be6504e21d4e4144e10f12a8af93ec77c03c2ce234bca775f6fcd728f3d8fdd553ffd701b9232836322624e586894d402&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521346013Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/d, Name: checkpoints/4160/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4160/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=36e1b54dca53b1e0d2d00191d2674d7001dd178f66fadad50073159575ff2ffee0b2976b19f74564405cd2278e66ae05b06d17294a6fd92434186ee4a42dfaf2294053ec293c19b417daa9c69a84cb42dddf489ecd62d0f4084c4fa6cdc274a0524b64af30b1e1549c72efb47f77865d09aa18ea680a8051e757962575038fe053322c5a9f1407558fe42811e43277816ac0e68f80793e2b5a53d76548dff85ddc55b1ab2dfdb965bdf85ce5499ae43487849dc98de5d62b6e8bff3c0a294774425141c0d1816fd1a3b4443578220a850d080060d7d85e45face0ddc7bf9eba68fca1ea6ea3e2c0deecbb7d5d7480fe47205264240acb2f7e5dc69e9cdc3b280&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.52135632Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/array_metadatas, Name: checkpoints/2048/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2048/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b305aaff5142b0b581f6db11af809b6fe3c18191d0c7ee0f3c380be5c1d88a8ffd55625301adc2a47379c6cb998cd7688d856579c520d703e629925d597f9c78185a78690d4f64d50867d7132748139ee3acaac136584a1766425c174e0a83a7f744db014e6b01d091deb7d5b23f3aa1a4c974c6207dbbbc7c08c35254211bf93ce8c54e528c67bb25621ffe3a9be127cea737ac6e26fb776cf9a78e5c44d2e2155b5361f82475afc909981e3aae53ee1533f8e84ed23642b431692f57b6bdc7903c131f2203e0dac603a34a6b67ab8b6d239fd7843a834a3b242d45d1632b7d1656cbfc85dc99c48625705078f5ed55acde9bb82edb649f1a40aa1576c8f183&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521382895Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0, Name: checkpoints/128/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/128/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0701a33e706da0b0d49dc733759f0f32a177d6d50d807fffd48248f70daec4e98a3670341023909ccedab89dbad7776c5e56ebb7c6f08656d7e108db0b81831caca73f837e5f5b0075925d6e53cc1b5b2f3e796c8df52b67cde6cdc759d7e5524f8360f6665ec4c943ceff5417dd3aa8c3d9a568e17e5b8e871f7fae4e155ea9ba87a28274462ae6a98e604ea683b6285f3b5dcdcc46e29586d839096a23a7e7ae9648d18606974ecf8be2e38b33f3cdfee78bc6d0d9cc023d8b424da223701ce33be5a47bcbb2e0bdacc005302a2e371f8cd53de5d8406aaaf85ef09b3b6412bbee6432198b07aa064212901ed0ec93e52da58f5073c493763fe4a155a6f26b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521400992Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default, Name: checkpoints/1408/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1408/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9cb2ecd40792b957f2aa2daac37c1064e2874a31f8248694bb057a7196b7528c1e356faf7a09148dbf808f675963413f2ad1875add787622c3ae05ce966975be86e93e9bc2534692583fcc30c0eb5c1e3626acda091ea8dcec594426e8937f96c20ac37db33db24b89b2b6a30da1a078cda13d5ce4280f99b37779808b82cab58814f8b40d720e0c35dec18909808f849856c9a02e58ee4c1e24ba1edca81f184d47325f3e010a3b9ff444b1b9762b6a308a2828490f055d649f1c810f865c172008b12de72d9572ec795c1bdfd4db4d4e9c80d32a4c8e65943a98303a097d2b1450b278dc2eecab786ccd3699c02eebe87b81bd457b013e5614741563520944&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521437432Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0, Name: checkpoints/4544/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4544/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0bcf5eb4546c367259b4062345ad1f854f4c778293ff3da98bfcc2ec503380997d1b4e32fb7e86bb005a2263d699392f07cbb467fab4d0d89353bdd976195b383d86f72ede92020a67540628faf876114dee6921d3d20f95bac1322830790dfe499e2563e17c6def88b0b1f431f861a6202c962ac0c6002089425400fae96497d16d2b3affabafdcb77b0c8aa4cf4b8465280e5b2ab1351b6da6ad9949aa8d6c555b754fdca918cd1d573dcccbae6604015c6a40d9d3b555838dd450c8fb1389ea9322de77c47d545cafb359666b5bdd9e3c7ad3afbaf5215b0ab17c0d8ce319362dfae74a1403982a7819c71d2201938d93a677739793b135234e2046dbb28f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521428721Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/array_metadatas, Name: checkpoints/960/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/960/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6b2650df8dfdb0e5c451c9697ecc7702c97b31f936d047aaccfcbc7c3e112a9d836f689d3a206009d8e02c441d443ca928d37a9061e5418b36f5d7ebf9d1ad87937b826dda13771b7a7ddbd6889a7452a4560e7ceb7eaaf4039929b3cbdc887e701a951cf23a1f42d1a95524c0655582eeb9944ab5c1cf72dbfdc486666be8e95404966019c10895d803935580d72e1ac269250b845aa60a9ba445bd47ea96847ef7284bb6f04cef3d2d7c058ce53512a3e0a1502027d627a609dfb3103eb671bd791732100c3233d221d8a63d8092bd49866da0ef43b8cf6b93d9ac3a608fa4fcc1949b41db4047b9a52d3566b0ffeb6b63bb0b23e3612faab61928afa5c200&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521440993Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904, Name: checkpoints/3904, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3904?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=94057bad465db9c4ebe13a2f49ee5d4878f0af7047f1dd65db9c42d42f3327510bc2963e0d1834c4b1145ab45dad449fb5ec3feb0b33009da093f88faf13f49b9c0da03e1a848913119345826c412dd137ab1eae458083ecc7a9a6f19355e76ea8cd683590bc51d572af0bf4dfabe8896d082889c5d0132a9f49d8464cd8b87afdec32252ec79c469d5f8851aca5cf97e12d9ddad0a2f7a4b432fa0af5678348640dbaab45fd9c683b988968acee343b914c7b11dee76e92abd5e98a6ab625dfe2c2a583774e3d04cd303b7c0e62c7ad007f88a6106e96ca150fb9a2dcb5140cffb653788b83112c470644b2702012cbd4d29c0e75fb8efc08169a8b14271cc7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521484504Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696, Name: checkpoints/5696, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5696?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=92573f90c5034b6638f58b74d8f158df6d5650dfa4da9cc7102c7a008566db83bb46fa6500781a3e73aa5ab41a08831acffc9b61f044131f00ad42380b6cadf512ddb6e5ef356d8d8a14db250969112a4098080fd7b1abf389922d14c7a941c64de1a3bcd09a5dc5c4c1aa0233bb616c91a36d7320bb7f031e1f7a6512f0dd4009aa34d210b98cf5ec9e0c43cc2e01b03c721e2252b47205533984777b0f48d5294f4e829433f8084f60805d38db39cc08d83e0dc7cefcac0b3524e79abe3b022b0b7d0ca213eef59e151c2e8b444b0ac6e8d26f76bf52fef9dcaa33de491c41d8919aeb9954fa16306c4caa1676c8a2f5b104eda1059fbedbe4d2cd26dc040f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521507978Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default, Name: checkpoints/4480/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4480/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=30e330effeca8481ac413664710bed3b284b4497ec1503bb3e033399488c086a53d11b511f213d110d482736218218e31f10fef01d499d3ac2a25e623fe9ca532d4b9e5810467a9fe0388ea850cd6a04f3f388849fb62a5321ad989aa2e90e1b96cd636eb8a178655d0d1bb361de10748d1a97e06614079c923ac0fdfe7f502c800513f1bcf11eef7b2148a2b5c9465e762aba0bca3e3e813a799e8d9e40b02aa94853a4e735f305e04bff2fefec39fb19f8a473708ff581f7cc23c5f16ff14760599c960ea04cea6f0b1688fa59eba7b047247d9d46ed89392cc2a278495118831092f3b9134b7ac35d8a3ce28ff518abe1a43b7f6c1b9153e8d1eba58d9074&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521521046Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d, Name: checkpoints/5312/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5312/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3dbf864a499c11ec3ba766b7d0d74f84922c397c9f85113ff4150ad29c88a1577fd3dee5e8b76a71445b599afb5b38afbac27feda74fc2b62d7ef354295f5eaedff4c7cf1dd3eb0fce3be7a0194f32a4ad7abfc789198379b1c5d254293e9b196c507afb1fcc51fe16c408758a7e20d5d7ef1f7975124d5b9f29acc09db0e024ef4a719eca54e2b955f33823f938ee256a2a6fb285e4476e777682e4759c03facfff81910a7dab3cc5dff38efafcc8bb7ff0c00866439c4524cefadfa1f43a47ae99376c10d0171b9c10ee05f84ad6cd59ad6b96f9cd4926f591e4e38ae4011e3041ba223f956ad10854681af85cf47ad08d10eb96e31341fbdefdf3f19b3b80&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48323033Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/array_metadatas, Name: checkpoints/3328/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3328/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=75739b942e7038d900743c95257b4d1426629339e228972f94bfb105944646e3fa6a32b16ec3cbcf9d07c36fe6fefa83280d48ee138edda480d3bc4c69226f6055606bb96393f058a1d7c84a2ffadeb36b6aea499721876d62bfbc3569cd541bbf529226261d3fd6a5b438fc4136825ac573cbf88743e6cb7eec6f96dfcff5935e4204f4d39cb790506dc7c77731d555836e3fc03d9fa9f36b7711e97825dd3f98cafe3303c8bc15cfa10b164bd953f9e27aeabe3663b16cdf608002981c73e7f5b25c115179da33093b0e2a6a05f1645fd420f1a3402ee129fe5ef28f7a7b536247ac3a55752a23be8e11cbff56cd5e09449a304d033d276efef8068198f876&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521528507Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d, Name: checkpoints/1344/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1344/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b8525db92537d74ca7531fe9b989419a44c44690a6190d26d385ab4f7405325db67a6b3044a2193c4556388e094a25e2d87122e8ee32cfbe218d8cf21415867a322cce1d52436e8520df53d1f6edf0393740a2dc298313ef508b5746f13eaffebf73d06d64207e53359ae67d56e32054b47c74fb22b3f490252ba9284ab82c1559f8711807036476e22a08cc7304e174b54e704f1f0d279dd7722007ab5734572cefdd05d0a0e154dccf1e2eff42d735d2583bf7446f1c2c83195987322e9cea0708e78b9b94f36457af6f7a4961fbbb1dffbfcb652af1c6004fdb95d916824c68c458d972e9d0859dd98c2652b7f2266c2317ea15f0b40eb46eeefa945bb46f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.63463426Z","level":"INFO","msg":"handler: operation stats","stats":{"operations":[{"desc":"uploading wandb-summary.json","runtime_seconds":0.371770495,"progress":"6.7KB/6.7KB"},{"desc":"uploading output.log","runtime_seconds":0.356208437,"progress":"1.0MB/1.1MB"}],"total_operations":2}}
+{"time":"2025-08-07T18:58:51.79900197Z","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
+{"time":"2025-08-07T18:58:52.084306801Z","level":"INFO","msg":"stream: closing","id":"wp_1.0_alpha_0"}
+{"time":"2025-08-07T18:58:52.084338051Z","level":"INFO","msg":"handler: closed","stream_id":"wp_1.0_alpha_0"}
+{"time":"2025-08-07T18:58:52.084356454Z","level":"INFO","msg":"writer: Close: closed","stream_id":"wp_1.0_alpha_0"}
+{"time":"2025-08-07T18:58:52.084481708Z","level":"INFO","msg":"sender: closed","stream_id":"wp_1.0_alpha_0"}
+{"time":"2025-08-07T18:58:52.084769871Z","level":"INFO","msg":"stream: closed","id":"wp_1.0_alpha_0"}
diff --git a/wall-prob-tree-13x13/debug.log b/wall-prob-tree-13x13/debug.log
new file mode 100644
index 0000000000000000000000000000000000000000..74f255b27bfd4816631b5212ce70288ef3c4c859
--- /dev/null
+++ b/wall-prob-tree-13x13/debug.log
@@ -0,0 +1,28 @@
+2025-08-07 18:02:11,436 INFO    MainThread:3869025 [wandb_setup.py:_flush():80] Current SDK version is 0.21.0
+2025-08-07 18:02:11,436 INFO    MainThread:3869025 [wandb_setup.py:_flush():80] Configure stats pid to 3869025
+2025-08-07 18:02:11,436 INFO    MainThread:3869025 [wandb_setup.py:_flush():80] Loading settings from /root/.config/wandb/settings
+2025-08-07 18:02:11,436 INFO    MainThread:3869025 [wandb_setup.py:_flush():80] Loading settings from /workspace/timaeus/projects/rl/wandb/settings
+2025-08-07 18:02:11,436 INFO    MainThread:3869025 [wandb_setup.py:_flush():80] Loading settings from environment variables
+2025-08-07 18:02:11,437 INFO    MainThread:3869025 [wandb_init.py:setup_run_log_directory():703] Logging user logs to /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/logs/debug.log
+2025-08-07 18:02:11,437 INFO    MainThread:3869025 [wandb_init.py:setup_run_log_directory():704] Logging internal logs to /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/logs/debug-internal.log
+2025-08-07 18:02:11,437 INFO    MainThread:3869025 [wandb_init.py:init():830] calling init triggers
+2025-08-07 18:02:11,437 INFO    MainThread:3869025 [wandb_init.py:init():835] wandb.init called with sweep_config: {}
+config: {'env_size': 13, 'env_layout': 'tree', 'wall_prob': 1.0, 'env_corner_size': 1, 'env_terminate_after_corner': False, 'obs_level_of_detail': 0, 'img_level_of_detail': 1, 'env_penalize_time': False, 'net_cnn_type': 'large', 'net_rnn_type': 'ff', 'net_width': 256, 'ued': 'plr', 'prob_shift': 0.0, 'num_train_levels': 2048, 'plr_buffer_size': 4096, 'plr_temperature': 0.1, 'plr_staleness_coeff': 0.1, 'plr_prob_replay': 0.5, 'plr_regret_estimator': 'maxmc-actor', 'plr_robust': False, 'num_mutate_steps': 12, 'prob_mutate_shift': 0.0, 'chain_mutate': True, 'mutate_cheese': True, 'train_proxy_critic': False, 'plr_proxy_shaping': False, 'proxy_name': 'proxy_corner', 'plr_proxy_shaping_coeff': 0.5, 'clipping': False, 'ppo_lr': 5e-05, 'ppo_gamma': 0.999, 'ppo_clip_eps': 0.1, 'ppo_gae_lambda': 0.95, 'ppo_entropy_coeff': 0.001, 'ppo_critic_coeff': 0.5, 'ppo_proxy_critic_coeff': 0.5, 'ppo_max_grad_norm': 0.5, 'ppo_lr_annealing': False, 'num_minibatches_per_epoch': 4, 'num_epochs_per_cycle': 5, 'num_total_env_steps': 200000000, 'num_env_steps_per_cycle': 128, 'num_parallel_envs': 256, 'console_log': True, 'log_gifs': True, 'log_imgs': True, 'log_hists': False, 'num_cycles_per_log': 32, 'num_cycles_per_eval': 32, 'num_cycles_per_gifs': 1024, 'num_cycles_per_big_eval': 1024, 'use_fixed_eval_levels': False, 'evals_num_env_steps': 512, 'evals_num_levels': 256, 'gif_grid_width': 16, 'level_splayer': 'mouse', 'checkpointing': True, 'keep_all_checkpoints': True, 'max_num_checkpoints': 1, 'num_cycles_per_checkpoint': 64, 'seed': 42, '_wandb': {}}
+2025-08-07 18:02:11,437 INFO    MainThread:3869025 [wandb_init.py:init():871] starting backend
+2025-08-07 18:02:11,650 INFO    MainThread:3869025 [wandb_init.py:init():874] sending inform_init request
+2025-08-07 18:02:11,656 INFO    MainThread:3869025 [wandb_init.py:init():882] backend started and connected
+2025-08-07 18:02:11,660 INFO    MainThread:3869025 [wandb_init.py:init():953] updated telemetry
+2025-08-07 18:02:11,669 INFO    MainThread:3869025 [wandb_init.py:init():977] communicating run to backend with 90.0 second timeout
+2025-08-07 18:02:12,236 INFO    MainThread:3869025 [wandb_init.py:init():1029] starting run threads in backend
+2025-08-07 18:02:12,575 INFO    MainThread:3869025 [wandb_run.py:_console_start():2458] atexit reg
+2025-08-07 18:02:12,575 INFO    MainThread:3869025 [wandb_run.py:_redirect():2306] redirect: wrap_raw
+2025-08-07 18:02:12,575 INFO    MainThread:3869025 [wandb_run.py:_redirect():2375] Wrapping output streams.
+2025-08-07 18:02:12,575 INFO    MainThread:3869025 [wandb_run.py:_redirect():2398] Redirects installed.
+2025-08-07 18:02:12,578 INFO    MainThread:3869025 [wandb_init.py:init():1075] run started, returning control to user process
+2025-08-07 18:58:49,629 INFO    MainThread:3869025 [wandb_run.py:_finish():2224] finishing run devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0
+2025-08-07 18:58:49,630 INFO    MainThread:3869025 [wandb_run.py:_atexit_cleanup():2423] got exitcode: 0
+2025-08-07 18:58:49,631 INFO    MainThread:3869025 [wandb_run.py:_restore():2405] restore
+2025-08-07 18:58:49,631 INFO    MainThread:3869025 [wandb_run.py:_restore():2411] restore done
+2025-08-07 18:58:52,061 INFO    MainThread:3869025 [wandb_run.py:_footer_history_summary_info():3903] rendering history
+2025-08-07 18:58:52,076 INFO    MainThread:3869025 [wandb_run.py:_footer_history_summary_info():3935] rendering summary
+2025-08-07 18:58:52,083 INFO    MainThread:3869025 [wandb_run.py:_footer_sync_info():3864] logging synced files
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..33499c868f5caf17bd102e8be61a05491dc8953c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754579215036600070, "commit_timestamp_nsecs": 1754579215175892946, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/d/be4b55bc5ae4c3a796799d281420123c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/d/be4b55bc5ae4c3a796799d281420123c
new file mode 100644
index 0000000000000000000000000000000000000000..0ad9d68267f48d424606a8dc82078aaf39242379
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/d/be4b55bc5ae4c3a796799d281420123c differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1d010ea1dd79e49de0758a26a91262e69b700ff7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/0ebf82e3d5dfce72a6c4a2118ee8451d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/0ebf82e3d5dfce72a6c4a2118ee8451d
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/0ebf82e3d5dfce72a6c4a2118ee8451d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/14065f5f7e608cd567d4b276f5b9cebf b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/14065f5f7e608cd567d4b276f5b9cebf
new file mode 100644
index 0000000000000000000000000000000000000000..5ad71912f79cc11434a82967867300a43f126e2c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/14065f5f7e608cd567d4b276f5b9cebf
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f6d1d8e8d7594a83d04e7405ff8485c5ecd437070c6aa413833dcae52731513c
+size 252114
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/63debc889d91fc62cf4853bdbd30107f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/63debc889d91fc62cf4853bdbd30107f
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/63debc889d91fc62cf4853bdbd30107f differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/aa9f6b4f8103559231db4a9d6e075637 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/aa9f6b4f8103559231db4a9d6e075637
new file mode 100644
index 0000000000000000000000000000000000000000..50ffddcfe853c2a94d902e820410b54bccbf8ed6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/aa9f6b4f8103559231db4a9d6e075637
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9c307d3a62b645e80da9e96fa084d4f62d3d857ad3905d82ca485753db9d59d0
+size 330386
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/c3fa4628eb352e5e16eb72654572e80f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/c3fa4628eb352e5e16eb72654572e80f
new file mode 100644
index 0000000000000000000000000000000000000000..a4b005040916aaea15beba2861fef0e11508f7f6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/c3fa4628eb352e5e16eb72654572e80f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e97ccb88fdb00863592dc265a5a9266237296373a1efa8294debaa2b92cf5374
+size 122665
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/cbc3fb6e656138c13a854d072e37b3e4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/cbc3fb6e656138c13a854d072e37b3e4
new file mode 100644
index 0000000000000000000000000000000000000000..a0f57957f39fdfa377a26ef07772b5e5c484ec3b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/cbc3fb6e656138c13a854d072e37b3e4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e59a1f050515736c26287716af489a0bbd645092
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..27de82c2af424c9dea56ad58e32b4282949d3dd6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754579814437199379, "commit_timestamp_nsecs": 1754579814621894121, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/d/9f987405d525163af31281885402e56f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/d/9f987405d525163af31281885402e56f
new file mode 100644
index 0000000000000000000000000000000000000000..caba5c845c3a0d20fd14bb62efb58aadaf21cc20
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/d/9f987405d525163af31281885402e56f differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e6aad67d1cbe68c0c92f55719450257e0bacae6b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/240b58bede71d1eb2aa826f900cc5d69 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/240b58bede71d1eb2aa826f900cc5d69
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/240b58bede71d1eb2aa826f900cc5d69 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/34249cedf3b3adde04124d415145194a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/34249cedf3b3adde04124d415145194a
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/34249cedf3b3adde04124d415145194a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/98a8c28b6e60870c3686098b14af3534 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/98a8c28b6e60870c3686098b14af3534
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/98a8c28b6e60870c3686098b14af3534 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/9a4dffa36085bbf7e523a43f926ff88c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/9a4dffa36085bbf7e523a43f926ff88c
new file mode 100644
index 0000000000000000000000000000000000000000..fb8a43bf569a130f504ed122f4c24215a77e9b3c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/9a4dffa36085bbf7e523a43f926ff88c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d6c99d2f0a13326c4255e2aea0b0bcff13c591879bf0fae1982d46421404da0d
+size 261708
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/a6a3ae1d6fc2453df7f3c981b1dd58ce b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/a6a3ae1d6fc2453df7f3c981b1dd58ce
new file mode 100644
index 0000000000000000000000000000000000000000..0d0a6ad593b69b77a41fd03abb129e90f173923d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/a6a3ae1d6fc2453df7f3c981b1dd58ce
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1f4c4e9f2caf26bc6ec519434c3829f917bd6f30baaaea2f24095c895bb62908
+size 439173
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/c5ffd9f8b2f900f1442d0e1c1c3baffe b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/c5ffd9f8b2f900f1442d0e1c1c3baffe
new file mode 100644
index 0000000000000000000000000000000000000000..85f35e7991cac7b6fc244bb2f277438119f4e90b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/c5ffd9f8b2f900f1442d0e1c1c3baffe differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/ce72fcb943bfba2d4c290d97ea5c3b6d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/ce72fcb943bfba2d4c290d97ea5c3b6d
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/ce72fcb943bfba2d4c290d97ea5c3b6d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/e9bfbc687b961fe1d3b467a07f3eb873 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/e9bfbc687b961fe1d3b467a07f3eb873
new file mode 100644
index 0000000000000000000000000000000000000000..ba0bc4f0bfcdc1b34bb1dda11a42d3dca2425543
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/e9bfbc687b961fe1d3b467a07f3eb873 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c169dc0add6cfeba91b65d687d0c566489eff624
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..fb6ba1aa3e549506e00d8e2c818ca03aeda376d2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754579848832453239, "commit_timestamp_nsecs": 1754579848965540441, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/d/86741095c5eca89b9e87b9f199227b76 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/d/86741095c5eca89b9e87b9f199227b76
new file mode 100644
index 0000000000000000000000000000000000000000..2b6ca440db37c7a7dbe49e1457c227862f164904
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/d/86741095c5eca89b9e87b9f199227b76 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f5096700c30a966bb3a920d336cfa10b7aaa37fc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/104eb64abaa0780b984efb23ffc01faa b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/104eb64abaa0780b984efb23ffc01faa
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/104eb64abaa0780b984efb23ffc01faa differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/19897cc70c3e579a4701bc6bedd3cd30 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/19897cc70c3e579a4701bc6bedd3cd30
new file mode 100644
index 0000000000000000000000000000000000000000..ce044d9d1047840ed8d384293dba341f06b036ca
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/19897cc70c3e579a4701bc6bedd3cd30
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:20364a4ddb009fb02e29b9fddc888cdcaf58f12b168101f46888d223c7098ff5
+size 379443
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/19d7954c4fc6d10bb10543ed6b9ca050 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/19d7954c4fc6d10bb10543ed6b9ca050
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/19d7954c4fc6d10bb10543ed6b9ca050 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/5ce17b0db9c323acfe5943e387973b2f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/5ce17b0db9c323acfe5943e387973b2f
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/5ce17b0db9c323acfe5943e387973b2f differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/bf85e76fde4bbd102db0d4c6fdd6a9cf b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/bf85e76fde4bbd102db0d4c6fdd6a9cf
new file mode 100644
index 0000000000000000000000000000000000000000..672bb6f9048bd81707d4155519e167d60612365d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/bf85e76fde4bbd102db0d4c6fdd6a9cf differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/ec20cd05d23e0d82ae89a7fdccacf860 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/ec20cd05d23e0d82ae89a7fdccacf860
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/ec20cd05d23e0d82ae89a7fdccacf860 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/f73893b3800ac13b8528776462c2ae26 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/f73893b3800ac13b8528776462c2ae26
new file mode 100644
index 0000000000000000000000000000000000000000..99d88252a9dd199b262d61eefad8720a2e23e3b9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/f73893b3800ac13b8528776462c2ae26
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:37fab05d04240367b22d6f34f7918dc471b8a02f6f7705a8bbe6b138a2a6aa0d
+size 287373
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/fe9ee65c7176fe9f0f16713c96d8f1da b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/fe9ee65c7176fe9f0f16713c96d8f1da
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/fe9ee65c7176fe9f0f16713c96d8f1da differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..faacfc1c9760495270e9338aeeceabd95b08440e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..2f831e406005405c2c8159f85954e48cec017589
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754579883058944004, "commit_timestamp_nsecs": 1754579883230950856, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/d/380d7793a8bdcd3462d916d7c8af2e68 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/d/380d7793a8bdcd3462d916d7c8af2e68
new file mode 100644
index 0000000000000000000000000000000000000000..5942b0c4eaa6f0f0f046a6407dcc2ecb7e497414
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/d/380d7793a8bdcd3462d916d7c8af2e68 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..cbccc9d60a65cb08f7c71c709c953243618cf574
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/2118a11b11e7606bf962398b0a72c5d8 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/2118a11b11e7606bf962398b0a72c5d8
new file mode 100644
index 0000000000000000000000000000000000000000..9a65e669770db91c2e93a3794aeda99f6882987a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/2118a11b11e7606bf962398b0a72c5d8
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e893ef3899c4f9d22c1e5048b6a858e7f3a7ca518fe25a6802852122135f9bb7
+size 255121
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/2b5a6e379eed95e06c285ea9cb075783 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/2b5a6e379eed95e06c285ea9cb075783
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/2b5a6e379eed95e06c285ea9cb075783 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/334dd77d8c389bf87668c319e829c1aa b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/334dd77d8c389bf87668c319e829c1aa
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/334dd77d8c389bf87668c319e829c1aa differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/6aff9b64f2e5f3c07a85ea28b9884ac5 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/6aff9b64f2e5f3c07a85ea28b9884ac5
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/6aff9b64f2e5f3c07a85ea28b9884ac5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/a576c7d2d5bc2e11831bb056632d7208 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/a576c7d2d5bc2e11831bb056632d7208
new file mode 100644
index 0000000000000000000000000000000000000000..ceb604835011711ddba38735e5a76eae1ba79485
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/a576c7d2d5bc2e11831bb056632d7208
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4b481cb6fe36013769040ae92cc27b3f22327358612fce5e19c3dc93f09a043d
+size 303319
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/c6dd4a3a80e3b7ff69e701acc7a025bf b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/c6dd4a3a80e3b7ff69e701acc7a025bf
new file mode 100644
index 0000000000000000000000000000000000000000..f74f6c498d12db80e0b7f371618b7ae3f88b0665
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/c6dd4a3a80e3b7ff69e701acc7a025bf differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/d7642041ab8645de8ee4f92df130161f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/d7642041ab8645de8ee4f92df130161f
new file mode 100644
index 0000000000000000000000000000000000000000..48781eff184230e8ea2364a3317604593ba02295
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/d7642041ab8645de8ee4f92df130161f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ac5e4c6b883e09d1d157297dc1049e75fd4346535485863b05dfa6387b832e8e
+size 106315
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/e1b733222094b0ba868d5fa0ba49382a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/e1b733222094b0ba868d5fa0ba49382a
new file mode 100644
index 0000000000000000000000000000000000000000..646bd352ae9ac409fad1c5ac1e50e6adbee00fe3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/e1b733222094b0ba868d5fa0ba49382a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/e3a89fd2ec20aee27d9bc92049be9aac b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/e3a89fd2ec20aee27d9bc92049be9aac
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/e3a89fd2ec20aee27d9bc92049be9aac differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..6909539d2f4918eabb8a33d3cbb399d14723e5f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..ecb228acb6901c05fe48e2ce775af767d98c389c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754579917883779784, "commit_timestamp_nsecs": 1754579918041131197, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/d/56df66be4dcc9f5676da93ddcc6a027b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/d/56df66be4dcc9f5676da93ddcc6a027b
new file mode 100644
index 0000000000000000000000000000000000000000..bed2f886f7a0f41f246f76f9034b6546aa44cbe9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/d/56df66be4dcc9f5676da93ddcc6a027b differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..2e4123e2b1c9655e192004db464f1f9d7138896c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/032e3b53cb46ce3dccc7981bec58c9e3 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/032e3b53cb46ce3dccc7981bec58c9e3
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/032e3b53cb46ce3dccc7981bec58c9e3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/0f5f6db246530443cc03a9150a681fa2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/0f5f6db246530443cc03a9150a681fa2
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/0f5f6db246530443cc03a9150a681fa2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/24e38a1b5d97cf9d6d9970a4d3d99472 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/24e38a1b5d97cf9d6d9970a4d3d99472
new file mode 100644
index 0000000000000000000000000000000000000000..c448baef44363acf56b91cc229826664683618ee
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/24e38a1b5d97cf9d6d9970a4d3d99472
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:40c2ed603eefb0209884cb0f49a938e2f142ba180b7368a1ae048db4082a81b6
+size 442744
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/2dfe31636dbf2a8e51fdea4e29f339ed b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/2dfe31636dbf2a8e51fdea4e29f339ed
new file mode 100644
index 0000000000000000000000000000000000000000..dfbfe1a61305bab6f716f83c77874d9e3176999d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/2dfe31636dbf2a8e51fdea4e29f339ed differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/2f4cb6327def353fc2355370b2bc687c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/2f4cb6327def353fc2355370b2bc687c
new file mode 100644
index 0000000000000000000000000000000000000000..0b0cdbc17f53528fdf44a4faf91e8fdfa7564689
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/2f4cb6327def353fc2355370b2bc687c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:decba9c4820f08e7b0b615fa39083134d3448002ef73413f20ed8ab4476deee5
+size 261945
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/3d87bbbb193cee16a780982110392639 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/3d87bbbb193cee16a780982110392639
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/3d87bbbb193cee16a780982110392639 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/43131c490fa1863ee8c914cfa6ec1c22 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/43131c490fa1863ee8c914cfa6ec1c22
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/43131c490fa1863ee8c914cfa6ec1c22 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/8a0b255aff1a30c53a7d280277fea03c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/8a0b255aff1a30c53a7d280277fea03c
new file mode 100644
index 0000000000000000000000000000000000000000..fbb741d1a644f133e4214aa020307b7cafb9709f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/8a0b255aff1a30c53a7d280277fea03c differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..423dfa6381c26a1b3f1042ea55535feb59468423
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..bcb0ea37a3b12eedaef9aaf8930041947925670a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754579313816993300, "commit_timestamp_nsecs": 1754579313959187028, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/d/b7fdc205657daeb4f71ba77f8283d42e b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/d/b7fdc205657daeb4f71ba77f8283d42e
new file mode 100644
index 0000000000000000000000000000000000000000..7fd416dee05e3e0bb18a3cf71a9371cbbe8fe4a9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/d/b7fdc205657daeb4f71ba77f8283d42e differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..2ca459fc1581517bc8135a13bcf49f87cc78ed6f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/25493ac3743e211ad2d18df1b7071f2c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/25493ac3743e211ad2d18df1b7071f2c
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/25493ac3743e211ad2d18df1b7071f2c differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/2be25adc9d7f28355836cbfe8376291e b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/2be25adc9d7f28355836cbfe8376291e
new file mode 100644
index 0000000000000000000000000000000000000000..2fc4a2d0381e96dfdf95f36443e78103a5ee1d26
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/2be25adc9d7f28355836cbfe8376291e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d1674e11e30b38265cf54e3e8149d571ede8245ae28732a14528fc074348498e
+size 411136
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/4c96f29691ae9bd4414a6ed1e026a6a3 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/4c96f29691ae9bd4414a6ed1e026a6a3
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/4c96f29691ae9bd4414a6ed1e026a6a3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/65ce2ee8bc413428a66f268c8f856b88 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/65ce2ee8bc413428a66f268c8f856b88
new file mode 100644
index 0000000000000000000000000000000000000000..a75f32c681daab16ddc4871e15914b7bdf31e0a0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/65ce2ee8bc413428a66f268c8f856b88 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/744dd8b7cfe759b97a205d44c26c93cf b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/744dd8b7cfe759b97a205d44c26c93cf
new file mode 100644
index 0000000000000000000000000000000000000000..2910fbc6db2094404ed932c7d9bcdb23beb99906
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/744dd8b7cfe759b97a205d44c26c93cf differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/aa864fe06c46bfd13c477c8c0461e3a3 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/aa864fe06c46bfd13c477c8c0461e3a3
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/aa864fe06c46bfd13c477c8c0461e3a3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/bf48914af98105298f0145ee59280252 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/bf48914af98105298f0145ee59280252
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/bf48914af98105298f0145ee59280252 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/ec31cddb61102f429a7582b45d6b5deb b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/ec31cddb61102f429a7582b45d6b5deb
new file mode 100644
index 0000000000000000000000000000000000000000..d3ddfe0efe2e3cccc640c4087ffaf9e4148ed828
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/ec31cddb61102f429a7582b45d6b5deb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4063794189fd18141feb7807a0ae13bfffdb3889a25e8dc0fe44110f51555ddc
+size 286991
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..97a0f774407c8d0d5dbb8def49afbe7afa7730cd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..92b49250515cbaf32494a4d4ce6fc6678c04b97d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754579952187795427, "commit_timestamp_nsecs": 1754579952346193488, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/d/ad8c02e74866db8eaa8fad61e87be5e9 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/d/ad8c02e74866db8eaa8fad61e87be5e9
new file mode 100644
index 0000000000000000000000000000000000000000..a1bf8fd0133b5fa73e80dbca88a472277ed5e014
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/d/ad8c02e74866db8eaa8fad61e87be5e9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..2f64686e03df7dd598c106e40ef4e988aa1367b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/0ba46f5ce02066435edd9ce66170b142 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/0ba46f5ce02066435edd9ce66170b142
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/0ba46f5ce02066435edd9ce66170b142 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/1df01566f60f3bd247df2c28b0e379c0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/1df01566f60f3bd247df2c28b0e379c0
new file mode 100644
index 0000000000000000000000000000000000000000..b22bb0ddce53a51d8037fd559f9a7d5423e56a87
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/1df01566f60f3bd247df2c28b0e379c0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/532e2d5f1baff38a210040ec646a492d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/532e2d5f1baff38a210040ec646a492d
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/532e2d5f1baff38a210040ec646a492d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/7e92c6075e72e7eb6fec8d5454302cab b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/7e92c6075e72e7eb6fec8d5454302cab
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/7e92c6075e72e7eb6fec8d5454302cab differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/9a6f6f80b65a3214d6c599318bd00504 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/9a6f6f80b65a3214d6c599318bd00504
new file mode 100644
index 0000000000000000000000000000000000000000..acb477d0d72e6734904e7dcfb4974f8afa53ab48
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/9a6f6f80b65a3214d6c599318bd00504
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:662caeef4875c0ced73a61b060239079a09924f3e8ba130a600ba9909f9bb9cf
+size 439007
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/a98b594633221a2d39dca7eb1306f774 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/a98b594633221a2d39dca7eb1306f774
new file mode 100644
index 0000000000000000000000000000000000000000..7f53bf898adcb83d7a955d186a6f75894b6baacc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/a98b594633221a2d39dca7eb1306f774
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f4a4aecc6a6df10ae6c1821eb40ca01dc2b5852326aaedbdf5e0721da3909ed1
+size 261610
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/fabefb0faeb8479abf0093d016d369e8 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/fabefb0faeb8479abf0093d016d369e8
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/fabefb0faeb8479abf0093d016d369e8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3011ba8089d6379c607d516d14a1c43733b58e31
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..123d37e587dfffb5a89fcb50d531f5c2798dc9b0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754579986718454247, "commit_timestamp_nsecs": 1754579986851830346, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/d/eda20e5e2c67162fd278a6c6b2ac6dcb b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/d/eda20e5e2c67162fd278a6c6b2ac6dcb
new file mode 100644
index 0000000000000000000000000000000000000000..48f784d48a6d03c2b7ee5eb6d79644c627f6ae2d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/d/eda20e5e2c67162fd278a6c6b2ac6dcb differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..2255d7f3769370f69979a3b67ba1d58af9fbad8e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/893c524411022693da196f9c708f6c5e b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/893c524411022693da196f9c708f6c5e
new file mode 100644
index 0000000000000000000000000000000000000000..e7d6541036f639b8d02974ff4c8ea7e1ab088a4a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/893c524411022693da196f9c708f6c5e differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/9f40fab18cf0cdf70e740b4af919c16e b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/9f40fab18cf0cdf70e740b4af919c16e
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/9f40fab18cf0cdf70e740b4af919c16e differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/baa61bc93f9871c324bd04f70afbbc80 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/baa61bc93f9871c324bd04f70afbbc80
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/baa61bc93f9871c324bd04f70afbbc80 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/cac19a674d4df8164c95c1d75b967e38 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/cac19a674d4df8164c95c1d75b967e38
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/cac19a674d4df8164c95c1d75b967e38 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/da1988eabc9e008acbbcc87ad9e80424 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/da1988eabc9e008acbbcc87ad9e80424
new file mode 100644
index 0000000000000000000000000000000000000000..4d2746c95080bceb84fbe21cd29f964430059a0d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/da1988eabc9e008acbbcc87ad9e80424
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bb2c68479bf0a692d29d96a33ed66dbca7be1962d862557f55d7b9be92f64da1
+size 510740
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/e18a271c51d2a6f12f45d8bb7993e030 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/e18a271c51d2a6f12f45d8bb7993e030
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/e18a271c51d2a6f12f45d8bb7993e030 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/e35a878e67066209be052fa81262353d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/e35a878e67066209be052fa81262353d
new file mode 100644
index 0000000000000000000000000000000000000000..577d09c34c33e206b5fa388abbc598d18fa5af90
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/e35a878e67066209be052fa81262353d
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:919b32d265f804859f6afbf50d68ac414f97f46a02f4df31cbba4a260f9a7338
+size 193891
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/e8db925d623a6bed37df7e65e277b316 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/e8db925d623a6bed37df7e65e277b316
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/e8db925d623a6bed37df7e65e277b316 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5be3e238bf805e57ef3135aeb3dc5a0dca533e1f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..00438a4b3b6e8387b7ca4c67091e7258b1aa8c3b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580021419364477, "commit_timestamp_nsecs": 1754580021582954576, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/d/6b681201cbeab0315f1eb3cd45ba2264 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/d/6b681201cbeab0315f1eb3cd45ba2264
new file mode 100644
index 0000000000000000000000000000000000000000..9f570d37fa840d39f9f3d1de1dfac97172290a32
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/d/6b681201cbeab0315f1eb3cd45ba2264 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..aa8cb12b2fa9648512fbb159baceae79c24bf58d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/4b3a451a327b867ac7227b20c2588720 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/4b3a451a327b867ac7227b20c2588720
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/4b3a451a327b867ac7227b20c2588720 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/6ed8dc60ac8a342f26738cd890139f2f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/6ed8dc60ac8a342f26738cd890139f2f
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/6ed8dc60ac8a342f26738cd890139f2f differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/93049ac0151d5e051fd486717a5b92d4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/93049ac0151d5e051fd486717a5b92d4
new file mode 100644
index 0000000000000000000000000000000000000000..f9871c0166b6bc3601c566aaffd58d85b71cf412
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/93049ac0151d5e051fd486717a5b92d4
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2e28324c4228970dc660c3f4b39db5118411684574529e6590ba7b445c47572d
+size 372468
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/950cea80a4e22df328021a2e55cd8512 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/950cea80a4e22df328021a2e55cd8512
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/950cea80a4e22df328021a2e55cd8512 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/ba3f8a39735851d273790b875331e3f9 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/ba3f8a39735851d273790b875331e3f9
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/ba3f8a39735851d273790b875331e3f9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/caf16fd5d85a7f7a7455cd0f1acae211 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/caf16fd5d85a7f7a7455cd0f1acae211
new file mode 100644
index 0000000000000000000000000000000000000000..7ce237b9a49f8164175db70f415c411c31e53be1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/caf16fd5d85a7f7a7455cd0f1acae211 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/f72f06833f3b9eda1d0779c2ea631fb7 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/f72f06833f3b9eda1d0779c2ea631fb7
new file mode 100644
index 0000000000000000000000000000000000000000..205da3b5e8e44ee340e5848b768972fe4edd6792
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/f72f06833f3b9eda1d0779c2ea631fb7
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4a2f7d9d16b3bcd648f65f84b3a106c7c599220dedeb4f3d64ecb534f18d84d7
+size 331703
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8d41fe098d4abb7408812d44ee0552f64f42e57b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..faf7acaecae94f436df872666866774c82876647
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580055831450316, "commit_timestamp_nsecs": 1754580055973094232, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/d/1cd8e67a9b78f7dc1c0f9db56699abab b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/d/1cd8e67a9b78f7dc1c0f9db56699abab
new file mode 100644
index 0000000000000000000000000000000000000000..4441b1539487442923ed0bc7abc63538e025dae3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/d/1cd8e67a9b78f7dc1c0f9db56699abab differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3a095d0d66a5e90268aa5e5d1499fab08d47c07d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/016b5638fde93eb96da07a06d1ec1a29 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/016b5638fde93eb96da07a06d1ec1a29
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/016b5638fde93eb96da07a06d1ec1a29 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/46366fb808d55275a47c6f970e966eca b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/46366fb808d55275a47c6f970e966eca
new file mode 100644
index 0000000000000000000000000000000000000000..eeef2c3f1287606a287afe9ce8e17581c7e5bd7e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/46366fb808d55275a47c6f970e966eca differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/4c9348416f9a37bcbec31d7d386aed6d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/4c9348416f9a37bcbec31d7d386aed6d
new file mode 100644
index 0000000000000000000000000000000000000000..b6f335fc5aa08bfcced3d56757532ea7808dd724
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/4c9348416f9a37bcbec31d7d386aed6d
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1fa2bb6a40ab72419c4c64099bd761802f459bc48dea52d070ca7a49838e566f
+size 287362
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/4d947ded903a3d03e7d9ea3fb2d39e0a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/4d947ded903a3d03e7d9ea3fb2d39e0a
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/4d947ded903a3d03e7d9ea3fb2d39e0a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/9bb0427f134ca6574630a82f29dba5a5 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/9bb0427f134ca6574630a82f29dba5a5
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/9bb0427f134ca6574630a82f29dba5a5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/ae7317f9512dea8bae9043e9fbfe2f30 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/ae7317f9512dea8bae9043e9fbfe2f30
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/ae7317f9512dea8bae9043e9fbfe2f30 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/dd01a8c95a6ac8f306f723df55a566fb b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/dd01a8c95a6ac8f306f723df55a566fb
new file mode 100644
index 0000000000000000000000000000000000000000..76b747c48e3c3be1646770ce23a8992efb93ee4b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/dd01a8c95a6ac8f306f723df55a566fb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7cbf35a4b82e5a7524579593af7f589501a3f5149999f495b52fcb05321fe39c
+size 450915
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4cfd297ec9b583f358fa76a80668aef61cd34ba5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..e1336eb7cbe93806814c777d894786d4bc2c8a61
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580090560820717, "commit_timestamp_nsecs": 1754580090735961527, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/d/32a74e73c1159d76de6057408d41e49d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/d/32a74e73c1159d76de6057408d41e49d
new file mode 100644
index 0000000000000000000000000000000000000000..24fc6ab9f9b8a4a5f2002a7fed6c26b22f582a24
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/d/32a74e73c1159d76de6057408d41e49d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e4fdb9ab13ebf53a3dd60a21410873bc8f8d8cf7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/04142de45197fe1e95c8584c8442844d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/04142de45197fe1e95c8584c8442844d
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/04142de45197fe1e95c8584c8442844d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/15058f008845f681ac98164c21c25cfd b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/15058f008845f681ac98164c21c25cfd
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/15058f008845f681ac98164c21c25cfd differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/1f5022d0cceba2485a4966436a0e6478 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/1f5022d0cceba2485a4966436a0e6478
new file mode 100644
index 0000000000000000000000000000000000000000..91ef272d0e3f5620f25f9ea4f90962bc35943e62
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/1f5022d0cceba2485a4966436a0e6478 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/21c41450a859fb1c4fc6787da008c8f8 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/21c41450a859fb1c4fc6787da008c8f8
new file mode 100644
index 0000000000000000000000000000000000000000..c28e1b3cdb59a3d16a490db4053da6071236852e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/21c41450a859fb1c4fc6787da008c8f8
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9d9003b7481646baa5b09deb2aedae54b130f2f45e73fed35623f7ba67961808
+size 451143
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/31ec8309a3f1d4764f203899f20bb374 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/31ec8309a3f1d4764f203899f20bb374
new file mode 100644
index 0000000000000000000000000000000000000000..bb2b501c0847f966fd92b7a6fd7d71649c2b9ed2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/31ec8309a3f1d4764f203899f20bb374
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:206b639236639ffe0d9d4ede8f2c93a2a8da36df0cfac4a7fb89796ee0135e4b
+size 287409
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/4ec932cbf2109cc7985fe89fafe66c75 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/4ec932cbf2109cc7985fe89fafe66c75
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/4ec932cbf2109cc7985fe89fafe66c75 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/838f7ecce33d37a26f7c002be957f615 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/838f7ecce33d37a26f7c002be957f615
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/838f7ecce33d37a26f7c002be957f615 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/ddd9cbfab98498961d0b9da3caa9e5c8 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/ddd9cbfab98498961d0b9da3caa9e5c8
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/ddd9cbfab98498961d0b9da3caa9e5c8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/ed79236ca73dd7ad8fb41750e39d772d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/ed79236ca73dd7ad8fb41750e39d772d
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/ed79236ca73dd7ad8fb41750e39d772d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b5c57a58288f5209b5bca4d94194b7ff160cb987
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..08f33f6070d698f5ab381f08d94d7a80b059289d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580125135339828, "commit_timestamp_nsecs": 1754580125272429566, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/d/29ba45f303c256d4011ce0ca5259aa62 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/d/29ba45f303c256d4011ce0ca5259aa62
new file mode 100644
index 0000000000000000000000000000000000000000..46d8b96bdbbb0bdf8c88628fbd0f0b151c35aebf
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/d/29ba45f303c256d4011ce0ca5259aa62 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c93193a06c514d0e09b678eff322cbd307bc9548
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/21accee996812e62ea5fde7243d0e7cf b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/21accee996812e62ea5fde7243d0e7cf
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/21accee996812e62ea5fde7243d0e7cf differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/3930770780cfccdd7a79c067fd3a0be4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/3930770780cfccdd7a79c067fd3a0be4
new file mode 100644
index 0000000000000000000000000000000000000000..7f1a26b8db10e7603b75437fafc52fea61d4e40d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/3930770780cfccdd7a79c067fd3a0be4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/3a658b2bee3c817fb2de495d9489225a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/3a658b2bee3c817fb2de495d9489225a
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/3a658b2bee3c817fb2de495d9489225a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/564fb6eaaec5a8856142376f12771586 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/564fb6eaaec5a8856142376f12771586
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/564fb6eaaec5a8856142376f12771586 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/99d26fec0cb307a7eeceab121990d581 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/99d26fec0cb307a7eeceab121990d581
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/99d26fec0cb307a7eeceab121990d581 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/c4385e1ec115fd1980cd71d66008d970 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/c4385e1ec115fd1980cd71d66008d970
new file mode 100644
index 0000000000000000000000000000000000000000..5d3e7c863052c31349d831a43e55a6643a63e438
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/c4385e1ec115fd1980cd71d66008d970
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1c9c3a5defabeeb6bb8a3bf0febc501b088e3a25719e8e9a8c718875dcadc829
+size 389325
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/d505e8c0e66c95e6ebc3ce3b032dbfa7 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/d505e8c0e66c95e6ebc3ce3b032dbfa7
new file mode 100644
index 0000000000000000000000000000000000000000..9ad007ad839b0f98c319259e62b81787161560f0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/d505e8c0e66c95e6ebc3ce3b032dbfa7
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:222c1e683df9e43cae50deea24add5ff8fc5c5f40316c1db5a899bff42c506f0
+size 275072
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8744828cebe45df20152a8b96a7087c6fb1303c9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..f9ce488f6972d0c4b8cb003cee563c9e5a5529e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580159836087151, "commit_timestamp_nsecs": 1754580159983338644, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/d/93f1cfad3b2562209bc8ffc78f698d64 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/d/93f1cfad3b2562209bc8ffc78f698d64
new file mode 100644
index 0000000000000000000000000000000000000000..77e06e3a339eee3d432a8578d8a0d16001c11504
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/d/93f1cfad3b2562209bc8ffc78f698d64 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d6278603533a131ba7685549fc83f0b9cf311868
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/24059d0432858b18a0eb2a9557590963 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/24059d0432858b18a0eb2a9557590963
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/24059d0432858b18a0eb2a9557590963 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/53a70bcf89c205568ac0dbca4b539773 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/53a70bcf89c205568ac0dbca4b539773
new file mode 100644
index 0000000000000000000000000000000000000000..89cf5000c2e814038a49df83419f8854ac09d0cb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/53a70bcf89c205568ac0dbca4b539773 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/6a35290584552b8d01d141201c7ca47d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/6a35290584552b8d01d141201c7ca47d
new file mode 100644
index 0000000000000000000000000000000000000000..bc6b3f9713a81c9c5e69ec7ce7bb9bbab5802f4c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/6a35290584552b8d01d141201c7ca47d
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b2716dfaf361a9da81b510424e1156636a4562902da667669dff99f1c41634df
+size 244848
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/6f00d8a9b9bea1dc271ed3e96388d7bb b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/6f00d8a9b9bea1dc271ed3e96388d7bb
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/6f00d8a9b9bea1dc271ed3e96388d7bb differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/92b3f191a0c3a406229e36f6179be13b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/92b3f191a0c3a406229e36f6179be13b
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/92b3f191a0c3a406229e36f6179be13b differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/c3b57d3ab4fb2f8500f5878a5a5fb849 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/c3b57d3ab4fb2f8500f5878a5a5fb849
new file mode 100644
index 0000000000000000000000000000000000000000..8608cc7153b616c789e714ee9228642ecf9e62c7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/c3b57d3ab4fb2f8500f5878a5a5fb849
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:44f46859b3cf888295febfaf1ba81d391be3610ebed14a132776ed8e0a2b7f1c
+size 456006
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/cd89b5a1b1a28358637b788191ebf99d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/cd89b5a1b1a28358637b788191ebf99d
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/cd89b5a1b1a28358637b788191ebf99d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/dd17660e484aa896697a1ddc8099157d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/dd17660e484aa896697a1ddc8099157d
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/dd17660e484aa896697a1ddc8099157d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..77141c866cfd480391170d2e26277126d0caa217
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..447794e09422ee0917a3107dce23dc59505a3b08
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580194635011615, "commit_timestamp_nsecs": 1754580194783407708, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/d/1028d91105c9be9dc80a31943bdae1d9 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/d/1028d91105c9be9dc80a31943bdae1d9
new file mode 100644
index 0000000000000000000000000000000000000000..d2fcb376efaba98e5f5f3f1b77e6b8f05646dbda
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/d/1028d91105c9be9dc80a31943bdae1d9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..097216f0a7c47bd1144c1e052700226bc6f2e1d0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/4425d6801fe13b458ec7d1467d673426 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/4425d6801fe13b458ec7d1467d673426
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/4425d6801fe13b458ec7d1467d673426 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/4e62b8f04c9c71282884687dee6185df b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/4e62b8f04c9c71282884687dee6185df
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/4e62b8f04c9c71282884687dee6185df differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/6ff847505bb548492c4bfde29a893992 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/6ff847505bb548492c4bfde29a893992
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/6ff847505bb548492c4bfde29a893992 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/7a81a83e37ace46460e1ca2764fb3821 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/7a81a83e37ace46460e1ca2764fb3821
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/7a81a83e37ace46460e1ca2764fb3821 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/93f7c0a0013c727c9fbc80056c37d404 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/93f7c0a0013c727c9fbc80056c37d404
new file mode 100644
index 0000000000000000000000000000000000000000..0f5032980e59d916b096bfc707458b038dc2f98b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/93f7c0a0013c727c9fbc80056c37d404
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:09142cef109803756d0fb9b8949b59c864d511e73076ca00874a52c220953ab0
+size 510868
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/96a5ea00277968e24209ff27385c996e b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/96a5ea00277968e24209ff27385c996e
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/96a5ea00277968e24209ff27385c996e differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/9d539ba4a88bac57cff042ea9f066b95 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/9d539ba4a88bac57cff042ea9f066b95
new file mode 100644
index 0000000000000000000000000000000000000000..0cc80970074b1c7cca3f7d3acd2ad2bada365dcb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/9d539ba4a88bac57cff042ea9f066b95
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:eec0415cb3c8985f2afdcf5e0d243d2a80ada61bac6a059ccac7d4c1d8bc0905
+size 227739
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/c791fd67e4728ff2c980cfb08e2311ff b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/c791fd67e4728ff2c980cfb08e2311ff
new file mode 100644
index 0000000000000000000000000000000000000000..4f8378e288e80b2a9fa97606611f4abede7f1a47
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/c791fd67e4728ff2c980cfb08e2311ff differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8f38cf3a555ba8c62438cb77564238f476af3b84
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..68f3b16128ba27eb0b49b515e8c6740a4f5aa7a9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580229262591165, "commit_timestamp_nsecs": 1754580229434456028, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/d/e2395c0531d60f9ea0878ab8291198aa b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/d/e2395c0531d60f9ea0878ab8291198aa
new file mode 100644
index 0000000000000000000000000000000000000000..52dc9de62a5f315c5b30537df83f1198d1de6b41
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/d/e2395c0531d60f9ea0878ab8291198aa differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9a5c5ab0f85384d693c7128f09b61f77ec96b3d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/135f477c9fba1ce746558f2e49938118 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/135f477c9fba1ce746558f2e49938118
new file mode 100644
index 0000000000000000000000000000000000000000..df33bfd8e63922d06545838072a51936d5496d9d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/135f477c9fba1ce746558f2e49938118
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f2d4cd88d1fc71c4bdf8ddc4f9230af385840f912a94a3683aed40ddeead71fb
+size 270524
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/27818e74dd04d05f7ba61c75c0db092f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/27818e74dd04d05f7ba61c75c0db092f
new file mode 100644
index 0000000000000000000000000000000000000000..620dfdf29b71b1c0fbb08aa15ad9e433f9a71ebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/27818e74dd04d05f7ba61c75c0db092f differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/95c96f273752e162c3f11b62f62789f2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/95c96f273752e162c3f11b62f62789f2
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/95c96f273752e162c3f11b62f62789f2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/9c74b79e5fd3be552cc3c86c63437eda b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/9c74b79e5fd3be552cc3c86c63437eda
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/9c74b79e5fd3be552cc3c86c63437eda differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/a6dda7df59776f1f3d515749f70555d6 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/a6dda7df59776f1f3d515749f70555d6
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/a6dda7df59776f1f3d515749f70555d6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/cbc73bdd1d7f045d711c882203f7770c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/cbc73bdd1d7f045d711c882203f7770c
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/cbc73bdd1d7f045d711c882203f7770c differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/d12b1d56dcda152a29e89b1e149cccdd b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/d12b1d56dcda152a29e89b1e149cccdd
new file mode 100644
index 0000000000000000000000000000000000000000..f16944e7a58f822c844752d8a120a528b543fd20
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/d12b1d56dcda152a29e89b1e149cccdd
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:57c9df5a5d9aba64927ee79c3dcf858be5638322a3b2bd8f4cedd1a34e8a7f4e
+size 434164
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/e76ed1ee91e2091fffa8410402d106e7 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/e76ed1ee91e2091fffa8410402d106e7
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/e76ed1ee91e2091fffa8410402d106e7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..89d527d85b5c76eb9c17e7a85d88f068cd800d64
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..6813fda7f684971c7e358801569918b4c8941e33
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580264054079638, "commit_timestamp_nsecs": 1754580264226989322, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/d/133417dd88e8b89c55cb4a8aeacc7763 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/d/133417dd88e8b89c55cb4a8aeacc7763
new file mode 100644
index 0000000000000000000000000000000000000000..119b7a419c4c444edf73f0798f06194bd00467db
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/d/133417dd88e8b89c55cb4a8aeacc7763 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9984b39ce9b3dae761e4f0be7332ace1749b41ea
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/07d239e770e5855b4788d13d75e2bc91 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/07d239e770e5855b4788d13d75e2bc91
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/07d239e770e5855b4788d13d75e2bc91 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/28434335b34a012678371061427854bf b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/28434335b34a012678371061427854bf
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/28434335b34a012678371061427854bf differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/9d95a022c8f10b82fb3f3813f4c27a46 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/9d95a022c8f10b82fb3f3813f4c27a46
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/9d95a022c8f10b82fb3f3813f4c27a46 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/ad4357335f1a234b0c9287a0ed692c7f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/ad4357335f1a234b0c9287a0ed692c7f
new file mode 100644
index 0000000000000000000000000000000000000000..40e51d438b729473e756447c0c05b07fd3784db0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/ad4357335f1a234b0c9287a0ed692c7f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:949917f2e6196e810d933bef38e19c05f02d5ffd4a19911edfd2fd15fdc263f6
+size 295896
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/bfe08001f4f4f106703a1464e0dada7f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/bfe08001f4f4f106703a1464e0dada7f
new file mode 100644
index 0000000000000000000000000000000000000000..75b962927571f60448e823b70ca2e6e052afc54d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/bfe08001f4f4f106703a1464e0dada7f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cae75610fa1993ae2206161ee6eb4847db308794ddc799b93fade04d784bf8c3
+size 442404
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/e7111fdfc5cb76c05227e62e1e6cd7aa b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/e7111fdfc5cb76c05227e62e1e6cd7aa
new file mode 100644
index 0000000000000000000000000000000000000000..1ec6494aeb7007d3c20f2675229c47845939e5c9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/e7111fdfc5cb76c05227e62e1e6cd7aa differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/ed7b91aaeef943401c8236110136d108 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/ed7b91aaeef943401c8236110136d108
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/ed7b91aaeef943401c8236110136d108 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/f80212047dc8d57133a824b0f4895014 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/f80212047dc8d57133a824b0f4895014
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/f80212047dc8d57133a824b0f4895014 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..05e28c77578e3cc01b11a7965002352475f39cf6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..737e14e1231776fb0d18e326500adca1140280af
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754579347384972468, "commit_timestamp_nsecs": 1754579347561492809, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/d/86622bf5e40c0c294243a03161035c47 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/d/86622bf5e40c0c294243a03161035c47
new file mode 100644
index 0000000000000000000000000000000000000000..fe5e069ee7686e63ddaae3af16bb4a37438c3577
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/d/86622bf5e40c0c294243a03161035c47 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3b7c696e70ac8c0a7dd89b479cd73317b7413879
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/115cb89b1e23935a36665a44caef850a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/115cb89b1e23935a36665a44caef850a
new file mode 100644
index 0000000000000000000000000000000000000000..3380afa17910841c8242c32589929407284934c2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/115cb89b1e23935a36665a44caef850a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5d7a09e1796b93b88842e0a204e3758f03f0e00d47ee6452c161ad60dd690f06
+size 406071
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/47b9b42340d2eaed5c9e5c6e1beaab7b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/47b9b42340d2eaed5c9e5c6e1beaab7b
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/47b9b42340d2eaed5c9e5c6e1beaab7b differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/5eac2bc8e1884b0a310ec354713b6536 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/5eac2bc8e1884b0a310ec354713b6536
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/5eac2bc8e1884b0a310ec354713b6536 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/81be5d927a1cdb69e7467e0e651d3c73 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/81be5d927a1cdb69e7467e0e651d3c73
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/81be5d927a1cdb69e7467e0e651d3c73 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/d48dab013b44cc4fb24e385091380e17 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/d48dab013b44cc4fb24e385091380e17
new file mode 100644
index 0000000000000000000000000000000000000000..f4d06912cd9674b40e41d858f190044f33ad277a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/d48dab013b44cc4fb24e385091380e17 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/e12f6a3db68a90ffa95374d0f339ffcb b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/e12f6a3db68a90ffa95374d0f339ffcb
new file mode 100644
index 0000000000000000000000000000000000000000..c3105fdc6009227abd604f4021f18c3c5c455e6c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/e12f6a3db68a90ffa95374d0f339ffcb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a8ff2b6769bdf60b4956a88d308678efbe4ca4e35f6af637c20c662d94bb7be3
+size 291827
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/fd648e1ba5b1008a67cb3b1f7a6b261d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/fd648e1ba5b1008a67cb3b1f7a6b261d
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/fd648e1ba5b1008a67cb3b1f7a6b261d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..be9797a9119fa1af30b07f359c373e49edb19754
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..1967cf624df49aa2851ec98ff6b86cffb61fb626
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580299361442187, "commit_timestamp_nsecs": 1754580299533473697, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/d/37b3a33b755e775657d37f684a77a046 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/d/37b3a33b755e775657d37f684a77a046
new file mode 100644
index 0000000000000000000000000000000000000000..dc4f4d0c6ab23a9498f898de54185ab768677aa4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/d/37b3a33b755e775657d37f684a77a046 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..7aa02ca287e329bc94607b0a9ed65aee9908d6ba
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/510e83f18e413538e91f66729178752f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/510e83f18e413538e91f66729178752f
new file mode 100644
index 0000000000000000000000000000000000000000..defde297cd72a66fca56f942f1dcc8724662dcb0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/510e83f18e413538e91f66729178752f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8b90c48c70d029dca8403351aa4168261e7c3c2edbeef3a4e3a2909a45344217
+size 469768
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/77583ffb900d3dfb531abfbfdd4fc017 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/77583ffb900d3dfb531abfbfdd4fc017
new file mode 100644
index 0000000000000000000000000000000000000000..91bee0d18a4d28b097649f599ecc985f150e3dfb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/77583ffb900d3dfb531abfbfdd4fc017 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/7d9919fb39f2f7eb21d7a7baec26808c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/7d9919fb39f2f7eb21d7a7baec26808c
new file mode 100644
index 0000000000000000000000000000000000000000..cabc7afa69afb184dc037818efaf1fc1d7e2c56e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/7d9919fb39f2f7eb21d7a7baec26808c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c1cc97daa558eb6ed62c121c060e937251ae6c5365cbc0b563cd002b2a489fa3
+size 185418
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/8016085994a5d10f7f92276770a81f3f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/8016085994a5d10f7f92276770a81f3f
new file mode 100644
index 0000000000000000000000000000000000000000..a1e6bba64a7f82838e4e647fd31a034b8a5deed9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/8016085994a5d10f7f92276770a81f3f differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/92fbf7c6c83b1ac28ee990a7cc060e47 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/92fbf7c6c83b1ac28ee990a7cc060e47
new file mode 100644
index 0000000000000000000000000000000000000000..8dacfe7e0c1f19b409a1d61d0ce6036a8ac74fbf
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/92fbf7c6c83b1ac28ee990a7cc060e47 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/93e12363a3bb488f3ec10102967d44db b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/93e12363a3bb488f3ec10102967d44db
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/93e12363a3bb488f3ec10102967d44db differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/aa1b3a847cdc69864dc4980273591882 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/aa1b3a847cdc69864dc4980273591882
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/aa1b3a847cdc69864dc4980273591882 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/b4e8b7a0709b1f18da62036deafeb49e b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/b4e8b7a0709b1f18da62036deafeb49e
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/b4e8b7a0709b1f18da62036deafeb49e differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/ea81aa584fe60dd265a1df8863a42647 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/ea81aa584fe60dd265a1df8863a42647
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/ea81aa584fe60dd265a1df8863a42647 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..59b26cbbf1656d495a94c3465fbf16659d2447ac
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..45f9798f9a0094ef1a8a3b708a2a269a58deb59b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580334046422044, "commit_timestamp_nsecs": 1754580334216713880, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/d/7bb66a226f2522d7c347b9f20875d439 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/d/7bb66a226f2522d7c347b9f20875d439
new file mode 100644
index 0000000000000000000000000000000000000000..aaf92b8ca7866e06e34c9fa5538122adb1fa3841
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/d/7bb66a226f2522d7c347b9f20875d439 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..ef781b5a4d26fbd624cffd52766d3d826de831ab
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/200e2595672b80c237b64aa35d585124 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/200e2595672b80c237b64aa35d585124
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/200e2595672b80c237b64aa35d585124 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/2ac35eda190fe8f6884e1b7341ce4b7e b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/2ac35eda190fe8f6884e1b7341ce4b7e
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/2ac35eda190fe8f6884e1b7341ce4b7e differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/300ead550e6fe591e22387a2d6d5ae02 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/300ead550e6fe591e22387a2d6d5ae02
new file mode 100644
index 0000000000000000000000000000000000000000..fe640af736b7e1dddd4d44d67199407a5905bc07
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/300ead550e6fe591e22387a2d6d5ae02 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/71680f44f2fd187bc9624f82f79c96d7 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/71680f44f2fd187bc9624f82f79c96d7
new file mode 100644
index 0000000000000000000000000000000000000000..352e20233d6d6d04c4179757978ab9992709d0a5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/71680f44f2fd187bc9624f82f79c96d7
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1f3ad6984b864c997b7420f7fdbfb01d2c58e55291dfd4eda19a21ef65c456bd
+size 300654
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/9bd3d5e67d66178e1e4ae8134894ba7d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/9bd3d5e67d66178e1e4ae8134894ba7d
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/9bd3d5e67d66178e1e4ae8134894ba7d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/bc9b593afb7dacf4c4cea5e7f69a34c9 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/bc9b593afb7dacf4c4cea5e7f69a34c9
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/bc9b593afb7dacf4c4cea5e7f69a34c9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/f3c9773476d82585a2cf7e3e82759bdb b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/f3c9773476d82585a2cf7e3e82759bdb
new file mode 100644
index 0000000000000000000000000000000000000000..bbdbd6a4fb1b392c7eacce2812c816c75c66328c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/f3c9773476d82585a2cf7e3e82759bdb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e56dcdca8232fb545866cc453fbf76d4f742d7c8ce809a24f4840a8a07b91ba8
+size 397913
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..66fc0b76899e6add68919167c24949ec3bebc06d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..52282481bdc5d30103dd36314769d9789333bef0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580385354771684, "commit_timestamp_nsecs": 1754580385529518863, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/d/08e2e5d120dca8877e311713b17f67ee b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/d/08e2e5d120dca8877e311713b17f67ee
new file mode 100644
index 0000000000000000000000000000000000000000..bcd45f0441e5c9cfa28cec8138a874cb11a39a94
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/d/08e2e5d120dca8877e311713b17f67ee differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4d6fc06d6fcbfdea8f881785dbae8c100b0abc64
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/12be06fa19064e1f1efaa25fb4076376 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/12be06fa19064e1f1efaa25fb4076376
new file mode 100644
index 0000000000000000000000000000000000000000..7fd58244825326d994da8a81908412bb59b4ba08
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/12be06fa19064e1f1efaa25fb4076376 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/622867bd2e70d0e0e385e7e08e4b721e b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/622867bd2e70d0e0e385e7e08e4b721e
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/622867bd2e70d0e0e385e7e08e4b721e differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/89a264fa5ca6139df168b9e7960ea60a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/89a264fa5ca6139df168b9e7960ea60a
new file mode 100644
index 0000000000000000000000000000000000000000..88e84a80c68ff9cf038106668362aa594de31099
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/89a264fa5ca6139df168b9e7960ea60a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:20a3d79c663765a13ca9539159e5e3198fd1678cf74f88854e8e04682eb4c40b
+size 329984
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/ab9619820e2c414a5ff92be2e75610e6 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/ab9619820e2c414a5ff92be2e75610e6
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/ab9619820e2c414a5ff92be2e75610e6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/b44a7962f3411d8c8c6cb3654c159cbd b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/b44a7962f3411d8c8c6cb3654c159cbd
new file mode 100644
index 0000000000000000000000000000000000000000..40ac7e9ad530c5709d0d8ba465936a0137b8eaaf
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/b44a7962f3411d8c8c6cb3654c159cbd
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:323112aeb3644247a107974de6c799558f0b3a1bae67ed80e3c76c9b09d5db56
+size 370784
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/e50e7d5da3671593d695986177c9796a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/e50e7d5da3671593d695986177c9796a
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/e50e7d5da3671593d695986177c9796a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/ec0fecef7c40d8cbe7b6bd766e234427 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/ec0fecef7c40d8cbe7b6bd766e234427
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/ec0fecef7c40d8cbe7b6bd766e234427 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c16d55ec1a37e238676575d323df4059cd56b3d6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..98b53a02c0a4800a50955a0535d0f6a29a4fac3a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580420269439687, "commit_timestamp_nsecs": 1754580420445011337, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/d/9a0e6e98fd0a933ef78ffd126903e234 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/d/9a0e6e98fd0a933ef78ffd126903e234
new file mode 100644
index 0000000000000000000000000000000000000000..4c5ab70e8d25656edadd11214ff7d0cbe487ab73
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/d/9a0e6e98fd0a933ef78ffd126903e234 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f1cc6282475770b334b5d7c9b659cbf603f18130
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/02d229525d0d5825084dd641f0245704 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/02d229525d0d5825084dd641f0245704
new file mode 100644
index 0000000000000000000000000000000000000000..08af62e5a15ea3f4d4e91486338d30a094cf6ca8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/02d229525d0d5825084dd641f0245704 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/0dd1c3c25fa23fec02909b7fdff97c59 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/0dd1c3c25fa23fec02909b7fdff97c59
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/0dd1c3c25fa23fec02909b7fdff97c59 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/1209d59c1d892966703ce2db8802c1f1 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/1209d59c1d892966703ce2db8802c1f1
new file mode 100644
index 0000000000000000000000000000000000000000..c97bb30b738dce7428abfe6083f1baf4d77ce2f4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/1209d59c1d892966703ce2db8802c1f1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/1534e3b5d2b8bb234b2147c648fca84a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/1534e3b5d2b8bb234b2147c648fca84a
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/1534e3b5d2b8bb234b2147c648fca84a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/21f622c1d71e934137518ee35b0a5ab5 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/21f622c1d71e934137518ee35b0a5ab5
new file mode 100644
index 0000000000000000000000000000000000000000..cc8002f9d957520e5ef3e3b2f278853f80da18aa
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/21f622c1d71e934137518ee35b0a5ab5
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5ff037540eaed329f86116ffdce6d3879001d1134b316691be0e1dee6d716977
+size 227787
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/4347bf01b3e33259eb710368c602b362 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/4347bf01b3e33259eb710368c602b362
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/4347bf01b3e33259eb710368c602b362 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/471750ed5e684ae4a9b1bcc230bdc86c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/471750ed5e684ae4a9b1bcc230bdc86c
new file mode 100644
index 0000000000000000000000000000000000000000..f28d59e1f9187cfd65b688b4bceb57a81f0acdcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/471750ed5e684ae4a9b1bcc230bdc86c differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/4e6388aa3ff4b86e86928112afe80170 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/4e6388aa3ff4b86e86928112afe80170
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/4e6388aa3ff4b86e86928112afe80170 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/b6c6addfbd2073713b8db261bdef76a4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/b6c6addfbd2073713b8db261bdef76a4
new file mode 100644
index 0000000000000000000000000000000000000000..8f7eb06ce1db779b0fc3f78e94699cd2e35bface
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/b6c6addfbd2073713b8db261bdef76a4
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:830201e6b7683d4b426d21176f09f22b6701c6b270e29d7f471049cee8c8b03b
+size 503632
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/d7848953a807c2a01d11df98a0236e18 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/d7848953a807c2a01d11df98a0236e18
new file mode 100644
index 0000000000000000000000000000000000000000..c0ab9f3080484ca41581c1c962a6c54b2d9f05b3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/d7848953a807c2a01d11df98a0236e18 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..20ea6f10f8001814cc32b5fb205cdf4046ee8735
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..a270f58ba89500d4aaa917c20c774f3511ea7ab9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580454833248075, "commit_timestamp_nsecs": 1754580454989145396, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/d/baefbe13d8ed4ae1162f4f0ef9a85ad0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/d/baefbe13d8ed4ae1162f4f0ef9a85ad0
new file mode 100644
index 0000000000000000000000000000000000000000..3bd68211ceaee7fbf7ff3340a6e5a2e61239be7c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/d/baefbe13d8ed4ae1162f4f0ef9a85ad0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d9563331cd0e8277279f423ef0ddae71ab7f11cd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/014ea0658d769ccf2260d1d3634e9a7b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/014ea0658d769ccf2260d1d3634e9a7b
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/014ea0658d769ccf2260d1d3634e9a7b differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/66c8e6428f49043f7137102f8fcf83c5 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/66c8e6428f49043f7137102f8fcf83c5
new file mode 100644
index 0000000000000000000000000000000000000000..a362894f7e61c7bd6f8aeb2fff5190de4297065c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/66c8e6428f49043f7137102f8fcf83c5
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ae6cbf796388d502c40fe0c354fe6f882633cabc6a1ec189486340af1ddd2971
+size 261813
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/a609d53a78843c35340b79948d4388fd b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/a609d53a78843c35340b79948d4388fd
new file mode 100644
index 0000000000000000000000000000000000000000..283abb2b0202a3aa40fb30ce2628bad5aec80978
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/a609d53a78843c35340b79948d4388fd differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/a8d5d94c4c2ae474351e57b95ae726f3 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/a8d5d94c4c2ae474351e57b95ae726f3
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/a8d5d94c4c2ae474351e57b95ae726f3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/aeb2e5f5a9c7d50704a7594b58ba3c22 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/aeb2e5f5a9c7d50704a7594b58ba3c22
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/aeb2e5f5a9c7d50704a7594b58ba3c22 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/c07ecb7d151e2ac360dd65cdab891a7a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/c07ecb7d151e2ac360dd65cdab891a7a
new file mode 100644
index 0000000000000000000000000000000000000000..978bc576be0f1f8e97e4b40055d6d2e3d53063d6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/c07ecb7d151e2ac360dd65cdab891a7a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:409c1c6ab95531ec9f3166e28e89817aecb4652a30881cc3145f06b1681a5dc7
+size 439219
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/d6b65be93548e56f1d377c408ba5c9f7 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/d6b65be93548e56f1d377c408ba5c9f7
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/d6b65be93548e56f1d377c408ba5c9f7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/e11733936832b5cdc97066bad8a1e31e b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/e11733936832b5cdc97066bad8a1e31e
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/e11733936832b5cdc97066bad8a1e31e differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4448af109a829cc63109018a7e39db3847769f99
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..09b6b107fff178305b91b1ce258b3b8684c1d109
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580489331728381, "commit_timestamp_nsecs": 1754580489483945610, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/d/b2a5a938e04ddae808ba651d8ea1817c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/d/b2a5a938e04ddae808ba651d8ea1817c
new file mode 100644
index 0000000000000000000000000000000000000000..e6d31a6cee83e1669e615dfe929bb8e12d23d317
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/d/b2a5a938e04ddae808ba651d8ea1817c differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d7823ba10e54fa22b6b37f511acdc98aae45ca7c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/11aea1f2606a075ca9a87b7260db7d51 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/11aea1f2606a075ca9a87b7260db7d51
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/11aea1f2606a075ca9a87b7260db7d51 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/462504cda8ccfdae50fa8bc6e3a457b9 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/462504cda8ccfdae50fa8bc6e3a457b9
new file mode 100644
index 0000000000000000000000000000000000000000..5b2adec9dda010187a9daededc86bbcc2a923c31
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/462504cda8ccfdae50fa8bc6e3a457b9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/46a1759e0bd7a84059ec598089ae6e84 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/46a1759e0bd7a84059ec598089ae6e84
new file mode 100644
index 0000000000000000000000000000000000000000..4a186227aee7ecf6f570f84d25573e8c277fa481
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/46a1759e0bd7a84059ec598089ae6e84
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:402dff7c7873bd72da1e7bb57d00640e6aa7abdf35684d50982012fe66dc2fbd
+size 408690
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/98542ee80fc359b6097b2999ed0e8f34 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/98542ee80fc359b6097b2999ed0e8f34
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/98542ee80fc359b6097b2999ed0e8f34 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/9d717a0f76f8f5ca67e51bdb85196c0b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/9d717a0f76f8f5ca67e51bdb85196c0b
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/9d717a0f76f8f5ca67e51bdb85196c0b differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/a7c3cce902b66cf42d9ffb3d3cff8a43 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/a7c3cce902b66cf42d9ffb3d3cff8a43
new file mode 100644
index 0000000000000000000000000000000000000000..5c473a2b6b4d30a5728a1bbbea9f2acd8ab2a1fe
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/a7c3cce902b66cf42d9ffb3d3cff8a43
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0d656337d87b32460faad7009e8c6c296988603db4d651f3b0ef24e4cc60deee
+size 329891
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/aa8481d37a8dc493d0efa101e18e553a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/aa8481d37a8dc493d0efa101e18e553a
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/aa8481d37a8dc493d0efa101e18e553a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/c2e65d960f26a3b6689da0f048bc873c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/c2e65d960f26a3b6689da0f048bc873c
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/c2e65d960f26a3b6689da0f048bc873c differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/f98c500dacfd02ee886308053934cb21 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/f98c500dacfd02ee886308053934cb21
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/f98c500dacfd02ee886308053934cb21 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4a7d55aa2c896d48f8445398810e196462e43934
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..6cdb5e624cad0ff0c7780be3f0e21063c735805b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580525042149849, "commit_timestamp_nsecs": 1754580525178324479, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/d/91cf084b928c047751c24b0d7f4de991 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/d/91cf084b928c047751c24b0d7f4de991
new file mode 100644
index 0000000000000000000000000000000000000000..9a899562c006bf8de831ccf8a75c19150d01ea98
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/d/91cf084b928c047751c24b0d7f4de991 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..30d1b0cbd16548d4136a579feb6b4c560dc66731
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/11884c6fbea89e419aed4fcebbfc69b2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/11884c6fbea89e419aed4fcebbfc69b2
new file mode 100644
index 0000000000000000000000000000000000000000..972b3ad7b7315336b54764ef3f5fa7db9019acd3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/11884c6fbea89e419aed4fcebbfc69b2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/36b5551c5be1cd5404eb973730a7aaf3 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/36b5551c5be1cd5404eb973730a7aaf3
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/36b5551c5be1cd5404eb973730a7aaf3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/44ff4e8f12e8c5fb8969cf4800d7dae0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/44ff4e8f12e8c5fb8969cf4800d7dae0
new file mode 100644
index 0000000000000000000000000000000000000000..e1f307212b1535ee58f5db850ef928c6adce8ac7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/44ff4e8f12e8c5fb8969cf4800d7dae0
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e13361e97888b449dfaff85137403970fc7b97adf83955ec671e09c351fdec3a
+size 287153
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/59e4f150cda9826d01cad531d3517597 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/59e4f150cda9826d01cad531d3517597
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/59e4f150cda9826d01cad531d3517597 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/5b612214808a203a5b5ad8c849556f9e b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/5b612214808a203a5b5ad8c849556f9e
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/5b612214808a203a5b5ad8c849556f9e differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/961afdb90944ec575301d0abd70d2d15 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/961afdb90944ec575301d0abd70d2d15
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/961afdb90944ec575301d0abd70d2d15 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/e41eb6b2053ba81412c0715cccbb81ea b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/e41eb6b2053ba81412c0715cccbb81ea
new file mode 100644
index 0000000000000000000000000000000000000000..401fe519de38b1f4633f59b9599193313a3b2e61
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/e41eb6b2053ba81412c0715cccbb81ea
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1a03f9a90e53682e4fdbe0481ea848ccad89cc6a7492af044974aedb8816fc88
+size 381780
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/fcd5f862e634f6051fa6adce367af18f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/fcd5f862e634f6051fa6adce367af18f
new file mode 100644
index 0000000000000000000000000000000000000000..33620ed2a2385bbe722e697f64180469f149d915
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/fcd5f862e634f6051fa6adce367af18f differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0d161cb4cb5cc75e0ef238441afc65355c6ddc13
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..acea8cf6b9500df1e671f1f69e1b16a3df12db31
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580559758438644, "commit_timestamp_nsecs": 1754580559944446978, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/d/68f9db33a9538bee686cc7fcad82bec3 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/d/68f9db33a9538bee686cc7fcad82bec3
new file mode 100644
index 0000000000000000000000000000000000000000..cc9b9c90a39a4c75451efa811c9557cb61f036dc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/d/68f9db33a9538bee686cc7fcad82bec3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d2cac69450606332a48830970b39d1d7c65229a8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/35587d76e9257b05b611ef250b0cf6dc b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/35587d76e9257b05b611ef250b0cf6dc
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/35587d76e9257b05b611ef250b0cf6dc differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/6a1173c644c01745bb4c1e77a975c416 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/6a1173c644c01745bb4c1e77a975c416
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/6a1173c644c01745bb4c1e77a975c416 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/855c65aa943fa6d0a93f05193908c0f8 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/855c65aa943fa6d0a93f05193908c0f8
new file mode 100644
index 0000000000000000000000000000000000000000..acfeca7124b0884728c9c81c8c9224f668c74598
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/855c65aa943fa6d0a93f05193908c0f8
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ca069742ed4b902ce5e6eaaf44be0c85f06cc49a22e7a84550a6b95fa9a815dc
+size 413676
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/9c62398c1396318b8bbb8fdd2fcbe148 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/9c62398c1396318b8bbb8fdd2fcbe148
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/9c62398c1396318b8bbb8fdd2fcbe148 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/bacad9195ba5187ca70d126bdc1f8e64 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/bacad9195ba5187ca70d126bdc1f8e64
new file mode 100644
index 0000000000000000000000000000000000000000..fa91368bbed43deefc93ac23475426179246cc77
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/bacad9195ba5187ca70d126bdc1f8e64 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/e4044a4214427f85748b4a5690bf6292 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/e4044a4214427f85748b4a5690bf6292
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/e4044a4214427f85748b4a5690bf6292 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/e9db3f54734f35b52ae729cfbd446ead b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/e9db3f54734f35b52ae729cfbd446ead
new file mode 100644
index 0000000000000000000000000000000000000000..4b71680d0ee96488b4412e0c73a144720fd3e8ba
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/e9db3f54734f35b52ae729cfbd446ead
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:18602e87b57ddd60bcbc16aa52c2c274cc68f36417288043ed507e119479575d
+size 321423
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/f2187ff63d982de0770aaa6741db5e1b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/f2187ff63d982de0770aaa6741db5e1b
new file mode 100644
index 0000000000000000000000000000000000000000..6abf872d70937bd1d3e08e0210d74869c0b0c812
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/f2187ff63d982de0770aaa6741db5e1b differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..6fdb58e4317c0327477d361e3a5c9a313a4a5e15
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..1821d6f92625d9b47c3126e4d6917f36f535b5cb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580594364613286, "commit_timestamp_nsecs": 1754580594530575847, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/d/ac592409ae727718b200798dac0dddf8 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/d/ac592409ae727718b200798dac0dddf8
new file mode 100644
index 0000000000000000000000000000000000000000..bae8227b411dea310b567a75de96c867ad016b9e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/d/ac592409ae727718b200798dac0dddf8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..00a25da80e8aa3f9bd5e5a3872a602cdda560c29
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/0c5fc9eb8e613fcc9bcb7211dd20c67d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/0c5fc9eb8e613fcc9bcb7211dd20c67d
new file mode 100644
index 0000000000000000000000000000000000000000..65529f6532287fad4b450ca493011b55e8bf7f92
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/0c5fc9eb8e613fcc9bcb7211dd20c67d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/3d5eae8199b45927dfd3bd19ed130d65 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/3d5eae8199b45927dfd3bd19ed130d65
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/3d5eae8199b45927dfd3bd19ed130d65 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/614433fa59a2480fde06276deb96932b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/614433fa59a2480fde06276deb96932b
new file mode 100644
index 0000000000000000000000000000000000000000..7d98383f1f9caabc2847afdaf6f96a092d1f621d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/614433fa59a2480fde06276deb96932b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c5d87c45466fc77843c1090f87d7e6911f615538b43cb42ef13bb12fe51c5116
+size 278826
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/944e606be351d41f7dc5b1bf37efcccd b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/944e606be351d41f7dc5b1bf37efcccd
new file mode 100644
index 0000000000000000000000000000000000000000..a63f91fc977f7f1d76b519747bf79f318fb46cb8
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/944e606be351d41f7dc5b1bf37efcccd
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8a15e013d56298fcdc95a0245b224b40be6662b8060fc91ec7237dbaeff6f2af
+size 460115
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/9a91444fedb5e2c7fb3491d76dc93d23 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/9a91444fedb5e2c7fb3491d76dc93d23
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/9a91444fedb5e2c7fb3491d76dc93d23 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/a4e61607a9e690ae4194dee2c200e9c0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/a4e61607a9e690ae4194dee2c200e9c0
new file mode 100644
index 0000000000000000000000000000000000000000..581f642cfea9b6f5c4ccbcdb7930595d3e85ce53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/a4e61607a9e690ae4194dee2c200e9c0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/c182ef9a8c10c118e3b330abaa6c7bd5 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/c182ef9a8c10c118e3b330abaa6c7bd5
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/c182ef9a8c10c118e3b330abaa6c7bd5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/ef12e7e655a619dd3df029721fb3a6f2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/ef12e7e655a619dd3df029721fb3a6f2
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/ef12e7e655a619dd3df029721fb3a6f2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0e035e76263f98f372e4d00b52e461f8f7980105
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..d8f6328c93b207a72c4ce21ee47ee33014849d90
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580629884565828, "commit_timestamp_nsecs": 1754580630044179811, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/d/933c6ead671a6d017c4b2186dbb56d8e b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/d/933c6ead671a6d017c4b2186dbb56d8e
new file mode 100644
index 0000000000000000000000000000000000000000..54bbce3394f4e001535cbdacc2ecd7c8ead88267
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/d/933c6ead671a6d017c4b2186dbb56d8e differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c0e251eaaab269e41f528f3e5e862532242b48ea
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/285a80af53acabb4524a036d6d61a5d1 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/285a80af53acabb4524a036d6d61a5d1
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/285a80af53acabb4524a036d6d61a5d1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/3592c1883d9c392d8d571ef602c6ffb7 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/3592c1883d9c392d8d571ef602c6ffb7
new file mode 100644
index 0000000000000000000000000000000000000000..628cfc3e8dfdefbc463a7eb7bd87c235605a9a04
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/3592c1883d9c392d8d571ef602c6ffb7
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3eed23b4afb50e4f20adf43b6d99c59f3e887e30ab686bacb7674312892a86e3
+size 379647
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/60a43c98c79c9a42e1b201747949b713 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/60a43c98c79c9a42e1b201747949b713
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/60a43c98c79c9a42e1b201747949b713 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/85a35b20eb3a795b238cf2f33056ca67 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/85a35b20eb3a795b238cf2f33056ca67
new file mode 100644
index 0000000000000000000000000000000000000000..65529f6532287fad4b450ca493011b55e8bf7f92
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/85a35b20eb3a795b238cf2f33056ca67 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/988b3e1f757f9611710028097f8df169 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/988b3e1f757f9611710028097f8df169
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/988b3e1f757f9611710028097f8df169 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/a0f43e574e44af6506d367ac2d42ba75 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/a0f43e574e44af6506d367ac2d42ba75
new file mode 100644
index 0000000000000000000000000000000000000000..fc735a905a36bd2363e73524cd2cd4e9115d0c82
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/a0f43e574e44af6506d367ac2d42ba75
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b92ccbcd11e78311babf1c4009b2911ac6f8dab834d5598b41d08903c4697cae
+size 287393
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/cf2cfec90797d568d32425fd6abf0915 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/cf2cfec90797d568d32425fd6abf0915
new file mode 100644
index 0000000000000000000000000000000000000000..eb058d67724a6c59fb7aa316c1f2495fc45af1e9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/cf2cfec90797d568d32425fd6abf0915 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/fac0acc8f6057d8acf650a72795d5dc7 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/fac0acc8f6057d8acf650a72795d5dc7
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/fac0acc8f6057d8acf650a72795d5dc7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..54813bc2357409dd8e3f96a31258729193223b99
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..e9f996f30049e24c8f3bc4ee17b14a881fffb1ff
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754579381218159698, "commit_timestamp_nsecs": 1754579381356522712, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/d/d150ba742fde5b6bb4ec8d0df218dfd2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/d/d150ba742fde5b6bb4ec8d0df218dfd2
new file mode 100644
index 0000000000000000000000000000000000000000..65c203007bfcc43818fed483c3dd08d8a3752147
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/d/d150ba742fde5b6bb4ec8d0df218dfd2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8a0c964517aae529e773193eb1a9d367ae64f25d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/1af1487fbd03dca13b80214e7318455d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/1af1487fbd03dca13b80214e7318455d
new file mode 100644
index 0000000000000000000000000000000000000000..be7555f79263c0c98e37f1a99daf76c1ed6b4468
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/1af1487fbd03dca13b80214e7318455d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/33642bf74e06884a3a89e1e7034723cd b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/33642bf74e06884a3a89e1e7034723cd
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/33642bf74e06884a3a89e1e7034723cd differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/398d655902ce2df11f1f3949f6830417 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/398d655902ce2df11f1f3949f6830417
new file mode 100644
index 0000000000000000000000000000000000000000..384464eb59e2f7a8983589599129a1d9b9fdac70
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/398d655902ce2df11f1f3949f6830417
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:517ad91e4ce556d527b35ce013d7dd60007b2efadd2561bb279b65da2667f9bc
+size 408381
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/3c56f7fc0931f7870cb43659f1542fce b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/3c56f7fc0931f7870cb43659f1542fce
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/3c56f7fc0931f7870cb43659f1542fce differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/52f68df65b5ba985e762db1a385b1843 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/52f68df65b5ba985e762db1a385b1843
new file mode 100644
index 0000000000000000000000000000000000000000..7c92fd4e61aa5e6b61880d3c7ac08c62009ee6b8
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/52f68df65b5ba985e762db1a385b1843
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c296eecd10f49c11c91f296ae5b94da37efa0baae55671e59a8fcf491c1e41e0
+size 329550
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/9c1fab23f0909afc7d193b89310fbf83 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/9c1fab23f0909afc7d193b89310fbf83
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/9c1fab23f0909afc7d193b89310fbf83 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/a429deb7e34a627040b925c374eba42a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/a429deb7e34a627040b925c374eba42a
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/a429deb7e34a627040b925c374eba42a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/c0fd424279a52ba656cc2769190538ab b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/c0fd424279a52ba656cc2769190538ab
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/c0fd424279a52ba656cc2769190538ab differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5af698878a073773d5b32e468545dfbf6094c607
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..3eaebaa2927d55b8b25792c473c191a035d81233
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580664833889003, "commit_timestamp_nsecs": 1754580664983597490, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/d/432dedad71fbf1b5f69684f542705a21 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/d/432dedad71fbf1b5f69684f542705a21
new file mode 100644
index 0000000000000000000000000000000000000000..d645b0e3553b84d6b5397ce7fd81901b473aaafd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/d/432dedad71fbf1b5f69684f542705a21 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0111ea0c644b7d605a91ac74728e05e40121e74a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/0aa4e89925570740f413145494968b70 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/0aa4e89925570740f413145494968b70
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/0aa4e89925570740f413145494968b70 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/20d8eb499cfce19fcae5d043fb408d70 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/20d8eb499cfce19fcae5d043fb408d70
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/20d8eb499cfce19fcae5d043fb408d70 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/2ef774d902184a5ab6c34f5fba6f27e9 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/2ef774d902184a5ab6c34f5fba6f27e9
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/2ef774d902184a5ab6c34f5fba6f27e9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/70cb52f98543605da67fc38a0d217741 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/70cb52f98543605da67fc38a0d217741
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/70cb52f98543605da67fc38a0d217741 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/80c7594b024b466555aee6c0127bec56 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/80c7594b024b466555aee6c0127bec56
new file mode 100644
index 0000000000000000000000000000000000000000..e945d80eef1eb738711772a5058e0dbf1c719fb6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/80c7594b024b466555aee6c0127bec56
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:af1a76fec31695c5e293ec80cf0913c2f77c8b37dc38e840dc178a5a88ea10db
+size 397907
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/c0ce8fa8d312846737795a459433b2d2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/c0ce8fa8d312846737795a459433b2d2
new file mode 100644
index 0000000000000000000000000000000000000000..1aa4392c421b7438a669259723845978ed027f22
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/c0ce8fa8d312846737795a459433b2d2
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6a00fc87b2420beee72b67e796bbaa33636f8710e75846881eac4324813ded16
+size 306496
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/e634e4a060433dd2a9cae2c96a68b344 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/e634e4a060433dd2a9cae2c96a68b344
new file mode 100644
index 0000000000000000000000000000000000000000..a722b6038c877863c9d2ffb736da9f3abf725869
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/e634e4a060433dd2a9cae2c96a68b344 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..de0831dcc09782107687c125f89b0178ee5d6b63
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..3f79148e99caed31600245e10ac27659b4f0e555
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580699462806331, "commit_timestamp_nsecs": 1754580699636626458, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/d/0694bec1b305d7bd8f7fa11bd6975020 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/d/0694bec1b305d7bd8f7fa11bd6975020
new file mode 100644
index 0000000000000000000000000000000000000000..6cb30f081231fa8930031bbc2a696b9cff31a5c2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/d/0694bec1b305d7bd8f7fa11bd6975020 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..77e0427bafeaa49b3dbaadd2c24632f5a41c44ed
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/067409c76170d05997db51ad51a93298 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/067409c76170d05997db51ad51a93298
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/067409c76170d05997db51ad51a93298 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/6f9af543e9cc92e9f7c73e2f9f9132ec b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/6f9af543e9cc92e9f7c73e2f9f9132ec
new file mode 100644
index 0000000000000000000000000000000000000000..2286e0a883a105b7eac9ebe9a3e5cc1e44d0f64b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/6f9af543e9cc92e9f7c73e2f9f9132ec
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b3de6d8779ca9dc66145b5af568292df2219cb18a248035055dbd5a14275e2d9
+size 457468
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/7600b4096436fe009dfb8a48b644a7b4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/7600b4096436fe009dfb8a48b644a7b4
new file mode 100644
index 0000000000000000000000000000000000000000..2a31377ca030aced51ef70fb458c6b5c47404cd3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/7600b4096436fe009dfb8a48b644a7b4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/a71fbac403e28c7a114da0a3a9d5e8e4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/a71fbac403e28c7a114da0a3a9d5e8e4
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/a71fbac403e28c7a114da0a3a9d5e8e4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/b3e7266483b095547a1d10df58bd9a7a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/b3e7266483b095547a1d10df58bd9a7a
new file mode 100644
index 0000000000000000000000000000000000000000..d0b57d98f363e57bd2697d230d4b2c737053b096
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/b3e7266483b095547a1d10df58bd9a7a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9e22a04b8917ac9406811a4cc9b2bd05451f7c97daf743cbb42f7d69f962787f
+size 257787
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/b46578ecfcb63edf9ccfa4db47097491 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/b46578ecfcb63edf9ccfa4db47097491
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/b46578ecfcb63edf9ccfa4db47097491 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/f98a3eaab54cc0f8926368d602c73462 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/f98a3eaab54cc0f8926368d602c73462
new file mode 100644
index 0000000000000000000000000000000000000000..5453a1374e590d2226963ed30ead0b2fc5cfd149
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/f98a3eaab54cc0f8926368d602c73462 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c5cf29c8bf3c203724422f77560fdbb3b554a22d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..19eeaa756aba38f936484e529c7c6379724f7ba7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580733820321208, "commit_timestamp_nsecs": 1754580733961210345, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/d/011cf7df872448039e2fa63449cf71e9 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/d/011cf7df872448039e2fa63449cf71e9
new file mode 100644
index 0000000000000000000000000000000000000000..9bc19684f9aabe606d400775b2cc1b327ef6c22a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/d/011cf7df872448039e2fa63449cf71e9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a716079ac59a8546ddcd90d22c870638f3f85d29
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/11845097145e69a2188b7b0e8faea939 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/11845097145e69a2188b7b0e8faea939
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/11845097145e69a2188b7b0e8faea939 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/1ebf7640934dc431bc462ff074d80858 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/1ebf7640934dc431bc462ff074d80858
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/1ebf7640934dc431bc462ff074d80858 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/527e2eb739ba13dab5647eb0c4e3e0f4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/527e2eb739ba13dab5647eb0c4e3e0f4
new file mode 100644
index 0000000000000000000000000000000000000000..906278bf37a39d2f7a113ea355c3c402bdffb09d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/527e2eb739ba13dab5647eb0c4e3e0f4
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:da73cf10a9627792992d4628fccf65e4f02a8c5be62eda9ffa7252f4150e5eb7
+size 698277
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/bd4bfb1c94207da30fb9d80df5de3036 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/bd4bfb1c94207da30fb9d80df5de3036
new file mode 100644
index 0000000000000000000000000000000000000000..5bc76d10045321b46b8be35225a8b90ffa8c669c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/bd4bfb1c94207da30fb9d80df5de3036 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/cb3cd644243877414083af7bcab35340 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/cb3cd644243877414083af7bcab35340
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/cb3cd644243877414083af7bcab35340 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/fb573b178babb36539875939539739aa b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/fb573b178babb36539875939539739aa
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/fb573b178babb36539875939539739aa differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b95514183fbc085756de050ac45c6d12a00e8982
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..98436ff7bf598ef9912ebaed5851f81b1d084661
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580768632363050, "commit_timestamp_nsecs": 1754580768762546672, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/d/17e6c77d0ec0c26e2bfd5f4dc54c24b6 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/d/17e6c77d0ec0c26e2bfd5f4dc54c24b6
new file mode 100644
index 0000000000000000000000000000000000000000..40eb2abf57ca3e3543fe3d2046fd604087910bbc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/d/17e6c77d0ec0c26e2bfd5f4dc54c24b6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a65f3987ad2613eac3e7f80e963f46b9589bb8bf
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/11cd2b4612900040cd05b50654e2a1b9 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/11cd2b4612900040cd05b50654e2a1b9
new file mode 100644
index 0000000000000000000000000000000000000000..d86b8ac8a26f3e377cbcf7dc3a5bb201cfc01852
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/11cd2b4612900040cd05b50654e2a1b9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/22b956c6107b539c20584cc44c6ee6b4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/22b956c6107b539c20584cc44c6ee6b4
new file mode 100644
index 0000000000000000000000000000000000000000..38a146815b0a97a0479fc21786865dbdc09f5633
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/22b956c6107b539c20584cc44c6ee6b4
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e292f35e380adf4eedd87b30af7ffd9d0c7f6dd7d56fad8791093f886a780bd5
+size 735880
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/2f15464539f551cb62c481d953d08bef b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/2f15464539f551cb62c481d953d08bef
new file mode 100644
index 0000000000000000000000000000000000000000..639019c6ee679c6a6b4de6cb7037cad618d453de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/2f15464539f551cb62c481d953d08bef differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/354f3b8d30478ec21f368701683e767d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/354f3b8d30478ec21f368701683e767d
new file mode 100644
index 0000000000000000000000000000000000000000..ec745d97353f97209ef50f0eb738585830219f53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/354f3b8d30478ec21f368701683e767d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/5a8cbba33d3c351b3e519f1bfd71808c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/5a8cbba33d3c351b3e519f1bfd71808c
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/5a8cbba33d3c351b3e519f1bfd71808c differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/86f8c334c12a26f768e8361b8586b567 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/86f8c334c12a26f768e8361b8586b567
new file mode 100644
index 0000000000000000000000000000000000000000..85f35e7991cac7b6fc244bb2f277438119f4e90b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/86f8c334c12a26f768e8361b8586b567 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/b09bc2d346f0b5de57ea518462727b77 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/b09bc2d346f0b5de57ea518462727b77
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/b09bc2d346f0b5de57ea518462727b77 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..80624f6060969d5674bf576c156ee1b4c9aea8fc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..be2e3dd7046598d6db18ec90d7da47e146669b7f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580802219107729, "commit_timestamp_nsecs": 1754580802359252125, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/d/5627800d4d482a3085ca584a45976965 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/d/5627800d4d482a3085ca584a45976965
new file mode 100644
index 0000000000000000000000000000000000000000..bf6a85e7e3e05088756c8e5d7bbf07dbb080f5d2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/d/5627800d4d482a3085ca584a45976965 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..166cbb7f4d32aec80c4a8d70aa1c89ae288fe94d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/54ce76c63b761012b8e2f3442ff1ae0d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/54ce76c63b761012b8e2f3442ff1ae0d
new file mode 100644
index 0000000000000000000000000000000000000000..e29f94a2800d23f5bee472479e2aa7d5d8a842c2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/54ce76c63b761012b8e2f3442ff1ae0d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/6803b43051ac4bec22e32c523f21b585 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/6803b43051ac4bec22e32c523f21b585
new file mode 100644
index 0000000000000000000000000000000000000000..2735a35619974b808f36830e60b941f17e26f5c6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/6803b43051ac4bec22e32c523f21b585
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:71a0006df4cdad33c355bd809a692060171608c1c66f17684f855e56601610bc
+size 317597
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/a6d1067f7ff17500850e303d5d1adadc b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/a6d1067f7ff17500850e303d5d1adadc
new file mode 100644
index 0000000000000000000000000000000000000000..0ffe28d2e12f527978f35941a8fedeb23a8251d7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/a6d1067f7ff17500850e303d5d1adadc
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3a0f8421ef635d9abeb273ba20872bc63cfd7f4aca94779f5741450351ca9726
+size 245311
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/bf42099566cb75900273ce81c2ff9ff3 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/bf42099566cb75900273ce81c2ff9ff3
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/bf42099566cb75900273ce81c2ff9ff3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/c7471136f185c6eeab8e5f8c525b3e34 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/c7471136f185c6eeab8e5f8c525b3e34
new file mode 100644
index 0000000000000000000000000000000000000000..968243947a537ffefb366e5c7f370bf42d08d629
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/c7471136f185c6eeab8e5f8c525b3e34
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:514569e0b89c4550b380d620175f9b16dc2ab80b98b76dd0dbd182e2e6b6ed2f
+size 140382
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/d1518c43622e4f5bfe6e73280b05633b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/d1518c43622e4f5bfe6e73280b05633b
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/d1518c43622e4f5bfe6e73280b05633b differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/ef2cfddab49138428f02c8600e4817b5 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/ef2cfddab49138428f02c8600e4817b5
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/ef2cfddab49138428f02c8600e4817b5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/f739714f207d1da0d4c3a4cd3380e296 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/f739714f207d1da0d4c3a4cd3380e296
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/f739714f207d1da0d4c3a4cd3380e296 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..846c60965048870a4638ed695cdd137a822e82df
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..6f69a656bbc916c71b6970ffb484da785c1fa742
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580837133941787, "commit_timestamp_nsecs": 1754580837275822997, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/d/7bb4962eddf81dc0fa0a4cbce7d5b8ae b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/d/7bb4962eddf81dc0fa0a4cbce7d5b8ae
new file mode 100644
index 0000000000000000000000000000000000000000..b23bc94be175a031fdce943717898fe7e05bb34d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/d/7bb4962eddf81dc0fa0a4cbce7d5b8ae differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a87e5cb2f2e6b749ef2059de4bad08ada324200f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/0437ae54528ede4dccf6c0fd68011b50 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/0437ae54528ede4dccf6c0fd68011b50
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/0437ae54528ede4dccf6c0fd68011b50 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/2af1b58309b1805503e472c1de79fc77 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/2af1b58309b1805503e472c1de79fc77
new file mode 100644
index 0000000000000000000000000000000000000000..a036237ee6f91d8f6daf6634efc2bdf8e0e33001
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/2af1b58309b1805503e472c1de79fc77
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8923d8cdb975e7650792eac23e4d7bc3d8d3b2436551fe3d5392a1256eeea790
+size 451298
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/65917fea134fe988f6faf22b5489d237 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/65917fea134fe988f6faf22b5489d237
new file mode 100644
index 0000000000000000000000000000000000000000..2ca786ecc8cbc43fbe27d6c781ffde889f1063a5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/65917fea134fe988f6faf22b5489d237 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/70829fdc66affb6d17cd7b8dca68643c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/70829fdc66affb6d17cd7b8dca68643c
new file mode 100644
index 0000000000000000000000000000000000000000..5d2522497a70419c71449301079b76b790390afa
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/70829fdc66affb6d17cd7b8dca68643c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:349f31e07e5d62e81dfffae4800fdeb7dbc24b1bb9331d674806c42f1d122c65
+size 287365
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/76c85a8f10f81bd6990dd7800d2e86ea b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/76c85a8f10f81bd6990dd7800d2e86ea
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/76c85a8f10f81bd6990dd7800d2e86ea differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/8ecc786c59baaa844f7ec810a0360f3e b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/8ecc786c59baaa844f7ec810a0360f3e
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/8ecc786c59baaa844f7ec810a0360f3e differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/a6f46e62c662cae369d8c1b243d839c0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/a6f46e62c662cae369d8c1b243d839c0
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/a6f46e62c662cae369d8c1b243d839c0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/dbc9659eafa6ac54332983e14597f6f5 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/dbc9659eafa6ac54332983e14597f6f5
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/dbc9659eafa6ac54332983e14597f6f5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..ac96ec4d4883cdc87e05a0d05a58d3e3769c7c31
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..5c84af6be675a889ef0802ad5043565274d73787
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580870550853896, "commit_timestamp_nsecs": 1754580870729994279, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/d/48dfa1c83fe39dedd65ed422aec49ec3 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/d/48dfa1c83fe39dedd65ed422aec49ec3
new file mode 100644
index 0000000000000000000000000000000000000000..01ea06d4dc25e5e26ef22ae1e136cdddc5bc8e16
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/d/48dfa1c83fe39dedd65ed422aec49ec3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b06e7d50ab2f4f69a240378c6866b7c09b1bc30e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/0ead436ea26bf0ee33064022a47725d6 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/0ead436ea26bf0ee33064022a47725d6
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/0ead436ea26bf0ee33064022a47725d6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/583f495623cf2ae71ca83daa9632d238 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/583f495623cf2ae71ca83daa9632d238
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/583f495623cf2ae71ca83daa9632d238 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/6b742d6c41d48bedb63102c84f1414d3 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/6b742d6c41d48bedb63102c84f1414d3
new file mode 100644
index 0000000000000000000000000000000000000000..d7c877ff4b50ccb147df417625b93f6125674341
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/6b742d6c41d48bedb63102c84f1414d3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/99050fd88078f9c04c76fc5a6dd7234a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/99050fd88078f9c04c76fc5a6dd7234a
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/99050fd88078f9c04c76fc5a6dd7234a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/d094d661476d7ba6cf74d6a674473ed2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/d094d661476d7ba6cf74d6a674473ed2
new file mode 100644
index 0000000000000000000000000000000000000000..6ad161ebc7101a5a6d44952928f326c1cff27001
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/d094d661476d7ba6cf74d6a674473ed2
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5d31b314e86ccc8a390a20be8a09d042acd030b64deb2338bb101223904652cc
+size 294062
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/d1ed51ac61beb0b1d04f6b9e7272be58 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/d1ed51ac61beb0b1d04f6b9e7272be58
new file mode 100644
index 0000000000000000000000000000000000000000..5782a6c7465863900f376c9d04b44f1f6d0d57b1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/d1ed51ac61beb0b1d04f6b9e7272be58
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b025d472f2038cba5b841ae5be402e63cc04c80c49b1601c7a73d35b6b481687
+size 406520
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/dd98270b2029bdf9bdade2616f538ebe b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/dd98270b2029bdf9bdade2616f538ebe
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/dd98270b2029bdf9bdade2616f538ebe differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..30ae6ebcbc506695fad1e0ab8fc03aee6588fc61
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..23fa045af8c6c8a62d68894413f61eeb7f298804
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580904990010147, "commit_timestamp_nsecs": 1754580905155578858, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/d/36f2f090c4f7a969110740dd4013a1da b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/d/36f2f090c4f7a969110740dd4013a1da
new file mode 100644
index 0000000000000000000000000000000000000000..9a4b3f7c651b3c9ea06448f622985768ce66b402
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/d/36f2f090c4f7a969110740dd4013a1da differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f4b1f51a70f78ac73f05787e3d45a80189ef3c73
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/1e416f0fae980ab8592fb9acd9148477 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/1e416f0fae980ab8592fb9acd9148477
new file mode 100644
index 0000000000000000000000000000000000000000..58fb6613dd971dc47a199a6d033dcd5f5837f761
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/1e416f0fae980ab8592fb9acd9148477 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/5944fcb123ca1ad039f9c550a14303a6 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/5944fcb123ca1ad039f9c550a14303a6
new file mode 100644
index 0000000000000000000000000000000000000000..6246923f2674db0f3b5ec2787d737b53cdc0337e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/5944fcb123ca1ad039f9c550a14303a6
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:163d06f06fcf797dff70c069b6d8aa24afabc8069c17966492f87bdb16ff4aa0
+size 313200
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/64ff45408fddfa20da11f503e885fba4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/64ff45408fddfa20da11f503e885fba4
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/64ff45408fddfa20da11f503e885fba4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/7651e135a80ba314f1073a24ecf12ccc b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/7651e135a80ba314f1073a24ecf12ccc
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/7651e135a80ba314f1073a24ecf12ccc differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/ad28a0293b5e54125e025f606d7eb075 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/ad28a0293b5e54125e025f606d7eb075
new file mode 100644
index 0000000000000000000000000000000000000000..33dcdcccaec718a10f56ba5729cf1dd97c2b2215
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/ad28a0293b5e54125e025f606d7eb075
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:416d4924bc316fffb16a5d0d68e6bc5414de6d578d49ba452e3ed9f46d175dbf
+size 355391
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/c8c213c1e5a211eeab5372fa8712c540 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/c8c213c1e5a211eeab5372fa8712c540
new file mode 100644
index 0000000000000000000000000000000000000000..4859214e6b68fa6e697de9748417fcf43be8081d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/c8c213c1e5a211eeab5372fa8712c540 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/f44ecc512aa3aea42e79e6c537ed20cb b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/f44ecc512aa3aea42e79e6c537ed20cb
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/f44ecc512aa3aea42e79e6c537ed20cb differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..012b5140f069d36c6ceca237d3fed758ac027876
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..2ce60c84af04378e6e3b1dbe49fc6a41ddc72b61
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580955442270927, "commit_timestamp_nsecs": 1754580955582410587, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/d/15911bf626a477aa546e414eb349248e b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/d/15911bf626a477aa546e414eb349248e
new file mode 100644
index 0000000000000000000000000000000000000000..428477c2d7aee7a3fc86ab02e7d178fc5159cddd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/d/15911bf626a477aa546e414eb349248e differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8fe0eb58fa13a970cbc585a1b29a6be4bfb9af4f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/12fbd7b5e42344862b16ed466987d842 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/12fbd7b5e42344862b16ed466987d842
new file mode 100644
index 0000000000000000000000000000000000000000..db0bc8693a0faff736d6ba854390fc260f91db83
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/12fbd7b5e42344862b16ed466987d842
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dda9830317e9240c9dcb6c08a99ab87cdca0da6d85624307f630c29639e4ef24
+size 330100
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/14d3bf358fcb3badf34c5c37ba1be1ff b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/14d3bf358fcb3badf34c5c37ba1be1ff
new file mode 100644
index 0000000000000000000000000000000000000000..7d20d6cf25e93e7b4f8078e484d8c358471e8c87
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/14d3bf358fcb3badf34c5c37ba1be1ff
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:db4037d7a833b014e0743927154a8c8ebe8ff8181503b96ef82ec8c3f8be4f1f
+size 370894
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/22c7cd04eafefc1d93139eed375bb1c0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/22c7cd04eafefc1d93139eed375bb1c0
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/22c7cd04eafefc1d93139eed375bb1c0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/2310ed40dea059b984cecdd44a839285 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/2310ed40dea059b984cecdd44a839285
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/2310ed40dea059b984cecdd44a839285 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/2bd23ef00577406f78875f79e2c720c5 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/2bd23ef00577406f78875f79e2c720c5
new file mode 100644
index 0000000000000000000000000000000000000000..49c1b88ef51f9f868cc2a2c185b7ba9b48c4f4e6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/2bd23ef00577406f78875f79e2c720c5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/300fcf395856319412efee6a6d2d6900 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/300fcf395856319412efee6a6d2d6900
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/300fcf395856319412efee6a6d2d6900 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/7295ea97af4c5714648b92a880ec4f7f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/7295ea97af4c5714648b92a880ec4f7f
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/7295ea97af4c5714648b92a880ec4f7f differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/f1a6ff0ad04cf96618ecadadfa702e07 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/f1a6ff0ad04cf96618ecadadfa702e07
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/f1a6ff0ad04cf96618ecadadfa702e07 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c152449ad7192b1a792087bd9c00b79cac8d9849
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..798274139df7b35c944a6bc7899c44fbad382500
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754580989838224019, "commit_timestamp_nsecs": 1754580989980889869, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/d/bf034692870c9859490ffc823c0b432b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/d/bf034692870c9859490ffc823c0b432b
new file mode 100644
index 0000000000000000000000000000000000000000..2ac405eb47e8a207455a6a0e703a6f88c7d59d4b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/d/bf034692870c9859490ffc823c0b432b differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5206845755b2b75330d77ea1e0b22267946238bd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/054867bbc6a7cc783e02382c17b9a3b4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/054867bbc6a7cc783e02382c17b9a3b4
new file mode 100644
index 0000000000000000000000000000000000000000..4859214e6b68fa6e697de9748417fcf43be8081d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/054867bbc6a7cc783e02382c17b9a3b4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/0ce40dcdbf58dd2f61c2bfd4bc814f09 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/0ce40dcdbf58dd2f61c2bfd4bc814f09
new file mode 100644
index 0000000000000000000000000000000000000000..9a235b099f98a92eb1c50870b1b49ffe73f2bdcf
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/0ce40dcdbf58dd2f61c2bfd4bc814f09
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0b3f6674502b968f80e2696232fae9ae25f71c1f8709c5c7e0e92ff1d72bc21c
+size 404944
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/116b854f122b9c019f4152656ebb8051 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/116b854f122b9c019f4152656ebb8051
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/116b854f122b9c019f4152656ebb8051 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/306dfab7efa3c17f34a4c0c9c33d7202 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/306dfab7efa3c17f34a4c0c9c33d7202
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/306dfab7efa3c17f34a4c0c9c33d7202 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/5334d1b20402b460984849c5c8587202 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/5334d1b20402b460984849c5c8587202
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/5334d1b20402b460984849c5c8587202 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/843d9d8c5f42e4114457e189c0c788f8 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/843d9d8c5f42e4114457e189c0c788f8
new file mode 100644
index 0000000000000000000000000000000000000000..f7b4863d1b158b92c4a515ad607b6491452de0a2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/843d9d8c5f42e4114457e189c0c788f8
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ee5b5bd800b621cb79e62817c388a10848ffee70b8ac3528e232e0c5a230901d
+size 296036
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/a9f5fadb8a16b4af3899eb9555cf37f4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/a9f5fadb8a16b4af3899eb9555cf37f4
new file mode 100644
index 0000000000000000000000000000000000000000..ccc7bd0e9ba843f8de8acb35ae6bc36ac9a22bc8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/a9f5fadb8a16b4af3899eb9555cf37f4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0aca9ecd713d2194748dcdce42746d1362457c46
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..931c286c44b060f5726fcbe531c8a4872d9931f1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754579415644385519, "commit_timestamp_nsecs": 1754579415787894029, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/d/b1aa192d093871166dac5d3ad2d4d131 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/d/b1aa192d093871166dac5d3ad2d4d131
new file mode 100644
index 0000000000000000000000000000000000000000..e614002e25fa40475bac388a704ca3c64a3627b7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/d/b1aa192d093871166dac5d3ad2d4d131 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..cdcd457c11ea4dff6fe3e6953ffe376121c331ff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/0744827d1e1c2755906fe9be9208f0f9 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/0744827d1e1c2755906fe9be9208f0f9
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/0744827d1e1c2755906fe9be9208f0f9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/36226cde082f851cfda9844757fdce2a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/36226cde082f851cfda9844757fdce2a
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/36226cde082f851cfda9844757fdce2a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/53bbbb4aa7e0d38a9a10218a7eec55bf b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/53bbbb4aa7e0d38a9a10218a7eec55bf
new file mode 100644
index 0000000000000000000000000000000000000000..57fd15311b5f7765b60fafa4cf8a67ef9093a9a5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/53bbbb4aa7e0d38a9a10218a7eec55bf
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a8715b5d3ac70cfad5911c34e91d707d54a24afdc962c1860306938a6d9b8c56
+size 505015
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/61e75cb89514c87b2a2719ac852c4769 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/61e75cb89514c87b2a2719ac852c4769
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/61e75cb89514c87b2a2719ac852c4769 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/73cd9ed172392c93ffb36640c6073fd4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/73cd9ed172392c93ffb36640c6073fd4
new file mode 100644
index 0000000000000000000000000000000000000000..c26a8777022d314b8d9130857a288e769bd0b652
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/73cd9ed172392c93ffb36640c6073fd4
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:344dfc2e797259e0265f7cc47368eeff2db96117c51e96f248b1eb639eea23e1
+size 193515
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/cd80ccd6703f50323d4e2d32bf1fe9d4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/cd80ccd6703f50323d4e2d32bf1fe9d4
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/cd80ccd6703f50323d4e2d32bf1fe9d4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/d9b53f73c0edb7e29120af755748dd7d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/d9b53f73c0edb7e29120af755748dd7d
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/d9b53f73c0edb7e29120af755748dd7d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/de91bc1c034e10c71441067ae97e1490 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/de91bc1c034e10c71441067ae97e1490
new file mode 100644
index 0000000000000000000000000000000000000000..5d0e222321bb812081e936f6a02e842b34bebf77
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/de91bc1c034e10c71441067ae97e1490 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..79e925f18b5a21d48483dea01591e470b2d1ef77
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..63504ed2c8f66ca2be521f1abe3940ca90e5395c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581023754228102, "commit_timestamp_nsecs": 1754581023932724680, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/d/5e8ca80870b67a9f2084e324d69995b1 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/d/5e8ca80870b67a9f2084e324d69995b1
new file mode 100644
index 0000000000000000000000000000000000000000..c28fd42368d15e3a830f5df74b336cb8b0e6360a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/d/5e8ca80870b67a9f2084e324d69995b1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..7b2e49d9f6c355ca701b19f03213efacaf4653ea
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/1528c63698ccb626e299692796656645 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/1528c63698ccb626e299692796656645
new file mode 100644
index 0000000000000000000000000000000000000000..0e302a2adf02f01376836f5e7df6955a39b6140b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/1528c63698ccb626e299692796656645
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a3ea547af0c99aab87d69091d22e9c54ea998294b13c812bda8261a804bb0398
+size 443040
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/43b20833bd98982a019415f77cb361fe b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/43b20833bd98982a019415f77cb361fe
new file mode 100644
index 0000000000000000000000000000000000000000..f67a95ae94266e981abbbc9d833cc636d26f91fd
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/43b20833bd98982a019415f77cb361fe
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3394b2247ef87f5f30bb49319c96861492776bcebec45d214c5d1d698b039325
+size 296016
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/55a46d34676f3f9e10e6aa751b45c191 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/55a46d34676f3f9e10e6aa751b45c191
new file mode 100644
index 0000000000000000000000000000000000000000..7fee846d2a558fd485a32231a33b420678c2972f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/55a46d34676f3f9e10e6aa751b45c191 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/a340c69360d73ecb9b84b04a847ad552 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/a340c69360d73ecb9b84b04a847ad552
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/a340c69360d73ecb9b84b04a847ad552 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/b8091ffc61504af7512793c5158c6dc9 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/b8091ffc61504af7512793c5158c6dc9
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/b8091ffc61504af7512793c5158c6dc9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/d14ad0c934741c56cb6309f47f67447a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/d14ad0c934741c56cb6309f47f67447a
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/d14ad0c934741c56cb6309f47f67447a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/de99e95e4d1865206c1f488e85db1583 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/de99e95e4d1865206c1f488e85db1583
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/de99e95e4d1865206c1f488e85db1583 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/f42d7965b017b690856e8cc831067f6f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/f42d7965b017b690856e8cc831067f6f
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/f42d7965b017b690856e8cc831067f6f differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4d810ad8dd71a8ae48b7fd87cd739cdc9f5ef826
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..20a0ed242fdaaa4caf2d467d35915ab802f39837
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581058462416235, "commit_timestamp_nsecs": 1754581058640616086, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/d/99c0856bfd8ab79a60944db843748fc4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/d/99c0856bfd8ab79a60944db843748fc4
new file mode 100644
index 0000000000000000000000000000000000000000..584c69d38a379198b4057dbb573013b77fd659b3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/d/99c0856bfd8ab79a60944db843748fc4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9e8ea88c19f71c2e0270d0f7b57614b07bb978e0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/1a0abe2aace2dfa409fd91e983ac21d2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/1a0abe2aace2dfa409fd91e983ac21d2
new file mode 100644
index 0000000000000000000000000000000000000000..d8bb78b5102af6cb375c3b4b0d3b86effa17dfe6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/1a0abe2aace2dfa409fd91e983ac21d2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/2035e50ba15fa1c30cc18dd8555de790 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/2035e50ba15fa1c30cc18dd8555de790
new file mode 100644
index 0000000000000000000000000000000000000000..771740f595f237c4051c6eb489d9538fedf7ddba
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/2035e50ba15fa1c30cc18dd8555de790
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4722bb92c9e1aa856937a09776fe9e78dfcb7b18ef5761c4896ca23bf5cecae5
+size 330046
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/355114784848ee62a728ec57f18e0636 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/355114784848ee62a728ec57f18e0636
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/355114784848ee62a728ec57f18e0636 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/6823bc19edf217bda01a6733d1dab71d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/6823bc19edf217bda01a6733d1dab71d
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/6823bc19edf217bda01a6733d1dab71d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/7a7919d9e0f09622649898536176fff8 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/7a7919d9e0f09622649898536176fff8
new file mode 100644
index 0000000000000000000000000000000000000000..85f35e7991cac7b6fc244bb2f277438119f4e90b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/7a7919d9e0f09622649898536176fff8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/8ece03f93432f1cd8235675a39f8b09a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/8ece03f93432f1cd8235675a39f8b09a
new file mode 100644
index 0000000000000000000000000000000000000000..fe930f7dbcc9f136e996a78c9ec3802072f27270
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/8ece03f93432f1cd8235675a39f8b09a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:405acfc421a6a4a0975132be02d2a8716dd7f20c8c69077c728c6fa545e73ccc
+size 408664
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/a3dc52c4550d9eb4d36124b4102886d2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/a3dc52c4550d9eb4d36124b4102886d2
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/a3dc52c4550d9eb4d36124b4102886d2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/abceadf27a60755f0bcb85aa0821d00b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/abceadf27a60755f0bcb85aa0821d00b
new file mode 100644
index 0000000000000000000000000000000000000000..4859214e6b68fa6e697de9748417fcf43be8081d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/abceadf27a60755f0bcb85aa0821d00b differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0909d4324fc0c99ea41a442cf99dcc54b949b7ad
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..6a9f9082f34bd18e4a0be141fee962140ef4fe77
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581094290276821, "commit_timestamp_nsecs": 1754581094460816858, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/d/81fc644d373ec2a9db7109f384cafae0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/d/81fc644d373ec2a9db7109f384cafae0
new file mode 100644
index 0000000000000000000000000000000000000000..652fd81cee704be0a86d9049d23d4ccd984f6de4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/d/81fc644d373ec2a9db7109f384cafae0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..bd1a806a639bad5ec5a6b9855b79c1da4bd1c25e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/07af63f87b42267b32ccb3f38851eb37 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/07af63f87b42267b32ccb3f38851eb37
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/07af63f87b42267b32ccb3f38851eb37 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/1656c0f1f582d4a3b90e3b48922d36a2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/1656c0f1f582d4a3b90e3b48922d36a2
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/1656c0f1f582d4a3b90e3b48922d36a2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/41d07ebd0de7fb4b2ed7ff1679c80847 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/41d07ebd0de7fb4b2ed7ff1679c80847
new file mode 100644
index 0000000000000000000000000000000000000000..392b8b5c789430b11156f76259f077c8390ee472
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/41d07ebd0de7fb4b2ed7ff1679c80847 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/4dbe4b99aecc67261886bcb0b5bb4eb5 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/4dbe4b99aecc67261886bcb0b5bb4eb5
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/4dbe4b99aecc67261886bcb0b5bb4eb5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/a3462313446182bad856f93d82425ecf b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/a3462313446182bad856f93d82425ecf
new file mode 100644
index 0000000000000000000000000000000000000000..49f01c1146e5af15530b32d9273424e824a0a621
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/a3462313446182bad856f93d82425ecf
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f6b9506dc427ff81b957541b7f5848f3846cff801b7edb354734368959f01eb5
+size 368674
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/b7b09c15a83f58216fe3f736a8782b71 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/b7b09c15a83f58216fe3f736a8782b71
new file mode 100644
index 0000000000000000000000000000000000000000..12f73997d415fb01ace1f25d3e4a08688fadb5b6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/b7b09c15a83f58216fe3f736a8782b71
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:332d9596acaeb359964acab670655ca539f8c79318fa2a122f829ae3064007be
+size 296018
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/c69e7fe3440ea132a3d21a129c32c0c8 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/c69e7fe3440ea132a3d21a129c32c0c8
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/c69e7fe3440ea132a3d21a129c32c0c8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..be88d4771adb3df95da97c20e5ccaa907992aeb0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..c72f242475b294074b2703966622e5c5f938e7f6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581128835257413, "commit_timestamp_nsecs": 1754581128976617581, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/d/94a472f8d5a721270d70d284c5604cf4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/d/94a472f8d5a721270d70d284c5604cf4
new file mode 100644
index 0000000000000000000000000000000000000000..fdae37e1b35458987c107d733e45b50c576ce4c8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/d/94a472f8d5a721270d70d284c5604cf4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..972f16651dfbbbec86a67e9549113046d57aeadf
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/1956ecdd20d1c9bf14412170fed5083f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/1956ecdd20d1c9bf14412170fed5083f
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/1956ecdd20d1c9bf14412170fed5083f differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/2fc25c971b42cc5e15814d2e8ad32d3e b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/2fc25c971b42cc5e15814d2e8ad32d3e
new file mode 100644
index 0000000000000000000000000000000000000000..9d9585c5158d6651dc13f8104261a6a2f413df25
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/2fc25c971b42cc5e15814d2e8ad32d3e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:885a09e98eacf4ef9eeb9cb9a9e243785b2aac17199bf0d5ea345087fd45274e
+size 227872
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/345a8fdce94897c08239ef861b4da438 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/345a8fdce94897c08239ef861b4da438
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/345a8fdce94897c08239ef861b4da438 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/38aad302f5ad0f4c5a175f2e48062cb1 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/38aad302f5ad0f4c5a175f2e48062cb1
new file mode 100644
index 0000000000000000000000000000000000000000..82c92d3710a6510dd1864f29a81215269c8dc488
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/38aad302f5ad0f4c5a175f2e48062cb1
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9800fe1802946d910755c59d841e815a3eef7e669177744f68b47c7f47abe161
+size 439478
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/3cb1474bcb74fde7ec71fd1a2f0214fe b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/3cb1474bcb74fde7ec71fd1a2f0214fe
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/3cb1474bcb74fde7ec71fd1a2f0214fe differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/6b36bb4bf5661034e8537b088a61304f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/6b36bb4bf5661034e8537b088a61304f
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/6b36bb4bf5661034e8537b088a61304f differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/6dea575bc2a95d6b58131fe76e3b3c76 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/6dea575bc2a95d6b58131fe76e3b3c76
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/6dea575bc2a95d6b58131fe76e3b3c76 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/c2c7d7ff06ffb690386167ae5dd89018 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/c2c7d7ff06ffb690386167ae5dd89018
new file mode 100644
index 0000000000000000000000000000000000000000..b93fa1395e4e258e8670308b706c376236bd04a7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/c2c7d7ff06ffb690386167ae5dd89018 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c53ddbc478b930eaccee1f24c3c39cca4930adde
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..bddff4a43b91b8f890bd3b3f8623d3c8065c0508
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581163035488627, "commit_timestamp_nsecs": 1754581163174540835, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/d/192decdc1fe89291526cf7bda3ccbbd2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/d/192decdc1fe89291526cf7bda3ccbbd2
new file mode 100644
index 0000000000000000000000000000000000000000..3a57ef084e50459744b6aaaf18b549476e5012a8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/d/192decdc1fe89291526cf7bda3ccbbd2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..980e61930f7b4bdcbbf32c64a2da946a6b6b2f95
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/0509e67757555fc46418e19d62038362 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/0509e67757555fc46418e19d62038362
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/0509e67757555fc46418e19d62038362 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/0e22e5313a642da2f22b90a04dd70218 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/0e22e5313a642da2f22b90a04dd70218
new file mode 100644
index 0000000000000000000000000000000000000000..c86c2c1cdcdbf7f076a411fb1d69a3e7ea03c893
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/0e22e5313a642da2f22b90a04dd70218 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/c26c2636f332e4a9c6f09ab5876bb86d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/c26c2636f332e4a9c6f09ab5876bb86d
new file mode 100644
index 0000000000000000000000000000000000000000..04637300c3f03d759706d0a8d630915beac4e6e9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/c26c2636f332e4a9c6f09ab5876bb86d
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4aacccca50fb2f8e477ddcf8809cf0beb6002c80fc7774bd541c099c39839921
+size 304694
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/d7de2afd74d5bd011755209c1a8bfe91 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/d7de2afd74d5bd011755209c1a8bfe91
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/d7de2afd74d5bd011755209c1a8bfe91 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/f053ab6c5b6da0bfe49cc05ed80f819b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/f053ab6c5b6da0bfe49cc05ed80f819b
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/f053ab6c5b6da0bfe49cc05ed80f819b differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/f80b2ca144369e9ee4da8cb6f1ceb22f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/f80b2ca144369e9ee4da8cb6f1ceb22f
new file mode 100644
index 0000000000000000000000000000000000000000..9d10a645a62ac11700d63449a29e0299097ce20e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/f80b2ca144369e9ee4da8cb6f1ceb22f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:110593106b2cd0ce418fa48b68ee953c157ae8beac9e9edeea9ef3dbb6fc794b
+size 396009
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/ffcb782b6cd80eb08de41941ec503137 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/ffcb782b6cd80eb08de41941ec503137
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/ffcb782b6cd80eb08de41941ec503137 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..977ec67f86769ea781b7a3c1de6627fe66b8e8bf
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..23d19b42be8bad2d3885a042e4e483f9bfc866e3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581197421147434, "commit_timestamp_nsecs": 1754581197562973615, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/d/f12f77239302db0f63d4648e6e177ebc b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/d/f12f77239302db0f63d4648e6e177ebc
new file mode 100644
index 0000000000000000000000000000000000000000..9ce5addc8ef369530fbcae858f5f537ba88248d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/d/f12f77239302db0f63d4648e6e177ebc differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a9dcc4122d8a137f16ef2171d8f089c22896c246
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/243ee0113f157765ed86a63488113b0d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/243ee0113f157765ed86a63488113b0d
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/243ee0113f157765ed86a63488113b0d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/2dae31892af474376170f1041a2ba118 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/2dae31892af474376170f1041a2ba118
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/2dae31892af474376170f1041a2ba118 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/8a19a8331af228035d54e9825e3160a9 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/8a19a8331af228035d54e9825e3160a9
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/8a19a8331af228035d54e9825e3160a9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/b49d86900c913610fd3d76dd94dda5ac b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/b49d86900c913610fd3d76dd94dda5ac
new file mode 100644
index 0000000000000000000000000000000000000000..3af55a14d54264b54c34e0512e16400a673f3946
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/b49d86900c913610fd3d76dd94dda5ac differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/c491dc4324575582a546699562c877e8 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/c491dc4324575582a546699562c877e8
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/c491dc4324575582a546699562c877e8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/cc850217b7dc32bcbc099666e4c13e80 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/cc850217b7dc32bcbc099666e4c13e80
new file mode 100644
index 0000000000000000000000000000000000000000..43472df81af9058885d5926362136bffa1d30209
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/cc850217b7dc32bcbc099666e4c13e80 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/e6649e130c9faba6cfacbccd93b6d87b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/e6649e130c9faba6cfacbccd93b6d87b
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/e6649e130c9faba6cfacbccd93b6d87b differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/f7a03761716f03aace07a1a022468c54 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/f7a03761716f03aace07a1a022468c54
new file mode 100644
index 0000000000000000000000000000000000000000..bfb2abefa663c73ee5d25d2844e05c47724a4e9d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/f7a03761716f03aace07a1a022468c54
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3277ec303f9e4252f063f48b5d6e53dab03e2d929d34f725ed519f047f867936
+size 698428
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..510b836a4b24e9295606235fd0e4788204bb85a5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..fd2aa00c0aefdba42134262e18c5d7d1dd48dd36
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581232289774537, "commit_timestamp_nsecs": 1754581232464672193, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/d/acfea08c348f5b4ee4fa464ba78cd2c4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/d/acfea08c348f5b4ee4fa464ba78cd2c4
new file mode 100644
index 0000000000000000000000000000000000000000..0b2edede436a17aa87fbe941243051446f9c3e65
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/d/acfea08c348f5b4ee4fa464ba78cd2c4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..2892ac8a980f79d0a335a22c57aadb91bb7263fb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/417d497dff2d0b54f6fb96958c159cec b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/417d497dff2d0b54f6fb96958c159cec
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/417d497dff2d0b54f6fb96958c159cec differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/4f35baf1f300feb9dfc279520228c651 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/4f35baf1f300feb9dfc279520228c651
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/4f35baf1f300feb9dfc279520228c651 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/5ba77ef4792c8df1d1f1ce4674e0c12d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/5ba77ef4792c8df1d1f1ce4674e0c12d
new file mode 100644
index 0000000000000000000000000000000000000000..ce14d07d9bf7c18e728c3fd83bbf5c9b711a1970
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/5ba77ef4792c8df1d1f1ce4674e0c12d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/6de3bda0d9cc9da54179fc363882cd97 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/6de3bda0d9cc9da54179fc363882cd97
new file mode 100644
index 0000000000000000000000000000000000000000..729a24e653992c502078116af39fc8f30187a088
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/6de3bda0d9cc9da54179fc363882cd97
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0aee8a81d7b5b346cbc6c06880c72ce63e76930b419850bc3228cea8da8cd773
+size 405115
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/7f3ba96b731bf3a5149d4008d4cf8298 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/7f3ba96b731bf3a5149d4008d4cf8298
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/7f3ba96b731bf3a5149d4008d4cf8298 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/c932cdad1330d3d1b395f0e4cf1518d2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/c932cdad1330d3d1b395f0e4cf1518d2
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/c932cdad1330d3d1b395f0e4cf1518d2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/dd2ba7d6b8e271b2585276a57c6549e2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/dd2ba7d6b8e271b2585276a57c6549e2
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/dd2ba7d6b8e271b2585276a57c6549e2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/f75393ba9140e39f96ab07e137f4ee09 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/f75393ba9140e39f96ab07e137f4ee09
new file mode 100644
index 0000000000000000000000000000000000000000..33f036b6a5448de02e73a97b4b6507485f75e4f7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/f75393ba9140e39f96ab07e137f4ee09
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c552c39fcabd45a39fbe7986e93446919629c79675c074799c47eda55d69cfa0
+size 296099
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..ab4046b27f75b0907f2d0074efbb296c308dd1a1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..40115cf3b3502feef9f9983a961227c148e58a22
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581266627875757, "commit_timestamp_nsecs": 1754581266770693933, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/d/3b0c315fd4750b94b5b2870221a83139 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/d/3b0c315fd4750b94b5b2870221a83139
new file mode 100644
index 0000000000000000000000000000000000000000..8c7b4ed60f6156ea5a6536fba115791ca79d7d92
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/d/3b0c315fd4750b94b5b2870221a83139 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..54c00f8f749bb8e13471ecf3edfebda963bc0fd8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/16ce0534d90b5dbd1e8e20db3cf8ed09 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/16ce0534d90b5dbd1e8e20db3cf8ed09
new file mode 100644
index 0000000000000000000000000000000000000000..fa72838350368bfe4c4bc5e0dcf21b31da9d5f0e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/16ce0534d90b5dbd1e8e20db3cf8ed09
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:70170e5047981a78d82076e375d5316dbcb789e9b84b2fb8d657e6ce05b97686
+size 296100
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/45542df2d5f7a6908aa90c66fdd768a6 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/45542df2d5f7a6908aa90c66fdd768a6
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/45542df2d5f7a6908aa90c66fdd768a6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/4c024a86616391fb9789592a80e8be0c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/4c024a86616391fb9789592a80e8be0c
new file mode 100644
index 0000000000000000000000000000000000000000..e5fa97a9616fa9825892b01b0b16e4142d58b0e8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/4c024a86616391fb9789592a80e8be0c differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/4f99fc56a0a886bf63e5e27108967943 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/4f99fc56a0a886bf63e5e27108967943
new file mode 100644
index 0000000000000000000000000000000000000000..f28d59e1f9187cfd65b688b4bceb57a81f0acdcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/4f99fc56a0a886bf63e5e27108967943 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/56f7721c3123000c1fab62f0c50fbd45 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/56f7721c3123000c1fab62f0c50fbd45
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/56f7721c3123000c1fab62f0c50fbd45 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/620ff3313b97b35df84d27f115788462 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/620ff3313b97b35df84d27f115788462
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/620ff3313b97b35df84d27f115788462 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/ada77f7166f58a478973368d68e768e9 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/ada77f7166f58a478973368d68e768e9
new file mode 100644
index 0000000000000000000000000000000000000000..65529f6532287fad4b450ca493011b55e8bf7f92
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/ada77f7166f58a478973368d68e768e9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/d69c602879cf589e15423a7239afbe24 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/d69c602879cf589e15423a7239afbe24
new file mode 100644
index 0000000000000000000000000000000000000000..151debe0656dfac9b8f7033414f717b21682a36f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/d69c602879cf589e15423a7239afbe24
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:228cabb2b3db67ab6efa7e5f5660884c234bdcbb22fd9c687ea08acb8199eabe
+size 401299
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..70f12246566cfbfe3a5e9febf61d50487f8e3978
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..c816cebcb9b8945a77905d9234b2717a12aee63e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581301926802588, "commit_timestamp_nsecs": 1754581302072651100, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/d/3f9c0a816fb3bc36a5141bb80c205690 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/d/3f9c0a816fb3bc36a5141bb80c205690
new file mode 100644
index 0000000000000000000000000000000000000000..e63cd4bd8730574e0b796fa1334972e7c8feeb9f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/d/3f9c0a816fb3bc36a5141bb80c205690 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0631758fc25022d262085d3a9b1c4262e7c106d0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/229821761392ff174ee95f150468596d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/229821761392ff174ee95f150468596d
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/229821761392ff174ee95f150468596d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/235df01f5b95ebd10679c38ef704fe84 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/235df01f5b95ebd10679c38ef704fe84
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/235df01f5b95ebd10679c38ef704fe84 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/23edb7474760c1abe5068b74b1ccfd4b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/23edb7474760c1abe5068b74b1ccfd4b
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/23edb7474760c1abe5068b74b1ccfd4b differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/39d924bd2ce7d26f0e30aefae8196467 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/39d924bd2ce7d26f0e30aefae8196467
new file mode 100644
index 0000000000000000000000000000000000000000..c8da8e8992b6dd4b4299f011356da74c5051f12e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/39d924bd2ce7d26f0e30aefae8196467
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:51b6b80b04b786ed38e8e980a97496c318def539989ab8576839736805308a4a
+size 420359
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/4da0d8495c5bfddc444dd1619a1848af b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/4da0d8495c5bfddc444dd1619a1848af
new file mode 100644
index 0000000000000000000000000000000000000000..0184822e4363c94348adb4ac1f1eae46e03c9aa1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/4da0d8495c5bfddc444dd1619a1848af differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/83ae15e97ecb299d6fd8fe957b314981 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/83ae15e97ecb299d6fd8fe957b314981
new file mode 100644
index 0000000000000000000000000000000000000000..e4968f69f43cee2a7f762fb551e6575e44b7cddf
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/83ae15e97ecb299d6fd8fe957b314981
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:70a2ca6f530441d853f971c3b4b460a5868032ec63cc26427db2886180e756cf
+size 278941
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/ac8cc2c7d12fa666eee10819758a3379 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/ac8cc2c7d12fa666eee10819758a3379
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/ac8cc2c7d12fa666eee10819758a3379 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/ec4cd95d2d2022b11a17b558a1beddc6 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/ec4cd95d2d2022b11a17b558a1beddc6
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/ec4cd95d2d2022b11a17b558a1beddc6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b827e6610e3d927bd51acda4f0cb44e44ecd6600
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..bd0783285b690eaa96f8970dec5987757dd606c4
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581336318211069, "commit_timestamp_nsecs": 1754581336463622345, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/d/cca67c09364c589bb6a32a5180b42963 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/d/cca67c09364c589bb6a32a5180b42963
new file mode 100644
index 0000000000000000000000000000000000000000..b6203241be64cccccc184c00c4d2fe1dd556d30e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/d/cca67c09364c589bb6a32a5180b42963 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d5eac3c912367a86dbf0f8aa4443e55cf4024287
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/35bb756be9ad82a23d2e8630c17151dd b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/35bb756be9ad82a23d2e8630c17151dd
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/35bb756be9ad82a23d2e8630c17151dd differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/46bbe50bccc393a4faa144b4fdf50c98 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/46bbe50bccc393a4faa144b4fdf50c98
new file mode 100644
index 0000000000000000000000000000000000000000..efb900269f95486bc95f3743b5f3e2ec823f5152
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/46bbe50bccc393a4faa144b4fdf50c98
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:41b70e23280e6789fe8de2456f6c11f52b42a79cf8e8dfac86977d1a42098a98
+size 329898
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/583962a6f2114e3fae8ad621bf475247 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/583962a6f2114e3fae8ad621bf475247
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/583962a6f2114e3fae8ad621bf475247 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/64f7526dd2b20d830501afecca0b7d20 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/64f7526dd2b20d830501afecca0b7d20
new file mode 100644
index 0000000000000000000000000000000000000000..84f38e400bdee97fa4a7285b8c222daaa7a378b2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/64f7526dd2b20d830501afecca0b7d20 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/90d912f6bc577c4d24a430d84fe7c497 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/90d912f6bc577c4d24a430d84fe7c497
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/90d912f6bc577c4d24a430d84fe7c497 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/d653b4daa70601549dd2d102e2e28109 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/d653b4daa70601549dd2d102e2e28109
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/d653b4daa70601549dd2d102e2e28109 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/d7bf06fad06da2cfbcd4c3299a202437 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/d7bf06fad06da2cfbcd4c3299a202437
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/d7bf06fad06da2cfbcd4c3299a202437 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/dc4fc5c3c7b133695d6683aa6c74bfb9 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/dc4fc5c3c7b133695d6683aa6c74bfb9
new file mode 100644
index 0000000000000000000000000000000000000000..bd4e05139a4744fdbefa101c6b404451764a0382
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/dc4fc5c3c7b133695d6683aa6c74bfb9
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a3b78ed158eab916adc0832ed3b31461891b566f9d7c75278d04692508f0250e
+size 408831
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..7bd5a7c7ce5416e98f18a11025e9d77af570bae6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..cea907b376bd082d6a1920b6ed69463752905f3b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754579451136131346, "commit_timestamp_nsecs": 1754579451277981909, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/d/4b2209c2ce14409d1a268a4cc3324578 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/d/4b2209c2ce14409d1a268a4cc3324578
new file mode 100644
index 0000000000000000000000000000000000000000..fa8b1eb2ee871cf50e649dfc021053f4dd73ac4b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/d/4b2209c2ce14409d1a268a4cc3324578 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..2a11b84be15c1a01beccdce760db8306345b25ac
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/3e0bb1d94a17bc1198ff57e5e298f25d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/3e0bb1d94a17bc1198ff57e5e298f25d
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/3e0bb1d94a17bc1198ff57e5e298f25d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/712c606bf3a2cc247b5054fdf2e57267 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/712c606bf3a2cc247b5054fdf2e57267
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/712c606bf3a2cc247b5054fdf2e57267 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/88a18efee7916245989edfc12d1f6325 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/88a18efee7916245989edfc12d1f6325
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/88a18efee7916245989edfc12d1f6325 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/8a8e105184da775f5f26dba53a310eac b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/8a8e105184da775f5f26dba53a310eac
new file mode 100644
index 0000000000000000000000000000000000000000..7500171f44de4c31b0986d23a92dd2a334de723f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/8a8e105184da775f5f26dba53a310eac differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/9d85ffef3249f0874c304a5d71ef8097 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/9d85ffef3249f0874c304a5d71ef8097
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/9d85ffef3249f0874c304a5d71ef8097 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/c90480b2adca42150bfe86ed1b3e69b1 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/c90480b2adca42150bfe86ed1b3e69b1
new file mode 100644
index 0000000000000000000000000000000000000000..f9b75711f68192fac13881010fd004eae0e87804
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/c90480b2adca42150bfe86ed1b3e69b1
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f61f689f9ebad51395200f3e20912669b18deb54009f7929bc41bba437c94199
+size 413315
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/d8abaaddb26020b2a349b438e068670c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/d8abaaddb26020b2a349b438e068670c
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/d8abaaddb26020b2a349b438e068670c differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/e328073556e5fd67479fb6ca29eee754 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/e328073556e5fd67479fb6ca29eee754
new file mode 100644
index 0000000000000000000000000000000000000000..db05ce46035e9e1488d6da3cfad7d39909ae9c60
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/e328073556e5fd67479fb6ca29eee754
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8397e1a878294deac439139f4e50452707a63fc8e3c2cf1bd58ac503fbd78796
+size 287179
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0bf548cfefcf03fa40d78a32f113292c5f68eab1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..a9460b544e73f081b8b45fd19494fa9689f95a6d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581371056294805, "commit_timestamp_nsecs": 1754581371225739631, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/d/96054f82c723cd05109976ebbba09201 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/d/96054f82c723cd05109976ebbba09201
new file mode 100644
index 0000000000000000000000000000000000000000..cdf3ba86721b668d20546d8e7a91853a16833c80
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/d/96054f82c723cd05109976ebbba09201 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..7041d93b48c0e2e9090e7c3249dd6348e9e00132
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/1b0224e12c7f18b3139987cf8357b56b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/1b0224e12c7f18b3139987cf8357b56b
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/1b0224e12c7f18b3139987cf8357b56b differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/3e480b78d493c6b6c92f9cf7227bbcf5 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/3e480b78d493c6b6c92f9cf7227bbcf5
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/3e480b78d493c6b6c92f9cf7227bbcf5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/42fd4a2e07b8479166e8cb0defaf249c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/42fd4a2e07b8479166e8cb0defaf249c
new file mode 100644
index 0000000000000000000000000000000000000000..de7db5e2a1c885b23af55c0c5066c8bcbdf8a0d3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/42fd4a2e07b8479166e8cb0defaf249c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:25b6c6301fadf1f47ccc8b449880c89d04ee0699a2bd312d6bb226c86c249fea
+size 321551
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/4909ff8641e9990a279a4dffd0cf1633 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/4909ff8641e9990a279a4dffd0cf1633
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/4909ff8641e9990a279a4dffd0cf1633 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/68fcf82d19d6309db417b6e6d14e12e9 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/68fcf82d19d6309db417b6e6d14e12e9
new file mode 100644
index 0000000000000000000000000000000000000000..51affba055eacdba6536904e59cb89db4879e912
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/68fcf82d19d6309db417b6e6d14e12e9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/8772788a55861c3631b014c293e5328a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/8772788a55861c3631b014c293e5328a
new file mode 100644
index 0000000000000000000000000000000000000000..a09388df3d2e5a7fc215d052116220853c7f9ced
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/8772788a55861c3631b014c293e5328a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a129a4863e04c1b42248d8ec780afe64491aca342dbbb3076bd88bb928b326b8
+size 417056
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/b68996db650c5eb6efdaf6ed80f219bf b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/b68996db650c5eb6efdaf6ed80f219bf
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/b68996db650c5eb6efdaf6ed80f219bf differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3356f5386034489caa3b0899b63780b2c937617e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..f4eb423ece27ffc8a6e7e55e2f8d71b291e7396a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581406137854089, "commit_timestamp_nsecs": 1754581406279344182, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/d/d0b43f0a2a22656c3198cb1dfd85a309 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/d/d0b43f0a2a22656c3198cb1dfd85a309
new file mode 100644
index 0000000000000000000000000000000000000000..fef743d14319dd93208895446518e46e6ee8c463
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/d/d0b43f0a2a22656c3198cb1dfd85a309 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..cb5e2820d5c3a7f8debae43917b0d4941dbbc21e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/019442e6fee270fc52fdb961986f8948 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/019442e6fee270fc52fdb961986f8948
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/019442e6fee270fc52fdb961986f8948 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/17cc4e703c55af0c6a420cb6089f1456 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/17cc4e703c55af0c6a420cb6089f1456
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/17cc4e703c55af0c6a420cb6089f1456 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/9b0c60f87d1217a33ee26a22b2dc41a1 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/9b0c60f87d1217a33ee26a22b2dc41a1
new file mode 100644
index 0000000000000000000000000000000000000000..27088c6ff198d0ea558396ed255f7ba26a50a99f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/9b0c60f87d1217a33ee26a22b2dc41a1
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:188e759a717d376dd22373bb915180d3aefe8fa4d2e035207bf668b4c7f0621a
+size 296011
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/9c083ea83d93ab8e4b6d21f94da73df2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/9c083ea83d93ab8e4b6d21f94da73df2
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/9c083ea83d93ab8e4b6d21f94da73df2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/b8da4fcad32d95f3344bebe84feeb33f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/b8da4fcad32d95f3344bebe84feeb33f
new file mode 100644
index 0000000000000000000000000000000000000000..447028d66f289f51fee1dcf0601231bd60d83e17
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/b8da4fcad32d95f3344bebe84feeb33f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:da316b350973fc4dd7ac31cae7b835f2f45ccfd17f83252c6b5246c3e530d45d
+size 403105
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/c42357c1cb7bd5c4a57adeef57e79c27 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/c42357c1cb7bd5c4a57adeef57e79c27
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/c42357c1cb7bd5c4a57adeef57e79c27 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/e533f5e6b1bbceeec49b462402916235 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/e533f5e6b1bbceeec49b462402916235
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/e533f5e6b1bbceeec49b462402916235 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/f553c9a28366ce2d5d3ba4b9e64de98c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/f553c9a28366ce2d5d3ba4b9e64de98c
new file mode 100644
index 0000000000000000000000000000000000000000..abf967dd249a071bf8c719b0abd0cf5e40748156
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/f553c9a28366ce2d5d3ba4b9e64de98c differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e22f9851056ae3bf8370adf07d2cf84d82bbab60
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..fd19384837acbf5de0efafc0fb0b503327f12079
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581440482723799, "commit_timestamp_nsecs": 1754581440660937229, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/d/020e9f40148d94a66937a2707021cf09 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/d/020e9f40148d94a66937a2707021cf09
new file mode 100644
index 0000000000000000000000000000000000000000..cdce39ec42b758499d3aecca2a1f85c180f48a76
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/d/020e9f40148d94a66937a2707021cf09 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..26a6e0060be901bc5cf967252f3e551c3f72550a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/31f56790053af52df0935d815098f420 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/31f56790053af52df0935d815098f420
new file mode 100644
index 0000000000000000000000000000000000000000..80b2298c150f3e69593068811d3beea33549e335
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/31f56790053af52df0935d815098f420
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e06a3de9c07066d2f427d962ebb0850debda32dd6088a2f077225ce59961bf1a
+size 438926
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/38f50f1c6e9215d5b6418c76ffc92127 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/38f50f1c6e9215d5b6418c76ffc92127
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/38f50f1c6e9215d5b6418c76ffc92127 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/40d714629c3cf2eb595ec74a5dd0bfdd b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/40d714629c3cf2eb595ec74a5dd0bfdd
new file mode 100644
index 0000000000000000000000000000000000000000..41b69fd74866df5b745627a201dbf4190118d48d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/40d714629c3cf2eb595ec74a5dd0bfdd
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ae4424e8aef30b1c549c0f22a968c42e6c7360eb92f44fbb9eea08330375ba84
+size 262090
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/4891a06b5c5273e846397c3e37a3c603 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/4891a06b5c5273e846397c3e37a3c603
new file mode 100644
index 0000000000000000000000000000000000000000..0455438c511e012789bdd1b918a4efe208425f9e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/4891a06b5c5273e846397c3e37a3c603 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/5d51fcd335355f22dc5c0ccba612e1e5 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/5d51fcd335355f22dc5c0ccba612e1e5
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/5d51fcd335355f22dc5c0ccba612e1e5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/b7bcc835b87cfd8e6c7c064453e56375 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/b7bcc835b87cfd8e6c7c064453e56375
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/b7bcc835b87cfd8e6c7c064453e56375 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/f4006915c971a515633ca0cc7817a77d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/f4006915c971a515633ca0cc7817a77d
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/f4006915c971a515633ca0cc7817a77d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/fdc032fce44f58e8bc4ee3335786fde1 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/fdc032fce44f58e8bc4ee3335786fde1
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/fdc032fce44f58e8bc4ee3335786fde1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..44d3073d7a769e7d3eb2ea0b9a1923b38fd0cd5d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..26dd8299bca09209c092925e7cb5eab04aa988ea
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581475957744438, "commit_timestamp_nsecs": 1754581476125060106, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/d/729a0715284bc101069714e18cd1b863 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/d/729a0715284bc101069714e18cd1b863
new file mode 100644
index 0000000000000000000000000000000000000000..66426c56df7cb8f30592ba484bfd7260fa5a7ba8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/d/729a0715284bc101069714e18cd1b863 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..21d55c4ea959f187b9243f3fd66f7c7bc7c639ad
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/46f5998c0a54ca741be483e8ec6ae2d3 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/46f5998c0a54ca741be483e8ec6ae2d3
new file mode 100644
index 0000000000000000000000000000000000000000..71e2ad5497af4d7b0059fe324122a939c747b366
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/46f5998c0a54ca741be483e8ec6ae2d3
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4c05949ff6951f46193d67c1453d03ba5b05c2e8bd29c362f81e69b50b3dca33
+size 379443
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/8571532fdda3cb005f95a5d6f83c5f77 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/8571532fdda3cb005f95a5d6f83c5f77
new file mode 100644
index 0000000000000000000000000000000000000000..20d06ea7d9f11669f6c0748dfb09107b90947a82
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/8571532fdda3cb005f95a5d6f83c5f77
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1f4ddaa3576e73bcf5df94f814f55e3d9af8b716e160cba6333d7dbb421b7a15
+size 321451
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/9a984192e0f0d3a346136a974ce68559 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/9a984192e0f0d3a346136a974ce68559
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/9a984192e0f0d3a346136a974ce68559 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/c03baf0561855b714f5518281576490a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/c03baf0561855b714f5518281576490a
new file mode 100644
index 0000000000000000000000000000000000000000..4859214e6b68fa6e697de9748417fcf43be8081d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/c03baf0561855b714f5518281576490a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/c383bd34f5c487a8da757de12b9577cd b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/c383bd34f5c487a8da757de12b9577cd
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/c383bd34f5c487a8da757de12b9577cd differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/c9a409c1c87811c2f0cba113b94605b0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/c9a409c1c87811c2f0cba113b94605b0
new file mode 100644
index 0000000000000000000000000000000000000000..4505e13d06a4428af1eda1bcfc65c46fb5825506
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/c9a409c1c87811c2f0cba113b94605b0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/ef0cab94b12cd5de8bb4dd51d72cff2f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/ef0cab94b12cd5de8bb4dd51d72cff2f
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/ef0cab94b12cd5de8bb4dd51d72cff2f differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4b512bed6cc6145bab76008440cf6b00c950f448
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..380a9ba84762220e5f8e893cfbb7930dbd83f4ce
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581527622024008, "commit_timestamp_nsecs": 1754581527771542212, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/d/e248910649c15cb2aabcef13051aac78 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/d/e248910649c15cb2aabcef13051aac78
new file mode 100644
index 0000000000000000000000000000000000000000..e360c1b76ee90b7412ff28b5bdb7001641f5a703
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/d/e248910649c15cb2aabcef13051aac78 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5daec7ca6b7e1f240ed016dcbc9f1db452c4f72a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/6c7d883f4ab37bdd64cae99bedb16f3c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/6c7d883f4ab37bdd64cae99bedb16f3c
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/6c7d883f4ab37bdd64cae99bedb16f3c differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/6f52b34166384ce80880dc82e7f15a28 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/6f52b34166384ce80880dc82e7f15a28
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/6f52b34166384ce80880dc82e7f15a28 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/b7f6855903e50441c09827657f2ddcd7 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/b7f6855903e50441c09827657f2ddcd7
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/b7f6855903e50441c09827657f2ddcd7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/bb957fbe09326a8491cfa1970aa69aca b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/bb957fbe09326a8491cfa1970aa69aca
new file mode 100644
index 0000000000000000000000000000000000000000..a766a93071312a2e47cbab4fa6223c98ba6668b1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/bb957fbe09326a8491cfa1970aa69aca differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/c4b828c8126fd5a0b429d215f5270959 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/c4b828c8126fd5a0b429d215f5270959
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/c4b828c8126fd5a0b429d215f5270959 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/c8bea4ba6d9763f1e306c52fad1894a3 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/c8bea4ba6d9763f1e306c52fad1894a3
new file mode 100644
index 0000000000000000000000000000000000000000..ab5f542fa90bc9a49998c8e2f3af5cad95dee611
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/c8bea4ba6d9763f1e306c52fad1894a3
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:87c1c4a4cd00bb6f0191ae31d4e5719ef31ddac0894d2f52b709d5fd08fb421a
+size 442703
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/f714b2a410d6a8400d81689823b07574 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/f714b2a410d6a8400d81689823b07574
new file mode 100644
index 0000000000000000000000000000000000000000..b20dae451746a919c897f1d72bb1d0eed03cb7f9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/f714b2a410d6a8400d81689823b07574
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3f1b9c27f085f80b7300fcbd4ea7aece9964ec91e735ed57450eb5137523de64
+size 296053
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/f916adb92e0e60501d43e7309ae63163 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/f916adb92e0e60501d43e7309ae63163
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/f916adb92e0e60501d43e7309ae63163 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/fdf8f115b60e1bba3fd85f20581b1dd9 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/fdf8f115b60e1bba3fd85f20581b1dd9
new file mode 100644
index 0000000000000000000000000000000000000000..4859214e6b68fa6e697de9748417fcf43be8081d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/fdf8f115b60e1bba3fd85f20581b1dd9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..cc22f01d9474d8e637b1b0d87c2b4c085725efae
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..1e6a413eada290b5ae6b102c193fbb72f91d8e1c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581562354577262, "commit_timestamp_nsecs": 1754581562533980864, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/d/9d2d9323908bf8fdb96135f1882aeef1 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/d/9d2d9323908bf8fdb96135f1882aeef1
new file mode 100644
index 0000000000000000000000000000000000000000..f4d6501839152ab64453e4a416717d127ecb08a0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/d/9d2d9323908bf8fdb96135f1882aeef1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..bdfcd4e70c3e92bcdcd5690b59c0dd315e665b9d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/3063f343d910625db31a68208a01ce61 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/3063f343d910625db31a68208a01ce61
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/3063f343d910625db31a68208a01ce61 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/3a1cf7c95cf887b4cc9300c956fe423f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/3a1cf7c95cf887b4cc9300c956fe423f
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/3a1cf7c95cf887b4cc9300c956fe423f differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/7cb6f65f197e9e5da0fed888eaac8a55 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/7cb6f65f197e9e5da0fed888eaac8a55
new file mode 100644
index 0000000000000000000000000000000000000000..704c7f5dc7c6c69becb093a783c605f41e015841
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/7cb6f65f197e9e5da0fed888eaac8a55
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:705ad8ddf2495986593fade17d87b616373faff7b14a513e0e97ce7e908c9cda
+size 363990
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/b0d44ea8e7585fe60d156183266ba17d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/b0d44ea8e7585fe60d156183266ba17d
new file mode 100644
index 0000000000000000000000000000000000000000..107dde63c0163938a033c9e276e1d4f15f6bda05
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/b0d44ea8e7585fe60d156183266ba17d
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:42b489101d551aa51dfbdda3cc5b2edf91e1f86bd1637352248a1ea855c326cb
+size 340558
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/bc6135a8ab2d1bbb5681a35eee1d25f8 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/bc6135a8ab2d1bbb5681a35eee1d25f8
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/bc6135a8ab2d1bbb5681a35eee1d25f8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/db7de9b492b0461dbd5becddc29a7060 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/db7de9b492b0461dbd5becddc29a7060
new file mode 100644
index 0000000000000000000000000000000000000000..b4067c6ebd9964fd07fd5002e94157e5034a2a27
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/db7de9b492b0461dbd5becddc29a7060 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/eba3f06c4cbffe54f1b444b8c4e2236b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/eba3f06c4cbffe54f1b444b8c4e2236b
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/eba3f06c4cbffe54f1b444b8c4e2236b differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..79e2a6d2ba12f12d6f5891424c544b7456448bbd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..d33e319ce01d61773701c3c453f41929bf566837
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581597155360944, "commit_timestamp_nsecs": 1754581597335212205, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/d/501e60af1270501bde0688602683312d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/d/501e60af1270501bde0688602683312d
new file mode 100644
index 0000000000000000000000000000000000000000..67d49a5af0512938745ff41379d8a115a9499128
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/d/501e60af1270501bde0688602683312d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..43d410f661583aee5deb914cc15663c7034654c8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/0eea05552d80341076b55a7cf2df8f52 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/0eea05552d80341076b55a7cf2df8f52
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/0eea05552d80341076b55a7cf2df8f52 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/2633d9e55530680f7f586641d9173b11 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/2633d9e55530680f7f586641d9173b11
new file mode 100644
index 0000000000000000000000000000000000000000..d2fba8d661a5c3b92e4d2ea77af650dd2b660013
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/2633d9e55530680f7f586641d9173b11
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:727a9fc6fe5eea9223299558623b708fd8061432c233abf0a9337de7d4006925
+size 457451
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/375817f533476f2bd532b9f8805bb46b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/375817f533476f2bd532b9f8805bb46b
new file mode 100644
index 0000000000000000000000000000000000000000..3895d25450b2f51da5277b71f1443679304eeef1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/375817f533476f2bd532b9f8805bb46b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:73afb13d660b7e765cd08980a6e44a57b3b7d15faa8cf9d2aa7517f15e7bf578
+size 280666
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/a90d3bb4e6abc511c93584a2c1ba23ec b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/a90d3bb4e6abc511c93584a2c1ba23ec
new file mode 100644
index 0000000000000000000000000000000000000000..2a31377ca030aced51ef70fb458c6b5c47404cd3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/a90d3bb4e6abc511c93584a2c1ba23ec differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/ce33aba45f8c0a16bfea15ae849a6cd0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/ce33aba45f8c0a16bfea15ae849a6cd0
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/ce33aba45f8c0a16bfea15ae849a6cd0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/d4f97dc824dccc2d4181eee7c1c4e7a6 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/d4f97dc824dccc2d4181eee7c1c4e7a6
new file mode 100644
index 0000000000000000000000000000000000000000..4f42a22aeb064de648d4b2ad05a5fc4613beddd4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/d4f97dc824dccc2d4181eee7c1c4e7a6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/dc24c21da078618da65bc4a62b39faf6 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/dc24c21da078618da65bc4a62b39faf6
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/dc24c21da078618da65bc4a62b39faf6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..57569737158a45e1b79122506ec025365b4e013a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..122100c36a0abe63056bb4766d3def6080c5919f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581630856860943, "commit_timestamp_nsecs": 1754581631043404985, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/d/2254ff37a7005173d8cba0949c0bde60 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/d/2254ff37a7005173d8cba0949c0bde60
new file mode 100644
index 0000000000000000000000000000000000000000..7b193041c59df66d7667ac56ebc713beb1adda0e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/d/2254ff37a7005173d8cba0949c0bde60 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8519094e500fe0510658810800c4e2ca3ca2a6f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/28af640d49187a84b2191a96c22dc1c5 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/28af640d49187a84b2191a96c22dc1c5
new file mode 100644
index 0000000000000000000000000000000000000000..9e93441e2d35255a167f6ac06f2bb978a03c8b26
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/28af640d49187a84b2191a96c22dc1c5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/2e56c1184a259abf1c122855c9c2fe19 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/2e56c1184a259abf1c122855c9c2fe19
new file mode 100644
index 0000000000000000000000000000000000000000..729578a7f62f8ad76b6b32c981fd5023b792cc96
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/2e56c1184a259abf1c122855c9c2fe19
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0a9d924ed300c229abbaf6b0cd0f7b1d7a82f9ff4975909eec683e0b431e6bab
+size 396422
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/5aafe24daec0af9d88fa81114cb67626 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/5aafe24daec0af9d88fa81114cb67626
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/5aafe24daec0af9d88fa81114cb67626 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/86e1fb8c9172faeafe5c1c6113ab8583 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/86e1fb8c9172faeafe5c1c6113ab8583
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/86e1fb8c9172faeafe5c1c6113ab8583 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/93fa082fc7fc382678627ad5643c3f75 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/93fa082fc7fc382678627ad5643c3f75
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/93fa082fc7fc382678627ad5643c3f75 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/d4e5d62107598d42a7156b3d4ea752f2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/d4e5d62107598d42a7156b3d4ea752f2
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/d4e5d62107598d42a7156b3d4ea752f2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/e92b8a096f4e93f96cd532bafec4c022 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/e92b8a096f4e93f96cd532bafec4c022
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/e92b8a096f4e93f96cd532bafec4c022 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/f53a454a4da99bb7ab185d5f2b76677d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/f53a454a4da99bb7ab185d5f2b76677d
new file mode 100644
index 0000000000000000000000000000000000000000..ad2e3b212c8961f83813d8672051b8793acaeb0e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/f53a454a4da99bb7ab185d5f2b76677d
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3253caa4fa90415124bee136b54adb7e588890494edb41fb90a0debae8695d5b
+size 304589
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..ffe876e41109a42758b1bf53e0075544fe224ff7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..d851ab0fc219c9d6d97a46a4f5b43abdf9fef12c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581665872219299, "commit_timestamp_nsecs": 1754581666057694098, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/d/f8adeb5f0f8b0d5e3186389b76941269 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/d/f8adeb5f0f8b0d5e3186389b76941269
new file mode 100644
index 0000000000000000000000000000000000000000..15b4986bf1766b3f3d2d7bba3d2f4e5e6a180fcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/d/f8adeb5f0f8b0d5e3186389b76941269 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..598870ccfa2f492ce806b84c7fe4b37ae7ceeb1e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/3723d60ab2a4f26eeda4c13a43d22db5 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/3723d60ab2a4f26eeda4c13a43d22db5
new file mode 100644
index 0000000000000000000000000000000000000000..6a18477a0f9fe63e8cbfd94307b7dfa02c8b10d8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/3723d60ab2a4f26eeda4c13a43d22db5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/3e62c5cf54e5628edaf5d035c8b36d58 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/3e62c5cf54e5628edaf5d035c8b36d58
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/3e62c5cf54e5628edaf5d035c8b36d58 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/42e2ac434e34595ee71bb11ae5615e4b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/42e2ac434e34595ee71bb11ae5615e4b
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/42e2ac434e34595ee71bb11ae5615e4b differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/5386935fd27f76cf0f85bcecd156943a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/5386935fd27f76cf0f85bcecd156943a
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/5386935fd27f76cf0f85bcecd156943a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/5991ee7228b3d71632b644b5dbd88db3 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/5991ee7228b3d71632b644b5dbd88db3
new file mode 100644
index 0000000000000000000000000000000000000000..886b49a62d83a7c2053b7a7c21ea01f518d77c9a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/5991ee7228b3d71632b644b5dbd88db3
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4f21c8a769a523d1d15191a73c36174d7754a42265719a2372d8afe46aa945d5
+size 442963
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/78ecc53c64c1a303749cd6c112d69400 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/78ecc53c64c1a303749cd6c112d69400
new file mode 100644
index 0000000000000000000000000000000000000000..4e412893d3937be337863a4e9387ed66a6fb0353
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/78ecc53c64c1a303749cd6c112d69400
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a163ea1f559c1149ec197f63e1865f3aeed419772ddc292abeb1bd7a3f66fcca
+size 296005
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/a79aebca6e3d1d4077ed8afec89a56f8 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/a79aebca6e3d1d4077ed8afec89a56f8
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/a79aebca6e3d1d4077ed8afec89a56f8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/b314fbbea19e223a9ed0891b577f2c07 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/b314fbbea19e223a9ed0891b577f2c07
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/b314fbbea19e223a9ed0891b577f2c07 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4de37d0344dabad6698ce69f6bf39e4ad03ffaf0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..f37fa7eefd968154a54ebb00e0814fb19e00ec7a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581700580276879, "commit_timestamp_nsecs": 1754581700748040116, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/d/6dcb9d8ba1c34cb7883b1b04edef9b22 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/d/6dcb9d8ba1c34cb7883b1b04edef9b22
new file mode 100644
index 0000000000000000000000000000000000000000..30fed359e1decc12344493e3eb95ade68cbd5283
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/d/6dcb9d8ba1c34cb7883b1b04edef9b22 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9d7f0181130dc8a76791ce393663e07774fbfb68
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/15a47401043129c383ae5016e686f665 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/15a47401043129c383ae5016e686f665
new file mode 100644
index 0000000000000000000000000000000000000000..d41e08bbcfec980b85747a048cf178b6b0942171
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/15a47401043129c383ae5016e686f665
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cbbc7ac4311e31f62b0d70be17d6b4c36c4da5a5500fb1aa4d7b1ec9bd581a6f
+size 261808
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/1cda59978a21a6bed462bf4ff38066f9 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/1cda59978a21a6bed462bf4ff38066f9
new file mode 100644
index 0000000000000000000000000000000000000000..470e24a35f6be90a107eac0271b13b9fffe26e63
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/1cda59978a21a6bed462bf4ff38066f9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/301281dbb3ad4a60aae98dfbfebebc17 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/301281dbb3ad4a60aae98dfbfebebc17
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/301281dbb3ad4a60aae98dfbfebebc17 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/62efc6af18b6a9b3b38a91b8c502fa31 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/62efc6af18b6a9b3b38a91b8c502fa31
new file mode 100644
index 0000000000000000000000000000000000000000..2a31377ca030aced51ef70fb458c6b5c47404cd3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/62efc6af18b6a9b3b38a91b8c502fa31 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/9ab9613dffedcf74fd58055d6de6ff83 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/9ab9613dffedcf74fd58055d6de6ff83
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/9ab9613dffedcf74fd58055d6de6ff83 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/cc3199f51eb916ace4cec44a24a03e34 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/cc3199f51eb916ace4cec44a24a03e34
new file mode 100644
index 0000000000000000000000000000000000000000..487e78ef8398c11b73f0a4d637e9362e765aa2f6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/cc3199f51eb916ace4cec44a24a03e34
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:56156629fef74ae23acdcf97f512ed7a8c69ed7a5b06f09e82cbda94b527ba55
+size 437304
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/ded20f31d2832a60a30989487668570a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/ded20f31d2832a60a30989487668570a
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/ded20f31d2832a60a30989487668570a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/e5801ca666a3a275d03288f66273f26b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/e5801ca666a3a275d03288f66273f26b
new file mode 100644
index 0000000000000000000000000000000000000000..dfbfe1a61305bab6f716f83c77874d9e3176999d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/e5801ca666a3a275d03288f66273f26b differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3fa058b290d4bd52cd892b5480d997a518c5a45d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..eff5273b2fba516cf5df4cdffd25c7dacbb807f7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754579485760947412, "commit_timestamp_nsecs": 1754579485944579176, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/d/ead4dd5cc4fd2e9e5ae9bd4ec6ee6706 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/d/ead4dd5cc4fd2e9e5ae9bd4ec6ee6706
new file mode 100644
index 0000000000000000000000000000000000000000..a5cb38b61078b7b6fb5ce1744934f3460877f9a6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/d/ead4dd5cc4fd2e9e5ae9bd4ec6ee6706 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3169f7a7b892df3a775eb1bd7278a6889bc7f031
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/37654fa09e2fa4a1cf1d93035d690e3c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/37654fa09e2fa4a1cf1d93035d690e3c
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/37654fa09e2fa4a1cf1d93035d690e3c differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/5d867d435a80c8719dda8433eb9edd69 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/5d867d435a80c8719dda8433eb9edd69
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/5d867d435a80c8719dda8433eb9edd69 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/bbc543d5cd785634fb24b3ea9bc5ae8e b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/bbc543d5cd785634fb24b3ea9bc5ae8e
new file mode 100644
index 0000000000000000000000000000000000000000..7ab37247ae2d38331f20f9850669443ae04e8bf5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/bbc543d5cd785634fb24b3ea9bc5ae8e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:65fd489217bd6e528fe7b5950ee100f39ce0acc2aac6f614ce23e04f08a49574
+size 438658
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/bbfde9863958f9b008224d33998ff903 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/bbfde9863958f9b008224d33998ff903
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/bbfde9863958f9b008224d33998ff903 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/c1edf814426945daca0c906eacb4a413 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/c1edf814426945daca0c906eacb4a413
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/c1edf814426945daca0c906eacb4a413 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/cc0adff86a8d6731ac96be0224d93c1c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/cc0adff86a8d6731ac96be0224d93c1c
new file mode 100644
index 0000000000000000000000000000000000000000..ad6ca7965107e19203726a6f00645898ee23ed58
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/cc0adff86a8d6731ac96be0224d93c1c differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/dac806a79a2a8f795b69708c37045475 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/dac806a79a2a8f795b69708c37045475
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/dac806a79a2a8f795b69708c37045475 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/db5d2994037403cd317ccc8cd6efdaf4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/db5d2994037403cd317ccc8cd6efdaf4
new file mode 100644
index 0000000000000000000000000000000000000000..4bd186ec889dfc8f00536536d699245e3bcdf3fd
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/db5d2994037403cd317ccc8cd6efdaf4
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:233d8408583d760018d245339ec2674819243430b10d16a3a95a31542a6bfc3e
+size 261761
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..021e671f3231a7ec8db43cba6a3345a938109406
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..5d32f26effc2c4dc8f0754314b296e8bd33f41b4
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581734939077670, "commit_timestamp_nsecs": 1754581735081854455, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/d/3cb92d9736df27225aef1625d354d75d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/d/3cb92d9736df27225aef1625d354d75d
new file mode 100644
index 0000000000000000000000000000000000000000..2233ab93bfcead721d28cd16bb47963c3401cccc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/d/3cb92d9736df27225aef1625d354d75d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3b5e18b2a8304a31aa8bd78b776688f446a857d6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/4e193f82b240cad937f9c03c2795e28f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/4e193f82b240cad937f9c03c2795e28f
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/4e193f82b240cad937f9c03c2795e28f differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/64f25661ea1c834b93b34a7bb0c34744 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/64f25661ea1c834b93b34a7bb0c34744
new file mode 100644
index 0000000000000000000000000000000000000000..607d41840d2d210584a90f8a463ebf062b3b582d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/64f25661ea1c834b93b34a7bb0c34744
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7b8eeb15880caa92550cae9607e0640c1595b88834a8543a31ae0aab8ace651c
+size 394768
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/6d8cf9091368c573e90b0a4b10fff572 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/6d8cf9091368c573e90b0a4b10fff572
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/6d8cf9091368c573e90b0a4b10fff572 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/82b18b8c3ebb65de3c953098d54662e2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/82b18b8c3ebb65de3c953098d54662e2
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/82b18b8c3ebb65de3c953098d54662e2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/a921a5efc5ca13753b47885862e7eee4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/a921a5efc5ca13753b47885862e7eee4
new file mode 100644
index 0000000000000000000000000000000000000000..2a31377ca030aced51ef70fb458c6b5c47404cd3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/a921a5efc5ca13753b47885862e7eee4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/abd81bde906e5db2865de1ee2a6e614b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/abd81bde906e5db2865de1ee2a6e614b
new file mode 100644
index 0000000000000000000000000000000000000000..2653cdc11a56e5a7b04d52465140858f5552ad12
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/abd81bde906e5db2865de1ee2a6e614b differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/b7bc43f2c882ca0c2d29e289a1fc6c7f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/b7bc43f2c882ca0c2d29e289a1fc6c7f
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/b7bc43f2c882ca0c2d29e289a1fc6c7f differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/fb0c4686ec479f92a27dcde0e7bfb54b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/fb0c4686ec479f92a27dcde0e7bfb54b
new file mode 100644
index 0000000000000000000000000000000000000000..6df0df9e47be0347d2bedfce4d32807b7f5da5e3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/fb0c4686ec479f92a27dcde0e7bfb54b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bc784af93538dfc8643847fa71fb61715addda7661a0dbb7d175c42915e5aa59
+size 270374
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..7be0e8dc4fbd050e4c8edd472e3fc5b74bd8115d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..baf9b61f7cef8641b16ab180b2c9228a5db4a51e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581769633762187, "commit_timestamp_nsecs": 1754581769776539782, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/d/b79037869b8b5d408e052e67ba7f36bb b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/d/b79037869b8b5d408e052e67ba7f36bb
new file mode 100644
index 0000000000000000000000000000000000000000..c23dd7098d14b1ba3e455dcbf4d087b1c51b9e05
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/d/b79037869b8b5d408e052e67ba7f36bb differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1944203b83878ef418159e9f04c5f053ba465f03
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/2795d84a6d58848e3ccd405458d66059 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/2795d84a6d58848e3ccd405458d66059
new file mode 100644
index 0000000000000000000000000000000000000000..f73a2601e3b1bc9205721182235784e3f1ead95a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/2795d84a6d58848e3ccd405458d66059 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/7b5524b2470405ea9dc2d4ea37b291ec b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/7b5524b2470405ea9dc2d4ea37b291ec
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/7b5524b2470405ea9dc2d4ea37b291ec differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/828e19f6f7bd8b771fc28459d14e1cbf b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/828e19f6f7bd8b771fc28459d14e1cbf
new file mode 100644
index 0000000000000000000000000000000000000000..d698338eb365a85ed83193d53df268e2c4e3b48a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/828e19f6f7bd8b771fc28459d14e1cbf
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5234e5f33a11b714a4b482809640c3263d843ab8f5908e93f377d139f68f3063
+size 159844
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/887b894efd5edee0476c191ea277e230 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/887b894efd5edee0476c191ea277e230
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/887b894efd5edee0476c191ea277e230 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/8d04f6fe1432b3aa7ed67125d3cd5243 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/8d04f6fe1432b3aa7ed67125d3cd5243
new file mode 100644
index 0000000000000000000000000000000000000000..ec745d97353f97209ef50f0eb738585830219f53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/8d04f6fe1432b3aa7ed67125d3cd5243 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/b5fa45614d0de7bacd77b3efbf79cec2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/b5fa45614d0de7bacd77b3efbf79cec2
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/b5fa45614d0de7bacd77b3efbf79cec2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/eac4193ee16d9bfaa41865e13c3b6fdc b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/eac4193ee16d9bfaa41865e13c3b6fdc
new file mode 100644
index 0000000000000000000000000000000000000000..a5bf1b1119ec7b252559c4d223d5b2439c57acac
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/eac4193ee16d9bfaa41865e13c3b6fdc
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:822e88d2f158e26a4b84727e365fb76429e12b1fcee97d6c13d691f04999e1f2
+size 505725
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/f3d359be47673fce217c29e61d2b6adc b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/f3d359be47673fce217c29e61d2b6adc
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/f3d359be47673fce217c29e61d2b6adc differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9e52c3f81a1e3d0645dfd0fbc6d0cb93afd9508c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..71a7a99862833a35bd4e36c4da909f13ef54632d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581804452907146, "commit_timestamp_nsecs": 1754581804627288718, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/d/98024786fd0247a10380e5d0f4d6de15 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/d/98024786fd0247a10380e5d0f4d6de15
new file mode 100644
index 0000000000000000000000000000000000000000..e2395b9b8c775322e6ef8519532a03497622d7ff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/d/98024786fd0247a10380e5d0f4d6de15 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9fecdb028b23cd35cf18cfaef2d947c6cb3064fc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/14b7c552dc1b74ec53090eae4b725fbb b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/14b7c552dc1b74ec53090eae4b725fbb
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/14b7c552dc1b74ec53090eae4b725fbb differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/245ed70da79606b6b70a0f98c6b41ba4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/245ed70da79606b6b70a0f98c6b41ba4
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/245ed70da79606b6b70a0f98c6b41ba4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/397eeeeaeac47b359117e5ca491b1c02 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/397eeeeaeac47b359117e5ca491b1c02
new file mode 100644
index 0000000000000000000000000000000000000000..3b4f3df31e2a4a6b98bf1455c57cfa2626f49e13
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/397eeeeaeac47b359117e5ca491b1c02
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:073f41680e279a2c44c35cdcc9d90b7e853d463085181b2e4bd01ab3bfb9e851
+size 262051
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/581aacbd3570be195b82c78479ead98c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/581aacbd3570be195b82c78479ead98c
new file mode 100644
index 0000000000000000000000000000000000000000..7b5c9418869130626da821b97e369297056dc838
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/581aacbd3570be195b82c78479ead98c differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/683fb585640cb88a79613db7e769871a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/683fb585640cb88a79613db7e769871a
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/683fb585640cb88a79613db7e769871a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/71f35085b5a9115b2ed9ff7a49533279 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/71f35085b5a9115b2ed9ff7a49533279
new file mode 100644
index 0000000000000000000000000000000000000000..e943ad5645e795bc9f6e0d22a7fbdb023b9423ff
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/71f35085b5a9115b2ed9ff7a49533279
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8cdd41f70a3710523fa25d759afde7ffb80eb95cdeb55d88516b56246a14997e
+size 476832
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/9fb66d38ff303827a02a36a21ee4b6a0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/9fb66d38ff303827a02a36a21ee4b6a0
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/9fb66d38ff303827a02a36a21ee4b6a0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/c6d3711fe581101b44701c89521e0e52 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/c6d3711fe581101b44701c89521e0e52
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/c6d3711fe581101b44701c89521e0e52 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a4ba047a39b2c586d315511c69e3016340be5173
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..936372288290b74d913528b693b0ed25e5b7e33d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581838656515628, "commit_timestamp_nsecs": 1754581838830735619, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/d/911c38b3dbb32822d5d9099dae891d8d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/d/911c38b3dbb32822d5d9099dae891d8d
new file mode 100644
index 0000000000000000000000000000000000000000..cbfc0ffc1a27a8b669c621c53924a939371f53aa
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/d/911c38b3dbb32822d5d9099dae891d8d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4d8b049bb3d1ccafb45db53ef2f3cc2b8e21f126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/0b76e2536223c6dd78d0166313b971d8 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/0b76e2536223c6dd78d0166313b971d8
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/0b76e2536223c6dd78d0166313b971d8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/5357d6a1e5298a4a34ebec1b4c6e344d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/5357d6a1e5298a4a34ebec1b4c6e344d
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/5357d6a1e5298a4a34ebec1b4c6e344d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/7bb2446841d028b77ce365f656ca6bbe b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/7bb2446841d028b77ce365f656ca6bbe
new file mode 100644
index 0000000000000000000000000000000000000000..d960dee1e69abb4ee4122be3bdd2f83db6e783b3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/7bb2446841d028b77ce365f656ca6bbe differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/a6d84a43e1bbae4882e860f0c4548355 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/a6d84a43e1bbae4882e860f0c4548355
new file mode 100644
index 0000000000000000000000000000000000000000..1926e8c4ceb8009550b7a90e5e382a1e44664a71
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/a6d84a43e1bbae4882e860f0c4548355
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6c5f235a76ceee6a5e17ca74a7c42b362594e10d8d807bed93685f61e73086dc
+size 485539
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/ae4b05b72b882d07f04d216c31b71990 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/ae4b05b72b882d07f04d216c31b71990
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/ae4b05b72b882d07f04d216c31b71990 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/c6c3c8b3d2b5f201aab22df943704572 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/c6c3c8b3d2b5f201aab22df943704572
new file mode 100644
index 0000000000000000000000000000000000000000..aa9cc89a3478de335b192ea281b829c44dc7ba2d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/c6c3c8b3d2b5f201aab22df943704572
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d4a0e9ddb6d578409bbd4456805732dafcad1219f742c44741b9af8913d65844
+size 253554
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/d3c275eed7669d1127efc0f837dcb7b8 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/d3c275eed7669d1127efc0f837dcb7b8
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/d3c275eed7669d1127efc0f837dcb7b8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/f811401d30ce6642dfa8624be3f3be93 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/f811401d30ce6642dfa8624be3f3be93
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/f811401d30ce6642dfa8624be3f3be93 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..46cffbc770643cd7db089508064028b5df0234d8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..09a0e7915accb14647c36e7c8c0320ea9df9a50e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581873016781949, "commit_timestamp_nsecs": 1754581873160690132, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/d/cf624868434f70c3825ae2a6854ec61e b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/d/cf624868434f70c3825ae2a6854ec61e
new file mode 100644
index 0000000000000000000000000000000000000000..07bbdc3893f166ddfe78d441608b461b9e5ad5d1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/d/cf624868434f70c3825ae2a6854ec61e differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0917c64be2765eb1f2ea48bb7f8fd980094c70c6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/30b733a14e29dabebc1c548df65cc7cf b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/30b733a14e29dabebc1c548df65cc7cf
new file mode 100644
index 0000000000000000000000000000000000000000..2910fbc6db2094404ed932c7d9bcdb23beb99906
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/30b733a14e29dabebc1c548df65cc7cf differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/488194b903df129ea7931e9eef58f123 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/488194b903df129ea7931e9eef58f123
new file mode 100644
index 0000000000000000000000000000000000000000..daff71b5adcfaf036cda27b51fedad002644e369
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/488194b903df129ea7931e9eef58f123 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/509073e7b788797e8d3bc577d1f0f9bc b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/509073e7b788797e8d3bc577d1f0f9bc
new file mode 100644
index 0000000000000000000000000000000000000000..d8a60c1bed6d63d1b228dbc50481e65bb8eff6a3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/509073e7b788797e8d3bc577d1f0f9bc
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2679374d28e0473ab3ecd1182af8972e8bfcaba99a365da30cfc9074bc32dcc7
+size 287436
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/72307a39e35ffc35d136fecd4851fb66 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/72307a39e35ffc35d136fecd4851fb66
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/72307a39e35ffc35d136fecd4851fb66 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/ac629881d7ff2d1d90f4a233518897c1 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/ac629881d7ff2d1d90f4a233518897c1
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/ac629881d7ff2d1d90f4a233518897c1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/ba4d0445f50a8876a55b52b752b7a9ee b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/ba4d0445f50a8876a55b52b752b7a9ee
new file mode 100644
index 0000000000000000000000000000000000000000..578d548e07e1a7c6ef41c7c3dd9102780bc16f4e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/ba4d0445f50a8876a55b52b752b7a9ee
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3242f8ac2c75ea63ef44509ad30a249fe139cb0e82b3028aa72d88dd4eddc2a2
+size 451542
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/c18040ce08a71a7b0a174a4eee643ef5 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/c18040ce08a71a7b0a174a4eee643ef5
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/c18040ce08a71a7b0a174a4eee643ef5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/d3a27513e0a883ceb040224e3b7f5a6a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/d3a27513e0a883ceb040224e3b7f5a6a
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/d3a27513e0a883ceb040224e3b7f5a6a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f42043cf0166abd56f409c576f1adcac5d0eaca6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..bad52842a38768919878e88e025eb73351443ace
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581907722297787, "commit_timestamp_nsecs": 1754581907860791975, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/d/7040f93326ea4e67a1e6122dcbf3eee1 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/d/7040f93326ea4e67a1e6122dcbf3eee1
new file mode 100644
index 0000000000000000000000000000000000000000..f72f6e0ff7d06880f25a287049a82e6a551d13aa
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/d/7040f93326ea4e67a1e6122dcbf3eee1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c0a390c1064dce6ec336e51d90eefe00c197df1c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/0c0c81f20d96df2acfc9b10e9a2e737c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/0c0c81f20d96df2acfc9b10e9a2e737c
new file mode 100644
index 0000000000000000000000000000000000000000..486eda7a1a5df402b9049bd05c7b31bb80b6876f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/0c0c81f20d96df2acfc9b10e9a2e737c differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/38894fbd397dfb5d2d839706f70eb7eb b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/38894fbd397dfb5d2d839706f70eb7eb
new file mode 100644
index 0000000000000000000000000000000000000000..0143498c9f97faecb241e73f211f00dd5fc05a1d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/38894fbd397dfb5d2d839706f70eb7eb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7cce99d3280c9e35be2f5f6823bd436af8a8884e484dcac5f063c0a24d3e3533
+size 316385
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/5a54ed69a3e51af4f6a95a0901329188 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/5a54ed69a3e51af4f6a95a0901329188
new file mode 100644
index 0000000000000000000000000000000000000000..38576f6712c343875b05e3239fefc1132f7cf175
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/5a54ed69a3e51af4f6a95a0901329188 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/8b3ca8312cef9af65d4150291994305b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/8b3ca8312cef9af65d4150291994305b
new file mode 100644
index 0000000000000000000000000000000000000000..442902fd20f360d000bf7cb1c40e349e35605614
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/8b3ca8312cef9af65d4150291994305b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2f2be6e975ad46b37f9c3fcad008544a75d87a7c03a393aa7c09f252aad4c464
+size 380065
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/b336a7d01e8e172156100ce0a5e3c9a7 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/b336a7d01e8e172156100ce0a5e3c9a7
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/b336a7d01e8e172156100ce0a5e3c9a7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/b9ab2aed6538413cd1a4b46500809863 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/b9ab2aed6538413cd1a4b46500809863
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/b9ab2aed6538413cd1a4b46500809863 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/c98c9496b34ae7b42eb0f89fc159866b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/c98c9496b34ae7b42eb0f89fc159866b
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/c98c9496b34ae7b42eb0f89fc159866b differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/d4736f34515ba9a4898d5aa3c675f187 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/d4736f34515ba9a4898d5aa3c675f187
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/d4736f34515ba9a4898d5aa3c675f187 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..ed22d00bd444e130a22151e80434ea989890a2b0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..7563782f50fa771b8a03fa3ae9f0135c066189cc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581943234013516, "commit_timestamp_nsecs": 1754581943386858124, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/d/413b672e031fa5a95fe905e48f4bd525 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/d/413b672e031fa5a95fe905e48f4bd525
new file mode 100644
index 0000000000000000000000000000000000000000..651e30cb3b13570f11abd8154c7e45a5d2f085c7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/d/413b672e031fa5a95fe905e48f4bd525 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e1129b9a50fc263a57ac1dc9b6dc753bf14955a8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/1e03d0173aad08ff56af3b32943a69fc b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/1e03d0173aad08ff56af3b32943a69fc
new file mode 100644
index 0000000000000000000000000000000000000000..ebbe9673c34995fae3d566d3858a4bbf08c32eda
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/1e03d0173aad08ff56af3b32943a69fc
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cc869077c88104750646f2e28651fbb8f3a8f61cc77f46ee27822638d0cd4cf8
+size 296045
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/2add0824d41fa0c677b0bb2b0de2d5a2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/2add0824d41fa0c677b0bb2b0de2d5a2
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/2add0824d41fa0c677b0bb2b0de2d5a2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/3881f1971739b67db2001c25bbeb0cd4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/3881f1971739b67db2001c25bbeb0cd4
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/3881f1971739b67db2001c25bbeb0cd4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/42944918c1372f3b59e2ad831ada7033 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/42944918c1372f3b59e2ad831ada7033
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/42944918c1372f3b59e2ad831ada7033 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/bfe9124c0d87e82a9a1b6778313dbc5f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/bfe9124c0d87e82a9a1b6778313dbc5f
new file mode 100644
index 0000000000000000000000000000000000000000..3cb1a2aa90f3094a2928350e76962b6354ea713b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/bfe9124c0d87e82a9a1b6778313dbc5f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:debe20c781a88ddb1727baa1b07609b5a210c5a5e01e6aba325f562a9cfc929c
+size 393035
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/c534a5088ffd6071125081ff7abae11a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/c534a5088ffd6071125081ff7abae11a
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/c534a5088ffd6071125081ff7abae11a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/c6e0959b6ced0932675ad020b2368259 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/c6e0959b6ced0932675ad020b2368259
new file mode 100644
index 0000000000000000000000000000000000000000..348874716c5f2aee045a05e6d4289c76b922a9e6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/c6e0959b6ced0932675ad020b2368259 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/ff7e8bb43afffa16fc477c99510c4182 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/ff7e8bb43afffa16fc477c99510c4182
new file mode 100644
index 0000000000000000000000000000000000000000..2910fbc6db2094404ed932c7d9bcdb23beb99906
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/ff7e8bb43afffa16fc477c99510c4182 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1c1b1302bc53ef4f67533b3c5adf4f63f32d0143
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..67410b004c0b0f44bd57b1d48f493257bf9a8c5b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754581977448960032, "commit_timestamp_nsecs": 1754581977627391686, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/d/aeb20ec517578eb3ac45991d1201cd7e b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/d/aeb20ec517578eb3ac45991d1201cd7e
new file mode 100644
index 0000000000000000000000000000000000000000..79e237b1d991c4beca373150f2b63b7be007bc4b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/d/aeb20ec517578eb3ac45991d1201cd7e differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..92e8f80b11b9b16db69d9e14519bb5f82e10e9d3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/0014690b394226465336beacba700fc9 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/0014690b394226465336beacba700fc9
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/0014690b394226465336beacba700fc9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/178edffa3a36cc1e70849ec68675b8eb b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/178edffa3a36cc1e70849ec68675b8eb
new file mode 100644
index 0000000000000000000000000000000000000000..ec745d97353f97209ef50f0eb738585830219f53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/178edffa3a36cc1e70849ec68675b8eb differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/1b11783c297a9c5e2ae992e4a711f790 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/1b11783c297a9c5e2ae992e4a711f790
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/1b11783c297a9c5e2ae992e4a711f790 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/1e3a5f7b03fb48de5c95b6619b0c55c4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/1e3a5f7b03fb48de5c95b6619b0c55c4
new file mode 100644
index 0000000000000000000000000000000000000000..1f0af83f38c572f664fae34ea9a55148e2879574
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/1e3a5f7b03fb48de5c95b6619b0c55c4
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:212bf92f2dd8351cb7a24a52b294c757a14f587f9fa647d6c8429d790d52f481
+size 439523
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/403675af73b0f1c01eaebf45494d5c3e b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/403675af73b0f1c01eaebf45494d5c3e
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/403675af73b0f1c01eaebf45494d5c3e differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/4b8664203727e6d5283cf7dcd6570d41 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/4b8664203727e6d5283cf7dcd6570d41
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/4b8664203727e6d5283cf7dcd6570d41 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/b16406f1dd98ef168d96cde8f9721e73 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/b16406f1dd98ef168d96cde8f9721e73
new file mode 100644
index 0000000000000000000000000000000000000000..6e996b6fa557f4f25781bed03dc087ce738fef4c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/b16406f1dd98ef168d96cde8f9721e73
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5d0dbeb35943467000477645a65755210085cd346419c776e8ae49ab82fdd012
+size 261746
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/bbc1810a8b137e38de28e528573b3e02 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/bbc1810a8b137e38de28e528573b3e02
new file mode 100644
index 0000000000000000000000000000000000000000..6091a9f2498fca5c4a16973cffd0db8b30db335b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/bbc1810a8b137e38de28e528573b3e02 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..31bc2681618de6c1e8a8e8d24221eb93c681536c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..4ec02edff2f24ac24f5a92e8c4406ca3703af37c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754582011129225307, "commit_timestamp_nsecs": 1754582011270049267, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/d/5fb0574ff9067e5d18cd80dd4d82f855 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/d/5fb0574ff9067e5d18cd80dd4d82f855
new file mode 100644
index 0000000000000000000000000000000000000000..f130fbfde817ed667dea0c2e997b5d54c1f747b2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/d/5fb0574ff9067e5d18cd80dd4d82f855 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a0cd62006838f220568442bcc31c0d9a83b1c767
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/0f3a88809ff6a86d57dc4be6fd900443 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/0f3a88809ff6a86d57dc4be6fd900443
new file mode 100644
index 0000000000000000000000000000000000000000..dfbfe1a61305bab6f716f83c77874d9e3176999d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/0f3a88809ff6a86d57dc4be6fd900443 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/49b765118cb23e604fded968cc92323b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/49b765118cb23e604fded968cc92323b
new file mode 100644
index 0000000000000000000000000000000000000000..7a5887ed759f4963f4196cefa7a5b3f6189e52a1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/49b765118cb23e604fded968cc92323b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:107aef96622a2a9901efa3f6c1170b38efe78924745a9a62527613a0938bbe9b
+size 278920
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/6431816e8100a44f60e8fbdea3dfd5b7 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/6431816e8100a44f60e8fbdea3dfd5b7
new file mode 100644
index 0000000000000000000000000000000000000000..4859214e6b68fa6e697de9748417fcf43be8081d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/6431816e8100a44f60e8fbdea3dfd5b7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/66b36cf3af96610436f0552f1fc3f70f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/66b36cf3af96610436f0552f1fc3f70f
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/66b36cf3af96610436f0552f1fc3f70f differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/ba6c84eb30c8b04b808943be0b310626 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/ba6c84eb30c8b04b808943be0b310626
new file mode 100644
index 0000000000000000000000000000000000000000..d70a50f8fdb09a0c4c183b480fb6d62da1e86efa
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/ba6c84eb30c8b04b808943be0b310626 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/ef50b9ccb02a94be66c2af83f19b45fe b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/ef50b9ccb02a94be66c2af83f19b45fe
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/ef50b9ccb02a94be66c2af83f19b45fe differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/f6ad270798c972e8a70af68cc08fcff7 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/f6ad270798c972e8a70af68cc08fcff7
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/f6ad270798c972e8a70af68cc08fcff7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/fdda237eb7ce577d14039e28520184a1 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/fdda237eb7ce577d14039e28520184a1
new file mode 100644
index 0000000000000000000000000000000000000000..9a345fff4202170073922b48ae7b6027a3bdb29b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/fdda237eb7ce577d14039e28520184a1
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1decc776d62654a118d8a2b71fd1ae2404efe3dda8fea28c3dae4f5d1d3a6198
+size 422282
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e19cfbcdb31bfff3b4c8614e37a0fc4dec27c3cd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..08417d4d5814c2ea20c97d0ff8ba914004c79527
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754582047277642455, "commit_timestamp_nsecs": 1754582047447480518, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/d/2c52bcb4795aedeec822f1bb036f23d2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/d/2c52bcb4795aedeec822f1bb036f23d2
new file mode 100644
index 0000000000000000000000000000000000000000..fd0d6e1af3c7220aae29781c9929390337e6869c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/d/2c52bcb4795aedeec822f1bb036f23d2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a3c9bd7cd240c3a550abf8e08864144b4bfef0c9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/319f21dc8f415f85c669daaabe8f9de2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/319f21dc8f415f85c669daaabe8f9de2
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/319f21dc8f415f85c669daaabe8f9de2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/419fa0a76ba1c1952b2878530927cb70 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/419fa0a76ba1c1952b2878530927cb70
new file mode 100644
index 0000000000000000000000000000000000000000..dfbfe1a61305bab6f716f83c77874d9e3176999d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/419fa0a76ba1c1952b2878530927cb70 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/529f96b6398c75c3b5abe25dc335b055 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/529f96b6398c75c3b5abe25dc335b055
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/529f96b6398c75c3b5abe25dc335b055 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/5b29a585ccb9e5436b503814559f7067 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/5b29a585ccb9e5436b503814559f7067
new file mode 100644
index 0000000000000000000000000000000000000000..0888f1001d0ffaf8287c1034f53bfeea0ac3dc96
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/5b29a585ccb9e5436b503814559f7067
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:18e6ba1dc26aee7d9c8bafcfd8b937db67be3b682246411895bf8784e83c4047
+size 253486
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/640f6360a8a96262536ecbeb62cfa2af b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/640f6360a8a96262536ecbeb62cfa2af
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/640f6360a8a96262536ecbeb62cfa2af differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/6b74e4cf8b7b4bc8040fb14a8db08463 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/6b74e4cf8b7b4bc8040fb14a8db08463
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/6b74e4cf8b7b4bc8040fb14a8db08463 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/6e71aa21fac556c6847fa1ea97592a94 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/6e71aa21fac556c6847fa1ea97592a94
new file mode 100644
index 0000000000000000000000000000000000000000..0225dad4767d76a7813da3c0573e395a4f43f048
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/6e71aa21fac556c6847fa1ea97592a94 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/a6469f100009ec813dc89697ba9cd763 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/a6469f100009ec813dc89697ba9cd763
new file mode 100644
index 0000000000000000000000000000000000000000..f4878e58e28d3301ec56f0ec2b0d8a16cbaa871c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/a6469f100009ec813dc89697ba9cd763
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c6d75dfe9312bdd44c889f18897f61cd7df5ea07c5b9102f69978b55736b289b
+size 439287
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/e26db2d88323f4b2400bd4371c3be3ce b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/e26db2d88323f4b2400bd4371c3be3ce
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/e26db2d88323f4b2400bd4371c3be3ce differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..929c8abdc3f442b1161598ebfe7c5a2321bd9af2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..e367a257aa76aaa147d11b9a5c4a2bb5fbefb807
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754579522763381415, "commit_timestamp_nsecs": 1754579522940625066, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/d/9545907eaf69cc6760977c8b41d9d07a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/d/9545907eaf69cc6760977c8b41d9d07a
new file mode 100644
index 0000000000000000000000000000000000000000..a6e42dbc01163be726e85caf7653017c34e9d7e1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/d/9545907eaf69cc6760977c8b41d9d07a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4733f0f052efd982a1872eeece86c49527dff03a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/01b5a6ff384f5ee798f9d80b4a541955 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/01b5a6ff384f5ee798f9d80b4a541955
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/01b5a6ff384f5ee798f9d80b4a541955 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/07c0a03d57f0303b66bade1ceede7167 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/07c0a03d57f0303b66bade1ceede7167
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/07c0a03d57f0303b66bade1ceede7167 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/368b831b402b4e44632b8f16425f14ba b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/368b831b402b4e44632b8f16425f14ba
new file mode 100644
index 0000000000000000000000000000000000000000..c96aec092cf2bff1254baab0734befa9e1a1e1fd
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/368b831b402b4e44632b8f16425f14ba
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:775a69e0e3012aca2de4d1739143c81b19652da45068fbb0f6fff54634b633f5
+size 336215
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/45d55f43c31194f261dcef6a2b7cbead b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/45d55f43c31194f261dcef6a2b7cbead
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/45d55f43c31194f261dcef6a2b7cbead differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/9214c9ea3e786fb4df3d67e4ce5e6701 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/9214c9ea3e786fb4df3d67e4ce5e6701
new file mode 100644
index 0000000000000000000000000000000000000000..417a8ae65c8269665c073cca702cc96f97f3b61d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/9214c9ea3e786fb4df3d67e4ce5e6701 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/cf167974db9e68870c935d163878cd04 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/cf167974db9e68870c935d163878cd04
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/cf167974db9e68870c935d163878cd04 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/d6b2dff90e7fe0fbd7184f2551d1c61b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/d6b2dff90e7fe0fbd7184f2551d1c61b
new file mode 100644
index 0000000000000000000000000000000000000000..003bc20442642bd4272a273806e26365dcd7dc20
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/d6b2dff90e7fe0fbd7184f2551d1c61b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f903c8e55a787d873ec1f1587cffca3b58741d2cc48bf2597075686d18d96b27
+size 363618
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..6b2137b2528e93dabd9405f121b73f25489fc922
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..70552ffe2c7f8b631dd935d8a3ce5d289f3868cd
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754582098886234646, "commit_timestamp_nsecs": 1754582099050061452, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/d/bec88e5ca050d849da13ac0ee9dfb568 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/d/bec88e5ca050d849da13ac0ee9dfb568
new file mode 100644
index 0000000000000000000000000000000000000000..c11466cfaa75d95bb1df7a171e1f4388b1bca743
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/d/bec88e5ca050d849da13ac0ee9dfb568 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..57b908b5b7bd1eebf28915a27812f577eb334cff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/208fed29bd0a506ca816a54acdfa1e56 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/208fed29bd0a506ca816a54acdfa1e56
new file mode 100644
index 0000000000000000000000000000000000000000..f2e929ed784eef8d87259ec873c21c2b4f1d7ccf
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/208fed29bd0a506ca816a54acdfa1e56
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b7bda317639f973d36aebe735d79f178bd5957ebe1e105ba67d8e1a65fa65e1c
+size 278926
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/3f736cca478c7e0d00e1963988917227 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/3f736cca478c7e0d00e1963988917227
new file mode 100644
index 0000000000000000000000000000000000000000..2a31377ca030aced51ef70fb458c6b5c47404cd3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/3f736cca478c7e0d00e1963988917227 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/499b24639ee21b7aeeed770854dde2b7 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/499b24639ee21b7aeeed770854dde2b7
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/499b24639ee21b7aeeed770854dde2b7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/63875870db4dee4dd72552782699e431 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/63875870db4dee4dd72552782699e431
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/63875870db4dee4dd72552782699e431 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/6f0e4da191e44ecaf9c78192cbfdd3b4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/6f0e4da191e44ecaf9c78192cbfdd3b4
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/6f0e4da191e44ecaf9c78192cbfdd3b4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/73be373d01cf5031a30a46bbabd19df1 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/73be373d01cf5031a30a46bbabd19df1
new file mode 100644
index 0000000000000000000000000000000000000000..127823415e8b1fb00f982d84f578d0e25728a20a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/73be373d01cf5031a30a46bbabd19df1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/8af4364c5ad5c47b5425d39f68af054f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/8af4364c5ad5c47b5425d39f68af054f
new file mode 100644
index 0000000000000000000000000000000000000000..96a56b73258682bc703709ca7ae0fd40fe006270
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/8af4364c5ad5c47b5425d39f68af054f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ea9724cd0a231de8425356d556b8f65092b015516b18d83d34be10d6d291c342
+size 460051
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/b19504c4df3de753c3dd4070aa17ea39 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/b19504c4df3de753c3dd4070aa17ea39
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/b19504c4df3de753c3dd4070aa17ea39 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/c850e58786b79f4e820204746f1edae9 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/c850e58786b79f4e820204746f1edae9
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/c850e58786b79f4e820204746f1edae9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3b0cc691d4063bced4709b416ec70676166c5f98
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..70956d267af06697ca5d3fd7f96c2838b384f318
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754582133528558529, "commit_timestamp_nsecs": 1754582133664585490, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/d/3aa96bc076c4c4fb7d410ceb67249ccf b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/d/3aa96bc076c4c4fb7d410ceb67249ccf
new file mode 100644
index 0000000000000000000000000000000000000000..8c02299404840146064f9c3eb27c8ce5185b23b7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/d/3aa96bc076c4c4fb7d410ceb67249ccf differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..6a9a908b02476608cbd7d641993ef4a5175dafee
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/1991ed1e10326388e3929434d80bb932 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/1991ed1e10326388e3929434d80bb932
new file mode 100644
index 0000000000000000000000000000000000000000..a8a67f01739819c61dba52943006e4781ba0ff0b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/1991ed1e10326388e3929434d80bb932
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e25d44eb6cef8de060794e1dd7a710bceb014bef13584d1b96efa46e6e761f48
+size 287451
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/51d41f7d11bdf6ef049489d723edb148 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/51d41f7d11bdf6ef049489d723edb148
new file mode 100644
index 0000000000000000000000000000000000000000..9b1d59858ca249a965b7a7fe941c8930c856e587
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/51d41f7d11bdf6ef049489d723edb148 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/54acd42cf12fc01e240e1f1675d649b6 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/54acd42cf12fc01e240e1f1675d649b6
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/54acd42cf12fc01e240e1f1675d649b6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/82660dfa03fbf45bfead42a8a6eb3774 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/82660dfa03fbf45bfead42a8a6eb3774
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/82660dfa03fbf45bfead42a8a6eb3774 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/922635c47cf7254c931bf9785c64c172 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/922635c47cf7254c931bf9785c64c172
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/922635c47cf7254c931bf9785c64c172 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/9912e45446580d2a4c191426126f2ce1 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/9912e45446580d2a4c191426126f2ce1
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/9912e45446580d2a4c191426126f2ce1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/db8aa2fdc2868972a5d158b6f4fe76e9 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/db8aa2fdc2868972a5d158b6f4fe76e9
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/db8aa2fdc2868972a5d158b6f4fe76e9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/dd8bee54ed91e91ab2227eefb0b64e7a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/dd8bee54ed91e91ab2227eefb0b64e7a
new file mode 100644
index 0000000000000000000000000000000000000000..6bb46ad75b54184215cf4282549a1c81efd2ac8b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/dd8bee54ed91e91ab2227eefb0b64e7a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:91569808b080902842f8458a22deb34012687655a40c7c086354e5864e442669
+size 417759
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..ea2e812013c069bf298e4e8ee2110f3257bc1b7b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..d5275c6b5627ffc56058df56bdfcd649396b1125
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754582168637658985, "commit_timestamp_nsecs": 1754582168784669963, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/d/6af08ae2f858c20eb00b0a08b7ac0c8b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/d/6af08ae2f858c20eb00b0a08b7ac0c8b
new file mode 100644
index 0000000000000000000000000000000000000000..127b4bbcd04e48028790160ccf3ec93ae3fe34a0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/d/6af08ae2f858c20eb00b0a08b7ac0c8b differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f220185ede147240e994ff5193006aa32248174b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/150ba43034b57be4a8c7e4b216ca2233 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/150ba43034b57be4a8c7e4b216ca2233
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/150ba43034b57be4a8c7e4b216ca2233 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/35c382075a2cd1ee1b858ce87f8d2fbd b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/35c382075a2cd1ee1b858ce87f8d2fbd
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/35c382075a2cd1ee1b858ce87f8d2fbd differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/35e9a280ee1bf2fde5523442ed64a415 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/35e9a280ee1bf2fde5523442ed64a415
new file mode 100644
index 0000000000000000000000000000000000000000..c86807c736121308b2e48c7c8ec428cd75a549fa
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/35e9a280ee1bf2fde5523442ed64a415 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/49d33d076910ced05911075a401f527c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/49d33d076910ced05911075a401f527c
new file mode 100644
index 0000000000000000000000000000000000000000..8df451bc2f8aeb5ad322889782fb5b032fe992c9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/49d33d076910ced05911075a401f527c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f61e876626267c6a40956612220848ab8298c3534a3f24eeb8eb7c4e506ec4dc
+size 460304
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/8942f1af8da1557ff84b7ff8f4176d78 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/8942f1af8da1557ff84b7ff8f4176d78
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/8942f1af8da1557ff84b7ff8f4176d78 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/9041a044f9740d0c7cf624924ac73f80 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/9041a044f9740d0c7cf624924ac73f80
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/9041a044f9740d0c7cf624924ac73f80 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/9495287317b56964ade04f87e03fb646 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/9495287317b56964ade04f87e03fb646
new file mode 100644
index 0000000000000000000000000000000000000000..e4cec05c2aa40d31196589a5d6f8abcce90a1454
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/9495287317b56964ade04f87e03fb646
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:982ed84a77f4b3a6cccf78a4a9bb7939ac66d7afd16011be87b21a84e8014eb7
+size 278901
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/b717dccfc4d671bef16dcf2c99cae0d3 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/b717dccfc4d671bef16dcf2c99cae0d3
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/b717dccfc4d671bef16dcf2c99cae0d3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1f94a138ec0372bf8d05cebe9708645bf66dae25
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..4fe2ea7014c3c09f264df24e1c4cdef70159b262
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754582202879735358, "commit_timestamp_nsecs": 1754582203048719843, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/d/6e4a2e619ecdca314240b0f8c7d1b677 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/d/6e4a2e619ecdca314240b0f8c7d1b677
new file mode 100644
index 0000000000000000000000000000000000000000..2c24161ed9dd15ce3b5c207494cc008467007a48
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/d/6e4a2e619ecdca314240b0f8c7d1b677 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..431de7266b9aa7c615f8cc70502ea9b79e25caee
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/0e3e314ba95fa90942e44b0e77385bea b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/0e3e314ba95fa90942e44b0e77385bea
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/0e3e314ba95fa90942e44b0e77385bea differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/0e7134d43fc5928254ca317931f8963e b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/0e7134d43fc5928254ca317931f8963e
new file mode 100644
index 0000000000000000000000000000000000000000..65529f6532287fad4b450ca493011b55e8bf7f92
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/0e7134d43fc5928254ca317931f8963e differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/33c6517b8a5c975e316d05c2161905b6 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/33c6517b8a5c975e316d05c2161905b6
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/33c6517b8a5c975e316d05c2161905b6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/71d2846c5401cd3454d36c4800ac9fb1 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/71d2846c5401cd3454d36c4800ac9fb1
new file mode 100644
index 0000000000000000000000000000000000000000..5aab524b4a096642a53f3b6a5db60fd86f7f19b3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/71d2846c5401cd3454d36c4800ac9fb1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/b74be78bac76ff58da5a9da71fef321a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/b74be78bac76ff58da5a9da71fef321a
new file mode 100644
index 0000000000000000000000000000000000000000..e6318aa74b4285ce341e000706192ab5a9e58819
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/b74be78bac76ff58da5a9da71fef321a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:565caae1fcedb609ccc42788fce572c1f3302a3f8a994f90e58b6e7def0d9d10
+size 434552
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/c12f49dfcd319f64ae928c7d21d8f3dc b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/c12f49dfcd319f64ae928c7d21d8f3dc
new file mode 100644
index 0000000000000000000000000000000000000000..9a639bcaa3357d4683e82ee2cca9dfe7f9fa74de
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/c12f49dfcd319f64ae928c7d21d8f3dc
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:39ce326240feeec0369a83f8f01ee30289f6d1e9ac545cbb1bcd9e312caf1e3c
+size 304624
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/cfff90758bcf909a587a33fa2f4a3ae1 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/cfff90758bcf909a587a33fa2f4a3ae1
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/cfff90758bcf909a587a33fa2f4a3ae1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/fd0fc73357ed9a51362a73bc84773b33 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/fd0fc73357ed9a51362a73bc84773b33
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/fd0fc73357ed9a51362a73bc84773b33 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..fda122c7be3738bdf05dd5e0f72f4705a6d56902
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..1b3bf99407e5db98cac164517e6d1c1b162da070
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754582235269559042, "commit_timestamp_nsecs": 1754582235434720134, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/d/69e80c174fba07ed85419e586336bd81 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/d/69e80c174fba07ed85419e586336bd81
new file mode 100644
index 0000000000000000000000000000000000000000..0c10771b73550f47ae391e452393366fcd612989
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/d/69e80c174fba07ed85419e586336bd81 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..2c512be7dcb993a8ddd1fbe1143aebab34d55008
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/0328838b6df83a5d443446ee4dce0f32 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/0328838b6df83a5d443446ee4dce0f32
new file mode 100644
index 0000000000000000000000000000000000000000..0e90f5742c4949088d8c1f68320081bfac38b826
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/0328838b6df83a5d443446ee4dce0f32
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:112a941efa6472abbbcf6bb497741afa343b4fe35ada8349fc4a96e66d4e5330
+size 364117
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/06cf8235e8d32b1fdf8beec280dee2fb b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/06cf8235e8d32b1fdf8beec280dee2fb
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/06cf8235e8d32b1fdf8beec280dee2fb differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/22bf81cd763a44cd05ca2b4e00b51b71 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/22bf81cd763a44cd05ca2b4e00b51b71
new file mode 100644
index 0000000000000000000000000000000000000000..6475bc401b94d8f82670bf6565033ff8263fba1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/22bf81cd763a44cd05ca2b4e00b51b71 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/304115fa3fdffc8daf39d7b9a63c40e2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/304115fa3fdffc8daf39d7b9a63c40e2
new file mode 100644
index 0000000000000000000000000000000000000000..e35d51f85ea41d8d896f7589aa7715d059d63d76
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/304115fa3fdffc8daf39d7b9a63c40e2
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:356e306b3101ee9e600cb587040895d71d6b76ba6271b32265736c0889847457
+size 336819
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/72a1ab188db11e9f5d7192c599e36976 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/72a1ab188db11e9f5d7192c599e36976
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/72a1ab188db11e9f5d7192c599e36976 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/731662827a88cd3121193ecabdd63764 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/731662827a88cd3121193ecabdd63764
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/731662827a88cd3121193ecabdd63764 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/a5edec3b2570d8e40fdf5df134a914de b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/a5edec3b2570d8e40fdf5df134a914de
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/a5edec3b2570d8e40fdf5df134a914de differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..233e0e9dc4ef198db125fd1571a54f9b26aca63a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..62a1fa1c3fc13d7c9b16dfdd348da186317892e4
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754582268853763286, "commit_timestamp_nsecs": 1754582268984378658, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/d/94d28ebcece016eff3a6f7ce48501979 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/d/94d28ebcece016eff3a6f7ce48501979
new file mode 100644
index 0000000000000000000000000000000000000000..d6f7857173a97156960f947d0bd95b757d86e0d4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/d/94d28ebcece016eff3a6f7ce48501979 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..56f8b816377b4d8a0c2db1fcb41a9778a44535ff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/19387ffb5598d72dcfc1af15b81a4288 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/19387ffb5598d72dcfc1af15b81a4288
new file mode 100644
index 0000000000000000000000000000000000000000..bac4c357a946521da582b587a7cc3ee9093fbbd3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/19387ffb5598d72dcfc1af15b81a4288
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f92569ba977b494b6ff5e7297b125b26c2c2dea8a0451cd33c2e00b682ca0769
+size 327954
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/4abf08ad9ff543fc451aaddb6571ee79 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/4abf08ad9ff543fc451aaddb6571ee79
new file mode 100644
index 0000000000000000000000000000000000000000..017e0196679d97979bc245a9a6ea22dcd53b8740
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/4abf08ad9ff543fc451aaddb6571ee79
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:88bed05be0e60fb864d20308b8bffe7640eb4efe14f572c2f8061053a1d2940b
+size 372645
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/5b83e43ac62009049848575396a377b7 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/5b83e43ac62009049848575396a377b7
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/5b83e43ac62009049848575396a377b7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/c27d27c43e890a2f40cbd5531b420986 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/c27d27c43e890a2f40cbd5531b420986
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/c27d27c43e890a2f40cbd5531b420986 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/db8f5ca33655f78a317c07e745485d7d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/db8f5ca33655f78a317c07e745485d7d
new file mode 100644
index 0000000000000000000000000000000000000000..8ebc9a5279f3138e82c83fe15b9c0d2d35f4ea9f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/db8f5ca33655f78a317c07e745485d7d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/e7aec90750e7e75872a0441d859c9507 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/e7aec90750e7e75872a0441d859c9507
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/e7aec90750e7e75872a0441d859c9507 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/f4be0a25dfffecc203d0bec702ad750e b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/f4be0a25dfffecc203d0bec702ad750e
new file mode 100644
index 0000000000000000000000000000000000000000..85f35e7991cac7b6fc244bb2f277438119f4e90b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/f4be0a25dfffecc203d0bec702ad750e differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4569befee4f08db0bba48c893b755fa85ff2e41f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..01b879e31b5ed46d15a668d2d5731df8f96222da
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754582304036877519, "commit_timestamp_nsecs": 1754582304188612801, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/d/142cca6f748aab7a76f445ee1e11f93a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/d/142cca6f748aab7a76f445ee1e11f93a
new file mode 100644
index 0000000000000000000000000000000000000000..e0c26eb153cfa520e13456d5c628af13eb591bf6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/d/142cca6f748aab7a76f445ee1e11f93a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a7e80540510e820591cec180ff6cb048f6b282a7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/377f08a125187fe7ec2ee17e668c0433 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/377f08a125187fe7ec2ee17e668c0433
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/377f08a125187fe7ec2ee17e668c0433 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/63bc6acdc16578add22e0f85c7c7b717 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/63bc6acdc16578add22e0f85c7c7b717
new file mode 100644
index 0000000000000000000000000000000000000000..001a5c69e5686f81aff4bf3174afa24eefb2003b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/63bc6acdc16578add22e0f85c7c7b717 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/7b430006749f0a454a60f78b1a22cd29 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/7b430006749f0a454a60f78b1a22cd29
new file mode 100644
index 0000000000000000000000000000000000000000..065e3c2840feee940e19f81f7cebf36898800d24
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/7b430006749f0a454a60f78b1a22cd29
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:84d736546e21fa48059cbc09fb6199e1f3b10684bea17dca9c57cda6da38806f
+size 456404
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/9180c1a81d9b6c196d223b58229cb093 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/9180c1a81d9b6c196d223b58229cb093
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/9180c1a81d9b6c196d223b58229cb093 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/ae6ac0fe7c0537787c0cb278ac7bafd5 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/ae6ac0fe7c0537787c0cb278ac7bafd5
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/ae6ac0fe7c0537787c0cb278ac7bafd5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/bba18a4656339fa15a4a90f9c3f27503 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/bba18a4656339fa15a4a90f9c3f27503
new file mode 100644
index 0000000000000000000000000000000000000000..1d728012e29191a536c38f5b9c91a4248ad85f8a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/bba18a4656339fa15a4a90f9c3f27503
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aaf60561979c94138bc8747e63539e51697930ed03e0ccf0432d2fd4daf5bbd4
+size 244997
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/f0398f630720f6455626e65243386f96 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/f0398f630720f6455626e65243386f96
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/f0398f630720f6455626e65243386f96 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/f49d0d9d1a1a221e2f61665f531d15b0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/f49d0d9d1a1a221e2f61665f531d15b0
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/f49d0d9d1a1a221e2f61665f531d15b0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..fad1ac9036871992e87a7ebc43d59d94b1ba4a3a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..1cd7fa5f1b4fcc4f16e00735838d692c3a2929d1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754582340226407188, "commit_timestamp_nsecs": 1754582340360198096, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/d/5854cff067151dca2bf8abc70a50c19f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/d/5854cff067151dca2bf8abc70a50c19f
new file mode 100644
index 0000000000000000000000000000000000000000..46d06243887c057192fad09a31c6bd54c104cf41
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/d/5854cff067151dca2bf8abc70a50c19f differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a27adac1f922953f6d38506b2312e78837ebe950
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/2346249c3b04b56b8adf47e265c68760 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/2346249c3b04b56b8adf47e265c68760
new file mode 100644
index 0000000000000000000000000000000000000000..4859214e6b68fa6e697de9748417fcf43be8081d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/2346249c3b04b56b8adf47e265c68760 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/75c2ab2733c57ef8ac39f8e6678cbc97 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/75c2ab2733c57ef8ac39f8e6678cbc97
new file mode 100644
index 0000000000000000000000000000000000000000..3a14df57c9058ec709754ea96627acb7a298cd63
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/75c2ab2733c57ef8ac39f8e6678cbc97
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1d3b7f8662875decfae1989bd6cc70de31340fd05ef3d095c5563076efa41b1c
+size 371052
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/7e595f0558392eddb3039ffdc0c75ee7 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/7e595f0558392eddb3039ffdc0c75ee7
new file mode 100644
index 0000000000000000000000000000000000000000..3a04c90becd788fb470ef27d65daf7655b3c3122
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/7e595f0558392eddb3039ffdc0c75ee7
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6a287f5b05d581c9bcf96bec88dc6499d46322f42e41b2135761eeee656626b9
+size 281287
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/d7fc8e5f189bf00f63a700a7d9ea7cad b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/d7fc8e5f189bf00f63a700a7d9ea7cad
new file mode 100644
index 0000000000000000000000000000000000000000..f28d59e1f9187cfd65b688b4bceb57a81f0acdcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/d7fc8e5f189bf00f63a700a7d9ea7cad differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/daf281e0302c1fda421ae7f137ddc947 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/daf281e0302c1fda421ae7f137ddc947
new file mode 100644
index 0000000000000000000000000000000000000000..f13880cce21e8943a341e65e239077718bb94855
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/daf281e0302c1fda421ae7f137ddc947 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/e6f26a4f3be83eb3ee8c4128beb9f77a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/e6f26a4f3be83eb3ee8c4128beb9f77a
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/e6f26a4f3be83eb3ee8c4128beb9f77a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/ff4719766d9ce1652b85d9830055b1a1 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/ff4719766d9ce1652b85d9830055b1a1
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/ff4719766d9ce1652b85d9830055b1a1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1d2457e8b9a3d0d5d19dfb36d473ddbde7275d5c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..cab8bc23cd64448802b9ffd2bf8ab4dbd0748299
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754582374852149635, "commit_timestamp_nsecs": 1754582375023719868, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/d/89f6ebe9a40307f1402092c75a30f162 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/d/89f6ebe9a40307f1402092c75a30f162
new file mode 100644
index 0000000000000000000000000000000000000000..9d5a6890664177ae307cf4aca8a773462538db5e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/d/89f6ebe9a40307f1402092c75a30f162 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b387b6bf5e9c1a77b66d48289260b0325679a44c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/0813e22e0930c3a9cef07eaed3e0c623 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/0813e22e0930c3a9cef07eaed3e0c623
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/0813e22e0930c3a9cef07eaed3e0c623 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/120f98e39fc7a0d88e132ff440a0d2e7 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/120f98e39fc7a0d88e132ff440a0d2e7
new file mode 100644
index 0000000000000000000000000000000000000000..2b50b315de4ca488a519e6ed767f77a1065a1d22
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/120f98e39fc7a0d88e132ff440a0d2e7
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4fda9ba3848a025bee8aa5d32e317aa3bc97bf9f87ecc92a04d28a734e6fe771
+size 398107
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/4481bd7afea1202541fe7b337bd0481d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/4481bd7afea1202541fe7b337bd0481d
new file mode 100644
index 0000000000000000000000000000000000000000..ded7a815faba3cf503fb1756148b3f0595a0c6ea
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/4481bd7afea1202541fe7b337bd0481d
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:95e083f482ccdc4a9d3e1bf4c57816bf0592fc2b63656494c88e48485babee0f
+size 340678
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/ab69870a967aee8ae3a3708ea20d7dbd b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/ab69870a967aee8ae3a3708ea20d7dbd
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/ab69870a967aee8ae3a3708ea20d7dbd differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/ac2129f157b4ecd8066124de7e358522 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/ac2129f157b4ecd8066124de7e358522
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/ac2129f157b4ecd8066124de7e358522 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/dc1db6a3826c52e7a83257ddf1c30a33 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/dc1db6a3826c52e7a83257ddf1c30a33
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/dc1db6a3826c52e7a83257ddf1c30a33 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/e765dc78500b24726234bc32ca81b33c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/e765dc78500b24726234bc32ca81b33c
new file mode 100644
index 0000000000000000000000000000000000000000..237ba36ee3d50f72eb496c302accbd61d471d725
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/e765dc78500b24726234bc32ca81b33c differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a211bd61189ce50bede9bddac71afce762cedfc6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..a5a4a957feca24a656206108d5f619db29fa7f99
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754582409486546079, "commit_timestamp_nsecs": 1754582409647307404, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/d/56edb80e3c84b50b6157ffc3c6149c8f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/d/56edb80e3c84b50b6157ffc3c6149c8f
new file mode 100644
index 0000000000000000000000000000000000000000..7d3d8e76aba537f5cb6fe616238dc8dabc99beda
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/d/56edb80e3c84b50b6157ffc3c6149c8f differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8ee3727bbc2c7c2709ca1174444234b07ed66213
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/1df0299962a5e93df60f242edad4ae11 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/1df0299962a5e93df60f242edad4ae11
new file mode 100644
index 0000000000000000000000000000000000000000..f38c5b7a951e5be24cb300e50de61e956f89649d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/1df0299962a5e93df60f242edad4ae11
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:585a6297fc4587bf52d07102ea8898b9c50e69958476875c9490ed2b1dda52ae
+size 244958
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/211756e4fc3286b3a4d895ee3c1763cc b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/211756e4fc3286b3a4d895ee3c1763cc
new file mode 100644
index 0000000000000000000000000000000000000000..b8c756c3806884b7db65eb7bea8d827a01682b0e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/211756e4fc3286b3a4d895ee3c1763cc
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ec765c6f1ae17d2dced53dc44285b4210d9ee098a92c5d52f136ba248ea69994
+size 494203
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/6165ca0ae647f18cc2a7d7b5f815988a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/6165ca0ae647f18cc2a7d7b5f815988a
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/6165ca0ae647f18cc2a7d7b5f815988a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/b4efdf49c5a3f70484bbfc3b8e0e093c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/b4efdf49c5a3f70484bbfc3b8e0e093c
new file mode 100644
index 0000000000000000000000000000000000000000..6de91a75f804e3ac0c806f160e5d4a9d1fcd3b99
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/b4efdf49c5a3f70484bbfc3b8e0e093c differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/cc80d69a6d2ff76de9a3d36ac8d42d06 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/cc80d69a6d2ff76de9a3d36ac8d42d06
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/cc80d69a6d2ff76de9a3d36ac8d42d06 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/cf83c7dd41974efdec044687bae3e911 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/cf83c7dd41974efdec044687bae3e911
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/cf83c7dd41974efdec044687bae3e911 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/e656d7393260d527e45a6ceda4ada5d2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/e656d7393260d527e45a6ceda4ada5d2
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/e656d7393260d527e45a6ceda4ada5d2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/ea6b9c2fcc45e254e922ace8559912db b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/ea6b9c2fcc45e254e922ace8559912db
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/ea6b9c2fcc45e254e922ace8559912db differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..ec0d83c9cc71fd0c4f9ecde49c72e7616a0965b3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..2664d25ece52405c9f04a8f3bbfc5926c9d4eda9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754579557432891445, "commit_timestamp_nsecs": 1754579557580627193, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/d/d65b386ab81675fc354f9754f043a524 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/d/d65b386ab81675fc354f9754f043a524
new file mode 100644
index 0000000000000000000000000000000000000000..6e275696387bf3624efa3e338057ddf5b0470796
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/d/d65b386ab81675fc354f9754f043a524 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1e8ba30abffc023c43aab0f527ed38a6ea95fcda
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/1b7f6d12e7f57f8ab227191a7f3fbcb7 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/1b7f6d12e7f57f8ab227191a7f3fbcb7
new file mode 100644
index 0000000000000000000000000000000000000000..936c7d7d4d07bce17e3f08e9dd055a3ba3a43a54
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/1b7f6d12e7f57f8ab227191a7f3fbcb7
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:23f4a8ff0293c01d8d3dc4cf0aedac7b4b2a468df0180d49711153a580fb4c24
+size 442508
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/376c0ce0510eb05e59de80ba85edea17 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/376c0ce0510eb05e59de80ba85edea17
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/376c0ce0510eb05e59de80ba85edea17 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/948cb7e84db59fbbb325f442d2b871e4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/948cb7e84db59fbbb325f442d2b871e4
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/948cb7e84db59fbbb325f442d2b871e4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/af812c069cb38026fb08104494de5347 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/af812c069cb38026fb08104494de5347
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/af812c069cb38026fb08104494de5347 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/bb0af4c49835d4b8e346c8c764a0c6b1 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/bb0af4c49835d4b8e346c8c764a0c6b1
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/bb0af4c49835d4b8e346c8c764a0c6b1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/eb37db0748dc518562c7911c2ff72837 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/eb37db0748dc518562c7911c2ff72837
new file mode 100644
index 0000000000000000000000000000000000000000..fde6b51b45ccf9e14e00c507b8472168e3f000b7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/eb37db0748dc518562c7911c2ff72837
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7320a25506e32c76fa7e947f955f2a1385d48207806370ecafaa718f19e9f29b
+size 261720
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/efc037a63aa17ee2e330d34c2cbbe07a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/efc037a63aa17ee2e330d34c2cbbe07a
new file mode 100644
index 0000000000000000000000000000000000000000..6485a6e54615ae92737ad611f4c53ff9d21e7500
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/efc037a63aa17ee2e330d34c2cbbe07a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/f13e7d70c690bfc8ebfe5d5863e29830 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/f13e7d70c690bfc8ebfe5d5863e29830
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/f13e7d70c690bfc8ebfe5d5863e29830 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8702910d22b8fa5ecf3828a75f51db0d63fcdb3b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..27cb8a2cdfc7e23e9e4ba9a53949c40b23ef5396
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754582443283971476, "commit_timestamp_nsecs": 1754582443460245171, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/d/e382283069af756dfebfb85bfde1ba3c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/d/e382283069af756dfebfb85bfde1ba3c
new file mode 100644
index 0000000000000000000000000000000000000000..32c494266dd95158da1e9a849e572efc4366c61e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/d/e382283069af756dfebfb85bfde1ba3c differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8402d5050e279dfc711e678b9f9842183a1dbe61
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/09d9ec642b647a62e93313eabebdaab4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/09d9ec642b647a62e93313eabebdaab4
new file mode 100644
index 0000000000000000000000000000000000000000..17477d0d85afdabe981dfd15959c72d1434d7280
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/09d9ec642b647a62e93313eabebdaab4
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d509c7ae781407d33366f65d51c4b54ca92463441589779f991b1335756edb6b
+size 278936
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/3684e0c5fac0a400362ab61c4784a933 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/3684e0c5fac0a400362ab61c4784a933
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/3684e0c5fac0a400362ab61c4784a933 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/6185d3d44b5e5c9055befc2eff2037f2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/6185d3d44b5e5c9055befc2eff2037f2
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/6185d3d44b5e5c9055befc2eff2037f2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/8b7823472ad2c653d79e4d748c5873fb b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/8b7823472ad2c653d79e4d748c5873fb
new file mode 100644
index 0000000000000000000000000000000000000000..75525c87d9197b2c12ab79545eaaff817279ed93
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/8b7823472ad2c653d79e4d748c5873fb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:77f04234b63926816a2969d88cbf4be1cf8ef97c19a7f7c6bd3ee250ec88605e
+size 422506
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/bdf6778ffbd894aad796f683a58827df b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/bdf6778ffbd894aad796f683a58827df
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/bdf6778ffbd894aad796f683a58827df differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/c046e05c53d8a8cc3504734cc454af64 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/c046e05c53d8a8cc3504734cc454af64
new file mode 100644
index 0000000000000000000000000000000000000000..d3c10c1a7375ac54ebab893ef317a00112270525
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/c046e05c53d8a8cc3504734cc454af64 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/c42da0a324a38b93dff7d776a6936c37 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/c42da0a324a38b93dff7d776a6936c37
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/c42da0a324a38b93dff7d776a6936c37 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/dc5b04ed8454745906b8644aabfcc10a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/dc5b04ed8454745906b8644aabfcc10a
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/dc5b04ed8454745906b8644aabfcc10a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/edcfcaa245025b1327eda689487da96d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/edcfcaa245025b1327eda689487da96d
new file mode 100644
index 0000000000000000000000000000000000000000..65529f6532287fad4b450ca493011b55e8bf7f92
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/edcfcaa245025b1327eda689487da96d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..2216802c3faefbd198d879808f27d4230485a7bc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..442ebf464d0eb5646a5f144a7d51ed33e63a8902
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754582479421613362, "commit_timestamp_nsecs": 1754582479620112475, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/d/3faf3ba284bfd5f6989d5cc0d11f8f5f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/d/3faf3ba284bfd5f6989d5cc0d11f8f5f
new file mode 100644
index 0000000000000000000000000000000000000000..73f0a87fa67ccba5ddddb01671e31f2e5d984429
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/d/3faf3ba284bfd5f6989d5cc0d11f8f5f differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..286d5d6ab0ec2b5ccf9a6f70b7e5825cc433f2f6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/042df637e41f88d9638041d095117bc8 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/042df637e41f88d9638041d095117bc8
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/042df637e41f88d9638041d095117bc8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/20828b697f92106a48a3640951d792a6 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/20828b697f92106a48a3640951d792a6
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/20828b697f92106a48a3640951d792a6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/235b3e7f5b8732da8ea15fecded19fb2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/235b3e7f5b8732da8ea15fecded19fb2
new file mode 100644
index 0000000000000000000000000000000000000000..b4b8cc4b85e19301e9f56577be94d16802fd0c44
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/235b3e7f5b8732da8ea15fecded19fb2
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fbe31eb9d6fad5d14801902b409e5e453e8e98c2b7fc8f6902600ceda4ac1580
+size 422250
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/2d296502594065183ec12d588ed8edea b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/2d296502594065183ec12d588ed8edea
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/2d296502594065183ec12d588ed8edea differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/40d0bbb189cbc2a9e278ade801deece0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/40d0bbb189cbc2a9e278ade801deece0
new file mode 100644
index 0000000000000000000000000000000000000000..a74cdda8a6d74a51d157a3412389a4fef40beaf4
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/40d0bbb189cbc2a9e278ade801deece0
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:00bc38e3c47f4bb57465bf408fc595761e2224dbec98ebb57a0bef3128059637
+size 278948
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/4f1f20f41538cf0360573392c0f6f268 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/4f1f20f41538cf0360573392c0f6f268
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/4f1f20f41538cf0360573392c0f6f268 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/abad64f3d917cd4db57d6866e59cde3d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/abad64f3d917cd4db57d6866e59cde3d
new file mode 100644
index 0000000000000000000000000000000000000000..72a2ebb3c41ad8852d2b8533decb5bca09bcb7d8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/abad64f3d917cd4db57d6866e59cde3d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..2b39859782590db4de6f588791c9bd42460f5b7a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..007e5e7c1d0d2cc158c001cc88bc4e0eab34553a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754582513520388930, "commit_timestamp_nsecs": 1754582513674095719, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/d/a370a69f1ba6e4d34e6ea09c36da446d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/d/a370a69f1ba6e4d34e6ea09c36da446d
new file mode 100644
index 0000000000000000000000000000000000000000..8ab7bd6a1a83d10f9c0fc0049e7f1f8aff79ce95
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/d/a370a69f1ba6e4d34e6ea09c36da446d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5828da212a6563382f80456ed24265a0420c12c6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/0509468c166fee8019c6ae42d7b0224f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/0509468c166fee8019c6ae42d7b0224f
new file mode 100644
index 0000000000000000000000000000000000000000..40634046af9aca434de326184cff71c5a1131bd2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/0509468c166fee8019c6ae42d7b0224f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a2d26572a0f2fe1023ca14c042d9db34005d32f8329e89211341f96cd92366cc
+size 528216
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/44af2a784c820e6b5cf3148b29d7c243 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/44af2a784c820e6b5cf3148b29d7c243
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/44af2a784c820e6b5cf3148b29d7c243 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/8bbf7e6343b1d092865583bd465f0a64 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/8bbf7e6343b1d092865583bd465f0a64
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/8bbf7e6343b1d092865583bd465f0a64 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/a28c99bc4b78e20cd9d41f617f35334e b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/a28c99bc4b78e20cd9d41f617f35334e
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/a28c99bc4b78e20cd9d41f617f35334e differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/a80664dc98e196de4bcd3f5a5e15cfa6 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/a80664dc98e196de4bcd3f5a5e15cfa6
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/a80664dc98e196de4bcd3f5a5e15cfa6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/af185e5143bfd4ec2fe03c1665f58f1b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/af185e5143bfd4ec2fe03c1665f58f1b
new file mode 100644
index 0000000000000000000000000000000000000000..d3bc31bb4d0e134beea01b061a17a7a7bbba3ff1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/af185e5143bfd4ec2fe03c1665f58f1b differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/b2bba28a29234d38961a1f88147b0f54 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/b2bba28a29234d38961a1f88147b0f54
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/b2bba28a29234d38961a1f88147b0f54 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/d0eca04fd9a29ad360dba533b4cf01a6 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/d0eca04fd9a29ad360dba533b4cf01a6
new file mode 100644
index 0000000000000000000000000000000000000000..ec41e8f4fcf1c7cf725c17d8ab2c1c7bb3256f53
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/d0eca04fd9a29ad360dba533b4cf01a6
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0499f7b9e18ab84936301a7987a6345e831388c9a7509b887cc1c9da2ac66a63
+size 210910
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..71104ddea8ddcc99532c54af409e222ad51f7981
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..8042f69a6c55d2157a28a22cc84d50f21b4f4d4e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754582548056981180, "commit_timestamp_nsecs": 1754582548241747843, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/d/acfcc9aa5445b68b325027088b1acace b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/d/acfcc9aa5445b68b325027088b1acace
new file mode 100644
index 0000000000000000000000000000000000000000..44b650f225dd242b4c4b3f195be96ddbe2971828
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/d/acfcc9aa5445b68b325027088b1acace differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0d53a09a6949bbe6dde8135d59e4834aedd7ddd2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/1b9c42469fb3470494aacc7e2b0249a3 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/1b9c42469fb3470494aacc7e2b0249a3
new file mode 100644
index 0000000000000000000000000000000000000000..6250cbf23a0d09d83458d3289e8f335d15794b1c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/1b9c42469fb3470494aacc7e2b0249a3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/482ba630b0b23933aebedde5e35b98b8 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/482ba630b0b23933aebedde5e35b98b8
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/482ba630b0b23933aebedde5e35b98b8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/65cec42acf5abb1fa2499276234b6cda b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/65cec42acf5abb1fa2499276234b6cda
new file mode 100644
index 0000000000000000000000000000000000000000..d7a8a5711d9948732e87392e9904080158728e13
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/65cec42acf5abb1fa2499276234b6cda
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b7e79386787113808591f23cce9ca9ecf3a0aed7da6fd3cbbc529c71729aaaae
+size 411999
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/844e1e88687c5c5c7f7809f87342323c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/844e1e88687c5c5c7f7809f87342323c
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/844e1e88687c5c5c7f7809f87342323c differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/a79388175eae3f549f07f1d017049754 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/a79388175eae3f549f07f1d017049754
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/a79388175eae3f549f07f1d017049754 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/cafffc94286657a10a4f3fc51bf2f304 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/cafffc94286657a10a4f3fc51bf2f304
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/cafffc94286657a10a4f3fc51bf2f304 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/f2b4d49cec84f246f382f283d599db88 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/f2b4d49cec84f246f382f283d599db88
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/f2b4d49cec84f246f382f283d599db88 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/fd52a2813009e105900375a96f457911 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/fd52a2813009e105900375a96f457911
new file mode 100644
index 0000000000000000000000000000000000000000..d4a3ec8d5e39f58b5c9d04a67767da4e79dd2f71
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/fd52a2813009e105900375a96f457911
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:07b2c68bf17f355a268fc9bb5d92bf5bc7e7ef9d13c1ab836cd8f7f05055400f
+size 287512
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/fe7432225727035875b375712bb5dca4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/fe7432225727035875b375712bb5dca4
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/fe7432225727035875b375712bb5dca4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c7e71936d89cc663f252f5d5097309fbe741ce3c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..6322e596bdd3eb6722046cb409a828d8eef71725
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754582581665960449, "commit_timestamp_nsecs": 1754582581842439209, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/d/68004481854206adcc2698a5c48b7011 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/d/68004481854206adcc2698a5c48b7011
new file mode 100644
index 0000000000000000000000000000000000000000..d8754135d99468a8f5d1731fc4939223e87d40ba
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/d/68004481854206adcc2698a5c48b7011 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..aad0373045ddd4806642778dc58ff55ea92b3536
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/051efa471a7bcc7ad392fa3826fccb67 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/051efa471a7bcc7ad392fa3826fccb67
new file mode 100644
index 0000000000000000000000000000000000000000..193d616ceebf32fbbf724329b5f56f6b050f089e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/051efa471a7bcc7ad392fa3826fccb67
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7ea6a61c00d9e788c6514cb9b520379490369f11c5b2240d732be9df37f30148
+size 443194
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/5ffc964b427ca5df75a2cfef73e7d352 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/5ffc964b427ca5df75a2cfef73e7d352
new file mode 100644
index 0000000000000000000000000000000000000000..690f586864a8bfa5ebab2cb498eebd93411e39d1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/5ffc964b427ca5df75a2cfef73e7d352
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3fc743234957950dfe14f95952f8fc8400a80a32c50d89b3f27b875a03827d5f
+size 261901
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/690309b6c62cd40ff5d67bb6e14bf45e b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/690309b6c62cd40ff5d67bb6e14bf45e
new file mode 100644
index 0000000000000000000000000000000000000000..ec745d97353f97209ef50f0eb738585830219f53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/690309b6c62cd40ff5d67bb6e14bf45e differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/9413d6eaa51d66ded39dd336cf260333 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/9413d6eaa51d66ded39dd336cf260333
new file mode 100644
index 0000000000000000000000000000000000000000..68a9d16589524b3605dd9d53bde935104828bb50
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/9413d6eaa51d66ded39dd336cf260333 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/c8d7dab0839588063642d42911b2b275 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/c8d7dab0839588063642d42911b2b275
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/c8d7dab0839588063642d42911b2b275 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/e3b1a7999a9da11f54f24aaac47fdc93 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/e3b1a7999a9da11f54f24aaac47fdc93
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/e3b1a7999a9da11f54f24aaac47fdc93 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/fbe573406f5fb14072e0fa1d6b1a454a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/fbe573406f5fb14072e0fa1d6b1a454a
new file mode 100644
index 0000000000000000000000000000000000000000..dfbfe1a61305bab6f716f83c77874d9e3176999d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/fbe573406f5fb14072e0fa1d6b1a454a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..38b438bca6358c65df860170ad125a04c9d1568e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..6becb5f75385d03d9e2293294c7c975009ee3f87
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754582615723412201, "commit_timestamp_nsecs": 1754582615876664258, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/d/4644b6ed1fcde32185c8cbf7b66db735 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/d/4644b6ed1fcde32185c8cbf7b66db735
new file mode 100644
index 0000000000000000000000000000000000000000..796284ff1eeddfe0660b78a6a3451d5eb42e5968
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/d/4644b6ed1fcde32185c8cbf7b66db735 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e8c3be5290acff4b5a2a4686960fba61da16b5e1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/261f2938579cbdd7de2223e98ff98946 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/261f2938579cbdd7de2223e98ff98946
new file mode 100644
index 0000000000000000000000000000000000000000..1813d1ed28193b2872bac1f24ab8f8c4b82b80dc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/261f2938579cbdd7de2223e98ff98946 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/3b4fc60f4395701e543f37e2f588bf1d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/3b4fc60f4395701e543f37e2f588bf1d
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/3b4fc60f4395701e543f37e2f588bf1d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/3cb813b08f0a44557f9b38b46109b9d4 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/3cb813b08f0a44557f9b38b46109b9d4
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/3cb813b08f0a44557f9b38b46109b9d4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/965f86928986376022220e0bcaad4995 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/965f86928986376022220e0bcaad4995
new file mode 100644
index 0000000000000000000000000000000000000000..9b713684d42a9b6eb23a77a928a67eb2e1d52206
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/965f86928986376022220e0bcaad4995
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:06b17244b5c7d8865b5023e840d0fe6b62e557a1028f07eb7cd537ec90ceb0ba
+size 227895
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/9929bd4c3b86a7a8b30329186746ee25 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/9929bd4c3b86a7a8b30329186746ee25
new file mode 100644
index 0000000000000000000000000000000000000000..fa47f9d07aa7b14da380b4fa3a0354585eb53369
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/9929bd4c3b86a7a8b30329186746ee25
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6e8aa0823bbe866de396cf439a95a21fe08a26df19c8352fbd39a8d3cbd6358e
+size 473559
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/daa5aa9bb6ca3b72e51d126b53371ff8 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/daa5aa9bb6ca3b72e51d126b53371ff8
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/daa5aa9bb6ca3b72e51d126b53371ff8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/eaf8210c8222d2591972e3f70831152e b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/eaf8210c8222d2591972e3f70831152e
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/eaf8210c8222d2591972e3f70831152e differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/fda3e0327093ba7f0ea46d91a11e6994 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/fda3e0327093ba7f0ea46d91a11e6994
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/fda3e0327093ba7f0ea46d91a11e6994 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a3ea30994e1a1ed223a2d9f06ea02dddbaceabb3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..d22128e6a95290c7b26982b508f5cf14453df3b1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754579280436752593, "commit_timestamp_nsecs": 1754579280569732661, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/d/9c25b18793e63b0baeb604cda92d05af b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/d/9c25b18793e63b0baeb604cda92d05af
new file mode 100644
index 0000000000000000000000000000000000000000..1942dab98aa1294626d9e8a8eb5c138704f56546
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/d/9c25b18793e63b0baeb604cda92d05af differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8a30bab90358605a74559d48ed9805c70eb10dc6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/126055cd3d00a897517fba8fe31ba9fb b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/126055cd3d00a897517fba8fe31ba9fb
new file mode 100644
index 0000000000000000000000000000000000000000..9fedc636ee212fda3f358cef18580198f4bcab8d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/126055cd3d00a897517fba8fe31ba9fb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7302229f4406be4d9653fe40341bdf9b05b03ed4a43685704bd0d32f159c2982
+size 365240
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/23f597b72785a1dd0e8470f9ce0d849b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/23f597b72785a1dd0e8470f9ce0d849b
new file mode 100644
index 0000000000000000000000000000000000000000..c774c46b0e3c2be9079dc62df5056e35cf8e52dd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/23f597b72785a1dd0e8470f9ce0d849b differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/4b806a00be0c9ffc63c5322a3063d5cc b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/4b806a00be0c9ffc63c5322a3063d5cc
new file mode 100644
index 0000000000000000000000000000000000000000..85f35e7991cac7b6fc244bb2f277438119f4e90b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/4b806a00be0c9ffc63c5322a3063d5cc differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/59fd1910d006e205540d2b533d167a97 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/59fd1910d006e205540d2b533d167a97
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/59fd1910d006e205540d2b533d167a97 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/9947246588ae543f8535d03c3ab3bf10 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/9947246588ae543f8535d03c3ab3bf10
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/9947246588ae543f8535d03c3ab3bf10 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/a44dee2066770bef964febe5517e97a8 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/a44dee2066770bef964febe5517e97a8
new file mode 100644
index 0000000000000000000000000000000000000000..ec745d97353f97209ef50f0eb738585830219f53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/a44dee2066770bef964febe5517e97a8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/c00d0760616c5ab4bf83c9e64d3f4034 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/c00d0760616c5ab4bf83c9e64d3f4034
new file mode 100644
index 0000000000000000000000000000000000000000..081262f90a3d3b3db83125ec0c19c6a7fd4fedfa
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/c00d0760616c5ab4bf83c9e64d3f4034
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4d88ecd513b502ca9ced034aef2e103273d1af444abac4e7fe9fe18477b83836
+size 372098
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..472b06ceef830769b5bb43951fac745bc73e9dae
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..8282e1aab053251ae9c856a7c745f574d046f087
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754579591583512518, "commit_timestamp_nsecs": 1754579591756694165, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/d/fb4daa6b4198ce882731b651d43385a2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/d/fb4daa6b4198ce882731b651d43385a2
new file mode 100644
index 0000000000000000000000000000000000000000..762dd556d601300bc6d51fbc2997f2fb7b1663e4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/d/fb4daa6b4198ce882731b651d43385a2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..af8bb5c3d972997bf15d7a23a83a8a46ebae9347
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/377d026f0aab50058ae80bedf304b1e3 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/377d026f0aab50058ae80bedf304b1e3
new file mode 100644
index 0000000000000000000000000000000000000000..62524b7a464de73845146b9ebbe198bee45ca381
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/377d026f0aab50058ae80bedf304b1e3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/3e8e8fed443c80e2f3e186a314ddb756 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/3e8e8fed443c80e2f3e186a314ddb756
new file mode 100644
index 0000000000000000000000000000000000000000..85e2d9ec5bf3b67cfbe6aa4b453863b3a57065a6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/3e8e8fed443c80e2f3e186a314ddb756
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c7909cf718c00efe854ecedc82d40f9ad6cfdf7a7d8c3a338d1c803c718b3bf5
+size 340182
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/54009da412c634fdc95794b7715c5e4d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/54009da412c634fdc95794b7715c5e4d
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/54009da412c634fdc95794b7715c5e4d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/76f64f59c2dc7f59d82bbc9718c14b58 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/76f64f59c2dc7f59d82bbc9718c14b58
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/76f64f59c2dc7f59d82bbc9718c14b58 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/855ea34f8e952f658081558230c568af b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/855ea34f8e952f658081558230c568af
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/855ea34f8e952f658081558230c568af differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/910b919247ffaa26751e9b5cf9a76e35 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/910b919247ffaa26751e9b5cf9a76e35
new file mode 100644
index 0000000000000000000000000000000000000000..18d6f12246b1fbd4091710c32efe9ffb0a2a1960
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/910b919247ffaa26751e9b5cf9a76e35
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cf36dd6ac990c08978229aef89a38b650788d6228596a17aca46d1d3eba467c1
+size 397551
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/d1ebea8667a82f8e1e200ce74d484778 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/d1ebea8667a82f8e1e200ce74d484778
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/d1ebea8667a82f8e1e200ce74d484778 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3f4ab6f59731431df36731cf418d680eba2bff28
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..81c4a1757d51dff1a4910fb3b2416a241c041279
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754579625847607672, "commit_timestamp_nsecs": 1754579626016933764, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/d/2f5d80bff8e3a996c1516946839689f7 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/d/2f5d80bff8e3a996c1516946839689f7
new file mode 100644
index 0000000000000000000000000000000000000000..ac4de206b8d789ef3134f69990cf90e07816d0ae
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/d/2f5d80bff8e3a996c1516946839689f7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5d4c2d56bbab3c3ce0a8e24881aabc087f958d09
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/088f7d6dfab3af59fcbcae83c70f7bd5 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/088f7d6dfab3af59fcbcae83c70f7bd5
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/088f7d6dfab3af59fcbcae83c70f7bd5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/5976e7a4d37254dc200f3715bfe06e19 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/5976e7a4d37254dc200f3715bfe06e19
new file mode 100644
index 0000000000000000000000000000000000000000..75ee6f4c347744778e511cef3abe0b0ad9e7affe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/5976e7a4d37254dc200f3715bfe06e19 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/8fc67794062d86206f4e5ffeacdb4663 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/8fc67794062d86206f4e5ffeacdb4663
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/8fc67794062d86206f4e5ffeacdb4663 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/b15e7a77e2df0f7001638184726667e6 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/b15e7a77e2df0f7001638184726667e6
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/b15e7a77e2df0f7001638184726667e6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/c131f8a71412437c593df286c02a89ec b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/c131f8a71412437c593df286c02a89ec
new file mode 100644
index 0000000000000000000000000000000000000000..3da99235771934d716e2fe9c22fd62f264eee61d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/c131f8a71412437c593df286c02a89ec
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8ee894f1218a189a5fabb1fc4374c92d683ae576a448a69cef950bb77b949c00
+size 261504
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/d0c8a4d169b22215a85f45dedb6f6b81 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/d0c8a4d169b22215a85f45dedb6f6b81
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/d0c8a4d169b22215a85f45dedb6f6b81 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/e43f55d994b7b1dcf8cedc94e810e352 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/e43f55d994b7b1dcf8cedc94e810e352
new file mode 100644
index 0000000000000000000000000000000000000000..dc80b49ec77f8d6f2f8e8e4d4434a0d8dca29d6c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/e43f55d994b7b1dcf8cedc94e810e352 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/e4b44ddff0c1064ffbf2596de65fc022 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/e4b44ddff0c1064ffbf2596de65fc022
new file mode 100644
index 0000000000000000000000000000000000000000..8f8d795971346c96ff06768a346d7dc23fa917f4
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/e4b44ddff0c1064ffbf2596de65fc022
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:31fd85d5253dfb2acac671f9defef2b147884809e3da8ed7c82a9b06139fd788
+size 476557
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..18e86041a1417a10a48b082e740162b87231e662
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..c63def06dd97f05dc6eb4e7af1d111435d76d5e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754579660362373662, "commit_timestamp_nsecs": 1754579660548644289, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/d/ea459fc7391e0b43100c18f17b82449f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/d/ea459fc7391e0b43100c18f17b82449f
new file mode 100644
index 0000000000000000000000000000000000000000..ec3b7105a969301b0d39392d891a21853e49262b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/d/ea459fc7391e0b43100c18f17b82449f differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1541e086252f93799a01fa2a79a88591dc666f3a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/0e21ba8c9065d2b5a97b2a64e9e7818f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/0e21ba8c9065d2b5a97b2a64e9e7818f
new file mode 100644
index 0000000000000000000000000000000000000000..dfbfe1a61305bab6f716f83c77874d9e3176999d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/0e21ba8c9065d2b5a97b2a64e9e7818f differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/22f67a8cad13b5700cd9730d11d109d2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/22f67a8cad13b5700cd9730d11d109d2
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/22f67a8cad13b5700cd9730d11d109d2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/25d613e6203026432d8604c0d3bb7fcd b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/25d613e6203026432d8604c0d3bb7fcd
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/25d613e6203026432d8604c0d3bb7fcd differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/2e34a99175f69fb81eabb32a9e3fbb28 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/2e34a99175f69fb81eabb32a9e3fbb28
new file mode 100644
index 0000000000000000000000000000000000000000..c4d2ec904233d6a6b870455d1398be099a69f319
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/2e34a99175f69fb81eabb32a9e3fbb28
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:55b85444cc2b9f97bc1f7e58c607e06ce7a50e5a6fe76aad0424cd48d219846d
+size 471411
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/615c9986c9365f8f29c7b01638d24983 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/615c9986c9365f8f29c7b01638d24983
new file mode 100644
index 0000000000000000000000000000000000000000..dd5ed2859bdbb82201955ed1cef8b12732b30a6a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/615c9986c9365f8f29c7b01638d24983
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:62d1e546f52b46d48a43f1f4a78a633882727dec340091c396dea6460d8df7a7
+size 227685
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/73820de5cf54bd27bc271c723d2f5aad b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/73820de5cf54bd27bc271c723d2f5aad
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/73820de5cf54bd27bc271c723d2f5aad differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/856d2b78d447f2e9376cc7c94a3e4638 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/856d2b78d447f2e9376cc7c94a3e4638
new file mode 100644
index 0000000000000000000000000000000000000000..97fd977217d47d479bd14cca806263fd2334796b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/856d2b78d447f2e9376cc7c94a3e4638 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/ba17fd717487c557d42a5806a17dece2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/ba17fd717487c557d42a5806a17dece2
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/ba17fd717487c557d42a5806a17dece2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/bafef53a1b17f686fcfbd2c61812b15c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/bafef53a1b17f686fcfbd2c61812b15c
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/bafef53a1b17f686fcfbd2c61812b15c differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..83365496d3cddc60277ee63014e29edf732f4feb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..9cfeb775b19fc114f5bf55b162ccf560cd6af3f4
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754579694980235298, "commit_timestamp_nsecs": 1754579695161611662, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/d/029397df6de7b3810251b1fdd4ced99a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/d/029397df6de7b3810251b1fdd4ced99a
new file mode 100644
index 0000000000000000000000000000000000000000..5c6f98347db723b3a91fa20ffa4d3c7f4357505e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/d/029397df6de7b3810251b1fdd4ced99a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..2d88d7661f0ba62c421628f6facb08031687ccff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/28734af768bf608b05c4b7610f12e283 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/28734af768bf608b05c4b7610f12e283
new file mode 100644
index 0000000000000000000000000000000000000000..519ce225a951109815e6b457409ea0bcebed1610
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/28734af768bf608b05c4b7610f12e283
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6fd519fc3f6f6ae7bf185352a09d3765ad52997c4ece65ece7e2703e71792fb7
+size 485178
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/4913843e712a8c4e42db40b0662047d0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/4913843e712a8c4e42db40b0662047d0
new file mode 100644
index 0000000000000000000000000000000000000000..d5aaec640449e69d5931ea94722320d7bc80b293
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/4913843e712a8c4e42db40b0662047d0
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:88c41dfc7999428372ef9f8ff115ff66de77856df8ba310e836958f71cf1ae19
+size 253182
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/4c2a3b925401b2de608dd4a639b42677 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/4c2a3b925401b2de608dd4a639b42677
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/4c2a3b925401b2de608dd4a639b42677 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/84bb1b717e2f00d1b593c6320641d3b7 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/84bb1b717e2f00d1b593c6320641d3b7
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/84bb1b717e2f00d1b593c6320641d3b7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/a5510b57bf1e653ec20c2cf6308e97a5 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/a5510b57bf1e653ec20c2cf6308e97a5
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/a5510b57bf1e653ec20c2cf6308e97a5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/b43150734c26fe390dcb4cb62c5483df b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/b43150734c26fe390dcb4cb62c5483df
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/b43150734c26fe390dcb4cb62c5483df differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/d9fe6f0e6122c48477185e0bc5355941 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/d9fe6f0e6122c48477185e0bc5355941
new file mode 100644
index 0000000000000000000000000000000000000000..8429ee6c5c4357929915c2a210b21dceb80ff8e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/d9fe6f0e6122c48477185e0bc5355941 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..902b53b2ab82cf2df6819e59a2e92e7326990a39
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..2e8bf128a3c4e51cf2ca057aa34217c576de599e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754579729448234503, "commit_timestamp_nsecs": 1754579729619106170, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/d/29d420e5d8c5fa545aa57c3c158f0a67 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/d/29d420e5d8c5fa545aa57c3c158f0a67
new file mode 100644
index 0000000000000000000000000000000000000000..1565fd56f29e690cb33baee4febd0c4fcb13081e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/d/29d420e5d8c5fa545aa57c3c158f0a67 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..6f1c94d80b5c0b87c830196597cd8502935a49f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/3f62d38d285bfc0a89a63a70060bf0f5 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/3f62d38d285bfc0a89a63a70060bf0f5
new file mode 100644
index 0000000000000000000000000000000000000000..a5a84af100b8cfeab5a2fb90a3e8e8db02a9f3ac
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/3f62d38d285bfc0a89a63a70060bf0f5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/47448ad82b29345f1b679fe33b1090e8 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/47448ad82b29345f1b679fe33b1090e8
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/47448ad82b29345f1b679fe33b1090e8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/4acc4afb3fe1857adfcc0dd50a534af0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/4acc4afb3fe1857adfcc0dd50a534af0
new file mode 100644
index 0000000000000000000000000000000000000000..ebd1bfe6a9e014fe23db22760dc52257f5fbd186
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/4acc4afb3fe1857adfcc0dd50a534af0
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f5c376402139898f4c826ab9e246f1e15ec1933aa349cf5840fc41ee2aa500d3
+size 253201
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/77a7ab178cbfdc05b79926be71edca4d b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/77a7ab178cbfdc05b79926be71edca4d
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/77a7ab178cbfdc05b79926be71edca4d differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/7eddf021947f8bcfe9a58e03a88345bf b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/7eddf021947f8bcfe9a58e03a88345bf
new file mode 100644
index 0000000000000000000000000000000000000000..3046b06512825b26897c679ff5d628702a32c726
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/7eddf021947f8bcfe9a58e03a88345bf
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6a6ac55d9b1a798a6670c76b5ece8a5599f2a164d6575c700e87bd408430a7d4
+size 485159
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/d80952e74129e272de2d17904503cd1a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/d80952e74129e272de2d17904503cd1a
new file mode 100644
index 0000000000000000000000000000000000000000..14058bd86ef5e373348d468579704393c568c73e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/d80952e74129e272de2d17904503cd1a differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/e6e1008f736071d3de612c14d6d99bb5 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/e6e1008f736071d3de612c14d6d99bb5
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/e6e1008f736071d3de612c14d6d99bb5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/f655013ad1d7c3dce2c986b8b076d2e2 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/f655013ad1d7c3dce2c986b8b076d2e2
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/f655013ad1d7c3dce2c986b8b076d2e2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..451c38aac38af19b545e904bf81bd55c7c1a5c54
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..11dc48d80abc83a82e7ec5baa081aa2680a3ce3e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754579764165644708, "commit_timestamp_nsecs": 1754579764337675355, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/_METADATA b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/_sharding b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/d/49a263b6e58a51222c53fc823b290b0f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/d/49a263b6e58a51222c53fc823b290b0f
new file mode 100644
index 0000000000000000000000000000000000000000..ff7f42ef1cf90daf19de9899275b8ae519997885
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/d/49a263b6e58a51222c53fc823b290b0f differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..330232775f3b8095d969a0e910b4f279716f5555
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/22b2b63ff8bf0e464649d53382f1c12b b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/22b2b63ff8bf0e464649d53382f1c12b
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/22b2b63ff8bf0e464649d53382f1c12b differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/2846cc8d40dc92f16bab4f8b8d0b2b7a b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/2846cc8d40dc92f16bab4f8b8d0b2b7a
new file mode 100644
index 0000000000000000000000000000000000000000..79fcd2ed0e2f6f0bed340677d404d87eb3fbd035
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/2846cc8d40dc92f16bab4f8b8d0b2b7a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5be82126e91f7a70482ce974a3caeba8b00f75444633b5be15a60b0636ddf223
+size 270130
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/585a148904e70abcf197179bb80d0a7c b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/585a148904e70abcf197179bb80d0a7c
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/585a148904e70abcf197179bb80d0a7c differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/992bb5f3ded52591f7ed8300299ceb40 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/992bb5f3ded52591f7ed8300299ceb40
new file mode 100644
index 0000000000000000000000000000000000000000..85f35e7991cac7b6fc244bb2f277438119f4e90b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/992bb5f3ded52591f7ed8300299ceb40 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/bd31c088042b3f40dcdf32c50d52296f b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/bd31c088042b3f40dcdf32c50d52296f
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/bd31c088042b3f40dcdf32c50d52296f differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/db664d00dea849f80075a6eac074ea01 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/db664d00dea849f80075a6eac074ea01
new file mode 100644
index 0000000000000000000000000000000000000000..e81f5cb3efb79679def761e7bf79edca48fe14fc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/db664d00dea849f80075a6eac074ea01 differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/fc0ebd7aee925c3353be40990679c917 b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/fc0ebd7aee925c3353be40990679c917
new file mode 100644
index 0000000000000000000000000000000000000000..8df2e93efe00b5c52066d24cd497d860f4c6d1fd
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/fc0ebd7aee925c3353be40990679c917
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0ad0ba75182747e216c920f2b14c913281b2a7a038fc8576f2ed929341e46f02
+size 468275
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5f273edfe41c774491debe9e533425d032e70123
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/config.yaml b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/config.yaml
new file mode 100644
index 0000000000000000000000000000000000000000..daec424d716136c6214f4cdbffa72e5f0ce85f49
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/config.yaml
@@ -0,0 +1,1130 @@
+_wandb:
+    value:
+        cli_version: 0.21.0
+        e:
+            x1qaiddm18zjlareqkr9lea6iitxnpr7:
+                args:
+                    - train
+                    - corner
+                    - --wall-prob
+                    - "0.0"
+                    - --env-size
+                    - "13"
+                    - --log-gifs
+                    - --env-layout
+                    - tree
+                    - --wandb-project
+                    - jaxgmg-wall-prob-tree
+                    - --wandb-name
+                    - wp:0.0_alpha:1
+                    - --num-total-env-steps
+                    - "200_000_000"
+                    - --prob-shift
+                    - "1"
+                    - --keep-all-checkpoints
+                    - --num-cycles-per-checkpoint
+                    - "64"
+                cpu_count: 24
+                cpu_count_logical: 48
+                cudaVersion: "12.8"
+                disk:
+                    /:
+                        total: "102005473280"
+                        used: "43119042560"
+                email: david@timaeus.co
+                executable: /workspace/timaeus/.venv/bin/python
+                git:
+                    commit: 7415666d6a2e5d63453ea9fadaf6cecf63256a59
+                    remote: git@github.com:timaeus-research/timaeus.git
+                gpu: NVIDIA GeForce RTX 3090
+                gpu_count: 1
+                gpu_nvidia:
+                    - architecture: Ampere
+                      cudaCores: 10496
+                      memoryTotal: "25769803776"
+                      name: NVIDIA GeForce RTX 3090
+                      uuid: GPU-43bd7524-bf94-73a9-1460-2b36dd09a208
+                host: 6e66e15ca303
+                memory:
+                    total: "135066595328"
+                os: Linux-5.4.0-216-generic-x86_64-with-glibc2.39
+                program: -m jaxgmg.__main__
+                python: CPython 3.11.11
+                root: /workspace/timaeus/projects/rl
+                startedAt: "2025-08-07T15:04:00.936848Z"
+                writerId: x1qaiddm18zjlareqkr9lea6iitxnpr7
+        m:
+            - "1": train-all/proxy_corner/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": step/env-step-all-after
+              "6":
+                - 3
+              "7": []
+            - "1": eval-batch-tree/avg_benchmark_regret
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/proxy_corner/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/shift_proportion
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/cheese-corne_dist_wavg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/solvable_num
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_advantage
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": step/ppo-update-after
+              "6":
+                - 3
+              "7": []
+            - "1": eval-batch-tree/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/lvl_benchmark_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/proxy_corner/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/proxy_corner/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/proxy_corner/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/eta
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/prop_walls_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_actor_approxkl3
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_grad_norm_pre_clip
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_loss
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/avg_benchmark_regret
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/lvl_benchmark_regret_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/shift_proportion
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/cheese-corner_dist_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/mouse-cheese_dist_finite_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/mouse-cheese_dist_solvable_wavg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/num_walls_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/visit_patterns/last_visit_time_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/proxy_corner/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/proxy_corner/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/num_walls_avg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_entropy
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/avg_benchmark_regret
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/proxy_corner/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/proxy_corner/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/rollouts_gif
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_advantage
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/avg_benchmark_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/rollouts_gif
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/proxy_corner/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_actor_loss
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/avg_avg_episode_length
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/avg_avg_episode_length
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/rollouts_gif
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_actor_loss
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/lvl_benchmark_regret_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/proxy_corner/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/prop_walls_avg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/scoring/avg_scores
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_critic_clipfrac
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_critic_loss
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_entropy
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/mouse-cheese_dist_solvable_avg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/prop_walls_wavg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/scoring/scores_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/visit_patterns/num_replay_batches
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/rollouts_gif
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/avg_avg_episode_length
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_advantage
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/avg_benchmark_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/visit_patterns/prev_batch_level_ids_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/lvl_benchmark_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/levels16_img
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/levels_cheese_is_in_corner_avg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_loss
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_actor_approxkl3
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/avg_avg_episode_length
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/lvl_avg_episode_length_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/proxy_corner/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/proxy_corner/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/proxy_corner/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/proxy_corner/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_actor_approxkl1
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/proxy_corner/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/lvl_avg_episode_length_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/lvl_benchmark_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/proxy_corner/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/lvl_avg_episode_length_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/solvable_wavg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_critic_loss
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/proxy_corner/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/proxy_corner/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_actor_approxkl3
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_actor_clipfrac
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/avg_benchmark_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/visit_patterns/avg_last_visit_time
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_actor_approxkl1
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_actor_loss
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_entropy
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/proxy_corner/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/avg_avg_episode_length
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_grad_norm_pre_clip
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/rollouts_gif
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/proxy_corner/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/mouse-cheese_dist_finite_wavg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/lvl_avg_episode_length_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/lvl_avg_episode_length_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/num_walls_wavg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/visit_patterns/avg_first_visit_time
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/proxy_corner/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_actor_clipfrac
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/lvl_benchmark_regret_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/lvl_benchmark_regret_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/avg_avg_episode_length
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/proxy_corner/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/cheese-corne_dist_avg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_critic_clipfrac
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/lvl_avg_episode_length_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/avg_avg_episode_length
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/proxy_corner/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/shift_proportion
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/proxy_corner/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/rollouts_gif
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_loss
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_actor_clipfrac
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_critic_loss
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/proxy_corner/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/rollouts_gif
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/avg_benchmark_regret
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/proxy_corner/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_actor_approxkl1
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_grad_norm_pre_clip
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/lvl_avg_episode_length_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/avg_benchmark_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/proxy_corner/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/mouse-cheese_dist_finite_avg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/solvable_avg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/visit_patterns/first_visit_time_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_critic_clipfrac
+              "5": 11
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/lvl_benchmark_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+        python_version: 3.11.11
+        t:
+            "1":
+                - 12
+                - 45
+            "2":
+                - 12
+                - 45
+            "3":
+                - 2
+                - 3
+                - 7
+                - 13
+                - 14
+                - 16
+                - 61
+            "4": 3.11.11
+            "5": 0.21.0
+            "12": 0.21.0
+            "13": linux-x86_64
+chain_mutate:
+    value: true
+checkpointing:
+    value: true
+clipping:
+    value: false
+console_log:
+    value: true
+env_corner_size:
+    value: 1
+env_layout:
+    value: tree
+env_penalize_time:
+    value: false
+env_size:
+    value: 13
+env_terminate_after_corner:
+    value: false
+evals_num_env_steps:
+    value: 512
+evals_num_levels:
+    value: 256
+gif_grid_width:
+    value: 16
+img_level_of_detail:
+    value: 1
+keep_all_checkpoints:
+    value: true
+level_splayer:
+    value: mouse
+log_gifs:
+    value: true
+log_hists:
+    value: false
+log_imgs:
+    value: true
+max_num_checkpoints:
+    value: 1
+mutate_cheese:
+    value: true
+net_cnn_type:
+    value: large
+net_rnn_type:
+    value: ff
+net_width:
+    value: 256
+num_cycles_per_big_eval:
+    value: 1024
+num_cycles_per_checkpoint:
+    value: 64
+num_cycles_per_eval:
+    value: 32
+num_cycles_per_gifs:
+    value: 1024
+num_cycles_per_log:
+    value: 32
+num_env_steps_per_cycle:
+    value: 128
+num_epochs_per_cycle:
+    value: 5
+num_minibatches_per_epoch:
+    value: 4
+num_mutate_steps:
+    value: 12
+num_parallel_envs:
+    value: 256
+num_total_env_steps:
+    value: 200000000
+num_train_levels:
+    value: 2048
+obs_level_of_detail:
+    value: 0
+plr_buffer_size:
+    value: 4096
+plr_prob_replay:
+    value: 0.5
+plr_proxy_shaping:
+    value: false
+plr_proxy_shaping_coeff:
+    value: 0.5
+plr_regret_estimator:
+    value: maxmc-actor
+plr_robust:
+    value: false
+plr_staleness_coeff:
+    value: 0.1
+plr_temperature:
+    value: 0.1
+ppo_clip_eps:
+    value: 0.1
+ppo_critic_coeff:
+    value: 0.5
+ppo_entropy_coeff:
+    value: 0.001
+ppo_gae_lambda:
+    value: 0.95
+ppo_gamma:
+    value: 0.999
+ppo_lr:
+    value: 5e-05
+ppo_lr_annealing:
+    value: false
+ppo_max_grad_norm:
+    value: 0.5
+ppo_proxy_critic_coeff:
+    value: 0.5
+prob_mutate_shift:
+    value: 0
+prob_shift:
+    value: 1
+proxy_name:
+    value: proxy_corner
+seed:
+    value: 42
+train_proxy_critic:
+    value: false
+ued:
+    value: plr
+use_fixed_eval_levels:
+    value: false
+wall_prob:
+    value: 0
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_0_245ea0be1162eb7bd894.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_0_245ea0be1162eb7bd894.png
new file mode 100644
index 0000000000000000000000000000000000000000..f5cf122f82821d99b976d3abafc71cc0b2c92162
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_0_245ea0be1162eb7bd894.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_0_3eb4d166767168218e89.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_0_3eb4d166767168218e89.png
new file mode 100644
index 0000000000000000000000000000000000000000..a3eb2021d2c4ebadacda2768ad83ae0a27d18337
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_0_3eb4d166767168218e89.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1024_1374590aed3def983454.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1024_1374590aed3def983454.png
new file mode 100644
index 0000000000000000000000000000000000000000..c7a7b7d3e3df2e96423872a73913fa84b4faecd2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1024_1374590aed3def983454.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1056_442967fb0c186066c763.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1056_442967fb0c186066c763.png
new file mode 100644
index 0000000000000000000000000000000000000000..47deca1a3a7159339f1e107e4a7e1b1b37328e0b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1056_442967fb0c186066c763.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1088_571cf1a869d9386e6573.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1088_571cf1a869d9386e6573.png
new file mode 100644
index 0000000000000000000000000000000000000000..4da225ab754ac959bb2de7adac8001b8159ffd25
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1088_571cf1a869d9386e6573.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1120_a46705d5f58ad4628473.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1120_a46705d5f58ad4628473.png
new file mode 100644
index 0000000000000000000000000000000000000000..b186b7ecba30ed2791ea94890b1486a4603f8dc6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1120_a46705d5f58ad4628473.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1152_52fe64face4d9408a6eb.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1152_52fe64face4d9408a6eb.png
new file mode 100644
index 0000000000000000000000000000000000000000..caf38dd5892140c3ad6da6845426f2ed6ebcc3d8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1152_52fe64face4d9408a6eb.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1184_64d0006faf7f47aeafe8.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1184_64d0006faf7f47aeafe8.png
new file mode 100644
index 0000000000000000000000000000000000000000..080e869c57b42530eea310dac468b585c213b6bc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1184_64d0006faf7f47aeafe8.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1216_ae6920c754a5fde7c2d4.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1216_ae6920c754a5fde7c2d4.png
new file mode 100644
index 0000000000000000000000000000000000000000..733a80fe968ecfc22722873ec606f35a265ed0c5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1216_ae6920c754a5fde7c2d4.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1248_782dd31e1f85721c0a1e.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1248_782dd31e1f85721c0a1e.png
new file mode 100644
index 0000000000000000000000000000000000000000..7d92e53a7e7dd69a92079d948f2900145ab33230
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1248_782dd31e1f85721c0a1e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1280_0c83f6dccb507b926081.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1280_0c83f6dccb507b926081.png
new file mode 100644
index 0000000000000000000000000000000000000000..b61fb9671ac869e8f5a1ed2dc408fb646292aa19
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1280_0c83f6dccb507b926081.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_128_2c67140aa2d6c9f36fde.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_128_2c67140aa2d6c9f36fde.png
new file mode 100644
index 0000000000000000000000000000000000000000..d74235e2ea9f482b70ef3709de2ac82f572f69cd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_128_2c67140aa2d6c9f36fde.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1312_d6f8260f4d0c957f33eb.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1312_d6f8260f4d0c957f33eb.png
new file mode 100644
index 0000000000000000000000000000000000000000..7366f7369c1e0842756d0ee83954b7415af07a3f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1312_d6f8260f4d0c957f33eb.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1344_fa97f29c45fa5900e4fe.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1344_fa97f29c45fa5900e4fe.png
new file mode 100644
index 0000000000000000000000000000000000000000..40c4a4fb53ab83f60ffde752368a3b02d0b7c50f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1344_fa97f29c45fa5900e4fe.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1376_f13f0542228dc8a00c95.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1376_f13f0542228dc8a00c95.png
new file mode 100644
index 0000000000000000000000000000000000000000..4a3b981e1930db1445e10d6d3e6bbca35ae5e8ec
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1376_f13f0542228dc8a00c95.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1408_f2f9723d6dea6cadae45.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1408_f2f9723d6dea6cadae45.png
new file mode 100644
index 0000000000000000000000000000000000000000..8f244809a1e9324807f58d60487d57dae3b43c32
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1408_f2f9723d6dea6cadae45.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1440_0a81d55e13cf183650c4.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1440_0a81d55e13cf183650c4.png
new file mode 100644
index 0000000000000000000000000000000000000000..ae63772bad4d0257a3bc5d5083660e1a9c82ac9b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1440_0a81d55e13cf183650c4.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1472_5100ad658b877ac8e213.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1472_5100ad658b877ac8e213.png
new file mode 100644
index 0000000000000000000000000000000000000000..a210fd883fd2a0d95a41c0cfbd6691075535310d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1472_5100ad658b877ac8e213.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1504_55a4b55d6d658014d31d.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1504_55a4b55d6d658014d31d.png
new file mode 100644
index 0000000000000000000000000000000000000000..b8427e13ab70a46eb7a71ca11dacbc19b3934133
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1504_55a4b55d6d658014d31d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1536_17540189535549e1246c.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1536_17540189535549e1246c.png
new file mode 100644
index 0000000000000000000000000000000000000000..f252e50eac1a67e44e6bb1be4dc4580e510628cc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1536_17540189535549e1246c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1568_cab60acc856a23bc97c7.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1568_cab60acc856a23bc97c7.png
new file mode 100644
index 0000000000000000000000000000000000000000..3c5f9f3bde38d26f9a6cbf8ad0d773c2642690de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1568_cab60acc856a23bc97c7.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1600_5de0df0cde7ccfb7f42a.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1600_5de0df0cde7ccfb7f42a.png
new file mode 100644
index 0000000000000000000000000000000000000000..f15670fcd310318e5d3a881c61c544f5535ab970
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1600_5de0df0cde7ccfb7f42a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_160_f872d8ea428ce60ce106.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_160_f872d8ea428ce60ce106.png
new file mode 100644
index 0000000000000000000000000000000000000000..435ab3e1ee90ddb9c58a72e608fdaee5bc2fcbee
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_160_f872d8ea428ce60ce106.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1632_c8316aa625895a55905a.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1632_c8316aa625895a55905a.png
new file mode 100644
index 0000000000000000000000000000000000000000..b3d7bb3c3dc2142730e086bf0b21503f76278d33
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1632_c8316aa625895a55905a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1664_547523c22cad9132fd23.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1664_547523c22cad9132fd23.png
new file mode 100644
index 0000000000000000000000000000000000000000..705f966c63837102f46e90bbd5e4ae8d00cb68bd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1664_547523c22cad9132fd23.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1696_3c1a1f59df06895a1af8.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1696_3c1a1f59df06895a1af8.png
new file mode 100644
index 0000000000000000000000000000000000000000..faedd45b394c337d639ce6d6bcc5d944a3ccba40
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1696_3c1a1f59df06895a1af8.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1728_a1d3bb6f3cc0fb71fd2b.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1728_a1d3bb6f3cc0fb71fd2b.png
new file mode 100644
index 0000000000000000000000000000000000000000..02e80f031120d08eca510a1b7fa635b34f0b5d02
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1728_a1d3bb6f3cc0fb71fd2b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1760_353e51a386bcfa7d6c3f.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1760_353e51a386bcfa7d6c3f.png
new file mode 100644
index 0000000000000000000000000000000000000000..219f4c991cf1c4235e0d2e29d99619fe704f3405
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1760_353e51a386bcfa7d6c3f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1792_eab6f4fd798395113a2b.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1792_eab6f4fd798395113a2b.png
new file mode 100644
index 0000000000000000000000000000000000000000..4a022de884cff67f79d071b1d4d68c15e7134109
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1792_eab6f4fd798395113a2b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1824_1510d12d582e3f3d6a86.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1824_1510d12d582e3f3d6a86.png
new file mode 100644
index 0000000000000000000000000000000000000000..8246147b70d8c08ae7ad02add7940f7a8f2a8d67
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1824_1510d12d582e3f3d6a86.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1856_c3188b86bf8bae6a9c9b.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1856_c3188b86bf8bae6a9c9b.png
new file mode 100644
index 0000000000000000000000000000000000000000..2c5d7cb3fb9896ca1e1dba5569be888d03ae12b1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1856_c3188b86bf8bae6a9c9b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1888_27b3336a42a5baf6524e.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1888_27b3336a42a5baf6524e.png
new file mode 100644
index 0000000000000000000000000000000000000000..54f59f1cdd8619a4ffc77d4bee3908bcf1219252
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1888_27b3336a42a5baf6524e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1920_f7d38aa2f80c0987b43f.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1920_f7d38aa2f80c0987b43f.png
new file mode 100644
index 0000000000000000000000000000000000000000..5265736fafdf0e342ff535f3f8d666f94a2a7b7d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1920_f7d38aa2f80c0987b43f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_192_87421e9b4674e051761d.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_192_87421e9b4674e051761d.png
new file mode 100644
index 0000000000000000000000000000000000000000..fdf910a4f237ee08cf8ce067036dce5c022800eb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_192_87421e9b4674e051761d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1952_a6f8957516eaf7d5ddb0.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1952_a6f8957516eaf7d5ddb0.png
new file mode 100644
index 0000000000000000000000000000000000000000..85efbc74665c3b664d8dce014b96e58446424984
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1952_a6f8957516eaf7d5ddb0.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1984_7c481796cf2ccbdec5b9.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1984_7c481796cf2ccbdec5b9.png
new file mode 100644
index 0000000000000000000000000000000000000000..8261fa6d7f714e0f10e1df9ba249dec6760f3887
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_1984_7c481796cf2ccbdec5b9.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2016_e41b9a85fab7d9d29182.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2016_e41b9a85fab7d9d29182.png
new file mode 100644
index 0000000000000000000000000000000000000000..85f07430f129f275eff2538b05f13acd39d1feb1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2016_e41b9a85fab7d9d29182.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2048_ee239042d4de02b38d01.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2048_ee239042d4de02b38d01.png
new file mode 100644
index 0000000000000000000000000000000000000000..eabcc554618c11eb13af8d712e38a50146cb0666
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2048_ee239042d4de02b38d01.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2080_419cf40b0804e5c21620.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2080_419cf40b0804e5c21620.png
new file mode 100644
index 0000000000000000000000000000000000000000..cf2f54237db3d37f6f3190d88afc5a50b155969b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2080_419cf40b0804e5c21620.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2112_dadf8ed3c65f5d1dddd2.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2112_dadf8ed3c65f5d1dddd2.png
new file mode 100644
index 0000000000000000000000000000000000000000..734ce9679bd0b1b627bf1563335b7be0c92e344a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2112_dadf8ed3c65f5d1dddd2.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2144_6ae952f8a15cf5a72590.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2144_6ae952f8a15cf5a72590.png
new file mode 100644
index 0000000000000000000000000000000000000000..05bd98661d893a0ea4849cfe44cf1bf0be8b2d59
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2144_6ae952f8a15cf5a72590.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2176_67effaffa12dfe293e39.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2176_67effaffa12dfe293e39.png
new file mode 100644
index 0000000000000000000000000000000000000000..73dc0ad704002c2381b2ae8fc173621ac668c658
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2176_67effaffa12dfe293e39.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2208_dc48a49482a490a02b80.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2208_dc48a49482a490a02b80.png
new file mode 100644
index 0000000000000000000000000000000000000000..886760957117976c78c83a510ca49d585a246e71
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2208_dc48a49482a490a02b80.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2240_6376fa873abf3c1fcd01.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2240_6376fa873abf3c1fcd01.png
new file mode 100644
index 0000000000000000000000000000000000000000..3bdbce0b2d6b384796390f0fdfaf2f2fcd7a260a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2240_6376fa873abf3c1fcd01.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_224_0cb141bb60054edcca6e.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_224_0cb141bb60054edcca6e.png
new file mode 100644
index 0000000000000000000000000000000000000000..18aa8579f04318ee1508b86d1f1f3bf67e4dae88
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_224_0cb141bb60054edcca6e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2272_3d4389b3dd0dc7e85011.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2272_3d4389b3dd0dc7e85011.png
new file mode 100644
index 0000000000000000000000000000000000000000..588a926ac2620c60ab033a543fa3daba571997f0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2272_3d4389b3dd0dc7e85011.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2304_9431dd5e09d772caf8d5.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2304_9431dd5e09d772caf8d5.png
new file mode 100644
index 0000000000000000000000000000000000000000..58ddeb61f51283f35f459fb396e10a882d7d7926
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2304_9431dd5e09d772caf8d5.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2336_1b4a10a9b9eaafcb4db3.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2336_1b4a10a9b9eaafcb4db3.png
new file mode 100644
index 0000000000000000000000000000000000000000..fbfd354a13bae600c8c273c6121f28492dac21ae
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2336_1b4a10a9b9eaafcb4db3.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2368_ced1d4719b580b949691.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2368_ced1d4719b580b949691.png
new file mode 100644
index 0000000000000000000000000000000000000000..8b0da57b2ad8bfd9c3a638e1a07ab95fdbc617e9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2368_ced1d4719b580b949691.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2400_e80e315d5a5e45e0bd23.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2400_e80e315d5a5e45e0bd23.png
new file mode 100644
index 0000000000000000000000000000000000000000..f9dc16e5b4f470a6e3d5c3d07e3e3ef66837de09
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2400_e80e315d5a5e45e0bd23.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2432_c765f0d8ebc9fd75d472.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2432_c765f0d8ebc9fd75d472.png
new file mode 100644
index 0000000000000000000000000000000000000000..66a073b1f71a39d8e9280f8b13d3648d08e3c368
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2432_c765f0d8ebc9fd75d472.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2464_4b5463277148d4a55f85.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2464_4b5463277148d4a55f85.png
new file mode 100644
index 0000000000000000000000000000000000000000..22bbc3b33952f408303625ff97428f970a7dda3f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2464_4b5463277148d4a55f85.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2496_61ab328458af55348c02.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2496_61ab328458af55348c02.png
new file mode 100644
index 0000000000000000000000000000000000000000..a933284b9a237f26a85fc326911bfef4902b8c8f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2496_61ab328458af55348c02.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2528_1ef7b2cd6d4d650c2290.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2528_1ef7b2cd6d4d650c2290.png
new file mode 100644
index 0000000000000000000000000000000000000000..993dff54dd0ae9ed4af7ccfd7975df12e6364999
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2528_1ef7b2cd6d4d650c2290.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2560_3583f1f6e5c65d69f18e.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2560_3583f1f6e5c65d69f18e.png
new file mode 100644
index 0000000000000000000000000000000000000000..0d5239ceddafd739ff2b3a6fd473ae61fa212205
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2560_3583f1f6e5c65d69f18e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_256_0759627a396d07ae3547.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_256_0759627a396d07ae3547.png
new file mode 100644
index 0000000000000000000000000000000000000000..6fad8f30f56f6ac37836d7760a4cc57a5c99363d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_256_0759627a396d07ae3547.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2592_b807f5291c6727300ade.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2592_b807f5291c6727300ade.png
new file mode 100644
index 0000000000000000000000000000000000000000..66e34edfff42f1c10807411dccfff318ad78a585
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2592_b807f5291c6727300ade.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2624_505e7f73a43d9b512903.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2624_505e7f73a43d9b512903.png
new file mode 100644
index 0000000000000000000000000000000000000000..19ddac3fca777777a4069849fd35c7df001653e9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2624_505e7f73a43d9b512903.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2656_4a8b7f258b18a39d5c42.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2656_4a8b7f258b18a39d5c42.png
new file mode 100644
index 0000000000000000000000000000000000000000..808505c7a249827885e6fca150be8514e3684b54
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2656_4a8b7f258b18a39d5c42.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2688_27e5289961801fabda3b.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2688_27e5289961801fabda3b.png
new file mode 100644
index 0000000000000000000000000000000000000000..b3efd74c5e0775c628bbc4ce4d3452a5585d5492
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2688_27e5289961801fabda3b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2720_72521bae6ff2f1abd348.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2720_72521bae6ff2f1abd348.png
new file mode 100644
index 0000000000000000000000000000000000000000..af0d6f8666cc45413e7faa2505a5f61db0c38994
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2720_72521bae6ff2f1abd348.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2752_02b1d9aff955b0a32ca5.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2752_02b1d9aff955b0a32ca5.png
new file mode 100644
index 0000000000000000000000000000000000000000..f02a126c70bc03622a131a4d1e93739e48e39e9a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2752_02b1d9aff955b0a32ca5.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2784_24e7c81b345d7f33d104.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2784_24e7c81b345d7f33d104.png
new file mode 100644
index 0000000000000000000000000000000000000000..8f6abb2c733bbe17b1fadee3235416dca88d12e0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2784_24e7c81b345d7f33d104.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2816_f77d960166a47dd55e93.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2816_f77d960166a47dd55e93.png
new file mode 100644
index 0000000000000000000000000000000000000000..d82424a52d578e91899edee1a978187c754887c8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2816_f77d960166a47dd55e93.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2848_0efeea61fc4f28e8da19.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2848_0efeea61fc4f28e8da19.png
new file mode 100644
index 0000000000000000000000000000000000000000..bc7263e63dafc6a89bdff545adf3daa6b124fcba
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2848_0efeea61fc4f28e8da19.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2880_7c7a9874d9c3763bc42f.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2880_7c7a9874d9c3763bc42f.png
new file mode 100644
index 0000000000000000000000000000000000000000..27ee6666722e110c93bcb94fbc9a74437a362c35
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2880_7c7a9874d9c3763bc42f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_288_713de999a72ac6ef2f9e.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_288_713de999a72ac6ef2f9e.png
new file mode 100644
index 0000000000000000000000000000000000000000..93f8fec8803bc255e807d963a10e3aceb117c255
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_288_713de999a72ac6ef2f9e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2912_9382486c82ffd219fb30.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2912_9382486c82ffd219fb30.png
new file mode 100644
index 0000000000000000000000000000000000000000..f743b2e45ed687cb1a421ba0ae25ff99c04db1db
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2912_9382486c82ffd219fb30.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2944_01299fd62d01d152b42c.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2944_01299fd62d01d152b42c.png
new file mode 100644
index 0000000000000000000000000000000000000000..b8e70ea7f53c552fc01ac8c0819a309de789cf5c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2944_01299fd62d01d152b42c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2976_f6d479b594c8f00adf10.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2976_f6d479b594c8f00adf10.png
new file mode 100644
index 0000000000000000000000000000000000000000..f72e74a9f8d461d39823d0ca4a37d2eb8b9c07a8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_2976_f6d479b594c8f00adf10.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3008_c6aaf386e296c0f08697.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3008_c6aaf386e296c0f08697.png
new file mode 100644
index 0000000000000000000000000000000000000000..4ba3389bdf44ae0b7e3d7bd006cd04fcaad5e190
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3008_c6aaf386e296c0f08697.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3040_ee2538a1cfe344887405.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3040_ee2538a1cfe344887405.png
new file mode 100644
index 0000000000000000000000000000000000000000..8cba7646635c73eaf917a8479aab582c2f195e01
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3040_ee2538a1cfe344887405.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3072_7d219c5b1ee15d0cae75.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3072_7d219c5b1ee15d0cae75.png
new file mode 100644
index 0000000000000000000000000000000000000000..4c6550a8958508b2aa3652ad9ff4b789375a9d36
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3072_7d219c5b1ee15d0cae75.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3104_b7be01863c5c5fea9c59.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3104_b7be01863c5c5fea9c59.png
new file mode 100644
index 0000000000000000000000000000000000000000..e1f86917f85a3ee92852b17bac154e197a3b384a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3104_b7be01863c5c5fea9c59.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3136_46fdcab63e1d9a6f993b.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3136_46fdcab63e1d9a6f993b.png
new file mode 100644
index 0000000000000000000000000000000000000000..bcf4168952277955f44c2e0bc4f55585e77c6c0b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3136_46fdcab63e1d9a6f993b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3168_3a4aca64c32560a480be.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3168_3a4aca64c32560a480be.png
new file mode 100644
index 0000000000000000000000000000000000000000..269b4839cd4caba37384f3f1746038c52c8cb488
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3168_3a4aca64c32560a480be.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3200_7d8defdf7f953346edc7.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3200_7d8defdf7f953346edc7.png
new file mode 100644
index 0000000000000000000000000000000000000000..16ae1c34381464ca17b7acda2b914da1cdb62f51
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3200_7d8defdf7f953346edc7.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_320_45a13ae971cd33b035f6.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_320_45a13ae971cd33b035f6.png
new file mode 100644
index 0000000000000000000000000000000000000000..e721e0fe7f9268f1185249e5ee9dfe830baf223a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_320_45a13ae971cd33b035f6.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3232_88bc1f212b2e6c7b04e1.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3232_88bc1f212b2e6c7b04e1.png
new file mode 100644
index 0000000000000000000000000000000000000000..3dd88b7fecc3857f0fd1a52028b44e66ad4d8a49
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3232_88bc1f212b2e6c7b04e1.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3264_cf4e3d815c1edcb3508e.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3264_cf4e3d815c1edcb3508e.png
new file mode 100644
index 0000000000000000000000000000000000000000..b9ae54368c8231994f720bd4be8a2595a3e0195f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3264_cf4e3d815c1edcb3508e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3296_cfb0aaa90d8cd76c2be3.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3296_cfb0aaa90d8cd76c2be3.png
new file mode 100644
index 0000000000000000000000000000000000000000..3009b07d07e43eb877763700a29a53fc350b255c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3296_cfb0aaa90d8cd76c2be3.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_32_da4858d2caf755bf1fe1.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_32_da4858d2caf755bf1fe1.png
new file mode 100644
index 0000000000000000000000000000000000000000..c16c8daeb0b4f0d2370f06b5cc47ce98ecb88d3a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_32_da4858d2caf755bf1fe1.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3328_d7718c9ebe69da05d100.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3328_d7718c9ebe69da05d100.png
new file mode 100644
index 0000000000000000000000000000000000000000..f08794fa1c665e11a388db80bed3dca5e7f34ed1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3328_d7718c9ebe69da05d100.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3360_e0d67984e3098b5a94a5.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3360_e0d67984e3098b5a94a5.png
new file mode 100644
index 0000000000000000000000000000000000000000..f4e110e6d9ffd30482a68339ab8812d47b9cf53f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3360_e0d67984e3098b5a94a5.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3392_995283d4562d8a4340fb.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3392_995283d4562d8a4340fb.png
new file mode 100644
index 0000000000000000000000000000000000000000..ba78d542f3c1b74ee8c7a7caf689fd84a66a824e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3392_995283d4562d8a4340fb.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3424_2059823038307e567f08.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3424_2059823038307e567f08.png
new file mode 100644
index 0000000000000000000000000000000000000000..da29d9ea9aa9a3d31c5f8374745145bf0c62d30b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3424_2059823038307e567f08.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3456_31cc23a7550a9e19418a.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3456_31cc23a7550a9e19418a.png
new file mode 100644
index 0000000000000000000000000000000000000000..c8dbfda957c79a49c95703e1679b28aa743fd069
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3456_31cc23a7550a9e19418a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3488_4295364d55e1e4167f4f.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3488_4295364d55e1e4167f4f.png
new file mode 100644
index 0000000000000000000000000000000000000000..68b6b599da4461d187c154fae16bf69cab480a49
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3488_4295364d55e1e4167f4f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3520_164a0f2ca7bac0f9a857.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3520_164a0f2ca7bac0f9a857.png
new file mode 100644
index 0000000000000000000000000000000000000000..c7f67d200ca693b0d36f1dace85dcf5255abbadb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3520_164a0f2ca7bac0f9a857.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_352_09d5bd05f458dbcb0285.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_352_09d5bd05f458dbcb0285.png
new file mode 100644
index 0000000000000000000000000000000000000000..1bbc25e0ce697a581f5fd1345c23ed9fe4b6065d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_352_09d5bd05f458dbcb0285.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3552_997734507a1c532190b4.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3552_997734507a1c532190b4.png
new file mode 100644
index 0000000000000000000000000000000000000000..0d0ced6b3adc771ac3281060d697b991633362c5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3552_997734507a1c532190b4.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3584_c58d5b49d95aa8e1cfcf.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3584_c58d5b49d95aa8e1cfcf.png
new file mode 100644
index 0000000000000000000000000000000000000000..96b31a7bbddf2a15d775307a92f007448ae024c7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3584_c58d5b49d95aa8e1cfcf.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3616_635e9fbe4b446bf86951.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3616_635e9fbe4b446bf86951.png
new file mode 100644
index 0000000000000000000000000000000000000000..54d601f11fe2d0626aa83843abc60e951f22544e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3616_635e9fbe4b446bf86951.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3648_c3e8c7bb22b46f66e583.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3648_c3e8c7bb22b46f66e583.png
new file mode 100644
index 0000000000000000000000000000000000000000..d05a89a7009135d670e3fdd907de3d4215117b40
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3648_c3e8c7bb22b46f66e583.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3680_6b1ac02a479332fa4335.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3680_6b1ac02a479332fa4335.png
new file mode 100644
index 0000000000000000000000000000000000000000..abd010e64334788b5975c11549b3a241472e13ec
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3680_6b1ac02a479332fa4335.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3712_7f2f1bb6e71086ee0d88.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3712_7f2f1bb6e71086ee0d88.png
new file mode 100644
index 0000000000000000000000000000000000000000..147ec686cb2b8373bf9211a26b1463a9241ba5e9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3712_7f2f1bb6e71086ee0d88.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3744_d33c91822d06fbe72d1d.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3744_d33c91822d06fbe72d1d.png
new file mode 100644
index 0000000000000000000000000000000000000000..fab52e58e940d52f2238787bf259f8624a37d07b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3744_d33c91822d06fbe72d1d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3776_992f8f74a74ca64793f2.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3776_992f8f74a74ca64793f2.png
new file mode 100644
index 0000000000000000000000000000000000000000..7f52a103e812896052b21d22877c331b2099f14d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3776_992f8f74a74ca64793f2.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3808_807e03f76c7c79c23db2.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3808_807e03f76c7c79c23db2.png
new file mode 100644
index 0000000000000000000000000000000000000000..cbfa534ec77059fe1c5f4d197ed1ccdfb8aa7cbe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3808_807e03f76c7c79c23db2.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3840_9672883e4d2051059e87.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3840_9672883e4d2051059e87.png
new file mode 100644
index 0000000000000000000000000000000000000000..5bf324a331aad3e928c477cb177052a828889036
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3840_9672883e4d2051059e87.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_384_9cd25391cd58b080407b.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_384_9cd25391cd58b080407b.png
new file mode 100644
index 0000000000000000000000000000000000000000..af6d6fe76e0fb5514d7ead34cc5e32686236a670
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_384_9cd25391cd58b080407b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3872_1ae030c5b42b8c754408.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3872_1ae030c5b42b8c754408.png
new file mode 100644
index 0000000000000000000000000000000000000000..f42c7643072b71ec3107e7ce5845d56d464d3491
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3872_1ae030c5b42b8c754408.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3904_8c94264be9a5f4e93c05.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3904_8c94264be9a5f4e93c05.png
new file mode 100644
index 0000000000000000000000000000000000000000..8d60828642aac1a66f5c55277bf4caaccd9a01d5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3904_8c94264be9a5f4e93c05.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3936_7e6dc48938cceb01eec5.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3936_7e6dc48938cceb01eec5.png
new file mode 100644
index 0000000000000000000000000000000000000000..f3a080d6544d5c6d05d88305fe3df9b3681762b0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3936_7e6dc48938cceb01eec5.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3968_7bd196be8be9fff9d547.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3968_7bd196be8be9fff9d547.png
new file mode 100644
index 0000000000000000000000000000000000000000..5eb70a04c5dc22a0f68b669fef99c56e58ad7cf4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_3968_7bd196be8be9fff9d547.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4000_32e26a3c53517cac5148.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4000_32e26a3c53517cac5148.png
new file mode 100644
index 0000000000000000000000000000000000000000..7da9ea512b6f237a2d2727acc7f70a060f39f383
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4000_32e26a3c53517cac5148.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4032_4ab89598fad0210a9af0.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4032_4ab89598fad0210a9af0.png
new file mode 100644
index 0000000000000000000000000000000000000000..a8164cf778248103852ed898819807a5aab4c68b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4032_4ab89598fad0210a9af0.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4064_ba2ac7573eeb47664b44.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4064_ba2ac7573eeb47664b44.png
new file mode 100644
index 0000000000000000000000000000000000000000..60a7b24c1235bf4a79b7788684854a86eccf5b07
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4064_ba2ac7573eeb47664b44.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4096_c911a29ad5826c5690f9.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4096_c911a29ad5826c5690f9.png
new file mode 100644
index 0000000000000000000000000000000000000000..8ac20da2ac42c431634d2fd378b70b5e2ede9bf1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4096_c911a29ad5826c5690f9.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4128_ced3a332bb0072c48d7e.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4128_ced3a332bb0072c48d7e.png
new file mode 100644
index 0000000000000000000000000000000000000000..3e93ffca7437103f5c05001af69ed39726b20e75
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4128_ced3a332bb0072c48d7e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4160_c304e8fdd1ec86a9faa2.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4160_c304e8fdd1ec86a9faa2.png
new file mode 100644
index 0000000000000000000000000000000000000000..1703c7cc91c895cb7d612dd091e92cea12ae4de8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4160_c304e8fdd1ec86a9faa2.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_416_dcd6d36dde74eab97ed8.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_416_dcd6d36dde74eab97ed8.png
new file mode 100644
index 0000000000000000000000000000000000000000..f8ca2723a36062ed6ef2a7f82f5c2bb96eea2c64
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_416_dcd6d36dde74eab97ed8.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4192_5527643e8fdef6dc6ff2.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4192_5527643e8fdef6dc6ff2.png
new file mode 100644
index 0000000000000000000000000000000000000000..63fa249c2a0145d4bd916bf4888a92740583726d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4192_5527643e8fdef6dc6ff2.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4224_6674e7fc6a750a641325.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4224_6674e7fc6a750a641325.png
new file mode 100644
index 0000000000000000000000000000000000000000..45b68a9c8575068fc2eec146e465d31d7c26e68b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4224_6674e7fc6a750a641325.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4256_a59fb6f3aa0ec383f360.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4256_a59fb6f3aa0ec383f360.png
new file mode 100644
index 0000000000000000000000000000000000000000..76c2adb000fe712fc3344756e3bb7b74319a92cb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4256_a59fb6f3aa0ec383f360.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4288_085d0f6fdbd9c7da4fc7.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4288_085d0f6fdbd9c7da4fc7.png
new file mode 100644
index 0000000000000000000000000000000000000000..0d61696aca863e357dd686f95d71ad837c119354
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4288_085d0f6fdbd9c7da4fc7.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4320_060e978de0038abb5dbe.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4320_060e978de0038abb5dbe.png
new file mode 100644
index 0000000000000000000000000000000000000000..065cb9ccf7c84cb73ac175b8e0e0f7204d09ed8a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4320_060e978de0038abb5dbe.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4352_25adb5790e59d5e495c4.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4352_25adb5790e59d5e495c4.png
new file mode 100644
index 0000000000000000000000000000000000000000..6ebc57a361f1ae71897741f9320808c8ec758d15
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4352_25adb5790e59d5e495c4.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4384_970cf0c5b23e2c41b102.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4384_970cf0c5b23e2c41b102.png
new file mode 100644
index 0000000000000000000000000000000000000000..28813400e8012e53d3a011972d0123b7a5ef04f5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4384_970cf0c5b23e2c41b102.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4416_dce91cd3f5d3ce65c523.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4416_dce91cd3f5d3ce65c523.png
new file mode 100644
index 0000000000000000000000000000000000000000..49b0b1521b8d77d0f00249e1d5dc77fb38a43148
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4416_dce91cd3f5d3ce65c523.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4448_26be89162a8629b871bf.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4448_26be89162a8629b871bf.png
new file mode 100644
index 0000000000000000000000000000000000000000..9b6339cc40f5b63b62d0256c53713026a5f356f9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4448_26be89162a8629b871bf.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4480_b78e2cad17d1e3e57107.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4480_b78e2cad17d1e3e57107.png
new file mode 100644
index 0000000000000000000000000000000000000000..3762596e43d0da3126b58dae4f810eba3e0e268d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4480_b78e2cad17d1e3e57107.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_448_2993cc592513e322b016.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_448_2993cc592513e322b016.png
new file mode 100644
index 0000000000000000000000000000000000000000..ae8eef5cb1e24003b159d546c24c8938ab55101a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_448_2993cc592513e322b016.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4512_646edd3c3bf3742af77d.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4512_646edd3c3bf3742af77d.png
new file mode 100644
index 0000000000000000000000000000000000000000..03fe47ec9e1e4a7aba2c35614c02ca1e571b6480
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4512_646edd3c3bf3742af77d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4544_a029939ea4ad8384f89e.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4544_a029939ea4ad8384f89e.png
new file mode 100644
index 0000000000000000000000000000000000000000..1b59cab1565e5c63827afb61fe606e4d6420d33e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4544_a029939ea4ad8384f89e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4576_28b65fbf36b17ee7f1ac.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4576_28b65fbf36b17ee7f1ac.png
new file mode 100644
index 0000000000000000000000000000000000000000..d3eaa6a61f60eadc30318a44ddbcbf279b1106dc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4576_28b65fbf36b17ee7f1ac.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4608_ad8285891a794cfc0b5c.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4608_ad8285891a794cfc0b5c.png
new file mode 100644
index 0000000000000000000000000000000000000000..201bda687db2c6872ddcd6cd1b8e6cc3732b2377
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4608_ad8285891a794cfc0b5c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4640_46ae0e0e48f5468a23de.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4640_46ae0e0e48f5468a23de.png
new file mode 100644
index 0000000000000000000000000000000000000000..709405941c41acf7817d9e46230df24d0e38ed27
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4640_46ae0e0e48f5468a23de.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4672_5cda7a2770923622b883.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4672_5cda7a2770923622b883.png
new file mode 100644
index 0000000000000000000000000000000000000000..44fb6f0ea7d0b0caf8ecbb42596f94f5f7eb6793
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4672_5cda7a2770923622b883.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4704_169e4f1341edd788fcad.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4704_169e4f1341edd788fcad.png
new file mode 100644
index 0000000000000000000000000000000000000000..e848189f9ecd4a6388f2aa87cbabfa58bcb62b31
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4704_169e4f1341edd788fcad.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4736_661e854c0b214a68030e.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4736_661e854c0b214a68030e.png
new file mode 100644
index 0000000000000000000000000000000000000000..8e340e2d92916f99b4a46b6e5af9635e9d4b0391
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4736_661e854c0b214a68030e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4768_c4df59cbf286830a35c7.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4768_c4df59cbf286830a35c7.png
new file mode 100644
index 0000000000000000000000000000000000000000..6c4c33664141425ffd4607056331ba15cb945a44
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4768_c4df59cbf286830a35c7.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4800_c6d9fb60f745bb0635c2.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4800_c6d9fb60f745bb0635c2.png
new file mode 100644
index 0000000000000000000000000000000000000000..571b4ff879e3e644dc1e5e67c9d1f4f497f29602
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4800_c6d9fb60f745bb0635c2.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_480_a9223172f67cab048ad2.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_480_a9223172f67cab048ad2.png
new file mode 100644
index 0000000000000000000000000000000000000000..6c064c8bd41066d314fea899153ff159c8d2975f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_480_a9223172f67cab048ad2.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4832_fb1a28ff4160dc5ae4d8.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4832_fb1a28ff4160dc5ae4d8.png
new file mode 100644
index 0000000000000000000000000000000000000000..c86a8661809a5a085fc8b1aa96e9bcda45393b65
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4832_fb1a28ff4160dc5ae4d8.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4864_ae7b9cf5fd45f56576ef.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4864_ae7b9cf5fd45f56576ef.png
new file mode 100644
index 0000000000000000000000000000000000000000..05fea2e144266bfba1d17116160a1417652b7c19
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4864_ae7b9cf5fd45f56576ef.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4896_537131f8b589bfd3a04f.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4896_537131f8b589bfd3a04f.png
new file mode 100644
index 0000000000000000000000000000000000000000..4d1f2cd7a2a901f3a9183f18b1faa93689b36581
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4896_537131f8b589bfd3a04f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4928_e51c570e27a7cf334c5e.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4928_e51c570e27a7cf334c5e.png
new file mode 100644
index 0000000000000000000000000000000000000000..49ab3de75b21878a5b1d37948af4cb0fa7310d75
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4928_e51c570e27a7cf334c5e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4960_393f7d3dcbd9db77175c.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4960_393f7d3dcbd9db77175c.png
new file mode 100644
index 0000000000000000000000000000000000000000..1494ed29b6751ccabbef9d49dcd7e51bbfe70b80
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4960_393f7d3dcbd9db77175c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4992_dd106df305e5f26c31d0.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4992_dd106df305e5f26c31d0.png
new file mode 100644
index 0000000000000000000000000000000000000000..fe2fa524127218403b0bf07416d360a0c10978d3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_4992_dd106df305e5f26c31d0.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5024_2a2772b2ce736437eafd.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5024_2a2772b2ce736437eafd.png
new file mode 100644
index 0000000000000000000000000000000000000000..bc13809e97a6967146650950673b8a2005757c4f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5024_2a2772b2ce736437eafd.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5056_d7512b2a313799d88231.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5056_d7512b2a313799d88231.png
new file mode 100644
index 0000000000000000000000000000000000000000..87bd2db24f418a405213132cfaaefbc3f39d162f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5056_d7512b2a313799d88231.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5088_67d8c78113c1a7810739.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5088_67d8c78113c1a7810739.png
new file mode 100644
index 0000000000000000000000000000000000000000..8542cf3767a5976ebc31e4d6d7aa125d013b3a90
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5088_67d8c78113c1a7810739.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5120_c35a4292835da347b47d.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5120_c35a4292835da347b47d.png
new file mode 100644
index 0000000000000000000000000000000000000000..8ff75410ec5b85d41c02e9cde8f093e173644103
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5120_c35a4292835da347b47d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_512_36dbdc441b153e42ba4a.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_512_36dbdc441b153e42ba4a.png
new file mode 100644
index 0000000000000000000000000000000000000000..b940e76cf2b2d8a4b1a9d0ba3597cef248e15dd4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_512_36dbdc441b153e42ba4a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5152_9b9f8658baabff632b59.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5152_9b9f8658baabff632b59.png
new file mode 100644
index 0000000000000000000000000000000000000000..4342ae75c6f235f66dc288bc357cb8c40bbfed07
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5152_9b9f8658baabff632b59.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5184_c419c882e7c0f7ca3da3.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5184_c419c882e7c0f7ca3da3.png
new file mode 100644
index 0000000000000000000000000000000000000000..465d5062a9301b89d203e039b6fe88c0c5fd4102
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5184_c419c882e7c0f7ca3da3.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5216_4de441c52e08bd82efa2.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5216_4de441c52e08bd82efa2.png
new file mode 100644
index 0000000000000000000000000000000000000000..5986c8801859b549e727bbc8479c04a1e72057af
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5216_4de441c52e08bd82efa2.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5248_75577d8ace8354a664c1.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5248_75577d8ace8354a664c1.png
new file mode 100644
index 0000000000000000000000000000000000000000..6448ed13300c9126e34b2a98e4ef279d3a1aae62
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5248_75577d8ace8354a664c1.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5280_ee3a2874b441382eafe6.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5280_ee3a2874b441382eafe6.png
new file mode 100644
index 0000000000000000000000000000000000000000..3aaa4dd344c2964b0804e29479a3d907785870bf
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5280_ee3a2874b441382eafe6.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5312_d0fb888d743f65e43c5f.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5312_d0fb888d743f65e43c5f.png
new file mode 100644
index 0000000000000000000000000000000000000000..260683c8dc602c8cc0122588bc59df83e47dbb70
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5312_d0fb888d743f65e43c5f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5344_23ce83e24168a74c9faa.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5344_23ce83e24168a74c9faa.png
new file mode 100644
index 0000000000000000000000000000000000000000..33932fff32e8497cc7526e1657e9d40c386b9efc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5344_23ce83e24168a74c9faa.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5376_a589bafc74ff362e454c.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5376_a589bafc74ff362e454c.png
new file mode 100644
index 0000000000000000000000000000000000000000..e8b690035cdf003f32fcda9d3ed83ad0f7d86b6d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5376_a589bafc74ff362e454c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5408_bf5e8b30155ff4225292.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5408_bf5e8b30155ff4225292.png
new file mode 100644
index 0000000000000000000000000000000000000000..97b5221bc957b4c4b9d50ef075615831868acfe8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5408_bf5e8b30155ff4225292.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5440_941b0cd5e5f8715818c2.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5440_941b0cd5e5f8715818c2.png
new file mode 100644
index 0000000000000000000000000000000000000000..c1db5aff73950560b20f7678657bd9c3b86565a0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5440_941b0cd5e5f8715818c2.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_544_933ae7db2c226b732ed4.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_544_933ae7db2c226b732ed4.png
new file mode 100644
index 0000000000000000000000000000000000000000..cdcfc5488591e282540dda5b9942e129d57435de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_544_933ae7db2c226b732ed4.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5472_b0494c6a4ef7be86849c.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5472_b0494c6a4ef7be86849c.png
new file mode 100644
index 0000000000000000000000000000000000000000..3ca6f58fae02f082b7aceacd7aaf0a1d806fdaee
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5472_b0494c6a4ef7be86849c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5504_ef841e93aee1d9585e55.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5504_ef841e93aee1d9585e55.png
new file mode 100644
index 0000000000000000000000000000000000000000..04139b254a242ee8939ab90e5b7178de13b300b0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5504_ef841e93aee1d9585e55.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5536_446b5b4f915297265979.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5536_446b5b4f915297265979.png
new file mode 100644
index 0000000000000000000000000000000000000000..0658ac81b8e503f15f57ea59f131925951ff6cac
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5536_446b5b4f915297265979.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5568_4b88d37337bde0f9c8be.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5568_4b88d37337bde0f9c8be.png
new file mode 100644
index 0000000000000000000000000000000000000000..85ec99c57329d51aebd85f90b5381c7e9f431457
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5568_4b88d37337bde0f9c8be.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5600_f2c5f8913d2e9d7785d5.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5600_f2c5f8913d2e9d7785d5.png
new file mode 100644
index 0000000000000000000000000000000000000000..b6b1b9e4698fe74492b67181fb3e370b717b4247
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5600_f2c5f8913d2e9d7785d5.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5632_577820521ce567958326.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5632_577820521ce567958326.png
new file mode 100644
index 0000000000000000000000000000000000000000..2c048b0acb68f9784d4f71ff0dd6f75562e7de5a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5632_577820521ce567958326.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5664_6ff7214aa96ee016ba5a.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5664_6ff7214aa96ee016ba5a.png
new file mode 100644
index 0000000000000000000000000000000000000000..5f3c899f649f406b4fbd40406518c871a816a83e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5664_6ff7214aa96ee016ba5a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5696_21a9f99491955adf7a95.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5696_21a9f99491955adf7a95.png
new file mode 100644
index 0000000000000000000000000000000000000000..53b0981b232fd3ea6e1670616b143bea956e4615
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5696_21a9f99491955adf7a95.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5728_51b3f92cd05d11ee2451.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5728_51b3f92cd05d11ee2451.png
new file mode 100644
index 0000000000000000000000000000000000000000..629a38af6a64101d90399d3465ac10ad4fabe101
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5728_51b3f92cd05d11ee2451.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5760_56660ad1b00f4f26e0a1.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5760_56660ad1b00f4f26e0a1.png
new file mode 100644
index 0000000000000000000000000000000000000000..874a4f5c9ef7b582a30cb08aaeb0fb0bd2ef4276
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5760_56660ad1b00f4f26e0a1.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_576_7016d56c0f4f7ce4e542.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_576_7016d56c0f4f7ce4e542.png
new file mode 100644
index 0000000000000000000000000000000000000000..b66a3dd2b7eacb7d7316deb278099412f4789a2b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_576_7016d56c0f4f7ce4e542.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5792_b033b4b9679bb421e6dd.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5792_b033b4b9679bb421e6dd.png
new file mode 100644
index 0000000000000000000000000000000000000000..b26b703c9042af9926b78e16236d33c410ca8e77
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5792_b033b4b9679bb421e6dd.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5824_1caabeb8d9e6938d9521.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5824_1caabeb8d9e6938d9521.png
new file mode 100644
index 0000000000000000000000000000000000000000..15bcf500df4f0fd2a47f734ab6548e66d650dd8b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5824_1caabeb8d9e6938d9521.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5856_a9fe699e19cd51caeba7.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5856_a9fe699e19cd51caeba7.png
new file mode 100644
index 0000000000000000000000000000000000000000..ad649cce6a3db387150250b065a63323f25add67
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5856_a9fe699e19cd51caeba7.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5888_b62ddf26429b50f5342b.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5888_b62ddf26429b50f5342b.png
new file mode 100644
index 0000000000000000000000000000000000000000..223216396109097456ac4b2af2754d1458fe47f9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5888_b62ddf26429b50f5342b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5920_8cb9fc0b252238341b67.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5920_8cb9fc0b252238341b67.png
new file mode 100644
index 0000000000000000000000000000000000000000..cd2362ab3df8d2b092aaa3a98d53361646516241
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5920_8cb9fc0b252238341b67.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5952_5f2089b6d652eb32abe0.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5952_5f2089b6d652eb32abe0.png
new file mode 100644
index 0000000000000000000000000000000000000000..dedfc6052f693c310fd4e6657f21186eef4f189d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5952_5f2089b6d652eb32abe0.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5984_3b5bdcf45b42cd7f1dfe.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5984_3b5bdcf45b42cd7f1dfe.png
new file mode 100644
index 0000000000000000000000000000000000000000..189433021a16730858b52b0a8326fa66b5c4c750
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_5984_3b5bdcf45b42cd7f1dfe.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_6016_82de052c056f3b871c0a.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_6016_82de052c056f3b871c0a.png
new file mode 100644
index 0000000000000000000000000000000000000000..362a86073ed7f198f23a478106d52efb8c3f871c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_6016_82de052c056f3b871c0a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_6048_f04bffb7b2d8fb3e6ad2.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_6048_f04bffb7b2d8fb3e6ad2.png
new file mode 100644
index 0000000000000000000000000000000000000000..0643244b37a721a5bda41ba3cf10a56f7be5809b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_6048_f04bffb7b2d8fb3e6ad2.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_608_ea0e8c30476e47b9914e.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_608_ea0e8c30476e47b9914e.png
new file mode 100644
index 0000000000000000000000000000000000000000..8aedb57a1866dd276e5c1c5b5a84553e5c1e73f7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_608_ea0e8c30476e47b9914e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_640_146d35e74db1aa8d89e0.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_640_146d35e74db1aa8d89e0.png
new file mode 100644
index 0000000000000000000000000000000000000000..a6a940a353090d17666261ee516c4ce927176f60
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_640_146d35e74db1aa8d89e0.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_64_e5fa372f1deec2d3b94d.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_64_e5fa372f1deec2d3b94d.png
new file mode 100644
index 0000000000000000000000000000000000000000..6ea5190be1039ae99ee570411d2005e61d5ddd93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_64_e5fa372f1deec2d3b94d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_672_ec3d2ecd2ab61ca507b5.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_672_ec3d2ecd2ab61ca507b5.png
new file mode 100644
index 0000000000000000000000000000000000000000..979ccd996074fc496852d163aa110c970813007a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_672_ec3d2ecd2ab61ca507b5.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_704_26478416d9948f1a6469.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_704_26478416d9948f1a6469.png
new file mode 100644
index 0000000000000000000000000000000000000000..62707fcb9c4f47c91b4c4b6817b72588f6a133d9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_704_26478416d9948f1a6469.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_736_3189a910beb6d64bed7e.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_736_3189a910beb6d64bed7e.png
new file mode 100644
index 0000000000000000000000000000000000000000..9e3e74348caaa720eb7754c94df915d735d78633
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_736_3189a910beb6d64bed7e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_768_9a53434e232af779e4dd.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_768_9a53434e232af779e4dd.png
new file mode 100644
index 0000000000000000000000000000000000000000..29a3f3fb9a84fd15754b4333767ef0090115bf9a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_768_9a53434e232af779e4dd.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_800_1df5772c9e8200dcb21a.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_800_1df5772c9e8200dcb21a.png
new file mode 100644
index 0000000000000000000000000000000000000000..a3042e1048d8b47618bd6363db32da5149161034
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_800_1df5772c9e8200dcb21a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_832_7611024f11b7c085c103.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_832_7611024f11b7c085c103.png
new file mode 100644
index 0000000000000000000000000000000000000000..fd9ec7024cf761616d42133ec5bb3228d9b60eb7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_832_7611024f11b7c085c103.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_864_a65cff6a5819aacc6667.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_864_a65cff6a5819aacc6667.png
new file mode 100644
index 0000000000000000000000000000000000000000..140fd67628d3f5f2c55dc0796540d23d364d5c77
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_864_a65cff6a5819aacc6667.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_896_c3927a6f3950222f5b2c.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_896_c3927a6f3950222f5b2c.png
new file mode 100644
index 0000000000000000000000000000000000000000..d77463da8041375c0e06e9e2494b3b9183ff7d4d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_896_c3927a6f3950222f5b2c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_928_cc2a93a395fc585ad5eb.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_928_cc2a93a395fc585ad5eb.png
new file mode 100644
index 0000000000000000000000000000000000000000..934af81351184853691abebb2aff2950aaabf4e0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_928_cc2a93a395fc585ad5eb.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_960_2d25485fecb8f640765e.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_960_2d25485fecb8f640765e.png
new file mode 100644
index 0000000000000000000000000000000000000000..c13804af72f378a08e18e4c374650c960c7b3358
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_960_2d25485fecb8f640765e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_96_e16fc76e4fe578fa7899.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_96_e16fc76e4fe578fa7899.png
new file mode 100644
index 0000000000000000000000000000000000000000..5e5fdf4cf8ece2d615f27e67583c170cd57408e9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_96_e16fc76e4fe578fa7899.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_992_d7b78dc81c7d635e6a33.png b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_992_d7b78dc81c7d635e6a33.png
new file mode 100644
index 0000000000000000000000000000000000000000..daa794849bd9e0243bbeac8157c7af65db6a7a32
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/images/ued/layout/levels16_img_992_d7b78dc81c7d635e6a33.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_0_f9a2c7b12fae2212b0df.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_0_f9a2c7b12fae2212b0df.gif
new file mode 100644
index 0000000000000000000000000000000000000000..4c6e8a3828b2e20f624fb6b354d664b21df47c7a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_0_f9a2c7b12fae2212b0df.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f9a2c7b12fae2212b0df142becd66daecf0fbeeaa486496b064851a1354eb8d1
+size 105420
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_2016_00d269aa1bc9116f8bd7.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_2016_00d269aa1bc9116f8bd7.gif
new file mode 100644
index 0000000000000000000000000000000000000000..011ac2e1318ae38bbd0d6cdfc99a34f97a691c1a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_2016_00d269aa1bc9116f8bd7.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:00d269aa1bc9116f8bd7268206d19ff568aeec95db6d20fcd8184076f0d0afd8
+size 263870
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_3040_725b9a6669dd81146e9c.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_3040_725b9a6669dd81146e9c.gif
new file mode 100644
index 0000000000000000000000000000000000000000..bb7991db9c92c50605626942ea32598449982518
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_3040_725b9a6669dd81146e9c.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:725b9a6669dd81146e9cb941d8b453ba48d6b792ce0a8f61d276e1d6d4ed2677
+size 259059
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_4064_94f43fd1734d82639b65.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_4064_94f43fd1734d82639b65.gif
new file mode 100644
index 0000000000000000000000000000000000000000..93f308b06a68f21e10f76dec474489c9cd784269
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_4064_94f43fd1734d82639b65.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:94f43fd1734d82639b655db6d4a2596f23997a91d3986f25a438a181afa91eac
+size 259851
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_5088_b3c2b0820fb0978fe942.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_5088_b3c2b0820fb0978fe942.gif
new file mode 100644
index 0000000000000000000000000000000000000000..8b38e8727dcef67dffc6feb9b8b095ba80e5144e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_5088_b3c2b0820fb0978fe942.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b3c2b0820fb0978fe942cd07f11d5dddee11d83c212e03ab2d9822556f064521
+size 257166
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_992_8d84766ff56b911dc3c5.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_992_8d84766ff56b911dc3c5.gif
new file mode 100644
index 0000000000000000000000000000000000000000..a0c5193472efb11540e8ef92648103f5d8f73fd7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_992_8d84766ff56b911dc3c5.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8d84766ff56b911dc3c537800459a7729fb817e10fd96cf7c2bf2cda1706ee79
+size 264056
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_0_4d343630d22822dd23f9.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_0_4d343630d22822dd23f9.gif
new file mode 100644
index 0000000000000000000000000000000000000000..2a54c425f9bdd0c502c020936d15b70cd33dea26
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_0_4d343630d22822dd23f9.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4d343630d22822dd23f92b49e2636379b3581666f68a15e9370951a3be0e793f
+size 116198
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_2016_43487c06f7c8cf0ce4b3.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_2016_43487c06f7c8cf0ce4b3.gif
new file mode 100644
index 0000000000000000000000000000000000000000..cccede8074eae187c4297c95ed03ed8a2b48e6b8
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_2016_43487c06f7c8cf0ce4b3.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:43487c06f7c8cf0ce4b36e2e848f3990661a69ee17d0727fe536b76e742affa4
+size 316819
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_3040_9c0d784d9b58feb05295.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_3040_9c0d784d9b58feb05295.gif
new file mode 100644
index 0000000000000000000000000000000000000000..3839420d6ca0acd3a9c0c5138350c17fd6747b99
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_3040_9c0d784d9b58feb05295.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9c0d784d9b58feb05295cff56f47b21279b394c39f4097a93e353414c860bac1
+size 312492
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_4064_46e3a9cb8d942efee5a7.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_4064_46e3a9cb8d942efee5a7.gif
new file mode 100644
index 0000000000000000000000000000000000000000..f73b908e8e256fe23afc287adcfd244e5513a7cb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_4064_46e3a9cb8d942efee5a7.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:46e3a9cb8d942efee5a711235d126281a89e145f46715495f630a6b5db962c45
+size 313835
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_5088_6c6bfa49d6645a4ec744.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_5088_6c6bfa49d6645a4ec744.gif
new file mode 100644
index 0000000000000000000000000000000000000000..d3f1a389a2c3937297d1ceafc53ade7cabbb7ac7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_5088_6c6bfa49d6645a4ec744.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6c6bfa49d6645a4ec744fa0e20c52fa65ce644e80919f1a68e45be9114b583f8
+size 314868
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_992_ea0b419521e17e14c4f8.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_992_ea0b419521e17e14c4f8.gif
new file mode 100644
index 0000000000000000000000000000000000000000..c8bc4a0e9a9b50da2da8c9fee29e6189081d2202
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_992_ea0b419521e17e14c4f8.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ea0b419521e17e14c4f8e386a40b9672774e7d5b18932fa55460b4208ff79ca9
+size 314579
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_0_bc8a36e71dc82e452ec5.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_0_bc8a36e71dc82e452ec5.gif
new file mode 100644
index 0000000000000000000000000000000000000000..f578e21f325d0ce061304b7352d92dbb6326d96e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_0_bc8a36e71dc82e452ec5.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bc8a36e71dc82e452ec577a13578efa32ae67c7d76fc95a718e71be066800272
+size 115532
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_2016_94d3910b42650d354441.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_2016_94d3910b42650d354441.gif
new file mode 100644
index 0000000000000000000000000000000000000000..d0dcc47051dc8ac1fd49c96576c64787bd9013c3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_2016_94d3910b42650d354441.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:94d3910b42650d35444187c7ade9be7fe76efc93598fd03ed7af46a9ebe7b7cd
+size 313079
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_3040_d0223c2c92823109e63d.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_3040_d0223c2c92823109e63d.gif
new file mode 100644
index 0000000000000000000000000000000000000000..17573150d3d42a8d8d7aae7b836948a22b0ea667
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_3040_d0223c2c92823109e63d.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d0223c2c92823109e63d4d7f39c6839b382df9ffc89182ca5ef05d9b39431c83
+size 309844
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_4064_871827662648763adffe.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_4064_871827662648763adffe.gif
new file mode 100644
index 0000000000000000000000000000000000000000..b4b9df4428196eae790833de3adacb29ae3c33da
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_4064_871827662648763adffe.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:871827662648763adffe96c3deaca8fee6d40d129b7f0e5cdfd77be1ad88b5f8
+size 310614
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_5088_d89e30114bcc244cca07.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_5088_d89e30114bcc244cca07.gif
new file mode 100644
index 0000000000000000000000000000000000000000..f69c6e1fe5f31a298b5b0b23ce8cfc361d56c329
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_5088_d89e30114bcc244cca07.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d89e30114bcc244cca071629f0fccfc25dc43d572f879bb0bf13d8acd51d5e10
+size 309458
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_992_100ea78a33a1369d62ea.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_992_100ea78a33a1369d62ea.gif
new file mode 100644
index 0000000000000000000000000000000000000000..9dcbf385f8e47534ff188e4c608318b24c02740e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_992_100ea78a33a1369d62ea.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:100ea78a33a1369d62ea12b01d4e726b3f43b3a220df534a6ce673c8d1534f5e
+size 309656
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_0_92fcce4df8a2aebf7a46.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_0_92fcce4df8a2aebf7a46.gif
new file mode 100644
index 0000000000000000000000000000000000000000..e34e366a877de73aa6fb133abcafaf945f517ca2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_0_92fcce4df8a2aebf7a46.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:92fcce4df8a2aebf7a46fbfa7c393c029351596ed4e7bbf5357baaea23e9ace5
+size 108075
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_2016_58d1901943a7bbaa90fa.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_2016_58d1901943a7bbaa90fa.gif
new file mode 100644
index 0000000000000000000000000000000000000000..d94c5334d4f6a35b6f396cdb155bbc1b381daea8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_2016_58d1901943a7bbaa90fa.gif differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_3040_2241126301d22704f30f.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_3040_2241126301d22704f30f.gif
new file mode 100644
index 0000000000000000000000000000000000000000..90a31c5827e593f89bb22be2d3ead7c58e76f890
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_3040_2241126301d22704f30f.gif differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_4064_31bfb8afdbba855b8609.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_4064_31bfb8afdbba855b8609.gif
new file mode 100644
index 0000000000000000000000000000000000000000..06fdaf38eeb671268b59caa1b2c9253fddbbee86
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_4064_31bfb8afdbba855b8609.gif differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_5088_cc92313eab6f5c385d8f.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_5088_cc92313eab6f5c385d8f.gif
new file mode 100644
index 0000000000000000000000000000000000000000..271b8199f9548481b7e5e49d71c7f1b4248f59f7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_5088_cc92313eab6f5c385d8f.gif differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_992_61ad0ecfb211ed4c0769.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_992_61ad0ecfb211ed4c0769.gif
new file mode 100644
index 0000000000000000000000000000000000000000..d0399e3ccb89def49bd6f42fd97dc8cd7ef8ef14
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_992_61ad0ecfb211ed4c0769.gif differ
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-all/rollouts_gif_0_aaa4468a8c6a23e11cd1.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-all/rollouts_gif_0_aaa4468a8c6a23e11cd1.gif
new file mode 100644
index 0000000000000000000000000000000000000000..c0384992477ddcc759e4d03b3b92a94dbbee1bd2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-all/rollouts_gif_0_aaa4468a8c6a23e11cd1.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aaa4468a8c6a23e11cd1452715e0e1b79c585bdb9129626f40df6afd6b4ec634
+size 116935
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-all/rollouts_gif_2016_e428956eff714d710b89.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-all/rollouts_gif_2016_e428956eff714d710b89.gif
new file mode 100644
index 0000000000000000000000000000000000000000..abd04a5eec178189dac78e7ce9e5468e60a009b2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-all/rollouts_gif_2016_e428956eff714d710b89.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e428956eff714d710b89c03c7ea068a063da4299f1f47208a8d4d113b8e79eaf
+size 195778
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-all/rollouts_gif_3040_a0da2a31a518817b6e32.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-all/rollouts_gif_3040_a0da2a31a518817b6e32.gif
new file mode 100644
index 0000000000000000000000000000000000000000..04e9a2821b3cf423e0498b4660543639d1a9d7c8
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-all/rollouts_gif_3040_a0da2a31a518817b6e32.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a0da2a31a518817b6e3286218abb18bd1131c52f1100749a7592700b9fff97ef
+size 308029
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-all/rollouts_gif_4064_9f611d045168c7ee20d2.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-all/rollouts_gif_4064_9f611d045168c7ee20d2.gif
new file mode 100644
index 0000000000000000000000000000000000000000..caeb412f63406d55a4547e20ab164ee004186358
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-all/rollouts_gif_4064_9f611d045168c7ee20d2.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9f611d045168c7ee20d2763ab1e23a2fba3a3a8e63d00c739c9e9bfa7479ce93
+size 310347
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-all/rollouts_gif_5088_16ca120b4aa0033a7e82.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-all/rollouts_gif_5088_16ca120b4aa0033a7e82.gif
new file mode 100644
index 0000000000000000000000000000000000000000..a6121373ce9f9b332543f8d3a6f5f35c18ba6f9e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-all/rollouts_gif_5088_16ca120b4aa0033a7e82.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:16ca120b4aa0033a7e82af3200508d2e90bfd4ddaf9ecaf2542ed10f9ac59f4e
+size 169261
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-all/rollouts_gif_992_287901f7a7d2c26d0a85.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-all/rollouts_gif_992_287901f7a7d2c26d0a85.gif
new file mode 100644
index 0000000000000000000000000000000000000000..6af3c84028e5d9685cdfb8f6fce6febd0831317c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-all/rollouts_gif_992_287901f7a7d2c26d0a85.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:287901f7a7d2c26d0a85f9c4bec3ac9a563b38f9d30c08c87e7f1b60343ef51c
+size 301237
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-generate/rollouts_gif_3040_a0da2a31a518817b6e32.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-generate/rollouts_gif_3040_a0da2a31a518817b6e32.gif
new file mode 100644
index 0000000000000000000000000000000000000000..04e9a2821b3cf423e0498b4660543639d1a9d7c8
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-generate/rollouts_gif_3040_a0da2a31a518817b6e32.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a0da2a31a518817b6e3286218abb18bd1131c52f1100749a7592700b9fff97ef
+size 308029
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-generate/rollouts_gif_4064_9f611d045168c7ee20d2.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-generate/rollouts_gif_4064_9f611d045168c7ee20d2.gif
new file mode 100644
index 0000000000000000000000000000000000000000..caeb412f63406d55a4547e20ab164ee004186358
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-generate/rollouts_gif_4064_9f611d045168c7ee20d2.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9f611d045168c7ee20d2763ab1e23a2fba3a3a8e63d00c739c9e9bfa7479ce93
+size 310347
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-generate/rollouts_gif_992_287901f7a7d2c26d0a85.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-generate/rollouts_gif_992_287901f7a7d2c26d0a85.gif
new file mode 100644
index 0000000000000000000000000000000000000000..6af3c84028e5d9685cdfb8f6fce6febd0831317c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-generate/rollouts_gif_992_287901f7a7d2c26d0a85.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:287901f7a7d2c26d0a85f9c4bec3ac9a563b38f9d30c08c87e7f1b60343ef51c
+size 301237
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-replay/rollouts_gif_0_aaa4468a8c6a23e11cd1.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-replay/rollouts_gif_0_aaa4468a8c6a23e11cd1.gif
new file mode 100644
index 0000000000000000000000000000000000000000..c0384992477ddcc759e4d03b3b92a94dbbee1bd2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-replay/rollouts_gif_0_aaa4468a8c6a23e11cd1.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aaa4468a8c6a23e11cd1452715e0e1b79c585bdb9129626f40df6afd6b4ec634
+size 116935
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-replay/rollouts_gif_2016_e428956eff714d710b89.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-replay/rollouts_gif_2016_e428956eff714d710b89.gif
new file mode 100644
index 0000000000000000000000000000000000000000..abd04a5eec178189dac78e7ce9e5468e60a009b2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-replay/rollouts_gif_2016_e428956eff714d710b89.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e428956eff714d710b89c03c7ea068a063da4299f1f47208a8d4d113b8e79eaf
+size 195778
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-replay/rollouts_gif_5088_16ca120b4aa0033a7e82.gif b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-replay/rollouts_gif_5088_16ca120b4aa0033a7e82.gif
new file mode 100644
index 0000000000000000000000000000000000000000..a6121373ce9f9b332543f8d3a6f5f35c18ba6f9e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/media/videos/train-replay/rollouts_gif_5088_16ca120b4aa0033a7e82.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:16ca120b4aa0033a7e82af3200508d2e90bfd4ddaf9ecaf2542ed10f9ac59f4e
+size 169261
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/output.log b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/output.log
new file mode 100644
index 0000000000000000000000000000000000000000..e11d57dd32c5a620d5b983a36b4e25bdb9c5fc26
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/output.log
@@ -0,0 +1,25061 @@
+  env_size                      : 13
+  env_layout                    : tree
+  wall_prob                     : 0
+  env_corner_size               : 1
+  env_terminate_after_corner    : False
+  obs_level_of_detail           : 0
+  img_level_of_detail           : 1
+  env_penalize_time             : False
+  net_cnn_type                  : large
+  net_rnn_type                  : ff
+  net_width                     : 256
+  ued                           : plr
+  prob_shift                    : 1
+  num_train_levels              : 2048
+  plr_buffer_size               : 4096
+  plr_temperature               : 0.1
+  plr_staleness_coeff           : 0.1
+  plr_prob_replay               : 0.5
+  plr_regret_estimator          : maxmc-actor
+  plr_robust                    : False
+  num_mutate_steps              : 12
+  prob_mutate_shift             : 0
+  chain_mutate                  : True
+  mutate_cheese                 : True
+  train_proxy_critic            : False
+  plr_proxy_shaping             : False
+  proxy_name                    : proxy_corner
+  plr_proxy_shaping_coeff       : 0.5
+  clipping                      : False
+  ppo_lr                        : 5e-05
+  ppo_gamma                     : 0.999
+  ppo_clip_eps                  : 0.1
+  ppo_gae_lambda                : 0.95
+  ppo_entropy_coeff             : 0.001
+  ppo_critic_coeff              : 0.5
+  ppo_proxy_critic_coeff        : 0.5
+  ppo_max_grad_norm             : 0.5
+  ppo_lr_annealing              : False
+  num_minibatches_per_epoch     : 4
+  num_epochs_per_cycle          : 5
+  num_total_env_steps           : 200000000
+  num_env_steps_per_cycle       : 128
+  num_parallel_envs             : 256
+  console_log                   : True
+  wandb_log                     : True
+  wandb_project                 : jaxgmg-wall-prob-tree
+  wandb_entity                  : None
+  wandb_group                   : None
+  wandb_name                    : wp:0.0_alpha:1
+  log_gifs                      : True
+  log_imgs                      : True
+  log_hists                     : False
+  num_cycles_per_log            : 32
+  num_cycles_per_eval           : 32
+  num_cycles_per_gifs           : 1024
+  num_cycles_per_big_eval       : 1024
+  use_fixed_eval_levels         : False
+  evals_num_env_steps           : 512
+  evals_num_levels              : 256
+  gif_grid_width                : 16
+  level_splayer                 : mouse
+  checkpointing                 : True
+  keep_all_checkpoints          : True
+  max_num_checkpoints           : 1
+  num_cycles_per_checkpoint     : 64
+  seed                          : 42
+configuring environment...
+configuring level generators...
+  mixing level generators with {prob_shift=}...
+configuring level classifier...
+configuring level mutator...
+configuring level solver...
+configuring level metrics...
+configuring eval level generators...
+configuring parser and parsing fixed eval levels...
+(unsupported size for fixed evals)
+configuring heatmap splayer...
+seeding random number generator with seed=42...
+configuring curriculum with ued='plr'...
+configuring eval batches from 4 level generators...
+  generating 256 'train' levels...
+  also solving generated levels...
+  generating 256 'orig' levels...
+  also solving generated levels...
+  generating 256 'shift' levels...
+  also solving generated levels...
+  generating 256 'tree' levels...
+  also solving generated levels...
+configuring evals for 0 fixed eval levels...
+configuring actor critic network...
+  net_cnn_type='large'
+  net_rnn_type='ff'
+  net_width=256
+  initialising network...
+  number of parameters: 198982
+initialising the checkpointer...
+setting up optimiser...
+initialising train state...
+configuring PPO updater...
+begin training loop!
+(note: first two cycles are slow due to compilation)
+                                                                                                                                                                                                                                                                         steps/s]
+===========================================================
+  step:
+    env-step-all-after:                         32768
+    env-step-replay-after:                      32768
+    env-step-used-after:                        32768
+    ppo-update-after:                           20
+  train-replay:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     79.82981
+    avg_avg_return:                             0.1919119
+    avg_reward_per_step:                        0.0048217773
+    proxy_corner:
+      avg_avg_return:                           0.073624715
+      avg_reward_per_step:                      0.0008544922
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     79.82981
+    avg_avg_return:                             0.1919119
+    avg_reward_per_step:                        0.0048217773
+    proxy_corner:
+      avg_avg_return:                           0.073624715
+      avg_reward_per_step:                      0.0008544922
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  perf:
+    env_steps_per_second:                       3072.330569459758
+    ppo_updates_per_second:                     0.5565518154684599
+    eval-batch-train-duration:                  10.672633411362767
+    eval-batch-orig-duration:                   0.29637806583195925
+    eval-batch-shift-duration:                  0.29042063280940056
+    eval-batch-tree-duration:                   0.2900292417034507
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.1154785
+      cheese-corne_dist_wavg:                   16.308754
+      levels_cheese_is_in_corner_avg:           0.0078125
+      mouse-cheese_dist_finite_avg:             7.3464355
+      mouse-cheese_dist_finite_wavg:            5.2355084
+      mouse-cheese_dist_solvable_avg:           7.3464355
+      mouse-cheese_dist_solvable_wavg:          5.2355065
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000004
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.008591815
+    visit_patterns:
+      avg_first_visit_time:                     0.0
+      avg_last_visit_time:                      0.0625
+      num_replay_batches:                       1
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011464998
+    avg_actor_approxkl3:                        0.0014054141
+    avg_actor_clipfrac:                         0.036254883
+    avg_actor_loss:                             -0.0010393184
+    avg_advantage:                              0.097278416
+    avg_critic_clipfrac:                        0.26132813
+    avg_critic_loss:                            0.025304899
+    avg_entropy:                                1.363961
+    avg_grad_norm_pre_clip:                     0.23948975
+    avg_loss:                                   0.010249172
+    max:
+      max_actor_approxkl1:                      0.0026931292
+      max_actor_approxkl3:                      0.002446468
+      max_actor_clipfrac:                       0.13415527
+      max_actor_loss:                           7.827394e-05
+      max_advantage:                            1.2874098
+      max_critic_clipfrac:                      0.9995117
+      max_critic_loss:                          0.03081631
+      max_entropy:                              1.3678575
+      max_grad_norm_pre_clip:                   0.41760215
+      max_loss:                                 0.01408309
+    std:
+      std_actor_approxkl1:                      0.00079628534
+      std_actor_approxkl3:                      0.0008090679
+      std_actor_clipfrac:                       0.04133498
+      std_actor_loss:                           0.0008025807
+      std_advantage:                            0.22287124
+      std_critic_clipfrac:                      0.34504935
+      std_critic_loss:                          0.0034872352
+      std_entropy:                              0.0047579287
+      std_grad_norm_pre_clip:                   0.079162985
+      std_loss:                                 0.0022817047
+  eval-batch-train:
+    avg_avg_episode_length:                     76.90827
+    avg_avg_return:                             0.24662751
+    avg_benchmark_regret:                       0.7461717
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.003982544
+    proxy_corner:
+      avg_avg_return:                           0.063186914
+      avg_reward_per_step:                      0.0006790161
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-orig:
+    avg_avg_episode_length:                     84.00958
+    avg_avg_return:                             0.06699096
+    avg_benchmark_regret:                       0.9232489
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.0008468628
+    proxy_corner:
+      avg_avg_return:                           0.06699096
+      avg_reward_per_step:                      0.0008468628
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-shift:
+    avg_avg_episode_length:                     76.21381
+    avg_avg_return:                             0.2578784
+    avg_benchmark_regret:                       0.734959
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.0039520264
+    proxy_corner:
+      avg_avg_return:                           0.08564603
+      avg_reward_per_step:                      0.0009841919
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.87308
+    avg_avg_return:                             0.12361042
+    avg_benchmark_regret:                       0.86102474
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0038604736
+    proxy_corner:
+      avg_avg_return:                           0.12361042
+      avg_reward_per_step:                      0.0038604736
+    rollouts_gif:                               float32(132, 225, 225, 3)
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        1048576
+    env-step-replay-before:                     557056
+    env-step-used-before:                       1048576
+    ppo-update-before:                          640
+    env-step-generate-before:                   491520
+    env-step-all-after:                         1081344
+    env-step-replay-after:                      589824
+    env-step-used-after:                        1081344
+    ppo-update-after:                           660
+    env-step-generate-after:                    491520
+  train-replay:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     28.328436
+    avg_avg_return:                             0.7798295
+    avg_reward_per_step:                        0.047546387
+    proxy_corner:
+      avg_avg_return:                           0.088643484
+      avg_reward_per_step:                      0.002746582
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     28.328436
+    avg_avg_return:                             0.7798295
+    avg_reward_per_step:                        0.047546387
+    proxy_corner:
+      avg_avg_return:                           0.088643484
+      avg_reward_per_step:                      0.002746582
+  perf:
+    env_steps_per_second:                       306417.4449802745
+    ppo_updates_per_second:                     42.02804977510154
+    eval-batch-train-duration:                  0.5295405620709062
+    eval-batch-orig-duration:                   0.49399377778172493
+    eval-batch-shift-duration:                  0.48831842094659805
+    eval-batch-tree-duration:                   0.48240647464990616
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.106201
+      cheese-corne_dist_wavg:                   13.599809
+      levels_cheese_is_in_corner_avg:           0.005859375
+      mouse-cheese_dist_finite_avg:             7.5029297
+      mouse-cheese_dist_finite_wavg:            5.2366996
+      mouse-cheese_dist_solvable_avg:           7.5029297
+      mouse-cheese_dist_solvable_wavg:          5.2367
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.24569221
+    visit_patterns:
+      avg_first_visit_time:                     3.0236816
+      avg_last_visit_time:                      8.903809
+      num_replay_batches:                       18
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013151083
+    avg_actor_approxkl3:                        0.0011655659
+    avg_actor_clipfrac:                         0.04511719
+    avg_actor_loss:                             -0.0011333146
+    avg_advantage:                              -0.0033074142
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0020198948
+    avg_entropy:                                1.154582
+    avg_grad_norm_pre_clip:                     0.80009645
+    avg_loss:                                   -0.0012779495
+    max:
+      max_actor_approxkl1:                      0.002782328
+      max_actor_approxkl3:                      0.0023681293
+      max_actor_clipfrac:                       0.119018555
+      max_actor_loss:                           0.00076064444
+      max_advantage:                            0.2101565
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00386511
+      max_entropy:                              1.1667609
+      max_grad_norm_pre_clip:                   2.2085636
+      max_loss:                                 0.00061298744
+    std:
+      std_actor_approxkl1:                      0.0007115727
+      std_actor_approxkl3:                      0.0005858848
+      std_actor_clipfrac:                       0.03294868
+      std_actor_loss:                           0.0010164059
+      std_advantage:                            0.0637953
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0008773323
+      std_entropy:                              0.00798986
+      std_grad_norm_pre_clip:                   0.4592993
+      std_loss:                                 0.0010722978
+  eval-batch-train:
+    avg_avg_episode_length:                     25.069248
+    avg_avg_return:                             0.8991811
+    avg_benchmark_regret:                       0.09361804
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.05127716
+    proxy_corner:
+      avg_avg_return:                           0.049637273
+      avg_reward_per_step:                      0.0015106201
+  eval-batch-orig:
+    avg_avg_episode_length:                     22.62509
+    avg_avg_return:                             0.9330275
+    avg_benchmark_regret:                       0.05721235
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.044700623
+    proxy_corner:
+      avg_avg_return:                           0.9330275
+      avg_reward_per_step:                      0.044700623
+  eval-batch-shift:
+    avg_avg_episode_length:                     25.369234
+    avg_avg_return:                             0.8965316
+    avg_benchmark_regret:                       0.09630586
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.04864502
+    proxy_corner:
+      avg_avg_return:                           0.07224189
+      avg_reward_per_step:                      0.0024032593
+  eval-batch-tree:
+    avg_avg_episode_length:                     80.8703
+    avg_avg_return:                             0.09657614
+    avg_benchmark_regret:                       0.888059
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0065307617
+    proxy_corner:
+      avg_avg_return:                           0.09657614
+      avg_reward_per_step:                      0.0065307617
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        2097152
+    env-step-replay-before:                     1114112
+    env-step-used-before:                       2097152
+    ppo-update-before:                          1280
+    env-step-generate-before:                   983040
+    env-step-all-after:                         2129920
+    env-step-replay-after:                      1114112
+    env-step-used-after:                        2129920
+    ppo-update-after:                           1300
+    env-step-generate-after:                    1015808
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     15.228127
+    avg_avg_return:                             0.8875388
+    avg_reward_per_step:                        0.09979248
+    proxy_corner:
+      avg_avg_return:                           0.03714469
+      avg_reward_per_step:                      0.002105713
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     15.228127
+    avg_avg_return:                             0.8875388
+    avg_reward_per_step:                        0.09979248
+    proxy_corner:
+      avg_avg_return:                           0.03714469
+      avg_reward_per_step:                      0.002105713
+  perf:
+    env_steps_per_second:                       351717.359392611
+    ppo_updates_per_second:                     42.90923958354043
+    eval-batch-train-duration:                  0.5440238285809755
+    eval-batch-orig-duration:                   0.45090318843722343
+    eval-batch-shift-duration:                  0.4795105680823326
+    eval-batch-tree-duration:                   0.49066372588276863
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.05835
+      cheese-corne_dist_wavg:                   7.3042517
+      levels_cheese_is_in_corner_avg:           0.0063476562
+      mouse-cheese_dist_finite_avg:             8.24292
+      mouse-cheese_dist_finite_wavg:            8.0033865
+      mouse-cheese_dist_solvable_avg:           8.24292
+      mouse-cheese_dist_solvable_wavg:          8.003387
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.20709923
+    visit_patterns:
+      avg_first_visit_time:                     8.744141
+      avg_last_visit_time:                      24.127686
+      num_replay_batches:                       34
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001737424
+    avg_actor_approxkl3:                        0.0014953966
+    avg_actor_clipfrac:                         0.077093504
+    avg_actor_loss:                             -0.00017415649
+    avg_advantage:                              0.004598803
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0009835637
+    avg_entropy:                                0.8916214
+    avg_grad_norm_pre_clip:                     1.8862818
+    avg_loss:                                   -0.00057399593
+    max:
+      max_actor_approxkl1:                      0.0062300297
+      max_actor_approxkl3:                      0.0056054443
+      max_actor_clipfrac:                       0.3446045
+      max_actor_loss:                           0.0074380534
+      max_advantage:                            0.14575559
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0019990432
+      max_entropy:                              0.91654646
+      max_grad_norm_pre_clip:                   7.0937424
+      max_loss:                                 0.00691517
+    std:
+      std_actor_approxkl1:                      0.0014350791
+      std_actor_approxkl3:                      0.0013155559
+      std_actor_clipfrac:                       0.08659374
+      std_actor_loss:                           0.0022385814
+      std_advantage:                            0.04397228
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00044243067
+      std_entropy:                              0.015436716
+      std_grad_norm_pre_clip:                   1.7144165
+      std_loss:                                 0.0021995544
+  eval-batch-train:
+    avg_avg_episode_length:                     14.555192
+    avg_avg_return:                             0.9559071
+    avg_benchmark_regret:                       0.036892045
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.10105896
+    proxy_corner:
+      avg_avg_return:                           0.0283042
+      avg_reward_per_step:                      0.0017929077
+  eval-batch-orig:
+    avg_avg_episode_length:                     14.06256
+    avg_avg_return:                             0.9606924
+    avg_benchmark_regret:                       0.029547412
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.078186035
+    proxy_corner:
+      avg_avg_return:                           0.9606924
+      avg_reward_per_step:                      0.078186035
+  eval-batch-shift:
+    avg_avg_episode_length:                     15.055287
+    avg_avg_return:                             0.9547225
+    avg_benchmark_regret:                       0.03811497
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.09322357
+    proxy_corner:
+      avg_avg_return:                           0.0383878
+      avg_reward_per_step:                      0.0021743774
+  eval-batch-tree:
+    avg_avg_episode_length:                     81.79464
+    avg_avg_return:                             0.06417692
+    avg_benchmark_regret:                       0.92045826
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.006690979
+    proxy_corner:
+      avg_avg_return:                           0.06417692
+      avg_reward_per_step:                      0.006690979
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        3145728
+    env-step-replay-before:                     1605632
+    env-step-used-before:                       3145728
+    ppo-update-before:                          1920
+    env-step-generate-before:                   1540096
+    env-step-all-after:                         3178496
+    env-step-replay-after:                      1638400
+    env-step-used-after:                        3178496
+    ppo-update-after:                           1940
+    env-step-generate-after:                    1540096
+  train-replay:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     19.858643
+    avg_avg_return:                             0.85460114
+    avg_reward_per_step:                        0.072143555
+    proxy_corner:
+      avg_avg_return:                           0.04789766
+      avg_reward_per_step:                      0.0024414062
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     19.858643
+    avg_avg_return:                             0.85460114
+    avg_reward_per_step:                        0.072143555
+    proxy_corner:
+      avg_avg_return:                           0.04789766
+      avg_reward_per_step:                      0.0024414062
+  perf:
+    env_steps_per_second:                       400170.69957822043
+    ppo_updates_per_second:                     42.05807532544468
+    eval-batch-train-duration:                  0.4152907710522413
+    eval-batch-orig-duration:                   0.326359286904335
+    eval-batch-shift-duration:                  0.32584942132234573
+    eval-batch-tree-duration:                   0.3550463803112507
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.968018
+      cheese-corne_dist_wavg:                   10.890238
+      levels_cheese_is_in_corner_avg:           0.005859375
+      mouse-cheese_dist_finite_avg:             9.131592
+      mouse-cheese_dist_finite_wavg:            8.083917
+      mouse-cheese_dist_solvable_avg:           9.131592
+      mouse-cheese_dist_solvable_wavg:          8.083917
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.16515262
+    visit_patterns:
+      avg_first_visit_time:                     17.839355
+      avg_last_visit_time:                      40.32495
+      num_replay_batches:                       50
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.003268449
+    avg_actor_approxkl3:                        0.0031281204
+    avg_actor_clipfrac:                         0.16115113
+    avg_actor_loss:                             0.002531331
+    avg_advantage:                              -0.0036568588
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0006088447
+    avg_entropy:                                0.8469615
+    avg_grad_norm_pre_clip:                     4.5925846
+    avg_loss:                                   0.0019887916
+    max:
+      max_actor_approxkl1:                      0.009396882
+      max_actor_approxkl3:                      0.009847899
+      max_actor_clipfrac:                       0.43811035
+      max_actor_loss:                           0.014287708
+      max_advantage:                            0.108937025
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0011348305
+      max_entropy:                              0.89847624
+      max_grad_norm_pre_clip:                   11.9520035
+      max_loss:                                 0.013688702
+    std:
+      std_actor_approxkl1:                      0.0026259464
+      std_actor_approxkl3:                      0.0024002744
+      std_actor_clipfrac:                       0.121889696
+      std_actor_loss:                           0.0037880493
+      std_advantage:                            0.03491499
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00028622002
+      std_entropy:                              0.025768125
+      std_grad_norm_pre_clip:                   3.365522
+      std_loss:                                 0.0037333395
+  eval-batch-train:
+    avg_avg_episode_length:                     12.110701
+    avg_avg_return:                             0.96581995
+    avg_benchmark_regret:                       0.026979184
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.12335968
+    proxy_corner:
+      avg_avg_return:                           0.019823734
+      avg_reward_per_step:                      0.0017700195
+  eval-batch-orig:
+    avg_avg_episode_length:                     12.675526
+    avg_avg_return:                             0.9644146
+    avg_benchmark_regret:                       0.025825247
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.08948517
+    proxy_corner:
+      avg_avg_return:                           0.9644146
+      avg_reward_per_step:                      0.08948517
+  eval-batch-shift:
+    avg_avg_episode_length:                     12.559465
+    avg_avg_return:                             0.96497595
+    avg_benchmark_regret:                       0.027861487
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.112602234
+    proxy_corner:
+      avg_avg_return:                           0.026458103
+      avg_reward_per_step:                      0.0022659302
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.1674
+    avg_avg_return:                             0.057121743
+    avg_benchmark_regret:                       0.9275135
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0065307617
+    proxy_corner:
+      avg_avg_return:                           0.057121743
+      avg_reward_per_step:                      0.0065307617
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        4194304
+    env-step-replay-before:                     2162688
+    env-step-used-before:                       4194304
+    ppo-update-before:                          2560
+    env-step-generate-before:                   2031616
+    env-step-all-after:                         4227072
+    env-step-replay-after:                      2195456
+    env-step-used-after:                        4227072
+    ppo-update-after:                           2580
+    env-step-generate-after:                    2031616
+  train-replay:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     15.932232
+    avg_avg_return:                             0.8792381
+    avg_reward_per_step:                        0.07647705
+    proxy_corner:
+      avg_avg_return:                           0.054719172
+      avg_reward_per_step:                      0.003112793
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     15.932232
+    avg_avg_return:                             0.8792381
+    avg_reward_per_step:                        0.07647705
+    proxy_corner:
+      avg_avg_return:                           0.054719172
+      avg_reward_per_step:                      0.003112793
+  perf:
+    env_steps_per_second:                       355435.8013029417
+    ppo_updates_per_second:                     42.97057075173255
+    eval-batch-train-duration:                  0.5213071871548891
+    eval-batch-orig-duration:                   0.45417709928005934
+    eval-batch-shift-duration:                  0.4691693736240268
+    eval-batch-tree-duration:                   0.4844901682808995
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.867432
+      cheese-corne_dist_wavg:                   1.889983
+      levels_cheese_is_in_corner_avg:           0.0063476562
+      mouse-cheese_dist_finite_avg:             10.065186
+      mouse-cheese_dist_finite_wavg:            6.3888407
+      mouse-cheese_dist_solvable_avg:           10.065186
+      mouse-cheese_dist_solvable_wavg:          6.3888397
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14273581
+    visit_patterns:
+      avg_first_visit_time:                     29.433838
+      avg_last_visit_time:                      57.35791
+      num_replay_batches:                       67
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0022193752
+    avg_actor_approxkl3:                        0.0021205673
+    avg_actor_clipfrac:                         0.11506348
+    avg_actor_loss:                             0.00092761236
+    avg_advantage:                              -0.001569898
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00041989965
+    avg_entropy:                                0.7266666
+    avg_grad_norm_pre_clip:                     4.421083
+    avg_loss:                                   0.00041089562
+    max:
+      max_actor_approxkl1:                      0.005185619
+      max_actor_approxkl3:                      0.0049015107
+      max_actor_clipfrac:                       0.26965332
+      max_actor_loss:                           0.0046605673
+      max_advantage:                            0.12106371
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00047990168
+      max_entropy:                              0.7553063
+      max_grad_norm_pre_clip:                   8.902334
+      max_loss:                                 0.0041270596
+    std:
+      std_actor_approxkl1:                      0.0015178868
+      std_actor_approxkl3:                      0.0014685356
+      std_actor_clipfrac:                       0.088130936
+      std_actor_loss:                           0.00196093
+      std_advantage:                            0.028823959
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.0155996e-05
+      std_entropy:                              0.016750623
+      std_grad_norm_pre_clip:                   2.5181477
+      std_loss:                                 0.001962092
+  eval-batch-train:
+    avg_avg_episode_length:                     9.350157
+    avg_avg_return:                             0.97512555
+    avg_benchmark_regret:                       0.01767362
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.15029907
+    proxy_corner:
+      avg_avg_return:                           0.015400455
+      avg_reward_per_step:                      0.0017776489
+  eval-batch-orig:
+    avg_avg_episode_length:                     11.014711
+    avg_avg_return:                             0.9701046
+    avg_benchmark_regret:                       0.020135274
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.10810089
+    proxy_corner:
+      avg_avg_return:                           0.9701046
+      avg_reward_per_step:                      0.10810089
+  eval-batch-shift:
+    avg_avg_episode_length:                     9.918795
+    avg_avg_return:                             0.97421813
+    avg_benchmark_regret:                       0.01861928
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.13874817
+    proxy_corner:
+      avg_avg_return:                           0.021238893
+      avg_reward_per_step:                      0.0020217896
+  eval-batch-tree:
+    avg_avg_episode_length:                     81.91377
+    avg_avg_return:                             0.05528325
+    avg_benchmark_regret:                       0.9293519
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0088272095
+    proxy_corner:
+      avg_avg_return:                           0.05528325
+      avg_reward_per_step:                      0.0088272095
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        5242880
+    env-step-replay-before:                     2555904
+    env-step-used-before:                       5242880
+    ppo-update-before:                          3200
+    env-step-generate-before:                   2686976
+    env-step-all-after:                         5275648
+    env-step-replay-after:                      2555904
+    env-step-used-after:                        5275648
+    ppo-update-after:                           3220
+    env-step-generate-after:                    2719744
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     9.487837
+    avg_avg_return:                             0.9292957
+    avg_reward_per_step:                        0.14718628
+    proxy_corner:
+      avg_avg_return:                           0.025202204
+      avg_reward_per_step:                      0.0024414062
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     9.487837
+    avg_avg_return:                             0.9292957
+    avg_reward_per_step:                        0.14718628
+    proxy_corner:
+      avg_avg_return:                           0.025202204
+      avg_reward_per_step:                      0.0024414062
+  perf:
+    env_steps_per_second:                       299058.5805214575
+    ppo_updates_per_second:                     42.672200475528314
+    eval-batch-train-duration:                  0.5261323312297463
+    eval-batch-orig-duration:                   0.4759828792884946
+    eval-batch-shift-duration:                  0.44307946041226387
+    eval-batch-tree-duration:                   0.4449170045554638
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.910156
+      cheese-corne_dist_wavg:                   10.8981285
+      levels_cheese_is_in_corner_avg:           0.0087890625
+      mouse-cheese_dist_finite_avg:             10.748291
+      mouse-cheese_dist_finite_wavg:            7.353492
+      mouse-cheese_dist_solvable_avg:           10.748291
+      mouse-cheese_dist_solvable_wavg:          7.3534927
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13088135
+    visit_patterns:
+      avg_first_visit_time:                     39.198242
+      avg_last_visit_time:                      68.2522
+      num_replay_batches:                       78
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014384231
+    avg_actor_approxkl3:                        0.0015142476
+    avg_actor_clipfrac:                         0.08140259
+    avg_actor_loss:                             0.00024780654
+    avg_advantage:                              -0.0013630652
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0005545495
+    avg_entropy:                                0.5178395
+    avg_grad_norm_pre_clip:                     3.7153862
+    avg_loss:                                   7.2417693e-06
+    max:
+      max_actor_approxkl1:                      0.0032342046
+      max_actor_approxkl3:                      0.0027770586
+      max_actor_clipfrac:                       0.16479492
+      max_actor_loss:                           0.0041348394
+      max_advantage:                            0.13457876
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0007883376
+      max_entropy:                              0.5547846
+      max_grad_norm_pre_clip:                   7.281164
+      max_loss:                                 0.0038665088
+    std:
+      std_actor_approxkl1:                      0.0007584556
+      std_actor_approxkl3:                      0.00070764136
+      std_actor_clipfrac:                       0.041161783
+      std_actor_loss:                           0.0011953157
+      std_advantage:                            0.031137923
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          9.071403e-05
+      std_entropy:                              0.019069308
+      std_grad_norm_pre_clip:                   1.9439998
+      std_loss:                                 0.0012023239
+  eval-batch-train:
+    avg_avg_episode_length:                     9.254499
+    avg_avg_return:                             0.9758471
+    avg_benchmark_regret:                       0.016952049
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.15983582
+    proxy_corner:
+      avg_avg_return:                           0.01752265
+      avg_reward_per_step:                      0.0021133423
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.621718
+    avg_avg_return:                             0.97109365
+    avg_benchmark_regret:                       0.01914617
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.11894226
+    proxy_corner:
+      avg_avg_return:                           0.97109365
+      avg_reward_per_step:                      0.11894226
+  eval-batch-shift:
+    avg_avg_episode_length:                     10.015025
+    avg_avg_return:                             0.9724443
+    avg_benchmark_regret:                       0.020393189
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.14814758
+    proxy_corner:
+      avg_avg_return:                           0.027078882
+      avg_reward_per_step:                      0.002281189
+  eval-batch-tree:
+    avg_avg_episode_length:                     81.91395
+    avg_avg_return:                             0.04778102
+    avg_benchmark_regret:                       0.9368542
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.010269165
+    proxy_corner:
+      avg_avg_return:                           0.04778102
+      avg_reward_per_step:                      0.010269165
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        6291456
+    env-step-replay-before:                     3112960
+    env-step-used-before:                       6291456
+    ppo-update-before:                          3840
+    env-step-generate-before:                   3178496
+    env-step-all-after:                         6324224
+    env-step-replay-after:                      3112960
+    env-step-used-after:                        6324224
+    ppo-update-after:                           3860
+    env-step-generate-after:                    3211264
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     9.905418
+    avg_avg_return:                             0.9232092
+    avg_reward_per_step:                        0.14309692
+    proxy_corner:
+      avg_avg_return:                           0.021797936
+      avg_reward_per_step:                      0.002380371
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     9.905418
+    avg_avg_return:                             0.9232092
+    avg_reward_per_step:                        0.14309692
+    proxy_corner:
+      avg_avg_return:                           0.021797936
+      avg_reward_per_step:                      0.002380371
+  perf:
+    env_steps_per_second:                       354899.22219279665
+    ppo_updates_per_second:                     40.71783787260257
+    eval-batch-train-duration:                  0.49709565844386816
+    eval-batch-orig-duration:                   0.43423594534397125
+    eval-batch-shift-duration:                  0.4224974550306797
+    eval-batch-tree-duration:                   0.46210659574717283
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.218994
+      cheese-corne_dist_wavg:                   13.594377
+      levels_cheese_is_in_corner_avg:           0.010253906
+      mouse-cheese_dist_finite_avg:             11.140625
+      mouse-cheese_dist_finite_wavg:            10.994559
+      mouse-cheese_dist_solvable_avg:           11.140625
+      mouse-cheese_dist_solvable_wavg:          10.99456
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13272205
+    visit_patterns:
+      avg_first_visit_time:                     48.861816
+      avg_last_visit_time:                      85.243164
+      num_replay_batches:                       95
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0024686821
+    avg_actor_approxkl3:                        0.0025191724
+    avg_actor_clipfrac:                         0.12216797
+    avg_actor_loss:                             0.0024538978
+    avg_advantage:                              -0.007073868
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.000598967
+    avg_entropy:                                0.47188815
+    avg_grad_norm_pre_clip:                     6.163942
+    avg_loss:                                   0.0022814933
+    max:
+      max_actor_approxkl1:                      0.008839497
+      max_actor_approxkl3:                      0.007529452
+      max_actor_clipfrac:                       0.26245117
+      max_actor_loss:                           0.00749246
+      max_advantage:                            0.11387187
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0010577396
+      max_entropy:                              0.53472936
+      max_grad_norm_pre_clip:                   11.426088
+      max_loss:                                 0.0072697364
+    std:
+      std_actor_approxkl1:                      0.0017383306
+      std_actor_approxkl3:                      0.001618387
+      std_actor_clipfrac:                       0.06679323
+      std_actor_loss:                           0.002003386
+      std_advantage:                            0.034712367
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0002610896
+      std_entropy:                              0.024021912
+      std_grad_norm_pre_clip:                   2.7072625
+      std_loss:                                 0.0019769417
+  eval-batch-train:
+    avg_avg_episode_length:                     9.022773
+    avg_avg_return:                             0.97654
+    avg_benchmark_regret:                       0.016259123
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.16375732
+    proxy_corner:
+      avg_avg_return:                           0.016765604
+      avg_reward_per_step:                      0.0022277832
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.651239
+    avg_avg_return:                             0.9705741
+    avg_benchmark_regret:                       0.019665763
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.12008667
+    proxy_corner:
+      avg_avg_return:                           0.9705741
+      avg_reward_per_step:                      0.12008667
+  eval-batch-shift:
+    avg_avg_episode_length:                     9.034622
+    avg_avg_return:                             0.9762617
+    avg_benchmark_regret:                       0.016575728
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.1579895
+    proxy_corner:
+      avg_avg_return:                           0.020122532
+      avg_reward_per_step:                      0.0022277832
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.19
+    avg_avg_return:                             0.045446105
+    avg_benchmark_regret:                       0.9391891
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.009979248
+    proxy_corner:
+      avg_avg_return:                           0.045446105
+      avg_reward_per_step:                      0.009979248
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        7340032
+    env-step-replay-before:                     3571712
+    env-step-used-before:                       7340032
+    ppo-update-before:                          4480
+    env-step-generate-before:                   3768320
+    env-step-all-after:                         7372800
+    env-step-replay-after:                      3571712
+    env-step-used-after:                        7372800
+    ppo-update-after:                           4500
+    env-step-generate-after:                    3801088
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     8.460297
+    avg_avg_return:                             0.9397604
+    avg_reward_per_step:                        0.17150879
+    proxy_corner:
+      avg_avg_return:                           0.009376423
+      avg_reward_per_step:                      0.0010070801
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     8.460297
+    avg_avg_return:                             0.9397604
+    avg_reward_per_step:                        0.17150879
+    proxy_corner:
+      avg_avg_return:                           0.009376423
+      avg_reward_per_step:                      0.0010070801
+  perf:
+    env_steps_per_second:                       337661.3175031377
+    ppo_updates_per_second:                     42.074006273908076
+    eval-batch-train-duration:                  0.49656920600682497
+    eval-batch-orig-duration:                   0.41893579810857773
+    eval-batch-shift-duration:                  0.4125999938696623
+    eval-batch-tree-duration:                   0.4141693627461791
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.373535
+      cheese-corne_dist_wavg:                   3.7347965
+      levels_cheese_is_in_corner_avg:           0.012451172
+      mouse-cheese_dist_finite_avg:             11.644287
+      mouse-cheese_dist_finite_wavg:            11.049461
+      mouse-cheese_dist_solvable_avg:           11.644287
+      mouse-cheese_dist_solvable_wavg:          11.049461
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12084212
+    visit_patterns:
+      avg_first_visit_time:                     59.193604
+      avg_last_visit_time:                      99.275635
+      num_replay_batches:                       109
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0047752406
+    avg_actor_approxkl3:                        0.004779653
+    avg_actor_clipfrac:                         0.1618042
+    avg_actor_loss:                             0.005564989
+    avg_advantage:                              -0.001360748
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0004644163
+    avg_entropy:                                0.39864418
+    avg_grad_norm_pre_clip:                     6.448733
+    avg_loss:                                   0.0053985524
+    max:
+      max_actor_approxkl1:                      0.012250962
+      max_actor_approxkl3:                      0.011976658
+      max_actor_clipfrac:                       0.29785156
+      max_actor_loss:                           0.014297147
+      max_advantage:                            0.10612708
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00054690684
+      max_entropy:                              0.42428374
+      max_grad_norm_pre_clip:                   9.328656
+      max_loss:                                 0.014117374
+    std:
+      std_actor_approxkl1:                      0.0035954348
+      std_actor_approxkl3:                      0.003263585
+      std_actor_clipfrac:                       0.08055641
+      std_actor_loss:                           0.0041373614
+      std_advantage:                            0.030479938
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.4790606e-05
+      std_entropy:                              0.016025363
+      std_grad_norm_pre_clip:                   2.5135927
+      std_loss:                                 0.0041353083
+  eval-batch-train:
+    avg_avg_episode_length:                     8.332034
+    avg_avg_return:                             0.97885066
+    avg_benchmark_regret:                       0.013948495
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.1762619
+    proxy_corner:
+      avg_avg_return:                           0.01085419
+      avg_reward_per_step:                      0.0017166138
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.712524
+    avg_avg_return:                             0.9702952
+    avg_benchmark_regret:                       0.019944685
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.11644745
+    proxy_corner:
+      avg_avg_return:                           0.9702952
+      avg_reward_per_step:                      0.11644745
+  eval-batch-shift:
+    avg_avg_episode_length:                     8.48747
+    avg_avg_return:                             0.97768867
+    avg_benchmark_regret:                       0.015148798
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.16648102
+    proxy_corner:
+      avg_avg_return:                           0.0109621575
+      avg_reward_per_step:                      0.0017089844
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.49606
+    avg_avg_return:                             0.04398122
+    avg_benchmark_regret:                       0.940654
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.007286072
+    proxy_corner:
+      avg_avg_return:                           0.04398122
+      avg_reward_per_step:                      0.007286072
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        8388608
+    env-step-replay-before:                     4063232
+    env-step-used-before:                       8388608
+    ppo-update-before:                          5120
+    env-step-generate-before:                   4325376
+    env-step-all-after:                         8421376
+    env-step-replay-after:                      4096000
+    env-step-used-after:                        8421376
+    ppo-update-after:                           5140
+    env-step-generate-after:                    4325376
+  train-replay:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     15.394929
+    avg_avg_return:                             0.88122606
+    avg_reward_per_step:                        0.06625366
+    proxy_corner:
+      avg_avg_return:                           0.029095279
+      avg_reward_per_step:                      0.002166748
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     15.394929
+    avg_avg_return:                             0.88122606
+    avg_reward_per_step:                        0.06625366
+    proxy_corner:
+      avg_avg_return:                           0.029095279
+      avg_reward_per_step:                      0.002166748
+  perf:
+    env_steps_per_second:                       244607.66450817377
+    ppo_updates_per_second:                     41.598145147538254
+    eval-batch-train-duration:                  0.537520476616919
+    eval-batch-orig-duration:                   0.49856856651604176
+    eval-batch-shift-duration:                  0.4970418903976679
+    eval-batch-tree-duration:                   0.4962847903370857
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.441895
+      cheese-corne_dist_wavg:                   10.044083
+      levels_cheese_is_in_corner_avg:           0.015136719
+      mouse-cheese_dist_finite_avg:             12.07251
+      mouse-cheese_dist_finite_wavg:            11.104448
+      mouse-cheese_dist_solvable_avg:           12.07251
+      mouse-cheese_dist_solvable_wavg:          11.104449
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.110520616
+    visit_patterns:
+      avg_first_visit_time:                     68.83032
+      avg_last_visit_time:                      115.28467
+      num_replay_batches:                       125
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0030426981
+    avg_actor_approxkl3:                        0.0026330308
+    avg_actor_clipfrac:                         0.13023682
+    avg_actor_loss:                             0.0022785768
+    avg_advantage:                              -0.002259449
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0003603017
+    avg_entropy:                                0.47211334
+    avg_grad_norm_pre_clip:                     7.139093
+    avg_loss:                                   0.0019866142
+    max:
+      max_actor_approxkl1:                      0.007879993
+      max_actor_approxkl3:                      0.007165962
+      max_actor_clipfrac:                       0.26208496
+      max_actor_loss:                           0.008760764
+      max_advantage:                            0.09766638
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00038819114
+      max_entropy:                              0.5149652
+      max_grad_norm_pre_clip:                   12.534905
+      max_loss:                                 0.008461587
+    std:
+      std_actor_approxkl1:                      0.0017447595
+      std_actor_approxkl3:                      0.0015116002
+      std_actor_clipfrac:                       0.05848652
+      std_actor_loss:                           0.002194206
+      std_advantage:                            0.026420936
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.8670344e-05
+      std_entropy:                              0.019894468
+      std_grad_norm_pre_clip:                   2.7507439
+      std_loss:                                 0.0021936127
+  eval-batch-train:
+    avg_avg_episode_length:                     8.107676
+    avg_avg_return:                             0.9790164
+    avg_benchmark_regret:                       0.0137827555
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.179039
+    proxy_corner:
+      avg_avg_return:                           0.0109565435
+      avg_reward_per_step:                      0.0017166138
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.319661
+    avg_avg_return:                             0.972229
+    avg_benchmark_regret:                       0.018010799
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.121170044
+    proxy_corner:
+      avg_avg_return:                           0.972229
+      avg_reward_per_step:                      0.121170044
+  eval-batch-shift:
+    avg_avg_episode_length:                     8.152344
+    avg_avg_return:                             0.9793354
+    avg_benchmark_regret:                       0.013501963
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.172966
+    proxy_corner:
+      avg_avg_return:                           0.014153257
+      avg_reward_per_step:                      0.001914978
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.85029
+    avg_avg_return:                             0.036492314
+    avg_benchmark_regret:                       0.9481429
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.00756073
+    proxy_corner:
+      avg_avg_return:                           0.036492314
+      avg_reward_per_step:                      0.00756073
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        9437184
+    env-step-replay-before:                     4653056
+    env-step-used-before:                       9437184
+    ppo-update-before:                          5760
+    env-step-generate-before:                   4784128
+    env-step-all-after:                         9469952
+    env-step-replay-after:                      4653056
+    env-step-used-after:                        9469952
+    ppo-update-after:                           5780
+    env-step-generate-after:                    4816896
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     9.311058
+    avg_avg_return:                             0.9311974
+    avg_reward_per_step:                        0.162323
+    proxy_corner:
+      avg_avg_return:                           0.019515332
+      avg_reward_per_step:                      0.0017089844
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     9.311058
+    avg_avg_return:                             0.9311974
+    avg_reward_per_step:                        0.162323
+    proxy_corner:
+      avg_avg_return:                           0.019515332
+      avg_reward_per_step:                      0.0017089844
+  perf:
+    env_steps_per_second:                       300602.01092136453
+    ppo_updates_per_second:                     41.987075503548475
+    eval-batch-train-duration:                  0.5189360361546278
+    eval-batch-orig-duration:                   0.47950541507452726
+    eval-batch-shift-duration:                  0.44810990151017904
+    eval-batch-tree-duration:                   0.4307793704792857
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.611816
+      cheese-corne_dist_wavg:                   1.96206
+      levels_cheese_is_in_corner_avg:           0.01586914
+      mouse-cheese_dist_finite_avg:             12.404053
+      mouse-cheese_dist_finite_wavg:            11.125816
+      mouse-cheese_dist_solvable_avg:           12.404053
+      mouse-cheese_dist_solvable_wavg:          11.125817
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11553523
+    visit_patterns:
+      avg_first_visit_time:                     78.369385
+      avg_last_visit_time:                      132.31323
+      num_replay_batches:                       142
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0044495123
+    avg_actor_approxkl3:                        0.004333844
+    avg_actor_clipfrac:                         0.15199585
+    avg_actor_loss:                             0.0047109104
+    avg_advantage:                              -0.006174842
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0004669005
+    avg_entropy:                                0.35130566
+    avg_grad_norm_pre_clip:                     7.7404456
+    avg_loss:                                   0.004593055
+    max:
+      max_actor_approxkl1:                      0.014416249
+      max_actor_approxkl3:                      0.013434471
+      max_actor_clipfrac:                       0.31140137
+      max_actor_loss:                           0.015860036
+      max_advantage:                            0.114818215
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00056921993
+      max_entropy:                              0.3942153
+      max_grad_norm_pre_clip:                   12.02597
+      max_loss:                                 0.015706193
+    std:
+      std_actor_approxkl1:                      0.0035673878
+      std_actor_approxkl3:                      0.003033273
+      std_actor_clipfrac:                       0.07066676
+      std_actor_loss:                           0.0036149279
+      std_advantage:                            0.029572569
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.644494e-05
+      std_entropy:                              0.016627586
+      std_grad_norm_pre_clip:                   2.692537
+      std_loss:                                 0.003611082
+  eval-batch-train:
+    avg_avg_episode_length:                     8.430086
+    avg_avg_return:                             0.97769535
+    avg_benchmark_regret:                       0.015103782
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.1798172
+    proxy_corner:
+      avg_avg_return:                           0.012051861
+      avg_reward_per_step:                      0.001876831
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.399711
+    avg_avg_return:                             0.97168076
+    avg_benchmark_regret:                       0.01855902
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.12359619
+    proxy_corner:
+      avg_avg_return:                           0.97168076
+      avg_reward_per_step:                      0.12359619
+  eval-batch-shift:
+    avg_avg_episode_length:                     8.431026
+    avg_avg_return:                             0.97790074
+    avg_benchmark_regret:                       0.01493671
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.17169952
+    proxy_corner:
+      avg_avg_return:                           0.011384881
+      avg_reward_per_step:                      0.0019454956
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.269516
+    avg_avg_return:                             0.043934092
+    avg_benchmark_regret:                       0.9407011
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.008285522
+    proxy_corner:
+      avg_avg_return:                           0.043934092
+      avg_reward_per_step:                      0.008285522
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        10485760
+    env-step-replay-before:                     5210112
+    env-step-used-before:                       10485760
+    ppo-update-before:                          6400
+    env-step-generate-before:                   5275648
+    env-step-all-after:                         10518528
+    env-step-replay-after:                      5242880
+    env-step-used-after:                        10518528
+    ppo-update-after:                           6420
+    env-step-generate-after:                    5275648
+  train-replay:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     17.158012
+    avg_avg_return:                             0.8698908
+    avg_reward_per_step:                        0.059661865
+    proxy_corner:
+      avg_avg_return:                           0.02741339
+      avg_reward_per_step:                      0.0016784668
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     17.158012
+    avg_avg_return:                             0.8698908
+    avg_reward_per_step:                        0.059661865
+    proxy_corner:
+      avg_avg_return:                           0.02741339
+      avg_reward_per_step:                      0.0016784668
+  perf:
+    env_steps_per_second:                       256156.3312373569
+    ppo_updates_per_second:                     41.74088952423553
+    eval-batch-train-duration:                  0.5363381830975413
+    eval-batch-orig-duration:                   0.46734824031591415
+    eval-batch-shift-duration:                  0.46415638644248247
+    eval-batch-tree-duration:                   0.4810940995812416
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.784424
+      cheese-corne_dist_wavg:                   1.9859986
+      levels_cheese_is_in_corner_avg:           0.016113281
+      mouse-cheese_dist_finite_avg:             12.609375
+      mouse-cheese_dist_finite_wavg:            9.363049
+      mouse-cheese_dist_solvable_avg:           12.609375
+      mouse-cheese_dist_solvable_wavg:          9.3630495
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12858099
+    visit_patterns:
+      avg_first_visit_time:                     87.69531
+      avg_last_visit_time:                      150.34082
+      num_replay_batches:                       160
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0019287507
+    avg_actor_approxkl3:                        0.0021612635
+    avg_actor_clipfrac:                         0.1055603
+    avg_actor_loss:                             0.0018761896
+    avg_advantage:                              -0.0010877416
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00034859296
+    avg_entropy:                                0.40781766
+    avg_grad_norm_pre_clip:                     6.8055606
+    avg_loss:                                   0.0016426685
+    max:
+      max_actor_approxkl1:                      0.006607978
+      max_actor_approxkl3:                      0.0064846883
+      max_actor_clipfrac:                       0.23937988
+      max_actor_loss:                           0.008737746
+      max_advantage:                            0.10572153
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00038716418
+      max_entropy:                              0.43470013
+      max_grad_norm_pre_clip:                   14.588701
+      max_loss:                                 0.008511048
+    std:
+      std_actor_approxkl1:                      0.0016471429
+      std_actor_approxkl3:                      0.0015609004
+      std_actor_clipfrac:                       0.06218301
+      std_actor_loss:                           0.0022386317
+      std_advantage:                            0.026158808
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.4116362e-05
+      std_entropy:                              0.017230127
+      std_grad_norm_pre_clip:                   4.135634
+      std_loss:                                 0.002236103
+  eval-batch-train:
+    avg_avg_episode_length:                     8.689327
+    avg_avg_return:                             0.9771527
+    avg_benchmark_regret:                       0.015646461
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.17881775
+    proxy_corner:
+      avg_avg_return:                           0.010537172
+      avg_reward_per_step:                      0.0017089844
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.933169
+    avg_avg_return:                             0.9695975
+    avg_benchmark_regret:                       0.020642314
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.11715698
+    proxy_corner:
+      avg_avg_return:                           0.9695975
+      avg_reward_per_step:                      0.11715698
+  eval-batch-shift:
+    avg_avg_episode_length:                     8.750086
+    avg_avg_return:                             0.977098
+    avg_benchmark_regret:                       0.015739493
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.17036438
+    proxy_corner:
+      avg_avg_return:                           0.008792313
+      avg_reward_per_step:                      0.0016326904
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.3432
+    avg_avg_return:                             0.04601825
+    avg_benchmark_regret:                       0.93861693
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.007408142
+    proxy_corner:
+      avg_avg_return:                           0.04601825
+      avg_reward_per_step:                      0.007408142
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        11534336
+    env-step-replay-before:                     5767168
+    env-step-used-before:                       11534336
+    ppo-update-before:                          7040
+    env-step-generate-before:                   5767168
+    env-step-all-after:                         11567104
+    env-step-replay-after:                      5767168
+    env-step-used-after:                        11567104
+    ppo-update-after:                           7060
+    env-step-generate-after:                    5799936
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     9.48061
+    avg_avg_return:                             0.9308206
+    avg_reward_per_step:                        0.15853882
+    proxy_corner:
+      avg_avg_return:                           0.01387064
+      avg_reward_per_step:                      0.0029907227
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     9.48061
+    avg_avg_return:                             0.9308206
+    avg_reward_per_step:                        0.15853882
+    proxy_corner:
+      avg_avg_return:                           0.01387064
+      avg_reward_per_step:                      0.0029907227
+  perf:
+    env_steps_per_second:                       247214.6442607859
+    ppo_updates_per_second:                     41.97146837400174
+    eval-batch-train-duration:                  0.6756844073534012
+    eval-batch-orig-duration:                   0.6261077132076025
+    eval-batch-shift-duration:                  0.6090064328163862
+    eval-batch-tree-duration:                   0.6196136707440019
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.871826
+      cheese-corne_dist_wavg:                   8.282659
+      levels_cheese_is_in_corner_avg:           0.016845703
+      mouse-cheese_dist_finite_avg:             12.783691
+      mouse-cheese_dist_finite_wavg:            7.5684013
+      mouse-cheese_dist_solvable_avg:           12.783691
+      mouse-cheese_dist_solvable_wavg:          7.5684013
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12597781
+    visit_patterns:
+      avg_first_visit_time:                     98.30762
+      avg_last_visit_time:                      166.27173
+      num_replay_batches:                       176
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0021549459
+    avg_actor_approxkl3:                        0.0018172025
+    avg_actor_clipfrac:                         0.08071289
+    avg_actor_loss:                             0.001350948
+    avg_advantage:                              0.0013351297
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00042724112
+    avg_entropy:                                0.3202788
+    avg_grad_norm_pre_clip:                     4.151968
+    avg_loss:                                   0.0012442899
+    max:
+      max_actor_approxkl1:                      0.0050199362
+      max_actor_approxkl3:                      0.004074202
+      max_actor_clipfrac:                       0.15002441
+      max_actor_loss:                           0.004625108
+      max_advantage:                            0.12882644
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00050508295
+      max_entropy:                              0.34607327
+      max_grad_norm_pre_clip:                   8.609577
+      max_loss:                                 0.0045366595
+    std:
+      std_actor_approxkl1:                      0.0016088566
+      std_actor_approxkl3:                      0.0011484717
+      std_actor_clipfrac:                       0.041402925
+      std_actor_loss:                           0.001689085
+      std_advantage:                            0.028226858
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.8124977e-05
+      std_entropy:                              0.012778499
+      std_grad_norm_pre_clip:                   1.9130828
+      std_loss:                                 0.0016932697
+  eval-batch-train:
+    avg_avg_episode_length:                     8.923067
+    avg_avg_return:                             0.97702634
+    avg_benchmark_regret:                       0.015772779
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.17512512
+    proxy_corner:
+      avg_avg_return:                           0.011637603
+      avg_reward_per_step:                      0.0017318726
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.455402
+    avg_avg_return:                             0.9717593
+    avg_benchmark_regret:                       0.018480461
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1258316
+    proxy_corner:
+      avg_avg_return:                           0.9717593
+      avg_reward_per_step:                      0.1258316
+  eval-batch-shift:
+    avg_avg_episode_length:                     8.962069
+    avg_avg_return:                             0.97567403
+    avg_benchmark_regret:                       0.017163439
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.1642685
+    proxy_corner:
+      avg_avg_return:                           0.0097202305
+      avg_reward_per_step:                      0.0015106201
+  eval-batch-tree:
+    avg_avg_episode_length:                     81.44501
+    avg_avg_return:                             0.059040505
+    avg_benchmark_regret:                       0.9255947
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.008522034
+    proxy_corner:
+      avg_avg_return:                           0.059040505
+      avg_reward_per_step:                      0.008522034
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        12582912
+    env-step-replay-before:                     6291456
+    env-step-used-before:                       12582912
+    ppo-update-before:                          7680
+    env-step-generate-before:                   6291456
+    env-step-all-after:                         12615680
+    env-step-replay-after:                      6324224
+    env-step-used-after:                        12615680
+    ppo-update-after:                           7700
+    env-step-generate-after:                    6291456
+  train-replay:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     21.01725
+    avg_avg_return:                             0.8351573
+    avg_reward_per_step:                        0.052581787
+    proxy_corner:
+      avg_avg_return:                           0.04508355
+      avg_reward_per_step:                      0.0027770996
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     21.01725
+    avg_avg_return:                             0.8351573
+    avg_reward_per_step:                        0.052581787
+    proxy_corner:
+      avg_avg_return:                           0.04508355
+      avg_reward_per_step:                      0.0027770996
+  perf:
+    env_steps_per_second:                       322615.14374195685
+    ppo_updates_per_second:                     42.191371987929124
+    eval-batch-train-duration:                  0.5343624800443649
+    eval-batch-orig-duration:                   0.4778124736621976
+    eval-batch-shift-duration:                  0.44647820852696896
+    eval-batch-tree-duration:                   0.4872571676969528
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.842285
+      cheese-corne_dist_wavg:                   8.278916
+      levels_cheese_is_in_corner_avg:           0.016357422
+      mouse-cheese_dist_finite_avg:             12.901123
+      mouse-cheese_dist_finite_wavg:            4.8921514
+      mouse-cheese_dist_solvable_avg:           12.901123
+      mouse-cheese_dist_solvable_wavg:          4.8921523
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13016143
+    visit_patterns:
+      avg_first_visit_time:                     107.67139
+      avg_last_visit_time:                      183.3877
+      num_replay_batches:                       193
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0028755923
+    avg_actor_approxkl3:                        0.0027729631
+    avg_actor_clipfrac:                         0.11956177
+    avg_actor_loss:                             0.0011015966
+    avg_advantage:                              -0.005952975
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0007865275
+    avg_entropy:                                0.38631585
+    avg_grad_norm_pre_clip:                     4.7523913
+    avg_loss:                                   0.0011085444
+    max:
+      max_actor_approxkl1:                      0.0059284708
+      max_actor_approxkl3:                      0.0056558824
+      max_actor_clipfrac:                       0.20898438
+      max_actor_loss:                           0.004526629
+      max_advantage:                            0.122485995
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.001584515
+      max_entropy:                              0.41309917
+      max_grad_norm_pre_clip:                   8.624797
+      max_loss:                                 0.004296875
+    std:
+      std_actor_approxkl1:                      0.0015115849
+      std_actor_approxkl3:                      0.001419746
+      std_actor_clipfrac:                       0.049851567
+      std_actor_loss:                           0.0016481611
+      std_advantage:                            0.039423898
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00047125487
+      std_entropy:                              0.013065064
+      std_grad_norm_pre_clip:                   2.090729
+      std_loss:                                 0.0015983524
+  eval-batch-train:
+    avg_avg_episode_length:                     8.856112
+    avg_avg_return:                             0.9772177
+    avg_benchmark_regret:                       0.015581494
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.1785965
+    proxy_corner:
+      avg_avg_return:                           0.012432713
+      avg_reward_per_step:                      0.0018920898
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.322649
+    avg_avg_return:                             0.9717214
+    avg_benchmark_regret:                       0.01851843
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13108063
+    proxy_corner:
+      avg_avg_return:                           0.9717214
+      avg_reward_per_step:                      0.13108063
+  eval-batch-shift:
+    avg_avg_episode_length:                     8.677536
+    avg_avg_return:                             0.9767395
+    avg_benchmark_regret:                       0.016097926
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.16989899
+    proxy_corner:
+      avg_avg_return:                           0.0103926
+      avg_reward_per_step:                      0.0016784668
+  eval-batch-tree:
+    avg_avg_episode_length:                     81.79895
+    avg_avg_return:                             0.056998964
+    avg_benchmark_regret:                       0.9276362
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.008735657
+    proxy_corner:
+      avg_avg_return:                           0.056998964
+      avg_reward_per_step:                      0.008735657
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        13631488
+    env-step-replay-before:                     6782976
+    env-step-used-before:                       13631488
+    ppo-update-before:                          8320
+    env-step-generate-before:                   6848512
+    env-step-all-after:                         13664256
+    env-step-replay-after:                      6782976
+    env-step-used-after:                        13664256
+    ppo-update-after:                           8340
+    env-step-generate-after:                    6881280
+  train-generate:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     8.890427
+    avg_avg_return:                             0.9344133
+    avg_reward_per_step:                        0.15933228
+    proxy_corner:
+      avg_avg_return:                           0.02338437
+      avg_reward_per_step:                      0.0027160645
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     8.890427
+    avg_avg_return:                             0.9344133
+    avg_reward_per_step:                        0.15933228
+    proxy_corner:
+      avg_avg_return:                           0.02338437
+      avg_reward_per_step:                      0.0027160645
+  perf:
+    env_steps_per_second:                       299352.0385780925
+    ppo_updates_per_second:                     42.09231340149669
+    eval-batch-train-duration:                  0.5295706400647759
+    eval-batch-orig-duration:                   0.48233415465801954
+    eval-batch-shift-duration:                  0.44764196407049894
+    eval-batch-tree-duration:                   0.4879027269780636
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.978516
+      cheese-corne_dist_wavg:                   15.487722
+      levels_cheese_is_in_corner_avg:           0.015136719
+      mouse-cheese_dist_finite_avg:             13.030029
+      mouse-cheese_dist_finite_wavg:            9.399413
+      mouse-cheese_dist_solvable_avg:           13.030029
+      mouse-cheese_dist_solvable_wavg:          9.399414
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13127491
+    visit_patterns:
+      avg_first_visit_time:                     119.18262
+      avg_last_visit_time:                      197.36597
+      num_replay_batches:                       207
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.005725468
+    avg_actor_approxkl3:                        0.0052572372
+    avg_actor_clipfrac:                         0.13235474
+    avg_actor_loss:                             0.0056214677
+    avg_advantage:                              -0.0052027777
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.000372672
+    avg_entropy:                                0.26256657
+    avg_grad_norm_pre_clip:                     6.995527
+    avg_loss:                                   0.0055452376
+    max:
+      max_actor_approxkl1:                      0.016110104
+      max_actor_approxkl3:                      0.013518421
+      max_actor_clipfrac:                       0.21374512
+      max_actor_loss:                           0.011761365
+      max_advantage:                            0.11518663
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00043532246
+      max_entropy:                              0.28426105
+      max_grad_norm_pre_clip:                   10.796954
+      max_loss:                                 0.0116936825
+    std:
+      std_actor_approxkl1:                      0.004157096
+      std_actor_approxkl3:                      0.0035739595
+      std_actor_clipfrac:                       0.05323372
+      std_actor_loss:                           0.003469983
+      std_advantage:                            0.02685954
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.7409902e-05
+      std_entropy:                              0.012805185
+      std_grad_norm_pre_clip:                   2.7777202
+      std_loss:                                 0.0034694325
+  eval-batch-train:
+    avg_avg_episode_length:                     8.788179
+    avg_avg_return:                             0.97635484
+    avg_benchmark_regret:                       0.016444284
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.17896271
+    proxy_corner:
+      avg_avg_return:                           0.01347254
+      avg_reward_per_step:                      0.0019607544
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.784967
+    avg_avg_return:                             0.9737534
+    avg_benchmark_regret:                       0.016486418
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13572693
+    proxy_corner:
+      avg_avg_return:                           0.9737534
+      avg_reward_per_step:                      0.13572693
+  eval-batch-shift:
+    avg_avg_episode_length:                     8.805744
+    avg_avg_return:                             0.97656786
+    avg_benchmark_regret:                       0.016269589
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.1721878
+    proxy_corner:
+      avg_avg_return:                           0.013665337
+      avg_reward_per_step:                      0.0020828247
+  eval-batch-tree:
+    avg_avg_episode_length:                     81.511215
+    avg_avg_return:                             0.053276785
+    avg_benchmark_regret:                       0.93135846
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0115356445
+    proxy_corner:
+      avg_avg_return:                           0.053276785
+      avg_reward_per_step:                      0.0115356445
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        14680064
+    env-step-replay-before:                     7405568
+    env-step-used-before:                       14680064
+    ppo-update-before:                          8960
+    env-step-generate-before:                   7274496
+    env-step-all-after:                         14712832
+    env-step-replay-after:                      7438336
+    env-step-used-after:                        14712832
+    ppo-update-after:                           8980
+    env-step-generate-after:                    7274496
+  train-replay:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     18.321056
+    avg_avg_return:                             0.8595043
+    avg_reward_per_step:                        0.055908203
+    proxy_corner:
+      avg_avg_return:                           0.04048314
+      avg_reward_per_step:                      0.0026550293
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     18.321056
+    avg_avg_return:                             0.8595043
+    avg_reward_per_step:                        0.055908203
+    proxy_corner:
+      avg_avg_return:                           0.04048314
+      avg_reward_per_step:                      0.0026550293
+  perf:
+    env_steps_per_second:                       298980.03279799386
+    ppo_updates_per_second:                     42.30111680565629
+    eval-batch-train-duration:                  0.536870701238513
+    eval-batch-orig-duration:                   0.4790507098659873
+    eval-batch-shift-duration:                  0.45101726055145264
+    eval-batch-tree-duration:                   0.47485813312232494
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.962158
+      cheese-corne_dist_wavg:                   10.090436
+      levels_cheese_is_in_corner_avg:           0.015380859
+      mouse-cheese_dist_finite_avg:             13.116455
+      mouse-cheese_dist_finite_wavg:            9.416349
+      mouse-cheese_dist_solvable_avg:           13.116455
+      mouse-cheese_dist_solvable_wavg:          9.416349
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11928457
+    visit_patterns:
+      avg_first_visit_time:                     126.40747
+      avg_last_visit_time:                      217.31494
+      num_replay_batches:                       227
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0020500568
+    avg_actor_approxkl3:                        0.0020077915
+    avg_actor_clipfrac:                         0.10087281
+    avg_actor_loss:                             0.0016623553
+    avg_advantage:                              -0.0130427005
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00047438047
+    avg_entropy:                                0.33584705
+    avg_grad_norm_pre_clip:                     6.024643
+    avg_loss:                                   0.0015636989
+    max:
+      max_actor_approxkl1:                      0.004084859
+      max_actor_approxkl3:                      0.0036114608
+      max_actor_clipfrac:                       0.15576172
+      max_actor_loss:                           0.004938256
+      max_advantage:                            0.09661633
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0010186622
+      max_entropy:                              0.3681169
+      max_grad_norm_pre_clip:                   10.54088
+      max_loss:                                 0.004730652
+    std:
+      std_actor_approxkl1:                      0.0010092681
+      std_actor_approxkl3:                      0.00084392016
+      std_actor_clipfrac:                       0.03500025
+      std_actor_loss:                           0.0018181907
+      std_advantage:                            0.029050648
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0002602894
+      std_entropy:                              0.016039591
+      std_grad_norm_pre_clip:                   2.3528836
+      std_loss:                                 0.0017296899
+  eval-batch-train:
+    avg_avg_episode_length:                     8.4877405
+    avg_avg_return:                             0.976125
+    avg_benchmark_regret:                       0.016674113
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.18478394
+    proxy_corner:
+      avg_avg_return:                           0.011384412
+      avg_reward_per_step:                      0.0018310547
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.971611
+    avg_avg_return:                             0.97320014
+    avg_benchmark_regret:                       0.017039679
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1317749
+    proxy_corner:
+      avg_avg_return:                           0.97320014
+      avg_reward_per_step:                      0.1317749
+  eval-batch-shift:
+    avg_avg_episode_length:                     8.545177
+    avg_avg_return:                             0.97585964
+    avg_benchmark_regret:                       0.01697779
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.17777252
+    proxy_corner:
+      avg_avg_return:                           0.009072828
+      avg_reward_per_step:                      0.0015563965
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.445435
+    avg_avg_return:                             0.046764165
+    avg_benchmark_regret:                       0.93787104
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.008720398
+    proxy_corner:
+      avg_avg_return:                           0.046764165
+      avg_reward_per_step:                      0.008720398
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        15728640
+    env-step-replay-before:                     7766016
+    env-step-used-before:                       15728640
+    ppo-update-before:                          9600
+    env-step-generate-before:                   7962624
+    env-step-all-after:                         15761408
+    env-step-replay-after:                      7766016
+    env-step-used-after:                        15761408
+    ppo-update-after:                           9620
+    env-step-generate-after:                    7995392
+  train-generate:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     9.181728
+    avg_avg_return:                             0.93449837
+    avg_reward_per_step:                        0.17523193
+    proxy_corner:
+      avg_avg_return:                           0.031484663
+      avg_reward_per_step:                      0.0031433105
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     9.181728
+    avg_avg_return:                             0.93449837
+    avg_reward_per_step:                        0.17523193
+    proxy_corner:
+      avg_avg_return:                           0.031484663
+      avg_reward_per_step:                      0.0031433105
+  perf:
+    env_steps_per_second:                       256740.5940423867
+    ppo_updates_per_second:                     42.06223632677516
+    eval-batch-train-duration:                  0.6658940147608519
+    eval-batch-orig-duration:                   0.6111423959955573
+    eval-batch-shift-duration:                  0.6058132722973824
+    eval-batch-tree-duration:                   0.6215416956692934
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    11.169434
+      cheese-corne_dist_wavg:                   5.6168795
+      levels_cheese_is_in_corner_avg:           0.014404297
+      mouse-cheese_dist_finite_avg:             13.090576
+      mouse-cheese_dist_finite_wavg:            11.21724
+      mouse-cheese_dist_solvable_avg:           13.090576
+      mouse-cheese_dist_solvable_wavg:          11.217241
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13224618
+    visit_patterns:
+      avg_first_visit_time:                     139.60156
+      avg_last_visit_time:                      227.22363
+      num_replay_batches:                       237
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0027455585
+    avg_actor_approxkl3:                        0.0025423744
+    avg_actor_clipfrac:                         0.10499268
+    avg_actor_loss:                             0.0023015484
+    avg_advantage:                              0.013637301
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00051698065
+    avg_entropy:                                0.2907294
+    avg_grad_norm_pre_clip:                     6.3323107
+    avg_loss:                                   0.002269309
+    max:
+      max_actor_approxkl1:                      0.006992806
+      max_actor_approxkl3:                      0.008598617
+      max_actor_clipfrac:                       0.23730469
+      max_actor_loss:                           0.012898185
+      max_advantage:                            0.10988587
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0010737462
+      max_entropy:                              0.3372552
+      max_grad_norm_pre_clip:                   13.653421
+      max_loss:                                 0.012751082
+    std:
+      std_actor_approxkl1:                      0.0020262222
+      std_actor_approxkl3:                      0.0021456785
+      std_actor_clipfrac:                       0.06367822
+      std_actor_loss:                           0.0034656478
+      std_advantage:                            0.031869806
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00029920784
+      std_entropy:                              0.020717947
+      std_grad_norm_pre_clip:                   3.395256
+      std_loss:                                 0.00341749
+  eval-batch-train:
+    avg_avg_episode_length:                     9.269306
+    avg_avg_return:                             0.9750446
+    avg_benchmark_regret:                       0.017754605
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.17285156
+    proxy_corner:
+      avg_avg_return:                           0.014421279
+      avg_reward_per_step:                      0.002029419
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.95718
+    avg_avg_return:                             0.97266304
+    avg_benchmark_regret:                       0.01757679
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13745117
+    proxy_corner:
+      avg_avg_return:                           0.97266304
+      avg_reward_per_step:                      0.13745117
+  eval-batch-shift:
+    avg_avg_episode_length:                     8.936403
+    avg_avg_return:                             0.9765308
+    avg_benchmark_regret:                       0.016306687
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.1690216
+    proxy_corner:
+      avg_avg_return:                           0.013217729
+      avg_reward_per_step:                      0.0021591187
+  eval-batch-tree:
+    avg_avg_episode_length:                     81.99818
+    avg_avg_return:                             0.04864011
+    avg_benchmark_regret:                       0.9359951
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.011047363
+    proxy_corner:
+      avg_avg_return:                           0.04864011
+      avg_reward_per_step:                      0.011047363
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        16777216
+    env-step-replay-before:                     8224768
+    env-step-used-before:                       16777216
+    ppo-update-before:                          10240
+    env-step-generate-before:                   8552448
+    env-step-all-after:                         16809984
+    env-step-replay-after:                      8257536
+    env-step-used-after:                        16809984
+    ppo-update-after:                           10260
+    env-step-generate-after:                    8552448
+  train-replay:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     17.553453
+    avg_avg_return:                             0.8648903
+    avg_reward_per_step:                        0.056152344
+    proxy_corner:
+      avg_avg_return:                           0.03274242
+      avg_reward_per_step:                      0.0020141602
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     17.553453
+    avg_avg_return:                             0.8648903
+    avg_reward_per_step:                        0.056152344
+    proxy_corner:
+      avg_avg_return:                           0.03274242
+      avg_reward_per_step:                      0.0020141602
+  perf:
+    env_steps_per_second:                       258920.55238767498
+    ppo_updates_per_second:                     42.21687680300857
+    eval-batch-train-duration:                  0.6667724465951324
+    eval-batch-orig-duration:                   0.5813607573509216
+    eval-batch-shift-duration:                  0.5958411479368806
+    eval-batch-tree-duration:                   0.599687309935689
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.988037
+      cheese-corne_dist_wavg:                   10.106454
+      levels_cheese_is_in_corner_avg:           0.015136719
+      mouse-cheese_dist_finite_avg:             13.152344
+      mouse-cheese_dist_finite_wavg:            17.514132
+      mouse-cheese_dist_solvable_avg:           13.152344
+      mouse-cheese_dist_solvable_wavg:          17.514132
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12673053
+    visit_patterns:
+      avg_first_visit_time:                     149.54028
+      avg_last_visit_time:                      242.27979
+      num_replay_batches:                       252
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0055922954
+    avg_actor_approxkl3:                        0.005483232
+    avg_actor_clipfrac:                         0.1668274
+    avg_actor_loss:                             0.0049898927
+    avg_advantage:                              0.003940544
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0002472542
+    avg_entropy:                                0.32322797
+    avg_grad_norm_pre_clip:                     8.900673
+    avg_loss:                                   0.0047902917
+    max:
+      max_actor_approxkl1:                      0.0137656275
+      max_actor_approxkl3:                      0.0131358765
+      max_actor_clipfrac:                       0.25390625
+      max_actor_loss:                           0.0126344
+      max_advantage:                            0.11203158
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00027721515
+      max_entropy:                              0.3456738
+      max_grad_norm_pre_clip:                   13.157925
+      max_loss:                                 0.012439765
+    std:
+      std_actor_approxkl1:                      0.003668823
+      std_actor_approxkl3:                      0.0032003813
+      std_actor_clipfrac:                       0.05848756
+      std_actor_loss:                           0.0038527106
+      std_advantage:                            0.022298247
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.674282e-05
+      std_entropy:                              0.00948374
+      std_grad_norm_pre_clip:                   2.8227274
+      std_loss:                                 0.0038575658
+  eval-batch-train:
+    avg_avg_episode_length:                     8.157749
+    avg_avg_return:                             0.97926855
+    avg_benchmark_regret:                       0.013530634
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.18702698
+    proxy_corner:
+      avg_avg_return:                           0.012421491
+      avg_reward_per_step:                      0.0019073486
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.762249
+    avg_avg_return:                             0.9738915
+    avg_benchmark_regret:                       0.016348302
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13772583
+    proxy_corner:
+      avg_avg_return:                           0.9738915
+      avg_reward_per_step:                      0.13772583
+  eval-batch-shift:
+    avg_avg_episode_length:                     8.067763
+    avg_avg_return:                             0.9795568
+    avg_benchmark_regret:                       0.013280669
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.18151093
+    proxy_corner:
+      avg_avg_return:                           0.011359625
+      avg_reward_per_step:                      0.0019378662
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.853714
+    avg_avg_return:                             0.036768407
+    avg_benchmark_regret:                       0.94786674
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.007575989
+    proxy_corner:
+      avg_avg_return:                           0.036768407
+      avg_reward_per_step:                      0.007575989
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        17825792
+    env-step-replay-before:                     8912896
+    env-step-used-before:                       17825792
+    ppo-update-before:                          10880
+    env-step-generate-before:                   8912896
+    env-step-all-after:                         17858560
+    env-step-replay-after:                      8945664
+    env-step-used-after:                        17858560
+    ppo-update-after:                           10900
+    env-step-generate-after:                    8912896
+  train-replay:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     17.072758
+    avg_avg_return:                             0.8680906
+    avg_reward_per_step:                        0.060455322
+    proxy_corner:
+      avg_avg_return:                           0.07151189
+      avg_reward_per_step:                      0.0042419434
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     17.072758
+    avg_avg_return:                             0.8680906
+    avg_reward_per_step:                        0.060455322
+    proxy_corner:
+      avg_avg_return:                           0.07151189
+      avg_reward_per_step:                      0.0042419434
+  perf:
+    env_steps_per_second:                       325490.5659969086
+    ppo_updates_per_second:                     40.426346594539766
+    eval-batch-train-duration:                  0.5314514739438891
+    eval-batch-orig-duration:                   0.4716844502836466
+    eval-batch-shift-duration:                  0.4509525643661618
+    eval-batch-tree-duration:                   0.4835870498791337
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.820068
+      cheese-corne_dist_wavg:                   12.784975
+      levels_cheese_is_in_corner_avg:           0.017089844
+      mouse-cheese_dist_finite_avg:             13.230957
+      mouse-cheese_dist_finite_wavg:            10.328146
+      mouse-cheese_dist_solvable_avg:           13.230957
+      mouse-cheese_dist_solvable_wavg:          10.328146
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11778931
+    visit_patterns:
+      avg_first_visit_time:                     158.99341
+      avg_last_visit_time:                      263.17334
+      num_replay_batches:                       273
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0037956813
+    avg_actor_approxkl3:                        0.0038592555
+    avg_actor_clipfrac:                         0.11948853
+    avg_actor_loss:                             0.0041411095
+    avg_advantage:                              0.0020085138
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00065227924
+    avg_entropy:                                0.2808469
+    avg_grad_norm_pre_clip:                     6.1609225
+    avg_loss:                                   0.004186403
+    max:
+      max_actor_approxkl1:                      0.011239922
+      max_actor_approxkl3:                      0.009266689
+      max_actor_clipfrac:                       0.2446289
+      max_actor_loss:                           0.008868443
+      max_advantage:                            0.11487609
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0015348195
+      max_entropy:                              0.30988547
+      max_grad_norm_pre_clip:                   10.32832
+      max_loss:                                 0.008933992
+    std:
+      std_actor_approxkl1:                      0.002578918
+      std_actor_approxkl3:                      0.002396353
+      std_actor_clipfrac:                       0.05220966
+      std_actor_loss:                           0.0025214052
+      std_advantage:                            0.03398873
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00037297086
+      std_entropy:                              0.016141482
+      std_grad_norm_pre_clip:                   2.3165011
+      std_loss:                                 0.0024864015
+  eval-batch-train:
+    avg_avg_episode_length:                     7.974647
+    avg_avg_return:                             0.97955513
+    avg_benchmark_regret:                       0.013244009
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19001007
+    proxy_corner:
+      avg_avg_return:                           0.011716597
+      avg_reward_per_step:                      0.0018463135
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.974964
+    avg_avg_return:                             0.97275233
+    avg_benchmark_regret:                       0.017487496
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13522339
+    proxy_corner:
+      avg_avg_return:                           0.97275233
+      avg_reward_per_step:                      0.13522339
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.9200163
+    avg_avg_return:                             0.979669
+    avg_benchmark_regret:                       0.013168491
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.18502045
+    proxy_corner:
+      avg_avg_return:                           0.010647047
+      avg_reward_per_step:                      0.0017929077
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.86846
+    avg_avg_return:                             0.035420462
+    avg_benchmark_regret:                       0.9492147
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.008300781
+    proxy_corner:
+      avg_avg_return:                           0.035420462
+      avg_reward_per_step:                      0.008300781
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        18874368
+    env-step-replay-before:                     9633792
+    env-step-used-before:                       18874368
+    ppo-update-before:                          11520
+    env-step-generate-before:                   9240576
+    env-step-all-after:                         18907136
+    env-step-replay-after:                      9666560
+    env-step-used-after:                        18907136
+    ppo-update-after:                           11540
+    env-step-generate-after:                    9240576
+  train-replay:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     16.1825
+    avg_avg_return:                             0.8780298
+    avg_reward_per_step:                        0.063934326
+    proxy_corner:
+      avg_avg_return:                           0.04145659
+      avg_reward_per_step:                      0.0025024414
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     16.1825
+    avg_avg_return:                             0.8780298
+    avg_reward_per_step:                        0.063934326
+    proxy_corner:
+      avg_avg_return:                           0.04145659
+      avg_reward_per_step:                      0.0025024414
+  perf:
+    env_steps_per_second:                       327410.6335997087
+    ppo_updates_per_second:                     41.67122413939816
+    eval-batch-train-duration:                  0.5117313107475638
+    eval-batch-orig-duration:                   0.47564816288650036
+    eval-batch-shift-duration:                  0.4474066225811839
+    eval-batch-tree-duration:                   0.4845115663483739
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.649414
+      cheese-corne_dist_wavg:                   12.774936
+      levels_cheese_is_in_corner_avg:           0.017578125
+      mouse-cheese_dist_finite_avg:             13.305908
+      mouse-cheese_dist_finite_wavg:            11.223909
+      mouse-cheese_dist_solvable_avg:           13.305908
+      mouse-cheese_dist_solvable_wavg:          11.223912
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11683669
+    visit_patterns:
+      avg_first_visit_time:                     170.41919
+      avg_last_visit_time:                      285.19873
+      num_replay_batches:                       295
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0031138796
+    avg_actor_approxkl3:                        0.0029437933
+    avg_actor_clipfrac:                         0.087573245
+    avg_actor_loss:                             0.0010683155
+    avg_advantage:                              -0.00087995076
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0004522026
+    avg_entropy:                                0.2699683
+    avg_grad_norm_pre_clip:                     3.692108
+    avg_loss:                                   0.0010244487
+    max:
+      max_actor_approxkl1:                      0.012084001
+      max_actor_approxkl3:                      0.013711171
+      max_actor_clipfrac:                       0.17993164
+      max_actor_loss:                           0.00935031
+      max_advantage:                            0.09235799
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0009334675
+      max_entropy:                              0.29616827
+      max_grad_norm_pre_clip:                   7.5593934
+      max_loss:                                 0.009216004
+    std:
+      std_actor_approxkl1:                      0.0029735593
+      std_actor_approxkl3:                      0.0031177825
+      std_actor_clipfrac:                       0.044363067
+      std_actor_loss:                           0.0034086928
+      std_advantage:                            0.030176055
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00026904856
+      std_entropy:                              0.011249971
+      std_grad_norm_pre_clip:                   2.149501
+      std_loss:                                 0.0033399926
+  eval-batch-train:
+    avg_avg_episode_length:                     8.150538
+    avg_avg_return:                             0.9777987
+    avg_benchmark_regret:                       0.015000419
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.18509674
+    proxy_corner:
+      avg_avg_return:                           0.014649663
+      avg_reward_per_step:                      0.0019454956
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.665815
+    avg_avg_return:                             0.97380567
+    avg_benchmark_regret:                       0.016434144
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13549805
+    proxy_corner:
+      avg_avg_return:                           0.97380567
+      avg_reward_per_step:                      0.13549805
+  eval-batch-shift:
+    avg_avg_episode_length:                     8.247185
+    avg_avg_return:                             0.97751874
+    avg_benchmark_regret:                       0.015318696
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.18115997
+    proxy_corner:
+      avg_avg_return:                           0.018026207
+      avg_reward_per_step:                      0.0020980835
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.1681
+    avg_avg_return:                             0.041600943
+    avg_benchmark_regret:                       0.9430343
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.014175415
+    proxy_corner:
+      avg_avg_return:                           0.041600943
+      avg_reward_per_step:                      0.014175415
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        19922944
+    env-step-replay-before:                     10125312
+    env-step-used-before:                       19922944
+    ppo-update-before:                          12160
+    env-step-generate-before:                   9797632
+    env-step-all-after:                         19955712
+    env-step-replay-after:                      10158080
+    env-step-used-after:                        19955712
+    ppo-update-after:                           12180
+    env-step-generate-after:                    9797632
+  train-replay:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     15.182545
+    avg_avg_return:                             0.88507545
+    avg_reward_per_step:                        0.065979004
+    proxy_corner:
+      avg_avg_return:                           0.034099594
+      avg_reward_per_step:                      0.0020751953
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     15.182545
+    avg_avg_return:                             0.88507545
+    avg_reward_per_step:                        0.065979004
+    proxy_corner:
+      avg_avg_return:                           0.034099594
+      avg_reward_per_step:                      0.0020751953
+  perf:
+    env_steps_per_second:                       311661.483233717
+    ppo_updates_per_second:                     42.20216209955223
+    eval-batch-train-duration:                  0.5253747813403606
+    eval-batch-orig-duration:                   0.4831597190350294
+    eval-batch-shift-duration:                  0.49731444474309683
+    eval-batch-tree-duration:                   0.4782530199736357
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.536865
+      cheese-corne_dist_wavg:                   3.7576563
+      levels_cheese_is_in_corner_avg:           0.018798828
+      mouse-cheese_dist_finite_avg:             13.489258
+      mouse-cheese_dist_finite_wavg:            14.836547
+      mouse-cheese_dist_solvable_avg:           13.489258
+      mouse-cheese_dist_solvable_wavg:          14.836549
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11267251
+    visit_patterns:
+      avg_first_visit_time:                     185.55981
+      avg_last_visit_time:                      300.21582
+      num_replay_batches:                       310
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001966689
+    avg_actor_approxkl3:                        0.0019380084
+    avg_actor_clipfrac:                         0.081530765
+    avg_actor_loss:                             0.0017457858
+    avg_advantage:                              -0.011036521
+    avg_critic_clipfrac:                        0.06680298
+    avg_critic_loss:                            0.0015188793
+    avg_entropy:                                0.248282
+    avg_grad_norm_pre_clip:                     5.8037477
+    avg_loss:                                   0.0022569436
+    max:
+      max_actor_approxkl1:                      0.006008498
+      max_actor_approxkl3:                      0.004469067
+      max_actor_clipfrac:                       0.16845703
+      max_actor_loss:                           0.0065542366
+      max_advantage:                            0.12050289
+      max_critic_clipfrac:                      0.71606445
+      max_critic_loss:                          0.0075938245
+      max_entropy:                              0.2685389
+      max_grad_norm_pre_clip:                   10.989037
+      max_loss:                                 0.009212605
+    std:
+      std_actor_approxkl1:                      0.0012601884
+      std_actor_approxkl3:                      0.0011933729
+      std_actor_clipfrac:                       0.0425852
+      std_actor_loss:                           0.0020675438
+      std_advantage:                            0.024091462
+      std_critic_clipfrac:                      0.20047747
+      std_critic_loss:                          0.0021302116
+      std_entropy:                              0.009479283
+      std_grad_norm_pre_clip:                   2.9151134
+      std_loss:                                 0.002616795
+  eval-batch-train:
+    avg_avg_episode_length:                     7.5953064
+    avg_avg_return:                             0.9804879
+    avg_benchmark_regret:                       0.012311257
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19159698
+    proxy_corner:
+      avg_avg_return:                           0.008529242
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.6437435
+    avg_avg_return:                             0.97423685
+    avg_benchmark_regret:                       0.016003048
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13652039
+    proxy_corner:
+      avg_avg_return:                           0.97423685
+      avg_reward_per_step:                      0.13652039
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4717255
+    avg_avg_return:                             0.98081076
+    avg_benchmark_regret:                       0.012026692
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.1892395
+    proxy_corner:
+      avg_avg_return:                           0.010715855
+      avg_reward_per_step:                      0.0017547607
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.839935
+    avg_avg_return:                             0.036335602
+    avg_benchmark_regret:                       0.9482996
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.009979248
+    proxy_corner:
+      avg_avg_return:                           0.036335602
+      avg_reward_per_step:                      0.009979248
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        20971520
+    env-step-replay-before:                     10682368
+    env-step-used-before:                       20971520
+    ppo-update-before:                          12800
+    env-step-generate-before:                   10289152
+    env-step-all-after:                         21004288
+    env-step-replay-after:                      10682368
+    env-step-used-after:                        21004288
+    ppo-update-after:                           12820
+    env-step-generate-after:                    10321920
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.6561165
+    avg_avg_return:                             0.9463751
+    avg_reward_per_step:                        0.18765259
+    proxy_corner:
+      avg_avg_return:                           0.025913384
+      avg_reward_per_step:                      0.003540039
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.6561165
+    avg_avg_return:                             0.9463751
+    avg_reward_per_step:                        0.18765259
+    proxy_corner:
+      avg_avg_return:                           0.025913384
+      avg_reward_per_step:                      0.003540039
+  perf:
+    env_steps_per_second:                       354132.29018970224
+    ppo_updates_per_second:                     44.44391571706301
+    eval-batch-train-duration:                  0.48158077895641327
+    eval-batch-orig-duration:                   0.41988038178533316
+    eval-batch-shift-duration:                  0.41676315385848284
+    eval-batch-tree-duration:                   0.4482523426413536
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.490723
+      cheese-corne_dist_wavg:                   5.543587
+      levels_cheese_is_in_corner_avg:           0.020019531
+      mouse-cheese_dist_finite_avg:             13.6936035
+      mouse-cheese_dist_finite_wavg:            7.6724615
+      mouse-cheese_dist_solvable_avg:           13.6936035
+      mouse-cheese_dist_solvable_wavg:          7.6724625
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.10547203
+    visit_patterns:
+      avg_first_visit_time:                     199.98853
+      avg_last_visit_time:                      316.34204
+      num_replay_batches:                       326
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014585446
+    avg_actor_approxkl3:                        0.0017074961
+    avg_actor_clipfrac:                         0.058441162
+    avg_actor_loss:                             0.0015402305
+    avg_advantage:                              -0.01416271
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0003760354
+    avg_entropy:                                0.19164424
+    avg_grad_norm_pre_clip:                     3.9242923
+    avg_loss:                                   0.0015366039
+    max:
+      max_actor_approxkl1:                      0.0053779366
+      max_actor_approxkl3:                      0.0053790645
+      max_actor_clipfrac:                       0.13110352
+      max_actor_loss:                           0.006633694
+      max_advantage:                            0.095492005
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00048291127
+      max_entropy:                              0.22235608
+      max_grad_norm_pre_clip:                   8.242585
+      max_loss:                                 0.006631514
+    std:
+      std_actor_approxkl1:                      0.0015024253
+      std_actor_approxkl3:                      0.0014393646
+      std_actor_clipfrac:                       0.036697485
+      std_actor_loss:                           0.0020834894
+      std_advantage:                            0.025950305
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.098833e-05
+      std_entropy:                              0.0143715795
+      std_grad_norm_pre_clip:                   1.9694443
+      std_loss:                                 0.0020817276
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4852505
+    avg_avg_return:                             0.98073745
+    avg_benchmark_regret:                       0.01206164
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19530487
+    proxy_corner:
+      avg_avg_return:                           0.008771102
+      avg_reward_per_step:                      0.0016174316
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.620703
+    avg_avg_return:                             0.9741409
+    avg_benchmark_regret:                       0.016098907
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1379242
+    proxy_corner:
+      avg_avg_return:                           0.9741409
+      avg_reward_per_step:                      0.1379242
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4086475
+    avg_avg_return:                             0.98147875
+    avg_benchmark_regret:                       0.011358669
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19116211
+    proxy_corner:
+      avg_avg_return:                           0.009329265
+      avg_reward_per_step:                      0.0016403198
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.30221
+    avg_avg_return:                             0.04126348
+    avg_benchmark_regret:                       0.9433717
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.01260376
+    proxy_corner:
+      avg_avg_return:                           0.04126348
+      avg_reward_per_step:                      0.01260376
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        22020096
+    env-step-replay-before:                     11141120
+    env-step-used-before:                       22020096
+    ppo-update-before:                          13440
+    env-step-generate-before:                   10878976
+    env-step-all-after:                         22052864
+    env-step-replay-after:                      11173888
+    env-step-used-after:                        22052864
+    ppo-update-after:                           13460
+    env-step-generate-after:                    10878976
+  train-replay:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     15.462639
+    avg_avg_return:                             0.8792526
+    avg_reward_per_step:                        0.063079834
+    proxy_corner:
+      avg_avg_return:                           0.043501507
+      avg_reward_per_step:                      0.00289917
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     15.462639
+    avg_avg_return:                             0.8792526
+    avg_reward_per_step:                        0.063079834
+    proxy_corner:
+      avg_avg_return:                           0.043501507
+      avg_reward_per_step:                      0.00289917
+  perf:
+    env_steps_per_second:                       323525.28720824665
+    ppo_updates_per_second:                     42.103891007926165
+    eval-batch-train-duration:                  0.5211493093520403
+    eval-batch-orig-duration:                   0.47605955600738525
+    eval-batch-shift-duration:                  0.4490069216117263
+    eval-batch-tree-duration:                   0.4866259526461363
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.486084
+      cheese-corne_dist_wavg:                   4.658023
+      levels_cheese_is_in_corner_avg:           0.021484375
+      mouse-cheese_dist_finite_avg:             13.854248
+      mouse-cheese_dist_finite_wavg:            13.083725
+      mouse-cheese_dist_solvable_avg:           13.854248
+      mouse-cheese_dist_solvable_wavg:          13.083728
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.108347565
+    visit_patterns:
+      avg_first_visit_time:                     213.1289
+      avg_last_visit_time:                      331.19946
+      num_replay_batches:                       341
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0018133427
+    avg_actor_approxkl3:                        0.0014243753
+    avg_actor_clipfrac:                         0.065057375
+    avg_actor_loss:                             0.0008763779
+    avg_advantage:                              -0.0037128716
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00027457005
+    avg_entropy:                                0.2511672
+    avg_grad_norm_pre_clip:                     4.808613
+    avg_loss:                                   0.0007624958
+    max:
+      max_actor_approxkl1:                      0.0038132728
+      max_actor_approxkl3:                      0.0027265209
+      max_actor_clipfrac:                       0.11450195
+      max_actor_loss:                           0.0031027612
+      max_advantage:                            0.100207865
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00031892603
+      max_entropy:                              0.277525
+      max_grad_norm_pre_clip:                   9.611718
+      max_loss:                                 0.0029882826
+    std:
+      std_actor_approxkl1:                      0.0010798043
+      std_actor_approxkl3:                      0.0006829378
+      std_actor_clipfrac:                       0.03123265
+      std_actor_loss:                           0.0011800954
+      std_advantage:                            0.023218049
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.5918762e-05
+      std_entropy:                              0.01303819
+      std_grad_norm_pre_clip:                   2.22873
+      std_loss:                                 0.0011815449
+  eval-batch-train:
+    avg_avg_episode_length:                     7.67326
+    avg_avg_return:                             0.98043126
+    avg_benchmark_regret:                       0.012367873
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19139862
+    proxy_corner:
+      avg_avg_return:                           0.007950392
+      avg_reward_per_step:                      0.0014877319
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.758976
+    avg_avg_return:                             0.97355795
+    avg_benchmark_regret:                       0.016681898
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.133255
+    proxy_corner:
+      avg_avg_return:                           0.97355795
+      avg_reward_per_step:                      0.133255
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.6550226
+    avg_avg_return:                             0.981175
+    avg_benchmark_regret:                       0.01166244
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.18802643
+    proxy_corner:
+      avg_avg_return:                           0.004683033
+      avg_reward_per_step:                      0.0010375977
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.66075
+    avg_avg_return:                             0.037407875
+    avg_benchmark_regret:                       0.9472273
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.010169983
+    proxy_corner:
+      avg_avg_return:                           0.037407875
+      avg_reward_per_step:                      0.010169983
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        23068672
+    env-step-replay-before:                     11665408
+    env-step-used-before:                       23068672
+    ppo-update-before:                          14080
+    env-step-generate-before:                   11403264
+    env-step-all-after:                         23101440
+    env-step-replay-after:                      11698176
+    env-step-used-after:                        23101440
+    ppo-update-after:                           14100
+    env-step-generate-after:                    11403264
+  train-replay:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     15.077475
+    avg_avg_return:                             0.88610464
+    avg_reward_per_step:                        0.06655884
+    proxy_corner:
+      avg_avg_return:                           0.053229682
+      avg_reward_per_step:                      0.0035095215
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     15.077475
+    avg_avg_return:                             0.88610464
+    avg_reward_per_step:                        0.06655884
+    proxy_corner:
+      avg_avg_return:                           0.053229682
+      avg_reward_per_step:                      0.0035095215
+  perf:
+    env_steps_per_second:                       300264.56514443684
+    ppo_updates_per_second:                     42.91957963921848
+    eval-batch-train-duration:                  0.5299907755106688
+    eval-batch-orig-duration:                   0.47879120148718357
+    eval-batch-shift-duration:                  0.449161765165627
+    eval-batch-tree-duration:                   0.4897224772721529
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.447266
+      cheese-corne_dist_wavg:                   4.6500797
+      levels_cheese_is_in_corner_avg:           0.021728516
+      mouse-cheese_dist_finite_avg:             13.940186
+      mouse-cheese_dist_finite_wavg:            8.591243
+      mouse-cheese_dist_solvable_avg:           13.940186
+      mouse-cheese_dist_solvable_wavg:          8.591245
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.107137986
+    visit_patterns:
+      avg_first_visit_time:                     224.61792
+      avg_last_visit_time:                      347.29004
+      num_replay_batches:                       357
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0081897965
+    avg_actor_approxkl3:                        0.007897737
+    avg_actor_clipfrac:                         0.15009765
+    avg_actor_loss:                             0.009513527
+    avg_advantage:                              0.007959468
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00026167487
+    avg_entropy:                                0.22674756
+    avg_grad_norm_pre_clip:                     9.204797
+    avg_loss:                                   0.009417616
+    max:
+      max_actor_approxkl1:                      0.01803232
+      max_actor_approxkl3:                      0.01837327
+      max_actor_clipfrac:                       0.24853516
+      max_actor_loss:                           0.02209565
+      max_advantage:                            0.10283917
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0003044834
+      max_entropy:                              0.24936274
+      max_grad_norm_pre_clip:                   15.230183
+      max_loss:                                 0.02198335
+    std:
+      std_actor_approxkl1:                      0.0048993058
+      std_actor_approxkl3:                      0.004842362
+      std_actor_clipfrac:                       0.06095697
+      std_actor_loss:                           0.0062285913
+      std_advantage:                            0.022135496
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.8681054e-05
+      std_entropy:                              0.013243236
+      std_grad_norm_pre_clip:                   3.8040657
+      std_loss:                                 0.0062216567
+  eval-batch-train:
+    avg_avg_episode_length:                     7.513747
+    avg_avg_return:                             0.98129606
+    avg_benchmark_regret:                       0.011503124
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19332123
+    proxy_corner:
+      avg_avg_return:                           0.007941527
+      avg_reward_per_step:                      0.0014648438
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.753206
+    avg_avg_return:                             0.97340095
+    avg_benchmark_regret:                       0.01683889
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13380432
+    proxy_corner:
+      avg_avg_return:                           0.97340095
+      avg_reward_per_step:                      0.13380432
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.488877
+    avg_avg_return:                             0.98101264
+    avg_benchmark_regret:                       0.011824829
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.1894989
+    proxy_corner:
+      avg_avg_return:                           0.006202794
+      avg_reward_per_step:                      0.0012359619
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.94594
+    avg_avg_return:                             0.036538437
+    avg_benchmark_regret:                       0.94809675
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.008735657
+    proxy_corner:
+      avg_avg_return:                           0.036538437
+      avg_reward_per_step:                      0.008735657
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        24117248
+    env-step-replay-before:                     12320768
+    env-step-used-before:                       24117248
+    ppo-update-before:                          14720
+    env-step-generate-before:                   11796480
+    env-step-all-after:                         24150016
+    env-step-replay-after:                      12353536
+    env-step-used-after:                        24150016
+    ppo-update-after:                           14740
+    env-step-generate-after:                    11796480
+  train-replay:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     15.222261
+    avg_avg_return:                             0.88454485
+    avg_reward_per_step:                        0.06365967
+    proxy_corner:
+      avg_avg_return:                           0.03156022
+      avg_reward_per_step:                      0.0020446777
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     15.222261
+    avg_avg_return:                             0.88454485
+    avg_reward_per_step:                        0.06365967
+    proxy_corner:
+      avg_avg_return:                           0.03156022
+      avg_reward_per_step:                      0.0020446777
+  perf:
+    env_steps_per_second:                       303736.0703110489
+    ppo_updates_per_second:                     43.99082934480281
+    eval-batch-train-duration:                  0.542927042581141
+    eval-batch-orig-duration:                   0.4516233494505286
+    eval-batch-shift-duration:                  0.4721462270244956
+    eval-batch-tree-duration:                   0.4865130567923188
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.437988
+      cheese-corne_dist_wavg:                   4.6404247
+      levels_cheese_is_in_corner_avg:           0.022949219
+      mouse-cheese_dist_finite_avg:             14.052002
+      mouse-cheese_dist_finite_wavg:            13.101789
+      mouse-cheese_dist_solvable_avg:           14.052002
+      mouse-cheese_dist_solvable_wavg:          13.101791
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.10549174
+    visit_patterns:
+      avg_first_visit_time:                     232.5503
+      avg_last_visit_time:                      367.2351
+      num_replay_batches:                       377
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.003612027
+    avg_actor_approxkl3:                        0.0038624175
+    avg_actor_clipfrac:                         0.10800171
+    avg_actor_loss:                             0.004906025
+    avg_advantage:                              -0.001959698
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00026173596
+    avg_entropy:                                0.23087318
+    avg_grad_norm_pre_clip:                     7.4314003
+    avg_loss:                                   0.00480602
+    max:
+      max_actor_approxkl1:                      0.012990713
+      max_actor_approxkl3:                      0.011867324
+      max_actor_clipfrac:                       0.18383789
+      max_actor_loss:                           0.014824996
+      max_advantage:                            0.08155584
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00031364168
+      max_entropy:                              0.24867587
+      max_grad_norm_pre_clip:                   13.085979
+      max_loss:                                 0.014715988
+    std:
+      std_actor_approxkl1:                      0.0028738612
+      std_actor_approxkl3:                      0.0027927815
+      std_actor_clipfrac:                       0.044108894
+      std_actor_loss:                           0.004446273
+      std_advantage:                            0.02280745
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.589107e-05
+      std_entropy:                              0.008509517
+      std_grad_norm_pre_clip:                   2.9584944
+      std_loss:                                 0.004446194
+  eval-batch-train:
+    avg_avg_episode_length:                     7.63395
+    avg_avg_return:                             0.98083526
+    avg_benchmark_regret:                       0.011963866
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19206238
+    proxy_corner:
+      avg_avg_return:                           0.007885975
+      avg_reward_per_step:                      0.0014801025
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.793915
+    avg_avg_return:                             0.97347075
+    avg_benchmark_regret:                       0.016769059
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13231659
+    proxy_corner:
+      avg_avg_return:                           0.97347075
+      avg_reward_per_step:                      0.13231659
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.545778
+    avg_avg_return:                             0.9809223
+    avg_benchmark_regret:                       0.011915143
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.18677521
+    proxy_corner:
+      avg_avg_return:                           0.005078764
+      avg_reward_per_step:                      0.0010375977
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.96983
+    avg_avg_return:                             0.03314314
+    avg_benchmark_regret:                       0.9514921
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.008850098
+    proxy_corner:
+      avg_avg_return:                           0.03314314
+      avg_reward_per_step:                      0.008850098
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        25165824
+    env-step-replay-before:                     13041664
+    env-step-used-before:                       25165824
+    ppo-update-before:                          15360
+    env-step-generate-before:                   12124160
+    env-step-all-after:                         25198592
+    env-step-replay-after:                      13041664
+    env-step-used-after:                        25198592
+    ppo-update-after:                           15380
+    env-step-generate-after:                    12156928
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.3019867
+    avg_avg_return:                             0.95139706
+    avg_reward_per_step:                        0.19732666
+    proxy_corner:
+      avg_avg_return:                           0.008599134
+      avg_reward_per_step:                      0.0007019043
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.3019867
+    avg_avg_return:                             0.95139706
+    avg_reward_per_step:                        0.19732666
+    proxy_corner:
+      avg_avg_return:                           0.008599134
+      avg_reward_per_step:                      0.0007019043
+  perf:
+    env_steps_per_second:                       353595.1874663489
+    ppo_updates_per_second:                     41.687265907355275
+    eval-batch-train-duration:                  0.5151966912671924
+    eval-batch-orig-duration:                   0.4682615092024207
+    eval-batch-shift-duration:                  0.47935581859201193
+    eval-batch-tree-duration:                   0.4866466782987118
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.337891
+      cheese-corne_dist_wavg:                   10.03562
+      levels_cheese_is_in_corner_avg:           0.024902344
+      mouse-cheese_dist_finite_avg:             14.09375
+      mouse-cheese_dist_finite_wavg:            9.513411
+      mouse-cheese_dist_solvable_avg:           14.09375
+      mouse-cheese_dist_solvable_wavg:          9.5134115
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1082927
+    visit_patterns:
+      avg_first_visit_time:                     243.20654
+      avg_last_visit_time:                      388.31714
+      num_replay_batches:                       398
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002035286
+    avg_actor_approxkl3:                        0.00202365
+    avg_actor_clipfrac:                         0.06447144
+    avg_actor_loss:                             0.0022252523
+    avg_advantage:                              0.0005569464
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00033950104
+    avg_entropy:                                0.18019299
+    avg_grad_norm_pre_clip:                     5.1452675
+    avg_loss:                                   0.0022148096
+    max:
+      max_actor_approxkl1:                      0.00838594
+      max_actor_approxkl3:                      0.009898944
+      max_actor_clipfrac:                       0.16113281
+      max_actor_loss:                           0.013051489
+      max_advantage:                            0.108632505
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00039435312
+      max_entropy:                              0.20997185
+      max_grad_norm_pre_clip:                   11.45971
+      max_loss:                                 0.013045631
+    std:
+      std_actor_approxkl1:                      0.0019479664
+      std_actor_approxkl3:                      0.002199739
+      std_actor_clipfrac:                       0.035080027
+      std_actor_loss:                           0.0030492097
+      std_advantage:                            0.025783418
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.7520216e-05
+      std_entropy:                              0.012725098
+      std_grad_norm_pre_clip:                   2.7053394
+      std_loss:                                 0.0030486758
+  eval-batch-train:
+    avg_avg_episode_length:                     7.642956
+    avg_avg_return:                             0.9806438
+    avg_benchmark_regret:                       0.01215535
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.1958313
+    proxy_corner:
+      avg_avg_return:                           0.007962114
+      avg_reward_per_step:                      0.0014572144
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.833731
+    avg_avg_return:                             0.9728538
+    avg_benchmark_regret:                       0.017386071
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1322403
+    proxy_corner:
+      avg_avg_return:                           0.9728538
+      avg_reward_per_step:                      0.1322403
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.5955043
+    avg_avg_return:                             0.98097175
+    avg_benchmark_regret:                       0.011865682
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.18909454
+    proxy_corner:
+      avg_avg_return:                           0.005352839
+      avg_reward_per_step:                      0.0011138916
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.672
+    avg_avg_return:                             0.037607536
+    avg_benchmark_regret:                       0.9470277
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.008468628
+    proxy_corner:
+      avg_avg_return:                           0.037607536
+      avg_reward_per_step:                      0.008468628
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        26214400
+    env-step-replay-before:                     13402112
+    env-step-used-before:                       26214400
+    ppo-update-before:                          16000
+    env-step-generate-before:                   12812288
+    env-step-all-after:                         26247168
+    env-step-replay-after:                      13434880
+    env-step-used-after:                        26247168
+    ppo-update-after:                           16020
+    env-step-generate-after:                    12812288
+  train-replay:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     17.344145
+    avg_avg_return:                             0.86808914
+    avg_reward_per_step:                        0.060699463
+    proxy_corner:
+      avg_avg_return:                           0.036484085
+      avg_reward_per_step:                      0.002319336
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     17.344145
+    avg_avg_return:                             0.86808914
+    avg_reward_per_step:                        0.060699463
+    proxy_corner:
+      avg_avg_return:                           0.036484085
+      avg_reward_per_step:                      0.002319336
+  perf:
+    env_steps_per_second:                       357078.6401552383
+    ppo_updates_per_second:                     41.654912078225166
+    eval-batch-train-duration:                  0.510289590805769
+    eval-batch-orig-duration:                   0.4743003351613879
+    eval-batch-shift-duration:                  0.4454435929656029
+    eval-batch-tree-duration:                   0.4694236842915416
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.243408
+      cheese-corne_dist_wavg:                   9.1287365
+      levels_cheese_is_in_corner_avg:           0.02758789
+      mouse-cheese_dist_finite_avg:             14.194336
+      mouse-cheese_dist_finite_wavg:            16.708744
+      mouse-cheese_dist_solvable_avg:           14.194336
+      mouse-cheese_dist_solvable_wavg:          16.708746
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.112611786
+    visit_patterns:
+      avg_first_visit_time:                     256.60034
+      avg_last_visit_time:                      400.2942
+      num_replay_batches:                       410
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0055995914
+    avg_actor_approxkl3:                        0.005690304
+    avg_actor_clipfrac:                         0.11221924
+    avg_actor_loss:                             0.0065094107
+    avg_advantage:                              -0.0015220136
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0004056058
+    avg_entropy:                                0.23457693
+    avg_grad_norm_pre_clip:                     6.7512674
+    avg_loss:                                   0.0064776368
+    max:
+      max_actor_approxkl1:                      0.018122232
+      max_actor_approxkl3:                      0.01822162
+      max_actor_clipfrac:                       0.1986084
+      max_actor_loss:                           0.016896918
+      max_advantage:                            0.106832445
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0009310087
+      max_entropy:                              0.2834271
+      max_grad_norm_pre_clip:                   10.100853
+      max_loss:                                 0.017093802
+    std:
+      std_actor_approxkl1:                      0.0042645577
+      std_actor_approxkl3:                      0.004312265
+      std_actor_clipfrac:                       0.047570724
+      std_actor_loss:                           0.0045474954
+      std_advantage:                            0.028311996
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00028440793
+      std_entropy:                              0.01842913
+      std_grad_norm_pre_clip:                   2.19945
+      std_loss:                                 0.004564301
+  eval-batch-train:
+    avg_avg_episode_length:                     7.9197516
+    avg_avg_return:                             0.9784466
+    avg_benchmark_regret:                       0.014352569
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19589996
+    proxy_corner:
+      avg_avg_return:                           0.009329669
+      avg_reward_per_step:                      0.0016937256
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.239547
+    avg_avg_return:                             0.971213
+    avg_benchmark_regret:                       0.019026868
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.12778473
+    proxy_corner:
+      avg_avg_return:                           0.971213
+      avg_reward_per_step:                      0.12778473
+  eval-batch-shift:
+    avg_avg_episode_length:                     8.279291
+    avg_avg_return:                             0.9782179
+    avg_benchmark_regret:                       0.0146195805
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.18817902
+    proxy_corner:
+      avg_avg_return:                           0.005813708
+      avg_reward_per_step:                      0.001159668
+  eval-batch-tree:
+    avg_avg_episode_length:                     83.17414
+    avg_avg_return:                             0.033571664
+    avg_benchmark_regret:                       0.9510635
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.007545471
+    proxy_corner:
+      avg_avg_return:                           0.033571664
+      avg_reward_per_step:                      0.007545471
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        27262976
+    env-step-replay-before:                     14024704
+    env-step-used-before:                       27262976
+    ppo-update-before:                          16640
+    env-step-generate-before:                   13238272
+    env-step-all-after:                         27295744
+    env-step-replay-after:                      14057472
+    env-step-used-after:                        27295744
+    ppo-update-after:                           16660
+    env-step-generate-after:                    13238272
+  train-replay:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     15.605448
+    avg_avg_return:                             0.8804437
+    avg_reward_per_step:                        0.06222534
+    proxy_corner:
+      avg_avg_return:                           0.061895363
+      avg_reward_per_step:                      0.0037231445
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     15.605448
+    avg_avg_return:                             0.8804437
+    avg_reward_per_step:                        0.06222534
+    proxy_corner:
+      avg_avg_return:                           0.061895363
+      avg_reward_per_step:                      0.0037231445
+  perf:
+    env_steps_per_second:                       311013.93030338513
+    ppo_updates_per_second:                     41.26877852538669
+    eval-batch-train-duration:                  0.5211605560034513
+    eval-batch-orig-duration:                   0.4878499861806631
+    eval-batch-shift-duration:                  0.4843485001474619
+    eval-batch-tree-duration:                   0.48320755921304226
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.264404
+      cheese-corne_dist_wavg:                   14.517762
+      levels_cheese_is_in_corner_avg:           0.028564453
+      mouse-cheese_dist_finite_avg:             14.222168
+      mouse-cheese_dist_finite_wavg:            14.022561
+      mouse-cheese_dist_solvable_avg:           14.222168
+      mouse-cheese_dist_solvable_wavg:          14.022563
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11077675
+    visit_patterns:
+      avg_first_visit_time:                     267.64575
+      avg_last_visit_time:                      419.30298
+      num_replay_batches:                       429
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0018724315
+    avg_actor_approxkl3:                        0.0017986976
+    avg_actor_clipfrac:                         0.07133179
+    avg_actor_loss:                             0.0017678151
+    avg_advantage:                              -0.00023262216
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00032005654
+    avg_entropy:                                0.21380197
+    avg_grad_norm_pre_clip:                     4.42386
+    avg_loss:                                   0.0017140413
+    max:
+      max_actor_approxkl1:                      0.004964179
+      max_actor_approxkl3:                      0.003676056
+      max_actor_clipfrac:                       0.107543945
+      max_actor_loss:                           0.0038066413
+      max_advantage:                            0.097170115
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00044586573
+      max_entropy:                              0.22986962
+      max_grad_norm_pre_clip:                   8.10828
+      max_loss:                                 0.00372323
+    std:
+      std_actor_approxkl1:                      0.0011954359
+      std_actor_approxkl3:                      0.0008501008
+      std_actor_clipfrac:                       0.025902025
+      std_actor_loss:                           0.0012662124
+      std_advantage:                            0.021191608
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.058058e-05
+      std_entropy:                              0.007866739
+      std_grad_norm_pre_clip:                   1.9012206
+      std_loss:                                 0.0012657773
+  eval-batch-train:
+    avg_avg_episode_length:                     7.5185423
+    avg_avg_return:                             0.9813962
+    avg_benchmark_regret:                       0.011402972
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19676208
+    proxy_corner:
+      avg_avg_return:                           0.009609021
+      avg_reward_per_step:                      0.0017318726
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.713583
+    avg_avg_return:                             0.9738386
+    avg_benchmark_regret:                       0.016401216
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13692474
+    proxy_corner:
+      avg_avg_return:                           0.9738386
+      avg_reward_per_step:                      0.13692474
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.46239
+    avg_avg_return:                             0.9813486
+    avg_benchmark_regret:                       0.011488875
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19089508
+    proxy_corner:
+      avg_avg_return:                           0.007034773
+      avg_reward_per_step:                      0.0013809204
+  eval-batch-tree:
+    avg_avg_episode_length:                     83.742325
+    avg_avg_return:                             0.023645023
+    avg_benchmark_regret:                       0.9609902
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0061187744
+    proxy_corner:
+      avg_avg_return:                           0.023645023
+      avg_reward_per_step:                      0.0061187744
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        28311552
+    env-step-replay-before:                     14581760
+    env-step-used-before:                       28311552
+    ppo-update-before:                          17280
+    env-step-generate-before:                   13729792
+    env-step-all-after:                         28344320
+    env-step-replay-after:                      14581760
+    env-step-used-after:                        28344320
+    ppo-update-after:                           17300
+    env-step-generate-after:                    13762560
+  train-generate:
+    shift_proportion:                           1.0
+    avg_avg_episode_length:                     8.153671
+    avg_avg_return:                             0.94008386
+    avg_reward_per_step:                        0.16937256
+    proxy_corner:
+      avg_avg_return:                           0.0054138033
+      avg_reward_per_step:                      0.00045776367
+  train-all:
+    shift_proportion:                           1.0
+    avg_avg_episode_length:                     8.153671
+    avg_avg_return:                             0.94008386
+    avg_reward_per_step:                        0.16937256
+    proxy_corner:
+      avg_avg_return:                           0.0054138033
+      avg_reward_per_step:                      0.00045776367
+  perf:
+    env_steps_per_second:                       297368.69595073885
+    ppo_updates_per_second:                     41.73369206301104
+    eval-batch-train-duration:                  0.5301350047811866
+    eval-batch-orig-duration:                   0.4735927926376462
+    eval-batch-shift-duration:                  0.44351580925285816
+    eval-batch-tree-duration:                   0.46362120565027
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.31665
+      cheese-corne_dist_wavg:                   3.7344315
+      levels_cheese_is_in_corner_avg:           0.029052734
+      mouse-cheese_dist_finite_avg:             14.314453
+      mouse-cheese_dist_finite_wavg:            14.930093
+      mouse-cheese_dist_solvable_avg:           14.314453
+      mouse-cheese_dist_solvable_wavg:          14.930093
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11153705
+    visit_patterns:
+      avg_first_visit_time:                     280.333
+      avg_last_visit_time:                      435.31226
+      num_replay_batches:                       445
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002506825
+    avg_actor_approxkl3:                        0.0019090867
+    avg_actor_clipfrac:                         0.069281004
+    avg_actor_loss:                             0.0017535862
+    avg_advantage:                              0.0007289422
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00029239187
+    avg_entropy:                                0.18743478
+    avg_grad_norm_pre_clip:                     4.589385
+    avg_loss:                                   0.0017123474
+    max:
+      max_actor_approxkl1:                      0.00577128
+      max_actor_approxkl3:                      0.0044539114
+      max_actor_clipfrac:                       0.1217041
+      max_actor_loss:                           0.005871393
+      max_advantage:                            0.11107957
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00035372694
+      max_entropy:                              0.21807173
+      max_grad_norm_pre_clip:                   9.049134
+      max_loss:                                 0.00582207
+    std:
+      std_actor_approxkl1:                      0.0014439024
+      std_actor_approxkl3:                      0.0010899078
+      std_actor_clipfrac:                       0.027354088
+      std_actor_loss:                           0.001442343
+      std_advantage:                            0.02362065
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.8299795e-05
+      std_entropy:                              0.014770314
+      std_grad_norm_pre_clip:                   2.1165657
+      std_loss:                                 0.0014412565
+  eval-batch-train:
+    avg_avg_episode_length:                     7.6013985
+    avg_avg_return:                             0.9801644
+    avg_benchmark_regret:                       0.012634791
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19571686
+    proxy_corner:
+      avg_avg_return:                           0.008471869
+      avg_reward_per_step:                      0.0015869141
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.76903
+    avg_avg_return:                             0.9732436
+    avg_benchmark_regret:                       0.016996218
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13591766
+    proxy_corner:
+      avg_avg_return:                           0.9732436
+      avg_reward_per_step:                      0.13591766
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.432041
+    avg_avg_return:                             0.9813436
+    avg_benchmark_regret:                       0.011493829
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.18835449
+    proxy_corner:
+      avg_avg_return:                           0.0052333307
+      avg_reward_per_step:                      0.0011672974
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.291824
+    avg_avg_return:                             0.014430877
+    avg_benchmark_regret:                       0.9702043
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0046081543
+    proxy_corner:
+      avg_avg_return:                           0.014430877
+      avg_reward_per_step:                      0.0046081543
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        29360128
+    env-step-replay-before:                     15171584
+    env-step-used-before:                       29360128
+    ppo-update-before:                          17920
+    env-step-generate-before:                   14188544
+    env-step-all-after:                         29392896
+    env-step-replay-after:                      15171584
+    env-step-used-after:                        29392896
+    ppo-update-after:                           17940
+    env-step-generate-after:                    14221312
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     8.506849
+    avg_avg_return:                             0.93832433
+    avg_reward_per_step:                        0.17623901
+    proxy_corner:
+      avg_avg_return:                           0.020279221
+      avg_reward_per_step:                      0.0018005371
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     8.506849
+    avg_avg_return:                             0.93832433
+    avg_reward_per_step:                        0.17623901
+    proxy_corner:
+      avg_avg_return:                           0.020279221
+      avg_reward_per_step:                      0.0018005371
+  perf:
+    env_steps_per_second:                       324570.58078896423
+    ppo_updates_per_second:                     41.70089690059331
+    eval-batch-train-duration:                  0.537455934099853
+    eval-batch-orig-duration:                   0.48230205103755
+    eval-batch-shift-duration:                  0.4788084626197815
+    eval-batch-tree-duration:                   0.45248145889490843
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.3359375
+      cheese-corne_dist_wavg:                   3.7418475
+      levels_cheese_is_in_corner_avg:           0.030029297
+      mouse-cheese_dist_finite_avg:             14.356445
+      mouse-cheese_dist_finite_wavg:            14.935757
+      mouse-cheese_dist_solvable_avg:           14.356445
+      mouse-cheese_dist_solvable_wavg:          14.935759
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.111432
+    visit_patterns:
+      avg_first_visit_time:                     288.63208
+      avg_last_visit_time:                      453.32935
+      num_replay_batches:                       463
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0021015443
+    avg_actor_approxkl3:                        0.001948613
+    avg_actor_clipfrac:                         0.068884276
+    avg_actor_loss:                             0.0022512872
+    avg_advantage:                              -0.0023259397
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0002995048
+    avg_entropy:                                0.18434618
+    avg_grad_norm_pre_clip:                     5.7883353
+    avg_loss:                                   0.0022166937
+    max:
+      max_actor_approxkl1:                      0.0056895064
+      max_actor_approxkl3:                      0.003998462
+      max_actor_clipfrac:                       0.12609863
+      max_actor_loss:                           0.005805582
+      max_advantage:                            0.09001708
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00033710536
+      max_entropy:                              0.19994415
+      max_grad_norm_pre_clip:                   10.334983
+      max_loss:                                 0.005758069
+    std:
+      std_actor_approxkl1:                      0.0016786552
+      std_actor_approxkl3:                      0.0012569824
+      std_actor_clipfrac:                       0.03622491
+      std_actor_loss:                           0.0018339445
+      std_advantage:                            0.024295388
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.2564956e-05
+      std_entropy:                              0.011059625
+      std_grad_norm_pre_clip:                   2.942953
+      std_loss:                                 0.0018340059
+  eval-batch-train:
+    avg_avg_episode_length:                     7.713502
+    avg_avg_return:                             0.9805841
+    avg_benchmark_regret:                       0.012215077
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19358063
+    proxy_corner:
+      avg_avg_return:                           0.011290589
+      avg_reward_per_step:                      0.0017929077
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.686493
+    avg_avg_return:                             0.97380376
+    avg_benchmark_regret:                       0.0164361
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13721466
+    proxy_corner:
+      avg_avg_return:                           0.97380376
+      avg_reward_per_step:                      0.13721466
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.577286
+    avg_avg_return:                             0.9808148
+    avg_benchmark_regret:                       0.012022637
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.18749237
+    proxy_corner:
+      avg_avg_return:                           0.0072309813
+      avg_reward_per_step:                      0.0014419556
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.309166
+    avg_avg_return:                             0.015327325
+    avg_benchmark_regret:                       0.9693079
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0043563843
+    proxy_corner:
+      avg_avg_return:                           0.015327325
+      avg_reward_per_step:                      0.0043563843
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        30408704
+    env-step-replay-before:                     15761408
+    env-step-used-before:                       30408704
+    ppo-update-before:                          18560
+    env-step-generate-before:                   14647296
+    env-step-all-after:                         30441472
+    env-step-replay-after:                      15794176
+    env-step-used-after:                        30441472
+    ppo-update-after:                           18580
+    env-step-generate-after:                    14647296
+  train-replay:
+    shift_proportion:                           0.9453125
+    avg_avg_episode_length:                     16.665651
+    avg_avg_return:                             0.8727492
+    avg_reward_per_step:                        0.059448242
+    proxy_corner:
+      avg_avg_return:                           0.089267746
+      avg_reward_per_step:                      0.0053710938
+  train-all:
+    shift_proportion:                           0.9453125
+    avg_avg_episode_length:                     16.665651
+    avg_avg_return:                             0.8727492
+    avg_reward_per_step:                        0.059448242
+    proxy_corner:
+      avg_avg_return:                           0.089267746
+      avg_reward_per_step:                      0.0053710938
+  perf:
+    env_steps_per_second:                       357547.57727527077
+    ppo_updates_per_second:                     40.69371489627544
+    eval-batch-train-duration:                  0.49532005097717047
+    eval-batch-orig-duration:                   0.45138860307633877
+    eval-batch-shift-duration:                  0.47358640283346176
+    eval-batch-tree-duration:                   0.48500657826662064
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.354248
+      cheese-corne_dist_wavg:                   11.825219
+      levels_cheese_is_in_corner_avg:           0.028808594
+      mouse-cheese_dist_finite_avg:             14.379883
+      mouse-cheese_dist_finite_wavg:            9.538113
+      mouse-cheese_dist_solvable_avg:           14.379883
+      mouse-cheese_dist_solvable_wavg:          9.538114
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11541639
+    visit_patterns:
+      avg_first_visit_time:                     299.469
+      avg_last_visit_time:                      472.3164
+      num_replay_batches:                       482
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.004232708
+    avg_actor_approxkl3:                        0.0031170126
+    avg_actor_clipfrac:                         0.10006714
+    avg_actor_loss:                             0.0030199965
+    avg_advantage:                              -0.008673621
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0002956558
+    avg_entropy:                                0.22963457
+    avg_grad_norm_pre_clip:                     6.0504913
+    avg_loss:                                   0.0029381898
+    max:
+      max_actor_approxkl1:                      0.0087877605
+      max_actor_approxkl3:                      0.0070603294
+      max_actor_clipfrac:                       0.14904785
+      max_actor_loss:                           0.0058115087
+      max_advantage:                            0.09814811
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0004930958
+      max_entropy:                              0.2522918
+      max_grad_norm_pre_clip:                   9.488444
+      max_loss:                                 0.005761402
+    std:
+      std_actor_approxkl1:                      0.0019292382
+      std_actor_approxkl3:                      0.0015515113
+      std_actor_clipfrac:                       0.03341938
+      std_actor_loss:                           0.0018556423
+      std_advantage:                            0.02105312
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.8e-05
+      std_entropy:                              0.012783523
+      std_grad_norm_pre_clip:                   2.255876
+      std_loss:                                 0.0018636193
+  eval-batch-train:
+    avg_avg_episode_length:                     7.5568933
+    avg_avg_return:                             0.9810773
+    avg_benchmark_regret:                       0.01172186
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19939423
+    proxy_corner:
+      avg_avg_return:                           0.009476676
+      avg_reward_per_step:                      0.0017318726
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.730809
+    avg_avg_return:                             0.9738343
+    avg_benchmark_regret:                       0.016405582
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13757324
+    proxy_corner:
+      avg_avg_return:                           0.9738343
+      avg_reward_per_step:                      0.13757324
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.6733913
+    avg_avg_return:                             0.98087007
+    avg_benchmark_regret:                       0.011967359
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.18915558
+    proxy_corner:
+      avg_avg_return:                           0.007505637
+      avg_reward_per_step:                      0.0017166138
+  eval-batch-tree:
+    avg_avg_episode_length:                     83.99127
+    avg_avg_return:                             0.019941788
+    avg_benchmark_regret:                       0.9646934
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.005897522
+    proxy_corner:
+      avg_avg_return:                           0.019941788
+      avg_reward_per_step:                      0.005897522
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        31457280
+    env-step-replay-before:                     16384000
+    env-step-used-before:                       31457280
+    ppo-update-before:                          19200
+    env-step-generate-before:                   15073280
+    env-step-all-after:                         31490048
+    env-step-replay-after:                      16384000
+    env-step-used-after:                        31490048
+    ppo-update-after:                           19220
+    env-step-generate-after:                    15106048
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     8.246056
+    avg_avg_return:                             0.94296527
+    avg_reward_per_step:                        0.17150879
+    proxy_corner:
+      avg_avg_return:                           0.015260773
+      avg_reward_per_step:                      0.0015869141
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     8.246056
+    avg_avg_return:                             0.94296527
+    avg_reward_per_step:                        0.17150879
+    proxy_corner:
+      avg_avg_return:                           0.015260773
+      avg_reward_per_step:                      0.0015869141
+  perf:
+    env_steps_per_second:                       291377.92881933757
+    ppo_updates_per_second:                     42.123012545617364
+    eval-batch-train-duration:                  0.5377235850319266
+    eval-batch-orig-duration:                   0.47902264539152384
+    eval-batch-shift-duration:                  0.47941474709659815
+    eval-batch-tree-duration:                   0.49368455819785595
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.3984375
+      cheese-corne_dist_wavg:                   9.15003
+      levels_cheese_is_in_corner_avg:           0.029052734
+      mouse-cheese_dist_finite_avg:             14.389648
+      mouse-cheese_dist_finite_wavg:            13.140354
+      mouse-cheese_dist_solvable_avg:           14.389648
+      mouse-cheese_dist_solvable_wavg:          13.140356
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11693195
+    visit_patterns:
+      avg_first_visit_time:                     312.62573
+      avg_last_visit_time:                      490.41235
+      num_replay_batches:                       500
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0027721
+    avg_actor_approxkl3:                        0.0024480305
+    avg_actor_clipfrac:                         0.07340088
+    avg_actor_loss:                             0.0022905865
+    avg_advantage:                              0.0013311086
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00033626825
+    avg_entropy:                                0.17461157
+    avg_grad_norm_pre_clip:                     5.68111
+    avg_loss:                                   0.002284109
+    max:
+      max_actor_approxkl1:                      0.0073217046
+      max_actor_approxkl3:                      0.0070257364
+      max_actor_clipfrac:                       0.15100098
+      max_actor_loss:                           0.009643875
+      max_advantage:                            0.11005092
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0004238466
+      max_entropy:                              0.19386268
+      max_grad_norm_pre_clip:                   11.960476
+      max_loss:                                 0.009637228
+    std:
+      std_actor_approxkl1:                      0.0018034182
+      std_actor_approxkl3:                      0.0016850958
+      std_actor_clipfrac:                       0.036547557
+      std_actor_loss:                           0.0024069971
+      std_advantage:                            0.025067192
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.701167e-05
+      std_entropy:                              0.012375935
+      std_grad_norm_pre_clip:                   3.1212108
+      std_loss:                                 0.0024055669
+  eval-batch-train:
+    avg_avg_episode_length:                     7.6594224
+    avg_avg_return:                             0.9804879
+    avg_benchmark_regret:                       0.012311305
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19689178
+    proxy_corner:
+      avg_avg_return:                           0.0077956202
+      avg_reward_per_step:                      0.0015106201
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.70747
+    avg_avg_return:                             0.9737159
+    avg_benchmark_regret:                       0.01652396
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13549042
+    proxy_corner:
+      avg_avg_return:                           0.9737159
+      avg_reward_per_step:                      0.13549042
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.6474147
+    avg_avg_return:                             0.980844
+    avg_benchmark_regret:                       0.011993386
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.18826294
+    proxy_corner:
+      avg_avg_return:                           0.0040430045
+      avg_reward_per_step:                      0.0009689331
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.37248
+    avg_avg_return:                             0.014491163
+    avg_benchmark_regret:                       0.97014403
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0042648315
+    proxy_corner:
+      avg_avg_return:                           0.014491163
+      avg_reward_per_step:                      0.0042648315
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        32505856
+    env-step-replay-before:                     16842752
+    env-step-used-before:                       32505856
+    ppo-update-before:                          19840
+    env-step-generate-before:                   15663104
+    env-step-all-after:                         32538624
+    env-step-replay-after:                      16875520
+    env-step-used-after:                        32538624
+    ppo-update-after:                           19860
+    env-step-generate-after:                    15663104
+  train-replay:
+    shift_proportion:                           0.96484375
+    avg_avg_episode_length:                     15.949404
+    avg_avg_return:                             0.8758305
+    avg_reward_per_step:                        0.0602417
+    proxy_corner:
+      avg_avg_return:                           0.0560166
+      avg_reward_per_step:                      0.0037841797
+  train-all:
+    shift_proportion:                           0.96484375
+    avg_avg_episode_length:                     15.949404
+    avg_avg_return:                             0.8758305
+    avg_reward_per_step:                        0.0602417
+    proxy_corner:
+      avg_avg_return:                           0.0560166
+      avg_reward_per_step:                      0.0037841797
+  perf:
+    env_steps_per_second:                       312014.39948280505
+    ppo_updates_per_second:                     41.6681412741234
+    eval-batch-train-duration:                  0.5278077106922865
+    eval-batch-orig-duration:                   0.48368245642632246
+    eval-batch-shift-duration:                  0.4809945076704025
+    eval-batch-tree-duration:                   0.49201850313693285
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.43042
+      cheese-corne_dist_wavg:                   4.636937
+      levels_cheese_is_in_corner_avg:           0.02734375
+      mouse-cheese_dist_finite_avg:             14.471191
+      mouse-cheese_dist_finite_wavg:            13.145423
+      mouse-cheese_dist_solvable_avg:           14.471191
+      mouse-cheese_dist_solvable_wavg:          13.145423
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11381073
+    visit_patterns:
+      avg_first_visit_time:                     324.4214
+      avg_last_visit_time:                      505.28833
+      num_replay_batches:                       515
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0027026297
+    avg_actor_approxkl3:                        0.0025217254
+    avg_actor_clipfrac:                         0.084625244
+    avg_actor_loss:                             0.0025960484
+    avg_advantage:                              0.0033034591
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00023358758
+    avg_entropy:                                0.21704626
+    avg_grad_norm_pre_clip:                     5.954766
+    avg_loss:                                   0.0024957962
+    max:
+      max_actor_approxkl1:                      0.009382483
+      max_actor_approxkl3:                      0.008562093
+      max_actor_clipfrac:                       0.15734863
+      max_actor_loss:                           0.009403514
+      max_advantage:                            0.12230557
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00026268166
+      max_entropy:                              0.22988337
+      max_grad_norm_pre_clip:                   10.137419
+      max_loss:                                 0.009305027
+    std:
+      std_actor_approxkl1:                      0.0023376178
+      std_actor_approxkl3:                      0.0019372308
+      std_actor_clipfrac:                       0.03843925
+      std_actor_loss:                           0.0026652617
+      std_advantage:                            0.021246016
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.1849072e-05
+      std_entropy:                              0.0066644754
+      std_grad_norm_pre_clip:                   2.313794
+      std_loss:                                 0.002666138
+  eval-batch-train:
+    avg_avg_episode_length:                     7.598668
+    avg_avg_return:                             0.9808474
+    avg_benchmark_regret:                       0.011951737
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19683075
+    proxy_corner:
+      avg_avg_return:                           0.007817629
+      avg_reward_per_step:                      0.0015487671
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.578331
+    avg_avg_return:                             0.97440577
+    avg_benchmark_regret:                       0.015834041
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13978577
+    proxy_corner:
+      avg_avg_return:                           0.97440577
+      avg_reward_per_step:                      0.13978577
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.48067
+    avg_avg_return:                             0.98115283
+    avg_benchmark_regret:                       0.011684587
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.18804932
+    proxy_corner:
+      avg_avg_return:                           0.004765855
+      avg_reward_per_step:                      0.001083374
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.19734
+    avg_avg_return:                             0.015052548
+    avg_benchmark_regret:                       0.9695826
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0046920776
+    proxy_corner:
+      avg_avg_return:                           0.015052548
+      avg_reward_per_step:                      0.0046920776
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        33554432
+    env-step-replay-before:                     17399808
+    env-step-used-before:                       33554432
+    ppo-update-before:                          20480
+    env-step-generate-before:                   16154624
+    env-step-all-after:                         33587200
+    env-step-replay-after:                      17399808
+    env-step-used-after:                        33587200
+    ppo-update-after:                           20500
+    env-step-generate-after:                    16187392
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     8.584059
+    avg_avg_return:                             0.93910635
+    avg_reward_per_step:                        0.18823242
+    proxy_corner:
+      avg_avg_return:                           0.009129211
+      avg_reward_per_step:                      0.0008544922
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     8.584059
+    avg_avg_return:                             0.93910635
+    avg_reward_per_step:                        0.18823242
+    proxy_corner:
+      avg_avg_return:                           0.009129211
+      avg_reward_per_step:                      0.0008544922
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  perf:
+    env_steps_per_second:                       328826.37756325013
+    ppo_updates_per_second:                     41.431210798010746
+    eval-batch-train-duration:                  0.6144559839740396
+    eval-batch-orig-duration:                   0.5313631910830736
+    eval-batch-shift-duration:                  0.5539796110242605
+    eval-batch-tree-duration:                   0.5490169133991003
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.283447
+      cheese-corne_dist_wavg:                   14.526724
+      levels_cheese_is_in_corner_avg:           0.026367188
+      mouse-cheese_dist_finite_avg:             14.368652
+      mouse-cheese_dist_finite_wavg:            12.247418
+      mouse-cheese_dist_solvable_avg:           14.368652
+      mouse-cheese_dist_solvable_wavg:          12.247418
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12145704
+    visit_patterns:
+      avg_first_visit_time:                     343.17188
+      avg_last_visit_time:                      521.3025
+      num_replay_batches:                       531
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0018085014
+    avg_actor_approxkl3:                        0.0015559065
+    avg_actor_clipfrac:                         0.06173096
+    avg_actor_loss:                             0.00079956086
+    avg_advantage:                              0.0037590247
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00028287782
+    avg_entropy:                                0.19113444
+    avg_grad_norm_pre_clip:                     4.467663
+    avg_loss:                                   0.00074986526
+    max:
+      max_actor_approxkl1:                      0.005292917
+      max_actor_approxkl3:                      0.0034134071
+      max_actor_clipfrac:                       0.11608887
+      max_actor_loss:                           0.003306901
+      max_advantage:                            0.08852404
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00033880014
+      max_entropy:                              0.21339163
+      max_grad_norm_pre_clip:                   6.432695
+      max_loss:                                 0.003279624
+    std:
+      std_actor_approxkl1:                      0.0015493499
+      std_actor_approxkl3:                      0.0009825123
+      std_actor_clipfrac:                       0.03201267
+      std_actor_loss:                           0.001050466
+      std_advantage:                            0.023765266
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.6136031e-05
+      std_entropy:                              0.013820934
+      std_grad_norm_pre_clip:                   1.4338065
+      std_loss:                                 0.0010483204
+  eval-batch-train:
+    avg_avg_episode_length:                     8.180105
+    avg_avg_return:                             0.978819
+    avg_benchmark_regret:                       0.013980096
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.18694305
+    proxy_corner:
+      avg_avg_return:                           0.008510628
+      avg_reward_per_step:                      0.0014953613
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.114321
+    avg_avg_return:                             0.9720614
+    avg_benchmark_regret:                       0.01817843
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1330719
+    proxy_corner:
+      avg_avg_return:                           0.9720614
+      avg_reward_per_step:                      0.1330719
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.9938
+    avg_avg_return:                             0.97934294
+    avg_benchmark_regret:                       0.013494537
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.17908478
+    proxy_corner:
+      avg_avg_return:                           0.006139566
+      avg_reward_per_step:                      0.0011825562
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.95655
+    avg_avg_return:                             0.00796455
+    avg_benchmark_regret:                       0.9766706
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0036849976
+    proxy_corner:
+      avg_avg_return:                           0.00796455
+      avg_reward_per_step:                      0.0036849976
+    rollouts_gif:                               float32(132, 225, 225, 3)
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        34603008
+    env-step-replay-before:                     17694720
+    env-step-used-before:                       34603008
+    ppo-update-before:                          21120
+    env-step-generate-before:                   16908288
+    env-step-all-after:                         34635776
+    env-step-replay-after:                      17727488
+    env-step-used-after:                        34635776
+    ppo-update-after:                           21140
+    env-step-generate-after:                    16908288
+  train-replay:
+    shift_proportion:                           0.95703125
+    avg_avg_episode_length:                     15.673864
+    avg_avg_return:                             0.8805559
+    avg_reward_per_step:                        0.062347412
+    proxy_corner:
+      avg_avg_return:                           0.057751205
+      avg_reward_per_step:                      0.003753662
+  train-all:
+    shift_proportion:                           0.95703125
+    avg_avg_episode_length:                     15.673864
+    avg_avg_return:                             0.8805559
+    avg_reward_per_step:                        0.062347412
+    proxy_corner:
+      avg_avg_return:                           0.057751205
+      avg_reward_per_step:                      0.003753662
+  perf:
+    env_steps_per_second:                       295625.8705361094
+    ppo_updates_per_second:                     42.30630623843751
+    eval-batch-train-duration:                  0.3074246384203434
+    eval-batch-orig-duration:                   0.21790319681167603
+    eval-batch-shift-duration:                  0.22038648836314678
+    eval-batch-tree-duration:                   0.25554955936968327
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.035645
+      cheese-corne_dist_wavg:                   9.108324
+      levels_cheese_is_in_corner_avg:           0.02709961
+      mouse-cheese_dist_finite_avg:             14.389893
+      mouse-cheese_dist_finite_wavg:            10.439625
+      mouse-cheese_dist_solvable_avg:           14.389893
+      mouse-cheese_dist_solvable_wavg:          10.439626
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.119755834
+    visit_patterns:
+      avg_first_visit_time:                     359.5127
+      avg_last_visit_time:                      531.2468
+      num_replay_batches:                       541
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0030830277
+    avg_actor_approxkl3:                        0.0027641135
+    avg_actor_clipfrac:                         0.095184326
+    avg_actor_loss:                             0.0035928308
+    avg_advantage:                              -0.0039865887
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00027276107
+    avg_entropy:                                0.21648097
+    avg_grad_norm_pre_clip:                     6.5521173
+    avg_loss:                                   0.0035127313
+    max:
+      max_actor_approxkl1:                      0.005619347
+      max_actor_approxkl3:                      0.006867479
+      max_actor_clipfrac:                       0.1772461
+      max_actor_loss:                           0.010274249
+      max_advantage:                            0.08421737
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0004105873
+      max_entropy:                              0.23597772
+      max_grad_norm_pre_clip:                   12.343827
+      max_loss:                                 0.01021845
+    std:
+      std_actor_approxkl1:                      0.0015175234
+      std_actor_approxkl3:                      0.0014798808
+      std_actor_clipfrac:                       0.03642226
+      std_actor_loss:                           0.0024822932
+      std_advantage:                            0.021710666
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.152136e-05
+      std_entropy:                              0.009869865
+      std_grad_norm_pre_clip:                   2.5220735
+      std_loss:                                 0.0024898543
+  eval-batch-train:
+    avg_avg_episode_length:                     7.45991
+    avg_avg_return:                             0.98147607
+    avg_benchmark_regret:                       0.011323108
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19919586
+    proxy_corner:
+      avg_avg_return:                           0.007925012
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.717096
+    avg_avg_return:                             0.97340846
+    avg_benchmark_regret:                       0.016831387
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13735962
+    proxy_corner:
+      avg_avg_return:                           0.97340846
+      avg_reward_per_step:                      0.13735962
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4339886
+    avg_avg_return:                             0.9815848
+    avg_benchmark_regret:                       0.011252626
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19010925
+    proxy_corner:
+      avg_avg_return:                           0.004371679
+      avg_reward_per_step:                      0.0010223389
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.00522
+    avg_avg_return:                             0.0047553773
+    avg_benchmark_regret:                       0.97987986
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0029144287
+    proxy_corner:
+      avg_avg_return:                           0.0047553773
+      avg_reward_per_step:                      0.0029144287
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        35651584
+    env-step-replay-before:                     18153472
+    env-step-used-before:                       35651584
+    ppo-update-before:                          21760
+    env-step-generate-before:                   17498112
+    env-step-all-after:                         35684352
+    env-step-replay-after:                      18186240
+    env-step-used-after:                        35684352
+    ppo-update-after:                           21780
+    env-step-generate-after:                    17498112
+  train-replay:
+    shift_proportion:                           0.96875
+    avg_avg_episode_length:                     15.323824
+    avg_avg_return:                             0.8831338
+    avg_reward_per_step:                        0.06439209
+    proxy_corner:
+      avg_avg_return:                           0.043549787
+      avg_reward_per_step:                      0.0027770996
+  train-all:
+    shift_proportion:                           0.96875
+    avg_avg_episode_length:                     15.323824
+    avg_avg_return:                             0.8831338
+    avg_reward_per_step:                        0.06439209
+    proxy_corner:
+      avg_avg_return:                           0.043549787
+      avg_reward_per_step:                      0.0027770996
+  perf:
+    env_steps_per_second:                       260051.17263391838
+    ppo_updates_per_second:                     42.32942839625422
+    eval-batch-train-duration:                  0.6346735302358866
+    eval-batch-orig-duration:                   0.6044590659439564
+    eval-batch-shift-duration:                  0.6312424773350358
+    eval-batch-tree-duration:                   0.6400297740474343
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.016357
+      cheese-corne_dist_wavg:                   4.614995
+      levels_cheese_is_in_corner_avg:           0.02758789
+      mouse-cheese_dist_finite_avg:             14.510986
+      mouse-cheese_dist_finite_wavg:            7.7456193
+      mouse-cheese_dist_solvable_avg:           14.510986
+      mouse-cheese_dist_solvable_wavg:          7.7456174
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000002
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11485486
+    visit_patterns:
+      avg_first_visit_time:                     370.93164
+      avg_last_visit_time:                      545.3247
+      num_replay_batches:                       555
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0031348236
+    avg_actor_approxkl3:                        0.002812584
+    avg_actor_clipfrac:                         0.09234009
+    avg_actor_loss:                             0.00353205
+    avg_advantage:                              0.0055203266
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00023868878
+    avg_entropy:                                0.20602372
+    avg_grad_norm_pre_clip:                     6.345275
+    avg_loss:                                   0.0034453706
+    max:
+      max_actor_approxkl1:                      0.009255333
+      max_actor_approxkl3:                      0.00818559
+      max_actor_clipfrac:                       0.17419434
+      max_actor_loss:                           0.009805341
+      max_advantage:                            0.10702616
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00027661736
+      max_entropy:                              0.22629744
+      max_grad_norm_pre_clip:                   12.31147
+      max_loss:                                 0.0097300615
+    std:
+      std_actor_approxkl1:                      0.0019182288
+      std_actor_approxkl3:                      0.0018356892
+      std_actor_clipfrac:                       0.037657738
+      std_actor_loss:                           0.0026191163
+      std_advantage:                            0.02126091
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.2935553e-05
+      std_entropy:                              0.0087234555
+      std_grad_norm_pre_clip:                   2.9781246
+      std_loss:                                 0.002619761
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4131174
+    avg_avg_return:                             0.98117876
+    avg_benchmark_regret:                       0.011620359
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20175934
+    proxy_corner:
+      avg_avg_return:                           0.00796454
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.642227
+    avg_avg_return:                             0.97418857
+    avg_benchmark_regret:                       0.016051313
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13898468
+    proxy_corner:
+      avg_avg_return:                           0.97418857
+      avg_reward_per_step:                      0.13898468
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.400233
+    avg_avg_return:                             0.9813893
+    avg_benchmark_regret:                       0.011448114
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19148254
+    proxy_corner:
+      avg_avg_return:                           0.0042167488
+      avg_reward_per_step:                      0.001045227
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.87871
+    avg_avg_return:                             0.0067321965
+    avg_benchmark_regret:                       0.977903
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0015869141
+    proxy_corner:
+      avg_avg_return:                           0.0067321965
+      avg_reward_per_step:                      0.0015869141
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        36700160
+    env-step-replay-before:                     18677760
+    env-step-used-before:                       36700160
+    ppo-update-before:                          22400
+    env-step-generate-before:                   18022400
+    env-step-all-after:                         36732928
+    env-step-replay-after:                      18677760
+    env-step-used-after:                        36732928
+    ppo-update-after:                           22420
+    env-step-generate-after:                    18055168
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.8507123
+    avg_avg_return:                             0.94413745
+    avg_reward_per_step:                        0.17883301
+    proxy_corner:
+      avg_avg_return:                           0.0156622
+      avg_reward_per_step:                      0.0017700195
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.8507123
+    avg_avg_return:                             0.94413745
+    avg_reward_per_step:                        0.17883301
+    proxy_corner:
+      avg_avg_return:                           0.0156622
+      avg_reward_per_step:                      0.0017700195
+  perf:
+    env_steps_per_second:                       309539.8886441675
+    ppo_updates_per_second:                     41.952159678461356
+    eval-batch-train-duration:                  0.5247107679024339
+    eval-batch-orig-duration:                   0.465800859965384
+    eval-batch-shift-duration:                  0.435817439109087
+    eval-batch-tree-duration:                   0.48955119401216507
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.000488
+      cheese-corne_dist_wavg:                   11.814225
+      levels_cheese_is_in_corner_avg:           0.02758789
+      mouse-cheese_dist_finite_avg:             14.616699
+      mouse-cheese_dist_finite_wavg:            9.565245
+      mouse-cheese_dist_solvable_avg:           14.616699
+      mouse-cheese_dist_solvable_wavg:          9.565244
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11534098
+    visit_patterns:
+      avg_first_visit_time:                     381.96313
+      avg_last_visit_time:                      560.3032
+      num_replay_batches:                       570
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0033389472
+    avg_actor_approxkl3:                        0.002496831
+    avg_actor_clipfrac:                         0.06258545
+    avg_actor_loss:                             0.0026442898
+    avg_advantage:                              0.0028813798
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00027057627
+    avg_entropy:                                0.14676723
+    avg_grad_norm_pre_clip:                     4.1233706
+    avg_loss:                                   0.0026328107
+    max:
+      max_actor_approxkl1:                      0.012498559
+      max_actor_approxkl3:                      0.008378673
+      max_actor_clipfrac:                       0.14135742
+      max_actor_loss:                           0.009140142
+      max_advantage:                            0.09774727
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0003046904
+      max_entropy:                              0.16393955
+      max_grad_norm_pre_clip:                   9.902828
+      max_loss:                                 0.009123772
+    std:
+      std_actor_approxkl1:                      0.0029648028
+      std_actor_approxkl3:                      0.0022444741
+      std_actor_clipfrac:                       0.035303596
+      std_actor_loss:                           0.0028214771
+      std_advantage:                            0.023073426
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.463476e-05
+      std_entropy:                              0.0125239985
+      std_grad_norm_pre_clip:                   2.6952178
+      std_loss:                                 0.0028213593
+  eval-batch-train:
+    avg_avg_episode_length:                     7.6102324
+    avg_avg_return:                             0.98054504
+    avg_benchmark_regret:                       0.012254139
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19742584
+    proxy_corner:
+      avg_avg_return:                           0.007932212
+      avg_reward_per_step:                      0.0015792847
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.721619
+    avg_avg_return:                             0.97414815
+    avg_benchmark_regret:                       0.016091635
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13906097
+    proxy_corner:
+      avg_avg_return:                           0.97414815
+      avg_reward_per_step:                      0.13906097
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4620123
+    avg_avg_return:                             0.98139757
+    avg_benchmark_regret:                       0.011439873
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.18960571
+    proxy_corner:
+      avg_avg_return:                           0.004719901
+      avg_reward_per_step:                      0.001121521
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.00743
+    avg_avg_return:                             0.0038759336
+    avg_benchmark_regret:                       0.98075926
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.002029419
+    proxy_corner:
+      avg_avg_return:                           0.0038759336
+      avg_reward_per_step:                      0.002029419
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        37748736
+    env-step-replay-before:                     19267584
+    env-step-used-before:                       37748736
+    ppo-update-before:                          23040
+    env-step-generate-before:                   18481152
+    env-step-all-after:                         37781504
+    env-step-replay-after:                      19267584
+    env-step-used-after:                        37781504
+    ppo-update-after:                           23060
+    env-step-generate-after:                    18513920
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     8.255164
+    avg_avg_return:                             0.94287395
+    avg_reward_per_step:                        0.18252563
+    proxy_corner:
+      avg_avg_return:                           0.004033807
+      avg_reward_per_step:                      0.0006713867
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     8.255164
+    avg_avg_return:                             0.94287395
+    avg_reward_per_step:                        0.18252563
+    proxy_corner:
+      avg_avg_return:                           0.004033807
+      avg_reward_per_step:                      0.0006713867
+  perf:
+    env_steps_per_second:                       321459.57410704135
+    ppo_updates_per_second:                     41.57502405832831
+    eval-batch-train-duration:                  0.5205559134483337
+    eval-batch-orig-duration:                   0.4755412247031927
+    eval-batch-shift-duration:                  0.4722100645303726
+    eval-batch-tree-duration:                   0.4607278574258089
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.970947
+      cheese-corne_dist_wavg:                   3.6938748
+      levels_cheese_is_in_corner_avg:           0.02709961
+      mouse-cheese_dist_finite_avg:             14.648193
+      mouse-cheese_dist_finite_wavg:            14.0650835
+      mouse-cheese_dist_solvable_avg:           14.648193
+      mouse-cheese_dist_solvable_wavg:          14.065082
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11800142
+    visit_patterns:
+      avg_first_visit_time:                     393.573
+      avg_last_visit_time:                      578.2324
+      num_replay_batches:                       588
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0021331045
+    avg_actor_approxkl3:                        0.0018761549
+    avg_actor_clipfrac:                         0.061120607
+    avg_actor_loss:                             0.0013117008
+    avg_advantage:                              -0.0010230461
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00029536206
+    avg_entropy:                                0.16291653
+    avg_grad_norm_pre_clip:                     3.7788494
+    avg_loss:                                   0.001296465
+    max:
+      max_actor_approxkl1:                      0.0049706376
+      max_actor_approxkl3:                      0.0039427597
+      max_actor_clipfrac:                       0.095214844
+      max_actor_loss:                           0.0048985407
+      max_advantage:                            0.12299389
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0003294743
+      max_entropy:                              0.18714318
+      max_grad_norm_pre_clip:                   7.6575055
+      max_loss:                                 0.0048828973
+    std:
+      std_actor_approxkl1:                      0.0012663435
+      std_actor_approxkl3:                      0.0009887904
+      std_actor_clipfrac:                       0.022572549
+      std_actor_loss:                           0.0012933958
+      std_advantage:                            0.024112668
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.8374674e-05
+      std_entropy:                              0.014659758
+      std_grad_norm_pre_clip:                   1.6394981
+      std_loss:                                 0.0012899531
+  eval-batch-train:
+    avg_avg_episode_length:                     7.633028
+    avg_avg_return:                             0.9805039
+    avg_benchmark_regret:                       0.0122952
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19521332
+    proxy_corner:
+      avg_avg_return:                           0.0083596315
+      avg_reward_per_step:                      0.001625061
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.718425
+    avg_avg_return:                             0.97414535
+    avg_benchmark_regret:                       0.016094485
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13884735
+    proxy_corner:
+      avg_avg_return:                           0.97414535
+      avg_reward_per_step:                      0.13884735
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4830823
+    avg_avg_return:                             0.98130834
+    avg_benchmark_regret:                       0.0115291225
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.18943024
+    proxy_corner:
+      avg_avg_return:                           0.004654824
+      avg_reward_per_step:                      0.001045227
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.064514
+    avg_avg_return:                             0.003382684
+    avg_benchmark_regret:                       0.98125255
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.00019836426
+    proxy_corner:
+      avg_avg_return:                           0.003382684
+      avg_reward_per_step:                      0.00019836426
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        38797312
+    env-step-replay-before:                     19922944
+    env-step-used-before:                       38797312
+    ppo-update-before:                          23680
+    env-step-generate-before:                   18874368
+    env-step-all-after:                         38830080
+    env-step-replay-after:                      19922944
+    env-step-used-after:                        38830080
+    ppo-update-after:                           23700
+    env-step-generate-after:                    18907136
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.053296
+    avg_avg_return:                             0.9517883
+    avg_reward_per_step:                        0.20819092
+    proxy_corner:
+      avg_avg_return:                           0.006467533
+      avg_reward_per_step:                      0.0010986328
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.053296
+    avg_avg_return:                             0.9517883
+    avg_reward_per_step:                        0.20819092
+    proxy_corner:
+      avg_avg_return:                           0.006467533
+      avg_reward_per_step:                      0.0010986328
+  perf:
+    env_steps_per_second:                       296405.11830247793
+    ppo_updates_per_second:                     42.19563377752742
+    eval-batch-train-duration:                  0.538545816205442
+    eval-batch-orig-duration:                   0.47639001719653606
+    eval-batch-shift-duration:                  0.4529964476823807
+    eval-batch-tree-duration:                   0.46852928679436445
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.883545
+      cheese-corne_dist_wavg:                   9.979569
+      levels_cheese_is_in_corner_avg:           0.026855469
+      mouse-cheese_dist_finite_avg:             14.681885
+      mouse-cheese_dist_finite_wavg:            16.765581
+      mouse-cheese_dist_solvable_avg:           14.681885
+      mouse-cheese_dist_solvable_wavg:          16.765583
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.116568424
+    visit_patterns:
+      avg_first_visit_time:                     401.9038
+      avg_last_visit_time:                      598.3262
+      num_replay_batches:                       608
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0024553614
+    avg_actor_approxkl3:                        0.0024432538
+    avg_actor_clipfrac:                         0.065264896
+    avg_actor_loss:                             0.0026765703
+    avg_advantage:                              0.005788371
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00031169734
+    avg_entropy:                                0.14202711
+    avg_grad_norm_pre_clip:                     4.4334984
+    avg_loss:                                   0.0026903918
+    max:
+      max_actor_approxkl1:                      0.006060837
+      max_actor_approxkl3:                      0.0064946315
+      max_actor_clipfrac:                       0.109375
+      max_actor_loss:                           0.0071400553
+      max_advantage:                            0.11721128
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00036874815
+      max_entropy:                              0.16856766
+      max_grad_norm_pre_clip:                   8.80909
+      max_loss:                                 0.0071468353
+    std:
+      std_actor_approxkl1:                      0.0013972413
+      std_actor_approxkl3:                      0.0014827513
+      std_actor_clipfrac:                       0.024874061
+      std_actor_loss:                           0.00226973
+      std_advantage:                            0.024792638
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.987907e-05
+      std_entropy:                              0.013918513
+      std_grad_norm_pre_clip:                   2.231134
+      std_loss:                                 0.002262691
+  eval-batch-train:
+    avg_avg_episode_length:                     7.5766077
+    avg_avg_return:                             0.98105705
+    avg_benchmark_regret:                       0.011742154
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19998169
+    proxy_corner:
+      avg_avg_return:                           0.008023104
+      avg_reward_per_step:                      0.0016098022
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.633155
+    avg_avg_return:                             0.9740633
+    avg_benchmark_regret:                       0.016176596
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13951111
+    proxy_corner:
+      avg_avg_return:                           0.9740633
+      avg_reward_per_step:                      0.13951111
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.5314584
+    avg_avg_return:                             0.9809412
+    avg_benchmark_regret:                       0.011896236
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.1902237
+    proxy_corner:
+      avg_avg_return:                           0.0046083815
+      avg_reward_per_step:                      0.0010375977
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.02247
+    avg_avg_return:                             0.004425264
+    avg_benchmark_regret:                       0.98020995
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.00064849854
+    proxy_corner:
+      avg_avg_return:                           0.004425264
+      avg_reward_per_step:                      0.00064849854
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        39845888
+    env-step-replay-before:                     20381696
+    env-step-used-before:                       39845888
+    ppo-update-before:                          24320
+    env-step-generate-before:                   19464192
+    env-step-all-after:                         39878656
+    env-step-replay-after:                      20414464
+    env-step-used-after:                        39878656
+    ppo-update-after:                           24340
+    env-step-generate-after:                    19464192
+  train-replay:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     16.003216
+    avg_avg_return:                             0.8777257
+    avg_reward_per_step:                        0.060699463
+    proxy_corner:
+      avg_avg_return:                           0.02728263
+      avg_reward_per_step:                      0.0018920898
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     16.003216
+    avg_avg_return:                             0.8777257
+    avg_reward_per_step:                        0.060699463
+    proxy_corner:
+      avg_avg_return:                           0.02728263
+      avg_reward_per_step:                      0.0018920898
+  perf:
+    env_steps_per_second:                       359341.40077599516
+    ppo_updates_per_second:                     42.76543322989237
+    eval-batch-train-duration:                  0.5401921141892672
+    eval-batch-orig-duration:                   0.44265621341764927
+    eval-batch-shift-duration:                  0.47250046860426664
+    eval-batch-tree-duration:                   0.4892426962032914
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.861816
+      cheese-corne_dist_wavg:                   4.575697
+      levels_cheese_is_in_corner_avg:           0.02734375
+      mouse-cheese_dist_finite_avg:             14.736328
+      mouse-cheese_dist_finite_wavg:            14.973437
+      mouse-cheese_dist_solvable_avg:           14.736328
+      mouse-cheese_dist_solvable_wavg:          14.973441
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.114774205
+    visit_patterns:
+      avg_first_visit_time:                     411.95386
+      avg_last_visit_time:                      613.3181
+      num_replay_batches:                       623
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0015467225
+    avg_actor_approxkl3:                        0.0011845535
+    avg_actor_clipfrac:                         0.05251465
+    avg_actor_loss:                             0.00036237133
+    avg_advantage:                              -3.0854993e-05
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00020307713
+    avg_entropy:                                0.19538571
+    avg_grad_norm_pre_clip:                     3.6458473
+    avg_loss:                                   0.00026852413
+    max:
+      max_actor_approxkl1:                      0.0027593188
+      max_actor_approxkl3:                      0.0021033813
+      max_actor_clipfrac:                       0.076538086
+      max_actor_loss:                           0.0025343522
+      max_advantage:                            0.09089345
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00021872991
+      max_entropy:                              0.20914742
+      max_grad_norm_pre_clip:                   6.5349402
+      max_loss:                                 0.002454047
+    std:
+      std_actor_approxkl1:                      0.0008513663
+      std_actor_approxkl3:                      0.0005137511
+      std_actor_clipfrac:                       0.020940648
+      std_actor_loss:                           0.0008397892
+      std_advantage:                            0.020197975
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.27465e-06
+      std_entropy:                              0.00975557
+      std_grad_norm_pre_clip:                   1.454638
+      std_loss:                                 0.0008404233
+  eval-batch-train:
+    avg_avg_episode_length:                     7.5525475
+    avg_avg_return:                             0.98164755
+    avg_benchmark_regret:                       0.011151589
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.1998825
+    proxy_corner:
+      avg_avg_return:                           0.00825128
+      avg_reward_per_step:                      0.0016098022
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.595117
+    avg_avg_return:                             0.9747864
+    avg_benchmark_regret:                       0.015453424
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13973999
+    proxy_corner:
+      avg_avg_return:                           0.9747864
+      avg_reward_per_step:                      0.13973999
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.39098
+    avg_avg_return:                             0.98179865
+    avg_benchmark_regret:                       0.0110387225
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19248962
+    proxy_corner:
+      avg_avg_return:                           0.0040909555
+      avg_reward_per_step:                      0.0009765625
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.99238
+    avg_avg_return:                             0.0048559513
+    avg_benchmark_regret:                       0.97977924
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.00035858154
+    proxy_corner:
+      avg_avg_return:                           0.0048559513
+      avg_reward_per_step:                      0.00035858154
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        40894464
+    env-step-replay-before:                     21037056
+    env-step-used-before:                       40894464
+    ppo-update-before:                          24960
+    env-step-generate-before:                   19857408
+    env-step-all-after:                         40927232
+    env-step-replay-after:                      21069824
+    env-step-used-after:                        40927232
+    ppo-update-after:                           24980
+    env-step-generate-after:                    19857408
+  train-replay:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     17.522701
+    avg_avg_return:                             0.862589
+    avg_reward_per_step:                        0.05697632
+    proxy_corner:
+      avg_avg_return:                           0.02642273
+      avg_reward_per_step:                      0.0016479492
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     17.522701
+    avg_avg_return:                             0.862589
+    avg_reward_per_step:                        0.05697632
+    proxy_corner:
+      avg_avg_return:                           0.02642273
+      avg_reward_per_step:                      0.0016479492
+  perf:
+    env_steps_per_second:                       357212.33653659583
+    ppo_updates_per_second:                     42.38259335235438
+    eval-batch-train-duration:                  0.5123239075765014
+    eval-batch-orig-duration:                   0.47380540426820517
+    eval-batch-shift-duration:                  0.4798412034288049
+    eval-batch-tree-duration:                   0.48805466666817665
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.895508
+      cheese-corne_dist_wavg:                   4.5935545
+      levels_cheese_is_in_corner_avg:           0.028076172
+      mouse-cheese_dist_finite_avg:             14.743164
+      mouse-cheese_dist_finite_wavg:            10.480674
+      mouse-cheese_dist_solvable_avg:           14.743164
+      mouse-cheese_dist_solvable_wavg:          10.480675
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.121726915
+    visit_patterns:
+      avg_first_visit_time:                     421.83984
+      avg_last_visit_time:                      633.2639
+      num_replay_batches:                       643
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00346998
+    avg_actor_approxkl3:                        0.0028410063
+    avg_actor_clipfrac:                         0.09312134
+    avg_actor_loss:                             0.0017264482
+    avg_advantage:                              0.0016994278
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00026975592
+    avg_entropy:                                0.21043968
+    avg_grad_norm_pre_clip:                     4.460849
+    avg_loss:                                   0.0016508866
+    max:
+      max_actor_approxkl1:                      0.0077336766
+      max_actor_approxkl3:                      0.005569212
+      max_actor_clipfrac:                       0.14819336
+      max_actor_loss:                           0.004947399
+      max_advantage:                            0.09929365
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0003185959
+      max_entropy:                              0.2372886
+      max_grad_norm_pre_clip:                   6.500188
+      max_loss:                                 0.004896552
+    std:
+      std_actor_approxkl1:                      0.00211928
+      std_actor_approxkl3:                      0.0014623572
+      std_actor_clipfrac:                       0.036809172
+      std_actor_loss:                           0.0015872787
+      std_advantage:                            0.022896321
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.0950563e-05
+      std_entropy:                              0.013513877
+      std_grad_norm_pre_clip:                   1.3740317
+      std_loss:                                 0.0015879684
+  eval-batch-train:
+    avg_avg_episode_length:                     8.231863
+    avg_avg_return:                             0.9789641
+    avg_benchmark_regret:                       0.013835092
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19581604
+    proxy_corner:
+      avg_avg_return:                           0.0076746056
+      avg_reward_per_step:                      0.0015640259
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.960939
+    avg_avg_return:                             0.9726908
+    avg_benchmark_regret:                       0.017549042
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1348114
+    proxy_corner:
+      avg_avg_return:                           0.9726908
+      avg_reward_per_step:                      0.1348114
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.8728447
+    avg_avg_return:                             0.97991264
+    avg_benchmark_regret:                       0.01292478
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.18599701
+    proxy_corner:
+      avg_avg_return:                           0.0038625712
+      avg_reward_per_step:                      0.0009384155
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.152374
+    avg_avg_return:                             0.0039652158
+    avg_benchmark_regret:                       0.98067
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        6.1035156e-05
+    proxy_corner:
+      avg_avg_return:                           0.0039652158
+      avg_reward_per_step:                      6.1035156e-05
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        41943040
+    env-step-replay-before:                     21528576
+    env-step-used-before:                       41943040
+    ppo-update-before:                          25600
+    env-step-generate-before:                   20414464
+    env-step-all-after:                         41975808
+    env-step-replay-after:                      21528576
+    env-step-used-after:                        41975808
+    ppo-update-after:                           25620
+    env-step-generate-after:                    20447232
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     8.380802
+    avg_avg_return:                             0.9388919
+    avg_reward_per_step:                        0.17559814
+    proxy_corner:
+      avg_avg_return:                           0.011432482
+      avg_reward_per_step:                      0.0010681152
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     8.380802
+    avg_avg_return:                             0.9388919
+    avg_reward_per_step:                        0.17559814
+    proxy_corner:
+      avg_avg_return:                           0.011432482
+      avg_reward_per_step:                      0.0010681152
+  perf:
+    env_steps_per_second:                       358779.4880890834
+    ppo_updates_per_second:                     42.704309711583214
+    eval-batch-train-duration:                  0.5441095223650336
+    eval-batch-orig-duration:                   0.44934000074863434
+    eval-batch-shift-duration:                  0.4743151552975178
+    eval-batch-tree-duration:                   0.4853787049651146
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.976318
+      cheese-corne_dist_wavg:                   9.996189
+      levels_cheese_is_in_corner_avg:           0.028564453
+      mouse-cheese_dist_finite_avg:             14.8203125
+      mouse-cheese_dist_finite_wavg:            3.2947416
+      mouse-cheese_dist_solvable_avg:           14.8203125
+      mouse-cheese_dist_solvable_wavg:          3.294742
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.121114135
+    visit_patterns:
+      avg_first_visit_time:                     435.64307
+      avg_last_visit_time:                      647.22534
+      num_replay_batches:                       657
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0015159113
+    avg_actor_approxkl3:                        0.0011325376
+    avg_actor_clipfrac:                         0.047424316
+    avg_actor_loss:                             0.0007238788
+    avg_advantage:                              0.0028993115
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00028728312
+    avg_entropy:                                0.15973781
+    avg_grad_norm_pre_clip:                     3.582145
+    avg_loss:                                   0.0007077825
+    max:
+      max_actor_approxkl1:                      0.0026973211
+      max_actor_approxkl3:                      0.0022032894
+      max_actor_clipfrac:                       0.09643555
+      max_actor_loss:                           0.002520733
+      max_advantage:                            0.10961473
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00032539846
+      max_entropy:                              0.18285221
+      max_grad_norm_pre_clip:                   7.428718
+      max_loss:                                 0.0024980202
+    std:
+      std_actor_approxkl1:                      0.0008701334
+      std_actor_approxkl3:                      0.0005922153
+      std_actor_clipfrac:                       0.02153538
+      std_actor_loss:                           0.00085830456
+      std_advantage:                            0.023939097
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.9493811e-05
+      std_entropy:                              0.014278192
+      std_grad_norm_pre_clip:                   1.5901389
+      std_loss:                                 0.00086025766
+  eval-batch-train:
+    avg_avg_episode_length:                     7.672513
+    avg_avg_return:                             0.98048466
+    avg_benchmark_regret:                       0.01231445
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19648743
+    proxy_corner:
+      avg_avg_return:                           0.008467596
+      avg_reward_per_step:                      0.0016326904
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.598061
+    avg_avg_return:                             0.9744891
+    avg_benchmark_regret:                       0.01575075
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13973236
+    proxy_corner:
+      avg_avg_return:                           0.9744891
+      avg_reward_per_step:                      0.13973236
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.540404
+    avg_avg_return:                             0.9807441
+    avg_benchmark_regret:                       0.0120933335
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.18914795
+    proxy_corner:
+      avg_avg_return:                           0.0041879364
+      avg_reward_per_step:                      0.0009918213
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.15384
+    avg_avg_return:                             0.0025736392
+    avg_benchmark_regret:                       0.98206156
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        6.1035156e-05
+    proxy_corner:
+      avg_avg_return:                           0.0025736392
+      avg_reward_per_step:                      6.1035156e-05
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        42991616
+    env-step-replay-before:                     21954560
+    env-step-used-before:                       42991616
+    ppo-update-before:                          26240
+    env-step-generate-before:                   21037056
+    env-step-all-after:                         43024384
+    env-step-replay-after:                      21987328
+    env-step-used-after:                        43024384
+    ppo-update-after:                           26260
+    env-step-generate-after:                    21037056
+  train-replay:
+    shift_proportion:                           0.96484375
+    avg_avg_episode_length:                     17.793678
+    avg_avg_return:                             0.86426824
+    avg_reward_per_step:                        0.057006836
+    proxy_corner:
+      avg_avg_return:                           0.045723602
+      avg_reward_per_step:                      0.0025634766
+  train-all:
+    shift_proportion:                           0.96484375
+    avg_avg_episode_length:                     17.793678
+    avg_avg_return:                             0.86426824
+    avg_reward_per_step:                        0.057006836
+    proxy_corner:
+      avg_avg_return:                           0.045723602
+      avg_reward_per_step:                      0.0025634766
+  perf:
+    env_steps_per_second:                       311118.4523657609
+    ppo_updates_per_second:                     42.19432741037373
+    eval-batch-train-duration:                  0.5148327732458711
+    eval-batch-orig-duration:                   0.47928938269615173
+    eval-batch-shift-duration:                  0.4485570667311549
+    eval-batch-tree-duration:                   0.4674711963161826
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.0217285
+      cheese-corne_dist_wavg:                   9.0987015
+      levels_cheese_is_in_corner_avg:           0.029785156
+      mouse-cheese_dist_finite_avg:             14.837891
+      mouse-cheese_dist_finite_wavg:            14.979229
+      mouse-cheese_dist_solvable_avg:           14.837891
+      mouse-cheese_dist_solvable_wavg:          14.979233
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1206577
+    visit_patterns:
+      avg_first_visit_time:                     448.9131
+      avg_last_visit_time:                      661.2517
+      num_replay_batches:                       671
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0044505945
+    avg_actor_approxkl3:                        0.004008755
+    avg_actor_clipfrac:                         0.10249634
+    avg_actor_loss:                             0.0041965223
+    avg_advantage:                              -0.002956511
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0003781819
+    avg_entropy:                                0.19916806
+    avg_grad_norm_pre_clip:                     5.303225
+    avg_loss:                                   0.0041864454
+    max:
+      max_actor_approxkl1:                      0.01514821
+      max_actor_approxkl3:                      0.0116418265
+      max_actor_clipfrac:                       0.17553711
+      max_actor_loss:                           0.011932253
+      max_advantage:                            0.08030659
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.000878044
+      max_entropy:                              0.21870413
+      max_grad_norm_pre_clip:                   9.59761
+      max_loss:                                 0.011827966
+    std:
+      std_actor_approxkl1:                      0.0038336474
+      std_actor_approxkl3:                      0.0030903684
+      std_actor_clipfrac:                       0.04264867
+      std_actor_loss:                           0.0032382088
+      std_advantage:                            0.026896846
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00028180165
+      std_entropy:                              0.0074931383
+      std_grad_norm_pre_clip:                   2.198191
+      std_loss:                                 0.0032529314
+  eval-batch-train:
+    avg_avg_episode_length:                     7.8329983
+    avg_avg_return:                             0.97991896
+    avg_benchmark_regret:                       0.01288021
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.1984253
+    proxy_corner:
+      avg_avg_return:                           0.008048197
+      avg_reward_per_step:                      0.0016098022
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.623016
+    avg_avg_return:                             0.973981
+    avg_benchmark_regret:                       0.016258754
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1396637
+    proxy_corner:
+      avg_avg_return:                           0.973981
+      avg_reward_per_step:                      0.1396637
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.648688
+    avg_avg_return:                             0.98074555
+    avg_benchmark_regret:                       0.012091908
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.18828583
+    proxy_corner:
+      avg_avg_return:                           0.0044724094
+      avg_reward_per_step:                      0.0010223389
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.285706
+    avg_avg_return:                             0.0025566933
+    avg_benchmark_regret:                       0.9820785
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        3.0517578e-05
+    proxy_corner:
+      avg_avg_return:                           0.0025566933
+      avg_reward_per_step:                      3.0517578e-05
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        44040192
+    env-step-replay-before:                     22511616
+    env-step-used-before:                       44040192
+    ppo-update-before:                          26880
+    env-step-generate-before:                   21528576
+    env-step-all-after:                         44072960
+    env-step-replay-after:                      22511616
+    env-step-used-after:                        44072960
+    ppo-update-after:                           26900
+    env-step-generate-after:                    21561344
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     8.009755
+    avg_avg_return:                             0.94347554
+    avg_reward_per_step:                        0.17633057
+    proxy_corner:
+      avg_avg_return:                           0.011766899
+      avg_reward_per_step:                      0.0010070801
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     8.009755
+    avg_avg_return:                             0.94347554
+    avg_reward_per_step:                        0.17633057
+    proxy_corner:
+      avg_avg_return:                           0.011766899
+      avg_reward_per_step:                      0.0010070801
+  perf:
+    env_steps_per_second:                       360453.45812097896
+    ppo_updates_per_second:                     40.956051005763165
+    eval-batch-train-duration:                  0.5146245341747999
+    eval-batch-orig-duration:                   0.4813806312158704
+    eval-batch-shift-duration:                  0.4499991424381733
+    eval-batch-tree-duration:                   0.4856385700404644
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.050049
+      cheese-corne_dist_wavg:                   9.107192
+      levels_cheese_is_in_corner_avg:           0.029785156
+      mouse-cheese_dist_finite_avg:             14.862549
+      mouse-cheese_dist_finite_wavg:            16.783934
+      mouse-cheese_dist_solvable_avg:           14.862549
+      mouse-cheese_dist_solvable_wavg:          16.783937
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12086428
+    visit_patterns:
+      avg_first_visit_time:                     461.30664
+      avg_last_visit_time:                      677.2739
+      num_replay_batches:                       687
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0021114454
+    avg_actor_approxkl3:                        0.0016662115
+    avg_actor_clipfrac:                         0.060162354
+    avg_actor_loss:                             0.0013831387
+    avg_advantage:                              0.0036019625
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0002678449
+    avg_entropy:                                0.17297444
+    avg_grad_norm_pre_clip:                     3.6527805
+    avg_loss:                                   0.0013440868
+    max:
+      max_actor_approxkl1:                      0.0068363356
+      max_actor_approxkl3:                      0.0053866403
+      max_actor_clipfrac:                       0.1138916
+      max_actor_loss:                           0.0046332525
+      max_advantage:                            0.08560252
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00029761437
+      max_entropy:                              0.20042992
+      max_grad_norm_pre_clip:                   6.439553
+      max_loss:                                 0.004579157
+    std:
+      std_actor_approxkl1:                      0.0018950135
+      std_actor_approxkl3:                      0.001238744
+      std_actor_clipfrac:                       0.028513474
+      std_actor_loss:                           0.0016500589
+      std_advantage:                            0.023117539
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.8352848e-05
+      std_entropy:                              0.0135876415
+      std_grad_norm_pre_clip:                   1.4869354
+      std_loss:                                 0.0016445651
+  eval-batch-train:
+    avg_avg_episode_length:                     7.8092666
+    avg_avg_return:                             0.9799981
+    avg_benchmark_regret:                       0.012801059
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19585419
+    proxy_corner:
+      avg_avg_return:                           0.007895776
+      avg_reward_per_step:                      0.0015869141
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.705168
+    avg_avg_return:                             0.97408295
+    avg_benchmark_regret:                       0.016156865
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13845825
+    proxy_corner:
+      avg_avg_return:                           0.97408295
+      avg_reward_per_step:                      0.13845825
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.594952
+    avg_avg_return:                             0.98152405
+    avg_benchmark_regret:                       0.011313396
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.18965149
+    proxy_corner:
+      avg_avg_return:                           0.0041670525
+      avg_reward_per_step:                      0.0009689331
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.047615
+    avg_avg_return:                             0.0051940456
+    avg_benchmark_regret:                       0.97944117
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        9.918213e-05
+    proxy_corner:
+      avg_avg_return:                           0.0051940456
+      avg_reward_per_step:                      9.918213e-05
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        45088768
+    env-step-replay-before:                     22937600
+    env-step-used-before:                       45088768
+    ppo-update-before:                          27520
+    env-step-generate-before:                   22151168
+    env-step-all-after:                         45121536
+    env-step-replay-after:                      22937600
+    env-step-used-after:                        45121536
+    ppo-update-after:                           27540
+    env-step-generate-after:                    22183936
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     8.03522
+    avg_avg_return:                             0.9431162
+    avg_reward_per_step:                        0.17837524
+    proxy_corner:
+      avg_avg_return:                           0.010365553
+      avg_reward_per_step:                      0.00079345703
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     8.03522
+    avg_avg_return:                             0.9431162
+    avg_reward_per_step:                        0.17837524
+    proxy_corner:
+      avg_avg_return:                           0.010365553
+      avg_reward_per_step:                      0.00079345703
+  perf:
+    env_steps_per_second:                       323077.1836840659
+    ppo_updates_per_second:                     42.084135365566326
+    eval-batch-train-duration:                  0.5077054053544998
+    eval-batch-orig-duration:                   0.40523047279566526
+    eval-batch-shift-duration:                  0.43413251638412476
+    eval-batch-tree-duration:                   0.45955124124884605
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.96875
+      cheese-corne_dist_wavg:                   9.097813
+      levels_cheese_is_in_corner_avg:           0.030517578
+      mouse-cheese_dist_finite_avg:             14.912109
+      mouse-cheese_dist_finite_wavg:            15.88905
+      mouse-cheese_dist_solvable_avg:           14.912109
+      mouse-cheese_dist_solvable_wavg:          15.889051
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12170896
+    visit_patterns:
+      avg_first_visit_time:                     474.7041
+      avg_last_visit_time:                      690.33936
+      num_replay_batches:                       700
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0020866077
+    avg_actor_approxkl3:                        0.0019812852
+    avg_actor_clipfrac:                         0.06361695
+    avg_actor_loss:                             0.0010646876
+    avg_advantage:                              0.0042058877
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00026594545
+    avg_entropy:                                0.16968222
+    avg_grad_norm_pre_clip:                     3.6880004
+    avg_loss:                                   0.001027978
+    max:
+      max_actor_approxkl1:                      0.007289625
+      max_actor_approxkl3:                      0.0067203795
+      max_actor_clipfrac:                       0.13745117
+      max_actor_loss:                           0.0036156196
+      max_advantage:                            0.10901207
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00032690095
+      max_entropy:                              0.20400712
+      max_grad_norm_pre_clip:                   6.2049804
+      max_loss:                                 0.0035686682
+    std:
+      std_actor_approxkl1:                      0.0018508084
+      std_actor_approxkl3:                      0.0016082964
+      std_actor_clipfrac:                       0.031949893
+      std_actor_loss:                           0.0012584096
+      std_advantage:                            0.022536684
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.01098e-05
+      std_entropy:                              0.018670479
+      std_grad_norm_pre_clip:                   1.3399744
+      std_loss:                                 0.0012546503
+  eval-batch-train:
+    avg_avg_episode_length:                     7.6443343
+    avg_avg_return:                             0.98067844
+    avg_benchmark_regret:                       0.01212076
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19668579
+    proxy_corner:
+      avg_avg_return:                           0.007891279
+      avg_reward_per_step:                      0.0015869141
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.7183895
+    avg_avg_return:                             0.9731947
+    avg_benchmark_regret:                       0.017045118
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13746643
+    proxy_corner:
+      avg_avg_return:                           0.9731947
+      avg_reward_per_step:                      0.13746643
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.547433
+    avg_avg_return:                             0.9815139
+    avg_benchmark_regret:                       0.011323581
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.18943024
+    proxy_corner:
+      avg_avg_return:                           0.0040154983
+      avg_reward_per_step:                      0.0009536743
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.33333
+    avg_avg_return:                             0.0014881198
+    avg_benchmark_regret:                       0.983147
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        1.5258789e-05
+    proxy_corner:
+      avg_avg_return:                           0.0014881198
+      avg_reward_per_step:                      1.5258789e-05
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        46137344
+    env-step-replay-before:                     23560192
+    env-step-used-before:                       46137344
+    ppo-update-before:                          28160
+    env-step-generate-before:                   22577152
+    env-step-all-after:                         46170112
+    env-step-replay-after:                      23560192
+    env-step-used-after:                        46170112
+    ppo-update-after:                           28180
+    env-step-generate-after:                    22609920
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.8218937
+    avg_avg_return:                             0.9442991
+    avg_reward_per_step:                        0.18127441
+    proxy_corner:
+      avg_avg_return:                           0.007290003
+      avg_reward_per_step:                      0.0013427734
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.8218937
+    avg_avg_return:                             0.9442991
+    avg_reward_per_step:                        0.18127441
+    proxy_corner:
+      avg_avg_return:                           0.007290003
+      avg_reward_per_step:                      0.0013427734
+  perf:
+    env_steps_per_second:                       246423.286804985
+    ppo_updates_per_second:                     42.23785644316588
+    eval-batch-train-duration:                  0.630574431270361
+    eval-batch-orig-duration:                   0.6112514594569802
+    eval-batch-shift-duration:                  0.6222288217395544
+    eval-batch-tree-duration:                   0.6247572507709265
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.989014
+      cheese-corne_dist_wavg:                   8.188165
+      levels_cheese_is_in_corner_avg:           0.03100586
+      mouse-cheese_dist_finite_avg:             14.970459
+      mouse-cheese_dist_finite_wavg:            13.197729
+      mouse-cheese_dist_solvable_avg:           14.970459
+      mouse-cheese_dist_solvable_wavg:          13.197731
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11687534
+    visit_patterns:
+      avg_first_visit_time:                     483.00732
+      avg_last_visit_time:                      709.2512
+      num_replay_batches:                       719
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012540056
+    avg_actor_approxkl3:                        0.0011362798
+    avg_actor_clipfrac:                         0.0463562
+    avg_actor_loss:                             0.0009642883
+    avg_advantage:                              0.0013068777
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00025474143
+    avg_entropy:                                0.14035062
+    avg_grad_norm_pre_clip:                     2.7519073
+    avg_loss:                                   0.0009513086
+    max:
+      max_actor_approxkl1:                      0.0020658288
+      max_actor_approxkl3:                      0.0021156776
+      max_actor_clipfrac:                       0.075927734
+      max_actor_loss:                           0.0026351037
+      max_advantage:                            0.09076935
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0003114554
+      max_entropy:                              0.15960768
+      max_grad_norm_pre_clip:                   4.8328466
+      max_loss:                                 0.0026371838
+    std:
+      std_actor_approxkl1:                      0.0005949606
+      std_actor_approxkl3:                      0.00048700208
+      std_actor_clipfrac:                       0.016478311
+      std_actor_loss:                           0.00071995176
+      std_advantage:                            0.022533588
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.4727235e-05
+      std_entropy:                              0.013289175
+      std_grad_norm_pre_clip:                   1.003545
+      std_loss:                                 0.00072038174
+  eval-batch-train:
+    avg_avg_episode_length:                     7.5029287
+    avg_avg_return:                             0.98129
+    avg_benchmark_regret:                       0.011509152
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19732666
+    proxy_corner:
+      avg_avg_return:                           0.008005785
+      avg_reward_per_step:                      0.0016021729
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.667786
+    avg_avg_return:                             0.97373664
+    avg_benchmark_regret:                       0.016503232
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13980103
+    proxy_corner:
+      avg_avg_return:                           0.97373664
+      avg_reward_per_step:                      0.13980103
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4348907
+    avg_avg_return:                             0.98150593
+    avg_benchmark_regret:                       0.011331463
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19038391
+    proxy_corner:
+      avg_avg_return:                           0.0040247436
+      avg_reward_per_step:                      0.0009689331
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.33333
+    avg_avg_return:                             0.0
+    avg_benchmark_regret:                       0.98463523
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0
+    proxy_corner:
+      avg_avg_return:                           0.0
+      avg_reward_per_step:                      0.0
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        47185920
+    env-step-replay-before:                     24084480
+    env-step-used-before:                       47185920
+    ppo-update-before:                          28800
+    env-step-generate-before:                   23101440
+    env-step-all-after:                         47218688
+    env-step-replay-after:                      24084480
+    env-step-used-after:                        47218688
+    ppo-update-after:                           28820
+    env-step-generate-after:                    23134208
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     8.372688
+    avg_avg_return:                             0.9399536
+    avg_reward_per_step:                        0.17904663
+    proxy_corner:
+      avg_avg_return:                           0.010336768
+      avg_reward_per_step:                      0.0007324219
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     8.372688
+    avg_avg_return:                             0.9399536
+    avg_reward_per_step:                        0.17904663
+    proxy_corner:
+      avg_avg_return:                           0.010336768
+      avg_reward_per_step:                      0.0007324219
+  perf:
+    env_steps_per_second:                       326633.0063157199
+    ppo_updates_per_second:                     40.617693544755106
+    eval-batch-train-duration:                  0.5302239554002881
+    eval-batch-orig-duration:                   0.47954050824046135
+    eval-batch-shift-duration:                  0.4527173200622201
+    eval-batch-tree-duration:                   0.49011307395994663
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.074951
+      cheese-corne_dist_wavg:                   9.105772
+      levels_cheese_is_in_corner_avg:           0.029541016
+      mouse-cheese_dist_finite_avg:             14.966309
+      mouse-cheese_dist_finite_wavg:            13.198999
+      mouse-cheese_dist_solvable_avg:           14.966309
+      mouse-cheese_dist_solvable_wavg:          13.199001
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1239204
+    visit_patterns:
+      avg_first_visit_time:                     495.1797
+      avg_last_visit_time:                      725.3513
+      num_replay_batches:                       735
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0025369907
+    avg_actor_approxkl3:                        0.0024521095
+    avg_actor_clipfrac:                         0.06470948
+    avg_actor_loss:                             0.0023551383
+    avg_advantage:                              0.0050862175
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00026969003
+    avg_entropy:                                0.1373374
+    avg_grad_norm_pre_clip:                     4.354236
+    avg_loss:                                   0.0023526463
+    max:
+      max_actor_approxkl1:                      0.008243335
+      max_actor_approxkl3:                      0.007365523
+      max_actor_clipfrac:                       0.1081543
+      max_actor_loss:                           0.0048343134
+      max_advantage:                            0.09121394
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0003538702
+      max_entropy:                              0.15314841
+      max_grad_norm_pre_clip:                   6.956009
+      max_loss:                                 0.004831837
+    std:
+      std_actor_approxkl1:                      0.0020030122
+      std_actor_approxkl3:                      0.0018047208
+      std_actor_clipfrac:                       0.025628107
+      std_actor_loss:                           0.0015269268
+      std_advantage:                            0.022488711
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.2201093e-05
+      std_entropy:                              0.009249435
+      std_grad_norm_pre_clip:                   1.7839676
+      std_loss:                                 0.0015262844
+  eval-batch-train:
+    avg_avg_episode_length:                     7.74832
+    avg_avg_return:                             0.9802897
+    avg_benchmark_regret:                       0.012509502
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19663239
+    proxy_corner:
+      avg_avg_return:                           0.008026951
+      avg_reward_per_step:                      0.0016021729
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.954398
+    avg_avg_return:                             0.9729753
+    avg_benchmark_regret:                       0.017264517
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13749695
+    proxy_corner:
+      avg_avg_return:                           0.9729753
+      avg_reward_per_step:                      0.13749695
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.711005
+    avg_avg_return:                             0.980512
+    avg_benchmark_regret:                       0.0123253595
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.18714142
+    proxy_corner:
+      avg_avg_return:                           0.003994339
+      avg_reward_per_step:                      0.0009841919
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.13333
+    avg_avg_return:                             0.0035009275
+    avg_benchmark_regret:                       0.9811343
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        9.918213e-05
+    proxy_corner:
+      avg_avg_return:                           0.0035009275
+      avg_reward_per_step:                      9.918213e-05
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        48234496
+    env-step-replay-before:                     24707072
+    env-step-used-before:                       48234496
+    ppo-update-before:                          29440
+    env-step-generate-before:                   23527424
+    env-step-all-after:                         48267264
+    env-step-replay-after:                      24739840
+    env-step-used-after:                        48267264
+    ppo-update-after:                           29460
+    env-step-generate-after:                    23527424
+  train-replay:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     16.347792
+    avg_avg_return:                             0.87268406
+    avg_reward_per_step:                        0.058563232
+    proxy_corner:
+      avg_avg_return:                           0.029290913
+      avg_reward_per_step:                      0.0019226074
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     16.347792
+    avg_avg_return:                             0.87268406
+    avg_reward_per_step:                        0.058563232
+    proxy_corner:
+      avg_avg_return:                           0.029290913
+      avg_reward_per_step:                      0.0019226074
+  perf:
+    env_steps_per_second:                       355901.7662119334
+    ppo_updates_per_second:                     42.48843440675474
+    eval-batch-train-duration:                  0.5204371921718121
+    eval-batch-orig-duration:                   0.47766338381916285
+    eval-batch-shift-duration:                  0.44866973999887705
+    eval-batch-tree-duration:                   0.47839506808668375
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.113525
+      cheese-corne_dist_wavg:                   10.000551
+      levels_cheese_is_in_corner_avg:           0.030517578
+      mouse-cheese_dist_finite_avg:             15.005127
+      mouse-cheese_dist_finite_wavg:            7.80678
+      mouse-cheese_dist_solvable_avg:           15.005127
+      mouse-cheese_dist_solvable_wavg:          7.806782
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.117979154
+    visit_patterns:
+      avg_first_visit_time:                     504.18433
+      avg_last_visit_time:                      745.3816
+      num_replay_batches:                       755
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0021081145
+    avg_actor_approxkl3:                        0.0016755539
+    avg_actor_clipfrac:                         0.06925049
+    avg_actor_loss:                             0.0012776464
+    avg_advantage:                              -9.073908e-05
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00018369433
+    avg_entropy:                                0.1884714
+    avg_grad_norm_pre_clip:                     4.588692
+    avg_loss:                                   0.0011810224
+    max:
+      max_actor_approxkl1:                      0.0044257157
+      max_actor_approxkl3:                      0.002735624
+      max_actor_clipfrac:                       0.10522461
+      max_actor_loss:                           0.0035950574
+      max_advantage:                            0.08335084
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00019607518
+      max_entropy:                              0.20431122
+      max_grad_norm_pre_clip:                   7.63087
+      max_loss:                                 0.0035041317
+    std:
+      std_actor_approxkl1:                      0.0011318556
+      std_actor_approxkl3:                      0.00067146966
+      std_actor_clipfrac:                       0.024500825
+      std_actor_loss:                           0.0012382814
+      std_advantage:                            0.01894482
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.4415248e-06
+      std_entropy:                              0.009208378
+      std_grad_norm_pre_clip:                   1.8722107
+      std_loss:                                 0.00124116
+  eval-batch-train:
+    avg_avg_episode_length:                     7.654579
+    avg_avg_return:                             0.9812621
+    avg_benchmark_regret:                       0.011537079
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19856262
+    proxy_corner:
+      avg_avg_return:                           0.007784155
+      avg_reward_per_step:                      0.0015792847
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.707558
+    avg_avg_return:                             0.97404027
+    avg_benchmark_regret:                       0.016199503
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13942719
+    proxy_corner:
+      avg_avg_return:                           0.97404027
+      avg_reward_per_step:                      0.13942719
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.723126
+    avg_avg_return:                             0.9800608
+    avg_benchmark_regret:                       0.012776645
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.18717957
+    proxy_corner:
+      avg_avg_return:                           0.0041543236
+      avg_reward_per_step:                      0.0010223389
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.33333
+    avg_avg_return:                             0.0007327936
+    avg_benchmark_regret:                       0.9839024
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        7.6293945e-06
+    proxy_corner:
+      avg_avg_return:                           0.0007327936
+      avg_reward_per_step:                      7.6293945e-06
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        49283072
+    env-step-replay-before:                     25493504
+    env-step-used-before:                       49283072
+    ppo-update-before:                          30080
+    env-step-generate-before:                   23789568
+    env-step-all-after:                         49315840
+    env-step-replay-after:                      25526272
+    env-step-used-after:                        49315840
+    ppo-update-after:                           30100
+    env-step-generate-after:                    23789568
+  train-replay:
+    shift_proportion:                           0.9609375
+    avg_avg_episode_length:                     16.703444
+    avg_avg_return:                             0.8690827
+    avg_reward_per_step:                        0.058258057
+    proxy_corner:
+      avg_avg_return:                           0.05628324
+      avg_reward_per_step:                      0.0033569336
+  train-all:
+    shift_proportion:                           0.9609375
+    avg_avg_episode_length:                     16.703444
+    avg_avg_return:                             0.8690827
+    avg_reward_per_step:                        0.058258057
+    proxy_corner:
+      avg_avg_return:                           0.05628324
+      avg_reward_per_step:                      0.0033569336
+  perf:
+    env_steps_per_second:                       352294.21359100734
+    ppo_updates_per_second:                     41.217792967861996
+    eval-batch-train-duration:                  0.5357339633628726
+    eval-batch-orig-duration:                   0.4596718903630972
+    eval-batch-shift-duration:                  0.45343328826129436
+    eval-batch-tree-duration:                   0.4727223291993141
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.116699
+      cheese-corne_dist_wavg:                   2.8189802
+      levels_cheese_is_in_corner_avg:           0.030517578
+      mouse-cheese_dist_finite_avg:             15.003906
+      mouse-cheese_dist_finite_wavg:            14.101478
+      mouse-cheese_dist_solvable_avg:           15.003906
+      mouse-cheese_dist_solvable_wavg:          14.10148
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11556065
+    visit_patterns:
+      avg_first_visit_time:                     509.78027
+      avg_last_visit_time:                      769.2224
+      num_replay_batches:                       779
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0021377963
+    avg_actor_approxkl3:                        0.0017529726
+    avg_actor_clipfrac:                         0.06780396
+    avg_actor_loss:                             0.0014682625
+    avg_advantage:                              -0.0014549287
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00016958156
+    avg_entropy:                                0.18955152
+    avg_grad_norm_pre_clip:                     4.870947
+    avg_loss:                                   0.0013635018
+    max:
+      max_actor_approxkl1:                      0.0042145904
+      max_actor_approxkl3:                      0.0036477228
+      max_actor_clipfrac:                       0.119506836
+      max_actor_loss:                           0.0039098915
+      max_advantage:                            0.072163045
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0001955099
+      max_entropy:                              0.20560607
+      max_grad_norm_pre_clip:                   8.650336
+      max_loss:                                 0.003815256
+    std:
+      std_actor_approxkl1:                      0.0010549296
+      std_actor_approxkl3:                      0.00089704973
+      std_actor_clipfrac:                       0.025844092
+      std_actor_loss:                           0.0012786948
+      std_advantage:                            0.018267306
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          9.503237e-06
+      std_entropy:                              0.009816957
+      std_grad_norm_pre_clip:                   1.8235549
+      std_loss:                                 0.0012791266
+  eval-batch-train:
+    avg_avg_episode_length:                     7.7062397
+    avg_avg_return:                             0.9804659
+    avg_benchmark_regret:                       0.012333322
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19595337
+    proxy_corner:
+      avg_avg_return:                           0.008416829
+      avg_reward_per_step:                      0.0016326904
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.618179
+    avg_avg_return:                             0.97477674
+    avg_benchmark_regret:                       0.015463091
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14059448
+    proxy_corner:
+      avg_avg_return:                           0.97477674
+      avg_reward_per_step:                      0.14059448
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.5257697
+    avg_avg_return:                             0.9807259
+    avg_benchmark_regret:                       0.01211153
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.18992615
+    proxy_corner:
+      avg_avg_return:                           0.0041801636
+      avg_reward_per_step:                      0.0010375977
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.33333
+    avg_avg_return:                             0.0007262248
+    avg_benchmark_regret:                       0.983909
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        7.6293945e-06
+    proxy_corner:
+      avg_avg_return:                           0.0007262248
+      avg_reward_per_step:                      7.6293945e-06
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        50331648
+    env-step-replay-before:                     26116096
+    env-step-used-before:                       50331648
+    ppo-update-before:                          30720
+    env-step-generate-before:                   24215552
+    env-step-all-after:                         50364416
+    env-step-replay-after:                      26116096
+    env-step-used-after:                        50364416
+    ppo-update-after:                           30740
+    env-step-generate-after:                    24248320
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.8326902
+    avg_avg_return:                             0.9436753
+    avg_reward_per_step:                        0.17697144
+    proxy_corner:
+      avg_avg_return:                           0.007292779
+      avg_reward_per_step:                      0.0007019043
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.8326902
+    avg_avg_return:                             0.9436753
+    avg_reward_per_step:                        0.17697144
+    proxy_corner:
+      avg_avg_return:                           0.007292779
+      avg_reward_per_step:                      0.0007019043
+  perf:
+    env_steps_per_second:                       321230.484615232
+    ppo_updates_per_second:                     42.16789271083732
+    eval-batch-train-duration:                  0.5206742128357291
+    eval-batch-orig-duration:                   0.47377862222492695
+    eval-batch-shift-duration:                  0.47055076342076063
+    eval-batch-tree-duration:                   0.464422132819891
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.141602
+      cheese-corne_dist_wavg:                   2.815874
+      levels_cheese_is_in_corner_avg:           0.030273438
+      mouse-cheese_dist_finite_avg:             15.026611
+      mouse-cheese_dist_finite_wavg:            16.802015
+      mouse-cheese_dist_solvable_avg:           15.026611
+      mouse-cheese_dist_solvable_wavg:          16.802017
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11499069
+    visit_patterns:
+      avg_first_visit_time:                     519.8037
+      avg_last_visit_time:                      787.2883
+      num_replay_batches:                       797
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0046866536
+    avg_actor_approxkl3:                        0.004531344
+    avg_actor_clipfrac:                         0.08706055
+    avg_actor_loss:                             0.005402558
+    avg_advantage:                              0.0063179745
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0002963952
+    avg_entropy:                                0.14126514
+    avg_grad_norm_pre_clip:                     5.721456
+    avg_loss:                                   0.0054094903
+    max:
+      max_actor_approxkl1:                      0.01349281
+      max_actor_approxkl3:                      0.012789961
+      max_actor_clipfrac:                       0.16516113
+      max_actor_loss:                           0.015181053
+      max_advantage:                            0.0858143
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0004685989
+      max_entropy:                              0.17897937
+      max_grad_norm_pre_clip:                   12.350267
+      max_loss:                                 0.015137206
+    std:
+      std_actor_approxkl1:                      0.0031869228
+      std_actor_approxkl3:                      0.0030040967
+      std_actor_clipfrac:                       0.033667084
+      std_actor_loss:                           0.0035403853
+      std_advantage:                            0.022713067
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.0509432e-05
+      std_entropy:                              0.013344525
+      std_grad_norm_pre_clip:                   2.1364608
+      std_loss:                                 0.0035307715
+  eval-batch-train:
+    avg_avg_episode_length:                     7.498556
+    avg_avg_return:                             0.9811692
+    avg_benchmark_regret:                       0.011629894
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19862366
+    proxy_corner:
+      avg_avg_return:                           0.0077651525
+      avg_reward_per_step:                      0.0015869141
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.568327
+    avg_avg_return:                             0.97518593
+    avg_benchmark_regret:                       0.015053907
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1406784
+    proxy_corner:
+      avg_avg_return:                           0.97518593
+      avg_reward_per_step:                      0.1406784
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4418964
+    avg_avg_return:                             0.9811208
+    avg_benchmark_regret:                       0.011716596
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19100189
+    proxy_corner:
+      avg_avg_return:                           0.0039810105
+      avg_reward_per_step:                      0.0010070801
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.33333
+    avg_avg_return:                             0.0007581494
+    avg_benchmark_regret:                       0.98387706
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        7.6293945e-06
+    proxy_corner:
+      avg_avg_return:                           0.0007581494
+      avg_reward_per_step:                      7.6293945e-06
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        51380224
+    env-step-replay-before:                     26640384
+    env-step-used-before:                       51380224
+    ppo-update-before:                          31360
+    env-step-generate-before:                   24739840
+    env-step-all-after:                         51412992
+    env-step-replay-after:                      26640384
+    env-step-used-after:                        51412992
+    ppo-update-after:                           31380
+    env-step-generate-after:                    24772608
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.9829235
+    avg_avg_return:                             0.9458387
+    avg_reward_per_step:                        0.1796875
+    proxy_corner:
+      avg_avg_return:                           0.012747375
+      avg_reward_per_step:                      0.0016479492
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.9829235
+    avg_avg_return:                             0.9458387
+    avg_reward_per_step:                        0.1796875
+    proxy_corner:
+      avg_avg_return:                           0.012747375
+      avg_reward_per_step:                      0.0016479492
+  perf:
+    env_steps_per_second:                       299912.81522355805
+    ppo_updates_per_second:                     41.60696542523001
+    eval-batch-train-duration:                  0.530954566784203
+    eval-batch-orig-duration:                   0.4755782065913081
+    eval-batch-shift-duration:                  0.44909458700567484
+    eval-batch-tree-duration:                   0.4759259456768632
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.133057
+      cheese-corne_dist_wavg:                   14.520637
+      levels_cheese_is_in_corner_avg:           0.03173828
+      mouse-cheese_dist_finite_avg:             15.064697
+      mouse-cheese_dist_finite_wavg:            10.507814
+      mouse-cheese_dist_solvable_avg:           15.064697
+      mouse-cheese_dist_solvable_wavg:          10.507817
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.114961475
+    visit_patterns:
+      avg_first_visit_time:                     528.96387
+      avg_last_visit_time:                      803.2339
+      num_replay_batches:                       813
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0025920083
+    avg_actor_approxkl3:                        0.0025041907
+    avg_actor_clipfrac:                         0.063250735
+    avg_actor_loss:                             0.0024305917
+    avg_advantage:                              0.0077930177
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00024590184
+    avg_entropy:                                0.13454849
+    avg_grad_norm_pre_clip:                     4.058748
+    avg_loss:                                   0.0024189942
+    max:
+      max_actor_approxkl1:                      0.009687853
+      max_actor_approxkl3:                      0.008270479
+      max_actor_clipfrac:                       0.110961914
+      max_actor_loss:                           0.0077418042
+      max_advantage:                            0.08050287
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00030328782
+      max_entropy:                              0.1588669
+      max_grad_norm_pre_clip:                   7.3416266
+      max_loss:                                 0.0077070408
+    std:
+      std_actor_approxkl1:                      0.0021533673
+      std_actor_approxkl3:                      0.0018739426
+      std_actor_clipfrac:                       0.027633125
+      std_actor_loss:                           0.0019964725
+      std_advantage:                            0.021735176
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.0478088e-05
+      std_entropy:                              0.012362871
+      std_grad_norm_pre_clip:                   1.7750965
+      std_loss:                                 0.0019872482
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4367275
+    avg_avg_return:                             0.98142797
+    avg_benchmark_regret:                       0.011371218
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19854736
+    proxy_corner:
+      avg_avg_return:                           0.007834294
+      avg_reward_per_step:                      0.0015945435
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.582985
+    avg_avg_return:                             0.9743651
+    avg_benchmark_regret:                       0.01587472
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14095306
+    proxy_corner:
+      avg_avg_return:                           0.9743651
+      avg_reward_per_step:                      0.14095306
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3522944
+    avg_avg_return:                             0.98147947
+    avg_benchmark_regret:                       0.011357982
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19226074
+    proxy_corner:
+      avg_avg_return:                           0.0040768594
+      avg_reward_per_step:                      0.0010223389
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.22221
+    avg_avg_return:                             0.0046286173
+    avg_benchmark_regret:                       0.9800066
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        6.866455e-05
+    proxy_corner:
+      avg_avg_return:                           0.0046286173
+      avg_reward_per_step:                      6.866455e-05
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        52428800
+    env-step-replay-before:                     27164672
+    env-step-used-before:                       52428800
+    ppo-update-before:                          32000
+    env-step-generate-before:                   25264128
+    env-step-all-after:                         52461568
+    env-step-replay-after:                      27197440
+    env-step-used-after:                        52461568
+    ppo-update-after:                           32020
+    env-step-generate-after:                    25264128
+  train-replay:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     16.760098
+    avg_avg_return:                             0.8705915
+    avg_reward_per_step:                        0.0574646
+    proxy_corner:
+      avg_avg_return:                           0.028764408
+      avg_reward_per_step:                      0.0016174316
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     16.760098
+    avg_avg_return:                             0.8705915
+    avg_reward_per_step:                        0.0574646
+    proxy_corner:
+      avg_avg_return:                           0.028764408
+      avg_reward_per_step:                      0.0016174316
+  perf:
+    env_steps_per_second:                       290462.713327718
+    ppo_updates_per_second:                     42.48261728133903
+    eval-batch-train-duration:                  0.5390412118285894
+    eval-batch-orig-duration:                   0.4817129597067833
+    eval-batch-shift-duration:                  0.4521587463095784
+    eval-batch-tree-duration:                   0.4742760546505451
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.114014
+      cheese-corne_dist_wavg:                   14.511754
+      levels_cheese_is_in_corner_avg:           0.03173828
+      mouse-cheese_dist_finite_avg:             15.109131
+      mouse-cheese_dist_finite_wavg:            12.312545
+      mouse-cheese_dist_solvable_avg:           15.109131
+      mouse-cheese_dist_solvable_wavg:          12.312545
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.117221326
+    visit_patterns:
+      avg_first_visit_time:                     542.12964
+      avg_last_visit_time:                      820.2339
+      num_replay_batches:                       830
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0029347788
+    avg_actor_approxkl3:                        0.0024143553
+    avg_actor_clipfrac:                         0.0800354
+    avg_actor_loss:                             0.0021194264
+    avg_advantage:                              -0.002256022
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0001836107
+    avg_entropy:                                0.18599401
+    avg_grad_norm_pre_clip:                     5.1982245
+    avg_loss:                                   0.002025238
+    max:
+      max_actor_approxkl1:                      0.006437948
+      max_actor_approxkl3:                      0.0044959746
+      max_actor_clipfrac:                       0.13513184
+      max_actor_loss:                           0.005189595
+      max_advantage:                            0.09214711
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00021243322
+      max_entropy:                              0.21574521
+      max_grad_norm_pre_clip:                   9.091887
+      max_loss:                                 0.005080527
+    std:
+      std_actor_approxkl1:                      0.0017849611
+      std_actor_approxkl3:                      0.001401037
+      std_actor_clipfrac:                       0.038908314
+      std_actor_loss:                           0.0020077967
+      std_advantage:                            0.019094586
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.2404061e-05
+      std_entropy:                              0.013200808
+      std_grad_norm_pre_clip:                   2.4641287
+      std_loss:                                 0.0020062625
+  eval-batch-train:
+    avg_avg_episode_length:                     7.49065
+    avg_avg_return:                             0.98158586
+    avg_benchmark_regret:                       0.011213263
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19857025
+    proxy_corner:
+      avg_avg_return:                           0.0078001176
+      avg_reward_per_step:                      0.0015945435
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.548216
+    avg_avg_return:                             0.9742027
+    avg_benchmark_regret:                       0.016037123
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14128876
+    proxy_corner:
+      avg_avg_return:                           0.9742027
+      avg_reward_per_step:                      0.14128876
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.5248013
+    avg_avg_return:                             0.9813928
+    avg_benchmark_regret:                       0.01144466
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19003296
+    proxy_corner:
+      avg_avg_return:                           0.0044954596
+      avg_reward_per_step:                      0.0011749268
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.013885
+    avg_avg_return:                             0.006651827
+    avg_benchmark_regret:                       0.97798336
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.00015258789
+    proxy_corner:
+      avg_avg_return:                           0.006651827
+      avg_reward_per_step:                      0.00015258789
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        53477376
+    env-step-replay-before:                     27623424
+    env-step-used-before:                       53477376
+    ppo-update-before:                          32640
+    env-step-generate-before:                   25853952
+    env-step-all-after:                         53510144
+    env-step-replay-after:                      27623424
+    env-step-used-after:                        53510144
+    ppo-update-after:                           32660
+    env-step-generate-after:                    25886720
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.9825883
+    avg_avg_return:                             0.9443964
+    avg_reward_per_step:                        0.18936157
+    proxy_corner:
+      avg_avg_return:                           0.015049964
+      avg_reward_per_step:                      0.0014038086
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.9825883
+    avg_avg_return:                             0.9443964
+    avg_reward_per_step:                        0.18936157
+    proxy_corner:
+      avg_avg_return:                           0.015049964
+      avg_reward_per_step:                      0.0014038086
+  perf:
+    env_steps_per_second:                       351460.34091923846
+    ppo_updates_per_second:                     41.282267638714124
+    eval-batch-train-duration:                  0.5312115792185068
+    eval-batch-orig-duration:                   0.4723008722066879
+    eval-batch-shift-duration:                  0.4557574950158596
+    eval-batch-tree-duration:                   0.5359416492283344
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.1171875
+      cheese-corne_dist_wavg:                   3.7203438
+      levels_cheese_is_in_corner_avg:           0.031982422
+      mouse-cheese_dist_finite_avg:             15.118652
+      mouse-cheese_dist_finite_wavg:            15.007338
+      mouse-cheese_dist_solvable_avg:           15.118652
+      mouse-cheese_dist_solvable_wavg:          15.0073395
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12183925
+    visit_patterns:
+      avg_first_visit_time:                     557.791
+      avg_last_visit_time:                      833.35156
+      num_replay_batches:                       843
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0018824966
+    avg_actor_approxkl3:                        0.0013882281
+    avg_actor_clipfrac:                         0.048718262
+    avg_actor_loss:                             0.00084012357
+    avg_advantage:                              0.004675155
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00027212425
+    avg_entropy:                                0.1348004
+    avg_grad_norm_pre_clip:                     3.0492914
+    avg_loss:                                   0.00084138504
+    max:
+      max_actor_approxkl1:                      0.0040561864
+      max_actor_approxkl3:                      0.0031863726
+      max_actor_clipfrac:                       0.08618164
+      max_actor_loss:                           0.0029212045
+      max_advantage:                            0.105513275
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0003156565
+      max_entropy:                              0.16050333
+      max_grad_norm_pre_clip:                   5.6106777
+      max_loss:                                 0.0028973157
+    std:
+      std_actor_approxkl1:                      0.0010967113
+      std_actor_approxkl3:                      0.000733064
+      std_actor_clipfrac:                       0.019644976
+      std_actor_loss:                           0.0008679484
+      std_advantage:                            0.022662986
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.9608638e-05
+      std_entropy:                              0.012769973
+      std_grad_norm_pre_clip:                   1.1617341
+      std_loss:                                 0.00086794654
+  eval-batch-train:
+    avg_avg_episode_length:                     7.5193167
+    avg_avg_return:                             0.98170257
+    avg_benchmark_regret:                       0.011096582
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.1974945
+    proxy_corner:
+      avg_avg_return:                           0.0077543473
+      avg_reward_per_step:                      0.0015792847
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.700586
+    avg_avg_return:                             0.97369325
+    avg_benchmark_regret:                       0.016546631
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14001465
+    proxy_corner:
+      avg_avg_return:                           0.97369325
+      avg_reward_per_step:                      0.14001465
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.5099936
+    avg_avg_return:                             0.9811957
+    avg_benchmark_regret:                       0.011641756
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19075012
+    proxy_corner:
+      avg_avg_return:                           0.0041157436
+      avg_reward_per_step:                      0.0010375977
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.33333
+    avg_avg_return:                             0.0007082856
+    avg_benchmark_regret:                       0.9839269
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        7.6293945e-06
+    proxy_corner:
+      avg_avg_return:                           0.0007082856
+      avg_reward_per_step:                      7.6293945e-06
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        54525952
+    env-step-replay-before:                     28049408
+    env-step-used-before:                       54525952
+    ppo-update-before:                          33280
+    env-step-generate-before:                   26476544
+    env-step-all-after:                         54558720
+    env-step-replay-after:                      28049408
+    env-step-used-after:                        54558720
+    ppo-update-after:                           33300
+    env-step-generate-after:                    26509312
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.310726
+    avg_avg_return:                             0.94975996
+    avg_reward_per_step:                        0.20254517
+    proxy_corner:
+      avg_avg_return:                           0.0036167693
+      avg_reward_per_step:                      0.0004272461
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.310726
+    avg_avg_return:                             0.94975996
+    avg_reward_per_step:                        0.20254517
+    proxy_corner:
+      avg_avg_return:                           0.0036167693
+      avg_reward_per_step:                      0.0004272461
+  perf:
+    env_steps_per_second:                       301113.09070552525
+    ppo_updates_per_second:                     41.9413650486501
+    eval-batch-train-duration:                  0.5287092290818691
+    eval-batch-orig-duration:                   0.47317484114319086
+    eval-batch-shift-duration:                  0.45077081210911274
+    eval-batch-tree-duration:                   0.470432017929852
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.179443
+      cheese-corne_dist_wavg:                   3.719343
+      levels_cheese_is_in_corner_avg:           0.032226562
+      mouse-cheese_dist_finite_avg:             15.165283
+      mouse-cheese_dist_finite_wavg:            13.218589
+      mouse-cheese_dist_solvable_avg:           15.165283
+      mouse-cheese_dist_solvable_wavg:          13.218589
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12021642
+    visit_patterns:
+      avg_first_visit_time:                     571.05615
+      avg_last_visit_time:                      846.29736
+      num_replay_batches:                       856
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0030102448
+    avg_actor_approxkl3:                        0.0026670524
+    avg_actor_clipfrac:                         0.06347046
+    avg_actor_loss:                             0.002651089
+    avg_advantage:                              0.0022609432
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00024431138
+    avg_entropy:                                0.1246584
+    avg_grad_norm_pre_clip:                     4.7615767
+    avg_loss:                                   0.0026485862
+    max:
+      max_actor_approxkl1:                      0.00532079
+      max_actor_approxkl3:                      0.006083889
+      max_actor_clipfrac:                       0.10925293
+      max_actor_loss:                           0.0069060735
+      max_advantage:                            0.07992464
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00030015357
+      max_entropy:                              0.1458771
+      max_grad_norm_pre_clip:                   8.054624
+      max_loss:                                 0.0068769646
+    std:
+      std_actor_approxkl1:                      0.0016614376
+      std_actor_approxkl3:                      0.0015527986
+      std_actor_clipfrac:                       0.02593474
+      std_actor_loss:                           0.0018958655
+      std_advantage:                            0.022203559
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.4012617e-05
+      std_entropy:                              0.012242788
+      std_grad_norm_pre_clip:                   1.89078
+      std_loss:                                 0.0018892987
+  eval-batch-train:
+    avg_avg_episode_length:                     7.5342684
+    avg_avg_return:                             0.9814912
+    avg_benchmark_regret:                       0.011307929
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19851685
+    proxy_corner:
+      avg_avg_return:                           0.0076187197
+      avg_reward_per_step:                      0.0015563965
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.673753
+    avg_avg_return:                             0.9736234
+    avg_benchmark_regret:                       0.016616443
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14013672
+    proxy_corner:
+      avg_avg_return:                           0.9736234
+      avg_reward_per_step:                      0.14013672
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4721518
+    avg_avg_return:                             0.9816557
+    avg_benchmark_regret:                       0.011181733
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19134521
+    proxy_corner:
+      avg_avg_return:                           0.0040805563
+      avg_reward_per_step:                      0.0010375977
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.25
+    avg_avg_return:                             0.0021117139
+    avg_benchmark_regret:                       0.9825235
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        3.0517578e-05
+    proxy_corner:
+      avg_avg_return:                           0.0021117139
+      avg_reward_per_step:                      3.0517578e-05
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        55574528
+    env-step-replay-before:                     28442624
+    env-step-used-before:                       55574528
+    ppo-update-before:                          33920
+    env-step-generate-before:                   27131904
+    env-step-all-after:                         55607296
+    env-step-replay-after:                      28475392
+    env-step-used-after:                        55607296
+    ppo-update-after:                           33940
+    env-step-generate-after:                    27131904
+  train-replay:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     17.402924
+    avg_avg_return:                             0.86622584
+    avg_reward_per_step:                        0.055511475
+    proxy_corner:
+      avg_avg_return:                           0.02392321
+      avg_reward_per_step:                      0.0014648438
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     17.402924
+    avg_avg_return:                             0.86622584
+    avg_reward_per_step:                        0.055511475
+    proxy_corner:
+      avg_avg_return:                           0.02392321
+      avg_reward_per_step:                      0.0014648438
+  perf:
+    env_steps_per_second:                       300528.6827253406
+    ppo_updates_per_second:                     42.06349942864405
+    eval-batch-train-duration:                  0.5413265647366643
+    eval-batch-orig-duration:                   0.4832795010879636
+    eval-batch-shift-duration:                  0.4863756261765957
+    eval-batch-tree-duration:                   0.4933133078739047
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.17749
+      cheese-corne_dist_wavg:                   3.7145743
+      levels_cheese_is_in_corner_avg:           0.033203125
+      mouse-cheese_dist_finite_avg:             15.235352
+      mouse-cheese_dist_finite_wavg:            16.819757
+      mouse-cheese_dist_solvable_avg:           15.235352
+      mouse-cheese_dist_solvable_wavg:          16.81976
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11888375
+    visit_patterns:
+      avg_first_visit_time:                     582.0161
+      avg_last_visit_time:                      859.2539
+      num_replay_batches:                       869
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0017863851
+    avg_actor_approxkl3:                        0.0016685164
+    avg_actor_clipfrac:                         0.0715149
+    avg_actor_loss:                             0.0010815727
+    avg_advantage:                              -0.003547673
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00020212171
+    avg_entropy:                                0.19817653
+    avg_grad_norm_pre_clip:                     4.336822
+    avg_loss:                                   0.0009844571
+    max:
+      max_actor_approxkl1:                      0.0039879093
+      max_actor_approxkl3:                      0.002802142
+      max_actor_clipfrac:                       0.10522461
+      max_actor_loss:                           0.0031984914
+      max_advantage:                            0.08484244
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00022195619
+      max_entropy:                              0.21480533
+      max_grad_norm_pre_clip:                   7.530354
+      max_loss:                                 0.0030935132
+    std:
+      std_actor_approxkl1:                      0.0010753146
+      std_actor_approxkl3:                      0.0006601598
+      std_actor_clipfrac:                       0.022888927
+      std_actor_loss:                           0.001056886
+      std_advantage:                            0.020099133
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.1282377e-05
+      std_entropy:                              0.011003894
+      std_grad_norm_pre_clip:                   1.2387694
+      std_loss:                                 0.0010594212
+  eval-batch-train:
+    avg_avg_episode_length:                     7.540824
+    avg_avg_return:                             0.981075
+    avg_benchmark_regret:                       0.011724172
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19798279
+    proxy_corner:
+      avg_avg_return:                           0.0076845116
+      avg_reward_per_step:                      0.0015792847
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.614298
+    avg_avg_return:                             0.9746065
+    avg_benchmark_regret:                       0.015633326
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14068604
+    proxy_corner:
+      avg_avg_return:                           0.9746065
+      avg_reward_per_step:                      0.14068604
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.523221
+    avg_avg_return:                             0.9814762
+    avg_benchmark_regret:                       0.01136127
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19141388
+    proxy_corner:
+      avg_avg_return:                           0.003986456
+      avg_reward_per_step:                      0.0010070801
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.33333
+    avg_avg_return:                             0.0
+    avg_benchmark_regret:                       0.98463523
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0
+    proxy_corner:
+      avg_avg_return:                           0.0
+      avg_reward_per_step:                      0.0
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        56623104
+    env-step-replay-before:                     28966912
+    env-step-used-before:                       56623104
+    ppo-update-before:                          34560
+    env-step-generate-before:                   27656192
+    env-step-all-after:                         56655872
+    env-step-replay-after:                      28999680
+    env-step-used-after:                        56655872
+    ppo-update-after:                           34580
+    env-step-generate-after:                    27656192
+  train-replay:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     16.704395
+    avg_avg_return:                             0.86829305
+    avg_reward_per_step:                        0.05722046
+    proxy_corner:
+      avg_avg_return:                           0.015627418
+      avg_reward_per_step:                      0.0009460449
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     16.704395
+    avg_avg_return:                             0.86829305
+    avg_reward_per_step:                        0.05722046
+    proxy_corner:
+      avg_avg_return:                           0.015627418
+      avg_reward_per_step:                      0.0009460449
+  perf:
+    env_steps_per_second:                       306374.8793370572
+    ppo_updates_per_second:                     42.001702761017576
+    eval-batch-train-duration:                  0.535180913284421
+    eval-batch-orig-duration:                   0.4964532433077693
+    eval-batch-shift-duration:                  0.4841123027727008
+    eval-batch-tree-duration:                   0.5084190368652344
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.180664
+      cheese-corne_dist_wavg:                   17.200356
+      levels_cheese_is_in_corner_avg:           0.032470703
+      mouse-cheese_dist_finite_avg:             15.255615
+      mouse-cheese_dist_finite_wavg:            13.23005
+      mouse-cheese_dist_solvable_avg:           15.255615
+      mouse-cheese_dist_solvable_wavg:          13.230052
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.119106054
+    visit_patterns:
+      avg_first_visit_time:                     594.3926
+      avg_last_visit_time:                      875.33545
+      num_replay_batches:                       885
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0020894087
+    avg_actor_approxkl3:                        0.001440765
+    avg_actor_clipfrac:                         0.058520507
+    avg_actor_loss:                             0.00072649005
+    avg_advantage:                              -0.0013032737
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00018145696
+    avg_entropy:                                0.20170477
+    avg_grad_norm_pre_clip:                     3.5906625
+    avg_loss:                                   0.0006155139
+    max:
+      max_actor_approxkl1:                      0.0037110797
+      max_actor_approxkl3:                      0.0029437602
+      max_actor_clipfrac:                       0.09484863
+      max_actor_loss:                           0.0029896772
+      max_advantage:                            0.085607946
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00020117196
+      max_entropy:                              0.21874806
+      max_grad_norm_pre_clip:                   5.710108
+      max_loss:                                 0.0028942064
+    std:
+      std_actor_approxkl1:                      0.00092032336
+      std_actor_approxkl3:                      0.00063590734
+      std_actor_clipfrac:                       0.02057203
+      std_actor_loss:                           0.0012055518
+      std_advantage:                            0.01917965
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          9.041238e-06
+      std_entropy:                              0.010311124
+      std_grad_norm_pre_clip:                   1.1596819
+      std_loss:                                 0.0012094273
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4911337
+    avg_avg_return:                             0.98098135
+    avg_benchmark_regret:                       0.011817796
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19919586
+    proxy_corner:
+      avg_avg_return:                           0.0076224916
+      avg_reward_per_step:                      0.0015640259
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.57741
+    avg_avg_return:                             0.97508764
+    avg_benchmark_regret:                       0.01515219
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14111328
+    proxy_corner:
+      avg_avg_return:                           0.97508764
+      avg_reward_per_step:                      0.14111328
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.5066447
+    avg_avg_return:                             0.980978
+    avg_benchmark_regret:                       0.011859406
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19104767
+    proxy_corner:
+      avg_avg_return:                           0.004092836
+      avg_reward_per_step:                      0.0010223389
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.33333
+    avg_avg_return:                             0.000759668
+    avg_benchmark_regret:                       0.9838755
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        7.6293945e-06
+    proxy_corner:
+      avg_avg_return:                           0.000759668
+      avg_reward_per_step:                      7.6293945e-06
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        57671680
+    env-step-replay-before:                     29458432
+    env-step-used-before:                       57671680
+    ppo-update-before:                          35200
+    env-step-generate-before:                   28213248
+    env-step-all-after:                         57704448
+    env-step-replay-after:                      29458432
+    env-step-used-after:                        57704448
+    ppo-update-after:                           35220
+    env-step-generate-after:                    28246016
+  train-generate:
+    shift_proportion:                           1.0
+    avg_avg_episode_length:                     7.8047433
+    avg_avg_return:                             0.9434977
+    avg_reward_per_step:                        0.17382812
+    proxy_corner:
+      avg_avg_return:                           0.0029820064
+      avg_reward_per_step:                      0.00039672852
+  train-all:
+    shift_proportion:                           1.0
+    avg_avg_episode_length:                     7.8047433
+    avg_avg_return:                             0.9434977
+    avg_reward_per_step:                        0.17382812
+    proxy_corner:
+      avg_avg_return:                           0.0029820064
+      avg_reward_per_step:                      0.00039672852
+  perf:
+    env_steps_per_second:                       278367.1962541716
+    ppo_updates_per_second:                     41.327800222028685
+    eval-batch-train-duration:                  0.5057417508214712
+    eval-batch-orig-duration:                   0.4224007371813059
+    eval-batch-shift-duration:                  0.41603020019829273
+    eval-batch-tree-duration:                   0.4655158808454871
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.232666
+      cheese-corne_dist_wavg:                   12.720218
+      levels_cheese_is_in_corner_avg:           0.032470703
+      mouse-cheese_dist_finite_avg:             15.25
+      mouse-cheese_dist_finite_wavg:            7.8264003
+      mouse-cheese_dist_solvable_avg:           15.25
+      mouse-cheese_dist_solvable_wavg:          7.826401
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12015348
+    visit_patterns:
+      avg_first_visit_time:                     608.05444
+      avg_last_visit_time:                      889.2427
+      num_replay_batches:                       899
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001992828
+    avg_actor_approxkl3:                        0.0012656273
+    avg_actor_clipfrac:                         0.045196533
+    avg_actor_loss:                             0.0005484226
+    avg_advantage:                              -0.00080735073
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00024895175
+    avg_entropy:                                0.12752412
+    avg_grad_norm_pre_clip:                     2.5271099
+    avg_loss:                                   0.00054537435
+    max:
+      max_actor_approxkl1:                      0.0039977655
+      max_actor_approxkl3:                      0.0026534216
+      max_actor_clipfrac:                       0.068847656
+      max_actor_loss:                           0.002173772
+      max_advantage:                            0.08134693
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0003028197
+      max_entropy:                              0.1393664
+      max_grad_norm_pre_clip:                   4.1696815
+      max_loss:                                 0.002169602
+    std:
+      std_actor_approxkl1:                      0.0010827965
+      std_actor_approxkl3:                      0.0007253384
+      std_actor_clipfrac:                       0.019490615
+      std_actor_loss:                           0.0008127329
+      std_advantage:                            0.02208716
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.381266e-05
+      std_entropy:                              0.007345343
+      std_grad_norm_pre_clip:                   1.0641963
+      std_loss:                                 0.0008152528
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4419274
+    avg_avg_return:                             0.98092407
+    avg_benchmark_regret:                       0.011875099
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19861603
+    proxy_corner:
+      avg_avg_return:                           0.007795273
+      avg_reward_per_step:                      0.0015945435
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.501827
+    avg_avg_return:                             0.97540814
+    avg_benchmark_regret:                       0.014831696
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14165497
+    proxy_corner:
+      avg_avg_return:                           0.97540814
+      avg_reward_per_step:                      0.14165497
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.373165
+    avg_avg_return:                             0.9819126
+    avg_benchmark_regret:                       0.010924843
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19298553
+    proxy_corner:
+      avg_avg_return:                           0.004169388
+      avg_reward_per_step:                      0.001045227
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.33333
+    avg_avg_return:                             0.0
+    avg_benchmark_regret:                       0.98463523
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0
+    proxy_corner:
+      avg_avg_return:                           0.0
+      avg_reward_per_step:                      0.0
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        58720256
+    env-step-replay-before:                     29851648
+    env-step-used-before:                       58720256
+    ppo-update-before:                          35840
+    env-step-generate-before:                   28868608
+    env-step-all-after:                         58753024
+    env-step-replay-after:                      29851648
+    env-step-used-after:                        58753024
+    ppo-update-after:                           35860
+    env-step-generate-after:                    28901376
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     8.116423
+    avg_avg_return:                             0.94206464
+    avg_reward_per_step:                        0.17358398
+    proxy_corner:
+      avg_avg_return:                           0.012874369
+      avg_reward_per_step:                      0.0012512207
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     8.116423
+    avg_avg_return:                             0.94206464
+    avg_reward_per_step:                        0.17358398
+    proxy_corner:
+      avg_avg_return:                           0.012874369
+      avg_reward_per_step:                      0.0012512207
+  perf:
+    env_steps_per_second:                       325369.70568411186
+    ppo_updates_per_second:                     41.884745433033
+    eval-batch-train-duration:                  0.5375515511259437
+    eval-batch-orig-duration:                   0.4808464413508773
+    eval-batch-shift-duration:                  0.47568155638873577
+    eval-batch-tree-duration:                   0.4697398189455271
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.166016
+      cheese-corne_dist_wavg:                   3.7126424
+      levels_cheese_is_in_corner_avg:           0.032470703
+      mouse-cheese_dist_finite_avg:             15.26416
+      mouse-cheese_dist_finite_wavg:            16.83115
+      mouse-cheese_dist_solvable_avg:           15.26416
+      mouse-cheese_dist_solvable_wavg:          16.83115
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12224931
+    visit_patterns:
+      avg_first_visit_time:                     621.78516
+      avg_last_visit_time:                      901.16846
+      num_replay_batches:                       911
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0016799746
+    avg_actor_approxkl3:                        0.0014715664
+    avg_actor_clipfrac:                         0.050610352
+    avg_actor_loss:                             0.0009941384
+    avg_advantage:                              -0.0039242744
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00022628508
+    avg_entropy:                                0.1391931
+    avg_grad_norm_pre_clip:                     3.451104
+    avg_loss:                                   0.00096808776
+    max:
+      max_actor_approxkl1:                      0.0041952822
+      max_actor_approxkl3:                      0.0036808543
+      max_actor_clipfrac:                       0.099121094
+      max_actor_loss:                           0.0035933051
+      max_advantage:                            0.08881307
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0002689364
+      max_entropy:                              0.17594013
+      max_grad_norm_pre_clip:                   6.215795
+      max_loss:                                 0.0035439415
+    std:
+      std_actor_approxkl1:                      0.0010702423
+      std_actor_approxkl3:                      0.0008883347
+      std_actor_clipfrac:                       0.022770496
+      std_actor_loss:                           0.0011141144
+      std_advantage:                            0.021371644
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.6908478e-05
+      std_entropy:                              0.01555739
+      std_grad_norm_pre_clip:                   1.4662346
+      std_loss:                                 0.001109792
+  eval-batch-train:
+    avg_avg_episode_length:                     8.135836
+    avg_avg_return:                             0.9798329
+    avg_benchmark_regret:                       0.012966299
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19307709
+    proxy_corner:
+      avg_avg_return:                           0.00761861
+      avg_reward_per_step:                      0.0015563965
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.67887
+    avg_avg_return:                             0.97435343
+    avg_benchmark_regret:                       0.01588635
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14004517
+    proxy_corner:
+      avg_avg_return:                           0.97435343
+      avg_reward_per_step:                      0.14004517
+  eval-batch-shift:
+    avg_avg_episode_length:                     8.062473
+    avg_avg_return:                             0.9792233
+    avg_benchmark_regret:                       0.013614111
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.18626404
+    proxy_corner:
+      avg_avg_return:                           0.0042061177
+      avg_reward_per_step:                      0.001045227
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.33333
+    avg_avg_return:                             0.0007040465
+    avg_benchmark_regret:                       0.9839312
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        7.6293945e-06
+    proxy_corner:
+      avg_avg_return:                           0.0007040465
+      avg_reward_per_step:                      7.6293945e-06
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        59768832
+    env-step-replay-before:                     30441472
+    env-step-used-before:                       59768832
+    ppo-update-before:                          36480
+    env-step-generate-before:                   29327360
+    env-step-all-after:                         59801600
+    env-step-replay-after:                      30474240
+    env-step-used-after:                        59801600
+    ppo-update-after:                           36500
+    env-step-generate-after:                    29327360
+  train-replay:
+    shift_proportion:                           0.953125
+    avg_avg_episode_length:                     17.135033
+    avg_avg_return:                             0.86529887
+    avg_reward_per_step:                        0.055999756
+    proxy_corner:
+      avg_avg_return:                           0.044118684
+      avg_reward_per_step:                      0.002960205
+  train-all:
+    shift_proportion:                           0.953125
+    avg_avg_episode_length:                     17.135033
+    avg_avg_return:                             0.86529887
+    avg_reward_per_step:                        0.055999756
+    proxy_corner:
+      avg_avg_return:                           0.044118684
+      avg_reward_per_step:                      0.002960205
+  perf:
+    env_steps_per_second:                       329172.45213128254
+    ppo_updates_per_second:                     43.345194741159055
+    eval-batch-train-duration:                  0.5341155976057053
+    eval-batch-orig-duration:                   0.45289277471601963
+    eval-batch-shift-duration:                  0.47996948938816786
+    eval-batch-tree-duration:                   0.49013275001198053
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.143066
+      cheese-corne_dist_wavg:                   13.604383
+      levels_cheese_is_in_corner_avg:           0.033203125
+      mouse-cheese_dist_finite_avg:             15.235596
+      mouse-cheese_dist_finite_wavg:            6.0359354
+      mouse-cheese_dist_solvable_avg:           15.235596
+      mouse-cheese_dist_solvable_wavg:          6.0359364
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12246414
+    visit_patterns:
+      avg_first_visit_time:                     633.3169
+      avg_last_visit_time:                      920.176
+      num_replay_batches:                       930
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0045622005
+    avg_actor_approxkl3:                        0.0040518083
+    avg_actor_clipfrac:                         0.10987549
+    avg_actor_loss:                             0.0054384866
+    avg_advantage:                              -0.0073617515
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00018729651
+    avg_entropy:                                0.19140337
+    avg_grad_norm_pre_clip:                     6.7265983
+    avg_loss:                                   0.0053407312
+    max:
+      max_actor_approxkl1:                      0.009429183
+      max_actor_approxkl3:                      0.0077355467
+      max_actor_clipfrac:                       0.15930176
+      max_actor_loss:                           0.011723775
+      max_advantage:                            0.1030103
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00021356586
+      max_entropy:                              0.21550855
+      max_grad_norm_pre_clip:                   9.630975
+      max_loss:                                 0.01161821
+    std:
+      std_actor_approxkl1:                      0.0023706208
+      std_actor_approxkl3:                      0.0018494293
+      std_actor_clipfrac:                       0.039033476
+      std_actor_loss:                           0.0031658325
+      std_advantage:                            0.018939715
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.1037199e-05
+      std_entropy:                              0.01083605
+      std_grad_norm_pre_clip:                   2.344731
+      std_loss:                                 0.0031657217
+  eval-batch-train:
+    avg_avg_episode_length:                     7.5721846
+    avg_avg_return:                             0.9811153
+    avg_benchmark_regret:                       0.011683892
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19644165
+    proxy_corner:
+      avg_avg_return:                           0.0078112143
+      avg_reward_per_step:                      0.0015945435
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.546993
+    avg_avg_return:                             0.9744191
+    avg_benchmark_regret:                       0.015820721
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14134979
+    proxy_corner:
+      avg_avg_return:                           0.9744191
+      avg_reward_per_step:                      0.14134979
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.547476
+    avg_avg_return:                             0.9817751
+    avg_benchmark_regret:                       0.011062333
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19016266
+    proxy_corner:
+      avg_avg_return:                           0.0042751417
+      avg_reward_per_step:                      0.0010757446
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.33333
+    avg_avg_return:                             0.0015236367
+    avg_benchmark_regret:                       0.98311156
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        1.5258789e-05
+    proxy_corner:
+      avg_avg_return:                           0.0015236367
+      avg_reward_per_step:                      1.5258789e-05
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        60817408
+    env-step-replay-before:                     30998528
+    env-step-used-before:                       60817408
+    ppo-update-before:                          37120
+    env-step-generate-before:                   29818880
+    env-step-all-after:                         60850176
+    env-step-replay-after:                      30998528
+    env-step-used-after:                        60850176
+    ppo-update-after:                           37140
+    env-step-generate-after:                    29851648
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.8061504
+    avg_avg_return:                             0.9462342
+    avg_reward_per_step:                        0.19616699
+    proxy_corner:
+      avg_avg_return:                           0.0036231459
+      avg_reward_per_step:                      0.00024414062
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.8061504
+    avg_avg_return:                             0.9462342
+    avg_reward_per_step:                        0.19616699
+    proxy_corner:
+      avg_avg_return:                           0.0036231459
+      avg_reward_per_step:                      0.00024414062
+  perf:
+    env_steps_per_second:                       294914.34862099873
+    ppo_updates_per_second:                     42.099902840134
+    eval-batch-train-duration:                  0.5226126564666629
+    eval-batch-orig-duration:                   0.47963804565370083
+    eval-batch-shift-duration:                  0.44794304110109806
+    eval-batch-tree-duration:                   0.4974216315895319
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.17627
+      cheese-corne_dist_wavg:                   3.7190447
+      levels_cheese_is_in_corner_avg:           0.033447266
+      mouse-cheese_dist_finite_avg:             15.268799
+      mouse-cheese_dist_finite_wavg:            16.827578
+      mouse-cheese_dist_solvable_avg:           15.268799
+      mouse-cheese_dist_solvable_wavg:          16.82758
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.120030776
+    visit_patterns:
+      avg_first_visit_time:                     645.2539
+      avg_last_visit_time:                      936.2825
+      num_replay_batches:                       946
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0015382307
+    avg_actor_approxkl3:                        0.0013490914
+    avg_actor_clipfrac:                         0.04534302
+    avg_actor_loss:                             0.0010263924
+    avg_advantage:                              0.0044826986
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00029299807
+    avg_entropy:                                0.11743861
+    avg_grad_norm_pre_clip:                     3.371838
+    avg_loss:                                   0.001055453
+    max:
+      max_actor_approxkl1:                      0.0065905973
+      max_actor_approxkl3:                      0.0041519855
+      max_actor_clipfrac:                       0.07836914
+      max_actor_loss:                           0.0035915058
+      max_advantage:                            0.12332523
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00037596974
+      max_entropy:                              0.14037089
+      max_grad_norm_pre_clip:                   5.8747315
+      max_loss:                                 0.0036224155
+    std:
+      std_actor_approxkl1:                      0.0015063602
+      std_actor_approxkl3:                      0.00091210543
+      std_actor_clipfrac:                       0.021150403
+      std_actor_loss:                           0.001160485
+      std_advantage:                            0.022513364
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.9226936e-05
+      std_entropy:                              0.013197468
+      std_grad_norm_pre_clip:                   1.5161695
+      std_loss:                                 0.0011529507
+  eval-batch-train:
+    avg_avg_episode_length:                     7.526821
+    avg_avg_return:                             0.9804068
+    avg_benchmark_regret:                       0.012392329
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19986725
+    proxy_corner:
+      avg_avg_return:                           0.0076264767
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.553637
+    avg_avg_return:                             0.9742155
+    avg_benchmark_regret:                       0.016024293
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14100647
+    proxy_corner:
+      avg_avg_return:                           0.9742155
+      avg_reward_per_step:                      0.14100647
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.553008
+    avg_avg_return:                             0.98148584
+    avg_benchmark_regret:                       0.011351598
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19077301
+    proxy_corner:
+      avg_avg_return:                           0.00421571
+      avg_reward_per_step:                      0.0010604858
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.166664
+    avg_avg_return:                             0.005036221
+    avg_benchmark_regret:                       0.979599
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        6.866455e-05
+    proxy_corner:
+      avg_avg_return:                           0.005036221
+      avg_reward_per_step:                      6.866455e-05
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        61865984
+    env-step-replay-before:                     31424512
+    env-step-used-before:                       61865984
+    ppo-update-before:                          37760
+    env-step-generate-before:                   30441472
+    env-step-all-after:                         61898752
+    env-step-replay-after:                      31424512
+    env-step-used-after:                        61898752
+    ppo-update-after:                           37780
+    env-step-generate-after:                    30474240
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.9807444
+    avg_avg_return:                             0.9429884
+    avg_reward_per_step:                        0.18392944
+    proxy_corner:
+      avg_avg_return:                           0.0064590247
+      avg_reward_per_step:                      0.0043640137
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.9807444
+    avg_avg_return:                             0.9429884
+    avg_reward_per_step:                        0.18392944
+    proxy_corner:
+      avg_avg_return:                           0.0064590247
+      avg_reward_per_step:                      0.0043640137
+  perf:
+    env_steps_per_second:                       327838.9619648863
+    ppo_updates_per_second:                     42.47845151143477
+    eval-batch-train-duration:                  0.548471437767148
+    eval-batch-orig-duration:                   0.47770200576633215
+    eval-batch-shift-duration:                  0.45814387034624815
+    eval-batch-tree-duration:                   0.4908181596547365
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.160889
+      cheese-corne_dist_wavg:                   3.705575
+      levels_cheese_is_in_corner_avg:           0.032226562
+      mouse-cheese_dist_finite_avg:             15.249268
+      mouse-cheese_dist_finite_wavg:            15.925876
+      mouse-cheese_dist_solvable_avg:           15.249268
+      mouse-cheese_dist_solvable_wavg:          15.925874
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12151745
+    visit_patterns:
+      avg_first_visit_time:                     658.1843
+      avg_last_visit_time:                      949.25024
+      num_replay_batches:                       959
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0021905464
+    avg_actor_approxkl3:                        0.002038366
+    avg_actor_clipfrac:                         0.0609375
+    avg_actor_loss:                             0.0025268833
+    avg_advantage:                              7.2854804e-05
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00023055189
+    avg_entropy:                                0.13375635
+    avg_grad_norm_pre_clip:                     4.475915
+    avg_loss:                                   0.0025084028
+    max:
+      max_actor_approxkl1:                      0.00631862
+      max_actor_approxkl3:                      0.0054897848
+      max_actor_clipfrac:                       0.11682129
+      max_actor_loss:                           0.009488538
+      max_advantage:                            0.07416642
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.000273211
+      max_entropy:                              0.15266755
+      max_grad_norm_pre_clip:                   8.0157995
+      max_loss:                                 0.00947892
+    std:
+      std_actor_approxkl1:                      0.0016196807
+      std_actor_approxkl3:                      0.0014815729
+      std_actor_clipfrac:                       0.030045807
+      std_actor_loss:                           0.0022955781
+      std_advantage:                            0.020704163
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.9398909e-05
+      std_entropy:                              0.013685342
+      std_grad_norm_pre_clip:                   1.9859824
+      std_loss:                                 0.0022952266
+  eval-batch-train:
+    avg_avg_episode_length:                     7.518072
+    avg_avg_return:                             0.9809769
+    avg_benchmark_regret:                       0.011822293
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19872284
+    proxy_corner:
+      avg_avg_return:                           0.007971071
+      avg_reward_per_step:                      0.0016021729
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.527355
+    avg_avg_return:                             0.97534776
+    avg_benchmark_regret:                       0.0148920715
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14150238
+    proxy_corner:
+      avg_avg_return:                           0.97534776
+      avg_reward_per_step:                      0.14150238
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4799805
+    avg_avg_return:                             0.98143595
+    avg_benchmark_regret:                       0.011401494
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19067383
+    proxy_corner:
+      avg_avg_return:                           0.004470573
+      avg_reward_per_step:                      0.001045227
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.14285
+    avg_avg_return:                             0.0049544424
+    avg_benchmark_regret:                       0.9796808
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.00010681152
+    proxy_corner:
+      avg_avg_return:                           0.0049544424
+      avg_reward_per_step:                      0.00010681152
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        62914560
+    env-step-replay-before:                     31981568
+    env-step-used-before:                       62914560
+    ppo-update-before:                          38400
+    env-step-generate-before:                   30932992
+    env-step-all-after:                         62947328
+    env-step-replay-after:                      31981568
+    env-step-used-after:                        62947328
+    ppo-update-after:                           38420
+    env-step-generate-after:                    30965760
+  train-generate:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     7.7735763
+    avg_avg_return:                             0.94513917
+    avg_reward_per_step:                        0.19448853
+    proxy_corner:
+      avg_avg_return:                           0.014967607
+      avg_reward_per_step:                      0.0014038086
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     7.7735763
+    avg_avg_return:                             0.94513917
+    avg_reward_per_step:                        0.19448853
+    proxy_corner:
+      avg_avg_return:                           0.014967607
+      avg_reward_per_step:                      0.0014038086
+  perf:
+    env_steps_per_second:                       324782.28295965766
+    ppo_updates_per_second:                     45.5136908290889
+    eval-batch-train-duration:                  0.517734250985086
+    eval-batch-orig-duration:                   0.4830701192840934
+    eval-batch-shift-duration:                  0.47605071403086185
+    eval-batch-tree-duration:                   0.4786248942837119
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.247559
+      cheese-corne_dist_wavg:                   9.109592
+      levels_cheese_is_in_corner_avg:           0.032226562
+      mouse-cheese_dist_finite_avg:             15.260254
+      mouse-cheese_dist_finite_wavg:            15.020863
+      mouse-cheese_dist_solvable_avg:           15.260254
+      mouse-cheese_dist_solvable_wavg:          15.0208645
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12069549
+    visit_patterns:
+      avg_first_visit_time:                     669.3882
+      avg_last_visit_time:                      966.345
+      num_replay_batches:                       976
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013644095
+    avg_actor_approxkl3:                        0.0014256741
+    avg_actor_clipfrac:                         0.045391846
+    avg_actor_loss:                             0.0012015695
+    avg_advantage:                              0.0024044071
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00022890104
+    avg_entropy:                                0.12518458
+    avg_grad_norm_pre_clip:                     2.6638913
+    avg_loss:                                   0.0011908354
+    max:
+      max_actor_approxkl1:                      0.005028086
+      max_actor_approxkl3:                      0.004693965
+      max_actor_clipfrac:                       0.09887695
+      max_actor_loss:                           0.004099572
+      max_advantage:                            0.09758663
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00025978364
+      max_entropy:                              0.14641464
+      max_grad_norm_pre_clip:                   5.7112694
+      max_loss:                                 0.0041128783
+    std:
+      std_actor_approxkl1:                      0.001256643
+      std_actor_approxkl3:                      0.0011329754
+      std_actor_clipfrac:                       0.021014335
+      std_actor_loss:                           0.0012637
+      std_advantage:                            0.021173539
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.5622067e-05
+      std_entropy:                              0.011932041
+      std_grad_norm_pre_clip:                   1.153522
+      std_loss:                                 0.0012644103
+  eval-batch-train:
+    avg_avg_episode_length:                     7.5195546
+    avg_avg_return:                             0.9812309
+    avg_benchmark_regret:                       0.011568254
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19888306
+    proxy_corner:
+      avg_avg_return:                           0.0077310973
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.547562
+    avg_avg_return:                             0.9749945
+    avg_benchmark_regret:                       0.015245352
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14135742
+    proxy_corner:
+      avg_avg_return:                           0.9749945
+      avg_reward_per_step:                      0.14135742
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.5173426
+    avg_avg_return:                             0.9812542
+    avg_benchmark_regret:                       0.011583221
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.18992615
+    proxy_corner:
+      avg_avg_return:                           0.0040056584
+      avg_reward_per_step:                      0.0009918213
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.02777
+    avg_avg_return:                             0.005208621
+    avg_benchmark_regret:                       0.97942656
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0001449585
+    proxy_corner:
+      avg_avg_return:                           0.005208621
+      avg_reward_per_step:                      0.0001449585
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        63963136
+    env-step-replay-before:                     32505856
+    env-step-used-before:                       63963136
+    ppo-update-before:                          39040
+    env-step-generate-before:                   31457280
+    env-step-all-after:                         63995904
+    env-step-replay-after:                      32505856
+    env-step-used-after:                        63995904
+    ppo-update-after:                           39060
+    env-step-generate-after:                    31490048
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     8.153197
+    avg_avg_return:                             0.94283783
+    avg_reward_per_step:                        0.18725586
+    proxy_corner:
+      avg_avg_return:                           0.007209515
+      avg_reward_per_step:                      0.0008239746
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     8.153197
+    avg_avg_return:                             0.94283783
+    avg_reward_per_step:                        0.18725586
+    proxy_corner:
+      avg_avg_return:                           0.007209515
+      avg_reward_per_step:                      0.0008239746
+  perf:
+    env_steps_per_second:                       328926.60809962737
+    ppo_updates_per_second:                     40.60739220402967
+    eval-batch-train-duration:                  0.540122278034687
+    eval-batch-orig-duration:                   0.47817212250083685
+    eval-batch-shift-duration:                  0.4473264068365097
+    eval-batch-tree-duration:                   0.4851005179807544
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.216553
+      cheese-corne_dist_wavg:                   13.63052
+      levels_cheese_is_in_corner_avg:           0.033447266
+      mouse-cheese_dist_finite_avg:             15.258301
+      mouse-cheese_dist_finite_wavg:            11.427348
+      mouse-cheese_dist_solvable_avg:           15.258301
+      mouse-cheese_dist_solvable_wavg:          11.427348
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12534764
+    visit_patterns:
+      avg_first_visit_time:                     683.198
+      avg_last_visit_time:                      982.19995
+      num_replay_batches:                       992
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0033992908
+    avg_actor_approxkl3:                        0.0025226444
+    avg_actor_clipfrac:                         0.06289673
+    avg_actor_loss:                             0.0026045637
+    avg_advantage:                              0.0070015327
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0002801592
+    avg_entropy:                                0.1253843
+    avg_grad_norm_pre_clip:                     3.566058
+    avg_loss:                                   0.0026192588
+    max:
+      max_actor_approxkl1:                      0.0076542976
+      max_actor_approxkl3:                      0.0062460103
+      max_actor_clipfrac:                       0.10595703
+      max_actor_loss:                           0.010130665
+      max_advantage:                            0.09997761
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00043613082
+      max_entropy:                              0.14932618
+      max_grad_norm_pre_clip:                   7.6133227
+      max_loss:                                 0.010167089
+    std:
+      std_actor_approxkl1:                      0.0020333903
+      std_actor_approxkl3:                      0.001559879
+      std_actor_clipfrac:                       0.021970956
+      std_actor_loss:                           0.0025124522
+      std_advantage:                            0.020778611
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          6.92281e-05
+      std_entropy:                              0.013218477
+      std_grad_norm_pre_clip:                   1.3406084
+      std_loss:                                 0.0025155686
+  eval-batch-train:
+    avg_avg_episode_length:                     7.469773
+    avg_avg_return:                             0.9815322
+    avg_benchmark_regret:                       0.011266934
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20082855
+    proxy_corner:
+      avg_avg_return:                           0.007751689
+      avg_reward_per_step:                      0.0015869141
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.559696
+    avg_avg_return:                             0.9743483
+    avg_benchmark_regret:                       0.015891485
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14128113
+    proxy_corner:
+      avg_avg_return:                           0.9743483
+      avg_reward_per_step:                      0.14128113
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4266996
+    avg_avg_return:                             0.98145187
+    avg_benchmark_regret:                       0.011385569
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19231415
+    proxy_corner:
+      avg_avg_return:                           0.0046075145
+      avg_reward_per_step:                      0.0011062622
+  eval-batch-tree:
+    avg_avg_episode_length:                     85.00952
+    avg_avg_return:                             0.006374211
+    avg_benchmark_regret:                       0.978261
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0001449585
+    proxy_corner:
+      avg_avg_return:                           0.006374211
+      avg_reward_per_step:                      0.0001449585
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        65011712
+    env-step-replay-before:                     33062912
+    env-step-used-before:                       65011712
+    ppo-update-before:                          39680
+    env-step-generate-before:                   31948800
+    env-step-all-after:                         65044480
+    env-step-replay-after:                      33062912
+    env-step-used-after:                        65044480
+    ppo-update-after:                           39700
+    env-step-generate-after:                    31981568
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.650532
+    avg_avg_return:                             0.94827247
+    avg_reward_per_step:                        0.18148804
+    proxy_corner:
+      avg_avg_return:                           0.016467443
+      avg_reward_per_step:                      0.0020446777
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.650532
+    avg_avg_return:                             0.94827247
+    avg_reward_per_step:                        0.18148804
+    proxy_corner:
+      avg_avg_return:                           0.016467443
+      avg_reward_per_step:                      0.0020446777
+  perf:
+    env_steps_per_second:                       275251.50710483483
+    ppo_updates_per_second:                     41.835279265921024
+    eval-batch-train-duration:                  0.5346528626978397
+    eval-batch-orig-duration:                   0.48418385814875364
+    eval-batch-shift-duration:                  0.4770699432119727
+    eval-batch-tree-duration:                   0.45523427054286003
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.246338
+      cheese-corne_dist_wavg:                   13.620979
+      levels_cheese_is_in_corner_avg:           0.033447266
+      mouse-cheese_dist_finite_avg:             15.282715
+      mouse-cheese_dist_finite_wavg:            7.8356314
+      mouse-cheese_dist_solvable_avg:           15.282715
+      mouse-cheese_dist_solvable_wavg:          7.8356314
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12064522
+    visit_patterns:
+      avg_first_visit_time:                     692.7883
+      avg_last_visit_time:                      999.2876
+      num_replay_batches:                       1009
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0018157841
+    avg_actor_approxkl3:                        0.0011873079
+    avg_actor_clipfrac:                         0.03931885
+    avg_actor_loss:                             0.00020261123
+    avg_advantage:                              0.0014874806
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00021111185
+    avg_entropy:                                0.11702619
+    avg_grad_norm_pre_clip:                     1.8114865
+    avg_loss:                                   0.00019114096
+    max:
+      max_actor_approxkl1:                      0.0049563823
+      max_actor_approxkl3:                      0.0033202735
+      max_actor_clipfrac:                       0.083618164
+      max_actor_loss:                           0.0017694365
+      max_advantage:                            0.08126968
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.000255077
+      max_entropy:                              0.14027561
+      max_grad_norm_pre_clip:                   4.189929
+      max_loss:                                 0.0017670457
+    std:
+      std_actor_approxkl1:                      0.0013894808
+      std_actor_approxkl3:                      0.00077848515
+      std_actor_clipfrac:                       0.017975362
+      std_actor_loss:                           0.00058605784
+      std_advantage:                            0.020442085
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.678304e-05
+      std_entropy:                              0.013512415
+      std_grad_norm_pre_clip:                   0.9557111
+      std_loss:                                 0.000586122
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4818845
+    avg_avg_return:                             0.98164845
+    avg_benchmark_regret:                       0.0111506935
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.1993866
+    proxy_corner:
+      avg_avg_return:                           0.008500935
+      avg_reward_per_step:                      0.0016555786
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.526482
+    avg_avg_return:                             0.97527826
+    avg_benchmark_regret:                       0.014961566
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14163208
+    proxy_corner:
+      avg_avg_return:                           0.97527826
+      avg_reward_per_step:                      0.14163208
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.427204
+    avg_avg_return:                             0.9819347
+    avg_benchmark_regret:                       0.010902726
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19193268
+    proxy_corner:
+      avg_avg_return:                           0.0051796106
+      avg_reward_per_step:                      0.0012359619
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.878784
+    avg_avg_return:                             0.0071647307
+    avg_benchmark_regret:                       0.9774705
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.00039672852
+    proxy_corner:
+      avg_avg_return:                           0.0071647307
+      avg_reward_per_step:                      0.00039672852
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        66060288
+    env-step-replay-before:                     33554432
+    env-step-used-before:                       66060288
+    ppo-update-before:                          40320
+    env-step-generate-before:                   32505856
+    env-step-all-after:                         66093056
+    env-step-replay-after:                      33554432
+    env-step-used-after:                        66093056
+    ppo-update-after:                           40340
+    env-step-generate-after:                    32538624
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.9247174
+    avg_avg_return:                             0.9460462
+    avg_reward_per_step:                        0.19674683
+    proxy_corner:
+      avg_avg_return:                           0.01231494
+      avg_reward_per_step:                      0.0046691895
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.9247174
+    avg_avg_return:                             0.9460462
+    avg_reward_per_step:                        0.19674683
+    proxy_corner:
+      avg_avg_return:                           0.01231494
+      avg_reward_per_step:                      0.0046691895
+  perf:
+    env_steps_per_second:                       302908.41509601957
+    ppo_updates_per_second:                     42.104620180544224
+    eval-batch-train-duration:                  0.5326464455574751
+    eval-batch-orig-duration:                   0.4703638730570674
+    eval-batch-shift-duration:                  0.44140310771763325
+    eval-batch-tree-duration:                   0.48027016688138247
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.300781
+      cheese-corne_dist_wavg:                   9.129539
+      levels_cheese_is_in_corner_avg:           0.033935547
+      mouse-cheese_dist_finite_avg:             15.307617
+      mouse-cheese_dist_finite_wavg:            15.027486
+      mouse-cheese_dist_solvable_avg:           15.307617
+      mouse-cheese_dist_solvable_wavg:          15.027488
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.121571496
+    visit_patterns:
+      avg_first_visit_time:                     703.354
+      avg_last_visit_time:                      1014.2737
+      num_replay_batches:                       1024
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.003262744
+    avg_actor_approxkl3:                        0.002734672
+    avg_actor_clipfrac:                         0.0659607
+    avg_actor_loss:                             0.002829789
+    avg_advantage:                              0.0005477191
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00022135706
+    avg_entropy:                                0.1169778
+    avg_grad_norm_pre_clip:                     4.2510967
+    avg_loss:                                   0.0028234895
+    max:
+      max_actor_approxkl1:                      0.0067948774
+      max_actor_approxkl3:                      0.005901524
+      max_actor_clipfrac:                       0.099121094
+      max_actor_loss:                           0.007014905
+      max_advantage:                            0.095791936
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00025420362
+      max_entropy:                              0.13379583
+      max_grad_norm_pre_clip:                   7.0832415
+      max_loss:                                 0.006995172
+    std:
+      std_actor_approxkl1:                      0.0018722431
+      std_actor_approxkl3:                      0.0015570689
+      std_actor_clipfrac:                       0.023551164
+      std_actor_loss:                           0.0019045528
+      std_advantage:                            0.020979356
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.696849e-05
+      std_entropy:                              0.013290492
+      std_grad_norm_pre_clip:                   1.656687
+      std_loss:                                 0.0019032109
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4646473
+    avg_avg_return:                             0.9807873
+    avg_benchmark_regret:                       0.0120118605
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20043182
+    proxy_corner:
+      avg_avg_return:                           0.009209758
+      avg_reward_per_step:                      0.0016784668
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.480225
+    avg_avg_return:                             0.97569597
+    avg_benchmark_regret:                       0.014543898
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14189911
+    proxy_corner:
+      avg_avg_return:                           0.97569597
+      avg_reward_per_step:                      0.14189911
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.331552
+    avg_avg_return:                             0.9822111
+    avg_benchmark_regret:                       0.010626288
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19309235
+    proxy_corner:
+      avg_avg_return:                           0.0069984132
+      avg_reward_per_step:                      0.0014419556
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.74504
+    avg_avg_return:                             0.008672318
+    avg_benchmark_regret:                       0.9759629
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0005874634
+    proxy_corner:
+      avg_avg_return:                           0.008672318
+      avg_reward_per_step:                      0.0005874634
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        67108864
+    env-step-replay-before:                     34078720
+    env-step-used-before:                       67108864
+    ppo-update-before:                          40960
+    env-step-generate-before:                   33030144
+    env-step-all-after:                         67141632
+    env-step-replay-after:                      34111488
+    env-step-used-after:                        67141632
+    ppo-update-after:                           40980
+    env-step-generate-after:                    33030144
+  train-replay:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     17.613087
+    avg_avg_return:                             0.86238444
+    avg_reward_per_step:                        0.054748535
+    proxy_corner:
+      avg_avg_return:                           0.006710192
+      avg_reward_per_step:                      0.0004272461
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     17.613087
+    avg_avg_return:                             0.86238444
+    avg_reward_per_step:                        0.054748535
+    proxy_corner:
+      avg_avg_return:                           0.006710192
+      avg_reward_per_step:                      0.0004272461
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  perf:
+    env_steps_per_second:                       353680.2302414649
+    ppo_updates_per_second:                     42.40757673694137
+    eval-batch-train-duration:                  0.6166571667417884
+    eval-batch-orig-duration:                   0.5770777883008122
+    eval-batch-shift-duration:                  0.5871786074712873
+    eval-batch-tree-duration:                   0.583170561119914
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.310059
+      cheese-corne_dist_wavg:                   3.7349381
+      levels_cheese_is_in_corner_avg:           0.03491211
+      mouse-cheese_dist_finite_avg:             15.374756
+      mouse-cheese_dist_finite_wavg:            15.933402
+      mouse-cheese_dist_solvable_avg:           15.374756
+      mouse-cheese_dist_solvable_wavg:          15.933402
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12065369
+    visit_patterns:
+      avg_first_visit_time:                     710.8906
+      avg_last_visit_time:                      1031.2285
+      num_replay_batches:                       1041
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.003719844
+    avg_actor_approxkl3:                        0.0030845755
+    avg_actor_clipfrac:                         0.09125977
+    avg_actor_loss:                             0.0029372976
+    avg_advantage:                              -0.002036298
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00015480821
+    avg_entropy:                                0.19984987
+    avg_grad_norm_pre_clip:                     4.1177793
+    avg_loss:                                   0.0028148515
+    max:
+      max_actor_approxkl1:                      0.0073182685
+      max_actor_approxkl3:                      0.0076578795
+      max_actor_clipfrac:                       0.14611816
+      max_actor_loss:                           0.010378816
+      max_advantage:                            0.07561612
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0001687452
+      max_entropy:                              0.21716623
+      max_grad_norm_pre_clip:                   6.2995515
+      max_loss:                                 0.010257192
+    std:
+      std_actor_approxkl1:                      0.001844599
+      std_actor_approxkl3:                      0.0017228407
+      std_actor_clipfrac:                       0.031721786
+      std_actor_loss:                           0.0025152187
+      std_advantage:                            0.017468087
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          6.5962763e-06
+      std_entropy:                              0.010139537
+      std_grad_norm_pre_clip:                   1.354304
+      std_loss:                                 0.0025181398
+  eval-batch-train:
+    avg_avg_episode_length:                     7.487769
+    avg_avg_return:                             0.98141855
+    avg_benchmark_regret:                       0.011380623
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20048523
+    proxy_corner:
+      avg_avg_return:                           0.0078046126
+      avg_reward_per_step:                      0.0015869141
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.605684
+    avg_avg_return:                             0.97414523
+    avg_benchmark_regret:                       0.016094618
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14073181
+    proxy_corner:
+      avg_avg_return:                           0.97414523
+      avg_reward_per_step:                      0.14073181
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.5105696
+    avg_avg_return:                             0.98160046
+    avg_benchmark_regret:                       0.011236981
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19197083
+    proxy_corner:
+      avg_avg_return:                           0.0039045205
+      avg_reward_per_step:                      0.0009765625
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.80962
+    avg_avg_return:                             0.0079389475
+    avg_benchmark_regret:                       0.97669625
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0006942749
+    proxy_corner:
+      avg_avg_return:                           0.0079389475
+      avg_reward_per_step:                      0.0006942749
+    rollouts_gif:                               float32(132, 225, 225, 3)
+===========================================================
+ 34%|███████████████████████████████████████████████████████████████████████▊                                                                                                                                              | 67.1M/200M [21:10<32:20, 68.5k env steps/s]
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        68157440
+    env-step-replay-before:                     34668544
+    env-step-used-before:                       68157440
+    ppo-update-before:                          41600
+    env-step-generate-before:                   33488896
+    env-step-all-after:                         68190208
+    env-step-replay-after:                      34668544
+    env-step-used-after:                        68190208
+    ppo-update-after:                           41620
+    env-step-generate-after:                    33521664
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     8.129513
+    avg_avg_return:                             0.9459585
+    avg_reward_per_step:                        0.1824646
+    proxy_corner:
+      avg_avg_return:                           0.010863757
+      avg_reward_per_step:                      0.0019226074
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     8.129513
+    avg_avg_return:                             0.9459585
+    avg_reward_per_step:                        0.1824646
+    proxy_corner:
+      avg_avg_return:                           0.010863757
+      avg_reward_per_step:                      0.0019226074
+  perf:
+    env_steps_per_second:                       264360.1722653167
+    ppo_updates_per_second:                     41.35133741333478
+    eval-batch-train-duration:                  0.525652589276433
+    eval-batch-orig-duration:                   0.4739384800195694
+    eval-batch-shift-duration:                  0.4492587074637413
+    eval-batch-tree-duration:                   0.4909424437209964
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.269775
+      cheese-corne_dist_wavg:                   4.6341457
+      levels_cheese_is_in_corner_avg:           0.03540039
+      mouse-cheese_dist_finite_avg:             15.378906
+      mouse-cheese_dist_finite_wavg:            13.239577
+      mouse-cheese_dist_solvable_avg:           15.378906
+      mouse-cheese_dist_solvable_wavg:          13.23958
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.121332355
+    visit_patterns:
+      avg_first_visit_time:                     723.03
+      avg_last_visit_time:                      1048.3555
+      num_replay_batches:                       1058
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0027492538
+    avg_actor_approxkl3:                        0.0022107665
+    avg_actor_clipfrac:                         0.06109009
+    avg_actor_loss:                             0.0016681232
+    avg_advantage:                              0.0073587596
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00026438263
+    avg_entropy:                                0.12644482
+    avg_grad_norm_pre_clip:                     3.0038192
+    avg_loss:                                   0.0016738698
+    max:
+      max_actor_approxkl1:                      0.005903324
+      max_actor_approxkl3:                      0.005425114
+      max_actor_clipfrac:                       0.08605957
+      max_actor_loss:                           0.003938147
+      max_advantage:                            0.08877945
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0003235959
+      max_entropy:                              0.15396538
+      max_grad_norm_pre_clip:                   5.0129585
+      max_loss:                                 0.003966401
+    std:
+      std_actor_approxkl1:                      0.0015273663
+      std_actor_approxkl3:                      0.0012074726
+      std_actor_clipfrac:                       0.02268615
+      std_actor_loss:                           0.0011820757
+      std_advantage:                            0.021893485
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.278433e-05
+      std_entropy:                              0.018197497
+      std_grad_norm_pre_clip:                   1.0595912
+      std_loss:                                 0.0011770984
+  eval-batch-train:
+    avg_avg_episode_length:                     7.5523305
+    avg_avg_return:                             0.9813752
+    avg_benchmark_regret:                       0.011423931
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.19940948
+    proxy_corner:
+      avg_avg_return:                           0.0076268935
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.555438
+    avg_avg_return:                             0.97440267
+    avg_benchmark_regret:                       0.015837144
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14105225
+    proxy_corner:
+      avg_avg_return:                           0.97440267
+      avg_reward_per_step:                      0.14105225
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.487362
+    avg_avg_return:                             0.98149
+    avg_benchmark_regret:                       0.011347444
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19181061
+    proxy_corner:
+      avg_avg_return:                           0.0038886245
+      avg_reward_per_step:                      0.0009765625
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.727936
+    avg_avg_return:                             0.0076380577
+    avg_benchmark_regret:                       0.97699714
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0010681152
+    proxy_corner:
+      avg_avg_return:                           0.0076380577
+      avg_reward_per_step:                      0.0010681152
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        69206016
+    env-step-replay-before:                     35258368
+    env-step-used-before:                       69206016
+    ppo-update-before:                          42240
+    env-step-generate-before:                   33947648
+    env-step-all-after:                         69238784
+    env-step-replay-after:                      35258368
+    env-step-used-after:                        69238784
+    ppo-update-after:                           42260
+    env-step-generate-after:                    33980416
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.989834
+    avg_avg_return:                             0.94368076
+    avg_reward_per_step:                        0.17782593
+    proxy_corner:
+      avg_avg_return:                           0.0036167693
+      avg_reward_per_step:                      0.0004272461
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.989834
+    avg_avg_return:                             0.94368076
+    avg_reward_per_step:                        0.17782593
+    proxy_corner:
+      avg_avg_return:                           0.0036167693
+      avg_reward_per_step:                      0.0004272461
+  perf:
+    env_steps_per_second:                       258782.19236089577
+    ppo_updates_per_second:                     41.643478612619155
+    eval-batch-train-duration:                  0.6520724911242723
+    eval-batch-orig-duration:                   0.6099497564136982
+    eval-batch-shift-duration:                  0.5935979522764683
+    eval-batch-tree-duration:                   0.6355709061026573
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.21582
+      cheese-corne_dist_wavg:                   13.613995
+      levels_cheese_is_in_corner_avg:           0.035888672
+      mouse-cheese_dist_finite_avg:             15.3984375
+      mouse-cheese_dist_finite_wavg:            10.549686
+      mouse-cheese_dist_solvable_avg:           15.3984375
+      mouse-cheese_dist_solvable_wavg:          10.549687
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.121136844
+    visit_patterns:
+      avg_first_visit_time:                     731.19653
+      avg_last_visit_time:                      1066.2429
+      num_replay_batches:                       1076
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.004307672
+    avg_actor_approxkl3:                        0.0043083276
+    avg_actor_clipfrac:                         0.07767945
+    avg_actor_loss:                             0.004950652
+    avg_advantage:                              -0.006184597
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0003781182
+    avg_entropy:                                0.1164104
+    avg_grad_norm_pre_clip:                     5.236521
+    avg_loss:                                   0.0050233
+    max:
+      max_actor_approxkl1:                      0.012059638
+      max_actor_approxkl3:                      0.01114302
+      max_actor_clipfrac:                       0.12207031
+      max_actor_loss:                           0.010726198
+      max_advantage:                            0.083517194
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0006907497
+      max_entropy:                              0.13549191
+      max_grad_norm_pre_clip:                   8.696361
+      max_loss:                                 0.010723478
+    std:
+      std_actor_approxkl1:                      0.0033106199
+      std_actor_approxkl3:                      0.0031062658
+      std_actor_clipfrac:                       0.030025829
+      std_actor_loss:                           0.0032839829
+      std_advantage:                            0.020462818
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00014530988
+      std_entropy:                              0.0076690316
+      std_grad_norm_pre_clip:                   2.2057476
+      std_loss:                                 0.0032907014
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4411
+    avg_avg_return:                             0.9817277
+    avg_benchmark_regret:                       0.011071412
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.2017746
+    proxy_corner:
+      avg_avg_return:                           0.007959499
+      avg_reward_per_step:                      0.0015945435
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.591632
+    avg_avg_return:                             0.9743396
+    avg_benchmark_regret:                       0.015900277
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14098358
+    proxy_corner:
+      avg_avg_return:                           0.9743396
+      avg_reward_per_step:                      0.14098358
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4174824
+    avg_avg_return:                             0.9813782
+    avg_benchmark_regret:                       0.0114592565
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19186401
+    proxy_corner:
+      avg_avg_return:                           0.0048580132
+      avg_reward_per_step:                      0.0011672974
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.6996
+    avg_avg_return:                             0.0077515375
+    avg_benchmark_regret:                       0.97688365
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0018386841
+    proxy_corner:
+      avg_avg_return:                           0.0077515375
+      avg_reward_per_step:                      0.0018386841
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        70254592
+    env-step-replay-before:                     35651584
+    env-step-used-before:                       70254592
+    ppo-update-before:                          42880
+    env-step-generate-before:                   34603008
+    env-step-all-after:                         70287360
+    env-step-replay-after:                      35651584
+    env-step-used-after:                        70287360
+    ppo-update-after:                           42900
+    env-step-generate-after:                    34635776
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.699404
+    avg_avg_return:                             0.94720674
+    avg_reward_per_step:                        0.1894226
+    proxy_corner:
+      avg_avg_return:                           0.004252488
+      avg_reward_per_step:                      0.0005187988
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.699404
+    avg_avg_return:                             0.94720674
+    avg_reward_per_step:                        0.1894226
+    proxy_corner:
+      avg_avg_return:                           0.004252488
+      avg_reward_per_step:                      0.0005187988
+  perf:
+    env_steps_per_second:                       313577.5149406243
+    ppo_updates_per_second:                     41.69255228589706
+    eval-batch-train-duration:                  0.5349822882562876
+    eval-batch-orig-duration:                   0.48133160173892975
+    eval-batch-shift-duration:                  0.44768380746245384
+    eval-batch-tree-duration:                   0.48392935283482075
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.33374
+      cheese-corne_dist_wavg:                   9.134427
+      levels_cheese_is_in_corner_avg:           0.036132812
+      mouse-cheese_dist_finite_avg:             15.448486
+      mouse-cheese_dist_finite_wavg:            12.343716
+      mouse-cheese_dist_solvable_avg:           15.448486
+      mouse-cheese_dist_solvable_wavg:          12.343718
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12081723
+    visit_patterns:
+      avg_first_visit_time:                     741.59546
+      avg_last_visit_time:                      1078.3425
+      num_replay_batches:                       1088
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002893906
+    avg_actor_approxkl3:                        0.0028655545
+    avg_actor_clipfrac:                         0.061016846
+    avg_actor_loss:                             0.0031637556
+    avg_advantage:                              0.0074142194
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00024217817
+    avg_entropy:                                0.112428464
+    avg_grad_norm_pre_clip:                     3.5217896
+    avg_loss:                                   0.003172416
+    max:
+      max_actor_approxkl1:                      0.009750091
+      max_actor_approxkl3:                      0.009467706
+      max_actor_clipfrac:                       0.12988281
+      max_actor_loss:                           0.012781732
+      max_advantage:                            0.082609594
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00031510272
+      max_entropy:                              0.13284037
+      max_grad_norm_pre_clip:                   6.2709293
+      max_loss:                                 0.012771997
+    std:
+      std_actor_approxkl1:                      0.0022514847
+      std_actor_approxkl3:                      0.0021836672
+      std_actor_clipfrac:                       0.027159926
+      std_actor_loss:                           0.002813178
+      std_advantage:                            0.020918688
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.6611035e-05
+      std_entropy:                              0.010958064
+      std_grad_norm_pre_clip:                   1.3105417
+      std_loss:                                 0.0028057806
+  eval-batch-train:
+    avg_avg_episode_length:                     7.5836797
+    avg_avg_return:                             0.9808211
+    avg_benchmark_regret:                       0.011978126
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20096588
+    proxy_corner:
+      avg_avg_return:                           0.007976658
+      avg_reward_per_step:                      0.0015792847
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.492928
+    avg_avg_return:                             0.9750005
+    avg_benchmark_regret:                       0.015239321
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1416626
+    proxy_corner:
+      avg_avg_return:                           0.9750005
+      avg_reward_per_step:                      0.1416626
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.5529222
+    avg_avg_return:                             0.98080796
+    avg_benchmark_regret:                       0.012029466
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19169617
+    proxy_corner:
+      avg_avg_return:                           0.0049902005
+      avg_reward_per_step:                      0.0010986328
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.69566
+    avg_avg_return:                             0.008428377
+    avg_benchmark_regret:                       0.9762068
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0021438599
+    proxy_corner:
+      avg_avg_return:                           0.008428377
+      avg_reward_per_step:                      0.0021438599
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        71303168
+    env-step-replay-before:                     36044800
+    env-step-used-before:                       71303168
+    ppo-update-before:                          43520
+    env-step-generate-before:                   35258368
+    env-step-all-after:                         71335936
+    env-step-replay-after:                      36044800
+    env-step-used-after:                        71335936
+    ppo-update-after:                           43540
+    env-step-generate-after:                    35291136
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.9602785
+    avg_avg_return:                             0.9454168
+    avg_reward_per_step:                        0.18600464
+    proxy_corner:
+      avg_avg_return:                           0.0036652086
+      avg_reward_per_step:                      0.0002746582
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.9602785
+    avg_avg_return:                             0.9454168
+    avg_reward_per_step:                        0.18600464
+    proxy_corner:
+      avg_avg_return:                           0.0036652086
+      avg_reward_per_step:                      0.0002746582
+  perf:
+    env_steps_per_second:                       353565.43606447085
+    ppo_updates_per_second:                     41.9238596704357
+    eval-batch-train-duration:                  0.5067093083634973
+    eval-batch-orig-duration:                   0.48122553154826164
+    eval-batch-shift-duration:                  0.4533320292830467
+    eval-batch-tree-duration:                   0.4675193438306451
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.407715
+      cheese-corne_dist_wavg:                   11.836029
+      levels_cheese_is_in_corner_avg:           0.037109375
+      mouse-cheese_dist_finite_avg:             15.481201
+      mouse-cheese_dist_finite_wavg:            10.549106
+      mouse-cheese_dist_solvable_avg:           15.481201
+      mouse-cheese_dist_solvable_wavg:          10.549107
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12173802
+    visit_patterns:
+      avg_first_visit_time:                     754.3098
+      avg_last_visit_time:                      1090.3191
+      num_replay_batches:                       1100
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.006243032
+    avg_actor_approxkl3:                        0.005222106
+    avg_actor_clipfrac:                         0.08688965
+    avg_actor_loss:                             0.0060176337
+    avg_advantage:                              -0.003859464
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00026870376
+    avg_entropy:                                0.12583338
+    avg_grad_norm_pre_clip:                     4.8668437
+    avg_loss:                                   0.006026152
+    max:
+      max_actor_approxkl1:                      0.012333753
+      max_actor_approxkl3:                      0.009616756
+      max_actor_clipfrac:                       0.122558594
+      max_actor_loss:                           0.010435987
+      max_advantage:                            0.077717185
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00038539903
+      max_entropy:                              0.14521447
+      max_grad_norm_pre_clip:                   8.284331
+      max_loss:                                 0.010444105
+    std:
+      std_actor_approxkl1:                      0.0027712972
+      std_actor_approxkl3:                      0.0022641278
+      std_actor_clipfrac:                       0.026234457
+      std_actor_loss:                           0.0026977651
+      std_advantage:                            0.02022343
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.958113e-05
+      std_entropy:                              0.009498861
+      std_grad_norm_pre_clip:                   1.8596416
+      std_loss:                                 0.002701846
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4282246
+    avg_avg_return:                             0.9814986
+    avg_benchmark_regret:                       0.011300543
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20249176
+    proxy_corner:
+      avg_avg_return:                           0.008541113
+      avg_reward_per_step:                      0.0016403198
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.602388
+    avg_avg_return:                             0.9741137
+    avg_benchmark_regret:                       0.016126169
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14095306
+    proxy_corner:
+      avg_avg_return:                           0.9741137
+      avg_reward_per_step:                      0.14095306
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.39443
+    avg_avg_return:                             0.98148835
+    avg_benchmark_regret:                       0.011349084
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19268036
+    proxy_corner:
+      avg_avg_return:                           0.0043793707
+      avg_reward_per_step:                      0.0010528564
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.68081
+    avg_avg_return:                             0.008514527
+    avg_benchmark_regret:                       0.9761207
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.004432678
+    proxy_corner:
+      avg_avg_return:                           0.008514527
+      avg_reward_per_step:                      0.004432678
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        72351744
+    env-step-replay-before:                     36536320
+    env-step-used-before:                       72351744
+    ppo-update-before:                          44160
+    env-step-generate-before:                   35815424
+    env-step-all-after:                         72384512
+    env-step-replay-after:                      36536320
+    env-step-used-after:                        72384512
+    ppo-update-after:                           44180
+    env-step-generate-after:                    35848192
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.645303
+    avg_avg_return:                             0.94672716
+    avg_reward_per_step:                        0.18203735
+    proxy_corner:
+      avg_avg_return:                           0.012506961
+      avg_reward_per_step:                      0.0012512207
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.645303
+    avg_avg_return:                             0.94672716
+    avg_reward_per_step:                        0.18203735
+    proxy_corner:
+      avg_avg_return:                           0.012506961
+      avg_reward_per_step:                      0.0012512207
+  perf:
+    env_steps_per_second:                       296213.1446630464
+    ppo_updates_per_second:                     42.26456412199507
+    eval-batch-train-duration:                  0.5381406303495169
+    eval-batch-orig-duration:                   0.47646399307996035
+    eval-batch-shift-duration:                  0.43983736634254456
+    eval-batch-tree-duration:                   0.4859447283670306
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.450195
+      cheese-corne_dist_wavg:                   9.160035
+      levels_cheese_is_in_corner_avg:           0.036865234
+      mouse-cheese_dist_finite_avg:             15.483154
+      mouse-cheese_dist_finite_wavg:            15.94639
+      mouse-cheese_dist_solvable_avg:           15.483154
+      mouse-cheese_dist_solvable_wavg:          15.94639
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12170412
+    visit_patterns:
+      avg_first_visit_time:                     766.8093
+      avg_last_visit_time:                      1105.1245
+      num_replay_batches:                       1115
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0025492231
+    avg_actor_approxkl3:                        0.0020498056
+    avg_actor_clipfrac:                         0.0578064
+    avg_actor_loss:                             0.002099445
+    avg_advantage:                              0.0044880267
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00021043129
+    avg_entropy:                                0.12107938
+    avg_grad_norm_pre_clip:                     3.7063954
+    avg_loss:                                   0.0020835816
+    max:
+      max_actor_approxkl1:                      0.0047056247
+      max_actor_approxkl3:                      0.0036145379
+      max_actor_clipfrac:                       0.08068848
+      max_actor_loss:                           0.0052036643
+      max_advantage:                            0.08613706
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00024253619
+      max_entropy:                              0.15126471
+      max_grad_norm_pre_clip:                   5.8605967
+      max_loss:                                 0.0051918104
+    std:
+      std_actor_approxkl1:                      0.001047475
+      std_actor_approxkl3:                      0.0007506369
+      std_actor_clipfrac:                       0.01738945
+      std_actor_loss:                           0.0012273581
+      std_advantage:                            0.020456716
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.8547877e-05
+      std_entropy:                              0.011673198
+      std_grad_norm_pre_clip:                   1.1607875
+      std_loss:                                 0.0012330766
+  eval-batch-train:
+    avg_avg_episode_length:                     7.386411
+    avg_avg_return:                             0.98245394
+    avg_benchmark_regret:                       0.010345227
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20310211
+    proxy_corner:
+      avg_avg_return:                           0.008044061
+      avg_reward_per_step:                      0.0015945435
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.603577
+    avg_avg_return:                             0.97416735
+    avg_benchmark_regret:                       0.016072538
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14089966
+    proxy_corner:
+      avg_avg_return:                           0.97416735
+      avg_reward_per_step:                      0.14089966
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3760304
+    avg_avg_return:                             0.98219043
+    avg_benchmark_regret:                       0.010647021
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19367981
+    proxy_corner:
+      avg_avg_return:                           0.0039899647
+      avg_reward_per_step:                      0.0010070801
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.56972
+    avg_avg_return:                             0.010482687
+    avg_benchmark_regret:                       0.9741525
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.004447937
+    proxy_corner:
+      avg_avg_return:                           0.010482687
+      avg_reward_per_step:                      0.004447937
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        73400320
+    env-step-replay-before:                     37289984
+    env-step-used-before:                       73400320
+    ppo-update-before:                          44800
+    env-step-generate-before:                   36110336
+    env-step-all-after:                         73433088
+    env-step-replay-after:                      37322752
+    env-step-used-after:                        73433088
+    ppo-update-after:                           44820
+    env-step-generate-after:                    36110336
+  train-replay:
+    shift_proportion:                           0.96484375
+    avg_avg_episode_length:                     16.78661
+    avg_avg_return:                             0.86872065
+    avg_reward_per_step:                        0.05682373
+    proxy_corner:
+      avg_avg_return:                           0.04291138
+      avg_reward_per_step:                      0.002746582
+  train-all:
+    shift_proportion:                           0.96484375
+    avg_avg_episode_length:                     16.78661
+    avg_avg_return:                             0.86872065
+    avg_reward_per_step:                        0.05682373
+    proxy_corner:
+      avg_avg_return:                           0.04291138
+      avg_reward_per_step:                      0.002746582
+  perf:
+    env_steps_per_second:                       353101.6632186195
+    ppo_updates_per_second:                     41.370203597431576
+    eval-batch-train-duration:                  0.5272697163745761
+    eval-batch-orig-duration:                   0.4787049535661936
+    eval-batch-shift-duration:                  0.4521892173215747
+    eval-batch-tree-duration:                   0.4818836944177747
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.45459
+      cheese-corne_dist_wavg:                   11.826254
+      levels_cheese_is_in_corner_avg:           0.037109375
+      mouse-cheese_dist_finite_avg:             15.45752
+      mouse-cheese_dist_finite_wavg:            16.846796
+      mouse-cheese_dist_solvable_avg:           15.45752
+      mouse-cheese_dist_solvable_wavg:          16.846796
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11836701
+    visit_patterns:
+      avg_first_visit_time:                     774.69507
+      avg_last_visit_time:                      1129.2473
+      num_replay_batches:                       1139
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0021780638
+    avg_actor_approxkl3:                        0.0018035466
+    avg_actor_clipfrac:                         0.06456909
+    avg_actor_loss:                             0.0014279423
+    avg_advantage:                              -0.00036855033
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00015311151
+    avg_entropy:                                0.16888694
+    avg_grad_norm_pre_clip:                     3.9363396
+    avg_loss:                                   0.001335611
+    max:
+      max_actor_approxkl1:                      0.0067283325
+      max_actor_approxkl3:                      0.0051406706
+      max_actor_clipfrac:                       0.11633301
+      max_actor_loss:                           0.007269905
+      max_advantage:                            0.082473755
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00016515248
+      max_entropy:                              0.19481157
+      max_grad_norm_pre_clip:                   6.6496234
+      max_loss:                                 0.007180752
+    std:
+      std_actor_approxkl1:                      0.001550867
+      std_actor_approxkl3:                      0.0011117713
+      std_actor_clipfrac:                       0.02393028
+      std_actor_loss:                           0.0016665336
+      std_advantage:                            0.01746847
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.4365616e-06
+      std_entropy:                              0.012722495
+      std_grad_norm_pre_clip:                   1.4177117
+      std_loss:                                 0.0016652629
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4651155
+    avg_avg_return:                             0.9815871
+    avg_benchmark_regret:                       0.011212058
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20111084
+    proxy_corner:
+      avg_avg_return:                           0.007992557
+      avg_reward_per_step:                      0.0016021729
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.482057
+    avg_avg_return:                             0.9752275
+    avg_benchmark_regret:                       0.015012335
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14190674
+    proxy_corner:
+      avg_avg_return:                           0.9752275
+      avg_reward_per_step:                      0.14190674
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.366802
+    avg_avg_return:                             0.98175037
+    avg_benchmark_regret:                       0.011087017
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19311523
+    proxy_corner:
+      avg_avg_return:                           0.004914083
+      avg_reward_per_step:                      0.001121521
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.44657
+    avg_avg_return:                             0.015568736
+    avg_benchmark_regret:                       0.9690664
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0043792725
+    proxy_corner:
+      avg_avg_return:                           0.015568736
+      avg_reward_per_step:                      0.0043792725
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        74448896
+    env-step-replay-before:                     38043648
+    env-step-used-before:                       74448896
+    ppo-update-before:                          45440
+    env-step-generate-before:                   36405248
+    env-step-all-after:                         74481664
+    env-step-replay-after:                      38043648
+    env-step-used-after:                        74481664
+    ppo-update-after:                           45460
+    env-step-generate-after:                    36438016
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.614419
+    avg_avg_return:                             0.94686496
+    avg_reward_per_step:                        0.19281006
+    proxy_corner:
+      avg_avg_return:                           0.013487896
+      avg_reward_per_step:                      0.00091552734
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.614419
+    avg_avg_return:                             0.94686496
+    avg_reward_per_step:                        0.19281006
+    proxy_corner:
+      avg_avg_return:                           0.013487896
+      avg_reward_per_step:                      0.00091552734
+  perf:
+    env_steps_per_second:                       262106.15780730554
+    ppo_updates_per_second:                     41.64255844092797
+    eval-batch-train-duration:                  0.5391971291974187
+    eval-batch-orig-duration:                   0.4826009003445506
+    eval-batch-shift-duration:                  0.4842310259118676
+    eval-batch-tree-duration:                   0.4387121368199587
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.432373
+      cheese-corne_dist_wavg:                   2.8387942
+      levels_cheese_is_in_corner_avg:           0.038085938
+      mouse-cheese_dist_finite_avg:             15.468018
+      mouse-cheese_dist_finite_wavg:            15.946504
+      mouse-cheese_dist_solvable_avg:           15.468018
+      mouse-cheese_dist_solvable_wavg:          15.946506
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.118991226
+    visit_patterns:
+      avg_first_visit_time:                     781.208
+      avg_last_visit_time:                      1151.2996
+      num_replay_batches:                       1161
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00270316
+    avg_actor_approxkl3:                        0.002178944
+    avg_actor_clipfrac:                         0.05390625
+    avg_actor_loss:                             0.0018165013
+    avg_advantage:                              0.013661853
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0003470426
+    avg_entropy:                                0.110398665
+    avg_grad_norm_pre_clip:                     3.3823395
+    avg_loss:                                   0.0018796241
+    max:
+      max_actor_approxkl1:                      0.0063774516
+      max_actor_approxkl3:                      0.0042571644
+      max_actor_clipfrac:                       0.091430664
+      max_actor_loss:                           0.005058298
+      max_advantage:                            0.1130414
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0007086842
+      max_entropy:                              0.12925473
+      max_grad_norm_pre_clip:                   6.033793
+      max_loss:                                 0.005183801
+    std:
+      std_actor_approxkl1:                      0.00157152
+      std_actor_approxkl3:                      0.0010700191
+      std_actor_clipfrac:                       0.019916564
+      std_actor_loss:                           0.0014419595
+      std_advantage:                            0.021481846
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00012688826
+      std_entropy:                              0.010965213
+      std_grad_norm_pre_clip:                   1.4326824
+      std_loss:                                 0.0014403573
+  eval-batch-train:
+    avg_avg_episode_length:                     7.480901
+    avg_avg_return:                             0.9817934
+    avg_benchmark_regret:                       0.0110058
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20057678
+    proxy_corner:
+      avg_avg_return:                           0.009412577
+      avg_reward_per_step:                      0.0016860962
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.497206
+    avg_avg_return:                             0.97489214
+    avg_benchmark_regret:                       0.015347718
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14182281
+    proxy_corner:
+      avg_avg_return:                           0.97489214
+      avg_reward_per_step:                      0.14182281
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.424714
+    avg_avg_return:                             0.98164123
+    avg_benchmark_regret:                       0.011196176
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.1918869
+    proxy_corner:
+      avg_avg_return:                           0.0062111653
+      avg_reward_per_step:                      0.0013656616
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.48605
+    avg_avg_return:                             0.014147856
+    avg_benchmark_regret:                       0.9704873
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.004753113
+    proxy_corner:
+      avg_avg_return:                           0.014147856
+      avg_reward_per_step:                      0.004753113
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        75497472
+    env-step-replay-before:                     38567936
+    env-step-used-before:                       75497472
+    ppo-update-before:                          46080
+    env-step-generate-before:                   36929536
+    env-step-all-after:                         75530240
+    env-step-replay-after:                      38600704
+    env-step-used-after:                        75530240
+    ppo-update-after:                           46100
+    env-step-generate-after:                    36929536
+  train-replay:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     16.461052
+    avg_avg_return:                             0.87163484
+    avg_reward_per_step:                        0.05758667
+    proxy_corner:
+      avg_avg_return:                           0.027242772
+      avg_reward_per_step:                      0.0017700195
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     16.461052
+    avg_avg_return:                             0.87163484
+    avg_reward_per_step:                        0.05758667
+    proxy_corner:
+      avg_avg_return:                           0.027242772
+      avg_reward_per_step:                      0.0017700195
+  perf:
+    env_steps_per_second:                       270888.5023415855
+    ppo_updates_per_second:                     42.07261082233418
+    eval-batch-train-duration:                  0.6650583194568753
+    eval-batch-orig-duration:                   0.6270623123273253
+    eval-batch-shift-duration:                  0.6124147921800613
+    eval-batch-tree-duration:                   0.6080886228010058
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.425293
+      cheese-corne_dist_wavg:                   9.145928
+      levels_cheese_is_in_corner_avg:           0.038330078
+      mouse-cheese_dist_finite_avg:             15.496582
+      mouse-cheese_dist_finite_wavg:            15.045975
+      mouse-cheese_dist_solvable_avg:           15.496582
+      mouse-cheese_dist_solvable_wavg:          15.045975
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11897123
+    visit_patterns:
+      avg_first_visit_time:                     791.44507
+      avg_last_visit_time:                      1168.3066
+      num_replay_batches:                       1178
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0021655287
+    avg_actor_approxkl3:                        0.00134082
+    avg_actor_clipfrac:                         0.05463257
+    avg_actor_loss:                             0.00078193395
+    avg_advantage:                              -0.00840232
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00015898008
+    avg_entropy:                                0.16164875
+    avg_grad_norm_pre_clip:                     3.0490193
+    avg_loss:                                   0.0006997754
+    max:
+      max_actor_approxkl1:                      0.0038539171
+      max_actor_approxkl3:                      0.002534056
+      max_actor_clipfrac:                       0.08996582
+      max_actor_loss:                           0.0028051063
+      max_advantage:                            0.07920921
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0001849915
+      max_entropy:                              0.18373993
+      max_grad_norm_pre_clip:                   5.2789264
+      max_loss:                                 0.0027224757
+    std:
+      std_actor_approxkl1:                      0.00097179634
+      std_actor_approxkl3:                      0.00057417393
+      std_actor_clipfrac:                       0.01995842
+      std_actor_loss:                           0.00081491086
+      std_advantage:                            0.017334621
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.2574637e-05
+      std_entropy:                              0.009349159
+      std_grad_norm_pre_clip:                   1.1688602
+      std_loss:                                 0.0008168078
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4325333
+    avg_avg_return:                             0.9818746
+    avg_benchmark_regret:                       0.010924622
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20217896
+    proxy_corner:
+      avg_avg_return:                           0.008289294
+      avg_reward_per_step:                      0.0016174316
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.484829
+    avg_avg_return:                             0.97499806
+    avg_benchmark_regret:                       0.015241772
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14176941
+    proxy_corner:
+      avg_avg_return:                           0.97499806
+      avg_reward_per_step:                      0.14176941
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.389552
+    avg_avg_return:                             0.98152435
+    avg_benchmark_regret:                       0.0113130845
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19352722
+    proxy_corner:
+      avg_avg_return:                           0.0057092234
+      avg_reward_per_step:                      0.0011825562
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.63275
+    avg_avg_return:                             0.0096288845
+    avg_benchmark_regret:                       0.97500634
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.004638672
+    proxy_corner:
+      avg_avg_return:                           0.0096288845
+      avg_reward_per_step:                      0.004638672
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        76546048
+    env-step-replay-before:                     39092224
+    env-step-used-before:                       76546048
+    ppo-update-before:                          46720
+    env-step-generate-before:                   37453824
+    env-step-all-after:                         76578816
+    env-step-replay-after:                      39092224
+    env-step-used-after:                        76578816
+    ppo-update-after:                           46740
+    env-step-generate-after:                    37486592
+  train-generate:
+    shift_proportion:                           1.0
+    avg_avg_episode_length:                     7.913762
+    avg_avg_return:                             0.94365454
+    avg_reward_per_step:                        0.18359375
+    proxy_corner:
+      avg_avg_return:                           0.008796238
+      avg_reward_per_step:                      0.0007019043
+  train-all:
+    shift_proportion:                           1.0
+    avg_avg_episode_length:                     7.913762
+    avg_avg_return:                             0.94365454
+    avg_reward_per_step:                        0.18359375
+    proxy_corner:
+      avg_avg_return:                           0.008796238
+      avg_reward_per_step:                      0.0007019043
+  perf:
+    env_steps_per_second:                       308993.3049333574
+    ppo_updates_per_second:                     42.124297313557456
+    eval-batch-train-duration:                  0.517947256565094
+    eval-batch-orig-duration:                   0.4853159999474883
+    eval-batch-shift-duration:                  0.4784558517858386
+    eval-batch-tree-duration:                   0.4695222703740001
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.505615
+      cheese-corne_dist_wavg:                   14.545424
+      levels_cheese_is_in_corner_avg:           0.03930664
+      mouse-cheese_dist_finite_avg:             15.546387
+      mouse-cheese_dist_finite_wavg:            12.351459
+      mouse-cheese_dist_solvable_avg:           15.546387
+      mouse-cheese_dist_solvable_wavg:          12.351459
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.121550724
+    visit_patterns:
+      avg_first_visit_time:                     804.1028
+      avg_last_visit_time:                      1183.1934
+      num_replay_batches:                       1193
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0018058749
+    avg_actor_approxkl3:                        0.0015609851
+    avg_actor_clipfrac:                         0.048620608
+    avg_actor_loss:                             0.0011916035
+    avg_advantage:                              0.006002913
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00024283161
+    avg_entropy:                                0.12034426
+    avg_grad_norm_pre_clip:                     2.7784765
+    avg_loss:                                   0.0011926753
+    max:
+      max_actor_approxkl1:                      0.0037934126
+      max_actor_approxkl3:                      0.003598048
+      max_actor_clipfrac:                       0.08251953
+      max_actor_loss:                           0.0039024148
+      max_advantage:                            0.08868694
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00029581384
+      max_entropy:                              0.14065053
+      max_grad_norm_pre_clip:                   4.773935
+      max_loss:                                 0.0038865795
+    std:
+      std_actor_approxkl1:                      0.0009947497
+      std_actor_approxkl3:                      0.00088650157
+      std_actor_clipfrac:                       0.019455116
+      std_actor_loss:                           0.0012324852
+      std_advantage:                            0.02127767
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.7286796e-05
+      std_entropy:                              0.009706705
+      std_grad_norm_pre_clip:                   1.0634673
+      std_loss:                                 0.001225546
+  eval-batch-train:
+    avg_avg_episode_length:                     7.491109
+    avg_avg_return:                             0.981707
+    avg_benchmark_regret:                       0.011092184
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20232391
+    proxy_corner:
+      avg_avg_return:                           0.008895785
+      avg_reward_per_step:                      0.0016555786
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.492109
+    avg_avg_return:                             0.97485256
+    avg_benchmark_regret:                       0.015387258
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14174652
+    proxy_corner:
+      avg_avg_return:                           0.97485256
+      avg_reward_per_step:                      0.14174652
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4674506
+    avg_avg_return:                             0.98099977
+    avg_benchmark_regret:                       0.011837669
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19325256
+    proxy_corner:
+      avg_avg_return:                           0.007808366
+      avg_reward_per_step:                      0.0015106201
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.63112
+    avg_avg_return:                             0.011145153
+    avg_benchmark_regret:                       0.97349
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0055770874
+    proxy_corner:
+      avg_avg_return:                           0.011145153
+      avg_reward_per_step:                      0.0055770874
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        77594624
+    env-step-replay-before:                     39714816
+    env-step-used-before:                       77594624
+    ppo-update-before:                          47360
+    env-step-generate-before:                   37879808
+    env-step-all-after:                         77627392
+    env-step-replay-after:                      39714816
+    env-step-used-after:                        77627392
+    ppo-update-after:                           47380
+    env-step-generate-after:                    37912576
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.3863025
+    avg_avg_return:                             0.9488213
+    avg_reward_per_step:                        0.20510864
+    proxy_corner:
+      avg_avg_return:                           0.012342932
+      avg_reward_per_step:                      0.0010986328
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.3863025
+    avg_avg_return:                             0.9488213
+    avg_reward_per_step:                        0.20510864
+    proxy_corner:
+      avg_avg_return:                           0.012342932
+      avg_reward_per_step:                      0.0010986328
+  perf:
+    env_steps_per_second:                       244472.94267019344
+    ppo_updates_per_second:                     41.96772694159588
+    eval-batch-train-duration:                  0.5402435809373856
+    eval-batch-orig-duration:                   0.4807848483324051
+    eval-batch-shift-duration:                  0.48024440836161375
+    eval-batch-tree-duration:                   0.45587954204529524
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.470703
+      cheese-corne_dist_wavg:                   14.53965
+      levels_cheese_is_in_corner_avg:           0.039794922
+      mouse-cheese_dist_finite_avg:             15.5
+      mouse-cheese_dist_finite_wavg:            15.049036
+      mouse-cheese_dist_solvable_avg:           15.5
+      mouse-cheese_dist_solvable_wavg:          15.049036
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12292969
+    visit_patterns:
+      avg_first_visit_time:                     815.9385
+      avg_last_visit_time:                      1202.331
+      num_replay_batches:                       1212
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0016136704
+    avg_actor_approxkl3:                        0.0013041011
+    avg_actor_clipfrac:                         0.04411621
+    avg_actor_loss:                             0.00072510575
+    avg_advantage:                              -0.0028264346
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00022863173
+    avg_entropy:                                0.120947614
+    avg_grad_norm_pre_clip:                     2.4388244
+    avg_loss:                                   0.00071847404
+    max:
+      max_actor_approxkl1:                      0.004515808
+      max_actor_approxkl3:                      0.0033460965
+      max_actor_clipfrac:                       0.08654785
+      max_actor_loss:                           0.0030419994
+      max_advantage:                            0.08578879
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00027513446
+      max_entropy:                              0.14821032
+      max_grad_norm_pre_clip:                   5.661094
+      max_loss:                                 0.002997394
+    std:
+      std_actor_approxkl1:                      0.0012362562
+      std_actor_approxkl3:                      0.000737215
+      std_actor_clipfrac:                       0.018664721
+      std_actor_loss:                           0.0007529512
+      std_advantage:                            0.021326499
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.475015e-05
+      std_entropy:                              0.014964381
+      std_grad_norm_pre_clip:                   1.2833297
+      std_loss:                                 0.0007484913
+  eval-batch-train:
+    avg_avg_episode_length:                     7.583619
+    avg_avg_return:                             0.9808168
+    avg_benchmark_regret:                       0.011982364
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20267487
+    proxy_corner:
+      avg_avg_return:                           0.010850973
+      avg_reward_per_step:                      0.0017776489
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.487301
+    avg_avg_return:                             0.9750301
+    avg_benchmark_regret:                       0.0152096925
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14176941
+    proxy_corner:
+      avg_avg_return:                           0.9750301
+      avg_reward_per_step:                      0.14176941
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.462225
+    avg_avg_return:                             0.98171043
+    avg_benchmark_regret:                       0.011126983
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19306946
+    proxy_corner:
+      avg_avg_return:                           0.008988786
+      avg_reward_per_step:                      0.0018997192
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.349365
+    avg_avg_return:                             0.016438792
+    avg_benchmark_regret:                       0.9681964
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0051498413
+    proxy_corner:
+      avg_avg_return:                           0.016438792
+      avg_reward_per_step:                      0.0051498413
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        78643200
+    env-step-replay-before:                     40239104
+    env-step-used-before:                       78643200
+    ppo-update-before:                          48000
+    env-step-generate-before:                   38404096
+    env-step-all-after:                         78675968
+    env-step-replay-after:                      40271872
+    env-step-used-after:                        78675968
+    ppo-update-after:                           48020
+    env-step-generate-after:                    38404096
+  train-replay:
+    shift_proportion:                           0.9453125
+    avg_avg_episode_length:                     16.901773
+    avg_avg_return:                             0.8684977
+    avg_reward_per_step:                        0.056274414
+    proxy_corner:
+      avg_avg_return:                           0.05707067
+      avg_reward_per_step:                      0.0037231445
+  train-all:
+    shift_proportion:                           0.9453125
+    avg_avg_episode_length:                     16.901773
+    avg_avg_return:                             0.8684977
+    avg_reward_per_step:                        0.056274414
+    proxy_corner:
+      avg_avg_return:                           0.05707067
+      avg_reward_per_step:                      0.0037231445
+  perf:
+    env_steps_per_second:                       295522.8516930591
+    ppo_updates_per_second:                     42.364997443137035
+    eval-batch-train-duration:                  0.5453281458467245
+    eval-batch-orig-duration:                   0.4786058748140931
+    eval-batch-shift-duration:                  0.4814696563407779
+    eval-batch-tree-duration:                   0.45760099217295647
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.511963
+      cheese-corne_dist_wavg:                   3.7678337
+      levels_cheese_is_in_corner_avg:           0.040771484
+      mouse-cheese_dist_finite_avg:             15.51001
+      mouse-cheese_dist_finite_wavg:            8.75938
+      mouse-cheese_dist_solvable_avg:           15.51001
+      mouse-cheese_dist_solvable_wavg:          8.759381
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12040709
+    visit_patterns:
+      avg_first_visit_time:                     827.84595
+      avg_last_visit_time:                      1219.2185
+      num_replay_batches:                       1229
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0021852944
+    avg_actor_approxkl3:                        0.0016635674
+    avg_actor_clipfrac:                         0.06583252
+    avg_actor_loss:                             0.0011022268
+    avg_advantage:                              0.004686761
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00014051351
+    avg_entropy:                                0.17414911
+    avg_grad_norm_pre_clip:                     3.625494
+    avg_loss:                                   0.0009983345
+    max:
+      max_actor_approxkl1:                      0.0045918887
+      max_actor_approxkl3:                      0.0032252073
+      max_actor_clipfrac:                       0.115478516
+      max_actor_loss:                           0.003976412
+      max_advantage:                            0.07918733
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00015234333
+      max_entropy:                              0.20014016
+      max_grad_norm_pre_clip:                   5.8366823
+      max_loss:                                 0.0038808933
+    std:
+      std_actor_approxkl1:                      0.0012223229
+      std_actor_approxkl3:                      0.00082007056
+      std_actor_clipfrac:                       0.027043385
+      std_actor_loss:                           0.0011399728
+      std_advantage:                            0.016643167
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.75313e-06
+      std_entropy:                              0.01167107
+      std_grad_norm_pre_clip:                   1.1364603
+      std_loss:                                 0.0011401912
+  eval-batch-train:
+    avg_avg_episode_length:                     7.453165
+    avg_avg_return:                             0.98136634
+    avg_benchmark_regret:                       0.011432817
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.2025528
+    proxy_corner:
+      avg_avg_return:                           0.008310274
+      avg_reward_per_step:                      0.0016326904
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.483986
+    avg_avg_return:                             0.9750463
+    avg_benchmark_regret:                       0.01519351
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14179993
+    proxy_corner:
+      avg_avg_return:                           0.9750463
+      avg_reward_per_step:                      0.14179993
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3984632
+    avg_avg_return:                             0.98140657
+    avg_benchmark_regret:                       0.01143083
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19423676
+    proxy_corner:
+      avg_avg_return:                           0.0066984147
+      avg_reward_per_step:                      0.0015487671
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.68242
+    avg_avg_return:                             0.0077902554
+    avg_benchmark_regret:                       0.97684497
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.003868103
+    proxy_corner:
+      avg_avg_return:                           0.0077902554
+      avg_reward_per_step:                      0.003868103
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        79691776
+    env-step-replay-before:                     40861696
+    env-step-used-before:                       79691776
+    ppo-update-before:                          48640
+    env-step-generate-before:                   38830080
+    env-step-all-after:                         79724544
+    env-step-replay-after:                      40861696
+    env-step-used-after:                        79724544
+    ppo-update-after:                           48660
+    env-step-generate-after:                    38862848
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.7383833
+    avg_avg_return:                             0.94584507
+    avg_reward_per_step:                        0.18450928
+    proxy_corner:
+      avg_avg_return:                           0.014883666
+      avg_reward_per_step:                      0.0021362305
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.7383833
+    avg_avg_return:                             0.94584507
+    avg_reward_per_step:                        0.18450928
+    proxy_corner:
+      avg_avg_return:                           0.014883666
+      avg_reward_per_step:                      0.0021362305
+  perf:
+    env_steps_per_second:                       250892.76268197305
+    ppo_updates_per_second:                     42.055153947808556
+    eval-batch-train-duration:                  0.5386041291058064
+    eval-batch-orig-duration:                   0.48208558931946754
+    eval-batch-shift-duration:                  0.4757365817204118
+    eval-batch-tree-duration:                   0.46664653439074755
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.545898
+      cheese-corne_dist_wavg:                   6.4645514
+      levels_cheese_is_in_corner_avg:           0.040771484
+      mouse-cheese_dist_finite_avg:             15.503174
+      mouse-cheese_dist_finite_wavg:            11.452608
+      mouse-cheese_dist_solvable_avg:           15.503174
+      mouse-cheese_dist_solvable_wavg:          11.452608
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.120321184
+    visit_patterns:
+      avg_first_visit_time:                     838.8574
+      avg_last_visit_time:                      1237.2756
+      num_replay_batches:                       1247
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0024250953
+    avg_actor_approxkl3:                        0.002184975
+    avg_actor_clipfrac:                         0.059503175
+    avg_actor_loss:                             0.0019895984
+    avg_advantage:                              0.0038413934
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00020367675
+    avg_entropy:                                0.11442502
+    avg_grad_norm_pre_clip:                     3.2414854
+    avg_loss:                                   0.0019770116
+    max:
+      max_actor_approxkl1:                      0.0060882615
+      max_actor_approxkl3:                      0.0044577047
+      max_actor_clipfrac:                       0.08728027
+      max_actor_loss:                           0.0052489843
+      max_advantage:                            0.09546709
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00024092635
+      max_entropy:                              0.13203067
+      max_grad_norm_pre_clip:                   5.3202877
+      max_loss:                                 0.0052399347
+    std:
+      std_actor_approxkl1:                      0.0013246932
+      std_actor_approxkl3:                      0.0009766398
+      std_actor_clipfrac:                       0.01958966
+      std_actor_loss:                           0.0014126586
+      std_advantage:                            0.019731242
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.991081e-05
+      std_entropy:                              0.0103471195
+      std_grad_norm_pre_clip:                   1.2448608
+      std_loss:                                 0.0014153188
+  eval-batch-train:
+    avg_avg_episode_length:                     7.51188
+    avg_avg_return:                             0.98137885
+    avg_benchmark_regret:                       0.011420314
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20259857
+    proxy_corner:
+      avg_avg_return:                           0.008772227
+      avg_reward_per_step:                      0.0016479492
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.520014
+    avg_avg_return:                             0.9750576
+    avg_benchmark_regret:                       0.015182209
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14155579
+    proxy_corner:
+      avg_avg_return:                           0.9750576
+      avg_reward_per_step:                      0.14155579
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4806833
+    avg_avg_return:                             0.9816324
+    avg_benchmark_regret:                       0.011205043
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19385529
+    proxy_corner:
+      avg_avg_return:                           0.005904706
+      avg_reward_per_step:                      0.0013656616
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.68785
+    avg_avg_return:                             0.009288783
+    avg_benchmark_regret:                       0.9753464
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0028686523
+    proxy_corner:
+      avg_avg_return:                           0.009288783
+      avg_reward_per_step:                      0.0028686523
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        80740352
+    env-step-replay-before:                     41222144
+    env-step-used-before:                       80740352
+    ppo-update-before:                          49280
+    env-step-generate-before:                   39518208
+    env-step-all-after:                         80773120
+    env-step-replay-after:                      41254912
+    env-step-used-after:                        80773120
+    ppo-update-after:                           49300
+    env-step-generate-after:                    39518208
+  train-replay:
+    shift_proportion:                           0.9453125
+    avg_avg_episode_length:                     16.582373
+    avg_avg_return:                             0.8677561
+    avg_reward_per_step:                        0.057434082
+    proxy_corner:
+      avg_avg_return:                           0.04967186
+      avg_reward_per_step:                      0.0031433105
+  train-all:
+    shift_proportion:                           0.9453125
+    avg_avg_episode_length:                     16.582373
+    avg_avg_return:                             0.8677561
+    avg_reward_per_step:                        0.057434082
+    proxy_corner:
+      avg_avg_return:                           0.04967186
+      avg_reward_per_step:                      0.0031433105
+  perf:
+    env_steps_per_second:                       320850.2127829043
+    ppo_updates_per_second:                     42.22541686723647
+    eval-batch-train-duration:                  0.5292986398562789
+    eval-batch-orig-duration:                   0.4898183597251773
+    eval-batch-shift-duration:                  0.4757794924080372
+    eval-batch-tree-duration:                   0.5006077960133553
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.625977
+      cheese-corne_dist_wavg:                   11.860869
+      levels_cheese_is_in_corner_avg:           0.040283203
+      mouse-cheese_dist_finite_avg:             15.542969
+      mouse-cheese_dist_finite_wavg:            15.952671
+      mouse-cheese_dist_solvable_avg:           15.542969
+      mouse-cheese_dist_solvable_wavg:          15.952673
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.121519536
+    visit_patterns:
+      avg_first_visit_time:                     852.13794
+      avg_last_visit_time:                      1249.2585
+      num_replay_batches:                       1259
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0024585219
+    avg_actor_approxkl3:                        0.0022354468
+    avg_actor_clipfrac:                         0.07831421
+    avg_actor_loss:                             0.0022971062
+    avg_advantage:                              -0.00056990737
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00014120922
+    avg_entropy:                                0.17986147
+    avg_grad_norm_pre_clip:                     3.790985
+    avg_loss:                                   0.0021878497
+    max:
+      max_actor_approxkl1:                      0.005616502
+      max_actor_approxkl3:                      0.0050042076
+      max_actor_clipfrac:                       0.122924805
+      max_actor_loss:                           0.0055921385
+      max_advantage:                            0.06886995
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00015552815
+      max_entropy:                              0.2042704
+      max_grad_norm_pre_clip:                   5.845312
+      max_loss:                                 0.005481733
+    std:
+      std_actor_approxkl1:                      0.0012686395
+      std_actor_approxkl3:                      0.0011767314
+      std_actor_clipfrac:                       0.02841692
+      std_actor_loss:                           0.0017775693
+      std_advantage:                            0.016586803
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          6.2979057e-06
+      std_entropy:                              0.011642011
+      std_grad_norm_pre_clip:                   1.1144418
+      std_loss:                                 0.0017690471
+  eval-batch-train:
+    avg_avg_episode_length:                     7.409665
+    avg_avg_return:                             0.981755
+    avg_benchmark_regret:                       0.011044167
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20309448
+    proxy_corner:
+      avg_avg_return:                           0.008769225
+      avg_reward_per_step:                      0.0016555786
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.545522
+    avg_avg_return:                             0.97422385
+    avg_benchmark_regret:                       0.01601601
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14146423
+    proxy_corner:
+      avg_avg_return:                           0.97422385
+      avg_reward_per_step:                      0.14146423
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.363704
+    avg_avg_return:                             0.98197347
+    avg_benchmark_regret:                       0.0108639635
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19480896
+    proxy_corner:
+      avg_avg_return:                           0.006225149
+      avg_reward_per_step:                      0.0014190674
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.46419
+    avg_avg_return:                             0.01349378
+    avg_benchmark_regret:                       0.9711414
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0046844482
+    proxy_corner:
+      avg_avg_return:                           0.01349378
+      avg_reward_per_step:                      0.0046844482
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        81788928
+    env-step-replay-before:                     41615360
+    env-step-used-before:                       81788928
+    ppo-update-before:                          49920
+    env-step-generate-before:                   40173568
+    env-step-all-after:                         81821696
+    env-step-replay-after:                      41648128
+    env-step-used-after:                        81821696
+    ppo-update-after:                           49940
+    env-step-generate-after:                    40173568
+  train-replay:
+    shift_proportion:                           0.96484375
+    avg_avg_episode_length:                     17.456038
+    avg_avg_return:                             0.8644669
+    avg_reward_per_step:                        0.0546875
+    proxy_corner:
+      avg_avg_return:                           0.032458786
+      avg_reward_per_step:                      0.002105713
+  train-all:
+    shift_proportion:                           0.96484375
+    avg_avg_episode_length:                     17.456038
+    avg_avg_return:                             0.8644669
+    avg_reward_per_step:                        0.0546875
+    proxy_corner:
+      avg_avg_return:                           0.032458786
+      avg_reward_per_step:                      0.002105713
+  perf:
+    env_steps_per_second:                       246566.86109179357
+    ppo_updates_per_second:                     45.005354923555316
+    eval-batch-train-duration:                  0.6547440541908145
+    eval-batch-orig-duration:                   0.6095777796581388
+    eval-batch-shift-duration:                  0.5548521196469665
+    eval-batch-tree-duration:                   0.6470617773011327
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.70166
+      cheese-corne_dist_wavg:                   18.153736
+      levels_cheese_is_in_corner_avg:           0.040283203
+      mouse-cheese_dist_finite_avg:             15.563965
+      mouse-cheese_dist_finite_wavg:            16.856634
+      mouse-cheese_dist_solvable_avg:           15.563965
+      mouse-cheese_dist_solvable_wavg:          16.856636
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12169161
+    visit_patterns:
+      avg_first_visit_time:                     865.78125
+      avg_last_visit_time:                      1261.2622
+      num_replay_batches:                       1271
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0032110699
+    avg_actor_approxkl3:                        0.002279804
+    avg_actor_clipfrac:                         0.08131714
+    avg_actor_loss:                             0.0009858672
+    avg_advantage:                              -0.0007313555
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00012820272
+    avg_entropy:                                0.17673568
+    avg_grad_norm_pre_clip:                     3.6001923
+    avg_loss:                                   0.0008732329
+    max:
+      max_actor_approxkl1:                      0.0071763517
+      max_actor_approxkl3:                      0.005237102
+      max_actor_clipfrac:                       0.13513184
+      max_actor_loss:                           0.0033035493
+      max_advantage:                            0.085546374
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00014729146
+      max_entropy:                              0.19549999
+      max_grad_norm_pre_clip:                   5.6544456
+      max_loss:                                 0.0031868115
+    std:
+      std_actor_approxkl1:                      0.001983574
+      std_actor_approxkl3:                      0.0013168678
+      std_actor_clipfrac:                       0.03561568
+      std_actor_loss:                           0.0012182523
+      std_advantage:                            0.015978832
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          8.477415e-06
+      std_entropy:                              0.00982655
+      std_grad_norm_pre_clip:                   1.3185488
+      std_loss:                                 0.0012192449
+  eval-batch-train:
+    avg_avg_episode_length:                     7.44038
+    avg_avg_return:                             0.98144144
+    avg_benchmark_regret:                       0.011357738
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20334625
+    proxy_corner:
+      avg_avg_return:                           0.009535071
+      avg_reward_per_step:                      0.0017089844
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.487172
+    avg_avg_return:                             0.9746264
+    avg_benchmark_regret:                       0.015613405
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14183044
+    proxy_corner:
+      avg_avg_return:                           0.9746264
+      avg_reward_per_step:                      0.14183044
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.460266
+    avg_avg_return:                             0.98126125
+    avg_benchmark_regret:                       0.011576176
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.1931076
+    proxy_corner:
+      avg_avg_return:                           0.009369101
+      avg_reward_per_step:                      0.0019989014
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.24974
+    avg_avg_return:                             0.017689543
+    avg_benchmark_regret:                       0.96694565
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0055389404
+    proxy_corner:
+      avg_avg_return:                           0.017689543
+      avg_reward_per_step:                      0.0055389404
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        82837504
+    env-step-replay-before:                     42336256
+    env-step-used-before:                       82837504
+    ppo-update-before:                          50560
+    env-step-generate-before:                   40501248
+    env-step-all-after:                         82870272
+    env-step-replay-after:                      42369024
+    env-step-used-after:                        82870272
+    ppo-update-after:                           50580
+    env-step-generate-after:                    40501248
+  train-replay:
+    shift_proportion:                           0.96875
+    avg_avg_episode_length:                     17.273495
+    avg_avg_return:                             0.86592484
+    avg_reward_per_step:                        0.05508423
+    proxy_corner:
+      avg_avg_return:                           0.029586101
+      avg_reward_per_step:                      0.0018615723
+  train-all:
+    shift_proportion:                           0.96875
+    avg_avg_episode_length:                     17.273495
+    avg_avg_return:                             0.86592484
+    avg_reward_per_step:                        0.05508423
+    proxy_corner:
+      avg_avg_return:                           0.029586101
+      avg_reward_per_step:                      0.0018615723
+  perf:
+    env_steps_per_second:                       295470.296028999
+    ppo_updates_per_second:                     42.561510452459196
+    eval-batch-train-duration:                  0.5327158849686384
+    eval-batch-orig-duration:                   0.4826861498877406
+    eval-batch-shift-duration:                  0.47977462131530046
+    eval-batch-tree-duration:                   0.4749913392588496
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.688721
+      cheese-corne_dist_wavg:                   19.075413
+      levels_cheese_is_in_corner_avg:           0.040771484
+      mouse-cheese_dist_finite_avg:             15.57959
+      mouse-cheese_dist_finite_wavg:            18.659784
+      mouse-cheese_dist_solvable_avg:           15.57959
+      mouse-cheese_dist_solvable_wavg:          18.659786
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11920218
+    visit_patterns:
+      avg_first_visit_time:                     871.5847
+      avg_last_visit_time:                      1283.23
+      num_replay_batches:                       1293
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0027962795
+    avg_actor_approxkl3:                        0.0023768821
+    avg_actor_clipfrac:                         0.071905516
+    avg_actor_loss:                             0.0014159402
+    avg_advantage:                              0.0009667233
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00011819531
+    avg_entropy:                                0.16134422
+    avg_grad_norm_pre_clip:                     3.2110384
+    avg_loss:                                   0.0013136936
+    max:
+      max_actor_approxkl1:                      0.005014601
+      max_actor_approxkl3:                      0.0045612985
+      max_actor_clipfrac:                       0.11352539
+      max_actor_loss:                           0.004893645
+      max_advantage:                            0.068838954
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0001324694
+      max_entropy:                              0.17786673
+      max_grad_norm_pre_clip:                   5.1622715
+      max_loss:                                 0.0047900835
+    std:
+      std_actor_approxkl1:                      0.0012074575
+      std_actor_approxkl3:                      0.00093862007
+      std_actor_clipfrac:                       0.024321126
+      std_actor_loss:                           0.001475785
+      std_advantage:                            0.015368699
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.096435e-06
+      std_entropy:                              0.007423124
+      std_grad_norm_pre_clip:                   1.1073934
+      std_loss:                                 0.0014758249
+  eval-batch-train:
+    avg_avg_episode_length:                     7.390025
+    avg_avg_return:                             0.9822489
+    avg_benchmark_regret:                       0.010550266
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20398712
+    proxy_corner:
+      avg_avg_return:                           0.0079587065
+      avg_reward_per_step:                      0.0016021729
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.563896
+    avg_avg_return:                             0.97507966
+    avg_benchmark_regret:                       0.015160131
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1412201
+    proxy_corner:
+      avg_avg_return:                           0.97507966
+      avg_reward_per_step:                      0.1412201
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.500104
+    avg_avg_return:                             0.9814204
+    avg_benchmark_regret:                       0.011417031
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19330597
+    proxy_corner:
+      avg_avg_return:                           0.005379471
+      avg_reward_per_step:                      0.0013580322
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.68103
+    avg_avg_return:                             0.010764846
+    avg_benchmark_regret:                       0.97387034
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0051574707
+    proxy_corner:
+      avg_avg_return:                           0.010764846
+      avg_reward_per_step:                      0.0051574707
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        83886080
+    env-step-replay-before:                     42860544
+    env-step-used-before:                       83886080
+    ppo-update-before:                          51200
+    env-step-generate-before:                   41025536
+    env-step-all-after:                         83918848
+    env-step-replay-after:                      42893312
+    env-step-used-after:                        83918848
+    ppo-update-after:                           51220
+    env-step-generate-after:                    41025536
+  train-replay:
+    shift_proportion:                           0.9609375
+    avg_avg_episode_length:                     16.979843
+    avg_avg_return:                             0.8704781
+    avg_reward_per_step:                        0.056396484
+    proxy_corner:
+      avg_avg_return:                           0.034576025
+      avg_reward_per_step:                      0.0022277832
+  train-all:
+    shift_proportion:                           0.9609375
+    avg_avg_episode_length:                     16.979843
+    avg_avg_return:                             0.8704781
+    avg_reward_per_step:                        0.056396484
+    proxy_corner:
+      avg_avg_return:                           0.034576025
+      avg_reward_per_step:                      0.0022277832
+  perf:
+    env_steps_per_second:                       351076.57994272135
+    ppo_updates_per_second:                     40.6392941384369
+    eval-batch-train-duration:                  0.5248755533248186
+    eval-batch-orig-duration:                   0.4731182521209121
+    eval-batch-shift-duration:                  0.45807017013430595
+    eval-batch-tree-duration:                   0.4781404873356223
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.671143
+      cheese-corne_dist_wavg:                   6.4803815
+      levels_cheese_is_in_corner_avg:           0.040283203
+      mouse-cheese_dist_finite_avg:             15.612061
+      mouse-cheese_dist_finite_wavg:            5.1616397
+      mouse-cheese_dist_solvable_avg:           15.612061
+      mouse-cheese_dist_solvable_wavg:          5.16164
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11973865
+    visit_patterns:
+      avg_first_visit_time:                     883.3152
+      avg_last_visit_time:                      1299.1648
+      num_replay_batches:                       1309
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0030999396
+    avg_actor_approxkl3:                        0.0025141702
+    avg_actor_clipfrac:                         0.077697754
+    avg_actor_loss:                             0.0017852128
+    avg_advantage:                              -0.009220976
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00014250547
+    avg_entropy:                                0.16894324
+    avg_grad_norm_pre_clip:                     3.5404758
+    avg_loss:                                   0.001687522
+    max:
+      max_actor_approxkl1:                      0.006884098
+      max_actor_approxkl3:                      0.005047801
+      max_actor_clipfrac:                       0.118774414
+      max_actor_loss:                           0.005073645
+      max_advantage:                            0.07069093
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00018880417
+      max_entropy:                              0.18306278
+      max_grad_norm_pre_clip:                   5.877016
+      max_loss:                                 0.004989979
+    std:
+      std_actor_approxkl1:                      0.0017121057
+      std_actor_approxkl3:                      0.0015099387
+      std_actor_clipfrac:                       0.031535525
+      std_actor_loss:                           0.0015693975
+      std_advantage:                            0.016240677
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.9082718e-05
+      std_entropy:                              0.007268409
+      std_grad_norm_pre_clip:                   1.3578054
+      std_loss:                                 0.00156541
+  eval-batch-train:
+    avg_avg_episode_length:                     7.453385
+    avg_avg_return:                             0.9814335
+    avg_benchmark_regret:                       0.011365621
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20220184
+    proxy_corner:
+      avg_avg_return:                           0.007626894
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.531929
+    avg_avg_return:                             0.9749774
+    avg_benchmark_regret:                       0.015262474
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14148712
+    proxy_corner:
+      avg_avg_return:                           0.9749774
+      avg_reward_per_step:                      0.14148712
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.517231
+    avg_avg_return:                             0.9803833
+    avg_benchmark_regret:                       0.012454185
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19322205
+    proxy_corner:
+      avg_avg_return:                           0.003864057
+      avg_reward_per_step:                      0.0009689331
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.69368
+    avg_avg_return:                             0.007792894
+    avg_benchmark_regret:                       0.9768423
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0029067993
+    proxy_corner:
+      avg_avg_return:                           0.007792894
+      avg_reward_per_step:                      0.0029067993
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        84934656
+    env-step-replay-before:                     43483136
+    env-step-used-before:                       84934656
+    ppo-update-before:                          51840
+    env-step-generate-before:                   41451520
+    env-step-all-after:                         84967424
+    env-step-replay-after:                      43515904
+    env-step-used-after:                        84967424
+    ppo-update-after:                           51860
+    env-step-generate-after:                    41451520
+  train-replay:
+    shift_proportion:                           0.9609375
+    avg_avg_episode_length:                     17.074402
+    avg_avg_return:                             0.86418664
+    avg_reward_per_step:                        0.055786133
+    proxy_corner:
+      avg_avg_return:                           0.034083784
+      avg_reward_per_step:                      0.002105713
+  train-all:
+    shift_proportion:                           0.9609375
+    avg_avg_episode_length:                     17.074402
+    avg_avg_return:                             0.86418664
+    avg_reward_per_step:                        0.055786133
+    proxy_corner:
+      avg_avg_return:                           0.034083784
+      avg_reward_per_step:                      0.002105713
+  perf:
+    env_steps_per_second:                       296106.09279587737
+    ppo_updates_per_second:                     44.483752735044035
+    eval-batch-train-duration:                  0.5233873352408409
+    eval-batch-orig-duration:                   0.4849451705813408
+    eval-batch-shift-duration:                  0.4824300743639469
+    eval-batch-tree-duration:                   0.4942878894507885
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.521484
+      cheese-corne_dist_wavg:                   8.26549
+      levels_cheese_is_in_corner_avg:           0.041503906
+      mouse-cheese_dist_finite_avg:             15.621094
+      mouse-cheese_dist_finite_wavg:            9.664453
+      mouse-cheese_dist_solvable_avg:           15.621094
+      mouse-cheese_dist_solvable_wavg:          9.664454
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11995606
+    visit_patterns:
+      avg_first_visit_time:                     894.7483
+      avg_last_visit_time:                      1318.2246
+      num_replay_batches:                       1328
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0040868833
+    avg_actor_approxkl3:                        0.0035234273
+    avg_actor_clipfrac:                         0.088049315
+    avg_actor_loss:                             0.0032898884
+    avg_advantage:                              -0.0017898113
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00013173159
+    avg_entropy:                                0.17099866
+    avg_grad_norm_pre_clip:                     3.7405918
+    avg_loss:                                   0.0031847556
+    max:
+      max_actor_approxkl1:                      0.010505711
+      max_actor_approxkl3:                      0.010212865
+      max_actor_clipfrac:                       0.17272949
+      max_actor_loss:                           0.012222715
+      max_advantage:                            0.07555115
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00015776292
+      max_entropy:                              0.18863048
+      max_grad_norm_pre_clip:                   6.855262
+      max_loss:                                 0.012112882
+    std:
+      std_actor_approxkl1:                      0.002251663
+      std_actor_approxkl3:                      0.0024173702
+      std_actor_clipfrac:                       0.035108536
+      std_actor_loss:                           0.003153121
+      std_advantage:                            0.016169433
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.0033236e-05
+      std_entropy:                              0.00966266
+      std_grad_norm_pre_clip:                   1.3633971
+      std_loss:                                 0.0031518864
+  eval-batch-train:
+    avg_avg_episode_length:                     7.406
+    avg_avg_return:                             0.98208976
+    avg_benchmark_regret:                       0.010709409
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20342255
+    proxy_corner:
+      avg_avg_return:                           0.008174161
+      avg_reward_per_step:                      0.0016098022
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.55935
+    avg_avg_return:                             0.9742769
+    avg_benchmark_regret:                       0.015962912
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14130402
+    proxy_corner:
+      avg_avg_return:                           0.9742769
+      avg_reward_per_step:                      0.14130402
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4727516
+    avg_avg_return:                             0.9809247
+    avg_benchmark_regret:                       0.011912709
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19326782
+    proxy_corner:
+      avg_avg_return:                           0.0057182154
+      avg_reward_per_step:                      0.0013580322
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.54889
+    avg_avg_return:                             0.014894469
+    avg_benchmark_regret:                       0.96974075
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.005493164
+    proxy_corner:
+      avg_avg_return:                           0.014894469
+      avg_reward_per_step:                      0.005493164
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        85983232
+    env-step-replay-before:                     44072960
+    env-step-used-before:                       85983232
+    ppo-update-before:                          52480
+    env-step-generate-before:                   41910272
+    env-step-all-after:                         86016000
+    env-step-replay-after:                      44072960
+    env-step-used-after:                        86016000
+    ppo-update-after:                           52500
+    env-step-generate-after:                    41943040
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.8092947
+    avg_avg_return:                             0.9441372
+    avg_reward_per_step:                        0.19012451
+    proxy_corner:
+      avg_avg_return:                           0.016831195
+      avg_reward_per_step:                      0.0026855469
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.8092947
+    avg_avg_return:                             0.9441372
+    avg_reward_per_step:                        0.19012451
+    proxy_corner:
+      avg_avg_return:                           0.016831195
+      avg_reward_per_step:                      0.0026855469
+  perf:
+    env_steps_per_second:                       322175.66567690996
+    ppo_updates_per_second:                     41.9179206860289
+    eval-batch-train-duration:                  0.5202729227021337
+    eval-batch-orig-duration:                   0.4844462275505066
+    eval-batch-shift-duration:                  0.48051390144973993
+    eval-batch-tree-duration:                   0.47632174100726843
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.400635
+      cheese-corne_dist_wavg:                   1.940001
+      levels_cheese_is_in_corner_avg:           0.041992188
+      mouse-cheese_dist_finite_avg:             15.640381
+      mouse-cheese_dist_finite_wavg:            15.964607
+      mouse-cheese_dist_solvable_avg:           15.640381
+      mouse-cheese_dist_solvable_wavg:          15.964609
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12083484
+    visit_patterns:
+      avg_first_visit_time:                     909.18945
+      avg_last_visit_time:                      1335.282
+      num_replay_batches:                       1345
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0028835281
+    avg_actor_approxkl3:                        0.0020866774
+    avg_actor_clipfrac:                         0.054589845
+    avg_actor_loss:                             0.0020367983
+    avg_advantage:                              0.0034038876
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00020120792
+    avg_entropy:                                0.11073818
+    avg_grad_norm_pre_clip:                     2.691726
+    avg_loss:                                   0.0020266639
+    max:
+      max_actor_approxkl1:                      0.0075181145
+      max_actor_approxkl3:                      0.0045634387
+      max_actor_clipfrac:                       0.10070801
+      max_actor_loss:                           0.004276227
+      max_advantage:                            0.09628755
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00023909708
+      max_entropy:                              0.12864517
+      max_grad_norm_pre_clip:                   4.7403545
+      max_loss:                                 0.004284242
+    std:
+      std_actor_approxkl1:                      0.0017718346
+      std_actor_approxkl3:                      0.0011173456
+      std_actor_clipfrac:                       0.022358188
+      std_actor_loss:                           0.0012693687
+      std_advantage:                            0.019812431
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.6203565e-05
+      std_entropy:                              0.010929043
+      std_grad_norm_pre_clip:                   0.97557366
+      std_loss:                                 0.0012671228
+  eval-batch-train:
+    avg_avg_episode_length:                     7.385007
+    avg_avg_return:                             0.9819605
+    avg_benchmark_regret:                       0.010838666
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20383453
+    proxy_corner:
+      avg_avg_return:                           0.008047585
+      avg_reward_per_step:                      0.0016021729
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.553814
+    avg_avg_return:                             0.97517776
+    avg_benchmark_regret:                       0.015062051
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14138031
+    proxy_corner:
+      avg_avg_return:                           0.97517776
+      avg_reward_per_step:                      0.14138031
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4065976
+    avg_avg_return:                             0.98185873
+    avg_benchmark_regret:                       0.010978662
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19376373
+    proxy_corner:
+      avg_avg_return:                           0.0068812855
+      avg_reward_per_step:                      0.0015563965
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.53667
+    avg_avg_return:                             0.012082199
+    avg_benchmark_regret:                       0.972553
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0056152344
+    proxy_corner:
+      avg_avg_return:                           0.012082199
+      avg_reward_per_step:                      0.0056152344
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        87031808
+    env-step-replay-before:                     44531712
+    env-step-used-before:                       87031808
+    ppo-update-before:                          53120
+    env-step-generate-before:                   42500096
+    env-step-all-after:                         87064576
+    env-step-replay-after:                      44531712
+    env-step-used-after:                        87064576
+    ppo-update-after:                           53140
+    env-step-generate-after:                    42532864
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.632962
+    avg_avg_return:                             0.9483874
+    avg_reward_per_step:                        0.19577026
+    proxy_corner:
+      avg_avg_return:                           0.0075687817
+      avg_reward_per_step:                      0.00061035156
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.632962
+    avg_avg_return:                             0.9483874
+    avg_reward_per_step:                        0.19577026
+    proxy_corner:
+      avg_avg_return:                           0.0075687817
+      avg_reward_per_step:                      0.00061035156
+  perf:
+    env_steps_per_second:                       261929.2229880901
+    ppo_updates_per_second:                     41.395423739323036
+    eval-batch-train-duration:                  0.5392445530742407
+    eval-batch-orig-duration:                   0.4750508554279804
+    eval-batch-shift-duration:                  0.45213479828089476
+    eval-batch-tree-duration:                   0.49319722782820463
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.36499
+      cheese-corne_dist_wavg:                   1.9265807
+      levels_cheese_is_in_corner_avg:           0.042236328
+      mouse-cheese_dist_finite_avg:             15.672119
+      mouse-cheese_dist_finite_wavg:            16.864697
+      mouse-cheese_dist_solvable_avg:           15.672119
+      mouse-cheese_dist_solvable_wavg:          16.864697
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12105963
+    visit_patterns:
+      avg_first_visit_time:                     921.48364
+      avg_last_visit_time:                      1349.365
+      num_replay_batches:                       1359
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0031105354
+    avg_actor_approxkl3:                        0.0025632477
+    avg_actor_clipfrac:                         0.062451173
+    avg_actor_loss:                             0.0026357796
+    avg_advantage:                              -0.0033740941
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00018036313
+    avg_entropy:                                0.111977376
+    avg_grad_norm_pre_clip:                     4.020416
+    avg_loss:                                   0.0026139836
+    max:
+      max_actor_approxkl1:                      0.009222383
+      max_actor_approxkl3:                      0.006958718
+      max_actor_clipfrac:                       0.099853516
+      max_actor_loss:                           0.007621953
+      max_advantage:                            0.087261915
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00022080861
+      max_entropy:                              0.13216878
+      max_grad_norm_pre_clip:                   6.1333485
+      max_loss:                                 0.0076258024
+    std:
+      std_actor_approxkl1:                      0.00235714
+      std_actor_approxkl3:                      0.0017477348
+      std_actor_clipfrac:                       0.023383664
+      std_actor_loss:                           0.0018355151
+      std_advantage:                            0.018759074
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.8882385e-05
+      std_entropy:                              0.007943994
+      std_grad_norm_pre_clip:                   1.5386097
+      std_loss:                                 0.0018399918
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4329624
+    avg_avg_return:                             0.98155034
+    avg_benchmark_regret:                       0.011248833
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20347595
+    proxy_corner:
+      avg_avg_return:                           0.0077733323
+      avg_reward_per_step:                      0.0015869141
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.508888
+    avg_avg_return:                             0.97552216
+    avg_benchmark_regret:                       0.014717665
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14164734
+    proxy_corner:
+      avg_avg_return:                           0.97552216
+      avg_reward_per_step:                      0.14164734
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4427004
+    avg_avg_return:                             0.9814879
+    avg_benchmark_regret:                       0.011349523
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.1933136
+    proxy_corner:
+      avg_avg_return:                           0.005484527
+      avg_reward_per_step:                      0.0013122559
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.46333
+    avg_avg_return:                             0.012667563
+    avg_benchmark_regret:                       0.97196764
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.005744934
+    proxy_corner:
+      avg_avg_return:                           0.012667563
+      avg_reward_per_step:                      0.005744934
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        88080384
+    env-step-replay-before:                     44859392
+    env-step-used-before:                       88080384
+    ppo-update-before:                          53760
+    env-step-generate-before:                   43220992
+    env-step-all-after:                         88113152
+    env-step-replay-after:                      44859392
+    env-step-used-after:                        88113152
+    ppo-update-after:                           53780
+    env-step-generate-after:                    43253760
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.38566
+    avg_avg_return:                             0.9490825
+    avg_reward_per_step:                        0.20074463
+    proxy_corner:
+      avg_avg_return:                           0.0042506647
+      avg_reward_per_step:                      0.00033569336
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.38566
+    avg_avg_return:                             0.9490825
+    avg_reward_per_step:                        0.20074463
+    proxy_corner:
+      avg_avg_return:                           0.0042506647
+      avg_reward_per_step:                      0.00033569336
+  perf:
+    env_steps_per_second:                       325647.31867645273
+    ppo_updates_per_second:                     41.099164422840936
+    eval-batch-train-duration:                  0.5256453808397055
+    eval-batch-orig-duration:                   0.47713194508105516
+    eval-batch-shift-duration:                  0.45034332666546106
+    eval-batch-tree-duration:                   0.47312641609460115
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.409424
+      cheese-corne_dist_wavg:                   9.141805
+      levels_cheese_is_in_corner_avg:           0.041015625
+      mouse-cheese_dist_finite_avg:             15.692627
+      mouse-cheese_dist_finite_wavg:            15.968207
+      mouse-cheese_dist_solvable_avg:           15.692627
+      mouse-cheese_dist_solvable_wavg:          15.968209
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12203317
+    visit_patterns:
+      avg_first_visit_time:                     935.94946
+      avg_last_visit_time:                      1359.3606
+      num_replay_batches:                       1369
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0034585109
+    avg_actor_approxkl3:                        0.002691213
+    avg_actor_clipfrac:                         0.06038208
+    avg_actor_loss:                             0.0026613194
+    avg_advantage:                              0.00428703
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00019547949
+    avg_entropy:                                0.10528558
+    avg_grad_norm_pre_clip:                     3.1600804
+    avg_loss:                                   0.0026537739
+    max:
+      max_actor_approxkl1:                      0.007021498
+      max_actor_approxkl3:                      0.0060646217
+      max_actor_clipfrac:                       0.10559082
+      max_actor_loss:                           0.008034109
+      max_advantage:                            0.07861185
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00024174263
+      max_entropy:                              0.12712224
+      max_grad_norm_pre_clip:                   5.407583
+      max_loss:                                 0.00801458
+    std:
+      std_actor_approxkl1:                      0.0018300477
+      std_actor_approxkl3:                      0.0013325136
+      std_actor_clipfrac:                       0.021878708
+      std_actor_loss:                           0.001764267
+      std_advantage:                            0.0188829
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.9444837e-05
+      std_entropy:                              0.014290249
+      std_grad_norm_pre_clip:                   1.2587409
+      std_loss:                                 0.0017610135
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4264116
+    avg_avg_return:                             0.98170286
+    avg_benchmark_regret:                       0.011096261
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20331573
+    proxy_corner:
+      avg_avg_return:                           0.007850324
+      avg_reward_per_step:                      0.0015945435
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.472191
+    avg_avg_return:                             0.9750212
+    avg_benchmark_regret:                       0.015218662
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14195251
+    proxy_corner:
+      avg_avg_return:                           0.9750212
+      avg_reward_per_step:                      0.14195251
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4268513
+    avg_avg_return:                             0.9811723
+    avg_benchmark_regret:                       0.011665117
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19384003
+    proxy_corner:
+      avg_avg_return:                           0.0045433952
+      avg_reward_per_step:                      0.0011062622
+  eval-batch-tree:
+    avg_avg_episode_length:                     83.77314
+    avg_avg_return:                             0.024394898
+    avg_benchmark_regret:                       0.9602403
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0042877197
+    proxy_corner:
+      avg_avg_return:                           0.024394898
+      avg_reward_per_step:                      0.0042877197
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        89128960
+    env-step-replay-before:                     45514752
+    env-step-used-before:                       89128960
+    ppo-update-before:                          54400
+    env-step-generate-before:                   43614208
+    env-step-all-after:                         89161728
+    env-step-replay-after:                      45547520
+    env-step-used-after:                        89161728
+    ppo-update-after:                           54420
+    env-step-generate-after:                    43614208
+  train-replay:
+    shift_proportion:                           0.9453125
+    avg_avg_episode_length:                     16.997011
+    avg_avg_return:                             0.86720324
+    avg_reward_per_step:                        0.055786133
+    proxy_corner:
+      avg_avg_return:                           0.052152604
+      avg_reward_per_step:                      0.0032958984
+  train-all:
+    shift_proportion:                           0.9453125
+    avg_avg_episode_length:                     16.997011
+    avg_avg_return:                             0.86720324
+    avg_reward_per_step:                        0.055786133
+    proxy_corner:
+      avg_avg_return:                           0.052152604
+      avg_reward_per_step:                      0.0032958984
+  perf:
+    env_steps_per_second:                       351553.14814924874
+    ppo_updates_per_second:                     41.94941180430978
+    eval-batch-train-duration:                  0.5360950473695993
+    eval-batch-orig-duration:                   0.48363489005714655
+    eval-batch-shift-duration:                  0.4705117894336581
+    eval-batch-tree-duration:                   0.46719457022845745
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.395752
+      cheese-corne_dist_wavg:                   9.138584
+      levels_cheese_is_in_corner_avg:           0.041015625
+      mouse-cheese_dist_finite_avg:             15.703613
+      mouse-cheese_dist_finite_wavg:            11.47508
+      mouse-cheese_dist_solvable_avg:           15.703613
+      mouse-cheese_dist_solvable_wavg:          11.47508
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11985939
+    visit_patterns:
+      avg_first_visit_time:                     945.8916
+      avg_last_visit_time:                      1380.304
+      num_replay_batches:                       1390
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0030652958
+    avg_actor_approxkl3:                        0.0026124518
+    avg_actor_clipfrac:                         0.07570801
+    avg_actor_loss:                             0.0022552514
+    avg_advantage:                              0.00034319572
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00012049142
+    avg_entropy:                                0.16480279
+    avg_grad_norm_pre_clip:                     3.7903626
+    avg_loss:                                   0.0021506944
+    max:
+      max_actor_approxkl1:                      0.009620855
+      max_actor_approxkl3:                      0.009677105
+      max_actor_clipfrac:                       0.18640137
+      max_actor_loss:                           0.010717403
+      max_advantage:                            0.085701525
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00013176358
+      max_entropy:                              0.18999016
+      max_grad_norm_pre_clip:                   8.3562355
+      max_loss:                                 0.010603026
+    std:
+      std_actor_approxkl1:                      0.002258893
+      std_actor_approxkl3:                      0.002375476
+      std_actor_clipfrac:                       0.046289206
+      std_actor_loss:                           0.0030717663
+      std_advantage:                            0.015522361
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.261275e-06
+      std_entropy:                              0.011597512
+      std_grad_norm_pre_clip:                   1.9298391
+      std_loss:                                 0.0030684941
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4333525
+    avg_avg_return:                             0.9813782
+    avg_benchmark_regret:                       0.0114209745
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.2035141
+    proxy_corner:
+      avg_avg_return:                           0.008049596
+      avg_reward_per_step:                      0.0016174316
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.490241
+    avg_avg_return:                             0.9748572
+    avg_benchmark_regret:                       0.015382682
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14174652
+    proxy_corner:
+      avg_avg_return:                           0.9748572
+      avg_reward_per_step:                      0.14174652
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.453542
+    avg_avg_return:                             0.9810213
+    avg_benchmark_regret:                       0.011816191
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19345093
+    proxy_corner:
+      avg_avg_return:                           0.005009355
+      avg_reward_per_step:                      0.0012588501
+  eval-batch-tree:
+    avg_avg_episode_length:                     83.90765
+    avg_avg_return:                             0.022236096
+    avg_benchmark_regret:                       0.9623991
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.004623413
+    proxy_corner:
+      avg_avg_return:                           0.022236096
+      avg_reward_per_step:                      0.004623413
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        90177536
+    env-step-replay-before:                     46202880
+    env-step-used-before:                       90177536
+    ppo-update-before:                          55040
+    env-step-generate-before:                   43974656
+    env-step-all-after:                         90210304
+    env-step-replay-after:                      46202880
+    env-step-used-after:                        90210304
+    ppo-update-after:                           55060
+    env-step-generate-after:                    44007424
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.756396
+    avg_avg_return:                             0.94519395
+    avg_reward_per_step:                        0.19619751
+    proxy_corner:
+      avg_avg_return:                           0.009208846
+      avg_reward_per_step:                      0.0008239746
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.756396
+    avg_avg_return:                             0.94519395
+    avg_reward_per_step:                        0.19619751
+    proxy_corner:
+      avg_avg_return:                           0.009208846
+      avg_reward_per_step:                      0.0008239746
+  perf:
+    env_steps_per_second:                       350643.54517476674
+    ppo_updates_per_second:                     43.8815994770475
+    eval-batch-train-duration:                  0.5458830306306481
+    eval-batch-orig-duration:                   0.4576271362602711
+    eval-batch-shift-duration:                  0.4808343565091491
+    eval-batch-tree-duration:                   0.4787349943071604
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.4313965
+      cheese-corne_dist_wavg:                   2.8457081
+      levels_cheese_is_in_corner_avg:           0.041259766
+      mouse-cheese_dist_finite_avg:             15.706787
+      mouse-cheese_dist_finite_wavg:            16.870848
+      mouse-cheese_dist_solvable_avg:           15.706787
+      mouse-cheese_dist_solvable_wavg:          16.870853
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999997
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11865617
+    visit_patterns:
+      avg_first_visit_time:                     955.6072
+      avg_last_visit_time:                      1400.2275
+      num_replay_batches:                       1410
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0019206078
+    avg_actor_approxkl3:                        0.00149553
+    avg_actor_clipfrac:                         0.0411438
+    avg_actor_loss:                             0.001115823
+    avg_advantage:                              0.005365953
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00018682565
+    avg_entropy:                                0.097937666
+    avg_grad_norm_pre_clip:                     2.5848694
+    avg_loss:                                   0.0011112982
+    max:
+      max_actor_approxkl1:                      0.006805964
+      max_actor_approxkl3:                      0.006174679
+      max_actor_clipfrac:                       0.107299805
+      max_actor_loss:                           0.0066352887
+      max_advantage:                            0.082482815
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00020909036
+      max_entropy:                              0.11284905
+      max_grad_norm_pre_clip:                   5.742322
+      max_loss:                                 0.0066227852
+    std:
+      std_actor_approxkl1:                      0.0017555248
+      std_actor_approxkl3:                      0.0014236363
+      std_actor_clipfrac:                       0.024477929
+      std_actor_loss:                           0.0017031041
+      std_advantage:                            0.01903136
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.2468029e-05
+      std_entropy:                              0.008392302
+      std_grad_norm_pre_clip:                   1.3713338
+      std_loss:                                 0.0017018123
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3709917
+    avg_avg_return:                             0.9818927
+    avg_benchmark_regret:                       0.010906508
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20401001
+    proxy_corner:
+      avg_avg_return:                           0.008219867
+      avg_reward_per_step:                      0.001625061
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.469931
+    avg_avg_return:                             0.9755062
+    avg_benchmark_regret:                       0.014733621
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14196777
+    proxy_corner:
+      avg_avg_return:                           0.9755062
+      avg_reward_per_step:                      0.14196777
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.356117
+    avg_avg_return:                             0.9816022
+    avg_benchmark_regret:                       0.011235312
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19412994
+    proxy_corner:
+      avg_avg_return:                           0.0053497544
+      avg_reward_per_step:                      0.0012893677
+  eval-batch-tree:
+    avg_avg_episode_length:                     83.85594
+    avg_avg_return:                             0.020397076
+    avg_benchmark_regret:                       0.96423817
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0049743652
+    proxy_corner:
+      avg_avg_return:                           0.020397076
+      avg_reward_per_step:                      0.0049743652
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        91226112
+    env-step-replay-before:                     46759936
+    env-step-used-before:                       91226112
+    ppo-update-before:                          55680
+    env-step-generate-before:                   44466176
+    env-step-all-after:                         91258880
+    env-step-replay-after:                      46759936
+    env-step-used-after:                        91258880
+    ppo-update-after:                           55700
+    env-step-generate-after:                    44498944
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.3857985
+    avg_avg_return:                             0.9483503
+    avg_reward_per_step:                        0.20083618
+    proxy_corner:
+      avg_avg_return:                           0.01332101
+      avg_reward_per_step:                      0.0031433105
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.3857985
+    avg_avg_return:                             0.9483503
+    avg_reward_per_step:                        0.20083618
+    proxy_corner:
+      avg_avg_return:                           0.01332101
+      avg_reward_per_step:                      0.0031433105
+  perf:
+    env_steps_per_second:                       298344.43053237366
+    ppo_updates_per_second:                     42.22323694646893
+    eval-batch-train-duration:                  0.5384674239903688
+    eval-batch-orig-duration:                   0.4807341592386365
+    eval-batch-shift-duration:                  0.47613896522670984
+    eval-batch-tree-duration:                   0.5069303773343563
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.426514
+      cheese-corne_dist_wavg:                   1.9435337
+      levels_cheese_is_in_corner_avg:           0.04248047
+      mouse-cheese_dist_finite_avg:             15.743652
+      mouse-cheese_dist_finite_wavg:            18.6713
+      mouse-cheese_dist_solvable_avg:           15.743652
+      mouse-cheese_dist_solvable_wavg:          18.671305
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11995846
+    visit_patterns:
+      avg_first_visit_time:                     964.834
+      avg_last_visit_time:                      1417.2446
+      num_replay_batches:                       1427
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0021472129
+    avg_actor_approxkl3:                        0.0016480256
+    avg_actor_clipfrac:                         0.04714966
+    avg_actor_loss:                             0.000922574
+    avg_advantage:                              0.0053790146
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00018863093
+    avg_entropy:                                0.11415501
+    avg_grad_norm_pre_clip:                     2.6176841
+    avg_loss:                                   0.0009027344
+    max:
+      max_actor_approxkl1:                      0.0060490016
+      max_actor_approxkl3:                      0.004814015
+      max_actor_clipfrac:                       0.08972168
+      max_actor_loss:                           0.0036178883
+      max_advantage:                            0.07246959
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00021919364
+      max_entropy:                              0.13044287
+      max_grad_norm_pre_clip:                   5.097963
+      max_loss:                                 0.0035876299
+    std:
+      std_actor_approxkl1:                      0.0015612339
+      std_actor_approxkl3:                      0.0009850673
+      std_actor_clipfrac:                       0.018838018
+      std_actor_loss:                           0.0010994587
+      std_advantage:                            0.019092914
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.500665e-05
+      std_entropy:                              0.008621366
+      std_grad_norm_pre_clip:                   1.0097754
+      std_loss:                                 0.001096419
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4068017
+    avg_avg_return:                             0.98149955
+    avg_benchmark_regret:                       0.011299594
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20323944
+    proxy_corner:
+      avg_avg_return:                           0.008361244
+      avg_reward_per_step:                      0.0016479492
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.534275
+    avg_avg_return:                             0.97452635
+    avg_benchmark_regret:                       0.015713524
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14146423
+    proxy_corner:
+      avg_avg_return:                           0.97452635
+      avg_reward_per_step:                      0.14146423
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.445731
+    avg_avg_return:                             0.9811293
+    avg_benchmark_regret:                       0.011708103
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19358826
+    proxy_corner:
+      avg_avg_return:                           0.0068035424
+      avg_reward_per_step:                      0.0015640259
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.4527
+    avg_avg_return:                             0.013168402
+    avg_benchmark_regret:                       0.9714668
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0044937134
+    proxy_corner:
+      avg_avg_return:                           0.013168402
+      avg_reward_per_step:                      0.0044937134
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        92274688
+    env-step-replay-before:                     47251456
+    env-step-used-before:                       92274688
+    ppo-update-before:                          56320
+    env-step-generate-before:                   45023232
+    env-step-all-after:                         92307456
+    env-step-replay-after:                      47251456
+    env-step-used-after:                        92307456
+    ppo-update-after:                           56340
+    env-step-generate-after:                    45056000
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.3378763
+    avg_avg_return:                             0.9476361
+    avg_reward_per_step:                        0.1956482
+    proxy_corner:
+      avg_avg_return:                           0.011087319
+      avg_reward_per_step:                      0.0009765625
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.3378763
+    avg_avg_return:                             0.9476361
+    avg_reward_per_step:                        0.1956482
+    proxy_corner:
+      avg_avg_return:                           0.011087319
+      avg_reward_per_step:                      0.0009765625
+  perf:
+    env_steps_per_second:                       311039.0739750399
+    ppo_updates_per_second:                     42.45980040088011
+    eval-batch-train-duration:                  0.3141829613596201
+    eval-batch-orig-duration:                   0.22394394502043724
+    eval-batch-shift-duration:                  0.24809002969413996
+    eval-batch-tree-duration:                   0.24060146138072014
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.376465
+      cheese-corne_dist_wavg:                   10.025637
+      levels_cheese_is_in_corner_avg:           0.043701172
+      mouse-cheese_dist_finite_avg:             15.789795
+      mouse-cheese_dist_finite_wavg:            15.077621
+      mouse-cheese_dist_solvable_avg:           15.789795
+      mouse-cheese_dist_solvable_wavg:          15.077625
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12136695
+    visit_patterns:
+      avg_first_visit_time:                     974.6577
+      avg_last_visit_time:                      1432.2463
+      num_replay_batches:                       1442
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0029069323
+    avg_actor_approxkl3:                        0.00242996
+    avg_actor_clipfrac:                         0.0597168
+    avg_actor_loss:                             0.0025191957
+    avg_advantage:                              -0.0019825327
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00020429585
+    avg_entropy:                                0.10365664
+    avg_grad_norm_pre_clip:                     3.457259
+    avg_loss:                                   0.0025176874
+    max:
+      max_actor_approxkl1:                      0.004615413
+      max_actor_approxkl3:                      0.0042256345
+      max_actor_clipfrac:                       0.08728027
+      max_actor_loss:                           0.004434686
+      max_advantage:                            0.06121677
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00023215682
+      max_entropy:                              0.11991446
+      max_grad_norm_pre_clip:                   5.8435817
+      max_loss:                                 0.004437408
+    std:
+      std_actor_approxkl1:                      0.0011729385
+      std_actor_approxkl3:                      0.001015374
+      std_actor_clipfrac:                       0.017931111
+      std_actor_loss:                           0.0012342226
+      std_advantage:                            0.019494195
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.6470098e-05
+      std_entropy:                              0.0065420326
+      std_grad_norm_pre_clip:                   1.14501
+      std_loss:                                 0.0012343142
+  eval-batch-train:
+    avg_avg_episode_length:                     7.447823
+    avg_avg_return:                             0.9817134
+    avg_benchmark_regret:                       0.011085759
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20293427
+    proxy_corner:
+      avg_avg_return:                           0.008766423
+      avg_reward_per_step:                      0.001663208
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.487065
+    avg_avg_return:                             0.97525656
+    avg_benchmark_regret:                       0.014983277
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14189148
+    proxy_corner:
+      avg_avg_return:                           0.97525656
+      avg_reward_per_step:                      0.14189148
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.456237
+    avg_avg_return:                             0.98152006
+    avg_benchmark_regret:                       0.01131741
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19242096
+    proxy_corner:
+      avg_avg_return:                           0.008354504
+      avg_reward_per_step:                      0.0017547607
+  eval-batch-tree:
+    avg_avg_episode_length:                     83.80168
+    avg_avg_return:                             0.024201632
+    avg_benchmark_regret:                       0.96043354
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0052337646
+    proxy_corner:
+      avg_avg_return:                           0.024201632
+      avg_reward_per_step:                      0.0052337646
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        93323264
+    env-step-replay-before:                     47742976
+    env-step-used-before:                       93323264
+    ppo-update-before:                          56960
+    env-step-generate-before:                   45580288
+    env-step-all-after:                         93356032
+    env-step-replay-after:                      47742976
+    env-step-used-after:                        93356032
+    ppo-update-after:                           56980
+    env-step-generate-after:                    45613056
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.466447
+    avg_avg_return:                             0.9475294
+    avg_reward_per_step:                        0.19552612
+    proxy_corner:
+      avg_avg_return:                           0.023319203
+      avg_reward_per_step:                      0.0045166016
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.466447
+    avg_avg_return:                             0.9475294
+    avg_reward_per_step:                        0.19552612
+    proxy_corner:
+      avg_avg_return:                           0.023319203
+      avg_reward_per_step:                      0.0045166016
+  perf:
+    env_steps_per_second:                       353076.016207808
+    ppo_updates_per_second:                     41.11277001842919
+    eval-batch-train-duration:                  0.5235841665416956
+    eval-batch-orig-duration:                   0.47907912265509367
+    eval-batch-shift-duration:                  0.45477990433573723
+    eval-batch-tree-duration:                   0.4688526876270771
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.338623
+      cheese-corne_dist_wavg:                   1.9350127
+      levels_cheese_is_in_corner_avg:           0.044677734
+      mouse-cheese_dist_finite_avg:             15.794922
+      mouse-cheese_dist_finite_wavg:            16.879602
+      mouse-cheese_dist_solvable_avg:           15.794922
+      mouse-cheese_dist_solvable_wavg:          16.879606
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12293993
+    visit_patterns:
+      avg_first_visit_time:                     988.35645
+      avg_last_visit_time:                      1447.3928
+      num_replay_batches:                       1457
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0018819649
+    avg_actor_approxkl3:                        0.0012573136
+    avg_actor_clipfrac:                         0.03652344
+    avg_actor_loss:                             0.00040441184
+    avg_advantage:                              0.0029081944
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00018094774
+    avg_entropy:                                0.088184305
+    avg_grad_norm_pre_clip:                     1.6071787
+    avg_loss:                                   0.0004067014
+    max:
+      max_actor_approxkl1:                      0.0037940736
+      max_actor_approxkl3:                      0.002575968
+      max_actor_clipfrac:                       0.07019043
+      max_actor_loss:                           0.0015179925
+      max_advantage:                            0.06746429
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00022105232
+      max_entropy:                              0.10687841
+      max_grad_norm_pre_clip:                   2.4835763
+      max_loss:                                 0.0015290803
+    std:
+      std_actor_approxkl1:                      0.0009867601
+      std_actor_approxkl3:                      0.0005682763
+      std_actor_clipfrac:                       0.01432263
+      std_actor_loss:                           0.000693131
+      std_advantage:                            0.018978592
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.7164679e-05
+      std_entropy:                              0.0078070136
+      std_grad_norm_pre_clip:                   0.47531042
+      std_loss:                                 0.00069607754
+  eval-batch-train:
+    avg_avg_episode_length:                     7.47705
+    avg_avg_return:                             0.9814855
+    avg_benchmark_regret:                       0.01131363
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20352173
+    proxy_corner:
+      avg_avg_return:                           0.00956819
+      avg_reward_per_step:                      0.0017547607
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.502632
+    avg_avg_return:                             0.97547203
+    avg_benchmark_regret:                       0.014767802
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14167023
+    proxy_corner:
+      avg_avg_return:                           0.97547203
+      avg_reward_per_step:                      0.14167023
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.467803
+    avg_avg_return:                             0.9816514
+    avg_benchmark_regret:                       0.011186077
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19277954
+    proxy_corner:
+      avg_avg_return:                           0.008614847
+      avg_reward_per_step:                      0.0018997192
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.21004
+    avg_avg_return:                             0.019599184
+    avg_benchmark_regret:                       0.96503603
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.005050659
+    proxy_corner:
+      avg_avg_return:                           0.019599184
+      avg_reward_per_step:                      0.005050659
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        94371840
+    env-step-replay-before:                     48267264
+    env-step-used-before:                       94371840
+    ppo-update-before:                          57600
+    env-step-generate-before:                   46104576
+    env-step-all-after:                         94404608
+    env-step-replay-after:                      48300032
+    env-step-used-after:                        94404608
+    ppo-update-after:                           57620
+    env-step-generate-after:                    46104576
+  train-replay:
+    shift_proportion:                           0.953125
+    avg_avg_episode_length:                     17.146599
+    avg_avg_return:                             0.8650715
+    avg_reward_per_step:                        0.055664062
+    proxy_corner:
+      avg_avg_return:                           0.058616918
+      avg_reward_per_step:                      0.0036621094
+  train-all:
+    shift_proportion:                           0.953125
+    avg_avg_episode_length:                     17.146599
+    avg_avg_return:                             0.8650715
+    avg_reward_per_step:                        0.055664062
+    proxy_corner:
+      avg_avg_return:                           0.058616918
+      avg_reward_per_step:                      0.0036621094
+  perf:
+    env_steps_per_second:                       301055.1870046159
+    ppo_updates_per_second:                     41.923772996660375
+    eval-batch-train-duration:                  0.5382219878956676
+    eval-batch-orig-duration:                   0.479644225910306
+    eval-batch-shift-duration:                  0.4515952439978719
+    eval-batch-tree-duration:                   0.4772979347035289
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.322754
+      cheese-corne_dist_wavg:                   10.049608
+      levels_cheese_is_in_corner_avg:           0.044677734
+      mouse-cheese_dist_finite_avg:             15.796387
+      mouse-cheese_dist_finite_wavg:            17.775953
+      mouse-cheese_dist_solvable_avg:           15.796387
+      mouse-cheese_dist_solvable_wavg:          17.775953
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.121467695
+    visit_patterns:
+      avg_first_visit_time:                     998.3723
+      avg_last_visit_time:                      1464.2737
+      num_replay_batches:                       1474
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0036007196
+    avg_actor_approxkl3:                        0.002635564
+    avg_actor_clipfrac:                         0.076983646
+    avg_actor_loss:                             0.0022499296
+    avg_advantage:                              -0.0045632864
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0001197309
+    avg_entropy:                                0.1571816
+    avg_grad_norm_pre_clip:                     3.8728752
+    avg_loss:                                   0.0021526134
+    max:
+      max_actor_approxkl1:                      0.00699451
+      max_actor_approxkl3:                      0.007078453
+      max_actor_clipfrac:                       0.13293457
+      max_actor_loss:                           0.007801216
+      max_advantage:                            0.06891477
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00013399661
+      max_entropy:                              0.17846648
+      max_grad_norm_pre_clip:                   6.4937625
+      max_loss:                                 0.0077128634
+    std:
+      std_actor_approxkl1:                      0.0020548087
+      std_actor_approxkl3:                      0.0016603608
+      std_actor_clipfrac:                       0.03294518
+      std_actor_loss:                           0.0019784665
+      std_advantage:                            0.0154167265
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.179219e-06
+      std_entropy:                              0.008450094
+      std_grad_norm_pre_clip:                   1.4631921
+      std_loss:                                 0.0019798642
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4347234
+    avg_avg_return:                             0.98187274
+    avg_benchmark_regret:                       0.010926432
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20245361
+    proxy_corner:
+      avg_avg_return:                           0.009490289
+      avg_reward_per_step:                      0.0017242432
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.51792
+    avg_avg_return:                             0.97487056
+    avg_benchmark_regret:                       0.015369263
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14155579
+    proxy_corner:
+      avg_avg_return:                           0.97487056
+      avg_reward_per_step:                      0.14155579
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.403043
+    avg_avg_return:                             0.98123467
+    avg_benchmark_regret:                       0.0116027985
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19345856
+    proxy_corner:
+      avg_avg_return:                           0.0077189812
+      avg_reward_per_step:                      0.001663208
+  eval-batch-tree:
+    avg_avg_episode_length:                     83.9792
+    avg_avg_return:                             0.021243375
+    avg_benchmark_regret:                       0.96339184
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.005432129
+    proxy_corner:
+      avg_avg_return:                           0.021243375
+      avg_reward_per_step:                      0.005432129
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        95420416
+    env-step-replay-before:                     48889856
+    env-step-used-before:                       95420416
+    ppo-update-before:                          58240
+    env-step-generate-before:                   46530560
+    env-step-all-after:                         95453184
+    env-step-replay-after:                      48889856
+    env-step-used-after:                        95453184
+    ppo-update-after:                           58260
+    env-step-generate-after:                    46563328
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.581947
+    avg_avg_return:                             0.943511
+    avg_reward_per_step:                        0.19607544
+    proxy_corner:
+      avg_avg_return:                           0.010671442
+      avg_reward_per_step:                      0.0009460449
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.581947
+    avg_avg_return:                             0.943511
+    avg_reward_per_step:                        0.19607544
+    proxy_corner:
+      avg_avg_return:                           0.010671442
+      avg_reward_per_step:                      0.0009460449
+  perf:
+    env_steps_per_second:                       356399.2878656766
+    ppo_updates_per_second:                     41.51277917861036
+    eval-batch-train-duration:                  0.5077170487493277
+    eval-batch-orig-duration:                   0.4218494901433587
+    eval-batch-shift-duration:                  0.4198552407324314
+    eval-batch-tree-duration:                   0.45496894512325525
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.31665
+      cheese-corne_dist_wavg:                   9.144566
+      levels_cheese_is_in_corner_avg:           0.045410156
+      mouse-cheese_dist_finite_avg:             15.804199
+      mouse-cheese_dist_finite_wavg:            15.986019
+      mouse-cheese_dist_solvable_avg:           15.804199
+      mouse-cheese_dist_solvable_wavg:          15.986021
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12273402
+    visit_patterns:
+      avg_first_visit_time:                     1007.68384
+      avg_last_visit_time:                      1482.1995
+      num_replay_batches:                       1492
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002388351
+    avg_actor_approxkl3:                        0.0019566843
+    avg_actor_clipfrac:                         0.0482605
+    avg_actor_loss:                             0.0018986276
+    avg_advantage:                              0.0052489303
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00018612055
+    avg_entropy:                                0.10263591
+    avg_grad_norm_pre_clip:                     2.9456992
+    avg_loss:                                   0.0018890518
+    max:
+      max_actor_approxkl1:                      0.006055998
+      max_actor_approxkl3:                      0.0055850334
+      max_actor_clipfrac:                       0.07470703
+      max_actor_loss:                           0.005073838
+      max_advantage:                            0.06749147
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00022053107
+      max_entropy:                              0.121510856
+      max_grad_norm_pre_clip:                   4.9430857
+      max_loss:                                 0.005044155
+    std:
+      std_actor_approxkl1:                      0.0016722452
+      std_actor_approxkl3:                      0.0012546766
+      std_actor_clipfrac:                       0.018811597
+      std_actor_loss:                           0.0013105563
+      std_advantage:                            0.018981703
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.8642666e-05
+      std_entropy:                              0.0110595655
+      std_grad_norm_pre_clip:                   1.0098523
+      std_loss:                                 0.0013061616
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3475056
+    avg_avg_return:                             0.98187655
+    avg_benchmark_regret:                       0.0109226
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.2042923
+    proxy_corner:
+      avg_avg_return:                           0.007682565
+      avg_reward_per_step:                      0.0015792847
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.494282
+    avg_avg_return:                             0.97509164
+    avg_benchmark_regret:                       0.0151481945
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14168549
+    proxy_corner:
+      avg_avg_return:                           0.97509164
+      avg_reward_per_step:                      0.14168549
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.388447
+    avg_avg_return:                             0.98143893
+    avg_benchmark_regret:                       0.011398548
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19360352
+    proxy_corner:
+      avg_avg_return:                           0.0060987575
+      avg_reward_per_step:                      0.0015106201
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.360405
+    avg_avg_return:                             0.014318574
+    avg_benchmark_regret:                       0.97031665
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.005836487
+    proxy_corner:
+      avg_avg_return:                           0.014318574
+      avg_reward_per_step:                      0.005836487
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        96468992
+    env-step-replay-before:                     49381376
+    env-step-used-before:                       96468992
+    ppo-update-before:                          58880
+    env-step-generate-before:                   47087616
+    env-step-all-after:                         96501760
+    env-step-replay-after:                      49414144
+    env-step-used-after:                        96501760
+    ppo-update-after:                           58900
+    env-step-generate-after:                    47087616
+  train-replay:
+    shift_proportion:                           0.94921875
+    avg_avg_episode_length:                     17.124506
+    avg_avg_return:                             0.8643011
+    avg_reward_per_step:                        0.055419922
+    proxy_corner:
+      avg_avg_return:                           0.044393238
+      avg_reward_per_step:                      0.002746582
+  train-all:
+    shift_proportion:                           0.94921875
+    avg_avg_episode_length:                     17.124506
+    avg_avg_return:                             0.8643011
+    avg_reward_per_step:                        0.055419922
+    proxy_corner:
+      avg_avg_return:                           0.044393238
+      avg_reward_per_step:                      0.002746582
+  perf:
+    env_steps_per_second:                       299509.3244771764
+    ppo_updates_per_second:                     42.00913587801479
+    eval-batch-train-duration:                  0.5394436828792095
+    eval-batch-orig-duration:                   0.48134484235197306
+    eval-batch-shift-duration:                  0.4525892809033394
+    eval-batch-tree-duration:                   0.4837885806336999
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.294189
+      cheese-corne_dist_wavg:                   10.034916
+      levels_cheese_is_in_corner_avg:           0.04638672
+      mouse-cheese_dist_finite_avg:             15.805176
+      mouse-cheese_dist_finite_wavg:            13.2897835
+      mouse-cheese_dist_solvable_avg:           15.805176
+      mouse-cheese_dist_solvable_wavg:          13.289785
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12208922
+    visit_patterns:
+      avg_first_visit_time:                     1019.8811
+      avg_last_visit_time:                      1498.3213
+      num_replay_batches:                       1508
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0015366568
+    avg_actor_approxkl3:                        0.0012389923
+    avg_actor_clipfrac:                         0.046307374
+    avg_actor_loss:                             0.00023659795
+    avg_advantage:                              0.0009254913
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00011438055
+    avg_entropy:                                0.14329267
+    avg_grad_norm_pre_clip:                     2.3287082
+    avg_loss:                                   0.00015049557
+    max:
+      max_actor_approxkl1:                      0.0031638655
+      max_actor_approxkl3:                      0.002395501
+      max_actor_clipfrac:                       0.072021484
+      max_actor_loss:                           0.0012945328
+      max_advantage:                            0.078118086
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00013700215
+      max_entropy:                              0.15424763
+      max_grad_norm_pre_clip:                   3.630616
+      max_loss:                                 0.0012154034
+    std:
+      std_actor_approxkl1:                      0.00091378676
+      std_actor_approxkl3:                      0.00054594927
+      std_actor_clipfrac:                       0.017874958
+      std_actor_loss:                           0.0004855616
+      std_advantage:                            0.014627739
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.0644001e-05
+      std_entropy:                              0.005609219
+      std_grad_norm_pre_clip:                   0.8462162
+      std_loss:                                 0.00048445218
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4923625
+    avg_avg_return:                             0.9816988
+    avg_benchmark_regret:                       0.011100335
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20184326
+    proxy_corner:
+      avg_avg_return:                           0.007765499
+      avg_reward_per_step:                      0.0015869141
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.493063
+    avg_avg_return:                             0.975317
+    avg_benchmark_regret:                       0.014922881
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1418457
+    proxy_corner:
+      avg_avg_return:                           0.975317
+      avg_reward_per_step:                      0.1418457
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.5238132
+    avg_avg_return:                             0.9812507
+    avg_benchmark_regret:                       0.011586757
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19108582
+    proxy_corner:
+      avg_avg_return:                           0.0040497594
+      avg_reward_per_step:                      0.0010223389
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.546524
+    avg_avg_return:                             0.011318803
+    avg_benchmark_regret:                       0.9733164
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0051727295
+    proxy_corner:
+      avg_avg_return:                           0.011318803
+      avg_reward_per_step:                      0.0051727295
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        97517568
+    env-step-replay-before:                     49840128
+    env-step-used-before:                       97517568
+    ppo-update-before:                          59520
+    env-step-generate-before:                   47677440
+    env-step-all-after:                         97550336
+    env-step-replay-after:                      49872896
+    env-step-used-after:                        97550336
+    ppo-update-after:                           59540
+    env-step-generate-after:                    47677440
+  train-replay:
+    shift_proportion:                           0.96484375
+    avg_avg_episode_length:                     17.818825
+    avg_avg_return:                             0.85749865
+    avg_reward_per_step:                        0.055267334
+    proxy_corner:
+      avg_avg_return:                           0.044172652
+      avg_reward_per_step:                      0.0028381348
+  train-all:
+    shift_proportion:                           0.96484375
+    avg_avg_episode_length:                     17.818825
+    avg_avg_return:                             0.85749865
+    avg_reward_per_step:                        0.055267334
+    proxy_corner:
+      avg_avg_return:                           0.044172652
+      avg_reward_per_step:                      0.0028381348
+  perf:
+    env_steps_per_second:                       322691.319208602
+    ppo_updates_per_second:                     42.211244061478936
+    eval-batch-train-duration:                  0.5189746962860227
+    eval-batch-orig-duration:                   0.49045342300087214
+    eval-batch-shift-duration:                  0.482781695201993
+    eval-batch-tree-duration:                   0.4927693735808134
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.203857
+      cheese-corne_dist_wavg:                   11.825682
+      levels_cheese_is_in_corner_avg:           0.046142578
+      mouse-cheese_dist_finite_avg:             15.771484
+      mouse-cheese_dist_finite_wavg:            8.777513
+      mouse-cheese_dist_solvable_avg:           15.771484
+      mouse-cheese_dist_solvable_wavg:          8.7775135
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12306318
+    visit_patterns:
+      avg_first_visit_time:                     1033.3181
+      avg_last_visit_time:                      1512.3169
+      num_replay_batches:                       1522
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0062373593
+    avg_actor_approxkl3:                        0.005601394
+    avg_actor_clipfrac:                         0.107141115
+    avg_actor_loss:                             0.0060147606
+    avg_advantage:                              -0.00073209626
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0004162516
+    avg_entropy:                                0.15778296
+    avg_grad_norm_pre_clip:                     4.8630943
+    avg_loss:                                   0.0060651037
+    max:
+      max_actor_approxkl1:                      0.01366345
+      max_actor_approxkl3:                      0.011055566
+      max_actor_clipfrac:                       0.16955566
+      max_actor_loss:                           0.021930322
+      max_advantage:                            0.06922549
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.001346459
+      max_entropy:                              0.18561569
+      max_grad_norm_pre_clip:                   11.02133
+      max_loss:                                 0.022417935
+    std:
+      std_actor_approxkl1:                      0.0034001153
+      std_actor_approxkl3:                      0.0029958347
+      std_actor_clipfrac:                       0.035030458
+      std_actor_loss:                           0.004815567
+      std_advantage:                            0.02896653
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00045564346
+      std_entropy:                              0.010082303
+      std_grad_norm_pre_clip:                   2.2073417
+      std_loss:                                 0.0049323915
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4455676
+    avg_avg_return:                             0.98134947
+    avg_benchmark_regret:                       0.011449719
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20178986
+    proxy_corner:
+      avg_avg_return:                           0.008723976
+      avg_reward_per_step:                      0.0016708374
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.468488
+    avg_avg_return:                             0.97496754
+    avg_benchmark_regret:                       0.015272306
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14199829
+    proxy_corner:
+      avg_avg_return:                           0.97496754
+      avg_reward_per_step:                      0.14199829
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.5185223
+    avg_avg_return:                             0.9814892
+    avg_benchmark_regret:                       0.011348205
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19200134
+    proxy_corner:
+      avg_avg_return:                           0.006486688
+      avg_reward_per_step:                      0.0012969971
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.42273
+    avg_avg_return:                             0.015238751
+    avg_benchmark_regret:                       0.9693964
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.005821228
+    proxy_corner:
+      avg_avg_return:                           0.015238751
+      avg_reward_per_step:                      0.005821228
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        98566144
+    env-step-replay-before:                     50397184
+    env-step-used-before:                       98566144
+    ppo-update-before:                          60160
+    env-step-generate-before:                   48168960
+    env-step-all-after:                         98598912
+    env-step-replay-after:                      50429952
+    env-step-used-after:                        98598912
+    ppo-update-after:                           60180
+    env-step-generate-after:                    48168960
+  train-replay:
+    shift_proportion:                           0.9375
+    avg_avg_episode_length:                     17.64213
+    avg_avg_return:                             0.85922927
+    avg_reward_per_step:                        0.054992676
+    proxy_corner:
+      avg_avg_return:                           0.058925018
+      avg_reward_per_step:                      0.0036621094
+  train-all:
+    shift_proportion:                           0.9375
+    avg_avg_episode_length:                     17.64213
+    avg_avg_return:                             0.85922927
+    avg_reward_per_step:                        0.054992676
+    proxy_corner:
+      avg_avg_return:                           0.058925018
+      avg_reward_per_step:                      0.0036621094
+  perf:
+    env_steps_per_second:                       313672.6680669338
+    ppo_updates_per_second:                     43.980544734825564
+    eval-batch-train-duration:                  0.5398929482325912
+    eval-batch-orig-duration:                   0.46342712827026844
+    eval-batch-shift-duration:                  0.4791666530072689
+    eval-batch-tree-duration:                   0.48474827129393816
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.207764
+      cheese-corne_dist_wavg:                   11.815935
+      levels_cheese_is_in_corner_avg:           0.047607422
+      mouse-cheese_dist_finite_avg:             15.77417
+      mouse-cheese_dist_finite_wavg:            7.8821573
+      mouse-cheese_dist_solvable_avg:           15.77417
+      mouse-cheese_dist_solvable_wavg:          7.8821583
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12248751
+    visit_patterns:
+      avg_first_visit_time:                     1043.4858
+      avg_last_visit_time:                      1529.2278
+      num_replay_batches:                       1539
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0035869174
+    avg_actor_approxkl3:                        0.0026388147
+    avg_actor_clipfrac:                         0.07625122
+    avg_actor_loss:                             0.001849701
+    avg_advantage:                              -0.002135565
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00025765164
+    avg_entropy:                                0.15397689
+    avg_grad_norm_pre_clip:                     3.0314567
+    avg_loss:                                   0.0018245502
+    max:
+      max_actor_approxkl1:                      0.009248025
+      max_actor_approxkl3:                      0.005590208
+      max_actor_clipfrac:                       0.111083984
+      max_actor_loss:                           0.0061640926
+      max_advantage:                            0.07083911
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0007313559
+      max_entropy:                              0.1660805
+      max_grad_norm_pre_clip:                   5.0759416
+      max_loss:                                 0.006356778
+    std:
+      std_actor_approxkl1:                      0.00228296
+      std_actor_approxkl3:                      0.0014209285
+      std_actor_clipfrac:                       0.026695617
+      std_actor_loss:                           0.0018540813
+      std_advantage:                            0.022669844
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0002668836
+      std_entropy:                              0.0072441027
+      std_grad_norm_pre_clip:                   1.2610378
+      std_loss:                                 0.0018312706
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3865967
+    avg_avg_return:                             0.9817188
+    avg_benchmark_regret:                       0.011080373
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20349884
+    proxy_corner:
+      avg_avg_return:                           0.009633348
+      avg_reward_per_step:                      0.0017318726
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.4848995
+    avg_avg_return:                             0.9751435
+    avg_benchmark_regret:                       0.015096331
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1418457
+    proxy_corner:
+      avg_avg_return:                           0.9751435
+      avg_reward_per_step:                      0.1418457
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4913726
+    avg_avg_return:                             0.9813317
+    avg_benchmark_regret:                       0.011505768
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19178009
+    proxy_corner:
+      avg_avg_return:                           0.007902354
+      avg_reward_per_step:                      0.0015487671
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.47948
+    avg_avg_return:                             0.012800688
+    avg_benchmark_regret:                       0.9718345
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0057525635
+    proxy_corner:
+      avg_avg_return:                           0.012800688
+      avg_reward_per_step:                      0.0057525635
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        99614720
+    env-step-replay-before:                     51052544
+    env-step-used-before:                       99614720
+    ppo-update-before:                          60800
+    env-step-generate-before:                   48562176
+    env-step-all-after:                         99647488
+    env-step-replay-after:                      51085312
+    env-step-used-after:                        99647488
+    ppo-update-after:                           60820
+    env-step-generate-after:                    48562176
+  train-replay:
+    shift_proportion:                           0.94921875
+    avg_avg_episode_length:                     17.0206
+    avg_avg_return:                             0.86586016
+    avg_reward_per_step:                        0.055633545
+    proxy_corner:
+      avg_avg_return:                           0.04508326
+      avg_reward_per_step:                      0.0027770996
+  train-all:
+    shift_proportion:                           0.94921875
+    avg_avg_episode_length:                     17.0206
+    avg_avg_return:                             0.86586016
+    avg_reward_per_step:                        0.055633545
+    proxy_corner:
+      avg_avg_return:                           0.04508326
+      avg_reward_per_step:                      0.0027770996
+  perf:
+    env_steps_per_second:                       276080.868937706
+    ppo_updates_per_second:                     42.08787597182552
+    eval-batch-train-duration:                  0.533024387434125
+    eval-batch-orig-duration:                   0.45951830223202705
+    eval-batch-shift-duration:                  0.44740965217351913
+    eval-batch-tree-duration:                   0.48892988730221987
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.237549
+      cheese-corne_dist_wavg:                   13.630929
+      levels_cheese_is_in_corner_avg:           0.047851562
+      mouse-cheese_dist_finite_avg:             15.7873535
+      mouse-cheese_dist_finite_wavg:            12.377164
+      mouse-cheese_dist_solvable_avg:           15.7873535
+      mouse-cheese_dist_solvable_wavg:          12.377166
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.120961934
+    visit_patterns:
+      avg_first_visit_time:                     1051.8586
+      avg_last_visit_time:                      1549.2854
+      num_replay_batches:                       1559
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.003612473
+    avg_actor_approxkl3:                        0.0031554725
+    avg_actor_clipfrac:                         0.08867798
+    avg_actor_loss:                             0.003800539
+    avg_advantage:                              0.0018752958
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            9.596169e-05
+    avg_entropy:                                0.15140586
+    avg_grad_norm_pre_clip:                     4.301542
+    avg_loss:                                   0.0036971138
+    max:
+      max_actor_approxkl1:                      0.005932017
+      max_actor_approxkl3:                      0.0059848772
+      max_actor_clipfrac:                       0.14404297
+      max_actor_loss:                           0.008533653
+      max_advantage:                            0.08049929
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.000116024145
+      max_entropy:                              0.16437978
+      max_grad_norm_pre_clip:                   6.0242167
+      max_loss:                                 0.008428606
+    std:
+      std_actor_approxkl1:                      0.0014276392
+      std_actor_approxkl3:                      0.0015417803
+      std_actor_clipfrac:                       0.030048208
+      std_actor_loss:                           0.0024224562
+      std_advantage:                            0.013618971
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.799437e-06
+      std_entropy:                              0.009238701
+      std_grad_norm_pre_clip:                   1.306687
+      std_loss:                                 0.0024235286
+  eval-batch-train:
+    avg_avg_episode_length:                     7.382032
+    avg_avg_return:                             0.9814843
+    avg_benchmark_regret:                       0.011314867
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20451355
+    proxy_corner:
+      avg_avg_return:                           0.008906671
+      avg_reward_per_step:                      0.0016784668
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.578961
+    avg_avg_return:                             0.9742885
+    avg_benchmark_regret:                       0.015951298
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14117432
+    proxy_corner:
+      avg_avg_return:                           0.9742885
+      avg_reward_per_step:                      0.14117432
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4710083
+    avg_avg_return:                             0.9810569
+    avg_benchmark_regret:                       0.011780561
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19226837
+    proxy_corner:
+      avg_avg_return:                           0.0068910336
+      avg_reward_per_step:                      0.0015029907
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.56753
+    avg_avg_return:                             0.01041008
+    avg_benchmark_regret:                       0.9742251
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0057754517
+    proxy_corner:
+      avg_avg_return:                           0.01041008
+      avg_reward_per_step:                      0.0057754517
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        100663296
+    env-step-replay-before:                     51675136
+    env-step-used-before:                       100663296
+    ppo-update-before:                          61440
+    env-step-generate-before:                   48988160
+    env-step-all-after:                         100696064
+    env-step-replay-after:                      51675136
+    env-step-used-after:                        100696064
+    ppo-update-after:                           61460
+    env-step-generate-after:                    49020928
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.656249
+    avg_avg_return:                             0.9460352
+    avg_reward_per_step:                        0.20056152
+    proxy_corner:
+      avg_avg_return:                           0.014408082
+      avg_reward_per_step:                      0.002746582
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.656249
+    avg_avg_return:                             0.9460352
+    avg_reward_per_step:                        0.20056152
+    proxy_corner:
+      avg_avg_return:                           0.014408082
+      avg_reward_per_step:                      0.002746582
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  perf:
+    env_steps_per_second:                       353854.9474846353
+    ppo_updates_per_second:                     42.25916471644466
+    eval-batch-train-duration:                  0.6211036397144198
+    eval-batch-orig-duration:                   0.5832123458385468
+    eval-batch-shift-duration:                  0.5867452481761575
+    eval-batch-tree-duration:                   0.6231877747923136
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.279297
+      cheese-corne_dist_wavg:                   13.635317
+      levels_cheese_is_in_corner_avg:           0.048339844
+      mouse-cheese_dist_finite_avg:             15.773193
+      mouse-cheese_dist_finite_wavg:            13.276791
+      mouse-cheese_dist_solvable_avg:           15.773193
+      mouse-cheese_dist_solvable_wavg:          13.276791
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12363884
+    visit_patterns:
+      avg_first_visit_time:                     1063.1741
+      avg_last_visit_time:                      1567.231
+      num_replay_batches:                       1577
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0022706825
+    avg_actor_approxkl3:                        0.0021198583
+    avg_actor_clipfrac:                         0.048797607
+    avg_actor_loss:                             0.0018626858
+    avg_advantage:                              0.0061854906
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00017118348
+    avg_entropy:                                0.09415466
+    avg_grad_norm_pre_clip:                     3.1825864
+    avg_loss:                                   0.0018541226
+    max:
+      max_actor_approxkl1:                      0.0062582036
+      max_actor_approxkl3:                      0.004854828
+      max_actor_clipfrac:                       0.07897949
+      max_actor_loss:                           0.005132448
+      max_advantage:                            0.07043004
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00020082072
+      max_entropy:                              0.11157936
+      max_grad_norm_pre_clip:                   4.87576
+      max_loss:                                 0.005128156
+    std:
+      std_actor_approxkl1:                      0.001413165
+      std_actor_approxkl3:                      0.0012595447
+      std_actor_clipfrac:                       0.019347465
+      std_actor_loss:                           0.0012267253
+      std_advantage:                            0.01813053
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.308979e-05
+      std_entropy:                              0.0076301107
+      std_grad_norm_pre_clip:                   0.94391805
+      std_loss:                                 0.0012290897
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4095607
+    avg_avg_return:                             0.9816608
+    avg_benchmark_regret:                       0.01113835
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20322418
+    proxy_corner:
+      avg_avg_return:                           0.009065795
+      avg_reward_per_step:                      0.0016708374
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.593734
+    avg_avg_return:                             0.97486436
+    avg_benchmark_regret:                       0.015375438
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14102936
+    proxy_corner:
+      avg_avg_return:                           0.97486436
+      avg_reward_per_step:                      0.14102936
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.5395975
+    avg_avg_return:                             0.98078257
+    avg_benchmark_regret:                       0.012054864
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19213867
+    proxy_corner:
+      avg_avg_return:                           0.006423043
+      avg_reward_per_step:                      0.0014038086
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.56854
+    avg_avg_return:                             0.011845225
+    avg_benchmark_regret:                       0.97279
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0054016113
+    proxy_corner:
+      avg_avg_return:                           0.011845225
+      avg_reward_per_step:                      0.0054016113
+    rollouts_gif:                               float32(132, 225, 225, 3)
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        101711872
+    env-step-replay-before:                     52101120
+    env-step-used-before:                       101711872
+    ppo-update-before:                          62080
+    env-step-generate-before:                   49610752
+    env-step-all-after:                         101744640
+    env-step-replay-after:                      52133888
+    env-step-used-after:                        101744640
+    ppo-update-after:                           62100
+    env-step-generate-after:                    49610752
+  train-replay:
+    shift_proportion:                           0.921875
+    avg_avg_episode_length:                     17.28204
+    avg_avg_return:                             0.8635173
+    avg_reward_per_step:                        0.05493164
+    proxy_corner:
+      avg_avg_return:                           0.07165962
+      avg_reward_per_step:                      0.0045166016
+  train-all:
+    shift_proportion:                           0.921875
+    avg_avg_episode_length:                     17.28204
+    avg_avg_return:                             0.8635173
+    avg_reward_per_step:                        0.05493164
+    proxy_corner:
+      avg_avg_return:                           0.07165962
+      avg_reward_per_step:                      0.0045166016
+  perf:
+    env_steps_per_second:                       295773.0456211396
+    ppo_updates_per_second:                     42.146134070736736
+    eval-batch-train-duration:                  0.5310475565493107
+    eval-batch-orig-duration:                   0.48838671762496233
+    eval-batch-shift-duration:                  0.4505424480885267
+    eval-batch-tree-duration:                   0.48239443730562925
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.170898
+      cheese-corne_dist_wavg:                   9.1260195
+      levels_cheese_is_in_corner_avg:           0.048339844
+      mouse-cheese_dist_finite_avg:             15.766602
+      mouse-cheese_dist_finite_wavg:            6.9865603
+      mouse-cheese_dist_solvable_avg:           15.766602
+      mouse-cheese_dist_solvable_wavg:          6.9865594
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12458694
+    visit_patterns:
+      avg_first_visit_time:                     1076.8328
+      avg_last_visit_time:                      1581.1292
+      num_replay_batches:                       1591
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0018857783
+    avg_actor_approxkl3:                        0.0012567261
+    avg_actor_clipfrac:                         0.04689331
+    avg_actor_loss:                             4.8636834e-05
+    avg_advantage:                              -0.0066840574
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.000103833525
+    avg_entropy:                                0.13599916
+    avg_grad_norm_pre_clip:                     2.3797486
+    avg_loss:                                   -3.54456e-05
+    max:
+      max_actor_approxkl1:                      0.004041954
+      max_actor_approxkl3:                      0.0033095772
+      max_actor_clipfrac:                       0.08129883
+      max_actor_loss:                           0.0018259473
+      max_advantage:                            0.077386856
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00011930037
+      max_entropy:                              0.15097743
+      max_grad_norm_pre_clip:                   4.378358
+      max_loss:                                 0.0017567792
+    std:
+      std_actor_approxkl1:                      0.0010824378
+      std_actor_approxkl3:                      0.0007380886
+      std_actor_clipfrac:                       0.020252576
+      std_actor_loss:                           0.0008114268
+      std_advantage:                            0.014395981
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          8.61443e-06
+      std_entropy:                              0.008474778
+      std_grad_norm_pre_clip:                   0.9380537
+      std_loss:                                 0.00080890284
+  eval-batch-train:
+    avg_avg_episode_length:                     7.534234
+    avg_avg_return:                             0.981008
+    avg_benchmark_regret:                       0.01179116
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20167542
+    proxy_corner:
+      avg_avg_return:                           0.008575222
+      avg_reward_per_step:                      0.0016326904
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.53611
+    avg_avg_return:                             0.97518027
+    avg_benchmark_regret:                       0.015059609
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14147186
+    proxy_corner:
+      avg_avg_return:                           0.97518027
+      avg_reward_per_step:                      0.14147186
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4692435
+    avg_avg_return:                             0.9808745
+    avg_benchmark_regret:                       0.011963006
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19322968
+    proxy_corner:
+      avg_avg_return:                           0.0056989724
+      avg_reward_per_step:                      0.0012817383
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.464874
+    avg_avg_return:                             0.013777969
+    avg_benchmark_regret:                       0.97085726
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0046081543
+    proxy_corner:
+      avg_avg_return:                           0.013777969
+      avg_reward_per_step:                      0.0046081543
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        102760448
+    env-step-replay-before:                     52625408
+    env-step-used-before:                       102760448
+    ppo-update-before:                          62720
+    env-step-generate-before:                   50135040
+    env-step-all-after:                         102793216
+    env-step-replay-after:                      52625408
+    env-step-used-after:                        102793216
+    ppo-update-after:                           62740
+    env-step-generate-after:                    50167808
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     8.323127
+    avg_avg_return:                             0.94194967
+    avg_reward_per_step:                        0.17172241
+    proxy_corner:
+      avg_avg_return:                           0.012171706
+      avg_reward_per_step:                      0.0009460449
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     8.323127
+    avg_avg_return:                             0.94194967
+    avg_reward_per_step:                        0.17172241
+    proxy_corner:
+      avg_avg_return:                           0.012171706
+      avg_reward_per_step:                      0.0009460449
+  perf:
+    env_steps_per_second:                       300004.6690958682
+    ppo_updates_per_second:                     41.61871825321558
+    eval-batch-train-duration:                  0.5181149179115891
+    eval-batch-orig-duration:                   0.4735943740233779
+    eval-batch-shift-duration:                  0.44773482345044613
+    eval-batch-tree-duration:                   0.40458426903933287
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.163574
+      cheese-corne_dist_wavg:                   11.811733
+      levels_cheese_is_in_corner_avg:           0.048339844
+      mouse-cheese_dist_finite_avg:             15.805176
+      mouse-cheese_dist_finite_wavg:            11.482757
+      mouse-cheese_dist_solvable_avg:           15.805176
+      mouse-cheese_dist_solvable_wavg:          11.482757
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12353197
+    visit_patterns:
+      avg_first_visit_time:                     1087.7795
+      avg_last_visit_time:                      1596.4255
+      num_replay_batches:                       1606
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0027937458
+    avg_actor_approxkl3:                        0.0022012293
+    avg_actor_clipfrac:                         0.045574952
+    avg_actor_loss:                             0.0016825277
+    avg_advantage:                              0.005694609
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00020136421
+    avg_entropy:                                0.09564905
+    avg_grad_norm_pre_clip:                     2.4902565
+    avg_loss:                                   0.0016875608
+    max:
+      max_actor_approxkl1:                      0.008458809
+      max_actor_approxkl3:                      0.0054104403
+      max_actor_clipfrac:                       0.091552734
+      max_actor_loss:                           0.0061482033
+      max_advantage:                            0.086600244
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00025457537
+      max_entropy:                              0.113752395
+      max_grad_norm_pre_clip:                   5.7847233
+      max_loss:                                 0.0061707455
+    std:
+      std_actor_approxkl1:                      0.0026386462
+      std_actor_approxkl3:                      0.0017855524
+      std_actor_clipfrac:                       0.025494767
+      std_actor_loss:                           0.0018952356
+      std_advantage:                            0.01860654
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.6932346e-05
+      std_entropy:                              0.009422064
+      std_grad_norm_pre_clip:                   1.2221103
+      std_loss:                                 0.0018956244
+  eval-batch-train:
+    avg_avg_episode_length:                     7.433708
+    avg_avg_return:                             0.98125446
+    avg_benchmark_regret:                       0.01154474
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20224762
+    proxy_corner:
+      avg_avg_return:                           0.009419812
+      avg_reward_per_step:                      0.0017089844
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.845851
+    avg_avg_return:                             0.9736696
+    avg_benchmark_regret:                       0.016570233
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13967133
+    proxy_corner:
+      avg_avg_return:                           0.9736696
+      avg_reward_per_step:                      0.13967133
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4678955
+    avg_avg_return:                             0.9807171
+    avg_benchmark_regret:                       0.012120307
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19233704
+    proxy_corner:
+      avg_avg_return:                           0.009222457
+      avg_reward_per_step:                      0.001953125
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.168816
+    avg_avg_return:                             0.018250067
+    avg_benchmark_regret:                       0.9663851
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0053100586
+    proxy_corner:
+      avg_avg_return:                           0.018250067
+      avg_reward_per_step:                      0.0053100586
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        103809024
+    env-step-replay-before:                     53149696
+    env-step-used-before:                       103809024
+    ppo-update-before:                          63360
+    env-step-generate-before:                   50659328
+    env-step-all-after:                         103841792
+    env-step-replay-after:                      53182464
+    env-step-used-after:                        103841792
+    ppo-update-after:                           63380
+    env-step-generate-after:                    50659328
+  train-replay:
+    shift_proportion:                           0.94140625
+    avg_avg_episode_length:                     16.929966
+    avg_avg_return:                             0.8661976
+    avg_reward_per_step:                        0.055908203
+    proxy_corner:
+      avg_avg_return:                           0.060420766
+      avg_reward_per_step:                      0.0036621094
+  train-all:
+    shift_proportion:                           0.94140625
+    avg_avg_episode_length:                     16.929966
+    avg_avg_return:                             0.8661976
+    avg_reward_per_step:                        0.055908203
+    proxy_corner:
+      avg_avg_return:                           0.060420766
+      avg_reward_per_step:                      0.0036621094
+  perf:
+    env_steps_per_second:                       327191.4941231834
+    ppo_updates_per_second:                     41.25306872821267
+    eval-batch-train-duration:                  0.5329891853034496
+    eval-batch-orig-duration:                   0.480787156149745
+    eval-batch-shift-duration:                  0.4741654107347131
+    eval-batch-tree-duration:                   0.46912720147520304
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.216309
+      cheese-corne_dist_wavg:                   11.826418
+      levels_cheese_is_in_corner_avg:           0.049072266
+      mouse-cheese_dist_finite_avg:             15.804932
+      mouse-cheese_dist_finite_wavg:            11.484578
+      mouse-cheese_dist_solvable_avg:           15.804932
+      mouse-cheese_dist_solvable_wavg:          11.484581
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12192717
+    visit_patterns:
+      avg_first_visit_time:                     1098.6455
+      avg_last_visit_time:                      1613.1855
+      num_replay_batches:                       1623
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0016927625
+    avg_actor_approxkl3:                        0.0011708601
+    avg_actor_clipfrac:                         0.04534302
+    avg_actor_loss:                             0.0005474688
+    avg_advantage:                              -0.002112999
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.000101377
+    avg_entropy:                                0.13666148
+    avg_grad_norm_pre_clip:                     2.495974
+    avg_loss:                                   0.00046149577
+    max:
+      max_actor_approxkl1:                      0.0031561481
+      max_actor_approxkl3:                      0.0017916528
+      max_actor_clipfrac:                       0.068237305
+      max_actor_loss:                           0.0014868695
+      max_advantage:                            0.05641681
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.000110433146
+      max_entropy:                              0.15076646
+      max_grad_norm_pre_clip:                   4.5616612
+      max_loss:                                 0.0014040836
+    std:
+      std_actor_approxkl1:                      0.0007292727
+      std_actor_approxkl3:                      0.0004696396
+      std_actor_clipfrac:                       0.01653994
+      std_actor_loss:                           0.00067849207
+      std_advantage:                            0.014166348
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          6.2742793e-06
+      std_entropy:                              0.008600248
+      std_grad_norm_pre_clip:                   0.81973183
+      std_loss:                                 0.00067546533
+  eval-batch-train:
+    avg_avg_episode_length:                     7.414266
+    avg_avg_return:                             0.981411
+    avg_benchmark_regret:                       0.011388197
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20358276
+    proxy_corner:
+      avg_avg_return:                           0.008746667
+      avg_reward_per_step:                      0.0016403198
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.751302
+    avg_avg_return:                             0.97468835
+    avg_benchmark_regret:                       0.015551481
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14012146
+    proxy_corner:
+      avg_avg_return:                           0.97468835
+      avg_reward_per_step:                      0.14012146
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.5105166
+    avg_avg_return:                             0.98050016
+    avg_benchmark_regret:                       0.01233729
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19116974
+    proxy_corner:
+      avg_avg_return:                           0.007629416
+      avg_reward_per_step:                      0.0015411377
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.63098
+    avg_avg_return:                             0.013130201
+    avg_benchmark_regret:                       0.97150505
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.005722046
+    proxy_corner:
+      avg_avg_return:                           0.013130201
+      avg_reward_per_step:                      0.005722046
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        104857600
+    env-step-replay-before:                     53772288
+    env-step-used-before:                       104857600
+    ppo-update-before:                          64000
+    env-step-generate-before:                   51085312
+    env-step-all-after:                         104890368
+    env-step-replay-after:                      53772288
+    env-step-used-after:                        104890368
+    ppo-update-after:                           64020
+    env-step-generate-after:                    51118080
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.467963
+    avg_avg_return:                             0.9475528
+    avg_reward_per_step:                        0.203125
+    proxy_corner:
+      avg_avg_return:                           0.003545083
+      avg_reward_per_step:                      0.00033569336
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.467963
+    avg_avg_return:                             0.9475528
+    avg_reward_per_step:                        0.203125
+    proxy_corner:
+      avg_avg_return:                           0.003545083
+      avg_reward_per_step:                      0.00033569336
+  perf:
+    env_steps_per_second:                       320550.57376007165
+    ppo_updates_per_second:                     41.91241634918168
+    eval-batch-train-duration:                  0.4975314885377884
+    eval-batch-orig-duration:                   0.41342685278505087
+    eval-batch-shift-duration:                  0.4067766945809126
+    eval-batch-tree-duration:                   0.4238560926169157
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.207031
+      cheese-corne_dist_wavg:                   11.821991
+      levels_cheese_is_in_corner_avg:           0.050048828
+      mouse-cheese_dist_finite_avg:             15.833984
+      mouse-cheese_dist_finite_wavg:            10.583747
+      mouse-cheese_dist_solvable_avg:           15.833984
+      mouse-cheese_dist_solvable_wavg:          10.583746
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12149474
+    visit_patterns:
+      avg_first_visit_time:                     1108.4126
+      avg_last_visit_time:                      1631.362
+      num_replay_batches:                       1641
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0025431595
+    avg_actor_approxkl3:                        0.0018749107
+    avg_actor_clipfrac:                         0.05522461
+    avg_actor_loss:                             0.0016811695
+    avg_advantage:                              0.0020331924
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00016359109
+    avg_entropy:                                0.10301792
+    avg_grad_norm_pre_clip:                     3.1722653
+    avg_loss:                                   0.0016599473
+    max:
+      max_actor_approxkl1:                      0.006194988
+      max_actor_approxkl3:                      0.0048154835
+      max_actor_clipfrac:                       0.10083008
+      max_actor_loss:                           0.0043855626
+      max_advantage:                            0.064909875
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00019991476
+      max_entropy:                              0.12246074
+      max_grad_norm_pre_clip:                   4.7648597
+      max_loss:                                 0.004365952
+    std:
+      std_actor_approxkl1:                      0.0015732266
+      std_actor_approxkl3:                      0.0010827088
+      std_actor_clipfrac:                       0.019065702
+      std_actor_loss:                           0.0012751129
+      std_advantage:                            0.018061014
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.4713926e-05
+      std_entropy:                              0.009454993
+      std_grad_norm_pre_clip:                   0.9366902
+      std_loss:                                 0.0012765675
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4297895
+    avg_avg_return:                             0.9817928
+    avg_benchmark_regret:                       0.011006353
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20348358
+    proxy_corner:
+      avg_avg_return:                           0.009713712
+      avg_reward_per_step:                      0.0017166138
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.787842
+    avg_avg_return:                             0.97506905
+    avg_benchmark_regret:                       0.015170798
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1399231
+    proxy_corner:
+      avg_avg_return:                           0.97506905
+      avg_reward_per_step:                      0.1399231
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4734454
+    avg_avg_return:                             0.9806697
+    avg_benchmark_regret:                       0.012167788
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19152832
+    proxy_corner:
+      avg_avg_return:                           0.009366647
+      avg_reward_per_step:                      0.0017089844
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.4039
+    avg_avg_return:                             0.0148069225
+    avg_benchmark_regret:                       0.96982825
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0056991577
+    proxy_corner:
+      avg_avg_return:                           0.0148069225
+      avg_reward_per_step:                      0.0056991577
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        105906176
+    env-step-replay-before:                     54198272
+    env-step-used-before:                       105906176
+    ppo-update-before:                          64640
+    env-step-generate-before:                   51707904
+    env-step-all-after:                         105938944
+    env-step-replay-after:                      54198272
+    env-step-used-after:                        105938944
+    ppo-update-after:                           64660
+    env-step-generate-after:                    51740672
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.4783626
+    avg_avg_return:                             0.9493735
+    avg_reward_per_step:                        0.19818115
+    proxy_corner:
+      avg_avg_return:                           0.014027665
+      avg_reward_per_step:                      0.0013122559
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.4783626
+    avg_avg_return:                             0.9493735
+    avg_reward_per_step:                        0.19818115
+    proxy_corner:
+      avg_avg_return:                           0.014027665
+      avg_reward_per_step:                      0.0013122559
+  perf:
+    env_steps_per_second:                       312248.10953253915
+    ppo_updates_per_second:                     42.666592661668076
+    eval-batch-train-duration:                  0.4973468380048871
+    eval-batch-orig-duration:                   0.4370224103331566
+    eval-batch-shift-duration:                  0.4200691692531109
+    eval-batch-tree-duration:                   0.4478120245039463
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.234131
+      cheese-corne_dist_wavg:                   1.9277701
+      levels_cheese_is_in_corner_avg:           0.05102539
+      mouse-cheese_dist_finite_avg:             15.847656
+      mouse-cheese_dist_finite_wavg:            16.885895
+      mouse-cheese_dist_solvable_avg:           15.847656
+      mouse-cheese_dist_solvable_wavg:          16.885897
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.122040085
+    visit_patterns:
+      avg_first_visit_time:                     1118.7092
+      avg_last_visit_time:                      1644.324
+      num_replay_batches:                       1654
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0021141511
+    avg_actor_approxkl3:                        0.0018806396
+    avg_actor_clipfrac:                         0.043127444
+    avg_actor_loss:                             0.0016245795
+    avg_advantage:                              0.0010472896
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00015629575
+    avg_entropy:                                0.08760687
+    avg_grad_norm_pre_clip:                     2.4098701
+    avg_loss:                                   0.0016151204
+    max:
+      max_actor_approxkl1:                      0.0048706573
+      max_actor_approxkl3:                      0.0034416413
+      max_actor_clipfrac:                       0.06530762
+      max_actor_loss:                           0.006611713
+      max_advantage:                            0.093669415
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0001744004
+      max_entropy:                              0.10929648
+      max_grad_norm_pre_clip:                   4.406833
+      max_loss:                                 0.0066043944
+    std:
+      std_actor_approxkl1:                      0.0011096349
+      std_actor_approxkl3:                      0.00090203213
+      std_actor_clipfrac:                       0.014615854
+      std_actor_loss:                           0.0016470421
+      std_advantage:                            0.017643083
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.2112616e-05
+      std_entropy:                              0.009520623
+      std_grad_norm_pre_clip:                   0.8938374
+      std_loss:                                 0.0016480404
+  eval-batch-train:
+    avg_avg_episode_length:                     7.555211
+    avg_avg_return:                             0.9811522
+    avg_benchmark_regret:                       0.01164693
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20301056
+    proxy_corner:
+      avg_avg_return:                           0.008872997
+      avg_reward_per_step:                      0.0016479492
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.644419
+    avg_avg_return:                             0.9743443
+    avg_benchmark_regret:                       0.01589552
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14078522
+    proxy_corner:
+      avg_avg_return:                           0.9743443
+      avg_reward_per_step:                      0.14078522
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.533821
+    avg_avg_return:                             0.9805965
+    avg_benchmark_regret:                       0.012240928
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19205475
+    proxy_corner:
+      avg_avg_return:                           0.006356351
+      avg_reward_per_step:                      0.0013580322
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.63157
+    avg_avg_return:                             0.011298917
+    avg_benchmark_regret:                       0.9733363
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0052261353
+    proxy_corner:
+      avg_avg_return:                           0.011298917
+      avg_reward_per_step:                      0.0052261353
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        106954752
+    env-step-replay-before:                     54624256
+    env-step-used-before:                       106954752
+    ppo-update-before:                          65280
+    env-step-generate-before:                   52330496
+    env-step-all-after:                         106987520
+    env-step-replay-after:                      54624256
+    env-step-used-after:                        106987520
+    ppo-update-after:                           65300
+    env-step-generate-after:                    52363264
+  train-generate:
+    shift_proportion:                           1.0
+    avg_avg_episode_length:                     7.6002035
+    avg_avg_return:                             0.94677997
+    avg_reward_per_step:                        0.1904602
+    proxy_corner:
+      avg_avg_return:                           0.0017836129
+      avg_reward_per_step:                      0.00030517578
+  train-all:
+    shift_proportion:                           1.0
+    avg_avg_episode_length:                     7.6002035
+    avg_avg_return:                             0.94677997
+    avg_reward_per_step:                        0.1904602
+    proxy_corner:
+      avg_avg_return:                           0.0017836129
+      avg_reward_per_step:                      0.00030517578
+  perf:
+    env_steps_per_second:                       352741.7413958177
+    ppo_updates_per_second:                     42.87753040028573
+    eval-batch-train-duration:                  0.5357853770256042
+    eval-batch-orig-duration:                   0.483737176284194
+    eval-batch-shift-duration:                  0.4436009982600808
+    eval-batch-tree-duration:                   0.4956208048388362
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.199707
+      cheese-corne_dist_wavg:                   3.7270598
+      levels_cheese_is_in_corner_avg:           0.051757812
+      mouse-cheese_dist_finite_avg:             15.865723
+      mouse-cheese_dist_finite_wavg:            15.988477
+      mouse-cheese_dist_solvable_avg:           15.865723
+      mouse-cheese_dist_solvable_wavg:          15.988477
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12304889
+    visit_patterns:
+      avg_first_visit_time:                     1129.145
+      avg_last_visit_time:                      1657.1951
+      num_replay_batches:                       1667
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0024346027
+    avg_actor_approxkl3:                        0.0017905707
+    avg_actor_clipfrac:                         0.0439209
+    avg_actor_loss:                             0.0012639185
+    avg_advantage:                              0.004030513
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00015829828
+    avg_entropy:                                0.09296539
+    avg_grad_norm_pre_clip:                     2.5021994
+    avg_loss:                                   0.0012501022
+    max:
+      max_actor_approxkl1:                      0.0067473054
+      max_actor_approxkl3:                      0.0044773035
+      max_actor_clipfrac:                       0.07507324
+      max_actor_loss:                           0.0043655485
+      max_advantage:                            0.073474884
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0001779494
+      max_entropy:                              0.115753494
+      max_grad_norm_pre_clip:                   5.0131383
+      max_loss:                                 0.0043389765
+    std:
+      std_actor_approxkl1:                      0.0017084166
+      std_actor_approxkl3:                      0.0011979009
+      std_actor_clipfrac:                       0.020290405
+      std_actor_loss:                           0.0013906659
+      std_advantage:                            0.017672963
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.0235497e-05
+      std_entropy:                              0.011398716
+      std_grad_norm_pre_clip:                   1.0394542
+      std_loss:                                 0.0013868512
+  eval-batch-train:
+    avg_avg_episode_length:                     7.416996
+    avg_avg_return:                             0.9819842
+    avg_benchmark_regret:                       0.010814971
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20359802
+    proxy_corner:
+      avg_avg_return:                           0.007997092
+      avg_reward_per_step:                      0.0016021729
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.662847
+    avg_avg_return:                             0.97480214
+    avg_benchmark_regret:                       0.015437685
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14064789
+    proxy_corner:
+      avg_avg_return:                           0.97480214
+      avg_reward_per_step:                      0.14064789
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4600945
+    avg_avg_return:                             0.9812315
+    avg_benchmark_regret:                       0.011605929
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19451141
+    proxy_corner:
+      avg_avg_return:                           0.005530852
+      avg_reward_per_step:                      0.0013885498
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.595276
+    avg_avg_return:                             0.011514356
+    avg_benchmark_regret:                       0.9731208
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0056762695
+    proxy_corner:
+      avg_avg_return:                           0.011514356
+      avg_reward_per_step:                      0.0056762695
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        108003328
+    env-step-replay-before:                     55115776
+    env-step-used-before:                       108003328
+    ppo-update-before:                          65920
+    env-step-generate-before:                   52887552
+    env-step-all-after:                         108036096
+    env-step-replay-after:                      55148544
+    env-step-used-after:                        108036096
+    ppo-update-after:                           65940
+    env-step-generate-after:                    52887552
+  train-replay:
+    shift_proportion:                           0.95703125
+    avg_avg_episode_length:                     17.435736
+    avg_avg_return:                             0.8644898
+    avg_reward_per_step:                        0.054840088
+    proxy_corner:
+      avg_avg_return:                           0.049304627
+      avg_reward_per_step:                      0.0030517578
+  train-all:
+    shift_proportion:                           0.95703125
+    avg_avg_episode_length:                     17.435736
+    avg_avg_return:                             0.8644898
+    avg_reward_per_step:                        0.054840088
+    proxy_corner:
+      avg_avg_return:                           0.049304627
+      avg_reward_per_step:                      0.0030517578
+  perf:
+    env_steps_per_second:                       243058.49802857285
+    ppo_updates_per_second:                     42.25522764652364
+    eval-batch-train-duration:                  0.6712520513683558
+    eval-batch-orig-duration:                   0.6131483260542154
+    eval-batch-shift-duration:                  0.6047895094379783
+    eval-batch-tree-duration:                   0.6095614247024059
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.154297
+      cheese-corne_dist_wavg:                   10.007722
+      levels_cheese_is_in_corner_avg:           0.052734375
+      mouse-cheese_dist_finite_avg:             15.895752
+      mouse-cheese_dist_finite_wavg:            17.786697
+      mouse-cheese_dist_solvable_avg:           15.895752
+      mouse-cheese_dist_solvable_wavg:          17.786701
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12301344
+    visit_patterns:
+      avg_first_visit_time:                     1140.5039
+      avg_last_visit_time:                      1673.3408
+      num_replay_batches:                       1683
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0025263287
+    avg_actor_approxkl3:                        0.001596447
+    avg_actor_clipfrac:                         0.052484132
+    avg_actor_loss:                             0.00031788085
+    avg_advantage:                              -0.0027456225
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00011112663
+    avg_entropy:                                0.13702893
+    avg_grad_norm_pre_clip:                     2.707156
+    avg_loss:                                   0.00023641523
+    max:
+      max_actor_approxkl1:                      0.0061656293
+      max_actor_approxkl3:                      0.004957158
+      max_actor_clipfrac:                       0.10241699
+      max_actor_loss:                           0.0026198262
+      max_advantage:                            0.06212932
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00013872444
+      max_entropy:                              0.15126163
+      max_grad_norm_pre_clip:                   5.674571
+      max_loss:                                 0.0025359693
+    std:
+      std_actor_approxkl1:                      0.0016303147
+      std_actor_approxkl3:                      0.0010345541
+      std_actor_clipfrac:                       0.020893555
+      std_actor_loss:                           0.0012741898
+      std_advantage:                            0.014247477
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.295984e-05
+      std_entropy:                              0.0073600826
+      std_grad_norm_pre_clip:                   1.3432345
+      std_loss:                                 0.0012765559
+  eval-batch-train:
+    avg_avg_episode_length:                     7.489444
+    avg_avg_return:                             0.98164254
+    avg_benchmark_regret:                       0.011156633
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.2027359
+    proxy_corner:
+      avg_avg_return:                           0.00795179
+      avg_reward_per_step:                      0.0016021729
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.639949
+    avg_avg_return:                             0.9750026
+    avg_benchmark_regret:                       0.015237237
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14067078
+    proxy_corner:
+      avg_avg_return:                           0.9750026
+      avg_reward_per_step:                      0.14067078
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4255085
+    avg_avg_return:                             0.9812558
+    avg_benchmark_regret:                       0.011581611
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.1933899
+    proxy_corner:
+      avg_avg_return:                           0.006920425
+      avg_reward_per_step:                      0.0017318726
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.547455
+    avg_avg_return:                             0.011463297
+    avg_benchmark_regret:                       0.97317195
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0058288574
+    proxy_corner:
+      avg_avg_return:                           0.011463297
+      avg_reward_per_step:                      0.0058288574
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        109051904
+    env-step-replay-before:                     55541760
+    env-step-used-before:                       109051904
+    ppo-update-before:                          66560
+    env-step-generate-before:                   53510144
+    env-step-all-after:                         109084672
+    env-step-replay-after:                      55574528
+    env-step-used-after:                        109084672
+    ppo-update-after:                           66580
+    env-step-generate-after:                    53510144
+  train-replay:
+    shift_proportion:                           0.953125
+    avg_avg_episode_length:                     17.097902
+    avg_avg_return:                             0.8640847
+    avg_reward_per_step:                        0.055358887
+    proxy_corner:
+      avg_avg_return:                           0.05131928
+      avg_reward_per_step:                      0.0032043457
+  train-all:
+    shift_proportion:                           0.953125
+    avg_avg_episode_length:                     17.097902
+    avg_avg_return:                             0.8640847
+    avg_reward_per_step:                        0.055358887
+    proxy_corner:
+      avg_avg_return:                           0.05131928
+      avg_reward_per_step:                      0.0032043457
+  perf:
+    env_steps_per_second:                       317741.3805306263
+    ppo_updates_per_second:                     42.01981183762624
+    eval-batch-train-duration:                  0.5230996003374457
+    eval-batch-orig-duration:                   0.48336427193135023
+    eval-batch-shift-duration:                  0.48591513093560934
+    eval-batch-tree-duration:                   0.4514737781137228
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.188232
+      cheese-corne_dist_wavg:                   10.006256
+      levels_cheese_is_in_corner_avg:           0.052734375
+      mouse-cheese_dist_finite_avg:             15.908691
+      mouse-cheese_dist_finite_wavg:            16.88986
+      mouse-cheese_dist_solvable_avg:           15.908691
+      mouse-cheese_dist_solvable_wavg:          16.889862
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12225742
+    visit_patterns:
+      avg_first_visit_time:                     1149.8708
+      avg_last_visit_time:                      1686.2825
+      num_replay_batches:                       1696
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0028177758
+    avg_actor_approxkl3:                        0.0014385948
+    avg_actor_clipfrac:                         0.048999023
+    avg_actor_loss:                             0.00047970578
+    avg_advantage:                              -0.001541918
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00010214431
+    avg_entropy:                                0.13927798
+    avg_grad_norm_pre_clip:                     2.2235746
+    avg_loss:                                   0.0003914999
+    max:
+      max_actor_approxkl1:                      0.0056360546
+      max_actor_approxkl3:                      0.0025214702
+      max_actor_clipfrac:                       0.06726074
+      max_actor_loss:                           0.0020401496
+      max_advantage:                            0.06412244
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00011605351
+      max_entropy:                              0.152032
+      max_grad_norm_pre_clip:                   3.8717825
+      max_loss:                                 0.0019669123
+    std:
+      std_actor_approxkl1:                      0.0014003472
+      std_actor_approxkl3:                      0.00065014954
+      std_actor_clipfrac:                       0.01784004
+      std_actor_loss:                           0.00079549284
+      std_advantage:                            0.014139961
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.1303407e-06
+      std_entropy:                              0.008449719
+      std_grad_norm_pre_clip:                   0.67160124
+      std_loss:                                 0.0007950266
+  eval-batch-train:
+    avg_avg_episode_length:                     7.452798
+    avg_avg_return:                             0.9817008
+    avg_benchmark_regret:                       0.011098339
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20290375
+    proxy_corner:
+      avg_avg_return:                           0.008257749
+      avg_reward_per_step:                      0.0016174316
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.628475
+    avg_avg_return:                             0.9746323
+    avg_benchmark_regret:                       0.015607482
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14076996
+    proxy_corner:
+      avg_avg_return:                           0.9746323
+      avg_reward_per_step:                      0.14076996
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3744326
+    avg_avg_return:                             0.98135674
+    avg_benchmark_regret:                       0.011480737
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19403839
+    proxy_corner:
+      avg_avg_return:                           0.00625287
+      avg_reward_per_step:                      0.0014801025
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.67934
+    avg_avg_return:                             0.008559175
+    avg_benchmark_regret:                       0.976076
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0053634644
+    proxy_corner:
+      avg_avg_return:                           0.008559175
+      avg_reward_per_step:                      0.0053634644
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        110100480
+    env-step-replay-before:                     56066048
+    env-step-used-before:                       110100480
+    ppo-update-before:                          67200
+    env-step-generate-before:                   54034432
+    env-step-all-after:                         110133248
+    env-step-replay-after:                      56098816
+    env-step-used-after:                        110133248
+    ppo-update-after:                           67220
+    env-step-generate-after:                    54034432
+  train-replay:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     17.28136
+    avg_avg_return:                             0.8625746
+    avg_reward_per_step:                        0.054870605
+    proxy_corner:
+      avg_avg_return:                           0.027006887
+      avg_reward_per_step:                      0.0015869141
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     17.28136
+    avg_avg_return:                             0.8625746
+    avg_reward_per_step:                        0.054870605
+    proxy_corner:
+      avg_avg_return:                           0.027006887
+      avg_reward_per_step:                      0.0015869141
+  perf:
+    env_steps_per_second:                       321933.98376137286
+    ppo_updates_per_second:                     42.67719002703801
+    eval-batch-train-duration:                  0.5251517444849014
+    eval-batch-orig-duration:                   0.47836503479629755
+    eval-batch-shift-duration:                  0.47700001206249
+    eval-batch-tree-duration:                   0.4531862074509263
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.192139
+      cheese-corne_dist_wavg:                   14.515104
+      levels_cheese_is_in_corner_avg:           0.053955078
+      mouse-cheese_dist_finite_avg:             15.907227
+      mouse-cheese_dist_finite_wavg:            15.088796
+      mouse-cheese_dist_solvable_avg:           15.907227
+      mouse-cheese_dist_solvable_wavg:          15.088794
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12266447
+    visit_patterns:
+      avg_first_visit_time:                     1159.9629
+      avg_last_visit_time:                      1702.1958
+      num_replay_batches:                       1712
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0020162293
+    avg_actor_approxkl3:                        0.0009943922
+    avg_actor_clipfrac:                         0.044006348
+    avg_actor_loss:                             -0.00012549483
+    avg_advantage:                              -0.00042671087
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            9.265125e-05
+    avg_entropy:                                0.15604267
+    avg_grad_norm_pre_clip:                     2.487794
+    avg_loss:                                   -0.00023521188
+    max:
+      max_actor_approxkl1:                      0.0036878213
+      max_actor_approxkl3:                      0.001991638
+      max_actor_clipfrac:                       0.07470703
+      max_actor_loss:                           0.00156134
+      max_advantage:                            0.0559808
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.000106072956
+      max_entropy:                              0.16913855
+      max_grad_norm_pre_clip:                   3.9686537
+      max_loss:                                 0.0014519158
+    std:
+      std_actor_approxkl1:                      0.0010726952
+      std_actor_approxkl3:                      0.0004105389
+      std_actor_clipfrac:                       0.014192085
+      std_actor_loss:                           0.0009829472
+      std_advantage:                            0.01362672
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          6.041837e-06
+      std_entropy:                              0.008988732
+      std_grad_norm_pre_clip:                   0.80996114
+      std_loss:                                 0.000981638
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4181232
+    avg_avg_return:                             0.98131067
+    avg_benchmark_regret:                       0.011488493
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20330048
+    proxy_corner:
+      avg_avg_return:                           0.008117218
+      avg_reward_per_step:                      0.0016098022
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.747953
+    avg_avg_return:                             0.97450495
+    avg_benchmark_regret:                       0.015734892
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14020538
+    proxy_corner:
+      avg_avg_return:                           0.97450495
+      avg_reward_per_step:                      0.14020538
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3946867
+    avg_avg_return:                             0.9811214
+    avg_benchmark_regret:                       0.011716011
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19397736
+    proxy_corner:
+      avg_avg_return:                           0.0058205184
+      avg_reward_per_step:                      0.001373291
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.67926
+    avg_avg_return:                             0.011369377
+    avg_benchmark_regret:                       0.9732658
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0051498413
+    proxy_corner:
+      avg_avg_return:                           0.011369377
+      avg_reward_per_step:                      0.0051498413
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        111149056
+    env-step-replay-before:                     56688640
+    env-step-used-before:                       111149056
+    ppo-update-before:                          67840
+    env-step-generate-before:                   54460416
+    env-step-all-after:                         111181824
+    env-step-replay-after:                      56721408
+    env-step-used-after:                        111181824
+    ppo-update-after:                           67860
+    env-step-generate-after:                    54460416
+  train-replay:
+    shift_proportion:                           0.9609375
+    avg_avg_episode_length:                     17.042511
+    avg_avg_return:                             0.8634264
+    avg_reward_per_step:                        0.05529785
+    proxy_corner:
+      avg_avg_return:                           0.038416192
+      avg_reward_per_step:                      0.0024108887
+  train-all:
+    shift_proportion:                           0.9609375
+    avg_avg_episode_length:                     17.042511
+    avg_avg_return:                             0.8634264
+    avg_reward_per_step:                        0.05529785
+    proxy_corner:
+      avg_avg_return:                           0.038416192
+      avg_reward_per_step:                      0.0024108887
+  perf:
+    env_steps_per_second:                       301735.2249503646
+    ppo_updates_per_second:                     42.5909239852701
+    eval-batch-train-duration:                  0.5261985203251243
+    eval-batch-orig-duration:                   0.4782128492370248
+    eval-batch-shift-duration:                  0.4558940129354596
+    eval-batch-tree-duration:                   0.4680021405220032
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.234131
+      cheese-corne_dist_wavg:                   17.227697
+      levels_cheese_is_in_corner_avg:           0.053466797
+      mouse-cheese_dist_finite_avg:             15.918701
+      mouse-cheese_dist_finite_wavg:            15.09542
+      mouse-cheese_dist_solvable_avg:           15.918701
+      mouse-cheese_dist_solvable_wavg:          15.095418
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12121648
+    visit_patterns:
+      avg_first_visit_time:                     1168.3413
+      avg_last_visit_time:                      1721.2122
+      num_replay_batches:                       1731
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013051678
+    avg_actor_approxkl3:                        0.0008078191
+    avg_actor_clipfrac:                         0.03510132
+    avg_actor_loss:                             0.0004444141
+    avg_advantage:                              -0.0021224713
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            8.1849474e-05
+    avg_entropy:                                0.11842879
+    avg_grad_norm_pre_clip:                     2.3490229
+    avg_loss:                                   0.00036691
+    max:
+      max_actor_approxkl1:                      0.0027117739
+      max_actor_approxkl3:                      0.0014098156
+      max_actor_clipfrac:                       0.052734375
+      max_actor_loss:                           0.0018609408
+      max_advantage:                            0.054704428
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          9.603578e-05
+      max_entropy:                              0.13370913
+      max_grad_norm_pre_clip:                   3.7894547
+      max_loss:                                 0.0017811472
+    std:
+      std_actor_approxkl1:                      0.0006995468
+      std_actor_approxkl3:                      0.0002974191
+      std_actor_clipfrac:                       0.011211908
+      std_actor_loss:                           0.00067670696
+      std_advantage:                            0.012742494
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          6.5652475e-06
+      std_entropy:                              0.008432804
+      std_grad_norm_pre_clip:                   0.6805598
+      std_loss:                                 0.00067642354
+  eval-batch-train:
+    avg_avg_episode_length:                     7.361845
+    avg_avg_return:                             0.98200583
+    avg_benchmark_regret:                       0.010793313
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20411682
+    proxy_corner:
+      avg_avg_return:                           0.0077778366
+      avg_reward_per_step:                      0.0015869141
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.601807
+    avg_avg_return:                             0.97455037
+    avg_benchmark_regret:                       0.015689477
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14095306
+    proxy_corner:
+      avg_avg_return:                           0.97455037
+      avg_reward_per_step:                      0.14095306
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.40675
+    avg_avg_return:                             0.9807977
+    avg_benchmark_regret:                       0.012039712
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19403839
+    proxy_corner:
+      avg_avg_return:                           0.005451122
+      avg_reward_per_step:                      0.0012588501
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.595184
+    avg_avg_return:                             0.010204639
+    avg_benchmark_regret:                       0.97443056
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.005744934
+    proxy_corner:
+      avg_avg_return:                           0.010204639
+      avg_reward_per_step:                      0.005744934
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        112197632
+    env-step-replay-before:                     57376768
+    env-step-used-before:                       112197632
+    ppo-update-before:                          68480
+    env-step-generate-before:                   54820864
+    env-step-all-after:                         112230400
+    env-step-replay-after:                      57409536
+    env-step-used-after:                        112230400
+    ppo-update-after:                           68500
+    env-step-generate-after:                    54820864
+  train-replay:
+    shift_proportion:                           0.91796875
+    avg_avg_episode_length:                     17.33567
+    avg_avg_return:                             0.86187994
+    avg_reward_per_step:                        0.05456543
+    proxy_corner:
+      avg_avg_return:                           0.07563374
+      avg_reward_per_step:                      0.0046691895
+  train-all:
+    shift_proportion:                           0.91796875
+    avg_avg_episode_length:                     17.33567
+    avg_avg_return:                             0.86187994
+    avg_reward_per_step:                        0.05456543
+    proxy_corner:
+      avg_avg_return:                           0.07563374
+      avg_reward_per_step:                      0.0046691895
+  perf:
+    env_steps_per_second:                       298626.3797666132
+    ppo_updates_per_second:                     41.89663840918
+    eval-batch-train-duration:                  0.5230731684714556
+    eval-batch-orig-duration:                   0.481029256246984
+    eval-batch-shift-duration:                  0.4533807896077633
+    eval-batch-tree-duration:                   0.4680178603157401
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.243652
+      cheese-corne_dist_wavg:                   17.207405
+      levels_cheese_is_in_corner_avg:           0.05493164
+      mouse-cheese_dist_finite_avg:             15.950439
+      mouse-cheese_dist_finite_wavg:            16.896603
+      mouse-cheese_dist_solvable_avg:           15.950439
+      mouse-cheese_dist_solvable_wavg:          16.896606
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.122839436
+    visit_patterns:
+      avg_first_visit_time:                     1175.814
+      avg_last_visit_time:                      1742.2773
+      num_replay_batches:                       1752
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0025328202
+    avg_actor_approxkl3:                        0.0019131348
+    avg_actor_clipfrac:                         0.05446167
+    avg_actor_loss:                             0.0011217281
+    avg_advantage:                              -0.004641607
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            9.901323e-05
+    avg_entropy:                                0.11713844
+    avg_grad_norm_pre_clip:                     2.418103
+    avg_loss:                                   0.0010540964
+    max:
+      max_actor_approxkl1:                      0.0043490487
+      max_actor_approxkl3:                      0.0033896312
+      max_actor_clipfrac:                       0.084472656
+      max_actor_loss:                           0.0035453578
+      max_advantage:                            0.0578596
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00012797603
+      max_entropy:                              0.13305038
+      max_grad_norm_pre_clip:                   3.6565816
+      max_loss:                                 0.0034718628
+    std:
+      std_actor_approxkl1:                      0.0011659309
+      std_actor_approxkl3:                      0.00090843224
+      std_actor_clipfrac:                       0.018135937
+      std_actor_loss:                           0.001149056
+      std_advantage:                            0.012487106
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.7632954e-05
+      std_entropy:                              0.00825885
+      std_grad_norm_pre_clip:                   0.7162988
+      std_loss:                                 0.001145933
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3789997
+    avg_avg_return:                             0.9820233
+    avg_benchmark_regret:                       0.010775872
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20417023
+    proxy_corner:
+      avg_avg_return:                           0.007891809
+      avg_reward_per_step:                      0.0015945435
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.567791
+    avg_avg_return:                             0.97474813
+    avg_benchmark_regret:                       0.015491728
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1411438
+    proxy_corner:
+      avg_avg_return:                           0.97474813
+      avg_reward_per_step:                      0.1411438
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4076557
+    avg_avg_return:                             0.98144305
+    avg_benchmark_regret:                       0.011394447
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19425964
+    proxy_corner:
+      avg_avg_return:                           0.0056622014
+      avg_reward_per_step:                      0.0014190674
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.43091
+    avg_avg_return:                             0.013777121
+    avg_benchmark_regret:                       0.9708581
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0058059692
+    proxy_corner:
+      avg_avg_return:                           0.013777121
+      avg_reward_per_step:                      0.0058059692
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        113246208
+    env-step-replay-before:                     57901056
+    env-step-used-before:                       113246208
+    ppo-update-before:                          69120
+    env-step-generate-before:                   55345152
+    env-step-all-after:                         113278976
+    env-step-replay-after:                      57901056
+    env-step-used-after:                        113278976
+    ppo-update-after:                           69140
+    env-step-generate-after:                    55377920
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.836787
+    avg_avg_return:                             0.9431859
+    avg_reward_per_step:                        0.21002197
+    proxy_corner:
+      avg_avg_return:                           0.015866162
+      avg_reward_per_step:                      0.0014648438
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.836787
+    avg_avg_return:                             0.9431859
+    avg_reward_per_step:                        0.21002197
+    proxy_corner:
+      avg_avg_return:                           0.015866162
+      avg_reward_per_step:                      0.0014648438
+  perf:
+    env_steps_per_second:                       322006.13640004204
+    ppo_updates_per_second:                     41.130051985295786
+    eval-batch-train-duration:                  0.5307105705142021
+    eval-batch-orig-duration:                   0.49391817580908537
+    eval-batch-shift-duration:                  0.2785045253112912
+    eval-batch-tree-duration:                   0.23230547923594713
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.2890625
+      cheese-corne_dist_wavg:                   17.223345
+      levels_cheese_is_in_corner_avg:           0.055664062
+      mouse-cheese_dist_finite_avg:             15.958252
+      mouse-cheese_dist_finite_wavg:            16.893484
+      mouse-cheese_dist_solvable_avg:           15.958252
+      mouse-cheese_dist_solvable_wavg:          16.893486
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12279681
+    visit_patterns:
+      avg_first_visit_time:                     1186.0735
+      avg_last_visit_time:                      1757.2979
+      num_replay_batches:                       1767
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0024398253
+    avg_actor_approxkl3:                        0.0016299535
+    avg_actor_clipfrac:                         0.04332886
+    avg_actor_loss:                             0.0012399924
+    avg_advantage:                              0.00024719976
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00015494981
+    avg_entropy:                                0.084224984
+    avg_grad_norm_pre_clip:                     2.2591133
+    avg_loss:                                   0.0012332425
+    max:
+      max_actor_approxkl1:                      0.0065096533
+      max_actor_approxkl3:                      0.0034129636
+      max_actor_clipfrac:                       0.059448242
+      max_actor_loss:                           0.003155782
+      max_advantage:                            0.06816667
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.000190889
+      max_entropy:                              0.103359096
+      max_grad_norm_pre_clip:                   3.3459184
+      max_loss:                                 0.0031708034
+    std:
+      std_actor_approxkl1:                      0.0018223184
+      std_actor_approxkl3:                      0.0008406097
+      std_actor_clipfrac:                       0.014205673
+      std_actor_loss:                           0.000892391
+      std_advantage:                            0.017535938
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.7094673e-05
+      std_entropy:                              0.008092007
+      std_grad_norm_pre_clip:                   0.6263837
+      std_loss:                                 0.0008928063
+  eval-batch-train:
+    avg_avg_episode_length:                     7.377913
+    avg_avg_return:                             0.98213696
+    avg_benchmark_regret:                       0.010662182
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20401001
+    proxy_corner:
+      avg_avg_return:                           0.009040237
+      avg_reward_per_step:                      0.0016708374
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.3002405
+    avg_avg_return:                             0.97218984
+    avg_benchmark_regret:                       0.018049968
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1375122
+    proxy_corner:
+      avg_avg_return:                           0.97218984
+      avg_reward_per_step:                      0.1375122
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4232883
+    avg_avg_return:                             0.9816661
+    avg_benchmark_regret:                       0.01117134
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19350433
+    proxy_corner:
+      avg_avg_return:                           0.009323444
+      avg_reward_per_step:                      0.002204895
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.42203
+    avg_avg_return:                             0.01436625
+    avg_benchmark_regret:                       0.97026896
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0059814453
+    proxy_corner:
+      avg_avg_return:                           0.01436625
+      avg_reward_per_step:                      0.0059814453
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        114294784
+    env-step-replay-before:                     58359808
+    env-step-used-before:                       114294784
+    ppo-update-before:                          69760
+    env-step-generate-before:                   55934976
+    env-step-all-after:                         114327552
+    env-step-replay-after:                      58392576
+    env-step-used-after:                        114327552
+    ppo-update-after:                           69780
+    env-step-generate-after:                    55934976
+  train-replay:
+    shift_proportion:                           0.9375
+    avg_avg_episode_length:                     17.223118
+    avg_avg_return:                             0.86427766
+    avg_reward_per_step:                        0.054840088
+    proxy_corner:
+      avg_avg_return:                           0.056536943
+      avg_reward_per_step:                      0.0033874512
+  train-all:
+    shift_proportion:                           0.9375
+    avg_avg_episode_length:                     17.223118
+    avg_avg_return:                             0.86427766
+    avg_reward_per_step:                        0.054840088
+    proxy_corner:
+      avg_avg_return:                           0.056536943
+      avg_reward_per_step:                      0.0033874512
+  perf:
+    env_steps_per_second:                       330774.1578881813
+    ppo_updates_per_second:                     41.97697576134206
+    eval-batch-train-duration:                  0.5370671227574348
+    eval-batch-orig-duration:                   0.4728301567956805
+    eval-batch-shift-duration:                  0.487166372127831
+    eval-batch-tree-duration:                   0.4639712907373905
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.301758
+      cheese-corne_dist_wavg:                   14.522394
+      levels_cheese_is_in_corner_avg:           0.055908203
+      mouse-cheese_dist_finite_avg:             16.006104
+      mouse-cheese_dist_finite_wavg:            9.700896
+      mouse-cheese_dist_solvable_avg:           16.006104
+      mouse-cheese_dist_solvable_wavg:          9.700897
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12285252
+    visit_patterns:
+      avg_first_visit_time:                     1195.8489
+      avg_last_visit_time:                      1772.3494
+      num_replay_batches:                       1782
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002757431
+    avg_actor_approxkl3:                        0.002009901
+    avg_actor_clipfrac:                         0.059381105
+    avg_actor_loss:                             0.001774885
+    avg_advantage:                              -0.0036900034
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            8.2040475e-05
+    avg_entropy:                                0.12174537
+    avg_grad_norm_pre_clip:                     2.753842
+    avg_loss:                                   0.0016941599
+    max:
+      max_actor_approxkl1:                      0.006137784
+      max_actor_approxkl3:                      0.004156425
+      max_actor_clipfrac:                       0.0847168
+      max_actor_loss:                           0.0033731926
+      max_advantage:                            0.06795502
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          9.126293e-05
+      max_entropy:                              0.13509613
+      max_grad_norm_pre_clip:                   4.42784
+      max_loss:                                 0.0033003285
+    std:
+      std_actor_approxkl1:                      0.0012509989
+      std_actor_approxkl3:                      0.00084254914
+      std_actor_clipfrac:                       0.017089687
+      std_actor_loss:                           0.0010296219
+      std_advantage:                            0.012807574
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.5930227e-06
+      std_entropy:                              0.008134759
+      std_grad_norm_pre_clip:                   0.7595377
+      std_loss:                                 0.0010286052
+  eval-batch-train:
+    avg_avg_episode_length:                     7.445641
+    avg_avg_return:                             0.9821182
+    avg_benchmark_regret:                       0.0106809735
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20259094
+    proxy_corner:
+      avg_avg_return:                           0.009147492
+      avg_reward_per_step:                      0.0016708374
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.529112
+    avg_avg_return:                             0.97597015
+    avg_benchmark_regret:                       0.014269665
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14150238
+    proxy_corner:
+      avg_avg_return:                           0.97597015
+      avg_reward_per_step:                      0.14150238
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.417103
+    avg_avg_return:                             0.981425
+    avg_benchmark_regret:                       0.011412429
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19284058
+    proxy_corner:
+      avg_avg_return:                           0.0061188187
+      avg_reward_per_step:                      0.0014343262
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.40297
+    avg_avg_return:                             0.01322586
+    avg_benchmark_regret:                       0.9714093
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.005821228
+    proxy_corner:
+      avg_avg_return:                           0.01322586
+      avg_reward_per_step:                      0.005821228
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        115343360
+    env-step-replay-before:                     58982400
+    env-step-used-before:                       115343360
+    ppo-update-before:                          70400
+    env-step-generate-before:                   56360960
+    env-step-all-after:                         115376128
+    env-step-replay-after:                      58982400
+    env-step-used-after:                        115376128
+    ppo-update-after:                           70420
+    env-step-generate-after:                    56393728
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.965624
+    avg_avg_return:                             0.94398475
+    avg_reward_per_step:                        0.19821167
+    proxy_corner:
+      avg_avg_return:                           0.01118
+      avg_reward_per_step:                      0.0012817383
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.965624
+    avg_avg_return:                             0.94398475
+    avg_reward_per_step:                        0.19821167
+    proxy_corner:
+      avg_avg_return:                           0.01118
+      avg_reward_per_step:                      0.0012817383
+  perf:
+    env_steps_per_second:                       327433.29988647724
+    ppo_updates_per_second:                     42.670312557203616
+    eval-batch-train-duration:                  0.5261676255613565
+    eval-batch-orig-duration:                   0.47455639112740755
+    eval-batch-shift-duration:                  0.44902293663471937
+    eval-batch-tree-duration:                   0.4730723835527897
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.317383
+      cheese-corne_dist_wavg:                   9.128139
+      levels_cheese_is_in_corner_avg:           0.055664062
+      mouse-cheese_dist_finite_avg:             16.024414
+      mouse-cheese_dist_finite_wavg:            13.307286
+      mouse-cheese_dist_solvable_avg:           16.024414
+      mouse-cheese_dist_solvable_wavg:          13.307288
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12280571
+    visit_patterns:
+      avg_first_visit_time:                     1205.7434
+      avg_last_visit_time:                      1790.1836
+      num_replay_batches:                       1800
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0018398411
+    avg_actor_approxkl3:                        0.0015478913
+    avg_actor_clipfrac:                         0.041357424
+    avg_actor_loss:                             0.0013225693
+    avg_advantage:                              -0.0044740383
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00018749597
+    avg_entropy:                                0.08226004
+    avg_grad_norm_pre_clip:                     2.252155
+    avg_loss:                                   0.0013340573
+    max:
+      max_actor_approxkl1:                      0.0046412535
+      max_actor_approxkl3:                      0.0032382435
+      max_actor_clipfrac:                       0.0670166
+      max_actor_loss:                           0.003319237
+      max_advantage:                            0.069628954
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00025869982
+      max_entropy:                              0.097651854
+      max_grad_norm_pre_clip:                   6.2871914
+      max_loss:                                 0.003304776
+    std:
+      std_actor_approxkl1:                      0.0013446808
+      std_actor_approxkl3:                      0.0009145017
+      std_actor_clipfrac:                       0.015987791
+      std_actor_loss:                           0.000968371
+      std_advantage:                            0.017133454
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.7084694e-05
+      std_entropy:                              0.0084751975
+      std_grad_norm_pre_clip:                   1.1458954
+      std_loss:                                 0.00096483366
+  eval-batch-train:
+    avg_avg_episode_length:                     7.393648
+    avg_avg_return:                             0.98118913
+    avg_benchmark_regret:                       0.011610027
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20362854
+    proxy_corner:
+      avg_avg_return:                           0.008996879
+      avg_reward_per_step:                      0.0016555786
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.521645
+    avg_avg_return:                             0.9756872
+    avg_benchmark_regret:                       0.014552612
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14164734
+    proxy_corner:
+      avg_avg_return:                           0.9756872
+      avg_reward_per_step:                      0.14164734
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.371569
+    avg_avg_return:                             0.9815182
+    avg_benchmark_regret:                       0.011319214
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19419098
+    proxy_corner:
+      avg_avg_return:                           0.0059579285
+      avg_reward_per_step:                      0.0013427734
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.18823
+    avg_avg_return:                             0.017341439
+    avg_benchmark_regret:                       0.96729374
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0060043335
+    proxy_corner:
+      avg_avg_return:                           0.017341439
+      avg_reward_per_step:                      0.0060043335
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        116391936
+    env-step-replay-before:                     59506688
+    env-step-used-before:                       116391936
+    ppo-update-before:                          71040
+    env-step-generate-before:                   56885248
+    env-step-all-after:                         116424704
+    env-step-replay-after:                      59539456
+    env-step-used-after:                        116424704
+    ppo-update-after:                           71060
+    env-step-generate-after:                    56885248
+  train-replay:
+    shift_proportion:                           0.9375
+    avg_avg_episode_length:                     17.295078
+    avg_avg_return:                             0.86204964
+    avg_reward_per_step:                        0.054473877
+    proxy_corner:
+      avg_avg_return:                           0.056541897
+      avg_reward_per_step:                      0.0035705566
+  train-all:
+    shift_proportion:                           0.9375
+    avg_avg_episode_length:                     17.295078
+    avg_avg_return:                             0.86204964
+    avg_reward_per_step:                        0.054473877
+    proxy_corner:
+      avg_avg_return:                           0.056541897
+      avg_reward_per_step:                      0.0035705566
+  perf:
+    env_steps_per_second:                       262797.5962432949
+    ppo_updates_per_second:                     41.47196921666103
+    eval-batch-train-duration:                  0.5331672187894583
+    eval-batch-orig-duration:                   0.4845546679571271
+    eval-batch-shift-duration:                  0.4829788552597165
+    eval-batch-tree-duration:                   0.4546734541654587
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.298828
+      cheese-corne_dist_wavg:                   9.123511
+      levels_cheese_is_in_corner_avg:           0.056884766
+      mouse-cheese_dist_finite_avg:             16.041016
+      mouse-cheese_dist_finite_wavg:            13.302998
+      mouse-cheese_dist_solvable_avg:           16.041016
+      mouse-cheese_dist_solvable_wavg:          13.3029995
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.124333575
+    visit_patterns:
+      avg_first_visit_time:                     1216.3381
+      avg_last_visit_time:                      1807.2378
+      num_replay_batches:                       1817
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0026556938
+    avg_actor_approxkl3:                        0.0016903557
+    avg_actor_clipfrac:                         0.05531006
+    avg_actor_loss:                             0.0015068904
+    avg_advantage:                              -0.004354071
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            8.201688e-05
+    avg_entropy:                                0.118811466
+    avg_grad_norm_pre_clip:                     2.4908934
+    avg_loss:                                   0.0014290872
+    max:
+      max_actor_approxkl1:                      0.004262041
+      max_actor_approxkl3:                      0.0030165415
+      max_actor_clipfrac:                       0.0847168
+      max_actor_loss:                           0.0040923003
+      max_advantage:                            0.054747522
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00011924525
+      max_entropy:                              0.1403948
+      max_grad_norm_pre_clip:                   3.5872896
+      max_loss:                                 0.0040185004
+    std:
+      std_actor_approxkl1:                      0.0010743971
+      std_actor_approxkl3:                      0.0006746324
+      std_actor_clipfrac:                       0.016384996
+      std_actor_loss:                           0.0010576079
+      std_advantage:                            0.012188534
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.1689615e-05
+      std_entropy:                              0.009058693
+      std_grad_norm_pre_clip:                   0.72242105
+      std_loss:                                 0.0010575996
+  eval-batch-train:
+    avg_avg_episode_length:                     7.367812
+    avg_avg_return:                             0.98163444
+    avg_benchmark_regret:                       0.011164721
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20388794
+    proxy_corner:
+      avg_avg_return:                           0.0077502257
+      avg_reward_per_step:                      0.0015869141
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.072543
+    avg_avg_return:                             0.9737286
+    avg_benchmark_regret:                       0.016511198
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13866425
+    proxy_corner:
+      avg_avg_return:                           0.9737286
+      avg_reward_per_step:                      0.13866425
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3820195
+    avg_avg_return:                             0.9811597
+    avg_benchmark_regret:                       0.011677733
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19432831
+    proxy_corner:
+      avg_avg_return:                           0.0056625325
+      avg_reward_per_step:                      0.001449585
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.470184
+    avg_avg_return:                             0.011099964
+    avg_benchmark_regret:                       0.97353524
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.005859375
+    proxy_corner:
+      avg_avg_return:                           0.011099964
+      avg_reward_per_step:                      0.005859375
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        117440512
+    env-step-replay-before:                     60030976
+    env-step-used-before:                       117440512
+    ppo-update-before:                          71680
+    env-step-generate-before:                   57409536
+    env-step-all-after:                         117473280
+    env-step-replay-after:                      60030976
+    env-step-used-after:                        117473280
+    ppo-update-after:                           71700
+    env-step-generate-after:                    57442304
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     8.01402
+    avg_avg_return:                             0.942485
+    avg_reward_per_step:                        0.18215942
+    proxy_corner:
+      avg_avg_return:                           0.0042564883
+      avg_reward_per_step:                      0.0008239746
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     8.01402
+    avg_avg_return:                             0.942485
+    avg_reward_per_step:                        0.18215942
+    proxy_corner:
+      avg_avg_return:                           0.0042564883
+      avg_reward_per_step:                      0.0008239746
+  perf:
+    env_steps_per_second:                       352514.0686916254
+    ppo_updates_per_second:                     42.16417421159978
+    eval-batch-train-duration:                  0.5554121136665344
+    eval-batch-orig-duration:                   0.4564051069319248
+    eval-batch-shift-duration:                  0.48065339121967554
+    eval-batch-tree-duration:                   0.47877846751362085
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.310791
+      cheese-corne_dist_wavg:                   1.0366862
+      levels_cheese_is_in_corner_avg:           0.057617188
+      mouse-cheese_dist_finite_avg:             16.04956
+      mouse-cheese_dist_finite_wavg:            14.204868
+      mouse-cheese_dist_solvable_avg:           16.04956
+      mouse-cheese_dist_solvable_wavg:          14.204868
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12373739
+    visit_patterns:
+      avg_first_visit_time:                     1225.4719
+      avg_last_visit_time:                      1822.3257
+      num_replay_batches:                       1832
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0021081492
+    avg_actor_approxkl3:                        0.0016914079
+    avg_actor_clipfrac:                         0.04638672
+    avg_actor_loss:                             0.001379542
+    avg_advantage:                              -0.0049693133
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00016880492
+    avg_entropy:                                0.09574026
+    avg_grad_norm_pre_clip:                     2.8136299
+    avg_loss:                                   0.0013682042
+    max:
+      max_actor_approxkl1:                      0.0038117152
+      max_actor_approxkl3:                      0.004069406
+      max_actor_clipfrac:                       0.076171875
+      max_actor_loss:                           0.004766631
+      max_advantage:                            0.07202339
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00022259945
+      max_entropy:                              0.1199162
+      max_grad_norm_pre_clip:                   5.239839
+      max_loss:                                 0.0047378154
+    std:
+      std_actor_approxkl1:                      0.0012326738
+      std_actor_approxkl3:                      0.0009559252
+      std_actor_clipfrac:                       0.016683578
+      std_actor_loss:                           0.001274512
+      std_advantage:                            0.0171581
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.2372795e-05
+      std_entropy:                              0.008841282
+      std_grad_norm_pre_clip:                   1.2861905
+      std_loss:                                 0.0012717223
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3586826
+    avg_avg_return:                             0.9817872
+    avg_benchmark_regret:                       0.011011947
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20441437
+    proxy_corner:
+      avg_avg_return:                           0.007711483
+      avg_reward_per_step:                      0.0015792847
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.966726
+    avg_avg_return:                             0.9738071
+    avg_benchmark_regret:                       0.016432751
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13897705
+    proxy_corner:
+      avg_avg_return:                           0.9738071
+      avg_reward_per_step:                      0.13897705
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.460365
+    avg_avg_return:                             0.9810773
+    avg_benchmark_regret:                       0.011760088
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19371033
+    proxy_corner:
+      avg_avg_return:                           0.005353516
+      avg_reward_per_step:                      0.0013046265
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.48797
+    avg_avg_return:                             0.010742672
+    avg_benchmark_regret:                       0.97389257
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0058670044
+    proxy_corner:
+      avg_avg_return:                           0.010742672
+      avg_reward_per_step:                      0.0058670044
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        118489088
+    env-step-replay-before:                     60424192
+    env-step-used-before:                       118489088
+    ppo-update-before:                          72320
+    env-step-generate-before:                   58064896
+    env-step-all-after:                         118521856
+    env-step-replay-after:                      60424192
+    env-step-used-after:                        118521856
+    ppo-update-after:                           72340
+    env-step-generate-after:                    58097664
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.7717896
+    avg_avg_return:                             0.94603235
+    avg_reward_per_step:                        0.19073486
+    proxy_corner:
+      avg_avg_return:                           0.022208195
+      avg_reward_per_step:                      0.0022277832
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.7717896
+    avg_avg_return:                             0.94603235
+    avg_reward_per_step:                        0.19073486
+    proxy_corner:
+      avg_avg_return:                           0.022208195
+      avg_reward_per_step:                      0.0022277832
+  perf:
+    env_steps_per_second:                       315565.0726329408
+    ppo_updates_per_second:                     42.1424344593612
+    eval-batch-train-duration:                  0.532892775721848
+    eval-batch-orig-duration:                   0.4805462220683694
+    eval-batch-shift-duration:                  0.4798065051436424
+    eval-batch-tree-duration:                   0.451126080006361
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.290039
+      cheese-corne_dist_wavg:                   9.12773
+      levels_cheese_is_in_corner_avg:           0.057861328
+      mouse-cheese_dist_finite_avg:             16.03711
+      mouse-cheese_dist_finite_wavg:            16.897856
+      mouse-cheese_dist_solvable_avg:           16.03711
+      mouse-cheese_dist_solvable_wavg:          16.897856
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.124241695
+    visit_patterns:
+      avg_first_visit_time:                     1238.0564
+      avg_last_visit_time:                      1834.2205
+      num_replay_batches:                       1844
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002903463
+    avg_actor_approxkl3:                        0.002501595
+    avg_actor_clipfrac:                         0.049481202
+    avg_actor_loss:                             0.002234707
+    avg_advantage:                              0.0010339052
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00014453336
+    avg_entropy:                                0.09343994
+    avg_grad_norm_pre_clip:                     2.2655683
+    avg_loss:                                   0.0022135335
+    max:
+      max_actor_approxkl1:                      0.005824318
+      max_actor_approxkl3:                      0.0055517675
+      max_actor_clipfrac:                       0.06713867
+      max_actor_loss:                           0.0058352947
+      max_advantage:                            0.06725204
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00017259216
+      max_entropy:                              0.10934414
+      max_grad_norm_pre_clip:                   3.6226113
+      max_loss:                                 0.005818963
+    std:
+      std_actor_approxkl1:                      0.0011974933
+      std_actor_approxkl3:                      0.0013235937
+      std_actor_clipfrac:                       0.0139968535
+      std_actor_loss:                           0.0015531946
+      std_advantage:                            0.016711105
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.1153304e-05
+      std_entropy:                              0.007886298
+      std_grad_norm_pre_clip:                   0.7301584
+      std_loss:                                 0.0015570152
+  eval-batch-train:
+    avg_avg_episode_length:                     7.389201
+    avg_avg_return:                             0.9814925
+    avg_benchmark_regret:                       0.0113066435
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20437622
+    proxy_corner:
+      avg_avg_return:                           0.007692573
+      avg_reward_per_step:                      0.0015792847
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.562889
+    avg_avg_return:                             0.97494686
+    avg_benchmark_regret:                       0.015292922
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1409607
+    proxy_corner:
+      avg_avg_return:                           0.97494686
+      avg_reward_per_step:                      0.1409607
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4162126
+    avg_avg_return:                             0.98175263
+    avg_benchmark_regret:                       0.011084824
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19410706
+    proxy_corner:
+      avg_avg_return:                           0.00477444
+      avg_reward_per_step:                      0.0012359619
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.402695
+    avg_avg_return:                             0.013431802
+    avg_benchmark_regret:                       0.9712034
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0059509277
+    proxy_corner:
+      avg_avg_return:                           0.013431802
+      avg_reward_per_step:                      0.0059509277
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        119537664
+    env-step-replay-before:                     60882944
+    env-step-used-before:                       119537664
+    ppo-update-before:                          72960
+    env-step-generate-before:                   58654720
+    env-step-all-after:                         119570432
+    env-step-replay-after:                      60882944
+    env-step-used-after:                        119570432
+    ppo-update-after:                           72980
+    env-step-generate-after:                    58687488
+  train-generate:
+    shift_proportion:                           1.0
+    avg_avg_episode_length:                     7.4549303
+    avg_avg_return:                             0.9482001
+    avg_reward_per_step:                        0.19482422
+    proxy_corner:
+      avg_avg_return:                           0.009170923
+      avg_reward_per_step:                      0.0008239746
+  train-all:
+    shift_proportion:                           1.0
+    avg_avg_episode_length:                     7.4549303
+    avg_avg_return:                             0.9482001
+    avg_reward_per_step:                        0.19482422
+    proxy_corner:
+      avg_avg_return:                           0.009170923
+      avg_reward_per_step:                      0.0008239746
+  perf:
+    env_steps_per_second:                       328051.34136449546
+    ppo_updates_per_second:                     41.22844887614597
+    eval-batch-train-duration:                  0.5287599032744765
+    eval-batch-orig-duration:                   0.47693198174238205
+    eval-batch-shift-duration:                  0.45054786931723356
+    eval-batch-tree-duration:                   0.3868501828983426
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.299316
+      cheese-corne_dist_wavg:                   9.129091
+      levels_cheese_is_in_corner_avg:           0.05810547
+      mouse-cheese_dist_finite_avg:             16.041016
+      mouse-cheese_dist_finite_wavg:            13.309883
+      mouse-cheese_dist_solvable_avg:           16.041016
+      mouse-cheese_dist_solvable_wavg:          13.309885
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12451963
+    visit_patterns:
+      avg_first_visit_time:                     1247.3147
+      avg_last_visit_time:                      1848.2671
+      num_replay_batches:                       1858
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0017260928
+    avg_actor_approxkl3:                        0.0013861329
+    avg_actor_clipfrac:                         0.03865967
+    avg_actor_loss:                             0.0007950947
+    avg_advantage:                              -0.002341956
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00013840626
+    avg_entropy:                                0.089562014
+    avg_grad_norm_pre_clip:                     1.8329586
+    avg_loss:                                   0.00077473576
+    max:
+      max_actor_approxkl1:                      0.0040273434
+      max_actor_approxkl3:                      0.0031136246
+      max_actor_clipfrac:                       0.05859375
+      max_actor_loss:                           0.0018992443
+      max_advantage:                            0.07002181
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00015273341
+      max_entropy:                              0.10115914
+      max_grad_norm_pre_clip:                   4.2852354
+      max_loss:                                 0.0018781949
+    std:
+      std_actor_approxkl1:                      0.0012488539
+      std_actor_approxkl3:                      0.0008097908
+      std_actor_clipfrac:                       0.016398642
+      std_actor_loss:                           0.00067603256
+      std_advantage:                            0.016710728
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          8.961747e-06
+      std_entropy:                              0.006301613
+      std_grad_norm_pre_clip:                   0.7210435
+      std_loss:                                 0.0006752356
+  eval-batch-train:
+    avg_avg_episode_length:                     7.394853
+    avg_avg_return:                             0.9817632
+    avg_benchmark_regret:                       0.011036005
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20340729
+    proxy_corner:
+      avg_avg_return:                           0.008619369
+      avg_reward_per_step:                      0.0016479492
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.736667
+    avg_avg_return:                             0.9744599
+    avg_benchmark_regret:                       0.01577999
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14008331
+    proxy_corner:
+      avg_avg_return:                           0.9744599
+      avg_reward_per_step:                      0.14008331
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3989773
+    avg_avg_return:                             0.981273
+    avg_benchmark_regret:                       0.0115644485
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.1937027
+    proxy_corner:
+      avg_avg_return:                           0.008275188
+      avg_reward_per_step:                      0.0018692017
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.11565
+    avg_avg_return:                             0.017292354
+    avg_benchmark_regret:                       0.96734285
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0071258545
+    proxy_corner:
+      avg_avg_return:                           0.017292354
+      avg_reward_per_step:                      0.0071258545
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        120586240
+    env-step-replay-before:                     61407232
+    env-step-used-before:                       120586240
+    ppo-update-before:                          73600
+    env-step-generate-before:                   59179008
+    env-step-all-after:                         120619008
+    env-step-replay-after:                      61440000
+    env-step-used-after:                        120619008
+    ppo-update-after:                           73620
+    env-step-generate-after:                    59179008
+  train-replay:
+    shift_proportion:                           0.9375
+    avg_avg_episode_length:                     17.22625
+    avg_avg_return:                             0.8639714
+    avg_reward_per_step:                        0.054840088
+    proxy_corner:
+      avg_avg_return:                           0.059141546
+      avg_reward_per_step:                      0.0035095215
+  train-all:
+    shift_proportion:                           0.9375
+    avg_avg_episode_length:                     17.22625
+    avg_avg_return:                             0.8639714
+    avg_reward_per_step:                        0.054840088
+    proxy_corner:
+      avg_avg_return:                           0.059141546
+      avg_reward_per_step:                      0.0035095215
+  perf:
+    env_steps_per_second:                       299342.9947523305
+    ppo_updates_per_second:                     42.13407995086032
+    eval-batch-train-duration:                  0.5290894899517298
+    eval-batch-orig-duration:                   0.4754656245931983
+    eval-batch-shift-duration:                  0.433544659987092
+    eval-batch-tree-duration:                   0.43050725664943457
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.353516
+      cheese-corne_dist_wavg:                   1.0473113
+      levels_cheese_is_in_corner_avg:           0.05859375
+      mouse-cheese_dist_finite_avg:             16.066162
+      mouse-cheese_dist_finite_wavg:            15.101268
+      mouse-cheese_dist_solvable_avg:           16.066162
+      mouse-cheese_dist_solvable_wavg:          15.101268
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.123448186
+    visit_patterns:
+      avg_first_visit_time:                     1253.7576
+      avg_last_visit_time:                      1865.2532
+      num_replay_batches:                       1875
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0025624956
+    avg_actor_approxkl3:                        0.0019687435
+    avg_actor_clipfrac:                         0.06507569
+    avg_actor_loss:                             0.0019867227
+    avg_advantage:                              0.00028318528
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            9.4216426e-05
+    avg_entropy:                                0.12679085
+    avg_grad_norm_pre_clip:                     3.6891594
+    avg_loss:                                   0.00190704
+    max:
+      max_actor_approxkl1:                      0.00505888
+      max_actor_approxkl3:                      0.004904681
+      max_actor_clipfrac:                       0.11645508
+      max_actor_loss:                           0.009290051
+      max_advantage:                            0.05889511
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00011954986
+      max_entropy:                              0.13859558
+      max_grad_norm_pre_clip:                   7.68285
+      max_loss:                                 0.009214597
+    std:
+      std_actor_approxkl1:                      0.0011139275
+      std_actor_approxkl3:                      0.0010011799
+      std_actor_clipfrac:                       0.023676604
+      std_actor_loss:                           0.0022060152
+      std_advantage:                            0.012483334
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.4665084e-05
+      std_entropy:                              0.008637689
+      std_grad_norm_pre_clip:                   1.520077
+      std_loss:                                 0.0022095556
+  eval-batch-train:
+    avg_avg_episode_length:                     7.414119
+    avg_avg_return:                             0.9815924
+    avg_benchmark_regret:                       0.011206728
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.2029419
+    proxy_corner:
+      avg_avg_return:                           0.008328294
+      avg_reward_per_step:                      0.0016174316
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.068079
+    avg_avg_return:                             0.97302175
+    avg_benchmark_regret:                       0.017218092
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13867188
+    proxy_corner:
+      avg_avg_return:                           0.97302175
+      avg_reward_per_step:                      0.13867188
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.352331
+    avg_avg_return:                             0.9821451
+    avg_benchmark_regret:                       0.0106923785
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.1948471
+    proxy_corner:
+      avg_avg_return:                           0.0060526687
+      avg_reward_per_step:                      0.001449585
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.17253
+    avg_avg_return:                             0.018363731
+    avg_benchmark_regret:                       0.9662714
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.006149292
+    proxy_corner:
+      avg_avg_return:                           0.018363731
+      avg_reward_per_step:                      0.006149292
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        121634816
+    env-step-replay-before:                     61964288
+    env-step-used-before:                       121634816
+    ppo-update-before:                          74240
+    env-step-generate-before:                   59670528
+    env-step-all-after:                         121667584
+    env-step-replay-after:                      61997056
+    env-step-used-after:                        121667584
+    ppo-update-after:                           74260
+    env-step-generate-after:                    59670528
+  train-replay:
+    shift_proportion:                           0.9453125
+    avg_avg_episode_length:                     17.085297
+    avg_avg_return:                             0.8628237
+    avg_reward_per_step:                        0.05569458
+    proxy_corner:
+      avg_avg_return:                           0.064074494
+      avg_reward_per_step:                      0.0038452148
+  train-all:
+    shift_proportion:                           0.9453125
+    avg_avg_episode_length:                     17.085297
+    avg_avg_return:                             0.8628237
+    avg_reward_per_step:                        0.05569458
+    proxy_corner:
+      avg_avg_return:                           0.064074494
+      avg_reward_per_step:                      0.0038452148
+  perf:
+    env_steps_per_second:                       300621.4228591278
+    ppo_updates_per_second:                     42.69748046243853
+    eval-batch-train-duration:                  0.5140616642311215
+    eval-batch-orig-duration:                   0.4715217063203454
+    eval-batch-shift-duration:                  0.4536582576110959
+    eval-batch-tree-duration:                   0.4769227113574743
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.383301
+      cheese-corne_dist_wavg:                   10.9446535
+      levels_cheese_is_in_corner_avg:           0.060302734
+      mouse-cheese_dist_finite_avg:             16.083252
+      mouse-cheese_dist_finite_wavg:            15.108922
+      mouse-cheese_dist_solvable_avg:           16.083252
+      mouse-cheese_dist_solvable_wavg:          15.108924
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.123435125
+    visit_patterns:
+      avg_first_visit_time:                     1260.2695
+      avg_last_visit_time:                      1882.2805
+      num_replay_batches:                       1892
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013954133
+    avg_actor_approxkl3:                        0.0010754591
+    avg_actor_clipfrac:                         0.042108156
+    avg_actor_loss:                             0.00073552964
+    avg_advantage:                              -0.0021473598
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            8.384099e-05
+    avg_entropy:                                0.122037195
+    avg_grad_norm_pre_clip:                     2.4962068
+    avg_loss:                                   0.00065541285
+    max:
+      max_actor_approxkl1:                      0.003850421
+      max_actor_approxkl3:                      0.003992215
+      max_actor_clipfrac:                       0.09777832
+      max_actor_loss:                           0.0030203685
+      max_advantage:                            0.055294454
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          9.565591e-05
+      max_entropy:                              0.1426638
+      max_grad_norm_pre_clip:                   4.6708717
+      max_loss:                                 0.0029215217
+    std:
+      std_actor_approxkl1:                      0.0008541004
+      std_actor_approxkl3:                      0.00077222555
+      std_actor_clipfrac:                       0.017479874
+      std_actor_loss:                           0.0008890812
+      std_advantage:                            0.012794867
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          6.309655e-06
+      std_entropy:                              0.011320172
+      std_grad_norm_pre_clip:                   0.7919881
+      std_loss:                                 0.0008881028
+  eval-batch-train:
+    avg_avg_episode_length:                     7.319472
+    avg_avg_return:                             0.98225355
+    avg_benchmark_regret:                       0.010545615
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20456696
+    proxy_corner:
+      avg_avg_return:                           0.007840945
+      avg_reward_per_step:                      0.0015792847
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.109699
+    avg_avg_return:                             0.9725183
+    avg_benchmark_regret:                       0.017721463
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13830566
+    proxy_corner:
+      avg_avg_return:                           0.9725183
+      avg_reward_per_step:                      0.13830566
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.339227
+    avg_avg_return:                             0.9817594
+    avg_benchmark_regret:                       0.011077999
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19469452
+    proxy_corner:
+      avg_avg_return:                           0.005769893
+      avg_reward_per_step:                      0.0012588501
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.19824
+    avg_avg_return:                             0.01679164
+    avg_benchmark_regret:                       0.96784353
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.00617218
+    proxy_corner:
+      avg_avg_return:                           0.01679164
+      avg_reward_per_step:                      0.00617218
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        122683392
+    env-step-replay-before:                     62390272
+    env-step-used-before:                       122683392
+    ppo-update-before:                          74880
+    env-step-generate-before:                   60293120
+    env-step-all-after:                         122716160
+    env-step-replay-after:                      62390272
+    env-step-used-after:                        122716160
+    ppo-update-after:                           74900
+    env-step-generate-after:                    60325888
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.9097476
+    avg_avg_return:                             0.944769
+    avg_reward_per_step:                        0.19946289
+    proxy_corner:
+      avg_avg_return:                           0.019037686
+      avg_reward_per_step:                      0.0022888184
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.9097476
+    avg_avg_return:                             0.944769
+    avg_reward_per_step:                        0.19946289
+    proxy_corner:
+      avg_avg_return:                           0.019037686
+      avg_reward_per_step:                      0.0022888184
+  perf:
+    env_steps_per_second:                       295779.73164621857
+    ppo_updates_per_second:                     42.30878498518644
+    eval-batch-train-duration:                  0.5182156022638083
+    eval-batch-orig-duration:                   0.4338620910421014
+    eval-batch-shift-duration:                  0.47836274560540915
+    eval-batch-tree-duration:                   0.48691566940397024
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.358154
+      cheese-corne_dist_wavg:                   11.837468
+      levels_cheese_is_in_corner_avg:           0.061767578
+      mouse-cheese_dist_finite_avg:             16.117188
+      mouse-cheese_dist_finite_wavg:            10.615355
+      mouse-cheese_dist_solvable_avg:           16.117188
+      mouse-cheese_dist_solvable_wavg:          10.615355
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12394777
+    visit_patterns:
+      avg_first_visit_time:                     1267.5281
+      avg_last_visit_time:                      1894.199
+      num_replay_batches:                       1904
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002175333
+    avg_actor_approxkl3:                        0.0019548286
+    avg_actor_clipfrac:                         0.049102783
+    avg_actor_loss:                             0.0017397322
+    avg_advantage:                              -0.0020412202
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00015739632
+    avg_entropy:                                0.103758834
+    avg_grad_norm_pre_clip:                     2.6457622
+    avg_loss:                                   0.0017146718
+    max:
+      max_actor_approxkl1:                      0.0052088965
+      max_actor_approxkl3:                      0.0044917557
+      max_actor_clipfrac:                       0.08459473
+      max_actor_loss:                           0.0035031112
+      max_advantage:                            0.07885432
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0002649845
+      max_entropy:                              0.1179726
+      max_grad_norm_pre_clip:                   4.296362
+      max_loss:                                 0.0034562647
+    std:
+      std_actor_approxkl1:                      0.0013704575
+      std_actor_approxkl3:                      0.0010520406
+      std_actor_clipfrac:                       0.018531982
+      std_actor_loss:                           0.0009869563
+      std_advantage:                            0.016301105
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.3824243e-05
+      std_entropy:                              0.007545634
+      std_grad_norm_pre_clip:                   0.8719623
+      std_loss:                                 0.0009835996
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3644505
+    avg_avg_return:                             0.98216826
+    avg_benchmark_regret:                       0.010630908
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20433044
+    proxy_corner:
+      avg_avg_return:                           0.009237364
+      avg_reward_per_step:                      0.0016860962
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.590592
+    avg_avg_return:                             0.9749074
+    avg_benchmark_regret:                       0.015332424
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1410141
+    proxy_corner:
+      avg_avg_return:                           0.9749074
+      avg_reward_per_step:                      0.1410141
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.6567135
+    avg_avg_return:                             0.9799718
+    avg_benchmark_regret:                       0.012865593
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.1924591
+    proxy_corner:
+      avg_avg_return:                           0.007177994
+      avg_reward_per_step:                      0.0015640259
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.134605
+    avg_avg_return:                             0.01741606
+    avg_benchmark_regret:                       0.9672191
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0061950684
+    proxy_corner:
+      avg_avg_return:                           0.01741606
+      avg_reward_per_step:                      0.0061950684
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        123731968
+    env-step-replay-before:                     62717952
+    env-step-used-before:                       123731968
+    ppo-update-before:                          75520
+    env-step-generate-before:                   61014016
+    env-step-all-after:                         123764736
+    env-step-replay-after:                      62717952
+    env-step-used-after:                        123764736
+    ppo-update-after:                           75540
+    env-step-generate-after:                    61046784
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.394387
+    avg_avg_return:                             0.9498508
+    avg_reward_per_step:                        0.18878174
+    proxy_corner:
+      avg_avg_return:                           0.020482484
+      avg_reward_per_step:                      0.005340576
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.394387
+    avg_avg_return:                             0.9498508
+    avg_reward_per_step:                        0.18878174
+    proxy_corner:
+      avg_avg_return:                           0.020482484
+      avg_reward_per_step:                      0.005340576
+  perf:
+    env_steps_per_second:                       300440.1711827554
+    ppo_updates_per_second:                     42.155662623356555
+    eval-batch-train-duration:                  0.5313923675566912
+    eval-batch-orig-duration:                   0.4821786880493164
+    eval-batch-shift-duration:                  0.507002049125731
+    eval-batch-tree-duration:                   0.4960661754012108
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.347168
+      cheese-corne_dist_wavg:                   12.73831
+      levels_cheese_is_in_corner_avg:           0.060791016
+      mouse-cheese_dist_finite_avg:             16.168213
+      mouse-cheese_dist_finite_wavg:            12.411692
+      mouse-cheese_dist_solvable_avg:           16.168213
+      mouse-cheese_dist_solvable_wavg:          12.41169
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1253084
+    visit_patterns:
+      avg_first_visit_time:                     1277.6252
+      avg_last_visit_time:                      1904.2163
+      num_replay_batches:                       1914
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0018624405
+    avg_actor_approxkl3:                        0.0014207824
+    avg_actor_clipfrac:                         0.035437014
+    avg_actor_loss:                             0.0005633506
+    avg_advantage:                              0.0044877557
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0001470467
+    avg_entropy:                                0.08773905
+    avg_grad_norm_pre_clip:                     1.6560389
+    avg_loss:                                   0.000549135
+    max:
+      max_actor_approxkl1:                      0.0046140016
+      max_actor_approxkl3:                      0.0031930152
+      max_actor_clipfrac:                       0.06213379
+      max_actor_loss:                           0.002568435
+      max_advantage:                            0.071508884
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00018136454
+      max_entropy:                              0.09910023
+      max_grad_norm_pre_clip:                   2.6107738
+      max_loss:                                 0.002551092
+    std:
+      std_actor_approxkl1:                      0.0014103566
+      std_actor_approxkl3:                      0.0009025625
+      std_actor_clipfrac:                       0.016498068
+      std_actor_loss:                           0.00069967576
+      std_advantage:                            0.016775815
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.4197716e-05
+      std_entropy:                              0.0087854965
+      std_grad_norm_pre_clip:                   0.53933346
+      std_loss:                                 0.00070122106
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3836064
+    avg_avg_return:                             0.9819584
+    avg_benchmark_regret:                       0.010840753
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20433807
+    proxy_corner:
+      avg_avg_return:                           0.009854928
+      avg_reward_per_step:                      0.0017471313
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.949648
+    avg_avg_return:                             0.973211
+    avg_benchmark_regret:                       0.017028876
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13887024
+    proxy_corner:
+      avg_avg_return:                           0.973211
+      avg_reward_per_step:                      0.13887024
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.384885
+    avg_avg_return:                             0.98135394
+    avg_benchmark_regret:                       0.011483549
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19287872
+    proxy_corner:
+      avg_avg_return:                           0.011033499
+      avg_reward_per_step:                      0.0022277832
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.23352
+    avg_avg_return:                             0.014925394
+    avg_benchmark_regret:                       0.9697098
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.006248474
+    proxy_corner:
+      avg_avg_return:                           0.014925394
+      avg_reward_per_step:                      0.006248474
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        124780544
+    env-step-replay-before:                     63176704
+    env-step-used-before:                       124780544
+    ppo-update-before:                          76160
+    env-step-generate-before:                   61603840
+    env-step-all-after:                         124813312
+    env-step-replay-after:                      63176704
+    env-step-used-after:                        124813312
+    ppo-update-after:                           76180
+    env-step-generate-after:                    61636608
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.874511
+    avg_avg_return:                             0.94422704
+    avg_reward_per_step:                        0.21051025
+    proxy_corner:
+      avg_avg_return:                           0.01395881
+      avg_reward_per_step:                      0.0012512207
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.874511
+    avg_avg_return:                             0.94422704
+    avg_reward_per_step:                        0.21051025
+    proxy_corner:
+      avg_avg_return:                           0.01395881
+      avg_reward_per_step:                      0.0012512207
+  perf:
+    env_steps_per_second:                       352760.48895744217
+    ppo_updates_per_second:                     40.304375783587076
+    eval-batch-train-duration:                  0.4673218671232462
+    eval-batch-orig-duration:                   0.48009303491562605
+    eval-batch-shift-duration:                  0.4840024644508958
+    eval-batch-tree-duration:                   0.4902936890721321
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.380859
+      cheese-corne_dist_wavg:                   9.144967
+      levels_cheese_is_in_corner_avg:           0.061035156
+      mouse-cheese_dist_finite_avg:             16.191895
+      mouse-cheese_dist_finite_wavg:            13.318312
+      mouse-cheese_dist_solvable_avg:           16.191895
+      mouse-cheese_dist_solvable_wavg:          13.318314
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12542647
+    visit_patterns:
+      avg_first_visit_time:                     1284.9243
+      avg_last_visit_time:                      1918.3027
+      num_replay_batches:                       1928
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0029180418
+    avg_actor_approxkl3:                        0.001734647
+    avg_actor_clipfrac:                         0.04221802
+    avg_actor_loss:                             0.00025621612
+    avg_advantage:                              -0.00096946233
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00012834152
+    avg_entropy:                                0.08763043
+    avg_grad_norm_pre_clip:                     1.5658782
+    avg_loss:                                   0.00023275647
+    max:
+      max_actor_approxkl1:                      0.0052616037
+      max_actor_approxkl3:                      0.0035407883
+      max_actor_clipfrac:                       0.0748291
+      max_actor_loss:                           0.0023911446
+      max_advantage:                            0.07916069
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0001562025
+      max_entropy:                              0.107887164
+      max_grad_norm_pre_clip:                   3.0115733
+      max_loss:                                 0.0023536086
+    std:
+      std_actor_approxkl1:                      0.0017138396
+      std_actor_approxkl3:                      0.0010103496
+      std_actor_clipfrac:                       0.01929083
+      std_actor_loss:                           0.0010632479
+      std_advantage:                            0.016037123
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.2713332e-05
+      std_entropy:                              0.010342179
+      std_grad_norm_pre_clip:                   0.5946163
+      std_loss:                                 0.0010656737
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3393655
+    avg_avg_return:                             0.9819279
+    avg_benchmark_regret:                       0.010871284
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20398712
+    proxy_corner:
+      avg_avg_return:                           0.009868796
+      avg_reward_per_step:                      0.0017242432
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.679516
+    avg_avg_return:                             0.9739726
+    avg_benchmark_regret:                       0.016267223
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14046478
+    proxy_corner:
+      avg_avg_return:                           0.9739726
+      avg_reward_per_step:                      0.14046478
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3570576
+    avg_avg_return:                             0.9817485
+    avg_benchmark_regret:                       0.011088915
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19372559
+    proxy_corner:
+      avg_avg_return:                           0.010000782
+      avg_reward_per_step:                      0.0020523071
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.210686
+    avg_avg_return:                             0.015436123
+    avg_benchmark_regret:                       0.96919906
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0062713623
+    proxy_corner:
+      avg_avg_return:                           0.015436123
+      avg_reward_per_step:                      0.0062713623
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        125829120
+    env-step-replay-before:                     63799296
+    env-step-used-before:                       125829120
+    ppo-update-before:                          76800
+    env-step-generate-before:                   62029824
+    env-step-all-after:                         125861888
+    env-step-replay-after:                      63832064
+    env-step-used-after:                        125861888
+    ppo-update-after:                           76820
+    env-step-generate-after:                    62029824
+  train-replay:
+    shift_proportion:                           0.953125
+    avg_avg_episode_length:                     17.166182
+    avg_avg_return:                             0.86472887
+    avg_reward_per_step:                        0.054901123
+    proxy_corner:
+      avg_avg_return:                           0.04553646
+      avg_reward_per_step:                      0.00289917
+  train-all:
+    shift_proportion:                           0.953125
+    avg_avg_episode_length:                     17.166182
+    avg_avg_return:                             0.86472887
+    avg_reward_per_step:                        0.054901123
+    proxy_corner:
+      avg_avg_return:                           0.04553646
+      avg_reward_per_step:                      0.00289917
+  perf:
+    env_steps_per_second:                       354146.81910673954
+    ppo_updates_per_second:                     41.293824142472076
+    eval-batch-train-duration:                  0.5374498479068279
+    eval-batch-orig-duration:                   0.4836604967713356
+    eval-batch-shift-duration:                  0.48839851934462786
+    eval-batch-tree-duration:                   0.45263681933283806
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.358154
+      cheese-corne_dist_wavg:                   1.9416748
+      levels_cheese_is_in_corner_avg:           0.06274414
+      mouse-cheese_dist_finite_avg:             16.207031
+      mouse-cheese_dist_finite_wavg:            15.121599
+      mouse-cheese_dist_solvable_avg:           16.207031
+      mouse-cheese_dist_solvable_wavg:          15.121601
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12231899
+    visit_patterns:
+      avg_first_visit_time:                     1291.2095
+      avg_last_visit_time:                      1938.2612
+      num_replay_batches:                       1948
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0024879815
+    avg_actor_approxkl3:                        0.001656382
+    avg_actor_clipfrac:                         0.054779053
+    avg_actor_loss:                             0.0010441315
+    avg_advantage:                              0.0009876275
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            7.5336684e-05
+    avg_entropy:                                0.1233752
+    avg_grad_norm_pre_clip:                     2.5284216
+    avg_loss:                                   0.00095842464
+    max:
+      max_actor_approxkl1:                      0.005770711
+      max_actor_approxkl3:                      0.0036025993
+      max_actor_clipfrac:                       0.09240723
+      max_actor_loss:                           0.003796616
+      max_advantage:                            0.05087918
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          9.06106e-05
+      max_entropy:                              0.13593611
+      max_grad_norm_pre_clip:                   4.015458
+      max_loss:                                 0.0037016855
+    std:
+      std_actor_approxkl1:                      0.0015310571
+      std_actor_approxkl3:                      0.00083601783
+      std_actor_clipfrac:                       0.019217795
+      std_actor_loss:                           0.0010860636
+      std_advantage:                            0.012056422
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.8349525e-06
+      std_entropy:                              0.008564497
+      std_grad_norm_pre_clip:                   0.8366879
+      std_loss:                                 0.0010833202
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3279138
+    avg_avg_return:                             0.9817892
+    avg_benchmark_regret:                       0.011009969
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20394135
+    proxy_corner:
+      avg_avg_return:                           0.007626894
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.66336
+    avg_avg_return:                             0.97391593
+    avg_benchmark_regret:                       0.016323922
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14050293
+    proxy_corner:
+      avg_avg_return:                           0.97391593
+      avg_reward_per_step:                      0.14050293
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.326659
+    avg_avg_return:                             0.9823766
+    avg_benchmark_regret:                       0.010460847
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19449615
+    proxy_corner:
+      avg_avg_return:                           0.004131162
+      avg_reward_per_step:                      0.0010223389
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.44032
+    avg_avg_return:                             0.01273632
+    avg_benchmark_regret:                       0.97189885
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.005897522
+    proxy_corner:
+      avg_avg_return:                           0.01273632
+      avg_reward_per_step:                      0.005897522
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        126877696
+    env-step-replay-before:                     64356352
+    env-step-used-before:                       126877696
+    ppo-update-before:                          77440
+    env-step-generate-before:                   62521344
+    env-step-all-after:                         126910464
+    env-step-replay-after:                      64356352
+    env-step-used-after:                        126910464
+    ppo-update-after:                           77460
+    env-step-generate-after:                    62554112
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.524011
+    avg_avg_return:                             0.94844294
+    avg_reward_per_step:                        0.19949341
+    proxy_corner:
+      avg_avg_return:                           0.013176389
+      avg_reward_per_step:                      0.0015869141
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.524011
+    avg_avg_return:                             0.94844294
+    avg_reward_per_step:                        0.19949341
+    proxy_corner:
+      avg_avg_return:                           0.013176389
+      avg_reward_per_step:                      0.0015869141
+  perf:
+    env_steps_per_second:                       300543.41789824754
+    ppo_updates_per_second:                     41.65010068733954
+    eval-batch-train-duration:                  0.51162573043257
+    eval-batch-orig-duration:                   0.4715046314522624
+    eval-batch-shift-duration:                  0.43997284211218357
+    eval-batch-tree-duration:                   0.4877877049148083
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.337402
+      cheese-corne_dist_wavg:                   13.640957
+      levels_cheese_is_in_corner_avg:           0.06225586
+      mouse-cheese_dist_finite_avg:             16.231445
+      mouse-cheese_dist_finite_wavg:            6.1244063
+      mouse-cheese_dist_solvable_avg:           16.231445
+      mouse-cheese_dist_solvable_wavg:          6.1244073
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.121755935
+    visit_patterns:
+      avg_first_visit_time:                     1297.6162
+      avg_last_visit_time:                      1954.2805
+      num_replay_batches:                       1964
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0022672748
+    avg_actor_approxkl3:                        0.0018095849
+    avg_actor_clipfrac:                         0.04197998
+    avg_actor_loss:                             0.0014612658
+    avg_advantage:                              0.0016221735
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00014404663
+    avg_entropy:                                0.085046664
+    avg_grad_norm_pre_clip:                     2.2186947
+    avg_loss:                                   0.0014482425
+    max:
+      max_actor_approxkl1:                      0.00477313
+      max_actor_approxkl3:                      0.0036566132
+      max_actor_clipfrac:                       0.072509766
+      max_actor_loss:                           0.003108466
+      max_advantage:                            0.07724756
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.000188659
+      max_entropy:                              0.10175889
+      max_grad_norm_pre_clip:                   3.8800633
+      max_loss:                                 0.0031114828
+    std:
+      std_actor_approxkl1:                      0.001203214
+      std_actor_approxkl3:                      0.0009367062
+      std_actor_clipfrac:                       0.015760686
+      std_actor_loss:                           0.0011037709
+      std_advantage:                            0.016432915
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.6359923e-05
+      std_entropy:                              0.0076058162
+      std_grad_norm_pre_clip:                   0.8429561
+      std_loss:                                 0.0011026586
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4118147
+    avg_avg_return:                             0.9813909
+    avg_benchmark_regret:                       0.011408251
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20298004
+    proxy_corner:
+      avg_avg_return:                           0.008035312
+      avg_reward_per_step:                      0.0016021729
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.609053
+    avg_avg_return:                             0.9744461
+    avg_benchmark_regret:                       0.015793677
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1407013
+    proxy_corner:
+      avg_avg_return:                           0.9744461
+      avg_reward_per_step:                      0.1407013
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3470163
+    avg_avg_return:                             0.9814733
+    avg_benchmark_regret:                       0.01136407
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19480896
+    proxy_corner:
+      avg_avg_return:                           0.0046002693
+      avg_reward_per_step:                      0.0011062622
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.35452
+    avg_avg_return:                             0.014132169
+    avg_benchmark_regret:                       0.97050303
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0061187744
+    proxy_corner:
+      avg_avg_return:                           0.014132169
+      avg_reward_per_step:                      0.0061187744
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        127926272
+    env-step-replay-before:                     64946176
+    env-step-used-before:                       127926272
+    ppo-update-before:                          78080
+    env-step-generate-before:                   62980096
+    env-step-all-after:                         127959040
+    env-step-replay-after:                      64946176
+    env-step-used-after:                        127959040
+    ppo-update-after:                           78100
+    env-step-generate-after:                    63012864
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.6386814
+    avg_avg_return:                             0.9473103
+    avg_reward_per_step:                        0.18182373
+    proxy_corner:
+      avg_avg_return:                           0.009318207
+      avg_reward_per_step:                      0.001739502
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.6386814
+    avg_avg_return:                             0.9473103
+    avg_reward_per_step:                        0.18182373
+    proxy_corner:
+      avg_avg_return:                           0.009318207
+      avg_reward_per_step:                      0.001739502
+  perf:
+    env_steps_per_second:                       273573.9870654461
+    ppo_updates_per_second:                     41.54661100156136
+    eval-batch-train-duration:                  0.5240594055503607
+    eval-batch-orig-duration:                   0.46652551740407944
+    eval-batch-shift-duration:                  0.45051452051848173
+    eval-batch-tree-duration:                   0.4827181212604046
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.279053
+      cheese-corne_dist_wavg:                   12.739276
+      levels_cheese_is_in_corner_avg:           0.061767578
+      mouse-cheese_dist_finite_avg:             16.253906
+      mouse-cheese_dist_finite_wavg:            7.0271587
+      mouse-cheese_dist_solvable_avg:           16.253906
+      mouse-cheese_dist_solvable_wavg:          7.0271587
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.121876754
+    visit_patterns:
+      avg_first_visit_time:                     1305.092
+      avg_last_visit_time:                      1972.2749
+      num_replay_batches:                       1982
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0034080104
+    avg_actor_approxkl3:                        0.0024406277
+    avg_actor_clipfrac:                         0.050042726
+    avg_actor_loss:                             0.0020423345
+    avg_advantage:                              0.003815719
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00015228962
+    avg_entropy:                                0.088648446
+    avg_grad_norm_pre_clip:                     2.4278786
+    avg_loss:                                   0.0020298306
+    max:
+      max_actor_approxkl1:                      0.008499727
+      max_actor_approxkl3:                      0.0056848107
+      max_actor_clipfrac:                       0.080200195
+      max_actor_loss:                           0.006142614
+      max_advantage:                            0.07022917
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0001798046
+      max_entropy:                              0.099557236
+      max_grad_norm_pre_clip:                   4.02743
+      max_loss:                                 0.0061132926
+    std:
+      std_actor_approxkl1:                      0.0022588682
+      std_actor_approxkl3:                      0.0014092103
+      std_actor_clipfrac:                       0.017376892
+      std_actor_loss:                           0.0015309686
+      std_advantage:                            0.01669211
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.6249707e-05
+      std_entropy:                              0.0056472514
+      std_grad_norm_pre_clip:                   0.8451423
+      std_loss:                                 0.0015283766
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3424244
+    avg_avg_return:                             0.9818013
+    avg_benchmark_regret:                       0.010997837
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20413208
+    proxy_corner:
+      avg_avg_return:                           0.007987396
+      avg_reward_per_step:                      0.0016021729
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.974757
+    avg_avg_return:                             0.9730714
+    avg_benchmark_regret:                       0.017168447
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1389389
+    proxy_corner:
+      avg_avg_return:                           0.9730714
+      avg_reward_per_step:                      0.1389389
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.275846
+    avg_avg_return:                             0.9819655
+    avg_benchmark_regret:                       0.010871971
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19500732
+    proxy_corner:
+      avg_avg_return:                           0.0048436867
+      avg_reward_per_step:                      0.0012664795
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.16822
+    avg_avg_return:                             0.01604626
+    avg_benchmark_regret:                       0.96858895
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0062789917
+    proxy_corner:
+      avg_avg_return:                           0.01604626
+      avg_reward_per_step:                      0.0062789917
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        128974848
+    env-step-replay-before:                     65601536
+    env-step-used-before:                       128974848
+    ppo-update-before:                          78720
+    env-step-generate-before:                   63373312
+    env-step-all-after:                         129007616
+    env-step-replay-after:                      65601536
+    env-step-used-after:                        129007616
+    ppo-update-after:                           78740
+    env-step-generate-after:                    63406080
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.6381893
+    avg_avg_return:                             0.94625866
+    avg_reward_per_step:                        0.19534302
+    proxy_corner:
+      avg_avg_return:                           0.009368457
+      avg_reward_per_step:                      0.0043945312
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.6381893
+    avg_avg_return:                             0.94625866
+    avg_reward_per_step:                        0.19534302
+    proxy_corner:
+      avg_avg_return:                           0.009368457
+      avg_reward_per_step:                      0.0043945312
+  perf:
+    env_steps_per_second:                       389251.6275050009
+    ppo_updates_per_second:                     40.56272953001378
+    eval-batch-train-duration:                  0.4679603399708867
+    eval-batch-orig-duration:                   0.4416306568309665
+    eval-batch-shift-duration:                  0.4092181995511055
+    eval-batch-tree-duration:                   0.42434967774897814
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.233643
+      cheese-corne_dist_wavg:                   14.520074
+      levels_cheese_is_in_corner_avg:           0.06274414
+      mouse-cheese_dist_finite_avg:             16.266357
+      mouse-cheese_dist_finite_wavg:            15.124855
+      mouse-cheese_dist_solvable_avg:           16.266357
+      mouse-cheese_dist_solvable_wavg:          15.124859
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1211229
+    visit_patterns:
+      avg_first_visit_time:                     1312.842
+      avg_last_visit_time:                      1992.3169
+      num_replay_batches:                       2002
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0032789577
+    avg_actor_approxkl3:                        0.0025642707
+    avg_actor_clipfrac:                         0.05078125
+    avg_actor_loss:                             0.0019485716
+    avg_advantage:                              0.00056923786
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0001400064
+    avg_entropy:                                0.086728886
+    avg_grad_norm_pre_clip:                     2.7460907
+    avg_loss:                                   0.001931846
+    max:
+      max_actor_approxkl1:                      0.007006021
+      max_actor_approxkl3:                      0.005711163
+      max_actor_clipfrac:                       0.08569336
+      max_actor_loss:                           0.0054966104
+      max_advantage:                            0.068455875
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00015630465
+      max_entropy:                              0.10139009
+      max_grad_norm_pre_clip:                   4.839008
+      max_loss:                                 0.005476335
+    std:
+      std_actor_approxkl1:                      0.0022275234
+      std_actor_approxkl3:                      0.001818563
+      std_actor_clipfrac:                       0.022627404
+      std_actor_loss:                           0.0016373965
+      std_advantage:                            0.016761558
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          9.60968e-06
+      std_entropy:                              0.00739263
+      std_grad_norm_pre_clip:                   1.0865103
+      std_loss:                                 0.0016331768
+  eval-batch-train:
+    avg_avg_episode_length:                     7.35662
+    avg_avg_return:                             0.98144084
+    avg_benchmark_regret:                       0.011358317
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20410156
+    proxy_corner:
+      avg_avg_return:                           0.007933338
+      avg_reward_per_step:                      0.0015945435
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.610893
+    avg_avg_return:                             0.9739417
+    avg_benchmark_regret:                       0.016298123
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1409607
+    proxy_corner:
+      avg_avg_return:                           0.9739417
+      avg_reward_per_step:                      0.1409607
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3175015
+    avg_avg_return:                             0.98183286
+    avg_benchmark_regret:                       0.011004558
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19489288
+    proxy_corner:
+      avg_avg_return:                           0.004962574
+      avg_reward_per_step:                      0.0012893677
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.44498
+    avg_avg_return:                             0.0114178825
+    avg_benchmark_regret:                       0.9732173
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.005996704
+    proxy_corner:
+      avg_avg_return:                           0.0114178825
+      avg_reward_per_step:                      0.005996704
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        130023424
+    env-step-replay-before:                     66158592
+    env-step-used-before:                       130023424
+    ppo-update-before:                          79360
+    env-step-generate-before:                   63864832
+    env-step-all-after:                         130056192
+    env-step-replay-after:                      66158592
+    env-step-used-after:                        130056192
+    ppo-update-after:                           79380
+    env-step-generate-after:                    63897600
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.833501
+    avg_avg_return:                             0.9462998
+    avg_reward_per_step:                        0.18667603
+    proxy_corner:
+      avg_avg_return:                           0.012823055
+      avg_reward_per_step:                      0.0012817383
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.833501
+    avg_avg_return:                             0.9462998
+    avg_reward_per_step:                        0.18667603
+    proxy_corner:
+      avg_avg_return:                           0.012823055
+      avg_reward_per_step:                      0.0012817383
+  perf:
+    env_steps_per_second:                       295832.58649790933
+    ppo_updates_per_second:                     42.58536318400908
+    eval-batch-train-duration:                  0.5358041096478701
+    eval-batch-orig-duration:                   0.466114136390388
+    eval-batch-shift-duration:                  0.43819670751690865
+    eval-batch-tree-duration:                   0.47168635483831167
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.135498
+      cheese-corne_dist_wavg:                   10.010429
+      levels_cheese_is_in_corner_avg:           0.06298828
+      mouse-cheese_dist_finite_avg:             16.302002
+      mouse-cheese_dist_finite_wavg:            16.929916
+      mouse-cheese_dist_solvable_avg:           16.302002
+      mouse-cheese_dist_solvable_wavg:          16.929916
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.121101394
+    visit_patterns:
+      avg_first_visit_time:                     1322.5317
+      avg_last_visit_time:                      2009.2844
+      num_replay_batches:                       2019
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013066264
+    avg_actor_approxkl3:                        0.0010173294
+    avg_actor_clipfrac:                         0.032977294
+    avg_actor_loss:                             0.000304669
+    avg_advantage:                              0.002698741
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00015209732
+    avg_entropy:                                0.091138534
+    avg_grad_norm_pre_clip:                     1.833383
+    avg_loss:                                   0.00028957913
+    max:
+      max_actor_approxkl1:                      0.0027787276
+      max_actor_approxkl3:                      0.002279013
+      max_actor_clipfrac:                       0.047729492
+      max_actor_loss:                           0.001611704
+      max_advantage:                            0.07794094
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00021293192
+      max_entropy:                              0.10099451
+      max_grad_norm_pre_clip:                   3.9144597
+      max_loss:                                 0.0016103795
+    std:
+      std_actor_approxkl1:                      0.0007238622
+      std_actor_approxkl3:                      0.0005492761
+      std_actor_clipfrac:                       0.012052937
+      std_actor_loss:                           0.00069798215
+      std_advantage:                            0.016000615
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.4784342e-05
+      std_entropy:                              0.007214015
+      std_grad_norm_pre_clip:                   0.744842
+      std_loss:                                 0.00069845206
+  eval-batch-train:
+    avg_avg_episode_length:                     7.2892475
+    avg_avg_return:                             0.9820811
+    avg_benchmark_regret:                       0.010718081
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20494843
+    proxy_corner:
+      avg_avg_return:                           0.0076264776
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.577222
+    avg_avg_return:                             0.9744208
+    avg_benchmark_regret:                       0.015819006
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14111328
+    proxy_corner:
+      avg_avg_return:                           0.9744208
+      avg_reward_per_step:                      0.14111328
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.2583723
+    avg_avg_return:                             0.98180753
+    avg_benchmark_regret:                       0.0110299215
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19504547
+    proxy_corner:
+      avg_avg_return:                           0.004925683
+      avg_reward_per_step:                      0.0013122559
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.415016
+    avg_avg_return:                             0.014042928
+    avg_benchmark_regret:                       0.97059226
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0059890747
+    proxy_corner:
+      avg_avg_return:                           0.014042928
+      avg_reward_per_step:                      0.0059890747
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        131072000
+    env-step-replay-before:                     66551808
+    env-step-used-before:                       131072000
+    ppo-update-before:                          80000
+    env-step-generate-before:                   64520192
+    env-step-all-after:                         131104768
+    env-step-replay-after:                      66551808
+    env-step-used-after:                        131104768
+    ppo-update-after:                           80020
+    env-step-generate-after:                    64552960
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.623761
+    avg_avg_return:                             0.9466789
+    avg_reward_per_step:                        0.17785645
+    proxy_corner:
+      avg_avg_return:                           0.0070815003
+      avg_reward_per_step:                      0.0005187988
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.623761
+    avg_avg_return:                             0.9466789
+    avg_reward_per_step:                        0.17785645
+    proxy_corner:
+      avg_avg_return:                           0.0070815003
+      avg_reward_per_step:                      0.0005187988
+  perf:
+    env_steps_per_second:                       245207.57059796902
+    ppo_updates_per_second:                     42.26019883032797
+    eval-batch-train-duration:                  0.6841503651812673
+    eval-batch-orig-duration:                   0.6034998735412955
+    eval-batch-shift-duration:                  0.606827674433589
+    eval-batch-tree-duration:                   0.6156022138893604
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.053223
+      cheese-corne_dist_wavg:                   1.9166739
+      levels_cheese_is_in_corner_avg:           0.0637207
+      mouse-cheese_dist_finite_avg:             16.350586
+      mouse-cheese_dist_finite_wavg:            16.933535
+      mouse-cheese_dist_solvable_avg:           16.350586
+      mouse-cheese_dist_solvable_wavg:          16.933535
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12195131
+    visit_patterns:
+      avg_first_visit_time:                     1332.7556
+      avg_last_visit_time:                      2021.2966
+      num_replay_batches:                       2031
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0023322513
+    avg_actor_approxkl3:                        0.001831582
+    avg_actor_clipfrac:                         0.04182739
+    avg_actor_loss:                             0.0017944884
+    avg_advantage:                              -0.0010970944
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.000119385346
+    avg_entropy:                                0.09394265
+    avg_grad_norm_pre_clip:                     2.3497045
+    avg_loss:                                   0.0017602382
+    max:
+      max_actor_approxkl1:                      0.006933442
+      max_actor_approxkl3:                      0.0052037095
+      max_actor_clipfrac:                       0.06896973
+      max_actor_loss:                           0.006348919
+      max_advantage:                            0.053342223
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00013090146
+      max_entropy:                              0.10298128
+      max_grad_norm_pre_clip:                   4.9094906
+      max_loss:                                 0.0063112555
+    std:
+      std_actor_approxkl1:                      0.0018576906
+      std_actor_approxkl3:                      0.0014719574
+      std_actor_clipfrac:                       0.018172957
+      std_actor_loss:                           0.0017322731
+      std_advantage:                            0.015277699
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.844816e-06
+      std_entropy:                              0.0066427393
+      std_grad_norm_pre_clip:                   1.0994521
+      std_loss:                                 0.0017311387
+  eval-batch-train:
+    avg_avg_episode_length:                     7.330405
+    avg_avg_return:                             0.9822838
+    avg_benchmark_regret:                       0.010515387
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20488739
+    proxy_corner:
+      avg_avg_return:                           0.009503629
+      avg_reward_per_step:                      0.0016937256
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.697954
+    avg_avg_return:                             0.9741231
+    avg_benchmark_regret:                       0.016116677
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14026642
+    proxy_corner:
+      avg_avg_return:                           0.9741231
+      avg_reward_per_step:                      0.14026642
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3253117
+    avg_avg_return:                             0.98169786
+    avg_benchmark_regret:                       0.011139551
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.193367
+    proxy_corner:
+      avg_avg_return:                           0.0054638423
+      avg_reward_per_step:                      0.001449585
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.41503
+    avg_avg_return:                             0.013363204
+    avg_benchmark_regret:                       0.971272
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0059661865
+    proxy_corner:
+      avg_avg_return:                           0.013363204
+      avg_reward_per_step:                      0.0059661865
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        132120576
+    env-step-replay-before:                     67076096
+    env-step-used-before:                       132120576
+    ppo-update-before:                          80640
+    env-step-generate-before:                   65044480
+    env-step-all-after:                         132153344
+    env-step-replay-after:                      67076096
+    env-step-used-after:                        132153344
+    ppo-update-after:                           80660
+    env-step-generate-after:                    65077248
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     8.205696
+    avg_avg_return:                             0.94258356
+    avg_reward_per_step:                        0.18582153
+    proxy_corner:
+      avg_avg_return:                           0.0048305024
+      avg_reward_per_step:                      0.00036621094
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     8.205696
+    avg_avg_return:                             0.94258356
+    avg_reward_per_step:                        0.18582153
+    proxy_corner:
+      avg_avg_return:                           0.0048305024
+      avg_reward_per_step:                      0.00036621094
+  perf:
+    env_steps_per_second:                       300700.8072440096
+    ppo_updates_per_second:                     42.241106862881196
+    eval-batch-train-duration:                  0.5163216860964894
+    eval-batch-orig-duration:                   0.4805743619799614
+    eval-batch-shift-duration:                  0.4738573292270303
+    eval-batch-tree-duration:                   0.4769784351810813
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    10.020996
+      cheese-corne_dist_wavg:                   12.705334
+      levels_cheese_is_in_corner_avg:           0.0637207
+      mouse-cheese_dist_finite_avg:             16.375977
+      mouse-cheese_dist_finite_wavg:            9.74553
+      mouse-cheese_dist_solvable_avg:           16.375977
+      mouse-cheese_dist_solvable_wavg:          9.745531
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.123018265
+    visit_patterns:
+      avg_first_visit_time:                     1342.3914
+      avg_last_visit_time:                      2037.2554
+      num_replay_batches:                       2047
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0020412055
+    avg_actor_approxkl3:                        0.0016247559
+    avg_actor_clipfrac:                         0.04545288
+    avg_actor_loss:                             0.0014565758
+    avg_advantage:                              -0.0028381855
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00012407897
+    avg_entropy:                                0.09551441
+    avg_grad_norm_pre_clip:                     2.4190013
+    avg_loss:                                   0.0014231008
+    max:
+      max_actor_approxkl1:                      0.0050144293
+      max_actor_approxkl3:                      0.003462331
+      max_actor_clipfrac:                       0.08117676
+      max_actor_loss:                           0.0035354514
+      max_advantage:                            0.055241108
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0001444557
+      max_entropy:                              0.11116068
+      max_grad_norm_pre_clip:                   3.9227629
+      max_loss:                                 0.003501306
+    std:
+      std_actor_approxkl1:                      0.001544355
+      std_actor_approxkl3:                      0.00090275763
+      std_actor_clipfrac:                       0.017349249
+      std_actor_loss:                           0.0010642499
+      std_advantage:                            0.015843932
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.02072345e-05
+      std_entropy:                              0.009240836
+      std_grad_norm_pre_clip:                   0.69187695
+      std_loss:                                 0.001060555
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3898935
+    avg_avg_return:                             0.9815043
+    avg_benchmark_regret:                       0.011294819
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20233917
+    proxy_corner:
+      avg_avg_return:                           0.007722488
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.657879
+    avg_avg_return:                             0.9741055
+    avg_benchmark_regret:                       0.016134344
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14057922
+    proxy_corner:
+      avg_avg_return:                           0.9741055
+      avg_reward_per_step:                      0.14057922
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4798913
+    avg_avg_return:                             0.98086977
+    avg_benchmark_regret:                       0.0119676925
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.1932602
+    proxy_corner:
+      avg_avg_return:                           0.0044077695
+      avg_reward_per_step:                      0.0011367798
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.40268
+    avg_avg_return:                             0.013424844
+    avg_benchmark_regret:                       0.97121036
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.005973816
+    proxy_corner:
+      avg_avg_return:                           0.013424844
+      avg_reward_per_step:                      0.005973816
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        133169152
+    env-step-replay-before:                     67600384
+    env-step-used-before:                       133169152
+    ppo-update-before:                          81280
+    env-step-generate-before:                   65568768
+    env-step-all-after:                         133201920
+    env-step-replay-after:                      67633152
+    env-step-used-after:                        133201920
+    ppo-update-after:                           81300
+    env-step-generate-after:                    65568768
+  train-replay:
+    shift_proportion:                           0.9375
+    avg_avg_episode_length:                     17.182453
+    avg_avg_return:                             0.8640954
+    avg_reward_per_step:                        0.054992676
+    proxy_corner:
+      avg_avg_return:                           0.06801874
+      avg_reward_per_step:                      0.004058838
+  train-all:
+    shift_proportion:                           0.9375
+    avg_avg_episode_length:                     17.182453
+    avg_avg_return:                             0.8640954
+    avg_reward_per_step:                        0.054992676
+    proxy_corner:
+      avg_avg_return:                           0.06801874
+      avg_reward_per_step:                      0.004058838
+  perf:
+    env_steps_per_second:                       324399.8354307092
+    ppo_updates_per_second:                     40.87816367654566
+    eval-batch-train-duration:                  0.46507050283253193
+    eval-batch-orig-duration:                   0.4575728131458163
+    eval-batch-shift-duration:                  0.4109713099896908
+    eval-batch-tree-duration:                   0.4244585568085313
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.937012
+      cheese-corne_dist_wavg:                   1.9058418
+      levels_cheese_is_in_corner_avg:           0.064941406
+      mouse-cheese_dist_finite_avg:             16.391357
+      mouse-cheese_dist_finite_wavg:            16.039454
+      mouse-cheese_dist_solvable_avg:           16.391357
+      mouse-cheese_dist_solvable_wavg:          16.039455
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12258176
+    visit_patterns:
+      avg_first_visit_time:                     1350.8735
+      avg_last_visit_time:                      2054.3076
+      num_replay_batches:                       2064
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0019320038
+    avg_actor_approxkl3:                        0.001230084
+    avg_actor_clipfrac:                         0.046020508
+    avg_actor_loss:                             0.00014120036
+    avg_advantage:                              0.0011403786
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            7.609852e-05
+    avg_entropy:                                0.121618845
+    avg_grad_norm_pre_clip:                     2.314525
+    avg_loss:                                   5.7630765e-05
+    max:
+      max_actor_approxkl1:                      0.0039669415
+      max_actor_approxkl3:                      0.0022203252
+      max_actor_clipfrac:                       0.06359863
+      max_actor_loss:                           0.0015015453
+      max_advantage:                            0.056412578
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          9.1008e-05
+      max_entropy:                              0.13827749
+      max_grad_norm_pre_clip:                   3.424009
+      max_loss:                                 0.0014297004
+    std:
+      std_actor_approxkl1:                      0.00097122113
+      std_actor_approxkl3:                      0.00046107755
+      std_actor_clipfrac:                       0.0144376205
+      std_actor_loss:                           0.00088932214
+      std_advantage:                            0.012332593
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.8878986e-06
+      std_entropy:                              0.008726634
+      std_grad_norm_pre_clip:                   0.59616154
+      std_loss:                                 0.00089253724
+  eval-batch-train:
+    avg_avg_episode_length:                     7.308337
+    avg_avg_return:                             0.9822918
+    avg_benchmark_regret:                       0.010507346
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20462799
+    proxy_corner:
+      avg_avg_return:                           0.0076229204
+      avg_reward_per_step:                      0.0015640259
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.831114
+    avg_avg_return:                             0.9738753
+    avg_benchmark_regret:                       0.016364526
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13942719
+    proxy_corner:
+      avg_avg_return:                           0.9738753
+      avg_reward_per_step:                      0.13942719
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.2712984
+    avg_avg_return:                             0.9823548
+    avg_benchmark_regret:                       0.010482622
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.1951294
+    proxy_corner:
+      avg_avg_return:                           0.004350231
+      avg_reward_per_step:                      0.0011138916
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.45611
+    avg_avg_return:                             0.0119544435
+    avg_benchmark_regret:                       0.9726808
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.005958557
+    proxy_corner:
+      avg_avg_return:                           0.0119544435
+      avg_reward_per_step:                      0.005958557
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        134217728
+    env-step-replay-before:                     68157440
+    env-step-used-before:                       134217728
+    ppo-update-before:                          81920
+    env-step-generate-before:                   66060288
+    env-step-all-after:                         134250496
+    env-step-replay-after:                      68157440
+    env-step-used-after:                        134250496
+    ppo-update-after:                           81940
+    env-step-generate-after:                    66093056
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.7006288
+    avg_avg_return:                             0.94410527
+    avg_reward_per_step:                        0.1743164
+    proxy_corner:
+      avg_avg_return:                           0.0069995164
+      avg_reward_per_step:                      0.00061035156
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.7006288
+    avg_avg_return:                             0.94410527
+    avg_reward_per_step:                        0.1743164
+    proxy_corner:
+      avg_avg_return:                           0.0069995164
+      avg_reward_per_step:                      0.00061035156
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  perf:
+    env_steps_per_second:                       324410.02597126324
+    ppo_updates_per_second:                     41.91568747605798
+    eval-batch-train-duration:                  0.6452942229807377
+    eval-batch-orig-duration:                   0.5729303229600191
+    eval-batch-shift-duration:                  0.5767564615234733
+    eval-batch-tree-duration:                   0.5428612446412444
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.835693
+      cheese-corne_dist_wavg:                   9.089795
+      levels_cheese_is_in_corner_avg:           0.064208984
+      mouse-cheese_dist_finite_avg:             16.43872
+      mouse-cheese_dist_finite_wavg:            16.94003
+      mouse-cheese_dist_solvable_avg:           16.43872
+      mouse-cheese_dist_solvable_wavg:          16.94003
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12324187
+    visit_patterns:
+      avg_first_visit_time:                     1362.6653
+      avg_last_visit_time:                      2070.3445
+      num_replay_batches:                       2080
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0016599139
+    avg_actor_approxkl3:                        0.0012856618
+    avg_actor_clipfrac:                         0.036724854
+    avg_actor_loss:                             0.0006705183
+    avg_advantage:                              0.0050753425
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0001283105
+    avg_entropy:                                0.092933014
+    avg_grad_norm_pre_clip:                     2.0359666
+    avg_loss:                                   0.00064174057
+    max:
+      max_actor_approxkl1:                      0.0035551835
+      max_actor_approxkl3:                      0.0036808369
+      max_actor_clipfrac:                       0.06555176
+      max_actor_loss:                           0.0054381723
+      max_advantage:                            0.07235873
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00014010434
+      max_entropy:                              0.117823295
+      max_grad_norm_pre_clip:                   3.4906654
+      max_loss:                                 0.005379608
+    std:
+      std_actor_approxkl1:                      0.0011043509
+      std_actor_approxkl3:                      0.0009359556
+      std_actor_clipfrac:                       0.016203392
+      std_actor_loss:                           0.0014971644
+      std_advantage:                            0.015945088
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          8.8236075e-06
+      std_entropy:                              0.009938414
+      std_grad_norm_pre_clip:                   0.7995564
+      std_loss:                                 0.0014925386
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3558016
+    avg_avg_return:                             0.9816472
+    avg_benchmark_regret:                       0.01115197
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20275116
+    proxy_corner:
+      avg_avg_return:                           0.0076267896
+      avg_reward_per_step:                      0.0015716553
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.534023
+    avg_avg_return:                             0.97510374
+    avg_benchmark_regret:                       0.015136083
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14156342
+    proxy_corner:
+      avg_avg_return:                           0.97510374
+      avg_reward_per_step:                      0.14156342
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.294227
+    avg_avg_return:                             0.9819272
+    avg_benchmark_regret:                       0.010910255
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19432831
+    proxy_corner:
+      avg_avg_return:                           0.004323956
+      avg_reward_per_step:                      0.0010910034
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.26561
+    avg_avg_return:                             0.014396599
+    avg_benchmark_regret:                       0.9702386
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0063705444
+    proxy_corner:
+      avg_avg_return:                           0.014396599
+      avg_reward_per_step:                      0.0063705444
+    rollouts_gif:                               float32(132, 225, 225, 3)
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        135266304
+    env-step-replay-before:                     68780032
+    env-step-used-before:                       135266304
+    ppo-update-before:                          82560
+    env-step-generate-before:                   66486272
+    env-step-all-after:                         135299072
+    env-step-replay-after:                      68780032
+    env-step-used-after:                        135299072
+    ppo-update-after:                           82580
+    env-step-generate-after:                    66519040
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     8.005706
+    avg_avg_return:                             0.94094175
+    avg_reward_per_step:                        0.18319702
+    proxy_corner:
+      avg_avg_return:                           0.005185305
+      avg_reward_per_step:                      0.0008239746
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     8.005706
+    avg_avg_return:                             0.94094175
+    avg_reward_per_step:                        0.18319702
+    proxy_corner:
+      avg_avg_return:                           0.005185305
+      avg_reward_per_step:                      0.0008239746
+  perf:
+    env_steps_per_second:                       315650.4368863204
+    ppo_updates_per_second:                     42.211162252210016
+    eval-batch-train-duration:                  0.5341290319338441
+    eval-batch-orig-duration:                   0.4791708495467901
+    eval-batch-shift-duration:                  0.47233398258686066
+    eval-batch-tree-duration:                   0.451246683485806
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.820068
+      cheese-corne_dist_wavg:                   9.979493
+      levels_cheese_is_in_corner_avg:           0.063964844
+      mouse-cheese_dist_finite_avg:             16.46875
+      mouse-cheese_dist_finite_wavg:            16.945936
+      mouse-cheese_dist_solvable_avg:           16.46875
+      mouse-cheese_dist_solvable_wavg:          16.945936
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12269685
+    visit_patterns:
+      avg_first_visit_time:                     1370.3892
+      avg_last_visit_time:                      2089.2788
+      num_replay_batches:                       2099
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002076708
+    avg_actor_approxkl3:                        0.0015032456
+    avg_actor_clipfrac:                         0.043536376
+    avg_actor_loss:                             0.00047986564
+    avg_advantage:                              -0.0002126384
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00013887118
+    avg_entropy:                                0.10207529
+    avg_grad_norm_pre_clip:                     2.119744
+    avg_loss:                                   0.0004472259
+    max:
+      max_actor_approxkl1:                      0.004982124
+      max_actor_approxkl3:                      0.003349661
+      max_actor_clipfrac:                       0.07458496
+      max_actor_loss:                           0.002411431
+      max_advantage:                            0.070323646
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0001548749
+      max_entropy:                              0.1136661
+      max_grad_norm_pre_clip:                   4.0421853
+      max_loss:                                 0.0023664131
+    std:
+      std_actor_approxkl1:                      0.0013722441
+      std_actor_approxkl3:                      0.000823051
+      std_actor_clipfrac:                       0.017716657
+      std_actor_loss:                           0.0008217924
+      std_advantage:                            0.016681137
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          8.186542e-06
+      std_entropy:                              0.008593643
+      std_grad_norm_pre_clip:                   0.8506604
+      std_loss:                                 0.0008217603
+  eval-batch-train:
+    avg_avg_episode_length:                     7.395791
+    avg_avg_return:                             0.9813917
+    avg_benchmark_regret:                       0.011407434
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20367432
+    proxy_corner:
+      avg_avg_return:                           0.0076272064
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.516966
+    avg_avg_return:                             0.9754968
+    avg_benchmark_regret:                       0.014743013
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14154053
+    proxy_corner:
+      avg_avg_return:                           0.9754968
+      avg_reward_per_step:                      0.14154053
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.39565
+    avg_avg_return:                             0.9812675
+    avg_benchmark_regret:                       0.011569899
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19293976
+    proxy_corner:
+      avg_avg_return:                           0.0041548726
+      avg_reward_per_step:                      0.0010375977
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.54498
+    avg_avg_return:                             0.011459088
+    avg_benchmark_regret:                       0.9731761
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0059127808
+    proxy_corner:
+      avg_avg_return:                           0.011459088
+      avg_reward_per_step:                      0.0059127808
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        136314880
+    env-step-replay-before:                     69107712
+    env-step-used-before:                       136314880
+    ppo-update-before:                          83200
+    env-step-generate-before:                   67207168
+    env-step-all-after:                         136347648
+    env-step-replay-after:                      69107712
+    env-step-used-after:                        136347648
+    ppo-update-after:                           83220
+    env-step-generate-after:                    67239936
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     8.113895
+    avg_avg_return:                             0.9444668
+    avg_reward_per_step:                        0.17694092
+    proxy_corner:
+      avg_avg_return:                           0.010755792
+      avg_reward_per_step:                      0.0009765625
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     8.113895
+    avg_avg_return:                             0.9444668
+    avg_reward_per_step:                        0.17694092
+    proxy_corner:
+      avg_avg_return:                           0.010755792
+      avg_reward_per_step:                      0.0009765625
+  perf:
+    env_steps_per_second:                       323936.86749375693
+    ppo_updates_per_second:                     41.89925625922293
+    eval-batch-train-duration:                  0.5393177028745413
+    eval-batch-orig-duration:                   0.4882343402132392
+    eval-batch-shift-duration:                  0.46249027736485004
+    eval-batch-tree-duration:                   0.46352032851427794
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.82959
+      cheese-corne_dist_wavg:                   12.683544
+      levels_cheese_is_in_corner_avg:           0.064453125
+      mouse-cheese_dist_finite_avg:             16.499268
+      mouse-cheese_dist_finite_wavg:            10.649212
+      mouse-cheese_dist_solvable_avg:           16.499268
+      mouse-cheese_dist_solvable_wavg:          10.649212
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1231272
+    visit_patterns:
+      avg_first_visit_time:                     1378.9832
+      avg_last_visit_time:                      2099.364
+      num_replay_batches:                       2109
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0026381246
+    avg_actor_approxkl3:                        0.0019517804
+    avg_actor_clipfrac:                         0.04697876
+    avg_actor_loss:                             0.0015631969
+    avg_advantage:                              -0.007757198
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00012855658
+    avg_entropy:                                0.097345866
+    avg_grad_norm_pre_clip:                     2.5534298
+    avg_loss:                                   0.0015301293
+    max:
+      max_actor_approxkl1:                      0.007990263
+      max_actor_approxkl3:                      0.0057017105
+      max_actor_clipfrac:                       0.08203125
+      max_actor_loss:                           0.0049834293
+      max_advantage:                            0.06377947
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00017097376
+      max_entropy:                              0.11098543
+      max_grad_norm_pre_clip:                   4.3679295
+      max_loss:                                 0.0049417396
+    std:
+      std_actor_approxkl1:                      0.001745609
+      std_actor_approxkl3:                      0.0011749027
+      std_actor_clipfrac:                       0.018519972
+      std_actor_loss:                           0.0012170057
+      std_advantage:                            0.01577942
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.5388234e-05
+      std_entropy:                              0.007991212
+      std_grad_norm_pre_clip:                   0.88174677
+      std_loss:                                 0.0012130684
+  eval-batch-train:
+    avg_avg_episode_length:                     7.321185
+    avg_avg_return:                             0.9822584
+    avg_benchmark_regret:                       0.010540763
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20531464
+    proxy_corner:
+      avg_avg_return:                           0.008596225
+      avg_reward_per_step:                      0.0016403198
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.60129
+    avg_avg_return:                             0.9745617
+    avg_benchmark_regret:                       0.015678128
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14089203
+    proxy_corner:
+      avg_avg_return:                           0.9745617
+      avg_reward_per_step:                      0.14089203
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3157787
+    avg_avg_return:                             0.9819485
+    avg_benchmark_regret:                       0.010888901
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19396973
+    proxy_corner:
+      avg_avg_return:                           0.004787543
+      avg_reward_per_step:                      0.0012130737
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.24681
+    avg_avg_return:                             0.015269094
+    avg_benchmark_regret:                       0.9693661
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0063552856
+    proxy_corner:
+      avg_avg_return:                           0.015269094
+      avg_reward_per_step:                      0.0063552856
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        137363456
+    env-step-replay-before:                     69664768
+    env-step-used-before:                       137363456
+    ppo-update-before:                          83840
+    env-step-generate-before:                   67698688
+    env-step-all-after:                         137396224
+    env-step-replay-after:                      69697536
+    env-step-used-after:                        137396224
+    ppo-update-after:                           83860
+    env-step-generate-after:                    67698688
+  train-replay:
+    shift_proportion:                           0.953125
+    avg_avg_episode_length:                     17.430176
+    avg_avg_return:                             0.8612808
+    avg_reward_per_step:                        0.054107666
+    proxy_corner:
+      avg_avg_return:                           0.04951488
+      avg_reward_per_step:                      0.00289917
+  train-all:
+    shift_proportion:                           0.953125
+    avg_avg_episode_length:                     17.430176
+    avg_avg_return:                             0.8612808
+    avg_reward_per_step:                        0.054107666
+    proxy_corner:
+      avg_avg_return:                           0.04951488
+      avg_reward_per_step:                      0.00289917
+  perf:
+    env_steps_per_second:                       298012.36467898317
+    ppo_updates_per_second:                     42.134043824997455
+    eval-batch-train-duration:                  0.5250581307336688
+    eval-batch-orig-duration:                   0.48090658988803625
+    eval-batch-shift-duration:                  0.475246449932456
+    eval-batch-tree-duration:                   0.4949402129277587
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.880859
+      cheese-corne_dist_wavg:                   12.694169
+      levels_cheese_is_in_corner_avg:           0.06518555
+      mouse-cheese_dist_finite_avg:             16.503662
+      mouse-cheese_dist_finite_wavg:            10.654329
+      mouse-cheese_dist_solvable_avg:           16.503662
+      mouse-cheese_dist_solvable_wavg:          10.65433
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.123228
+    visit_patterns:
+      avg_first_visit_time:                     1385.323
+      avg_last_visit_time:                      2117.318
+      num_replay_batches:                       2127
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001788257
+    avg_actor_approxkl3:                        0.0012179165
+    avg_actor_clipfrac:                         0.04550171
+    avg_actor_loss:                             0.000631168
+    avg_advantage:                              0.0009289454
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            8.7096225e-05
+    avg_entropy:                                0.1297255
+    avg_grad_norm_pre_clip:                     2.6149805
+    avg_loss:                                   0.00054499065
+    max:
+      max_actor_approxkl1:                      0.0038542084
+      max_actor_approxkl3:                      0.0025807116
+      max_actor_clipfrac:                       0.07788086
+      max_actor_loss:                           0.003462118
+      max_advantage:                            0.0540573
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00012737662
+      max_entropy:                              0.15707701
+      max_grad_norm_pre_clip:                   5.0386124
+      max_loss:                                 0.0033850581
+    std:
+      std_actor_approxkl1:                      0.001075064
+      std_actor_approxkl3:                      0.00063327624
+      std_actor_clipfrac:                       0.018406887
+      std_actor_loss:                           0.0010086761
+      std_advantage:                            0.011932389
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.6199154e-05
+      std_entropy:                              0.0126845
+      std_grad_norm_pre_clip:                   1.0188147
+      std_loss:                                 0.0010073528
+  eval-batch-train:
+    avg_avg_episode_length:                     7.313959
+    avg_avg_return:                             0.9820097
+    avg_benchmark_regret:                       0.010789457
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20479584
+    proxy_corner:
+      avg_avg_return:                           0.008098023
+      avg_reward_per_step:                      0.0016098022
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.518929
+    avg_avg_return:                             0.97564286
+    avg_benchmark_regret:                       0.014596965
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14155579
+    proxy_corner:
+      avg_avg_return:                           0.97564286
+      avg_reward_per_step:                      0.14155579
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.2888527
+    avg_avg_return:                             0.9821465
+    avg_benchmark_regret:                       0.010690962
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19508362
+    proxy_corner:
+      avg_avg_return:                           0.004359465
+      avg_reward_per_step:                      0.0010986328
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.35689
+    avg_avg_return:                             0.014534583
+    avg_benchmark_regret:                       0.97010064
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.006034851
+    proxy_corner:
+      avg_avg_return:                           0.014534583
+      avg_reward_per_step:                      0.006034851
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        138412032
+    env-step-replay-before:                     70090752
+    env-step-used-before:                       138412032
+    ppo-update-before:                          84480
+    env-step-generate-before:                   68321280
+    env-step-all-after:                         138444800
+    env-step-replay-after:                      70123520
+    env-step-used-after:                        138444800
+    ppo-update-after:                           84500
+    env-step-generate-after:                    68321280
+  train-replay:
+    shift_proportion:                           0.94921875
+    avg_avg_episode_length:                     17.715755
+    avg_avg_return:                             0.85847056
+    avg_reward_per_step:                        0.054260254
+    proxy_corner:
+      avg_avg_return:                           0.05367022
+      avg_reward_per_step:                      0.0032653809
+  train-all:
+    shift_proportion:                           0.94921875
+    avg_avg_episode_length:                     17.715755
+    avg_avg_return:                             0.85847056
+    avg_reward_per_step:                        0.054260254
+    proxy_corner:
+      avg_avg_return:                           0.05367022
+      avg_reward_per_step:                      0.0032653809
+  perf:
+    env_steps_per_second:                       296537.3918860831
+    ppo_updates_per_second:                     42.11532532347939
+    eval-batch-train-duration:                  0.5362053113058209
+    eval-batch-orig-duration:                   0.48079582024365664
+    eval-batch-shift-duration:                  0.47780056204646826
+    eval-batch-tree-duration:                   0.4451049054041505
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.849121
+      cheese-corne_dist_wavg:                   12.688299
+      levels_cheese_is_in_corner_avg:           0.064453125
+      mouse-cheese_dist_finite_avg:             16.53711
+      mouse-cheese_dist_finite_wavg:            7.951135
+      mouse-cheese_dist_solvable_avg:           16.53711
+      mouse-cheese_dist_solvable_wavg:          7.951135
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12426703
+    visit_patterns:
+      avg_first_visit_time:                     1393.658
+      avg_last_visit_time:                      2130.3967
+      num_replay_batches:                       2140
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.004108695
+    avg_actor_approxkl3:                        0.002874143
+    avg_actor_clipfrac:                         0.070336916
+    avg_actor_loss:                             0.0024721234
+    avg_advantage:                              0.0024704295
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00022193951
+    avg_entropy:                                0.13730364
+    avg_grad_norm_pre_clip:                     3.3579738
+    avg_loss:                                   0.0024457895
+    max:
+      max_actor_approxkl1:                      0.008962133
+      max_actor_approxkl3:                      0.007935956
+      max_actor_clipfrac:                       0.13134766
+      max_actor_loss:                           0.008748461
+      max_advantage:                            0.057053983
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00071764644
+      max_entropy:                              0.17407626
+      max_grad_norm_pre_clip:                   7.3028708
+      max_loss:                                 0.008765385
+    std:
+      std_actor_approxkl1:                      0.0025294428
+      std_actor_approxkl3:                      0.0019089693
+      std_actor_clipfrac:                       0.028260441
+      std_actor_loss:                           0.0024663263
+      std_advantage:                            0.020874832
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0002773733
+      std_entropy:                              0.015746633
+      std_grad_norm_pre_clip:                   1.6075038
+      std_loss:                                 0.0024559288
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3184743
+    avg_avg_return:                             0.98182
+    avg_benchmark_regret:                       0.010979123
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20472717
+    proxy_corner:
+      avg_avg_return:                           0.008489204
+      avg_reward_per_step:                      0.0016326904
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.482885
+    avg_avg_return:                             0.9748413
+    avg_benchmark_regret:                       0.015398542
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14182281
+    proxy_corner:
+      avg_avg_return:                           0.9748413
+      avg_reward_per_step:                      0.14182281
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.365163
+    avg_avg_return:                             0.98162043
+    avg_benchmark_regret:                       0.0112170195
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19407654
+    proxy_corner:
+      avg_avg_return:                           0.00445258
+      avg_reward_per_step:                      0.0011444092
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.3842
+    avg_avg_return:                             0.013771897
+    avg_benchmark_regret:                       0.97086334
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.006248474
+    proxy_corner:
+      avg_avg_return:                           0.013771897
+      avg_reward_per_step:                      0.006248474
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        139460608
+    env-step-replay-before:                     70713344
+    env-step-used-before:                       139460608
+    ppo-update-before:                          85120
+    env-step-generate-before:                   68747264
+    env-step-all-after:                         139493376
+    env-step-replay-after:                      70713344
+    env-step-used-after:                        139493376
+    ppo-update-after:                           85140
+    env-step-generate-after:                    68780032
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.879652
+    avg_avg_return:                             0.94674057
+    avg_reward_per_step:                        0.1880188
+    proxy_corner:
+      avg_avg_return:                           0.014984194
+      avg_reward_per_step:                      0.0016479492
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.879652
+    avg_avg_return:                             0.94674057
+    avg_reward_per_step:                        0.1880188
+    proxy_corner:
+      avg_avg_return:                           0.014984194
+      avg_reward_per_step:                      0.0016479492
+  perf:
+    env_steps_per_second:                       745532.5432581233
+    ppo_updates_per_second:                     40.80187747141086
+    eval-batch-train-duration:                  0.3180874437093735
+    eval-batch-orig-duration:                   0.22409984935075045
+    eval-batch-shift-duration:                  0.2605914482846856
+    eval-batch-tree-duration:                   0.22487111389636993
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.889648
+      cheese-corne_dist_wavg:                   7.2761536
+      levels_cheese_is_in_corner_avg:           0.064208984
+      mouse-cheese_dist_finite_avg:             16.553223
+      mouse-cheese_dist_finite_wavg:            15.156285
+      mouse-cheese_dist_solvable_avg:           16.553223
+      mouse-cheese_dist_solvable_wavg:          15.156283
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12359009
+    visit_patterns:
+      avg_first_visit_time:                     1401.2048
+      avg_last_visit_time:                      2148.1746
+      num_replay_batches:                       2158
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0023785483
+    avg_actor_approxkl3:                        0.0017431593
+    avg_actor_clipfrac:                         0.04373169
+    avg_actor_loss:                             0.0010225056
+    avg_advantage:                              0.0039052137
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00012762779
+    avg_entropy:                                0.09558139
+    avg_grad_norm_pre_clip:                     1.9677066
+    avg_loss:                                   0.000990738
+    max:
+      max_actor_approxkl1:                      0.007026245
+      max_actor_approxkl3:                      0.004598405
+      max_actor_clipfrac:                       0.07543945
+      max_actor_loss:                           0.0046963855
+      max_advantage:                            0.06643534
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0001610077
+      max_entropy:                              0.11786942
+      max_grad_norm_pre_clip:                   4.082467
+      max_loss:                                 0.0046743
+    std:
+      std_actor_approxkl1:                      0.0018502994
+      std_actor_approxkl3:                      0.0012993781
+      std_actor_clipfrac:                       0.019047268
+      std_actor_loss:                           0.0011853914
+      std_advantage:                            0.015652478
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.6726734e-05
+      std_entropy:                              0.011331926
+      std_grad_norm_pre_clip:                   0.803948
+      std_loss:                                 0.0011856833
+  eval-batch-train:
+    avg_avg_episode_length:                     7.413987
+    avg_avg_return:                             0.981332
+    avg_benchmark_regret:                       0.011467169
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.203331
+    proxy_corner:
+      avg_avg_return:                           0.008957146
+      avg_reward_per_step:                      0.001663208
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.483451
+    avg_avg_return:                             0.974983
+    avg_benchmark_regret:                       0.015256813
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14186859
+    proxy_corner:
+      avg_avg_return:                           0.974983
+      avg_reward_per_step:                      0.14186859
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.361183
+    avg_avg_return:                             0.98129916
+    avg_benchmark_regret:                       0.011538295
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19392395
+    proxy_corner:
+      avg_avg_return:                           0.004704356
+      avg_reward_per_step:                      0.0011520386
+  eval-batch-tree:
+    avg_avg_episode_length:                     83.97127
+    avg_avg_return:                             0.021002572
+    avg_benchmark_regret:                       0.9636326
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0062713623
+    proxy_corner:
+      avg_avg_return:                           0.021002572
+      avg_reward_per_step:                      0.0062713623
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        140509184
+    env-step-replay-before:                     71237632
+    env-step-used-before:                       140509184
+    ppo-update-before:                          85760
+    env-step-generate-before:                   69271552
+    env-step-all-after:                         140541952
+    env-step-replay-after:                      71270400
+    env-step-used-after:                        140541952
+    ppo-update-after:                           85780
+    env-step-generate-after:                    69271552
+  train-replay:
+    shift_proportion:                           0.91796875
+    avg_avg_episode_length:                     17.220366
+    avg_avg_return:                             0.86451936
+    avg_reward_per_step:                        0.05496216
+    proxy_corner:
+      avg_avg_return:                           0.08475584
+      avg_reward_per_step:                      0.0052490234
+  train-all:
+    shift_proportion:                           0.91796875
+    avg_avg_episode_length:                     17.220366
+    avg_avg_return:                             0.86451936
+    avg_reward_per_step:                        0.05496216
+    proxy_corner:
+      avg_avg_return:                           0.08475584
+      avg_reward_per_step:                      0.0052490234
+  perf:
+    env_steps_per_second:                       325590.24025103793
+    ppo_updates_per_second:                     41.72195519643769
+    eval-batch-train-duration:                  0.5329299019649625
+    eval-batch-orig-duration:                   0.48078222293406725
+    eval-batch-shift-duration:                  0.4516191929578781
+    eval-batch-tree-duration:                   0.4857135023921728
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.855713
+      cheese-corne_dist_wavg:                   7.2824
+      levels_cheese_is_in_corner_avg:           0.064697266
+      mouse-cheese_dist_finite_avg:             16.597656
+      mouse-cheese_dist_finite_wavg:            16.959494
+      mouse-cheese_dist_solvable_avg:           16.597656
+      mouse-cheese_dist_solvable_wavg:          16.959496
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.123027734
+    visit_patterns:
+      avg_first_visit_time:                     1404.9614
+      avg_last_visit_time:                      2165.1858
+      num_replay_batches:                       2175
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0033968918
+    avg_actor_approxkl3:                        0.002276002
+    avg_actor_clipfrac:                         0.0670044
+    avg_actor_loss:                             0.0016968166
+    avg_advantage:                              -0.0030240044
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            6.038797e-05
+    avg_entropy:                                0.12835063
+    avg_grad_norm_pre_clip:                     2.7853305
+    avg_loss:                                   0.0015986599
+    max:
+      max_actor_approxkl1:                      0.0050822906
+      max_actor_approxkl3:                      0.0033357856
+      max_actor_clipfrac:                       0.107666016
+      max_actor_loss:                           0.0052534025
+      max_advantage:                            0.05459732
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          7.2079085e-05
+      max_entropy:                              0.16077101
+      max_grad_norm_pre_clip:                   5.6990805
+      max_loss:                                 0.005124766
+    std:
+      std_actor_approxkl1:                      0.0013483419
+      std_actor_approxkl3:                      0.0008446834
+      std_actor_clipfrac:                       0.020903232
+      std_actor_loss:                           0.0013656879
+      std_advantage:                            0.010915601
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.6440496e-06
+      std_entropy:                              0.009842353
+      std_grad_norm_pre_clip:                   0.9210603
+      std_loss:                                 0.001360488
+  eval-batch-train:
+    avg_avg_episode_length:                     7.327841
+    avg_avg_return:                             0.9823735
+    avg_benchmark_regret:                       0.010425705
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20394897
+    proxy_corner:
+      avg_avg_return:                           0.008256231
+      avg_reward_per_step:                      0.0016174316
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.501738
+    avg_avg_return:                             0.97538
+    avg_benchmark_regret:                       0.014859805
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14170837
+    proxy_corner:
+      avg_avg_return:                           0.97538
+      avg_reward_per_step:                      0.14170837
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.2641363
+    avg_avg_return:                             0.98227644
+    avg_benchmark_regret:                       0.010561025
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19509888
+    proxy_corner:
+      avg_avg_return:                           0.0040217526
+      avg_reward_per_step:                      0.0010223389
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.20039
+    avg_avg_return:                             0.018607214
+    avg_benchmark_regret:                       0.966028
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0063552856
+    proxy_corner:
+      avg_avg_return:                           0.018607214
+      avg_reward_per_step:                      0.0063552856
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        141557760
+    env-step-replay-before:                     71729152
+    env-step-used-before:                       141557760
+    ppo-update-before:                          86400
+    env-step-generate-before:                   69828608
+    env-step-all-after:                         141590528
+    env-step-replay-after:                      71761920
+    env-step-used-after:                        141590528
+    ppo-update-after:                           86420
+    env-step-generate-after:                    69828608
+  train-replay:
+    shift_proportion:                           0.91796875
+    avg_avg_episode_length:                     17.306786
+    avg_avg_return:                             0.86428046
+    avg_reward_per_step:                        0.05480957
+    proxy_corner:
+      avg_avg_return:                           0.069864735
+      avg_reward_per_step:                      0.004180908
+  train-all:
+    shift_proportion:                           0.91796875
+    avg_avg_episode_length:                     17.306786
+    avg_avg_return:                             0.86428046
+    avg_reward_per_step:                        0.05480957
+    proxy_corner:
+      avg_avg_return:                           0.069864735
+      avg_reward_per_step:                      0.004180908
+  perf:
+    env_steps_per_second:                       302155.2351862714
+    ppo_updates_per_second:                     42.58465459022238
+    eval-batch-train-duration:                  0.5125726368278265
+    eval-batch-orig-duration:                   0.46485996898263693
+    eval-batch-shift-duration:                  0.4495692839846015
+    eval-batch-tree-duration:                   0.488765224814415
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.910156
+      cheese-corne_dist_wavg:                   12.6878805
+      levels_cheese_is_in_corner_avg:           0.064697266
+      mouse-cheese_dist_finite_avg:             16.640625
+      mouse-cheese_dist_finite_wavg:            8.873314
+      mouse-cheese_dist_solvable_avg:           16.640625
+      mouse-cheese_dist_solvable_wavg:          8.873313
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12320474
+    visit_patterns:
+      avg_first_visit_time:                     1411.501
+      avg_last_visit_time:                      2180.303
+      num_replay_batches:                       2190
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014713284
+    avg_actor_approxkl3:                        0.0009799671
+    avg_actor_clipfrac:                         0.04038086
+    avg_actor_loss:                             0.00072984997
+    avg_advantage:                              -0.0039460342
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            6.614253e-05
+    avg_entropy:                                0.12398987
+    avg_grad_norm_pre_clip:                     2.324002
+    avg_loss:                                   0.00063893135
+    max:
+      max_actor_approxkl1:                      0.0030285465
+      max_actor_approxkl3:                      0.0016659058
+      max_actor_clipfrac:                       0.06652832
+      max_actor_loss:                           0.0021002837
+      max_advantage:                            0.05474955
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          7.4788084e-05
+      max_entropy:                              0.13982475
+      max_grad_norm_pre_clip:                   3.657175
+      max_loss:                                 0.0020062143
+    std:
+      std_actor_approxkl1:                      0.0008575192
+      std_actor_approxkl3:                      0.0004430173
+      std_actor_clipfrac:                       0.014654671
+      std_actor_loss:                           0.00055435987
+      std_advantage:                            0.011366198
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.6067225e-06
+      std_entropy:                              0.00642187
+      std_grad_norm_pre_clip:                   0.6308273
+      std_loss:                                 0.000553603
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3777084
+    avg_avg_return:                             0.9815351
+    avg_benchmark_regret:                       0.011264114
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20235443
+    proxy_corner:
+      avg_avg_return:                           0.0077291806
+      avg_reward_per_step:                      0.0015792847
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.552698
+    avg_avg_return:                             0.97495055
+    avg_benchmark_regret:                       0.015289216
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14109802
+    proxy_corner:
+      avg_avg_return:                           0.97495055
+      avg_reward_per_step:                      0.14109802
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.313337
+    avg_avg_return:                             0.9816607
+    avg_benchmark_regret:                       0.0111767035
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19390106
+    proxy_corner:
+      avg_avg_return:                           0.004013637
+      avg_reward_per_step:                      0.0010147095
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.27286
+    avg_avg_return:                             0.016994204
+    avg_benchmark_regret:                       0.967641
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0061569214
+    proxy_corner:
+      avg_avg_return:                           0.016994204
+      avg_reward_per_step:                      0.0061569214
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        142606336
+    env-step-replay-before:                     72286208
+    env-step-used-before:                       142606336
+    ppo-update-before:                          87040
+    env-step-generate-before:                   70320128
+    env-step-all-after:                         142639104
+    env-step-replay-after:                      72318976
+    env-step-used-after:                        142639104
+    ppo-update-after:                           87060
+    env-step-generate-after:                    70320128
+  train-replay:
+    shift_proportion:                           0.9453125
+    avg_avg_episode_length:                     17.308723
+    avg_avg_return:                             0.86016476
+    avg_reward_per_step:                        0.05429077
+    proxy_corner:
+      avg_avg_return:                           0.05746541
+      avg_reward_per_step:                      0.0033874512
+  train-all:
+    shift_proportion:                           0.9453125
+    avg_avg_episode_length:                     17.308723
+    avg_avg_return:                             0.86016476
+    avg_reward_per_step:                        0.05429077
+    proxy_corner:
+      avg_avg_return:                           0.05746541
+      avg_reward_per_step:                      0.0033874512
+  perf:
+    env_steps_per_second:                       320691.81133578694
+    ppo_updates_per_second:                     42.17394506748705
+    eval-batch-train-duration:                  0.5398888206109405
+    eval-batch-orig-duration:                   0.4835264254361391
+    eval-batch-shift-duration:                  0.4851407930254936
+    eval-batch-tree-duration:                   0.48900154512375593
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.895996
+      cheese-corne_dist_wavg:                   14.502667
+      levels_cheese_is_in_corner_avg:           0.06616211
+      mouse-cheese_dist_finite_avg:             16.666748
+      mouse-cheese_dist_finite_wavg:            11.564883
+      mouse-cheese_dist_solvable_avg:           16.666748
+      mouse-cheese_dist_solvable_wavg:          11.564886
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12312964
+    visit_patterns:
+      avg_first_visit_time:                     1420.081
+      avg_last_visit_time:                      2197.337
+      num_replay_batches:                       2207
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0035633096
+    avg_actor_approxkl3:                        0.0031523013
+    avg_actor_clipfrac:                         0.08226319
+    avg_actor_loss:                             0.0039952225
+    avg_advantage:                              -0.003498536
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            6.298643e-05
+    avg_entropy:                                0.13556264
+    avg_grad_norm_pre_clip:                     4.3410716
+    avg_loss:                                   0.0038911533
+    max:
+      max_actor_approxkl1:                      0.008270496
+      max_actor_approxkl3:                      0.006757955
+      max_actor_clipfrac:                       0.1184082
+      max_actor_loss:                           0.010319144
+      max_advantage:                            0.05744064
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          7.8941135e-05
+      max_entropy:                              0.15143791
+      max_grad_norm_pre_clip:                   7.521673
+      max_loss:                                 0.010230546
+    std:
+      std_actor_approxkl1:                      0.0020505518
+      std_actor_approxkl3:                      0.0016966837
+      std_actor_clipfrac:                       0.030053178
+      std_actor_loss:                           0.002768865
+      std_advantage:                            0.0108192125
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.86698e-06
+      std_entropy:                              0.010477753
+      std_grad_norm_pre_clip:                   1.5364769
+      std_loss:                                 0.00276963
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3539767
+    avg_avg_return:                             0.9816667
+    avg_benchmark_regret:                       0.011132423
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20317078
+    proxy_corner:
+      avg_avg_return:                           0.009452613
+      avg_reward_per_step:                      0.0016937256
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.578621
+    avg_avg_return:                             0.9743223
+    avg_benchmark_regret:                       0.015917499
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14109802
+    proxy_corner:
+      avg_avg_return:                           0.9743223
+      avg_reward_per_step:                      0.14109802
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3025665
+    avg_avg_return:                             0.9821558
+    avg_benchmark_regret:                       0.010681652
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.1942215
+    proxy_corner:
+      avg_avg_return:                           0.0040548416
+      avg_reward_per_step:                      0.0010375977
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.212555
+    avg_avg_return:                             0.0161167
+    avg_benchmark_regret:                       0.9685185
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.006126404
+    proxy_corner:
+      avg_avg_return:                           0.0161167
+      avg_reward_per_step:                      0.006126404
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        143654912
+    env-step-replay-before:                     72941568
+    env-step-used-before:                       143654912
+    ppo-update-before:                          87680
+    env-step-generate-before:                   70713344
+    env-step-all-after:                         143687680
+    env-step-replay-after:                      72974336
+    env-step-used-after:                        143687680
+    ppo-update-after:                           87700
+    env-step-generate-after:                    70713344
+  train-replay:
+    shift_proportion:                           0.92578125
+    avg_avg_episode_length:                     17.391941
+    avg_avg_return:                             0.85915416
+    avg_reward_per_step:                        0.05429077
+    proxy_corner:
+      avg_avg_return:                           0.06534745
+      avg_reward_per_step:                      0.0039978027
+  train-all:
+    shift_proportion:                           0.92578125
+    avg_avg_episode_length:                     17.391941
+    avg_avg_return:                             0.85915416
+    avg_reward_per_step:                        0.05429077
+    proxy_corner:
+      avg_avg_return:                           0.06534745
+      avg_reward_per_step:                      0.0039978027
+  perf:
+    env_steps_per_second:                       259577.23817676512
+    ppo_updates_per_second:                     42.06798940546203
+    eval-batch-train-duration:                  0.5424328716471791
+    eval-batch-orig-duration:                   0.48518211022019386
+    eval-batch-shift-duration:                  0.47058485727757215
+    eval-batch-tree-duration:                   0.43549239449203014
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.855469
+      cheese-corne_dist_wavg:                   16.283611
+      levels_cheese_is_in_corner_avg:           0.06738281
+      mouse-cheese_dist_finite_avg:             16.70288
+      mouse-cheese_dist_finite_wavg:            15.169529
+      mouse-cheese_dist_solvable_avg:           16.70288
+      mouse-cheese_dist_solvable_wavg:          15.169531
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12368304
+    visit_patterns:
+      avg_first_visit_time:                     1425.0732
+      avg_last_visit_time:                      2217.2188
+      num_replay_batches:                       2227
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002431344
+    avg_actor_approxkl3:                        0.0015592468
+    avg_actor_clipfrac:                         0.052801516
+    avg_actor_loss:                             0.0010164379
+    avg_advantage:                              -0.00077141623
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            6.365072e-05
+    avg_entropy:                                0.12477249
+    avg_grad_norm_pre_clip:                     2.8543239
+    avg_loss:                                   0.0009234908
+    max:
+      max_actor_approxkl1:                      0.0064542107
+      max_actor_approxkl3:                      0.0037230686
+      max_actor_clipfrac:                       0.083984375
+      max_actor_loss:                           0.0031450735
+      max_advantage:                            0.06850326
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          7.974673e-05
+      max_entropy:                              0.13547
+      max_grad_norm_pre_clip:                   4.8233547
+      max_loss:                                 0.003046234
+    std:
+      std_actor_approxkl1:                      0.0013521072
+      std_actor_approxkl3:                      0.00077850104
+      std_actor_clipfrac:                       0.019765763
+      std_actor_loss:                           0.001068574
+      std_advantage:                            0.010722881
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          6.2831236e-06
+      std_entropy:                              0.0054371892
+      std_grad_norm_pre_clip:                   0.874757
+      std_loss:                                 0.0010676415
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3701787
+    avg_avg_return:                             0.98192173
+    avg_benchmark_regret:                       0.010877411
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20301056
+    proxy_corner:
+      avg_avg_return:                           0.007892338
+      avg_reward_per_step:                      0.0015945435
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.4993515
+    avg_avg_return:                             0.9753784
+    avg_benchmark_regret:                       0.014861395
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14157104
+    proxy_corner:
+      avg_avg_return:                           0.9753784
+      avg_reward_per_step:                      0.14157104
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.2840657
+    avg_avg_return:                             0.9821004
+    avg_benchmark_regret:                       0.010737026
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19446564
+    proxy_corner:
+      avg_avg_return:                           0.00423654
+      avg_reward_per_step:                      0.0010604858
+  eval-batch-tree:
+    avg_avg_episode_length:                     84.035645
+    avg_avg_return:                             0.019116139
+    avg_benchmark_regret:                       0.9655191
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0071105957
+    proxy_corner:
+      avg_avg_return:                           0.019116139
+      avg_reward_per_step:                      0.0071105957
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        144703488
+    env-step-replay-before:                     73531392
+    env-step-used-before:                       144703488
+    ppo-update-before:                          88320
+    env-step-generate-before:                   71172096
+    env-step-all-after:                         144736256
+    env-step-replay-after:                      73564160
+    env-step-used-after:                        144736256
+    ppo-update-after:                           88340
+    env-step-generate-after:                    71172096
+  train-replay:
+    shift_proportion:                           0.9375
+    avg_avg_episode_length:                     17.330534
+    avg_avg_return:                             0.86132586
+    avg_reward_per_step:                        0.05432129
+    proxy_corner:
+      avg_avg_return:                           0.058880106
+      avg_reward_per_step:                      0.0036315918
+  train-all:
+    shift_proportion:                           0.9375
+    avg_avg_episode_length:                     17.330534
+    avg_avg_return:                             0.86132586
+    avg_reward_per_step:                        0.05432129
+    proxy_corner:
+      avg_avg_return:                           0.058880106
+      avg_reward_per_step:                      0.0036315918
+  perf:
+    env_steps_per_second:                       327028.1450252615
+    ppo_updates_per_second:                     40.755610482833525
+    eval-batch-train-duration:                  0.5402250615879893
+    eval-batch-orig-duration:                   0.5054699694737792
+    eval-batch-shift-duration:                  0.4826575582846999
+    eval-batch-tree-duration:                   0.4816164830699563
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.878174
+      cheese-corne_dist_wavg:                   12.685519
+      levels_cheese_is_in_corner_avg:           0.06762695
+      mouse-cheese_dist_finite_avg:             16.725586
+      mouse-cheese_dist_finite_wavg:            9.777572
+      mouse-cheese_dist_solvable_avg:           16.725586
+      mouse-cheese_dist_solvable_wavg:          9.777574
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.123861924
+    visit_patterns:
+      avg_first_visit_time:                     1430.6372
+      avg_last_visit_time:                      2235.2024
+      num_replay_batches:                       2245
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0019655877
+    avg_actor_approxkl3:                        0.001341456
+    avg_actor_clipfrac:                         0.04597168
+    avg_actor_loss:                             0.00017708128
+    avg_advantage:                              -0.0004995792
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            5.983066e-05
+    avg_entropy:                                0.11915363
+    avg_grad_norm_pre_clip:                     1.8433198
+    avg_loss:                                   8.784297e-05
+    max:
+      max_actor_approxkl1:                      0.0041067535
+      max_actor_approxkl3:                      0.0027754433
+      max_actor_clipfrac:                       0.067993164
+      max_actor_loss:                           0.002287291
+      max_advantage:                            0.061090887
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          6.7651185e-05
+      max_entropy:                              0.12832406
+      max_grad_norm_pre_clip:                   2.5738342
+      max_loss:                                 0.0021961525
+    std:
+      std_actor_approxkl1:                      0.0010791013
+      std_actor_approxkl3:                      0.0006851739
+      std_actor_clipfrac:                       0.017252142
+      std_actor_loss:                           0.00072823255
+      std_advantage:                            0.011001463
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.2645648e-06
+      std_entropy:                              0.005065623
+      std_grad_norm_pre_clip:                   0.4455598
+      std_loss:                                 0.0007281212
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3313155
+    avg_avg_return:                             0.9818337
+    avg_benchmark_regret:                       0.010965439
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.2032013
+    proxy_corner:
+      avg_avg_return:                           0.0076272064
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.5117445
+    avg_avg_return:                             0.9752482
+    avg_benchmark_regret:                       0.014991622
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14139557
+    proxy_corner:
+      avg_avg_return:                           0.9752482
+      avg_reward_per_step:                      0.14139557
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.2790365
+    avg_avg_return:                             0.98173
+    avg_benchmark_regret:                       0.011107444
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19444275
+    proxy_corner:
+      avg_avg_return:                           0.0046042004
+      avg_reward_per_step:                      0.0011901855
+  eval-batch-tree:
+    avg_avg_episode_length:                     83.824524
+    avg_avg_return:                             0.021753302
+    avg_benchmark_regret:                       0.96288186
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0078048706
+    proxy_corner:
+      avg_avg_return:                           0.021753302
+      avg_reward_per_step:                      0.0078048706
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        145752064
+    env-step-replay-before:                     74088448
+    env-step-used-before:                       145752064
+    ppo-update-before:                          88960
+    env-step-generate-before:                   71663616
+    env-step-all-after:                         145784832
+    env-step-replay-after:                      74121216
+    env-step-used-after:                        145784832
+    ppo-update-after:                           88980
+    env-step-generate-after:                    71663616
+  train-replay:
+    shift_proportion:                           0.9296875
+    avg_avg_episode_length:                     17.356289
+    avg_avg_return:                             0.8622858
+    avg_reward_per_step:                        0.054138184
+    proxy_corner:
+      avg_avg_return:                           0.07416559
+      avg_reward_per_step:                      0.0044555664
+  train-all:
+    shift_proportion:                           0.9296875
+    avg_avg_episode_length:                     17.356289
+    avg_avg_return:                             0.8622858
+    avg_reward_per_step:                        0.054138184
+    proxy_corner:
+      avg_avg_return:                           0.07416559
+      avg_reward_per_step:                      0.0044555664
+  perf:
+    env_steps_per_second:                       355382.69999910507
+    ppo_updates_per_second:                     42.41845016538334
+    eval-batch-train-duration:                  0.5106826163828373
+    eval-batch-orig-duration:                   0.47893097065389156
+    eval-batch-shift-duration:                  0.48804657254368067
+    eval-batch-tree-duration:                   0.4946923302486539
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.8515625
+      cheese-corne_dist_wavg:                   9.087035
+      levels_cheese_is_in_corner_avg:           0.06738281
+      mouse-cheese_dist_finite_avg:             16.73584
+      mouse-cheese_dist_finite_wavg:            14.27353
+      mouse-cheese_dist_solvable_avg:           16.73584
+      mouse-cheese_dist_solvable_wavg:          14.273534
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.123556346
+    visit_patterns:
+      avg_first_visit_time:                     1437.2927
+      avg_last_visit_time:                      2252.387
+      num_replay_batches:                       2262
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0020145748
+    avg_actor_approxkl3:                        0.0016443313
+    avg_actor_clipfrac:                         0.053448487
+    avg_actor_loss:                             0.0013953578
+    avg_advantage:                              0.0030367174
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            6.15393e-05
+    avg_entropy:                                0.119005166
+    avg_grad_norm_pre_clip:                     2.9847646
+    avg_loss:                                   0.0013071221
+    max:
+      max_actor_approxkl1:                      0.0054908926
+      max_actor_approxkl3:                      0.00409481
+      max_actor_clipfrac:                       0.095214844
+      max_actor_loss:                           0.0034117466
+      max_advantage:                            0.07198024
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          6.8947265e-05
+      max_entropy:                              0.13118969
+      max_grad_norm_pre_clip:                   4.7559752
+      max_loss:                                 0.0033283723
+    std:
+      std_actor_approxkl1:                      0.0016882264
+      std_actor_approxkl3:                      0.0010214405
+      std_actor_clipfrac:                       0.020164201
+      std_actor_loss:                           0.001056302
+      std_advantage:                            0.011061949
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.7632653e-06
+      std_entropy:                              0.006453287
+      std_grad_norm_pre_clip:                   1.0009047
+      std_loss:                                 0.0010537651
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3081446
+    avg_avg_return:                             0.9825399
+    avg_benchmark_regret:                       0.010259256
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20433807
+    proxy_corner:
+      avg_avg_return:                           0.008704703
+      avg_reward_per_step:                      0.0016479492
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.527513
+    avg_avg_return:                             0.97500193
+    avg_benchmark_regret:                       0.015237957
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14149475
+    proxy_corner:
+      avg_avg_return:                           0.97500193
+      avg_reward_per_step:                      0.14149475
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.328192
+    avg_avg_return:                             0.98188204
+    avg_benchmark_regret:                       0.010955406
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19364929
+    proxy_corner:
+      avg_avg_return:                           0.004568044
+      avg_reward_per_step:                      0.001159668
+  eval-batch-tree:
+    avg_avg_episode_length:                     83.84921
+    avg_avg_return:                             0.024178267
+    avg_benchmark_regret:                       0.96045697
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0079422
+    proxy_corner:
+      avg_avg_return:                           0.024178267
+      avg_reward_per_step:                      0.0079422
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        146800640
+    env-step-replay-before:                     74711040
+    env-step-used-before:                       146800640
+    ppo-update-before:                          89600
+    env-step-generate-before:                   72089600
+    env-step-all-after:                         146833408
+    env-step-replay-after:                      74743808
+    env-step-used-after:                        146833408
+    ppo-update-after:                           89620
+    env-step-generate-after:                    72089600
+  train-replay:
+    shift_proportion:                           0.94140625
+    avg_avg_episode_length:                     17.501593
+    avg_avg_return:                             0.8600309
+    avg_reward_per_step:                        0.054016113
+    proxy_corner:
+      avg_avg_return:                           0.05265696
+      avg_reward_per_step:                      0.0031433105
+  train-all:
+    shift_proportion:                           0.94140625
+    avg_avg_episode_length:                     17.501593
+    avg_avg_return:                             0.8600309
+    avg_reward_per_step:                        0.054016113
+    proxy_corner:
+      avg_avg_return:                           0.05265696
+      avg_reward_per_step:                      0.0031433105
+  perf:
+    env_steps_per_second:                       320153.14420504554
+    ppo_updates_per_second:                     42.07753294451973
+    eval-batch-train-duration:                  0.5484686875715852
+    eval-batch-orig-duration:                   0.4838695302605629
+    eval-batch-shift-duration:                  0.4629313014447689
+    eval-batch-tree-duration:                   0.4553796639665961
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.805176
+      cheese-corne_dist_wavg:                   9.975977
+      levels_cheese_is_in_corner_avg:           0.068359375
+      mouse-cheese_dist_finite_avg:             16.742432
+      mouse-cheese_dist_finite_wavg:            16.076328
+      mouse-cheese_dist_solvable_avg:           16.742432
+      mouse-cheese_dist_solvable_wavg:          16.07633
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1241189
+    visit_patterns:
+      avg_first_visit_time:                     1444.2148
+      avg_last_visit_time:                      2271.2903
+      num_replay_batches:                       2281
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002193486
+    avg_actor_approxkl3:                        0.0015985022
+    avg_actor_clipfrac:                         0.04628296
+    avg_actor_loss:                             0.00095962285
+    avg_advantage:                              -0.0025229184
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            6.0052844e-05
+    avg_entropy:                                0.11685385
+    avg_grad_norm_pre_clip:                     2.3597667
+    avg_loss:                                   0.0008727954
+    max:
+      max_actor_approxkl1:                      0.0038120942
+      max_actor_approxkl3:                      0.0033914454
+      max_actor_clipfrac:                       0.07751465
+      max_actor_loss:                           0.0033097756
+      max_advantage:                            0.05555141
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          6.783512e-05
+      max_entropy:                              0.13007703
+      max_grad_norm_pre_clip:                   4.357381
+      max_loss:                                 0.0032158692
+    std:
+      std_actor_approxkl1:                      0.001362134
+      std_actor_approxkl3:                      0.0009983181
+      std_actor_clipfrac:                       0.021791218
+      std_actor_loss:                           0.0012918025
+      std_advantage:                            0.010914035
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.9307197e-06
+      std_entropy:                              0.007238664
+      std_grad_norm_pre_clip:                   1.0084273
+      std_loss:                                 0.0012904985
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3331275
+    avg_avg_return:                             0.98146117
+    avg_benchmark_regret:                       0.011337995
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20352936
+    proxy_corner:
+      avg_avg_return:                           0.008204412
+      avg_reward_per_step:                      0.0016174316
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.495918
+    avg_avg_return:                             0.97493756
+    avg_benchmark_regret:                       0.015302267
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14179993
+    proxy_corner:
+      avg_avg_return:                           0.97493756
+      avg_reward_per_step:                      0.14179993
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.330472
+    avg_avg_return:                             0.9817537
+    avg_benchmark_regret:                       0.011083767
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19454193
+    proxy_corner:
+      avg_avg_return:                           0.006107228
+      avg_reward_per_step:                      0.0016708374
+  eval-batch-tree:
+    avg_avg_episode_length:                     83.96402
+    avg_avg_return:                             0.022016037
+    avg_benchmark_regret:                       0.9626191
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.007293701
+    proxy_corner:
+      avg_avg_return:                           0.022016037
+      avg_reward_per_step:                      0.007293701
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        147849216
+    env-step-replay-before:                     75235328
+    env-step-used-before:                       147849216
+    ppo-update-before:                          90240
+    env-step-generate-before:                   72613888
+    env-step-all-after:                         147881984
+    env-step-replay-after:                      75235328
+    env-step-used-after:                        147881984
+    ppo-update-after:                           90260
+    env-step-generate-after:                    72646656
+  train-generate:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     7.5268135
+    avg_avg_return:                             0.94764
+    avg_reward_per_step:                        0.19558716
+    proxy_corner:
+      avg_avg_return:                           0.023542644
+      avg_reward_per_step:                      0.0026550293
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     7.5268135
+    avg_avg_return:                             0.94764
+    avg_reward_per_step:                        0.19558716
+    proxy_corner:
+      avg_avg_return:                           0.023542644
+      avg_reward_per_step:                      0.0026550293
+  perf:
+    env_steps_per_second:                       326723.24784910167
+    ppo_updates_per_second:                     41.88111950100931
+    eval-batch-train-duration:                  0.5315832402557135
+    eval-batch-orig-duration:                   0.47137515153735876
+    eval-batch-shift-duration:                  0.45047799218446016
+    eval-batch-tree-duration:                   0.48913529235869646
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.799072
+      cheese-corne_dist_wavg:                   9.975565
+      levels_cheese_is_in_corner_avg:           0.068359375
+      mouse-cheese_dist_finite_avg:             16.76294
+      mouse-cheese_dist_finite_wavg:            17.875023
+      mouse-cheese_dist_solvable_avg:           16.76294
+      mouse-cheese_dist_solvable_wavg:          17.875025
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12451587
+    visit_patterns:
+      avg_first_visit_time:                     1450.418
+      avg_last_visit_time:                      2286.3296
+      num_replay_batches:                       2296
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0022187952
+    avg_actor_approxkl3:                        0.001560085
+    avg_actor_clipfrac:                         0.039837647
+    avg_actor_loss:                             0.0014451446
+    avg_advantage:                              0.0032928614
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00011676922
+    avg_entropy:                                0.08826079
+    avg_grad_norm_pre_clip:                     2.6223295
+    avg_loss:                                   0.0014152686
+    max:
+      max_actor_approxkl1:                      0.006735493
+      max_actor_approxkl3:                      0.0041411463
+      max_actor_clipfrac:                       0.0703125
+      max_actor_loss:                           0.0037838127
+      max_advantage:                            0.06548536
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00013911912
+      max_entropy:                              0.11099814
+      max_grad_norm_pre_clip:                   3.850971
+      max_loss:                                 0.0037353535
+    std:
+      std_actor_approxkl1:                      0.0017187176
+      std_actor_approxkl3:                      0.0010851406
+      std_actor_clipfrac:                       0.018352177
+      std_actor_loss:                           0.0010975716
+      std_advantage:                            0.014725957
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.2092971e-05
+      std_entropy:                              0.011253002
+      std_grad_norm_pre_clip:                   0.60883886
+      std_loss:                                 0.0010940181
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3425274
+    avg_avg_return:                             0.981876
+    avg_benchmark_regret:                       0.010923181
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20320892
+    proxy_corner:
+      avg_avg_return:                           0.00818831
+      avg_reward_per_step:                      0.001625061
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.472597
+    avg_avg_return:                             0.9751484
+    avg_benchmark_regret:                       0.015091445
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14198303
+    proxy_corner:
+      avg_avg_return:                           0.9751484
+      avg_reward_per_step:                      0.14198303
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.309286
+    avg_avg_return:                             0.9819164
+    avg_benchmark_regret:                       0.010921029
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19476318
+    proxy_corner:
+      avg_avg_return:                           0.005970368
+      avg_reward_per_step:                      0.0016708374
+  eval-batch-tree:
+    avg_avg_episode_length:                     83.5896
+    avg_avg_return:                             0.02599147
+    avg_benchmark_regret:                       0.9586437
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.00856781
+    proxy_corner:
+      avg_avg_return:                           0.02599147
+      avg_reward_per_step:                      0.00856781
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        148897792
+    env-step-replay-before:                     75694080
+    env-step-used-before:                       148897792
+    ppo-update-before:                          90880
+    env-step-generate-before:                   73203712
+    env-step-all-after:                         148930560
+    env-step-replay-after:                      75726848
+    env-step-used-after:                        148930560
+    ppo-update-after:                           90900
+    env-step-generate-after:                    73203712
+  train-replay:
+    shift_proportion:                           0.9296875
+    avg_avg_episode_length:                     17.714333
+    avg_avg_return:                             0.85720426
+    avg_reward_per_step:                        0.05316162
+    proxy_corner:
+      avg_avg_return:                           0.06274861
+      avg_reward_per_step:                      0.003692627
+  train-all:
+    shift_proportion:                           0.9296875
+    avg_avg_episode_length:                     17.714333
+    avg_avg_return:                             0.85720426
+    avg_reward_per_step:                        0.05316162
+    proxy_corner:
+      avg_avg_return:                           0.06274861
+      avg_reward_per_step:                      0.003692627
+  perf:
+    env_steps_per_second:                       236293.79895599594
+    ppo_updates_per_second:                     41.3298891381322
+    eval-batch-train-duration:                  0.5338910697028041
+    eval-batch-orig-duration:                   0.4806630341336131
+    eval-batch-shift-duration:                  0.4809887604787946
+    eval-batch-tree-duration:                   0.4496001908555627
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.762451
+      cheese-corne_dist_wavg:                   7.288629
+      levels_cheese_is_in_corner_avg:           0.068359375
+      mouse-cheese_dist_finite_avg:             16.7854
+      mouse-cheese_dist_finite_wavg:            16.973291
+      mouse-cheese_dist_solvable_avg:           16.7854
+      mouse-cheese_dist_solvable_wavg:          16.973291
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12531942
+    visit_patterns:
+      avg_first_visit_time:                     1458.1057
+      avg_last_visit_time:                      2301.3508
+      num_replay_batches:                       2311
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0026262773
+    avg_actor_approxkl3:                        0.00192992
+    avg_actor_clipfrac:                         0.055133056
+    avg_actor_loss:                             0.00028504827
+    avg_advantage:                              -0.0035113022
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            6.116592e-05
+    avg_entropy:                                0.12767716
+    avg_grad_norm_pre_clip:                     1.98088
+    avg_loss:                                   0.00018795408
+    max:
+      max_actor_approxkl1:                      0.0054157334
+      max_actor_approxkl3:                      0.0043186913
+      max_actor_clipfrac:                       0.09338379
+      max_actor_loss:                           0.0027229143
+      max_advantage:                            0.048819363
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          7.399544e-05
+      max_entropy:                              0.15010396
+      max_grad_norm_pre_clip:                   3.2316887
+      max_loss:                                 0.002630955
+    std:
+      std_actor_approxkl1:                      0.0015209941
+      std_actor_approxkl3:                      0.0010763389
+      std_actor_clipfrac:                       0.02270361
+      std_actor_loss:                           0.0011608269
+      std_advantage:                            0.010747092
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.1415873e-06
+      std_entropy:                              0.009815799
+      std_grad_norm_pre_clip:                   0.6360296
+      std_loss:                                 0.0011631097
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4081626
+    avg_avg_return:                             0.981418
+    avg_benchmark_regret:                       0.0113811735
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20127106
+    proxy_corner:
+      avg_avg_return:                           0.007693914
+      avg_reward_per_step:                      0.0015792847
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.638691
+    avg_avg_return:                             0.9743936
+    avg_benchmark_regret:                       0.015846236
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14073181
+    proxy_corner:
+      avg_avg_return:                           0.9743936
+      avg_reward_per_step:                      0.14073181
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.4621277
+    avg_avg_return:                             0.9808657
+    avg_benchmark_regret:                       0.011971766
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19236755
+    proxy_corner:
+      avg_avg_return:                           0.004811602
+      avg_reward_per_step:                      0.0012283325
+  eval-batch-tree:
+    avg_avg_episode_length:                     83.13571
+    avg_avg_return:                             0.02841757
+    avg_benchmark_regret:                       0.95621765
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.010353088
+    proxy_corner:
+      avg_avg_return:                           0.02841757
+      avg_reward_per_step:                      0.010353088
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        149946368
+    env-step-replay-before:                     76152832
+    env-step-used-before:                       149946368
+    ppo-update-before:                          91520
+    env-step-generate-before:                   73793536
+    env-step-all-after:                         149979136
+    env-step-replay-after:                      76152832
+    env-step-used-after:                        149979136
+    ppo-update-after:                           91540
+    env-step-generate-after:                    73826304
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.4953785
+    avg_avg_return:                             0.9504849
+    avg_reward_per_step:                        0.2012024
+    proxy_corner:
+      avg_avg_return:                           0.017724194
+      avg_reward_per_step:                      0.0018005371
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.4953785
+    avg_avg_return:                             0.9504849
+    avg_reward_per_step:                        0.2012024
+    proxy_corner:
+      avg_avg_return:                           0.017724194
+      avg_reward_per_step:                      0.0018005371
+  perf:
+    env_steps_per_second:                       351628.5804576528
+    ppo_updates_per_second:                     40.471367795988435
+    eval-batch-train-duration:                  0.5323088364675641
+    eval-batch-orig-duration:                   0.48112067859619856
+    eval-batch-shift-duration:                  0.48699628934264183
+    eval-batch-tree-duration:                   0.49444082379341125
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.725098
+      cheese-corne_dist_wavg:                   9.068371
+      levels_cheese_is_in_corner_avg:           0.068847656
+      mouse-cheese_dist_finite_avg:             16.80127
+      mouse-cheese_dist_finite_wavg:            16.982546
+      mouse-cheese_dist_solvable_avg:           16.80127
+      mouse-cheese_dist_solvable_wavg:          16.98255
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12518778
+    visit_patterns:
+      avg_first_visit_time:                     1467.3574
+      avg_last_visit_time:                      2314.2632
+      num_replay_batches:                       2324
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0026001765
+    avg_actor_approxkl3:                        0.0022476006
+    avg_actor_clipfrac:                         0.05241699
+    avg_actor_loss:                             0.0021725087
+    avg_advantage:                              0.006281668
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00012507335
+    avg_entropy:                                0.09976641
+    avg_grad_norm_pre_clip:                     2.4760835
+    avg_loss:                                   0.0021352787
+    max:
+      max_actor_approxkl1:                      0.007374423
+      max_actor_approxkl3:                      0.005352855
+      max_actor_clipfrac:                       0.083984375
+      max_actor_loss:                           0.0067825373
+      max_advantage:                            0.068127096
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00014767614
+      max_entropy:                              0.12558502
+      max_grad_norm_pre_clip:                   3.9971175
+      max_loss:                                 0.006752936
+    std:
+      std_actor_approxkl1:                      0.0019741822
+      std_actor_approxkl3:                      0.0015158989
+      std_actor_clipfrac:                       0.020113632
+      std_actor_loss:                           0.0017080251
+      std_advantage:                            0.01511106
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.1548993e-05
+      std_entropy:                              0.011272813
+      std_grad_norm_pre_clip:                   0.6994521
+      std_loss:                                 0.0017036978
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3664403
+    avg_avg_return:                             0.98196083
+    avg_benchmark_regret:                       0.010838326
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20261383
+    proxy_corner:
+      avg_avg_return:                           0.0077265627
+      avg_reward_per_step:                      0.0015792847
+  eval-batch-orig:
+    avg_avg_episode_length:                     11.020128
+    avg_avg_return:                             0.96969867
+    avg_benchmark_regret:                       0.020541145
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1359787
+    proxy_corner:
+      avg_avg_return:                           0.96969867
+      avg_reward_per_step:                      0.1359787
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3406963
+    avg_avg_return:                             0.9816072
+    avg_benchmark_regret:                       0.011230234
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19387817
+    proxy_corner:
+      avg_avg_return:                           0.0044758315
+      avg_reward_per_step:                      0.0011672974
+  eval-batch-tree:
+    avg_avg_episode_length:                     83.05325
+    avg_avg_return:                             0.02978693
+    avg_benchmark_regret:                       0.9548483
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.011550903
+    proxy_corner:
+      avg_avg_return:                           0.02978693
+      avg_reward_per_step:                      0.011550903
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        150994944
+    env-step-replay-before:                     76709888
+    env-step-used-before:                       150994944
+    ppo-update-before:                          92160
+    env-step-generate-before:                   74285056
+    env-step-all-after:                         151027712
+    env-step-replay-after:                      76742656
+    env-step-used-after:                        151027712
+    ppo-update-after:                           92180
+    env-step-generate-after:                    74285056
+  train-replay:
+    shift_proportion:                           0.89453125
+    avg_avg_episode_length:                     17.408958
+    avg_avg_return:                             0.85894644
+    avg_reward_per_step:                        0.053863525
+    proxy_corner:
+      avg_avg_return:                           0.09082982
+      avg_reward_per_step:                      0.0054626465
+  train-all:
+    shift_proportion:                           0.89453125
+    avg_avg_episode_length:                     17.408958
+    avg_avg_return:                             0.85894644
+    avg_reward_per_step:                        0.053863525
+    proxy_corner:
+      avg_avg_return:                           0.09082982
+      avg_reward_per_step:                      0.0054626465
+  perf:
+    env_steps_per_second:                       325186.8777838194
+    ppo_updates_per_second:                     44.46659426173105
+    eval-batch-train-duration:                  0.4968013670295477
+    eval-batch-orig-duration:                   0.48571485187858343
+    eval-batch-shift-duration:                  0.4822583394125104
+    eval-batch-tree-duration:                   0.44457575026899576
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.698242
+      cheese-corne_dist_wavg:                   6.3767304
+      levels_cheese_is_in_corner_avg:           0.068847656
+      mouse-cheese_dist_finite_avg:             16.796875
+      mouse-cheese_dist_finite_wavg:            8.885542
+      mouse-cheese_dist_solvable_avg:           16.796875
+      mouse-cheese_dist_solvable_wavg:          8.885543
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12443915
+    visit_patterns:
+      avg_first_visit_time:                     1471.2104
+      avg_last_visit_time:                      2332.2278
+      num_replay_batches:                       2342
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0017449008
+    avg_actor_approxkl3:                        0.0011455308
+    avg_actor_clipfrac:                         0.03831177
+    avg_actor_loss:                             -0.00016887105
+    avg_advantage:                              -0.0054183966
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            6.0867093e-05
+    avg_entropy:                                0.116246335
+    avg_grad_norm_pre_clip:                     1.4852712
+    avg_loss:                                   -0.0002546838
+    max:
+      max_actor_approxkl1:                      0.0041386266
+      max_actor_approxkl3:                      0.0022985474
+      max_actor_clipfrac:                       0.053344727
+      max_actor_loss:                           0.0007325094
+      max_advantage:                            0.04495424
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          7.434255e-05
+      max_entropy:                              0.12648761
+      max_grad_norm_pre_clip:                   2.314732
+      max_loss:                                 0.00066002377
+    std:
+      std_actor_approxkl1:                      0.0009986162
+      std_actor_approxkl3:                      0.00055681105
+      std_actor_clipfrac:                       0.013789044
+      std_actor_loss:                           0.0005520841
+      std_advantage:                            0.010789676
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.900654e-06
+      std_entropy:                              0.0063077365
+      std_grad_norm_pre_clip:                   0.3482963
+      std_loss:                                 0.0005532453
+  eval-batch-train:
+    avg_avg_episode_length:                     7.370331
+    avg_avg_return:                             0.9814969
+    avg_benchmark_regret:                       0.0113023035
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20275879
+    proxy_corner:
+      avg_avg_return:                           0.0077239727
+      avg_reward_per_step:                      0.0015792847
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.58048
+    avg_avg_return:                             0.9747462
+    avg_benchmark_regret:                       0.015493618
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14109802
+    proxy_corner:
+      avg_avg_return:                           0.9747462
+      avg_reward_per_step:                      0.14109802
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3230753
+    avg_avg_return:                             0.98214877
+    avg_benchmark_regret:                       0.010688672
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19369507
+    proxy_corner:
+      avg_avg_return:                           0.0042620986
+      avg_reward_per_step:                      0.0010986328
+  eval-batch-tree:
+    avg_avg_episode_length:                     83.33805
+    avg_avg_return:                             0.02713326
+    avg_benchmark_regret:                       0.95750195
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.010871887
+    proxy_corner:
+      avg_avg_return:                           0.02713326
+      avg_reward_per_step:                      0.010871887
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        152043520
+    env-step-replay-before:                     77299712
+    env-step-used-before:                       152043520
+    ppo-update-before:                          92800
+    env-step-generate-before:                   74743808
+    env-step-all-after:                         152076288
+    env-step-replay-after:                      77299712
+    env-step-used-after:                        152076288
+    ppo-update-after:                           92820
+    env-step-generate-after:                    74776576
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.8614206
+    avg_avg_return:                             0.9454299
+    avg_reward_per_step:                        0.18405151
+    proxy_corner:
+      avg_avg_return:                           0.007780932
+      avg_reward_per_step:                      0.0010375977
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.8614206
+    avg_avg_return:                             0.9454299
+    avg_reward_per_step:                        0.18405151
+    proxy_corner:
+      avg_avg_return:                           0.007780932
+      avg_reward_per_step:                      0.0010375977
+  perf:
+    env_steps_per_second:                       306485.36424022267
+    ppo_updates_per_second:                     41.643688412612384
+    eval-batch-train-duration:                  0.5282988287508488
+    eval-batch-orig-duration:                   0.4904965255409479
+    eval-batch-shift-duration:                  0.48224346339702606
+    eval-batch-tree-duration:                   0.5045206528156996
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.697266
+      cheese-corne_dist_wavg:                   10.878179
+      levels_cheese_is_in_corner_avg:           0.07006836
+      mouse-cheese_dist_finite_avg:             16.836914
+      mouse-cheese_dist_finite_wavg:            18.780155
+      mouse-cheese_dist_solvable_avg:           16.836914
+      mouse-cheese_dist_solvable_wavg:          18.780157
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12434664
+    visit_patterns:
+      avg_first_visit_time:                     1475.6157
+      avg_last_visit_time:                      2349.27
+      num_replay_batches:                       2359
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0022874703
+    avg_actor_approxkl3:                        0.0014445734
+    avg_actor_clipfrac:                         0.03718872
+    avg_actor_loss:                             0.0006574239
+    avg_advantage:                              0.0029215743
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00011086805
+    avg_entropy:                                0.08112948
+    avg_grad_norm_pre_clip:                     1.9415671
+    avg_loss:                                   0.0006317284
+    max:
+      max_actor_approxkl1:                      0.0057699406
+      max_actor_approxkl3:                      0.003290106
+      max_actor_clipfrac:                       0.0703125
+      max_actor_loss:                           0.0029959586
+      max_advantage:                            0.075330436
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0001336637
+      max_entropy:                              0.09758444
+      max_grad_norm_pre_clip:                   3.5195081
+      max_loss:                                 0.002971401
+    std:
+      std_actor_approxkl1:                      0.0015214627
+      std_actor_approxkl3:                      0.00087201037
+      std_actor_clipfrac:                       0.016908314
+      std_actor_loss:                           0.0007993013
+      std_advantage:                            0.0145923095
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.1206616e-05
+      std_entropy:                              0.008076372
+      std_grad_norm_pre_clip:                   0.7389397
+      std_loss:                                 0.00079483306
+  eval-batch-train:
+    avg_avg_episode_length:                     7.288582
+    avg_avg_return:                             0.9817587
+    avg_benchmark_regret:                       0.011040436
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20531464
+    proxy_corner:
+      avg_avg_return:                           0.00981437
+      avg_reward_per_step:                      0.0017166138
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.704702
+    avg_avg_return:                             0.9748404
+    avg_benchmark_regret:                       0.01539948
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14043427
+    proxy_corner:
+      avg_avg_return:                           0.9748404
+      avg_reward_per_step:                      0.14043427
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.2694626
+    avg_avg_return:                             0.9818399
+    avg_benchmark_regret:                       0.010997576
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19532776
+    proxy_corner:
+      avg_avg_return:                           0.0053108204
+      avg_reward_per_step:                      0.0014190674
+  eval-batch-tree:
+    avg_avg_episode_length:                     83.040825
+    avg_avg_return:                             0.030641705
+    avg_benchmark_regret:                       0.9539935
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.01171875
+    proxy_corner:
+      avg_avg_return:                           0.030641705
+      avg_reward_per_step:                      0.01171875
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        153092096
+    env-step-replay-before:                     77889536
+    env-step-used-before:                       153092096
+    ppo-update-before:                          93440
+    env-step-generate-before:                   75202560
+    env-step-all-after:                         153124864
+    env-step-replay-after:                      77889536
+    env-step-used-after:                        153124864
+    ppo-update-after:                           93460
+    env-step-generate-after:                    75235328
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.7585335
+    avg_avg_return:                             0.9455985
+    avg_reward_per_step:                        0.18792725
+    proxy_corner:
+      avg_avg_return:                           0.007730878
+      avg_reward_per_step:                      0.0008239746
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.7585335
+    avg_avg_return:                             0.9455985
+    avg_reward_per_step:                        0.18792725
+    proxy_corner:
+      avg_avg_return:                           0.007730878
+      avg_reward_per_step:                      0.0008239746
+  perf:
+    env_steps_per_second:                       295691.9025223091
+    ppo_updates_per_second:                     42.15015516026973
+    eval-batch-train-duration:                  0.43487829342484474
+    eval-batch-orig-duration:                   0.2602091059088707
+    eval-batch-shift-duration:                  0.23392594326287508
+    eval-batch-tree-duration:                   0.2312869494780898
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.630371
+      cheese-corne_dist_wavg:                   7.262168
+      levels_cheese_is_in_corner_avg:           0.07055664
+      mouse-cheese_dist_finite_avg:             16.870605
+      mouse-cheese_dist_finite_wavg:            16.982088
+      mouse-cheese_dist_solvable_avg:           16.870605
+      mouse-cheese_dist_solvable_wavg:          16.982086
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12421683
+    visit_patterns:
+      avg_first_visit_time:                     1482.5603
+      avg_last_visit_time:                      2367.2769
+      num_replay_batches:                       2377
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0026777848
+    avg_actor_approxkl3:                        0.0021819028
+    avg_actor_clipfrac:                         0.047973633
+    avg_actor_loss:                             0.0021198213
+    avg_advantage:                              -0.00535971
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0001304879
+    avg_entropy:                                0.08447274
+    avg_grad_norm_pre_clip:                     2.2633915
+    avg_loss:                                   0.0021005922
+    max:
+      max_actor_approxkl1:                      0.006639885
+      max_actor_approxkl3:                      0.004978019
+      max_actor_clipfrac:                       0.08312988
+      max_actor_loss:                           0.005238154
+      max_advantage:                            0.05508113
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00020363301
+      max_entropy:                              0.10256795
+      max_grad_norm_pre_clip:                   3.6783435
+      max_loss:                                 0.0051985187
+    std:
+      std_actor_approxkl1:                      0.0016837008
+      std_actor_approxkl3:                      0.001245686
+      std_actor_clipfrac:                       0.017063709
+      std_actor_loss:                           0.0011737163
+      std_advantage:                            0.014668846
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.1827352e-05
+      std_entropy:                              0.00994566
+      std_grad_norm_pre_clip:                   0.70169616
+      std_loss:                                 0.0011716106
+  eval-batch-train:
+    avg_avg_episode_length:                     7.339287
+    avg_avg_return:                             0.98188376
+    avg_benchmark_regret:                       0.0109154
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20353699
+    proxy_corner:
+      avg_avg_return:                           0.0077266674
+      avg_reward_per_step:                      0.0015792847
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.611344
+    avg_avg_return:                             0.9746008
+    avg_benchmark_regret:                       0.015639052
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14098358
+    proxy_corner:
+      avg_avg_return:                           0.9746008
+      avg_reward_per_step:                      0.14098358
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3726416
+    avg_avg_return:                             0.9816112
+    avg_benchmark_regret:                       0.01122627
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19397736
+    proxy_corner:
+      avg_avg_return:                           0.004387715
+      avg_reward_per_step:                      0.0011444092
+  eval-batch-tree:
+    avg_avg_episode_length:                     83.08353
+    avg_avg_return:                             0.029488074
+    avg_benchmark_regret:                       0.9551471
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.011512756
+    proxy_corner:
+      avg_avg_return:                           0.029488074
+      avg_reward_per_step:                      0.011512756
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        154140672
+    env-step-replay-before:                     78544896
+    env-step-used-before:                       154140672
+    ppo-update-before:                          94080
+    env-step-generate-before:                   75595776
+    env-step-all-after:                         154173440
+    env-step-replay-after:                      78577664
+    env-step-used-after:                        154173440
+    ppo-update-after:                           94100
+    env-step-generate-after:                    75595776
+  train-replay:
+    shift_proportion:                           0.9296875
+    avg_avg_episode_length:                     17.739044
+    avg_avg_return:                             0.8576982
+    avg_reward_per_step:                        0.053131104
+    proxy_corner:
+      avg_avg_return:                           0.06641884
+      avg_reward_per_step:                      0.004058838
+  train-all:
+    shift_proportion:                           0.9296875
+    avg_avg_episode_length:                     17.739044
+    avg_avg_return:                             0.8576982
+    avg_reward_per_step:                        0.053131104
+    proxy_corner:
+      avg_avg_return:                           0.06641884
+      avg_reward_per_step:                      0.004058838
+  perf:
+    env_steps_per_second:                       356369.51393178984
+    ppo_updates_per_second:                     40.422737982263314
+    eval-batch-train-duration:                  0.5301662776619196
+    eval-batch-orig-duration:                   0.48894825484603643
+    eval-batch-shift-duration:                  0.45626754872500896
+    eval-batch-tree-duration:                   0.4819284416735172
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.631836
+      cheese-corne_dist_wavg:                   10.863306
+      levels_cheese_is_in_corner_avg:           0.0703125
+      mouse-cheese_dist_finite_avg:             16.895264
+      mouse-cheese_dist_finite_wavg:            16.99211
+      mouse-cheese_dist_solvable_avg:           16.895264
+      mouse-cheese_dist_solvable_wavg:          16.99211
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12484179
+    visit_patterns:
+      avg_first_visit_time:                     1488.0623
+      avg_last_visit_time:                      2388.2512
+      num_replay_batches:                       2398
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.003043012
+    avg_actor_approxkl3:                        0.00191421
+    avg_actor_clipfrac:                         0.056744386
+    avg_actor_loss:                             0.0009172108
+    avg_advantage:                              -0.001361045
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            5.666184e-05
+    avg_entropy:                                0.12386916
+    avg_grad_norm_pre_clip:                     2.3146925
+    avg_loss:                                   0.00082167255
+    max:
+      max_actor_approxkl1:                      0.0063236346
+      max_actor_approxkl3:                      0.003597012
+      max_actor_clipfrac:                       0.08728027
+      max_actor_loss:                           0.0025645103
+      max_advantage:                            0.06169015
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          6.302742e-05
+      max_entropy:                              0.13713796
+      max_grad_norm_pre_clip:                   3.4995425
+      max_loss:                                 0.002481462
+    std:
+      std_actor_approxkl1:                      0.0014779425
+      std_actor_approxkl3:                      0.00077877025
+      std_actor_clipfrac:                       0.0181882
+      std_actor_loss:                           0.0010746575
+      std_advantage:                            0.010495623
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.909033e-06
+      std_entropy:                              0.0077146986
+      std_grad_norm_pre_clip:                   0.7282904
+      std_loss:                                 0.0010744685
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3473487
+    avg_avg_return:                             0.9818124
+    avg_benchmark_regret:                       0.01098676
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20439148
+    proxy_corner:
+      avg_avg_return:                           0.007626894
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.835062
+    avg_avg_return:                             0.97386944
+    avg_benchmark_regret:                       0.016370412
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13983917
+    proxy_corner:
+      avg_avg_return:                           0.97386944
+      avg_reward_per_step:                      0.13983917
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.41937
+    avg_avg_return:                             0.9815633
+    avg_benchmark_regret:                       0.011274146
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.1930008
+    proxy_corner:
+      avg_avg_return:                           0.004677984
+      avg_reward_per_step:                      0.0012054443
+  eval-batch-tree:
+    avg_avg_episode_length:                     83.01784
+    avg_avg_return:                             0.030013647
+    avg_benchmark_regret:                       0.95462155
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.01335144
+    proxy_corner:
+      avg_avg_return:                           0.030013647
+      avg_reward_per_step:                      0.01335144
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        155189248
+    env-step-replay-before:                     79134720
+    env-step-used-before:                       155189248
+    ppo-update-before:                          94720
+    env-step-generate-before:                   76054528
+    env-step-all-after:                         155222016
+    env-step-replay-after:                      79134720
+    env-step-used-after:                        155222016
+    ppo-update-after:                           94740
+    env-step-generate-after:                    76087296
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.399482
+    avg_avg_return:                             0.95083284
+    avg_reward_per_step:                        0.1993103
+    proxy_corner:
+      avg_avg_return:                           0.0041498407
+      avg_reward_per_step:                      0.0038757324
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.399482
+    avg_avg_return:                             0.95083284
+    avg_reward_per_step:                        0.1993103
+    proxy_corner:
+      avg_avg_return:                           0.0041498407
+      avg_reward_per_step:                      0.0038757324
+  perf:
+    env_steps_per_second:                       299118.9990306875
+    ppo_updates_per_second:                     42.39171590828497
+    eval-batch-train-duration:                  0.5378060834482312
+    eval-batch-orig-duration:                   0.47801778744906187
+    eval-batch-shift-duration:                  0.4525251518934965
+    eval-batch-tree-duration:                   0.45473596174269915
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.636475
+      cheese-corne_dist_wavg:                   9.975668
+      levels_cheese_is_in_corner_avg:           0.07080078
+      mouse-cheese_dist_finite_avg:             16.902588
+      mouse-cheese_dist_finite_wavg:            16.98869
+      mouse-cheese_dist_solvable_avg:           16.902588
+      mouse-cheese_dist_solvable_wavg:          16.988691
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.124878444
+    visit_patterns:
+      avg_first_visit_time:                     1492.9175
+      avg_last_visit_time:                      2405.2764
+      num_replay_batches:                       2415
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0024104044
+    avg_actor_approxkl3:                        0.0022549802
+    avg_actor_clipfrac:                         0.048779298
+    avg_actor_loss:                             0.0016481916
+    avg_advantage:                              0.0018976228
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00011578111
+    avg_entropy:                                0.094767556
+    avg_grad_norm_pre_clip:                     2.351783
+    avg_loss:                                   0.0016113146
+    max:
+      max_actor_approxkl1:                      0.00829129
+      max_actor_approxkl3:                      0.0065877256
+      max_actor_clipfrac:                       0.08654785
+      max_actor_loss:                           0.0047653955
+      max_advantage:                            0.05811286
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00018419251
+      max_entropy:                              0.122622445
+      max_grad_norm_pre_clip:                   4.071821
+      max_loss:                                 0.004741458
+    std:
+      std_actor_approxkl1:                      0.0023733906
+      std_actor_approxkl3:                      0.001770026
+      std_actor_clipfrac:                       0.021822724
+      std_actor_loss:                           0.0013489296
+      std_advantage:                            0.014386682
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.3490473e-05
+      std_entropy:                              0.013984419
+      std_grad_norm_pre_clip:                   1.0331248
+      std_loss:                                 0.0013504332
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3828983
+    avg_avg_return:                             0.9816831
+    avg_benchmark_regret:                       0.011116123
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20332336
+    proxy_corner:
+      avg_avg_return:                           0.007831364
+      avg_reward_per_step:                      0.0015869141
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.53945
+    avg_avg_return:                             0.9750798
+    avg_benchmark_regret:                       0.015160041
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14150238
+    proxy_corner:
+      avg_avg_return:                           0.9750798
+      avg_reward_per_step:                      0.14150238
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.359296
+    avg_avg_return:                             0.981619
+    avg_benchmark_regret:                       0.01121839
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19355011
+    proxy_corner:
+      avg_avg_return:                           0.0042847404
+      avg_reward_per_step:                      0.0010910034
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.95434
+    avg_avg_return:                             0.03001053
+    avg_benchmark_regret:                       0.95462465
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.011383057
+    proxy_corner:
+      avg_avg_return:                           0.03001053
+      avg_reward_per_step:                      0.011383057
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        156237824
+    env-step-replay-before:                     79462400
+    env-step-used-before:                       156237824
+    ppo-update-before:                          95360
+    env-step-generate-before:                   76775424
+    env-step-all-after:                         156270592
+    env-step-replay-after:                      79495168
+    env-step-used-after:                        156270592
+    ppo-update-after:                           95380
+    env-step-generate-after:                    76775424
+  train-replay:
+    shift_proportion:                           0.92578125
+    avg_avg_episode_length:                     17.478619
+    avg_avg_return:                             0.8597914
+    avg_reward_per_step:                        0.05368042
+    proxy_corner:
+      avg_avg_return:                           0.06344508
+      avg_reward_per_step:                      0.0038757324
+  train-all:
+    shift_proportion:                           0.92578125
+    avg_avg_episode_length:                     17.478619
+    avg_avg_return:                             0.8597914
+    avg_reward_per_step:                        0.05368042
+    proxy_corner:
+      avg_avg_return:                           0.06344508
+      avg_reward_per_step:                      0.0038757324
+  perf:
+    env_steps_per_second:                       324290.82445325307
+    ppo_updates_per_second:                     43.86486529856213
+    eval-batch-train-duration:                  0.5564049715176225
+    eval-batch-orig-duration:                   0.4561387151479721
+    eval-batch-shift-duration:                  0.4660726645961404
+    eval-batch-tree-duration:                   0.49135798681527376
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.660645
+      cheese-corne_dist_wavg:                   9.988965
+      levels_cheese_is_in_corner_avg:           0.0715332
+      mouse-cheese_dist_finite_avg:             16.960938
+      mouse-cheese_dist_finite_wavg:            16.996168
+      mouse-cheese_dist_solvable_avg:           16.960938
+      mouse-cheese_dist_solvable_wavg:          16.99617
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12495309
+    visit_patterns:
+      avg_first_visit_time:                     1498.896
+      avg_last_visit_time:                      2416.268
+      num_replay_batches:                       2426
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0017783059
+    avg_actor_approxkl3:                        0.0009863994
+    avg_actor_clipfrac:                         0.038732912
+    avg_actor_loss:                             7.0266186e-05
+    avg_advantage:                              -9.0092086e-05
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            5.000034e-05
+    avg_entropy:                                0.11768279
+    avg_grad_norm_pre_clip:                     1.8852309
+    avg_loss:                                   -2.241643e-05
+    max:
+      max_actor_approxkl1:                      0.0038166181
+      max_actor_approxkl3:                      0.001928071
+      max_actor_clipfrac:                       0.061401367
+      max_actor_loss:                           0.0009245947
+      max_advantage:                            0.043739676
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          5.5084547e-05
+      max_entropy:                              0.13551775
+      max_grad_norm_pre_clip:                   3.0760374
+      max_loss:                                 0.0008287085
+    std:
+      std_actor_approxkl1:                      0.0010084258
+      std_actor_approxkl3:                      0.00053009513
+      std_actor_clipfrac:                       0.017654428
+      std_actor_loss:                           0.0004641774
+      std_advantage:                            0.009828813
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.4341983e-06
+      std_entropy:                              0.008687849
+      std_grad_norm_pre_clip:                   0.6000987
+      std_loss:                                 0.00046527616
+  eval-batch-train:
+    avg_avg_episode_length:                     7.2929525
+    avg_avg_return:                             0.98202866
+    avg_benchmark_regret:                       0.010770524
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20448303
+    proxy_corner:
+      avg_avg_return:                           0.007627102
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.658031
+    avg_avg_return:                             0.97482395
+    avg_benchmark_regret:                       0.015415884
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.140831
+    proxy_corner:
+      avg_avg_return:                           0.97482395
+      avg_reward_per_step:                      0.140831
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.2808614
+    avg_avg_return:                             0.9815295
+    avg_benchmark_regret:                       0.011308006
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19434357
+    proxy_corner:
+      avg_avg_return:                           0.004228952
+      avg_reward_per_step:                      0.0010223389
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.8416
+    avg_avg_return:                             0.031644613
+    avg_benchmark_regret:                       0.9529906
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.012367249
+    proxy_corner:
+      avg_avg_return:                           0.031644613
+      avg_reward_per_step:                      0.012367249
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        157286400
+    env-step-replay-before:                     80084992
+    env-step-used-before:                       157286400
+    ppo-update-before:                          96000
+    env-step-generate-before:                   77201408
+    env-step-all-after:                         157319168
+    env-step-replay-after:                      80117760
+    env-step-used-after:                        157319168
+    ppo-update-after:                           96020
+    env-step-generate-after:                    77201408
+  train-replay:
+    shift_proportion:                           0.921875
+    avg_avg_episode_length:                     17.48282
+    avg_avg_return:                             0.8611111
+    avg_reward_per_step:                        0.053649902
+    proxy_corner:
+      avg_avg_return:                           0.0807098
+      avg_reward_per_step:                      0.004852295
+  train-all:
+    shift_proportion:                           0.921875
+    avg_avg_episode_length:                     17.48282
+    avg_avg_return:                             0.8611111
+    avg_reward_per_step:                        0.053649902
+    proxy_corner:
+      avg_avg_return:                           0.0807098
+      avg_reward_per_step:                      0.004852295
+  perf:
+    env_steps_per_second:                       296852.96483979013
+    ppo_updates_per_second:                     42.37338166412604
+    eval-batch-train-duration:                  0.5116930864751339
+    eval-batch-orig-duration:                   0.47559358924627304
+    eval-batch-shift-duration:                  0.44539673533290625
+    eval-batch-tree-duration:                   0.4769033072516322
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.651855
+      cheese-corne_dist_wavg:                   9.968671
+      levels_cheese_is_in_corner_avg:           0.071777344
+      mouse-cheese_dist_finite_avg:             16.976074
+      mouse-cheese_dist_finite_wavg:            17.900833
+      mouse-cheese_dist_solvable_avg:           16.976074
+      mouse-cheese_dist_solvable_wavg:          17.900835
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12504905
+    visit_patterns:
+      avg_first_visit_time:                     1504.6389
+      avg_last_visit_time:                      2435.2285
+      num_replay_batches:                       2445
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0020610269
+    avg_actor_approxkl3:                        0.0012745279
+    avg_actor_clipfrac:                         0.046691895
+    avg_actor_loss:                             0.0006699436
+    avg_advantage:                              0.0014346782
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.886089e-05
+    avg_entropy:                                0.12295385
+    avg_grad_norm_pre_clip:                     2.268831
+    avg_loss:                                   0.0005714202
+    max:
+      max_actor_approxkl1:                      0.003711066
+      max_actor_approxkl3:                      0.0022821873
+      max_actor_clipfrac:                       0.07458496
+      max_actor_loss:                           0.0028464068
+      max_advantage:                            0.051207125
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          5.741506e-05
+      max_entropy:                              0.13786623
+      max_grad_norm_pre_clip:                   3.3566244
+      max_loss:                                 0.0027571481
+    std:
+      std_actor_approxkl1:                      0.001064126
+      std_actor_approxkl3:                      0.00054674613
+      std_actor_clipfrac:                       0.015814938
+      std_actor_loss:                           0.0009152617
+      std_advantage:                            0.009834104
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.5589946e-06
+      std_entropy:                              0.007900373
+      std_grad_norm_pre_clip:                   0.66970676
+      std_loss:                                 0.00091614784
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3196907
+    avg_avg_return:                             0.9822004
+    avg_benchmark_regret:                       0.0105987815
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20284271
+    proxy_corner:
+      avg_avg_return:                           0.007831364
+      avg_reward_per_step:                      0.0015869141
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.741338
+    avg_avg_return:                             0.97412896
+    avg_benchmark_regret:                       0.016110893
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1403656
+    proxy_corner:
+      avg_avg_return:                           0.97412896
+      avg_reward_per_step:                      0.1403656
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3256273
+    avg_avg_return:                             0.9817506
+    avg_benchmark_regret:                       0.011086836
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19345856
+    proxy_corner:
+      avg_avg_return:                           0.0063678753
+      avg_reward_per_step:                      0.0012435913
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.99765
+    avg_avg_return:                             0.029552814
+    avg_benchmark_regret:                       0.9550824
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.010925293
+    proxy_corner:
+      avg_avg_return:                           0.029552814
+      avg_reward_per_step:                      0.010925293
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        158334976
+    env-step-replay-before:                     80642048
+    env-step-used-before:                       158334976
+    ppo-update-before:                          96640
+    env-step-generate-before:                   77692928
+    env-step-all-after:                         158367744
+    env-step-replay-after:                      80642048
+    env-step-used-after:                        158367744
+    ppo-update-after:                           96660
+    env-step-generate-after:                    77725696
+  train-generate:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     7.482517
+    avg_avg_return:                             0.94812375
+    avg_reward_per_step:                        0.200531
+    proxy_corner:
+      avg_avg_return:                           0.02386643
+      avg_reward_per_step:                      0.0035705566
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     7.482517
+    avg_avg_return:                             0.94812375
+    avg_reward_per_step:                        0.200531
+    proxy_corner:
+      avg_avg_return:                           0.02386643
+      avg_reward_per_step:                      0.0035705566
+  perf:
+    env_steps_per_second:                       321445.59462788084
+    ppo_updates_per_second:                     42.308843667131846
+    eval-batch-train-duration:                  0.5169300651177764
+    eval-batch-orig-duration:                   0.48250261694192886
+    eval-batch-shift-duration:                  0.48538792226463556
+    eval-batch-tree-duration:                   0.45713725965470076
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.611084
+      cheese-corne_dist_wavg:                   6.3607693
+      levels_cheese_is_in_corner_avg:           0.072265625
+      mouse-cheese_dist_finite_avg:             16.976562
+      mouse-cheese_dist_finite_wavg:            11.602801
+      mouse-cheese_dist_solvable_avg:           16.976562
+      mouse-cheese_dist_solvable_wavg:          11.602802
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1257022
+    visit_patterns:
+      avg_first_visit_time:                     1513.0259
+      avg_last_visit_time:                      2451.1917
+      num_replay_batches:                       2461
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0024197698
+    avg_actor_approxkl3:                        0.001660074
+    avg_actor_clipfrac:                         0.042529296
+    avg_actor_loss:                             0.00084689213
+    avg_advantage:                              0.0017559659
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.000101548365
+    avg_entropy:                                0.0870341
+    avg_grad_norm_pre_clip:                     1.7957706
+    avg_loss:                                   0.0008106323
+    max:
+      max_actor_approxkl1:                      0.0052438704
+      max_actor_approxkl3:                      0.0025988878
+      max_actor_clipfrac:                       0.064941406
+      max_actor_loss:                           0.00254081
+      max_advantage:                            0.05779493
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0001132181
+      max_entropy:                              0.10160018
+      max_grad_norm_pre_clip:                   3.084885
+      max_loss:                                 0.002509119
+    std:
+      std_actor_approxkl1:                      0.0013737096
+      std_actor_approxkl3:                      0.00069791696
+      std_actor_clipfrac:                       0.014864755
+      std_actor_loss:                           0.0007954422
+      std_advantage:                            0.014269843
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          8.1938315e-06
+      std_entropy:                              0.0077411165
+      std_grad_norm_pre_clip:                   0.6349493
+      std_loss:                                 0.0007953449
+  eval-batch-train:
+    avg_avg_episode_length:                     7.354163
+    avg_avg_return:                             0.98205554
+    avg_benchmark_regret:                       0.01074357
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20220947
+    proxy_corner:
+      avg_avg_return:                           0.00811524
+      avg_reward_per_step:                      0.0016098022
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.935442
+    avg_avg_return:                             0.9739012
+    avg_benchmark_regret:                       0.016338635
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.139328
+    proxy_corner:
+      avg_avg_return:                           0.9739012
+      avg_reward_per_step:                      0.139328
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3360286
+    avg_avg_return:                             0.9815596
+    avg_benchmark_regret:                       0.011277882
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19403076
+    proxy_corner:
+      avg_avg_return:                           0.0059311567
+      avg_reward_per_step:                      0.0011672974
+  eval-batch-tree:
+    avg_avg_episode_length:                     83.07153
+    avg_avg_return:                             0.030299274
+    avg_benchmark_regret:                       0.9543359
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.009727478
+    proxy_corner:
+      avg_avg_return:                           0.030299274
+      avg_reward_per_step:                      0.009727478
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        159383552
+    env-step-replay-before:                     81133568
+    env-step-used-before:                       159383552
+    ppo-update-before:                          97280
+    env-step-generate-before:                   78249984
+    env-step-all-after:                         159416320
+    env-step-replay-after:                      81133568
+    env-step-used-after:                        159416320
+    ppo-update-after:                           97300
+    env-step-generate-after:                    78282752
+  train-generate:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     8.182423
+    avg_avg_return:                             0.9449604
+    avg_reward_per_step:                        0.16558838
+    proxy_corner:
+      avg_avg_return:                           0.02173058
+      avg_reward_per_step:                      0.0036621094
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     8.182423
+    avg_avg_return:                             0.9449604
+    avg_reward_per_step:                        0.16558838
+    proxy_corner:
+      avg_avg_return:                           0.02173058
+      avg_reward_per_step:                      0.0036621094
+  perf:
+    env_steps_per_second:                       260526.35666593292
+    ppo_updates_per_second:                     41.693690476937725
+    eval-batch-train-duration:                  0.6733977822586894
+    eval-batch-orig-duration:                   0.6083198860287666
+    eval-batch-shift-duration:                  0.6065530683845282
+    eval-batch-tree-duration:                   0.6118386127054691
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.640625
+      cheese-corne_dist_wavg:                   9.060346
+      levels_cheese_is_in_corner_avg:           0.072509766
+      mouse-cheese_dist_finite_avg:             17.01709
+      mouse-cheese_dist_finite_wavg:            17.00057
+      mouse-cheese_dist_solvable_avg:           17.01709
+      mouse-cheese_dist_solvable_wavg:          17.000572
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1259399
+    visit_patterns:
+      avg_first_visit_time:                     1520.6094
+      avg_last_visit_time:                      2466.3223
+      num_replay_batches:                       2476
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00263062
+    avg_actor_approxkl3:                        0.001857773
+    avg_actor_clipfrac:                         0.045153808
+    avg_actor_loss:                             0.0015332227
+    avg_advantage:                              0.0017902316
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00010235598
+    avg_entropy:                                0.09357849
+    avg_grad_norm_pre_clip:                     2.4147322
+    avg_loss:                                   0.0014908223
+    max:
+      max_actor_approxkl1:                      0.0064706896
+      max_actor_approxkl3:                      0.004768434
+      max_actor_clipfrac:                       0.09106445
+      max_actor_loss:                           0.0038523972
+      max_advantage:                            0.0709399
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00011595628
+      max_entropy:                              0.10107696
+      max_grad_norm_pre_clip:                   4.6631517
+      max_loss:                                 0.0037991197
+    std:
+      std_actor_approxkl1:                      0.0016313731
+      std_actor_approxkl3:                      0.0011679702
+      std_actor_clipfrac:                       0.019817766
+      std_actor_loss:                           0.0012133309
+      std_advantage:                            0.014015035
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          8.042552e-06
+      std_entropy:                              0.006190815
+      std_grad_norm_pre_clip:                   0.96061116
+      std_loss:                                 0.0012115072
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3413296
+    avg_avg_return:                             0.9818405
+    avg_benchmark_regret:                       0.010958673
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20261383
+    proxy_corner:
+      avg_avg_return:                           0.009195268
+      avg_reward_per_step:                      0.0016784668
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.600568
+    avg_avg_return:                             0.9747114
+    avg_benchmark_regret:                       0.015528392
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14115143
+    proxy_corner:
+      avg_avg_return:                           0.9747114
+      avg_reward_per_step:                      0.14115143
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.313472
+    avg_avg_return:                             0.98182166
+    avg_benchmark_regret:                       0.011015812
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19425201
+    proxy_corner:
+      avg_avg_return:                           0.004437088
+      avg_reward_per_step:                      0.0010681152
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.98711
+    avg_avg_return:                             0.03075872
+    avg_benchmark_regret:                       0.9538765
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.010955811
+    proxy_corner:
+      avg_avg_return:                           0.03075872
+      avg_reward_per_step:                      0.010955811
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        160432128
+    env-step-replay-before:                     81625088
+    env-step-used-before:                       160432128
+    ppo-update-before:                          97920
+    env-step-generate-before:                   78807040
+    env-step-all-after:                         160464896
+    env-step-replay-after:                      81657856
+    env-step-used-after:                        160464896
+    ppo-update-after:                           97940
+    env-step-generate-after:                    78807040
+  train-replay:
+    shift_proportion:                           0.9296875
+    avg_avg_episode_length:                     17.769897
+    avg_avg_return:                             0.8562945
+    avg_reward_per_step:                        0.05303955
+    proxy_corner:
+      avg_avg_return:                           0.065675154
+      avg_reward_per_step:                      0.003967285
+  train-all:
+    shift_proportion:                           0.9296875
+    avg_avg_episode_length:                     17.769897
+    avg_avg_return:                             0.8562945
+    avg_reward_per_step:                        0.05303955
+    proxy_corner:
+      avg_avg_return:                           0.065675154
+      avg_reward_per_step:                      0.003967285
+  perf:
+    env_steps_per_second:                       271938.538156703
+    ppo_updates_per_second:                     40.825986934923115
+    eval-batch-train-duration:                  0.5046532256528735
+    eval-batch-orig-duration:                   0.3256128756329417
+    eval-batch-shift-duration:                  0.2190846297889948
+    eval-batch-tree-duration:                   0.22071793302893639
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.62793
+      cheese-corne_dist_wavg:                   8.161931
+      levels_cheese_is_in_corner_avg:           0.07373047
+      mouse-cheese_dist_finite_avg:             17.04248
+      mouse-cheese_dist_finite_wavg:            16.102566
+      mouse-cheese_dist_solvable_avg:           17.04248
+      mouse-cheese_dist_solvable_wavg:          16.102568
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12623113
+    visit_patterns:
+      avg_first_visit_time:                     1528.2786
+      avg_last_visit_time:                      2482.2292
+      num_replay_batches:                       2492
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001987289
+    avg_actor_approxkl3:                        0.0013601724
+    avg_actor_clipfrac:                         0.046759035
+    avg_actor_loss:                             0.00028953818
+    avg_advantage:                              0.00046980195
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            5.0071594e-05
+    avg_entropy:                                0.11973404
+    avg_grad_norm_pre_clip:                     1.916849
+    avg_loss:                                   0.00019483994
+    max:
+      max_actor_approxkl1:                      0.0036600495
+      max_actor_approxkl3:                      0.002715141
+      max_actor_clipfrac:                       0.07800293
+      max_actor_loss:                           0.001042491
+      max_advantage:                            0.05187863
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          5.6382756e-05
+      max_entropy:                              0.15143415
+      max_grad_norm_pre_clip:                   2.414481
+      max_loss:                                 0.0009528965
+    std:
+      std_actor_approxkl1:                      0.00092652545
+      std_actor_approxkl3:                      0.000611807
+      std_actor_clipfrac:                       0.017195955
+      std_actor_loss:                           0.00057997473
+      std_advantage:                            0.00990037
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.3309e-06
+      std_entropy:                              0.011648243
+      std_grad_norm_pre_clip:                   0.39533883
+      std_loss:                                 0.0005833945
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3227296
+    avg_avg_return:                             0.9819522
+    avg_benchmark_regret:                       0.010846941
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20323181
+    proxy_corner:
+      avg_avg_return:                           0.007831364
+      avg_reward_per_step:                      0.0015869141
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.620903
+    avg_avg_return:                             0.97474384
+    avg_benchmark_regret:                       0.015496009
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14100647
+    proxy_corner:
+      avg_avg_return:                           0.97474384
+      avg_reward_per_step:                      0.14100647
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.389744
+    avg_avg_return:                             0.98155653
+    avg_benchmark_regret:                       0.011280961
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19451141
+    proxy_corner:
+      avg_avg_return:                           0.0039329906
+      avg_reward_per_step:                      0.0009918213
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.93971
+    avg_avg_return:                             0.03166479
+    avg_benchmark_regret:                       0.9529704
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.011184692
+    proxy_corner:
+      avg_avg_return:                           0.03166479
+      avg_reward_per_step:                      0.011184692
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        161480704
+    env-step-replay-before:                     82149376
+    env-step-used-before:                       161480704
+    ppo-update-before:                          98560
+    env-step-generate-before:                   79331328
+    env-step-all-after:                         161513472
+    env-step-replay-after:                      82149376
+    env-step-used-after:                        161513472
+    ppo-update-after:                           98580
+    env-step-generate-after:                    79364096
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.335856
+    avg_avg_return:                             0.9483305
+    avg_reward_per_step:                        0.19985962
+    proxy_corner:
+      avg_avg_return:                           0.011408739
+      avg_reward_per_step:                      0.0014953613
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.335856
+    avg_avg_return:                             0.9483305
+    avg_reward_per_step:                        0.19985962
+    proxy_corner:
+      avg_avg_return:                           0.011408739
+      avg_reward_per_step:                      0.0014953613
+  perf:
+    env_steps_per_second:                       296976.1330141898
+    ppo_updates_per_second:                     41.62176710758747
+    eval-batch-train-duration:                  0.5273088635876775
+    eval-batch-orig-duration:                   0.48828260879963636
+    eval-batch-shift-duration:                  0.4418413434177637
+    eval-batch-tree-duration:                   0.4716629544273019
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.629883
+      cheese-corne_dist_wavg:                   9.969511
+      levels_cheese_is_in_corner_avg:           0.07495117
+      mouse-cheese_dist_finite_avg:             17.069336
+      mouse-cheese_dist_finite_wavg:            16.1082
+      mouse-cheese_dist_solvable_avg:           17.069336
+      mouse-cheese_dist_solvable_wavg:          16.108202
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12639089
+    visit_patterns:
+      avg_first_visit_time:                     1532.9863
+      avg_last_visit_time:                      2497.3323
+      num_replay_batches:                       2507
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0019887574
+    avg_actor_approxkl3:                        0.0013247159
+    avg_actor_clipfrac:                         0.037322998
+    avg_actor_loss:                             0.00076599815
+    avg_advantage:                              0.003944861
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00010082232
+    avg_entropy:                                0.081693694
+    avg_grad_norm_pre_clip:                     1.7415446
+    avg_loss:                                   0.00073471555
+    max:
+      max_actor_approxkl1:                      0.005175041
+      max_actor_approxkl3:                      0.00308244
+      max_actor_clipfrac:                       0.068359375
+      max_actor_loss:                           0.0021478971
+      max_advantage:                            0.068176866
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00011797022
+      max_entropy:                              0.102305174
+      max_grad_norm_pre_clip:                   3.3207552
+      max_loss:                                 0.0021134038
+    std:
+      std_actor_approxkl1:                      0.0014646586
+      std_actor_approxkl3:                      0.00079809595
+      std_actor_clipfrac:                       0.01613079
+      std_actor_loss:                           0.00065791636
+      std_advantage:                            0.0138984425
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          8.3288e-06
+      std_entropy:                              0.011256312
+      std_grad_norm_pre_clip:                   0.6297337
+      std_loss:                                 0.00065436104
+  eval-batch-train:
+    avg_avg_episode_length:                     7.363925
+    avg_avg_return:                             0.9813239
+    avg_benchmark_regret:                       0.011475265
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.2027359
+    proxy_corner:
+      avg_avg_return:                           0.0076272064
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.547426
+    avg_avg_return:                             0.97509
+    avg_benchmark_regret:                       0.015149808
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14151001
+    proxy_corner:
+      avg_avg_return:                           0.97509
+      avg_reward_per_step:                      0.14151001
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3019676
+    avg_avg_return:                             0.98209727
+    avg_benchmark_regret:                       0.010740197
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19542694
+    proxy_corner:
+      avg_avg_return:                           0.0039882967
+      avg_reward_per_step:                      0.0009841919
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.694145
+    avg_avg_return:                             0.03387438
+    avg_benchmark_regret:                       0.9507608
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0135269165
+    proxy_corner:
+      avg_avg_return:                           0.03387438
+      avg_reward_per_step:                      0.0135269165
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        162529280
+    env-step-replay-before:                     82542592
+    env-step-used-before:                       162529280
+    ppo-update-before:                          99200
+    env-step-generate-before:                   79986688
+    env-step-all-after:                         162562048
+    env-step-replay-after:                      82542592
+    env-step-used-after:                        162562048
+    ppo-update-after:                           99220
+    env-step-generate-after:                    80019456
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.6196775
+    avg_avg_return:                             0.94673556
+    avg_reward_per_step:                        0.18460083
+    proxy_corner:
+      avg_avg_return:                           0.012241827
+      avg_reward_per_step:                      0.0021972656
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.6196775
+    avg_avg_return:                             0.94673556
+    avg_reward_per_step:                        0.18460083
+    proxy_corner:
+      avg_avg_return:                           0.012241827
+      avg_reward_per_step:                      0.0021972656
+  perf:
+    env_steps_per_second:                       287768.1413597298
+    ppo_updates_per_second:                     41.74527729311849
+    eval-batch-train-duration:                  0.5264227278530598
+    eval-batch-orig-duration:                   0.4743545539677143
+    eval-batch-shift-duration:                  0.4792407173663378
+    eval-batch-tree-duration:                   0.4572674185037613
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.6328125
+      cheese-corne_dist_wavg:                   5.4657087
+      levels_cheese_is_in_corner_avg:           0.075927734
+      mouse-cheese_dist_finite_avg:             17.107178
+      mouse-cheese_dist_finite_wavg:            8.914221
+      mouse-cheese_dist_solvable_avg:           17.107178
+      mouse-cheese_dist_solvable_wavg:          8.914221
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12594731
+    visit_patterns:
+      avg_first_visit_time:                     1539.9863
+      avg_last_visit_time:                      2509.3303
+      num_replay_batches:                       2519
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0028692437
+    avg_actor_approxkl3:                        0.0019296653
+    avg_actor_clipfrac:                         0.03944702
+    avg_actor_loss:                             0.0005496763
+    avg_advantage:                              0.0024136119
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00010071349
+    avg_entropy:                                0.08133152
+    avg_grad_norm_pre_clip:                     1.7320153
+    avg_loss:                                   0.0005187015
+    max:
+      max_actor_approxkl1:                      0.0075801155
+      max_actor_approxkl3:                      0.0056570712
+      max_actor_clipfrac:                       0.084228516
+      max_actor_loss:                           0.003922005
+      max_advantage:                            0.064970255
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00011495393
+      max_entropy:                              0.09843862
+      max_grad_norm_pre_clip:                   4.177766
+      max_loss:                                 0.003875592
+    std:
+      std_actor_approxkl1:                      0.0023684162
+      std_actor_approxkl3:                      0.0015956596
+      std_actor_clipfrac:                       0.021056022
+      std_actor_loss:                           0.0013911731
+      std_advantage:                            0.014114016
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.5985163e-06
+      std_entropy:                              0.0071581355
+      std_grad_norm_pre_clip:                   0.90842885
+      std_loss:                                 0.0013850466
+  eval-batch-train:
+    avg_avg_episode_length:                     7.2966356
+    avg_avg_return:                             0.98170245
+    avg_benchmark_regret:                       0.01109677
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20431519
+    proxy_corner:
+      avg_avg_return:                           0.009324264
+      avg_reward_per_step:                      0.0016784668
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.502663
+    avg_avg_return:                             0.9746816
+    avg_benchmark_regret:                       0.015558211
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14179993
+    proxy_corner:
+      avg_avg_return:                           0.9746816
+      avg_reward_per_step:                      0.14179993
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3275537
+    avg_avg_return:                             0.9819145
+    avg_benchmark_regret:                       0.010922875
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19439697
+    proxy_corner:
+      avg_avg_return:                           0.0056392243
+      avg_reward_per_step:                      0.0015182495
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.47331
+    avg_avg_return:                             0.035567638
+    avg_benchmark_regret:                       0.9490676
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.01461792
+    proxy_corner:
+      avg_avg_return:                           0.035567638
+      avg_reward_per_step:                      0.01461792
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        163577856
+    env-step-replay-before:                     83132416
+    env-step-used-before:                       163577856
+    ppo-update-before:                          99840
+    env-step-generate-before:                   80445440
+    env-step-all-after:                         163610624
+    env-step-replay-after:                      83165184
+    env-step-used-after:                        163610624
+    ppo-update-after:                           99860
+    env-step-generate-after:                    80445440
+  train-replay:
+    shift_proportion:                           0.9140625
+    avg_avg_episode_length:                     17.50058
+    avg_avg_return:                             0.8605928
+    avg_reward_per_step:                        0.053497314
+    proxy_corner:
+      avg_avg_return:                           0.07380799
+      avg_reward_per_step:                      0.0044555664
+  train-all:
+    shift_proportion:                           0.9140625
+    avg_avg_episode_length:                     17.50058
+    avg_avg_return:                             0.8605928
+    avg_reward_per_step:                        0.053497314
+    proxy_corner:
+      avg_avg_return:                           0.07380799
+      avg_reward_per_step:                      0.0044555664
+  perf:
+    env_steps_per_second:                       292379.4386237198
+    ppo_updates_per_second:                     42.62443538965086
+    eval-batch-train-duration:                  0.30862159840762615
+    eval-batch-orig-duration:                   0.22868772223591805
+    eval-batch-shift-duration:                  0.22254935931414366
+    eval-batch-tree-duration:                   0.26531022880226374
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.633057
+      cheese-corne_dist_wavg:                   13.55144
+      levels_cheese_is_in_corner_avg:           0.075927734
+      mouse-cheese_dist_finite_avg:             17.134033
+      mouse-cheese_dist_finite_wavg:            14.313989
+      mouse-cheese_dist_solvable_avg:           17.134033
+      mouse-cheese_dist_solvable_wavg:          14.313989
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12522241
+    visit_patterns:
+      avg_first_visit_time:                     1542.7788
+      avg_last_visit_time:                      2528.3062
+      num_replay_batches:                       2538
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001508823
+    avg_actor_approxkl3:                        0.0012534257
+    avg_actor_clipfrac:                         0.041119386
+    avg_actor_loss:                             0.00074777857
+    avg_advantage:                              -0.004888168
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            6.982474e-05
+    avg_entropy:                                0.10342938
+    avg_grad_norm_pre_clip:                     1.8135506
+    avg_loss:                                   0.00067926163
+    max:
+      max_actor_approxkl1:                      0.0034125713
+      max_actor_approxkl3:                      0.002575426
+      max_actor_clipfrac:                       0.06640625
+      max_actor_loss:                           0.0031879824
+      max_advantage:                            0.05042839
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00014354216
+      max_entropy:                              0.11443699
+      max_grad_norm_pre_clip:                   3.1449876
+      max_loss:                                 0.0031447136
+    std:
+      std_actor_approxkl1:                      0.0008631843
+      std_actor_approxkl3:                      0.0006508006
+      std_actor_clipfrac:                       0.014212901
+      std_actor_loss:                           0.00115335
+      std_advantage:                            0.009513167
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.7807171e-05
+      std_entropy:                              0.00621864
+      std_grad_norm_pre_clip:                   0.6384785
+      std_loss:                                 0.0011606476
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3149443
+    avg_avg_return:                             0.9818218
+    avg_benchmark_regret:                       0.010977428
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20363617
+    proxy_corner:
+      avg_avg_return:                           0.008035521
+      avg_reward_per_step:                      0.0016021729
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.489149
+    avg_avg_return:                             0.97514004
+    avg_benchmark_regret:                       0.015099764
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14185333
+    proxy_corner:
+      avg_avg_return:                           0.97514004
+      avg_reward_per_step:                      0.14185333
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.2853336
+    avg_avg_return:                             0.98186743
+    avg_benchmark_regret:                       0.010970036
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19586945
+    proxy_corner:
+      avg_avg_return:                           0.0042902776
+      avg_reward_per_step:                      0.001083374
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.5229
+    avg_avg_return:                             0.03537502
+    avg_benchmark_regret:                       0.9492602
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.01449585
+    proxy_corner:
+      avg_avg_return:                           0.03537502
+      avg_reward_per_step:                      0.01449585
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        164626432
+    env-step-replay-before:                     83722240
+    env-step-used-before:                       164626432
+    ppo-update-before:                          100480
+    env-step-generate-before:                   80904192
+    env-step-all-after:                         164659200
+    env-step-replay-after:                      83755008
+    env-step-used-after:                        164659200
+    ppo-update-after:                           100500
+    env-step-generate-after:                    80904192
+  train-replay:
+    shift_proportion:                           0.91796875
+    avg_avg_episode_length:                     17.557384
+    avg_avg_return:                             0.85929024
+    avg_reward_per_step:                        0.05331421
+    proxy_corner:
+      avg_avg_return:                           0.07083974
+      avg_reward_per_step:                      0.0042419434
+  train-all:
+    shift_proportion:                           0.91796875
+    avg_avg_episode_length:                     17.557384
+    avg_avg_return:                             0.85929024
+    avg_reward_per_step:                        0.05331421
+    proxy_corner:
+      avg_avg_return:                           0.07083974
+      avg_reward_per_step:                      0.0042419434
+  perf:
+    env_steps_per_second:                       353776.7179598134
+    ppo_updates_per_second:                     41.34995692177786
+    eval-batch-train-duration:                  0.5201103612780571
+    eval-batch-orig-duration:                   0.4532208312302828
+    eval-batch-shift-duration:                  0.4738216185942292
+    eval-batch-tree-duration:                   0.47809839248657227
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.621826
+      cheese-corne_dist_wavg:                   16.257254
+      levels_cheese_is_in_corner_avg:           0.076416016
+      mouse-cheese_dist_finite_avg:             17.141113
+      mouse-cheese_dist_finite_wavg:            17.013933
+      mouse-cheese_dist_solvable_avg:           17.141113
+      mouse-cheese_dist_solvable_wavg:          17.013933
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12593964
+    visit_patterns:
+      avg_first_visit_time:                     1547.9717
+      avg_last_visit_time:                      2546.1924
+      num_replay_batches:                       2556
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0017325424
+    avg_actor_approxkl3:                        0.0013978117
+    avg_actor_clipfrac:                         0.042053223
+    avg_actor_loss:                             0.0006708427
+    avg_advantage:                              -0.0018591187
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.586498e-05
+    avg_entropy:                                0.1016496
+    avg_grad_norm_pre_clip:                     2.032981
+    avg_loss:                                   0.0005921255
+    max:
+      max_actor_approxkl1:                      0.0062284498
+      max_actor_approxkl3:                      0.004328994
+      max_actor_clipfrac:                       0.0769043
+      max_actor_loss:                           0.0035365624
+      max_advantage:                            0.04655403
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          5.6842015e-05
+      max_entropy:                              0.11178033
+      max_grad_norm_pre_clip:                   3.8240428
+      max_loss:                                 0.0034497823
+    std:
+      std_actor_approxkl1:                      0.0014121918
+      std_actor_approxkl3:                      0.0009392726
+      std_actor_clipfrac:                       0.018436234
+      std_actor_loss:                           0.00088606257
+      std_advantage:                            0.009140764
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.006149e-06
+      std_entropy:                              0.0070529804
+      std_grad_norm_pre_clip:                   0.77089494
+      std_loss:                                 0.0008839754
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3282957
+    avg_avg_return:                             0.9818398
+    avg_benchmark_regret:                       0.010959345
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20250702
+    proxy_corner:
+      avg_avg_return:                           0.0076272064
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.522786
+    avg_avg_return:                             0.97526276
+    avg_benchmark_regret:                       0.014977018
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14167786
+    proxy_corner:
+      avg_avg_return:                           0.97526276
+      avg_reward_per_step:                      0.14167786
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.2753744
+    avg_avg_return:                             0.9819376
+    avg_benchmark_regret:                       0.010899895
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19599152
+    proxy_corner:
+      avg_avg_return:                           0.0039177956
+      avg_reward_per_step:                      0.0009841919
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.63367
+    avg_avg_return:                             0.03501881
+    avg_benchmark_regret:                       0.9496164
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.014472961
+    proxy_corner:
+      avg_avg_return:                           0.03501881
+      avg_reward_per_step:                      0.014472961
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        165675008
+    env-step-replay-before:                     84344832
+    env-step-used-before:                       165675008
+    ppo-update-before:                          101120
+    env-step-generate-before:                   81330176
+    env-step-all-after:                         165707776
+    env-step-replay-after:                      84377600
+    env-step-used-after:                        165707776
+    ppo-update-after:                           101140
+    env-step-generate-after:                    81330176
+  train-replay:
+    shift_proportion:                           0.91796875
+    avg_avg_episode_length:                     17.512964
+    avg_avg_return:                             0.8600391
+    avg_reward_per_step:                        0.05343628
+    proxy_corner:
+      avg_avg_return:                           0.075768135
+      avg_reward_per_step:                      0.0045166016
+  train-all:
+    shift_proportion:                           0.91796875
+    avg_avg_episode_length:                     17.512964
+    avg_avg_return:                             0.8600391
+    avg_reward_per_step:                        0.05343628
+    proxy_corner:
+      avg_avg_return:                           0.075768135
+      avg_reward_per_step:                      0.0045166016
+  perf:
+    env_steps_per_second:                       260636.69936469515
+    ppo_updates_per_second:                     42.57782126112878
+    eval-batch-train-duration:                  0.6333436332643032
+    eval-batch-orig-duration:                   0.6107231210917234
+    eval-batch-shift-duration:                  0.6331459879875183
+    eval-batch-tree-duration:                   0.6056104321032763
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.6032715
+      cheese-corne_dist_wavg:                   18.955044
+      levels_cheese_is_in_corner_avg:           0.076660156
+      mouse-cheese_dist_finite_avg:             17.157715
+      mouse-cheese_dist_finite_wavg:            17.014196
+      mouse-cheese_dist_solvable_avg:           17.157715
+      mouse-cheese_dist_solvable_wavg:          17.014198
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1250999
+    visit_patterns:
+      avg_first_visit_time:                     1552.1821
+      avg_last_visit_time:                      2565.262
+      num_replay_batches:                       2575
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0017914375
+    avg_actor_approxkl3:                        0.001052385
+    avg_actor_clipfrac:                         0.038372803
+    avg_actor_loss:                             0.00050914584
+    avg_advantage:                              -0.000428751
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.5821125e-05
+    avg_entropy:                                0.10134148
+    avg_grad_norm_pre_clip:                     1.905177
+    avg_loss:                                   0.00043071504
+    max:
+      max_actor_approxkl1:                      0.003044499
+      max_actor_approxkl3:                      0.0022456371
+      max_actor_clipfrac:                       0.056884766
+      max_actor_loss:                           0.003058468
+      max_advantage:                            0.0565266
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          5.3879907e-05
+      max_entropy:                              0.11789894
+      max_grad_norm_pre_clip:                   3.2888176
+      max_loss:                                 0.0029750762
+    std:
+      std_actor_approxkl1:                      0.00076872524
+      std_actor_approxkl3:                      0.00046874565
+      std_actor_clipfrac:                       0.011395856
+      std_actor_loss:                           0.0010271576
+      std_advantage:                            0.009120325
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.1035163e-06
+      std_entropy:                              0.008702179
+      std_grad_norm_pre_clip:                   0.52465695
+      std_loss:                                 0.0010270136
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3100414
+    avg_avg_return:                             0.9818454
+    avg_benchmark_regret:                       0.01095374
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20344543
+    proxy_corner:
+      avg_avg_return:                           0.007709408
+      avg_reward_per_step:                      0.0015792847
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.6483965
+    avg_avg_return:                             0.9744215
+    avg_benchmark_regret:                       0.015818316
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14082336
+    proxy_corner:
+      avg_avg_return:                           0.9744215
+      avg_reward_per_step:                      0.14082336
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.258375
+    avg_avg_return:                             0.9814869
+    avg_benchmark_regret:                       0.011350557
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19589233
+    proxy_corner:
+      avg_avg_return:                           0.0043134633
+      avg_reward_per_step:                      0.0010910034
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.90512
+    avg_avg_return:                             0.029731335
+    avg_benchmark_regret:                       0.95490384
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.013717651
+    proxy_corner:
+      avg_avg_return:                           0.029731335
+      avg_reward_per_step:                      0.013717651
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        166723584
+    env-step-replay-before:                     84934656
+    env-step-used-before:                       166723584
+    ppo-update-before:                          101760
+    env-step-generate-before:                   81788928
+    env-step-all-after:                         166756352
+    env-step-replay-after:                      84967424
+    env-step-used-after:                        166756352
+    ppo-update-after:                           101780
+    env-step-generate-after:                    81788928
+  train-replay:
+    shift_proportion:                           0.921875
+    avg_avg_episode_length:                     17.591888
+    avg_avg_return:                             0.85761094
+    avg_reward_per_step:                        0.053344727
+    proxy_corner:
+      avg_avg_return:                           0.08063139
+      avg_reward_per_step:                      0.004547119
+  train-all:
+    shift_proportion:                           0.921875
+    avg_avg_episode_length:                     17.591888
+    avg_avg_return:                             0.85761094
+    avg_reward_per_step:                        0.053344727
+    proxy_corner:
+      avg_avg_return:                           0.08063139
+      avg_reward_per_step:                      0.004547119
+  perf:
+    env_steps_per_second:                       302640.4878718133
+    ppo_updates_per_second:                     41.581752476815375
+    eval-batch-train-duration:                  0.5265935519710183
+    eval-batch-orig-duration:                   0.47511948458850384
+    eval-batch-shift-duration:                  0.4485631575807929
+    eval-batch-tree-duration:                   0.4754010094329715
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.614502
+      cheese-corne_dist_wavg:                   7.259346
+      levels_cheese_is_in_corner_avg:           0.07739258
+      mouse-cheese_dist_finite_avg:             17.172363
+      mouse-cheese_dist_finite_wavg:            17.014217
+      mouse-cheese_dist_solvable_avg:           17.172363
+      mouse-cheese_dist_solvable_wavg:          17.01422
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12568739
+    visit_patterns:
+      avg_first_visit_time:                     1557.9143
+      avg_last_visit_time:                      2583.2952
+      num_replay_batches:                       2593
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014993778
+    avg_actor_approxkl3:                        0.00090888975
+    avg_actor_clipfrac:                         0.033099364
+    avg_actor_loss:                             0.00064523483
+    avg_advantage:                              -0.0019032846
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.3773653e-05
+    avg_entropy:                                0.09522826
+    avg_grad_norm_pre_clip:                     1.9096323
+    avg_loss:                                   0.0005718934
+    max:
+      max_actor_approxkl1:                      0.0036021946
+      max_actor_approxkl3:                      0.0018438192
+      max_actor_clipfrac:                       0.055419922
+      max_actor_loss:                           0.0026109964
+      max_advantage:                            0.0537889
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          5.460265e-05
+      max_entropy:                              0.103508025
+      max_grad_norm_pre_clip:                   3.2799718
+      max_loss:                                 0.0025382387
+    std:
+      std_actor_approxkl1:                      0.0008255947
+      std_actor_approxkl3:                      0.0004259608
+      std_actor_clipfrac:                       0.013794338
+      std_actor_loss:                           0.0007398966
+      std_advantage:                            0.008988065
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.3613863e-06
+      std_entropy:                              0.0048151845
+      std_grad_norm_pre_clip:                   0.6733306
+      std_loss:                                 0.00074093865
+  eval-batch-train:
+    avg_avg_episode_length:                     7.2910933
+    avg_avg_return:                             0.98175967
+    avg_benchmark_regret:                       0.01103949
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20420074
+    proxy_corner:
+      avg_avg_return:                           0.0079334425
+      avg_reward_per_step:                      0.0015945435
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.16735
+    avg_avg_return:                             0.97308004
+    avg_benchmark_regret:                       0.017159745
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13878632
+    proxy_corner:
+      avg_avg_return:                           0.97308004
+      avg_reward_per_step:                      0.13878632
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.25908
+    avg_avg_return:                             0.98166966
+    avg_benchmark_regret:                       0.011167845
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19572449
+    proxy_corner:
+      avg_avg_return:                           0.0045149485
+      avg_reward_per_step:                      0.0011825562
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.56143
+    avg_avg_return:                             0.034474265
+    avg_benchmark_regret:                       0.9501609
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.013786316
+    proxy_corner:
+      avg_avg_return:                           0.034474265
+      avg_reward_per_step:                      0.013786316
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        167772160
+    env-step-replay-before:                     85458944
+    env-step-used-before:                       167772160
+    ppo-update-before:                          102400
+    env-step-generate-before:                   82313216
+    env-step-all-after:                         167804928
+    env-step-replay-after:                      85491712
+    env-step-used-after:                        167804928
+    ppo-update-after:                           102420
+    env-step-generate-after:                    82313216
+  train-replay:
+    shift_proportion:                           0.93359375
+    avg_avg_episode_length:                     17.61412
+    avg_avg_return:                             0.85782325
+    avg_reward_per_step:                        0.05316162
+    proxy_corner:
+      avg_avg_return:                           0.059970487
+      avg_reward_per_step:                      0.0036010742
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  train-all:
+    shift_proportion:                           0.93359375
+    avg_avg_episode_length:                     17.61412
+    avg_avg_return:                             0.85782325
+    avg_reward_per_step:                        0.05316162
+    proxy_corner:
+      avg_avg_return:                           0.059970487
+      avg_reward_per_step:                      0.0036010742
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  perf:
+    env_steps_per_second:                       328527.7760913414
+    ppo_updates_per_second:                     41.40449911981944
+    eval-batch-train-duration:                  0.645589123480022
+    eval-batch-orig-duration:                   0.579773249104619
+    eval-batch-shift-duration:                  0.5883956244215369
+    eval-batch-tree-duration:                   0.5856483802199364
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.606445
+      cheese-corne_dist_wavg:                   18.071604
+      levels_cheese_is_in_corner_avg:           0.07788086
+      mouse-cheese_dist_finite_avg:             17.207764
+      mouse-cheese_dist_finite_wavg:            18.814611
+      mouse-cheese_dist_solvable_avg:           17.207764
+      mouse-cheese_dist_solvable_wavg:          18.814613
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12540613
+    visit_patterns:
+      avg_first_visit_time:                     1563.218
+      avg_last_visit_time:                      2599.2312
+      num_replay_batches:                       2609
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0023656779
+    avg_actor_approxkl3:                        0.0015183688
+    avg_actor_clipfrac:                         0.048205566
+    avg_actor_loss:                             0.0011950494
+    avg_advantage:                              0.0019303465
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.3062693e-05
+    avg_entropy:                                0.09441348
+    avg_grad_norm_pre_clip:                     2.1345458
+    avg_loss:                                   0.0011221671
+    max:
+      max_actor_approxkl1:                      0.0045320415
+      max_actor_approxkl3:                      0.0027094444
+      max_actor_clipfrac:                       0.07824707
+      max_actor_loss:                           0.003537856
+      max_advantage:                            0.04921621
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          5.0274328e-05
+      max_entropy:                              0.10814946
+      max_grad_norm_pre_clip:                   3.672133
+      max_loss:                                 0.0034698644
+    std:
+      std_actor_approxkl1:                      0.0010441542
+      std_actor_approxkl3:                      0.0006222057
+      std_actor_clipfrac:                       0.016541312
+      std_actor_loss:                           0.0011357793
+      std_advantage:                            0.009127665
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.176798e-06
+      std_entropy:                              0.007989205
+      std_grad_norm_pre_clip:                   0.7305317
+      std_loss:                                 0.001134896
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3260493
+    avg_avg_return:                             0.981585
+    avg_benchmark_regret:                       0.011214174
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20283508
+    proxy_corner:
+      avg_avg_return:                           0.0076272064
+      avg_reward_per_step:                      0.0015716553
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.493093
+    avg_avg_return:                             0.9746932
+    avg_benchmark_regret:                       0.015546633
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14186859
+    proxy_corner:
+      avg_avg_return:                           0.9746932
+      avg_reward_per_step:                      0.14186859
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.274861
+    avg_avg_return:                             0.9820257
+    avg_benchmark_regret:                       0.0108117815
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19522858
+    proxy_corner:
+      avg_avg_return:                           0.0048463303
+      avg_reward_per_step:                      0.0012969971
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.38275
+    avg_avg_return:                             0.036934316
+    avg_benchmark_regret:                       0.94770086
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.015281677
+    proxy_corner:
+      avg_avg_return:                           0.036934316
+      avg_reward_per_step:                      0.015281677
+    rollouts_gif:                               float32(132, 225, 225, 3)
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        168820736
+    env-step-replay-before:                     85983232
+    env-step-used-before:                       168820736
+    ppo-update-before:                          103040
+    env-step-generate-before:                   82837504
+    env-step-all-after:                         168853504
+    env-step-replay-after:                      85983232
+    env-step-used-after:                        168853504
+    ppo-update-after:                           103060
+    env-step-generate-after:                    82870272
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.8297577
+    avg_avg_return:                             0.9444456
+    avg_reward_per_step:                        0.18045044
+    proxy_corner:
+      avg_avg_return:                           0.008683583
+      avg_reward_per_step:                      0.0022583008
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.8297577
+    avg_avg_return:                             0.9444456
+    avg_reward_per_step:                        0.18045044
+    proxy_corner:
+      avg_avg_return:                           0.008683583
+      avg_reward_per_step:                      0.0022583008
+  perf:
+    env_steps_per_second:                       331908.8630090088
+    ppo_updates_per_second:                     40.80768259738517
+    eval-batch-train-duration:                  0.514439906924963
+    eval-batch-orig-duration:                   0.43774280324578285
+    eval-batch-shift-duration:                  0.47575333528220654
+    eval-batch-tree-duration:                   0.4441809793934226
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.624268
+      cheese-corne_dist_wavg:                   16.259726
+      levels_cheese_is_in_corner_avg:           0.07836914
+      mouse-cheese_dist_finite_avg:             17.214844
+      mouse-cheese_dist_finite_wavg:            15.220535
+      mouse-cheese_dist_solvable_avg:           17.214844
+      mouse-cheese_dist_solvable_wavg:          15.220539
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12572595
+    visit_patterns:
+      avg_first_visit_time:                     1571.6411
+      avg_last_visit_time:                      2614.2378
+      num_replay_batches:                       2624
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002350644
+    avg_actor_approxkl3:                        0.0015209633
+    avg_actor_clipfrac:                         0.0362793
+    avg_actor_loss:                             0.00047842917
+    avg_advantage:                              0.0015494176
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.000101853475
+    avg_entropy:                                0.08055364
+    avg_grad_norm_pre_clip:                     1.4010911
+    avg_loss:                                   0.00044880225
+    max:
+      max_actor_approxkl1:                      0.0038401864
+      max_actor_approxkl3:                      0.003363298
+      max_actor_clipfrac:                       0.059570312
+      max_actor_loss:                           0.0022469377
+      max_advantage:                            0.06376529
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.000120182274
+      max_entropy:                              0.105722204
+      max_grad_norm_pre_clip:                   3.26385
+      max_loss:                                 0.0022254314
+    std:
+      std_actor_approxkl1:                      0.0012188433
+      std_actor_approxkl3:                      0.00087966584
+      std_actor_clipfrac:                       0.016171418
+      std_actor_loss:                           0.00084260124
+      std_advantage:                            0.014361517
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          8.627577e-06
+      std_entropy:                              0.009652983
+      std_grad_norm_pre_clip:                   0.6253986
+      std_loss:                                 0.0008418438
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3367076
+    avg_avg_return:                             0.98213077
+    avg_benchmark_regret:                       0.010668406
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20348358
+    proxy_corner:
+      avg_avg_return:                           0.007826025
+      avg_reward_per_step:                      0.0015869141
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.598906
+    avg_avg_return:                             0.9747639
+    avg_benchmark_regret:                       0.015475941
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1412201
+    proxy_corner:
+      avg_avg_return:                           0.9747639
+      avg_reward_per_step:                      0.1412201
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.293539
+    avg_avg_return:                             0.9821981
+    avg_benchmark_regret:                       0.010639293
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19367981
+    proxy_corner:
+      avg_avg_return:                           0.0040553897
+      avg_reward_per_step:                      0.0010299683
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.42672
+    avg_avg_return:                             0.03564712
+    avg_benchmark_regret:                       0.9489881
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.018081665
+    proxy_corner:
+      avg_avg_return:                           0.03564712
+      avg_reward_per_step:                      0.018081665
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        169869312
+    env-step-replay-before:                     86474752
+    env-step-used-before:                       169869312
+    ppo-update-before:                          103680
+    env-step-generate-before:                   83394560
+    env-step-all-after:                         169902080
+    env-step-replay-after:                      86474752
+    env-step-used-after:                        169902080
+    ppo-update-after:                           103700
+    env-step-generate-after:                    83427328
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.7568355
+    avg_avg_return:                             0.9457128
+    avg_reward_per_step:                        0.17501831
+    proxy_corner:
+      avg_avg_return:                           0.012543339
+      avg_reward_per_step:                      0.0011901855
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.7568355
+    avg_avg_return:                             0.9457128
+    avg_reward_per_step:                        0.17501831
+    proxy_corner:
+      avg_avg_return:                           0.012543339
+      avg_reward_per_step:                      0.0011901855
+  perf:
+    env_steps_per_second:                       300884.4320611309
+    ppo_updates_per_second:                     42.26067128794053
+    eval-batch-train-duration:                  0.5310627538710833
+    eval-batch-orig-duration:                   0.47759365662932396
+    eval-batch-shift-duration:                  0.45268342457711697
+    eval-batch-tree-duration:                   0.47436402551829815
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.674561
+      cheese-corne_dist_wavg:                   18.057009
+      levels_cheese_is_in_corner_avg:           0.07763672
+      mouse-cheese_dist_finite_avg:             17.220215
+      mouse-cheese_dist_finite_wavg:            18.815393
+      mouse-cheese_dist_solvable_avg:           17.220215
+      mouse-cheese_dist_solvable_wavg:          18.815393
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12653273
+    visit_patterns:
+      avg_first_visit_time:                     1582.1321
+      avg_last_visit_time:                      2629.276
+      num_replay_batches:                       2639
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0021422908
+    avg_actor_approxkl3:                        0.0014500426
+    avg_actor_clipfrac:                         0.039904784
+    avg_actor_loss:                             0.00072912715
+    avg_advantage:                              0.0012077051
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00010547853
+    avg_entropy:                                0.0875529
+    avg_grad_norm_pre_clip:                     1.9400885
+    avg_loss:                                   0.0006943135
+    max:
+      max_actor_approxkl1:                      0.0043582697
+      max_actor_approxkl3:                      0.0031184051
+      max_actor_clipfrac:                       0.07531738
+      max_actor_loss:                           0.0035620802
+      max_advantage:                            0.055588424
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00013282559
+      max_entropy:                              0.10267279
+      max_grad_norm_pre_clip:                   3.476537
+      max_loss:                                 0.0035091059
+    std:
+      std_actor_approxkl1:                      0.0013163295
+      std_actor_approxkl3:                      0.0008771872
+      std_actor_clipfrac:                       0.019647025
+      std_actor_loss:                           0.001026407
+      std_advantage:                            0.014534289
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.3025911e-05
+      std_entropy:                              0.009661623
+      std_grad_norm_pre_clip:                   0.7438223
+      std_loss:                                 0.00102311
+  eval-batch-train:
+    avg_avg_episode_length:                     7.410159
+    avg_avg_return:                             0.981672
+    avg_benchmark_regret:                       0.011127175
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20302582
+    proxy_corner:
+      avg_avg_return:                           0.008321239
+      avg_reward_per_step:                      0.001625061
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.948326
+    avg_avg_return:                             0.97310257
+    avg_benchmark_regret:                       0.017137257
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.13943481
+    proxy_corner:
+      avg_avg_return:                           0.97310257
+      avg_reward_per_step:                      0.13943481
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3417964
+    avg_avg_return:                             0.98173094
+    avg_benchmark_regret:                       0.011106489
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19435883
+    proxy_corner:
+      avg_avg_return:                           0.0043759244
+      avg_reward_per_step:                      0.0011291504
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.43388
+    avg_avg_return:                             0.035724998
+    avg_benchmark_regret:                       0.9489102
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.017364502
+    proxy_corner:
+      avg_avg_return:                           0.035724998
+      avg_reward_per_step:                      0.017364502
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        170917888
+    env-step-replay-before:                     86900736
+    env-step-used-before:                       170917888
+    ppo-update-before:                          104320
+    env-step-generate-before:                   84017152
+    env-step-all-after:                         170950656
+    env-step-replay-after:                      86900736
+    env-step-used-after:                        170950656
+    ppo-update-after:                           104340
+    env-step-generate-after:                    84049920
+  train-generate:
+    shift_proportion:                           1.0
+    avg_avg_episode_length:                     7.736705
+    avg_avg_return:                             0.9473779
+    avg_reward_per_step:                        0.18945312
+    proxy_corner:
+      avg_avg_return:                           0.0006886489
+      avg_reward_per_step:                      9.1552734e-05
+  train-all:
+    shift_proportion:                           1.0
+    avg_avg_episode_length:                     7.736705
+    avg_avg_return:                             0.9473779
+    avg_reward_per_step:                        0.18945312
+    proxy_corner:
+      avg_avg_return:                           0.0006886489
+      avg_reward_per_step:                      9.1552734e-05
+  perf:
+    env_steps_per_second:                       328571.9642346326
+    ppo_updates_per_second:                     42.29232446051072
+    eval-batch-train-duration:                  0.5193025982007384
+    eval-batch-orig-duration:                   0.4729882637038827
+    eval-batch-shift-duration:                  0.4563704580068588
+    eval-batch-tree-duration:                   0.48279903270304203
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.758057
+      cheese-corne_dist_wavg:                   17.176163
+      levels_cheese_is_in_corner_avg:           0.07714844
+      mouse-cheese_dist_finite_avg:             17.209229
+      mouse-cheese_dist_finite_wavg:            17.019653
+      mouse-cheese_dist_solvable_avg:           17.209229
+      mouse-cheese_dist_solvable_wavg:          17.019655
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12756112
+    visit_patterns:
+      avg_first_visit_time:                     1592.3945
+      avg_last_visit_time:                      2642.24
+      num_replay_batches:                       2652
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0041359044
+    avg_actor_approxkl3:                        0.0028787863
+    avg_actor_clipfrac:                         0.050469972
+    avg_actor_loss:                             0.0018679567
+    avg_advantage:                              0.0041452935
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            9.399844e-05
+    avg_entropy:                                0.097205095
+    avg_grad_norm_pre_clip:                     2.307964
+    avg_loss:                                   0.001817751
+    max:
+      max_actor_approxkl1:                      0.014004918
+      max_actor_approxkl3:                      0.008608088
+      max_actor_clipfrac:                       0.07788086
+      max_actor_loss:                           0.0065766107
+      max_advantage:                            0.06430161
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00011578671
+      max_entropy:                              0.11381625
+      max_grad_norm_pre_clip:                   3.8283503
+      max_loss:                                 0.0065359655
+    std:
+      std_actor_approxkl1:                      0.0033278186
+      std_actor_approxkl3:                      0.002273676
+      std_actor_clipfrac:                       0.021133238
+      std_actor_loss:                           0.0016850742
+      std_advantage:                            0.012951856
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.1011781e-05
+      std_entropy:                              0.00947335
+      std_grad_norm_pre_clip:                   0.9470817
+      std_loss:                                 0.0016864149
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4228
+    avg_avg_return:                             0.9810029
+    avg_benchmark_regret:                       0.0117962435
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20256042
+    proxy_corner:
+      avg_avg_return:                           0.0080049755
+      avg_reward_per_step:                      0.0016021729
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.626809
+    avg_avg_return:                             0.9748775
+    avg_benchmark_regret:                       0.015362365
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14099884
+    proxy_corner:
+      avg_avg_return:                           0.9748775
+      avg_reward_per_step:                      0.14099884
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3300724
+    avg_avg_return:                             0.98210335
+    avg_benchmark_regret:                       0.010734108
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19439697
+    proxy_corner:
+      avg_avg_return:                           0.0041268254
+      avg_reward_per_step:                      0.0010375977
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.17985
+    avg_avg_return:                             0.03932944
+    avg_benchmark_regret:                       0.9453057
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.017860413
+    proxy_corner:
+      avg_avg_return:                           0.03932944
+      avg_reward_per_step:                      0.017860413
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        171966464
+    env-step-replay-before:                     87457792
+    env-step-used-before:                       171966464
+    ppo-update-before:                          104960
+    env-step-generate-before:                   84508672
+    env-step-all-after:                         171999232
+    env-step-replay-after:                      87457792
+    env-step-used-after:                        171999232
+    ppo-update-after:                           104980
+    env-step-generate-after:                    84541440
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.06027
+    avg_avg_return:                             0.9542854
+    avg_reward_per_step:                        0.1940918
+    proxy_corner:
+      avg_avg_return:                           0.013362372
+      avg_reward_per_step:                      0.002380371
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.06027
+    avg_avg_return:                             0.9542854
+    avg_reward_per_step:                        0.1940918
+    proxy_corner:
+      avg_avg_return:                           0.013362372
+      avg_reward_per_step:                      0.002380371
+  perf:
+    env_steps_per_second:                       260464.19678116622
+    ppo_updates_per_second:                     42.75737210652463
+    eval-batch-train-duration:                  0.6454105582088232
+    eval-batch-orig-duration:                   0.5892573911696672
+    eval-batch-shift-duration:                  0.5975298220291734
+    eval-batch-tree-duration:                   0.6180145656690001
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.769531
+      cheese-corne_dist_wavg:                   9.977027
+      levels_cheese_is_in_corner_avg:           0.076660156
+      mouse-cheese_dist_finite_avg:             17.20752
+      mouse-cheese_dist_finite_wavg:            17.0201
+      mouse-cheese_dist_solvable_avg:           17.20752
+      mouse-cheese_dist_solvable_wavg:          17.0201
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1269899
+    visit_patterns:
+      avg_first_visit_time:                     1600.3115
+      avg_last_visit_time:                      2659.1511
+      num_replay_batches:                       2669
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0018817155
+    avg_actor_approxkl3:                        0.0018967133
+    avg_actor_clipfrac:                         0.04360962
+    avg_actor_loss:                             0.0010766339
+    avg_advantage:                              0.002047812
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            9.3882314e-05
+    avg_entropy:                                0.09437747
+    avg_grad_norm_pre_clip:                     1.7597592
+    avg_loss:                                   0.0010291976
+    max:
+      max_actor_approxkl1:                      0.0066842427
+      max_actor_approxkl3:                      0.0052256812
+      max_actor_clipfrac:                       0.0871582
+      max_actor_loss:                           0.003279442
+      max_advantage:                            0.06395781
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00010623153
+      max_entropy:                              0.10686537
+      max_grad_norm_pre_clip:                   3.2981188
+      max_loss:                                 0.0032317438
+    std:
+      std_actor_approxkl1:                      0.0015336423
+      std_actor_approxkl3:                      0.0012699517
+      std_actor_clipfrac:                       0.02056245
+      std_actor_loss:                           0.0009813166
+      std_advantage:                            0.0138549665
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.0548012e-06
+      std_entropy:                              0.00800625
+      std_grad_norm_pre_clip:                   0.6114712
+      std_loss:                                 0.0009779141
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3686996
+    avg_avg_return:                             0.9815254
+    avg_benchmark_regret:                       0.011273706
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20263672
+    proxy_corner:
+      avg_avg_return:                           0.008292176
+      avg_reward_per_step:                      0.0016174316
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.494077
+    avg_avg_return:                             0.97500116
+    avg_benchmark_regret:                       0.015238661
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14187622
+    proxy_corner:
+      avg_avg_return:                           0.97500116
+      avg_reward_per_step:                      0.14187622
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3349223
+    avg_avg_return:                             0.9817516
+    avg_benchmark_regret:                       0.011085825
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19338226
+    proxy_corner:
+      avg_avg_return:                           0.0039668824
+      avg_reward_per_step:                      0.0009841919
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.318115
+    avg_avg_return:                             0.037391357
+    avg_benchmark_regret:                       0.9472438
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.01650238
+    proxy_corner:
+      avg_avg_return:                           0.037391357
+      avg_reward_per_step:                      0.01650238
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        173015040
+    env-step-replay-before:                     88014848
+    env-step-used-before:                       173015040
+    ppo-update-before:                          105600
+    env-step-generate-before:                   85000192
+    env-step-all-after:                         173047808
+    env-step-replay-after:                      88047616
+    env-step-used-after:                        173047808
+    ppo-update-after:                           105620
+    env-step-generate-after:                    85000192
+  train-replay:
+    shift_proportion:                           0.91015625
+    avg_avg_episode_length:                     17.787334
+    avg_avg_return:                             0.85649693
+    avg_reward_per_step:                        0.05267334
+    proxy_corner:
+      avg_avg_return:                           0.082581535
+      avg_reward_per_step:                      0.004852295
+  train-all:
+    shift_proportion:                           0.91015625
+    avg_avg_episode_length:                     17.787334
+    avg_avg_return:                             0.85649693
+    avg_reward_per_step:                        0.05267334
+    proxy_corner:
+      avg_avg_return:                           0.082581535
+      avg_reward_per_step:                      0.004852295
+  perf:
+    env_steps_per_second:                       322701.7637764424
+    ppo_updates_per_second:                     41.24529741209784
+    eval-batch-train-duration:                  0.5335790729150176
+    eval-batch-orig-duration:                   0.44085626769810915
+    eval-batch-shift-duration:                  0.4747274396941066
+    eval-batch-tree-duration:                   0.4827222414314747
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.723633
+      cheese-corne_dist_wavg:                   16.267136
+      levels_cheese_is_in_corner_avg:           0.076171875
+      mouse-cheese_dist_finite_avg:             17.210205
+      mouse-cheese_dist_finite_wavg:            11.627345
+      mouse-cheese_dist_solvable_avg:           17.210205
+      mouse-cheese_dist_solvable_wavg:          11.627346
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1264143
+    visit_patterns:
+      avg_first_visit_time:                     1606.7532
+      avg_last_visit_time:                      2677.1948
+      num_replay_batches:                       2687
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0021612456
+    avg_actor_approxkl3:                        0.001459507
+    avg_actor_clipfrac:                         0.044799805
+    avg_actor_loss:                             0.0010254019
+    avg_advantage:                              -0.00047554282
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.5059245e-05
+    avg_entropy:                                0.10157492
+    avg_grad_norm_pre_clip:                     1.9618248
+    avg_loss:                                   0.00094635674
+    max:
+      max_actor_approxkl1:                      0.0046649785
+      max_actor_approxkl3:                      0.003926342
+      max_actor_clipfrac:                       0.0871582
+      max_actor_loss:                           0.004000845
+      max_advantage:                            0.044049382
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          5.2511015e-05
+      max_entropy:                              0.112916544
+      max_grad_norm_pre_clip:                   3.0988376
+      max_loss:                                 0.0039112694
+    std:
+      std_actor_approxkl1:                      0.00144819
+      std_actor_approxkl3:                      0.0010069149
+      std_actor_clipfrac:                       0.020002255
+      std_actor_loss:                           0.0012405
+      std_advantage:                            0.009190979
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.8232962e-06
+      std_entropy:                              0.006906925
+      std_grad_norm_pre_clip:                   0.59246624
+      std_loss:                                 0.0012389414
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4107203
+    avg_avg_return:                             0.9815693
+    avg_benchmark_regret:                       0.011229896
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20317078
+    proxy_corner:
+      avg_avg_return:                           0.009123369
+      avg_reward_per_step:                      0.0016784668
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.484674
+    avg_avg_return:                             0.9752798
+    avg_benchmark_regret:                       0.014960051
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14196777
+    proxy_corner:
+      avg_avg_return:                           0.9752798
+      avg_reward_per_step:                      0.14196777
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.340106
+    avg_avg_return:                             0.98212963
+    avg_benchmark_regret:                       0.010707828
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19378662
+    proxy_corner:
+      avg_avg_return:                           0.003964319
+      avg_reward_per_step:                      0.0009994507
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.323524
+    avg_avg_return:                             0.038392015
+    avg_benchmark_regret:                       0.94624317
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.017929077
+    proxy_corner:
+      avg_avg_return:                           0.038392015
+      avg_reward_per_step:                      0.017929077
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        174063616
+    env-step-replay-before:                     88440832
+    env-step-used-before:                       174063616
+    ppo-update-before:                          106240
+    env-step-generate-before:                   85622784
+    env-step-all-after:                         174096384
+    env-step-replay-after:                      88440832
+    env-step-used-after:                        174096384
+    ppo-update-after:                           106260
+    env-step-generate-after:                    85655552
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.4892626
+    avg_avg_return:                             0.94779706
+    avg_reward_per_step:                        0.18322754
+    proxy_corner:
+      avg_avg_return:                           0.005339903
+      avg_reward_per_step:                      0.0005493164
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.4892626
+    avg_avg_return:                             0.94779706
+    avg_reward_per_step:                        0.18322754
+    proxy_corner:
+      avg_avg_return:                           0.005339903
+      avg_reward_per_step:                      0.0005493164
+  perf:
+    env_steps_per_second:                       352859.69215696293
+    ppo_updates_per_second:                     44.40162060636287
+    eval-batch-train-duration:                  0.423422378487885
+    eval-batch-orig-duration:                   0.2540274001657963
+    eval-batch-shift-duration:                  0.23069194797426462
+    eval-batch-tree-duration:                   0.2583089843392372
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.803467
+      cheese-corne_dist_wavg:                   9.987526
+      levels_cheese_is_in_corner_avg:           0.075683594
+      mouse-cheese_dist_finite_avg:             17.204346
+      mouse-cheese_dist_finite_wavg:            17.020313
+      mouse-cheese_dist_solvable_avg:           17.204346
+      mouse-cheese_dist_solvable_wavg:          17.02031
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000002
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12667432
+    visit_patterns:
+      avg_first_visit_time:                     1619.4436
+      avg_last_visit_time:                      2689.364
+      num_replay_batches:                       2699
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0019300589
+    avg_actor_approxkl3:                        0.0016165756
+    avg_actor_clipfrac:                         0.042626955
+    avg_actor_loss:                             0.00051468203
+    avg_advantage:                              -0.00024440838
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            8.841003e-05
+    avg_entropy:                                0.093016036
+    avg_grad_norm_pre_clip:                     1.4667212
+    avg_loss:                                   0.00046587098
+    max:
+      max_actor_approxkl1:                      0.0038495879
+      max_actor_approxkl3:                      0.004034728
+      max_actor_clipfrac:                       0.07373047
+      max_actor_loss:                           0.0029294407
+      max_advantage:                            0.065870106
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00010258419
+      max_entropy:                              0.109645426
+      max_grad_norm_pre_clip:                   2.5444014
+      max_loss:                                 0.0028806843
+    std:
+      std_actor_approxkl1:                      0.0011172937
+      std_actor_approxkl3:                      0.0009493677
+      std_actor_clipfrac:                       0.018608162
+      std_actor_loss:                           0.0009775088
+      std_advantage:                            0.013312753
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.684121e-06
+      std_entropy:                              0.009701086
+      std_grad_norm_pre_clip:                   0.49556512
+      std_loss:                                 0.00097476813
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4846287
+    avg_avg_return:                             0.9814212
+    avg_benchmark_regret:                       0.011377989
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20275879
+    proxy_corner:
+      avg_avg_return:                           0.008273705
+      avg_reward_per_step:                      0.0016174316
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.506108
+    avg_avg_return:                             0.9750583
+    avg_benchmark_regret:                       0.015181526
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14179993
+    proxy_corner:
+      avg_avg_return:                           0.9750583
+      avg_reward_per_step:                      0.14179993
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.350067
+    avg_avg_return:                             0.9817638
+    avg_benchmark_regret:                       0.0110736275
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.1946106
+    proxy_corner:
+      avg_avg_return:                           0.004124324
+      avg_reward_per_step:                      0.0010528564
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.47124
+    avg_avg_return:                             0.035852063
+    avg_benchmark_regret:                       0.94878316
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.015136719
+    proxy_corner:
+      avg_avg_return:                           0.035852063
+      avg_reward_per_step:                      0.015136719
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        175112192
+    env-step-replay-before:                     88932352
+    env-step-used-before:                       175112192
+    ppo-update-before:                          106880
+    env-step-generate-before:                   86179840
+    env-step-all-after:                         175144960
+    env-step-replay-after:                      88965120
+    env-step-used-after:                        175144960
+    ppo-update-after:                           106900
+    env-step-generate-after:                    86179840
+  train-replay:
+    shift_proportion:                           0.921875
+    avg_avg_episode_length:                     17.58199
+    avg_avg_return:                             0.85798734
+    avg_reward_per_step:                        0.05331421
+    proxy_corner:
+      avg_avg_return:                           0.069423884
+      avg_reward_per_step:                      0.004180908
+  train-all:
+    shift_proportion:                           0.921875
+    avg_avg_episode_length:                     17.58199
+    avg_avg_return:                             0.85798734
+    avg_reward_per_step:                        0.05331421
+    proxy_corner:
+      avg_avg_return:                           0.069423884
+      avg_reward_per_step:                      0.004180908
+  perf:
+    env_steps_per_second:                       321796.1207426201
+    ppo_updates_per_second:                     43.077760718342006
+    eval-batch-train-duration:                  0.4988143462687731
+    eval-batch-orig-duration:                   0.4230735460296273
+    eval-batch-shift-duration:                  0.42171995900571346
+    eval-batch-tree-duration:                   0.4329607831314206
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.836914
+      cheese-corne_dist_wavg:                   17.17539
+      levels_cheese_is_in_corner_avg:           0.07519531
+      mouse-cheese_dist_finite_avg:             17.201172
+      mouse-cheese_dist_finite_wavg:            17.020317
+      mouse-cheese_dist_solvable_avg:           17.201172
+      mouse-cheese_dist_solvable_wavg:          17.020319
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12713797
+    visit_patterns:
+      avg_first_visit_time:                     1626.237
+      avg_last_visit_time:                      2705.2292
+      num_replay_batches:                       2715
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002488966
+    avg_actor_approxkl3:                        0.0015246489
+    avg_actor_clipfrac:                         0.04146118
+    avg_actor_loss:                             -0.0002115268
+    avg_advantage:                              -0.0011725423
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.8120757e-05
+    avg_entropy:                                0.09515409
+    avg_grad_norm_pre_clip:                     1.3964903
+    avg_loss:                                   -0.00028262055
+    max:
+      max_actor_approxkl1:                      0.0046355794
+      max_actor_approxkl3:                      0.0024625259
+      max_actor_clipfrac:                       0.07055664
+      max_actor_loss:                           0.00096648
+      max_advantage:                            0.03995645
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          6.331356e-05
+      max_entropy:                              0.10830453
+      max_grad_norm_pre_clip:                   1.9372277
+      max_loss:                                 0.00090381055
+    std:
+      std_actor_approxkl1:                      0.0013934908
+      std_actor_approxkl3:                      0.00071124756
+      std_actor_clipfrac:                       0.015994018
+      std_actor_loss:                           0.00059154927
+      std_advantage:                            0.009426523
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.3715653e-06
+      std_entropy:                              0.007455576
+      std_grad_norm_pre_clip:                   0.32411864
+      std_loss:                                 0.0005916914
+  eval-batch-train:
+    avg_avg_episode_length:                     7.492796
+    avg_avg_return:                             0.98156786
+    avg_benchmark_regret:                       0.011231311
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20218658
+    proxy_corner:
+      avg_avg_return:                           0.007831364
+      avg_reward_per_step:                      0.0015869141
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.477129
+    avg_avg_return:                             0.9751301
+    avg_benchmark_regret:                       0.015109767
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14202118
+    proxy_corner:
+      avg_avg_return:                           0.9751301
+      avg_reward_per_step:                      0.14202118
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3637786
+    avg_avg_return:                             0.982031
+    avg_benchmark_regret:                       0.010806486
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19515991
+    proxy_corner:
+      avg_avg_return:                           0.0039176582
+      avg_reward_per_step:                      0.0009841919
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.265305
+    avg_avg_return:                             0.038199455
+    avg_benchmark_regret:                       0.94643575
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.014564514
+    proxy_corner:
+      avg_avg_return:                           0.038199455
+      avg_reward_per_step:                      0.014564514
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        176160768
+    env-step-replay-before:                     89358336
+    env-step-used-before:                       176160768
+    ppo-update-before:                          107520
+    env-step-generate-before:                   86802432
+    env-step-all-after:                         176193536
+    env-step-replay-after:                      89391104
+    env-step-used-after:                        176193536
+    ppo-update-after:                           107540
+    env-step-generate-after:                    86802432
+  train-replay:
+    shift_proportion:                           0.91015625
+    avg_avg_episode_length:                     17.655024
+    avg_avg_return:                             0.85765374
+    avg_reward_per_step:                        0.05307007
+    proxy_corner:
+      avg_avg_return:                           0.08097324
+      avg_reward_per_step:                      0.00491333
+  train-all:
+    shift_proportion:                           0.91015625
+    avg_avg_episode_length:                     17.655024
+    avg_avg_return:                             0.85765374
+    avg_reward_per_step:                        0.05307007
+    proxy_corner:
+      avg_avg_return:                           0.08097324
+      avg_reward_per_step:                      0.00491333
+  perf:
+    env_steps_per_second:                       815896.4388489489
+    ppo_updates_per_second:                     42.08569670245208
+    eval-batch-train-duration:                  0.30158089753240347
+    eval-batch-orig-duration:                   0.21405053324997425
+    eval-batch-shift-duration:                  0.24539714585989714
+    eval-batch-tree-duration:                   0.2236007796600461
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.831787
+      cheese-corne_dist_wavg:                   8.189863
+      levels_cheese_is_in_corner_avg:           0.075683594
+      mouse-cheese_dist_finite_avg:             17.196533
+      mouse-cheese_dist_finite_wavg:            17.01936
+      mouse-cheese_dist_solvable_avg:           17.196533
+      mouse-cheese_dist_solvable_wavg:          17.019361
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12705883
+    visit_patterns:
+      avg_first_visit_time:                     1631.533
+      avg_last_visit_time:                      2718.2263
+      num_replay_batches:                       2728
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0023035237
+    avg_actor_approxkl3:                        0.0017323971
+    avg_actor_clipfrac:                         0.05148926
+    avg_actor_loss:                             0.0011357128
+    avg_advantage:                              -0.0041365186
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            5.0680053e-05
+    avg_entropy:                                0.11334288
+    avg_grad_norm_pre_clip:                     2.172346
+    avg_loss:                                   0.0010477098
+    max:
+      max_actor_approxkl1:                      0.0041690622
+      max_actor_approxkl3:                      0.0030729468
+      max_actor_clipfrac:                       0.07739258
+      max_actor_loss:                           0.0025545359
+      max_advantage:                            0.044031978
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          5.9684502e-05
+      max_entropy:                              0.12769708
+      max_grad_norm_pre_clip:                   3.9710636
+      max_loss:                                 0.0024789788
+    std:
+      std_actor_approxkl1:                      0.0012826571
+      std_actor_approxkl3:                      0.0008794724
+      std_actor_clipfrac:                       0.018671716
+      std_actor_loss:                           0.0007385662
+      std_advantage:                            0.009703406
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.8164716e-06
+      std_entropy:                              0.00763325
+      std_grad_norm_pre_clip:                   0.7455885
+      std_loss:                                 0.0007407658
+  eval-batch-train:
+    avg_avg_episode_length:                     7.407333
+    avg_avg_return:                             0.98186314
+    avg_benchmark_regret:                       0.010936031
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.2028656
+    proxy_corner:
+      avg_avg_return:                           0.010282049
+      avg_reward_per_step:                      0.001739502
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.471389
+    avg_avg_return:                             0.9751215
+    avg_benchmark_regret:                       0.015118345
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14202118
+    proxy_corner:
+      avg_avg_return:                           0.9751215
+      avg_reward_per_step:                      0.14202118
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.454098
+    avg_avg_return:                             0.98093414
+    avg_benchmark_regret:                       0.01190334
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19324493
+    proxy_corner:
+      avg_avg_return:                           0.0043857973
+      avg_reward_per_step:                      0.0010910034
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.187515
+    avg_avg_return:                             0.03849963
+    avg_benchmark_regret:                       0.9461355
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.015983582
+    proxy_corner:
+      avg_avg_return:                           0.03849963
+      avg_reward_per_step:                      0.015983582
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        177209344
+    env-step-replay-before:                     89784320
+    env-step-used-before:                       177209344
+    ppo-update-before:                          108160
+    env-step-generate-before:                   87425024
+    env-step-all-after:                         177242112
+    env-step-replay-after:                      89784320
+    env-step-used-after:                        177242112
+    ppo-update-after:                           108180
+    env-step-generate-after:                    87457792
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.4073315
+    avg_avg_return:                             0.9491774
+    avg_reward_per_step:                        0.19659424
+    proxy_corner:
+      avg_avg_return:                           0.010626903
+      avg_reward_per_step:                      0.004333496
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     7.4073315
+    avg_avg_return:                             0.9491774
+    avg_reward_per_step:                        0.19659424
+    proxy_corner:
+      avg_avg_return:                           0.010626903
+      avg_reward_per_step:                      0.004333496
+  perf:
+    env_steps_per_second:                       311456.6620997226
+    ppo_updates_per_second:                     41.659840957714565
+    eval-batch-train-duration:                  0.5219542048871517
+    eval-batch-orig-duration:                   0.4823674652725458
+    eval-batch-shift-duration:                  0.48496949300169945
+    eval-batch-tree-duration:                   0.4844749327749014
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.83374
+      cheese-corne_dist_wavg:                   16.267727
+      levels_cheese_is_in_corner_avg:           0.07543945
+      mouse-cheese_dist_finite_avg:             17.20874
+      mouse-cheese_dist_finite_wavg:            17.02126
+      mouse-cheese_dist_solvable_avg:           17.20874
+      mouse-cheese_dist_solvable_wavg:          17.02126
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12749004
+    visit_patterns:
+      avg_first_visit_time:                     1638.9146
+      avg_last_visit_time:                      2730.2395
+      num_replay_batches:                       2740
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0028544196
+    avg_actor_approxkl3:                        0.0019564996
+    avg_actor_clipfrac:                         0.043542482
+    avg_actor_loss:                             0.0007844243
+    avg_advantage:                              -0.00059729966
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            8.201142e-05
+    avg_entropy:                                0.09950096
+    avg_grad_norm_pre_clip:                     1.9199219
+    avg_loss:                                   0.00072592904
+    max:
+      max_actor_approxkl1:                      0.0065291487
+      max_actor_approxkl3:                      0.004080518
+      max_actor_clipfrac:                       0.080566406
+      max_actor_loss:                           0.0041704793
+      max_advantage:                            0.047430575
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          9.549453e-05
+      max_entropy:                              0.11453088
+      max_grad_norm_pre_clip:                   3.7576852
+      max_loss:                                 0.0040993732
+    std:
+      std_actor_approxkl1:                      0.0018241154
+      std_actor_approxkl3:                      0.001301425
+      std_actor_clipfrac:                       0.020431269
+      std_actor_loss:                           0.0012149807
+      std_advantage:                            0.012652178
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          6.65989e-06
+      std_entropy:                              0.00866421
+      std_grad_norm_pre_clip:                   0.6678711
+      std_loss:                                 0.001213386
+  eval-batch-train:
+    avg_avg_episode_length:                     7.39612
+    avg_avg_return:                             0.98178613
+    avg_benchmark_regret:                       0.011012981
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20331573
+    proxy_corner:
+      avg_avg_return:                           0.008097387
+      avg_reward_per_step:                      0.0016021729
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.464394
+    avg_avg_return:                             0.97527945
+    avg_benchmark_regret:                       0.014960421
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14211273
+    proxy_corner:
+      avg_avg_return:                           0.97527945
+      avg_reward_per_step:                      0.14211273
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.2871933
+    avg_avg_return:                             0.98242354
+    avg_benchmark_regret:                       0.010413839
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19543457
+    proxy_corner:
+      avg_avg_return:                           0.0039346213
+      avg_reward_per_step:                      0.0009918213
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.36612
+    avg_avg_return:                             0.03742905
+    avg_benchmark_regret:                       0.94720614
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.013710022
+    proxy_corner:
+      avg_avg_return:                           0.03742905
+      avg_reward_per_step:                      0.013710022
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        178257920
+    env-step-replay-before:                     90439680
+    env-step-used-before:                       178257920
+    ppo-update-before:                          108800
+    env-step-generate-before:                   87818240
+    env-step-all-after:                         178290688
+    env-step-replay-after:                      90472448
+    env-step-used-after:                        178290688
+    ppo-update-after:                           108820
+    env-step-generate-after:                    87818240
+  train-replay:
+    shift_proportion:                           0.953125
+    avg_avg_episode_length:                     17.651726
+    avg_avg_return:                             0.85932153
+    avg_reward_per_step:                        0.053009033
+    proxy_corner:
+      avg_avg_return:                           0.040870752
+      avg_reward_per_step:                      0.0024414062
+  train-all:
+    shift_proportion:                           0.953125
+    avg_avg_episode_length:                     17.651726
+    avg_avg_return:                             0.85932153
+    avg_reward_per_step:                        0.053009033
+    proxy_corner:
+      avg_avg_return:                           0.040870752
+      avg_reward_per_step:                      0.0024414062
+  perf:
+    env_steps_per_second:                       749636.5330092896
+    ppo_updates_per_second:                     41.83660636601092
+    eval-batch-train-duration:                  0.3132707020267844
+    eval-batch-orig-duration:                   0.2593652633950114
+    eval-batch-shift-duration:                  0.22330968361347914
+    eval-batch-tree-duration:                   0.2541426606476307
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.823242
+      cheese-corne_dist_wavg:                   9.980788
+      levels_cheese_is_in_corner_avg:           0.075683594
+      mouse-cheese_dist_finite_avg:             17.216309
+      mouse-cheese_dist_finite_wavg:            17.019936
+      mouse-cheese_dist_solvable_avg:           17.216309
+      mouse-cheese_dist_solvable_wavg:          17.019938
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12599581
+    visit_patterns:
+      avg_first_visit_time:                     1641.3186
+      avg_last_visit_time:                      2751.3186
+      num_replay_batches:                       2761
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014867189
+    avg_actor_approxkl3:                        0.0012495812
+    avg_actor_clipfrac:                         0.0423584
+    avg_actor_loss:                             0.0006634152
+    avg_advantage:                              -0.0061047636
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.9493516e-05
+    avg_entropy:                                0.10959073
+    avg_grad_norm_pre_clip:                     1.9051214
+    avg_loss:                                   0.00057857117
+    max:
+      max_actor_approxkl1:                      0.0034660217
+      max_actor_approxkl3:                      0.002192163
+      max_actor_clipfrac:                       0.068359375
+      max_actor_loss:                           0.0020488799
+      max_advantage:                            0.036403418
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          6.6884924e-05
+      max_entropy:                              0.11722359
+      max_grad_norm_pre_clip:                   2.6517968
+      max_loss:                                 0.0019611681
+    std:
+      std_actor_approxkl1:                      0.0009188248
+      std_actor_approxkl3:                      0.0005926857
+      std_actor_clipfrac:                       0.015215123
+      std_actor_loss:                           0.0006238348
+      std_advantage:                            0.0094772
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.77223e-06
+      std_entropy:                              0.004056031
+      std_grad_norm_pre_clip:                   0.44980237
+      std_loss:                                 0.00062404945
+  eval-batch-train:
+    avg_avg_episode_length:                     7.382519
+    avg_avg_return:                             0.9819228
+    avg_benchmark_regret:                       0.010876356
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20347595
+    proxy_corner:
+      avg_avg_return:                           0.009119949
+      avg_reward_per_step:                      0.0016708374
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.469027
+    avg_avg_return:                             0.9748885
+    avg_benchmark_regret:                       0.015351351
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14202881
+    proxy_corner:
+      avg_avg_return:                           0.9748885
+      avg_reward_per_step:                      0.14202881
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3066998
+    avg_avg_return:                             0.9815236
+    avg_benchmark_regret:                       0.011313895
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19480896
+    proxy_corner:
+      avg_avg_return:                           0.004160836
+      avg_reward_per_step:                      0.001045227
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.4449
+    avg_avg_return:                             0.037066747
+    avg_benchmark_regret:                       0.9475684
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.013320923
+    proxy_corner:
+      avg_avg_return:                           0.037066747
+      avg_reward_per_step:                      0.013320923
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        179306496
+    env-step-replay-before:                     90931200
+    env-step-used-before:                       179306496
+    ppo-update-before:                          109440
+    env-step-generate-before:                   88375296
+    env-step-all-after:                         179339264
+    env-step-replay-after:                      90931200
+    env-step-used-after:                        179339264
+    ppo-update-after:                           109460
+    env-step-generate-after:                    88408064
+  train-generate:
+    shift_proportion:                           1.0
+    avg_avg_episode_length:                     7.5916176
+    avg_avg_return:                             0.9504369
+    avg_reward_per_step:                        0.18832397
+    proxy_corner:
+      avg_avg_return:                           0.0
+      avg_reward_per_step:                      0.0
+  train-all:
+    shift_proportion:                           1.0
+    avg_avg_episode_length:                     7.5916176
+    avg_avg_return:                             0.9504369
+    avg_reward_per_step:                        0.18832397
+    proxy_corner:
+      avg_avg_return:                           0.0
+      avg_reward_per_step:                      0.0
+  perf:
+    env_steps_per_second:                       329443.2285622311
+    ppo_updates_per_second:                     42.15634004753591
+    eval-batch-train-duration:                  0.5126048093661666
+    eval-batch-orig-duration:                   0.4860963234677911
+    eval-batch-shift-duration:                  0.4928285265341401
+    eval-batch-tree-duration:                   0.48969076573848724
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.833496
+      cheese-corne_dist_wavg:                   4.5763993
+      levels_cheese_is_in_corner_avg:           0.07495117
+      mouse-cheese_dist_finite_avg:             17.22705
+      mouse-cheese_dist_finite_wavg:            15.220414
+      mouse-cheese_dist_solvable_avg:           17.22705
+      mouse-cheese_dist_solvable_wavg:          15.220416
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12641843
+    visit_patterns:
+      avg_first_visit_time:                     1648.6123
+      avg_last_visit_time:                      2765.3252
+      num_replay_batches:                       2775
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0027131583
+    avg_actor_approxkl3:                        0.0020985135
+    avg_actor_clipfrac:                         0.046038818
+    avg_actor_loss:                             0.0018281917
+    avg_advantage:                              -0.0012550519
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            8.24811e-05
+    avg_entropy:                                0.09023982
+    avg_grad_norm_pre_clip:                     2.4916234
+    avg_loss:                                   0.0017791924
+    max:
+      max_actor_approxkl1:                      0.006428494
+      max_actor_approxkl3:                      0.005449713
+      max_actor_clipfrac:                       0.079833984
+      max_actor_loss:                           0.0048758974
+      max_advantage:                            0.05258739
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          9.423517e-05
+      max_entropy:                              0.09859249
+      max_grad_norm_pre_clip:                   4.3126674
+      max_loss:                                 0.004832457
+    std:
+      std_actor_approxkl1:                      0.0017516273
+      std_actor_approxkl3:                      0.001378626
+      std_actor_clipfrac:                       0.020218886
+      std_actor_loss:                           0.0013085525
+      std_advantage:                            0.012823546
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.87392e-06
+      std_entropy:                              0.0056028515
+      std_grad_norm_pre_clip:                   0.9120093
+      std_loss:                                 0.0013083391
+  eval-batch-train:
+    avg_avg_episode_length:                     7.391781
+    avg_avg_return:                             0.9818855
+    avg_benchmark_regret:                       0.010913665
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20348358
+    proxy_corner:
+      avg_avg_return:                           0.0076272064
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.480701
+    avg_avg_return:                             0.97537464
+    avg_benchmark_regret:                       0.014865222
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14198303
+    proxy_corner:
+      avg_avg_return:                           0.97537464
+      avg_reward_per_step:                      0.14198303
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.2795234
+    avg_avg_return:                             0.98195827
+    avg_benchmark_regret:                       0.010879165
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19584656
+    proxy_corner:
+      avg_avg_return:                           0.004093593
+      avg_reward_per_step:                      0.0010223389
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.72108
+    avg_avg_return:                             0.034825645
+    avg_benchmark_regret:                       0.94980955
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.012428284
+    proxy_corner:
+      avg_avg_return:                           0.034825645
+      avg_reward_per_step:                      0.012428284
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        180355072
+    env-step-replay-before:                     91291648
+    env-step-used-before:                       180355072
+    ppo-update-before:                          110080
+    env-step-generate-before:                   89063424
+    env-step-all-after:                         180387840
+    env-step-replay-after:                      91324416
+    env-step-used-after:                        180387840
+    ppo-update-after:                           110100
+    env-step-generate-after:                    89063424
+  train-replay:
+    shift_proportion:                           0.953125
+    avg_avg_episode_length:                     17.528658
+    avg_avg_return:                             0.8594816
+    avg_reward_per_step:                        0.053466797
+    proxy_corner:
+      avg_avg_return:                           0.042321973
+      avg_reward_per_step:                      0.002532959
+  train-all:
+    shift_proportion:                           0.953125
+    avg_avg_episode_length:                     17.528658
+    avg_avg_return:                             0.8594816
+    avg_reward_per_step:                        0.053466797
+    proxy_corner:
+      avg_avg_return:                           0.042321973
+      avg_reward_per_step:                      0.002532959
+  perf:
+    env_steps_per_second:                       298890.40733710426
+    ppo_updates_per_second:                     41.70211774345779
+    eval-batch-train-duration:                  0.5262683872133493
+    eval-batch-orig-duration:                   0.47873177472501993
+    eval-batch-shift-duration:                  0.4777676612138748
+    eval-batch-tree-duration:                   0.44032012298703194
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.817139
+      cheese-corne_dist_wavg:                   9.979897
+      levels_cheese_is_in_corner_avg:           0.07470703
+      mouse-cheese_dist_finite_avg:             17.22876
+      mouse-cheese_dist_finite_wavg:            17.023457
+      mouse-cheese_dist_solvable_avg:           17.22876
+      mouse-cheese_dist_solvable_wavg:          17.023455
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12663415
+    visit_patterns:
+      avg_first_visit_time:                     1652.9116
+      avg_last_visit_time:                      2777.2412
+      num_replay_batches:                       2787
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0020679918
+    avg_actor_approxkl3:                        0.0015827339
+    avg_actor_clipfrac:                         0.05076294
+    avg_actor_loss:                             0.0011757491
+    avg_advantage:                              0.0011092544
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.8038975e-05
+    avg_entropy:                                0.11769462
+    avg_grad_norm_pre_clip:                     2.6550055
+    avg_loss:                                   0.0010820741
+    max:
+      max_actor_approxkl1:                      0.004967405
+      max_actor_approxkl3:                      0.003530771
+      max_actor_clipfrac:                       0.08215332
+      max_actor_loss:                           0.002789082
+      max_advantage:                            0.050462306
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          5.3644246e-05
+      max_entropy:                              0.13453548
+      max_grad_norm_pre_clip:                   4.1523514
+      max_loss:                                 0.002698729
+    std:
+      std_actor_approxkl1:                      0.001163328
+      std_actor_approxkl3:                      0.0008106847
+      std_actor_clipfrac:                       0.019741274
+      std_actor_loss:                           0.0009467739
+      std_advantage:                            0.009835063
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.667394e-06
+      std_entropy:                              0.0074954666
+      std_grad_norm_pre_clip:                   0.92837125
+      std_loss:                                 0.00094424904
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3242207
+    avg_avg_return:                             0.98196316
+    avg_benchmark_regret:                       0.010835984
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20419312
+    proxy_corner:
+      avg_avg_return:                           0.0077824215
+      avg_reward_per_step:                      0.0015869141
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.489727
+    avg_avg_return:                             0.97489834
+    avg_benchmark_regret:                       0.01534149
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14186859
+    proxy_corner:
+      avg_avg_return:                           0.97489834
+      avg_reward_per_step:                      0.14186859
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.2539396
+    avg_avg_return:                             0.98244
+    avg_benchmark_regret:                       0.010397487
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19652557
+    proxy_corner:
+      avg_avg_return:                           0.0041695144
+      avg_reward_per_step:                      0.0010070801
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.33121
+    avg_avg_return:                             0.037300806
+    avg_benchmark_regret:                       0.9473344
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.014472961
+    proxy_corner:
+      avg_avg_return:                           0.037300806
+      avg_reward_per_step:                      0.014472961
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        181403648
+    env-step-replay-before:                     91947008
+    env-step-used-before:                       181403648
+    ppo-update-before:                          110720
+    env-step-generate-before:                   89456640
+    env-step-all-after:                         181436416
+    env-step-replay-after:                      91979776
+    env-step-used-after:                        181436416
+    ppo-update-after:                           110740
+    env-step-generate-after:                    89456640
+  train-replay:
+    shift_proportion:                           0.9296875
+    avg_avg_episode_length:                     17.685127
+    avg_avg_return:                             0.858803
+    avg_reward_per_step:                        0.05291748
+    proxy_corner:
+      avg_avg_return:                           0.06098335
+      avg_reward_per_step:                      0.003692627
+  train-all:
+    shift_proportion:                           0.9296875
+    avg_avg_episode_length:                     17.685127
+    avg_avg_return:                             0.858803
+    avg_reward_per_step:                        0.05291748
+    proxy_corner:
+      avg_avg_return:                           0.06098335
+      avg_reward_per_step:                      0.003692627
+  perf:
+    env_steps_per_second:                       244640.3262995742
+    ppo_updates_per_second:                     42.24581850409386
+    eval-batch-train-duration:                  0.6775416387245059
+    eval-batch-orig-duration:                   0.6084385830909014
+    eval-batch-shift-duration:                  0.6164264706894755
+    eval-batch-tree-duration:                   0.6137876594439149
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.824219
+      cheese-corne_dist_wavg:                   9.087664
+      levels_cheese_is_in_corner_avg:           0.07446289
+      mouse-cheese_dist_finite_avg:             17.247803
+      mouse-cheese_dist_finite_wavg:            11.628297
+      mouse-cheese_dist_solvable_avg:           17.247803
+      mouse-cheese_dist_solvable_wavg:          11.6283
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.126336
+    visit_patterns:
+      avg_first_visit_time:                     1657.6843
+      avg_last_visit_time:                      2797.353
+      num_replay_batches:                       2807
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002573474
+    avg_actor_approxkl3:                        0.0015322702
+    avg_actor_clipfrac:                         0.049298096
+    avg_actor_loss:                             -1.4510495e-05
+    avg_advantage:                              -0.0036603243
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.787969e-05
+    avg_entropy:                                0.115943186
+    avg_grad_norm_pre_clip:                     1.6654478
+    avg_loss:                                   -0.00010651385
+    max:
+      max_actor_approxkl1:                      0.004812749
+      max_actor_approxkl3:                      0.003404773
+      max_actor_clipfrac:                       0.08703613
+      max_actor_loss:                           0.0015470278
+      max_advantage:                            0.054172873
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          5.720256e-05
+      max_entropy:                              0.13819368
+      max_grad_norm_pre_clip:                   2.7039175
+      max_loss:                                 0.0014489111
+    std:
+      std_actor_approxkl1:                      0.001121456
+      std_actor_approxkl3:                      0.0007378318
+      std_actor_clipfrac:                       0.017930059
+      std_actor_loss:                           0.00083670614
+      std_advantage:                            0.009503416
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.799701e-06
+      std_entropy:                              0.007765716
+      std_grad_norm_pre_clip:                   0.5391139
+      std_loss:                                 0.00083717395
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4800835
+    avg_avg_return:                             0.9812846
+    avg_benchmark_regret:                       0.011514546
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20314789
+    proxy_corner:
+      avg_avg_return:                           0.0076272064
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.4667845
+    avg_avg_return:                             0.9753251
+    avg_benchmark_regret:                       0.014914745
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14207458
+    proxy_corner:
+      avg_avg_return:                           0.9753251
+      avg_reward_per_step:                      0.14207458
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.316265
+    avg_avg_return:                             0.98179233
+    avg_benchmark_regret:                       0.011045061
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.195755
+    proxy_corner:
+      avg_avg_return:                           0.0040093027
+      avg_reward_per_step:                      0.0009994507
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.126625
+    avg_avg_return:                             0.04063744
+    avg_benchmark_regret:                       0.94399774
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.016441345
+    proxy_corner:
+      avg_avg_return:                           0.04063744
+      avg_reward_per_step:                      0.016441345
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        182452224
+    env-step-replay-before:                     92471296
+    env-step-used-before:                       182452224
+    ppo-update-before:                          111360
+    env-step-generate-before:                   89980928
+    env-step-all-after:                         182484992
+    env-step-replay-after:                      92504064
+    env-step-used-after:                        182484992
+    ppo-update-after:                           111380
+    env-step-generate-after:                    89980928
+  train-replay:
+    shift_proportion:                           0.9140625
+    avg_avg_episode_length:                     17.906586
+    avg_avg_return:                             0.85739607
+    avg_reward_per_step:                        0.0524292
+    proxy_corner:
+      avg_avg_return:                           0.073307574
+      avg_reward_per_step:                      0.004425049
+  train-all:
+    shift_proportion:                           0.9140625
+    avg_avg_episode_length:                     17.906586
+    avg_avg_return:                             0.85739607
+    avg_reward_per_step:                        0.0524292
+    proxy_corner:
+      avg_avg_return:                           0.073307574
+      avg_reward_per_step:                      0.004425049
+  perf:
+    env_steps_per_second:                       355815.29563668714
+    ppo_updates_per_second:                     42.70580801999625
+    eval-batch-train-duration:                  0.5456542782485485
+    eval-batch-orig-duration:                   0.48372773453593254
+    eval-batch-shift-duration:                  0.45463569927960634
+    eval-batch-tree-duration:                   0.47803089395165443
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.804932
+      cheese-corne_dist_wavg:                   18.977106
+      levels_cheese_is_in_corner_avg:           0.07519531
+      mouse-cheese_dist_finite_avg:             17.25
+      mouse-cheese_dist_finite_wavg:            18.822187
+      mouse-cheese_dist_solvable_avg:           17.25
+      mouse-cheese_dist_solvable_wavg:          18.822191
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1264244
+    visit_patterns:
+      avg_first_visit_time:                     1661.8938
+      avg_last_visit_time:                      2813.2104
+      num_replay_batches:                       2823
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0020056493
+    avg_actor_approxkl3:                        0.0014137792
+    avg_actor_clipfrac:                         0.052142333
+    avg_actor_loss:                             0.0006870398
+    avg_advantage:                              -0.00525503
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.8394977e-05
+    avg_entropy:                                0.12639105
+    avg_grad_norm_pre_clip:                     2.132886
+    avg_loss:                                   0.00058484636
+    max:
+      max_actor_approxkl1:                      0.004932048
+      max_actor_approxkl3:                      0.0025233875
+      max_actor_clipfrac:                       0.08105469
+      max_actor_loss:                           0.0024897605
+      max_advantage:                            0.049393654
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          6.9042086e-05
+      max_entropy:                              0.14403632
+      max_grad_norm_pre_clip:                   3.3726556
+      max_loss:                                 0.002385283
+    std:
+      std_actor_approxkl1:                      0.0010793497
+      std_actor_approxkl3:                      0.0006694637
+      std_actor_clipfrac:                       0.017762732
+      std_actor_loss:                           0.0010937682
+      std_advantage:                            0.00945228
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.270522e-06
+      std_entropy:                              0.010920006
+      std_grad_norm_pre_clip:                   0.71568453
+      std_loss:                                 0.0010958918
+  eval-batch-train:
+    avg_avg_episode_length:                     7.445659
+    avg_avg_return:                             0.9811525
+    avg_benchmark_regret:                       0.011646705
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20362854
+    proxy_corner:
+      avg_avg_return:                           0.0076272064
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.46433
+    avg_avg_return:                             0.9751549
+    avg_benchmark_regret:                       0.015084991
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14209747
+    proxy_corner:
+      avg_avg_return:                           0.9751549
+      avg_reward_per_step:                      0.14209747
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3300323
+    avg_avg_return:                             0.9821075
+    avg_benchmark_regret:                       0.010729901
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19537354
+    proxy_corner:
+      avg_avg_return:                           0.003941047
+      avg_reward_per_step:                      0.0009918213
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.26048
+    avg_avg_return:                             0.038610935
+    avg_benchmark_regret:                       0.94602424
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.014762878
+    proxy_corner:
+      avg_avg_return:                           0.038610935
+      avg_reward_per_step:                      0.014762878
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        183500800
+    env-step-replay-before:                     93028352
+    env-step-used-before:                       183500800
+    ppo-update-before:                          112000
+    env-step-generate-before:                   90472448
+    env-step-all-after:                         183533568
+    env-step-replay-after:                      93061120
+    env-step-used-after:                        183533568
+    ppo-update-after:                           112020
+    env-step-generate-after:                    90472448
+  train-replay:
+    shift_proportion:                           0.95703125
+    avg_avg_episode_length:                     17.710354
+    avg_avg_return:                             0.8581898
+    avg_reward_per_step:                        0.052856445
+    proxy_corner:
+      avg_avg_return:                           0.04398435
+      avg_reward_per_step:                      0.0027160645
+  train-all:
+    shift_proportion:                           0.95703125
+    avg_avg_episode_length:                     17.710354
+    avg_avg_return:                             0.8581898
+    avg_reward_per_step:                        0.052856445
+    proxy_corner:
+      avg_avg_return:                           0.04398435
+      avg_reward_per_step:                      0.0027160645
+  perf:
+    env_steps_per_second:                       325129.90355498216
+    ppo_updates_per_second:                     44.96411711550951
+    eval-batch-train-duration:                  0.5106569621711969
+    eval-batch-orig-duration:                   0.4816548116505146
+    eval-batch-shift-duration:                  0.4753106376156211
+    eval-batch-tree-duration:                   0.45802867971360683
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.807861
+      cheese-corne_dist_wavg:                   18.98759
+      levels_cheese_is_in_corner_avg:           0.07470703
+      mouse-cheese_dist_finite_avg:             17.240234
+      mouse-cheese_dist_finite_wavg:            16.1244
+      mouse-cheese_dist_solvable_avg:           17.240234
+      mouse-cheese_dist_solvable_wavg:          16.1244
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1265792
+    visit_patterns:
+      avg_first_visit_time:                     1669.3176
+      avg_last_visit_time:                      2830.3513
+      num_replay_batches:                       2840
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002404502
+    avg_actor_approxkl3:                        0.0017324373
+    avg_actor_clipfrac:                         0.0543396
+    avg_actor_loss:                             0.00092782546
+    avg_advantage:                              -0.0018554492
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.8877155e-05
+    avg_entropy:                                0.12896563
+    avg_grad_norm_pre_clip:                     2.0802312
+    avg_loss:                                   0.00082329847
+    max:
+      max_actor_approxkl1:                      0.0051059257
+      max_actor_approxkl3:                      0.003059519
+      max_actor_clipfrac:                       0.091918945
+      max_actor_loss:                           0.0039631333
+      max_advantage:                            0.047417045
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          5.4116375e-05
+      max_entropy:                              0.13534951
+      max_grad_norm_pre_clip:                   4.0575643
+      max_loss:                                 0.0038537774
+    std:
+      std_actor_approxkl1:                      0.0011915418
+      std_actor_approxkl3:                      0.0007109744
+      std_actor_clipfrac:                       0.019450871
+      std_actor_loss:                           0.0011233962
+      std_advantage:                            0.00980427
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.081698e-06
+      std_entropy:                              0.004806042
+      std_grad_norm_pre_clip:                   0.8138114
+      std_loss:                                 0.0011222338
+  eval-batch-train:
+    avg_avg_episode_length:                     7.36432
+    avg_avg_return:                             0.98183435
+    avg_benchmark_regret:                       0.010964824
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.2047348
+    proxy_corner:
+      avg_avg_return:                           0.0076272064
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.462347
+    avg_avg_return:                             0.9752541
+    avg_benchmark_regret:                       0.014985744
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14209747
+    proxy_corner:
+      avg_avg_return:                           0.9752541
+      avg_reward_per_step:                      0.14209747
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3006697
+    avg_avg_return:                             0.9815667
+    avg_benchmark_regret:                       0.011270701
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19570923
+    proxy_corner:
+      avg_avg_return:                           0.003940231
+      avg_reward_per_step:                      0.0009918213
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.26323
+    avg_avg_return:                             0.038328297
+    avg_benchmark_regret:                       0.94630694
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.014060974
+    proxy_corner:
+      avg_avg_return:                           0.038328297
+      avg_reward_per_step:                      0.014060974
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        184549376
+    env-step-replay-before:                     93618176
+    env-step-used-before:                       184549376
+    ppo-update-before:                          112640
+    env-step-generate-before:                   90931200
+    env-step-all-after:                         184582144
+    env-step-replay-after:                      93650944
+    env-step-used-after:                        184582144
+    ppo-update-after:                           112660
+    env-step-generate-after:                    90931200
+  train-replay:
+    shift_proportion:                           0.91796875
+    avg_avg_episode_length:                     17.90403
+    avg_avg_return:                             0.85673404
+    avg_reward_per_step:                        0.052337646
+    proxy_corner:
+      avg_avg_return:                           0.07310407
+      avg_reward_per_step:                      0.004333496
+  train-all:
+    shift_proportion:                           0.91796875
+    avg_avg_episode_length:                     17.90403
+    avg_avg_return:                             0.85673404
+    avg_reward_per_step:                        0.052337646
+    proxy_corner:
+      avg_avg_return:                           0.07310407
+      avg_reward_per_step:                      0.004333496
+  perf:
+    env_steps_per_second:                       299132.94777381344
+    ppo_updates_per_second:                     42.48288050026259
+    eval-batch-train-duration:                  0.5354519812390208
+    eval-batch-orig-duration:                   0.48444083239883184
+    eval-batch-shift-duration:                  0.48082767333835363
+    eval-batch-tree-duration:                   0.4533717203885317
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.792236
+      cheese-corne_dist_wavg:                   18.97158
+      levels_cheese_is_in_corner_avg:           0.07446289
+      mouse-cheese_dist_finite_avg:             17.246582
+      mouse-cheese_dist_finite_wavg:            18.825373
+      mouse-cheese_dist_solvable_avg:           17.246582
+      mouse-cheese_dist_solvable_wavg:          18.825373
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12679318
+    visit_patterns:
+      avg_first_visit_time:                     1674.4768
+      avg_last_visit_time:                      2848.317
+      num_replay_batches:                       2858
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0024035021
+    avg_actor_approxkl3:                        0.0016833221
+    avg_actor_clipfrac:                         0.056573488
+    avg_actor_loss:                             0.0007160717
+    avg_advantage:                              -0.0016389302
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.984802e-05
+    avg_entropy:                                0.1230753
+    avg_grad_norm_pre_clip:                     2.304954
+    avg_loss:                                   0.0006179204
+    max:
+      max_actor_approxkl1:                      0.0048574833
+      max_actor_approxkl3:                      0.0034439515
+      max_actor_clipfrac:                       0.107666016
+      max_actor_loss:                           0.001970465
+      max_advantage:                            0.043640137
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          6.2462306e-05
+      max_entropy:                              0.13518573
+      max_grad_norm_pre_clip:                   3.3302286
+      max_loss:                                 0.0018752558
+    std:
+      std_actor_approxkl1:                      0.0013479246
+      std_actor_approxkl3:                      0.0009050168
+      std_actor_clipfrac:                       0.025861727
+      std_actor_loss:                           0.00055066426
+      std_advantage:                            0.009625466
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.6243766e-06
+      std_entropy:                              0.006355782
+      std_grad_norm_pre_clip:                   0.7322677
+      std_loss:                                 0.00055099337
+  eval-batch-train:
+    avg_avg_episode_length:                     7.38757
+    avg_avg_return:                             0.98202777
+    avg_benchmark_regret:                       0.010771412
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20453644
+    proxy_corner:
+      avg_avg_return:                           0.0076272064
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.465357
+    avg_avg_return:                             0.97494066
+    avg_benchmark_regret:                       0.015299141
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14201355
+    proxy_corner:
+      avg_avg_return:                           0.97494066
+      avg_reward_per_step:                      0.14201355
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.2960863
+    avg_avg_return:                             0.9817389
+    avg_benchmark_regret:                       0.011098473
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19579315
+    proxy_corner:
+      avg_avg_return:                           0.003986592
+      avg_reward_per_step:                      0.0010070801
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.297966
+    avg_avg_return:                             0.03801565
+    avg_benchmark_regret:                       0.9466195
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.015129089
+    proxy_corner:
+      avg_avg_return:                           0.03801565
+      avg_reward_per_step:                      0.015129089
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        185597952
+    env-step-replay-before:                     94142464
+    env-step-used-before:                       185597952
+    ppo-update-before:                          113280
+    env-step-generate-before:                   91455488
+    env-step-all-after:                         185630720
+    env-step-replay-after:                      94175232
+    env-step-used-after:                        185630720
+    ppo-update-after:                           113300
+    env-step-generate-after:                    91455488
+  train-replay:
+    shift_proportion:                           0.921875
+    avg_avg_episode_length:                     18.046085
+    avg_avg_return:                             0.85472924
+    avg_reward_per_step:                        0.05218506
+    proxy_corner:
+      avg_avg_return:                           0.06829347
+      avg_reward_per_step:                      0.0040283203
+  train-all:
+    shift_proportion:                           0.921875
+    avg_avg_episode_length:                     18.046085
+    avg_avg_return:                             0.85472924
+    avg_reward_per_step:                        0.05218506
+    proxy_corner:
+      avg_avg_return:                           0.06829347
+      avg_reward_per_step:                      0.0040283203
+  perf:
+    env_steps_per_second:                       265086.0347339197
+    ppo_updates_per_second:                     41.495959533257235
+    eval-batch-train-duration:                  0.5338886054232717
+    eval-batch-orig-duration:                   0.46100854966789484
+    eval-batch-shift-duration:                  0.4498855946585536
+    eval-batch-tree-duration:                   0.488003077916801
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.793457
+      cheese-corne_dist_wavg:                   8.175063
+      levels_cheese_is_in_corner_avg:           0.07519531
+      mouse-cheese_dist_finite_avg:             17.247803
+      mouse-cheese_dist_finite_wavg:            17.024904
+      mouse-cheese_dist_solvable_avg:           17.247803
+      mouse-cheese_dist_solvable_wavg:          17.024904
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12756068
+    visit_patterns:
+      avg_first_visit_time:                     1678.8499
+      avg_last_visit_time:                      2864.3013
+      num_replay_batches:                       2874
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001947121
+    avg_actor_approxkl3:                        0.0011731455
+    avg_actor_clipfrac:                         0.040667724
+    avg_actor_loss:                             -5.601491e-05
+    avg_advantage:                              -0.00022611106
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.6733967e-05
+    avg_entropy:                                0.12639014
+    avg_grad_norm_pre_clip:                     1.688845
+    avg_loss:                                   -0.0001590381
+    max:
+      max_actor_approxkl1:                      0.0040793773
+      max_actor_approxkl3:                      0.003069153
+      max_actor_clipfrac:                       0.075683594
+      max_actor_loss:                           0.0015562521
+      max_advantage:                            0.049767494
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          5.179484e-05
+      max_entropy:                              0.13893335
+      max_grad_norm_pre_clip:                   3.301922
+      max_loss:                                 0.0014416858
+    std:
+      std_actor_approxkl1:                      0.00105205
+      std_actor_approxkl3:                      0.00069582387
+      std_actor_clipfrac:                       0.016998934
+      std_actor_loss:                           0.00080467074
+      std_advantage:                            0.0097319195
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.7069561e-06
+      std_entropy:                              0.007899083
+      std_grad_norm_pre_clip:                   0.67972094
+      std_loss:                                 0.0008018589
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3810296
+    avg_avg_return:                             0.9814856
+    avg_benchmark_regret:                       0.011313597
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20373535
+    proxy_corner:
+      avg_avg_return:                           0.008083535
+      avg_reward_per_step:                      0.0016021729
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.462555
+    avg_avg_return:                             0.97499144
+    avg_benchmark_regret:                       0.015248418
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14211273
+    proxy_corner:
+      avg_avg_return:                           0.97499144
+      avg_reward_per_step:                      0.14211273
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3613887
+    avg_avg_return:                             0.9810486
+    avg_benchmark_regret:                       0.011788833
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19443512
+    proxy_corner:
+      avg_avg_return:                           0.004009031
+      avg_reward_per_step:                      0.0009994507
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.26358
+    avg_avg_return:                             0.038342882
+    avg_benchmark_regret:                       0.94629234
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.01436615
+    proxy_corner:
+      avg_avg_return:                           0.038342882
+      avg_reward_per_step:                      0.01436615
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        186646528
+    env-step-replay-before:                     94765056
+    env-step-used-before:                       186646528
+    ppo-update-before:                          113920
+    env-step-generate-before:                   91881472
+    env-step-all-after:                         186679296
+    env-step-replay-after:                      94797824
+    env-step-used-after:                        186679296
+    ppo-update-after:                           113940
+    env-step-generate-after:                    91881472
+  train-replay:
+    shift_proportion:                           0.94140625
+    avg_avg_episode_length:                     17.635925
+    avg_avg_return:                             0.8576728
+    avg_reward_per_step:                        0.05307007
+    proxy_corner:
+      avg_avg_return:                           0.053322155
+      avg_reward_per_step:                      0.0032043457
+  train-all:
+    shift_proportion:                           0.94140625
+    avg_avg_episode_length:                     17.635925
+    avg_avg_return:                             0.8576728
+    avg_reward_per_step:                        0.05307007
+    proxy_corner:
+      avg_avg_return:                           0.053322155
+      avg_reward_per_step:                      0.0032043457
+  perf:
+    env_steps_per_second:                       300510.40439080825
+    ppo_updates_per_second:                     41.80662254104179
+    eval-batch-train-duration:                  0.5272713582962751
+    eval-batch-orig-duration:                   0.44557756651192904
+    eval-batch-shift-duration:                  0.46809044294059277
+    eval-batch-tree-duration:                   0.47647363785654306
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.780273
+      cheese-corne_dist_wavg:                   8.177735
+      levels_cheese_is_in_corner_avg:           0.07519531
+      mouse-cheese_dist_finite_avg:             17.252197
+      mouse-cheese_dist_finite_wavg:            17.023739
+      mouse-cheese_dist_solvable_avg:           17.252197
+      mouse-cheese_dist_solvable_wavg:          17.02374
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1265851
+    visit_patterns:
+      avg_first_visit_time:                     1682.2856
+      avg_last_visit_time:                      2883.2358
+      num_replay_batches:                       2893
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0021335983
+    avg_actor_approxkl3:                        0.0015525748
+    avg_actor_clipfrac:                         0.051794436
+    avg_actor_loss:                             0.0011375912
+    avg_advantage:                              -0.0046257917
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            5.3826807e-05
+    avg_entropy:                                0.11479489
+    avg_grad_norm_pre_clip:                     1.9530407
+    avg_loss:                                   0.0010497099
+    max:
+      max_actor_approxkl1:                      0.0048482665
+      max_actor_approxkl3:                      0.003154559
+      max_actor_clipfrac:                       0.07556152
+      max_actor_loss:                           0.002579589
+      max_advantage:                            0.04383123
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          8.940979e-05
+      max_entropy:                              0.12628964
+      max_grad_norm_pre_clip:                   3.2321982
+      max_loss:                                 0.0024812962
+    std:
+      std_actor_approxkl1:                      0.0012274293
+      std_actor_approxkl3:                      0.0008270532
+      std_actor_clipfrac:                       0.018129192
+      std_actor_loss:                           0.00073859957
+      std_advantage:                            0.008961146
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.42469435e-05
+      std_entropy:                              0.0051872027
+      std_grad_norm_pre_clip:                   0.6220459
+      std_loss:                                 0.00073680066
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3652487
+    avg_avg_return:                             0.98174775
+    avg_benchmark_regret:                       0.01105138
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20503998
+    proxy_corner:
+      avg_avg_return:                           0.0077447286
+      avg_reward_per_step:                      0.0015792847
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.462191
+    avg_avg_return:                             0.97512776
+    avg_benchmark_regret:                       0.015112088
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14214325
+    proxy_corner:
+      avg_avg_return:                           0.97512776
+      avg_reward_per_step:                      0.14214325
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.33109
+    avg_avg_return:                             0.98156595
+    avg_benchmark_regret:                       0.011271458
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19506836
+    proxy_corner:
+      avg_avg_return:                           0.003940231
+      avg_reward_per_step:                      0.0009918213
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.3487
+    avg_avg_return:                             0.037584294
+    avg_benchmark_regret:                       0.9470509
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.014251709
+    proxy_corner:
+      avg_avg_return:                           0.037584294
+      avg_reward_per_step:                      0.014251709
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        187695104
+    env-step-replay-before:                     95420416
+    env-step-used-before:                       187695104
+    ppo-update-before:                          114560
+    env-step-generate-before:                   92274688
+    env-step-all-after:                         187727872
+    env-step-replay-after:                      95420416
+    env-step-used-after:                        187727872
+    ppo-update-after:                           114580
+    env-step-generate-after:                    92307456
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.3622
+    avg_avg_return:                             0.9522534
+    avg_reward_per_step:                        0.2081604
+    proxy_corner:
+      avg_avg_return:                           0.0074397684
+      avg_reward_per_step:                      0.0012207031
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.3622
+    avg_avg_return:                             0.9522534
+    avg_reward_per_step:                        0.2081604
+    proxy_corner:
+      avg_avg_return:                           0.0074397684
+      avg_reward_per_step:                      0.0012207031
+  perf:
+    env_steps_per_second:                       300044.4108517268
+    ppo_updates_per_second:                     42.13061545666242
+    eval-batch-train-duration:                  0.40717495512217283
+    eval-batch-orig-duration:                   0.24702782928943634
+    eval-batch-shift-duration:                  0.31243210658431053
+    eval-batch-tree-duration:                   0.4874813435599208
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.748779
+      cheese-corne_dist_wavg:                   7.278237
+      levels_cheese_is_in_corner_avg:           0.07543945
+      mouse-cheese_dist_finite_avg:             17.23877
+      mouse-cheese_dist_finite_wavg:            17.018858
+      mouse-cheese_dist_solvable_avg:           17.23877
+      mouse-cheese_dist_solvable_wavg:          17.01886
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12697771
+    visit_patterns:
+      avg_first_visit_time:                     1691.0908
+      avg_last_visit_time:                      2902.1365
+      num_replay_batches:                       2912
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002313936
+    avg_actor_approxkl3:                        0.0018132782
+    avg_actor_clipfrac:                         0.041082766
+    avg_actor_loss:                             0.0014566019
+    avg_advantage:                              0.001520209
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            9.470182e-05
+    avg_entropy:                                0.08613338
+    avg_grad_norm_pre_clip:                     1.9147545
+    avg_loss:                                   0.0014178194
+    max:
+      max_actor_approxkl1:                      0.008331361
+      max_actor_approxkl3:                      0.0045833997
+      max_actor_clipfrac:                       0.07336426
+      max_actor_loss:                           0.0040478352
+      max_advantage:                            0.06729996
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00011436147
+      max_entropy:                              0.10063547
+      max_grad_norm_pre_clip:                   3.4647741
+      max_loss:                                 0.0040023937
+    std:
+      std_actor_approxkl1:                      0.001839605
+      std_actor_approxkl3:                      0.0012944305
+      std_actor_clipfrac:                       0.017611172
+      std_actor_loss:                           0.0012024222
+      std_advantage:                            0.01367252
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.8215435e-06
+      std_entropy:                              0.008885493
+      std_grad_norm_pre_clip:                   0.76027167
+      std_loss:                                 0.0011999311
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3460217
+    avg_avg_return:                             0.9815895
+    avg_benchmark_regret:                       0.011209639
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.2050705
+    proxy_corner:
+      avg_avg_return:                           0.007950456
+      avg_reward_per_step:                      0.0015945435
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.465185
+    avg_avg_return:                             0.9753569
+    avg_benchmark_regret:                       0.014882974
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14204407
+    proxy_corner:
+      avg_avg_return:                           0.9753569
+      avg_reward_per_step:                      0.14204407
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.2732773
+    avg_avg_return:                             0.9815719
+    avg_benchmark_regret:                       0.0112655
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19616699
+    proxy_corner:
+      avg_avg_return:                           0.003940231
+      avg_reward_per_step:                      0.0009918213
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.5984
+    avg_avg_return:                             0.03627829
+    avg_benchmark_regret:                       0.94835687
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.013755798
+    proxy_corner:
+      avg_avg_return:                           0.03627829
+      avg_reward_per_step:                      0.013755798
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        188743680
+    env-step-replay-before:                     95911936
+    env-step-used-before:                       188743680
+    ppo-update-before:                          115200
+    env-step-generate-before:                   92831744
+    env-step-all-after:                         188776448
+    env-step-replay-after:                      95911936
+    env-step-used-after:                        188776448
+    ppo-update-after:                           115220
+    env-step-generate-after:                    92864512
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.7708464
+    avg_avg_return:                             0.9450772
+    avg_reward_per_step:                        0.19100952
+    proxy_corner:
+      avg_avg_return:                           0.0037410087
+      avg_reward_per_step:                      0.00076293945
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.7708464
+    avg_avg_return:                             0.9450772
+    avg_reward_per_step:                        0.19100952
+    proxy_corner:
+      avg_avg_return:                           0.0037410087
+      avg_reward_per_step:                      0.00076293945
+  perf:
+    env_steps_per_second:                       322800.98388332396
+    ppo_updates_per_second:                     42.888689703665115
+    eval-batch-train-duration:                  0.5094991745427251
+    eval-batch-orig-duration:                   0.47565383091568947
+    eval-batch-shift-duration:                  0.44472522661089897
+    eval-batch-tree-duration:                   0.4650337677448988
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.758789
+      cheese-corne_dist_wavg:                   17.17467
+      levels_cheese_is_in_corner_avg:           0.07519531
+      mouse-cheese_dist_finite_avg:             17.244873
+      mouse-cheese_dist_finite_wavg:            16.126358
+      mouse-cheese_dist_solvable_avg:           17.244873
+      mouse-cheese_dist_solvable_wavg:          16.126362
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12712353
+    visit_patterns:
+      avg_first_visit_time:                     1705.001
+      avg_last_visit_time:                      2917.3403
+      num_replay_batches:                       2927
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0028049515
+    avg_actor_approxkl3:                        0.002013053
+    avg_actor_clipfrac:                         0.042242434
+    avg_actor_loss:                             0.0011417189
+    avg_advantage:                              0.0034303358
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            8.753827e-05
+    avg_entropy:                                0.07099569
+    avg_grad_norm_pre_clip:                     1.7421006
+    avg_loss:                                   0.0011144923
+    max:
+      max_actor_approxkl1:                      0.007868699
+      max_actor_approxkl3:                      0.0064167576
+      max_actor_clipfrac:                       0.07470703
+      max_actor_loss:                           0.0056261756
+      max_advantage:                            0.056960285
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00010789499
+      max_entropy:                              0.07986798
+      max_grad_norm_pre_clip:                   3.405257
+      max_loss:                                 0.0055875657
+    std:
+      std_actor_approxkl1:                      0.002367895
+      std_actor_approxkl3:                      0.0017253312
+      std_actor_clipfrac:                       0.018993547
+      std_actor_loss:                           0.0013939128
+      std_advantage:                            0.013026364
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          8.98547e-06
+      std_entropy:                              0.007286753
+      std_grad_norm_pre_clip:                   0.6467246
+      std_loss:                                 0.0013930222
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3109856
+    avg_avg_return:                             0.9817177
+    avg_benchmark_regret:                       0.011081461
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20481873
+    proxy_corner:
+      avg_avg_return:                           0.0076272064
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.464724
+    avg_avg_return:                             0.97524905
+    avg_benchmark_regret:                       0.014990758
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14211273
+    proxy_corner:
+      avg_avg_return:                           0.97524905
+      avg_reward_per_step:                      0.14211273
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.239067
+    avg_avg_return:                             0.9822948
+    avg_benchmark_regret:                       0.010542658
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19685364
+    proxy_corner:
+      avg_avg_return:                           0.0038945442
+      avg_reward_per_step:                      0.0009765625
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.80869
+    avg_avg_return:                             0.03442867
+    avg_benchmark_regret:                       0.9502065
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.011299133
+    proxy_corner:
+      avg_avg_return:                           0.03442867
+      avg_reward_per_step:                      0.011299133
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        189792256
+    env-step-replay-before:                     96501760
+    env-step-used-before:                       189792256
+    ppo-update-before:                          115840
+    env-step-generate-before:                   93290496
+    env-step-all-after:                         189825024
+    env-step-replay-after:                      96534528
+    env-step-used-after:                        189825024
+    ppo-update-after:                           115860
+    env-step-generate-after:                    93290496
+  train-replay:
+    shift_proportion:                           0.9140625
+    avg_avg_episode_length:                     17.753849
+    avg_avg_return:                             0.85726607
+    avg_reward_per_step:                        0.052764893
+    proxy_corner:
+      avg_avg_return:                           0.0734739
+      avg_reward_per_step:                      0.0043640137
+  train-all:
+    shift_proportion:                           0.9140625
+    avg_avg_episode_length:                     17.753849
+    avg_avg_return:                             0.85726607
+    avg_reward_per_step:                        0.052764893
+    proxy_corner:
+      avg_avg_return:                           0.0734739
+      avg_reward_per_step:                      0.0043640137
+  perf:
+    env_steps_per_second:                       239690.74395606807
+    ppo_updates_per_second:                     42.103258770577014
+    eval-batch-train-duration:                  0.6272630598396063
+    eval-batch-orig-duration:                   0.578965101391077
+    eval-batch-shift-duration:                  0.5540765076875687
+    eval-batch-tree-duration:                   0.5666549736633897
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.780029
+      cheese-corne_dist_wavg:                   4.5775743
+      levels_cheese_is_in_corner_avg:           0.07519531
+      mouse-cheese_dist_finite_avg:             17.260498
+      mouse-cheese_dist_finite_wavg:            15.225198
+      mouse-cheese_dist_solvable_avg:           17.260498
+      mouse-cheese_dist_solvable_wavg:          15.225198
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12632394
+    visit_patterns:
+      avg_first_visit_time:                     1709.4045
+      avg_last_visit_time:                      2936.2102
+      num_replay_batches:                       2946
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0021714934
+    avg_actor_approxkl3:                        0.001571306
+    avg_actor_clipfrac:                         0.056359865
+    avg_actor_loss:                             0.0010055492
+    avg_advantage:                              -0.0006423009
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.236934e-05
+    avg_entropy:                                0.11735622
+    avg_grad_norm_pre_clip:                     2.3745735
+    avg_loss:                                   0.00090937753
+    max:
+      max_actor_approxkl1:                      0.00465916
+      max_actor_approxkl3:                      0.0031644169
+      max_actor_clipfrac:                       0.10986328
+      max_actor_loss:                           0.0032373173
+      max_advantage:                            0.05601573
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          4.6487567e-05
+      max_entropy:                              0.13401043
+      max_grad_norm_pre_clip:                   5.130069
+      max_loss:                                 0.0031233411
+    std:
+      std_actor_approxkl1:                      0.0012112199
+      std_actor_approxkl3:                      0.0007044709
+      std_actor_clipfrac:                       0.020926822
+      std_actor_loss:                           0.00094400544
+      std_advantage:                            0.009175745
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.176195e-06
+      std_entropy:                              0.009197295
+      std_grad_norm_pre_clip:                   0.84035754
+      std_loss:                                 0.00094521133
+  eval-batch-train:
+    avg_avg_episode_length:                     7.357901
+    avg_avg_return:                             0.98185754
+    avg_benchmark_regret:                       0.010941681
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20458221
+    proxy_corner:
+      avg_avg_return:                           0.0077412943
+      avg_reward_per_step:                      0.0015792847
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.462341
+    avg_avg_return:                             0.9751928
+    avg_benchmark_regret:                       0.015046982
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14209747
+    proxy_corner:
+      avg_avg_return:                           0.9751928
+      avg_reward_per_step:                      0.14209747
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.269031
+    avg_avg_return:                             0.9820008
+    avg_benchmark_regret:                       0.01083665
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19634247
+    proxy_corner:
+      avg_avg_return:                           0.003987277
+      avg_reward_per_step:                      0.0009918213
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.213715
+    avg_avg_return:                             0.038494248
+    avg_benchmark_regret:                       0.94614094
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.016716003
+    proxy_corner:
+      avg_avg_return:                           0.038494248
+      avg_reward_per_step:                      0.016716003
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        190840832
+    env-step-replay-before:                     96894976
+    env-step-used-before:                       190840832
+    ppo-update-before:                          116480
+    env-step-generate-before:                   93945856
+    env-step-all-after:                         190873600
+    env-step-replay-after:                      96894976
+    env-step-used-after:                        190873600
+    ppo-update-after:                           116500
+    env-step-generate-after:                    93978624
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.895911
+    avg_avg_return:                             0.94582164
+    avg_reward_per_step:                        0.19741821
+    proxy_corner:
+      avg_avg_return:                           0.006324011
+      avg_reward_per_step:                      0.0007019043
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     7.895911
+    avg_avg_return:                             0.94582164
+    avg_reward_per_step:                        0.19741821
+    proxy_corner:
+      avg_avg_return:                           0.006324011
+      avg_reward_per_step:                      0.0007019043
+  perf:
+    env_steps_per_second:                       261556.22126854362
+    ppo_updates_per_second:                     42.254683060202325
+    eval-batch-train-duration:                  0.6737899351865053
+    eval-batch-orig-duration:                   0.5559108490124345
+    eval-batch-shift-duration:                  0.6396374441683292
+    eval-batch-tree-duration:                   0.612332602031529
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.834473
+      cheese-corne_dist_wavg:                   17.16764
+      levels_cheese_is_in_corner_avg:           0.07348633
+      mouse-cheese_dist_finite_avg:             17.240234
+      mouse-cheese_dist_finite_wavg:            14.322973
+      mouse-cheese_dist_solvable_avg:           17.240234
+      mouse-cheese_dist_solvable_wavg:          14.322975
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12792602
+    visit_patterns:
+      avg_first_visit_time:                     1722.9934
+      avg_last_visit_time:                      2947.34
+      num_replay_batches:                       2957
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002250981
+    avg_actor_approxkl3:                        0.0017221585
+    avg_actor_clipfrac:                         0.045581054
+    avg_actor_loss:                             0.0003927204
+    avg_advantage:                              0.0021296917
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            8.1535734e-05
+    avg_entropy:                                0.09773947
+    avg_grad_norm_pre_clip:                     1.7433628
+    avg_loss:                                   0.00033574877
+    max:
+      max_actor_approxkl1:                      0.00592612
+      max_actor_approxkl3:                      0.003605533
+      max_actor_clipfrac:                       0.06774902
+      max_actor_loss:                           0.0034213904
+      max_advantage:                            0.054823935
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.000107203814
+      max_entropy:                              0.116182454
+      max_grad_norm_pre_clip:                   3.0764678
+      max_loss:                                 0.0033642035
+    std:
+      std_actor_approxkl1:                      0.0015902615
+      std_actor_approxkl3:                      0.0010812622
+      std_actor_clipfrac:                       0.017979942
+      std_actor_loss:                           0.0011129093
+      std_advantage:                            0.01247029
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          9.078816e-06
+      std_entropy:                              0.008536895
+      std_grad_norm_pre_clip:                   0.5714237
+      std_loss:                                 0.0011152156
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4212923
+    avg_avg_return:                             0.9819585
+    avg_benchmark_regret:                       0.010840714
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20516205
+    proxy_corner:
+      avg_avg_return:                           0.0076272064
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.463419
+    avg_avg_return:                             0.97513074
+    avg_benchmark_regret:                       0.0151090585
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14211273
+    proxy_corner:
+      avg_avg_return:                           0.97513074
+      avg_reward_per_step:                      0.14211273
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3290515
+    avg_avg_return:                             0.98195034
+    avg_benchmark_regret:                       0.010887105
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19540405
+    proxy_corner:
+      avg_avg_return:                           0.0039405003
+      avg_reward_per_step:                      0.0009918213
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.22279
+    avg_avg_return:                             0.038423784
+    avg_benchmark_regret:                       0.9462114
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.015426636
+    proxy_corner:
+      avg_avg_return:                           0.038423784
+      avg_reward_per_step:                      0.015426636
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        191889408
+    env-step-replay-before:                     97484800
+    env-step-used-before:                       191889408
+    ppo-update-before:                          117120
+    env-step-generate-before:                   94404608
+    env-step-all-after:                         191922176
+    env-step-replay-after:                      97517568
+    env-step-used-after:                        191922176
+    ppo-update-after:                           117140
+    env-step-generate-after:                    94404608
+  train-replay:
+    shift_proportion:                           0.9375
+    avg_avg_episode_length:                     17.757393
+    avg_avg_return:                             0.857218
+    avg_reward_per_step:                        0.05279541
+    proxy_corner:
+      avg_avg_return:                           0.053684644
+      avg_reward_per_step:                      0.0032043457
+  train-all:
+    shift_proportion:                           0.9375
+    avg_avg_episode_length:                     17.757393
+    avg_avg_return:                             0.857218
+    avg_reward_per_step:                        0.05279541
+    proxy_corner:
+      avg_avg_return:                           0.053684644
+      avg_reward_per_step:                      0.0032043457
+  perf:
+    env_steps_per_second:                       261304.78703414963
+    ppo_updates_per_second:                     41.6425258177724
+    eval-batch-train-duration:                  0.3059558253735304
+    eval-batch-orig-duration:                   0.23649114929139614
+    eval-batch-shift-duration:                  0.2695303214713931
+    eval-batch-tree-duration:                   0.22741705272346735
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.8220215
+      cheese-corne_dist_wavg:                   17.201965
+      levels_cheese_is_in_corner_avg:           0.07421875
+      mouse-cheese_dist_finite_avg:             17.248291
+      mouse-cheese_dist_finite_wavg:            17.025703
+      mouse-cheese_dist_solvable_avg:           17.248291
+      mouse-cheese_dist_solvable_wavg:          17.025702
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12748021
+    visit_patterns:
+      avg_first_visit_time:                     1728.2236
+      avg_last_visit_time:                      2966.3335
+      num_replay_batches:                       2976
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00206958
+    avg_actor_approxkl3:                        0.0010467296
+    avg_actor_clipfrac:                         0.041571047
+    avg_actor_loss:                             -0.00020808447
+    avg_advantage:                              -0.00090088975
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.1748033e-05
+    avg_entropy:                                0.112655416
+    avg_grad_norm_pre_clip:                     1.3685966
+    avg_loss:                                   -0.00029986587
+    max:
+      max_actor_approxkl1:                      0.0036394317
+      max_actor_approxkl3:                      0.0016640739
+      max_actor_clipfrac:                       0.056762695
+      max_actor_loss:                           0.00054906216
+      max_advantage:                            0.046450675
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          4.6118967e-05
+      max_entropy:                              0.1231522
+      max_grad_norm_pre_clip:                   1.9146007
+      max_loss:                                 0.00045909185
+    std:
+      std_actor_approxkl1:                      0.000996771
+      std_actor_approxkl3:                      0.00041364273
+      std_actor_clipfrac:                       0.013654669
+      std_actor_loss:                           0.00052181364
+      std_advantage:                            0.009107315
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.6401192e-06
+      std_entropy:                              0.00616269
+      std_grad_norm_pre_clip:                   0.32856476
+      std_loss:                                 0.0005225372
+  eval-batch-train:
+    avg_avg_episode_length:                     7.347694
+    avg_avg_return:                             0.9814313
+    avg_benchmark_regret:                       0.011367861
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20594025
+    proxy_corner:
+      avg_avg_return:                           0.0077094086
+      avg_reward_per_step:                      0.0015792847
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.463666
+    avg_avg_return:                             0.97494334
+    avg_benchmark_regret:                       0.015296487
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14208984
+    proxy_corner:
+      avg_avg_return:                           0.97494334
+      avg_reward_per_step:                      0.14208984
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3055205
+    avg_avg_return:                             0.9815297
+    avg_benchmark_regret:                       0.011307782
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19558716
+    proxy_corner:
+      avg_avg_return:                           0.0039630686
+      avg_reward_per_step:                      0.0009918213
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.88868
+    avg_avg_return:                             0.03108992
+    avg_benchmark_regret:                       0.9535453
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.012397766
+    proxy_corner:
+      avg_avg_return:                           0.03108992
+      avg_reward_per_step:                      0.012397766
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        192937984
+    env-step-replay-before:                     97910784
+    env-step-used-before:                       192937984
+    ppo-update-before:                          117760
+    env-step-generate-before:                   95027200
+    env-step-all-after:                         192970752
+    env-step-replay-after:                      97943552
+    env-step-used-after:                        192970752
+    ppo-update-after:                           117780
+    env-step-generate-after:                    95027200
+  train-replay:
+    shift_proportion:                           0.93359375
+    avg_avg_episode_length:                     18.034039
+    avg_avg_return:                             0.8550104
+    avg_reward_per_step:                        0.05227661
+    proxy_corner:
+      avg_avg_return:                           0.058172613
+      avg_reward_per_step:                      0.003540039
+  train-all:
+    shift_proportion:                           0.93359375
+    avg_avg_episode_length:                     18.034039
+    avg_avg_return:                             0.8550104
+    avg_reward_per_step:                        0.05227661
+    proxy_corner:
+      avg_avg_return:                           0.058172613
+      avg_reward_per_step:                      0.003540039
+  perf:
+    env_steps_per_second:                       327551.7067431835
+    ppo_updates_per_second:                     40.90182547579856
+    eval-batch-train-duration:                  0.5240151630714536
+    eval-batch-orig-duration:                   0.4416998429223895
+    eval-batch-shift-duration:                  0.476206467486918
+    eval-batch-tree-duration:                   0.47847185656428337
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.869873
+      cheese-corne_dist_wavg:                   15.3825035
+      levels_cheese_is_in_corner_avg:           0.07397461
+      mouse-cheese_dist_finite_avg:             17.249756
+      mouse-cheese_dist_finite_wavg:            12.525792
+      mouse-cheese_dist_solvable_avg:           17.249756
+      mouse-cheese_dist_solvable_wavg:          12.525794
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12725046
+    visit_patterns:
+      avg_first_visit_time:                     1739.1794
+      avg_last_visit_time:                      2979.2214
+      num_replay_batches:                       2989
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001670884
+    avg_actor_approxkl3:                        0.0011777446
+    avg_actor_clipfrac:                         0.042163085
+    avg_actor_loss:                             0.00027396888
+    avg_advantage:                              -0.0002840313
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            3.7150476e-05
+    avg_entropy:                                0.11401587
+    avg_grad_norm_pre_clip:                     1.8187335
+    avg_loss:                                   0.00017852824
+    max:
+      max_actor_approxkl1:                      0.0034735436
+      max_actor_approxkl3:                      0.0022290484
+      max_actor_clipfrac:                       0.06518555
+      max_actor_loss:                           0.0021325676
+      max_advantage:                            0.047127843
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          4.0323437e-05
+      max_entropy:                              0.12705228
+      max_grad_norm_pre_clip:                   2.9459522
+      max_loss:                                 0.0020350225
+    std:
+      std_actor_approxkl1:                      0.00095762283
+      std_actor_approxkl3:                      0.00064179057
+      std_actor_clipfrac:                       0.016291626
+      std_actor_loss:                           0.0009383618
+      std_advantage:                            0.0085388385
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.8893298e-06
+      std_entropy:                              0.006538553
+      std_grad_norm_pre_clip:                   0.55307275
+      std_loss:                                 0.0009351583
+  eval-batch-train:
+    avg_avg_episode_length:                     7.446417
+    avg_avg_return:                             0.9814839
+    avg_benchmark_regret:                       0.011315275
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20375061
+    proxy_corner:
+      avg_avg_return:                           0.007788093
+      avg_reward_per_step:                      0.0015869141
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.466049
+    avg_avg_return:                             0.9751771
+    avg_benchmark_regret:                       0.015062693
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1420517
+    proxy_corner:
+      avg_avg_return:                           0.9751771
+      avg_reward_per_step:                      0.1420517
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3849487
+    avg_avg_return:                             0.9818837
+    avg_benchmark_regret:                       0.010953726
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19428253
+    proxy_corner:
+      avg_avg_return:                           0.0042612464
+      avg_reward_per_step:                      0.0010604858
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.80136
+    avg_avg_return:                             0.03344491
+    avg_benchmark_regret:                       0.9511903
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.011276245
+    proxy_corner:
+      avg_avg_return:                           0.03344491
+      avg_reward_per_step:                      0.011276245
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        193986560
+    env-step-replay-before:                     98402304
+    env-step-used-before:                       193986560
+    ppo-update-before:                          118400
+    env-step-generate-before:                   95584256
+    env-step-all-after:                         194019328
+    env-step-replay-after:                      98435072
+    env-step-used-after:                        194019328
+    ppo-update-after:                           118420
+    env-step-generate-after:                    95584256
+  train-replay:
+    shift_proportion:                           0.92578125
+    avg_avg_episode_length:                     17.679684
+    avg_avg_return:                             0.85859525
+    avg_reward_per_step:                        0.053009033
+    proxy_corner:
+      avg_avg_return:                           0.06508024
+      avg_reward_per_step:                      0.0038757324
+  train-all:
+    shift_proportion:                           0.92578125
+    avg_avg_episode_length:                     17.679684
+    avg_avg_return:                             0.85859525
+    avg_reward_per_step:                        0.053009033
+    proxy_corner:
+      avg_avg_return:                           0.06508024
+      avg_reward_per_step:                      0.0038757324
+  perf:
+    env_steps_per_second:                       264954.0668594547
+    ppo_updates_per_second:                     41.85930471225876
+    eval-batch-train-duration:                  0.5327203404158354
+    eval-batch-orig-duration:                   0.42660355661064386
+    eval-batch-shift-duration:                  0.45671179983764887
+    eval-batch-tree-duration:                   0.44447231851518154
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.8532715
+      cheese-corne_dist_wavg:                   9.977398
+      levels_cheese_is_in_corner_avg:           0.07446289
+      mouse-cheese_dist_finite_avg:             17.25
+      mouse-cheese_dist_finite_wavg:            17.02549
+      mouse-cheese_dist_solvable_avg:           17.25
+      mouse-cheese_dist_solvable_wavg:          17.025492
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12672763
+    visit_patterns:
+      avg_first_visit_time:                     1742.9712
+      avg_last_visit_time:                      2994.2751
+      num_replay_batches:                       3004
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002532112
+    avg_actor_approxkl3:                        0.0016130833
+    avg_actor_clipfrac:                         0.04713135
+    avg_actor_loss:                             0.001231163
+    avg_advantage:                              -0.0028121544
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.1811567e-05
+    avg_entropy:                                0.11567607
+    avg_grad_norm_pre_clip:                     1.9378246
+    avg_loss:                                   0.0011363928
+    max:
+      max_actor_approxkl1:                      0.0054544266
+      max_actor_approxkl3:                      0.0037622089
+      max_actor_clipfrac:                       0.0736084
+      max_actor_loss:                           0.004513544
+      max_advantage:                            0.043677747
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          5.0279614e-05
+      max_entropy:                              0.13124265
+      max_grad_norm_pre_clip:                   3.1694531
+      max_loss:                                 0.0044205817
+    std:
+      std_actor_approxkl1:                      0.0014458029
+      std_actor_approxkl3:                      0.0008882797
+      std_actor_clipfrac:                       0.016599
+      std_actor_loss:                           0.0012985251
+      std_advantage:                            0.008777328
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.383474e-06
+      std_entropy:                              0.008594447
+      std_grad_norm_pre_clip:                   0.7066259
+      std_loss:                                 0.0012962099
+  eval-batch-train:
+    avg_avg_episode_length:                     7.358597
+    avg_avg_return:                             0.981608
+    avg_benchmark_regret:                       0.011191215
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20488739
+    proxy_corner:
+      avg_avg_return:                           0.007842706
+      avg_reward_per_step:                      0.0016021729
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.466975
+    avg_avg_return:                             0.9747211
+    avg_benchmark_regret:                       0.01551878
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14200592
+    proxy_corner:
+      avg_avg_return:                           0.9747211
+      avg_reward_per_step:                      0.14200592
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3081675
+    avg_avg_return:                             0.9815946
+    avg_benchmark_regret:                       0.011242778
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19487
+    proxy_corner:
+      avg_avg_return:                           0.0038945442
+      avg_reward_per_step:                      0.0009765625
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.776146
+    avg_avg_return:                             0.033934474
+    avg_benchmark_regret:                       0.95070076
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.011428833
+    proxy_corner:
+      avg_avg_return:                           0.033934474
+      avg_reward_per_step:                      0.011428833
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        195035136
+    env-step-replay-before:                     98762752
+    env-step-used-before:                       195035136
+    ppo-update-before:                          119040
+    env-step-generate-before:                   96272384
+    env-step-all-after:                         195067904
+    env-step-replay-after:                      98795520
+    env-step-used-after:                        195067904
+    ppo-update-after:                           119060
+    env-step-generate-after:                    96272384
+  train-replay:
+    shift_proportion:                           0.921875
+    avg_avg_episode_length:                     17.661457
+    avg_avg_return:                             0.8585225
+    avg_reward_per_step:                        0.053131104
+    proxy_corner:
+      avg_avg_return:                           0.06720303
+      avg_reward_per_step:                      0.004058838
+  train-all:
+    shift_proportion:                           0.921875
+    avg_avg_episode_length:                     17.661457
+    avg_avg_return:                             0.8585225
+    avg_reward_per_step:                        0.053131104
+    proxy_corner:
+      avg_avg_return:                           0.06720303
+      avg_reward_per_step:                      0.004058838
+  perf:
+    env_steps_per_second:                       297319.7403113273
+    ppo_updates_per_second:                     42.43587147990876
+    eval-batch-train-duration:                  0.5188061874359846
+    eval-batch-orig-duration:                   0.47765565756708384
+    eval-batch-shift-duration:                  0.4928294410929084
+    eval-batch-tree-duration:                   0.4470509272068739
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.889893
+      cheese-corne_dist_wavg:                   9.980992
+      levels_cheese_is_in_corner_avg:           0.07421875
+      mouse-cheese_dist_finite_avg:             17.24292
+      mouse-cheese_dist_finite_wavg:            17.022793
+      mouse-cheese_dist_solvable_avg:           17.24292
+      mouse-cheese_dist_solvable_wavg:          17.022797
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12674065
+    visit_patterns:
+      avg_first_visit_time:                     1751.6257
+      avg_last_visit_time:                      3005.3428
+      num_replay_batches:                       3015
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0023620974
+    avg_actor_approxkl3:                        0.0016026
+    avg_actor_clipfrac:                         0.048638918
+    avg_actor_loss:                             0.00037460885
+    avg_advantage:                              -0.0010255295
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.365358e-05
+    avg_entropy:                                0.1219188
+    avg_grad_norm_pre_clip:                     1.4842314
+    avg_loss:                                   0.00027451685
+    max:
+      max_actor_approxkl1:                      0.0059271166
+      max_actor_approxkl3:                      0.0040099574
+      max_actor_clipfrac:                       0.08520508
+      max_actor_loss:                           0.0021412554
+      max_advantage:                            0.046618223
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          4.82372e-05
+      max_entropy:                              0.13951738
+      max_grad_norm_pre_clip:                   2.1670976
+      max_loss:                                 0.0020309507
+    std:
+      std_actor_approxkl1:                      0.0014102794
+      std_actor_approxkl3:                      0.00093951914
+      std_actor_clipfrac:                       0.019910654
+      std_actor_loss:                           0.00068866723
+      std_advantage:                            0.009591142
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.466332e-06
+      std_entropy:                              0.00942356
+      std_grad_norm_pre_clip:                   0.37977567
+      std_loss:                                 0.0006858981
+  eval-batch-train:
+    avg_avg_episode_length:                     7.437175
+    avg_avg_return:                             0.9818667
+    avg_benchmark_regret:                       0.010932442
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20459747
+    proxy_corner:
+      avg_avg_return:                           0.0077094086
+      avg_reward_per_step:                      0.0015792847
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.462191
+    avg_avg_return:                             0.9750827
+    avg_benchmark_regret:                       0.015157167
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14214325
+    proxy_corner:
+      avg_avg_return:                           0.9750827
+      avg_reward_per_step:                      0.14214325
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.333688
+    avg_avg_return:                             0.9818009
+    avg_benchmark_regret:                       0.011036498
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19506073
+    proxy_corner:
+      avg_avg_return:                           0.0038945442
+      avg_reward_per_step:                      0.0009765625
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.314445
+    avg_avg_return:                             0.037902925
+    avg_benchmark_regret:                       0.9467323
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.01423645
+    proxy_corner:
+      avg_avg_return:                           0.037902925
+      avg_reward_per_step:                      0.01423645
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        196083712
+    env-step-replay-before:                     99319808
+    env-step-used-before:                       196083712
+    ppo-update-before:                          119680
+    env-step-generate-before:                   96763904
+    env-step-all-after:                         196116480
+    env-step-replay-after:                      99319808
+    env-step-used-after:                        196116480
+    ppo-update-after:                           119700
+    env-step-generate-after:                    96796672
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.4972076
+    avg_avg_return:                             0.94700456
+    avg_reward_per_step:                        0.1885376
+    proxy_corner:
+      avg_avg_return:                           0.007823352
+      avg_reward_per_step:                      0.0009765625
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.4972076
+    avg_avg_return:                             0.94700456
+    avg_reward_per_step:                        0.1885376
+    proxy_corner:
+      avg_avg_return:                           0.007823352
+      avg_reward_per_step:                      0.0009765625
+  perf:
+    env_steps_per_second:                       298196.39692175144
+    ppo_updates_per_second:                     41.9752120324348
+    eval-batch-train-duration:                  0.40865332167595625
+    eval-batch-orig-duration:                   0.22379132453352213
+    eval-batch-shift-duration:                  0.2505305940285325
+    eval-batch-tree-duration:                   0.22422412410378456
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.88916
+      cheese-corne_dist_wavg:                   3.6855493
+      levels_cheese_is_in_corner_avg:           0.07519531
+      mouse-cheese_dist_finite_avg:             17.249756
+      mouse-cheese_dist_finite_wavg:            17.024767
+      mouse-cheese_dist_solvable_avg:           17.249756
+      mouse-cheese_dist_solvable_wavg:          17.024767
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12736754
+    visit_patterns:
+      avg_first_visit_time:                     1759.3601
+      avg_last_visit_time:                      3021.2969
+      num_replay_batches:                       3031
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0017132981
+    avg_actor_approxkl3:                        0.0012340534
+    avg_actor_clipfrac:                         0.03932495
+    avg_actor_loss:                             0.00029013617
+    avg_advantage:                              -0.0008518716
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            8.116189e-05
+    avg_entropy:                                0.09361892
+    avg_grad_norm_pre_clip:                     1.6988213
+    avg_loss:                                   0.00023709818
+    max:
+      max_actor_approxkl1:                      0.004659199
+      max_actor_approxkl3:                      0.003022446
+      max_actor_clipfrac:                       0.07458496
+      max_actor_loss:                           0.001729886
+      max_advantage:                            0.058871567
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          9.641631e-05
+      max_entropy:                              0.113943234
+      max_grad_norm_pre_clip:                   2.570958
+      max_loss:                                 0.0016683885
+    std:
+      std_actor_approxkl1:                      0.0013828563
+      std_actor_approxkl3:                      0.0007360419
+      std_actor_clipfrac:                       0.017377997
+      std_actor_loss:                           0.0007369154
+      std_advantage:                            0.012689213
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          6.362186e-06
+      std_entropy:                              0.009542016
+      std_grad_norm_pre_clip:                   0.5379505
+      std_loss:                                 0.0007336675
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3494697
+    avg_avg_return:                             0.9814304
+    avg_benchmark_regret:                       0.011368803
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20484161
+    proxy_corner:
+      avg_avg_return:                           0.007788831
+      avg_reward_per_step:                      0.0015945435
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.463257
+    avg_avg_return:                             0.97505707
+    avg_benchmark_regret:                       0.01518279
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.1421051
+    proxy_corner:
+      avg_avg_return:                           0.97505707
+      avg_reward_per_step:                      0.1421051
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.3325777
+    avg_avg_return:                             0.9818771
+    avg_benchmark_regret:                       0.010960368
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19493866
+    proxy_corner:
+      avg_avg_return:                           0.0038945442
+      avg_reward_per_step:                      0.0009765625
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.60487
+    avg_avg_return:                             0.035932563
+    avg_benchmark_regret:                       0.94870263
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.011390686
+    proxy_corner:
+      avg_avg_return:                           0.035932563
+      avg_reward_per_step:                      0.011390686
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        197132288
+    env-step-replay-before:                     99876864
+    env-step-used-before:                       197132288
+    ppo-update-before:                          120320
+    env-step-generate-before:                   97255424
+    env-step-all-after:                         197165056
+    env-step-replay-after:                      99876864
+    env-step-used-after:                        197165056
+    ppo-update-after:                           120340
+    env-step-generate-after:                    97288192
+  train-generate:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     7.703296
+    avg_avg_return:                             0.9484533
+    avg_reward_per_step:                        0.18591309
+    proxy_corner:
+      avg_avg_return:                           0.01491408
+      avg_reward_per_step:                      0.0027160645
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     7.703296
+    avg_avg_return:                             0.9484533
+    avg_reward_per_step:                        0.18591309
+    proxy_corner:
+      avg_avg_return:                           0.01491408
+      avg_reward_per_step:                      0.0027160645
+  perf:
+    env_steps_per_second:                       327102.9825940456
+    ppo_updates_per_second:                     42.617101290023626
+    eval-batch-train-duration:                  0.5115273538976908
+    eval-batch-orig-duration:                   0.48704317957162857
+    eval-batch-shift-duration:                  0.47601662762463093
+    eval-batch-tree-duration:                   0.45067067816853523
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.916504
+      cheese-corne_dist_wavg:                   9.080468
+      levels_cheese_is_in_corner_avg:           0.07519531
+      mouse-cheese_dist_finite_avg:             17.234375
+      mouse-cheese_dist_finite_wavg:            17.01484
+      mouse-cheese_dist_solvable_avg:           17.234375
+      mouse-cheese_dist_solvable_wavg:          17.01484
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12713432
+    visit_patterns:
+      avg_first_visit_time:                     1769.1951
+      avg_last_visit_time:                      3038.1882
+      num_replay_batches:                       3048
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002392954
+    avg_actor_approxkl3:                        0.0018210888
+    avg_actor_clipfrac:                         0.04573364
+    avg_actor_loss:                             0.0014117853
+    avg_advantage:                              0.0016724989
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            7.593477e-05
+    avg_entropy:                                0.08840579
+    avg_grad_norm_pre_clip:                     2.042443
+    avg_loss:                                   0.0013613469
+    max:
+      max_actor_approxkl1:                      0.005620292
+      max_actor_approxkl3:                      0.003713924
+      max_actor_clipfrac:                       0.068359375
+      max_actor_loss:                           0.004086411
+      max_advantage:                            0.07203686
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          9.7376826e-05
+      max_entropy:                              0.10594846
+      max_grad_norm_pre_clip:                   3.3422313
+      max_loss:                                 0.0040459507
+    std:
+      std_actor_approxkl1:                      0.0011401242
+      std_actor_approxkl3:                      0.00075930654
+      std_actor_clipfrac:                       0.01410846
+      std_actor_loss:                           0.001153946
+      std_advantage:                            0.012222121
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.3407173e-06
+      std_entropy:                              0.009499874
+      std_grad_norm_pre_clip:                   0.74078876
+      std_loss:                                 0.001157197
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3220634
+    avg_avg_return:                             0.9819243
+    avg_benchmark_regret:                       0.010874852
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20585632
+    proxy_corner:
+      avg_avg_return:                           0.007848863
+      avg_reward_per_step:                      0.0015869141
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.46222
+    avg_avg_return:                             0.9750664
+    avg_benchmark_regret:                       0.015173384
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14213562
+    proxy_corner:
+      avg_avg_return:                           0.9750664
+      avg_reward_per_step:                      0.14213562
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.408575
+    avg_avg_return:                             0.9812095
+    avg_benchmark_regret:                       0.011627879
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19412994
+    proxy_corner:
+      avg_avg_return:                           0.004138702
+      avg_reward_per_step:                      0.0010375977
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.66221
+    avg_avg_return:                             0.03573532
+    avg_benchmark_regret:                       0.94889987
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.013076782
+    proxy_corner:
+      avg_avg_return:                           0.03573532
+      avg_reward_per_step:                      0.013076782
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        198180864
+    env-step-replay-before:                     100335616
+    env-step-used-before:                       198180864
+    ppo-update-before:                          120960
+    env-step-generate-before:                   97845248
+    env-step-all-after:                         198213632
+    env-step-replay-after:                      100368384
+    env-step-used-after:                        198213632
+    ppo-update-after:                           120980
+    env-step-generate-after:                    97845248
+  train-replay:
+    shift_proportion:                           0.921875
+    avg_avg_episode_length:                     17.913052
+    avg_avg_return:                             0.8570845
+    avg_reward_per_step:                        0.05230713
+    proxy_corner:
+      avg_avg_return:                           0.066580266
+      avg_reward_per_step:                      0.0039978027
+  train-all:
+    shift_proportion:                           0.921875
+    avg_avg_episode_length:                     17.913052
+    avg_avg_return:                             0.8570845
+    avg_reward_per_step:                        0.05230713
+    proxy_corner:
+      avg_avg_return:                           0.066580266
+      avg_reward_per_step:                      0.0039978027
+  perf:
+    env_steps_per_second:                       392230.39413433586
+    ppo_updates_per_second:                     42.24317096419274
+    eval-batch-train-duration:                  0.4365055412054062
+    eval-batch-orig-duration:                   0.4021034426987171
+    eval-batch-shift-duration:                  0.4186759013682604
+    eval-batch-tree-duration:                   0.41429199930280447
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.93042
+      cheese-corne_dist_wavg:                   17.186636
+      levels_cheese_is_in_corner_avg:           0.07470703
+      mouse-cheese_dist_finite_avg:             17.238525
+      mouse-cheese_dist_finite_wavg:            17.024004
+      mouse-cheese_dist_solvable_avg:           17.238525
+      mouse-cheese_dist_solvable_wavg:          17.024008
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12702316
+    visit_patterns:
+      avg_first_visit_time:                     1775.9084
+      avg_last_visit_time:                      3053.3672
+      num_replay_batches:                       3063
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013448031
+    avg_actor_approxkl3:                        0.0009588385
+    avg_actor_clipfrac:                         0.035888672
+    avg_actor_loss:                             -4.697887e-05
+    avg_advantage:                              -0.0016665442
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.0881554e-05
+    avg_entropy:                                0.11410663
+    avg_grad_norm_pre_clip:                     1.5515182
+    avg_loss:                                   -0.00014064471
+    max:
+      max_actor_approxkl1:                      0.003095198
+      max_actor_approxkl3:                      0.0022118075
+      max_actor_clipfrac:                       0.05859375
+      max_actor_loss:                           0.0012010159
+      max_advantage:                            0.05521828
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          4.7834015e-05
+      max_entropy:                              0.121185616
+      max_grad_norm_pre_clip:                   2.7157989
+      max_loss:                                 0.0011063422
+    std:
+      std_actor_approxkl1:                      0.0009497316
+      std_actor_approxkl3:                      0.00059579243
+      std_actor_clipfrac:                       0.016191006
+      std_actor_loss:                           0.0005374878
+      std_advantage:                            0.008779111
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.1892753e-06
+      std_entropy:                              0.0042651193
+      std_grad_norm_pre_clip:                   0.504088
+      std_loss:                                 0.00053614797
+  eval-batch-train:
+    avg_avg_episode_length:                     7.3775287
+    avg_avg_return:                             0.98135686
+    avg_benchmark_regret:                       0.011442275
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20536804
+    proxy_corner:
+      avg_avg_return:                           0.0076272064
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.47196
+    avg_avg_return:                             0.9751724
+    avg_benchmark_regret:                       0.0150673855
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14204407
+    proxy_corner:
+      avg_avg_return:                           0.9751724
+      avg_reward_per_step:                      0.14204407
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.389846
+    avg_avg_return:                             0.98137873
+    avg_benchmark_regret:                       0.011458725
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19438934
+    proxy_corner:
+      avg_avg_return:                           0.0038945442
+      avg_reward_per_step:                      0.0009765625
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.49617
+    avg_avg_return:                             0.035954457
+    avg_benchmark_regret:                       0.94868076
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.013923645
+    proxy_corner:
+      avg_avg_return:                           0.035954457
+      avg_reward_per_step:                      0.013923645
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        199229440
+    env-step-replay-before:                     100728832
+    env-step-used-before:                       199229440
+    ppo-update-before:                          121600
+    env-step-generate-before:                   98500608
+    env-step-all-after:                         199262208
+    env-step-replay-after:                      100728832
+    env-step-used-after:                        199262208
+    ppo-update-after:                           121620
+    env-step-generate-after:                    98533376
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.545292
+    avg_avg_return:                             0.9484216
+    avg_reward_per_step:                        0.18063354
+    proxy_corner:
+      avg_avg_return:                           0.00909514
+      avg_reward_per_step:                      0.0009460449
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     7.545292
+    avg_avg_return:                             0.9484216
+    avg_reward_per_step:                        0.18063354
+    proxy_corner:
+      avg_avg_return:                           0.00909514
+      avg_reward_per_step:                      0.0009460449
+  perf:
+    env_steps_per_second:                       298915.5436743963
+    ppo_updates_per_second:                     41.614397448624054
+    eval-batch-train-duration:                  0.5364610189571977
+    eval-batch-orig-duration:                   0.4923221170902252
+    eval-batch-shift-duration:                  0.5005645146593451
+    eval-batch-tree-duration:                   0.4865581849589944
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    9.99585
+      cheese-corne_dist_wavg:                   17.195963
+      levels_cheese_is_in_corner_avg:           0.07373047
+      mouse-cheese_dist_finite_avg:             17.220459
+      mouse-cheese_dist_finite_wavg:            16.123737
+      mouse-cheese_dist_solvable_avg:           17.220459
+      mouse-cheese_dist_solvable_wavg:          16.123737
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12822571
+    visit_patterns:
+      avg_first_visit_time:                     1795.9175
+      avg_last_visit_time:                      3064.3242
+      num_replay_batches:                       3074
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0029367572
+    avg_actor_approxkl3:                        0.0023499287
+    avg_actor_clipfrac:                         0.050427247
+    avg_actor_loss:                             0.0021528024
+    avg_advantage:                              0.0014507454
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            7.412806e-05
+    avg_entropy:                                0.09888681
+    avg_grad_norm_pre_clip:                     2.4266348
+    avg_loss:                                   0.0020909794
+    max:
+      max_actor_approxkl1:                      0.0072810333
+      max_actor_approxkl3:                      0.005416679
+      max_actor_clipfrac:                       0.08618164
+      max_actor_loss:                           0.006009464
+      max_advantage:                            0.06011355
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          8.262266e-05
+      max_entropy:                              0.11915319
+      max_grad_norm_pre_clip:                   4.798775
+      max_loss:                                 0.005938304
+    std:
+      std_actor_approxkl1:                      0.0020507385
+      std_actor_approxkl3:                      0.0014679388
+      std_actor_clipfrac:                       0.021293517
+      std_actor_loss:                           0.0015033531
+      std_advantage:                            0.012078679
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.1707312e-06
+      std_entropy:                              0.011185217
+      std_grad_norm_pre_clip:                   0.8955571
+      std_loss:                                 0.0015003559
+  eval-batch-train:
+    avg_avg_episode_length:                     7.4134235
+    avg_avg_return:                             0.9817493
+    avg_benchmark_regret:                       0.011049833
+    avg_benchmark_return:                       0.99279916
+    avg_reward_per_step:                        0.20472717
+    proxy_corner:
+      avg_avg_return:                           0.007898478
+      avg_reward_per_step:                      0.0015945435
+  eval-batch-orig:
+    avg_avg_episode_length:                     9.463073
+    avg_avg_return:                             0.97508776
+    avg_benchmark_regret:                       0.015152076
+    avg_benchmark_return:                       0.99023986
+    avg_reward_per_step:                        0.14211273
+    proxy_corner:
+      avg_avg_return:                           0.97508776
+      avg_reward_per_step:                      0.14211273
+  eval-batch-shift:
+    avg_avg_episode_length:                     7.410384
+    avg_avg_return:                             0.98172855
+    avg_benchmark_regret:                       0.011108886
+    avg_benchmark_return:                       0.9928374
+    avg_reward_per_step:                        0.19413757
+    proxy_corner:
+      avg_avg_return:                           0.0039324807
+      avg_reward_per_step:                      0.0009765625
+  eval-batch-tree:
+    avg_avg_episode_length:                     82.24053
+    avg_avg_return:                             0.03979186
+    avg_benchmark_regret:                       0.9448433
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.018875122
+    proxy_corner:
+      avg_avg_return:                           0.03979186
+      avg_reward_per_step:                      0.018875122
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+training run complete.
+finishing checkpoints...
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/requirements.txt b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/requirements.txt
new file mode 100644
index 0000000000000000000000000000000000000000..1d52de8af834763af26ecb7bea4d057f7b5df2d1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/requirements.txt
@@ -0,0 +1,252 @@
+kiwisolver==1.4.8
+aulis==0.1.0
+tokenizers==0.20.3
+xmltodict==0.14.2
+hf-xet==1.1.4
+timaeus-cli==0.1.0
+jsonlines==4.0.0
+DataProperty==1.1.0
+timaeus-aether==0.0.0
+nvidia-nvtx-cu12==12.6.77
+cachetools==5.5.2
+devinterp==1.3.2
+wcwidth==0.2.13
+pydantic_core==2.33.2
+tqdm-multiprocess==0.0.11
+protobuf==5.29.5
+decorator==4.4.2
+pyasn1==0.6.1
+zipp==3.23.0
+pre_commit==4.2.0
+typepy==1.3.4
+etils==1.13.0
+shellingham==1.5.4
+aiohttp==3.12.13
+prettytable==3.16.0
+readchar==4.2.1
+sentencepiece==0.2.0
+seaborn==0.13.2
+paramiko==3.5.1
+cftime==1.6.4.post1
+tabledata==1.3.4
+colorama==0.4.6
+GitPython==3.1.44
+cycler==0.12.1
+tcolorpy==0.1.7
+cfgv==3.4.0
+psutil==7.0.0
+aioitertools==0.12.0
+aiosignal==1.3.2
+humanize==4.12.3
+termcolor==3.1.0
+zope.interface==7.2
+pathvalidate==3.3.1
+nodeenv==1.9.1
+mbstrdecoder==1.1.4
+chardet==5.2.0
+zope.event==5.0
+cloudpickle==3.1.1
+opt_einsum==3.4.0
+MarkupSafe==3.0.2
+requests_ntlm==1.3.0
+pybind11==2.13.6
+fonttools==4.58.4
+gast==0.6.0
+packaging==25.0
+proglog==0.1.12
+Jinja2==3.1.6
+wadler_lindig==0.1.6
+fancy-einsum==0.0.3
+iniconfig==2.1.0
+pluggy==1.6.0
+idna==3.10
+nest-asyncio==1.6.0
+better-abc==0.0.3
+matplotlib==3.10.3
+portalocker==3.2.0
+tqdm==4.67.1
+smmap==5.0.2
+rich==14.0.0
+distro==1.9.0
+xarray==2025.6.1
+bcrypt==4.3.0
+regex==2024.11.6
+aiobotocore==2.22.0
+jmespath==1.0.1
+frozenlist==1.7.0
+typer==0.16.0
+cryptography==45.0.4
+python-dotenv==1.1.0
+pytest-xdist==3.7.0
+boto3==1.37.3
+grpc-google-iam-v1==0.14.2
+threadpoolctl==3.6.0
+aiohappyeyeballs==2.6.1
+pywinrm==0.5.0
+platformdirs==4.3.8
+nvidia-cuda-runtime-cu12==12.6.77
+aiofiles==24.1.0
+google-api-core==2.25.1
+proto-plus==1.26.1
+nvidia-cufile-cu12==1.11.1.6
+mpmath==1.3.0
+PyYAML==6.0.2
+filelock==3.18.0
+typing-inspection==0.4.1
+syrupy==4.9.1
+distlib==0.3.9
+annotated-types==0.7.0
+more-itertools==10.7.0
+multiprocess==0.70.16
+grpcio-status==1.71.2
+six==1.17.0
+requests==2.32.4
+mdurl==0.1.2
+pyinfra==3.3.1
+typing_extensions==4.14.0
+greenlet==3.2.3
+lxml==5.4.0
+einops==0.8.1
+ssh-python==1.1.1
+networkx==3.5
+attrs==25.3.0
+pyasn1_modules==0.4.2
+toolz==1.0.0
+joblib==1.5.1
+diskcache==5.6.3
+multidict==6.5.0
+gitdb==4.0.12
+transformer-lens==2.15.4
+execnet==2.1.1
+safetensors==0.5.3
+rsa==4.9.1
+zstandard==0.23.0
+google-cloud-compute==1.31.0
+rouge_score==0.1.2
+importlib_resources==6.5.2
+cffi==1.17.1
+pyspnego==0.11.2
+aioboto3==14.3.0
+Pygments==2.19.1
+tabulate==0.9.0
+typeguard==4.4.3
+click==8.2.1
+nvidia-cuda-cupti-cu12==12.6.80
+word2number==1.1
+pycparser==2.22
+gevent==25.5.1
+transformers-stream-generator==0.0.5
+sqlitedict==2.1.0
+datasets==3.6.0
+netCDF4==1.7.2
+msgpack==1.1.1
+urllib3==2.4.0
+pillow==11.2.1
+PyNaCl==1.5.0
+parallel-ssh==2.14.0
+nltk==3.9.1
+python-dateutil==2.9.0.post0
+yarl==1.20.1
+pyparsing==3.2.3
+ml_dtypes==0.5.3
+evaluate==0.4.3
+numexpr==2.11.0
+pydantic==2.11.7
+ssh2-python==1.1.2.post1
+moviepy==2.2.1
+markdown-it-py==3.0.0
+googleapis-common-protos==1.70.0
+xxhash==3.5.0
+dill==0.3.8
+sacrebleu==2.5.1
+identify==2.6.12
+fsspec==2025.3.0
+dm-tree==0.1.9
+chex==0.1.89
+huggingface-hub==0.33.0
+certifi==2025.6.15
+imageio-ffmpeg==0.6.0
+scipy==1.15.3
+pytz==2025.2
+pytablewriter==1.2.1
+optax==0.2.5
+google-cloud-tpu==1.23.2
+nvidia-curand-cu12==10.3.7.77
+charset-normalizer==3.4.2
+wandb-workspaces==0.1.16
+s3transfer==0.11.3
+absl-py==2.3.0
+accelerate==1.7.0
+wrapt==1.17.2
+narwhals==1.43.0
+virtualenv==20.31.2
+peft==0.15.2
+triton==3.3.1
+sympy==1.14.0
+nvidia-cusolver-cu12==11.7.1.2
+contourpy==1.3.2
+nvidia-cusparselt-cu12==0.6.3
+nvidia-cufft-cu12==11.3.0.4
+ruff==0.12.0
+nvidia-nccl-cu12==2.26.2
+nvidia-cusparse-cu12==12.5.4.2
+google-cloud-secret-manager==2.24.0
+pytest==7.4.3
+google-auth==2.40.3
+beartype==0.14.1
+sentry-sdk==2.30.0
+distrax==0.1.5
+grpcio==1.73.0
+pyarrow==20.0.0
+tensorflow-probability==0.25.0
+imageio==2.37.0
+wandb==0.21.0
+propcache==0.3.2
+tensorstore==0.1.76
+numpy==2.3.1
+nvidia-cuda-nvrtc-cu12==12.6.77
+nvidia-nvjitlink-cu12==12.6.85
+setuptools==80.9.0
+tzdata==2025.2
+transformers==4.45.2
+pandas==2.3.0
+scikit-learn==1.7.0
+lm_eval==0.4.5
+botocore==1.37.3
+torch==2.7.1
+plotly==6.1.2
+ipython_pygments_lexers==1.1.1
+comm==0.2.3
+matplotlib-inline==0.1.7
+ptyprocess==0.7.0
+jupyter_core==5.8.1
+executing==2.2.0
+pexpect==4.9.0
+pure_eval==0.2.3
+asttokens==3.0.0
+stack-data==0.6.3
+traitlets==5.14.3
+jupyter_client==8.6.3
+parso==0.8.4
+ipykernel==6.30.1
+pyzmq==27.0.1
+tornado==6.5.1
+ipython==9.4.0
+prompt_toolkit==3.0.51
+debugpy==1.8.16
+jedi==0.19.2
+jaxtyping==0.3.2
+jax-cuda12-plugin==0.6.2
+jax-cuda12-pjrt==0.6.2
+jaxlib==0.6.2
+jax==0.6.2
+nvidia-cublas-cu12==12.6.4.1
+nvidia-cudnn-cu12==9.5.1.17
+treescope==0.1.9
+flax==0.11.0
+simplejson==3.20.1
+orbax-checkpoint==0.11.20
+aulis==0.1.0
+devinterp==1.3.2
+timaeus-aether==0.0.0
+timaeus-cli==0.1.0
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/wandb-metadata.json b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/wandb-metadata.json
new file mode 100644
index 0000000000000000000000000000000000000000..7f4265ef28eeddd39d3a1de90e2158d4406c00c7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/wandb-metadata.json
@@ -0,0 +1,60 @@
+{
+  "os":  "Linux-5.4.0-216-generic-x86_64-with-glibc2.39",
+  "python":  "CPython 3.11.11",
+  "startedAt":  "2025-08-07T15:04:00.936848Z",
+  "args":  [
+    "train",
+    "corner",
+    "--wall-prob",
+    "0.0",
+    "--env-size",
+    "13",
+    "--log-gifs",
+    "--env-layout",
+    "tree",
+    "--wandb-project",
+    "jaxgmg-wall-prob-tree",
+    "--wandb-name",
+    "wp:0.0_alpha:1",
+    "--num-total-env-steps",
+    "200_000_000",
+    "--prob-shift",
+    "1",
+    "--keep-all-checkpoints",
+    "--num-cycles-per-checkpoint",
+    "64"
+  ],
+  "program":  "-m jaxgmg.__main__",
+  "git":  {
+    "remote":  "git@github.com:timaeus-research/timaeus.git",
+    "commit":  "7415666d6a2e5d63453ea9fadaf6cecf63256a59"
+  },
+  "email":  "david@timaeus.co",
+  "root":  "/workspace/timaeus/projects/rl",
+  "host":  "6e66e15ca303",
+  "executable":  "/workspace/timaeus/.venv/bin/python",
+  "cpu_count":  24,
+  "cpu_count_logical":  48,
+  "gpu":  "NVIDIA GeForce RTX 3090",
+  "gpu_count":  1,
+  "disk":  {
+    "/":  {
+      "total":  "102005473280",
+      "used":  "43119042560"
+    }
+  },
+  "memory":  {
+    "total":  "135066595328"
+  },
+  "gpu_nvidia":  [
+    {
+      "name":  "NVIDIA GeForce RTX 3090",
+      "memoryTotal":  "25769803776",
+      "cudaCores":  10496,
+      "architecture":  "Ampere",
+      "uuid":  "GPU-43bd7524-bf94-73a9-1460-2b36dd09a208"
+    }
+  ],
+  "cudaVersion":  "12.8",
+  "writerId":  "x1qaiddm18zjlareqkr9lea6iitxnpr7"
+}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/wandb-summary.json b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/wandb-summary.json
new file mode 100644
index 0000000000000000000000000000000000000000..db4c35eed0fa62a3ce11f32d876c9eed910fa98e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/files/wandb-summary.json
@@ -0,0 +1 @@
+{"ued/distances/levels_cheese_is_in_corner_avg":0.07373046875,"eval-batch-train/avg_benchmark_return":0.9927991628646851,"eval-batch-tree/avg_avg_return":0.039791859686374664,"train-all/avg_avg_episode_length":7.545291900634766,"step/env-step-generate-before":98500608,"ppo/max/max_actor_clipfrac":0.086181640625,"ppo/std/std_entropy":0.011185216717422009,"eval-batch-tree/rollouts_gif":{"_type":"video-file","sha256":"cc92313eab6f5c385d8fd15c99061c6f7d076e6bb0b5f42beaf80d33a9c0e330","size":68726,"path":"media/videos/eval-batch-tree/rollouts_gif_5088_cc92313eab6f5c385d8f.gif","width":225,"height":225},"ppo/max/max_actor_approxkl1":0.00728103332221508,"ppo/avg_actor_clipfrac":0.05042724683880806,"ppo/max/max_actor_loss":0.006009464152157307,"ppo/avg_loss":0.002090979367494583,"perf/eval-batch-tree-duration":0.4865581849589944,"eval-batch-train/proxy_corner/avg_avg_return":0.007898477837443352,"train-all/avg_avg_return":0.9484215974807739,"train-generate/proxy_corner/avg_reward_per_step":0.000946044921875,"perf/eval-batch-train-duration":0.5364610189571977,"train-replay/avg_reward_per_step":0.05230712890625,"train-replay/proxy_corner/avg_reward_per_step":0.003997802734375,"train-replay/proxy_corner/avg_avg_return":0.06658026576042175,"eval-batch-tree/avg_benchmark_return":0.9846352338790894,"ued/layout/prop_walls_avg":0,"ued/eta":0.5,"step/env-step-replay-before":100728832,"train-all/proxy_corner/avg_avg_return":0.009095139801502228,"ued/distances/solvable_num":4096,"ued/visit_patterns/avg_first_visit_time":1795.91748046875,"ued/distances/mouse-cheese_dist_solvable_avg":17.220458984375,"_timestamp":1.7545826156690476e+09,"perf/eval-batch-orig-duration":0.4923221170902252,"eval-batch-orig/proxy_corner/avg_avg_return":0.9750877618789673,"eval-batch-shift/avg_avg_return":0.9817285537719727,"eval-batch-train/proxy_corner/avg_reward_per_step":0.00159454345703125,"ppo/max/max_actor_approxkl3":0.005416679196059704,"ppo/std/std_actor_loss":0.0015033531235530972,"ppo/std/std_actor_clipfrac":0.0212935172021389,"eval-batch-orig/rollouts_gif":{"width":225,"height":225,"_type":"video-file","sha256":"b3c2b0820fb0978fe942cd07f11d5dddee11d83c212e03ab2d9822556f064521","size":257166,"path":"media/videos/eval-batch-orig/rollouts_gif_5088_b3c2b0820fb0978fe942.gif"},"perf/env_steps_per_second":298915.5436743963,"train-generate/shift_proportion":0.9921875,"eval-batch-tree/avg_reward_per_step":0.0188751220703125,"eval-batch-train/avg_avg_return":0.9817492961883545,"eval-batch-orig/avg_reward_per_step":0.14211273193359375,"ued/layout/num_walls_avg":0,"ued/distances/mouse-cheese_dist_finite_wavg":16.123737335205078,"train-replay/avg_avg_return":0.8570845127105713,"ued/visit_patterns/avg_last_visit_time":3064.32421875,"eval-batch-shift/avg_avg_episode_length":7.410384178161621,"eval-batch-orig/avg_avg_return":0.9750877618789673,"eval-batch-tree/proxy_corner/avg_avg_return":0.039791859686374664,"ppo/avg_grad_norm_pre_clip":2.4266347885131836,"train-generate/rollouts_gif":{"size":310347,"path":"media/videos/train-generate/rollouts_gif_4064_9f611d045168c7ee20d2.gif","width":225,"height":225,"_type":"video-file","sha256":"9f611d045168c7ee20d2763ab1e23a2fba3a3a8e63d00c739c9e9bfa7479ce93"},"eval-batch-shift/avg_benchmark_regret":0.011108886450529099,"eval-batch-orig/avg_avg_episode_length":9.463072776794434,"ppo/max/max_advantage":0.06011354923248291,"eval-batch-tree/proxy_corner/avg_reward_per_step":0.0188751220703125,"perf/ppo_updates_per_second":41.614397448624054,"ppo/std/std_loss":0.001500355894677341,"_wandb":{"runtime":3584},"ppo/std/std_actor_approxkl1":0.002050738548859954,"ppo/avg_actor_approxkl1":0.002936757169663906,"ppo/avg_critic_loss":7.412806007778272e-05,"ppo/max/max_critic_loss":8.262266055680811e-05,"ppo/std/std_critic_loss":4.170731244812487e-06,"eval-batch-train/avg_benchmark_regret":0.01104983314871788,"_runtime":3584.620653852,"step/env-step-used-after":199262208,"ppo/max/max_critic_clipfrac":0,"step/env-step-replay-after":100728832,"train-all/avg_reward_per_step":0.180633544921875,"ued/distances/solvable_avg":1,"ppo/avg_advantage":0.0014507453888654709,"train-replay/avg_avg_episode_length":17.91305160522461,"ppo/max/max_grad_norm_pre_clip":4.7987751960754395,"eval-batch-shift/proxy_corner/avg_reward_per_step":0.0009765625,"train-generate/proxy_corner/avg_avg_return":0.009095139801502228,"eval-batch-tree/avg_avg_episode_length":82.24053192138672,"ppo/avg_actor_loss":0.002152802422642708,"eval-batch-train/avg_reward_per_step":0.2047271728515625,"ued/distances/cheese-corne_dist_wavg":17.19596290588379,"eval-batch-train/avg_avg_episode_length":7.413423538208008,"eval-batch-orig/proxy_corner/avg_reward_per_step":0.14211273193359375,"ppo/avg_entropy":0.09888681024312973,"train-all/shift_proportion":0.9921875,"eval-batch-shift/proxy_corner/avg_avg_return":0.003932480700314045,"step/ppo-update-after":121620,"_step":6080,"step/env-step-generate-after":98533376,"train-generate/avg_reward_per_step":0.180633544921875,"ppo/max/max_loss":0.0059383041225373745,"train-generate/avg_avg_return":0.9484215974807739,"ued/distances/solvable_wavg":1,"ued/scoring/avg_scores":0.12822571396827698,"ued/visit_patterns/num_replay_batches":3074,"ued/distances/mouse-cheese_dist_solvable_wavg":16.123737335205078,"ppo/std/std_critic_clipfrac":0,"ppo/std/std_advantage":0.012078679166734219,"eval-batch-shift/avg_reward_per_step":0.1941375732421875,"ued/layout/levels16_img":{"path":"media/images/ued/layout/levels16_img_6048_f04bffb7b2d8fb3e6ad2.png","format":"png","width":55,"height":55,"_type":"image-file","sha256":"f04bffb7b2d8fb3e6ad222108c8dcd8a71d7ea4afe79e0708022e5b8ab7056e5","size":416},"ppo/max/max_entropy":0.11915318667888641,"perf/eval-batch-shift-duration":0.5005645146593451,"eval-batch-orig/avg_benchmark_return":0.9902398586273193,"eval-batch-train/rollouts_gif":{"sha256":"d89e30114bcc244cca071629f0fccfc25dc43d572f879bb0bf13d8acd51d5e10","size":309458,"path":"media/videos/eval-batch-train/rollouts_gif_5088_d89e30114bcc244cca07.gif","width":225,"height":225,"_type":"video-file"},"train-generate/avg_avg_episode_length":7.545291900634766,"eval-batch-tree/avg_benchmark_regret":0.9448432922363281,"ppo/std/std_grad_norm_pre_clip":0.8955571055412292,"ued/distances/mouse-cheese_dist_finite_avg":17.220458984375,"step/ppo-update-before":121600,"ued/layout/num_walls_wavg":0,"ppo/std/std_actor_approxkl3":0.0014679387677460909,"ppo/avg_actor_approxkl3":0.0023499287199229,"step/env-step-all-before":199229440,"eval-batch-shift/rollouts_gif":{"width":225,"height":225,"_type":"video-file","sha256":"6c6bfa49d6645a4ec744fa0e20c52fa65ce644e80919f1a68e45be9114b583f8","size":314868,"path":"media/videos/eval-batch-shift/rollouts_gif_5088_6c6bfa49d6645a4ec744.gif"},"train-all/rollouts_gif":{"size":169261,"path":"media/videos/train-all/rollouts_gif_5088_16ca120b4aa0033a7e82.gif","width":225,"height":225,"_type":"video-file","sha256":"16ca120b4aa0033a7e82af3200508d2e90bfd4ddaf9ecaf2542ed10f9ac59f4e"},"ued/layout/prop_walls_wavg":0,"train-all/proxy_corner/avg_reward_per_step":0.000946044921875,"train-replay/rollouts_gif":{"sha256":"16ca120b4aa0033a7e82af3200508d2e90bfd4ddaf9ecaf2542ed10f9ac59f4e","size":169261,"path":"media/videos/train-replay/rollouts_gif_5088_16ca120b4aa0033a7e82.gif","width":225,"height":225,"_type":"video-file"},"ued/distances/cheese-corne_dist_avg":9.995849609375,"train-replay/shift_proportion":0.921875,"eval-batch-shift/avg_benchmark_return":0.9928374290466309,"step/env-step-used-before":199229440,"ppo/avg_critic_clipfrac":0,"step/env-step-all-after":199262208,"eval-batch-orig/avg_benchmark_regret":0.01515207625925541}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/logs/debug-core.log b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/logs/debug-core.log
new file mode 100644
index 0000000000000000000000000000000000000000..c1f5bcf10b23fc3aad024d79bcb4ef2b551aa5e7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/logs/debug-core.log
@@ -0,0 +1,16 @@
+{"time":"2025-08-07T15:04:00.985058025Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmp4lmdszd5/port-3804420.txt","pid":3804420,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
+{"time":"2025-08-07T15:04:00.986018532Z","level":"INFO","msg":"server: will exit if parent process dies","ppid":3804420}
+{"time":"2025-08-07T15:04:00.985970442Z","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-3804420-3804733-4287269021/socket","Net":"unix"}}
+{"time":"2025-08-07T15:04:01.151723745Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
+{"time":"2025-08-07T15:04:01.162269721Z","level":"INFO","msg":"handleInformInit: received","streamId":"wp_0.0_alpha_1","id":"1(@)"}
+{"time":"2025-08-07T15:04:01.519105491Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"wp_0.0_alpha_1","id":"1(@)"}
+{"time":"2025-08-07T16:03:48.038277331Z","level":"INFO","msg":"handleInformFinish: finish message received","streamId":"wp_0.0_alpha_1","id":"1(@)"}
+{"time":"2025-08-07T16:03:48.047308371Z","level":"INFO","msg":"handleInformFinish: stream closed","streamId":"wp_0.0_alpha_1","id":"1(@)"}
+{"time":"2025-08-07T16:03:48.99423244Z","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"1(@)"}
+{"time":"2025-08-07T16:03:48.994290118Z","level":"INFO","msg":"handleInformTeardown: server shutdown complete","id":"1(@)"}
+{"time":"2025-08-07T16:03:48.994309389Z","level":"INFO","msg":"server is shutting down"}
+{"time":"2025-08-07T16:03:48.994433672Z","level":"INFO","msg":"server: listener closed","addr":{"Name":"/tmp/wandb-3804420-3804733-4287269021/socket","Net":"unix"}}
+{"time":"2025-08-07T16:03:48.994503624Z","level":"INFO","msg":"connection: closing","id":"1(@)"}
+{"time":"2025-08-07T16:03:48.994553775Z","level":"INFO","msg":"connection: closed successfully","id":"1(@)"}
+{"time":"2025-08-07T16:03:48.994566368Z","level":"INFO","msg":"connection: ManageConnectionData: connection closed","id":"1(@)"}
+{"time":"2025-08-07T16:03:48.994586182Z","level":"INFO","msg":"server is closed"}
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/logs/debug-internal.log b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/logs/debug-internal.log
new file mode 100644
index 0000000000000000000000000000000000000000..b0cfbfed58d16897ba30a05fdbe1fd997beb9a60
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/logs/debug-internal.log
@@ -0,0 +1,1167 @@
+{"time":"2025-08-07T15:04:01.162502789Z","level":"INFO","msg":"stream: starting","core version":"0.21.0"}
+{"time":"2025-08-07T15:04:01.519009005Z","level":"INFO","msg":"stream: created new stream","id":"wp_0.0_alpha_1"}
+{"time":"2025-08-07T15:04:01.519090627Z","level":"INFO","msg":"stream: started","id":"wp_0.0_alpha_1"}
+{"time":"2025-08-07T15:04:01.519103154Z","level":"INFO","msg":"handler: started","stream_id":"wp_0.0_alpha_1"}
+{"time":"2025-08-07T15:04:01.519113927Z","level":"INFO","msg":"sender: started","stream_id":"wp_0.0_alpha_1"}
+{"time":"2025-08-07T15:04:01.51914196Z","level":"INFO","msg":"writer: Do: started","stream_id":"wp_0.0_alpha_1"}
+{"time":"2025-08-07T16:03:46.862040498Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default, Name: checkpoints/3840/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3840/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6dce1d452c73edbc495b7bae1cc67045a4426dcd616770cf2ecc58c44e0d76c37e8dff1228b4d5c157390d15314203aef80258fc386820b20540a5a57d1d45abc1c9bf739ee5231ea1b3a1a2f67de4f7c2662a22d3bc8a406723d8be3a281f36602b9c32a8c526c0d4ce97db83a1adb9b7f40dc1ba584ab045f62057979f9fa6ebbe9e629d8daf99753a22cd4b02071b3d5c85edad033aad227a6bd79a7a83f828072e8cd578b3ce65f9c5560dc26948affe5ea3b595182d0bbe222e99d0d90348678e05595fc092c70ad24b85ac55982171c982f42adb4c1168227693e3ce2e42e17005ae6cd549d160e1dc1effed778fe23602e7d795003330301e1342f0c4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862108201Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0, Name: checkpoints/1920/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1920/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=83c57940cf2ef4eff944c6ac784cf4fd41d7bf273fabc8b87306f55b9d81192842627a9969a1173b58f9ee5821367b6e4a490c54690a5acf0d3c86ae9137164148f9c06103cd9032afb6255ed193a7016e8e94516ecccf4b0f4a5863561b33efae62f67b8b27513a3aea43080723d24adc11fc25f125722f56ed9f8af1a37b0068ea4efe9c9c7f272b93fe64ff24767f07f80563372d1d2009d943b5ca5761b755ede498a91d4de87d5e56ceefaca4626a13e4f9066674167063d438d213a3b90ac0574129c6182cf77ffae7abbe4203a565096ef9c90cccd3e386a96117a571279b3c622ef7e81f51f75ca018b29248c4f1e077a41f7ab59e1130d6b650e1a4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862163932Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/d, Name: checkpoints/448/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/448/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=27e737dbaecc9a5a2ad3efd4168f975e94a7accca82680ef85ea2e9cd657f6bde0d8b57eb314e52b238cba287e1773b893e5ded0af03fa184db235d16a60883bc59a3fc86d0c9dc6ad898c5fee6a7465c8065b42fdff1462f8c43accc154c117181ac28b8c62f9483b56f25c89965099499712ced5440bb0c49c72483503e49544fd1894fe5d67aef6ac75bd5f4dd441d3ddfaf51c6a2e1f71da389d2df1e3f4aa4665c1ca9333153958f89830c519c9971391b4f74402b0568d57c481d351ed24ab01fef2e242847301d9972423d2847d6c10a10d6cf219cb9c020c456edeec8137710e64192fa273997d5292f9a2527a5f6c1ddc3bb053467b28bd5096f743&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862202143Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184, Name: checkpoints/5184, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5184?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2fee0f83801e9a50f5bc28bf4947a766cba9f386356e7ebbb3893a0a5f9f5bf1c923ac8a673bf46bd8fd54d41d9fcde48b37b59ca27dfdd2cf81f1a178d5b7987a3f1c55e108437c8a35bbffd4d6e11a02d4b03e1bab5137b0a3155d9e2d7c99c289f611d465951973a38bf9a27914cdc90aaaa25859bb31127d6b523a37f69f853666471d74fb12d9c21564d41f6fab39c24d0caddd265e66a2ef28268479f8ef83907bddcf451e3e15234627bd37de645d5e3fd68055b4a73fd37abf8f9839e2888261cd6695136240f76c79a2a93dec79d09545fbf381003fd6fd975b9561f3d593bc11885eb9da815cadbc261cbf85a4e97f352423b78abfd0f551ef0680&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862211206Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032, Name: checkpoints/4032, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4032?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=47b600b0a1f7e38a5afc7d348e3ea5b7e853ba136a8f9827a75726ba682e6c1fcbcac47118aee06996c03e2a3a9aa750e60235490a66285a7f7e28a4092b5c72a483b9dff4f034d835b97fa640f408bb3c1f0881be7bd7a506254704a27ee8e43b26493f15f1eb7c4cd08d682adcc0c0a2fd1b11f9c0b8c3f7ba518809d55a866b98669a3d53b3a971bb942aa9541bdde32261004d925a25981c748992a05c0a5687c8947c120bd06e1505e6ab3465fe60711b3ef14f7a8c88d608249999eb2116d653e80044171f79038c3062d51ebc6b34c835af8975b9bb469bf74e75549c0ce811a23ac1ec6d5fd35e1fb14af868dbaa320e504841e98a8c4145073badf6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862231466Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints, Name: checkpoints, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=742c283a209d2339fbd0706083f809d7768f4c802ef5f933c9d94e8bdcd4a068e73e9b39fe927d22df5fc2dfbf83b218d97e52058d7026626df3b819aacf850ba3362de0ff5c7e2ca95d4ede9ec662c9045c1167648cfbcbab2b8545220e7cefead530af11765772f8374f831a78da371f5daa2ccef4d56992cf1ff30b8b9b91940176a2b093b666d923e748fbe728101c6c96c3cb72bed7cb509880deb9ee3165292b1ea2701e66803245616a7e2922e2353daf5c155b4235c93bcb102a2529513b7a0dc100540cf03ae9039909f6ac5d299ed0a78b8e27166a1f726cf6570e1d19b16e9e6b18714ccc710d5ddbed8ba59c5bfa852873d76c6ef77623f40340&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862262737Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/d, Name: checkpoints/4864/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4864/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b5439d81b061fd42e31b7e37968cffb20b9bdf7f907318e0a562186772c2a342a7ad0239059dc3423790d3006f6e6ad59e78168602282f46c7afe10d79dd55c129ea84b0b6cbc5fb3fd4b3fe0e47760b4a2ca618f34001f6fd3109f5f563eb8877b79bf5150c43d9acdebdc8865ddfa204d1bbd810bd74e4dece2f7b3eea95adf9be1b705af40bcf9cc2dc883a2d384a7775322905033fa07f0801aeb2978377e08564ab4f49e70f47ad8fd407b137bbc52abbda61d3d5fe802d424bcc46d860bb411f1c5a5c57cb0b19d7c753655346088bdaae27abab246f8b8bb2139777cf6fcea54f3cd771f635b5782b2b316e3de4a1b81e30aa94fc4b6ca370952bad2e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.86230701Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/array_metadatas, Name: checkpoints/3136/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3136/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=7175628c98e3356465130e5a36c509d2aeb808cc6f8d19440fda3738a60c4f80eb3e40f8ccf0e454cc4a9740e0b6aec5a9746230ef3a004d428b63abe9e0a9af0f7ac443f1f4ec0b04038a8071fdbe9c61d9d3fdf4e38db8fd368598c381fec8d4f0e097dafb68afb5e48ba92f552b9198c50b60c758ef4e33331f53e9faefdc99f9f405371077ea6bbcdd79e0695c7b335ffbce2e0aef21e6990d5c0eb447384f182f5c7339669ddf6861cf27ac241e920d8068025a2533d53aeeb3fdd9fb4a759eed71a5f08a2a77cd7e5c4f3058c02245aaffb14902a98f882036034ae4f3e42ad2e5fced2c21fc12ed25a501e125fb6af947f7884cf1a0e025dafb61119a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862323906Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800, Name: checkpoints/4800, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4800?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=7af759eb3dc7cfc447b6c22aa0dd4a17c3ce467a093f9d9dc8089d154f6323dd6020504785614f09e43d88bccbc4de0e9dbdf160d3e51c9848bcb6768e0e2aaae63bec901eda13154569fe094fc57c40012cbcdf68cc3494f8446436c9a1c8303c2463f8b1aff83dcbf63e7d4c3efdb1be287f662a7d15cb089e5636eb15c86e15728d86fcfd7ef1e861d8ecc5829bf6cc5c7f9d11fc9bbefcca812875d695166f873e5379612612de1d8f8ac7119181d1f5d7fb568cf2c996977404f3d94849a205e25de92b5c174945afdbe4fea0151bbbf08eef276dd2d6f45de986078dd8ca1ad34ea38ab0cde42c8d4dbf83ba63759316364e40c36191bbf99c54477540&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862398231Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/d, Name: checkpoints/3136/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3136/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=75319e5fea66b156264113c47a920f6e07b7f831e230affe2d61088272dc6ee0b896cad70d21a319cc92129178f4319c2c85b394096ff9bc25a91f680846064dee97fba3d00732af14d7df7520bc677e8d5efb623ff19d2c0fd43506d08e6294963fdf8802fb57cfab0d925bab0ed0a5c3ad4e31bc127f4bed3414b0f4d4c018672dc1283ef55269a53d5476332f2170ca495e04bdab641630fb1fa14d885fbd82746af96f42d58a2bd37067e610c9859986f3b457bf9f7dcc512cdeb138e4844685c5011a2fbd580477b84d7070ca4a2683523d1d6c7ae2e107a8989b42ad2be4ba954fb07bf23aee789709d907a313b724d913bd4069e6fecae048f0a00e48&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862407099Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d, Name: checkpoints/3264/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3264/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=142548b428791d4937195257d1a5eab31cd00c29dd87c9c624927f5a7d034831988b9754c1cad73644703e1ae0d72485b456eeba69acadda829670e0ab796a03dc5bbfd5f7b0fa05f5563cfb6d8d3e7bc0e1ab297215e7bd26805e605b01c63057b34b1ea9bcfaee5a2ae892d0cc61b1242f4f613293da2872d71f1142df3f75ea64fd9a395d1399d0356758002d6cf7d1987018c7f6185f70ec085628d7a0cb34166da1c5ab25d4774ef3dbbb6ef997771d70a5ed8383a26b5628dfe6beeb9b2a6d520b0460eec0e0519cc5606e9731c8b5df5f8abdcd0c8e76b8d83d303e15770c4d37648f90d0bde1a106f9f90b2449a87fdbf4ad0b49c687054b3591f9ca&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862478683Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0, Name: checkpoints/2112/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2112/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9b8210bde0c94ef01f0db90e2f4ea84af30b86fcd44fb4ef9e5d4b950e4f07253dce456d47debbb5b2cbe055ee65d5d6ba36b155a2bc1b41ee9d04fe3cbd2b0a14a7a3ba3a2279210f5b8f5c686d720aa9804a3c7871e25a4cbacb99c27407caa1d9e9a3ec349bb855031512ade6351363258e56eefc4db9d4ee312b5e747e54e1a642fe825ed89ae5e2b69c332b365ac6e1a345a58a2aeb6f5e589458b98e0c5bcb83c6ec2e422ec5ac8a186c7931ce6e8990920b412aa42437a242707307360c4eb742febd668a1b39215aae9bba2b942e063beea7b8bfeafc6f6cafae717090266c0871e369afe072433976f0ed4b102ad2e5ccc3a0d6577fd4ff2226334d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862517124Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696, Name: checkpoints/5696, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5696?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=13583f6c794168352247af3f6c29dfd48b70ba8b43afde39eaa3b0faf07ad3a88aefcf6d5b1c91a5f502752a8f3541c97a54c189e85170045cbef2c0b8035a42097609069d00505cf42d40543021c828a857b835f887cc6b7d8cd9bd306648ac634914f4e3edd29f09ad0629345b563ea3da1bbbdda19947952b90b8bfe1499960214eed5f971631adae529f247e92fdf791401d179929e9e7deecb76f4627b7733365bc9030bd315908a5114e897c37d6b19599f0bda12f9f078bc6d481ffdc3623b00cf4f76e401f265eb40fad22c8ab3b07719366e315a4ee5d9b2bc5619ba6376f7a053c5c38fa07857e65a144870915a2dd11ab4a55a34f3a80510364ad&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.86258431Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d, Name: checkpoints/4032/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4032/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a0aabd0718fb30136f14199de7b7b62844ef7d6906c29fc114011a5188528bc4329e03d621496918ab317ad3d13a7a31e1ff6c81e4166154f7cf6b0dc2ed285cb7b6ee9e9c8ed2292502196fa299506ebcf50c47350d4a12b40aef18ed2f131680e26381cd059458e1d02c926cb20d5b58e1f0c9c625db596c1b2a247fe58d7f520673dec73a4f40987036520b3c56cb8a8220250124e794f7cc961d85cf45c29fa2a413a96d203e193f59b2d73a0ce6b781132a63b79c5ff2e32feb7de73a97bbf6ad648a4c9abf629b748a5e90cc91b08eb507204aab3d9f54ba268d144f308f08101184b2316ad59b5cf2ee4a758aa5fba0e7665440f6421072c5652908c0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862596811Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/d, Name: checkpoints/6080/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/6080/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=7f7d89a9988c545c690eba5a4cc352088ca1e726bf1b9147534f8f34bec249bab392a6465fb629771680d0e563a32d1e3c5a0a1ed2914daa933b4f79ab492a9fb9549d1598183396699bdf5a39652c1428ae986d4917c0c4cf1f5708ce57dbe9c2a5084401e81a9f4f450fb3a37d3288fbbdabcd52fa86606a91b4b12e0d9aea5a9bc34c82620400c2bcd47f676703fb2c5cc26899a581de264d9772440d95d8dfb887029e646ec79030dd25a715d5f972259cb3b19b033b2c46cf976a23b6a76ac0e4396c1e7c8bbc5a31bc82ee9bf3d9b894f57bf4911c28b86040fbd32f496edfbac9f50d5a53b5fc71fffe5d27afa6254d4af7b760c4b086a582e7850c62&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862625077Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/d, Name: checkpoints/2240/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2240/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2885f9baaa031d8d8a4fdab76dd5315a3754509d6287164405342295bf18f7ba022615f87e0452c3c48f4989ad5246943dc80caabdc4e913276f9a11b39c9ab9f932bdad37aed6f4d4bdb6352d6887208dbe4dc7e5854729d85f1390c06c5d02c5f1fbcb4574afb1cb7c08c32796215ab5c77654922aa3d3a8d69406bdc06584cd4a461ed1ce4b52dec0d09eff1bc71fdc56bd96678fe19dc42320b5177f5baa048fda9220596031e70343576f7b4b9e7a837ccdabd16ff7f5984618468230c117fd3f4e1f22eddef496bfae7cb84398e3df3b1688286b56f3ea0948173b045d1c5d1118a2d7020406a81063437376dfbc1c229718227e5b1121064dad406d38&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862684472Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d, Name: checkpoints/192/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/192/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=4640de238c4eb643909e8c32bd20f19d913ce91a819fe29187fd7b369c721cfdf09717a00c751da10cb659a26c6dedd43c537bb9cf4732f39d4efca78d6ae103fff27ae20fbf790dd5db03fe2831be5093ad9803bf1c93c37fecb4de191efb3da3f4fb2a79a988c22b19beb2e68b0d6931299882dd4487965327cfcca7b01d4cd69c9d47b43a825e4f89fa91402f20b68b344e5fcd15cddc4d2412152383ebb54a56b7d5fc8edd607838f0e9bf223ea77c93295c6920417b81417a735ebaf4ffba017f922ef351454ad36f799f24411891a29513d516d65b754ac1c62c51a6b78e38da997a4b66ef4561e1a2c88d68d672205e8b856dd44d5027102e7579178a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862716365Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984, Name: checkpoints/1984, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1984?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1e73429aaee4c228624e2f6185ce9b2ebeffe170effa018a74773bff7556f2642df790dce945669ec3be2ab0c11d00f799ffa8012bb78929dcaa25b353e679bf184ff5cdfe32b51029f4a308c9b02b35e61aa9a860cf806216c912493120f09ad125d5218ae55b597c8f7d147129f86fa8d67904dacec9ebf9f2714e6583eebc4bfc8f66c4fa26652e461085e6837102e0de33cfb8e7474d91548481ec1a7647946ee53e9fb2a02c7b259af3a1c33be321833aaf0745fce6a06846f61a82fac6f83028b8d8b85cb2c5e5bb37d837e137c376a3c089bf05eb9d9f4b6fbbd522b69c32a835b9b99526492af2621d8b1a2a5c329869bae0de0fd52599112d330b45&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862878406Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304, Name: checkpoints/2304, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2304?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=5560346899b0887195cb8210333e7abc593af3085419c2ce20aa4d12264c640bd61d117d51f24c9d405da45e1a3494ec2e2bc1112f9589c3548d285babb263f558a353e0c418c4e40d31e829c74dc4bc54021c4dbacb431caaebd67d29b19c9e6a0c5212d942fcc041e16cff826a10b7eb44dd3e9a4d1b217a99e4899227fdeba9b85f1ec684f666e6f92335424f2fb062d97db0aa49a257f243d3a087adb69f002cfeebda16e4322012e5d6bf19f68aa862e93022ee8681a19de39dd7c2c3f1999b21ba12fabc71b0aa4cc0837e2e62f0e3fc9346015775ac2b69707a383a6d0c6d35689feef359bbe4fe7a9b874e7542b59d8c7a7a36e9c83f336678ef9f11&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862882609Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default, Name: checkpoints/1408/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1408/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=70c820c9a59214fa3a9ca1d29d2b4bc8c0109491d891a0cb20ed6a933fb6a791a5d2794451757eed4e6a81bbdebc7e323fe43eac3da5712d2e0df40f3ce219122d113487d5151ae5451eb2c2b5854ea8dc7e0a58dd3fde4e255296d47423208da31d85b1437c0196a4ef7c190934051993db131a3845fe24ee3c20117b40c16bdead128a279bdd820f590314acb5a59496669342c52cf1a114113a9386365e0c0e28b4f0f79b9cc00c6e6206ea92e1f22697a54c3aaf617721c62994c4046bff61718774e948f87e761f4597419972982faf622c40b9ad59d5026746d0b79b92926f4871d4ce4febcdc13579ada83f6bd6eb87f2e8c0ca95c27826791152b1b5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.863051509Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default, Name: checkpoints/5760/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5760/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=3686f4c4ef10657d24991d9e8e2c621da644d0556b85eeb257dbd4249ed199b9a5a58833d83c064f4b6c08b70db22863348dd63346e3b7132284ff5cb5216836794ee49c74a801d947de9416b718cd5c024d79105b63b7638955198dd0a044cf4614690673d4c4129cc4dda9ee26f654249976db5213ca935487f14f8167eed6e93d37d5bf4c74512d9b5d7717bc8e57269412ec0bedbf8dee8d82d998f0ad7a96f300ef0c327ce4c4ef57f86f951b7d86d7417b5a1ea09c0df38974e51e688513b614f0bc87aa92df81a0d7530cffaac13042254d4c0dbb0aac5d27679e05223c0aae9d6f4231d06ddd91d1b3b5a2d3f179ee7dcb2a92af0db1e8a106d72c13&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.863094195Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/d, Name: checkpoints/2816/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2816/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=09872fbf2d50112a23f8f1c23845195272ddb2c547f61f9138d613daac1114de7e4caf566f5ed203a1d95d8eca1d1b2939380fc709534e08056d1bf3cfb6b5efff0705ea46fbf941fa95167949dc74243eaf40b2b9db528c1a8269a94406214e7dfd48906ac69d98dbd29c99b303cbdc1c96bbf95745f698bd91d5a98e80b3be85bc42d3ff5611c865a245a4373b313b12024f0ccd1498f6bebaa767c85dbab9f4a0c9983343911ecccc34860377229ac7cdda4e43046cd871e35eae93dbc997d1b86bcebc9680db07d20cfeac964a772560c8d79c4b636462a4df4d5884b35da806972867d9bdbeabe14ef967008e09153e4a913122d425d88bd4210ec6bfda&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.863311857Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d, Name: checkpoints/640/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/640/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=51812f760a7cae60587b52a34c94df25b81d987a4e0c238e0f8f63351f008c442f7f31aeb04ef5e45250af80ae9c061306301566656615a2056fa0d74ba36e8c47251656fef335fbe6704de08250b4ef98a21c1e72bf003851a7fd111b0cfdfceb0bc27f55becbbc6dd79447f1c47e51d299a1394591bb69cf4675a6a857c72527734e2e65bc09309b7f05cb7be21dba4a80cf5b96414a8511a21d4233321425e183354de95ca4af9c35a8cb41739bd0ea61897beebfd45cdb8f5b11bfe4fa89202493da087171e9bd530ed3acbf96af9e5990cc99f82a66ac995f522fc934e5c65e454157ad4ae6d2c840231670c4efeeb57d8f4994f380271ed868863aad8d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.863362602Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/d, Name: checkpoints/128/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/128/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=68b9e7ea199720fc1fe199ca2f8fa56c380a4e7ce95bd5431186f31121adc256a98aa35d879216e0d3ab01100b7d2e884ebaa5f2b988eda22bb7c19ed892a4edeb6bfecc4df1cc52051d584142f99d61071e37f8028efca77fe4343778eca06556aab13f9a33d6ce62f93f967f55a269989c13cc44a44f252b99d36d7c30d0f8657bf03abfd87c789c6ba9b48797de2f88a6608560ec1aaf8b110cec225c4ccc89aa3caf3e68db25729b5df594d395d67c93365e8eb8f823977ea9bdaf742060bada75ae140b728216f418f1967c16a04e439d41211a3406b307b620222f4fd3bf022874844e60c773f3ffce5171d1e6e679fd3a88f058a83a099fd7660f7eaf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.863433382Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0, Name: checkpoints/2176/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2176/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=3e2ac055027055f69c80b19ac1c6ea81a5a25adf53dd4ec998791c7a4deb7bc2166885af9ec57a01c2d5e2d643b81c474d076ea993698071ccd29ad70ddbdcadad1a4065f25d7a93878d5d9bcb56e09d7e80df43119db1bae7c36ee0d7e1dfb0a6246e1c49f6b64a51969a25cd01e559b642df314684f055b21583e5e790795440cd5683a5401cfb2341649fabee78eece478e3e6356f9a57fa84ae74d1957d11e2fe00ad31887720436c70247034470c3a9f26538d97ee1b9c92812d281ae85123ca0c2a97e515a5451dd8e94628b4b484c287d60048bf0256e4e79d4e2513167a5501e6fc1f1f570d56298fbb35b2a2b1b4a0201eb8dbb45f1a9d7ebe28c66&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.86350335Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default, Name: checkpoints/1600/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1600/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b8f9491ffa03944bfc534a686d60f8968f2645f2e14cc5fbda88e0b23fdab1ab244efd1cbfab34372520b7470af29a00736cec7d625ec158723fa0ee116abf8fbe07003c1552d3b2259afc3b55debd9df042842426ca0eebfd9d3c53f3b8434bb52064d90056f61836f674afa0746e32c7b47256bde292decb9e0eb8976036a733dd11bb36c00ff8f6f6cc893e3369a395d90a7532ece2bcd96250dd1ef1620097efb715a5839938e2c3a17afa2410e0e22e5090836438c856ecf6dbd51fd01bf3a26859574403c242bdf39f6f8cb9c85e9dbba094d64294544edbd6d64c05cb4a59bde378a1a952a5e12a42978862d19a749baf2e77e6dbc46cc58919357f85&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.863519521Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/d, Name: checkpoints/4736/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4736/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=7c0c05ca60a6f5186fb028cd57b4e155db10a5659903954d473c21c4b8fffcf3b8693bf4d7e81752bfb7acad2866a72241fae4e70bf2a553bf6c9a113c531131e81f30f736946f102c169af4a6b7317c44212151902a580ccd3241ea2d8521fc887dd313f8a39bb2eecd7de0a819c54f2f2e111468569438c5c08906c04830373ac14a59e8a83a4d7b064980b382c84d9499970d28ff8c354a7779900241989cec5a7b25aedd738e026b2e32214f6b73694b8b142c5fa36b1ffe6ea21cdd5c78bc28404753c00de205f9598d5fa519486584f4b0a5ea7c7f61f52321230da83f3744e498c517b4ca48a1441199c9f79de19ab9b4fdaaacb21c618b53cdf94686&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.863550546Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/d, Name: checkpoints/2176/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2176/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=55b8d961df11507fcae5229f25a7c5ef9a7e409f9f46b9cef9940eeade55b04e381cea2c6ef19cce335158f6eab003ab5a8fe2956cebd05a74945a5ecbe912cbbeb25a78afd0342098323d4b0271e12edcce980ba7862dac1e22d4b0255ec24b2aa2c5fe29f91f48071aa67b7e317d7cfb4adf89c0952052fce436efe42d0211fc42418e91f96b63bdc5aae644cf6760f91c85cadefcc4a3b670aa2f7e10f6dfc9593c16180d701c9a9efbc4b1d24a3834a737fbd4fb4ba3e4c4e5118d5ca484892469e35d54841e750b9e8dec7f1e745ed6b32da907504f98f5f7c3df696507e36da7a86e613f0f594db6954a06c0a39f6cd74a5addf96a89164046bf30bec1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.863677778Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0, Name: checkpoints/3392/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3392/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=66c6e7f304c7a72d8ce0ce16c8665c2ee8fb9edebcbe99f9f925a8c6f1596987dd7658814f5ac1e9e0f1fa11fe8712c75169619489edc509ea1a9688a75f023f377b4ae6be3b1307e3378e9bb167e5a25792d4580e0976d1f556a7d89add3e75d49e38e7832ca33d5a1d14758e6f346b088557df9126a2746c968b453826146e20bf71303d76cb60f742184b59d827e99273d4f0b680d3801209d808171d38a698d7ee73369c856a7c33978b1f95b3bfb47530a7940a7ee1490492187bcf0f05e634fa2f02346e7f1574af080fac6eaf70f021adbff7426ceba10f67f6efb5ab52eca10ea262fe148b7a254d0aa0216bab9615c5dac0fad91f7e18ddae35ed9c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.863726184Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0, Name: checkpoints/6016/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/6016/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a6cc0d32ebbf31f0c060036f9f5de23d83edd202e3f0c2e3f0ef28113dfa8ac3e522c5de0ca5bed0ccff2d87fa84bae6733e882279c0c089ee5a399ffb796fac1f97b2570ea66017d6794acf2250de02fbef078bb765f401cbce4c4098e7c95430326b9f8ca64fb8d74105c8d999f1da3d4aa4bba5a362c7519a5768fd5eaab53cd931d1a34ab23221b6575c6fdd8c5a272f728d674ef214b9788a386e153d20362ac345031ae156b41c38e1b99c8c4a3e52466304f54247955c120ede7a36d0a018c0043a8646e2de74f73c366f258ff33431badb573285235d94a6ac3cde7f93b7b5906fc909333619b3ba4563330fb3dffee649c74060a2a9e47e46ed8294&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.863668435Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/array_metadatas, Name: checkpoints/512/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/512/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6e166fd5a0d58fe44e78ccfd275b68176b32cad671ccb74c7f02988a66458f758e9451b0ec3bc715bb2793522f6f03162b06a4acbc4d68862c0bea6f59b50982020e065ddb8fa6aa810e54b866564ebbb71f8dd7279e6c78d656d5e0c5fac15cb09f496bd38a08cacc214e738ef04ef0718cdb0c3ca871ef061bd808ed2f3820d3f7a060ab1554ab0a5ae349f9a66a2d2147e19de29274285ae0fa6a32cba52464ef401dfa91ea4d6ef7231cb1010bc5f9d44c78263733e7c3bda7acb389d8937921db8d10841d1917f76006918c9e40c1a14d788f5224f01f8b9819053ee3be041cecfbb438b7ae81c64744ec1c6003dabbffbc762ad1a0010d8edfcac5cec1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.86384461Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0, Name: checkpoints/384/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/384/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=aa8b93c3d424686e650fd7b82d6fbad9bcf516d351fc3ac90dcadd5036f2e55d1c133972e7be63bab79448152f861ad474913a146a2c98676e6211b8fa78444e8087bc2142b60667a1652573725d2446d88961b34622355a4cb518fcc6262a3b6d04a7c8fd54902d391bf3af75b6f38bdf8c4e05a759ce64bf8d55e7941b6ed57903ea373344cf36ee0d486ff00d49401ec6d6d75275c724b77ff5d6a22151878642613a54ad7ee7bfe1873d4de3107364de35565a0702292dfd3c8a823eb9f387b4507186917f322eabd492475a5dda571c2a3d327bdff4c0662c5ef03749feb7546c111f357174348fd94c33da52a0a16124dcc925b2446a8a51cc50e9929b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862614248Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d, Name: checkpoints/0/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/0/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=60f4fcf34b6f920fa3dfa9086054a31a33b1445cd26f581d6c640a60d2f2c3a4ef6cc1df7bc6ad0e304a611a3630af2c090eec394b3e784fd6cdbd90afcd7eca16ee4859a804b19df6fd1cabcc78f6251ca34d68f051c9079a6c2a6ca3f5d29156c2fa08e48360e76c040ce6a46a1d76ec163b13f67f58a0dfaf27a18eb7fc94850e0e502e19646d409f4edf412f74e4cade18198edb68d52ece9d0b87582bb19afe0bb11a6464408848a6d73d2e4387cde0d8641d8efb93070efbbd00df172feea5543aac5c1e646655452835ef759ea78b88f7123b2c6152e52873af1e875488ae2f53595cbf865355ff823d2fb9d95d01051e908ede79f6fecafc6209bfd5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.865029476Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0, Name: checkpoints/2944/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2944/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=20bc5871a6f350d1052dc0ddeceebd18516bc26f3edde7117141d7aa20798cce564a007132be86d1411fcdbd53b5834948ff1a7ce77b12f836fe94e4c59957467a44dc05560ea3272f4e2e0792ade1f732f627216cdf4af5f56cb645d8d6db46f0d844c8ecd0277918a02890da4ac45de9db174ccd07fd32f466e2182214f5f5205ed0fb4d6f1afd5a7b6002d4ede9efb87e45b860b0b3cfa68548cb14b8480648cb3bdef7b73e513623a57d9b4706d5482901c6905ef9ad9326d9d09d5e88ca677a9b5507a2f24c2eb52257ca2602b97094d16bdc49a0ee00eefd3299c888a8c42a710b77ea03dee7414b5e9965a334f9e33c865832b2cdf939120f41e6a93f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.86525643Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0, Name: checkpoints/5888/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5888/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=7c65213f26bedab91a9ecca1191431fda89663ce1473d47d93a505879ea71ddca8debe46c83cabdf2db21e289c3f15f04d4f42eed268a36c84b319fb683c1d9c18f45e6d3f107a32464ddf6a46e1971d746c59b317684343e06039090a0666cf83db37b91ef9ace774d9b5ec4898d3e7acd14a95b5631f7f62190248028ff9147346c335ad56369d64b5c6444d00101a6c78f987c6cfc32926bdd5165f0f020fe184d2589dbd81ea154c8890a7c085414b50793c5985ba4c51125a5a37d62ab79039e35df81b7d33f68476cecd8df74ac084c4d03607f537d5ec268497dc7725d4e0eb384cea333ef356f24cc365dd2a48a830ba7c5bf24297fb48784b02c004&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.865365095Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default, Name: checkpoints/1536/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1536/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=c51cc15d7fc873e17471a6bd76bcc20edbe65de7dde4f0a1628263382ae3b45f759909a2bf62d19a9eee951c69e4c56e51bf16d74466f9b8b4733bda033b35315b5732b8052b062d21d67938b417fa9498c21ff2078e0e348d960d307979a9e5f114848534d8eaba6db1dd7022a8503ceb4b849198aa203a604c542de66218a02c1c29e9ec0320ae997d0394ca50476e337995ecd35cd1f5ca19514a5ba87d6f6b4345d3b52c7f5629c41cf05265a2a96684bc55c9570e58dd98b46d9d26eac385b50c6705877861906a93c3f830a9f86d806ce2fe1f3ad09c834f0ea84d49cb9ded5ac0cda49f780f31083c2c953f27aeb09af4da1a93e04c5a0b5aa9f5bfca&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.865500164Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376, Name: checkpoints/5376, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5376?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=3e3dfa606f47d480a7cc827d4099fe46c4743ade83bfd03ec81c5edaf2e9432fc93546e480042e3e5ae4b38b36f602299f99b1af5b8adf2b926286b7366c48a30e91a55034e31eae8ec383ea278136782abd43a3d6f473a836c539922bcc78f79d2cd368a269137e2d8d7fdb1d52fb916eb054a849e52c2a969ae306b0bfafc319c5191adba402bf7deb91385980caea066ffae82ee3c7175ccd61096829afdf44d82d29557ae6cf5e61d85bba41d8cb67fc6a3072f00033f28d9beceb541b1cf45f96cd39a1b54c64f05ca93caf2fdc48c60027fc17e50571c3619553823aa1cf505f92fc9adcd9069c70da9017f197cd859534153629dc75a257e2ca5ef901&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.863393941Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d, Name: checkpoints/4096/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4096/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=93f29b4fbfcade7cdad571785d4c7bb48f31176c751570ea2032352149d7f152643033a9d75bd158f63143081a9268e0b396e0a8202eb998475efa72b0fae4f97c355b685d188a1dca9e8d25937315b188b4fb2fdc6bd255853e960b1fc1fb1d1bb90258f1ba0f4e19788a1bc73198e8fa6de5fc75132d84e87c4b31ab02f2e05c4ce09b217c71f9cc516f23307b76618be9affbb2c901bb38efd0126684e5ff744bb9a4fe6bf3d9785ddb7a25f906b837f1536fb6f2e654eaa9cbf3e4818da66c1501639a7410e5b2e4904169d80ed0cde93521b8c3a06b104ddf60fa88e5e71f3834ca75275cef02fdc3b53d6fbc8716d5e2746cc16facd8b43e085d7691a6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.863845039Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0, Name: checkpoints/1984/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1984/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=8e1eb230649d5d61859f1afceb256ac00e9e25e9caa98a5d9d3af419f9856bf4eaf3c1260e14e721aad0560bc69929fd29c1e0935bf5d617e620feb319c1f86ef0e760f9de60539693b52ea0261677fabb2c5317b69e2c4c36fa4c7d140cfe7007543940e0b26e744865a9bc1bf4d7f49bc83afb36000c4c20993d2fec82c6cebe320333b72af76f09ebcc6f9b18eccf260930ef54d4965d65d0c5790b934f9e9c5fa9d161e7d47e34e65bf5bdee98d69973df3b4921422fb213b3542e3ede24044ef1815bda5d881121166a0288b43bfac8e3206488981ffa4f187f307a387e251ac1790b064ac6414904d2f5bbecfcf6c925d5f5d3a2e1e2c48d9ac7611c59&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862119863Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d, Name: checkpoints/4800/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4800/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b94278e62fa0f7dba3f861ad36ba2a449e55d89e47c3c4b018502ef5a84129c8e149ff2982b6c37a61c49877a0697b22a601dfc1a0575a3c3b8fa8bf69425ce2ce5ecc41df4b8294a6899dcf4de843cf905ab4806807d0087353fd4cbb9ecf73ce15f8a02bc5d67724a097cfa9e17c3b8753e3cc4b7e0b1a4ac8648585f21cfb40bba6a897226c2dee14e32d37b0ce2808f14445e883b3719d6dff2863a1dba584cf4087e6dc3d2f345ec92b18fa80b1eaf82f8f8b622b3b0bb8edfcc428733dd6d53647edd32ba9f757a4faaffeb1e09a5867d1f75e2cb32cad4bfb541d10f5cbf289d124ec62200a932496f6c872373812e7c7f125f7a7c95915922ee871a2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862234053Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/array_metadatas, Name: checkpoints/5120/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5120/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=66b137fce7ee7584cc70c2246983d953e30eaafc1c8b957d7556999a88fd3f1f5063e213a3a78b69583105231c32f2ddf6df2fb3d73b92c0be8291c5d5b9b5acdb06c33e26a7fc81c4c58a4f8cbef3e2a6650cb3ecffbb4aaedf18b5206ea3788a80b45f2202cdf6e160e5e135c15d53a47512c5b102bc4a743f23837af641d913771b9aae12dfc52a72dd91bede4207a9a9cea5fcae3c331d1aaf315793c0b66b97a179830a0d1bd8f2d3c89547d37fd771efccf36366f704dad060371ced129775bc3c9ffc920a3607a5d86a929a23e1627853c06037f50ba1fc795955ef82c2ac917a4a26f48c68226da2c62822175494d051c71affbd91d78a08c645da49&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862245321Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/array_metadatas, Name: checkpoints/3648/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3648/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=621a9d9c1cab4a3954801e5f3f5ea87759f990599e19c9ae112d4082e91e3c4b69fae9dedd97979d5f47b42d53924fa32029b087ca76d671d00bf76badc7b86572e241d39165c6b9aa157a6b940cbed8dac44b60ed68b6651ce3006ed340f80dda3ca00c39ea9dc386369e20b4f249840e6312c5707733758707d5b1f6d7bbe2ca3b1313da7ebc1e9035d6755bebef38b0d5a52a67ef803087b8b44dcaa23d6d7ee02e15d2e0f226a0abd3e8e19203c7352e3862f8bcf2df8fcc593760d5aba89a274cba28055572d179d9ca8c273809e2d283a0a1d6976c305288e4cce6c861ce86ec02e3d28c44b632c955f9a8dffa901b42f21ff0e83e86e8d97b012aa699&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862332654Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0, Name: checkpoints/1024/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1024/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=ad7763c93eae748001de1963a65dbea1cf8b4f8d883d89aa0806c95e8ba565e232a6b948efac54e2afd4c952415d28769867620527fb04a036c283444ad151d1f2b589f3bfc72091c721815bcdab9017688a3586201d08f9a249480ed18f4a7488781242b869ded6e046161f9dd32f35e61af8e82795bebd3c6d7ebd3d3f94d8dbd0a3581dcf818cba4d9ea2db9a6f4c5f3da79b8a011b0a8b5b22599aad035b7a5aaf264cec70d2fd61fe59dc823aaf86218892300d97547c53f5ae3f328f2d98b9a7739d221eb32468163b9cd87b741cb8006db9039a605c30de0d720418449683521a490db4ffd385b1d41a42bfd6f62113763abac50598ba92befc286be1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.86205367Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0, Name: checkpoints/4224/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4224/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=44de1aa1ab4c78da4f8af6931a747de7720759691543eb8aa7cace2b5edbe2881e4bc5529a8aa65fb91a690b10216ba5f2a833bd5eea79977d93b867aa7e3a1b4d54d81a71411195a8f5ec32e7254e71b480223d9107509eba0558b34316ff1bdd06d473e209b4d28ef4e56e9ff73e1a72a62f284895fb9e249ecd4c50a19313eefc5f3ab70320529f3bd3959483c633653de53a4b720abbfdd645dcb8fec99e3cf347f17b4b5e45862581b99540a6ee1200ee94cc4c80e9432e24801a98ace8463b0a8f8f89b4906526fd500791622a1cc11a799c73a785f0d4a024a5bc4974926f49934de995502c806c74292a4c0d7958d4cfcf064a899ce321b625d67f9c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.864974001Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0, Name: checkpoints/4800/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4800/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2001a027e238a5f7f28e90df9decffaf8913845051ec6fe85e994db352f07b315ca38a7023b68266c343a6004d98abf514bdce24740d0b1523129e13abf28bbc0d56fd5de6cfd713d483be3ebba66d1a7075835a41b6ed06f0e0507377c5b1fee0dcb4dfabaa7ffae89740d515bfcc3f0ca241503d08feddf2695e1879b0cd83ffd1e2646f35220b39c09beec5d5f111bada532b9c260defaca06687bfc8e198daad301f7011ae511a717081932d7f439a6e962cb79f57c58b3c5c34f44d19a9ed95acab3f4a7d79d6e2368d7126441ecfa5fbdb0cf91490d54127c875a092936f29878fa1397282bec44c78c37e8fa9feba2f92d5f89cb8777fdd35a06a24da&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.865502205Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default, Name: checkpoints/2432/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2432/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6ba29da74d58fa6e215a372e0c738d59b1417dd82a59bed22f194b1b12053c245c66f59adba1b4fc04eb5662023ea6fa1127b7b49867bd5b3f5db5e49cf67864e219a92f4cf82a7b28d14f812ca6ae432f6aa7a4c8b49c4f7917ab6ac5a150d5ee106f6ec42f4b7e9d99c365686823eb20242f6635c231185a7f2545f3347d94eac6c5e61f8c3fb0d8d35c38b21f8ea2d2568731f13780b00d26b8d2b7e180f4b1be25357cd8d937e52d4ca650c51d691af6590e66de52242e1e9d2f80c616b79395f2760bcd48a56a07fcd85de65f76d1a82875e6046415fdd06b8bdfba68aa8e962dc2369c24b8183caf0dfdbe9e137e69a247aef505892de5178cde241aa5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.864978937Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960, Name: checkpoints/960, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/960?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=0c4dc6912f004bf589efb6d8b8949c34d7a28160e24820aefb23c7fc617b609c0ab15b3edfda0c8d0b60c65b976b645945bec7bd6fa3040546e090bb3567bf5b5911b99569553c41314d3f2ddaebd5adb087befdfdccab279c471392ed73d4b50f37ba7a8fd8da815aa2092673e5d1f1ba80810af0c898257d4f402be0d823c584dbc73086f499949b1964d982a28be0038d7541992ab88df804600a6c2dba46bca0a1c51c4c0c8d55e128412a06beb0bd51f69556fe463ceafa63ea16ecb868232538942aede829753fae0626b5e190f2180b0a0365cb9e1c77ea545b4326b0cd52a7d9fd244a82347e6e314818e0c8df6c9a07b6a1cc65f9d0c71d4b63e4c6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.865094376Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648, Name: checkpoints/3648, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3648?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=adf2e08c47451a0bd97353837740b15624e3dd07eae794f45727cf82f0570b54f728679b072566e534d0bf6a0eed0038f329cd2516c56a6652c39e64fb7046f4c5f10a722a0c93ca85742c5c005e4bb09bb643daf8cfb1d128a4ec8196762236000b514d4f8d46886e2e79cced86211ab25d8d5316abdb9290583c63936832ecf2a2b51b7e765d5d7bb02bd086dc21ae5c015e5fa348de32b0cc41199f65216aff420bcdff4ee14ebe4b6712752d96d8aee1d8c133d51ae63c5f5b80450dc39a74fec8748cb10cfe461d741857854c0142d70e67c8087ddc0db0509cdbc77cc716e23ea6a45478a3e890d3130f8db5ebe67576edab721765184bb0c1c79cde1c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.865143606Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0, Name: checkpoints/4864/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4864/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=4b39bfaea3e79cfbf5caf9f389aa976f6ce331d96cfd5fe25bcf59bff916c0addf77b23ac75ae932fc1d440c1cbd42dae96b6f0aff104026c02728c6555ba6c6e03d8b4f134ad39abaa7b615f1e69d198c764085e612006191511407057a6bb2ee6a76d0bda2b89791272e4e78847d2b9bd6e374018249ed9bfcbb0bae2df27d08deb7711dea50ceceecaaddaf1cda9a090c8e678a091e8da86a1a45f0ed4f1368b1f199d2a3e6ea420e0ab53594a820c9f0d512c81b58f491700880aa3adf245f99d73e07c365d199fe8a8e6407b87aba3b6b8eb352940dece9e5d0ee076446fa1878c37d0d2bb65bff99dbd8a8e90c9aad0fea018bf13df30e4c5e04cddb41&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.86582226Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d, Name: checkpoints/1024/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1024/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=7f64618edd87aef91facd27ca50553024a7279f9f25de656fa000832aca08d3193768e285da712b0fd8efc59897e7d5ad9eaf9c9962b66ba3fd69fc11f650a26ae3849024f89051ad541675acb4d8a6c7b9a2896cd8a4e95b574363d300b787282d540814b53bef1ba63b426a68d9587fd58823d64f74937d4744587b188de6e6c9d838497726cee094026413b7654c7a5b47868bde87d134de6a58b6cab180f7fd5c32fce857bbf98427f46ee6ca5d8b6a2cfcb71386033c44b1cf712adba81f6e181169fad735b042811b5c5c21e81b1cd36458f9d2ed8f6cd597ee77a55ac7aa72113349fdf455725e1adbd2e176ce2fdb6ebd68c485dc23f867c9b20ab42&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.86581681Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008, Name: checkpoints/3008, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3008?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1ec6838e15ca56b6846455af3a27b6e3564bfb944c0d1f119a697ae233578df44fa3203cc1f2543fa2504d55825efe8064e1405c6b2231697492d87529d64ae6c35364d0ad6a3185e7abc0200e82a4b581fa34a717965b19d490476301164d6fba778330a7d1a6a3ee5c8d509f03c81fb1e2b8d92d0ace275f3f0865baa744ef4f27006f34891e90fc9faa5e0c973c1a1fed47b8af62efe7c0f6bf4db5366f9973db1a38130ab9b346a67ec50eb2aa24eaddeef3fd2a6a318545f86a4f079d353d347d2ef61e4e305f61a2bb7f666f87d2d35bff555bdcb505fcef37c75ae9231d91ab39cae22fe15ecf3fd9060baf2a91190edca16d5000f6bae9f4c720ebe8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.865332188Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/d, Name: checkpoints/1152/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1152/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2a4be5ce4a82df90511813c6040271fd5a6457cf79464d004ad597e679f03bdf1f304fbc2c3e461e9875386eac8ccef66c220be5a0415f94e2b6cae2ed072f81224d16911b47dcfecfe757ddc968dbc49929198bd6d2ab2a74ea017b10c842168fb045aa668210dc77118ca017fd0b61501c036c2f88eceb50d9c64e3df117340dd40a6bfbbb6108b595eee65914b9c8b756f165319a94f2c61989960694e0c0fa8a626ba3cf59a3bfbd8ad1a27a6f78fd7c60ab6c3bfebee143e267f2f341f97931b6e74db719dc16c39331fae20ca5e6b85ff6525d62efcb9988a8a576fe92c06b59f62d4f5c770c657f98dd850e833d4b45ee1c39db900fd592a852317071&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.865037932Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0, Name: checkpoints/256/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/256/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=8ccddd9c784d32069cda64d0c08280aa956120a6b13f5dc16624f10c009d0f7738edd90eb059273666e3aeecd80ac86b5bd5ee9bbf7eec1b5744ddb00d79aaeecb2b5a4afbb6b9c0bcb5232c753708a8da3f0c68b54e88c07a5ee0d09042cae701674b9c5e3adc96ef4c9d22482f4380d86707db50895781e620caa4ca8ee8ad7bd015db2d70024e87643ab31c59c2dc2810b04c7369cd5432e42ba357f294810ea046e9f671b65c7334a3e41e1c8ac83cf9b34aaddbaa49053905f16dd878f002d41eade84ba9616913520789e4e3f52df892d615300c8924f3c09c52d8eae34d94f007d7a8546c865b6b07ef810324874e0b5eb979b705b395060cf5ba85de&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.865734201Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/d, Name: checkpoints/3392/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3392/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=48c10e5f691e272154cc870c6cf4de51902333518dd13b61e9d2d640724019083e77bcd862a1675b51d42ddefd0ae4acc62511763b2940dcb8dadfb391d59082d9ba428c560853fd9ef8eb5c4da1d8e31df016b8b5cd59a38e17f0a3165799ad06ad4d5644e696f5427f45b0b583ce903722b0557247c491dc89b650e948fb902d69ef724ee6865accdba253b7547322804aece71e8d61c1f9377a7b5a86e04d7ca6bd4e8f0939690f62f4cfa7852995c4dedfd14f064b6ff8b9cf57bf0577c8c6e690fb721bf07f0d3e8eac47f1eab64264074e6f1282e822e8e5eff69ee05837eedf071ae02b488f33822ccf9f7b7a159f35d3863a592bfca5ff72cdcec1a6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.865858597Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default, Name: checkpoints/3648/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3648/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=7ff7027c71f4df962a8a7a6c8ba650dc9af3733d098e134b9569f548161873add6e072cee61efdca806ab2aabac41c1257ea23475dc22cd72e7771f9551596919376a94ba91f44e44647f9b1354cbf1cb83e255ca84e8431ba5d395ac5cd6388404883f609720a470a7ce067b0c6b8147b74f6b287873e445b6c03b146c6d51108dca0c99e6c2eafc6770cf4eff21c7ed4bff0fd11aba6bf5357028d3e854080ce8910d33ecf88e4436b7024b047d5d9a824b79a47d7bec41e4e6acf871ebf1389cd78c517d39604f303ea7a08ca6edd4d568b5c0c3a87998fc68f8b1684da1427137b24bdcd3c951d353d6ed460299523e9bd1de50c0f67d5b1a47fcf118034&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.865732645Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/array_metadatas, Name: checkpoints/832/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/832/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=63b6ef09280429d9b23669fbeb77a8f73ccdd0f9f667ed421c3312fd24922baaad5873f40011d03e374eab6a2d20304032f25f58a656c0a4e272e6c02d03388c6b7eb9abf51a2cd528ca418158698d74bbca82a42968d898f63ab15c191ff9572cc5040052d17eff86f9df833d524fea538a93a463e6e30ec7f69b1473c9abe65d9edbaa2bcb9e76eeafa56ea330ced62d947f0f090fbcadc4af0981744c4fcecc4cce528375e769f64d1b756cfc9ce777482c12c192c85c55161bb93905e0a668ade75e2f5649e1561a41e3860c7257583ad2d05ab791cde753d5ea32a26e553433de98577a7e9d108fb3b9076f0b457d5a77d4c14721bfd7923554498989d7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.863856003Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/d, Name: checkpoints/4608/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4608/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=069456844b51973efbd556a4934046c79dd3bf8d82a1720f82e8d09c8519aa5bf964c71f6646de01e272a55aeb9a543162c1526d9598e72b34f7b219a1d67092e237f1596d880c5b0aaf244e006e4e1627d90401df18467294feafb1095468bd06d3e4e7d7c849ac007e51b6a71e6bd3a5804f80203dc13c4f09f9401adc7329589895233f2689997d47ecea554e85f776b117553e49702e965d26e81fdeb3fdc6abb890ff82f94164526fb41078a509c1430f605eac5b7fc329ee3acc98dbc56be5893fbc3fe9e3d569e5f808ecd23cd538f2911ed7b72aa6986caf2d73f62fc122cd81fb66a5dc3fb9cb1bdce341e680e0426c432809a871d216cf17ffb1e6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.86589966Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default, Name: checkpoints/4224/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4224/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2fecf4f0726d4ad8fc5efb377bdf15cd2c83755f409609f9d430b0f5a8505887c90efc29129ffb23e570c3075fe6cf8074acf27ba02c03756ae06f838c79d45cba596a4b6b8eee5ade505405478ce03f4f262b7e7bfafed657d6cc1b4cee4459f5a7c122f9279ab7fedaf132ef8aefff0fdcb25591ba56c4581bd3c4ee1ddb956f6cb3a30aa50fd87a1b17c746a6423daa6e387ca30e5fa9e3c37bcec51feb22d2ea42715fe9693a2b3b92b9104c7129388b400e55c5b2330fd88ce9dfbe38ecddc4fd8809ebe7a233f1aa7d7cb2e7b172afa15e55fe95823b7675710f072ca016305054ae925b8811aad8171172250c692cefccac345ff2f9f6af5ef5d4ed8e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.865909427Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/d, Name: checkpoints/512/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/512/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=410c0d9f4d619880bca91c3f83bc59771ffc7446f8809dbb20f3d529fca18bacdd91da64d2fcafb170efe840f38979d91f39c24b977667e095189d9db130f0d0c439f13ad787e5c207b9f0a7b30fc04d7c84747022d9401a4c5016d3ffd2b2e607d6c82cdb4900a48b1c2f9a748f42ba5aaaa8eb95b5126f3aa96b359de09c2d78487c48478e1a7ecb32ae78b51dd4fa6225d17de5620ff9a52c8a4d0361733fe290dbd678018fc6fe888d1dff831a5237e818de92c4dd447bc46a4c79fbfd5c7a0095898f92f079f357bbc95d83fd7848b8c5d4ac0ef9f2965f95c63ac6a3e6c73ad6b246085aca0f1ad458c8d3cb5b057312b43e02ceb0f0ff076e1598f2e0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862242321Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/d, Name: checkpoints/0/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/0/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1a095dc4c09a3bf3ef7c8eb04e0c870459f1421c86e59cdc93bcfca465ab6178cf11db3582177f9ac79f46c4b0dcac572eaa02ba905822df796ddcd34757f028e813e5a3272f94800d22bdf9d25b5d93afc083142fedef1eedb09967d95c2bdc64a1735d6cd39c31854a07c76036bac76e3ebac538d20bfcc4ba855c7cd263f5285d07adc15d7b6ae110b5edef0a9decb037bf1fa4bf40e228e3f3182000e66db57dd7c49c9e5e83d3bf180edea33ce95fa5d722db449121e9fcc46fa86b2776604d29df94b16d1b3c3cfb180884b80c94ba4821d398362a8fe01b1a12eb6a7c4aedcf877f9b8ef951b440ef6b4f1121f5d4b7bce7edc354dacc334ba8be1c85&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.866035213Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/array_metadatas, Name: checkpoints/3072/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3072/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=c2f89d9cb8bd00badb2aaa5ae7523d38d2f1453718804dea8ac7b57763c8283ec84df13534e34e57f4d12d49657a5d192d420de62770da6a15c9e9404dd645d575226189b6affbce9c867318fa59a76e5e8e6b5d5079ae07ff59b59af715ba0a9c6d75b1c4a6949ae7b7c85d958b0fe624ba6b82225828c2b7d01ae9048a8cc1bac0ffe1e47d413669347a2ed9a5e8fb7dce4dfd17fb0ce984ba310db3ff5adce6cee4a17b547533a312964ed07ad203ff9983ab5bd4f195c848a82dc3c2db602581f4d871a6505761198f1c0bdef705c97d3628a1e236ebafe42d19f1e60fd85d1c64c69eb998c4b4e2bbdb9644c9357bd801cf9a9c6652f2553527dda4c368&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.865581982Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/d, Name: checkpoints/320/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/320/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=08b976e22306a073af93f697e17193685a4cc8b9f7dcc50b73b6d7ad0cbeaf7e34953da2c696ad26149d4d7dbd3f53fa0dd53458afad4defd393b05067d7ffe627a6f8b2c22fa3c444ad63ba82862285dec7964e617834e17054e8087f583b3971438c309514884d45b437c8da2897c444b464f31e154b29c6ea91973c97c3ed1bd129fe6769c2bd00c2687f3ae4a188bf33ac340953be60521d71af4f37b5090c2e7d72f1394646b3348727e487d3b776b81b38d17f05e5909280bbf2e1136b271dda1d28100b8b57aea245d0a2a1fa4defd4ab7ee05706ab4fe9d5a0c2b7ffd6d0e39d92b1b4b22fdc11acfaba0a5b6df4df082ffa41bb025fb1dd1fc886ed&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.865946469Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/d, Name: checkpoints/64/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/64/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=09b9a18d9f1e298cb5440e8cbe90aac899ff2432d89bbebba511f5dbc64856a59b3dee55460e2a7d322821a43d19c3efc5f5f736df0b710e684ede4d12d2f48e112aa5963402dc3619b559540cc8167040c35fd3ebe7fe6b5e8883a39d1be859d1f13c7b7d8546684f1e5c9be0cf075a846bd882f5ac945edba1e0da2e4405bc9b94f9d83e9347b91b3ffd3c1619b9de4529ba0bc951a02b93e31ba215e9d5e72787f3c155335ea9c952a4e73c582f0c3de4de847a7703fe83fc3dc003eb11f6a736e0c0cdedd5ec919b349cee60982ba6e03a754a119775eacabd828edcf0b678ac06541adffb17ee684d85904d4b4a7f3404614b0d4c0f7ff1c74e0bdd0013&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.866057508Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default, Name: checkpoints/3712/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3712/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9765877d9f4c6e9d1eb89a8e3706f9c6bbb32e492e0a5d0063e642cfd1cb6fc0e73fe6cfccb91e3a8be6e84b68fe44c57811edcda15762636f3aea9d39527868161dd1f30a23da4730f21a981d90167b40c6cc0091b12713611be234bcf2cf04fe621299b69bf78e51da47163879c152ae873d4b956827fac353dfa6535a9181ac904176a9c2892f7b26416fc77b93d1ed2bfe2c116d6e448cc73bb87026d064a2ee1bd35f75c3bccdf53ec23f153bc85562b4abed3dd9166fbd2e007a7b6aa842f16f0f8c9823a6eddf79d1e355c2f8dbc801d941751f7a8a12af96532bfb084b30e6bb587c046c2bd87dcbeeca39dc9b84a6bb8d98fc464149ac0614ef890d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.86612748Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0, Name: checkpoints/1472/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1472/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=0f98cffa1ed714bd4652036096db06497d363162cdcd0cd4334b02f891ffb13e50825801e47d1f094f34c6d57c7c76d62532c9fbd0e104d4cba47126d4a52012a237a8541510c26f68d1ad8abb3c8a5ee5517288dc7f2aa75197f78502e790adb847d08cbc268e14660b6db530c31807f156934c790a5cd02d2b21db0db7f1372273c9c851d444a5e0c05d56350e04d167bc911624a8df6afc8803f282f9b939ec15fb052e4314f63995fbc4f30bb79992dd9ab9f2a11c81cffe0ecdb0e7514c8c8a43649646414e1a56483d8761a06668e15f42ea56af7d8c22d5c65b1cbf78926ece7ba28d3049c9038e8c5d5e90c21ec297e3c9526e712f9f795dc5e46f5c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.866167422Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d, Name: checkpoints/2816/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2816/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2dcd5c8eab6a998e01ae9675c3e53c8d1cb030ac03c7028d63b8c7d00f1d8283484f6234c035c832f3b64ccef2ba60e115002900a6f68b0fde9b3c1cd91ca9110b14b60946e47b94b2c282112061ddcef3a8e4904bb0fc923c309225a61dfb028a06dda2dacc5ecbb78d292202f552483dd2a52abe8a3030b1a11ac07a6d8400dd9a3c7df19d58316fed6883f2925a5203d7e59224d27e501d05bf20f0e123809441622c4f288d4991ff2c6604eb1ad709f8c09261436233a62b56f996500ade1a7b62a1803a412012dd9487a897064b32e7db5109e5a7913e9e561a1526a8161f8ac25e96abd97290532e73fdd430ca166c8cce017279bda44c2c6a3d054a3a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862782669Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/array_metadatas, Name: checkpoints/4928/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4928/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=83c0a4dda6517fe61b7330f54a10fd6cb7691a5ecb56d9294ebb5f1a979d2dfc16bcae21663aa5a65dd6227da2a47bc0705f4121a6226e2c9d6ae5948384317b76bedb4a1a28bcbca08befea9b98b153c6482b6bf244a7a2c67e2cd006b9bfcd0353d745818b27ca068f9e4503ba2b00c0614a4adfcc9d69ab410d322f7d12a68bb01dddcbe5a294bd5e9afe856238dba5bc56b8eb0c2e0a08f001192cfc748c2c37b188e233dc531d2edf40727a94cb2405f6ea801765c62c9ca2f4e587696e9e8675542f88f999aca327ea4ba5bee5075ad78187e6a96b378c1435cfcff401af450c582b95efe93646dd33cdca7d9a787715234fe7bb1454c4ba6f2e42c5e1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.866223567Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default, Name: checkpoints/1088/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1088/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2ce516895d77b6d951c395b9cab6661f1fbbded24b332044660d16a9f271a8a29808c3c478abb9cecfa6c0724c076593e4b804b63548c31fbc02f41646a2f40e6913916123420889cc38679b67086a87ce374f595a6a72c12a7d05bd3e3adb488909dea6f9ff6076e03082fb37794e827f3cef283f0c3d893811c469d441355c03c8fbfb4a1c3d9dea32739e5a0f7f8b4d9b680d8975e60c0fe255e096e5c7aa750128b18ad296b7320220d30536f7d455e94cf4f143ee3415727dec640348fac80e3b216057794c567697e426d611653639e6feff69a7e8866c49133d1780bd72f10a53fe015a759b746fc863ce7adf6de7ecd514a051872a450ff2105af2fc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.866229762Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default, Name: checkpoints/2560/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2560/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a33259df21474ed609f6bdb5a344df5963a70372126b01ff829a9e84aaa900045ccc0beb2b61e735a7ae6aad6a8c44e7aa68955d24323f88e690861726046d27828d806986ec2b59a8ca5d0e59a77e21d7763e5261c292aadf114ff6e63e73aeb3d984e34d27f3d5e1ac45a69a8b42707a9da32747d5d69ae8f6fae7c8577b3119b5ab55de2930b84d79718fb69a7d29351709c6859cc1b51f7effca5f873947185c831a912b7323a91a9d54633bc3a11779a4ec5e6edcf810578041efe564f581ed70f3c12acc8b524642c8bf8f80b5252f4a3a86797e72728dc2e48a55c6e943e2cad2ef60dc9535825ce1b9e796dd6c120c95c403ed7470dbd73d45be6e13&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.865832196Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/d, Name: checkpoints/3456/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3456/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6b31d638ecc9de3e5d1d562b8bfc87cf7cc365dff7ac53b6725014cbdf65ab96e416885730a396607c7d65c07673b2d58cfb892d9d08468443dedc9c51191055c57c1de6b80290bc472b3665dd82eecae24b39d340cc7431bdda27973628819d25ffa1acd39b96e85ea580e52af8a6a2e47ad9f01a784fd5e4a44efefc69447cab6b18ec89db7e51ab37314317c1238a3e30827fda3b5c66c8ff21aea2c14747140313bab9151562f6ecf765abecd5cb3cef4e7e9d443dec6a09352d609e29b83dfee1e4eb9a5ac86cf76c684add1f8b78a88ca384b8537188ff834dc1d780f0566e8c85817786350bb3e42ce9ab1fd636c186982f0ded866d5f281ddaa41282&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862209097Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/d, Name: checkpoints/2368/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2368/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=7bc1970e44f29bbb6e3b5e4c19b4b6cd69b868db3ad832855874a6be8e1f7db4b24708507b1b1c7e0e7576670e3426dbeb9ebea2cb5758aca8bd78f7341de2c7a5d78377a008465e2e76333a0ae91575525cee32d8e647a492aad5050e2ff48875d86a355c24b4386a8f40c788e0ce42838581a189eb3d1994848c8c4c6458d97ddf426b149fd61f11235deaa820ff0120f963f9525b08add629e7b94e8afbd3368df2e36c4aed3db8525999a37dddce994f3478284663f7aee2b86daf4888f4cb56a08940ea46941ab80821dcb66f00382328c076423c0bb94a9015be5d4f46ee6f7627f19f2bde2725dac878404dcd94317306e257d7c245f59a081ab0e47f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862326205Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/array_metadatas, Name: checkpoints/4544/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4544/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=41c82e7cd5e37b8a3c1b919449320420ffd28b46e3027fb95488d850280f6f7ab6437426da41439b01c653dfb462737152bce4c8109b1f75ef794b6b334723d81008507e6a15bca02d22f4367a07082ffa78ae78cd7031ab2ffa6b16bee61de7cb62602db15d97286ab7c39ecb5630ac82284943dabda2353834222c20fbad0642002ced011f34d88447172159d4ff0644dda914c269a2c0abea7ccc096b0ea1b529d1dd3fae68eb8e0e9e6cced193ef3a2ed7fead81f6344dcbc44f9110d3632aa3149a08746413ec7211c8d301d631767c86671f633e869662f4231acca8b23dae17e4f4aa3d3166ae8693c12cfe1824e347d4408538f98c9144dabeb078dc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.866337878Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d, Name: checkpoints/3328/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3328/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=806442f31e53f7981786c8e4f2e8121eab8bb48e00db33b668998d0fc97c3fe6abbf4d8e38844bb8adf45f0e92f7a1f68a862506b011e6c36f3f6a78acf9488841852fd6effdbe81b7bd832cae27fe2463e55cc4ca79f2fc6db9246a61319a4678a887c5573f3aaa8896523eb4f6048367c2138783ba5ede093275dab411df0ce580966c98b6d86374b2f036e4b87a0c5296c770a04c620e9f02058cb9679793f5deb356cf26040c58f2e90c2954b084c1321e9299a0fef612f1514349483cd67a85461b558914203587281bd9c41dcb0a819b630d27ba953f2ba080058ff77eb5639fc9b6d3f63f3058a782c004651f11bbefeb047bc3ead0ea5054c361430c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862070621Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344, Name: checkpoints/1344, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1344?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=29ce164e36ca2804f1a1cfe8b7e8056d22c0080ae76daa77a5eb2fcf41892380aeec9bc46e07d3af72176119de86efb8d444642ad17f8402f1846f062723c01ee9f5f6ca5591b6326844354ddd59905ad92f7f9cd30b473b9e839f6a1321db6340cec89cbcbdca9db249488f3c7dc851755b797deaa730bd7d85a5341201cbccab70ef21683a29b8c5c009c83caf1a254990dbc9dae65a11cb14feb82d325a504a769aa67f09dd357e2e1191d5a7c87f2abd17aad781a79e2c6973d84359f957f0346a1e4ce0e2d60f2864dd6523de93309b0b316192f4723f766ca5a8e646c13af31e00ee62b92d38ed9f13c7875f969514eec31c847ad4effff260f6b29d0e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.866283677Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0, Name: checkpoints/2752/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2752/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a22cdbed034a4ecc9bdeb8eebd4aa8f36d5db01b9d72b55f0a8bef0f87923c37aae76d8267b09669228b397a5fabf6f871fa7bb624004159cb7ac3edc574b370c047f7a775f223872fe2b9cb00e1e83e4acc59c37e98478853fc113b52fce72dbed54dba132b7edecab11527ccb3af74a56691fb89d7d2dff5ed5bd280f4b4e8de1d0cbce6a77ef93ae6cfb02da0fd71bc9909903453d3ab5e9d81eaabdc1fd13d5d4da6600ce6d813b238f7a956b7293651695fb48380a01cfe0592c2d5cf96b1b8616905980ac6061b246c840a8144f4279a1d5b94504f20489abbe09514159fc2dfc429a753757f57d8cc344b0e26089191cb071b6ed669cd4023fd1f344d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.86377166Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328, Name: checkpoints/3328, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3328?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=4f4783218b6a652d8a9179bdd36547f860fe0af9220d7a378c14f208c520845d956a8c1a432cf721282ba33ea6508266e871c92a15496f7b9eb6c95b7226d0b0b932f7c01ef78e20309a83bf0979c6004f11016f87401e99f7a876da3dac15181d280386c0ebd87ed21417cfe2ef7e6488b8a58851518336a5c7b277a2c8082c8fe8e087a4e02a8ae23dd01e3dcc998768edfd5a373e0543929a684655cee9a6f5f12d3077bdaaf5c48148e40446b300d4b6dc988d2260d1795d4b2c0c2eba8edc0533e2fc8e42f1f3d52e63e4feebde206efd878e31d6d795142470785f453783fdfe584634847251bdd75307062595ae9d5fdd76c7ae30556bbd6dc96ccc58&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.866291398Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0, Name: checkpoints/5952/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5952/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=036e928c8b58e812ec41f15d5114075abb4fdddd77421e31a66858fd6800e49275299c8ba994f4c16440515fdf9839f0bfcaeef3c8f695fcfa4228e439f45633101a08a65817d6bc5b72eb8bb585c6ab98c6df07854c52475b0c3ea1d957139a6ec41c22c12122c9710ab93de55a7bbe6f0407e034c6fbf3bcd294bcf230c220aff750d7c5b5e545846b9ab92a7b0bc04de5d913b74ea89132086b3893b971a95335e134ef2712b800308574bbc027c8a63e2808508a9088e16e807798119ac5805af6aef60cd417f29117933ccf5c24742d6c86b6e1b0643c3eb0ba76611486661b6d3d38f2b6a1c86a994a9c63a4e8fc98b43115c131718ebef6e20c59c40c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.866344251Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288, Name: checkpoints/4288, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4288?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=74cdbf4c03fc96deb963fc5039ad0da2852cb02a4b2b6a8ad3b0c0880382910fc39259b114ac8484cad47353f409f6803e4597ed41014f06231ac879da1d12724e92a13d6e28ca8840e78926fc58f5c9ea5570a593758a506594dc28469aca3e750bd3c422a6483002d2ee4fe1fac2fa94fccfa3e2b70ab67360cf8e132c502a02a9eb89793e006c11fe7c03d61691fbceb66ac48d446800268cc9c6b3ab76e7190aa26b33694435c1cccb5f402626fc3e9e723d8bd566f8b91f88fb86ceeb5844a48e7576bf3999b0051aa6ee74f1e851efd798d000196ff556a8b29f09d2ff8b7066ebb07f0866a341889f16433838524167c740f16b2eb0d86c42bcb9fa77&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862515533Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d, Name: checkpoints/2944/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2944/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=39c0d655ed9fe4349baab9d1a1c6915d0d2574f39bcd20e79be41853ac2ceda1aaef3380066c6880f370e4db8e64c9d6d0f72b660b19701d19ddff97af6c4f08191826cda1728326f444e4e23adaf2e59585202942aa21308a98b3a039b10fd2fffd8420aefd731d5d32d5a9f9e05b7eb40c9a81e070e3923b27f0e7e6bada39f6c3e541ccca8a04ff5cefe1a2d209b49ce43d27ee7b34a065073647ab7ad8dc38f8155e2ad33d7c3cb99f7699562cb8f60090cebadf36c0088c9b80eff89e59f944a2c3c01328ffc2f2b68c7f6c9d657fcc713e6551a04acc25ca814cd5cb3bcef92bb32d1dd6b1372fc15a0aee027d09816be964a30c2b2156d248da0a2142&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.866565228Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/d, Name: checkpoints/832/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/832/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a37c1811948416b90b23a52f7111b5f14115c1a9587aefc3b81cc7952adb30a02f2e8a5743f04c5338a8edb1962945d72738546fdd7326cb1861a4e5d1e6d12ff82eb52bbcdc9c4bb59cdb8cf3440a73fbdb035a32753c09505e0edf33143a0411f335479d1089ee89c6fcf9cfbcb2354e6967686863784a1124df43f6a294e5269f131d549a295dff410ef24c9aba0d24294681a1b50157a2e270c8e9a5752dca987d6f616b549657c9c74f64d2b0b1c816352ad11fcae3a7c7a93837dfc48025c65332e63b92b3b743852ca79b323c34b9be253b4839be62bc9299b2218ad18a569e7c173d7df2824649b39a12cc756305933073ff83d1381d203bcae87082&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862115536Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/array_metadatas, Name: checkpoints/3264/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3264/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=947a07c3f7d826ed8c71ba4da6d1a374524d99a32a989797d935a2bfaba9a1fc385d422ba9388614fb1f1e80fb9d48ab96b9ff0789cc997787d06bcca14df1e91d98d6a915375f9a4e17ab82de4a1c288a5d8e7b22e76003ccd286e7d65fcf4d70234b9edd9850e22e979aeffa3a40aa18e1b9baabe359161a4b1023bfeaf8222b9d3489ad1ffef22e285521df96835808f296bac6b16fbe6b3e95915060ad4932ba7d5440db725780fa09bc12bd7f6b9373f7d2590b91f350c257b4f0002ea35810d57cd1390f3e63150a94124e91f8b30c79cfa84d900db0baac57a7d33f18b4a77f4e0868ff62248f96c56e2c090ce272696c91164c53d6db1bb2a0c8a555&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.866708054Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0, Name: checkpoints/5440/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5440/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=4deccb6822f56a81d07df85f85549e063b719fcdc6522c083c3bc30b1027c5274505e61b55c7d52f51abfba782c37e4d8b4bb7e71151bbe39cf071edd427a0d46e68aea5952a51a4033ea63ee355320595170c02de9ac27952857a575a6eef603b69d1cf86474dbbbcd2c2c54506d8f96409c1a847b1eebe730e9cf6edfe2f7860a1ef473c5295369125234adc6207f59459e60f1c7b310bf8caec48dced3c2d4b7f27f41eba03000d2dd823cd215fcbe7420546af0440b99859d8d6b3e1e879b51d37957ff60cf64221e1629996fc184699cd83caa49ffd56d6e0d63d48be4075fbe56b9baf96ceefe5a87ed86387da8c3909fb734a7f396372f60f37a1c464&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.866570283Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/array_metadatas, Name: checkpoints/4096/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4096/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=75da327f6d1bae00ec1b3725dcf23d6173eeaef0fb832c6c7859abede6ca08b1f4f9d13885a4539a4591b90c7d1a5d333aa9e304e104150117c518d0545621e5873eade4b4e4bf2167a04829b80dfeec776ab3c838a59f12e287119eff7e6e054c7ab745024298a19466ac5def848ba842c199c5dd42b307c3a709c05a3d181a3007e2c9a770f545a4f2d702ad4e7299575e65b550a1f314b3cd6d92d74cc275fe5da007af1f96fda242770c9005f4ea3061cf1c399f27bd44952c78f171c12516c597841f4213856fd63f3e58202b87656b781225449755584ed40021fd8d900983ca6b014434c541353ad75e4c07742b1f61dccb52c9cdce191451d0aead0a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.866798204Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default, Name: checkpoints/384/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/384/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6d5113e1ef397d7e50085e6788e97c14b6bcb943d215f386f9d734de245866f89ab9f09e63ca42f7b55ce9b0c2f15c1eb183309f5793b0aff9483ef4e64b456350f217412b6dc1ffa89821eecc40633d24e5bbcadd4f0fc6154f376bd65b5726f95dd82a7431fb7cad98267e9d3e5d532ce3f5dc8e5a903e3a86a5f0fabd89402875dd93faed05f1f73c0f42248b96a7dfbe64627564a541ea0fd22fe6bdcd546329c99658755844724791b32b2252b362ecafbe4b10642c866c225e600324290387c24327ddaf9f087b05268722384996e8e4abcb69ffcc427de664ae3dd4f6bd81f3feceae001604c1b0c01492546f1501f439d16aa5b721e82ef741dd61f8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.866798134Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/d, Name: checkpoints/576/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/576/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=8d7b6a436391193e65fd9cb8d2a6d50f106715c2c245f30cf399a7e62ce1389e5b7e2b32ad823d0d4d7362dc25554a9210a8c2d09a19371e531af5f320139c1515ccba724dfe09e38251eb1563a2b6a41898ccf161432b048bd9c4c02fe3fd3344ed07786dae0e9d1ff0472408bf50d5b2d58e127c5b93becb49aa0744b68d695830223a75b4dabbceae5e48dd02956a27463f852801285a894977a1bb3d062de3c3051f6e2a9013462ed80c8ed504e9b1c60122206a04467f784c6cbef4a90fa015c4cbeabae95d63544d1f13c8bda23e69ed2d2cf996b26e823918b40827eb1a1ba8923e95d002feb6309c5e739f831dae2460747ac253cf6d9b52ceed814a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.866850146Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/array_metadatas, Name: checkpoints/3008/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3008/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b561f5710e42a54651d0469b81d0241394a1ae6c1a4a1e6b4602e3a78f9cdf0c27b6ff8ba124040a5f8c72aae1eccd5e6baacbdf0f5f2dd90cd7bfffe683490525fc3fcfc62eb9eee21c361c28ca2efdc3a2636f6fe2fa9db6004f175df1988cf99d39d6911dec8c44a0ec7631ccb20a1c93c0943ddc6ce5e760612d75629784df79ee65a11593b3d5634e89a70818ef15b22555a57dc0ddd36a3d42e190ac9b4a2eb41d0f1f79d3c11b3ff692cf0fc970de7ba5cdfd0e748dc5a2a5ead892736840f0cc06b0d2de23de2d25c4b5d46080485f2aa85be76c5bb988064750572f331a395fb257bea426721edbf80a2e9d6293e5466d73e4ffd73acb1aa09b7023&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.866848641Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/d, Name: checkpoints/2880/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2880/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b7c65a142bf8b6ed0f3ee9bffa421b54f60d57c7aee8d05789e4c8b3403a5400b0952a54f45585017eb21a80d21ba4386b9ad3ff0afd3876ac1ca964fc3ea6ab039621457dbf3be97a071bab6ba99a1ac97c3256cff6da63c3f636264c9b68a2c5b471ba2cba61faa5f376a4ac5fa27e8f4c7bc433f9e04baefe4cc75135b24e9bb6c08a896d13dfbf45ce8ea01aaa123835cabdc457f1027b9e791be1f8721965e365154fc981a42175b163f60042ea519422dbbebe43c25152a8f536683ddd267c7c079f2f3014432b9c5dce1980636034ce240c3c42c79e380cc891562fdfa0bbc121b756628607749bb0e0c30026246c4a2a0c3811ab857e8292855cdc09&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.866922373Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/array_metadatas, Name: checkpoints/5632/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5632/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1263fae300d1c20e4fbe714c8d0c16f3d1fc7390f046bb371813c83ad96893d6c10739d7c326a0a69f9dfe4e67761419e1fb4a325cf84f51f0dd52c5972b57f651eb4626480817380b6d8295d532ca8cf0d72b4ebc603a60c8320f4491a98f250352111ebced89ffa16a2e30c2cb0e8a3e4d29486a206b7c285dbc4e4a62e78a984ee40d55c7084d5a55cbf356d884d31cca422a22b2d8ac73637f755619b6772fef39d576b1463c2d13a93e83490695f16fabd01cffe6ff26f479a16b14162551716cf7fdc4d889a9f05f286bafc0adeeae3bbff238fea3aeead1476f8fefd84cf3853ba5b8660bbe8e5cf8181d29fe414d5c54fdf29fdb87496bcd8cbc145c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.866322194Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/array_metadatas, Name: checkpoints/3328/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3328/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2e50e9e56007a20044272066757b4ba8f64e4d1744d8b0012b30f483fd02c79dfc5b5181739ad00a57adbf29cd735e205f9d9ca8d329f447465758df26860b2d4f208ed6fac0c69e608e4415f0b5492928cbf7cab4404dbc90b07d4fb76355210c24fb659f8c8a51a3146e705c1cf40fa4ba8b7ce1f106129ab727f903aa608a39d9a232bd3ff80748526826d972036423ea49998f0e24f8d3e9a9a43a46cd9f47d6a87d5fdad7181ffd0223b0cb4a834487b656acaf46707b2018352bdeeaad986b29e01a04edf0466d14e9eae340b17e9b9e55f54bd2dca2401ea2ba9f2e6802b44f38bd8e5bbe90bfee725f5c975f9d2acc2b19d26944f2b0829560a7a67a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.867022832Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d, Name: checkpoints/256/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/256/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=4691c434dfdb93ae9d10a0f8473ae7beb3ab5fc706a5fb583ca941e32e43d6c6700895e8e649e470c85b86d6ff3ec773186aeb9e4fcccd96de14feab199c76fad3c599c0edf3236c55657261f46e25a904d7d1c5002c9b45b5a2324644de8517864ddf33a071aa270c41b748c900875152f4b6135444b7c437d964dc538f70b2272769e011db95c68bdbd6d321432904ce0e9b90ab7cf1d285d55433966098a57a5c3b23945c7aeea9f6b32e992ce19af4b290ca8a9d8d3f5f2f780a49035026e1274f544dd9936b9180ed73f6b3ccfe98fdd942e600a4b5aca88edfbb593190539e117ea0bdeb72a9a0135c6059b5754e2b63583c5981e3fabe21f6faecf085&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.867063992Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d, Name: checkpoints/2624/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2624/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b67d498ea576029d66e010f01c91c5df7f81906489e7cfdf657883a0b79300308f4df6ce9a932b8748d6c0b01510284d2d732c23876245d53b8b6dc60c26bef40b23183f21df992fe21e96ba84d65ab2f50226495953a0614d923b96df57a25305180b4ccffaa0a9281b8c71d7f8aa432a8d8b21e3bcc5041ad173c2ef3557d04217ba42876eddd59670ee27a3567c90ed47032c759f4c69d8252a7fb287981df1d38f1b5d75003c282b63c9b5d7c0161ea6f0e7e5b4b0d422f4b46688da477eaac4cc7b13896c6303c46001c9b0d8574347781d2d0a99e26b85e9c6afb2bbb375431022425b69ba561f3fe463dec846e3f199a83d69d6672d7c1841614cc16e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.867128107Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0, Name: checkpoints/896/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/896/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=81cef0a11d72f86f83384943e96fe6fddf7fc4ad594bd83c42b72ab6e9242603bd02cd93ae2d6da080b77b3d40ef488382f29386c7eabf1c373596670dbe166cc66c935ec0e2caa3add471d5fea2facd0db1cbc1d2269d6cdfb9774980878fb3530d41a873f65f8b8e695788d74ca91060862df6953783acfd8142b9fd720d9bde71dba208ee29e4ed64da73d34fba577a92ab5fc8f4db35b788c5e0c5a747af78c7da07b1a549d4ed7d78fc0f1a45141cc1a91babc787fcc9ab9ce336caddb56ddeb009adb2a56364361bfbdcab0fb9a74bd1de92981f4c5b4d3f363ae0521c4c3b99e3f94ea1355d8fda24cf298732853b751b9438e9dd57d36cb51454ace8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.867142286Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456, Name: checkpoints/3456, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3456?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=25606484bf018a4c27e1a84de17e0278ef3fd37973d06b0b8c2895bed13d3229698287ed6f28f4a1a18677f0e5cc428171e1ccca738d0fcdc7cfb5a78ee60aa7bafe38d9af9a7c8b8d9d1f70dbe1b093ae5757bb9a839a11eaad6ffdc8232d011edbf20dc7a3be5c1ca8593c4e77bd6108eeccb11c6018bbda494e3f0c6b6e227db93ff05722558dcaaf0662a7d4a85e9402bca3b32cc3f0c39dfd31b421784ed2ad0986f79b69fbb99fa2335c1fe52b6db498239808e555e51747a6b64767ec35c7b935d33823c418a886c23869dd89af8efcc754bcd19737b36f8b591c1edad9ce283a8eaf249cc2a1b0a587deded1c36f3b5be91a91128cde8d4e0ed558fc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.867172592Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728, Name: checkpoints/1728, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1728?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=565e097e103b37040047ed5911ebd942b322235283057ae69192087b13dc01cdfc178ebf98007805c68454ff38cd1a23339db63cc7de425ef90c0344f68a6fe834dd3378fdf62287e16a458c5eeb796599b93a86553ce17344d9e15f08ba49978bf0720b072c8751d213925de494de37c9b9628b11b4903d9806a4765058a7e0a29538e036ef2f3ef9c2f14d1f08fa6d2a2234eea583d25bc9da7e928299420a18dccf0d45a1c4a89db0dd220153a927f1fcced5a918fefa06515175677dc75cddd525e0fd59d3138519615f008cfbe1a1f0334ab585c87762ec1600623964e658e940992b98856c3b8ebc89f65d40c308afb37ec78efde6bb430afca7282ebf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.867268162Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712, Name: checkpoints/3712, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3712?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=3381182ce5bd1d15343662555d36d1540adce717165e59bba7417c2a53b3effd43e8408aca0c897a280592973ca0207a49108e0ceb3a5c24843d7cb7edccba63ee04de54fabb7b78ad7490197f9b88e9d5f21ba0d4fa914cabd58c4095733d0bf0182f0b9b1ed8b79c8ec8f0c8a02910030014e50e046587caab020f400ddb4e051c14768d0aface0b8f6c874c30928a833f832154f7b028ac4104dc6465eeed4f0b0195dbf4872195b2e41fadc5bf022f955866a98042600c0ab555d31453a199ac41689a8536696f19efda5c8e0c9ce1a1935427021dbd94462cd8db7a0a7cb1427e4aac5036d1489cdd5bf2347c6b35321eaf08a25696ca68d1310775afbf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862162626Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/d, Name: checkpoints/3968/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3968/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=7a1749933a7b3955bc170d3af263c8d4c542d33f2673777ece45d2deebaa355129c1454599556e62c81e147e48181e7dcaedb962d94c10a26d94bf15f7a819493ac1300a64c8e761e5ae7fa61a9bf0334cb1a29a5f543e8b68074b9a9bb8e118c0f14ecde8845f6160b9bf4543511091f1df0c411f57fdfa6b392aebec1d3b821e333e42c0cba9b3b811b551c83a7f70e2354cb55c3aa54e03662c53aa7508d5c3b00c529a9a4a3b192a5f31f76e54daa7fcf14338bafd890c08bc5184bef373aadf753bd4844cc8943bdc3cca703cf7a4b95925b5d9d5aea12ab121308a7f1531404cdf03bbce98acc672ea227c3fdffbc9b5913f76c60c9dda7a10bc8806b4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.862489545Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default, Name: checkpoints/5952/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5952/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=0cf6a53b4d7987caffa6adabccad201b094b83c4e75a9a99ee87be3d8d40a738aea6b72fe4d78c7486d4da0f4c39d3422cbc0719539ae69711ec1c9311d0810c731bc49610a8e8b7cbb1332227ffd8e7c2893f1799dcfb326ee97bc8b672c5f67efebd8bcf32531ad1903524137e3c10c6492638f7bb9ccf9546433660432a549041d220f7444371db5d6868eb0f5367de2bf41cdbbafc336b444b1c12331b7249834cbed2c80749926a3156cfc2582cb3b27febb8cc670c62a01c34c687653fa5251460873a60cfeea3e876607f9270a3e700e89d66c7d38b08e6ef1355d3719149d8f52a1f6950cdb9c6263c494a4c4a9c6e7059a21d9a1af3fa669dde0f15&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.866915765Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/d, Name: checkpoints/1408/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1408/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=c0ad6517193f2e9049586cae636846450000e45d49a6e89df305a2bb3de2ab9c94c910d6f54acd5d3fcf4b016a9c8818cccf41134368a8cf62ae9d9c59dc917db1104c66f94fefa14b989cf59c815c8746a18d5c7d6b1e32b0e57d3ffa456da3a43a3f41a4e20a510ba344421e8908e90bd485903519eaec545352e5fc780ef28efb8f9ef7b82b517213f957b8fecf74ccce36de39375aba5112e3082db5c8b0f439861d097e4053e12e706bdbc2cc6767d7e4157d79ae14db50f33561c4d4fd8b0b0e332290965ba7a4bef3ff72bc33ac949232a2c6f147bc30df7880e17f6b9ce4d255d5d42ea82df2ff60c70d15072da9da43fb8c7f4ce346fc149f8248d8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.867164708Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/d, Name: checkpoints/3072/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3072/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6d194c5e11e4e2e4cc96514a4e3e975ead205ca5e6885274e3d5f1caf2cb944852b3c152a0ebaed3270c0d7178fa70f36062e8b73e3ead945e3d09c97b2fdde8404088c3570fd01abd3213f8c3acbd3f249b8c043c1068a1bc45ca0088d4312c46a06dd292d4106b247f864b6ca06f0e8593e4eecfd9d1c5f96b61f842d43806c84b527b7021b7fd2e708d6d0534e20d52b7eb58faac8354212bcb7efa54696f73bb56c7d9823486c9d7953aec804c56771b1ca5fc6eb7f9206367c7d8a350d411f755ff4fbfac824b4512f892c48544bc931567ec2c83f3246e02ac59a9dd405e09ba09ed65acd9672d5f8604cab11fc46e11e0e91e6bfc5228103583bd0e9b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.868719803Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/d, Name: checkpoints/1024/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1024/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1107b54eeaff53379fa9b42869b00bdec52cb1da234f5765eb845d6d165b34d8f491000b8709e391d7fbeee2db16eb86591ea553b654c82ebbdad178a62dbd192bfa8b863a83f25c0e8721c4951aa21a345809af4d684ef91568f5048a4d182be13df64918745bf2e54d5ed702cb0659da9f44f4400765c9a11298ce2caf2b359ca887aedd8c88372e7b716323391724090d3a930b0926ca950bb1ff3cc94fcc07cef0adc652c42a9d898c6cde0be2a1e9987edb584a79de9247588941179fedf5f532bf18ba300bba61596af3ac379829f4de1c3dadcbd262dfb74f3651ca0db7702f64cddd98f46ddceac44d29833f9584740f51c72756c640cb60e4bc5d23&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.868769603Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d, Name: checkpoints/4160/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4160/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=07079f89fc833959d16126e981f830bc758e850812d674b2932a3daef46bd0a1759734e9e8c457444dc7467ee596fffd10e86806769538cc674c2d7cf0b1a8a043abb4d8707e24718d43cf7114179cd97758e883c348f0c41d580e78000542ffc456b31a885fcd603d3ae5c1279d823e8bbb17131535d919d24a6d4908bee72823200bc3325658b68ddbbc0fe8b908940a15a3295b2113d778bc726dcdbbdd403f3fb38e11be03b3e614f1aa91cd63258e473185ad17aa23e83b65d371e51e3f18e4515d4cfaffbe6898db77733d882d82efa92318678feb9bdfa5a44b4dc6dabe96572381a17ca00a9f57dd1cbccd72f1070c04d199ab034d5b00e7e2e9e466&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.868853188Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280, Name: checkpoints/1280, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1280?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=54b4325007f12dfb4fe7a099076fdf94272a79c956f6e538905f2ee231263b48273a1cf0b18e80bb25e4d8bc1e99d41676bccc2fed243e4065fb8efa0acbbbac178d9eb3018dd4c9ce3115c949d32c022cf0521db638efd3637a1f3a166c82420c7951776e0b503cd96c94e9aabecba1c32462ef340825a3ebce9b5b9e1081d9a934a8daf78ae702b5c875e2c294a40a46470e3a19b5ad529b1bb02bbec243c5a760bd44d85b3160cc9a3578a526d8a66f3aa022be43c36e62436ebc2b0b29ee995a2dd2b3b18e3fe698bd9a763b75e83b48b8c72c53b98adfe9433c5a763a9278217f4ceace8e3a86afe845600944fbf7ffc5f93a5e5895396e114fa6aa3e82&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.868911436Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0, Name: checkpoints/2816/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2816/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=3ca6e1c7b903aac7144da2dc86868c2dbdb285fed33b115484559d472963a3343272a290dacd497d30dadd5f2c9e4f5db5c96bd50b4c116a871794bbd42736af875f534fe33c07e35bcaa7c4edc770783d40538140bc6074ebe7e299c87d744fb6b38501c278e92ddc08b1fe10329af911db64060eb31ff42158898874b8cf28aca0d137e9ca7e9035182b36d148680f90bbbd11807f2161b368ea3daf4d987efbacd86ad0b58bd689df7d5208a1bafb132cc08792e13cb73dc5368736ec322aaab44ea9ff45fd0633269a23f9dc2d8da61a2911675a50b1f9bf7d804469cf1423673366bdd19fe9299a1a791a5d66e869327919513793705e773cc76d8a7ec8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.868943708Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default, Name: checkpoints/3456/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3456/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=0b09d251d3e328cd85b9bce450ab07af0e921b2a8e7e068c935f458596d5fa7a76b4aca75f5934e64120e0cfb79b98eb81ce71439651e5f56186cffadc5c4bd3a5ce70231db7132487b531bcd57e5298b9d508e8aeee1bf9c96471e4419e5d9ca5db10108875a871b30a81a1f5a9abc783ebe56657d7871ea28a543e33a9066458317baaeced44631bb1ee06d6d6b44d9107a6506383fa1ae8ad10e9f6acee8e330c86a01b0f37dab5d9d3b02fa449cb954e953c751788f520812c211596876d9b338f887116514dc8cda99a9b919056bf91f204294843ff310603ee8d50be88c6813f8baa2e977d09f4707f2cbeb032ac842625678b52f62b4d5b1ed5e1b2db&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.868960973Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default, Name: checkpoints/4288/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4288/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=882157b1926dd71cce1520752b3879fff5d21889c4d62199191dc51f9bd2fe57db836249cccb234f992f9b7872c747b548303e0ab16ada0572cbc5162e85fa09a66d954391fade46ef1476fabf32b0ca4681a327394a8fb2f6988cd11a5ac0b36ef7d1af1154ee70d2446ae20168c77cbaf93d324410790b270e5f862c9ef1c2dbe287d5f44713c85fef134e90c42116269e9a992ee2e45db1cdea1b8ce633803cd406f460716aa1f678646c83096cc59498b412bad2ecf33ac671953f42a73ed182a6a7f9f96c09d387c92dc37c6326d503a13e529dffd59b64a9c7b2fa0ddf61876587156770e8547d1cfdc57a02ac058bb60fd21be32392fe5c8854ca2ff7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869040243Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0, Name: checkpoints/4672/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4672/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2aca023ef64d288a9a65c5f8438eb3eaac674cfde978db31f86fe0fc31d77cfd2e45cfe68afe9bbeb131b02e5bfa7fac9b05868e91c89f4f1ce4e85c229c9b751e32aef91dfbe19fcb18712f4cfa85e8d455ff0f0f29b241dbabc67a426f4bc13296aad0dd3295d03fb471ad6fb3ef9345e283cf7a0a7b439ff6a948ab2267286fdad8a80766ca867e7fe925d1bb03887381f3a620460ccea8b97883ec8aa4a081e4e09a7aaab0f0400e18b78aa4bab1aab442ec84dbd52f508a436377574c2b83d9207702fba62cc234e872f17eca555e4450e45ac9ec42b041c93ba58d84ce5b00b925be86d79ac0fff2f820c9a0ff91b1350077357dd7f7691661b2fa9f5d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869071471Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0, Name: checkpoints/0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=38272e8985c788b6e54fcabafc5458484735e79de269d35e9112361e60987c9331912f1bd13f71b576d984b2d11b46ece10ca2fc87ee31162c7d8ab003eeb4ff8c7cdac7874f8e08768c415b4b8a10c536971b3de417804c96619409ab2839d93a8151ddac78eea28f63aded89fef6410cb067e8e70d6ade67d578e024dca116a9dc405de0892edc960d440dee2211db7cf4d93c4517f5b8d830717b337a187b48bcd23d4bf9475a5c7d18d728f2e28207c7a6c699a22a68ebfba131a807762a6c6b0eba8da25a8e18f34c59dd599aee120f40117d6760c9fc2cc4034b7046924b573c9bc7268f085da68ab97e453f1f7362b995630859695d868c77ba7c1257&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869100726Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/d, Name: checkpoints/5824/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5824/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=4c8d6d6ad7fa57093951dfcceed199d51e79dc2dfe88aba99210326ba00aa578a7e9661a49ae9d21905a6e4d4dafb67538b270023c463d9a9ab6416ed7ef55df2a5c9a576023fa3e1e24be43b3ee5dbf88747f564f5703cf7ba95572a4c7a5a8ca6b922faeae4b8733f4d239dcb5eb8d67eceead39b260d432cf76e0a3ec6bbfe34702cc2dc8ce4d24bda66b904897c71d002b4679aede9e2ff21e8ceb8d2585876e43b6d6e6c902bcd771fa1ab7b4a35187975632e6cec76492b9692e236292592dcbf11711124f621381a7626a0ca8b4a0548a3a45008c78aa90d5c8146758c019a8ef601f7d3509a61af7d7ee855008a7257a952abd52f9d5045e18a96999&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869145528Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/d, Name: checkpoints/1984/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1984/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=26a3a2f77b185cafb11b77b5f9e419cb9c282243289a436ab45082daf950adb06bcd1d9c9efb1065e51a533ae5b4b13d45e140dbe3a9f843895dc9560b234c605643f08cb2fd07b82652d184c6d9606d06c13822d6000b81996594105457e896956e76b6e6e75ee7dcfc20f6d784d9995372537a97d98542525bd336e8debc7c27243107bc51113289bfeea0a2f4a93d93452af16b154aad748c1ccc417953cd5dff6143a1b7ddc01a68af710be1b1a8a99a9abe8f1aa610c046ab54ee7b5eaa6830997f911e4a04420eb711d71c9a00eb4d9c4094e6195e53c1f09952b7e216657fafb84d263f558192c5fd0292fb30b7a11041f1debb8930fc1f368134206e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869182853Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0, Name: checkpoints/2368/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2368/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=8c52ccb020b1884ea2094e501cc3e22a93a694b8f06970056497966c724ee1614ed5861dcdd8f8ac75a05c24116c607e96ac53d8119b3258b7f567a74161a57d8b68d984d223da227b126953751df169f93a088d73f1a70d580d3395ff0be0a66ae89dfd0238d7e899f82de3dc6d1e87a19bd7ab63466070c045fe5f027685e008e00114871bc190ab65e0f1e0678c34fa4b8f90023180b10ac1977a9d0bf458bc305f3bf5bdbed7388522612c58a42ca50b5e6b54eab4b84c3ead05c22f6df340adf42767dc377bf48146265a7292686f30ef2898777987ea44e6bde189ed657d3be7fb3ec6a5a9ab6dddf5159340b25b8806ff1249de1bf88f9483e25e1bd3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869201567Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/array_metadatas, Name: checkpoints/1536/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1536/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=227694c370c4adec3e03c5dfdbb5ce256c7b5d89929c2ca8ae5a153fc60222a3d3742c505eec4c1e5019cb031f993b9c62fe7b2472b950bd7b92f066ab45d4227492004ca94c0c5693cfdeec42f7c923bc26f7dd47c08b7210717206b5d6f59b0f4e2db56a918065edca9ba211acca920c9241b4a75ee18c3dbd55863db52ced03f503563fc65778aaeb3497fe6aea18eb1f767474e13dec46e2c9699ec66ef0c0bc4fe206fad6ded1e4bdcbddc3e8b61eb081a81fa47b42701c35037806d89270506c91f612c2d5c60387365f8a3c98b54d3277ef130c68c1fde6af0db965e9b40cd1db80b6c53c47a8b44d2c4544d69ab4a7bd7822a15b60d48226c7434756&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869264968Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/array_metadatas, Name: checkpoints/3904/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3904/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=74553ba66b0084b68586f43d40650de3a2fdd024698cca5eb078647b0bb977b6625867d11f04fc38edf1dcbf172ef913d745a0aa24604a7e7e6715d5f7d29016fcc07c83f91e5aee5a1eca533789d7f87e48621709fdcd31ae5ea8e61cba4aa9c98092e254b080817f6bdbfdef8c66490518e80a5a8e06adf637ceb72f286f887b74af691aca27cc8fd340afc51829483dc3e78c25954803f7e88b3605f5be7de4dadcb4c85c731ae3babe211f8f933bfff9e4ab75f20c47762ceee01eaec9f893ab277e0257dd12b3a015584e5c713150eeff20c2af8d0c3be10a506357e459b0cd8472473e67f741dc021ef9e5a2f702c168b615f7738964b27a67a1768572&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869260015Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512, Name: checkpoints/512, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/512?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=c1dc685e599a534741be85037edeb8906de32edd740dc5dca292a43f497d7fa92e6c897fdfd5d6c6dd43a96c77ed6639301e49aebad4d12bf84a956bc98f8c07b3d3abd4c4ef77bd3611aad55fa8ca73e1b5d94f6899ba6f6b2293aa0023b656a8befae0ed2bb73c3787541307303c0f5896d0291ef6802d19df23d46555ec33c2f13747329a584c7f156d34b25fd3e7acae7a3ee82c4b933cc93c51881a0f5d877375f3acf44f5cec90ff128e09d3148f342766a891e4de1f37d746121296796e56d359c3e6e1de117764cd0a66f388bff6500efec71dc2562b6979ada520aebd27b6171c58220e00ec7a66e5ed6f72ef423a826021999729557d4119b01e0a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869292722Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/array_metadatas, Name: checkpoints/960/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/960/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1688ae9e9f5bbc36a353915d0d255bea062c584c52ca3d272bb22bee592cb81631af0ec50ad8e79511ef1e657242a1263975b8c8f1f0eb6ee0b65d0014902d6b10d78dd4cb5763c7ef03c7dd1a39c674487a4cf3378e780df8e8b24d7d0ab2c164186bf9a094bf7eee2d5b82d52f17772282c6fed6acb67eb8f807817ea9e3505de55f49b69fb038f77a1df74341abc7e1289ee2a972335a5e1484429591021d85677cf0c8be008135c0434cc0636450929ae135e7c0061e0efdd62c6c60563c2d1deaadd039f7cce3554c03f3032171e0879da441025003d0cc3a8b9b84c585033562f624929792aebdd0b636a0395db51de720d88aacb9fb707d1e27104df7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869286153Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/array_metadatas, Name: checkpoints/0/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/0/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=28b9c0492b69dcd5133ab4fd9297a9a4b6946756e52c42c2d3a9858ae91b232cc91ae02a8e1f986177a0ccd18722847ac6cdc034a490dad84f5d2ca096b6e08e10f5ed9ed7ebccef0bc2055f1fc10ef8b0bb34ebb1f7288c5a944deee6a99105f43504174672587fd7ec4fecf76cbf5c0df86e2c181a4921070f08c071d571369dc86edc9f6835536a908c317f36b814a27a6d23886297616c35a51e76bb1e7c3c1714e589cd9fd882ef206aa63e4c9c482b70b74d0a9da6e79b8ae281a86ce2ce3c51fa7c93ae822c8b8f1effe08396fe125f4c9d3360089c95df2ae589a2a83f8b7c4a107553c164ab872b82c2027de12c05acb1f93d99417112a3cd7f0d50&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869482036Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/array_metadatas, Name: checkpoints/320/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/320/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=910eb9d90586b6e169f1ce74910a951408bd37975e7d008a09ef4998bcef8390d0256c0439376c998b3b5fc44f9238a934ca658df185ede8a26d0906bf39290274a4f564ecbad6af3209cdbeedbd70ec09e65f63d53d9b49f9520e0f2102321b999f4bb1820f85ce76809be264c07c7a6ff4f4c7d581a86a0e79b9096cb5b8e64fbbe01b34306c69a9db1afe1b74284ecb049584dd90ce766f0fdfe56f5bc3819f343633c9090e6c9f150dbea63d6cf165820cc56dfaedc303f970ae4bdfab8c3a8cee7f0c59d8b585e19272fbcd311e855338a5f7871f60933bac6697a18fb383049f2657e8e753686e1428570daed31f88cfee2ad1ecb127e7d796fd9cd321&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869436896Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0, Name: checkpoints/1728/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1728/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=5be4a9f742a1a7b60d7cde2002443172aa6467f566cd26c1070a61c97e349ee8f29cd7f0a5c5ce25af17cd65545ebcc5a23660628258bd5fcb715bfc53e5a93f70cdedb9b9c7aac73c4d357351b45ec0ecfbe0e6296d43a7a5e8751741fb1c187980a0a693c37c5efdd344d135b88181394255302d6abdeb0d521c10f9f4c988607a5fd16b491dbb55e17298b1c9287c111a6921614bfb474fbe5d5474ea88aab942ed32411f80a263ada87d846b0e12cfbab8c126bb5212bd9860c0e91e77d947fb6ef1ae1ba6beb8fd02ddbe9c6a96208cd5857d42b1dd46a90f288114cd5781010c6139e38cb6bd3860a5440b573195cef1344361ae3beecde81bd17fa434&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.86926433Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default, Name: checkpoints/128/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/128/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=79408da39352c0d66178bdeedeafc54116b05ecc5e33f85a8ff2584f959e582a293f3d729ca404cf566fc54d67bd717274b94e3218e74a19f102e139d8d749afa94264ed54a9f155ec6110cd4868daa18f50c6a5832bde9657291559220d3d4bde074ad1662674524ac80f4b7f377f1994225263c6e699c0f3da1c6fe245635258f280cfc865cb0a0e49116d4869be270d6d671e7a0cac98b058f0a138ea76c9a63b6557d7226fecbc134b62a4ad26254f9d128cf35782bc629c743e5ae6d38efc70563a114331abd7da76f082a74c3c885d61ed7f8e8170f0c3461b005d0c79e8592c9278f7b7124368170b2e053f1e0cd6b1b4f92e4f3f1444b2fb18bcec44&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.868773373Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/array_metadatas, Name: checkpoints/5184/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5184/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=44b0f64347f6eabeb418d2b6864df52c19d177ffe21deb43bb56f1c124e6dea6f5b5155c64aa80b241a801f6b1803e9cdef5b295e7175980ef6c48aaceb67ab09a77e6ce95d007bb1369f4ec08fccc75db837e475c927bcdd346851e1954175625d8134f57e54754aa5dd6643f6707d249f54a38bfe3e9ceeb30927481fb207a57c136bf6e5b3266d5da38454f77ffd8f07dbefc5ad1ed1eb484c7de6d6f25d5717e8aca319e3785f35e5f3ac115e5157a58ee1f2a772f6e6402d2cc4cff52bfd008eaaa7c5eb4dd01272e3700f418827df3e36f444a3c038e848b004b5e87748d390100b521f725c24c53147752935bf632fde217c2b5dfb929690e940cd430&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869574237Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136, Name: checkpoints/3136, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3136?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6de9746b2e43101e43ad5c972a481467d094266e461f2de48d266fdc499e2d9433d744c21ec8ce024aee851502949131a2fd6950fd12b1dd709c38c0e40349eb0175e6556eada367cefed762f8d29d1fa335095509bb6d474286bf13c0ede15de3f9d12cb10ee4a1196b51e7d358ed4fbf0e42b21a1b1e8edd6a5ace5942e6e87cbd0a3583f957c9982889307b3bcb4a3b7f7a31c5c9058aee636aaa1c10b50537d20376b4e2780b196840ed2ca847dd2b906e19e8c69785970f4ad4cc784a638f0b1f9ca8baf38768b6439d68654dbef0ef3402cf3804a81415f25f6dcc282ae6d2cffd7de813d84fc069630e99853fc3b4989640def735e32d091e3997d3be&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.868716215Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/d, Name: checkpoints/2560/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2560/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1c65a0334486c931902eef8105a907bb0ce0cf5b02ae4b7311e18725a0c56be909aa8c7b209a8ebff2ccc7b41fdb2a220249292b27211808b070b9ae5a8fdca103ea0d21dcd605165b7efe5c2aad8ae0ba12318988a9612cf0bd99202baf09ddd72f12f09635fa896cedd9295cddc7b25c23333ff6ee91e8fda1c1058efbff4828045f8ef46e9e47843d4ae4bebe5d6179f5f2a2cbb7e8ddee7f743b11d1c1b20ab1ed8046f50296a5fb74ddb8fcb7d839f9a210e4116f43addf8ed91cfbc163b380be0ae7b938e342116607c29674003a717fe83b256195754cb8c5735c0c1915465853124d5f86580b5dfe3413f0c2877c9e4feb9e673873651acc1d179150&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869661534Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/array_metadatas, Name: checkpoints/64/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/64/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=07e6ab552b26b4d82b95e40ea1f93eb83e061de2eeb6e7631e09cf7aef8c9c51ce841d500ee50c6a0e7c293dbca03fcefd0f62773e4c4d297e30d09053c27b4d59d7317fc3761eed754d0ad8b6fbe2c8c18ff6bc03db979d47a17ba18d90d70a448b455c15ae3d0a48cad3924aa763c52d890830a9031e421b9918aa7de34b60a07ecc4218c375090aa79f073253558e5b7f6f55c44489426d468815635800ba24d72a5285054537758b2f585dda0bdb6f7bf915e3055237ba25122703306e7e9d7783b9505701f1415bdbad1caba2168a9e9319657d77ce8f5fbf383912eafc8d2c351de0d450300a9e0d1837397b11311c6c55bbe8a38b0188388bf37bd463&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869681547Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/d, Name: checkpoints/3328/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3328/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1aa72487cd1c140b190a7c425535748b7d48b7d4a882f038355e018f9cce3e3caeee0743da6887cbe68d8d5bbb3b579f88db1051d3b8ed2fd387a5792eaf4baf5b6e07d8337ad630f1bfde67b9e5e1a3521582658fb60b2f7674a46b57ea0dee8e4aa9012cc7ffaec43a0e37e2d1b67535f2f2ddb976595043d200bb373e8188cea3e1f8c7edbc23944f53561ee41ef39b189c1ed73abea7a6f5844fbbcf2b75f128a4c0bef5d7a99a692418e46b0a193d99206a5aa669ce6f882255b2cb8d623e1b49ccffca48f6edc16388d24d4d021ae382fd1c32b29af0f984769fc632a7955a3ab03ae70d7f6ce57e6f24afb4072fa436f1714f42db5e303947e709259c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.86966897Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0, Name: checkpoints/2432/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2432/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=62a4eebdc6d25e210ff61458932c448d1d247ab84774ce198547717f03c28f4c35ade1eec519ecde7a58ad08ac4b72b762aeae8d78b82038fc3993fb755387ada8fba38657e25ab27dfb3809dc0cf9849377ee444953e44035ea46167752c44170c816ca233aa45b7a22098dd78053eddfcfd2899db3d2d2c6e2a9dbc80fdfbc31f4683547a78767860ddd77112411b6274ab291a2a8f84bb48c8daf393cf32d50d830201842fb09b8f5d1b9e960e52137b556508914c315ec5b39e415ecc9c681a582f3b7bedfae1fa5169800e8c90662865e9471a0e313e2f2579c21c18116750bd02c14c301c2cdc255e8e0549399d1435fe11aa9f489359ffe825ad1a060&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869834736Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880, Name: checkpoints/2880, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2880?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2a9da4fc0280a7b808ecdf2ad0e88d22ee219c8158c953b9106d6ad61ace73c7a6b91841460951cceb16784cae9d488a9ff41be392afe7ec8f38b49540399525cf3873f01c5979f41541a885993107d0e4ac6486ff20cc4b7d7f7ce09ec8ec8eb35e5975b1a921abcf98a840b13fa8ff29a5952626a32659cc9230f4197a67bd0af843c6f17d89d032b06613fc749b1ad8c4e5e35c8d0a57d6bd4592c9a77147fefd61c9163e76b2f5c0975e243b39d6ba9c7a5169db0bd6d1a8a8913174a7eea33aebcf2ec73d85817f02a6faba77fbf09d345f8b64dd102e642931ee0e67e321c7fa04a2d67b4781c4dd6ef9505adea3bdcaec61fa24b33be0ec607f5a6874&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869505876Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216, Name: checkpoints/1216, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1216?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=5366677e3e6f75361908d83fe279445624c4244f9a8b8463a37f5f25e7868e3b33347b53fd4eeec93cf8126aa7569230ad793ae22b42797d3551e036969eacec5adb93cb2f9ad53c76019e6c0ceb2dacd74b65dd51549418b398911942a4e5dc1ce8f697be0220d1911e4cc086153db559a62ed62d02f9f538f1574bf1325a49399d3efbe14f80bed970aee6daaab845752ebdf5604413269c6f62da1560ca6d4008323f5505138eaf05e0928feb6e284b6958edbd6b506b0ec12999440cc56572a6c8cf688c9ab399100a3b906805dde30cefaa14b28dd5ec9183719f25c3ef5dcba85edb7a3f4a92223ff15d33d06ad6dc6eda2cac8a3fb3d8e9e4a55cf84f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869927571Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/d, Name: checkpoints/4288/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4288/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=3aae52a59ec9af3b8cb88dffddb639dfc3ac24d3c539b5092b1b1f785c5c3572b10daa6fc15cafc4556f895405ce4ef394d5f5a91711e7fb34bb2e13e4b13c0e43f7b0a8a4dd549859a2feb764f5361ce37cef709d558ec84cfd3a49437d6a9cf1d4459c1661c980ca1a82f152ca1eb7e364ded153c353408ad913cbd946d6e96fee5268151fb333b530ab0dda9baa001a388a3509d0546f53cf14ec0d5cd7e4a670cc56e1a4f4d53b4eef39f3c89521c0820012ec0fde2f4307566a1d1b64efe7ff3857ff5cb8db1ded77f50a05a7cdbd68061f13136d430b84bc6cd804bf346043e8d82bbdf08d74f5094df486e13b6ebdb0e39a2c9cf278988e10f282453d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869937617Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d, Name: checkpoints/5504/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5504/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b91aaa3ce7e1de1c9d8f74df82c42ca19bf7806f596251d4d4024f9cdefca51484cc10c62b24e7cf7a49a13f2de725e308219059679cae65cdaf8954457ac74a7313ec49693c81a3562156a4b83b3f48734c51b95af744ab41042d0c8354d69e171597cf332f87c70e4600c6dcfc366c92e1b2133faee8c575cdc0ed030cbe05078f957bcf2f67fc1d0e539de9678f88ccca2a125f803c31646ce92a0256e595130201fe6bc63ab62564707ebf63f2599707c6d0341b9a165cc4a22947283beccc5d40270451243c12e7b4c46463cfbbd3cbc576eacf454fa8c7ffb6063a69f9773a5f22b45d2d8090a2b19da0e0e1972c27809a6ce1633f50102ed8b4eeebca&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869714625Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0, Name: checkpoints/4992/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4992/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=62189d3cd063b4fd4696003cb677bab2b2495fcd0985fc67c0217ebff31f1131db5b023f02d39acf7f35203209f860870ecdcc3f12564bbd03cb1304fefb6c999621c5eb4b4537aa855170f83a653b77fb0c0fb38f498ff78310d3a92825a779297fbb5056b7b2bcfbe207380bda603efb484daffb33d6e8f1742c6d77bad5d02b741fdf6c6ff2c12274a17eb0f7a8ccb0513f07ee300c41a8de5e3f89aa5f2e9cf304e2d747ff0eff843f6f946ae0d272d9f39201163289b4bd885b37dbbc1af4c5d277727c39caf9ed12ebe5b54cf630ef73971d4838e6c1a5271ebb59d509ef9777483bda276a7c7f0c4eb4db4d3187ebcc0c66ff2735df548c85efd6c86f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870035245Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0, Name: checkpoints/4288/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4288/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=80b153635b63a6af04998f8ad3d2bb313040a7bbfbc14198836e1f17f4ac9ca2338005f137891f97313350e60ff3524a8382fc7d6df759f199d4993675030673f0233b5180e6fef514717b8addde60b7b8c7d4d78cbe40b49a00bad8ec7dc1ec7a2deed0536acc5d121d8ae691b56381a277b7ecab8c326765d6091181a2797b1fe87a6b31a0a4ab25283ddaea4516080bc80aa4fcf187d178ea8ee8b3910c4223336d40714fdc8298030f62a70a4304adecb6292efb4985711d1ab2208755ceabe833d57bfda8d5de6d3af474cac75ac11d38d6e5ee1889df5cbe4393401f27011bcba9b467c1cd1fd9ed58f313b17b048bc731a2056b29d45a963d3457be8d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870217268Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d, Name: checkpoints/5056/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5056/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=384dca2611ebdde41060e1c687eca9b35d796fa867e28bd92e803c2a6b3f0d0a4d16c6950068afc2a96b90401784dae890457edc6f20998221bf9a3f870a24193fcc465e2d75cbf343bbc9e9fa0ad6471ae25cf9aea1d617ed707205f3d85b447ce066a9043ef49f5e632e70b2fdb66d2aa80f733a860751a22d3a07565c1ef859973f68c342f5e31d05c63a93f4ed925baf9891731e43c97c3d7170883ccedee19b7b9b736d0ba54c7d5f02840abb7e4c8a867e4adc9927e93776d7805f8909b257a571d105085896528218dbc4c5c661cfdc1479d5b4f691d7f8c08e8d046c92d2cde3e3c36770254cff61fcb023e0a0449b0dc341af14da49395f5ff98171&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.868846432Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d, Name: checkpoints/832/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/832/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=143d0896bac3bf56f0eb4da4128dc48215d33d2308fd3219503962fd9d3617c2988be7cd7ee01bde778df519e82ada42967eb88725f8f825d7f9812c8622e5161b611407f6cfde456bc1b7e3ff07ba0a47c0b1462171f56d178b6527fd1a489a44f8d2a489dff7e1c005f874df06e3d378021e27666c054394a57f02dbd20e9fbe6670f9fc4d391c67ae3b2ebf8e35a7d1545689fc5f6bfb1ee00b230ec68e9a2878e5121faf94a5026eb97673bbb48d90e77533a1dfe4bd92d4a7a38e2772c067fb189f86d42a792db689c2a874c69a273be3bf2357b04877df6b94a7378f7bf20533467d4ba33ff44ac5f735e5b9c0d0d486df940897408e42757b306974a9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870308159Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default, Name: checkpoints/5632/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5632/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=79ef5945799e4307e6e3b10836d6ad82a89669724e5f89153bb337c5ab5ae4a9d0c641684e046539b21d47318a109b1ad47585f5c031e469cc3ace825dcfcefaa511027c2096922a8be547c67c73adb5f2221e10d2faf46257d80e57d7c9ff40cc043caf57c2d6ed2ddb4514c2deabc12e28e918fec22a1a627ca60cd1e708f0e500d980e3f6ccf8cef35144279b5c7e81d715a5467ebf2bb49de9f0ef2313ea45091bc347e2c4c41c9dd5019f8111ca02aebb6cf1142c397a7ab0e17f166cde70f80ad63952d1a3eb6adc3d5c53d272f15295bd922c1b0005cb0baf9f5ae14024f6899d57b4590229482b2dae45cab1b7423558e70892dfd382130b722170e6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870348786Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384, Name: checkpoints/384, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/384?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b78cb8b2ce2968dba97834b70f7473d906acfc31afb8057bd106611732d66fa912e0becf47938c006992e384b315d4977e05bc8f1bc6aae8fc04d5427b67339960075aa71e41898a260bd1881154c44e8bed46a2f91ab4d163d651cf597807cca6c18b3995412890f5d7ea7d1e65312e9c079c23585abe2ecac099bbc3cc188b2174e2756e2525092f09bc5a86e5f43e3c0791e0809729d5252f92bca40714ccea8af7e9bed2b92789612a7f3b2a5651801c2d250061382832b6b260273e123d522709bced0c1026175b08a0b65768e2b420d989e6b410237cc5d28b567de29a33153d090381717505d721a6904d1e58501f2d05765236528c41357af33106c7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.868860014Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0, Name: checkpoints/1408/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1408/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=c4f02fbfdd88d7f1435022071c652d55be3a69f24c6ffb2483e20e9cb727bd47940b8aa7fa72d03446c775b26e412c47dbba10b39434101eb6a6855b616a2deee10ca58c6101876037adfa571be5064da50aecf7bda85864588804a860a3216e26a86d31302b4ee78a0324001d627bd56459d13459e275544027b446dc8921b1e0d158c4808e0f6d8ed1f9f3a7c9f47dbedeb6ff952d5345c97fbbef45bd0aa0151fc61c9bdb219cd8eb953796271feb477f797b152aa13c1955a84f52ff5267d9360f092869711f6eedd77625dd68f218079770d29441e098df0462a8ba6767f3393c0cb6c52896367c5e924763942f1d88a413f8ac9d07142c33592ecf2b15&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870372461Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d, Name: checkpoints/5440/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5440/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=232a6bc361db93c97d65d60af99bb968ada221d8cbbd2f94a97126452db587a0805b9e8eff328962fc299a58cf6549bf3ec0bc511acc9e65fc81b3891e0541076ec438df307573b33488afc66db49cc9f538018a7a9376cc9b31451ac9a592ee72d92da93e56b3d003a8f56e74fe4c6168fb07596c201c08b898698d700890b60be31c291876980de75c810191c7931be9e75110ed89df065d28315fb8ab1bb7e761a4cc71a9b203850f11b7cee9e9af03ee145fbee685eff1b573a997120c494fb9c5ffa2613ccdbabb6da15cd300241716b2faaceff5fc1d84b62522f17e078a391665b1224450643a4c237bb7dadc4e0fd45da652de307fe92758111e6c72&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869044416Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968, Name: checkpoints/3968, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3968?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=44193dcc3768c12913eb9e171b02fddcc2d8e4e3836a5501ad4ee09e2a748568f098cd3461c5474690f0c1ba4cc5cc7f9c9e16838fa63fd5105d3fe06d56989eba1d9413ec45f9f031adea9299a295e24251632b22f33de5a38efa1732a20133fbff4dba14b0d7701d8557d392c171823f7f3a27ac5e88c98c9a706a167662eb911e65c6a0e53804e17eb729d50dd94aa4e1bb18da04057f57938d8803a9f54bc410a87411ca6fe2da556070dff5e52b10b5d1eeb1627cf20dd9be6d5d9669ae726cabf06fc52de8b863cae57f4d62e122b68e3935fbf158cfa6a8af123783262fb9445ce0bfe9761bc90e1a556d224f011acb3ce9a1bf745b09afee6d42b1ce&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.868973599Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/d, Name: checkpoints/5632/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5632/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6cba8e654b440ccf1076da6096175a21fe139798f6c3d7444a4378c4f47bc38330324faaa8719541ceb1e01e51e04f5cebb64128db8185f980794eac43ba7d6f6268aba4c299334067effb921074e1b9510edf0f2f1fe92fbb8b2af21c398f294dc92cee5e9641429ae79647898ed1e0c6fa4f7c3cf31e2bff7540bd64bdecf312eaab83605235600cb79cc32bfa749179bdf35e6742212bbadf95de412e2b7ebae04310b1081138e9e49db49dc745abac3930c5a9aa564a1547998fa5841bae9ec29fbfe361e8f812b79b4c8fff4a4c2d670091a76191763a39421ad6b6386be17dc3c9775b038a82c5672636ddf5a2926333218af68199be893adb608a7a31&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869132956Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/d, Name: checkpoints/4224/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4224/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=97e8028b1e0d85c4e3aa4f8a30164fac0201600c1a01199eb03259a581afe6e265f1c9a0a49afce703cddb2fb10ee25d90c8a3cab9138b74cdf6deaccaaf55fdf4b256f61985565b36791f125d6a0d79d2aceead53811556b1ba7f62e2a692401c6ddf9a9b86ef6240af86682dba3cc83cd8798ca7cfe04fa7fffa53267a3361613fef1eacf8577c2624bdc66962a592bcf56372641f05a8b0508900a6b3b128f9f9d1d82a2e6b9c91841018fccf8ac5f1ee2f0e58854b86d6686a6b46bf332e9483d2ac1a7d46f66b9df3f5f5a5c0f1ec7d73d3288ee9b745d2da9b29c80caee17cd50a40edcf5f60389c0392c13e18018cf86058626543245d6621bca925a1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869351207Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520, Name: checkpoints/3520, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3520?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=69f9446c5e4c47d4aabc96a1e61c95c9b76c74ce34c190109f892157eaf1e20dc142dc90533023222d727f91e93ebeddf1472a32c00b38e6da0ccb7be80506d2a35a0addaa64605a70d063934a227858c0fe37d1ea5f49653453cfb54ecb3517cdfcf4096e20ac8ff9a2a22d03f7b337769a048de6d8c445f2805ab953063b613a305fcdb27447acaf969b6e223ddba42f8ba00c137a197f7c460258e0f8b08edce5aff26a9b0edb0407186216c84a29586aefb9641a99ca68128db2209183c28d3c91939f97f85d0548a12356a442a6c6329f10c836ceecf6170b9d62c2f12f212ea385df54a60de70a29c78e78e344d1dd0941902ff3d7a152d45f8ab4c313&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869393126Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default, Name: checkpoints/1728/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1728/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a08a6062377bd59652e4684d845c396cfc916851c8e0edfd8463cdb37658269b8f54e9e08be0a7f9c6757ec8261933fa2515397bdeed66ef90dec13c81b486b1790508d39e67b3271a9b951b0173bdfcb7b019ad3fb85e0d7967f1eb26e99d354b584c12228bfa4ffac2343bceb8a4db3bc9648e9476fe9c61c271e787866d11f3c585c84fa92d2fff8bbbfc2b31b377db27ffa6e13c7e5a60ba28e7cdd93f60d07264b3f98a0bdfe5f2dd6121a2e273e55fe62b6af56ffaa5bf5c6e78f7e16d9217c7d8247264871cf85b55f3a434ef8706c0e6ac84bbbfa64907e46f23fefd061b09ae7f52eb6fc2ebec64c44aa8b76537b2dfe85d0b00d3c470db7b6fbf93&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869271497Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/d, Name: checkpoints/704/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/704/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=03928d0a230febdb09676df2e4a63efdcad39a92e3bd80148c4c059d4f9251e2e7d5e92bb026f2fb003f00e16f3737c8305812df4b132f224e3d4a0c1015aaeec912c7e8b9a50067a8b6ca669b0efbdf4d0f78dd0632213cd0d4c1fb28abbe8978423191f185d364567999a7290a0321f7f16b79b5bc3e900e614a6bccec10cef50bcf13938e313f741f0ad517733eafc74bb0f4bc8340b2b1f1a7c7878874eb4112870827d714a4fd07059a71b6b2fe93159c6f5c71daf6e4e5ea6f3aeebfbfb56419017b9b147a0bd50ca656db3fda5bf9117a2646d7bfc56eaefdbf5c547fb3f9ed078a01fe449bdd6cd2b8dfbf9f2122c7630f3679f858f08ad56900cda9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870482825Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default, Name: checkpoints/0/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/0/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=0b8eeb064a7bc92f492f09789f6362052b4e9e0e7843c20895e061c1ab5b04b9b6aa7085d7037dd446c68dadbc3845acf5c3308a54f2bbc40f6ffb1fcd985ad424ee70b8bc49c7c971c5ede39c0eb7558f21911e92c0f2df3ab6cd14755af76e7df3b19ec0261817a4661ca12ab83758abefe90eced0eda12460c887976e21395888978243e6e3df7b69a0b3207d428209ac11cf21fd9d29fd8e0c4b746a465798a3f94193c6eb9735bccd709c193c5254a3c1dd8eb84fd9efc347aa043413ac5334ed5b7f1518f09a00a2f7d7885a020176afffbf894fe12cfdf6585a8cc2d74da734ace897e8d2d0766931847d19f5f797d54fc55fb09d850316a4342c39ef&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87053173Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0, Name: checkpoints/1600/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1600/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=403c0c540d7a542425bade4b987aed09f5fa09ccadc3d6f021437e4372be713a6306e042226f8d51964f5ae5e87d2a2cd0f447e62ae88a268066cac67272696975028b4ceef09612958d69cd9045551f478e6a765af6c07bf39cca286b7675912ab9c28678ff7f1aa02e67eaa3aa234271cc184f4ef9b3f38986ab6ce6bbf6e0cfecb6ae86482436f2c6d47049acb6445ff0ff3affe7966a5274eedb43b6c7844d4174eb1c71064ff003fe2f9c89c4394e9ac56b4949e18390657dfb089dab6612f282413549ee2bf0a52dda5df8b4aa154770aef5c86d9f94212b6054b97694261627ebb315b011b0e24a6ab67facbad014ad7bc2e4339d5c545cc065257741&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869076599Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d, Name: checkpoints/3584/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3584/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=be4e169a38abbfe5961e4a96e38f6a97d4076793b83c252b24a5ddc2619a58cf11b607caf7e279b3246788a73bc011b90ce95f00a6b1d7438e91c2d6f1644f271ecc4b511ed8351e9684305c143cdf1ee460b2a13de14864bf9d51f6438820246ca374eb8c1fa2e840e1f26216ed2e0da0cd9e0317f01a96bfe4e7bcb7c18ed8d88bfa72453dfbad02b75616ecf787860b8db371bb2c0eac969f2c08c15ea6b28e2005b5f340fd3a1aaab0902b5115ebf2ac61db5898e2fe6f096e62991d1b7528f706aeba3d6a4eb1daff82b82284fb56d05834d31687b1de97e65d42a470f2cd21fcc12b3764593f175530580383bb8153f743499202c90046d361455b13a7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870587157Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default, Name: checkpoints/1984/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1984/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=8f014de4baa135cc6d697a36b67a61cb7ec017ff1bef9b1fc32d2b99380db3fe3a54d781c6187daa67c7d1c9314ba2be0fa59d88e3de096b07440311710be9e0bffa996c5fd7cf9ae65d7847d45b42063657c6728396cb96be735dd795714d364ad47223b721ee79b43ea5a12d190f691f909c24aacbe53b38292b20714854536790f47a89cb17e781fa5dc1cda772b8d0abfbebea5547947d8ac0f5fd2f13ee1b82c8aa21386ad67c297c0b3c49e65ad20c59ee41ec6a1fc433e5b77d1fd6145a7e59fb7e9e9fb3dd7ce93744d177f139b8439f770c1d580e6b5946037f13e8aafcbe8087cf81efc58b54d1f0c3411a8ace600683af6cfbc651bd91d65a0ba1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870623572Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default, Name: checkpoints/4736/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4736/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=74e790478a7e04598a1c700462bacca24e4edd2c2dd0b7df11763523d67fbbbfd835c42118597f087b43d05b4ef51fe0ae99bde5b39eb5405a771984bf5b210392a40aa08833274f3a64e7ee40eeac1096209421e0af8ee17e865b4ae49f4d15ef06fee1ef4eb3d232a508298917c0c47776a500b47f4aa1b5328fbb00331e98475aca322ef7b17bc0358c3b3ee8a39a87a28472eff7bfbfcfebd4ea3d76c462ffaa79181507545c7c31fb78f121c1b83a921f487ff6e45ad87f9e05c90871d711891cd37378c319fd7e5b3f8a8942a484a8dfde74ede4d3906417d0164315476fb97b9de5cea501a847a7495d8287f3e567999c5b31b58ca4bdc5fc151f849a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870653078Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/array_metadatas, Name: checkpoints/4736/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4736/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a8021b11134bee583015492799058b0da164b350036db846d1c6fc8763582f037f6c1d329768b7d9675e633a111f09beff53a650057b8530f3b76bcb732d2c496353d1515666a9a4f946360cd878c519d480b5b2ea728758e7e39d138773340f84a1e366fec3e925a192efa466ca8e7f96400b0701bd0055f9bf11845ea0f7a248ce1f57ab5bfb7a1ea7d5f467392f738aea87f0bf2149fe2738ca43181ebe2bbd3a96013775a4535b847ea8ea799015825945e6f57d8f138f2a33c43c65514f6c8b474e08e8df0071d8934314c609626b3546727a39f9813bc1a36fbb0468e84fbc58dc36b0635ecc35ad0215b5aa6c6181309c5432b6e121342cb2c103920d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869950258Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default, Name: checkpoints/4352/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4352/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=0e8e55298d10f0f377d9a807a497939b6c26f1ee2f408768e96cfd5ce9335cdcff4d8b8676826b2e9dc5b1abd55b757c71ec0bcc988af3d84d4daeb499b7f1a5ae37da23111548fdfcbfff4ef0a09154303fe95ee904327efcb24ba3cba824b701625b755480d814b87b378389d0265d81a7f528da7b1382c277ef29f82b9bb6a6738b944958ea7d8c7869715466e4a050c1872526b18fbb44f2808851a3cc76349b2587620a0bf03b236401983079f885feb79aa6679019d61b3023e9727f492139e32507d070e23f1cf94d05565a14804ec1bd0ce74beab6ace0457c95939fdc73857e2557f591bab6226d645d766007539c61a105442d05749e40ffc0ea52&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870141686Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0, Name: checkpoints/3584/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3584/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=7a2f1b5ac42833a3f27fa63568af98fc857cff09bbe80d48a2b28ee477075d59d319393e5ddd679be3affd02728416f05c98505b3ccb326999e9ed22f16ca652d2e42aa1da851f29d607c588d3a16258ba66aa1d112eae47ae4c158ea0bd3a3b40fd60607fe8648587ccd3ebfe2b9b2f0ffcb10eed754fd78e71decc38dbe2622444f546296f1d380ac5a60841c4c7cacf4d389a0234f6e84be2f08b8f378893f4ed9b06be7e461bbbcd99a84ee55145caa4e4f8210d9ae869669f86af0a2ff123ba52d3bab122709c19d856f2803899222da2d4b79bb03e65a615668de06620fd2e2dab0b7b092b3c86e57d1869fab39e03725936473e0c30536be85b702cdd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869812499Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536, Name: checkpoints/1536, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1536?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=01c36e168afaffb8e7a7f02997adf41d06b7b7dc033a39b26bd3ff1cc888354e98cbaa42e984705ebb9fdf4a3fb4553366b8e3571a3549371d4ae86409f5fb784ce4e961f7fc5e7d3c215786f06d142a4b65a737e8695c1b593662ac23ea7df7f4ba287f4a6f35e19eef79ccd20fe46a25af3f8a25d2edfd622a17e25f3238f5fe3b9a523383508a6affcf3c31f97fb23feef5fe901432a53d16d8c0bf949a0f3bff6069d9e4409168e52f4eb2fcbf53f6ec9c6467f35a6872af1cb15ef3a1716a223deee14b9247faacd13a5f8a87ad921edb1f5cd79e410f57a39243290715188b896f45c4baed99574a3d32a7377e60fa81b62c7878084bd4bf7d9b4a067a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869521161Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200, Name: checkpoints/3200, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3200?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=7b6f4c86858cfb8e5e0f839b32605afc8b8226bde3118763f9fc9300e4bee7fb3f6a531fb8b61bbead37082c4838b0334a337d85433b41efbaa105e01843624af008b30fc3995e44641cd1d87df24376b0ffe5b8d7eb63c2dcc3eb85566e5c1d6aae676f2e2267bddba085be23349f63782efddc246ed05ebce2ad081a8e4a5d5280215dd404e6e5bf0c67c1f5a8410e2cb0904a4203bb60ade435a1e135d14c09bddef0665215e2d3e0a8fba488f9d6e7333c77006fecbb2f382f23b020fa0b1c0ee63a5209d4781317e5be44dd5eb06f60a388c93c3527db13e21740bea50ceea28a6e5a5d93e2bc3fd0b42f60a989726a5682927e4cf4dbf25a4b948b5616&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870762557Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0, Name: checkpoints/1152/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1152/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=384087390c02209a45d8d74d04f4dfa238977aed8da548a212e17cf2ff885e8b36502feef01e087eddf298ae28d6e7137cd1742331845e3c0467e12025831b21a9ec2bc7c2b209d0838d5872759f17bf5f41d96a423ef99f02231891bb46eb32b937388a51111561b83f7ed491d57076a31532541769100e5a21aa32dd4e20d8786ce80bbfa2321fffbf74a59c64249d1752e844e56031701099a360a71b9dbf855f14ec345f0c3395929b4df7b3e56d6a7a5e01909cb97e9b1289a035e56aebd5c9b5f614590a7864a71d8d107f176fdddcd2f50505a93f9add619465a161198b2edc9b4db3bb829370a7dac260d2ea98ffd18f50da3ef383dbd080f1b67e0c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.871984726Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0, Name: checkpoints/5824/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5824/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=412f81a605947fab5fd26496811cf35826dab14f925189ccc27233624eeb2e8f998d641510551a02317e227deee45e98545b1f866725c741404c3c18a695a9807fb297633e05619f170dc402f417db751661d5c8fd0511052a9b134d163e62fad8bf55d1b9555aae5f67390b8628c7ae9177846aeb62a42c8072eae97ef3df739bda19494f1fca1b0a2ca9f41aca3ed1298ca2fff765ac800b76ad9b656c2085626c13d36f1aba82ed6495861f306fe5d49aa533207ade9c135aefcd6d7332eb7a16d0d7e65dccb17dd99fa03ab101af4ff72dc0497ca690108a6a0f5b3bb1b8832454207972ef4a727a9b06997f73826bb2a294b0e355607b92d0daa593943d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872084808Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d, Name: checkpoints/1792/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1792/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9f78011dad969979ba971d3c8775e2ed5fd0c60734ad24ade7a6f5b62042b31dbe466d5cdd9bb1fc2af043ab223592509b1e862470101d90db175d211c3e489d22fd978b94259cd64f809e9fae2f31759d553b1b62b989e966838fe93fb1bb43b7f1f6bd9f0e41eb19c85c62f96d9082b1fc903489262cbdab0fc2ba4fd8dc3cd5b2221d4f67e7a685afb271e6f53bffc88b77b4338895bb14e5fe66679921e656a426008dd6fbc3197f431edd762e0064e475bdf395c4e470fd10c2a7020179fbf13cc6bad43c0c64a17a49b4a17f003e4571857c7664e566bb81393aa4f5f87ec8f755fddcbbed7af1d8a0e2a2c2a052e73db50e0594b0dc62308f70fee078&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872096019Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/array_metadatas, Name: checkpoints/4352/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4352/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=74ff95ac2b32d477c853552af197cac8dbe7ad6846df001da8afcbe6a3c404bc32ba6638d76d5ce35af6f863f4ec5da4ccab0682b38709ab576212c03951d39e1a11a38e21e9a8c4696b814809dce408ef44f557bfe762cd45b07b8fee199b2afbd1a2b2c74628ce41430e4035c869ea2a5020d9fb466e2ba4e376f9da1dc025eba68721a38c49ee93456a1572489c9051125ddcb20dbb3c9abc76f14043a9f5446e72029ba9675e31395afa54da95dc4eaf286be419eb1c9eeee8d6eee90dc018f3b3063af95e401b8e878d06742abd73d1554178b1ec21923336c122b4b6b541ce643024d4f6e24743decf49dc3857046aff4f169c5ec2cb06fe403702bb65&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872024774Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d, Name: checkpoints/3456/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3456/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=07801dc03777384054e4d4e3e64ffaf6e8c4572c7b37fda11ba0485b7a4d1b86464ef36078bb435fbb0fa98d676dac4683623f60161b9a640c0ac17a4bf8f97bb922185f485590a475948c677423eb0f7d60b0cf3b5261e9af48ff87839e7f3faecb6ec5e1b2c74d05b5cf7ff28ecdf7d37df2ca3b0140a134c4463ce140fc84e4eca7cdbfa32d3cf7cc1941e76134e909abcb305e40fd0caa3d1f40f7f137a34d5a66f3f5292e25faead717cc637492c4ed4b4b13542feed69f376d4f4b1f3e5db3e7d6b1bf720434e51985c90506cd1ae54d9b2c7efa54b42ba07605d0250f88950555fd0ccf7f001497ac24750339cbf5152dc11a8253f334ef3bd5a79697&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872168821Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default, Name: checkpoints/3584/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3584/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=854bb99136a18545b7d01feced32b1cdc517eee05a30d72fa95dc722727cc6bb88c0474ed06841722098f4283ce385b25ddf8df24ffa4f265d33007b6c78f307da33d61261cb6589951808b5b27cca9825c7f19b452901a81ea837b7dad3c80e77939ea10660eb848fe443c5acae393de1e6a90c302d7963c31d224509fd0cb9f207ac5de1ab3b975bf5ec28801ababe70da3cbbe93599b4eb6166a74727200cf52ddb82ca7218a01496a2aba18d83648b0ce0a42e2c4ccc72909efb09ec7c83516dabe5292d8e699b7f2bf52b0f22d28e502f683c2f1cf6ac1aec019832a3c566a0479b7eae4fbc422e6e56cbef64ec100411b05549334ba697983ddd801ca2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869411732Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/array_metadatas, Name: checkpoints/4160/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4160/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=16361db8877fb91aa01de3dd924624c9a3d63125b5b1dfa40351aec871a2951fa34e21f525e0ad882e374fe65dc4847eca202849cb879026cec7f46f65e9d90fa705aaed139f97d51f0281996940482c636dd3843f2615d7f6d3f91a750c383b2f15a2e645420f4ffad5cfddca0c63b2bc5cc8ec0da01f7c1e0bda3bb8c9ac71abc01e1c564d9d486205e06f810fc5cbff586853e0c0770b415dead68f50552964888a165bc11707e02c44deff4423158b049b0684a7a6141e4ed2bf207df4a87c5e44070eadcedfe3d00d403364a858019b0abf4eb9f1285f7f614a2c71f28aa06ff8b94ff0fc1174dbbf7813f73934251f37982f10a281395290bb9bfedc6b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869504156Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default, Name: checkpoints/1152/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1152/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a3581459fdf097850a436aeace2190f6f3d1bea08708a52e669ac69933629fc19d84b4827372e5d80d7cf316a4bfd1f5b03bf2fb1fe0793b4f983bb9025c2a545c43c5e08ab9dd692926d5fc20051d8c21293375f110f6722bd3156eaa8e7d9e37f81fb52fea69fd79f4f602d0e88304687f08823034e93b6b6eaf69785937568e07d11bcf52f88bc9b73e9eb66bda089d844d64b43fce940466ff33d399fa7ccb143f33c07c1748a0c4d37be0c39bd9e91890f4a468a7e5d630b8464858cf976ed19078fa2275efc560637f7b66e649ea8e91161a62331e28db7679d6ddaaf693879db6a0c982d1a839a4d3dc0cf5383bbb5f7585241b2cf0d7949fdc1a452b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869476255Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320, Name: checkpoints/320, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/320?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a6dacb9a0b5baa00908bbf1f12abae8b3283d5b9fd531383ce6c56643e808751067d634945b26f3d316e7b5a16853b2617c0465f9a94eef33d29774315613aa649e2827e7d997077b2c10b9274c08597f5e0d0a1c3cf25e3065233b250d4cb060f363a7c89a234509682b32e713f086553dbcbf49db1df5e25524dd04c5277b9df2d6b09b48e4f3b4ea1c5c90967c31b1af2d6f10d72ad425180c90c5db5f96c1f8a150a2c6ba6312644c7e6c52519b37441463ae4cdb054fdacd910b29df6132dc06d9701c3993a10cb5ea910e0d0959977b5fd9783504d3e19747970b445f5e9ce9ee56af1919858bebfc92f7f0ed84c47b7b570f91dda09896a985c8740b3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87222911Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/d, Name: checkpoints/1664/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1664/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=3d442b3f5cbf07cbbd4d7590bb38f26bb1c5916cba90a062065338b7a023c6b6679d4e8f40dda0b30bdb768bf0a5098143d8bcc6a3e6cf41804ceb9afc27766667ee9540522c109414f386222a6270101a006c813bb1a55cff95bfdc55e740464311944ec5009b19bca329244b2dd27073120e0f0d675bd0c8941b47b7c6ca2fc75442ae51f2df87e4eb0a88aac53beae2eac7613e936c46610523957ac0672b15b72c4348e3cd91254075c63f92c9a3e38bd729f0e66e698b14aac741a842dcdb1c304ae364683f194009eeb04475ae652ac5f36e93b8e37ff920cbffc0fbdf02f71b53cc3d9e54b86209d19f31be868deff842b5ac1c4f886c46756ed3e6e5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87226246Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default, Name: checkpoints/2944/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2944/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=04ab901cd63e502bdf74cb1cb986020765056c8ce5ea127c917f2fbea847882461920de2c7a5859a04dc5f718a78f5c1adefecf2163ec73632002e238a9c9d75357b79b20e891456c1a089c48e4dc8357fda3ef7477409967cfd44c2c678c66d856e0cfb5f4f295c8dd529379545e8116859bd7363ec5985479e29fd06933147513c030ed7e225cb3a0647969fbc7d941ebee6f7ffbf88869cad6513b7bf8b55ee1e4a540f72330b6ceafbc6a5ad8d98ea48622cc944c05917fae9da440f325ebb9fad24340d4f22ea8ea88954f6abe03bdcb7cd8bf681d4415d0cdb22bfc81de106c77a9f4d9890c6a28f1db22be0704b1a7f08b1c72f13bcd20af628a0a0e2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869548582Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/d, Name: checkpoints/896/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/896/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=14d5becd3b4066f4434546e635d70c2fa4ca3551037a406414cc5161b9bc17a49e6bc8dc6f18d3294493b7e290d3eaa7a0c72a55d94036780c1555197da79b957e35ea3a49ef49121b522c05a54d9ccc579715973875d80e10d6b1a585d806dcca396af1b687bff3365d48bcb97d6919c7845e9077292524c7847e2a39be5c751d054e9a44253082d4c1f199288b81fe44a47bc8004dfc10d4fbd6272168e1001e9919f9ee8b097224c86f086912371b328df4749e727375961b05cb22feb911c92c635f7c88f064e9b71a701ab9957a6de29af79b0ee5166ff53519f29dd1ba7db296e58b74032c951b17e68805bb32b212560814807357bc394a381cc5ef2f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870632264Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/d, Name: checkpoints/3648/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3648/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=136cba2baec9b0cb8009de6415fe85908bcf839a44c36f62c50ec8d1c3e9c2d6ec777eb93fe86ad214c1a4c6720f5bb457b38a998897c43bb2c18b1e6a2399e6a8af805960ff75d568bdecfab28829c590462d6ca321346af59064295a79d20efe44b54019e9cdebc03c19cfb852b00507c9911f4cdf7f64c9acb2a9dde854ea1cd97f16e4153ec2c4eb38aa7b0a7abe63172c1e94f32023be5df72b4a1d343d9e08a795cdff42a54db8fde0e68a125ce0c4af02d1091ea8e865e86a3417bcff3710957d4123462ed8959eee23353ec7397f171a6d4db73509d698b8a6ff093f4f40f881feeb5d4310c8a71687264a6a3b016d7e37f81a4d62e28b76b55b10f1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869287196Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/d, Name: checkpoints/3904/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3904/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=5714a1a9876a06becc3becd541b36bbf8bbe90b0078df98f588e76057120f7d423fd96054c63830b287bdaec3ea24da0b82e65d3b1ef0e9277530d328f4cd8b975dd246d52896948185754361148b39ac9b9a5547ea635e90dc7b0385f47312273f492a4f2f34a40950915626b4d6429d8d3190699230d924c348abf24845bdcb09aa06b5f0c7da9e88c1db4ca6232e2dec1d54228d6e071c6aafb48894f67445629e75979d447b5db503803df75111882c7dccfbe996aa7cc0acc71e764ccb16900495e5123b865735d6321dd5cd6e9b3d77d0cca502950a3b0f927a45aa83efaa7f7acb25e094df6ea5ee99897755c3b90d0a34377ec39c6c029179e633c45&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869253376Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default, Name: checkpoints/3008/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3008/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=503c1b8e63d16e02db8ef9a03a053d86c1a08398aec727246f44fa5b89f141944d32e0ff5968e8f2d056ab373da9037f0202f5ec05722495dd3f7dcb78423cdafc285fe2dd3865b2868cafc26cec3ff604f3c876cdbfe2e65b6cba089284a1432a2835109f934974e359d1fab42ca994506b706ce7413dca343356e6f7bbbddcac77df971a48377a1bfda340a694c19d92f1ca1f935ade92804353bfdf74e9b0a6ce12d2ee5003c10ab526fbce9ce5c8f54ef4d402d521fbacda0caf7d977a1338af2b13a22a32cc2674352adb4d2cf126d700ad0f50734232b87b294466661d9b33c7eea50fecc708e49e73342c270056994140fa771d108e14264aeaaceadf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870745686Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/array_metadatas, Name: checkpoints/5248/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5248/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=7ade2515a396dc43867933974d378500c119ecc967898085f6fc2ec7d53e64e60fd82ecdfe37efdceb1c4d5d817ab3fdd22529d3b4d58afb3a6aefcbab78fd9bd5ec8a4d0fa1919c47e8a028f6642ae1fc19218d34e6d324c87afb1f7b6f303b7e5527d24f69fd8f3f28731c129c5283e4ea21d176608bfb5ab272b97d95bd259a6adcc718f679a5e34183c64c35ef9d3943b7ed59f8858cbe8d5a820d69e8ddaa1b0ab1eed64fb36ec74cc1c75305808b48f36be9f3b9e0d880c61f3c1f0d34a1fb2a4ba9be43cb38217829a872374322a623936c8d2e6ccbee9ae00d0e9eaeaad08f3b784cbcfd50101517f604bff7e081fd1055bc9e0ccfed0d5d1df0593b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.868974801Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default, Name: checkpoints/2816/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2816/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6941cb0a6405fe634c43e07d787d757e8b2ca45f94fb75a32f659f59ae77fe91ec7f44aaa8ca08bca6ebb95d15d8bafe2830bb9de1ef81307a1f699ad8e1b23e02fe69f40fb3db49eda2e1a47da15588e406db45bc421f4e5e593a78857279879e4742cf862657b4e655d032bc2f92fe83b99cf7b1302d25409cc4559237b17c6734cfa1ee01e4268d5377b4d257a9b962c48ce1eda7484c605530d000a3536755a2cb168e181a4ca8261249fe3755284f9722665ea1afc1a64a8efe740ed63685569bc0f0a542e001c6a100975a024b89554d13979c8e8bd55e6c7d76399d7898e0ad4d0486f75c870877725c4282ffc545c802fc74a806a8ef7009a9768122&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869729409Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0, Name: checkpoints/3840/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3840/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a636537e73148ebf4eb4539d51d26489584b50340bf1f5c2bd10ce9983e53181c5c58aa859bc75934b29b9fe575facfc0a9ddbc31471873e118a9170860c97af0eb494a1ba537bd83e2c041f93ee354b313b01395748733625e2a17b0925c25328434b80d6e93c1336633f6df03174c54bcce113a68177dc2532761ed1ed92635ca576e824c0c5e8170ce23d734e2ffe55cabb496f5098b282d7e78caefc830ba0335b10aecb0e1d2616a0ab840683fbaccbacc4aeae56f41ed17050a74f77bb2bad70b66dde5953c1e9af5ddabe706c51c658226dff35a05a3a0e5f62253d8df16a341f78b6bcaa368a1baa191ae8465f02f4f39c310632521be5707bb152f6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870346216Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056, Name: checkpoints/5056, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5056?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=258a572f7b9a089f4999a3d9474551b0f641ebedba48600e4bb23255e271528f2398401c3b2f2cf7a36c5eab4eb4a7c2a4905c579b64c37ac30265705e4a7f5c8c87b9aa436df186c9d6797c7b7bf8e5ebc686d4071d250a588432da25e01a09c8e03718a947e795b84105cf5f9f09e8c8e5203abb0b516765f6bdd81bc91232f4b307572e9d0b207a1028312d07b25b4de1f9250de2cffd1ccac2c04bada82d852546d4e9363c715d36aa17d8fb6eaa845409891ab935641ef420538cce20989548e75b71e64609fcb9d744252e9bef2485dff71b5d72d69b9b809c8cb67894767af5213ddc78343fe61f62235c12374055969dd3cac7572656c8c10e8b0c3d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870449137Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default, Name: checkpoints/1792/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1792/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=621cc46485d75a967cc3694fbfb485caa2cce1c9f64d0ce21de94471282608188fedb81b150fe9e73e6c898e49e9ab771b957614142d78f88f3eef72c2c55c5df96ee037710fbac8ac26dbb030979f8756460fe5c6dbea3ec09b5721294ad6bb8c33fb63c66820fc6ce2386b7907ef5109615545e31f8cfb492dd96816cd3d54c3ab3cc718cbbf0da8c153d8f22151f11008894a5fc8d2070e3427a4768156eea7054b960d93f04343316f8fbf5b3cf349ece6694102c5677a64f5b783a86ddb2ca7be3c826fecffd7b8c4d0d0ce8d4f8fad543e874e7f726720d80eb8f95409057bfc93712ed4b5ce47243c1175dd4d737abe66e32ad80d73384c909edfde71&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870494977Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/d, Name: checkpoints/5952/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5952/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9045045d5a1e719380467da6f18aea28d4dd96b54b8e809029f3e69999810e8fa842074af7cbe78122b67fd66f8d7786c3e5f8091f19d430d45b057ecfcc46b9e7f9134b8cbf5b3c1cb7d85fc1ac71d7cdd37e487d5c670f80bf819fdf86dcdeea16237e33fbe5315502a1d78a1504a49f4a855318c6220516c6efaf9fbb689ae8a885bf79eccbd6b873b1588df843df5d4b3fefb7ecfb8e12cb60f10ad0c25c43af0375d9be26401ea5a272f0960296d66defc7b1ddf9e5cf94e709b0342d644d32af1249b375d9febd8cc9eaa36af3cbd9dbf6eda7f1c9faef0fa1a45e04f2fba74b3663b7a2f0dfd71d9fcbbf1067829fe2d07dffcaa68e439c7de3ea72e4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870525339Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default, Name: checkpoints/2048/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2048/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6bf0b4a3feaaf74c225dbd862bc8d305432196851e3f9f1226dff4d284221a3c10c016c06ab7244af131dc7b7555bd45938285135c5f276967d57ed31433a4de0d3805f40afabcd7a63bd2ca1ccdbdd3bcd8c19d1b051a6172437aa210d1fc95a4add24b413a5e50c00c07e0c54b247844ac1147092cca628e987b59f93f13b91cc08b111a8ca7f739e8c0d20b736662dae93f3e26801e92eaea3e09b00e1928b49a37d0e94b4b03aafb319b5b5a0a1c1693da4568ae101b170262cf1b95417aecb64ee9a497d354b97e6abf76326781ac9d68b942dbc255c5072da6617d61d0ba802f92166183c73d6a2a926a5af819962c3d03e072162b1f07fac72d3651cd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870460498Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/array_metadatas, Name: checkpoints/3456/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3456/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a989098e519e71173fd9cc2c5260620ae3d159196b1c8eef6bed4d3c01ce02b6bc3c3284e296dc590de0dc4b0561cc0828cca4c338fffe6c802d25cf82fc96a224642cc4e5a47d674419f0b503cd74c2b9ff2679d241b820efd61586cfb3b4e6c090208250d18f0f787b2e36077013862a77db71a3353b38d9db8c86bcfbae8a862e724f0d6712c1ae50894d00a052d8ad0dba01ce047d9ea3d3a2666b7f77162955aff7ac6cd145c1b9209bf48f4786f8d9543464854ace12db9505c51b241a649aa207bdea6a4d8fb90bf5bea152f6a47d8bc3da9874676239cbdfe64a641eace094ea540dc7b206a8da5b71ea7f6389866201a325708e7390accf6ee68e2f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870400067Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default, Name: checkpoints/2752/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2752/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=254f7c539a1d1f6d67250630d43eeb018be3835b82bdd54745f4e2ce821d1da75a3063d636c6ab98b74b660d24fc6e9ece3222d1659a0f80f0dd79e8083bad94030bcb10a2ee44b9afafce28c39c467287f4a5a45cb04c6c0c6bd8dbd7ea44b00f281d5c366c9fc983e9b67cb37a1d4572c1ff8c1a95c504cf873d1cdda3c71d563ae31f8c4dbd983711046841686340b47d2ce069a1e49df0d9f255b025f731c6bf154bc15c4e3ac722d44e6dce81f9f787d4cbd07f379af82daabd7a93f7f201513de7da28bed203a0c2e83bbbce3c7698772db7cc446abdf2cec78220fa7c8cc127433b7933dfa170c350159c796f5d1da00411eb2a773b83ec81c67969bf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870384102Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/array_metadatas, Name: checkpoints/640/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/640/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1d5ec9036b8ee8e34974fa0b56d361dd1be87c22a9dd5d8dafb8385f85bf4cb63473811a7203de6cfad1e76ad786251d0e2ddcad4f09c542cfa1b0bdaa2537160f1646d878d0378d21626310d302d5bda7611a9b95da9d1c08a53b28ba2378566cf662fc6772ca236136d8bbbc16c9f5f63c20f2f4c79be1c97a032126436e3828a3a98a6f6a6d50d3304cb0c04a8f06f4c836e6cd8b5ea6cd66b976a1245f20545ffa3e84fa24b987d6e614834c7c3a647c2706dd177057621cc1d03f5f98175f0f0ccb6dffc1cbf5e3ef7597a5e36ada7dc1408c4eb158f7485ae9c81154146ae17e8bc1154be4289bbcbd55f044734bbb0f999dba7ee3c7bee3a1efc8b68f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870458885Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default, Name: checkpoints/5696/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5696/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1d6e2c189ec97df2df16e84fd638b4b977ca0eff3694786ae4edbeb2bb742cff07527d5155e9fd99ba8fa834ebe4c529225f426dd0bfe5580736ef2ae274cecdd08f1fc77a276d18a319ec6f683895dc7ffc2a48bf72357b393a960b6013c50c3c5870746e709782f3fc23a4cfa6ce1a257e4a007aa4a07e7fdc98f16e7667011787e3472e8f4b96558ccb3eea201144810f6cb88ae2b37fadf7a8f585aaa9824f91188dbf63252cc24f697841fe5c6717a22bded4e456bfa850e3bc5016061fd011ee3c6f920a7b7021c36d7fa44a584da1b6142405b2a44966e2c74292a755978eea07f264e790372e63cbd9f15c965a3f01564e257b10065a77c369d03508&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869456412Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/array_metadatas, Name: checkpoints/384/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/384/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2dcafdb2d4aa7f079882f8d041e93315ab40b22e15137a0c9440cbce1d690455e433dcf9473a51e072e7c15b96311a28536dcfd2689ba34b1f0d7e41647eec3076c1a7bb120a6bd7895d9230d382d8e5863de971a5ed01186f5f491116cae7f6d04e6b8bd7900b2d9636bf1d626c9f27d67b4ccdcd700c4dd540d5857551553560268a42ce528cfbae106aac0e5e0acebcf64245b4e040ae461b39c8645a5be90776a8c681e2158f549043637b0ba6732d733e756446f822a246f95b07d280e1f37f8174e8c370777c77f1752fdc9d2ee81e2738a281ab8ab7654733d1c4d99c9120e813b5ca24f6b72f763644bf61ca0115672b49cb3a7498d3dfdc8bd2f27a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870652448Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440, Name: checkpoints/5440, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5440?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9bbfb3d302f298de194ea4b459ef9158f6af8c2fddba9e9d0c1c2a1378adccb9e4f962addc6597b3c4e05b17ab87febdffb4ca428541e66240e4333d556810857febda7a3e8a22d0a411e6103b09dabee5777fef509f190861d695a878fb5e76ca9dfab4b25942ca3c7266061f4bcb2fda4697a29fd5ee370673d18cf2355ed7a81a6a1d21db0350c785f020ed3900fe315512924b6db0b7c5428616a13f7e4147717c3303f1d9ba8e7cd2b80e4ed72288c8c647060fcf5fcc86e70c329e110dc58a059161e84c270f0eae0b06d2f6de2c19cf3cfc18390cf7f081d963fe1268620334b0d477a0644c403d5469680856ac0445bd2620c40407cc143b0f923575&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872410282Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default, Name: checkpoints/704/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/704/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=7e3081c50efa5be26c86d60b23047a8189b1c942bb1854c6e20f2e09179af9388401d5d115987530bf30c411c9290de29aa4d66af49aec07f1ebb690be8188210a966eca09fba5ffc98536f204aeac23dd6034a399b2bef589ebab2c798f21a0e01c3d93feb996a8f071ec590db068110801f7d28813cb034713b512729e83a6e7f48ae0111f8835a0c5061cdd7243bebcfdef6abebea2a0b0dac88edece3f8d2f0f240af0caac7d3893969fb03c0333e03b0df54bd9243679de1ad2bb32ecd3b8f17407c8f659a9a283974cb80e4349b97362d6e475d6885b404e2aa7a41bd3fb3758634c194a23b07ac35f542c431e537bb2e7f51ef8832986d95a85fbc8f4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872436329Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/array_metadatas, Name: checkpoints/256/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/256/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=04c92135fd7bad23835a8139a7f92ee5e67f1c2f104c4afcd38f81c86dd1a5959d90539cd2c18010e7e0cc07c7a28a4b4cba7883306d853f99f7e37f0037b0c59c7a02eff210b7e59dd84da464a619ee9ce03fd0c1405b291357be62ed73e7db3cb549f8f632b9c17425046cf651c10c745741263b4a9d2166ce117540e73f7e13a2e2911c2c95773255eba18153d9c8190a17eca97136da47a60a67e44586e9ca2c7ed833316b3eabb285b77c9f48c61cf8b7da0a14ac5a6baa27c5a371a9f6fe4d860bc468e1a1af3c2b820c47a23f6343184a84144dca925f2e16107315fc23dd984c7b4b7b7a34c86d6b36bd9adee41cd33a3cae40019876e04c5dd71ca7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870646277Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0, Name: checkpoints/2560/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2560/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=5fd750315043256a1635ac75478eb6519b5a87c8f7a3bb0a7868bb6ad2895a2629631c1b65777b0c70d11642c1ec3eb99b4521946cf5b9ea8b44c66bcd4763ff09131d79a2033d966fb79889005c7e4c8b780621db9cca9835937f8b26709551209ff4df8cb74d7c610afe561648f73a45ce7389d49cce62a8941399c4b79e3c5145e930158d3048f27f68e19f256f563cd3469cd19d59d09be8a6eeba9ef4b473f3e4843738fb8e4bb8b3c1146dac6d9c90e809db83c782fc676d2433080471ab4a45b3b32354d0a3f4418d5de71d37c6331af9b88491582d0a3c94e658d1788782f9192b2c68f5f232803c4c2c9d6aa2486d2e8feec494322db678bf71cafe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872344252Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d, Name: checkpoints/5120/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5120/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1e37efcff709b36021c42fe25885464a61953b4ec26a5190b147f662c5b4e75ab5c7531fc7c450e3d9b95c5aa277363f1193481f57739d805a1ecf6c3ac0f56940317f6f2e4f82575838c8a89502cf578dbdc8b35e6e933c06aecadfedc807b6c7eba663b88cd02c0afd21ccbfbc8561cf9f04ca9c6f99f8792612a3bdf986b61f5d0612dec534782924c8bbce19045ea8b006961afcd0b12be930bee0741506bc2687cbde485b05370b85e1010df3b8fc23e62c53383531b0ce6b2f2e4fdbb94ec1c70d6b17b13c54444149a78ca979e67f729598b8519a5032a742ee7778a8fc0c513edf8d26dc5fd80f45d4290e95d9975eccf642e53698f4127f6860bc64&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872498616Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d, Name: checkpoints/64/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/64/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2c10f204f31c49a0f1cca33b132fbaa8166e3eb235dfeb75a44da10a6af39a70065e18aab5b4066102636bf59a1a32759322777f5904e00edc5719c7abdf4a63836f65fc9ff4d95b5a677fefaf47a47443396854c5f06a4591a3251a2849c5e5f33a89e72a4b3cb87b293aa8df02861ed233a77f1a901cb34d391f9a1fa5fa24684988aa07268c40465b1449a59ee8b8d3d83b6fc310efd3fda1b8f0153b811cc3519cc4a01076525691ff9c5e1acfaf9eaa121c60baf27e78136d2198efb1fb94d1a0c799cdd5b6cba32db63e1832fc87bd750494b0d7b63c5c541e916a4a89bd90fb4b6acff7dff9bf30c6544349389c547770c3f9ca22da67124fd41b2167&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870751923Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d, Name: checkpoints/1728/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1728/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9e6006845239a6b1f20603cc225cfd264ab97031d5e6ef14ca936e837028b397eeea3b8b8723e5fa01d4b68a065ac9c558c1ed330fa85e2b2c779619bf05b060c36a3fd3024cdcd95d6c508eb9f8e9be61b7783d13f3b44de3942fe21d893f327eded6668708e8cc45411d0f5e440dd9b2091479147850e7d4be75066640755b88f0e0b3890ec05b94650003957f6ac547401cd1eeabfede99404d4fa508ba8c7bea03a59680a85b813aac889ed34d206f626cef1c999044eb91c58c22fb90058657391865670e11869719d801177522155a0db7e5a13d13b79029207fec1d58fb0d92951cc4af9f3788cdbed1dbd5041227e52853a791bab503ecf55bfbe008&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.871738143Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480, Name: checkpoints/4480, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4480?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6e82f6f9f2d6a75a8bc35f548bff9eda6d1e304603832b614780d30de44499d51ee08f0f46a1e6b55e4924fa5a1511213570e440d8a036ccb69dfdc1ee2f94eb94ba9716b082a53f095a5f05f3df3557fe5ae29b156f8a428bf60e17bf3eb93246272dad405861fc653a4ba4a7de7b385190c26ed6af7ea025339859653ada5fcd948343f93bf5fff8a2ab8f0ca2bc0cb0345e78fc69938f100eb785d79d80a3bab451ac2c2df778dab595a2b8d1ec49deb151d69b8b5b876d84e3d5c328f18ce6370d8cad048caba4ae409ab89485fe585105377df10ee273c97f4e15fec64a73a08e0fb89e8998f82cce015fdb440cbc67a5dc91a30181e3dc17b15d8c48af&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87258652Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/d, Name: checkpoints/4352/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4352/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=7e4579f35e405da9665440a42b3030944ea407dee9d81bb6909fc567e0c8552e197c0b2d356451ef3c3ec98cdc66bec3d3fb7e18d41e985e8e0bd9a05fd645cf6d24f2c3a0d959dee60ad2cccd240639919e71f43add12b351b470472b89f509b098ccb9311d8e8a7aed52ee43fcd533387eb4e917232b65b5c2a6d46c568940a99ebd548e8a85095c9d6033b0a46f706edbb746d7c07052e2b6eed6e109be96a0099e2ea62d7e470a7834fdeaa803ca4b543ae40b4fca78f9b353b91554948d05520e1483fdd3780ec654158a1e2d0e2925881f4080fe8012be94281e887b4eecd53289238f3e05d69a4273f591cc1f9dfa92a000af3374967bb86e11b3101c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872610056Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0, Name: checkpoints/0/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/0/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=3c3470b3e6ae26565bee731c43a9b7e3baed85a1bb88a5045cd79d591564ca9b2b0c856defa5bf20a148595f809dd77f5c8d88575e50677662a99d3d951202b878b011b104fd0411457a326851b749ad2437b1763bb4a2363c532efe29e23744c04cb2b9a9b46c47991c7af13bdedaa3f5be6396552975a00307decb8759acdcf2931144fdd5538f6dc23d6a5044268f3554e2bb6ba6266fda62fa406a3fb190d754f257e5ce5dfd43ff103547408e00cc30d00e3e416262ee00edb541cdab31fe70d969ce5a2a5e577025a44dcd45709d4a318f44a9a5592e5525d2726dc013520c88398f192b2de116010ca6de17ca184dbf33a3aeddc802fe2f759eb1781a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.871849216Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792, Name: checkpoints/1792, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1792?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b91e12372e8f561738d236355eabc57e8127f08d48adb35b15f15de8c6c854bb14cc2b8e3123f837a6bd998d0f71f989077efe56c276ed1acb318c1ee136cd9ee3a4cf154358399198fcd6c2555c20cb07f11f663655bfa72daac13133be00702ead1f1b7d11b4ee852a5274bb1105775a827b602f7166542e86e33ffeb857b85a90e731ecc86a1eb307bc6f714b2fcf43ee7d304a588528df0e17b30434621ed5647e8f640d8647af5c40d5dc411956f99ccc126a1c2507da8528fb8997ad209995011ff73a4f1d3d760d7b4070ad976da8a3c38d19dd03acf84eb7c2e9cf48e18025e191e65d9b8ba0624d951a2df4318a23bcd7676f1fff6d8ac497051c63&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872704108Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d, Name: checkpoints/4864/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4864/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=05b6d9cc7053bb1f730c806346f0885ec7f78b24505203a25bf995cc10ce1379668591a24a8d147f0ccba91b9f82d4979debde8db18e51c9c1f19d74ad5547959677213498eb027cfb29a05a1591b554b7a29b6ac83da54564005f0a7962c2bef7df61a3bc669d27188344fe50a6ced091f910b2f89032cbf45d1129d81fd250fb7c065920b6c58db67752f79f08786fbede181273a810ee6417b3d4d8d4e45d3e7392e32ef4f989795aa5b664e258e389320faad33df36c1a5a3571ce5769dd23d0086bd6b66e7839bc51462d3f9f44a492437ad1e8e3ac6d96fee0d8aa9aa39b37c566ff0b5b348648a9e17fe7023962b2b0edd5586ec4ac273ae4966b3928&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870749676Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/d, Name: checkpoints/1280/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1280/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=aba1a3da7a122c3057d0de564ae33ce6a0a7619c8a7dbd7bd463ef41d61e1e453481c1732a4e1ce716cc8073f4ccb341dda609ca02dd13d98bfa574aae23488cdbfe1530ebb9d9c1dadf826534f75706a05740e45d28dbbb2f80f51d82ca301c75657bd74c53f5c9f5be188930f2b92111d99b893c5b8ea62ee319e2f8f11068fa990581b7c49feeb3581e7b612703715cab8d35b25a7565ac2757499e8a6383cc12a69ad9ad8a6564882958abab75ed2e5ed6763a51b59820d4e1f8dbe7dc4be5ad052c041dc1dafc92758088d5a6ac2ecadac86af2b020e4ff800251526b2c9425075bdba728c65b00e9546e66fbf82d866888b59bc0532028a75bb9ba83c2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872739862Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/array_metadatas, Name: checkpoints/5760/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5760/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a1308558622c387acfa4925d0db9ba4bfa047edf7ae33786ecad5b4ad9bce493130115ca74be812766c1562bdec7cabf1072573a8345b3df85bc54d42b28d720a684d0a8631b5cd3b17f43824999fb26bebcb196696ba4d7317c377c1fee0b33efe94298ceb4bbdb43601e261ecb0a783c9883250422c73ee163e75a1869728e4284585a96328c4afd7b0ba2d658b374baf103401610da32fae8aeddc357c9ef8a014ede6654a58ebc629605c707cbb74c2aed41bdf7df27cd9697107c8e9234fc6098a6efdaa387b97dc39974b87f491f3c312e0895358c840f88a51ad695e47dfaec2117471712b5aac83c63014a27165d79d42bab98fba3be46c6bff13bdb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872730807Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/array_metadatas, Name: checkpoints/128/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/128/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=bb65838ec188114736c69c1f22bdbd23b895415b3fb567c56b86cc34d3ec5e8b9c33984d4c96d1cb604dd52f4226a2172e4be2c9fe0aad48499ddacb37085c2592428afd396721c9d60f484f10c924897d891d2e2ec97d3bfe14a9cb5dac08d0162fd25c59e769fda5b869b43029e7c5495613cd87e64ab3cb384c4589f9bd89771a4ed22805c168878851726c7d9e1fea5b5403d10d9d7cef191d09bfb364adf7c2512128435bab59942f78e81a5694fb650ca9790eac70aed6b0b424fedf6dd9b42c00d36ddc2087b17828eb57a268a6fd10f0155a2546202edc9707990a7348bedf3bcf39e0f3069ac5d4a87d0557838ace4c75154bef95a379a7e7f089c7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87185612Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default, Name: checkpoints/3968/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3968/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2d6add1a6c6c736e6320e0c09380258af303cb4f29c9837d65aa41b8f48303934308753a7ba71fddf466225fae3ce0acaa1d5e0fc1e6476aee23734f6ccdbc4acdb10363ca0432eefb41a321df42f949735f1ed67d2c5925daf46284f8acdeff96c6257cb0191d5470911443983f08d86b3031907afa29d10aeeceaed27a152e47e6c174b4a9c0b7aacfa29bc3d7741ea3086c3ee5db95d8cdae3eb82f1d22fa485ec1619c42e07d9a48be52f15700e66be63b979a31f3497b8fe57b95fcdc82572fa896ef492ddf7b9118250a00ed8f282572f69596569e53ed0526f871452bb78574eff86ddecf2589044eabb8f5af5f3e5f8d141264a2000c0975e1365de3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872829407Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d, Name: checkpoints/4608/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4608/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=c168b79ab22047d14db474bc249f631ba8720f869750a7cb3c75993c21b10042ec9dbdd21b958dd83d6697af48509197d2b08553da27db2b4a6cfbc589c8b02518b8e45538f5208722a8fb0a6f96a881f436dad2a3fd1b5eebc35ee7c54df12c1a6f4d7b74132d1ecd04e6f148420c1b9dab563bef2393f73cd7f4270a1322df9da7972711f892126ad1f1eb96fd1571831064a3ba42c0680eef18e81cf17624004fad3dda2046f75d5e2aedae87884702309f5e3979c8dbec4f6455fb829763589a1dbcbf0b0cbf14a0f24362f66d17d583e059a981a55416c0a5e66d1ce91d30cb7029ff98c3385c997610a0e358cdbc8aa14184cabcf0524cdbd3fa2cef6d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870306626Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/array_metadatas, Name: checkpoints/1280/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1280/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=88b4f5c5f1bb281f1c843eff2e8068746eacded954969b4e227c7efc190981bfe354fdd704dde568fb1943f6347914b13753ccf75cc36ad8a59da1228f0bcb0363ee74a77d0acd9ff2a0b951697a037c3c4a618e8e4e13c1af0a4e3d26bc2e6e422c41eace42676db4ea6f5d74069875562d82f6978d63173f16f31531a19b5d3439996c3aba5743158edeb2b0beb4aa192408bdceca5d186eaa98567df065657ef8e9011c48cdcf57aa71614aeee4ae2b5372c8494c6744cd0d202a5d280539fd6bf3a713e957e8d938ccbf23dd1947043382ead51acb98e19f5defd9c9f624b81410456efb0611795076fbce8b52cea3d574ebef93e955fc2abfe15b3defab&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.868846332Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/array_metadatas, Name: checkpoints/5376/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5376/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=91de4a77a7920216782768f1ad1468b0468e7dae827b1cb7b5db6bb237cc54a42c572907c03f938f4df5da3303ec8bdc3ec4fe14eac252bcc0ac169b67e21f107ec5876ae8888036d858e399fae70f6c7a148eaabb1ef82476d52924c33d29505d4818d293ae50853293b0753d2db1710d1cf2b198de045237aac6ccec975e622b5263332571a26ae782067db3834253b4658c33fa8dcab0b1386e9f11396054bd13108add407814eaf0f137b3196f299a8170b81c0c20d3283809c2045804c7f760f896b12657cdf3ef92facce15f5639d9b98bdb64a628b6f508f5938cbe22243671274ffe6fa2b77c9cdc7eb0464c40caa187b962ddb8f7ad0eb1e01a25b4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.871863887Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default, Name: checkpoints/320/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/320/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=5cab1a673e6b70ab5a4997b7efbea0308e21d790efb7624e6a838f916f72ae6c164b9310c4ce671e79cc075c91276a8409c80e91a1e8438328289bc6ffe2abc4847fa993e231a1ab19e68be2f75087387ecc767b9f8cca1b8852450673ac592ca5595cd222202598f92ed389f5ca42518e3a4d55366997a9c63503f1d1516907f7ad7afcb758b9cb90f17d6c7e755bf227a8a6ba0e64fbe61a5445cec55eb106ac474c56e967bbf5ef734d57635b130e46b3344e3fda34b113cad1a6e799d6ea7a3754b212928449fc1f34d4e8e019164d2faff92c6649375ecfe151efb9fde8a90966fa8ed5b177065de3394f8a1a30f5f0921ff1af3ea23066bc088f9c52a5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872086387Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default, Name: checkpoints/4864/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4864/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=216bba3c2ac12ce024d5552b6920556f2e1afdc66204c0a4bb2ac74b9c50d85cadb9bb80d204ebc3b131bf0d99b433e081b12d97e3f92d1026f765f92ea39d0c79cef0053165d87f3debdd2c6c227140620a9eb61dee670d2b71af6663ce43ee99dbfc770d3c7d402af959f5a695375ea53cfb830b1f0fb7ade0aa71175740f40805a9be79c4b8b9be785a05ec0610d0f8bde92c116003ab734e6d6f7d6d8602468de58674df6d1d09f3a23353131a34cffc13bbbaebe36629c0aa1e029704a589db54b6b5c9341d4536d43a04a451dc80375c9fb1adc4f323ab2550f48f915418e081586dc143566fb978924a5fc75e9941751400fee6d37b96bb6e541d8611&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870045807Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/d, Name: checkpoints/4416/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4416/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=8ce2487eb8ac2f8e6673b889be0d4d4c5d60058af81aa061101755f7b7216389224f7473db94094a7069edcbc0cb5e8ab5681dd116879576e0fa243563b8cd9fca4a204c5e1772f703ba1ee2830d9a572da911eb5d0758e711a5b140b2b6ba56e5086ec34c2ca12660728343aeb1e982711b10aba0104610d6361815fffe472ad4e314bbd560a8da2c918b1a1b0b1d9dad02993b467f94bd15ef171c388412b5e7b3dea4675868d9eca49e076f6487a5670a60b44d0649ac1a570a888a6c651f7e8e532abfb5b48dcb5a7668118a4fb53c360c31b3dac41f8eaa296940beace388c05e7750fb059a8bfbb67a69639d0511bbbfd4c43462fc6acb67ca5d28d5f2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.871961097Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832, Name: checkpoints/832, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/832?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=7b3a0d361b9b445e868fa53e72048a2dd5e494f5f2e0b2ea50df083294b1b9fcbb4eb2985224d35016421709e1816142d19fd9ca6784a9d8c0fdafd28aea835734f7657aeed8fbfd6c370f9f7d4588beb08032d7365004368e6f66decf5d6e750f3243dc38f717910b53fe784c5017146a77aea5fa886d49038ed4513723cf548fca9ec915c2d3e8211a2a0cb74139b8533ea4110f77dcdc8bee27f8783287959880af88c314d99f55d4c9ba90404415458d2bf9134e88b939501c8e29e33cb8d6cd4464e7179754f69e9dc35ef3825fe3cf3ce74638106b0e2cdb3c197d816ab3a1873dcfeca563527afbe0dc443d1ae61c6421e7f3e730af4a3e4090bb76a4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.871998102Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0, Name: checkpoints/6080/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/6080/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=4770497a28cb919f0a306b8a768925f1ff88ee43300c2b07025ca77c3a495f94b5c42673a91a895416ea62c784564611cf11ba7d3eb05da0741ad6584eb8ca6b9a7bf988ee295264f032fe3fc81769b15b63d9589a6e7a32e252fc23d8f6c75c0ec2c762649f24e2b9ffb29c84c5cb6e771b5aede7dcb6103a069670e011c09573a7a628602c5f55b4478c9e71cf084d076b807fcb5639c970c372b210dc36e15abed984f5cbc0d104d035f50215b3f8a00bafbc3e10c2b4ef608f393acfffff2d29f5e5b907b7ff1cc86fb904133cbfb32fafd5bab3c1ad17e331beffe68f3d7891443bf3a3a5428d4c1107bf620dd5a3b2aa60aaf60bb64c1c3435763b6cc4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872015824Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d, Name: checkpoints/3008/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3008/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=26a1301c91c398d4e509cd32b729e4948849d89e78286915369e2577256c2b5c7365a3b3bc8773f4a8210d79d1a9ed5701f224d1af1e38049ef19ab51021d2066d9f409405f7369e2570cb0f7f2939e2e18ce4258b8ae226dabf4fd0812dcafda0a9482d44f70991be9cb668762f16a2d28f9034b4e3e8cea888808c286cbbf6a4634b58a9a492cb6bea09c4c732682bc1f002918b6f940b21877d94772c52d88a10cf8a57c288ae5a838b1a415a155f24ee89ebc8a75f3ba068567aac40cce0ef884745484481be7ae344be64d0dd3c3238c4137c7b704ccdf4b9d025cd92e20b01be75a18642f000ccaec2964368b682efb4cd8585654a79087fe23a92c21d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87301401Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840, Name: checkpoints/3840, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3840?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b7c158ee94810eddf607bb01f35895ab1f1f294cf76db176cfdfff1994588a747f39ea9f220b017174ffb2d038c2d6f6f604407e650f976420c139a0d6b465fe09a3c09cdc077753fe527d67aa33ddb087f4e75e041e09fb8209f01020c279789c3ff1da36c6e2c257e227bdf5b02f463000438bf2349718186a2d3cd175644889a43bf040a8716bb14121b82ae31d30afeb624f9856ca3b9768ab1b45848f192e2e392b3757396a39609091745e7bd4e16e8190df068cececdb67d6eb05d73e3e56b7f187f28417b47a2d11a9c91f1ac1ee0e6e5bbce99c439ffcd1bd251db7633714d1ab05a477fe579106aac8b99e32042a10778f2d6c991eca41dd350f7e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872222387Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default, Name: checkpoints/4800/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4800/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=79572537e031d8bdf7c78726bc7b752ffa5713b1da541bf8b48b3f0bba3210d81a0a02061be8456bd73d68468efaf38fbaff53dcf51fab5f952d162f9ace031cb21d43ad456ede4d0ecd9263014f9f3c9211d3e99ebd46e00d4ab3947a52a11327fcd8daf1ec79617124b5a9f6db0661280615f2d2f7a66059a140fa86dc83f6f9e8271f8a9c6998281e3c1f486b2959293bf13f022e940d94e9129fe41bea168dcc301d271d136a71f8a58af1abed8fbfa7583bac9cb750107a54b342048d1a7f247f4c7ed87a4371e2ba53c477c7cb0558b4003d9327a450dc9a6e0635d165eb36731577832b2744c9f0e679351a3add4eaafb33e34ff679f255c5ce32b4c7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872211665Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/d, Name: checkpoints/1856/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1856/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=75dfd13629bc2438a7e0bb2160eb92ccadda538549bb2017ce84eeaea2a0ef1e639d1474ae109aa59e1b6ed2a1b9839c435359955336dfcf8520c60c1f059f571b7e7c3127f0b01560cfd31bdcf947b0a70923b630d6f22d9dcc48bf18e46c9ce09c0aefcb043830ed5c1e11916638e1b642449d0369954ad1598fb26255a40de37aa0daa9d63f405d03699d1a9bfea8497a30950e8b69cc31c39d9aff146e98f0a5b02e37d553ab8d172aa4c3df4a78cf3a3aeb1af6e3d3b730f7c8b2deab16fe33aea066041df960eb1517e34f749b48c6d83d4147aac681d211f88b2f19e980b26b7c82d8599640daf662bd3ea27ab0ed3f6cbf270f77829cbcb297fb3947&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872249896Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/d, Name: checkpoints/1792/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1792/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9ac2a8f975f9ab241dd3b442fa0c9340c45c7a56ac437ce4a44d58ad94986bcc25b4fafe7e1885b43d95450356ac00870b2c3092e071813400a13a3585b7cf265f4c9272ab076beb13bd2000905eb93b64a3ceaccff90625bd5b656e9464fa7569560997bab33b35a4f8161bffb7e1abb97cdce738270665e1abf5580241445c497546aee358d763b04ee5e1b5e9e861fc64b68e08ffc971325df454b4527702a2beed7e9706668c6bc8dfc176b75b1187f5444f55e5a39d4dc550e4a50f7d59797f7c8a519d57e22214828df8216d359f5a0a501091a4e692c55f0250d41f9d44f9773d356213118bd6dbb2f21dcf672063ad326390e7ab62f6d463f3730733&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.873072972Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/d, Name: checkpoints/3200/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3200/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=485e9cada52564ff2e73451de394a3b60e18617e7aff39516381d6383ea55b4848b053a3bdfe2cb3e9cc4bb0c09842ca571554306cffc686e26a31ba45f0192a80777bb5313d11df2cb766af1eef4e69fecf99ab2d039d64b53630179c102c395c7008cfebbbdd0d47109a167e62aa71bb642c2d8743bad837c04ed27021dcab5fa8532b30c32e68b67a79d44c2031d03c209b38d35c46d3f09b783b3c407706000e590ac1e1f31fb7f8b82075eb7c4134f9420e4c32eef9c422a723e93405f4ad9482492d309562e2be09d70e2cd16e3d33348bd215e95a5da522d74c41ca1d5aa3b3101bc8c7ecd614603fb917d558a4d56d87d28f36243e909eb54dd88d72&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.873099684Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0, Name: checkpoints/2624/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2624/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a4728e11f1d9d8e8d05542b63141e408c4bc5610a7c918ab11fafe770fdc645be6e44b2de37c6cba3b6e90df35697867843c537dd2ab9fd9f264fd7a58c04a5ab3c53eb6bdb1c406639f7e1742cede252f35ebb64514806c830b480759ece5ad5ede9acfed4e50d519a978d6a9df188b67cce056d2522e92e34e086ef667a0bece269b1ce188454fb83a8ef4cc1a4ec7627dcc55a243a6e34908c581abfc251534d1383cf721572e0dbe4d67a0717ff158ec6e7e05ac7a3e318f607a599d567ffe57f3d78c174e9da27260222dadc0eb2e98f7a43b81878c438960e1ffe61c4c0c7ab917a14cfb2e74338c58c106aaf106300f6e6e709ae227abec9151e79738&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.873130349Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default, Name: checkpoints/5312/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5312/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=aaf1aecd885996e9c7b9eeb0d17e43dbffdc31dbd660bfa88840d9544567f6ba6bc88cb6d5b62631ba84a4b59e03abbfa84dc8d73ef02289d478d9d2726bd85e7fe146bab0032928e0b6c600568b54072586a58d6f04db6a5962b1e3f30e307b2e3d3130904416b2e5a1112ad1957b8f17b7fbeff1443b0a99705b1812182e2e69fac9bb668c4400b0a8978b18a776634041eaec56952b948f80e5811a97bdc07113a142d07d2e57c4d423c54f4cb86fbc6d17f205e0cc8ae55477cf75b89874f356a4997e91ec40ddd310534ac61b8c15d512a4f369b1e4685587dc3dd652be80a76ec06215162805d25789126427af6af6d23cdfab412d2d4b56f432167304&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872364673Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/array_metadatas, Name: checkpoints/4416/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4416/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=391b3dd8eafb142d87db72e9f1d366a59d402c4cae9528d2ae4dabc44b43df2e374046cc28eabf459cc05e4d22089ae5e13e8dabd17d6ab13c399d555d7629cd898901d72a1545dcde9a3c3d5159ef562b0e752d04a90ffa1844369243a335b8d8f2c9b314bc532fedab192a89b7a9ed4afe50164d81b7a92f23fe8ae4d16986abdefa25d18edce4572e0ec7402e4d7323c0ffb98a95b1df3c8fcfe1f48f0f1ea8fdd00cd16d661b768a32454e079d22ebd0e5d8420a3f92e98c6bfb26797cf239f9ff15d9a6032b662c0267c9cc95aab5331dcadfb9373e983fa7edbef1eb65441709da406a90ef59859d28789941251f6094afa273aa73f5ad92358e8bfb6d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872339965Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/d, Name: checkpoints/4544/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4544/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=3e810e2903dcc5497d4c16afa7fd863d39de7682d34f7415112304bda97efc9b365b94031cbdf0ff3409e5d078d77df2279df08cc7488751e88d1d2ad12a135615b1c870942d8f863ce166d34e9a71bfbac02b14bc7b990d1e8846eb556578c2fc542528c100770fcdbd1c4338511e888a3267889f6bd86ea329ca83e5f35275ab6e1ff4bf4753e13166ba49ad3d80ac57de2a5e3fb962fcbe3e823cd9f04eb45815c73f62a9e7d0ec93bb05a30ff99d7629415ef194156fd0482d46ff69fc6fce7dfb53df48e140a21ab749659d2a5e9e49f3d9b8a6e689f3f1ad992e64229e0f24dee37a365bcd5f35be4a60a2d86333d7edec64d9a0c32a7d82e25095172c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87317261Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352, Name: checkpoints/4352, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4352?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=868254aba5feadcb1f3880ae655112223c927173ed5120c2abfe552513dcdfa73cd0aaf2812bcc84a04c9e9f81752242285fd450bfebfbdbf399af7ea2b0bf040ac4ebc297bf974400e1147d04ac3424719499a0c4dbe51ca10733f2ce77239c9d2af0f2caf3d7f39f1e3ee9d362972c5763993e76145d53ae802dcedbfee1c1285b043f1fd67785380ec9793ae848c0fb22f6dbc69f5b8328158c62cd5f24628aedfa43edc37201321ae5c8661ad80652c45522a545b462d97da05e32687680afe6ab6be57064fc79be2c69baefc4062c386d5b16340ae6c2fdd4013d4720ec17a3c29701f1f153df6ce2575071d1283c76cec413a6fca25dbf23e29854e3f1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.873188159Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/d, Name: checkpoints/4928/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4928/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6adbed79e53bb46e7d93fdffb7afdb2392ba82ebc77afe60c89f01025caae841c0c4bdeaf495990cea0338b70867a3f1e538dfe36018540fbfd1e8b671ffc5cd69e120f70f8e9a4ef16507c2cef10bbbc2fe8a909c7c3cc8286cab6c932025757334d5dda93deed9d0ac06d27249dc0652ff1bc2b9624df18d993b9422cff53fce0a0071b4086cf85f412f92307a02d9719383aae94da171d81b33f51259f8f12f759585fa93c74c39481ad343e699455935b2f8ec9ec425ce8b7c80583ee3ea147249930f64b99f7925c9affb45186692853b254bad339818d63bbdf5f9ec31c3c59f6e27f10ae6a68a58776ec19634476e0f8aa59ae3059ccc017451d843a4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872346474Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default, Name: checkpoints/1344/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1344/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=18a2a53b4a3aed8331c0ff7b71462b5ae3b9b4bb497d624d5a1d815ff34b4700714be39e1bb30de5a9f4c49ae9e2bfd210dcde4f31d840f37b7c3daf2cec8dbcfa3766d60ba665b4d4d6df70a042438c75f5cf261fe9e9dea0bc5b8b1791e7cc89356e978867cf01b17148b88aafb75f5770dd0c90e2f1e14723b5895593c37ca8ad242da9a4fc21c3c4485eef70dc57af94b7930fdd67b0cf444d9f1ad2d3f74c714637ac9da9077bfe99e514cb97999575c29ebc216d961ea0b77901e843bfc4a3349c2174295f265fed909db906eaa4850abc059f7e4df201c5b066f1cd850f75e2a3a581dc5b05c9e426da783a16593d0b22ef2bc6a553c3c71a80e2e2c9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.86919882Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d, Name: checkpoints/2752/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2752/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2784d8d6bd2209139f0a08a993de68418ba96a95c3b687fa6a7c2dd092d420a77d442f1b1ade11c53332aa0be6d9ecdefb4e0b56e6a0e552d8dc6b8b8b285f6b857b0b3f070f740fcd26d3dcd05b8b9454f9850b1f1eac8f9b7451501c726e9b5e804c15748e5e00dcc74c2aec49b00dfec3b363d0bccd9ec410b41e175a94093714389c2ad48e48383fd6c2ac8ebcafe35aef12ad13f3109830070cff8819bba59a10b8cfc996fc550761042ae53fef9ab50d3c3a39205523b72e51d4ce971f0b1c78ee848f0dc94136d3fa62b11e74d89c2c9707291310b874961a11926a927a90a44005f6332d541a2ea9f0ca774b8eb67dedfefbd30310b301b945e3c880&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.873349468Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/array_metadatas, Name: checkpoints/2048/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2048/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6d726a1671d82b488a021b0769b744a09436957d75c5319b91ac16f590d90a77fb45ca25e1de9eea77e787cca05b93cf1c689ae4cf7e5e17743402da9d68bfa565e4eb7920735f18db1df6231a2b1e981e594d26bdcdd3a44180c19ccd96d00d8da69b3fef38e088b7b71ca604908d4683762b73a97fb966fdceabb847abc8fb99885faf2ddb9c41e8644ffd27c49975363451f0e13f6ea38aff169e2a3b34c2417a0f6f68319afbe1409cf212ef10bc7c3e4725964f0efb7d36abfad7f8bdd4fae35e5bf614756b15caed075103427f2ae978031cb6168c828fe0aa0cf4b1cedc2b67cb04adc80cd2d291836061da693bc037b52b4fe9606eeb59947b9053ab&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870271783Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/array_metadatas, Name: checkpoints/1088/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1088/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a55a161f3e18cdd280d3bff630dfb90781f74545e6c3d0d739e8ff26743e2e82dc9299022066da51ebf90bd5d4fec71a2550869bf28a44d69facf2bd84d0da127006e6fd7104e3d679557c4dd9642b3e7c597a5177eeba408e189828fb8d13d20ddb54d983e5fb4b48933f1ebfcea65e894853364dd040d0bfab4cb892e5b3dcceb5e1bb862c08e9b8a5d553364f75ca73121eaf4667a238228c47b8b1da32276f1ba41c354d3034d900e9943e6130dc26730657d125e4d1b9a4e72c308f5f6c6076f6d3c15aaf407b2b7d9439a681734042b16b644b5cabd666c874a8586b7da93c13ed9d21bf218edb823941d5e29bd2dcb238d350ed78a7974c6d8f3abbc1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869628243Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/array_metadatas, Name: checkpoints/3712/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3712/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=5569471c38732f2b3c02df139cee27e0a0f7392f590aa6824011c0740c8b8674c7ef5c2beb9642ac738c14bc3d3e3cda27f106c951770691ae32a89458f9547fb433aea9520de8537290430c0d80a45e0218acbe5a45fad180d1556bc37732a9fee00a5056584c4424ccbc95cb937e04305fc67069d5fd5446943416bb9396ef916886d932d7e2baf92e58426d86b44439b29ffec2b5f6fd75b0584de0c48bf262e6d4a2865ab6b75da5de33a6c5077948ed783ff7bc88e5f6a4467784aa12019e9aadfd89d24729d63dcff44a265ac15a3903cf7532d046baf5b75ae177e8c446ebaebce81734dbe5808c876fec2fb15ca908c1eb1e73aa6e93eafc60fc553f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869115476Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default, Name: checkpoints/5440/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5440/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=85f6d5095bac24c4c6eb03ccf3b3f12588017f259861d5f482b84606b84602af7c1b9c43e0d80c4d91b0958ead481905cd898b201319d27439f1d40dc846fb12b01637cf75188b5f8e7912670a54260db3656775eafdcd5cca766021e62a0927d2ef40747a14963f14c598eca810160d04f4cc1ff97987a9eabff6937dd12a611c627632007d20e3c359f8ce5ab1983963b285f66e04645deb656d64f4a0df6292f4029c82ceaff7453974b9042112f63637fbab794304c41873e6cd3680c3116d56644cec1c1398aad1cdfef9c00fe681e89b9f81dadbb9d30e5182b13ad5b978d24eae4de788adfe8d5bf81bf9cbca7bd176315b13a48623e4f698b6044fb6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869200796Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d, Name: checkpoints/3840/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3840/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9b5dc11c92f99c7939cf8bbd61bb3b9c3e1756b463ef931172a259c1f38b8dffc96ed6116245013bcad854a42e1813636f40565fa8f6f65fd93cb59351e35b5159b0ed8a94dfc9ff2439c61c4dfd3f3c1f3144beba4da73d8aca9f094aaf14e2d11f4c8d74f93c29cdf1cf1f26e12ab7e5107f7d75abf37c07722d2c084b9b982a6e3cd6a844110b84325e569a883a6929cdf82ce0d35f0a068dbda5fdbaa4e5aaa0f0ca3c8a5170a8d1a64946eb7173782a152d91615a9c82844cb20bd5f1f41f7b2ecb45a1a79146312f3584bc38a0b18c16a11118b161488d02a75e75fd09364adcf90ce9f71d5aa54adf3ab547654471c42170e8b64c810cea376bbf3224&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87343651Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856, Name: checkpoints/1856, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1856?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=55b1415d71273cac80f037dd4ceef365bdfe8c5ad00f60e9f959456c28d873ec6ade500c23823d37f17bb58b37ed3126de2d57d53df124d18654fff2dfcaeb63329feb849f1a1b2d16c630e326c809ee57a1ca150aeaacff2b3c978fc66ec9006fcad94b54aa5b8903e1c628239134dc6362a4f1ff89146f8153feaf91f72e5d770e7a1a3a4bb611c04dcec72af6a093dc8c558cc99bd283cb7e41cc3c71759015d96f884b03ab9f902ed4df7ddaf7ecb710ef3870fac025c895539b39efe9f0ba4ab663d1a5e5e94b08f77432a4d654e797f16f5203203d486bb774befefd0e65e2d4fd390b8154be36aebb3efd1bde96bea864c4738fec9e3493d71c87356d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.873461701Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d, Name: checkpoints/2688/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2688/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=3164644eddbea716c997d20d5db1e0947ecc29442f007db75ad57f8dc836656273bba028f4e7fb7543bf907d2c10f89a2ef175dd4ef651471589720d5b352bf250d0efdc15f743a97b3928b347b7407fc8a63ec24f5e92dd9ad9c0bd97eecb13c552893580c1560db06d9b21568c332f15669049f4f4cbf452e5fbd906ab0dcac80c1661812d0f62957d6736fb93b85246170cf11e4f9dd236f6608f325916264be7374badfafdc391a1fd8aa40039e33e14a1a8d0c16520391685ecc38a9e2849a7c0165c687315ef0f11b5bbce3de6e49e156ada89affb6bd90c61c63bd034759b355fd5aa03074797d07c0e81bff8a0a868e77974cc0289358769884d2aab&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872454695Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/array_metadatas, Name: checkpoints/5056/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5056/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a4df0ba65898e7c810e16a10c692e39e7b5025078586f83972afd1ba8b37c91714f7d9adad51fe351bc909fc3a5333bff5816d6d0bf4e59c542bbd65c171d0d5f7cd479a5e09c165167cf532739d45225eba3d7bdcfd8c3d5f570c4e5eda0945696d188744304db538ac93322d86d5e152fb7cbfd49484e550daa1dd250b6d86e1150062561b75a69e074a7b0908ad74c8a8a1df130f8c60ac4a75c574ce965c3094ea0e15ef45402110c401fe66efbd62b72f84af69769de8f68e937b359311c4a69217258675818aea4f3d41516995a4d991946a4cf492a4da76f14d6fce1765ddeb15c75ffe95e25e02201f5fb51a22d88b53411996f85a9f258e4f855b75&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872421554Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d, Name: checkpoints/1216/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1216/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=898d34b361d34f81cefd070a518fead6f20d203644eada63ab695f56fd837f3eb3d19dc3187a31dcefba3854b280fd5d10ebab34f92be6d74b3a675c8c5c35bf3766db35926322f131ed4507bceb6f0c096436348f7a3aacdbb8ff1ccba0369107b0c3da8a8c90e7df5fe482d0b4e7cd368af62940cac1f6ab9642378a33357ed26e901502f2a857ffa475838de28cbf222fc34642c0e07cf75f2f09498ca62a0964e5a1a9f2af50440f8b266199970e341ff30a2ee569b556854c41054b4b0727a3c569cd284fed9c565b61d5139cc18bd8494c1e3a88009699de52c2563b594111ea99053bed1e147dfb7860a93dcc8e9ddfa05278a0b1af9f42c72bc52d37&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872456794Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0, Name: checkpoints/2304/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2304/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9c46ce6c1282888458728e90e8d079ec1cbf68c4b720ced15d27f1e471e484eeeab0c6d470cd105ad5fc4362bbdc61fa803619e3a268e6c98228f277dd463f3fc25b0f672de228cc516a42169802877d898968e411cfd3c57f95c48ad7b4bfa01b70129de751e6578515c9ba4ebae41be915b5eed629b26f0011f36422c4ca5e07c5412b22cb124a41c38e4fdb3890935b710683265840ce3ab07409d292a06b8090cf09dfb816605133250239ca04cb39d7ce975c22444d8dab42c4db6e002bb16d160c34df982f74db21f92dd7ab967a8f8a21afd713d25582189f8c94abc8775cff7a351dc76b0d2940b3fa4e3011e6c0837a0c9336c92d69112b66eeb4d3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872425894Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default, Name: checkpoints/4992/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4992/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=251d6204642437e708760fe5887b584a37d86e7725cf52b8102b217cfc20b2a220fecdd456ef39d10a5de150bc956ef0cbae543977ed3469b3499f4de858f12a7285fd9cfde4f5c211ed84f76918636747248a0e2668152a2804322a73bc7048ba6b29e2f1324cf79f83cba68baa9e963242b2c57bf58018d4eb358e04acf47e756abbcfb4633703114281beb9871fdd10e4b41871708e780f0a12d4f272da8b2cae93b8a23187da9a3fcc2a685ab449289a1ee627fa9e62936f269e685cc59f2853ced67ebdd97c7bf787b5807897e928c36f6abd62016009daec5f7748d76cba3c428d0eb58aeb7034610667202628d9a09805b1011e1e0284a18c0f626959&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872467503Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d, Name: checkpoints/5248/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5248/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=bc80198494c8398fd08ee34ddf2023b79da02cde63c36544b3f4bef13c2939510484f5af74a78d9b082e3e39cf317fdd952a705f64efbabc2facb24bfc6dc6dbd1faf713740831e0ab8f73c2d04627412ea3ff4a616dde878c4f7db3dba266d991642dc3a513230717e0ab616254dadeb08cdba8fdf1d1f1684d0fc00f3a18a2ef0a4b4d0c179dfeb637c5016601eb6e5bf73fc083626f7f47e4f5c61290d5cf8599ee916b105234e6e800104c509021bdc1a7f3832871ef853b92b917a4766beb35bda938cd920a42cef2a2e95d69b98692ab236179eab507baf112580b6b535c2381cb09eb19540bb6a8536dfcb734e3b985e7a4d3fd31d3b49c1ffd4e44cd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87248624Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120, Name: checkpoints/5120, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5120?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=85b2ed78b09341ff7eda662488ca6d4059852476c69bca69430ca838a19572d223035e8e92a299ce6796eb1fed36f7bbfc62d855b0008b5e59c2a3942f759b6b6ec0e7d4b18fc478a27fde44e7dc3dfd6415063a06e71fcfcc7a588b9845ce30e48b67f15a6c63de2bb2f640d0153f114f40cd842b9a85070fd4fd5210738d9d6e659fb88062a7e5ec7d5fd8d745f07b70103ae74d86434e57119f10c94160cd82dccc9c3e5c95dab0c13badd61eb87a0ad249374e189fc15909c2712a38eda3a70941fc4b2b88a10e6964f8bad0386bbbf388a273ebc3c43ea234fb6346b8bba51110e3f07f4dcd1fd1f4254cccc51a604fef969216f33f83dee9b6cce5dbe8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872585852Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/d, Name: checkpoints/640/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/640/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=440257c5200f47cdee3b91fde701f2ef97776fa2f70a6654991496b246dab4df502baa5a737f1783166ad9d26520612d64be52f1e05636fbb2d83155a20186939bf403e1fc862a9c0b389176d295168b993d77315bc8808352c2016de0213919a18c85e7f2b43cc9fd6f96352df533e3faa829316434aa20476767a77fc2e4152f8174fb542e292248830cf81c2f8944d46762a1823db7ee707f1783da95d91fd81f1cb1fc9b60d14971393346594b6939aa413db4c01d6bd2851c0bff94e7eeab8e6ffa3afa2572711bf4a56d1dd7f162bc744660f82917b510d1ae111bf6b65443d7b82931f0f2e05a50df635569f55d26520d3d4d7ebdd2de8828497d29db&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870646999Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/array_metadatas, Name: checkpoints/4608/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4608/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9aff871edacd8d10c2082341038b3a9a6f8c8c8e2307730142a1f9a89990fb9086f1ad3072b7493c870a66b2b17abe1c44cd1a4805483b6ca1b03f48e70d7460a774fc5622ed37534b327d8c11abff9f401c74adf9d6e059a73801d64f4dfe1d310076a69362f849d9bd83efa841b4bbbcb488cb970de3cb9e86c07a8de0218649ecf7dff92181f9a7ac53e6a8f2efd10d3b7ae94bdf40592b2ac75825294cc9afc48acb1610a5414f685c1c5b1143c85508d917922b81a04283264ed35143e78a2231865ce2e7d3bc56ff61ab8e0370806e0c71a54ca8189fee0d90b07cb830ca2b4f152b4b27bf4bd376999640ea0518932a18dc0a962800de360fe4c4e077&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.868960976Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/array_metadatas, Name: checkpoints/2688/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2688/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=814a067e572bcde02aa90278dd70154b39178fc94594184633dd0b3957060b408c458ae72ce98de542a7b7483fe9a05c7173a5db48e7bdd66c9ef6f44cff79dfba295aa7abb45e8a01ffc3498980c820216bba022f764fe3f142bcb0c3a991064c7aac64b2ca7e87612f232e442772f67e9cb8635d2d82bfc4e6c6fd66dcd9a6df4a5f7a37387dfb91f46c09146d97fc45e1846cc7d53f73cf6fbc4b6d2b0aaa7ab8dfeed6de4be0bfa74aa2b2f12b5e24b9e9c3bd1a529f541eb27037b7edabbbe43e528b461747c5d3a87c1d3a417376ea531eb7362cf2c1e3f3dba47155b27be842a811e75cf3d908906dd329c663876a24fd3a170610454900e4d549cbb4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.869173715Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/array_metadatas, Name: checkpoints/1664/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1664/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1dcc86a0ec525b298eb2daf874539fded02d3d3b64bf220b9791404dae6fb686d55a120e7b8953de23886331132b78f2a2b13f0b7a0900bd17288d91970c2d0cc3f5a2565d0b8891c33b2052c5c8dc6fbd779a20c8d60a8331b99c577f5251299b989d4775e1cc8925dff6499b215aed51ddf7637b69af772cf4814edd58bd7ece2cb66fe4bfbd7fdae7defc43fcd7a7b677441636d8c16ec78e8587b3ef9df916ab0e7ecdc7e9a6461dfd8eee6beb4c9c125d8ef2e97cdf8d8bf7e38ef11414c9210380743f570d7b7a4c0972fd23240e876fad6920ec314c50c95a9900fa136f175b920a81c30ce429c2d6e1a5a17188990aa7a2833bcd8b0810d52de067ac&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872630089Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default, Name: checkpoints/1024/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1024/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=37b887e5105253cf68d49ca8c12a822240297fdcc07061a9804db73745bf4aff46d26768f49a88c37757997ac0357f74a8c34caa340101161e1646387f85529c65ac75431a39f0db3bd995be39b057226c9f204efee990446c3ed2a67008f9b22bf1e4f9c664ca7c44382dd94a0f7e2d7a513545eda80dd4d0933b33f7fd8dad2dbac22fc3b0cfef1064544c4e9d69b157fa522622db0603433e6aa8ff4b37916a790163fd9e6c2cbe099cbbc88cc2b8c85c5eed51b6193b4c9d78d585204c0891d3bec772c06aed527acdfcc1143c01b645d6ce2490d8fa61f81749b1fd85690eeb0007c5a794d32bdd66d6a4bfacf5f03ca893679788847aa361af1d5e3d8e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.873628782Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/d, Name: checkpoints/1344/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1344/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=222c86e2e7a2cf935d59cd79edab0dc2d53d664325a7e4faf5bb62973d634985166585643518756bd01271b825ba20720f7a6d3e4f87d3b6a9e4d69ffa18e917e7924262ec073050486fcfe5ba649fe24b0f79f51578014bd506bde38bc4018164dd439d04d4787f4ef4809420278d89c27fe6ca591c97ed73b64cf885f6568cd6230ccb7ca09631fe7dac08f0defe0bf402c4ab6f307317ca0289002f4499dcc319119c1c67721055ab96a0eadcfb79681cfee7515e45ce8c354914336ba310b18306ad9a43a1d015533b047eabc3401bf0f9a6c568b6e8c48712240dd98e6c0bb978d9b37060b7ea88ed19d4d202c94c2a77b133a23484709ff1dd31571ba3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.873634384Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d, Name: checkpoints/5696/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5696/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=3f777497a701b467cc90ce30ec939cdd8bd8a6511443de16cede16e75e917d64d34d1e7e1e7ba7c51f13b3deeef4a4d04112a07f432b93adc00432d237c3d0a0da6bd4a571495bdbe02b04f2d5fec147432e188651790351f8e9a8fb00bb3791dc8adce4a17e7534124ea21ac1b2cca520aa9e1f213f4dcb72543230042645548dd83a25a59d3c5f87f33f19bbee639dc3f823458ca994df4a9bd0b6e00d3070f1bacec0221c5c396e3b550d2b891e51415a632f1c798c696af892770653e7430821fae53a9c7b59641f8a086e95042b6b9e365e329d1c3d195dbcb39e1a74dc10c5011fdcfe1cc4ae19f7edaf821779cd1f5a193a62ce5c367a9df3f4e7a882&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.873635399Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0, Name: checkpoints/4928/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4928/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=0e550325e6a3d86255fa9bd45dbd3fdb79b71c41fe7c0e89d62bb0ebbe8ea325c7a57e683c7eb1abc381f052d603d59ede82a96254af0e9098e753edb3376e2c4fb12013e68f054a2c2b72a3d9efdc7792bb0d73e5d3478376ba593c2d8b836be9e9ab9294857c1ea478a36541c3f1aaf62a401f351dc391f89d57ef0db086f7694cf1c8880065853a4403feb975d8dcc1267df18d023e7cbafacf656104667cea50f0e9992fe84112780644c1370467dd448b113f6d480ddb4d8aca6b0dc7726ed2dd24d602211def022a320485384103ac7d35ae7d67ab08a07e1e68fe7148b4746190e4c732f8531d1d333330661d78d5a509768b9ce966f80c05d485d7a8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872724088Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/array_metadatas, Name: checkpoints/3200/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3200/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=73a9c77bbe6181cab5ecf2dddb3fc4d209cdfbef42a3351856aa438d812342b462accc8e156d471e43871b1dab64e87ad93809d52a10e880cdf37dcfc860447c8ae79c5ee4bf78e488c8fcf93f2dc7b99f7b89299b0859c5a402a21b1d5fb63c9abb55aa555462dc1b3276a716b2395d8c990a8354a7352659ce7a15ce4805b03ed7615f35c9146b77e9cf21e7462d32147be9d6c856cffd39801cbb42e05c4f0b0e3ef656cb9bd3e93d600bcf9b136160a94b5b2f3dc7b780c09fed591f0caab0998256258d0789158d9376e7f583c5758ea2db342d951b12efcfdcd80d5b91e4ea2b3d38d35da86d751a02ad35fe0dba6c496f7473299f224d9710c24ff5b0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872758386Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default, Name: checkpoints/5056/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5056/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=986b3be6a5bc618c7b2b19dbe2b1c5a456bab8ef05d6865f6db2a07a1750829ba2e0665988aef6b96bb96b787a768998e594c1f1e4fbe528faaf3c7fa8b72d38a49bda5d5898eea85e63e8d3801c0c6362456cb8f5e15dade70521b39019a4bd1f3702f944ca51b3ce065344160bd50538e50bd1b6f4b47e8ccc47c50adf5f96499d9166bf725fc822cd79040e1a8555962a649e8dc7e51f90d631846114948bcbfe768663703b20be2b046a27893cbd00099815ae1915949dc98e4017f89c6d5a1fc9d1b29b9f810228467892cbc99a88b90c0b04440b3dbb9e38844c83351ccb10589c67a6a2c86c3aff53a477f71d0480147c4d446a6ed789b97fc93b9277&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872817179Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default, Name: checkpoints/448/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/448/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=956979dbeb8b04742c2c3f19d2287b34877836e5e43ac017e4c008cfb635277236f062ca100c5bbaa454fe95b2f5bddf0fe88cb11f46910a8ceb676c5bdc1294fae430bf7fc8bb69d470357844081df9a91b4e2c7bb85a7dfb90c2ac8ca6d3a01359f28b53cd52663cdeeb344b6d9a7e7690168a598ac2857d7ac250a0bfe0a88169722ab2c986556921d48d4e04de9066e1dba7ca1d0b008c8dd64eec4e11dc194c284ec3cbce8b4397079b81ad25a09e3ae9ce79b95596dc6e7450b12912820046c9788f14285091bf65f71de006e4913443140a45a90e0d30e60f6cd6cb34976119be9f67cc302db0f37dfd2094fb3c02023221a02148f8a4136e41514077&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.868710645Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864, Name: checkpoints/4864, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4864?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=4eed45fdd557e9236c1df1a2b681fcf749cb4591cb93bf43cc0b8bb5878db7562c3fdd10c26cae6b67b79ed034240def68117f6bec03e26c25210af5c221a9aea20b550b9a92cfe68f7bf4e12214861e5dec7652736df28a0b5db16a2a6c11ecbe5852bbc9538757cb33d3a3733ef29c24e3b0f12ac940a5e92a94ec477d596b272d9bf336cb55ba554da64ac5e675506bb8593c607543b8bf106a4d0a16a5090ef0c736f3a3e3510033f6bdd341f79e8819e19bef88fe22ec62097c49652dead4075aff9afcbffb9fb5a3d27924c4043d5590573904abd456c214682f54beca1067ff580ee03a2af4cee98899d5b1644351aeae198b49a7ccda3dce5aba8f75&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872798958Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/d, Name: checkpoints/3584/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3584/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=5cb12f8fb228431d61efc31c1721575fce0fa00341302d22533b5b888202c4f850dbb99fae466c6948f7f7bac56f6028d92c15f917f0a49382d1f73ce28a2a1b9f0a36baaa109095c13f05c00814dc40216bfbe0530b139f9c0b1628cda0bb902ff0213dce40e3585eec2936b53145e5a12213329e2e237d983fe6ec3762f818f371e4a8bd8b4a5faaeb2f563c2e2470095d6b8b802ee2273c9d7bf0f30801850fc73e6cea9f49f54231a51bae942505f1c2f19e7c68a8889cc117c4ce1b72109cb662d2ca6f17ab87040026057f13db046a503675927ac4021e68098faf8d0c274fd0dfe7f5c64010e92558e052881143fccc3befe004cf72f1601066f1638a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872848465Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0, Name: checkpoints/2880/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2880/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2c6ca5523065753d7a47cdd4dfdaadb9ac9ad1e0e91eb43880bdc299af039373dbc01083fc666487896974cd91713146c9fc70488e8ea13fce30059b8c09709a87019e55494bf0bf118cda774c9f9fe044361b84e053530275091c91781ccda6823e2d680795f34cb5d4e9cdec3e445055a921b13977f9f216e001751aae47a5b0768b1c94c8e7b438e4c3124f48c3860fd2b11f048d27f345ea39a8ddfe0b8eaea19970cd4ea3233703ed56655757fb0f03d0d15d5bfaca29b632b2eb6f79c50d4c0479613f1a12ce91f7f3baec799ff35d0ebcfc00a80d71bc5e3e944c926dfbffdabbc179073c0c7b1f25a23ae45a2b865e45f7a03646ff44cc06b44ddcc0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.871951902Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0, Name: checkpoints/5312/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5312/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=4d34c372d4403108467cde888c879f7b6e61391bfd9878c5ec1a8de781fb454985eac8954fc473b3d0317739e742e6326f5becca528667fa999d45e433ccdda06e0e497f112949d27068ee08a85408f645482b240b0140b469ef70359b49e84007abbdace33f67da4f8c797d6de9290e43e032fbdfe7cf30b6322265437c717db0a8f2fc37cbc1b309ff9cfc7e9096ff3a2691b86ed2f4cc7520167265dc6d7d0d35e90b9aa45ede0e810c9cdba46891e5a582c73ac6906ad64ed7d293eef633f3e7f02fe597066cb29062a262667018aec28a374febf55071289cc6c9690519a712d7a8f6073c85d8df4a3f126fb29cc48512a00915df4e020b55db4dbd7e7b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872938076Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d, Name: checkpoints/5376/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5376/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=80a4e023a7c8c5606626ab5bcbd28b4b62217c58243bf989ca6e1094d3c95ebe81b38a874e416b6f3bcbc0139c97ada1035db370b8f723a5e6361ea29a089fa28f2c34e7bd6c7084ecc85b6a1a5097c8591d5081a0d5474615c302ae32876b4dd7c6a7fd3937020d007e83a403d2345f2399125666836ae2511f7be6a4c40a24b5611179c269dca3fd9119ce576144fd251ac6881290fc3e4b4905787ac28901e51737df5fc7fd8598c7180dcf2477db80453ab0f2ab2079340fc9e51ac122c6ad57f73a3b11d52d5efef21c5a871c347f087101013cde39be6478daa04e919ed854bf357e24f4322c332d2e45cc66bd95e45450e8973659718c22873d350dd1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.873006407Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default, Name: checkpoints/256/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/256/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a6ba39e77b11850426c947045b36437a4b215456262ac599375360b0b53d19d15744068877e76ce9cd7b5704bdd4a9599db5d4de5f5a93e25fc1ae776866d0f7674e620dc4a4e5157105590db88bbc50e53bafa571f12c0d00fc6ed8394d0f90f61715d20aa22f971591da5aa45da02bef4bd74ff1e710287b88805874c079019c7267c64fba4b857ffe7f1663eb67a1226824895873376c03f894cb92ac8823339f528269f64479e719988e1f5aede93495651a2f48821e78f6094e2342bbaf70e10318312228d984fffffd9dee19cc6ba64d14dbcbdc508c6ebc082b814876c188a4a6703e5fa490e46e68d99b134e803d32a2fa354c6cbd9a053bc62eea1a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.873072922Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0, Name: checkpoints/704/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/704/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=42ed252d627bfe119003610fee57ea09ff52b6537e46e8ac212da2d599f2d92285d491fd1db8e978316cab207bdaaab607e1fa0400766101a0b9e0b241b1d442a356d1114adf4618ee38b99eeec4673b2d94b912c67f4b08ec85b81fbebed3d5bbf3f09e8499f635531cf74c95d6022f1f96cb09d670ef83d5577fdbc2a45cd61f3a5fd5df322c1611e9a4f1e270a5cdc8aad754916047460c936bb1b57ef719b250aa1104012db418384358832f41439378e6d71ff8b458a04977b805c553bbfacdb89b1ebfc13d34bab469792d66a5c440ef00dbf4b4b012b1ee4eaa8e7a27578ba480f4a4eb05abe62c01efb9ba13bdf5e8eef8a2debda7cea3d808290a8f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872280638Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688, Name: checkpoints/2688, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2688?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=99a3856e272cfda74ad4f2816dd26c5a94e04bd178d922b57c29a46dfb5841b09168adb3200a0852e4f841448f37fc46a61dc5086478da2883195ce6d406e0c01ddec3f3f00110710d1003c6c176f10e8985a293375c1f8ef8aabd88b2ca6236cb48298985c8815557230e6155eaf583403444c593a1ea504bc2cf5bfc9e81e895c6d1594dfd569721c8dbb54da2cdcfe73ec5e32395fd620626f59565df6e7a77ef5c7ee3d825fcd1a4d97ad70cd7bc680f476948ef629b20ca8eda1b174b072169448aabe975226ef125f16e6c62062c6ec2bef921bd412d6ee78cbdaf39e8962e9b7d32412edad2ec28c60d7fdd0227c1497f9716e99d63d04117bda37f9d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872062687Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0, Name: checkpoints/832/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/832/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=5d77bc6707f4267cbca12634ef8851bfcf76f8811deaaa482d47b14344c53bfef552aa6a3e0494c0e384197c73c08566984c7175ae5b7bb1a9f8f287e332a096bc37ccb711ea7df6b11ad4f6ae9d8325e5cef1ebfa165c655ee6397164170b153c4598912349fb18f3d40d55674532d48107416cb566a6d951f3a1544c9f568700a98012ee8044e852b5c01d569a85e797908b8a9d06f01a90f525cb378d07ed22382f9c059ef740425195da05b8b418a5ee5ada3795c075b2ca58d532077f4eb6137ce1f64b680e08c987b388f43708def4f49e0b57da59a772bcc2368de4b735e2f0f5b199b619caef12f323f8b81b61261b63cf352f8ef3c444dc11fb126b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.873091916Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/d, Name: checkpoints/3520/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3520/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=274f04c524a29cd765d39a02d25f02b05d1caf944b0a6d2e7b3a5ca4c97db49802c96b431cc3845fdb972ba54aff25347f15a401476eae2fd0ed7e9f1a170cf3f462b65d4f6fa74cf087d89e68aca898d27fee94e27a032a77d87c202e091df14479c409f05d8bd0014866c05eb4e4edc39048958a06cf0ee982e8ef19ba05bc584cdf83a9621e22ffd73157fb2d70d296667fef3faeebab8426f829011c4a0d3468fc3fb3b730ab90dfbbe127173a186d9485229edd92808a105cbffc7168138444bdb3c053f3d9c2f8b6e5753b6e9139cc30d0d793f31ae50ef96ff9bcfc30adccf5a41e30b6a6f5673b6137768298ee6adbfad82e3eab2b55694562a965b9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872279632Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888, Name: checkpoints/5888, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5888?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=c331f4aa203deda055dd84295a9987cf91acbb02121e8db3db46a1618a0f2570924c2f233134ec086cb462b6b11434a6c723b914ecc08fb125bfa330fcc4d1587b95ab582f2eb16796ef0e7812c3bfcc72fb4d83f0b7f905831f43217b157d5a7345b7dbb75b78865dbfe74b250ce1a791962875da697d482509539ea63865312f4e577a375bd29cc789c9f1b3aabcf4d22367423eb35b7bc7e901207331d0d81ba021c9288f004977218738e5f589012a5f9c71d50690f6295e37680640bb048e49b0066512189e155a1d424614b90e64e93efb7926cabdc1c03b010ac79f8c68c3889e9f49adf55fddb23676c10494ade2b7cee789e5742ee17fcea5505a57&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.8732258Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624, Name: checkpoints/2624, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2624?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=63e150ed75f1c39e440589991ebbf4897130bcc57a33faad2102d1d3dee7c2a45cdcc777325b56db671568a347655486fed752d21ae0f0b70e8c0cbb2a81a3910e25bc5c20aa68edaad3ac3e7677e72bbef383128ac7ffcadbaeff4fdc01ff2d3662e426f1dc4b029dc559443c36b9e514ba87c6d1ec52f37ec7c64066187b291d3303f1f6e04897773786dbd251f69ddc21899329d0b5270ff51fdf89b7a5b612124ad10a904df974a066d0b6f8e3170e314a5f4c3026583f13c674c5411475b39211b1e66521a77b822096a87b2a7c8a098eef64249528cce445ef179dfff50c124955ee030994ee59183e06168f011f665c9035a65fb249048f7d91e371d6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.873315818Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/array_metadatas, Name: checkpoints/1792/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1792/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=68e52bbe54271b599248fa289d5115beda7a3dbfee6d5822b88997ca0ac67878d78e08cc1023519aef254d338476f3b3bfb4d3d56de597a79ea46f53df5b2864d64ea2f5d3e725a53e0a438e236c06caef7bd4b4ef1316c7e58061b4ab4087e5da743df9fd1fbd238135c25cce78ded57db0bed45aa49e2e003cbdc76f0a60a317fe425e132db0642730f59ad5619cb38dbe79abf9a630bf56ea091660839e4ca284cef907153a07eec4eb942cf65e6d9808266a2ac5061b1ff605464452a650183e0fe66d3cecb9011a49ad14c5d75743e0605af41de3ee8106507dc068690d4123a1d47f37e0a2ca19b94e62e27fea292e0cfd00faf2db767cf463b7106d22&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.873420241Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d, Name: checkpoints/2496/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2496/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=ab6ea195f2b20cc850be1f7372c5ba8fd517931956fb294f2158e1c2c808765c7a1e03e0ec58003c9bc9033fcf7a547c58e9b2f11554336d70ea785e0704d576b83e675f82ca9f526c3b808ce31eb98d8d6b608b635ed33789b9e85cf3c882588818052f59f1b1d79daaf057642ecc343c7ab61220a17558300a41264fe356696d1833e6e3da982159c196b08ba081d0f4f5c158604b007ea76ede96e7397e0702f95dec485a9e826312098035f43e4eabda90dd2adf8d6eeeebaf84d920fddde3dbce9daef2fd5c39eaf2cfa5e42f8e12c0afa53e9725dad48afc9dca2acd1e6a356beb526acfe3ab6b506f49d6ee5e747298b75ecf1a1a46a81e756f944853&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.873474551Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d, Name: checkpoints/1408/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1408/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=22443a50ba103d11efa02bb77944b20c5eb020a7cd176b2e2cf9f5f23848fb5356b3db1398f3562048ed760e2f8982d08f9637404fa26682499cbe3ab0fd7cf8050fe91b6564b35e64e55e3328a648f8a971b847ee970ac070934661f68ee2b84bbdca8aab9143b0013bef4fe7b658e8d6fe58ac2b42cb333bd54cdf263eff0585fc33a4ae2c3bdec9f68f311e34311666496094bdfe346fb8db9f014544e2f4f7a5f9f65cb14bddb1298190d510ed70907d4a65901717f19df97a96a4a763c01684257c6cf3ac73aeb8ff8ab84d5912a6d89e50587c97b81359eb598fbe58890cb7cb4f3e2bc70011ca1ccf4933474356b6098d3d13a0f3e7c309abf7afdeca&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.873515133Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/array_metadatas, Name: checkpoints/5504/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5504/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9a8f01c43e8e3c6fb246daabf6a63c4187982550a206455cf7152cdd53e53cddbe352a0de6fed154dc59e2581e8bd43aac2bef5430f458ee4cab155630ae3a477667b0b357aea5024729ded8842b56e746e8075fc29cfcfead155c49b49e6bdfe35030a023df5b2bc9f25a8015d8c4a5fe631d0d0106078a5778e1ed1ac5a190ff69027b8c19612617c5cec83c216b75e2a5b1cd5046745f378a76895a677847dfc77d61e5ee555452407cc5933e1864618560657d16534a52604f347f031d8eae39434b9e1909dd530a6c49c5567a92078fcd9d329e7c4299e3d9e0dc5d444265314c6fb3ef10040a9a135b8e5ec0978de714949c4df0b43d7c37d9f9f9bfb5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.873321892Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/array_metadatas, Name: checkpoints/3776/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3776/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=27303806657711c24075d6b6e68456b0351df64aa4aa8efe1426d41014c47cacc3db2c9a3ebaf7c516c95a566ffd638376a8b6a506f365c6e51a4154d02961c7846adb5b8690123a8186f3212314ff4b523613ea44426b4511c088d15d97a0c251e9d25dd5c5925d77602e98fa5e25aef8016d816fb2b14d76c38a39ab235e4a7452bc52d6098bad73a137360ff1f559a4961ed41d893f406027cedd3b7372f15fabbd1b46ba34c7aa3ec1b1d781d295d66a58c634fdc8658c1491a1d648c9f5e461dbf9f0c90d5d0b95a8c8476214ab8dfce21bd276a63e50ec8e6481e1e500498b450388474367bc6215457c12b3deaa90108e3d5964609c9463dc30c5505f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.873286967Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816, Name: checkpoints/2816, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2816?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=17d12449885458d204ab4756e6c7029e3cb12d36027e996af37fb4255f6701ce945a78bc0f6a916336140ab2bb7b9d8d25b773bd30e64ec36524b0f30e89e747cfd19ea1e4a39e200e82d2f8ce932099bfb98fcfebf19c08dc8337d8841b215f2679de3df200f7fed70e3b564a59ca132f17e63160004775e5baccf0f490036cff8578d95d772c5bf1668757a82b29c5315799ee5322e38d755e135fe09c39c12e93e0f5b00cfa4e90cd16bbfbae059f5488f60f5c17858d122b37374ccac4aca31bd751ebf9b11190c444c8c66f28022bf7febb57317a293f08d78cfb2716ac376e50c0683232d9f1eabbd81c235adba661b3f50ffa37b0f6556f456fd4166f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.873516145Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/d, Name: checkpoints/5376/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5376/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=3cb417c5b91aa675ba71d9ba4d5e769914c24b960e5d44186f9a6807d57c2f965715e13f0c47f6b124ec30e715ff686eb4e665fadb373462adb2624b2c137c859486739e081f719036687dd980407f3b7021f9d9aa5003dbd506ffe61eababab7b033dbf0b7d64256fb7e2658e4d607356d06eae6c5ea56181143d6968da63309df1e824129efec755050d9360098ff84ca4d0ad495da1b86d21bb48d28a6baa538cd5a2fc2a99ef9f8641dc0bd34927518a7f8891fdc153e18930726efa41975c8cf012ec65156df19a0c001f80188c911cd9b11f8433aebceba611194fdd78fa2d885a2195c5b4ff4f84416e662a312cc552a69f57c94990b5c07478342dc0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.873551451Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/array_metadatas, Name: checkpoints/2496/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2496/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=afbcd613062189cb0c507a239f7f2b2acb4551982fad9764899ae89d67844c02a2db7f6237efa618a546f4797effa2636c77cd6923c260bbd50587f7b5ba1a0afd00853084d0d88a8ff68a5e0f117cff96b5149fccea944c84aef331c1ab82f71afffaf457b8e5ae6c78c0598dec1384c31cb9bcbcb36bcb737e18b7ed897a9c19e1e08f3130cca3115d76802140a742f4516398c5bdda4316a30d987d3837cfe32852e4bf715bc57592561d1a4fb7cc8d1327d1e6edb2ce53d821d0bebaead93eb58b16de3d6211dafe8076dbf5da21d90f141191babb840710b688fb9c15b3b68d056f3c52ed2a5f11fc76e0fb6cc50cc00757d907027ed64388ad4ec31ec8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.873549456Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416, Name: checkpoints/4416, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4416?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=4497f904625a9cc7f5ed8cbf1ae109f2580c4556436891005be1330a7b8306621cf7dda34a7d6d1873d6b680a7dc5b8997d5cdb92c2f5454e6b7d3ac682e11956b0ec9e966e547b5388e3972a2e06ce3fa44d12a677142bf914d507e318733fff83b90fc3c6780821ec6318006c4d2c517476232462df09d03a48f32d946ea89ebc2e93f16556752ffa19786b7f7bf8502f6b11ab12a61493ea9d3c1783466677f0fcdf95e55bca504e5870412cead1fd1c13d4e773b7971d58ca7947a2dd1a0bfcb0077b5b3270c2f88be5c2bc38c0e05e196b05208fabade7edeed2c637b70956827d6944f0b88330e10600072fc8db41c4921c133dc89e5865c692518f353&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.873614623Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/d, Name: checkpoints/192/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/192/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=4219a8b43a6856fbfd7d64933e25200b38c17ea2f8defa86b15370ee47f9b64f7a67ed65e14e4f7ad369ee359fa623625bb967c07f5f46eb92ce7b8df8ee632a869810ea00ed4978b08b39fb026a0aa19c9ca16a244692b1e3197fe72324b2bbd4686bdc1416ccf2e410e38af49adef94b292327bb5f2f0d2ecb5bc490b0dbf54eff37bda91f16b8f9b77738777f71426b7993c1c39e2247f8816f78ea13254ad59812972cec087c4324247c46c0dd0e42fae5bccbe80dfbecfff8e9d7cdc3b013b9de242d335583519545c1fa85ee431e2057970fc63c768a381c0b64881435a4ffe592517675d8a58be5d760802fb8e6a25ad72382bd77bc8fb3d3f08731f9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.872720879Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584, Name: checkpoints/3584, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3584?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=253c7ca140830553f361d00e35d832de954c1e15a47f74b6dc1d48d843e80ac486127ca0e78dc964bd3d30996e5fb6177d25cedab844da609c24846c41c26a0da361cc4023de86b9cb5c640d41924dbb922271a6347aa3ba2bbf7f0aabacc8023524cc4b74bd96b8801c58298e4bd33aa821fa44ac60360166bc5be4944bbe652a4f582f3d36f21c04d974cae3892ddb16bf2895fd8c17c41d651228b430cb92e31b37e10c379d7865e828e19af093cfe01e7ad9e2170b366954551c3553a277bad9dc3427c7157548617bc5c541b21c47d82327359fb9559c8014eececea7e0acae2418d58667cfe5b9b3723cf6c930d6a4b4df2e51a23d44168428543a442f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875493382Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default, Name: checkpoints/2240/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2240/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2e789140caf34c0d4726039ed5be45f2ef1fc93c945db29301828bf7a73f14e33ab2f5879894252239bab340f0f0663d317580cea1a2e4f44968b67095448c3ea2ad694ebab84ae689004a4648f14aa1e8eee6ad3add3913e89fc38be123b2ab4944825cf742bb0493aa8d478bfe28d65586e730e1be4a7f87f575088e0942b9967d3d8549ff95efab142906fc5d2aa2c780c0719e43de46258ab9244df4e598a8221b3a6d0c798d284bfd01626b06680ae3554914743f13390a44d026264817de2f4f7117852eef467b2bec13cfbf2a86b77ea539c698fc9cf5d63904b1396aed415ae57d43701ae4d95313cfd0528f4a19a652fb871d4048af0f32e0ec5aba&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875582126Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0, Name: checkpoints/128/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/128/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2f7b5f2d8dde7ae5cc1cf6eeef9a4f2c380f95173cf9fccf7ccdddca228fa6235bd7d254a90d2d5754177d636dc7ed9976b5d2d50d146dbd4879bf84ccdbfc884e77b158852e6716cafcb05a93d5bc6de6badc3ecf469bee8d44cb3d8834a459e8c4bc249dd7432a14cb84ea1552e23299e166f3fc29b4ff5d59f0c2527db945eecb8f0d0c2029141ec9fba3d292880567197a0bfc7c7ab25e0ba7e28adfe376804b1ae540f502b8045b6d4d7244e3e365641077d471c27115f7dc7ecdf035242378e68121b91c6d73814e5326405d192b9bae2d0d55066c00f491e9470057d4fdcdd61b265d3541ddad53b8bad3347ebfef90267a68c49ab0a2b73bbf9bf808&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875607367Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/d, Name: checkpoints/5888/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5888/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=8db5d6b3a3d4206feda103e3eacb253641e915bb92b046388356408e5115c4c4d7ad8dde4f914d01ff3a6416d34326412191cb251af748c20aa35e3df671ae795234ed25bafa6cb0a4374db83f4710de3c0d55a1e6b48e1ff13b55c4168c1820344569cefab515dfc4a2a2e3b7096046bdc70f6e2842a2f7d3157131eac9b9e9197bd00932f6fc3d3de413a4e65789cd3e682959ef2c32e229d202a25cb6486ad6892e8d23cca116ec15ea72c14870e301da9f0a8e4120b36c279f9f5e03dc5479352c049c8d7af6e4a89d4b38e446b73c39e1b99f67527b8f33d028a9de1e60f618e936d5886c7494753507b21e13189c97ec68496d87181a574efc7a8c9263&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875623474Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d, Name: checkpoints/320/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/320/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=65a23ac7c443155d8b8f42802fd46cd59897dd1a4b9c61cafdfaf12e12ae2f13d88e810012420126d2dcad5db9641c53712ba79c91c1942891e7a0aff4da189bdcba0018f52e0c769d534ff7965cbd0b42dc4b6b088fd997a9a911a8d42d3954735cfb4c271640c95708aaef7e4a98c422119d6e8f2b4e1120dc91c1f5b8738f0516ba9d76f4c5004631925ae5a0d6c22a71984a070034eff93e195e277b02bb4d45172cff35da300fc8997b636fd846eae684522996593717090cab692bee1092e6b3a9dda34140409e6d0a12cfcb37c24231d865ba018ae7b9ce18d19cc2a108603ca27e1e872dc14e426204a3db550e47af682ecffa3eb639082aadf6a2d2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87559327Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/d, Name: checkpoints/1536/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1536/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=0ce1a683fa44026333cef33afa130097a671395b0bcd9b40f3b35d57475822e43e2d29fb791f36c987044c172b4f535b2eae7f0d8f0a0ffc94b4d58c57dbbd06def820b4db032184f9204e1812a573e4e6ec6f89d9ca28ee0e75b862eaf013790935691dec41e3797a7bde36f3a58b4f7007485483739196a4ea010c8a23b34e8ba87647cf61023059acb9185237eab4c7a1f1ef4052a67c88c5bce11f3a59c429bb425c4d0bec0bbbde588a5fff5b4d0328d850561a0cb77f474e87818f55dff9a554c259f128c3a2e71991e6180a357c5a867af61086badb68a47b155496bd89106bf20838ee9f24687387bf60c1061dfa27d6d79fbf37d997ed49b98878f1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875595499Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256, Name: checkpoints/256, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/256?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=85064193ff251b56f61fea5cef74915f753d14ab02add3951be4b4da2bd9f56fa4f4726fdbacd42259797f30d7e033c17cc0a7ca7ec7c574a7edb483e4a11f3cd5bdaf48fe3b8ae75bf9e2cc6868094a08890e31226ee4e65110c4a1980bf08a1e5dfd197276dc31cbcccbbda474379798d7841b9829eb5683641573eccc00b8634ec94c08fb27622551a594b404a6ce987da37f4553d748c858f519f8308907a28fecb6fe382b8f90c20f3e228cbdb1e936ca782b33a81fddf96b71b062ff9661daf4bd70d46709e2ac248d4dc78fda56bad8476a7a290694d695b7cdcd5da23e91633b9b703128dd1f07f0a84b14898290e6ee97accc21ce333e0510440c99&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87562777Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d, Name: checkpoints/3712/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3712/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2b7a5353b65474896c4b7e52e5e4e3eaa147bced0f57a58a92e212d049c1f12d4f3fed802ae003bfa75f09333c10c2b87dc27a284cef44ec091bdfea44082024da8723e642d8cc2485bd7eaead9456735179ad8b3686f93dd7aefe2e374f00c0f09bf65aa5c2eb97bb65f576ec2515656bd095ddafc04c68517ee363ea4e8cd685f69d488e38c18e464b17aac4bf0a8c011c0356fb7cc596f8e7f1e62dcdf1ad4c1221b26deb5168e2d5ce06862454126a3363c0f480619c83a82ca35c19e3aaef0d2f55a65fad8f4a3c2f551d030cd6440049f1cc6180122ce1bc4825aa7597b7f0ae141c135174c60bb59c834b6ad10fa97ad0cbfd7552e0dc7233ee7cd33d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875708582Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d, Name: checkpoints/1472/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1472/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a37da17c711782d0af6196dbdfcf1b4f5910f8a72be7df17f35659ec985d25a155f9f3244a0c83c28548b1983284c3a61dadb28e34b35d5eac116562edc5c36f32ffd69744b50a2074ea7dd10ac8467557405c9a5093c2649a91fe43862a6034dc0429c628f876dbc594502e21da2473286d42e5cb713b686e913db5936532e3e98d9f65309c59c42ec99fa31455a34a69a69201dac4d406917b51ae4ea8e5cc6d6c9dd6f372a40b41eb6d79e6f126d608ec33dda8d917e4d01a0ae29e47847bc36748980914ad454ef55d64a193077f5dfdee12282f430889a15a1fb7f8024f9da806e8322e03e10f08d804544524887511dad0ce8b19255547741dbbaf2562&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875712676Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024, Name: checkpoints/1024, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1024?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=20703d1b5ab19a9a55777b140dcf895127a13d969d4ce92c3f52da30d144ef15f8aeaf6d7fd4c3b9a1b580c30d41dd6419126db1be7cd72fc4bdf228de1082339f718926960a4a1f57a8934ae302e7376641c5ea7fa1cfe77afc559810ba3999c1bc0fd214d0498650c747f859aefb4b5735e69197e684a756c3baaeb20fa23329d61fd775fabd33b896b078d457b0c1f21a052d2bb84b11be2f65a3d1261c79b378b7ebd5c9afa4884ae11324107ad26edd455ed8a5b0f480b931b753bdab3791457cc249f78637ffcf443d3060db7e0045e88e3f21a6b4a17a09e71f7c964129753d5b87b59df14dcff4692f4aeab34548281ce283358cf9cffed5a080afc8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875740243Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default, Name: checkpoints/512/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/512/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=0f4383260f71f7f8a0efee8ccd2970e75a038b2f8f0fe04938a787c9d03798e1343610ca041dba53691a2b9e74d5f3f63133f48161d583d85a474d2d8833c87ff4baa6ad089d78c56478e2f04e319e501df0321de3d600c7c552574d41e9ac53c2b3bdd5355ae8f1ea40f46b2ac9fde0335e9267b87ad7d6d30cbd24c2d870347526d69c48cafd00f8c140750948b1755e5e16e7953562bb1eef71c9186f61622623d0b9bcf03f610fae069168f1512c21d19d6a020e0a49ac12ab00c7aef5d5b4479f835c9d0c1e551307bded17e4d7fb1d9a2123ad321b9ebb15ea527180e1ffb9c9b60494ed8082aa3e2cc426f06839487a30be4ccbf874bd11accc76762c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875801237Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/array_metadatas, Name: checkpoints/2112/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2112/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=606487ebbdc0bc2c2bff644a54e563a603967d9a77153f6e05c9bbd1d3fd18cec576e739d807f4cb12c793f928a9fad948d37ee34eff2e4116160050c57270827447bdb715413c7f4bdbf93856d95b40d0125c5271e8a1c02b4f6719f2ac3055b44887c1b1dbccd45f32bc5877ecb2c4e70571e88c7c348990b493ff28a1471ce0f11bd1c5a4eb28250460e181f081aa0596f035d51cfb104940b2241723b3644df0e39aa24625c22ee3f107f98e25bc18c949d973eea10846958016643138cd24120fab97d6010a34d42069dfb4c95a5cdf8f004100b09b64283a6b01c03e5fbe09a48c605346532155210456c211dfe65ed9d86bdd77950d1fe2fffee517ff&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875632717Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0, Name: checkpoints/5120/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5120/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=01c55bafdfa9bbd059ace1961f12db0c7a73b13c60eec3c63e5ab4336c03cf33a5dd79d29fe599adfc2421f6dfc89b023b7fccbd7ce49d6ff70b32f78a3f0885d898d99a404e490165fc3451dd6ed3184b362ac7338366474ea4a72d0da76338fbcd854ae2f1a69f9853e67d176373a54ffd13517801c4ae057d4f56f9548fcfa3a07a22823038865d1dd9c3c6a4baf7f4b59a8cc5887c2babbb75c4d96fd56f88b0537072639f1676b0cb8aedf97eb82a82157e6c1394178bcf263572e3ccdac741587cb3e322b65fea44c97d3b5b5001d18ca17c03b2310ab56d0dbee63a2ede4e547beadc7bf4b2ae27030309cbdbeb62a7bc6ed8806339827908bd4cb68a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875766147Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/array_metadatas, Name: checkpoints/1920/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1920/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=32dec63246ac62b10f564c6b65cf63338f3ecc9c45928c0441d4e60c5c86472287f9456557cee5fb5ce2153e4f28064a40dec56612a4733e89f63807e502634ed1e1c8d0d682f6ab0b32037320be45716cc85e042da57d2b6fe4e1ee9714d205e52ab98b99addba07bb9dba2bd1c9a05f1fbc94b330851aef4c1f02e4e8bd1a7deb86d6c86f63bd4a6c7ea9e778bad484ea682c82482b902fa0ffee3af583cb49c18b795f2ac5fd1a10d41fed062a5f2912db8a9d0d2ee6f8b33dfe076ccb5b14b5f47d0f498f3b41c02d25d5316bd7fbba554410dcccfca0d56fc467feec76cc773d0913336c6969093596530b33d70900c0be05af37b6a09f887cfecfde3e2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875824179Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d, Name: checkpoints/3648/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3648/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b35afda13ef599d903c734b669ceb83af3391d4312796ad6464987e83c4149e21ab261c5f99bab9f4aac6a84b5aab5b24960def94754713cd24459f103380940bf1ca3193b62600b42ce5f7ddadede1bc3e30d3e7081b719b69e2c313f0aae7ca69ee311ad5cdb4378d7ac441e827c4ed0cbd15e6afb8e0c6212cfa3dbf48dbaceb889e476173301b84ae3b506d7f2378b545915a2db473b80583dc4410e27b7938be111f45a8cf3686cb109a0988eefe61a8cf2d3971591d0ebdb21e5ab96a879eab8e79cc0f206170b281f8e41a86aa17f01e95ec81ff13fb7c7b77c9c34d0aec7876603dc73f353e8b266db3be0debbf09cea2751a6315a97eb60c5ad92a1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875850883Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d, Name: checkpoints/3072/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3072/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=66c605c74e6504cd7dc8fa14dc2c6238c79942a52ab828908dbdb54af4ed98c72004c14352c95f173e632eddde4ae09b5d5c77f9297bd2e5a770c51eb5dd0cc7e1888392cfa7ab823d57d9e065b91f3c7c12e404b8b562200a5106289a427e85e7e4f45c59ba2e192f5f01480f7046ba112332e7a15301f30adb4c824e731379a7dc7b8757462ec99f89b28f18c514922cd2fca5221440e908a9141629ad5428736bf9401362e14024e8555f09537c6e917fad8007ff8a31dcdb7491d37d1c0d8b2b089f4a9370729cd398274220d3300c72db8e40fcbd2a5bc291d48b60dc4083f9775ce4e76649c5632c7416880d2b6bcf2ec9621e5a392745cabc8cfda682&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875843995Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default, Name: checkpoints/5120/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5120/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b0dd3c868e01df8d79d63f6eaa3c3bb94e91b188bdbddfcb79b0a5c65ad50677ccf276021f5d750cfbef75ebe998f8438ef5ac7c1974b2e5dcc0015874e6588cf00885b5e40743bb8c4fd0fdf59c7c4e90a88fde2ef57543cf9276d10837c733044775513c0783023c7ef8d9c1a144636c5a5bdbeb149bd1f82b66a17fb537df8ed4ae4e50e24061034c76fbf1a4bca19f18ef38d11ad743cc83863232c29011e668b429b933c0a478cbeccf9c89d0b3ff9ffb6693e6ac70f56a7f099c2b3b5061ef27c976b5e7306a5951be9ff4d88da3b742f669282e8315d19390245487d17b418b6c4f822985c95714f8d3831ea77a9ba1893ea168274448b8422d25fbbe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875883553Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default, Name: checkpoints/576/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/576/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9df1d06c8229ef8a5a64ca13b760203888d354da643448523e9af069acdf8e162b6584b89fdfa785b6ab161787ceb69b061c0f4e4a4e5f442f87dd83e05a58edebc6e78775ff5057d20698a9301b9e285e840655df0118e84b916ececd4669697f92539719ea9e18b4a246cf6a7d2fbd8db6a22207c8a2b0a3fe10abec95a3380ba6384c4b697359d77a1a3563c781011c67a6ab74e0036fccab8ace280cc8e57ea992bef58b50295f3269b85a91d14380fce5b52cde699f81e20ad65f1121027eaee226ed21c964e261805634e6f5ba96e10e6eb9243666830ff94b47c7d18b66edafbd96826c435936470a971e20a64dc30a8d907e67a14e6a2e9fa8128dae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875685969Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/array_metadatas, Name: checkpoints/1216/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1216/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=62e35deb96f798d47ee147d26e86c877e1def1c3a1b914c0688e23af83d86ecd34ba6eca33bd7d413df2a1ee6b6925258832aecb2a4a14dd2f87aff92447bc8b755ccbd0186a0b95e8df64efc81d5ce090d779908dd4a22690cbacb32e299d2aef850d11fdebdfa4bed0aaebf2e1cf828dd5e2604cfe4998dc8cb7fd2dea5b07efe0128327fa58b56028045f97d216e94062c8d4bba0dd64b605814685a13e31f27f35eba46f1fbf9dec693ca732f6c915f145fdafa6d5f16113ee81085ec452c964955a88c557234064c40d6f1de7b20f074e8fa1d994f105c6018c0fcf07378f68f774f5e1226c16ad36d48183577c0c2308b59db23e68146a6c28194c3201&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875904532Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/d, Name: checkpoints/5312/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5312/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=7e3eef30d805ef4e21d637a6377af7482db3e787b023a7b8700a4cf6add16f9e5af05bb2b38e571977b1483e18e6ee1a1cdbf6a87067e8c0b35e69a4fdde82149bbf3086794de62f40a9429a8dfed9932eb2379fef3260b121da3430704ed4d51ccd1947cb3c430c86db19143470e44af080a81e886c5ddc34ab6e99538dbbc734b02e01960bb66e0f9f74fc58d6a686f7d872dd2fc701a550e29f8fc45c7e6c0ae0283807664087a145f6f1ac8798a310bbb3cb96ae7e9365df0492d9a1a224b3241a39e02ebb0470aa0e13fc08e6ba0c864ed21b8b4e2d192020208217c2205b161a8d744959b9ed0797bb72b3d52c4c1946b3edeaba63a50130eb2a6c1e8b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875944977Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d, Name: checkpoints/2112/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2112/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1290cda2086c56c3eece88c781453ca892548c2c601cfebb3a285d82b7f08a4cff2bd145561b8e717c726bf18c7572033c80d8d818f54c936fed6c8ea335b590f6878da40b3b9145f6639b34218251cf9ca2c696b19d1d82bf6bd6a3ef95c928bb453b4fc00ca0e882ec3cf5a91242d90f52d0c2b86cc53caef123cdafb27efabfc92cd0e3f79b4177b1e1469b0c3f2b5b85e788f23d02e0f5ec5d7243c718fcddd4483b41740a5c3b5d262e570b3589c1de87a9942b4fc16b460dacbc8aa09303c8c56db46d88d8d378fd92c8dae7b5042062a701aa8759b599ab025fb454e2ef9ebe6a5eeedbd4c2c8dd1b8f1f794ba9e319f25731b1e8b35dc8bc32109a6b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875943796Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d, Name: checkpoints/6080/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/6080/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=7e0be69caef35ed9bc33470adb5adb09986aae2f5bf352035ef55fb8e954c4c4a2260bd549fb139b93f6ec045fed614770afe78ba7053c43c90481874a5a4c15a732a200e788d0417188c4bbb8dd5cffc8528026e3229d35dd4bd28894251609929bcd0c7dc078bd6895ee3631d1b9071d04769ba3a5ea9498e59658fe5de2f6153db0215f47c99bdeebf5e4073d8349d7862d8afd28b8763e0a6ac3b15cf9680d9c22a8e5ba47f77ccfccd67ee2ccc0d08b64c8425e2738328c9800138d270b693b0f3d2cca9a31709e8a763b8339b560062b6e1576c44c64ecfdf40d7cb65da3c3489b6ad0f085edea005a72b4e0980b99dbc48d910ce44c030c3c93305cd4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875704099Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d, Name: checkpoints/4480/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4480/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=857d46f30fddb75de007db5bc551d41a10479769d811387f66ae82426133f067fa859b8044c6d7362654ac5182db495a8c574d52b41ae25af4d22c5d9d128e67d1146eb32b4ea61ad5f9ec0cacef9ab794e8b9310ff4392c540a889608c31b7e4d42c719373e27dc79425741786407715484fc687a3e785655093a3d8166918c80bfc93ed02c0ea4ba6c876affcdfddcd2b3819bce047a18edacb0920934b766bbbba7905e147a5520cfca9fa59cec3a29094e832de5e8083de26cc76d811cd3f9d75430008aa05618cd0c9cfc13bf05b90836f806d587841f4faf158d3941d783d25de2998b625eded88153cade91cf0323eeb9a4244179ccb0f4408c5908cd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875963033Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0, Name: checkpoints/3648/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3648/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9e97f3862c6981b7a4283e3f7a67e8709db2774f480d557ca7ac9f3a7dc86ea5f4ab3e88a938c5b74bbc59468cc97af1581c37b652f3a010daef5d487128a79a2556945339e31ff126cffd36c6b728d7565028046c8b7911b5db5eb36cccce4cbbd026dbf3efaea4ebdcbfa4655000ad2a71fa322ea7e3958c135e60700c8f2f163426b6fe0e6ff9772941696cdb8c449482fb79d762913e58aa369fb51a916fd51ac6d9efcb487216f4902e15c4647428c87c0f3eeee6bec1b1ce9e9a1c6df49faf2222b6a439af1b156ecc33cceb67278f22f4eb8a59668750f2529e73794ae1a5c2e94309974fb0db72313f365d67e63a1f395286e99e627c44c3720c7b87&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875983613Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0, Name: checkpoints/1344/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1344/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=354882ee790bb4df28702a1a97f8e53bf1928466a2fa657542ab73a6d01b1b8b75cb79d73dfc30da412448f3ad6ab3f027ba7ee32ecfff6eb85c6a11c5ceb0af124cc23c53e884b15ac141714e2423e39f8ad66ea567ba616c513abc00441aa638766b07657fd5d3036aa730f31167174c0f0404cfaa2d64b2f258602257740994b88406e2c904a8d40a4c8141eb76ddcbf2e8e1a244c06e740962a2209685ae1105816f44fdff168c1df479e024122f464fac3fa7d8213d60952e6429ba4a132ae161aa1034b933b24941812ef2583ddec4927af4407feda46526cc682f871616385de0afdba9cda9aaffb910a08a249393d23a652b39a0a4952944008b4337&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87600471Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d, Name: checkpoints/704/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/704/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=23d6d6b709a36a2ce816b824c9aa59ed9c82807c0dfea216d687ef8e95754f922925766d3b7a31b0de4e141b7bca31c7be4b5de0018d34ace1f80322f9775673520f5c23fd1533e6dc5309e0986d854ce2454f6c86bd8831fe8e0b71e8fdbcc731d7e76f5338c2387a9409a839280b3d0bb306fd4251b3851b6afe45c6652f99f1c578bfc5a6770c0b23fe35503ef9aa32a154867e9cc2c30c8f958ca7e375136332e7d72e95c984952b5173fbf67ac24f028ac132bc9678e921f3bb3265ea2c5bea38e34c6c1a1f3f271b569e08e22c9e9a9e6cac293b22e5c6d2c16f18aff27462cede23dc75d0786ab73780edd2c76b3bc298b0a1166ab72f7a57cd44612c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876020666Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default, Name: checkpoints/192/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/192/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a03ca287d1da12025255c4011171a6b6f806c4b35b9488dc538db80cf18172c025b77817608dc1eec55f815e5529ffd8b3ac238e56a9ddfb9edb441ca22732d870da69d20488cc47a367edf2bd3ac30440c4d594b9cb1ca9244c911d8df1c67027c7ad7241f5f594efa94491d24763f37e715a40913c45b6a53ecde2a6158b7412cdd359dd2a6b8ae321fc874dcc19fe06542ab1f60e0f7508f93e96d53059719a23f15f98b8a6616206f6f5a13e207f0d9b909f1a0eb34168066ea68604d38eef77a61fe9313b8731befa484f401d217d86744eb65c2ac5e93f0b1cd3b270699557a6b6439f0f2d9db904bbe85ca5091ecacce4958a470d822b0fa07c92d49c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876043024Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/d, Name: checkpoints/5568/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5568/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=ae6c47ef0f33eec679f29e646c98a17b8403d78b8887d67f8e39dca1fa9f63eb2b01fe586b5c3d49d3449cac7dccf916319220a9ab28db356d3d7c99b78e494dd4ec79c7c01fb85670e4aab53417927fe129f3da7c48bffd077d96f131db4a9e904105206f97026c87499ec94edab2f90dfbdac491cb6a132d16bcd3ba7a47d33dd5708ad35b3b4de7bb134297cac39293cd531df693e4369e62cfbb77cad143c289b389df00e586d186841d47f0c94e2c7f0589e4486bb0b211a279ec36085035682b4f1410145d84d4a45c3214ddcffd6abe7fb92a5cf02b3e9f8743d5ff072e53f073e4e970642cf048dc8155abbaefc3a6bf1ba09ac53921bf23d9d4cc73&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876061348Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d, Name: checkpoints/768/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/768/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=aefbb0939cdad766151fa87432d0c5f740700ffdfec391b23b4a2a90cda3a66fe5f7a3b71bf732bd2a628a29e6d632b5bec10667bb2f167901adde4d6c20b464b645ca6a0a14321bb2c630f7ed32caebbdacf70f0d0486580c35fca710351edc96dca74c98d9ded90c55c93d9acd11e562447532f4db29d5cba00c4032ea939fea70bd7dae484c266c3769aebfc41e4d1724055632b3824897127ed66ec9892dd5a77fbb6cc2e5ba356511b95915c58a40bd97e0ac6d005a89f288969ad2f9e93e0552d238570d176477b404a37e5363b69e229d1b6ef48647296560fe90a3bb1d7bca6461a1f0146e5ad615a991df2ef0b0e6ffe42254a013b4ecca2f5c1ec4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876032924Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0, Name: checkpoints/3136/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3136/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=3aa7f53c280eb97c6df88b2990de9348939c44f220b526965f5393f40a1a9a6bdef53b8d877ac2d1aa7b417230ff2d888a4c6a21b899e028e36cb21616e5a37069092ffcf02f8bf3d41889c6ce9c3e82dc74ad05dca0ad667934d674061fd9b28b78bc86b94caa98f3595979fce13dd68af64346379e1d7e7443f1ff28e0d4e4dc81ff18e72db4697e59fd4c821fbf474bae76b5ee8a1ec623eb85e4a6b4c25d966543d428958e81f2106222331937a0aa8db5a98da915dc0b99f692b5c24ae6ca1347e4e6e77e52f849dac9c9cc30adc1ad86353f2905f089cd4799a1884a99dfc23a20d618f78777d7b48985ff85ca8e239a93d54dbb2487c56f60440d1bea&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875723795Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/d, Name: checkpoints/1472/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1472/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b6286e92aea5b82e152e31c02ff05db5caf4ec7b309f380e343b373c0b43fec36acdb7ca4dd0b0d0f72a5433bfdf734492f13ebf57fc47c2217f0c5348f7f066afcc1e2f450754dddbc6bbd307660dfbc876ba733da554058db09c92deb137490704407c914a207a143a78661e1fbf934793d935c2bb76158a8a5fa3f4e35934de843a92b66258b43a4d0d84798e28ffff0d7238080a9cce275d80f246484c967ead442caed1f0bd311a1e6235965365d64a7dc0405ff7a14c513b8b466981b94a1f31f5380712a7181168662e7c417a245f0eb4e4160af8adc79759942c53fd80ccb3cc60db3df042df718d030713d13b6dba0d014498c22b04be62a2855c30&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876110937Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/d, Name: checkpoints/2432/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2432/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a7d3fbb067c9c6af0bf056651aae604a71e0fca2f12ffa602fec939969c4fd9d6c08ceab65c15f7539330b5c74e1074eb277fc4b2e3306f1fc31b14d9b94e4c1c666e0d399bb08e6377bbdfcc583c3606fff3740e371b545c30c691fd1125d87f695e9a753e9d3df35bd90bd58d8f29c9b5ebcdfd7587a2932c62b72dbfe509c357ab318f20e425bcc3c031d31e9ccb48c56599a5ab98265aa63617bcffdabcc67aa477f66327cd0174aa6bb475d15151be20f3010c3484ad706c186d8b9735839729e7785cf51560285cf955a89ccb1858e0c9ea6b989fb5b3227d87ce96997a48a79aa9fca1be2ab77db1f3c0d9133fc1d6c561d160d07b38cef985402090a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876139728Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928, Name: checkpoints/4928, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4928?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=ae8b7ac94c6b1bce5668135914374df8d84f4eb07ae86d8054a91821195fd08f658b6396a92217af5166c5344c858ffd42e8310daf03f4b1e0f5793ffcf144bfbaeea0a0e15a6c94401e358da62dc56b71f22d23709b96e3e50786d38df7f9e83850f1f6875be3fc43300f96e04b330f61d3d5b3191b0d026cb344b63652585936f25c747d8bb9cb772f92c243b7e2d71f3a0c3551bbab5292dd906f7699f5ea0a76cd4f2b499e4119aa8b2216d10da98dd3a68fdfb67efa61eb52edd39e4facfed6991ccd0b4d4f6a56da85ed8f84fd1e0a747afc309bfe7f65c07d4483fd4d2d893a5d9631fd9d34103d34ed39580b577b210299acf4cd2b9c8a8f4928cb22&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876155337Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/array_metadatas, Name: checkpoints/1024/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1024/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6cb02bc15b60b248f7715b219d4e6f2d44d63abec536c2555b84e071bd7d2e04c78819e733baf397bcb7189659ccc681246de1c4c941f11204d64ada7b92867fe2288252e283787f112fa26eeff682568bce685ef903b68952076adc34508f3e0187095b2530eba711e043361925b76a453996299ed4c89056cb4251985f73cc676bc1d837bee2a3052235acfdd7bb4bedc654f51e44f77a1c9dd01fa869a8c09bcebd435dc0bff11c174b8314f6e2fdee5454862417a1d886199457eec14f008519c145e826662c105cae0002030020b5477d1f0219798aa6eb5a331023abed68bdf0e8d86de0933812eb12a0c4cd14eb4bde84d032c748c33bdf441bf35658&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876174305Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672, Name: checkpoints/4672, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4672?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=ade95be320a37ed698938073ce12edf8195d593cdd5579143bd6dc9f915eff3a52a58c8c44f40631b450bc914133011414e9f6e2661110d8ae97e87c83be4edac7c862559009401bee436d802a20c05e9ed03c85e59720dc9e4e6a4ff675d82d689e1248cb1b8761dfcc841f5b50cffd594fc5b9ef57d238dd60a4064e9dffdbe1a73f0c2f5fa034060f0cafe01c18eec628f0367b111de83cd0ccee4f3e56fef4217639970b38a58ebb7aa9c09e4b1a9125dc52f2078102f1a4d127c376266dab3b831a61c806782a834fd832001a9819d40b2c335bacd1f8a02a38bc5cc191f5981789680845a0800b4709a1dd4b67631b9cd4f9fcc260540dec903dc0f427&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87621208Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d, Name: checkpoints/1152/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1152/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1f9fe7a2cdb1035bc5aae9a6a20ddb9fd23ad7926a5fb014848cb686127f66255210a8b8fb1c5d5b63489c782de5db7e08b41440df8a84329bc11d8b1dea91cec0add61a288a3fd9d6a386f4261d51df08d06164e7701db50e4a72a49c53bb2950bfca3bbce79e9acd2eb441f8152f055832d5a1ca04b1f6e3f894aa101aed4e9cdf7cf77249533b06c109427a99e491df8dcd553741ff5d40797358e390ba342596cd71331888e7981bd082b2910d1b767aee61faeca7a31823f978053897e063456c262dfec1e79c790a37b13b2a03a261d65d8b01a7737014d6a76e359b32348b010ee9feab86269df68bb57ce94560f67a47c4d74657a7a602e4b05e51b6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875728212Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016, Name: checkpoints/6016, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/6016?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=c24a979c9f99dd6eb12844fe08140ec781804996ed9e35cbca2bd4d1c352dfec04a868f105572c56f9d4c115d16cfd3b0cb543e15705d51b4fecdd5737bf3aaa2c2e48b4cbf019be3a0ed5a33a3c3bb0b65b2b39bb466773cf7a14e10576a4e476aeaf5ff34173fdf31f4a3f87f6ed0d007e70b19ea56476b590adafa514fbc987b149612e3dfef5f0635b9cd2ba545b5f5c01c12c6f3255e1f5b526b7193d491dc2cfef06b7fb5aa7a91a5062513976d3446c32807eb3b878b11eac83a16215920cb985bada31cb96f7f714fdb131e3b9a4ef0ba32fbf9c8987a26d34ae9115d8a5ed177903c20b017a3a532e10d5c58643636661b773e201a1668dd4c68e0d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876234959Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0, Name: checkpoints/4352/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4352/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=0c5623d9d4ebf2f7db0b77ee4a31075007b8eeb01c6d913257f33a6565294a23544bb1187d664d47b6050e2c4ddca68925cfa044015221cd7e25b458e3a2da440dae2d13a41eec831c79143550adb2887b6f5e299938e4d048a5d607ef2ecd4cfdeeed78af63bca36e2a062a74da0faa4fabd26aa5b43ce772f2828dcce42f45d90a8d26a3f1dfc2191a7f643d35565dca00bb47a22cb0467ab7f19a241515ef3e910fa11463fe558f67462b46af0a884b5788b5c517523c281a8d2ae2e3f887974fc868927d48cd3e8433319802b952e3671b2f603cdff4683cdb166b1130f817053b4729218d565ffc53578d618fcaaa652177f3dbb2336e93e49cd31a5e5a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876232081Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d, Name: checkpoints/3968/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3968/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=7ec85c4c224531e2d8c6d10aa92989f54e9a69c7082aee27fa1b29a4adac8ec5abcbc6a80a087d9d64ef8e8faf3aaed007a3663b0eaf3c027d097abb376d8c5e461b3cc305d768868505c1d416ba044a1853f4149ae2ebc23055b64323e3d7d1f501a708a1b588c403196d66bf40ae0a80c58608397ffc78c3476f2f2855f4719e461f0a7f901591bcb6c93ca697e249ed42f606e715ebf7dfdeedd497b5aeee0207c40fe19d37a4ee3af07bb19c76503810d1703d5b065ad12bd11cb888c59418dac5ee586f32781be380541e4b5cd7e94cf5283ec3301b383c8f1a75489adbe38e7015d1416ed2ae70634609202681f899f287385a2fd9cb0a406bd121efc9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876240002Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920, Name: checkpoints/1920, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1920?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=53f4d0d3418deb81ff8857e666532affc362abc10d0d458012177617bb88f23546afa376127d11600715110157f27e9eb408424aede2e92a50d45182035207e528019697561cab731e7eeb5ffae1f6ccdaa3c4064eb0b7f7996da434c8335119a1e64a073f5db75d2a7a87256aa1e4791ac077b57844e15f06bb82243799f5958af770d977f718b7b6229273d73cbdd3783b42cec182a1613668972e92d377738b61a6d257836f5e3fc9a3dec2170b53324ec50bc4207e9e5606f38e0b76dfad4bc95e775e49a07c73ec2208aa1d1fbc915bfc7af2b83b5db03ce549071c3c7bd1c2d4b5a63b6e31d9c5f10e3077be4f3485a0241a14101d3e8cc9302d75c465&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876293085Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/d, Name: checkpoints/1216/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1216/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6eb7973324df16848f2af6a3528bdd73f0719a0e58bd41bc4017d259703b07402eb951d441d07a151929f0174886faf2fbbbd9ff2825f7c978161fe245e2a226ea7b4443b47512430c72cd78dae1046195f9e8d7068c83522c73d616bda0c3cc7b103f9d2dae57610340d311a7ae05c1eb64dd6493dd645ed436d6e2eaaf468ec3c066d9a85aa6cfd9fdf7ef3df327448c521ddf5c363d0b6e8e7bbc289258afa1562353a94a9338523ede702d5b200add7ac43f3c8588d56e981636eaba996126f29a119566157b8367665f25b10faf6eb5e9479236c7c1fc32d4627da5e6b4b7db5d2efc22b5d3a5d52ed7f62cba912dd0067d17a96bda7e9fb6705c7b9495&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876298393Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/array_metadatas, Name: checkpoints/2880/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2880/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=90bef6535d8e1e4807a9c39230a0275b543010fd621940e2e3e1ee74ce3eca1ec0819a18e7b1ac344c137e845d6ae4eabfb98b078b48f29cfbce73eaf711e28cf610e45db15c9d2e502b14e52aef923583111af4c6ad472cbb4511d55303659e29ea5306122f51b3c82cd6de613417200e966cf05094ca33b31ca754577a2425a1d18e800317f7fd4b3099e37aaa3b5febc47f1414b1a249ce20d658fc8965a1661c4586ebe5a358625abfbcd88bda8d95ea2f146356e13dd8a1f95bce9f80d28ac54056628a8f70dbcc9c29ce690ca3ae838abfb85a5891ea1a487a20bca1dbf09d857ed8619b51819c5da6d1fbc6e611ab055a4e16872b01eea459ddbc1d72&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876307656Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/array_metadatas, Name: checkpoints/2432/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2432/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=064990531fa503da42a144b6bdddcdcd2e3d42060d3dd9e4c11bfc0d636a09b78679113821f2201d3f1e295f02fd4a2461a72fca9f6a01a39f1952f15a55eb1b9d81a0bc2c461abf8199b4b7df42d586d5399c5da19a67cdc20c63e363f9386f26f6d87fae13ebda119e2a7e96087f6b7e16b1ee7efdd62eeade93e8a2cb5cad522bba10e654725e97366134147e09896ff0cd52da727152c7684d5009c60c423b392f8fc733d3babec6cff0cae45c7e93d17529140177b13d18fc2c0681e57e91da92fd01d29ea421ff34d3f21b82b96be7430e007891fb5af97ebd4c7f7e96262b3cbfbec9ab1d4f738d7324eaaf49158be1fa7c3dbc804c3b6ff75afa4bcb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876325733Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0, Name: checkpoints/4416/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4416/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=951e80c57f5720bffad68e5e07808248e9f007443d1184d66c3aa53d368f4aaf76fb08f6e2b69856e855ea913cdf55563968ce4a14a2669c90f45aec94acd2768dce8a549c5358a3f5502ff1269f44551c51f70b480b4fcf154334e0a692563c77c5ae3a1825a6ad92ebb473caf42fc71130d54629db1f1e72c7b053c0a4ad97d02acdd948e4724adc03a40a644048e44914840f4491f445fc266e6694701de2b874ae0d850ac00110019ebf7d707706e7c8e7c384ff049f2c7c0a61f29af9efb32bdb7f5c1c8333e38a16ce2ae927c8911e925818091394c50d76c0a428370cc4c4e91af05d8ddcebad2b6674d734b09e8763fd086c2f5db4fbb3511a13f3f6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876347317Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/d, Name: checkpoints/6016/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/6016/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=ab357dff706b2242d463377abbb8107ee5b2a4f5d96997907d232bc266a63c904547824bcc2910b4547da5ec34cf6587b9097fdaf9139bc049769f0b8eb15131ac4d563fc9c4ab884707ebf54762d1e22fd3c055f2d22b9626a9f7c921676f68196f1ef542393a2f3eb80e0197c0d292de6888c57abf9fc8325c239ccdea1e850915e037305e8ae54b690a66e50effd034e16387f47e08d09d665ecbbeacd661836b669eb771b3955d4f096aa3e519ed9503457714648b9846be73e37687aee051dba24e5e8994c9847979acabd647be5062b07a2894615bae73727228da27a8211a35475f933ad572867473b9c12d09b58b15ac639283523dc5292b954446f0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876368621Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default, Name: checkpoints/960/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/960/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=455e91e62a2f68ddaaf6307ebb07d98c6a70c0cc86b9d56e19f585eac2bbefde4021bce4b37317b894c52992d4f7c2e64d7c326232640d96a135585150227ba8910714006a38f948801a9707bba60e044d84e5d1191354ce410280c140f4be445d539cda1e96d4cc740f9ba928f386f689dad627b3ef94232eb18f593f54185862e687cb9c8823eda9ef0d64a65efa279414b4092c6234e231bc09b2d3e6136509c8795d3afb9c451bb8bdc15c3c14696041b82ab410ecfc435909992e85063c2abce1fd382aae2d22b81733930727680b7039acf64aa95a955551264dd15da3dea326dfc5a52fc9108182db739d425a20ff7db3d0e02b926f698b422ab0128f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876194696Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/d, Name: checkpoints/5760/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5760/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=48d86ad06da0a3d11a999a08682ba07995f6499e719f5212e2bc53efffb1357234d674506f63320d89de162b0b498d8488cef9790a35794e65e29dcaee53fc26fd239ccc0ad7ab56ad77ba959b0694bc5364deeec0b5c328a1140d0a8972c44ac3f2ad1eb28c90a40d1da99331544622a4a6b398ab7ae5406db2c2adc5cbe8510d0ff85f4653a7b940e28682b51ba21f11ab345f2b1822e2ee62e1a161952456202c19c4b814d480a9efb81b310d9dd7cfbea5ea8044c12250e7fdd46e95d2abd3ff9cafddca1921e2ed31f5d5f4ea52153a759864da63d2bc94fb39b18dcfa25295a17b8c63ced1700407f73de2f2344de6d9f4e683495083919b1e6474ac25&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876406962Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096, Name: checkpoints/4096, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4096?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=33eacb9cc3e1987908d2983abd45ae8adcfa065839bd136333f273af5316c2798b375e2c08b0330f5c5d341c6626b457682863634a6d8e26b03c6825e73103f5b890d3eb0f957157b479cf56bfa6e77884c8796b4965f6921b428acf4d5a50a69107d54044533d79c88bdb7cbfb5b636c5e23ad0a141e51f43e1a18829eda35f5d1a1d16c4188da2ff503c5e793f868454621850ede99d73e617def990958312c415dc8c11cc49cfad76da7d73d9d38050a36ca50a2ab76ce53176f1cff1a696352c69a0d659d20d1a922a208a57909fedf48813a8f14dc1cc1fd8b7d8935327e087beca854cc08420d0b4b0a0ae563630f91adc3b7374f67e850c1ba5190b2f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876408305Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d, Name: checkpoints/896/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/896/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a71a5c56e485be44fe5491661f4c66678421468c9eb38ddaae67e17b7a586d176a7559121109f85c8205723761f584eb570ec629df246655f7db6a438df6be728063bf8c617c06a7c7fe1888929e24713ce5c71cf7de278c8f7b943a1c51aa6df48f26ab5fd0f05beeb84bd3eb9253ed09f08e65dfbc261c47fdb5c26fe57efaba6887a64c75501331d3d82f656c49115567390ff83b5d9060bf58db9c3e0f9511f006f5048d3eece20d88165f064dacca429f8854f6f875384d429f87551f01999a34d892f1877ca077596b6b272c9246d1340c7fab1e9c9a1b6bb42c209eb469fb4a447a15e1ea2ef4a02114fc52ffc69754999dcb688acc9ab6b5c3e6e4d1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87643767Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default, Name: checkpoints/640/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/640/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=55504588d6b80fc4fcd2cce6f0eac8bcd9df411bdb43e0cab982532d5c8afbcf8a2c266cac83b7b95fe24bbb285001bf0aa0da55b93729884c95d3c300c265a4ef7f6621dcd027b62a04497a1f4cc327a434316385e0be954f792310ae5448728d62d109468f2c7d0e8a7a03f860378d6ec712df9e16486088e8d1d8356519df1f8ec287817d57fe484ab9108201da11dc593637d53bb48dd89bd5f941d72d25b87467830e4b83836c9c9c9a46485ea82079c9706bbd7345da6b304cd25a2d40cab2b68763158b33e8100c3478c2fda9b09c2f1abb1dfb69c4616d5a3185bcf64f56fdfff33f8792fdcbba7c346ed83b2bd28af9cb88ed77795f3f61d90bb223&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876244551Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/d, Name: checkpoints/4032/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4032/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=4b133238e10958d7ce844620a9c8609bb1ef0d5a714768cfed1a229df536ee439e3d8dc874b79f5a663a2a1e0c251e5847bc29786a6ce041b44640a4342f2e908d72fc0f765026140ba53bf7545656e66fac622bb48ae4f93b9e79123eac5172fae3e26f7090acdbc2dc6dae9fed93f167039332f20625c2447a8453dc4d2eb7ec09bfd02ba8016e6c0fd408a3924ba82d385e57b88ef363ca6227bc5928fd1129afacd598662f31f9eac90c74ce8722a686a1a22f92cb9706fba49f50bc0e690c054aa9e39da70f6922aee8ceadfde39d0a270a29e2f958714da7bb088435f2328dae70c79834d98f592f537c23ea4412425fb0b83b7eed97f4ba323d50883f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876462236Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d, Name: checkpoints/2240/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2240/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=96c6e8afa18d270643569364f7b5057bdd76c765fff5b6c56c198ca076f3f8fcbcc28db50efc13ef54a588740b4e0443dda80c7c189bf8bbb5c3319585168146b6f0f9747ae99a55c0e77b0eaefb74432581be5f2b6df24c56358d91b3fe51b6b0fc444d3ae5e8742726b9ef128523cd35a8203ce39a827a65759c6edb2189669cbed320b9605285221424b9c04ca2c8829c72dad0037c5ae9a45bbbd9e4cf46fab38eb7f89c81aa4d29b4b84de0167cdcdab3fdee477d8bde493aa91b65d510352f3485f5f4fd40ce94f10318f60dbe013e7a209f58be044cf3277a9e7a57dddf31e2d5dbe11efda1d3dbdb958c9e6c66d1ca20ad031189fb5babb8033bfa52&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876498658Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default, Name: checkpoints/6080/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/6080/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9afcc69e710d637f53b4509f6ad54664ed0be555a9f2874c322b2073c752ff05ef75dc49f279c6679a16bc090fe470493a43c6473c2ec435ea32e0a3380e23c6b47d06d94ee95bd8db01f6d9a80b362704c9cbcd9aab1dabf74e78f5bfedc5806b27c0855a5f52c10fbf45928e6b93ddffc5b08286c6aaa3198c780a267b6bc9ed70eda33583dcca7f15ab8c1824f2baf8ff755a1c321e7f666c2d84bb1e5a4be385fb7a11ba7e5c6b5f9c22cb6d20c573b2ce7c1eee25500938a10366a56c1b5ea914b8190a77421605292c724c4b6fbd40fcdd235f27de824ab038a94f4f14b01636011058b333e572336e28213680d746199c6db9b4b97c760d0371104d7c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876444142Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default, Name: checkpoints/1472/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1472/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1b9fc418b512db335d4b3f8fccf8d4d4b14c21e6358775ef6688a2d872bbce7e995501777eb5c420a4c8e3798fe9c460a9b0094962b466bb092fb65d410eaf16cc939ad1607fac7b573751d6c7a24a46b1cb1f903d5b0ccd4808ce9f569e948eda31b1d69512634b35c2894cd1328c45f7273dafde56570bc5a669b845a0cbfb039e789fd895c2b324985c6a943ae56438ea8ddb47e193cf84f16a3cfb004a611d702e698a04cb304f67e82fb759d8d968676da5208c56e143ffe5510398e82fce2fdfca2a548f5e14bc3ea0e61b47841a708c5d0f4fabc272ab552d78caeacf7fc1ca96adfa8db3aa83d3d77984795afcbea2c2af5d93e786cbeda0e38a9742&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876533254Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/d, Name: checkpoints/256/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/256/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=4b0ab0a536124ff5632836dde594f9f746ef29ba1e84285e755767ac2939eb973c70cf2e04bf7dfa52819ea17b9bca586beefcdd7e8a52a6b8bd4baf3ebe1745cf528979827e5db657574c5b4241575e8f61d9a4a694a375bc2c59deeae50d67a6ea4c8c919dae78fe452175ec76cf8380d5c5abf5478863774943a922d3718330e94d94eeafe270a1709df4b47cfd3c29d7daeff70d2fde50e7bd17230a1a15aaf897639a5b30f367f1f0de79149cd922a6012944ae6d551fa8cb67a4bc8d486f04f49d25352dbefe424998eb6de882215f738831cee578a790cd928ecb6b85ffc99768b955f0ed9638121238c23b191bf1d7704687658b983d534bc7e87004&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876142091Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0, Name: checkpoints/640/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/640/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a01f46750ba383cd77d97ef12ada87367e86d2033a3ed529cb15e14776893f6c134da8fcdea7d26f1354810acecb26cbc01cc38dcf1aad3f0b4e822e4459f0d09c0a8d50c2e482a9c72fe18d06181cbcd3149a5321a3c329185fcc873565d48fcf17e9dac698fd097bd7c3aada6329593b87a61bb07f630179b4ed5ce658854fdb23aace921bb27c51bfdc34018c986fa06c0c09647ac21aef4fe1c71f74916f43dc814f4db839bb8c701896b239733713db4723927274402cf9787d1d9f75515437413600f9db218b24970f8dd31b7d09a4305c2bf086870b74a40fa2f29fa5dda79bed2a980d08136306964db749c8ddd119b7cd6277725c6ac552d4c82913&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876541583Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/d, Name: checkpoints/4992/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4992/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=126cb28e9c1b72e192e2555fc8cccb5f87a4a49184ae1b73ad4d69d46f565fe216644dce8880e6472f97034137293fe500cb7f73618d7974cf01569cfbc265ee421e0ced5d30eee2627641d52f38a4af25a1cc259be1fe665d0a9d10690d62a000f3cdd093147e615a930791bf35f54f2bcf4d161943fe6c3ce0ac6163d81092649364dae3d9e072982a1e289e4512329a308c52c835f2c46914e5e299de001134b68630f4b6605c3c88a72efc0e03e2b360223093eff34df5e476d8d064e9fd7a6f3670dec193b79d7e6ae5a11d5322aad2136b1f916aec454a7170460029be3df251021f67bde5805d1d50c46c2b37afbd0db84e56b23d7d942bd6125704f2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876560752Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0, Name: checkpoints/5632/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5632/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=73650ed2b1b3ef5d8929afd0fcf0f4db6a37ac6e84029dad040dbdc1618fbefcb062ea56aeb047012ab008d258ee58841fda6fa4c82f48d8016259aef2eb2cad863403b838d21e93e1c583d60d64adbfb45995e954ad12b05aac6f2665daccef8590b050aa1771860501a5c210afc2f682a9b37d58f337641b47fa8354655f3fdd3269974e6b30f73dd361f3703c5d67b5716b447fea6dbbd03c7fe2d6ade8f9a232bc076a52192314c0a26a976933989478fa4d011707f1677221f3da0385f0fad17657c7fdf394c6ba8e5ff4ba1594a348e86ab3784826263b44ab22faaad04ba946203d5c79dd8cc35c99934b09cc58fc81ce171f9e10120db3cf8ec706c2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876595939Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default, Name: checkpoints/2112/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2112/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a5c583539d8db4c0111eb2c318ddfd83e0300344cca6d46473389b85b4dd134e35455fdf794e714156254ada4e80efe27dc9c8d7d019e0b1f396a3b08b5bac55de75826ca5b66917907e8de980cd0cabaa15dfc845be70670c83f0f097cf229e7354d2f51f2ab1d4a148fa5d59675bf470fb8ac4dd339c7d9cfab7ff55c4dd68b7dc1083dac7264a641a71464bca5d730e1f6b1cdcf7003032b24cc2934c3906ea44dbec3bc9e0bf726ab90fdb34dd6dccae5af93e6f250a5d2dad4194322fe471830134c3a5ba037e8cd09142fae7403430421970c7f360d0ffe481ed0c36c2565a487bbf4f930d196b7e8a764a693f8ce6250d106f1a467bd97cb42b7a22ea&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876601654Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d, Name: checkpoints/2304/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2304/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2e96dab0b3e5de930550a411f348abe692b49269eea4b0126837b668802ef1738ed36adc49234bc5f6363f82f5207352ccef96812c09060f7ec72bc0d64a6254c1ef2cbaa566291a5cc869bcea3ba7383756278c2d8f88834624d72ddeade43e4c45efcd0e4d5b77752640c3cbd31d7235665c5e9c28e8eda002e000a8b9dfea201dd2610eb7ece4ef2f17ca97b96e1c80d3b77bfba36798c613d671d5ab596ede22c50978b5658c9d868776a1bb02310c726c64fc68ee5d623e63e3c8f099febbfdb6470ab92e31c7f062fbfa27631de228bf9f089537bcf0ff5763cb97496360477d24ab7e99244616ffed8040da1f866d1b2f40a70685d068d2748779186b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876601209Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/array_metadatas, Name: checkpoints/1600/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1600/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6dc492177c956f9b89bed358fee4781b5dc784ede2cbc845c76193260a0ce13595313dbc14c8c3a2ba7fc54396d1c7333a062ba65745bec10d0436437f4d6d81806486e1a1d98b7f986fe0fed2a22ae72206b86b40d384feb42c581894c972a4a3ffbaaf0b7e6be92e0b3185b5aeb3661fb8fe51b685e8afa11bb003f08af008fd13ff52e593b653395889db725cc43b188d98622aa922e4bc48bb5077ceab957834bf0748f6e3b18e19fbccb11f8d7fad1cfbeee59e958c16fe1dd99350c39234efcee8c70eccd6675d09abf53bd8c3b87b839eb29e14b77db96787bd568b705c69616a7814763d2947659053d53f2cec4881170cc275055c689cbd8231c22a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876639113Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0, Name: checkpoints/192/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/192/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b7c15c10a6714d111cae99c8de58a6548940fbc995d9238e7a1fd69d5c8ac6ceba634dfb968e1f2e7e3535535637995ca25c95e027d607668b25f4efa432211a222a123723362f95c093d8e4634f6444a9b52ea45ec096cf76cddaecec8bde6d49e42db8db1bf5f83fcd5484e9dab3983168140fa408495a8100b8051cfd2e3aa8f3cafa77874306d6ea1bc95cc7da76b4522af5485efd115a2eec0e28112fe6ee90d6e22088b348f5b0f3f813ddaae2e88967b162da07cae753fbab5fb115b6779a2a0c0168eb581e29fc0bfb5e8d61dec89d9fdf0f45f95972714a9ebfef757ce9f3cdfa64e9444ed76ecf6a1bad3621b5a231bab7d756ce3ff65f4f5dd0f4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876637703Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/array_metadatas, Name: checkpoints/4800/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4800/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=bb653e0ac422c07ef92f913d39b9867eb99b38c10045f1d1c03bbc5d38f179192d4724a3a9f989a9330d8e9d9517180c45a8b5802c4c26576b6200ee4e96dad865b34f8702de3132f50fca7e0afd08c0bb3c3f710f2f14d555dd80586ff000f6caeb41b988519ce674b414074ee8156f4ca107aae7ce1d101f22820668b8546dcf29aba0b556f9819364de979731cbd4fb715beebc543d75dc25ca9016071763349245efd915c38def13339ddea5b02576c40fcd63610be0676d22ccdc17db0023973e98a9bc1beb670e19e27a80e15225eabeef9ac02fc2c5c039a1061bca31f6d715d85469f9d914f3a3b09006ccee5c37240fb1804ebbd72312d05754a597&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876655331Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d, Name: checkpoints/5568/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5568/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=900e13a6d8c4d72bc02ccf87e4fe30ae289038b695f6c49e9ecbb9b665b334fe318ca1e3e3b9ec0be21365813a027d38b59d059f534aa6fd7e7882356c607a64637a55fdaa340a333765d1747273c94dfe751c905306cb6f7244f639a1143973774842981587a67a7a6c3d23d0c54dab0142ec32f551df81e1557ef1f74ea08120b98d17020f98d8d9b699c0ef478b6dd50804b6a7bca0fcafb4e63b492ea7f9e23fb874938ece854901bc41f4d341e1e5485f35a597ba0a5d3de4af69f1962e53da3cbb8cf00c17f147a82877f6d6b877aed510f7074a42ae09508071aa0f0bd0f401ad7ece13cdc0769231c450b2ed6ed537ec11cbbd6e90d6d1b9c5397fa7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876684164Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default, Name: checkpoints/4096/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4096/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=8d4cffde4474fafcce3c70ba40cb607b75eaf71c621333fa682d20a8d589c31a4168fd96f08eeb07ebb2102e10650994845346aae0b4942c8c30b5f9fe2192ad540f2e1ab5410b2523e472372b42c9b59024704ed447e89dff63e0a5a754db7c98fb93ad726076efc280eb5dfe4d4ca0957c6947919235a349da252eaab0d99ef2c9031de3382e6b860b401cce6ecbb8a24d123649730661878b6791004ac1d89fdff55118e3ec17d290272bd0ea3be84656ae5b9b3540b45918a6786167b73bf816abfbfcff1c5cb8216e3fe7d271f842f12b7e9b50c0dbe8af761b5681f2b53566d7e2288e16534fa82b3d547217cafcd608434a01e644f6870410a1264913&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876680482Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default, Name: checkpoints/3328/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3328/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b4a817eff9b6093d9a0ec9a43958c27e6c270f803c9d416d71a4201fc95aea1d9c7e61259711cd1bcb820441f0e519f68718754a823e8e3f268a2070a62df38e2c1199f0a76279ba61c5d8c02dcf8a856530ccca089cd4ae4256e1c43f47f29f18291a7549176a942ae9451ce5892c08d5c40a34a3671913d9aff61661ef14b56dbb0305fde6f1bb16a3f47111e6bd019791da12d438e418a367893ab5566df77c6c5abf6ac34b16d3dc83c7e48f442e8d4bcb6bb731a19013dcedda24827a88a799003d3be847bae8967d2861441d32503b0a1dcb0e37c7b6d0a7646dbcb666801f4bba01a8cfb056819d8a71220f204fc1de5e42244682f4d89dba742f566c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87670427Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0, Name: checkpoints/3328/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3328/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=82efc446ef95de1cba2cc299bd1567f36ded62ab6f27addba7fa7be4a48afd6c0f399a980a1588d906b60e6176db5e370f0f5b03df08ed27bd849af138d0b2575b80850e3a388e267cb03e9fff737435c5aaacfaa50503e3be729a738ae171db647c6750467837a63e72fb8241c4c8d721e21fc5b1796dd781dad7718c1827fc90e1789dc3782e67f6beff38f94e1efc5483b7e80ee26c9b077028f8ac61b4e3859df20396f3b490bd42fb8dc916bec9799fae726c8ed95db8bc853107c146510a5d7b2c4f1043c9a9bb489f15f4e737a088a2a7e92dcddaea24b2368d1f9ff3265e964b3f0b89e555dbba62997a87ce0e8afab6ac74dbb83af1bc0f0fa7a2fc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876717154Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/array_metadatas, Name: checkpoints/3392/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3392/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=5b2e3038cc76a0127fd09c95cbbed65cf8adff5012a237a6201aed53db400deff62a6cca1c37281d0f0096346bf2a74fc1c95e24653746bb7a9c8315d98bda81e3c90b05ec99c9eb6171f15152abd7eb31b4817370fd5b712da97af113bb7a0a6cd5b9194810c45ae0c7269f7bbe0159745e4d977417e631e6345640c12d047ec3340e65a5efa45d86fdc4dbf48b3e08fdf7224aed70cc5935564cf88904f3ff6ffff8328ab0f3b3d45b58c03e69870af3ce2303998f121d15641ef93b87bc397f1efcc83c6c12560f07506e9767e395095e69b2f302800e4574455dec00564606938543ae23dcdecd2a8b633699e21c3fcb693301c7deee5665dce2b5668efc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876744438Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752, Name: checkpoints/2752, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2752?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a5c8c55ba10f0e503e99a38f3da5213260250975def74bb49a80de5226160b354790a1aa0541bdad1b3ac2025f5fd3f44e4ca410b96eb3a9760e037469fef6b26aab6dba675a00530a05792b283def7008b8622c6097ab12b6f6706c3176b1d8469dbc331f50a6299e67b49a6b3705eb07ff71f9a9bce1dd5f86597df46cec1b7b13518e634358165d46f2b8ff0cb0912be7ba63b012c9589a5585771385c4470712c3f22df291f2b71a4675b2d5f229126530380bba1aa97fe5526068b25c5bab8bce3eb4d3b511f533f11ee9f84cfbf79e772d7b43ca63cc76708748fba3a0cca3a83549c6b09df878ffafd0e2792720c9b5ffd09992ff9bd896a52cd4104d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876637097Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/array_metadatas, Name: checkpoints/1728/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1728/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=af2c8314301cee76120a8c310cca58d9cf5f270e37d46eaa4eb651ac66cf53993cf454a337dfcd3c6bd27dbd60b49f8f957aa297bf5e49a64ee9589b38823482d504a0db759245541d916c45c4656d7c2bc1da970b091843bd13f6c8789c6101a28c72b0869c7c40cbde10e3e0068905526d699638599fecb404cb57f1086ece85e99d927e9b22ffd8f4bc1b7163591e2a2dd95e37edf851c869c4bc3b08dc30562feff62f14863c2504168c544911a85124c87f1b7a8930dc44b9e8abbbf4f2d741a0729e7705712294c153cf985b1e772a4beadd66e8a251fb998718a62e3bd104669672c611531e2b69dbc560a04e6106304f2bb05c600f44bdccb144e457&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876799547Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/array_metadatas, Name: checkpoints/4864/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4864/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=bca14650a8a106852fad0352b48598fe254170c9214f4e810ebeca6bfa56053277abd529560edfcaba52e2c3c18a757824a0005577a672eda9e66d6de481fa70d39677d91ebba49eb2c0fe9e084e03d8afcdf756173e704385809fe3c86aca0ac6cbcd1f707e9a3c9ab6c1a7631f05ddbf0e24a8f0d5eebecc4bc36da5a1f44e9c5f7e1703ae6d9e034a62eee1a96dfb4af87e83b2f2196143f550f91b3536dd7b98c81cc2c675f13685da0f72eea408a9c9823ce87f85d3dede30ca9c183913e6cd86f21184263d01e078e4c658addc255e89ad5206e670fbc26ca4041325cf66929ee825feac8199dde0901b133650344d2e1023cea3e02d186bd599705d4d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876834215Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/d, Name: checkpoints/2688/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2688/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a833d89ba64f9119c038949e43f0101b2503b5207f06e214369ee822540cff2685b7589ad8337c7645faf65eb5adcbb2aef0d1c91c0e3bda010e86bc06324b6736c7212d4267e29f283181b0ba5579b55be2b167d531b61b550b27b2f168af099113cef12548f03852cd776f0a115a18c71b8cc3e281ab58af81d18ee50e644d266d77dd7dce434ca33514d1bc05bcbce91c2a4b821ec34e3e1edd6774a9e94196af952a45049105e766c0737e9c9b52193990f5a823d750ae75e145ea5e07a7a274ea617f758a48fe9c218c0855e7869bca969de665c808cf326b142e3dd1e91dae9c504f678b010093f925b5c4c4ecf3f8df07f22186491c0ae65da6ed4026&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876842273Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d, Name: checkpoints/1536/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1536/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=226ca73b8d6607a232781353c5be0dfc33b9d39eaa33cb189f53a592e2c2eb51e49de49ee72bc9c24f4ee889a21d5803d51a00ca38b3756af658e59aba3db60a5dc3aa9a31041a6122e895d318841ac0fc86dceb7681870333ff65e57dd09958b88cb6028bb0c0579fa2eb6f1fb406bc4250aa76252a9ceb3e069a0d8789a2230e4984a95676c2d9b0c89ea5b23b6774546974f722fb4715290add0838c607b3169d66efebc423b6ac2a54a2711704231d633f581c6dabfdf6f04f2488ae4013363919bff6fd3fb7adffcb0309b60ac4c6a7265eb618fe124dc0b87e34e3f379347fd5e140573cd537dd02601d362c17b8ffab9cc9b36fb2466ffd0d64386735&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876713484Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/d, Name: checkpoints/4160/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4160/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=3ed23e17948737891601e2e5a812576ee1c7b38f5a2c5606bdb15c724f3b4f2d39c743ed1dd9bdcd68b20591124ff5bd22c7f5386aec23d37613c795bb694d84d66a342d1868c114421a423524b2b6e9ae30ed8bcb49a1b033b022f7ae150e7207f9eb60204aa457582767dc16799a0b70ba1f2995015b57f8e5a61157c56e3f7a48ebc7d9ed5519fcbb2db1e9c51add917449d648aa954f27e9eff5edda76f159176887a4dca7fb6bfef065fc2ecc6898a172255abe8494d76213d7eef900ff0345b6627f3330e8b50f8a8dd679b95afe541b8d1113f7fbad355057b559c27511bc88d840ab7309b9a28853967c2b11d4fbc9f20f38af0ac013ed0477c1fbbe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876856543Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d, Name: checkpoints/4288/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4288/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=65d3556b87d7452852bc1074a6df9c92ad3c9031eeb48343319fe40256a9862aa328e91990ca9c534cc882222bdc7b52e1336a3ce4e21211bc33db5b1cd23ad7f7e1b7cc0b5a65a903b1936f4e8e8eabbe305a42ee2eee127052c6801a3ca0dbb7ba3b65cf44d893782f8d39b1d0cb1773130a946a8e526d3bdf0de6d8862014875bd79e3a244fdc0b08522d9cfeb1a2df8882d5060262979ef2806b508e819bf27290ad9d89a1c12ae6ee6abedbcbebb82a64d5d11b3a83880472d76681fb027f28e5ed2bf2ce1deabaee09925f37c636b7c86f2136477c57f8d1bbd52ba456d0599732f3119409f69a3e8fed760dd4bfafafd6cd527eadc72004aca231976a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876442772Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default, Name: checkpoints/4672/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4672/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=44f66530c8063e84f011b54ae12ca261d23258c6668903d9bbca4e6766fbc9ae0ef6c312aba96de6f2150ed5b296554889d2bedd62c2eba8e27b554a658b02bfc34693f9c560fad5c157f9e02717753d2aab5f7a15105ae767969b964eed6696e8cdf334db35b790eeedb8c684bfd24b662dba7e64cfb04997486599f839278db616e9baa71ba901b8d1fd3e264b64c4b669f469e1d949358248e2798101199a8c221183192ff2940f805ab3bb7ff7217cb75f9e18251dbe94868d564ec5eeac82f1e0b00eca099b226fe0caceabf3e8a9b33de63dba100fae6320119d7d142c2c1e390cd3079f9ed5bf144b89c9735c018e510240d868916fc1d905ef760d64&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876607548Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default, Name: checkpoints/2496/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2496/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6dc7d81b409a269a2dd36185852fbcc259e5a0b4921c6ad4f4ebdc0b3002211095bce6a88f4556ae455c344cc4b7b20598f12e4d106126ea19abd85ae17d82061f21ab742f15b0c4cf63c2144cd81c3f1e405b184891706d053fc3c53bc91e3ad1cc5eb6b05f7ded52d6e2da5eec8ac4e40690f5b7cb2a41affefad78fc1e746b98ad975ace2467a1deb1b07aa80ef67c7701aa04591a05b08570594606e0680476ecf8a18756de70303d147411b9b43e0f39e8dfda7bf2d0161043cc9974a670973c56fcf5a623bf578da7d9daeb82db5fd6153b3885bb2a5b6175e16e079466cd1abd2b4ba351e0ba326f7826af22452947f51be64ba57322b3f41f0c24013&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876878916Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/array_metadatas, Name: checkpoints/2752/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2752/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=04ce7e16483e02a7964223aac86e1ae7d5d921735960d0b052a5cb06aeadb0d294a1b5213442dac99cb21214d219c20509f0a7d472d1aa33e77bb73f23b191292eb3ab832dd4ce99ca4a589a8c2fccf43445ce55d637cd27ec014f18dcfc9f16e901b0c06b49ffe2824ccb32f9a80d35781877877a779a58faf5337a550b5f2eaeea4618e1c303aba0285269f3b2d667400a5212dbe22c7a69edb1b04b1a60359e5d58e31dd8c3c55faa7526f3c7d5eb1db10687e30eb3c987c77ae0de60cffb5ec44cdf6ce0daa267fd7d47037fbabf87b9432aebb8de4222e2899d8b509e20e303cba4d6caa0df6cb650570c4cb4f8a2cb53ea3d05bcecd30d529844a7ca29&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876933447Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d, Name: checkpoints/1664/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1664/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=3dc20214b0b9e2d5f890da597d5891d07a4d3969b039c53337276de751fe555c5ea00c40c2231b867969b62ff7c1a286049c69e73ca27a0890c32a0385d8a1b4975c146648f537101e1e1efdd6889d38f91b9edf0f3a32ed2b6ed9fd240752bf832f25ec8efed18ed6ed979c3a87bbe4c8e2deccd3b32e8271d79b6889e26b9f8ba10b4bc4f2c18a21fb924e33fadf71778c54910a94945d8c0000c9451b2d1d3449c62e987a327479556047c80085b0b385dff8554a02081f0f27dfb20274631a5a6b0fec0c6b151dad478494abe8fe20a9d644911e1508c8c2b289d74851bd13f3470540f7e85c5d24e750bf967a79266aad2605dfefd3d5572c56855b9672&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876535842Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d, Name: checkpoints/2560/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2560/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=bd8096dda7c319098b39e9d81de127050a9d58edf02373b493fdf517043ad8a09e5b2c87b7bee46cef01d27715ba41ce31bd5a7d6ed94f8d1fc7f05ecae3125252d5cc6303ec842f8790eb9762071af688f80e4f64ff29d31e92a1fd67ffb4271b3363f4867af5e92f1e4359aa597984266a7212e72bbf445ad5c479ba5e158f9009dd0c4cb398c36dfd97777545e3d98401ad8bb0559c6551f12bdfaee2a40a732aa6f4aedd8ced4ea89c3f818d047ecb5a18a37972d71675ca16b29ad7df72fcd32120acbfa1704ebf6691e81b8e206f019256a573a24fb1af1a2c05049440b2785744af9bb7e0bc95b8c78dfc683cb7146f990b1c457d6a12bce1ada2472c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876959067Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/array_metadatas, Name: checkpoints/3840/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3840/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=656e8afff4d7acc9ca3e50de29df9097f8fa73eb2cc56302dcd889b330667bf913149849378ceee39ba8da68caad041a240cb4c7e41457269063be2786b570327d8ce2b21b261d0383c4a3fe6918d0c88d247c1b5caaf6b083fc90ed13d398ad68042232b9050ae9355375a3a6756212864ef8623ebd80a908efc49c4f6fba719fedb94b558c93787d0517ad6ed7c3eed81149ce491ccc9a72b37f29f381b9135843b89f541568819ee3e685a1a17d4e5cb06ce5b321d3c7a4936d6a79f2e538c1519bd56197fa0ae23a696a9b7dd7adcf8dbbd1bffe3367f843f3f07efd343593953299bb1a047a169900c08609345a3384ce2e48e5582512156aad84ffed64&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876793639Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080, Name: checkpoints/6080, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/6080?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=83396d1e52c03ba3d18b366c432405466f7b1d9f72a788ad4f34d24f86e27b142191815e0c3b5b1a73c2553d754a5cb625daef8b643784c3aa00206ddb956b984a942474f9c2aac6ee6b3960ab65b70a006291c5dbdc2149a595d930dfba1a61fcd15b9cfdf81680a3165469c122f1ca1a38b5803fbaf39e551dd2b94b89d1ee88245700f55e9817cafef21746405b34d0ab28a37807aad7f2d2651adf6d140336437ccc2dd05dde7889d4d66f6b1550c153c478fbc7b15256c76b9034fd6667983ab634d6b9b32e995ef7812b9d643907dfbdf5e8c79a1968702b040d00660ce69a00c43d2f12dca79ddf26dd78f9ed1d25c4eed2d9dc6bc7e528c8c6d4fa83&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876784154Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312, Name: checkpoints/5312, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5312?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2760dd9e8c4677fb1c6200ae199073d36d3b806ab55fda552691e7e2fee5d472416c8dbcbbd16dfb6c237ac52371e3bb2cfac6a2c2faf6a58160085a217b77eb1c9dd813530fa38bd443bb7d4cd58fddffe3ed4e9a2b28fe090cf015377b27bed1c0ed9bee01420b4921f15646b5ba50026995b9b4cb52f487a67f4a8697c12bbbecd8d9d5bb9f926e96484d46aa591202c7a4b3d7ca0337221d95aea2b88bddb8748ec35f1d68d047426fd284c3caf8613869173ca1ecfa7baf5942e5eaf633f7e6ab3cea7b0941f9f5a0dc438464195f02b513b573e60f174b31c6deea63157967aa49ee7e9ebc3b3661642460dc9d71c7deae440dd280b49a8f13db970280&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876991723Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d, Name: checkpoints/5888/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5888/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6bfa224e5d4886d79ecc69e4deeb33f363de6bd79a980942f7578ecc78dddaa996ee97ebc324f24f94134aa81fe97b2a6efeec95a275e67ddc149d71ea5c597800b8cd3afd8ff35ada0142cec37180a42965e085c055105e9b9cfd472ccd23ca7d6e08b2f63a959d13f5d7c37110abd3c9179c4fa3eeb93f3914a1a19eae7b1d1bfe18e5d40228403246947c859d8f4bedba607ef81b71aa40253ac5d412a3b0273e9b5320205068cc3c014a6a4ec0a0bc42e16a2cb47cbee00752b528b6b81f28234f0fb9449e521b105ba62c0186601bb7389c4df58cd84dfb98463748bb57598de065395fd48d99915c13ace73c98b686c26385841a66902b6f20a0ac08e1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876939105Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072, Name: checkpoints/3072, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3072?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1f2c6933841adba970b251df6ad29babcc89d7ad77af989123c747cfacdc8991cf40ca72e71b1e51cae3976bbba8822ca880f9a645abce5765b7cdad0306d6c4c7ac56584e7cd052f80629557b64fc862c36b30e19d2f387badb9d29c75010f8385238482d8f550ce4709d5227550a58029e15a4b9535f54b7f8934afbdd5110bfaceed0e0b9bb631b9fa5ea391bac91ceb51c159ecdcdad6b3dda60bb2475e0ad284a8ea3c5726a6e083930424c092f08ad5bcc9363270c291a2f933bde9fe4f094cba3af561137425a6f29766f3bf259d14bf20eb1d9968241442bcf17006188a8d0ff7b74bfb35163555a14f4f9728aa236176b86f7a2fa539a4d685d6a72&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876723212Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/array_metadatas, Name: checkpoints/1472/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1472/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=4ad41cabac3505edc1355034787dc46f7a949751ec694b4964cd93bc1359001c5b7d3c9b6cfc1cd173c47f9c64c2fefcf9630ba937e0a902e525e10b4936e27e6e6b0179403d2af36f8745ee6d002368dab4b70539d0a2bcb07d92e63933b366d6ae22130785c17e66efdc835b2da79871d5507d04d71e7f676efd460b2887c155fc8b07b0acf4ddce754cbb381b371ecb8259a408cfca0f9d256417f383272866e4ca4035335255e76c5b915957cfa7789ebd376750ea6ca41d3e9339b0085dfe6030fa8623981ae98772e1942edf0d4ed4df2b812141756cde34da7ccbdee1923f9a699d6378a6bcbf583cbbd9e1add299a0734b788933646f2cd82817de1a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877031673Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d, Name: checkpoints/1600/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1600/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=32a049371f8ea8fe110fa0faa7140d6633239d25c15d9a9845a724061c47f183441a6774755675e28f606406800145eff6db145282a76cb7f256d415a56926bd6340ef40be45bf19bb8da45bcd4928765e1fb330449b078a68200317cd3f838bf6d03eff2f7984b390d7b62f0883ce42ca19e5ac5494b504f624fd5ffabd56f7ac041b11a2234d2e768e55aa8739e61b7d7fd40c1bc6de069f7f2fb58a21f1fa95474e63a94dcb856fb75d32761d68c8042fd71484ed44fcbcc6c0b88fd5fe3de6eacbf6c97b3bf1899f08910e0aac670568dc214bc669f8e949eb169062a21d2135f4a41d4419d4682c89b935dcc14f3a42a1f7664b18f4f489158617b50762&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87697845Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/d, Name: checkpoints/3008/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3008/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=173e00f493d89fddcda55bac367dda15a0ef70ff87b1ca7b0ab3bf5ae02d5287352fb46311efac4ea01affeedcff56a8501057fbb1becf25dfcd079abecb1a67868a2c5180700f6e88474aacc2a340d5cfe3cac91673b6aa7b994a2b84e4aaee9932277ae7ee482470b7b72e7ea8753af01e5747566f539d8414f840eeaa7fd4182cf596d2bd6d20ea8071dc63f2a4c367b8b829953c4481855132380695eac8317ee9dd74c002a2fc59090e3ba6205e5dd54ecb1b09f3342a479c14396d70e1d29b94c11b95c44a63d2bb593ca1175afd784862970850b0bbe1042ced2209a789fbfb4a1b207baafc99508934d45eb7c43e8046fd297ad9dce727a29dda5dc2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876723755Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d, Name: checkpoints/384/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/384/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=5ea1838aa20135a6da0c64b2319ed91e9308f0324dd0268380222cfa3a1d5fc23a64803e404002cf280ec7034196ab06bdff5e1b98f77b9cdfe30013afd87d4393a18ae95aef5a118051cb464edd83b28cb4dc44ef30e9f5f2baa100f871f2584f38baac7480427274d3003ec538a7271c92052ac15e3e70234bc25d299ac835e8bdf90cc52f000795c6495cccedb612a1908179c803703c20d4e810727eee70f0f75bb2ff0474db96e71136bc53d11717fa2b827b41fa63b9c057274c923db9b9f463abb22eb9a53992ef705dc10888938f554714af77a5f7702ab80b714043e8cda6b4cc3324ae5c3305a487484fefceaf4de05d7e38db393c121ecfa95ff9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877069245Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default, Name: checkpoints/896/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/896/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=31d77e0649b4197e5f75728d65c72074ee8ec4397de8108582163bd9cb8662394442dcbc320f39b88a29e330949657a74326ce8cfceb1a0b1a50c26603c4646827f9e291b5f8637013aad7b49ce38caed9c44196dbb0e2a6ed6cb8405d67ca76bfc8a96faaf587e403342e746b9ed67b6238fed0c1bc259b50be859a870c1d6f130cd507a7d7381fc2a9fb1276546266cb452dce16b48416286a1c1f47da4de28b030058e226ff72d4db8477c5240f4e81ea26912a778962867ba9b70e083badb535d241d03efee85f563b766c1a1843477e51a455211f33368eea8c59ab27f761473547c8153ff9a72a62dfb51d5ab00f26cae03ada2135982f8d90b2c22fe6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877088448Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/array_metadatas, Name: checkpoints/4288/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4288/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=8384ff78aaa4533b8e5ff6e6100319cc8d37f992ef210c0a783cbe16d6f725121ba7bf4a630bd7655903ec0b56815f27656c263a81908126a6241fdcf86de996f4658be27528f2f7a71a5526af2c53c3b69415402e00b8b4a8d911d52f913773d53e0cdeee27ade982ceb8f8bd8534987fd81b73f9d291c664b8273f1e5386e2eb165a937d33c1f63d8d64dbeecd5a588ecd1c45536a8f47d306476ed0e7a991705c2c24ffe5283044879a7208b4dd64d134877477cd280b306727ebc90428ab371882ef005c88f4759972927607e52723232d7feb058003e9376dc9707f79bbc7924a91715fe2d83bd9e5773944add21e7f532f8e31dc2b370f5346fe659ec4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877127757Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/array_metadatas, Name: checkpoints/3584/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3584/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=848c15fb322bcbe2e11460adc810f537389fbdb90a4e8cfa179fc7a86f5290320c8f7ccf4eec25ebf0ef091d282469f179bc08637a92dd122cdf855f04f245c0ca9984d3166c7faa17222029d9b11fc2e00c75f5944c4bea6ad49265101a7ef33fe5bf92c14d0a98cc7ec69901f01d27c191d49e5904ecde0e27fecc9f0c7bd93021750bae757c23e1b3406aefc1794f37a24069a28827aefee1f77a8221faf82b41ae4b1bff06245942ea7556aab591014fc3f2fd14aa5a31ffd983b47fc1c6a250afe395798e6b09aea5cb0342a9ba98ec7ca2b669a2648585495e62c1a158197a70b300be5c49694cb8cc304af46fc09cb8b414705848b86e4da797c8dab1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877068234Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152, Name: checkpoints/1152, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1152?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=14773ddcb51e46d843e40e70a378482e5af5c4224b368d5762c9e0df37b1a99152145c0a9b594966ede21ad68adb64c09de03b43977a39ee4f2396a5fba0a4c07ad800af8a3c99cde01d3e882d4e6593819b3f2f8ecfaf331a35992c4a73b5b000e50908103570c80c97a539267f9fe5e3ad87905f698add8ede6e786718b4a9c73dcb695300284eba2d265085dc707150c9542fdee1a4839935087776d86018d8fade2f707f2155635c2e86561b8303fa10cc9eb34e6d83561fab3968073cf74b2a976af2a96c234f0f46ebdf26ee730a4d73d05edbcf4e0e590ce08266fe184afb22c63b39af2e379a85641ccbe149780162101956b87cb7a09cc2884d6200&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87659042Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/array_metadatas, Name: checkpoints/4672/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4672/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9d5abfc6c548f0b9661893a31628aed20aa644eae173f5b5b0f6d9b50ff937c608acfd24ab999414c0e7bc966d3f7d1a26f29edc535bc7330328fed230bc9c67c94e7d081bb8185001eeab866a31fb0ea1ff3c6dc577c2f4ba8612c2dd1937bd9909ea012355c2cb70cb2970963680608bd9ff97bc61a37deba195f3abf7ee370ffdcbcf8f5593e1379df2a24f24544f1be669fbdd8a3c68cfcd284a3639f508a2b0d973c707c574ac0e96004a8984505e78ab1c7c11d0939a174d5e415e9a2f8674f0ef0e551be700cc85e2a0d541df9ba2fa8160497241cf12da37f8b695578d924806123cc4662d1b5e9aaf83690ee39bf0abd7051f350434f3cca661faf8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877134331Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/d, Name: checkpoints/2048/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2048/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=61ca31a65c43a6fb6d5462ae40358e76a144123c823958fddd08fafe8c922b43b12294b07cd567592af501ab5490163432d337fd87a2fdc4c066a4e4a36a69653233dde1bc85bd4426935600cf47aea299b49b326a63c1e3a317561b24ac3dee9f82f465fcba3eaaa2bc2a7fd0e76a38229ac4cd115d89b05ca777095734eaa1d88acc8b5434e7cf2d89ab541b5690f8b2da221d0c42bfbe126872972583b5c2aac59bc2520a721a4512873839401ed8fbaa83e5a82961f18e45e34ad537ba34912206743891c2183a59a815a7066adead5a00807edf232743eae985a96c576b0f2f237b06778806ab4456912ee036ee21612c9c9335de89d45d66b1bdd9ee4b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877220873Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default, Name: checkpoints/3264/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3264/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6014e87acd9c4bfaecb82a82817fbb9d007dafa58c0201da0f193be0f4dbb3ba2c333838f94b722eee758992d311962ef4b769d4e5a7af14990428a39e8a935ba9346c035e62386565fa01596da0540b3f2c5d8a3d91161d3b4433557214cc4414dc2bb877302f3812e23f67dd37dbad4e3d59606b843abf652d2e986051dc64d34d30e26cc912661c0eb7ec45dffd8f1dbdaef0812cd4f7e9209b627251c37f538a64a88107b7b2b5382fde194b4aeb40d3293f7dcc82a5e8d76dd65c13c427c43e18b785173ab34400d66af4e9c2ba5eb7639c5f270777fff0263fb1b71b0b5b7dfaa3a8fcf4101a718ef3ad233c1d6cd871d58cb0a325fad53b508394f85f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877237433Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default, Name: checkpoints/5376/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5376/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b6d367792aa2f048db008309ff6912bc3a53313ac2abd49f712163c9d860140d9875ee17128a08c6254464e14b6f0c7dfcb6b1a795d17878a1b0110fbf8d9e01e91c00ab62720ade87c4624012d686dce13151348f3d547194c9295ce23471406ed52ebe95e6e199e2732b91fb1244d79a8670694c2845f6f8c92c7080cff3f532e98907ed6edea9bbdb40a89df9891595bb61dd5cd7ec870f837bb52c83976ff3eb64f0a48f01c9d6e148c4a3b89f4f1ad409bc01855b89ef0ca99d77550f7af54d66c74fde41c85b47e2c6e18a9dd3c781006bb8143951be6df674f371d0e71a5f50eb164bdf8f2693610ddc617cb048810584832648fe9888eed2d280615d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87725911Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472, Name: checkpoints/1472, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1472?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=24c5c492b65d644e2a8f93292313026bffe60a21587e28692c0ec1be745028b26bee01f7069123bbbd1f427027284fa4ec3aaa15656415ea81d35cec6900bbd3d274e8c7cd49df30a3706ab805ed9f611d16600f6c6982da68a43fca171711c4174e29939bd2de2374d7131caf033dce7780e6f82289d7f8e5bb9229ffefe27672cd06a2dff1f112e6380626abf22e20ac96a23aa389a72df28e1951b12fce1bd498d459dbbfe91ca8bb05922289871cdd8a821aaab221fc26fa563ef0d73713dfe6a04f5731b3bd038642b246f2395d9a730ade097ac8dfc16fdd0184bb4ab99ddd82492a7cfa1a18a98010884040e4a348db37d338a22b2d05c786eeb41750&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876557582Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d, Name: checkpoints/1984/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1984/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=4b5e5bc5e20e86428d2d63a93fa8ee57f5cd67cc35c28b821583fc8a8535366f763bc5ac70197d6e08cade147fa5c6fb45f3c1a7845f0948f06a1376962f9a1379eb44b0d9b4d9834a6465bd4877f7c9d257853b27af1f14552c103cf79a52f89cb6ecf1b3861e4d862973d64790e64e680d52175c189749162767014c1876b992997458c73b2fee5624e56efd250be3ceb0d66ea4d26b6cc80c21aa5827166e60d673f45427ef2a9d71ca129d8e5207dfe266d5b22121a73f5e96aad06b951f82433b73401833ecbe176032f17fccbd082efbd4abb85c35a0665b8f410aa44d8d67b0c48c59e81820e0566e1a0a484b215fb65d6c3910f54ed868d90c696ce9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877295999Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0, Name: checkpoints/1792/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1792/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=279dbac3e24695fa97c8f970363f970a11bfd91eab9932b79984a5b02e5b82335800e0bc23e9ffa141dfb348c40c36f1b6d0553aa4608c0625fdfb26e29a334aa47bac6c4cb7bfe82e9aff8d895dd3719a1ef604aa146e7c247399123de47d245bcb3f08fc30ffe74b1dfce7284a349ddd30ab62deefbc940f493f3a999f16501b67fd327ed2f9ff8deccd4bf5b3f43cd413f2f29ac814ed1f343e4176f642777c3df3f95a5514f03f80cb82a9ba99ffc3f3f581be103dd52ddb4e089c703a2e43bdb7176d4aaea251a95008c9e5cfee7c391d95d3007291f71428b600737adfe30c73076c6ade513e80e0fe60c995966bc365ed3141c0655d42a2d9eb60e17f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87732034Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default, Name: checkpoints/64/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/64/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=8be36d2c1ce00aaddde58223c2498ab948b8730390dbad0c6eeea75472869fb40122025c08cbcc030b79946e82121236cfa974edf2082349022438f2b0a780b024df83310885afbcfcb6a21a93ae89ed5b1b4a6c2a474018776c19cf7172954927223884dd41781c424f037de6a92d3964c03aea1ebb0e9ea15f8d0df7e16e8011a2fc93cd462c55b9df169847bb8b41c7189cbbba1ba2cde022d741bac097dddb04cf678fc1b25f9c169a2907c4a55b73982a30ad5cbae4a6716f15d803b7d128bd2ebadc67956d9b4312e49546efe1be9f9e06021b03edf29dfc488bf0c8ecb362c9bdbf0670ef32874d17aa0e859faa35128558458df55043f406ce7549b5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875826244Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default, Name: checkpoints/2368/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2368/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=03da04751d0176de9205955f9f5d383b2c70d26c57b8d42429a94475c3786ab2927d4062b2187e92ca378a3444994fd0816bf59412296477c4989303e7781c11489bd9a9a8259e1ba2c7c33b7f30aaf027f03d2340ad70c1d10570b701b920c8d70d0f22e787a23814d7e6bbd926a39d9b495649bbe5ab21a9f91410a2e8179bd1c297bb4c1ec9c6bad145c74751d283a5fe09c2d06000e609dccbd5b2d2c55f0d7d7b4d1b9ab4deb99cce4ece06e2aa365fb791a078c7cc252f1946e1982de52a67e2e2bca216f9b636008d637686550d2e98e941be040ab16486965a927bfe530c86f3c5505374dfedca32a8ea930d7fb66029a5ea79ee4c59946a87082f5c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877314937Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/array_metadatas, Name: checkpoints/2304/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2304/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=0834c360351afef2efd2bd55ce35c1875eaf3903fc2b6955f5120b87f1763ba7db57e27427f343dae798acfea017cdd98d3ec6a563defd73fe53e462a0f76b0b829e8ca50979bc639a8f36923044be9671f8ae6abcfdd79e5378cb5376f25a34036a35e8dacb6861df38c285df3f93a7143801040856640e39133c720b0a821ea513c588c261779ce8c1eb4267b5188b297cf4d0800000a52db23bcd18eaa3f5c2f456f8df45895d6f0b70c8e34ec247338becd6ed8e12e89e30bd6c1a1e8374cc6c0f5eb25be2956dddfeeb8a13d8109556379f3f68d8cca8343f46210f94fecce889b526ee208fd3c162f07ea436c8192b369b9651b6a1bb9c919f14b5a7b0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877375118Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0, Name: checkpoints/1216/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1216/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=4353e7b49674ec9cff5bd9bce66219137070bfa315f121818818652e52ab6cb08158eb4e7dfa8a6fccd5ce61ae7e8ac194de8eefc88a803ef9afd5791432bc0ba60dc94ef745b549b72955ea2176f193dd192433e57cbe8cd844d8c652533f24bc9e803f386e8ddd09559dae0151499c741e727c01c8c3dd78400a102b9af6e3b6f57cb8aaf12fc65b879997ff256d8f3cbf075526ac4fa06e078c72681472e4b9019978c07d00b226381dfcbbeb8f823889b65af4ef7528918d51c9b2812e66b5eb4af33ac4e20a1afc5d397fd8c710bac641e47bb7a59799b262582011e7b8229306f803c53affc688162c0aa58c050a9c97470a989d0ee1461a224f755828&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877396506Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d, Name: checkpoints/512/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/512/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=967b2a91337af6be820f403d06275d6e96147ce4fafbfd23320151a410dcbee5c02ffd94527e484e34f9b19f82e48b284b858718ea0267a24a1efae2948cef152a28b097ff46f1760963b4c806167f035586fe5e5105a4fbe7410544c73ad799d7bcc4bb6540f30281fd04edcd50936f3f6dc7f9d16ba5ac51ea2a631e50f11d4885b05dcd7768a608a3b0a36b6d2ff1d7a7eb60170e4032b38791c5f4baee5f251353a8b82e441dc7ff59e55397a363d42effdd1d85515776dae151337914c1d29edbda0de318e030f321c08968bce13b58d058c6396d87dc667e6a0595b3c2bc07d7c1aa6670f1b8b092a42e1233f548628a59bfb3e947d045074526051958&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877402321Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/d, Name: checkpoints/5696/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5696/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=c14bb6505f7adaead75b482743d7e49c9e030dfbe2178f47fefc5a4dd7bd5ccc504031601a0ad7211442666d2a9b3d4eb46d3d7f812b2fbba5971cc7e1e828d5d02d139cb42466e3c0cf935d41a8f1002cb4dc3ccefbf1f9940e890a425d2787ae551d7de31a62405dfcf5fafbe1850e3973006e342b853d6237620a10e332b8df1f25a7b41dcff8a7051db874e445581755802b19b9a1f166d2bafda6552d3e5d048a227bf22f2912ec5596f2dfd2af8a7fa1a7c176785195cdb2ff90f6a93f91bc63f088d44e4e58c9ad18b7ce0238ff0394b49bf2c3f8a07106cb16740521353ade9e14fcfc27e84ef087c9a554e68aedcdb007d8448fd47ff37c6e40b953&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877414064Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/array_metadatas, Name: checkpoints/5952/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5952/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=8087266e090b37334e776b4873ccfcfb9fd8ea4020fea17b81ddae2ce5e9defd7e77a1c02321794aca3cfc29cdef8c7877c5b8178997104ff0483d966b1f10a9f7f2f51c95e811e41e736f56f9e8c3e165ccb13960982c9137f710757df1cb672c95472acbbf8e6c2fca8a39b4df424e04acd9757e5c63e8449756817137f8ffc77fcbf05d2fbe11feebbf2ac8cda6db7c03459517056592d497370106dd7800f36de03fe3e9f722e0d85791fcbf8ba8f38dad21198e18575df8d9a3483f50d455f89df79ff136d4cbcd02588e284d690873600f12650c29bdf83ff70311d4aaa4f2fb131ae5dd2a33af0e0fc0fdf574c8711d31d9138039314e279af0802cb5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877369305Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/array_metadatas, Name: checkpoints/1152/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1152/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=05566a5bff900fde53e5a8bcddfd2527ffb05bb9facdfa843a81384aa55c82920f9ba55b1c5059930fb4608dd1c3b45603bd4806ab821728b878ca40b10312e716088503da83d3f6b03f34aaf28c41638ac6383ae0ccaddd7b522bece03588704586a57dca92499a370d39d030f9dd5d09b720c6f03b329fd79b70d423e8ebed7bb0ddc58ed2ba95a23c5c9bbc74eeffb66434e765a020ec4a4242302fd918f021deeec59f74bf3c2c31d46ae5cd42622cf5d94754d73b2fda2c7281e0708e9b7dd52a9341a6405afe6321307049338dab44ba7164cb80b699a83afbd603a8719bfebf2c985300096c7313b9782ba984005df552a1291fc67789cc77d01bda52&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877404602Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608, Name: checkpoints/4608, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4608?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=4c24d7046e26cfb067b3fe41260dc684518b10eeda36be0bb0405cc3594860d840e8dc7a9783e3bcf6673199ae784c2fb6f83ffce4988d9adad4b8d680242af08fd21ca06731ef448d19d5051f55b7fb17f2ec1075abf5229db79a4a4387762a4e8b05162d6453d523b58441d643e109e07c658bf066ad22171b9d9a4f74bba3fbf1213782291d4aff46b8f6a94c651113cd8ba3e8b5356854f478b4a5237d792c2d38901d498b1159acecf7653e4df25313da0b2239a0dc4fc9798dc5e784ca3df71f9fcdde97f1618e0e3a14cff9c8f662428804fb2e8c48caaed0d1868afdd3dc2eedfb018ad7b84bada1b05f06b49f0ceafc68bc7087a5063c2cafb2991d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877167971Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d, Name: checkpoints/3200/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3200/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=119318f7c197512eb7e8356f1be189d8d9c70477c7c442706798e166ddd95d149965b2a3002e6a8fb3846236a9ae49afa67edc4f3f9e9c24fe1227e30cb8a42c64a8065402318ef36405f023da149acee235c62789e0964c67a0f28a22121a501b58b5b6232d2eb7abacd3c76958c8ae0a21eec8f281a7c97687cec42a49b5543aab8dda4c0a0d2696ba63cd6e4f1828a6f99f2d23660cdd84b07b70dd3529fbd3e50692aa63713384839b79270130dd27ef68dcf61670f569ed680ef021197d4276c964bc38f30ac93c2a6a3830d56e7fbe214b7087410f8c15a55374b33fcd8d7ab032f3a6f08259f4425ed4a00ea6bc19544210725cb7935b4ac9739dc02b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877058111Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/array_metadatas, Name: checkpoints/2624/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2624/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=5f5d85412ac43eb2ba4857a558627d3fd4acddc91289a8d45a18fd7978f68f5ab498be76d442d3644a6cb9a111d113add30fdf0ba8133068926d61b607d10d5ba6e53074aae5393767e79ea89f4f639891a244731428375f5ffe296027863f865533d3e1589164917a47274f3f45bb28e6041a0d06f156c841e6bbfca7264bfa095d924660a5d0c8dc8582f2dcc7dbca11c6c7cdc1b205b7a579dd7f07c6db035abe2c95dafba9564932c56311674452684a81069dcea8eeabf5d237a096adf4303844566c0f119fdb5d106c1d324ac6c503ca7bac0b54bddc4120a5ac3c94c1169996730e7b59b73554a5b339e578e0738f71f06732205231c947f017a3569a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877413227Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d, Name: checkpoints/5312/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5312/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=17f36e99277919fd391fc7f298b0f0db9d88494a9b5fb73c3572d1acced011a36afd7a256d96aad4a21b83b61075ace1d4646f1e8ed8ada0244b1941a9aa68fca14c263270b9a4105ae00dfc983c231623a51f2f264877de653a2bf1c460aecdd86325b9dc7f68f0b69042593d4b586bf45bff587d898286918e9631a2abc19ceadebea52bf7651e51453d8230b65e44ee19a66d0b0e0ed8df151f408b3dd21b38550a4394d08e36c7225e436cb0827ef779163eff7c082cec79fb343d7b4d67ecc41ff202c1e9f3bfd7962d1ce649d109ae21c8adfcbc64d600ebfa7af2422514e66e513d2b67b10bcf1a1f061f67583ffffffcc12c6c80bfaa93fbc75b6d50&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87717258Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/d, Name: checkpoints/5056/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5056/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=482add317cf0d006dd2009b02eeda2bfde14c65fc05db020752dd5e796fadd242d9880b6c5824be9c0e8d7855996a8501d17b20116e7e49df158a573bf414a66bc6f0f11940ff5693f3652dff364038cc23930bd9b43a59564c344d626af0e79a4193430ce81520e2aa05a94c9cb4b002070e4394df58c0ebeb62a8e1e962d444d9cc43540e8f252fe4e9f06988bb3c985adf35b65f031a48cd0ba04e26b66dce51137e7a6357c4fb0603ec4b6c81cd841478e6cc1be1b8346279ffb477f0fcdea40c9e69ecfa1bc17eb5ca5fd30ea215b3510810dfc317295fe2985073d2b51b3e391cbc34b82e05f3c00bece6fad2e4c56e3c36bed6d70148c6725375edd6f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877038115Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0, Name: checkpoints/960/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/960/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=46004e40cbe2dde543e60e47acbb37d094edfcbd855218cde892d105daa1b0c96b4c0273e5d4cc89b01477a99f5f609918a29badecbbf6114b5f133d26a75687cbe8471486eed0cfa72df8653ef22b0e79d6e148aad2cc96354fb196d3b23f122af4db7099feaa8a4f93792dbe3808123d7a227446d812adcbd6900bd1387b3bad9b797852d2d2db2d8f9bb61720992eef3b70b048be7b070f9513f8bd7d83113cc04f4ad1009e128524919c11b8486fa727797a70d4f1120d2a3dcdc669f8442366bcdd04f20751de5793b97df4a8ba0198f65e3bec7ad74727d73f5eafd25fc2adf06f576e0d12ff4796fcfc8b4f73e5740ce065df295ab8b0527985563be0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877418335Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default, Name: checkpoints/5568/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5568/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=87ac785d0584cdfa7d9d8ef65a8e941d1312d7602c90781bf4197a07cca1e313d7a18c925ab84df7ba147cbd432614257c4419a3b1c95f114e4e578b718283b76c5f9c6a48774531e8705f0f3364224a0849ded10a03ca5100921f452be4a7a969c7849fc161aeb81d62c87a4c35939c12558c64d7044210c10b28fdc313a672970b3d1442655dea59c73a7d1b3ec6311b48eaa0bd40f5d57a96e00f3aa15c34ffc95fe553e456dca6b0728ca7b3badb7f4555d13b4b875282af9a91ee1bbd66956a9dd2aa5e8950cbc643720270608173aad997d681e0d7bdb4904a8cc6d727b7c35fc9eee6cd24e4afe28f9e66d8940941a1432b40152daf972c3a4363b619&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87587547Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d, Name: checkpoints/2368/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2368/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=316539a7625bc584165b171f569bc66e4db9a6d437c278140650003994ddf981ab63f5bbc542c218c574a28eeb62e19b7272d5a3bcd7baae98960cfa60606acbcbe3feedb1f101f537d9ac0642bd5be52d7ab66e4245351b9fab0e2b05896bdade975947bf82f61d42127f2bccde861ad0bd6c443dc9c88bb14d79ca54a3c8c333c36abb55b17008a5adb0a3f31a00c65aa4f712f06840c9d8874b779fdc4063e11a213d3f07b072a03d30be12ee289e9c0f1f0067966f5eb00bbbfd6619f9bbe83d2a222b27298058835d05c011466f74c6e91303b4b30fd47024fcb83e58e167ff75ad91413b3c4ecf3cac60e9a101322e1ba8057b834c182ac22cd3f20438&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87363573Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264, Name: checkpoints/3264, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3264?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=84d562edc352b927a02dcdee3e0a7c4b4802ceec05d4e7fcdbd4cd8e4e1a7377997b1d6483fa7711b10b303dcc7ff9149212eaad056afe502da38a8112cdfb7c6696251e030db56d9849df1aefe43433ec32cac8fee94b5411196e2a3609afc824f701fcb836d4ebe014ed361fec95d70f8b79fc55f9234c61b02b3d46d2d2e7bd562705c1365d08b5ea1d7996911cca0ed41dc03ae2112fa34845f5d39b666410085a414889ab49c99eb1eec60e29a0db8ffadafb293ed13835e1efd64e64052e8ee7198bf1158cb66102bd8a9a76d304164ac010dc1c6d92cd2520eb1500d78eb9cc664055d95e3cdd4c9dbf636f6852fd445db547086c746e90f72088141a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875894202Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default, Name: checkpoints/5888/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5888/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=5171ec180e3c65d6095ef0fce443440b1ae10c3e4229bfa98666df8fae28a56cfd128e0ba8abaa5baba140fed43081278ad1d385882f872abd35c889d1822eb72d52083c51bf58dd4da2f4d8431a2fdf42d2e3a6979f73310061afb74a28b2a2a9dba40eec5fad574d70977cbff493097e363a47086b8673c25d5acf8c9b3a144a581bbdbb1358074833cd2d0be087464c5cf17b8442e7b47d2d324802fc5d2528a6224b926846739c4e2ba07b33c9588e2566f02a611b369fc925c2609efb5d16ada55691db703f0602488433dd6278314c3348d93e8a6c897f9b48a2a1eb5816a834e3f5183ff31e040d8d53cdb7d3ebf8fe7fd334fb85b5b04e271ec7316c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877182679Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d, Name: checkpoints/576/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/576/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=563af5ecaa638678a51bb1db7384f84d66857eadf25519c26ae9893ad8656c967178342c5c79637bb8edf0fc18ceed8bfb934dad90c55e82ed8249db09103b3ed99fa0313452d44a087d5f7637991f4ab4fe38244f03eba8e7faf3c1f8fe8911db66eb26dd20318d5108677b58e4d8637a08fb26400421259deb3c2366faa7ea186bf135531f32ef953624a27a01276fc6fa04539c928770cc82b4d3bb2c0c38902ea743f7a14f67882a9533cce71c25d0d48ded3e677f834f023eb3791b7ac6883121ef24983fceb393302f9a26b2718c894c8fb0a80e802a92d37935a8870a9d57fe173a81666c2d29ac298f709b1041ea1b651e6283b882c4d0213eb61f5c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875889701Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176, Name: checkpoints/2176, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2176?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=ae9fe7a5d51e8f4754f10eb192191b4c37555153a5c2e021251d491dce5ae01367bee3b9f157b6b2c63f81d2da5246bccc516c0474dd17f9afae1bb114c64cd4030a7c5cd62f3dff12c312752a8cbfd390cbccbde9442ccbe56448bbc4ac856a898a37bf160cf3e66e33b7d21f9d50bd5823a89c26e70784e07b0efafff1ed60b2668ff5fe5b00c0323d94cbc7e35eb4abbc6b64f94e60746acf6e0b9b9c45d77690c1054300ff3c1ec04a814f450b98309f4b5a4fa68ca37b20f5c7a2e91d4ac21aea525b284ef55436efc1dded33f2346db72b53bfa7cf0e3c3edc41e6ce7a04daee23c6122d343718edcab2db9897257ad8f333a6c86e7395afdaec838450&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877327431Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112, Name: checkpoints/2112, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2112?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b3ed502c754376105b4bb24d2c131681e17f9606d6fe66c45970724c5f0d6304e5d28f944611c0995c4286636ea94ed4dab563064c4c09f355ca1f823e1608aa0557f54837fed77da3090a046c2b72b2febe845400fe7e33e1b09188ae4cc150863363dd6b65335fff81279ab40afe2de34b7252a0b7e249d7a01e615be3f6040949aa6284ca565440c967b7690bbd1f76dad2d84b6ba595bf6a2d49156b22791812fc26caebcd9bfa968daf64ef48b83a3afd26ad64b014045e48aef13afe5cb29e38e0ccc683acfad04913b098c3bd95e8badd43f5bbd542e0ea2017604df929a09ceaa808a6c6575f6587c77a2fe48ea8e8d2d3e18abaafd910d5d5c44df4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.878288681Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0, Name: checkpoints/5248/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5248/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9f0e1eecd25bb69f2bc6af414e596fcfef1bf019ed8cf4b3e77af6185aca1315ba0daf051bbeb544bc4adb15713ec58cc82050c551b0eceb48030b362a5db97f3364f582398d6ddfb3544e7f75fcfe863ae9b1308f472c147bc621fff0790a2fc2e80b8c8b4827f9f1e13ebc534887182643652d6595efe82bde483da83389a9236493cd5c31b6bb74d822b94f866383c4e8283be1fcb1141fa4ca0b41f7829c05a71984214d2a1b1665f4681b34427a7cc1354c9dc1704ec2650fff0f92f25971af6380164d79246e0ccb82f4c5a401336635310b7c514acf13230a6e340c203f75d4ca839a48e6e9c04e6443f46bdc1ab9fba109f0118a990709de26816dc7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875910692Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0, Name: checkpoints/3968/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3968/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=5b81a5afbb76634f8aee472435d22c0f2eb71c9c818274788578ff516f3fed81255ee6f4a6e883ce117fd2f20a9bf65a1f6fbb54bdf8d1cd4d72cdec9564dd668843451399d79c7dec129b6142f8e8f85fef2acf98223f1cc1b844a257140fa024e3a3c4c609943f0846529d2bffc92ded82ad1d6fc0703a436a36e69d1b98cce1186050a9188f0aa2b61de2c273cebf0510a46fc496c62cd1cc1fa62f1a71cced105cc12663fac571c7e8fad280ce8e5e336f91fc4161ac14b1d2c549677f5d164cc8381b93d27e29681d50fd53ec189e4d005dee38f9064294bc897721c65489478e854a846675338f8c0c3f30e08525fb2475d1a0498c5776aded80b72809&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87593771Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/d, Name: checkpoints/4096/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4096/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=0eacde64325a21c1366141f594caa810db817fc60022791aa4c35dedbf0366686e7be994ff3044b4db1411f2cd6ef844ba4ed8bd6f9c6fe8be49016b742deb5106faf2196dfc6f0d47dfa8c83c5472f7c79cd7610a4835e9787cd82505167258b4479dc252bb60e54a8a2aafcfa42d522781de51d77aced3ef6ff108637e84e0af5692fb82fa4f14c91c97ad1729656b4c43ab3e2509eceedc9da357297dcdb23573126c9aeda854e483528df8adf16948f89157e17dcfa6386ab7214c699b79c404482dff476c89b4a19fbe0169822fc375a990d2f55fee08941538d85f1cdd43ad9c342f6d6a8f5e5bc69910afe0a0864a07f2f86b3a88372daec346ae4646&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875958139Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/d, Name: checkpoints/1600/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1600/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6cec5ee1dd934324f68da6f1c8cacc9c4ac24db89da8ac238813fd8f8b13f1b8fbe9d57055f24c84a987534eff214f5ea5a11f9f5719628d8b172aa798fce212d30e494715dd0beff035b6e25a16cc29078b7de0321d7771c45852aa886f76e9cdc937ee44e5179dfa7a2e8e6218217eb1262c5a88393a9ea65589fa108f577f1bddb6b33abd4e44c842ef5af877ebd204d46a754b289d929e2616352f86e09edf3861f45d47aa58d814762180bd54ce3a5e8b10090330fe2e6e7b6f923a750a169a8667090596bab25dba68d4169e1d34d0b792e8873eb0e26bf716a24d447da82675ad969b9eb49660a1e4d70a3e34ac6200d4adbde30b88a09213684ff095&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875971756Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0, Name: checkpoints/4544/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4544/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=27f4df6bd56708a10d3fa225d667df006563b42eb26f7807abc2499c9723d76c1c52ace7640fedcb981facbaeee35dd467bc87581668077e4b799968c7cbbb4ffa5aecdf4f266856983386d59651132d1b335d43dcb232b1e998191d5ca0c7469c94ddc49b80195141b32f43445e871f180c3f5823d3b5661876aa64668fc75c3991db8a5940b2355a277a4641c71b4e0cfcff993aebefb87d1ba1ac5a5c45d2310e212e7283a3a2e1b7804127a9b3989823471102c7faf9220fd66a13c490f9b9cfa900f07148b3543379e20b0478f480372caf672c4683fb29b312a0c313fed58c1282048abaa14d26ca8d83e0e16d006f350368c1cb9d2ac320911cb28970&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875965116Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d, Name: checkpoints/4672/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4672/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=40cbae200af09f456842e7ef68187f6cd566f30337f49ce1bf3e19ee7c1918cee1a675980a5d5c0b0d448d1c24732c304045eefcf6c1623a60c8bd485e94a7f7d647ed9fb733ec86ca3f115d8cb016100e0cc975d44323f4d3882c19351966cc2ec3932881a59fbf5f3b2441e1c940ea3d06da5350d15f81798bc2b1adee8660a44cbeb696befd1d05dc2b51284a8908c5a15b4378a1a69f193e98a45c72011b3a258e106de01d5f94b37a3d89ea771fd43e6f9de2af88683cef8741831bee742e52c8134abad0e66efc7d9c08a1b75e1dad0aa937666357e93ed4e906ae8af27eb36a8068cca01ab37c667e6a5914661d461312463316dd06ebdc4140282bbb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87598319Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0, Name: checkpoints/3520/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3520/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=3517a457c73023b50c133ec3529fa19eb00ead237388feb5430ce8375af80052276ff62bf44ba59902d896a62ceb990f7a72210a89e03b5120597a07213f9244756cf6e37f00b02af731dcbc6535f0f50e78f368d26d8502c70a0fd2574bc8f772ae832ee9f0d547fd6d3fc49dbbf0412c6503723c5a3593732e773299100e22c4954f8157d5d686a51a5fd2d66c9678faa632069c1d4f48d298ee0e3cb53793951fc07e503756650e249089c476363c9840c8716b8026611b9a6381ec8a0e0e6477f53bb2083cec938258dc0a48bb6b1b2762d77b58f9d8ab7aa1beb2d5e0143299fd27d981079af4c459819384e2937f80d6b9e208e2ed21704dbc830a7444&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87596237Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default, Name: checkpoints/1856/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1856/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=788ee231571a4a9009597628c2bb03f27e05cc426a0d6f416622b0589552fa7e92a2653f9e4124f3518660c41a201c063aa03de806ae3a33ab1c710ba8687333b6067b956ce38b3124f9542983dbefbee125fa573a4487b7190298fd82b2398781004eb44b867a39b4f109c35ecd6ce0c5d43785449695085107b230570b0136b5fd690e0704818a574864c2be7ad20fe570ad76a6eb472cc8f3ae7879928e4497342654d906ad6a70b640ca74ba9c9c2638d96c8263729ef9343cbb26ed25cbeab06a60277bcbf84615266e4b4d23e7acaaf1f4c8e9e35373b078065bf5b632e33dd869adaaa5e85dc1af1f3c7c3663933c781cdaff48cab2b795f5e841c77c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875997474Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/array_metadatas, Name: checkpoints/4480/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4480/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a3bb9c8691237ba6e529590360760db58982b0a2ac82c5518d579da852d6ef6d787365c3b285bd688850dbbe4969fd7013ef7f766764a54b4522c082b6b7d730f7abfaf61bf051934979752c39011f092a235f3524fe9ad1554617784b98b165faebb7aa1c283be18b8307b2b5b3b7e03a6558d5ecd4d63534e59aa717c0506268f73f64967e8951fdb0b37239dc60649cb307a078461b0420974029e6babf3ab880b634cf9f21c03835c61564af997ee4e9cf01552f5128ee2ec7beeed0c190733c115c7755646ce1e6aaa034dee00c1949e48461a2442fbe7248667221556d5dd175a922ab027bac7b4b446cde0869500716f7feac467eafcc9476b4c65e3c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87602934Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d, Name: checkpoints/128/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/128/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=c1c2a8813236098c75f507eeff04e0a5a584c80aadbb1c9ec51278825ec9c0be3d824d8dce051639054b96c848387969d388ec0cd9291a8b74792142d497b091b129b882941e56044f4817c76ba2f34ed44dec4e8aa6828e6b63d9dd4caecabaab3acb2b10e35f2fde39862462a204e9d6d1610b5596445e588384799b463acfbc5f6e4742c812c1588338447ce9fd5728160430518fa06347f836c4ef17fb52aa395a852572a63ea1c60cbcf9702d44bb6130d8a0859803691aa466d6ac352d481ac02a3300639a7f376ce992d008391f32d5ea01e1f23d2b50305168065f8b0eca6fc6576270183c9bca35a4a3a67ec260424ec1fd6d1ab8f7fcbcf7075e5c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876055697Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/array_metadatas, Name: checkpoints/1408/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1408/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=16713833c75506a9a16e9555e76b87432015b0264791ef3ce6d11c964edc3f6b1ed777a2e0502ca6b1bc335251741ca4e920d0415ec1a8d9b14f99b194d124b4a9fad4684e119451d3526df0191c968a62067a11ed5f232f9cb130f8b8a68b9cc0dfc70b0caae4c8f21976652d5d4e4539d322a60e82cc0c85849644363e3642f9e3ed1f2f51b1d7e76eaaae822c3190a95dfbea9c2f7936227fdfba04094db3e2aa97fdf2ddf4950cc446820d9ad45580f3cbcfc262259b742590ba2063033cc60b4f7812302f8d313f1147b653847418932714209270c046aead3dc81d23b791859832ef5558194990b3c9a7ccd4b6d8558d285c18b0ae26737812d9c4e301&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876056996Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0, Name: checkpoints/3264/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3264/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=bde5832bb2fbacb5ad81de8aa8c3e3f82adf0083e5fac277dc41355b2f26c54ee37be36953943490f3c3ee523e6bc657a667bcf3a6141e2ec7fc9575c6f83198bebae6dc19cc9c58e03601d4a43c435594b95299fa3dfaa72a82ceeb2ee3a092a29b359352fa1d4465d0902e366f5a858f262d6d628d53b478bd6e00a677c82b17016c1d77d6cd003bed0bc6740e4103853f3aac66386428297416cbd96b68a0039aee97efd9f6858ae021267b2a1b79656f068941ce23818a0ddac85d954a91b6c8c4e23e36d457dd8e806961de5d3c31d3ea8095a7e2cb50f3900ee99e3b3e9cee02162dd5a68456345f4cf565eded04b98316b1644063a71d33be951c3652&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876047366Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952, Name: checkpoints/5952, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5952?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=bf45c1268db7d5b5847b082446f17389630d2115a46dc09e87ae0f081355bfe1221d7c661d39da5247e5be9702109d57272cf01015d0420c25cfb48bb13c0a47d41e8d52e7b4d38894a66a7a7758643a30cffd72fd41a08b875257a02b2984b0b8b4c94bd9bec24a9257ebcc7a0896f0808fada164228df43616f1ac5c6518402121d903b215ce271ef834d76ad88bc519388db99b5f5cd8d5f83d1b8b678c6e77f59a2409f67e1c470b2fe55d99a14b539b5d66f94652ffdf3835ec39a71d4eecb6170a8532d46aa836628a09362dbdfea9bdbdc43abdfa58a1c6290fe057367ae3cb79843075aafafc033ff6c66dffc8211b2e6601a43920d03d753735a751&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876091852Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0, Name: checkpoints/5376/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5376/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=07f608af6126bc0053b4aef95c7d61edfbaaf2f64958f28476d1a14a477e5508c686574ce43a758ef041061955d8bd6330da1700384ab5d0b4bb043f82829fa50f71b776b564feb8568cc77c99d74182f28cc0833f5eac15c17d830a7b85050a95921096386b05699a1d6af1f0427ff9439dd1d83feb49b08ce59ecb2d6b6ddd090a19cc0488f5a3d1f67de24bf35498e12e8c3f959c8d2e4f33e02f24e5fb0707405c67192fdf2dde6b6e6bb2992e2641acba2b3ed235b8c49ffaa9ff56ab1b5bef137d62f83ea034c22e695211f68d968f1ebd6c7c345178e09bb6acaa1e4a8363127e48328457f11e147f7a6792c1168015f4a04cf90e7ce42674495cc48f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876112937Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d, Name: checkpoints/5632/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5632/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=0ce9d8d20a088035e193eb89bdfaed2628de652a4b1aca9084b173eeeb2526878d2730a030f0a3b87cd3c256b3368d40560f47a7b2e6fcfee417583391fc93b4b3d55068c551c01697bfcf957f2935dca39c02738352f4d0b1ef57a8a9d152880c642652a7d7f913f90558c0eb8612582b5dc7b20bb9044f24e632c8402a050d8e2a7a55837b2e8b0aa6da4062e306ce9b31acc113b3d9467ac9242e0f96a3374fdf999a14abf2c56867658ef2117b6dd4a7a60f1bc08e3930635cda5930d3bfaac827ba3ba1325a8afcfbdae1e3ae0542662b84fc1149a4e1fca85d6f2c69e8ba5e4cec7d6e452567fb5932f0f2954f50b9afadeb105856e0951fa4784f80b6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87609958Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d, Name: checkpoints/1856/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1856/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=649dc985e3ed399bf251682850dae3c0dcf728cc988da9bf204175f1244b6b34983260dfe770ee0f5262d46120f6170786a56dd78f04b10cbcd16e078870aef4899083a7bd5e512a7bd5908a3381a40759d7963a741723af412c563ada8450713e856dc73c1754359b6da24713ed1d4a2b09875aa8a338172a1ca6de2200296d96f013f5605e4627cf88ab49e8558dfd653e89b7ccb9f964a6800c81acd752d31ef28cea3f8903a1f0a4c8418be949beabf6e31599ee193c861d55e4da38b9d08274106099094f8ecb5ddd9082169f8070ee58696233dde2fc341e3cc9c1fdc05d841f3705ca6d9b4629055d1306b81807c31abd8d8c90e6afc9eac506db1ae6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87611322Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/d, Name: checkpoints/5184/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5184/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=8b8f8ef3db671a7f189f6e7e1eea206d129f8ceb048f3c569b0b42a5995b4a8424913add5c4f61f3a991e58e504e49a2f2d86d8290df923032827298e88cfd5601471eb057e351c2cdbf3d3cf386aada01a76afa22b7e6df39e6b736290aec98775c9bbb4984005cf3616bbd62c81e7570d607235cc0d0ce609b0a681ea0a971a0d2bfc01a6eb4f526cdcc01e071733dbd73c27ff422fa1b023a982f412bfb2aa5f7318ee35174523c5cb0784f991886bb943269800230b5bdfe359c45bacb6c9ee480dfb66344475598d28add23a8daa5546e16c4ea5347c499e044ad879448e5a1c2b3d9472f1f90bcb6d466e95d2d55ffbd815d065ea9539cacc7821e8c51&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876127487Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824, Name: checkpoints/5824, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5824?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=c4272d9bb89b02d042b3772437ea91f49813ca893752311c9609d46855e4d46bf3ea51b418e0cf34892192c4a30cf3a238f515368885d7a2daefbeb4565459d86ca4f59d25027eb29a3aaeff15b5fe70788f39022d7a963c67e219e9a09ce380133a048bdfbfcee76903ebeaf7108203b675d38d568029644fa17497d08e3e0cf2b2e909ef66f267dd01f262bbd3522955bc4df6aff03c94ba185854ebc8393d3c61aacae95e9c2ee75508fca6ec641aafa068c2bed8b2e4ba005b9290e9c823115c3606a0330733ed43f4de6a18083ee964d6218e73edd0a410dd09567667f519cafc9768ade6f728d206ad0c710b435bb42e9eacb255db0a60e4b472e25a26&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876141079Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992, Name: checkpoints/4992, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4992?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=4151860e986ddf3441343679aa120499554ec03530400bec1d30d49c5363f064c5a84bf9fc47e563fcfa2ec690a5fb88063f8f99994d4effa66eeba264469c6b05c403ab70ee418f52933f81f892cd495c1d4b44c7188598b4087671875181cfe2c43dc5bddf7338a17d86da13f0f20e6d0ede4e8fd6ed91da25df6660ca5242945e805e8e981c1154373cf2358b7e2dbfd0649144aa8441c491f62f7f02e50f1c55626bfe5177245b8a57beeb88195244a48bf1bd67930d054174aabe85c620f9995b72766f097097a53eb8bd563aa01cb4314c5a0dfc675780a800b55a4b9d5c1e18435bb432cd8b0667f898fd2132908c44cd5e29bc2f509a27be30cc203f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.878406522Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048, Name: checkpoints/2048, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2048?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1a50ffb79f832ba82ffb7ca7ab6acd03e800d189a3174eb7575d7564b5395c2ddb28f7c7b2b3c4bab5fb1dc1581188ff3d1083d2c112a32f25bd258a9d9c99170090ffe1d064a8b3ab93b67439e75fe48e6a688f224e25d07b6528195c35f2a7389e636b6fe242cbd3b5139dcf5b9d9edb7417402e787d0165b60b11727e5bc3bcfe390cee75205b384e222cc0e023e2fb7d32d3d3fe6c8d736805abe2944d2cbb571b5c9524f81e29911df084ffc399f57fc4adbdf5611674d4b97f1963b829c26cab0d4b395766329ad7b15723e5779337ffa27a86cb487dff41abfddc31d417ca7657109423bd93e69589caab4e4547e18e9becc463cfd715189606239348&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.878436032Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/array_metadatas, Name: checkpoints/1344/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1344/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9bdc92d09d07902719d89478c689a646f943b2a985a41dd898d0db29f56f473da5c7dd02c142f55a7bd5b1e2ab61344f8acdb669433ee8cd51fc3ffa19ba87c030cc2f847519fd633810bfd1b844b01efa8f1f26bea398776817a0f018d7efe311c34b5bfb3a5d82c479a937b59af32abbc9b2d517b45b0b6bb3ff26c881b98c13cdaa40d924cd5d29f58ecc99dbbdc98d8bc17631ae30367c880e55f68e2004b8418c0b244745447ac0664123f2249fdb6548d67acdaba81d02f5e281add339f892ea492c3f1df2dba001536051ebf29331eedb659a311a81938eca1ad60070d58038cb2aadebc5d30a8ee6145cde212ecc8f220cc921a186c8caba65216411&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876156394Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/d, Name: checkpoints/2624/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2624/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=829f5dd3429b6340c2b806795f23724e748692b219d112dd3defad74e8acf1feff025ee68a0b48114285b29ba35176cdb521df7a038647bacb0c57699e3f636a188dcb9e78e0fac38d576f87d22eacc4ab17a97491e75b0319013ba4225e4b72e03194301caf307f712193327048f0d5e71f7cd107ba6f0f6ba5cb1ed27373d5a65ce817bbaac99e1ab1b3e2c905e8834637111b1a0026ed18d2e717cebc7e2b89b25566843c521342313a18484e1cae123b9c10533dc1189d2684a0d3e25fe6b2ed164ad0534b0d8586f477f050e7cd5171b16a632a4cd529615cc3605a82a013ecc13f904dea345b42cac2680364271125bf7f6cee2dea0c4ca988877814eb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876178623Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664, Name: checkpoints/1664, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1664?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=7721b9b2e62eee14c735fdf56df5732e52c31393397c9c2c0f74a09ebe2e61d64db986ab89be4d0d571b9ce2f4852c30295c7a13672acd82a46618ae41db7514f51892b48b39cfc2d6ef91bdeb341807e253b89f0eb5103aae81c2ed3d933f83dbae61c8571d49a3e1a2ce1c8cc4c7b76e67a1ee224498e17653e00bc28b270f7b8e7fa7195194a95e1e2bc1c5a044aeddffa7e7be36c60ef675affff6757f17f7f1a6b10b5fa0fac398f8b8eb5a3d7575c37479faca63d3dde5fd8384fc5474106d87bd097893d8dd3a072f478cce55b9a28626a700927c78082e6fdaa8dd7cfc7ca4d8f50544929f875b791fc888135027a8533e5c51a5cec5940e9691045d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876192989Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504, Name: checkpoints/5504, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5504?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=5b46cf5b6dee2d9ee0614ba7513b60ad9bcedd65b27cb96488d53596dbadab8039e0e9fa0966d44aba89cc942307e37ec8e86aad9e13d0bb277e644a386dd48d7a8b309e90b60ec1ef18583d60df57b1324c7b861ec34ca9fd49f59fa487a874a545bdfaeacfae947a17ecb94fe82e5811eb005a1b145b453417c0f81c109c517ee08d469a5959ceb7d4524bea14a605743986021b853810abd1f1a54f2b2ce0a8031ea08b282236f8dc8c32d25f1d7139bb7a836651dfaf11aaa58ba67e650567705695652b4dd969a1bfca0255d8a43e447f2619a33dc16df3010f3344e65975b3a0228163c19b1a43f9577b967dfb28e81b86a8bc8135983e6c4ebdff2d5d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876191234Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0, Name: checkpoints/4032/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4032/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=5e926e26a3212ff9cc8e6bb9f5dc1e7b9eaf0412298c3853404f95fd8a061f42866d2a322fccf592427eb93709b2942556a030ff477f03424aa4d239ae0564d3664802958e6cb863d1109276eacf37d8f8b7105bf27982ec34e3c725671fa49248c28e7d415d3f48f67f8b4919310d68c55b7cf7cf5b4b12c597a59d435b5c2229f2af0c4b346fb4e797c35a7ced1bdddc83674abd312d480ce255f3ef34ae9a28dca5a28d21b2a17db12001787024fd1ec5acef42f7fa41194faab4a1cc5b2b156779683906033681a74df09de0d6e15199e80bd873cc70ce197cee026fca9e0685015cbbe5bddcdff191b7d5677294b10291cf286d389072614b3c3ace005d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876189059Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default, Name: checkpoints/4608/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4608/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b601c2d38a84c79ec56f15935d0bf77aded79f029c5fae4a13476291230951bbc9edf3d828b3ed41a605f438c878460cf39b222c1ee002376022b44b1005654fcd0398ef40c506f3b17cfa8260d79d9dd94968354a476cfe50a830ef6c24f1bd79f97d8ad9da12e162bc0a1a598f7e18241bcf9663a9da687045995bc7aabfa115f3c5e66ba8e97d79f9a6f223e1a381154f00b400d11d3b44eb1151e0a31d98d115a09bdb8a52e26c19c9bac16bb79708837b8f066c8d1c72bb9acaf2ba2fe10973db7f9a453ec9f30f7b4fe0ab9e67399ec623081585ce6a736378dcfdb143571ea3274111741e656f5bc455ad192d1d8b9e61422a51755f4af84a3ace8aa2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876210457Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/array_metadatas, Name: checkpoints/4224/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4224/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=935b424d82f46743b51d40d9e946f91efc2a10e847ad199622277d0d5aca3f1a7af05490faa3d0ef518fd4c75b8cce0c15f3eb77921c9132c1055bf71625566710dda75f007807d2486d1dbb8ae15873cb288ee9eda5209ba6b2eb27bc5beb56c0b0c7925119a1d4a9abdd67934dd414c5de03475ca4f071168e4fc3b9e47f4d578ae21e31cc721c48a10fe5a81e2ef639e53fcc9f5104950a4e3239848d9fcf99412128568d138aa8a86f91d9e9a6d0529ee808bec5df8ebd94a7020b1238fda34d2a3d0abebf5085bfc4e8e7bfda75335c19e69dd95b30c6b071e429f0b65ed0e7ba2ba10eafb179034bed3a0b4bd26e1941228b0f2d0c451a39823cf922b8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876210968Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0, Name: checkpoints/512/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/512/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6dc04cdf14b872983074617d174f2b2c9e6cf7a08fe141881351d6e0d1e28af8d570f8a5da4f4974ee66b6ceca1bd003e10b60b9b648283bf0be1e0dc43e431b93397cce3b5398a86950430b57a221d48049b3707c92955afcdc867bcf124a10e86b3bd67191867277d9130c52746f3ca977b81d2c446dbe73a33be5311481a6fd4aa58168e9ab601dbd260ce5bbf67afe9b1d378f1679fa150ef1b4e72827cbd534ea516599a837bbd06b11fdab4f97254a46d99705c0ca6af9d4cf694861eb251820a9eb5ce93580d0dc2e349e8a47eb6b49657e4a07f5a1fc095b444eb23be02a899f97745fbd0d11b25557606d8b40d0ad969806425defba680d242aa0f2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876228721Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d, Name: checkpoints/4992/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4992/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=11d33014f4f325b984426681a568e9395723f9ecd2d4a84c332428364e124644bc20b2b799e3a8c6d96869d7e7c68480ae2b0c1542ed23e8700b072b486d1dd945d1893469116179bd80bcabbd98657de691afd30dd1004457831d4337c2c186631a21571095813b504dd28900c0f0945776fdd485e74de38e4a2ae7025ffe0d8ca5d70a90bb0595f638f5348dee8f1ad6b7549a13d7c25ea141f409b8476a2647bfb42402d51ad5aebd3042ee582906c2b42e68159afa1da6501692f15f7f3ed179bd6d5aa2715eebc9dcadfaa90e71a404486914fbfe8a120b8f9624083915dc015a7a2e4416741e0437e87c3162668190f23ebe6594bf0e9f0643ab8108f6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876230595Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0, Name: checkpoints/1664/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1664/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=74e4a8d8743f0f42b1f497a1e5930a1905f581d02c409e8aac3a5df7ceda56a1cd21c4af9ad24dbea08c49ce2c946d76ab10e09a028ca72f5cbbc40d289430dd28153d3a479d2ade5785d69d4de5a427ec62c337fcdaa5982e9f807de3ac8f48a90b19e6af09d2e857435a2f9e1ea0563558313aaa4c122ded952df667a8a878ec3362b313a9a382e528992aeed0eb398b58a668467d0e479ffd2994b769621bd39a6fb915bcf42313a59fa10b6961863f7b2a0e94e9b968a639e5a75538896c0077895c5659e5d2abe62d391bafed93640277eac53d6799fba53e16d8fd98b2b24fe41d7b624185dbeb39716cc0cb6aef9f143bfc8a741f9ecbf2e513c24951&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876246852Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/array_metadatas, Name: checkpoints/1856/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1856/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1fb314ec9a9f137bf55e6b69b4110c8f6745e7aaccdc9e5b200406984f8fe9ab2f8bc3584cd29c12da111c626125f3fe788f5b86a707c2458930d97df1d5229e8b7fdf20b1dfeef5b831c86f0e86c7b0b4e8f05c750ddb14b9b9cbc2c994b4c1b54e266a5443b48dc801638c8e5fd3af509645b71ec3800ee4de588c919e85ef70806534f24f35a77f888ce9bbe278e2f552368e3ad43f0cff6fa4b4c9fc351fcd16df4f3521c47a5a44215e8c570b3c071b95c1b524c20f960dce289169f7d96a66b2ac1c3a124dcca579e8a701e4fe4ea095e2f3c249b0b20a304cf23624b7e3fd93dfeaaef98554e8f4770a740a14bfe7e3ad5ac5b8956b3c0c4bc422b46a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876229313Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default, Name: checkpoints/5184/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5184/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1a4a5952ff78ee69b962533250819b41298a685ba6670bae419b6ac8724236736c5eaabc35b1c537d6acb3ab269afa53e0fc6fa8ef9853ae777c86279ee0069106733b8dba8768a143469b6c43c0a109ef978f41fead4115a0f1668f506af76a95dd3c574fcce8c593aac112cef420a8c8863b396f85e72f56e92b646fe2e3484dcf0276677dc6c87343c5529a7c9f8a9ebb35bba57a22359ca99bbe8c5f22e8c2d52378fe083b7c1db7cd5a265c03868ba3f4bc92ac175248f81e9e45f410ce0f7c3526ec42b1866ec34366ae7cc1bd2ce97886e5b8ca365527a2026f2d48d60a4b54e4fc5a35717388ff3101bf71366a7e3fd9774953e5e01d9f829222be41&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876274452Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/d, Name: checkpoints/5120/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5120/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=105b6888e4f52f770a68ab2c9aac2b15ee2523857dfa4a840799bab41d83083f103a6d55de077f70d14f148622845f0d5888f0c83a257b3adb7fde9d6ab0623e2814296704ae38ac6060c8cb0bf9ea4308ce1cbd90ca1eb3150b59ef16b4d7d297e6d8147d0245a910a1ca851c698d26c6359521b1fb9093404a2e0dc92950499fcb38c1e0828bfb8745a5dae2a27112c0b1f6f8c704a81f3ea614363b151b384f964da92e4f2de81af4d5189eb4218b827e14048e38c6bbe3a72626bccbf824f7bbc86e173fca2c150de5c222be61952017890c824db6fd868dd2a77073f405fbe6af5a1b2bca46520d4e912aa7df29adc6c01cb207f8cf260d240cdf7db363&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876275735Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/array_metadatas, Name: checkpoints/2368/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2368/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=017d9ebfebcda7b2d8dbbd757b60f6736e0570d9e3e2304cd073e3b38bdd460ef1e3261d32d5d5f033c59ea33cb308d3a28c57abbbb587269cef299de8747aafd80c1db4ed6c248a395026cd7b60400a4291e5d8e1e5837be122b2e2ab551ac32c67e5242ba1b9a9d9a0093538230735785fb2524f501a2f1cc9e6e815e31b86ceca3556827f33902fcacc27a7852d9ae5ddea00b5dfb9f27df07ae71b6b4431fd5c88e03b0baa8500d0f48dfef9dd1ebc55e348640f3d82d608aaddb61f97825faf1f88fd27268aad7c692b4b3f5cc6b97d15193b825516037497a27d35a410f1cbffccf4650dc538c2be443aa061bd6ed8bfc0798af2e6ebc962bf4cd54271&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876275628Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/d, Name: checkpoints/960/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/960/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=8e64c12051dea4820d075c9dfba02aa00a06de8f79c453a1f7a8f28a41e0c5e88bd916e5f17df73da8cf5582f8f2b2d412a0ad4e6881a4ae56cb5316c8187f51725f8a111aab99ddab0b36107993350c5194a25a445688e665045ad0a01c1cf81ee14612d61f18686e73688243b0f1356658aabcf3239a6587bf3dbc2f3f76a0ce4ba009e13edcc9416e18b8930f7fa242c10d12292c9506b12e39bbc23367224f11641ba121d111f412f573e9adb325f2c3bb296d7141fb8e279d4fe7aaf5ac92c4c23ccb07a6ddeecd3b936eff302201e2ab7255060c52dc858f13e03db89fee7418d3e3b6021fe56aae7debd4cd712bf8674ba8b8b1ac6ea8d58f504316de&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876275074Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/d, Name: checkpoints/1920/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1920/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=94b7e6f61e3b355ef2531f81f744afb723cdad1e011e262de345b1a40c71b4303810d05a75da736e2f5bc8c792fda43732d0e7837b5bc65aaa54aadebe8a8903b1bec8569ec0e01823c07629ecc8ecb068fb0d0b1a376315c677983ced39d23c09a8776061e9fcc3b05b115a2641e6bf37baf5b3cd708e1817dcfc27ebcdbb64954f6f16abeddd01005ef88df864208a7cb70282a3f9b7cd925ce60594b58edc769fe666918f541ccfd8c64b704548fb205ac5fe1ee8e1294f3ed11a1abea21afa86b41b8d867ad85755cd8016a3dd9a9e8fd7a01c7b528e9bbabfd38ef52b0e3585e1a7e426329b96bc93390881b650e4a6379c2013dc6347302b0369c55600&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876289256Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default, Name: checkpoints/5824/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5824/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6f6865e112f56efb76c773454bd2a04fcd6c9acf5d3cc2d4be8cc0ad985108f7640208275fdcaef55faae701ae93aed48a740cd892d3a8a5855f17746f6961987502d3b02673b54be2af4cb2cefbcc32a27247c7765b8a5875d16d9681c83d7dfa4e10b14e82dd85abb46092d0dd0b83c2368146d41d108ce5364560b32d364876af8f7287f1f7acc6819410193c1cf4da6fcce4b6cec1e73ab7ae8fd5c019f9f94ab9d67b571dff8c0bd45d3cf32fa522b2756cb92f1e0b725836185f4da876152ef1c62b7017afbe127085eb90cd2509f88541d2938f9fa0a75db185f5eb0d2cad1c3aa7ef224c61695d6f277720588cdd19e8c30ad7a0928b83726ef3b761&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876302739Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d, Name: checkpoints/3776/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3776/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6f373206af38634991c8548513697d731dd8936e6ed061fc287e5638f90654d32e7125f23e9b83149bd0052142c197833362596b1edf801301c61bdc9494d3dad166deb2e02f6cb1de253cbc64622308fb44e811307d30d633d0db908c95c240fc791f4644b483b94357c9001ac21235e3784c66898e2393c7a869415f9db1c46958760b2757de67daed5e4c76d6935e18972d938389abbb37ae3338480fcceb29e487d401962f36822ec609b45a9b071c24e541d138f474c82e8275febbc33b751e3cd7ac99da15734f0493f4881f2b8f522d4f814b05a7106d187a8facc18605b9ab1ffa37559430f286e9bd1bdf4ff60873ba39898df266017f8e7aaf4ac9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876297987Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d, Name: checkpoints/4416/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4416/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6fe8d68f22b12ebd0b7ea18aae197c52cfbe55df2deb0f678054905ca6a3cc6cc32f5efead67be2536e3b4d21b5f90505df72982faa4ef37e82234f770371c6375ba88b4fcb5976751cbd6136c3b890233b30b17d3055c5644d39c08249a0f382b61150ef99c8cf5f6dc46d00c034bb0f030d0633fff5fa555546b6d7c6b9c29c8705b8320e4536ec9d3490b3824a0b0eb7a0468387a56fbbd7a90748af2ebcd0bec9ee4f2b1b942c4d573705b4a41466619894c347c90809d45d856165ab2247494776e469bad1cf41c662ce32b1fbb5f03c329a551377a99b0e4637e6f2b32c2cdfc517614e6abaf2489c25375c47f8699c56a960a242c78a91c02f5f31fe1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876305232Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496, Name: checkpoints/2496, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2496?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=8a4eaee274ee72f4312f07f5ff7464c4083a0dfdd4a346aae1938e4c9ad870678a1d09b90b91c01dac0645cf2e62812fa1c7c1f713c265c38ad759cd8495fff784eedc87af590f06764622d55f27da3b20fd03826524ca535dd314a63cccd65fd758cdb9f1d8b13a1c6cb1e121403b9cf63c0bba08a418894929f2f0c08fdd44b2ee50e81aad9b1b81d52a1dbf530f8c0169ba2f1e31ca68c2ac33e03f6f402fd2c86a63c4108766ae336422a7ebfb75de46b0b39b42c01a10ed5fbf5616455102ef39cbc35da6d9e88b4e9b46ead5affd36dec11253c06f9543e9174780bcefe2a70113b900a3d3685e20bb319880d455e9738c9bba156f3381cf1e6153d907&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876322728Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0, Name: checkpoints/64/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/64/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a9ed7208e0feff00bc35bab1ebb7d5f5c543a9e5f62740e7e00348d2a5c356684b74675dde23ae990b294175981ba6ec152daa01f7dddac0e76d48ef35a2df486622a71643278740e4c24718c69291762099abe24003647df21df17ec9c89dff10068a2d32ee0dbaf731e62e58b12f04312d97429b86ffb1c903e1ca9a27862606fd8bf9da69af9fe04e7445d16bd7dfd3dd813f42571292952c4cb653c26726215e13c95c1ad99fd4b5b91dc63023be215101fc4e0eb41f3a03f02aa5e0d899862604ea5aa6dc7c8c334d9fdeb93b291154c484b1d686ab9c6275a0726cb211de3f18cdef79cf9d81d137fdc604217768e9c7cc8adf31465dbf541155c942b1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876341656Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192, Name: checkpoints/192, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/192?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=59c4d8a4f17bfcfea52464add363f9739485ccebcca43f2efe7f0fa2ddfe5fe272e66dda77e7d1cfcf760cc33359a5d4f8a1f34edd19fd8a41674b4befcf3f7e4bace3e777aacdf9ae104fbfe30ee300e08aba059a5b85a00de9d9a34954dee57ffed93a58f99f9b6b00a00454e496bf6fdbb6b75f2edd8a075e6cc6a59435f74e3dea268d23b8500cb43c50ae6a86d1718742ae28912c143b0a857036af717d0a2db35be8a79352a1ab08eaf63ebb788d4983e3392e27783c466f92a5be86615196bf459a642a47c7f179d72c6c89f635b2fab26c61814f504b6c8c0ad130cc687d5a4ea0b4f5443b7127579e646bdd8d9260b9d7828432a3004f8b1da13c53&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876356621Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default, Name: checkpoints/4032/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4032/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=0a6d90c4a06a8d4a5325e111d6065dca92bc16945b5b8825e2ef83b8b4db89f489f560c1247410521a13b3c47afe2a3df24e49a1762f53741e87fabf7e28ab2ee281b6835d33ae408b35c6d155d720c6d9cf10d73685b17568ac2ef3dc2a25a14fc87ece73f326d57ba5f8e3d578c6a7c1b3ceb9394d8ad519c114a581291588d081d3dc94ee90177df28f7f704445961ec7527a47d1c2a183d55a1bb70256b88fa81d4c301bbb48f87e92cd2e1adba42a9b054048b67eaabe7efd5a675df907e9eaed3d1683b6250360bc22e275561241d164838821964a40e3220bcb51b2c6e6a0f3182ffd88d5d150de7a2cd2dc0d119b5fec8389b59025be2e76e9a51349&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876356623Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0, Name: checkpoints/1088/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1088/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=8b197ee9753a87fbb4adb25881a71aafc8ed122858a790000872f12db5ab4e8eebda6bd595de0de29761a4e2dd459a0cc763ad731ffed7174423aac86335f7dfe7ad7838c3fe61d03f57d5be291dfa8de7222817404e864081792eeaada4ec6b4b83f79153ec6c8b224f076d9ea57760bb62672ae073fb598bfe922e1f8b7afb89c45350940260c2ae44cbcbceaf27746207b2e264d671dd9a71b6ad874c26dc63bb7db727beba1c31624b73afd6d808cff0bbe80bc8ec6afe4d0b346df43009dcf8cb54f08339a880cdee856a714150c74b895d4820e16b8507f3723b2f372e1f67645b7277d0c601083ba872ba2dac746c0b22d31271916efcee05d3532ef1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876339049Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448, Name: checkpoints/448, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/448?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=63607b4dd606f1c5f8a6014111ce36f9431937f92f05ab80ea30f50f86c1cf3c5e7c0b23c63667fd2209f72c5a3181f99067dce4e8314136b3e450067683e8d97247529ac7a4062284057ace7cd697e81533bb93763c206f722d697f809ff57e6574f376c6a321a076661dfa06072831535d47294d72d5ac17677b9422ae7478eeecace89276e8ab6325f2671034adc1312dbd8c17d7e5273e0ee88a9c0edf994759ee0ac00421ee94830703f264af4c45fe02cbcfb583c715f0acddcacb2957d7d7e5c55483cf3cb65e8a61d152f129416d678974c4bbea42a69bfaf169d6c5334ccc1bd0febf2b55022229bcab91931dcfde900c25702faa1009a0e1e8a92f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876335254Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d, Name: checkpoints/5952/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5952/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2b71eda4c3628ff311c2a6369898d603d3eef49d126551502cd3d153325a37cf7c4242151bffc7aae322fb492b9d94701cada15fc7255c9dcdeb876435b5eb0c880cc3a4f9ecf5951b153e11bca6d39a6b3f489a3249bccdb5c97c732b720f2f9338cf683eef0c9ab5ee466b885c4ccc35890f256556707a434bb833bc82dff74efd45706ccc09b7a58a59c479e937d462659aa8b2891fae80bead32dba9ae9ecf6fea222c8c35b732e40428bf3c653625ea91f4177f2136d0b0ca72a674e9e786a72e86a5b462fc4870d3d945ed6f9602de24819e13254f2c8f1158f36d8dadb0c551c044b13de77bf405143f307a9419b9d9c345ad384febc384796b1a757e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876419281Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0, Name: checkpoints/3904/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3904/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=538cedca91939413c2141757fc50f23800156e7415f18dca97bab599d1c2bec30efd4086d8c02b964a2b2a9dd4645b57a58b5c6fde3ef1c6f73d220d9124d5a3995f3f4940a37dd81c0ce5634a161b145bc3c3c47485b48a0a49cf9282f0def42f9608a4178b45db753730a93c70af9716c4e14624ab61d16d07fa73ca722c5a3a806029a47c9b51e6fdb66db6ed5a4ebd304adebf88ebf2004da0fce51d3e13b6acd004657e63ae17bb8be436fdf45ccace7b58413cad0b2189c849077d09c1926654e09af3226c754c6c0819253dca883482f76b4dd1f4bc26893c248476afdfe048058eb9adf6c3f6bbd0da3cdc7d901f4bd74bedf7bf8882d9014cab01d1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876426601Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default, Name: checkpoints/4416/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4416/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=653f9c68e58bd52616e5064439f5d3fbb9184bf2f1855e4cf40bebf9483afc07ff161fbad5e067fb8c17a80bb598895e4a9005cade599519819ed4037301b1e33a8e3c9f1a784ae83ccdca8b448b47f590477bcbd9f88e5aea43d4ab344af0b0e93c7def5001735b9024844693dbda090fe04d9e0f4816c6d94a0a6b046454dfd389ed9932fd300af3d9c01739895e74aaf6cbb4bc7a37d104580c7ac8b4fa9d04e7c037d01b85b356efc053202eefbfcd3579d60b7b87e130e8d6725410f21be32600a2a77a2e5a79147d13e18d4c4e3341d6b07440b13b86d3a1986b05843ee0c8356cbd6d2b450568ad413b7e1cbc7b5dce2121a8f0d5520fe08242593364&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876423455Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default, Name: checkpoints/4928/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4928/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a7ddc37c9b2a537c75b506236996dfb0e22f160b489b1d2a0becc851798a6bda4f02bb42cac83c6927a8a5aa2556a8ccf742a95fd36c1262570383fec622d2903ecd716b15ac6a5ddd9c72d7681162fbcfb4b20ef001a47c4d243e75a0959c23b428f01608969dbdcf76e2d93a7b1255fc4c764711abf013bab9e474a904f942ab82b1adaa2869e1d346413c5721bfb64d1393e7774cfabdaa6dbb6f92d1922c908d96364c5f46c0102ae2ac296020c4136ad35e12a774af4e83a6f56cfa22e3873b1aceb794d6d20f02b6b463c6fef70478e17ff9d88dd0073b014efa734d86e2fc474a37fb92e41a4eee14d96d9e97be79431d9d3999ec4b11d560c4edb125&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876442338Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default, Name: checkpoints/2688/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2688/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=7ca1caca02a457e01690f87dd435de40cfb22fcbce15952504ec9d86e96ce4ede1fd98f9b88650282a913ba3e1634a515855b70910da06709affd493671951e7cd206bebc3b2fff7a544331de7c3e654f909d4c82d5a8efb1922ac8c7b6a9bb48d0864a35c262465990ccd4524683037841bc0d98bd5b3f66ae2c579ee538d6b2fa3cda240038feec37bfd069e181953abcba2e5e162da1e56f3d9cb27649e17780a4d54ae4d3ccd352f66fc84e25fc36ca02e45f9f3904def6a458a019d3f2de3ea7e21803837d05e0493ea050fe5c830ec899a92added2f2d515d80a83ffd466b1b6bf8f0492292ad4eb9136137eb4ef9540f2aa8512d3147e72c0e2d4120e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87644769Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128, Name: checkpoints/128, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/128?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=4ca94cea4f6f114c08f484c672447ac34e473712ee95cffc166dfa82afb0bd6ebd347663ed1c42a7a114680393655ddd40070ccadee72be5b56ccd27664271f6f1725cc1fb25871e0e846ab7dbd5a9214b5eb6206688286a2dfba428455ac973a1c71b7f9aad767c3000b906ac4cab016b2460052d104c54fe56b4ad9ce318965e657125878cce6686462e83d3d32318ed6c8d58ec73a04c0b28fd3f2121e9efcaeff6ca5d9955187b17f793e9ec7e2fb11c624eb58297a547e73360e0c19d450ef4f891d35fa6d1983fdc0a295d94889b2330c6e448a28aa10cd91e844087f28ef6593822a81b870117aee2c9da88f30ccf8fe2624fdaa062d74fd423ae719c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876437386Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0, Name: checkpoints/3008/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3008/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=06a0e4f5bf7ea81fe3fa7007b86180dd87992fbaeee6ce4a5e77a50c217e9eeccb97e3fffe52e212444ada4520e1256ec718ed346da7ceb77a4b8bb9ea1bcc5e40ac3b9abb6d660c8f1fd6b3bdd5fe49e1c0d2e9edbbfd4557db31cea828507632aead0d9b9d64107fae08ca078fb647940d2fd36e99a4d015277071b5ea5f7ae27ffc5f0cc31e85d1ece3298b5f6461c961f9459e70adc3ddb72fbf4844a2953aadc8ed651d02d5beb839c48981f519591d23727137b0cd0ee63e82b9fc57b4d3138a752f41527323d157c8f76803cfc3be7d1f8a4ac07c7f6a8f7bf4e14f4b9fb053bf7c8a289aa920a42857b78017e4aa48f08eae0e7ed42bb6490dbc1a1b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876450749Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/d, Name: checkpoints/2112/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2112/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2ae535f345be6a5b1b0ff64a9881330fdf09aad4eadb8a50e66e7ea35a2049dadf9360872e2c34920ecfe12850ddef450c4babdbb704c153b507ceb638e225b91b3cc8eee36ad01bda4373f2e4d1dddd6771a0dcac877d94555c995c8f8b1d6607a21a84eb475f1535ba7457d8bff1a8de5923ce0945131a7fa56469f7766c5b267ae3a5c81448158fcdd14a7e17c95933f9302b3613d986b0cef983e92f41fda012ca2ab19fc15a693eb77c1d090b103cc671d4e4692688dff9ee0b95583cb3795c388a43a905103b241f42d221edef8613915005d924ad3a257666fd937dd2fd32375bf34fdb693f19d3cccbe5418da2e45d20d481a8c1ad108708d5f60e48&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876446244Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0, Name: checkpoints/768/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/768/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9820157b91bf334fe9e318c7377973381fbcc94a8d990aa502e84e69be5d9403669b1a83ef6bddce4c585300599679a5b422bc5fbaee5deee5760ad42d7ade28dad8e3e5a35bd93767c98521ca6e65b6b35653cf279f745e3463799334e01583a3637f1e966760ca3bf4a169d6c6735d74978f88d182baeb6dc8d26d28fd3c72dbce5d53834998c6e1da1b311095273a3e6a827f53fb3fe3804a544042926bbf7ac762233d5c980409d5ef824984b964a4d23000952f79cb9418805bfe0a6c9483e14e7a2718c9e342b19402cee4b8ab5da43fe2fe5634f85232a9e4f32d87a4bb2de590df22f0d0086efb1e864acdc958d544262d26f2d9fababd205a1655b4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876469516Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/array_metadatas, Name: checkpoints/2944/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2944/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b8a913d20a31bedf06a7766b7d5551710f190ab00160c59c6070f2023b6413784685a9ea47a3d37d343ac8de2501ca64d6dce49edbe3ed0486f89ecd1c9c37df90b8c5d8e68905f788c59de63143575ed7550f7bd59b78459a03de46953815eb473cbe4affa19b4b0f6daa04a86e09bd204de9229b371ae6f3fd088a57191e76527dd710c21c40671fe225245ebb57895d8f72ee983f4bb2bf7cae1fa6481b41b6a4348faf449747516f8daa47dedb622652f8ec0d56938e242e539803d25691727c7bffdc91edb73cc593cf8366a736a4d6be51f1dca2c35aee398b11ec5eccc5b23968aaab49c1c4743fa20d5bea165342ca75bc5f90cb7eb775eb942907f7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876457794Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0, Name: checkpoints/4160/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4160/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=90eeef9e490607447109150b59ac79d83b40aa084f15b9c095c95ae648e20b3db5f7f8b7323a933dbbfeccc03280cb2705517147c62b9d63689b7bb117a975e999018ab79a6b6863aba2993a222c998f7a7f2847965f6205b93925de069da39a4ee863a7d181a469d623fe2fc73b76b7ded52c3d25741a58b56accd3730875f21710e7198350ea56638df5b4ebe0728dbc5629621f8416e29df1b252a1221d96ac6ff59270278e8518b84f82230f2528120bc1a22a6059692701566babe5551cb4c6fd79d78b0d8ff5f674ecb1457267f6037c519c063b8bda2ae2f1c3d0a4e202ac1cdfd86cadf15a0cac095278aa0255a90212b8c2dd8bc984baa4dcced897&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876476147Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d, Name: checkpoints/1344/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1344/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=aeebd7c28983e8aea5727e7820dc287afc25fc466b42ac25fb91e88ef496fe2dc8fa2cf3e586fe0c26235bae939b7ba0c0a436d440272618089340500c835dde181bf375dadef1c2372026d482faf03c0c7f52d7a70f430e11b859f2545f4aa427f93772ce6edd9b17d43aa6265a3dc4a4e91a4840778063bb5b9e73d3b00ce9d60f59aeedc09e32b1fbf2b00fbde486e5819934b24a7074848f410a0291cf67071dd27ff038c175f51dc6925e635dcff57c0121ef57185e8f9b8752ba913a15821f50698cfc0f741cad737af27f513948febe96746073c06ecee7cf26521a7ec9e705e0ab140502ec9cc286dbeab3adfe1ef84fe97f8fa56e8b0eeaea886889&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876494161Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default, Name: checkpoints/3072/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3072/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6ebeb3510eb723e846911d8b4b36fdc0858d91be415f2106a244563ba88db3878b8f757e7a20e6e6e61ab6400b059f456feec54a47f5ea8eca41532eb299f64abc14d842fe14d522eaf38b7a97757797a5679734342597dc2c6cc2ddfd1c5b6ad78a9e34ee057fae5c43a26ab8dbdad4f50d1285c8e8e4a88b33f1b9b3cfb3aeff008d796408c2268f7fa15eeee59a61253f6cd3252bf1ac5279d33d69b519abba4466a4f10ab04c2fe1a28251c5d9daaca98d59237aa4d7296cd5809b8b75458cc71d92cd06a97a77f0389c2b848b6d4f7239be5e60926524832a986be6a26e7f2ad0b192aa61a1624d99fb404f574f1fea755d5e84acbfff2c5e5c0d89965d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876488797Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d, Name: checkpoints/2048/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2048/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=3489823a77ed83ec290b7151046df40a5ffab407b37217ae57556b8d934dae05362f085055ee19d4a919c37fcb2f58be7074bf4f9676470438153b213149563a2e1871b2c1993f53187596ae178e06fb54a858e6f0e83d166fb6427aaae2590fc5f809e0e39bc53a372b06b3a04e1027bf57222589310de1aeae8de43946452478e177498474173c8875e2ea73dd679039fb10d494e501c3034c8a81758c422a2ff212f439da84fdd9ca0f8fa94d38e700ca8665019f4b3660002bded4928fa316a80453e418d1c0432514a96c268be167fe2716005b08be9936ea37250ded6d5b8507000731538a08565403e615f060bf2fa8aecd2c2b837fb22750345e8ef4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876484388Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d, Name: checkpoints/2176/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2176/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=22d99b55637af0d5e3ac23859d78945f19b8ffc0d9369160b8ca20cb5e0056a2479e264b951c5f7a17d9e8e4b236fcc45bb95b4f7ffc9aabbff68f1ec67a9dac4ee975f37b8d82c1990d3c71ab8cf5e2bc006267b39224b42228015bf7aea766b35f96a6351b4ebd638834c0f84fe6d16a6ad9cc5c6c993761f41c8ea1f4d77e620055e43a106fd8b250b6a18f40a5cadcbf1eccb49b4f380d142c1bc1c25f975f354f0f4ce7c56cc8d98d51300869a6edc85ead9a5651b83dcfa8262da84e560e9ea98894128ab24c7b318fc653a20baaf5381f862e1ebd4e7d19e808102da704a96a32fc565c07964c5e65cc487508188485c152ef1f468b7a76e26b1680da&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876514659Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0, Name: checkpoints/4480/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4480/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=bf5fdfeea99c0806226fd328cd44e29f5e91caa4510d126192aa9cec1c6a8e0194b095ba5bbb3f408b481b083b7a20ec77194fdfde27c077c22dca4921ef041db3b53fd96f976451fee920778b9446d91ec50b5fe1888416b598409d3563b6755c8747a10c745e4a225331d9a9512a0b21cf52a6620b1008396f83955984517a12f4153a78d9be41b2178900e3553ec784d9566f67e2c8505d6b4a8047b16e6f9ac0f4aef97f8885807a4f61d1690cace4e5cb178f7fbce8addcf27053e682d420207588d976616f3f45c24d0f5fafe63f378e20e0e2b0881ba9325488cc89c73e04efd02e72ab036ee27660353911b0bd63c10fa38d006a734af8e2cebcad4f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876530003Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d, Name: checkpoints/5184/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5184/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2ddd7372ca42ab781e9eed6bd827b5c9365b997cbc38f945f71e3e740468b82ee816587387deb4c9fd8d8d7ccc5f92ab8796f0809b654552909fcc9e9034db760344c1e507480aeb164b570bc6479bd8bf1342c5eef7c86c10f565f60752fb533de3e6a718c8be7aefafa4f84fcea76f71ec2777a9962626716d0df865f3dd0d3647daca45fa2ac9fbe8578c0da2259aaa77c5120544e8b82fe673948b74c9fc01406ecfa8c0b0ed422e4705cf9f599917f44a15930f2c32512269c7b20ef12f0fdd5f96bde4bb48f6d28e9c236e2702c30275fe80292aad890cf1ded424c4ccbed0c0be94a9d0487f8d74468198900eec85cf503e59909dad385c97b3d8dba8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876536629Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default, Name: checkpoints/2304/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2304/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6458d8927dcdb4fd45c4d59d5e3717eb1ff9fad693a596dfee425054fca0c1dbdf9e9d0ac43800c1aa2ebf7fbaaa774059a69e5e592e54832c078618083149df3697a41ed9579f97dba234015c5c9d7ff2d6a04bdb8d1c7f9b2bc2f7553fb60e82477b7b70e605365357d6947aef2b2ca6a474e5868fb7970213afb09fb2375194db6c75a5ae196e6e13d83adf6606ef5324e46a0d7fc9bc8a9448ae9a3be000a774d3053c536d4c870a66e39ac87175b70f136917c5090337c17501b0a9258343294c9c7afb4a390c3e2e115c557fe261d31b7c712a4b79e2de7c95cab57f49cc258e4ed8a622053d67eeca69ddbaec54eeb78f71bfa71d75a7499d741ec710&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87654103Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d, Name: checkpoints/4544/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4544/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6987958fb7ac1cba8cd3b31240a6050a63a9a2004a5b20abfef1d7c21ef131ae30dcbecad7e12083fa9d42f212bc1bab0af687219dd4c159c95d26bf9cff144ea3b178a8598ec96d61f9899bb2c6baa9261eb53d437b159c64abfb73bb7337401e736250ad706c97848fa42566e466b3d31102f4e610c2f0cfdd229e64f7fc130ce961e9f5a4142bf91f6630f6635880d3f2c2eccf7aa68c4affd05a59abfbecca26541ea7ab2ea43e4540e33888b6b352e67b23260e7727e33c326f24add0b10ac23ad5a5a3c0cdadabafaa54689dedc96e6d4e3df446cddefa2aa88bc37d8dccae4f36de0b0fbdac2f46d20759e5a2bb6792c82d04ed7a2c8c2aae7f2d07e2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876563302Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/array_metadatas, Name: checkpoints/5888/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5888/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=76e7ec2e4576c56d3d3873f795b52d65a964a59a59683f80265384b653fe46a66fe105da944c0e4f43d2885881b995217e9d831942283809069d7aa5cfaa4b180ce3e3e234be6dff2f5d3513f8d4581032aacd84c29f1882e4b2130540d656744ea3d1bfd15439cc1d9ad26bebade50b1f9595c26e00f3fb88a954939ac8c1ecb8789ae8adf9def1b604a2302fc824e5bcced6f9d9e24da3bc5f51f97daeb16ecbe263685f2f0ba9764b12662b4a609f616a5375f37c7f7ac55d675e4a46413c8207b4bc2bcad65e67ce118fa1d275ecf9e539dc4e848e5b9a98a314bcb5ef09e34347af5981487eb6eb6da06675e53fa337e9b26125230ffa8fba4465d51225&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876579417Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/d, Name: checkpoints/5440/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5440/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9430ab54f9d3dca79fdd5be68a31317a1cbe9804d7a6eac270856e9892d01065541e11457049a2fd3667fd714e64dd0094b4f3a5d37ba579bfe53b436010e40ee18a2150b4148220ff0b33136b63032533109930ffd341566ec42f16f9eadf0c136161e315252f2f6acc7ac0d6bf65738d6b0d23bcaf098d7c5e316383f349ebbb9230bd23b122f5a421ee69af129f2e1b12c0d81a321a873a1645391a5b848410d0f6f36661e5bf890466f480a3b0a504dbea8f6589ea2d66c074b3db10f8657234fa7297291cf8ade6b04eec0345db0a3aede6901b8c5fb261220a68eb4d46525e7c48594e9573d2d1349037525eb0d1e5c6a65147ca1a9a3e914579d36e85&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876584091Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768, Name: checkpoints/768, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/768?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2ee109f419e0dda1c7fd5c2a053aeab478412041e7b2a7a84f35b1a3bbf7c1164e3a1c733fc44081d3d4a31e7dbbe29081a40aa52e9a7e17ebb6a0a5aa0eb36b55a4cce99e19655a85d43ee21054689484fbea1f1ebcf80a1984099d4ffad714ea6315a9b1e23fc64429da89d419d1981ab53546c4c9fb2163d28a11f7cc83be1dd3473d842b3ed70026f604ae5bacb582e28abf327be22fff29903e377d26a040dffe72af29b53f9d58ee5525c661ec4e44b79e6e51c94131304d3be9d7d9192f6acbc54cb8961100b7c4d8a2c664ec6668e735312a662293d77ae7aa05c3dd2670021180db5f33d1eeba57a58959598813feedef021e037485c52e35f8a941&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876584533Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/d, Name: checkpoints/3840/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3840/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=436072db7ac3ded6e52bf1d88ba1166b57b5652ce7e7fa77f3e48f271d0908029a37874b7f18afba5098136bd6ce46744cdb66fa3011e00840b68c3f23b57cc6939cccb4ad6477ab473d0ca9d73f087bbb5d8f2a4390949259046e88d7946829fd84002a0b20141ba69e0399ec95458e657e8d4762456d08cec8d0e2857df9dc560a760db38505dcf2fe83d1b71eebf762c9f0bc0f745bb16c82505b67204e4dd59ba043be21343bff1260a6424ccdcec93999fe08e593e4ec671d4a11379fe7d04cb642ffd34b6e9885a3fda5d2f275df5048f7588cf59e85635ea3fe51e2e3dd8441d0d971fa499f2b9d187818782f2b77c45abc29881608a348f648e9d1dd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875771082Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default, Name: checkpoints/3520/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3520/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1bb0f7650cb5a5b4e40fc8b26391cb2a000e6fb156706923b0f0c64fe0b085ed0c2ce9e4f84dd451bf785f96333982fe04155a2f057a274c595168fd62009fee628e2deb991cc48f9b191e654cc6727e582107e8a75d22228b9eeffedf6596c1b3bbc8dc25cf29714200123a2fb16045f970cbbb15f81f6731ab9f373ba8e2a70b3a2b302fa4310901d71a69c36b85ae6285959707ef96ba0ef4895173412ef6971ba14414a8f80d16fca3f968d24d3013cbae52a2f2f9696a57a48aca953e41120553e01013878b5f8bb8766ab297be109b933131e8d7336f89531902de53050b3aab9d83c7ddb7f4619c5e4a562f910f888eb143a181d911322731b96a13d5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876627174Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default, Name: checkpoints/3136/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3136/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=aec6adfd525bf86a2aed08c8baa8d806df5c0a104673c52556fbf370403bdd19cf1bbfe53e351d57d41c54f48aab96ab93400a540d6d31e822f6fa1c310fec1d07337486fc1b91b23694802732e36a65313908ec43e4a0dabea9fea53c6af0b79ff066f815c5febeca3478414500c67bbf6f80e1e39559311d77967ffeca69b2e6007284a301f146b339ea041320e6c5ff80d39246261b15fdab038664a65d30eaf19cb2e5533b1f2334162f6dc4520b55e4911f558f9357b22d9a1fe4c4f7856b4719a876100e9d904d661520a7f8becf695c4cebfc905628fdeefc63648f051803e1f7c5808f36daa5bf8437252a2ed8d2262992c36710713013b375be7a03&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876635134Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/array_metadatas, Name: checkpoints/4992/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4992/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=41074521e8044b6f02817daa7ac1ffe8d268cf3a64d92040e4d065947222fe0f51dbe1d66e6367fcfe1ff7fd1366f3825eb7212d8a04e205375dda2ed092e2f0e73cf30165f48dc2a08c0a55832ba04f87e634f12e87fa2ebd7dc2fbe7c40de0047b4a4474b0046c0c91b35de1e36d9278dd5cc5e936406709fc858b1aa1d61b181b811e6238390eb9d9e6d94d614114f887e3ecb279d12fe409a143b68ce6ad9bcd32d1c2b0a7761f8b95266adbda4d6f9282cd5950d772e1816a8b8a448eb059109bdf17a6c1a36d41be83604025fc4157f3a39350b4bdbd9a58cd853d010c841ffeb85e0a6fc4bda3bb5f5a6d6e66cc573be8f3cc2094657b940fbad9e3c5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876628956Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248, Name: checkpoints/5248, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5248?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=c15e253655073df7089348bbd2aa09177daa8829e05f5b2fae3ca878d059ac512b91652163e937aad8e5b1785266cb709c8fd278d2f71c95c3d66f300e8a6b0746504f5a22948c05ea631bef3d43d5756fab8e2e5234bc856b8c05ebe17718d4dc493a6278df876ee5150ae012ace93a80d8f273f0394222453c24f4242efbc3313079115a558ee0cd303ee45deac44ff095f05dcb85a65a3b2dee7ba9e7c14b8d027c732116e291302105c240950b315309822c60aed34e263a94aa1a05155ba9a8f6f2d4bce173b8bec937b668e92394fc18f17575e3ca08dee6b03f655e60485d3fcaabe96e6f1d88ae92cb8f0020ae23650805eeadcb80f2ae2b66d8c91a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876635289Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0, Name: checkpoints/4608/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4608/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=27c048f62be8c1706e4333565e78d448bcdf8a875b2395bab393ceb4408c8da048372801bbef0bab6b0207c5a135ffbd59146f43fe45284fe1092a9f77caf3ceb79d19be99f3e91222a76a6c99574adc67ae17ea7fddbef12dcc7184d7c29fc4bd6a7bdc5d1abb0afcd5a51da650281868aea294902710b76ef6b3396d4343817f944c5e92bd5723e2107efd613ed2b3877656a9b3717274ff7b0d1b31a23abc3d8cc9ba96048a9b1ee96c22d0155c2d460397d30618a60d5b42ebc3d2d52afc351bbe01601b6310ed38d2b34a47fcd69d62f4f1ce3eb55e0c4597e55ec0ec978cb0e1c56a70f1931d568628ced542a079cf630fbcef0db6910a438686d8cbf4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876642483Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0, Name: checkpoints/1280/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1280/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9aa3d6c22ebadf16a6583649a4c376b186ad8e499c86186c67dbbf2f759a2021aad79d585000ad3dbab00b3ea74952270bac62cdbfdbf816a51fe847cc0f993a6f5784ab6974af180d383f5c49fe3f9e98315f4a24127c6020f279cd6d063abbe62921815d2e41cb3453b4d914003700a392f21fe4fb2ca83c2e95e4d457cda2b5350d3703e1c4325b329d0f6e768d43b6e5627dd0ad44a242d51f8dd3f1c962c7a3f765c56ec20d83cb24267d45e822b786e4b9c44ecb33a81e06f44e4080dcd99522212f0a0d176f7b1737f2579287232cc0c60a87f054c8d4f47ce586ded7a681e400d7ef15a3cce151251f366bd1050add15c56162315b95292f5f4cf117&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876647314Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/array_metadatas, Name: checkpoints/576/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/576/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b933b04a1b5b5585b2d98b7f59f731dd8882762439d400eab564e7a1b66b95f62a8b8bbfc78da61d866f06b930c077c9c865aeea248b8cbab81229755eae9a85bd322e36dff4564c5ce39842940b720a704d4250ccc6bf6e8e8fa576680e3a220afa5387bed56b5cacbaed436a5647b23f001c072459ad9d00a8b82c0565a4f5619278bce27d6fc1048655363df893e0f48b424b8a8e84ece5fc0d0c0b019bf7290a64d48b2c2e46711f8b2d6952ff8de14da9c7729d793b2343781d64362c77d558126efdc44a3790f3a18cae3379e378db0504959e41f6c7cd2aef0cd9d5a7df4da0049ce1d5d45e512833b236929fd5108ff7f2f92b1ef49eac62cfe4e1e7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876659212Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default, Name: checkpoints/768/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/768/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=0596380207c39c1692ab1e6703678170b1e629c358a345b0c90cc16e8355509abe41142d5f05b77eb0ccb371b449987cc68428557ad8a5ac36e30dde11fcf5548ab00d320471bba599dda08bced6461b2a58907ae16fc2855516914ea525541da4e1859098ecb8346e36b11af292498731df7906bbbe1b64ff3b54ac9720e55c87150b5480baff64ed8c175687acb5b95ec82b2d5f348579fa490b9102b5e004f2b529dc46c8f2cb1424b8feebad98ac576e4f1ba3a50353a8e2feb590d73bbe3414ceb69a6ee527cbf12fc43916b97133b00946dea614aa174fef4442f7a4c193b7033278b85a4a4940ddecb7a58c5efc92221617405fc2319537aa0cdcdd25&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876664154Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d, Name: checkpoints/2880/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2880/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9c031f62a031af30d513d55f83e4817f4f219779ca3c4a44e8b2ef6324576ec3293b1df21479f808e04e397b175c800ef7a44c3114cd702090133d882f743ad44ab55ef9105f26afad01240678ff451f935b3f2385d663a129deb591b0aa3764da24d450af22df2b4c94c13fc02de5491d7e1c39b30831785e2752e122aa7f1535fe680f1383b200facce1a18f4860b346e350977c296db57e2b7875a3eb4e4b7751f8a057c4b68b88d94c8ea947ca84ecad3c51f5e4440921a77bb7f7b51199e314f5594879f8c438e2337df7636fe46daa36ee5e7d934afbb490d9389ce27d87dbccf6fd230f072946d3055f50ff7e072bfe1ab13fde6653b9dbad633a9cba&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876695306Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/array_metadatas, Name: checkpoints/5824/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5824/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=0ea2a788f6e528b6d591727086ac94cc6d1385b15c6ae97c38a4541f50a0037eabb90a8dcd692e1dbe4d8734ee7feef2dc568fb35976e954addc5cce15222cc2ed4c13a8d02e65a661ce12ce87f2d565bf786e3684cc4ae747a1ac928b21c5b2b576befb087033d18932e23c983a4097e07b1f0e2dc4355a6fbe00eca81bf3074fcfd3775333f9303048cbcce17923094a48a880f83022ae129f762a7cd3dae91d264a203778811c5f0d11fb5bf99a8037b08c6896c41e865c9c4279020c1f1e56ac7786bbccebf57f29717f60cbd756c3f587ef5fb42768cf4a9aa971f034b984cbf6bcd3fb164e04e9ba9fdcdc30a4680111d4bd19a7f60d2c9a23ff525673&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876706159Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/array_metadatas, Name: checkpoints/6016/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/6016/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=339316e0c779d67db114f09697a175b76d808e1d7a327b719a1b208c7bdeafec5f72a13c00c2b7fbe2fd8acaa11bbc32f463f4e583eca976e8f251058c2780e81ba6be8c079f28b7f558e5a9b989a8d714087bbace2f30d5b93a68ddecc23dcfd64255b482383f1df14f0383d4bca943c85d635a6b3a82d39d3a65bc87bd9602418ac989691e2aa5598b9b781736657378e4d91cc341a7d14f43fe31b92f1334e62fe4d867561ce6a3db80bbe716134ce38aa06efba8985e3a6514045d7be63bf4a9a5bb5fc82953d87bc650cbdbd5400a0ec36e650532719081b21afb4a309f727b7636cf682394e530179b46d7bb1475ba3b2ad51988bc42ee6d357c510bb5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876724867Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d, Name: checkpoints/1280/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1280/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=c1857a8214458f6547c9c70e3ee41f99a9aa15cd110f1c5338b3f5455b6a98f0587f59a3759cd798bc50880e94a83f805ab68f3b1f2cef1774d310ed7e019347ae77ef2a6ea98ab37e646e01d48a4149fc095c8729098fda7c53a492a7fb987201311e67f30f98eaa0ac532f5821585997e8b47dd4bf1d5cd0de5046c8c4e050fd5561a922714a15de00b1b0a36af20d35c32d8c88ebe447b2cb8a1789b2e08e4c4968e700a0e138686806661736271fb57c6826cde13cfc924ff3722db4b8406b8ed592bf7a425fe174d4f6c8695f6c4b6eef07cd45c6d5d2fbb7a11dbdea9c551e11046d4e358931c51bcee7af4b8c0daa4d4fd879580391928e0d815d48c5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876768453Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0, Name: checkpoints/5184/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5184/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=453c81af65731de747bcde400fe589fdca9a5578de5344e088500da1887ad095de9875221fb1e7e3ac7541d3006c0b4a85e3f66b6c863b85a744b17560da941a7f6a63b18fcc092282680ac6c3d5d69d5f0f3edba11b0e8c339df773f59e3888be8a1fff6887d45af3d6ce42023d97ba5ea80b4e6542957944683dc2f1ae94e62d4117352b87d0c2e6cae09b8781c8207628de14e45912e47d96c14da5f45fc475e790e0d2da79aa51e7bb5a4772a5133ff56ffa4c32bf3e5dbf2e71b27f57c5dc31c1b55909f8f204f1638c2ac31106e2be46866546dd8b6e5b95a969453a14aaf90dc7694024012c3ad40e080e89e9ff5387118748aad47da542d533852ab9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876776435Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/d, Name: checkpoints/3776/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3776/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=3d49ede71e4275ce45fd9a52231520b493aa63c458d1b703d101c9abf499f1ee7e2b80b8f5e6d38ca6ee6cc03fb05f85069e7c6fd1db260df4cdb8463ffa664cffb6209f380151092e0e54e6f2e34ca5daf4e1593a48b78602c07e7037ff8454a1a1f6b13dd6379d58cc0b4936e8c7fd2041354bfcdd8dff1fc5e5ac9cb4271034d6219253daff384cc490ccb21af05a804c8432e901e43eeb8ac60a617d12ca49798522799b62a632fb07519d953e6b5fd6ecb2f6ead32b85457a27e5ea5c4c85c56fd56c4ee7d73508c095857bb9766d4edaf66d7b97eed98382330efb18c09750c9c7ad095504c19e220cc166761b3baa4e12ab8e0c2cf89fb84f932d7d27&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.878972486Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/array_metadatas, Name: checkpoints/5696/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5696/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a5703a6630051c50033e204517143ddfbf8a51468212a394bcb2362883f5145ab225129276f40d59358bcafb898b65fa71b688b1c3357da0571cf826fbd01cbea612a9c813eac6ce7616f6734aba8394215ebcf9fd0da8da3c66a57da3534b38eb5518e7f1e3ea81552b5ff9add74f7a511e0e0d8a4bc51bf52fecd77fb3a4a94087d35d0ea3dfc0b06e95509849271de1c331f3316457f4a290979362cb8491bd374e1371cbf35cd0176ab5d8e529b72bafe2240458a5385e26fdb702fd3c318d7b1b97f4d5a98b5cfd1f1d42a9df46b670455da25b48bcdef893b56d4d1e9aaada9fbca3afc32c42b64a3af303ef386d4f43d284ee9fd5c37e7a48c07af00d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876810316Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d, Name: checkpoints/4224/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4224/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=7a3cafc3046e5b72f91313c41ba2f0e2b3b9077c64dd3e553199b080495f1dc2ca90585fed53aef3743b87867531b5499ca82d1f2a64e84431c9e97457a9cee22b5f9536344176c3cfc08ccbb1cc02edf4173f38e6f51a49fdff2eda11e8eb83bd26026e56b515a380a6570c284f17e9ee3540fea8d3405af72bfe6ce71b89daa53500a5065dcded66bf32f8d6aeb13712da45d82bb6bfc2c50e2c6b2df57ec3cb4de8d569316632199a60e46d78a05b0259be98d0a79070c7ecd65a19df8209096ecd62542acd1705b4b4ed5f5e1576955721fef1a910d94d5a403d8ecb97b0f5a00d666f7cb26cfab2ad17a6219f25618610bfdc809ad936190363f5c31580&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.878967941Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d, Name: checkpoints/3392/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3392/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6c477f24e0baad52c8337a38835b3ac8348036abbdcf8dfb36ae21ee542ac7b56847b76a2bd0eac4e55c7621a6e00ce9a3c94bcf8e3b3cfb96ef2b33e8d4c54259dee4314d23f889846c714f5c1a9fad5cc763a05dbf70a83a8a2a9ff4ef3876af199d3356f9502778e204b0360a6d639bf11778ba2413c48d2ce993e1ddca6681fb12a6fae57afb51ba42f1249a780aa19e3ca127624254bd3f4bcef7536c4d966cb4f79976e4f39e6edaf91f66684d57559290b9ad24ac621348d9fbdb37bd5ca6d167d747435fc1a5ab96b914e8bcf13794002842b9277abd13ce87fc5f7c157d516a8baa7946958ce9c52cffafcb105a2d95508a832f6e01d5b0b199ee92&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.879016584Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d, Name: checkpoints/3136/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3136/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=5473ac5e714109a8b0620893453c31804c1e6f357b8f9adb7140fccb00c69051a968f3a0a027bd6870619dc9f28e25b6b65b29eb1fc38bbf0d519b9607233b9c1f2fec0f58290316a77dcac462153f5e866816b5959aa75c92f66221dd24dcf547dc7279b4608d803d4a814bc14083e535fb13e51c65f33e4ec6f8e636f19ada2770ef59e96071f0a86d8ab520f62366455bfecb435218c92d2dc0c1936b9f9f40deb7f7f2ab0d7bd5235a5930a30923b98ac698484691dca3e7702d80202d9aab07f0b97bc54cdbd86ba6473758de160cfa2bb79861de18db3ed0ddb8968a92002d8c71a81452c9ff85049c4c5fef2b02848fba206ced5aa7f5c9969f277d93&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876850518Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d, Name: checkpoints/4736/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4736/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=600958ded07ff68f860de7f2481284299dbb12fe9cf74c225f9affcbb3f3bcb6d99da11cf9a9ffa91853e308925bc3e59a75fc2f30891731f4941f78dce2e97d436c184aeeaa4930f366300f396992435c89490ceeb0fad9ae315d5a5df6b7d6a1b322d9f0033d13c6a8716a353e619762c76b8283f8788ec39751694d35477e46d72ed94bd5c363eea1cfed172485fece8bb34aa988ab0b8f293e6cf2e95d102d0f88b987863a78420f402fc082683e4095866ff4d6b2b6093c669d5f6d58e1c9653a9bda56873dcf9e4eca4dfe723a004d72bc6ef6d6c980fb728c58ef1a7339ccf0492316ccd727ae3131965f2469402ea83f467728052e3318416855892c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876852596Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/d, Name: checkpoints/4480/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4480/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b93613a2b3fe0da16971e5e38bb86b26a723c2e8763fbac42d1ac0c7b6d9820fc2f8b8916848f90256476d38c3cb12940c12b4fb39f7006ffffeba999096774cf21b94ab0932643d1e4d5de951ddbf6a392de324afaf62108c8b1ffb6c8e1dd39a7e3c5e8bb35c5982265447680e6c477b7d709934b5c04bce208b02eb101495ec523023bc4bd8bf0d01d85377a5e29485b34db1a07eb59b8138d0c8f254025ee27eb00bc0033a34a7f616139c9b59c5588342eb252e94dda80250c5f39ecbbd0652ead65b1d6b680caf069a1d1fa94d05cb715120b99cd9512ae5d8fa10e331ad1967983c1498b31b95cd08b62da323dcc8f832678194b1e437efa71e70a474&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.879128247Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/d, Name: checkpoints/2496/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2496/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=06915efc8bea5aaf8ca1e44a6add839872ad55c060fcc4bcd4b1f91a7291d6338dd997ee935aca3156a46f18f1b7f42c65294ffb35bccf1144e5e1017e0a6485dcd4fa686fb9abfdf4c7d9bf2172fec49bf70cc82dc68322f04fc8716a792c4982e331e11922984f64b3417b6bf9ea6fb31e3348a973012a790e403fe36ecd96aec5185981098b353e5c2aead78d29f39ba6851e31acc4a71a57c22b154aed78b9512b89cea6712ebfe9e703dcd1d68a5b2d8caf1fc2b81784afd9f6d0afa299de8e3541c429874e0281a885dfff7fa851bdfa55f8d22043f1431a08e8b810e83a77493616130c3dc92d166265eb4e7f3b9a842b2d0f66ac95d37d7e20f7950c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.879208149Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408, Name: checkpoints/1408, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1408?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=192d8aa6b1824f047b4f91f7ff0a2d95cb4cf42b547ddb07c4651e635757cfa4175acd773476a3c0e28ca50871dc4338d479837ed1571a0dd57ed8cf46d6eb5f590c09f7f54353ff284588c85591769d4cf7cbe5c1b73f58b08a49cd4d5c27d07b6e4a49cb4ce5b01f86b7d34158238ea2835aa2f79816975853bcf341f37f06b059c2396f1e90641887352cfcb4ad12010ab70b8d2aa8b7ffd4db3fbed8a01ff2ee9cb7dae4b6d8dc692a3e8c35dc369da1d4e81cfd555f78e093696f6eca9dd7cdf0fc8b4b39de966bec5850d91ef31d06e826715a0dfae66f379264276fcbd172c5e77c74397884c97f6ee2f2704d1557d7befd6c85b4dcce26ed6c85d4de&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.879229181Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/d, Name: checkpoints/5248/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5248/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=81c9dc3a392ad2253c710f8f527ce457297229f9c783a04200a53bdc6d893b6d7eaade835bbd3a77803c6c3033c841e19547d9702519220707d6667d206ee11acd319ad1ea7bfa5edf766faf3097003a2aa6420e6b583e94b806896cdc88214dbb88234d78448104550ede1b3e3ba6a7f44d7457b579b796a99f4cdb8524b10911f205252340e04b522cb5b87dd188dae255b0fe4ddcc77480102e34fc6543160c8e8df1bf5520e008433eb89827ee5827c5ae7b34bdce8b76a3120f4bc6970c5807629b90a4d4088bfc6f72bdcc9d14c782a03aa2f1e5d4da88a81c64f961e86e97304a231dee6d1d5f6836cd559d4dac41b731729495a0116a60be5aeccfe9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.879306371Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/array_metadatas, Name: checkpoints/448/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/448/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=772f362d7d84914c67c603b4d06ee56529584b9ef8c353873d2cefd451e956fd5c4740a210c353caaf6ae34a1c73389c1d57bdab542229aa3c57aeefd6b014482147dca78636e8d63005d4aac957c03df712f24bc5020fcc24a29923b1740454371ccd0378b5b5888c95b076422b7ec610352b602929dc441f8140fa575fb2ba1d221a3c179ea5618ecc9360b6a268d67ba44b3604d6ce738e748cc12541533f2e917ddf280f23b3acb07f903fc2e9cdc769a4909bde2898f6b397ed0bfec0c765a51bc68363ce649a626712b8b57ddb5b622092714b3df321b5eed869ddb38dc8df4ba917e97daa97c2d0ddef6db7f5d6f97849c87a6d234342d81a04cdecae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.879320941Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736, Name: checkpoints/4736, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4736?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=91f4981dc92ba1ef398a2fb58eb47151ed981dc42ebf73a834303ea6f593e129accdf7baf31d8a21d4447639968d8ffb8d08402c787f15ba33302264a05989ed23ea065fbc0cfa0a767bb918929c8ad0bc47160c19d3cdc88a4869a4380d4c174e94f01f02ff3d253a1eea03144495913bb16e56771bbf6a422ad352ac579482407d0a3803e5953ea26d34373bfbd89dd6bada9022f072248b00cda71aa6d0766c5074a113c151852086c661d177a982512d1eb8c007a9608307e2ccf98aad097f346d25cf4e5562c79fc12eb710351e548c6ca519ee55a338024cc9bc5372952b748f9fce62ff246ea4a396b052696d0878d389696415f51f18165c80b6ef53&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876901561Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default, Name: checkpoints/3200/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3200/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=292570f8b1540851bd64cab8dfc190a053772fb8aae2dd8723e16084aca313365f8712b4d8370da3705bb5caeacf903938f2bbced0300f748ee1c806097233f3e4b796a052eec10a87191e001ad34fbff9ee0f5ee08b0bd402f27bf147717b0a5859913c092cda9a2ef2755a5c69c99479d44d046fde4ed67db205cdd3d516ae0bb5124167d7bc130f5fe10a3438e37cadcae821d37598e4f9ec38327e812696149059d7af2b8c82638280bff49ac3a3cd7b02d6b13b42b646ea3b35a48650028d6a5264b79e9cad17f9b329e0c0b8e38783fab87a21f309fd4ac802dbf17d46084aea6c4ab23a2bd80cfdc8364cb32770513de89667cbbcd0d8a4160206e789&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876889923Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/array_metadatas, Name: checkpoints/704/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/704/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=383f8cc752f0d823266680843aa4f811bb06ebf5b6fb683ab476d600264e7de958e986b9152cd47400572e6694e682e7b1b4b29776ab0d92d00aa377f8096fba9fb8b2bb12a31d2a39b78a67f2c2d8002736e48664ac88fc8112e806d94f655134f7b2948427c46637200f365deeafe1a0ef94ecc86287588931b3f7179a8278c5d1e07e369d58ad8cdf94c7e5746c472f9eebe709f73af6e86820f7c7273c9e7648caeaacd0da6b97c2bfa8b37d16444dfefa72173c44b92ce97039b5b5f637dab96451b6150793b5c2cb54e19ac06ce4da7e3e3b4827fe3d67b5d1caef38fb1bfb7c2688913ad7ca328c74fd4ce75a87e5204659a88ec1e6503f4c634914ad&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.879414636Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/array_metadatas, Name: checkpoints/768/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/768/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=15b1e7faae1b61c7c2a655682bed1a3cad3690513ba7d7034ab609a419f3ee70cb930da2e60e809d0d698c30f8467c67a03cf7a73b669e40fde063c223f2da286b26c33f72ff9d5b2153203ff5605feec3944ce287c49754120538e4694799351501379d64ed54273baf9f464814860156116c08ea7348c2c1f42016c0c8004b072ab7b3daeaa0d228c08820f086f152c3c779ad93b1e393c74b51889949c9ef569239cff9bf8e8f47377df1eee51af5b62e10ce74a8de88b44d7412e4d39ccdf89acc907023eaf6bcb125d98b7675f3a89127489e150f4406df2a6ead331479f5e4e870b4f9cf7c373207454dec8a39049a540c828b2d3106c05e8ccf400717&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.879406076Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d, Name: checkpoints/4928/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4928/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=5701bf8da8280df3cbf60220a71623d19d18c792437935b5ed0d5430ad261dab0c14087350fc912ca545f7fb4cfd3e8af652127052d1f8442a400306fd8900600cf13d0163a9cc12045eb8a0e5176f6e9b6667bfd7e4b8f34fc7e5071d2a692006216403fde1af640682a37acccb14e5f94d42d5ef8db372c9dfd7bc37f8873c71f9c0be66b3772d95c500d1ba008e5bf1528c372bf4ccd46e4111bfbfec1a314cdec20e265e58af2d4b6dbc989434c3107989849e4842fc62e11f481d3dd27489e4bd675055a93907598447b86ce58867ee58f2af818e830520e2e8fee8879c30a24b9724176b8229fb112508b7f1c7ea3a2bbfe8c19809f0e9254d7ce37153&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87643698Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d, Name: checkpoints/5760/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5760/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=be6f75963d79235c4f87c5c8f1b1cbb65c3e805833cfd829075d881bd5be18f5b4c1dec0ae503fc507e93c9db6dc44e1749c1f9f29df9e57ceaf0e71a3932f9c8a639ddb44f26a3776fe5986bf9d9e6621e25ff46160fdd0bb0ad9135bd441a27a021447ebe837ecc64d62c5703ed662ce84c3b6f003b94b44c394e621725b9278951c1118d6f8107381425cc78801b4d6629838b773481684586f7686da7b5981cdd82f31a0e41bf75f0fcf223d24166cb27c0b15a4b32034e582783f45b07af60ecb6d5f240edffe3e264f08581257ce9fe51a415d104fb0f73c2f3551c0be542bf8c42e641221709530911d2fd98b9bc704a84fb5357ce7306fd0985493d5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87946413Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/d, Name: checkpoints/384/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/384/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=76b4fcd0a399e4dd8ad9163235d24ef3cf3099890872833654645f2f5b4f49626d06ab5dc768bfbf3e8add840d147fc1ae514d118ffc69ba8996b7cdb06ec067e3e23e331dbe6e858be03ad7b9124b322ffb01312df2d4dd2aed9cb4b22eca4c81af0533586e292b80edbd0d006355793fe2272d5e5a2346663648854dff3b5f59162896c5e14140e8135c080fa127bf2364b9c6e8d82a7c7907f3dd6bb1ef869cd5fa9e6f9ea3d8d567b647af1545574fc732c0cbd5c4804dcfd9f308b06f87d895ca3b7f11dd70f4833751a12ed0966764a29bdbfa739c6265a46069f3f4c195ff4af3268997486080d32c21b4ff0a4f60f6741316b7f84055c84f15db7c9c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876900514Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default, Name: checkpoints/4544/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4544/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=88f5a39e1a12447d210d0b2c03ae21918a2b7fd0269e37bce382c3bc9be8c81f02aeac6858cf7f51568e7b3bf9e4d3a474498e7abf309646dec6afbd25cc783b5ba3f59f078a6fb4290e2e073995bc970530dc8006e69ea99e2b6c6859d64a550e09ea48bb882a3027349e62247f37e8122ab3ffb001f1b0bbd2be594344da5cdea5798e15693856ff931e0f921d3ff685d3adde78b99dc2a6ea20edf0c910ec22575efb5e7bce5ae639f85eb62709e6079b23aea42633a4dd2085e581b91b1dddd39e8f42dbb533253217f4028339e826b685d5a30646a920242bd736cab8e81ebc06bddc3be1a0c47e15509e197e6998fd1daaeb9476706ba350226786363d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877440323Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/d, Name: checkpoints/3264/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3264/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6fa04062fc68f9a4b3c476a2edb58016cecc27c461311480871d699d23df3d00f058ce49f6bfe42882660320c9f2cdb39603a93ccd7665e6ff74d1265e34c1e0669d6e538549775d7cc72097370443bef787fb0a82879563d8cccf733c86303009fbdd92f61716726105b08078aee3bd5411b6d262f05a0ac5470d31534011d1cab5d35b1e28ffdf447c3b204986aa0235c1ffb3ad2bf8a937f5444d4de735e394ca7734f420cf3102ee8d4752706de7558edcaa35c4b4663d17401920cea21d106c9b7bac222487c0c6c8c8bb2f0963b7c53e10e365a526c4f789f3acc1fc38110e632f669c1f8e788baeec2cdc397ef7bf4cd1d816f6664283a8c2ee438227&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876940437Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/array_metadatas, Name: checkpoints/1984/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1984/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b68236114335cb50a64ca304d110cd6a4bcbafee95c05296789eb5556a657b4ded041fabbdc6969a21506d98fbb65c1ce5398c6e91bf4e9d10191ddd81c0d7859f3e9268085c8880de7eb8829d2ce8818d729778d4249314f756cb820ee896c6aa7a15e3a8354c7bb68b4510c7dd1a5cd51db18b4f27ec57741ad22346ff889490b7d58f994ca250664d3621a9829f92280596426b8e1748639dcc9f3da3e31d186ec79152e7090af8baedfdfdb9ac1ba13249b02234db256ddeaf8929df25ae646de3ca699f9319ad722dc77ee4f7594b3f65368839b9b0bcc5ecc31593f413611b7a99e0fcec7bea3fa6846d7fcfb8e03a455f7866a01123562ef6f07a3147&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876942647Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/array_metadatas, Name: checkpoints/2176/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2176/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=280224f30b71aea5521457e78c72746128bf3b3e9b7d72f535a54dcf328aa40237b845af2c6d251a71aff69e27fcb1e289a4136d021d1117ee1e22e1bb7bce0d045e66b8bae50c52a5b745a5ec8115d6b51d6e2970f923270f26958426f39a1c1cfcbd0865a28c6c78a0239e106e7d15f7a352fbf7384909561b37eeea9491998bea0764fe7655b604f01133957bda1d49215bf76a5c00bff1b5f5c91748299ce09b93274fc994f245f094f695120351eec80a51df68357295dd534115b98ccc4a9dfe375cbefe112c924a4411b34433ba2422693cf9158ee7ffe8dbd87ddeff83ab34bf9103d246fac916b30113515b1c392b50589d962e7a0a903ffac675ff&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.875802105Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/array_metadatas, Name: checkpoints/896/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/896/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9fe5fff80352f84f779bc24be1a6773d4ba61db9f3ce9564a8af8f3496e835a52f93ee7048e25104d10cb9839da9dd5e37aa2417b7a055b395e900e4359a9b27cdcbc01d32329f38468898d0c0539acfbd0d754b4915fbf19139d9847026da18f14355bd35c017410312270b6e5bf8790ec91815b8952a785a6491519da274c3765d4e1c002758a51dfc267aad47b5adad76ebfb730a97a06259efb88d4aed208f548909b8d53b2a56ef41274427da605bf843644a08fff3201710a48f749b17c7acc13cc3cbc7667b06551c465b8fafd2bb330c3e925c078c81a5f37603aed5e65572c47552ad3943c0ad20a26e8023236a2011ab38025a7868c77f106c3d4c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.879541177Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d, Name: checkpoints/448/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/448/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=3f4388315e0608cb51eaabe9e30e398412e9c93001d974f72de04d9b51de763615710a8194550e7671b0d7964fd4944259d4173c9cdb3759558edace688e30722a241ae41fe474b9eb1431cd7396a771ff46cc8d385832e2efe95117ee1ea59910b86ca89588f6860caf2bc2e0d6ed0d12fa96c788ecf764dc643e676ebb62cd46b86378f1daeadcadc4aee264e1c659b8d2af8da5bd11f8690dff8c4ae701a657e030d71c70dfc1c3905f48d0e21a261ecbb37f7d72a30345401c02e54e3e5409a460f12466aa44e5add590df0c3f50889ecc308d22d985498d9eacb7583610be547369f4971fad5407705f48da4d87c983828b64924222a8957059c2c674fe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876999092Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d, Name: checkpoints/3904/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3904/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=330c844ea9d8ad25657a3b59c1abb4ccdccde11d4af0fcb06f4a2fb462f2dcbb5f06979418c328cc4b254e01979dc8739fc806bff90fc4e6acebeb271be6b3119f3678393407f236ebf88b9d0a09d9c56027cfeb0e9466bd1ff2b1af8ef670dd8dd58d9ebb1a7d60a6cf358d2d539d367b707291b3bf0189f70c4e986792cc944e6cb6aaf8b27936def55b1ea1d457ce19b1ca5dab928593a2d9dcc5ae118a99dee901f66baff747d8fc258c9432e0f6c0d48cbec7b1e45a9b88e59f73a58deb4ca643314a953986ceffefee3ea6432e44e6375635ab43ef93aa6db05c367ede7ebe90724864464544ea1d7f87c5389f4c53c0e4b421ec9b46cf2939674078a0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876995871Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0, Name: checkpoints/4736/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4736/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=868699c6388d1edec7106e00d035ce38574ef8b744f888a13d337e98095da3050858fde7c5657552910a3bae7256444d9e52a1e6f6d9a3bbad05025dfc76999543678e362dade49fcd6f37d217a4ba86840ab3cbbc2b01044a1d846f55cce78479f0996edf4dd56ff60c219762e9809c233b1f9f2de6ae5d7818a5b7fe0efd1898c7445f5c12093676ac331c403e9fead98072ed58df54005260e68cb401c4c7aacc6422cec0e73ae53c90bd53d24c1f86aaf6a8e24046e6459227ed7e67c9f06906e47263b89d14c95e7d4bdf0e448d7dffebc5e7c4909aab5d65731b1422053bc7c331883c6c2783a25b372512ce8e10401c16a2c55cd71f45a248b525c33f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876445452Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544, Name: checkpoints/4544, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4544?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=48cbe7a1eb3275f7145496ad4910dfb320cad1a96f7bbc52e77858693e9cecc9beea0144e6fd58b9ce218fe79a57245742969576f3332851683fae3dcf4af0577a23caaa2c2b96ed43d4643a23dc06fc3665e2230fe27f82db5735bc588a986a8fc3bf0fb57587f43c8920655028c3808146433c31a50d972e3cefe8f31a75d902d87be0cbd364122f1348e2c27c479395e9082534a5c3f8aec60de0147fb82858b912e58c9af6444dff5044ae1c7357508e3004220cd7c09e4feeb8bc486e727c70f7d89105f794ce1a8efed4b43217643d0be6058b44009b05fd4de50ff1918db370920d233f3b260245b1da59503a018082af74b5df68144f266e40e5e714&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877022596Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default, Name: checkpoints/1664/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1664/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=24092247263d8f5c735fe2389c82550cd91a8356f83670e8c9a5fa63c53f44640a5cd2ba15b0003a4bffd570188091d666fac8e257ef8c32141aa256535f47d35b8051b35842c6ea4e6d40d02c2aeb994d8e96942f8fc90fde012052dc9ee6b8477a32da53f670fd4c286539f46697c3f37a5928d25f994045d262e045ea58ce27d02173db854a1f08f29d2b93b264212d31685e5bbc4e5723e460e6d51b439097328cd051de1bff69594dc8a4b277622390c5033c7f4608a5ae0b615b50c2b156068f0b1a3926a66072e44e89e1581f8f27cf63ee41ac68bcac508b1b03e9363dadff16acf5c87f2f0be56955eee1212ce34e7fe0005fc5b33250558302f47b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877028699Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default, Name: checkpoints/5248/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5248/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9479ee227f816490afe02d7e263f0beb21d21a3349d17dc52bfb6587807a8fbe47938c5ea2d1b9df5f895b35160462cd99b9d5b910414d9b2657c6a25eecd2058f0b4da4ffe477e7029e54d64e67e5189b2500915955a5276a86de43c666ba998e3cd6aceca919445e711c0a7b25b5031fe79f05c5925e3b0e7fec5ce2a8225a14ef0cffd16a41c718fc889aa024441e4069809f04764464ab87f3b2f90a483045e99c0933fe748a3d45b274d6849333af58c7263ee49e3558d5f290a5182edd48814d1aa07799d217c4233e39d4c5358f62cfa95aa2cfeb567de488bafb158980895adf2841ad20538260ea065ef672338fb6ca8b68659813e0a0d7c4b4969f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876650309Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d, Name: checkpoints/1088/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1088/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=0f5a0dabc3e6339d624a50f28129039e9f015444426a7f00a4cd20fa35a4f9d5196bf9d6e842b5733557a3df01c10f67cec3809fb7d82f6af1409a98b827f8dbe5229d7c7c4da00b041209ed14f26dff0230156960b13288ee780788ea82cb67fc77b06bcd0b145440f09a336a972ef39861700f71bb6024fd4ddea9028ddc6c85fa8ed8920c04c496814f057020e57ae883c817194f00b5a9cc1127d4e33c251aa4db6368fcb8064f02a8a0b6612723c20c167683a5e6f7596421426623d4f5d5df27b01479c3deb408e9ba8e6d38a980d0e36e5d466015832d5677036f189ef885abc94c42c737f93f935f0ac43dfc61341ab950c34e2e5d5ea11c1f5dcfb9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877059777Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/d, Name: checkpoints/2944/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2944/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6d756360c4460f9f5d93f55f386c9d7ee931cbc9a0c73cca2225769e5998f642bd8d83b90b18a0f06c097c9f843471bb89e180987e08be3c87373fdfb65c35c812b611560c08346f0a5b4ac609ca5c19cc11025eb9fcce18cbe99cce41151d7405d0dd02e6e38f0296f50d887bbfbeedcca291b9f69bd45dddc77ed6a1df4d6211da490ecc08e343ad480c223c750538340e327dae459f6af2b70781240d582e1cca09187ba4c034baad438bd4f3b397419aca92f5f417e239863bf4b861ab42b101503d6dee51787eef13df451c39f41c29df55f0dde601328790557a3d0ce540ad785a400ac715e9e948cd3d52efb1e2a394a52146069fbb9fdef53cb3c259&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877081286Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/d, Name: checkpoints/768/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/768/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=0ad58d3e1b5e6fd5a7a79a97afbfc37b49f8aeb969f4e3edf1eed339dd9f877e6f59a3a2319d1e779d9b6dc2a3afc1dc86b77f94dd9d4daf70ac40449a59d267dcc5352e199d792947ff65b3e1b348cf46f26653752270379301e1ba0fb4be93f530c17d07afc1fee37aa1d2ec6c16835ec86f4d6c7f3ccdf2efb36a38786421992f84c1c039a9aadb4661fae9403752f154af744e0e47cd56770d74dbeb412664bf3915891e5caede3843423221d4110b4db4defd7ec78f75de2aa4d19fa91422fc7356623bd4cf0cfd8ff81a899b1bbf805eaf3455e32a5d731a2e9a5b0b75ee03eddef44e87e06d9b54915c51f000c56f5acdbe4e14febac9aab058c745e5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877094756Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776, Name: checkpoints/3776, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3776?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=000a68d3af48ef194fade06f1bbfe3d7c82106e9e33d58852d68e5bc7c863995308f0226efe38d3dae45d2f3e1fecde542486cbaf0b416fb8ff6da37f73f8e7b6c04179c390343049655d2d991cff4d3f3230905b0f54e27e16c0b9704ea5861d66f8e14fd98b491a5eaf72db53d0c0182ab695dd3412ce13a30e1f616ebf8205d98a2be60b7802411f5b7f2e892ffc9bcdc51da5aaaa71886c26ce3afcd0058e57b1534ae746e843154437f72f9bf10065b2b7d862640b20b4917d33c2cab8b10ab1aaa6864933c59e77d6a24fd3a24d2652a782f4456946563f31bcb13df444979f4dcb963ce8a69c2857da17ce615c515addce64b5e7723cfdda130a3b0d1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876626769Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d, Name: checkpoints/960/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/960/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1f3b6c82fe431d30c1e0322f75a5a5662d7d020fddb382b6aefb633f424b4ac7a6020c3b4286cbe03696898329ee3571ea9cff8ad6facca51813bbd61c76c1ceb56e9b6a8efcf3f2fa7187a6bcc6c0f64b1abf05515a19f1d255165f009115635d7094fb2d09124d925f7f7e33da96edb96dc803b623d40548b52071be03df3c3ef70c6cb0c43fe1960180d012b77134a4f305eade39e436b5530b05d83bebb10b5d349ee5c274096912960a6540c61d0a2241a895cbdb166cbe2ac71af1c80d9f9b176fdd89bb5b968d4262cc0b1321469be80f2382a853d2deb7d103fca32f5aa4fc409e55e4d1340bb769b763ebd097a1d705c71dc2ec07785d3a31eb184e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877179257Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0, Name: checkpoints/2688/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2688/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=81a044456ea5b99ec515e2fa9b91925f452b4b0ea2bb5bf18fc9f44c2cb9c626bf770db9af6ed33b1651908a86273d472dcb234db30c6368227ba861bd1ae16b62c64d47a8a1afdd5342935a35623b37393d87c8e216c91233c5bad79f53a9d57e0effd0e5b176152429ee9d19c90dcd421a2426f100f090da2ec5a372f0b66b116e2cebfe3126619341a331aeeac41c71abbb1c0ea29c112baa946040f9fdb5cf544817b14564b8fab4b694c2f22af6b5b486fa95d6b2d38ba2a3ecae0e8faf76e0b40a0d8bed44a285db91411a94768ceb572cc22a1af8c3c8de14f0c8ae74a594fca0079cc1962c428876131a88f32e03c0e7ef126d4040d570e98cca0597&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877195733Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d, Name: checkpoints/2432/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2432/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=90df211a5942dc4eee0d304e3c869a5479e2facb7e2358af53d2917f09de38f5821bf811a9863fcd918078f974a0c0391f1b3c8de4cc5ced8fde1b4b3310fcde6c23bb3e20e2c6f7a580771c88885d7fb0d9074afaee651c5d3d1c843288a164ba5e714dfbbfd8e7491c48300d86a5ae946a8ddf7796eda8431f3ff2d0f5531844ed15e6e3d46cd4bf3f3a192d6788fd6d4f0018f9a300724f24bd7d8e931a970e9d3e042de1d01ad9fc36c7ee1201980923e9a90267aeec8159dbb1a6a2c9da34495fb14e9cc7f64ad4d0aae02d5af6485e0a41a32b3343166df313aace3274177108f92e0c3cabf79df16a60ddd72ce3e8ce6ce28be6d47e11ba0986b362aa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877231631Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568, Name: checkpoints/5568, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5568?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b3a73485cf0ba95197dc42823aff82a8f300d5dfd5a729c503a574537c5b2586443b8af56ecd3d166818c218843beced92a566faa7f9dba2f0f22a45233340223b81f1c585c75a73ab2e7205226c18ab8b38f7610712070e32cce175ce23edc9ef1eed8398c3dad61a65a5a2bc4b0ea74732ed30674e26e882006c8c1ef5c4be2c20e4ee64a69af37e68aca3f2829d2bada50f11347a4b733187c701ca2632d53dac07ff12ebe75d64b9550ffae8c221255ddf91e4d65ceb7a3a5cf633c3a0e1600390cbace380caee5998bcaa533a5c322f55c8893c55a59ac899d5c9b2a6cd15257e56e110bb7b0cd280e1016ee66b3cc0be8207ec0500fc6ef8e2875d25d0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877359607Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default, Name: checkpoints/4480/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4480/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=bb8968709e18db6d0e417cf4838b717e26d5e571b52f108be7e59e065e865d1551beee5fe10e538bcb6ac672737358334557beb397cd8eff68889335c958ec60e80bd61031a303fc6465f67abae49a12fca7dde04a0d4e55bd991d8245fc291d7a39efa2e36959a481bb7ea9ad1c27b8b131d74af0ff332dc87a527b2b0478b88de9f947a316d0126de6e998bd62f7ea9cabf8a4164a89da0edd09a2ffc669843f25638c6f503e76911bcde65d2b8ac1d2446c9be3177d84d05b731cc4b0597b9693f4b27b8aa65d6670706685a92c3d8a742f8c6298bd8b4da623239c1ab2b1bd56e45e7669a6615d18f0254f1db744ce0eecade4f0d36aa1588e17f48498a6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877351561Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760, Name: checkpoints/5760, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5760?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=bd24f2e7b43b063dc7f5f9db3e504504bd7ca801cf3ae9f5a47bd97e433b588321ff2513f1653d47d8d9ec07389f50614866594816d3d892d332397138bd0f6958946cd4b6e4a217bb0b649dd252a153a318273e0dc0bb17784f16f33e0351a89b09cc5052eea7354d188c1bbc8654326e7954b51d50e7d6680c8e7e112ab6ba54ce16924561d7d33ab34e1f59bdb59a874d563f1beaff1755c77d8e63cd224c7a6038163846fb8421a2bc384bbd0f8e6b8b7dfce2d6f74829199559a589a36417aec63e476313a46090bd1ffeb5d0382b6aeb44a746b5bde94fc4f2ce3d979b6f1aee0686b2a0ef701c2c0a9912a6b8e550fb102304fd89027a0a61abdc3f01&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877483953Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0, Name: checkpoints/3072/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3072/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=04f3d62455838f0bd874aef0ebe90844d41306cf26d20d578e71c5cb90286575a8432d5a291cd4b34a76e741f0b4dbd1efaf0c79eea4541d3986876c6b3e88fb8384e3fb35fb202504b4cdae5f449f9318971501ee0edf3287a84f045d416fe2fe6f4463c6b595325ac6c75dfc5a108bb40f21c74d7063881e549b478bb77992105e84c04648af126dd9d6506ff4030f1543384a3af46e2d6262528690cf91803d5a64a8a59c06f5791face78657e40009d6258342eed750477f37c101759023365b2658e69cf918cf476d719d02a81478cafb59ea13f9621cd4f075a9839f7b0b0e05be98a099b5a743d7ce7e6491a7c6ff8098345b52c4e06574336678ed61&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877175725Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600, Name: checkpoints/1600, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1600?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1e4f834e7520dc26f3ef52406555e40ab858ed4c2a7d266a5e80f102ed032503c1216e524e66d94a088d3be2133ad075a8e21991e4356c6ea508d8fd0c6e52a1f979990dfad03c11736419f2c853c0d0faf95e0025f6f87d25609b73e586f6cf34dcca4465c52b63ad952f0243f7789339f08f0c5579629b66c0df47b5ab9f796420303d425545ea7d5e51be529ea380fb05aecd937f3712fa6c8b83946d8d4fdca24c4f31b6ca91e01471a1a4af7cdd4722e1efa05e54f9e98398ba01b15f26de923a488107b5c08611e48ff6cd45ae246f96c5bbfccee7db39ecc0d085e9ab6dd3a5117d01f488ce40c0271f301d519347af8007eea080dfbde80f489ea5d9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877525746Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d, Name: checkpoints/5824/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5824/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=588f3b740623c12de0ae9d0e7e7540d8005dbb8678da58bf68abb5aa7fd837478dff690bef4bec4fb35fc3e45927fc8142259a558f0657351b9c33b2235955d16e6805815615ea245c6a500432a7f7667ddb8dbc9c0309ce9bc1c69098fd22defb38fcac5eaa32316c16f18ff978688c2cbefeaa6cc27b8c6ea04e4eed7aab43de63694395671322334a1cdd03a0c6c8eec9a14773726c09a502e0a51eaac9a179863719bd511d93309e12fbd7504fd1bf7367b95a8018f78222acc5d87ad4f8741641ea3550327819c7dbca0bee851b8185f18ae1a504de520bc4b3c908891f42c2e9c4ce1a5e3a2463fdce40dbb761bc82293178cd69e3ff1a84dd22bc35b1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877178071Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0, Name: checkpoints/5568/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5568/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=3a09fc2a9af0f8f8deae55d0c214eb90df53da34af444892e7cc4a465fcf5428501b0c5fadea9d7dde16b6dabca92022ad211ca10ccf13514eacc2602e32062c5a7d7812aac8b9c3503075920117d32e3b306e83a2754731f7f641a2a1cc1a65d4e3cd9d1e1979e19632d7a76172e3c06b2a3964b9875ed2771ca4ebf7dd6fc53b837447fde1370854c9abe535b510c5d6f084165fe226a5c3b77bce276364a3c32732370e32aef0d2f17abce397b58b2551a029608b5b5bda82397303d368ac0b30398898ebd1f3480317fb4f3bb6ce1f83026123da7449804e384342a2de4063d621eae1ce67f9b07ab224613c593214173c09732423ffb9eec1f3e3d385a2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876549241Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0, Name: checkpoints/448/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/448/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=5d391b7cafddc70cd47d3495384c479f8012420f7377a820b2f936a71f3780c333fa66c13a34efdf260f3747ae7c1cf0e19bf7c0284e1a02cb28c6640e7d7235731454c8191b9a23d13670b581bef4c048e7a5e0a53409b4f0be5fbb39fc7bf1e847c768e5a9359fd2a7d1508186a503e6b815bdd58bde43153c8ff33da199c5b79fdfbcb383d95bcc01de97d73996f905a62cb998fd0b2656ee68d7f5979a441e7ad29047ebd4a29941f4f45dc53d90a8f2aff520308fb3430b0cb9e3c5d03cbd3a68480687698b9dd42e8737d4909b370b8f487de4250f189b94767e93272f64855ebeb009d40d8f6b496ddebd61f688315457d9c07436d1dc44b69f2c42c3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.877515109Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088, Name: checkpoints/1088, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1088?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2bab1d1a48d0f94390d832a30b3f5fef0583f3dda5b1dc12f8a7535a3308570f6e762b2d5d6dae95c83cf4b2d6b4bc32b413075b0d571596f68cb35c3642b4370d84ef917ca69f536f61472c54d96ee545ead0d07a6964895da424f026d97e5ac2b6b11c174d2946d9b4ed1804c6da64811de05f2a00b955407659365268333e344cdaf60634919631e720fe9582873ed505a3e83ca12249f110d124554f381d986293bb8486187bc593c91f5dff1f5479d555062aaa489e142b6a8855a4cea9222c0775612153d2fd34ac3e0bc273bdbb4e1dad0d846eb1cb65e30a8ce96333f0adf6051cac80c21625c443392605206bca6400442f4be4e879e3303701d77d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876466575Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default, Name: checkpoints/832/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/832/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=59ee619b8d28f21911f8be01d82fdb2bdfe31f1bdd25a64fdd563934a61313e58f3186e86969176f4cf7f972d630cce76daad53db30b17ebb7f5240fbad536d7ab57b4cba15365c75115e27894444e036697cda704cb541de7ee6ea53f9f34b0e897cc852a5f73a1083cc812e2199f9893f48e65f9cdeda76666dd0c08d49cd142f2a7877e2a58034ea6de05d997a5a8690ede56beef781e8a0ffef821a2aa74221f77163b43435c14c4f2caff1c697298b086661b7fc3deb5043a2687fb95ad10e9457e3c11ed87d18c406a369bc2409bbdbea556767ae56088e23c96f354e59799262d63e9e01ba68cd3b3f94e428e834b57e2c412a1e4ee57370d33b10b50&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.878028976Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default, Name: checkpoints/6016/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/6016/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1454a1a894f79e14b034eee1ea263679ab8a2910fe0bc571734de80d13d99dd69e93dd590db0defd0eeee408302c753d8a0caabcc0de11ad96badbc93dd70593d6439f03454d6e4909625a9a810404b09435173c0eb3973188c35e1cd6ef2326c367a4b5eea9aa2182dd27548566e1b314d2ba0f3b309ba2afe525ddd1f81415081cb7e6c530362a855b048feadc9567747ef22089d36e02449b5a75d9002f0ed34168ad1baf73f1f1aa4fd8d6f6e724bd8ce72173d1e31157c76b2a257a1c07a47e74613facf15d61379a14ecb18c2fd8483ac89dde3609281554b105af2150fd12340ae869b8a3166dc5b0035daf91f00040735f78dcccd39b7150b0d7d3e4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87803236Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/d, Name: checkpoints/5504/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5504/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=77ec17813c7c91bd3f6ccfee2a534b73bcd438b43567446e3c3c36f14a4ef6cbd3f5ebacb3fda3d0a29dd5a3ca2d110658c647f924a06601e7dfd9db7ff86f723ae6db76a41e0ae9d0bbad9cc61595cde4495cc0bf46657a79c036a5765abf1dcb8d08222fb17228d3db78246d61f9f8bb8706793dfbecf248ddc84a441af91b18dfe31dfd162a54c2aaf9eeea6de53e141111e45cd446de7745309d52e2d45b8f793645fff09ace3f2da5f95ae21a7a998610935a5388d7d299ef9c0291341ca1480e4b1a3914fc22a58bac4978f25081f9ed13884b9887a1e76267ba694a650c49367316733ab07c46c9447e658a477af44087b1deb28197ff6fe9546e0f0d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.878154544Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640, Name: checkpoints/640, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/640?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6f83cb88ef5ab8c4b5b48208921ed16b8564afce458a6448ab1f1220dffdad8b338510c3d26a99aa52279216dd97f87836fa5bfbe0c8a66f3ffd7277dc37bc95f0135759916a139dd0f703f1a3a657d87694d8e8e1ab0c49aa78aa88dd12f84c9f39d91ddc1bce7f467aa60cb212ac777282a564ca907775db2ec6fa38112c75b9757de6173bf9bfff47a13ac803a5065cc9745e808dbeefd9240729a9b7950578033b61e69a3f0986d2c253306710b8fe0ea8dafe2201767146352f19fad2439ee9d7c3a4d2ce52746649d2826c4f41c3504a355fea24311d2aeff2ded65ad66e4479d931e240ac4b6d824ff87ca475237e07f27b3636f1bb9f9c36263647a4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.878275676Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0, Name: checkpoints/1536/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1536/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=94934ef591dd7b9df03bbe1069cbd057ba361d6b48820259ec671fe4cd25d1ba39aba364c639d2ee0746a947ea2c8ce880475293892d5bd40a74a9b1e948d3e694c3ef31c61f2ac5a8599bedad729cacedbf316ed38d2704f1f825da05061e0c79ca61e3ff08e2d73d5637a423909201201ff0dd6b53e5c5203b215efdde0bf89b17f8b14c7651447027f06bbae91d249a6d1bab2c823fadd29d02c13b62f325a78638958d30c90e99f1f69ff4f02d352b4d14242c49b5167d92dc3ca2fc796ef8fb33b0cafdeeb616f92498ec5d2d9f18e5a372985b191ac3bda2b7842176dd8532abdff91f398edb3f03340cdd1ba6507430ad47fb77be05e05eabcfa6fedc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.878333301Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0, Name: checkpoints/576/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/576/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=86eb418bc0d22f683c3c33decc5e3c2b73b7f49e01e1b93565e626925ed3e37cf7795324c69afdfc0682e2137eaaf78135b4ff3a20811cd7ac1b55ff327e5f5936f63bcb17b3c05519e3ef882656b3a03cac304f46b357b86fcb62530ba1deebf1bf30e97b60f515d711549833de365173a48f685d06478ed52010afaea8cc098aa99b6bdf56e760559e19a12e2dfe0d63ff307aa04712fea7f996f3714ea2d79a3c6bed596f7b7c9df3b6f5309a984c884ca2e94f9e3fd672a681a78dc9d3f0ffb592569f4a988e300d1323b0011dc00845ab3a39b9c2be64893ec4b0c098d1d7d9ee7dcc4246e145253edb29894ce27ec5e3364f4d7920a11b54134c9c426e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.878359729Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0, Name: checkpoints/3456/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3456/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6d985c6d8fc862a0ded9bc2b8fd4e7beca77f27b30683af8b2026292cec6862a0ad87d21e39b838ab042e0a29bd1f075b80eaf2bbca5a8e8d7193327b29e273d65a5193a2960224fcd8a35c3401c5e183e9e049a1a7f740f10e383552a7bfb765acc8daf605c0c5227a80101a8f1a732876b55da774d5fa995353aa7712bc8e64a54ceac0bf3c6012e7669f0a654e61e12c36f72f977932d3dacff7c09b85aa9f330fdef156d1f0c0901f8e65d6f50e0a89e2e252813d45a5b925fe9a80db444ff6ca8e3ab24bccd00b4c9b374350e4edb1b58afc71d8fd075df5cd63f691d9db579be8f704286fdd1a8a2debdb0024958356987e4d7c2f8d3b9361220745673&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.878396157Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/array_metadatas, Name: checkpoints/2816/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2816/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=25cdc93f728b26d244dc22bac862dcd3ae8b9b6215a0ce402aa849e3742765a69a93a09cd12697e6618144fb0bd07987119bb2434f1796b1c8463835fde79bb32bc6509ea27b61408428de1fe600736f36a623adac1132e6182035eded25ddb85d93aecb7f5d723e7da5adaa614282eb5f77d031ea5329a4d43ceaa50601fb70177e450721a4735664ec811f882c7c51042a486720b8a3cf6ee127642312f55da4326401a2880787f0e5d0cf500a13fc9bff746bbc7bdbcce096a89848d50ec26b0507b94ffeac47f9dc1f7edc970a218f9c0e5ccd8e9c8d7ea77fa708c0470db5c29b221c4c0e1bc5ebf655c1e1033237134262be8d16b49d6e2a7545ccb449&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.878404945Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d, Name: checkpoints/1920/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1920/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1b20be54e50c777b767097fbc4de4577efb1939365155acbc86b12691e7e796df5fde8d37305a26867d98903e3d3b43773e5daca2e6876c5d462e4f8a8898de5a3f50941403ab820c511e29bd2bcbab42fcf8e30d73ce341c82c79fd9875a4e87f039242cab1157b4412dd8cddf948a10707c86ab00ff30d1ee1c9c8b7a21c8d4615d9d1e4e4586681e7de1ffc323c19a324e34ae281ee9131b91e088b46ff1cfee1c5e35fbd75bc21ae27c0a659d9b839d4a0fc201f7b8490489284de0a1516043d7023dbbf4cf87fcc32c8b11991044e1a180d741eccb641388376fb728c6c17266d949c40fb8aa26a362b55b72ec04927b1bc421e53a9fb3051d0310641f2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.878417178Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432, Name: checkpoints/2432, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2432?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b07ce3245001485555b3dbd0933cfc4bd593ae5f21b5c731d2ee2edc683027efa59c595761abdf8f29c4b220c103eb4e63d83d383c03ac7c2dcafb9fe363c7313114bbc825f5134645040cf030f122ead3cb2ca4cc98df948a32bc83160a03cb847b2c141ba372e6b4f43a5966715f91ba197db1c45c7249139655b360beb85cbffaaaf37b74996217f5fc6af3c275487c4be6f46535733eb1dfd29fb475c535b25c3e243e931d159224db91a6e23bf81f73cce32f1cd913328b87f138eb16a25417d27420bcc01d30bec8e7eef9f3d90a7d8c60040f1a172935c982ea59a1488ccb77e3a718cf9ece16a24054b8c55684efc520125475d0635d0daedfd654ff&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876142539Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/array_metadatas, Name: checkpoints/3968/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3968/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a4a66521c5203259488aec85a588875c8b025c7846b5a97634bb962fc284279e236df4434e0f98e1462d4abc17a4de01950a433eb41fda0eac112d5bd7d38933cda65c20c90a901622bb42f9184ee518ff2bb5a3f134876276faacf30b72f558549ecdb0a5d8a659064d9b72dc43e628eaaa925b7e135352ddb17532f6aa2f39709fd13bdf65161fcd289d13540db9b0c22e7f69e06d29a006e68c959875025ed7eb2882b84971c704b89875199c8c5de73b9f4f1afd6924df2efcd81ce83611bbbb44f171eb9f6f1603dd4e2ed205167385b237348a3824981a3cae07145644357c9c24ce857aeea21e79353db2ee3d4128d986bb6061fc37481392029f300a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.878529071Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default, Name: checkpoints/1216/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1216/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=bb110f56bed4608597093c5fa8ffc81122b2d11516c219c49720b84dc2556921edb914f2e6d297c9225fe41795af07cbf2698e8773e05242a14fc94b210cf105626b75381c2929c2ee20c923582434f0fdba2db9e8234fda5935b556d7c4302fb4590fe38614668bf326c0ea0080fa2c18f4bd1054620da4a5f00dbb866632eb255df1d18c430646324a22c3964866905bb8eb3cb40f741aef32cfe69339d9b0aff3e91e02e2000d9f9d35c8bd1879cbc0a6e47d734b9224225091f1235a379bd4dd03151bb1273ab4d75932654703d62c0f1dcf0e3e1d3e3e007e22d0d53de19c1c7a2b5eb393073e0dbf1771748aced17e2a56d72e832a5d521e664375837c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.878579021Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560, Name: checkpoints/2560, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2560?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9e7be9ada17aba182277eb7290864d25bb4829c2456672d0a375d715cddbb5c984a5f48db7c0baf7f28af3cd79b29c7e2fa9af6fca353de7481eae74b34497769e52be99c023781516f126323f1e398f2607102268695a6a2e8d4605e9fc2818fe6abf08899fc59a0545763395500e149bce201d32af0144fd367e98d4ac78a0cfbf4b152e5d91fe451fb6ebecea4136402c24c23b40f867e856fbf534a62a6bd46e83f8414318198c0d262573d65c0827575eebca67b73e8c1fbb6ea4b4c1508d98f74f73543c4736b5471bc043a605b2abe5bdfe5471653f0ff078485e4ded08791f9232320ee694cedb085cf70e2fd0802cd47d82d4789b70930d0f457f69&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.878602888Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default, Name: checkpoints/3776/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3776/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=c22d622925093cf012d4163315dfcf5d4b2e3e448b0e3b682ba5d7a29eb2bed7d68e9b766b078bd05ae7eae69726fc8c8dafccfa5500a73b4de843d9f3560f37040ea0eb68625d1ab6d803dd9c372e62c4e6b4a97192001f6c8546065ca3650779c1de8a9da87d3f9327d0aa10f77e708ea20d449b09323effaf05d62b9fe2e665997744625a3bebba0836207ec0210aa532abacea09917fcf5738a649b64c6a057c2d00e048b35991463c3e23e40098eab503f0fba7696fca450cc8be85b90d99eff791f94a150dd58c4a741ef8b355b3c46b7f4800de5ba2af89696a7d8246eadbc9dfca7e30a117621fab30b83ba3d103cb9eb1bc46c9ff19cc10187ca0d0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.878629679Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default, Name: checkpoints/2624/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2624/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=3d9380501596baf2e1fbcc48d148854edd1584e232fb252aa218bfdcbb3d7a13a7b27b1b51337e70e561ada923f427847f4a7c03e0227ffa1a60b0f7e02f307f0846a3eb5ac168300c727edaa827c254b95d7cefd9534e7caf6acec6d82765aa19953f1b69b64160b69cc2a0e6a06b20f6c99062153a1c01f396cafde8f815eed0f64e4bf997e8715cd209ab7f8fae2661bd9ffd14e6e0a3403eed561ca21e12d76b4bf0130849f13bf46739bf2a0cfcae8602c1bbe57777a9f43088cfa7fae3ab4f2641af37ec16ac477b49f641ba896ad36ed464affb8cb7e56d70077bd59c715f77fc45c7c1121cc92b867bedbf4799849ba97b634c0d1aee86a26a1f093a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.878658909Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default, Name: checkpoints/5504/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5504/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=0784b70dc3925ed50153764833c9fe0445d7f1ca62b8eee8903b212be7e2ea92bb923d7aef1eac923582ac9e6c6f1821cd66b41891cb0dab6fe759799514619645c2ff3503e0828cf224b7456beac734c7beee568d766876ac7bf2ced54433584ecd8d1ad934dae6af39b257d6b3505607fb2ac79724a4067f73239c9d1788430fb51471f4583eb47ae4753882f900a54c09df7d59c4c3e6d260b2f031eb41cceb37da4815cfdb9037d67bbf531d46cbc5c7172282c6f4eac56509740023ec1adca68f82fbaa9ebaa58b31418ae5882c1a0a59a20079224859855c4ceaf3d0e6f559abde481235dd0e5b643f60cd2f41641a2b9d58b967bbd30168439da8b223&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.878781043Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64, Name: checkpoints/64, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/64?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=712aef0e506695c6588e490fd0cc86d76dfa89f5cf74cea80558682591b837fbfb00a63a23fe0a997cc006f5ce5b4730f417eccd102347fe8a626a5fc242625576c5f07e9e13ea41764075b4d14c5f3f541350f7cfc2c4c4a99b41d2223cda1e9f8a40202d6cdf30f73fbbadb24ee111c423f808547a76fd0f21aed840f44d9032ea27bfeb7ef3d0ffe1198642946b04365e3900de2da877ad5a94c8ec2dd5f4f733936c936091e18a8e45b124130a53f215294b476542f8b307d4a6b73d9221cbc29d79e1d49d468d884bbd6c7d40aee896dd4d616e2da2ded8b3cacb5bcac44f6d92db5d795847704e2851027855f097a9a0556ece08b269389e2ecd9bb8be&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87889843Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0, Name: checkpoints/3200/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3200/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=5e0718e9e0852b86916ef95d97d1fdb8cd986da308344259d6be26ce62e55f95644acb123fadadb845fdefc4ee0da1261c452709eff2b7f6e2bde38318f0c3fc3b5edb81b7b31e0eb3e1a7374f8642be4311e2e4d9375995ad7ce176a4a71f8500ab40404a3c422a2c2f6abc95f2c1acb1312e98198fb4c891229b32213ff63a19960d3096c803b5b8220e94896078c7f245c8f6709a8e4dd7ac874fcde7d37d4b5d0e4c8f9c770528f6715c9cacb0f9b308b9a517671df6e2b696c6f26d9a5c3733ab3b8a248c3605de1175b55f889338c40e2fe7fd36877765dd7634a2fe8499b17997d05fe2e60c3435d822937e74698dabd88d2456e650445f6e11ef8c92&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.878919266Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/d, Name: checkpoints/3712/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3712/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=5dcac9b472df39b21292546a2ae607720a147f99b6e700bfb0381f04f83cde85d257c313e906422b27b3407e3b1d982c60042b7cec61349eb62fc9738d5c716f0f80b2c463df71c9bef0db5dba2a92ebf4616bae653f647205327002b719fbcef8549cda0dc4bf2fd8c0e872ce8d6e3104c5e59dce3212dc14c30398cbbb84f463bd12b9f3b296d09123a404dfc56f05e5f7c3b45ff7f2e6f2fd96b59b2cb3eb9b999a691d784f7967b2cbbba8e8e342b1795d59a855acf5bfe5b6b545fd7cc41584139f37b9c8ba8efefd254f81f9486ea30ad9bddbb2798c74695838a6a27aa9a7b510538b9118c91b53531a5b36641095a10ba380120aa3e64b7879726079&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.878943702Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/array_metadatas, Name: checkpoints/5312/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5312/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=22db56fd2108ef673b7a00fc377b6312b04a861b356f6751a9dfd717eb68027677133cd832e5038302b3c914a04aabff584dd5167950955598e9d6dec1b2d7d1cc1456ff4ccc772d6b814a930cf989a015cc00a86d63eb2bbe6e0829a656647f2b9ed363455226c0ba7013e7b8dcec367f18273604e53564a3afd69e456ba3868cb81dc2cbee13557627415b6d61d89b250a10678d99b0eaf52552714394b14ef8a7d7a2a27b09d5da400ebd76d96a1aa78b25254c4fc15ef2a94f339a30ad0bcba7530e60beb51db17c87f608e0756d16ddeaf2fc9cf7999e68f2df946c2c645edb2a133dfc9453dab80fd2f04b78c553bb87c7d2ca5085caa9a50d4d352c72&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.878942727Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/array_metadatas, Name: checkpoints/3520/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3520/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=65a4bcae38c52046506cfeb2e0a7f21391cbcfe586746f7a05d7dee1cdf6623c3abdfc9ba3254704c0f46ba5e1db17b15bdf238644f231f8b03359098ed6d156af1a99e236e26ec59e8fe400df5b8d6fbb45197cbfdcb140eb1a4f9c0b8c9c6776a8d25ed0456ea279847ef03858a61c761a708667f266ad7907bd033502a23556de217c7a46eb355d87260ed83869084ab5c0a15d5c8d59bc2865c519c50cae6d16cab6b92d31e8d818259357fbee85a97546c974bf59a66ab0f1ab3b7ae7b202d617de9eda174f2bceeaaa62f601d0c1a8be267de8f847639b54be3a55209039c7d3d554de1e57492c0b08f9b4abb9833ec2a0eaaabb9428fdc1bd1d3523d4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876823355Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0, Name: checkpoints/320/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/320/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=176336a8c759e426d1fd990ea5b383ecfbb64833eb4ad36b34ddce468fe37c866a5dd2fb0393aa749001395796cff347516bd09b86c39d66bf3e3367754fb0d841b1fbac38420f3a96fe748a1bbf46e4d439288b7a5f9b26468134775de651dd709756ba76acb553fed107a518821750cf76bce79ad0126a51b60b70614c418ceaf99c2d2ca5d8afe5f735d52e828f8c8161c09d8dd16d5135a9c6db6c12dac3f9ae9e3f6dec0627adea457c197bcc58a0f3e754bf062b7fed79d52dea5015b6c3cf7ee23fa65c061ee638ad22ae000ca1409d19323649cf243099f679280ed0c240e3c80f409885cd4cd96b9aecb126e2c5e5a6fe19038d9bec8fca38ffc85f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.879419689Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0, Name: checkpoints/3776/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3776/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=c41f96281794964466faf3bdabb8ff2f0ccd527edd5a14dfacd441abc1cff92afd2fcb7d342049ad1fa2b299fc614964c0f8210bbc4628e9f79164c6fd57a84896909694f70255e2fbbb6ef76242704508fe56450aa166e9adc0d814bf47e32ce54c7a12f4b705cb9daefc10ec5a897021afbd55377b798f942ef582dc3438522aa22fb48bfffe3b18af9934fca79ed371eec42ebfbf5eab3b021a72834e39b23deb4c0a9505f87613946182faffdb2b406468d6806cf58ac20c59513b9fd161eaa9f2ac1233c46b7b1e8df45e63b3fbc55abd4c784a824214dc6b13204eecf77503b2772b2e9679939a8a1a1c16f2eb1fcb7d1e2cba0d527c63596b0e2e533e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87947793Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0, Name: checkpoints/2048/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2048/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=8a76a42a44e673ffa10a549b915c31882b2e2cd357bd764790ff90cce691473543d2fa074ab599b7ada97efc15d1daa2b2d718d4089d238ede0215803af0128bc21a3e3c3bfe885447f15ce406be4ed0de74a8999d8b92b5706415c74941128e0d00e40c02f8adc6729bf9939346308a7c41ef0909625e4b490e025cf617c547722efba46a9a4475d8c9485d8af2befac62c049330fa9abe47f343a597c520a719e3c9cd1cd1ca31134f4e58824bce3a3a8aa461a309c246773bbc41ec9d9fac8ed93b1f76bae64988de6a3f32c544a0038d5d16c107dcae7ad4f99f8f42a613ac628c16debc2fd1fe30c3cf66706e3a011421a0736c4d7b38da471b1ad0a4bd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.87689482Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0, Name: checkpoints/4096/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4096/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=ba9beb338e5a061a9d4753de9fcc49cede4cb23043cc9f986b16de85d0f8543acc51132198a7a4ce2b9c8935444b97fbec645b97f0cbfdd3244609d33f6342f69a3f034877cf50ef82b322022b520dc17c0229f0b042e82eeb2f1918c92f01fb1feea5419874bf5b894d65d9dbfea74b1782e9f392a1cdecf3d632ad203f11646c51651e813fe84acb1cf7739ecb55f073774bf877ae5a5ab281025d2e7b3e2a6fdfa9178ce3b659020a76404275d00c5738894ef5d9a1b75c53063c430e2eb617bbe3574b49c7e6b1c5e1fce0164857b80696fe69421bc0113455c69b1ac72777125004c76e4f023704f772619100dff5d394e08ed2cb83c6d5c7416490190f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876917416Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default, Name: checkpoints/3392/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3392/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=4ed5d51d4f6f8a58c14e88c154d4335deac445c2067324b95425c14a944321fb18d96190b6d44d479b15174c24dd38a9ff893181b642fcc14651742b20999c84bc0496aaf6e43d7ee72355f5e406a61bd30c570836d2829f61f5b9ab4ebaa6d6a99a915be578f0d74a5308880fbab16f217a82c83c86a467254ea6fcf6b321a17d7452863af2ebab19e2e48a914389ed1a64c8c3fefdcd1bf9411770040f57a59820a853bd44a5f40a8d4c5438a36a49bd449d5930bdaaf884747e8407aae0c472855bd99673bcc0ef155e7c0b332181310161dd499c63253dfeb35e42d6d2e3b49276c3c900fc113ccea55393e215f159122f2b8fdf117de216bca8323dcef2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.876993147Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default, Name: checkpoints/2880/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2880/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=19a69397917d443aac7a86475d1040643014fca166b56809cb0a28c5612f3406626db8665bf4bbbfc5922f78e0779608accf99cb40770af6a22f43bfcddb515b9c59f42be7f6d24bfaeb4ddb91220d75fa76e77331e7a106c6140a1945db5475f261780952384a3aa03b24c99bfd40bfab2bf3d08cb093be4714f11d4c7e7603f662bdb4be503848d6c75c7c7592d671cb5097513c375b593ba3654853b710872964bef6bb282e3194a56df0d75bd2713abb2e8971337ef99e3ecab0e86961a80fa6b5fd4de44a2f10b31bd6298c78fdd6d716eaf43bde9b8e22e88eb93b54fa877d866429c3741c4492e1379456ea684291908779d136507bfab6a63b2e932c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.879566131Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0, Name: checkpoints/2496/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2496/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=5e257fa71dfbec7e5efcfc79553a914ea45fe5dbfb9942dac85dff6b17dd1af4a14b0651a6eb4e04d6850ad9092216249bc53c2a12010a29a0a41b9a7e82df49b3beadd45675c947210495cad3c175421aaca02af5b2f76d1f4d66180e15fec3c3922fa6c26d52be765695ced02284eb11875b63c632e16a177897287c5441658c52f09859853fcc7663f2690878a98efe365b78808c8b76042213881a56e2a28dc80481d4622d1f2c0ef2c81f4ada29bd7d045bc6929a7534223bfb317c9a46018d0eca2a21dcd94f80fa87c492b0dcb92dc4bcaadb96f9d4604a6c0c82b957130d438e726a2c66331db372c8fe04024cab0d75ae8f02ea43f7f663a8607fbe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.879564126Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/d, Name: checkpoints/2752/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2752/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=620d19cb941db237cfb053c066cb75cc8d7c46174da5071f53a554c166366f25a28fe9254714d3a20bf65622dbf8bcb670ed77c7aef7e6672224251d6288fea27c0918e0bf3e6b39919b4ad4ac3f72db077e45f279aecedc22a3d230ef8804847173da429e1640eeadcbbee13b26a192fd93d53bfabe26e4065b79c52ee76d013d82364e8174336bbbd5c8172c07a4483ef5c0bdbc00b722d44fa8319cb539829a11b122fc5e73ff21fef43191a80e2ffafd04efc05f8405fc4f7b9830e0b94d1d7543df7907a96b897940e98185bbb1c9a2982a634a936b7db696c052e2084043eb063a9c4e194862ea5ed7f29e2bbeb83fe2420cded97d5d8846fd19f1091a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.915920991Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576, Name: checkpoints/576, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/576?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=2a4f2c6019963e31ebc54205d3a2db58249477d3856c9e6c39e53f81987f2415f60666e70cbf9be59dd5f4a11485b15ba2fe1cf7f884873ae47427e77fa847b1d6b1108c93272ad4096aa4559e9512fa628c04405cd16ec6d1afb8b7375bdf5a82168291ca48886dcb017ce7d0699bb54e4047386ba3e93c61aeef6a71ecd73bb6870c24013e6e4ebde12529d226feb78a8391a4aaecfb7f77b153e091d5eab69b2fc95b1e1493c94c4447a04d98fdeaa695a4c4c3db3184a5ba2635551d461029c59f5158d4ed89f61963b9795f42024107081a766a5011ce8cb2a55bec6aea70e2558929a3eb215747bdf8e38e9e016ba7e7428306f59a8b59cdb0beb823f8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.915961114Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904, Name: checkpoints/3904, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3904?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a81f678b40e9d3b0f3f5d042f4f47e17ce1701fab4ec06a909f238c49992214c48cc15bbace85b446a863833aab58672fb866446b3b9cf7bf3d9f34aa4657452cdeb3912e9fdbfd3363c9498490269f3b78b15965d664a5ffb33d441b29b487f92775b3e9eed690b8a4ac0dea354de3f62a21d9028ea13ece3af6813b25c1d0d8afd88c205e1617e31e74b52006d843cd82a194761c69d9d4c693d73f9e25c055ac7f30056311505d0bc2260d563314fdf433c7fff3e3e26a366912603abef98c7d7e9855891c1f7dc0930b941a6e92096d2b118460a8c8746c0dd7e193cf6a92812a554dd549beba362e9e1b2f3971793149758c6faa0d20a7edb6d64d00589&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.917916027Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default, Name: checkpoints/2176/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2176/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=24c86400c403dc2c245c17fb6567f86170e1a729ca541f2d53e4dbbb6449c168da2dcb3f1f2ee015904cdfe0722eea1ac9648b7a680f9d790e780cc4ceda84bd48fa3ec9ca2583bef44b851a4fc0d6e47525c1fad3cee2296f122c98e3cff3a4809fa532eddac10b89facaa7d5b32fde4f036f9b4f7829263d2a0138d868fabba5f0e7284ca0de9835cbeed6594f3f5e827c61834171d443ff5e377cad9075aa0f00e292ab04616b46048ab9d862d8fd7834b7b02ee2c4564d56963ccc36c6889e32a6deefabaa60bd0c2ef756101c080e32eb84dda3bfd09914f4488d85baa2cd3d8ac1bf036287085b465df95538364d0dd96f2d49ff159b652a678d0eefd4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.918084365Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/d, Name: checkpoints/4800/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4800/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=4215b425ec49e41af2e738b82c072c9bcc69d18e6b22309c2be982b8808d9f58323f7c25682d9884f8a1366cc0f20bea2b10b60d11f6539d47581967a107646c0178aa00c1231e94c6acf87d355445cbf18fb93f2fc6955625f3e61a1b5a1c39149295f33bf2c2366b6b8a645642199b630867ced31664cc371be8d1922b14dc455e08f0ccee87f60daaa641599d3a7fb25702242fa4a067e0125634b8a59f76fe51fbe1842a2597509b987c6eb6835802d344adb2d8580ae1a437e17f3a5282ee746bf2050b89e785129d19a99fd486ae54b17b0192901c1e11d51a6ba295c7e0d71fad912e296dd31468e33b6915a4704cce4861228a96f6f3e9a1f6caf098&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.918084739Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0, Name: checkpoints/5760/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5760/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9e110a461bf55308fc3014d9276093b4babb03e00ad19372a3a3ce38fcc6d7bafee44859d9830f2c13543ccfd1df99e90431a3a589f203d6b42d60f601a446c01b478a9173727c3e4398aa1e78efcfcbe304af5af590def54a3dc64cfc872028c5c17f851c6b81fa642b14d6dedcfa179c9e5376dbf11d409507ba7ed1e5f5938e6ce553ffd36111a71051920d8ee7a035520627310f2810afb041377f5bc05e69aea5de585ab7ad38458a698609e795940679568048fdbe987988a9665936106a0fd6cbcca8f3240fe63ae43ce62c5f60b0f5e8f07f0b08f2305230ecd00c8ba948aafaab5c114d935079a5ac94d65f293d18153920c60feeb4647a21624f1b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.918153343Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704, Name: checkpoints/704, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/704?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6d6472ed7082b66959829f1d38d6170c14a7d2c597e574f64a2b5ff7ff598f78a4fe69cbb905d9426f362ac7424c75892df0f0ecb79e2d54c9dadf7d3d5ca6494e51b639a3ed5bfb152a569d73d1fd0999689cb817f47267a03216279328de2fe0dfaacd093c46a5c5a4d615f7a0eeff28b7ba8e440ed8c01ab4aa6a549804d83ea453f260cab0be6b9a359f60b4b1991d94ac352ff6df8f331c6083f5ca380a2b3a8db6d9cc9cf43c12d832d9e95b2811b6fea0139e507e020df83933fd9d0fe155636a0f47b54eabcfe7a00447e7900dcffa395e965fa55c7c9fb741dbb0ead929eaa688df0b888c0ff3b81e5cfdc58a01074c2e5fd9cdcb09f73d86c45155&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.918185379Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/array_metadatas, Name: checkpoints/6080/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/6080/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=b8fd3ca9ba03f8b1f00ebea562bacdf0c73d021143fede6ed3609cf325cf73c64202239c078e637d3edefb438de2afe1fe83cff27e34300d065115903eb7bf4f985f1f7fa845be1546280216387c2a0a9c013687e2baab987093251aa5c2431197a8e16a91a1d91ce1438bea6c195d5e073ef83a98ca0750a0dafd6485ec06cf7a07553eb246ea942a0d830715ec3a5ed075764131f64bd92ea6b6f2e7a2216c2cf0885742466bb4c073a98e6f6a63dcc9e949791b65f0ac574c06fa89add6394a16171171adf7965c5ebc3900c7d672de4e42099aae093e801de72c4f83a1c673ba7d6d364a33d1c64ea9087d08ed0b3dd4f73105898519e48164c3bbb6ef2f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.918202372Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/array_metadatas, Name: checkpoints/5440/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5440/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=28136c37a52cf4c525d20a21b3e25fc814f198f941c2c74c71dc7d3ddfa69c47542db898b56ffa55961d2e53933b113b4263114c86d1cb5d5e343006845b5233106a41029a55ef725addaf07cee71a029a1092dde602689644c28068ccfbb969daaba84ea4e40d15789810b6d274c37826cabbf09299459976ecc530d2095257fbbfd753453e2ece31c2c69af4aa5efabf6e376f553d8d6fe5b79b29deb821e864d4a1c0d6224f35465931cd8d597c8bd66b935eeb12f2693cfd457e311258ea3b86a8bddbdb8ae86cea79452c92b659455f9b1b5a7d85dee13716c1627f52c1e623e654fca323b8977246661e013d2b6800c2b4f1e378dd38f5d1ed39b0d334&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.91826029Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0, Name: checkpoints/5056/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5056/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9fc6b7afe6d8dd526f419df789ad05a27b55f142f6b05636c4ececd970f5472ce694c434e7a2e5c62345ba9dc9165ac0cdc444adbd7d20a5a818cdaf1d2dc7b026d38fd1c925f5bb53244116d5ae85566c98898e59797f31252a8e237004b11165eb1993203ddd5f4d42efa132258b52814420e90ed21257aba436cdeb7be35958728e654f3c0c3d7e76d8c40cdc9aa3c27160e23c38f8bddc1e0ddf669d74a966cbcddd512de142f5ef737fba74e7d06ae23445003948c11d23ec2647a3c16e0084b6bd2d7bfe1f5f8b715237d1f2465252e9f297062c80b7f5aca26760e858a6b884a0d286fe3a2fe58aad15b0a097978370dee41ae16b12d4a3f28fe5c90f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.918289211Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944, Name: checkpoints/2944, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2944?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=3fd7b73b60e261fda570be81d1901b99c2004a1ca0d3cc03d5db7eebd1e514d76b5ae1257a04827c82e0f86338cf1d10dc31c4b500ddcb0bace9d687e03093e5b14f6bebd6cd34524069c64840b145b1b8f3113ae5bb25c4dc6f9e97a36c4b32843ea57da9d21410878753ec0f36286aa8bd404266343db3158189a7afa2812ffe2b279132c5c71da91d10af50ff3e237537118785ebb3aba570ed7e35de2c6dfdeeb01bf87f4c53e31f51a61ebd4f719fa80bb1f931a67bad2c05ef10553f069af1209be206aeaf61c6b43b8dc8063ca0410bd30df2c689faebd42e05183949190c3c8d89dbf8e9a80adebe5445e256369c7ab432ab4b86840d72458651464d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.918323515Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/array_metadatas, Name: checkpoints/5568/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5568/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=015e6c5cd10d8f789f6d693e7a641dd56736af04a019dad7f1bec251cfebecea6379647c2d4b442771b00716b9fec5ec199bfcc50b5b6f0d6bdbd3366c9a8ffda01a33863aeb32b32700f2bd20fa41b8bbdab6df6593b46fab525063d8fb6262b26ce7016c5fe2a4d1a5cc42f552dcee3c97262cb0095993f27722c2290af776a099b223868921ecd9f6f0b6e7b05ef95db6e7493cf4c0e4a4adc475ade7c9eafdc1d3a21689513fa7317bb574888c8543d46d93750effdaef87902aa3efec746c2ed09a5004f13097adb2277a214612d3dcc43bb18598b6935ceb285bc37631dfa29d15bf1ee4e8a8380f7ca201a8b0e660d59aeeb7c73e99a8b77b424a46fe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.91831461Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default, Name: checkpoints/3904/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3904/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6d11984054faea5ff1cc1dc844da7983402a59ec267a90ad045fbd6a7aa45d2f7c12244bbd9fb15df00644189a3d4e5a90230652765d096ab2ab12e486081859c789086f448380c2f32a18d92cb0070b5e34fd5910c0b52cd690a0419ea10789aa17c1532f01f4cf3556598e5fb493824a272e6a1646a91c4457bcda48fe309e1c77a3dccc0c78d7c4b45c5caed68333b1ee375a79cfcf3f2cbef486ff0db792f68d6ce855b5f214b6559773ab9f4e4bacaea53b58da22b9ef8ea085c39221c86bf40e53b6e5d2698345ffc27bb84f76978c037e0287d43311347ce0d25496122e35d4107b3006e6d4cf487789e0d1f26798e3901f805e26a2d63ee00ad700cc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.91838923Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392, Name: checkpoints/3392, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3392?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1b8e159f82c0e77d42cbb0b7e05d7c5395f7de5612204ca5ab0c82e62b0f3ea9179921655ba912343f46308be2ab1ddbb8cbb0c03b70cd1b715d0d49272663e1d29088539f16a59d55f8d47f83f8d940a1ba2bb0c01b6f2dc2fae3ddff1a99cb86bfe26b4b72a427cc006c3f42a315a8dab0f0bba7fdfa9c3d81f70b9ad64a9d0b5d8f669bdb75bd0cae6d0a02041665d0a17867bc015d32058fee70766a87c8beace5a9d0df865144bd2c548ee759068182e74e1d2583d7890011656d89ffa4575fdea16857168f9be4abe495df453c53087a318cfdb97b50e291b510d99fa556750d4f191b215512238aaf31af976052a7d6131039040f666c33c54650f496&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.918398587Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default, Name: checkpoints/1280/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1280/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=c10754b63d49ae5d6944a6113618cf4e9d09a03428ad624a72683ad7dc86d382e1b754c02003bb0ec21d8e46c599cbf42e155254394ea6150d8674aa5f62215651cf22bc62bf4f62cace623fb3061c4fa041e489c1bec2b9f460d21a026d5064688ed1844d4673c65aebf0bed69580e7bd84c69b99bf60a0390701ea90588248e0962cc03462378a303e69a2edf512451fda1506ececf375b903ce4beb2ebf1e8fda6359e8261d208254333f73e1adab0d897a7313cab80c143293612831f69cd442e31248cde2300136b0970466b090d63ef27e608b468bd1be5889c1a34b9ac3390d2425aa98bb98bf1e8d4dac14b4e1931e0391ea42d9648b3abeddc0d9fa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.918386687Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224, Name: checkpoints/4224, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4224?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a87757ec86fd53722dab49ab9f5d195c2f63378c291728d4bbf4b46ef7c202f31f3940c7ddc8080a13a0b29897c01df0b3ee4e316fe57a70d19344f983580650e6eb44e1337a423591b9b9c27aa0da5429f8dd3c21d4127e2dcd17199ff2392428428219255c8069e5b171bdf1a8f448031fdcf46a99d0033c07fdc928d94f579f416378e5358b366a1c55e5f128819d3ace9e71d0d811f8e33a2f490caa07e8f7bbec8460e78c03ef6ffc311a61a3e925eef640a4af4191cc33836cf129ec53e57cf52a224e41a934f829e7904180b5e1685e9d4dcc0d303374d263d31d07d5b66a259de68b92ea3304d6f9f5acfc25ec24d8db400eb15ae999b6883d637b4e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.918552709Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/array_metadatas, Name: checkpoints/2240/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2240/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=13022e02a9a0ba6503e15a16258aab0b998fe15864401264f6396e97bfa7cf7b4115524674e311c4a587ba10abccf531cbc695aeba1393abb4a04dec0b9ff306b8da3b9a62a9ae01998e7ace2251763cda9040ba275b6fddb1bb1f8fa7f68d2dad49f8bd9a347d4eb413c9d5bfe4cc6560a32f77349f1d0ea10234e8bd4daf79c6b5472f727f49876da0f608b83c8bb4873b2d4a0eb909c8dab4d2358fb40fdad7ba806ffe7a230f802ae15a2904eec669c332c644c27ebd73ebe9ffa0718d14bb7eaa4993503d9e3cd98fb127cc96bf07b887ceaba7e415be666efd0de26b95ac45cedf883ce4d3d8ddb1594e30956ef93bd4cd519b3554ab8d819285eb1e66&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.918593098Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632, Name: checkpoints/5632, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5632?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=5eb3dde2b12783e5a02f367a24d16232ea543d9f60d56ce07f696f023c104f9121ad7bec0501441a4c693a1500ccf44aa90fba9d95d2c378ab5668c8ec8a94764088df5c5aebfe20cb5cc44b43cf001e4d3c189f9d245cc8589b2beac18acae54243cb5863fcfd84525bf02ae4f88a3da02f3f161862e5fe59fefa80f25215eb4f52f9e0abd6602367cad78523d67f1d18dc9efd57e68658240bd7cb418668f8ea606c606bace3ea07c853a9107b4be7ca18e74674b5bfdd8e0d29ac7aa0a398da9de3914be467831401536acb66f8e817717aa6e453dffcee46f376be9e65cbfb53d5549ad5026eaf6d94702089182d78418acb9517a3de0c162c0fdd8be8c4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.918662131Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/d, Name: checkpoints/1728/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1728/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=0fbaf6810559c93034add1108d77a9ee6b443af5b8a5b66f2145ce16bbde0e38c4cbebc54fc97e136ada5c846d22f24ed5645e44f8b9152dfc426f74715e80c9fc66880b38d85135b05c2498967a446c53739758013814a4d058de7534d147555355c21e222567d63b7cadbb4e1866554177b7c680b328e132aa56da2c3c690e4b3567c3e602b2d2e0d43783b42aa8f822e9e4608a9b06f42f2798c44d19f426d2352a449592f9c53f96c149298fa4a2905147c0343b8dc502693beba3ef7f1251e2689b441370fb0940009fdb986ca01914f761762f7da68086dc6c8a6c31c67cab8fd9c77d75cec3772f8a96408b45b639dccf51a6c72e6e5cf2e2284a4c19&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.870341762Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896, Name: checkpoints/896, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/896?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6ae36a192386dd28cfa27a006fb3c23d11b0634cb158661d85a4fda9772c9c9acc35575de193db84ec794f7c2ae8c43417ec7d6f97494a13d7bdd38f5bf95ce2d19e560327bc15948eed74f559a1a4da511e75b91ac8f9f469b2633b6f157d3e8a97702dccf26c5d889d8e097199f7c190060111df22a5f73647f4f0c7f03d1e2cc42d07a54aa88599e853dbb7f592706764ae34728bd8bf87ea399666e47b57e7121f9cba69930e3564bf4b6f674b3011284ff32b53f8a7bf56da4f6b63c9b4367a7916791a00fec82598c4a3c511e71f7265a138c88a6aa9bc3ca3ee6f1cc7104af82ce575b2951b328c37dfb34e593ea45983132e3841fe91b19b3f741712&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.918853098Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/array_metadatas, Name: checkpoints/4032/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4032/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=ac2467bb5c5017159a0e9b27d997f944708a230660d29a002a6a0803bbd064ff64a905bc898f890bba3b3fc94eafcda857f1811bd43e5d1cfe3407958d587f6fedf1796196b15b86688eb1f79c81e53bc4781a13873a441afece599de30faf27de081a307e32a175d7855d8c649e0d883fcaa57bb5dd53cc67cc17277da6a7c0663af3fe53037aeb166704f2c22bf2046d67b81f9eecbdbf1bed31226c06443b56125a35acecc6545eaf438f40f871a78b9ae7c30e797b6d901191addb717d186bc68b55a32b6cd669bf42ca747ce392fd31ce10ab38a51cf43d259b4d59146c705cf838e1a666352c91baca351402a8a09fa91eb54e3c8e4652ad7686bdae93&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.918859727Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/d, Name: checkpoints/2304/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2304/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=14499b0a2a1224c09530a19c6c707133fee8e38101ab4231e7b2522d55508052f00a863aa20fbc4e5656f8081ba3998b317ee1139491d44690121662248622f701de78f90b39a22326bc78a3ba54bf6532a6559d23f4ee477fb27ce1be0667d836e8c246ebe2f0632cd61eb5f274a8f5a775b2e940defbd3aae09f7d55194a60c3aa68f2cb9b295c10a97e7f2b5ea249a62a41613a0400cf1279b5e8403b6b9fa27d9b7ca6f6e0ac7b865e585a4f8612de743c313e7da8d2296d1057cdf35d46372a3b74fd8f903a22c61863effb0f8b10d5856e3f47d0f068047af58172fba43144ae87030e0cb9524407bc358cb292ecce636b76a7c61dd7caf984c67a08f9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.919000724Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0, Name: checkpoints/1856/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1856/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=402ce7aab61af1b1f5dfad7718c7bbec48b735f71b2f25d79a01391c9b591c1ee1a5fb986c9c03767a8cd8d5022f903b273689b361b48569a14a5a3d09b061c4b690c58ef5311baa4fae7961a13a180357ca852da0164b4cc7c055860aa5f772e5ec2f5e67649b01fe316166e39d70425612a1342271fc3051056f5426077b7e3b260fad4f8687319921762ecda7da8417891dc3740024b2ec36dd56945fa5ec46846b5b06b9fa3f6afb21a94249e806f5571bd62b6b13024210e6f4e4a83bd2a9e75a66c1701be8f49fe38b1556436aa6485ae0f9140f9925dda44bb2dd7f2917a72b18f2c5b5132905c4601abe6aa47f62ca4df2ea763e2b75c7455eadcada&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.919066944Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/d, Name: checkpoints/1088/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1088/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=3be720f6748b4bc35169c95811c412a06a9dee23744cce03367a48c8fbc209843f3a38db691d7c886fed61130387f049c602588376f213d1714bbbcbfa1fcaf2df72ea7dbb7a60f9ba573d08ecff5e1b56d51bb458598ef1af82de43f18b9bf4e603b3bc06b9672960fbde8782c43c04eb0cbb0b1119db9f351c4875a2513d60ffa0932cdf8100a6134bcd114cecf450e2a032b4b7e72b9f66171b5fcd978d4352ca93b328f788201d164f34e65ee6637f4d571c7e9977f765e436857dafa94c4e5a5d8e97e591c96f0f954bc22d466d2a33e4d32d23afbdb7812cd2c3b7f77ea0523cc795b9af5be229d093cd53f2b2d3c08a5bd9e511899295a9b8bc19e163&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.919152064Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default, Name: checkpoints/1920/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1920/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=7a1e7d20bf6ed19ada94c4f4cca2e48579b8801fa10f8c64169f6935415175a748b4bf107d998c2538175e6509adaf9a87a8818ace10713008ab07b8ccedfcf578851258b1c65a8904bcb7cbe3fc5a5cd4deb86920bb37cc13ed81cf153fac821661689f07681268f78bdb1eb40f928b2bfd8c6f48321fd56ca8cccd4450880f19e88b2fbd3e062e55d4dde9e589e99c0ba03eace1a01b55a836b2deb0434184777295eddd073a1bc23759a201b2d72d734888eb1d8bd63c040841970b7ff0f9e92c01c2d7ac780ff1c66210e7c5f537aa1b19198136c870e35a16a29fb825bf37c37089c7120eab5cab88b6e4cbe6a753472edb7e6c425de05d518735104b5e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.919150924Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d, Name: checkpoints/4352/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4352/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=6949b27fc8aaae0ecabb6f2d9c60df17876288ad200ab0b90955ff97532edc4f4196e4b81a5152f558ca26206dfef407c9ee037397907f8e65783509937c7142dfb7907bbb99b395d6397bcfba18c2b1304b8d26611245d2dab914000b2425eb9ff2cfead3d9db40c4d42d78c5b02b4c135c6c0354516b64069cb214489273354926d567d4b61f9c8d4a67ecd6e30ddec5c0aa2a933c743712b35bfd71393732f62a7a0f2a05303766743d201fdbf4495ff9dde8056b3f797942387589a5e5b959cfac3ed8857c4c100ceb5932fe2751d6f5d5b4b8176837a7ffabdd3031401425e21d0a8945e6445660e71ba860d03cd20ba61c51a0e38aa25cf6b39bb3a8f6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.919194475Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/array_metadatas, Name: checkpoints/2560/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2560/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=1e9d9631236f399c274ca8ff2e542530913c2bb05dc785a5e05d3fd8e10d0146b2172d70dd0551c6e4f8974a207d0cf0119d709c9bd15e5b59bfa4573f75fee3925539b7fc296b35cdd8a1fba7fdf6101277424f2070f9561481c990cf5253a0d68061552a3ef384b3bfa09fe0cc883e4ce7e6da7ff334d19dc31b68518d51e2c347430b9e94446116b29afd63c6287456596f84002d566203df0a7bf114ac04dec11915d12ce632f98bc668f5f86d129d0c13ab195987398de6af302582a4d4d5cdd06c3399e15aab6810e8dedafb69ac0729343412b083fa44b8bc52a8f909f3cb41f632ced3c106a27076eb7cf58986a1482ee4b2061dd839bcabd1acbbfa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.919196709Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0, Name: checkpoints/3712/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3712/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=9eef5f59f2457ede3db80562356fcb49adb7040621999c7bc0c59c7edd2bc57a9024ee0c0efb3a2e050b9a7cd4de9654e668311e1fddb769bacfb2859bfdcb3fd28c39a1fc9cea608a115f0d40a34a1190df44290ba0bd436f0fda276ef199c196bbaf452c366457bc65624a72df19494c00236088683cbdb7f42297f6e1bfb57094839ee41ccb8556c1de17ef9f35829bfaa014f00480b683e76370ecf1d3ea8048e297b892263cf9fc82542eaeb912e05ae474a4876c2729b9a1057809d309a8b4fcc72c0da730c121a7d1e8e9f2165d42d1037179db74ceb1e4806f66aabc6d0a537f62f7d10b448dc3eb2741fc2c0b6afe24efed65a0c23f78b24f2fad59&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.919246718Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d, Name: checkpoints/6016/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/6016/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a21ee77268cf45a0024949a08ae0d5b54d43cea4fa6320b81edb678a3a6bc2d004c4677d04118042bbe9d4263056648cdf65123943583ff7579e1ae83f4a29d7279a07973b0cb9ab6268c2a85e4ed8fdb16945fe07a2a74935b28b17a96712a18ec19a66a55668954e7c67db1b2c47eeecddec3ecb0a69383fc9e7aceb20068d46abfdc8767e556e052a5368b3b6ca50d3bc346835b4013766a94484e8ca3297c578d81addccd03a737c6e1a5a2db6d610088b7ecad1768613bfcfbb5916b8dfa4f57a46788c68b2cf50389ba84227c3733272ee3f0f66dd3006f30fb481cc57c306fabc0dd892406418211e03391116fec9a3310ea54ef397e8624b0667e7ff&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.919197524Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160, Name: checkpoints/4160, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4160?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=85c0e8c45ff465fef54f140f5ff4b7e6ecb371a04efa9c51d9b904a6a4ccbe5f1830c1d76f84792f51e5f48a95d7108db5118a557d76cced4172d2fdeb012b4ff3199d9c73b47ab2e0e298ad3daa1fe593d7098a76f5fe1d1f20426a1e45d7fc03b9cd272557dc27b7fb0152d246ce4a30526a220535275db90e0cf15252150e88fbb59b697383154ed3cd940af0ceccede12d716fc137265d919e2cf9706e10e4ee70d0cbe2b97d0f1eff4a36ee15dcccc30713e365b750401f35684cf5b0e056ea9d7f1f8cd8486aad982e5189e72354a66054f9c4358edcb58c62bd277c4f79c1ea787ce27629bb9f3a5389c7c498e7c36ec409544eb454d38d39c63f1bdf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.919287033Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0, Name: checkpoints/2240/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2240/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=989eb7de495ca6e1659677e7d7189acb01888504c7e2f6d816eefba33271355b0ef7180517fed433de1b321d06cdc9cef0fd295da7064c09ceedeb71f0f1ef41066078665df05c73a7272d20fccdd7ea02c127aef8a778adaf90e7bdfeb6225dff13fe951a7d3e13b3acf57b3305787744d31dda0f11321fa8eecc1e0cf2b6b3e32198f92dfee092625f7735d21573693155d957584dfe6d94a8dfcd360bd4ca462a6d8f415f79372fe2c308e3c4cf86208b09be8cdd18b4bf42a6cc0cf790aee08fb167fc4e1a156bec8d7b2f2c476cc73a5d06afdd64c39aae5ebe0a13c7f22fababcef555f708b73af88bb17911afdc97db88d2577868e3a03fb6fda327d5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.91930745Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default, Name: checkpoints/4160/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4160/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=711777a936254b6520b10133a95edc46ca4f8895a4ccc99c20445d721245ee7e4ecf7dffe34abf785983f1e81cc4a7e97567261181523200507f4ef9deb52f2900ec4c5b34c390f63e7367a67e1c67bfca23ecb91f51bfeba439002a8f7abcc0bdd5149b849524c0da2e78d9aa04569e556a9300b242a4e086444904fe05cade0164ca7e087992ecfbd813754289d834a7fd33dd96c22834d2a0c0024d016b1c7d6630bd96f8d814a51beae75f73672d6905428c16d5bd93fa91232680e7cc0dade836dd3e5408437422af7343079c0e5dba5718c052bb34002e953feca733835d7fe7663b07fb4f551beb835dead50e1b17b6e2a0cc697f6ee46941d1765614&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.919383355Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0, Name: checkpoints/5504/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5504/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=03fb0abec8e9e09f0f03633eccbfefca2fea0a3f3e8b9370bf00a4582afed5070e4e24953b8ae0808fc5bc6619c7012b77a19c5e6aea7ab8d473032dac488a1cf14f4cc7f1a97fb4058ba9a40dca352350531ad4c31b17052c5979bf237822fc34d4b478442f43bafb78f7d067f95d656a66f61a7b128a3b268a47746abeef18b81d76c53919ecc2d2b323f28baefc4c5b25ba167ea96a6a913d292153e929750d77229a93ef05bd52e5d44c87071449a98e0ef1ee2b476018f59f063d3474556d3256793e10aecc7f7c33a47b79636743b76344739d9e8008593a70e247b95227fc6569c57bc64a8d55b31d1b94c4d05d1b65483c7a1aad0f0ef5c470c4e0c0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.919423399Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d, Name: checkpoints/3520/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3520/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=845474b1b7a56528f8f51bd21436f1ccc45ccffd1a649c14bf1dc56ab85b8cc1f8661f9af5d314538a0e78c5fe9101b6f1bcaa4b40f98270d8e5feb31564e9ab1ffa8b6311e67d1434b8b527d3c13ec1b1fed4705ffc5ac1daad5552df121df562998ca0acad7591bbed70b78d1b732e7f938656df72a2ecd9f24803ca2606cd61362e7214778738c8bc46dc921f2da7aa42a5feb0477e0977f5593c6d1cb839018f86e0dea834de9ca908c7b98132a4de2c9cc22a69824df70912883d265b8b00e57a4294c733d29fd5903c1ff0b9194b3fa1c0293fba2702bb6f3dbe54f48f8685300e1531510e1b8ea6b393de289c70d45b374d74a028e822525669d03598&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.919568787Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/array_metadatas, Name: checkpoints/192/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/192/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a7d15dd9b8cc196824c6e6b332f989d983ab13bd483637116630801997681a5e2d70dbf51a877da8e44fbb1f178d95bbacb6bca756f55e4716b642181597e6247dbfa8a1ad7cd9bd4409c70ebbfd1327d917964242232e7002aa34f4cd02761c59664f7d035cfccf50851546d90b742b7b49b3450e340b0b6e9b45ea1dd476a1fb5f2e193fd8cd1b0b05b1a31be5f56425ae59741fa645b3503ee5a557375bb96786daf5d119f6ee2ba196438df6d2699b2dafe24b68c3e7d97ae188c3a68458b04582c72aa7e57ad4165ca2942c40670b1f88cac8a9b78b9e9546cc3f9f0be9e8949c401ef257ac180fd12bd77085e93949f1e19cd463fc4d26c29eaa7e0062&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.91989457Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240, Name: checkpoints/2240, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2240?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=a2ecdd26369062b4670a85b2db2fe73babe21b779211d1cb813096f6b1a065c5ceed260ff34bc2a5377674ab30576aba9f5fa038ef94efb47aec444473bc906916e3b33c2f0dc4887591b9b47ffc8ae71fbaba0a7877ed2e8d1afc3264d21b54dfea3c1f3ebc7527e5129270700894de23d665941b8018bbdb63a739d111e7c20f5dfc308a9d93aabd603532e271e5ee8b621275966007d5afbe2aab9cf44076a09259f36cf9cffbb902227094bbaeb827030b78591ab227c61ba6b0cf9c503509a5d7c957a6998d36e3f8a03469bef09929836999f8aaa317530250f0ccfd6306c3837eb1ad4db9a4bbc1e5c077d1aedb295cb26b0210bf3482aa0c76ac9fef&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.919920392Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/d, Name: checkpoints/4672/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4672/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=691ddf0e8eb6bb633914251b94c572fc446f8e9da7e159a2baeb6c74ae3ee47901c141466d4b0d146fa502c52ae44e8af668e20b6c2dae54bca1cd2ff81cc3f40a4680b06e419fbd573c835678d305c4036659aa0140eac43ae9a47010f2c218c37e8c965c3c294b71af279747801dda02bd23f3ef2976cac9910eeec8b7581d0524b400c9ad2088613680871292a285c386fa6b5aa0f9ff9958f4db5b27c267649c1229eaf6f9f509980ecf30bec8f9372c6f12100870e5ace3a827040b5bfb096f30f25ca86ec3463eaf8c9591cd8e65d4d3a9062c54bcd8c850f61514cf8d72a6cc2fadab1f65d4b1fc84facb33fa5bba58defd37b1d170e6083f264f6b6b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.919976243Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368, Name: checkpoints/2368, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2368?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=5609aeb383783e871a7c3195ccc80f2e4a2bf89155ecaf7aabd2dcfbbe033c44c2991d5fca85535fb105fafae8914d22054c5f78464a9d7163da871ac576b365fae094445fbe31f79f70df6e70f97941690b7404d6925e5371142cc1e944c1e3d0325e4343e07300b7e225c9d1e92bb6feefaff16e0928d7d15121097ac2bb8e9769932ec994726445453873fdc45fb511064ef272c21fcb5cc54292e952d723176f2213abdc4dadf7839b168658890f7417d6357d4d12efbf6205c9cca86fd800b79fa40c5f917c7b07554fc6775b1da35789b393fdefcae296e3abdf904dbc4431cfbe8862d42292a69dbe66505d8f24f65becb1993129d3b07f0ad6286549&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:46.919975063Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0, Name: checkpoints/5696/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5696/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160346Z&X-Goog-Expires=86399&X-Goog-Signature=c4cdb1e6e22becf3186fb37f439d51ae55976654446582306eff3cc6ee7653bf18057519f1863382963d5953c947acb1205e4efa687e60d0b4aedc21bfa9ab09677e2bcb7b58a94469e37f3496be69cd0703b2f21c10c32526bd56c8675d992e9005234394e05c045a2bd70ba4667129786f88b9c79ec47005b788ddefe1b4c776191d488112116a4e94edcea7964f35dcaf19cbd3e07c1a52ef8b17bd9be36469d56b4c1e4b9d233a0d38ef4764ec7b01ea9ce6ccb1647f1ef1a8cf7c91a768b44ea3b2b2f39d0e33769b842d0ff912c7bd29f903d13da485cad7ea20506b0b3323037bb2ecd25510c96b2329ef297cc348fa1636acb629b25b4cd0b9760294&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323491082Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928, Name: checkpoints/4928, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4928?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=4d24d8324ff72513883527b10f928faa377662c8940cac7c4a0e9eef2b56c2a513af4fb73fdd99dd69339d61c48a832721b47fb5d7eed37f13f6174798ba0d6f734d29d6c0b7cd5fc6135a1bc3e56d01a683f5951415a10982b426e6c0adc733fc1f89207cbacd033077db9d2fbeaacce2b338225cb252b3c8a96b4703f92935e5c2cdd5a11f96077d71d007031bc0b75959f40ee3fb1d6060b8b7514f2b3e35847b708ad368e8c1164b74d85ba4e41930b0c85616213491a276ae99a1e516d5a042542ac46f243868535c399e09dba902051776e1a3eaa98af55c959b2c3fd8feafad78f004bdf3548c6bd80ac614877d72cb606e072a9d82ecf2f60e752ffe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323518138Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default, Name: checkpoints/704/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/704/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=967b9845a1838f96be1ae88cbd7f8ce7292c11579bc21431e8888e52664c35bea14e6099b2c2a243d4757c4c5dfb984e0aed42d5381aecea2178f37168e5903e34e35b7b366bf3618b9bc6ae69d9c475c75d5fb03ddbb254209c368fc0cebbd59ac5814294867babe11c0002d734e088a9d487d03be03d3845db87b9efac9fd2b78930bfe403a9ca1e33bba184fc6ee14156186ac9974b73180fcf7ffa2b5c357208981a237fe830304df42237068108f60941f43da9983a5723fecd90ebe4e6ebaeb657229fa45069a2524dd7e14890f4c3a037962f8f53ac06eea9ec690712dddf6ca6c9de3074d51854691867dace13a8525d92a59efc53c1b94ee27bd512&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323629357Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/d, Name: checkpoints/6016/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/6016/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3af7c268bd31cff2fc484344806228e21e84243c86ae1cc7dfe346e89aa3c996a1351ddbbb6867ae09af95e197d85b7f448f9183fd6b29a9a5d1d833f8e98c87b67f0146fdc34e6af7c59338f4a94e26d2decf262929be3f4d74b002e3693abb6c27666c0c9661b1fc22ea7b20697430b6168f83b2d87f6c2231a08c104f98d3fc6d75328abf377c266e6a0244e17be567ebd1b108ab8f8fabd6b6059b35ebedd7e88f4d2d08c2af1e600b3f7f1430abef72b1bd0dc105c47b8c966b02af65d74a9646d1331b463a70cc2e89e0384ee0951178152cdd91677f61c222ad5e746bd62e26229e808d7ae53210b4c6f8d77955f0e34186d6fc87052695d63b2d8d62&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323666261Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/d, Name: checkpoints/5632/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5632/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1e346ae8803c8dd7edd71123dba097727f5bb2a9ff4d93c90a175882581d6b6708772e900dbc7d81da3b8099795e818529c23a481fffbbaf5880d5caad411bb8e723c4a474ff24784c37c9eb27a211d8fe2ebfa9bd056ae2ef0cc309d3dba4efed0242845bc95c76abab50f366b99046baccb7d8390105574e6a60f14763cc9b4920b1da67d7df7acc3625e6c8121ca64b96b10aec9572ed7e202d6572609a4623b845ee0eb0878f4ecf3649eb1bc4adef8a431f87106ed184c236dfbfaf88924d7926cf6b5a53374887d48c0dea1fd898335b1fa1fefadbc30fbe301b6ea64d417e1c6d9701eb9c16df1b3913c8889586c2076afd793492db2042a5bc109e01&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323707759Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/array_metadatas, Name: checkpoints/3776/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3776/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1830a63af33b44fd98bbbfd389a003698be3a95d36b724fb4424595140cf7bf510c38e42a10107351ed19f63ea2dd57630f520a394571e655e70bf1b01ca1b9380e2dc43f562100e80427cb6a84c59a4c03862bd991c2481b5bdaa7693b34ed8ed024436da7084a7f5d947687ee884935ff466b1f401d0efb811ccfd0af2faab3e9cb1229e21a5ca223cb4247b2673a13d8c153c9f09a6e76782156b67a96d3eee4a043efbd936d8f894bd427916a396f91fcf0494f81ef761711f1db142039da8bcb9d7a6ce92d5e620faa49f30e44303fbe3aac0f750f4f63262627b9c183e18b997fa3b286e27a70172db071bb7243d9cde32400d585fe9f55e42142e69d6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323769635Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0, Name: checkpoints/5312/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5312/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b832ca78d1bd24a1cb4b7506351118cab92db73d804e20b3dc232108532c56abb1f1ffc90e9ced5e0f0d4155bb94b771cc4ace8ee7c7fcc115ff035d9b8edd6839e99d8bd6acbfea8fa89183b4d506f6839a568b275bc07b6413b0a3484be206d022e914846d155959b80924b1181d8c478ce59c1e6c8e47a14b39c593da8a06e2a8b6887ab3094b15b4b8395e56e006721b8d8d6a6ef44f064cd82036be39f79d4908fbddf78e2f9d818e884195074cb1b51306aac72e38fcc3f55bff8780baced9c8823272d6b7f29442c874125797ac55c75a5fbfd19ad043bd9659c32af8c37b7205ec40b7d3d5cfcba6d1e9f75f1ef1f897f7187661913994a05a33e7be&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323823067Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d, Name: checkpoints/4544/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4544/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1754c4772b94112b5bca101fb4cb9bb78be34056c7505af50bf35e5cff97b9bcee40e8414b1e81fdffd99b139debbe699b11bc084bdd7a10291b4f2da635a710cf8fc0a739b9d4cd3249f9efe750a8d4da4fef1872de63eef9054deef5fb29774c380dadb2b5d455207523f524c53d4957809edfd9b05eb8ce843aaddcdc996eb9cfe439aa86d3aae4a1b9c752ef6aa3093bad36069ff401565aa98068c57168c2b8426ded9b6859388e7497d851c993b17272f194342b621b9e7df09f240f40f4fe016c311a6b902abc34ccb59d6200252b844b8b3b69a01413836ffbbe19d6f7482b697322479434554a95cc8c0c77ba7af2733f74e193f3801158c95e13e0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323673338Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default, Name: checkpoints/3968/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3968/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=68b61996e0e1e6c464f5198e605e519c49d33012508ccdeaca5dec66600fa1be0c4c723b7bdd739f43ebf5021ebac5eb215102c7b2b5a64d4a7a78039a234ae237da77ef469383dfd29bfeed1c0d9492b3f7dc908d3c90cc906b9347cbf0b7f86bc22b2cf57cba02bf2058184822137652959f0728f4168530cacb26f2f56f6f807e44afe4969b62a12ca36b95dfbece61bce799686da108638890de1aa40a518dcd18c03af94a906982c45a6b6c5793bfe4a0c50dde99c2a34a93715133ada065ffd5932a062d1d7b7af77c5837fbf092dcf6081a6081717290de917fddfee828e916e2a4fcc4a836520115bf3548aa7f72c6e82779b44a86ca2e367cb90d3f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323855822Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0, Name: checkpoints/1088/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1088/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=252e045393da70f3cb469097ebe01124443f9b59fa75a3a41b3caad1e31e7de8bfb66819f5fce3a345e3d151fb127af0fa3e5daa781a874ad81e3699dc8d157e3e2ee02a2ddf3e9b4c7c6dd65ae8a1d0dbcb4337668f0b6f56a0e3e64cdb2b1e4223a2cd9d7d5b32fc93ca08da0549e5b237e694fda8e1edb1b990265fc5af40274b9fcea0ab65f88caab9167386d652b14940d5fdf53f35af948d82d3d2375b81863d4197c65aa5c0c0f7d44805d9f625932545bc554ff8d20c381baf94facc6a881940f3a745d33984f99ba98c863e54e88b45bfe6f97eeeefab306cfc93f9b762bb55f9f30eddcb17caa44520482cc814a078f6e73ab0e358bd4a56505127&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323897944Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64, Name: checkpoints/64, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/64?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=736c61e009dda33b6bea31570d115414ab9fbdc07110931e37b8bd9c488b73419094f8ce2282f055e967e604538c67ec8ff6292dd7fd353b11ed0ea24571bd5ef5b1a484435815430ca1aa862fd89d2d307f5170c64b85bcb1e9cb08b610a858c78fcda6594dfbee62849fadfaddd4be4ed4ee76a6fe00f36efd2f69c8cde329b84106faa807681b07af81e014db4a60b5e79f78691d4d877093ae7a65ec9459653b82f6de6be3c25293b43064d8ea53127bea14f9573e2b9c68277ed6a5e79060db9a989907337bf0fb506f61f4d6af5c74520cbf9943a066d038b188d3894c534233eb8fe27b25cc43efe700aaa805569e29daedef90da703e0de781292e3b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323923966Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d, Name: checkpoints/6080/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/6080/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1ca49a2c39694f728355008d02098e53876b9c20c9575c3cf07ed9291b9cb7d8f7d59165607da55abbb3a035a21db2364c6c3d6895f1958039424222a0c2c9ac4480741bc37d7d5f99482282258e6fa5b0ad6536e6ed78bdc120e608394623e2129b638544e3134333b88cefa3cd67c5c155ef90d662c93b00740fd15b7fa7553c65cb8af86f3ff56155a8ebef687a95445fe09cc88fa49f09919d10988c1eda4faf0c22db64baec0fc049e375fce5b88735fc8420d2d85e7880c52ef11e364308cacfeadea2cffe66e2a9efd7b4ea989a7390c6d4ef97b641ab6723ceee83e79f2f09fb7611d077f6c1ec47b1443d9c9925e64c664484a141194db104988489&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323956793Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/d, Name: checkpoints/5952/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5952/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=20cc442128111a407315d136fc30419fbdc48d1538b057c66a56babdfaaff989038bf51f8e4a1b9a2f31437223d5dec8b3b498da311974385b69a20cc06c4e4074e6c7a34a5a8991e416dacbf6e3b9dbd7b72da9ecefcfedee62984358cc1610411a33e5a6949d17e95eb608b1a0d3a17d4f5d4f1abfd66f76ad9742b7490e16a326c0ab1354371e07a6cc235097b99da688cf664592457c87ffaf8fe7085e81230c8fc1c6459e1c58804a3d837371f1c10645ce5e2334a19da6177ea0b9e3acf22f932d7a1363b3f2cd0a74bea3329c2316941252aa60414ed43c2a43d8cfded3e9da2a817ae669df9d414fbf444255cd3e921c5cfe71e4ca68a6e78cdb98a8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323641579Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0, Name: checkpoints/768/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/768/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=c01aa64472e3351efa196956443631a3435b45c2e5e2624ea425717b9aedd60d78d2867566cf9ba2f68e936ad57ce2f966e9a1140f2c7bebcde76ff22d285fb7ed676502935928cc0f438636357768385c0f5eb6a8b5be8fd281ba0515b60043a0b420f52cc5eb22671bf7f1368cbf3e32aad38145cb70d90ee211ead110d7297486367abab0294d0cf70626a3dc880a310c87cc892693cbf3415c20d8c90af0daef51b4ed7ab9adff2a498a0cb09804ea741c6c98d4e44853e058f35d272c84a34d749ba3942ab336e24f349006f732f3a684ebee0b14e3c82955fffb027b9f72121acb336b9daed13042d346ae1fba45d2a5831f6e3b1408b942211b21e4a9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324016147Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default, Name: checkpoints/2112/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2112/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=ad455b3af83a4a95da51cc565fcfe444347aba8495085c62606663a7804a24cef67b048adbca642de82685579cec5a02fd92e93ce479d1ebd4496e1ecf886a287e11d59ba5e698c9121c2cb0266ea36933acf841c053d7cadb8dda6b0ab1eca7c94c29ff91ed7bfbf07837ec02ab0667e0e984a74f69768bd2d2a10edec4b3bbb545180703b87494580b721152471c4fba1c5f22d1ba56f086014610da0f855b518170c53ae024cf67f64f5f02a45560efd007d3109f53527a8b3b91f00b73f39fbd5c1ec7c01b0d80aece0d03e4151bacd6e635f769cf1984f931927fa1933c8ae8d99858b2067e8810ab066977357df2831ea2007ddb8be5767a9eba141fbc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32404025Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default, Name: checkpoints/2048/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2048/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=31bfb68410a3fcff56df85f41b0517bcdc11444bcf4075f3dc79506b5633c63eb9ad90db1a2e6d361d51a225ffaef07f61a70b2f1c7251f49cce99010ccd8079690eebeb279c68f7095bbc552629e56abf7cc05a87fd1f1044ae6e800ea09227d1af1066509a5b2ee04573fb29174c18c835d94d30f07dc58b3b8923ff16ae352dbfc3a343143bb3bde0931a418d20b558764bbf328cf9e1f632acbd6c99a98b5c8a562731a93f4069456a695a2f9aa33ba2bea4aa4bca6cfed9d3b22e7fc80df6ce0ccbddbe9d06d6d1e48b4ac596ff1b5db6801164a0357f808d07644556dd3c097115970b84736f05014236fe23082658e2f737ad1159ec249fada4f90abb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324046933Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default, Name: checkpoints/1024/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1024/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=abd59e17ca387f82268629e9361e8548d1c969618d5cb5d1179f87267f50638a921f7f20caae486424286f6b621cd2e24002ce30b953df21d8e0c119e0bfdd5c33d06bc16704750c5ec380a6c7de9887e2d4aa2cd88896efd1c73117bab7f8eec1899d62e1651e491b4f2ef5aa5e39eaaebecade0dbca278eb4e2ec1610ea87efd8ae9382fcc37a338ee685e81fe31c4692e9df3251c32b22725d5fe69d9a466d483684587195d6b08446e322becc056101b0a4489b29853229447e2ea6a40f839d4126bdf04b170e24c3deb92ce77601622a0690e5e8f9b96c33511314ca5e7ca5b1a800e91ddb86ae9035bc4b052f480a7ffde592c93b8987e81abe8c177c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324087722Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d, Name: checkpoints/2240/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2240/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3d9ffdc53cc9b1bdeb4ef62463c8ee8b3f157d64e51724effea13ef2e757777003368543646c5b0972de1c54382b431f73c7e2f9fd0af085e6cec067a72b4a4a7b92af883c9763d63787b3b4fd6948936c02af287fc62a8beb96576dad5e20cc0517105161dd9c7b60ead2cdcee649adf001d37e4a8c710b9661b9b38a76f4c1d8fbd0ee1a4c8cdc69dfb6891349f84b3eefe66c54235f76a150050e25e5fc20e8d7487f01b1cf155a83a1ae27e4dd8f1abf36306209022d8a9b3e0731a28f60a48570405e27dfd53999287088d9396ec6eb2a5a5190ea2694b516131992667b5f6dea213dd19915f17fd308446921902fad5abcb5fe74791378a40a2fc0f6b1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324116096Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/array_metadatas, Name: checkpoints/1088/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1088/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=418d5c488ce90f901a090d1fa01c6f92c54b7a15572d6c2a67705765c53b4e7932a2735b7795ac95161166a50bba9724691a90c498b44a5449e147fd8c3b09820eb7c31729fb044a63600a5e2ce41f83ba499f41975c6d1ed325721d2c57e82e23d27a4d75f9e7537479d45bcb7d8591c735832724af14bd44a25c209540a3e428967c77d86da07f4e12b643912f6ae83f763915045a249a8d4805272e225aa46471ea4288efa8009ea69b38981b8fb3f0186076c8f120b23d948a4b0d954c1c8c798f42056877af01a7f2284f8834d6aebf16b5eee3eafc34d0a1b6bc397991759a143ad80410d0916906c7c9cdeafeb4fa580ddbe317cb3f3c055ccb5a44ee&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324150718Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/array_metadatas, Name: checkpoints/5952/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5952/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=88f88d5c436d6a6e7d5e119431e230ca7f9a3e8977cbe985e9c36fab41170d8d14dfbad1be5f28d906a7e883879d18757055d1146e8f7ec5ddc2d4b340cd83fda7529cb966cae8d895d87a7f86a04f50a02fc284528434a09612fb1dde34f00ec84888bf5cade55c1b549f1d661b2100eeafc8e2ac53104a375449ea6c532404e97d33bbbba7a6d0553267f88403eda58d9408248336ba190483482a59ad7d7d320cb760f131e2c7865a930deadf17bf7135730c79e485eff7beb9bef18a79ba60fe3cd8b8b578c1f501200c7a62a40a10615444e20bd2a320650774ed9d3660e2fd15986a30eaa4b73fd0648ea134ff9c403d6736db10f2c42cadf8ab4a2aae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32417842Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d, Name: checkpoints/0/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/0/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8f5e54f9c571c40ae8ecce7edad54dc8e0c895034d82c37f65089f1b93af539ce842dff0468da7cd0841629e939b2d9bd03d1ab39897d26d7e2fe1638d9d730c19f8ed5d1b7aba8935904b8835b6468c5edf56e548a09931e44a373bb49f593c8945d74dc7c01061fe7b0f0d907c12648b04b98f0970c55d3269c9fa368b3c56679bfbcb5eabcc02f4b7b60ce35013ea51ff0fe2cab065849c67c073d66049fea3c981f8012d91541baa3d3bcd3ea4b41d176d247c5166eca4d83374cd4499d3a530bbe1c3c16ebd7c930c2b63b48da4a4a118c78de228ced3c6dfaad5cdb84c170048f0c1a6c1d9daedd3e87b70c80a72e971840d0f92418fb9f72c3e217029&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324194608Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default, Name: checkpoints/5248/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5248/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=07e3b73d452fe6b046a30f81bc7186a4a6acd564126c9287a090928a3aa27b3bcbbecce55b6c9fa24c4dea10fe3c6776922cbd1a7c2575eaf0bf67e77b7889a399fc46b286a155ceee1e08bcd87edf37e84bac5a1c64c5ecbb14393b477d90a5d43e93401a40ae8b59a66d7ce406b2a93a70417336dae78964bb32febec8fe9c4e862dea6ef438db01a1db3cd89658ff55899bbab43aa7655155306d77e1785ad02a83ca1835e830c6a4263bfc51cbfc55fcbcbc6644b37ff9aa75fcd1381d5d90ba47e919dfe14f56a00f4c0226d8b5aeebd667c5fa75029bd4bf8ce81f1bf9dabf6d4f765c2728adc08999ed1fcb1ca1b34b4581ee65857548bb2730c99f3d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324222573Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0, Name: checkpoints/4352/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4352/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=66dce75f6bb78be92a03a40e01946e880019223ac0d6e00af551f8d629811bf7bac8e187691740a143d2ab8843f3a1feb80d8a3d7cd2ebd7467822a98623d9f8c8e305fdf40fdc173cdfe6704fd374d9a0b6be94f5fc4da7efc7db5f0a9951527b57b696202fdcfef0a2500e0387e8e1a6dbdc09c929f436da9e76fdb40ca014479edff6414db108650d03b4e263278c0ba65ffd819110cf25c4b335fbfcb326fb35ba0e145f00da0a0635b55ba959db2dde7e7c5aef0b67b3cca1b6031784a1edfe52265d1f1e0441093632aa75991f0d9e4b39083fb596646beef3c02c67d4fee8cc297f43f5fe632741db686303416b983393b2355db56a6b174f41fcd1f6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324263597Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/d, Name: checkpoints/4544/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4544/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a7a31217750682e93d2ab0a887ab01fe767318514a929d75b19b3e8548a7e02a8e1f7deb9796da32e0853f327ab93471ec5825863f445be5d0e270f2ef46a8fc0764b1ee94b173774e6a788a0bd08a9a4f483c33626f0a27ea8314df92cbd42a2221128d0b6c5bbfe327977ecddb69f832c4f8e6d52a0d8f7bce21161f3bfc0b5f578d091eb865aaefa97b25b4751bed2fcb980fe2694e54d19edc4d39f2669d8a1a153d22fa8f1d8863732003bd479a86d3f3cbd46fef5eea6c47a4749546bdfb080e65b3f97c620d9e31e70a355b3a1e84a9a59e0abf4bfd42ac42a9bdc8f61685bf59da749466c8db61b9b4329128e7e8bffe5e244758bbe587ed1f922623&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324282579Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/array_metadatas, Name: checkpoints/3136/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3136/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=7d3a1f5f1da266f6239975ad1d8215c765abc042d5825baadffe035394ef5425a70ec61148ec79dd83113b1cc7d3e4fd3ea65d377b407cd8605f7317b35dffe7017858104136c0fe109d5c3c36a00ae01e636b07b287f2f4a6e5dcd2978a0e9ca348ed50bf8f2fae74f3b484df68f30e0ca1310016703584649a44827ed06e09cba3cf5e925642f78f5be98d0297936af960095c567c6e24764057f6da8b7cd98a5b6e12c59df98f7c968364c23b96e2b955d67460c7cb0e0a9af511fb582cb07dfb8ecb803c78085768139cdc0a3131c81c848963bc7853d893806a66f491933cf8458333e3b37f83153c30cc0919810d23872f01ee3535e6c97dddd01c1c5f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324316195Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default, Name: checkpoints/1088/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1088/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=24caab85da8bdcc41401cc5ba670d2bad62bf79ba444d085460e31d6dff6c94f39f404c55fbbccb8869c25f209169310428e9499154695aca584299edf116490afd15c8dd8164838c052ac54276202c7e118a7cb18950434f90fc7dd793e8913d5167c02e5b900ee8c46ad0d809cde5486ec2a6355c24c11c159c9fa9158e4e79503c29af2e68fbdf3b785e8cf1200553eea8c401dc542f27588eeed9e0f853b438cac488eb3539772b55a17503b67d43774b48b7927f54c34a156adbc9d733398dc56b56c5e373f400bb493cb99bd5d9ff2e60ad841335a907c3e32672de413ddf096f59ea4f6f1d1f18b5c69d3f62dca8bbac0b6baadf7b5b20845d6a8251a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323813251Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0, Name: checkpoints/384/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/384/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8393fbd2c6a40fb8280de3cbd0751f96baaa9fe797922f7036a186e2afdee7009e35ee0c6000a85de3abef69a42e7979b05ea85badb81f09c8c960e443c4b67b89d10a51c4aef2225093111a4d984a7e8defcfe48a9a3ad5d31f887fa8bc87854ffa3a8651d636622b9d57bbf2a7bdc37717c01355fe7537891554b1d5bb5c9a40c38ba3b6c3dfa992f77cba5b20e3e38c2db2861644aa4592d5057a72658f9e95067129579b1471b9f5abd1b26dfb4f4769e6807b18628f4df503238cc3489b70580d9b824673914f323e00d5db0958c4da69aff2866211f94b4bf11b624e86456914de8f29a970a515cd97b7f55325f29c041729ca538d2b30bb0d307cf12d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324367502Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d, Name: checkpoints/5184/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5184/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=5ef84cdbc262867e6c58e1dafa373bd799b45a997574b6fd20039af3d727c8d5133107dcf770e634a8aa827d1d1c3e70d76524d7141d19169fccd4f31900306ec17f355c4bd3aa87e46d838d2877d15d7c87dca50191d3c4d82c7169e2f600c89948b78667fb58b8b57f79acc1ef4e9296aa538c9fc0c5273b4f015b1ed9e517a07cd74c6152bdf00a4922b987e37e161b30a5e894757f50cc202b583daca7dfc06674ff85eb72ff7bf8d3db319137b9887f3943b9e96a72083a415a08e47d9443a2543ae4a94dbbb3e3a3060e3ec3c7e400caa809b1c3484029bac98bee7500f0500a51240aa035c6e0db8d3dc4d6cd94cbc639bca507f9682238978381db59&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324379707Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0, Name: checkpoints/3648/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3648/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=4adc1749652c2adc3a1dfbbbfb1d18f04cd94ae4213950549ba07a0bbdeb97125ddf3bbd255595d8fe0de493ada11f33df3e636baa9ec2ea3478864f070cb7b3e4cda54873e835a51893da25dccdd075276841b40f7ccf9e6fd9bd78e3212823b93588e0683d05ece25b36b97e624e87570717221721e6dab0eb3842f86a9a3b7d8ba0cef199444d42d19ca86f9f692223d6048c6b1626e740290969a4f87bdf92b5a989905dbdce410a1f0e7e160d652b3f24cdae2c6f836cf7bec2cf8cf056a5f35fc5503dc845e4d34e82aabf793063b526c97935d89cea2cb1dd87142c0254583b05f2d233cde11f28d5f7d4fb209cd0433c474d9ac2f80a383af3f208d0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324421981Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d, Name: checkpoints/1344/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1344/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=46b49c87850a6aa6c99b346ee00d5524635b37f4dc2bb0d571b1967dcfd816f8c2c4ee81ed014f4b0911c8fbdfb92b9c2b50317028d76c24d8e085132ae7fa7a03ae06d84d4b13f389560796712bd147c5bedf102acde0272d8e7fd7b894ac2c8b183665e12dd85ddd0764063d8cf1cf42816d014efe73b2d7b78cced8e47ef2e5d01c1500994d315b8569e8da4c45097d2a918aa366817c1d499ca7506b08a3da3ff35e8bc18586c398adfe3b224446e9969fb6fce6188ecc7a96b72fc2f99f47711ba7f01c691da8ab99e5f05ebd4dc84e5bf05f01b729c2f61397b3085c1f5cf9f8fa1145bdf6a71abd78f21bc8f3276d2b4ce7161dfcc653b5ef4f4f6dc7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324461589Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d, Name: checkpoints/4032/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4032/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=31675fa32962ee50e2751d04f37287d7a7e423e38ec612049ac2433ec79c3f129cd5c6a5516d5a3dc5e2a3fb06e548b0f16b4954ec223c079b87ad05c9cea3ae5d6f20e01505ce2b9f423e6a66d4796abe96261277c9e47833426437b66ca5d452102fe32f633de16dd9f9ad6fc6cdb36157694d28f815edec3245c802cf72570d292939e1719ef2f4cb7fd2c542ee29a6fb2dd3c79f815428babbd052d6cca2d90c9bf7c77359fabe14cc72846fc02bac6751a38a017c15c3b387cfab715a8656d967a47d63527e72728aa428859b1543c528a8ccdeba879b46837c84ce7c93caff01ff658ff2424e899c819de2851f61974f46ba070188dfafee005dfad86f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324456489Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d, Name: checkpoints/1664/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1664/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=5b09d502f42705c1bf6172d5826845c680f3bf9de852c37a00de037e8f05b806666d17bd9736babe4d1a1f59f4de78a2c2160ae63bedb60baa0118510ababff0d3a3f60a067aa717393913ec8295ae135147bae925fa994b6eb09579448b1a2556a88b7793939c9503529a529ce21c0f80bf7ff38bcc3a334eb542f843df6698bc499ed3641053fd957a46fdbe24145c81c699995e7e31c449f478ad9f112c8a7742163f5c8cf6ce1d1f2b370b186640e070057c51f91e4321477d4e3cb282c2b2dc60eb34f22b67f0510da31237c5cf16b69e7cf09769f5a265778af94f865b99d6d06d00d19551a45e8b865dea1087cbf5754009cb42b58fa0d0de37b738de&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324509353Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d, Name: checkpoints/768/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/768/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=c2a1d59ec3d50ef70c6b7885f23a2fa34fbc79203ead961afe519fda6fa1d014b52e3f476a533a3a6828438060668f6c2a4048902ee1736b9391bd8bbf49b08f55d3fa6b3917785c050e0f4aeb11bfa848ff37acbfdfda8c9288878bf1480eb0fbe42f3516c063110efcbd9ac7fba102a73cd392d89a314a80ff291cb7d08a2635dd2886fa50f111cbe965e7487255cc007c8bdd7d5c1ef97e9d48166f4ceacda10ac3e8bcef78cb3bc4d136afef9f2664a6c4b43afd1074d0f543d3b9e86f0fd4c129df541b4f6799a723e9bf3a7b3b9820ba920ccd9cf6b37f808552ba3c0a7228324909e9071bc4df8d0066ac2913ca9dfc2605b9ee83cb460a1f2759abcd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324535769Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default, Name: checkpoints/1344/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1344/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=4be03ad1fc84371a4fd854c1b70388b54d36eb73c870e5c826dee9316067c7a90dd7786c3da8ecaef804c669a44f0d062b9388ff02ed13c7c8f4bf557d1b073dc9a95a405ad2e95fc227731e860a27b551acc15ff5fe65fc420ce396dbe907b85299e88908e9e06a9aed92ac00db7645342a468309607fa0228bedfb10855ede69ba70db8f15d06c8591f557fa4a4d84dec46a43c015a9c0c6575189ade16e047e3a14f45fc2d551b7d41e479667770ece38686f55da3eca2d9c9a8b951e73e96c7b93a55cb67ae2fe604968160708f8a44f990cbdd5741eadfdefe621c1b1857ca98db120acc66d1fa0fea146d3f733a9226f1741c84f2c6361ae27469e04a4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324547567Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/d, Name: checkpoints/2880/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2880/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a38e251e5366238a1645dd53849b224f939f373403227c440ed90140f1fab3224b9595f6939c5cc19d1cd359c846703ff8d274231acb598be753bf420b8025eb195b5bec464b5a18f982542485a7c2893244d646d2a9360c8c19dc6088edbda2cfacbff810f423a6d2f549c9846fa68ff0ada420c3e662bd245027b4be393b8ab92b0b588ae9fb13211571db83b37f42a7a8e12f14c6a07b2661a55677452ce0e925b14b8b2adcb58493ec092003d97b4c22f8d27eeb6ce95ec1ed02c2e294c63adf8107a4a55c6599fb8d8beac6536d43943795d2fe515e91996f08943d46401715b3ef645809853b03645b201ee73ed185d94dae058a9cc3ea39a9a4fefd7a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324578163Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056, Name: checkpoints/5056, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5056?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=c106c62eadd6762aed66eb2f8470a4b6c696d0ef1db9a23d8b32d9aeec975a0fd5ed8c28f40cbf1a364f8b748682584957e173de0a3c1dbecedd5f5f6b931738f46e0af6004a215fd036642ddb7d97e0f301a33e7e877a805aa45716da3d62285f96ba6bb4f42634dbe10b567ece7b82eb193474ede461820cd76bf13c4632ec4445aef627e4e343969be175052e5f611962e3981a7f19f58ce5e9aab29d2fc2bc3eb47b59f7f8b2e1a23da740ef61f7229b4b580a88ddc3371cb4a23a1d46b838a57fe964c456bbc8e575b373086876298ecc62114a05fbead40d8ef28f326234e284fee85676fdef9098c7fcda58916e4176e914b055f12b35376ed0cc39db&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324605571Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default, Name: checkpoints/832/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/832/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=66423ee23bba84fd1a9cbd05d8459623ffb7c72292775ae271f7dd71bd7d2c4cecf144c6ef882bad0638afb5a8ea6cc45b034b495fd48346e65f11f2e7d51ee99de41a45e9243c181d8e5953091a5b6c70ff96a06830d3fc03d44b07a6c88206ef7307b3668070952f5b4dfa833696bcd8e1624d2a1baf939e7d738c415fbe61aea9ccc58aaa3c2d83e191bdf27aca625326c24ee8fb6ec6f0ded2c25dc953d5c85ac6bd4cea962fd36ef811e0e6d543c6ebc60ad42d6e34098694d2c9058c52500f5989809a0449514a9bc573ca27278abd84a69ea452ecd2718ee2db0026191480bb59f3c28d1df46e0e3c5d00d42ccb1ce90dc35c43b68869ad98c4271e94&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.3245396Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/array_metadatas, Name: checkpoints/4352/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4352/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=5f4303a709c29cb153e07883360d689773907df4dbf7085c9a817e6d6c80b374f79533e88f7e011478d41ff0f8492bf73d43849603fcff77ca2116e7ad4b9c0c31a23dfeffb12e5d2ab6a7e23ed6bcacd1f02713ff6b34e3d8c3a2bc3a21dbc58f6103a553f3d2df704ffbfb4128871dc283cde7af66f32065a71aa7bea4245269fd7f7c6338e6be362c0c2f0e0458ed9448dd0aa145af4963fda5c334f54784fd9f396e5d307b6b3c6efd6758b9453d9f48d2bd143db8bcfef97aa5280bd1bb1b3e9170b1ad3ccf6ab23f10d86f9d3fddd0d002374bf50e018d771e2ef6caa4edd9b3a074c7193ff5abc92cc45097bdc8dee5c14fdb160f700f1a4b4cb65b71&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324685687Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/array_metadatas, Name: checkpoints/1152/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1152/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=22dd5591b2dc231e20704d29a678f4f958896f3ae57ecc42c69630668e3b33d9e947713673a0954921b6fec6cabfc4274d9cfd495fa176b10bb901a0e0bdcacbead9df3841dc4855775bbeffaa25315bf6cf1ad247b130018d4686601184c17a38fd826f7349bf8d3718e1fe87841530e1fa12928b399ff64294d0292baa7fcecec52799ae91f4e534cafa481ff044cf9f72774f62ef11f75fdf4a13ae4d7f6a79b6e99c50d596695e3a28abd4bffcd8de419de3829adf7043e8a9dc3d548fe63f3c1bf8778a2c41cc171a011f9aa94842d1c1c72a662f8f6b011d3dd53d909fe37e392c31358f051316bf3d1f4fde65bb9a192d47beb6e811836554f9c2e486&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324713176Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632, Name: checkpoints/5632, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5632?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1767c92904fab52f3b28815c3a7365906e29eb51610563fe55987a9864b552e826478c55b51f15b74103215e9b2ac9f0b13dbc63ff19a4cf5940a3f8362c8792743b741376db2933578e5ab79f45989ed91400eb0684e4f9ffa26ddf82e53addc35ac06cdec18c57a7da6b69cacd09227615c44dd7193baac03a7999384de7dfbb7dd14fcb174bb01eaf1dd962741a738cdc8e8c379aad6ad0977228e93decf3e35d87d4cc580b2ed42e38d64379086c03ec0f3553e3b94782ada4b0e361de26260464c12ee00c70c03e2e039637f89a21535230196e475928b1a2f0967096de98d3ec6651eab634c18479501dedc863682b2b05fabb2f4034ce948f17c8b7b8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324729062Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/d, Name: checkpoints/1344/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1344/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=7f0b0eb792e720344dff5eaab69b018e27ddf7de53fe345494d77444a074a9fa1ef06f6037bc3f5bafeb97e8051590a923612392772970b186fdabfae917f96372b48af7677e5daaeedaeb3d1af75c01c5f0d3255a2b640d11a3b0bb0b323c4582d12826039905d2daf2a87f2d1ecb55d309383f5220e20afa7ee6939e01126d37a8e6743158b4e834591f7f5f00aa4b752219e55377e31e11a1c13f63e0b97675f6a2d947c43b8b1c8a65eff24b32287e953808ee9ad1ee5469784069f9c31adf57dd82f194b518a07a8f57c16243d752eef26d82d9440a80710bd7d9f7833b04998b423afa51c41b06407144334abc78f942cc122118ca4864900b417edbce&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324754362Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d, Name: checkpoints/704/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/704/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8fbfecd57a34ed4c5df9b87906a207163e1b94557cee47beedd07650852f7a492502f14a105bc2ab6aca4107ef5d660dfb559b97f7cd44c621b6763bcddb1adf8e3288c7771a0e4c5e8e6bf573fa755a83a6d973f43447ddb0009067bea9aff36bf04ad012f74c5bdb43695c800a567c6a1d0eb7abf15b06cf06df434e2953ecf1c52b38a90053c11cf738209a19bd0625200d2c2b42fd072d04aee555f0fb1e3d4fbefcc8112d797ff44c5c3969be4c8b92c15c9a3753b8a17ab70f3de3a2001c1af741f40a7614890d1eb6d9eacc0044dd3a605559486b836b129f78c8fc28edb3f815f89c9d28d898aeca1f9322e0e0faacaeaaebe62ce5fb06c8a9cbfb6f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324771078Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/array_metadatas, Name: checkpoints/1920/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1920/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=115c8c3365563eab62e5bc1468a2136c915522949cd1543498fb3286d8da09b51daa2882496541c98a13acab5ba2aa23878adb6de653f031db55b2cd2287331b45d419bfa7d510f61ea55bb0eed32492da4d8ecfc353fb4306383ca59ce72ba982cb085396d96d1ae2236aefb69160fcec0058e1c96f1e3e15171fcad549606d51941f5f2ee2894df61f7130fcfc14daa4c172f234ef7c93aaffa32e45eac1ac5ff7691e23a6b81c39c14a1d5ff4b8ead0f0b9eb618abcdef77d3cfda2c5435d16b20275bbd67cfbd56b5c38e1691898e34742059cf8b3607c7c0ce22dfb9f1fa5217efb7127cb98f99ea7a15e7687804a2b5b5fb958f840c629a234c786288d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324781764Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/d, Name: checkpoints/4480/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4480/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=06910adc3114ff63de46898b306f9315def7954436759f392674b7f8b8409348450dd44f42df5d168af53573cf5a7c9124c4d9a81a3a10695422dac305bdeefc2bb54dcbe49359d4325353f696dd2bb8b892255d4febb736a8e0579141b610408b277851066f9701e69bf223f74d629dc4881315d22a590411a0734e3ee23413e6f21eefa5f3446c14a7bdbee0b113a4a78d708df19c63d002a2ccfd215ed60813f455a62c19b833c519981123e080df57c9c5697bf1a22a1a42bf433dd985bce42c382235ed89ec72304a4f1020eb6d777fdb716e58a095d179eb2c630be168248a688858d6c618122033c0475f9aa9e28827c451147008fedd5fb18f824281&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324805563Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312, Name: checkpoints/5312, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5312?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=077e17d3a94e6551d2d8b0cc353334ed37fbe4727e7d35843b3a833d17fc4ab18f61f81d089edcf9546f6b150b2513b12a5115c4eeb487a6ff9051971d70db126051d0ae3579710a1188ae5116daf72cb39a82182c33ebaca61ae102ddc6b7f142fe966b892de87fa930b7dfbe9bd8a9fa6a99c4ac7f100c9bdee742a018411253cf047f13dfeb06972fdbcc16060daa0c69cef5180fb9320242a1312c4e91cee220ffa6da3a60d1a07c767c9c17f72222a91c2d76b7c218e6867a992a2109392897e7cf3f43e88a1ad1e2d2becd7228111b82a19f3641c057b2d64db8b084ff3a41fb5722a3b51b4338577ddb4ea502fc6535da6f12618842317459c2186730&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32482259Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/array_metadatas, Name: checkpoints/2048/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2048/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=6a014bca865e49978068f80a63273c8de5927e9fdf09505bcd8d1c99d5cd38c6b373b702cc274dcc42006a4cb6785872a13d9b761f72bf3a2ea831b27821b527d3d8d3330be63528fdb9cb591281b2f32d1f947045a7f417c22282ed0d217ec137b8a2932fe5e38807404ba4b7ca14649a0d05589e735c183e61ce718b8f0826d3d9db33e958e9da7c7655c4389606e13c3a2314ce087ba71044bac39390aa466e46013d005cb9c617531299ca2671b5349083059dd5014e40f333afed86b021309f3cd933b248f51293aa9454f400f9bbb7bcec5f9ed1504d2a33ed4ad00733e051f13e1265c10330d9398dc51895a047b9dca6fe939196aaf69a14cfde7908&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32484109Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default, Name: checkpoints/2944/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2944/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2cf1f957d9671d6a32be979f16fc893b8f4db6b841894d3cc1c64123ff7ecf5040d95d85cd9159ec8e8c3da114f890d9fff0339cf8cbbca5f89305c907056517c20ec939d369d7596ae5fdc73543c26601c2a07e96f7f3dd24e2f4b94397234d391e1d77af507e126b36ffba8062c0575d2a66517c2a7042a48ee6ae9f642f84655fdabdd4011abb8db031e0226f26f077c93105ae1c6490c0f22e141223aa8cf1d8094082226cef5991028e34486a173d3b8566c1e0ea48b10b557e56db2ca5664fd071fa159df02418949373c650972b28b318075346c35682c6ef9287338b58f0b2a624115f22e50786272f6140f2d9e9de2acc0e0679ad6ad7849d35756b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32471377Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088, Name: checkpoints/1088, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1088?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=bda2466021ebc78312916c08958dae2da9c9ec5ea5895368653e4c874472c49c9424c53dfe3e57e70132f4b7f12feec7dc460c69b34efb049c3670158b0f9dc78281bac58d783a700a4cfb5875e7672e9335c5a2666f4b2524d1ee159e4aeb15e1578f3df8395c9ad05ba7facd3ac002173312baea16c7479f24ba9b422b389675d4cef796a60dc37ab2ea0090cd1f4dc277de04a7c5173d6b9a2fe6f5c961ad2650028fb53caa508366162d698ca541474ec376b94371775de324d28f05930aefbab39e5617632a026932a09833a6b4c125b8b91b1c6ff5208443918aa983f33cfa3701a265fe7abaca7c2aa065e7de2207ba0c57ebb86e5f3f144410fb35b2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324889936Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/d, Name: checkpoints/2176/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2176/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=613437a36e22c1d64e36e39dc6927a3f55e62f5955cf9105d33907fe77bb18912595635de6c2eb241ba5d6b888c62ef10d904803bd9137a32d5e9498d31bad24650d983aa3339def59a42165bff9a8710c820dcf1e4e8efc9ddf255b2394a02e032eaa18d3024ffd3e284c8caae0d400b276ea4b1d8b61d4463cd175466c6d431b53b8783db32402bef39388c3fb20afbfdb8fe2a75c6c7cb37ff87914fdd1d44d0e4dcc5ab2ba3b38ddc5e41ee21b19b2e1ff30d641d3737e1ccb75d22dc5e6dae1435a69d8de1dddd7996494e18293a619d70da077d9c1533d28611692f3bc0d994f191efc1f1480f687f5ccd871e120c4edec100c031333dc559b3507fa38&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324195991Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216, Name: checkpoints/1216, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1216?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1e75d95dcf8acfec69dc9e8febd9a91ca61c59f0e63bd79cde6f0bc3507e6d469e4c01b4494c2e3a958892f9437a6eaebbc2a9b13bb8ee02cdcccad7c35d7418ced09d28da23baebd8ab1f55bed5911ab7aef22f262df87f8776cc2fa8aadd55f8587c7e84d3df89db63263e7becc29dfb8666e01ec973494f819f1b1df3e100ca630c55428e7b210e60eeee78f86ba8a9e825eacfbca81b39b20497cb564a0b8602e3a88be577411696ef7c8b7f55c7e6ebbaf20cf8f621412e41c5b42afe5527276f475cbe245fec8192c69e615031a6154c69262b53b938b720fb15c57022b118f721d90931a5dd85cd9a2d55a1bb5d14031cb3dbcc7bbf6cf8549fd47c52&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324964111Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0, Name: checkpoints/1920/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1920/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8957a8c4eacfc1eddabbaa80f5c6f8d09a4c05cdecf303c3a98247c011476252b312b4cf0ea77595a750414a4b1bb516fbe9587be94cd666c1519d1457e9132e225e841027d021aad24f15f97279c8f43f05db53c60823b1c10ddaacc65b3e99e734cbbe3e44c49cc27e107374ac95ddd80e2f60f623ef2fe8cc254a53266a86d1abc41386e19dbad11ac431c6adb9d27949dd6b5b320da87f80e96d0f459f5c7627b3cba680a1af3308266fcab8d6de5053e21ef1f6283806a79e0cec281e6d17fbde9548f5167a09892b3483f74486d655d9defb52aca3c9151e6c3b1264f56bae623ffb16344893ef7b9aea7199bfdde3580b4d888dc11299c7f166614b44&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324968651Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704, Name: checkpoints/704, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/704?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=784a35c84ef2861a0261605b71a0149c00e053c697b36c278688a50f540007733cc385d166a6148de68c4a01228884189ffa6710b74e0cf5446a7d4c16589d072babac586f86896ad0164dc893e8d1271072d3c40caab1cc6d2f3929ad97f2d5cb23b680034c435e5ce80ecc60319a3263e84c4b182f8769fc27976c64557dadfca625abd413617d7b69f434f23046bce9b0d0d840b0b09faa3b140a4b37bbba16d554471fa490417924bda91d3ba8d1429c89a485e248e3d47a0c93ec32de1b1a1e633cd8fba2e919db5938a276f24b94e5c245e684b08d2e4339d0f6d799c5205ac5fdaf628f1ab48c75a2bc772ca910254d2e9e0d510d68cc755ee5e5a1b1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324987746Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944, Name: checkpoints/2944, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2944?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=494287ede53b9fad8e4af87ea920ce5145c3e140816c655047e961900f39c369114b8d5fb0c8ff34c2faa94a102a2bc7fdfe4822d3b991496db01a29fae941c5651f3da78d01fcbe56a9e6975e88e514ab81bb3bf775decb813be77d06713e0780c217872e3abb2afcabe0373ca91f5f0ebc8aa4e0d5e4eeeb93e58e2cd33eefa9d6f217ed2baef35f092ebdc9289439d08939c846d8467b99111e72120ef6d202097394130cde009c42f668afa57fd1d2869cf70b4cda1ae8447a10106e54ad682355a9b8560c5cbad2fbd68abe6e58db680d8cf0453c06079677e9204582824a74013a284c2f1ec300fd63ca709facdda692b2ddfa8cefd44d91fda321cbf9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325043789Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default, Name: checkpoints/5696/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5696/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=aad10d8d4ffc299dbf34020780e30275185a0e35c02fc4ef370e29d83d0ff6a8b0ccd7012f77b858f3205c352dca23593596c4a2ea45cf8536c9f99e7924a1263be3df813f202214e8a670e2c27bc4715ad776767aec8429519ffc96abf783b99dfc90893cc4e9d67bacc0d7189d3063111a4fd27c7c872a749d19d3efc9e33b6a9b17c2d25f25565231c0b655601673b117d9492397e1ec5c4c1486a5552d572551fae9364b09650f2bdc72a7187f99ed14e9642ce0dd9ff6f5506bfc54b517757120ef879d6704991dc377434d969b8e638bf55a65b31a96377332cf1e599f04012b55b5ac04ddb72fc6c1ed20d19876ed89395e4653984e929b39767829ab&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325096866Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184, Name: checkpoints/5184, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5184?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=952925e4e89d4165ee4b0e6778cab16d70f703175c9de6b1b9a61cc195dc0ee946097b1842a798dee3d9f43a1c9cc0708bbeb000760550e1807aef8c2b7fd2a9dd3579c499b2320b8c3037c64619934177dbf0a0ff4e9694b66d0be8f7bf866896d3f0fd5b090564d2afa3b446dc3e820ca5c7acca3b0544bf2384b88c3d188661d840b21bc8d6fe7764d361a9b386d1bcf215fe7b0e7ade05ad8a53eaba6a2d2554adaa48fb3170020e56a8a279a51b17661c9d09d76b0d173db182060ceaaac91ec36af09bf780590e8c0d65c6f59e6c4a5d2804bee8a9cdf1098ecf02aeb2c2f1cbbeb09f71a222a05186daec340d3af43c54213b0cf26e956219c98b53e1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325112869Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/array_metadatas, Name: checkpoints/5440/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5440/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=bbd56205d4bb260ed2fffe8908cc8c211c22f94b6c2e15066e9290fefad796677fe697e76ab3d082bb094a718163619ce62ef0e7e5c034db4c3151afbae6c1a0a30bc1cfce51c246a407a7194575b1c8793e4584e12afe0dadd59f5cd7ad0cf4c8843e887dbf6479e92366ad1c621c06747dfe193f9743d1e6bb427932be0bbf63bcaad1a7686712ccd5444837af7dd6858fff45095f95f9db9cc2f8dffcc08a3096e7db810953d01296beab6e419a2ca5d86f8ae6cb6d43031dcb7abce3343dfa422716a33d803f22b8871dd00c1590a01bc2e5492d3dddf61b495e50964fb004b8a56d4e201462cfdb20284524e1a2ddf79a32647e68a7677f087795459771&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325112992Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/array_metadatas, Name: checkpoints/0/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/0/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=79c0a9e8fad336f1aef692aa5f63c469cb120228d410e7cb944668105066a78842793657bf038f48cf76cbf729881038de2edd6e5ff41b748a6ab81763d5d6e1d5a3c505633a1b885c1a6731babca528a20799e6e41230097a739b59479b2ccfa0131326c743db996c41113d70c52e267ade1c792e55ec5ffdd03961b6568eb81f8e55951c5918fc118132fd50ce95f5e0b6b18c10c61ad43801c67b440d559e9b788b0c2c245b84af37abfdea1bbc13afcdd6d8d396260e0823e9fef8abf2665cede0d9f1dde328ab5cc3477dcc6d184cdc062ecb20e5defabaa0d926e0fb26a4482fb3cb17a70e4d0cf337accf96e67406737ba153f70917a6ce7a56a674e2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325172927Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default, Name: checkpoints/448/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/448/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b165219ed60f2c920d5c6a78e434611373914a34f3d4563afc3b1356b9215296904d05d7cadcbff008c0b53ba3d2c7283c77f200737335f5336a19a84c43e65a9e05c58decbe06c72c52e588bbf984353655868af830756839a087024ca2778b1cc2f5085b65315ac30645bcf91feaf06a86ae0545f435b23b29604ce3417c1124f901542748bf413bb5b4a51f06198d0e97786cf40754f04f74a4c4ad8cab489bc5d864f00d7a46b357fa61b92d95950772d80eb4ea71eb71786eff74cf06d322ef27b98fe6db2db704efe9e4cda7ef2b77695d5f7335b6445e851254110dc8f9bf417cab815efa490d4e88c93bbf46b0b85664b3810ae9b17317daa152024a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325186125Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/d, Name: checkpoints/896/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/896/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=4a3b6e16b26c9d6cd38310f2abfaeafc2934789bbad152a926dcb83189cde5a059096f86dd1e605326e11c65733de4cb6632c6304e0a2033f658f0c46e5f5ee035f889256c7308c1730a9e9e097450a57f4f881a5705a05255b883f715de9e402290970b32706430c2ff51cfd8124789f437977f43809900c3d1cdb4a2f12777d6e6bc9af1778b0c85139160431ddbd2b6b716ea73eb274ba8b65d37a9fc474249125c6032f3ce7d3a78eaaa9be821c7dec776550cb923201bb4a2a418f70397732f81fba1183ba9a8cdea0b2e700b15178098b50d1e43e0362ae3e43d26e0cd74af524aad985ebc9fb78fedd06b0986824deb54e8fdc6238c32af8b47512ccd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325099102Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320, Name: checkpoints/320, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/320?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=01234a81f35087ef2965a46b43a0572ebd159d971602875073f6e657442e37781eb72ee8192f70fc3de45dfec2942066828c8e9dde189dff6f722d783f9511e7cfdd4372368f76f5074221822f3f84b089943ba4dcfd2428ee4a597889de4398be8a843126c717efd56e98891ba9d7f004e102de0678494a6c05307528cdbd235de3fb4d7175ab8a45a7390cc6d613564220066c2795cfde9e86a3f039bdc7ed33e4e3ebb8ec0ab105369b41943f10123208132ae77419c393cef5b86d1f8b03e3ed904e7802538a1e536a5f0974aac7e56a47fb0f9a889ea0fc0b19b58c2e2d5de20d24553c0ad1a1adca7b8115b71d5e1ec322342992343bda5c613e102a28&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325232237Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0, Name: checkpoints/3520/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3520/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8a2f6b733d6f0d3fd9ee081e0e380e80da7d382c7c1b55a5ac2b000688809e330c1c34aeb3be08e41492aebaa347d90b2896e2af18f046897cd72bfb2da677a5f1a34c52ffc7b977aed8bd09872d3aaaa15b3d43d7b3844f80440278411592aa5d756d1a7e3019589e6f5f7b689e7de9516d520a70eb6e100ca60f9bf449cdf92b2dad552d5089647b89cb2b8cc38ecc296c790f235564796fbf0d61ace7a9cfc8bc7014fa86760d7fdac4ed3906e10fd458605c803939ff636baba155c146187961bbbe49c0474ecbbbb3a6cdcced84bdc9a37459746485c80290584357ab991fffe6caa85f1dd99f9f8daaef7ec935b5919e2d791fdfc8cbc5070a7ee44e50&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32526161Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0, Name: checkpoints/1984/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1984/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=7567bb48a3ffff3ead3b693e83bf2584c14b1c88584ca5af01e63a62c7db15a1cb38cc9739274213e18e4cb3b7aa2eb470322e656b84d1a20f840ba547ecb33f1c40b934d667f6d583c27e5a74a4ba3363b21ee300ab968ac170e18f9df4e2d604081dae5cf727b7fac97e4fc8ed3d25d20adf92020576f74aee07f0e90fa2a7cd7e14afd2b451d66fcef67443e3c43e8381961aac0ea94acd403b0385bfc0d1ae2c60be16d8553dcd58f5b36b0c8fe083bfa5966c64ecb83ba4e8c91513c4147572bc5754ae027aa364f6fef214f10975b6dec6a5070ad536dbd4f5cdb61d1ccfda4bc6f73b91ed17f08937c5faf57270be23c45953be51e74d7058ea31c136&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325281358Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d, Name: checkpoints/5888/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5888/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=0e913367330d1934a0cf15f26ce0618b7cc8da9f1a855d3668987c9c77eba7770b5e00fabec12ebcff8ab9985ed5315524bc0e8b8d96b5b0ad05cb3e865a596b5765b5b75879f8e8b404afe65619ef9267f90ec50e15ad6454972a8a726d9e0bda4287d2134a9317dfb6af3e088a7da8d221a0a2187f02aa71fb0a29e3be53767825319e87a9fbaf1cf7f73385b6d62c7e2570c4cd434e0ac8ffbfcf2cd8cdc89d0129564b0e03a8ee7b27da40b14d38524e40e3548e5cfa38ef13d349d8f1498b494a0ecbc7275e30dd1d1c04050904e5d07a89aff5a02102d05ad3a2ad3d50dd629da273152e372341b70d48b9a1686c01e55ae11ea64128e8a306a0e644cf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325290047Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default, Name: checkpoints/1920/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1920/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=757301e7880b26dee9c393a3f74da514b55ae34be34912b36f565c2e8fa5730a445b7c2b291708dde86393e6233c27c1439e773edd706d0fcc6afd864c3a57787173dfbe6ded8e81b7f0968d38b265715a1f4a8bf5be04abd561ab553723315658964cfdd12b5f356670e1df3f26f14ad7c25f7ef7df96417331e68c9e834e8d2ea17ecb55f3e2a4674def9ee272760882f964ff08a6e2753cee5c6598072999a7c9d25f44584e2da93376591d5dbd70cd55653e67909303c44c1265261f311de16a39637b460f7700500dafd18fbca5dd6fee9042b5200cb82c00110d489bf8b322f5d2bd479e944d95e2081d6bb0113a504747f2df3b6ab24333aeabb5eee8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325323555Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default, Name: checkpoints/3456/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3456/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=c038470444ae1748d71a9a58a9cc6835659cc688cd3fd8c7554c26dc8c0404619767ed2a886f8248643c6927a0fafda978168fecfc93e6ff1d769c967fb5126dba09a6f05b4581acd5caee5be64c41cbab3ac512cdf346f8d65f498f45c9f2590f901bb46dc8a9a8c47b7785ea84af2dfc86f8ed6b40681459323767e454ccf91740fe4ddbb4e6f911cab52d90edeef4e165d2d823aa0851e633ad1e212ced138fb7a5edf52e211a71eabef5171b4d4566b53620796e4ef9d38fce966fbbe60db038e217a62f37f93164af308849d464e45dd0ff30d12c9c8de6f11cf4d1148c05576a333cd42847dd46996428d334f98ae9284390049e7cbcd614d7dd3eae2f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325353914Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d, Name: checkpoints/576/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/576/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1a75924db91cb44a47b7a9286891273ef3aea1d659f3bea2d56879156e171ba7e6b937166b3f34b0a4cf38521082bf2e9b189bf3aba6b5dcfd2075c170759db5bac3cbc774ac313708ab70d23c5d21fa0e76b37a0baa12f458cae2f1f791ae4be4e7ad81f88f35a617fa4ece5cf391908c72c8970b2be6fb273a43ca44749f0ddc2ff3f15740f83fbdb24f4e11948262c3ed04764f15aa3cd0a2225d944ccca24c3f6c3188ea8bf4ecd451d6656a46df095056acf1066f80bd686b3822bf42ddf559304f37fb45314c21b36cf9fdedcf8a221ca7c589a0c8cb00f018482f179f9109f1719149708ecc53f26e37665486e7e08ed4230c9257ec584ffcc5e5564d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325299382Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/d, Name: checkpoints/704/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/704/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=7b27fb273f074ec0fbe74a9300afe160d553d3a343ce3c052668baec4157539f1c5114a3489a71902ec0c11ea415bfc5f9895072202400af398a823f0275a46440c161f7ce51e7cff6d6bfed0b14feaf22be6d6ec5809218fe6d6296fd79f84e9f6a5ed7254fd8c2369857ee34a2081e9db8c55e1f08a12ce9776b2c3a8dddb3867556b1fd10d3d19a8968d543c569a7cdd518fabb0598fd7689dd525445171e7f07f71c270d7c117f5ec422222429fc2f9e36d7dd33a18c3f56c4d40e568ef579b604daf876aafbfd868c70c58a7c228bd14d317a585f97c23565ec433d90cbf1e4d60439c917e38a9132be101d7790b5db116f68101a07f78f31f0c7a7bf90&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324491051Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default, Name: checkpoints/2624/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2624/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=64f44a9da1fd1a5bec49bb3b207d6e9a2bc45a99ff1d662141ef9b00d965653158fec8228444bd5b57908630590f3c852ed49ac8bc82bb2a7a9d5cdd3d5316ea8d7172d82b8463f004aeb9eb370fd69db250db0ee1a237d97a0c4da4ee057423a6171b5e9d5db8cf097675078093252f40efe69ee2ece392323bd67ef3edee1471c6b770ce934c91e820a834bdf5cdbef6c511da103d7438a30e816e4aee8dc33d6b8a22a194d903fb75842ce0240065a5478fdc0058073684d084c5009486dabd299905f32faedd2967bb908f089e2a1c8be27492e1def6b1be2aea590aedbe77f576be78eb66f15bb3e87ef555ebbf31a8b727d9c12a6767cf9dc47f525315&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325409843Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880, Name: checkpoints/2880, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2880?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=be29be10cb8e661eeea6e07678ea646044dde87b6027af15ba224f15a740b826a220cdfab0a3cffa1b857e35f7853c51522ace6c000246ba00ed08162207613b36281a115f0db9c6b1bbeedf54a228c579d700e86d8bd86ed7303a309e22a3521e96a8525da5293e3c69ac5c0134d4fbaf603c4dcc9508a32d42681fdea7bab5840accc4704c60742ff650714b61cc8a44e6e2600d159954cb08ba409cd4e41925ee4891fee90318e364481c75bf148844e5fe06d54c26cafff67d27e6e9a7f1be706c8c2489a7b84dc23e8d4bd96de443e787df97a42baac0db61ac66f9c1e802e416407dbd14872838cc3ff7d0fede83db13fe549f44599ab09ead6d5c0cb5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325434048Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0, Name: checkpoints/5504/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5504/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=6b618042d48657943f87e56f0635d6206b69068cda32502be35008fb178f44787ea89f118632c86bbea7dfd4498ff31c92198ab4c61fb3cd219b1a724d56920c642b4aa1938781e756638e7f2afd2e13e8700be99a24d47af22c6183d758022a697ff0db2a72deeccf5dad041bb215113ae58f95d2231a0f071e735414b6228e403a1768a926dc6f832943e884868099530f348c34afc6f0ffa2afa0ea3c581104d91ad56394014a65a9ca19092c95e16e4eda15716895ddc4b11768a82660a25b64c75d206aaf71534f5ee1beb933fd2ce3f3f9ac3f6defe4f76d43f002907723bf9b100c5cf47aeac7c0a809213e96b7404b87662b583ce831799db340b689&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325349621Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d, Name: checkpoints/2752/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2752/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2cadffdeee8365152e0f57c0d52f1605dd6e2df7ae74fa482fdca822aa30b511dae796934e3243243fa239d72d0a9279f6434ce7d527feb7093cf485da5b4fc755376c26aaafaba139d7b7e6246b575595b6a7f40c868ae2a4120509fcbc4effa465160585af712035b5fcac3f2e2e033a67563c99e5dc531f466f9f3cbf5cc160d758a967f36203bc555db0d4b3159b3ea94918846d3931ed1aebe753234bce4c34ef55afa583c89b38b3f5e122bf1d779c32f35cd06f451af8a159c5c4f893dc07fb4fc84bae5a97f4d754b9af58993ca505439ee954aec92a11d848f87c4f51816eb78c8cf7429b4f72705ee24bf5e19cdb8e4947b2db869c0a785fb5260a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325485545Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0, Name: checkpoints/4736/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4736/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=7d4c4db5921beb211e2d3237694076fff05ac1fd8ec3e9afbd2cae663353a9377ae52c9f01eb02be2d8c02e1cdd035df3ed5a6ad1a526812caa2236a92bee80479ea6edb6d0b38512a0660c129c78fcd5f32c516a0564da3ed69865c3c4cf41b066b4af6faafcee5ae49ffcaff28b89818b50df3df7b670432e74c263ddc3f1c7665c93c203e974d956e9de477709ee0468b60c88b12d4531b0c3ead04dae113d0866679a11bfeecd447e5c13f50c5b000d30b1bce136affede96987dd6a595020dd1d1c98e6ddd9d5e19bbe2bbfe3916e50a5063632fd4545c64135702ded9cdc1bd395088d503ade345f32ceb3ff8cbc7bc61428119b1e0f82ca3581d79248&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325531173Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0, Name: checkpoints/4864/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4864/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=184349478392ce9926ced1c8f3425dcc60874cbe8a923590fa223a3cfffc4704c2fd6d801b624f40062937baaa3749e8cfe2da841ad35b8f94a36aeca06153058054f1f0d94acc71e6d55f56a0c44a6aab1acb680a89a91277dc6471b3a62204267437c8e9ec31419f5ae0b1913b535c2f999a8a8b00dfc1a906c4162ec19471dabeb2777e18a2c20d39c621fe822115f3163af29c510be303ef667541c51badce67dd2143f6262e0a430d24c4aca41d13d4356dbc473859cacc31b9a33ace15b48e6be285076f806d820f056b64942148f114bb22fbb036d357833fdcc07cabf4c947309f4c8b924d7951931628ac8dac47b0a261f1006a1be6b5e031049d87&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325550965Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/array_metadatas, Name: checkpoints/3072/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3072/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=83c61918dfc2228b953c530ead263feee9c5acd1b7a24f8d6f88ca0e95a8a48bbda5b9a6c5c9273b0b36999199e7f7eb48a1e1a89e58c9369e3ae249ad6c593d676e0808e4c765ff8527b3611a56c0a55858ce7129da893145415c3e9773eb6f2ceda47dcdaf3d8e3a303b69851c9137c7bcd5a54a5843edb839cb0142a3adc59954c6c02df924dffda5c564af9391c8f46311f0cc864634ad73a101e438086337a97563589f790b5da2b3da0d86a55bb7059cb354db5eeaf321c7779ddd7b697dd280d307050a78b10590e3033a19ba31723c20439313d0bf9a4c8a695d359b9079b4af7eaf9d4bd0a7d741d8c54a2cf49fa24caadc2d009bb90a4af372acd7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325540601Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568, Name: checkpoints/5568, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5568?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=44c0d7258cfded2feca8e710c478e034a42842fe72bcd6211e82fe324cbe5c9c2a5f81417902e38579d99c4b64e4794193e056ddf0fc5f38598afefecaf1b14f5dcbef1ba0f290404d66445e23e36a08652555bf04015f3cc42d8b0243158da25c78a2857f3620a85d63f2f5a9ec713295c0ac7f7357483f4bc163056f4b6fc655838dade7c8c07eaf64ec44cef18aabd8e235472d9e65a0153dc5fee2c85225a35321505fe5c7009e79f1b91d9b44f6f2c4818354e6a9091c561d7d1c969549071b8ee4518f85df776ab9b9881af24984b0d7161338ea16290e946965c27aa75a426c79799bac6936b8e4d9e0888f50da21b3dd78b5d8b1bfa47d4cade5ad46&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325429983Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/d, Name: checkpoints/3520/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3520/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a3998d8d55c30579006fd2ea1929c450bc4f311db61a866b2d8e703ba39ff21a0f8acb52e848bba40139d6c23eccc3f9e3df5b915704d9751d8426148d74cdb2c32a5e67a192a66da598cb411995692cf83fd5180ee0cc0670f580888cbe61cf81502e9212f81001ef5d7c5f90ddc322160c3ae489bc5e5eed5a2340f8acb548550479e3a721ae23d8b28cee6554e307a1dfd792d5b90f85b832ee33f7d04a8e0ff61c135b30ccd0a6bf862968c712b67dd11d1522028c7fd0c6ef583cdb8e809610f8369dae41c0a0ccf6cc594c785e48ffc51a8a63b59e29a78fdd58d1f4983de27ab7346cdc70f3c45c0bc409545c686cbe7661af9d96eae74cfadc7f3def&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325601669Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/array_metadatas, Name: checkpoints/4800/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4800/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8d229b70d990ef30d33a65a184dae24fc7caf9d028ea5c51f4e2ed053bea7818cfa80ddf7a325daf21570f5ad04f8e8c8ccac1b1437248596a961c65b90f08015d28836a93a9061051caa90750def4bf2ad008903c4c14988fdef914d9823c8ffbb597d4954dda8113220fbd531a6dcfa5d46af31fbb50fa61ec0c35d96c929b30c0e4562805a7605204f47faaa603aa3cd6af6dea1e8fa9b0ccd8d0afb2d01fa0bd9b0d25f72887dced313cae21f2273d0297176c6b59898537a71d61e3541f162acced6d906f9ea1205167ac0c923395917270c682ac9dbc5b0dda6506154292764a69dbd6020f545fcc8c17065b6b66dac2a0448bd1c983493f6eb5d153eb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325504349Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/d, Name: checkpoints/5760/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5760/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b0686dceff37dc85e6881c1691c37555049ffe0e6515684ed973782de4cf406a5a5a7fb03e05d1e853460acf82ac5a45818d2b43eb18dd27977c2d7b0c970436e8d2852b42fa2ec70a7c9e326bb587d0e92716bd0d9a4e62679877cebdde27f4bb8de4a5215f313607b77a29ed97e4acdd7a230e83420246289f4bba492bca41fe82e7d875416ca62798981baba7014e525a18faeff98344e87576af7c945b2a06c19faf2081159fe1ff7030b93073dad02e1961b5588ea2eecf9bd1c6729ed3ad51b1ef905bc797af3cf6844b424965a7db9a85215860d25710b913ceeb9ce76d40ec0894cfb51e967c537fc0113ca201f6d7ccb07f271456a4285f3c97323c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325641758Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0, Name: checkpoints/1600/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1600/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=6466689c8341743abe49b913893a6902eec516772a2047ba362a2da1742208cc82aab95d2d819a7ddc2449e77e25a8d49b315f7fa7a1076ade642c3d398ea69aca523c1546a6f842355fa32d5e6fbd12478af0d9b3d9d5c198f8aa1f5815441024d0193cc9bd0941a4cd7f470fb286eb43b700f807ab3ace9889800de5c55a6daf0b94e2eaa0e8dc40bebccaeac9f82ff6c30c9c4a2f3e2ed8c112f96aa56bb7217669adff1a3b17f428291e8695e32a71d23bd66cb6d4c46a589e9afa3f5330a890601da0b543d5adbd60ee864361fa95c65a794178ab05e54ec5b90eb2dc47e757e5fe6feac8088c1cefe195f4803869f52ed8a42f1a18a740721bd00210f7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325657745Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0, Name: checkpoints/1024/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1024/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=00b1d2f966c46e9bb0b610b3f766494e1b63c7e83b0f94a37d624634efe70c3a57239c7aa16e9fc2ddaa66ead358cc833002058f7f557d161f65e3dafc42f52e77bff44f739d12ced6c8fe44e6bacb89421bffbb371bf6be3630cddbd3910c83754b9e402591dcd85787b17f251be9d08d97b0097bbb69b84508f39aa9246e8b49b5009c73bb35e45e6ac22ef83db15eaee886fb81b7aaca5761034e80eb559434b0b1a89e180692f6424b83561521e962ddf89eb37fbc9887b23403bb961eb1e8c47e5f143ee1fcf970344a25d6a0d56e1651d7ee5c4e34bb767d4fffe7d86829e479fd5aa06812d37ab7261d0930e379b138b8e29a3a09dcc35cde6722819c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325693609Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/array_metadatas, Name: checkpoints/4480/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4480/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=78dee0f8b4cc1c4ee4572849b45ee602de97253fee32e42e356fc449fab135344676fc58c6b3ddf098866a718454c5f301806648ab8e223c80a99e149e4da2d3917af92ca2aa70cd594f470d61da16471693eeeca52f5168634d56c43891ce1102de58a1425107cb90aada5051e3300e9198721b4268fc00335ed282960240dae22b5c3a862b08d88b63387e5ea4f08ea5aefebf85af4d7ec1ff2736bc590ba4ebe5c8d98867eaab2baf7a33610692fb0d2313a98fb376290264d0812f6385131ea4f67ebf05d9863af090cc83c3e1498471e49da2893f77207ddf8ce5953028fe8dcfb89002cd37700549539ac4247af1071db012080cc5ad028da94097fdd8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325605214Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/array_metadatas, Name: checkpoints/1728/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1728/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=33c5e741f7c58754012adab298904f1def5c03ab63d3c4c39fc3a8104ef98b4de52bf7e4de92966867739467a2665ce06d089f5828ed93b6802a6a60b77b8da67eceace510b13f966ff9c09b801b73d2e3804ae9c4a894c5dc49e2db66df815b9fd8917cc34f59fcb841dab9600d8830208255ff590da781661ae549b6fc3b1cfa54fb22d80309136f847a81387b5bda5b7a4323e300ad528be7bd930d7d894d9b648db41ff38d9648447d945a7851c5b235959caca7faf7fbfdef8412998c7b9a77735c7dfd7f5aaa103db4fd4249c6ae175333a9a690e707fc39b77b7125ea9835c0d3408b3981c4c1fa1a95e5c8998087af46677df12b1792480df85b3c93&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325716934Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/array_metadatas, Name: checkpoints/5760/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5760/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=5b554a0e553a8e69d5b2515291282aafa4576248a8c16eb3fd088c39c938543301d2a3a86c6947408e3210ac3970dce866044d4adf664ac2e4040c76c7223ab17741eaac00619a607fb141f87bb9085d039bccf67891d5f51f6022588571cdff6919551a654ee3f11814014503853a92364308f4ef9ee836473dee292454f2544c8ec77967e6716e4ca02a1c964e786e869e4d5f0187a55a44e91fae63caa5a488c3793119c685e88a3d98ea5f6d5108ca9bd69cccd9960b36712a123205ef63a45ee9a24fdb1f3defb43ef62f075c99ece2de015e5adabc94babfe1eb09776600fb8b418052f3274de7ab7c94eb2ce11fb53f6d7e106829f15bbe2239acbb3c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325752712Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0, Name: checkpoints/704/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/704/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=42ee9153fe8151c61e79f3a27893d33df5ffb94057c099b9fdc187dfebd0afb0c9d6141551de0a90bae74b2e6af8b0f1a3c5b37a961bf4c134744c8eacf0d31771325c656afe5f83edb2be4ed490753225f1ef4d676a9dd0fc18fdbb4a4f18a1702469555062e72ff65f8ef7fb8317dec086332bb57da313e6bf9e1536bc3206bdb69c1a7afaa33a4cbd10f2198b8ec1290fc8082e0f7c84d1725feec065e8ba6db18ec7a93117844c8257c92fe5b2e38fc5365892d3cfe3d73e51201514261eaca64dc3fe20bdf09c3e3f557e404df10a2c7462744f606d100826927fa53e2426cd330918944bc075e6195b871e69296383b4483e3c98b508052c0739d4efd8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325780496Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/d, Name: checkpoints/4096/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4096/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=672239f957697d4c76178f1fca045738679e650e42e1544d09b3160122aa2b337cba988b46a910fd2f0995b659fd0b7cd367ec4f6c79b39c49b9c96fb0ba3ba1c95749c093d379316825e7cdbc6c2707de0384c2dc40d2e9a687fef95165ccf60f43eeaa3c532fb338751fd92659f05c0522e70622678e436e41195cca9ed4cea7fc1b38c13c1ca5d8a42b7e635987d1cfe58fec5d3d01ac65bbed60c9c5a64ae0691d773175138a610436d89a0700f3dc24041dd6eaf95157c4020e74d491e2db7b50e08752da8544c6a07479459863c14e88cd88e1ae103c86ac10f5ebe703788409ed93902b7a2828878f56d9647be6300004b190dfb9c78746dc453d63d4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325794698Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/d, Name: checkpoints/320/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/320/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=67b483b508dbd2271fa2d38f3fb232149c99a0ceda2c68a764f0fcf676472b309ac6201092f721c26612c7f90abe283ca62615344be8d480ae9d99d0dcbb8fef32474979c6406e39116b33c78ad2ff8283767b218b7b583b9f7f38affb39611935ba1ddf53032681d88d0e88d894433bbe9b8b1a1742036dce4f51a4d48028cb721347204f87c4249a097c50a745c34a298ebd3262b98a13e327dc77e47780c7ba92d40c21f89e8b1948bf73da577242cc17a3b28fffc26c93779422ee5dedef9cd19f125d58956323b5acf254b1f3aa0b76542662cfd0f4625905f11a4581b50ec5733b3857f14bd342efa0f6ddb5091f680c6b7668609c12409422837f1106&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324909324Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default, Name: checkpoints/896/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/896/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=50250e2db6daa4176f91cfd059b554c17bca476761c6287187dace2b6c5007020f09c5295661cf3f48b84da19f8a7029144a94f4e8d66fc88625a27d3e56677011444da28fa01c7feea8d080858203211accb1ef4483209fa314023fca88d4a2d19363b6b0805de26306bca8c834191fc02ec2eaa760e7143633b29e08ec7bd02e235db4d120c2a2ffca6cb8267d0e7baccb03f18571d1f0d6c1f7376d2c829b9b630c7f212934b820f136a061ec025281ce9d7dc1506e8940d0109254cc7519d2dd3a75097488e25c1f88565385f55577639b4fe6c6004894984943c274d0a625d0660c8ca4390f841cc5843a8ad9ebea617d28c991ac49ba335c66bfa11e82&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325862005Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/array_metadatas, Name: checkpoints/3264/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3264/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=4692fd6b92983bdc18e835c4f187f93605e5f7b51b92b1d80f0d462fed7a4c9711c37beffa84ba51f3aab44c6b8417ee8f19bc8fed2f46f07ba9f0b4e89b063a70eafef4aa722a0190b7bd6708a0b62c899c70b0e8e80a00cc3924b8e3695fc5db28fa7efddbf90239bc5225981cb4e06e76a5346e42c790816fe40b88ba55c5a7914b178984f3affb2297224c65658c4aa2c7bbd0ded7e1f9368607a6240d1adb25e2291663bad731f02c96a98648c40f21e5690ff04bfb528e47e8d920d8587ede3b055fe0c8b0e20534e9c597ea926d523cf84ff8a5c73498767110c135511a8803c57be9e192e0cb78d0c579443c83ee9fec7b413753c244c5a6751854c7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325877837Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d, Name: checkpoints/320/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/320/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=459c6728fbf4e493d90ecd08d7ebfb7258ee39b4c9747043f1df2207d417f32ab96ca4e2023502ece739d9d5e142250d459ee743bcbabcf372a51e72b641a4f97029d20733d3fc0784a5502e1d2fcb2ab34d461c18157e45446fc7782cd7a1582cf3d1e55be81fb2c90599fea76cf402814b151a859a3820c6ac10961b3e5c7be6f9de38439e5193b568561189eb85cc9f3d76ad3badffe08eff0c124aa520d03cd8ca22a88c60c55ca0f15c8f8c5df188adcaaee4f48c10b1cca09b2d9fe96f303336f7ce662c16f7c83ce8b673d6543aaae65cce5c1e76d25d9637cd94233d1cc57ed1f89e4747fc6ced9c18a55f187b39d8286d567ff862b9c4e9d4387115&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325907738Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d, Name: checkpoints/4160/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4160/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=0d144190a69648ae0a5a4fa3a6d2af8956c6ac3b465bd48f1a8b433c800c84f36eb2941197607f91422c20686bed2bd81e886a337cce38f50bcf5f660b8ccbfd448b10b1c98a7bef7ad8fe17be6eb8b44fb9e98a53ab4b6b1c673d87ba73654c57022569eb76caff7b0715fcf09b856f65df3586fa89e46b8d5b08430c6250c5df39eca35bb87f9cc1dca4d05a14be5f1455c9342733f4bbf731bf551ff2e09c733942f188876731a636bee46f28866a7e848ae066a7871493cbb861add92002798508a130f5887d3a1b1b3e86a2222df93fa6e01a467260aacc62df416b558f3450c38cc50c71a22ed022fb464660d8b0836de877d93737919c655a41e3d404&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325924764Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888, Name: checkpoints/5888, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5888?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=579f71ce41e19a7c636e34127974b294c9521003b8191fa5d204ed7a270cba30f86de4c700142baa575f1f4784addff3419930a69c70be0e023ab61ab8843ab841865825c533b7ee5f3975525c7c72819d706cd7e026d8d5cb9439dc9ae852c9cdaca26bad6383b42e9fc8751e446ec9f76762009e170bf55f6e8d84afc7359220a554dd4d292be180647428848df677083ecd5d2aa228f0acf557d838292d5bc377689012f3d4cc37e26a328acb596ab38003442024c7c158a5cf58df6e7876097182d4a904cdcb2c747be2bfe01e71d9f6c1d7108262fe118b989dc131d58db1596a9ab7f1b5cb260627eede1b39da6fdb6a364f8f9e8c28dfe25efc8baed2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325966953Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0, Name: checkpoints/3904/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3904/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=83f11597b140ce8dd04fd81d04a70fec6678141b7697b59b5dc849533c96df061a76594c2dd3a510b3bf985ade7e9b842e80f47d967144bb61dc49a5e3f3716b61bc949e0a466dfc54aa4775b4562ff4b10ded2b2b8fd9156c4cb554407f7f0b5e983389dbc896a1b18f28990cfd0ae3d3f8933ab1b7acec86d31fa10c9bfd4ce0978e83ea5f1551402ca9a5129f9860343ccbd4d2ccc62e0d63ecdf2cc5c0412fce0a21a87d9952de765bcb4b99946ca59e72ad16b418b6731911fbdf247b33b9625250fd1f2236b6bd338ad8c6f116a5dc0279ef900ec4d421eb882966f4a708bcf0de24d2b3005d16c1c5fc60951d15469b0d3cc2f3f7cb2a82ced4f7fe8d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326011709Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/d, Name: checkpoints/192/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/192/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1db79e83fede3ac2a6636406de673241b0c5abbda8bdc4b623a49acf3f2ddd2038537e222ec0bfb91846c0c8336a3a6926a4eb0df90bf125c0a9bad1d6f3aefd16192247cab64c414b208e2a151075ce5651315634b252ed33ec090b96d4973d3e5cf3e2b14c5b0feddcb2580c2532acc01a8ec2ea04bed9d3b716aeae2455d8b7b5545b5972f98e4700b84308616f48f49d610af859e47a444897d15b4f8e89a4a4b1f8a7819231763de34241d7ef2207d7edc9933479b447d9ae3f153779b15245e4796e3f76d685b3222419241427dacf306a9ab50a83dd6ab0dc568e6ccc24b8291d59ec26bdcad65ab112722ead295b177665c09f02846d214a6bdac7bf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326002311Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920, Name: checkpoints/1920, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1920?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=4926fd40db9ad6e484638dc1779ba695a85433edddf6f254ba29ae5e6f2975aecaec5cdcf7d82b7d78da38a7d90654baf901c3ae67aade643e4765de48b97fc533215a2f5a571e17a1580f4a577bd425ff4d5ebd3decf6ef82a888a661eb5358f39c937cdd86ee3b49f182cbcac930a177aad107bf9265d3ff4b6f82ce42794801d69a68fe5a6cbf08f4ece4d6b3b09a3753c87da045526fe5f95051ca1d0f55db0da929cee5fb1bedba22276f67d5ceb5d8863318b756fe2f559d5dcd8b1c20f38a80533c22c05ce29eb768f012df97c57ce1b3b2bdf42c87d881111b07af71d04b9a188e07da1abc1beb4cc5be388316c18de142b991ecfd8ee9c11de88937&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32603579Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0, Name: checkpoints/5760/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5760/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=0fdd08f54a8294fa90327a5d1571848b3848d35f26d63fbf136488eecb4e7a0d7802c3c7072f2a128c3a230599417ef04d1f9a191a66e6039f8ddeb87824f702c3005d68eba1b4e9cea25dfe7458ce552de46105662ed4c4ee55240e02accaf1c78a58c3ec51973eb717fa52a5befe8bbba714c84e48656501c6938423cb223fbe5584d12d477faa4aad00b38a1a4147d95f9024a05e36aedac163f2a5b26349e29d0fa57ff728944de64fd421d77dbf968a0a485f0eed198f985f721f7759f1dafa92be3276605541f579ef990a7fd23ac8b43b596d4290c18d7b71069b395bb5627f5bce1129f6e8078d469e0e1f65e5d3be61cdbde95119e6ce88570e172b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32605703Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0, Name: checkpoints/4480/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4480/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2831c6560e627fadb98ff025b7781b7e84a99011dad936aed874bf96ff1d2130fb7082b477f7a490fd49f74a6dc657f582f135977b418d2f8d51dba2e561892d4224e944bbc60a8c55b6c3a8e3fb550fa4a3701a5d603ffbfcf4a9c66206ba0f478f9463c87892a193b42c3b68c00009c18fc2dea430df184dbe2d32aa8fd84afab1bfb018706dcb33f9b172507922b09b17b306c12e7ad5d7b7648c8e9dec32e1cc1a6580bb54e2082e2d6d15d04f1c5b8848bf55808d62b7a6447bca743e1b0ee588c1d176c61da3b6045ae8cae36f49a4534272a63207fff1608fc8f21c4e18e73db0ba40521ce32753363497dc93f5a7c5d9bac3fccbe0e180f21407c1e6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326103771Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0, Name: checkpoints/4032/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4032/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b2c5a0cd899c6e9b093a4b27b21dd436c905c499b028db17423af2658c55afd8dd9840a6d30087fe62d18a8902da137ee537e781614ccd22f13db9c53285f116fd53b3ff14f9ce1d499a57f96d2d542907bd8d3e9c4937b29c3778fb0abb6229e08e6bc9c4e4419063c2c12d39ebb360e40bcfcc06905cd9184f3f5871af4218e0ecea31298eb294fc20959b91d3e8579b40fbac19e59507827359aee63d1385d2e3e73b7fd77ff6663c957bbeab1c9921d155bac385ea673833ed6f04dfde0d3bf3b96fd478268693d35147a7436c16b1bcd413d867d4c5ea938e8aa1051571cce5daa24bc0bb24901ece881217663e85b95470fda5a2c69415dc4dddca6d86&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326126381Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d, Name: checkpoints/2880/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2880/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=812a201698124a203cf6b56cc61156336fccbf977470c2d05f29186ad71c5324c79595929c79ad207741e486ce2acd2a56495b585839bd866e34b15e4b273713b384da0625f9e748495153811d2739f364e13cf028aa985e7917427ff85a5f6f66ce8438cbc3e263894fe75cd5080b59aa5ab567b32bfeb9bcac41d435ae3e1a639270430c6165f2a5fae7db3dfe0cccd4b26018ef328aa559fcd302bd42d7872e1528de03b553561e8aa0d5d1ae267da76f4cb78ed250ec73fbad8e39f7e73a14c88fff77aaad593f4c40b82292fa55f9ca5fcf4e7901afb82d6119d0aa70be2435aedfa2636ef5ed27c9b94f34cffad9525f845e48321b8a29f5d1540fadf5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326130698Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default, Name: checkpoints/5568/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5568/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=642b0b45df91297e72836c7f80871c8f6b19886e18cd3cce5b2c7cc48782ae710d7bac3d6f3e939d8620bcbece3c2954ca6e9efe667eb16a2ab2c4ec900cce1a9fd4c39503d82c18f43d0ee7ff2d05bc8cf238a1b55b685f57aec1c730a8614c1d023e924ebba375ba878b1ac73d2bcf6362f0b378c434071d9ef9257a22345d9a24074fef209a9fe855af07b563c4587aa74d0d04c505b4531993d7d014a848a8d9790203472ffdcd22d96efb3041b6fcf7299b42bd57367edb6b4a7558545f45cffff78f9197d29fda519767bd8c1d4af577ae8704575f28ca2d6a345a0d083861a4e5c6aacca3c142ab26fea9ede4780629ca9166967c052e88380dd7ee50&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326183786Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d, Name: checkpoints/4992/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4992/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=4346c7a49024d986b23b4bcdfa45a7afe353f8ea4273ac517e30ba31a3b26a5d1678edb6e94a4e97b571275b5a1458b9046bb6f016d1a1d2f81e3f81a2f9f5241595d0dd90b3347f33817fac1cfca1fab2e2b486a32f1991ef44adde8e64f97a3e31722c07dda9db0e955d9220ad94b0883ba2b308acfa602238693b8699071fea21a3845bbfe7d549bd3f934d6695d8cce13b88e900179027d0d97e41f4829c2bc15e12377dc4e129b9d69aac3b142e2addfd8a40c657c705c59f69acfab97dd1ae8a49b4d089a2ada6cc9eab60e413e6d7aebb5adc88ae65a4b8a47d39624fa47b1e4062d484fcb4303cc152ef59663fb91577d0ed6aeb4a55e5f5a6cbdc6a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326199307Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/array_metadatas, Name: checkpoints/4864/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4864/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=ace80d7ba14fb0f6618ca4477721a75ccaa8eea6001938c972d3c8f5f440a58a5b3c17b1ed3bd56774d02dd10cd51513b73a1a0115c28c7053cb018939799cc3885eee7e40dff53fff23af838d7cb1807638ca259328b5633f5e967b268517e758be8bc7b223e3610bb262e4111e9eeadcf3fa051c44840e7cbbf7ab5269836694267658339d3fcaf0afb73770fa05d18c59a4d7ff5a70ac71391b8298e472161cd0b4a1651a0d0ab4f41deecfe72f3d4d3f4fd9a75b83a89d3dbc4333bdbbc70d0cc36f97a2669de27eef242a0c46bc3c281574db9146967715d69baf247ced005e1fcf6a802cec42152abd4796500b9712505273a1276f2fdf1d80f4608a12&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326008444Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/array_metadatas, Name: checkpoints/1024/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1024/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b184a53bc8bf2f2e0a853a9de5ec99cc460c7a1d32e328ba20eb1232ff7906d4a71de3a15d4780096af8932348113c375e6e7fb31436419db3789a4761dda05e8b7717748d1b847e48381731ee542bd2f20923f856102b4b783b9a09c913bc2481720f2703a29b7f1ce1dd0fe83b1b4f68b75bf97c2188b1e3c67252982a3890494d682d12d18e4b9a1bb985b1e35edd26970d3ab7294a087766d2e62b4ee12b11303b74adc456b00ce5aed48f07f147da08180c05f229fea45f235473778afa3a709f3016da86081e6c800c279bf91b647352785b2e2da5529a57ef0bbd70fe6e9de3c6bd6b9ff0fc72d3bf5c695bd195872760748173ea16f11aca120676f9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325852885Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560, Name: checkpoints/2560, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2560?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=56aca65b2d88a2a85a4095c36b659e07c6217df595b3c94eee9b870c362a3bf82cc58eb57d0df7cb0f4682225e26c4687d69f06c85e68208e743a8d1826e3e889204be4bdddd561496a45cdd31d32c2e2bec40fbe28d89b7349c45f0a7c689294f9c4ef250ce93c36a209bf7bfd2c3f62ec07be4254a3b963924a98595efb7da913fe20fe78d4dff65a4daf44c8ddf7b0690d0db0e8fe2d73318ecb083c85a4f450669b6ea44b04cc28dde04bb07548fbbfc7f23041c593a6adc977ab62a32e1953c89be64c8c472ad4717251e3df7d32b733387e4e89515c34b046645c2fe7b847a26ad82dfda9b9ab9ff661a6275e686364f826b56d3df02d2de2d69bce458&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326173241Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0, Name: checkpoints/3072/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3072/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b1c3242a9a9afea425ac38a24aed9b28a2452376fe6f63727c94b84eeb00202d50542aaa4713b05c060affd9cbb46f426fe0e596a87fc2a99b6aa30a011bfcbe0caa13788f134425f909dc19eb9521b11daa53c2cf8737c641291144b3437d811f72482376a3dbc6a1ac03c2b0c4acc4abe1cee6262f8df0dee52820303b9ad38505f8aa229a75f27a0a40a29301b57bf859e3fcd8e55d9b667e6674cb4f19172239e850a9e037c3ce1e374e78c469636a98ce674c1f7b1a211ceb0075b692c597cccd4e6d8cb1ea5a139d112cdc9ec6f84c29b004b8319b8e712821ceb395aced57f1d177821b3c8bc7f6083bd5c2dde6c86154bee70df39b2b3746123934a1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326316519Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0, Name: checkpoints/4928/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4928/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=488bc4611f3efeb548df71b9c1efd53e0a517d767d895d6daf0f952ab40fae643cdfca70ef36b85bef79d949e86c6e4b4ead9617d1ea68e4828e3c699178cf6cb4524457ee35487fe831d2ba8a4bc580fa483206cc79cf6cd0f0bdaea81e71731274566e851da4ef9a3541c840c9cf38dafbf4cd8408d3571255cb5ff1816af998bb4f40362edfa99b9154052019457a1927ef8a6320b8360c06c930a46fa9b7b40344ba42db2d179c82ed85cd33558eec2e5adb82e2e24aa72add295b6288a5e2bcac012db624d02bea2e85046be51e2af7c6a693a582eff5309b7fb7e5224c3836a140e2b77562caf1da5560413ce88bf02b85a8e4dd53ff26e24d35ad38a6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326133741Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default, Name: checkpoints/6080/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/6080/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b9c3f15df2fdfeaf8c1c8ad7ea8a8386d630479f56f6d3a62f99df92da6b2f35b3504d620742febfaf6f5d802554e16060b957fed71dc3f609380fb00f68a38554eda06e0aca6caa94fefb1488c62c21b6c021524563a651ae033cbb8e9a0c0aa45bfc2ecbde5c0794dd4a8bf5d755dae16299f7767e5dc200deb18fc5ccc5d917d645888491cfe9d25578be4c7a09b44fa3f835eabc8743481cd831000c12fb94668371dc5c03878ef78a5fea9316101b97260fc009db167f63e9f84e6d65a6227b8f4b87c6340a5b0d8465861855d7d4ec2fb80a86cf59ba60d7de9f09565c4808c53e94f5ca9be17da58dc26574109e8517fa55876d790f8397a125ff77ce&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325206814Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default, Name: checkpoints/3584/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3584/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2a36415729fb2399d5841bd0f4b29e9f751ab0f6a6c36ae2d75af6b0e705cf392856ff874eb2a6ca88f8f7504f3cc99bc6addc2140c39e963a6e8e9f57966b35ea26098ac382f7c3fbc35da16485b3b579ea3f724f20860862ff28cc7179d9f497f7d03d6c886321f24bc72b26b72cd7eeda5ccb872b8be0fd1f0924500bcd8c2b4b40091ca77bc4302a7447f92d4a66fda858d877d5fda6ad97f4c86bc1f6596e825a0c3f3bbbbb32d16514ecd3ecc3025df9140375ddd577767530d2caa629a7e081e3a2c9a5a8e50070c5cf02175619268002a300216c7a8f7f50e508bc9036e4c09e77c2cb91defafb4c9252205fa609bdc0e28e4e2c0e7370da145ceb6b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326337292Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280, Name: checkpoints/1280, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1280?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1e4e6778f868dc5e9b583fed13f78ee81d9b11af30a718a612ed9bb8bcb9664139628f04e696459ceb469d708aa88dad65c599d523caef748584d5ac88da16bd779f5b3ce429a47b48c892454c32123792c902a06ea1001c9967172f2f0774252a2053f6f2c2b242d82b394e854f7f72dcf85168f69760dfb5bd2c5e3186edce5ffe2f7802df08c261f7ec65ab857aa79a2b800f08efb7f4ed2ea241ccf7df41cc07712a176c4487a69212b79ca741f449d2425b7fd531f402db0ec9880603efc7cbdc7bcdabec187f9c7e91d9753345c5a43fcdcef6d62c62048959a14815e89011e83ea310df25938c5417f8e3641b96e42fc05b4c9212a877fc4c0c5f7cd3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.3263338Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0, Name: checkpoints/4800/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4800/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=9e8afd9f9a1ec851ec7ed9779367e93626222d1c9b3d352a7593fd095a19007ad011bea21a9eac6a633913d71cafc1f8206987b1055b34db2be0899bd1689e3a7533c62442051ed11699751efee51c36c752949e08a6bd5d5f649445258d1d89633db5dee6bffd9410e7e76363e9d426dadf11a127059d855667a033698ab9eac881bbebc2c6360b62243498374c5d951b2934a4d7658807e745c43681d4030afc99d130a3bb0690ba148ddeb6be2b7f51dde578718742a58903886696daba4f8ad883abd01c9314e2b8a10e334e807f1b3bc0ba8b4444eb5f2793aba84eb9c98e3a62cd97d7281a4ec209df085f441ce70088fd822d12a4443164d69b958211&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325707655Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default, Name: checkpoints/384/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/384/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=986c87e69133fdcab9422111be1f8813ebfd1eb33a8b60144ad8c8dd87b60c5b4dc9fe15151eff2164681a9f247e929aa0eeaa9784b848fc793ea828f1c114aa648499329ecc922f760ceefd45b02e30a304ff29f0af201dd8127ddead575311d04bf28500cefad18f9c0c912698dd7f730ec687c450333c0d368c60712b8b9658fc1d13eb946e28fff5898f4b8ae2f1543f9228edb39b1f8495aca4aa0cefaabd03786eeb96565729782acdf04e99a936e211e848c0d0665f7f0d925e99d5ee9974f3652484dc44c654d38b96bfca936819e481e831dd5f0dbf8a29ef6ac58ed2fbd4c7f86ad0b304ae2eee9e1444dfbbabc4d0cfb281fb2c52bedc6bb02f81&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326962003Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/array_metadatas, Name: checkpoints/3840/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3840/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=5fd2a15a1b2df84865a93030f261320b52eab4e3e81c8c064daa9aaf042edcc26063ddd90b4f842f931947a5c999d29f71ae86d19eac395b4e00ea3c43231c6df5cb161f1727b62fc80f3411ad9056e0afaf3939b0309cb132f631452b5d715edf897e1e58ea13f8581d80898e3efcaaaf5aef1e7cc6593da3d31d6364b8ad6c8012b738be6711c71553122798512e55d08735cb135a87e69f030a4e9dc25e8de950147e7b3cd3b2a300c5af0faeecdfba48a21904606cc9a35238dbe45752e8c3d5f6965dfe368f558dc2deb5d1acc5ee5e227c46e971a3ffd3934d57a658cae633a5362e3655d7925bd5536fb8d824cbf4dd0ef884bdfdca213f8c01474343&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325876574Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default, Name: checkpoints/0/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/0/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=08efc23737e411d8930a53207b43d76ac96f7a486c30c8187f95ca25327780c4f97751e9d6d6638d46071d196945a1ae2d7bac39da0f6378a7658754e0d2cda28ecc6a010115ac6d3866a2051971122afa8e2519ee657241f8eafe2989b7b70db82f8fd8665873c939379e5dab6701f43611888c495525a9384cd05938a830b38216420292f32915a799c4e7a710aa7f8cb4ce84a0411ed1f8236f779b14581a4b7f756c5d69505e85bf75a25ea7f9f18ac26184e4265d94eb35dc30ea48db7a3c90919a0d57f1e02b31cf4edda50bf05227897777baa3fdcd30a223e37d06427401178d670cd5af2e446fa4a2563549053bec07723dee8b29482cc52f16099e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327060964Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0, Name: checkpoints/2560/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2560/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a7a3a4da62218a1c3836c8a675a17714b35a8e0b51750fba90f6a79e9f7ce429f020477c572fdef64302510be2ef7bcd22073329bf95e0a1f442788b19595739075e9a659ac486c1efbdec43463300ec15c24beb484bbe015781b15c11256246cfbf3ab870ce4eff942158e7d3986bb6667752e0a2876afdc9b974a1465f27fe6e80db6dfdd9382eb9712945ebae29cda9f1c8a063644b212169ce285430a95a8af7cb6cace29571947c64249cd277a91d292755d95b557dd3b7ca96a96740ca0d644b552b968ff07946f11fa34605b20c01966f8da05f1d8817af46526e69880a002a43fa546dd83c174b507ea0d37242e4605a49a7132d993d5f8ed2254796&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32624763Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default, Name: checkpoints/5888/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5888/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3fa68df81e3c8b31c73377e987b134ba395ad696f6910ce891e09f45caa95b2f495688a4419fb277f1888b95440d4bf0703f4aef6279d74a4de6ee2a99823a4b93f9dc4594456ce8fdc3a92e15a9c190b00b111a7f57f5933e97d99573dc56cbf0d0230852ac1cfc2c802612b4a361596c065eade9968a07c47b5753bb19a7a55461af85f809e4fd2e29bb2fca3da857d736ecff055a662abd77e128ff8628e0c5587e7a2bcbd2eee2f53f4c809ca363fbd053c638471afc37fdb78ff9c9c2b703c6fd57f9a41cad54e18e874bff2ec27a7930cc192f2e3329727c2d95b7398e8f894298e2d80faefd004426e328c603a5fbdddefac8e564658332c2bcd2370d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327192003Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default, Name: checkpoints/1152/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1152/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=598114a2d0c81393308fc6fd6ef568e83aab87d97133d91632e6321d012fe8ff07b5c5afbc27ac8c5479227c673f4ea4b2625ce02569947be383be642ea777b369e60e360c448c5f8a34e7ff65fabdbb9ae4586f696d5938f5c795e8bb630bdb179cd54ac26db396cdd758783355b137ac0b26b52a235790bb68b75487d3cffef3eb5902ac7c85763f3956541bcd70ec4198510a85507c7f5af01b46ae97af8c2463b43b1b4bb2d38f8a877e95de292bec1961df59d85b547d33e92701c843faea9a3665bfaef2886eeaece6c6bd3c021973605e33b9804ce1ab891ee9cddc99e31dbcfe222ba87a2e26692bf268e2152b660c03f87147a374f152d5f32cc397&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327242716Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304, Name: checkpoints/2304, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2304?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=ace25e3fc95c80180ee4184199bef57c9e6fb377f16f8b71cbbed0c91863fefcd7c1af11cbd0bc4f68efd1fd8fdf1af5e741e285e7bbad85e888c699465634cb69f5eb616783e6a27468d4fa5333f4a23e14330dc5eea10c9cb317efde0e21f15be51f334a76f97afd96c8d313602f2b5bfa1e338d19c42390107f066aed8dd477966794f44c7489ae65d702890e495762b71ccb674aa63180d0412fa2db54fe7a51384fd74638a8c5d2a8bc4c0e41b914061e93b897f72869c31643e0d7d202658395999f34c11b119e0a6ebbc41865ca7ce8c55344eb796873c91ea02c728ff55f390d5e0e695694fa11f81f43eb200dac7f0dd37470e29b1b210e90221af7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326035914Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/array_metadatas, Name: checkpoints/5504/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5504/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a50eddfc3553b22f01d6041ecc567f2b022a7ad1bbae6f7f645f7f3553e17953df40a7718099a8b32987150341eaea869f9b3dfa8dfed6755229f45ea80538728d26127501a98d954d9a2935582d4e3ddd35cdee4a50e267fb7a933c1abcbc9adbc3159f39b6b2c0d6177986f7fd08bcbc96f65627ffedae6dcf05fa07795eeca68c24cf6f6e78e254e1237f4056861d676578b632733b7c6efff65ffa7e467cc65456a818dce83b872a91192afa7ff95508e83c6b6689f0df68ff051f68e7a03b1191ad49daee10cf68feaea8dd4e0eee11454370a362fcfcbf78717b5d1055c15a29f8ea885ea7abbab8be3dc13599a7816e7c98a3123976ecf77b52ce0373&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327271444Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640, Name: checkpoints/640, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/640?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=85399147401158a8112e63a9b4b67969a6f8e194abd101d7ab8e2e15b49ca9e3210f6fff41122d69fc9f835bbb73229d927dee2c79b904fd41cf01db5b4c295183d643a259d0691d024eca3aabf0882d2319ee25d0c2f72fa8a5db0b65cf833819dfa3f97a30bffaa04ccc91c19ca62df0c153bd976478be7525f2e6024dabd8fb53e89c94ba5abd61fb648cf91dfc894c657ee06c41efd0aa3dbd2afc98f22392de0e11525a77f77e87ebe5ee9753861d4fe1a80fbb3a48ddc95eeeb866d5354e365098ce6cbab8cc19d5a8145b1120eeb5472a7a574405066de7cddeb41e6a64ed276674b0eb926f9bc79a1d05051a5bd084788d4debfb48badd2031abf769&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327276374Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624, Name: checkpoints/2624, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2624?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=438aa3908883365084a4cdbba14a9d8b23f3db4c6e980b858ba560d1c0a845b368fb61a4465e0a9563319e48502dcce76c561c5f31c27ea70851a152015d3991af5668e48cdf91d91154dc6e46c8ddf3aa31c916d05883524c0fc40eafbb2f186aaa6d523f32dc756e95fca0a6b55fb6737394bebf23eee52ad0ef438cb4509124f3b2ff71cce5ad5c5410c9a3328ce225b65935d9288aa17700f464f770b0ca07933ab514e1c3e82ccb5da29d4db8a681664d69777b0017fd2a6dd49c3dc021a83c9a1d22aad6449b4959d93de82622317af2d9ee3ac3a07f79ebb5c9360dd9c2828e643fbefa34887ee7474bc7e7bc819b9b93b2baeeb5259a27d332b2c9fd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327339454Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/d, Name: checkpoints/4032/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4032/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a3eb2c422f2361158c4298226a2794ca14bb76f1e05b9311fae90666723a5b26b0604148f7abe471b744b6ced9689ba5d8d3a0abbae52650a0cd9feaa2e05c145e999628b46e555d5a38cca7805a35b5009dd10b38f6fd46810c0584dc9524d75f1ee6d6ecafdf86280aa14997b9b6eb29d8e9ed01ffcd125500c4986efe4590a85fd112f30b075e5c8100a2b36397c7d3d4a5ae36afa3fb09d84ec379b84ffbdf635a2419a6a4cab26c5a4425a6db52a35c7f8ad7fc1c87708c99620f757411f91d882ef60d27483a2e42f4410bee725b8a5e48e0486245c1f434a192b864707dc1dd2217faf601470cce58d702a09410934b197091196ca3fbfbe3dfb97dda&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327338676Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/d, Name: checkpoints/3136/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3136/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=14573ca3272eb81dd0151908c49ca4f6b5391d86df01ecfbdfc4e84fb2659f6fc6af70da9e35e06f674f878179580cb7c948abc0e412ea8cd7f4af2325fde0c38b0371699668b6e7f80aa953d76b82985ca44c71dde1c7a30eaddea84f1cd8d0777a85df93936af8c4286d3e0afc37bc2029ad4822239b79e4ff9783f1e62129357efbe8c4b2bd8ad41686b571aa11a5e76c846fe85bf771e717230b10cac44e15861a7df0eb0b2a0de34f80bdad328e76938b85a74a2501c099a1e2ede750fea54143fb67a37b143a0df7e556a2269e58031204b66558292eba43525d6f7396070b5e989306ce433466363ad75a8992cbaf8aad41e9bb0128b7921b7cb4d320&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327366381Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/array_metadatas, Name: checkpoints/2432/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2432/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2d04e1d1a62059654b77578d7e3dc8590ea37cf81099a2f833b1abf8302e33430bbd139af49769b7c492078e1c9960adbd1041341aa992e2f40f26b17fe8a389a4e4d9fba1ff30e2e3eeb6b6c026a7542b5459721aaecdcae2b3df2e7563d9bd4d3de6e731d5a4d74ebe059d26457566240f7de00c4662a089434f11bb903889987563f58d0cfa394653e57c01fadff5eeff126ec091419db392cf660a45e5f8441973cf2a3ccfe1ddf92049f440510c34969c7be9202eebebf8b320bb743369a8792e023bace1fcef877a52bbe8864299fe0916ee981cd1d8cb9ee2c47ca5932ea601e598a319eb6c5c073dcf718d450486d5e856e69878cd790b2ed14e6caf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325687692Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default, Name: checkpoints/2240/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2240/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=7aac0bf4fd20297ae47c26d776e8a452b0778f8dfbb2fdfbf4412a39f2977b8cb079cb40be1c1bb0ff65b3d29454f1f3afd3094c9daa5dc711dfd286796da110a5e6bd767859c30dbdb9ff2754fe04a166194a7d78d546d17b8edc4453bea0083d6c05998aada67af1d6f121bca920584f318b291248195e46a4d9b0354335a6f34362d0c1dc81f66b744e3044bd6403741306f1999087b7c0deaf37bc4328d3422a3ad79f6f7c481023832c75c70a1fb9965252a606e7e98893d5770a237ffc63478113b2203f1d8022e461f0daf2a1c889f5d89d51e442caefcfc6b8817d333e2c6ccb6bd057e5f2f908f8571fc9a53e183324ebe41fabee5e00eeb71276ba&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327433114Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d, Name: checkpoints/4352/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4352/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=7371e292f79e19bbcf852d70ed138d67c6300c04bb030918c10e4dbff4948f8aa9596607da5996d23dbcd522b38efc887e0ede878bd288489e15703f9912c96672f24d896ddbdcd64011fdc1dc1c0df66534b261a04ef7a43ed4ad3df567646a538794d8e30fa663762184e43df5f7c825501b75940d614ce9ab7b0d4e416290a74d8ad3d23c82148c338b94be61b800c357cfba6860889a0515a9aa65b5a743ef92a931335bee8fb696ba9142ca5cfab49094bd67a46a09f1cc706c6908d3a8e05c699ff3c61b0a76d8c01c40986d67baa41e5a4dfb5232012481d1a622cb15e354caa0384e212af576ba321aacaf3608f321636c27720e50ac164e0448a3ec&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327483678Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128, Name: checkpoints/128, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/128?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b5da2f7b5090528ba8fd3d22cda3de455ee02a3da3b4295d64e812e176972d5af024fac54ceaaa4429c22e95a5c4367b38142e8462dfae1c271880d15268863d83ffb44d8b750c6f8b12bfe00b3eb58e931d16d88d529052e839f81044eda8e9bc7aea0775150129bc72a7bec23bf21a1ea4c79efba5949e65f16a52eaa3e8dd96a2ab2dbdd31ffba1857a76a42f9efb047ebfd6bd8d8acaf29888eea54d6642b97ea4cf8bea8088c5775352a5f1536087920fc0305f08b30bd8f5314357980c27d185b6e63c6eee7f40d6f39b3c5090a5ccb1b24feb53bf26a4c489bececc69f39252e3609abafa8f0b55cfe6bfc0a1e36c125087ebbe58c4a8b1bc69c0b663&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327482543Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840, Name: checkpoints/3840, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3840?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=9cbafe9093fe25309d0974b4cfdb68e381f4432ae46ba989a474a3a69ed0af841f22e9054dbba36dfce538e3214970b046ec5d9dfd8ac9ea57ebc08d355284a6dadf62ea08ed303214054fa3e235900d3ceaaf5a421778c72fbc8d00c99badbd023106c79676bdf4b550ad9b2badd6c441596fe9d8ffb92e5b993a03f422f8b3d8218e577978ae01c7d7fca5bfcd7092dcc700f8d82e3fba483b031727e287e65ec926a3a556755ff9bcb66dec253ffb75993de72295d7a5080088e6eec2ac217896a08edc8b5b844bc8f8df457b1471a8266bdb12e7d300a94e59451113afb4f8d0bf10b63d6c740763d5b7b003b4164c202bda49fe119d25b3070569fd3006&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325752852Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/array_metadatas, Name: checkpoints/4672/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4672/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=957b34013310bd6e9218ec4259aa015cc78f86270895326e5c731b5b80bcd248826e526f9ecb2dfb4227f7f4615dc91753d1fee8ec57c1d116ce91ca6ecedc7da922b1af02fdbbaeab1cf1af438a560107a972e5c52f67cdecb3898f17477bba180c6ccc79fa75ba94b559ecc080952c72f2052d6d9481491000056e2b53b46833059dba58ab4b1f7f1e7b4de74a038f5c7b29aa774046e9bba2370208c669bd8302b53ad209f25421c97d6414afe62ed448067e5533950e9b81403ec0f760d2334270f6ec83efa3b2b06667a378fc34403d81f8fc3a7df1facde12da233dd5cab1f530bb85a947d11e2dfae7c3a2d02b506d1ccdc818aef1ede7184662b23e2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327536954Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/d, Name: checkpoints/1472/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1472/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=285ec113bcd8f95b776a3e6c3a34d8d3a31062816194a9d5d33300f6648c4267227f61a9c822fc7daa60a989474ce728eb5312e250c9cd78a51ab6ca8c57816d248806cf7978ea2980fc117c5f265bdb4a53e53c2ae46f04f5b68cf67f4cfc9d9b895af72d2710d9cf5144e249cff7d304d98280498cfa88795aa63b6448f2318620d5a1e771f7be2c8818252da7f3e397295907736e22c9d5361de512d9c079f8741493d15da48bac2d527c30558d334936148192e59d0d7bef66ba88d2911d5cbf5e4963d250f66cf227f435501bf7780841c48840a37dafe097892faf94383caf1ca813a28c442e8ee70811b1ea84e6a4b970992abdba9c8ebe82972c45e2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327489851Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768, Name: checkpoints/768, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/768?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a0a4de200914e73b1a7078644b2cca227301c2d3190c07952939d12745b3870d45975d90ea5bbc048a44d57a33c59860d4ce1c1f52039f3638043ddead02ce72d6238e16720b8e8843be53fa91e41279f8c8ac700a908ae0bc31622dc7da63fec7b0217acdd26ea62fedac5555d81738773631bfdafa801d6dab3cbd604ef6c531dbfb7f9f2dff43de554d0f1dcf3ab708d6dff2b0732b0d4df4f198590b27898604c8c5490d0e374285f7999c19758636cb6180181e95bcc6ef05862759df0fa6291f1eee343000f7e9abd5ee3bb9c347d6f862ebb0802c173ab9fede902bfb7f1f7ab82b1576939b57ae178748a5d9b29bf95cb0efecde15430237462f6eed&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327581168Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/array_metadatas, Name: checkpoints/2176/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2176/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=944d8a955a769a1c2477fefa7d534e595e91e9db933e2dd1729b148e207c63dea495c3f3ce7c89b7fc9fe7a05341a2c980bc356a417a5718c9dfea4f7c7a17185ac765441d930f7de8dde50ef1c5fff5c64869c9decf880556c114a0c117e96b76da3082a48d326f9b62fe1cc531669621ae760995b3b600fc531864479584896127bef2960f14a26af7829de90297d57771a4921f5eb39f2bf49d817ac435b7cafcfd15cd1417d6ab5c6d8a7a1af9594fd6d5d0be5d0a149143ee414930045f6ebaa81549b68527e8c50b3306ed6832903a720225c6e8e07bd167023eacf59b1b3a83f7a2a88e26e493df677b1f889839f7b92133da9923c6e3cd96f22172c7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327338114Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992, Name: checkpoints/4992, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4992?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8d6cb18c9222e7e954585fb0b1ed6c88142488e080a799cc0308e877ccdca29c8d907a54faf9805cb841a0982585b779ca24c913966e5d27f35aa2473ccaf7af7e5fb90f60da5207369c7244d053bb976972c7bc4da6acea8be5d7b489c9e9fba652b6e0e5f62996eff9c405ffae370548829cfc384a6881b5cd8b537f8eafcf6dafa0c98647fc84f290b2ed4db6462f9ab3895fdc98e18af2fea128558fb440e42b29439c068c9d5e51a2491090a6c4b3ea9ba669d73069c4be27e4c1e24ee5be38fcce30d79a31a879ba2b7c1d782eb396839367e1e12c1833177530632f88bbbff7de7942bed4a43d4ede6d1a416d4a6600005df54f2e356ab829ed7688f0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327602011Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0, Name: checkpoints/2496/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2496/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=0111a26e70dd122abb33145817165adc8b9305afd5866f82ac76964f09e8fcb8dac798f02436cc776635b2410496dc0c8a066fc6254e0bcb43501f63343fa515e9685ab19e4c565d034369bccec40d70094d624569b3079aef63a0f75509bc2cba618b813ea28837148985c0eb784a471c537242c3df8db75ddcb412c1737eab9c0bf5f31a24df2ad8b9166058409d2ca7774c8d29c5e44375f9b70b004a084b30e60b4e7a11a667cc4b26cad05d30481526832527b583f894470966256c565a087bbc8632556fe11008cd1105c70727cdbbd18d2aa7267d82a19623f8dc684807b5056ff22dfebe13e0c66eb8aac3c8c3639784f604fe51e61000ab6ea05c2a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327647475Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832, Name: checkpoints/832, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/832?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b157a3d0245025fcadd4b4fb6f6e7e97d45fa4a22fc67c782d0b4f5282181b703b3d7d6e5a533867b7309b9ede830ea2751590fc61d83a3e4d8f22fdedb851238ec1d037d2c787e2e5147fdc18602bf42341953da287842f310ca1f01eb61fd7b107abbb78111c58adf28d2a40b1476370625f1bd3ac3555ff5c54fba888a460a3a0f9863a3f03853f0a09a6abe2d19263358d934d7235f314f31982ee72bba8a034c1cea1dae7c907e6c330832b1e3060eb4df07ed2512f2c6a5778f29734633aa5a0574ec53a917b53097c45f65b219f9916589509d560509c0d93a4b941cc2a4b50882be086182713ce325034b087202018a99ca307c55498f38174ffbba1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327467099Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d, Name: checkpoints/5632/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5632/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8d2251141d0b8d52dffd5549bdb529c4e183965f2dac2d46eec0521498d6a57f6f4d0a3c59b93b98ada9d18c0e7155461193193dae934afc4fe267bbb80cf5da65f822cad4f70bbc0bf3e49f5591186b05cd60785ac6e1cd1ce4fd9db04bb3335d13a09eeed15fd000b65853e04142db3fae029ea8f1bf90b40fb8480e8e1811d90b42fc2b4587b53c283440d950416bcca33da7c71ed51d546fe959a7ea090b05c0b1a957f4d0466fb61d321c586d7d9bf6a6c23444b1294a6c7c5389af030ab9dce0e3483134ba7b3cdfcb503a5e1a29bd35c6b521f314fd6ef1bddec87a83e778f33b77134b67d08076a71a2dbf5bac5dc8b2acfd750eb213ab982566fa35&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327708774Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default, Name: checkpoints/960/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/960/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=c1a5f8b586e60dc8d83a715e961723cb8b95ddbaadb744450b419b0d626572f0437611be38a145ce0d7b3450efe16860848f6b818cb597d55cb3f90ff034f33c092fa69c69d302766493d7fc8a00cb66f6f3a3c14b3e0746dfdc977481997a250d620a6501d00e1473828e1cd5405eede2faa6dc17e7723257b1ebd379a96d31020707a11802ed0b0ee8db9ed661d73e763b02703badc5ae3cafc84ef7cf9552ec2815ddbede1a745bca3c4b8d579d3089ddb9112ba9015a94863977c41202cb27f876a355c6eef56fc2771b7ccfcd5be4c0d734670845836aab37c4bc87afc38f320e2cc6a5d36d82fc658612c42121cfa6d1d4a8f5fb5989624da3bb058696&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327721599Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/d, Name: checkpoints/4736/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4736/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=934093d450fee60f61c98c36b77d481366604b37443af45dc866547e3e83826c81e80d564b8b3b109e38054efb633c61e4a3d04be389d5e36fb89fb1e47cbdecbfb346558d280b8c14d45c63bad34fb3adddb9111d21595ccbe47a70d74c0f04939d4080cdfbe2574b462a5e11078b4fa65bd1007e73fd8a4cf0f6098dbb73ac4b56fb03f56a45a48a1b06326d9b633053a9a2c5460c89181209b927386b216b76505308bb64d7fe7b1f1b23bac92ae5118797519e178e19a9bf1130cbd84439b155a401b9bf6f858fe2b042b13e575ee260dcd8587604fec51593cc0a904a02a4cd52ba8c92b20c2287687114c5143a5a2a6d92d9162b6e528ace77c478e277&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327800881Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d, Name: checkpoints/3520/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3520/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=7e2cf7e762ba1b9279fe235b6e7d7c673273c945d5d90c66fbac9631cc8dcada774f1c18761baa8d67b1f37a121f46fe18af845d4607f62c5fd54ea8133f65e15502134adcb901415158e2be17d37a8c77f043e55a6fe33d859614477a186a7d19752657c866c93c87410018573cba3fc36f1e0c0f6437436f99eba7ecdd12480bfd889265a0c192f296eae1756b17cefd120c2679ea88f96947756d94a7bd9f300ac0e51fc38c319be6dc82616cc598ccbf0b420b9a54c2cef093879eb1fdc028822305b876293497eeb8d65bc3225865b785b63e2d20c3aadda8232cc1ccf623bf5c7db8b1487dc5f13d7d5b291ff0fa0736147807f4893098748ecea79795&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327808289Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/d, Name: checkpoints/5504/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5504/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=51a0f38b9a4535b5272f88bafe7a312601e44a4bf5682a15e66983b1e2e25e9ab851dea13b5bc0952f91dbb17389eca3709b793e5700af342adafc3b213f6e6dcf975f9557acd1fbaf9a6fab0965cc458ca9bbc00f6c8a61f5b61f08a308f3482cc6a5e7f0913c2219a31977c644352aa50097e8a625bea22ade3a1eb2d811db1f81677a0fb5a9bb7d903f3b9fffb2c7b7a2506e9f048fc8612baf572d9a71cc91fd93f438f0590699b5c94fe3135cb467fe5e4ac08a054d5081832c743e315a0dd5aadd2e7e2cdd092d74c293c7456e27369ff21d66a48c4f6cc940abd94a8c3df2b67ed85e74bb3427a9b2f476ca7de1a93b38e708c56812f54c062a6c646c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327456514Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792, Name: checkpoints/1792, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1792?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=05fa16fc8be51893e5bb33eb0bf09aef8bc64dfadf1b7695c6c24f9ec88a7024b11963ab4ddedaa4291d89cb1f37ec3c264ccd46c31acc2cde1817bc995dadb87dacf1c75b9b7c0b18afe6ddd73acf583e6c8d8b20a09b6261a0b4a2ead7fe1258734dbd5f77e3b2b5ec29dc4487d814566e3e26a7b22f7fb47fc5055f7ac40d6b253af825fb13cd49d9a83fe616289d5029448970aed939a4a6ec54e851880d16e0bf0b3fed6024ffb04215e37cfdd8320d0a2b060ea425ec8f777adad8c87c3c8a1c07dc6a19775878853b4039fa00f76d008455338ef5e29e2e8715463e81674fcc301311cbae1ac83923065bc477e91811a08a826ddd20d8cd8692539a61&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32571912Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/array_metadatas, Name: checkpoints/3904/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3904/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b89531b585a6c631fbeb1b23c72d2ff70924e2ec059cbf303aab1c9f48bc7a991ebeb5501792c2736f28a71959345bef40f35776a43bcd5fdb2f41eb2c6dd0fc51d62e966ed5c0f841607ea84356e675310a036addfbd6b88db449b54f861d25d3cece979ce531ef73e7b9e974cf354f9d4e7025d6b653bb233836b492c1fd795174681a6411d62e35311ecd95b1cf00335e467b521633c704139fb4d913560721a5db4504bd316f249dc6e540cad22abadce7821fbf76469d05c86f1cf943fd9b79cb2a1f05618c4b2ed7ccc19ce2ed6edccd2bc7a9da7450e29721b6642c39edd39e1e8246480cacdecf101bb716ebdfa3d0b7ac295fb4f177acf58ab53126&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32789333Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default, Name: checkpoints/4160/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4160/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=0a2c797ef4abb715a81b70a6b4ce87fe044144ecd2f74880bf21f6eab81f2059a5700285c6edff5738ce09278b963b9288d283cbb44fc58662ad9261d78447742b8612ca32bd9e42062ddf45f8f6bbc7ba52abed008a11b3bf0d4b26f18b22e49513e11b1f1eaae11961d2d850b8e3d284eeb3e5a1f2662bce280f08a1616c84974695c924c84c501d6e51f70313fbfb70b0d0886212079652be72419ab64aadbf935bcc217fcce4129063d4e19bdedf5f23a0f316d70af4a11b4e3f3e3e1d2736fecc8ed5eded66ed61ea4819444502e9bc54c135ae0e953a3f4ca33266953c3cf544c3a3a704bc666f328b4f292df8ce9309728fef4e4650532b9a4a981f8f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327930932Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0, Name: checkpoints/1792/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1792/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=6a2558526805cb123793d2301f0b0cc370f224e7063bc1d7f25e7df430c36253c115214442928e8d75254227aa1b0fca13f6aff87ad33929f223daa6ac3b5448b587386157e4744b1e0424d3e7a63910ece1a423b293fc954bd8277af5805af3efc20663083433477f9f016d13f0aa30d17dc7d163304ba0d90a15240e174f21c4241f1bf877116436b936f141f499dd9d6ea516ab7d3bc72e9a993c24466d55db190885f741f8f3b1808ed34bf86d7877a21e4a6f92afcd4929f2646ad475cbb17842f1083078bd42af934b3a1b6dcab8159de69ce52e328c5630121decb1e283522e2c8d3669b27d2189f76d08280071db92f1063a3adb33cc885900a53b74&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327653019Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/array_metadatas, Name: checkpoints/64/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/64/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=5a501560f6216a9d3e5fd2a04c92a3e75042b143886a6165f176116608cdf7a11562e29445c3c4db682270341dade31f64905bd6a221e5dfd3e14169d3e86c87c8a36c34a55bf2ebec080024db7936e4d44c7a59cc9015e24eaa38b0367cbbd78bbf4c54797ff807a85661d773126bc7cb38d1daa6dacbc87e3d7687b10a1924c7c7b5c915eb48e44a0ef27f371d7188acceff92b713a50171bbc724d39a7caa8ef50d63d04607c2e46e5e88433c8a0bf597323bb7212cc5bd5e809f4d6774ff2dd9582ce8301b1678cc693b390d950680139512ed51ee2af0034f0e531136e31b75715baffe4a33c047f56b687e82783edaf47fee0c023aa97eb54d9f5f037f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327043626Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/array_metadatas, Name: checkpoints/1792/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1792/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=7cdebc9a4674107bdd1aa8effb4a26d09129488625c1fce6f8a69da6c15013302315c5a920d24714f24985c6a9b4f9834046cfce00f2739202a953a14cb4254a158c62ba1713cdc56f6fbbc550514c1151a84722808d29870eaf87a5451f84f94b14cd1422f9649ff90b87d58818d8327332a540af2b68febd01a1161a246f3ebcbe96c84f28264696b4600f74994cdc39173de0107776aefdf89742986c69828460e5266f7e606a8702fc47889a12c9c86b2d157eedbda08e36ae841965e2d38154f78c71193017a125bc372d805879e7732fff104023d238d655eac28865c914565b474ac5c2a930922e79da607fe05d802ffd225094da89c762aedfba2f42&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328014179Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default, Name: checkpoints/640/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/640/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=22c8b9bd109f75435fe2bb2dd3f0cac3c16a3c4a9269dc5f5000c4d4a5e630bf8ac75dfc2f1b0c0128a672e6b257e23346c7f71a4702f38d9c5ba3863043185c69dd9e154915b2733c8e4a6585068a6404e8583d42f359d586b2d647f0a9470cf3dfa724765debedb1315b3215d3859f3849ca04c0626b7da3ccf31146aa5edd5d9cd60db6d9828002f9c68be76d9bbb2364e6e1192468491f5f82a8d3280b88bdec7dafd6d5705aaf29804e9a0d691418517468ea9da13598acf51ce8c56231d288bea2f95cf2cb00679deceebe184952d687bb2ea49b9958d536fe3c27e329aa305cc6ded4ccfd4fd3aa8cddb0ad6629af28ac0fe284904e07767a2b14627c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328050447Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0, Name: checkpoints/3968/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3968/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=5d2307ef7a8e06b59b53a9d9a8e237dab2d93f187f9379722334d85d6ff763c72e734a85d1e21f01238cbef5c9aa54e685da8a49782c77bbe41f1160570078346b5e1c501399ba9815ba2176b1e5d919da33cc11af39962b6faf5660b65473073a399c7f50ad6a4013b48c1d627712add60c903df065a4e00fbf1bee86ead06828c2618d494703389e6fd14c586613fd742dade13607533087025678fbbb00883f8c3bfa086ffa4828c2e9f3236c00571d6a6752a20931de5d704f798f9a05d9ed675726e5376f2016d4d4d2c0d27369644c49ada63c6df641bcd41e6cef3c678e4ad9dfffd5c85be35f069173ff23fe22013a6c4fb8020b28dd30d72c037aa7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325232274Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080, Name: checkpoints/6080, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/6080?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=9a11342c39f282521c36fb9b94944460a58bb4fbadf4f7e31fb6ffbd3771a95f84f0c369d798be4209a10a2ae0c2c44d1e6d159eb5a3f84e143ecd01c47fa58ebe491405e14787f25c3e64cba42675a7902ce562a778d889a631d5801d998258ed6fffc75324d56e3227f19fd6a58ea2d6688a70f0b3d4d7e0219aec60d4f72b5476f4b2ba4ce0eedbed1fef6da5a2aa199c3503c518d7a630761782801dc8663bccc704bef45e24819ef56c8aab02f4bf29442835a2f7fafc3d4ffa31faf97d0a7f8b8cfdc963abefbeac7bf481633dddb843ae441e0fb492b252e1126aa17679a3cbe41cc648f1700cdddc461a102a925c8624aed88707cfa6371aa6e81b80&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325015176Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/d, Name: checkpoints/2688/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2688/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=34ee1b61acc09734a2b4e02fd6169b026a57241161961e19228b98896c151e23e2dc33f694c6efdfcdf711529946fa04d1e4568e6f06e63a8af8c6c0762f2117d50e0f5d8e4313e2313c2f5768e2af0d30fef69dba5ba2162caf33350f5b869ca3ba94e623e1c0eb2b456ec1d64afbd81c61de50700bbb28e4e54a3518c6571cbe8125df236cd4cb65df428c5cd26ae55c4e6b81facb05d2a86fca7028ac04484b928e29dfc3fa48114540e7fba60fb1fed2c4d49ef2f0e41c71211e5347755be6b94a54cb0931a68a31fc1143752d95e41aaca30e878209c375b94609d92f40566ed5d9c0c5fc60063218d7e94e1acdc91ee412a83fd91c4c5d83e67caa9433&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324760975Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/array_metadatas, Name: checkpoints/4096/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4096/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=04db37f2e4d21b09d71568125f96d8a5b727834f80d00f7d9e005d0559832c46499d934157064dd0f012af609fa757ac17f6d9b43420c21112c9ae6814dfd4ddfe9067dc67fea6703d23b1d55cbf59d5b5537b4a3c098ad1ab67c1eafdca35a7d29a7dfc136cff510be2463a79dfb7122a086c5dfb36558261179bf68012b18709ee23dcf9635c1161ccc4ea170277d9a5bff8a268a3052ddbd6eebe04c9f236b542bdaf9683927b670e22c9d5778a5fb316a85e676134e8bfc6748b3d2e0b6bb0a95e93108a9305a5f23f2900879fa9d73ed3536e6fa56c027ea806b8136459c4c35c676a90b7f8fcbea504d187fdb79269fa998111cecb25f9c969b59f6b16&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327910891Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d, Name: checkpoints/2176/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2176/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b673827b038cd9e229de68aed824679b4ac4f9e48a7ad6886694ce6f8c5ef9ef4889ff0edc3f401166df4678563f672e0ffa0f4bec480f4119dde058c29a2f8d206f6437aababedc8925d3b95021ab83e3cd8fce8c8d560740888c3fa30f5860783dc1d835ae008971ea9a426ca71dbd46bb7aef00bf820046e90e976c266a20ba41137e826dc918c6634f93760ae3e534e5bafc8d1eb11ffb6a6a7b3b9f4acdbda1ae4644c3bc36ac4190175921a2e9a97689f0de23650d0400531235f32ac2a203399601fa1930dd3d241146a92c72dc9090bc304e86314df2485bce15dbad7ba4c947c1512855195b31b2b418fd4e58855bec9f94ed6e7adfac041c3674ab&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328106414Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default, Name: checkpoints/4608/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4608/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=18b6ffe7c246f3dff59f819f4206f2e3cef68b6952aa106fc19679f970e47cb981ba133d4667f36f788ca1da637238e75c23907226161d6bd3a9e5fe4e6e3a927680bdffb9b2638b1f35e512daf05f9fe0cf8aaa37088332597a959adfcacdea2f21f593c6989bfa8890109b427808388152a25158456845134c16e57714ca537358c0fb5cb9b4d53c8b0a7844e5ceb9a252827e161f1b1226b7d65e3d5366221619fc2e2ab7adc6d51300a1df27f71f72f9087c30791f98b5c816e2a1867d6ea80499a811bae01383463af9bb62f933c2519b31d0b8c83767a375c10734246847afe8895c3f117ea92f9801d6b3751f21b51ea04ffa886956c694d3d9ea74e2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328143724Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default, Name: checkpoints/3648/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3648/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=89751fa7ff2e8ad06dc2b52ea538ab157dc62f2cfd30f6309cf02d7a6ae8c33d12c11cb7bc8c08b90d3e9f508db63280c80952f48081cae089ca4cad8a53823d946afebbc881c3234a6e993b18964f33576e8cd800c5bcaa39cc83ead5dcbdcfdd90bb5f7f2147cc119c735e2abc2d7d950453e296208463f30d4fba33a23f1695cdde13af8eab7e516167b55b5916ccb3db938e8a2a3419cc13f75472a322e037fb0e1538eacefb71a6d79eadf0c3754f3a31bd59800262ea27d009e244fe6f1185b0085aa2b88f4f1fd8cbb5b907875c5b83f59286bd652ab444ad91efb26d28be30333c9c98780bdd2a13ded66b4ea3d7d0314e394a8c3c29c17559858b8c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328170959Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d, Name: checkpoints/4416/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4416/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=6b6610440a2347a92c04e638f7fa5d1833e5544ad097a9016e41e37623c8a3935d2865de76cfce46f61bc799049a5b48c669675cd60144a022b7bb063ed21ae4fca57b1a71149ffe10c16652b6a2f396076424c278bb9eb5acbf6ac7cfd263d27ee69832f747b6a0eb9e6003b572707aa7bcd0923b50ac1d0b35fc74e4ad30122cbf326c31c927398ffa046055dd4e61eebd3fb607830c60562501a206b9c6f97f09e79da8a146609c7408d4b5575ad0455683ffd09c0edac2ab8d85e731cfa48f1ccc400af1c4a517748c4206be583cbd609f8af1ec2fc57739b491d4473ae462334bb4eae59acf542699e85ba2f8061e97952efc1ef85e59d18e400e79e795&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328113646Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0, Name: checkpoints/3840/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3840/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=201625daf5bace1a5fbb1ce2521aadd07ae3f970ddd42eb6404ddcfc185adae4c78305bb932f0a67b03e144c1c124efdcd390515349ddb499bc79058eb8d5079c09edadf800b073a197439fa142839a900fcc22188a6415191c5fa3081aec2e7d75c832d986d129ffc79f445fff2d2f408c7105cdf58a672e078cf01f4c064e4a57a08ca2240837f706e5ca4440ac51ee7f52b6cd38992a13903b0d1264cc04d54162f30fae9752dfdc27292e5699fe91599d41c59d2e3ea73d00e46258efbcad71f526d6fb4d9551e3de5eea5920a130bf3aaa93d57d4dd64ef9462a271211d1971bb70f8db54a4b77ad5dabbc3428d1210886cdf4920856a04fa91eb71deb2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324690074Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default, Name: checkpoints/5504/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5504/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=67a013bfef41597a5c81f605d7d4ef38af06b9a6d17b9fc759acff622f731f56dfff3881989de68a925af0144fdf11436bb8d0d468345119a1e29fa3312511508e4aece393f7339acaa24958d4645039a9ae6a57c25316b27374c68fa15bb4e03003d5003b53960390ea729d1c8b8b2425fe7de80d7863b9cec31f532bcec282fe680cdd5f8b50bc968ef6009d569b7bc255744d8772aab89b146d19d984664c6f8f67526dd1ab103a85f0965b7973bfab37d82496f4a101590de324c43c8890fa4a06c22595f3f110b81c9eea6c5c3ba739bba7dedc70dde0607e5389f5a2c2557a82f3e1d2a61cfe3119226649b07af2cc2ead3e36eb6e12102c0098950b9f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32823413Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default, Name: checkpoints/4224/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4224/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=7f9e6b9f70f412a0b0a6a67a8815c89fd21dc0ff5d31919707ced70ce154fe4146041902d3c06c0b4237e172929a79a77c2315e38df4ce90486cb58ed4f1b783c92a717379fe0b8625179b1b2a7dcbce817d1c26b4297cf35a759beda83e5b39c52c1a36e259794b013a01084e14b2bf66c9db2899917eb6432b788822594de5c1f28bc2c66b7e80a86aac99e3a51937093becac68eae67c98bf444c02a2c1c969092660f7c68dd92f6c78c1a291c03449c571bf33dbae37ffa88282443097ec016ab48e0694b51f728e8d9e14dc220625eb5b7169681655c0baa5f2dc14096ee40f9f5aa5b773a4404bd6c58c40a776b4256824aa40ac6c62b5a56be6514c99&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324865932Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default, Name: checkpoints/128/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/128/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=9a99ddacb7a2d844f939f08e808921052021158c799c3b98406301c63ad39806d0e2c133f20a43eab76afc383caf8b7059157c9c1ca05b5e57909306990f5e1aa43a3ee1a4b990d6ce9e822ba18a7aecf1c9c0cd18f52a6c23b113768c7f54f97934c14c05598980c1b95e4b56183891f982d3a9282ef6ac5a4f5e0e50714f20a57387a7871a729aaa20f7900594cb0a28f024b45b9a94e8f6ecbb8475689aeaeaf810990366a1aab4338704b5d82442b382d524617db3d1b3416f71c2118ec054c6d4e437e94b5395825f60ec941a94740803671b272a17f2c504f8fa07a592de108ed994897d43a4c6ed54dc8f3d4221d388e2d219fbee3093fc28d7567bc3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324801611Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default, Name: checkpoints/5952/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5952/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=735ad46516e6dd68eb1d73bddd19071bb602b620d76af880b12f5a049a911fe7ac1f7edc4e04990958d209575a1a511973e158b59204e67406adad272c0a02414403f60bbfa2cd3b4df662451ddd5b5ccf0c44f969ab08eb28a3dfd9fecf1ef7f66defa252ff10ccaf0da10b99b1b185fad88d4f091fa48fca377efa97b1e1900b64071dcea2d5b8bad48cd75b2ced6668c3bc217530ccfb2534c7763afc9cbdf398a39826a23076af0d5b71a4d9af22c9aacff78c785659f542e35832187226f48757f58d54eb4464ab7a5d8b7ee74241a628549d1c14a26da37e9eb11db1ebf29e8b12791c442acd8ec2cbbae12633bb5667cb8b7e47d7bc2b4a75ab96096a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324778038Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/d, Name: checkpoints/5312/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5312/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=bfe223de10ef0391fd68e3772ba1c5aea632cc6ac741f1bace226579b5e2e1be7e222916206d9ac1e1da5cd776fc7ba66be24c73d24a014eaf6c1b7aaa351ed2832717052542e7c5dbec7b7e5fe54eae4639dffd9af6656ad451f69a7c7c45654a68edf514e25eaeb1972d5757d8c51643e91e04c94ee10e63461e6b9ea9795cb963035ff89d17408b90d24a89edfc361d673e326d1343671c08743975c8255f4fa33b61e91049e7783884a01a717fb9b8b5843d92baa658239290c88b761ca4da6b655ff3805578d4a04699aff0707642cefea1e6c97aa663deaabe3ea1d351c595f05d07bbb18ff46bc15e726fd90ea28570c3bae60b61e98f40b5b1461f94&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32833165Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d, Name: checkpoints/1216/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1216/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=0cf7295706c49b61f241896a5ce41006557ca014ea537a5183712042dbf20a58042fbfdaf4b122619b7c8bc91d7ca6c30818f760bbb78422ce883c8a9053d1cece927dc1a5d2efb727e068af67e77983bed60d237cb75e8fd3cc67b2b119b37bdd715a2a38dd1d5fd0c5d926a4023ed9b529d167255b45ada59a8598ede3bfd53acece2c9518bdb485f904ee3fe81d3a9de861bda0f3af5ec61baebd03d83a896de5182d7d3573100f0cecc867237aaf932c678ab60b10e95bd6e226a79e68b89a464c8bd8d226e69eabf7179514596a96da985f1b98d83c741d01b70aa618c0866fdc1ef5a1b4909aa1e7dca1a41ed7a2693020ec97013a0fbdc1e74a93f306&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324737936Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/array_metadatas, Name: checkpoints/384/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/384/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=c0bb323218dc956a9fbe1b72dd57140e248c3597d84485043b777d8429c2774d7cde37a6c44f91c4a2eaaa6c12d0eacfdd1b8b3c1e49d217f631c2845816341e2c70405d39ce537734a44e95ddffa7b9d5cad3693ca35558d1c372d883b7a8e3191a6b4666d3ec3f274192c087592092cf1670874dcc29f8c02083aeb407cdebf03abc3df90f2270705ace6352b5ec5f68b1f5ecf08f5ea26e81b51899511b861fd507620c3954068b054e890dc13bede5cd56dd3ae60ee76859c87af1463c01bc165a08c79f4898d98d1204a5bd8f044eea7709584e59b05382119e230394a9e1b4805be0616cbd377bae5fd3cc621c262522f31546d4706233e6363e0b97dd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324610332Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480, Name: checkpoints/4480, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4480?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=0d874974fc750e48d6eb034b1236dd483e402a974842fda0465ec3e3eba29130262523d45ebfc925f64a31ac442ad4cf26ac51c0da6deecd84224f002d6d4642000e905b1a4532e0351229630e017baf32ce0403f969b10c7fcff75381eee76ad91c04bb601f2aa30893a7713a45009eec856636fc75019873ad703d354d0e9dabb4217550a3d2c485f8217fc1ee494852188419513c7b0d680a6bd55e2f03895fc8c8b20f102322558d63fd9dac69c402f6be92f34eb201d20e6d8b047153b44197f85a18b7e2684e666639db1d897bd07a299835f7883cafe803fef7c579dded69c899894b14beb2a8c7b4e9db755f52dbcac624010bee3db43e673e6b1f16&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328355636Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688, Name: checkpoints/2688, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2688?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=942a484a1d0f4954f327e87b4cc7d810789115387b5cc6024f6ace6e6ca80e88b0e90ccf8f077fba9d65d0319d0098b8141f392e71575ad8e5bc0507ddc7a64d759c37b19c9e6961b6244904285d8cbd9344aeb600eee99bad5ba0594b20aada2b707102e417b1091c2183832d1ba9696bc9e9ee5e648f0643c60f5a920ab61e977ac0523f4a9c3597c9f32a54aee0c16db4f2ad4e6df933faa2b2f23dd65fd8fa18c85ef860861f8ce490f651a97957591916aa720512b44871073924c0a303b9eb214ce8663a615d74d37176ee0f1575dfb0f5dfc8f56872a42fd76f59712a40de4c9041f12bc996aea0c88456fb76b0e94fcbdf525d121d787921f5f68182&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328404115Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/d, Name: checkpoints/5184/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5184/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8b7ca49808fa8d5ff3979643775b3622030c11dd99521801b8bb15a817cb76c36c1357c0560172cc09fc47ad70d093a22b47cd0401386277a6be680e09eedd8f37b2da9d3660001dfe61e467d7c15f987367d94619806ae6370b8d5b0f066ddaba75bcfc471a011dcaee3899fa718bee88c5213951d39d415a9de6d1358c875be2a464b3a70ec34f601366020d456f517b417140be9132ac743f249a429948f245f1b1e11f9a6aaf8cb6126c93e452ae3407cbef3dac39a2835b4c0a3843dd8171fc21e7d7fdfc16abd6e8d84fe96bfd1a7536ac6ca10a94132d6b56b1262d5cdb2cab6a6d700f19323a84c19e1cc93b9159334e96ecd12de4b03392f09ca3b3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328470313Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/array_metadatas, Name: checkpoints/1536/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1536/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=788c8c7e4490ec7491e357160e37244e6c6c505623ae67b3c78fc59f7fb11c6eae59245119b671d240955f48ffb5d331234f23196415c9f27b50aff9f462f57131bc38ca3cc87d48ff956abf79963d42aefbc7147bf077c3bc4f56d433ed393d03a1a8e1d752140f8c0f071fca44be2e912c71a62469613b2870a4db259eb345b7e53bd8334018f2d1629904d359fa8997d86c9c717f02feb53ba8073aaf0b7b4e970aa9d7b86231ee8e8c998a8e2a5e9482944071fec4bda0c8bcf6f9ed33dc5e4d10b34f467033f91ad21753039fb7fec68088d1968c85fce4c77f817960893f122371df56ee6700054fc3f8df66f690d581d3661e6b57efb96bed7c0a5958&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328471789Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/array_metadatas, Name: checkpoints/5248/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5248/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=657a697e4885b9b1bbcef55a52aa59c15205e1231fa63c7a24b7664714fdaed0f3512a5a67f9e1c333799d8c9c9628a63dd88d420b27ec9aa0e801be3fcaba76ebf2a1c0a7fb01dbcc81f035939e043f943408ae0e504290175aee06bb46c4b7acbf7be38074d2ac768ce1adf88b90401f077830319c2356557617ee31b884215f081710d161225b011e5db2efeb9bb037f762917035571eb209dae460da94fabc84f3642d84b1f97ab876f40a26f80a2f0c7ee1eea4059557291ca5002b722089f0e1543b1965fbc1bd56b7aeb821ced8c188764c38d387ba16065943cfedea34c02e369b9213a99836aa45e921e9967cba459e9b909d89bdfd7b8193344bc5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328418351Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/array_metadatas, Name: checkpoints/128/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/128/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=819dd91de6538c8a1b8f81e9babf187a2cee514501ac4de183b3ecf83f2a5478d4a057517ac2c2689e42432115ec51ff58720b6e63cbd5a9e6fc527832e70af41e5bb7a71128f7db4fbaa75d49dde24b302b5663bbf233de05e67b41f37f708ebed1fe94b1ef22a53dd1af05a8cf4fdbc0d60f648ba6280214d34180fdc271dc4b2c64c6abd16eef16dfe248093cc19d41a9648b3aba4247d7911b3ca42d0ccd71326b189b677e1b6a465cd1666e749ca0e245cab3e09150f00fe497cbacce87b56f270b0bc6bfcc6280e200ce56f99e89ae5ce5b8644e26a2609ed01f67f458543191f0f408cccb2a169ebbfff2253827dd68c10f38e2266a67836a3b8e4ce8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324436335Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152, Name: checkpoints/1152, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1152?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=bd43843a30829162eb66c52064dc7006f49cd486fb7b190685608d456315a94aaf26ed2b8465885aa9e94825dbccb71e73c7bba8cfebbc4ebf9bdf32a575360b8a933fd0c7fd123058543abef21b4f8fde39d9fb531109330ea2ea4afe43f73ef17f6cdf37a786df90473bfb573544c242fbb77501f5a9245f3a38edaca080744c710f54b4ac6fb01c3afb6539039e10bc166ebe5863ac66fe17b9f4ac3525576f46a93c9ade8b83cc4d2af9d67541aefa728f444790b982dead1bec16d22d2dbd64bc994d3f573107b91de6dbb495ecae5fc26f3cf242e6051bd452216f345e64df91e471346c137331a02febf7387cb50ec46bb946b8277474181e53ccb8b7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324277779Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440, Name: checkpoints/5440, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5440?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=9b0c39e16374eb03291e89ea56a435d6eaf191aa0cd07ec9f9b53b0a882fc7b82921775d901c88b1cf24cf81afed74b0f15bd279a31281038dbc9692c971fe81294935daadb26ecf6c0e0986553a2b468e97b41d39ebff59cbad65b626040dfbfb7df15774315bb3844e402f44ebedf0c5108dd2370b2e9a735060e275c8d4e357b4a8af17f9e9d6b6fa393d9dc7f3a16ead9e4e5fe49f288893ab9fd0d8ecf20620bd40a48789571a5196b02834036ab3b029f31ddcdcbeaf474acd59e679dbfbdaef88a60a5eb26da85959182bb3cbca68539f746027a3880e5ee5d3f37cf23581126378dacc6dbd489788af35e0a916e92314e168616fabe502beb6a3d356&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328605953Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0, Name: checkpoints/512/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/512/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=c243c396a4c14064505cbed17070f60de84ad46ba465d12d86875aa27ecdaa2635834b5cbb2db3d65ddc62a3cddf3db3dc6608b924d5d724d0b357643bd5947c77686695b15703ae626eadfebccdc963eb85eb48bceaedaaf3390e71ae82ef8e95f13c28f722945d11c4192efa137ccb67b1f23fdfff5deb5adbd8506b69a2b4ba14fdc582b12f2f16b0eb8c75f1921601ec979c2204430f19a149cb889736e61d753d1e0fb2be2b64ca0fd8c1c5c4433a85437f606e4f71f58cd15e004902e1fa2b348ef7b5b98403a5b96ce555bc74f9b34e1a245bf342a12ee5fb3e3f23fb121b27b81481dd9bc6efa750fd0a9a99138586bb67f2b8ced2e016c510a829ea&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324145756Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d, Name: checkpoints/3712/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3712/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1310df63dfb0255976d2eb518f0b4c0da670d3778f5f730fca36cb613a691f5608b1372285514fac8b114ca265b14a672eb702258972f25db3a268e7f058429b9cae6750b1088e58e3822de40d797118b45ad6f3172f2216d29c6a3a8eda86aac1cf747ff0b19b7de6be5fcb197b1a677139f4c21a03b0c7528d2ce85acdc987f60a9d875bf7b44798b85b17a9b0c3c501b2eb581524907c72a37fb765366fd0226e58b8d0672307c5cbc862dbf18aaf63465c06eadcf04dd308d352908401dd2e0a15355dc41e6688f8747bd43765f61b0214369bcdebad4300d49e49b89bdffd33ceda845e2505f39b0a2a278b368376c5a48b218be06b5136bb96f1a9e2e2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324552088Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264, Name: checkpoints/3264, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3264?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=482fe0ef805afb2157560df18553d66b7e077e1f984092898450be012b103ba217b79fdcc2f0f1d8f5841ab0bb200bdbf5c4d361ae1d0115037fa9d0ed6186d9fa164572d5a6aef6ec75b2f33ce68e829c9795fe168e2177091bddbc53391c7ed24837904501a5b125083a1643e4aac0d95c333e85441bc3a3fec7deed9c69ea2a4d55e144e6b29cc05fc95c67d4fdfe1ca9cce6aa3b9e3441f9863c859819f0b0877aa1699d69fc1d63c9b1319cc746ba8259126d83528b6614000190e51d70b798af166a88dfc889483ac5f9080388141ccc19e4d09b1167d5b2c2f6b6f01c9b633e6d24ef310a15be21a4f572ca6144212298bda0596bc3db35618e2ae5e8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328715088Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/d, Name: checkpoints/3904/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3904/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=80204c694d0f20f38b1518c296c8ce8bc1ed5557637987dcc319a7b3debae75d7f9e13f49ce2c5e69bb29970dff3635ad7c4bd1a6976f7cfa6affe6930996277a15176b2df263b9902823323017abad00cd6be0130dae93851894f30d6ef5a406c474f5a50c9e71d787595511abda007077641327d88033689d46533e7e230b3b3c84185189f3aa4013aa514fc34f9f62834e19869456ab96bec27663d124f22f1aed7d31bd54319a7367a8f047570e688f6054fba8aedc1cd3b2fcc89bbe5f79d45c4a841719af7d17e9d9625638e9cef6da388bc193751a680fafe34b41ec4b9371a1e50dbc2b44dd5b84a1a8d7468c1120e1ce91bcd358885be1009d340f5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328793425Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368, Name: checkpoints/2368, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2368?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=02f6fb464bd384a16390e0adbc0932a599d845376d3ebe8eba71c1c9b49cac65e15d12d0bb0992ec4c31faa5628e29ed9dbe77cff310cc11c4013864cdfcef1cedde00daf162aaf3ec42d7c93e13094a70e7939b5b5e671752cddbab8b95ae66635f7cbfd3200e3dfc96397b94190884bba500b317c8dfc2d460211d6776613b91ed259db3f355d811fa3a262e282c258c0508fce2c58896a58abbfbcd08ab193f15d8bdc0adecf17fa0eef2f0cf32dad011233186e24fea1c0fb5917f5c5e7a4774e1361f3c15d6a2e184780b118a8417c486508b07ccd5829972db4869e7a745d61473ce2384a18c9ff19b339f105ff7e87564dcb43f01091577c70facd762&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324088133Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default, Name: checkpoints/1472/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1472/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a855f1446494cbe44ae37ce32e374e36bda8a31f1179a02011fb0b7da54d3d60a43ce230e815d626890d50cc9a2154c95e65c4a90bcb95a94a5e96fb3cd548b765fc1f50ea040f39a8a6dc0a0b3fbe78d4dfbb4e487dfb0e55ac2bc0d78d9d005fd7e72c2364d8b663ddbb2c3ef961a9ed84fcf1c5f777ed1c76fa7962c4a5562c87262bf9771885333bb5a8f4aa34986628fe724ab754ebdf828e2685a689fcd5d02fac9788632ec759b8e5da8cc67f1975da2a59a0586dbd7d3ebb04c90be3e61a1946783eb5c869e7d518043ed5cfaf61478e1b2c47e2461023475ba88633cb08e6b558d399076bc67c367ad42cb6bdcbcbb6a51f20f8ca65570b1cae66f0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328833382Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0, Name: checkpoints/0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=14baf967e2e672ef23f6e49e0c9909f913ee13d00197c8a32dfa0de62c5739056a66927a4658d12735c083614030a56a524da75588b9742facc20080aff50fb43afb387ef43b20f7b12f72250e2f3a76c0c6af85eeda551a99757576955d4087738469be512fd1db4b8ca05d3bf1d2c11d887b1270e014b64b850ede91f5f74111061b964db644786124b12aaed98a786b264d183257692bddda4a1ea1c4df48d7325039728a560e3ac5c1ad3af1c6ca2d467d381ac260d2060c889d0feaee633092fba0447af6439584881b19625bc8c4699eec0536542d6e6409dfb615c9f88a3abe65f9bac5a4b13314f5f52ec44cd6ff4db4f916f9314e826c3c3a835cab&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328855805Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default, Name: checkpoints/3840/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3840/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2db60571f39324ad39ca386556c50404b8bd546ce523d28b60ff845c557fb84977812e53683d30aae80b69313458d2a6b439f33003e8d93ab6d1b23aa0a06218f5b1182ef0cf1cd8d361e03c58a89f20f7c493117c1ef8e64c54d87a19efdabd1ee9cb1c9c43b0ca1ed7b30205a3df1389c62142138ef4a080849afa2163db56f157a82b1055c404c0de8aec9910df0925aef5342cdff0b1d9719568153a809ebe1f772ac229d3de6052cb84474e2190b56260723749532a7cf93572b336d2fa6dbf391dfb8fdc101c569149e10c48b47c5e7fad3839bb7935fd0fd68db2ae2c85525e90e2f633d2cda2fb3f0ab12d3aeee0485e86c1562ee008513bc8c6156d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328898878Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default, Name: checkpoints/4544/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4544/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b23345a36d2ac4c0f307275312078688d6ef2971c3812719be72996197a934cdf38f6ee5d7c0ee6780b7a038e765114fb927612c947245e6fa09b041427005b43d3dbbc83fc798f310466162d7d3452a5d12cc4ae1591f7cb123a9da94117ba88b07721f25af8018a8b425ac52c275b9f041d58003c2f57301775dcde9326e2e901a82d1f4b7c95ff3f061a1cc935a2bd9128f0b80e01f3b9cc14ce70b9d737ef728b32ef1b299091f5f75a69975d17ab45b1c9d830b08af32642f6582b73c7506e1011ba99e9bdc0fb70b7b87345864ee25625ee3c68119eda1599380182ccb26d2cb1096dd7bb8fd8837cd288aac516a109e498ee2470e67c28cccb93098e9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328740224Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448, Name: checkpoints/448, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/448?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8828f7688854dd20541f993cc69affc20f49f48f2fddbbb294718624f5cbd1bf39c7c26dd52cc4cfb502bc4b280a0e970c887755f9041fefc0fc3904ab6674b6efad10777447d305e52f008a3faa2792d3887e96ec61d1526c94a0b53b2a4b3757932fdf6f174f6966bd39b6c3872ba7bc12674cf19775b1e274bc8e47c9c2367329397fcb71233e2463b06b854d75c45f87ced17caceff67bfd88d83505e0d2b6cd51682183dd47bb759170ae27d4fbd530e90a0320624d5ef78c30c583f44ffc71fa75300a15c79fd8886b5e96ef36cd9e86b13950d9dd295184b08ee63f42904965a877be83b75e0fc8c7cc7322615f95798be1e670eabf746cca4afcd2d4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328913226Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200, Name: checkpoints/3200, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3200?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=ae75c50c03d0f7adc65ca9defade3d2cdbf3cb166308b870271289ccf3fa9768d41c4d278efe5506eecc99d54ffd77ae8b56fb9ced506edbb7e7ebe40902a63258deaa0645fcf48d010c8efa2f2cad4b79c98f0c704c0ce3a1051b200dd7f491be384fe7b3f96546cb64bb8a7a874b6bf43ca6e28d93302446116e9a9de40aa2cc79222373cfc4e99d1b3d539b344a82533ea570dc4290fab1cf3e593d8b8ab691177f8b76792996bb247ded2788694b5cb80e5e82f45e71323e611654b27588cee8ccce688a273c6e5e12c238914d2bfa7db64ffcdb7ef54bf8bbf1bb28dc5255bdce225827e09fec85eb7d510e123f031db6a36880b46bca6ff2a36298b984&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324295593Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d, Name: checkpoints/1728/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1728/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=95ada8b0d56f5d57ca41c4b3fbca1e747449b0d9a5d12e49a5cd457100d1f23da74dd7e6377719aeb821bc4c215194e84176ff3fdcb95358171a378a711d8c88ad110a7e10c45c35f0e6e2afe5e14d8f5c0c83e6c5f7bce372ffc4800020955fafd84dea29990ac96a9159d8c31c98cc63fd062c439922d20e94b8836bfe1fdc811d392c6f534dc227e68a9831c2695f18e9cae7b6f801f9b401d9444a56a857253e511ea4e9b5040709bffe5f2af376eecda6e16fa6eeae04983669c2f3ee88e2030b6bb9eef188d200b9b84c2e97637c7d3915181ab5948bd9c1288d501321a54647d9e95201d31d1c844d415751dc95132c91876dec4d7df1fd487210aec5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.329031087Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432, Name: checkpoints/2432, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2432?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=547143888c52a1252d6af865bddee041cd5942d049d71269fe727be25ad277a5a39ea6aef53eaa5c3784f1f72e953c8422fd80e68e38664caad6d641afeee967c32a07f57d3e1779b59c0050ce1ce453a1098bccf91a955cf89dacc6297af7b4e49506bba4ff9f6bee24b1713b4564666c386753debf49dfee1bf66356dc7ad70e7412de9a3ee921122236d358a6e2022a0646ac537378f65620c0eaa0dd38287058543695c03d4abbcff32db33de1cb2c017459dc275b29217c77459834186d4f0ffac28c5666ca97170af0b8bb324c8e3b91dd46fbee9f1ef08484724c1e0e3717ff4d9505871659a9dafad2616fb66e96b359a17b6f21921ffb73ef6e928d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324275756Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/array_metadatas, Name: checkpoints/5312/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5312/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=7941531b252a54b701d276f364ca4a3bd7fd8d28eaf21caf236133331463c4261407d094328b64a7be66e282833923fe18a9877276c8a7653007c8bd2141ed395d47bc813106e112efdfd92b65c8085f11b15634896238e2e794eccd222bf0dccd13aed570f677b6375afb4085f78939e3b8fd8d3270f95575f4eaaa7bcdd3346901d98325d5ba019904dd81d2991517447a2336fa17032567d6b70e365a4991fad1ddcbff2b1115ccb00a462d4a32052e1733bfbcb169f968721772f755e60e2047718d68b1c694534d138ce6d90f11fbeda1bb9436f01700d8997752b857aaf78e9a179527807ab91e6ced4c71886140adcd596d5b5fc057a53e462a6a36d1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323924868Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0, Name: checkpoints/5952/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5952/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=9dc6f40134ab24604f4fb39187ecee4ae0a4139bcf6f7f73c88eab57721b63e4ef81b99f35b5054086b24bda3f8ea1c75e9f7b31fab9291aaedfacf67094025037862d57d749f84c27f948aa662f2748eacd4379bcca19ba9717144c555039867d4091771f53a895fa63fde2ca182713fd46fad29c1de56ab6c2d75fa1e87a547d44f99801308ed96138832bd8a2db7963c489379701407113d22b7464a50043082d4bbf88610a7999cc83a54ea27b483e9116966dc0fa7060ae6b585d3866a275990dcc7bb887e4acdc5d3b3f83277c9db3ac12d3d4868b5dac3ea4e9ae930efba326ca9becc031fd6de573c09d05b9afc30c775c23ed59dfaac3350502afc3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323735247Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d, Name: checkpoints/1600/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1600/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=36b16a1bfe83fedc7b961668dd5a8a6255e032d4033555ae10c083120a13d44215ce70b1f25fac782e81d590a899ec908538a885b77e5da9bce20e39cbf8a69b74e0d51d6b7b7bfde03ff9feccd4768bebca4143fd1b78075406aed779bc5a8098d18e4ef2aebe8146dc9998087ecbd20f1fb94890100f064e65583a321c27d9b4d09a7a31cb027b3782eb2866c6b8aa9e94858b86e3afc490a234886d0476a3a73c8668fae82a3e318c093a8c8d0fde496dd73ac6619428fc2b58dba8d74e05536968e6fb87440f906676ba71a66741c399fd13654284c473057295242ec2ced67fb7ce4232836279ddcab9ac1c2ed14897c0120b6e2096332185924a4d1eb2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323896826Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d, Name: checkpoints/256/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/256/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b2f607145b704ad9d9df43f0ce9359fc67c008dc0889b87d14b3d0697fc9911abe7b9006e72a7f1d0d73418f236e85046c1b622424e67c671938062df1b8f2a5fc5345272c6ae70c8986afd1ca18fad55c2ff584aefa8e1d1c05d57c7d5fa17da7cfee8c5eadd1f5fbbd34f62a324335318e1bc4162e963b932a1c791c693f359946bfe00808c48d59a721b43abaad2c5e9c511980f1a88d4cd43a24a8b02cad9aa1beaafe0ff6fdac3f50a25a16813b116b980cbf43490171e2ad69c8c219a91c408a1ab07e16c7d5618dfff5825a6ac1b29104634473aebe4843294b3c9098db1ad77d965fb2556b16bab87438ae7ad54951c1d151df60850d6ccb1bd9f284&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324051433Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120, Name: checkpoints/5120, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5120?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=59e856f8dc23f7c2dee2d4bf6815a939c882cbf67921da8c759650b6f68c18321bab2077818db900e761d3caf576a8c079c31a281d6f979fe89993049839a142b16974410edd435935fff7c62208967ad5370c6460cb7ed0588a179785a809742579b094570139903d7e13e12cd28e2bfb2b7ba8719c74e6826aed97c883075d87e648671929d4b6830ff43e5f5eb7300ee6166ebbbbef65fb76c63644c8c44ce4a9c8a79c38090af3aa1db59d596b90b6bf68f3eab32a999bd7be8611b9d209e77a7e31c1f0cbb7d4df4f78cdf6a62186b561e864f9ed5e67e85598d353ca372cc3055ed48a0e6a71e288e3c4e5222f825b5558c12fba0add8c2465c41fc292&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323991405Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default, Name: checkpoints/2688/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2688/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=71ee5e7910a5620f120927a61571b8e31503d89ef2cb774e6f434313dcb182ccb679761f704754a3c7bb3b455c77525ad01ea723419bf184bb1ac6e8eb378b6d658e7fb1c3b24caed8fbaee31294ad86831a7db86ec94e98269d9f34bc5bcae6951593bad4484ae365df1b60788f9526f536ea7a41e1de933179ca7948b0d046f72a91289f6378b31105a462af106a6d0e6a38dfb4e07a05994a7f43ec0d2a58f29e297d766efac2ccbe63d0b2cac70ab3e2f809db02188516e333779d0d0ae7a5af39a2b1b7328e0a9807de836c7b1817a728e4b254a55060de1b9b4c43a5a225188339ab530492f874811895841b352939bb749e92344f174bedcd847f27f9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323569673Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584, Name: checkpoints/3584, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3584?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=4be0890ad4d9c0c6658bcc55a443dfad5f0feec868983390b2ffd867f77f7040cda8b8682a7a7a54b2edce63489c824ce843ae7021fa903583d4a5fe7fca81c54f56dbd25c64f5bde610b7fc0eb7425aade1af67265323b00527423bd612a5d1a7b522a80b2d212618d8239acb4a7785851f12424d1c432b5e2b9eb3c0c9538f6aad7ca9efa41c7012c0963421822b3bac4686a81461a779a5b9ded50abc6cd7faadb1908cd2920ba3e53f2ac6482c2b583b132f6bbc6cd95f46dc985c502f56c6b70f855e31db7440c4524159e4d83103968e5890e45a3416ade72b13842e65c6cf7978c9f0badd4fd6d5c62652c02987dabff5610b8b958b40a94627379626&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323522081Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/array_metadatas, Name: checkpoints/4224/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4224/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a9c4e8e10560a309848e60f9ef4a2aa4eb7f9466a11970b23bea5074898f0502f617f3277b8df92a5ab38f5c0483d9df18984473fdc99e1ec9c7aa8eee084adffb5e7439a59cc95d85ccc9213d132c6749d5e324370bc9ef06761e8af527dd633eb5a88bf5e852679fbc426b9bd78d7169c2b8404e83bb810f874c6ae2a5d843a50282192921ee9cddb5afb7e7778140b05175aa55d6c6738a24a836ffe7acb9cb71fc6cae4ef57e21c44833eb2ff5981e910424cf1fcbfdbde93b8d9e0008148cc38c71c64e7ad48deb7ee3cd1505a18a474cbeb4d19aee6669bd1131faa9802d9199bf8d7cf28705f08f5071fbf48f5a36c4036fb62364eba9beb1bd441c90&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323806966Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default, Name: checkpoints/4416/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4416/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=0a4cf31af2673a7c2ee1b39bcf25a7938af257526b1d624c97c6b29e9d73351988ad55cca37cc6c30cf5b49cbdd247b1ab31632129101886a2b38ce8f66de07bd36fe867a8b67a3babc3f731476f729e1487ce8abe6083f84aca2a9210b07ccdee1b92b47bfe56e814ca863dfa0f2056e9c3768501cedfd3a657c0e137f8b351f23f467a8ae6bfe457c48ff9388799fd408cde453fcb436e2b8525b4e06d1f2ec60348f20d9dc205170d4ba9090ef2f85f7eaa3abe63d34ba4a6888949b2722a6618a844b566e4220204565bc9c1be577ac0467eea0f345f227b04d859aabe445a7e98d16a91c106b1993ec194decb3ab4cfaa456dcd68f117b09ad0fde3366b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323920364Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984, Name: checkpoints/1984, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1984?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a4081e4dc4c678acac4dec7761f022fa72c617cd1e6ebfe2f24506426e2367935d1cb8f7205486ccc6d254f5cc4f99f5fc7bd0d3af5bd6067706e649c511dd4786a99193e51ec225a4cac798b563d2c3990948a5684f0b47af4a7d8c0a14732806bce042b6401fd8f1acfaf5e8b58e2c6d6795e4e4e479f1e90f2ffd026ae7fbf112fed72b2dbff0e8a11b628f1099bc554f3ae7c758dc13c470cb3fa20e4a994a651686b6296d2a29a675d699d74f024bcf90d74ebd9a1863ec6be377847937ce90bcabd98ae36cc8a2c2aeb519a8c2f9c5a5559d29f1a3d84a8cf9ccf9a741edab1174c679a91d241f54cf51204dc4d6c23155010dade081b9cf8e8a357cb5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323925967Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0, Name: checkpoints/2816/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2816/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=c498299f5332c0dbb7169a0ccf3920a448ee2cd0b4093520a8efbc04bb4e2cca8b9fe8195e3c2686541792aa7a7f56765db5e64062ed665d1bc806d32de2f43a7d60a2d8cf77a5b2d3297ab2699ee3ac75442428b9b239caa5065376a7989bc3d71888918a82c9ece0509a639af5b9bd9d8486297bdda588e6fcdfee68e93fac9061b3982d0644bd49e4e74a727c30893f6e44597a45379b13914d576ad1190a3e2513f9f9d96041121d436429a6cd5b9a6b88d163229b6685456541c57456b52a4c18c3c0df4bfaab1ca727c771f5e5bde4099364549a2fdc06de889a0778f08ab1224117e40dfa80faab317c7059318daac1724a1f62925abd71317bc86cbe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323959526Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default, Name: checkpoints/4672/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4672/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=5426ca9aa851bd6a7edef30e556334adda0abb418b19ed3c3daa242d7adde312f27abf833613f415529e03bbf33a510eb35e409f1828cb0ffc1180148f8754da83c51ced1dfcc5ce93a531fc15f99e0c83dd333eab58c4159930bb53aef5dcf6b1bf3e1b74fe5b6b1c8dd3a8a123b95a61fdf32969a5cf316c89872101ab73d0f20b8aa3c9303d2f7dbecfe55d89819b0bf463df71fae925169c563edde5a6603d40bf503539b890ab35b81a2ab2822caa2c26a954fb4e13732814fcbe949b951d6256ff6322f2ce5c5af0f86ead396ac0138db22a10bc7272fbcef44dc9cfb81b1d9d6d350cb48f3ed53f22e171d810bd70ac3d196202b9589bf8afb33729b8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.323991246Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/d, Name: checkpoints/2048/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2048/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=351465b3be33c01b6bc6dba02154a0b2712dd4d7ec269b44eedd8f16aaa91ee1f3948f224c1a57bd3406a441f5f0ca2ef77617a50cd21930e509887e7f22a6a49123c0daee2e5c459c2dfe65948b7e7f97090e4a9d0b682f30d1785ac95b54c844ba6525606fb06ab1875fb6374d50a09d0424f743616dcc4aaea03451bed9b1c68559f2c5366a4215e18dfdaf78ce00ad5adecc9fb5354936b827a6e81c5b6f6f2cb5cf2810c5752d3ee1bb758d7ab1c92e01c25d7b7ebc3c2b4c053d1be5f891d08b140463b84be173216433b08a39a46252f4419c7c2cbc3c2fe6975f0af53b966e5ba64f3ec19db9b5c6e95dcf27a293b571274d24174ff40b6f601b57ed&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324052719Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d, Name: checkpoints/3776/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3776/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1cbc52cc54e6d7d9111fbef4093e9e8fbed9bc386a6f2865df9713224307d9ce902f1382c5f1ef179aa877b727d6f88a0a4f5a45b1555fc9b2d6f77c236909819d022d60b09291bc245ba19ea2497cc6e7b682627aebfb8331d640004a71588dd7ba42a5243aefa1e955f7757c3e6b24cd9c52424459281c839c736f61b9823438f5955382b61e3fdbc51a13f8359b7493a2d5a9263e32bebd82269284b15377d1d715d85f31cb015d7bb2ed8bf956af68d720de050ff299d0f053eb1d75539d11dd7dd586d3fa56c9e5176f1f2efe94be3edef52d087ca115d307f2f44528c92cb733aa132b7b0a9b2b6ae204a4137c189a7a1a56c598c3aa3ee74d952e50ed&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324059749Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/d, Name: checkpoints/6080/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/6080/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=7f89ed2cf86d9cd86eeba93fbe95c9c9ac4ec4e69c8a5715833b1a668739e5f665cf49da48f4062e3f0b614396cfa9697d2ec2e37d87263a121b965e8b8ccfdf8de80e5e232bf0138e042a8777a85532af1feacb0e64607a64f1f88e13f2d78e1e2d5167a7f942998aead130cc160c42d6265fd0231381b2b1994bbf33b7978a6c852ed81bad0f59d75984e7d62b1280098a3868b2a789348e51657f29efd5baeb10a16a9a8c409ac9c364ae48bff05608c5b6d0be2c4b80e8dfed908a117cbfcdd6996d10bd9f5813d4f1ac8bd16514241e2427a004775eb1d6f38e57e092f23b9dd7577b4328f45bfa3e2daa64c036a761b27276c0e320e3e98988a325b55a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324059482Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default, Name: checkpoints/5120/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5120/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=869cbd1866d6fc3952c16c88bee28f1f6c5a5cfceabd62e1b9f123e735c9f511765328e48f718b6508619ff93df6ca8b864772f1bb5dc1a42e0ee07a2eccdd75c7ffa533d1346e34b23367307a0f204b359c91eea7e70959277f6bde3d51fb4f5e1200d631706dab1d5c93acf937475e66d263f902eaba2ecf9367510d0539ea539c2651ab123fc0896747f68cb4c859453d76f7213650837b85d491c4dee504e194d3e7260a7aeded4a4ec9dee91133356f4ad7eddd7713514f0b2c54e84967b8728f207b8fb82f4b4d31e2a0b5aab317026c01e2984015a3e081638aca17411df25bd51efdbb277c5a95b0bd476c8868b0ff29ec42486d9ad8847b068ae70b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324059469Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d, Name: checkpoints/5504/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5504/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1e373a09ef7b9d8f39380aa136df5bd3600d4eb79457d40906433b7087e0ce743359190f1202cc6e79160c8d7230629081546dcee214ff45f87796522f5afe00d1f1939c89efdf1d9528ed253bdb4b605add12b6d33402d7327562a0e3080f28e8146148826861e5ea85a8c48ae780a31232329142ca3a3d36d3042f6f99392d3ceefaa8efa37ee634a8a56469c9dc1c1d336793341a73419a9c1a21fcfccb375683c9daa9def173f1b5e4fd2d1836d70e8ac060f2516e16b2571116c688813a450f243d7c43a92af29cb71a8892fac1f1c9b5c6d885f1fb0d79e418543345f73ccbf68505311dbe20c9ce448a2401d1ad0d89812bf28f3b21a18376b920828d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32412323Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/array_metadatas, Name: checkpoints/2816/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2816/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=501fd21654b298859a48dd2fc29b71cb1e1c61237d8a872f4ee47d3b6879f4e6a675547f947317e5ff3931ddc6f132c0367c6ecaf1e472669ae44cefaac3e5120732337efb2103d8d9eb056a704d5e4fe3a1d9e2f9dd2407193451a2c85e12c8084d1baa509e3029f108b456fb13b6dacbbf3295d203c8a058afd2d949e6e6986bc7dcc1342ace572f2b3045464c9ee2b9b5a290e305ac7b5eb541420fa89855c1efe25a8a9ad55b26d9209e02187ae2e8aeb3e14f1140475b6f14467a96421fe6688960dc57a42f417d1ab6e3e7a1f4777e03f8ffb8885d6d664ad02639dea98c96142bb0b114f5535f0efaebe80f975dad62f85a3398b5d45db8d5b71af7f0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324131308Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5504, Name: checkpoints/5504, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5504?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=260e47c17ca1897911d5555c8ab24bc6760186f7ee2f9094b00ec510445c4e218342f27829c3fff6a1d46e15519ccb60ac0093f7da25bc18d9a19d0a4c4cd00a6774a5f7b9f4c313e761eb5cdd6fe06f03f50d2b20782377629308b82e5d749f31db6c3ce54aba45eef77ab3a95efce58d74154833e0e813c67abcffbf23e8e59f43d75a78f21ade0945285b4886abe6ee7eb134a387dc7e3b4ce3d6702c41397016ee421ffde6bcb5f448484f5f63d800b9ff1c9f40740f00ff9f1523ec07946605eabebc9eb434f017b5d39b819754428b87da5c1634c34aed4f55de2a7137fff61b7131aee950e5bc55c0a65a6f822d2c797f4e312dfc30866ab8f4289972&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324166655Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/d, Name: checkpoints/4288/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4288/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=9ca64cf557c73bc6f7b57aa3d7bc5de91646526c1b0ddb5ebe63958473a81b3506381a61eb8121583d8b348608cd97633db35809c3e92bec0da0746ebd2ba531de69058937310fad193ade248af64833ba52d7740bdef1e7339b6600ab7fb6ea706551fd30dbae1fa27dccff1679449a68e5cd6a99ab5b199472c543a137e6baa22c71eab60ee0d692687d59171652f13acbe87165bbbadeac1e3aaa6466374a51199c541e6ea78b21b862d727cfff029be5724c48ac8280774d3773f88de7c2157826c16fbfc8811b1e1c80e032349d46e58a16b01ef86e543026f469f7e41fb8cc80285200e2cff047723f3d2208fe1a320ee3856b6738fe9981bed8951f93&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32417817Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d, Name: checkpoints/3456/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3456/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8d60815bd9af42716538e0ba8f8f9099fe032a6f143b0d2423d163adc823be0474d6e089475d5671f70d792c01d09eaf14ba6be7631aa45bf2fbced17729e92efc4fe908265960d1ed34252dcf26562ff11c0178951c88065cee5ebc7c6be8fe8728a8cf6331c370029328d1217b7cfcb14de77697c33b954b155b6d27ab9e463ca3af0229d92af11a8253391d09a17e7a89124934b23ed995dedf2a8f0c8206e60f5ee6215e8c8fee00a6fec90f0ad085fad73a8ebd7308b281765592b9dc76783a681ae4b98c5ca82ca19187149f0b27d773e55c3de1c385f6764646123f9c37a30f1eef3558c0e8385d8e1d39cd3657f8b8d4350597e0ae62fa0b75600613&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324193123Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0, Name: checkpoints/448/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/448/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2bd7c7b767561618a2cf81ce93c6985e738649c916cadeb10312c740d1827997231c501177117bbef41a34e00e98b75e3fcaf7f301c80b8935bb99661691ba502b4b2156b6aa33b6bdf1f43a2a74b81f9e60bc45ec2d594528dd22116776f389a8cabc5f77502c02ec3bd57d41ee0537a4218271122206690bc091db39582a0e96c6cedceed8532d7f677fa24f4d0f2f675a6229ae1bce876917efe6bfb4da58ba85ee8a9715eb7e2ff4b8e37fd0eaec2a9b94637f64acc370e0e31949b65c46b555a63ab5ee8b9b3cdae26b03ceac3014918f8361943cabd5141c3da232437e0817e6f8d014f6b8f8506a4368937231b51368b057d5dc6f7ed00cbfc620c6be&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324202693Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664, Name: checkpoints/1664, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1664?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=7bdf9f7c245752c63543ba23be317e167274617f210ea4fc5900691b6129268f53a08ff4d07237fb134cad0cfd44cd210966780787e1c06088368b128eed4df6bb3843d86d62e934273e182aeb090279b00330f670725497d87444358de0f4f4c5aff752546c06d42d45292da518cd96651cc77f07c75e5cd4966c1e0730a6ca42cf0f33aeff9a0fdd04cc608fb4c1ff72e35c68f420447ad7e4a9151fb3d81495a0b88738e660c74c171d66c2631856ca90ea84464e456626c68cd6faeaf7a33476a4e154e4984539029b938106e309d9a201b043e20b8d6e409cf102fc4b715e99b7275ed1d8cf7f5e60aec58fdef8dafa091bf63c0e134f9a288390b56d39&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32418133Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default, Name: checkpoints/4352/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4352/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=5e6b397217709dad609b0add9cbd569cc5baa4c79cfa002a2548cbaba51ea3ec2ddcfaa1b9cc7454fa3c2835f4be2e245e922906ae3e5058b94965a26a73d78dd8e8ce6059c22d4f5f8146f651ce13a4f7aada6c67f2c4aea5a6473443aa52b5bcc643552044a349631c0c7548682810c9bf10837c11d394f61b22c25dc1ca75df945b96c3c76a7a681bff92e2d9027bae18d93f6845f5f1668067974cad7a79517854dc3cb17b9dc950eaa5eaf1d98466aada3dc9987065bcb1970acef753497c2a121650662666330073006bf66e7bdd1181cea301a03c983165b674b7f7641abf341305ca589b9a0577211e1fa76bfc3130546bbdbfc34642c4924bceef8e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324200933Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d, Name: checkpoints/1152/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1152/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=08f5dfdcf961dbc33b90f374db04122a36fbe693efe27e4c32cc9988d0497623061cd35e173bcd3d6842fe47b200d5a20c528c33c304acd646b200de833ce2d41819474277c53d119a74c876d751822304601c4c9f6f59fc21dac78988b37b4c099f375ca062266edd9f1b9b59d9aa9d77f42f91b540653b33b4d192bf35a2b2add28ec9f1e238515afa359dc076b955f1a4ccda5bba021e2b43e4f14e3870748a9c5a1ae56b0e4b3ecddd4c0b8c5bbaee809b80536346f9ce04a542d71ec339f2a40f9ea89eb92fcad8f40227372680d3353725d7fa05a0771ccd8581ee5a44a6f5501a51fd1a355a88a47735a36e0765e4997333a5cac7736f96ef0c220b7b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324189528Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d, Name: checkpoints/1856/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1856/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b5f38362b19f06c425c0a88b504b0b019a05485fe940a7611537a367b9fcd97b34071e0b0fedbdeaa45e3979a674f935a627d46b6cf6d209720f06b9a6af411b157a9f87027a4a1d40f84f148133f9ec5aee84dbbc20f45be11595ffa046c7971c0438087fc14c2ce53e7666f1d2a5d9b32f4ea2e61cca993a058b85cfd6173274e7bd1dde69ee48e292489cd8a8f37965f389ffbfa12f9f1b6c9935c007be78f114c19c36545ed45ddd406731a0db357370c46779d5a54e10fb85ced89cfc64f3054f1ca83a3ebdf94bb7b6ab98c01002a59befcf1f90cbe3f0caec9130fdddc5ba3959526551461b75044018ea44a30d8100dc6df58da9930c25b2fe10d652&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324219479Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0, Name: checkpoints/896/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/896/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=420626a94526195d38eb7c872808867e7a72164adb5fd80b32dd3e9e03831ae79ff16c9c9b987b4994b37c7bc5e7e0d23ad9fb9ec795575a1fa0c10636578eae2b6c35853347f7e5cb4bd755148948a29eab5ca97cc0544e90e4fac63fd96e0b8b2b6f1a49a858e7268dde9f0d02ffe9020c74b7035a91b3189b92a34a033c427171d46866ccdbaa5734113446f5b540192df9e4e346755dde1290d38d33b286b9b7e618c30a257658344848b156fb77e114fcd6a325dad3154811f52d3f15139a6e6f852192d807f7965eeb133da466d9958a18c05cfe1e8f1ebd1ac020c0595b55ac482ef990f48c97eb3defb8cbc1523757d491e7fdf10a1c9e32d11efb0e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324246066Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0, Name: checkpoints/3136/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3136/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=106fc8b5f4b1617dbb5553b443a8a6a1c053d89cf4e76c28db8442eb66fb1ddf8326c07c6b859dfe9e5c978bf83610f5af5f7cf6cbb8cb867800c4b2ef30455481221e2f45a85650c16a68948b329cc83c865370dc3ba16c693449a62f151e555603b5752bfd9b73113859d42d2c9ee1af35f8bdf357ffe315f2c8a60b993c43e6e511affe4478b293fcea1ba6d5c1afbda67fd48c27ce632ec0a445ce72c2dde6aab48b7000b544dffedaed95a10852a590ec333535e7624a22df8ec0b92c7912ec804735fb50f4fd0813bede8e4e3abe11d08f0b5ccb3304f5179be5127c27f3312be57ed3c56072341f8ef15c586e030fa8161c45040527442b5b9c7d1e1d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32424413Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default, Name: checkpoints/4480/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4480/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=433dccb15844f9d0a23a41b8db132a869fb100273fa086cd30d0d0ce81c34db8d9a6ab4b53abc71210b5f7c6d67c74046376a1fef012347ffb562a621f81af4a1b9d2d7ae547f8041813fff7d06c815b158a5725d5bfc23e9fcf780d65a8ced46ea582324154b2b5af42c0c81c8cf5962c4ca178a34e692982f557c2c0be369e9cb2ab0ffae93c91b9a87248a96ca9f58dcdb7db7abf17630fcd2ed22a03562d2b8195a24387563b6d264afa25b8fed8cf5a0b030c7388be43ce42a1c336c31628274b7b444bbadf0813833d4052d5da6a69a6ef8e4beb91bd4cdd80355f67d1fbe9ad0d58a5c62c6c234cb3b3181861edc484461256d4746224643ac7b06ef2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324272608Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/array_metadatas, Name: checkpoints/832/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/832/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=87f3388a0949563c046a547c7674ee5da8f56d1762ce554b96ee8d5f4a6c021be2db6c15bbc7ec36c3f1fdb25939cc21132de17dce2020e3eeccf041372d159c7e1bc56ac399fe5648d575bf7b824c16cf048959691d5f144fabe2d38a169ddd35837625404a7774b22b63c7a90b7ddeb6739e8975c79e04a5c34d8a6a3b0fdbdfc79fba83fc47684696e4fb5415d6ab7059fca10e57450d36839bf7b528b7e37d731c6efdfa60b4de710df8a36403f142ac35eb216c15f3b41734a4ab3334a54c1bc0db7bb1577722196a4a6b72055b4049aaf2f27fce8916d8972b527f30372f4c443f5ed8f471aa56e46bfad56bd33b585a2c2d5aab0ce0de02666795d37a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324318186Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/array_metadatas, Name: checkpoints/1664/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1664/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=9313fc32321899558716e89579b494dafbae4a75fe50c0ed88cccd9c5d7904541685f17301d898fc8dac7f80890b1354b39514d63e9742cbc1b3d03e5f0a8a591dbf33dfaeba0258ddd852a0eb35bae094037d300cf313cbe846c3dd518e2fc46206c4d3c8562d682209bd0c6c7a75880ccbe55b5f6476edcdf8ef96930767785e8dcd9ebe5666647989ac579621c721937778d8afb3067949f56555b55d7a8f3ca6e5e477d86ebfc1d6bbaf2928b6abb3dca151d8a8f7c4b84a34f41d4bfd7df3f6f1056a0e9e7932de79c52d4e804f858fd4ae9481a27b9aee868e37a0139f9dc0bf3ba4fc5adb4cf714e445428bfd6d88b7a35225f9f6b57857fa178e4d16&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32430364Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d, Name: checkpoints/3840/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3840/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=5e377027ee6336f491423131a0f048aa5593c6155227c2f58e2f14a941c13ef9715104e1a0f9855d34c326f2419fa7c62ecf8f704a5c49c8df615c0d41c6356f876145d974337cdda8b0b52cb5875dff79d34a48e5c3f66ffa3065ceb5615ebda3fa40b1f09d88489bbe66e23c8b0b16152ca2e333116c4f18da0d9418ce130aaac0a4499d2ff3a39c6f66fc96522630ffd141f4eb6ea23891cbf87e81e530ee08cae0b3d8bb4201d9d50bc255b40ebea72f84a2bae3e2272bf027a1d3108e2a876cba46668da39ceaa77acd4c52a11cd1be84919c504a00bd2e0406c373e75baf3e650643a957f1634a293d0291500e37a7fc1211a073d669c8d7f668bcba32&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324340666Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d, Name: checkpoints/2944/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2944/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=70008383a278a39034a192d19cba33f294c96db963957c835ec046db6e7c73fa7aa2b372b10c19c35a650d0514f731646bc2657652df638b51d29328feca4223d36c57614c4ac1d79ece49992b9e0c85250067b3565ca2a5e29562be6d5a213f531a732114c832b3478c72404a3475cd914db7c59b68e3734d3b884a95cdc7b9a8fc06725cc8719bef2a9c0207b7f183b467c3ff8e8d833dbd34bdd3889782b9ea6c2849c629edab20ed9c3e5472e07f171ff1db1ae8e4c3274d75b299425464a73ed49d54b1bd31e7fc03a88e51272edb0984f870d4dd9bd4e03bd8fc233864ec619d842e6d60095825ac977ba920be48769175a44ebcb709339517ed87f332&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324334208Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d, Name: checkpoints/1792/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1792/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=691324f8518b12c25f2cc57ef2c7f61d9b88cc1657f4eec520b42f0817e3ccd43ac1e69e01957fe3fed89120161047a5c4dc0f89c934c1b74c908af394854b94e5fea695193afcc2007b4ee3a3b0b289e93d1511a0e3633fdebd2d59c5c58457c62b1ac18f5d8812a4478018ad0bc5ae2a638e5d08858166ba3c5960be0ccee8fdafee374c947a31b3a2cb12554e14aef7cb6da2003c1433cbe8052da02ca3a75a5d53763d290b330ee0da08e0f68c4a6bafbd11cc4893165ac80a42bec93e2518425996246eaa8349a4292a70892f805d4997695d7fff2632af436354dbebc9ac861705a2d7b47d679045f04b97464e86f6221f89902781b5dd27e740ad6580&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324354536Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/d, Name: checkpoints/128/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/128/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3e891d989168d8188d09e4de36130b8c37de2cc17bd30eb1ca906f663b13e2f335c075a0ab43cba66a907393e3c1a75f51d10360aa80c28d031e45b9727798a0cd0ca58ba2d3fd04e40f9ae4737f11a2be443b062284ce5bf63c0494666207cffd11b854ebc0551a3d75eeac44948e8c917beba0269d5eebf04179d5d25d8ac240fcdfba83f986d7000a2332f5c3be06e2384f07c14a526757b62eacb83ca62cfb9bb5ecd9b30a837b722cba348b39aa19dfcbe0767ff8e6b69f17a584067778a68479eabf430553eec869bd47baba1e00b80d37cb2aa4d925f9508018fb5a5966f1d7ee04141265e54965ec43117a296176485af6c56520d279f4728f2652c0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324371153Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/array_metadatas, Name: checkpoints/3712/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3712/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=19e6823348a3c5868572ca938979fb1c1aecc041f6d59584d66e38ab40f83005a60dd76043aae6cfd0c5aafd78a401325b763d15bf17d7300518bf1fc578d87e99ba008bb7151f85e4ef8731777a8407ab88c943af268b727be5a5cca56a08335e547b39080f2067d5f8b3ffbde465af68158f1cdc1f2258aeef439f8f171a0b760271ee1f2aa95d8bee5aa506f643e9338bee59ab409c2ced2aa2557cfa463fc1d7d2dc32052c4236b925a7621f504152d26ecfcb86af1bd6a8eca2305cca8b47d2e1fee6323f559cc5f98699183bfce8e8f18e6ea92f3373a8d93bb2cb5ad8954b769adce48f1a2e2b71b2164df74c6ac89fcb2e10010058d61599202abc31&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324358819Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0, Name: checkpoints/3776/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3776/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=428de74c4c87a6692de4538c1964638474a43d5b57effa353ce1a338296d67231ca49f5bbfff84762a9bfb8e9754d66af4df088137d5f04c1e01c55bc34841be0daa6b4fa8c907f62d8f607229b3501c9def4e4d4fa9c4bad921b9882223c895db40ce445e73b3b7181333abe8f5dd8430d9f15923a92d72b99d8da2cf431038dd030df49cc8a125ebf23e2e6dc18ce9b0d851d0ee24a2c65d26ebb52ddc77ce309b63d19e13643f13ff29962a313c9146265273912dae8586df0f354ff966a6f0af59893f8ae673d7b2c4ac7efc04178e5a40f19f8fe8e25cf98a219a4bcdfc2a7815706f5e2220b02277e03294c561e6b278e444e3e2826b55ca9fd90a610a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324349413Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d, Name: checkpoints/5952/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5952/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=7077c36200c13f428639ec935350739e31778e2d97c54e52d2710048906d11bbc6690442e7017e4203f7b778dbe7c3837aa3e5afc493a4fe5ab991167a6cedbd69c8495b2e497f0cb6e1912b792212ddeeb7fe34980e3a1d6091e1c0bc42c3fd58280788cf96a54801af5135ade379b679c64c21a0b23ef0e6516243b98403ca4179d99d0f406972bf2d5b8e25ec8fdd2dae8c97e798cae5d284546b8c8e22cf398e66c6dba8b401840f05371c517d220a19f6d8129b4d66f3d2f8d720a23086a498636ab579a2c348d9977defa5c3191b209118834d3acce67dbf3855e75fd14c98c1b7a366da644ae0de858a97c91c96f90db030f631c6228747f63bacbdef&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324369426Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600, Name: checkpoints/1600, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1600?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3b1938b3f91b97cc8d55115f50e407ba6e5ca3620f1649f2f817f917dd8ed6e4b8b89491487f462521db198abc7954adc63f0b22f40dbb4c308383b02ebc57ff48885d3d8c4790c254d8ee8434afb4673d21fb58aeb399e0413f4ac421bc3eb9d348ce150cae6ae16828bc2a9937a11dbe7e2db42c194fa295f0a78daf4357631dcba91320234ead40ae822d749ccdeaacb5b42c2943d7a6599a3dad05b41bfcf67ba47c229527ce91666c6a3cf225ce8ed1423944f95836b3551fbb26d053c5502baed835c18cdbc3039f481d18b83279aa115219e1faa84e2404efdf1e889ddecea3e836f61d5eeb568f741e86c791d9ecaf6d10b5b71af07c5f09b8bd4590&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324368814Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default, Name: checkpoints/3136/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3136/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=31f3d15f20257013114b7bbe6535100dbb5e2eb552ffa37781bc8126f6bb815c306e5e276856daab26f5c26b622ed917b81d822c2ba119c575fa64f2999c41390a6daf264b563d41ca2bcb4a3b74950e2263681e72f8dd25f1cc4684aaad03aa07ffae0c8918cee1bb09f785f58285aa55ae925f8100b497b51c1ce6b614fab6c7ae8ede8c7778fd0f9698ab435f0be1277760bc83a257f8ee31c687217e68738a2e843b2fbd282b5de409e8241a5f494e80b820e6ef874d01f13018221eb745f9a9d016d5fe4b47b9e477ff716c1b504255c47e758bb96ac2729dfce905cc7ed6c11c535080a5266c9b3e2d8eafcb52cf0f197ca0243ed24007e33873491b49&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324363323Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/d, Name: checkpoints/3328/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3328/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b51d9d9baa1f202123f952ee8ddad5b788cdaed2f309280e52e14e508b0ae1533b4bf3912cb3660044bed83d8730eeb59f6c728095a36b829db30407e1fe791feda6e12d384bdccedf652991a7f58df844f1f3c42f17ab42674448991a36d5077ecd97376c72bf6c00fd68189aebb9d25b880d4f1a5c7043418940521f468637d1c1ee618df4ad0252bb441e26a173897e8298a5ce73252fb57c7cbd74510c78f90e23c5dbb2c373b7de6527cd1e8edd45281bbb206e4605e4ef0f77bb932213d46981b913b1a1d2247721dc4396552600116b133d91895e7b92dc31255c7ced63507c8e229b8694587017ff296cada6f4ce555d9247253f2a115df605648af3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324384018Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d, Name: checkpoints/4224/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4224/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2b34c33838c48163ab5ff84103e4a619a1c8b50d7409b8611f07a9a78e2205453a564d34f50374163513adfba0f76c2adadbbdb6cb7717cac46e5a06152808f99c493d279339e5c59913762d07eaf3741243e7b66cf341fb1517d9c79178e24a5fa01680cdee03b785ba329fc9d07727c540484343a307d6cb6829bc2554009a4e56554a1abfed56df538dca014d13c834c989f2ab0723cdce27247ff2316dadb538de1f6e1b260dd159c52dd892e07a86e239c7f8e19cb7f65797db43e8b7b484952aaa851b9f027706d650e48af1f492d93ec6cfcfc82853974132b6c82eab01f46efd2e20cf5fac5e111d1011e9831bb64c79973118ab7da9efcc19ebb6b7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324401794Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/d, Name: checkpoints/1152/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1152/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2fe252ca5d1ca8fb3e503b7f9021f7e47a9917fb4072d34445cf390db4af43225484b1d6dc55937545f93fb9dd4d0499577ea310a6508a9021f5e2aeed27e49beabd34a6366532ae94563a22a5c4fd641c914eab342c3f22fb164dc3195360ef697b51ba9acb3e14cd040cfd382fafa9abfb84ca64ab7797b018710ff16183ba97f2ab7b9236a5e68b4246aeb3cab35501f5c67b2263cfc49fd2aeef93d40072297678f75fdb3c175c2dcd084eedc8279e47904d4006e616ed0b941deb5fcfd2a153d28e69a8048604e9c43bb4bdafc0c5bfe18c5b50d4e81c19d41048184472c7cd907b347378f07ccd55a2ba5b7fc9b44bccbe9dd6256fb510a5b85777c824&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324400437Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d, Name: checkpoints/5056/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5056/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=104587a053709a8d8eb952c7d6aa75b7872a1cec8514d35b3ead7d7be93f888404e9a8f8908187d1f73e794fd79222786a2b4f658817f25e2f47d663c00db3c88c845ee118c78a9351ed60935f60b5cfc761e3460b4a3a9687b5189e38959b0faf323d80e6697d419453fa4d73eaa235be008b634f2d7e5d034b7e895d41a8a28cf97514f0eac626bf93966d26a8e20449389a2060e380218e7e0eafe606e7c197df021070bf03b80b0436fcf5a37cc63e97a6e646ff6ed06d0469e72f61b80d2d73d011b0e2652ebd9dad24dc13835f706ab68b2e1bc602c436fa3684585deb5e543f4104237d70b5fb6f26eb7a094bcbf1a557705a4c708571f5f7f8d9936b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324399533Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/d, Name: checkpoints/5248/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5248/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=6c12eada5cc62d1569c56241851b3e2d833dfd838058cc0f4dbcdbad9bb6516802bebb2a171dd7c89a9cc27fcc631ce2cbb5dd4380874b4d74a232ca86304f84a0109e4a7062f0d737d6b23d4719453a870c07b084d40d70becc649e2cb8201b9a448e257f3f30fc80f6a065a97bad92f89ad23dd9503173e002e6fd2babd4fd30b148f6a6f01908de95eb0e3273345c66e86b3914ae6f89bb5a31717a445da2c1542f590b1ab8ce1b0deda3cb620b96d304463cfc837e402ef04942ce5ff9abf28484be60c9c817d70cffbb9ff4eb059b711cc3386d7a97238331eb12b868f01014719f705ef9357eb8321d62a2c4e0741ce6fa6303c560d7350235f73827f4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324418893Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d, Name: checkpoints/5312/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5312/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=39feaba3be5c23f5d3642d3a27909326923962fa0d2551892e661462d03a3a3b657750175466663c65ddc9ab07f55897c7f4c1377fd9eebf4cc6c1569b44aabd1a12e3b9c132dc0f5e9a893ac08a84a02cbb124ca9e62fb822fdaa88a70f644d76465eed62d129a212d1e2a5f6873b6ef0a5950b884f9453e2bc4370648b86e971c3fb640d34f1fbdc0df6592f2b6fd151f4542ae732565bcf003506186cb23e13ded07f9b6e39594a10903a1e6ba1cf681fa43d3d4d3f2c38d63256549377798f33b3faa2a960a376baaf53580a3c908a7b486cde5438871a434e29498b1e820796888b30c3ccdf66c6177bb21249c005968e80410f55170e810f5bd396d28b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324435397Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032, Name: checkpoints/4032, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4032?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=97446048e99a8778cfa149740e237f512e57b214c56feaea7512968cee4cc858ce6e9209e559e2288076e8cab14c6dfaa64994a0abcb66ea050d3828747b53897b64c82d765fd1893100c3c37a60b984919fce0d972ca748c7667ff6fd79e7301bf9492423087c3c704dee63d4470eaf17452ad95059da57ba873377ff7b63ba08fc334edf4e85d812cd416f02e17e675835f8c9a1d94403a26299adfc3d670e123e26bbced766b286aefa1745aac0b9301483fce97c2d208a151894bc4b51d30b5978f8e7aa07c9aaef854de538422a768e4060f32a47ba2555cebd4095aceca729dbf9066be5454b678c2063358b3c7557ca29e6a7309eec2be5f3819c94c9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324428914Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/array_metadatas, Name: checkpoints/3648/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3648/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=63c006e3415ba6ed6043e560d52eb685fa5d602f7f4a49292025f75851b7b9dc7da168958c459fa81fcf5d3640ee73247642ba2a415c69c60f841a8a6f07c48e1a657f5a04b1ae2548ad28fe99895ae5764bda44e543a412cd5ed55ecb1d0627964c15747766b848408d9dc67a729b2f3fa98f232267cd727a86695c3861278a7336bcb4de1f66e67bdd578fe40a2010658933698a478ac0357e5f18b7537dbbb395f8002c77b37281e6d7095f9b17ff3662cb9a232f36ac019926451988295e5b868803d2f17df62152afa31bca2ebf7dcf972460973e1a6ec659377f2902ba050c78c26bc29547660df006a1f521a7d65a72195a2e9acda6d04ec313892d4f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324439132Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default, Name: checkpoints/2432/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2432/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=62ef80b8854239ad5bf0ad8690d4d78e61f00984977fd1165e0f2af0d65c4bf628bc3becf3727a2fd684a706c5641e0d010b2115cc1be849d49525c8f7dffb3deb275ae58afa918f54d28a564086cd278942f6fc20833c3636c1291acb8f841eafe65d8dcc2f357646b4bb0eddf7bcbbc7dae979ef1c404462cac4021b6cdfd23857658f7d93398f8b4a659f3cd686c79ece128f75bf1a95a4c8fa28aa521bb858f05f2e6db95d981a5ffe5edbe506bc3da91070b80fcccdafaf91c117bf50c5874441e1ebd28a0f02c2667638744d16b1674c6cb4e98af2ddfc86d7f3fd5180d14964fb5f87c53fddc4c116b9e19c9cb73db010fd6a4ec0e58c912a5958dfc9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324457704Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d, Name: checkpoints/5760/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5760/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=4fed4760289315c3e230cd7127c6a1dacd114b777bc48a6501a92e8596e20a2aa2fa0cfda4ae474621136601da38537368d1dab0d771fda443fa2f36cd4602c623709de407768ec346d83fcd6b07f6e7d90f38a3e3567e2fb8d9a4d3e61756a5e0cfab894861b3d694a4a54ab2fb611bb2b98ede5ff3ff410bf7e67ba45c62f1a1ee62c61798841a4eea3b419df7c81170af6c339ce2c9862c5ad86290c9555d62a6aa88b05ed28a88dcc9931f76d8d419f36a05e48610114a61ac834af9ff758dcc006c2aceaf990ea1d7562a2d63d2b95d09e57a2bf848381854ff90a8443b313caec8400a4ce18cee373223a83d649009d7c0d1f0ce1bc96d17c37d52486d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324447233Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0, Name: checkpoints/1664/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1664/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=871965856ca8ed2f888348e5cc5cfddc366ce8b8891b19ce9f132e0ec7adf4f166903d5f839eb1a170abf7e40b7690bd2aec20fe3e14215a09e684897b1ebbde66fa5d615bd5a473bc28913141b14ac2433dbd87118c499a2ce986b9a02f5d53ad3a3a78a1f320a1ca4c708fdf5c030ae8483946bfe4296057da4adda0ccb2bd301241aa4d0bc8dd169c56fb90ca3394d0d662ebc8155e4c3d9e058a2fdb032652cf858fbd01fbe316ef6cafa88f86cc39be82da3bd9d7b30412dc622841b216cfa18691abdc6c523d02dbaa6b37a92711dec8ef3fe8309d5d375a835709fb89742f554ac22e4733934eadc2a3a46f3f2c2af267e78b38ca1fb2e1345c205f7f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324463459Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d, Name: checkpoints/2560/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2560/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b9e7c108d8a8edccc645c55052019a97cf55cdd659dbbef278b9b59bff02a8c3e1077b325132297a27d506e20c85036dc00b6358f2140c156af9fac83a032a7f8006a4d8879c430b5a3243f7181314ce8c2d47ce3b1422499a2785d9d6b1f2df35fcbe44af1acecfe07af7c68778114220e4d319e3a0ff291dffff4069a8cddd06a55082ee2a85f39311639667a8a54a590228d7411300fbe15fab050de4f5247bd9929251ce43ead4f47542977fec3c9727afcafc2cd59cbb8fff1bd6b936f82e3256eab832a7c6991924e337a6f31e7a2889d8a475342a33ef7aae730653e364c9ed5e115ed8a7381b46685c348ce2fc51ecc904e1cac775801e3442b8bf99&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32448085Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default, Name: checkpoints/3392/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3392/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=514bd7d8eb59dd35faccbda4c703421132b9101a4a050971ae3885489721eca015165a999bb8bf8fb35720e98e1761854436fd5f9ae8c85a8a25d445f99b64b1c61a0798524b98a1579f5035b28f7a5a3f3e2ee8555b31bb3b5723567e719a6f4903bba34c3683cbbc5e47272637756cadc15fdcd7587b103e4a00ea72abf5c8247b3d443c8cc16bd28236c0e86dccccf862daf49430a959c88c8ead2ba5b427a7ec9b959bad2f3ca5b9db7d47081337fd92aabf37f78803eca3795473b82272d4f28f5f1f7101fdb73d1baeed98f37f242cb6a72b7313ba165894d67a05c0dc57feb1b4b403b8ea0a948d544ca75bcfac58f9f57557efd996d13169f6e489fb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324481431Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d, Name: checkpoints/3648/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3648/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=91f310a67f188407786e69b73b6e91f78632abc54742807c257cb19931c1b03c778a73f02a10caa7135c2ab9ac589d46ed7fffd7e54685e523603a7c873973a2aca667db3f681eb5604c5f1744bcd30dcb4266255a13331c01b6c6b91ba2c291e6bc1e0f3eea90aee116e9f0c65d67ff5754a19f254ef847efe1b3296a9344372b34c01545f5a2e21e3bff96568686fc8a64779a2415b0afd0546639a13d91a48f12130a957ab04b8d7eebd437d1ef94cfae403f33df4beea2ba3f4e701911fdea01db47e697d1ec1024bb315b65f6d84134c57975284b33feccacded1f9af1507daa99cb47d358db9bf31dba9c8abbfceb4d5bdaffbe6c8d20f43d0139ef21f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324496021Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default, Name: checkpoints/2880/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2880/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=195bf14330234b1da137dbe7fb4d85bd0080433ab915e2ac51e4e68c385dff9c0461039fae6f255b7c0cf878afadc21f1083c7f5172c0924a555c70b64784f5948359a362a0829db7a9c71560a963c56a95f6e2488da323d68a5a363ad41f5c745d7569b359c0b099ac51265d4f9aaf9d2f990cfd172c4f729ea62c7e043cb24069ba59d343b1889ad28267b5cc74402589b162a9c83011fa6e7c8a20c0ef0d26a02cbd891f678e0c2c24749eae86bf65e0e8550de1a2df184b5b1027d630b82d4a628f02a3d075efba032c306db2bbf1385bc7f9a9a5a90d0f46c7592f0b51d1aa427d21b3d655659ffb5502dd63ca0718a93dd9ef57d407bb53667429cef5a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324522258Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/array_metadatas, Name: checkpoints/1216/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1216/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=79641effbbca9412793348bb02545828b67628e31a69802e8a880ca0087ca6d7f879c347e4c97c127918314d27f511913950cc59fede5c37a63539e6ab1e7672c2db9fe4673a8d109e6a46603b1f4f74d6fc7d4b6edf5bb5646a9f6fa459879bce1b170853a567f493c9f9a95704fd713fa69a6a947fe5e5906e2d64f94073714dd16f72101b3ba64d3de4ab53b0e154e902611ed4117d27c67b46e6ab169f45c916fa60cd7e02bbf7d34e09b161eb694eb64d2ed4e5dc3c366a7be01a7c838c4c25669653c1ec5d395da6b684dc75dbcd9297c852e232b20701680277ba1e414d78b3c2685868cf73d75fae68c2cc0e36d75b30446bd0856327d10bbd794374&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324522188Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/d, Name: checkpoints/2496/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2496/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=48593825950c86924cc241b2327ad012b4adeb3f1dd5c7cc08ec5362ebf3415f74aef04960534199bc8e3ed97f5d92e0fb1c3f9cad476adef610ab59ba4d0104955fbe497431570bf203dcd326f0a1a62115e69d8ed7fca9b1457b30b4ea9bf1707e61411f53c18127fa46a65e753702ed831bdd5be8935a4997ccbb309d9c7d0161dc3ba7bf053901a01b262a4e62ee2769b9070d0e18fdaafee8f1557a76e05370177cb9f0fce1f914364897f19b8587e3b06c9119d8a643b77d788dc976e34dae240de5262a34c51807f04fc646b915350d37a7b7a647fb7c3a7eb530e2675abf5b9111ded5a9abf406de348b39726dde3a49434c0498d4eeaf56ed40a968&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324512111Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/d, Name: checkpoints/1408/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1408/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=85d187c8ca3c84b8b80a43d66c921ebf294b895baaed9799c234a282ecaed02bf98ecedc88dd3af7c8e69577fe6fe60c553d80be41169be0ecf477f820bb2815ad8fcc18e3066bfaf3649e706703116409ef37db00b60ab44d869f3cc4521b3598fdad93e94cafb6b4715bdb7f6e5eb514abb2208070d64f3d683aa6388ffdc4c66a90543b05cfc50a95967b3dac5d7eb984bdf45ce010b099cdbea8435c1c1fe6bda1d46c0a37573d4e32893b548b2570b6099a070cb076d08811a1ec8a836c242cac35926fec310a90555b68a15df8c5695864409a1efe84ada8c15f47354a8ec21d324ebb1e3e5108bdc646b2331e0edb3554712e60c4720a1c7354a9ec63&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32454452Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/d, Name: checkpoints/3712/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3712/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3992d5a4570f0d356452a0e1bc7cf3711d8a0816659d645f9ac2eecf5e3d06e4cc3d4601157c72426fb40d3796b981d72dbb5b1e7ef41e6110d8534fa7c4f85112304448f4838e87a3798fb2dd0408cdce24d760ccae10544757dd9597a407907862ce9a1f84c71212044aeb1b946ecb0eff9581c33581be4d1a764e3823954cae2d950b2e3960f73c5dfee85efb0846ea411832c36574b17594b06682579de273283e4f6a90aabf9dfa95a5fd4a3a4819c4ef3b8deb967c70506718ae6de64e4a7ed3036d7d17c65d537bb4c8a8d6b175a637a1f5a93113ebef96ae328f0af38acea8f385ef664d0feb10d8f93b98581f5a535b8e41c34ba684692400386fe7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324551747Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d, Name: checkpoints/2048/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2048/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=c266e96971f4ddcde7f86c3cadda600a502d9350c63bbda29c4bf7e04fcabb7f935f03aa155f0f2c765c7378641e08cbea803a2138e07c773998ae1ee5b11330e7a75af5b6ba3373057dd36ff17e12db2ad2916a754bcfa33eadb8a9acb0ba05c66b51ab2cd050aea80ebc1f3d1e7e7bf9989ff2160f912261069f9a2528b860b4e30ceba7722e980d6f26a4c45015d8ec3e760438341b2c7949ee5c7b131b8016aa41d6b83f538b85542c68962bde9f00aaa7755f8f26107dfeadd2d388100b1c247654d0ab5d22a595d471e88e783772ec9d552d30b1a9c8a3f470e04bc034b74779ab2b68f2740bd44c2d7f5c7103c4762d2617b72720bb0f19fd894d766a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324560369Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d, Name: checkpoints/384/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/384/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=6743a3e6ba4d7477c7fd66afc9e6eb37fe3467536a6f6222011d3af3f97aef48e9ad165ffe5df0bc00787593c93830e95cc01804af765f95146adf41304f0b8f14c6363b3ccf97ff97eae791bccdcedf25cbb9c0f75b7313bf26cc9877eaec12deba62d188097a1fb204eaedd9ad3c5c77e2e09bb47d7332cdd51e02609aa2169f3851f4d08f05811e09e6c4cfb2449257ef6d65daaae772dc35e4d15333e59576dcbf4dd619a6b7acf57f2eb60fd9c7426d56537bd473b5f68daaadaa97eb411a210c3afc4109e1d1bd81013d2161e4dc9881be0304510a602cb260169c7c087b31a9c12a6e2903786d867cb796e6ac8595141ce2e8f0e93443bb2618487a50&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324578554Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0, Name: checkpoints/3200/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3200/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=c18134deadee8ac974e464eaa8d889c3a118afc5c5a07d2a32e99d415b5a13aca73d858d2f6dec1baec9538defd33a65d11ba95af67cb1fffa7eee79bd469063f0fcd632756d6505ea4d7b44521f877003f437aa46aeea52c76e935b0aedfd69a62dcf385c9db7f566df36b535b06dd2efb580cc1cf8e249adef2b0d5ff61cd221e38e37689d317021f55ba9a4f8e0a06c3e27d81d6169fac538c42af73f452dbbfdb9778fdf271d01ab9206a04fbd3d20146b571fb6a0d7ef89d9ecd831453eae18ea7116da407ff75c96118e12725f1077772e7662d2044016486a26a6b530f0e51c330905554c13b56836717d45fea98b15be34fadbab5511d0ed7809f5c6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324580545Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/array_metadatas, Name: checkpoints/3200/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3200/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=4a58280fc2c6d4e7265b0794af37426bcb09fddec23d66bdd12a7e45fce6bd3351217615661b3429db721d5b61710a5f6bd84d306fdba5e9ee4b248549c68910594c568a7f00728b8cc05c00c95b9f2e284d2bfe9faf3d2a9d84d380d4a4d2058e8c920cc9830c0954a1c4a87ca85cb8de33b4685689bc777b45a29b8b1fdb87642e83e09d22305cc3dd97f658b18ffe8d85f0216ebc1f65a5acd8202cbd2a318e3427f16ca55f0e43f5f99850002102d2a31ab841deb22d9fe092eb83d011f0bec0e844c20da034822eeea1433af4e6866856707b8ae93d82e3f6d446af1dc538edb5fccbecd74e89011856f911c93e33276ed0f454d0ac5160796deeaa0c81&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324584583Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968, Name: checkpoints/3968, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3968?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=05e37ab1628cfc8253f76377184c9509ca1c429563b3f4ced8efe6a22c9b53e9677f049994a89c1eb55ae9f7d4c9c171f221a211ab8b61b228de7e82d34320100a7c1404557dbf20ff3c1e170a6133dcce02307ef402014d398e0ae939f25a8b3ba56bc2e04dbdd120c60b2432beb7e94a5e4f14374853539f02217c1e493deeca8f69e2d46e201d2344dc5397dec0208fd72d143564af8809f585c9bfc8fdfe27993d401bcce2b3a1afca7e4bebbcd6ffb620a51463d088989b1c0eb3d9ec3a6a7b8513607fe9342e98ac23fa51948c4497c862d1ad743d82107f722697602c9bf9886b861f637f00a3719b5d68b36b18cbc38385ba5880dc9c4df931dc7ee4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324592793Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728, Name: checkpoints/1728, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1728?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=6f86767da0bb4245a223ca6c0eceee864fe864d2478e0da4f8d3981fe6c9b3f044def9ea98eb405924e97446adcb475e996d27226cccae21727caedab3a003a896fb16fffa13fe18aaaf115a9ee1f38a3e60bf83f2364db43d5e8dfa43dbabe0b595722ee08c33ae1011fc0025f3f2188aab560262af79d669088a7ffac426b6e494124fdfe42cd6be95c1aad436dfb4d613e6d40c0acc2f5c019e5e94a58cb930f7c38d5d65942683483e14e334879a3afc9801a1ab936a4b85f86ee274aea2026ef70d07da426aa5be470ab95be76ba59e7c813012cd83338224bd3fbc73f1095521845e5ad127dc1588a1d4bc2bfcf24d63ed89b96225d46ec8d1f5215290&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.3245898Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0, Name: checkpoints/1216/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1216/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=550a2d56908b1595bcb997a78b6ddebebec0448dc1a91e0509302b209eb362e4ddc3008a14309c8f556517b1652097a91b676d2f1c1a7d472610c21e86d0d95075726fad8f6e4efcbe1459d5f5209f1556c7e6667235d5c611e816c9af8d9f380d42cb80dcc1a4002ffae7dd6ea59fd669041a2061c739edf411c0fb834f571619054e3d52922e693bcb39c95f244b7c2128305c5b5a0e64897787b1086da723a2ec02a1cdfa776517e68b2c8db5a0a483f764f1cd67fbd952e41649d97e75fdc371f50b3577375bb4c6b19bdacf144fcf89143007b6307af590024b096eee9ce2d3a84a1cd7763b2de646e34d257e1066662c7bf827cb0eb71ff366a9e8b507&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324596924Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/d, Name: checkpoints/3392/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3392/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=114859d499ade47f36aea82dcb4b32640fe97fbbc4d5dd48cc086dc577196639808d48dc462db3eca8b1e15d40d2fe84fe060a7b50f4feff93442fd6f678fadb33b7b9b4bcfbae013f378a3c103fe9560eb12938d6412a946be87315a0c927c5f43f7e5da41104aee4e1983ffb98c8fbaf6daf7005cb6f96f513481f9c2d48e2954fb124d3b9f12b97331f41284bd8c23f39b3f2cd55b9211eb6ce80b24e6bd1ed89ece9e41027c977fd1f2944e22fad5855e956dde02fc82990afe7f8e2d20059b814e4d043025a06702a825ef5033564355b8c8b98ff91c175ff92233cfb8bd7533e1103cbdd8a6d993f0819e484d76807d6ae8a50e42976aff7f1ec29eb61&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324575313Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/array_metadatas, Name: checkpoints/640/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/640/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=9a23513a4371c4a7e1936e5b41a4628643906ae349b44690523d7460484c1ed07d034074b005171c2b64b98860f2f9a767de355c4178bd8b352d4030234b46c86ddb0c5fc8fd15cc801050aa06f96e68578367eba3bca68434a035387ea719c3d859b5cd8d7623868b5797c43143661cd369c3057f70d574fe360a85c9e605a0e682f4cd6c813ec9407c90fabf622ce1019f35c65117b2c148017a5a0b74ca2776a1301c26d922889dce7a90c696e169490dec50274f8e7916c204bf2686c9c9ee2eae51034c0e48a139b1d09c7b15f14c56fc40caf980139969f825f179f942a232ce9c8e45939991109290ae0bd0ee501b5a7cca928d14875b8b08ae6c5fc1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32459507Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default/d, Name: checkpoints/1664/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1664/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=4662ba8cad76a0b5f43fce8b018c7fcec965f1354b767901d6d6c05d98fec8d7de3a932cee27aa19c21ebae7e46a5e603378484f552dc72256728dfa125adb10ebfd8b99357ef67820396e4fdacf1e5bedac6fb85dd020d0820e8ea1c33515a6da927fe823d6d9f03d38891a461728489b9d881ebf5807a531e8c8024a34de626e2e6dcd7a5cc08d512a8f4f5d5954ebbbf424f9eddb784d01f37cd5045972366ebe9fe8f4201fdbf7b42fa08bdcf6d38a6970649f744497b69b7c07a4a95f2457ca68d7f57ed5e2ab1d41583263a43d4ad0442c2258d1a25655f8cebc8582c7ba6cff1ad4209304f71122ae6fed0048e3cdeb69b0f47292fcefc0c7ae697b41&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324602097Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/array_metadatas, Name: checkpoints/5568/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5568/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=12b8da6c87ae505b47f5918078cc3e55691a97ae8ddfd8e97fbd106582eca72f2e64359a2a56c50a9eaa03dbec9492a1c88a6a2a18bcaf58156bfd14a7ed8d54eb771d148cadade6ad520780b2cead1ed6dd51b02c20c3399f3b0809c13032727baf0d448d01757bfcb962f2e55e54a33d95caf5783d991b328607b0cb1abefbb9c7a0c6ee0aa4ba958c04fa3b1e88f5d2a172ae10cad10fb51e811cfc2b50e179be10d99bca0dfc7437b733fcb94837cb151ac1361663fe187db71cf68f7ba3ea54a292de5cdb927f1d185cb0608d5f68ee540c07d64da2d3abb772870cf953c749b3458f66b7834c13ffc748d3a0e2a9269f783cc5f9f7baf9e3006f0e8d0c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324605372Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default, Name: checkpoints/5824/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5824/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=15a5f2b9b14cff3def78856dfdb95f5aba65eab4a3c99b67bfc3e515de9f2305d8f5a7518eb54f8ea30670e2f4e63bd4d574ac6bf334c1fa5e8a4e26d3ee4ccf539a18a25e18ef64fc51c7c7f5278198e28e23b823045de1cfc56fa679e4f834bf321d76decb1a8770bca7f2b5d850cf0eddeb495ee061e17053efeb813d94ba36c9766a9d0bc27b0464ff9ce064dee8cff0b64e36d22101f8668d4f89feb671cdd7eb6b673aa8c1a0e5ed073677223226f0f1a1316cb3b438bb278fa45dfe3adcc7d8a39a35d9f317f163fd005eb3712fcc079a0cc494c606f7a38c0a611ef7bdcec4baf3558d644066317fb15d03f7869faf020bd4a9c6dfa5f0d3cc2ef274&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324623052Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d, Name: checkpoints/5376/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5376/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3308cebb91667d868e8d5dcd3f1691a81ed4aec0a9e86e0eba5bfa1c32b37f1460b4279995e637d1878bb34ee2750485526e02c3082a54847da764c95be2fbae4cc0a771d36acfd69a996895f8ecf0caef8e6c3c71c3d99231c0fc0f494f2f5b131ec92ef612bea0525303fc6b035d3c3617c600616a22730f14bd881a65630384fd8c77c78b787571a843005b12158333b78c960004ff06915b7846d9ce53a1667e6621349695434885d5eaa6db8c960ebf4c5939a8bfe0ea00cf6afc14816413fbec0812c9654d3f987a228b17865cab84c004c00083541124c86b8424745def27f49a9f2425f589664865e52bb00bf7bb60afe9483616ee4584387fce173e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324621248Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672, Name: checkpoints/4672, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4672?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2e2f5f4057a9662ad7a81a3b4c843a37cc9d4f4921c88169cb32e6799e360d70fc11e59a138fca86b4189f9431cf755c42b9f8c9494b3c8c38a322c2ec4e80b2587895d6eeec84916f486bf01dfad71dd49bfa89a1839d8ab49dc06634d8e6f5c50aea4e8c91a856d34ff9acc4da284e9e9254f12ac6828a1e7e1567e2bdef394bffb73e103ed05b9f3fb0410044f06d89170b75a1b8e69953aaf672007916e9a2e810709770ba49edb16492903c9b8dfc867f3c13fc0b2a9802cd31874c898850cc5deee1c93e2375123ae1ab93bbb45396d5e537d5f043057fb0d6a803482b6c04a53e136b9a496431996fbe97aae84a98105443ab64fa7a0ff8e895f4ad08&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324631726Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/d, Name: checkpoints/256/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/256/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2c1be396170ce23640f72a0f01e32fa02ed58eafba39ef59228a2c9f8b0b596f61da9732ea0a882aa5bd9d4cc907bc757b11c8e2faa220e162c74853047bc67d61c9f26b1d800ccdfd4f4dfb9afc1269ecb7fad41dcacf1b704ed4209b821ad7af8d2be0c1e75728f06c5d4b615a8db082f3ee5c5b10f844d49c499541ed1dc961745b0b7f7e3bb3da191d5d7e6237d6d44db9f3327e2143986a02a188e19c18d41ee3559e2a91ff83e73ecfa4f9a765b84010a3e67749688c8b7d1d78434fbbd329b88d02bb329749cedeacce2d7364c20377ed4d97710314572548d39264894f20fc26f8679355b19a0acede7242f18ba4cb0b74f5d07c9b4d5c6724d78097&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324661839Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default/d, Name: checkpoints/3776/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3776/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=6dceb8cc6a90a335bd9ee7bb32ca60f093d610f4e0188a92eaecfb67fee5c810d661159306146487d5056c883e5843ad6534b9e5597b80af784da8d85928d53fd28a6bc38a6546176125ff361cfe1bacf0e36c7d7be3d407c76cbc436fd4dbd5cb91bd6e0277176c058beefbc81db6bc9cbf1a238401e6b6f30598f73c0f95a96cc579a6915e6f912f08e4e7090e9a0d14623334eb8fd42728fb6a81662787811e414fac7639917ef4892a42365a63c0b5c7eacd31e4191ef8b542fd2a48a84367545e7fb230349278e974a51bb85c9491dee4050245b977c00a04f0f355343f58fdd00a16a017b1adc7fdde8bff5eb7bf12829d0a3526686bca952c006c85dd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324647594Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default, Name: checkpoints/3712/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3712/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2a44e64976579d5a08c0cb51133f822c276bb32c57b4e21495c68cefa8ea25df960c9f95d20a953fca632a51f138d4e07ed0713978d0a468d3555a5613c41b9707a25d57c47879e8af7cf582a59cd2b22e3d43361be3853458ad9bc5a408eae6e540abe23c28a176cf42491903e9171cc9f83e8a176a37552a2ae3d08c8cc9df5840b6caaffe7b1ec21d844497ade279c1240a0c2e56e258af8a075e1ecaa50c5a8ae29da1bda1f4d73c8e2dd055bd921de9d6c2e6378d23b90afd7fa9ee5c58a444ce82af51851bfa3646b90b23a8cd7283c8d5495a110d9d3f786692568cbd07f05df5150d039c74bebd45b85b8e211354dbe9e19642f75932701e62f4896f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324653932Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/array_metadatas, Name: checkpoints/5056/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5056/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=6c090da6521e1ab1738a1aa8a5030fa643bf16eb8c0688eed4f754338c275ee5f3b5c2afe780c9c19dc124ab4112d24796196264f7915a849691aec8a6d0f40cea2ab51fc967897da7779c5ba4977f87b028335850888ec2580e140ac8a6494154e8be0bd5e12758a71030f9cf4fda62792240f31980827a20ecd9a09f4aa711c31ec27eb809f57fc4033b9dce1af977561c89ff3490982de9b113870a704e57dea1bd5f5afce7c0c447282aeba8532a274e9971a74f6b0a024a67957c422ec993c79ed81dda409d5fa16da0acd8b40bff0b59e2f48d7f959fc3d9f34e273499b1659ec21c9fd7b0138bb7ba0c6bf046903c62db0d5f28b8475f728a0fcac5bd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331226602Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default, Name: checkpoints/576/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/576/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=88d8314308c86b862ca6d88c8db4b5dab705861c6d0aee000299367d1ea29c2a1a964dcb9fd408a7c121966093140962ce34beb42fd1e4944c57d59ccf4cd904d7c85029b7cef132ca4936b2bb2813211826bb3e2577c27392ba1e04a01765e645062e0f8566a9bb6f73fe9411f62d80d8bf329360b058c77b8c964099f540c7901488249245cfc826b98a706f890c7335c08e84bfd6aabe6277267721a065fa386f93b075b683cc86e64c0cb1f37e5f01bd061f3ffd15dbdde6c63dee9c96531a40a22c81b88b063402e77b3b9d28d3c2a8505440f8379a1303b7fb3775d9e660868fc5d406935d72f3c34aeae2e5a5a1895113ade8540ca28845c866c78f08&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324652646Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0, Name: checkpoints/1152/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1152/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=84f454f29f041d76c23b664aa6bc9c1d8e8d76b9992efa45952ceb958ecaa65599209a1b6bb46534c6b7eca3364b7170a9aabcbadc9132002884abd5020e9cbf93f449af0b1ea7b5a75bbe963738446e770b0368f3ae6c6e1cf3fb707792e9f745e130cc50fde7fdcde1ac08990120e0a28220abf33a70a46d44b395f9802a2466e0f4f9d377e0bb7b77ef071c74d27d584e3ef76c118b78f4c87ae9bb2e4aa602a29e678f75ee382f15a531b3dad1698600c7fad8b26c01afe51c59e111c97086a9cd20cfdcf5a99301e22591afc18fcca090a1e434ddfe0d0754d276fdc325d674a199c9d9b7ca17c95ec6503e530d368bd307606d9c73fc68b44b7c556b98&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331256192Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/array_metadatas, Name: checkpoints/896/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/896/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=93d6393708375d4380b529694c4e2ea33c7d5410193f76e3ff31c536ad08484ef99d93ba6123dab875b50671b85d654748a19151589d4d4ea7ca3b52e65ef85a4aa7a15bebe6c5cf1a40da3572676d34caa390b66268908a4ed0f33c98f585ceca8beff948588877fbc7c0072e63458fa78b5088f15a0c02876a1642053e4cef6b617a0ce3bf6d750c2f1a2c3d61b23fb1330c93d4139b0b08fd6040b713f06c088d521935cc6a4905e3734c5e880ff897039097f18df9fa0af462060fa82eb3496ef592f85d7ecf6602ecef259f1509033f5f028a3c9e7b2d4d37d855b3cc8b839505abff846fb2ebc30842308a427bedacaf6f2d55f55152b1ef1923badedc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331276184Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default, Name: checkpoints/1408/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1408/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=7da4090e02e140fbf372cb7ec2055eee7d645bbc9ff37473c9da04ee357bcbe8f6d1ae764381a879cd6441326b7658c5110d71055d9ce72b44eeb7e522bc7668099e5aa632c7c000429e0a3a4bc6a66c2aa1f2f73ee1e9266e0fc95566fc6154ce77f9696890963baf7e61a5f14b8097335a9cc83b7e50387035aefffec6a578f3a518dbbed7f98117ec76dff62b670f2e13f6530d8e8dc8b7a95915f9042c0023ddb09269d9da1e25d281527b9b53fbba240f1136e0bd9f9172e0aa6988e6d9db0f72d4d981ac77b29399e99e0c6e7f4d6d646286dc54e3e0637dbfa900f738db2975c38874e2d9e0a2154f0e87bcd8428fdb59f37c94f004a7b9fc970c7378&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331311224Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0, Name: checkpoints/1408/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1408/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a41a51d1a42eb15d5fc60d6ff94c38c92e3a0baf0b1bd0cb4a2387d39fd211251ddeca364c3435ac4c5362fbce39abea2acc85a5750bd695ba3ee37e500f959180075a669aa99e28a37a142297986f66c5f987dc933a18400c1628b3a75ce1c680c67795d60c57962d04b2c2f33235232b86f17cb6b59e3bed126b9df25119280d152634fe3d98f5101f45c1b5c002d6115836637a798502d6954a4c27aff52ae302c23c88d2fe06189004c435a6d71b0a09c09e405a56461e10ce28e40096fb326590b6e2796415ecc447ef853ae6282bcd12d6c49dadbb3354e692caf2e244e8aa7ed593f14ce5f512f717e26241d16fbe635bfb71dcdf6598a50dea9e33b1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331339411Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/d, Name: checkpoints/3264/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3264/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1ab8b72d014471ee0defc48c59f292720b00834af06743700ca09064891367aee099107994008b799b8d829b80e29ce6b940ba6001cb68e392f693d75b09f839e524f4c93a31d7d8b22ff3083d78a6627a7038c7a563bb6ca1e8b1e29ca1cd0b4196bd1e085df13f375fedba7261ad7282c494ddc7c4e025ad0a84531ae12923ecddca54d54acfd78f6c98397cadb7cf915fc9ae3b24699c1bc460e85d25d56d7f811c0fea732c81df21553777788e0b35d9a7c98b4ed3726dfca90013768dbb724647660177f5e49d01198b9c26821ae0ee1847a65cd6439cf3e97aec6ebb0b56a48126a90e2a3ce55f7683659abb7f1e929548d50df431045dc1ce599a5177&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331253012Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5952, Name: checkpoints/5952, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5952?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=af3e915e4d79ba3ac7338b3d8574c6a3686d52f80280156535b95e7ec2a3238bd4be889dacf63d3898d6f5862dda7918759a98e73d917e40585e45d03cf629e7b95ef8e7cc600a08afc34d47c3f02a28e0edd928ad14b5e70c517b2289baddba4ba9580cc10fd3e3c9620f9f1635aa1edece1f81ecd17b430876fee3d32908d3f7a22c41992f8531c0dac5a52b9a0da03f0096e836fcd4c984126a916d73d1151d206a3e72f1e6575bc579e7a7f9c3711bcd88f43bc2b5d7d9c38ec9c34358fb54115690f485fa95f38bcf89428c167e070729d71cb32289489c68501aebf751bd9c155e9962baba03dfb6322760288b05cf1cd82b092672eb52201afd99f5c4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324661426Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0, Name: checkpoints/5568/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5568/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b6f631c821d329b8fa9e454dbda09e00a782425a44e862087485aa92b971c212560ca4cc42a0c2bf9929d0c68e24c946a7fafd4aa98d5d1311aa7bda76fe7e66640a98a3f689f7abee2359bb8ad07b942bc106d70ec7b032500c801d65987e3123d35a90ed52faf473e9061fdb57675cf0530af8190d529046694f09776eea23f5208307cd2e8d0771509fdda2c56aa9b5adf841f85b64e5d7ddf651bed7f56902951f9edd654adc896f2c5d593b458ee1fbf3f942620916ab761aec3de9a8659eb64d5a72b0a7dcc31138e45a9d201f15875b27758ff6a24d403c7fcf705615e4ce05373ccd27eae009ad82c416377ebad78a6e0da3cdf5143d1952c7873f0d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331430472Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960, Name: checkpoints/960, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/960?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=ab90bdef91f2fd520eeea9959aa6f00d2fb624715a503da434e7b9331ce5d8bd778088c3e0cf198a435b5abb739e8a3dd10f630aca7ce6d295bc173297600e33067ece1dc6eff981b548222002878931bad9790b8f4aae75acd8c33f3162ed925e08a99ff5ed5cd740e37fbad680fd26e258f5272ac85fd5177ce5693c9edd277130ba2bf3ea8a6ded738c187e3de9712feed004134705b9ae456af917138dca63497b58628f222519404320aeca93125df4f810e8418c81428d9909ad125ee6aa0cba820024cda94109a4ced54f3b1fc1b3a811c486978fdd906da9f0f60ddb0f6653aa799311613183929f43206b34c824e4e6134bd573d9cb0478ef39ae84&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331432823Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/d, Name: checkpoints/512/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/512/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=01b67d4faf71fc50bd008b678fd68305cea44966af84df6d137543333d1b576a94f6c8be07015f96266b73dad97531999df4a980f23475c9de3614d549ce72100090959536766d0865f7d19c49a6ca060f92feea955f0026ff5617f164bc8253eb34a50fe13d23c3fa27a05e4b7ad5eb9d933334ef850a09c91062b3c94a137690d947a6b4d1bfb86c25d9e73e1011019b94b5fe3ece419a04cb8107c574d5b99f7f58477d52670321bb506850cf37f7f243bc06d47909d97e22ffc1d9fa2344f4992e47c31e32d87459bc919d51cc7241ac4329749bf84aab3e098628421ee4127a2e762a85f2e8d757007e284c7a1ab65e8159f410421b3ff7980dc5416c16&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331470471Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248, Name: checkpoints/5248, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5248?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=88365fe1c1e255f53fce3ebaa97cbd4de8b92a7f06926a7076992b388084f8f6b060f26d7aa807ffb52497034a0fa52f685e6d6f10c0bbb34d4bbb8a9e2add8d3779f4ab5d418d1af353f8cfa8563b1e3b5b3566bc98ff2427fae9cad4f0a266b2d4270783567bd55a66d12aafea193a6a37560901fb2df6d4d1a3614098035018a0dbe6d523ceeae3484b90f31eaca5029c587feb2870f276bd2f0a89a7e9a98a013bdb594347f04d3603ff2455cbc733bdc98081ef5f750cd0e153276105bce17c92a98e0fd6ec8e2174b05a8a7df8c69b0c6ed94ac0b9d3d392a8f6dc45f14d535c5ca84ba5331617b733efe9870f180c6066e474146dfba068e3281cfe4c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331496843Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/array_metadatas, Name: checkpoints/3328/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3328/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=bc109e362c25536186e3733208bb85d730c651a76fedc19329df30284b06a06cb78fd653cb60f9cdc8f877884ca223b9f034a25675e7483343465541bd4265ff006bed25b09f378870b4d80e348e24fec1b5f8c02a18f32bdfb752907932e061d4ae84bbbfef7bc48168504ac6bfb166e4bf5df62d9a974119142eb74d42052f8f5e3a7aedf00bc9e5a54811dcf4c89fbb143cef06e9df116c1b37227c9e43525fd1aa01cc0d3dc3ce3b9aa4f28c07f1e7c46b790e526d36c985e18ec2bb1c0a3e2c14dc0222e18a033dc5ff5cc0e25009f21787094747f7efa7cd1f111a9fdb3e614c65258de3081b9b1d7e230b69ddd786c9ae253410901a651938158c8326&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331526982Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0, Name: checkpoints/6016/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/6016/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=770b68fd0df22a7bf468318ac8dc2aae0cfbe68bda727b64c36daa1122e4765ae8831da5285aecf7f03c99317784c03be5d1b632ce20a504b154b5230b8a4fefada5126f6f26bebedb1ca2c65ce959ae476f534fa05206dcb9bda255dbb1e09d383780ff2b225f5d7651b6ac31d9ae1ab96315c343ee58a5695917b5b680f86107561914e9c693b2d58a45d9ef45c60653952067aae10a1fec8e9d54b796334154d528f7eda8c59a6083edfccd6b67d6c19ace5ace6393ff9207a0ffc135a739b9b0a08d688cf7a845839ac68650275cd3b7a19438af52c1d836521248e72c0053e4d9e74cccdbbd05741d49fbf62fc79cc9d0c8ce688ffcd388a25993c8a101&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331299989Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/d, Name: checkpoints/1728/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1728/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=800637abcb97f8fdf9056111172b2da75eee1b07537d4f381b660e0c20c56d6a111f957fee6ef95ff11e23cd3d6e6b1ee81b2b5f63f3c7a9007bdf7cb3e6a846c2ee01a83551afbc72eb19b27496105e792900a520c0c88bcb8029fe8e71d2cc07589bd3dee18ea8a3dc28448e15898ab4dea145ec24ed5fc52e02b317256cf24b9c7103de5c920b464a6934bf5e60733704203057a9b420918cb943959ebc6a3c12c946f4f92665afdfe6b4d54362aa76db4cca8232b990fc091df8380f2319621b3449ac6f77e2ee9f8295a7ec277cda45b45b18b91e2514195c8d0d8985d743543298210f74f366b6282a38652ac1074588f7cb87e40ee1462218f41fb0b0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331357299Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default, Name: checkpoints/3072/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3072/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=abf1f7a56aa112e4fce5ce77f77dcf9947473b784ee6e50868eddfdf36a2a299d495b41df054298a52211b5a31e8eb760402e68af004e3894bb13e0bc0732e9aecc4b2757c8713828884050a36c207af3da380673230bb375c2800922a96d41af21260ff2c0fee5be26d19d255be81d718b968a3ee67ea7f04ebb1abedf7f90b5843f4bba3db84f9ad695e977d97053fdd466e497347128f2307273c36b55c6cd845bbd7dd5364821ddb00239cc489a25447820ae96501bcc59388ee8b88b0b96d64deb4f2e9e69956d021ab5b7c071251d622a1ec91c7a3e446f1f09fdd2d10139472922ca1b3c571862f66c16c07b92c482f8113d31e30a2e04c9740d97e71&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331626605Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/array_metadatas, Name: checkpoints/2752/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2752/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=5071bc3d4bc22bb090e68cfcd02c2f3204c5371803f5ecf354379c4b64d696ce2c4d2a174f3da2ee5769d77608ef1a947380989a424b4b6d4f2a0ef1295f37f2ef56e45078a3bce73e6a237102f18a718b1698147e5a707ca43b45a9abf07e2ef17b8bd2c4052b51a7f5bc846033c02879259f4ec469e6cb2c319219cf1fa881c705cbc0848a57d72912bf366695a6d092d740be67fce411f2cebaa724e21006d1bc4dd11f50dc261f6293b5d3839ef4408c3a5d549c33e99f56523e080f57f76b5a82f0687c00a7ef87b57de9693ba2619cd5e7498af893a293f48c8975dbfa4b067bc7a02ad2d2bc234ff1db60ff749d6105a398ae4f6179a0cbbd6af3e113&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331657596Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/d, Name: checkpoints/3456/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3456/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=9edf81950dd572797b4b5b9ba1eb26ee0cfd8c41fce767262fed1764f47eb08f710c5b19b5dab2a06a94343a654c173ef4c481f1c92af2fd92122a4ceabc1ab1d6c3a4688c4a0883f36113816ca314c7d80a2ce1321924376b6242fed041ccff0de9666a7da0cff9397f0d1724166f5de2a29c6a3376dff3717dae5061dc1908cbf771a0880eb58fb1a5696ab1122fb059ac9d96bf9c0498895f7b33fadc932e64aa20760a71f84938ad58c11fd130edb413b51fc29d9efd8a9723899cb3fa79b2fae7c34c3f390818c1fdf7b81e18ef77c8c012c45d03b7d4fdd3e5cab645094812c364b3f451673bde6a17867446b0a82188295a1f69b6b34b19aba7e77670&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331659791Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d, Name: checkpoints/1984/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1984/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=74add51dff2a6ae60f9d8c34a1116f472810858326f5079d8f6037ec24370187fb70ad480f45b4bd8a3fb2d39b69348264caa389dc7063cf59903331718544b1195dd036083d9204de96a785da87290937b59f11ef498ae77e0af5e396761caf494ff2cf4d00fab4ce320d2c062f21f170548930efb63559d4b2481a7162a187fac8c7110de81d48f97b7747ddf4173e943e37c3366b026d612acd238656ae263fd8caf9e46218ed3d4c5c7bab043e41a215a6261070978594b13c05de8b3aee2b19c25f1be3d8a235d927eab2bb18015119adf11079792912ac099b08617cda7a3f02a9895c7050e329b888467f380672d7d0e66fd09f0c0732d43fbc0ef4ae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331687268Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/d, Name: checkpoints/3584/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3584/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=bbf7e57451af597bbabbdc4b4dfcba8aa2da5431fc2f41680a09bf9ff84142b3e046216fcb1e222110492dcab8650081e14680490d97d650afd31beee2c2e431a187c35e5506aaaf92c2d9842e69d6cf82b6b5064bb8a468d66494042b73ed303b333e7bdc544ece356058fa81ca7ce5b108cfcf9dc9c71c99ce327878ad6ea9474ba8dad0e6d18e8eaa4851354bfa95c958c1446b5fb3e2e74d97df00d9dc43a124f849209f3179dcfc763c1e496a2cd597cc1ae4cfb25aeb39b31a91990577d384cf355f7422026115e6b53d1d1bd2d6d901685f1df9cf57938ac08f6c0a5b137e7de39cb73b9e98232e1423fec0b6a579be4ae49d0f6f569f12d914539bff&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331496462Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0, Name: checkpoints/3392/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3392/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3f26821dd9923a1d6881ed7db1b6dc7066a516afbd393ac52e06dba09aa66e9bb689c0d7007ecaf9d7e4455d753fc9755779912a9a2a71e2fb8e2c2acbf121af676551869aabe52e9e3d5ec4ca5c6c4bb11a69009a0da87b9b54ba1372b2cf19aeb7ba3dc14851da5e38fda4a9f5282fa3eed6f265261a23a198620dda64fd0743c7e6e32485a91a391b35f636e88d91bf2302752555f0a416310a114c9d892fefaec1c3fe9340edf56e8d758bba46cc2978e925a8f91bda29a50cb0d031924cf976465bd7cb699a5efba85cefbea1289796c2a84b58f5331bc6fe53e75aa6aa192198efe004b741b600a712865a13e78da164ab0ed231b86b32ee2f4d7d9c25&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331740756Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/array_metadatas, Name: checkpoints/4288/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4288/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2f281f0f56bb4b88c9fd60ecabf12feaa4f941096ea23c9c7aa4f1bf3aaf7ebd121ae9e01aa1f5ea94156d2e4e80d641b87511c5e4caee35733260c719b6000d805967995a02c8add938306a505c02ea8fabb228660704b68db61e81227818fb9e59cbc5a35631c20c7639512d25152b20e056a6e354e0142397cfe83f1cfdb637107df625e925452bfd596e18415106d0a0adc5af35a7c7e20a360e4fd118f9a70ac6183cd0d7c079b2d84ec16d774514cc797cf96c37598a088a4e1f24374746eedebd02868ccdb2455706a2eaa14956549f53b8e9a335344945a34f75dce5e3e13cb389cb6805ce9d516022866c372a92ebe1356b405873d6ed9531875b67&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331746333Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/array_metadatas, Name: checkpoints/3584/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3584/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3b4b157561f0331c89d87f7a3821e2225db0c985c72aeefe4b7ec23799571a8722e12e10d3dbba3c68279dd2c66dc11f7610274580d3d8a95caa566753dea554900195fca39369b09eb43c06c15dfb1ae6c63ef083d26ba528ca69ad720bc1b84f32836df1f51e001ab78fc3b85f0d3d4813c0f7f55581a30354aa7bf072326671377c5e2d92bdc41848a9336540bd5b757937151845547869f6d8127ecce9f4e414f5a650469c1c528f976cdaa42fd4eb29eb36c9716d5cb8dc57ee5337d79099938f7ab02b2151838394e70fbe03587ac8ebc7102c71ea84a6c98be787f4d192079125ad10468e13055dc14756c16e7bed1952d9445673a3af4407a87cf2a1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331803263Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default, Name: checkpoints/512/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/512/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=94f7b398dc1bec3ffeb64f3ac77057c5f6000f74ef8087ce7f3ab0d441bfb22d817ce50690064165a46147c06cfcd85c315efc3fdfd3fb306e5f73ba897f6003ece9beab8885b8c8a21c668cadb4851c0683daeeaf3dab4cb5e3280cb5547f304dd7ad609e4913e27152d891997a5b99e91d4e8cbb40837f9553af261933f84a0c843697abb20addfebcb097c29d684e14106d16077cf36f5a23be0949149eeb47a9b0ce2971efefafff114f5e70d9b6e3839b0fe3255f096e92aacd42d8d3f54170db7bb72df7f2f806214939914f2cb8e9913e82d4c583e25af115f2669a0fcda5f2e68c8302c4c6fa30075b8e46780bb94770bd83909036c51f2d22b4ea36&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.33181341Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d, Name: checkpoints/1024/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1024/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=c03745355dbec4c529b1fb7177f89067c91f97d063d7ce685be5b33357ee7af75ca781d5c305378faba49cf849f5e12677ad0e0944df67d44326e6ff0bf5e8f057787028b484656003b37e855f8a4819e5a35397c718e24a9471d40d672c1c9a40236b675d712391622c2869d5a7cfb0d99748f6d92cabe97a1398e81830fd38286ed00b43a8131313c648c29eeefcefaac11c57a73ce4f2d26c7304ef947ee6aa8b4d8212fbda86bc91815d47340b35cc459abf974ab1322efac64f19bb4e5a6638e35d144deb8877de36bf29c90b313d2832471fe2050bd6c35876fa0ed1f324428daad76d57f54947e1ef239993e32d98b244ec1b30e7e264a80201f081f7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331716431Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/array_metadatas, Name: checkpoints/1408/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1408/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=99348765469d3c920d84ab4271a25d5dfa09abd88af73c3f06825ea19a24296e6ce4b14795636c5353053c7abc601768aa199b23215a06acacc107af1cf84c7c87ac1016fb0af25b3f4e71995225684fd485f93394241c6c4af094594a6ce24e732c0dd2d463c636b5b682365056a84dfce8cc4ab05dbd501b9840ae5a20cf64a8ad3292449717c1533b307531f7b2b5eb2bc4283b61610cedf0c344fb3b7897aeec3079eb2c32965ff333529dd608edc33a98b7ec4f8f401fd9ede0b7035ef8c103eb01c8ed83ec0afb7172fb236d2183a35e7c3b904e8a88cf019b5ccfbeb9db1759dc9322055626600e46058d0ab3c518f2cd2805b6cd846656d685dbe009&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331887136Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0, Name: checkpoints/1472/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1472/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2bf0bc4f2a20cb3e4965269dd93a5cb02b0eba2badd602cc19af85275030d6485c691d6b87f1819851c197af0550c6e1ed224a20b59aa428c5d2479963a059343d3a4b8efc14765c65adbb0de37f7be3c2f9210aa59b302ee0e57c6332f3be2b2a9d2e4dae20798a0eca0583bf32b12bacf40a1a798ab3fa5e16eac791ce95baba06f340710a2abd0a89c8f6100bdac2c90a8854fa2315af30aa96356c08c800e2eba1c08a89c363d2db5268c96559ef8bd8d5c01b6f154780866fa8cf69319d38c76a331959a0f78f86db8b528b8af21f6c03f0d2ea366fef1360c1da9cde3108428591251d9181293aa278a213ec05eda9c3bab9daa5d207cb695771da3d96&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331950245Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472, Name: checkpoints/1472, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1472?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=971cb420607468ff0e98326c36fd7779ea714c2faebf40f360940205e739aafec139e999fc94da51f0424cb2233c4c73bef21289699283f6c87d4be144f4ce3fe0006696bb2b49112e10b64c1119777dadedaded66a1504a8fe3f9f4776a30018d142bde4691f74152f73c2dcea733863f61895d9087643bba4b6aa60fd6cf8c76d021df68d2511ada57124afff4ccee3eea7152e7dfe81a39d00b6773fd09e9ca2d608ef1f02d9d2213fec4205b96e3dfd10525397e7b199dec52bcaf2dff310bc9577d92a28c1ffb9e6ef4fcbebf54dd181fb945717f483a2608a404efb8c6694c8799d4893a627a2b9387e2ec06edd347d3fc6f3a1bf898f16476b260fd86&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331688174Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default, Name: checkpoints/4928/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4928/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=0e39cdc9f013375e78f33797f00553a6639b0aee115a9062a55f1b0dcd34b3fce6e464ceb60528218b8f756f521fcb174dcc5497fff4959b4460dd048836ba563b890826a3bb537e1593a3f9300dd001598408b4fe9309ce798e67faad3343cd783c7c613c7c3c3511b85ac2740d435c16b5a1da1a6c300928e055d09da258a9ce19b2e3e52e4a0a43e494cd760470bd678f0ddc8093521fcaff111a54abfbc077d35347f492738785aa1be3d62355a7fce21624a408aca6b9af65abee96ca202c35247187f5e9acacfec19accab52fc012a6f5499922fd2aae60c970f956d762fcc8f854938e5a8b58744c5db994fcd9d1426e00fefbd6b1de29a30cf4a5f3a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331978232Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default, Name: checkpoints/1792/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1792/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=c1d30f7ff657b50df0daa01d15bed80d02908e6bf10ab66d3057cf8ddb2c5f67460685ff05732abf55cfa8e82aa7a884d0af1d8ae3868857aaa8489a8f25777c520932ec91c3b67a6b29cb7adfdcd7acfeaa4b1c0e59c46cf44ee273a9e993ff5a7aeea8a9a5b8b9c5290a0ae212cf79f7c9777b12c7da9d61bed0365c0c133287c72ed12f48c4cf4617e3beae9cdf534035501d53370f529e4437923cb9e281c06af1f822b42d50ecde268ff7f9cdaf4fd5e853ac7752744a3a1a22c09c408b9123618f2af2c09ffdfcac691d40559b9c932f828e8552272e95ac8fa3fae0463bbca4f183fb0e7e8d1a8dfa632ca391abb184a0450d9583c4ccb047c7327bb2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331992537Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/d, Name: checkpoints/1856/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1856/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=97d8239bb41bc67ef0e579777855a0cf2f6f10fe5b03d92348a230accdba38be5debca2196ce3b16d7bbe9a6d7c03dfb080dd386a678813d8474aaf521d35bacb9d6ceda96ff20ab2f888a95be5d7d21669eefa588a46254e620ba836c36c88ceafe069e3ecb7b7f680b8dda91d8fdd530001faa0416d59304c602500647b26e78a58049aa079926084755205cde366f3ec3e3b7383967daab5406a304403615255032e0d2b62f77686b222862dac3d5d4ebc6c8ee59615c5e74d8469e94098c4400703838b74c608662aba3be7eb8b7c3283b286c0dd2166528f04abd2302ea8fb0226816ed49bc23d1ac1aab303e35eb1923c7ee35fddfda9445e6a55f9838&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332006459Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/array_metadatas, Name: checkpoints/5376/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5376/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=56af9507365f5fa9b5b4ebcd7bd72cc6b8cdbf64a18428ce977e8ff4bb4d9948f60271e756163da0049641383aff4cab399cca911766775884fd142c31c7dc9b62493b355c4d93739464b421d4eb0b55f3136ee1406dccfde5669b052ab7e63989ca717cc5238adc1f2fadecaf0f0cf8ba20feb79bd3abf03054886eec42f00e3231463220a0f10ff26aae9baa43fd1f18171457bfe850bfc8d5e9029460bbab0dfb06a14b5b8ec764c86b9e66580b587f8b99a670d4f0f15837b2082e7b78fb61041316c734a098e1f4974771f5c288f4eec844b081a3cfce2bf7b4fb6eb2fe637b3ac743c2ed8d10d1090d63ae5f185b65a061c84784a996a85807691870bc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.33181702Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256, Name: checkpoints/256, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/256?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2acd11ebdcfaa4166c5f7452b242c73dcbd5e301fd0c88983716035717478e9096df3b981999ea645091cf0516b6e754528d5a44054080e96d46718dfad348e7b40a4d4f3b3ff5cf6466ec47658caa4624b5f65c4138f60a9d8b45057a2f3060b57698c188dfc934c146f59fff86b0d8afff849093d2a0e03f7885dc477fce90e88e97762bbac9e04184dfa663dbfb7445f3342b4d2bdbb6813e0f5df975c8c286601b9415f08cb7213b9222b6ab538f9be6fd9ef0ea7a86b35f378dee43400ac507af9ae17c9ed6ee3109471bc2b3671d65615327efe54a142d829c35b330321a914747f0e5ccec4754cdad226c440a6df6da5677116fa225dfb2a322106bc8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332056964Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/d, Name: checkpoints/5888/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5888/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=77636055d873d8bc3b1b71e8d168be465524b32dab2ae0963121b1684d460b1865e82312a75fb4a259e07a6d027f3f774bbc6b1af0c7724b784ba1b39ddfa6bbd8450dfc6bc0cf5a1e9ba9480faf1a993fcb16e2ddbc135d86d4e25adb78477d86068deb82d2e0c8b0b00504c52f962256697cefc1f1e4e816e4703796318b7fd2830a79f5d6b246fc04d7dbce5116e4f527e52323d777fd682509519072c76b68f530eb4b5669b0b96b2502d840e9fcc6e7d3aec74c399b0ef41f3deda78a96605cb39e8cf7180f0481241d2249d9face3c500a1682e042ffd8ddadfdcb75074228fdfcec214d2a1bd71ac936004111906a4f668bca1ade2f73b7b05f9958cd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332085106Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/array_metadatas, Name: checkpoints/2880/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2880/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=bc88f13588ba15b209abacb2d2875ae81e687a65a8370bca5da69455e914e081a04f873710e57691afaabefdc82c74a10fc836364eb3c67e24359ac5e8d8c07828f933091759df14e3b73afaa8928c0ca780a94cdfb8f249b53a3c16e790adf861059eb7d3b6e028095999e0511bba41b716e4b48bb330917152d3a9a1d7a08349f67d0f2c597e08fdc4b75af176261f490dfd8b261c44de9449d9b1868a757936747c6b05f31c3c8316e93911f804c4f1bc4e1698d9dde2667c9ebb01d1989a8469967caea54ebe752188454c5ee74ce7351607e1df399392afb95b93bfe390bdf07f498c8174129c865ea0e5b99474758b498f6edbb3db34b00d0c6e1b57f8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332096067Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/array_metadatas, Name: checkpoints/3392/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3392/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1afd10c4d63f978adae94aba1e0f7ed63c275ff588c8fb7cdd35145fcc723ae9a2e832b6d3400fdf3c4a142e8672ef99cacdac9c617c004fe3354fba32b75e7b7ea61cfe9e6b613c202487859d19db43ca97e5241d38919a466837b160c07ee520478f46fd39641ffc1749ecf3e96aeef8e6b3addc296f1fc9e6b2f20eb6145424c793836fa0fca146feea39eff2b4c5057d3d29f508f4c27f23e4e42a2d8a1fb84cf4060c973d2dfbc449dc6bd45ff1f0d46def3ea7ceb3e29a98c3b47c4ea50bb4c051f560783c34aa6aef9b88c7bcc49173502b47aca56c7bf9d095200a202610c74a6119d635ed512a82fe89d05931078f92a38cc7cbf5a3d8074f3b5fe2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331549618Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0, Name: checkpoints/4544/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4544/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b63736a3b6468d34e1362677f22f877b111eabbb9454b09c5a0574ef73b9540cbc3d9ef23b37975b84cd084fdca2b65ed2e6c104aa7eb433402127a5cc8edc746d1ac187e71304b7a3e6b758d713b52e649ab8cbd86cc22c0a55f72f72076d6f2c710a6484e7155960423198e16ca38c6699c0439ab26b4d19039e481195f63f08694f1bddc307da7407175beef6dcb1f27d4f3d048080c377043db98c20dd1767fcf0f226b2c12085d5a25d973e0e26bef9bacf3c9b34d77a5b0f88bad9dddeafae463e3abaeaba7655b46673617961a0ee961fed0cfb716217c14d99cacc8231ee44d78e6bc8057d8e15e8791c932f30d003a6f95631d7222ed049ca878b2f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.33166731Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d, Name: checkpoints/1088/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1088/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2f5d7c35a1f60b4f56692f597052905adf8218b9e3ea08a8ecd5281484ab783adaff21095b4cc7133b94b7f04ecb35a2df3f65437b77c54605d9bbdecbd1aedff2d31010ac8100cf305f1bc317b497d6fe6e9b2507fee99b5962f04159b540fc37510b42aa017eb616d54ee52a6c0e89c042ecc6d9823c6149b7494682a37024ab60c7beaff1b326cbe3ee11944e44694bac77e3ae072f14b2bda854f87aea54502570812ba356eccc4749e4d86782ac83069a98b7217d9bce28f19189cec32bd46cb4681186fd12a680eb220576db4247e3b35687993f166af7f78c6da0a3b5ac8f6dde954e889a283846a19bff5fe0d1cbc4c7ca38b4b3d28bbb87ac59f7a6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.33206554Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0, Name: checkpoints/640/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/640/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=408ceb901f560f15da9efa67a40f18475fa26794d3e87dd8d554502a6f8e8bad7b15453eea5fb0a42d52dd878497bd820d0712e7087c1164d2795c6987e2696f7229a91f978827819eb91a5a89f3446d5d5f5797560a4f2559e4ed113ec211961b7d9e469d0d5a2f0f219a8f61b6619038c259cce0525b3cbf5c483986215e690d172ff3083810e91b8d3c9e7514462c1ba2be2e4f5045154fdb02ae03b6d4b37077385f281b820999c1113f0845c1fcce0dab63452683798b64ef820733c258ada0a917f7a69c5a82288c6d42a2e8b756d03b57c999059996e52f54441501cabcb8df5a9a1127a02e1d6901ca7ca35ad45c1ca7b448d61b3a22ac4a60d1e177&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332174605Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d, Name: checkpoints/2496/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2496/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=94aaf1b79314d67fc5287de32ccc93d75413a3648ac48a42bc5359e1d8bf7f138d55158b8312640a85d02a065bb32b9f23a3950f0171834f92e88a7a6367b800137f3876a53d2a3084529c837c325ebc31ebaa766ba3d9fb7dceb86f1736ef8f8b6be1f7d570d8deff2a2a73d6d670b59e3a23a6f30d5a7b7e9e0b444122e87769a15692f15816d0a3668c34a54fa921cb2d17d1bdef67a9f9fb74451a353b8e7b2c5ff0d92c7975ee26b940592e6dcd00dd4b7c84794837e41db95d00375f2faeb5be6149330b65d59ce886e192d1989b8163f653250e00d09bcb5e78aacb9ae130db86b1dd9c250841c5cc7b7f111c8888164d71ece0ef0da57f55b317dc13&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332181Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/d, Name: checkpoints/1920/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1920/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a70e324b01f37303bcd276483aab9d6226801e9f925ba193f6cea3c2475226577304a4385a7b3fc91fb003a437512086204bc1e492076cc5e1c98e23f63d63f8df49dfea4e447c107237748d43824c12dc19218c3b4a75fb8fb9933b3daa85491156c676fff5cacc0bbb0ca5b157ad2bdbc1418be6b1cd20b785e7b43280de8f010c273e33d944c2130d36eb6e05fbfb4687d183d0304c15916760abb036737e82f72a5cc572ba499561fc47a458646db35be77b61d2ba6779267ea72fadefcc76bc17e4eeec38dc532a38f5bba9f5949817e1d549e1e83d31c63a12d2a4dea6fd86490346a1877d61c5b56de49f0b8c0da3696543217806e9fa2fe8256e3f8b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332213634Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/array_metadatas, Name: checkpoints/5888/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5888/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=95db96395d3b369b012871d23239e416e8f437d7857205a619d6f61e086f91c09a9fde9b280d8f8d12e771047f2ff1f5fe3d12d53b930ee0732653db2302084e936f2f20855bcd8efbadca81b23b8726c3ad2a866f503c7492363cf0288dcfee965fa75d5d61ea6f4c0019657f4b2ae349c56dc56643c1a02ebd8da82ed3c0b3f9616a628cb027fad5b71a43122451ef75c81d70725e16b59802ed303961cfdbe4656fce000816e10e3eae69d22593ce3e3a206c26aa82564b2e1989b8871137d597cc62e138fa8e202939439b7f3ce3f7d8b3105e321899452e730a07b8ec27d5a673225a65c027998568fc9d0030b8ac6e1d34714e12b18d938dd629dbb791&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332220408Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default, Name: checkpoints/3520/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3520/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=08fe3664ec4c0d37eca325bc32b3d3251fe30a82d8cc179699e0f9b2ff3f54c5083f8c1103166ae1af7347429549b7b80fd02ad3780402acd04de7a41aa8f6245461c91012c5e2df60d7c8ba446fb117f7129849a14dece0c5ed49a94f23bff2f5ccfbaee078e86f8eafa771b20d96c958f213711dea37c3f9f853f0889cbe08bc1c98fbd50d4447cefec6bcd32db25600d1eae5f7cc286ef9ae7db73ac32d26733ff5a5c14e3b9041b804a87f67a8d7c2175e4a7f61ec8ed7f1df47b3eaa8213b30654f98ab81d57bd7a53ef2d76281a086c449e0b092a4ac6d4f2160fc8b86c03354860500d7d2234142acc9cf6fc8d51ee3a8df62d5e940a1ff653db19926&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332216036Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608, Name: checkpoints/4608, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4608?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2ffba74abcc9b7c23686d84877c263a8dc3c33f9add75856ec403772aec0eab26cda5464caff9c5b794b0579b623f069ea2d83faf421afa0ac8751193bedea34451b5b04b656765c7bb9d76f49128b7feb9a26cc55b23cfb5c430b236b86a26b2b942b431256f4e5b2952abead372a09b405d32214a256b3bcde5d3a9657b704f7d909b71bc4a49b40faf5612aac8e72234bdf267aaf45d2ec032c3646b25eeb4e4c8242bcb64171a201044c457438b3b365131674241c03531831855c88bc2cda6344775b35db54c7a42e235fecbc2450086c2c9f72e4e56621fb6e21ebbb904e4f0d41021a02297a72453b10884790f4dd59ce4ac530bcf302310e691684f1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331638553Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/array_metadatas, Name: checkpoints/1472/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1472/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=5cda3eece3cee5ed8938009e45399ac8052fc1bd230af1e17a8f827209c40c8ac27d8c692f94b7ad7f9fbca40920f955b0644b6a3fa4b7ecbb3092892f746a39f103e221461f0174f7eee4e24220788fd0b5dbf05a25642acd7f2ae4ce974ce48f987319883e2d1ebd9c323605d61d96eeb03c0f4f936fc8d7d38b7e5e28c5291d761342a5cbff5306b4108f042ddbbf8eda928eb41a44b3ce88dfc83f562ca80d310318f287601e745e8d472efa9dba65ebabd149ce50dc20c08b86c940bf8f399e6a5c32915823803a0fa6f12a0079434afcc2b8a369ea620f2a420bb33e5fd33715f5c7ed41b233153f74d4ebff6a1dd183629f632e2e83fb63b528c9eee6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332264113Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024, Name: checkpoints/1024, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1024?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8f43654da77cd281f1032375bc63fed58d61814e02cde0d116cb2517170641ba955101420be1940e5b2d2f7d1bece446da9b79e72b6721df61260371c98e43282721ccb62e82338e1ca22cc50913aab48ea192999ef439ca3a13834feae94e92ed94e7d97ed559fb589bcda608d7cb8606bf1375cfaaddac73d68afa5bb6a77efd25e7da2f4d5221e50d5736fc96d5c59e5693eb27c414b0716df410c8588f059a70e9ae8547cc47465030eb80d7c1260e59a62378baac288b0b9097d9bacee5dfb59e69ed92c8e34e9380f8daef5c68c62671cb926a967effcd3008e303e739e5bbce390ede1ec257d29496640ca54086cb0bf652bfcde7f287989478a23298&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332069298Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d, Name: checkpoints/3072/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3072/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=28906e69b52907726f4244284fe1a62572d4a590d64697f3b92aaac1a4c134d9c0d803a6f4a7f73575751d7e576adfc0a34a0253fbca8e378b371cf241f93aa048464fea0daf66ab1689ad9eeca560509a2f5536e854589c082a837d27e8a47874825a0a59de6bb374a60dbdac10418e48b2d29dfb5eaeaa637f7679ec04fe2bdcc88cb5dab52bc211df7e7d3dc6cbc0a7d0a393d183b8373d09de52f462b637bec03ef890bc06b731f7aa9cda60eb375aac6d1a778876d7233e6fd04431ccaffbb26eed4e176e48b6491218e7d71e02d7c0faf5fb22a2a272266a77c1cb6f9a97c7af0f97de22f42a3411abcd274a746596df45f8f814919697db66aee4a898&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332304745Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/d, Name: checkpoints/640/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/640/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=4927e76ba44d08f4b1e5298ecc40522b52e5b4a875b677cd2e7d0ca6328f6a0d8d7a9ffa81c3ac964e5a5ff7cf08ae8e5971f0ea04f60b5eec17d1bcda8cd8913541bf2dca4d86453d402a82eadb81dcc3515beec20e038a1450753dd23166e27093d7a74bf933a82053050e4a58477ee892a6ef4f8ed55b26075d5ed80e6896384a93ac727d04b0cdf44a5f613815eb46762e24a0307104c63b0979f56f201d5a73a4c5f60fb504076c7822b7a616d7fac9d2319c3cfb24481473595daf44f5fb0247d946d3b02925560be4cf0dcd58d508b5573fbb7de5761d8ad6b9c3f275da4108b6b389de55181345ce81abf63d31a2ad422eb3efffb35f29e11bbdd135&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332310488Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520, Name: checkpoints/3520, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3520?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=33f21e4d1c0b0cedcb9a48339fc718931636f7c45c63d94ce2a68baf1a1294fd966496f39d5a29db7c69c1b2e25f32152508775396eeee4b6ed9fe14508956e06be60bcff99542cfcc9e818af553e4b2ef7255478f48f4c19c05531bd4dd80f5b1a24823d379664bd606aca2cae87509bad42b628de483a65831b37ef246f8c217894ede31ecbc387e6447cafa3b279a524921100dfa1162497eff98f6f41a361ecade7f7c5ea82623862c87a42b1dfd7409fca97bf39e98055bc7e74974c0f9bfc8ba2f2f8c467022bef91647290431cd5958a8f1ccfa15fe82d38781da33dd50c5d1539ef25092c0060a6f500251cf0a8088377b44b50f7369d0c9555e651d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332327508Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328, Name: checkpoints/3328, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3328?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=9afee71b28d1e59ff20e58170cb3ebd59eaea6aebbc099107ac3b9a2e141a76a32ae0770eba97aa331f27aeddf17c3cc04f0d3f1f04cc2bd49cc9c48bb2ad3683dcfe8dcde041ded1eb99497c9f451584fa3c9f1e13364ab97f01d6401d964e3d532a9061effca0f636bd9bb70a3afed74f1a918e4918dfdbf54a875e0b9c83c42d8198a0bf14e303579166f61f07798a3c7f174fef7ec5aed6a4f2b27d8814261bfd3b79170c6ccd7c37acd654a598898475bd4b1de0f451f90670f55834b11171b1eb306b85016e60800d1d05e2af2a757140c05bc67afeafb956b15a87dc536bd2dfc20f22b7b947669ca1b7ac2f45fe04ae7ae9ddeb413e5a1d6254f9411&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332346837Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0, Name: checkpoints/2944/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2944/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b788de1e7af51eb930afdf25e870e07593778bb83be2a87f22e1fa266b78b0b0d8116bddd842cb200a87e6f4c18709023021c4b2d624e41f440de09e3504880c01a19eeabd5e5fb2620a507dd9768a8b0afbdfd5ef12c8dc25db4aa41e1412ac55975e38346e9a1ce9e90a2409cd55e42a48fc06956602731731fec1143af760bb6a443bebb743f0eacf1e43c3595757caadb3885ecefbd3407c9e4bed893eb065148e30cb3f4971886df8a4c4bff30e84230d66fc0b2471504e938a7f476cb1fee26b077568c2fc7951fae1286f695133bcba7693888adc5688ef7b566c8f6a5a0e0c111f75168e2fa3acf9f7520d776879da68e6419e31da2fbc5ea3649f47&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332376284Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d, Name: checkpoints/6016/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/6016/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3a83c02777a7f908afba9e8bcaad126557ffcb958c6158fb7c307438bc70add28ac22f2f1a1c1e68e3da02909a02f7cbd6ac5e91f6fdbf7d37b1ddc84c3404b0755e0a8c4055aa63bd4c31ae6ec31a2e71cfe73740fac3d08e01687ec5137dfe35729382d3af5b7b1069ea0a32deba789cd90e958447a0f0e22cfc4d6f49faaa0e42884082c988a9d0a0ababae8fda66f5824744f7e637f95083525742a128a4ffc7951618f027df93f3d1dfeba3e5ba0be006f222d5ba9f29aba8ba18c23eec5e4aea2059d9f3a1faf5eb8aa1656ef71ba44ba874baf069da6b8379dadb0ead1fdc6f948d591f700fd229ae81c15d33874ffa1c53d7c62dcdbe308e2ebc1361&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332315205Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d, Name: checkpoints/960/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/960/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a7ae462ff39110b765d6ca7c6a77fe7c493e3d88ccbcc8bc410bd74397f7b0e5e028f70e44dbf08e4ac0bb1a51cc55892d76da17c609894928df898cf20f333d393dd18cc97d8d7848f677663e36e859b1f9ce6a587514c6df887e7e56b8edd06189739031e10967ca9bf4b5450012320fb106dacc470e904b31a5ba442aaa2e6ab44866659b4d742e32021fda773cd59ad7522c584c58d6047f84fdf55790fb7f3cbfba01821cd32c71d5886cc4e565391ca6e5bb2e4618b6753f4d562f103339e8eb02d5b43395349b5763b64e34d5e673a1d437a194dab363f6b382919e99e271619e34c810dd8f5c3fdd58b41c68beb06100081937bf530e8c58f8a37114&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332415789Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default, Name: checkpoints/3904/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3904/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=36348b211da61742f287e72672766305d0d5373bd956447b4a657a6359f1217383f7986bab00b980f36bf01a46f7abe878c1f12d674f5ac8724f484a4806b6568cc30e0480a900c393ff5f7bc0135e10b33435f800276c8e2e151440b32dc7be642d1a1b42ce982dd80f97a3e8259215de530414cfa5c361f5dfbced3ef2c7cf4eeeef43e329fca52640a62c886337f25296234752b4c6fe10821c26dc043f62055342e585920f8b6492e2ba121a24d5aed0c69669efbf0ddb1868295e70c13affa93dd9f815c862488c2001c720bb761139471aff2165de40d5b8844f6a200c2e2459a5074eb98509c4510e672653346d8db53a3f1bffec57d93e872b1e4f1d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332435644Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/array_metadatas, Name: checkpoints/5696/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5696/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=17365f8bc88027b772dd4c7ff6d6d88bbbdf93992c97bf3cfdfc3fcc127e1af584edd56722872c12be231843108c5ee2aa7d06fe5dfdecc0a06be390f7420b212983fbdc26e81766ab99577c0fc774a5de3e4e29d838bc4fda794842d44d4fb4d7053bcf91891d7bab88b5f8aeed4033932fa876a04ece4e5290abb3b0390d2fd93373eebd406d204ac0346ddbad6027a4d0c7046036f8d0ec04638f7dac7b6c2a04d067ed77e78cdaa184bfd290d9bc02d30e34ceb81728f9c03534e98162515334beb28eaa4cd3e7cd66569cdfc42336a22959aef916054e11a3651ba17a9f3671e23914f9ce6b084838a4c2994cac3efe70dd857b80d35a3cfd3838e391da&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332451349Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0, Name: checkpoints/1856/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1856/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=5eecd0d036a2b3fa51c23ddfb3771ad8c78e0d5f395d5eabb66098f9fc670e06043f7ce8db433101097160cf59d092adc184dcf0c05d28f2f4f6cb5bc16a617748411e142da3bfc1de51cddb5c83d822f88f4c52986dfd6d6d52695c8c643ae868f27e2e2fe0c79360db59f39a787777ab07ad4781afe7b914ab8aa5c8e4ba41dad92d0277826305f40bb0a77a8012cb9a2e24193668c358c6d5ec6a5bc6b4d6d1bd12a2f01134a98e3dbf5c2a5acf04a180eca225d3d5551c97c82e730f6d077e5da0ba48cc3d169525a70763c9832f705ea6d082b691bb0e8b9489bd779313a33fa1948fdb1c8225dac4534342df213f45f10d6fe1297f42c1e119de8fa7f8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331991836Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240, Name: checkpoints/2240, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2240?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=03faaccea30b0c67377ce535274704fb223af1e327414595e0c2f3b991f5b086c49878d91c674da9684dd95b9318d4c8aaed8431d952304b7fe1e44db4534388bfb45ed962f3280a2c815a41fa808ec355a6f2b1d2ad941d03a803ed6c0c6c73fa2e616427cf57d02c0ed43d6cbf362411bf9de385fb39408ce189dea1612d6412c5008040b4cdc3eb1daf0155c3ee8d7e2441c2cfcec03eee2d899ea4a1c5e486224bfc644bb81c397319498e813fa52df6dc6d27f024ff6bc3771fa56d460c67f2c62b57fd83cb94cbb15223042c26ffd404128e3eb597df238fda897b9cb3330b2ccc02d1b0cf19c0ccd8973a6387c38516191ce94570441da6d878cd64ae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332474112Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/d, Name: checkpoints/4800/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4800/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=210bf991e7fcb52bffeceaea9ad95b61eb91cda7d0d56771fd3699f90d2c65d5bdd790542b4b7dab977f94b946e2d7f6d292b52f7c95ad99d71e38d35d58529fd29730791aae82e81313da78378dadce07cb27c7bc5f0d76513e6f2baad9a28e6401423299a32574f1219c65d3b645f5591a2e22c05eafacca1675e8c9a5fa48d3a3527e2b0eb2db89764f46ae248642ec96efc559bc3838d8beb80389bbcd3f3e25c2287db44bf094f3535c2ef7d4f231c1970b424d001998b6eef80abfb6b870c72070e757e9f7c8ff70c42c7ac585320ac23b59b819916b7fbb76e1f8614d406b28acfdf6a6b9997fe4e44772be2976c23dbec31f8d02b134a1853b83705c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332493982Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/d, Name: checkpoints/5824/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5824/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=0a7af21b1d2b356b675e894270aea24a17674c42073003c262ee807be16a1e4c70829a45fefff290bbe37f3d005bdb4611452910685ee42236938d1d5a7b0d8f63516943517e3dca0553fb55732e357234c870db41348fc968ec76028dc64ec43331928c066fbe3950d01b77d0628437b064cb0ab772289a2f4f081c4a5d31136d4c0705e160d2054f8d8f80fe01b128e8621cfa820817c0dc47f9c9a525c5b2dfc4678fc53da0441c6a833fa95a2a87cadb7e991ba41bde7e73448a5314ed8b4704d31aabcd1beefd28086d428dd5c227099a1c435d0129ea7455e40323e06bb67287a031c6ffe211189fbb9781451d0fe8d760f4adc36feaccffec54d5d7a4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332510565Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192, Name: checkpoints/192, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/192?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a27fb6b27632a29616cf7bfad32b36902870ce77dc648bcb10046607cdd4bc0c0cd0af6bd10dbb53b8977b96f8d9e6c5c11c40ee559842225726022f9c83782131913b4eb4e5841eba10560ba3bec1ae98f51a9e3bb5afd2956bda2eb91ffb1258c0fb3dd7e774a60b1b35e8073436d25da4fd039dc5c89ac6de3a8f3f250fbb4dedd8568259e4ce78dd95301ae2420ce7180a03ae910e220eadbb996b4bd30dc277469097646414fc1099aaad28f22252222732e93546896a00d16b23933adbc40b3ed2a15a8764853e238dea6626c04702f1f14a0b86972dcf49e06d42ed05848fc91c9a29ff4724ec9362de75e098515b9df4b76d3d483a94bacc1a5e0745&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332369299Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0, Name: checkpoints/4608/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4608/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=4661756bc5da2656efbd559ae8e931494e4f9a086b81a327ebfdfbd8ddf6c6d01007fb0c41655f093975e0991ea0401e6e6e2c23a17c1f9c08cb28e92170cbd9d417f73fa18f3bdf4202f4a91a4109d511830ffeef827a687979d5d4148efb82fa2de641a46bc65966b11c9f6ee20f1e0c49f9e744040797c521ff620c1a64b81d89512b426920d4f5b64d78b488a29c79376d8ad7660e69a875d53f2d272d3aeb89461cb3c4dddbf575365d35c977b85ac832b50c3f1135aa8b66a8fec46ca7cd6d72adaead7b1d11fffad186180811c79394d91efd1675e138fde079adb54702139f333d0400cdae7f3f1a28efc6fb167c0446e1ee4b59170559012a5d0a29&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332525542Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/array_metadatas, Name: checkpoints/2944/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2944/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=46215cca3adf54b651dff95f40481f2008741261322f16035d45adadffdeb5bf91b4cd7db313769cff1d653aea0637f7b025e01f8cd43e0a569fd1950ba9e009159f7464f1558d71b89f8a38f50870f6018f5900691a2842596a45082ee6fc3c5d3dcc565262331adf7c7cf9eec632a926c172d39d201ebb929be8bb75ad260b93f6ec60653d55f1e973588051dccc1b995f0ce9aec161b64dbbe0c6a13cb137577a1d4b410b3cf3ee9f5172b501291da000bebfb4a5f4ee1e9dd6830017738344da665a60fba6739d772f13342b71b0b0857ea842c286e8da2c6226ef0bb2ffe0c00c5d95e9ae42738e298674e9cd97b8d0fb768d291555460dd3ac60be41f3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332475111Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default, Name: checkpoints/4096/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4096/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1bcf2d363c4306742158a826ec06098292a1e970fff41a2467ab7c154d434817263dd3b055bbb32f458a97a2ed6dcef0eb1c7f699b684bdb4bca47a713d52bc4a710632643d9e358dd35e139608a9dec8db3ef0e689f8cf7522d9cc0049d1fd8207ef821d7bddb0813ba369ae2eb2310b2264628d2e60cf5e46c68412ef718d7aa5211fdad43e1280a68a58e7611aa26725b77fdf86392ed08ece26a5cf79f2b6001a1df0f2e51add5b848666e9b67895791e22ebce0b94dfed3012f218d97a62a569621c303fba647cbd3bd4e81e42d31a1ff5b053368d130a1d029440255e53349805d15fd40846a5fe53ed02f8a63d700ae56131bad312bb9360c50d4e0a8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332176813Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default/array_metadatas, Name: checkpoints/1856/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1856/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=56ce125026202be6919fba5d635975e1da83a3a690a090401074ca9424ba843089c6ed0c3d739e7b12935426d5d684b1d03f2a3f4fc8687c6f32895393f21717928ae2d00f5d81a1bcc7579781db11d2c851086f579a3600aa482b05347c149150fef68ef1429c27ac78f17da116906378ab9d2ab21ca284b4c07ffcfd66d3fc61fb175e2bd57fd0dd8a4b67525a516eee69c1e81e8934c94f36dffd5a5b7c1adbbc4d1f822b4b198284d1e7c41e75a5353bcd974677eb19628bfedadd558536d60672e77aa8860b62aefd94b7f62a2e0fb9e4f6d900854c14609f5ed2879baf43410ae9e54de4f0b841b8db8fbade906562a4eb5dedf269bc3bccffc9a08e9a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332418246Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d, Name: checkpoints/1280/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1280/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=957e8f47cc5b9b909368ee02154bc31ddf7c1cc0d25dbccec3192906b17c0dda39f9ff8c62d0c49886cc62db35cb4c0cf9c81112b4e5667a654bd332a50ce65d2a2f2639456ed0e053e962366b3fdea8f0a11f0d6f710192b512bb859ffef123aa831842fecf598c74c06dfd46994f8df574dbc8016b34b0ef7ba0bdbf2c75b62c00bf1e4f7ca72eb781db438eedd0fdd12b0ded7dcea4d0b2f4e0f4c7b3984433a04bb3d5ead85b4bd6b9d9f9e4c6e7f28aefde784c057dfef86638586d4bbc5c62bea3f818a2a6d20b8586415460cdc52506a5111adc3561878c5255f5201842d35f0c5f5dbf9e4a455d19acbf50b4b441070b4758001457bce57a4b8f7ff6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332189119Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/array_metadatas, Name: checkpoints/4608/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4608/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=c4e666f64e1dc1ac2c31f20810bffa61ded52ef70fd7c080075f8b36adabee144538de2f575bf898a9449f0c56971f998880e71c95c2c7f0ddd237e37be7ca4ec9f958b99f90e1baa3201f644bf293afad01aaa3cd25ef1f145e6b5e422a5ec1f95c882c7afe2076ef0b0a98af062c66df353713f3d6241c4d8884c54b31637eb1c028470e0ce096de0fba88ba6a11072dacbc7c8ee21ad87518ecfb795119f1d63288314d394cd9b000d4130face4297b007968218bdc44d2dde087472766f27af623964e44fb1dd7c01012bd75b59b02d35311a48646934ec4d536a7aff38c88921d80b4199d19db5018f2f755fb81f67931c1cbc0a3c69e1b9f677c0b8fe9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.33145616Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/array_metadatas, Name: checkpoints/2368/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2368/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=837fe9f4f426389fead17378e41f87195a00b70b1e74a09434e781c6455e6a9faddd85bf19c11d4725d5468526d489ae1dd54e316f507d66e5d66682d01bbe0c8025868298cd9ab923e4f06bfdf7dc89d1a4f26432d26c00e4a85276500af93a71b0e480ea4c0a95c587ae729ab9a5f0f9f63e725e5656ed1f47a41d4c7efafe5bdabaecd61172e2529c74c71d926cb1135700a67aa1bc81311c7f08f60e666d025deb5a1ed660e501273182e5c3df0c4883afc3cce20826ca49182914ffa17fca3392bf57ed3c16f384292edeb37ad9aab9979e83f0005c0006f1027430415426ccaf1758b0a92f724d46d8251cb3266202b1e454804bbd75c88e776b26b125&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324689613Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d, Name: checkpoints/1472/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1472/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=af1353cae0961c692b72d33e80e810dce018d86f80ab249526d96e873e26a4a9366f5ecfba62d8f6963bc19e4dad79109e451ce0fe96080826639f24e43e88a67041ac9a26a5cb4f1227e93b03a7c4dad64b951aac48dc21eb1b3487d3a397ec0a1c039a680c739e1d518f29948a49427fcd0069e511f2a72b40a266b2867d552a158ca479f2aeca0f9ad275fe2c0e26566c61f02f190999c0b6c016ee2e3ed59f00b3389c2c10e2c70708daad7a3748897fc38c70c9928264ba69d5455b8ba2b9809bae21e62c717b81807c029f8a272227923735b30628201c8e8e00fe83b466e48137eb8e9c53c7ef60d763bc4f8d3ebd4f57e5135eed09ace3ec04d965d0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324245882Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/array_metadatas, Name: checkpoints/576/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/576/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=71e8ca4564b566d5f6e7e610446c5afff41f55865d2dfa59ded288b8fda8c2142e3076fa0d073a4f4246bf67be6c8e7591ea1039a3dc04ece8f649d54a3cc357c53031831f462248687f2d99e72f5074658de2c55440bd09f6d3bfdb09d994453ec3bfc6d7ef813289c4037acb732791cfecb4df69960510018a4b5d56b6a71215defb81ed38fa7af510dbf21c4cc438172150a55416f386899fc47e162fcfd0dd89bfc71d4c8dc0603cb59bf365f4a47e498e9f36264963d918e08bd3522b40cd249e63f7c8ea9613b3d080f5627da111ad32d6d9edad11a3c831b19a2058f35cb8d1de24bbac713c5fbf246cfe1827be115bb918f88327e6e5b1c9169378a1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324744603Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0, Name: checkpoints/3712/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3712/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3d91aaa108ea8752aca3600f961e5ba0aa33fe1fe942b1d92580be1719ea6159b6169f92c41ac8dcc06dcc4eb467869ddc17f0a6be28ef82d024135070df11a59848269a4bb831dcf8b6eb1645056c1b98f8cbb9c59f710b630453063bc97f2faa0048942dddd9193f4419002c8a7c55a086b384e88a1277deca19cb862a402b04d5f88f8a2c306397bda9d5047f1c4ba0bbac9590c8f77fa2c3f432158622f67a184d0fdac407c0d354994a06e687d2ca262df11acf611f175af18d316297817a3c19213075b1adaa100df0acc7ff76bb74343496ef2a7d1c5d3cb30cd7621431787519c58dfa07e81cbb8ec7b5c937ba0713ccf3f119db02e91aa8f54cb8a0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324746013Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512, Name: checkpoints/512, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/512?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=c02ec5badc70b2847920762d36400c698a1e0d8b4436cc0bc8919ab9941d14fa665ed5311bbd6b120b27377f1f450ef04e13a7cd987afb6845d3d481ad94689fc1a96a9799e59f0690c41bc6d9eb3a4613d193c3d578eea7a134b0e5ecae6517d37c02b8bb8f57ecfcbe7072db0086956481c0bfc351c5fa0ee51dbfb82f10abf34891e8eb8585a25e64f3272ffbaf2c69a9ed750f873cefb0833ce87d0cebc0b0a192e9553fd239dbd856042020eee9c9fc73c9a1177c34f4f7a367627902873ec290792630833becbfe547ed999dae9a15a225b9a1e226079ecb966fc94d33ce4dd98e1ed08c7c998f751699e64b190187b3311d4ab0e21c478aff8aa43ae5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324759541Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0, Name: checkpoints/2688/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2688/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a2c1d509f018adfa91abf20bb04d3797577e34a019f03a42bf462aaca79603a5a5314d40ce376cb867e07d088fab642768280fe509dcb36236ffaaee180a8cd4786a7ea1f54c8ea8e121723d9b39e06cf3f26e28ba2e829b87a46c7add15be198c8b9da6f54e63c4b34f876b5970338751592dbde425c245df50d62bfcec28bdc2f0fb6b886823491a9a694195d959ad8f0f9075388716589e1948a78393bbc550e39d0881fa8b138712d3d1dd11fac8cad9952ea6c3928a014299f0e0441a3993d24e554ad49912be7eed3f83f93340310a3784de947893bbc03bd0f12e3ed93b93c33614b432312bd0e35608c623609d8a2874beeaaebcae8f0932d076e812&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324768639Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352/default/d, Name: checkpoints/4352/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4352/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=937c767128f52a6e346fe88f6aca9866fef59d1010f143a4a3b8b69b95d2e486769e9302c0cd67b2366c53109ed39780a604303e95034f93736c7942c65a5e92f6c0e968efad4fd22feb1297469dd0a48742ff1d655101b5e1e668ef49f5d346f7fb6bcd8c35aaaa1c3ea33140e6d2d983facf64814296aa964078d17dd714e927380dd9bcb8b4c5257ddeec31bb9c74fda62693d46698ab7ed2ff0dbe02a1b553be43fd4e0eb60c64ed9077ad7fe18a8680a3b597ac941c04d97b73d6e45567bbdc2065e69c47244cb684ed6af2a9dacff9ce742a840f0ddf4f275628c0e3746a017b87c8e3e53be69e8b48aa5c1646fb0e80d2c5db42cbb478958c171e9dad&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324773431Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default, Name: checkpoints/1728/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1728/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=0f24c631f7e193f4c6fe4ef3cea59da7eb89b9921af1e032000e4ca8e556c4ed805d32f687b6bae736ffcf7e76b1ca16917a013ebb772d26f7671c4437794b1e4ff32b4822031d54b489f89bee838052570dce7504cb21c26fce3b9f913e222f5fb9c2ca1f3e987061096e52dd75d414500f5651efbae81f1a9e495e172c577394976447d6b329b4cb537e3ca809e94716f630a9e4efec5e2ccd09badaa639872d879158729a973296b78ef31be656a957aaee4362e854ab3d7c1df1ef0a55a6cf55d579b698536cc6aaa37a37f6ffb5b3fe27856c590599d35f71b66732165945ba92e71410cec02d8c7338ac33a1f9359db23573798c9ca93863cdcf67919e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324787818Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default, Name: checkpoints/320/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/320/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=37678eac1bf7109ef06179f3574dd8e9a66b4bf3a559ce3857d1d5abc3a73dbf6c61f7ef8f3159e86af3d4f1c447d5d076c10bb532f2ec8243ace60f6853afdb99b1bedd95de5ac5a42a5da168bb46290501eb6c35f352cfb6d07eb3e16f427d9a75f7722f569f466e895087049060f4f359aaa7f076b9efa6c54f304e7f3e964d6643602fd30fccb0cab48db2fed95b0ca059c0e37cefca6d29c3b7b0865b87cee03dda2fac03b721877c17bb4edc1ff668a46f105926e8a3ae448761ebfee420a29a7e034b0b0416cce989f2eebc38311e8f700875aae2835bd7f058a2eedfc267e6625165e156208998f76f9192932e990d0254d06b0ba0281b0c0ea90165&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324789355Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3520/default/array_metadatas, Name: checkpoints/3520/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3520/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=aef1e3d73a3b2176fb343881b66d7805e7e0144bb56d408144100ccf17423a04218ebc3d415cf606c47d4271959d5637fa87afdbf74cac7147dbd7984036b093fbe6c2b6284c58b33a367f02bb2a56d2bbb458d08ec288042e020d970c5491d3c80eaf81020908198e069033e1ed47dda60a5814171661e8519d6785d1b64fc8d78d5b9b4a3c4946fe37c9b45fe705fe375ee478137fb0c08cff6bb2a3ef8059a8032dde773569d7a400a43f9d664dc64bfc771bb80129537d25c5286eb4a3b0de73e00575ed531ed231aed7d684c9c3df980df5766b79ef52b1103c87d709b83f741621d0fc6163ae488ccb773cb3b1269aae97039e65b9da4b8e0b705f10d3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324787297Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/array_metadatas, Name: checkpoints/1344/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1344/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=96be292331f044f8ab5493ef849418de6552d5bc7e2cd04a53b1ac2427301fcfc50d185e6f7b04d719aaec63e99248ca23b10d7e3de9473dee4f3ccfbbb6f06ffb72e4db47256d28c3a2d9d3f29722c90675872432b8d948ed06f17710b70be987149551b4c47286e83547eb394aef66406c7742ec2df2677d823703429838884c76e688edf8721e591b761410436b41e4d7d18a0cbfbf7802cfccc58fa65e9ff021159c20fc2fde108444f0d79b6e1a77ab4c04f0a8e1d55e332bdc2386603715d3c2db91c9315c709b003a4a3f93572a4a2bdc2a9cef11eab2b83f420424372f2769154b2d7fea502f418ad21e3f160d12f2244f2ee1611a4eb068361eade3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324792192Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/d, Name: checkpoints/4992/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4992/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b76e459fa159ce51285fa98c0467e0bda684c52cc022c9595542c204e816583407e2f31b6a5fb0756d577a5975e15ccb839e43377dbe52e74beb93245cdc9ddbfc7cf2b20332a3697a6f3db80c26cc463dc211a4359b3813ff0eeea3ec340793492da6bd2ea80391577a091a409a850e7202ee454d5fa269f0ad3666d676245da3fff317f4cd22423e5dd7196ed7a231c20b8faa82a0a955e56069c712bfa8afc9beeb29a227299e234b9a1680765c0ab36d6ed522ed532ff3b23a3c435bec10cdf39d9202b14a4ef56b12f3058f9e7d488903ec83f855cc98ae96cb79b33150772ca37a6f6b69212c9d0bf6369077af5e7330455606ddcdb1c0dd1f0cd02d00&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324796572Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0, Name: checkpoints/5440/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5440/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=9afa4894ca6576829f35453a99341d77d6d9b71fa7f4f1c010e485539fca96189edb0753b0c35e41871b93ddf1baea153387036fd8fe7da50ae3fc75515fa9f4b88fa6095f04b40734d56258f80bf0cece701037e758205ede1ffd602bf8ce472bda9ab04c0a4b5b3c950bf390601895bfb4750b0daaabc841cb7db84135efcf181eb7f2b02da085fe000a6a31f6c9c64156c37def646404c68015fa1509d788b221617e29c7d3250d4c6b6ceaf7ea948926b2378b1cb7c4740ea42cd768e81db88eddd82fd55b389b74923974318b5778c6738f5eb2c510abfe5627b6a6a137e2de460b69478e08f0eecfa2ef6c20e6d80fb5177b80e51c87be977e82037108&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32479695Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d, Name: checkpoints/3968/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3968/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b8623e86925d5c22901edc838ea910cf3e6a171ca5333efb2563839d4c4b4b153465c75f3270e97156b2d54e6f90ebbaff96a83ccc5df274924c3b68222551e9b388bb1063aa036e82796454e12e7b5508c4b52b75e701d6b51d73623eac15a9e6ded3b170b23212ea6ec38f2df68c8329014bb1204de6cacde9f9f827f23c5e4c3d3c3357c3bb438b2ec9a39f132a0831cbd1edf89d9e6ee4493ebdba61fa2c926c5a96ba00c08868d32d2dca96cb5c7a2a248450fdf6dd7695487f5dcb7e0719a5dad747bdcc3bdd47a5bc90a4dc1e20965bda6e112b83019c700cadb971965a2daf9b66abddaea83c240c1f20ff4a32094be77b83e4e11d135bff64037a8d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324810742Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856/default, Name: checkpoints/1856/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1856/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=487cdae2d04af788b962abe5655b1ceeabcf988f7b0135bc1301326153c767d958d4114265eff2c61663213943617b4fdc5e61dcf7ab772bbfbd5c980253eba93bad658db09dea37ead67fa70eefdbb18bd300ab461d395c927c61e37a99321ee77d5309f69341fb2b03fcd258a3a4a98f7ad782a05b7fe3590f5b4b20153603b0775e841ce27ea85a79c7ffe7074a1740653180a3cdcc4096941dd7a3e9557ed223d66e7bf8bb27d9ed5651e3ac7de08b8e284054603b93101fbb18129d5d9c3f1a2b35380ed99c0fcaa2f0f539320dd00a92c1a00584f8460ce91ba71c78077a2e68bd388360a5bb876967b6e0f0f54321092f66101c3593d86019ae294870&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32482095Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d, Name: checkpoints/1408/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1408/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=92589e71536f1ce245e4b0de16bc60997e48163feae0034f1ca8d21d04cf08cd48ab0dbd2baae4cb74906741f801adce5e785eeb212c97b3efc4ac233e8fd1e3f1879de8d7cb62bd2d9037087bd50a62306fed3975979eccbffad2836919fc784b2277582d65cace373ce3962e55b6c832427bedb079c8cfb90c200996d6314b74ced7a24ee1da0086eb79d05a171489ad90413aa96fdde130ddbd7e007755073aa959ae0dbb085d4c603e464d070561e95783c4e9af621cacd477682a886c00a0ed1ddcbf4afd557ad9800ed74e822f8abb48f7aa4b6cbac1e465dfb042ca5c16d2126e5b2659fd9e498ef68f04c6cbfc8a51a5df7792a7083db63799692889&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324836017Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416, Name: checkpoints/4416, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4416?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8720913f99a7da01163f8f2ae7b0dd7748a30fe18cbb879389b635aef152c106fec1e3375bf647867c6035d06088c44a3b935dc8cf803c2ea9fdfb30a26de156fc6072d57351d43d754d4355c1093f73d39cf7c658c9225a5b7b0afb811ad35dc889d2932b26394ccd82950edf14f15a76e310cd0c99153db8567387c777edff789015befbc3c9856fa15b88e5b81a40e7c26855053ba34021ded3843c5d2fb16fe88bd6d9564ea07d709ceda11dd33731bcd36f5b91490240db1c49968c443787aad09212950ff9d2f7820d0679c35e175c6b17ddfe4cd03dfbf0cd12a453370add3e2b0c6ad3c12419f7de0d1862aefaefe940359a6f8883c9d102daa7968e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324823101Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d, Name: checkpoints/3392/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3392/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=10213965b4ef60e6e180de9507ca17250e4455faa3fa7eed12217abb48a30cd1cce3bef8cfbec78fb2a2e837358d7fbfee61db917e1294a66f5a33cacd67df52a24c50c2262e334b4f4c4fe9ce484bf7a848afbb48b9c71090b7331157cf54ca5aa7a5529b3d4bd1ce47298e2e8b5af2c460758e07877e22fe2018affe5b47413380fcf4a54a603b15c8b63c3a436eda626957d57838d07fe7d2377f4b14d43754501cd43ff8265eb5046f6fef8368e0b9e4d86c8ab9acbeca16d92c09ea173c1ce0e09919572e795c2e922f0322c262d73b43416ad466d1951b1618e04bbe836a8d1bd5cfa1d6a8ebb8a48122ae3a892e938c6732f85a05647fcee1691cfacd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324847903Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d, Name: checkpoints/64/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/64/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=15626b4648ddee9b37e7cd16449d3f9e966e772b6c42e22b627ebf8b6c52fd249183416b27ff2c4a05e1f658399167b05fc39187cc980a3c9b6b8ce8c4bb55e0700843127fc50dad1ed5c08d4b7b490feb4dd6a86b494cd8d34c4d601a9a537cb70dbb83878eaff08159c3fda9106ebc7ea58d75c62f1b58dfe185b0918ed9f9c5171208fb583430daffcd7e4e1d4302d08269096020af1e67162dcd44b3559921c32bcc155b8c9f301fd7169abe5aaa054a4433c4370a66f2bcd04e22bbbf8dc269f64912030486da940bc8b83db094a7926f60ad77935e6d8b03b6331c0d550953ddc10db53a1890b9a51f0312c4ecc0b490de4a8f040f0db64028d7a96332&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324845818Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0, Name: checkpoints/256/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/256/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a816ae8d8c81e1c41db7aa53307602986869c595eeb31b0d4ed26bb702b9e1be6423b29cca2fb64a91a85f7870abbb95997fde2c0fca55b78ae29e924c3a8d47e81cc0df75b269bb14c7f8a9039a484447017941987876c350e32551673d0d1f5357f6e5a7b554fc0f7f5d276e09282cc02b31e35470e40f714585b819518cf24e2016b010bb61c97950a4a4d31f86d77b409899045a41c44ec1f7670b37f62a75929c3ad9508f0b2d384e85d0923b43970523728620f4fd76c5f9bd2891167d2adb924460b1df5b2788e994efc06b3d00b0bedba06462671c9777b8831dec7f8a7cdd22b34c7c812fbc77e729b3838594a1cdae3204a6f59f70c9bc34e52415&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324850766Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0, Name: checkpoints/4992/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4992/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=20e7f2b99bc865a5188748c111898303ded9c06f8d6fe84ba38ee555976479f80c932530edb6c20e981ab2c297f4ed7df6bbe66395ce17c6252dd037d02bfe58a2ca67d52071fceb27727a6080e1e1c2a511946a18fb624a03a2fe57c79fcb82d8e0f4b7117d07e91f425be337502fd8537fe58545e3e77523f9f1282a4330e180d9633abc936541d91a5464f95292d9ae20a7011d8935e3f219c831e355d23c3f30d1cce592860739a52aede66f2fda8598d8c602372971162fae1d835711c8f74d316f071767961223aa48391695c7611eaeddf371d8e475e48483a6573a0cf50a6e0623ae97d676916e901ec1418f2919a238724b2a6d751923d11c15e111&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324696816Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/d, Name: checkpoints/1600/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1600/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=7feb889fb3e14430ba51017e15ef221e4ad0e1ad695fe6838796e34707d6aa16bdd2e1f4695806a11c20430fccaf35e4f0097c8d2213cd65de2e0345cc96f147cbabb9376a53170ef43a82130b5310707c73b86c8d6bc18a879d0e88441e5757fbf84d71e69f55c02659de2766b39e48b031e2c6fc871bc777617ea60e8221a5979c309dbcdd54cf4444efa1854740e9f4ee4b0d7ae35dc703b2200d212d07054bb2e3ce13c782795dcac29c46df9c9bba344e2246bc430a13979fb8e45a5d68aabd5633a717a648925a00325a26e9b1fc7c75a62b18a2c0b5db120d96f1bc84250c81151bb7fa7f1b955b7d7701e982ec097f5dce0b611178e4308cefce99e5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324902479Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/array_metadatas, Name: checkpoints/960/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/960/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=9151c122013a63ecff7fd4c834c586add8fe121c117b71566877b42a356e3bbff1ed26211d96281edfba91a86d9f264f28f65bbb27ef2cdd155578638a6729bce0f2576a8b2bde71aefef450dcb0bcfd9bee7edc0b2d9f37709040f08ec9dff5e62ca7e77a91a648139984f5aa5be94b036c51b076dd72b781164cdc263e05e86eb1eaf6683d5310312ebb337b38dc7e114904349f8d5f4eeb05d249b874c2250aca11590726aa92bf75a2e8e4b27fabf8a20c483dc1600de7d46dd463cd0edf1484af7632c312d1c93f51c4ab07a1cd4c6084375daa459761a6d4332a428aedbc4ee72ae6910915005d6992f8783180367eadeb811a37f77ff167fdb89bb715&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324906762Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/array_metadatas, Name: checkpoints/1984/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1984/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a53a61c8ec8756ee68aa8f563311695d26f48a26115d6522173a4accee13f5a977ff0a27b04e2f2889a0cdfdd2b9b56ecf73105d797ee606626e6e7e51f3895cb40247f6b46b6b5d55b50bd6c414e9fbbdb5595f83e5778ebce61a1b09c3259b6157e6182a8b7f5ecfbf631e063bd222a07d31bd862c4cba40ef68b02418846a29f622ea8dcba578382819167fefa5632c033bfc7140f61dfa9c8039e546cdbe6bb923bbc8787136fc8c951e713c4f12ca1de992bf06d11316e14d6845d54040f2e452b83332baedf3b8b133b0f10459303ae3f211f52dcb3082a8892d623a10a0e4ff4d17a2d10d3cdb25fdb750e0f054b30b4b12b16509180920bbedb8894d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32492266Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/array_metadatas, Name: checkpoints/3008/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3008/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=9cd60009bf1013c887ed879dc7fb7ad822ba93e975a0d00df824213516c5a9c6442a57922fbbb54b301e5e46e05685b1149823ecaacc8448596bb855dfcd840789a4985e0be2e6f81cb17c67f5365feb6f9e67eae47f26a89b5ef1287a9f3d00482a2fe87c324c3691526da7c37b31206b878f3fb6fa531f3fd72138623901f8bea19a7e4dab7d82ac93084ed9affe4403b37e7026799483173df5f58bd448c2b0104dd41c339ae16e86a59327a76904e6ecf16c1479f94901b6056532bc38ea21caafc40dbab2ebf25f7c95db29d0cacdd2c37bd97e333fad014fd111fad57d11fd729eb709b8b450da0a73e5d3be8af7f93f51f7e97a5e05444bae2b37d4f5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324933477Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0, Name: checkpoints/1344/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1344/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a4829abdb43bc7bdb2adf6964d180eb0e7a17a1ae59d6f401a4049da30bacdd4c56efa5a533c1ad224d4ebe689a9a6b2510a8e9a624a24495a667fd1ad3f6d5426697511e37573f149f81806d621487a055b993d8703acf434bc95bd8015320ac7ed4718bf960a366c67c93b849ccd366de422a3d6e1c0611fc6fb3170acc2ecdb7f15e8886d4fccba8574002c5c4b9fb48a50e72ce7c1feb512636592c5769894071006c026926c4e9e5671b05a4cf681ba06c82914a207821ec40cc26178c79e6ff6cd525b8fb4125326d961a4b28cce5a56749a8a322033cd333851096f006535c24ceb3082970ab6465e29be0ce89f02a627dc6f2deb2c821c866c3638f9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324948198Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/d, Name: checkpoints/2368/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2368/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=5fd36b4d37938674f381bb53e845cb8c00b62943f8dc4297d9244f4e904239dc883ccd9041e0a5445df79d5c1d29e4f52abc58e0d20c2d9d8a96272740b9ac1398ec7aa3930657b7c507da64186b81ef6e37be0f04e1fcbac1324c3a434339e9e9540ad1b814df4903561443e46605d1daab7f40b54dd692d9630a0a40a667ffb166621c5e0ac992e5dabbe9bf81a852130287bb4e1467d5bd4d4e2c86541ae07e4cf0fb52848499cf1d89a28ede12d5caa8b3ffdd270ddb24aa783c84db24f402408455763bd20560a3591076b841e590af3f412242705eb9c6e6d8b35c9e8e67b02063398ca7cadad27cba5fae2263cb66e0ad8350cf16189325af3c0e3679&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325019706Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/d, Name: checkpoints/4224/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4224/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=41880117e877b4e5d368ac9e4e14525711043ce67fb2ff11b5f21360a90f3b0d2fffec35b142209917ff65f4440cf6c2d58f15e8bdcc7907cae67acd2a770ad8a3e55c38f9dfa35ad28eb097e760db0d6b47e8a5fb37b1c4f90dddf41def932603f302069c7858fbf20a2f007369d71f1381e7c94012e1e196eeb9f7107ff893ecb69b0a71c1cb8926f5bd3468e5901e948c4a0fdc126952557796221868752c2b9ec6e47931cdc62fac3081c32f1ef47b3bb5c0162eae4a5ff8608e7ce672e09edd5ee8c4189502229f70f8d2625b9c6ebf381d5c6258c7550d25537938be0f6e13bce05e7f360dc5cf5f9f0e15a7678ea3d01395358fb921b1227da74818e2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325007817Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4352, Name: checkpoints/4352, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4352?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=15a1b4a0a4831453387402f22cbcef1bb1cb1f08f9349bd6697fc71804ca6203d6bc2b56911bb5acb921515b3d9293c4f6318265f8d41c1bbd8be07b20e3b190de3cefa34bf7e55371d4dafba94b39ab8db326e2aff721a5157546e99220e3df7d6070e110b13aeb302691fb5b7e007d17b4f9f83c8a0eaabb00b58b6eafb98ba4c13183bb20c3336b9553cd703870d5707c74437cc8281859cdb8f3129c4b10cb142de3faefdafb14764bdf0f216d6991d865a0d783bc7d45df5a86500c411b8c832fedb4aaf11bd690126fdf8f9b631cbd4965b2b9093ae8d67a0cbca4f6486a9945d9147ce49bc9b2d7aa87c70e217ffa0bbffbc8d8271488827e5af518cb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325022478Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d, Name: checkpoints/4928/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4928/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b760eb94baa8e65cd6bedfdb9c789e82ed921707d68c25a883a50fb28956b607ab97f45dffc6f290badccb151c909249e9a851c3f1dd9c05ca48fe505caed2ad91c1863cccc4c6331afef385e38c05e538bf95f8296edd6c4416b23cd918a562286eed1efa4304bc0cbcf44876f75aa185bd32e3c5e05e325a1d843fec1106bedda284746502f22f2aa02075a9e1051b1671b1ae2c30273010d8f77e759a7d7188c74d6d9fad0306f37c78519fe297471d77cb1fdc9f06df89a382e2e4c68be722536ec19f6a750f12c5bb9eab469665afe39503176807f7267bd4a3112da50ca82a829022e9677ef5a9d670591b1b7d045710efed60a16cb182d9306344adcb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325030724Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/array_metadatas, Name: checkpoints/2688/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2688/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=34d8b1228da0fe7efe7da30ff62403c4935ff3c3b404d654b530c8c0d65370bfa61b9a2743af52a80d9225cac90abdff01f1f081e9161898c19193f7f7ae38a75cc2cf69517b3f9b3073fa5635e7385b131622abb9bc68956cdf830509baaf04081c54082380ad8345f831a6110781ab0fcfe162f283362966cfd01b51aa4935cc269d1dbbc746a3ccf798436d40a24bfbff67a668047d54a09fc4089ea5d53814c9eead757f2aea9bb293993539ab644ecc016d12659039ee61d295f6521baff909d1f59fdd875895a248897e474ff9f7adf11607b1c43b73e9b8beaeb44c691a189d28b8c92285fd0dfbb3b40c55c84e3e4f9cbcbc0d50b0f5d898c8b0db6d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325106449Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0, Name: checkpoints/4096/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4096/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=95dae2274098916eca6049ffa1cb432d054307957516376403db0b5c44f2404728a29d6e3f482c627df694f13223a8e16eeacaae1b728a73e0b58cfe3a817754204314f643a678f425c31ce06ca6eb891884474e7d6add53e6952a84cf7c93d463fcec64b82a5b3999ea52188383bcd46c8d05290637dd4e419eaec16fde121133ca48075589534d06f66d62006e5581e6ac93df00827d5e68283c8c16b5c6d5d9330645de4f91443c2fbe2d87867040c51dca8da0353da9d173e7739aa2b4e9d084dab93bdcf7b8c85fc089babc7188d063e774bc840cbd0ff1b053aaefecabe2838114efedb1e22f180847d49fdaf1fc38fed79589b4e027903da186a0f252&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325119798Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default, Name: checkpoints/2496/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2496/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=514792906e6959d1811bb0c65cd4b2e678801e5acefe1c4ce53f1653e98f58306e4be15b9fec579ca0b7db9645b76c5ef4e3c3a0bf1167bd748cf643a8274f6741ebd3ff07749cda29cd5b77136ad272936003eea1fadd5d90d78306b05e6def542e1a94da2022a4b28d04790eac38724f0896b0eb6b6768dcc4f226356d7e00e25f5b190b2feaa2eec4866dbf1ef0b954ee038aa49151645d699745f20ee0e5b7c7d7c05f5b811e2e6ab58af27735ffd2e0a437befd6fc0527ac67d075aed375bfd24797630ff7a8ead9db507c555f3efdd33ce4cebb8b060b4ee0efe8392e886f2c7d2730601379fd053876c8684abbe633306cb66997f96757332bbd47f73&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325048799Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default, Name: checkpoints/4800/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4800/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3e6d7518c3097a1de32cbbf8974601860e5ef0382652690dfa7d44a7fb2ad2bc95df4231b815d48c275b9c8fc79fd575f2f3e8a3acc919c97094b1c48190647c4f2305e1564740b434ebbd4cdbeca84e1905d5e6f634ab71afcd6f60d05998ad92d267fb04e6d67f89066c926674d7a945bd0da7d2f975819ad49e52bb515a86e1e5468ff2bfedc3635f3e642a127f768dd2be47829bb6e6650d86fff75a753ced361584f82efe79764d4dbbcd8f36da7d0553f52608f69fdc0eac536871bfba02a8566272b0804f21797fc3ff5b49c3ad6b49ad4b1edc2808e2dd711a0e0795b366645eaa80227a60d3dbb4b680b3bde0965f730dffc601e6a5607295e082da&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325227182Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0, Name: checkpoints/3008/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3008/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a1b7f7f693e99b23a84ed35edea72e17cddb391052e6b058fbb23363d3bf87ab78860155b08c2dc55dfb7cf990c10d7bdc272813b440ddc93016c3c433c78abfaec4bb6510c8641f00b61cba3e1a1d6a7914ab77fb2805940b93f5ee93d82c5444fe481fcbfbd5a71d3f17c3df1ea821edc6a0c952ffb7fa355f3e9b367ff029cc61962d4e211e09fdcf089e589e2a4875143e708e9b2570786b7450e44e3a666ab543d60f6425a7b517a207b3a039802abaabdfe0f133ba334ac820a11cd1d9e084a725d0313d50e33b4a86a79162d123c17e85ed1c166203ec3070a4bbeb0ad7152757ab453d45ebc1fbe9be16b650877ea0601f1b5c0800d2bd0a26ca18fc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325261143Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/d, Name: checkpoints/4864/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4864/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8f78ee460f96a0c44fb1927df16aaba33224a5916ff9c2ec71654c897cd84113614f9759759ea1478f5d3709e9b9fae0a96aef0dbf534a4a63f4d6594222c1039dc698c91e32fe3e95b969227e3cd84f413d715e64fb4e635327ae05d99ae5bd53ac56d3be5b1097fdd3ccab8cdac2ee6dd586fa1b106ce504a579f2aa4d76cff8719069d79eb2231046068b11404a9e80e9ade47b558c54b2092eba7004b3c561b6d44546bf136bf195606a11b93e27df6eb25d902bae0ea8c787c2ee5631a133188e3f143f16448b9217f65e0993afa680b628d0ef29cbb7caa8da7c7e57e3c57052bef1f361f38cd2877013d08c65e3ba7b4981f812c1bf2b290477df9f45&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325267061Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d, Name: checkpoints/4672/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4672/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=71b356ebfdf603510783f44261d738e949275078571cc0b7e766e5f195f62e0863638923109cc3e8c105b509f2ea75fe558dfdfbcc38f71c5427da8e30ae570045cb8d75b560235fb91a664b325a99e9b35d3df1d5c5003da9295dfbdd99de25040102a3be925f3fb7206236ed966a053ee826719df06331d18ee88978a1552a56185ae512eaa8de3f184315506c96e7577e4c5a2a87b6115ee2ec58eebde494c9b8f27d8e75917d58e15fc5763b7d6e22c09d5cd64d6703d82648290d9ca556ce6360d89f48edfd8d0335562642eb9af48cb746a91722b47bb3aab50333c7396f0cd3660c8b9907a1edc7eae7542b13c1432f5aec4838d9091f7d4b0af77b08&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325267593Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736, Name: checkpoints/4736, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4736?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=c2b9b5e427265c56403535f712c6b4748aa9be8754c41c4ae4f7f71f514df29075a45a575e30d16f02ea086095950ed8375b8c8e14a12ed2e4fabdde76b340d22d113b9c96acfd6ff1ac51c1f4b753466a37e9eb2c3f3ac4af8f8f4d8f041b0fa7a384d93f394d4b397605c3954f60c9de2d850d4b0cfe663fa041d72fec2935bec05e9a4249f11e4c514450b5ad8d729fda58df2f7bfbb6f8067584b43a0f71aab2c8e7d9c547e937316c5e96e9cae791fc8e5b496e94b2ebc1a812542ed107d9a163da4fa08c54a9f4a326edf98f99ffa3ebce458dad79b90ceec4dd87f1c9861e30793726c5d80fd51bd51547fed81b153fd0659925df782f0735404542fd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325279996Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136, Name: checkpoints/3136, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3136?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=60fe16ec59d4d36744cfb834a1be6a8b4d409687e9ae779eb4f4a66253487da5742abcbf647a124b908c30c96e18e22e016449d056fd868790aaaa3c90cc60f94cd862ee22021d08f6402d9940296fcf0b55f8da81f1e03a8ccc7cd042ef8448e3d69558ba87dcadd32c3bcc41a6ac373bdb5a15082d5221801b48261edccfb0530887b945413ff68d234d5c49cc16a7a74495031794d9b75d6cff497dcc1972a23492100de0f334efeddfac24e02325e83aa61f8a265bc1e729ab3fc75371fe05525ff79c05d9fbbbe20ef563a81f2fe6dd42bed0ec0fd88a9f3b8541e0956c764aefa771c7c1611dcc0b98c19e1796e7ffcdf181c91e80dda06d8eb4c6af00&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325310829Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0, Name: checkpoints/2048/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2048/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8b488a7f0f3d1120587530c129d71a6bb04626ed2e005458e31a438c899f7aa8c490e7a15679105009953572b49d2fbd8d3a04e7afce3d854b360a31f4ac2ad9b39192838f70db9d5f4c95bb5d7ad70ed4034494c8616a896717e78aa2766556eef6eca1a9d8aba064aff06c804fea6df1305178d4a7012c3dfe6605497128fdc98fd36ab8476faa5859367a0338fa1cac80d380ef1a860237c362737dc9672061c3a4ff500e6362e6c8d313ed067f8992dd2e7afdf35219a975cc51574027a1bfc6667c79e7f85de202396ab4f41b7ac88bf53db00e7192b630eb423f3cf6221598e6e83a9a1f7b17419da1a0bc9164d089813d49200539c9a9c02c1ec94e0e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325290458Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/d, Name: checkpoints/2560/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2560/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1cea31bc08663b999cfc2eaab50beff8d31957b9abbae060f1f1dfc050afdee1bc63c454a32dad7ac7b92a0a03c9f0cc16961cbc905034a17e1833e767281464e284e398ae6528148399cc077a553d3ed05394dbfb76e884207d41f29a795eb4c3bdfb74d11bfe22efb183ea6fbce35aeb103aa16ad91cb2b5ab6ac7c587235725d2136a2f200dfe8b208ccb6f6ec669c257226d10c06e59ad3f904e7d084d09117d4c4df1b5c77ee573a440a45519b7a3d9747a1b148726b77e84ee70f1b46522ce483898f3048920854bfdda023d9a9f50f353b51142c7071e2cc472f2c685f6b0252bd26c41cbe347136f754bbdccb593cd54465b8628616aac700b8613f1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325348616Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0, Name: checkpoints/128/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/128/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=ae24527a3750fbdd48f5b3e5d62a5e233050358d505a32ba8ae445625892f2873193bdc1c392ac39a35a4c03235f18d6424ca281386dda1dcad352c37b204d9043c7402d3d024e66ddec70ea97a01fd2ce4de8f6bb5d968f3e09ede503894ed746951b87a01b089824e7e744d7da92ab1b47fb0dafc194d6ae94bc10a800a41685469de994f7ece54b837196e01a7351f792baa9b1fc7b2bb822b04863c1b99974b37825ab57bbd9dd28fb6d454cc6f02df92af5bd0a9b9ed0bc45c9aaca2843fbd24f7cb0c808fbbd1b491fd95f4e7e25648b39cf0f07d415451959b97305cf2c7a5a20473ecb6290243563c7bc540a1f205e29ff88941c5a1f8af38c7bc784&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325367663Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default, Name: checkpoints/3328/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3328/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1800fb910f1acaa89fb3b6a97dd77828fd49898273facf3116492294428676d44d8d2d201d4a3319d4e01562759b9ad9e25c30cb10d1bb95ed40080a1363bc857395d5eaf0958301c465c6e5f5fb108e2c26fbb4d5ca192e778e3c5acd319dd4c6e3081f4c53d0b52511f0388c1d8fa5d2dd2912ad1e714099c376736fd8d59bd127685e38fd092c38284957a270f48699fafcf8e32e46e868fd4af85b9876bfd22c26b62539da27e679efd03574a39362fedaee3c351ee058c4b328f430668a61fedeb7ad501575749fd4556962936e96d9f12e341c0124c28a43a2ae905bd0b817568540106dafb5084400cd5d3a0fc82ddce0305b1239b3f588f0b61ac79e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325369324Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default/array_metadatas, Name: checkpoints/4992/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4992/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8114975b386726fe54418f1e6646bf3d59fdb608af7e39f64fc78a1c8fac83c82741a45bf980a3cf9e31b6fecd830d3cc09e8e66b536c8aa475d30b113906d129a808ad7c156b4bcfdafda817e0e8662e81d527c1c92db4c7c9f4c91e0f1e3e1833032d73102f89074c472b7c78ae4eb787cd422a62816258c5681a8b0d887551fdef34e78d228dd6e6b08c4bab9d67bc4c5ac2d60a3f0821e7725c42a2954a362c1d13ae206f8f8718bc60a96ce4266f28a9d8e4e7ea6ee9a645aa52ece12a7722e789d9be5c30273fd5058899d07b9ee64dd19c19ed5ad9584fccd167a55d960a4ca430e3f3cd2de434daf372083d09b955628343adc33c8bc31d236d2a824&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325366534Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default, Name: checkpoints/4288/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4288/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=4af1b9e6fa1d1452e95bd99acecff12fcd1ff8a09cf7148ca18de259f05501be5edb274c2e68016742f6e14710a79e9f7c45a768c4f648519bafbd4fd69f5e79073adfa1f39dd651bb9656f38358cb4c30f6aec67cd2ee004f0176970bc4e8d76d200726cc21b1ef69e72773aa1f76399b7b97f8204fa6d27cf79b4a369f6b60e10836e94d30277878ac7a8b86ee7ce968f4f1ecf12675cdcd523929f99724d666adbc6598b475a59dc10d4c8ffcaaab8e33697650480081a58f815cd1887c462a5b35ba2ddb34567eefe33da3763b0f1f6be2e2a2a575bd8ce9a458f0c56ae6d950bb9160d065c4cfe04b68898e279c6817470a2f7faa07f06ca23d3c3cf194&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325357499Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default, Name: checkpoints/2368/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2368/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1173e42350d2f751d620c57b7022513d86824f8112f1df7bc96648735ee3beb7a72189880c6098c5f1f4d8319f9a59d291354e27ea4c48d99b7bd9b0f95fa2ef967deab625ffb878661c21a3d500c80f1def26bcdd1c4c79242b9a19ca781cfc4b8f82d6c2ad065f1f33885254045958f87d1ed4116eccbc3cca24d5180d6ff2f1390666117d6660119f0eaab598af0b004fbb99cc475f0ca6de91d5ca9c2ef219518957a915ed8bd4867b31beb759b45a8174e714df4b77f686503fec20679c6c3b4083c9615fbf76ac40f414f28641a14ff53e9e1f829433b5b339641d42b00dbaf716e002c467ea81fb367b61e542a6abd8b1bdf5ebe5a172e9f0e31e87fc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32541763Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/array_metadatas, Name: checkpoints/512/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/512/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3be6929e61aff78ec9f2fbf95965479c7ef02a887c9bf7bf0860e6bda905b523c6951ef12f2a1f362aa0868d6337dca3f0424d4085a7f448898938295e608bd24f7fe2c1c65b46b23d4b3befe93a9cd0af4a963887efd4fae5530a8b03bb836b422097e5f9609e3102d2defc40454c4056c355675fb8f947be12d292e8bbf9898a7c26e741605857bfe14f914b9ae5425625247776b560671c96b951d2d8d9a3915c6313c3ad8adfb46463ccf4c7184334105c3bc4ba52099f5317a5e361ef2cf72e7a43f964120a6e1b1ad1b7a93c3fa16a39b90a40544089fbe118f8c5ca7592c17b70d7f4c8b4264ddd63090336f4f6151c221831522d7e5919aaefbde2fb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32541439Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default, Name: checkpoints/5632/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5632/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=6c4bd2688398a0dcb9a6d9386069a4fa14438cfc97cfccfe606ac67d6b5005d24e928e436f05c948f07de5186d5e0a839972ed1b41fdd21f562419fe4cc13f82ed9b6af3319ab9f4b0cfa990f91dac2809c990b09d542e2d07e26b4b7bdbd0e240b5a55d1a98cd2102f0dd74efadb8ef45df3efe5170ccc6bc1a7b463d40efc6a7e9fd6bc5ae60489b453ca69b293e89ca0331927a7b535d9fd481efde41e2aef131eca8ca3e7affb6f119a2d9e52fb44a213cef9fa21e020e37eb91912949a8764e5a77f25a2e75760d564207ab409edd918e495934b660cdf29265cb4d6b5324571e125ee8038cd4348faf49e0be0736c89b18a802484dd2f12572911f6b74&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325429543Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/d, Name: checkpoints/960/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/960/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=21d561eb9d305a76e95acedd32549e063cda7b28a1aca472cfe0a7e5ed84f2a694fc1f90a27a6d8f9b91a6183f337f7deb83e32496d35158f2ee3198ed00f86aab3c1430c2f72c1a4eb9b234020a7d31544634759f25505cf6083625be975573428fec142a093c634a440abff7afac8801fc362cc72aaef8c46dba3ce4139e7b3bfdee80e74608b175be7580b8bf94576081ba9ab52d3b16c7bb72fc0d4cc5d406c6bb16473f4a0e490fca25ceea0e20fc0239bb1b1cfce79f83832da47da670ebafc81df4a79da6ee189ea9ba4a26996f1e14abeb149e10f74e02771431ad97f66ce211af33275ca539dfcff13ab5ccf338155fd88845a4a05f8b56f063cdcb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325452647Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/array_metadatas, Name: checkpoints/1280/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1280/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=ab52c0c0edba8f649322ea52ed4802d10f927776cd4122a3c1e87c7de887609520de451f30b5732916462ac26267af6a05838057c738b351cee79a5c62aae8f1db6fc1b3ded435b5471e55e6e5da35c93b611716f652cf3f76ebf5c823edf21ba9d7e53862487cad81a49b6d56f8e336794441f323089ab4df761e3f390d274004a6256f775302a53c5c6b55f4b4b3a69f157181ef7fdba40a428f8daa7686fd300837d85ba90bb51a4e2762d0f12bf9908315570131311f07aa621f2f368d611d785debe920badad91f74a83ad770c2b00f5c9b9dd79c503cf892ecbec859fba3e7d41caddd993b60cd4c8e3cfe568a5ad4294f1bbe4054cfd81f9e2e4dae10&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325450666Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2048, Name: checkpoints/2048, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2048?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=478c7985fe1f4a75333fa33125b918b45b318aff8edfc29a21f7b13c480c2b2f4dcd53a019b3ea9532e1ec660fcd63a6f3ec868e83afc46b05bfb2c93ab6e88083a5a53bbac9eb355f0d0e68a971e27f93a1e2248e818085fc438b5375e21c41cda4d3e78ea0bc1df5dcd200e1846fa6e84d27d88dd0dc30b816111fcc0c0f3e14b639daa5e6adecd8db0cc8ba2cc898ee53b7b0bcb9ff33d26ea34498c4897762295525f833e947fe0338e5f4c1baa810cada92a9aec1bfc32ce1bc95d5d07300a09b1e666f48e5726fa9c9d77aef70d1e4bd78329eb55e1bba7f8f4fb9ea768fede9013c815a5e28a4ec704fb4cc7e2442710cb64e4f660f19e166d90e52e6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325484699Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0, Name: checkpoints/3264/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3264/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=24b2d1ce1dd193ed356a10d3121557490d9ff914f2a313ae3ebad3da547a31680d349b9dffb5b8589150758bd5dd18b9021162cbd1263f8b35089d25f6c413b9e059076e65ab26c0094d212c65c96e9324bd931f7749331a5412975aee87ea849397a49d3301c1e9b4848693f8c14d4a2b054b77c7fdb8315101af86c56b14a5cd4aa8b7dacc6c4694454cd88bc837c6c20b348b57a09125e0c45b62d1d8dbe87a291e9fdf4bbaf492d156fd468b60c2bea8f1f138f413d2ef039fc5b7b30e6ebe2ef37395abf5b0f558bc751d1bf76cb0de79f14197b50fbf66bfa572b6e04ae70f646bd7d388d2f5101f0f1e91009173da456cd997e67998445c1ac60caedb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325513611Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d, Name: checkpoints/3200/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3200/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2e74de2dfe776a153e6d5bb8d41ab0a16211d1e9a1c871754125e7b017d7a0ae6837fdfa5a0b6ff3bb6f27c4945d5bbbf71decafdfb59666bcd2d799cf19b68274e134eb135c0206bebb10807bd775c619323b269cfeb1f1d497b72c58b98882b4780fb0c922d6ce1a0563670454cb66d0301667cb65c5443b47afcb5dc60fc51771d0e834fa540759d983d6a58133122b96f3715a46b9372cc74b93813aed5a2e64142cb6019b66101e7a82d3e9f96c40d566f954c15ba371df5b1430668a8d1df9bce78bfef5781d9e6d1d8cdd36e45d9518a3dc30b85d4ced417a21dffd0df328f29ef625efa2624732014860a15e9a6b30015c21707cacbcfdc4adf5a1b2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325532399Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1408, Name: checkpoints/1408, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1408?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2f10b228f1a04742956726653587a9bc3d2e6207f3b4a9f2913efb4ee3d2dbeba2691d8df738d003e26ba29420255d1c740b6c91252effad4170feda41f59bc36a789b0b62f5c92f67939de275ef1f95d0a2cf92f54c3dd72eead4c08c0f1f9329e0b49de7510f098183e28adb4af81747bedf25f7d0f2be37ac470bc704724b1d3c3bf4a629dbdb899b5ca1dff8ddf3129c9b3032ced935a1ed9787ee9bbf6508942de87ac1afb6ec248b3320382a983c6dc742004da37033aebb6564632de1a7de310ff66f02b356d19be3924e93e8415431e1e790e1e835b2c1513642b846b9d258a23d0e752ad64155e6df4af19a9cd3e962f8f52e535d3892891a586761&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325542265Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176, Name: checkpoints/2176, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2176?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=552798e3ae3c48415cda0ac4f600e0637945e46ab9e122e161c11cd52396b7d9fd83b65d54e4dd538fba8810487788f9e3bdb8a3811c899113993f5ee081c5b8f87b1e6a58b9f420e8d04c2a21faab4ba8fb077dc16c82f0497c7fde88bfbe82c59cd8b7b696203ae08f974c6914d268eb7bd03ddd43cfeab042986602af2fef862d904ed06c3606283e87b7dbf7e63bd7d2cb40d9c66eb01aee19bd628ea3da9a7eb284abeba1eabbb26c335d6f85cfe3c430c003e7b9373cad4ab17c8ef3f3a0332384c60d599b42ea8619f4a897e696a566fc660f2e23f9b6a7191fe835b722b5cd8ece8037708c791603b3ea5e7dbfb84038286aa23c62cd83ffc2716057&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325596324Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/array_metadatas, Name: checkpoints/5184/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5184/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=18824bb233f809823f7bfb303fb26d599ef67aabca692c8769245819829cc18ef92ea6d13a7472cd1d1f183a75f260560feda137c274a388b989be01e1d3542c71253867610e5ded2af51805ef22e7c9c5dc1a0d14790a3a3cc69973d7cb7c474305ab8d051757be70cbed1d7235436cf20245e175a0eea6e27e5aec61340c8235d48ebb79a19cd5ff27617b8f602d027b3688ab4edb6ebbb27dc552edd9a408f82fbd043b7d95051093e90f09128ff94b686c4fb2d279e6e5c28273197b7d11d058fa0379427e19d29d62220584ca8449f1f0301140b8e8fb8f71d5ba31cff69eeccb800ea6ba2a3abd090e7546719121490a0eb81fb7ec33620b5ab0b110fd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325595266Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536, Name: checkpoints/1536, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1536?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2ba7b3dafb87dfd0040e811aea99bbe72084d4687d63b42c6bbb33c67bcc2d78888defd55260b72162337302295d46786a7f6c8848ae04c6ed7d83cb96ececf603318e9dc98f74bc4ce7849221b7b5705cea2288f48131d994e99180537d84aaab5205bc8fd0f346a9e3b0accda6cd449759c63242f725c07a19195ddbe951e51e27cee3e7a321d014cd8b61e5b985d6cc143bd0d2b83efb688b6a7aff9ffb19789a3e1ec689b7e039d5a5bc922b4596910b8db66a353c1822346a2056018e0a819eb0f72e5d8ded7a194590872667d9f6084484acc44ae2f5c5561383ea82fc2d3ce5473383aea06d9380db5b2038a44fb1485ba6de5b4f7b92c2dee03c15a3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325617323Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default, Name: checkpoints/768/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/768/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=5a407bff18a4defa707495bfe49d90c90f2b7fd14e2233c6ffcc4d40aad466f6eef32e145156c9cd98c53b8cf6c986743fd726c1c53f38f3760d2edb3676de80fb530c569610d9cb5701fb8ceb47ad2702a0c0d932eeb9f9477aa742f3196b0efa6ca046a92ab579f6c017b957d3fd46816b32ed1cd044f7d7b9bdc8535285f30bca5601d92114b5eb83b6c48e70cce970572cbbc583689fb93ba29f1451b637d73c1df92758a3f66999d1f6f53de0481b1ccff15ab97cafb96b5915a9d2b753ab42c0bf3635f3502b7ebf2f793273c724dce880c9b86339b1a2288cd3dc01ccd8be01d96ab2715711dc44a0e0e75a519c92014d2220a3ef1107e481ecb52be5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325615238Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/d, Name: checkpoints/4160/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4160/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=9b1556578374a9215acfc8324c7e7a620211bc77085a1c418a60fffb1ad9ef4f57edfdc4e1241d8bc35781d2b616b0ac00aab315868d5883431e9cad2f729717356e3e3f214e0f044bfda2b841873bedc585fb3df8f03a99005c772b7ba7d1567be95423b05f19962b85f9a10d69f4e5aa7a68ef545f7415a93d91f327929b2938985ca31650845efad146cad080b6ddf6338570a191d1b2da857e8cfba8e4420523431c69189f53cbc0917dee1ac67a447c15aa48f95b67443a99b1f1dc6dc1a737484487e7ee8fa9343b32c4b785c52d4f53433c2416e4b54f765c8ae38d4b98395cde3216656f3a76e2a37f9645efeb5796b57ab7b2d0bd178bf375d23766&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32562906Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/d, Name: checkpoints/0/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/0/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=bfba865a21998f2678fbc1a5c994f8e9120cc93d2ad3531641e7074bb40904ce0f1c157336819ed3139a67b4bd5900d8158637b70a104a4716eeacb78fa879d22b333a373cbdfacea8633d03684bd991920f5651932bb7d66ccafee1087c454a6418962bc68d12c43eaeba101e30b049109547d58ea5554de860bd7fafc4ef2c7db60b50ce470860e0ce4b8de41ba583030d91d73511302b0fad3c107b444cd9801c3b13546d5f204cc88df65ddb73491619456d98b1a41c99af7ae657c13fd5e7a44cf2b2049fb5388ed7570a32ef579e4e9fa3b705f3a752e9e981877444534bf71aca29d79964d4e8f555c77fb2a5a579c54b33e48769c8a5aec21dddc667&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325697716Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default, Name: checkpoints/4864/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4864/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=9331140129de85f4517d10fdc5fbdd417794ac9f7153edefb52d516ccc57ace80dd9b5992bb26b19aaf0ba003d6dc4d16b3feb315a5a449e2d07d7dd3cee554c99ea789148d010b4ab1fd156d2da99d26627ade8c430521e8be55e51d976b8a1ca34afbb7f12bc81e04d5faf4dd3c652084a82abdb40199e202495e90381460d4f17d0b9ef6516f3db89328acd434c1954eda23065e785fa3d4f09da15f162aef76a80f5a63a7e02036b2c6b819f5c11660b2cbb7eb0372d5590b3cc0ad05353fa4ff836a5c6ea6e0363d6ef23231cc4586943e53ef7d517a02e0cac34f69a4cc4f676bb83374d4ad101a6d26bc721485fd6e11b9d5843ef73b892452e471ce2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32574755Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default/array_metadatas, Name: checkpoints/4032/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4032/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=464837122d56829f5385c48d03448be4b91a571e36415221d759124e2f6d928d34c69d42233c43963a0b3a709b22d9c302bd11e4dd6dcbf603bb74258b155f7e595e56f1eef03d445fe6848a3a4eaf069e390385d1709af615068659c914c6d623cf5ed145016f9f25b82338406386fb0e65b89ca6de9fb215c4624ecc116590ae2762421ddc77935f472065112e8cbbd3bd2c145c4dce0ed3af98569d5fb2dd4a2182a430b57c7e4cb1993507ff2db5f5aa6db9131b0c086ef410a88a80374bc9f66dcdaf5188e25dcf7161556441370d064e8c6b2a5d4898838d248d124c7f8a828b88a467dd2d053bd476c18677399a6869cfda8bd388fd4de33a428ae922&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325762856Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d, Name: checkpoints/640/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/640/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b8d5d78d4b036b9556955d87ae653bcf62504e6292bc6fd5abd8f8e238be8e091daab73afc33da4c90d83f0400855f06186c38d78761eb567f76e9e9b366abdc4dd2a88678b468a58289305ace599ec1e72045bc6b48f1c5efd50c247c3fa5c4e778ac1bc3f923e3afbb67ed40dba91cbc8e8642948a9db864e5d37228bfd1545d3f2b6c648eb5743e1d9b3cb352fa48427ae65d2ff1f4038a3cddb9af624aaf072cb24ce19a897e99cc4369ec57e3dc8738115fbdbfff533ff09ee5359c19e3958a06d531554cfaa78a491f020e39cf0a7ff78929271e89fdd0bab6c62187651c85770b2259a015c570ea1f2570d86eb0210fb4dbd58bfdefc7c80c354f6f0a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325767096Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d, Name: checkpoints/4736/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4736/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1521699de2a0a309de8a4f61eb50a7045b26003a411154f8547f9a73bde47207ebb04085e5a95ddc1e6351dadc10437fc6e1a224f3b3004e60b8d3128a0132926a4254d76cffc3139998c802675bf1db106cc1fb2dadfdd0128afaac85fa17a6930420e86d7ab7c21b7bb113b5a173a7403e9ba4320389e581d1d03deb5ccef254596bfe792a64f8eda0ad56723c06e4b3b349de8e386359ec8bd1555e0c356b67d4a8a400ca1d5c839bfe24032e8b6e1a525c82070943c261f2bfbfbef8e85bf94cf726b63d1982094e5c51f5b5ef64cbafd920ea7436a1fbbe28601748f2db70e2b79df706ef660bb76b2eb81b6c069c6e890c0cc811d51f054ba49bad3a38&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325788981Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/array_metadatas, Name: checkpoints/320/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/320/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3da080f6fa917e0965c71036f5d3b3a0bfe4c177a3df062e851e46a47ac3398621769e5d25204697562634c231ab54af7c59d5d63d6769a113285460eab443e8ac9c19731374e75581278ba877bb9a8b455354f44c0f7e217095f7bdf98440aa63aab0b054ca0f83c7c02bb32729b0317410655bbace898eba9ca29d73baac040795cc98cb0319332d2bb6092a9f4e523d899f334e8d3f00856479f489a39aa440a86b3e49cdfa178bfd39fec93174db2cb3522733eae38cc96480dbc84a92ea13847516f9e790338dc78f9f335fec7aef9067989afa60f7cf60f253bd9ee33ba582825957c71882ba7ea4ba5b6149d689f5b1601e7b6a8410a3de906dc44e15&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325819423Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/array_metadatas, Name: checkpoints/192/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/192/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=9fae2fa87c14a6a8f3f6c2539aa498b1ac66c84883eaa980c4eb037047d83523f961b683af1a25a459760e246cda9d7f54c1bfc03f60e3a1a5e049344337ad9ff3a201e9bf4bc3ca8a386a3a26319028d9f01ee27183b51b3ccf943016d562fb2c397129acce494d0159165c05e2ebdb16190a903ec9c3240345e389ae352d7a0d165f9a18e9e9bbe46e06a4200c918a41ae9bc3a280c8c0d437631fdd4a92cb59e5612e04e5cb58724877a7dbcee93353abb5e3eba542a7ae5c805215f56f7b3e75c68657cfc6a9c885f2822604513a0eb8a8760f6c8e9dce5a8342154aeabd5765d2e92ce559b56a69c77b0d5bd828f0d2d8d68c08c1d4693cb7686ac8261e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325869919Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d, Name: checkpoints/5120/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5120/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=4eb02d91c724283fee6a798fe2c5f87f39b20673c369333a7cbf03c9ea5eca1df65c589fa936902131979123a6cee005e984c90bf253a9d50b90d4c1342ebed0e3bf94dbc06bad1b348b49acca4403593531061766ce8d3b5641df1d701514497b9ecaf6b19915676b9bbc151001b3f1e578a1dc4fa12f95bfd4f62d4ab1b8e3a4785a51dd56729b2f4a0b290987c6fbfdd70f3bd7a7e99f0a7aef23072a12e8ce9b208460544200bf2b43d96173ef57dc6853bcc8e5a8139e7f6f72f811247d3820bb0e92b41c05cc0f22db3716df8c8c5b03ca2afab1138760c9d81472ec69241ff071ae642a4ed22fc6c05af2db2748c644f98790d95ab5de751824635cec&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325918062Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0, Name: checkpoints/1280/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1280/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=661805b028ece2855f7a276527d16867aead0f3607919d44d16a7a997b6b2d5b27330893e6ac96123075cb73b1007115b6ed78faddc9a594137c1a4cffdb9c893cbf6a67916a63737787050f103f9e081fff927f50cc2a5c6d79de50a83addaba3c7bd7ee3aa92c2eef116c63efdfa108f5ed0e4201c5574d63c3ba05d8f00afc1573fdf9689cc3374f49e4bc1c23dc2abe22abf25376be77aea982a1b19fadd8a851cd95dd15f89db99232e735fa3e7efaebb0ceabc93494394904a238b9bd12da1dc69db97520baa3bbcb228a5fe9c72b606be5a696185adad557fe966e3406ed267c31e0919e7f87f1e5f8727eda5d5c7e52aa4c87f2991231cfb002b678a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325920214Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default/array_metadatas, Name: checkpoints/4736/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4736/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=7098316ccb1cdda3df44930dd2a1618a516f8d0a04e5f46804cd87fbe5c4902429f7b16e6486017daab136f30f7f1241ae1479d5aa31a206b6cae342289463efadec5124085e8f7146027a4d4bb5f904f5bd4110fb0337c5e19d2458ad8575e0c6432f2b38c667cfcfc9dc88cf5899330751b1bcc42837d9a7632c31da487368a5c9367ab655a04beb067bcf13568c29453548c597d794b2c153c23ea51a68d38c5067084664ba1d0c88fd88c57b7929b15a6fd4a51aeba7de8ca5093def518efa7f98c01cc94b473aa8915b09c5603d0247b463f2271538b9829eabd731f818507d7f6b06ffdf398e6a2186472c39c4e3db6c68c7a4bd23a5ceb24e6c2cee1c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.325991559Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d, Name: checkpoints/4864/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4864/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=131d3cd0df1bb7be6e254588e9b4182fa4084a85f7d9a269bfc27b141570ad3d23d11c489a474126de21a29df6c7aeb45a852dfe8a1e4bd56cc169b8801f0eaedda4392ba3f6695ddc9e864e4639dfdebd16825e10df08517f8342b6e89eef9be80ae5a80ef742725b45caf606245118ce3b06a1a262fe3467c0d278604dbcd1078902be0deb40a282ab5f3ed1efa5d32d497d245db84dc5d01cda66981c59204f3c68afef40f14d6ddc4722bd5405b2273f4645804688f310d03ac387b74406e68cc4c4bcf020e46d3f4aa5f0fdd66b58602a5e167ed3d29bb859bc017c2cc770fa38f9cbac8659ded6c3fd027a36bbc500e7baba80b6b4bddb1e5fa2e8d9ff&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326075334Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default, Name: checkpoints/2176/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2176/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3e1f79086d12b10efb8f27c91bee7e4d806caa212cd35bf2b7a91f878f085ed5441ac744ce0518fcfa0c6ed0567a910a2bd07c45d3b39b1b08e98d1a28b845939c17ea5fa874d450d4b44ceb2fb7a36f553cbc4bf40302fc415249617712b59b3095f967e478231d9f643f7277e267d11307ae8ff58a5c85cb6d4b8326aa29c01f0ebea50d4b6c0618e1636c85bb9e9d92cfe97d0722639503ec66b872abe5793e4cbb5585043638196612c668d96d4b9f2396b6fdd6661811ad8d407d801d231937c76d55a62b4b8088a42d87e6a46b1c8a4ee4f4928cd1afc5021062ac1ea8f1b013b968f511045b1c37e16bbac9ffbd9640682560934a3905bc7ed6b5a721&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326077555Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456, Name: checkpoints/3456, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3456?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2e66c5142c3ef8d0204f1f064ec30be5efd729dc06a685d00dba027f1d2326dd2ea7a8f0c8dbdd69c7db89f3233f83d36892f317b68f2cf80fe6078f7ba425dc1e197605628d9051c7b4891de887886dae7b6ef931b6c0786324644b6b5a4c8c67e038d5ec6c7264320a3b86c580d0886d0d396f8b52bb9d39ed76b26598406a217d942f52764c63d467d1a1502ed1d88c646ff47718c5ed32224ffe825d05aa48ee22c2cd594c7849e8fba4beb622e55c8467e65b0642f88dee5e6476c18e063b4e84df08ad57dd90013f9390a9466c9b6ebd80be7d6165b82f11ab5437b825257f97e1154bf8411e8f809cf4a553c715ec695695d03e8b32a242edce89c7a3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326165526Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d, Name: checkpoints/4800/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4800/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=ae34d099279c4e5be78dd0ecec182fd3da6523e4d8e98d7bcf1ce308b622b0dbccc4e110937d1d0fb78665b68ad3cf275614db1298a1e7dd0d12bf8f558ec5657f036a443c91cdf9935f7debbac953300f041beaece55c83b7f840922f77f462d43d7f372624bccdfc8dcf5ed0ba91411c5b02338c0c66d7a54ab42ce0ab48c85f2dd02119885bcc954efe811a3f879f8d4d1bc1f82ebd19142c77329111f3c447745d9c0ba3740283166e2251b04c6cc0e98ad54339fbfeb33a164941d019ca07bc73017279d0deb1f3dfc0f0b4a87b88a24ea1795619dcfc7c6692a6aef6976376d47526cf987c9e9fcca7ba4929a002c7a6538554bff87b8f33eb39af5760&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326141027Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4992/default, Name: checkpoints/4992/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4992/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=63ae782fe0f9197fa9cfbe9aa7f64f3a0496023d9d531eae8dd9e92417bcc1f26fdf7ed69a33f900da81bfe4dd9905fc370ab7a81a69afba8f31e82340ebe2e953fa775b4e4c882e12e35822fd5548878a4d2565971c0c2ce4c2d4267bb767482e18f232ca613af9429589e474dee98c9e38c27aecd3ed024e080aff4f5e008d197657490c9ee6aae0acedbbebd82a2291bec4f149f8415582a7804fb705e668d206d46cfb9049b1e77b3bd2df530b6724ea2e04f44a00968e641c3685b547bd930e0aac649c36c3b7f930863ea6c2105ea911bb44e4694fdf4dda248c62db38e9565f41bf90e6e5184933b45b133986d2692e12747c8eaab510ad659c284193&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326224305Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0, Name: checkpoints/1728/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1728/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=91818ef71a78b75eff403d4e8633da4509e5a2fbd3ec2db65ccb26ce49b22c7d92b65c06efbf911a8ba5612c62de456478281fe7824f716b89faff5d8657c224221f9b7de3c90a87bc05ea04f56bae8c73512cdabb7c1d49c3c532063941c2dee5a21376ba71b646837a42b711d955104bd02ebc1bf2aa87256f5b3b5885cec621fb2ef03fc81dc7838e64ad5fa78c2cfd20a4d6e97245b4b722810f7dd23bd0592ce4acef72926f2225b435ca975cebcfd209aa38bee635242adad8b6e9a4cbb7380a559b4a88bb3cf62e0a23aa2e97be5a1a6b89d5ac0d869ee32b53396a08a1a499ee3239c08ab600aa24582c5ec7b4ce41fd66bce2db5d602883d3c2a972&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326195584Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4736/default, Name: checkpoints/4736/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4736/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=65a3122a2c25173751b39e142606c05c98e7fdba03bc9ca7fc867be2764e792ed672e206d75c1ebfc70d743b07f17406993175204f362f9b57e7309b0854c867fe1cca97c2532bbf40805be1f2bab8f8962bcd5f154196fc04586b457cbc9d969da422009caa290ce8929f050c21d10f4ae1ce1efe6defddcf249fe6bb992f0bff85f0159d1fa1292894e567303493599e1bcee87dbafae8c745d9bb616e34c27c88132659ba20bdfb5074b5e1c657c4855d2cc5a97a6c8413df62603c762017eb081f5f99e1c957c7254d31db056ee4be073a88c8a0ac0c03aef2eee0626754ad38c3e4721359a2e5d0c133728dbf53c0dc9798e08d4eb3990478a82a847a12&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326204973Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0, Name: checkpoints/4672/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4672/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=16b4835ee2c63963caa248be0bf03f2b3b6fb1fe54abc98273a39b9888cb3df130e64014a7398bbbbcc9d675bb9ef03f315622bf8b078dde462e37a97ada8f8e0a92988de46345f79e4759845d447430fdc964e1401d835cffc5c5e2e16b307af393cd44d4350e943e8a44eaa6e3f162b928eabeb4442d1cce4b20a6d5f9ade96e4722f7e6987ee9ae6bb0786200059fb320813658f47bf92208fd8d77c1a13d7f605e77ea1976a8eecb7618ed5d7925f2fa675833f32aee35c3b32e58bb821f2cb045a8783fd59872bce04b9431226a597a351eed829e5b69e0f107ad6ad207503173cb49f7c2f11fa13f6c9e363d500a4b219dcb2472f2d89299ff42fe6ba8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326246707Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d, Name: checkpoints/4288/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4288/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=4cfa0e097c6d6278814ac89315ad4e2f329de6e642b38e17400d6c05bc273c0ac85ceeb885d70c84f0c793cc35a5ed3fa22429573704c6d21df5392915a407a899ce81b1ebe4d5d9824ee1f63081eb5f4bd432f8ec07e76eb7f1873f811ec72dcaf249e604c21cfbb56a4b633665fbf33f9ac327369fed9c7c5931322431c733fff4234fd8209c89bcbcefb7175dc062ace271204ef475953a39b9482b6638b38b473ac7c3e8fcc9f16e73db38355f34c4f0a9c63b9e7ab7a5d509ffe005868e5c08b13ff39d30c47b1cb8ee9a45f6a3e4de09efaa1be8223b87f1a02087540ea5671d8eaf3b1368ef9b4cbb2e5e7176617ce004aa780312143fb510d54343e5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326236542Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2944/default/d, Name: checkpoints/2944/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2944/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8d97aebe57209ab8aba79055cb639cfed6280a0f5905359249ce75bc7b1399cfb4362adbbc751840e5749001159e7166531f4831286b744fed64672f5332975d0c8ebe9f1b54743fcc4f436266a653851023d488875b1bc3441b294b8732b7f08116642f67c1246d028dd0622c2ac9cc2c090d11b63a5f320f8a960930993fb6d86ed7e9d566e1deb92c2bc6b679c3129ef757bad21620bf29164cf102cd15ab6faac7c53e995e48632c9a723e24c3abc06530abdf8101df9d2570f820d1d8368ff3d573ecc2117c93f50433aa45ee9501ec5514c7bb32a2af37522504ec7a1a9fcc5c9a97e125fd02b980318a51483acc186ba959cdba87a09cf99940d8c4c7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326268473Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default, Name: checkpoints/192/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/192/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=91ccee7386016ef88029c386d027d40981699f99c7caed7bdb87795f2553021751007c5cc03eaae75c44342f9905d0b2aa78efa82beeae184c4a92101bd77aa26fcc3304b5aded46b2050acf99a8db78aa1b119d241734d96c379160289577b41f468e6a6b5154ea01a6328fb8f7dd482e7a2f047843b7eb82b44758821b72bc7249df0f594ecb714aaad602c4b4e166f3bdac02d966465d9f193a54b7bbe4ad89ceb7efc85df43be9b15faea429c65532d8cc68d4daafc3e747cef2cb7b8f7d0dfdf5df9f502c520936b6dbfac1e6c92724e4f7541f195b789853ecc88ad2ddc8d48ffdd4c108dd8cf50e68a5ba66cbf516dfaab57ff697f9943262ffcaf02c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326316901Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0, Name: checkpoints/5184/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5184/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3bf6bc5e0414749eefa7d310e0c09da1443224f36dbb8be95fc09e7def2166603f48708ccedc0a0d16ae035dcb98193a67ebacd37e9dfc5e7c11f5096d58f64ed6b91375e63b8274ef12930e5f0b862b043e0e28f061e07e9f4fb804fb3e558bc52806fb6a80ecf360c0b5e649c737db59849bbc31db3458c11ab41ddbd737bdd677403c9d23cca101285a9790ca7179caf1e3f230704a535c575ed911773d08377af60f42011496d5c28b0aa1b200611f1b3f3e092e187c18b6855cc8f44c55d279ff98669e2e093a39b165e2b4ad0a2371d35b8a5ea27c313af47bb10ae359fc97d8bac9ec89917fe59b606b0c8545ec55b62c778c0809434b168a69e2461c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326316456Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default/array_metadatas, Name: checkpoints/1600/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1600/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a879ccd2cce376ef8e91ea93f88f83410bbb8921678073741120f90cb69a7997bed5063cee2d844eeac1be02402d3a263969e8ba882ea2ee2dbd870a57d52cb4f6553128e7dee1aaabac7157a30dd81412e6e727f7aa0e30859c55555f1833699dc071a8d9b3b321ec9f3e823b342e9894234252f33c3212572e5598295ba358dd6acd6b57e2436707de784935df0b7ddd5b2259db8c42595547619a7e103a1663138f8b5ac9f136a84f395dad19b82d93c0c07666fbdc0f3e4a1416c54c441132185c19a8f2ebf0e80ff479a509208917ba56d1804bcffdcedaa55d240952d04e94dd793ad777684154ad6c7b1c8cfaa9ecec4ba731dd011ef336174219b080&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326335503Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/array_metadatas, Name: checkpoints/3968/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3968/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1cf5c54dd86da38b74324614368ed220b598b8d094632e6ecc686ff42d8fdbe4929e70f4a62ab5f227a32beede6f0d749d7ceb91a78ab8b77b2e565e73d1f5c0898327c40af937c785f22042c69e9e571f754d679f9cd272aab5a2ff3c129fc2abd2dbc7c511f1e5f8903ddb300e8670c4e47751497f64648e1328c78b4ecfceed3abbde3f6fe268c7fb8a0d57f996525352903c37fca4d5d8e18804f14dc7ff1be02c8d2cf0a5762b4a952f79d9fcfa32920eca54284847e270c1eebc7e5d58da66f46c6f6a93609171f5e669f240055bff1e8bca6ab7b3178de56bcbc3e8e717bbec7eaaf3f1814aed93da0f63b4c040a5a8c9ca5458a4a0cefa75d18457b5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.326251095Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default/d, Name: checkpoints/1984/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1984/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b11a2a18f167122e54d1ada62b451cebd94eedb1f4dde157be81c5e808683de4d90b67d28c3e7ea295e2340625d2aa7432de4f17d50c49340f578540122ce14190a5997403dbadb98ffe4213f6f9d408d6b5fa3e47554f44aea375484a33d975a6902eab08b64cfd35a587385e01dff7f741a0fbfb2884fab4a6c3f97c9302b58e8b5e79c4c2500ac777d941097349a446194d79e073918d1c3ebdfd6318b21cfa9de2a7d9b4da5f3c8f1eb002e9f50361b2bd02ff07368b461559246558b2b1c96b42b84c11ad412c0dcb15dd2eee310d9a1ec3b1632bb4682f22a22666adaca8fde8a598caccbb489964dfb76d719aa6deba20c3209fa6adbf52d8d3e46fa5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327277327Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/d, Name: checkpoints/5440/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5440/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=29b0ae9db5e3b3d15a5444f8fcb499724d5dfaa3041f71f3432bee6692e75d48a0b565767d8c5226bc19b01e922d3adcc777c90fba4fec6c99f915bef3176efc29adb16a617910c9cd7dc64edd398f6292d54cb7de085807b9be8532a089a8f39ec7e75aba7fe8f3a5eb6c17e2e32a873f34e8a49dae1c3cef771e74d7e1e51b3494afeb72b475bbc879e9d68c3ca63e9126ef416224781203466e9339a18105220d104ec42c68d1da5373af60da6df0b0f0eef18d5f86d0fd5b32319d47688c327051d91afb95bec52bb2ab4757d3545c9587bd2768b68e001a9461c9b21c85aed8f3af7a8dee06c057251b16d709299da58fadc49d4280fc8fb8d4cc733071&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32741704Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/d, Name: checkpoints/5376/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5376/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=29f05eb862fcb475a7287f044ceb7b4d572577a4832df05339377bc8ae7a5ce1bdd4b79dab01268c012ac4b4868ada22b53f6e601189f43df55dcdb3dea07393c94918901847d411b26b609e2968dd9a146b20056ab31a4d3de84c38ebbc2b746194e61a86cdbdbeb9a73b22b8b505cc8d1d92d619cbd592f19adcb1d6330ea06473d59a13587633564c6190aa1f4699dbb70d5e6fd26dd7b3a199b666334f929e64c4c4cc2e53cb639f3f5f4653863335fdbf9b6ebf49840a52a9589266035e9d7c53994ff7fb6ec6cd91c927a7a6b60b81e2e12be9e48f2b7194d05daa1d659d9e09f4a2a9dc76cd87cc2e1d60b2959498ed4cfcf5f2c5d6b928c08dc00387&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327426928Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384, Name: checkpoints/384, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/384?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8b1561962e424655ec785b8799831930524e61e708694e067b08accdbd7e887cf6f46f7b542a171ba74aea0215760ebd07b95a76e7d4914797447cdfac12264c8dc0d1097679acd5fe840bf8727bf2f4aa05fd2bb642a346e97bc97a4900fc00def2b602dca968832d60cf164d5ca5497e798ed2a5438583d67cd84023e338a7be169b03eb55044bc40233570db85cc4578987ec0180b8fb96730cd64392c7d59a3fd1524f906992f84b817efd3382a3c46571aff33b750253c1ad88f5f10deb63ff46b9111bcb01862f720278bf2c551cb9dfcbf4c05cbce7ce540503dcf7ab8560fd5d920546bb83cfd8c8b69590fad30c8bca584ea6e14eb1179b3fac1b54&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327504083Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d, Name: checkpoints/5248/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5248/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1bd4a292a76e5b25a9a3b7c226f6207a176bc4a526b4dd8281c3a7efe3b307340d07e148d87e9547236f91b80a1ca646fbc94d0913af67395e133a964c738863cc6b18d01416426fb5aca7bf74bcc35a2a90effb1c441ebce2a47f9eaf9dd831084bd577b24138b2689dec762560228eb539ad79708cd1364d8a08e76ae757ac2d7bbda3525daacc61c659278a2d4e84bfa017ed5327f049fa11e90220ff18899db4db36210ff27f0f1126899146c008ff51f9e8633ac86b544e6bad5e43b0d1ab75c370cd5d139c6bf264f1ebde3217f47f1d8c8d070c2dac502f89ede8d9de980380dedde7a41a4ddffa86fe75848cb2500c3039985a3044caf3cb264a472b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327516029Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default/array_metadatas, Name: checkpoints/256/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/256/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=72d9a76a8d1cfa3c00462258f21e9f6a36f781fb5e033f55f6bf354c15bce4dc4d60916134ed7e0902cc742fda54f15da7a21c4606af37ffa510cb16a9b6dde3dd2a1f3923df50fd6e8af95aeba6d94104d5d54e6b486a75cca8991f396081d047d8a8298326bf937dcb76c90108663877716e44f60bc8873408bb8148a1d34b375b4c5081e37e3ace93827e7fe8cb73f787798d93df9b1c8630699f0c4b47d018f8b0e5918b1d5c45aae9c4ec76d317a0e04a7c4d0281fc3a5fa9398e053e7e9e18c458e6bbf0874996c60108619fae51d8809230b0faaaf91d3d0a10f46f301d10d20dd65e2beb5405bb82b6f4410b8b4d0ad30d065ce6f054fed3c92482a0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327543601Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0, Name: checkpoints/5248/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5248/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=4ddb1cbf28de11b78914a4eddf5ef67bf90abacb7721ee3079e3b22c3317e58e2aa61aac3caa8914a9d0c173dcd9fa8ab8c943a4b34340b4b77b1bebaab3c779db3e8e220549feac06071456aceb0d76351e700cd96a0e346ffc4426a1efc27f1edee3a36ea682bd48bcb62e34198d4f30f3d841a480f5636d3f9a98935aea89f9eb204fa6aaa7a06159b3cbce18e2d71413a591ba17ee194bf4265873b71789d65a12c63b458934eb41a982417b60ec3ed0728d86ff5927b64c43b7f5c5373c272f33249af3a40c11134c9ac94da297dfcaa488debf9dffcfdd945dee6c1dd847fde080b51f8c366a39af0b2448fe975141d7e898788e824c11fcfa67538d6f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327590791Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0, Name: checkpoints/1536/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1536/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=af77bc659731300bdf848a1898834d0bfac83baaa44efac4c6e355659c574c7da0fd0fd672e4aa168fb78ef64494726c93e59bb198fbcb16dfd6a1b1c2b6df2faca153bcac3a0041681f4ef833beccd7e3406e549edad151c21d4ed4168c65a4848c5bf9db0bf56523f30a4dda9f8c2a3cc5a394646ecc59870e22c73afe3d72c15f11df50cfba4fb201657a070a44f923fd70715ea969d71ebafb60c1bd498eefa7a6d7bdb1686d0cb29d3e3926b2abab09ca11ed2f51b0e0b467a76d1d88f31911975f7d3faaf96dd9ecda56472d76d7b8747c10b7a2e9b0789bb1518978f8177967fc64206c53a0e53a69b11cd7a7b55715a68f99098e479e02fbc929acfe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32760341Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0, Name: checkpoints/960/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/960/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=31d86757a2a04d4ec275c29680d43ebf646d049e6cd7a38d86347159a74fb690bc64a60fb6aa284fcddc326c8333dfa9312feb6024d56472b0d5e92a04bea1f11fae16697531b3408a001e88081441986778d468098302c626724665c2eb154bdb6e0ca25bb927e5e90bb94a11394acf7391a58dfa2000ec1ed866e69020ec31f01bd23955220b392b2ecece00096ea842e124bb9c2ff6b80a6e6df2180d97a9898700e381a8fb08abbaafa0e097561cf9e9e5f683ff11efd213b543d044dce7daa04bd89f4292833cd6f09b8fe816fc3f67674fc8416ec6fecbad5c1c6b3c780ca5a793313f5a29b5d033ef65d323a544dcd3569792e7a08a2d087460a027ba&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327607049Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4032/default, Name: checkpoints/4032/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4032/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=c468ab1993b62bc4ec96d0e0fc5ffe41bc544661d9ae6d575c5187acbb8a19f11d389a34c4cd63bf69c0c8d5b314a0852510d3cc2c3c8251ae66d9e78f8b94ed59cebf99ed7e3460b6aaf29b4d666c180779d3fc20822e504c96d46d2a8f49cd192eda6d7c41b299c64805b3658751ca2347fb414cad89bc064960e9837de1df7a885d118952b0108af765561a4467dc0a987975860e616e74b14dab4152694fe1cb646870fe9942168ed6eec3bb489b70c19c0fb75d0810f287e35ba281838bc7ee00fdd3ef8e6749746e06cfb2a17049f31ac203122edf2db821b50cf0e16ee327fe951b07b1ec6a7d3d29a9d16c6a99580f4bff447ddab788cff67e0065b0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327618994Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/d, Name: checkpoints/5120/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5120/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3ffbf47b0ad01d3a3c105631c26d276565a4504b25b3be68ef3857c050384091a47a2a3646f39cb1989d4d3cad05997a22e333ed8384d0e8edd9b82f0e47d5739720d9d78e136557df4af27cfb83dc346fe6e1df0a758292e4b97528dcd38292f77ec02ac98d317520e58f03db325dc333e6a58843d6624e8199100ff31f4d5d722ecfa38aed7231eea5a052fd6e80f919e98ec8ea076c4d04a4c049c541ed92100990b7abd3305e06d1dd09effbbe4c13967ef5fcdb40b35d9c9c3a35f5c70dc49dcac7cb1902ba694240f4cc4cd5fdd3ab45338701da0264b28bbbb220d6e085a0b3e93793c132b3ad8ee8889bb85fa1a9ba83e44153be5cc9b3be3cdb7928&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327715455Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d, Name: checkpoints/2816/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2816/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8e0815111d1b579d887b13602e83b4befb5ee4d240df30588dad569f582cbcef08a25a572d858270f76e2e6b1a0e27354c42c39e020a0b1a68a8c9f5369fb4d8ca6958b1d466adc31d4280d6162736bfebae30d3e8733f4589f761e81247f4644b82ad4063964445baa4639ecfac8412d2a51a6032b82c14d8d89bb05086f7fd0dee764c12b47b91e333ed158f8ffd247c9a3b0638532d5c8124555ef8b730d56c12c3304fa523944ef0672e9fe3a6ca3d5addf43fb1dd06983d5f5b9e26426e2ef46ef2b0339234c773d0b83e518f7a3eb2b8eaa13031e2cd2a399542052fd550b2e859c16816eeed6964001da5ec945b1d60cceccc9a00aad32bb1445e5dd0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327720253Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default/array_metadatas, Name: checkpoints/2560/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2560/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=65a844c6a2b371009b9615114f9741f658ee94949adb9e7252e0cc96956219ec74eb6954b08bcce07a8141d97c02e7f93c20dde59e3618168ddb6b6aa8e4fc92d2d657ab2dcdcad1888f4d4f05deae89b740b2166425d3dcb5abcf9df42a6582d2d4339832e83b9958c91da5ce05ba4eb60273bc28bb0f41499a9467f2e4ce3b30032da6b1266d855d70e709a0f2b82fdbc54a00bffd70c7acd9b9f0cbf3e154b92ce7d2f98152bbf8bd1d9015572e4e36b11b20ef707490fbc166675dfa0bff49398fc7cac41c40dfbb0506a67a863a038c9eebeaae58f6cf060d0b54e2cee904e808ea87aef528852f6a166892fcf6b0c7c72d93972c6bade92b867e65a211&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.3277777Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576, Name: checkpoints/576, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/576?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2eadfa3299ce7e34f3905b21d6364cbaf5f037177b9eb51d5e6b9942f04f71a9b9198bf57865eaa8787cdc340a1c96c1b4df7933ee9fe944fa1522a2b01994b228cad1e899d46e9910d22bd1a6d82d8dce53e236c01bcf9713d121878a735adb0c46a1217aec14713c06ca45e5029b62e0f79f5afbb980ade1655b78db9492ea4996af8c9260b5f689553f6b9bc8ec3e3c840e793eba271f2b9907fe197f38ae486d257687fb8c0d072790c3875ecb4ec50ea996badcac205a2cb405cdd13ea915d67615b535ab28abd36f886f8ed8f3642f9962c775899ecc55479aa8298ba1f02822976c24bf60b6433d72dc4b941dcdec31b3fe9a587485525c3ab012a901&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327825864Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4800, Name: checkpoints/4800, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4800?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=52c1905a8568d90e186ff5e5cc3c18169ea83bc4c08574973de6f1826c39ad32a8bff18b538b513e5fdad324c0bd55b1a8eade75179ba3d2168995edbd462bf4e2bc46088745e4255bf5b3bc7612460bd9b6ad308baaece1a5bac06945ff0fb7d7385f2934b134d4891ccee5f496f512b4ba7729b4033e4fe3c5dbfb01d4690a72f1963e7da74786b28edb75a064d96f06236db19c4fbe4ee2e0041e887df3993361cfabd46a63d47aefd283d249377f72f126b3f163829ae9c68e21cdccc8edb271cb89bd6f8fa2994244250d38e75f45bfd7fc082b58ae54e8dd34eb4fbead88d99c800fd99fa01d0edb9744f672dee1e38d61e73649a05b672c2fbca7eb4a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327843565Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0, Name: checkpoints/2432/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2432/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8c75b9158924759ca810dc361302a2b4213411140f9e55d96bedc7c3b5819d818f68ece2145e62d1c068847725aa6ff42e5e504e9785bdde0d013cc95cf02681b024ed26e9c81d4884b5d9f23c7f2d1eb66957626a7eab2755b817b485ded7342b71abd38eef4a17e53b1d5275cf6d2789285f1c277a71489c101a60664a38cd7d4cedda91db9ec88936c139662427b556275b48c14f547fba0af2ad7d84c409ba83fd3090274e3d681408cecccfecbbf0d6335b4af1f84ebda1e375f479de96b5d08fa23e2345db78ee609af96b0585565a4bb8a425ea9db6a9a2039f0aef1742a787bca5da2bc1489468083440b411129bbc89c11ef596b4cebaf75889df63&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327868887Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/d, Name: checkpoints/2432/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2432/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=772ee5ce9e55defbb44293e44d34678f11937074209bfd8bb771ae2d704a58466a3bc951241cf65e65975cbc181a7c45c649f4cdfb6da875bf62b785c279c3bcefb951329ba9ad19181eeec1c8a9a03da7d516d613e6b328f3fe67bbb22ff44f33f0924384c5ac3cec046f1f2ad0b64a31f7af62884c59a35dcb0e1d8aafa7ff8338a2d05792a6760a40222aaa743f6024e0cc74ecf2083559b93d0062787d6a18f825543a3d5575b8eb6b1810b4481767df5e1ea68cd12126a41206eefe6bc6615d3eb7584e5ae9e787234d3a9cd1af9b898b8f7ad1ebe72b4a67b3e59d5b9b7943cb9d20d77320f8db3efa4dbdd191b74355744979cdd98b21b13211b3edba&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327916091Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/array_metadatas, Name: checkpoints/4928/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4928/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=962c523799173d746f95b4ec3ecbfdab8a2a396c7948d99e71b075d97d6dcaeba5b5f2ed7c919292c342ea05bed374c767486afdbaf6baa1ea4a04eb91afcf65dc4ca7f3ed3704f8af9899dddbbd3cbfcdda82333b9e840913bb0067e5a0a2df5018fe17bb1283eaf4a62240aaa58aae6d0f1f82a72a9bf552278d3ae47fb1aa723863f467b5ea5610eb2f83aaefc5cb877003b044ab7d3f050eb147dc1787aeb19cac6b926f34f8505fc352461f26e257af384aba9b39dcdeafce72d72085f9b1b79ae42e0ede89f9ced5fa007a05fc01dac5400fc6c5b1d685a3b2ce6ddf3bbd281c45af4634b9b53de42d0d0031ae97e70a21daff600734de78e2d6b5355c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327947199Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696, Name: checkpoints/5696, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5696?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=06bc94d2b28c9460bd17c84915a09059eedad3b51b6c03992df8f6f0dad2d48bbb49950f9d55c19e50d452996093b79942b396e6a8bae58e9af6ad784a5b034351f4e1b6128fed356e39e7018a936a86148a10615ae96cdf30d154978b800f139be7f1d5d76a219a4447897f0e06481bd64b1d77c21a85ae3b085ea0ae04d5bd1a3226c092163190b452faa576d9a42feb0b3e8363d4eca0a423861fb97a6db6bd271b8cd54e09c7cb10d97ae4b62a843c2a61d999066320b1215e245370b2504898f82013323999d6b2a6b2a2b33a0722438e9e83c6974e1fdcfded5ede8a8614c0cb499f0dcf2555f8112b54aab27f15225a796e9bbe55d1a43d398fa86e61&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327969581Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776, Name: checkpoints/3776, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3776?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=bc45c3d40a538718a6fefe1668f766818e13c269f52a4927dedd5996fc87472fcc0c20c254997c710fbec6ec27c4a545b5b21776c2047432afdf511c75f16f1dabf79521ce8640d7561b827b461a4041fb6dcb52cc3d3cc9183ae3aff96565253d1dfb4af73b1b2df46e5d7ee2182f8fa50119bc90e3f9d558243ab57640743970c148b188fe0cb51750e46aa5336c75152ce352a0877da506f88925019de80b2555bcefff4e93b69dd0ddee63930819f9712d1857489735dd08b7fef713a02b5abb639b2cb6edf86e80d1d2ab259c191675cc9779b0bdf2911f48790a40ec96adb738fd1cc5f81fa0dac96ada8059aec653a7e1a536c9e9b67e30b1996a4455&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327993498Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default/d, Name: checkpoints/2816/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2816/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3b48ff675a7ac5dfb0617677e88d13d5b06f9a6c08dfade4c848d972d8aef0131fbff2ddb3a0ef5bc5ab4161a33d9fe7cad9505a66b99ba8f2cbc4834d037ee811640ea59e787e5ff42ed282c6cc5e35236497320f86bd6206ec0461b047886fe7926dabe18c81add6f5d78b07e6db4b2be5d0a07784c9418ffa0ff8dbddd524e291f68fe73302fd4537ea3b2d3a6a98ed708b92c190241f1a0646bcfad1411510921c54fe0c522004bc43c3c4ed7ab5a9d0232d905e664133c25a320012573f161292c3345efc7b53503465277d5019c0e21dc4d978641f3944fd86feda2b24bb5c4003fa3e8c632a480562de6c8a519476a0a7ce04e14ee69138687f3c6940&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.327990163Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d, Name: checkpoints/2368/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2368/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=0432caa012efe143c33d8875401d430a4a6a06a4ea2cfe14c77aee34dc7473a81cd66719713840f2048af933704c8e2935e7ba079088c4303b527b7b25e9b800804de0762dfe793fcc23efd319ca24c97d3603933bc0cbd59ba9e0b79a061ea96a359d157b2c813bc821584786a44d83b73a12e6ca07df04a5414f008bc218b6ca62ae2fb1311bcdc6faa7566f7cc653a9d4befbbfea275bc36086daddd466c137b4fb913956c88440aada4da4e080e48bdcad9c5f0fa0372fc2f6befd220cc899f5fc1a95d228a76366cd2376fe41ff2b519ab7e1cd700bf61307035077a1df568ad6f493796a1254fd0d5f662bad1790b74135f0e765fa87acd27b0ccf8723&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328046677Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904, Name: checkpoints/3904, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3904?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=25f639d428f0679a99b399c984a1b9cc77d6226ef84ad4750ae1aa3a7ffdc82204e9f3376b8ece7ecdab5d2835dfe6c13223aa3fa4739d91d4bdce9900ce597fa91b2150de41cbfe65da7777cb673e8109b0fd0fd83528e417f0072e83aeca60d68758be8a49931647cf40a8bdcf5e231b53bc7bb68a2dc25ee892e0ab6a42d3d0490da1981995dff284b1dc0c4b493a379bb734ebfabd23e8aefab8e5ff3054c31e30f089f8720645fe3a740a7310323ecddaa7ae5323b9a9eece4cb8eaa6ff79098db2687f2bbeee4fe32bb63d0bce72ba577adf86c846cceadce16c3ff95b1dd684af0c978c99d2e0fa6537ee1a53f17bd8e51ac34dc112912f0eefecc115&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32805242Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default, Name: checkpoints/5440/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5440/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=07950b200a2495c81f3c4db0374899f4bc955de0f667e43d54a6da04917fcb126f25784c704a3804bb979fda1823cd2c012c79b4a54f168f61da02ed9417db577348465ffc6860069d2b39c88e076271da40c92fe1be7e8c9caf88a099829b6bc35d24f7bb828b8054acdc2284d24629245eb8718c4d5d15456a802b9cc650bf95833db2ea87a68e87660d16a3aaa150250c917bbc9a549df1a116c4550aadfff3e1030021c2a49db0cdd83d3d83ca67afbd6322bf5f4670178ea12d621e24b396f095ee3464b9fb0f6f5287880dd68b1482bd162272e53f76af02d05d8a77a3031f949a5dee44c29be108259da2184133e525f8e5c8b597bf3570ff2ebedd66&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328124669Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0, Name: checkpoints/2176/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2176/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b824b5ea22e61f10310776a39228e13e28b020ad6adf4ef03d4bb2d7eb35ff933e25329a5a6eaba44d84044227f1425d8cc6a841fe81f4013a44fb2499d72ada1bdfe8fbe3314ba186a520a29687e43bdcbbe3881a6bba2e488df02a7db82c752b6eadc799685327c795127af359bbd929e2987446441019dca0d39828e516ac352bd35d10c3fb6c4d84ec0222974bf440e5e453a98614e208fa9de9f3f1b743ccbbd02713f31a93e2e5b4382441b1a49c8bfd668204a90a6b6401cc7cd26c6d7551a5e3e85292dfdc42b399ba2e7f085c2800bf32003da1976d2ecebb781575d58a328702005d0fbf5ed40277ad1ab2fc86d3ccdaf02593af24b1f629ce9e80&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328162744Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496/default/array_metadatas, Name: checkpoints/2496/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2496/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=338a6d5c4c9b34349a015691e91df6afd1e3dc468d96fb884a45698762907b9b703fd31c51ba7b94cb44c1201dab8b111061a5a57682591d2fd7b34bbc25265112c63a29823f53b5efb5ebf625e0d96530eec6a6cc56d45e41133a491ffb8f3ca48bc1105fd8fa97754eb0f9240d0101a684534acb131480765a5db2c4ff7a06f07a2efe6ba5aa6eae451b255bef26e31d153fd923754c4d539bbf5d9a63bdf573e54177f13e127bb24db92c913f4922cbdede09069b43596ecc1a594e486103f336b0fe36ab4f97b004ed88fab10ed6f20e45ffa5baad97778cc226637c1bb05086ba89929999ac5c21be7642322a18089a506137aaa732d9728cd8b82257b5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328229054Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2496, Name: checkpoints/2496, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2496?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8a36214ba424dc00071f49c789bdf6dc37b5d17e15203a6a2edb0b2b7839107d091af83c6c774b3e652acdf790c7ceacc222994f8e61953a92627c2c713112b754721053644529e6cbb3d6badb19e46df05e71218941d62dcee14e8fbb7274dadf62392f9ebd87a33c0b5bb16eea816f58381ce78cc12d98d9b51e81fa51f31f7c27e2b46843b5b7dd94f25d54e012fbd78ee8be6f3e260256572ae89aafc417f59ad93a819b35ea80997c3dc21a32f441063d0fa6e9f68797f44f365e45b9d3f3ad2fcd2c0a0be4d85cb255cf624e4b78ea93044e663e9326eae3ade52b44150e246c3e1a853165919bb7a2b45a0e458cd46aaf4c01399e09c70b2525084f6d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328250895Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544, Name: checkpoints/4544, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4544?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=33b798e9211a07181e84f601e954bd056bfb30d9716313f76ea7deb90f4dbe7ab5868f501aea8a7716593841169dbce307675208a9add1bfbf62b484220e7b401e959a5035286636510818db673271c1662f41c68ff0cd1dc41c1d347cd49732db0beabe352165fba724de2835e6fb98968ab019d2ae324ecaed6a1f13887cac671873dd8b3db8d67a4428ee0d466a426fec03dddefc81ca8293429c87a2b2fc56f36aebc21ae0fa7b9e0410bfaceb5faf21cb8dcdc5f1fbfcb310205f63b0c37d4149837c11ebd6c7063274839607230429c84d54716af3bb31fd87255d473f8bfccd9468e2061dbc7986de1a59e3a212770e567273cf8662c775b971e1a482&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328243014Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/d, Name: checkpoints/2752/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2752/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=926c596d39832678f6f8f8e0626655f6ece394fcfce45a3848cf666c094e960504309628455c8e703fdc1f7cc20d368a5de5c3d9dcd72d4cabcbca600fa756cd241931acbd8a659eda8a9c53c39647a49e34b53275cf0511589ac16b44c3ec643ae13d2121711b9de714fcbacf65aa12b603726924b7afbd60e232f83332bf21c83fda5cb7477d757ce0518fad07276629b32715e812f72907ae0de382e877e436fa93d9b54f5104e8b7f8535c54c13a59dd3e7dc1f0694670375db3ba965a3104dfb36f05380ad323208bbac69bf9b411530d6efae03fcbf1d198c2af0965ac92d9bf462a109e89923fe5b9b47af5c1fd9963eaf4606bcd4cb3730acdcdb688&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328289241Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d, Name: checkpoints/896/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/896/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=c2a23c50e58d2fad20707132a49056a2695bd94fc572d65906994e8ae3f8c3cf09f92176f084939e7437a268273f5d272abc7bffaa40d4e77f1520b275147b339a3abbe58503590916e126b1e2669b27fcecb54a050c9be5a4e28aefcc36f47540de335a5de0fba627c41fd62ac4f0b973e7f9aca8ae02b475ad2d479212ed11cca880289be1acd986168c6aee942583dc4174fcb6d11e8e9937a6eae8737f0433f6c1dbd8adc128aab4509881a936be281caf072026a658cc2a5dcb0ca7ffce47354779c56f87cfed44682ab6661810a2b098b5b53785bc7ad8304bd971c37d66242db7dcb166ef5e7be249f7ba340fe6bf9f69de6f9ffa4bf17fd78ff896bd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.3283008Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376, Name: checkpoints/5376, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5376?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=7319bd3dce002663b6db8751da1828d0acd70ef1a4925927eaf956049a47825df7accad13049f1a7c0c6290ba6f2d9d8c25939476f033b559a6abb5ffb7b1e23f3cace8547c02ccb264e60076eb647e1e55f77f35aa1fd07f8d15936c5df81a6884dd924756607e79fa05e3dfd4243b3a8aa11b3a97e4f673f160033e079bb113bd0f6d7b15d7c5cba67f7b5f0cafa8d3cb8140e02e7c3838e666801ef87fb1038f0ebe864f04bd8dd5e765aa037b0f6ebfe0c00a46fd5f10ed9b7001a8395107a87ce01b92736b996f26e07c90f68646cf11a8f707cbe38bb1e3017261e391b43096d2c7678a7875515eb2a7bef695ccaef77950c4adc708e437f09d50081d9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328314405Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1344, Name: checkpoints/1344, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1344?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a18e1776c5d12b3d4b8b396d2462576c04ad82a3290377af140a1a22b4ca0bbcf851131a277a10582500e5585c36ee7289ca94e267a754b9332a6a5c8c93c7c2cdd605e43294bdb7ad7442fd474de8129d17a286eb5e2e2ef96663ed3b6cf00601266825eee30cadbc315e25419cc3de01a99a791af6f92d0d8107c1e0a66be55cdaa02a372597622c7689194543d76fc044e31b3def4430a47dfd9a2b46e3327b8cb64043bc351a9224f3b49ee6d1c8c5249b9f8c69cd223bd1bd2868f45a41177615fe46fc6c3260321d96bb5f8398ddbd94bcc422800692fdf72208c536fbc834a2785fab00a3b6887687f709fad28e3e344d3a2e0812a47b9a1b0aec2fc8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328413984Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288, Name: checkpoints/4288, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4288?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=75ecc0979d95207fc42c0b6fe34df48f661c5054eb9611193a0676d67a88a7abaa2cf445564a8afcc581080dd59ac8032b850d436d28445f804fd1f15751437c32e241f124ef256c7c59af77f32af39f2052d0d7ca3af1374fbee19c116d93e0880f1c8b202b2bbb4834b149004106422f6dca6a4b13c7b911a5c5250d54e25f2ca5fef7b2f9bf4bfeef0bd039d3f76a947b18e371daad43baedcc759723c81ca04ce83ee8f6d88ce0a14e7259b809810861f819a492207e45a09e9aacabb96f5e4d9a7393778e62fc4ff84d7edf487e2b003d0b10348e77e6d96b4e86a023029e0649d1981346360f6c2232835bbd3e742307a346aa5b5fd6b6856ca4ee2437&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328503803Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d, Name: checkpoints/5696/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5696/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=0e3a5397d7df7c1dde3541cf1f99e7c9ecda8135cda2990ac6d87add71c367d46117aad9a77e449d64ee8190bec741e5f067e576ab8b6990787de0123439aa46c69d47802581fae083e78cffb0b55ff951b47f061ff6fd769150a8510df0e6a62fb718a6c20caea74cfeb55504a2293cc0a40fae5ca5d0dccd5e1fc889297157ef2c06db45e60eb49c6d5e350e947e46be2055daa0e6701b049479ce9327903825b3337fdcbc8158b28d10bd9ea54d15d1ca6fd6980d5a442dec12f3dc8db588e76ff9d7049c28ad99121b98d808ca8d251d8c9a5e4f7634c4a93938df5edbd50f06c9809987ee6c0fed998748aeb3b0450297fbb1b6d6650cc5f08b8e2de368&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328469124Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default, Name: checkpoints/1280/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1280/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=361a4baa3e950d2cedd4bcca028037e2439ac0a37bbff81a2f5ad15d44fd361549fa33b2c3b0a8380d3353b7d9c510c4b2955bfe15b343d0555424df5c48fea95c401de4f81ff4653ce7f2ad4141b9d5623ef4ff449b43792f9f9aa299f0ca1785417dedc60580b8193e270da22f8795cd216ad4d4d85dc8b98814c8f43137c67efc343768a4c003fa15dab33777cd7dfbc6f0c63a36e8e636c778725de56cf769472df3350faf1f2e586a226082e60565004c6bb0a7af3468351114d731c0a6c4169d10defa1c7ec53434c1b84b5a52633333554d2172ad9147943366899dacf69b9d3d154083429561dfc10c6c20d77d7e24d39ce9db8cf07622c5fbfef124&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328539179Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3840/default/d, Name: checkpoints/3840/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3840/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3bae2d50ef128972293ed083ae7b4bd4e67701d64baf08a6775d3126131e91964891d6fbd8f77bb09e8a5619fa60cb3f18e304777d847fb8849d23abf4b240b1f6b871db34ca5fdc90e3a3d4f77711063b624226bcc1c19e1758f0d18c7de8598689e6cabba752382dde70dd3dc6619699153b1af517f98d3d0a40370e9b94d201a4ec23f6a9f29d3121e73f7cf18ecd550f196a262f9fd1b3d08c1d277425ec0844aabb3adfb2b144ae46ae25bd3a28785e2d659b8c581ab395a81e35161fded098d1f2f1e0e77ecb4aecd27c0fee21f182659453bc6a165bd0beb50302e8ae06057616307ccbf6d0fb3cd26defa6c06408b2a672934f5bb06c43609ee04f39&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328662062Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016, Name: checkpoints/6016, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/6016?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=22729d096ffc61414e0971d2f9a9a8465871aa8a6a7f4702e7cad10c359df5c975e1a76fc70a9a16756f60333d7f5af0730de658fe75b8ebf8a505e86073ca9127478e2ad8064b780c81c36c67b564287cfbd7ea3badf0a8ea6e591274ba69df7ac575949a27429e3400d283b3190424849a4fa9c776dd949691d9aa911bedef48dc185ae65e149e2de0107d7829371b02e74da0212202433813e3cb1c7f6ddb35da926ca7fc9f185422bd1d6fecfaa00c968e1adcbee6805be919d87d561e993cf9e82025d54ba8e833d416f22c0545da17c785b1a6e51147f6809e9806f168e73ee7d5a7a1ad2340154447898401399fc433b7ade9bef6b93e8529648c3ea0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328750188Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112, Name: checkpoints/2112, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2112?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=924d00816f568a609b589a48602aa695317ce784091d7f5f6a5e067bd235b1c66d335e8cdeffab4f95dfa4163ed254be8c15b597c80b1707d86f61fd196bec01cb1f091cdbdf089b9d21cdb86494bc60f2a08937baa861b55c303dce6a48815afcf1e6d82b9da653d1d811b93ca0550012f00ca4875f2af062e111ef0b8f013317a0f6751a73866a0596d1eaa2273c2a83baf1ddb2044802d6670973a8abe6783951ec760bff60f9256bbb7556389346cf09e519991df7abb8e23928956667a0dd0f89aafbb01edf077574d2efd36205675b09c000318fde29c770ff020014c003f8e747fe80f4aa42bb4d9e862c258700d03a1ff119bae6eca15c8b3a9a65b9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32876056Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648/default/d, Name: checkpoints/3648/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3648/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=0181be52576b78fe6047d4f9a6a2c0b1c94bd4baa339d50f8975abd43b9ea88469aed6c7475f7870d964eb20d99d24ddeb04e995eb1f4332ce9977ebfaf304d80a6609bab3501fcc8241ba2a23ceec60a6af26ad14ae5436c1b17890245f8507492f42ab9a170ce26da714d511dc3a54e41bb451ff4001bb9494f3daec7bc618e76f330c9a0adca766af0fbef7ccc41c6a0738862d2a612be38743d7ecd1a4f121c6dd7f7f1225a13933915ee9da42f6a6d3ee42b6385490226730e28acf2ea0e1b9910954c1e1afc17176d36b12c89ff99262695e50c242042e65dad015e6523fea10d7bfcfc2e7c1a875bf98712ef00b875321eedc1b146eaca16167bb0add&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328827018Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/array_metadatas, Name: checkpoints/5120/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5120/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=765a8f48450b30e3c2e310ac1789e56639dc3801a56cf4379e59452fde3749be0d2f0c40341154fe733e16fba35d4b5550f1e7f5318c50b19d6ada32bbd793bb478da757024a758d995393628edc71e3836e48425ec9203cd60e03419509ab27f9d01641f51727d7ee93b79e7ef0d9a521af608837999d1b40735e98fd85c180388ebec6fee382ee619914fcb1ff472cf052297739345621af743a234303ac54a01b87af7fa1f91b20a726b77f305762fe5798d810bcd0d6b973511a245335ad6041e608348833989daf3acab92a272a01f0d76ad27ff7b8f471c9a330ba10395c172b6cb81b9cc26f930856d66fcee436db43a009a73f4e02905dce53d2d686&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328879573Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d, Name: checkpoints/5440/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5440/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=5181fabb5a445f145796cbfbea406dd1d2345214306c63e08e6ecacf24e49c6ae1db0fbd009adfd29e46fcf5252c428d4afa7bf05c49a482a253b9b4b506d5dcd32ad441a8298fa77fbdd5182d11d08246e0384f142a8b192c3497d477ac212f504d1b1647719475ee92855b7f34d66b0d3ed287a415eb771b3fa2afa029b9dc81a4cef3c987c45141a940a28102044c52d90c2486f5b8f280650326bbfafa5bd9ac879308513ba187f51eb06e9420544f1e1bd5b9979f9a7015a146ac3957b23681d9922e91c284e0a80e6ef61cbb19d18cb17b1f68987438563a84d90e20712c87652ea05fb002d396caeee2d223e8c2e7add2957ab7476b52c3a8c88dbbf9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.328909306Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0, Name: checkpoints/4416/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4416/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=36998645af4f731c949f9632bced1b21e147350eb93c88c1fc77d56d8be2ddb4e25a0c85af9febc5e5ff00d7713b8525949f3fdf5d365f6d4daae48a5d83c90ac96bcd683ec6743d465ca8d4effdc0177ff671cc28a8f665e6b09d558928e4e75492cb50e20b3b4fe1dba8ca446995dfb2fa63e08edc706dfb41755943a0473f6f6d65e7dfbf00b40298736448a271f883a968eb4ff062fa3cf94cc0a9e2062db559051950f2b311a0d5b9c93d2af6d10590a54f0154479ad1026942a2a9dfe4130f9ee48a2aa1ce17cb1753dd8709f2bea8f88deaa976cbc6d56940200ba62cc1951f888b12c8b4243ae63f532d2afac713b8186d8ff698f98d7e888dd1ab27&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.329039116Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0, Name: checkpoints/4160/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4160/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=ada5837392d5764aebfdfaf4d0fc0e00bd0c3a57dd0610dccd2f258841985640ede9ebefb53d493a748e83756f10f0e6d613948e638d94fbd7ec764a8e3297b474c11218ec3abfac80db871a7c6461d7c9e0943180a0db23c316a038022be3921a0a612d219611888911b821b89f2f355bd9e1857b59d46d33640f2ac1d8048de2975bcf5dbc0cbd630a4cb1f2d31afe8f579e6e3281de51976d8cf335bb4203f309658f1871996c66ac1a1b5385d3dc51b02e318be02c821d34802b8ee27f789ad68a8f80065a20212f21c6a7254ab0b426883c80be8d274e22416747392ba08d141342bd684d5816b5792a24ca8b73e8dfc64c1de74b43b12d45fbe22b8631&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.329021265Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0, Name: checkpoints/2240/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2240/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3e6e69abea6cc2fb9310d558e3f6a328f98ec2aeaa31303c1d7e6e48899a5202ed0172f9beb8a79f84f948ec469daa0934681a8baceb9511f746d20e112dce1e1ad91c51a823c03be85d02fd05ff99cf0c3a4bf89b48430600d5d1b8876c3a884ba8530d44fe4fe7c33f38b3746becffdee7677d1fdc140e0a506e4a88e8db41dfc1f4d69af89080b3c33144dfc4ad53d3febd43a1a133003ad815d96f54925dabf0f9b64b7bf228141cf182ee70c978586d9d07f219cfc84302c7831bcda001d862a7f80ec47af93b5aab303467044a87ff834f1312b5a3cc846e9ccfb3ea1b0403259a444a615bee2506aa7183593ffd3c8b2d4eff5ccfba90325a087f4265&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.329062027Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d, Name: checkpoints/512/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/512/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=59bfffbc62d8f8c7c13289f5a7cc6d8dd05fa243805bed170108c749d7697e7011eff793e10579f9b4b5f7b37e52d57b97e9f1f27d403d29a396e17d72d91599f89b31bed091902f23d122cd20ac24f091352e2c98aa3e37252ba7220dc29d083eec9fe47604a5abda32b9c996220ba561a53f93880049a94252041065c34f8c4f130be2e6d49816a4f336ccd19da0026b4efc724f5b915ead669f5463581eab1aa8488fc33723f507334e4dcc9d2b0e1f88cf99782eaf656fdcc1f03bbed30935541bbee18e3600fdcc31ef0dfbd4a8c9a32e190da0dc068939ee9e9cffbf7879e69c10819e4f9fef0422e8049884913303b76e67bce46a9395803ded84f538&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.329179207Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3392, Name: checkpoints/3392, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3392?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8362f5c0c4a84018b72a4a10482e4093491d41ca753f51f359c4a13337fc4c11d81360b4e14285571e1ee903a0522274e564a3364210f16eed03f1624b966700cd4ce29f203ea70b20f033504091e980c942e13f47ccc82590fd0cd24ff5d3801a609d449ca0f75f7ecdda5590035254ade988e35bd2bea742d2cf5adf87c45b47b17207fd2570ecfe9d04fa82660e159bcf1c864301e8cdbbf0c28ff5993a879ef7de5c207a5c3e98278b5aee1d1053125fc39b1d8711fca333309e5a873d1526b8a5bdf393bf6e6658e41152105d42571d8ea24b0a17141ef1029ee37479a832dd44e69bda739799da7421b1b36958dfd0deadbc1c3e106bf40856d3997141&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.329185874Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/d, Name: checkpoints/448/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/448/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a68e50a4fb677e66b6b9a857acf98ea77f8a8f3f3a4a0b9912c7c638824fe32c76581df5c972b09204d592e397ae3560ed43ce7095f8446a4b67115a46c7d96a34768fcdbdf838b6c68b7dd26e790e932a6d6211c8c343a748797cfd537cd84aeac0e59cefeeda241eeee7de3ec6cad740949a2204348820cd413e65a6abcd6130b0df741ff962bf4eb50d5c0ae2d05559fb345566683e05637aa7ac988ceafff9a9080ff326d87dc1d4d8e210382f3c8f8059582fde2e7b85358c5edb64aec887051e63d808d7d50c4fac8abea53564a32c984785f9783e9911990015286fc4100618beec7564a919d41f2bb7df15fb1e2f56c96d72bb381b8bc8d881ca4780&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.329198095Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d, Name: checkpoints/3264/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3264/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3dd91b11ff002e94fad87b47816a366b2c59fab5e91390e7588a92743083fedc582dce8832aee9079b4a6098eaf74c65aeb5dd1cbb5fa38167a9b2438981633a99e9e583fdcafcb8dac185a49899e579c2a98aecf1988d4c6a35a209afae5b2f6a254c133033c6d051eb2ab15942296f16e8e1b3850563dbd520797d5ac5318e95221e686f23c1e60af86aa4c5efbfa145b3636f5b5ea3e7f887c318dc3ec84cdc66798b8ed8d195bd7b90500ff2105618938fc0fbb8053366e6269ace59864b7fee95e30e414435b675cd96252fab1bcded60ae8cfa1835fb0f91e5c8e02580af546aee72eb6b70dac4c256a2ec681b723dbb59de1f302fd5106d601d04f065&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.329446678Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/d, Name: checkpoints/4416/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4416/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3e9993e2a0bf18b52d2ddfbc9f0a2cca69c2d17310c542da0a19264ba4b4c9872fb625357a76c3f5547c5c5c0ccee6495420de35795ad4bb4d1c0123e05509cdb8f686cd317d06a4496e8e2ca58c208ef8ba8b2e4a3c8b4355be4e0917e14b2fe35406cf2ef3a98afb4918ef98b618c8a2a8ac8663f4effb601e5f430390c4ddb3e374f2cec566d6ad679f8bdb0413a1c7f5bd8e035d777173384811e9517b1fbe1a4bf242b25792e78a17f0734f6205c71c0a9fc04cbb7974dddbe437b38b94c9db6c9b164d71b8dcfc46d7cb9e81eb93e59cf49246e80a4119db3c989879e71a912d4f47ca6cc17d4d4bd808e7cdcd80ae84ce1a6346dc5bf0dd651438211f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.32945945Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224, Name: checkpoints/4224, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4224?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=91ef39d0b2c989293d4349273d27ac6828c68b6881d3b660adf367263783ebdec0b4cd1b1d387ed123886250069ff5438d0bb7755a8ae71aa8f36e90c79597f5308e917c27e72b1ef0f6bfc7647b0302db72aae309dc02e1795107a4d4357736e426f03ba425b839d44279a6c723213ac6359e110bdbf36f0f7d7e8fa780c48acd857633a7b59bf71ae03218b3514c1b9eb9613098bfb226bb6a212ae3dbbc534dbc2738a878c0b27f9cf3880abfb6ab214566191f755791d93037bb8476c85c9212617630984be4cc94fc55d0c84bf014fc0554097a17f843852900288704b5c460f949122ee2ebf56e35d957d2dfc16a809af94192e1b5ab641fcfb173b5f0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.329465709Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/array_metadatas, Name: checkpoints/3456/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3456/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=938bd03c84f9e1cdc8300335ee547d84191d0e02dd19f56c6c81e0e4a4dc7a5a1edf4fc5500d10af17f3eec9af54790fcd13f1e2f66c4f15050d23f5342acf2e8429606ec32877dddfd1958b9d367afab0724367d7a5154cbe1fac3b997a9abc7c1f457df94cd3eba414c1ddafb9bc2e6a98ffdb0a35f938791f88ebc0ab9f2b3ecdcb623c325950509f159102f4c6f03a4344608aeabee7890a7ce8129f0aa3fd2828eb3dc2290649231ee6bf3d3c036349c2f5d3a52bb1eafd39f65764d1c213d4d491d2b3f9ee70b9b8f14fe85896b2f35d9e33ec478ca9821801e52753f6d0502171ac022cd78c01dd037c4a1eb16806ec98dabc8da83d1e444883ae9f4f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.329452782Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752, Name: checkpoints/2752, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2752?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=753ff52082c8533053fad208b3df7ab5e782bc08c7f5ed4854edd0b9979ff32a0fe96906c16045ab4d791ff0806531bee0041cc61e0f36f75eab351b567033c46c4fe8602014537c3f5dcee16452f66cb76cd808ca8020cd92b2fc3d70e36bd985e3039df1f934424cc65958082943ed648bbba5d328e152233cec047b08a8dbefc1fc2e63e89fcbd9791d3e3b3e0902180efe88f3b518d9372943845c50bcfb52a8c7c5a501281bfff0864d669ae6d2b5cfa33b6875e568e26b55e8c4170891bc62413d575412fc3bb2a3d5333c18d214b803526cb353c338be3e839c10841257480a73f68ce07beddec884dd8465f8849a12c54b28910ac6ebfdc6a7dce6a7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.329452657Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0, Name: checkpoints/0/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/0/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=769c6e61a459a45db27a00d3f3253b6afa7f45159c982885e91cab24f1f79ac0f86dcf41f5750cf41509d8a938b23185ed12fbfb67a67025593fcffb1a14e7a6d03ba649c5279b09d1d5527fedf266fb9eeffef0172a65dcf993ab566b5fc726796021d024a46d8d57ac5791ae820a3a0978f2d5c84a385bd7a8f94c114e6526d5d1cced1061ffa0e4a6fef11c50c26b0f8d3551df8889c737fc7d6a495b6fe895919ad9c0a90942ccf5620224215dab825ff5a3ed70b9b8ea7eb82a6c8dd477854c3b98d96a43325e2801333ef534a94e20ca92fa7bf2388c8a53285a900c878abb424bcbd69333485aa3cf3e2facffe1f2776d80d9e6f4d6db049fd07ce383&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.329522658Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0, Name: checkpoints/5824/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5824/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8611f246bf6ba8ab0397a8584fbf0bb8f74c777b64af10d6361a7acc5ac1324ed0f8bc45c8fe477872a43736841c1a746d88f2bbc51f3b4ad7043c8d15b1cefefc3f875bdb350d00f8c591b0004293e955dbc443acc015ea783e68bb79b9c29211bff2a96758fff2de968313f38394c773c901bc75c1c1ffc3a9b2bbd2833bb14e3002ef4c0552cc080868fc3480a52a773f64508c523bdb46dfb1774598aac623942020f555ac91953e21d5b0956255ef15592533af06cbdfa59d12cf3a1eea501f2bf8efe0fe21a9506a12e486b82ac9420e32a901e055671ee115afbd797ab31a8a21b3ec2f4a8199386d2dd7f03f34e4d9155fb50ed75a7955d2427dc059&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.329540146Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096, Name: checkpoints/4096, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4096?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=c4e116cf2318b76c775efb4873103ef594549d867accfc38e5dce65be4ab18c2fe9bff83f5f83c01f58a6f46d0ec67ca40fc6affa424e51c53e340575b388913691df5ddebc98d80e8547b13865b5abefddac7b8039c43d4eaf18722705f81f31350e2500a26cf9cb4358837eb5d3f5437bcc38fcff2a698c48dbf76602e84aa6da7c828e1f401606e294b5e138bcf5e7c2cebe8c2f1bc768b53dff7a37063531a3cc9e18f0fd271f6017f744b0e05430fd42fbe9fdaca010b2cdffbd814c974706eb4e1ad325f93c02477db4bd7711e7d79bb5244dc586b2ca7afc3168be99caee9dad88bc88b059ecb62a7ff268335925fcab8162655ef7e7cce1251899dd1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.329665865Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160/default/array_metadatas, Name: checkpoints/4160/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4160/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=22ccfb20eb436fe69df09c781c41a326741cfc0f04e2201417788c6b68a9583e8f7fcc9da4b808da7b2f46dcc8cfcc1c273ac36291cad7d07687c6af263d483f30a220945a35810374e3a8fb89abac5c1bd87beb5f2dc86ed2a3db70feed2229b710a6acf555151da4a06ef98bdaa9b7aa88ec55f53dadb572e5114ec4667d033f86cc2f318de163faa56789a0280bcae6bdeeaa2d106885ca594199f15e4a01b1b3af74a0d5d876c91fd34b9e18ac5f04595f17c58f2d74f92d4f6dbec29d5743aa09bfc5bb16c929a839d86da0d3accac424b8ffabb36d7690f512bac7456d22601eb27922891aa7cbbed8b5344ebb0e372b4865c8f8fd1826f329768a4c34&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.329706769Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/array_metadatas, Name: checkpoints/2304/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2304/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=99e20ac6f84fd1ff45f4adf12d25cc5ab62d7e844557bcbedc8aec50ee83545a563b56a8f2079c6462e9b19e179c40bf5b81b0b563f4f0a1b55e216f76a5e1e1a08a17e0e007d2418fd1f82ca180d3027a00a33c1c06f29011582a1868e2152184342ad07c37668b5d03ecaa70056406a1615f8239db6799867a7fe1fd4dbedd59c12d690fddbb31c9ae36dfda94ce0a7558afaadb22f1c54496190422f8dfcdbc0b4c2067dc7a420dbdc0186344d66c993dae045ac504571b75138705c3b4b0d5369d5f4d762be64d0fb93e0f1c1eb7a6b61123738b4c5768d0c196b594c53af5bc736dc5c1c264bf7af62439feb18e7235eb07f7220965c8cc1664f786296f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.329720278Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1088/default/d, Name: checkpoints/1088/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1088/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=0a5c7c11280a6c68163ef87b142feb1804c5dc67a99026a92acdb3acea4febb77d48848d04654ae739570eaaf01f8d2c97e723400d533babd07b897104fa3be347250a6cfbd4593be086c0e49f2a7d362df0018a3f40179342a9891fe8bb2a6b02d345c4120f352cb09437c5c08ad8f338e85b7f2918caca570a1befc6f7af03935b417c76e8d494ceca103d1b233029e4ee0304dae858acb46c52f3dbe6eefaf932734f5395f857771b785b69f1d305f771eeefefdbceae2b813ac902f607371bf8b37d86897a3a4534739a96005413a55b39b5f15d95d872b2ebbce2585a042b4b1d6b5f4fc63b4345ecb574dc75b1689473fc9bcd049a8e3d26a07b8b6cf4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.329730327Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/d, Name: checkpoints/3008/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3008/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=10a77a6bc2704bd8de8ea6b69c93ac0cff523006e3feb51653ea397d76cc057df1d31d2cc91203123c1418381b676d04c04a299141d3a220ce05898a2f31b904f061f0cb10ff2494b69167148e11de838cb6c42fdec40975887c350d56c1083e7d9b76892751368b9426e93b0b5731d7fff84743961e2b39bdf95b3ba74f02defc9c80151ec8f2bf7c8148b00ced24fe382ea5398516c9b9aafc550d72318047f3e79bad1dbdb9b23f5ba46029066c23f6fc59754b06737de3c557f10e68641249531a2107512472927b76a2644f4c79c5972bc160e886cca7d87d04fcddc89df8864b32e326d45c512a21ae9463661bf42ae3ff9bbbf3ebb1913112d817016b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.329738663Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/d, Name: checkpoints/5056/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5056/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b201830ec6fcd1c8b0b1776f938bd9bddb6a780714632eaec7951b0201eff30ad97e17157c27d0ba91fa8db7de7f9315976ac664f6d73a5708db525f329cb4d6dab180cdc5e1b14c7a6e5f0c93a749b05064dda8a1279b0ac6b6ce793b1f155576fdd1222e4a1923569c93e16dbd19e00da4228ba61581a3cfc81f6a4b63960080406ca7b0d236471ef2c33517e2226741ca11c6f5f3f5afe98aa1ca70831064db414e88aa4ab02c818fc3cba3d29de841799a286b40e97e28334d92eb485870db64c734a94b5d1eebaca7c68f90c18cfb32e16f6fc5d08ff37e23a546858af5ac8d3125bc68821cd5b6878d6b1cf0c2877ac5030944b52ae1460849997b8193&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.329732441Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0, Name: checkpoints/2112/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2112/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=aadd844056ad88997932f1806a4c206faefa9aa589fdf7edf2b1cf985373451446d8714725b746977b0a9958338b96d72684554d2be2f56da7147b7b7b32fa5296d79cc906011caac710b9d087ff5e8a6401c47753d99858fafee3c52424a981bed679fdeecbb4cbc150c0baec749298fc0460d6a1ba5236d7949573a2ab73787043c7cb8b294ecb306d98f61f0c5d70335ac8f2222c973ee767dab167c9ab197821afd030edccf3339cae8b1d101802f9cb7cb9fd0087c72fc23f4b39bd86794f0745836a88467415dc7ada933faad71284f81ce2422a111e00285ce7c1711caf09e03a01dcfb92234de50aa8ead659faab81ef23b346865721d84637d1a137&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.329743706Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/384/default/d, Name: checkpoints/384/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/384/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=ab75e439977d2622ccdc7f8a2adc34bf05478d155203126d487d4ac016a951d2b6403445507eeb96792b390ecd612b97e8ee1bb22ebde87f1db41247cb8f7ceff85c79fb9187edb09e03d8b452e8be4994f402ae07804574962ba964fa54dcebd9cb615e41109abbbea5e21cbaa1f651162469b7d7dff43ed8a381302a46d0b7932af0c0e28988f30be90969232acd4f409cb608d8e4f39aa58c0e710e949b8590f56e3429dfb6dff90bd5b42025f7187e812307e967bd520ad73c4773d647f7553731fd1836360c845fd151c70b0f1b742a210a671bcef1d4ccac70aa829506013021e7f75e2e3973feba988cea1b4d64a90e12d8042db321ee92001701953a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.329757238Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/704/default/array_metadatas, Name: checkpoints/704/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/704/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=70003c8c91d0912a20198c446357c77a9ff26dc6e75c7074d150753fdcad7e0f0823800deac8bd17f70d12b88373507319117e22d1f64f786e0f78e7a21fbda3c57b86d50b610b44e502ab5b354dc7579889f29fcdf9d3e885a67cbe145630cad6caf3dd7bad965edadb1af11d389d7dee1ce90818bcc0dc393816aa2eb91a962b4c81abc88723e5cacc90ff0aaf05969e077384856cf7aae7989c04a20b796ac84de9ef7675971d74171e713248b707f54732e08564d1e0558fcc03fca6a7089f63625abad3df37f9674a3190607cc7d399d103a5bf900a44809dfa246a3152c29f639868b1b37c364a503367c66b07ce68b8bcc4341f7671d8d941182e41e9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.329756836Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5184/default, Name: checkpoints/5184/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5184/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=82ab8876c490c409b160b889f63579b6c368638969e862bd86daa50a6964a85046637abe8a4d93c41511512b5fdd49494a3935042c2ab4c794dc4738ca4b513cc20b7727da29f6b6a92826fa46be58e255a7bebb926c46b9ac10858a2c605ef35ba5b5f5150215941153037267090c887e18975af04779944b3bad1e1f5499a6171434b3476b24faf3199961b8647a277a5b3b7c131cb4861aacbb8b7f751b9eafdd485f7579d29074c8a3dd6e7378684e2080b9ca01b20b44cfebfb4f9e565f9ea516e0f2bf06e4f2ed7b6df7f614f2a92e6a37b8a3060f5fa593ec94b1b63bb2505c2d69c87df2963d1d49ef7dfd5a5de2dcdc3175bc5398c0d1e0873e198f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330603227Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d, Name: checkpoints/192/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/192/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2ce4a0c7976f6b7e6a55f5f744d76f9a1e337d69c034e73e46198718b0186efe20db0aae4dbd0fd2383216bc671a2bd6aae17ed861bf0e6145e90e39d5f29d49beee8aed59f44198b00931be40b57e0de2127fbf078e4e0778112a1e067baa55e6aa23a25176a5e2f8e82cd99e793ccb9e0ff79f6387c4cf2ba10e174fe54c74d0a98043255c5d376561aef32f6a1fe029d70bc39d9fca81fe5bf343e3081279fc4126fd80447a5645f045b6390fcd3df74b25ba9c4dce1df1a1556a0f7ba8fc6b475df8a6b9685a33e74b15e50eab55c33fa0918583719ab7338b64540d4cc32fcbec782a3471d8860f2cee7fad321bf568c49b8d2609852ab8422d1ed480fc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330600489Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default, Name: checkpoints/64/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/64/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=62b00a6e662f7f91fa076d354d3fb7f4cd323b6f756f89e00098c30a63110a91fd6bd6364a47148dac9091c8f1f35d0c786c425830daae12764089c7f0451fd5dd811d02ea256c8c3a9be5d14a3f324bf780adbd030113cac214373daf6042c3b27fb304adc5c158edb11519673a485a94505b953d78e98ba1f27efe9aa70a0936b7acfc55f640df47b69de20bcc77336f0bd96c56f2884a578d948b10c7e67574066f4d698e12c5d48ea742e6b9c11cc9fda797d8513ea8b8e78fbf875e6dd9512311221974873b4283bbcca09b9d7ed2f63579a865e1245c5c25a8016d093b6373eaada119a73ee8477ed6951dbf7948cb4482721739d28f7ba9c90a9b2716&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330647189Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0, Name: checkpoints/320/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/320/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=9895f7fb9103bc80b6e3bcfea0ee3abc00051aef8d3903689405380437562221e55fd1c6c126aedf244dc5b6f2f82b928fb0196e86454aae042df013714ace76705fbaf5083e641ec7553430b9cfefcffd29d67b329c98534efccb2bef461c04f8ef5191685380bd2cd73153d43fb388b15e8887eac13df00fd66968c60c6967a188b2f3bf1031cade529b9d763690f24ca0fc22468fb58eadbd9bd3a22bb8b3221fd2e183c8de2cc230d524956f27e4eb7d4405918dd8c76e361383f226762ec4430c6f8fe8dda51e038425cfdc72879863a9f53c28e7b60fa594cdc4173fe438f9d9999938fa2437848f2389309f632d6e9e7ec14344eafcf17931f9dd8194&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330628791Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default, Name: checkpoints/5056/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5056/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=4a5836c3018e7f1753c3da1bd4822161055f589a2f456a7c40a17608ac37f73d2204b5979e71c4bbb020c563800e3e24fde20d1c0aa156af72bd10ec5f13c46ecdb04c6bc67e80f967f7e3ca60e80c26a3f37d860de7c1e296b293c9318c241cb03f166e1da584069869476b5d02d753c712a41de14f0a69728cb9704ba06e23fb0d46129e425367d07ed2ae9deb7543befdda821e8060229d79b7f08da070d5b4ecf7fba5242d937036fcd4f6402a7fd7026fff5046ce202f6479f0bf6357a2a51db6473b58e089b93453b3327386907c54bf7e1051825a2a94931a219a1d601980af3817bd1db1f995f4768b4809c391a52fbe8b6383851b49a01842b5350b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330657481Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0, Name: checkpoints/192/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/192/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=20e504816d715a4ad997305a4404697a01a533eeba434fec2528420a0020a6b405491208c3a0e7ab6ba96e8e531ccd30edc45a16a3061a28f48a1e35564cc3485c4408592745f502f35b1c6672ed33176f46968414e5382521b3010469bbc9d98a1b5c6aa243e24dd1cf23c2c4b4429b29dc88e85eae96d62ec86da7cabd6376ebd3aa75314e42666d0e36ab5a814eaffe15975ec61f715f17edeefa50acada72d24441cd8e5f453c5bb68940ca578a1c95149dff90b63c78826ff5e2ee613ccb7e188ef113a2a88dd3729f5c87d9097fbdc4f2fe2c8cd7072cd35cffa5f075b70c5fcfbad69ba8924dbc7318859e2ea0a11893c1889e9ec2bb1cca44ffe0087&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330659281Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3264/default, Name: checkpoints/3264/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3264/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3e30806d696abcd40fcc5f3ef60adfbd89313d86384ca0abea13cfaec91fb62c0e767f140576925254f65ebcd161e54f855c33cb24880982f9e3ecd7b6d33ff335c6bf7f37dbdcab3e866055c99617e58011738792d833dd04c933d7a878e75ff2343f66f09b90cef50af2676c9d6b6aa2fc4ee78013e3d8b58bc99ee35bb6c5e98e6835bfce9695344165794648dbe1812d27f3e6990dd775e2f0ef8102f84da39ae86884bb76c84b732675aea72a2171cc7a6430943d09ebead32e010a10026ad20e830762e4f73c87c8f6d24fa52cb5d0f254e74460ee1684bb544afc5d4a29639f9a42b435b2f865b4961aba388f7084f1eefa768939921bd14d19256980&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330658264Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/array_metadatas, Name: checkpoints/2240/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2240/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=14beab1ef6e4dc4cbd831d0616eabfd53ee74304dbd7b4c274f32e949d654bdb998b1b0fe05fae322a06c9a5bd1245ec7eb623d5f07ae348c42852e49c57bb8c97a5422990c4a829c4f24d3ec96de7feeae232eec459fd3811ed3cd5410657f209b73a61ac48e8aeffcb5a3ad010acb8e508dcfff7d9f8083b2c0be51f31d535fa3b2cf0dea1f6c9febf0df792ab14163816b052c8a5da0018e6b84f899ec30bbff12168503c7d52b22eaa6f64230e984f7dc8bba18efac057f885f466fe906eee2ae808d37375cdb9c06a229f23c21308990dae571465ac15235ccf300bcee712b395661f6dc1004e26a7e1f4789cab992fabbc1cc568c1ce4b296016713350&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330739673Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1280/default/d, Name: checkpoints/1280/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1280/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=9ee2316e433a78cb9a0adb35985e8982467541605b4048eb56c701a6a7e8a57e3b426439b9a2ef74464a15d4ff94ceeabaabfc9da85883b6300387e7ddd55f0bafa432c0d70507306e8ad43d2d4e6d3095f5230ed44ff1171f0d5479ff2e1b7510dd86c81b76e60c524f1b050ca3ee2c82bc2f6a3c16fe45f8f73b61c81abcc63a043e77a3ddca26065e7475cab5d96473ce0ddc559c78918a8d45375d968831f72483f86943560c4fb1c87b1550486ed28e63219bb4be16c4e1a7a1608c17bece3474fb5175a1f52690bd6501586292882f992916e41bcbd2badc66c5e4297e72f26e5ec655adea4150df9a47870beb422d6ca69ff92dec681e2b8d43d59809&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330738587Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default, Name: checkpoints/3008/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3008/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=6853aed694c4058720d7adab0f80e1685adeab792a80b20b2440572ba227b82cdc4cb5ef38155eedcb56419cbf5a0e5e1a88a03dcae7c845b555c47d45f5be07bfeaea22e7dc65d221ed28c98a306698f9644543c46d8dcefbb3adf9a36e67ac9425ef91670220716803749eef20064205b2e01fcbba9f2876dceddaa3e329908de9ed769f219c8294236e8f4aff24f2f4bc784d0cb418801b911d8738a95f8301fc06aa39bab632a53ffda2be335b78acc48d5b1e732c86d46e8913744cd781c0c27f7e76d3d00d45501c28ac68b17a94615d2489445ba3f2062ed1a110fb9e05b0956c369419827abcfa9b4d63bba8485bd55e80002253a2fec4fe45d7d902&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330761903Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d, Name: checkpoints/1920/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1920/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=9a5ed9e3f686ebf7cf3929b01bf7a22f9e329bec2cef8f9993108b4c6234f5cbf27ab12a9b3bab17baa0a0b8669ee25c25e3add2e3a6db13d9d1c24697f5e1725faa634f3d27071d76c662a460d0e4fcabb48092381f700eae33f1db79c524bb292abf580677b6b08b049529c163025b902e61042187cf68d3a5d5b9da22103a6853bbe4f186abc03052289ff6933cdd7d51be1f2046a3ea05920dcd4cec63ef3942d2a9997c5bff43edabb4106d801a7c5b44dfb7efc50179df2cc7db5c7b1b66792126ec9a151672da10ffd4fb94fdfe7e6e1444ceb15a974fc1bd77fcc57f40cfe1e896e52e4fdffce78dbea4229f41b59eb19b12fe4fe316f0545d5780ba&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330757508Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4864, Name: checkpoints/4864, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4864?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=29944cb370757a53f9e0caaf90a771ccc4344dcfd7dc6c6e2ae828481f8198bd41dcd050bdea06759c39036a0cb8df448b21050bbf40ad439c2099ea200617dbe30d0eea1ad82f5e3c9fb5df165d657e3d82e8ca9ed0c941cf516a7d4bb8674857912503a31779b47fda7f1b4af82bd48cfdf1c699ef81163828ce4831faa64af176c0bd5a6ed2d4866a38ba1c7e5aaad96662f84543bc6f0b1af10486058cfe7d3329dcebea82a5092f8fcdc97d3b479bfe32f7acfac716a26bad3fd1506eb1c816870cf3d319022653eec84986697dbf6bf1f643b76e0d72590d8d1066d7c4846840b0da37930cc4cf72aad920f848ccb90603695095fa56bf5543be730b6f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330778214Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d, Name: checkpoints/3008/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3008/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=bf829daac7c712e3cdbc91736d1888d84b2c1c757a9fbf6acea53c074032be3536ce0408d8ac318334d6b4b46abee44e21c29d238212fdccebcb87f92c1bb0b98147f68816cd0113989101de9628917d414c3a3d35f8f3e0b9377173384af4226583c5366b8b2ca34c40c61b634d7d6217620a9b2bb541f2a247eb6f59a880eefbb5de75fd2f174b94b6dd641b1af7f82a540fd244c441ab895c5e9f807c3ba72bc863bb0110c41f3fe335af19221a7aea9b2d4f332e757d6f0bf659a13a48026e044a6918eff452892affaca7f93e4d2b33fbaeb91752d91c0a62e45fb8a26a459c1962f9e506523d7740c2323ffcadbf20578b1902204a1737b0067cc5dc3b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330808245Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d, Name: checkpoints/4480/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4480/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=36de487ddebde36422b9ba7f5968df4e15ca4d98b5ff87438e06c13a32c05334e3e56e302196e1a350ca8937e3345ef2361e95af23d06c8444eadefd163db8e39059e10bcfb6b20396b5ce6d1d37b9e5e02159313d7fc40d756a296b6902d5c7581778477c84b83bc5a46e8d3db1477f61a3aa944ad807fbe0112edf1aae1190794987e2e9501a5464b41cbc955578871c518a8f995ccc25c4ca2f1922bae073a44a785bb5d640e484f60ee79a6481e28c088b96652fe6daba85e274d7eaee9778d2b561af332e55855cec8a60d4a8e3b06a3b83a8278ee25e4e19901428803dc6daf2befc5025e31d435179e31d9d2d3dbebb7072ddb9c6cedb2590b300ee63&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330808483Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0, Name: checkpoints/5120/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5120/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=41aa11b7c38b4b85c4937a25d0adff9b2893f65b9db1e7fdc22faef47cabb9757aae097524f709bf539b8912b87ded1782ea51b76be14822bcc736c2b1c002448088ecdaa10293ec29a13c7e33d99af583c89bbe331ef5165591902fe031c88c45861dbd01349317bbe99f8b12f5954d05725e47d06ff7073e055bf58f2bca7a5e0c082e59140261f09a8aab848feedb0078d2568765b8f711379267c18708668f3fdf5006ed66a3d03d8cae24030df44cee8e2adf72e800a0c0ce5560dfb2301263be11200784101340568cecb8c640c8d484703ee1dfe9c97e49ed0e52ac337f47a52c88e09b05f4cded68ab7e98094e8e149de0ce746fb25ef0b9394bcb1f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330820007Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default, Name: checkpoints/2752/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2752/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=0c73c6ffa9220c2357ff77b361e568364ad1ceade213156be43d3e537942e6a5b0681b6c494df15b41b8dc1d59c89d5bc42b26baa88c19db380f0fb30e744ba6b21d928023a459dcd56f1868dd435f255a2d308a248568471b9a5370dbfebcb2a1e8cd4037711a3fbe081d0b457ba7945ea67ebba677ae5205841e02647885a22fbf0ece7becfda471263560a2a8108031f571aaf2bf649effb0ff6406e37101d89e8c963e21f416c210441461f5d3fc0bf8ffbf7cf9f9adb56d6750205982f7e73bb97ee19f1338fd4e880dbcb7ed0ba9fa335d607b2b3ea3823d685b002df9a83352ec4724a8a6dbf19b52c488e0d8ceba7e52767cdde81d21189f6328280e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330812966Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4416/default/array_metadatas, Name: checkpoints/4416/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4416/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=40d3291c5e3e5ccdebc65c5f1eda14e4b57657a4afda4c7857cc70fac87df3c18bf427e82c34a77e14a415e4d363f99bdc862df4eadcb7f75a88cc6a5364f4e2171f985645359b4496e94f03c28043b5241db393c2a2f48f9fd1953866940ab9f46132172facc3f19134860a473c237a3955bd3902ed11972a5235066a0d60be2582cbca446734b77ae0ac8a3674f6a01a8e347971647cb1b3efc6bff0d639c22ec4ca8d1eebac7bb64e85503e54775f4afb003fcf6eeb645b2a2ab9c9c8537b2709c1316e912f6e7d61bd4625115f419236d85edeba36db1d2e9dcee584935ee7ba893b2740d3ad877b293993b28ce1972c3acb2b6d6caaa491c629ee17acfa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330847085Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d, Name: checkpoints/448/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/448/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3fe1f1c2824fecd2e4afabdb507655b4119295ed41832aaf0954142d8bc98eb863294b0479430aca5871a406157ecac226cf99af676db42b265a2f3ea3c7eb0020e1628eb5969b08f5752a9b0f2338fc19d7971bfadadd394284f1d9f8225f42d8d733889ebc2438d3699f7163f5d6c0aed0ce8978c93a7456dae99392cb27ebf10279d05651ace64b90e747fa756cbb8241aeecc098ef78a4f75ebd736810def1af3d170ab5ce5f709f181d41b68868e820a3d2070c75b1ea7259fa683afe715ebc6934711e1ef6d011988306a0e24f218db3b4d2332bbd7e923baf0d96ca126ab1c5f57d182254c076984e0dd8606cf7dbb14d1b42aa160e6c3b2d2f94e571&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330859047Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/256/default, Name: checkpoints/256/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/256/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=88fb45a13d3bc001855e0f5dbcd4d52ee22b5a58efd4c811647e3af44c2152e24eccaaea3777c0874002519b6e2352c54ea6e0de3a7a9f6060181636e8f1ae27526b406fb9a08e8dd2405596dacd71018c57615c358052054cb906bcc81594b6664c4c32310a9b58270dc9bcecaa2a0a66f6f8de8350134cdaa7546c78ecb9bd08949b6b33abd7de53642dd0dd19793fecd378c8d3cd5632ad8c1727e1cb7dc169c0705145814ea010bb665e10aa5e8556b2ca0d7651cad26e12a6f3c4e27dd7251656ff5b3ad633456311d404a6b3e13ea58abbbe82f2b8d6bbae94660d2890284e2752f092d515c072bbdc31a60446c6c936e26f3b2d6d1defd8ecb8dde8cd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.33085829Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3712, Name: checkpoints/3712, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3712?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=99c312f5d123bd4369ad00337a8d9fbd4ac369d337dc1cff2dd3f6f2339a7eb0bec77ebc17572a718412cd86f881b772f2a4a730de3f018294befaaebcd643b516d75d8e9d633fdf500f945d2582bc144aad404b03067af894b782165e6f8cd5047016d3f2adfd042350dbd41261d1f1320918f058008b1e369a5ba2d1b83efd72aa668dfdddb15e61195cbc5451e2e1876835251a24d83fd53bdce281c01ee4808e6def5a9b81e1a21f238072bf214d590db5161d6609f878f93f16f6994559f926aa8ab2008a8322804f4db43b4bc81b7bbbea536874ec0e483c660fe3461d6daba7bd861b42dd7fcc6fbcc7ff31e85ee079ca5ed1d3cd706ff63f6743cbff&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330852464Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3968/default/d, Name: checkpoints/3968/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3968/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=4a68a0f705f4201d4bc4bb7c7ef8a43efb89df32e1b96b416dfd2f6417d534887952df14b00e26de87010c4237688773d717a1cb8f9e5e4e3657b733c0e50b74b90721907758b12a4c0fd9c29f0ad8e21bc3efaa36e82913001c76ec9ba0f164df90c334481a0ea40643de9518b76955605d0c5a5bab0c6c0c6a921795e8786cb9e22da03658c3f14366b65c91fba3790d124d12988a95efd1e30df946688d2cb1a2ce9dc927db3f8b27eaf97e33bf30305dba32bba91a62ed246245c6ac8ef961f7f76c9a2cd22baa1523423d6083fb8e209f93600967e97ae8d1d2529a5fbd3b217bf7711c7b7213e47d9139669443e1826547474f643238e05bad6eca6fba&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330868039Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default/d, Name: checkpoints/1216/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1216/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=83ce7c068c165152be80b1c4be3247ee4eb92ec6d8f61c734dbfc4578bbaf444a5af37d0bf91e1dd4bb71995513b6d66b43dee610c1a4463cd0f2fd0827a25900990d944c1ba2d916ffbb12f08f7b05e6b8839d6b846e35ed5bbe79c34dcb092fd5f733d04980232beed240ddfca02183b963679baaec207b0e4a5c2631180527b5aa3bb71c789c1be1c840cd66c5864190d337e76589437746581d3718457fb99a9adb11512e8aa808d4a153841982dca8e0a164e42a0e44cb7a60908a3c109342d815fdbb7c8832d93c37b8592e6d480fe23f1f08bef17228abd35f42f5578f1991f2cda1f893d2ea3180eacde7580e95a9003db468c709289befc18274aec&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330905593Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1856, Name: checkpoints/1856, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1856?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=583a9bf7b99f7c4b072c1d0fded5fe05768e5386cf75578809c96b870936e76acadd7e8a5884d7ca81ce474070114df34fa330d07846b2fe13cbb417b005331aae91d42e29563b433436b2059089e6d1430d7e47135ac0e42c51299ddc2f31fa227b79e992b3f86ec652d1fa90389b64cbdff87d7f3cf8be053000136e5fdfd4a8ab251f82ee8ad2f9bc05a0f7c472b8d1b12fec6b95876004e20d7391e4040cdd98d148a29f3176d8bfe2ccf9ad08661ea37e609bf1f8a7c46bc94948cd8aa1c1ed5eef70cf66fdec7a8451dd37fba7261178179aa70ed054ad4fe35fb6b6cf0abab72eae8cf37bdb3ae2021deb889bfdfc8992bf6e2fe16a6301ed68983122&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.33090383Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d, Name: checkpoints/3904/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3904/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8830b2cdf4cb04b0aa138bd7efa74353ca93fc29137505215a85fa2fa3d2aa3a76b862fbc49d6cc7e403acec38e3b0af2c1d0e13d36cafd1509b1832a5107ef0c0cc8916d9180a7f1bc1527d12b2defd2c1194ffeb5c78bc8a06c3810f7a716e09f60cd62b4084efcaee241e2e89f563598279e64b8ea2f27e8e39f871b6b83084a5a74aa8e3354e5885af6fe032ca94e4915215c0af9b716c560b0c9802cbddb484dac21741246489564130ee36d5553a58c3214d3caa9a418cad9bdc6620a96fe95625f9b6ed49d5bef0be8fcaa15b9091267ac7556512bd531a516499eeef534a5e4b3f323a8e2ef60b7009052d0063181312cfc42a1ff4672f3087a3b6a6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330918912Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072/default/d, Name: checkpoints/3072/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3072/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a172e8f75f71f2471e56e8bc2d2772b75b0163ea3e2bc925b158383e6008699833dd6a18ab3f037a6060915e88868406fbfa18044c4ad7fce7bdccaacd5f41bcca7028e125b7cebe2306daac628247dc7b401f2252c6c4ec436dd2b59a7619407bd88a4b5e2b3871f72eac9a3f444fad39e09249f1925c7ead80e629422f9d62944216bc95bce855f7661988af8691b68010945f7f2a838ef82dd2200d78cff13e5b9e60dc3643e2f9ebff9db9da08d0835fddfff04754a027cae8d41b639cd302228a8ac3a0b6e4abf83d6760e4f2e787b0beac0b03f1363575f4ee4d668c56f7278f30105a44592801c1f73e038ccab119c86b0810961572d783852c8aef2c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330918168Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/array_metadatas, Name: checkpoints/5824/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5824/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=40608ee125862b9e3b6aa663503ba4f1ea6e2648396537c561ef3ccac2678a38cb2ddec393d2e2ec7b751e47da1a2159e08032e18d575ab5f6f4ee7c64cc1c0178d124165b567bedc2c668e7413fc557b4bb1fad2d5b80b0df662e113eeb35604a17650c1650f5d8f7b04679ba58dfd2dc7629e1544e747fc36f0644c71a53e59826fac1e0b4884b470bdd319961b6379638740f80e4fb282cef35fadf5956ad4a71a336acb6e12bdadd4f84eb45fee32886a1c2d80b3445a72715d7e19a289a4b1145cc1ddc1f67cae74f6236c491dcaa9bf937c36668a0920cd700c79cf16b6f1c0a27e8486872734309ca02d2d5cedefdfdb73eb57abfe93085626137bf2c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330949207Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/896, Name: checkpoints/896, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/896?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=aa34729bc8508a0d4203f803cf71a990894fe01b565dc25c5ebdb6f0a43ba02ea2d217b81cc22bb608258ac6d2ebe8ca118a0b10f0913b14791f52d800a015aac9102ce954775f264cc321b7f4f13b3d2a75200775a486a4b3f4063b91b8db1fcc62ebbcf83f2d213c3ddaa1a2d3ca4acc660bf504add75f869fec62ecba37198e1e3f8e17ce576c0b45b134a650768b0f467ab9c66356f505c050cdb5c6e6619379e34330fc1ad8fc7afcb102db6b2520c06e06fed3ebca6786b450f65146e9bb764004becfc8976151e9005221fc748e20e93af374d9a1525ccd336c974f0dc39b56e0334416358228d6c3467b5e9e5f9c778cd15374a2a0075c8b5f746c41&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330937907Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0, Name: checkpoints/3456/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3456/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3c1e329182c82a3669e43199002823c57c425e6be131318af5a7fab2c3f945fea388f74a4e9668d074687e312b358314f8867ae2d0bbe7ec48d55bb8087876ac31f5ae4538bebe674b09d687ddd8f3dab77fb764dcf7c375d7f0f5c82b96cefd6c8e0fd8ad904df3b10c44bec69c9830ef1dffafdcd6443dd695b2ed9b27a650f2584bab75f7c8e8bd8ce4527ddc5e5ab67aad087a3c1b5312711ddd8a93ffdaaea38937dcbee71f6d39b57ab70d42c97016cfd6c64ad41dbf2b102c473d939bb92b466534c586cb22f5baf21f1a473ea9bbde3f07424754cb1e7d7ba47b0f16bf4ae10c08098149914ceeb331b89d394b9ca47f6e26cfae35e0ca65e3098f8d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.330974034Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816, Name: checkpoints/2816, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2816?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8175891f13417783dc9d15f24cb92c15afe22ef0868c943fd192480cd7c216f62f1855feee3faee3cb5f9852e1fbc25cd0d90bef4f243d44c093654df8b21d11d1582abc28c3c1ec8944e016ea66cde4a6c3fa2b038f70ac97b4562124c1f19afe775f3555b44faa3298e60bf556e2d72c4425a48c9954699427993d944b21d78129ff0cc6e931b88768e1717af29ce28f1cf265ad2b18771bbce572b626dd7e46ac69ec239024219c426a6a2705784d20b2c946827cad8b17c077f6717c84a88b9f74f204ebbf84b562af9a5ef9236e1257d65fe13cc396b2b4205aff3a7b2bfba0c0980b2d4d1d2d4727a7031dad967bcc9712b3a3404bd0da1bd43cf8f259&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331001189Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1600/default, Name: checkpoints/1600/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1600/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8b253a35b70fe44e5d9aea60fb49b7b3368b058a9c573a2769b7457a3bdb77790f41b00e25dd70ef53a1e5306bb55c7714e0c0b192fe7eb97dd67fd5ccbc0b4f64ed99e3a100d4a9f6bb989864f1bec47d77cc5f464167c52b92c418f3a8601f318fa6467cba9df4ac4f6dd023df1c61c1c996b459f9b02a93c44a9e1130fde94fc726e4e0d3743727c71b7d8cdf5cb4078d084a19c1fb9d5ec7dbc53c23062b3d7dcd291375971cceaf5969732caea3d6b7ffec1e853b622434283685e795b42bc44f1f38d4314e7581306ae9af2be9b705a3bf8e0c9029061f8c7da4004133843291c2e46280b31406ab13a6d2fd14f6fc853b2bbcb770f6f73fe93ec82b7c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331000386Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0, Name: checkpoints/2304/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2304/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=773695646e3639b6a2f60576fe5e6ee035274d707a591ad16cef1e197bf7b108640877e1e5c9a623103f5cd5a39666429940c3931ed5011ff0cad1d293ce7d327e12916fda031ed4808bd7ece6b4463f8dcc1c59f6d7a082deb89b018708fa551d1715e7a2d9b8624281b36d46bfea5a621aacc31e3009b76aecda5b1a97adfc126e82034858e730eec8ae6da1f957d6d910f9e4a01ef58684015f499238f00b2db5ad47a3c893199ad60018a93238e0495f1306ff8aaa42dc86a6494ad09f5e52cd2f70ba70ca983b55298e89538d33ac17b23d0fa11334874d6b643ec76c3188a62ac589e1feef36581c3ddd880ba0a84f64e47a18962ff2fbf0738e6afe9b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331002453Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/d, Name: checkpoints/2624/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2624/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=11741877f270347fae6b8168980774dc36214788c63e2b9bd899cbf6ffc2d6ccfa38a67ec959d3182be01d81be1daa8d42ada30840283ee2e9d876d006c94cbb2a55e1ca61b6613b56ef16f83da4bb88271b11e98f6ac467c1c7337da9efdb912f4cf5ed307db2e5736e5efcd03cfcba2a38fd1073732148daca94d0b49bd85f750c7dbeee5c4bc7c9a130ac32138ac1cd99eb27a9248b4459089660294ad6b554797ae647995ab6ced20caf21eea86612e00c5e650c216b035de4f1e345dcbbc9e269fd0aab551e25a03edbb3fe3a12b0c73f660660238b3cb678e095be337b86ec92011258008681053d776c5f38f867a8bed9c86803b88e66fdc696815512&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331028691Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d, Name: checkpoints/5568/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5568/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=59a2cd30c0516f642592044c605ab39a0f55385cead7b9631788cefc0a68669b64b0bc442674d4c4ad2bd61569f4ece0d6b6afe8accde767ca2c5688553017d6fd4e1071ff6ef839540e27628b9edeaf29a90c816992bdf199b11acbf24d4adf2feeaca1257eae7687661937ef0016665aa173036d0d41089ab0844f072a25201c973ee77a7db6ae7d7bda80c5c8c9172fc5c42d4432510f826f6af86f69178ca43e383df4310fabde70a1f494781c171bfb8dc92999b39cb0d37202be3c732422e7fc661e52910570ebfd0b36660a0113f1db3facd416432f71cb9013a67368dc6e002c714781374a5b7777ed2084b1ef77f14186c1e2386a8d8eb17aca26a7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331029382Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints, Name: checkpoints, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=44c698d0aa95977618d5ee15ef29ca9ea263517b5fa032b36e7c493a11434119c870b54f7787949588b154eb3616d9335d37d243444cb643ca0665682a6e055bce50ac88576ede7d1773df5f115f21c64f9fc3d1ec0196a07225568d05c214c359f1fd666040b472754c270dbe5002f70d27850e2c58b76caed61c16e2681f2fcebecc94332615b33b1d97179c4dae06e46ff3d63550acff89c8f7ceb427f498f3c2d2f2fdf772d815dee6da84f74daa91dacd3b1a2d847f2be6e531faa7a3c90079af01ccd608309338a595cd182d37ef6c18b9b771c907a130ecf3af9516d4b6ef4c636450cd7e0b32ae017becbe1fcce4aa803d1628c209288d5cc4d42bfa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331045053Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d, Name: checkpoints/4096/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4096/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=083b45b46e3dd259a63893d5e584202831daf52484a30a77caa90b13a7f2113813f477a6b259874d618d421d425dfdedcfef5d4582590d941aadb0af31e7ec79184c7d2d676fe9524e9439afa355ee3632c9295134f606e880fc2102b76b160cec8b4a40cee11097c1d72c457c92ec312a84a3725b43856f12cdf30565b717d9aba5a762b434cb82630c20a493b83b60e57046068115a2a614bd7b4e665b43e78ebcd9403d371c08a87e9f2e83fc8c266418bf4693b44355c8e94439f3ecd3fe9183cf888cfe78e047a8d35858e691b2b3a8e82916ceaa28d3adb768fff44d095f9e656de34d1cc14d19c5904775ef3143c99adf04a2ba0a9d6945a4014f07a2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331069517Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/d, Name: checkpoints/4608/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4608/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=abcd6da7bfabfb866dbab8e68079e4173ae873cb5ca1d50c9dc64b597cdeb558afeacaf78c0b9d97ec9de3582e38adfe02f70f67f02cfe7d6349f4bc12822efe726764120cd5efb33fc8c6ec730976793a74101f8f7856da33db151ed04954a9c21ea1be5c2b2e8eb5d5474d3f9ac5ed4380ba73ab7ed49709f464e47ca30fdc686f2c744bd202dd421384c6e31c523b1474bde73011286db0a3350f97092c78da94865f5e96c5374022f40c0881c16768bd6bec1131d036fb533592790f51770fe6b60fd79b8eaaf16f7ca666b162f658b7f31b2daff447930a34fbdae81c22a867accd1a0a227ca559da8aad5acfe39494a0c1e99cccdc9d4fc69d179de48a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.33109238Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3776/default, Name: checkpoints/3776/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3776/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=75395a187bdf8fd1f4307fffa4672e090a7358a78940dd7050fdc5ea243c474fe0723c2450b7b6f9eef1e5e4a3d5314b96214d84e6cf04e295d2d88b99969e6d58f633180efbf65278251fab1f6367acfe885019ce1b69daa95503e96ff3dc44b9df8c80c28a29636db315899f47da32de20c4ddb029ddf68d998246e29b3b370605883839d24c6479e3209685b1f687d7a215667b770775a8374aa79b84ab8ccabc641202f1f9a560de9463ab928944232dced505fac3d98ecfac74281821b73e5f0245dd9daff58d5f671986f0a7e9b8dc02bb6d34a23a69d4e4fdf99ee59277d8e287e9dcd049b09921ad91cce0d4de11fcb8c485f4eb98da47e9d1390159&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331093895Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/array_metadatas, Name: checkpoints/2112/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2112/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=bb06f239c5456ea4efdd49da39d7aa47d0ba07714dad23fd4519bb4c68613bcb8d3435a08b0ebbbc2b0f7c6620c9adb33deddb028e49cf8eea56e2bcc314102342270feaa8a4c2f94c35bd459d62d4b67dc708c9a182d32f9208cf964a29cb0c5c8744cef9d23fe69f015e3baae55cb9d53621f1e09c8eac324f2275b2fdc8c607deea94f5b1ce3675a153cf164e7c77fbcff7df248a72b0bf1c8da35ea56e4577bec595be1eb929c4d1fb812ce3bbb6df47ada732baad9eaaba05b878ba6594927f2cb35a071e0228660f912337c62bace269753050804c11f6313ad2530367593186fdf5ba269a3dd4d7e0a9d266a39a4e7c6484fc0c2f55a0384ff5ee1fba&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331118366Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/d, Name: checkpoints/64/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/64/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=4aaf6e383804be0db1a262e883672725fe88927b4da426e446a64bd04599c3d55c43581e2b41a95a3302565943126a0fcdfcc55c11d53b8ba37e53c92b2f1cfabc1924fda3d40edf7b24e629758763082d6c43cfcbd534d9398de2ed08e3ddc3f9c149e5f8362d9f58c545a1da43de0c7f4583283c1ad3b9c3efc9cd5e87aed2a484632a0015be9ea85593fedbc1f43e2562db5fbbb095026a3131d65ebee92299c47cc99e5c20c76cbe8743b111b359a9be6d0ff45897d0f738ea0d0b7277020454770d4f16f6eec79a004ca661c5d9b09728b0f6b4fa960aa155e514883c4f7a6d6a96cc6ac1a8a70fa33765bb43d32892251852e440cb5d2dc51cc611f913&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331119835Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d, Name: checkpoints/128/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/128/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=7ed85906e3b575ae734a2f8893d209ef486763fa567b6de74e35f3f48ea82c449866cdb91a4dad36571029025b094209159aefb25d6b87c8e39a71b22e306669bff266d8e33b7731f8e287c1906a834c0ffae08a5764b120d4c7f72b0099fba614425a2536b7ffe22699c3035a27afef8d5b25f74ad93d151c2b29027922daf9d33ec73ebf26ec940ad5838dc3ee272909a8227c712a50a552945be1dc7d3859dc54451707e20ae9586cc798e7ec1ee22d78929970894ef913347f3231dd6d4f1bc8ffa8bb38c2e6e720fc8f042f5462185dbdb28fd16cd43b4aa62cac2b75a633cb03e7563de6062d185d2b4d4ae8cc42962b5a754ce572fcc37166d9be84e3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331126942Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4160, Name: checkpoints/4160, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4160?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=621d0742bd351a6f18a37648ce3a6f955db099bc5e51c937d8bdac9c66df93aac062896dd6e1b83cfdfba16e9aec1f27538666dc1c12a6395a73881f69d070b90e7af12ca39e81628f9a12540c99b1014b12ab27a7412de516bd328565ccbefabc85c30c972fb77cd68eb8b52bccf136c84193ef192734bc342df69add004f472d8a03223d62dd2c8b3b50e64635f4a4512150e0e011292e6e5fdb663c1d632f0b3163809fe185798bfabe0871ba87b6fee4c376bd1489ea37902f5f7efa35963b97d74ceb527b43db9dd859d9bd5f53838b7fd465822f5617a3666e62e3ddfa1b541459e07c7496ec72fea36fd22fcfafb706b0573d1b4e21a0515f8d17f193&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331147063Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2560/default, Name: checkpoints/2560/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2560/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a2bdebef03f5cccc9e980a068c2a10c4b7a6802a39c9b13d2d6fe6e94682d8cbb9c795aa6bfc231791f1cf6cd654edec16de9354106119436be2b5ab9ba84b017139096e0f89439909b264175888a2138089d761da968075100bd3693c9c4d9efa3d8523da6dc3369fa5ec1d66892994c950c0d34ac8326638cb4f4e4b14d0834ab8dfe40b858fafcab08f764367c8997a7d46a779e4606c6acb21152ac16d8a6cba2ad3ac39598ffb9827f853d958538fe4f78d7937283131eb6eb70e66e5f166daa532026dcb471a47d6dee88663c03b3f4c2fca166b03d4fd4fe5bd09112fad804af19be1b1f4c6be0a30fba2e76d94e5a990722582fb422d88763971377d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331166111Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0, Name: checkpoints/3584/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3584/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=880d57b28ece8316b1fc8136a7a14e81420124e092f0b3fbb7d6710e99b5bce802cbc149e2051329238afc05d6af899a17b737f47315c433ef49d7fb9fa635befa00f8203434aed2c9a2c635e242b892770f6adb499727cb9f748c67dbfc920c4ddd56a0591d573fcaee9a53059df3b9c0e3b940dd30760ce94d30d0fece5177afe7468699973036c14b1745c944f587cfe3a4dfa34c619b799409a493410db09c921c5a22178128827814c397410eb81175368620b4089b89ea1601b04e6a63c0a63e9c3d75c7c61609263e397cd1bd88b87d8b9c87eca17912d6448f66addfa445e5af95f0da2c8e0febb88f28ff40aa1c7336e2c873c97e319329b087a96e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331182083Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default/d, Name: checkpoints/3200/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3200/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=5af0314737a373735708391556e064e184197d374640d35e5190a7bf64028f1d74a8326eb2f4d8ec84fb21284d94d7e6e8d174efdea0a2a45455aa0b560dc8dcd6705f3e9c501d1b14fe856164a7f4ce1693a3f53b47d9352fd57dbe92ba8c202122190cb32d23a6d0013c6862b1dab08a86af1e8f2ebaeb443490434dadbd9b323b39200e3ccc7bec4fef58b4a8753e362930a2f3ef57f7b0e2fd707e0547e9bde98b3a17b059f9dbebe2533006d3de7f4cf286585c33f1c8e7b01bba69bc984613d5d2c0f758eebf08f8cb726aa0c64cfb985bf87e02f519b7dfb12b2bf60d24aa40f9600f60eda96890a1be7283eb142d620d43df5d81d96a462aebdecec9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331176005Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0, Name: checkpoints/4288/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4288/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=c3b411803d9a5caee08bbbe3f8221cbc0863e877e53b1ba549bc46d4c7ba1c4f372eaca34addbbe05cb423437e5b9263357f7ce3c9e6acb9ca5b57f58b347f8bd760db9f456e4b4a535e965d5dee97581919651f11cd11b330842048cbb72bdd9c15b148316c994f2c47cfc1d5535737e70323cb5c15709de645794b2806704f6e8ffdb3cffffe196c3b1b22e2caf18332cb5f229794eb48301a77960c10797b528487d161a04f13e3e52bb43d48a45b564c85a2d455969efcd212b6c5f146ee8629107b5f5d5885508ddbdbd49f18a7e5777f22ad4cd5a026a15127ac2556f25dd53dfb4c093fdf493c2a89db599e7add2cf93ea41d1d3c7497c5a8eb44420b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331180381Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d, Name: checkpoints/2304/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2304/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=0a56d7538348212a95a2ec1691308242f0df35b9b6d889544fe77087e5a0c58b3fe68828b4651b2944fb596ba323a78719f5a9a133b66a9ad32a330935492e5fbe9b0c3360ab5cf18bb5e0cb70ca6ddfa3319bef9f39871b306ba8977132b395a38776087775509739778929729240956bb092314e2d612f1e1e20a074b348e792cc40431d5d4a4fc0ddc3422767d110b971f8f04e4d8100d113b5b26db9d9e56e739bf1655c81b5e29b4da95f7d5138760d8ab76115e10490659ab773ac112a87a9f867ea1ace8ab660adb3070a6334ad4bcf9abdb4e570f4b76c277ee9674c577794a84490ad89abb4df7773f4e439538dd8a3d098f546170094085adab29e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331184041Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3200/default, Name: checkpoints/3200/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3200/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=16dcae891c6192650c5620634b183dfe423ada56be428829bb4ddc1973a7c1640b7affb8b853bf1bc1e87f74e6c3d1e08ec99f6623cea59ca8afaecb7917cc81d09afcb5a9e992321af45d6be617c89623ef2382249126051bd351c100aa4b8e4a8a114357547911b2b2059e2d62f95399f7199f8fb31f05bbce70bf27506494915460d68cfd01fcdcb03520f9dc2ffba5fef26ad5814327d362d117c1b6a14a6270aed9cc15a467814c725eb56c6c12af89744612c20d87079a67c9dc4872931e683ebec348a960f9bcad31aafc9dd867152bc99ec16efcb4aa347b87661dae95b63c0a558d67c0f86c2ac5c52f58f268bc9616d23ca6ef361d44a84fcd802a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331226465Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d, Name: checkpoints/2112/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2112/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=0cba9b078dc66e5d06062fd0973c81311ce239d3a79084ed96e42057278b1647943efdd1c57caeeadb8595052822d427eaa8a281614e1085194992cb8b2ef48375c798972cf68b8c2993f2a79b81b169e481b73187ba76497e0891ef17f9346fa20581e3baf3324ccd77fe12da593b9962e5f7d435715e301fa91bfd769ca132d8a1181b2fc2191e60801dd03f2068d5252e3c06806ff986e1e4561dbbc8d413566bc423f59a42ab2efba4c600e1ac5c06673710e1a220712de4e4ded8fada47bf74f59567c2f9b12d27d92cae27a377762d350de6e4f17cf6a0283863474d6bbaffd6b42dcf6d69994e7f810184cb7535d2b4f71001d41daca686a6134c5299&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324660124Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0, Name: checkpoints/5056/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5056/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=6a841a18f9c981fb2938529852b66f52dcd489f4e8d51abdf3aa23208685f25350b9d51d5a2adce3fe05e8e09542cd5ccc7b17262267ce048db6ec2142d0bb11818dcf358d11a2e6d7f0f19eddde093d1be416cbe6ef6f6364f2b1188d2cb72aba36b7dbea3ca74206c3d82b10dcb2df619b26ff49c194032071756a1396b1e732816b3e12edf8a89549ff1a4d4379791da427206c422045b3c82d1555d62b5cf44f4b45f7ac8b26bc54f30580d5a207dd62a6a32d89339cbee0d12cf57b2a7d126e6e7a6bdfd85e9061918908b5d35269a07273b929c66c9499c12d89e67badb888e2fb62955a96245d22c46c4f0ea52026561d7db87cf85a2a59d6f3caf692&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331260901Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/array_metadatas, Name: checkpoints/768/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/768/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=5c73eaf8d60e26db599126469f33f367ffd97205335a0edb35f549372bbfc420ba1b4d35a7a80c6dae81ab2bbde085fd0c88baa199c63f7ec774e94b940bfc5a0b6b9f11eab7c34eb06a3cb1bf99d1d0278f8732e39ce4d95d8d71925d1237f23a6a75319f70473b21c3592ed7f27704db3138aa684e52b78690a4af1af3a9ab4b5a4f07a242c24fed7a7fcb4b0fabd020bc3722939b7648e9596faa8ef653da01f6579fbaef0688fd6061e2b3dad23ead7bbe19560c4d794301a07773e74761696f9512850267e4c33a388bcee70ae5a38c8ff781a0e30330f68bd8304f6d2d3cc3a532b44dc19918120d2c8844bf57d7583fbf823198dd41e872f836b511d3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331316219Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default/d, Name: checkpoints/2304/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2304/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=385ae1adab59682c4d23467b6e8f644dd2e00c3a8839eb75a11d99376482027aa578fc80a39c8b306fd69e663a85854c0fd0cc7971c8e69f52ea1decc8713cb7e44e5240b85ba056333ccdded95b0693bdba1b0d9a2aa055eaff8aa699515fb82810929cfd3e469cb8a58ac673802f870ae2a49d49cd84bbae96c6510cf0043f0426a4bd90648dc581ad06e3124e8b96b6b43362b6b51dc91a834b41a55005213d700858edf77c071b0e57af5096e13ce969e28568dae496b7ff2a8d98224ad3032cdd5a427d266e3504df715710f82cc2d86de1f70d4741c032d2ac2ea07ae2648c0c2390638d133cb1986e0c7cfd57e37a4b30dbde8c724c7bd60fff904718&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.324656735Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default, Name: checkpoints/1536/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1536/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=6757d47d699ad62c21280b39379b7090f9144eb872a241b050e80295941e9084b603799e265f2d634a18ab1878dd182b5974622a8cfd7063372ee91e42c2c54ae234c72cca2c620454238dd7cd0ee1fca60cb522cfca5b462cd64a79b0b6fe17e9acc871f9e5030c86129a5e39eb8ec4143be280a9aa2a5d1d126c4d1d962bef1419db08d7ac673c203513bc7207cef58099e8387193dd25ac93224f6d7e4050fac957b9c9a4ce0f7b0b51fc4ef0859331212def4b2e83db1d162528171ce2cae8fc7718f6ac060b9a2c820d2896af0886987c9773a2d0c449cafda5864b450f790e34ad62c992b281dc86183f1e1b424448e2ba2569bbf20681625da76af0fc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331354958Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4672/default/d, Name: checkpoints/4672/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4672/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=40fa24dc09ed2c8c6da277750b34a2dc252ff00a227242392d8aba00dbed99bea8effd3347df8323a249c0c074231f556b2c40d20d10fcd17508688d690906d6a72b47372f2fc9a1d72037c24b241c60257e0bf7662ac193f4f13a5a84bf12ed17f081882692e370cf46fe939848f53827ef0b06c91cee44ba41f5a6d6542abe31bf07afb27b79bbe811768a2fda33616cd053efe9bd9cd0ae384f92c9dd0d05b82a5dd578cbcdd711c3b22e6951016afa9ab98a68cd80321e3df864cd71f8b3cceff01bc53cc4e7aa372268db5fd862a0b2e64b7793ee478ffc108762307e7c90647c4414753d9b71f2f5e19e4d8ac6d3aa41db16e72a408778b951bb5386b1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331359747Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5312/default, Name: checkpoints/5312/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5312/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=0fc5f32269277fde93b0af09a6657af12049b9cb710158f1d19757f25eaeca6730ac8bbc73cd5a4da9a0d80f0f37969706305f7dca8697150ad38a7e1fa5166d1215baa7c7c50b7a1944837f10fa5c0d52629e9deaf7643ac8be62f0661d73ff19bd98b4b70c974cd2e372e97a0b8911e3cb2b69283c2c57b712219db61eb45112ca44aadef94f24a9d4e6a4b950bc9d601cdc200531569e6131b68b90acbe062beff6c24e4ade31467d129b7af8a6473efe8dd91130488fe86b48969cedc3324f353404e6af35384cbb79048a090076ffb353073e0416ac5f58f736ef85977d9716db607a4bc97c86647630ed3454f303d464001673e40f62383cb89bd2b2c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331386031Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0, Name: checkpoints/5376/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5376/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=96a47217d7e14a75652033bafa9975be650ce683a2bb7081d64a15358273934854c8c9d866f5557f3e8f865f2d2bd32c04abbd1f6ae29ea41e08ee5dc80748594bf75e71d1bd89b81b12f924d465f6764dadebacdfa4b18d775ed20535977a8375c0f9c016759e5adbab1f501889d9e655a5464b326e292ae9259099093157ad9712f5278f4caa50d48c52321973e99491adf5b6b1dac1e43091dbcce5079743aed1fe4203135aa8373bcfd97780d5ca88d2caf73ce12a1c3870589a9dfaeebb3f098ab328f81577c84c84296a0775583d45858e5549dcc8f9b6409dfaf13655fd5e8041397618006d631a4769e8416dbd38a8133e553e66611dd3061bcdfd44&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331373488Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2112/default/d, Name: checkpoints/2112/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2112/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=c0d90ae36e0194327de399b360c49eb9940d75eabc5ca1035aa6bcab87a7b9250d0b09f0fea0f4d27930e79999d2d0736d5c4359981b1c990f151eb098306204f86273df0e3ef4b3a6a3064c2883a43c7ee333a1c5e7c3f5ce456d68495dd5dba745071cecb89fcd4cfea2f86e11553e6ca81aefaf996540184cc51b06bb33e32e67ed5323134e7e6bf2105d16250bf65ddbf78d59965024d61b16a2134d8957a5c66d503992a160f5567c1ee71ddfa9e3982b8c724ec44026e43608904b35b66fa5f28e05e7052f670a3160358c488cfd6b10f2bc9b5ab3cdbc87d6e9901d760c8361413915f550f9a60c4ee1afa587e0b0f71bc1aa85621a3bbeb897e6cf88&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.33137776Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d, Name: checkpoints/2624/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2624/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=5f5be1de77d588b7d15a746b8f92cf619e49e88e0b34f0a9db1a7fcc34947efc81de4e8dd14f062b7139b9282e0db552776a298b962aea5cbaf7fa741e0cbfa454f21171ebf3867626f2b976984a82e437d79f8c826d070c063a3a3dc21d84cea3b0d82bf5342b7f00f38433c9a73c9555430ec94f4ac920fa4f8dd849a26836f1f00ce05dd3a9baef5db3c5e5967b7c08839c505c838eb9fccfff9cadd8d87dbbcc9008e36d9bc1662bfcf28cc0da463ec17a4bcf8e7267b3f84ad78347eb45f1171d1303418d64b959ed43234b039bed6accc412f1e472b38c7f75d7e1cacd7a5f373add52d288cf2f4ec81a41b01d64ae9b60a23f31e3745e7080dd1d7d8f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331404142Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2304/default, Name: checkpoints/2304/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2304/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=015a59f2591c88246ea22a276f7c777d5e7588d690b779f28eab8173348c4cae7312c4085ba169de819b16233fffcc79d8d3f743618127cd669f6ead84671171bf0ca19903b87e70cedc756b4dba824f6016e24afd4c40491e72f6c2114e7637f8bc099c370b854eb09a3384c8be53849310b10d5f67caf637cb9e6a65f0e62a3440ecc2b334ec5f7a46a7e5c27f99b2fe771825170e84353b74e53eecf2b0c12f7c38c74738b103408a22e19a35c21462aa5d399a4bc83382b2f6e0123f5ecb4e06e2c272355fd2eb8d7d9d18356f7d658263c99eea2977b49ef392abc60c22e8ed1a1999af0ac071da5d8f9e5af2dff6eb7628e40b19f31b0f1ae79d039b47&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331458484Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default, Name: checkpoints/6016/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/6016/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b4e7d9717d31e87d9b30b4fd7510c192765813e902558b198e0cd5623bd797da589d9d8ba75d2c5da0a59568074d69e3f1785adea59078adb638e78920cb45ee0a28c48b939e705a3785fb1e3047aa7e8b9313ac70111d87617bffb185caf43bedc2771f50048a3cab80f733a357248addbcbf4cf5abd26c10ff7ce6046907105a2881c714cd0ef8d1aa5efc423b0f90b15edd63a0deca2847ceb849d8ddf5911bd0a93e21df9701dffb0964ae7630461c5b2e7589bcd6f455970343b2e0f704f14615030a3f6915f6b5e44a4a0e5ec6e03875e4cd334acee31a0ad971b4991a82d04731d2010d3555413eb7de26c9057b319f0b7c18b51089cf477170915889&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331526324Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3072, Name: checkpoints/3072, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3072?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=9ca7205f330a48a52b088cbd56b4335dfb4a1e61272750efbbab6218d6dfcde41752adb0dcba177870e9dd60cd42c0a9d6a7afce53bcf0e976c033fcaa2192baba6a40db31490d8500f065a52c525c6681280016a4bd7f5770f9030a5951d5973d399bd9b0aeb582d91c0cece6c5289999955a0d426a58fee7d247e3d2fced08d8a581f49238e13ffba269290c55dc7a70f9c2ce88241edddcb02f1ffcc2b78fecbee291f1085576fb2f1f68f6049443b5119cb7c1a73f12121e2a305fa96781631261ea3b5205c1c2b9ed567c8b1601b12cc1aa2e9e32198320f98ca3278b2f8aec67faa4c2913e52497db49f3aaa617d3d2005d0c6a67be8aa9ffa8568281a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331523742Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0, Name: checkpoints/2880/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2880/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=12a846573b48d8e83aeffbe54d76bbb07e15fcbe22c0ba41a0cab0a69693f5446dbcedf2b35bfae577e751ec274435be95d4aca35e520d87bdf94561ec2e32e3da3b7473a0f9fdbd5d8867c245f3265649b45baaf20d2ea36ef3de62337d7905505fa0f338029ff785f750520a12224ee5025ed1ca260758cf6a021800e44184d87b61091086e879cf3767b1ea2d8b5f4a8471bfa8165d33c3d582b7066af8b8203c782882368a86859236b9f4e1d2fa5f27bafe13ec4bac741d2dfe1e09edd9ea437d217c51ba822b13588edb12c97bc3dc9a99e36e4142daf2c1b898645bd9357a614fd310f3336a87caf54b3c6569c541686d8fa1d4b8aa94102923e487ad&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331544973Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/d, Name: checkpoints/576/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/576/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=8e8e52e9dba56e0ac89704b73bf5c1261c73db4dd8d23864e4940f1a295edc92a81d0eda9755ff1428b9e88f9e05555bbc6dadee947e137520b0a18c52a7df28a3acddb5b6f754b5f1219ae50d59c139c206139416fbed376d371a2afeade0ad7f92da3a28429d2eb9a1d89ea2891f80854a42e04e42393aa734b1863e3772b6a6a672a7a2b5e790c5e3803a9b974c7e0440c83e42007ef5688e10ad9215423d2a18e1e2507536df98339422abe6b921c4c6af66eb471bbe8b8c19ac7381209a3e0e9cb5bb9131539d6da357c217413e3f5f34045ac1b19fe3d65d7c31377673e6bc6e0436c8a7f222db93e5b3ce6c3d7779db0f73cc73ea2e91642af75ddb28&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.33157704Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1216/default, Name: checkpoints/1216/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1216/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=c28a654765e9a57465b08bb4e79073eea02aade6b4101be24165ec54ba4dfbbaa0ec839400ea2a8417af9d1d52b0e07a96cf0ae7fcb0f859062ca9ef0e73df8b855e2e4bc405f3b52321166807e6ca877c62dff4d0be39e8336626b312f228e0591cbfe72b3fdd254c6dc179df80d5036c325eedb6a63fb1657dd80d064223b3bc9d8b09ca501a4db34148c521e8e4b08cb3fc080115e385695885a7ba535e63c74e3fb3af5b03c210d64de4f20fb2823587d2a8e5600246a56465c5023f043ecbc597892075156ea8c16147d94e1165eacf238b60432281051be3c6feea693a49cba60beecdf4919e8b7d1e5ba1224ac0fc2515b37c2f5ed84351da3043687d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331625456Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2816/default, Name: checkpoints/2816/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2816/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=b7962bcf292287362224302e899943c23915698c4b25cb8a58741bf83c4556320a6c0384f82c0d88c1fc243a266bc52ad6d8d81ee8df4a10172a3920452109c818ad476f29ba71c5531e8e5bee26da22bf0df810edf045f1635b062f0780f10356abdf60e66d77b57b242af6cbabc07f10f59cc2e2f83e287d7be9407615691c7f1545be9d0fb108d152f437e3c06d64de05023f44aae20af619661679c4ed3edce7f7c243f6009a89650dc3033de1a42599e23ed6038a040788e660bb69285560a1c4f21b9697740bf607542eb8f7f4b5cbbeb1eb74edec40242ff5194b2f63feca22787120e219875f44679ea1330f3319ac289363a4400a4fed7e7711de8d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.33163222Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/768/default/d, Name: checkpoints/768/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/768/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=93f93201afb1a579ff876a58911b6b75f5d4311a2a483a6985882707c1ee2af481840a06554c26da85c4c42a5573b7d95a741ca6a3e3e7dffe716d1b6914e5a65497e2031af7720489a94e5f8e1dc933a8a4bd67ec6ca5d953402b76c5c96dbf5ebaaf670a582dd1ef3d7e41e969be26a04827fec6d5351723e8a396c14c0c4ded188f4e229f5bdd895a57960592788c64293c8684decadb7c56890fa8a6c96063fb6a252b11ff1b9b2b92c5f000557d4e24aea614ffc5e35479704bc9c6320ce7e8216a2f256f90d04691bf63c570fd36f8afaa357479caba9b8e15c75154d4f01407ba7e4b3aae27ac99d69931b6f60ebedd0b1c8112c856cd092058c17862&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331638785Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0, Name: checkpoints/576/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/576/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2ccd5a8fce7c8e8f3c8f9491faffa24b66122f18ddbd60d74141848ef716701c2c34939b8ffff5dc2ed4baf39893c4a42d1f2ee24cf077cb3d362c760fd793bcc0a7b7c83ca1c1c91f001b6133df58bce7b9d8cb3c188a8c86f7a4c8ba4f5a2959fc337f399f9d2849554b4fea0c681c83026d78b675f842bb2006b116d98d30b9100d2638ff99ece12e194453244e3fdc7f9a1624aaea4bac8754bc73ee71272ef212563871a6c4645ff65323bb47f9db0514aaa58158f8289f952aa6b61387c864b5ec591fec3712b9925c5442b70c44b0560aa56fee4236eabaf855692251f2bb4a1c9578558481d3890e3a4ec47277a7160153d272d989629ad0d6d5228d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.33173143Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5376/default, Name: checkpoints/5376/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5376/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=653e80152840e35099188c8c945aeb3ade8b87cfd01c31d8b2ee9cb3578aac753c2a29aa8f9e0706c0820a4c81fcdf217bdc6f1f66b3b4e6810a9e38270e827f04d3f008242763a830cd40120b356f95d28d4120d5c17d2d3061b61fdd65709dd92a17ae2889a488200da585ca5ad9a24bec01e322a791599ce457837758c244f20200ea13832d798f143dc5e80ff1e4683def0133077c2838ed00519bb028b61dde1a5c3307d0ad06ac55ccd7a740de78a77b88e893f0be7e66ce9bf6cd3023b7cf1641aabf46c8ac7943089793686a6a3bd60d020c4f5a5bd2f9df161d650673367c18b74961c58bc96eaffb3a5f60d663a45d2ebbf620551001250874e050&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331814489Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d, Name: checkpoints/3328/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3328/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=9100052adc4380995ff297ba04ad5b3ac1e594a72a3987385704f8f83aa2886f0a9ef9387462d7550caa8b93b8200d3c74d4d124fcae026c7b161cd0d27b7dd781ea84c613bc557fe5ed05db13709abd1c08c9494ddf5556b2485d65628b602a66fb909e2cc920c8f577754fd76122d651b282d1f026b4a32fabf8be38d8f00cb0f75689df028083a7cd3ff823021ec13aecfc757ab31ba39be99ccd25b5f4950a1b6961243c6d8a82609d0821d4645324a3b7e228ae18ed786fbb0e69f44c5b0484c03ebd167eac688f7eb594ed8955230a43ef0c959c4a5efc848889c7d840d92c6d58ca6ec48f97c9a2f71fa50f2bc9ae7d7ef1053ac8911f2a316ddbf4a3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331820603Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0, Name: checkpoints/832/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/832/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=86af4ccb7eb5d407f855e16ecf0d83016f83fd30617b49052a8ec7a200e0bc2e83010a1cf92c9085a16c3c564d7460c1d1ecb2cdb7185de8c2dc03b755d0b274a09e36f72016e462c6cfcca8257b1247e34259b17792a3214516322d1ca282cf849a47e584fc696712623452a4fce91290d7d96ce56b0436b481252ef2c9c4a00e22a65d9f81276451f03aef69ad4320925a5b0dc4962efd1c6b41d5b0c937671ed13a073e8667703d731f1a0500cc1340f936d9f7197a48291cd2bef98ca174330d752cdcffe29941736b8bc1c8d6ca8b05e660181347a651b876ec62911216540478b76aef9200b87cda46f6238d4de2802e3b7f7daaa6b0c9003e71c8728e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331820245Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d, Name: checkpoints/832/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/832/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=43d576c018f2b953c49c0a214970313f26643035c5c3269b7faf46740f48b76b5473ae3260fb369fc1ad931e2dc0aba498dfcdfc469c49ff0cf89dcdc0bd56cf0d065dfbe67e312e9f97273d74843de3f24b9456c18e5c71eaf87687f426af57be02843d2d6921258d47fa85fa8ba92901a0b7e4612f8cd94db13f40b24b51dc10dfd74ff73f781a0be5ec2f957ef83547041814a4ec742dc18303ae8cb66076a69eda7eca813a0fe4b25c3118260b332aadbf141a0bd04e9b1e41bf724488de6742aaef919989da1b16d81ca5f928b45734f1fe967c8d961473d056d256b76fff81fc62a44599167903d0475e82d89bc642282ebf05add1d78c7816fce84e1d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331825922Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/832/default/d, Name: checkpoints/832/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/832/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=162eebdd3be2d5960da189ba2f03f04ad257e4de79f3f5cad4559f0c9a5c41dfa8572d0d84fb4d716720586a1e3cc56dd9d747c9edb3c1929b2a6275f21d89f6fa6674556ad2485b46fdf4784cc9b5c7711bc35ce71995ea5951c667327939203a7b0b2d2d2f10944a154080e7a54ee20ef3b8178dcc065b2f572084bd2329b0298e8caad7a115ee3edf53e313cc47db6413dec68060f8e74988e95b86e9a5e9927b6836e6ee0490908e3190a3ffc4d7b3c171935645982ce49cbc56d93fa9ceb0d15fd7d12bef80717b7a391be2f4a3bd14286be99fbeef7ea1abb36073100fef87c5a9f4d466f2dd9540705fe9639533e0bdd8cfd7c05de8988eb86d39dd85&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.33188417Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d, Name: checkpoints/1536/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1536/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=4bf1c6c6568115bc3ea18fb08ceff0cba123c083ee09d0dff11ed7d7e7748def0480ff5734138e0ad2e2a6d59a28ff497465fa42cdd1e73bc5132b7df2a31e59f4c5c3a7b020fea7d631e2849d704bd3e4f4d6864b23b018262b00909f5e18e72f12965dcc75180883bff8c799f5cd476ec47265e5ab132cdca969c41d64acfaf1728884d8714945955c11db7143ac0e7fab0becd693e9a2aa78e573bf2503d861eb20ad91068200a6ac4aef1957e4976c2acfe607a80e1dae323390f6580134b027522a15defd788a2605f5472e3525ec395de064fc586d9721652e0669cb38348cbd4f6685e661f8da2d50a75ccbd30069ee7d9050cb2c1248963637e96146&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331878131Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d, Name: checkpoints/5824/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5824/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=6ff0de25b72f468c9a393014915fdf4e6ccf26be59f98eb56f672999e6b9b1684009b4d69972a819e59e235824e6b70d7d8d084de0446e0fb42fd5d78f39161e900d483346de40cba96f67a3a4a3262756a44cde42a849831fe4d4b9ba7898f1754ff7dc613963f95b30a6e85f02029ae9319970e1620380bb395f70eefbb8d001513da1ae55ab39e1d2c965752ea085932d76e8a1dea778957cb46920d8b653b1fe98b7dc84441aa11e7b7a0ed05b99a6348eb4a238ed8151b8b14600628d7f1e46246e677f093f5ea25c815427bca194b9f0b84e89473b6d291f67ab2ec42a9e32eee25dbc869ba05d6cb681e618501ad6256ff620f4e7b5044b47828a23ee&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331897104Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0, Name: checkpoints/64/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/64/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=935194339e97d73418b894d87be34d9a55dfbacf2f4124c2e85896ca1ef2caa3a04648a2af06fe597fea6a85936b29ce57072849052528d1a3bf7dc921d661d3f374c4378c4aaa3fd49a571724dbd0244afb4214ab686228440d1c44b7480b87ff69422af0ae74f8f5173933aa80c65aa0f3d7f43e80790a2502c4c4c6ddf79457267cd702934c1f02abe230d4a567aaa0a1cfb361ef539a1eb6d38ee6316504f840783fc7f0e332e83f78aed4a9e694121ac4e6d1449ce6a148db25a5264b2690d3a4433a56bc1f652f6bfc820768d84b0f3fcadcb5af1afc07b31dbc20df465b6d8f21df269dda02d1c885f35fa4f9ac0084810d6853c8631630b957a94600&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.33188292Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1024/default/d, Name: checkpoints/1024/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1024/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1c72c2d33648ce9a201a3acfea5f2987cf5e1e6c26c5776f44e215c504110ac4f855cf59cc8e3647acd4bcac5ce9f5b05b73f7bd942f20e61b4785ba0d3fa8b8f796221d8b583a90ee4fda2112824c3273ed1e2821cfa04dbba0ebdd631d9fe37ab9bb347eda4d57903d4f5a29ba634936ec4d5ca7d66242edfb21236111d1fa119f58e97e26a5ec7f7e615ba0141501475fabcfd0e1d113b9512a23b7d884957b145120955ec9c4e7943b16cc59ae238ba4e58b0034982329bb64594dabbfb0099231d60de17b62e6856253520b64fde8beb5d7e2350846dd823cc378a05592fb93073e2e4fe9170f8bc801eef878ed91b11651df9892998ec6445b7f23b441&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331895891Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4544/default/array_metadatas, Name: checkpoints/4544/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4544/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=6c26681294e9471c96a08f334e7e1e88c85142d1e2c2dd8f2ea0d9efa15b685550e8cdd22ddce129a6e8fa305613c23786ce32843913882630c132edbb226b29f786074da0599e075801f231480f4fa76fedf93f6671f026b0371324b350f0b81e0ef022a66582609b5068920550d85f48324f60066faf5bb7725637bd3ec42d76050f309556e7f9cdc17b819bf68b44f3274079d246e6776ac65afe072eabfe09e74299ee5c44164a7bd7fbc7817dda3fb41e36030375a26e923b515697378997241f173623171ecc593e4753c34305cdded8fbc29bc41694ccf24072d14887897e6eab7ffed3dc265f95ae8e4f9b686790d69adc4f2ccd588ce2ea92879c0e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331914017Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0, Name: checkpoints/2752/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2752/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=70a1a1ebf9424ce0479c3fe20095ece63cd0bb3dd343228f0bc195a449cdd02a3006ae8141362bbec0a75650db44562972c3036957a36d7745e6083131799437714c5dcf19d81bb3d3d64803d601dfd5eb48d63a0b9d8e61400600f239844fb88c2e82d3bfd12180eeee1b29e021d9ec20bc312b74fce71c5bee2f987509ec8cf4e09d163c03b42a1bfb47a9457c45c99333f443511883e73a7f43d7bfaa578782c27bbdad40a4e1f7ee3deb87fc6250af33dff4c1cf5c11dac7b19ada344d9050eaeb9603657787441117631c680977bb48955a38b0c44427599f951da5b724d64bdb05a46b50eb82dac4353057814ec6547b7f83e83e4ecb10d18d767a5c87&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.331974998Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1536/default/d, Name: checkpoints/1536/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1536/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=9ccb3f7d59e52a3187fef8e999849a105935c6bc843ed2fa6764d2c7e6caaff9c99df423c66631f223da0736ac7a07d0093df578b720ac938d3bf6010fb152600c28898927c447d0b1ab24ad15c062a62b7d976e585314ef176d24726a1767132771c529b37c2e58ce6905444d1bd3fa6b630f4f1b015787e8aea7c653abf78df4c5699af0acbf23e9405901ebb58b43d752f35285bfdcd538ce6b13bb70e6bd35d19ce4893d93f57e3157a7ce8ee5f879817372f2e6116eb98a0c90eed9170f074f92353c5ebf5edb5404c41abc4b466256a498e9fc99e063f61b9eb1439eea0edbad1d7b9f15af0f7e35a4b4b99474679eee70917738b282ee412e704cd0f7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332038472Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5568/default/d, Name: checkpoints/5568/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5568/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=84e977f345148dba5c19f47be7c65aaab5d9b2e171cbcbfd7dc3a70e5908d102e610fe4fb7e6320c3e6ea166430376b7d97d8d825a79690e1bd723d2200386de7334d8ca15aff5229d14d6c289e57fbf4bd7a9cf2f76b2f4d801566b68e83a7c1ccee48cead24b44de055dc8b172aaf94e993b826ef843ae615de6e1540d2f67d0c3d3914802bb330f570a0c6e612e4cdd886ffb99ed20251fa356892b17d2fee3736fe31eea157121bd832e3b023e05fdedb84a9a35999b116fafdabf1c2c73cef8bb9e4943c7b85e38e231bd146e42bc36845319987c8862941a22152f86a0fe4f468aed3dadc78e427d99428864d14f1ae771c9d1362be899fd41a469a4d6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332050137Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3648, Name: checkpoints/3648, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3648?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=5d6e720862a10c96083d666369c4e9ddcd2838d5cc47a9070c4af4669be935ed61416d2c8a6ca68ba98595faffba6229e4bf841e253264f20f6548b91e6f5658353141628cd69a509013135e7f0a159aa2b7e990d4ef621839855b955397154d0bdc917ceac18ec0bfdd90d86799c1a85d2b09d0412c6f4c85942bed4522ca7e4ee2080a0a2b8c6bb0879dbba68f91aafe014c29a79b3888fcfbadad875c0a6829db82c2806c82e2b73517d2cd817a81f3107af7c2c7863a4b2ae8e6b11ae5f2ff12e1a783aecfaad85fddae575fb9a843b892648df45e8a474f0375842bcd84d7ed43fcd29ed78fc18fe302734c131196008bcd950b14e73df0b70105633a11&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332045132Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/d, Name: checkpoints/5696/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5696/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=486fde579edc8623963df753bcd0b3191d62f00428dc149d46bf11c5da8b1bc5b4f811dedbc12084ef40dedcd4cc7dad0860f146558b99706eba79f416b7be5a8bf9bb0602fb94821ade4f197cd8a90a7885e937025cd66cc6f6ca442dccca57f18becac4fbdf82c2f57d378cec43c6dbc9cacd4b55cb1c92ac8dd12d9c5617c1c71ea9d10439e8cef3fe317ffe6aac5218370cc97e0f8705b9f00155ebb5bd449ac218ff922b3fa741ed8f59106f63ac6f3fcf52121ddebc96191f7e92958f5ba7e434eb763c359666f793a89c8461e5560cd7a1574c75fea60ef3b4669c55e511ff2c16d1f428320cf4b125074c603f08918efac5682972a9d8d777e1daf25&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332082671Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1664/default, Name: checkpoints/1664/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1664/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=10da239d6d98664d4cad090c29c350f77a44b13172e38834d37979b0c08daae57910099a6d3ea9f0f31155491434658324b5b420370a5abf72d58073de67aac5abae11cefb7e7ac606cb7104480fa11751ed0dece3f31fb119c0f591f9f12e82cc89710aea02813c1185c7aa26e391c6f2d1545a5258da2dc3b67115498f73d71397e04912c629a8e056d3f378d1ced182f104279cd9727cc97642251c3be838c852abbb6405e3c4e515f1c09d414c95252290c0beaa6f30d1d9c461697c11fe83dc6cfcf5f09baae9a32e81eb081f015ad25a6d7c5f8ee880656461c29832897196dfaa63aae1349b66e00b4ac3e1ef65152384692d1f0e3bf2495ae4540081&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332124116Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6016/default/array_metadatas, Name: checkpoints/6016/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/6016/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a31ed7bccc819bbb05293544b7007d3898d7e2835d1e2cde7b51eb447de7681d3aa54637008c6f0eb7df5c21ae20a46697f32e379c3d44e5a957800d11e167c562d967a1348e6fd99642e7c0bce8cc3a6417095b11dd183dff863b750339f1b1e307ca273f4f74365a5928a2b0879743fce341689ebcb331ff60fa6e29a5e24e53576854941ee29adf2acd458f7f9da6d43ae45a3c8d3d687246f713371ebaa14af2aa76d48ff452284db9984227fb569a7645fad50b660eba29b90856e59d7ec241ed2983203582edcdcc99b47674b694f1434bcac59cd13964ba0861c64ff1864595d14ad977b73147ea3e8fd70215c906781c86d6ead33cbe9542132572bd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332129192Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1792/default/d, Name: checkpoints/1792/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1792/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=35e05fc0cda6f6837c44c8edca80aad04130216642e35b161e961c0eedd569a3bce3ed023e2cf646f0058e8a2ce880af44e0c470957f88e1ec681c7471a17df774cc48d20a00c77270d75e1a6e6e6d78b4015ce58eb88795781e7ae43f7a585d62eeac1078d84c604374f7e3a85e2e60854f6d13b4c34b34bd092091aee8b7f642c0a627663c2969d163729d63d82dc6676b3993f5bc3d52bfd06d6268efdcdf7bf5fcd91506170c88b4d724b31ac2f0e38f4d10d92bdacd5ba2e5f74a7b162c0fd130505848a8782813c46036dd5e9f282948bc00e8d2717154dd2a10058dae6e973641f1ae8e48426806dc19f3a0c4965f9d957c3997f67c09aaf370baf14a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332127652Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0, Name: checkpoints/3328/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3328/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=118f85c16870e22a51a6041f45e0e4f3d75eb6e8950eb00a22677d6d367033035403782e5b0f717c0a853000dac206fdcb60f679333412096cd24bbc7221045457f8f4adfd469661d599f2192269e0f5041b1a92283324b642734df848282c0e87245a735abbbaf0ef931031aacb4b74b86666bfbb8c496e8936acdceb31132066302ce434bb484c7b440858c1fadaedc0c69c00c6d0c8f054e36c24b3a4b70a7647a9193378ab1652b9ebf22f5389060410fe5bf253b113712373253aaee3f1f8e0181ac8192b9d542fd2c00bd94d3ae55885c80cb1803230232db633678fcc9f6b126dc7d9dc650e76812f2a92bcb60f779a41a53e9551ffdd020dda464aa6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332133856Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d, Name: checkpoints/2688/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2688/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=089b6a1377949301dd36955874b9183638309a35cdc2c169219c44619742fe97ea283aefa1f889ff247a23d1bf56fbf8c4a0a7af46ac47473dce4b258402e3c11e5b454f52d49f870450a6f3f26a5061e726043b7f53240e9f881a85231c3dca7fecffca0eeb8d78f10ea9e28539bff3b248349a057becb954cfa50e97dae082914d9a4c1449d8fa4d2b0c5f24e31c8481ca3ea68f20f55ab2258a0ece97b5da22400ffdc20be5102821118307381f79fe9890f49d21b3a10af6023bc7ee72c189764cd0c1083f181d6d07048264361f964edfd621eac8b383d1a6ec0495faeb903f3a5905f8a007374a6c58cbbea9e6fe5f390890377898303428a49b57833d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332143769Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/array_metadatas, Name: checkpoints/5632/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5632/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=857957c8696c4952974440c21a46b305e098d73e53636bacc3c1ba259fe60bf564e02e5afa3638b665114faef645451ef619d2fd63e9f128a7f9b04c59bdf8e6dcffc6fc8743ced21f565b80789b13b9a9213a0c6e6d72ca41f33b6e1b1adc39c5f946068ccba0609de81c352c2dbdd25899aeedc4a6bae780ddb98b120c7a3590958bf9812b0306e70ae5e746a623107b476744ba78deb17dec85b6f05b0db7f5dc69e2b121c17ccb4517e305061d822bceb3f6b0fa971a2fe607ed3ce6081ffacfbeea41fc731aac2370ce252e357d8af4d08441be57345de5699f19e651698e430bdffac7d0430ae377d7663a13981bc0bb6a319fd7ce01cf687af309e1ff&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.33214385Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0, Name: checkpoints/2368/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2368/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=5149ffa6be461032691ac05b5509bb348c4542bce39be0c71aa5fb2fd6b2bcdd2f8c4ca61ea7ac75dab7ff880452c346781ea12030d1ce6cddce3d2334b870054b112938b0473d1f3264a3c7744c7600a21544f67170e6348b07493a8ea7dfef3f3d455b6499d3d4195daec2cc538298366d06f8f9361c0727c3d56922b1bfe85d2e3b708f7078fc13a9941f821318e8d5b342a13f83af88fca8da6393fbc5756926c0096de6dbc6a4e32017ce12867a975ecae001bcd1a9f25c086dcd45686e1f1ff6b9d9a101e6325dbcd8f872806ad368e8b70c6c14aec91eae1437dd600ed5ffb03252c8f33a4f2b78dc2d584d4b2d1a950956aa668888facf512d3c00b7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.33218239Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/448/default/array_metadatas, Name: checkpoints/448/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/448/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=0ac385f8be75671a870e14241e175dedaebbedbc15c8eed438e65e9c9e59c75e346a1b45b2d27e4ce2b5c401d8e5ede276c110c2ee953072c3ec850a7e25b9630780f3f421f70a9d238cbced61076628905390939b1da1f79c65cacf1f5e622c37d7b168eb6c1b9a36bdce8a203ba5ea8f15140c0c2225c9a2e80b74e3ecc4d6f0bd8584156f809c8a4ed9093c21e38df6969e786006806bd005a3282db0ee3c2935a80908a73ef501a44bc56f4e628fe46a403c7292c5fd97502178b8e725e4c2d60e6deadea2d222848f5bb0c9b4332c69c23959b72b3083edaa78b97cf01a1b6fdb397776329a862c5254f30ec39c5cc23d8310f096d3721a9ce0974693c3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332208805Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2240/default/d, Name: checkpoints/2240/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2240/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=019543412dd0f4cd6a8f8e641a37255bfbbd0f66a03233eae95e5e3c3bb430376a7be5c0d0a6dafaf853421d00db3f9721c444b883dee5ad1aa6aec5b3a6bf7ed2b2255ec9c45f6f70f16f4c60cf95e84acc730a308ec2f82ed215ba848810e336e16e0e71a3a310451060e812a9dcc6405d46a0e81d82e34aad4674daccdd58786935fabcd8fea744bb815f17e545fda7ed114706bed949d7c08db8431234c64f8f7dd69e608e0d7edf9c0e3f01ced8d99b12938894c43af37bdaccc54f3eac4d7be12843ab549d5c7f7729784705b74715a69f12fe332a2423698cdf0062bfc011d5a42cdda31cec1fb394201aa2cf44dd4ce642cbc4dc2731fa11ec3c080c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332200076Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0, Name: checkpoints/2624/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2624/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1a3a527d9c21f5bda1a3504233f3dbf4c60417d9b00a753586afe3ceda3b67cf4735352667ded406e1c268758c6e469486c22e8d8cb3b314c5e364d07897955d926f98d64d7b4d466e16fa49d0961fceba7c2ca02fa5f9c8ce06dd95e0e55dc5769e5d52c787d0d76b5bbb4d100c6b2f5c5ca175937783cbb614b689635aae981a452495be543c8cbd0b3833cbb626389e95b0d3365600bc1f60de9e65108da7c3fd42452b095719b4caf26cdb89e083d05611b4ff02caba6fbf2fb3f5a1f602a68e78c18c95bf449cd503cd928eb2ba41f6012735b85ccd6262e6e8279028f9f0e394e3831d1ec95f0b00c3e731b49451283021f531bc6f53e342d7843123a8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332250037Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4928/default/d, Name: checkpoints/4928/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4928/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=5a2b2500282c67695efb7badb6349b27fc3f412a3d4cee406b2ceb14cd47600e36880995118dfe0376b7562f34eec8b7aacbedc79a12edb10593f0b99579fe194071d4959bd492af990dd4f0edbe67e3276288990dac466d79ea07d9e464bbce8ae91d08c7c700c19ba4e91be4a1d8b219b5b12656841bb9dc2c6ac4134054a49756344d42945d625d633b0e8403e351d9939454507f495be237657663a10c0634b8e1cfa3015d07aaab3cf6b3776e1489c5af2841ddf667836fe226abca6862cfef6e6842e6f2f740354e22821b998ec5223615d64f00b5be73a4cd444c7daf82f40d028291c7298004328046fd283009b178c5da63d87192e12d56b4d696e8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332257652Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760/default, Name: checkpoints/5760/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5760/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=c0b9140fa567cc729adf91d817ec5c52bf8093ffe75d368cb4a8f877a896b98073bca572cf38866ab5b92c1f89a074a853f263b6c8e54dc0cecec2168232f4127b6ddf93443ed1c6dd743db4a387e583e845e00675a767f85f8ffc6bb0fc60230f551be0449177db5b85a116f0421b05f3e46c80b58cde8ed5783eceb34ca6b71d734d57b3917b1a8096906bb86b1a2d2b971f5bc75d8c288fb9285ed5afc39c929b67c35ff38fae7865f8476dfd1c2204d132e2cf8a91c5ac52e96b5494538751b6b44b328ef189f987431736779779ce3b5c318508230bdb0f4a8a6950ff63adc157c56c148e04a9e2d8844620fa0f1802821182f85d0aa4ab8dc3d5d7ffc8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332256825Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5760, Name: checkpoints/5760, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5760?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3833d5bc70a5668eb67da0144cbb1b3942b54e1feb61f13eb1ebf1610f1ae4a07521df4c4706e5a897d908b45f57170a3b7cf87c5403a4ca1ce52df7d2e9a1587ede609b3ed66a5dd3a85c37b03e0a4ae3b22e1d668bb74118eba8f0b8b7d57fa027bd8185776cf2cf5517358fd5aaf333f151e9379da7843c877b1eb52c3b275110041d5f29e82fbded97fb34562e4e0ebfe20d8b518607d0268d6b234e7228a95d46cb0a9cb58134e79054cefff7d599ba6256ef37ab87a9e716dffd4d2e4f92230d16faeb464646f06ed8af019db68a4ccb2525d3b94bd3e4c675b12d072120e5ccb107d170b0d73b85c391c5419bb9501b9939c938629946e640db4a2c80&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332273947Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0, Name: checkpoints/5696/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5696/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=38f631f56c2e34022b8dc8a225dd035435c00c8336eb16e1b911739865ec9ca9e9cb214844201b891aea3bb7ed693d963b52aff84a90655d6ef4a8996b11053aa80d7b0db05bb02ce7feea7d1727f263ba4f3bef7a43b033b160873deb055fb879605ea8b13d513ddddb38e7ad0a454bfd0870d2dff578810c3d0c5be54564e2398eaeaa2b7126d07f8d0a13532cc4c5a0d77c135d415d860d154643638a3c5366e8ac6b1107716b95bc06dbe9ec77ba77a225e715c0eca64610bf0888fc7e23b19cbe62773e315cc81131aab94bf0dfdcc5e947cd84adf9528d8097e567512f1a8c6a055913300bb324788654434c9cd2bfe0d18acb34b52522c74698dda74b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332289523Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0, Name: checkpoints/6080/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/6080/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1be4d79638808f801c78535ecf21c9b54b93113436440aeee25d917fa2b58866b39a7cbbf5b95544777b236de83e944a2dbd8757d5a4fb3f8c0962ea2b3ffc5fb017743207983374874f5ace8f330055e8748c86889dced16bdc14f595a0cf2696485722ce9928eff9ffaad1264e1aff546c9832025a6a7f2d231b2e144efbcfc0a309a00bbdd4c64d20fc4c6ad3ebbdc0da858377690bcb10241ede5557c6def5218f678881737fa752c55c77e8fa9fe681ce1810d90e6111a5e14ced51509cc94bd924277798ed3b3ce6115dc3a5528f1dbe744314eafb7c53b982d8b58a97a544fcea00f23e9cbb46fe9c5308a67001656d88898ecad4c21185709ea6a9e4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332284575Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0, Name: checkpoints/4224/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4224/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=7cd4e820316e9a9a81d3e65ec807435de98a4d0970827d05ea9ae8c63a3be9bcdcd4ebb56375e4ffa77ddcdd3753996fb4090ffcaf36f61165575b99ebc7f9b53c315fbd693f1f5ec251c448d24840d15fe05e46e6907726bb0ca43c214bde3471121b1bdf13ab735bcb60495b845a6bcfd7d9e54391aa61f2bb5c70987e9f003ec75efc5edb7578bc57317b4fcbbacf1a2e7e25498dd6f6ebde18de7f086e54d146cc56ddd3fd9faa94a4f5b897433239d5327152cff8a4470e3976757b13d1fd7dbe2b5a48bd9e9756de255a8e718e7b2853cfa2323a041ad08e915e8624259d97ce4725234a26ae27788faa02dc904659e09a19debc75e1353702ad349e7b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332328763Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2624/default/array_metadatas, Name: checkpoints/2624/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2624/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=26ec8889a5cc02283aefef21b1d415897f97e142235c1c2465759923f5a31ddb1c94e232104c2e6295bd6704d93a4ffd546b55d4ea5f50f1547debfc9b62290e03b3a8d44e335477437907e9595e7379d5f16b147a1ed632c082b0aa8361efd997b5250c105f0f9c78028025c83fee57f7140f1519b4444cde8d6a408fb1ad45d22cea8d319c2c034dfea89080ce792816aa3afcfe774896a596c85fe928bc035c4a27532ada14243192796f547656eb8859652781a99474118ea5f1db94a27cbc1bd1b644258a18289807e35ab0c61fc949fd489ba74dc77f44f03b41313297875632c6a911d990853d7233bc1ca5d68c3de295ed869bdd20ed7e29cf12b2f9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332362191Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/1984/default, Name: checkpoints/1984/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/1984/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=3295fe801eed19b363049e304571c15578f6549c68c8a20bc390dca57ce669bb0758c55153ac6bef15dd291146dd85350028007fbf4b4b2335a6222e2d4ead8546ac0dad87d92f0d604334c417393f5ec5ae99f1ae6463ff09e66fe5504093a033d932e238ebea263869caad169bedb69f379ab3a73aa1e5bf51a6cca5657aaa68a7b21dc38e91cd56a2f02d4962fcfafcd0f2d8a131de22313f5ca55f084ce6413ba9527f2b4db19c8b1ac811f5b478f33634f7faff2426168115e587d3caf0dd068bb47aed82eb04f78c9584a03843801e2abc9612ed6c340a010cb244bf94f39d44eefc90cae0318748908716569ce9dd38481668da4f3cbb5fcaa7f12a6e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332374023Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d, Name: checkpoints/3584/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3584/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=89980ea452a94a5e31d85c5a54be68af107d20e28678fd1774965a78a5b306da4e89cb392b4ad4266263237a10676740ca9cf9126e3ede24bec863fc9d4a30fb9ecb6da8213eef4dba225655af04422f41a1d8d9fdb236993b698d4ffc9e5fd1b15e93d27234610fd49d8563b2c75a71016c714aab2d96300f6d633648020c74c0070f889a0756481402c67c275c94c54f7f37b9c9e08b9ae654ff3a9f163977c484aae7656c155f6a0089ff6651552984df530dd421a61b8852e73c23a1ef7d17bde286449676625c369f6a6cdfa6753dbd19f1568cfc36ef24079d32c750e61c64f0098b523287c20ef5ea04589023a9c16efd6c77816b385f8030900d8c1f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332418851Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d, Name: checkpoints/4608/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/4608/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=1d92ff531bf2acd7061fac9e796803397da92a8be2e136711007fbaa51d0f0b9b106a0c38dea48b854fe2104de9ceac7004c5d4e02c91f1a63fc882997c5ab1fb363352d2c37439be3023b704c3ecafeddf00b53bc10a723756aa6e84caaabf5249429c083288c7aaad9b52924e576a1cd61eafc0211b51e129023565b577ebefc025a1b97d5b7841dd702e7f28838e79d5359180140903f82101b6c86f754b7975b064fbbee166b35dd18f5e4d22c32fc3f93a1292e902c6829230761b3874c2d22ec4daaec99a6e768aacfcf8a867cef851667f8e50700c75dc8a44ab1ba7c50065a68f8158993a9aa25fda0bcae70c6201e50b24d4c76bfc130699e73a3a3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332424242Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0, Name: checkpoints/5632/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5632/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=6d4c97f555e34619fcb845443b577a5ebe5ca3d2039d0d96c3e4bf973b13a66f7bdb6e189e1e36f11c1911fc27dd530708cc744fe3148c3e96949be589e47b35fb5d2dda9fe6835b5676e3e4023e043dc1e9457366c97f45bfa8467f8e1b2675dd48337949f411650f365aa1b289fa4805eee2a79b6a43874a97d065b983fe77929b903b59f6601d4286bdc42113a3afe08b8cdce8c7445c60827111928ac463465a2cbd048c2daee599518c7c03cee5fae2376856382a5d987cf32aca84ca95afa6e5aa3dc90364ed505ae53d6deafd1368f0f5ccfcd4835497e247e32b63e8d4cd789d198d7a5c8a9d5c049b0fd0fd968489827837568e7b033265b23dad1e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.33241456Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d, Name: checkpoints/2432/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/2432/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=266609765e3f5a9e8a9a7fb65636e161d9e7793b214999beba049b88dd93137ad09f0f7ca33b69b3d3b5892ece637dea1600e260a181dcdbb10d740d465e1b73606d86d13e4509308da4ff95c6aa7f29732fb8327d4295f58e710652ffa1fafababa82e268d6173e6448dd614a2e35d02d5173826199b547e777d4ef43bcdb26f8234ef292319f4e785990b3d04f153b99371fb28bf1204a3cebe437895a8a006ccbc265d7396968b8f07295ccf46890110cf4b59e9cd48549fa1f67c80fd0692ae404dda190c4679d5652f07c4e03c5d6d2dea1150b6888c56572d5a259d9ca385b3cef9ba5625cc2ce5bd3d4baee29bae1db513ccabf5c945d838bea7c43e9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.33241771Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0, Name: checkpoints/5888/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5888/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=6a6c16194ef4d5a82dd98933881751f405020590aa7e9efae764652c4951bc31d5948d3d1ccc62212d1c501829c066926e47696836730bb8d458a88f13d9d87375afecd019443995b41c9922ab69361dc8596d2ce6250d4e0912a40ee279b9321b8494d2fed6c1ed338d58de9a213a12942f758989a2ba522b7d9d96728d7b3cf269178f61a846866df5848e30d7b32bc39072638421b9ae9b7b7c159b05ead5e6b2308dff98bd832abad37b96626de2a0df35a3023134832895e8638f07e357eb7290d897d46e71932f33fb5a139f502d84942b3f4f3575dbd3fe3a26b3a8ce8e0cc933a0295c8e7e5e4bd32df99ea47b716f588cef6337d59991b54d8361c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332421181Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3008, Name: checkpoints/3008, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3008?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=477eb7d6d7141893ae5f2e99398cb35296acecb89cb2331dbc36781b747624ae89962dfce25b8337912168bb19494b6810935d89a12672862602c2d28503dbe536290498f11d5f7bcf0ba1d91400b6d8e1ef1ab63afd7a4573c041d0be590c1f5c77944b3a8a6b938b1d6fa9f68da2fbd9b20e64ac5ffd7b6abba90b544638fa1e927f8d4b2c9013ffe09bb4aeafc5dc651099d030eeb9727240d873fea668975b8aa5f84623953cba4894b7fd5e00f4815500eaf05b9c94fa4d29162c3daffdebb6d71f03966f6d6fe2929f9567dc5d22f9f8894039abd3f52ead562c7a13ead4a8e8a01e5675227fb88deb0d0e646691b73c6a3c0cf2999feba3a46822b510&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332468085Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/5824, Name: checkpoints/5824, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/5824?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=26a793384c4207b14aa0ea2b50d59d14bc718a66f4d2282b080c513bbfdf997d6f1292a9b36326ac2899caee74306828aa564215492bee4715a591dba46a196248d9fe9e2e9db8a6fc323af683f02aca42b3d9525ad266869402881ce274a0b77b24289d2bc30c81524c68a446844ace74c035b7d478861349f47f41527bfd9fd5ea6dc73eab63672e0c3c9340d3e32ee72b525608539739eab4227072fab0b4b8b303bab6f8eb93a4c8397ce2ff4b20fdcda462eb1ef567247a26cec3c6e71d33b4c9d466f22adf704b2897a909f61deb4a2b31b893cfe107eea4ce90a7d5105d785f01f244739800a8a442594364ec1aa9ae4fbb4ea2b29eeee3e11ec9770a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332490606Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/6080/default/array_metadatas, Name: checkpoints/6080/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/6080/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=a898afa3b34cbf7b64967fdc3e23202b2bb5ceecbe59603786460f596d356a4b2619894c60b4cd4860ba7ca8fd613cb4350934e68f3e183d3d728124455fb4c80f5ce933d822acaffee2f89b8787408f7f9c562232dee8df08de691583eb22bcf1b9b0e3b8d80f9f5b3756c1ca3b6a85449f715a81a56c9feceb952404c58a0a7f07a3eb5621ca1b0f8c75d1d773913abd229548c4192f6693c440addbbc16e3660867eb6b5b30ea3c209b54779b316e37f6638de4a8df4b63c9d01caac0043092bf545e3e6687f632fca8dc62d456b1a0cfc94add0ad4ff328364a4f9896594a7a4b89b3d0bd80372d849ba76f417ea60c3f63116d2d8675f369badb8d837f6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.332522631Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d, Name: checkpoints/3136/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1/checkpoints/3136/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T160347Z&X-Goog-Expires=86399&X-Goog-Signature=2f8830c7e56156d080eb80861f13388c48b4d52858fac0f3a69445f439b1a5e2f6561561f1315a99bb84e16208c36907953d3c6385ee27d005bc25e1930fa46f1a933eb3ebf76ed44622f149da0dfa8cf2475d86a2231cb481fbcc4f7c757d4cdff97cf289ca1c560cced512602562fddcd50715b07b251ff72e2d4ac4e377b0b00fcdf6630aece43e24bddc196830e081638f9bb046281c813bd45e58a3d2dd8a33d00566a7affd24fbe62d32ab6f4f441dfc60472188ac1a88258a0576e7511a32eac7475850ee23e73a9312600fb3c0c76787f1d41621805741c33a0b2ae191022154c9dc7fcead778eade6b662d7314e4c2862432ca46795f27f2c46234f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T16:03:47.460575456Z","level":"INFO","msg":"handler: operation stats","stats":{"operations":[{"desc":"uploading output.log","runtime_seconds":0.584826354,"progress":"1.2MB/1.2MB"},{"desc":"uploading history steps 189-190, summary, console lines 24928-25060","runtime_seconds":0.095036407}],"total_operations":2}}
+{"time":"2025-08-07T16:03:47.765423718Z","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
+{"time":"2025-08-07T16:03:48.038333456Z","level":"INFO","msg":"stream: closing","id":"wp_0.0_alpha_1"}
+{"time":"2025-08-07T16:03:48.038371623Z","level":"INFO","msg":"handler: closed","stream_id":"wp_0.0_alpha_1"}
+{"time":"2025-08-07T16:03:48.038400199Z","level":"INFO","msg":"writer: Close: closed","stream_id":"wp_0.0_alpha_1"}
+{"time":"2025-08-07T16:03:48.038409997Z","level":"INFO","msg":"sender: closed","stream_id":"wp_0.0_alpha_1"}
+{"time":"2025-08-07T16:03:48.038571165Z","level":"INFO","msg":"stream: closed","id":"wp_0.0_alpha_1"}
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/logs/debug.log b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/logs/debug.log
new file mode 100644
index 0000000000000000000000000000000000000000..726f17e7e0dc5ef5ebcd66c2b3f712042a7ee466
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/logs/debug.log
@@ -0,0 +1,28 @@
+2025-08-07 15:04:00,939 INFO    MainThread:3804420 [wandb_setup.py:_flush():80] Current SDK version is 0.21.0
+2025-08-07 15:04:00,939 INFO    MainThread:3804420 [wandb_setup.py:_flush():80] Configure stats pid to 3804420
+2025-08-07 15:04:00,939 INFO    MainThread:3804420 [wandb_setup.py:_flush():80] Loading settings from /root/.config/wandb/settings
+2025-08-07 15:04:00,939 INFO    MainThread:3804420 [wandb_setup.py:_flush():80] Loading settings from /workspace/timaeus/projects/rl/wandb/settings
+2025-08-07 15:04:00,939 INFO    MainThread:3804420 [wandb_setup.py:_flush():80] Loading settings from environment variables
+2025-08-07 15:04:00,939 INFO    MainThread:3804420 [wandb_init.py:setup_run_log_directory():703] Logging user logs to /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/logs/debug.log
+2025-08-07 15:04:00,940 INFO    MainThread:3804420 [wandb_init.py:setup_run_log_directory():704] Logging internal logs to /workspace/timaeus/projects/rl/wandb/run-20250807_150400-wp_0.0_alpha_1/logs/debug-internal.log
+2025-08-07 15:04:00,940 INFO    MainThread:3804420 [wandb_init.py:init():830] calling init triggers
+2025-08-07 15:04:00,940 INFO    MainThread:3804420 [wandb_init.py:init():835] wandb.init called with sweep_config: {}
+config: {'env_size': 13, 'env_layout': 'tree', 'wall_prob': 0.0, 'env_corner_size': 1, 'env_terminate_after_corner': False, 'obs_level_of_detail': 0, 'img_level_of_detail': 1, 'env_penalize_time': False, 'net_cnn_type': 'large', 'net_rnn_type': 'ff', 'net_width': 256, 'ued': 'plr', 'prob_shift': 1.0, 'num_train_levels': 2048, 'plr_buffer_size': 4096, 'plr_temperature': 0.1, 'plr_staleness_coeff': 0.1, 'plr_prob_replay': 0.5, 'plr_regret_estimator': 'maxmc-actor', 'plr_robust': False, 'num_mutate_steps': 12, 'prob_mutate_shift': 0.0, 'chain_mutate': True, 'mutate_cheese': True, 'train_proxy_critic': False, 'plr_proxy_shaping': False, 'proxy_name': 'proxy_corner', 'plr_proxy_shaping_coeff': 0.5, 'clipping': False, 'ppo_lr': 5e-05, 'ppo_gamma': 0.999, 'ppo_clip_eps': 0.1, 'ppo_gae_lambda': 0.95, 'ppo_entropy_coeff': 0.001, 'ppo_critic_coeff': 0.5, 'ppo_proxy_critic_coeff': 0.5, 'ppo_max_grad_norm': 0.5, 'ppo_lr_annealing': False, 'num_minibatches_per_epoch': 4, 'num_epochs_per_cycle': 5, 'num_total_env_steps': 200000000, 'num_env_steps_per_cycle': 128, 'num_parallel_envs': 256, 'console_log': True, 'log_gifs': True, 'log_imgs': True, 'log_hists': False, 'num_cycles_per_log': 32, 'num_cycles_per_eval': 32, 'num_cycles_per_gifs': 1024, 'num_cycles_per_big_eval': 1024, 'use_fixed_eval_levels': False, 'evals_num_env_steps': 512, 'evals_num_levels': 256, 'gif_grid_width': 16, 'level_splayer': 'mouse', 'checkpointing': True, 'keep_all_checkpoints': True, 'max_num_checkpoints': 1, 'num_cycles_per_checkpoint': 64, 'seed': 42, '_wandb': {}}
+2025-08-07 15:04:00,940 INFO    MainThread:3804420 [wandb_init.py:init():871] starting backend
+2025-08-07 15:04:01,152 INFO    MainThread:3804420 [wandb_init.py:init():874] sending inform_init request
+2025-08-07 15:04:01,158 INFO    MainThread:3804420 [wandb_init.py:init():882] backend started and connected
+2025-08-07 15:04:01,161 INFO    MainThread:3804420 [wandb_init.py:init():953] updated telemetry
+2025-08-07 15:04:01,173 INFO    MainThread:3804420 [wandb_init.py:init():977] communicating run to backend with 90.0 second timeout
+2025-08-07 15:04:01,833 INFO    MainThread:3804420 [wandb_init.py:init():1029] starting run threads in backend
+2025-08-07 15:04:02,462 INFO    MainThread:3804420 [wandb_run.py:_console_start():2458] atexit reg
+2025-08-07 15:04:02,462 INFO    MainThread:3804420 [wandb_run.py:_redirect():2306] redirect: wrap_raw
+2025-08-07 15:04:02,462 INFO    MainThread:3804420 [wandb_run.py:_redirect():2375] Wrapping output streams.
+2025-08-07 15:04:02,463 INFO    MainThread:3804420 [wandb_run.py:_redirect():2398] Redirects installed.
+2025-08-07 15:04:02,467 INFO    MainThread:3804420 [wandb_init.py:init():1075] run started, returning control to user process
+2025-08-07 16:03:46,455 INFO    MainThread:3804420 [wandb_run.py:_finish():2224] finishing run devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_1
+2025-08-07 16:03:46,456 INFO    MainThread:3804420 [wandb_run.py:_atexit_cleanup():2423] got exitcode: 0
+2025-08-07 16:03:46,457 INFO    MainThread:3804420 [wandb_run.py:_restore():2405] restore
+2025-08-07 16:03:46,457 INFO    MainThread:3804420 [wandb_run.py:_restore():2411] restore done
+2025-08-07 16:03:47,993 INFO    MainThread:3804420 [wandb_run.py:_footer_history_summary_info():3903] rendering history
+2025-08-07 16:03:48,030 INFO    MainThread:3804420 [wandb_run.py:_footer_history_summary_info():3935] rendering summary
+2025-08-07 16:03:48,037 INFO    MainThread:3804420 [wandb_run.py:_footer_sync_info():3864] logging synced files
diff --git a/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/run-wp_0.0_alpha_1.wandb b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/run-wp_0.0_alpha_1.wandb
new file mode 100644
index 0000000000000000000000000000000000000000..f4e922b3b1c6a7ae48dbc924bd5d73114ff71680
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_150400-wp_0.0_alpha_1/run-wp_0.0_alpha_1.wandb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:07eee9c7eaf413c93911fa4bdf81b508e6aa87f3c0db5f913bb9bc8322cde228
+size 8356821
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..fa5c6529727efff912f39d3476b5215cf475c792
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754582808443565907, "commit_timestamp_nsecs": 1754582808639835624, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/d/be82f696e2516a77f4c9729461a77f3d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/d/be82f696e2516a77f4c9729461a77f3d
new file mode 100644
index 0000000000000000000000000000000000000000..ae63299e4c1cd59a7ca382f36dee04dd80b27a38
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/d/be82f696e2516a77f4c9729461a77f3d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..7dbc90eb280acf16d8b2b9748de17e3fca8ddc6e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/06e2871e1093f6abf6cd44d50e79a0fb b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/06e2871e1093f6abf6cd44d50e79a0fb
new file mode 100644
index 0000000000000000000000000000000000000000..37e2b6de65796649461cb96011283b8f7f9cb6bf
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/06e2871e1093f6abf6cd44d50e79a0fb differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/28a6025f88e9279c5309c909d88dbbc7 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/28a6025f88e9279c5309c909d88dbbc7
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/28a6025f88e9279c5309c909d88dbbc7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/28b1edc6f6ecb9ab524ee4e7f5d0351f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/28b1edc6f6ecb9ab524ee4e7f5d0351f
new file mode 100644
index 0000000000000000000000000000000000000000..4859214e6b68fa6e697de9748417fcf43be8081d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/28b1edc6f6ecb9ab524ee4e7f5d0351f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/335d7d00f234be6a0d3f214f71ab50c4 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/335d7d00f234be6a0d3f214f71ab50c4
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/335d7d00f234be6a0d3f214f71ab50c4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/4ab02c394b802aced363252db83c1dee b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/4ab02c394b802aced363252db83c1dee
new file mode 100644
index 0000000000000000000000000000000000000000..f3b34de2cf558602b48bab0b26220d0f11b870f8
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/4ab02c394b802aced363252db83c1dee
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:34965d16b82d11a432e616fbfa2c3c4f679bec461115a272a35d2a057704a386
+size 159177
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/5388bdff0ea392b04db2691e07a8fdd3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/5388bdff0ea392b04db2691e07a8fdd3
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/5388bdff0ea392b04db2691e07a8fdd3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/53fd2e3500942a61440c3e31fa6b56e4 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/53fd2e3500942a61440c3e31fa6b56e4
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/53fd2e3500942a61440c3e31fa6b56e4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/6ddd386a5f2055e307b0812182d36c5e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/6ddd386a5f2055e307b0812182d36c5e
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/6ddd386a5f2055e307b0812182d36c5e differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/a6c6be437ba217836fe168fa1012e2ba b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/a6c6be437ba217836fe168fa1012e2ba
new file mode 100644
index 0000000000000000000000000000000000000000..5a747c58bae0a32b2cc736c0d4bd35f8a4e19bac
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/a6c6be437ba217836fe168fa1012e2ba
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:22da68c026365f6c77a570d9595462fa8b8a83b0417281afc4d7b67a8ee5cb96
+size 538557
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c8f8b3e8480b42103254bafd946f4c1da7153f2a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..4963ff0e3ce857ae6079359a6bede779872b58b5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583371564781522, "commit_timestamp_nsecs": 1754583371737858959, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/d/f26b792bbd772ccca910e8ab900737af b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/d/f26b792bbd772ccca910e8ab900737af
new file mode 100644
index 0000000000000000000000000000000000000000..fd9ce059be134a29a2ea4e51e1a0730a99265ed9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/d/f26b792bbd772ccca910e8ab900737af differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..da1ebb0e38033354fe87e14be02d88938427f007
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/25ed29ab7e25621d6bf66c03f4a1edd5 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/25ed29ab7e25621d6bf66c03f4a1edd5
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/25ed29ab7e25621d6bf66c03f4a1edd5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/323d4900eee8dd04e45700a8d9fe1896 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/323d4900eee8dd04e45700a8d9fe1896
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/323d4900eee8dd04e45700a8d9fe1896 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/3809dee0d34360060a153f012326bc4c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/3809dee0d34360060a153f012326bc4c
new file mode 100644
index 0000000000000000000000000000000000000000..dfbfe1a61305bab6f716f83c77874d9e3176999d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/3809dee0d34360060a153f012326bc4c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/3a6c40edd34b38be79cc48a04ddd055a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/3a6c40edd34b38be79cc48a04ddd055a
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/3a6c40edd34b38be79cc48a04ddd055a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/639d2ac7a2e499151ee1128576e37f53 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/639d2ac7a2e499151ee1128576e37f53
new file mode 100644
index 0000000000000000000000000000000000000000..7175e5be20cee91782ac8ee7b75e7c6e3bf49339
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/639d2ac7a2e499151ee1128576e37f53 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/b1041ad8104208025272b9e5d0c95389 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/b1041ad8104208025272b9e5d0c95389
new file mode 100644
index 0000000000000000000000000000000000000000..4014789839407d3393f55b53cd3c90802c346a7d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/b1041ad8104208025272b9e5d0c95389
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:06dd540b3bb6134436f6635bd65d38864533f83d369ee233196c2bd0e724dcd1
+size 455759
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/e14d3bfc025b97a8b4026387458a292d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/e14d3bfc025b97a8b4026387458a292d
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/e14d3bfc025b97a8b4026387458a292d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/f1dd3874cdec8cae74c9ffbe8f7524e1 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/f1dd3874cdec8cae74c9ffbe8f7524e1
new file mode 100644
index 0000000000000000000000000000000000000000..486142fb81d650c78619c9a47a804696084d86fc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/f1dd3874cdec8cae74c9ffbe8f7524e1
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:78110ac970db921369f4efffe46f32d6ef47e87a38361ebb400a508a0574df86
+size 244529
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b0a7f4aa2b2f9176a11da6f6a20f97436add5db4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..3fd85179fbe26331110e45f38f9347d8d315391d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583405016904187, "commit_timestamp_nsecs": 1754583405162909859, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/d/23e333ee391df43fda9f7c91faa26761 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/d/23e333ee391df43fda9f7c91faa26761
new file mode 100644
index 0000000000000000000000000000000000000000..b32d0f4610b1070c7db14ec03cf8304185bd99d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/d/23e333ee391df43fda9f7c91faa26761 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8d23fa1e91940608cac72c7a41b541ad03d104ea
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/08b1c66999e4746acdd31d332fb5fa6c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/08b1c66999e4746acdd31d332fb5fa6c
new file mode 100644
index 0000000000000000000000000000000000000000..af0e7c6758f2f35fc06a3db3bbe5d1b7b38c4621
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/08b1c66999e4746acdd31d332fb5fa6c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:17a0fa633bbc5da82a2474d7b716ef7f47ca235ef1aafc7f4accbd25453c8c35
+size 338129
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/0f1471a8164c68710833c69cf69f6a1b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/0f1471a8164c68710833c69cf69f6a1b
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/0f1471a8164c68710833c69cf69f6a1b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/22daa64051ac5a6b61eca6af62d415a4 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/22daa64051ac5a6b61eca6af62d415a4
new file mode 100644
index 0000000000000000000000000000000000000000..28504d77cf27d39f2a65ae8ec73d51de850971e8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/22daa64051ac5a6b61eca6af62d415a4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/42ff215c7e1b37e133ac6dbf0a285c37 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/42ff215c7e1b37e133ac6dbf0a285c37
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/42ff215c7e1b37e133ac6dbf0a285c37 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/43db39653722f748fd24578a5858b4da b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/43db39653722f748fd24578a5858b4da
new file mode 100644
index 0000000000000000000000000000000000000000..50b87b938e7d659690339687a6fde1217e9c51d6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/43db39653722f748fd24578a5858b4da
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e84c2ee5dc7816a126574a66b70cb68d9a6409459fa93f25e4eb3e202a155a6a
+size 365573
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/86bd7773eca8d6c25b4f6c78e9889a3c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/86bd7773eca8d6c25b4f6c78e9889a3c
new file mode 100644
index 0000000000000000000000000000000000000000..f28d59e1f9187cfd65b688b4bceb57a81f0acdcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/86bd7773eca8d6c25b4f6c78e9889a3c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/a05864616b9f56fa969e9f4abdf1e7f0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/a05864616b9f56fa969e9f4abdf1e7f0
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/a05864616b9f56fa969e9f4abdf1e7f0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/d9c56cde5fc41db9669ba8782cb8ff8b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/d9c56cde5fc41db9669ba8782cb8ff8b
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/d9c56cde5fc41db9669ba8782cb8ff8b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..487ab0cf12bd52bd977b6e487de3b1742540c20f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..3e40a80cc461a5d6cebe536eabc80302a4f08709
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583438188779540, "commit_timestamp_nsecs": 1754583438362243502, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/d/44340a14fac645fa88e763c9b1cc4339 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/d/44340a14fac645fa88e763c9b1cc4339
new file mode 100644
index 0000000000000000000000000000000000000000..69f9bd1f17f68fc6ca6a55a4484920d4577396c1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/d/44340a14fac645fa88e763c9b1cc4339 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..19fbbd7476b3592f22954119a25c16b95b45aa55
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/0e6b36581815ccf58a4b12574166f9cd b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/0e6b36581815ccf58a4b12574166f9cd
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/0e6b36581815ccf58a4b12574166f9cd differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/898ebadcfddb282f9d6b549bec088d7e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/898ebadcfddb282f9d6b549bec088d7e
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/898ebadcfddb282f9d6b549bec088d7e differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/92409fdc57cd4b54702da8cc6321b9a4 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/92409fdc57cd4b54702da8cc6321b9a4
new file mode 100644
index 0000000000000000000000000000000000000000..d30c3c4a0cb2dd5cc590387c57f1605f5c9f3451
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/92409fdc57cd4b54702da8cc6321b9a4
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8c0d98021be40dea60df6d90dcba87ac5f1b7d591fb710ddaf5799c08e1b750f
+size 442152
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/a0dee141115c409288083e50297097f1 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/a0dee141115c409288083e50297097f1
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/a0dee141115c409288083e50297097f1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/c9f242ed2d2810c4ed3917cbdff88bef b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/c9f242ed2d2810c4ed3917cbdff88bef
new file mode 100644
index 0000000000000000000000000000000000000000..56d5f559990b0596cdd08b9ddc913f08d655488d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/c9f242ed2d2810c4ed3917cbdff88bef
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3b63296829ae1c7278fc3b5f52eecddb3eaad1f6e997920732d9cab0e5d51d91
+size 295553
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/d51525d8aa301ea1b30666c5e1ddbf41 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/d51525d8aa301ea1b30666c5e1ddbf41
new file mode 100644
index 0000000000000000000000000000000000000000..271e2ea2b24020ef9abb42b049072674ab647d01
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/d51525d8aa301ea1b30666c5e1ddbf41 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/e4668e87e296aaec4e616fc57f3d7f3f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/e4668e87e296aaec4e616fc57f3d7f3f
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/e4668e87e296aaec4e616fc57f3d7f3f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..cf221cdb605bb8e5835c85829637ed1f91f0f178
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..224e20a0f11ef5d5198ffed762aefdb278b5a64c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583471284048880, "commit_timestamp_nsecs": 1754583471456160694, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/d/d9fad09e8e3ea98177cba2b519e61f8f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/d/d9fad09e8e3ea98177cba2b519e61f8f
new file mode 100644
index 0000000000000000000000000000000000000000..ace4b905769a074383681ad8a368651d3fd58cc9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/d/d9fad09e8e3ea98177cba2b519e61f8f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8f7528ac3b44ea5dbd7f44c13d1bfab55d3161fa
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/28c61ada2b511ed98b59f8af2e29cee2 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/28c61ada2b511ed98b59f8af2e29cee2
new file mode 100644
index 0000000000000000000000000000000000000000..b6eb0d59c4019b5117f215c392b103b63aa716c9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/28c61ada2b511ed98b59f8af2e29cee2
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7c2ee34a6f828705e6999b4398139b68a63a7925589a134a4bb642416e47c728
+size 467716
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/356fd58ad58f2265c1e7243c7a4a3be7 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/356fd58ad58f2265c1e7243c7a4a3be7
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/356fd58ad58f2265c1e7243c7a4a3be7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/609efdf1326b3ca427b661c3e2b807e0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/609efdf1326b3ca427b661c3e2b807e0
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/609efdf1326b3ca427b661c3e2b807e0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/6963f1550847deaeae29c600557818a8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/6963f1550847deaeae29c600557818a8
new file mode 100644
index 0000000000000000000000000000000000000000..d3198c155b5a455e05a64e23722ee5f775389570
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/6963f1550847deaeae29c600557818a8
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fee2ddf82368451dbc6dbec3973f991a4031208471f97bafd1a9f3832a871bd2
+size 270147
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/96cc137f34602dbb9215076087285515 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/96cc137f34602dbb9215076087285515
new file mode 100644
index 0000000000000000000000000000000000000000..97ffd20eb29ed2b2c92b18645c9744b28879b421
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/96cc137f34602dbb9215076087285515 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/a2e5adc0e37433902dbdf9771b13d406 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/a2e5adc0e37433902dbdf9771b13d406
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/a2e5adc0e37433902dbdf9771b13d406 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/c70fd7929f22a4f53f964221ba79ff9f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/c70fd7929f22a4f53f964221ba79ff9f
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/c70fd7929f22a4f53f964221ba79ff9f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/cdf762c7dc13e0a519a43a1976f3858e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/cdf762c7dc13e0a519a43a1976f3858e
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/cdf762c7dc13e0a519a43a1976f3858e differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4fbbf72b1c1a152535103aa608d78e0b534f4348
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..e8dc84791ca0451186fede55a1fc1a3ddf296b55
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754582897531809478, "commit_timestamp_nsecs": 1754582897681792375, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/d/5e03ab4986fc5ace20222b5bc0cad0c0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/d/5e03ab4986fc5ace20222b5bc0cad0c0
new file mode 100644
index 0000000000000000000000000000000000000000..9d7a477234c95d7879c394e18bc59b2336dfabb6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/d/5e03ab4986fc5ace20222b5bc0cad0c0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1319f5aeccfaf21345cbd1f88725dfcf6ed8e17c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/0ff559359842a8d04f6b789f6d25e07c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/0ff559359842a8d04f6b789f6d25e07c
new file mode 100644
index 0000000000000000000000000000000000000000..51db1e27f2191688611cda8c398e2e1b47a524e7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/0ff559359842a8d04f6b789f6d25e07c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:91ce8d3fdf232c03cab8591d82533e95cbb7e28716bbc714cf60d63b39e252dc
+size 404466
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/3fc593b33fea1f2b71add13d7ce96eb5 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/3fc593b33fea1f2b71add13d7ce96eb5
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/3fc593b33fea1f2b71add13d7ce96eb5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/4f9a512b8dba3c0fa134c90b3dd4c0f8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/4f9a512b8dba3c0fa134c90b3dd4c0f8
new file mode 100644
index 0000000000000000000000000000000000000000..f28d59e1f9187cfd65b688b4bceb57a81f0acdcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/4f9a512b8dba3c0fa134c90b3dd4c0f8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/6a547ee5b403358f0473b049b5ca3427 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/6a547ee5b403358f0473b049b5ca3427
new file mode 100644
index 0000000000000000000000000000000000000000..030597f34c81ab3c30539942682a6133615a2f03
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/6a547ee5b403358f0473b049b5ca3427
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e27ae1eadc421cd573924ce2bb0fbfa76e3a87272338a6bf28dc43dda7a23d79
+size 295297
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/aa0577ed7c5a703511fa07ed702a197f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/aa0577ed7c5a703511fa07ed702a197f
new file mode 100644
index 0000000000000000000000000000000000000000..2a31377ca030aced51ef70fb458c6b5c47404cd3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/aa0577ed7c5a703511fa07ed702a197f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/ba33517e667177276265a24b09ecc6cf b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/ba33517e667177276265a24b09ecc6cf
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/ba33517e667177276265a24b09ecc6cf differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/e2a84093ebd9d172e53aba0127a4705f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/e2a84093ebd9d172e53aba0127a4705f
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/e2a84093ebd9d172e53aba0127a4705f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/f94572696458e1e6ae0c5c0eaa423d61 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/f94572696458e1e6ae0c5c0eaa423d61
new file mode 100644
index 0000000000000000000000000000000000000000..f4bf25222338beb557ac5283d2f2f25a15297664
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/f94572696458e1e6ae0c5c0eaa423d61 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..cad7fb4455c03039253d5d41578128757c2c62d4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..7d6649ca3987238819d8b638249aa173c9d265eb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583504558883764, "commit_timestamp_nsecs": 1754583504728554153, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/d/628429ce38737a30d5c68a40e1e766f1 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/d/628429ce38737a30d5c68a40e1e766f1
new file mode 100644
index 0000000000000000000000000000000000000000..d141026f6aefdf611449765c72d43a1570738612
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/d/628429ce38737a30d5c68a40e1e766f1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f00b43f222348f694e24d6dd41b7abf5e4b5e20d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/01404e097804f492fb2b61e1ec3547ae b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/01404e097804f492fb2b61e1ec3547ae
new file mode 100644
index 0000000000000000000000000000000000000000..3840447d4255f6431a0da4589fc2aad291218cbb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/01404e097804f492fb2b61e1ec3547ae differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/27ba5a225d0ec2513782819d0c81bbcf b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/27ba5a225d0ec2513782819d0c81bbcf
new file mode 100644
index 0000000000000000000000000000000000000000..b53710b74d7a357af85042dbfc254d90dff7e9f1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/27ba5a225d0ec2513782819d0c81bbcf
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:29a845641caa5bdb9c11de853efd16c1523225cf76a31f1c78684dac2b8b35f3
+size 414809
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/314f8abd2252b6fa78e6796b900db798 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/314f8abd2252b6fa78e6796b900db798
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/314f8abd2252b6fa78e6796b900db798 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/521ac19beb59256acd2bb3c326a28be6 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/521ac19beb59256acd2bb3c326a28be6
new file mode 100644
index 0000000000000000000000000000000000000000..e39805943bee5a8c5d53edbe573d2b19b354bf2b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/521ac19beb59256acd2bb3c326a28be6
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:98857ffc8407a0af950f0938d2effd79e5287953b62258ff2138057b239ed666
+size 284787
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/77d757bcc6f7dbe0cc1c403d33ebb540 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/77d757bcc6f7dbe0cc1c403d33ebb540
new file mode 100644
index 0000000000000000000000000000000000000000..2910fbc6db2094404ed932c7d9bcdb23beb99906
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/77d757bcc6f7dbe0cc1c403d33ebb540 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/d0891787f32d57095330b5d8c10856c7 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/d0891787f32d57095330b5d8c10856c7
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/d0891787f32d57095330b5d8c10856c7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/dace0b07ab3ac298b8ef0f528acc36e6 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/dace0b07ab3ac298b8ef0f528acc36e6
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/dace0b07ab3ac298b8ef0f528acc36e6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f69d6b2dd27822dff5430642b4b8de11b4adf253
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..d81889c46f099d92e49da1dccb8b34a7f23d5374
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583537974827738, "commit_timestamp_nsecs": 1754583538148766469, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/d/ff7c6cfa1a703a981ceed250a763e63b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/d/ff7c6cfa1a703a981ceed250a763e63b
new file mode 100644
index 0000000000000000000000000000000000000000..8a10bf4ed1813b9983c0805e69c4c1a8bf6e249e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/d/ff7c6cfa1a703a981ceed250a763e63b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..7a5fc095fe16854d757f6fe4432b1037af4eed63
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/0a56fd986aab451b088fe70c9dd2f243 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/0a56fd986aab451b088fe70c9dd2f243
new file mode 100644
index 0000000000000000000000000000000000000000..e03e359c5edf3d2e243e13eaf367119516558cb0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/0a56fd986aab451b088fe70c9dd2f243 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/119611fec5b278c7e4521b5c0934a48c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/119611fec5b278c7e4521b5c0934a48c
new file mode 100644
index 0000000000000000000000000000000000000000..65529f6532287fad4b450ca493011b55e8bf7f92
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/119611fec5b278c7e4521b5c0934a48c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/2adfff627fca0fa4f7d304cd8ea31e7f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/2adfff627fca0fa4f7d304cd8ea31e7f
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/2adfff627fca0fa4f7d304cd8ea31e7f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/4568a06361209e2574916aa056652c99 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/4568a06361209e2574916aa056652c99
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/4568a06361209e2574916aa056652c99 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/58af6d47676434860b05bb0ad93a2473 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/58af6d47676434860b05bb0ad93a2473
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/58af6d47676434860b05bb0ad93a2473 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/77c709c07d9a9fddaf89938fa8399f19 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/77c709c07d9a9fddaf89938fa8399f19
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/77c709c07d9a9fddaf89938fa8399f19 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/8dcff388050a83cc67f8a69f83d938bd b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/8dcff388050a83cc67f8a69f83d938bd
new file mode 100644
index 0000000000000000000000000000000000000000..b6ce720db092b6a0e3bac75f816eef13371231a8
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/8dcff388050a83cc67f8a69f83d938bd
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:20e64346197fb64504a936e3abfdbb2b7785912e12469f7f77722574510bea1a
+size 295610
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/a69df5845a474dc9e0b92293e8fd7613 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/a69df5845a474dc9e0b92293e8fd7613
new file mode 100644
index 0000000000000000000000000000000000000000..a1b7e7d0af2226fa4f9bffe72f7f45b5fce3195e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/a69df5845a474dc9e0b92293e8fd7613
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ac408629265a772ed8d776d44e6767570893648a7980e8416b05d594842d0423
+size 442282
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..26a9f39395482a334814022d530a93f50fb061ce
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..cbe9fb327b6c14af03fef4b457237221d4484c56
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583571235733766, "commit_timestamp_nsecs": 1754583571375241582, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/d/2054c76aa1817f45624e1d3a4abc2f9b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/d/2054c76aa1817f45624e1d3a4abc2f9b
new file mode 100644
index 0000000000000000000000000000000000000000..ba84087ce004f95ed2f40e00c9b762a5aed255a8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/d/2054c76aa1817f45624e1d3a4abc2f9b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c1c78e7b3d4826b5921a8dbcd58af59e14bf36b9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/0c2a45d3cdbeaf64bf3100983b1b1880 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/0c2a45d3cdbeaf64bf3100983b1b1880
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/0c2a45d3cdbeaf64bf3100983b1b1880 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/3b28798ef9f199f9abdafd844cf1cb45 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/3b28798ef9f199f9abdafd844cf1cb45
new file mode 100644
index 0000000000000000000000000000000000000000..0aa3cb90dea03d256bdb9b0022f26642059b2145
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/3b28798ef9f199f9abdafd844cf1cb45 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/40c779ddd9913f5798b0db0776e8daa3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/40c779ddd9913f5798b0db0776e8daa3
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/40c779ddd9913f5798b0db0776e8daa3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/4ab4956d88f689c196eef617a21f9c72 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/4ab4956d88f689c196eef617a21f9c72
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/4ab4956d88f689c196eef617a21f9c72 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/68968e27a1a89b8bee2a815604559c55 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/68968e27a1a89b8bee2a815604559c55
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/68968e27a1a89b8bee2a815604559c55 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/9e0ab5da9f821773ce5c431a1ef5f70e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/9e0ab5da9f821773ce5c431a1ef5f70e
new file mode 100644
index 0000000000000000000000000000000000000000..4a62193d2b5553cccd4171e7b17f89c2f47b8744
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/9e0ab5da9f821773ce5c431a1ef5f70e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b30d20f7c73c40ac91c057ac4d7da7dc8d1dc3c36180e86242761319d9e05342
+size 321081
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/b36d04854e981974598284690f6e7d35 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/b36d04854e981974598284690f6e7d35
new file mode 100644
index 0000000000000000000000000000000000000000..325a46988084d3671580ad24dd9aa53b42ba4417
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/b36d04854e981974598284690f6e7d35
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4b7e7a9b9e450500ca1234498efc014c1621639df0a75116924cda242ce480f4
+size 416978
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/b8a0499cc863a8b408e4fb4873a75388 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/b8a0499cc863a8b408e4fb4873a75388
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/b8a0499cc863a8b408e4fb4873a75388 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/ba36b70f6e7d00dafbf6efbd6ea2bded b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/ba36b70f6e7d00dafbf6efbd6ea2bded
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/ba36b70f6e7d00dafbf6efbd6ea2bded differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5d5240fca3a6f9d6435d603b5238b13d76344f4e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..9652301d4f6accda9f20ed68473e8dbf497e281a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583604469397214, "commit_timestamp_nsecs": 1754583604647365937, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/d/9fe7b2d766480d2a803392776e6a84bc b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/d/9fe7b2d766480d2a803392776e6a84bc
new file mode 100644
index 0000000000000000000000000000000000000000..4af5f296c5b013d350685b6823d202e496680232
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/d/9fe7b2d766480d2a803392776e6a84bc differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..797da0dd40557e0eaf1cf58fe64d2e97ebcab34e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/1a392d5bb9cfcb1f7dacafb12e994755 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/1a392d5bb9cfcb1f7dacafb12e994755
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/1a392d5bb9cfcb1f7dacafb12e994755 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/4cd3b6f0137636d59fb1141a7c5e4bc6 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/4cd3b6f0137636d59fb1141a7c5e4bc6
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/4cd3b6f0137636d59fb1141a7c5e4bc6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/5f19d930b81c2ec2e7314f53e1636c8a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/5f19d930b81c2ec2e7314f53e1636c8a
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/5f19d930b81c2ec2e7314f53e1636c8a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/6ad64be9265dd8215c0a5ee9b365c695 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/6ad64be9265dd8215c0a5ee9b365c695
new file mode 100644
index 0000000000000000000000000000000000000000..85f35e7991cac7b6fc244bb2f277438119f4e90b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/6ad64be9265dd8215c0a5ee9b365c695 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/8b4b0fbbf0944c0de18cf9ce622abc00 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/8b4b0fbbf0944c0de18cf9ce622abc00
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/8b4b0fbbf0944c0de18cf9ce622abc00 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/96bc19425282f706850acfef7c036cb0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/96bc19425282f706850acfef7c036cb0
new file mode 100644
index 0000000000000000000000000000000000000000..63fc115ca51342c4e2bea33278cc95baded1058d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/96bc19425282f706850acfef7c036cb0
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fa8f22ff4669e0d92ebdaf3e5c83f725aa67a5d944ba5e89af4cb907bb3111fa
+size 442380
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/a3042385b62c8640433e2d0f33e93b74 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/a3042385b62c8640433e2d0f33e93b74
new file mode 100644
index 0000000000000000000000000000000000000000..367426a123c8a70f855716ee1011467bbc582630
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/a3042385b62c8640433e2d0f33e93b74 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/be921a4bed3020b2fd76a62bae3d80f4 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/be921a4bed3020b2fd76a62bae3d80f4
new file mode 100644
index 0000000000000000000000000000000000000000..87de24046f61f15d06b2d9892f4860a021614dae
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/be921a4bed3020b2fd76a62bae3d80f4
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b39f788b6bc3cdd7fb4c8dbf0c99cc3b283dc43b4256e2600cf3e5cad5ba1022
+size 295678
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/e97a7fb589eaa78dafd960241b7166d9 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/e97a7fb589eaa78dafd960241b7166d9
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/e97a7fb589eaa78dafd960241b7166d9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b73b9ce27a24945455a228040c9f71f833e03577
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..d669886682e87593000e90dfaefb742808da4f4e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583637057129577, "commit_timestamp_nsecs": 1754583637234221718, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/d/9ea37a6a5051d15c80f402c7072b6e60 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/d/9ea37a6a5051d15c80f402c7072b6e60
new file mode 100644
index 0000000000000000000000000000000000000000..8999502efa1e43374b8c44fdac57895cc2da232b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/d/9ea37a6a5051d15c80f402c7072b6e60 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f284677bd67c3b4157dc1b4d1d31bf16b05839b6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/02431061da5bba50e4272301b0645f74 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/02431061da5bba50e4272301b0645f74
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/02431061da5bba50e4272301b0645f74 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/289cf2707931e35039628c13c0238882 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/289cf2707931e35039628c13c0238882
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/289cf2707931e35039628c13c0238882 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/6fc1b1f4e316605c72f1fa411162534b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/6fc1b1f4e316605c72f1fa411162534b
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/6fc1b1f4e316605c72f1fa411162534b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/790e410120487c74601e5c22a632693a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/790e410120487c74601e5c22a632693a
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/790e410120487c74601e5c22a632693a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/a65a484bb6ebd2c7fe174980f2b408bf b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/a65a484bb6ebd2c7fe174980f2b408bf
new file mode 100644
index 0000000000000000000000000000000000000000..7987e4b208cb40e35eaf106eb547d10c2b4fe052
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/a65a484bb6ebd2c7fe174980f2b408bf
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fed363322462c8168cf8cc006cf2587e04f30e4f15281f8a98b0bf493d64f6e1
+size 389031
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/ac136a93168bdae194f3733209fdaf32 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/ac136a93168bdae194f3733209fdaf32
new file mode 100644
index 0000000000000000000000000000000000000000..f74ee70e3e13c78ed928e76a4bae3592c2721991
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/ac136a93168bdae194f3733209fdaf32
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cba37cf68563977b36ad0d650230c1ef2cb0f6fef5ee8463d6a7527ea4bf6fd7
+size 336024
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/e35c543328b48a5f909f4190965427ed b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/e35c543328b48a5f909f4190965427ed
new file mode 100644
index 0000000000000000000000000000000000000000..ba25cd4341404d50ce3a2a8f68936bda5b44b3ce
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/e35c543328b48a5f909f4190965427ed differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..55a4cd270c2a5a6ae1bdd9efd36aa624851ffd91
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..ea26665fed16ad9f99244780b5c02e51a9bdf165
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583670158654914, "commit_timestamp_nsecs": 1754583670328887631, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/d/226a9cfc2b37d095276c65d81f602cd9 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/d/226a9cfc2b37d095276c65d81f602cd9
new file mode 100644
index 0000000000000000000000000000000000000000..e218b2fe293c5a50fe5d8167ef3d9603530d6d25
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/d/226a9cfc2b37d095276c65d81f602cd9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..48e1ec82f98704cdce2a03cd74a42d6a2baf8de2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/072f5ae4c2056e2511e5e84fca0795d8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/072f5ae4c2056e2511e5e84fca0795d8
new file mode 100644
index 0000000000000000000000000000000000000000..cc78a6b799eabe36d2596ebc64beabb8d843ac0e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/072f5ae4c2056e2511e5e84fca0795d8
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1f3f3ddb68fd816a19b73697319f10310d66273e13c805aa1023e46ddd373b54
+size 295672
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/28929e22d1295d4e9d8742d888fc30e5 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/28929e22d1295d4e9d8742d888fc30e5
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/28929e22d1295d4e9d8742d888fc30e5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/5bee4c8e5285c77153f52086b05a37e7 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/5bee4c8e5285c77153f52086b05a37e7
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/5bee4c8e5285c77153f52086b05a37e7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/63763dc1ea848fc0684e845b061e1560 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/63763dc1ea848fc0684e845b061e1560
new file mode 100644
index 0000000000000000000000000000000000000000..6ba8cffa1fde1631c1886c6e3dc34322186502c7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/63763dc1ea848fc0684e845b061e1560 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/859cdc4264d3df58a8910250827c3237 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/859cdc4264d3df58a8910250827c3237
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/859cdc4264d3df58a8910250827c3237 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/998ccb19d65d5d4a4fccf6385e8225c1 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/998ccb19d65d5d4a4fccf6385e8225c1
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/998ccb19d65d5d4a4fccf6385e8225c1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/a2a92c73d02b4c597f7df4032f844ac3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/a2a92c73d02b4c597f7df4032f844ac3
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/a2a92c73d02b4c597f7df4032f844ac3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/f95508e89f5dcf2d89ee4bdd9b78ad09 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/f95508e89f5dcf2d89ee4bdd9b78ad09
new file mode 100644
index 0000000000000000000000000000000000000000..c76b4b9fbb09cbe18f61f9edc7b114850edc0805
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/f95508e89f5dcf2d89ee4bdd9b78ad09
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d65e1f3470cf97ff2c4b19a3e994b82cb2927f7876256a0f4af0c15f3539d831
+size 442417
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..6a3e8cb0065b7d785763b7d16e9fd7a4b4654314
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..32fd0868168c9145fd9585d65ba6001e628a8218
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583704725627238, "commit_timestamp_nsecs": 1754583704879970717, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/d/9410d74937178c1225c782eeeb004f9d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/d/9410d74937178c1225c782eeeb004f9d
new file mode 100644
index 0000000000000000000000000000000000000000..182dbd8902ffc8e6b0829eb3a15e8265e6a30544
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/d/9410d74937178c1225c782eeeb004f9d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..03bbc02c3718a4f15c5d683da1aa93641164be9f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/0a2ab96a22725414c145abf8af9a471c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/0a2ab96a22725414c145abf8af9a471c
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/0a2ab96a22725414c145abf8af9a471c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/264aa2908052c8ce7ff4403458316ad9 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/264aa2908052c8ce7ff4403458316ad9
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/264aa2908052c8ce7ff4403458316ad9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/8275cb81e7516fbd42ee34357805bd84 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/8275cb81e7516fbd42ee34357805bd84
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/8275cb81e7516fbd42ee34357805bd84 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/a1368278bc9089efb5e1649964834e9a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/a1368278bc9089efb5e1649964834e9a
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/a1368278bc9089efb5e1649964834e9a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/b79f6eb9c727341a583d420f43252c9e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/b79f6eb9c727341a583d420f43252c9e
new file mode 100644
index 0000000000000000000000000000000000000000..63a45366c89c8e90e0286750873ed7ef858d6193
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/b79f6eb9c727341a583d420f43252c9e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0c16d87ce11c8c0ef12be26f68f186a4f4cde5e6f2ffce160ee203bbb8e09a2f
+size 253291
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/c5b9a730e97f78147762a435e80dbca5 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/c5b9a730e97f78147762a435e80dbca5
new file mode 100644
index 0000000000000000000000000000000000000000..20c9d21b31815fe54763b31c242ed02e069c76f3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/c5b9a730e97f78147762a435e80dbca5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/c6f089bd2aa9d6f8a5353566acd18706 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/c6f089bd2aa9d6f8a5353566acd18706
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/c6f089bd2aa9d6f8a5353566acd18706 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/db0ff1bd938599d1fba0749c16397dfd b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/db0ff1bd938599d1fba0749c16397dfd
new file mode 100644
index 0000000000000000000000000000000000000000..509f38f5260906f39f3aeaf10cedd6a54b71ee49
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/db0ff1bd938599d1fba0749c16397dfd
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5ea8ee0ac737269c8ceba0d072c1a3907ef1ea065585892a64788366064d0267
+size 484994
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..56265fcb9d41a56edd665f87fc358b20d5a60420
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..666f35c99f633bdaa41023609bc2287c633e7c4d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583738534075151, "commit_timestamp_nsecs": 1754583738720846902, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/d/3f03169dd42adf88f568bd64e56470a2 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/d/3f03169dd42adf88f568bd64e56470a2
new file mode 100644
index 0000000000000000000000000000000000000000..75dcccfc9f81981d977d1eb5e6576cc5e4c7f6cb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/d/3f03169dd42adf88f568bd64e56470a2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..85ed3a921412e81eaf7c2a74ffb4d79d5e7c8552
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/18cfc954aea7a075ca7ba8017ee10555 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/18cfc954aea7a075ca7ba8017ee10555
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/18cfc954aea7a075ca7ba8017ee10555 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/2faf607f0ff722e9e24da0f7ac43be4e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/2faf607f0ff722e9e24da0f7ac43be4e
new file mode 100644
index 0000000000000000000000000000000000000000..bfa806b310bf6681ac1066f6ed0058faeecca852
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/2faf607f0ff722e9e24da0f7ac43be4e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2af6f98ff1b7ff80c105ce96a651b6cf029e47bdbc963b1de7fdebda4f763e51
+size 436932
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/3859a4a79ddb73cb22a050456b9619b4 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/3859a4a79ddb73cb22a050456b9619b4
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/3859a4a79ddb73cb22a050456b9619b4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/45c4d5dbfd11692d99238f8e42e02d8f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/45c4d5dbfd11692d99238f8e42e02d8f
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/45c4d5dbfd11692d99238f8e42e02d8f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/46ec0b825d2ab81ca9c73a87067cd399 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/46ec0b825d2ab81ca9c73a87067cd399
new file mode 100644
index 0000000000000000000000000000000000000000..01981b05e4c2dd7e73d7bd739555f13b28e9fed8
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/46ec0b825d2ab81ca9c73a87067cd399
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:49de7e07c43b77266e83c4a0307e88db1cfbc375fada4ed0b6a7dbde520c0168
+size 227617
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/66644daf7f5377e527ec3e16950e9318 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/66644daf7f5377e527ec3e16950e9318
new file mode 100644
index 0000000000000000000000000000000000000000..9d14e3cc9cf9502948582f140277e2b2bc0385b3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/66644daf7f5377e527ec3e16950e9318 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/6980c0d2026f895a4405ba5977ff7b67 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/6980c0d2026f895a4405ba5977ff7b67
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/6980c0d2026f895a4405ba5977ff7b67 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/98daadf9bb4e5e3cb54c90bdf462ccfe b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/98daadf9bb4e5e3cb54c90bdf462ccfe
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/98daadf9bb4e5e3cb54c90bdf462ccfe differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b06bfe7fcdf964a23a5e57c88ddf2199de22a113
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..5a588bd9671519d233969a1ddab2c4756a1e7000
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583770350236870, "commit_timestamp_nsecs": 1754583770534396335, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/d/3070f1dd04cb02afd1b5ce31caddf5d7 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/d/3070f1dd04cb02afd1b5ce31caddf5d7
new file mode 100644
index 0000000000000000000000000000000000000000..4eb4fa6da317cc93e6685fcd25eac7b12f5d24ee
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/d/3070f1dd04cb02afd1b5ce31caddf5d7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9c4e5aacc5cb053b2400f43bf435ad321c2c3fcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/09103f6d933d8e646b0533f5a1d6b56e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/09103f6d933d8e646b0533f5a1d6b56e
new file mode 100644
index 0000000000000000000000000000000000000000..0138aa0a88283241929ab0e03d4efb42ecb524ac
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/09103f6d933d8e646b0533f5a1d6b56e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3fb61c8e46fcc9b5f76359d2840af2e481705992f94c7fde9972d8bb68c33a68
+size 123773
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/bbfa9f4c0577270bfd8c17b3c2b0db4f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/bbfa9f4c0577270bfd8c17b3c2b0db4f
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/bbfa9f4c0577270bfd8c17b3c2b0db4f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/c217f1d7b9ee3dd27421b3ce64c083b0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/c217f1d7b9ee3dd27421b3ce64c083b0
new file mode 100644
index 0000000000000000000000000000000000000000..4efe795ee36556d86c70c59db1f5f6a49d2a98df
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/c217f1d7b9ee3dd27421b3ce64c083b0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/cdda68a55aa0e32d76ed4a1d08385a49 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/cdda68a55aa0e32d76ed4a1d08385a49
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/cdda68a55aa0e32d76ed4a1d08385a49 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/d7196d33571ed3c79fdbc003de901a11 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/d7196d33571ed3c79fdbc003de901a11
new file mode 100644
index 0000000000000000000000000000000000000000..5bb96b3605ec995799b363ea2102c9762727ca85
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/d7196d33571ed3c79fdbc003de901a11 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/ddc6170fa1025762215884e722d282f5 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/ddc6170fa1025762215884e722d282f5
new file mode 100644
index 0000000000000000000000000000000000000000..361bbac3eaf937cae9dd7de1f5716f88b01705af
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/ddc6170fa1025762215884e722d282f5
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:30cf5c4554d93ee1971a4f71a30ced1531ff629edac889556c4bd88a7c1b27b2
+size 293455
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/e97e592102f2f73360be48f1fecd06c5 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/e97e592102f2f73360be48f1fecd06c5
new file mode 100644
index 0000000000000000000000000000000000000000..fb3197216d2fca02daad15324f6e5646b95a6255
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/e97e592102f2f73360be48f1fecd06c5
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7e40061949c0ef460655c8a3d593b756f7ce5c3a147c2bd830823d3b0de3787c
+size 247894
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/f1e45debfd9c65a773500acfe0f08879 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/f1e45debfd9c65a773500acfe0f08879
new file mode 100644
index 0000000000000000000000000000000000000000..f1b702d1a1ada1c8ca3209589754c9aa150dc750
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/f1e45debfd9c65a773500acfe0f08879 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..aef659185f7263a1fd8be7d3f53815fb290802dc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..f0a5cdab4f3266a920b6192fa49ef305dab0324a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583803482450252, "commit_timestamp_nsecs": 1754583803650968121, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/d/b9912a2e67c25884e38d68070b9b792f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/d/b9912a2e67c25884e38d68070b9b792f
new file mode 100644
index 0000000000000000000000000000000000000000..69bc507b415d8e84b203b6e6caf2f246b6d7b151
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/d/b9912a2e67c25884e38d68070b9b792f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9ee9417665f76fdd6207cd6c9761340aad578dab
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/0006571684894e34b313ef657038e210 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/0006571684894e34b313ef657038e210
new file mode 100644
index 0000000000000000000000000000000000000000..a53a70fb16465e90a83c4b7af44f404a7848cacb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/0006571684894e34b313ef657038e210
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7641ce88821aeae84c4f4aebf0bcce8c5597d2f6be01524987ad4432d1d96098
+size 278649
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/101ecfbefa7c84e4753c6a7054a61564 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/101ecfbefa7c84e4753c6a7054a61564
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/101ecfbefa7c84e4753c6a7054a61564 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/47d099195568b16d8a829d7ddb745141 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/47d099195568b16d8a829d7ddb745141
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/47d099195568b16d8a829d7ddb745141 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/5897853237cd6736a8f90bd3a46874b7 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/5897853237cd6736a8f90bd3a46874b7
new file mode 100644
index 0000000000000000000000000000000000000000..ec745d97353f97209ef50f0eb738585830219f53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/5897853237cd6736a8f90bd3a46874b7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/7d7288d4619429282939e8e6e99b3e4d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/7d7288d4619429282939e8e6e99b3e4d
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/7d7288d4619429282939e8e6e99b3e4d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/a7859ed79c1028d22bdc6501b9733c06 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/a7859ed79c1028d22bdc6501b9733c06
new file mode 100644
index 0000000000000000000000000000000000000000..14efb3066fda3e489a11ec3d4cc3f69a9e60b1a3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/a7859ed79c1028d22bdc6501b9733c06
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:05260175957e7b04daef1b9f54f48d6ab869bc34ce78da28dbf511bf2230184c
+size 459567
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/b7b70daaa06cbc8fd95b4e45a903aaae b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/b7b70daaa06cbc8fd95b4e45a903aaae
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/b7b70daaa06cbc8fd95b4e45a903aaae differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/e13e3e522e7a06af51b6cfdd6524d572 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/e13e3e522e7a06af51b6cfdd6524d572
new file mode 100644
index 0000000000000000000000000000000000000000..0b2ed10f865c2f9a716111125629ee3f61a19ecc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/e13e3e522e7a06af51b6cfdd6524d572 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4bdf4ae4a5ee07e67963c43e609fe29606105e35
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..5ab42519025cc6bc881ee627ac482000cab0c635
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754582929356278861, "commit_timestamp_nsecs": 1754582929541388612, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/d/b03c11a0a793b96c8054975068916a12 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/d/b03c11a0a793b96c8054975068916a12
new file mode 100644
index 0000000000000000000000000000000000000000..bcccf733d38884e98208ff4fe11fe0024300e468
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/d/b03c11a0a793b96c8054975068916a12 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3617f184c63d4f6b56a885498facc73774f4ed7a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/35176f604442bcc7cb753552c5a827f3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/35176f604442bcc7cb753552c5a827f3
new file mode 100644
index 0000000000000000000000000000000000000000..4859214e6b68fa6e697de9748417fcf43be8081d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/35176f604442bcc7cb753552c5a827f3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/8736b48fbbfd1ba7dcbc840773bf4f2b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/8736b48fbbfd1ba7dcbc840773bf4f2b
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/8736b48fbbfd1ba7dcbc840773bf4f2b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/944680b9ab4b92411114218f6df070ae b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/944680b9ab4b92411114218f6df070ae
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/944680b9ab4b92411114218f6df070ae differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/9c866ce67fcde302824861c38604f74f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/9c866ce67fcde302824861c38604f74f
new file mode 100644
index 0000000000000000000000000000000000000000..7ec2cac203425f3bf7192960f8fa03d66f1de7e3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/9c866ce67fcde302824861c38604f74f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/a326187bc642e0f4da8b87c22915beec b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/a326187bc642e0f4da8b87c22915beec
new file mode 100644
index 0000000000000000000000000000000000000000..c0ead2b6ad73b7b9630145c72f652969a740a93e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/a326187bc642e0f4da8b87c22915beec
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:334ffaefb5f19b51e355231a1fb957cb25e8227473ae380524cc51aa9b4ebc54
+size 278358
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/af99c8d9a321dec4daeeaa563d84e97c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/af99c8d9a321dec4daeeaa563d84e97c
new file mode 100644
index 0000000000000000000000000000000000000000..86fc765304e5849386216dc5a1724d45c7c0bae9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/af99c8d9a321dec4daeeaa563d84e97c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7aac70a35220b5921f1eb3acbc2d24aa6814d1719d7ce1e6aa821b9043361af5
+size 421677
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/c0c93153028b0432a0c28daa0839f5cb b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/c0c93153028b0432a0c28daa0839f5cb
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/c0c93153028b0432a0c28daa0839f5cb differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/d31c3a7e9b06527e8dbc9328b2c1fc06 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/d31c3a7e9b06527e8dbc9328b2c1fc06
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/d31c3a7e9b06527e8dbc9328b2c1fc06 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/d3b977b34ab08779d4fde833636784d6 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/d3b977b34ab08779d4fde833636784d6
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/d3b977b34ab08779d4fde833636784d6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a51098667c1da867de98a09767530b6017cb3908
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..f6e3821d48e93c0fca5dda952f5015e4449979d5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583835043799553, "commit_timestamp_nsecs": 1754583835185248899, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/d/5e098f731a2f8265dd31cce99a5916aa b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/d/5e098f731a2f8265dd31cce99a5916aa
new file mode 100644
index 0000000000000000000000000000000000000000..176c57d2a2b990025355cda43867a1d29a78d19f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/d/5e098f731a2f8265dd31cce99a5916aa differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..84acab16047308552eb343889eef2aef3313f25c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/1a3cfcd6fa0a1f8b601573ef0406374c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/1a3cfcd6fa0a1f8b601573ef0406374c
new file mode 100644
index 0000000000000000000000000000000000000000..c02ed10c01751dd82c4b876aa09ff10466887087
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/1a3cfcd6fa0a1f8b601573ef0406374c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8964a3d4e3422ceea41b8bcf7c10d02b5afbd86e0020ff2ec018c102ff8e2ba9
+size 374383
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/2494436ad71152d8fa3f0a09b179616a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/2494436ad71152d8fa3f0a09b179616a
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/2494436ad71152d8fa3f0a09b179616a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/53ccc80b4b6d640db47667a9c7cbe703 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/53ccc80b4b6d640db47667a9c7cbe703
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/53ccc80b4b6d640db47667a9c7cbe703 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/5e5f2f3e8e356e0fb29bf82e5fc586e4 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/5e5f2f3e8e356e0fb29bf82e5fc586e4
new file mode 100644
index 0000000000000000000000000000000000000000..416ef18b904414941019d6ce25580e2b2a0f4179
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/5e5f2f3e8e356e0fb29bf82e5fc586e4
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a5bc9853431bcb65646a46529faf06eb743516914b253a95e486c745b7cf6229
+size 295724
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/7f49f0007b0abbdde93331a2408271c6 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/7f49f0007b0abbdde93331a2408271c6
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/7f49f0007b0abbdde93331a2408271c6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/ab06a0305134150e1236285ed48387ba b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/ab06a0305134150e1236285ed48387ba
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/ab06a0305134150e1236285ed48387ba differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/cba07a06ea7459ae9a6ba7172b883ff3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/cba07a06ea7459ae9a6ba7172b883ff3
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/cba07a06ea7459ae9a6ba7172b883ff3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/fb1cef03d907ce33c127687fd2839a5f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/fb1cef03d907ce33c127687fd2839a5f
new file mode 100644
index 0000000000000000000000000000000000000000..34cd2b22ae7bbc9cc3c0b33c954a2b31a64be6b7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/fb1cef03d907ce33c127687fd2839a5f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..11fab7960236b962faa1cf3bc44922119fbdb66c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..dc1c3b2846ac691d8aee3b841c749279c31b668c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583869091072496, "commit_timestamp_nsecs": 1754583869255351527, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/d/31fee3bb97e58d94cb07d433427d39f5 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/d/31fee3bb97e58d94cb07d433427d39f5
new file mode 100644
index 0000000000000000000000000000000000000000..6b4f39f605d2fd3f62b2cc50a4d6ccbc516db7db
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/d/31fee3bb97e58d94cb07d433427d39f5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..710f01e7622c94bdf2a98e4252fdf72a7642778e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/057fa63e618aa072f1242b2332e0c73b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/057fa63e618aa072f1242b2332e0c73b
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/057fa63e618aa072f1242b2332e0c73b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/173aca346e30f44ddfec505154c6df4c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/173aca346e30f44ddfec505154c6df4c
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/173aca346e30f44ddfec505154c6df4c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/61e224db0ede6e3fbeee55944d19d059 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/61e224db0ede6e3fbeee55944d19d059
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/61e224db0ede6e3fbeee55944d19d059 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/981cf88349801bea79ff21a9ee502320 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/981cf88349801bea79ff21a9ee502320
new file mode 100644
index 0000000000000000000000000000000000000000..7b6a2ede2fd3a496c40d89f0a5fef04af7fca928
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/981cf88349801bea79ff21a9ee502320
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c513771137ac651bf32802009ff45138adccf6071efa5106a7be4a4ea3c6d183
+size 363767
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/9f619b3036e38de15d2dba2ca21fe104 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/9f619b3036e38de15d2dba2ca21fe104
new file mode 100644
index 0000000000000000000000000000000000000000..c84ae19dc95a86b1ea4f223f1a7e8559bd7d9768
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/9f619b3036e38de15d2dba2ca21fe104
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1ee0953f1c2ebd35b7e0a30c83e562914904d883c8705c5de18703314be98de6
+size 374124
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/ab0b66c60987083fdec68b9342cd6455 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/ab0b66c60987083fdec68b9342cd6455
new file mode 100644
index 0000000000000000000000000000000000000000..a1938d32fde300fcf2a61a0e5959fd4ccfc7310c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/ab0b66c60987083fdec68b9342cd6455 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/d136a8d625cfd7e18ca3956d03a589d3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/d136a8d625cfd7e18ca3956d03a589d3
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/d136a8d625cfd7e18ca3956d03a589d3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..20966617493bf92964f65d0ef2d4f36a3f0dacd9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..0eb72f4a14b8572776b0de1b1ebd7eec7080166a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583916832203562, "commit_timestamp_nsecs": 1754583916980537077, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/d/dc19ed8a0e8ff955f29739a2ca855d66 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/d/dc19ed8a0e8ff955f29739a2ca855d66
new file mode 100644
index 0000000000000000000000000000000000000000..3a3c82e5149b6308ed76c1d259cb59f7c01d9d1e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/d/dc19ed8a0e8ff955f29739a2ca855d66 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..79a3d5d8e1ff95fb9c37efefe1ddf58a961e9498
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/4b98720177d384cbd2ccda50ccbb172e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/4b98720177d384cbd2ccda50ccbb172e
new file mode 100644
index 0000000000000000000000000000000000000000..edbaebae86be5c53bb56d50de55cf8d33f5937a2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/4b98720177d384cbd2ccda50ccbb172e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c449d9d426382c42a2dd0d46c3ef439a4b556b6ebd726412e9fc2236a94fae01
+size 261822
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/6038b75c3ca31f3239c9490a2fc9be27 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/6038b75c3ca31f3239c9490a2fc9be27
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/6038b75c3ca31f3239c9490a2fc9be27 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/8d489f053e5a532ebeec762445d0ca2b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/8d489f053e5a532ebeec762445d0ca2b
new file mode 100644
index 0000000000000000000000000000000000000000..93f30dcceb537a6c8ec666eca71a0b6976bc61f5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/8d489f053e5a532ebeec762445d0ca2b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/998713ad647f6878138b3c1e4ce48866 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/998713ad647f6878138b3c1e4ce48866
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/998713ad647f6878138b3c1e4ce48866 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/9e5f384c0d58ff333355b8e150197a23 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/9e5f384c0d58ff333355b8e150197a23
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/9e5f384c0d58ff333355b8e150197a23 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/abe762c34ef8f25c0d33c54c3d69f0fd b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/abe762c34ef8f25c0d33c54c3d69f0fd
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/abe762c34ef8f25c0d33c54c3d69f0fd differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/c45e677fd02878c10b3ba75aaca46961 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/c45e677fd02878c10b3ba75aaca46961
new file mode 100644
index 0000000000000000000000000000000000000000..92ef168a80951e22433a50bf2781dcbd2f72207f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/c45e677fd02878c10b3ba75aaca46961
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:55afdc280aab333767260c45914c3cc7308367943e763b0abb23916ed1f9d496
+size 442351
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/c77a6b1bfa7fcceb1633f2100c9db9d8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/c77a6b1bfa7fcceb1633f2100c9db9d8
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/c77a6b1bfa7fcceb1633f2100c9db9d8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4da4c342a9299d94bfcbcd71958be056e10ca2ac
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..e3a80c27af56a93659ad0519c355719613e0d9c7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583949416821166, "commit_timestamp_nsecs": 1754583949558953336, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/d/8379a71c986442a7f6fb618e5bbd6451 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/d/8379a71c986442a7f6fb618e5bbd6451
new file mode 100644
index 0000000000000000000000000000000000000000..5b02522b3c1a0245f5bdfe0aa583b1f8e1aedcba
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/d/8379a71c986442a7f6fb618e5bbd6451 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4b0021c5220640d614b842347fc161ad8212643c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/06f976f79e4869e9f4f6c792776de41f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/06f976f79e4869e9f4f6c792776de41f
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/06f976f79e4869e9f4f6c792776de41f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/079722bcfde8e7deecd952b08a51f698 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/079722bcfde8e7deecd952b08a51f698
new file mode 100644
index 0000000000000000000000000000000000000000..c0918988847cd232aab5007343b67e30e98b5516
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/079722bcfde8e7deecd952b08a51f698 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/0b384fcf082c396ec7f8c508a54ef078 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/0b384fcf082c396ec7f8c508a54ef078
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/0b384fcf082c396ec7f8c508a54ef078 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/630c0bc146a039afc2bb0360d6bd5be0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/630c0bc146a039afc2bb0360d6bd5be0
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/630c0bc146a039afc2bb0360d6bd5be0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/9162dcb44225dd0f66c2fa80eba9de0b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/9162dcb44225dd0f66c2fa80eba9de0b
new file mode 100644
index 0000000000000000000000000000000000000000..4859214e6b68fa6e697de9748417fcf43be8081d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/9162dcb44225dd0f66c2fa80eba9de0b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/93a12c0def55eba04d111034da58bdc3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/93a12c0def55eba04d111034da58bdc3
new file mode 100644
index 0000000000000000000000000000000000000000..e652005c690fb172fa4a35bdbabcce8a2729102e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/93a12c0def55eba04d111034da58bdc3
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8673cfe623f579557d59ab4fb7ac4efaee4326b68379051781a23464b98cce42
+size 321199
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/a2810388c8fbe9ee3cf069ced4855508 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/a2810388c8fbe9ee3cf069ced4855508
new file mode 100644
index 0000000000000000000000000000000000000000..1f53c906d2d09c08b11f62762ebea2a8fe6bd5d1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/a2810388c8fbe9ee3cf069ced4855508
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:98bf0b0b9b61a91b7c43c4feaf59a50143bf82fd60f5d9a0e9d8f3a0751d4b95
+size 416876
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/c722a7a7ed9ecc1d56a309be12f44d6f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/c722a7a7ed9ecc1d56a309be12f44d6f
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/c722a7a7ed9ecc1d56a309be12f44d6f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a675e05c7ec633d60be0f12d09873e14f6984cff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..d98206712d5999a41811d912fb9f9a7d58369ca7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583982578431471, "commit_timestamp_nsecs": 1754583982764673758, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/d/2d32bb4d6ea26f93fa0d2e445d8c3d70 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/d/2d32bb4d6ea26f93fa0d2e445d8c3d70
new file mode 100644
index 0000000000000000000000000000000000000000..d37a399b3273c58ba18092b965b4472703cf2b3d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/d/2d32bb4d6ea26f93fa0d2e445d8c3d70 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..cead4ed304bf33beb3905b18671a79ede56b57ce
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/0c2b40b98593fdf024e9df1df96fefa9 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/0c2b40b98593fdf024e9df1df96fefa9
new file mode 100644
index 0000000000000000000000000000000000000000..1f2debc88120d237bb0f34968fec4e63b9e54d0e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/0c2b40b98593fdf024e9df1df96fefa9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/9e9fac7b54a5df39616eb17de7966f6d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/9e9fac7b54a5df39616eb17de7966f6d
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/9e9fac7b54a5df39616eb17de7966f6d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/b19fb6ba48fbe2bf18de2a0533a9715b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/b19fb6ba48fbe2bf18de2a0533a9715b
new file mode 100644
index 0000000000000000000000000000000000000000..4a40795922d9030c67c9dd663d50b0d318c5275d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/b19fb6ba48fbe2bf18de2a0533a9715b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f4af9df677f8b257cd5c77da3eda9b9ec977843b422b8c2bfefe8f614610892c
+size 408092
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/bd4324b8606a8ed216e5f66fdf7cb85a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/bd4324b8606a8ed216e5f66fdf7cb85a
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/bd4324b8606a8ed216e5f66fdf7cb85a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/c8daee29bbb44342edd48c44f16bf60e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/c8daee29bbb44342edd48c44f16bf60e
new file mode 100644
index 0000000000000000000000000000000000000000..9434ccc1e4aa5213d56bd4884d51c6c415df3e01
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/c8daee29bbb44342edd48c44f16bf60e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2f0524e42e893bfb50dd140f6a5ed7e60c96c6c9b55d02472a057d54bf3e7ceb
+size 295767
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/ea903a3a031bd9fc7b13a5c42b22ef37 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/ea903a3a031bd9fc7b13a5c42b22ef37
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/ea903a3a031bd9fc7b13a5c42b22ef37 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/f8c641c18481ac35a1cb01056fc05090 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/f8c641c18481ac35a1cb01056fc05090
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/f8c641c18481ac35a1cb01056fc05090 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..da34ed3b1958d81200e193e471770b39370be19c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..ada551b94c2c53cf1bce95b1e4fb6eb32fc313b6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584015568238894, "commit_timestamp_nsecs": 1754584015758806208, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/d/7d55811028933a5af88142e8629f4159 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/d/7d55811028933a5af88142e8629f4159
new file mode 100644
index 0000000000000000000000000000000000000000..efc5ee07119ff26e89bf66f07174110fc989a84f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/d/7d55811028933a5af88142e8629f4159 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..ad0a44b6d3997d50169b02246598f62b5c16fa7b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/1bc3bf01910de50fb3a0c8665460fee8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/1bc3bf01910de50fb3a0c8665460fee8
new file mode 100644
index 0000000000000000000000000000000000000000..2a17d97de1ae10b83dfaaac579143939db548893
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/1bc3bf01910de50fb3a0c8665460fee8
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7ea5f3724bc704157b0ca1c0e5457c1bb8c612682eb6b8896577f2bdb92791b1
+size 518768
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/42b2202051b4a403866c00e8440b5937 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/42b2202051b4a403866c00e8440b5937
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/42b2202051b4a403866c00e8440b5937 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/9b5005af93430771332d85d7fe14169a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/9b5005af93430771332d85d7fe14169a
new file mode 100644
index 0000000000000000000000000000000000000000..30a332d7bb8223968e4e9af770269d8a84dec05a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/9b5005af93430771332d85d7fe14169a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bc95086e31ae1abbc088ca92ac4fc91d964988fad808538c50aae3c5b25df6e8
+size 219177
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/bfc3e1edd700264c7bee02279f260ac9 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/bfc3e1edd700264c7bee02279f260ac9
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/bfc3e1edd700264c7bee02279f260ac9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/c4b38d6197a87c1be28fc6f54ddc519c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/c4b38d6197a87c1be28fc6f54ddc519c
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/c4b38d6197a87c1be28fc6f54ddc519c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/ec7cbbf41a282cb431db6e3c9e333380 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/ec7cbbf41a282cb431db6e3c9e333380
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/ec7cbbf41a282cb431db6e3c9e333380 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/f0a4c36f8a8cbad0f1849322a43a28e7 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/f0a4c36f8a8cbad0f1849322a43a28e7
new file mode 100644
index 0000000000000000000000000000000000000000..f28d59e1f9187cfd65b688b4bceb57a81f0acdcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/f0a4c36f8a8cbad0f1849322a43a28e7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/f31ba69df9fc741bda8ffaec309b062b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/f31ba69df9fc741bda8ffaec309b062b
new file mode 100644
index 0000000000000000000000000000000000000000..6534cafffe3a298dde7ab25d852ef53c9ff14ed8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/f31ba69df9fc741bda8ffaec309b062b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8e980fe8d189a6f09f7563051885e0a33494c03b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..4a0c7102090251fe446ac89ae5333dfcd7faf289
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584048653616643, "commit_timestamp_nsecs": 1754584048824251351, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/d/1815383e8a98f088ee2efa2f9e86ac39 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/d/1815383e8a98f088ee2efa2f9e86ac39
new file mode 100644
index 0000000000000000000000000000000000000000..fd5a6ad27934e39dcc8ae529e3e17e1de3872e7e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/d/1815383e8a98f088ee2efa2f9e86ac39 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..7ceb5739a68dd6bca6d4b89665ae49868c781f5b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/005c4ada358a224620aaeb1c46fa523c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/005c4ada358a224620aaeb1c46fa523c
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/005c4ada358a224620aaeb1c46fa523c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/254566bf1dab1d23b971299e5b86fdca b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/254566bf1dab1d23b971299e5b86fdca
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/254566bf1dab1d23b971299e5b86fdca differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/44263fe3a8f0a55b8a521bd41797a61e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/44263fe3a8f0a55b8a521bd41797a61e
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/44263fe3a8f0a55b8a521bd41797a61e differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/7c16d39b29f494ed0c7963ce920d6939 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/7c16d39b29f494ed0c7963ce920d6939
new file mode 100644
index 0000000000000000000000000000000000000000..557eda0fd3ac27cc5886075a4f314bb0a8ab51ee
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/7c16d39b29f494ed0c7963ce920d6939
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f318753e58f6de7166a13bc9a8ee0e021586c514688b917ca7a2f60de4774897
+size 365559
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/a857a88c8ac48d32379c1dc4d106af72 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/a857a88c8ac48d32379c1dc4d106af72
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/a857a88c8ac48d32379c1dc4d106af72 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/bae6b8dcdf80eec2d83358cca8f0853d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/bae6b8dcdf80eec2d83358cca8f0853d
new file mode 100644
index 0000000000000000000000000000000000000000..afa46b7204430f00fc8c4620357630d288694f1a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/bae6b8dcdf80eec2d83358cca8f0853d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/ddb81cd408c2cae8b4695ed8b29e3a01 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/ddb81cd408c2cae8b4695ed8b29e3a01
new file mode 100644
index 0000000000000000000000000000000000000000..b90655c437a4207a9dc75b554bee7d2a867c5796
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/ddb81cd408c2cae8b4695ed8b29e3a01
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a424f55c8a03d604cded0e7055635026d7df6d41b3fe9eabda86ecd8f93fa619
+size 372317
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4d502da2c2e9b6930773e1242363b7db45eacdeb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..29b0e806243b70175737148ca9ca03128706a4eb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584082389489103, "commit_timestamp_nsecs": 1754584082565263807, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/d/0ef0b79359d256c579146081ce2730ac b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/d/0ef0b79359d256c579146081ce2730ac
new file mode 100644
index 0000000000000000000000000000000000000000..16bc0708031570c39a1eb82f4dcfe9dc0a9ae5dd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/d/0ef0b79359d256c579146081ce2730ac differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..335567ddbc2df04907668f1fb66d03a5a0ca6a01
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/1be8dba722c4740d0d32bdfeac1bb952 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/1be8dba722c4740d0d32bdfeac1bb952
new file mode 100644
index 0000000000000000000000000000000000000000..e50d0cf300a45286d52a9d765492a7e333572de4
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/1be8dba722c4740d0d32bdfeac1bb952
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ed5d849252fca099997fd8591f586ad7b040b594efd7bb2108a263f6d1f81ec7
+size 295788
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/4b65e138b383d4812eacb9b75ea9ef3a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/4b65e138b383d4812eacb9b75ea9ef3a
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/4b65e138b383d4812eacb9b75ea9ef3a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/8c414a0996d77473d979955773a94ec2 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/8c414a0996d77473d979955773a94ec2
new file mode 100644
index 0000000000000000000000000000000000000000..78fbdfc35959c3cf6cc110dff58adac045ddb9ed
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/8c414a0996d77473d979955773a94ec2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/a89cbff66011ebc668cdccc94b0e5824 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/a89cbff66011ebc668cdccc94b0e5824
new file mode 100644
index 0000000000000000000000000000000000000000..71c77c0ca6c59062743736017954da41d1fb3b0e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/a89cbff66011ebc668cdccc94b0e5824
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9d76e0e090061607147af70d663c3b3b209eebd74b308ae7a1a5564ae5753d8e
+size 442367
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/b0cff3e367e9742aa1cb9187d7c8c283 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/b0cff3e367e9742aa1cb9187d7c8c283
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/b0cff3e367e9742aa1cb9187d7c8c283 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/b4210a5109eaed07c7af4cc12de0ba5e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/b4210a5109eaed07c7af4cc12de0ba5e
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/b4210a5109eaed07c7af4cc12de0ba5e differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/b56b6fa444c96ff2b70405563be33e49 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/b56b6fa444c96ff2b70405563be33e49
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/b56b6fa444c96ff2b70405563be33e49 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/d2589459a0a954c0a689608e5ecb2b14 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/d2589459a0a954c0a689608e5ecb2b14
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/d2589459a0a954c0a689608e5ecb2b14 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/e8f576307c99d671a051c230fc46a8b5 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/e8f576307c99d671a051c230fc46a8b5
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/e8f576307c99d671a051c230fc46a8b5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..47178ef1fb221f5d9da6c936963caab6c727346c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..59e77b5dff51091e621c88ac16d9845d9e9ed763
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584115732635288, "commit_timestamp_nsecs": 1754584115919414265, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/d/2b0e95a7ac964a58ff61b701bcbfeb03 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/d/2b0e95a7ac964a58ff61b701bcbfeb03
new file mode 100644
index 0000000000000000000000000000000000000000..bc03247e91e26ea0457d7c629ef1824742ff7c0d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/d/2b0e95a7ac964a58ff61b701bcbfeb03 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4dd00d4beb97cc8c29159d0090d1253f5e16684e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/1ea0f65b42965bc96947bae7aa2a8d70 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/1ea0f65b42965bc96947bae7aa2a8d70
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/1ea0f65b42965bc96947bae7aa2a8d70 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/3de63f7f6dc861ede8ec48df5198a17c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/3de63f7f6dc861ede8ec48df5198a17c
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/3de63f7f6dc861ede8ec48df5198a17c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/c3d7fcbf2f1bf5cab32326daf021e5bb b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/c3d7fcbf2f1bf5cab32326daf021e5bb
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/c3d7fcbf2f1bf5cab32326daf021e5bb differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/d178a604706996f23cba2cd8f5fcbe1e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/d178a604706996f23cba2cd8f5fcbe1e
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/d178a604706996f23cba2cd8f5fcbe1e differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/d1e363be172047ce43514c4daf2310cf b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/d1e363be172047ce43514c4daf2310cf
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/d1e363be172047ce43514c4daf2310cf differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/ecc852ecb448308a5820ff56809dfbab b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/ecc852ecb448308a5820ff56809dfbab
new file mode 100644
index 0000000000000000000000000000000000000000..f2a1c694bf1fabfaa419b99e9b8dc66ad9300b64
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/ecc852ecb448308a5820ff56809dfbab
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:eeae219ab55332673746ac2124653d374fc35bc86f2313444d7af77c042a39fc
+size 287241
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/ef86af8d816f9b19f9c5a40bab737f0d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/ef86af8d816f9b19f9c5a40bab737f0d
new file mode 100644
index 0000000000000000000000000000000000000000..34a15c219b84fc57dac1dacd5ad19686ffea35cd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/ef86af8d816f9b19f9c5a40bab737f0d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/fa994bd50a06db9db0670eb20ddce333 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/fa994bd50a06db9db0670eb20ddce333
new file mode 100644
index 0000000000000000000000000000000000000000..84b5910bf0ad3b3ebbcec418b806083a76839b3c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/fa994bd50a06db9db0670eb20ddce333
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c625603dc43d7485b06d9ac1d96b2775a3bc0398f4986ffde929419a90f5c666
+size 413376
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..161f24f4bd7bdc757658b31c25d0392e0584a657
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..e0f06cd3da96e2932380944dd226c05668ba74d6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584147821499390, "commit_timestamp_nsecs": 1754584147962754797, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/d/4c94fb35cdf99d7c5bc8dd15acd6e515 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/d/4c94fb35cdf99d7c5bc8dd15acd6e515
new file mode 100644
index 0000000000000000000000000000000000000000..3974aa04036407b865e2506894949ec5591c6b26
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/d/4c94fb35cdf99d7c5bc8dd15acd6e515 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..fb008117661d49864a1c0afc932b4461afa92a30
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/08b189de608693b6394846d27a40f08a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/08b189de608693b6394846d27a40f08a
new file mode 100644
index 0000000000000000000000000000000000000000..a721ab35c17f4edb13a2e5eca64dc1444296830a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/08b189de608693b6394846d27a40f08a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/0f5b9365cd070dc55f7d98dda03733d1 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/0f5b9365cd070dc55f7d98dda03733d1
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/0f5b9365cd070dc55f7d98dda03733d1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/13b62b197acd654122b44245b59ff7c3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/13b62b197acd654122b44245b59ff7c3
new file mode 100644
index 0000000000000000000000000000000000000000..f232ae8a1f22f181bfcaec0357eace3d721922db
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/13b62b197acd654122b44245b59ff7c3
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:af143acb5a5ddaab583b2839074f9e517c75425a03ee3966c341b8410d1aafb1
+size 734755
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/5a85a36cb127e212b51f012d7b04a095 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/5a85a36cb127e212b51f012d7b04a095
new file mode 100644
index 0000000000000000000000000000000000000000..2910fbc6db2094404ed932c7d9bcdb23beb99906
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/5a85a36cb127e212b51f012d7b04a095 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/98f7976c935ad792dc9219d9b554f679 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/98f7976c935ad792dc9219d9b554f679
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/98f7976c935ad792dc9219d9b554f679 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/a69745e1f44791ded8ecd074435d4946 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/a69745e1f44791ded8ecd074435d4946
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/a69745e1f44791ded8ecd074435d4946 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/b86b5e58d33f487e38020bdf0aa4ec5d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/b86b5e58d33f487e38020bdf0aa4ec5d
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/b86b5e58d33f487e38020bdf0aa4ec5d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/d4ba1dac530df948e1b040b7203bc5f3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/d4ba1dac530df948e1b040b7203bc5f3
new file mode 100644
index 0000000000000000000000000000000000000000..4e8b0b39da5b122050e690184c4c32955da942c5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/d4ba1dac530df948e1b040b7203bc5f3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..190ade0cad017feb79716bab7ad46c504cb724d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..b9d394b11ff7d679062c1a47a987bb02bc3a11cb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754582964060846132, "commit_timestamp_nsecs": 1754582964248808588, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/d/8a1b6714e3e3d2b2026cd07eea5d27ac b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/d/8a1b6714e3e3d2b2026cd07eea5d27ac
new file mode 100644
index 0000000000000000000000000000000000000000..f50d24ea7e77abb807185466e61104638479b164
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/d/8a1b6714e3e3d2b2026cd07eea5d27ac differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d1ae81d77a16194800db8a4ebae4136cf292205f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/0c86bb3880b0aa0fb0daa9a89bd91e23 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/0c86bb3880b0aa0fb0daa9a89bd91e23
new file mode 100644
index 0000000000000000000000000000000000000000..0f9fc484ecb62a69c89c01cb1b3ba1c358dda518
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/0c86bb3880b0aa0fb0daa9a89bd91e23
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c10c7375f74ab554243c05e314408dbea13e8a8641869406aa8896d6072dd298
+size 286958
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/1bdae94a492ab37b5bf2fe01d3a1972c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/1bdae94a492ab37b5bf2fe01d3a1972c
new file mode 100644
index 0000000000000000000000000000000000000000..f28d59e1f9187cfd65b688b4bceb57a81f0acdcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/1bdae94a492ab37b5bf2fe01d3a1972c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/39d69086a6a3fc1928b1029bce1a4b1c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/39d69086a6a3fc1928b1029bce1a4b1c
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/39d69086a6a3fc1928b1029bce1a4b1c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/6219d52e418b76d0bd0ac983e9681549 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/6219d52e418b76d0bd0ac983e9681549
new file mode 100644
index 0000000000000000000000000000000000000000..f79d297d6897715699148323a3065b1f0da5db36
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/6219d52e418b76d0bd0ac983e9681549 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/b6b43343b87c528bece6edba6cb8178c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/b6b43343b87c528bece6edba6cb8178c
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/b6b43343b87c528bece6edba6cb8178c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/b832e34f5dfdf9ada9a8733b7623033a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/b832e34f5dfdf9ada9a8733b7623033a
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/b832e34f5dfdf9ada9a8733b7623033a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/c3e12ffab5bb96027f033a83aa2eb624 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/c3e12ffab5bb96027f033a83aa2eb624
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/c3e12ffab5bb96027f033a83aa2eb624 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/d448888de4399f125355c76102f063d3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/d448888de4399f125355c76102f063d3
new file mode 100644
index 0000000000000000000000000000000000000000..e1206a75cdede11c667b054a72b8e02b8defb834
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/d448888de4399f125355c76102f063d3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/fbd878740d6fedada801f081db8dc3a1 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/fbd878740d6fedada801f081db8dc3a1
new file mode 100644
index 0000000000000000000000000000000000000000..f47633d39c4fd96c5a5721185fa885db5d529bb2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/fbd878740d6fedada801f081db8dc3a1
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:18d58f90a069778d2b626efa6e856fbd59d12b5745a93013f486ab63f0af231f
+size 448964
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f6ddec24b29315cabc92ea9c23a37282d690ebaf
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..b720842c3efc597b90e4dee9579e7787bedd61a7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584182127883214, "commit_timestamp_nsecs": 1754584182276188915, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/d/4b57cd8568242eb4f14d80ec9a7ffeac b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/d/4b57cd8568242eb4f14d80ec9a7ffeac
new file mode 100644
index 0000000000000000000000000000000000000000..9f047f3e2b072473d3be0cb5ca35cc646a5e0b65
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/d/4b57cd8568242eb4f14d80ec9a7ffeac differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9397397311e3fe2c78b3182f636f96509c398429
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/71a558a3239791335336ae952c710ed0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/71a558a3239791335336ae952c710ed0
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/71a558a3239791335336ae952c710ed0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/74bf33e1eaceb55589439f3e5d7ecdbc b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/74bf33e1eaceb55589439f3e5d7ecdbc
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/74bf33e1eaceb55589439f3e5d7ecdbc differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/7c7e99e756eb3213eedb1098bfa95440 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/7c7e99e756eb3213eedb1098bfa95440
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/7c7e99e756eb3213eedb1098bfa95440 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/8524cf83fdb94d1b46f9f5a665800f23 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/8524cf83fdb94d1b46f9f5a665800f23
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/8524cf83fdb94d1b46f9f5a665800f23 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/8bb85d1461a16a590bbab46ae1f98d14 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/8bb85d1461a16a590bbab46ae1f98d14
new file mode 100644
index 0000000000000000000000000000000000000000..38ff998957cda6e5d40e622d05e46d6514c17922
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/8bb85d1461a16a590bbab46ae1f98d14
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:69dfe3f9611f1bdd621f2a1e6354e689c20ea8c804b5e4223a325f610273df68
+size 193584
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/b5e44b88aa6e9804b148436affd5ecf5 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/b5e44b88aa6e9804b148436affd5ecf5
new file mode 100644
index 0000000000000000000000000000000000000000..cdd7a22b633e1f9a14ea504e699d37b9aaec0937
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/b5e44b88aa6e9804b148436affd5ecf5
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c89fe9aab5d89d1291c4d7420b965849f7c2c73e6c89a0082b412975ddbf3988
+size 544572
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/ecbebc54511139ebbc66a2765548c161 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/ecbebc54511139ebbc66a2765548c161
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/ecbebc54511139ebbc66a2765548c161 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/f130c5ef5c9f3d78735f6f61b569306a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/f130c5ef5c9f3d78735f6f61b569306a
new file mode 100644
index 0000000000000000000000000000000000000000..60e6f779b419da550008fa71e0ab20b3555e05b7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/f130c5ef5c9f3d78735f6f61b569306a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..817f681c3f5fec757bbc51014b92511716b06b5e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..8c3bf3cb196f6344a2f922d685fd46b6289940e9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584215356348373, "commit_timestamp_nsecs": 1754584215533787719, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/d/4684968b2782af44ef06ca1adf29feec b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/d/4684968b2782af44ef06ca1adf29feec
new file mode 100644
index 0000000000000000000000000000000000000000..6666f4704ebf45eb203c42ec4884437908a09154
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/d/4684968b2782af44ef06ca1adf29feec differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..006ef529e3947dbdf3a0fb5816fae1a43c2a6d34
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/297fed5df95534414b1cf481374d5819 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/297fed5df95534414b1cf481374d5819
new file mode 100644
index 0000000000000000000000000000000000000000..0d9e818e0160aba2557225062f6baf3b3cd903cd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/297fed5df95534414b1cf481374d5819 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/336f05f82f76880eb7d08681e73222ff b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/336f05f82f76880eb7d08681e73222ff
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/336f05f82f76880eb7d08681e73222ff differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/40e8f1a78d4d98da310e0879cd2a2c78 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/40e8f1a78d4d98da310e0879cd2a2c78
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/40e8f1a78d4d98da310e0879cd2a2c78 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/62067cf68ac17b9bd1124e8d3353a577 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/62067cf68ac17b9bd1124e8d3353a577
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/62067cf68ac17b9bd1124e8d3353a577 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/68b0aff4bdf148af8d01cad0da04226c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/68b0aff4bdf148af8d01cad0da04226c
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/68b0aff4bdf148af8d01cad0da04226c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/8c606b355aa358ab9f3dd4cb9bca2011 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/8c606b355aa358ab9f3dd4cb9bca2011
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/8c606b355aa358ab9f3dd4cb9bca2011 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/b3b2ff0bbdbb1fff35e6674ff00c4ce0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/b3b2ff0bbdbb1fff35e6674ff00c4ce0
new file mode 100644
index 0000000000000000000000000000000000000000..dbb8100aa958c25cc6227ac0f782173623d66323
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/b3b2ff0bbdbb1fff35e6674ff00c4ce0
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7ba89488547f8a18d778fc1a15cd7b7b85b6fde7ea502906690e5a995b880b4b
+size 295792
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/eca0aa7edd75e21b35d6d3806c7224c0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/eca0aa7edd75e21b35d6d3806c7224c0
new file mode 100644
index 0000000000000000000000000000000000000000..6463ec8509c312ad365b754e2f93de05fabfedbd
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/eca0aa7edd75e21b35d6d3806c7224c0
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:480e83620f60fe6706dbaaa75fefb133a1cc19c46f0d33198ea3c75bfa00ac87
+size 404475
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d1ebf166231f4caffc275e7e8d4980cd42970086
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..150b21d1d12591ba7f7b7efbb2111dd154c28ea0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584249384683981, "commit_timestamp_nsecs": 1754584249550207912, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/d/13a46b24b8dab5a0353045800fc8ea33 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/d/13a46b24b8dab5a0353045800fc8ea33
new file mode 100644
index 0000000000000000000000000000000000000000..26342d0b32cd19d6c1d70e6cdeb5b0300989ac68
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/d/13a46b24b8dab5a0353045800fc8ea33 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8ac7aa284e78be6b78d6d914b7ddb084fb8d177a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/08aabafd0c288fd5491a309f92d94d05 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/08aabafd0c288fd5491a309f92d94d05
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/08aabafd0c288fd5491a309f92d94d05 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/36d64b94bb5dc9d39c59c67bc81185ab b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/36d64b94bb5dc9d39c59c67bc81185ab
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/36d64b94bb5dc9d39c59c67bc81185ab differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/407e8c652f327b4a5324711eeb83fa02 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/407e8c652f327b4a5324711eeb83fa02
new file mode 100644
index 0000000000000000000000000000000000000000..13fd8b24e331af5052917bfd9c71b0cba89af43c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/407e8c652f327b4a5324711eeb83fa02
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9445546e1b4e0f9faa2aa8a9adf3e33ab4a101c4664206ab5befcd1381def627
+size 442187
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/591e13551b6e2e57432891f727c34b9f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/591e13551b6e2e57432891f727c34b9f
new file mode 100644
index 0000000000000000000000000000000000000000..211c9e7270a9b117f9597071d8fa782a3398b3ff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/591e13551b6e2e57432891f727c34b9f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/63a271fa2e3e10858163503eeeed8ed1 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/63a271fa2e3e10858163503eeeed8ed1
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/63a271fa2e3e10858163503eeeed8ed1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/6b8f9277eab502f3ad0994f51599fffd b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/6b8f9277eab502f3ad0994f51599fffd
new file mode 100644
index 0000000000000000000000000000000000000000..18254ff897afacdfb60db5737c5e90729073b3bf
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/6b8f9277eab502f3ad0994f51599fffd
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:43e90a7f98156b88008bbb6058ae54afa8a142d9551c77203f8ece65d171e103
+size 295852
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/8cdf3c8a98ff6913dd83d18adc67740e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/8cdf3c8a98ff6913dd83d18adc67740e
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/8cdf3c8a98ff6913dd83d18adc67740e differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9e0411fe5c8677f71f4bf885388ced5c04c48e87
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..bd770ad44dfdbda9ab2b865f9065c0c4df9e381e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584282575121785, "commit_timestamp_nsecs": 1754584282747983597, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/d/584082fd577f7ba818034eb4c2f1b097 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/d/584082fd577f7ba818034eb4c2f1b097
new file mode 100644
index 0000000000000000000000000000000000000000..42234a58396ffcecf5943327f91d4618da2ab369
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/d/584082fd577f7ba818034eb4c2f1b097 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d0edd3b20e53dc5e8bf7ebec34c53a97a99bb562
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/173362441822ab0bf9871d6ee6436ae8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/173362441822ab0bf9871d6ee6436ae8
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/173362441822ab0bf9871d6ee6436ae8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/177dec714e5d9ed05c4504bfe9a8f1c4 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/177dec714e5d9ed05c4504bfe9a8f1c4
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/177dec714e5d9ed05c4504bfe9a8f1c4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/32812f52d4befa83c9efc1bb1c5c7d64 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/32812f52d4befa83c9efc1bb1c5c7d64
new file mode 100644
index 0000000000000000000000000000000000000000..73e21d20ec0cc8b5bbe5e599f7d8b5040f9d35c9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/32812f52d4befa83c9efc1bb1c5c7d64 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/79d76a28589bb5a60e23481669eae86b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/79d76a28589bb5a60e23481669eae86b
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/79d76a28589bb5a60e23481669eae86b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/8531ce33fc56db52c28513eb6b762784 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/8531ce33fc56db52c28513eb6b762784
new file mode 100644
index 0000000000000000000000000000000000000000..ac8d9328ee08d8b00ca133b1c48648ee8b6d1f53
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/8531ce33fc56db52c28513eb6b762784
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ab4da073d8aa5fafd7fac15d2f53135f04a3303c36b18ccdecbf05ee9dc78bc2
+size 493566
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/87807d9ce709b91cdcb059b858ce4d9f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/87807d9ce709b91cdcb059b858ce4d9f
new file mode 100644
index 0000000000000000000000000000000000000000..53d9cc04309ceae853027892ec6f8bf8064bede0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/87807d9ce709b91cdcb059b858ce4d9f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:467fc4a86b9d6ddd7cd026334e664ebfe8c5a2a6d5a4eacdf9f5c1b6f00cffac
+size 244816
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/b621a4af7246f0219793a4da13bb2f70 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/b621a4af7246f0219793a4da13bb2f70
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/b621a4af7246f0219793a4da13bb2f70 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/c0b4937053ed29b36d31ff0d568ee926 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/c0b4937053ed29b36d31ff0d568ee926
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/c0b4937053ed29b36d31ff0d568ee926 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0474d7716a1b1cbd8e7f0eae11590f36d09a62f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..a1d4d043fe06fa3891d7baf05e9d6a9ea784da0f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584314892372108, "commit_timestamp_nsecs": 1754584315057029728, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/d/8314edee479d9a3efbdfe1b618dfdc37 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/d/8314edee479d9a3efbdfe1b618dfdc37
new file mode 100644
index 0000000000000000000000000000000000000000..b9f015ae0b2a133c80ebf978795413a6df724582
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/d/8314edee479d9a3efbdfe1b618dfdc37 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..fea44dde214633a778e8213d95709941c9ffb19f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/0c3169c6edefdf6593b76d08a99f8973 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/0c3169c6edefdf6593b76d08a99f8973
new file mode 100644
index 0000000000000000000000000000000000000000..6928b4aee31265340d4f55bf8633fb3fc8abb5ae
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/0c3169c6edefdf6593b76d08a99f8973 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/11364cc9854dac3e37c7513087c74350 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/11364cc9854dac3e37c7513087c74350
new file mode 100644
index 0000000000000000000000000000000000000000..5b742183249f83200f5bbdc6eb02ab280129e9ad
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/11364cc9854dac3e37c7513087c74350
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:88a8450d253859ac9c451ad738c20f47e9e7008fba2da353fe67606b241e2c38
+size 442126
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/2d0581141953bdc0fcb05c27457bee37 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/2d0581141953bdc0fcb05c27457bee37
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/2d0581141953bdc0fcb05c27457bee37 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/474df3e314a348c8e2a207ec8f760986 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/474df3e314a348c8e2a207ec8f760986
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/474df3e314a348c8e2a207ec8f760986 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/8edc120237220bbeda09bb365c6ecd86 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/8edc120237220bbeda09bb365c6ecd86
new file mode 100644
index 0000000000000000000000000000000000000000..4a4ea3f2f01d74d501ff7ebc406e963e95b80d68
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/8edc120237220bbeda09bb365c6ecd86
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dc30c17f6891271d233b174304e5515c9f7f4c515f5346515b24e4decd11b879
+size 295908
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/93521f5e92583a38ad4beba83b481952 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/93521f5e92583a38ad4beba83b481952
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/93521f5e92583a38ad4beba83b481952 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/a805c4fe05dbde02ace66056333454c6 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/a805c4fe05dbde02ace66056333454c6
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/a805c4fe05dbde02ace66056333454c6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3874eae78d20f45ee3bd2b783b0bdfe21b3f9434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..592848f9b995f386937df4243adef98619505a48
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584348120244646, "commit_timestamp_nsecs": 1754584348258283278, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/d/e5c23a24dac811f28aed354a848ddc36 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/d/e5c23a24dac811f28aed354a848ddc36
new file mode 100644
index 0000000000000000000000000000000000000000..cfe922d5d4fcaa9455a39d2eb58bcb8bdf48ca75
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/d/e5c23a24dac811f28aed354a848ddc36 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0bc10bec97c2480dcf76763eb5f62bd7af092b1f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/22ce06dae08fd30c3fb3e8d65283f312 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/22ce06dae08fd30c3fb3e8d65283f312
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/22ce06dae08fd30c3fb3e8d65283f312 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/45ed77cb804d46164077abbe4038a7c5 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/45ed77cb804d46164077abbe4038a7c5
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/45ed77cb804d46164077abbe4038a7c5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/788e3100e6764596a94ed8126f63a76f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/788e3100e6764596a94ed8126f63a76f
new file mode 100644
index 0000000000000000000000000000000000000000..cb5368439de8bcf64910570932b3ffbf3890a735
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/788e3100e6764596a94ed8126f63a76f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:595393cbd3ee773ef51604a2ea59fa50d0058a65eb4ba11d2c3bb2415fd2d961
+size 735630
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/9a323432dccdfe7d2481b8e7a6b12da4 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/9a323432dccdfe7d2481b8e7a6b12da4
new file mode 100644
index 0000000000000000000000000000000000000000..2910fbc6db2094404ed932c7d9bcdb23beb99906
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/9a323432dccdfe7d2481b8e7a6b12da4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/b9901329f0e950418a05fe0e6e348512 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/b9901329f0e950418a05fe0e6e348512
new file mode 100644
index 0000000000000000000000000000000000000000..462f2f6bd70276b1baa382093ed36be495eb745b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/b9901329f0e950418a05fe0e6e348512 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/d620e2ce78e0d56a0a2d64a66cb33386 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/d620e2ce78e0d56a0a2d64a66cb33386
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/d620e2ce78e0d56a0a2d64a66cb33386 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/f1acb7aa34185dc1d75c0e7b5f3e8e5a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/f1acb7aa34185dc1d75c0e7b5f3e8e5a
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/f1acb7aa34185dc1d75c0e7b5f3e8e5a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..ab49b8a2e94c159824644dd915e5f0b0bae33d9d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..285dde0bc75d8db9b4cb8e527ec7f2a5a5a4c041
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584382863867029, "commit_timestamp_nsecs": 1754584383036672095, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/d/7dda3ec84aaf83911934a2c2b69592bd b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/d/7dda3ec84aaf83911934a2c2b69592bd
new file mode 100644
index 0000000000000000000000000000000000000000..26f161c74af3ae86a225c7a10da3adaea3aad893
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/d/7dda3ec84aaf83911934a2c2b69592bd differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f7d866a80eb4d5700f07058733caee6e7168c422
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/1fc792643ede2e5402d43f90532b4c1c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/1fc792643ede2e5402d43f90532b4c1c
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/1fc792643ede2e5402d43f90532b4c1c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/40c2864c0da38497c09a78b0faa3bf61 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/40c2864c0da38497c09a78b0faa3bf61
new file mode 100644
index 0000000000000000000000000000000000000000..dae72081ebd73bdb7df4d5c35719daebd2cb6135
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/40c2864c0da38497c09a78b0faa3bf61 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/4435c7c639b69dc8fb7972ae695f2e50 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/4435c7c639b69dc8fb7972ae695f2e50
new file mode 100644
index 0000000000000000000000000000000000000000..9db2bba766bd82f030a6f850c920b9b8ade3f936
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/4435c7c639b69dc8fb7972ae695f2e50
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a22743dfc64dbc47dcf342bc6ad0d692909228f0a4ba2f96fddc78632633034b
+size 450991
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/6090064c83810976986d0719d58e5c1a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/6090064c83810976986d0719d58e5c1a
new file mode 100644
index 0000000000000000000000000000000000000000..4859214e6b68fa6e697de9748417fcf43be8081d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/6090064c83810976986d0719d58e5c1a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/69414957e851803c85b68aced0bec3d4 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/69414957e851803c85b68aced0bec3d4
new file mode 100644
index 0000000000000000000000000000000000000000..8d501107f5fd5a52bc4654452a2102fdb24faf92
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/69414957e851803c85b68aced0bec3d4
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:76613a7c034d2fd82b676fc765a240d1f7adb83e2b0dad94e187f0485574bacf
+size 287263
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/7ad5c81a032b6d3c45b0d1820ff6228f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/7ad5c81a032b6d3c45b0d1820ff6228f
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/7ad5c81a032b6d3c45b0d1820ff6228f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/7b9256f1c5bbfb9a400d936d1f689408 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/7b9256f1c5bbfb9a400d936d1f689408
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/7b9256f1c5bbfb9a400d936d1f689408 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/ae63e164430deb15495be55d48d87107 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/ae63e164430deb15495be55d48d87107
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/ae63e164430deb15495be55d48d87107 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/fc04ca13c61357e76c9ee0d73e85e8d5 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/fc04ca13c61357e76c9ee0d73e85e8d5
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/fc04ca13c61357e76c9ee0d73e85e8d5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..987ed3bbc532f56d599afe574cdac6a012ab1e83
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..1d1bb31d5124da9f738b2f4e07b228b1361bdc36
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584415492755304, "commit_timestamp_nsecs": 1754584415663662520, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/d/2ebe4140194ece99ccdd718e1319d5e0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/d/2ebe4140194ece99ccdd718e1319d5e0
new file mode 100644
index 0000000000000000000000000000000000000000..c35216b1387e488596862c321d4093e3b05e550a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/d/2ebe4140194ece99ccdd718e1319d5e0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..7bde25a8afd587c1b22cb8be9203648a55ca91ee
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/040f2e79748b040023dcedbe0a6e18d1 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/040f2e79748b040023dcedbe0a6e18d1
new file mode 100644
index 0000000000000000000000000000000000000000..86ae374519ad46687d0a7f64c4909bbef4eebeab
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/040f2e79748b040023dcedbe0a6e18d1
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aa3431de95aaf1d40ccd9be16d3ab5c191e561b087a3a617084514421f6c8d6d
+size 408173
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/0e9574abd50733c35cae599991d8b522 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/0e9574abd50733c35cae599991d8b522
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/0e9574abd50733c35cae599991d8b522 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/146b90889bcef8073e60f7c72ae34116 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/146b90889bcef8073e60f7c72ae34116
new file mode 100644
index 0000000000000000000000000000000000000000..7c1b5edee81ae51d4b6e1f3d8d0104c1ecb48ea9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/146b90889bcef8073e60f7c72ae34116
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9a4a227623ed2d7b2b3a43a5e932ac2aa0ced352401913c7bab7f8ab3e6da034
+size 295820
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/3ca64726a6b08d0c7c72e2934c920aa9 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/3ca64726a6b08d0c7c72e2934c920aa9
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/3ca64726a6b08d0c7c72e2934c920aa9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/6ae690ba5abd9575aba7f028469581a4 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/6ae690ba5abd9575aba7f028469581a4
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/6ae690ba5abd9575aba7f028469581a4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/9f53f5f5cb064a4c76b13bd669f90f08 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/9f53f5f5cb064a4c76b13bd669f90f08
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/9f53f5f5cb064a4c76b13bd669f90f08 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/d07ce87d4a77e53c405c72d842a15436 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/d07ce87d4a77e53c405c72d842a15436
new file mode 100644
index 0000000000000000000000000000000000000000..4af2fe7a2a6fd3690a41c8d4626c8c9cdcf28453
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/d07ce87d4a77e53c405c72d842a15436 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..66465463499a7dd9cc9e5cc931fe5ef2fe455b11
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..413eb84f49280bf3e9294f970affda4d12f62075
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584462845028253, "commit_timestamp_nsecs": 1754584463016945630, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/d/1ad7d38fa10d6214679e6d7859cbee60 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/d/1ad7d38fa10d6214679e6d7859cbee60
new file mode 100644
index 0000000000000000000000000000000000000000..8df3f8f9ca41d8cb7fed9ed6456ba503aaa0b504
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/d/1ad7d38fa10d6214679e6d7859cbee60 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5d861bfe24015dee48727208e2173751b62cd1f9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/2cfd1579a45dea476113c94d9265d101 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/2cfd1579a45dea476113c94d9265d101
new file mode 100644
index 0000000000000000000000000000000000000000..f89fec95c849dc96c8c7b2937e0509b442c6ce80
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/2cfd1579a45dea476113c94d9265d101
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:635845b628dc136aed58b246df5c276c280480a5126ad0e000c556213e0db542
+size 421763
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/3612a1996442416fe947ef3a661b650d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/3612a1996442416fe947ef3a661b650d
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/3612a1996442416fe947ef3a661b650d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/4a963933afd7968fa972067cd6e1a4e7 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/4a963933afd7968fa972067cd6e1a4e7
new file mode 100644
index 0000000000000000000000000000000000000000..327382ccd70c6c1435ecc7fbf19ac0f6c211c1e5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/4a963933afd7968fa972067cd6e1a4e7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/89c61675bd88b4fd2fe1e74281a6dbf6 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/89c61675bd88b4fd2fe1e74281a6dbf6
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/89c61675bd88b4fd2fe1e74281a6dbf6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/d2c0d68f44dd58c788dc3ab5e1693f3f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/d2c0d68f44dd58c788dc3ab5e1693f3f
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/d2c0d68f44dd58c788dc3ab5e1693f3f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/d82f71cd77d69530f15518baeddb49d2 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/d82f71cd77d69530f15518baeddb49d2
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/d82f71cd77d69530f15518baeddb49d2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/dfbb8e68a79b8f6fce16ddaa206092cf b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/dfbb8e68a79b8f6fce16ddaa206092cf
new file mode 100644
index 0000000000000000000000000000000000000000..f92686caa6fbe6cef504d3ce71fb1f2d60b1eadf
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/dfbb8e68a79b8f6fce16ddaa206092cf
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:880a97f5043dc4200bb731f264465bf22fbda106b7e5418a4c1e35b46c9c1d48
+size 278716
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/e85708f6fe1359518d16e244aba1a4d5 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/e85708f6fe1359518d16e244aba1a4d5
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/e85708f6fe1359518d16e244aba1a4d5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..db8f23bbe3cf3f2a4e9d76247f492b3ad3b6a0f6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..37205d3241db973e934191b4edb9ffbcad983d5a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584495884124710, "commit_timestamp_nsecs": 1754584496050190224, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/d/3c2bb11608ae7c323c02c6a8a8add708 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/d/3c2bb11608ae7c323c02c6a8a8add708
new file mode 100644
index 0000000000000000000000000000000000000000..bb4ad10a01259883d9bc437ed0a856202f33a3dd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/d/3c2bb11608ae7c323c02c6a8a8add708 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b08c05a52318c1a3910072ba645a31013b19f0a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/3a851ac0660633e16835349166d2ae94 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/3a851ac0660633e16835349166d2ae94
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/3a851ac0660633e16835349166d2ae94 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/6790aa479a7735fea90cc3be85de5cf3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/6790aa479a7735fea90cc3be85de5cf3
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/6790aa479a7735fea90cc3be85de5cf3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/680b88edfb00978daa90f1787c9ab2d6 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/680b88edfb00978daa90f1787c9ab2d6
new file mode 100644
index 0000000000000000000000000000000000000000..a405622cc4f58f0cda26fc41da2d083361ff1f40
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/680b88edfb00978daa90f1787c9ab2d6
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3459a6f6c5110598ee3115ea30f03e8a3aa4a49289ddd502bd89162bfba8444f
+size 340212
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/907dfde75583be2cdfec419eb1319969 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/907dfde75583be2cdfec419eb1319969
new file mode 100644
index 0000000000000000000000000000000000000000..b81c99fa1e4f39beb0db82985157c5f1fa851e1f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/907dfde75583be2cdfec419eb1319969
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:282c050261c0374f888f1df2282bb494c1f0bab8be77e127f6b278578ea99de9
+size 397793
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/9fb71e7c93dfafddc5a88bc044fa53b7 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/9fb71e7c93dfafddc5a88bc044fa53b7
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/9fb71e7c93dfafddc5a88bc044fa53b7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/c50b614f8dc02f56fcd637a5b8c0afba b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/c50b614f8dc02f56fcd637a5b8c0afba
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/c50b614f8dc02f56fcd637a5b8c0afba differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/df53ed3f3e9d720323696cf1ea456511 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/df53ed3f3e9d720323696cf1ea456511
new file mode 100644
index 0000000000000000000000000000000000000000..96a3e863774dabe2530f6b5d8e07815eb72b0aa6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/df53ed3f3e9d720323696cf1ea456511 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..73c0cd54439e851fb9bf6215b7492a28bea6d44a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..034486145f05a922d961d50f62dede994a09a0a8
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754582997262825288, "commit_timestamp_nsecs": 1754582997449701544, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/d/a0319df5c2bd179d74c5ee25c446aa7d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/d/a0319df5c2bd179d74c5ee25c446aa7d
new file mode 100644
index 0000000000000000000000000000000000000000..3146b5b1a26c6a65e7de4cbc3c0a8e394991d4c8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/d/a0319df5c2bd179d74c5ee25c446aa7d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e2e69f7d663886220f11685475935ee08a318854
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/476ac1a4f50cc86f0a77e2e7721387b2 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/476ac1a4f50cc86f0a77e2e7721387b2
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/476ac1a4f50cc86f0a77e2e7721387b2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/65eef76c240e89beb264fc811f1dc085 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/65eef76c240e89beb264fc811f1dc085
new file mode 100644
index 0000000000000000000000000000000000000000..6921cdfa7ecebc73578d5bf96d0e83e4fa7ddae4
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/65eef76c240e89beb264fc811f1dc085
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a87fe10accbc0c124f1f6b68cf524e9a93802ba2549119e36ee2e5162ea5fd2b
+size 404448
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/7265a5b699c90078d5d9aa587e6aa57b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/7265a5b699c90078d5d9aa587e6aa57b
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/7265a5b699c90078d5d9aa587e6aa57b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/a5f3a5ceb8f13454dab46f9038e6e9ba b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/a5f3a5ceb8f13454dab46f9038e6e9ba
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/a5f3a5ceb8f13454dab46f9038e6e9ba differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/bd229c1a4b00d7206c2e80b67e8f805f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/bd229c1a4b00d7206c2e80b67e8f805f
new file mode 100644
index 0000000000000000000000000000000000000000..24c048d73df1ca7e759917c7d27a904095d34a67
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/bd229c1a4b00d7206c2e80b67e8f805f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/d703d08cd6d9987493e7b86366c6e3f1 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/d703d08cd6d9987493e7b86366c6e3f1
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/d703d08cd6d9987493e7b86366c6e3f1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/d9edba30a28ad755640e4f35997b0abd b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/d9edba30a28ad755640e4f35997b0abd
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/d9edba30a28ad755640e4f35997b0abd differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/dfd28b1e061576dd27309906776c6bb0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/dfd28b1e061576dd27309906776c6bb0
new file mode 100644
index 0000000000000000000000000000000000000000..8c5f6fb8505bd8713b24c673c5f231c06d560e50
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/dfd28b1e061576dd27309906776c6bb0
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:260f2d080a4cdbe370826dbd52b70c56f9285fbb8616aee5c1ea5509b5095c02
+size 295586
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3968e3b644d13935b092dc0a78534070e9763836
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..4be831d57ec4cf039ab891c9ee4266e006abfc26
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584529644436616, "commit_timestamp_nsecs": 1754584529787512739, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/d/5b2a7a8cff0360dc4cee48ef71c09282 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/d/5b2a7a8cff0360dc4cee48ef71c09282
new file mode 100644
index 0000000000000000000000000000000000000000..6cb8b726b08d9dab9e276b3ceec59236914ed8c2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/d/5b2a7a8cff0360dc4cee48ef71c09282 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a08c91ef620ae3ab60ae429665046ecc9450086e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/150f0de932e023746a748e94b943fcc2 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/150f0de932e023746a748e94b943fcc2
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/150f0de932e023746a748e94b943fcc2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/16672c3c651f10e1f392ce4a418bdb2d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/16672c3c651f10e1f392ce4a418bdb2d
new file mode 100644
index 0000000000000000000000000000000000000000..bf20b105f119794af5565eaa4a3dbacaaff2290b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/16672c3c651f10e1f392ce4a418bdb2d
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:22e1f1d3852af24b6f78693acb314f034e8d5ed73608803665c3c50bd2aacf64
+size 425788
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/20dba384a949bee5be0209cad7f54612 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/20dba384a949bee5be0209cad7f54612
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/20dba384a949bee5be0209cad7f54612 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/388d9d4a94469c61e821ced62eb00b2f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/388d9d4a94469c61e821ced62eb00b2f
new file mode 100644
index 0000000000000000000000000000000000000000..cc42d482cd469d4b2c235db76ef93cbca7699a7d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/388d9d4a94469c61e821ced62eb00b2f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:014cc84daf7191a451751a85722b00d069f4e8524452acc79b7de84cec214baa
+size 244831
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/430c651db184d74457017c4ab788292d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/430c651db184d74457017c4ab788292d
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/430c651db184d74457017c4ab788292d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/806526dac50980cf6b76d5403efcc2c6 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/806526dac50980cf6b76d5403efcc2c6
new file mode 100644
index 0000000000000000000000000000000000000000..ef762daab69125848e7650e6dbfa5a6759309fd5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/806526dac50980cf6b76d5403efcc2c6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/b8844d1ebfa40ffb28324eaca7b0c921 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/b8844d1ebfa40ffb28324eaca7b0c921
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/b8844d1ebfa40ffb28324eaca7b0c921 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/c7c017f80849064d32871ec9c4642114 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/c7c017f80849064d32871ec9c4642114
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/c7c017f80849064d32871ec9c4642114 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..99bf4b9a637c0816715830ef440c60cddf2b3a45
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..a4cff2e2a195a03f4db60ba5631a789e42b1a9f5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584562079945159, "commit_timestamp_nsecs": 1754584562256983977, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/d/e783bb85ecf93f8bfd89f81c36a8523f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/d/e783bb85ecf93f8bfd89f81c36a8523f
new file mode 100644
index 0000000000000000000000000000000000000000..a03304e55a19991ce255ad9910ccb6a138f9f0a6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/d/e783bb85ecf93f8bfd89f81c36a8523f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..6b635faacc4bfb76f92e3b0fc1622f191184483e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/101066679961500e3830b3499cce3a62 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/101066679961500e3830b3499cce3a62
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/101066679961500e3830b3499cce3a62 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/11f246da9aa642cf654872f0448c2faa b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/11f246da9aa642cf654872f0448c2faa
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/11f246da9aa642cf654872f0448c2faa differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/12e5713a3842ea89aae80faa5a4db522 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/12e5713a3842ea89aae80faa5a4db522
new file mode 100644
index 0000000000000000000000000000000000000000..acaaaee81808cd50ea2108e887fa681c4c843044
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/12e5713a3842ea89aae80faa5a4db522
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b460fa9325be676d6f126c12996642c89ca0f8802137782354c603f9b6209069
+size 735600
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/590ed209fbb192d84b190dc52ecaf634 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/590ed209fbb192d84b190dc52ecaf634
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/590ed209fbb192d84b190dc52ecaf634 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/7a3fc5db032c64c637d29c319f3dcfde b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/7a3fc5db032c64c637d29c319f3dcfde
new file mode 100644
index 0000000000000000000000000000000000000000..541a3e657e1f496c6b096636b465fa64e165e7ca
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/7a3fc5db032c64c637d29c319f3dcfde differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/8da0c36ada624c018ce3024c087d25f2 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/8da0c36ada624c018ce3024c087d25f2
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/8da0c36ada624c018ce3024c087d25f2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/8fb7fb36faf6d714fc6d01ec348e23e3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/8fb7fb36faf6d714fc6d01ec348e23e3
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/8fb7fb36faf6d714fc6d01ec348e23e3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..91ebb1b9e0f84f6daac41e120f15a7682d7275bd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..fb71ed24eb74ef49e71506fb8d1d6d1be448ce99
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584595156753251, "commit_timestamp_nsecs": 1754584595336380639, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/d/e56c77cfa8d6620f2b8d74f4c91d0bdd b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/d/e56c77cfa8d6620f2b8d74f4c91d0bdd
new file mode 100644
index 0000000000000000000000000000000000000000..bb61acd53a5573fbea57c78545f548b163c06ce6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/d/e56c77cfa8d6620f2b8d74f4c91d0bdd differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..655ad9d41e2188e5842eca3414946371bf0f44e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/07fd76a341922a8691b38bef97d1551d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/07fd76a341922a8691b38bef97d1551d
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/07fd76a341922a8691b38bef97d1551d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/49198107e3fff2faab54082b464a6c90 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/49198107e3fff2faab54082b464a6c90
new file mode 100644
index 0000000000000000000000000000000000000000..a295b729be6a4d3bfa55afca5d3ece89b760984a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/49198107e3fff2faab54082b464a6c90
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1f79ecaf601c4caf24a334eba48f21871391eed8f31051f1960c6e53c46d43aa
+size 355370
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/5d089bd95138e5249b5afe12e51a094b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/5d089bd95138e5249b5afe12e51a094b
new file mode 100644
index 0000000000000000000000000000000000000000..9e081813b06e4d8f7d2d7485eb5f3f6862235a98
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/5d089bd95138e5249b5afe12e51a094b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/6d9b6d176a592eda5eee3fee623c152d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/6d9b6d176a592eda5eee3fee623c152d
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/6d9b6d176a592eda5eee3fee623c152d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/b11e71857d8ec1418f7b3c1de810f193 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/b11e71857d8ec1418f7b3c1de810f193
new file mode 100644
index 0000000000000000000000000000000000000000..85f35e7991cac7b6fc244bb2f277438119f4e90b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/b11e71857d8ec1418f7b3c1de810f193 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/e2e49bdc567282e08d7e1d3fc2ada7ec b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/e2e49bdc567282e08d7e1d3fc2ada7ec
new file mode 100644
index 0000000000000000000000000000000000000000..e164d842815ddf4656aea77b9aa4e13e01f85da6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/e2e49bdc567282e08d7e1d3fc2ada7ec
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bafb66583281aab377014019594f6af4ecc422f08f44640c387b15d57e569548
+size 348995
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/f12ecfe99eb47debf9b88a6065579445 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/f12ecfe99eb47debf9b88a6065579445
new file mode 100644
index 0000000000000000000000000000000000000000..65529f6532287fad4b450ca493011b55e8bf7f92
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/f12ecfe99eb47debf9b88a6065579445 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f0898d18b4fecf1dcdd70a20c9e97efb44e5e2c5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..58a26b4e92b9e4eecc31e52b4376dfc2621de2fb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584629574574766, "commit_timestamp_nsecs": 1754584629742728266, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/d/df6128cef191d10a65ab2d8b1fc7d403 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/d/df6128cef191d10a65ab2d8b1fc7d403
new file mode 100644
index 0000000000000000000000000000000000000000..bacaef184e12f3c9d46cc1b783471ce6a8f6c5ef
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/d/df6128cef191d10a65ab2d8b1fc7d403 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c0925874563cce4ee0c92eccc2ff1fd0fb75a5bd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/0ac01023d23c553503c7301a1129d7b6 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/0ac01023d23c553503c7301a1129d7b6
new file mode 100644
index 0000000000000000000000000000000000000000..0cc44816435f9ce67fec735920cc46d9f753797d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/0ac01023d23c553503c7301a1129d7b6
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:79c8918a8b895f63709dd4341472024e30cf91ed9e92405f26c5615910a23af0
+size 459398
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/6ad34ff10984e3e80175e7ee7fdb6995 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/6ad34ff10984e3e80175e7ee7fdb6995
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/6ad34ff10984e3e80175e7ee7fdb6995 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/8f095176b63a8e5dbe6b41eab90ab798 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/8f095176b63a8e5dbe6b41eab90ab798
new file mode 100644
index 0000000000000000000000000000000000000000..ddf67feb2110a61e44da2ee43724eba326848881
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/8f095176b63a8e5dbe6b41eab90ab798 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/bf26eb40f420990716d0854b51fcd64c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/bf26eb40f420990716d0854b51fcd64c
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/bf26eb40f420990716d0854b51fcd64c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/c7ba373f3d6c69d72bdc2ee5cbcf7894 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/c7ba373f3d6c69d72bdc2ee5cbcf7894
new file mode 100644
index 0000000000000000000000000000000000000000..85f35e7991cac7b6fc244bb2f277438119f4e90b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/c7ba373f3d6c69d72bdc2ee5cbcf7894 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/efa97b93b6f2f45e6909b48181b12627 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/efa97b93b6f2f45e6909b48181b12627
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/efa97b93b6f2f45e6909b48181b12627 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/fdb427fe39a700b8a1b559d8fac68184 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/fdb427fe39a700b8a1b559d8fac68184
new file mode 100644
index 0000000000000000000000000000000000000000..647355680603051ba6228b45ea1474d53972236a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/fdb427fe39a700b8a1b559d8fac68184
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:764b8c03bc3916995a8ef7974daf6135dc520c65536c6bb6d2493185841dd81c
+size 244839
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3227c8d216570a3ee870172df39e426f83c1bbf4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..053066062a367a138fb35aa2e959bdd67c36e159
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584663174141438, "commit_timestamp_nsecs": 1754584663348588365, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/d/a7eb49d1bba5c054b089f3bbbae5ec68 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/d/a7eb49d1bba5c054b089f3bbbae5ec68
new file mode 100644
index 0000000000000000000000000000000000000000..ae79bf82296aed2191043291a5f01dcb3f699a67
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/d/a7eb49d1bba5c054b089f3bbbae5ec68 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0601ef1f640a7b8f38cbacaafb779f16656b4d29
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/052a262838694fad8dbe32b93bce478c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/052a262838694fad8dbe32b93bce478c
new file mode 100644
index 0000000000000000000000000000000000000000..6096f5bc55eb67725a3acdf5fdebadaa06453cd4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/052a262838694fad8dbe32b93bce478c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/2fe8f4a21d19fc19a8ffea69e3b1a3b7 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/2fe8f4a21d19fc19a8ffea69e3b1a3b7
new file mode 100644
index 0000000000000000000000000000000000000000..85f35e7991cac7b6fc244bb2f277438119f4e90b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/2fe8f4a21d19fc19a8ffea69e3b1a3b7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/31610d9ceca48fdade9c23dcd7efe924 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/31610d9ceca48fdade9c23dcd7efe924
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/31610d9ceca48fdade9c23dcd7efe924 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/4081a4ef9d4d1762c9ccf509a01dd7cb b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/4081a4ef9d4d1762c9ccf509a01dd7cb
new file mode 100644
index 0000000000000000000000000000000000000000..f07c8445e603f7b25d8d92aa5d244d1f14c4288b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/4081a4ef9d4d1762c9ccf509a01dd7cb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0d3e563eb72e0dc6957ee34dae693e4c05372f37a59a5ec7d520f93ee4671e7e
+size 329950
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/9aec37f0dd8ddfc10c4dbd7718ee426e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/9aec37f0dd8ddfc10c4dbd7718ee426e
new file mode 100644
index 0000000000000000000000000000000000000000..5e9da0f07f8f917b3350ee6a4da9aa45a0390b8f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/9aec37f0dd8ddfc10c4dbd7718ee426e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d665c830d3aa1f6a0c02ac85dc6ac578f72467ded54db1ee47435d5635ccadd1
+size 408615
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/bc3618144d002ecf022e0b8ab9f1430c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/bc3618144d002ecf022e0b8ab9f1430c
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/bc3618144d002ecf022e0b8ab9f1430c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/cb1d3b2688d911472698ca1b346799ea b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/cb1d3b2688d911472698ca1b346799ea
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/cb1d3b2688d911472698ca1b346799ea differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3ae34e701d57d3f2b093ee44c30ce87aaee2ad3c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..303c398e4491d1cf1a877ea442134681e8db8fc5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584696221959758, "commit_timestamp_nsecs": 1754584696374454201, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/d/82a2980f3b5cbba961ebc4c79ee7393c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/d/82a2980f3b5cbba961ebc4c79ee7393c
new file mode 100644
index 0000000000000000000000000000000000000000..9ebb6a51dc78912678e4b2dacaf2c538b4fe9d33
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/d/82a2980f3b5cbba961ebc4c79ee7393c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..516124612a7615390a9c81947f863cb36bc570b1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/23ede4b1c9da2d84776996b8484838ef b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/23ede4b1c9da2d84776996b8484838ef
new file mode 100644
index 0000000000000000000000000000000000000000..ec745d97353f97209ef50f0eb738585830219f53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/23ede4b1c9da2d84776996b8484838ef differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/2d834ea37663a33f0548f2a4ea8e4465 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/2d834ea37663a33f0548f2a4ea8e4465
new file mode 100644
index 0000000000000000000000000000000000000000..85f35e7991cac7b6fc244bb2f277438119f4e90b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/2d834ea37663a33f0548f2a4ea8e4465 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/68115412b4f9db509254504cbd38ab19 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/68115412b4f9db509254504cbd38ab19
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/68115412b4f9db509254504cbd38ab19 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/83adbc50b9d75f8ddb92e8cf18c1debc b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/83adbc50b9d75f8ddb92e8cf18c1debc
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/83adbc50b9d75f8ddb92e8cf18c1debc differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/abd79adc576a29bf9198ba0cc0458cde b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/abd79adc576a29bf9198ba0cc0458cde
new file mode 100644
index 0000000000000000000000000000000000000000..b0127231b6b88c87a8615f32d238741d6a1e4e96
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/abd79adc576a29bf9198ba0cc0458cde
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9f95a559b06ee675fcd6a1138c2112cbf83b09b699f9bb2ebc2e6d41550d73e0
+size 355409
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/d64dd38e10433aed24d1914b60b6b775 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/d64dd38e10433aed24d1914b60b6b775
new file mode 100644
index 0000000000000000000000000000000000000000..0e715773ed7e94ed7621c86f41f503e5053ecb94
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/d64dd38e10433aed24d1914b60b6b775
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:91e9cd3a671e37c8b60e6a927fd95d4322df7b115dc475095e26a9a1cf267dff
+size 345098
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/f1380a20dce77f6294e58380afd9b5eb b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/f1380a20dce77f6294e58380afd9b5eb
new file mode 100644
index 0000000000000000000000000000000000000000..ed5766459a18c8b2d04988666f2b1721c1232f19
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/f1380a20dce77f6294e58380afd9b5eb differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9b930c209a9868740529b5468c2be3b7b096fa62
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..0130b420ada206d6c89e924d9af9192b1459c88c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584729645104027, "commit_timestamp_nsecs": 1754584729835414729, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/d/11ca3dd54ebff627014b2a3f7154fa51 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/d/11ca3dd54ebff627014b2a3f7154fa51
new file mode 100644
index 0000000000000000000000000000000000000000..1a2a7021f32e0d332a77e9d1a07bd68ea39f1c76
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/d/11ca3dd54ebff627014b2a3f7154fa51 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..442aed9763321130061e4cab4dfa8b494d252cab
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/033750d3c263ab10112410aadc44a494 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/033750d3c263ab10112410aadc44a494
new file mode 100644
index 0000000000000000000000000000000000000000..5867cacde3eaa2786f999ce6949e5b196d4a6454
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/033750d3c263ab10112410aadc44a494 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/1511474303bfbc153ac541a0c9473508 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/1511474303bfbc153ac541a0c9473508
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/1511474303bfbc153ac541a0c9473508 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/2cb4f6839f965a868fd4b59948c394e6 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/2cb4f6839f965a868fd4b59948c394e6
new file mode 100644
index 0000000000000000000000000000000000000000..90d7f3f72cd41f5a4bedb7c66aeb040aba056378
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/2cb4f6839f965a868fd4b59948c394e6
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ea1aeffb2a6a9f9d4ccbacbb334818a003d2ff141ba80230ede99f03fc16ad0e
+size 460014
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/7337a19d48dfab4e6885bb5741666ba2 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/7337a19d48dfab4e6885bb5741666ba2
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/7337a19d48dfab4e6885bb5741666ba2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/8e2cf0bf1d23db4f792a9245002539b8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/8e2cf0bf1d23db4f792a9245002539b8
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/8e2cf0bf1d23db4f792a9245002539b8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/c107040fecfb381a706fa0104aeac496 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/c107040fecfb381a706fa0104aeac496
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/c107040fecfb381a706fa0104aeac496 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/c1089596cf6ca508340abf38b248e1a8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/c1089596cf6ca508340abf38b248e1a8
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/c1089596cf6ca508340abf38b248e1a8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/c49e63a00315a6d9a0e4c356892fe828 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/c49e63a00315a6d9a0e4c356892fe828
new file mode 100644
index 0000000000000000000000000000000000000000..552a8d8414e29e4d0ca026766fab93728f37224e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/c49e63a00315a6d9a0e4c356892fe828
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:84518af25c74d57ba02840cb94e822801bbc8d8f060968f0324a3b0daf8671cf
+size 244869
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b29d3aa2e0489b15bdc520311e1f0ce9ea5d2b42
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..b49bb56d9c9a45e3be5a43a6b75b7ea60c6a0b2c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584761571275678, "commit_timestamp_nsecs": 1754584761747190779, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/d/77a74d6abe1e7c823649b0b25e195743 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/d/77a74d6abe1e7c823649b0b25e195743
new file mode 100644
index 0000000000000000000000000000000000000000..c4875a47f420ead129efb523034a75ce3dd2f7d1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/d/77a74d6abe1e7c823649b0b25e195743 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..54d016c89137f3312dd5ffb431be74c2660ef893
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/4a6271c6fab55f2fc30a37a895fec244 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/4a6271c6fab55f2fc30a37a895fec244
new file mode 100644
index 0000000000000000000000000000000000000000..53f0a80657ba81b7e0870a3bc10c9bbbaf7fb4ba
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/4a6271c6fab55f2fc30a37a895fec244
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b003ec2ef43ebf102c8050e102e738e61af3980fcc9d367dbc1901c65ba31d6f
+size 338375
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/67e4e1eecb716043b81a23ccb535e7db b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/67e4e1eecb716043b81a23ccb535e7db
new file mode 100644
index 0000000000000000000000000000000000000000..85f35e7991cac7b6fc244bb2f277438119f4e90b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/67e4e1eecb716043b81a23ccb535e7db differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/69e68d70afb892a5ffb7dc7dcccdf60d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/69e68d70afb892a5ffb7dc7dcccdf60d
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/69e68d70afb892a5ffb7dc7dcccdf60d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/9136a4987d3d27e60af30b9666dd0d20 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/9136a4987d3d27e60af30b9666dd0d20
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/9136a4987d3d27e60af30b9666dd0d20 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/9c17eb503575eab841d67749f552c450 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/9c17eb503575eab841d67749f552c450
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/9c17eb503575eab841d67749f552c450 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/ae1e9ea4943c856c46da165dae57ae59 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/ae1e9ea4943c856c46da165dae57ae59
new file mode 100644
index 0000000000000000000000000000000000000000..2910fbc6db2094404ed932c7d9bcdb23beb99906
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/ae1e9ea4943c856c46da165dae57ae59 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/e8f618fcc8b98b95ed77de4125cde042 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/e8f618fcc8b98b95ed77de4125cde042
new file mode 100644
index 0000000000000000000000000000000000000000..4b0e0e632a0f3645a197cab1adec69469b487f67
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/e8f618fcc8b98b95ed77de4125cde042 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/f84a6c6eec17a8cb8b1432f9b63e16a8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/f84a6c6eec17a8cb8b1432f9b63e16a8
new file mode 100644
index 0000000000000000000000000000000000000000..60d43211b62d824806e2dc671bf5f73d2239b482
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/f84a6c6eec17a8cb8b1432f9b63e16a8
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e734e015f315f78affa8b29e9a7f12500b8a6cacce64a38303541827ba01c75e
+size 353593
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..70184ea3ea27a9aed51931cfa829a9b57bd68d43
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..d0db48b2658b225765a85c2f08b53088f9f43712
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584793937842230, "commit_timestamp_nsecs": 1754584794125449743, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/d/a644018d1761f0a087d0f664333525f9 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/d/a644018d1761f0a087d0f664333525f9
new file mode 100644
index 0000000000000000000000000000000000000000..49fdac3d860cb6beba8bcd8b716a932c016fbb9f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/d/a644018d1761f0a087d0f664333525f9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f63144cb92fb16f635c77df2fac1d43cbabf3368
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/51f57f8d96d54bdb50af54811a8357be b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/51f57f8d96d54bdb50af54811a8357be
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/51f57f8d96d54bdb50af54811a8357be differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/a7cc308a2147fd0655564ee505070829 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/a7cc308a2147fd0655564ee505070829
new file mode 100644
index 0000000000000000000000000000000000000000..85e4865a794cc3462da84e3b01b4981daed0637b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/a7cc308a2147fd0655564ee505070829
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7a7ee6a0d40f12cc886231e1e58cab60e754c2df38c5552962f9c40824a503f4
+size 411269
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/a99179ba74cb0f79d1b494aac81ec409 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/a99179ba74cb0f79d1b494aac81ec409
new file mode 100644
index 0000000000000000000000000000000000000000..43819d26f5597a12660fcfc29035d582538c1a27
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/a99179ba74cb0f79d1b494aac81ec409
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b4252cfef9c777abea67fb790e8a23121544c9760c82cc40b85f5b8738a0edba
+size 327673
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/b0a9f18d571990ef0b8b7ffcc1cb5dbe b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/b0a9f18d571990ef0b8b7ffcc1cb5dbe
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/b0a9f18d571990ef0b8b7ffcc1cb5dbe differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/e4d971351f77229270554f90841da881 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/e4d971351f77229270554f90841da881
new file mode 100644
index 0000000000000000000000000000000000000000..03750400c56f81a370f3982b9162a428a86e4602
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/e4d971351f77229270554f90841da881 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4e625d3ab52a8134254b2be5a3fbc785ab0bf3fe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..a0506b3151781eb0b5a3548bae169a361ec4ea20
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584828066760516, "commit_timestamp_nsecs": 1754584828248041072, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/d/2b2254a6d4cc84c0b10c78c335a13852 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/d/2b2254a6d4cc84c0b10c78c335a13852
new file mode 100644
index 0000000000000000000000000000000000000000..536a41ac7498244f611315eb834afa932ea3b2a5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/d/2b2254a6d4cc84c0b10c78c335a13852 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..ce1df569311ec88062f00308a988321aa189c0ac
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/017e932e407e419fd57c716a5dd2b879 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/017e932e407e419fd57c716a5dd2b879
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/017e932e407e419fd57c716a5dd2b879 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/0443a508cbd418ae72f1ca5873d77f72 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/0443a508cbd418ae72f1ca5873d77f72
new file mode 100644
index 0000000000000000000000000000000000000000..c0ab9f3080484ca41581c1c962a6c54b2d9f05b3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/0443a508cbd418ae72f1ca5873d77f72 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/84bffdb7dc4c817312786b24a17a7935 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/84bffdb7dc4c817312786b24a17a7935
new file mode 100644
index 0000000000000000000000000000000000000000..69d397a3eb97a17c9da27c9287f4196d2be503ef
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/84bffdb7dc4c817312786b24a17a7935 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/93d392ec1ace0cecafe2f1e392048ebc b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/93d392ec1ace0cecafe2f1e392048ebc
new file mode 100644
index 0000000000000000000000000000000000000000..6ea98c14449a57656bec1e429956bf5b03aeccd6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/93d392ec1ace0cecafe2f1e392048ebc
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:56c65a518aceff0001e58932b46e2b70c4bd21b74f094f183f658bf80ce75e8d
+size 431927
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/b6ab7b778fec1927099cbdb8fc1da3fe b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/b6ab7b778fec1927099cbdb8fc1da3fe
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/b6ab7b778fec1927099cbdb8fc1da3fe differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/bac894a308a52ee006788efa5755352a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/bac894a308a52ee006788efa5755352a
new file mode 100644
index 0000000000000000000000000000000000000000..8efa1b69e98fb40f0d5c16addc26b26b248f65e7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/bac894a308a52ee006788efa5755352a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b4e1f632481c3cd8b4ffeea8dfa4e0eee9c4b37eda1ed91032ab2ac685d39710
+size 304295
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/d084f5845655c58075ad8a63ce156a0a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/d084f5845655c58075ad8a63ce156a0a
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/d084f5845655c58075ad8a63ce156a0a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f75576e16902becaa725929e3879d1dbe87d598d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..972ff67863ecc935c413f4006beb9b6f50db125c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583030520729936, "commit_timestamp_nsecs": 1754583030671646051, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/d/94f8a97f0bd40e7153dfda9a2556e34c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/d/94f8a97f0bd40e7153dfda9a2556e34c
new file mode 100644
index 0000000000000000000000000000000000000000..e5bad58c9008d705616d08fc45534cf9e29dd8cf
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/d/94f8a97f0bd40e7153dfda9a2556e34c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f163e107c10109675433698e1af5d900884629b2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/26987c34c83fb789566585ff314f76ef b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/26987c34c83fb789566585ff314f76ef
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/26987c34c83fb789566585ff314f76ef differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/415ca51e22a5e3339ea2a65cc73a87d2 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/415ca51e22a5e3339ea2a65cc73a87d2
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/415ca51e22a5e3339ea2a65cc73a87d2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/6a61973c4ddd830c40b27e193c8de3fe b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/6a61973c4ddd830c40b27e193c8de3fe
new file mode 100644
index 0000000000000000000000000000000000000000..d544120088db914cadd43d9b844daadecc2534dc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/6a61973c4ddd830c40b27e193c8de3fe
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:334ca4c426acae73fa880de332580e1aa0eb196e22689feca50868feb7dcc214
+size 295514
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/a284601796850586f6d6c0920e10c38c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/a284601796850586f6d6c0920e10c38c
new file mode 100644
index 0000000000000000000000000000000000000000..8c157eff196f70f0fbe883ae19e226e1d8e95994
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/a284601796850586f6d6c0920e10c38c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/baf8641f20d7656c30e8509748e25b39 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/baf8641f20d7656c30e8509748e25b39
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/baf8641f20d7656c30e8509748e25b39 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/d998aa4f18f0e24578b11d3fcd2b605b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/d998aa4f18f0e24578b11d3fcd2b605b
new file mode 100644
index 0000000000000000000000000000000000000000..5e18e8abce97526880103f2b2a828f18b1f16b8b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/d998aa4f18f0e24578b11d3fcd2b605b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6fb2c6cdebd9dd9376515feb3e01deba597a551ea0cd5597ecec6335db610cb6
+size 407868
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/f4beb804ab2743456ed425690a67aafc b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/f4beb804ab2743456ed425690a67aafc
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/f4beb804ab2743456ed425690a67aafc differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..97815681d282b2cee9abea169b99fd065eb99ad2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..32b9ccc6ae12607f0d3598fabae9d863b95a8652
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584861738926743, "commit_timestamp_nsecs": 1754584861885015465, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/d/b568bac36e7b7092dd787c6f83020337 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/d/b568bac36e7b7092dd787c6f83020337
new file mode 100644
index 0000000000000000000000000000000000000000..86e77ff2e0a9ad2d51999f2fd86def9fc213ab10
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/d/b568bac36e7b7092dd787c6f83020337 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..72c7c1e5eaded237c380e8437346f6075117ef45
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/122695790ca980b6a46e0fc10a92d983 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/122695790ca980b6a46e0fc10a92d983
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/122695790ca980b6a46e0fc10a92d983 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/4e1ac76c0fcd6c1b35a91c1a64d9370c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/4e1ac76c0fcd6c1b35a91c1a64d9370c
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/4e1ac76c0fcd6c1b35a91c1a64d9370c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/5a0d46c015b816309a22b187e7aec59b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/5a0d46c015b816309a22b187e7aec59b
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/5a0d46c015b816309a22b187e7aec59b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/910746fe855eb46605584c84c94fad52 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/910746fe855eb46605584c84c94fad52
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/910746fe855eb46605584c84c94fad52 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/a76ff3dc5e07ccfa25fa1ac4adab25d4 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/a76ff3dc5e07ccfa25fa1ac4adab25d4
new file mode 100644
index 0000000000000000000000000000000000000000..c0b2a96ceab3ad427dc82e20c6c82c925aee0dbe
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/a76ff3dc5e07ccfa25fa1ac4adab25d4
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:abfe1863f65b27f99e6bf84fd9dcf44ca005a1a8154c555c6457e0a09a601829
+size 302654
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/aaf4943dd197cbafaf9122b6fbec066f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/aaf4943dd197cbafaf9122b6fbec066f
new file mode 100644
index 0000000000000000000000000000000000000000..ee3574c768211e9093f158927578ac33c0ee36ed
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/aaf4943dd197cbafaf9122b6fbec066f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/f04e1eced8eb19291002fb8930ddb501 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/f04e1eced8eb19291002fb8930ddb501
new file mode 100644
index 0000000000000000000000000000000000000000..4ef53d531fb6db6725f187214f4f8733f6f97fad
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/f04e1eced8eb19291002fb8930ddb501
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:83550b2718bd593170b21897415dd86d56e426fd8ef8a9bb58d6ba2891945ac6
+size 398027
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8ab936907ad7b4da2defcc32c5c6f0465d13554f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..4aca7ecad24392e53b9a4b69fca1bde843c11d4c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584894556693522, "commit_timestamp_nsecs": 1754584894731678319, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/d/b6e13695016ce79c869fc99b11e05beb b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/d/b6e13695016ce79c869fc99b11e05beb
new file mode 100644
index 0000000000000000000000000000000000000000..1e476906464461f915bc13543c82483e4fbd1d68
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/d/b6e13695016ce79c869fc99b11e05beb differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3d67a8d33e9c01edaf0a7c43ddebabdc8ea879be
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/636457bfbb2d1d45f4e51fb47b1ea215 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/636457bfbb2d1d45f4e51fb47b1ea215
new file mode 100644
index 0000000000000000000000000000000000000000..3a2f60756b3f120f6f871c88e2ef206accb73b4c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/636457bfbb2d1d45f4e51fb47b1ea215
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b6f527a0c9dced5cef63122138846d0e866fde8d8e07448c5c47bf9683ec7ff7
+size 397973
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/7ea075fa85103ebcea8c62f057376f3f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/7ea075fa85103ebcea8c62f057376f3f
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/7ea075fa85103ebcea8c62f057376f3f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/8b579ea105f6ad0948028101b6847f5b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/8b579ea105f6ad0948028101b6847f5b
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/8b579ea105f6ad0948028101b6847f5b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/a75d8be54fd7b47d7f8e073c2fb9aa21 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/a75d8be54fd7b47d7f8e073c2fb9aa21
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/a75d8be54fd7b47d7f8e073c2fb9aa21 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/c406a5fecf5113e1a39defb19ea4302b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/c406a5fecf5113e1a39defb19ea4302b
new file mode 100644
index 0000000000000000000000000000000000000000..e863c16c8ab1e424df7c21f82f6a170cdcb260d5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/c406a5fecf5113e1a39defb19ea4302b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8408a5a37306b08a849a710a647bfbcbf77401a28445013f3b27fdfb6c7917fb
+size 306341
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/e1514051bc80eb2d343d1b4185e2146d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/e1514051bc80eb2d343d1b4185e2146d
new file mode 100644
index 0000000000000000000000000000000000000000..1fbdf83dbb80b6e580a1fb223dc06a13a28464fa
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/e1514051bc80eb2d343d1b4185e2146d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/ed9f3e29bb5eeec7761a8b336145e854 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/ed9f3e29bb5eeec7761a8b336145e854
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/ed9f3e29bb5eeec7761a8b336145e854 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0fd164cc34ce627a4ed489235d00fa25947acb05
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..58dc954875281e9dd9770af63b60a724dd48e829
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584927787602239, "commit_timestamp_nsecs": 1754584927960661752, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/d/b9632ed6afec7442bfe616e6c4ac9cc0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/d/b9632ed6afec7442bfe616e6c4ac9cc0
new file mode 100644
index 0000000000000000000000000000000000000000..8c7abc681a72fff48407e438b31d007fbf339e49
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/d/b9632ed6afec7442bfe616e6c4ac9cc0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8802b9402e2e0bcbff3a98b34205bd46361bf6d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/0a34eb1e8a1c066a311e92ef2513cea8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/0a34eb1e8a1c066a311e92ef2513cea8
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/0a34eb1e8a1c066a311e92ef2513cea8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/2d931438cbdc66b57b565a2c1c9713d8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/2d931438cbdc66b57b565a2c1c9713d8
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/2d931438cbdc66b57b565a2c1c9713d8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/46706b9e932be3c4f91d3577162f5f62 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/46706b9e932be3c4f91d3577162f5f62
new file mode 100644
index 0000000000000000000000000000000000000000..5bdf2e9241ac8c1ea162808d17b35961c79de338
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/46706b9e932be3c4f91d3577162f5f62
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:94c355724c5125aa6a027c67cd9d3af85f4b1661a11ea065e35ca9b11d198003
+size 296022
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/8a13ce0e978a0bb8d356838a1ebebc1e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/8a13ce0e978a0bb8d356838a1ebebc1e
new file mode 100644
index 0000000000000000000000000000000000000000..865ca0e9178d7761719c67f5d9fb843745d9650b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/8a13ce0e978a0bb8d356838a1ebebc1e differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/98a082070c96c84924d39dd7740643a7 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/98a082070c96c84924d39dd7740643a7
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/98a082070c96c84924d39dd7740643a7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/c08dfacd13fcbfe36ea43b3012c44a39 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/c08dfacd13fcbfe36ea43b3012c44a39
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/c08dfacd13fcbfe36ea43b3012c44a39 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/d63ac5b40bb0b72ea2da9b87848d1fd2 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/d63ac5b40bb0b72ea2da9b87848d1fd2
new file mode 100644
index 0000000000000000000000000000000000000000..3aa7a2e06588e0b119b6c2aefc7f52849475ad6d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/d63ac5b40bb0b72ea2da9b87848d1fd2
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:502e6b4caa62bca4721493f83c8ff50f2a450928cc97d8bdaa26ea2bf19305a5
+size 404895
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3a2e84911c0fec69a829d1e021f23336e1b567a3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..41298a2e78d8fb13fe492267a1a97c01572834e9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754584961016886809, "commit_timestamp_nsecs": 1754584961160645153, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/d/02fbfe43ea9b117dbc854cd21569751a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/d/02fbfe43ea9b117dbc854cd21569751a
new file mode 100644
index 0000000000000000000000000000000000000000..b0f059da79427e353e0c37cce5021bd88bd096b4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/d/02fbfe43ea9b117dbc854cd21569751a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..151783cffd0370fd0f7ec4ba8f0f2949c6605f1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/565e798aab5717538e8f7b7648d3bee0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/565e798aab5717538e8f7b7648d3bee0
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/565e798aab5717538e8f7b7648d3bee0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/652862d98e52aeaecea9210ef71ccd60 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/652862d98e52aeaecea9210ef71ccd60
new file mode 100644
index 0000000000000000000000000000000000000000..72fd63dc7ed129fa4bb79a6c88b99818abf57d6f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/652862d98e52aeaecea9210ef71ccd60
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:87708074dbcc7d326bcd23a424db24a170db903045661e76edb20ee209a4142b
+size 329937
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/74411e54412cf91cec9fcf4496b67ca0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/74411e54412cf91cec9fcf4496b67ca0
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/74411e54412cf91cec9fcf4496b67ca0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/8732dc0381ea72ca9b08fea6c87deef9 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/8732dc0381ea72ca9b08fea6c87deef9
new file mode 100644
index 0000000000000000000000000000000000000000..a58c5766bb0b0585d50b7025f55762a47ffdd979
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/8732dc0381ea72ca9b08fea6c87deef9
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dc8b472532820aad4630c1a882897d6096ef21dd071c13adca8d2442c2b4ab52
+size 336575
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/e16fba01b7fae67aabb5e0bb4f6bfcee b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/e16fba01b7fae67aabb5e0bb4f6bfcee
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/e16fba01b7fae67aabb5e0bb4f6bfcee differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/f3d98510524171e0d97f1ba97c87107b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/f3d98510524171e0d97f1ba97c87107b
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/f3d98510524171e0d97f1ba97c87107b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/f68ef10a1bd1b9e2aa57ac81d6f37235 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/f68ef10a1bd1b9e2aa57ac81d6f37235
new file mode 100644
index 0000000000000000000000000000000000000000..30c7a73329f1887d3f4876da6d563441cd2b01aa
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/f68ef10a1bd1b9e2aa57ac81d6f37235 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9db44f52021004df9fb64c2f92737ef08c999d4e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..68c086294cb019a2ad97e7f927a221744b74b73e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585006890219070, "commit_timestamp_nsecs": 1754585007080626263, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/d/0cfb262c07add8192c19490deeb00b35 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/d/0cfb262c07add8192c19490deeb00b35
new file mode 100644
index 0000000000000000000000000000000000000000..e4e56b64a486ba59f3dfcb619ee9b52ac7c35fb6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/d/0cfb262c07add8192c19490deeb00b35 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..31bea42b81318613f9cb2cf503ce64667396cac0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/097bc235f9a9ebfb22bc2b5769f1aa86 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/097bc235f9a9ebfb22bc2b5769f1aa86
new file mode 100644
index 0000000000000000000000000000000000000000..b01d6ce3548ade5d353891b24f8e88d0aef954f7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/097bc235f9a9ebfb22bc2b5769f1aa86
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f1a823a663db0369101264e3ece447d0d39bc771d233b1b4d142edf4cc65ad74
+size 278776
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/111c7019d7a4d976835acedd2ad43166 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/111c7019d7a4d976835acedd2ad43166
new file mode 100644
index 0000000000000000000000000000000000000000..ffea76edf3397c849c4b7a565d68d4341d63277b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/111c7019d7a4d976835acedd2ad43166 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/2764e6e5912f761ec09ba67b3fdabaf8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/2764e6e5912f761ec09ba67b3fdabaf8
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/2764e6e5912f761ec09ba67b3fdabaf8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/57ab1550db9edabb4c95242b38deff35 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/57ab1550db9edabb4c95242b38deff35
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/57ab1550db9edabb4c95242b38deff35 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/79b9deb4dba35c574ae372f5f1336893 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/79b9deb4dba35c574ae372f5f1336893
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/79b9deb4dba35c574ae372f5f1336893 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/7e0ba34f355b27f7482b25ab19d97e26 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/7e0ba34f355b27f7482b25ab19d97e26
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/7e0ba34f355b27f7482b25ab19d97e26 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/de2fbb8b9458f0ec3bcdbbd460cb0822 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/de2fbb8b9458f0ec3bcdbbd460cb0822
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/de2fbb8b9458f0ec3bcdbbd460cb0822 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/ee325a390ed2fcfd9e146891666bfdcf b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/ee325a390ed2fcfd9e146891666bfdcf
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/ee325a390ed2fcfd9e146891666bfdcf differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/ff9b49e2a363b175588a4d48e7161a82 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/ff9b49e2a363b175588a4d48e7161a82
new file mode 100644
index 0000000000000000000000000000000000000000..d3dc935ea0b60ea9a20430fefe8b9bc2bb837973
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/ff9b49e2a363b175588a4d48e7161a82
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0c461cff1677f976494a28aeae822de0a5b939defc557b311cd1172e65d5826b
+size 425994
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a8623edb74fde06c0be62d18ddab096345b2d9a8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..c866256491500c3edc16355883388c5c7af187c0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585040077271668, "commit_timestamp_nsecs": 1754585040259967603, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/d/14f0842044e0af69048060aec825f2d9 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/d/14f0842044e0af69048060aec825f2d9
new file mode 100644
index 0000000000000000000000000000000000000000..1942373519f6438c6fe6ceaf5d98ae4e99df26c5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/d/14f0842044e0af69048060aec825f2d9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a13989fbaecf544ed12177ebec6b35e09b15553d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/1bc8132889877e7f39ec376d50fee2b7 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/1bc8132889877e7f39ec376d50fee2b7
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/1bc8132889877e7f39ec376d50fee2b7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/1c82e6652e12cc388b7b12f34d1a7847 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/1c82e6652e12cc388b7b12f34d1a7847
new file mode 100644
index 0000000000000000000000000000000000000000..78c4dab5c631ad3cb17e82354954180120c01d0e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/1c82e6652e12cc388b7b12f34d1a7847
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fcee2a1fd5e693b16327fabc48d7ac106f8e5ffe7876ea536e48b8009a304749
+size 451373
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/21fb6830c41b74380dc4e7487360d8cf b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/21fb6830c41b74380dc4e7487360d8cf
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/21fb6830c41b74380dc4e7487360d8cf differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/30a88f68d29cd786b3c62b915db1de4d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/30a88f68d29cd786b3c62b915db1de4d
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/30a88f68d29cd786b3c62b915db1de4d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/7e943de4af521fd18ed8bbf0e213196d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/7e943de4af521fd18ed8bbf0e213196d
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/7e943de4af521fd18ed8bbf0e213196d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/939ebe2278e2cca43fe5cd43391357da b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/939ebe2278e2cca43fe5cd43391357da
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/939ebe2278e2cca43fe5cd43391357da differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/a1572db1892abdb444cd7c2b46040be8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/a1572db1892abdb444cd7c2b46040be8
new file mode 100644
index 0000000000000000000000000000000000000000..4d4520e74c453f0378fc71c6c90e95ac8cacd107
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/a1572db1892abdb444cd7c2b46040be8
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:197eaf233ef82daac74a2ba31737b03819cba0e4d1968da66f0e806c2701610c
+size 287358
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/b16ac7f93a04f0e390b45188134319b5 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/b16ac7f93a04f0e390b45188134319b5
new file mode 100644
index 0000000000000000000000000000000000000000..b0f7b09df2b3a6512dc442d4f127aa66740dcdba
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/b16ac7f93a04f0e390b45188134319b5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a30a203947f3155494d9648e50620f30f8343566
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..5f3c00a13fd40411a1daf93bbc4616be933edfcd
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585073337931908, "commit_timestamp_nsecs": 1754585073519092350, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/d/ae41e57351096c09d06199c712b035d3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/d/ae41e57351096c09d06199c712b035d3
new file mode 100644
index 0000000000000000000000000000000000000000..234bfd549cce27d1872125f9bf27f703e2f0497c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/d/ae41e57351096c09d06199c712b035d3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..24da33aa57dc6877171a6227bf4dd3fd475766eb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/458623a32cca2228650e9df682a4fe9d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/458623a32cca2228650e9df682a4fe9d
new file mode 100644
index 0000000000000000000000000000000000000000..d7e6cafd43bd00ad30d9aabe62d688def7ed3ace
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/458623a32cca2228650e9df682a4fe9d
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8e19cda959c3c78a0110d00a9f951adbf4a95f3e1793d4a0f5388a52e42128b1
+size 261959
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/6aa1e6df10c6c28f3b389911b6b42d35 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/6aa1e6df10c6c28f3b389911b6b42d35
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/6aa1e6df10c6c28f3b389911b6b42d35 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/7c871e9a99b35acba6dc6bc3c261731f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/7c871e9a99b35acba6dc6bc3c261731f
new file mode 100644
index 0000000000000000000000000000000000000000..640d3a3c1aa64a591a213d0a21b2dc334360ac84
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/7c871e9a99b35acba6dc6bc3c261731f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/a1e67179d06568798e74ac26aaa29621 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/a1e67179d06568798e74ac26aaa29621
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/a1e67179d06568798e74ac26aaa29621 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/a52df1b129f8622752e34499e7df0069 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/a52df1b129f8622752e34499e7df0069
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/a52df1b129f8622752e34499e7df0069 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/ab78cd7ca6e2cc8c7b7a4e63f323de74 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/ab78cd7ca6e2cc8c7b7a4e63f323de74
new file mode 100644
index 0000000000000000000000000000000000000000..753453ad22d4db4cfbbde6a56d89a8125fab182f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/ab78cd7ca6e2cc8c7b7a4e63f323de74
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:55a765d07cad29b0d38d805631011b5603ac3d7903b5e583a58a3475964bea1d
+size 472825
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/b3c68af9f748ec5cedb58a735514b39c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/b3c68af9f748ec5cedb58a735514b39c
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/b3c68af9f748ec5cedb58a735514b39c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/c4cdbf6c052d0a3c5362f0a20af49f4b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/c4cdbf6c052d0a3c5362f0a20af49f4b
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/c4cdbf6c052d0a3c5362f0a20af49f4b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..ee87235f8c0fbea25855b843b56c22db08a6efa6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..f104b222cd4972bb1a7d23f51e39b99d3c7fff0e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585106652091331, "commit_timestamp_nsecs": 1754585106830510050, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/d/426dea10e7b612861549e23121dc5a1d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/d/426dea10e7b612861549e23121dc5a1d
new file mode 100644
index 0000000000000000000000000000000000000000..d1f4ec516423b724b49be9b17917ce0db01329c3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/d/426dea10e7b612861549e23121dc5a1d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f6acfe0ee5c2cc47e7729488a5ee7d9bffde0d52
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/0adef324ea98b82d433d477d84108ffe b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/0adef324ea98b82d433d477d84108ffe
new file mode 100644
index 0000000000000000000000000000000000000000..f28d59e1f9187cfd65b688b4bceb57a81f0acdcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/0adef324ea98b82d433d477d84108ffe differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/23b50cb2710646d0f8888a1e20ed4206 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/23b50cb2710646d0f8888a1e20ed4206
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/23b50cb2710646d0f8888a1e20ed4206 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/2db9d6d257a1779b540f78875fa89289 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/2db9d6d257a1779b540f78875fa89289
new file mode 100644
index 0000000000000000000000000000000000000000..4f492f2a659545251ef9aa1b8411c3793b3d1d1c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/2db9d6d257a1779b540f78875fa89289 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/503b1a91b9581b4396c5af7f2795bc0f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/503b1a91b9581b4396c5af7f2795bc0f
new file mode 100644
index 0000000000000000000000000000000000000000..894585bbcd174d18480d7f6791003ae77f125ac3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/503b1a91b9581b4396c5af7f2795bc0f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bd51ad847402e685bc4ba24dcdb94eb6c447e35755a17b4b2e97edc9fe761074
+size 384402
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/512042c64f9d04e434de368943c77a5e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/512042c64f9d04e434de368943c77a5e
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/512042c64f9d04e434de368943c77a5e differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/a10e4eb9675a4080caed209d3ed2584e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/a10e4eb9675a4080caed209d3ed2584e
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/a10e4eb9675a4080caed209d3ed2584e differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/dc13b3abff983e32856b0effb8363a2a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/dc13b3abff983e32856b0effb8363a2a
new file mode 100644
index 0000000000000000000000000000000000000000..b78db44e6d9314d8d5ee9db5ad5e42ef93e4dd00
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/dc13b3abff983e32856b0effb8363a2a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/e5866a66eed10b448f57bce830da269d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/e5866a66eed10b448f57bce830da269d
new file mode 100644
index 0000000000000000000000000000000000000000..8b9c94e2fc7dd860eab644113ebd4ca477b5fb49
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/e5866a66eed10b448f57bce830da269d
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:66ca11af169f2eec21ce3990aa52ed8c438f465b57f70e7b9193c2445832efef
+size 353322
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..363d2422b332b23c49a87f2f747ea06629343ad3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..7ba31d2b96a2b5fbbe7111e8e1a958a25baf2f55
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585139718779925, "commit_timestamp_nsecs": 1754585139872668347, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/d/eb9acbd786bd1c5fb2200a699cbced74 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/d/eb9acbd786bd1c5fb2200a699cbced74
new file mode 100644
index 0000000000000000000000000000000000000000..617fc97d0f2faa2341446ef0c167f43e4e1c52b1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/d/eb9acbd786bd1c5fb2200a699cbced74 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a8c6ea3599d6dec32aef223853e3fd19a25fad26
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/0287ba4d553983916f79945b6ba6645e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/0287ba4d553983916f79945b6ba6645e
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/0287ba4d553983916f79945b6ba6645e differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/3ef62f7c18cdacdf7d7483d1a09341a7 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/3ef62f7c18cdacdf7d7483d1a09341a7
new file mode 100644
index 0000000000000000000000000000000000000000..65529f6532287fad4b450ca493011b55e8bf7f92
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/3ef62f7c18cdacdf7d7483d1a09341a7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/5b91ec5c4fd30dbbe29beb54d86df725 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/5b91ec5c4fd30dbbe29beb54d86df725
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/5b91ec5c4fd30dbbe29beb54d86df725 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/6ed0d14f21c9893fcfac0edb371329c3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/6ed0d14f21c9893fcfac0edb371329c3
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/6ed0d14f21c9893fcfac0edb371329c3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/714d1d48a9d2e6b6898c0fa00d22756e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/714d1d48a9d2e6b6898c0fa00d22756e
new file mode 100644
index 0000000000000000000000000000000000000000..d939a37f5db029113cf53ae911aabca40fd069a5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/714d1d48a9d2e6b6898c0fa00d22756e differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/7daab2913609df8f1bfae5a267638a9c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/7daab2913609df8f1bfae5a267638a9c
new file mode 100644
index 0000000000000000000000000000000000000000..df33bf63c7c5fe8b30a4ef399cf19950ca234f6b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/7daab2913609df8f1bfae5a267638a9c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a1e9c44d8b6864cea5658768bb4ffd856c70de0a5194cb97fbed7ccc2791271c
+size 295933
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/97412f725ada0fe5c2492ea274b9391d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/97412f725ada0fe5c2492ea274b9391d
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/97412f725ada0fe5c2492ea274b9391d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/d4244c8c0d4e96986db5793be0180ca8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/d4244c8c0d4e96986db5793be0180ca8
new file mode 100644
index 0000000000000000000000000000000000000000..8f958e6ca6e44650d287cac075701ad823a7f269
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/d4244c8c0d4e96986db5793be0180ca8
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3103f95dc1932142f6e7c99598bb7fb9474501e2b40879be3ab6c81b0129fbfb
+size 371024
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f87cafa0f47308fd224c367af8205bb3c2505eb6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..15151e9557652b262193c18e67f479c4280555e9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585172956321760, "commit_timestamp_nsecs": 1754585173136632593, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/d/48d10f5dc28847ff35241591afd10944 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/d/48d10f5dc28847ff35241591afd10944
new file mode 100644
index 0000000000000000000000000000000000000000..eee25f29bbc08dad3332a5db1eb967e1ed11baae
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/d/48d10f5dc28847ff35241591afd10944 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..42b34f915696c2c9f5d97454112f0f5f30bb17f9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/1ce993d8dcce5a3523a1cf8aebef0b5e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/1ce993d8dcce5a3523a1cf8aebef0b5e
new file mode 100644
index 0000000000000000000000000000000000000000..34b79194ec17ce64c08530c06ca76c22f65c7bbf
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/1ce993d8dcce5a3523a1cf8aebef0b5e differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/57f290eb4ec9214d380d1ad0ec3b1c6c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/57f290eb4ec9214d380d1ad0ec3b1c6c
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/57f290eb4ec9214d380d1ad0ec3b1c6c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/5dc2c5bcf5041b87739e42db5d9ddc68 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/5dc2c5bcf5041b87739e42db5d9ddc68
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/5dc2c5bcf5041b87739e42db5d9ddc68 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/d72b00acdb9c4d49005e069adf552154 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/d72b00acdb9c4d49005e069adf552154
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/d72b00acdb9c4d49005e069adf552154 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/e7d0e924bd067bef035bd0b676288da0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/e7d0e924bd067bef035bd0b676288da0
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/e7d0e924bd067bef035bd0b676288da0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/ee8a88812e9ccf0da70c34e35e43f5c0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/ee8a88812e9ccf0da70c34e35e43f5c0
new file mode 100644
index 0000000000000000000000000000000000000000..6b5d2caa255c13fdadeb7c43ffd62a93aa617b06
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/ee8a88812e9ccf0da70c34e35e43f5c0
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4d96f92091800ee07d628b9df61ec3dce82d4c86ecb575015511ed31fdd9b363
+size 426926
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/f1fcaa851415acb992c001f01d7862b6 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/f1fcaa851415acb992c001f01d7862b6
new file mode 100644
index 0000000000000000000000000000000000000000..0b7386555cbf6f6dff28145fc1769b3be43b3580
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/f1fcaa851415acb992c001f01d7862b6
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d9ae5e04f53eec344e5278c73d88faed3aa9854e69d878830fd4e6b519320ddc
+size 278848
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/f5e3ac31aee538a88b9e81498e379d68 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/f5e3ac31aee538a88b9e81498e379d68
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/f5e3ac31aee538a88b9e81498e379d68 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..24f74e09684aa3fe82a115725740b4c7cdb7b000
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..032df11a38456681d9df02850cfa164a559dc74a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583063559717261, "commit_timestamp_nsecs": 1754583063729990256, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/d/83c4c00e18daf9d6301adfbe344fc63a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/d/83c4c00e18daf9d6301adfbe344fc63a
new file mode 100644
index 0000000000000000000000000000000000000000..d95c074b2b5fed9840e9fb5e1430b4ca0c7b8164
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/d/83c4c00e18daf9d6301adfbe344fc63a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e68c24d0676d0115e8306348119fabfe9afceb89
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/0f55e80d8efc9419d83a37fa8b15e470 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/0f55e80d8efc9419d83a37fa8b15e470
new file mode 100644
index 0000000000000000000000000000000000000000..c853ce5dfdc043ea57f734c7ecdde5fff7c994eb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/0f55e80d8efc9419d83a37fa8b15e470
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6133014fad7e6e57e44d482b43776df271ee5786fbf6464b4ec08bb6745b1b33
+size 297033
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/19fd0801e9e9b9e9f0a5c46a9d4c63a2 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/19fd0801e9e9b9e9f0a5c46a9d4c63a2
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/19fd0801e9e9b9e9f0a5c46a9d4c63a2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/2bff70ef9499b0c2cd773fada90be06b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/2bff70ef9499b0c2cd773fada90be06b
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/2bff70ef9499b0c2cd773fada90be06b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/9cc20abe905fb7ca16e9448e8e4602dd b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/9cc20abe905fb7ca16e9448e8e4602dd
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/9cc20abe905fb7ca16e9448e8e4602dd differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/a8058069fc7156926e6890d80e04e6bc b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/a8058069fc7156926e6890d80e04e6bc
new file mode 100644
index 0000000000000000000000000000000000000000..f3991e29ad9cad086b17de02305fd4bc91fa0eb9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/a8058069fc7156926e6890d80e04e6bc
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c0e758b3800fa9a62e32597db3934b687a2d7e7d6326c5cecaca866b48e30c3f
+size 406022
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/dd6cd7e6c8aa0e77871016139dd60f1a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/dd6cd7e6c8aa0e77871016139dd60f1a
new file mode 100644
index 0000000000000000000000000000000000000000..72cf20208c4acb9580550d384c277085f703c292
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/dd6cd7e6c8aa0e77871016139dd60f1a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/de2a14b5718d58b8ac2f26e3b82e9443 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/de2a14b5718d58b8ac2f26e3b82e9443
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/de2a14b5718d58b8ac2f26e3b82e9443 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..64dc8d54d2c4c7c26b38877eb48573e3374a407c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..701edc59379a3d1d393305573cdda938b1a2e0b0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585206844378075, "commit_timestamp_nsecs": 1754585207024714300, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/d/b51b4a5281e127bd7c7e334f96c60fc4 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/d/b51b4a5281e127bd7c7e334f96c60fc4
new file mode 100644
index 0000000000000000000000000000000000000000..59a0a5df0ca4f6d809273d0753aa8303a7209052
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/d/b51b4a5281e127bd7c7e334f96c60fc4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..6837d59357134847152ee44cb1e056ce89fb9e41
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/17162f56f35f65942dca1f920f648d95 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/17162f56f35f65942dca1f920f648d95
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/17162f56f35f65942dca1f920f648d95 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/1a804c7301a1d69949ab2a7faa66c2f1 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/1a804c7301a1d69949ab2a7faa66c2f1
new file mode 100644
index 0000000000000000000000000000000000000000..d1cf604f6d936be94427f7fd2ced067f84e3de30
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/1a804c7301a1d69949ab2a7faa66c2f1
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bb272d3301413a8ca6f588e96196eaaaacc3b218d525d7eb17439ffb1b84cac4
+size 454091
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/4415171f55c337f993b2580975c0d907 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/4415171f55c337f993b2580975c0d907
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/4415171f55c337f993b2580975c0d907 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/760df65c1d18a7866d13d61c56818b32 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/760df65c1d18a7866d13d61c56818b32
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/760df65c1d18a7866d13d61c56818b32 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/a30c8adecb58b01dec9aefacff93a902 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/a30c8adecb58b01dec9aefacff93a902
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/a30c8adecb58b01dec9aefacff93a902 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/daa7d12e5b3d94d16853f3c16f82ab1b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/daa7d12e5b3d94d16853f3c16f82ab1b
new file mode 100644
index 0000000000000000000000000000000000000000..60c0426bf53e281088084f1bfaf64f27611dc4bc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/daa7d12e5b3d94d16853f3c16f82ab1b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/e6f142347fa91252634df254b6f88618 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/e6f142347fa91252634df254b6f88618
new file mode 100644
index 0000000000000000000000000000000000000000..76c9f61815dd7f0a2067893e625dab6d9f021f57
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/e6f142347fa91252634df254b6f88618
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:180989f80c6a9d7fdefa3b28147d8d9cf2bd45feebe8b7d2c1c7a9d0c59b7ed1
+size 244810
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/f93b1bc95e67f83b30f63e7ccb58ae3e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/f93b1bc95e67f83b30f63e7ccb58ae3e
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/f93b1bc95e67f83b30f63e7ccb58ae3e differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..02d9f97f7e8f770f91ac39d02e9e800d508d04b6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..a46ea0b6bb0b709719ddae34b7facdfbf8e14708
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585240156923355, "commit_timestamp_nsecs": 1754585240335426528, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/d/6dee5c404b1a67d47a3a6798d4135470 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/d/6dee5c404b1a67d47a3a6798d4135470
new file mode 100644
index 0000000000000000000000000000000000000000..50c3ef88db7a4eb6e390bcc6547b0008d3ddb84e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/d/6dee5c404b1a67d47a3a6798d4135470 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..6b4480a45c5e0ca4c8347360968ce19f33c09b87
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/1981728470f88259ed1f8998654a0172 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/1981728470f88259ed1f8998654a0172
new file mode 100644
index 0000000000000000000000000000000000000000..4481dc000ab799914d19be045c622c2a742e8bee
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/1981728470f88259ed1f8998654a0172
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e402627d4183bb3af6fb5fb23946a2cd0e9609c01d10f3cd652256caea202930
+size 486621
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/3f93a65879cae2122b647e32a512a4c4 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/3f93a65879cae2122b647e32a512a4c4
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/3f93a65879cae2122b647e32a512a4c4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/466a2ecc36dc7245f409ee570ab0a4c6 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/466a2ecc36dc7245f409ee570ab0a4c6
new file mode 100644
index 0000000000000000000000000000000000000000..65529f6532287fad4b450ca493011b55e8bf7f92
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/466a2ecc36dc7245f409ee570ab0a4c6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/48dda1f677dc72102e3f69bff9f1966f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/48dda1f677dc72102e3f69bff9f1966f
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/48dda1f677dc72102e3f69bff9f1966f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/49592ffd5d4633fa256ae0e9920831df b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/49592ffd5d4633fa256ae0e9920831df
new file mode 100644
index 0000000000000000000000000000000000000000..ecffe073b999bb60e00917004154c34568b4c531
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/49592ffd5d4633fa256ae0e9920831df
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9ef522cc4135cc2bec36edb858cf1d9db1811bbb90e1fef93f8fa9cdb9aec03d
+size 244801
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/b65dbc0d5d8efb6faa37a9cfa70c9d4a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/b65dbc0d5d8efb6faa37a9cfa70c9d4a
new file mode 100644
index 0000000000000000000000000000000000000000..0e41d65063bc42be5aef06924b56e44de2014024
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/b65dbc0d5d8efb6faa37a9cfa70c9d4a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/bbcd130594da8fb5f647b477b421e556 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/bbcd130594da8fb5f647b477b421e556
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/bbcd130594da8fb5f647b477b421e556 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/d7ae3f4d72ed3deb783464c4f51b6155 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/d7ae3f4d72ed3deb783464c4f51b6155
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/d7ae3f4d72ed3deb783464c4f51b6155 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/e4435c02588bf16da26187787d4ebca9 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/e4435c02588bf16da26187787d4ebca9
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/e4435c02588bf16da26187787d4ebca9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d5cf3c9ae72cc84900e9ffac1516c18beeee80e0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..758e99bec4f8b9563ea68e43621d553a40132802
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585273538550382, "commit_timestamp_nsecs": 1754585273681107085, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/d/3badcf43df10d496838e513baa98e531 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/d/3badcf43df10d496838e513baa98e531
new file mode 100644
index 0000000000000000000000000000000000000000..9d3ec68c1bc88f5b5be8bbd26a530eeb0a59eef4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/d/3badcf43df10d496838e513baa98e531 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0a7ad300cb63a51ed0ae5d2baa9d0d3f1d33f976
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/41da96eb31f5857df2f8e7c0da41b7d9 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/41da96eb31f5857df2f8e7c0da41b7d9
new file mode 100644
index 0000000000000000000000000000000000000000..9b1788b75f72a77853e1955710d30404a0cb0164
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/41da96eb31f5857df2f8e7c0da41b7d9
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b6b02c4dee391e3cce8736d23ee86b81efd90257d07c329a2476ebaa1537b6af
+size 295908
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/47b69fb6443d8c9e5eac30813c03c8b3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/47b69fb6443d8c9e5eac30813c03c8b3
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/47b69fb6443d8c9e5eac30813c03c8b3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/4bbc611eb3c6c238a3032afb7e6f50ce b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/4bbc611eb3c6c238a3032afb7e6f50ce
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/4bbc611eb3c6c238a3032afb7e6f50ce differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/90a5a0e9c70de7956ba32fe8ec55114e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/90a5a0e9c70de7956ba32fe8ec55114e
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/90a5a0e9c70de7956ba32fe8ec55114e differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/a9956250cc6217b5d83ea2eb7b2a3653 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/a9956250cc6217b5d83ea2eb7b2a3653
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/a9956250cc6217b5d83ea2eb7b2a3653 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/a9a9bfb8dc7cf3704c00b892f0a1e911 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/a9a9bfb8dc7cf3704c00b892f0a1e911
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/a9a9bfb8dc7cf3704c00b892f0a1e911 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/d3dc726d64ae4282d262021589860d86 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/d3dc726d64ae4282d262021589860d86
new file mode 100644
index 0000000000000000000000000000000000000000..82ecaf0ab4ed22dd6fb8c6118393d016c524c0a6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/d3dc726d64ae4282d262021589860d86
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e87ef2bed66d8067b02b0f1dc5159546bd4309bf73dd499b1a65197ac8351913
+size 442785
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/fcdbdd327594b2473ed517f9e36cb3ea b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/fcdbdd327594b2473ed517f9e36cb3ea
new file mode 100644
index 0000000000000000000000000000000000000000..61bd30409a82efb4dd13bfc8a7612da9d255633e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/fcdbdd327594b2473ed517f9e36cb3ea differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..19996da131187c02aca017addb0c4283fadd1359
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..c4617fbcaaf476b64350c66c53bbba9f78a96d31
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585306566748203, "commit_timestamp_nsecs": 1754585306746053730, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/d/2c0e9a31931415ba9be6b72516c8c7fc b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/d/2c0e9a31931415ba9be6b72516c8c7fc
new file mode 100644
index 0000000000000000000000000000000000000000..9c45bead7bfe667bb96295c96a3ccb3bf7503ac9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/d/2c0e9a31931415ba9be6b72516c8c7fc differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..dbad6bdffe6011acf0326fc1d3370b25f4104fcb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/0983f603f2df120bfbdc2fd37c2341b5 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/0983f603f2df120bfbdc2fd37c2341b5
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/0983f603f2df120bfbdc2fd37c2341b5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/13b40afdf9ae4b16b53e18711c0fa8c2 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/13b40afdf9ae4b16b53e18711c0fa8c2
new file mode 100644
index 0000000000000000000000000000000000000000..76d21ddbf672cd4daac993a775274c6b82848dd2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/13b40afdf9ae4b16b53e18711c0fa8c2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/26de0a31f24629b840ed9535267f1929 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/26de0a31f24629b840ed9535267f1929
new file mode 100644
index 0000000000000000000000000000000000000000..c0ab9f3080484ca41581c1c962a6c54b2d9f05b3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/26de0a31f24629b840ed9535267f1929 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/52da67b4da2e58684eadaad3ab1cfa7c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/52da67b4da2e58684eadaad3ab1cfa7c
new file mode 100644
index 0000000000000000000000000000000000000000..1ae679e6b512321d458cebc3d86cca7b7fce2f7e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/52da67b4da2e58684eadaad3ab1cfa7c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5b6cad6eb981fbdaa6d5458ca8f9beb1605991b5e620f7bed81d52eafaf45280
+size 295935
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/621ab6cdd2a7c5727a3657a30014264f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/621ab6cdd2a7c5727a3657a30014264f
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/621ab6cdd2a7c5727a3657a30014264f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/75aafd1ce923b5081e40d190050fac09 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/75aafd1ce923b5081e40d190050fac09
new file mode 100644
index 0000000000000000000000000000000000000000..cd8da373ae279421827eaadeb0534e6225487315
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/75aafd1ce923b5081e40d190050fac09 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/a2f089af329cad117ab27f72fdaf8653 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/a2f089af329cad117ab27f72fdaf8653
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/a2f089af329cad117ab27f72fdaf8653 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/c42b8c6b121336ed2ffb5bbf41ad674c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/c42b8c6b121336ed2ffb5bbf41ad674c
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/c42b8c6b121336ed2ffb5bbf41ad674c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/c575f0a1715910fae93719b4de435ce2 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/c575f0a1715910fae93719b4de435ce2
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/c575f0a1715910fae93719b4de435ce2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/f36b83f2a27b9181d64037c2a6397bcb b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/f36b83f2a27b9181d64037c2a6397bcb
new file mode 100644
index 0000000000000000000000000000000000000000..e5654227b8a14f71a87ba2641af897f88e11a7c0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/f36b83f2a27b9181d64037c2a6397bcb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5e7043a2b46c1e52f8ebba5611160304999d7d04520691978247c2cddb84aa29
+size 441064
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..decc8822eefa528ac02979d9739775ae02295573
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..a6e651927848a11ae686c43d01795310b78382be
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585339784588439, "commit_timestamp_nsecs": 1754585339953629126, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/d/1ca4c523f6409601d004ffa490718472 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/d/1ca4c523f6409601d004ffa490718472
new file mode 100644
index 0000000000000000000000000000000000000000..67047ee84768a5ef74a168febe05c8d63ceb5017
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/d/1ca4c523f6409601d004ffa490718472 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..14fcd59fccbb36fa1bca19a8b90d6fe2ee466fdd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/3bf05e2db708bd0e3745ad3a2cffac58 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/3bf05e2db708bd0e3745ad3a2cffac58
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/3bf05e2db708bd0e3745ad3a2cffac58 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/6bcfb16b21131103624295298ed0f99d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/6bcfb16b21131103624295298ed0f99d
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/6bcfb16b21131103624295298ed0f99d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/8ae758eee771393e268f90c1e512267c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/8ae758eee771393e268f90c1e512267c
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/8ae758eee771393e268f90c1e512267c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/909817f4312925c32930883b265dfcd3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/909817f4312925c32930883b265dfcd3
new file mode 100644
index 0000000000000000000000000000000000000000..f428abda5833023d39359a4dacb9e7c3d8a4ad8f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/909817f4312925c32930883b265dfcd3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/b552def1252864be8e4388aab6d4afa8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/b552def1252864be8e4388aab6d4afa8
new file mode 100644
index 0000000000000000000000000000000000000000..4e204c7a999c574e0ae2098a2d8418999d5fd592
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/b552def1252864be8e4388aab6d4afa8
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7cfedab8a3494e264f21cf38f20a0f5f08865cb26fe0304a17ebe1af30444d20
+size 340390
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/fb039818fea0ebf39fc88b41ac3eb8ec b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/fb039818fea0ebf39fc88b41ac3eb8ec
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/fb039818fea0ebf39fc88b41ac3eb8ec differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/fbdb7843441508891cdb58112fabb7c3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/fbdb7843441508891cdb58112fabb7c3
new file mode 100644
index 0000000000000000000000000000000000000000..4ff3015dbb1789bf05b1353c4d424ad752e0ed4d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/fbdb7843441508891cdb58112fabb7c3
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5efe3cd73f9fecdf5134c233c4d3ffb795f47d2ee9fbb5483022209c86291acf
+size 397904
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..19c9830e89c0837bd069caff1db708dc87f77546
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..caf40f49b5cf0a503b12948465b71f649c9f6851
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585372983984179, "commit_timestamp_nsecs": 1754585373148784823, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/d/b7ade72d3881b3539ed50e1b02358d6f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/d/b7ade72d3881b3539ed50e1b02358d6f
new file mode 100644
index 0000000000000000000000000000000000000000..d4f4a08c3004c546d66372494a5fee728de12d35
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/d/b7ade72d3881b3539ed50e1b02358d6f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c27fd2192a693bd687011df3dca9e906a3691621
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/78665b079923870a8d3095cfe40e40ef b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/78665b079923870a8d3095cfe40e40ef
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/78665b079923870a8d3095cfe40e40ef differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/b281d6bd7df490d5b5cc427de633bbbe b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/b281d6bd7df490d5b5cc427de633bbbe
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/b281d6bd7df490d5b5cc427de633bbbe differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/b9b62cf0996985bec877c4f5bb5660c4 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/b9b62cf0996985bec877c4f5bb5660c4
new file mode 100644
index 0000000000000000000000000000000000000000..81868e5a3d77057d3501bf76418dd124a60d14b2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/b9b62cf0996985bec877c4f5bb5660c4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/c98a3e0e9b4fc2e06e54b08bc9d1c7f6 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/c98a3e0e9b4fc2e06e54b08bc9d1c7f6
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/c98a3e0e9b4fc2e06e54b08bc9d1c7f6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/d115d9cec8a397bf20a0a5761cef7604 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/d115d9cec8a397bf20a0a5761cef7604
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/d115d9cec8a397bf20a0a5761cef7604 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/d39d76002a2a916c9a2d05df9751a8a1 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/d39d76002a2a916c9a2d05df9751a8a1
new file mode 100644
index 0000000000000000000000000000000000000000..e1e90f034ac0ed611c9b07ed1c1e4146e230e4c4
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/d39d76002a2a916c9a2d05df9751a8a1
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3d495e4f981484e7371c70b786bca56ce32cde915c4e8efb1ddedf56810d2e1b
+size 261954
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/eea40b3e4e75a43b4bdaef8a372d732c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/eea40b3e4e75a43b4bdaef8a372d732c
new file mode 100644
index 0000000000000000000000000000000000000000..1eddb3c24b184603452c024b016c189fca22c1e3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/eea40b3e4e75a43b4bdaef8a372d732c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0ebe8a6fdff93a49f85208f95fedb02b506aad52bf7decb6671a15377a36f4f6
+size 438893
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/f34fcaa327005163159d15b9fdf966b1 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/f34fcaa327005163159d15b9fdf966b1
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/f34fcaa327005163159d15b9fdf966b1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4d92c370becbe1f3d7423fdf22536b768521eb6b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..d14b85c04f3b648c54c66b9603a46e1c9be0e01c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585406960945008, "commit_timestamp_nsecs": 1754585407129759906, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/d/6420dcd52dd064cd9a4bba835e83a878 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/d/6420dcd52dd064cd9a4bba835e83a878
new file mode 100644
index 0000000000000000000000000000000000000000..a4018a142188383ac52ba0d3c89bb2c18078ebb1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/d/6420dcd52dd064cd9a4bba835e83a878 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..25ee1c9267d072d3bbc22624ffc763b160f6c560
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/122fe583a06a5952cbea3ea6a3de0660 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/122fe583a06a5952cbea3ea6a3de0660
new file mode 100644
index 0000000000000000000000000000000000000000..10797ec34ed6f8c8309c39fa2ee54326aba2ea26
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/122fe583a06a5952cbea3ea6a3de0660 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/2bd4db35f96c7dc81cfd5d4b12107d9f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/2bd4db35f96c7dc81cfd5d4b12107d9f
new file mode 100644
index 0000000000000000000000000000000000000000..33c4eab6394414beff0703a3f6dbd637f34149bc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/2bd4db35f96c7dc81cfd5d4b12107d9f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6bc73eb59df202a1141f74456a1472bf8831e854105ad323386beff951209ec6
+size 322783
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/2cb7a5176250bd6ad8e402b113a59ee0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/2cb7a5176250bd6ad8e402b113a59ee0
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/2cb7a5176250bd6ad8e402b113a59ee0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/42ad33b23806285f5b3dc1ebdff9979a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/42ad33b23806285f5b3dc1ebdff9979a
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/42ad33b23806285f5b3dc1ebdff9979a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/8655074efdb8e1a5c79181eb564b0968 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/8655074efdb8e1a5c79181eb564b0968
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/8655074efdb8e1a5c79181eb564b0968 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/bfd5b06d62565e26c6edf851315fd9b9 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/bfd5b06d62565e26c6edf851315fd9b9
new file mode 100644
index 0000000000000000000000000000000000000000..9e1a73f50a601d821fd44060b89dc7f938571884
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/bfd5b06d62565e26c6edf851315fd9b9
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:da6e41745b68565243e4c4be4a700e470d8a62c1ea2c48f6e4f116fab951d76a
+size 415007
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/e1195b687541bd3e1c4cc1a5b969dc13 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/e1195b687541bd3e1c4cc1a5b969dc13
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/e1195b687541bd3e1c4cc1a5b969dc13 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a8bf0dbcd1d859a9870886959f376edff3834736
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..d4f853bce856ba311c90e6346410b535d1f4a0ff
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585441767992945, "commit_timestamp_nsecs": 1754585441951542976, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/d/a09ddedacb56870f3d8a7febae8f88d0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/d/a09ddedacb56870f3d8a7febae8f88d0
new file mode 100644
index 0000000000000000000000000000000000000000..ed73f9076db6574edaefb331b256c6bb7a89ad8f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/d/a09ddedacb56870f3d8a7febae8f88d0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..24227fa36d7db00912e6fdfc881cbb17c103e47b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/346b2d64126145d1a1492dc28b43b4fc b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/346b2d64126145d1a1492dc28b43b4fc
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/346b2d64126145d1a1492dc28b43b4fc differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/57ccc73c9b47c46d323670d6cd160d59 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/57ccc73c9b47c46d323670d6cd160d59
new file mode 100644
index 0000000000000000000000000000000000000000..e26792d1ba5f6ae6bb12f27e4bbac310cbed63c6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/57ccc73c9b47c46d323670d6cd160d59
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b1175901d1995d1fa1be8e998b696ed8cf57710aaab4ee78f7304ace6bccdb54
+size 460032
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/7bf7914478846a02cf1bbf977274f8c3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/7bf7914478846a02cf1bbf977274f8c3
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/7bf7914478846a02cf1bbf977274f8c3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/a1173fd87aa29f75aceb9ff894c23414 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/a1173fd87aa29f75aceb9ff894c23414
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/a1173fd87aa29f75aceb9ff894c23414 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/a1ecb2e02ea5ebc88baeb996fd2aad27 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/a1ecb2e02ea5ebc88baeb996fd2aad27
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/a1ecb2e02ea5ebc88baeb996fd2aad27 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/b6653240ad51193c77eb0b482af4599b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/b6653240ad51193c77eb0b482af4599b
new file mode 100644
index 0000000000000000000000000000000000000000..5674b3b899bb3e1af4a66940d4d2f86cd486c930
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/b6653240ad51193c77eb0b482af4599b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d0c912af5097727d250ee64755bd5457b1c861fbe2798ec2b96f8656a1ad23f6
+size 278800
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/c93147c5e59e48526beb198e77553811 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/c93147c5e59e48526beb198e77553811
new file mode 100644
index 0000000000000000000000000000000000000000..48a2594e9570815effb82904dd081bb2c581c55e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/c93147c5e59e48526beb198e77553811 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/dda2a463dc6c40a5d7344d09bf00f37b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/dda2a463dc6c40a5d7344d09bf00f37b
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/dda2a463dc6c40a5d7344d09bf00f37b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/e65c278f026275c205df1308b6095583 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/e65c278f026275c205df1308b6095583
new file mode 100644
index 0000000000000000000000000000000000000000..65529f6532287fad4b450ca493011b55e8bf7f92
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/e65c278f026275c205df1308b6095583 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5ce25b8c8518ab9ca2d82762be312c40ee25ca97
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..db6a6ba0a7d6c506cab189db9cc0dfa35f3fa087
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585476191269622, "commit_timestamp_nsecs": 1754585476363734145, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/d/e94bb89479a47f402d6e9df6ac8c5c30 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/d/e94bb89479a47f402d6e9df6ac8c5c30
new file mode 100644
index 0000000000000000000000000000000000000000..bf9fcf618a26dc4b5a9a2229c8ba9d56574c8767
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/d/e94bb89479a47f402d6e9df6ac8c5c30 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a825f04625cdf6cf078cf3f3a7385921277e52b0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/3027a0a94b0c0ecbadfe46353314539c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/3027a0a94b0c0ecbadfe46353314539c
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/3027a0a94b0c0ecbadfe46353314539c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/3c5ae1b8853c11b2661ab873aa071273 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/3c5ae1b8853c11b2661ab873aa071273
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/3c5ae1b8853c11b2661ab873aa071273 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/5fe80ee08b12b88e2be8eea7eb8027d2 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/5fe80ee08b12b88e2be8eea7eb8027d2
new file mode 100644
index 0000000000000000000000000000000000000000..0a32f2f8515f263ffec6f71bd2b381bf3202b834
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/5fe80ee08b12b88e2be8eea7eb8027d2
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dc810f21b05167101980f44fbed38b636a6ef60d39ac0c7dd2d0f5e71e8a5cb6
+size 363838
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/6a7da64568588cccdfd4662003dfd87b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/6a7da64568588cccdfd4662003dfd87b
new file mode 100644
index 0000000000000000000000000000000000000000..31be3e8c704af8b13700f31452c050a63904cd18
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/6a7da64568588cccdfd4662003dfd87b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/803ba605a6821c572428593c603d025c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/803ba605a6821c572428593c603d025c
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/803ba605a6821c572428593c603d025c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/e2d9e24de945e232fda14a4dc94e16a8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/e2d9e24de945e232fda14a4dc94e16a8
new file mode 100644
index 0000000000000000000000000000000000000000..f6da90cb25deab7f406492b87d6ed881a4af652d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/e2d9e24de945e232fda14a4dc94e16a8
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:33b6e22c0f5ddeebbc47f0c469916742c514e357abf2d68ebcb089fac0ce7246
+size 336365
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/fbc32f05b2d4f8cd06ab55b5b1785beb b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/fbc32f05b2d4f8cd06ab55b5b1785beb
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/fbc32f05b2d4f8cd06ab55b5b1785beb differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a1121ea1e068a67ca90f3b9d875770bf17d5907c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..b23f87da37c6b727f7bb74705c86f074ff62f1c6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585509452893366, "commit_timestamp_nsecs": 1754585509621162107, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/d/55c0efa7c884c9b9ff1ce56b69702915 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/d/55c0efa7c884c9b9ff1ce56b69702915
new file mode 100644
index 0000000000000000000000000000000000000000..7616f03bf0c62e8d665a3707ded9e3759643db02
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/d/55c0efa7c884c9b9ff1ce56b69702915 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..90b57bf3ee70915dfb2404d752c6b355fcfb5d56
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/0cc6e6c982b35d143ad872c75b989d64 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/0cc6e6c982b35d143ad872c75b989d64
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/0cc6e6c982b35d143ad872c75b989d64 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/216cf48f19fc339502747611dc8b7b0e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/216cf48f19fc339502747611dc8b7b0e
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/216cf48f19fc339502747611dc8b7b0e differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/5d4c0ded748ae39e97271d8129581a6b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/5d4c0ded748ae39e97271d8129581a6b
new file mode 100644
index 0000000000000000000000000000000000000000..242e66b797161cba3ee398b00e06996593d04984
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/5d4c0ded748ae39e97271d8129581a6b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7d2853533af0c7c4361f2691ae1058d263acad971881a8b4bfbba0b0f44e80b8
+size 340306
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/a3d8ec11ef5a947bb253d395545514a9 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/a3d8ec11ef5a947bb253d395545514a9
new file mode 100644
index 0000000000000000000000000000000000000000..066019a3e9bd9f76f78ed3c765b87d7f630b0aaa
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/a3d8ec11ef5a947bb253d395545514a9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/b62cec65357c4ee13287faa0c0b5faa5 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/b62cec65357c4ee13287faa0c0b5faa5
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/b62cec65357c4ee13287faa0c0b5faa5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/bc299411c8ec3e4c22e3bd6c6b503c6a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/bc299411c8ec3e4c22e3bd6c6b503c6a
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/bc299411c8ec3e4c22e3bd6c6b503c6a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/c5cc2963f868804e70c7efff8fc0e0db b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/c5cc2963f868804e70c7efff8fc0e0db
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/c5cc2963f868804e70c7efff8fc0e0db differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/e2be54beffdc507d0d9cb6d51d1839ca b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/e2be54beffdc507d0d9cb6d51d1839ca
new file mode 100644
index 0000000000000000000000000000000000000000..e6e8d94ba3e2456330e69990c3d96163b8156a0b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/e2be54beffdc507d0d9cb6d51d1839ca
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ffe0e9d9cc164fb195a217ac267a86b0e2806b28bd81366ca4662ca2649e0c0a
+size 397968
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4372ed9e9bcaa51f52e6793dc9b17b52e0d88aad
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..4925bb3607e69c2b90320094388bf052bac3d76c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583096860123759, "commit_timestamp_nsecs": 1754583097041546775, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/d/ebea4f850d7d5750e443507ef739cfa8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/d/ebea4f850d7d5750e443507ef739cfa8
new file mode 100644
index 0000000000000000000000000000000000000000..ab6bdfdd3f09da5ff384e15f1304ac2c54b4bbd4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/d/ebea4f850d7d5750e443507ef739cfa8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e5aeb091175c80398cda1ddcfe7a8e7bb9b71a4b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/6775bcb1cec9481590c4b2a86d0e5561 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/6775bcb1cec9481590c4b2a86d0e5561
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/6775bcb1cec9481590c4b2a86d0e5561 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/707eae8c6ce3097943167e2e61334143 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/707eae8c6ce3097943167e2e61334143
new file mode 100644
index 0000000000000000000000000000000000000000..63052ecf30148cbb11921146ab447576bd075efb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/707eae8c6ce3097943167e2e61334143 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/7b925dab8c1886c43d96023514b3d19c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/7b925dab8c1886c43d96023514b3d19c
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/7b925dab8c1886c43d96023514b3d19c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/8ab3a96f250dcc817d7b3a7ca1565221 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/8ab3a96f250dcc817d7b3a7ca1565221
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/8ab3a96f250dcc817d7b3a7ca1565221 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/b57c489da78ae08fe0f8554d2c2a3e74 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/b57c489da78ae08fe0f8554d2c2a3e74
new file mode 100644
index 0000000000000000000000000000000000000000..cbb41d90e6a6a1669a35cb377a60d5c84e0423a5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/b57c489da78ae08fe0f8554d2c2a3e74
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:96a027eb6d23f2859f6cb6c0f9aa5cf9b62abc09fa2d270df2659fb8dd4ce49b
+size 295547
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/c269713c0bd9ba8138f30518b328a58b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/c269713c0bd9ba8138f30518b328a58b
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/c269713c0bd9ba8138f30518b328a58b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/eafe414a2add9bf7232aa34bdfd035b5 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/eafe414a2add9bf7232aa34bdfd035b5
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/eafe414a2add9bf7232aa34bdfd035b5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/f193c781721df09e038d554bdbb950be b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/f193c781721df09e038d554bdbb950be
new file mode 100644
index 0000000000000000000000000000000000000000..b5e4d58201732bd020eb48cf6eba347e29c4c76c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/f193c781721df09e038d554bdbb950be
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ae711767143b7e8a7456e691387d40d1d75f1e79d1fab5d760726a0de81d2420
+size 442251
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..cd0c4f868de8742ac187a29ba8998e9437490bf0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..b9e1956aabbc8347c22bac6af842e5bda6ecfa53
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585556236424302, "commit_timestamp_nsecs": 1754585556379842910, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/d/e9211edb4ecbc8cef281ffc0fe05834d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/d/e9211edb4ecbc8cef281ffc0fe05834d
new file mode 100644
index 0000000000000000000000000000000000000000..8e7179acd70864b868c7856637eb43f0357d4e83
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/d/e9211edb4ecbc8cef281ffc0fe05834d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a25f9fd5d65fb176fe539297b84573a2e560a2a9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/078bf3ab82971351cb8533a194697099 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/078bf3ab82971351cb8533a194697099
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/078bf3ab82971351cb8533a194697099 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/4271f2f02b270ba264ba9cf862b7473f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/4271f2f02b270ba264ba9cf862b7473f
new file mode 100644
index 0000000000000000000000000000000000000000..ba961aad5506f7023e837d8265b78fa09a069568
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/4271f2f02b270ba264ba9cf862b7473f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6a731fc8e7e63d99acb3c85fdca932bd7c947cf0da508f65eb631d208f0a1bdc
+size 377562
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/9a61fc9199d4832c9e0202a4611f3a97 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/9a61fc9199d4832c9e0202a4611f3a97
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/9a61fc9199d4832c9e0202a4611f3a97 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/a0fb460018f15287033c69a1957d2526 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/a0fb460018f15287033c69a1957d2526
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/a0fb460018f15287033c69a1957d2526 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/cbe4e858c74325974407271cf902180b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/cbe4e858c74325974407271cf902180b
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/cbe4e858c74325974407271cf902180b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/cfbd6511db8f6d464eec0bb9506024cc b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/cfbd6511db8f6d464eec0bb9506024cc
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/cfbd6511db8f6d464eec0bb9506024cc differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/e4a2a92a555231fa22c03d563b189c5c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/e4a2a92a555231fa22c03d563b189c5c
new file mode 100644
index 0000000000000000000000000000000000000000..aa9d1898e1669d21a8991543d246646d2fb21463
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/e4a2a92a555231fa22c03d563b189c5c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/e7d790bff641c7afc384b7a3538e1d98 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/e7d790bff641c7afc384b7a3538e1d98
new file mode 100644
index 0000000000000000000000000000000000000000..cf4452e46f4eeb15fd3d5ae306ce34a81353a2fa
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/e7d790bff641c7afc384b7a3538e1d98
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:129ef8f6a5f393c363d733e9b1e6ea5922efb5d525df9e797b1b67a0774d4003
+size 287384
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3222bb9cf4b61ce1355d42554c4d33e7c8f61e93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..1c3d87b20f91127b49198917c63d457626d564af
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585590325460521, "commit_timestamp_nsecs": 1754585590472615531, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/d/6258e40db720544cef808297eec02587 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/d/6258e40db720544cef808297eec02587
new file mode 100644
index 0000000000000000000000000000000000000000..0b380dc0013923f76c2af2d2c9d42568899509c6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/d/6258e40db720544cef808297eec02587 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b3fe2e5b43c85d8ee6f338322667773cfe138359
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/2658bbefb66e8194e214694f1c774611 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/2658bbefb66e8194e214694f1c774611
new file mode 100644
index 0000000000000000000000000000000000000000..f28d59e1f9187cfd65b688b4bceb57a81f0acdcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/2658bbefb66e8194e214694f1c774611 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/49887161769a1715ef30d68e84fe0606 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/49887161769a1715ef30d68e84fe0606
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/49887161769a1715ef30d68e84fe0606 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/7c5926b08279fdd08dc4904c05b48db3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/7c5926b08279fdd08dc4904c05b48db3
new file mode 100644
index 0000000000000000000000000000000000000000..b6510648e355a8fc76b2a39317820b68680e32aa
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/7c5926b08279fdd08dc4904c05b48db3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/a3015629bc80c29e21a3b7413ea35831 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/a3015629bc80c29e21a3b7413ea35831
new file mode 100644
index 0000000000000000000000000000000000000000..9892f907aec65bfd6d564d3ce6e7cf355e5ffe05
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/a3015629bc80c29e21a3b7413ea35831
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:af843c3f250b333f020956cdbfab46c45646ca60a8c86806f570d47889375835
+size 261996
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/b590b1b906abb751ff1e3183dc844490 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/b590b1b906abb751ff1e3183dc844490
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/b590b1b906abb751ff1e3183dc844490 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/b795ba9427b1423f828174ac5eaf8bd2 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/b795ba9427b1423f828174ac5eaf8bd2
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/b795ba9427b1423f828174ac5eaf8bd2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/cfbdeb7ecfd17b798d5e564dbadd50bb b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/cfbdeb7ecfd17b798d5e564dbadd50bb
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/cfbdeb7ecfd17b798d5e564dbadd50bb differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/d954cd8d10e4e1057626733a9b4832c0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/d954cd8d10e4e1057626733a9b4832c0
new file mode 100644
index 0000000000000000000000000000000000000000..d2c293cfacc7cf7371e8ca4d97b01ec9deea367e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/d954cd8d10e4e1057626733a9b4832c0
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a8e33630fc03c6db97b73b1e556eb2892c5ce46def56b48339791bb435c0dc70
+size 438821
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..cc945f895059ea693262f356444e05f8a9cb6930
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..d81110345139db259acddb11e6e8799690a11d68
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585624127894625, "commit_timestamp_nsecs": 1754585624280595182, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/d/0ab171a1c4b7ead86a60c53dcbb7cfae b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/d/0ab171a1c4b7ead86a60c53dcbb7cfae
new file mode 100644
index 0000000000000000000000000000000000000000..6045e1aa4b1834fef71c12876302a20c2d988514
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/d/0ab171a1c4b7ead86a60c53dcbb7cfae differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..89671260f76a3ba9cd4c2dbe55a6ae6b65553521
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/1386449b7eba9b1fda60827c3eed49c9 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/1386449b7eba9b1fda60827c3eed49c9
new file mode 100644
index 0000000000000000000000000000000000000000..2910fbc6db2094404ed932c7d9bcdb23beb99906
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/1386449b7eba9b1fda60827c3eed49c9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/2aef862650dece8931c24dcef5d4fe7d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/2aef862650dece8931c24dcef5d4fe7d
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/2aef862650dece8931c24dcef5d4fe7d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/3e104bcf090ab3fcad0faa8589c320f9 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/3e104bcf090ab3fcad0faa8589c320f9
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/3e104bcf090ab3fcad0faa8589c320f9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/60582817ed21b5c8e0df15894aeec880 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/60582817ed21b5c8e0df15894aeec880
new file mode 100644
index 0000000000000000000000000000000000000000..66705a93692149d9d8a86d4d59fa7459289a1a1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/60582817ed21b5c8e0df15894aeec880 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/9423cf31f5304ca39d0cad1e2c15c6a2 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/9423cf31f5304ca39d0cad1e2c15c6a2
new file mode 100644
index 0000000000000000000000000000000000000000..612663d4ec79d865f0de2a4309a4ba8e3d65c659
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/9423cf31f5304ca39d0cad1e2c15c6a2
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:866f2e559434bd008114afc187ea38f6147440d20e9b43f355aa379127f8e6f1
+size 459866
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/c5c2fce7d071b2918521ec6a0324319d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/c5c2fce7d071b2918521ec6a0324319d
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/c5c2fce7d071b2918521ec6a0324319d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/d5387e9d3a53f20c031c68e36c6024de b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/d5387e9d3a53f20c031c68e36c6024de
new file mode 100644
index 0000000000000000000000000000000000000000..f001cc522bda39cf3e705e17d73ab5b056c15451
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/d5387e9d3a53f20c031c68e36c6024de
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4bd495e7f119feb4e9ce941f2fc4bf60f442a3b9c333aee792ebe02158293dd3
+size 278835
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/e903dc2401d8a6f2dc507f69ec7cc578 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/e903dc2401d8a6f2dc507f69ec7cc578
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/e903dc2401d8a6f2dc507f69ec7cc578 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4cd1097e11cfa40750d04b9d5761747b60c7f14c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..9cc7cbd0d93fe22603fcbd24284f49d50804ae67
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585657347351914, "commit_timestamp_nsecs": 1754585657626983129, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/d/a33492e94ca64eb0d9d3fd794544d313 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/d/a33492e94ca64eb0d9d3fd794544d313
new file mode 100644
index 0000000000000000000000000000000000000000..ede7e2859662be34355fcdb1286a0d686babab32
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/d/a33492e94ca64eb0d9d3fd794544d313 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9f9654708c554c1ae37fd4a29ff11c4eadf8c7dd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/39a2fada611b93f50445d70887b30a08 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/39a2fada611b93f50445d70887b30a08
new file mode 100644
index 0000000000000000000000000000000000000000..d8327aa7be4db1ce3c827e00e7f95183b8464e29
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/39a2fada611b93f50445d70887b30a08 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/4af51e144bb6568372bc9225a8360550 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/4af51e144bb6568372bc9225a8360550
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/4af51e144bb6568372bc9225a8360550 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/4d6224faec4c27385403541bf2a7b291 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/4d6224faec4c27385403541bf2a7b291
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/4d6224faec4c27385403541bf2a7b291 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/509404e09d0dbf7936caa3d526543559 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/509404e09d0dbf7936caa3d526543559
new file mode 100644
index 0000000000000000000000000000000000000000..4859214e6b68fa6e697de9748417fcf43be8081d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/509404e09d0dbf7936caa3d526543559 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/74014576438054bbb2a2eb5ab213c1e9 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/74014576438054bbb2a2eb5ab213c1e9
new file mode 100644
index 0000000000000000000000000000000000000000..8ba292a2de82f9e0d52ca1a4b3e9f8da8c1191a2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/74014576438054bbb2a2eb5ab213c1e9
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cbc97c0ecfb888a26a83e63e9b167c099b954079ec20ca2b3ba1ca1e31d31c41
+size 694668
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/81fac9253020a58ecb7c73c3326e564b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/81fac9253020a58ecb7c73c3326e564b
new file mode 100644
index 0000000000000000000000000000000000000000..2910fbc6db2094404ed932c7d9bcdb23beb99906
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/81fac9253020a58ecb7c73c3326e564b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/aadaab50c458234951c815216ab528a2 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/aadaab50c458234951c815216ab528a2
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/aadaab50c458234951c815216ab528a2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/b2467eff990cdf518dc26f5ea41ae9b0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/b2467eff990cdf518dc26f5ea41ae9b0
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/b2467eff990cdf518dc26f5ea41ae9b0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/bc373548ecd965990b672ff0b7e6ff7a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/bc373548ecd965990b672ff0b7e6ff7a
new file mode 100644
index 0000000000000000000000000000000000000000..6f48bf83d80d22104dab4b18a40c519f214eca16
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/bc373548ecd965990b672ff0b7e6ff7a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..817a37fee680373e1da5054ec21ccf2a5117b9bd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..25a4d80544119cdbbebff52276218e57ee3e3d9b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585692170367490, "commit_timestamp_nsecs": 1754585692343814743, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/d/07221aa0893d7bdc567484697425e652 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/d/07221aa0893d7bdc567484697425e652
new file mode 100644
index 0000000000000000000000000000000000000000..87f93feb7a18564c0a8d87eaa0c7ebb7c8b8703f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/d/07221aa0893d7bdc567484697425e652 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3a2a0bd013eba4201add8f47a6f481d5540bcfd4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/24573dea43fa80181937aa4bed0eec82 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/24573dea43fa80181937aa4bed0eec82
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/24573dea43fa80181937aa4bed0eec82 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/2d0f8b9246216c14e2b81b5b04aebc67 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/2d0f8b9246216c14e2b81b5b04aebc67
new file mode 100644
index 0000000000000000000000000000000000000000..c0cc3b40df36b72e2a3388916fcf42e1107be085
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/2d0f8b9246216c14e2b81b5b04aebc67
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:380478515e962066bd3c5aa512a99e06d221245b42e21a80084601b843487fe6
+size 459723
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/41aaf88151cdc1c2adc5f512b41916e4 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/41aaf88151cdc1c2adc5f512b41916e4
new file mode 100644
index 0000000000000000000000000000000000000000..c2821d2c01c126364fc23326c2194d63084eaac2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/41aaf88151cdc1c2adc5f512b41916e4
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a5cc6d1d2bb225d2f3ff42119b8bc5265de57b920e465d898f23b285bb62a2f4
+size 278871
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/4309169e80e651721b665ee0346a1cf4 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/4309169e80e651721b665ee0346a1cf4
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/4309169e80e651721b665ee0346a1cf4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/71d0d105ea0b746855c730a99d847c77 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/71d0d105ea0b746855c730a99d847c77
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/71d0d105ea0b746855c730a99d847c77 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/8f57a94250e4e3b6c8392a2d86a9bb3f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/8f57a94250e4e3b6c8392a2d86a9bb3f
new file mode 100644
index 0000000000000000000000000000000000000000..65529f6532287fad4b450ca493011b55e8bf7f92
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/8f57a94250e4e3b6c8392a2d86a9bb3f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/b8e246748864c1a6cf66ead408639df1 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/b8e246748864c1a6cf66ead408639df1
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/b8e246748864c1a6cf66ead408639df1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/cd477b49bb4fbf5665542df848612df4 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/cd477b49bb4fbf5665542df848612df4
new file mode 100644
index 0000000000000000000000000000000000000000..f28d59e1f9187cfd65b688b4bceb57a81f0acdcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/cd477b49bb4fbf5665542df848612df4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/ee99a161683edb24c73031b86b7f38f9 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/ee99a161683edb24c73031b86b7f38f9
new file mode 100644
index 0000000000000000000000000000000000000000..517770abc43edc62957f0d99cc38ebeb528bb962
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/ee99a161683edb24c73031b86b7f38f9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0124c98565ed09da0cfa78e17a25ec14b1bec347
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..3fe2490b9a789e834a3dfc1df7a3771dd0476708
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585725377590523, "commit_timestamp_nsecs": 1754585725547589886, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/d/ec7ee3104ce7e07296082249f49f5d2a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/d/ec7ee3104ce7e07296082249f49f5d2a
new file mode 100644
index 0000000000000000000000000000000000000000..dcd149109ed406ca99d6e1603403f9e412b2e66a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/d/ec7ee3104ce7e07296082249f49f5d2a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d18d462f56b86ca366c9d75d31c7bc6244dde96b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/0297ef474d50a2ef090a395decafdf17 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/0297ef474d50a2ef090a395decafdf17
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/0297ef474d50a2ef090a395decafdf17 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/2affbb24e8637c1af68bea8cf1f93543 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/2affbb24e8637c1af68bea8cf1f93543
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/2affbb24e8637c1af68bea8cf1f93543 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/644aaecd70e0bfa033b32ff1fd2b174b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/644aaecd70e0bfa033b32ff1fd2b174b
new file mode 100644
index 0000000000000000000000000000000000000000..75e5315eb5d3e89d44f4ea1e2a6a942e42d41e8c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/644aaecd70e0bfa033b32ff1fd2b174b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:54bbaced785e40eeacc9e3df71bfc0c8688e081004aa1c235370a361cf367cdc
+size 261746
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/8273908e1d55753dea3cc58020f59ca1 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/8273908e1d55753dea3cc58020f59ca1
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/8273908e1d55753dea3cc58020f59ca1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/85f8b5a3f16481751987f67751ba2218 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/85f8b5a3f16481751987f67751ba2218
new file mode 100644
index 0000000000000000000000000000000000000000..009be8a7fa87777c487dc1837ace19ffdf1ac977
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/85f8b5a3f16481751987f67751ba2218
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cee52d226e6796a18bcfc54cf79c6dd7e65015f59a57ff1f73c95aaea586fa6f
+size 443054
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/917ec86dd612d75d798bee3694b9a73f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/917ec86dd612d75d798bee3694b9a73f
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/917ec86dd612d75d798bee3694b9a73f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/addde6dd52368c16e4df1d0fc36d5fc5 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/addde6dd52368c16e4df1d0fc36d5fc5
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/addde6dd52368c16e4df1d0fc36d5fc5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/c9f4452c67ba145b8c816e12e7455a93 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/c9f4452c67ba145b8c816e12e7455a93
new file mode 100644
index 0000000000000000000000000000000000000000..f28d59e1f9187cfd65b688b4bceb57a81f0acdcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/c9f4452c67ba145b8c816e12e7455a93 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/cd0563f084210b01ec871652d430f491 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/cd0563f084210b01ec871652d430f491
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/cd0563f084210b01ec871652d430f491 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/fe55f06ff5c6420a5038ae5c2f32af16 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/fe55f06ff5c6420a5038ae5c2f32af16
new file mode 100644
index 0000000000000000000000000000000000000000..34cdf651a85d3082d7175037c2efa3a2a2c480d8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/fe55f06ff5c6420a5038ae5c2f32af16 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4a34a815b850a4a5897c44731c3b86e85575a340
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..1390e442bbbc043c35d21d58bef87fc5c5bcd913
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585758680910991, "commit_timestamp_nsecs": 1754585758858137557, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/d/e9ad94343746dd712cf73c675883caa0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/d/e9ad94343746dd712cf73c675883caa0
new file mode 100644
index 0000000000000000000000000000000000000000..07612833f4f684f3b44157d6982ce1b348a22636
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/d/e9ad94343746dd712cf73c675883caa0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..7e73769544300857000eaf3ced38b886edf14d29
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/0080c850ac7e53eea8fd9ab0c27af53f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/0080c850ac7e53eea8fd9ab0c27af53f
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/0080c850ac7e53eea8fd9ab0c27af53f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/54b605b98ec89e31f9e1e2950ff93d37 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/54b605b98ec89e31f9e1e2950ff93d37
new file mode 100644
index 0000000000000000000000000000000000000000..15101f9dfd5af661e1465e0a96d57c326601b2a0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/54b605b98ec89e31f9e1e2950ff93d37 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/92c4e1152b153ddfdf5f65f6fbce3278 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/92c4e1152b153ddfdf5f65f6fbce3278
new file mode 100644
index 0000000000000000000000000000000000000000..5dc1f5fcbd479795cfe54974e833353db90f1ff9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/92c4e1152b153ddfdf5f65f6fbce3278
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0fc3e6715de930c920bb474e4a44d7208b6d88b2abfa5c6fa17a7f95eab89633
+size 253486
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/b26ce54608e9e58f7812758dd0191ece b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/b26ce54608e9e58f7812758dd0191ece
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/b26ce54608e9e58f7812758dd0191ece differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/b760d8af6b1e2781b95dd1c300471e65 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/b760d8af6b1e2781b95dd1c300471e65
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/b760d8af6b1e2781b95dd1c300471e65 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/d3ef6f25efec543afe1999f3f7c5faa8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/d3ef6f25efec543afe1999f3f7c5faa8
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/d3ef6f25efec543afe1999f3f7c5faa8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/e3a3d7afe8c5e956d600bf05dab23ff8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/e3a3d7afe8c5e956d600bf05dab23ff8
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/e3a3d7afe8c5e956d600bf05dab23ff8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/e7c102265c760dbc35b9abbd1d7d02b8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/e7c102265c760dbc35b9abbd1d7d02b8
new file mode 100644
index 0000000000000000000000000000000000000000..02ab796566833c7c5a487e37ce2e940e05bb98b7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/e7c102265c760dbc35b9abbd1d7d02b8
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b13848f84e42642be5ab048389ba54caffaa8f18f58a050a9acd53c602a587d6
+size 447463
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..57059d0a98f8e9bc3abec6410a1f1acd15b2b68f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..813cb5e6b71d1bf0f06a02dc5358b22d1e975378
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585793128176932, "commit_timestamp_nsecs": 1754585793270086111, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/d/545cf8193e20c77eb604802d9fee4033 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/d/545cf8193e20c77eb604802d9fee4033
new file mode 100644
index 0000000000000000000000000000000000000000..a9e6228d4559da0b23281b690595ef4e5c110d17
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/d/545cf8193e20c77eb604802d9fee4033 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c4c0dc90241d0075774867c3f1270091dfe7c2bc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/1c3aa9e431863b273400a5073bfa47ce b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/1c3aa9e431863b273400a5073bfa47ce
new file mode 100644
index 0000000000000000000000000000000000000000..fe30b87486055ce52a03e13948a604b7ed5d50ef
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/1c3aa9e431863b273400a5073bfa47ce differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/2bc057f7fd66833f70f187691a559169 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/2bc057f7fd66833f70f187691a559169
new file mode 100644
index 0000000000000000000000000000000000000000..c26f972fd48c3395d46aaf5957faa4a465d1e721
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/2bc057f7fd66833f70f187691a559169
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1db9f7f7ff8c11f53970c819537a803d1aa2aef33b11d9dd7cddc50bb5477286
+size 270325
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/50bcec50ca74813c3dd6e601b36d7090 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/50bcec50ca74813c3dd6e601b36d7090
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/50bcec50ca74813c3dd6e601b36d7090 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/8af70989ac6e10630ea41ba43c684765 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/8af70989ac6e10630ea41ba43c684765
new file mode 100644
index 0000000000000000000000000000000000000000..2a31377ca030aced51ef70fb458c6b5c47404cd3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/8af70989ac6e10630ea41ba43c684765 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/b12b1e534f759c429a2866672b8a2906 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/b12b1e534f759c429a2866672b8a2906
new file mode 100644
index 0000000000000000000000000000000000000000..6dad1bd79795b952613dcc34b340602a65e94c32
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/b12b1e534f759c429a2866672b8a2906
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d519816e8eb2b13b3dcb5b46e86b90caa0e77fc1cbf583388a3419caa84a5ce7
+size 468368
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/cd47bc21497d6e09c2f4c6907ce86ff3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/cd47bc21497d6e09c2f4c6907ce86ff3
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/cd47bc21497d6e09c2f4c6907ce86ff3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/d3f751b0bbc0de9a45de1e47b613ac3d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/d3f751b0bbc0de9a45de1e47b613ac3d
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/d3f751b0bbc0de9a45de1e47b613ac3d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/fba38df314c2744338429e4fd9cf9352 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/fba38df314c2744338429e4fd9cf9352
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/fba38df314c2744338429e4fd9cf9352 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3c5b39ad337fe8a867e784b2d6df0ad8fa7e748f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..88bb43b3acd8b8221b290bf4bf6f7a8c4044cb27
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585826647446288, "commit_timestamp_nsecs": 1754585826836576180, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/d/1115f96aca4579553b32e99ec40dcb47 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/d/1115f96aca4579553b32e99ec40dcb47
new file mode 100644
index 0000000000000000000000000000000000000000..584c84a8fae068bc13a8aa664c9234c7075589f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/d/1115f96aca4579553b32e99ec40dcb47 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1e250fed56756437c80c6855f9f9ac0945ca9e2b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/061eca80fff0e4f87f9604a32087e02f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/061eca80fff0e4f87f9604a32087e02f
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/061eca80fff0e4f87f9604a32087e02f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/0ea7e2b57e52d602e61352973e10fd38 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/0ea7e2b57e52d602e61352973e10fd38
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/0ea7e2b57e52d602e61352973e10fd38 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/1d1d676ea87126cce63c552aa083bf38 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/1d1d676ea87126cce63c552aa083bf38
new file mode 100644
index 0000000000000000000000000000000000000000..5a3c51498d8d6f98d4ff70b0c42e23a95a9583da
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/1d1d676ea87126cce63c552aa083bf38
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:45aa14156bf83d4736430f835f5c45a8e00d87b93045d9f648d81a660b8ad85b
+size 372555
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/36a495cae1ee407e0d8559b696ab4f80 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/36a495cae1ee407e0d8559b696ab4f80
new file mode 100644
index 0000000000000000000000000000000000000000..9d42a06cbfa6d9d815d9de6441498dcdcccb5cad
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/36a495cae1ee407e0d8559b696ab4f80
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:61d960e5078958563035d5094044d18ea7f41bbe5c0c89f969c4daa5b9979cf7
+size 365714
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/392840ad38f6fbc47b9c576f75914585 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/392840ad38f6fbc47b9c576f75914585
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/392840ad38f6fbc47b9c576f75914585 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/3bed923701d1b7266828c6b71946f16a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/3bed923701d1b7266828c6b71946f16a
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/3bed923701d1b7266828c6b71946f16a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/c2b18f68e670c77af4d412a4c34ec4cb b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/c2b18f68e670c77af4d412a4c34ec4cb
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/c2b18f68e670c77af4d412a4c34ec4cb differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/df7f6355697698626b4dda0067a88c3e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/df7f6355697698626b4dda0067a88c3e
new file mode 100644
index 0000000000000000000000000000000000000000..020fb3e44d485a21212f75a61c69ff87e3acf065
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/df7f6355697698626b4dda0067a88c3e differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..7c3f368c526623838bfe3aff24d55f6b5931738c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..d3e8e7607066f3df78f3b4ac7cdf7b751368fabf
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585860619510292, "commit_timestamp_nsecs": 1754585860780816880, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/d/da5f5b6a5cf7a351c514553308d45872 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/d/da5f5b6a5cf7a351c514553308d45872
new file mode 100644
index 0000000000000000000000000000000000000000..0dfe046069f8fe0c1be5ed38ab6772420c3bdb60
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/d/da5f5b6a5cf7a351c514553308d45872 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..72cb972a282b4d738372e94e721addd72191866f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/3e8228ec4f6d7a421a31f8df26da6579 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/3e8228ec4f6d7a421a31f8df26da6579
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/3e8228ec4f6d7a421a31f8df26da6579 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/5c626ca02019f8f341f51cf7e11406b8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/5c626ca02019f8f341f51cf7e11406b8
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/5c626ca02019f8f341f51cf7e11406b8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/713cfa8033b5405326e7f9bcc99542e3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/713cfa8033b5405326e7f9bcc99542e3
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/713cfa8033b5405326e7f9bcc99542e3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/880ef1f1894d038568d75a458aa1be38 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/880ef1f1894d038568d75a458aa1be38
new file mode 100644
index 0000000000000000000000000000000000000000..c2ca5827c43795c8c2abaa8e6468d2a7fe4c6d22
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/880ef1f1894d038568d75a458aa1be38 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/a573635f8eee854be47b38c025ea531b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/a573635f8eee854be47b38c025ea531b
new file mode 100644
index 0000000000000000000000000000000000000000..51698b76807692aab3bc07b2b4892fc45d7b6407
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/a573635f8eee854be47b38c025ea531b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:eb95511f663d4e827cf74a485c06aa2a0788989315bb619210615fc2998d589b
+size 296010
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/ae65567848b96d14fb6ab436a9faee26 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/ae65567848b96d14fb6ab436a9faee26
new file mode 100644
index 0000000000000000000000000000000000000000..f2196f23e1dfe24539953292aaada092f570039f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/ae65567848b96d14fb6ab436a9faee26
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:db8e0607d6793b298383793c5900faf7ef81f57a9107a098bb277b67f0d76af6
+size 404896
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/b105039bf299100f973ad77947ae057b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/b105039bf299100f973ad77947ae057b
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/b105039bf299100f973ad77947ae057b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/c394fbf19de28367bc2005f3b1b11f8c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/c394fbf19de28367bc2005f3b1b11f8c
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/c394fbf19de28367bc2005f3b1b11f8c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..eb2651dbacddd9d5b6509ea56b339619c1b02e57
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..530d50fa8090ad718ae8aa007ad980f1d0852509
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583128928741618, "commit_timestamp_nsecs": 1754583129072724759, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/d/f2bfcb21ec72aebbd7967b3774933536 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/d/f2bfcb21ec72aebbd7967b3774933536
new file mode 100644
index 0000000000000000000000000000000000000000..f050db964868c6481c9038128fef96e3a78d0892
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/d/f2bfcb21ec72aebbd7967b3774933536 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..33df38ae6bb0633a26a567aa96d705199685354e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/616adc2ca59c2d93b47c74f924247ab9 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/616adc2ca59c2d93b47c74f924247ab9
new file mode 100644
index 0000000000000000000000000000000000000000..0324b40fcb9b2f039856540ae8912b4c5b31709c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/616adc2ca59c2d93b47c74f924247ab9
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e7cea1ddf744df0871d51b49137b9ad2f968240b28b58d260a0548e1a1686fd1
+size 314678
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/b7c9e9e74ae5db09471d9d7a0e94366b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/b7c9e9e74ae5db09471d9d7a0e94366b
new file mode 100644
index 0000000000000000000000000000000000000000..a79399df04f787bb66dea8bdbf5d37bedb0a3a08
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/b7c9e9e74ae5db09471d9d7a0e94366b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4911d62c12f82f164432c4cd94ae363e81bdfda00f400d6e25adacb37fce3ea8
+size 388857
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/b89efa794fec44cfbe2fef549a2de3fd b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/b89efa794fec44cfbe2fef549a2de3fd
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/b89efa794fec44cfbe2fef549a2de3fd differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/de8bc8df4684b43b9ceb7ad4d6da8680 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/de8bc8df4684b43b9ceb7ad4d6da8680
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/de8bc8df4684b43b9ceb7ad4d6da8680 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/e0647e1219c98b39b7432d8e626694c5 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/e0647e1219c98b39b7432d8e626694c5
new file mode 100644
index 0000000000000000000000000000000000000000..9424ce30bda9411bed2158c53f6e2a96ffee20f3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/e0647e1219c98b39b7432d8e626694c5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/fa6ebfdfbdda034d9e737a063f103fd7 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/fa6ebfdfbdda034d9e737a063f103fd7
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/fa6ebfdfbdda034d9e737a063f103fd7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/fd1a7611504ffd15f0ff1f3badfa2673 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/fd1a7611504ffd15f0ff1f3badfa2673
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/fd1a7611504ffd15f0ff1f3badfa2673 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b21214f6cec35a23236ba3b1f8e35e40b6f89f22
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..bbdd6a4f29662fe9cc54e4fd02479c0eef2fc8a1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585894124257030, "commit_timestamp_nsecs": 1754585894277720213, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/d/a305f27b091c1cc6ee3b2879b7ab6524 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/d/a305f27b091c1cc6ee3b2879b7ab6524
new file mode 100644
index 0000000000000000000000000000000000000000..ebc08d5d6c3ad72293ed254f76bb6d8c3221eae6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/d/a305f27b091c1cc6ee3b2879b7ab6524 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..58fc45ff019787052ccfbc685f02b03aadb583ac
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/207ecc8f68349b0d29a1ea9d6fe3ccb6 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/207ecc8f68349b0d29a1ea9d6fe3ccb6
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/207ecc8f68349b0d29a1ea9d6fe3ccb6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/5937b0a7e4c36092fb3a9d209c8f5e3b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/5937b0a7e4c36092fb3a9d209c8f5e3b
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/5937b0a7e4c36092fb3a9d209c8f5e3b differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/6e3f3985b6c6c95c4120d57cd7d70f73 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/6e3f3985b6c6c95c4120d57cd7d70f73
new file mode 100644
index 0000000000000000000000000000000000000000..009dd79a5b3071dcd9f26ef7a950b29e4ca37fa9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/6e3f3985b6c6c95c4120d57cd7d70f73 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/ac6723d7d822c0d5a5dcc65bdfb061a8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/ac6723d7d822c0d5a5dcc65bdfb061a8
new file mode 100644
index 0000000000000000000000000000000000000000..9637c97faa0ad5864462a7535bd81501bb7d4120
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/ac6723d7d822c0d5a5dcc65bdfb061a8
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d3b5d23809aafb8f336f0c0aaddafcd29ff521342974d1bb2e48bda7eab63a12
+size 438938
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/b4832f195861a87b672f8f81476456af b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/b4832f195861a87b672f8f81476456af
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/b4832f195861a87b672f8f81476456af differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/c6d1ea88b20107b68b1ddd4766f1bd9e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/c6d1ea88b20107b68b1ddd4766f1bd9e
new file mode 100644
index 0000000000000000000000000000000000000000..b0f664a5cf8f5f0d59de982b1f495995edccf32b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/c6d1ea88b20107b68b1ddd4766f1bd9e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:75b81ad3a05216b99483e5308ec46f22f86ce73daa2b348b92f3a00b2fe16191
+size 262013
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/d233a36d694768c9c7e058edc00c9af1 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/d233a36d694768c9c7e058edc00c9af1
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/d233a36d694768c9c7e058edc00c9af1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/e49aa0d8684a8cd192a53be19c8dbdbb b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/e49aa0d8684a8cd192a53be19c8dbdbb
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/e49aa0d8684a8cd192a53be19c8dbdbb differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a7bfec62f46879629084dcc8e70a2ad4d70c07d1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..eedddcfea53f23b1283fdc16ddd3628cc0bff5ee
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585928942673196, "commit_timestamp_nsecs": 1754585929116768099, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/d/ea902fd977dbf6ba83909420cc36794d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/d/ea902fd977dbf6ba83909420cc36794d
new file mode 100644
index 0000000000000000000000000000000000000000..60921759860a072b614716ec470346a0861d73f4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/d/ea902fd977dbf6ba83909420cc36794d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..ff18d75ba4be18eb7389ea1fbc54098ec7acd8c2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/14e56fee2c090dac359cd2d8002d4884 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/14e56fee2c090dac359cd2d8002d4884
new file mode 100644
index 0000000000000000000000000000000000000000..faedeefcd5e18111aabbad63c85d39160da3b083
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/14e56fee2c090dac359cd2d8002d4884 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/17366d57b0d44fc604ce237f2a29957c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/17366d57b0d44fc604ce237f2a29957c
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/17366d57b0d44fc604ce237f2a29957c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/218613bfe0df1c90e8a9e46d5be70328 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/218613bfe0df1c90e8a9e46d5be70328
new file mode 100644
index 0000000000000000000000000000000000000000..ba26e27f30b1498ef296f129752886ccc30a2d67
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/218613bfe0df1c90e8a9e46d5be70328
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2d4de689f8010f6e4fb131f8a37a12a97ad1eda6488eef39fb3d0342c261575c
+size 348800
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/4fc8a51c42d7133577c9ed012daf1687 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/4fc8a51c42d7133577c9ed012daf1687
new file mode 100644
index 0000000000000000000000000000000000000000..2a31377ca030aced51ef70fb458c6b5c47404cd3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/4fc8a51c42d7133577c9ed012daf1687 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/abee3cdc3d3a1d04487d7ac5a515eac3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/abee3cdc3d3a1d04487d7ac5a515eac3
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/abee3cdc3d3a1d04487d7ac5a515eac3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/c23ec5639e6bafbd899eec00733810cd b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/c23ec5639e6bafbd899eec00733810cd
new file mode 100644
index 0000000000000000000000000000000000000000..3a4f2f0722ef73fd66386ae0d73a3440907b6d2e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/c23ec5639e6bafbd899eec00733810cd
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0b2d9b62249f3d59c25c826fbc8ab2f7b017846fd1bde986fb1df8d5f721ca1f
+size 389511
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/e14f865cc689de505cb3bd5d1837c550 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/e14f865cc689de505cb3bd5d1837c550
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/e14f865cc689de505cb3bd5d1837c550 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3d45f6e108933629b01ea983b977375e2e7b6ea4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..7e271c0a77ac8de80f9cd3acf06d4a9c709aa2cb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585962162940866, "commit_timestamp_nsecs": 1754585962364615576, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/d/393e96f0dffd62e60b4f9d82892e9e30 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/d/393e96f0dffd62e60b4f9d82892e9e30
new file mode 100644
index 0000000000000000000000000000000000000000..fca55c84d844d4f1d3414ec108515e37239dddee
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/d/393e96f0dffd62e60b4f9d82892e9e30 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..11c00a1df8f840eef4f7051ffca03484577ced09
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/08135e35d6fee1d325515158dd12c855 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/08135e35d6fee1d325515158dd12c855
new file mode 100644
index 0000000000000000000000000000000000000000..19a70cd8360fb383c61f45fef1d4e9cdcd2247c9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/08135e35d6fee1d325515158dd12c855
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e6dfec96e01fcb2fc968da0c0f8f801cc110800b0127268dd711119cefafd9b2
+size 382457
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/32788643d1fbe373aaaba030f5a7acd3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/32788643d1fbe373aaaba030f5a7acd3
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/32788643d1fbe373aaaba030f5a7acd3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/482fa089b0fce3e31c2e7aa4355feeca b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/482fa089b0fce3e31c2e7aa4355feeca
new file mode 100644
index 0000000000000000000000000000000000000000..0c01d2969d427d6de35cb8322e4609de1205c87c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/482fa089b0fce3e31c2e7aa4355feeca
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:993f57b0d980de72b44c9e8f028afc23391e86243850002dce3f3e00844dd895
+size 321497
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/53f7a7f32a60fa529407a024010de6be b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/53f7a7f32a60fa529407a024010de6be
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/53f7a7f32a60fa529407a024010de6be differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/57f40e35c86e20b3cd2c404ef69a0506 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/57f40e35c86e20b3cd2c404ef69a0506
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/57f40e35c86e20b3cd2c404ef69a0506 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/8092bb8f7526850210c5f2218353b9b3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/8092bb8f7526850210c5f2218353b9b3
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/8092bb8f7526850210c5f2218353b9b3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/a53be4c9a32a599cd671997bd30fe5a0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/a53be4c9a32a599cd671997bd30fe5a0
new file mode 100644
index 0000000000000000000000000000000000000000..9dc107a0cedc2f909aedfacf73026788e2dd793b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/a53be4c9a32a599cd671997bd30fe5a0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/e2759486197097d8e498cf97dc8f73d8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/e2759486197097d8e498cf97dc8f73d8
new file mode 100644
index 0000000000000000000000000000000000000000..ea747c59990b001f20961dd52dea851441461709
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/e2759486197097d8e498cf97dc8f73d8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/f1181608b58a961f98aad08c0613b6ad b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/f1181608b58a961f98aad08c0613b6ad
new file mode 100644
index 0000000000000000000000000000000000000000..65529f6532287fad4b450ca493011b55e8bf7f92
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/f1181608b58a961f98aad08c0613b6ad differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..7d3c9d1481bc951e0de79ca2917bafd752ca6649
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..78074b70df12b84e45fac4d552b2f9e66f800c3b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754585992957758323, "commit_timestamp_nsecs": 1754585993126526848, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/d/3aec500db9b1c94646ffa9aa78bffb2a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/d/3aec500db9b1c94646ffa9aa78bffb2a
new file mode 100644
index 0000000000000000000000000000000000000000..ced771594223edd08b9278e64f1f85316d8243d8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/d/3aec500db9b1c94646ffa9aa78bffb2a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8697fd42b71ab4943dda020112330d22d7e4d8ad
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/63bc6f0445ff5e872130066b3fce8cbf b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/63bc6f0445ff5e872130066b3fce8cbf
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/63bc6f0445ff5e872130066b3fce8cbf differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/69996d07f2c99529519db2fea22cae96 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/69996d07f2c99529519db2fea22cae96
new file mode 100644
index 0000000000000000000000000000000000000000..cd197dc4c2cfb79f8aa64d50f5d836e5b86ba518
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/69996d07f2c99529519db2fea22cae96
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3e078ea2be03e8a39bf5be0318f9e2c126ea3b858d1742e3e00a7d2a8436cbc8
+size 408527
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/7fe01396b3777d1c649f82e5cf2fb267 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/7fe01396b3777d1c649f82e5cf2fb267
new file mode 100644
index 0000000000000000000000000000000000000000..548a5a80424fdd9cad1f55660d51abdfec9b56f5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/7fe01396b3777d1c649f82e5cf2fb267
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1b7a1a4c955e7c36ad6ab108263abdef74e3a587117abf3fcddadd901bc64576
+size 330013
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/8af200be2d227e4e8557204d31d5c380 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/8af200be2d227e4e8557204d31d5c380
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/8af200be2d227e4e8557204d31d5c380 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/9c5f11f999eeba47cc616966083e47b5 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/9c5f11f999eeba47cc616966083e47b5
new file mode 100644
index 0000000000000000000000000000000000000000..4c0071f2d6362460419ebf6b4098ea3155d10e3b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/9c5f11f999eeba47cc616966083e47b5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/ec5603e50af4646503c4ee2260297cbd b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/ec5603e50af4646503c4ee2260297cbd
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/ec5603e50af4646503c4ee2260297cbd differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/eca9ca7c31f30fa55e4e5cdb014e0d3e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/eca9ca7c31f30fa55e4e5cdb014e0d3e
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/eca9ca7c31f30fa55e4e5cdb014e0d3e differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9b24ab03d4e82d5df6039ee0d7e8acb59df9cfd0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..5dea46356c419b016069a7b5aadf450673eff37d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754586025934902498, "commit_timestamp_nsecs": 1754586026081314515, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/d/5ee0ee4e00a8035595ce5ccfa30cd2e1 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/d/5ee0ee4e00a8035595ce5ccfa30cd2e1
new file mode 100644
index 0000000000000000000000000000000000000000..5812316bf2e2b2a26b2e71250da770c5cece783d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/d/5ee0ee4e00a8035595ce5ccfa30cd2e1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4bab381635b4e0feb9399e0564e18adbc6da3bd0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/07178b05ec22a0b1e7cb7f693e019a9d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/07178b05ec22a0b1e7cb7f693e019a9d
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/07178b05ec22a0b1e7cb7f693e019a9d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/540b23b6b4a948f875fd4fa7aeb12b61 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/540b23b6b4a948f875fd4fa7aeb12b61
new file mode 100644
index 0000000000000000000000000000000000000000..85f35e7991cac7b6fc244bb2f277438119f4e90b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/540b23b6b4a948f875fd4fa7aeb12b61 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/9218150f805196f26074893cc94c7d19 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/9218150f805196f26074893cc94c7d19
new file mode 100644
index 0000000000000000000000000000000000000000..10a19e9a36dcbba8df779109efcaa432a1aa804d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/9218150f805196f26074893cc94c7d19
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ebbbeed4b08b37b1e67654f2902d1dcd66d5c55c67aa09f1fd178b4cdf35d5f2
+size 357155
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/a580630605e05973fd375bad7f37cdcd b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/a580630605e05973fd375bad7f37cdcd
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/a580630605e05973fd375bad7f37cdcd differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/a6ddd6eb8118dc9b305a4eabec7a56ea b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/a6ddd6eb8118dc9b305a4eabec7a56ea
new file mode 100644
index 0000000000000000000000000000000000000000..e4b265fcd39df6da0d03ad68da99ab38c35e2532
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/a6ddd6eb8118dc9b305a4eabec7a56ea
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:05226e5266b0357f1b93105b782db2349228ec4b1cd5e45865b2493bd3343e31
+size 381108
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/a9861b362bba28989f5e129ec731ef8f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/a9861b362bba28989f5e129ec731ef8f
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/a9861b362bba28989f5e129ec731ef8f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/b4412398cf66e78932a4c6bd39bcc0ef b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/b4412398cf66e78932a4c6bd39bcc0ef
new file mode 100644
index 0000000000000000000000000000000000000000..5ec4f507a453272ad896720351e3404e8309eff1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/b4412398cf66e78932a4c6bd39bcc0ef differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..cb9ee4275cae7601a8e956a9b6783a362a8bb84f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..7c433b8671d82c492888b2181838bfa38d97c0e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754586058817970250, "commit_timestamp_nsecs": 1754586058960734375, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/d/5d4e71dde0b071edd8a1e7cd20692f4d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/d/5d4e71dde0b071edd8a1e7cd20692f4d
new file mode 100644
index 0000000000000000000000000000000000000000..3ac3e0e0738ae32d04bbf8aada5ff3c7b5728310
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/d/5d4e71dde0b071edd8a1e7cd20692f4d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..69d1c36e9d06bdc2d0f65cf1b65f8b853101a245
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/0961b9ab53cff549978be614a32e5eae b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/0961b9ab53cff549978be614a32e5eae
new file mode 100644
index 0000000000000000000000000000000000000000..4a8d18ab030a5ba19525f9353b33d3c4b35216aa
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/0961b9ab53cff549978be614a32e5eae
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:43745b33d96b57cdd50a6d4fdb7a0d4bdd9ec3abe19f8c12857c304af6e25b7f
+size 296003
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/1e05d5f9c50ed452e6479bd82ebccf6c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/1e05d5f9c50ed452e6479bd82ebccf6c
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/1e05d5f9c50ed452e6479bd82ebccf6c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/3fa7cec6d6ad91b3370a4caf7b26b8c9 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/3fa7cec6d6ad91b3370a4caf7b26b8c9
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/3fa7cec6d6ad91b3370a4caf7b26b8c9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/464db06505459dbc2985dca8530f907e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/464db06505459dbc2985dca8530f907e
new file mode 100644
index 0000000000000000000000000000000000000000..017f533fdc1bd0025b12108e1e2077d23b3f5053
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/464db06505459dbc2985dca8530f907e differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/509a9d8b82ad542e743eef334f07f035 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/509a9d8b82ad542e743eef334f07f035
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/509a9d8b82ad542e743eef334f07f035 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/537f3dc8c6bb07667700d00da9141df5 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/537f3dc8c6bb07667700d00da9141df5
new file mode 100644
index 0000000000000000000000000000000000000000..7f7017c814b435862822ffeba289b89574bf25a0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/537f3dc8c6bb07667700d00da9141df5
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9a19d53d9c6f69096a5837f828da1b62d5770073956568e06b873b62a1e88065
+size 442796
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/75ab92a52328b54e9c716fbcf1b129ec b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/75ab92a52328b54e9c716fbcf1b129ec
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/75ab92a52328b54e9c716fbcf1b129ec differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/ad92c4b61d2587bcd4c40c7c7226bfb8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/ad92c4b61d2587bcd4c40c7c7226bfb8
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/ad92c4b61d2587bcd4c40c7c7226bfb8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b65ed10eeb66c60903a301b47bd7616f3f56ec71
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..335966b459ff7249ca640f6378f44bb75b49c47c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754582864421174871, "commit_timestamp_nsecs": 1754582864561707508, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/d/2491765c7b0b216b39a5bbea7f290d4d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/d/2491765c7b0b216b39a5bbea7f290d4d
new file mode 100644
index 0000000000000000000000000000000000000000..08d1081c18c9490034a35f301bdcd83c9444fa91
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/d/2491765c7b0b216b39a5bbea7f290d4d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..ce98134390b52cb616bf7134a8d430da385fea6a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/5a31b52bb1efcf6c32cda498aae4a70d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/5a31b52bb1efcf6c32cda498aae4a70d
new file mode 100644
index 0000000000000000000000000000000000000000..bb6c09296e64edc783679ac6510cb2e1f061dbf2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/5a31b52bb1efcf6c32cda498aae4a70d
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c77332894a55abb6ae0326a6d4a82d4cd33290b0be290d674c16768241e7122c
+size 441180
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/746e252972ecd1b33cb93585a9b78622 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/746e252972ecd1b33cb93585a9b78622
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/746e252972ecd1b33cb93585a9b78622 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/7c381b146babca23b655d2c347f1d8e9 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/7c381b146babca23b655d2c347f1d8e9
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/7c381b146babca23b655d2c347f1d8e9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/a041bcef4aead068c7da5de93ff57de5 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/a041bcef4aead068c7da5de93ff57de5
new file mode 100644
index 0000000000000000000000000000000000000000..2910fbc6db2094404ed932c7d9bcdb23beb99906
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/a041bcef4aead068c7da5de93ff57de5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/a729b4f0d76c0c377fa852cd688e259a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/a729b4f0d76c0c377fa852cd688e259a
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/a729b4f0d76c0c377fa852cd688e259a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/c05b1e42e4179ba8e0cfb94ece170139 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/c05b1e42e4179ba8e0cfb94ece170139
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/c05b1e42e4179ba8e0cfb94ece170139 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/ed557e6ea73f6e4df18c7c0a027b07a9 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/ed557e6ea73f6e4df18c7c0a027b07a9
new file mode 100644
index 0000000000000000000000000000000000000000..4274e073a19acdc8b9261d366576fa854fa39870
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/ed557e6ea73f6e4df18c7c0a027b07a9
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0e55788085acc4b83ae24b3bf4b791149b3f21c5cdc2faae538ea5e7c511a65a
+size 295437
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..26a2d9bf0407d8d91e6294042e05eac1d1ceb687
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..29b840f5227346f69cc38fa32a94f05a238121b0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583160977526610, "commit_timestamp_nsecs": 1754583161151607684, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/d/c53db6804e5954e476728e6cfe371ba8 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/d/c53db6804e5954e476728e6cfe371ba8
new file mode 100644
index 0000000000000000000000000000000000000000..23dca39765c48737b835100fd9cfaa44834afdcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/d/c53db6804e5954e476728e6cfe371ba8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..ce4499a4a1681a6b4bb1ed21cb9f434a125d76aa
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/0f759a2ccc6311f5e38d2cc3e4bc38af b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/0f759a2ccc6311f5e38d2cc3e4bc38af
new file mode 100644
index 0000000000000000000000000000000000000000..aa8b2e1760e9b35afa14e6700514a6d88c334d55
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/0f759a2ccc6311f5e38d2cc3e4bc38af
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5d8ff86c16bd24e10cbafb0ab32b5dd0b788f660d01cb78dc779d8ab9dffbd8a
+size 442238
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/5eb90b6b2652acbe422bef872f0d77c1 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/5eb90b6b2652acbe422bef872f0d77c1
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/5eb90b6b2652acbe422bef872f0d77c1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/73d4f1a6cc298e911e326be7ce8cc2ee b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/73d4f1a6cc298e911e326be7ce8cc2ee
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/73d4f1a6cc298e911e326be7ce8cc2ee differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/ab0ec984cf2d0314629e9189f508d002 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/ab0ec984cf2d0314629e9189f508d002
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/ab0ec984cf2d0314629e9189f508d002 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/b2f6385b500e14b5bbfee8b149509ecf b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/b2f6385b500e14b5bbfee8b149509ecf
new file mode 100644
index 0000000000000000000000000000000000000000..3c2af63cbe23dd8fbc0ad669a8ecc5f7332fceca
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/b2f6385b500e14b5bbfee8b149509ecf
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4bf67cbd05a1e067e8e79d9fd05ccdfb445391b683cd4a19b1ff179343fc034c
+size 295525
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/b4ef3b2be7a5016d10f03ce7eda5a950 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/b4ef3b2be7a5016d10f03ce7eda5a950
new file mode 100644
index 0000000000000000000000000000000000000000..5d77b87ae3ad61174b012dddf58ab57488b0f3d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/b4ef3b2be7a5016d10f03ce7eda5a950 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/ca0ee342b53c987fc22fe2f6f8ef9ef0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/ca0ee342b53c987fc22fe2f6f8ef9ef0
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/ca0ee342b53c987fc22fe2f6f8ef9ef0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/da7bb203e11663fc209d600fff0f4391 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/da7bb203e11663fc209d600fff0f4391
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/da7bb203e11663fc209d600fff0f4391 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0accb7bf71765c9411520cc9333245c6abbaaeff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..3203196aada5b89af6b236a1f4ba37bf683579b2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583193721682288, "commit_timestamp_nsecs": 1754583193860004144, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/d/4a7bdfac0f4defd95cc8ec247d9c66ac b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/d/4a7bdfac0f4defd95cc8ec247d9c66ac
new file mode 100644
index 0000000000000000000000000000000000000000..e0fd2cbec067d191f89ecbf081600389374ca1b6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/d/4a7bdfac0f4defd95cc8ec247d9c66ac differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..41e142c3a184d2d653b4156cd78b22c9063fcd63
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/090a30c5dab266dbb4974e65663aed06 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/090a30c5dab266dbb4974e65663aed06
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/090a30c5dab266dbb4974e65663aed06 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/5912cb59a93d62a1d4fd33b8c594a98a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/5912cb59a93d62a1d4fd33b8c594a98a
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/5912cb59a93d62a1d4fd33b8c594a98a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/6190d5a6e8107e3b9c10d0c98df5cf3c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/6190d5a6e8107e3b9c10d0c98df5cf3c
new file mode 100644
index 0000000000000000000000000000000000000000..3485dbe0841a53c77aacaeb43edb297ac9c72de3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/6190d5a6e8107e3b9c10d0c98df5cf3c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2780cee7a08379c11bcab80f75933bd0714c267845ddc5a1ae66a89b1b95631f
+size 577518
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/9d69e4f4ce030e9fe49fbf13fa92718f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/9d69e4f4ce030e9fe49fbf13fa92718f
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/9d69e4f4ce030e9fe49fbf13fa92718f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/b69cc1e35d9dc1ea93c6b64e612bf67d b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/b69cc1e35d9dc1ea93c6b64e612bf67d
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/b69cc1e35d9dc1ea93c6b64e612bf67d differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/cd957f850e8afe0b9029f4a60ebf83be b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/cd957f850e8afe0b9029f4a60ebf83be
new file mode 100644
index 0000000000000000000000000000000000000000..9418142025b295780238490f4dc9ac353092d553
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/cd957f850e8afe0b9029f4a60ebf83be
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9d3e5281f839ae8a6af4839972573848623f39d9c355d2512438c90aa802b9b5
+size 159369
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/f933bbb3df1d4b6f13b9a56ff5797804 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/f933bbb3df1d4b6f13b9a56ff5797804
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/f933bbb3df1d4b6f13b9a56ff5797804 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..7475219e42f8af672e0290debc4500ca45be71ca
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..a1c445607ec5a7807e2a2440476f5cd067b66a4c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583226781985282, "commit_timestamp_nsecs": 1754583226955681246, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/d/a987ded4bddb1f44c02d6a518c55618f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/d/a987ded4bddb1f44c02d6a518c55618f
new file mode 100644
index 0000000000000000000000000000000000000000..a9dd9ca6e2a5ea2dfb6e0b2db35d31eddbde65c6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/d/a987ded4bddb1f44c02d6a518c55618f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9c25e294fbd606ce853215c98c1a00c6e00c249e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/161c954fbb0c9176e73739427ea338f7 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/161c954fbb0c9176e73739427ea338f7
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/161c954fbb0c9176e73739427ea338f7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/2fd69da09a4a9e2532118d516e42c19b b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/2fd69da09a4a9e2532118d516e42c19b
new file mode 100644
index 0000000000000000000000000000000000000000..15f3ba6c08404f3f00958ebdf2aeecf1df9b8d18
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/2fd69da09a4a9e2532118d516e42c19b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d413e18ba96bb9ac08cdc7fcde736baa754f693f2a631939dea4763d3adcc2f7
+size 286984
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/660e9fbb48e9b77a0700c5272131494c b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/660e9fbb48e9b77a0700c5272131494c
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/660e9fbb48e9b77a0700c5272131494c differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/ab37fb9dea1d56c5e322f7b2643c0329 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/ab37fb9dea1d56c5e322f7b2643c0329
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/ab37fb9dea1d56c5e322f7b2643c0329 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/b2d85c6cea1b06e62babccfc19e1e47a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/b2d85c6cea1b06e62babccfc19e1e47a
new file mode 100644
index 0000000000000000000000000000000000000000..1d3144b4b0b0d04d2f779273f9c4571a6d111c4f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/b2d85c6cea1b06e62babccfc19e1e47a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3ce54b8f9ede5982702fc1c0add7836b441349846b132d1219f814b9bc04d269
+size 450929
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/b30ba21ec2701e58a7aa058c7b8d0396 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/b30ba21ec2701e58a7aa058c7b8d0396
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/b30ba21ec2701e58a7aa058c7b8d0396 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/bc4ad98f2096ddf28a2a3bc917d132e5 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/bc4ad98f2096ddf28a2a3bc917d132e5
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/bc4ad98f2096ddf28a2a3bc917d132e5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/df344dd3d6b5a6348c264230e0ddeed1 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/df344dd3d6b5a6348c264230e0ddeed1
new file mode 100644
index 0000000000000000000000000000000000000000..cf28d01512c791ddb7ae8a982dfaef725c91ac02
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/df344dd3d6b5a6348c264230e0ddeed1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..ab3b4edd2edc46130b7486a5a73bb59c4ef95401
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..2a3b0a39c80b0b0c9abebcc7f7806d21ed0ec11b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583259928119261, "commit_timestamp_nsecs": 1754583260076653602, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/d/5efba6c57f667ae82a2a44d04fd327f4 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/d/5efba6c57f667ae82a2a44d04fd327f4
new file mode 100644
index 0000000000000000000000000000000000000000..3c5a238b59844664de9106871561c2edb7db5326
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/d/5efba6c57f667ae82a2a44d04fd327f4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0c7c8938421938d141077e709d546fa534c2be40
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/01c027ee5ac4364351554f6cbfffafda b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/01c027ee5ac4364351554f6cbfffafda
new file mode 100644
index 0000000000000000000000000000000000000000..39e06480490099071210c830587cc2330c1916b2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/01c027ee5ac4364351554f6cbfffafda differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/11f2f18926f9d238004c5d099dc64be3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/11f2f18926f9d238004c5d099dc64be3
new file mode 100644
index 0000000000000000000000000000000000000000..bb7cf73471f29d4f6e133b731b1dd0deb4b8219d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/11f2f18926f9d238004c5d099dc64be3
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f49ba6a85988a2ca64abf6f7981355f80664951f4b7380b3e3436c95d0c7a942
+size 407877
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/255ea22352d1e22cca48f8b5962bd417 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/255ea22352d1e22cca48f8b5962bd417
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/255ea22352d1e22cca48f8b5962bd417 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/e6447a432bdb659b243a5ef5308b2f57 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/e6447a432bdb659b243a5ef5308b2f57
new file mode 100644
index 0000000000000000000000000000000000000000..bce6ec8f964f39d0731228ab543b737f3049b524
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/e6447a432bdb659b243a5ef5308b2f57
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3e852c3709831dcf1950599c96620eee0d0910cb20d90996e048f9467aaedb30
+size 329570
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/f64648d1e0cf6bb57d7ecf33f4c6e75a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/f64648d1e0cf6bb57d7ecf33f4c6e75a
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/f64648d1e0cf6bb57d7ecf33f4c6e75a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/f9924bff6912d5b95dec0d6e02decd88 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/f9924bff6912d5b95dec0d6e02decd88
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/f9924bff6912d5b95dec0d6e02decd88 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/fa77d511b01f9c156ddfc45d4f48d3f3 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/fa77d511b01f9c156ddfc45d4f48d3f3
new file mode 100644
index 0000000000000000000000000000000000000000..65529f6532287fad4b450ca493011b55e8bf7f92
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/fa77d511b01f9c156ddfc45d4f48d3f3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c00d1641e18dd8a46536d49a847ac6fa03476b86
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..2270b5070a65f2747c8ab25cd0795a1fa1da44bb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583293083948338, "commit_timestamp_nsecs": 1754583293269001205, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/d/4f7e810e3656ecf02bfe0bd6729c822f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/d/4f7e810e3656ecf02bfe0bd6729c822f
new file mode 100644
index 0000000000000000000000000000000000000000..67515311eb83cb705d309a7f47608ca217809c76
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/d/4f7e810e3656ecf02bfe0bd6729c822f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..132d0a914a2d35300b76767570950488312acc0b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/46a429fad5fa2872c2e8a12ea643f8c4 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/46a429fad5fa2872c2e8a12ea643f8c4
new file mode 100644
index 0000000000000000000000000000000000000000..31493c73f3cd1e950fe548ec43cf602f7a739b50
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/46a429fad5fa2872c2e8a12ea643f8c4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/6729c14d9cceed4fc1637aaecf9b2646 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/6729c14d9cceed4fc1637aaecf9b2646
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/6729c14d9cceed4fc1637aaecf9b2646 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/7019856da506ae0d7194fe6e2b10854a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/7019856da506ae0d7194fe6e2b10854a
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/7019856da506ae0d7194fe6e2b10854a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/788d690ddd3f698ccad42befa041015a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/788d690ddd3f698ccad42befa041015a
new file mode 100644
index 0000000000000000000000000000000000000000..3d61ea64a1888001fe0bbfbbe504213f0fbdfd3e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/788d690ddd3f698ccad42befa041015a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c9cdb179bca573c4a585cf102e4ecad2c8166d7c8db032fc8da85ca30d13a4a5
+size 413126
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/9a4c3183a3d1a596929b6492fe8dcc0a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/9a4c3183a3d1a596929b6492fe8dcc0a
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/9a4c3183a3d1a596929b6492fe8dcc0a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/c57b5e88251d9485b28167e64cf33f6e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/c57b5e88251d9485b28167e64cf33f6e
new file mode 100644
index 0000000000000000000000000000000000000000..f0a53ec5eb992e986cf5d3f2caf10f721f995bd5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/c57b5e88251d9485b28167e64cf33f6e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:582476ffc3f6d35775492c6527924776fb50dd8fee7f6db08356c08185aac62a
+size 253118
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/caa00e5b53768d5cf20021cfab93242a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/caa00e5b53768d5cf20021cfab93242a
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/caa00e5b53768d5cf20021cfab93242a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/d5aa5990dc0f6c5c3b826fb932234c5a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/d5aa5990dc0f6c5c3b826fb932234c5a
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/d5aa5990dc0f6c5c3b826fb932234c5a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9549b640eadb3c5dd047f7eb9401b7c3339abb53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..6689feac33c947ef39c720d80644eab301b25f0e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754583326250177755, "commit_timestamp_nsecs": 1754583326440613637, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/_METADATA b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/_sharding b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/d/933f516f8abac5be84a4b204a6344d1f b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/d/933f516f8abac5be84a4b204a6344d1f
new file mode 100644
index 0000000000000000000000000000000000000000..071daadd007a140bcee79c2f8078dd6e2d55fd5b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/d/933f516f8abac5be84a4b204a6344d1f differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0291cacbdad15fe13af1fc011bdae16320383f51
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/197098672dfbd2d4278a296e2ce4063e b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/197098672dfbd2d4278a296e2ce4063e
new file mode 100644
index 0000000000000000000000000000000000000000..f01969c9c727ca4bdf8789c8fe795c070c98121d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/197098672dfbd2d4278a296e2ce4063e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:225dd9e6c42d26c6ddffb272e82a4ef764d44e220798e4e46f7da799275034d9
+size 286982
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/284604a1fea6067af72b1c0ece88be5a b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/284604a1fea6067af72b1c0ece88be5a
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/284604a1fea6067af72b1c0ece88be5a differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/48f6f3175bf1099a6103d71b022e3ba2 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/48f6f3175bf1099a6103d71b022e3ba2
new file mode 100644
index 0000000000000000000000000000000000000000..ec745d97353f97209ef50f0eb738585830219f53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/48f6f3175bf1099a6103d71b022e3ba2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/537a367324133fb97052f12e23c17696 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/537a367324133fb97052f12e23c17696
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/537a367324133fb97052f12e23c17696 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/67dbd5066111ca9f17e2eaf937e62254 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/67dbd5066111ca9f17e2eaf937e62254
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/67dbd5066111ca9f17e2eaf937e62254 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/80b6e48e813b6e65d76ff22b3e865c07 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/80b6e48e813b6e65d76ff22b3e865c07
new file mode 100644
index 0000000000000000000000000000000000000000..ab8c18312738e81ec882ea756f565687811e77ea
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/80b6e48e813b6e65d76ff22b3e865c07
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c90466e4d5dc2b313d383b4c1900c769b69eba41df50a22a04db069be4953b86
+size 451012
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/a7a5a42690540542a81a38a1b2732b23 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/a7a5a42690540542a81a38a1b2732b23
new file mode 100644
index 0000000000000000000000000000000000000000..4016906166fb41dd29ad78d62b36db32de15da53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/a7a5a42690540542a81a38a1b2732b23 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/eec35ac8a40716fec48469b977b35888 b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/eec35ac8a40716fec48469b977b35888
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/eec35ac8a40716fec48469b977b35888 differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..bdb8ce3f51e5f7975a53442004f7facb362decfd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/config.yaml b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/config.yaml
new file mode 100644
index 0000000000000000000000000000000000000000..220c36ee2ec340d1cc670c7374e935cfeab99a68
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/config.yaml
@@ -0,0 +1,1040 @@
+_wandb:
+    value:
+        cli_version: 0.21.0
+        e:
+            mdhm7ax1vj6umwzz7dpb0eix0yt41u7o:
+                args:
+                    - train
+                    - corner
+                    - --wall-prob
+                    - "0.0"
+                    - --env-size
+                    - "13"
+                    - --log-gifs
+                    - --env-layout
+                    - tree
+                    - --wandb-project
+                    - jaxgmg-wall-prob-tree
+                    - --wandb-name
+                    - wp:0.0_alpha:0
+                    - --num-total-env-steps
+                    - "200_000_000"
+                    - --prob-shift
+                    - "0"
+                    - --keep-all-checkpoints
+                    - --num-cycles-per-checkpoint
+                    - "64"
+                cpu_count: 24
+                cpu_count_logical: 48
+                cudaVersion: "12.8"
+                disk:
+                    /:
+                        total: "102005473280"
+                        used: "43219062784"
+                email: david@timaeus.co
+                executable: /workspace/timaeus/.venv/bin/python
+                git:
+                    commit: 7415666d6a2e5d63453ea9fadaf6cecf63256a59
+                    remote: git@github.com:timaeus-research/timaeus.git
+                gpu: NVIDIA GeForce RTX 3090
+                gpu_count: 1
+                gpu_nvidia:
+                    - architecture: Ampere
+                      cudaCores: 10496
+                      memoryTotal: "25769803776"
+                      name: NVIDIA GeForce RTX 3090
+                      uuid: GPU-43bd7524-bf94-73a9-1460-2b36dd09a208
+                host: 6e66e15ca303
+                memory:
+                    total: "135066595328"
+                os: Linux-5.4.0-216-generic-x86_64-with-glibc2.39
+                program: -m jaxgmg.__main__
+                python: CPython 3.11.11
+                root: /workspace/timaeus/projects/rl
+                startedAt: "2025-08-07T16:04:01.892209Z"
+                writerId: mdhm7ax1vj6umwzz7dpb0eix0yt41u7o
+        m:
+            - "1": ppo/avg_grad_norm_pre_clip
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": step/ppo-update-after
+              "6":
+                - 3
+              "7": []
+            - "1": ppo/max/max_actor_approxkl1
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_actor_loss
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/lvl_avg_episode_length_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": step/env-step-all-after
+              "6":
+                - 3
+              "7": []
+            - "1": eval-batch-orig/proxy_corner/avg_reward_per_step
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/visit_patterns/avg_first_visit_time
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/lvl_avg_episode_length_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/rollouts_gif
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/lvl_avg_return_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/lvl_reward_per_step_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/lvl_avg_episode_length_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/levels_cheese_is_in_corner_avg
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/visit_patterns/num_replay_batches
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/avg_reward_per_step
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/proxy_corner/lvl_avg_return_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/lvl_avg_episode_length_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/proxy_corner/avg_reward_per_step
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/visit_patterns/avg_last_visit_time
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/proxy_corner/avg_reward_per_step
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/rollouts_gif
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_grad_norm_pre_clip
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/proxy_corner/lvl_avg_return_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/proxy_corner/avg_avg_return
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/lvl_avg_return_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_actor_loss
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_critic_loss
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_critic_clipfrac
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/avg_avg_episode_length
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/avg_benchmark_regret
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/proxy_corner/lvl_reward_per_step_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/rollouts_gif
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/avg_avg_return
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_critic_clipfrac
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_critic_loss
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_entropy
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_loss
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/avg_avg_return
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/mouse-cheese_dist_solvable_avg
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/proxy_corner/lvl_reward_per_step_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_actor_approxkl1
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_advantage
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/proxy_corner/lvl_reward_per_step_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/num_walls_wavg
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_actor_clipfrac
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/avg_benchmark_return
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/lvl_reward_per_step_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/mouse-cheese_dist_finite_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/levels16_img
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/visit_patterns/first_visit_time_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/lvl_avg_episode_length_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/prop_walls_wavg
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/avg_benchmark_regret
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/shift_proportion
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/lvl_avg_return_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/proxy_corner/lvl_reward_per_step_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_actor_approxkl3
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/avg_avg_return
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/cheese-corne_dist_avg
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/cheese-corne_dist_wavg
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/solvable_num
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/prop_walls_avg
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/eta
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_entropy
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_loss
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/mouse-cheese_dist_solvable_wavg
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/avg_avg_return
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/proxy_corner/avg_avg_return
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/lvl_benchmark_regret_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/proxy_corner/lvl_avg_return_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/avg_avg_episode_length
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/rollouts_gif
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/avg_avg_episode_length
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/solvable_wavg
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_actor_clipfrac
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_advantage
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/lvl_reward_per_step_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/avg_reward_per_step
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/shift_proportion
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_critic_clipfrac
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/lvl_reward_per_step_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/proxy_corner/lvl_avg_return_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/proxy_corner/lvl_avg_return_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_grad_norm_pre_clip
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/proxy_corner/avg_reward_per_step
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_actor_loss
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/lvl_avg_return_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/lvl_benchmark_regret_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/avg_benchmark_return
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/avg_reward_per_step
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/lvl_avg_episode_length_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/lvl_avg_return_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/lvl_benchmark_return_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/proxy_corner/avg_reward_per_step
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/scoring/avg_scores
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_advantage
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/solvable_avg
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/visit_patterns/last_visit_time_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_actor_clipfrac
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_critic_loss
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/proxy_corner/lvl_reward_per_step_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/num_walls_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/visit_patterns/prev_batch_level_ids_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_entropy
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/avg_avg_return
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/lvl_benchmark_return_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/lvl_reward_per_step_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/proxy_corner/lvl_avg_return_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/mouse-cheese_dist_finite_wavg
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/avg_avg_return
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/rollouts_gif
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/avg_reward_per_step
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/avg_avg_episode_length
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/proxy_corner/avg_avg_return
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/lvl_reward_per_step_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/avg_avg_episode_length
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/proxy_corner/avg_reward_per_step
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/avg_avg_episode_length
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/proxy_corner/avg_avg_return
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/rollouts_gif
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/cheese-corner_dist_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_actor_approxkl1
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_actor_approxkl3
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/lvl_benchmark_return_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/proxy_corner/avg_avg_return
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/avg_benchmark_regret
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/mouse-cheese_dist_finite_avg
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_actor_approxkl3
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_loss
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/shift_proportion
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/avg_reward_per_step
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/num_walls_avg
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/prop_walls_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/avg_benchmark_return
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/proxy_corner/lvl_reward_per_step_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/avg_reward_per_step
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/lvl_avg_return_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/lvl_benchmark_regret_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/proxy_corner/avg_avg_return
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/scoring/scores_hist
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+        python_version: 3.11.11
+        t:
+            "1":
+                - 12
+                - 45
+            "2":
+                - 12
+                - 45
+            "3":
+                - 2
+                - 3
+                - 7
+                - 13
+                - 14
+                - 16
+                - 61
+            "4": 3.11.11
+            "5": 0.21.0
+            "12": 0.21.0
+            "13": linux-x86_64
+chain_mutate:
+    value: true
+checkpointing:
+    value: true
+clipping:
+    value: false
+console_log:
+    value: true
+env_corner_size:
+    value: 1
+env_layout:
+    value: tree
+env_penalize_time:
+    value: false
+env_size:
+    value: 13
+env_terminate_after_corner:
+    value: false
+evals_num_env_steps:
+    value: 512
+evals_num_levels:
+    value: 256
+gif_grid_width:
+    value: 16
+img_level_of_detail:
+    value: 1
+keep_all_checkpoints:
+    value: true
+level_splayer:
+    value: mouse
+log_gifs:
+    value: true
+log_hists:
+    value: false
+log_imgs:
+    value: true
+max_num_checkpoints:
+    value: 1
+mutate_cheese:
+    value: true
+net_cnn_type:
+    value: large
+net_rnn_type:
+    value: ff
+net_width:
+    value: 256
+num_cycles_per_big_eval:
+    value: 1024
+num_cycles_per_checkpoint:
+    value: 64
+num_cycles_per_eval:
+    value: 32
+num_cycles_per_gifs:
+    value: 1024
+num_cycles_per_log:
+    value: 32
+num_env_steps_per_cycle:
+    value: 128
+num_epochs_per_cycle:
+    value: 5
+num_minibatches_per_epoch:
+    value: 4
+num_mutate_steps:
+    value: 12
+num_parallel_envs:
+    value: 256
+num_total_env_steps:
+    value: 200000000
+num_train_levels:
+    value: 2048
+obs_level_of_detail:
+    value: 0
+plr_buffer_size:
+    value: 4096
+plr_prob_replay:
+    value: 0.5
+plr_proxy_shaping:
+    value: false
+plr_proxy_shaping_coeff:
+    value: 0.5
+plr_regret_estimator:
+    value: maxmc-actor
+plr_robust:
+    value: false
+plr_staleness_coeff:
+    value: 0.1
+plr_temperature:
+    value: 0.1
+ppo_clip_eps:
+    value: 0.1
+ppo_critic_coeff:
+    value: 0.5
+ppo_entropy_coeff:
+    value: 0.001
+ppo_gae_lambda:
+    value: 0.95
+ppo_gamma:
+    value: 0.999
+ppo_lr:
+    value: 5e-05
+ppo_lr_annealing:
+    value: false
+ppo_max_grad_norm:
+    value: 0.5
+ppo_proxy_critic_coeff:
+    value: 0.5
+prob_mutate_shift:
+    value: 0
+prob_shift:
+    value: 0
+proxy_name:
+    value: proxy_corner
+seed:
+    value: 42
+train_proxy_critic:
+    value: false
+ued:
+    value: plr
+use_fixed_eval_levels:
+    value: false
+wall_prob:
+    value: 0
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_0_28c67ad9511441387f2d.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_0_28c67ad9511441387f2d.png
new file mode 100644
index 0000000000000000000000000000000000000000..d1e40a6399f8bad04c5884b8045c0a705015774b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_0_28c67ad9511441387f2d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_0_b73c8ee4e56349010606.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_0_b73c8ee4e56349010606.png
new file mode 100644
index 0000000000000000000000000000000000000000..3f7004db24f59534257ac89b54b3d69bbdb63baf
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_0_b73c8ee4e56349010606.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1024_50ad94d4b72c19463789.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1024_50ad94d4b72c19463789.png
new file mode 100644
index 0000000000000000000000000000000000000000..23b38402dec72acbeb1039b66bcb28ff17731995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1024_50ad94d4b72c19463789.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1056_b751fa3c572605b4b761.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1056_b751fa3c572605b4b761.png
new file mode 100644
index 0000000000000000000000000000000000000000..e1de8e375af5f18dfd08eccd3397cf1a11d8cec9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1056_b751fa3c572605b4b761.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1088_eabf6a4e46267496e080.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1088_eabf6a4e46267496e080.png
new file mode 100644
index 0000000000000000000000000000000000000000..134c2cf6007de25320cea89db29f2100e2e0e92d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1088_eabf6a4e46267496e080.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1120_e65bcfa59847c6718f7b.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1120_e65bcfa59847c6718f7b.png
new file mode 100644
index 0000000000000000000000000000000000000000..4b3429285922b7dba1fea67af3baf3edb2a32bf9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1120_e65bcfa59847c6718f7b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1152_74285ea42afc3872a863.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1152_74285ea42afc3872a863.png
new file mode 100644
index 0000000000000000000000000000000000000000..a0f33057a83356e716c68710bd36ed21c0ae6f0c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1152_74285ea42afc3872a863.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1184_7428cf210c93c2f6231e.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1184_7428cf210c93c2f6231e.png
new file mode 100644
index 0000000000000000000000000000000000000000..0151e87814f4cf4a31de4f8742c84d3fa5c982ff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1184_7428cf210c93c2f6231e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1216_1d275fee356e772b9535.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1216_1d275fee356e772b9535.png
new file mode 100644
index 0000000000000000000000000000000000000000..890274d1bedd8f584b18504f686f6d2e65883e7c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1216_1d275fee356e772b9535.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1248_3b2f6e531b604817532b.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1248_3b2f6e531b604817532b.png
new file mode 100644
index 0000000000000000000000000000000000000000..509fdaa2329d137df9536024b74274a3a2f2e84a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1248_3b2f6e531b604817532b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1280_aae3f9b8b1b76c3c16b8.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1280_aae3f9b8b1b76c3c16b8.png
new file mode 100644
index 0000000000000000000000000000000000000000..9d4b31ac9d786dc5f03748f3058b2e3de94918a5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1280_aae3f9b8b1b76c3c16b8.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_128_efd969534d67d7478cb8.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_128_efd969534d67d7478cb8.png
new file mode 100644
index 0000000000000000000000000000000000000000..49070aa1a5ed0880c2b7c5a1633509533b02cbe7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_128_efd969534d67d7478cb8.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1312_22ae610e1490bc428ba7.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1312_22ae610e1490bc428ba7.png
new file mode 100644
index 0000000000000000000000000000000000000000..8aee4da02a8a8a17e6291e52b9b10288200d6185
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1312_22ae610e1490bc428ba7.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1344_f0a3fbd4b1383de3adb9.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1344_f0a3fbd4b1383de3adb9.png
new file mode 100644
index 0000000000000000000000000000000000000000..dea72194f6bf88729dd335c55a41112f5b7c65e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1344_f0a3fbd4b1383de3adb9.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1376_ab9800a4b453356f309a.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1376_ab9800a4b453356f309a.png
new file mode 100644
index 0000000000000000000000000000000000000000..391fa0cdcbea73cb1fddb868b4a74db8f45fd6bd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1376_ab9800a4b453356f309a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1408_298ae3042f1b960173d6.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1408_298ae3042f1b960173d6.png
new file mode 100644
index 0000000000000000000000000000000000000000..7e4a1fcdadc541135f81a50c3eb23d1f17c12f07
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1408_298ae3042f1b960173d6.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1440_d9eb55da1eacf72a1b5e.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1440_d9eb55da1eacf72a1b5e.png
new file mode 100644
index 0000000000000000000000000000000000000000..bcbf6d9805c91a1c5b3b65d3cda685504b727760
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1440_d9eb55da1eacf72a1b5e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1472_c9b336d9b9ff95ca4706.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1472_c9b336d9b9ff95ca4706.png
new file mode 100644
index 0000000000000000000000000000000000000000..f6c3e71e37b1911d19513c1b7ec34d533d0f01d6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1472_c9b336d9b9ff95ca4706.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1504_2fa689aa0fee63709629.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1504_2fa689aa0fee63709629.png
new file mode 100644
index 0000000000000000000000000000000000000000..422a22173bce51abd9a5aee6f0d4b704e14c074b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1504_2fa689aa0fee63709629.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1536_f561e0a6e3eed17e35b1.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1536_f561e0a6e3eed17e35b1.png
new file mode 100644
index 0000000000000000000000000000000000000000..0ae325aa222aef3d483b4e80cb87ef5e5b4da2db
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1536_f561e0a6e3eed17e35b1.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1568_1db9a6db76d0936fd887.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1568_1db9a6db76d0936fd887.png
new file mode 100644
index 0000000000000000000000000000000000000000..2b8a41ad7ce787ccb0f3dec1119385922579097e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1568_1db9a6db76d0936fd887.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1600_af7561d2a324639935e4.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1600_af7561d2a324639935e4.png
new file mode 100644
index 0000000000000000000000000000000000000000..44b70420c111335c21b1028fa7951950873357c4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1600_af7561d2a324639935e4.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_160_11564c3cd37db6a506ae.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_160_11564c3cd37db6a506ae.png
new file mode 100644
index 0000000000000000000000000000000000000000..3717e34a629abad9b3a852ace4550ecd64da85a1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_160_11564c3cd37db6a506ae.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1632_b6b6e9195100165c4cbb.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1632_b6b6e9195100165c4cbb.png
new file mode 100644
index 0000000000000000000000000000000000000000..f079f73d6e54492bb6225c60587595b883666aca
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1632_b6b6e9195100165c4cbb.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1664_e271a1d3d55e756be14c.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1664_e271a1d3d55e756be14c.png
new file mode 100644
index 0000000000000000000000000000000000000000..bc8cd9976962a8507323131271fed29576c6935a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1664_e271a1d3d55e756be14c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1696_bf71844688ef1e97c120.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1696_bf71844688ef1e97c120.png
new file mode 100644
index 0000000000000000000000000000000000000000..32bb9d62d829050a4b8a4b8f4b98fac05f6c1148
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1696_bf71844688ef1e97c120.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1728_82c19ab80b45d81587fe.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1728_82c19ab80b45d81587fe.png
new file mode 100644
index 0000000000000000000000000000000000000000..fbc6ba0940a211924baddb6d0cf6f64471be3335
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1728_82c19ab80b45d81587fe.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1760_347b7a684965a361830b.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1760_347b7a684965a361830b.png
new file mode 100644
index 0000000000000000000000000000000000000000..bf89828d804391d46271e7b3333e09dd817fac1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1760_347b7a684965a361830b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1792_e3e34b1b1164817a7040.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1792_e3e34b1b1164817a7040.png
new file mode 100644
index 0000000000000000000000000000000000000000..ec9a7270a04abcd176ab86a83904e7cba41426c1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1792_e3e34b1b1164817a7040.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1824_2f5221568738b57900af.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1824_2f5221568738b57900af.png
new file mode 100644
index 0000000000000000000000000000000000000000..18ac72222579be5a7745289b877820fef6d6a6bf
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1824_2f5221568738b57900af.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1856_8dcaaad12c9d53fe0d8f.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1856_8dcaaad12c9d53fe0d8f.png
new file mode 100644
index 0000000000000000000000000000000000000000..a279a9bde4ce8bafc725bdc8a835f4e1d6c924bc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1856_8dcaaad12c9d53fe0d8f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1888_149cf4eef0d5b8838c67.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1888_149cf4eef0d5b8838c67.png
new file mode 100644
index 0000000000000000000000000000000000000000..665c47a341cd15823db64934089bc2dba6e66ac9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1888_149cf4eef0d5b8838c67.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1920_efcbe37f50bb805bb00c.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1920_efcbe37f50bb805bb00c.png
new file mode 100644
index 0000000000000000000000000000000000000000..ee5670cd40851bc73c548e7e0e997ae7199a7a93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1920_efcbe37f50bb805bb00c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_192_8f32458f652c0674bb88.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_192_8f32458f652c0674bb88.png
new file mode 100644
index 0000000000000000000000000000000000000000..e2066bb25817fe2b8db520bf88cdb0df8acdc5d9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_192_8f32458f652c0674bb88.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1952_1b2bb0a873c3da6cb710.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1952_1b2bb0a873c3da6cb710.png
new file mode 100644
index 0000000000000000000000000000000000000000..0f472c6024159651cddcceb95c65ab417748c326
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1952_1b2bb0a873c3da6cb710.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1984_3e0b2b0772064ff0e4b3.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1984_3e0b2b0772064ff0e4b3.png
new file mode 100644
index 0000000000000000000000000000000000000000..0772241bf8f64e0cfd4c7e346a61474ef8e4d81f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_1984_3e0b2b0772064ff0e4b3.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2016_e432143a828a1c43cc87.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2016_e432143a828a1c43cc87.png
new file mode 100644
index 0000000000000000000000000000000000000000..c1a4e51e3f3d77346b1e67d9d7a559b5dedda82d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2016_e432143a828a1c43cc87.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2048_a444f8e2fd3970e784b3.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2048_a444f8e2fd3970e784b3.png
new file mode 100644
index 0000000000000000000000000000000000000000..82e03b4aa1acd8d66e9ca3b5547e58be550e0a9c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2048_a444f8e2fd3970e784b3.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2080_b2e506f3f3baf0b1cbc9.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2080_b2e506f3f3baf0b1cbc9.png
new file mode 100644
index 0000000000000000000000000000000000000000..594b8c8227070f268b407eb0cb109bf21b055b35
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2080_b2e506f3f3baf0b1cbc9.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2112_1881889aeb144e907dd4.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2112_1881889aeb144e907dd4.png
new file mode 100644
index 0000000000000000000000000000000000000000..7f71d16fe4bcd952ba3a2ff7f0d80ac489f2c1bf
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2112_1881889aeb144e907dd4.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2144_4b6d5390ea4684c5f9bd.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2144_4b6d5390ea4684c5f9bd.png
new file mode 100644
index 0000000000000000000000000000000000000000..332364286a3f0f9654ad117ebb04f16814d2f482
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2144_4b6d5390ea4684c5f9bd.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2176_d9c3d1692399bc7c52d1.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2176_d9c3d1692399bc7c52d1.png
new file mode 100644
index 0000000000000000000000000000000000000000..8e9dbdb90d0fbb3b9524c5ece1d413d5dd4ff83a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2176_d9c3d1692399bc7c52d1.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2208_3ea76579b6c4b46641cd.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2208_3ea76579b6c4b46641cd.png
new file mode 100644
index 0000000000000000000000000000000000000000..8b5542aead96efedab48a343afd53ed2bfd98481
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2208_3ea76579b6c4b46641cd.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2240_c1cff49179dbdb28d72c.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2240_c1cff49179dbdb28d72c.png
new file mode 100644
index 0000000000000000000000000000000000000000..92fa73890b6fba2b7fa3b3fe81eb3c2a4c00f49c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2240_c1cff49179dbdb28d72c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_224_e53b476f8217c0173557.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_224_e53b476f8217c0173557.png
new file mode 100644
index 0000000000000000000000000000000000000000..8e41d8fa92cfd890ae060baa01b4e9addd463539
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_224_e53b476f8217c0173557.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2272_82ea91fac2c93ef52d8e.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2272_82ea91fac2c93ef52d8e.png
new file mode 100644
index 0000000000000000000000000000000000000000..5610e2e505cec25f35596487a8fa083fba9a6588
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2272_82ea91fac2c93ef52d8e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2304_83e217ce703858b04791.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2304_83e217ce703858b04791.png
new file mode 100644
index 0000000000000000000000000000000000000000..21202517d6beb68586526d12d3e784453cf9b2a1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2304_83e217ce703858b04791.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2336_2599ecc2aa3252e968e2.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2336_2599ecc2aa3252e968e2.png
new file mode 100644
index 0000000000000000000000000000000000000000..6ba3192223fa384e82dd1217ca4db2192c44c410
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2336_2599ecc2aa3252e968e2.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2368_35feabb5d20de78c7ebd.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2368_35feabb5d20de78c7ebd.png
new file mode 100644
index 0000000000000000000000000000000000000000..7442aa763eefae46c842caf02e24d1dabfa1477f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2368_35feabb5d20de78c7ebd.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2400_13c2cc6b9ec7d782b229.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2400_13c2cc6b9ec7d782b229.png
new file mode 100644
index 0000000000000000000000000000000000000000..573f3a925d7c338e9ed5ed96662920dc9cfe74f1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2400_13c2cc6b9ec7d782b229.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2432_14c67765f7c04cda34be.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2432_14c67765f7c04cda34be.png
new file mode 100644
index 0000000000000000000000000000000000000000..b8b4f67c9d706d65a7359948057f2f7897424453
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2432_14c67765f7c04cda34be.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2464_98b6b9c98e279f6552e7.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2464_98b6b9c98e279f6552e7.png
new file mode 100644
index 0000000000000000000000000000000000000000..905495b31c13c2bbf6db9b000444b29ed8deeb42
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2464_98b6b9c98e279f6552e7.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2496_cd68647a162c688e0eef.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2496_cd68647a162c688e0eef.png
new file mode 100644
index 0000000000000000000000000000000000000000..4541844f1d8104c46425a0b11c3f38e6e97d26ac
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2496_cd68647a162c688e0eef.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2528_fc4eb84fcfeba02c0396.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2528_fc4eb84fcfeba02c0396.png
new file mode 100644
index 0000000000000000000000000000000000000000..deae08f2a2c05fac410c8add17e857148c694b6a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2528_fc4eb84fcfeba02c0396.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2560_9304e3ac7469229d0e4f.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2560_9304e3ac7469229d0e4f.png
new file mode 100644
index 0000000000000000000000000000000000000000..a9b91c7cf338e4076e330b899bcdd85b37b91545
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2560_9304e3ac7469229d0e4f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_256_45009770c7665c36b233.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_256_45009770c7665c36b233.png
new file mode 100644
index 0000000000000000000000000000000000000000..70d54c50216458b9d5635a107c88d91837abc540
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_256_45009770c7665c36b233.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2592_695b82cb35d70e9f4639.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2592_695b82cb35d70e9f4639.png
new file mode 100644
index 0000000000000000000000000000000000000000..2e82d3b7da14a14f0331a4e4c6ec57e8c02b914d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2592_695b82cb35d70e9f4639.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2624_ebb12694d4af73eb9777.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2624_ebb12694d4af73eb9777.png
new file mode 100644
index 0000000000000000000000000000000000000000..48c5136b57f4d1db7eef000b0664c7f5e60ec689
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2624_ebb12694d4af73eb9777.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2656_01db97bf9ec9fa52ec1a.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2656_01db97bf9ec9fa52ec1a.png
new file mode 100644
index 0000000000000000000000000000000000000000..c3b2894348b12f763e0946816628a1aef489a26f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2656_01db97bf9ec9fa52ec1a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2688_21ddcb053d034809f89a.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2688_21ddcb053d034809f89a.png
new file mode 100644
index 0000000000000000000000000000000000000000..c56c37d2d8303e0019ad5c2bcfb25f0c835aefa7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2688_21ddcb053d034809f89a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2720_4c81bcc833debc22fa20.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2720_4c81bcc833debc22fa20.png
new file mode 100644
index 0000000000000000000000000000000000000000..b1ac956bc5793f2c28b65439fbc3326b5b9502e6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2720_4c81bcc833debc22fa20.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2752_46651b02ec1be349c3ea.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2752_46651b02ec1be349c3ea.png
new file mode 100644
index 0000000000000000000000000000000000000000..e773cad932cf34d2bfb45d7532726d36e0160148
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2752_46651b02ec1be349c3ea.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2784_a88ae6492024a25460ca.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2784_a88ae6492024a25460ca.png
new file mode 100644
index 0000000000000000000000000000000000000000..a6a37769566f5ce072cbe1265ffd78807ab4c37d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2784_a88ae6492024a25460ca.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2816_7459b26653a7373472e6.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2816_7459b26653a7373472e6.png
new file mode 100644
index 0000000000000000000000000000000000000000..c47f258eb8c2266682e56392e4c049566efce259
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2816_7459b26653a7373472e6.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2848_4c5b1be67552f650900b.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2848_4c5b1be67552f650900b.png
new file mode 100644
index 0000000000000000000000000000000000000000..d8e4a5c5aff1170eb238ae0fc825e35392905d35
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2848_4c5b1be67552f650900b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2880_10d5fb4418986706aae8.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2880_10d5fb4418986706aae8.png
new file mode 100644
index 0000000000000000000000000000000000000000..5383aacfc9218e38d6de47a5b8338f713b834e84
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2880_10d5fb4418986706aae8.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_288_53c7478a7ec8125c3bde.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_288_53c7478a7ec8125c3bde.png
new file mode 100644
index 0000000000000000000000000000000000000000..9610f6f2ef02261e712245d97d19c6f6e1ea0258
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_288_53c7478a7ec8125c3bde.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2912_4e875df6c5ad97eacdc7.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2912_4e875df6c5ad97eacdc7.png
new file mode 100644
index 0000000000000000000000000000000000000000..97a8fa8f518b310036c256610a1af8d7708a7f13
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2912_4e875df6c5ad97eacdc7.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2944_9fce3fdd1c39172e8ed0.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2944_9fce3fdd1c39172e8ed0.png
new file mode 100644
index 0000000000000000000000000000000000000000..7f1f6927a55e308f9e7abfd5cb508064694334c2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2944_9fce3fdd1c39172e8ed0.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2976_c633ffee60e4a59ebb03.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2976_c633ffee60e4a59ebb03.png
new file mode 100644
index 0000000000000000000000000000000000000000..02c5801d7e3bfe85c5cf662834567c027d6cd0bc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_2976_c633ffee60e4a59ebb03.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3008_7106db1c8f3f96abb43c.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3008_7106db1c8f3f96abb43c.png
new file mode 100644
index 0000000000000000000000000000000000000000..955aa8debdbf85946607bc73faa2545398ae0c95
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3008_7106db1c8f3f96abb43c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3040_7887640ed8b9d1921c25.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3040_7887640ed8b9d1921c25.png
new file mode 100644
index 0000000000000000000000000000000000000000..b144ba2fe0bb99b07ff17eb41f3d07b3221b297b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3040_7887640ed8b9d1921c25.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3072_d25ae99e3bf302ae9cdc.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3072_d25ae99e3bf302ae9cdc.png
new file mode 100644
index 0000000000000000000000000000000000000000..d3188eed885c2592b343dd8c4bc68efedf18da5a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3072_d25ae99e3bf302ae9cdc.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3104_88e55b46939377b7091c.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3104_88e55b46939377b7091c.png
new file mode 100644
index 0000000000000000000000000000000000000000..4f1ffc34617d69085cdc03e927e692dacb15b447
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3104_88e55b46939377b7091c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3136_3ff6fdbc8177abb91396.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3136_3ff6fdbc8177abb91396.png
new file mode 100644
index 0000000000000000000000000000000000000000..2afe99bc8f73f9aeca8b4237b6de1997d5ea09c1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3136_3ff6fdbc8177abb91396.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3168_963f571cda81fe7d3053.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3168_963f571cda81fe7d3053.png
new file mode 100644
index 0000000000000000000000000000000000000000..93b29bf6e4abef5a31e91331581802dde0f56f76
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3168_963f571cda81fe7d3053.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3200_a0f534b17e55255eb772.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3200_a0f534b17e55255eb772.png
new file mode 100644
index 0000000000000000000000000000000000000000..c9e7edba00f6337de3b2e774435f5065224f7961
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3200_a0f534b17e55255eb772.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_320_6bcd6f95fec46cfc2bb9.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_320_6bcd6f95fec46cfc2bb9.png
new file mode 100644
index 0000000000000000000000000000000000000000..38eb6e3c7afbfe9f29b91d7d9e983ab414283f17
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_320_6bcd6f95fec46cfc2bb9.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3232_53076154321a55211cbc.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3232_53076154321a55211cbc.png
new file mode 100644
index 0000000000000000000000000000000000000000..80b28958de3cfcdee17eaf2d8f9a55d899887965
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3232_53076154321a55211cbc.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3264_bdc7016a5e81f6b80c46.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3264_bdc7016a5e81f6b80c46.png
new file mode 100644
index 0000000000000000000000000000000000000000..f03e43b4609d5eecc2b53b0f78eecf66ee47fe7d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3264_bdc7016a5e81f6b80c46.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3296_0b3d945149c484021f76.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3296_0b3d945149c484021f76.png
new file mode 100644
index 0000000000000000000000000000000000000000..67abd1c04cc0bee89d264cfb2a6a63dff4d6c324
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3296_0b3d945149c484021f76.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_32_47d2b0043faa62d98b4d.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_32_47d2b0043faa62d98b4d.png
new file mode 100644
index 0000000000000000000000000000000000000000..e7b74a83713dfac8c5fd2df140b380bf990af0d9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_32_47d2b0043faa62d98b4d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3328_365bdc40575711e833d7.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3328_365bdc40575711e833d7.png
new file mode 100644
index 0000000000000000000000000000000000000000..1267ceb82d1d23fe3c52dc10a5ee95d75d4d43a4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3328_365bdc40575711e833d7.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3360_6a3a09db5c8f19ca6509.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3360_6a3a09db5c8f19ca6509.png
new file mode 100644
index 0000000000000000000000000000000000000000..fa86d0b9d3092ff7e9e48c2e364bd7e2a59786da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3360_6a3a09db5c8f19ca6509.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3392_4f619b8c256d5bc1955a.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3392_4f619b8c256d5bc1955a.png
new file mode 100644
index 0000000000000000000000000000000000000000..9bab342f63a5b1da8fe1d5a662a2b65faa429c0f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3392_4f619b8c256d5bc1955a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3424_fce040f033aa7a2af8ec.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3424_fce040f033aa7a2af8ec.png
new file mode 100644
index 0000000000000000000000000000000000000000..2e6c95baaf54b250280629164362e95e28e51c40
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3424_fce040f033aa7a2af8ec.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3456_05353813addcf8e014b3.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3456_05353813addcf8e014b3.png
new file mode 100644
index 0000000000000000000000000000000000000000..33370dc1e1535430655922637d077683dbb2dc1a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3456_05353813addcf8e014b3.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3488_e30936676509234eafa0.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3488_e30936676509234eafa0.png
new file mode 100644
index 0000000000000000000000000000000000000000..b378cb0ac420a33dd384647e152da5e751471953
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3488_e30936676509234eafa0.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3520_5398f4567face08f569f.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3520_5398f4567face08f569f.png
new file mode 100644
index 0000000000000000000000000000000000000000..fc6c9e8512dd33e673de32d8a8fd10c0be53485a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3520_5398f4567face08f569f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_352_50652bc59950265bbaa1.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_352_50652bc59950265bbaa1.png
new file mode 100644
index 0000000000000000000000000000000000000000..c32c533ab727cad529b1872077404af9eae4c8e2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_352_50652bc59950265bbaa1.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3552_9a18f8b8b5f30857fff0.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3552_9a18f8b8b5f30857fff0.png
new file mode 100644
index 0000000000000000000000000000000000000000..81e5fa83dbc1862654f9326ead20fe36b0e72f67
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3552_9a18f8b8b5f30857fff0.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3584_38fdde6b2d4bb928d4ff.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3584_38fdde6b2d4bb928d4ff.png
new file mode 100644
index 0000000000000000000000000000000000000000..381a15e6b99950afb7dc5f1183f306d3fe3001fa
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3584_38fdde6b2d4bb928d4ff.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3616_ace255304a06b13cdb63.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3616_ace255304a06b13cdb63.png
new file mode 100644
index 0000000000000000000000000000000000000000..9d4cfa7a067104a14be9b8a9330e193f778c9b34
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3616_ace255304a06b13cdb63.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3648_997580daca615625c601.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3648_997580daca615625c601.png
new file mode 100644
index 0000000000000000000000000000000000000000..ae0aa404809739a504de15507bd48ac476971c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3648_997580daca615625c601.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3680_cc862e3aabb4964099f4.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3680_cc862e3aabb4964099f4.png
new file mode 100644
index 0000000000000000000000000000000000000000..69931c725798c095f2b73e9a917561ea4ff9ddc4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3680_cc862e3aabb4964099f4.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3712_00aff01e3df07d04722a.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3712_00aff01e3df07d04722a.png
new file mode 100644
index 0000000000000000000000000000000000000000..ccc87302bc12d177491aae704ebab99d5c42d2fe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3712_00aff01e3df07d04722a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3744_4af34b8a2c41c64ef66e.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3744_4af34b8a2c41c64ef66e.png
new file mode 100644
index 0000000000000000000000000000000000000000..f400e78b44af28dcaa952e9e2f5d6f3c56c97abc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3744_4af34b8a2c41c64ef66e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3776_736e35848128eea6b60e.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3776_736e35848128eea6b60e.png
new file mode 100644
index 0000000000000000000000000000000000000000..bb122800b49e7a5d082de5d6763e02ab5bf82321
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3776_736e35848128eea6b60e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3808_6646cd5e670217617e05.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3808_6646cd5e670217617e05.png
new file mode 100644
index 0000000000000000000000000000000000000000..eaf46532ad530cc4d1102906d35097a678b37f8d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3808_6646cd5e670217617e05.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3840_0e0b4e99da2d81b3f8a5.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3840_0e0b4e99da2d81b3f8a5.png
new file mode 100644
index 0000000000000000000000000000000000000000..8eecae7c5b953d824e8d5930214a44b6a8401b5f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3840_0e0b4e99da2d81b3f8a5.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_384_b403d556068046f8c2d7.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_384_b403d556068046f8c2d7.png
new file mode 100644
index 0000000000000000000000000000000000000000..f3766db1a859d533609bcb3b1c802f0e3ba24a58
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_384_b403d556068046f8c2d7.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3872_1f7a7ffc92543ff0d6e3.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3872_1f7a7ffc92543ff0d6e3.png
new file mode 100644
index 0000000000000000000000000000000000000000..9bd06d35a5002f0548afd45c4b8e564486fb74b3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3872_1f7a7ffc92543ff0d6e3.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3904_27a8e0311bf8f145053b.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3904_27a8e0311bf8f145053b.png
new file mode 100644
index 0000000000000000000000000000000000000000..1dffb5524f090a3b5cdfa2f0bae403792c158da4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3904_27a8e0311bf8f145053b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3936_14a15d81448cf603f9f5.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3936_14a15d81448cf603f9f5.png
new file mode 100644
index 0000000000000000000000000000000000000000..434425973da0fec4937260cfbcabf2a7ba4c0e49
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3936_14a15d81448cf603f9f5.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3968_bda0cd7810271ad62f6c.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3968_bda0cd7810271ad62f6c.png
new file mode 100644
index 0000000000000000000000000000000000000000..8eeb638ad179938d872172e99b03fd4304e264d6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_3968_bda0cd7810271ad62f6c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4000_9a1a2d4c482c9bad795c.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4000_9a1a2d4c482c9bad795c.png
new file mode 100644
index 0000000000000000000000000000000000000000..1268bcba0fc6e863f937a86e26efe2c34d3db9f1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4000_9a1a2d4c482c9bad795c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4032_1657ad5398b5f3faf15b.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4032_1657ad5398b5f3faf15b.png
new file mode 100644
index 0000000000000000000000000000000000000000..479e333aad23369355893ef60bcb821bf17ed158
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4032_1657ad5398b5f3faf15b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4064_0c9c53dc9fc524015333.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4064_0c9c53dc9fc524015333.png
new file mode 100644
index 0000000000000000000000000000000000000000..33e1552fe367ba0d97f02985a490d0a68eaef659
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4064_0c9c53dc9fc524015333.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4096_c5446e257ac8e30dad8d.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4096_c5446e257ac8e30dad8d.png
new file mode 100644
index 0000000000000000000000000000000000000000..cf7ef3aebda84a2cf32bf4e632018d01c6f990ec
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4096_c5446e257ac8e30dad8d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4128_eb960794fe6ac225202f.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4128_eb960794fe6ac225202f.png
new file mode 100644
index 0000000000000000000000000000000000000000..eb1a38a5d9f65a7f2fdb8346823b80e4ea8e851e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4128_eb960794fe6ac225202f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4160_6b31efc68363786054eb.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4160_6b31efc68363786054eb.png
new file mode 100644
index 0000000000000000000000000000000000000000..afa1f6100a035eff3ea2d803c2f66f4d7a29233a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4160_6b31efc68363786054eb.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_416_d8d6005c6840720844a1.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_416_d8d6005c6840720844a1.png
new file mode 100644
index 0000000000000000000000000000000000000000..ffcba2a112b2f5eae3476f68584ceb448064dca4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_416_d8d6005c6840720844a1.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4192_6d85de27a1093b096d37.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4192_6d85de27a1093b096d37.png
new file mode 100644
index 0000000000000000000000000000000000000000..5fc7bbb079703bb629a6b8e0e1bdb91f1e429c97
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4192_6d85de27a1093b096d37.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4224_effb46dd3e3f38890732.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4224_effb46dd3e3f38890732.png
new file mode 100644
index 0000000000000000000000000000000000000000..7f107aa6a54cb3f2e4c2502682177a4345731198
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4224_effb46dd3e3f38890732.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4256_1a1c31f082fedda491c7.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4256_1a1c31f082fedda491c7.png
new file mode 100644
index 0000000000000000000000000000000000000000..fbcda1128d9b283d6b7dc26072b64bf19a258573
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4256_1a1c31f082fedda491c7.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4288_7efc90dd953d1dc920fd.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4288_7efc90dd953d1dc920fd.png
new file mode 100644
index 0000000000000000000000000000000000000000..ab9fe1cdb4c51060ecf8d1f45ad9bdb5f5f261ef
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4288_7efc90dd953d1dc920fd.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4320_94764960c8baf8449a6e.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4320_94764960c8baf8449a6e.png
new file mode 100644
index 0000000000000000000000000000000000000000..5c81e9cd25a3f9cdcf7f46b87135133cc2988849
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4320_94764960c8baf8449a6e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4352_bf194539070406cfbafc.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4352_bf194539070406cfbafc.png
new file mode 100644
index 0000000000000000000000000000000000000000..97f789645d807a45b01f0bbd35f2034c8d65e172
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4352_bf194539070406cfbafc.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4384_f4d757684d56ef0041b5.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4384_f4d757684d56ef0041b5.png
new file mode 100644
index 0000000000000000000000000000000000000000..64f64e394bd4fea560d0fbb06cb04f9fc7c91dea
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4384_f4d757684d56ef0041b5.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4416_fa89d01ebf5c9ece481f.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4416_fa89d01ebf5c9ece481f.png
new file mode 100644
index 0000000000000000000000000000000000000000..a9b40ce0193f9f00ff1ef9942a2600e6cd32a16b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4416_fa89d01ebf5c9ece481f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4448_16683bf1b573d4ea50d6.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4448_16683bf1b573d4ea50d6.png
new file mode 100644
index 0000000000000000000000000000000000000000..1e8626b0f2edeac11da76d30e783432da1636712
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4448_16683bf1b573d4ea50d6.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4480_8591606d385f175ae17b.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4480_8591606d385f175ae17b.png
new file mode 100644
index 0000000000000000000000000000000000000000..f9a30fef38375fdb30b22cec3ac9a9b5ce9fc17d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4480_8591606d385f175ae17b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_448_41d31dc980613be5e0b7.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_448_41d31dc980613be5e0b7.png
new file mode 100644
index 0000000000000000000000000000000000000000..f2c7d2bb539d8db02438726c7f04adb4a11e247f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_448_41d31dc980613be5e0b7.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4512_cff667e40b5c91e9fa22.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4512_cff667e40b5c91e9fa22.png
new file mode 100644
index 0000000000000000000000000000000000000000..d65bd048cb29849aa9cd7037475298ce5d395656
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4512_cff667e40b5c91e9fa22.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4544_b14ab430fc4a2dfa8184.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4544_b14ab430fc4a2dfa8184.png
new file mode 100644
index 0000000000000000000000000000000000000000..fc956d849696277e1a5e41930be8e0962757e816
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4544_b14ab430fc4a2dfa8184.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4576_e2ea90a58d092cfb8bde.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4576_e2ea90a58d092cfb8bde.png
new file mode 100644
index 0000000000000000000000000000000000000000..4d045a5480fee812992f3b42e31bb761db79dec8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4576_e2ea90a58d092cfb8bde.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4608_d046aef3da8965f9c687.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4608_d046aef3da8965f9c687.png
new file mode 100644
index 0000000000000000000000000000000000000000..c473572bb318766688e88a6cd8b712bab3440394
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4608_d046aef3da8965f9c687.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4640_c4d436572d5db0383a58.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4640_c4d436572d5db0383a58.png
new file mode 100644
index 0000000000000000000000000000000000000000..8b9bd36bab45417f7d2c3e1ef7f6d9bf4922f389
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4640_c4d436572d5db0383a58.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4672_fa70cc8a87b2c7d6e3a8.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4672_fa70cc8a87b2c7d6e3a8.png
new file mode 100644
index 0000000000000000000000000000000000000000..40b45aa88ef91de2f77045f1243d07fe957ce572
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4672_fa70cc8a87b2c7d6e3a8.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4704_dcf31639a5e1c06f4629.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4704_dcf31639a5e1c06f4629.png
new file mode 100644
index 0000000000000000000000000000000000000000..d66bca73e89179a7d111099dada87b9d0a6fe4fc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4704_dcf31639a5e1c06f4629.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4736_10419b617d8b320c3f18.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4736_10419b617d8b320c3f18.png
new file mode 100644
index 0000000000000000000000000000000000000000..2662ff4365a859f57718d51b03ad5239abc1fe79
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4736_10419b617d8b320c3f18.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4768_d58e09f5c6dfde6765d0.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4768_d58e09f5c6dfde6765d0.png
new file mode 100644
index 0000000000000000000000000000000000000000..20fcc1f7b780ad2c178948e7e2df85b84929c764
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4768_d58e09f5c6dfde6765d0.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4800_c4a57fc9d882e902f894.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4800_c4a57fc9d882e902f894.png
new file mode 100644
index 0000000000000000000000000000000000000000..47503889d43ad9cb835da6f9d812b487dc59bb05
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4800_c4a57fc9d882e902f894.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_480_f9f9e354c709198a73be.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_480_f9f9e354c709198a73be.png
new file mode 100644
index 0000000000000000000000000000000000000000..6d92be59d3ecda4431fa0a3ff4d2bb0f03b57719
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_480_f9f9e354c709198a73be.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4832_026405c5987f394a4570.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4832_026405c5987f394a4570.png
new file mode 100644
index 0000000000000000000000000000000000000000..71fcdb9c589bf35f0f3b91303a0e02e8b70337e1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4832_026405c5987f394a4570.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4864_f0f4b1810d7c4c94adf5.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4864_f0f4b1810d7c4c94adf5.png
new file mode 100644
index 0000000000000000000000000000000000000000..be74e8778b747d61b046cd1cdb3503c9856e0ffd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4864_f0f4b1810d7c4c94adf5.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4896_c82c3e1740fd8f62141a.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4896_c82c3e1740fd8f62141a.png
new file mode 100644
index 0000000000000000000000000000000000000000..bda65b929e549bbe75c782fdd57905d382cff49c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4896_c82c3e1740fd8f62141a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4928_7f8b3129ff9478fcc7f8.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4928_7f8b3129ff9478fcc7f8.png
new file mode 100644
index 0000000000000000000000000000000000000000..2cddacea7fbef41814004c4e26d5ef37abb00500
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4928_7f8b3129ff9478fcc7f8.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4960_08b3ae9f5b8dd844fbd0.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4960_08b3ae9f5b8dd844fbd0.png
new file mode 100644
index 0000000000000000000000000000000000000000..424edbe792665a177aec7e5271be5b4d26c6615f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4960_08b3ae9f5b8dd844fbd0.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4992_577ff836e42b07323e0e.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4992_577ff836e42b07323e0e.png
new file mode 100644
index 0000000000000000000000000000000000000000..633218a797bf64fe30f65e05007a4fd5876964db
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_4992_577ff836e42b07323e0e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5024_6e8b1ee4979951f4fe0d.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5024_6e8b1ee4979951f4fe0d.png
new file mode 100644
index 0000000000000000000000000000000000000000..f71eb071a725e5c1adac03e01e6714dbd3ab8802
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5024_6e8b1ee4979951f4fe0d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5056_c30c95d9150b6e79fe1c.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5056_c30c95d9150b6e79fe1c.png
new file mode 100644
index 0000000000000000000000000000000000000000..f92feb6658a647242fdbaa8821f92c98e498436c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5056_c30c95d9150b6e79fe1c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5088_b12ea1970d5beba94ad1.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5088_b12ea1970d5beba94ad1.png
new file mode 100644
index 0000000000000000000000000000000000000000..4e6a2b978f7fc20f53bcff599394082ba310a3cf
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5088_b12ea1970d5beba94ad1.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5120_b1e4f47c00304b49c86e.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5120_b1e4f47c00304b49c86e.png
new file mode 100644
index 0000000000000000000000000000000000000000..9e7ff384c3b2832e53cddf45c21bed8d6ad0422f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5120_b1e4f47c00304b49c86e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_512_aa14f08534521b4cd4ae.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_512_aa14f08534521b4cd4ae.png
new file mode 100644
index 0000000000000000000000000000000000000000..1a5f3099cbc36da5131faa8d99d346283d7bb1c9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_512_aa14f08534521b4cd4ae.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5152_78ccd1ebe09bbbc0d8ea.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5152_78ccd1ebe09bbbc0d8ea.png
new file mode 100644
index 0000000000000000000000000000000000000000..b2bfc9efda1277307e40646aa8b1e604cced37e2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5152_78ccd1ebe09bbbc0d8ea.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5184_4a6680e10d2f9d85cc32.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5184_4a6680e10d2f9d85cc32.png
new file mode 100644
index 0000000000000000000000000000000000000000..3553c34f4763889768081ba177664246d1e8481d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5184_4a6680e10d2f9d85cc32.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5216_5590ca076135b9665a66.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5216_5590ca076135b9665a66.png
new file mode 100644
index 0000000000000000000000000000000000000000..2fce5add381b2e98745cd56ab07d024dfb1d162a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5216_5590ca076135b9665a66.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5248_a114d33c55ebdb3d6130.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5248_a114d33c55ebdb3d6130.png
new file mode 100644
index 0000000000000000000000000000000000000000..23510d831c9edd18bd73cd525f40823e44dd9791
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5248_a114d33c55ebdb3d6130.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5280_0d8447fb9817a288d9f6.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5280_0d8447fb9817a288d9f6.png
new file mode 100644
index 0000000000000000000000000000000000000000..e6a63bb0c1a2279184cd3be6813a513161e95682
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5280_0d8447fb9817a288d9f6.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5312_7e46f09178999546b6ff.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5312_7e46f09178999546b6ff.png
new file mode 100644
index 0000000000000000000000000000000000000000..3058a6083ee0ea8b12cfdc1095d1b4fae21d7fc4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5312_7e46f09178999546b6ff.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5344_cba3cc4aa127ddf1e1d9.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5344_cba3cc4aa127ddf1e1d9.png
new file mode 100644
index 0000000000000000000000000000000000000000..0d6180289b885cb940e916d2e38434413f99eb86
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5344_cba3cc4aa127ddf1e1d9.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5376_f719e12418709f4d2c23.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5376_f719e12418709f4d2c23.png
new file mode 100644
index 0000000000000000000000000000000000000000..437c34922e4e1ec40002438508ea092d1b336f7f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5376_f719e12418709f4d2c23.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5408_b807cc270b3760db682c.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5408_b807cc270b3760db682c.png
new file mode 100644
index 0000000000000000000000000000000000000000..3cc3bff93f563dbf079613b119e9599ca01ab385
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5408_b807cc270b3760db682c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5440_58d12e9aae19bf2fa711.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5440_58d12e9aae19bf2fa711.png
new file mode 100644
index 0000000000000000000000000000000000000000..d324a33fc38b0f836f696c590dcbd1e2dc10cbe8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5440_58d12e9aae19bf2fa711.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_544_6a6b43c60feea49692e3.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_544_6a6b43c60feea49692e3.png
new file mode 100644
index 0000000000000000000000000000000000000000..f7496da2cec4ff32c0f9ae3b84be80737e9c14e1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_544_6a6b43c60feea49692e3.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5472_9c403f5e6da4b8c713c6.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5472_9c403f5e6da4b8c713c6.png
new file mode 100644
index 0000000000000000000000000000000000000000..3a367e115858a59d4ccb9f07f08cfe8639b096ce
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5472_9c403f5e6da4b8c713c6.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5504_7bdcb0d8c47bd2593f5f.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5504_7bdcb0d8c47bd2593f5f.png
new file mode 100644
index 0000000000000000000000000000000000000000..ab87a996ca36c91b4018ef53fae9b66b6deafa58
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5504_7bdcb0d8c47bd2593f5f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5536_4031f442124b35e7927e.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5536_4031f442124b35e7927e.png
new file mode 100644
index 0000000000000000000000000000000000000000..56270bb41d632f4c08f1655535a1f5ef9fd7a72f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5536_4031f442124b35e7927e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5568_f71344375ee0407565b0.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5568_f71344375ee0407565b0.png
new file mode 100644
index 0000000000000000000000000000000000000000..de858ff73d2bc83eb0bfddf2f59f529cd36eb08a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5568_f71344375ee0407565b0.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5600_4b9a69a4b936e7ea0e3e.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5600_4b9a69a4b936e7ea0e3e.png
new file mode 100644
index 0000000000000000000000000000000000000000..7f879771a5a4647c581e7cd91299dc8ae1324ca9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5600_4b9a69a4b936e7ea0e3e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5632_522161f6d71736b17ee2.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5632_522161f6d71736b17ee2.png
new file mode 100644
index 0000000000000000000000000000000000000000..fd438e285b785821a556649e010bba498752e0d8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5632_522161f6d71736b17ee2.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5664_8b60e78c5ff6e86e85ab.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5664_8b60e78c5ff6e86e85ab.png
new file mode 100644
index 0000000000000000000000000000000000000000..9dde1c60104889b31f911d052f022b1bb1a186e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5664_8b60e78c5ff6e86e85ab.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5696_4f7c500b14cbe3f767eb.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5696_4f7c500b14cbe3f767eb.png
new file mode 100644
index 0000000000000000000000000000000000000000..a667e367010501b6919c753894d2f36e3b309f96
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5696_4f7c500b14cbe3f767eb.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5728_bf89c14ee6247058281a.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5728_bf89c14ee6247058281a.png
new file mode 100644
index 0000000000000000000000000000000000000000..89567e75b53b9b556a20f6d92de94537d4233338
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5728_bf89c14ee6247058281a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5760_709cbe854ce90f2830d1.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5760_709cbe854ce90f2830d1.png
new file mode 100644
index 0000000000000000000000000000000000000000..5b1481608217e6ee4ac350a13c18452c4d9619c8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5760_709cbe854ce90f2830d1.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_576_d0637fc2bdcb7abd6931.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_576_d0637fc2bdcb7abd6931.png
new file mode 100644
index 0000000000000000000000000000000000000000..b65c36c4746c553e7e2cb40559d961f244c26686
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_576_d0637fc2bdcb7abd6931.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5792_e5d494b0088070ae8024.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5792_e5d494b0088070ae8024.png
new file mode 100644
index 0000000000000000000000000000000000000000..25b3cc71af648b6880debc08dd91c5c44ca2c1d3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5792_e5d494b0088070ae8024.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5824_283285e130ba3c5bfbf8.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5824_283285e130ba3c5bfbf8.png
new file mode 100644
index 0000000000000000000000000000000000000000..0580a2b9cbe79b87bc750ea414607871e132b88e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5824_283285e130ba3c5bfbf8.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5856_4beb402c167eed9b4ce0.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5856_4beb402c167eed9b4ce0.png
new file mode 100644
index 0000000000000000000000000000000000000000..c29ea9eabda8b4eb047ea37f81ddf44d5e3c6315
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5856_4beb402c167eed9b4ce0.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5888_0ba96c48974aee302c8e.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5888_0ba96c48974aee302c8e.png
new file mode 100644
index 0000000000000000000000000000000000000000..520f0c8726971603c683eedaa0692aa4a8eb3491
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5888_0ba96c48974aee302c8e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5920_d0143e20491155c2091f.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5920_d0143e20491155c2091f.png
new file mode 100644
index 0000000000000000000000000000000000000000..0aefdd3902f35323ddbfc698f8a567039268a8d6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5920_d0143e20491155c2091f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5952_ad19e88d9c59d2f2c3ac.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5952_ad19e88d9c59d2f2c3ac.png
new file mode 100644
index 0000000000000000000000000000000000000000..6c75d032c9689ed4c0485d2e6a4ba8264353fefc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5952_ad19e88d9c59d2f2c3ac.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5984_7435d3afe6c251702cb0.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5984_7435d3afe6c251702cb0.png
new file mode 100644
index 0000000000000000000000000000000000000000..ea4441d7c351583654a8aecb322bd22e168d3ec8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_5984_7435d3afe6c251702cb0.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_6016_32629db18ac104ecfa65.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_6016_32629db18ac104ecfa65.png
new file mode 100644
index 0000000000000000000000000000000000000000..43f418d26d6ab08a11353bd0528643a0e63965b9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_6016_32629db18ac104ecfa65.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_6048_239d7b602cc1b202d152.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_6048_239d7b602cc1b202d152.png
new file mode 100644
index 0000000000000000000000000000000000000000..5f07626c8be82aef9f10ea5b0414b9ad363f1587
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_6048_239d7b602cc1b202d152.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_608_61d93036cb4f951c1653.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_608_61d93036cb4f951c1653.png
new file mode 100644
index 0000000000000000000000000000000000000000..dce2b75ad800b0c7e1302cbc6117dc90f8a26d90
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_608_61d93036cb4f951c1653.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_640_9a49d8d67b1c369de624.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_640_9a49d8d67b1c369de624.png
new file mode 100644
index 0000000000000000000000000000000000000000..4ab562c0dbf813be12dbcc65f4bb4de924d85ebc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_640_9a49d8d67b1c369de624.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_64_0ad77ca27898bb67b78d.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_64_0ad77ca27898bb67b78d.png
new file mode 100644
index 0000000000000000000000000000000000000000..2c2abd95f2dd7fab6365017662777142b96318e8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_64_0ad77ca27898bb67b78d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_672_8552ea93b45bdd917dc6.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_672_8552ea93b45bdd917dc6.png
new file mode 100644
index 0000000000000000000000000000000000000000..f5fa99ef59500c124bd669c390b88568a5a13f96
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_672_8552ea93b45bdd917dc6.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_704_6ee9718948f6761b6d18.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_704_6ee9718948f6761b6d18.png
new file mode 100644
index 0000000000000000000000000000000000000000..c104804c8853d2354846f1648ff7904a0fbf7ee7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_704_6ee9718948f6761b6d18.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_736_a4d91cd842a1ba77d961.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_736_a4d91cd842a1ba77d961.png
new file mode 100644
index 0000000000000000000000000000000000000000..b50b6209820ff4d568a6f942386b6ed913e49908
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_736_a4d91cd842a1ba77d961.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_768_e86b8d26131434abb295.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_768_e86b8d26131434abb295.png
new file mode 100644
index 0000000000000000000000000000000000000000..5f59893e943b46a83bd14a6017b2f1f141c741fa
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_768_e86b8d26131434abb295.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_800_11b4a191febad4563b4b.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_800_11b4a191febad4563b4b.png
new file mode 100644
index 0000000000000000000000000000000000000000..67d0e8e6c815026025979b0516a4226d3bd9e4f1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_800_11b4a191febad4563b4b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_832_e9b4117b33f08d2657fd.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_832_e9b4117b33f08d2657fd.png
new file mode 100644
index 0000000000000000000000000000000000000000..0cee31d4f0c61ad7cdf7e80b25ffe1c41fed77dc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_832_e9b4117b33f08d2657fd.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_864_a9553fd649505ffdf547.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_864_a9553fd649505ffdf547.png
new file mode 100644
index 0000000000000000000000000000000000000000..de072908443ee302a6ddafd98bbb9c4a01781a4b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_864_a9553fd649505ffdf547.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_896_74e58edaf06818c1ab43.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_896_74e58edaf06818c1ab43.png
new file mode 100644
index 0000000000000000000000000000000000000000..79f917e061228d2f4d113644619c1740d9892a51
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_896_74e58edaf06818c1ab43.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_928_0db2f39c460f77e0b595.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_928_0db2f39c460f77e0b595.png
new file mode 100644
index 0000000000000000000000000000000000000000..3df3866941c29dfffeffbc538e0d574bea8e60b4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_928_0db2f39c460f77e0b595.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_960_10c8501c32d19176c2b3.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_960_10c8501c32d19176c2b3.png
new file mode 100644
index 0000000000000000000000000000000000000000..e0adb47582c90fa666985938dd7dc468def1a8a1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_960_10c8501c32d19176c2b3.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_96_5d6ff1e483cd158d9d2e.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_96_5d6ff1e483cd158d9d2e.png
new file mode 100644
index 0000000000000000000000000000000000000000..da7c63e452caa7daece2eb15e62638c819c8ab04
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_96_5d6ff1e483cd158d9d2e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_992_db045fc44f2cf709b426.png b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_992_db045fc44f2cf709b426.png
new file mode 100644
index 0000000000000000000000000000000000000000..969bb8ad02529089d8eeda92239ac0d3c2109c96
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/images/ued/layout/levels16_img_992_db045fc44f2cf709b426.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_0_d2d9756f4ea98d9be959.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_0_d2d9756f4ea98d9be959.gif
new file mode 100644
index 0000000000000000000000000000000000000000..e519ea6ba26be71f05a68487c37b4667675ad154
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_0_d2d9756f4ea98d9be959.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d2d9756f4ea98d9be9594910c0fe5780928541b6fdc5f7f9aef2131e516d53dd
+size 109680
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_2016_f2f66af7b1bf030e09df.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_2016_f2f66af7b1bf030e09df.gif
new file mode 100644
index 0000000000000000000000000000000000000000..e640372b8154cc088af4b30531614aef5b3e9534
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_2016_f2f66af7b1bf030e09df.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f2f66af7b1bf030e09df2e2c884e28f0d3804b0f4acf808d9c8c6d003bf3760f
+size 260298
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_3040_c7fc4d034a9a17f8c668.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_3040_c7fc4d034a9a17f8c668.gif
new file mode 100644
index 0000000000000000000000000000000000000000..cf2841a08ad6eaffeadc21f1d69f71c2d768ac2c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_3040_c7fc4d034a9a17f8c668.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c7fc4d034a9a17f8c668b9d851430f1c7aba14cd5c060a45280c016047e71312
+size 244610
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_4064_e62d975a47004a0b87b4.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_4064_e62d975a47004a0b87b4.gif
new file mode 100644
index 0000000000000000000000000000000000000000..62baa21d79edc2aeaabebf56a7771ea2dddb4003
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_4064_e62d975a47004a0b87b4.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e62d975a47004a0b87b496c5face08b48d056c93ebd0b68fbba4ab5c298535ce
+size 253530
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_5088_1a999dd354e9cacb552d.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_5088_1a999dd354e9cacb552d.gif
new file mode 100644
index 0000000000000000000000000000000000000000..eef2acd6688256aefc11091139c5f3a6f3a3d8a3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_5088_1a999dd354e9cacb552d.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1a999dd354e9cacb552db7679fd97f0043f25b846aef9aa17212f1020212d7d8
+size 253901
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_992_824355b2d3a467bb3a18.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_992_824355b2d3a467bb3a18.gif
new file mode 100644
index 0000000000000000000000000000000000000000..a19122c36edbfb35a5cba975554a405492ff4a15
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_992_824355b2d3a467bb3a18.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:824355b2d3a467bb3a18eea23171790af9817a1d9c7b841863b494b7a30d178e
+size 256460
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_0_1f596720145f47629a86.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_0_1f596720145f47629a86.gif
new file mode 100644
index 0000000000000000000000000000000000000000..149c080a2e68bf0b899f094cbc557b3322d0f95f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_0_1f596720145f47629a86.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1f596720145f47629a86f0ba896516da350c951578bf2deecd0e07a06eb5de88
+size 116112
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_2016_bdb3e700f67de89cf9b4.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_2016_bdb3e700f67de89cf9b4.gif
new file mode 100644
index 0000000000000000000000000000000000000000..2e683c2a917d9ee7de55e900eadfffb3afc3d19a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_2016_bdb3e700f67de89cf9b4.gif differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_3040_b9e04603b6142e97e174.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_3040_b9e04603b6142e97e174.gif
new file mode 100644
index 0000000000000000000000000000000000000000..1df3ef6d41a510c8ef6de51a82b0fb230b9ae6e8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_3040_b9e04603b6142e97e174.gif differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_4064_38d4527781a7d45c4b3b.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_4064_38d4527781a7d45c4b3b.gif
new file mode 100644
index 0000000000000000000000000000000000000000..a7e056cb274802d1fc5d702fad5d3d1f6212d198
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_4064_38d4527781a7d45c4b3b.gif differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_5088_d9e7cbb78f0db799d05e.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_5088_d9e7cbb78f0db799d05e.gif
new file mode 100644
index 0000000000000000000000000000000000000000..ddd676a62245c4cd36f50f048df4004e7ff9592f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_5088_d9e7cbb78f0db799d05e.gif differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_992_9346e5fed31a0bf40872.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_992_9346e5fed31a0bf40872.gif
new file mode 100644
index 0000000000000000000000000000000000000000..51d6ae439e9e0d76d153acc47fcd1f7f119aa9ca
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_992_9346e5fed31a0bf40872.gif differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_0_487c9da232a04dd895aa.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_0_487c9da232a04dd895aa.gif
new file mode 100644
index 0000000000000000000000000000000000000000..37d500b38d2145217d941a7248cc709d56dfe885
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_0_487c9da232a04dd895aa.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:487c9da232a04dd895aabc678621cc28830a6a119f48f22041d9774851930688
+size 107287
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_2016_8b6d55e20bc2b3cd82c7.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_2016_8b6d55e20bc2b3cd82c7.gif
new file mode 100644
index 0000000000000000000000000000000000000000..704d67f2d916c50bec4c39f8ce06e3e8619c2ecd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_2016_8b6d55e20bc2b3cd82c7.gif differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_3040_db3cb8b14ba50eeb3576.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_3040_db3cb8b14ba50eeb3576.gif
new file mode 100644
index 0000000000000000000000000000000000000000..8e0aa6f53f378463c100521ff468a47039b6fa87
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_3040_db3cb8b14ba50eeb3576.gif differ
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_4064_08f09aceb06e4c59172f.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_4064_08f09aceb06e4c59172f.gif
new file mode 100644
index 0000000000000000000000000000000000000000..ffe49451e82f76b967cf36a0c83b7c22dcc8de5d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_4064_08f09aceb06e4c59172f.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:08f09aceb06e4c59172fa0d7ff90a5249c530fcc85d5ec99873897a149ca0b5e
+size 119563
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_5088_e00a0d045117b56c0b6c.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_5088_e00a0d045117b56c0b6c.gif
new file mode 100644
index 0000000000000000000000000000000000000000..0a7d594e4c8281707ed6e3ed68ee170f4699988e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_5088_e00a0d045117b56c0b6c.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e00a0d045117b56c0b6c0488ce5dd5a1af8081ef6978cfed7dc928415c8250ca
+size 126007
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_992_f7294f691ff5dbc9e887.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_992_f7294f691ff5dbc9e887.gif
new file mode 100644
index 0000000000000000000000000000000000000000..d12722988e7708a37dd20dd00be09c699c2bc679
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_992_f7294f691ff5dbc9e887.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f7294f691ff5dbc9e887d750daacfd32102924055638453d78caf24f44cf80fe
+size 127428
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-all/rollouts_gif_0_8bdbad86b104ab61e521.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-all/rollouts_gif_0_8bdbad86b104ab61e521.gif
new file mode 100644
index 0000000000000000000000000000000000000000..265310d5b11b0de103aee179b497ba63a047f5dc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-all/rollouts_gif_0_8bdbad86b104ab61e521.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8bdbad86b104ab61e52111b318b718ebb82910d3f3e674e600bd03bc7b94a2ef
+size 105986
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-all/rollouts_gif_2016_d617283381c3ba4ffa74.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-all/rollouts_gif_2016_d617283381c3ba4ffa74.gif
new file mode 100644
index 0000000000000000000000000000000000000000..2c58ec09e00ba65355642f223ded0a03bef75efb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-all/rollouts_gif_2016_d617283381c3ba4ffa74.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d617283381c3ba4ffa74020003f0deb9f1a975ab6deee7d116117826222d4509
+size 163746
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-all/rollouts_gif_3040_1deab0e6e9ee2422ffbc.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-all/rollouts_gif_3040_1deab0e6e9ee2422ffbc.gif
new file mode 100644
index 0000000000000000000000000000000000000000..d2033ceac955436606390287d7c1373c74b18103
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-all/rollouts_gif_3040_1deab0e6e9ee2422ffbc.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1deab0e6e9ee2422ffbc39bab6262d2bc706d17305155b32a05413f24249f3e5
+size 237092
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-all/rollouts_gif_4064_7bef146b11b7b5f9c1e2.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-all/rollouts_gif_4064_7bef146b11b7b5f9c1e2.gif
new file mode 100644
index 0000000000000000000000000000000000000000..4b141f829e05dfb8fb015c33a60605ee34816136
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-all/rollouts_gif_4064_7bef146b11b7b5f9c1e2.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7bef146b11b7b5f9c1e2c6a9d3f15d91c383d6a9ddd53db1bb87427b42cded21
+size 258805
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-all/rollouts_gif_5088_016bb4c85f0bf1211686.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-all/rollouts_gif_5088_016bb4c85f0bf1211686.gif
new file mode 100644
index 0000000000000000000000000000000000000000..78c9656166258e619f0527affdbec17cdcfbe5ff
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-all/rollouts_gif_5088_016bb4c85f0bf1211686.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:016bb4c85f0bf12116867c026ddb2387bcfe4c166e50d260da185e8024561cd3
+size 110715
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-all/rollouts_gif_992_c68b64c3490ca4eb840c.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-all/rollouts_gif_992_c68b64c3490ca4eb840c.gif
new file mode 100644
index 0000000000000000000000000000000000000000..f77532b6145f14da7aa29de68b2cb0de77945c51
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-all/rollouts_gif_992_c68b64c3490ca4eb840c.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c68b64c3490ca4eb840c4b9a7069413d75c049088c7a791e5c6a076bd89ca623
+size 259964
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-generate/rollouts_gif_3040_1deab0e6e9ee2422ffbc.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-generate/rollouts_gif_3040_1deab0e6e9ee2422ffbc.gif
new file mode 100644
index 0000000000000000000000000000000000000000..d2033ceac955436606390287d7c1373c74b18103
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-generate/rollouts_gif_3040_1deab0e6e9ee2422ffbc.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1deab0e6e9ee2422ffbc39bab6262d2bc706d17305155b32a05413f24249f3e5
+size 237092
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-generate/rollouts_gif_4064_7bef146b11b7b5f9c1e2.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-generate/rollouts_gif_4064_7bef146b11b7b5f9c1e2.gif
new file mode 100644
index 0000000000000000000000000000000000000000..4b141f829e05dfb8fb015c33a60605ee34816136
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-generate/rollouts_gif_4064_7bef146b11b7b5f9c1e2.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7bef146b11b7b5f9c1e2c6a9d3f15d91c383d6a9ddd53db1bb87427b42cded21
+size 258805
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-generate/rollouts_gif_992_c68b64c3490ca4eb840c.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-generate/rollouts_gif_992_c68b64c3490ca4eb840c.gif
new file mode 100644
index 0000000000000000000000000000000000000000..f77532b6145f14da7aa29de68b2cb0de77945c51
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-generate/rollouts_gif_992_c68b64c3490ca4eb840c.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c68b64c3490ca4eb840c4b9a7069413d75c049088c7a791e5c6a076bd89ca623
+size 259964
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-replay/rollouts_gif_0_8bdbad86b104ab61e521.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-replay/rollouts_gif_0_8bdbad86b104ab61e521.gif
new file mode 100644
index 0000000000000000000000000000000000000000..265310d5b11b0de103aee179b497ba63a047f5dc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-replay/rollouts_gif_0_8bdbad86b104ab61e521.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8bdbad86b104ab61e52111b318b718ebb82910d3f3e674e600bd03bc7b94a2ef
+size 105986
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-replay/rollouts_gif_2016_d617283381c3ba4ffa74.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-replay/rollouts_gif_2016_d617283381c3ba4ffa74.gif
new file mode 100644
index 0000000000000000000000000000000000000000..2c58ec09e00ba65355642f223ded0a03bef75efb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-replay/rollouts_gif_2016_d617283381c3ba4ffa74.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d617283381c3ba4ffa74020003f0deb9f1a975ab6deee7d116117826222d4509
+size 163746
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-replay/rollouts_gif_5088_016bb4c85f0bf1211686.gif b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-replay/rollouts_gif_5088_016bb4c85f0bf1211686.gif
new file mode 100644
index 0000000000000000000000000000000000000000..78c9656166258e619f0527affdbec17cdcfbe5ff
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/media/videos/train-replay/rollouts_gif_5088_016bb4c85f0bf1211686.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:016bb4c85f0bf12116867c026ddb2387bcfe4c166e50d260da185e8024561cd3
+size 110715
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/output.log b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/output.log
new file mode 100644
index 0000000000000000000000000000000000000000..9810d56ce9fc88f93cc386ec9403405b2a205c57
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/output.log
@@ -0,0 +1,23142 @@
+  env_size                      : 13
+  env_layout                    : tree
+  wall_prob                     : 0
+  env_corner_size               : 1
+  env_terminate_after_corner    : False
+  obs_level_of_detail           : 0
+  img_level_of_detail           : 1
+  env_penalize_time             : False
+  net_cnn_type                  : large
+  net_rnn_type                  : ff
+  net_width                     : 256
+  ued                           : plr
+  prob_shift                    : 0
+  num_train_levels              : 2048
+  plr_buffer_size               : 4096
+  plr_temperature               : 0.1
+  plr_staleness_coeff           : 0.1
+  plr_prob_replay               : 0.5
+  plr_regret_estimator          : maxmc-actor
+  plr_robust                    : False
+  num_mutate_steps              : 12
+  prob_mutate_shift             : 0
+  chain_mutate                  : True
+  mutate_cheese                 : True
+  train_proxy_critic            : False
+  plr_proxy_shaping             : False
+  proxy_name                    : proxy_corner
+  plr_proxy_shaping_coeff       : 0.5
+  clipping                      : False
+  ppo_lr                        : 5e-05
+  ppo_gamma                     : 0.999
+  ppo_clip_eps                  : 0.1
+  ppo_gae_lambda                : 0.95
+  ppo_entropy_coeff             : 0.001
+  ppo_critic_coeff              : 0.5
+  ppo_proxy_critic_coeff        : 0.5
+  ppo_max_grad_norm             : 0.5
+  ppo_lr_annealing              : False
+  num_minibatches_per_epoch     : 4
+  num_epochs_per_cycle          : 5
+  num_total_env_steps           : 200000000
+  num_env_steps_per_cycle       : 128
+  num_parallel_envs             : 256
+  console_log                   : True
+  wandb_log                     : True
+  wandb_project                 : jaxgmg-wall-prob-tree
+  wandb_entity                  : None
+  wandb_group                   : None
+  wandb_name                    : wp:0.0_alpha:0
+  log_gifs                      : True
+  log_imgs                      : True
+  log_hists                     : False
+  num_cycles_per_log            : 32
+  num_cycles_per_eval           : 32
+  num_cycles_per_gifs           : 1024
+  num_cycles_per_big_eval       : 1024
+  use_fixed_eval_levels         : False
+  evals_num_env_steps           : 512
+  evals_num_levels              : 256
+  gif_grid_width                : 16
+  level_splayer                 : mouse
+  checkpointing                 : True
+  keep_all_checkpoints          : True
+  max_num_checkpoints           : 1
+  num_cycles_per_checkpoint     : 64
+  seed                          : 42
+configuring environment...
+configuring level generators...
+configuring level classifier...
+configuring level mutator...
+configuring level solver...
+configuring level metrics...
+configuring eval level generators...
+configuring parser and parsing fixed eval levels...
+(unsupported size for fixed evals)
+configuring heatmap splayer...
+seeding random number generator with seed=42...
+configuring curriculum with ued='plr'...
+configuring eval batches from 3 level generators...
+  generating 256 'orig' levels...
+  also solving generated levels...
+  generating 256 'shift' levels...
+  also solving generated levels...
+  generating 256 'tree' levels...
+  also solving generated levels...
+configuring evals for 0 fixed eval levels...
+configuring actor critic network...
+  net_cnn_type='large'
+  net_rnn_type='ff'
+  net_width=256
+  initialising network...
+  number of parameters: 198982
+initialising the checkpointer...
+setting up optimiser...
+initialising train state...
+configuring PPO updater...
+begin training loop!
+(note: first two cycles are slow due to compilation)
+                                                    steps/s]
+===========================================================
+  step:
+    env-step-all-after:                         32768
+    env-step-replay-after:                      32768
+    env-step-used-after:                        32768
+    ppo-update-after:                           20
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     84.47618
+    avg_avg_return:                             0.045426786
+    avg_reward_per_step:                        0.0008544922
+    proxy_corner:
+      avg_avg_return:                           0.045426786
+      avg_reward_per_step:                      0.0008544922
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     84.47618
+    avg_avg_return:                             0.045426786
+    avg_reward_per_step:                        0.0008544922
+    proxy_corner:
+      avg_avg_return:                           0.045426786
+      avg_reward_per_step:                      0.0008544922
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  perf:
+    env_steps_per_second:                       2485.1837178749106
+    ppo_updates_per_second:                     0.4950293040315249
+    eval-batch-orig-duration:                   12.000743062235415
+    eval-batch-shift-duration:                  0.5560225192457438
+    eval-batch-tree-duration:                   0.5955950804054737
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             10.063232
+      mouse-cheese_dist_finite_wavg:            13.602753
+      mouse-cheese_dist_solvable_avg:           10.063232
+      mouse-cheese_dist_solvable_wavg:          13.602748
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000004
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.0021038568
+    visit_patterns:
+      avg_first_visit_time:                     0.0
+      avg_last_visit_time:                      0.0625
+      num_replay_batches:                       1
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0024435713
+    avg_actor_approxkl3:                        0.0026338014
+    avg_actor_clipfrac:                         0.15576783
+    avg_actor_loss:                             -0.0037967935
+    avg_advantage:                              0.048738517
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0067404
+    avg_entropy:                                1.3678831
+    avg_grad_norm_pre_clip:                     0.2607908
+    avg_loss:                                   -0.0017944764
+    max:
+      max_actor_approxkl1:                      0.004030235
+      max_actor_approxkl3:                      0.0036007024
+      max_actor_clipfrac:                       0.33911133
+      max_actor_loss:                           -7.4505806e-08
+      max_advantage:                            1.2869482
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0105511565
+      max_entropy:                              1.371342
+      max_grad_norm_pre_clip:                   0.5807057
+      max_loss:                                 0.0010989512
+    std:
+      std_actor_approxkl1:                      0.0010353023
+      std_actor_approxkl3:                      0.00091282226
+      std_actor_clipfrac:                       0.110722914
+      std_actor_loss:                           0.0015215236
+      std_advantage:                            0.11731085
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0018020744
+      std_entropy:                              0.003654496
+      std_grad_norm_pre_clip:                   0.10771038
+      std_loss:                                 0.0012596718
+  eval-batch-orig:
+    avg_avg_episode_length:                     82.98658
+    avg_avg_return:                             0.12104543
+    avg_benchmark_regret:                       0.86858404
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.0014648438
+    proxy_corner:
+      avg_avg_return:                           0.12104543
+      avg_reward_per_step:                      0.0014648438
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-shift:
+    avg_avg_episode_length:                     77.68495
+    avg_avg_return:                             0.22890788
+    avg_benchmark_regret:                       0.7638637
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0036849976
+    proxy_corner:
+      avg_avg_return:                           0.14200409
+      avg_reward_per_step:                      0.0014419556
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.065735
+    avg_avg_return:                             0.1411371
+    avg_benchmark_regret:                       0.84293854
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.003944397
+    proxy_corner:
+      avg_avg_return:                           0.1411371
+      avg_reward_per_step:                      0.003944397
+    rollouts_gif:                               float32(132, 225, 225, 3)
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        1048576
+    env-step-replay-before:                     557056
+    env-step-used-before:                       1048576
+    ppo-update-before:                          640
+    env-step-generate-before:                   491520
+    env-step-all-after:                         1081344
+    env-step-replay-after:                      589824
+    env-step-used-after:                        1081344
+    ppo-update-after:                           660
+    env-step-generate-after:                    491520
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     12.48398
+    avg_avg_return:                             0.90500486
+    avg_reward_per_step:                        0.10586548
+    proxy_corner:
+      avg_avg_return:                           0.90500486
+      avg_reward_per_step:                      0.10586548
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     12.48398
+    avg_avg_return:                             0.90500486
+    avg_reward_per_step:                        0.10586548
+    proxy_corner:
+      avg_avg_return:                           0.90500486
+      avg_reward_per_step:                      0.10586548
+  perf:
+    env_steps_per_second:                       261211.9448948108
+    ppo_updates_per_second:                     41.4909983284729
+    eval-batch-orig-duration:                   0.5319276247173548
+    eval-batch-shift-duration:                  0.4803112018853426
+    eval-batch-tree-duration:                   0.498010759241879
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             11.884277
+      mouse-cheese_dist_finite_wavg:            2.0258465
+      mouse-cheese_dist_solvable_avg:           11.884277
+      mouse-cheese_dist_solvable_wavg:          2.0258465
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.119080916
+    visit_patterns:
+      avg_first_visit_time:                     3.0361328
+      avg_last_visit_time:                      8.930908
+      num_replay_batches:                       18
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014158005
+    avg_actor_approxkl3:                        0.0013376612
+    avg_actor_clipfrac:                         0.08065796
+    avg_actor_loss:                             -0.00073929434
+    avg_advantage:                              -0.008993767
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00016916671
+    avg_entropy:                                0.65036076
+    avg_grad_norm_pre_clip:                     1.4057153
+    avg_loss:                                   -0.0013050718
+    max:
+      max_actor_approxkl1:                      0.004073262
+      max_actor_approxkl3:                      0.003992399
+      max_actor_clipfrac:                       0.3100586
+      max_actor_loss:                           0.0012977589
+      max_advantage:                            0.112734914
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00020247363
+      max_entropy:                              0.6681402
+      max_grad_norm_pre_clip:                   5.8051324
+      max_loss:                                 0.000724462
+    std:
+      std_actor_approxkl1:                      0.0011169584
+      std_actor_approxkl3:                      0.0009705827
+      std_actor_clipfrac:                       0.08193358
+      std_actor_loss:                           0.000837753
+      std_advantage:                            0.017586438
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.1491607e-05
+      std_entropy:                              0.0108471885
+      std_grad_norm_pre_clip:                   1.3973609
+      std_loss:                                 0.00083836535
+  eval-batch-orig:
+    avg_avg_episode_length:                     11.249315
+    avg_avg_return:                             0.9690336
+    avg_benchmark_regret:                       0.02059596
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.11708069
+    proxy_corner:
+      avg_avg_return:                           0.9690336
+      avg_reward_per_step:                      0.11708069
+  eval-batch-shift:
+    avg_avg_episode_length:                     83.44081
+    avg_avg_return:                             0.036139384
+    avg_benchmark_regret:                       0.95663214
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0011291504
+    proxy_corner:
+      avg_avg_return:                           0.9520311
+      avg_reward_per_step:                      0.0078125
+  eval-batch-tree:
+    avg_avg_episode_length:                     68.633896
+    avg_avg_return:                             0.23793182
+    avg_benchmark_regret:                       0.7461438
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.03224182
+    proxy_corner:
+      avg_avg_return:                           0.23793182
+      avg_reward_per_step:                      0.03224182
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        2097152
+    env-step-replay-before:                     1114112
+    env-step-used-before:                       2097152
+    ppo-update-before:                          1280
+    env-step-generate-before:                   983040
+    env-step-all-after:                         2129920
+    env-step-replay-after:                      1114112
+    env-step-used-after:                        2129920
+    ppo-update-after:                           1300
+    env-step-generate-after:                    1015808
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.337027
+    avg_avg_return:                             0.92013955
+    avg_reward_per_step:                        0.13461304
+    proxy_corner:
+      avg_avg_return:                           0.92013955
+      avg_reward_per_step:                      0.13461304
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.337027
+    avg_avg_return:                             0.92013955
+    avg_reward_per_step:                        0.13461304
+    proxy_corner:
+      avg_avg_return:                           0.92013955
+      avg_reward_per_step:                      0.13461304
+  perf:
+    env_steps_per_second:                       313424.12294447876
+    ppo_updates_per_second:                     42.18042127776411
+    eval-batch-orig-duration:                   0.5291581377387047
+    eval-batch-shift-duration:                  0.4737989353016019
+    eval-batch-tree-duration:                   0.4881916306912899
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             13.304932
+      mouse-cheese_dist_finite_wavg:            18.38484
+      mouse-cheese_dist_solvable_avg:           13.304932
+      mouse-cheese_dist_solvable_wavg:          18.384842
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.10465534
+    visit_patterns:
+      avg_first_visit_time:                     8.32251
+      avg_last_visit_time:                      24.23584
+      num_replay_batches:                       34
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0016768503
+    avg_actor_approxkl3:                        0.001772146
+    avg_actor_clipfrac:                         0.08185425
+    avg_actor_loss:                             0.0012970761
+    avg_advantage:                              0.0016201409
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0001138375
+    avg_entropy:                                0.29257536
+    avg_grad_norm_pre_clip:                     3.1869295
+    avg_loss:                                   0.0010614195
+    max:
+      max_actor_approxkl1:                      0.0053368965
+      max_actor_approxkl3:                      0.0062959725
+      max_actor_clipfrac:                       0.2421875
+      max_actor_loss:                           0.006682599
+      max_advantage:                            0.086853564
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00013407387
+      max_entropy:                              0.31901786
+      max_grad_norm_pre_clip:                   8.589235
+      max_loss:                                 0.006438614
+    std:
+      std_actor_approxkl1:                      0.0018271374
+      std_actor_approxkl3:                      0.0016087865
+      std_actor_clipfrac:                       0.07238825
+      std_actor_loss:                           0.0021140624
+      std_advantage:                            0.014941886
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          8.882873e-06
+      std_entropy:                              0.012910991
+      std_grad_norm_pre_clip:                   2.3964505
+      std_loss:                                 0.00210828
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.349625
+    avg_avg_return:                             0.9719045
+    avg_benchmark_regret:                       0.01772499
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13008118
+    proxy_corner:
+      avg_avg_return:                           0.9719045
+      avg_reward_per_step:                      0.13008118
+  eval-batch-shift:
+    avg_avg_episode_length:                     83.468735
+    avg_avg_return:                             0.03318685
+    avg_benchmark_regret:                       0.95958465
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.003227234
+    proxy_corner:
+      avg_avg_return:                           0.92339635
+      avg_reward_per_step:                      0.0075302124
+  eval-batch-tree:
+    avg_avg_episode_length:                     68.72314
+    avg_avg_return:                             0.23300678
+    avg_benchmark_regret:                       0.75106883
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.03450775
+    proxy_corner:
+      avg_avg_return:                           0.23300678
+      avg_reward_per_step:                      0.03450775
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        3145728
+    env-step-replay-before:                     1605632
+    env-step-used-before:                       3145728
+    ppo-update-before:                          1920
+    env-step-generate-before:                   1540096
+    env-step-all-after:                         3178496
+    env-step-replay-after:                      1638400
+    env-step-used-after:                        3178496
+    ppo-update-after:                           1940
+    env-step-generate-after:                    1540096
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     14.314285
+    avg_avg_return:                             0.89520967
+    avg_reward_per_step:                        0.076538086
+    proxy_corner:
+      avg_avg_return:                           0.89520967
+      avg_reward_per_step:                      0.076538086
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     14.314285
+    avg_avg_return:                             0.89520967
+    avg_reward_per_step:                        0.076538086
+    proxy_corner:
+      avg_avg_return:                           0.89520967
+      avg_reward_per_step:                      0.076538086
+  perf:
+    env_steps_per_second:                       272290.1422680207
+    ppo_updates_per_second:                     41.566806413817716
+    eval-batch-orig-duration:                   0.5302922185510397
+    eval-batch-shift-duration:                  0.49102029390633106
+    eval-batch-tree-duration:                   0.49116952903568745
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             14.333008
+      mouse-cheese_dist_finite_wavg:            18.498817
+      mouse-cheese_dist_solvable_avg:           14.333008
+      mouse-cheese_dist_solvable_wavg:          18.498817
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.102604136
+    visit_patterns:
+      avg_first_visit_time:                     16.131592
+      avg_last_visit_time:                      40.397705
+      num_replay_batches:                       50
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.006632115
+    avg_actor_approxkl3:                        0.006576372
+    avg_actor_clipfrac:                         0.10910034
+    avg_actor_loss:                             0.0073240064
+    avg_advantage:                              0.008265303
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00011873456
+    avg_entropy:                                0.18770443
+    avg_grad_norm_pre_clip:                     5.7745795
+    avg_loss:                                   0.00719567
+    max:
+      max_actor_approxkl1:                      0.02361317
+      max_actor_approxkl3:                      0.023294766
+      max_actor_clipfrac:                       0.22851562
+      max_actor_loss:                           0.030569648
+      max_advantage:                            0.05972141
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00024174436
+      max_entropy:                              0.20624757
+      max_grad_norm_pre_clip:                   11.095425
+      max_loss:                                 0.030441266
+    std:
+      std_actor_approxkl1:                      0.006442435
+      std_actor_approxkl3:                      0.006601558
+      std_actor_clipfrac:                       0.059419602
+      std_actor_loss:                           0.009275706
+      std_advantage:                            0.011650534
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.262988e-05
+      std_entropy:                              0.008061364
+      std_grad_norm_pre_clip:                   3.321183
+      std_loss:                                 0.009287623
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.148228
+    avg_avg_return:                             0.97227466
+    avg_benchmark_regret:                       0.017354831
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13246918
+    proxy_corner:
+      avg_avg_return:                           0.97227466
+      avg_reward_per_step:                      0.13246918
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.96215
+    avg_avg_return:                             0.038959306
+    avg_benchmark_regret:                       0.95381224
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0045700073
+    proxy_corner:
+      avg_avg_return:                           0.8873488
+      avg_reward_per_step:                      0.0072631836
+  eval-batch-tree:
+    avg_avg_episode_length:                     69.14557
+    avg_avg_return:                             0.22777672
+    avg_benchmark_regret:                       0.7562989
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.036575317
+    proxy_corner:
+      avg_avg_return:                           0.22777672
+      avg_reward_per_step:                      0.036575317
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        4194304
+    env-step-replay-before:                     2162688
+    env-step-used-before:                       4194304
+    ppo-update-before:                          2560
+    env-step-generate-before:                   2031616
+    env-step-all-after:                         4227072
+    env-step-replay-after:                      2195456
+    env-step-used-after:                        4227072
+    ppo-update-after:                           2580
+    env-step-generate-after:                    2031616
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     15.196661
+    avg_avg_return:                             0.88468003
+    avg_reward_per_step:                        0.06478882
+    proxy_corner:
+      avg_avg_return:                           0.88468003
+      avg_reward_per_step:                      0.06478882
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     15.196661
+    avg_avg_return:                             0.88468003
+    avg_reward_per_step:                        0.06478882
+    proxy_corner:
+      avg_avg_return:                           0.88468003
+      avg_reward_per_step:                      0.06478882
+  perf:
+    env_steps_per_second:                       292954.85811736947
+    ppo_updates_per_second:                     42.36599670808471
+    eval-batch-orig-duration:                   0.5347175598144531
+    eval-batch-shift-duration:                  0.5051723355427384
+    eval-batch-tree-duration:                   0.4849514374509454
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             15.154785
+      mouse-cheese_dist_finite_wavg:            17.705841
+      mouse-cheese_dist_solvable_avg:           15.154785
+      mouse-cheese_dist_solvable_wavg:          17.705841
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.108011544
+    visit_patterns:
+      avg_first_visit_time:                     25.97754
+      avg_last_visit_time:                      57.415527
+      num_replay_batches:                       67
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0016721301
+    avg_actor_approxkl3:                        0.0017455969
+    avg_actor_clipfrac:                         0.06575928
+    avg_actor_loss:                             0.0020082586
+    avg_advantage:                              -0.0078003523
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            8.590058e-05
+    avg_entropy:                                0.15581776
+    avg_grad_norm_pre_clip:                     4.6003385
+    avg_loss:                                   0.0018953913
+    max:
+      max_actor_approxkl1:                      0.0047601676
+      max_actor_approxkl3:                      0.0056851655
+      max_actor_clipfrac:                       0.13195801
+      max_actor_loss:                           0.0064409347
+      max_advantage:                            0.06662691
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00018702324
+      max_entropy:                              0.16569608
+      max_grad_norm_pre_clip:                   9.233726
+      max_loss:                                 0.006374471
+    std:
+      std_actor_approxkl1:                      0.0013340354
+      std_actor_approxkl3:                      0.0014868258
+      std_actor_clipfrac:                       0.037273686
+      std_actor_loss:                           0.002177246
+      std_advantage:                            0.010651983
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.9689454e-05
+      std_entropy:                              0.0034830566
+      std_grad_norm_pre_clip:                   2.5600839
+      std_loss:                                 0.0021879778
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.135098
+    avg_avg_return:                             0.97357285
+    avg_benchmark_regret:                       0.016056653
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13252258
+    proxy_corner:
+      avg_avg_return:                           0.97357285
+      avg_reward_per_step:                      0.13252258
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.76398
+    avg_avg_return:                             0.037973225
+    avg_benchmark_regret:                       0.9547983
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0059509277
+    proxy_corner:
+      avg_avg_return:                           0.8986113
+      avg_reward_per_step:                      0.0073242188
+  eval-batch-tree:
+    avg_avg_episode_length:                     69.61767
+    avg_avg_return:                             0.22528115
+    avg_benchmark_regret:                       0.7587944
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.03526306
+    proxy_corner:
+      avg_avg_return:                           0.22528115
+      avg_reward_per_step:                      0.03526306
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        5242880
+    env-step-replay-before:                     2555904
+    env-step-used-before:                       5242880
+    ppo-update-before:                          3200
+    env-step-generate-before:                   2686976
+    env-step-all-after:                         5275648
+    env-step-replay-after:                      2555904
+    env-step-used-after:                        5275648
+    ppo-update-after:                           3220
+    env-step-generate-after:                    2719744
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.808855
+    avg_avg_return:                             0.93038654
+    avg_reward_per_step:                        0.14840698
+    proxy_corner:
+      avg_avg_return:                           0.93038654
+      avg_reward_per_step:                      0.14840698
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.808855
+    avg_avg_return:                             0.93038654
+    avg_reward_per_step:                        0.14840698
+    proxy_corner:
+      avg_avg_return:                           0.93038654
+      avg_reward_per_step:                      0.14840698
+  perf:
+    env_steps_per_second:                       724893.9534221827
+    ppo_updates_per_second:                     42.320361315649095
+    eval-batch-orig-duration:                   0.3099415237084031
+    eval-batch-shift-duration:                  0.2716065142303705
+    eval-batch-tree-duration:                   0.24142913799732924
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             15.70459
+      mouse-cheese_dist_finite_wavg:            17.745556
+      mouse-cheese_dist_solvable_avg:           15.70459
+      mouse-cheese_dist_solvable_wavg:          17.74556
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.110263586
+    visit_patterns:
+      avg_first_visit_time:                     34.694336
+      avg_last_visit_time:                      68.28906
+      num_replay_batches:                       78
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002087685
+    avg_actor_approxkl3:                        0.0021993087
+    avg_actor_clipfrac:                         0.07398682
+    avg_actor_loss:                             0.0024267219
+    avg_advantage:                              -0.054153025
+    avg_critic_clipfrac:                        0.54193115
+    avg_critic_loss:                            0.0035858192
+    avg_entropy:                                0.121768825
+    avg_grad_norm_pre_clip:                     4.922919
+    avg_loss:                                   0.004097862
+    max:
+      max_actor_approxkl1:                      0.0073133777
+      max_actor_approxkl3:                      0.008099361
+      max_actor_clipfrac:                       0.1508789
+      max_actor_loss:                           0.0073377676
+      max_advantage:                            0.055915833
+      max_critic_clipfrac:                      1.0
+      max_critic_loss:                          0.010112045
+      max_entropy:                              0.13532704
+      max_grad_norm_pre_clip:                   10.112914
+      max_loss:                                 0.012046065
+    std:
+      std_actor_approxkl1:                      0.0021274071
+      std_actor_approxkl3:                      0.002108788
+      std_actor_clipfrac:                       0.04696177
+      std_actor_loss:                           0.0025442038
+      std_advantage:                            0.021010134
+      std_critic_clipfrac:                      0.46666008
+      std_critic_loss:                          0.003097706
+      std_entropy:                              0.007683881
+      std_grad_norm_pre_clip:                   3.119512
+      std_loss:                                 0.0035029582
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.137779
+    avg_avg_return:                             0.97460955
+    avg_benchmark_regret:                       0.015019968
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13254547
+    proxy_corner:
+      avg_avg_return:                           0.97460955
+      avg_reward_per_step:                      0.13254547
+  eval-batch-shift:
+    avg_avg_episode_length:                     83.39502
+    avg_avg_return:                             0.032581225
+    avg_benchmark_regret:                       0.9601903
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0049591064
+    proxy_corner:
+      avg_avg_return:                           0.9177175
+      avg_reward_per_step:                      0.0074386597
+  eval-batch-tree:
+    avg_avg_episode_length:                     71.19681
+    avg_avg_return:                             0.21176946
+    avg_benchmark_regret:                       0.77230614
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.03466034
+    proxy_corner:
+      avg_avg_return:                           0.21176946
+      avg_reward_per_step:                      0.03466034
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        6291456
+    env-step-replay-before:                     3112960
+    env-step-used-before:                       6291456
+    ppo-update-before:                          3840
+    env-step-generate-before:                   3178496
+    env-step-all-after:                         6324224
+    env-step-replay-after:                      3112960
+    env-step-used-after:                        6324224
+    ppo-update-after:                           3860
+    env-step-generate-after:                    3211264
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.9134655
+    avg_avg_return:                             0.9256674
+    avg_reward_per_step:                        0.14901733
+    proxy_corner:
+      avg_avg_return:                           0.9256674
+      avg_reward_per_step:                      0.14901733
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.9134655
+    avg_avg_return:                             0.9256674
+    avg_reward_per_step:                        0.14901733
+    proxy_corner:
+      avg_avg_return:                           0.9256674
+      avg_reward_per_step:                      0.14901733
+  perf:
+    env_steps_per_second:                       320854.2476068164
+    ppo_updates_per_second:                     42.459934639002114
+    eval-batch-orig-duration:                   0.5324812475591898
+    eval-batch-shift-duration:                  0.5037821847945452
+    eval-batch-tree-duration:                   0.487756154499948
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             16.086182
+      mouse-cheese_dist_finite_wavg:            17.786701
+      mouse-cheese_dist_solvable_avg:           16.086182
+      mouse-cheese_dist_solvable_wavg:          17.786705
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11218117
+    visit_patterns:
+      avg_first_visit_time:                     42.627686
+      avg_last_visit_time:                      85.23828
+      num_replay_batches:                       95
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0007212241
+    avg_actor_approxkl3:                        0.0006903023
+    avg_actor_clipfrac:                         0.025720214
+    avg_actor_loss:                             0.00060560595
+    avg_advantage:                              -0.007716791
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            9.004886e-05
+    avg_entropy:                                0.113802336
+    avg_grad_norm_pre_clip:                     2.6891947
+    avg_loss:                                   0.000536828
+    max:
+      max_actor_approxkl1:                      0.0026238249
+      max_actor_approxkl3:                      0.0022634924
+      max_actor_clipfrac:                       0.04736328
+      max_actor_loss:                           0.002666749
+      max_advantage:                            0.061311245
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00022698584
+      max_entropy:                              0.13542819
+      max_grad_norm_pre_clip:                   6.5610967
+      max_loss:                                 0.0025947753
+    std:
+      std_actor_approxkl1:                      0.00079530285
+      std_actor_approxkl3:                      0.00055886316
+      std_actor_clipfrac:                       0.0137926545
+      std_actor_loss:                           0.0008018068
+      std_advantage:                            0.010870881
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.7291553e-05
+      std_entropy:                              0.008880227
+      std_grad_norm_pre_clip:                   1.8241853
+      std_loss:                                 0.0007939182
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.185993
+    avg_avg_return:                             0.9721751
+    avg_benchmark_regret:                       0.017454362
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.130867
+    proxy_corner:
+      avg_avg_return:                           0.9721751
+      avg_reward_per_step:                      0.130867
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.79398
+    avg_avg_return:                             0.047154896
+    avg_benchmark_regret:                       0.94561666
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.009895325
+    proxy_corner:
+      avg_avg_return:                           0.84702694
+      avg_reward_per_step:                      0.006881714
+  eval-batch-tree:
+    avg_avg_episode_length:                     70.031624
+    avg_avg_return:                             0.22060648
+    avg_benchmark_regret:                       0.7634691
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.034324646
+    proxy_corner:
+      avg_avg_return:                           0.22060648
+      avg_reward_per_step:                      0.034324646
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        7340032
+    env-step-replay-before:                     3571712
+    env-step-used-before:                       7340032
+    ppo-update-before:                          4480
+    env-step-generate-before:                   3768320
+    env-step-all-after:                         7372800
+    env-step-replay-after:                      3571712
+    env-step-used-after:                        7372800
+    ppo-update-after:                           4500
+    env-step-generate-after:                    3801088
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.46225
+    avg_avg_return:                             0.9245291
+    avg_reward_per_step:                        0.12564087
+    proxy_corner:
+      avg_avg_return:                           0.9245291
+      avg_reward_per_step:                      0.12564087
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.46225
+    avg_avg_return:                             0.9245291
+    avg_reward_per_step:                        0.12564087
+    proxy_corner:
+      avg_avg_return:                           0.9245291
+      avg_reward_per_step:                      0.12564087
+  perf:
+    env_steps_per_second:                       241618.53596188867
+    ppo_updates_per_second:                     42.402577436058074
+    eval-batch-orig-duration:                   0.6726413760334253
+    eval-batch-shift-duration:                  0.6162501312792301
+    eval-batch-tree-duration:                   0.5847115982323885
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             16.48999
+      mouse-cheese_dist_finite_wavg:            17.835196
+      mouse-cheese_dist_solvable_avg:           16.48999
+      mouse-cheese_dist_solvable_wavg:          17.835196
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.11579266
+    visit_patterns:
+      avg_first_visit_time:                     52.09619
+      avg_last_visit_time:                      99.25659
+      num_replay_batches:                       109
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009382935
+    avg_actor_approxkl3:                        0.0009573305
+    avg_actor_clipfrac:                         0.022839356
+    avg_actor_loss:                             0.0011915129
+    avg_advantage:                              0.0043241563
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00017911122
+    avg_entropy:                                0.06398555
+    avg_grad_norm_pre_clip:                     2.2535887
+    avg_loss:                                   0.0012170831
+    max:
+      max_actor_approxkl1:                      0.003996996
+      max_actor_approxkl3:                      0.003354208
+      max_actor_clipfrac:                       0.045288086
+      max_actor_loss:                           0.004789412
+      max_advantage:                            0.050864756
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.000484206
+      max_entropy:                              0.07692629
+      max_grad_norm_pre_clip:                   4.2976646
+      max_loss:                                 0.004749484
+    std:
+      std_actor_approxkl1:                      0.0009896487
+      std_actor_approxkl3:                      0.0007704836
+      std_actor_clipfrac:                       0.012649978
+      std_actor_loss:                           0.0010845426
+      std_advantage:                            0.009690416
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00013533575
+      std_entropy:                              0.0064726225
+      std_grad_norm_pre_clip:                   1.0711634
+      std_loss:                                 0.0010786189
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.123999
+    avg_avg_return:                             0.9726421
+    avg_benchmark_regret:                       0.016987406
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13240814
+    proxy_corner:
+      avg_avg_return:                           0.9726421
+      avg_reward_per_step:                      0.13240814
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.3639
+    avg_avg_return:                             0.041517433
+    avg_benchmark_regret:                       0.9512541
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.007865906
+    proxy_corner:
+      avg_avg_return:                           0.88642794
+      avg_reward_per_step:                      0.007217407
+  eval-batch-tree:
+    avg_avg_episode_length:                     71.8952
+    avg_avg_return:                             0.20528571
+    avg_benchmark_regret:                       0.77878994
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.032569885
+    proxy_corner:
+      avg_avg_return:                           0.20528571
+      avg_reward_per_step:                      0.032569885
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        8388608
+    env-step-replay-before:                     4063232
+    env-step-used-before:                       8388608
+    ppo-update-before:                          5120
+    env-step-generate-before:                   4325376
+    env-step-all-after:                         8421376
+    env-step-replay-after:                      4096000
+    env-step-used-after:                        8421376
+    ppo-update-after:                           5140
+    env-step-generate-after:                    4325376
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     16.9012
+    avg_avg_return:                             0.87054634
+    avg_reward_per_step:                        0.05606079
+    proxy_corner:
+      avg_avg_return:                           0.87054634
+      avg_reward_per_step:                      0.05606079
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     16.9012
+    avg_avg_return:                             0.87054634
+    avg_reward_per_step:                        0.05606079
+    proxy_corner:
+      avg_avg_return:                           0.87054634
+      avg_reward_per_step:                      0.05606079
+  perf:
+    env_steps_per_second:                       288654.9641235881
+    ppo_updates_per_second:                     42.21965941531663
+    eval-batch-orig-duration:                   0.5314357029274106
+    eval-batch-shift-duration:                  0.4761794190853834
+    eval-batch-tree-duration:                   0.45005959924310446
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             16.773682
+      mouse-cheese_dist_finite_wavg:            18.76834
+      mouse-cheese_dist_solvable_avg:           16.773682
+      mouse-cheese_dist_solvable_wavg:          18.768341
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.116193354
+    visit_patterns:
+      avg_first_visit_time:                     58.956543
+      avg_last_visit_time:                      115.3186
+      num_replay_batches:                       125
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00969709
+    avg_actor_approxkl3:                        0.009649103
+    avg_actor_clipfrac:                         0.09423828
+    avg_actor_loss:                             0.00717612
+    avg_advantage:                              -0.0004518732
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            3.884155e-05
+    avg_entropy:                                0.11058372
+    avg_grad_norm_pre_clip:                     5.836568
+    avg_loss:                                   0.0070849583
+    max:
+      max_actor_approxkl1:                      0.031783707
+      max_actor_approxkl3:                      0.032405484
+      max_actor_clipfrac:                       0.15246582
+      max_actor_loss:                           0.0256912
+      max_advantage:                            0.047718585
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          7.860356e-05
+      max_entropy:                              0.11928111
+      max_grad_norm_pre_clip:                   11.71396
+      max_loss:                                 0.025620578
+    std:
+      std_actor_approxkl1:                      0.009854319
+      std_actor_approxkl3:                      0.009853854
+      std_actor_clipfrac:                       0.03895843
+      std_actor_loss:                           0.007428713
+      std_advantage:                            0.00719385
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.5746215e-05
+      std_entropy:                              0.004265172
+      std_grad_norm_pre_clip:                   3.1275153
+      std_loss:                                 0.007436557
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.060718
+    avg_avg_return:                             0.9741701
+    avg_benchmark_regret:                       0.0154594155
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13332367
+    proxy_corner:
+      avg_avg_return:                           0.9741701
+      avg_reward_per_step:                      0.13332367
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.34892
+    avg_avg_return:                             0.04007756
+    avg_benchmark_regret:                       0.95269394
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.008155823
+    proxy_corner:
+      avg_avg_return:                           0.87370944
+      avg_reward_per_step:                      0.0071105957
+  eval-batch-tree:
+    avg_avg_episode_length:                     71.05549
+    avg_avg_return:                             0.21489681
+    avg_benchmark_regret:                       0.76917875
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.033584595
+    proxy_corner:
+      avg_avg_return:                           0.21489681
+      avg_reward_per_step:                      0.033584595
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        9437184
+    env-step-replay-before:                     4653056
+    env-step-used-before:                       9437184
+    ppo-update-before:                          5760
+    env-step-generate-before:                   4784128
+    env-step-all-after:                         9469952
+    env-step-replay-after:                      4653056
+    env-step-used-after:                        9469952
+    ppo-update-after:                           5780
+    env-step-generate-after:                    4816896
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.319185
+    avg_avg_return:                             0.92713684
+    avg_reward_per_step:                        0.12878418
+    proxy_corner:
+      avg_avg_return:                           0.92713684
+      avg_reward_per_step:                      0.12878418
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.319185
+    avg_avg_return:                             0.92713684
+    avg_reward_per_step:                        0.12878418
+    proxy_corner:
+      avg_avg_return:                           0.92713684
+      avg_reward_per_step:                      0.12878418
+  perf:
+    env_steps_per_second:                       261118.12806666765
+    ppo_updates_per_second:                     41.55471033910409
+    eval-batch-orig-duration:                   0.5284560974687338
+    eval-batch-shift-duration:                  0.4919081823900342
+    eval-batch-tree-duration:                   0.48510470427572727
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             17.01416
+      mouse-cheese_dist_finite_wavg:            18.790684
+      mouse-cheese_dist_solvable_avg:           17.01416
+      mouse-cheese_dist_solvable_wavg:          18.790684
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.118802175
+    visit_patterns:
+      avg_first_visit_time:                     65.77222
+      avg_last_visit_time:                      132.31787
+      num_replay_batches:                       142
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011078824
+    avg_actor_approxkl3:                        0.0010161927
+    avg_actor_clipfrac:                         0.017779542
+    avg_actor_loss:                             0.00010549425
+    avg_advantage:                              0.0007509374
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.2496077e-05
+    avg_entropy:                                0.031826552
+    avg_grad_norm_pre_clip:                     1.058443
+    avg_loss:                                   8.491573e-05
+    max:
+      max_actor_approxkl1:                      0.0037703442
+      max_actor_approxkl3:                      0.0029311222
+      max_actor_clipfrac:                       0.036132812
+      max_actor_loss:                           0.0012315195
+      max_advantage:                            0.034939885
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.8692071e-05
+      max_entropy:                              0.0390324
+      max_grad_norm_pre_clip:                   2.7083976
+      max_loss:                                 0.0012179699
+    std:
+      std_actor_approxkl1:                      0.0009647866
+      std_actor_approxkl3:                      0.0007666839
+      std_actor_clipfrac:                       0.008763023
+      std_actor_loss:                           0.0004979909
+      std_advantage:                            0.0068407683
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.7711885e-06
+      std_entropy:                              0.0037190684
+      std_grad_norm_pre_clip:                   0.7178586
+      std_loss:                                 0.0004991108
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.048646
+    avg_avg_return:                             0.9738847
+    avg_benchmark_regret:                       0.015744857
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13344574
+    proxy_corner:
+      avg_avg_return:                           0.9738847
+      avg_reward_per_step:                      0.13344574
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.60619
+    avg_avg_return:                             0.036210626
+    avg_benchmark_regret:                       0.95656085
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0071029663
+    proxy_corner:
+      avg_avg_return:                           0.84858835
+      avg_reward_per_step:                      0.006843567
+  eval-batch-tree:
+    avg_avg_episode_length:                     70.33615
+    avg_avg_return:                             0.21996409
+    avg_benchmark_regret:                       0.7641115
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.03477478
+    proxy_corner:
+      avg_avg_return:                           0.21996409
+      avg_reward_per_step:                      0.03477478
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        10485760
+    env-step-replay-before:                     5210112
+    env-step-used-before:                       10485760
+    ppo-update-before:                          6400
+    env-step-generate-before:                   5275648
+    env-step-all-after:                         10518528
+    env-step-replay-after:                      5242880
+    env-step-used-after:                        10518528
+    ppo-update-after:                           6420
+    env-step-generate-after:                    5275648
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     17.243534
+    avg_avg_return:                             0.8619291
+    avg_reward_per_step:                        0.054748535
+    proxy_corner:
+      avg_avg_return:                           0.8619291
+      avg_reward_per_step:                      0.054748535
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     17.243534
+    avg_avg_return:                             0.8619291
+    avg_reward_per_step:                        0.054748535
+    proxy_corner:
+      avg_avg_return:                           0.8619291
+      avg_reward_per_step:                      0.054748535
+  perf:
+    env_steps_per_second:                       310765.088497973
+    ppo_updates_per_second:                     42.17077660752027
+    eval-batch-orig-duration:                   0.5313782449811697
+    eval-batch-shift-duration:                  0.4618430556729436
+    eval-batch-tree-duration:                   0.46105689089745283
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             17.226807
+      mouse-cheese_dist_finite_wavg:            18.816965
+      mouse-cheese_dist_solvable_avg:           17.226807
+      mouse-cheese_dist_solvable_wavg:          18.816965
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.121290445
+    visit_patterns:
+      avg_first_visit_time:                     72.20361
+      avg_last_visit_time:                      150.37427
+      num_replay_batches:                       160
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.003034634
+    avg_actor_approxkl3:                        0.0035803826
+    avg_actor_clipfrac:                         0.022558594
+    avg_actor_loss:                             0.0049594785
+    avg_advantage:                              0.0005592277
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.6790585e-05
+    avg_entropy:                                0.034563214
+    avg_grad_norm_pre_clip:                     3.4355085
+    avg_loss:                                   0.0049383114
+    max:
+      max_actor_approxkl1:                      0.014371714
+      max_actor_approxkl3:                      0.014842142
+      max_actor_clipfrac:                       0.06555176
+      max_actor_loss:                           0.01826027
+      max_advantage:                            0.03034389
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          5.758856e-05
+      max_entropy:                              0.052629616
+      max_grad_norm_pre_clip:                   11.488633
+      max_loss:                                 0.018232971
+    std:
+      std_actor_approxkl1:                      0.0035487255
+      std_actor_approxkl3:                      0.004200796
+      std_actor_clipfrac:                       0.01875298
+      std_actor_loss:                           0.005477576
+      std_advantage:                            0.005627062
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.2655432e-05
+      std_entropy:                              0.008254451
+      std_grad_norm_pre_clip:                   3.0484917
+      std_loss:                                 0.005477261
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.042346
+    avg_avg_return:                             0.97397476
+    avg_benchmark_regret:                       0.015654765
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13345337
+    proxy_corner:
+      avg_avg_return:                           0.97397476
+      avg_reward_per_step:                      0.13345337
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.35789
+    avg_avg_return:                             0.039315775
+    avg_benchmark_regret:                       0.95345575
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.009262085
+    proxy_corner:
+      avg_avg_return:                           0.8417796
+      avg_reward_per_step:                      0.006767273
+  eval-batch-tree:
+    avg_avg_episode_length:                     70.55466
+    avg_avg_return:                             0.22024284
+    avg_benchmark_regret:                       0.76383275
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.034080505
+    proxy_corner:
+      avg_avg_return:                           0.22024284
+      avg_reward_per_step:                      0.034080505
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        11534336
+    env-step-replay-before:                     5767168
+    env-step-used-before:                       11534336
+    ppo-update-before:                          7040
+    env-step-generate-before:                   5767168
+    env-step-all-after:                         11567104
+    env-step-replay-after:                      5767168
+    env-step-used-after:                        11567104
+    ppo-update-after:                           7060
+    env-step-generate-after:                    5799936
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.441621
+    avg_avg_return:                             0.9245204
+    avg_reward_per_step:                        0.12680054
+    proxy_corner:
+      avg_avg_return:                           0.9245204
+      avg_reward_per_step:                      0.12680054
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.441621
+    avg_avg_return:                             0.9245204
+    avg_reward_per_step:                        0.12680054
+    proxy_corner:
+      avg_avg_return:                           0.9245204
+      avg_reward_per_step:                      0.12680054
+  perf:
+    env_steps_per_second:                       294489.14465126354
+    ppo_updates_per_second:                     42.31412745979216
+    eval-batch-orig-duration:                   0.5302664050832391
+    eval-batch-shift-duration:                  0.4786914000287652
+    eval-batch-tree-duration:                   0.4769631242379546
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             17.441162
+      mouse-cheese_dist_finite_wavg:            19.735014
+      mouse-cheese_dist_solvable_avg:           17.441162
+      mouse-cheese_dist_solvable_wavg:          19.735016
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12388961
+    visit_patterns:
+      avg_first_visit_time:                     79.562256
+      avg_last_visit_time:                      166.25464
+      num_replay_batches:                       176
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00069501944
+    avg_actor_approxkl3:                        0.00061565824
+    avg_actor_clipfrac:                         0.010992432
+    avg_actor_loss:                             0.0011409634
+    avg_advantage:                              0.0011050825
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.127598e-05
+    avg_entropy:                                0.023449436
+    avg_grad_norm_pre_clip:                     2.022676
+    avg_loss:                                   0.0011381521
+    max:
+      max_actor_approxkl1:                      0.0017046976
+      max_actor_approxkl3:                      0.0019738534
+      max_actor_clipfrac:                       0.023071289
+      max_actor_loss:                           0.0056707337
+      max_advantage:                            0.05347246
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          8.309791e-05
+      max_entropy:                              0.031070422
+      max_grad_norm_pre_clip:                   4.138536
+      max_loss:                                 0.005667076
+    std:
+      std_actor_approxkl1:                      0.00055740133
+      std_actor_approxkl3:                      0.0005407804
+      std_actor_clipfrac:                       0.0063329055
+      std_actor_loss:                           0.0015059878
+      std_advantage:                            0.006713084
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.7322303e-05
+      std_entropy:                              0.0029021627
+      std_grad_norm_pre_clip:                   1.181753
+      std_loss:                                 0.0015083525
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.042921
+    avg_avg_return:                             0.97386444
+    avg_benchmark_regret:                       0.015765052
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.133461
+    proxy_corner:
+      avg_avg_return:                           0.97386444
+      avg_reward_per_step:                      0.133461
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.59505
+    avg_avg_return:                             0.037026763
+    avg_benchmark_regret:                       0.95574474
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.00983429
+    proxy_corner:
+      avg_avg_return:                           0.8687664
+      avg_reward_per_step:                      0.007003784
+  eval-batch-tree:
+    avg_avg_episode_length:                     75.36667
+    avg_avg_return:                             0.15566045
+    avg_benchmark_regret:                       0.82841516
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.029632568
+    proxy_corner:
+      avg_avg_return:                           0.15566045
+      avg_reward_per_step:                      0.029632568
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        12582912
+    env-step-replay-before:                     6291456
+    env-step-used-before:                       12582912
+    ppo-update-before:                          7680
+    env-step-generate-before:                   6291456
+    env-step-all-after:                         12615680
+    env-step-replay-after:                      6324224
+    env-step-used-after:                        12615680
+    ppo-update-after:                           7700
+    env-step-generate-after:                    6291456
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     17.524267
+    avg_avg_return:                             0.8589258
+    avg_reward_per_step:                        0.05355835
+    proxy_corner:
+      avg_avg_return:                           0.8589258
+      avg_reward_per_step:                      0.05355835
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     17.524267
+    avg_avg_return:                             0.8589258
+    avg_reward_per_step:                        0.05355835
+    proxy_corner:
+      avg_avg_return:                           0.8589258
+      avg_reward_per_step:                      0.05355835
+  perf:
+    env_steps_per_second:                       314607.63573227526
+    ppo_updates_per_second:                     42.23930092930966
+    eval-batch-orig-duration:                   0.5449402071535587
+    eval-batch-shift-duration:                  0.4924004301428795
+    eval-batch-tree-duration:                   0.482553387992084
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             17.635986
+      mouse-cheese_dist_finite_wavg:            18.85937
+      mouse-cheese_dist_solvable_avg:           17.635986
+      mouse-cheese_dist_solvable_wavg:          18.859371
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1257728
+    visit_patterns:
+      avg_first_visit_time:                     87.88452
+      avg_last_visit_time:                      183.36572
+      num_replay_batches:                       193
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00053600094
+    avg_actor_approxkl3:                        0.00041189627
+    avg_actor_clipfrac:                         0.011187744
+    avg_actor_loss:                             0.000609923
+    avg_advantage:                              -0.0036481465
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.9983547e-05
+    avg_entropy:                                0.026380071
+    avg_grad_norm_pre_clip:                     1.4467386
+    avg_loss:                                   0.00059353467
+    max:
+      max_actor_approxkl1:                      0.0017043028
+      max_actor_approxkl3:                      0.0016310262
+      max_actor_clipfrac:                       0.034423828
+      max_actor_loss:                           0.0022610612
+      max_advantage:                            0.02002716
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          8.128193e-05
+      max_entropy:                              0.030088421
+      max_grad_norm_pre_clip:                   4.06274
+      max_loss:                                 0.0022388294
+    std:
+      std_actor_approxkl1:                      0.0004840148
+      std_actor_approxkl3:                      0.00046072638
+      std_actor_clipfrac:                       0.011253154
+      std_actor_loss:                           0.00075987267
+      std_advantage:                            0.004427797
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.722443e-05
+      std_entropy:                              0.0024943422
+      std_grad_norm_pre_clip:                   1.0800654
+      std_loss:                                 0.0007592539
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.044148
+    avg_avg_return:                             0.9738029
+    avg_benchmark_regret:                       0.015826568
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.133461
+    proxy_corner:
+      avg_avg_return:                           0.9738029
+      avg_reward_per_step:                      0.133461
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.65474
+    avg_avg_return:                             0.037024315
+    avg_benchmark_regret:                       0.9557472
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0104522705
+    proxy_corner:
+      avg_avg_return:                           0.8723217
+      avg_reward_per_step:                      0.007019043
+  eval-batch-tree:
+    avg_avg_episode_length:                     77.77606
+    avg_avg_return:                             0.11529745
+    avg_benchmark_regret:                       0.86877817
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.026275635
+    proxy_corner:
+      avg_avg_return:                           0.11529745
+      avg_reward_per_step:                      0.026275635
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        13631488
+    env-step-replay-before:                     6782976
+    env-step-used-before:                       13631488
+    ppo-update-before:                          8320
+    env-step-generate-before:                   6848512
+    env-step-all-after:                         13664256
+    env-step-replay-after:                      6782976
+    env-step-used-after:                        13664256
+    ppo-update-after:                           8340
+    env-step-generate-after:                    6881280
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.762972
+    avg_avg_return:                             0.9285226
+    avg_reward_per_step:                        0.14230347
+    proxy_corner:
+      avg_avg_return:                           0.9285226
+      avg_reward_per_step:                      0.14230347
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.762972
+    avg_avg_return:                             0.9285226
+    avg_reward_per_step:                        0.14230347
+    proxy_corner:
+      avg_avg_return:                           0.9285226
+      avg_reward_per_step:                      0.14230347
+  perf:
+    env_steps_per_second:                       317644.8960892372
+    ppo_updates_per_second:                     42.142332489600456
+    eval-batch-orig-duration:                   0.5327810663729906
+    eval-batch-shift-duration:                  0.48858635406941175
+    eval-batch-tree-duration:                   0.4730956247076392
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             17.788086
+      mouse-cheese_dist_finite_wavg:            18.873564
+      mouse-cheese_dist_solvable_avg:           17.788086
+      mouse-cheese_dist_solvable_wavg:          18.873564
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12734038
+    visit_patterns:
+      avg_first_visit_time:                     97.203125
+      avg_last_visit_time:                      197.32715
+      num_replay_batches:                       207
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0005929548
+    avg_actor_approxkl3:                        0.00038720295
+    avg_actor_clipfrac:                         0.008929444
+    avg_actor_loss:                             0.0002723315
+    avg_advantage:                              -0.00044710343
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.1619415e-05
+    avg_entropy:                                0.023091262
+    avg_grad_norm_pre_clip:                     0.9668733
+    avg_loss:                                   0.00026004997
+    max:
+      max_actor_approxkl1:                      0.0015002936
+      max_actor_approxkl3:                      0.0012838377
+      max_actor_clipfrac:                       0.023803711
+      max_actor_loss:                           0.002562462
+      max_advantage:                            0.034433663
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.6661384e-05
+      max_entropy:                              0.029805925
+      max_grad_norm_pre_clip:                   3.1274295
+      max_loss:                                 0.0025487018
+    std:
+      std_actor_approxkl1:                      0.0005435451
+      std_actor_approxkl3:                      0.00033408197
+      std_actor_clipfrac:                       0.006670366
+      std_actor_loss:                           0.00077354757
+      std_advantage:                            0.0066773654
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.321989e-06
+      std_entropy:                              0.0033731475
+      std_grad_norm_pre_clip:                   0.73780847
+      std_loss:                                 0.00077378243
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040624
+    avg_avg_return:                             0.9738058
+    avg_benchmark_regret:                       0.015823705
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13350677
+    proxy_corner:
+      avg_avg_return:                           0.9738058
+      avg_reward_per_step:                      0.13350677
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.745705
+    avg_avg_return:                             0.03434448
+    avg_benchmark_regret:                       0.9584271
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.010177612
+    proxy_corner:
+      avg_avg_return:                           0.8473047
+      avg_reward_per_step:                      0.00680542
+  eval-batch-tree:
+    avg_avg_episode_length:                     78.06136
+    avg_avg_return:                             0.1074532
+    avg_benchmark_regret:                       0.8766224
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025810242
+    proxy_corner:
+      avg_avg_return:                           0.1074532
+      avg_reward_per_step:                      0.025810242
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        14680064
+    env-step-replay-before:                     7405568
+    env-step-used-before:                       14680064
+    ppo-update-before:                          8960
+    env-step-generate-before:                   7274496
+    env-step-all-after:                         14712832
+    env-step-replay-after:                      7438336
+    env-step-used-after:                        14712832
+    ppo-update-after:                           8980
+    env-step-generate-after:                    7274496
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     17.941479
+    avg_avg_return:                             0.8553901
+    avg_reward_per_step:                        0.05218506
+    proxy_corner:
+      avg_avg_return:                           0.8553901
+      avg_reward_per_step:                      0.05218506
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     17.941479
+    avg_avg_return:                             0.8553901
+    avg_reward_per_step:                        0.05218506
+    proxy_corner:
+      avg_avg_return:                           0.8553901
+      avg_reward_per_step:                      0.05218506
+  perf:
+    env_steps_per_second:                       321872.73140816065
+    ppo_updates_per_second:                     41.91162379989908
+    eval-batch-orig-duration:                   0.5328430477529764
+    eval-batch-shift-duration:                  0.48410843312740326
+    eval-batch-tree-duration:                   0.4481287682428956
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             17.867188
+      mouse-cheese_dist_finite_wavg:            18.884958
+      mouse-cheese_dist_solvable_avg:           17.867188
+      mouse-cheese_dist_solvable_wavg:          18.88496
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12806109
+    visit_patterns:
+      avg_first_visit_time:                     102.89575
+      avg_last_visit_time:                      217.2732
+      num_replay_batches:                       227
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009587283
+    avg_actor_approxkl3:                        0.0008205074
+    avg_actor_clipfrac:                         0.0069885254
+    avg_actor_loss:                             0.0029901469
+    avg_advantage:                              -0.0011539105
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.9838266e-05
+    avg_entropy:                                0.0133385435
+    avg_grad_norm_pre_clip:                     1.9545119
+    avg_loss:                                   0.0029867277
+    max:
+      max_actor_approxkl1:                      0.0035978504
+      max_actor_approxkl3:                      0.0025784955
+      max_actor_clipfrac:                       0.011962891
+      max_actor_loss:                           0.0077790543
+      max_advantage:                            0.019128442
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.4730416e-05
+      max_entropy:                              0.015664166
+      max_grad_norm_pre_clip:                   3.5671082
+      max_loss:                                 0.007772177
+    std:
+      std_actor_approxkl1:                      0.0010313598
+      std_actor_approxkl3:                      0.0008221028
+      std_actor_clipfrac:                       0.0033049001
+      std_actor_loss:                           0.0029982044
+      std_advantage:                            0.00482185
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          6.977345e-06
+      std_entropy:                              0.0013658312
+      std_grad_norm_pre_clip:                   1.0155128
+      std_loss:                                 0.0029982931
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.052204
+    avg_avg_return:                             0.9732562
+    avg_benchmark_regret:                       0.016373366
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13339233
+    proxy_corner:
+      avg_avg_return:                           0.9732562
+      avg_reward_per_step:                      0.13339233
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.37085
+    avg_avg_return:                             0.03919902
+    avg_benchmark_regret:                       0.9535725
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.009902954
+    proxy_corner:
+      avg_avg_return:                           0.85338503
+      avg_reward_per_step:                      0.006904602
+  eval-batch-tree:
+    avg_avg_episode_length:                     77.90552
+    avg_avg_return:                             0.10956207
+    avg_benchmark_regret:                       0.87451357
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025733948
+    proxy_corner:
+      avg_avg_return:                           0.10956207
+      avg_reward_per_step:                      0.025733948
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        15728640
+    env-step-replay-before:                     7766016
+    env-step-used-before:                       15728640
+    ppo-update-before:                          9600
+    env-step-generate-before:                   7962624
+    env-step-all-after:                         15761408
+    env-step-replay-after:                      7766016
+    env-step-used-after:                        15761408
+    ppo-update-after:                           9620
+    env-step-generate-after:                    7995392
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.013848
+    avg_avg_return:                             0.9291249
+    avg_reward_per_step:                        0.14498901
+    proxy_corner:
+      avg_avg_return:                           0.9291249
+      avg_reward_per_step:                      0.14498901
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.013848
+    avg_avg_return:                             0.9291249
+    avg_reward_per_step:                        0.14498901
+    proxy_corner:
+      avg_avg_return:                           0.9291249
+      avg_reward_per_step:                      0.14498901
+  perf:
+    env_steps_per_second:                       289041.8732813377
+    ppo_updates_per_second:                     42.15355435646924
+    eval-batch-orig-duration:                   0.5380737353116274
+    eval-batch-shift-duration:                  0.4881650796160102
+    eval-batch-tree-duration:                   0.4869498834013939
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             17.9646
+      mouse-cheese_dist_finite_wavg:            18.89194
+      mouse-cheese_dist_solvable_avg:           17.9646
+      mouse-cheese_dist_solvable_wavg:          18.89194
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12921318
+    visit_patterns:
+      avg_first_visit_time:                     112.86255
+      avg_last_visit_time:                      227.21606
+      num_replay_batches:                       237
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00065838516
+    avg_actor_approxkl3:                        0.0007528736
+    avg_actor_clipfrac:                         0.016644288
+    avg_actor_loss:                             0.0015140608
+    avg_advantage:                              -0.0019727368
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.0290163e-05
+    avg_entropy:                                0.026257187
+    avg_grad_norm_pre_clip:                     1.6888365
+    avg_loss:                                   0.0014979487
+    max:
+      max_actor_approxkl1:                      0.002480631
+      max_actor_approxkl3:                      0.0030214097
+      max_actor_clipfrac:                       0.028808594
+      max_actor_loss:                           0.0055369497
+      max_advantage:                            0.034026742
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          4.598165e-05
+      max_entropy:                              0.03305403
+      max_grad_norm_pre_clip:                   3.9907594
+      max_loss:                                 0.0055199075
+    std:
+      std_actor_approxkl1:                      0.0007112077
+      std_actor_approxkl3:                      0.00074228307
+      std_actor_clipfrac:                       0.009362982
+      std_actor_loss:                           0.0017005836
+      std_advantage:                            0.0052393544
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          8.167462e-06
+      std_entropy:                              0.0035912793
+      std_grad_norm_pre_clip:                   1.084046
+      std_loss:                                 0.0016995085
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.04504
+    avg_avg_return:                             0.9736682
+    avg_benchmark_regret:                       0.015961297
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13346863
+    proxy_corner:
+      avg_avg_return:                           0.9736682
+      avg_reward_per_step:                      0.13346863
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.345184
+    avg_avg_return:                             0.041755714
+    avg_benchmark_regret:                       0.95101583
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.011779785
+    proxy_corner:
+      avg_avg_return:                           0.84061885
+      avg_reward_per_step:                      0.006790161
+  eval-batch-tree:
+    avg_avg_episode_length:                     78.61375
+    avg_avg_return:                             0.10175642
+    avg_benchmark_regret:                       0.8823192
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025009155
+    proxy_corner:
+      avg_avg_return:                           0.10175642
+      avg_reward_per_step:                      0.025009155
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        16777216
+    env-step-replay-before:                     8224768
+    env-step-used-before:                       16777216
+    ppo-update-before:                          10240
+    env-step-generate-before:                   8552448
+    env-step-all-after:                         16809984
+    env-step-replay-after:                      8257536
+    env-step-used-after:                        16809984
+    ppo-update-after:                           10260
+    env-step-generate-after:                    8552448
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.135143
+    avg_avg_return:                             0.8535765
+    avg_reward_per_step:                        0.05154419
+    proxy_corner:
+      avg_avg_return:                           0.8535765
+      avg_reward_per_step:                      0.05154419
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.135143
+    avg_avg_return:                             0.8535765
+    avg_reward_per_step:                        0.05154419
+    proxy_corner:
+      avg_avg_return:                           0.8535765
+      avg_reward_per_step:                      0.05154419
+  perf:
+    env_steps_per_second:                       312224.70116236893
+    ppo_updates_per_second:                     42.19055857540202
+    eval-batch-orig-duration:                   0.5296919187530875
+    eval-batch-shift-duration:                  0.4853651151061058
+    eval-batch-tree-duration:                   0.4460789989680052
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.006104
+      mouse-cheese_dist_finite_wavg:            18.89771
+      mouse-cheese_dist_solvable_avg:           18.006104
+      mouse-cheese_dist_solvable_wavg:          18.89771
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.12979418
+    visit_patterns:
+      avg_first_visit_time:                     118.40796
+      avg_last_visit_time:                      242.26562
+      num_replay_batches:                       252
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014287774
+    avg_actor_approxkl3:                        0.0010883635
+    avg_actor_clipfrac:                         0.03267212
+    avg_actor_loss:                             0.000810985
+    avg_advantage:                              -0.0005914107
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            6.5039153e-06
+    avg_entropy:                                0.05471787
+    avg_grad_norm_pre_clip:                     2.2690024
+    avg_loss:                                   0.0007595192
+    max:
+      max_actor_approxkl1:                      0.0049831956
+      max_actor_approxkl3:                      0.0049997703
+      max_actor_clipfrac:                       0.079589844
+      max_actor_loss:                           0.004770139
+      max_advantage:                            0.018440306
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          7.677852e-06
+      max_entropy:                              0.059977774
+      max_grad_norm_pre_clip:                   4.9907484
+      max_loss:                                 0.004718275
+    std:
+      std_actor_approxkl1:                      0.001267124
+      std_actor_approxkl3:                      0.0011910859
+      std_actor_clipfrac:                       0.02107477
+      std_actor_loss:                           0.0015338898
+      std_advantage:                            0.003920413
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          6.080392e-07
+      std_entropy:                              0.00288569
+      std_grad_norm_pre_clip:                   1.5605406
+      std_loss:                                 0.0015331508
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041321
+    avg_avg_return:                             0.97380733
+    avg_benchmark_regret:                       0.015822163
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349152
+    proxy_corner:
+      avg_avg_return:                           0.97380733
+      avg_reward_per_step:                      0.13349152
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.087814
+    avg_avg_return:                             0.044811618
+    avg_benchmark_regret:                       0.9479599
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.010604858
+    proxy_corner:
+      avg_avg_return:                           0.87054753
+      avg_reward_per_step:                      0.007041931
+  eval-batch-tree:
+    avg_avg_episode_length:                     80.12445
+    avg_avg_return:                             0.0711081
+    avg_benchmark_regret:                       0.91296756
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.023864746
+    proxy_corner:
+      avg_avg_return:                           0.0711081
+      avg_reward_per_step:                      0.023864746
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        17825792
+    env-step-replay-before:                     8912896
+    env-step-used-before:                       17825792
+    ppo-update-before:                          10880
+    env-step-generate-before:                   8912896
+    env-step-all-after:                         17858560
+    env-step-replay-after:                      8945664
+    env-step-used-after:                        17858560
+    ppo-update-after:                           10900
+    env-step-generate-after:                    8912896
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.273907
+    avg_avg_return:                             0.85252583
+    avg_reward_per_step:                        0.051116943
+    proxy_corner:
+      avg_avg_return:                           0.85252583
+      avg_reward_per_step:                      0.051116943
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.273907
+    avg_avg_return:                             0.85252583
+    avg_reward_per_step:                        0.051116943
+    proxy_corner:
+      avg_avg_return:                           0.85252583
+      avg_reward_per_step:                      0.051116943
+  perf:
+    env_steps_per_second:                       258804.63291607192
+    ppo_updates_per_second:                     41.53619660489731
+    eval-batch-orig-duration:                   0.5436291694641113
+    eval-batch-shift-duration:                  0.4844811912626028
+    eval-batch-tree-duration:                   0.48379990737885237
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.028809
+      mouse-cheese_dist_finite_wavg:            18.902508
+      mouse-cheese_dist_solvable_avg:           18.028809
+      mouse-cheese_dist_solvable_wavg:          18.90251
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13019595
+    visit_patterns:
+      avg_first_visit_time:                     123.632324
+      avg_last_visit_time:                      263.1599
+      num_replay_batches:                       273
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013643836
+    avg_actor_approxkl3:                        0.0014788551
+    avg_actor_clipfrac:                         0.03135376
+    avg_actor_loss:                             0.0029425507
+    avg_advantage:                              -0.0005231465
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.1810871e-05
+    avg_entropy:                                0.040778104
+    avg_grad_norm_pre_clip:                     2.9128306
+    avg_loss:                                   0.0029076783
+    max:
+      max_actor_approxkl1:                      0.005274701
+      max_actor_approxkl3:                      0.005114679
+      max_actor_clipfrac:                       0.056396484
+      max_actor_loss:                           0.01022691
+      max_advantage:                            0.030253708
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.6303617e-05
+      max_entropy:                              0.04406521
+      max_grad_norm_pre_clip:                   4.396813
+      max_loss:                                 0.010192013
+    std:
+      std_actor_approxkl1:                      0.0015307394
+      std_actor_approxkl3:                      0.0014099245
+      std_actor_clipfrac:                       0.02086075
+      std_actor_loss:                           0.002975144
+      std_advantage:                            0.0048499303
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.7771443e-06
+      std_entropy:                              0.0021516832
+      std_grad_norm_pre_clip:                   1.4083853
+      std_loss:                                 0.0029754092
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.0428915
+    avg_avg_return:                             0.97376215
+    avg_benchmark_regret:                       0.015867341
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13346863
+    proxy_corner:
+      avg_avg_return:                           0.97376215
+      avg_reward_per_step:                      0.13346863
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.22436
+    avg_avg_return:                             0.054426752
+    avg_benchmark_regret:                       0.9383447
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.012199402
+    proxy_corner:
+      avg_avg_return:                           0.85547066
+      avg_reward_per_step:                      0.006958008
+  eval-batch-tree:
+    avg_avg_episode_length:                     80.436295
+    avg_avg_return:                             0.06342747
+    avg_benchmark_regret:                       0.9206481
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02368164
+    proxy_corner:
+      avg_avg_return:                           0.06342747
+      avg_reward_per_step:                      0.02368164
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        18874368
+    env-step-replay-before:                     9633792
+    env-step-used-before:                       18874368
+    ppo-update-before:                          11520
+    env-step-generate-before:                   9240576
+    env-step-all-after:                         18907136
+    env-step-replay-after:                      9666560
+    env-step-used-after:                        18907136
+    ppo-update-after:                           11540
+    env-step-generate-after:                    9240576
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.225643
+    avg_avg_return:                             0.8524932
+    avg_reward_per_step:                        0.051239014
+    proxy_corner:
+      avg_avg_return:                           0.8524932
+      avg_reward_per_step:                      0.051239014
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.225643
+    avg_avg_return:                             0.8524932
+    avg_reward_per_step:                        0.051239014
+    proxy_corner:
+      avg_avg_return:                           0.8524932
+      avg_reward_per_step:                      0.051239014
+  perf:
+    env_steps_per_second:                       346256.5903354198
+    ppo_updates_per_second:                     43.385828437216254
+    eval-batch-orig-duration:                   0.5067397588863969
+    eval-batch-shift-duration:                  0.4898585919290781
+    eval-batch-tree-duration:                   0.4815137358382344
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.069092
+      mouse-cheese_dist_finite_wavg:            18.907013
+      mouse-cheese_dist_solvable_avg:           18.069092
+      mouse-cheese_dist_solvable_wavg:          18.907011
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13065022
+    visit_patterns:
+      avg_first_visit_time:                     127.201904
+      avg_last_visit_time:                      285.18872
+      num_replay_batches:                       295
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00072592514
+    avg_actor_approxkl3:                        0.0007451146
+    avg_actor_clipfrac:                         0.028186036
+    avg_actor_loss:                             0.001542174
+    avg_advantage:                              0.0008277821
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            9.473266e-06
+    avg_entropy:                                0.065497264
+    avg_grad_norm_pre_clip:                     3.202175
+    avg_loss:                                   0.0014814133
+    max:
+      max_actor_approxkl1:                      0.0032307631
+      max_actor_approxkl3:                      0.0019606333
+      max_actor_clipfrac:                       0.060180664
+      max_actor_loss:                           0.004199285
+      max_advantage:                            0.027374804
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.1915752e-05
+      max_entropy:                              0.07469052
+      max_grad_norm_pre_clip:                   5.3898997
+      max_loss:                                 0.0041423105
+    std:
+      std_actor_approxkl1:                      0.00078606856
+      std_actor_approxkl3:                      0.0005763079
+      std_actor_clipfrac:                       0.019167017
+      std_actor_loss:                           0.0012953614
+      std_advantage:                            0.0045625474
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.3674837e-06
+      std_entropy:                              0.005031508
+      std_grad_norm_pre_clip:                   1.4163344
+      std_loss:                                 0.001297118
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.042097
+    avg_avg_return:                             0.9738386
+    avg_benchmark_regret:                       0.015790908
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349152
+    proxy_corner:
+      avg_avg_return:                           0.9738386
+      avg_reward_per_step:                      0.13349152
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.97974
+    avg_avg_return:                             0.044380445
+    avg_benchmark_regret:                       0.9483911
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.013282776
+    proxy_corner:
+      avg_avg_return:                           0.8638066
+      avg_reward_per_step:                      0.006958008
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.1501
+    avg_avg_return:                             0.096773386
+    avg_benchmark_regret:                       0.8873023
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02481842
+    proxy_corner:
+      avg_avg_return:                           0.096773386
+      avg_reward_per_step:                      0.02481842
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        19922944
+    env-step-replay-before:                     10125312
+    env-step-used-before:                       19922944
+    ppo-update-before:                          12160
+    env-step-generate-before:                   9797632
+    env-step-all-after:                         19955712
+    env-step-replay-after:                      10158080
+    env-step-used-after:                        19955712
+    ppo-update-after:                           12180
+    env-step-generate-after:                    9797632
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.380535
+    avg_avg_return:                             0.8517591
+    avg_reward_per_step:                        0.050811768
+    proxy_corner:
+      avg_avg_return:                           0.8517591
+      avg_reward_per_step:                      0.050811768
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.380535
+    avg_avg_return:                             0.8517591
+    avg_reward_per_step:                        0.050811768
+    proxy_corner:
+      avg_avg_return:                           0.8517591
+      avg_reward_per_step:                      0.050811768
+  perf:
+    env_steps_per_second:                       325261.3680399123
+    ppo_updates_per_second:                     42.268020981505444
+    eval-batch-orig-duration:                   0.5350687950849533
+    eval-batch-shift-duration:                  0.4921719478443265
+    eval-batch-tree-duration:                   0.49763752426952124
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.121338
+      mouse-cheese_dist_finite_wavg:            18.911455
+      mouse-cheese_dist_solvable_avg:           18.121338
+      mouse-cheese_dist_solvable_wavg:          18.911457
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13112259
+    visit_patterns:
+      avg_first_visit_time:                     131.72827
+      avg_last_visit_time:                      300.19238
+      num_replay_batches:                       310
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0005716956
+    avg_actor_approxkl3:                        0.00063321856
+    avg_actor_clipfrac:                         0.005395508
+    avg_actor_loss:                             0.0004222203
+    avg_advantage:                              -0.010614622
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.995909e-05
+    avg_entropy:                                0.018109184
+    avg_grad_norm_pre_clip:                     0.64167714
+    avg_loss:                                   0.00041909065
+    max:
+      max_actor_approxkl1:                      0.0017675726
+      max_actor_approxkl3:                      0.0017469346
+      max_actor_clipfrac:                       0.013549805
+      max_actor_loss:                           0.0023214407
+      max_advantage:                            0.015479445
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          8.238368e-05
+      max_entropy:                              0.024628647
+      max_grad_norm_pre_clip:                   1.4561155
+      max_loss:                                 0.0023321551
+    std:
+      std_actor_approxkl1:                      0.0005628181
+      std_actor_approxkl3:                      0.0004884472
+      std_actor_clipfrac:                       0.004194958
+      std_actor_loss:                           0.00092581694
+      std_advantage:                            0.0054467674
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.8595596e-05
+      std_entropy:                              0.003430283
+      std_grad_norm_pre_clip:                   0.45988938
+      std_loss:                                 0.00093183236
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.042217
+    avg_avg_return:                             0.9737718
+    avg_benchmark_regret:                       0.015857715
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13348389
+    proxy_corner:
+      avg_avg_return:                           0.9737718
+      avg_reward_per_step:                      0.13348389
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.67668
+    avg_avg_return:                             0.049678475
+    avg_benchmark_regret:                       0.943093
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.012779236
+    proxy_corner:
+      avg_avg_return:                           0.85398406
+      avg_reward_per_step:                      0.006919861
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.74191
+    avg_avg_return:                             0.07782903
+    avg_benchmark_regret:                       0.90624654
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.024131775
+    proxy_corner:
+      avg_avg_return:                           0.07782903
+      avg_reward_per_step:                      0.024131775
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        20971520
+    env-step-replay-before:                     10682368
+    env-step-used-before:                       20971520
+    ppo-update-before:                          12800
+    env-step-generate-before:                   10289152
+    env-step-all-after:                         21004288
+    env-step-replay-after:                      10682368
+    env-step-used-after:                        21004288
+    ppo-update-after:                           12820
+    env-step-generate-after:                    10321920
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.641609
+    avg_avg_return:                             0.92448145
+    avg_reward_per_step:                        0.13375854
+    proxy_corner:
+      avg_avg_return:                           0.92448145
+      avg_reward_per_step:                      0.13375854
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.641609
+    avg_avg_return:                             0.92448145
+    avg_reward_per_step:                        0.13375854
+    proxy_corner:
+      avg_avg_return:                           0.92448145
+      avg_reward_per_step:                      0.13375854
+  perf:
+    env_steps_per_second:                       293808.74093999714
+    ppo_updates_per_second:                     42.14101412209217
+    eval-batch-orig-duration:                   0.5327457608655095
+    eval-batch-shift-duration:                  0.488552731461823
+    eval-batch-tree-duration:                   0.48785456642508507
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.197754
+      mouse-cheese_dist_finite_wavg:            18.91954
+      mouse-cheese_dist_solvable_avg:           18.197754
+      mouse-cheese_dist_solvable_wavg:          18.919542
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13176101
+    visit_patterns:
+      avg_first_visit_time:                     136.62549
+      avg_last_visit_time:                      316.28003
+      num_replay_batches:                       326
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        -1.291342e-05
+    avg_actor_approxkl3:                        0.00011428052
+    avg_actor_clipfrac:                         0.0037353516
+    avg_actor_loss:                             -5.4956134e-05
+    avg_advantage:                              -0.025851361
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00011602874
+    avg_entropy:                                0.009031011
+    avg_grad_norm_pre_clip:                     0.21145105
+    avg_loss:                                   -5.972775e-06
+    max:
+      max_actor_approxkl1:                      0.00023816123
+      max_actor_approxkl3:                      0.0003776812
+      max_actor_clipfrac:                       0.009155273
+      max_actor_loss:                           0.00025060028
+      max_advantage:                            0.03500974
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00037948738
+      max_entropy:                              0.011704318
+      max_grad_norm_pre_clip:                   0.52958935
+      max_loss:                                 0.00026761345
+    std:
+      std_actor_approxkl1:                      0.000121755715
+      std_actor_approxkl3:                      9.952474e-05
+      std_actor_clipfrac:                       0.00265272
+      std_actor_loss:                           9.100517e-05
+      std_advantage:                            0.010352054
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          9.504853e-05
+      std_entropy:                              0.00148929
+      std_grad_norm_pre_clip:                   0.10944449
+      std_loss:                                 0.000110227666
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.042398
+    avg_avg_return:                             0.9736857
+    avg_benchmark_regret:                       0.015943814
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13350677
+    proxy_corner:
+      avg_avg_return:                           0.9736857
+      avg_reward_per_step:                      0.13350677
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.04951
+    avg_avg_return:                             0.05681371
+    avg_benchmark_regret:                       0.9359578
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.012550354
+    proxy_corner:
+      avg_avg_return:                           0.8532583
+      avg_reward_per_step:                      0.006958008
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.80986
+    avg_avg_return:                             0.072749436
+    avg_benchmark_regret:                       0.91132617
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02406311
+    proxy_corner:
+      avg_avg_return:                           0.072749436
+      avg_reward_per_step:                      0.02406311
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        22020096
+    env-step-replay-before:                     11141120
+    env-step-used-before:                       22020096
+    ppo-update-before:                          13440
+    env-step-generate-before:                   10878976
+    env-step-all-after:                         22052864
+    env-step-replay-after:                      11173888
+    env-step-used-after:                        22052864
+    ppo-update-after:                           13460
+    env-step-generate-after:                    10878976
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.451283
+    avg_avg_return:                             0.85078657
+    avg_reward_per_step:                        0.050567627
+    proxy_corner:
+      avg_avg_return:                           0.85078657
+      avg_reward_per_step:                      0.050567627
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.451283
+    avg_avg_return:                             0.85078657
+    avg_reward_per_step:                        0.050567627
+    proxy_corner:
+      avg_avg_return:                           0.85078657
+      avg_reward_per_step:                      0.050567627
+  perf:
+    env_steps_per_second:                       317789.0663674004
+    ppo_updates_per_second:                     42.29357834645101
+    eval-batch-orig-duration:                   0.5314835384488106
+    eval-batch-shift-duration:                  0.4903701301664114
+    eval-batch-tree-duration:                   0.4918368998914957
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.250977
+      mouse-cheese_dist_finite_wavg:            18.922985
+      mouse-cheese_dist_solvable_avg:           18.250977
+      mouse-cheese_dist_solvable_wavg:          18.922983
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1321807
+    visit_patterns:
+      avg_first_visit_time:                     140.90771
+      avg_last_visit_time:                      331.20483
+      num_replay_batches:                       341
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0059152306
+    avg_actor_approxkl3:                        0.0059119645
+    avg_actor_clipfrac:                         0.06467896
+    avg_actor_loss:                             0.002708471
+    avg_advantage:                              0.0026261103
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            5.2094274e-05
+    avg_entropy:                                0.07088747
+    avg_grad_norm_pre_clip:                     1.3511565
+    avg_loss:                                   0.0026636305
+    max:
+      max_actor_approxkl1:                      0.017937044
+      max_actor_approxkl3:                      0.016488854
+      max_actor_clipfrac:                       0.15698242
+      max_actor_loss:                           0.0071535213
+      max_advantage:                            0.028955936
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0001373874
+      max_entropy:                              0.111645326
+      max_grad_norm_pre_clip:                   2.659331
+      max_loss:                                 0.0070602316
+    std:
+      std_actor_approxkl1:                      0.0051799533
+      std_actor_approxkl3:                      0.004483136
+      std_actor_clipfrac:                       0.04053384
+      std_actor_loss:                           0.0018344293
+      std_advantage:                            0.0055106687
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.6033576e-05
+      std_entropy:                              0.022987619
+      std_grad_norm_pre_clip:                   0.5815236
+      std_loss:                                 0.0018213405
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041082
+    avg_avg_return:                             0.97374487
+    avg_benchmark_regret:                       0.015884664
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13350677
+    proxy_corner:
+      avg_avg_return:                           0.97374487
+      avg_reward_per_step:                      0.13350677
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.286316
+    avg_avg_return:                             0.040414684
+    avg_benchmark_regret:                       0.9523568
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.012062073
+    proxy_corner:
+      avg_avg_return:                           0.8557801
+      avg_reward_per_step:                      0.0069122314
+  eval-batch-tree:
+    avg_avg_episode_length:                     74.90925
+    avg_avg_return:                             0.16573687
+    avg_benchmark_regret:                       0.81833875
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.029067993
+    proxy_corner:
+      avg_avg_return:                           0.16573687
+      avg_reward_per_step:                      0.029067993
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        23068672
+    env-step-replay-before:                     11665408
+    env-step-used-before:                       23068672
+    ppo-update-before:                          14080
+    env-step-generate-before:                   11403264
+    env-step-all-after:                         23101440
+    env-step-replay-after:                      11698176
+    env-step-used-after:                        23101440
+    ppo-update-after:                           14100
+    env-step-generate-after:                    11403264
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.656412
+    avg_avg_return:                             0.8492528
+    avg_reward_per_step:                        0.049957275
+    proxy_corner:
+      avg_avg_return:                           0.8492528
+      avg_reward_per_step:                      0.049957275
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.656412
+    avg_avg_return:                             0.8492528
+    avg_reward_per_step:                        0.049957275
+    proxy_corner:
+      avg_avg_return:                           0.8492528
+      avg_reward_per_step:                      0.049957275
+  perf:
+    env_steps_per_second:                       243013.44990234083
+    ppo_updates_per_second:                     42.32643616554029
+    eval-batch-orig-duration:                   0.6710713971406221
+    eval-batch-shift-duration:                  0.6183969546109438
+    eval-batch-tree-duration:                   0.6013071732595563
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.313965
+      mouse-cheese_dist_finite_wavg:            18.92929
+      mouse-cheese_dist_solvable_avg:           18.313965
+      mouse-cheese_dist_solvable_wavg:          18.929293
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13268264
+    visit_patterns:
+      avg_first_visit_time:                     145.81763
+      avg_last_visit_time:                      347.24707
+      num_replay_batches:                       357
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011340663
+    avg_actor_approxkl3:                        0.0012163569
+    avg_actor_clipfrac:                         0.003503418
+    avg_actor_loss:                             5.1276435e-05
+    avg_advantage:                              0.0005997486
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            6.927114e-06
+    avg_entropy:                                0.01491255
+    avg_grad_norm_pre_clip:                     0.44899184
+    avg_loss:                                   3.9827457e-05
+    max:
+      max_actor_approxkl1:                      0.0028292593
+      max_actor_approxkl3:                      0.0025880623
+      max_actor_clipfrac:                       0.005126953
+      max_actor_loss:                           0.0022723544
+      max_advantage:                            0.0347296
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          9.62464e-06
+      max_entropy:                              0.02052832
+      max_grad_norm_pre_clip:                   1.3446499
+      max_loss:                                 0.0022606468
+    std:
+      std_actor_approxkl1:                      0.00094353216
+      std_actor_approxkl3:                      0.0008669575
+      std_actor_clipfrac:                       0.0012982568
+      std_actor_loss:                           0.0007860641
+      std_advantage:                            0.0040164664
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.4563527e-06
+      std_entropy:                              0.003722125
+      std_grad_norm_pre_clip:                   0.35780203
+      std_loss:                                 0.0007851758
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.042789
+    avg_avg_return:                             0.9737793
+    avg_benchmark_regret:                       0.015850183
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349152
+    proxy_corner:
+      avg_avg_return:                           0.9737793
+      avg_reward_per_step:                      0.13349152
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.576004
+    avg_avg_return:                             0.035312228
+    avg_benchmark_regret:                       0.95745933
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.010520935
+    proxy_corner:
+      avg_avg_return:                           0.843563
+      avg_reward_per_step:                      0.006790161
+  eval-batch-tree:
+    avg_avg_episode_length:                     73.3811
+    avg_avg_return:                             0.18654598
+    avg_benchmark_regret:                       0.7975296
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.030143738
+    proxy_corner:
+      avg_avg_return:                           0.18654598
+      avg_reward_per_step:                      0.030143738
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        24117248
+    env-step-replay-before:                     12320768
+    env-step-used-before:                       24117248
+    ppo-update-before:                          14720
+    env-step-generate-before:                   11796480
+    env-step-all-after:                         24150016
+    env-step-replay-after:                      12353536
+    env-step-used-after:                        24150016
+    ppo-update-after:                           14740
+    env-step-generate-after:                    11796480
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.482052
+    avg_avg_return:                             0.85061073
+    avg_reward_per_step:                        0.050476074
+    proxy_corner:
+      avg_avg_return:                           0.85061073
+      avg_reward_per_step:                      0.050476074
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.482052
+    avg_avg_return:                             0.85061073
+    avg_reward_per_step:                        0.050476074
+    proxy_corner:
+      avg_avg_return:                           0.85061073
+      avg_reward_per_step:                      0.050476074
+  perf:
+    env_steps_per_second:                       297408.9968129883
+    ppo_updates_per_second:                     42.225262188972174
+    eval-batch-orig-duration:                   0.5276815621182323
+    eval-batch-shift-duration:                  0.4831316079944372
+    eval-batch-tree-duration:                   0.48724567238241434
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.358887
+      mouse-cheese_dist_finite_wavg:            18.934303
+      mouse-cheese_dist_solvable_avg:           18.358887
+      mouse-cheese_dist_solvable_wavg:          18.934301
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1330197
+    visit_patterns:
+      avg_first_visit_time:                     150.00415
+      avg_last_visit_time:                      367.23242
+      num_replay_batches:                       377
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012294101
+    avg_actor_approxkl3:                        0.0012985212
+    avg_actor_clipfrac:                         0.05065918
+    avg_actor_loss:                             0.0009041258
+    avg_advantage:                              -0.002904845
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            8.416114e-06
+    avg_entropy:                                0.08788718
+    avg_grad_norm_pre_clip:                     1.9283288
+    avg_loss:                                   0.0008204467
+    max:
+      max_actor_approxkl1:                      0.0037204619
+      max_actor_approxkl3:                      0.0034629647
+      max_actor_clipfrac:                       0.091430664
+      max_actor_loss:                           0.0025086738
+      max_advantage:                            0.020449877
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.7245684e-05
+      max_entropy:                              0.09316976
+      max_grad_norm_pre_clip:                   3.8784401
+      max_loss:                                 0.0024278823
+    std:
+      std_actor_approxkl1:                      0.0011583839
+      std_actor_approxkl3:                      0.00090358924
+      std_actor_clipfrac:                       0.02341457
+      std_actor_loss:                           0.0006434327
+      std_advantage:                            0.0036937161
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.8769184e-06
+      std_entropy:                              0.0035357827
+      std_grad_norm_pre_clip:                   1.0143901
+      std_loss:                                 0.000643238
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.043111
+    avg_avg_return:                             0.97371185
+    avg_benchmark_regret:                       0.01591767
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13347626
+    proxy_corner:
+      avg_avg_return:                           0.97371185
+      avg_reward_per_step:                      0.13347626
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.56928
+    avg_avg_return:                             0.03458208
+    avg_benchmark_regret:                       0.9581895
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.011146545
+    proxy_corner:
+      avg_avg_return:                           0.87639093
+      avg_reward_per_step:                      0.00705719
+  eval-batch-tree:
+    avg_avg_episode_length:                     75.022766
+    avg_avg_return:                             0.16173266
+    avg_benchmark_regret:                       0.822343
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02848053
+    proxy_corner:
+      avg_avg_return:                           0.16173266
+      avg_reward_per_step:                      0.02848053
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        25165824
+    env-step-replay-before:                     13041664
+    env-step-used-before:                       25165824
+    ppo-update-before:                          15360
+    env-step-generate-before:                   12124160
+    env-step-all-after:                         25198592
+    env-step-replay-after:                      13041664
+    env-step-used-after:                        25198592
+    ppo-update-after:                           15380
+    env-step-generate-after:                    12156928
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.942015
+    avg_avg_return:                             0.9274956
+    avg_reward_per_step:                        0.13259888
+    proxy_corner:
+      avg_avg_return:                           0.9274956
+      avg_reward_per_step:                      0.13259888
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.942015
+    avg_avg_return:                             0.9274956
+    avg_reward_per_step:                        0.13259888
+    proxy_corner:
+      avg_avg_return:                           0.9274956
+      avg_reward_per_step:                      0.13259888
+  perf:
+    env_steps_per_second:                       324366.0470188799
+    ppo_updates_per_second:                     41.86669346779314
+    eval-batch-orig-duration:                   0.5416986299678683
+    eval-batch-shift-duration:                  0.4873342840000987
+    eval-batch-tree-duration:                   0.4819571580737829
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.395264
+      mouse-cheese_dist_finite_wavg:            18.938755
+      mouse-cheese_dist_solvable_avg:           18.395264
+      mouse-cheese_dist_solvable_wavg:          18.938755
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13329199
+    visit_patterns:
+      avg_first_visit_time:                     153.8003
+      avg_last_visit_time:                      388.29517
+      num_replay_batches:                       398
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012109344
+    avg_actor_approxkl3:                        0.0011335226
+    avg_actor_clipfrac:                         0.05974121
+    avg_actor_loss:                             0.0009570038
+    avg_advantage:                              0.00038227593
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.5540727e-05
+    avg_entropy:                                0.10396021
+    avg_grad_norm_pre_clip:                     2.4313521
+    avg_loss:                                   0.0008658141
+    max:
+      max_actor_approxkl1:                      0.005190431
+      max_actor_approxkl3:                      0.0057007624
+      max_actor_clipfrac:                       0.14355469
+      max_actor_loss:                           0.0059677893
+      max_advantage:                            0.038061023
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.459936e-05
+      max_entropy:                              0.112665206
+      max_grad_norm_pre_clip:                   5.49162
+      max_loss:                                 0.005870648
+    std:
+      std_actor_approxkl1:                      0.0014162556
+      std_actor_approxkl3:                      0.0012898428
+      std_actor_clipfrac:                       0.04924183
+      std_actor_loss:                           0.0014642872
+      std_advantage:                            0.007980166
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.9729143e-06
+      std_entropy:                              0.006097596
+      std_grad_norm_pre_clip:                   1.8068405
+      std_loss:                                 0.0014623695
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040653
+    avg_avg_return:                             0.97379076
+    avg_benchmark_regret:                       0.01583876
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349915
+    proxy_corner:
+      avg_avg_return:                           0.97379076
+      avg_reward_per_step:                      0.13349915
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.61691
+    avg_avg_return:                             0.03661963
+    avg_benchmark_regret:                       0.9561519
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.008293152
+    proxy_corner:
+      avg_avg_return:                           0.8877983
+      avg_reward_per_step:                      0.007156372
+  eval-batch-tree:
+    avg_avg_episode_length:                     75.74182
+    avg_avg_return:                             0.14816716
+    avg_benchmark_regret:                       0.8359085
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02873993
+    proxy_corner:
+      avg_avg_return:                           0.14816716
+      avg_reward_per_step:                      0.02873993
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        26214400
+    env-step-replay-before:                     13402112
+    env-step-used-before:                       26214400
+    ppo-update-before:                          16000
+    env-step-generate-before:                   12812288
+    env-step-all-after:                         26247168
+    env-step-replay-after:                      13434880
+    env-step-used-after:                        26247168
+    ppo-update-after:                           16020
+    env-step-generate-after:                    12812288
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.697435
+    avg_avg_return:                             0.848948
+    avg_reward_per_step:                        0.049835205
+    proxy_corner:
+      avg_avg_return:                           0.848948
+      avg_reward_per_step:                      0.049835205
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.697435
+    avg_avg_return:                             0.848948
+    avg_reward_per_step:                        0.049835205
+    proxy_corner:
+      avg_avg_return:                           0.848948
+      avg_reward_per_step:                      0.049835205
+  perf:
+    env_steps_per_second:                       310273.3802755485
+    ppo_updates_per_second:                     42.15703495542884
+    eval-batch-orig-duration:                   0.5192011930048466
+    eval-batch-shift-duration:                  0.4850870929658413
+    eval-batch-tree-duration:                   0.47833576053380966
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.461182
+      mouse-cheese_dist_finite_wavg:            19.841383
+      mouse-cheese_dist_solvable_avg:           18.461182
+      mouse-cheese_dist_solvable_wavg:          19.841385
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13378888
+    visit_patterns:
+      avg_first_visit_time:                     161.9773
+      avg_last_visit_time:                      400.26514
+      num_replay_batches:                       410
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0027353254
+    avg_actor_approxkl3:                        0.002846093
+    avg_actor_clipfrac:                         0.04807129
+    avg_actor_loss:                             0.0012004434
+    avg_advantage:                              0.00069456664
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.7893724e-05
+    avg_entropy:                                0.09248125
+    avg_grad_norm_pre_clip:                     1.5082784
+    avg_loss:                                   0.001116909
+    max:
+      max_actor_approxkl1:                      0.006599904
+      max_actor_approxkl3:                      0.0060775466
+      max_actor_clipfrac:                       0.12207031
+      max_actor_loss:                           0.0070732865
+      max_advantage:                            0.026895642
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          4.4833694e-05
+      max_entropy:                              0.10798258
+      max_grad_norm_pre_clip:                   4.5365324
+      max_loss:                                 0.006971675
+    std:
+      std_actor_approxkl1:                      0.0018575159
+      std_actor_approxkl3:                      0.0017986016
+      std_actor_clipfrac:                       0.031053944
+      std_actor_loss:                           0.0019380082
+      std_advantage:                            0.004362553
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          9.6192125e-06
+      std_entropy:                              0.007034952
+      std_grad_norm_pre_clip:                   1.2200037
+      std_loss:                                 0.0019333833
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.04348
+    avg_avg_return:                             0.97351843
+    avg_benchmark_regret:                       0.016111072
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13347626
+    proxy_corner:
+      avg_avg_return:                           0.97351843
+      avg_reward_per_step:                      0.13347626
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.582596
+    avg_avg_return:                             0.049989793
+    avg_benchmark_regret:                       0.94278175
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.011039734
+    proxy_corner:
+      avg_avg_return:                           0.89402425
+      avg_reward_per_step:                      0.007286072
+  eval-batch-tree:
+    avg_avg_episode_length:                     78.33265
+    avg_avg_return:                             0.10588452
+    avg_benchmark_regret:                       0.8781911
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02519226
+    proxy_corner:
+      avg_avg_return:                           0.10588452
+      avg_reward_per_step:                      0.02519226
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        27262976
+    env-step-replay-before:                     14024704
+    env-step-used-before:                       27262976
+    ppo-update-before:                          16640
+    env-step-generate-before:                   13238272
+    env-step-all-after:                         27295744
+    env-step-replay-after:                      14057472
+    env-step-used-after:                        27295744
+    ppo-update-after:                           16660
+    env-step-generate-after:                    13238272
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.707691
+    avg_avg_return:                             0.84890294
+    avg_reward_per_step:                        0.049804688
+    proxy_corner:
+      avg_avg_return:                           0.84890294
+      avg_reward_per_step:                      0.049804688
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.707691
+    avg_avg_return:                             0.84890294
+    avg_reward_per_step:                        0.049804688
+    proxy_corner:
+      avg_avg_return:                           0.84890294
+      avg_reward_per_step:                      0.049804688
+  perf:
+    env_steps_per_second:                       255691.1339058199
+    ppo_updates_per_second:                     42.01349597629137
+    eval-batch-orig-duration:                   0.5461920527741313
+    eval-batch-shift-duration:                  0.48972746916115284
+    eval-batch-tree-duration:                   0.4916125535964966
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.51123
+      mouse-cheese_dist_finite_wavg:            18.949917
+      mouse-cheese_dist_solvable_avg:           18.51123
+      mouse-cheese_dist_solvable_wavg:          18.949919
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13416389
+    visit_patterns:
+      avg_first_visit_time:                     168.08789
+      avg_last_visit_time:                      419.26123
+      num_replay_batches:                       429
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0008110756
+    avg_actor_approxkl3:                        0.0008834465
+    avg_actor_clipfrac:                         0.05168457
+    avg_actor_loss:                             0.00089547643
+    avg_advantage:                              -0.002069259
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            3.809683e-06
+    avg_entropy:                                0.09045421
+    avg_grad_norm_pre_clip:                     3.0996263
+    avg_loss:                                   0.0008069269
+    max:
+      max_actor_approxkl1:                      0.0023316352
+      max_actor_approxkl3:                      0.0023955698
+      max_actor_clipfrac:                       0.107910156
+      max_actor_loss:                           0.005666582
+      max_advantage:                            0.028811097
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          6.48842e-06
+      max_entropy:                              0.09751813
+      max_grad_norm_pre_clip:                   8.094991
+      max_loss:                                 0.0055729873
+    std:
+      std_actor_approxkl1:                      0.00072078314
+      std_actor_approxkl3:                      0.0006160025
+      std_actor_clipfrac:                       0.038356625
+      std_actor_loss:                           0.001570803
+      std_advantage:                            0.0023122332
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.0615602e-06
+      std_entropy:                              0.004272352
+      std_grad_norm_pre_clip:                   2.383772
+      std_loss:                                 0.001570043
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.043331
+    avg_avg_return:                             0.9735642
+    avg_benchmark_regret:                       0.016065344
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349152
+    proxy_corner:
+      avg_avg_return:                           0.9735642
+      avg_reward_per_step:                      0.13349152
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.54518
+    avg_avg_return:                             0.038992107
+    avg_benchmark_regret:                       0.9537794
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.013442993
+    proxy_corner:
+      avg_avg_return:                           0.8906704
+      avg_reward_per_step:                      0.0071792603
+  eval-batch-tree:
+    avg_avg_episode_length:                     75.774155
+    avg_avg_return:                             0.1495023
+    avg_benchmark_regret:                       0.8345733
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.028442383
+    proxy_corner:
+      avg_avg_return:                           0.1495023
+      avg_reward_per_step:                      0.028442383
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        28311552
+    env-step-replay-before:                     14581760
+    env-step-used-before:                       28311552
+    ppo-update-before:                          17280
+    env-step-generate-before:                   13729792
+    env-step-all-after:                         28344320
+    env-step-replay-after:                      14581760
+    env-step-used-after:                        28344320
+    ppo-update-after:                           17300
+    env-step-generate-after:                    13762560
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.136717
+    avg_avg_return:                             0.92965865
+    avg_reward_per_step:                        0.13272095
+    proxy_corner:
+      avg_avg_return:                           0.92965865
+      avg_reward_per_step:                      0.13272095
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.136717
+    avg_avg_return:                             0.92965865
+    avg_reward_per_step:                        0.13272095
+    proxy_corner:
+      avg_avg_return:                           0.92965865
+      avg_reward_per_step:                      0.13272095
+  perf:
+    env_steps_per_second:                       318486.8970000787
+    ppo_updates_per_second:                     42.263204077032356
+    eval-batch-orig-duration:                   0.53505003079772
+    eval-batch-shift-duration:                  0.48244002368301153
+    eval-batch-tree-duration:                   0.47217434644699097
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.57373
+      mouse-cheese_dist_finite_wavg:            18.957634
+      mouse-cheese_dist_solvable_avg:           18.57373
+      mouse-cheese_dist_solvable_wavg:          18.957642
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999964
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1346271
+    visit_patterns:
+      avg_first_visit_time:                     176.41284
+      avg_last_visit_time:                      435.29028
+      num_replay_batches:                       445
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0007601821
+    avg_actor_approxkl3:                        0.000669524
+    avg_actor_clipfrac:                         0.025665283
+    avg_actor_loss:                             0.00033575593
+    avg_advantage:                              0.0039559724
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.4985255e-05
+    avg_entropy:                                0.10264119
+    avg_grad_norm_pre_clip:                     0.7247879
+    avg_loss:                                   0.00024560737
+    max:
+      max_actor_approxkl1:                      0.0031947554
+      max_actor_approxkl3:                      0.0042121476
+      max_actor_clipfrac:                       0.14379883
+      max_actor_loss:                           0.0022639083
+      max_advantage:                            0.030300438
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          4.352058e-05
+      max_entropy:                              0.11608016
+      max_grad_norm_pre_clip:                   2.1514215
+      max_loss:                                 0.0021803707
+    std:
+      std_actor_approxkl1:                      0.001043553
+      std_actor_approxkl3:                      0.0011096069
+      std_actor_clipfrac:                       0.0431769
+      std_actor_loss:                           0.00069585274
+      std_advantage:                            0.005793078
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.2394573e-06
+      std_entropy:                              0.008391453
+      std_grad_norm_pre_clip:                   0.52278835
+      std_loss:                                 0.00069797586
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041131
+    avg_avg_return:                             0.9739078
+    avg_benchmark_regret:                       0.015721668
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13346863
+    proxy_corner:
+      avg_avg_return:                           0.9739078
+      avg_reward_per_step:                      0.13346863
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.74895
+    avg_avg_return:                             0.03384005
+    avg_benchmark_regret:                       0.95893145
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0105896
+    proxy_corner:
+      avg_avg_return:                           0.8884467
+      avg_reward_per_step:                      0.0071792603
+  eval-batch-tree:
+    avg_avg_episode_length:                     75.30701
+    avg_avg_return:                             0.15668087
+    avg_benchmark_regret:                       0.8273947
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.028411865
+    proxy_corner:
+      avg_avg_return:                           0.15668087
+      avg_reward_per_step:                      0.028411865
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        29360128
+    env-step-replay-before:                     15171584
+    env-step-used-before:                       29360128
+    ppo-update-before:                          17920
+    env-step-generate-before:                   14188544
+    env-step-all-after:                         29392896
+    env-step-replay-after:                      15171584
+    env-step-used-after:                        29392896
+    ppo-update-after:                           17940
+    env-step-generate-after:                    14221312
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.210329
+    avg_avg_return:                             0.924079
+    avg_reward_per_step:                        0.12844849
+    proxy_corner:
+      avg_avg_return:                           0.924079
+      avg_reward_per_step:                      0.12844849
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.210329
+    avg_avg_return:                             0.924079
+    avg_reward_per_step:                        0.12844849
+    proxy_corner:
+      avg_avg_return:                           0.924079
+      avg_reward_per_step:                      0.12844849
+  perf:
+    env_steps_per_second:                       307994.38746768824
+    ppo_updates_per_second:                     42.29299212427985
+    eval-batch-orig-duration:                   0.527878156863153
+    eval-batch-shift-duration:                  0.4864146504551172
+    eval-batch-tree-duration:                   0.48965595476329327
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.623047
+      mouse-cheese_dist_finite_wavg:            18.962175
+      mouse-cheese_dist_solvable_avg:           18.623047
+      mouse-cheese_dist_solvable_wavg:          18.962177
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13499391
+    visit_patterns:
+      avg_first_visit_time:                     183.84814
+      avg_last_visit_time:                      453.31738
+      num_replay_batches:                       463
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0019374135
+    avg_actor_approxkl3:                        0.002306117
+    avg_actor_clipfrac:                         0.034985352
+    avg_actor_loss:                             -0.00022712183
+    avg_advantage:                              0.014213383
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            7.4111456e-05
+    avg_entropy:                                0.09858698
+    avg_grad_norm_pre_clip:                     1.0408098
+    avg_loss:                                   -0.00028865304
+    max:
+      max_actor_approxkl1:                      0.0042884555
+      max_actor_approxkl3:                      0.004237158
+      max_actor_clipfrac:                       0.06225586
+      max_actor_loss:                           0.002048131
+      max_advantage:                            0.056621313
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.000174265
+      max_entropy:                              0.11090175
+      max_grad_norm_pre_clip:                   2.3090062
+      max_loss:                                 0.001990887
+    std:
+      std_actor_approxkl1:                      0.0012521598
+      std_actor_approxkl3:                      0.0012100722
+      std_actor_clipfrac:                       0.015615241
+      std_actor_loss:                           0.0007853631
+      std_advantage:                            0.012499476
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.7877966e-05
+      std_entropy:                              0.007971946
+      std_grad_norm_pre_clip:                   0.5465601
+      std_loss:                                 0.00078633515
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.066011
+    avg_avg_return:                             0.97346985
+    avg_benchmark_regret:                       0.016159624
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.1331482
+    proxy_corner:
+      avg_avg_return:                           0.97346985
+      avg_reward_per_step:                      0.1331482
+  eval-batch-shift:
+    avg_avg_episode_length:                     83.30194
+    avg_avg_return:                             0.028221386
+    avg_benchmark_regret:                       0.96455014
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.009933472
+    proxy_corner:
+      avg_avg_return:                           0.8163914
+      avg_reward_per_step:                      0.0065689087
+  eval-batch-tree:
+    avg_avg_episode_length:                     70.726875
+    avg_avg_return:                             0.21874686
+    avg_benchmark_regret:                       0.76532876
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.033470154
+    proxy_corner:
+      avg_avg_return:                           0.21874686
+      avg_reward_per_step:                      0.033470154
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        30408704
+    env-step-replay-before:                     15761408
+    env-step-used-before:                       30408704
+    ppo-update-before:                          18560
+    env-step-generate-before:                   14647296
+    env-step-all-after:                         30441472
+    env-step-replay-after:                      15794176
+    env-step-used-after:                        30441472
+    ppo-update-after:                           18580
+    env-step-generate-after:                    14647296
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.98944
+    avg_avg_return:                             0.84762466
+    avg_reward_per_step:                        0.048980713
+    proxy_corner:
+      avg_avg_return:                           0.84762466
+      avg_reward_per_step:                      0.048980713
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.98944
+    avg_avg_return:                             0.84762466
+    avg_reward_per_step:                        0.048980713
+    proxy_corner:
+      avg_avg_return:                           0.84762466
+      avg_reward_per_step:                      0.048980713
+  perf:
+    env_steps_per_second:                       291628.89767159865
+    ppo_updates_per_second:                     42.17363422959584
+    eval-batch-orig-duration:                   0.5199756370857358
+    eval-batch-shift-duration:                  0.4795578243210912
+    eval-batch-tree-duration:                   0.486543501727283
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.650879
+      mouse-cheese_dist_finite_wavg:            18.962652
+      mouse-cheese_dist_solvable_avg:           18.650879
+      mouse-cheese_dist_solvable_wavg:          18.962654
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1350781
+    visit_patterns:
+      avg_first_visit_time:                     196.47607
+      avg_last_visit_time:                      472.3108
+      num_replay_batches:                       482
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0021959664
+    avg_actor_approxkl3:                        0.0018515876
+    avg_actor_clipfrac:                         0.05112305
+    avg_actor_loss:                             0.00024638075
+    avg_advantage:                              4.007143e-05
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.8832821e-05
+    avg_entropy:                                0.12761658
+    avg_grad_norm_pre_clip:                     2.3034897
+    avg_loss:                                   0.00013318054
+    max:
+      max_actor_approxkl1:                      0.0055379597
+      max_actor_approxkl3:                      0.0051466487
+      max_actor_clipfrac:                       0.1307373
+      max_actor_loss:                           0.0018528542
+      max_advantage:                            0.039993286
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.257105e-05
+      max_entropy:                              0.13798067
+      max_grad_norm_pre_clip:                   6.100448
+      max_loss:                                 0.0017337915
+    std:
+      std_actor_approxkl1:                      0.0012942086
+      std_actor_approxkl3:                      0.0011809288
+      std_actor_clipfrac:                       0.030243954
+      std_actor_loss:                           0.00078396767
+      std_advantage:                            0.00769712
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.7338147e-06
+      std_entropy:                              0.0053889127
+      std_grad_norm_pre_clip:                   1.4240277
+      std_loss:                                 0.000782967
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.051142
+    avg_avg_return:                             0.9733087
+    avg_benchmark_regret:                       0.016320772
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13331604
+    proxy_corner:
+      avg_avg_return:                           0.9733087
+      avg_reward_per_step:                      0.13331604
+  eval-batch-shift:
+    avg_avg_episode_length:                     83.51046
+    avg_avg_return:                             0.02535754
+    avg_benchmark_regret:                       0.96741396
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.010299683
+    proxy_corner:
+      avg_avg_return:                           0.7256081
+      avg_reward_per_step:                      0.005859375
+  eval-batch-tree:
+    avg_avg_episode_length:                     70.077446
+    avg_avg_return:                             0.2222264
+    avg_benchmark_regret:                       0.7618492
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.035469055
+    proxy_corner:
+      avg_avg_return:                           0.2222264
+      avg_reward_per_step:                      0.035469055
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        31457280
+    env-step-replay-before:                     16384000
+    env-step-used-before:                       31457280
+    ppo-update-before:                          19200
+    env-step-generate-before:                   15073280
+    env-step-all-after:                         31490048
+    env-step-replay-after:                      16384000
+    env-step-used-after:                        31490048
+    ppo-update-after:                           19220
+    env-step-generate-after:                    15106048
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.220476
+    avg_avg_return:                             0.92535615
+    avg_reward_per_step:                        0.14572144
+    proxy_corner:
+      avg_avg_return:                           0.92535615
+      avg_reward_per_step:                      0.14572144
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.220476
+    avg_avg_return:                             0.92535615
+    avg_reward_per_step:                        0.14572144
+    proxy_corner:
+      avg_avg_return:                           0.92535615
+      avg_reward_per_step:                      0.14572144
+  perf:
+    env_steps_per_second:                       282011.2040945696
+    ppo_updates_per_second:                     42.27661595907413
+    eval-batch-orig-duration:                   0.5182789918035269
+    eval-batch-shift-duration:                  0.47243175748735666
+    eval-batch-tree-duration:                   0.4499817630276084
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.70337
+      mouse-cheese_dist_finite_wavg:            18.969376
+      mouse-cheese_dist_solvable_avg:           18.70337
+      mouse-cheese_dist_solvable_wavg:          18.969378
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13526511
+    visit_patterns:
+      avg_first_visit_time:                     203.76074
+      avg_last_visit_time:                      490.44434
+      num_replay_batches:                       500
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0015361159
+    avg_actor_approxkl3:                        0.0013412758
+    avg_actor_clipfrac:                         0.042785645
+    avg_actor_loss:                             0.00023084901
+    avg_advantage:                              0.0008035095
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.1955766e-05
+    avg_entropy:                                0.09316432
+    avg_grad_norm_pre_clip:                     1.2068276
+    avg_loss:                                   0.00014866257
+    max:
+      max_actor_approxkl1:                      0.0035260948
+      max_actor_approxkl3:                      0.0032431162
+      max_actor_clipfrac:                       0.09753418
+      max_actor_loss:                           0.0023262482
+      max_advantage:                            0.038056135
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.8658884e-05
+      max_entropy:                              0.10726006
+      max_grad_norm_pre_clip:                   2.4558253
+      max_loss:                                 0.0022429307
+    std:
+      std_actor_approxkl1:                      0.0010333176
+      std_actor_approxkl3:                      0.0008394872
+      std_actor_clipfrac:                       0.027353544
+      std_actor_loss:                           0.00073235435
+      std_advantage:                            0.006737499
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.6425616e-06
+      std_entropy:                              0.010121453
+      std_grad_norm_pre_clip:                   0.6456112
+      std_loss:                                 0.0007351903
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.060428
+    avg_avg_return:                             0.9732319
+    avg_benchmark_regret:                       0.016397625
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13323975
+    proxy_corner:
+      avg_avg_return:                           0.9732319
+      avg_reward_per_step:                      0.13323975
+  eval-batch-shift:
+    avg_avg_episode_length:                     83.47347
+    avg_avg_return:                             0.027743515
+    avg_benchmark_regret:                       0.965028
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.00907135
+    proxy_corner:
+      avg_avg_return:                           0.6424416
+      avg_reward_per_step:                      0.0052261353
+  eval-batch-tree:
+    avg_avg_episode_length:                     69.74445
+    avg_avg_return:                             0.22470069
+    avg_benchmark_regret:                       0.759375
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.03453064
+    proxy_corner:
+      avg_avg_return:                           0.22470069
+      avg_reward_per_step:                      0.03453064
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        32505856
+    env-step-replay-before:                     16842752
+    env-step-used-before:                       32505856
+    ppo-update-before:                          19840
+    env-step-generate-before:                   15663104
+    env-step-all-after:                         32538624
+    env-step-replay-after:                      16875520
+    env-step-used-after:                        32538624
+    ppo-update-after:                           19860
+    env-step-generate-after:                    15663104
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.882914
+    avg_avg_return:                             0.84923
+    avg_reward_per_step:                        0.049346924
+    proxy_corner:
+      avg_avg_return:                           0.84923
+      avg_reward_per_step:                      0.049346924
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.882914
+    avg_avg_return:                             0.84923
+    avg_reward_per_step:                        0.049346924
+    proxy_corner:
+      avg_avg_return:                           0.84923
+      avg_reward_per_step:                      0.049346924
+  perf:
+    env_steps_per_second:                       316886.2523485059
+    ppo_updates_per_second:                     41.255605896679164
+    eval-batch-orig-duration:                   0.5310201570391655
+    eval-batch-shift-duration:                  0.4856617096811533
+    eval-batch-tree-duration:                   0.4546196758747101
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.7583
+      mouse-cheese_dist_finite_wavg:            18.97285
+      mouse-cheese_dist_solvable_avg:           18.7583
+      mouse-cheese_dist_solvable_wavg:          18.97285
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13534576
+    visit_patterns:
+      avg_first_visit_time:                     215.87036
+      avg_last_visit_time:                      505.2378
+      num_replay_batches:                       515
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0019238631
+    avg_actor_approxkl3:                        0.0021035688
+    avg_actor_clipfrac:                         0.07111817
+    avg_actor_loss:                             0.0021953103
+    avg_advantage:                              -0.0005634271
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.5431599e-05
+    avg_entropy:                                0.12267911
+    avg_grad_norm_pre_clip:                     2.752561
+    avg_loss:                                   0.0020803472
+    max:
+      max_actor_approxkl1:                      0.004700764
+      max_actor_approxkl3:                      0.004540909
+      max_actor_clipfrac:                       0.13330078
+      max_actor_loss:                           0.005362672
+      max_advantage:                            0.023352265
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.93308e-05
+      max_entropy:                              0.12738657
+      max_grad_norm_pre_clip:                   6.1853266
+      max_loss:                                 0.0052473065
+    std:
+      std_actor_approxkl1:                      0.0012082913
+      std_actor_approxkl3:                      0.0010243715
+      std_actor_clipfrac:                       0.030794727
+      std_actor_loss:                           0.0016980524
+      std_advantage:                            0.0055135083
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.151616e-06
+      std_entropy:                              0.0024843542
+      std_grad_norm_pre_clip:                   1.3896025
+      std_loss:                                 0.0016985332
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.072437
+    avg_avg_return:                             0.97367483
+    avg_benchmark_regret:                       0.015954666
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.1332779
+    proxy_corner:
+      avg_avg_return:                           0.97367483
+      avg_reward_per_step:                      0.1332779
+  eval-batch-shift:
+    avg_avg_episode_length:                     83.29873
+    avg_avg_return:                             0.028319698
+    avg_benchmark_regret:                       0.9644518
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.010749817
+    proxy_corner:
+      avg_avg_return:                           0.6156764
+      avg_reward_per_step:                      0.005012512
+  eval-batch-tree:
+    avg_avg_episode_length:                     70.95145
+    avg_avg_return:                             0.21399125
+    avg_benchmark_regret:                       0.7700844
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.032310486
+    proxy_corner:
+      avg_avg_return:                           0.21399125
+      avg_reward_per_step:                      0.032310486
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        33554432
+    env-step-replay-before:                     17399808
+    env-step-used-before:                       33554432
+    ppo-update-before:                          20480
+    env-step-generate-before:                   16154624
+    env-step-all-after:                         33587200
+    env-step-replay-after:                      17399808
+    env-step-used-after:                        33587200
+    ppo-update-after:                           20500
+    env-step-generate-after:                    16187392
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.234543
+    avg_avg_return:                             0.9266223
+    avg_reward_per_step:                        0.1331482
+    proxy_corner:
+      avg_avg_return:                           0.9266223
+      avg_reward_per_step:                      0.1331482
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.234543
+    avg_avg_return:                             0.9266223
+    avg_reward_per_step:                        0.1331482
+    proxy_corner:
+      avg_avg_return:                           0.9266223
+      avg_reward_per_step:                      0.1331482
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  perf:
+    env_steps_per_second:                       349507.23704807466
+    ppo_updates_per_second:                     41.30491035570804
+    eval-batch-orig-duration:                   0.6382062556222081
+    eval-batch-shift-duration:                  0.575269821099937
+    eval-batch-tree-duration:                   0.5955205308273435
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.811523
+      mouse-cheese_dist_finite_wavg:            18.980598
+      mouse-cheese_dist_solvable_avg:           18.811523
+      mouse-cheese_dist_solvable_wavg:          18.980598
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13618931
+    visit_patterns:
+      avg_first_visit_time:                     230.43237
+      avg_last_visit_time:                      521.29565
+      num_replay_batches:                       531
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00094445184
+    avg_actor_approxkl3:                        0.00064176176
+    avg_actor_clipfrac:                         0.034100343
+    avg_actor_loss:                             7.0657525e-06
+    avg_advantage:                              -0.0025688317
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            3.2582215e-05
+    avg_entropy:                                0.11568707
+    avg_grad_norm_pre_clip:                     1.4848284
+    avg_loss:                                   -9.233023e-05
+    max:
+      max_actor_approxkl1:                      0.0033284384
+      max_actor_approxkl3:                      0.0023192181
+      max_actor_clipfrac:                       0.12768555
+      max_actor_loss:                           0.001724429
+      max_advantage:                            0.036234677
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          5.983698e-05
+      max_entropy:                              0.14405842
+      max_grad_norm_pre_clip:                   5.951214
+      max_loss:                                 0.0016243091
+    std:
+      std_actor_approxkl1:                      0.00079799275
+      std_actor_approxkl3:                      0.00057209
+      std_actor_clipfrac:                       0.036806703
+      std_actor_loss:                           0.00055925356
+      std_advantage:                            0.006771035
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.0832207e-05
+      std_entropy:                              0.010572413
+      std_grad_norm_pre_clip:                   1.4284998
+      std_loss:                                 0.0005616162
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.047394
+    avg_avg_return:                             0.97370183
+    avg_benchmark_regret:                       0.01592766
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.133461
+    proxy_corner:
+      avg_avg_return:                           0.97370183
+      avg_reward_per_step:                      0.133461
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-shift:
+    avg_avg_episode_length:                     83.48078
+    avg_avg_return:                             0.02502232
+    avg_benchmark_regret:                       0.96774924
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.010429382
+    proxy_corner:
+      avg_avg_return:                           0.69999075
+      avg_reward_per_step:                      0.0056381226
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-tree:
+    avg_avg_episode_length:                     70.54466
+    avg_avg_return:                             0.21628341
+    avg_benchmark_regret:                       0.7677922
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.032104492
+    proxy_corner:
+      avg_avg_return:                           0.21628341
+      avg_reward_per_step:                      0.032104492
+    rollouts_gif:                               float32(132, 225, 225, 3)
+===========================================================
+ 17%|▏| 33.6M/200M [11:10<39:15, 70.7k env steps/s]
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        34603008
+    env-step-replay-before:                     17694720
+    env-step-used-before:                       34603008
+    ppo-update-before:                          21120
+    env-step-generate-before:                   16908288
+    env-step-all-after:                         34635776
+    env-step-replay-after:                      17727488
+    env-step-used-after:                        34635776
+    ppo-update-after:                           21140
+    env-step-generate-after:                    16908288
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.11252
+    avg_avg_return:                             0.84675986
+    avg_reward_per_step:                        0.048614502
+    proxy_corner:
+      avg_avg_return:                           0.84675986
+      avg_reward_per_step:                      0.048614502
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.11252
+    avg_avg_return:                             0.84675986
+    avg_reward_per_step:                        0.048614502
+    proxy_corner:
+      avg_avg_return:                           0.84675986
+      avg_reward_per_step:                      0.048614502
+  perf:
+    env_steps_per_second:                       347582.3062011252
+    ppo_updates_per_second:                     40.58359376293051
+    eval-batch-orig-duration:                   0.5159283252432942
+    eval-batch-shift-duration:                  0.4828109499067068
+    eval-batch-tree-duration:                   0.47174943145364523
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.882568
+      mouse-cheese_dist_finite_wavg:            18.984291
+      mouse-cheese_dist_solvable_avg:           18.882568
+      mouse-cheese_dist_solvable_wavg:          18.984291
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13645191
+    visit_patterns:
+      avg_first_visit_time:                     241.81763
+      avg_last_visit_time:                      531.30396
+      num_replay_batches:                       541
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002728379
+    avg_actor_approxkl3:                        0.002376901
+    avg_actor_clipfrac:                         0.046868898
+    avg_actor_loss:                             0.0044354047
+    avg_advantage:                              -0.0048837997
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            7.216404e-05
+    avg_entropy:                                0.08979091
+    avg_grad_norm_pre_clip:                     4.2486405
+    avg_loss:                                   0.0043816953
+    max:
+      max_actor_approxkl1:                      0.0076979175
+      max_actor_approxkl3:                      0.005733275
+      max_actor_clipfrac:                       0.07775879
+      max_actor_loss:                           0.015892494
+      max_advantage:                            0.021573305
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00017535318
+      max_entropy:                              0.0982923
+      max_grad_norm_pre_clip:                   7.072675
+      max_loss:                                 0.015808893
+    std:
+      std_actor_approxkl1:                      0.0020091722
+      std_actor_approxkl3:                      0.0017422179
+      std_actor_clipfrac:                       0.025349906
+      std_actor_loss:                           0.004649545
+      std_advantage:                            0.006200966
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.6664096e-05
+      std_entropy:                              0.0043206355
+      std_grad_norm_pre_clip:                   2.703594
+      std_loss:                                 0.004641676
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.043104
+    avg_avg_return:                             0.9736682
+    avg_benchmark_regret:                       0.015961282
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13348389
+    proxy_corner:
+      avg_avg_return:                           0.9736682
+      avg_reward_per_step:                      0.13348389
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.82416
+    avg_avg_return:                             0.032905176
+    avg_benchmark_regret:                       0.9598664
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.011749268
+    proxy_corner:
+      avg_avg_return:                           0.77804774
+      avg_reward_per_step:                      0.0063095093
+  eval-batch-tree:
+    avg_avg_episode_length:                     71.10493
+    avg_avg_return:                             0.21566364
+    avg_benchmark_regret:                       0.768412
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.030441284
+    proxy_corner:
+      avg_avg_return:                           0.21566364
+      avg_reward_per_step:                      0.030441284
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        35651584
+    env-step-replay-before:                     18153472
+    env-step-used-before:                       35651584
+    ppo-update-before:                          21760
+    env-step-generate-before:                   17498112
+    env-step-all-after:                         35684352
+    env-step-replay-after:                      18186240
+    env-step-used-after:                        35684352
+    ppo-update-after:                           21780
+    env-step-generate-after:                    17498112
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.122776
+    avg_avg_return:                             0.84659624
+    avg_reward_per_step:                        0.048583984
+    proxy_corner:
+      avg_avg_return:                           0.84659624
+      avg_reward_per_step:                      0.048583984
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.122776
+    avg_avg_return:                             0.84659624
+    avg_reward_per_step:                        0.048583984
+    proxy_corner:
+      avg_avg_return:                           0.84659624
+      avg_reward_per_step:                      0.048583984
+  perf:
+    env_steps_per_second:                       314638.12450003414
+    ppo_updates_per_second:                     42.27737676185689
+    eval-batch-orig-duration:                   0.5394474631175399
+    eval-batch-shift-duration:                  0.489709765650332
+    eval-batch-tree-duration:                   0.48964987229555845
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.95752
+      mouse-cheese_dist_finite_wavg:            18.991692
+      mouse-cheese_dist_solvable_avg:           18.95752
+      mouse-cheese_dist_solvable_wavg:          18.991693
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13693143
+    visit_patterns:
+      avg_first_visit_time:                     253.4209
+      avg_last_visit_time:                      545.3635
+      num_replay_batches:                       555
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002368667
+    avg_actor_approxkl3:                        0.0023513192
+    avg_actor_clipfrac:                         0.04524536
+    avg_actor_loss:                             -0.00022182408
+    avg_advantage:                              0.0007948142
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.7571692e-05
+    avg_entropy:                                0.051107418
+    avg_grad_norm_pre_clip:                     0.3272493
+    avg_loss:                                   -0.00025914566
+    max:
+      max_actor_approxkl1:                      0.004985319
+      max_actor_approxkl3:                      0.0046488633
+      max_actor_clipfrac:                       0.0871582
+      max_actor_loss:                           0.00016481709
+      max_advantage:                            0.033266008
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.505894e-05
+      max_entropy:                              0.056134224
+      max_grad_norm_pre_clip:                   0.64437133
+      max_loss:                                 0.00013137393
+    std:
+      std_actor_approxkl1:                      0.0015021435
+      std_actor_approxkl3:                      0.0014998548
+      std_actor_clipfrac:                       0.029515715
+      std_actor_loss:                           0.00016566992
+      std_advantage:                            0.00818783
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.3119917e-06
+      std_entropy:                              0.0036187961
+      std_grad_norm_pre_clip:                   0.120577276
+      std_loss:                                 0.00016779824
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.042984
+    avg_avg_return:                             0.97361684
+    avg_benchmark_regret:                       0.016012674
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349152
+    proxy_corner:
+      avg_avg_return:                           0.97361684
+      avg_reward_per_step:                      0.13349152
+  eval-batch-shift:
+    avg_avg_episode_length:                     83.03879
+    avg_avg_return:                             0.031734284
+    avg_benchmark_regret:                       0.9610372
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0121536255
+    proxy_corner:
+      avg_avg_return:                           0.64894277
+      avg_reward_per_step:                      0.0052490234
+  eval-batch-tree:
+    avg_avg_episode_length:                     72.63236
+    avg_avg_return:                             0.19392624
+    avg_benchmark_regret:                       0.79014933
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.028434753
+    proxy_corner:
+      avg_avg_return:                           0.19392624
+      avg_reward_per_step:                      0.028434753
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        36700160
+    env-step-replay-before:                     18677760
+    env-step-used-before:                       36700160
+    ppo-update-before:                          22400
+    env-step-generate-before:                   18022400
+    env-step-all-after:                         36732928
+    env-step-replay-after:                      18677760
+    env-step-used-after:                        36732928
+    ppo-update-after:                           22420
+    env-step-generate-after:                    18055168
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.877354
+    avg_avg_return:                             0.9273713
+    avg_reward_per_step:                        0.13919067
+    proxy_corner:
+      avg_avg_return:                           0.9273713
+      avg_reward_per_step:                      0.13919067
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.877354
+    avg_avg_return:                             0.9273713
+    avg_reward_per_step:                        0.13919067
+    proxy_corner:
+      avg_avg_return:                           0.9273713
+      avg_reward_per_step:                      0.13919067
+  perf:
+    env_steps_per_second:                       293639.2431080551
+    ppo_updates_per_second:                     42.2515712888764
+    eval-batch-orig-duration:                   0.5151443965733051
+    eval-batch-shift-duration:                  0.47993138153105974
+    eval-batch-tree-duration:                   0.48822372779250145
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.021484
+      mouse-cheese_dist_finite_wavg:            19.000269
+      mouse-cheese_dist_solvable_avg:           19.021484
+      mouse-cheese_dist_solvable_wavg:          19.000273
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13765082
+    visit_patterns:
+      avg_first_visit_time:                     262.26318
+      avg_last_visit_time:                      560.32935
+      num_replay_batches:                       570
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0003528805
+    avg_actor_approxkl3:                        0.0006022724
+    avg_actor_clipfrac:                         0.02260132
+    avg_actor_loss:                             0.0005319204
+    avg_advantage:                              -0.013356207
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            6.430083e-05
+    avg_entropy:                                0.07190297
+    avg_grad_norm_pre_clip:                     1.4018593
+    avg_loss:                                   0.00049216783
+    max:
+      max_actor_approxkl1:                      0.0013604099
+      max_actor_approxkl3:                      0.0021268344
+      max_actor_clipfrac:                       0.0592041
+      max_actor_loss:                           0.002666574
+      max_advantage:                            0.030537128
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00018384858
+      max_entropy:                              0.08435884
+      max_grad_norm_pre_clip:                   3.1545634
+      max_loss:                                 0.0026315635
+    std:
+      std_actor_approxkl1:                      0.00039498575
+      std_actor_approxkl3:                      0.0004980095
+      std_actor_clipfrac:                       0.015993599
+      std_actor_loss:                           0.0008211201
+      std_advantage:                            0.0074289814
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.9676896e-05
+      std_entropy:                              0.0066867983
+      std_grad_norm_pre_clip:                   0.9374267
+      std_loss:                                 0.0008199178
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.043079
+    avg_avg_return:                             0.97352326
+    avg_benchmark_regret:                       0.016106242
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.133461
+    proxy_corner:
+      avg_avg_return:                           0.97352326
+      avg_reward_per_step:                      0.133461
+  eval-batch-shift:
+    avg_avg_episode_length:                     83.130455
+    avg_avg_return:                             0.030690359
+    avg_benchmark_regret:                       0.9620812
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.012908936
+    proxy_corner:
+      avg_avg_return:                           0.7188157
+      avg_reward_per_step:                      0.0058059692
+  eval-batch-tree:
+    avg_avg_episode_length:                     70.2807
+    avg_avg_return:                             0.221005
+    avg_benchmark_regret:                       0.7630706
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.028953552
+    proxy_corner:
+      avg_avg_return:                           0.221005
+      avg_reward_per_step:                      0.028953552
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        37748736
+    env-step-replay-before:                     19267584
+    env-step-used-before:                       37748736
+    ppo-update-before:                          23040
+    env-step-generate-before:                   18481152
+    env-step-all-after:                         37781504
+    env-step-replay-after:                      19267584
+    env-step-used-after:                        37781504
+    ppo-update-after:                           23060
+    env-step-generate-after:                    18513920
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.887942
+    avg_avg_return:                             0.9289421
+    avg_reward_per_step:                        0.14068604
+    proxy_corner:
+      avg_avg_return:                           0.9289421
+      avg_reward_per_step:                      0.14068604
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.887942
+    avg_avg_return:                             0.9289421
+    avg_reward_per_step:                        0.14068604
+    proxy_corner:
+      avg_avg_return:                           0.9289421
+      avg_reward_per_step:                      0.14068604
+  perf:
+    env_steps_per_second:                       290020.92110550776
+    ppo_updates_per_second:                     42.17325432284009
+    eval-batch-orig-duration:                   0.5419192919507623
+    eval-batch-shift-duration:                  0.48798430152237415
+    eval-batch-tree-duration:                   0.48650832287967205
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.072998
+      mouse-cheese_dist_finite_wavg:            19.006264
+      mouse-cheese_dist_solvable_avg:           19.072998
+      mouse-cheese_dist_solvable_wavg:          19.006266
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13813634
+    visit_patterns:
+      avg_first_visit_time:                     271.3147
+      avg_last_visit_time:                      578.2434
+      num_replay_batches:                       588
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00078689645
+    avg_actor_approxkl3:                        0.00094598497
+    avg_actor_clipfrac:                         0.02822876
+    avg_actor_loss:                             0.0014012154
+    avg_advantage:                              0.0011179461
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.6159352e-05
+    avg_entropy:                                0.058621462
+    avg_grad_norm_pre_clip:                     1.7001332
+    avg_loss:                                   0.0013556737
+    max:
+      max_actor_approxkl1:                      0.003371902
+      max_actor_approxkl3:                      0.0024092705
+      max_actor_clipfrac:                       0.060791016
+      max_actor_loss:                           0.0069679487
+      max_advantage:                            0.042823374
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.4266755e-05
+      max_entropy:                              0.07026039
+      max_grad_norm_pre_clip:                   4.5412683
+      max_loss:                                 0.0069155777
+    std:
+      std_actor_approxkl1:                      0.00091937254
+      std_actor_approxkl3:                      0.00057330716
+      std_actor_clipfrac:                       0.01426348
+      std_actor_loss:                           0.001775106
+      std_advantage:                            0.0072852266
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.5223934e-06
+      std_entropy:                              0.00706987
+      std_grad_norm_pre_clip:                   1.3134311
+      std_loss:                                 0.0017722141
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.047213
+    avg_avg_return:                             0.9733345
+    avg_benchmark_regret:                       0.016295016
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13341522
+    proxy_corner:
+      avg_avg_return:                           0.9733345
+      avg_reward_per_step:                      0.13341522
+  eval-batch-shift:
+    avg_avg_episode_length:                     83.11349
+    avg_avg_return:                             0.031020341
+    avg_benchmark_regret:                       0.96175116
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.010818481
+    proxy_corner:
+      avg_avg_return:                           0.7240617
+      avg_reward_per_step:                      0.005874634
+  eval-batch-tree:
+    avg_avg_episode_length:                     70.20655
+    avg_avg_return:                             0.21773264
+    avg_benchmark_regret:                       0.766343
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.029678345
+    proxy_corner:
+      avg_avg_return:                           0.21773264
+      avg_reward_per_step:                      0.029678345
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        38797312
+    env-step-replay-before:                     19922944
+    env-step-used-before:                       38797312
+    ppo-update-before:                          23680
+    env-step-generate-before:                   18874368
+    env-step-all-after:                         38830080
+    env-step-replay-after:                      19922944
+    env-step-used-after:                        38830080
+    ppo-update-after:                           23700
+    env-step-generate-after:                    18907136
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.190737
+    avg_avg_return:                             0.926345
+    avg_reward_per_step:                        0.14788818
+    proxy_corner:
+      avg_avg_return:                           0.926345
+      avg_reward_per_step:                      0.14788818
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.190737
+    avg_avg_return:                             0.926345
+    avg_reward_per_step:                        0.14788818
+    proxy_corner:
+      avg_avg_return:                           0.926345
+      avg_reward_per_step:                      0.14788818
+  perf:
+    env_steps_per_second:                       349883.8708774708
+    ppo_updates_per_second:                     40.49687614619054
+    eval-batch-orig-duration:                   0.5248655136674643
+    eval-batch-shift-duration:                  0.47985495813190937
+    eval-batch-tree-duration:                   0.4590910505503416
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.108887
+      mouse-cheese_dist_finite_wavg:            19.008934
+      mouse-cheese_dist_solvable_avg:           19.108887
+      mouse-cheese_dist_solvable_wavg:          19.008938
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13856584
+    visit_patterns:
+      avg_first_visit_time:                     284.71582
+      avg_last_visit_time:                      598.3396
+      num_replay_batches:                       608
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010972387
+    avg_actor_approxkl3:                        0.0007731265
+    avg_actor_clipfrac:                         0.010101318
+    avg_actor_loss:                             1.1836877e-06
+    avg_advantage:                              0.00024231631
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.4116248e-05
+    avg_entropy:                                0.038734615
+    avg_grad_norm_pre_clip:                     0.50267667
+    avg_loss:                                   -3.0492809e-05
+    max:
+      max_actor_approxkl1:                      0.0045277635
+      max_actor_approxkl3:                      0.0035147779
+      max_actor_clipfrac:                       0.036621094
+      max_actor_loss:                           0.00065174233
+      max_advantage:                            0.037939727
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.9613402e-05
+      max_entropy:                              0.046940483
+      max_grad_norm_pre_clip:                   1.0323417
+      max_loss:                                 0.0006226269
+    std:
+      std_actor_approxkl1:                      0.0011052613
+      std_actor_approxkl3:                      0.000803881
+      std_actor_clipfrac:                       0.0071416744
+      std_actor_loss:                           0.00040306867
+      std_advantage:                            0.005554579
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.9461345e-06
+      std_entropy:                              0.0043535517
+      std_grad_norm_pre_clip:                   0.22617717
+      std_loss:                                 0.0004049719
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.08316
+    avg_avg_return:                             0.97360694
+    avg_benchmark_regret:                       0.016022611
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.1332779
+    proxy_corner:
+      avg_avg_return:                           0.97360694
+      avg_reward_per_step:                      0.1332779
+  eval-batch-shift:
+    avg_avg_episode_length:                     83.142586
+    avg_avg_return:                             0.031526994
+    avg_benchmark_regret:                       0.9612445
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.010398865
+    proxy_corner:
+      avg_avg_return:                           0.6654074
+      avg_reward_per_step:                      0.0054092407
+  eval-batch-tree:
+    avg_avg_episode_length:                     70.52286
+    avg_avg_return:                             0.21523193
+    avg_benchmark_regret:                       0.76884365
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.030708313
+    proxy_corner:
+      avg_avg_return:                           0.21523193
+      avg_reward_per_step:                      0.030708313
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        39845888
+    env-step-replay-before:                     20381696
+    env-step-used-before:                       39845888
+    ppo-update-before:                          24320
+    env-step-generate-before:                   19464192
+    env-step-all-after:                         39878656
+    env-step-replay-after:                      20414464
+    env-step-used-after:                        39878656
+    ppo-update-after:                           24340
+    env-step-generate-after:                    19464192
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.479336
+    avg_avg_return:                             0.84354115
+    avg_reward_per_step:                        0.04751587
+    proxy_corner:
+      avg_avg_return:                           0.84354115
+      avg_reward_per_step:                      0.04751587
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.479336
+    avg_avg_return:                             0.84354115
+    avg_reward_per_step:                        0.04751587
+    proxy_corner:
+      avg_avg_return:                           0.84354115
+      avg_reward_per_step:                      0.04751587
+  perf:
+    env_steps_per_second:                       323541.05771438224
+    ppo_updates_per_second:                     44.59804942325976
+    eval-batch-orig-duration:                   0.5153286634013057
+    eval-batch-shift-duration:                  0.4862845269963145
+    eval-batch-tree-duration:                   0.4712440827861428
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.163818
+      mouse-cheese_dist_finite_wavg:            19.014532
+      mouse-cheese_dist_solvable_avg:           19.163818
+      mouse-cheese_dist_solvable_wavg:          19.014532
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1387075
+    visit_patterns:
+      avg_first_visit_time:                     293.36304
+      avg_last_visit_time:                      613.3469
+      num_replay_batches:                       623
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00072354614
+    avg_actor_approxkl3:                        0.0008629185
+    avg_actor_clipfrac:                         0.027697755
+    avg_actor_loss:                             0.0007818701
+    avg_advantage:                              -0.0007599713
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.0451901e-05
+    avg_entropy:                                0.047648005
+    avg_grad_norm_pre_clip:                     1.4443659
+    avg_loss:                                   0.000739448
+    max:
+      max_actor_approxkl1:                      0.002390927
+      max_actor_approxkl3:                      0.002169487
+      max_actor_clipfrac:                       0.04748535
+      max_actor_loss:                           0.004813535
+      max_advantage:                            0.037934363
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.6694636e-05
+      max_entropy:                              0.055434786
+      max_grad_norm_pre_clip:                   3.0860631
+      max_loss:                                 0.004772399
+    std:
+      std_actor_approxkl1:                      0.0008180731
+      std_actor_approxkl3:                      0.0006913515
+      std_actor_clipfrac:                       0.01802218
+      std_actor_loss:                           0.0012893552
+      std_advantage:                            0.005173594
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.6690855e-06
+      std_entropy:                              0.0030349346
+      std_grad_norm_pre_clip:                   0.7411117
+      std_loss:                                 0.0012905474
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.044805
+    avg_avg_return:                             0.9734734
+    avg_benchmark_regret:                       0.01615606
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13346863
+    proxy_corner:
+      avg_avg_return:                           0.9734734
+      avg_reward_per_step:                      0.13346863
+  eval-batch-shift:
+    avg_avg_episode_length:                     83.55742
+    avg_avg_return:                             0.027469618
+    avg_benchmark_regret:                       0.9653019
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.009750366
+    proxy_corner:
+      avg_avg_return:                           0.748729
+      avg_reward_per_step:                      0.0060577393
+  eval-batch-tree:
+    avg_avg_episode_length:                     69.79989
+    avg_avg_return:                             0.2244774
+    avg_benchmark_regret:                       0.75959826
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.030212402
+    proxy_corner:
+      avg_avg_return:                           0.2244774
+      avg_reward_per_step:                      0.030212402
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        40894464
+    env-step-replay-before:                     21037056
+    env-step-used-before:                       40894464
+    ppo-update-before:                          24960
+    env-step-generate-before:                   19857408
+    env-step-all-after:                         40927232
+    env-step-replay-after:                      21069824
+    env-step-used-after:                        40927232
+    ppo-update-after:                           24980
+    env-step-generate-after:                    19857408
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.471495
+    avg_avg_return:                             0.8440317
+    avg_reward_per_step:                        0.047546387
+    proxy_corner:
+      avg_avg_return:                           0.8440317
+      avg_reward_per_step:                      0.047546387
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.471495
+    avg_avg_return:                             0.8440317
+    avg_reward_per_step:                        0.047546387
+    proxy_corner:
+      avg_avg_return:                           0.8440317
+      avg_reward_per_step:                      0.047546387
+  perf:
+    env_steps_per_second:                       258468.08903687797
+    ppo_updates_per_second:                     41.98399308925895
+    eval-batch-orig-duration:                   0.5431217048317194
+    eval-batch-shift-duration:                  0.47494071908295155
+    eval-batch-tree-duration:                   0.489393824711442
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.208984
+      mouse-cheese_dist_finite_wavg:            19.02052
+      mouse-cheese_dist_solvable_avg:           19.208984
+      mouse-cheese_dist_solvable_wavg:          19.02052
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.138942
+    visit_patterns:
+      avg_first_visit_time:                     302.0083
+      avg_last_visit_time:                      633.3508
+      num_replay_batches:                       643
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0007054554
+    avg_actor_approxkl3:                        0.00077921426
+    avg_actor_clipfrac:                         0.022198487
+    avg_actor_loss:                             -0.00023634863
+    avg_advantage:                              -0.0019333747
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            6.6244115e-06
+    avg_entropy:                                0.029544542
+    avg_grad_norm_pre_clip:                     0.164331
+    avg_loss:                                   -0.00026258096
+    max:
+      max_actor_approxkl1:                      0.0019094581
+      max_actor_approxkl3:                      0.0012522093
+      max_actor_clipfrac:                       0.044311523
+      max_actor_loss:                           0.00026295148
+      max_advantage:                            0.034933865
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          9.0380745e-06
+      max_entropy:                              0.036473274
+      max_grad_norm_pre_clip:                   0.5110603
+      max_loss:                                 0.00023999595
+    std:
+      std_actor_approxkl1:                      0.0005363426
+      std_actor_approxkl3:                      0.00037367153
+      std_actor_clipfrac:                       0.0141510125
+      std_actor_loss:                           0.00023688097
+      std_advantage:                            0.0036418813
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.046903e-06
+      std_entropy:                              0.002829327
+      std_grad_norm_pre_clip:                   0.13812688
+      std_loss:                                 0.00023720243
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041555
+    avg_avg_return:                             0.973684
+    avg_benchmark_regret:                       0.015945515
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349152
+    proxy_corner:
+      avg_avg_return:                           0.973684
+      avg_reward_per_step:                      0.13349152
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.535
+    avg_avg_return:                             0.038716055
+    avg_benchmark_regret:                       0.9540554
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.010856628
+    proxy_corner:
+      avg_avg_return:                           0.64253986
+      avg_reward_per_step:                      0.0052490234
+  eval-batch-tree:
+    avg_avg_episode_length:                     72.295334
+    avg_avg_return:                             0.19922692
+    avg_benchmark_regret:                       0.7848487
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.030227661
+    proxy_corner:
+      avg_avg_return:                           0.19922692
+      avg_reward_per_step:                      0.030227661
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        41943040
+    env-step-replay-before:                     21528576
+    env-step-used-before:                       41943040
+    ppo-update-before:                          25600
+    env-step-generate-before:                   20414464
+    env-step-all-after:                         41975808
+    env-step-replay-after:                      21528576
+    env-step-used-after:                        41975808
+    ppo-update-after:                           25620
+    env-step-generate-after:                    20447232
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.162091
+    avg_avg_return:                             0.9240365
+    avg_reward_per_step:                        0.1381836
+    proxy_corner:
+      avg_avg_return:                           0.9240365
+      avg_reward_per_step:                      0.1381836
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.162091
+    avg_avg_return:                             0.9240365
+    avg_reward_per_step:                        0.1381836
+    proxy_corner:
+      avg_avg_return:                           0.9240365
+      avg_reward_per_step:                      0.1381836
+  perf:
+    env_steps_per_second:                       240341.0517727431
+    ppo_updates_per_second:                     41.85851865778881
+    eval-batch-orig-duration:                   0.5329099362716079
+    eval-batch-shift-duration:                  0.48008311819285154
+    eval-batch-tree-duration:                   0.45040817838162184
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.2583
+      mouse-cheese_dist_finite_wavg:            19.024832
+      mouse-cheese_dist_solvable_avg:           19.2583
+      mouse-cheese_dist_solvable_wavg:          19.024832
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13953921
+    visit_patterns:
+      avg_first_visit_time:                     312.11743
+      avg_last_visit_time:                      647.2012
+      num_replay_batches:                       657
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0015678926
+    avg_actor_approxkl3:                        0.0011934208
+    avg_actor_clipfrac:                         0.01994629
+    avg_actor_loss:                             7.943774e-06
+    avg_advantage:                              0.0024974358
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            3.3059983e-05
+    avg_entropy:                                0.048149787
+    avg_grad_norm_pre_clip:                     0.51355296
+    avg_loss:                                   -2.3676013e-05
+    max:
+      max_actor_approxkl1:                      0.005151589
+      max_actor_approxkl3:                      0.005150276
+      max_actor_clipfrac:                       0.045288086
+      max_actor_loss:                           0.0011999877
+      max_advantage:                            0.052823007
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          4.895677e-05
+      max_entropy:                              0.054657422
+      max_grad_norm_pre_clip:                   1.1007073
+      max_loss:                                 0.001162086
+    std:
+      std_actor_approxkl1:                      0.0013686202
+      std_actor_approxkl3:                      0.0012504437
+      std_actor_clipfrac:                       0.016156428
+      std_actor_loss:                           0.0005545741
+      std_advantage:                            0.009631876
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.302812e-06
+      std_entropy:                              0.005232148
+      std_grad_norm_pre_clip:                   0.3079821
+      std_loss:                                 0.0005552364
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.0825405
+    avg_avg_return:                             0.9737129
+    avg_benchmark_regret:                       0.015916545
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13275146
+    proxy_corner:
+      avg_avg_return:                           0.9737129
+      avg_reward_per_step:                      0.13275146
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.05072
+    avg_avg_return:                             0.04570255
+    avg_benchmark_regret:                       0.9470689
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.011474609
+    proxy_corner:
+      avg_avg_return:                           0.53858143
+      avg_reward_per_step:                      0.004463196
+  eval-batch-tree:
+    avg_avg_episode_length:                     71.031845
+    avg_avg_return:                             0.21711321
+    avg_benchmark_regret:                       0.7669624
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.030349731
+    proxy_corner:
+      avg_avg_return:                           0.21711321
+      avg_reward_per_step:                      0.030349731
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        42991616
+    env-step-replay-before:                     21954560
+    env-step-used-before:                       42991616
+    ppo-update-before:                          26240
+    env-step-generate-before:                   21037056
+    env-step-all-after:                         43024384
+    env-step-replay-after:                      21987328
+    env-step-used-after:                        43024384
+    ppo-update-after:                           26260
+    env-step-generate-after:                    21037056
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.620514
+    avg_avg_return:                             0.8421103
+    avg_reward_per_step:                        0.047088623
+    proxy_corner:
+      avg_avg_return:                           0.8421103
+      avg_reward_per_step:                      0.047088623
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.620514
+    avg_avg_return:                             0.8421103
+    avg_reward_per_step:                        0.047088623
+    proxy_corner:
+      avg_avg_return:                           0.8421103
+      avg_reward_per_step:                      0.047088623
+  perf:
+    env_steps_per_second:                       335892.6684661581
+    ppo_updates_per_second:                     41.85079863283752
+    eval-batch-orig-duration:                   0.5227886186912656
+    eval-batch-shift-duration:                  0.4932781821116805
+    eval-batch-tree-duration:                   0.4809527425095439
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.282715
+      mouse-cheese_dist_finite_wavg:            19.027283
+      mouse-cheese_dist_solvable_avg:           19.282715
+      mouse-cheese_dist_solvable_wavg:          19.027283
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.139949
+    visit_patterns:
+      avg_first_visit_time:                     321.10718
+      avg_last_visit_time:                      661.2068
+      num_replay_batches:                       671
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.011203276
+    avg_actor_approxkl3:                        0.009766036
+    avg_actor_clipfrac:                         0.0732544
+    avg_actor_loss:                             0.0031212938
+    avg_advantage:                              0.00020375477
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.0113584e-05
+    avg_entropy:                                0.087877765
+    avg_grad_norm_pre_clip:                     1.9735352
+    avg_loss:                                   0.0030384725
+    max:
+      max_actor_approxkl1:                      0.039031595
+      max_actor_approxkl3:                      0.037538018
+      max_actor_clipfrac:                       0.12634277
+      max_actor_loss:                           0.015433788
+      max_advantage:                            0.039645255
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.2138413e-05
+      max_entropy:                              0.09784022
+      max_grad_norm_pre_clip:                   3.5764377
+      max_loss:                                 0.015349524
+    std:
+      std_actor_approxkl1:                      0.011910803
+      std_actor_approxkl3:                      0.0108902175
+      std_actor_clipfrac:                       0.042008933
+      std_actor_loss:                           0.0044418448
+      std_advantage:                            0.0028932148
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.160952e-06
+      std_entropy:                              0.007252964
+      std_grad_norm_pre_clip:                   0.9819826
+      std_loss:                                 0.004440656
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.043261
+    avg_avg_return:                             0.9736412
+    avg_benchmark_regret:                       0.015988315
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.133461
+    proxy_corner:
+      avg_avg_return:                           0.9736412
+      avg_reward_per_step:                      0.133461
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.88062
+    avg_avg_return:                             0.0357567
+    avg_benchmark_regret:                       0.9570148
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.011749268
+    proxy_corner:
+      avg_avg_return:                           0.7284401
+      avg_reward_per_step:                      0.0058898926
+  eval-batch-tree:
+    avg_avg_episode_length:                     69.56882
+    avg_avg_return:                             0.22471449
+    avg_benchmark_regret:                       0.75936115
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.030982971
+    proxy_corner:
+      avg_avg_return:                           0.22471449
+      avg_reward_per_step:                      0.030982971
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        44040192
+    env-step-replay-before:                     22511616
+    env-step-used-before:                       44040192
+    ppo-update-before:                          26880
+    env-step-generate-before:                   21528576
+    env-step-all-after:                         44072960
+    env-step-replay-after:                      22511616
+    env-step-used-after:                        44072960
+    ppo-update-after:                           26900
+    env-step-generate-after:                    21561344
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.4193325
+    avg_avg_return:                             0.92316115
+    avg_reward_per_step:                        0.13632202
+    proxy_corner:
+      avg_avg_return:                           0.92316115
+      avg_reward_per_step:                      0.13632202
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.4193325
+    avg_avg_return:                             0.92316115
+    avg_reward_per_step:                        0.13632202
+    proxy_corner:
+      avg_avg_return:                           0.92316115
+      avg_reward_per_step:                      0.13632202
+  perf:
+    env_steps_per_second:                       315930.79343398166
+    ppo_updates_per_second:                     42.12766490283967
+    eval-batch-orig-duration:                   0.5354393450543284
+    eval-batch-shift-duration:                  0.48907792288810015
+    eval-batch-tree-duration:                   0.4837683318182826
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.287842
+      mouse-cheese_dist_finite_wavg:            19.030275
+      mouse-cheese_dist_solvable_avg:           19.287842
+      mouse-cheese_dist_solvable_wavg:          19.030277
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14003435
+    visit_patterns:
+      avg_first_visit_time:                     326.88892
+      avg_last_visit_time:                      677.2803
+      num_replay_batches:                       687
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00021373597
+    avg_actor_approxkl3:                        0.0002416609
+    avg_actor_clipfrac:                         0.0098022465
+    avg_actor_loss:                             -2.7796637e-05
+    avg_advantage:                              0.00038744803
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.8324537e-05
+    avg_entropy:                                0.06359176
+    avg_grad_norm_pre_clip:                     0.35655424
+    avg_loss:                                   -8.222614e-05
+    max:
+      max_actor_approxkl1:                      0.0008393405
+      max_actor_approxkl3:                      0.00073837815
+      max_actor_clipfrac:                       0.026855469
+      max_actor_loss:                           0.0004386548
+      max_advantage:                            0.04407245
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.1482296e-05
+      max_entropy:                              0.06966104
+      max_grad_norm_pre_clip:                   0.8517285
+      max_loss:                                 0.00039403245
+    std:
+      std_actor_approxkl1:                      0.00028137938
+      std_actor_approxkl3:                      0.00019395127
+      std_actor_clipfrac:                       0.008121585
+      std_actor_loss:                           0.00018180332
+      std_advantage:                            0.005921252
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.8394896e-06
+      std_entropy:                              0.0045641223
+      std_grad_norm_pre_clip:                   0.21959367
+      std_loss:                                 0.00018347678
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.042641
+    avg_avg_return:                             0.9736905
+    avg_benchmark_regret:                       0.01593898
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13344574
+    proxy_corner:
+      avg_avg_return:                           0.9736905
+      avg_reward_per_step:                      0.13344574
+  eval-batch-shift:
+    avg_avg_episode_length:                     83.04066
+    avg_avg_return:                             0.03113472
+    avg_benchmark_regret:                       0.9616368
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.010749817
+    proxy_corner:
+      avg_avg_return:                           0.7796081
+      avg_reward_per_step:                      0.0062713623
+  eval-batch-tree:
+    avg_avg_episode_length:                     69.3492
+    avg_avg_return:                             0.22948383
+    avg_benchmark_regret:                       0.75459176
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.032096863
+    proxy_corner:
+      avg_avg_return:                           0.22948383
+      avg_reward_per_step:                      0.032096863
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        45088768
+    env-step-replay-before:                     22937600
+    env-step-used-before:                       45088768
+    ppo-update-before:                          27520
+    env-step-generate-before:                   22151168
+    env-step-all-after:                         45121536
+    env-step-replay-after:                      22937600
+    env-step-used-after:                        45121536
+    ppo-update-after:                           27540
+    env-step-generate-after:                    22183936
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.529357
+    avg_avg_return:                             0.92479706
+    avg_reward_per_step:                        0.12911987
+    proxy_corner:
+      avg_avg_return:                           0.92479706
+      avg_reward_per_step:                      0.12911987
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.529357
+    avg_avg_return:                             0.92479706
+    avg_reward_per_step:                        0.12911987
+    proxy_corner:
+      avg_avg_return:                           0.92479706
+      avg_reward_per_step:                      0.12911987
+  perf:
+    env_steps_per_second:                       318952.67081085604
+    ppo_updates_per_second:                     40.59224880315324
+    eval-batch-orig-duration:                   0.5180508457124233
+    eval-batch-shift-duration:                  0.48036426678299904
+    eval-batch-tree-duration:                   0.4597249561920762
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.281006
+      mouse-cheese_dist_finite_wavg:            19.028574
+      mouse-cheese_dist_solvable_avg:           19.281006
+      mouse-cheese_dist_solvable_wavg:          19.028574
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14012751
+    visit_patterns:
+      avg_first_visit_time:                     333.16064
+      avg_last_visit_time:                      690.3169
+      num_replay_batches:                       700
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011434808
+    avg_actor_approxkl3:                        0.0010113245
+    avg_actor_clipfrac:                         0.047003176
+    avg_actor_loss:                             0.000744764
+    avg_advantage:                              -0.00549021
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            7.7681354e-05
+    avg_entropy:                                0.09849007
+    avg_grad_norm_pre_clip:                     1.1285778
+    avg_loss:                                   0.0006851147
+    max:
+      max_actor_approxkl1:                      0.0041115936
+      max_actor_approxkl3:                      0.0039597563
+      max_actor_clipfrac:                       0.11376953
+      max_actor_loss:                           0.0038380325
+      max_advantage:                            0.04185468
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00023666253
+      max_entropy:                              0.107402556
+      max_grad_norm_pre_clip:                   2.1754982
+      max_loss:                                 0.0037719845
+    std:
+      std_actor_approxkl1:                      0.0012085033
+      std_actor_approxkl3:                      0.0009676914
+      std_actor_clipfrac:                       0.036562912
+      std_actor_loss:                           0.0012173551
+      std_advantage:                            0.006421493
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.727921e-05
+      std_entropy:                              0.0055395113
+      std_grad_norm_pre_clip:                   0.7541508
+      std_loss:                                 0.0012105404
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.045498
+    avg_avg_return:                             0.97353464
+    avg_benchmark_regret:                       0.016094878
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13343048
+    proxy_corner:
+      avg_avg_return:                           0.97353464
+      avg_reward_per_step:                      0.13343048
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.88767
+    avg_avg_return:                             0.035057977
+    avg_benchmark_regret:                       0.95771354
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.008926392
+    proxy_corner:
+      avg_avg_return:                           0.8555447
+      avg_reward_per_step:                      0.006942749
+  eval-batch-tree:
+    avg_avg_episode_length:                     71.97919
+    avg_avg_return:                             0.20872444
+    avg_benchmark_regret:                       0.77535117
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.028121948
+    proxy_corner:
+      avg_avg_return:                           0.20872444
+      avg_reward_per_step:                      0.028121948
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        46137344
+    env-step-replay-before:                     23560192
+    env-step-used-before:                       46137344
+    ppo-update-before:                          28160
+    env-step-generate-before:                   22577152
+    env-step-all-after:                         46170112
+    env-step-replay-after:                      23560192
+    env-step-used-after:                        46170112
+    ppo-update-after:                           28180
+    env-step-generate-after:                    22609920
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.9234295
+    avg_avg_return:                             0.9269475
+    avg_reward_per_step:                        0.13949585
+    proxy_corner:
+      avg_avg_return:                           0.9269475
+      avg_reward_per_step:                      0.13949585
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.9234295
+    avg_avg_return:                             0.9269475
+    avg_reward_per_step:                        0.13949585
+    proxy_corner:
+      avg_avg_return:                           0.9269475
+      avg_reward_per_step:                      0.13949585
+  perf:
+    env_steps_per_second:                       309335.0988577977
+    ppo_updates_per_second:                     42.159779308257804
+    eval-batch-orig-duration:                   0.5411720303818583
+    eval-batch-shift-duration:                  0.49335472844541073
+    eval-batch-tree-duration:                   0.4850796917453408
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.275146
+      mouse-cheese_dist_finite_wavg:            19.02733
+      mouse-cheese_dist_solvable_avg:           19.275146
+      mouse-cheese_dist_solvable_wavg:          19.02733
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1402004
+    visit_patterns:
+      avg_first_visit_time:                     338.50708
+      avg_last_visit_time:                      709.1675
+      num_replay_batches:                       719
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010301914
+    avg_actor_approxkl3:                        0.0010287715
+    avg_actor_clipfrac:                         0.013586426
+    avg_actor_loss:                             0.0008257924
+    avg_advantage:                              0.0014838574
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            3.5319004e-05
+    avg_entropy:                                0.03560638
+    avg_grad_norm_pre_clip:                     0.76195693
+    avg_loss:                                   0.0008078455
+    max:
+      max_actor_approxkl1:                      0.0049983175
+      max_actor_approxkl3:                      0.0043413453
+      max_actor_clipfrac:                       0.037841797
+      max_actor_loss:                           0.0025137886
+      max_advantage:                            0.040738463
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          5.1526848e-05
+      max_entropy:                              0.043677334
+      max_grad_norm_pre_clip:                   1.5442901
+      max_loss:                                 0.0025070708
+    std:
+      std_actor_approxkl1:                      0.0012079492
+      std_actor_approxkl3:                      0.0010499732
+      std_actor_clipfrac:                       0.008303931
+      std_actor_loss:                           0.00090918405
+      std_advantage:                            0.008216132
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.3184806e-06
+      std_entropy:                              0.004739113
+      std_grad_norm_pre_clip:                   0.41960752
+      std_loss:                                 0.0009106637
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041382
+    avg_avg_return:                             0.9736536
+    avg_benchmark_regret:                       0.015975934
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13347626
+    proxy_corner:
+      avg_avg_return:                           0.9736536
+      avg_reward_per_step:                      0.13347626
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.652664
+    avg_avg_return:                             0.038072776
+    avg_benchmark_regret:                       0.95469874
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0129470825
+    proxy_corner:
+      avg_avg_return:                           0.7449588
+      avg_reward_per_step:                      0.006011963
+  eval-batch-tree:
+    avg_avg_episode_length:                     72.6373
+    avg_avg_return:                             0.18849236
+    avg_benchmark_regret:                       0.79558325
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.029266357
+    proxy_corner:
+      avg_avg_return:                           0.18849236
+      avg_reward_per_step:                      0.029266357
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        47185920
+    env-step-replay-before:                     24084480
+    env-step-used-before:                       47185920
+    ppo-update-before:                          28800
+    env-step-generate-before:                   23101440
+    env-step-all-after:                         47218688
+    env-step-replay-after:                      24084480
+    env-step-used-after:                        47218688
+    ppo-update-after:                           28820
+    env-step-generate-after:                    23134208
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.562578
+    avg_avg_return:                             0.9325499
+    avg_reward_per_step:                        0.13763428
+    proxy_corner:
+      avg_avg_return:                           0.9325499
+      avg_reward_per_step:                      0.13763428
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.562578
+    avg_avg_return:                             0.9325499
+    avg_reward_per_step:                        0.13763428
+    proxy_corner:
+      avg_avg_return:                           0.9325499
+      avg_reward_per_step:                      0.13763428
+  perf:
+    env_steps_per_second:                       256998.3678601871
+    ppo_updates_per_second:                     41.620788686169846
+    eval-batch-orig-duration:                   0.5430874302983284
+    eval-batch-shift-duration:                  0.49132414162158966
+    eval-batch-tree-duration:                   0.48866361379623413
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.263428
+      mouse-cheese_dist_finite_wavg:            19.026907
+      mouse-cheese_dist_solvable_avg:           19.263428
+      mouse-cheese_dist_solvable_wavg:          19.02691
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14025341
+    visit_patterns:
+      avg_first_visit_time:                     345.13696
+      avg_last_visit_time:                      725.3452
+      num_replay_batches:                       735
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        2.926392e-05
+    avg_actor_approxkl3:                        3.7085556e-05
+    avg_actor_clipfrac:                         0.0015319824
+    avg_actor_loss:                             -2.9019802e-05
+    avg_advantage:                              0.0009127277
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.0352812e-05
+    avg_entropy:                                0.010528968
+    avg_grad_norm_pre_clip:                     0.123695314
+    avg_loss:                                   -2.937236e-05
+    max:
+      max_actor_approxkl1:                      0.0002289257
+      max_actor_approxkl3:                      0.000101692145
+      max_actor_clipfrac:                       0.00390625
+      max_actor_loss:                           0.00012774672
+      max_advantage:                            0.040070474
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.7946844e-05
+      max_entropy:                              0.012405299
+      max_grad_norm_pre_clip:                   0.27516425
+      max_loss:                                 0.00013146519
+    std:
+      std_actor_approxkl1:                      7.81583e-05
+      std_actor_approxkl3:                      2.8009503e-05
+      std_actor_clipfrac:                       0.0010742708
+      std_actor_loss:                           4.9468712e-05
+      std_advantage:                            0.007316754
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.5164778e-06
+      std_entropy:                              0.0011686889
+      std_grad_norm_pre_clip:                   0.06631669
+      std_loss:                                 5.0156068e-05
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040646
+    avg_avg_return:                             0.9738435
+    avg_benchmark_regret:                       0.01578603
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13350677
+    proxy_corner:
+      avg_avg_return:                           0.9738435
+      avg_reward_per_step:                      0.13350677
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.9832
+    avg_avg_return:                             0.04324972
+    avg_benchmark_regret:                       0.9495218
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.013282776
+    proxy_corner:
+      avg_avg_return:                           0.6420794
+      avg_reward_per_step:                      0.0052108765
+  eval-batch-tree:
+    avg_avg_episode_length:                     70.68584
+    avg_avg_return:                             0.21861956
+    avg_benchmark_regret:                       0.7654561
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.027542114
+    proxy_corner:
+      avg_avg_return:                           0.21861956
+      avg_reward_per_step:                      0.027542114
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        48234496
+    env-step-replay-before:                     24707072
+    env-step-used-before:                       48234496
+    ppo-update-before:                          29440
+    env-step-generate-before:                   23527424
+    env-step-all-after:                         48267264
+    env-step-replay-after:                      24739840
+    env-step-used-after:                        48267264
+    ppo-update-after:                           29460
+    env-step-generate-after:                    23527424
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8415978
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8415978
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8415978
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8415978
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       315888.77981843986
+    ppo_updates_per_second:                     42.17932470506759
+    eval-batch-orig-duration:                   0.5199449472129345
+    eval-batch-shift-duration:                  0.4863064354285598
+    eval-batch-tree-duration:                   0.48371265549212694
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.254395
+      mouse-cheese_dist_finite_wavg:            19.025986
+      mouse-cheese_dist_solvable_avg:           19.254395
+      mouse-cheese_dist_solvable_wavg:          19.025986
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14026779
+    visit_patterns:
+      avg_first_visit_time:                     350.18335
+      avg_last_visit_time:                      745.35474
+      num_replay_batches:                       755
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0004659984
+    avg_actor_approxkl3:                        0.0004890806
+    avg_actor_clipfrac:                         0.015045166
+    avg_actor_loss:                             0.0011006481
+    avg_advantage:                              -0.00084018754
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.0732884e-05
+    avg_entropy:                                0.03287475
+    avg_grad_norm_pre_clip:                     1.9472039
+    avg_loss:                                   0.0010731398
+    max:
+      max_actor_approxkl1:                      0.0016232335
+      max_actor_approxkl3:                      0.0013194532
+      max_actor_clipfrac:                       0.041259766
+      max_actor_loss:                           0.0049918755
+      max_advantage:                            0.030181825
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.665697e-05
+      max_entropy:                              0.037937317
+      max_grad_norm_pre_clip:                   4.227571
+      max_loss:                                 0.004966647
+    std:
+      std_actor_approxkl1:                      0.000505017
+      std_actor_approxkl3:                      0.00035651398
+      std_actor_clipfrac:                       0.015111638
+      std_actor_loss:                           0.0015919803
+      std_advantage:                            0.004886111
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.0845637e-06
+      std_entropy:                              0.0029804087
+      std_grad_norm_pre_clip:                   1.4198382
+      std_loss:                                 0.0015919529
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040616
+    avg_avg_return:                             0.9738065
+    avg_benchmark_regret:                       0.015823077
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.1335144
+    proxy_corner:
+      avg_avg_return:                           0.9738065
+      avg_reward_per_step:                      0.1335144
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.76651
+    avg_avg_return:                             0.036224894
+    avg_benchmark_regret:                       0.95654666
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.012825012
+    proxy_corner:
+      avg_avg_return:                           0.6585027
+      avg_reward_per_step:                      0.005302429
+  eval-batch-tree:
+    avg_avg_episode_length:                     74.8525
+    avg_avg_return:                             0.1717251
+    avg_benchmark_regret:                       0.8123505
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.027359009
+    proxy_corner:
+      avg_avg_return:                           0.1717251
+      avg_reward_per_step:                      0.027359009
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        49283072
+    env-step-replay-before:                     25493504
+    env-step-used-before:                       49283072
+    ppo-update-before:                          30080
+    env-step-generate-before:                   23789568
+    env-step-all-after:                         49315840
+    env-step-replay-after:                      25526272
+    env-step-used-after:                        49315840
+    ppo-update-after:                           30100
+    env-step-generate-after:                    23789568
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8416866
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8416866
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8416866
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8416866
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       240434.1682972451
+    ppo_updates_per_second:                     41.371528134884855
+    eval-batch-orig-duration:                   0.5364713221788406
+    eval-batch-shift-duration:                  0.49100654013454914
+    eval-batch-tree-duration:                   0.4902202123776078
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.246094
+      mouse-cheese_dist_finite_wavg:            19.92318
+      mouse-cheese_dist_solvable_avg:           19.246094
+      mouse-cheese_dist_solvable_wavg:          19.92318
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14027303
+    visit_patterns:
+      avg_first_visit_time:                     354.52393
+      avg_last_visit_time:                      769.20166
+      num_replay_batches:                       779
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014129688
+    avg_actor_approxkl3:                        0.0014396323
+    avg_actor_clipfrac:                         0.040124513
+    avg_actor_loss:                             0.0009650592
+    avg_advantage:                              0.0060109217
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.7288463e-05
+    avg_entropy:                                0.07562589
+    avg_grad_norm_pre_clip:                     1.0352935
+    avg_loss:                                   0.0009030776
+    max:
+      max_actor_approxkl1:                      0.007326465
+      max_actor_approxkl3:                      0.005545332
+      max_actor_clipfrac:                       0.09094238
+      max_actor_loss:                           0.004933737
+      max_advantage:                            0.045225978
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          6.737218e-05
+      max_entropy:                              0.078725465
+      max_grad_norm_pre_clip:                   2.5368907
+      max_loss:                                 0.004890387
+    std:
+      std_actor_approxkl1:                      0.00190647
+      std_actor_approxkl3:                      0.0015745236
+      std_actor_clipfrac:                       0.033841494
+      std_actor_loss:                           0.0014340573
+      std_advantage:                            0.006557296
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.5299609e-05
+      std_entropy:                              0.002848667
+      std_grad_norm_pre_clip:                   0.72064966
+      std_loss:                                 0.0014402054
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.0433035
+    avg_avg_return:                             0.97358936
+    avg_benchmark_regret:                       0.016040172
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13344574
+    proxy_corner:
+      avg_avg_return:                           0.97358936
+      avg_reward_per_step:                      0.13344574
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.546814
+    avg_avg_return:                             0.040881712
+    avg_benchmark_regret:                       0.9518898
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.012893677
+    proxy_corner:
+      avg_avg_return:                           0.6957729
+      avg_reward_per_step:                      0.0056381226
+  eval-batch-tree:
+    avg_avg_episode_length:                     72.0752
+    avg_avg_return:                             0.20538506
+    avg_benchmark_regret:                       0.7786905
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.028289795
+    proxy_corner:
+      avg_avg_return:                           0.20538506
+      avg_reward_per_step:                      0.028289795
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        50331648
+    env-step-replay-before:                     26116096
+    env-step-used-before:                       50331648
+    ppo-update-before:                          30720
+    env-step-generate-before:                   24215552
+    env-step-all-after:                         50364416
+    env-step-replay-after:                      26116096
+    env-step-used-after:                        50364416
+    ppo-update-after:                           30740
+    env-step-generate-after:                    24248320
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.586778
+    avg_avg_return:                             0.93367225
+    avg_reward_per_step:                        0.14068604
+    proxy_corner:
+      avg_avg_return:                           0.93367225
+      avg_reward_per_step:                      0.14068604
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.586778
+    avg_avg_return:                             0.93367225
+    avg_reward_per_step:                        0.14068604
+    proxy_corner:
+      avg_avg_return:                           0.93367225
+      avg_reward_per_step:                      0.14068604
+  perf:
+    env_steps_per_second:                       319512.8658038724
+    ppo_updates_per_second:                     42.502094369732696
+    eval-batch-orig-duration:                   0.5413090344518423
+    eval-batch-shift-duration:                  0.4870415311306715
+    eval-batch-tree-duration:                   0.46467339619994164
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.229492
+      mouse-cheese_dist_finite_wavg:            19.024103
+      mouse-cheese_dist_solvable_avg:           19.229492
+      mouse-cheese_dist_solvable_wavg:          19.024105
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14027546
+    visit_patterns:
+      avg_first_visit_time:                     362.80615
+      avg_last_visit_time:                      787.28296
+      num_replay_batches:                       797
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0002967917
+    avg_actor_approxkl3:                        0.00021149935
+    avg_actor_clipfrac:                         0.0068115233
+    avg_actor_loss:                             1.1082506e-06
+    avg_advantage:                              -0.00080885226
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.307957e-05
+    avg_entropy:                                0.019794075
+    avg_grad_norm_pre_clip:                     0.39930713
+    avg_loss:                                   -1.2146035e-05
+    max:
+      max_actor_approxkl1:                      0.0016811716
+      max_actor_approxkl3:                      0.00066167134
+      max_actor_clipfrac:                       0.017333984
+      max_actor_loss:                           0.0009765411
+      max_advantage:                            0.026382864
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.6639387e-05
+      max_entropy:                              0.026719404
+      max_grad_norm_pre_clip:                   0.6637165
+      max_loss:                                 0.00096538075
+    std:
+      std_actor_approxkl1:                      0.00044297308
+      std_actor_approxkl3:                      0.00017098026
+      std_actor_clipfrac:                       0.0045292196
+      std_actor_loss:                           0.00034343108
+      std_advantage:                            0.0051853913
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.7280469e-06
+      std_entropy:                              0.0032524955
+      std_grad_norm_pre_clip:                   0.20342034
+      std_loss:                                 0.00034470597
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.044657
+    avg_avg_return:                             0.973567
+    avg_benchmark_regret:                       0.016062504
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13345337
+    proxy_corner:
+      avg_avg_return:                           0.973567
+      avg_reward_per_step:                      0.13345337
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.02251
+    avg_avg_return:                             0.044564873
+    avg_benchmark_regret:                       0.94820666
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.012702942
+    proxy_corner:
+      avg_avg_return:                           0.6295363
+      avg_reward_per_step:                      0.0051345825
+  eval-batch-tree:
+    avg_avg_episode_length:                     78.12718
+    avg_avg_return:                             0.118572704
+    avg_benchmark_regret:                       0.86550295
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.026435852
+    proxy_corner:
+      avg_avg_return:                           0.118572704
+      avg_reward_per_step:                      0.026435852
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        51380224
+    env-step-replay-before:                     26640384
+    env-step-used-before:                       51380224
+    ppo-update-before:                          31360
+    env-step-generate-before:                   24739840
+    env-step-all-after:                         51412992
+    env-step-replay-after:                      26640384
+    env-step-used-after:                        51412992
+    ppo-update-after:                           31380
+    env-step-generate-after:                    24772608
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.041197
+    avg_avg_return:                             0.9276975
+    avg_reward_per_step:                        0.14422607
+    proxy_corner:
+      avg_avg_return:                           0.9276975
+      avg_reward_per_step:                      0.14422607
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.041197
+    avg_avg_return:                             0.9276975
+    avg_reward_per_step:                        0.14422607
+    proxy_corner:
+      avg_avg_return:                           0.9276975
+      avg_reward_per_step:                      0.14422607
+  perf:
+    env_steps_per_second:                       296629.0300827503
+    ppo_updates_per_second:                     42.20082547329687
+    eval-batch-orig-duration:                   0.534979896619916
+    eval-batch-shift-duration:                  0.47475957963615656
+    eval-batch-tree-duration:                   0.47993149142712355
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.216309
+      mouse-cheese_dist_finite_wavg:            19.021915
+      mouse-cheese_dist_solvable_avg:           19.216309
+      mouse-cheese_dist_solvable_wavg:          19.021917
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1402772
+    visit_patterns:
+      avg_first_visit_time:                     369.53223
+      avg_last_visit_time:                      803.2241
+      num_replay_batches:                       813
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00043322067
+    avg_actor_approxkl3:                        0.00022171471
+    avg_actor_clipfrac:                         0.0065246583
+    avg_actor_loss:                             -0.0001815945
+    avg_advantage:                              -0.032144487
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00016404572
+    avg_entropy:                                0.025849435
+    avg_grad_norm_pre_clip:                     0.36200166
+    avg_loss:                                   -0.00012542107
+    max:
+      max_actor_approxkl1:                      0.0011622737
+      max_actor_approxkl3:                      0.0005334319
+      max_actor_clipfrac:                       0.014038086
+      max_actor_loss:                           0.00035520643
+      max_advantage:                            0.053536534
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00059791724
+      max_entropy:                              0.031617932
+      max_grad_norm_pre_clip:                   0.8106388
+      max_loss:                                 0.0004936575
+    std:
+      std_actor_approxkl1:                      0.0003761964
+      std_actor_approxkl3:                      0.0001507192
+      std_actor_clipfrac:                       0.004311083
+      std_actor_loss:                           0.00040066516
+      std_advantage:                            0.0114135565
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00014622266
+      std_entropy:                              0.0029129114
+      std_grad_norm_pre_clip:                   0.1795205
+      std_loss:                                 0.00042012706
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.04302
+    avg_avg_return:                             0.9734842
+    avg_benchmark_regret:                       0.01614529
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13343811
+    proxy_corner:
+      avg_avg_return:                           0.9734842
+      avg_reward_per_step:                      0.13343811
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.68362
+    avg_avg_return:                             0.03704438
+    avg_benchmark_regret:                       0.9557271
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.012489319
+    proxy_corner:
+      avg_avg_return:                           0.7377001
+      avg_reward_per_step:                      0.0059814453
+  eval-batch-tree:
+    avg_avg_episode_length:                     74.710045
+    avg_avg_return:                             0.17665233
+    avg_benchmark_regret:                       0.8074233
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.028060913
+    proxy_corner:
+      avg_avg_return:                           0.17665233
+      avg_reward_per_step:                      0.028060913
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        52428800
+    env-step-replay-before:                     27164672
+    env-step-used-before:                       52428800
+    ppo-update-before:                          32000
+    env-step-generate-before:                   25264128
+    env-step-all-after:                         52461568
+    env-step-replay-after:                      27197440
+    env-step-used-after:                        52461568
+    ppo-update-after:                           32020
+    env-step-generate-after:                    25264128
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8416569
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8416569
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8416569
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8416569
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       258163.7712540557
+    ppo_updates_per_second:                     41.99404184584139
+    eval-batch-orig-duration:                   0.5387417208403349
+    eval-batch-shift-duration:                  0.46955176163464785
+    eval-batch-tree-duration:                   0.45903918892145157
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.199707
+      mouse-cheese_dist_finite_wavg:            19.020416
+      mouse-cheese_dist_solvable_avg:           19.199707
+      mouse-cheese_dist_solvable_wavg:          19.020418
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14028598
+    visit_patterns:
+      avg_first_visit_time:                     378.2893
+      avg_last_visit_time:                      820.2344
+      num_replay_batches:                       830
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00012429491
+    avg_actor_approxkl3:                        6.16864e-05
+    avg_actor_clipfrac:                         0.0002746582
+    avg_actor_loss:                             -4.5183e-05
+    avg_advantage:                              -0.00031743862
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            5.2277555e-06
+    avg_entropy:                                0.0048200497
+    avg_grad_norm_pre_clip:                     0.042778026
+    avg_loss:                                   -4.738917e-05
+    max:
+      max_actor_approxkl1:                      0.0005426373
+      max_actor_approxkl3:                      0.00037338908
+      max_actor_clipfrac:                       0.0009765625
+      max_actor_loss:                           0.00011265604
+      max_advantage:                            0.03750086
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          5.7630727e-06
+      max_entropy:                              0.00705494
+      max_grad_norm_pre_clip:                   0.0927902
+      max_loss:                                 0.00010952806
+    std:
+      std_actor_approxkl1:                      0.00017256958
+      std_actor_approxkl3:                      0.000105310304
+      std_actor_clipfrac:                       0.00027802837
+      std_actor_loss:                           6.6433815e-05
+      std_advantage:                            0.0033670603
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.4300032e-07
+      std_entropy:                              0.0010484625
+      std_grad_norm_pre_clip:                   0.02044269
+      std_loss:                                 6.704877e-05
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.04377
+    avg_avg_return:                             0.9736141
+    avg_benchmark_regret:                       0.016015451
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13346863
+    proxy_corner:
+      avg_avg_return:                           0.9736141
+      avg_reward_per_step:                      0.13346863
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.622055
+    avg_avg_return:                             0.04788942
+    avg_benchmark_regret:                       0.9448821
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.012275696
+    proxy_corner:
+      avg_avg_return:                           0.5772556
+      avg_reward_per_step:                      0.004714966
+  eval-batch-tree:
+    avg_avg_episode_length:                     81.0034
+    avg_avg_return:                             0.05959376
+    avg_benchmark_regret:                       0.92448187
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025650024
+    proxy_corner:
+      avg_avg_return:                           0.05959376
+      avg_reward_per_step:                      0.025650024
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        53477376
+    env-step-replay-before:                     27623424
+    env-step-used-before:                       53477376
+    ppo-update-before:                          32640
+    env-step-generate-before:                   25853952
+    env-step-all-after:                         53510144
+    env-step-replay-after:                      27623424
+    env-step-used-after:                        53510144
+    ppo-update-after:                           32660
+    env-step-generate-after:                    25886720
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.4806595
+    avg_avg_return:                             0.9226674
+    avg_reward_per_step:                        0.131073
+    proxy_corner:
+      avg_avg_return:                           0.9226674
+      avg_reward_per_step:                      0.131073
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.4806595
+    avg_avg_return:                             0.9226674
+    avg_reward_per_step:                        0.131073
+    proxy_corner:
+      avg_avg_return:                           0.9226674
+      avg_reward_per_step:                      0.131073
+  perf:
+    env_steps_per_second:                       310709.14531517174
+    ppo_updates_per_second:                     42.198588304499985
+    eval-batch-orig-duration:                   0.5527736507356167
+    eval-batch-shift-duration:                  0.48958782479166985
+    eval-batch-tree-duration:                   0.49103728123009205
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.177246
+      mouse-cheese_dist_finite_wavg:            19.018242
+      mouse-cheese_dist_solvable_avg:           19.177246
+      mouse-cheese_dist_solvable_wavg:          19.018244
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14027646
+    visit_patterns:
+      avg_first_visit_time:                     391.01807
+      avg_last_visit_time:                      833.3574
+      num_replay_batches:                       843
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0024502114
+    avg_actor_approxkl3:                        0.0012990081
+    avg_actor_clipfrac:                         0.019195557
+    avg_actor_loss:                             -0.0006946015
+    avg_advantage:                              0.0009938157
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.1614116e-05
+    avg_entropy:                                0.04336001
+    avg_grad_norm_pre_clip:                     0.37921736
+    avg_loss:                                   -0.0007271546
+    max:
+      max_actor_approxkl1:                      0.008145399
+      max_actor_approxkl3:                      0.008092299
+      max_actor_clipfrac:                       0.061279297
+      max_actor_loss:                           0.0009274888
+      max_advantage:                            0.04131663
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.472538e-05
+      max_entropy:                              0.053529955
+      max_grad_norm_pre_clip:                   1.2831507
+      max_loss:                                 0.00089401205
+    std:
+      std_actor_approxkl1:                      0.0025212062
+      std_actor_approxkl3:                      0.0021859694
+      std_actor_clipfrac:                       0.017239762
+      std_actor_loss:                           0.0009137126
+      std_advantage:                            0.0056784833
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.280592e-06
+      std_entropy:                              0.0054718093
+      std_grad_norm_pre_clip:                   0.24789687
+      std_loss:                                 0.0009129369
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.062409
+    avg_avg_return:                             0.97300327
+    avg_benchmark_regret:                       0.016626198
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13315582
+    proxy_corner:
+      avg_avg_return:                           0.97300327
+      avg_reward_per_step:                      0.13315582
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.30304
+    avg_avg_return:                             0.040419895
+    avg_benchmark_regret:                       0.9523517
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.011291504
+    proxy_corner:
+      avg_avg_return:                           0.6291561
+      avg_reward_per_step:                      0.005126953
+  eval-batch-tree:
+    avg_avg_episode_length:                     80.86099
+    avg_avg_return:                             0.0676965
+    avg_benchmark_regret:                       0.9163791
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025634766
+    proxy_corner:
+      avg_avg_return:                           0.0676965
+      avg_reward_per_step:                      0.025634766
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        54525952
+    env-step-replay-before:                     28049408
+    env-step-used-before:                       54525952
+    ppo-update-before:                          33280
+    env-step-generate-before:                   26476544
+    env-step-all-after:                         54558720
+    env-step-replay-after:                      28049408
+    env-step-used-after:                        54558720
+    ppo-update-after:                           33300
+    env-step-generate-after:                    26509312
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.683772
+    avg_avg_return:                             0.93121266
+    avg_reward_per_step:                        0.14343262
+    proxy_corner:
+      avg_avg_return:                           0.93121266
+      avg_reward_per_step:                      0.14343262
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.683772
+    avg_avg_return:                             0.93121266
+    avg_reward_per_step:                        0.14343262
+    proxy_corner:
+      avg_avg_return:                           0.93121266
+      avg_reward_per_step:                      0.14343262
+  perf:
+    env_steps_per_second:                       240752.1048571073
+    ppo_updates_per_second:                     42.365583573677405
+    eval-batch-orig-duration:                   0.6376099046319723
+    eval-batch-shift-duration:                  0.6443522581830621
+    eval-batch-tree-duration:                   0.6168126119300723
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.160889
+      mouse-cheese_dist_finite_wavg:            19.017529
+      mouse-cheese_dist_solvable_avg:           19.160889
+      mouse-cheese_dist_solvable_wavg:          19.01753
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1402845
+    visit_patterns:
+      avg_first_visit_time:                     400.54272
+      avg_last_visit_time:                      846.2593
+      num_replay_batches:                       856
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00021655849
+    avg_actor_approxkl3:                        0.00016061401
+    avg_actor_clipfrac:                         0.003656006
+    avg_actor_loss:                             0.00014044007
+    avg_advantage:                              0.0008374767
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.7967952e-05
+    avg_entropy:                                0.013036819
+    avg_grad_norm_pre_clip:                     0.2551784
+    avg_loss:                                   0.00013638723
+    max:
+      max_actor_approxkl1:                      0.00080872467
+      max_actor_approxkl3:                      0.0005147784
+      max_actor_clipfrac:                       0.00793457
+      max_actor_loss:                           0.0013936264
+      max_advantage:                            0.026145577
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.7582675e-05
+      max_entropy:                              0.017581608
+      max_grad_norm_pre_clip:                   0.6090723
+      max_loss:                                 0.0013912576
+    std:
+      std_actor_approxkl1:                      0.0002584473
+      std_actor_approxkl3:                      0.00014243297
+      std_actor_clipfrac:                       0.0022186846
+      std_actor_loss:                           0.00040008474
+      std_advantage:                            0.0049342373
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.2976e-06
+      std_entropy:                              0.0019563849
+      std_grad_norm_pre_clip:                   0.15745294
+      std_loss:                                 0.00040012243
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.042145
+    avg_avg_return:                             0.97378576
+    avg_benchmark_regret:                       0.01584376
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.133461
+    proxy_corner:
+      avg_avg_return:                           0.97378576
+      avg_reward_per_step:                      0.133461
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.83886
+    avg_avg_return:                             0.046448544
+    avg_benchmark_regret:                       0.946323
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.01209259
+    proxy_corner:
+      avg_avg_return:                           0.6646534
+      avg_reward_per_step:                      0.0054016113
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.86165
+    avg_avg_return:                             0.09099771
+    avg_benchmark_regret:                       0.89307785
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.026000977
+    proxy_corner:
+      avg_avg_return:                           0.09099771
+      avg_reward_per_step:                      0.026000977
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        55574528
+    env-step-replay-before:                     28442624
+    env-step-used-before:                       55574528
+    ppo-update-before:                          33920
+    env-step-generate-before:                   27131904
+    env-step-all-after:                         55607296
+    env-step-replay-after:                      28475392
+    env-step-used-after:                        55607296
+    ppo-update-after:                           33940
+    env-step-generate-after:                    27131904
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.682053
+    avg_avg_return:                             0.84180063
+    avg_reward_per_step:                        0.046905518
+    proxy_corner:
+      avg_avg_return:                           0.84180063
+      avg_reward_per_step:                      0.046905518
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.682053
+    avg_avg_return:                             0.84180063
+    avg_reward_per_step:                        0.046905518
+    proxy_corner:
+      avg_avg_return:                           0.84180063
+      avg_reward_per_step:                      0.046905518
+  perf:
+    env_steps_per_second:                       320215.08434118016
+    ppo_updates_per_second:                     42.242067304236734
+    eval-batch-orig-duration:                   0.5173778422176838
+    eval-batch-shift-duration:                  0.4755626479163766
+    eval-batch-tree-duration:                   0.484731063246727
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.137695
+      mouse-cheese_dist_finite_wavg:            18.11473
+      mouse-cheese_dist_solvable_avg:           19.137695
+      mouse-cheese_dist_solvable_wavg:          18.11473
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14028111
+    visit_patterns:
+      avg_first_visit_time:                     413.14722
+      avg_last_visit_time:                      859.2407
+      num_replay_batches:                       869
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00034337502
+    avg_actor_approxkl3:                        0.0006865866
+    avg_actor_clipfrac:                         0.017871095
+    avg_actor_loss:                             -5.872435e-05
+    avg_advantage:                              0.00043514906
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.125886e-05
+    avg_entropy:                                0.04347394
+    avg_grad_norm_pre_clip:                     0.24865937
+    avg_loss:                                   -9.656885e-05
+    max:
+      max_actor_approxkl1:                      0.002440306
+      max_actor_approxkl3:                      0.002280907
+      max_actor_clipfrac:                       0.055664062
+      max_actor_loss:                           0.00019436004
+      max_advantage:                            0.027853847
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.439455e-05
+      max_entropy:                              0.048652105
+      max_grad_norm_pre_clip:                   0.7548952
+      max_loss:                                 0.0001496877
+    std:
+      std_actor_approxkl1:                      0.00061989017
+      std_actor_approxkl3:                      0.00066738296
+      std_actor_clipfrac:                       0.019082382
+      std_actor_loss:                           0.00016360343
+      std_advantage:                            0.005295987
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.1980734e-06
+      std_entropy:                              0.0032646956
+      std_grad_norm_pre_clip:                   0.20034012
+      std_loss:                                 0.00016331712
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.047895
+    avg_avg_return:                             0.9733356
+    avg_benchmark_regret:                       0.016293902
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13336182
+    proxy_corner:
+      avg_avg_return:                           0.9733356
+      avg_reward_per_step:                      0.13336182
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.68529
+    avg_avg_return:                             0.047310546
+    avg_benchmark_regret:                       0.945461
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.012329102
+    proxy_corner:
+      avg_avg_return:                           0.609513
+      avg_reward_per_step:                      0.004951477
+  eval-batch-tree:
+    avg_avg_episode_length:                     80.72333
+    avg_avg_return:                             0.07160239
+    avg_benchmark_regret:                       0.91247326
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025733948
+    proxy_corner:
+      avg_avg_return:                           0.07160239
+      avg_reward_per_step:                      0.025733948
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        56623104
+    env-step-replay-before:                     28966912
+    env-step-used-before:                       56623104
+    ppo-update-before:                          34560
+    env-step-generate-before:                   27656192
+    env-step-all-after:                         56655872
+    env-step-replay-after:                      28999680
+    env-step-used-after:                        56655872
+    ppo-update-after:                           34580
+    env-step-generate-after:                    27656192
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84170794
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84170794
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84170794
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84170794
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       245293.0370868776
+    ppo_updates_per_second:                     42.35558463523344
+    eval-batch-orig-duration:                   0.659337037242949
+    eval-batch-shift-duration:                  0.6160377003252506
+    eval-batch-tree-duration:                   0.6144875725731254
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.124512
+      mouse-cheese_dist_finite_wavg:            19.01251
+      mouse-cheese_dist_solvable_avg:           19.124512
+      mouse-cheese_dist_solvable_wavg:          19.01251
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14029124
+    visit_patterns:
+      avg_first_visit_time:                     421.3689
+      avg_last_visit_time:                      875.3323
+      num_replay_batches:                       885
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0019221356
+    avg_actor_approxkl3:                        0.0017770199
+    avg_actor_clipfrac:                         0.023895264
+    avg_actor_loss:                             -0.000271553
+    avg_advantage:                              0.00093602156
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            8.727653e-06
+    avg_entropy:                                0.062701754
+    avg_grad_norm_pre_clip:                     0.35135126
+    avg_loss:                                   -0.00032989093
+    max:
+      max_actor_approxkl1:                      0.004128145
+      max_actor_approxkl3:                      0.0035631147
+      max_actor_clipfrac:                       0.037841797
+      max_actor_loss:                           0.00053707603
+      max_advantage:                            0.02477336
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.4247289e-05
+      max_entropy:                              0.06962172
+      max_grad_norm_pre_clip:                   1.4948038
+      max_loss:                                 0.00047885865
+    std:
+      std_actor_approxkl1:                      0.0013251265
+      std_actor_approxkl3:                      0.0011715371
+      std_actor_clipfrac:                       0.013140566
+      std_actor_loss:                           0.0004515717
+      std_advantage:                            0.0044559515
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.5538582e-06
+      std_entropy:                              0.005275323
+      std_grad_norm_pre_clip:                   0.3852671
+      std_loss:                                 0.0004542562
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041441
+    avg_avg_return:                             0.97385454
+    avg_benchmark_regret:                       0.015774932
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13348389
+    proxy_corner:
+      avg_avg_return:                           0.97385454
+      avg_reward_per_step:                      0.13348389
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.29954
+    avg_avg_return:                             0.042733215
+    avg_benchmark_regret:                       0.9500383
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.011222839
+    proxy_corner:
+      avg_avg_return:                           0.6682004
+      avg_reward_per_step:                      0.0054473877
+  eval-batch-tree:
+    avg_avg_episode_length:                     81.18431
+    avg_avg_return:                             0.056991223
+    avg_benchmark_regret:                       0.92708445
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025611877
+    proxy_corner:
+      avg_avg_return:                           0.056991223
+      avg_reward_per_step:                      0.025611877
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        57671680
+    env-step-replay-before:                     29458432
+    env-step-used-before:                       57671680
+    ppo-update-before:                          35200
+    env-step-generate-before:                   28213248
+    env-step-all-after:                         57704448
+    env-step-replay-after:                      29458432
+    env-step-used-after:                        57704448
+    ppo-update-after:                           35220
+    env-step-generate-after:                    28246016
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.102514
+    avg_avg_return:                             0.92678463
+    avg_reward_per_step:                        0.14501953
+    proxy_corner:
+      avg_avg_return:                           0.92678463
+      avg_reward_per_step:                      0.14501953
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.102514
+    avg_avg_return:                             0.92678463
+    avg_reward_per_step:                        0.14501953
+    proxy_corner:
+      avg_avg_return:                           0.92678463
+      avg_reward_per_step:                      0.14501953
+  perf:
+    env_steps_per_second:                       320880.3725480953
+    ppo_updates_per_second:                     41.189630414034774
+    eval-batch-orig-duration:                   0.5362707385793328
+    eval-batch-shift-duration:                  0.48149868566542864
+    eval-batch-tree-duration:                   0.4812275869771838
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.10791
+      mouse-cheese_dist_finite_wavg:            19.011797
+      mouse-cheese_dist_solvable_avg:           19.10791
+      mouse-cheese_dist_solvable_wavg:          19.011799
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14029488
+    visit_patterns:
+      avg_first_visit_time:                     431.74536
+      avg_last_visit_time:                      889.2595
+      num_replay_batches:                       899
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0008642467
+    avg_actor_approxkl3:                        0.0007438976
+    avg_actor_clipfrac:                         0.009381103
+    avg_actor_loss:                             0.0005653063
+    avg_advantage:                              -0.0014438471
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.565098e-05
+    avg_entropy:                                0.01220522
+    avg_grad_norm_pre_clip:                     0.23711228
+    avg_loss:                                   0.0005659266
+    max:
+      max_actor_approxkl1:                      0.0032373103
+      max_actor_approxkl3:                      0.0022353674
+      max_actor_clipfrac:                       0.021484375
+      max_actor_loss:                           0.0016564317
+      max_advantage:                            0.02803129
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          5.267755e-05
+      max_entropy:                              0.016524883
+      max_grad_norm_pre_clip:                   0.5054159
+      max_loss:                                 0.0016567778
+    std:
+      std_actor_approxkl1:                      0.00092951115
+      std_actor_approxkl3:                      0.00071981654
+      std_actor_clipfrac:                       0.006253322
+      std_actor_loss:                           0.0004281892
+      std_advantage:                            0.0064085545
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.1061866e-05
+      std_entropy:                              0.0025194336
+      std_grad_norm_pre_clip:                   0.09339491
+      std_loss:                                 0.0004325315
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.0423355
+    avg_avg_return:                             0.9738486
+    avg_benchmark_regret:                       0.015780963
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13347626
+    proxy_corner:
+      avg_avg_return:                           0.9738486
+      avg_reward_per_step:                      0.13347626
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.72975
+    avg_avg_return:                             0.048484165
+    avg_benchmark_regret:                       0.94428736
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.011764526
+    proxy_corner:
+      avg_avg_return:                           0.6134844
+      avg_reward_per_step:                      0.005012512
+  eval-batch-tree:
+    avg_avg_episode_length:                     80.93979
+    avg_avg_return:                             0.061853737
+    avg_benchmark_regret:                       0.9222219
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025604248
+    proxy_corner:
+      avg_avg_return:                           0.061853737
+      avg_reward_per_step:                      0.025604248
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        58720256
+    env-step-replay-before:                     29851648
+    env-step-used-before:                       58720256
+    ppo-update-before:                          35840
+    env-step-generate-before:                   28868608
+    env-step-all-after:                         58753024
+    env-step-replay-after:                      29851648
+    env-step-used-after:                        58753024
+    ppo-update-after:                           35860
+    env-step-generate-after:                    28901376
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.437143
+    avg_avg_return:                             0.9239044
+    avg_reward_per_step:                        0.12930298
+    proxy_corner:
+      avg_avg_return:                           0.9239044
+      avg_reward_per_step:                      0.12930298
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.437143
+    avg_avg_return:                             0.9239044
+    avg_reward_per_step:                        0.12930298
+    proxy_corner:
+      avg_avg_return:                           0.9239044
+      avg_reward_per_step:                      0.12930298
+  perf:
+    env_steps_per_second:                       544713.705517484
+    ppo_updates_per_second:                     42.21465311823696
+    eval-batch-orig-duration:                   0.3161838836967945
+    eval-batch-shift-duration:                  0.2238013343885541
+    eval-batch-tree-duration:                   0.24898422323167324
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.09082
+      mouse-cheese_dist_finite_wavg:            19.009771
+      mouse-cheese_dist_solvable_avg:           19.09082
+      mouse-cheese_dist_solvable_wavg:          19.00977
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14029938
+    visit_patterns:
+      avg_first_visit_time:                     441.50757
+      avg_last_visit_time:                      901.17163
+      num_replay_batches:                       911
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0004211409
+    avg_actor_approxkl3:                        0.00030391137
+    avg_actor_clipfrac:                         0.012347412
+    avg_actor_loss:                             0.0011815381
+    avg_advantage:                              -0.00072118576
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.05763e-05
+    avg_entropy:                                0.03132477
+    avg_grad_norm_pre_clip:                     1.3539389
+    avg_loss:                                   0.0011605014
+    max:
+      max_actor_approxkl1:                      0.0011420993
+      max_actor_approxkl3:                      0.0009608018
+      max_actor_clipfrac:                       0.026245117
+      max_actor_loss:                           0.0030593686
+      max_advantage:                            0.03431511
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.0909843e-05
+      max_entropy:                              0.03995474
+      max_grad_norm_pre_clip:                   2.2171943
+      max_loss:                                 0.0030394574
+    std:
+      std_actor_approxkl1:                      0.00032478137
+      std_actor_approxkl3:                      0.00025449207
+      std_actor_clipfrac:                       0.007809569
+      std_actor_loss:                           0.001050659
+      std_advantage:                            0.005975726
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.3687714e-06
+      std_entropy:                              0.0038327929
+      std_grad_norm_pre_clip:                   0.68569404
+      std_loss:                                 0.0010506377
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.060243
+    avg_avg_return:                             0.9731943
+    avg_benchmark_regret:                       0.016435213
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13330841
+    proxy_corner:
+      avg_avg_return:                           0.9731943
+      avg_reward_per_step:                      0.13330841
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.63112
+    avg_avg_return:                             0.04872493
+    avg_benchmark_regret:                       0.94404656
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.012626648
+    proxy_corner:
+      avg_avg_return:                           0.60975885
+      avg_reward_per_step:                      0.0049591064
+  eval-batch-tree:
+    avg_avg_episode_length:                     81.184235
+    avg_avg_return:                             0.054117907
+    avg_benchmark_regret:                       0.92995775
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025619507
+    proxy_corner:
+      avg_avg_return:                           0.054117907
+      avg_reward_per_step:                      0.025619507
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        59768832
+    env-step-replay-before:                     30441472
+    env-step-used-before:                       59768832
+    ppo-update-before:                          36480
+    env-step-generate-before:                   29327360
+    env-step-all-after:                         59801600
+    env-step-replay-after:                      30474240
+    env-step-used-after:                        59801600
+    ppo-update-after:                           36500
+    env-step-generate-after:                    29327360
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84176224
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84176224
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84176224
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84176224
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       318259.1791333021
+    ppo_updates_per_second:                     42.11040170530875
+    eval-batch-orig-duration:                   0.5489604789763689
+    eval-batch-shift-duration:                  0.48326185531914234
+    eval-batch-tree-duration:                   0.4826893676072359
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.07666
+      mouse-cheese_dist_finite_wavg:            19.00777
+      mouse-cheese_dist_solvable_avg:           19.07666
+      mouse-cheese_dist_solvable_wavg:          19.007772
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14031234
+    visit_patterns:
+      avg_first_visit_time:                     450.06152
+      avg_last_visit_time:                      920.1824
+      num_replay_batches:                       930
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0026464905
+    avg_actor_approxkl3:                        0.002349662
+    avg_actor_clipfrac:                         0.042559814
+    avg_actor_loss:                             -0.00038081376
+    avg_advantage:                              0.0053610187
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.6963957e-05
+    avg_entropy:                                0.07551331
+    avg_grad_norm_pre_clip:                     0.39514196
+    avg_loss:                                   -0.0004428451
+    max:
+      max_actor_approxkl1:                      0.007936809
+      max_actor_approxkl3:                      0.006418906
+      max_actor_clipfrac:                       0.10888672
+      max_actor_loss:                           0.0006951764
+      max_advantage:                            0.028662086
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          6.965977e-05
+      max_entropy:                              0.084188476
+      max_grad_norm_pre_clip:                   0.8052295
+      max_loss:                                 0.00062925066
+    std:
+      std_actor_approxkl1:                      0.0022358296
+      std_actor_approxkl3:                      0.0018694678
+      std_actor_clipfrac:                       0.040777374
+      std_actor_loss:                           0.00057030946
+      std_advantage:                            0.010535559
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.6947726e-05
+      std_entropy:                              0.0054278253
+      std_grad_norm_pre_clip:                   0.18049856
+      std_loss:                                 0.0005761583
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.049528
+    avg_avg_return:                             0.9734544
+    avg_benchmark_regret:                       0.016175138
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13344574
+    proxy_corner:
+      avg_avg_return:                           0.9734544
+      avg_reward_per_step:                      0.13344574
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.68374
+    avg_avg_return:                             0.048111334
+    avg_benchmark_regret:                       0.9446602
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.011550903
+    proxy_corner:
+      avg_avg_return:                           0.6443685
+      avg_reward_per_step:                      0.0052719116
+  eval-batch-tree:
+    avg_avg_episode_length:                     80.74397
+    avg_avg_return:                             0.06588802
+    avg_benchmark_regret:                       0.9181876
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025650024
+    proxy_corner:
+      avg_avg_return:                           0.06588802
+      avg_reward_per_step:                      0.025650024
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        60817408
+    env-step-replay-before:                     30998528
+    env-step-used-before:                       60817408
+    ppo-update-before:                          37120
+    env-step-generate-before:                   29818880
+    env-step-all-after:                         60850176
+    env-step-replay-after:                      30998528
+    env-step-used-after:                        60850176
+    ppo-update-after:                           37140
+    env-step-generate-after:                    29851648
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.109109
+    avg_avg_return:                             0.92438364
+    avg_reward_per_step:                        0.13018799
+    proxy_corner:
+      avg_avg_return:                           0.92438364
+      avg_reward_per_step:                      0.13018799
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.109109
+    avg_avg_return:                             0.92438364
+    avg_reward_per_step:                        0.13018799
+    proxy_corner:
+      avg_avg_return:                           0.92438364
+      avg_reward_per_step:                      0.13018799
+  perf:
+    env_steps_per_second:                       343962.5516573274
+    ppo_updates_per_second:                     42.31571474772687
+    eval-batch-orig-duration:                   0.5302624525502324
+    eval-batch-shift-duration:                  0.4965982250869274
+    eval-batch-tree-duration:                   0.4682479640468955
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.060547
+      mouse-cheese_dist_finite_wavg:            19.006311
+      mouse-cheese_dist_solvable_avg:           19.060547
+      mouse-cheese_dist_solvable_wavg:          19.006313
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030921
+    visit_patterns:
+      avg_first_visit_time:                     460.5276
+      avg_last_visit_time:                      936.2866
+      num_replay_batches:                       946
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00028844003
+    avg_actor_approxkl3:                        0.00022790933
+    avg_actor_clipfrac:                         0.0065856935
+    avg_actor_loss:                             -0.00012091951
+    avg_advantage:                              0.0007381578
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.4508123e-05
+    avg_entropy:                                0.024779651
+    avg_grad_norm_pre_clip:                     0.21417251
+    avg_loss:                                   -0.0001384451
+    max:
+      max_actor_approxkl1:                      0.0007274983
+      max_actor_approxkl3:                      0.00060590974
+      max_actor_clipfrac:                       0.01586914
+      max_actor_loss:                           0.00032342318
+      max_advantage:                            0.026209652
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.5759175e-05
+      max_entropy:                              0.033096157
+      max_grad_norm_pre_clip:                   0.6266152
+      max_loss:                                 0.0003085886
+    std:
+      std_actor_approxkl1:                      0.00026640648
+      std_actor_approxkl3:                      0.00014554836
+      std_actor_clipfrac:                       0.0043520206
+      std_actor_loss:                           0.00023814233
+      std_advantage:                            0.0051962556
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.649796e-06
+      std_entropy:                              0.0032403101
+      std_grad_norm_pre_clip:                   0.13463227
+      std_loss:                                 0.00023932362
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.045263
+    avg_avg_return:                             0.9735725
+    avg_benchmark_regret:                       0.016057003
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13346863
+    proxy_corner:
+      avg_avg_return:                           0.9735725
+      avg_reward_per_step:                      0.13346863
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.49495
+    avg_avg_return:                             0.049849384
+    avg_benchmark_regret:                       0.9429222
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.012527466
+    proxy_corner:
+      avg_avg_return:                           0.64570147
+      avg_reward_per_step:                      0.005256653
+  eval-batch-tree:
+    avg_avg_episode_length:                     81.18417
+    avg_avg_return:                             0.0562529
+    avg_benchmark_regret:                       0.9278227
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025596619
+    proxy_corner:
+      avg_avg_return:                           0.0562529
+      avg_reward_per_step:                      0.025596619
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        61865984
+    env-step-replay-before:                     31424512
+    env-step-used-before:                       61865984
+    ppo-update-before:                          37760
+    env-step-generate-before:                   30441472
+    env-step-all-after:                         61898752
+    env-step-replay-after:                      31424512
+    env-step-used-after:                        61898752
+    ppo-update-after:                           37780
+    env-step-generate-after:                    30474240
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.8100605
+    avg_avg_return:                             0.9299513
+    avg_reward_per_step:                        0.1481018
+    proxy_corner:
+      avg_avg_return:                           0.9299513
+      avg_reward_per_step:                      0.1481018
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.8100605
+    avg_avg_return:                             0.9299513
+    avg_reward_per_step:                        0.1481018
+    proxy_corner:
+      avg_avg_return:                           0.9299513
+      avg_reward_per_step:                      0.1481018
+  perf:
+    env_steps_per_second:                       744148.3502936409
+    ppo_updates_per_second:                     41.83247220054731
+    eval-batch-orig-duration:                   0.2978918571025133
+    eval-batch-shift-duration:                  0.23577385675162077
+    eval-batch-tree-duration:                   0.25655673909932375
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.045898
+      mouse-cheese_dist_finite_wavg:            19.004826
+      mouse-cheese_dist_solvable_avg:           19.045898
+      mouse-cheese_dist_solvable_wavg:          19.004827
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14032912
+    visit_patterns:
+      avg_first_visit_time:                     470.5061
+      avg_last_visit_time:                      949.25073
+      num_replay_batches:                       959
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00012381731
+    avg_actor_approxkl3:                        0.00012330418
+    avg_actor_clipfrac:                         0.006427002
+    avg_actor_loss:                             9.006169e-05
+    avg_advantage:                              0.005298633
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.5106675e-05
+    avg_entropy:                                0.050604697
+    avg_grad_norm_pre_clip:                     0.38591486
+    avg_loss:                                   4.701033e-05
+    max:
+      max_actor_approxkl1:                      0.0004500585
+      max_actor_approxkl3:                      0.00038406622
+      max_actor_clipfrac:                       0.014160156
+      max_actor_loss:                           0.00046046637
+      max_advantage:                            0.032681286
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.4557288e-05
+      max_entropy:                              0.06311015
+      max_grad_norm_pre_clip:                   0.7176005
+      max_loss:                                 0.0004308016
+    std:
+      std_actor_approxkl1:                      0.00018046073
+      std_actor_approxkl3:                      9.8218545e-05
+      std_actor_clipfrac:                       0.0044243163
+      std_actor_loss:                           0.0002690832
+      std_advantage:                            0.004656946
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.9485117e-06
+      std_entropy:                              0.00499279
+      std_grad_norm_pre_clip:                   0.16312467
+      std_loss:                                 0.00026976087
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.044315
+    avg_avg_return:                             0.97374797
+    avg_benchmark_regret:                       0.015881492
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13344574
+    proxy_corner:
+      avg_avg_return:                           0.97374797
+      avg_reward_per_step:                      0.13344574
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.7762
+    avg_avg_return:                             0.035395287
+    avg_benchmark_regret:                       0.95737624
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.009643555
+    proxy_corner:
+      avg_avg_return:                           0.7749994
+      avg_reward_per_step:                      0.00630188
+  eval-batch-tree:
+    avg_avg_episode_length:                     80.47104
+    avg_avg_return:                             0.07711218
+    avg_benchmark_regret:                       0.9069634
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025749207
+    proxy_corner:
+      avg_avg_return:                           0.07711218
+      avg_reward_per_step:                      0.025749207
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        62914560
+    env-step-replay-before:                     31981568
+    env-step-used-before:                       62914560
+    ppo-update-before:                          38400
+    env-step-generate-before:                   30932992
+    env-step-all-after:                         62947328
+    env-step-replay-after:                      31981568
+    env-step-used-after:                        62947328
+    ppo-update-after:                           38420
+    env-step-generate-after:                    30965760
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.6805525
+    avg_avg_return:                             0.9285548
+    avg_reward_per_step:                        0.14474487
+    proxy_corner:
+      avg_avg_return:                           0.9285548
+      avg_reward_per_step:                      0.14474487
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.6805525
+    avg_avg_return:                             0.9285548
+    avg_reward_per_step:                        0.14474487
+    proxy_corner:
+      avg_avg_return:                           0.9285548
+      avg_reward_per_step:                      0.14474487
+  perf:
+    env_steps_per_second:                       318161.87941067066
+    ppo_updates_per_second:                     40.7968332875028
+    eval-batch-orig-duration:                   0.46355101838707924
+    eval-batch-shift-duration:                  0.4716293215751648
+    eval-batch-tree-duration:                   0.41931246779859066
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.03125
+      mouse-cheese_dist_finite_wavg:            19.003729
+      mouse-cheese_dist_solvable_avg:           19.03125
+      mouse-cheese_dist_solvable_wavg:          19.003729
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030185
+    visit_patterns:
+      avg_first_visit_time:                     479.82715
+      avg_last_visit_time:                      966.3721
+      num_replay_batches:                       976
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0001828423
+    avg_actor_approxkl3:                        0.00010085951
+    avg_actor_clipfrac:                         0.003894043
+    avg_actor_loss:                             -9.3383926e-05
+    avg_advantage:                              0.0053715687
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.1847346e-05
+    avg_entropy:                                0.0353594
+    avg_grad_norm_pre_clip:                     0.27898863
+    avg_loss:                                   -0.000107819644
+    max:
+      max_actor_approxkl1:                      0.00045741568
+      max_actor_approxkl3:                      0.00020525526
+      max_actor_clipfrac:                       0.008300781
+      max_actor_loss:                           0.000113924965
+      max_advantage:                            0.038707376
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          8.6189204e-05
+      max_entropy:                              0.04010334
+      max_grad_norm_pre_clip:                   0.67104816
+      max_loss:                                 0.000103391474
+    std:
+      std_actor_approxkl1:                      0.00015489032
+      std_actor_approxkl3:                      5.6501583e-05
+      std_actor_clipfrac:                       0.0030755661
+      std_actor_loss:                           0.00012724285
+      std_advantage:                            0.0102457125
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.8981373e-05
+      std_entropy:                              0.0026465976
+      std_grad_norm_pre_clip:                   0.16826464
+      std_loss:                                 0.00013168989
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.043349
+    avg_avg_return:                             0.9737557
+    avg_benchmark_regret:                       0.015873801
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.133461
+    proxy_corner:
+      avg_avg_return:                           0.9737557
+      avg_reward_per_step:                      0.133461
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.087616
+    avg_avg_return:                             0.043740705
+    avg_benchmark_regret:                       0.9490308
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0104522705
+    proxy_corner:
+      avg_avg_return:                           0.73463714
+      avg_reward_per_step:                      0.006011963
+  eval-batch-tree:
+    avg_avg_episode_length:                     81.05415
+    avg_avg_return:                             0.067799926
+    avg_benchmark_regret:                       0.9162757
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025627136
+    proxy_corner:
+      avg_avg_return:                           0.067799926
+      avg_reward_per_step:                      0.025627136
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        63963136
+    env-step-replay-before:                     32505856
+    env-step-used-before:                       63963136
+    ppo-update-before:                          39040
+    env-step-generate-before:                   31457280
+    env-step-all-after:                         63995904
+    env-step-replay-after:                      32505856
+    env-step-used-after:                        63995904
+    ppo-update-after:                           39060
+    env-step-generate-after:                    31490048
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.273892
+    avg_avg_return:                             0.92563474
+    avg_reward_per_step:                        0.13562012
+    proxy_corner:
+      avg_avg_return:                           0.92563474
+      avg_reward_per_step:                      0.13562012
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.273892
+    avg_avg_return:                             0.92563474
+    avg_reward_per_step:                        0.13562012
+    proxy_corner:
+      avg_avg_return:                           0.92563474
+      avg_reward_per_step:                      0.13562012
+  perf:
+    env_steps_per_second:                       299435.2720095406
+    ppo_updates_per_second:                     42.351645950433856
+    eval-batch-orig-duration:                   0.620633470825851
+    eval-batch-shift-duration:                  0.518906801007688
+    eval-batch-tree-duration:                   0.5658243345096707
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.017578
+      mouse-cheese_dist_finite_wavg:            19.002241
+      mouse-cheese_dist_solvable_avg:           19.017578
+      mouse-cheese_dist_solvable_wavg:          19.002245
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14032108
+    visit_patterns:
+      avg_first_visit_time:                     488.4087
+      avg_last_visit_time:                      982.22876
+      num_replay_batches:                       992
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00012965382
+    avg_actor_approxkl3:                        0.0001966025
+    avg_actor_clipfrac:                         0.008190918
+    avg_actor_loss:                             0.00023146936
+    avg_advantage:                              -0.00082878757
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.939967e-05
+    avg_entropy:                                0.026480714
+    avg_grad_norm_pre_clip:                     0.6248567
+    avg_loss:                                   0.00021968847
+    max:
+      max_actor_approxkl1:                      0.0006183217
+      max_actor_approxkl3:                      0.0004426528
+      max_actor_clipfrac:                       0.023803711
+      max_actor_loss:                           0.0015203729
+      max_advantage:                            0.037705183
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.730809e-05
+      max_entropy:                              0.03598895
+      max_grad_norm_pre_clip:                   2.1844413
+      max_loss:                                 0.0015051583
+    std:
+      std_actor_approxkl1:                      0.00026609548
+      std_actor_approxkl3:                      0.00013706491
+      std_actor_clipfrac:                       0.0058151996
+      std_actor_loss:                           0.0006210535
+      std_advantage:                            0.007961014
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.9644383e-06
+      std_entropy:                              0.004230299
+      std_grad_norm_pre_clip:                   0.5006994
+      std_loss:                                 0.00062121265
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.057084
+    avg_avg_return:                             0.9735956
+    avg_benchmark_regret:                       0.016033955
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.1333847
+    proxy_corner:
+      avg_avg_return:                           0.9735956
+      avg_reward_per_step:                      0.1333847
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.594284
+    avg_avg_return:                             0.04864106
+    avg_benchmark_regret:                       0.9441304
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.013114929
+    proxy_corner:
+      avg_avg_return:                           0.6775007
+      avg_reward_per_step:                      0.005508423
+  eval-batch-tree:
+    avg_avg_episode_length:                     81.18375
+    avg_avg_return:                             0.0504639
+    avg_benchmark_regret:                       0.93361175
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02558899
+    proxy_corner:
+      avg_avg_return:                           0.0504639
+      avg_reward_per_step:                      0.02558899
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        65011712
+    env-step-replay-before:                     33062912
+    env-step-used-before:                       65011712
+    ppo-update-before:                          39680
+    env-step-generate-before:                   31948800
+    env-step-all-after:                         65044480
+    env-step-replay-after:                      33062912
+    env-step-used-after:                        65044480
+    ppo-update-after:                           39700
+    env-step-generate-after:                    31981568
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.917517
+    avg_avg_return:                             0.92862964
+    avg_reward_per_step:                        0.14212036
+    proxy_corner:
+      avg_avg_return:                           0.92862964
+      avg_reward_per_step:                      0.14212036
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.917517
+    avg_avg_return:                             0.92862964
+    avg_reward_per_step:                        0.14212036
+    proxy_corner:
+      avg_avg_return:                           0.92862964
+      avg_reward_per_step:                      0.14212036
+  perf:
+    env_steps_per_second:                       314873.39352376247
+    ppo_updates_per_second:                     42.33295512451411
+    eval-batch-orig-duration:                   0.5204816339537501
+    eval-batch-shift-duration:                  0.48952846601605415
+    eval-batch-tree-duration:                   0.4822897808626294
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.012207
+      mouse-cheese_dist_finite_wavg:            19.001377
+      mouse-cheese_dist_solvable_avg:           19.012207
+      mouse-cheese_dist_solvable_wavg:          19.001379
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14034319
+    visit_patterns:
+      avg_first_visit_time:                     491.812
+      avg_last_visit_time:                      999.27295
+      num_replay_batches:                       1009
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00039130618
+    avg_actor_approxkl3:                        0.00036761176
+    avg_actor_clipfrac:                         0.0022460937
+    avg_actor_loss:                             -4.8012032e-05
+    avg_advantage:                              0.0001565798
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.3186419e-05
+    avg_entropy:                                0.010100305
+    avg_grad_norm_pre_clip:                     0.110094905
+    avg_loss:                                   -5.1519124e-05
+    max:
+      max_actor_approxkl1:                      0.0014620081
+      max_actor_approxkl3:                      0.0014398857
+      max_actor_clipfrac:                       0.010498047
+      max_actor_loss:                           0.000676
+      max_advantage:                            0.03265363
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.6756054e-05
+      max_entropy:                              0.014254235
+      max_grad_norm_pre_clip:                   0.48468837
+      max_loss:                                 0.0006723166
+    std:
+      std_actor_approxkl1:                      0.0005188191
+      std_actor_approxkl3:                      0.0003493845
+      std_actor_clipfrac:                       0.002602532
+      std_actor_loss:                           0.0003553814
+      std_advantage:                            0.005026523
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.6335297e-06
+      std_entropy:                              0.0017993103
+      std_grad_norm_pre_clip:                   0.099599436
+      std_loss:                                 0.0003551524
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.045755
+    avg_avg_return:                             0.9737859
+    avg_benchmark_regret:                       0.015843632
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13337708
+    proxy_corner:
+      avg_avg_return:                           0.9737859
+      avg_reward_per_step:                      0.13337708
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.33053
+    avg_avg_return:                             0.05201605
+    avg_benchmark_regret:                       0.9407555
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.014122009
+    proxy_corner:
+      avg_avg_return:                           0.62697875
+      avg_reward_per_step:                      0.0050735474
+  eval-batch-tree:
+    avg_avg_episode_length:                     81.136086
+    avg_avg_return:                             0.051727243
+    avg_benchmark_regret:                       0.9323484
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025619507
+    proxy_corner:
+      avg_avg_return:                           0.051727243
+      avg_reward_per_step:                      0.025619507
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        66060288
+    env-step-replay-before:                     33554432
+    env-step-used-before:                       66060288
+    ppo-update-before:                          40320
+    env-step-generate-before:                   32505856
+    env-step-all-after:                         66093056
+    env-step-replay-after:                      33554432
+    env-step-used-after:                        66093056
+    ppo-update-after:                           40340
+    env-step-generate-after:                    32538624
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.996761
+    avg_avg_return:                             0.9287466
+    avg_reward_per_step:                        0.13470459
+    proxy_corner:
+      avg_avg_return:                           0.9287466
+      avg_reward_per_step:                      0.13470459
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.996761
+    avg_avg_return:                             0.9287466
+    avg_reward_per_step:                        0.13470459
+    proxy_corner:
+      avg_avg_return:                           0.9287466
+      avg_reward_per_step:                      0.13470459
+  perf:
+    env_steps_per_second:                       348543.46390619676
+    ppo_updates_per_second:                     41.22764937318742
+    eval-batch-orig-duration:                   0.5335587058216333
+    eval-batch-shift-duration:                  0.4847958320751786
+    eval-batch-tree-duration:                   0.4998873993754387
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.004883
+      mouse-cheese_dist_finite_wavg:            19.000774
+      mouse-cheese_dist_solvable_avg:           19.004883
+      mouse-cheese_dist_solvable_wavg:          19.000774
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14031598
+    visit_patterns:
+      avg_first_visit_time:                     496.30103
+      avg_last_visit_time:                      1014.35425
+      num_replay_batches:                       1024
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0008777136
+    avg_actor_approxkl3:                        0.0005933725
+    avg_actor_clipfrac:                         0.010424805
+    avg_actor_loss:                             -1.6249251e-06
+    avg_advantage:                              -0.0057774577
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            3.6000627e-05
+    avg_entropy:                                0.02302742
+    avg_grad_norm_pre_clip:                     0.6280972
+    avg_loss:                                   -6.652033e-06
+    max:
+      max_actor_approxkl1:                      0.0020856285
+      max_actor_approxkl3:                      0.0015241579
+      max_actor_clipfrac:                       0.030029297
+      max_actor_loss:                           0.0021769404
+      max_advantage:                            0.022328973
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00012117466
+      max_entropy:                              0.029055163
+      max_grad_norm_pre_clip:                   2.2329028
+      max_loss:                                 0.0021688533
+    std:
+      std_actor_approxkl1:                      0.00074155856
+      std_actor_approxkl3:                      0.00045486077
+      std_actor_clipfrac:                       0.008707849
+      std_actor_loss:                           0.0008019207
+      std_advantage:                            0.0048142583
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.7310609e-05
+      std_entropy:                              0.00393901
+      std_grad_norm_pre_clip:                   0.51202965
+      std_loss:                                 0.00080050476
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.043714
+    avg_avg_return:                             0.97347057
+    avg_benchmark_regret:                       0.016158909
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13343811
+    proxy_corner:
+      avg_avg_return:                           0.97347057
+      avg_reward_per_step:                      0.13343811
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.521805
+    avg_avg_return:                             0.049491875
+    avg_benchmark_regret:                       0.9432796
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.013587952
+    proxy_corner:
+      avg_avg_return:                           0.6940273
+      avg_reward_per_step:                      0.0055999756
+  eval-batch-tree:
+    avg_avg_episode_length:                     81.1837
+    avg_avg_return:                             0.050479174
+    avg_benchmark_regret:                       0.93359643
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025611877
+    proxy_corner:
+      avg_avg_return:                           0.050479174
+      avg_reward_per_step:                      0.025611877
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        67108864
+    env-step-replay-before:                     34078720
+    env-step-used-before:                       67108864
+    ppo-update-before:                          40960
+    env-step-generate-before:                   33030144
+    env-step-all-after:                         67141632
+    env-step-replay-after:                      34111488
+    env-step-used-after:                        67141632
+    ppo-update-after:                           40980
+    env-step-generate-after:                    33030144
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     20.797203
+    avg_avg_return:                             0.8400707
+    avg_reward_per_step:                        0.04446411
+    proxy_corner:
+      avg_avg_return:                           0.8400707
+      avg_reward_per_step:                      0.04446411
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     20.797203
+    avg_avg_return:                             0.8400707
+    avg_reward_per_step:                        0.04446411
+    proxy_corner:
+      avg_avg_return:                           0.8400707
+      avg_reward_per_step:                      0.04446411
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  perf:
+    env_steps_per_second:                       295917.0397550212
+    ppo_updates_per_second:                     41.95996133784608
+    eval-batch-orig-duration:                   0.6468252157792449
+    eval-batch-shift-duration:                  0.6173746725544333
+    eval-batch-tree-duration:                   0.5930084129795432
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.001709
+      mouse-cheese_dist_finite_wavg:            19.000286
+      mouse-cheese_dist_solvable_avg:           19.001709
+      mouse-cheese_dist_solvable_wavg:          19.000286
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1415106
+    visit_patterns:
+      avg_first_visit_time:                     499.89038
+      avg_last_visit_time:                      1031.2952
+      num_replay_batches:                       1041
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002859767
+    avg_actor_approxkl3:                        0.0029440348
+    avg_actor_clipfrac:                         0.11480103
+    avg_actor_loss:                             -0.0011680288
+    avg_advantage:                              0.0027776114
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            3.2695996e-05
+    avg_entropy:                                0.14411543
+    avg_grad_norm_pre_clip:                     2.666897
+    avg_loss:                                   -0.001295796
+    max:
+      max_actor_approxkl1:                      0.0087242285
+      max_actor_approxkl3:                      0.007282709
+      max_actor_clipfrac:                       0.17773438
+      max_actor_loss:                           0.0022239434
+      max_advantage:                            0.04412067
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          4.511064e-05
+      max_entropy:                              0.16161649
+      max_grad_norm_pre_clip:                   5.5649624
+      max_loss:                                 0.0020816878
+    std:
+      std_actor_approxkl1:                      0.0026889357
+      std_actor_approxkl3:                      0.0022177654
+      std_actor_clipfrac:                       0.0571775
+      std_actor_loss:                           0.0014173583
+      std_advantage:                            0.0089355325
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.2982978e-06
+      std_entropy:                              0.00929004
+      std_grad_norm_pre_clip:                   1.4407653
+      std_loss:                                 0.001415239
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.286238
+    avg_avg_return:                             0.9724403
+    avg_benchmark_regret:                       0.017189242
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13201904
+    proxy_corner:
+      avg_avg_return:                           0.9724403
+      avg_reward_per_step:                      0.13201904
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.377106
+    avg_avg_return:                             0.050759315
+    avg_benchmark_regret:                       0.94201225
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.014480591
+    proxy_corner:
+      avg_avg_return:                           0.57559
+      avg_reward_per_step:                      0.0046463013
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-tree:
+    avg_avg_episode_length:                     81.18373
+    avg_avg_return:                             0.05195222
+    avg_benchmark_regret:                       0.9321234
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025611877
+    proxy_corner:
+      avg_avg_return:                           0.05195222
+      avg_reward_per_step:                      0.025611877
+    rollouts_gif:                               float32(132, 225, 225, 3)
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        68157440
+    env-step-replay-before:                     34668544
+    env-step-used-before:                       68157440
+    ppo-update-before:                          41600
+    env-step-generate-before:                   33488896
+    env-step-all-after:                         68190208
+    env-step-replay-after:                      34668544
+    env-step-used-after:                        68190208
+    ppo-update-after:                           41620
+    env-step-generate-after:                    33521664
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.32333
+    avg_avg_return:                             0.9221629
+    avg_reward_per_step:                        0.14041138
+    proxy_corner:
+      avg_avg_return:                           0.9221629
+      avg_reward_per_step:                      0.14041138
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.32333
+    avg_avg_return:                             0.9221629
+    avg_reward_per_step:                        0.14041138
+    proxy_corner:
+      avg_avg_return:                           0.9221629
+      avg_reward_per_step:                      0.14041138
+  perf:
+    env_steps_per_second:                       295721.43739330966
+    ppo_updates_per_second:                     42.270170251074816
+    eval-batch-orig-duration:                   0.5284535679966211
+    eval-batch-shift-duration:                  0.46330601815134287
+    eval-batch-tree-duration:                   0.4578986447304487
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.998535
+      mouse-cheese_dist_finite_wavg:            18.999777
+      mouse-cheese_dist_solvable_avg:           18.998535
+      mouse-cheese_dist_solvable_wavg:          18.999777
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1406821
+    visit_patterns:
+      avg_first_visit_time:                     504.88818
+      avg_last_visit_time:                      1048.3994
+      num_replay_batches:                       1058
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00064748514
+    avg_actor_approxkl3:                        0.000769015
+    avg_actor_clipfrac:                         0.017089844
+    avg_actor_loss:                             0.00032821656
+    avg_advantage:                              -0.0034758574
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            3.0648134e-05
+    avg_entropy:                                0.043688204
+    avg_grad_norm_pre_clip:                     0.99309933
+    avg_loss:                                   0.00029985243
+    max:
+      max_actor_approxkl1:                      0.0015756828
+      max_actor_approxkl3:                      0.0012776743
+      max_actor_clipfrac:                       0.039916992
+      max_actor_loss:                           0.0020447616
+      max_advantage:                            0.026745498
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          4.36887e-05
+      max_entropy:                              0.055301055
+      max_grad_norm_pre_clip:                   2.910242
+      max_loss:                                 0.002016851
+    std:
+      std_actor_approxkl1:                      0.00043637448
+      std_actor_approxkl3:                      0.00032665607
+      std_actor_clipfrac:                       0.008734725
+      std_actor_loss:                           0.00075241295
+      std_advantage:                            0.006936852
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          6.4814403e-06
+      std_entropy:                              0.006968651
+      std_grad_norm_pre_clip:                   0.8850771
+      std_loss:                                 0.0007509746
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.0752
+    avg_avg_return:                             0.97315663
+    avg_benchmark_regret:                       0.01647286
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13317108
+    proxy_corner:
+      avg_avg_return:                           0.97315663
+      avg_reward_per_step:                      0.13317108
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.30935
+    avg_avg_return:                             0.05230321
+    avg_benchmark_regret:                       0.9404683
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.014434814
+    proxy_corner:
+      avg_avg_return:                           0.638247
+      avg_reward_per_step:                      0.0051498413
+  eval-batch-tree:
+    avg_avg_episode_length:                     81.18402
+    avg_avg_return:                             0.050447993
+    avg_benchmark_regret:                       0.9336276
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02558899
+    proxy_corner:
+      avg_avg_return:                           0.050447993
+      avg_reward_per_step:                      0.02558899
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        69206016
+    env-step-replay-before:                     35258368
+    env-step-used-before:                       69206016
+    ppo-update-before:                          42240
+    env-step-generate-before:                   33947648
+    env-step-all-after:                         69238784
+    env-step-replay-after:                      35258368
+    env-step-used-after:                        69238784
+    ppo-update-after:                           42260
+    env-step-generate-after:                    33980416
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.371338
+    avg_avg_return:                             0.92432284
+    avg_reward_per_step:                        0.13513184
+    proxy_corner:
+      avg_avg_return:                           0.92432284
+      avg_reward_per_step:                      0.13513184
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.371338
+    avg_avg_return:                             0.92432284
+    avg_reward_per_step:                        0.13513184
+    proxy_corner:
+      avg_avg_return:                           0.92432284
+      avg_reward_per_step:                      0.13513184
+  perf:
+    env_steps_per_second:                       379355.8188068101
+    ppo_updates_per_second:                     42.348728077276945
+    eval-batch-orig-duration:                   0.48277711868286133
+    eval-batch-shift-duration:                  0.44577848445624113
+    eval-batch-tree-duration:                   0.4625497302040458
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.998047
+      mouse-cheese_dist_finite_wavg:            18.999918
+      mouse-cheese_dist_solvable_avg:           18.998047
+      mouse-cheese_dist_solvable_wavg:          18.99992
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14024505
+    visit_patterns:
+      avg_first_visit_time:                     505.70117
+      avg_last_visit_time:                      1066.3374
+      num_replay_batches:                       1076
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.000704545
+    avg_actor_approxkl3:                        0.00094550446
+    avg_actor_clipfrac:                         0.019866943
+    avg_actor_loss:                             0.00025661686
+    avg_advantage:                              -0.0058196136
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0005457943
+    avg_entropy:                                0.05496338
+    avg_grad_norm_pre_clip:                     0.96030915
+    avg_loss:                                   0.0004745506
+    max:
+      max_actor_approxkl1:                      0.0022594272
+      max_actor_approxkl3:                      0.002163638
+      max_actor_clipfrac:                       0.042236328
+      max_actor_loss:                           0.0024771784
+      max_advantage:                            0.045386136
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0014215903
+      max_entropy:                              0.07293456
+      max_grad_norm_pre_clip:                   2.9832344
+      max_loss:                                 0.0024771169
+    std:
+      std_actor_approxkl1:                      0.0006348658
+      std_actor_approxkl3:                      0.00054564077
+      std_actor_clipfrac:                       0.011679959
+      std_actor_loss:                           0.0008647586
+      std_advantage:                            0.010914045
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00042599914
+      std_entropy:                              0.007797879
+      std_grad_norm_pre_clip:                   0.6672698
+      std_loss:                                 0.0008393141
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.045964
+    avg_avg_return:                             0.97360903
+    avg_benchmark_regret:                       0.016020453
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13333893
+    proxy_corner:
+      avg_avg_return:                           0.97360903
+      avg_reward_per_step:                      0.13333893
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.57184
+    avg_avg_return:                             0.03901211
+    avg_benchmark_regret:                       0.95375943
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0082473755
+    proxy_corner:
+      avg_avg_return:                           0.81500125
+      avg_reward_per_step:                      0.006614685
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.71338
+    avg_avg_return:                             0.10118027
+    avg_benchmark_regret:                       0.88289535
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.026069641
+    proxy_corner:
+      avg_avg_return:                           0.10118027
+      avg_reward_per_step:                      0.026069641
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        70254592
+    env-step-replay-before:                     35651584
+    env-step-used-before:                       70254592
+    ppo-update-before:                          42880
+    env-step-generate-before:                   34603008
+    env-step-all-after:                         70287360
+    env-step-replay-after:                      35651584
+    env-step-used-after:                        70287360
+    ppo-update-after:                           42900
+    env-step-generate-after:                    34635776
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.939404
+    avg_avg_return:                             0.9267801
+    avg_reward_per_step:                        0.12945557
+    proxy_corner:
+      avg_avg_return:                           0.9267801
+      avg_reward_per_step:                      0.12945557
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.939404
+    avg_avg_return:                             0.9267801
+    avg_reward_per_step:                        0.12945557
+    proxy_corner:
+      avg_avg_return:                           0.9267801
+      avg_reward_per_step:                      0.12945557
+  perf:
+    env_steps_per_second:                       275886.8423694728
+    ppo_updates_per_second:                     41.566026559435734
+    eval-batch-orig-duration:                   0.5396435176953673
+    eval-batch-shift-duration:                  0.48771452624350786
+    eval-batch-tree-duration:                   0.484168435446918
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.997803
+      mouse-cheese_dist_finite_wavg:            18.999792
+      mouse-cheese_dist_solvable_avg:           18.997803
+      mouse-cheese_dist_solvable_wavg:          18.999794
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14024642
+    visit_patterns:
+      avg_first_visit_time:                     506.06152
+      avg_last_visit_time:                      1078.3779
+      num_replay_batches:                       1088
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0017924181
+    avg_actor_approxkl3:                        0.001041357
+    avg_actor_clipfrac:                         0.022644043
+    avg_actor_loss:                             -0.0007324621
+    avg_advantage:                              0.0009935978
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.8650959e-05
+    avg_entropy:                                0.068528146
+    avg_grad_norm_pre_clip:                     0.60762066
+    avg_loss:                                   -0.0007916647
+    max:
+      max_actor_approxkl1:                      0.006747393
+      max_actor_approxkl3:                      0.0039872625
+      max_actor_clipfrac:                       0.068603516
+      max_actor_loss:                           0.00052192993
+      max_advantage:                            0.029259384
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.3741364e-05
+      max_entropy:                              0.088226035
+      max_grad_norm_pre_clip:                   1.5464296
+      max_loss:                                 0.00046407626
+    std:
+      std_actor_approxkl1:                      0.0016662758
+      std_actor_approxkl3:                      0.0010826681
+      std_actor_clipfrac:                       0.016923696
+      std_actor_loss:                           0.0009167199
+      std_advantage:                            0.0057667093
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.3272587e-06
+      std_entropy:                              0.011918616
+      std_grad_norm_pre_clip:                   0.3332672
+      std_loss:                                 0.0009220769
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.083797
+    avg_avg_return:                             0.97325087
+    avg_benchmark_regret:                       0.016378712
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.1327591
+    proxy_corner:
+      avg_avg_return:                           0.97325087
+      avg_reward_per_step:                      0.1327591
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.6264
+    avg_avg_return:                             0.040399916
+    avg_benchmark_regret:                       0.9523716
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0056991577
+    proxy_corner:
+      avg_avg_return:                           0.8433637
+      avg_reward_per_step:                      0.0069122314
+  eval-batch-tree:
+    avg_avg_episode_length:                     77.56239
+    avg_avg_return:                             0.1474137
+    avg_benchmark_regret:                       0.8366619
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.026992798
+    proxy_corner:
+      avg_avg_return:                           0.1474137
+      avg_reward_per_step:                      0.026992798
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        71303168
+    env-step-replay-before:                     36044800
+    env-step-used-before:                       71303168
+    ppo-update-before:                          43520
+    env-step-generate-before:                   35258368
+    env-step-all-after:                         71335936
+    env-step-replay-after:                      36044800
+    env-step-used-after:                        71335936
+    ppo-update-after:                           43540
+    env-step-generate-after:                    35291136
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.478991
+    avg_avg_return:                             0.9240751
+    avg_reward_per_step:                        0.13833618
+    proxy_corner:
+      avg_avg_return:                           0.9240751
+      avg_reward_per_step:                      0.13833618
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.478991
+    avg_avg_return:                             0.9240751
+    avg_reward_per_step:                        0.13833618
+    proxy_corner:
+      avg_avg_return:                           0.9240751
+      avg_reward_per_step:                      0.13833618
+  perf:
+    env_steps_per_second:                       296646.5541645721
+    ppo_updates_per_second:                     42.259216192247614
+    eval-batch-orig-duration:                   0.5296773239970207
+    eval-batch-shift-duration:                  0.4820583248510957
+    eval-batch-tree-duration:                   0.485934660769999
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.998535
+      mouse-cheese_dist_finite_wavg:            18.9998
+      mouse-cheese_dist_solvable_avg:           18.998535
+      mouse-cheese_dist_solvable_wavg:          18.999802
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14026141
+    visit_patterns:
+      avg_first_visit_time:                     506.11865
+      avg_last_visit_time:                      1090.385
+      num_replay_batches:                       1100
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00088902324
+    avg_actor_approxkl3:                        0.0006186635
+    avg_actor_clipfrac:                         0.022406006
+    avg_actor_loss:                             0.00017122344
+    avg_advantage:                              0.0011785473
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.7981809e-05
+    avg_entropy:                                0.048109993
+    avg_grad_norm_pre_clip:                     1.2384914
+    avg_loss:                                   0.00013710435
+    max:
+      max_actor_approxkl1:                      0.0037479158
+      max_actor_approxkl3:                      0.0029343043
+      max_actor_clipfrac:                       0.072631836
+      max_actor_loss:                           0.0037344054
+      max_advantage:                            0.039421976
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.268643e-05
+      max_entropy:                              0.05826236
+      max_grad_norm_pre_clip:                   4.3141603
+      max_loss:                                 0.0036939327
+    std:
+      std_actor_approxkl1:                      0.0010501161
+      std_actor_approxkl3:                      0.00080395007
+      std_actor_clipfrac:                       0.018735208
+      std_actor_loss:                           0.0011666167
+      std_advantage:                            0.0072666225
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.97609e-06
+      std_entropy:                              0.005694583
+      std_grad_norm_pre_clip:                   1.0029899
+      std_loss:                                 0.0011651092
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.045647
+    avg_avg_return:                             0.97366405
+    avg_benchmark_regret:                       0.015965512
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.1334076
+    proxy_corner:
+      avg_avg_return:                           0.97366405
+      avg_reward_per_step:                      0.1334076
+  eval-batch-shift:
+    avg_avg_episode_length:                     83.08035
+    avg_avg_return:                             0.031616867
+    avg_benchmark_regret:                       0.96115464
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0063934326
+    proxy_corner:
+      avg_avg_return:                           0.837752
+      avg_reward_per_step:                      0.0067749023
+  eval-batch-tree:
+    avg_avg_episode_length:                     78.359146
+    avg_avg_return:                             0.117581084
+    avg_benchmark_regret:                       0.86649454
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.026535034
+    proxy_corner:
+      avg_avg_return:                           0.117581084
+      avg_reward_per_step:                      0.026535034
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        72351744
+    env-step-replay-before:                     36536320
+    env-step-used-before:                       72351744
+    ppo-update-before:                          44160
+    env-step-generate-before:                   35815424
+    env-step-all-after:                         72384512
+    env-step-replay-after:                      36536320
+    env-step-used-after:                        72384512
+    ppo-update-after:                           44180
+    env-step-generate-after:                    35848192
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.651619
+    avg_avg_return:                             0.9315232
+    avg_reward_per_step:                        0.14706421
+    proxy_corner:
+      avg_avg_return:                           0.9315232
+      avg_reward_per_step:                      0.14706421
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.651619
+    avg_avg_return:                             0.9315232
+    avg_reward_per_step:                        0.14706421
+    proxy_corner:
+      avg_avg_return:                           0.9315232
+      avg_reward_per_step:                      0.14706421
+  perf:
+    env_steps_per_second:                       319449.3726612536
+    ppo_updates_per_second:                     41.97091754248863
+    eval-batch-orig-duration:                   0.5110831307247281
+    eval-batch-shift-duration:                  0.4561091000214219
+    eval-batch-tree-duration:                   0.4782328298315406
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.998535
+      mouse-cheese_dist_finite_wavg:            18.999748
+      mouse-cheese_dist_solvable_avg:           18.998535
+      mouse-cheese_dist_solvable_wavg:          18.999748
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14026566
+    visit_patterns:
+      avg_first_visit_time:                     506.37524
+      avg_last_visit_time:                      1105.1008
+      num_replay_batches:                       1115
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010019528
+    avg_actor_approxkl3:                        0.00072143495
+    avg_actor_clipfrac:                         0.018432617
+    avg_actor_loss:                             0.000626537
+    avg_advantage:                              -0.0011508926
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.7832206e-05
+    avg_entropy:                                0.031119213
+    avg_grad_norm_pre_clip:                     0.86226475
+    avg_loss:                                   0.00060433376
+    max:
+      max_actor_approxkl1:                      0.0023445105
+      max_actor_approxkl3:                      0.0017254802
+      max_actor_clipfrac:                       0.035888672
+      max_actor_loss:                           0.003900775
+      max_advantage:                            0.031161964
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.4664503e-05
+      max_entropy:                              0.045055926
+      max_grad_norm_pre_clip:                   1.9122715
+      max_loss:                                 0.0038777038
+    std:
+      std_actor_approxkl1:                      0.0006564205
+      std_actor_approxkl3:                      0.0004957929
+      std_actor_clipfrac:                       0.010816421
+      std_actor_loss:                           0.0010761445
+      std_advantage:                            0.0052199056
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.597525e-06
+      std_entropy:                              0.0056100804
+      std_grad_norm_pre_clip:                   0.37732288
+      std_loss:                                 0.0010767987
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041094
+    avg_avg_return:                             0.97368383
+    avg_benchmark_regret:                       0.015945679
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13348389
+    proxy_corner:
+      avg_avg_return:                           0.97368383
+      avg_reward_per_step:                      0.13348389
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.93799
+    avg_avg_return:                             0.03708239
+    avg_benchmark_regret:                       0.95568913
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.006713867
+    proxy_corner:
+      avg_avg_return:                           0.80987084
+      avg_reward_per_step:                      0.0065841675
+  eval-batch-tree:
+    avg_avg_episode_length:                     80.077225
+    avg_avg_return:                             0.08533712
+    avg_benchmark_regret:                       0.8987385
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.026115417
+    proxy_corner:
+      avg_avg_return:                           0.08533712
+      avg_reward_per_step:                      0.026115417
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        73400320
+    env-step-replay-before:                     37289984
+    env-step-used-before:                       73400320
+    ppo-update-before:                          44800
+    env-step-generate-before:                   36110336
+    env-step-all-after:                         73433088
+    env-step-replay-after:                      37322752
+    env-step-used-after:                        73433088
+    ppo-update-after:                           44820
+    env-step-generate-after:                    36110336
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184384
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184384
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184384
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184384
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       325223.29345084546
+    ppo_updates_per_second:                     42.31041897144044
+    eval-batch-orig-duration:                   0.5013272725045681
+    eval-batch-shift-duration:                  0.4304606830701232
+    eval-batch-tree-duration:                   0.47720764111727476
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.998291
+      mouse-cheese_dist_finite_wavg:            18.9996
+      mouse-cheese_dist_solvable_avg:           18.998291
+      mouse-cheese_dist_solvable_wavg:          18.999601
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14026584
+    visit_patterns:
+      avg_first_visit_time:                     506.39673
+      avg_last_visit_time:                      1129.2004
+      num_replay_batches:                       1139
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        9.409913e-05
+    avg_actor_approxkl3:                        6.9580674e-05
+    avg_actor_clipfrac:                         0.0036315918
+    avg_actor_loss:                             2.4242514e-05
+    avg_advantage:                              -0.0003296041
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.5588378e-06
+    avg_entropy:                                0.016182272
+    avg_grad_norm_pre_clip:                     0.19275086
+    avg_loss:                                   8.83966e-06
+    max:
+      max_actor_approxkl1:                      0.0008360676
+      max_actor_approxkl3:                      0.0004225637
+      max_actor_clipfrac:                       0.02746582
+      max_actor_loss:                           0.00034891814
+      max_advantage:                            0.0038762093
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.5185777e-06
+      max_entropy:                              0.019824674
+      max_grad_norm_pre_clip:                   0.8581079
+      max_loss:                                 0.00033408948
+    std:
+      std_actor_approxkl1:                      0.00020941524
+      std_actor_approxkl3:                      0.00010283214
+      std_actor_clipfrac:                       0.006432292
+      std_actor_loss:                           0.00012288516
+      std_advantage:                            0.0017577914
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.846435e-07
+      std_entropy:                              0.0016867638
+      std_grad_norm_pre_clip:                   0.26240802
+      std_loss:                                 0.00012257554
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041443
+    avg_avg_return:                             0.9737972
+    avg_benchmark_regret:                       0.015832333
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349915
+    proxy_corner:
+      avg_avg_return:                           0.9737972
+      avg_reward_per_step:                      0.13349915
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.13406
+    avg_avg_return:                             0.043471865
+    avg_benchmark_regret:                       0.9492997
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.009468079
+    proxy_corner:
+      avg_avg_return:                           0.7232466
+      avg_reward_per_step:                      0.0058898926
+  eval-batch-tree:
+    avg_avg_episode_length:                     80.77075
+    avg_avg_return:                             0.06368304
+    avg_benchmark_regret:                       0.9203926
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.0256958
+    proxy_corner:
+      avg_avg_return:                           0.06368304
+      avg_reward_per_step:                      0.0256958
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        74448896
+    env-step-replay-before:                     38043648
+    env-step-used-before:                       74448896
+    ppo-update-before:                          45440
+    env-step-generate-before:                   36405248
+    env-step-all-after:                         74481664
+    env-step-replay-after:                      38043648
+    env-step-used-after:                        74481664
+    ppo-update-after:                           45460
+    env-step-generate-after:                    36438016
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.279081
+    avg_avg_return:                             0.9244156
+    avg_reward_per_step:                        0.13027954
+    proxy_corner:
+      avg_avg_return:                           0.9244156
+      avg_reward_per_step:                      0.13027954
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.279081
+    avg_avg_return:                             0.9244156
+    avg_reward_per_step:                        0.13027954
+    proxy_corner:
+      avg_avg_return:                           0.9244156
+      avg_reward_per_step:                      0.13027954
+  perf:
+    env_steps_per_second:                       320374.2782169867
+    ppo_updates_per_second:                     41.521766241998655
+    eval-batch-orig-duration:                   0.5235891677439213
+    eval-batch-shift-duration:                  0.4440241241827607
+    eval-batch-tree-duration:                   0.45759911462664604
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.998291
+      mouse-cheese_dist_finite_wavg:            18.999836
+      mouse-cheese_dist_solvable_avg:           18.998291
+      mouse-cheese_dist_solvable_wavg:          18.999838
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1402657
+    visit_patterns:
+      avg_first_visit_time:                     506.39673
+      avg_last_visit_time:                      1151.3005
+      num_replay_batches:                       1161
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0008464576
+    avg_actor_approxkl3:                        0.00045299972
+    avg_actor_clipfrac:                         0.009326172
+    avg_actor_loss:                             -0.00063514657
+    avg_advantage:                              0.002764238
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.5357685e-05
+    avg_entropy:                                0.021972202
+    avg_grad_norm_pre_clip:                     0.41904578
+    avg_loss:                                   -0.00064443995
+    max:
+      max_actor_approxkl1:                      0.002547547
+      max_actor_approxkl3:                      0.0012692192
+      max_actor_clipfrac:                       0.022583008
+      max_actor_loss:                           0.0006238837
+      max_advantage:                            0.045318127
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.697707e-05
+      max_entropy:                              0.03152558
+      max_grad_norm_pre_clip:                   1.053548
+      max_loss:                                 0.0006187513
+    std:
+      std_actor_approxkl1:                      0.0007519086
+      std_actor_approxkl3:                      0.0003443262
+      std_actor_clipfrac:                       0.0067358487
+      std_actor_loss:                           0.0008141388
+      std_advantage:                            0.0072521293
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.4816015e-06
+      std_entropy:                              0.0042216815
+      std_grad_norm_pre_clip:                   0.21180063
+      std_loss:                                 0.0008167321
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040855
+    avg_avg_return:                             0.9737451
+    avg_benchmark_regret:                       0.015884409
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349915
+    proxy_corner:
+      avg_avg_return:                           0.9737451
+      avg_reward_per_step:                      0.13349915
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.77402
+    avg_avg_return:                             0.036982447
+    avg_benchmark_regret:                       0.9557891
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0071258545
+    proxy_corner:
+      avg_avg_return:                           0.7833075
+      avg_reward_per_step:                      0.006362915
+  eval-batch-tree:
+    avg_avg_episode_length:                     80.47631
+    avg_avg_return:                             0.075928956
+    avg_benchmark_regret:                       0.9081467
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.0258255
+    proxy_corner:
+      avg_avg_return:                           0.075928956
+      avg_reward_per_step:                      0.0258255
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        75497472
+    env-step-replay-before:                     38567936
+    env-step-used-before:                       75497472
+    ppo-update-before:                          46080
+    env-step-generate-before:                   36929536
+    env-step-all-after:                         75530240
+    env-step-replay-after:                      38600704
+    env-step-used-after:                        75530240
+    ppo-update-after:                           46100
+    env-step-generate-after:                    36929536
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184384
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184384
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184384
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184384
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       322628.71628957574
+    ppo_updates_per_second:                     42.156591459703286
+    eval-batch-orig-duration:                   0.515947506763041
+    eval-batch-shift-duration:                  0.4702979186549783
+    eval-batch-tree-duration:                   0.4264480583369732
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.999023
+      mouse-cheese_dist_finite_wavg:            18.99986
+      mouse-cheese_dist_solvable_avg:           18.999023
+      mouse-cheese_dist_solvable_wavg:          18.999863
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030002
+    visit_patterns:
+      avg_first_visit_time:                     506.43042
+      avg_last_visit_time:                      1168.2678
+      num_replay_batches:                       1178
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013755212
+    avg_actor_approxkl3:                        0.0017578691
+    avg_actor_clipfrac:                         0.0022644044
+    avg_actor_loss:                             -0.0005331926
+    avg_advantage:                              0.0006239158
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            9.384531e-06
+    avg_entropy:                                0.007229119
+    avg_grad_norm_pre_clip:                     0.06591915
+    avg_loss:                                   -0.0005357295
+    max:
+      max_actor_approxkl1:                      0.0038293295
+      max_actor_approxkl3:                      0.0043105036
+      max_actor_clipfrac:                       0.0031738281
+      max_actor_loss:                           -3.2410026e-07
+      max_advantage:                            0.013902545
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.702135e-05
+      max_entropy:                              0.011647224
+      max_grad_norm_pre_clip:                   0.15256937
+      max_loss:                                 -6.6918756e-06
+    std:
+      std_actor_approxkl1:                      0.001415065
+      std_actor_approxkl3:                      0.0015378424
+      std_actor_clipfrac:                       0.00090156595
+      std_actor_loss:                           0.00034588214
+      std_advantage:                            0.0041949432
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.2610918e-06
+      std_entropy:                              0.0026176677
+      std_grad_norm_pre_clip:                   0.034254577
+      std_loss:                                 0.0003443693
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040616
+    avg_avg_return:                             0.973858
+    avg_benchmark_regret:                       0.015771486
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.1335144
+    proxy_corner:
+      avg_avg_return:                           0.973858
+      avg_reward_per_step:                      0.1335144
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.482124
+    avg_avg_return:                             0.039221946
+    avg_benchmark_regret:                       0.9535496
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.00831604
+    proxy_corner:
+      avg_avg_return:                           0.7753956
+      avg_reward_per_step:                      0.006263733
+  eval-batch-tree:
+    avg_avg_episode_length:                     80.09906
+    avg_avg_return:                             0.08031217
+    avg_benchmark_regret:                       0.9037634
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025924683
+    proxy_corner:
+      avg_avg_return:                           0.08031217
+      avg_reward_per_step:                      0.025924683
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        76546048
+    env-step-replay-before:                     39092224
+    env-step-used-before:                       76546048
+    ppo-update-before:                          46720
+    env-step-generate-before:                   37453824
+    env-step-all-after:                         76578816
+    env-step-replay-after:                      39092224
+    env-step-used-after:                        76578816
+    ppo-update-after:                           46740
+    env-step-generate-after:                    37486592
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.26302
+    avg_avg_return:                             0.92219543
+    avg_reward_per_step:                        0.13858032
+    proxy_corner:
+      avg_avg_return:                           0.92219543
+      avg_reward_per_step:                      0.13858032
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.26302
+    avg_avg_return:                             0.92219543
+    avg_reward_per_step:                        0.13858032
+    proxy_corner:
+      avg_avg_return:                           0.92219543
+      avg_reward_per_step:                      0.13858032
+  perf:
+    env_steps_per_second:                       293857.86994845467
+    ppo_updates_per_second:                     42.23422800944746
+    eval-batch-orig-duration:                   0.53013766836375
+    eval-batch-shift-duration:                  0.48771105799824
+    eval-batch-tree-duration:                   0.48810179997235537
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.999023
+      mouse-cheese_dist_finite_wavg:            18.999863
+      mouse-cheese_dist_solvable_avg:           18.999023
+      mouse-cheese_dist_solvable_wavg:          18.999863
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030063
+    visit_patterns:
+      avg_first_visit_time:                     506.43042
+      avg_last_visit_time:                      1183.1792
+      num_replay_batches:                       1193
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        4.5604975e-05
+    avg_actor_approxkl3:                        0.00013268064
+    avg_actor_clipfrac:                         0.0054992675
+    avg_actor_loss:                             -0.0002793659
+    avg_advantage:                              -0.0017815076
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.4417197e-05
+    avg_entropy:                                0.01810551
+    avg_grad_norm_pre_clip:                     0.3815388
+    avg_loss:                                   -0.00028526282
+    max:
+      max_actor_approxkl1:                      0.00029559026
+      max_actor_approxkl3:                      0.0003381842
+      max_actor_clipfrac:                       0.012939453
+      max_actor_loss:                           0.00044006575
+      max_advantage:                            0.03906256
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          4.245168e-05
+      max_entropy:                              0.02348992
+      max_grad_norm_pre_clip:                   0.9576002
+      max_loss:                                 0.0004323832
+    std:
+      std_actor_approxkl1:                      0.00015065137
+      std_actor_approxkl3:                      8.551848e-05
+      std_actor_clipfrac:                       0.0034862724
+      std_actor_loss:                           0.0005923385
+      std_advantage:                            0.007450188
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          6.058631e-06
+      std_entropy:                              0.002745159
+      std_grad_norm_pre_clip:                   0.2164195
+      std_loss:                                 0.0005929947
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040766
+    avg_avg_return:                             0.9738857
+    avg_benchmark_regret:                       0.015743794
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349915
+    proxy_corner:
+      avg_avg_return:                           0.9738857
+      avg_reward_per_step:                      0.13349915
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.78476
+    avg_avg_return:                             0.045971848
+    avg_benchmark_regret:                       0.9467997
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.009635925
+    proxy_corner:
+      avg_avg_return:                           0.68143713
+      avg_reward_per_step:                      0.005531311
+  eval-batch-tree:
+    avg_avg_episode_length:                     80.35359
+    avg_avg_return:                             0.081015676
+    avg_benchmark_regret:                       0.90305996
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025901794
+    proxy_corner:
+      avg_avg_return:                           0.081015676
+      avg_reward_per_step:                      0.025901794
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        77594624
+    env-step-replay-before:                     39714816
+    env-step-used-before:                       77594624
+    ppo-update-before:                          47360
+    env-step-generate-before:                   37879808
+    env-step-all-after:                         77627392
+    env-step-replay-after:                      39714816
+    env-step-used-after:                        77627392
+    ppo-update-after:                           47380
+    env-step-generate-after:                    37912576
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.832254
+    avg_avg_return:                             0.92951286
+    avg_reward_per_step:                        0.1459961
+    proxy_corner:
+      avg_avg_return:                           0.92951286
+      avg_reward_per_step:                      0.1459961
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.832254
+    avg_avg_return:                             0.92951286
+    avg_reward_per_step:                        0.1459961
+    proxy_corner:
+      avg_avg_return:                           0.92951286
+      avg_reward_per_step:                      0.1459961
+  perf:
+    env_steps_per_second:                       257399.08154646461
+    ppo_updates_per_second:                     42.38625886986838
+    eval-batch-orig-duration:                   0.6821423210203648
+    eval-batch-shift-duration:                  0.5979657266288996
+    eval-batch-tree-duration:                   0.6078707622364163
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.982666
+      mouse-cheese_dist_finite_wavg:            18.997786
+      mouse-cheese_dist_solvable_avg:           18.982666
+      mouse-cheese_dist_solvable_wavg:          18.997787
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14110388
+    visit_patterns:
+      avg_first_visit_time:                     513.0698
+      avg_last_visit_time:                      1202.3589
+      num_replay_batches:                       1212
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00035490704
+    avg_actor_approxkl3:                        0.00023166765
+    avg_actor_clipfrac:                         0.0051330565
+    avg_actor_loss:                             -0.00013000656
+    avg_advantage:                              0.003164676
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.9727258e-05
+    avg_entropy:                                0.019034747
+    avg_grad_norm_pre_clip:                     0.35271776
+    avg_loss:                                   -0.00013917768
+    max:
+      max_actor_approxkl1:                      0.0019167429
+      max_actor_approxkl3:                      0.0013737764
+      max_actor_clipfrac:                       0.010375977
+      max_actor_loss:                           0.00072287116
+      max_advantage:                            0.030985594
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.037439e-05
+      max_entropy:                              0.023573354
+      max_grad_norm_pre_clip:                   0.650368
+      max_loss:                                 0.0007144166
+    std:
+      std_actor_approxkl1:                      0.0004474512
+      std_actor_approxkl3:                      0.00030783308
+      std_actor_clipfrac:                       0.0031821448
+      std_actor_loss:                           0.0003680495
+      std_advantage:                            0.0059876833
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.943069e-06
+      std_entropy:                              0.00206686
+      std_grad_norm_pre_clip:                   0.14713837
+      std_loss:                                 0.00036856433
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.059071
+    avg_avg_return:                             0.9731776
+    avg_benchmark_regret:                       0.016451918
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13341522
+    proxy_corner:
+      avg_avg_return:                           0.9731776
+      avg_reward_per_step:                      0.13341522
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.07824
+    avg_avg_return:                             0.042277344
+    avg_benchmark_regret:                       0.95049417
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.008796692
+    proxy_corner:
+      avg_avg_return:                           0.7080753
+      avg_reward_per_step:                      0.0057296753
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.88111
+    avg_avg_return:                             0.09948406
+    avg_benchmark_regret:                       0.8845916
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02607727
+    proxy_corner:
+      avg_avg_return:                           0.09948406
+      avg_reward_per_step:                      0.02607727
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        78643200
+    env-step-replay-before:                     40239104
+    env-step-used-before:                       78643200
+    ppo-update-before:                          48000
+    env-step-generate-before:                   38404096
+    env-step-all-after:                         78675968
+    env-step-replay-after:                      40271872
+    env-step-used-after:                        78675968
+    ppo-update-after:                           48020
+    env-step-generate-after:                    38404096
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       256007.72550173354
+    ppo_updates_per_second:                     41.5307905291942
+    eval-batch-orig-duration:                   0.5400837920606136
+    eval-batch-shift-duration:                  0.4950002580881119
+    eval-batch-tree-duration:                   0.4802782153710723
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.990479
+      mouse-cheese_dist_finite_wavg:            18.99857
+      mouse-cheese_dist_solvable_avg:           18.990479
+      mouse-cheese_dist_solvable_wavg:          18.998571
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14020252
+    visit_patterns:
+      avg_first_visit_time:                     514.6157
+      avg_last_visit_time:                      1219.2734
+      num_replay_batches:                       1229
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.003117267
+    avg_actor_approxkl3:                        0.0031514938
+    avg_actor_clipfrac:                         0.019116212
+    avg_actor_loss:                             -0.0005401132
+    avg_advantage:                              0.0015339872
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.0242561e-05
+    avg_entropy:                                0.037804242
+    avg_grad_norm_pre_clip:                     0.32968023
+    avg_loss:                                   -0.00057279604
+    max:
+      max_actor_approxkl1:                      0.009436801
+      max_actor_approxkl3:                      0.008165753
+      max_actor_clipfrac:                       0.05029297
+      max_actor_loss:                           0.0015854989
+      max_advantage:                            0.014678299
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.3615376e-05
+      max_entropy:                              0.0463362
+      max_grad_norm_pre_clip:                   1.3936647
+      max_loss:                                 0.0015621824
+    std:
+      std_actor_approxkl1:                      0.002746196
+      std_actor_approxkl3:                      0.0025049613
+      std_actor_clipfrac:                       0.014794217
+      std_actor_loss:                           0.0010827493
+      std_advantage:                            0.0042011687
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.1112562e-06
+      std_entropy:                              0.0051513524
+      std_grad_norm_pre_clip:                   0.36477643
+      std_loss:                                 0.0010814031
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040735
+    avg_avg_return:                             0.9738499
+    avg_benchmark_regret:                       0.015779575
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13350677
+    proxy_corner:
+      avg_avg_return:                           0.9738499
+      avg_reward_per_step:                      0.13350677
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.52948
+    avg_avg_return:                             0.037078597
+    avg_benchmark_regret:                       0.9556929
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.008132935
+    proxy_corner:
+      avg_avg_return:                           0.8287892
+      avg_reward_per_step:                      0.0066986084
+  eval-batch-tree:
+    avg_avg_episode_length:                     76.28202
+    avg_avg_return:                             0.16793044
+    avg_benchmark_regret:                       0.8161452
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.027511597
+    proxy_corner:
+      avg_avg_return:                           0.16793044
+      avg_reward_per_step:                      0.027511597
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        79691776
+    env-step-replay-before:                     40861696
+    env-step-used-before:                       79691776
+    ppo-update-before:                          48640
+    env-step-generate-before:                   38830080
+    env-step-all-after:                         79724544
+    env-step-replay-after:                      40861696
+    env-step-used-after:                        79724544
+    ppo-update-after:                           48660
+    env-step-generate-after:                    38862848
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.4475
+    avg_avg_return:                             0.9323745
+    avg_reward_per_step:                        0.16210938
+    proxy_corner:
+      avg_avg_return:                           0.9323745
+      avg_reward_per_step:                      0.16210938
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.4475
+    avg_avg_return:                             0.9323745
+    avg_reward_per_step:                        0.16210938
+    proxy_corner:
+      avg_avg_return:                           0.9323745
+      avg_reward_per_step:                      0.16210938
+  perf:
+    env_steps_per_second:                       354071.39292332245
+    ppo_updates_per_second:                     42.307414841376435
+    eval-batch-orig-duration:                   0.5179440444335341
+    eval-batch-shift-duration:                  0.4795618997886777
+    eval-batch-tree-duration:                   0.4861435927450657
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.995117
+      mouse-cheese_dist_finite_wavg:            18.999498
+      mouse-cheese_dist_solvable_avg:           18.995117
+      mouse-cheese_dist_solvable_wavg:          18.9995
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1402331
+    visit_patterns:
+      avg_first_visit_time:                     514.69946
+      avg_last_visit_time:                      1237.3472
+      num_replay_batches:                       1247
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0007935675
+    avg_actor_approxkl3:                        0.0008136833
+    avg_actor_clipfrac:                         0.031781007
+    avg_actor_loss:                             0.0013036603
+    avg_advantage:                              -0.0038368092
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.645783e-05
+    avg_entropy:                                0.068610445
+    avg_grad_norm_pre_clip:                     1.3505903
+    avg_loss:                                   0.0012582787
+    max:
+      max_actor_approxkl1:                      0.0030138104
+      max_actor_approxkl3:                      0.0021333303
+      max_actor_clipfrac:                       0.08068848
+      max_actor_loss:                           0.0046313475
+      max_advantage:                            0.045913696
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          7.0273934e-05
+      max_entropy:                              0.077481814
+      max_grad_norm_pre_clip:                   3.2945757
+      max_loss:                                 0.004570684
+    std:
+      std_actor_approxkl1:                      0.0008280964
+      std_actor_approxkl3:                      0.000666427
+      std_actor_clipfrac:                       0.019731384
+      std_actor_loss:                           0.0015755403
+      std_advantage:                            0.009357325
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.968259e-06
+      std_entropy:                              0.006574636
+      std_grad_norm_pre_clip:                   0.8469485
+      std_loss:                                 0.0015722914
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.069243
+    avg_avg_return:                             0.97325224
+    avg_benchmark_regret:                       0.01637727
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13321686
+    proxy_corner:
+      avg_avg_return:                           0.97325224
+      avg_reward_per_step:                      0.13321686
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.156296
+    avg_avg_return:                             0.04626795
+    avg_benchmark_regret:                       0.9465036
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.004676819
+    proxy_corner:
+      avg_avg_return:                           0.89706075
+      avg_reward_per_step:                      0.0073547363
+  eval-batch-tree:
+    avg_avg_episode_length:                     70.341095
+    avg_avg_return:                             0.22554871
+    avg_benchmark_regret:                       0.7585269
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.029647827
+    proxy_corner:
+      avg_avg_return:                           0.22554871
+      avg_reward_per_step:                      0.029647827
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        80740352
+    env-step-replay-before:                     41222144
+    env-step-used-before:                       80740352
+    ppo-update-before:                          49280
+    env-step-generate-before:                   39518208
+    env-step-all-after:                         80773120
+    env-step-replay-after:                      41254912
+    env-step-used-after:                        80773120
+    ppo-update-after:                           49300
+    env-step-generate-after:                    39518208
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184444
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184444
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184444
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184444
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       480684.0670823698
+    ppo_updates_per_second:                     42.1582680864465
+    eval-batch-orig-duration:                   0.30309468042105436
+    eval-batch-shift-duration:                  0.23674813192337751
+    eval-batch-tree-duration:                   0.26733965799212456
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.998047
+      mouse-cheese_dist_finite_wavg:            18.99995
+      mouse-cheese_dist_solvable_avg:           18.998047
+      mouse-cheese_dist_solvable_wavg:          18.99995
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14028774
+    visit_patterns:
+      avg_first_visit_time:                     514.82886
+      avg_last_visit_time:                      1249.3062
+      num_replay_batches:                       1259
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0005617131
+    avg_actor_approxkl3:                        0.0005915515
+    avg_actor_clipfrac:                         0.03463745
+    avg_actor_loss:                             -0.00019888637
+    avg_advantage:                              -0.001686231
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.6062055e-05
+    avg_entropy:                                0.08563798
+    avg_grad_norm_pre_clip:                     0.9660209
+    avg_loss:                                   -0.0002764933
+    max:
+      max_actor_approxkl1:                      0.0016744481
+      max_actor_approxkl3:                      0.0012032307
+      max_actor_clipfrac:                       0.07702637
+      max_actor_loss:                           0.0011751242
+      max_advantage:                            0.028668761
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.3510347e-05
+      max_entropy:                              0.09043026
+      max_grad_norm_pre_clip:                   2.5638227
+      max_loss:                                 0.0010943895
+    std:
+      std_actor_approxkl1:                      0.00047618998
+      std_actor_approxkl3:                      0.0003728964
+      std_actor_clipfrac:                       0.024180027
+      std_actor_loss:                           0.0005906566
+      std_advantage:                            0.0049106767
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.9468423e-06
+      std_entropy:                              0.0029655038
+      std_grad_norm_pre_clip:                   0.7224222
+      std_loss:                                 0.0005910416
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.047813
+    avg_avg_return:                             0.9736441
+    avg_benchmark_regret:                       0.015985478
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.1333847
+    proxy_corner:
+      avg_avg_return:                           0.9736441
+      avg_reward_per_step:                      0.1333847
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.57826
+    avg_avg_return:                             0.04348986
+    avg_benchmark_regret:                       0.9492817
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.004890442
+    proxy_corner:
+      avg_avg_return:                           0.88856256
+      avg_reward_per_step:                      0.007270813
+  eval-batch-tree:
+    avg_avg_episode_length:                     73.99426
+    avg_avg_return:                             0.19427462
+    avg_benchmark_regret:                       0.789801
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.027763367
+    proxy_corner:
+      avg_avg_return:                           0.19427462
+      avg_reward_per_step:                      0.027763367
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        81788928
+    env-step-replay-before:                     41615360
+    env-step-used-before:                       81788928
+    ppo-update-before:                          49920
+    env-step-generate-before:                   40173568
+    env-step-all-after:                         81821696
+    env-step-replay-after:                      41648128
+    env-step-used-after:                        81821696
+    ppo-update-after:                           49940
+    env-step-generate-after:                    40173568
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184223
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184223
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184223
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184223
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       295086.18833301065
+    ppo_updates_per_second:                     42.21861450246578
+    eval-batch-orig-duration:                   0.5415649088099599
+    eval-batch-shift-duration:                  0.486674215644598
+    eval-batch-tree-duration:                   0.48499655071645975
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.999023
+      mouse-cheese_dist_finite_wavg:            18.999939
+      mouse-cheese_dist_solvable_avg:           18.999023
+      mouse-cheese_dist_solvable_wavg:          18.999943
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030015
+    visit_patterns:
+      avg_first_visit_time:                     514.8865
+      avg_last_visit_time:                      1261.2312
+      num_replay_batches:                       1271
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0030470928
+    avg_actor_approxkl3:                        0.0028060328
+    avg_actor_clipfrac:                         0.052557375
+    avg_actor_loss:                             -0.00069334
+    avg_advantage:                              0.0077442164
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.0826037e-05
+    avg_entropy:                                0.06795492
+    avg_grad_norm_pre_clip:                     0.45237967
+    avg_loss:                                   -0.000750882
+    max:
+      max_actor_approxkl1:                      0.0058980435
+      max_actor_approxkl3:                      0.0056375638
+      max_actor_clipfrac:                       0.075805664
+      max_actor_loss:                           0.00089020655
+      max_advantage:                            0.032413125
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          4.4895358e-05
+      max_entropy:                              0.073069505
+      max_grad_norm_pre_clip:                   1.0097294
+      max_loss:                                 0.0008364001
+    std:
+      std_actor_approxkl1:                      0.0017241659
+      std_actor_approxkl3:                      0.0015544717
+      std_actor_clipfrac:                       0.019918622
+      std_actor_loss:                           0.00053892814
+      std_advantage:                            0.0054130224
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.0787672e-05
+      std_entropy:                              0.0033117062
+      std_grad_norm_pre_clip:                   0.2832545
+      std_loss:                                 0.000540508
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.045279
+    avg_avg_return:                             0.97374934
+    avg_benchmark_regret:                       0.01588017
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13343048
+    proxy_corner:
+      avg_avg_return:                           0.97374934
+      avg_reward_per_step:                      0.13343048
+  eval-batch-shift:
+    avg_avg_episode_length:                     83.02437
+    avg_avg_return:                             0.037176184
+    avg_benchmark_regret:                       0.9555953
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.005218506
+    proxy_corner:
+      avg_avg_return:                           0.89362955
+      avg_reward_per_step:                      0.007232666
+  eval-batch-tree:
+    avg_avg_episode_length:                     76.18892
+    avg_avg_return:                             0.16808115
+    avg_benchmark_regret:                       0.81599444
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.026924133
+    proxy_corner:
+      avg_avg_return:                           0.16808115
+      avg_reward_per_step:                      0.026924133
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        82837504
+    env-step-replay-before:                     42336256
+    env-step-used-before:                       82837504
+    ppo-update-before:                          50560
+    env-step-generate-before:                   40501248
+    env-step-all-after:                         82870272
+    env-step-replay-after:                      42369024
+    env-step-used-after:                        82870272
+    ppo-update-after:                           50580
+    env-step-generate-after:                    40501248
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8418428
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8418428
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8418428
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8418428
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       347380.2768702502
+    ppo_updates_per_second:                     44.54863927172994
+    eval-batch-orig-duration:                   0.54961640201509
+    eval-batch-shift-duration:                  0.46689972281455994
+    eval-batch-tree-duration:                   0.4866087958216667
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999996
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          18.999996
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030851
+    visit_patterns:
+      avg_first_visit_time:                     514.9065
+      avg_last_visit_time:                      1283.2139
+      num_replay_batches:                       1293
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00027903073
+    avg_actor_approxkl3:                        0.00022488332
+    avg_actor_clipfrac:                         0.012561035
+    avg_actor_loss:                             0.0001794322
+    avg_advantage:                              -0.0010252779
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.0047325e-05
+    avg_entropy:                                0.05355203
+    avg_grad_norm_pre_clip:                     0.554536
+    avg_loss:                                   0.00013090378
+    max:
+      max_actor_approxkl1:                      0.00071802177
+      max_actor_approxkl3:                      0.00052438857
+      max_actor_clipfrac:                       0.040039062
+      max_actor_loss:                           0.0012295749
+      max_advantage:                            0.015105784
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.9590556e-05
+      max_entropy:                              0.058192864
+      max_grad_norm_pre_clip:                   1.5736009
+      max_loss:                                 0.0011831481
+    std:
+      std_actor_approxkl1:                      0.00025944744
+      std_actor_approxkl3:                      0.00013894396
+      std_actor_clipfrac:                       0.01118106
+      std_actor_loss:                           0.00047324554
+      std_advantage:                            0.0036533747
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.503259e-06
+      std_entropy:                              0.002287655
+      std_grad_norm_pre_clip:                   0.46282703
+      std_loss:                                 0.00047472955
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.046047
+    avg_avg_return:                             0.9735353
+    avg_benchmark_regret:                       0.016094211
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13339233
+    proxy_corner:
+      avg_avg_return:                           0.9735353
+      avg_reward_per_step:                      0.13339233
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.97531
+    avg_avg_return:                             0.03302046
+    avg_benchmark_regret:                       0.95975107
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0056533813
+    proxy_corner:
+      avg_avg_return:                           0.90229565
+      avg_reward_per_step:                      0.0072402954
+  eval-batch-tree:
+    avg_avg_episode_length:                     77.5676
+    avg_avg_return:                             0.14588821
+    avg_benchmark_regret:                       0.8381874
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.026504517
+    proxy_corner:
+      avg_avg_return:                           0.14588821
+      avg_reward_per_step:                      0.026504517
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        83886080
+    env-step-replay-before:                     42860544
+    env-step-used-before:                       83886080
+    ppo-update-before:                          51200
+    env-step-generate-before:                   41025536
+    env-step-all-after:                         83918848
+    env-step-replay-after:                      42893312
+    env-step-used-after:                        83918848
+    ppo-update-after:                           51220
+    env-step-generate-after:                    41025536
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8418417
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8418417
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8418417
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8418417
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       242073.47005431
+    ppo_updates_per_second:                     42.44279361571092
+    eval-batch-orig-duration:                   0.6601780075579882
+    eval-batch-shift-duration:                  0.6211519483476877
+    eval-batch-tree-duration:                   0.6183417486026883
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.999756
+      mouse-cheese_dist_finite_wavg:            18.999979
+      mouse-cheese_dist_solvable_avg:           18.999756
+      mouse-cheese_dist_solvable_wavg:          18.999979
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14031368
+    visit_patterns:
+      avg_first_visit_time:                     515.64233
+      avg_last_visit_time:                      1299.1599
+      num_replay_batches:                       1309
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011515204
+    avg_actor_approxkl3:                        0.00074045296
+    avg_actor_clipfrac:                         0.013153076
+    avg_actor_loss:                             0.00073730404
+    avg_advantage:                              -0.0029007392
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.115526e-05
+    avg_entropy:                                0.027922807
+    avg_grad_norm_pre_clip:                     0.94947404
+    avg_loss:                                   0.00071995886
+    max:
+      max_actor_approxkl1:                      0.003904033
+      max_actor_approxkl3:                      0.0042737615
+      max_actor_clipfrac:                       0.033813477
+      max_actor_loss:                           0.0034335898
+      max_advantage:                            0.015062034
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.9689483e-05
+      max_entropy:                              0.033670586
+      max_grad_norm_pre_clip:                   2.188613
+      max_loss:                                 0.0034230016
+    std:
+      std_actor_approxkl1:                      0.0011997047
+      std_actor_approxkl3:                      0.00096580636
+      std_actor_clipfrac:                       0.011944184
+      std_actor_loss:                           0.001473631
+      std_advantage:                            0.0056667984
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.0399756e-06
+      std_entropy:                              0.0037698478
+      std_grad_norm_pre_clip:                   0.80403525
+      std_loss:                                 0.0014765289
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.048369
+    avg_avg_return:                             0.97371364
+    avg_benchmark_regret:                       0.015915867
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13345337
+    proxy_corner:
+      avg_avg_return:                           0.97371364
+      avg_reward_per_step:                      0.13345337
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.503555
+    avg_avg_return:                             0.039216615
+    avg_benchmark_regret:                       0.95355487
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0051651
+    proxy_corner:
+      avg_avg_return:                           0.859798
+      avg_reward_per_step:                      0.006942749
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.261314
+    avg_avg_return:                             0.11053836
+    avg_benchmark_regret:                       0.87353724
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.0259552
+    proxy_corner:
+      avg_avg_return:                           0.11053836
+      avg_reward_per_step:                      0.0259552
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        84934656
+    env-step-replay-before:                     43483136
+    env-step-used-before:                       84934656
+    ppo-update-before:                          51840
+    env-step-generate-before:                   41451520
+    env-step-all-after:                         84967424
+    env-step-replay-after:                      43515904
+    env-step-used-after:                        84967424
+    ppo-update-after:                           51860
+    env-step-generate-after:                    41451520
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       295983.9787480636
+    ppo_updates_per_second:                     40.7215213749045
+    eval-batch-orig-duration:                   0.5276378048583865
+    eval-batch-shift-duration:                  0.4904890535399318
+    eval-batch-tree-duration:                   0.4794605756178498
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.999756
+      mouse-cheese_dist_finite_wavg:            18.999966
+      mouse-cheese_dist_solvable_avg:           18.999756
+      mouse-cheese_dist_solvable_wavg:          18.999966
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030546
+    visit_patterns:
+      avg_first_visit_time:                     515.64233
+      avg_last_visit_time:                      1318.2656
+      num_replay_batches:                       1328
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00018463019
+    avg_actor_approxkl3:                        0.00022515969
+    avg_actor_clipfrac:                         0.014807129
+    avg_actor_loss:                             0.00032507683
+    avg_advantage:                              -0.0012571857
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            8.654296e-06
+    avg_entropy:                                0.020491447
+    avg_grad_norm_pre_clip:                     0.4225082
+    avg_loss:                                   0.00030891257
+    max:
+      max_actor_approxkl1:                      0.0009270057
+      max_actor_approxkl3:                      0.00067697914
+      max_actor_clipfrac:                       0.03112793
+      max_actor_loss:                           0.0013803933
+      max_advantage:                            0.016063988
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.9212897e-05
+      max_entropy:                              0.023912914
+      max_grad_norm_pre_clip:                   0.8390989
+      max_loss:                                 0.0013627737
+    std:
+      std_actor_approxkl1:                      0.0002570019
+      std_actor_approxkl3:                      0.00019523411
+      std_actor_clipfrac:                       0.011823258
+      std_actor_loss:                           0.00041886934
+      std_advantage:                            0.002513386
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.633179e-06
+      std_entropy:                              0.0020414717
+      std_grad_norm_pre_clip:                   0.313092
+      std_loss:                                 0.00041898323
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.043224
+    avg_avg_return:                             0.9736974
+    avg_benchmark_regret:                       0.015932038
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349152
+    proxy_corner:
+      avg_avg_return:                           0.9736974
+      avg_reward_per_step:                      0.13349152
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.3806
+    avg_avg_return:                             0.043581896
+    avg_benchmark_regret:                       0.94918966
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0071868896
+    proxy_corner:
+      avg_avg_return:                           0.87812805
+      avg_reward_per_step:                      0.0071029663
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.91868
+    avg_avg_return:                             0.093428805
+    avg_benchmark_regret:                       0.8906468
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.0258255
+    proxy_corner:
+      avg_avg_return:                           0.093428805
+      avg_reward_per_step:                      0.0258255
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        85983232
+    env-step-replay-before:                     44072960
+    env-step-used-before:                       85983232
+    ppo-update-before:                          52480
+    env-step-generate-before:                   41910272
+    env-step-all-after:                         86016000
+    env-step-replay-after:                      44072960
+    env-step-used-after:                        86016000
+    ppo-update-after:                           52500
+    env-step-generate-after:                    41943040
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.981892
+    avg_avg_return:                             0.9285287
+    avg_reward_per_step:                        0.14788818
+    proxy_corner:
+      avg_avg_return:                           0.9285287
+      avg_reward_per_step:                      0.14788818
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.981892
+    avg_avg_return:                             0.9285287
+    avg_reward_per_step:                        0.14788818
+    proxy_corner:
+      avg_avg_return:                           0.9285287
+      avg_reward_per_step:                      0.14788818
+  perf:
+    env_steps_per_second:                       293165.49249367125
+    ppo_updates_per_second:                     42.341643737958
+    eval-batch-orig-duration:                   0.5206006225198507
+    eval-batch-shift-duration:                  0.4796842345967889
+    eval-batch-tree-duration:                   0.4500258285552263
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999998
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.0
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030783
+    visit_patterns:
+      avg_first_visit_time:                     515.6538
+      avg_last_visit_time:                      1335.3059
+      num_replay_batches:                       1345
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        6.6111344e-05
+    avg_actor_approxkl3:                        0.00017391494
+    avg_actor_clipfrac:                         0.008532715
+    avg_actor_loss:                             -0.00014094362
+    avg_advantage:                              -0.0007630191
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.881316e-05
+    avg_entropy:                                0.045898903
+    avg_grad_norm_pre_clip:                     0.36122957
+    avg_loss:                                   -0.00017743597
+    max:
+      max_actor_approxkl1:                      0.00085663376
+      max_actor_approxkl3:                      0.00041215465
+      max_actor_clipfrac:                       0.016723633
+      max_actor_loss:                           0.00067394786
+      max_advantage:                            0.047311604
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.7462449e-05
+      max_entropy:                              0.05325213
+      max_grad_norm_pre_clip:                   0.716857
+      max_loss:                                 0.0006360335
+    std:
+      std_actor_approxkl1:                      0.00025952287
+      std_actor_approxkl3:                      0.00010221974
+      std_actor_clipfrac:                       0.004700246
+      std_actor_loss:                           0.0003336088
+      std_advantage:                            0.0069052824
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.990691e-06
+      std_entropy:                              0.003973993
+      std_grad_norm_pre_clip:                   0.17832492
+      std_loss:                                 0.00033551178
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.045376
+    avg_avg_return:                             0.9735942
+    avg_benchmark_regret:                       0.016035303
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13345337
+    proxy_corner:
+      avg_avg_return:                           0.9735942
+      avg_reward_per_step:                      0.13345337
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.45351
+    avg_avg_return:                             0.043011393
+    avg_benchmark_regret:                       0.9497601
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0055999756
+    proxy_corner:
+      avg_avg_return:                           0.889915
+      avg_reward_per_step:                      0.007232666
+  eval-batch-tree:
+    avg_avg_episode_length:                     78.93483
+    avg_avg_return:                             0.11634577
+    avg_benchmark_regret:                       0.86772984
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.026245117
+    proxy_corner:
+      avg_avg_return:                           0.11634577
+      avg_reward_per_step:                      0.026245117
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        87031808
+    env-step-replay-before:                     44531712
+    env-step-used-before:                       87031808
+    ppo-update-before:                          53120
+    env-step-generate-before:                   42500096
+    env-step-all-after:                         87064576
+    env-step-replay-after:                      44531712
+    env-step-used-after:                        87064576
+    ppo-update-after:                           53140
+    env-step-generate-after:                    42532864
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.547319
+    avg_avg_return:                             0.92345804
+    avg_reward_per_step:                        0.14529419
+    proxy_corner:
+      avg_avg_return:                           0.92345804
+      avg_reward_per_step:                      0.14529419
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.547319
+    avg_avg_return:                             0.92345804
+    avg_reward_per_step:                        0.14529419
+    proxy_corner:
+      avg_avg_return:                           0.92345804
+      avg_reward_per_step:                      0.14529419
+  perf:
+    env_steps_per_second:                       254227.2700910235
+    ppo_updates_per_second:                     42.40636850362317
+    eval-batch-orig-duration:                   0.6670307265594602
+    eval-batch-shift-duration:                  0.6183781502768397
+    eval-batch-tree-duration:                   0.6176881659775972
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999996
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          18.999996
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030772
+    visit_patterns:
+      avg_first_visit_time:                     515.9033
+      avg_last_visit_time:                      1349.331
+      num_replay_batches:                       1359
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00038352804
+    avg_actor_approxkl3:                        0.0003284372
+    avg_actor_clipfrac:                         0.014630127
+    avg_actor_loss:                             0.00028063176
+    avg_advantage:                              0.00018157922
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.7261975e-05
+    avg_entropy:                                0.05331185
+    avg_grad_norm_pre_clip:                     0.5362951
+    avg_loss:                                   0.00023595091
+    max:
+      max_actor_approxkl1:                      0.001716867
+      max_actor_approxkl3:                      0.001574476
+      max_actor_clipfrac:                       0.056640625
+      max_actor_loss:                           0.002176858
+      max_advantage:                            0.04194051
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.3168446e-05
+      max_entropy:                              0.06326038
+      max_grad_norm_pre_clip:                   1.6032897
+      max_loss:                                 0.0021280588
+    std:
+      std_actor_approxkl1:                      0.0004969984
+      std_actor_approxkl3:                      0.00039900976
+      std_actor_clipfrac:                       0.016066618
+      std_actor_loss:                           0.0005558334
+      std_advantage:                            0.005761683
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.1899739e-06
+      std_entropy:                              0.0036702754
+      std_grad_norm_pre_clip:                   0.4185607
+      std_loss:                                 0.0005544769
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041927
+    avg_avg_return:                             0.973712
+    avg_benchmark_regret:                       0.015917469
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349152
+    proxy_corner:
+      avg_avg_return:                           0.973712
+      avg_reward_per_step:                      0.13349152
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.59064
+    avg_avg_return:                             0.0414626
+    avg_benchmark_regret:                       0.95130897
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.00667572
+    proxy_corner:
+      avg_avg_return:                           0.87606966
+      avg_reward_per_step:                      0.007095337
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.70032
+    avg_avg_return:                             0.10196076
+    avg_benchmark_regret:                       0.8821149
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025772095
+    proxy_corner:
+      avg_avg_return:                           0.10196076
+      avg_reward_per_step:                      0.025772095
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        88080384
+    env-step-replay-before:                     44859392
+    env-step-used-before:                       88080384
+    ppo-update-before:                          53760
+    env-step-generate-before:                   43220992
+    env-step-all-after:                         88113152
+    env-step-replay-after:                      44859392
+    env-step-used-after:                        88113152
+    ppo-update-after:                           53780
+    env-step-generate-after:                    43253760
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.315036
+    avg_avg_return:                             0.9257221
+    avg_reward_per_step:                        0.13235474
+    proxy_corner:
+      avg_avg_return:                           0.9257221
+      avg_reward_per_step:                      0.13235474
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.315036
+    avg_avg_return:                             0.9257221
+    avg_reward_per_step:                        0.13235474
+    proxy_corner:
+      avg_avg_return:                           0.9257221
+      avg_reward_per_step:                      0.13235474
+  perf:
+    env_steps_per_second:                       349352.35301157675
+    ppo_updates_per_second:                     44.5018081828484
+    eval-batch-orig-duration:                   0.5170977115631104
+    eval-batch-shift-duration:                  0.48284999281167984
+    eval-batch-tree-duration:                   0.47456331457942724
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999998
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.0
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14031169
+    visit_patterns:
+      avg_first_visit_time:                     516.1543
+      avg_last_visit_time:                      1359.3545
+      num_replay_batches:                       1369
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00042953022
+    avg_actor_approxkl3:                        0.0002617342
+    avg_actor_clipfrac:                         0.017999267
+    avg_actor_loss:                             -0.00025076795
+    avg_advantage:                              -0.00019467248
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.357342e-05
+    avg_entropy:                                0.05974513
+    avg_grad_norm_pre_clip:                     0.4311853
+    avg_loss:                                   -0.00030372638
+    max:
+      max_actor_approxkl1:                      0.0010747786
+      max_actor_approxkl3:                      0.00042153854
+      max_actor_clipfrac:                       0.034179688
+      max_actor_loss:                           0.00029790774
+      max_advantage:                            0.034323156
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.8712793e-05
+      max_entropy:                              0.07381788
+      max_grad_norm_pre_clip:                   0.9053063
+      max_loss:                                 0.00025520087
+    std:
+      std_actor_approxkl1:                      0.00032514732
+      std_actor_approxkl3:                      0.00011348653
+      std_actor_clipfrac:                       0.0095724035
+      std_actor_loss:                           0.00028371034
+      std_advantage:                            0.0054289703
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.9405015e-06
+      std_entropy:                              0.0072615766
+      std_grad_norm_pre_clip:                   0.193883
+      std_loss:                                 0.0002835717
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041616
+    avg_avg_return:                             0.9737731
+    avg_benchmark_regret:                       0.015856415
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13347626
+    proxy_corner:
+      avg_avg_return:                           0.9737731
+      avg_reward_per_step:                      0.13347626
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.70169
+    avg_avg_return:                             0.042832963
+    avg_benchmark_regret:                       0.94993854
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0065841675
+    proxy_corner:
+      avg_avg_return:                           0.8830323
+      avg_reward_per_step:                      0.007171631
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.4569
+    avg_avg_return:                             0.106955424
+    avg_benchmark_regret:                       0.87712014
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02594757
+    proxy_corner:
+      avg_avg_return:                           0.106955424
+      avg_reward_per_step:                      0.02594757
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        89128960
+    env-step-replay-before:                     45514752
+    env-step-used-before:                       89128960
+    ppo-update-before:                          54400
+    env-step-generate-before:                   43614208
+    env-step-all-after:                         89161728
+    env-step-replay-after:                      45547520
+    env-step-used-after:                        89161728
+    ppo-update-after:                           54420
+    env-step-generate-after:                    43614208
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184444
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184444
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184444
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184444
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       303766.34765521606
+    ppo_updates_per_second:                     42.14436709878601
+    eval-batch-orig-duration:                   0.542651941999793
+    eval-batch-shift-duration:                  0.4902782067656517
+    eval-batch-tree-duration:                   0.49474973045289516
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.0
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.0
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030814
+    visit_patterns:
+      avg_first_visit_time:                     516.1543
+      avg_last_visit_time:                      1380.3406
+      num_replay_batches:                       1390
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        2.7121601e-05
+    avg_actor_approxkl3:                        0.00031908482
+    avg_actor_clipfrac:                         0.008483887
+    avg_actor_loss:                             -0.00032651852
+    avg_advantage:                              0.0007372182
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.7096285e-06
+    avg_entropy:                                0.01420961
+    avg_grad_norm_pre_clip:                     0.14333622
+    avg_loss:                                   -0.00033937334
+    max:
+      max_actor_approxkl1:                      0.0006269585
+      max_actor_approxkl3:                      0.00070361746
+      max_actor_clipfrac:                       0.034301758
+      max_actor_loss:                           0.00011870265
+      max_advantage:                            0.016524851
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          4.869883e-06
+      max_entropy:                              0.01695216
+      max_grad_norm_pre_clip:                   0.51377314
+      max_loss:                                 0.00010612615
+    std:
+      std_actor_approxkl1:                      0.00025624433
+      std_actor_approxkl3:                      0.0001895385
+      std_actor_clipfrac:                       0.008709569
+      std_actor_loss:                           0.0003267579
+      std_advantage:                            0.0024224045
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.9169746e-07
+      std_entropy:                              0.0014858297
+      std_grad_norm_pre_clip:                   0.1434205
+      std_loss:                                 0.00032752802
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041321
+    avg_avg_return:                             0.9738325
+    avg_benchmark_regret:                       0.015797015
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349152
+    proxy_corner:
+      avg_avg_return:                           0.9738325
+      avg_reward_per_step:                      0.13349152
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.572784
+    avg_avg_return:                             0.04035466
+    avg_benchmark_regret:                       0.95241684
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0062026978
+    proxy_corner:
+      avg_avg_return:                           0.8892708
+      avg_reward_per_step:                      0.0072250366
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.69259
+    avg_avg_return:                             0.09705965
+    avg_benchmark_regret:                       0.88701594
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025970459
+    proxy_corner:
+      avg_avg_return:                           0.09705965
+      avg_reward_per_step:                      0.025970459
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        90177536
+    env-step-replay-before:                     46202880
+    env-step-used-before:                       90177536
+    ppo-update-before:                          55040
+    env-step-generate-before:                   43974656
+    env-step-all-after:                         90210304
+    env-step-replay-after:                      46202880
+    env-step-used-after:                        90210304
+    ppo-update-after:                           55060
+    env-step-generate-after:                    44007424
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.287756
+    avg_avg_return:                             0.9245759
+    avg_reward_per_step:                        0.12652588
+    proxy_corner:
+      avg_avg_return:                           0.9245759
+      avg_reward_per_step:                      0.12652588
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.287756
+    avg_avg_return:                             0.9245759
+    avg_reward_per_step:                        0.12652588
+    proxy_corner:
+      avg_avg_return:                           0.9245759
+      avg_reward_per_step:                      0.12652588
+  perf:
+    env_steps_per_second:                       347963.7097866982
+    ppo_updates_per_second:                     42.3469221990052
+    eval-batch-orig-duration:                   0.5223872223868966
+    eval-batch-shift-duration:                  0.47312305215746164
+    eval-batch-tree-duration:                   0.4849252151325345
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999998
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          18.999996
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     516.1543
+      avg_last_visit_time:                      1400.1829
+      num_replay_batches:                       1410
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00040743267
+    avg_actor_approxkl3:                        0.00041135438
+    avg_actor_clipfrac:                         0.01921997
+    avg_actor_loss:                             0.0007498229
+    avg_advantage:                              0.00011847961
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.52317e-05
+    avg_entropy:                                0.041663162
+    avg_grad_norm_pre_clip:                     1.4340781
+    avg_loss:                                   0.0007207755
+    max:
+      max_actor_approxkl1:                      0.001248668
+      max_actor_approxkl3:                      0.0011026014
+      max_actor_clipfrac:                       0.041992188
+      max_actor_loss:                           0.0036038645
+      max_advantage:                            0.03137684
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          4.360435e-05
+      max_entropy:                              0.05038979
+      max_grad_norm_pre_clip:                   3.078698
+      max_loss:                                 0.0035753148
+    std:
+      std_actor_approxkl1:                      0.00042915673
+      std_actor_approxkl3:                      0.00033028208
+      std_actor_clipfrac:                       0.01385869
+      std_actor_loss:                           0.0012330444
+      std_advantage:                            0.006698592
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          6.665328e-06
+      std_entropy:                              0.005057859
+      std_grad_norm_pre_clip:                   0.915522
+      std_loss:                                 0.0012322649
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.042393
+    avg_avg_return:                             0.9737603
+    avg_benchmark_regret:                       0.01586923
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13348389
+    proxy_corner:
+      avg_avg_return:                           0.9737603
+      avg_reward_per_step:                      0.13348389
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.666214
+    avg_avg_return:                             0.044955276
+    avg_benchmark_regret:                       0.94781625
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.005569458
+    proxy_corner:
+      avg_avg_return:                           0.88498735
+      avg_reward_per_step:                      0.007232666
+  eval-batch-tree:
+    avg_avg_episode_length:                     80.19301
+    avg_avg_return:                             0.089251354
+    avg_benchmark_regret:                       0.89482427
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025733948
+    proxy_corner:
+      avg_avg_return:                           0.089251354
+      avg_reward_per_step:                      0.025733948
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        91226112
+    env-step-replay-before:                     46759936
+    env-step-used-before:                       91226112
+    ppo-update-before:                          55680
+    env-step-generate-before:                   44466176
+    env-step-all-after:                         91258880
+    env-step-replay-after:                      46759936
+    env-step-used-after:                        91258880
+    ppo-update-after:                           55700
+    env-step-generate-after:                    44498944
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.130806
+    avg_avg_return:                             0.92688566
+    avg_reward_per_step:                        0.1269226
+    proxy_corner:
+      avg_avg_return:                           0.92688566
+      avg_reward_per_step:                      0.1269226
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.130806
+    avg_avg_return:                             0.92688566
+    avg_reward_per_step:                        0.1269226
+    proxy_corner:
+      avg_avg_return:                           0.92688566
+      avg_reward_per_step:                      0.1269226
+  perf:
+    env_steps_per_second:                       346706.67897212727
+    ppo_updates_per_second:                     42.3144074381614
+    eval-batch-orig-duration:                   0.510851813480258
+    eval-batch-shift-duration:                  0.48453014623373747
+    eval-batch-tree-duration:                   0.47220470290631056
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.000002
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.0
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     516.38306
+      avg_last_visit_time:                      1417.2185
+      num_replay_batches:                       1427
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00046207928
+    avg_actor_approxkl3:                        0.00047050903
+    avg_actor_clipfrac:                         0.011645508
+    avg_actor_loss:                             0.00087826606
+    avg_advantage:                              1.1775082e-05
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.7027289e-05
+    avg_entropy:                                0.034693748
+    avg_grad_norm_pre_clip:                     0.7659608
+    avg_loss:                                   0.0008520858
+    max:
+      max_actor_approxkl1:                      0.0024937468
+      max_actor_approxkl3:                      0.00315875
+      max_actor_clipfrac:                       0.036987305
+      max_actor_loss:                           0.007089818
+      max_advantage:                            0.0474782
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.0919953e-05
+      max_entropy:                              0.0469329
+      max_grad_norm_pre_clip:                   2.4875543
+      max_loss:                                 0.007065074
+    std:
+      std_actor_approxkl1:                      0.00066211337
+      std_actor_approxkl3:                      0.00071541965
+      std_actor_clipfrac:                       0.0095508145
+      std_actor_loss:                           0.0017337877
+      std_advantage:                            0.005363778
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.2335645e-06
+      std_entropy:                              0.004596428
+      std_grad_norm_pre_clip:                   0.5876164
+      std_loss:                                 0.001735864
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.043134
+    avg_avg_return:                             0.9735472
+    avg_benchmark_regret:                       0.016082259
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13347626
+    proxy_corner:
+      avg_avg_return:                           0.9735472
+      avg_reward_per_step:                      0.13347626
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.327675
+    avg_avg_return:                             0.042645317
+    avg_benchmark_regret:                       0.95012623
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0065689087
+    proxy_corner:
+      avg_avg_return:                           0.8963047
+      avg_reward_per_step:                      0.007293701
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.13987
+    avg_avg_return:                             0.11159793
+    avg_benchmark_regret:                       0.87247765
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.026046753
+    proxy_corner:
+      avg_avg_return:                           0.11159793
+      avg_reward_per_step:                      0.026046753
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        92274688
+    env-step-replay-before:                     47251456
+    env-step-used-before:                       92274688
+    ppo-update-before:                          56320
+    env-step-generate-before:                   45023232
+    env-step-all-after:                         92307456
+    env-step-replay-after:                      47251456
+    env-step-used-after:                        92307456
+    ppo-update-after:                           56340
+    env-step-generate-after:                    45056000
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.71809
+    avg_avg_return:                             0.927697
+    avg_reward_per_step:                        0.14907837
+    proxy_corner:
+      avg_avg_return:                           0.927697
+      avg_reward_per_step:                      0.14907837
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.71809
+    avg_avg_return:                             0.927697
+    avg_reward_per_step:                        0.14907837
+    proxy_corner:
+      avg_avg_return:                           0.927697
+      avg_reward_per_step:                      0.14907837
+  perf:
+    env_steps_per_second:                       314212.80370084185
+    ppo_updates_per_second:                     44.471192461456916
+    eval-batch-orig-duration:                   0.5140468748286366
+    eval-batch-shift-duration:                  0.47272935323417187
+    eval-batch-tree-duration:                   0.49203764367848635
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.0
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.000002
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030749
+    visit_patterns:
+      avg_first_visit_time:                     516.38306
+      avg_last_visit_time:                      1432.2427
+      num_replay_batches:                       1442
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00046637375
+    avg_actor_approxkl3:                        0.00038722492
+    avg_actor_clipfrac:                         0.011767578
+    avg_actor_loss:                             0.0001997026
+    avg_advantage:                              0.00016560103
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.6204221e-05
+    avg_entropy:                                0.026972104
+    avg_grad_norm_pre_clip:                     0.75479543
+    avg_loss:                                   0.0001858326
+    max:
+      max_actor_approxkl1:                      0.0020015682
+      max_actor_approxkl3:                      0.001190175
+      max_actor_clipfrac:                       0.029541016
+      max_actor_loss:                           0.0022078971
+      max_advantage:                            0.03171128
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.3063927e-05
+      max_entropy:                              0.03613988
+      max_grad_norm_pre_clip:                   1.5363785
+      max_loss:                                 0.0021962197
+    std:
+      std_actor_approxkl1:                      0.00046676828
+      std_actor_approxkl3:                      0.00029760206
+      std_actor_clipfrac:                       0.007851542
+      std_actor_loss:                           0.00076293416
+      std_advantage:                            0.0070095886
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.9340335e-06
+      std_entropy:                              0.0060194856
+      std_grad_norm_pre_clip:                   0.37142408
+      std_loss:                                 0.0007644146
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041927
+    avg_avg_return:                             0.97382027
+    avg_benchmark_regret:                       0.01580923
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349152
+    proxy_corner:
+      avg_avg_return:                           0.97382027
+      avg_reward_per_step:                      0.13349152
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.16221
+    avg_avg_return:                             0.045687802
+    avg_benchmark_regret:                       0.9470837
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0073013306
+    proxy_corner:
+      avg_avg_return:                           0.87734056
+      avg_reward_per_step:                      0.007118225
+  eval-batch-tree:
+    avg_avg_episode_length:                     80.621376
+    avg_avg_return:                             0.080564454
+    avg_benchmark_regret:                       0.90351117
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02570343
+    proxy_corner:
+      avg_avg_return:                           0.080564454
+      avg_reward_per_step:                      0.02570343
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        93323264
+    env-step-replay-before:                     47742976
+    env-step-used-before:                       93323264
+    ppo-update-before:                          56960
+    env-step-generate-before:                   45580288
+    env-step-all-after:                         93356032
+    env-step-replay-after:                      47742976
+    env-step-used-after:                        93356032
+    ppo-update-after:                           56980
+    env-step-generate-after:                    45613056
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.017965
+    avg_avg_return:                             0.9295762
+    avg_reward_per_step:                        0.13308716
+    proxy_corner:
+      avg_avg_return:                           0.9295762
+      avg_reward_per_step:                      0.13308716
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.017965
+    avg_avg_return:                             0.9295762
+    avg_reward_per_step:                        0.13308716
+    proxy_corner:
+      avg_avg_return:                           0.9295762
+      avg_reward_per_step:                      0.13308716
+  perf:
+    env_steps_per_second:                       320524.85872314795
+    ppo_updates_per_second:                     42.069205712263916
+    eval-batch-orig-duration:                   0.502321514301002
+    eval-batch-shift-duration:                  0.4597591310739517
+    eval-batch-tree-duration:                   0.4900182159617543
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.000004
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.000004
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030755
+    visit_patterns:
+      avg_first_visit_time:                     516.6035
+      avg_last_visit_time:                      1447.3811
+      num_replay_batches:                       1457
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0005990516
+    avg_actor_approxkl3:                        0.0005511603
+    avg_actor_clipfrac:                         0.007385254
+    avg_actor_loss:                             -3.96285e-05
+    avg_advantage:                              4.9759183e-06
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.4970572e-05
+    avg_entropy:                                0.01631133
+    avg_grad_norm_pre_clip:                     0.25437972
+    avg_loss:                                   -4.3454547e-05
+    max:
+      max_actor_approxkl1:                      0.0016222302
+      max_actor_approxkl3:                      0.0015286598
+      max_actor_clipfrac:                       0.020996094
+      max_actor_loss:                           0.00039148983
+      max_advantage:                            0.044258118
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.5625755e-05
+      max_entropy:                              0.020833801
+      max_grad_norm_pre_clip:                   0.66739
+      max_loss:                                 0.000382368
+    std:
+      std_actor_approxkl1:                      0.0005092184
+      std_actor_approxkl3:                      0.0004078725
+      std_actor_clipfrac:                       0.005952915
+      std_actor_loss:                           0.000248001
+      std_advantage:                            0.0072446493
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.2890687e-06
+      std_entropy:                              0.0028067038
+      std_grad_norm_pre_clip:                   0.1448533
+      std_loss:                                 0.00024868472
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.044546
+    avg_avg_return:                             0.97386825
+    avg_benchmark_regret:                       0.015761252
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13348389
+    proxy_corner:
+      avg_avg_return:                           0.97386825
+      avg_reward_per_step:                      0.13348389
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.21936
+    avg_avg_return:                             0.04334955
+    avg_benchmark_regret:                       0.949422
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.008262634
+    proxy_corner:
+      avg_avg_return:                           0.82966506
+      avg_reward_per_step:                      0.0067596436
+  eval-batch-tree:
+    avg_avg_episode_length:                     81.13724
+    avg_avg_return:                             0.057427686
+    avg_benchmark_regret:                       0.9266479
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025604248
+    proxy_corner:
+      avg_avg_return:                           0.057427686
+      avg_reward_per_step:                      0.025604248
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        94371840
+    env-step-replay-before:                     48267264
+    env-step-used-before:                       94371840
+    ppo-update-before:                          57600
+    env-step-generate-before:                   46104576
+    env-step-all-after:                         94404608
+    env-step-replay-after:                      48300032
+    env-step-used-after:                        94404608
+    ppo-update-after:                           57620
+    env-step-generate-after:                    46104576
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8418401
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8418401
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8418401
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8418401
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       242679.09988106476
+    ppo_updates_per_second:                     41.87104682395562
+    eval-batch-orig-duration:                   0.559554498642683
+    eval-batch-shift-duration:                  0.48652072809636593
+    eval-batch-tree-duration:                   0.48425340466201305
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.000002
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.000004
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030904
+    visit_patterns:
+      avg_first_visit_time:                     516.79175
+      avg_last_visit_time:                      1464.2424
+      num_replay_batches:                       1474
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        1.5972666e-05
+    avg_actor_approxkl3:                        6.1903243e-06
+    avg_actor_clipfrac:                         0.00017700196
+    avg_actor_loss:                             1.6806554e-05
+    avg_advantage:                              0.0026495038
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.1135618e-06
+    avg_entropy:                                0.0018316137
+    avg_grad_norm_pre_clip:                     0.0681107
+    avg_loss:                                   1.6031721e-05
+    max:
+      max_actor_approxkl1:                      0.00011013409
+      max_actor_approxkl3:                      2.8427745e-05
+      max_actor_clipfrac:                       0.00048828125
+      max_actor_loss:                           0.00015000068
+      max_advantage:                            0.018597543
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          5.685268e-06
+      max_entropy:                              0.0022244677
+      max_grad_norm_pre_clip:                   0.19954911
+      max_loss:                                 0.00015045471
+    std:
+      std_actor_approxkl1:                      3.281088e-05
+      std_actor_approxkl3:                      6.7342626e-06
+      std_actor_clipfrac:                       0.00014170149
+      std_actor_loss:                           4.6191646e-05
+      std_advantage:                            0.0020867735
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.3528976e-06
+      std_entropy:                              0.0002169898
+      std_grad_norm_pre_clip:                   0.052363742
+      std_loss:                                 4.6271536e-05
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.061617
+    avg_avg_return:                             0.9735955
+    avg_benchmark_regret:                       0.016034039
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13333893
+    proxy_corner:
+      avg_avg_return:                           0.9735955
+      avg_reward_per_step:                      0.13333893
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.35625
+    avg_avg_return:                             0.052767277
+    avg_benchmark_regret:                       0.9400042
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.012489319
+    proxy_corner:
+      avg_avg_return:                           0.7192688
+      avg_reward_per_step:                      0.005836487
+  eval-batch-tree:
+    avg_avg_episode_length:                     81.13661
+    avg_avg_return:                             0.05245497
+    avg_benchmark_regret:                       0.9316206
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02558136
+    proxy_corner:
+      avg_avg_return:                           0.05245497
+      avg_reward_per_step:                      0.02558136
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        95420416
+    env-step-replay-before:                     48889856
+    env-step-used-before:                       95420416
+    ppo-update-before:                          58240
+    env-step-generate-before:                   46530560
+    env-step-all-after:                         95453184
+    env-step-replay-after:                      48889856
+    env-step-used-after:                        95453184
+    ppo-update-after:                           58260
+    env-step-generate-after:                    46563328
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.065693
+    avg_avg_return:                             0.92697704
+    avg_reward_per_step:                        0.1378479
+    proxy_corner:
+      avg_avg_return:                           0.92697704
+      avg_reward_per_step:                      0.1378479
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.065693
+    avg_avg_return:                             0.92697704
+    avg_reward_per_step:                        0.1378479
+    proxy_corner:
+      avg_avg_return:                           0.92697704
+      avg_reward_per_step:                      0.1378479
+  perf:
+    env_steps_per_second:                       242786.79757541194
+    ppo_updates_per_second:                     42.402433011094566
+    eval-batch-orig-duration:                   0.6624909238889813
+    eval-batch-shift-duration:                  0.6186369080096483
+    eval-batch-tree-duration:                   0.6168292630463839
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.0
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.000004
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030805
+    visit_patterns:
+      avg_first_visit_time:                     516.79175
+      avg_last_visit_time:                      1482.238
+      num_replay_batches:                       1492
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0008298183
+    avg_actor_approxkl3:                        0.00081461575
+    avg_actor_clipfrac:                         0.017352296
+    avg_actor_loss:                             0.0006914935
+    avg_advantage:                              0.0022934813
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.7960996e-05
+    avg_entropy:                                0.033919483
+    avg_grad_norm_pre_clip:                     0.95520574
+    avg_loss:                                   0.00067155453
+    max:
+      max_actor_approxkl1:                      0.002992245
+      max_actor_approxkl3:                      0.0020435138
+      max_actor_clipfrac:                       0.044677734
+      max_actor_loss:                           0.0044945814
+      max_advantage:                            0.03660947
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.9158273e-05
+      max_entropy:                              0.042138673
+      max_grad_norm_pre_clip:                   2.9659727
+      max_loss:                                 0.0044659576
+    std:
+      std_actor_approxkl1:                      0.00076602807
+      std_actor_approxkl3:                      0.0005900086
+      std_actor_clipfrac:                       0.009590585
+      std_actor_loss:                           0.0013247078
+      std_advantage:                            0.007667317
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.530802e-06
+      std_entropy:                              0.0050323796
+      std_grad_norm_pre_clip:                   0.66198045
+      std_loss:                                 0.0013232217
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041449
+    avg_avg_return:                             0.9736531
+    avg_benchmark_regret:                       0.015976481
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13347626
+    proxy_corner:
+      avg_avg_return:                           0.9736531
+      avg_reward_per_step:                      0.13347626
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.30353
+    avg_avg_return:                             0.04327382
+    avg_benchmark_regret:                       0.9494977
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.007446289
+    proxy_corner:
+      avg_avg_return:                           0.81722546
+      avg_reward_per_step:                      0.0066452026
+  eval-batch-tree:
+    avg_avg_episode_length:                     81.101776
+    avg_avg_return:                             0.056649026
+    avg_benchmark_regret:                       0.9274266
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025520325
+    proxy_corner:
+      avg_avg_return:                           0.056649026
+      avg_reward_per_step:                      0.025520325
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        96468992
+    env-step-replay-before:                     49381376
+    env-step-used-before:                       96468992
+    ppo-update-before:                          58880
+    env-step-generate-before:                   47087616
+    env-step-all-after:                         96501760
+    env-step-replay-after:                      49414144
+    env-step-used-after:                        96501760
+    ppo-update-after:                           58900
+    env-step-generate-after:                    47087616
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       339968.8207598474
+    ppo_updates_per_second:                     42.20848022880767
+    eval-batch-orig-duration:                   0.5080584222450852
+    eval-batch-shift-duration:                  0.4287513568997383
+    eval-batch-tree-duration:                   0.4511179216206074
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.0
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.000002
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     516.79175
+      avg_last_visit_time:                      1498.313
+      num_replay_batches:                       1508
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        1.1801221e-06
+    avg_actor_approxkl3:                        -1.3402238e-09
+    avg_actor_clipfrac:                         0.0
+    avg_actor_loss:                             1.4757737e-06
+    avg_advantage:                              -0.00086455676
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.8472513e-06
+    avg_entropy:                                0.0003801695
+    avg_grad_norm_pre_clip:                     0.009246012
+    avg_loss:                                   2.0192301e-06
+    max:
+      max_actor_approxkl1:                      3.0992437e-06
+      max_actor_approxkl3:                      1.023227e-08
+      max_actor_clipfrac:                       0.0
+      max_actor_loss:                           4.976988e-06
+      max_advantage:                            0.003555715
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.619293e-06
+      max_entropy:                              0.00041825423
+      max_grad_norm_pre_clip:                   0.014927053
+      max_loss:                                 6.0976713e-06
+    std:
+      std_actor_approxkl1:                      1.505681e-06
+      std_actor_approxkl3:                      4.6061186e-09
+      std_actor_clipfrac:                       0.0
+      std_actor_loss:                           2.7190902e-06
+      std_advantage:                            0.002030513
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.9800736e-07
+      std_entropy:                              1.9501877e-05
+      std_grad_norm_pre_clip:                   0.0028424147
+      std_loss:                                 2.957574e-06
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040743
+    avg_avg_return:                             0.9738282
+    avg_benchmark_regret:                       0.015801305
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349915
+    proxy_corner:
+      avg_avg_return:                           0.9738282
+      avg_reward_per_step:                      0.13349915
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.53295
+    avg_avg_return:                             0.050834317
+    avg_benchmark_regret:                       0.94193715
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.011230469
+    proxy_corner:
+      avg_avg_return:                           0.7560437
+      avg_reward_per_step:                      0.0060806274
+  eval-batch-tree:
+    avg_avg_episode_length:                     81.053154
+    avg_avg_return:                             0.05676849
+    avg_benchmark_regret:                       0.9273071
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025604248
+    proxy_corner:
+      avg_avg_return:                           0.05676849
+      avg_reward_per_step:                      0.025604248
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        97517568
+    env-step-replay-before:                     49840128
+    env-step-used-before:                       97517568
+    ppo-update-before:                          59520
+    env-step-generate-before:                   47677440
+    env-step-all-after:                         97550336
+    env-step-replay-after:                      49872896
+    env-step-used-after:                        97550336
+    ppo-update-after:                           59540
+    env-step-generate-after:                    47677440
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8418433
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8418433
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8418433
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8418433
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       295241.507153959
+    ppo_updates_per_second:                     42.202128676643795
+    eval-batch-orig-duration:                   0.519199950620532
+    eval-batch-shift-duration:                  0.48418512381613255
+    eval-batch-tree-duration:                   0.44066044967621565
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999998
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          18.999996
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030755
+    visit_patterns:
+      avg_first_visit_time:                     516.79175
+      avg_last_visit_time:                      1512.3035
+      num_replay_batches:                       1522
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        -4.7023036e-07
+    avg_actor_approxkl3:                        7.1652875e-07
+    avg_actor_clipfrac:                         4.8828126e-05
+    avg_actor_loss:                             -4.511699e-06
+    avg_advantage:                              -0.00042758975
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            3.945393e-06
+    avg_entropy:                                0.0012511521
+    avg_grad_norm_pre_clip:                     0.020388773
+    avg_loss:                                   -3.7901548e-06
+    max:
+      max_actor_approxkl1:                      2.8581424e-05
+      max_actor_approxkl3:                      3.3744523e-06
+      max_actor_clipfrac:                       0.00024414062
+      max_actor_loss:                           1.1052936e-05
+      max_advantage:                            0.012527049
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          7.7311615e-06
+      max_entropy:                              0.0013094489
+      max_grad_norm_pre_clip:                   0.043849215
+      max_loss:                                 1.3103e-05
+    std:
+      std_actor_approxkl1:                      1.5649472e-05
+      std_actor_approxkl3:                      8.9736756e-07
+      std_actor_clipfrac:                       7.117861e-05
+      std_actor_loss:                           1.2231851e-05
+      std_advantage:                            0.003798844
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.916488e-06
+      std_entropy:                              3.779897e-05
+      std_grad_norm_pre_clip:                   0.011144632
+      std_loss:                                 1.26873265e-05
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041101
+    avg_avg_return:                             0.9737885
+    avg_benchmark_regret:                       0.015841002
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13347626
+    proxy_corner:
+      avg_avg_return:                           0.9737885
+      avg_reward_per_step:                      0.13347626
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.35898
+    avg_avg_return:                             0.05321672
+    avg_benchmark_regret:                       0.9395548
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.008605957
+    proxy_corner:
+      avg_avg_return:                           0.81424296
+      avg_reward_per_step:                      0.0065994263
+  eval-batch-tree:
+    avg_avg_episode_length:                     80.84459
+    avg_avg_return:                             0.06944238
+    avg_benchmark_regret:                       0.9146333
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025505066
+    proxy_corner:
+      avg_avg_return:                           0.06944238
+      avg_reward_per_step:                      0.025505066
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        98566144
+    env-step-replay-before:                     50397184
+    env-step-used-before:                       98566144
+    ppo-update-before:                          60160
+    env-step-generate-before:                   48168960
+    env-step-all-after:                         98598912
+    env-step-replay-after:                      50429952
+    env-step-used-after:                        98598912
+    ppo-update-after:                           60180
+    env-step-generate-after:                    48168960
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184444
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184444
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184444
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184444
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       316295.9724877914
+    ppo_updates_per_second:                     41.00827143037499
+    eval-batch-orig-duration:                   0.4998532859608531
+    eval-batch-shift-duration:                  0.43236831296235323
+    eval-batch-tree-duration:                   0.40021831914782524
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.0
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.0
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030749
+    visit_patterns:
+      avg_first_visit_time:                     516.79175
+      avg_last_visit_time:                      1529.2312
+      num_replay_batches:                       1539
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        -3.6579713e-06
+    avg_actor_approxkl3:                        1.8805624e-05
+    avg_actor_clipfrac:                         0.00012817384
+    avg_actor_loss:                             1.3551303e-05
+    avg_advantage:                              -5.6637437e-05
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.5970911e-06
+    avg_entropy:                                0.0013762871
+    avg_grad_norm_pre_clip:                     0.051366385
+    avg_loss:                                   1.2973559e-05
+    max:
+      max_actor_approxkl1:                      6.4505024e-05
+      max_actor_approxkl3:                      7.684244e-05
+      max_actor_clipfrac:                       0.00024414062
+      max_actor_loss:                           0.0002959855
+      max_advantage:                            0.02016002
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.2702307e-06
+      max_entropy:                              0.001595552
+      max_grad_norm_pre_clip:                   0.17561083
+      max_loss:                                 0.000295187
+    std:
+      std_actor_approxkl1:                      3.0312429e-05
+      std_actor_approxkl3:                      1.8248345e-05
+      std_actor_clipfrac:                       8.165946e-05
+      std_actor_loss:                           8.896714e-05
+      std_advantage:                            0.0016779072
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.4278872e-07
+      std_entropy:                              0.0001551317
+      std_grad_norm_pre_clip:                   0.043339185
+      std_loss:                                 8.89498e-05
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041212
+    avg_avg_return:                             0.9736536
+    avg_benchmark_regret:                       0.015975907
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13347626
+    proxy_corner:
+      avg_avg_return:                           0.9736536
+      avg_reward_per_step:                      0.13347626
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.00583
+    avg_avg_return:                             0.058008134
+    avg_benchmark_regret:                       0.9347634
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.009353638
+    proxy_corner:
+      avg_avg_return:                           0.81564695
+      avg_reward_per_step:                      0.0066833496
+  eval-batch-tree:
+    avg_avg_episode_length:                     80.236824
+    avg_avg_return:                             0.08674605
+    avg_benchmark_regret:                       0.8973295
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02571106
+    proxy_corner:
+      avg_avg_return:                           0.08674605
+      avg_reward_per_step:                      0.02571106
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        99614720
+    env-step-replay-before:                     51052544
+    env-step-used-before:                       99614720
+    ppo-update-before:                          60800
+    env-step-generate-before:                   48562176
+    env-step-all-after:                         99647488
+    env-step-replay-after:                      51085312
+    env-step-used-after:                        99647488
+    ppo-update-after:                           60820
+    env-step-generate-after:                    48562176
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       311072.6537832105
+    ppo_updates_per_second:                     42.25093776744211
+    eval-batch-orig-duration:                   0.5405755201354623
+    eval-batch-shift-duration:                  0.4885834902524948
+    eval-batch-tree-duration:                   0.48915123473852873
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.000002
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.000004
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     516.79175
+      avg_last_visit_time:                      1549.2339
+      num_replay_batches:                       1559
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        1.51779695e-05
+    avg_actor_approxkl3:                        6.0617407e-05
+    avg_actor_clipfrac:                         0.0002685547
+    avg_actor_loss:                             3.182702e-06
+    avg_advantage:                              0.00044929597
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.4539837e-06
+    avg_entropy:                                0.0020980665
+    avg_grad_norm_pre_clip:                     0.08348812
+    avg_loss:                                   1.8116243e-06
+    max:
+      max_actor_approxkl1:                      0.00022400699
+      max_actor_approxkl3:                      0.0002876747
+      max_actor_clipfrac:                       0.0008544922
+      max_actor_loss:                           0.00084136426
+      max_advantage:                            0.015701711
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.365688e-06
+      max_entropy:                              0.002587965
+      max_grad_norm_pre_clip:                   0.22829795
+      max_loss:                                 0.00084014994
+    std:
+      std_actor_approxkl1:                      8.6848995e-05
+      std_actor_approxkl3:                      7.29453e-05
+      std_actor_clipfrac:                       0.00024291685
+      std_actor_loss:                           0.00028852702
+      std_advantage:                            0.0014604965
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.80399e-07
+      std_entropy:                              0.00035155966
+      std_grad_norm_pre_clip:                   0.06338582
+      std_loss:                                 0.00028855415
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.42156
+    avg_avg_return:                             0.9723075
+    avg_benchmark_regret:                       0.01732203
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.1207428
+    proxy_corner:
+      avg_avg_return:                           0.9723075
+      avg_reward_per_step:                      0.1207428
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.13972
+    avg_avg_return:                             0.060617555
+    avg_benchmark_regret:                       0.93215394
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.00843811
+    proxy_corner:
+      avg_avg_return:                           0.86236537
+      avg_reward_per_step:                      0.0071411133
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.64399
+    avg_avg_return:                             0.102468036
+    avg_benchmark_regret:                       0.8816076
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025558472
+    proxy_corner:
+      avg_avg_return:                           0.102468036
+      avg_reward_per_step:                      0.025558472
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        100663296
+    env-step-replay-before:                     51675136
+    env-step-used-before:                       100663296
+    ppo-update-before:                          61440
+    env-step-generate-before:                   48988160
+    env-step-all-after:                         100696064
+    env-step-replay-after:                      51675136
+    env-step-used-after:                        100696064
+    ppo-update-after:                           61460
+    env-step-generate-after:                    49020928
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     12.094727
+    avg_avg_return:                             0.9118389
+    avg_reward_per_step:                        0.10144043
+    proxy_corner:
+      avg_avg_return:                           0.9118389
+      avg_reward_per_step:                      0.10144043
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     12.094727
+    avg_avg_return:                             0.9118389
+    avg_reward_per_step:                        0.10144043
+    proxy_corner:
+      avg_avg_return:                           0.9118389
+      avg_reward_per_step:                      0.10144043
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  perf:
+    env_steps_per_second:                       235730.8885992518
+    ppo_updates_per_second:                     42.162004582808216
+    eval-batch-orig-duration:                   0.6651998898014426
+    eval-batch-shift-duration:                  0.5999947069212794
+    eval-batch-tree-duration:                   0.6280277641490102
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.039062
+      mouse-cheese_dist_finite_wavg:            18.922655
+      mouse-cheese_dist_solvable_avg:           18.039062
+      mouse-cheese_dist_solvable_wavg:          18.922655
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.22120374
+    visit_patterns:
+      avg_first_visit_time:                     621.33057
+      avg_last_visit_time:                      1567.2571
+      num_replay_batches:                       1577
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0003245397
+    avg_actor_approxkl3:                        0.0006515849
+    avg_actor_clipfrac:                         0.030895997
+    avg_actor_loss:                             -0.00056629314
+    avg_advantage:                              0.00021947383
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00013435954
+    avg_entropy:                                0.15501198
+    avg_grad_norm_pre_clip:                     1.0019294
+    avg_loss:                                   -0.00065412535
+    max:
+      max_actor_approxkl1:                      0.0011880563
+      max_actor_approxkl3:                      0.0013136674
+      max_actor_clipfrac:                       0.057495117
+      max_actor_loss:                           0.0014805282
+      max_advantage:                            0.08691448
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00016427075
+      max_entropy:                              0.21938092
+      max_grad_norm_pre_clip:                   2.5102203
+      max_loss:                                 0.001361497
+    std:
+      std_actor_approxkl1:                      0.00037379123
+      std_actor_approxkl3:                      0.00030575905
+      std_actor_clipfrac:                       0.015712773
+      std_actor_loss:                           0.00090438186
+      std_advantage:                            0.016942427
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.8603714e-05
+      std_entropy:                              0.030529859
+      std_grad_norm_pre_clip:                   0.6172108
+      std_loss:                                 0.00090544124
+  eval-batch-orig:
+    avg_avg_episode_length:                     11.316671
+    avg_avg_return:                             0.9694184
+    avg_benchmark_regret:                       0.020211093
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.11342621
+    proxy_corner:
+      avg_avg_return:                           0.9694184
+      avg_reward_per_step:                      0.11342621
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.818954
+    avg_avg_return:                             0.050650164
+    avg_benchmark_regret:                       0.9421214
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.009315491
+    proxy_corner:
+      avg_avg_return:                           0.8989713
+      avg_reward_per_step:                      0.007331848
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-tree:
+    avg_avg_episode_length:                     80.32868
+    avg_avg_return:                             0.08503652
+    avg_benchmark_regret:                       0.89903903
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025794983
+    proxy_corner:
+      avg_avg_return:                           0.08503652
+      avg_reward_per_step:                      0.025794983
+    rollouts_gif:                               float32(132, 225, 225, 3)
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        101711872
+    env-step-replay-before:                     52101120
+    env-step-used-before:                       101711872
+    ppo-update-before:                          62080
+    env-step-generate-before:                   49610752
+    env-step-all-after:                         101744640
+    env-step-replay-after:                      52133888
+    env-step-used-after:                        101744640
+    ppo-update-after:                           62100
+    env-step-generate-after:                    49610752
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.677086
+    avg_avg_return:                             0.8500483
+    avg_reward_per_step:                        0.052825928
+    proxy_corner:
+      avg_avg_return:                           0.8500483
+      avg_reward_per_step:                      0.052825928
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.677086
+    avg_avg_return:                             0.8500483
+    avg_reward_per_step:                        0.052825928
+    proxy_corner:
+      avg_avg_return:                           0.8500483
+      avg_reward_per_step:                      0.052825928
+  perf:
+    env_steps_per_second:                       320748.9687153822
+    ppo_updates_per_second:                     42.40191359170568
+    eval-batch-orig-duration:                   0.5278326729312539
+    eval-batch-shift-duration:                  0.4862608453258872
+    eval-batch-tree-duration:                   0.4586260914802551
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.329102
+      mouse-cheese_dist_finite_wavg:            18.918125
+      mouse-cheese_dist_solvable_avg:           18.329102
+      mouse-cheese_dist_solvable_wavg:          18.918125
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.15386769
+    visit_patterns:
+      avg_first_visit_time:                     622.47974
+      avg_last_visit_time:                      1581.2205
+      num_replay_batches:                       1591
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00046124888
+    avg_actor_approxkl3:                        0.00039229193
+    avg_actor_clipfrac:                         0.019195557
+    avg_actor_loss:                             0.00079726
+    avg_advantage:                              -0.00046755618
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.468295e-05
+    avg_entropy:                                0.040278196
+    avg_grad_norm_pre_clip:                     0.8801404
+    avg_loss:                                   0.00076932326
+    max:
+      max_actor_approxkl1:                      0.0014423484
+      max_actor_approxkl3:                      0.0011017539
+      max_actor_clipfrac:                       0.04711914
+      max_actor_loss:                           0.003802652
+      max_advantage:                            0.05340624
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          4.8607224e-05
+      max_entropy:                              0.048163503
+      max_grad_norm_pre_clip:                   2.2776513
+      max_loss:                                 0.0037840095
+    std:
+      std_actor_approxkl1:                      0.0004325223
+      std_actor_approxkl3:                      0.00031261108
+      std_actor_clipfrac:                       0.01859864
+      std_actor_loss:                           0.0012307139
+      std_advantage:                            0.0053679524
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          9.0982285e-06
+      std_entropy:                              0.0040620156
+      std_grad_norm_pre_clip:                   0.7885273
+      std_loss:                                 0.001232709
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.079067
+    avg_avg_return:                             0.97309065
+    avg_benchmark_regret:                       0.016538871
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13249207
+    proxy_corner:
+      avg_avg_return:                           0.97309065
+      avg_reward_per_step:                      0.13249207
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.44487
+    avg_avg_return:                             0.053228647
+    avg_benchmark_regret:                       0.9395429
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.010864258
+    proxy_corner:
+      avg_avg_return:                           0.88362
+      avg_reward_per_step:                      0.0072402954
+  eval-batch-tree:
+    avg_avg_episode_length:                     80.800865
+    avg_avg_return:                             0.07147655
+    avg_benchmark_regret:                       0.9125991
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025627136
+    proxy_corner:
+      avg_avg_return:                           0.07147655
+      avg_reward_per_step:                      0.025627136
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        102760448
+    env-step-replay-before:                     52625408
+    env-step-used-before:                       102760448
+    ppo-update-before:                          62720
+    env-step-generate-before:                   50135040
+    env-step-all-after:                         102793216
+    env-step-replay-after:                      52625408
+    env-step-used-after:                        102793216
+    ppo-update-after:                           62740
+    env-step-generate-after:                    50167808
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.449589
+    avg_avg_return:                             0.9229785
+    avg_reward_per_step:                        0.13378906
+    proxy_corner:
+      avg_avg_return:                           0.9229785
+      avg_reward_per_step:                      0.13378906
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.449589
+    avg_avg_return:                             0.9229785
+    avg_reward_per_step:                        0.13378906
+    proxy_corner:
+      avg_avg_return:                           0.9229785
+      avg_reward_per_step:                      0.13378906
+  perf:
+    env_steps_per_second:                       295423.77665070194
+    ppo_updates_per_second:                     42.30008977023937
+    eval-batch-orig-duration:                   0.5247079962864518
+    eval-batch-shift-duration:                  0.48945844545960426
+    eval-batch-tree-duration:                   0.4835628690198064
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.733398
+      mouse-cheese_dist_finite_wavg:            8.162751
+      mouse-cheese_dist_solvable_avg:           18.733398
+      mouse-cheese_dist_solvable_wavg:          8.162752
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13831663
+    visit_patterns:
+      avg_first_visit_time:                     623.74805
+      avg_last_visit_time:                      1596.4998
+      num_replay_batches:                       1606
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00037900714
+    avg_actor_approxkl3:                        0.00042382543
+    avg_actor_clipfrac:                         0.015826417
+    avg_actor_loss:                             0.0006119803
+    avg_advantage:                              0.0010751027
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            3.929344e-05
+    avg_entropy:                                0.07117044
+    avg_grad_norm_pre_clip:                     0.7376336
+    avg_loss:                                   0.00056045654
+    max:
+      max_actor_approxkl1:                      0.0014638514
+      max_actor_approxkl3:                      0.0009866946
+      max_actor_clipfrac:                       0.045166016
+      max_actor_loss:                           0.0024589943
+      max_advantage:                            0.06325078
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          5.20595e-05
+      max_entropy:                              0.08405958
+      max_grad_norm_pre_clip:                   1.3740815
+      max_loss:                                 0.0024145918
+    std:
+      std_actor_approxkl1:                      0.00044864943
+      std_actor_approxkl3:                      0.00030289392
+      std_actor_clipfrac:                       0.012522729
+      std_actor_loss:                           0.0008797422
+      std_advantage:                            0.009077975
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.870938e-06
+      std_entropy:                              0.0070350324
+      std_grad_norm_pre_clip:                   0.41488093
+      std_loss:                                 0.0008785259
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.044699
+    avg_avg_return:                             0.9737345
+    avg_benchmark_regret:                       0.01589502
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.1333847
+    proxy_corner:
+      avg_avg_return:                           0.9737345
+      avg_reward_per_step:                      0.1333847
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.91222
+    avg_avg_return:                             0.04738433
+    avg_benchmark_regret:                       0.9453872
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.008102417
+    proxy_corner:
+      avg_avg_return:                           0.8643821
+      avg_reward_per_step:                      0.00705719
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.28056
+    avg_avg_return:                             0.11110147
+    avg_benchmark_regret:                       0.87297416
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.026062012
+    proxy_corner:
+      avg_avg_return:                           0.11110147
+      avg_reward_per_step:                      0.026062012
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        103809024
+    env-step-replay-before:                     53149696
+    env-step-used-before:                       103809024
+    ppo-update-before:                          63360
+    env-step-generate-before:                   50659328
+    env-step-all-after:                         103841792
+    env-step-replay-after:                      53182464
+    env-step-used-after:                        103841792
+    ppo-update-after:                           63380
+    env-step-generate-after:                    50659328
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.545702
+    avg_avg_return:                             0.84333783
+    avg_reward_per_step:                        0.047332764
+    proxy_corner:
+      avg_avg_return:                           0.84333783
+      avg_reward_per_step:                      0.047332764
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.545702
+    avg_avg_return:                             0.84333783
+    avg_reward_per_step:                        0.047332764
+    proxy_corner:
+      avg_avg_return:                           0.84333783
+      avg_reward_per_step:                      0.047332764
+  perf:
+    env_steps_per_second:                       320372.4228916423
+    ppo_updates_per_second:                     42.18444526674564
+    eval-batch-orig-duration:                   0.5140800541266799
+    eval-batch-shift-duration:                  0.45016533229500055
+    eval-batch-tree-duration:                   0.22050974983721972
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.865234
+      mouse-cheese_dist_finite_wavg:            18.982677
+      mouse-cheese_dist_solvable_avg:           18.865234
+      mouse-cheese_dist_solvable_wavg:          18.98268
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13891885
+    visit_patterns:
+      avg_first_visit_time:                     624.4885
+      avg_last_visit_time:                      1613.1992
+      num_replay_batches:                       1623
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00022622391
+    avg_actor_approxkl3:                        0.00018629276
+    avg_actor_clipfrac:                         0.0037597658
+    avg_actor_loss:                             0.00029936977
+    avg_advantage:                              0.0013991417
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.366348e-05
+    avg_entropy:                                0.01972931
+    avg_grad_norm_pre_clip:                     0.40877852
+    avg_loss:                                   0.00028647223
+    max:
+      max_actor_approxkl1:                      0.0005585328
+      max_actor_approxkl3:                      0.0008941155
+      max_actor_clipfrac:                       0.008422852
+      max_actor_loss:                           0.002708882
+      max_advantage:                            0.03567928
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.0712345e-05
+      max_entropy:                              0.02584557
+      max_grad_norm_pre_clip:                   1.0268928
+      max_loss:                                 0.002686989
+    std:
+      std_actor_approxkl1:                      0.00016915315
+      std_actor_approxkl3:                      0.00020747472
+      std_actor_clipfrac:                       0.0021331238
+      std_actor_loss:                           0.0007467582
+      std_advantage:                            0.0047387453
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.3776863e-06
+      std_entropy:                              0.0031141907
+      std_grad_norm_pre_clip:                   0.26683113
+      std_loss:                                 0.0007447246
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.0499
+    avg_avg_return:                             0.97358096
+    avg_benchmark_regret:                       0.01604852
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13337708
+    proxy_corner:
+      avg_avg_return:                           0.97358096
+      avg_reward_per_step:                      0.13337708
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.7247
+    avg_avg_return:                             0.05110943
+    avg_benchmark_regret:                       0.94166213
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.009162903
+    proxy_corner:
+      avg_avg_return:                           0.8322375
+      avg_reward_per_step:                      0.00680542
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.26216
+    avg_avg_return:                             0.1091651
+    avg_benchmark_regret:                       0.8749105
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.0260849
+    proxy_corner:
+      avg_avg_return:                           0.1091651
+      avg_reward_per_step:                      0.0260849
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        104857600
+    env-step-replay-before:                     53772288
+    env-step-used-before:                       104857600
+    ppo-update-before:                          64000
+    env-step-generate-before:                   51085312
+    env-step-all-after:                         104890368
+    env-step-replay-after:                      53772288
+    env-step-used-after:                        104890368
+    ppo-update-after:                           64020
+    env-step-generate-after:                    51118080
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.302767
+    avg_avg_return:                             0.92656255
+    avg_reward_per_step:                        0.13671875
+    proxy_corner:
+      avg_avg_return:                           0.92656255
+      avg_reward_per_step:                      0.13671875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.302767
+    avg_avg_return:                             0.92656255
+    avg_reward_per_step:                        0.13671875
+    proxy_corner:
+      avg_avg_return:                           0.92656255
+      avg_reward_per_step:                      0.13671875
+  perf:
+    env_steps_per_second:                       262522.3049178203
+    ppo_updates_per_second:                     42.255298734781526
+    eval-batch-orig-duration:                   0.6111975936219096
+    eval-batch-shift-duration:                  0.572127022780478
+    eval-batch-tree-duration:                   0.5482730958610773
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.927246
+      mouse-cheese_dist_finite_wavg:            18.991423
+      mouse-cheese_dist_solvable_avg:           18.927246
+      mouse-cheese_dist_solvable_wavg:          18.99143
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999964
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.13936369
+    visit_patterns:
+      avg_first_visit_time:                     625.24585
+      avg_last_visit_time:                      1631.3601
+      num_replay_batches:                       1641
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00066875067
+    avg_actor_approxkl3:                        0.0005145604
+    avg_actor_clipfrac:                         0.009881592
+    avg_actor_loss:                             0.00011863746
+    avg_advantage:                              0.000310786
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.3071115e-05
+    avg_entropy:                                0.036041684
+    avg_grad_norm_pre_clip:                     0.46269593
+    avg_loss:                                   0.000104131344
+    max:
+      max_actor_approxkl1:                      0.0028541521
+      max_actor_approxkl3:                      0.001504826
+      max_actor_clipfrac:                       0.020019531
+      max_actor_loss:                           0.002339148
+      max_advantage:                            0.058043897
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          5.228097e-05
+      max_entropy:                              0.04528491
+      max_grad_norm_pre_clip:                   0.98296094
+      max_loss:                                 0.0023251297
+    std:
+      std_actor_approxkl1:                      0.0007391637
+      std_actor_approxkl3:                      0.00047382552
+      std_actor_clipfrac:                       0.0058958167
+      std_actor_loss:                           0.000821839
+      std_advantage:                            0.009184004
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.6650684e-06
+      std_entropy:                              0.005125716
+      std_grad_norm_pre_clip:                   0.25063685
+      std_loss:                                 0.00082286936
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.042083
+    avg_avg_return:                             0.9739302
+    avg_benchmark_regret:                       0.015699372
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13346863
+    proxy_corner:
+      avg_avg_return:                           0.9739302
+      avg_reward_per_step:                      0.13346863
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.59793
+    avg_avg_return:                             0.04934007
+    avg_benchmark_regret:                       0.94343144
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.011627197
+    proxy_corner:
+      avg_avg_return:                           0.78706133
+      avg_reward_per_step:                      0.0063552856
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.92059
+    avg_avg_return:                             0.0981196
+    avg_benchmark_regret:                       0.885956
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025901794
+    proxy_corner:
+      avg_avg_return:                           0.0981196
+      avg_reward_per_step:                      0.025901794
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        105906176
+    env-step-replay-before:                     54198272
+    env-step-used-before:                       105906176
+    ppo-update-before:                          64640
+    env-step-generate-before:                   51707904
+    env-step-all-after:                         105938944
+    env-step-replay-after:                      54198272
+    env-step-used-after:                        105938944
+    ppo-update-after:                           64660
+    env-step-generate-after:                    51740672
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.017595
+    avg_avg_return:                             0.9237747
+    avg_reward_per_step:                        0.13085938
+    proxy_corner:
+      avg_avg_return:                           0.9237747
+      avg_reward_per_step:                      0.13085938
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.017595
+    avg_avg_return:                             0.9237747
+    avg_reward_per_step:                        0.13085938
+    proxy_corner:
+      avg_avg_return:                           0.9237747
+      avg_reward_per_step:                      0.13085938
+  perf:
+    env_steps_per_second:                       376903.20865920396
+    ppo_updates_per_second:                     42.32203005947291
+    eval-batch-orig-duration:                   0.49452049657702446
+    eval-batch-shift-duration:                  0.42965830583125353
+    eval-batch-tree-duration:                   0.4490619385614991
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.918701
+      mouse-cheese_dist_finite_wavg:            18.991955
+      mouse-cheese_dist_solvable_avg:           18.918701
+      mouse-cheese_dist_solvable_wavg:          18.991957
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1402129
+    visit_patterns:
+      avg_first_visit_time:                     652.55396
+      avg_last_visit_time:                      1644.3164
+      num_replay_batches:                       1654
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00010226556
+    avg_actor_approxkl3:                        0.00011585888
+    avg_actor_clipfrac:                         0.0037353516
+    avg_actor_loss:                             -0.00015063897
+    avg_advantage:                              -0.0016836999
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            3.505571e-05
+    avg_entropy:                                0.038174164
+    avg_grad_norm_pre_clip:                     0.20577411
+    avg_loss:                                   -0.00017128527
+    max:
+      max_actor_approxkl1:                      0.00038137104
+      max_actor_approxkl3:                      0.00033471582
+      max_actor_clipfrac:                       0.009033203
+      max_actor_loss:                           0.00031224452
+      max_advantage:                            0.07723069
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          4.396977e-05
+      max_entropy:                              0.044924997
+      max_grad_norm_pre_clip:                   0.53554505
+      max_loss:                                 0.00029303937
+    std:
+      std_actor_approxkl1:                      0.00012661221
+      std_actor_approxkl3:                      7.8584046e-05
+      std_actor_clipfrac:                       0.0021621955
+      std_actor_loss:                           0.00020228565
+      std_advantage:                            0.008678122
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.9486957e-06
+      std_entropy:                              0.0031325708
+      std_grad_norm_pre_clip:                   0.12358672
+      std_loss:                                 0.00020374559
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.070383
+    avg_avg_return:                             0.9734116
+    avg_benchmark_regret:                       0.016217897
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13320923
+    proxy_corner:
+      avg_avg_return:                           0.9734116
+      avg_reward_per_step:                      0.13320923
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.45575
+    avg_avg_return:                             0.049996838
+    avg_benchmark_regret:                       0.94277465
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.011703491
+    proxy_corner:
+      avg_avg_return:                           0.8050016
+      avg_reward_per_step:                      0.006500244
+  eval-batch-tree:
+    avg_avg_episode_length:                     78.96314
+    avg_avg_return:                             0.112728104
+    avg_benchmark_regret:                       0.8713475
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.026153564
+    proxy_corner:
+      avg_avg_return:                           0.112728104
+      avg_reward_per_step:                      0.026153564
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        106954752
+    env-step-replay-before:                     54624256
+    env-step-used-before:                       106954752
+    ppo-update-before:                          65280
+    env-step-generate-before:                   52330496
+    env-step-all-after:                         106987520
+    env-step-replay-after:                      54624256
+    env-step-used-after:                        106987520
+    ppo-update-after:                           65300
+    env-step-generate-after:                    52363264
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.685523
+    avg_avg_return:                             0.9230739
+    avg_reward_per_step:                        0.12390137
+    proxy_corner:
+      avg_avg_return:                           0.9230739
+      avg_reward_per_step:                      0.12390137
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.685523
+    avg_avg_return:                             0.9230739
+    avg_reward_per_step:                        0.12390137
+    proxy_corner:
+      avg_avg_return:                           0.9230739
+      avg_reward_per_step:                      0.12390137
+  perf:
+    env_steps_per_second:                       320602.6034054973
+    ppo_updates_per_second:                     42.31057969275113
+    eval-batch-orig-duration:                   0.49675558786839247
+    eval-batch-shift-duration:                  0.4955022893846035
+    eval-batch-tree-duration:                   0.4822005843743682
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.96875
+      mouse-cheese_dist_finite_wavg:            19.893835
+      mouse-cheese_dist_solvable_avg:           18.96875
+      mouse-cheese_dist_solvable_wavg:          19.893835
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14007601
+    visit_patterns:
+      avg_first_visit_time:                     653.322
+      avg_last_visit_time:                      1657.2468
+      num_replay_batches:                       1667
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        7.954241e-05
+    avg_actor_approxkl3:                        0.00014072453
+    avg_actor_clipfrac:                         0.006713867
+    avg_actor_loss:                             -0.00025652468
+    avg_advantage:                              0.0015279506
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.7440188e-05
+    avg_entropy:                                0.034854192
+    avg_grad_norm_pre_clip:                     0.16457045
+    avg_loss:                                   -0.00027765878
+    max:
+      max_actor_approxkl1:                      0.0007763827
+      max_actor_approxkl3:                      0.00038080578
+      max_actor_clipfrac:                       0.018432617
+      max_actor_loss:                           0.00015677698
+      max_advantage:                            0.04576689
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.6984424e-05
+      max_entropy:                              0.04136107
+      max_grad_norm_pre_clip:                   0.33324587
+      max_loss:                                 0.0001416017
+    std:
+      std_actor_approxkl1:                      0.0002581364
+      std_actor_approxkl3:                      9.263646e-05
+      std_actor_clipfrac:                       0.0047162445
+      std_actor_loss:                           0.00025962063
+      std_advantage:                            0.007792114
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.034794e-06
+      std_entropy:                              0.003804092
+      std_grad_norm_pre_clip:                   0.06311786
+      std_loss:                                 0.0002606006
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.04969
+    avg_avg_return:                             0.97356176
+    avg_benchmark_regret:                       0.016067734
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13334656
+    proxy_corner:
+      avg_avg_return:                           0.97356176
+      avg_reward_per_step:                      0.13334656
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.16173
+    avg_avg_return:                             0.053704273
+    avg_benchmark_regret:                       0.93906724
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.01033783
+    proxy_corner:
+      avg_avg_return:                           0.85364306
+      avg_reward_per_step:                      0.006942749
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.12134
+    avg_avg_return:                             0.115283325
+    avg_benchmark_regret:                       0.8687923
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.026069641
+    proxy_corner:
+      avg_avg_return:                           0.115283325
+      avg_reward_per_step:                      0.026069641
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        108003328
+    env-step-replay-before:                     55115776
+    env-step-used-before:                       108003328
+    ppo-update-before:                          65920
+    env-step-generate-before:                   52887552
+    env-step-all-after:                         108036096
+    env-step-replay-after:                      55148544
+    env-step-used-after:                        108036096
+    ppo-update-after:                           65940
+    env-step-generate-after:                    52887552
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.637249
+    avg_avg_return:                             0.8422433
+    avg_reward_per_step:                        0.047058105
+    proxy_corner:
+      avg_avg_return:                           0.8422433
+      avg_reward_per_step:                      0.047058105
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.637249
+    avg_avg_return:                             0.8422433
+    avg_reward_per_step:                        0.047058105
+    proxy_corner:
+      avg_avg_return:                           0.8422433
+      avg_reward_per_step:                      0.047058105
+  perf:
+    env_steps_per_second:                       318287.12325663335
+    ppo_updates_per_second:                     42.23647651822738
+    eval-batch-orig-duration:                   0.5307869901880622
+    eval-batch-shift-duration:                  0.4725782023742795
+    eval-batch-tree-duration:                   0.4515337599441409
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.996094
+      mouse-cheese_dist_finite_wavg:            18.99351
+      mouse-cheese_dist_solvable_avg:           18.996094
+      mouse-cheese_dist_solvable_wavg:          18.993511
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1399941
+    visit_patterns:
+      avg_first_visit_time:                     654.04834
+      avg_last_visit_time:                      1673.4207
+      num_replay_batches:                       1683
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00034576294
+    avg_actor_approxkl3:                        0.00025826882
+    avg_actor_clipfrac:                         0.018914795
+    avg_actor_loss:                             4.012026e-05
+    avg_advantage:                              -0.00035687617
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.2435732e-05
+    avg_entropy:                                0.05954448
+    avg_grad_norm_pre_clip:                     0.6400116
+    avg_loss:                                   -1.3206351e-05
+    max:
+      max_actor_approxkl1:                      0.0010122242
+      max_actor_approxkl3:                      0.0007727238
+      max_actor_clipfrac:                       0.061645508
+      max_actor_loss:                           0.001303941
+      max_advantage:                            0.034122884
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.4806525e-05
+      max_entropy:                              0.066369504
+      max_grad_norm_pre_clip:                   1.5205518
+      max_loss:                                 0.0012484186
+    std:
+      std_actor_approxkl1:                      0.00030898815
+      std_actor_approxkl3:                      0.00020283328
+      std_actor_clipfrac:                       0.016491942
+      std_actor_loss:                           0.00045665752
+      std_advantage:                            0.005232018
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.3457231e-06
+      std_entropy:                              0.0021665976
+      std_grad_norm_pre_clip:                   0.41896954
+      std_loss:                                 0.00045538018
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.044302
+    avg_avg_return:                             0.973538
+    avg_benchmark_regret:                       0.016091526
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.133461
+    proxy_corner:
+      avg_avg_return:                           0.973538
+      avg_reward_per_step:                      0.133461
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.16499
+    avg_avg_return:                             0.0535929
+    avg_benchmark_regret:                       0.9391786
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.016548157
+    proxy_corner:
+      avg_avg_return:                           0.81310165
+      avg_reward_per_step:                      0.006576538
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.95408
+    avg_avg_return:                             0.09655809
+    avg_benchmark_regret:                       0.8875176
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.026031494
+    proxy_corner:
+      avg_avg_return:                           0.09655809
+      avg_reward_per_step:                      0.026031494
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        109051904
+    env-step-replay-before:                     55541760
+    env-step-used-before:                       109051904
+    ppo-update-before:                          66560
+    env-step-generate-before:                   53510144
+    env-step-all-after:                         109084672
+    env-step-replay-after:                      55574528
+    env-step-used-after:                        109084672
+    ppo-update-after:                           66580
+    env-step-generate-after:                    53510144
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.632156
+    avg_avg_return:                             0.8423004
+    avg_reward_per_step:                        0.04727173
+    proxy_corner:
+      avg_avg_return:                           0.8423004
+      avg_reward_per_step:                      0.04727173
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.632156
+    avg_avg_return:                             0.8423004
+    avg_reward_per_step:                        0.04727173
+    proxy_corner:
+      avg_avg_return:                           0.8423004
+      avg_reward_per_step:                      0.04727173
+  perf:
+    env_steps_per_second:                       325350.8561622284
+    ppo_updates_per_second:                     41.3416347608687
+    eval-batch-orig-duration:                   0.5232550175860524
+    eval-batch-shift-duration:                  0.4828458186239004
+    eval-batch-tree-duration:                   0.45629935152828693
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.999268
+      mouse-cheese_dist_finite_wavg:            18.999598
+      mouse-cheese_dist_solvable_avg:           18.999268
+      mouse-cheese_dist_solvable_wavg:          18.999598
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14010385
+    visit_patterns:
+      avg_first_visit_time:                     654.81396
+      avg_last_visit_time:                      1686.3247
+      num_replay_batches:                       1696
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00028264083
+    avg_actor_approxkl3:                        0.0003608957
+    avg_actor_clipfrac:                         0.0045776367
+    avg_actor_loss:                             1.8114062e-05
+    avg_advantage:                              0.00016276898
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            6.8989534e-06
+    avg_entropy:                                0.029356057
+    avg_grad_norm_pre_clip:                     0.1974616
+    avg_loss:                                   -7.792528e-06
+    max:
+      max_actor_approxkl1:                      0.0011100931
+      max_actor_approxkl3:                      0.0011901216
+      max_actor_clipfrac:                       0.010131836
+      max_actor_loss:                           0.0007587299
+      max_advantage:                            0.0217852
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.2527389e-05
+      max_entropy:                              0.034674317
+      max_grad_norm_pre_clip:                   0.7777141
+      max_loss:                                 0.0007277528
+    std:
+      std_actor_approxkl1:                      0.0003683619
+      std_actor_approxkl3:                      0.0003704011
+      std_actor_clipfrac:                       0.002852504
+      std_actor_loss:                           0.00022072616
+      std_advantage:                            0.003724178
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.0677337e-06
+      std_entropy:                              0.0035827695
+      std_grad_norm_pre_clip:                   0.14939213
+      std_loss:                                 0.00022110934
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041113
+    avg_avg_return:                             0.9738046
+    avg_benchmark_regret:                       0.015824944
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349915
+    proxy_corner:
+      avg_avg_return:                           0.9738046
+      avg_reward_per_step:                      0.13349915
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.757744
+    avg_avg_return:                             0.058411133
+    avg_benchmark_regret:                       0.9343604
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.016716003
+    proxy_corner:
+      avg_avg_return:                           0.7677522
+      avg_reward_per_step:                      0.0062408447
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.46393
+    avg_avg_return:                             0.10608789
+    avg_benchmark_regret:                       0.87798774
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.0260849
+    proxy_corner:
+      avg_avg_return:                           0.10608789
+      avg_reward_per_step:                      0.0260849
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        110100480
+    env-step-replay-before:                     56066048
+    env-step-used-before:                       110100480
+    ppo-update-before:                          67200
+    env-step-generate-before:                   54034432
+    env-step-all-after:                         110133248
+    env-step-replay-after:                      56098816
+    env-step-used-after:                        110133248
+    ppo-update-after:                           67220
+    env-step-generate-after:                    54034432
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.682053
+    avg_avg_return:                             0.84191966
+    avg_reward_per_step:                        0.046905518
+    proxy_corner:
+      avg_avg_return:                           0.84191966
+      avg_reward_per_step:                      0.046905518
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.682053
+    avg_avg_return:                             0.84191966
+    avg_reward_per_step:                        0.046905518
+    proxy_corner:
+      avg_avg_return:                           0.84191966
+      avg_reward_per_step:                      0.046905518
+  perf:
+    env_steps_per_second:                       242703.03653531228
+    ppo_updates_per_second:                     42.3740722074448
+    eval-batch-orig-duration:                   0.6174082476645708
+    eval-batch-shift-duration:                  0.6009784080088139
+    eval-batch-tree-duration:                   0.6411435343325138
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.003906
+      mouse-cheese_dist_finite_wavg:            19.000391
+      mouse-cheese_dist_solvable_avg:           19.003906
+      mouse-cheese_dist_solvable_wavg:          19.000395
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14023474
+    visit_patterns:
+      avg_first_visit_time:                     655.3474
+      avg_last_visit_time:                      1702.2173
+      num_replay_batches:                       1712
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009479434
+    avg_actor_approxkl3:                        0.00058972166
+    avg_actor_clipfrac:                         0.008276368
+    avg_actor_loss:                             -2.2550114e-06
+    avg_advantage:                              -0.000662222
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            5.595313e-06
+    avg_entropy:                                0.025121478
+    avg_grad_norm_pre_clip:                     0.24876288
+    avg_loss:                                   -2.4578838e-05
+    max:
+      max_actor_approxkl1:                      0.0016976246
+      max_actor_approxkl3:                      0.0015165689
+      max_actor_clipfrac:                       0.03125
+      max_actor_loss:                           0.0024236217
+      max_advantage:                            0.017268538
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          7.914788e-06
+      max_entropy:                              0.028704707
+      max_grad_norm_pre_clip:                   1.0537726
+      max_loss:                                 0.0023975233
+    std:
+      std_actor_approxkl1:                      0.0005016681
+      std_actor_approxkl3:                      0.00039848743
+      std_actor_clipfrac:                       0.009298168
+      std_actor_loss:                           0.0008128593
+      std_advantage:                            0.0032900542
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.1262039e-06
+      std_entropy:                              0.002041557
+      std_grad_norm_pre_clip:                   0.2833342
+      std_loss:                                 0.0008120574
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.042548
+    avg_avg_return:                             0.97369206
+    avg_benchmark_regret:                       0.015937444
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349152
+    proxy_corner:
+      avg_avg_return:                           0.97369206
+      avg_reward_per_step:                      0.13349152
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.68608
+    avg_avg_return:                             0.059218377
+    avg_benchmark_regret:                       0.9335531
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.014778137
+    proxy_corner:
+      avg_avg_return:                           0.78858584
+      avg_reward_per_step:                      0.0063934326
+  eval-batch-tree:
+    avg_avg_episode_length:                     80.12227
+    avg_avg_return:                             0.08692375
+    avg_benchmark_regret:                       0.8971518
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025901794
+    proxy_corner:
+      avg_avg_return:                           0.08692375
+      avg_reward_per_step:                      0.025901794
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        111149056
+    env-step-replay-before:                     56688640
+    env-step-used-before:                       111149056
+    ppo-update-before:                          67840
+    env-step-generate-before:                   54460416
+    env-step-all-after:                         111181824
+    env-step-replay-after:                      56721408
+    env-step-used-after:                        111181824
+    ppo-update-after:                           67860
+    env-step-generate-after:                    54460416
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84183836
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84183836
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84183836
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84183836
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       318067.75600375014
+    ppo_updates_per_second:                     42.026564431398924
+    eval-batch-orig-duration:                   0.5373788010329008
+    eval-batch-shift-duration:                  0.49130765814334154
+    eval-batch-tree-duration:                   0.2809630883857608
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.003906
+      mouse-cheese_dist_finite_wavg:            19.000322
+      mouse-cheese_dist_solvable_avg:           19.003906
+      mouse-cheese_dist_solvable_wavg:          19.00032
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14028975
+    visit_patterns:
+      avg_first_visit_time:                     655.7168
+      avg_last_visit_time:                      1721.2168
+      num_replay_batches:                       1731
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        1.2527014e-06
+    avg_actor_approxkl3:                        5.4400743e-07
+    avg_actor_clipfrac:                         3.6621095e-05
+    avg_actor_loss:                             -3.5047067e-06
+    avg_advantage:                              -0.00023614867
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            5.029359e-07
+    avg_entropy:                                0.0012048021
+    avg_grad_norm_pre_clip:                     0.01800989
+    avg_loss:                                   -4.4580406e-06
+    max:
+      max_actor_approxkl1:                      3.4268312e-05
+      max_actor_approxkl3:                      4.6136383e-06
+      max_actor_clipfrac:                       0.00036621094
+      max_actor_loss:                           9.929761e-06
+      max_advantage:                            0.0026128888
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          8.8446393e-07
+      max_entropy:                              0.0015877248
+      max_grad_norm_pre_clip:                   0.039519332
+      max_loss:                                 9.086001e-06
+    std:
+      std_actor_approxkl1:                      1.0914228e-05
+      std_actor_approxkl3:                      1.0579155e-06
+      std_actor_clipfrac:                       8.717564e-05
+      std_actor_loss:                           1.3412862e-05
+      std_advantage:                            0.0012057053
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.0992975e-07
+      std_entropy:                              0.00029383478
+      std_grad_norm_pre_clip:                   0.011700874
+      std_loss:                                 1.35420705e-05
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040646
+    avg_avg_return:                             0.9738288
+    avg_benchmark_regret:                       0.015800737
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13350677
+    proxy_corner:
+      avg_avg_return:                           0.9738288
+      avg_reward_per_step:                      0.13350677
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.826126
+    avg_avg_return:                             0.057011038
+    avg_benchmark_regret:                       0.9357605
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.017372131
+    proxy_corner:
+      avg_avg_return:                           0.75324106
+      avg_reward_per_step:                      0.0060653687
+  eval-batch-tree:
+    avg_avg_episode_length:                     80.922676
+    avg_avg_return:                             0.06484022
+    avg_benchmark_regret:                       0.9192354
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025650024
+    proxy_corner:
+      avg_avg_return:                           0.06484022
+      avg_reward_per_step:                      0.025650024
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        112197632
+    env-step-replay-before:                     57376768
+    env-step-used-before:                       112197632
+    ppo-update-before:                          68480
+    env-step-generate-before:                   54820864
+    env-step-all-after:                         112230400
+    env-step-replay-after:                      57409536
+    env-step-used-after:                        112230400
+    ppo-update-after:                           68500
+    env-step-generate-after:                    54820864
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184444
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184444
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184444
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184444
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       355549.5139180237
+    ppo_updates_per_second:                     42.34671360389137
+    eval-batch-orig-duration:                   0.5159342791885138
+    eval-batch-shift-duration:                  0.46571902744472027
+    eval-batch-tree-duration:                   0.44626243971288204
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.003174
+      mouse-cheese_dist_finite_wavg:            19.000408
+      mouse-cheese_dist_solvable_avg:           19.003174
+      mouse-cheese_dist_solvable_wavg:          19.000412
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030704
+    visit_patterns:
+      avg_first_visit_time:                     656.1406
+      avg_last_visit_time:                      1742.2847
+      num_replay_batches:                       1752
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        -7.328805e-06
+    avg_actor_approxkl3:                        2.6101395e-06
+    avg_actor_clipfrac:                         0.00015258789
+    avg_actor_loss:                             -1.659887e-05
+    avg_advantage:                              -0.00089102006
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.6329037e-06
+    avg_entropy:                                0.0020871917
+    avg_grad_norm_pre_clip:                     0.020039117
+    avg_loss:                                   -1.786961e-05
+    max:
+      max_actor_approxkl1:                      5.7726775e-05
+      max_actor_approxkl3:                      1.5266633e-05
+      max_actor_clipfrac:                       0.00061035156
+      max_actor_loss:                           2.1236017e-05
+      max_advantage:                            0.0051924586
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.9396897e-06
+      max_entropy:                              0.002651569
+      max_grad_norm_pre_clip:                   0.037769094
+      max_loss:                                 1.9880543e-05
+    std:
+      std_actor_approxkl1:                      2.7526434e-05
+      std_actor_approxkl3:                      3.3896877e-06
+      std_actor_clipfrac:                       0.00018054442
+      std_actor_loss:                           2.2292537e-05
+      std_advantage:                            0.0021979595
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.593904e-07
+      std_entropy:                              0.0002790966
+      std_grad_norm_pre_clip:                   0.0067505
+      std_loss:                                 2.2327466e-05
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040766
+    avg_avg_return:                             0.9738055
+    avg_benchmark_regret:                       0.015824027
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349915
+    proxy_corner:
+      avg_avg_return:                           0.9738055
+      avg_reward_per_step:                      0.13349915
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.61601
+    avg_avg_return:                             0.059942104
+    avg_benchmark_regret:                       0.9328294
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.017478943
+    proxy_corner:
+      avg_avg_return:                           0.77114916
+      avg_reward_per_step:                      0.0062408447
+  eval-batch-tree:
+    avg_avg_episode_length:                     80.61969
+    avg_avg_return:                             0.08215015
+    avg_benchmark_regret:                       0.90192544
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02583313
+    proxy_corner:
+      avg_avg_return:                           0.08215015
+      avg_reward_per_step:                      0.02583313
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        113246208
+    env-step-replay-before:                     57901056
+    env-step-used-before:                       113246208
+    ppo-update-before:                          69120
+    env-step-generate-before:                   55345152
+    env-step-all-after:                         113278976
+    env-step-replay-after:                      57901056
+    env-step-used-after:                        113278976
+    ppo-update-after:                           69140
+    env-step-generate-after:                    55377920
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.1230955
+    avg_avg_return:                             0.92734724
+    avg_reward_per_step:                        0.13931274
+    proxy_corner:
+      avg_avg_return:                           0.92734724
+      avg_reward_per_step:                      0.13931274
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.1230955
+    avg_avg_return:                             0.92734724
+    avg_reward_per_step:                        0.13931274
+    proxy_corner:
+      avg_avg_return:                           0.92734724
+      avg_reward_per_step:                      0.13931274
+  perf:
+    env_steps_per_second:                       215259.14542724457
+    ppo_updates_per_second:                     42.38027865931183
+    eval-batch-orig-duration:                   0.6593140792101622
+    eval-batch-shift-duration:                  0.6320928232744336
+    eval-batch-tree-duration:                   0.6084831757470965
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.002441
+      mouse-cheese_dist_finite_wavg:            19.00022
+      mouse-cheese_dist_solvable_avg:           19.002441
+      mouse-cheese_dist_solvable_wavg:          19.000223
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030725
+    visit_patterns:
+      avg_first_visit_time:                     656.2468
+      avg_last_visit_time:                      1757.283
+      num_replay_batches:                       1767
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        5.3820055e-05
+    avg_actor_approxkl3:                        7.7859404e-05
+    avg_actor_clipfrac:                         0.004071045
+    avg_actor_loss:                             -3.4042354e-05
+    avg_advantage:                              -0.00089989265
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.6110875e-05
+    avg_entropy:                                0.022850439
+    avg_grad_norm_pre_clip:                     0.11382847
+    avg_loss:                                   -4.8837363e-05
+    max:
+      max_actor_approxkl1:                      0.00036003586
+      max_actor_approxkl3:                      0.00031896099
+      max_actor_clipfrac:                       0.01977539
+      max_actor_loss:                           0.0005578622
+      max_advantage:                            0.045050085
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.1301283e-05
+      max_entropy:                              0.031526197
+      max_grad_norm_pre_clip:                   0.33899754
+      max_loss:                                 0.0005388642
+    std:
+      std_actor_approxkl1:                      0.00014148412
+      std_actor_approxkl3:                      7.722746e-05
+      std_actor_clipfrac:                       0.005158287
+      std_actor_loss:                           0.00021532297
+      std_advantage:                            0.0060900846
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.510622e-06
+      std_entropy:                              0.004283266
+      std_grad_norm_pre_clip:                   0.07648909
+      std_loss:                                 0.00021417347
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041443
+    avg_avg_return:                             0.9737972
+    avg_benchmark_regret:                       0.015832275
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349915
+    proxy_corner:
+      avg_avg_return:                           0.9737972
+      avg_reward_per_step:                      0.13349915
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.854904
+    avg_avg_return:                             0.056548584
+    avg_benchmark_regret:                       0.9362229
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.01537323
+    proxy_corner:
+      avg_avg_return:                           0.76629937
+      avg_reward_per_step:                      0.0062026978
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.877
+    avg_avg_return:                             0.096625
+    avg_benchmark_regret:                       0.8874506
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025985718
+    proxy_corner:
+      avg_avg_return:                           0.096625
+      avg_reward_per_step:                      0.025985718
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        114294784
+    env-step-replay-before:                     58359808
+    env-step-used-before:                       114294784
+    ppo-update-before:                          69760
+    env-step-generate-before:                   55934976
+    env-step-all-after:                         114327552
+    env-step-replay-after:                      58392576
+    env-step-used-after:                        114327552
+    ppo-update-after:                           69780
+    env-step-generate-after:                    55934976
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       349624.0521795263
+    ppo_updates_per_second:                     40.66013237680459
+    eval-batch-orig-duration:                   0.5033566728234291
+    eval-batch-shift-duration:                  0.4511916972696781
+    eval-batch-tree-duration:                   0.48307353165000677
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.001953
+      mouse-cheese_dist_finite_wavg:            19.0003
+      mouse-cheese_dist_solvable_avg:           19.001953
+      mouse-cheese_dist_solvable_wavg:          19.0003
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030719
+    visit_patterns:
+      avg_first_visit_time:                     656.3247
+      avg_last_visit_time:                      1772.2971
+      num_replay_batches:                       1782
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00041347308
+    avg_actor_approxkl3:                        0.00030785752
+    avg_actor_clipfrac:                         0.013067627
+    avg_actor_loss:                             -3.1207874e-05
+    avg_advantage:                              0.0033428245
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.23727e-05
+    avg_entropy:                                0.023427961
+    avg_grad_norm_pre_clip:                     0.051728416
+    avg_loss:                                   -4.8449485e-05
+    max:
+      max_actor_approxkl1:                      0.0016321804
+      max_actor_approxkl3:                      0.00087480235
+      max_actor_clipfrac:                       0.030761719
+      max_actor_loss:                           9.23574e-05
+      max_advantage:                            0.02690041
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.9519519e-05
+      max_entropy:                              0.028121702
+      max_grad_norm_pre_clip:                   0.111519456
+      max_loss:                                 7.283357e-05
+    std:
+      std_actor_approxkl1:                      0.00050547614
+      std_actor_approxkl3:                      0.00029760387
+      std_actor_clipfrac:                       0.012443539
+      std_actor_loss:                           8.530101e-05
+      std_advantage:                            0.0065235444
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.155551e-06
+      std_entropy:                              0.0019572964
+      std_grad_norm_pre_clip:                   0.019870613
+      std_loss:                                 8.6582346e-05
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.043833
+    avg_avg_return:                             0.9735477
+    avg_benchmark_regret:                       0.016081786
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13339996
+    proxy_corner:
+      avg_avg_return:                           0.9735477
+      avg_reward_per_step:                      0.13339996
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.53401
+    avg_avg_return:                             0.053238228
+    avg_benchmark_regret:                       0.9395333
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0135650635
+    proxy_corner:
+      avg_avg_return:                           0.88842756
+      avg_reward_per_step:                      0.007232666
+  eval-batch-tree:
+    avg_avg_episode_length:                     71.54579
+    avg_avg_return:                             0.21452954
+    avg_benchmark_regret:                       0.7695461
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.029037476
+    proxy_corner:
+      avg_avg_return:                           0.21452954
+      avg_reward_per_step:                      0.029037476
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        115343360
+    env-step-replay-before:                     58982400
+    env-step-used-before:                       115343360
+    ppo-update-before:                          70400
+    env-step-generate-before:                   56360960
+    env-step-all-after:                         115376128
+    env-step-replay-after:                      58982400
+    env-step-used-after:                        115376128
+    ppo-update-after:                           70420
+    env-step-generate-after:                    56393728
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.870898
+    avg_avg_return:                             0.92887604
+    avg_reward_per_step:                        0.14102173
+    proxy_corner:
+      avg_avg_return:                           0.92887604
+      avg_reward_per_step:                      0.14102173
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.870898
+    avg_avg_return:                             0.92887604
+    avg_reward_per_step:                        0.14102173
+    proxy_corner:
+      avg_avg_return:                           0.92887604
+      avg_reward_per_step:                      0.14102173
+  perf:
+    env_steps_per_second:                       345144.327973186
+    ppo_updates_per_second:                     42.39300013138278
+    eval-batch-orig-duration:                   0.5238114073872566
+    eval-batch-shift-duration:                  0.4816470108926296
+    eval-batch-tree-duration:                   0.4902380621060729
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.000488
+      mouse-cheese_dist_finite_wavg:            19.000057
+      mouse-cheese_dist_solvable_avg:           19.000488
+      mouse-cheese_dist_solvable_wavg:          19.00006
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030766
+    visit_patterns:
+      avg_first_visit_time:                     656.54565
+      avg_last_visit_time:                      1790.1436
+      num_replay_batches:                       1800
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0004526226
+    avg_actor_approxkl3:                        0.0004808121
+    avg_actor_clipfrac:                         0.017059326
+    avg_actor_loss:                             1.1781917e-05
+    avg_advantage:                              -0.0007231593
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.4456103e-05
+    avg_entropy:                                0.04639429
+    avg_grad_norm_pre_clip:                     0.3482122
+    avg_loss:                                   -2.238432e-05
+    max:
+      max_actor_approxkl1:                      0.0014885669
+      max_actor_approxkl3:                      0.0014119572
+      max_actor_clipfrac:                       0.052001953
+      max_actor_loss:                           0.0010529598
+      max_advantage:                            0.031884372
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          4.1902502e-05
+      max_entropy:                              0.06963755
+      max_grad_norm_pre_clip:                   0.7978074
+      max_loss:                                 0.001015607
+    std:
+      std_actor_approxkl1:                      0.00040581598
+      std_actor_approxkl3:                      0.00039538555
+      std_actor_clipfrac:                       0.015118884
+      std_actor_loss:                           0.00045711623
+      std_advantage:                            0.0060902704
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.875614e-06
+      std_entropy:                              0.008874765
+      std_grad_norm_pre_clip:                   0.1761099
+      std_loss:                                 0.00045448728
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.042992
+    avg_avg_return:                             0.97376585
+    avg_benchmark_regret:                       0.015863717
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13348389
+    proxy_corner:
+      avg_avg_return:                           0.97376585
+      avg_reward_per_step:                      0.13348389
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.10474
+    avg_avg_return:                             0.05647647
+    avg_benchmark_regret:                       0.93629503
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.014579773
+    proxy_corner:
+      avg_avg_return:                           0.8115278
+      avg_reward_per_step:                      0.0066223145
+  eval-batch-tree:
+    avg_avg_episode_length:                     73.31821
+    avg_avg_return:                             0.20174295
+    avg_benchmark_regret:                       0.78233266
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.028335571
+    proxy_corner:
+      avg_avg_return:                           0.20174295
+      avg_reward_per_step:                      0.028335571
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        116391936
+    env-step-replay-before:                     59506688
+    env-step-used-before:                       116391936
+    ppo-update-before:                          71040
+    env-step-generate-before:                   56885248
+    env-step-all-after:                         116424704
+    env-step-replay-after:                      59539456
+    env-step-used-after:                        116424704
+    ppo-update-after:                           71060
+    env-step-generate-after:                    56885248
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8418417
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8418417
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8418417
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8418417
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       214446.67351302455
+    ppo_updates_per_second:                     42.38754393366683
+    eval-batch-orig-duration:                   0.638538115657866
+    eval-batch-shift-duration:                  0.6161757614463568
+    eval-batch-tree-duration:                   0.6275123497471213
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.000244
+      mouse-cheese_dist_finite_wavg:            19.000044
+      mouse-cheese_dist_solvable_avg:           19.000244
+      mouse-cheese_dist_solvable_wavg:          19.000042
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     656.5796
+      avg_last_visit_time:                      1807.2446
+      num_replay_batches:                       1817
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        3.0108378e-05
+    avg_actor_approxkl3:                        1.5244036e-05
+    avg_actor_clipfrac:                         0.00026245118
+    avg_actor_loss:                             -4.830584e-05
+    avg_advantage:                              -0.00083052577
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            6.25369e-06
+    avg_entropy:                                0.0018741576
+    avg_grad_norm_pre_clip:                     0.031737227
+    avg_loss:                                   -4.7053152e-05
+    max:
+      max_actor_approxkl1:                      0.0001296759
+      max_actor_approxkl3:                      5.1732568e-05
+      max_actor_clipfrac:                       0.00061035156
+      max_actor_loss:                           9.1828406e-07
+      max_advantage:                            0.028806865
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          8.436829e-06
+      max_entropy:                              0.0024123308
+      max_grad_norm_pre_clip:                   0.06886779
+      max_loss:                                 2.7586962e-06
+    std:
+      std_actor_approxkl1:                      4.9794497e-05
+      std_actor_approxkl3:                      1.5860236e-05
+      std_actor_clipfrac:                       0.00021239884
+      std_actor_loss:                           3.956507e-05
+      std_advantage:                            0.0037848474
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.5192857e-06
+      std_entropy:                              0.00027925702
+      std_grad_norm_pre_clip:                   0.018586375
+      std_loss:                                 3.9882027e-05
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.047136
+    avg_avg_return:                             0.97370946
+    avg_benchmark_regret:                       0.015920049
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.1333847
+    proxy_corner:
+      avg_avg_return:                           0.97370946
+      avg_reward_per_step:                      0.1333847
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.81425
+    avg_avg_return:                             0.06326699
+    avg_benchmark_regret:                       0.9295046
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.012016296
+    proxy_corner:
+      avg_avg_return:                           0.87407184
+      avg_reward_per_step:                      0.007156372
+  eval-batch-tree:
+    avg_avg_episode_length:                     69.74902
+    avg_avg_return:                             0.23141256
+    avg_benchmark_regret:                       0.752663
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.029464722
+    proxy_corner:
+      avg_avg_return:                           0.23141256
+      avg_reward_per_step:                      0.029464722
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        117440512
+    env-step-replay-before:                     60030976
+    env-step-used-before:                       117440512
+    ppo-update-before:                          71680
+    env-step-generate-before:                   57409536
+    env-step-all-after:                         117473280
+    env-step-replay-after:                      60030976
+    env-step-used-after:                        117473280
+    ppo-update-after:                           71700
+    env-step-generate-after:                    57442304
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.726248
+    avg_avg_return:                             0.92972636
+    avg_reward_per_step:                        0.1409607
+    proxy_corner:
+      avg_avg_return:                           0.92972636
+      avg_reward_per_step:                      0.1409607
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.726248
+    avg_avg_return:                             0.92972636
+    avg_reward_per_step:                        0.1409607
+    proxy_corner:
+      avg_avg_return:                           0.92972636
+      avg_reward_per_step:                      0.1409607
+  perf:
+    env_steps_per_second:                       319035.6166786255
+    ppo_updates_per_second:                     41.911041411027696
+    eval-batch-orig-duration:                   0.5061087608337402
+    eval-batch-shift-duration:                  0.42757342755794525
+    eval-batch-tree-duration:                   0.4199984623119235
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999996
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          18.999996
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     656.8479
+      avg_last_visit_time:                      1822.3013
+      num_replay_batches:                       1832
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00064863305
+    avg_actor_approxkl3:                        0.000399858
+    avg_actor_clipfrac:                         0.016900634
+    avg_actor_loss:                             -0.00039769243
+    avg_advantage:                              -0.0020217858
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.4320556e-05
+    avg_entropy:                                0.071547
+    avg_grad_norm_pre_clip:                     0.2930073
+    avg_loss:                                   -0.00045707912
+    max:
+      max_actor_approxkl1:                      0.0013802224
+      max_actor_approxkl3:                      0.0010231711
+      max_actor_clipfrac:                       0.032958984
+      max_actor_loss:                           0.00020425208
+      max_advantage:                            0.041974783
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.9626033e-05
+      max_entropy:                              0.085657746
+      max_grad_norm_pre_clip:                   0.58448136
+      max_loss:                                 0.00015353717
+    std:
+      std_actor_approxkl1:                      0.00042438853
+      std_actor_approxkl3:                      0.00025310702
+      std_actor_clipfrac:                       0.009357983
+      std_actor_loss:                           0.00044541777
+      std_advantage:                            0.0068235327
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.842612e-06
+      std_entropy:                              0.0066646296
+      std_grad_norm_pre_clip:                   0.10231751
+      std_loss:                                 0.00044734706
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.051582
+    avg_avg_return:                             0.97360206
+    avg_benchmark_regret:                       0.016027493
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13336945
+    proxy_corner:
+      avg_avg_return:                           0.97360206
+      avg_reward_per_step:                      0.13336945
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.349655
+    avg_avg_return:                             0.069794506
+    avg_benchmark_regret:                       0.92297703
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.010475159
+    proxy_corner:
+      avg_avg_return:                           0.83546066
+      avg_reward_per_step:                      0.006881714
+  eval-batch-tree:
+    avg_avg_episode_length:                     71.47662
+    avg_avg_return:                             0.21814166
+    avg_benchmark_regret:                       0.765934
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.028793335
+    proxy_corner:
+      avg_avg_return:                           0.21814166
+      avg_reward_per_step:                      0.028793335
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        118489088
+    env-step-replay-before:                     60424192
+    env-step-used-before:                       118489088
+    ppo-update-before:                          72320
+    env-step-generate-before:                   58064896
+    env-step-all-after:                         118521856
+    env-step-replay-after:                      60424192
+    env-step-used-after:                        118521856
+    ppo-update-after:                           72340
+    env-step-generate-after:                    58097664
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.807715
+    avg_avg_return:                             0.9302051
+    avg_reward_per_step:                        0.13516235
+    proxy_corner:
+      avg_avg_return:                           0.9302051
+      avg_reward_per_step:                      0.13516235
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.807715
+    avg_avg_return:                             0.9302051
+    avg_reward_per_step:                        0.13516235
+    proxy_corner:
+      avg_avg_return:                           0.9302051
+      avg_reward_per_step:                      0.13516235
+  perf:
+    env_steps_per_second:                       730169.6736005903
+    ppo_updates_per_second:                     42.15065817100253
+    eval-batch-orig-duration:                   0.3037732280790806
+    eval-batch-shift-duration:                  0.2318431232124567
+    eval-batch-tree-duration:                   0.261571167036891
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.0
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          18.999996
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000002
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     656.8479
+      avg_last_visit_time:                      1834.2119
+      num_replay_batches:                       1844
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0003222858
+    avg_actor_approxkl3:                        0.00032608875
+    avg_actor_clipfrac:                         0.011785889
+    avg_actor_loss:                             -0.00029064628
+    avg_advantage:                              -0.0015014773
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.1309708e-05
+    avg_entropy:                                0.07416971
+    avg_grad_norm_pre_clip:                     0.34110504
+    avg_loss:                                   -0.00035416117
+    max:
+      max_actor_approxkl1:                      0.00093478034
+      max_actor_approxkl3:                      0.00073804735
+      max_actor_clipfrac:                       0.025634766
+      max_actor_loss:                           0.00040920824
+      max_advantage:                            0.040739655
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.9825564e-05
+      max_entropy:                              0.09288955
+      max_grad_norm_pre_clip:                   0.61279655
+      max_loss:                                 0.00032770826
+    std:
+      std_actor_approxkl1:                      0.00031335684
+      std_actor_approxkl3:                      0.00020312612
+      std_actor_clipfrac:                       0.0071553895
+      std_actor_loss:                           0.00041521224
+      std_advantage:                            0.006659295
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.1309341e-06
+      std_entropy:                              0.0072278483
+      std_grad_norm_pre_clip:                   0.11547746
+      std_loss:                                 0.00041249025
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.05061
+    avg_avg_return:                             0.9738142
+    avg_benchmark_regret:                       0.015815342
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13336182
+    proxy_corner:
+      avg_avg_return:                           0.9738142
+      avg_reward_per_step:                      0.13336182
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.620575
+    avg_avg_return:                             0.06481572
+    avg_benchmark_regret:                       0.9279558
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.012504578
+    proxy_corner:
+      avg_avg_return:                           0.8486367
+      avg_reward_per_step:                      0.006942749
+  eval-batch-tree:
+    avg_avg_episode_length:                     72.57283
+    avg_avg_return:                             0.21101081
+    avg_benchmark_regret:                       0.7730648
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.028396606
+    proxy_corner:
+      avg_avg_return:                           0.21101081
+      avg_reward_per_step:                      0.028396606
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        119537664
+    env-step-replay-before:                     60882944
+    env-step-used-before:                       119537664
+    ppo-update-before:                          72960
+    env-step-generate-before:                   58654720
+    env-step-all-after:                         119570432
+    env-step-replay-after:                      60882944
+    env-step-used-after:                        119570432
+    ppo-update-after:                           72980
+    env-step-generate-after:                    58687488
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.075981
+    avg_avg_return:                             0.9267308
+    avg_reward_per_step:                        0.1416626
+    proxy_corner:
+      avg_avg_return:                           0.9267308
+      avg_reward_per_step:                      0.1416626
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.075981
+    avg_avg_return:                             0.9267308
+    avg_reward_per_step:                        0.1416626
+    proxy_corner:
+      avg_avg_return:                           0.9267308
+      avg_reward_per_step:                      0.1416626
+  perf:
+    env_steps_per_second:                       297022.04184177786
+    ppo_updates_per_second:                     42.208956840290675
+    eval-batch-orig-duration:                   0.5185747025534511
+    eval-batch-shift-duration:                  0.4829397303983569
+    eval-batch-tree-duration:                   0.4814680330455303
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999992
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          18.999994
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030907
+    visit_patterns:
+      avg_first_visit_time:                     657.3469
+      avg_last_visit_time:                      1848.2432
+      num_replay_batches:                       1858
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0004104829
+    avg_actor_approxkl3:                        0.00052820967
+    avg_actor_clipfrac:                         0.01591797
+    avg_actor_loss:                             -0.00037210618
+    avg_advantage:                              -0.00045258767
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            3.8483515e-05
+    avg_entropy:                                0.07772386
+    avg_grad_norm_pre_clip:                     0.48397928
+    avg_loss:                                   -0.0004305883
+    max:
+      max_actor_approxkl1:                      0.0016392913
+      max_actor_approxkl3:                      0.00085905043
+      max_actor_clipfrac:                       0.022094727
+      max_actor_loss:                           0.0012448169
+      max_advantage:                            0.044130206
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          5.0726252e-05
+      max_entropy:                              0.09353572
+      max_grad_norm_pre_clip:                   1.078185
+      max_loss:                                 0.0011831971
+    std:
+      std_actor_approxkl1:                      0.00043541857
+      std_actor_approxkl3:                      0.00023597751
+      std_actor_clipfrac:                       0.005446771
+      std_actor_loss:                           0.00080704497
+      std_advantage:                            0.008121348
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.5455467e-06
+      std_entropy:                              0.009459319
+      std_grad_norm_pre_clip:                   0.28691325
+      std_loss:                                 0.0008058954
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.070959
+    avg_avg_return:                             0.9731546
+    avg_benchmark_regret:                       0.016474938
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13315582
+    proxy_corner:
+      avg_avg_return:                           0.9731546
+      avg_reward_per_step:                      0.13315582
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.743355
+    avg_avg_return:                             0.06467064
+    avg_benchmark_regret:                       0.9281009
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.01033783
+    proxy_corner:
+      avg_avg_return:                           0.8661546
+      avg_reward_per_step:                      0.0071411133
+  eval-batch-tree:
+    avg_avg_episode_length:                     71.844986
+    avg_avg_return:                             0.21273082
+    avg_benchmark_regret:                       0.7713448
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.028823853
+    proxy_corner:
+      avg_avg_return:                           0.21273082
+      avg_reward_per_step:                      0.028823853
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        120586240
+    env-step-replay-before:                     61407232
+    env-step-used-before:                       120586240
+    ppo-update-before:                          73600
+    env-step-generate-before:                   59179008
+    env-step-all-after:                         120619008
+    env-step-replay-after:                      61440000
+    env-step-used-after:                        120619008
+    ppo-update-after:                           73620
+    env-step-generate-after:                    59179008
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84181815
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84181815
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84181815
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84181815
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       320538.737789413
+    ppo_updates_per_second:                     44.63275607052036
+    eval-batch-orig-duration:                   0.5197714772075415
+    eval-batch-shift-duration:                  0.4888989171013236
+    eval-batch-tree-duration:                   0.4865399897098541
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.000488
+      mouse-cheese_dist_finite_wavg:            19.000004
+      mouse-cheese_dist_solvable_avg:           19.000488
+      mouse-cheese_dist_solvable_wavg:          19.000004
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14028138
+    visit_patterns:
+      avg_first_visit_time:                     658.2466
+      avg_last_visit_time:                      1865.2437
+      num_replay_batches:                       1875
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00010957676
+    avg_actor_approxkl3:                        0.00033696878
+    avg_actor_clipfrac:                         0.013751221
+    avg_actor_loss:                             -5.625268e-05
+    avg_advantage:                              0.0005328848
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.28501e-05
+    avg_entropy:                                0.1271869
+    avg_grad_norm_pre_clip:                     0.6649938
+    avg_loss:                                   -0.00017201457
+    max:
+      max_actor_approxkl1:                      0.00058626826
+      max_actor_approxkl3:                      0.00074644585
+      max_actor_clipfrac:                       0.043945312
+      max_actor_loss:                           0.0007958021
+      max_advantage:                            0.04216987
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.6333399e-05
+      max_entropy:                              0.13381246
+      max_grad_norm_pre_clip:                   1.7728643
+      max_loss:                                 0.0006754678
+    std:
+      std_actor_approxkl1:                      0.00026095958
+      std_actor_approxkl3:                      0.00019917973
+      std_actor_clipfrac:                       0.013714439
+      std_actor_loss:                           0.0004082242
+      std_advantage:                            0.0072202664
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.4246383e-06
+      std_entropy:                              0.0044292305
+      std_grad_norm_pre_clip:                   0.55028385
+      std_loss:                                 0.00040785305
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.061236
+    avg_avg_return:                             0.9735446
+    avg_benchmark_regret:                       0.01608495
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13314056
+    proxy_corner:
+      avg_avg_return:                           0.9735446
+      avg_reward_per_step:                      0.13314056
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.74877
+    avg_avg_return:                             0.066265896
+    avg_benchmark_regret:                       0.9265056
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.011924744
+    proxy_corner:
+      avg_avg_return:                           0.8846053
+      avg_reward_per_step:                      0.007255554
+  eval-batch-tree:
+    avg_avg_episode_length:                     69.38121
+    avg_avg_return:                             0.22959635
+    avg_benchmark_regret:                       0.7544793
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.03060913
+    proxy_corner:
+      avg_avg_return:                           0.22959635
+      avg_reward_per_step:                      0.03060913
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        121634816
+    env-step-replay-before:                     61964288
+    env-step-used-before:                       121634816
+    ppo-update-before:                          74240
+    env-step-generate-before:                   59670528
+    env-step-all-after:                         121667584
+    env-step-replay-after:                      61997056
+    env-step-used-after:                        121667584
+    ppo-update-after:                           74260
+    env-step-generate-after:                    59670528
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8418379
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8418379
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8418379
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8418379
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       254900.28435043126
+    ppo_updates_per_second:                     41.46177215578167
+    eval-batch-orig-duration:                   0.5340557806193829
+    eval-batch-shift-duration:                  0.47655051946640015
+    eval-batch-tree-duration:                   0.4385417578741908
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.000488
+      mouse-cheese_dist_finite_wavg:            19.000082
+      mouse-cheese_dist_solvable_avg:           19.000488
+      mouse-cheese_dist_solvable_wavg:          19.000084
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030793
+    visit_patterns:
+      avg_first_visit_time:                     659.14307
+      avg_last_visit_time:                      1882.3223
+      num_replay_batches:                       1892
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0008550109
+    avg_actor_approxkl3:                        0.000622098
+    avg_actor_clipfrac:                         0.024536133
+    avg_actor_loss:                             7.391796e-05
+    avg_advantage:                              0.00080014503
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.3753369e-05
+    avg_entropy:                                0.10616847
+    avg_grad_norm_pre_clip:                     0.8559573
+    avg_loss:                                   -2.537383e-05
+    max:
+      max_actor_approxkl1:                      0.0027278916
+      max_actor_approxkl3:                      0.0017187205
+      max_actor_clipfrac:                       0.051879883
+      max_actor_loss:                           0.0014852546
+      max_advantage:                            0.022087276
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.5658514e-05
+      max_entropy:                              0.11174014
+      max_grad_norm_pre_clip:                   2.065105
+      max_loss:                                 0.0013856797
+    std:
+      std_actor_approxkl1:                      0.00054980285
+      std_actor_approxkl3:                      0.00037909002
+      std_actor_clipfrac:                       0.013375475
+      std_actor_loss:                           0.00073615054
+      std_advantage:                            0.0047660284
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.1809311e-06
+      std_entropy:                              0.0036875508
+      std_grad_norm_pre_clip:                   0.36938098
+      std_loss:                                 0.0007368266
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.04158
+    avg_avg_return:                             0.97389007
+    avg_benchmark_regret:                       0.015739435
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349152
+    proxy_corner:
+      avg_avg_return:                           0.97389007
+      avg_reward_per_step:                      0.13349152
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.784164
+    avg_avg_return:                             0.06270023
+    avg_benchmark_regret:                       0.9300713
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.009689331
+    proxy_corner:
+      avg_avg_return:                           0.85282445
+      avg_reward_per_step:                      0.0070114136
+  eval-batch-tree:
+    avg_avg_episode_length:                     70.989395
+    avg_avg_return:                             0.22238609
+    avg_benchmark_regret:                       0.76168954
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.029953003
+    proxy_corner:
+      avg_avg_return:                           0.22238609
+      avg_reward_per_step:                      0.029953003
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        122683392
+    env-step-replay-before:                     62390272
+    env-step-used-before:                       122683392
+    ppo-update-before:                          74880
+    env-step-generate-before:                   60293120
+    env-step-all-after:                         122716160
+    env-step-replay-after:                      62390272
+    env-step-used-after:                        122716160
+    ppo-update-after:                           74900
+    env-step-generate-after:                    60325888
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.227997
+    avg_avg_return:                             0.9242467
+    avg_reward_per_step:                        0.15332031
+    proxy_corner:
+      avg_avg_return:                           0.9242467
+      avg_reward_per_step:                      0.15332031
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.227997
+    avg_avg_return:                             0.9242467
+    avg_reward_per_step:                        0.15332031
+    proxy_corner:
+      avg_avg_return:                           0.9242467
+      avg_reward_per_step:                      0.15332031
+  perf:
+    env_steps_per_second:                       323799.691854548
+    ppo_updates_per_second:                     40.759650808131475
+    eval-batch-orig-duration:                   0.5179525408893824
+    eval-batch-shift-duration:                  0.48849310632795095
+    eval-batch-tree-duration:                   0.5017109634354711
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.999756
+      mouse-cheese_dist_finite_wavg:            18.999935
+      mouse-cheese_dist_solvable_avg:           18.999756
+      mouse-cheese_dist_solvable_wavg:          18.999937
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1403057
+    visit_patterns:
+      avg_first_visit_time:                     659.5491
+      avg_last_visit_time:                      1894.2571
+      num_replay_batches:                       1904
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.000118459495
+    avg_actor_approxkl3:                        0.0001244501
+    avg_actor_clipfrac:                         0.0062072757
+    avg_actor_loss:                             -0.00025616525
+    avg_advantage:                              0.0006085484
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.4396748e-05
+    avg_entropy:                                0.038132872
+    avg_grad_norm_pre_clip:                     0.2974917
+    avg_loss:                                   -0.00028709977
+    max:
+      max_actor_approxkl1:                      0.00045663392
+      max_actor_approxkl3:                      0.00025161833
+      max_actor_clipfrac:                       0.015991211
+      max_actor_loss:                           9.83011e-05
+      max_advantage:                            0.037260056
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.7622548e-05
+      max_entropy:                              0.04692162
+      max_grad_norm_pre_clip:                   0.66536576
+      max_loss:                                 6.142593e-05
+    std:
+      std_actor_approxkl1:                      0.00013634627
+      std_actor_approxkl3:                      7.033726e-05
+      std_actor_clipfrac:                       0.005050566
+      std_actor_loss:                           0.00027138327
+      std_advantage:                            0.0053046364
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.1477156e-06
+      std_entropy:                              0.00529214
+      std_grad_norm_pre_clip:                   0.13521636
+      std_loss:                                 0.00027295668
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.045807
+    avg_avg_return:                             0.97346437
+    avg_benchmark_regret:                       0.016165152
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13343811
+    proxy_corner:
+      avg_avg_return:                           0.97346437
+      avg_reward_per_step:                      0.13343811
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.969025
+    avg_avg_return:                             0.063873045
+    avg_benchmark_regret:                       0.92889845
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0107040405
+    proxy_corner:
+      avg_avg_return:                           0.8080946
+      avg_reward_per_step:                      0.006576538
+  eval-batch-tree:
+    avg_avg_episode_length:                     72.3527
+    avg_avg_return:                             0.20414439
+    avg_benchmark_regret:                       0.7799312
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.029090881
+    proxy_corner:
+      avg_avg_return:                           0.20414439
+      avg_reward_per_step:                      0.029090881
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        123731968
+    env-step-replay-before:                     62717952
+    env-step-used-before:                       123731968
+    ppo-update-before:                          75520
+    env-step-generate-before:                   61014016
+    env-step-all-after:                         123764736
+    env-step-replay-after:                      62717952
+    env-step-used-after:                        123764736
+    ppo-update-after:                           75540
+    env-step-generate-after:                    61046784
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.967503
+    avg_avg_return:                             0.92810667
+    avg_reward_per_step:                        0.13943481
+    proxy_corner:
+      avg_avg_return:                           0.92810667
+      avg_reward_per_step:                      0.13943481
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.967503
+    avg_avg_return:                             0.92810667
+    avg_reward_per_step:                        0.13943481
+    proxy_corner:
+      avg_avg_return:                           0.92810667
+      avg_reward_per_step:                      0.13943481
+  perf:
+    env_steps_per_second:                       239434.33561918887
+    ppo_updates_per_second:                     42.34247859886392
+    eval-batch-orig-duration:                   0.6584728602319956
+    eval-batch-shift-duration:                  0.5699711050838232
+    eval-batch-tree-duration:                   0.6022959146648645
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.0
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          18.999998
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030914
+    visit_patterns:
+      avg_first_visit_time:                     659.8347
+      avg_last_visit_time:                      1904.2139
+      num_replay_batches:                       1914
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00019789716
+    avg_actor_approxkl3:                        0.00015875675
+    avg_actor_clipfrac:                         0.006817627
+    avg_actor_loss:                             -0.00011154374
+    avg_advantage:                              0.0010646713
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.0056408e-05
+    avg_entropy:                                0.022552853
+    avg_grad_norm_pre_clip:                     0.19304661
+    avg_loss:                                   -0.00012406838
+    max:
+      max_actor_approxkl1:                      0.0006275178
+      max_actor_approxkl3:                      0.0003222308
+      max_actor_clipfrac:                       0.016845703
+      max_actor_loss:                           0.00043066358
+      max_advantage:                            0.039022684
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.8474435e-05
+      max_entropy:                              0.030570257
+      max_grad_norm_pre_clip:                   0.5380024
+      max_loss:                                 0.000418893
+    std:
+      std_actor_approxkl1:                      0.00019468255
+      std_actor_approxkl3:                      8.790932e-05
+      std_actor_clipfrac:                       0.00422287
+      std_actor_loss:                           0.00020985775
+      std_advantage:                            0.0065663755
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.860299e-06
+      std_entropy:                              0.0040419134
+      std_grad_norm_pre_clip:                   0.11429193
+      std_loss:                                 0.00021066936
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040616
+    avg_avg_return:                             0.9738059
+    avg_benchmark_regret:                       0.015823642
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.1335144
+    proxy_corner:
+      avg_avg_return:                           0.9738059
+      avg_reward_per_step:                      0.1335144
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.686554
+    avg_avg_return:                             0.04962626
+    avg_benchmark_regret:                       0.9431453
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.009124756
+    proxy_corner:
+      avg_avg_return:                           0.721592
+      avg_reward_per_step:                      0.00579834
+  eval-batch-tree:
+    avg_avg_episode_length:                     72.01132
+    avg_avg_return:                             0.20527379
+    avg_benchmark_regret:                       0.7788018
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.028968811
+    proxy_corner:
+      avg_avg_return:                           0.20527379
+      avg_reward_per_step:                      0.028968811
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        124780544
+    env-step-replay-before:                     63176704
+    env-step-used-before:                       124780544
+    ppo-update-before:                          76160
+    env-step-generate-before:                   61603840
+    env-step-all-after:                         124813312
+    env-step-replay-after:                      63176704
+    env-step-used-after:                        124813312
+    ppo-update-after:                           76180
+    env-step-generate-after:                    61636608
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.634449
+    avg_avg_return:                             0.93139523
+    avg_reward_per_step:                        0.14266968
+    proxy_corner:
+      avg_avg_return:                           0.93139523
+      avg_reward_per_step:                      0.14266968
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.634449
+    avg_avg_return:                             0.93139523
+    avg_reward_per_step:                        0.14266968
+    proxy_corner:
+      avg_avg_return:                           0.93139523
+      avg_reward_per_step:                      0.14266968
+  perf:
+    env_steps_per_second:                       297892.81491324527
+    ppo_updates_per_second:                     42.263924721108616
+    eval-batch-orig-duration:                   0.5187570992857218
+    eval-batch-shift-duration:                  0.43967538699507713
+    eval-batch-tree-duration:                   0.4794378150254488
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.0
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.0
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     659.8347
+      avg_last_visit_time:                      1918.3718
+      num_replay_batches:                       1928
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00024416106
+    avg_actor_approxkl3:                        0.00026486837
+    avg_actor_clipfrac:                         0.0077697756
+    avg_actor_loss:                             4.5792993e-05
+    avg_advantage:                              -0.0015517792
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.3045692e-05
+    avg_entropy:                                0.030222548
+    avg_grad_norm_pre_clip:                     0.5110121
+    avg_loss:                                   2.7093285e-05
+    max:
+      max_actor_approxkl1:                      0.0009153978
+      max_actor_approxkl3:                      0.00074116467
+      max_actor_clipfrac:                       0.019042969
+      max_actor_loss:                           0.000948983
+      max_advantage:                            0.028671503
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.9272811e-05
+      max_entropy:                              0.0379155
+      max_grad_norm_pre_clip:                   1.1180522
+      max_loss:                                 0.0009293997
+    std:
+      std_actor_approxkl1:                      0.0002791261
+      std_actor_approxkl3:                      0.00017830309
+      std_actor_clipfrac:                       0.004932045
+      std_actor_loss:                           0.0003831092
+      std_advantage:                            0.0069547747
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.5402854e-06
+      std_entropy:                              0.004362249
+      std_grad_norm_pre_clip:                   0.281587
+      std_loss:                                 0.00038242218
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041113
+    avg_avg_return:                             0.97373056
+    avg_benchmark_regret:                       0.015898999
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349915
+    proxy_corner:
+      avg_avg_return:                           0.97373056
+      avg_reward_per_step:                      0.13349915
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.43248
+    avg_avg_return:                             0.0545819
+    avg_benchmark_regret:                       0.9381896
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.005836487
+    proxy_corner:
+      avg_avg_return:                           0.81868887
+      avg_reward_per_step:                      0.006690979
+  eval-batch-tree:
+    avg_avg_episode_length:                     68.73959
+    avg_avg_return:                             0.23114192
+    avg_benchmark_regret:                       0.7529337
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.03062439
+    proxy_corner:
+      avg_avg_return:                           0.23114192
+      avg_reward_per_step:                      0.03062439
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        125829120
+    env-step-replay-before:                     63799296
+    env-step-used-before:                       125829120
+    ppo-update-before:                          76800
+    env-step-generate-before:                   62029824
+    env-step-all-after:                         125861888
+    env-step-replay-after:                      63832064
+    env-step-used-after:                        125861888
+    ppo-update-after:                           76820
+    env-step-generate-after:                    62029824
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184444
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184444
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184444
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184444
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       242179.075482704
+    ppo_updates_per_second:                     42.357611148085844
+    eval-batch-orig-duration:                   0.6711573777720332
+    eval-batch-shift-duration:                  0.6175394989550114
+    eval-batch-tree-duration:                   0.6147934310138226
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.0
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.000002
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030749
+    visit_patterns:
+      avg_first_visit_time:                     659.8347
+      avg_last_visit_time:                      1938.2217
+      num_replay_batches:                       1948
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0004897749
+    avg_actor_approxkl3:                        0.0004737754
+    avg_actor_clipfrac:                         0.015820313
+    avg_actor_loss:                             0.00014382401
+    avg_advantage:                              -0.000913388
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.1788149e-05
+    avg_entropy:                                0.07284831
+    avg_grad_norm_pre_clip:                     0.905211
+    avg_loss:                                   7.686975e-05
+    max:
+      max_actor_approxkl1:                      0.00096266856
+      max_actor_approxkl3:                      0.0009260135
+      max_actor_clipfrac:                       0.02734375
+      max_actor_loss:                           0.0014291909
+      max_advantage:                            0.0447827
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.5829652e-05
+      max_entropy:                              0.078514725
+      max_grad_norm_pre_clip:                   2.3087573
+      max_loss:                                 0.0013683308
+    std:
+      std_actor_approxkl1:                      0.00029635103
+      std_actor_approxkl3:                      0.00025967686
+      std_actor_clipfrac:                       0.0074886335
+      std_actor_loss:                           0.0006741366
+      std_advantage:                            0.0045068115
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.5360795e-06
+      std_entropy:                              0.003160849
+      std_grad_norm_pre_clip:                   0.59928113
+      std_loss:                                 0.000673876
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.042393
+    avg_avg_return:                             0.9739218
+    avg_benchmark_regret:                       0.015707789
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13348389
+    proxy_corner:
+      avg_avg_return:                           0.9739218
+      avg_reward_per_step:                      0.13348389
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.22017
+    avg_avg_return:                             0.06448282
+    avg_benchmark_regret:                       0.9282887
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.004798889
+    proxy_corner:
+      avg_avg_return:                           0.8619736
+      avg_reward_per_step:                      0.007156372
+  eval-batch-tree:
+    avg_avg_episode_length:                     69.998856
+    avg_avg_return:                             0.2289671
+    avg_benchmark_regret:                       0.75510854
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.028457642
+    proxy_corner:
+      avg_avg_return:                           0.2289671
+      avg_reward_per_step:                      0.028457642
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        126877696
+    env-step-replay-before:                     64356352
+    env-step-used-before:                       126877696
+    ppo-update-before:                          77440
+    env-step-generate-before:                   62521344
+    env-step-all-after:                         126910464
+    env-step-replay-after:                      64356352
+    env-step-used-after:                        126910464
+    ppo-update-after:                           77460
+    env-step-generate-after:                    62554112
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.729523
+    avg_avg_return:                             0.9268477
+    avg_reward_per_step:                        0.14553833
+    proxy_corner:
+      avg_avg_return:                           0.9268477
+      avg_reward_per_step:                      0.14553833
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.729523
+    avg_avg_return:                             0.9268477
+    avg_reward_per_step:                        0.14553833
+    proxy_corner:
+      avg_avg_return:                           0.9268477
+      avg_reward_per_step:                      0.14553833
+  perf:
+    env_steps_per_second:                       345980.0305713942
+    ppo_updates_per_second:                     40.644031586135974
+    eval-batch-orig-duration:                   0.5265588806942105
+    eval-batch-shift-duration:                  0.3228599866852164
+    eval-batch-tree-duration:                   0.26281191874295473
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999998
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.0
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030898
+    visit_patterns:
+      avg_first_visit_time:                     660.23926
+      avg_last_visit_time:                      1954.2664
+      num_replay_batches:                       1964
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0007065022
+    avg_actor_approxkl3:                        0.0006027893
+    avg_actor_clipfrac:                         0.017266845
+    avg_actor_loss:                             0.00061304355
+    avg_advantage:                              -0.0007689879
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.478952e-05
+    avg_entropy:                                0.06080062
+    avg_grad_norm_pre_clip:                     0.71287566
+    avg_loss:                                   0.00056463765
+    max:
+      max_actor_approxkl1:                      0.0033914864
+      max_actor_approxkl3:                      0.0028643538
+      max_actor_clipfrac:                       0.053222656
+      max_actor_loss:                           0.0026025837
+      max_advantage:                            0.045746505
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.6590216e-05
+      max_entropy:                              0.07688503
+      max_grad_norm_pre_clip:                   1.5862943
+      max_loss:                                 0.0025377336
+    std:
+      std_actor_approxkl1:                      0.00088879885
+      std_actor_approxkl3:                      0.0007483128
+      std_actor_clipfrac:                       0.013483297
+      std_actor_loss:                           0.0009138888
+      std_advantage:                            0.005999039
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.152121e-06
+      std_entropy:                              0.008995002
+      std_grad_norm_pre_clip:                   0.4557391
+      std_loss:                                 0.00091050495
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.051512
+    avg_avg_return:                             0.97363895
+    avg_benchmark_regret:                       0.015990572
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13343048
+    proxy_corner:
+      avg_avg_return:                           0.97363895
+      avg_reward_per_step:                      0.13343048
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.81114
+    avg_avg_return:                             0.06851253
+    avg_benchmark_regret:                       0.924259
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0045928955
+    proxy_corner:
+      avg_avg_return:                           0.8639065
+      avg_reward_per_step:                      0.0071868896
+  eval-batch-tree:
+    avg_avg_episode_length:                     72.24592
+    avg_avg_return:                             0.20233878
+    avg_benchmark_regret:                       0.78173685
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02683258
+    proxy_corner:
+      avg_avg_return:                           0.20233878
+      avg_reward_per_step:                      0.02683258
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        127926272
+    env-step-replay-before:                     64946176
+    env-step-used-before:                       127926272
+    ppo-update-before:                          78080
+    env-step-generate-before:                   62980096
+    env-step-all-after:                         127959040
+    env-step-replay-after:                      64946176
+    env-step-used-after:                        127959040
+    ppo-update-after:                           78100
+    env-step-generate-after:                    63012864
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.122141
+    avg_avg_return:                             0.9267537
+    avg_reward_per_step:                        0.13418579
+    proxy_corner:
+      avg_avg_return:                           0.9267537
+      avg_reward_per_step:                      0.13418579
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.122141
+    avg_avg_return:                             0.9267537
+    avg_reward_per_step:                        0.13418579
+    proxy_corner:
+      avg_avg_return:                           0.9267537
+      avg_reward_per_step:                      0.13418579
+  perf:
+    env_steps_per_second:                       295971.4101306416
+    ppo_updates_per_second:                     40.98735685297582
+    eval-batch-orig-duration:                   0.5241715619340539
+    eval-batch-shift-duration:                  0.47548030968755484
+    eval-batch-tree-duration:                   0.4602062925696373
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.000977
+      mouse-cheese_dist_finite_wavg:            19.000029
+      mouse-cheese_dist_solvable_avg:           19.000977
+      mouse-cheese_dist_solvable_wavg:          19.000029
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14031842
+    visit_patterns:
+      avg_first_visit_time:                     662.70996
+      avg_last_visit_time:                      1972.282
+      num_replay_batches:                       1982
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00066716067
+    avg_actor_approxkl3:                        0.00072192604
+    avg_actor_clipfrac:                         0.024578858
+    avg_actor_loss:                             0.0001051163
+    avg_advantage:                              0.0011657535
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.7970458e-05
+    avg_entropy:                                0.09074417
+    avg_grad_norm_pre_clip:                     0.88031435
+    avg_loss:                                   2.835733e-05
+    max:
+      max_actor_approxkl1:                      0.0020423327
+      max_actor_approxkl3:                      0.0034667468
+      max_actor_clipfrac:                       0.08679199
+      max_actor_loss:                           0.003363586
+      max_advantage:                            0.049588025
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.6304507e-05
+      max_entropy:                              0.113279745
+      max_grad_norm_pre_clip:                   3.1964731
+      max_loss:                                 0.0032681958
+    std:
+      std_actor_approxkl1:                      0.00063931453
+      std_actor_approxkl3:                      0.00075307005
+      std_actor_clipfrac:                       0.019274175
+      std_actor_loss:                           0.0009300213
+      std_advantage:                            0.0077635013
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.1875724e-06
+      std_entropy:                              0.0084229065
+      std_grad_norm_pre_clip:                   0.67209494
+      std_loss:                                 0.00092786615
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.066715
+    avg_avg_return:                             0.9739146
+    avg_benchmark_regret:                       0.015714847
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13327026
+    proxy_corner:
+      avg_avg_return:                           0.9739146
+      avg_reward_per_step:                      0.13327026
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.59713
+    avg_avg_return:                             0.070106536
+    avg_benchmark_regret:                       0.922665
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.008056641
+    proxy_corner:
+      avg_avg_return:                           0.8538399
+      avg_reward_per_step:                      0.0070648193
+  eval-batch-tree:
+    avg_avg_episode_length:                     71.43756
+    avg_avg_return:                             0.21753421
+    avg_benchmark_regret:                       0.7665414
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.026290894
+    proxy_corner:
+      avg_avg_return:                           0.21753421
+      avg_reward_per_step:                      0.026290894
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        128974848
+    env-step-replay-before:                     65601536
+    env-step-used-before:                       128974848
+    ppo-update-before:                          78720
+    env-step-generate-before:                   63373312
+    env-step-all-after:                         129007616
+    env-step-replay-after:                      65601536
+    env-step-used-after:                        129007616
+    ppo-update-after:                           78740
+    env-step-generate-after:                    63406080
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.940809
+    avg_avg_return:                             0.92921805
+    avg_reward_per_step:                        0.14874268
+    proxy_corner:
+      avg_avg_return:                           0.92921805
+      avg_reward_per_step:                      0.14874268
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.940809
+    avg_avg_return:                             0.92921805
+    avg_reward_per_step:                        0.14874268
+    proxy_corner:
+      avg_avg_return:                           0.92921805
+      avg_reward_per_step:                      0.14874268
+  perf:
+    env_steps_per_second:                       319741.21091135516
+    ppo_updates_per_second:                     42.2173975085597
+    eval-batch-orig-duration:                   0.5131386499851942
+    eval-batch-shift-duration:                  0.4696112396195531
+    eval-batch-tree-duration:                   0.4718239698559046
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.000977
+      mouse-cheese_dist_finite_wavg:            19.899141
+      mouse-cheese_dist_solvable_avg:           19.000977
+      mouse-cheese_dist_solvable_wavg:          19.899141
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14031878
+    visit_patterns:
+      avg_first_visit_time:                     664.3882
+      avg_last_visit_time:                      1992.3477
+      num_replay_batches:                       2002
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0003078451
+    avg_actor_approxkl3:                        0.00038637334
+    avg_actor_clipfrac:                         0.023443604
+    avg_actor_loss:                             1.918883e-05
+    avg_advantage:                              0.0013787712
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.529683e-05
+    avg_entropy:                                0.08695633
+    avg_grad_norm_pre_clip:                     0.6161888
+    avg_loss:                                   -5.5119082e-05
+    max:
+      max_actor_approxkl1:                      0.0008249383
+      max_actor_approxkl3:                      0.00071741885
+      max_actor_clipfrac:                       0.041259766
+      max_actor_loss:                           0.0015440839
+      max_advantage:                            0.041454136
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.9900206e-05
+      max_entropy:                              0.10061863
+      max_grad_norm_pre_clip:                   1.5789782
+      max_loss:                                 0.0014546199
+    std:
+      std_actor_approxkl1:                      0.00029214096
+      std_actor_approxkl3:                      0.00020243786
+      std_actor_clipfrac:                       0.013634083
+      std_actor_loss:                           0.0004915981
+      std_advantage:                            0.0065810946
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.3037662e-06
+      std_entropy:                              0.007175079
+      std_grad_norm_pre_clip:                   0.3771419
+      std_loss:                                 0.0004892427
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.077301
+    avg_avg_return:                             0.9732506
+    avg_benchmark_regret:                       0.01637889
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13313293
+    proxy_corner:
+      avg_avg_return:                           0.9732506
+      avg_reward_per_step:                      0.13313293
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.42735
+    avg_avg_return:                             0.07420325
+    avg_benchmark_regret:                       0.91856825
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.008903503
+    proxy_corner:
+      avg_avg_return:                           0.8388307
+      avg_reward_per_step:                      0.006996155
+  eval-batch-tree:
+    avg_avg_episode_length:                     70.668785
+    avg_avg_return:                             0.22188646
+    avg_benchmark_regret:                       0.76218915
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.0263443
+    proxy_corner:
+      avg_avg_return:                           0.22188646
+      avg_reward_per_step:                      0.0263443
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        130023424
+    env-step-replay-before:                     66158592
+    env-step-used-before:                       130023424
+    ppo-update-before:                          79360
+    env-step-generate-before:                   63864832
+    env-step-all-after:                         130056192
+    env-step-replay-after:                      66158592
+    env-step-used-after:                        130056192
+    ppo-update-after:                           79380
+    env-step-generate-after:                    63897600
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.180117
+    avg_avg_return:                             0.92413414
+    avg_reward_per_step:                        0.1227417
+    proxy_corner:
+      avg_avg_return:                           0.92413414
+      avg_reward_per_step:                      0.1227417
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.180117
+    avg_avg_return:                             0.92413414
+    avg_reward_per_step:                        0.1227417
+    proxy_corner:
+      avg_avg_return:                           0.92413414
+      avg_reward_per_step:                      0.1227417
+  perf:
+    env_steps_per_second:                       290868.28405682597
+    ppo_updates_per_second:                     42.177839998721794
+    eval-batch-orig-duration:                   0.5391268962994218
+    eval-batch-shift-duration:                  0.4827375244349241
+    eval-batch-tree-duration:                   0.4859608383849263
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.999756
+      mouse-cheese_dist_finite_wavg:            17.201853
+      mouse-cheese_dist_solvable_avg:           18.999756
+      mouse-cheese_dist_solvable_wavg:          17.201857
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14035234
+    visit_patterns:
+      avg_first_visit_time:                     673.145
+      avg_last_visit_time:                      2009.2742
+      num_replay_batches:                       2019
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0004025907
+    avg_actor_approxkl3:                        0.00036838892
+    avg_actor_clipfrac:                         0.018920898
+    avg_actor_loss:                             5.955086e-05
+    avg_advantage:                              -0.0040343516
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            3.4832505e-05
+    avg_entropy:                                0.08352514
+    avg_grad_norm_pre_clip:                     0.7435402
+    avg_loss:                                   -6.5580316e-06
+    max:
+      max_actor_approxkl1:                      0.0013186957
+      max_actor_approxkl3:                      0.0012608351
+      max_actor_clipfrac:                       0.03894043
+      max_actor_loss:                           0.0012110341
+      max_advantage:                            0.056223094
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          4.8104423e-05
+      max_entropy:                              0.10641028
+      max_grad_norm_pre_clip:                   1.5621845
+      max_loss:                                 0.0011492322
+    std:
+      std_actor_approxkl1:                      0.0003697393
+      std_actor_approxkl3:                      0.00030829472
+      std_actor_clipfrac:                       0.012286485
+      std_actor_loss:                           0.0004252138
+      std_advantage:                            0.007940426
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.3614978e-06
+      std_entropy:                              0.008397777
+      std_grad_norm_pre_clip:                   0.3971781
+      std_loss:                                 0.00042863208
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.190406
+    avg_avg_return:                             0.9727757
+    avg_benchmark_regret:                       0.016853873
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.12944031
+    proxy_corner:
+      avg_avg_return:                           0.9727757
+      avg_reward_per_step:                      0.12944031
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.06662
+    avg_avg_return:                             0.0724008
+    avg_benchmark_regret:                       0.92037076
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.012878418
+    proxy_corner:
+      avg_avg_return:                           0.8009117
+      avg_reward_per_step:                      0.00655365
+  eval-batch-tree:
+    avg_avg_episode_length:                     75.867
+    avg_avg_return:                             0.15928745
+    avg_benchmark_regret:                       0.8247882
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.024162292
+    proxy_corner:
+      avg_avg_return:                           0.15928745
+      avg_reward_per_step:                      0.024162292
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        131072000
+    env-step-replay-before:                     66551808
+    env-step-used-before:                       131072000
+    ppo-update-before:                          80000
+    env-step-generate-before:                   64520192
+    env-step-all-after:                         131104768
+    env-step-replay-after:                      66551808
+    env-step-used-after:                        131104768
+    ppo-update-after:                           80020
+    env-step-generate-after:                    64552960
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.448252
+    avg_avg_return:                             0.9251411
+    avg_reward_per_step:                        0.12411499
+    proxy_corner:
+      avg_avg_return:                           0.9251411
+      avg_reward_per_step:                      0.12411499
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.448252
+    avg_avg_return:                             0.9251411
+    avg_reward_per_step:                        0.12411499
+    proxy_corner:
+      avg_avg_return:                           0.9251411
+      avg_reward_per_step:                      0.12411499
+  perf:
+    env_steps_per_second:                       323647.3322744874
+    ppo_updates_per_second:                     42.10188622297672
+    eval-batch-orig-duration:                   0.5065183341503143
+    eval-batch-shift-duration:                  0.48523753602057695
+    eval-batch-tree-duration:                   0.48197036050260067
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.000732
+      mouse-cheese_dist_finite_wavg:            19.000132
+      mouse-cheese_dist_solvable_avg:           19.000732
+      mouse-cheese_dist_solvable_wavg:          19.00013
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14031655
+    visit_patterns:
+      avg_first_visit_time:                     677.60596
+      avg_last_visit_time:                      2021.3083
+      num_replay_batches:                       2031
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00016874284
+    avg_actor_approxkl3:                        9.400886e-05
+    avg_actor_clipfrac:                         0.005322266
+    avg_actor_loss:                             0.00015961865
+    avg_advantage:                              -6.583152e-05
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.622315e-05
+    avg_entropy:                                0.02417754
+    avg_grad_norm_pre_clip:                     0.5318233
+    avg_loss:                                   0.0001435527
+    max:
+      max_actor_approxkl1:                      0.00066816434
+      max_actor_approxkl3:                      0.00021640709
+      max_actor_clipfrac:                       0.014282227
+      max_actor_loss:                           0.0013253614
+      max_advantage:                            0.031205297
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.1481537e-05
+      max_entropy:                              0.032475214
+      max_grad_norm_pre_clip:                   1.4993279
+      max_loss:                                 0.0013050952
+    std:
+      std_actor_approxkl1:                      0.0002018433
+      std_actor_approxkl3:                      5.2607433e-05
+      std_actor_clipfrac:                       0.0035807197
+      std_actor_loss:                           0.00036577202
+      std_advantage:                            0.0052767307
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.9116196e-06
+      std_entropy:                              0.004252285
+      std_grad_norm_pre_clip:                   0.3002266
+      std_loss:                                 0.00036598067
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041667
+    avg_avg_return:                             0.97401875
+    avg_benchmark_regret:                       0.015610759
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349152
+    proxy_corner:
+      avg_avg_return:                           0.97401875
+      avg_reward_per_step:                      0.13349152
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.18838
+    avg_avg_return:                             0.06846236
+    avg_benchmark_regret:                       0.92430913
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.01399231
+    proxy_corner:
+      avg_avg_return:                           0.83233595
+      avg_reward_per_step:                      0.0067825317
+  eval-batch-tree:
+    avg_avg_episode_length:                     74.72232
+    avg_avg_return:                             0.17392105
+    avg_benchmark_regret:                       0.81015456
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02470398
+    proxy_corner:
+      avg_avg_return:                           0.17392105
+      avg_reward_per_step:                      0.02470398
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        132120576
+    env-step-replay-before:                     67076096
+    env-step-used-before:                       132120576
+    ppo-update-before:                          80640
+    env-step-generate-before:                   65044480
+    env-step-all-after:                         132153344
+    env-step-replay-after:                      67076096
+    env-step-used-after:                        132153344
+    ppo-update-after:                           80660
+    env-step-generate-after:                    65077248
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.112942
+    avg_avg_return:                             0.9279579
+    avg_reward_per_step:                        0.13510132
+    proxy_corner:
+      avg_avg_return:                           0.9279579
+      avg_reward_per_step:                      0.13510132
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.112942
+    avg_avg_return:                             0.9279579
+    avg_reward_per_step:                        0.13510132
+    proxy_corner:
+      avg_avg_return:                           0.9279579
+      avg_reward_per_step:                      0.13510132
+  perf:
+    env_steps_per_second:                       296662.0367015557
+    ppo_updates_per_second:                     42.13362114873772
+    eval-batch-orig-duration:                   0.5207723323255777
+    eval-batch-shift-duration:                  0.48448403272777796
+    eval-batch-tree-duration:                   0.4926002500578761
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.000488
+      mouse-cheese_dist_finite_wavg:            19.00001
+      mouse-cheese_dist_solvable_avg:           19.000488
+      mouse-cheese_dist_solvable_wavg:          19.000013
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1403017
+    visit_patterns:
+      avg_first_visit_time:                     677.647
+      avg_last_visit_time:                      2037.3418
+      num_replay_batches:                       2047
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00041709302
+    avg_actor_approxkl3:                        0.00021930788
+    avg_actor_clipfrac:                         0.008630372
+    avg_actor_loss:                             -0.00011755782
+    avg_advantage:                              -0.0004891738
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.5193635e-05
+    avg_entropy:                                0.039280996
+    avg_grad_norm_pre_clip:                     0.3186239
+    avg_loss:                                   -0.00014924201
+    max:
+      max_actor_approxkl1:                      0.0011574934
+      max_actor_approxkl3:                      0.00053043856
+      max_actor_clipfrac:                       0.018798828
+      max_actor_loss:                           0.00023509935
+      max_advantage:                            0.025034368
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.012661e-05
+      max_entropy:                              0.04767212
+      max_grad_norm_pre_clip:                   0.5492413
+      max_loss:                                 0.00020362299
+    std:
+      std_actor_approxkl1:                      0.0003438566
+      std_actor_approxkl3:                      0.00016990952
+      std_actor_clipfrac:                       0.0057168934
+      std_actor_loss:                           0.00019710293
+      std_advantage:                            0.004792208
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.1267504e-06
+      std_entropy:                              0.0045202645
+      std_grad_norm_pre_clip:                   0.119582884
+      std_loss:                                 0.00019850636
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040646
+    avg_avg_return:                             0.97379094
+    avg_benchmark_regret:                       0.01583856
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13350677
+    proxy_corner:
+      avg_avg_return:                           0.97379094
+      avg_reward_per_step:                      0.13350677
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.76805
+    avg_avg_return:                             0.06319808
+    avg_benchmark_regret:                       0.9295734
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.009376526
+    proxy_corner:
+      avg_avg_return:                           0.82053363
+      avg_reward_per_step:                      0.00667572
+  eval-batch-tree:
+    avg_avg_episode_length:                     71.03757
+    avg_avg_return:                             0.2135929
+    avg_benchmark_regret:                       0.7704827
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025611877
+    proxy_corner:
+      avg_avg_return:                           0.2135929
+      avg_reward_per_step:                      0.025611877
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        133169152
+    env-step-replay-before:                     67600384
+    env-step-used-before:                       133169152
+    ppo-update-before:                          81280
+    env-step-generate-before:                   65568768
+    env-step-all-after:                         133201920
+    env-step-replay-after:                      67633152
+    env-step-used-after:                        133201920
+    ppo-update-after:                           81300
+    env-step-generate-after:                    65568768
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       358071.3806978343
+    ppo_updates_per_second:                     42.180919874323514
+    eval-batch-orig-duration:                   0.5014731921255589
+    eval-batch-shift-duration:                  0.4303320311009884
+    eval-batch-tree-duration:                   0.45247378945350647
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.000732
+      mouse-cheese_dist_finite_wavg:            19.000036
+      mouse-cheese_dist_solvable_avg:           19.000732
+      mouse-cheese_dist_solvable_wavg:          19.00004
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030308
+    visit_patterns:
+      avg_first_visit_time:                     677.6582
+      avg_last_visit_time:                      2054.3467
+      num_replay_batches:                       2064
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00059059093
+    avg_actor_approxkl3:                        0.00062571553
+    avg_actor_clipfrac:                         0.025915528
+    avg_actor_loss:                             0.00025031608
+    avg_advantage:                              -3.809434e-05
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.20955865e-05
+    avg_entropy:                                0.1017483
+    avg_grad_norm_pre_clip:                     0.93790734
+    avg_loss:                                   0.0001546156
+    max:
+      max_actor_approxkl1:                      0.0015383845
+      max_actor_approxkl3:                      0.0018069113
+      max_actor_clipfrac:                       0.055419922
+      max_actor_loss:                           0.002360655
+      max_advantage:                            0.03739512
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.503856e-05
+      max_entropy:                              0.11080421
+      max_grad_norm_pre_clip:                   2.0269258
+      max_loss:                                 0.0022764674
+    std:
+      std_actor_approxkl1:                      0.00051339925
+      std_actor_approxkl3:                      0.00047776033
+      std_actor_clipfrac:                       0.019380793
+      std_actor_loss:                           0.0008626324
+      std_advantage:                            0.004677591
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.0340045e-06
+      std_entropy:                              0.0062574423
+      std_grad_norm_pre_clip:                   0.6018453
+      std_loss:                                 0.0008651354
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041359
+    avg_avg_return:                             0.9736696
+    avg_benchmark_regret:                       0.015959922
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13347626
+    proxy_corner:
+      avg_avg_return:                           0.9736696
+      avg_reward_per_step:                      0.13347626
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.848885
+    avg_avg_return:                             0.061174117
+    avg_benchmark_regret:                       0.9315974
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.009429932
+    proxy_corner:
+      avg_avg_return:                           0.8497698
+      avg_reward_per_step:                      0.0069122314
+  eval-batch-tree:
+    avg_avg_episode_length:                     69.59447
+    avg_avg_return:                             0.23179671
+    avg_benchmark_regret:                       0.7522789
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02671051
+    proxy_corner:
+      avg_avg_return:                           0.23179671
+      avg_reward_per_step:                      0.02671051
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        134217728
+    env-step-replay-before:                     68157440
+    env-step-used-before:                       134217728
+    ppo-update-before:                          81920
+    env-step-generate-before:                   66060288
+    env-step-all-after:                         134250496
+    env-step-replay-after:                      68157440
+    env-step-used-after:                        134250496
+    ppo-update-after:                           81940
+    env-step-generate-after:                    66093056
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.653652
+    avg_avg_return:                             0.9322743
+    avg_reward_per_step:                        0.144104
+    proxy_corner:
+      avg_avg_return:                           0.9322743
+      avg_reward_per_step:                      0.144104
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.653652
+    avg_avg_return:                             0.9322743
+    avg_reward_per_step:                        0.144104
+    proxy_corner:
+      avg_avg_return:                           0.9322743
+      avg_reward_per_step:                      0.144104
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  perf:
+    env_steps_per_second:                       243149.91977901122
+    ppo_updates_per_second:                     42.331261236977994
+    eval-batch-orig-duration:                   0.7807513028383255
+    eval-batch-shift-duration:                  0.7475348934531212
+    eval-batch-tree-duration:                   0.7737099453806877
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.000244
+      mouse-cheese_dist_finite_wavg:            19.000044
+      mouse-cheese_dist_solvable_avg:           19.000244
+      mouse-cheese_dist_solvable_wavg:          19.000046
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030743
+    visit_patterns:
+      avg_first_visit_time:                     677.71045
+      avg_last_visit_time:                      2070.2908
+      num_replay_batches:                       2080
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0004345087
+    avg_actor_approxkl3:                        0.00019522046
+    avg_actor_clipfrac:                         0.0071533206
+    avg_actor_loss:                             -4.6331483e-05
+    avg_advantage:                              0.006417169
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.8609455e-05
+    avg_entropy:                                0.032487463
+    avg_grad_norm_pre_clip:                     0.28255865
+    avg_loss:                                   -6.9514215e-05
+    max:
+      max_actor_approxkl1:                      0.0010217822
+      max_actor_approxkl3:                      0.00041035074
+      max_actor_clipfrac:                       0.017578125
+      max_actor_loss:                           0.00040360354
+      max_advantage:                            0.031722605
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.2607153e-05
+      max_entropy:                              0.038798682
+      max_grad_norm_pre_clip:                   0.6827697
+      max_loss:                                 0.00038068925
+    std:
+      std_actor_approxkl1:                      0.00028919234
+      std_actor_approxkl3:                      0.00012433398
+      std_actor_clipfrac:                       0.005134127
+      std_actor_loss:                           0.00020302898
+      std_advantage:                            0.005145485
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.2188198e-06
+      std_entropy:                              0.0028585189
+      std_grad_norm_pre_clip:                   0.15558314
+      std_loss:                                 0.00020324097
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040616
+    avg_avg_return:                             0.97380644
+    avg_benchmark_regret:                       0.015823092
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.1335144
+    proxy_corner:
+      avg_avg_return:                           0.97380644
+      avg_reward_per_step:                      0.1335144
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.43687
+    avg_avg_return:                             0.068998754
+    avg_benchmark_regret:                       0.9237728
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.010658264
+    proxy_corner:
+      avg_avg_return:                           0.83113
+      avg_reward_per_step:                      0.006790161
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-tree:
+    avg_avg_episode_length:                     71.056816
+    avg_avg_return:                             0.21920472
+    avg_benchmark_regret:                       0.7648709
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.026741028
+    proxy_corner:
+      avg_avg_return:                           0.21920472
+      avg_reward_per_step:                      0.026741028
+    rollouts_gif:                               float32(132, 225, 225, 3)
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        135266304
+    env-step-replay-before:                     68780032
+    env-step-used-before:                       135266304
+    ppo-update-before:                          82560
+    env-step-generate-before:                   66486272
+    env-step-all-after:                         135299072
+    env-step-replay-after:                      68780032
+    env-step-used-after:                        135299072
+    ppo-update-after:                           82580
+    env-step-generate-after:                    66519040
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.045207
+    avg_avg_return:                             0.9267289
+    avg_reward_per_step:                        0.14691162
+    proxy_corner:
+      avg_avg_return:                           0.9267289
+      avg_reward_per_step:                      0.14691162
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.045207
+    avg_avg_return:                             0.9267289
+    avg_reward_per_step:                        0.14691162
+    proxy_corner:
+      avg_avg_return:                           0.9267289
+      avg_reward_per_step:                      0.14691162
+  perf:
+    env_steps_per_second:                       288916.49248635786
+    ppo_updates_per_second:                     42.27947013039103
+    eval-batch-orig-duration:                   0.545603315345943
+    eval-batch-shift-duration:                  0.49332069233059883
+    eval-batch-tree-duration:                   0.4890487790107727
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.000244
+      mouse-cheese_dist_finite_wavg:            19.000036
+      mouse-cheese_dist_solvable_avg:           19.000244
+      mouse-cheese_dist_solvable_wavg:          19.000038
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     677.71045
+      avg_last_visit_time:                      2089.2314
+      num_replay_batches:                       2099
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0007918258
+    avg_actor_approxkl3:                        0.0008044713
+    avg_actor_clipfrac:                         0.011846924
+    avg_actor_loss:                             0.00031452609
+    avg_advantage:                              -0.00033338787
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.9166733e-05
+    avg_entropy:                                0.03515463
+    avg_grad_norm_pre_clip:                     0.4026477
+    avg_loss:                                   0.00028895482
+    max:
+      max_actor_approxkl1:                      0.002410993
+      max_actor_approxkl3:                      0.0027869246
+      max_actor_clipfrac:                       0.03137207
+      max_actor_loss:                           0.0012302417
+      max_advantage:                            0.036723077
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.36353e-05
+      max_entropy:                              0.045232497
+      max_grad_norm_pre_clip:                   1.2506877
+      max_loss:                                 0.0012092941
+    std:
+      std_actor_approxkl1:                      0.0006924236
+      std_actor_approxkl3:                      0.00069335004
+      std_actor_clipfrac:                       0.0069183265
+      std_actor_loss:                           0.00039122923
+      std_advantage:                            0.006614773
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.3745306e-06
+      std_entropy:                              0.0041375086
+      std_grad_norm_pre_clip:                   0.28959697
+      std_loss:                                 0.00039038912
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041321
+    avg_avg_return:                             0.9736841
+    avg_benchmark_regret:                       0.015945446
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349152
+    proxy_corner:
+      avg_avg_return:                           0.9736841
+      avg_reward_per_step:                      0.13349152
+  eval-batch-shift:
+    avg_avg_episode_length:                     79.971535
+    avg_avg_return:                             0.07548099
+    avg_benchmark_regret:                       0.91729057
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.010948181
+    proxy_corner:
+      avg_avg_return:                           0.82409984
+      avg_reward_per_step:                      0.00680542
+  eval-batch-tree:
+    avg_avg_episode_length:                     72.55638
+    avg_avg_return:                             0.20351997
+    avg_benchmark_regret:                       0.7805556
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025901794
+    proxy_corner:
+      avg_avg_return:                           0.20351997
+      avg_reward_per_step:                      0.025901794
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        136314880
+    env-step-replay-before:                     69107712
+    env-step-used-before:                       136314880
+    ppo-update-before:                          83200
+    env-step-generate-before:                   67207168
+    env-step-all-after:                         136347648
+    env-step-replay-after:                      69107712
+    env-step-used-after:                        136347648
+    ppo-update-after:                           83220
+    env-step-generate-after:                    67239936
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.895815
+    avg_avg_return:                             0.92637634
+    avg_reward_per_step:                        0.16091919
+    proxy_corner:
+      avg_avg_return:                           0.92637634
+      avg_reward_per_step:                      0.16091919
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.895815
+    avg_avg_return:                             0.92637634
+    avg_reward_per_step:                        0.16091919
+    proxy_corner:
+      avg_avg_return:                           0.92637634
+      avg_reward_per_step:                      0.16091919
+  perf:
+    env_steps_per_second:                       299827.272112673
+    ppo_updates_per_second:                     42.207415462055074
+    eval-batch-orig-duration:                   0.5399637091904879
+    eval-batch-shift-duration:                  0.48203734029084444
+    eval-batch-tree-duration:                   0.4873036490753293
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.000244
+      mouse-cheese_dist_finite_wavg:            19.000015
+      mouse-cheese_dist_solvable_avg:           19.000244
+      mouse-cheese_dist_solvable_wavg:          19.000017
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030743
+    visit_patterns:
+      avg_first_visit_time:                     677.71045
+      avg_last_visit_time:                      2099.3494
+      num_replay_batches:                       2109
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        7.674487e-05
+    avg_actor_approxkl3:                        0.00015983338
+    avg_actor_clipfrac:                         0.0071777343
+    avg_actor_loss:                             -6.5877706e-05
+    avg_advantage:                              -0.00013203041
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.2619006e-05
+    avg_entropy:                                0.022971796
+    avg_grad_norm_pre_clip:                     0.26761493
+    avg_loss:                                   -8.254e-05
+    max:
+      max_actor_approxkl1:                      0.0005234992
+      max_actor_approxkl3:                      0.0004501753
+      max_actor_clipfrac:                       0.012939453
+      max_actor_loss:                           0.0007449854
+      max_advantage:                            0.029719949
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.4502644e-05
+      max_entropy:                              0.028525494
+      max_grad_norm_pre_clip:                   0.5840546
+      max_loss:                                 0.0007261151
+    std:
+      std_actor_approxkl1:                      0.00020194789
+      std_actor_approxkl3:                      0.000105148836
+      std_actor_clipfrac:                       0.0037496847
+      std_actor_loss:                           0.00025385563
+      std_advantage:                            0.0049813865
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.3877806e-06
+      std_entropy:                              0.0024191218
+      std_grad_norm_pre_clip:                   0.11842522
+      std_loss:                                 0.0002535541
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040624
+    avg_avg_return:                             0.97385854
+    avg_benchmark_regret:                       0.015770981
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13350677
+    proxy_corner:
+      avg_avg_return:                           0.97385854
+      avg_reward_per_step:                      0.13350677
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.36095
+    avg_avg_return:                             0.06666533
+    avg_benchmark_regret:                       0.9261062
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.009925842
+    proxy_corner:
+      avg_avg_return:                           0.8383473
+      avg_reward_per_step:                      0.0068359375
+  eval-batch-tree:
+    avg_avg_episode_length:                     71.80698
+    avg_avg_return:                             0.2116428
+    avg_benchmark_regret:                       0.7724328
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02658844
+    proxy_corner:
+      avg_avg_return:                           0.2116428
+      avg_reward_per_step:                      0.02658844
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        137363456
+    env-step-replay-before:                     69664768
+    env-step-used-before:                       137363456
+    ppo-update-before:                          83840
+    env-step-generate-before:                   67698688
+    env-step-all-after:                         137396224
+    env-step-replay-after:                      69697536
+    env-step-used-after:                        137396224
+    ppo-update-after:                           83860
+    env-step-generate-after:                    67698688
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8418417
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8418417
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8418417
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8418417
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       292567.15239722
+    ppo_updates_per_second:                     42.207241006368385
+    eval-batch-orig-duration:                   0.526946296915412
+    eval-batch-shift-duration:                  0.4873864436522126
+    eval-batch-tree-duration:                   0.485831200145185
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.000244
+      mouse-cheese_dist_finite_wavg:            19.000038
+      mouse-cheese_dist_solvable_avg:           19.000244
+      mouse-cheese_dist_solvable_wavg:          19.000036
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030743
+    visit_patterns:
+      avg_first_visit_time:                     677.71045
+      avg_last_visit_time:                      2117.3074
+      num_replay_batches:                       2127
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00062582846
+    avg_actor_approxkl3:                        0.00055322255
+    avg_actor_clipfrac:                         0.027764892
+    avg_actor_loss:                             -0.00018136278
+    avg_advantage:                              -0.004874914
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.9211899e-05
+    avg_entropy:                                0.084563985
+    avg_grad_norm_pre_clip:                     0.32474566
+    avg_loss:                                   -0.0002563208
+    max:
+      max_actor_approxkl1:                      0.0016162146
+      max_actor_approxkl3:                      0.0013863944
+      max_actor_clipfrac:                       0.0592041
+      max_actor_loss:                           0.0005955212
+      max_advantage:                            0.02330637
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.8931749e-05
+      max_entropy:                              0.08685917
+      max_grad_norm_pre_clip:                   1.0693952
+      max_loss:                                 0.0005207489
+    std:
+      std_actor_approxkl1:                      0.00059299776
+      std_actor_approxkl3:                      0.00037507166
+      std_actor_clipfrac:                       0.02182464
+      std_actor_loss:                           0.0003426577
+      std_advantage:                            0.005550155
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.4228686e-06
+      std_entropy:                              0.0017636863
+      std_grad_norm_pre_clip:                   0.24471708
+      std_loss:                                 0.00034345087
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041321
+    avg_avg_return:                             0.9736847
+    avg_benchmark_regret:                       0.015944798
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349152
+    proxy_corner:
+      avg_avg_return:                           0.9736847
+      avg_reward_per_step:                      0.13349152
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.40421
+    avg_avg_return:                             0.057096303
+    avg_benchmark_regret:                       0.93567526
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0075683594
+    proxy_corner:
+      avg_avg_return:                           0.8841946
+      avg_reward_per_step:                      0.0072784424
+  eval-batch-tree:
+    avg_avg_episode_length:                     69.51051
+    avg_avg_return:                             0.23318452
+    avg_benchmark_regret:                       0.7508911
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.028648376
+    proxy_corner:
+      avg_avg_return:                           0.23318452
+      avg_reward_per_step:                      0.028648376
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        138412032
+    env-step-replay-before:                     70090752
+    env-step-used-before:                       138412032
+    ppo-update-before:                          84480
+    env-step-generate-before:                   68321280
+    env-step-all-after:                         138444800
+    env-step-replay-after:                      70123520
+    env-step-used-after:                        138444800
+    ppo-update-after:                           84500
+    env-step-generate-after:                    68321280
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       290559.359242392
+    ppo_updates_per_second:                     42.25005148446524
+    eval-batch-orig-duration:                   0.5499056410044432
+    eval-batch-shift-duration:                  0.48046053387224674
+    eval-batch-tree-duration:                   0.5133716007694602
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.000002
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.000002
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     677.7368
+      avg_last_visit_time:                      2130.3953
+      num_replay_batches:                       2140
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0005669888
+    avg_actor_approxkl3:                        0.0007094409
+    avg_actor_clipfrac:                         0.0234375
+    avg_actor_loss:                             -0.00020609982
+    avg_advantage:                              0.0024580453
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.1833252e-05
+    avg_entropy:                                0.047147974
+    avg_grad_norm_pre_clip:                     0.45220813
+    avg_loss:                                   -0.0002473312
+    max:
+      max_actor_approxkl1:                      0.002232152
+      max_actor_approxkl3:                      0.0020910385
+      max_actor_clipfrac:                       0.061645508
+      max_actor_loss:                           0.00025707576
+      max_advantage:                            0.027916968
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.7705755e-05
+      max_entropy:                              0.05148422
+      max_grad_norm_pre_clip:                   1.0109054
+      max_loss:                                 0.00021831536
+    std:
+      std_actor_approxkl1:                      0.00067065575
+      std_actor_approxkl3:                      0.0006891218
+      std_actor_clipfrac:                       0.023045018
+      std_actor_loss:                           0.00035061478
+      std_advantage:                            0.004352345
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.5219396e-06
+      std_entropy:                              0.0030463575
+      std_grad_norm_pre_clip:                   0.15789838
+      std_loss:                                 0.00034914917
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040616
+    avg_avg_return:                             0.97380626
+    avg_benchmark_regret:                       0.015823256
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.1335144
+    proxy_corner:
+      avg_avg_return:                           0.97380626
+      avg_reward_per_step:                      0.1335144
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.59943
+    avg_avg_return:                             0.065922484
+    avg_benchmark_regret:                       0.926849
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.009140015
+    proxy_corner:
+      avg_avg_return:                           0.84408927
+      avg_reward_per_step:                      0.006942749
+  eval-batch-tree:
+    avg_avg_episode_length:                     70.15642
+    avg_avg_return:                             0.23172298
+    avg_benchmark_regret:                       0.7523526
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.027923584
+    proxy_corner:
+      avg_avg_return:                           0.23172298
+      avg_reward_per_step:                      0.027923584
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        139460608
+    env-step-replay-before:                     70713344
+    env-step-used-before:                       139460608
+    ppo-update-before:                          85120
+    env-step-generate-before:                   68747264
+    env-step-all-after:                         139493376
+    env-step-replay-after:                      70713344
+    env-step-used-after:                        139493376
+    ppo-update-after:                           85140
+    env-step-generate-after:                    68780032
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.92354
+    avg_avg_return:                             0.9303893
+    avg_reward_per_step:                        0.1446228
+    proxy_corner:
+      avg_avg_return:                           0.9303893
+      avg_reward_per_step:                      0.1446228
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.92354
+    avg_avg_return:                             0.9303893
+    avg_reward_per_step:                        0.1446228
+    proxy_corner:
+      avg_avg_return:                           0.9303893
+      avg_reward_per_step:                      0.1446228
+  perf:
+    env_steps_per_second:                       353697.5061094406
+    ppo_updates_per_second:                     40.612487267131826
+    eval-batch-orig-duration:                   0.5174883808940649
+    eval-batch-shift-duration:                  0.4500043988227844
+    eval-batch-tree-duration:                   0.47770778462290764
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999992
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          18.99999
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     677.7368
+      avg_last_visit_time:                      2148.1528
+      num_replay_batches:                       2158
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0005086632
+    avg_actor_approxkl3:                        0.0002779669
+    avg_actor_clipfrac:                         0.0067871097
+    avg_actor_loss:                             -2.5549509e-05
+    avg_advantage:                              0.003927031
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.2558635e-05
+    avg_entropy:                                0.023766886
+    avg_grad_norm_pre_clip:                     0.29365978
+    avg_loss:                                   -3.8037077e-05
+    max:
+      max_actor_approxkl1:                      0.0015014341
+      max_actor_approxkl3:                      0.0007596066
+      max_actor_clipfrac:                       0.015014648
+      max_actor_loss:                           0.00042782724
+      max_advantage:                            0.025643468
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.311773e-05
+      max_entropy:                              0.029047122
+      max_grad_norm_pre_clip:                   0.66903925
+      max_loss:                                 0.00041009433
+    std:
+      std_actor_approxkl1:                      0.00040581185
+      std_actor_approxkl3:                      0.00019682196
+      std_actor_clipfrac:                       0.003995151
+      std_actor_loss:                           0.00018616715
+      std_advantage:                            0.0066792965
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.3874646e-06
+      std_entropy:                              0.003104386
+      std_grad_norm_pre_clip:                   0.15726563
+      std_loss:                                 0.00018559446
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040616
+    avg_avg_return:                             0.97384423
+    avg_benchmark_regret:                       0.0157853
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.1335144
+    proxy_corner:
+      avg_avg_return:                           0.97384423
+      avg_reward_per_step:                      0.1335144
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.25003
+    avg_avg_return:                             0.07083559
+    avg_benchmark_regret:                       0.9219359
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.012580872
+    proxy_corner:
+      avg_avg_return:                           0.8142996
+      avg_reward_per_step:                      0.006614685
+  eval-batch-tree:
+    avg_avg_episode_length:                     70.97339
+    avg_avg_return:                             0.22378626
+    avg_benchmark_regret:                       0.7602894
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02722168
+    proxy_corner:
+      avg_avg_return:                           0.22378626
+      avg_reward_per_step:                      0.02722168
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        140509184
+    env-step-replay-before:                     71237632
+    env-step-used-before:                       140509184
+    ppo-update-before:                          85760
+    env-step-generate-before:                   69271552
+    env-step-all-after:                         140541952
+    env-step-replay-after:                      71270400
+    env-step-used-after:                        140541952
+    ppo-update-after:                           85780
+    env-step-generate-after:                    69271552
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       314603.7789879964
+    ppo_updates_per_second:                     42.13084060832877
+    eval-batch-orig-duration:                   0.5130764301866293
+    eval-batch-shift-duration:                  0.4655130188912153
+    eval-batch-tree-duration:                   0.4918305892497301
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999998
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          18.999996
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     677.7368
+      avg_last_visit_time:                      2165.1362
+      num_replay_batches:                       2175
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00076633523
+    avg_actor_approxkl3:                        0.0005758505
+    avg_actor_clipfrac:                         0.023181153
+    avg_actor_loss:                             -0.00023666769
+    avg_advantage:                              -0.00016327649
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.0895341e-05
+    avg_entropy:                                0.050575674
+    avg_grad_norm_pre_clip:                     0.28308678
+    avg_loss:                                   -0.0002817957
+    max:
+      max_actor_approxkl1:                      0.0020096754
+      max_actor_approxkl3:                      0.0012517669
+      max_actor_clipfrac:                       0.0546875
+      max_actor_loss:                           0.00011185184
+      max_advantage:                            0.016168714
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.9634326e-05
+      max_entropy:                              0.053438433
+      max_grad_norm_pre_clip:                   0.71292686
+      max_loss:                                 6.402167e-05
+    std:
+      std_actor_approxkl1:                      0.0005695711
+      std_actor_approxkl3:                      0.00038698758
+      std_actor_clipfrac:                       0.018667856
+      std_actor_loss:                           0.00017428823
+      std_advantage:                            0.004336098
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.897825e-06
+      std_entropy:                              0.0018657407
+      std_grad_norm_pre_clip:                   0.18377614
+      std_loss:                                 0.0001749413
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.042398
+    avg_avg_return:                             0.97368574
+    avg_benchmark_regret:                       0.0159438
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13350677
+    proxy_corner:
+      avg_avg_return:                           0.97368574
+      avg_reward_per_step:                      0.13350677
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.53662
+    avg_avg_return:                             0.066357926
+    avg_benchmark_regret:                       0.9264136
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.009002686
+    proxy_corner:
+      avg_avg_return:                           0.85369515
+      avg_reward_per_step:                      0.0070495605
+  eval-batch-tree:
+    avg_avg_episode_length:                     71.84964
+    avg_avg_return:                             0.21526402
+    avg_benchmark_regret:                       0.7688116
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.026275635
+    proxy_corner:
+      avg_avg_return:                           0.21526402
+      avg_reward_per_step:                      0.026275635
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        141557760
+    env-step-replay-before:                     71729152
+    env-step-used-before:                       141557760
+    ppo-update-before:                          86400
+    env-step-generate-before:                   69828608
+    env-step-all-after:                         141590528
+    env-step-replay-after:                      71761920
+    env-step-used-after:                        141590528
+    ppo-update-after:                           86420
+    env-step-generate-after:                    69828608
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184444
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184444
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184444
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184444
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       293452.1731631084
+    ppo_updates_per_second:                     42.27962037813365
+    eval-batch-orig-duration:                   0.5246348464861512
+    eval-batch-shift-duration:                  0.48095804918557405
+    eval-batch-tree-duration:                   0.45462165400385857
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999998
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.000002
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030749
+    visit_patterns:
+      avg_first_visit_time:                     677.7368
+      avg_last_visit_time:                      2180.278
+      num_replay_batches:                       2190
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0026734106
+    avg_actor_approxkl3:                        0.0025661795
+    avg_actor_clipfrac:                         0.018841553
+    avg_actor_loss:                             0.000105536776
+    avg_advantage:                              -0.0036825996
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.2778973e-05
+    avg_entropy:                                0.047266312
+    avg_grad_norm_pre_clip:                     0.8372187
+    avg_loss:                                   6.4659944e-05
+    max:
+      max_actor_approxkl1:                      0.006765495
+      max_actor_approxkl3:                      0.0063583935
+      max_actor_clipfrac:                       0.05493164
+      max_actor_loss:                           0.00473894
+      max_advantage:                            0.019461691
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.1762677e-05
+      max_entropy:                              0.051898062
+      max_grad_norm_pre_clip:                   3.0546634
+      max_loss:                                 0.004705773
+    std:
+      std_actor_approxkl1:                      0.0022987772
+      std_actor_approxkl3:                      0.0019440127
+      std_actor_clipfrac:                       0.023558315
+      std_actor_loss:                           0.0014623087
+      std_advantage:                            0.0050039613
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.541079e-06
+      std_entropy:                              0.0027006466
+      std_grad_norm_pre_clip:                   1.0460151
+      std_loss:                                 0.0014642067
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040735
+    avg_avg_return:                             0.97382104
+    avg_benchmark_regret:                       0.015808448
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13350677
+    proxy_corner:
+      avg_avg_return:                           0.97382104
+      avg_reward_per_step:                      0.13350677
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.85101
+    avg_avg_return:                             0.06371635
+    avg_benchmark_regret:                       0.9290552
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.00894928
+    proxy_corner:
+      avg_avg_return:                           0.8435098
+      avg_reward_per_step:                      0.0069122314
+  eval-batch-tree:
+    avg_avg_episode_length:                     68.35251
+    avg_avg_return:                             0.23752582
+    avg_benchmark_regret:                       0.7465498
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02986908
+    proxy_corner:
+      avg_avg_return:                           0.23752582
+      avg_reward_per_step:                      0.02986908
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        142606336
+    env-step-replay-before:                     72286208
+    env-step-used-before:                       142606336
+    ppo-update-before:                          87040
+    env-step-generate-before:                   70320128
+    env-step-all-after:                         142639104
+    env-step-replay-after:                      72318976
+    env-step-used-after:                        142639104
+    ppo-update-after:                           87060
+    env-step-generate-after:                    70320128
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       339486.56604799343
+    ppo_updates_per_second:                     44.69229378448185
+    eval-batch-orig-duration:                   0.511968344449997
+    eval-batch-shift-duration:                  0.4721437534317374
+    eval-batch-tree-duration:                   0.4898985670879483
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.000002
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.000006
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     677.7368
+      avg_last_visit_time:                      2197.318
+      num_replay_batches:                       2207
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011321408
+    avg_actor_approxkl3:                        0.001156479
+    avg_actor_clipfrac:                         0.03164673
+    avg_actor_loss:                             9.956486e-05
+    avg_advantage:                              -0.00025170884
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.9040883e-05
+    avg_entropy:                                0.113475226
+    avg_grad_norm_pre_clip:                     0.57096237
+    avg_loss:                                   6.1005414e-07
+    max:
+      max_actor_approxkl1:                      0.0036760326
+      max_actor_approxkl3:                      0.004207927
+      max_actor_clipfrac:                       0.068237305
+      max_actor_loss:                           0.0014301948
+      max_advantage:                            0.039214134
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.880891e-05
+      max_entropy:                              0.123117745
+      max_grad_norm_pre_clip:                   1.5939044
+      max_loss:                                 0.0013231955
+    std:
+      std_actor_approxkl1:                      0.0010765238
+      std_actor_approxkl3:                      0.0010358063
+      std_actor_clipfrac:                       0.02047432
+      std_actor_loss:                           0.00069907424
+      std_advantage:                            0.008735593
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          6.0456728e-06
+      std_entropy:                              0.005522094
+      std_grad_norm_pre_clip:                   0.39837083
+      std_loss:                                 0.0007003742
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.051008
+    avg_avg_return:                             0.9730495
+    avg_benchmark_regret:                       0.016580012
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13330078
+    proxy_corner:
+      avg_avg_return:                           0.9730495
+      avg_reward_per_step:                      0.13330078
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.5632
+    avg_avg_return:                             0.059684888
+    avg_benchmark_regret:                       0.9330867
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.005706787
+    proxy_corner:
+      avg_avg_return:                           0.9021682
+      avg_reward_per_step:                      0.0074920654
+  eval-batch-tree:
+    avg_avg_episode_length:                     68.15044
+    avg_avg_return:                             0.24003774
+    avg_benchmark_regret:                       0.74403787
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.03023529
+    proxy_corner:
+      avg_avg_return:                           0.24003774
+      avg_reward_per_step:                      0.03023529
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        143654912
+    env-step-replay-before:                     72941568
+    env-step-used-before:                       143654912
+    ppo-update-before:                          87680
+    env-step-generate-before:                   70713344
+    env-step-all-after:                         143687680
+    env-step-replay-after:                      72974336
+    env-step-used-after:                        143687680
+    ppo-update-after:                           87700
+    env-step-generate-after:                    70713344
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184384
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184384
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184384
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184384
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       314892.3336183122
+    ppo_updates_per_second:                     40.670215406554725
+    eval-batch-orig-duration:                   0.5107407383620739
+    eval-batch-shift-duration:                  0.48440883681178093
+    eval-batch-tree-duration:                   0.45402168575674295
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.0
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.000002
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030752
+    visit_patterns:
+      avg_first_visit_time:                     677.7368
+      avg_last_visit_time:                      2217.2156
+      num_replay_batches:                       2227
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00037866388
+    avg_actor_approxkl3:                        0.00038154234
+    avg_actor_clipfrac:                         0.015972901
+    avg_actor_loss:                             -8.8800865e-05
+    avg_advantage:                              -0.0012838284
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.8591758e-05
+    avg_entropy:                                0.081361406
+    avg_grad_norm_pre_clip:                     0.2497233
+    avg_loss:                                   -0.00016086639
+    max:
+      max_actor_approxkl1:                      0.001224101
+      max_actor_approxkl3:                      0.0011180523
+      max_actor_clipfrac:                       0.03869629
+      max_actor_loss:                           0.00033227168
+      max_advantage:                            0.0438928
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.6209029e-05
+      max_entropy:                              0.08488792
+      max_grad_norm_pre_clip:                   0.80010945
+      max_loss:                                 0.0002667661
+    std:
+      std_actor_approxkl1:                      0.00036564967
+      std_actor_approxkl3:                      0.0002924176
+      std_actor_clipfrac:                       0.01273219
+      std_actor_loss:                           0.00019753278
+      std_advantage:                            0.00688869
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.186866e-06
+      std_entropy:                              0.002230713
+      std_grad_norm_pre_clip:                   0.1741036
+      std_loss:                                 0.000198919
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.044808
+    avg_avg_return:                             0.9737423
+    avg_benchmark_regret:                       0.015887182
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13342285
+    proxy_corner:
+      avg_avg_return:                           0.9737423
+      avg_reward_per_step:                      0.13342285
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.640175
+    avg_avg_return:                             0.05644344
+    avg_benchmark_regret:                       0.93632805
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0064849854
+    proxy_corner:
+      avg_avg_return:                           0.901059
+      avg_reward_per_step:                      0.0074310303
+  eval-batch-tree:
+    avg_avg_episode_length:                     72.286255
+    avg_avg_return:                             0.21192351
+    avg_benchmark_regret:                       0.7721521
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025871277
+    proxy_corner:
+      avg_avg_return:                           0.21192351
+      avg_reward_per_step:                      0.025871277
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        144703488
+    env-step-replay-before:                     73531392
+    env-step-used-before:                       144703488
+    ppo-update-before:                          88320
+    env-step-generate-before:                   71172096
+    env-step-all-after:                         144736256
+    env-step-replay-after:                      73564160
+    env-step-used-after:                        144736256
+    ppo-update-after:                           88340
+    env-step-generate-after:                    71172096
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       256503.46181115453
+    ppo_updates_per_second:                     42.044866919357034
+    eval-batch-orig-duration:                   0.5360521990805864
+    eval-batch-shift-duration:                  0.49051519203931093
+    eval-batch-tree-duration:                   0.44351241271942854
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999998
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.0
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     677.7368
+      avg_last_visit_time:                      2235.1877
+      num_replay_batches:                       2245
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0022370215
+    avg_actor_approxkl3:                        0.0017283474
+    avg_actor_clipfrac:                         0.022790527
+    avg_actor_loss:                             -0.0008919883
+    avg_advantage:                              0.0008033047
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            3.094793e-05
+    avg_entropy:                                0.08154311
+    avg_grad_norm_pre_clip:                     0.9769277
+    avg_loss:                                   -0.0009580574
+    max:
+      max_actor_approxkl1:                      0.01354054
+      max_actor_approxkl3:                      0.012631517
+      max_actor_clipfrac:                       0.0703125
+      max_actor_loss:                           0.0015106001
+      max_advantage:                            0.04108274
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.823452e-05
+      max_entropy:                              0.122524284
+      max_grad_norm_pre_clip:                   2.322534
+      max_loss:                                 0.001449073
+    std:
+      std_actor_approxkl1:                      0.0038206836
+      std_actor_approxkl3:                      0.0029281063
+      std_actor_clipfrac:                       0.017966872
+      std_actor_loss:                           0.0018539507
+      std_advantage:                            0.007345752
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.5235757e-06
+      std_entropy:                              0.014805686
+      std_grad_norm_pre_clip:                   0.73593503
+      std_loss:                                 0.0018650875
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.053194
+    avg_avg_return:                             0.97380054
+    avg_benchmark_regret:                       0.01582901
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13329315
+    proxy_corner:
+      avg_avg_return:                           0.97380054
+      avg_reward_per_step:                      0.13329315
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.05583
+    avg_avg_return:                             0.05390804
+    avg_benchmark_regret:                       0.9388635
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0071487427
+    proxy_corner:
+      avg_avg_return:                           0.90947115
+      avg_reward_per_step:                      0.0074920654
+  eval-batch-tree:
+    avg_avg_episode_length:                     76.60068
+    avg_avg_return:                             0.14833558
+    avg_benchmark_regret:                       0.83574003
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.024414062
+    proxy_corner:
+      avg_avg_return:                           0.14833558
+      avg_reward_per_step:                      0.024414062
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        145752064
+    env-step-replay-before:                     74088448
+    env-step-used-before:                       145752064
+    ppo-update-before:                          88960
+    env-step-generate-before:                   71663616
+    env-step-all-after:                         145784832
+    env-step-replay-after:                      74121216
+    env-step-used-after:                        145784832
+    ppo-update-after:                           88980
+    env-step-generate-after:                    71663616
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8418428
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8418428
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8418428
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8418428
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       249202.53298006312
+    ppo_updates_per_second:                     41.84606411659701
+    eval-batch-orig-duration:                   0.5498533369973302
+    eval-batch-shift-duration:                  0.48436242062598467
+    eval-batch-tree-duration:                   0.4840970356017351
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.000004
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.000004
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030892
+    visit_patterns:
+      avg_first_visit_time:                     678.28
+      avg_last_visit_time:                      2252.3674
+      num_replay_batches:                       2262
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013720152
+    avg_actor_approxkl3:                        0.0014671285
+    avg_actor_clipfrac:                         0.04748535
+    avg_actor_loss:                             -0.0001938142
+    avg_advantage:                              0.0009389417
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.0239408e-05
+    avg_entropy:                                0.16066092
+    avg_grad_norm_pre_clip:                     1.005272
+    avg_loss:                                   -0.00034935545
+    max:
+      max_actor_approxkl1:                      0.0033191894
+      max_actor_approxkl3:                      0.003117957
+      max_actor_clipfrac:                       0.09008789
+      max_actor_loss:                           0.0028511807
+      max_advantage:                            0.048379064
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.2656145e-05
+      max_entropy:                              0.17114606
+      max_grad_norm_pre_clip:                   2.3913467
+      max_loss:                                 0.0026964338
+    std:
+      std_actor_approxkl1:                      0.00079335325
+      std_actor_approxkl3:                      0.00077702844
+      std_actor_clipfrac:                       0.02820519
+      std_actor_loss:                           0.00082825415
+      std_advantage:                            0.0037678531
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.2919082e-06
+      std_entropy:                              0.005976154
+      std_grad_norm_pre_clip:                   0.62126696
+      std_loss:                                 0.00082884857
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.045374
+    avg_avg_return:                             0.97370183
+    avg_benchmark_regret:                       0.015927657
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13345337
+    proxy_corner:
+      avg_avg_return:                           0.97370183
+      avg_reward_per_step:                      0.13345337
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.12933
+    avg_avg_return:                             0.059691366
+    avg_benchmark_regret:                       0.93308014
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.008888245
+    proxy_corner:
+      avg_avg_return:                           0.88744974
+      avg_reward_per_step:                      0.0072631836
+  eval-batch-tree:
+    avg_avg_episode_length:                     76.44454
+    avg_avg_return:                             0.15540713
+    avg_benchmark_regret:                       0.8286685
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.024208069
+    proxy_corner:
+      avg_avg_return:                           0.15540713
+      avg_reward_per_step:                      0.024208069
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        146800640
+    env-step-replay-before:                     74711040
+    env-step-used-before:                       146800640
+    ppo-update-before:                          89600
+    env-step-generate-before:                   72089600
+    env-step-all-after:                         146833408
+    env-step-replay-after:                      74743808
+    env-step-used-after:                        146833408
+    ppo-update-after:                           89620
+    env-step-generate-after:                    72089600
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184384
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184384
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184384
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184384
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       243184.08263128588
+    ppo_updates_per_second:                     42.44087125402506
+    eval-batch-orig-duration:                   0.6720827333629131
+    eval-batch-shift-duration:                  0.6151381256058812
+    eval-batch-tree-duration:                   0.6160958837717772
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999996
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          18.999996
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030883
+    visit_patterns:
+      avg_first_visit_time:                     678.6565
+      avg_last_visit_time:                      2271.3047
+      num_replay_batches:                       2281
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00270116
+    avg_actor_approxkl3:                        0.0025877429
+    avg_actor_clipfrac:                         0.074761964
+    avg_actor_loss:                             0.0006228369
+    avg_advantage:                              -0.0012028364
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.9935529e-05
+    avg_entropy:                                0.15273997
+    avg_grad_norm_pre_clip:                     2.2043366
+    avg_loss:                                   0.0004850647
+    max:
+      max_actor_approxkl1:                      0.00541732
+      max_actor_approxkl3:                      0.005304508
+      max_actor_clipfrac:                       0.1595459
+      max_actor_loss:                           0.0033291727
+      max_advantage:                            0.042503178
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          8.24101e-05
+      max_entropy:                              0.1631206
+      max_grad_norm_pre_clip:                   4.2286487
+      max_loss:                                 0.003212069
+    std:
+      std_actor_approxkl1:                      0.0017378584
+      std_actor_approxkl3:                      0.0015510024
+      std_actor_clipfrac:                       0.038195264
+      std_actor_loss:                           0.0012202768
+      std_advantage:                            0.004326106
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.075202e-05
+      std_entropy:                              0.00654577
+      std_grad_norm_pre_clip:                   1.1810465
+      std_loss:                                 0.0012214108
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.044205
+    avg_avg_return:                             0.97394174
+    avg_benchmark_regret:                       0.015687812
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.133461
+    proxy_corner:
+      avg_avg_return:                           0.97394174
+      avg_reward_per_step:                      0.133461
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.55249
+    avg_avg_return:                             0.05268172
+    avg_benchmark_regret:                       0.9400898
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.006752014
+    proxy_corner:
+      avg_avg_return:                           0.8819948
+      avg_reward_per_step:                      0.007171631
+  eval-batch-tree:
+    avg_avg_episode_length:                     78.58155
+    avg_avg_return:                             0.10832159
+    avg_benchmark_regret:                       0.87575406
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.023864746
+    proxy_corner:
+      avg_avg_return:                           0.10832159
+      avg_reward_per_step:                      0.023864746
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        147849216
+    env-step-replay-before:                     75235328
+    env-step-used-before:                       147849216
+    ppo-update-before:                          90240
+    env-step-generate-before:                   72613888
+    env-step-all-after:                         147881984
+    env-step-replay-after:                      75235328
+    env-step-used-after:                        147881984
+    ppo-update-after:                           90260
+    env-step-generate-after:                    72646656
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.106154
+    avg_avg_return:                             0.9254292
+    avg_reward_per_step:                        0.120788574
+    proxy_corner:
+      avg_avg_return:                           0.9254292
+      avg_reward_per_step:                      0.120788574
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.106154
+    avg_avg_return:                             0.9254292
+    avg_reward_per_step:                        0.120788574
+    proxy_corner:
+      avg_avg_return:                           0.9254292
+      avg_reward_per_step:                      0.120788574
+  perf:
+    env_steps_per_second:                       293174.19132188405
+    ppo_updates_per_second:                     44.53423848498771
+    eval-batch-orig-duration:                   0.5228681471198797
+    eval-batch-shift-duration:                  0.4916157089173794
+    eval-batch-tree-duration:                   0.4871563035994768
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.0
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.000002
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030766
+    visit_patterns:
+      avg_first_visit_time:                     679.0134
+      avg_last_visit_time:                      2286.349
+      num_replay_batches:                       2296
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00033731275
+    avg_actor_approxkl3:                        0.00021589232
+    avg_actor_clipfrac:                         0.0095092775
+    avg_actor_loss:                             -8.4076404e-05
+    avg_advantage:                              0.00094202487
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.025373e-05
+    avg_entropy:                                0.030335281
+    avg_grad_norm_pre_clip:                     0.3051401
+    avg_loss:                                   -0.00010428482
+    max:
+      max_actor_approxkl1:                      0.0008168109
+      max_actor_approxkl3:                      0.00046839603
+      max_actor_clipfrac:                       0.02368164
+      max_actor_loss:                           0.00041979016
+      max_advantage:                            0.05408275
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.4177705e-05
+      max_entropy:                              0.0374068
+      max_grad_norm_pre_clip:                   0.64666164
+      max_loss:                                 0.000399791
+    std:
+      std_actor_approxkl1:                      0.0002637299
+      std_actor_approxkl3:                      0.00012342725
+      std_actor_clipfrac:                       0.005990146
+      std_actor_loss:                           0.0002560418
+      std_advantage:                            0.006564358
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.4298581e-06
+      std_entropy:                              0.0040967497
+      std_grad_norm_pre_clip:                   0.1486724
+      std_loss:                                 0.00025725632
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.055597
+    avg_avg_return:                             0.97380394
+    avg_benchmark_regret:                       0.015825598
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13336182
+    proxy_corner:
+      avg_avg_return:                           0.97380394
+      avg_reward_per_step:                      0.13336182
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.6662
+    avg_avg_return:                             0.06534238
+    avg_benchmark_regret:                       0.92742914
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.009498596
+    proxy_corner:
+      avg_avg_return:                           0.8476046
+      avg_reward_per_step:                      0.0069274902
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.56932
+    avg_avg_return:                             0.08936877
+    avg_benchmark_regret:                       0.89470685
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.024002075
+    proxy_corner:
+      avg_avg_return:                           0.08936877
+      avg_reward_per_step:                      0.024002075
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        148897792
+    env-step-replay-before:                     75694080
+    env-step-used-before:                       148897792
+    ppo-update-before:                          90880
+    env-step-generate-before:                   73203712
+    env-step-all-after:                         148930560
+    env-step-replay-after:                      75726848
+    env-step-used-after:                        148930560
+    ppo-update-after:                           90900
+    env-step-generate-after:                    73203712
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       317609.82797067304
+    ppo_updates_per_second:                     42.335020617381026
+    eval-batch-orig-duration:                   0.5204470055177808
+    eval-batch-shift-duration:                  0.4803063180297613
+    eval-batch-tree-duration:                   0.48368509765714407
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999998
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.0
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     679.0134
+      avg_last_visit_time:                      2301.3557
+      num_replay_batches:                       2311
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013881591
+    avg_actor_approxkl3:                        0.0012437704
+    avg_actor_clipfrac:                         0.039257813
+    avg_actor_loss:                             0.0003970707
+    avg_advantage:                              -0.00032555207
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            9.7832935e-06
+    avg_entropy:                                0.09677559
+    avg_grad_norm_pre_clip:                     1.5584699
+    avg_loss:                                   0.00030518675
+    max:
+      max_actor_approxkl1:                      0.003203369
+      max_actor_approxkl3:                      0.003546681
+      max_actor_clipfrac:                       0.10620117
+      max_actor_loss:                           0.005043652
+      max_advantage:                            0.020517945
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.0863119e-05
+      max_entropy:                              0.0992559
+      max_grad_norm_pre_clip:                   3.5158095
+      max_loss:                                 0.00495129
+    std:
+      std_actor_approxkl1:                      0.0008878705
+      std_actor_approxkl3:                      0.00090498914
+      std_actor_clipfrac:                       0.027392384
+      std_actor_loss:                           0.001916675
+      std_advantage:                            0.0044939993
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.681206e-07
+      std_entropy:                              0.002067433
+      std_grad_norm_pre_clip:                   1.0509759
+      std_loss:                                 0.0019167288
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.049263
+    avg_avg_return:                             0.97365975
+    avg_benchmark_regret:                       0.015969772
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.1334076
+    proxy_corner:
+      avg_avg_return:                           0.97365975
+      avg_reward_per_step:                      0.1334076
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.81497
+    avg_avg_return:                             0.06497802
+    avg_benchmark_regret:                       0.9277935
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.007347107
+    proxy_corner:
+      avg_avg_return:                           0.8567871
+      avg_reward_per_step:                      0.0070343018
+  eval-batch-tree:
+    avg_avg_episode_length:                     78.18022
+    avg_avg_return:                             0.11494653
+    avg_benchmark_regret:                       0.8691291
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.024429321
+    proxy_corner:
+      avg_avg_return:                           0.11494653
+      avg_reward_per_step:                      0.024429321
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        149946368
+    env-step-replay-before:                     76152832
+    env-step-used-before:                       149946368
+    ppo-update-before:                          91520
+    env-step-generate-before:                   73793536
+    env-step-all-after:                         149979136
+    env-step-replay-after:                      76152832
+    env-step-used-after:                        149979136
+    ppo-update-after:                           91540
+    env-step-generate-after:                    73826304
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.0741825
+    avg_avg_return:                             0.9280227
+    avg_reward_per_step:                        0.13925171
+    proxy_corner:
+      avg_avg_return:                           0.9280227
+      avg_reward_per_step:                      0.13925171
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.0741825
+    avg_avg_return:                             0.9280227
+    avg_reward_per_step:                        0.13925171
+    proxy_corner:
+      avg_avg_return:                           0.9280227
+      avg_reward_per_step:                      0.13925171
+  perf:
+    env_steps_per_second:                       298729.09357906587
+    ppo_updates_per_second:                     42.06711877439659
+    eval-batch-orig-duration:                   0.5459245881065726
+    eval-batch-shift-duration:                  0.46431804075837135
+    eval-batch-tree-duration:                   0.47529929503798485
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.000004
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.000006
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     679.0134
+      avg_last_visit_time:                      2314.2297
+      num_replay_batches:                       2324
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0006919043
+    avg_actor_approxkl3:                        0.00068111595
+    avg_actor_clipfrac:                         0.021990968
+    avg_actor_loss:                             0.0002171241
+    avg_advantage:                              0.00071744155
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.2425264e-05
+    avg_entropy:                                0.06521455
+    avg_grad_norm_pre_clip:                     0.7398616
+    avg_loss:                                   0.00016312221
+    max:
+      max_actor_approxkl1:                      0.002313617
+      max_actor_approxkl3:                      0.002304729
+      max_actor_clipfrac:                       0.06726074
+      max_actor_loss:                           0.0024465271
+      max_advantage:                            0.03863299
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.058318e-05
+      max_entropy:                              0.08199915
+      max_grad_norm_pre_clip:                   2.3384132
+      max_loss:                                 0.0023756954
+    std:
+      std_actor_approxkl1:                      0.00059701654
+      std_actor_approxkl3:                      0.0005964438
+      std_actor_clipfrac:                       0.015991595
+      std_actor_loss:                           0.0007633715
+      std_advantage:                            0.0064553996
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.5233556e-06
+      std_entropy:                              0.007719066
+      std_grad_norm_pre_clip:                   0.60483485
+      std_loss:                                 0.0007594224
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.076314
+    avg_avg_return:                             0.97410214
+    avg_benchmark_regret:                       0.015527378
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13308716
+    proxy_corner:
+      avg_avg_return:                           0.97410214
+      avg_reward_per_step:                      0.13308716
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.85863
+    avg_avg_return:                             0.066718295
+    avg_benchmark_regret:                       0.92605317
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0055770874
+    proxy_corner:
+      avg_avg_return:                           0.8682225
+      avg_reward_per_step:                      0.0071868896
+  eval-batch-tree:
+    avg_avg_episode_length:                     78.52835
+    avg_avg_return:                             0.106213644
+    avg_benchmark_regret:                       0.877862
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02418518
+    proxy_corner:
+      avg_avg_return:                           0.106213644
+      avg_reward_per_step:                      0.02418518
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        150994944
+    env-step-replay-before:                     76709888
+    env-step-used-before:                       150994944
+    ppo-update-before:                          92160
+    env-step-generate-before:                   74285056
+    env-step-all-after:                         151027712
+    env-step-replay-after:                      76742656
+    env-step-used-after:                        151027712
+    ppo-update-after:                           92180
+    env-step-generate-after:                    74285056
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       343141.22701331624
+    ppo_updates_per_second:                     42.29695781182051
+    eval-batch-orig-duration:                   0.5277079744264483
+    eval-batch-shift-duration:                  0.49257335625588894
+    eval-batch-tree-duration:                   0.49349424336105585
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999998
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.0
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     679.8623
+      avg_last_visit_time:                      2332.2427
+      num_replay_batches:                       2342
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011990959
+    avg_actor_approxkl3:                        0.0010090315
+    avg_actor_clipfrac:                         0.04041138
+    avg_actor_loss:                             -5.2741543e-05
+    avg_advantage:                              0.00031458656
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.4915264e-05
+    avg_entropy:                                0.13312899
+    avg_grad_norm_pre_clip:                     0.6389776
+    avg_loss:                                   -0.00017841293
+    max:
+      max_actor_approxkl1:                      0.0029120566
+      max_actor_approxkl3:                      0.0029353984
+      max_actor_clipfrac:                       0.118896484
+      max_actor_loss:                           0.00057101995
+      max_advantage:                            0.034338117
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.9882755e-05
+      max_entropy:                              0.13893229
+      max_grad_norm_pre_clip:                   1.3562793
+      max_loss:                                 0.0004409587
+    std:
+      std_actor_approxkl1:                      0.0008303132
+      std_actor_approxkl3:                      0.00071288564
+      std_actor_clipfrac:                       0.026636358
+      std_actor_loss:                           0.00029402942
+      std_advantage:                            0.0056941076
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.3406735e-06
+      std_entropy:                              0.0035195574
+      std_grad_norm_pre_clip:                   0.39584213
+      std_loss:                                 0.00029326105
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.07071
+    avg_avg_return:                             0.9739644
+    avg_benchmark_regret:                       0.015665121
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13316345
+    proxy_corner:
+      avg_avg_return:                           0.9739644
+      avg_reward_per_step:                      0.13316345
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.74214
+    avg_avg_return:                             0.06883985
+    avg_benchmark_regret:                       0.92393166
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0038528442
+    proxy_corner:
+      avg_avg_return:                           0.8882233
+      avg_reward_per_step:                      0.0074539185
+  eval-batch-tree:
+    avg_avg_episode_length:                     78.4142
+    avg_avg_return:                             0.11388308
+    avg_benchmark_regret:                       0.8701925
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.024154663
+    proxy_corner:
+      avg_avg_return:                           0.11388308
+      avg_reward_per_step:                      0.024154663
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        152043520
+    env-step-replay-before:                     77299712
+    env-step-used-before:                       152043520
+    ppo-update-before:                          92800
+    env-step-generate-before:                   74743808
+    env-step-all-after:                         152076288
+    env-step-replay-after:                      77299712
+    env-step-used-after:                        152076288
+    ppo-update-after:                           92820
+    env-step-generate-after:                    74776576
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.080641
+    avg_avg_return:                             0.9266826
+    avg_reward_per_step:                        0.122039795
+    proxy_corner:
+      avg_avg_return:                           0.9266826
+      avg_reward_per_step:                      0.122039795
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.080641
+    avg_avg_return:                             0.9266826
+    avg_reward_per_step:                        0.122039795
+    proxy_corner:
+      avg_avg_return:                           0.9266826
+      avg_reward_per_step:                      0.122039795
+  perf:
+    env_steps_per_second:                       295811.13191021694
+    ppo_updates_per_second:                     42.221098174544586
+    eval-batch-orig-duration:                   0.5356065323576331
+    eval-batch-shift-duration:                  0.48466713819652796
+    eval-batch-tree-duration:                   0.48495995067059994
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999996
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          18.999994
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     679.8623
+      avg_last_visit_time:                      2349.264
+      num_replay_batches:                       2359
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010813682
+    avg_actor_approxkl3:                        0.00063401024
+    avg_actor_clipfrac:                         0.014581299
+    avg_actor_loss:                             -0.00013647978
+    avg_advantage:                              0.00026585505
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.908672e-05
+    avg_entropy:                                0.052040506
+    avg_grad_norm_pre_clip:                     0.4164562
+    avg_loss:                                   -0.00017897696
+    max:
+      max_actor_approxkl1:                      0.0026628387
+      max_actor_approxkl3:                      0.001551489
+      max_actor_clipfrac:                       0.024169922
+      max_actor_loss:                           0.0013001198
+      max_advantage:                            0.0422253
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.4456298e-05
+      max_entropy:                              0.059212737
+      max_grad_norm_pre_clip:                   0.82055914
+      max_loss:                                 0.0012601675
+    std:
+      std_actor_approxkl1:                      0.0006476418
+      std_actor_approxkl3:                      0.00033479946
+      std_actor_clipfrac:                       0.0058566374
+      std_actor_loss:                           0.00060093694
+      std_advantage:                            0.00628427
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.7710626e-06
+      std_entropy:                              0.0037912382
+      std_grad_norm_pre_clip:                   0.16248728
+      std_loss:                                 0.00060247805
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.06076
+    avg_avg_return:                             0.9738917
+    avg_benchmark_regret:                       0.015737854
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13326263
+    proxy_corner:
+      avg_avg_return:                           0.9738917
+      avg_reward_per_step:                      0.13326263
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.214195
+    avg_avg_return:                             0.07962769
+    avg_benchmark_regret:                       0.9131439
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.005508423
+    proxy_corner:
+      avg_avg_return:                           0.8753748
+      avg_reward_per_step:                      0.007369995
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.4431
+    avg_avg_return:                             0.09198353
+    avg_benchmark_regret:                       0.8920921
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02381134
+    proxy_corner:
+      avg_avg_return:                           0.09198353
+      avg_reward_per_step:                      0.02381134
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        153092096
+    env-step-replay-before:                     77889536
+    env-step-used-before:                       153092096
+    ppo-update-before:                          93440
+    env-step-generate-before:                   75202560
+    env-step-all-after:                         153124864
+    env-step-replay-after:                      77889536
+    env-step-used-after:                        153124864
+    ppo-update-after:                           93460
+    env-step-generate-after:                    75235328
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.862547
+    avg_avg_return:                             0.92655987
+    avg_reward_per_step:                        0.14627075
+    proxy_corner:
+      avg_avg_return:                           0.92655987
+      avg_reward_per_step:                      0.14627075
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.862547
+    avg_avg_return:                             0.92655987
+    avg_reward_per_step:                        0.14627075
+    proxy_corner:
+      avg_avg_return:                           0.92655987
+      avg_reward_per_step:                      0.14627075
+  perf:
+    env_steps_per_second:                       309358.0052130106
+    ppo_updates_per_second:                     42.35864590818292
+    eval-batch-orig-duration:                   0.5254121497273445
+    eval-batch-shift-duration:                  0.4820322282612324
+    eval-batch-tree-duration:                   0.4800923466682434
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.000004
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.000006
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030772
+    visit_patterns:
+      avg_first_visit_time:                     679.8623
+      avg_last_visit_time:                      2367.245
+      num_replay_batches:                       2377
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00042751335
+    avg_actor_approxkl3:                        0.00042627202
+    avg_actor_clipfrac:                         0.016827393
+    avg_actor_loss:                             -5.6699093e-05
+    avg_advantage:                              0.00082105515
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.8448272e-05
+    avg_entropy:                                0.056814976
+    avg_grad_norm_pre_clip:                     0.44890237
+    avg_loss:                                   -0.000104289946
+    max:
+      max_actor_approxkl1:                      0.0011238836
+      max_actor_approxkl3:                      0.000955238
+      max_actor_clipfrac:                       0.048339844
+      max_actor_loss:                           0.00035574776
+      max_advantage:                            0.043761432
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.3853923e-05
+      max_entropy:                              0.070375085
+      max_grad_norm_pre_clip:                   0.880668
+      max_loss:                                 0.0003132418
+    std:
+      std_actor_approxkl1:                      0.0003441597
+      std_actor_approxkl3:                      0.0002344802
+      std_actor_clipfrac:                       0.011376815
+      std_actor_loss:                           0.00026931128
+      std_advantage:                            0.0060308585
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.437888e-06
+      std_entropy:                              0.0064940634
+      std_grad_norm_pre_clip:                   0.20872019
+      std_loss:                                 0.0002708273
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.045403
+    avg_avg_return:                             0.97381085
+    avg_benchmark_regret:                       0.015818665
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13346863
+    proxy_corner:
+      avg_avg_return:                           0.97381085
+      avg_reward_per_step:                      0.13346863
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.57892
+    avg_avg_return:                             0.07196383
+    avg_benchmark_regret:                       0.9208077
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.006187439
+    proxy_corner:
+      avg_avg_return:                           0.8712102
+      avg_reward_per_step:                      0.0072402954
+  eval-batch-tree:
+    avg_avg_episode_length:                     78.40184
+    avg_avg_return:                             0.112040795
+    avg_benchmark_regret:                       0.8720348
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.023857117
+    proxy_corner:
+      avg_avg_return:                           0.112040795
+      avg_reward_per_step:                      0.023857117
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        154140672
+    env-step-replay-before:                     78544896
+    env-step-used-before:                       154140672
+    ppo-update-before:                          94080
+    env-step-generate-before:                   75595776
+    env-step-all-after:                         154173440
+    env-step-replay-after:                      78577664
+    env-step-used-after:                        154173440
+    ppo-update-after:                           94100
+    env-step-generate-after:                    75595776
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184444
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184444
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184444
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184444
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       348171.3763330199
+    ppo_updates_per_second:                     44.569218896191
+    eval-batch-orig-duration:                   0.5193842062726617
+    eval-batch-shift-duration:                  0.48728446289896965
+    eval-batch-tree-duration:                   0.4852334884926677
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999996
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          18.999998
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030749
+    visit_patterns:
+      avg_first_visit_time:                     679.8623
+      avg_last_visit_time:                      2388.2437
+      num_replay_batches:                       2398
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002030495
+    avg_actor_approxkl3:                        0.0021856276
+    avg_actor_clipfrac:                         0.08560181
+    avg_actor_loss:                             0.003835234
+    avg_advantage:                              0.0044576963
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.20241e-05
+    avg_entropy:                                0.14739668
+    avg_grad_norm_pre_clip:                     3.0948083
+    avg_loss:                                   0.00370885
+    max:
+      max_actor_approxkl1:                      0.0064627603
+      max_actor_approxkl3:                      0.0045516677
+      max_actor_clipfrac:                       0.1550293
+      max_actor_loss:                           0.014983282
+      max_advantage:                            0.03890723
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          8.814496e-05
+      max_entropy:                              0.15352494
+      max_grad_norm_pre_clip:                   6.422443
+      max_loss:                                 0.01484728
+    std:
+      std_actor_approxkl1:                      0.00156188
+      std_actor_approxkl3:                      0.0014755848
+      std_actor_clipfrac:                       0.04732423
+      std_actor_loss:                           0.0049922816
+      std_advantage:                            0.006286406
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.0131827e-05
+      std_entropy:                              0.0032366302
+      std_grad_norm_pre_clip:                   1.9390454
+      std_loss:                                 0.0049896245
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.043033
+    avg_avg_return:                             0.97379756
+    avg_benchmark_regret:                       0.015831932
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13348389
+    proxy_corner:
+      avg_avg_return:                           0.97379756
+      avg_reward_per_step:                      0.13348389
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.6392
+    avg_avg_return:                             0.06914008
+    avg_benchmark_regret:                       0.9236314
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.004966736
+    proxy_corner:
+      avg_avg_return:                           0.8875987
+      avg_reward_per_step:                      0.007385254
+  eval-batch-tree:
+    avg_avg_episode_length:                     77.3013
+    avg_avg_return:                             0.13589898
+    avg_benchmark_regret:                       0.84817666
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.024284363
+    proxy_corner:
+      avg_avg_return:                           0.13589898
+      avg_reward_per_step:                      0.024284363
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        155189248
+    env-step-replay-before:                     79134720
+    env-step-used-before:                       155189248
+    ppo-update-before:                          94720
+    env-step-generate-before:                   76054528
+    env-step-all-after:                         155222016
+    env-step-replay-after:                      79134720
+    env-step-used-after:                        155222016
+    ppo-update-after:                           94740
+    env-step-generate-after:                    76087296
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.307804
+    avg_avg_return:                             0.9290514
+    avg_reward_per_step:                        0.1347351
+    proxy_corner:
+      avg_avg_return:                           0.9290514
+      avg_reward_per_step:                      0.1347351
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.307804
+    avg_avg_return:                             0.9290514
+    avg_reward_per_step:                        0.1347351
+    proxy_corner:
+      avg_avg_return:                           0.9290514
+      avg_reward_per_step:                      0.1347351
+  perf:
+    env_steps_per_second:                       321771.6179750531
+    ppo_updates_per_second:                     42.304733740528015
+    eval-batch-orig-duration:                   0.5191336954012513
+    eval-batch-shift-duration:                  0.47573081962764263
+    eval-batch-tree-duration:                   0.48471463192254305
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999998
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.0
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030752
+    visit_patterns:
+      avg_first_visit_time:                     679.8623
+      avg_last_visit_time:                      2405.2703
+      num_replay_batches:                       2415
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0003720084
+    avg_actor_approxkl3:                        0.00042735846
+    avg_actor_clipfrac:                         0.018426513
+    avg_actor_loss:                             -0.00013576905
+    avg_advantage:                              -0.0016914618
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.0216514e-05
+    avg_entropy:                                0.06888242
+    avg_grad_norm_pre_clip:                     0.3934537
+    avg_loss:                                   -0.00019454322
+    max:
+      max_actor_approxkl1:                      0.001111883
+      max_actor_approxkl3:                      0.0010153514
+      max_actor_clipfrac:                       0.045654297
+      max_actor_loss:                           0.00026847515
+      max_advantage:                            0.031311512
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.8538723e-05
+      max_entropy:                              0.08249543
+      max_grad_norm_pre_clip:                   0.8794385
+      max_loss:                                 0.00021552507
+    std:
+      std_actor_approxkl1:                      0.00027426874
+      std_actor_approxkl3:                      0.00025947625
+      std_actor_clipfrac:                       0.011551714
+      std_actor_loss:                           0.00022596403
+      std_advantage:                            0.0064014033
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.8951147e-06
+      std_entropy:                              0.006997358
+      std_grad_norm_pre_clip:                   0.20799243
+      std_loss:                                 0.00022739914
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041927
+    avg_avg_return:                             0.97378623
+    avg_benchmark_regret:                       0.015843246
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349152
+    proxy_corner:
+      avg_avg_return:                           0.97378623
+      avg_reward_per_step:                      0.13349152
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.59434
+    avg_avg_return:                             0.055511832
+    avg_benchmark_regret:                       0.9372597
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0029525757
+    proxy_corner:
+      avg_avg_return:                           0.8980431
+      avg_reward_per_step:                      0.007408142
+  eval-batch-tree:
+    avg_avg_episode_length:                     77.37379
+    avg_avg_return:                             0.1255445
+    avg_benchmark_regret:                       0.8585311
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.024246216
+    proxy_corner:
+      avg_avg_return:                           0.1255445
+      avg_reward_per_step:                      0.024246216
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        156237824
+    env-step-replay-before:                     79462400
+    env-step-used-before:                       156237824
+    ppo-update-before:                          95360
+    env-step-generate-before:                   76775424
+    env-step-all-after:                         156270592
+    env-step-replay-after:                      79495168
+    env-step-used-after:                        156270592
+    ppo-update-after:                           95380
+    env-step-generate-after:                    76775424
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8418417
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8418417
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8418417
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8418417
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       312405.9194012084
+    ppo_updates_per_second:                     42.22245957858826
+    eval-batch-orig-duration:                   0.5371163748204708
+    eval-batch-shift-duration:                  0.4788974542170763
+    eval-batch-tree-duration:                   0.4825291223824024
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999998
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          18.999998
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030772
+    visit_patterns:
+      avg_first_visit_time:                     679.8623
+      avg_last_visit_time:                      2416.248
+      num_replay_batches:                       2426
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001356275
+    avg_actor_approxkl3:                        0.0011901379
+    avg_actor_clipfrac:                         0.036413576
+    avg_actor_loss:                             0.0003946662
+    avg_advantage:                              0.0011853741
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.0510422e-05
+    avg_entropy:                                0.13382559
+    avg_grad_norm_pre_clip:                     1.1527017
+    avg_loss:                                   0.00026609583
+    max:
+      max_actor_approxkl1:                      0.003035935
+      max_actor_approxkl3:                      0.00238037
+      max_actor_clipfrac:                       0.10424805
+      max_actor_loss:                           0.0024098605
+      max_advantage:                            0.025568008
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.2504784e-05
+      max_entropy:                              0.1397275
+      max_grad_norm_pre_clip:                   2.3397183
+      max_loss:                                 0.0022875953
+    std:
+      std_actor_approxkl1:                      0.00071169203
+      std_actor_approxkl3:                      0.00059698406
+      std_actor_clipfrac:                       0.02600966
+      std_actor_loss:                           0.001017856
+      std_advantage:                            0.004877982
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          9.64408e-07
+      std_entropy:                              0.003316143
+      std_grad_norm_pre_clip:                   0.5789841
+      std_loss:                                 0.0010198444
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.086119
+    avg_avg_return:                             0.9733852
+    avg_benchmark_regret:                       0.016244259
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13303375
+    proxy_corner:
+      avg_avg_return:                           0.9733852
+      avg_reward_per_step:                      0.13303375
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.29814
+    avg_avg_return:                             0.047822192
+    avg_benchmark_regret:                       0.9449493
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0028076172
+    proxy_corner:
+      avg_avg_return:                           0.90840614
+      avg_reward_per_step:                      0.007507324
+  eval-batch-tree:
+    avg_avg_episode_length:                     76.3312
+    avg_avg_return:                             0.14494333
+    avg_benchmark_regret:                       0.8391323
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02494049
+    proxy_corner:
+      avg_avg_return:                           0.14494333
+      avg_reward_per_step:                      0.02494049
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        157286400
+    env-step-replay-before:                     80084992
+    env-step-used-before:                       157286400
+    ppo-update-before:                          96000
+    env-step-generate-before:                   77201408
+    env-step-all-after:                         157319168
+    env-step-replay-after:                      80117760
+    env-step-used-after:                        157319168
+    ppo-update-after:                           96020
+    env-step-generate-after:                    77201408
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       294194.71590130625
+    ppo_updates_per_second:                     42.1124381873681
+    eval-batch-orig-duration:                   0.5471622319892049
+    eval-batch-shift-duration:                  0.4872832428663969
+    eval-batch-tree-duration:                   0.4785300949588418
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.000002
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.000002
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     679.8623
+      avg_last_visit_time:                      2435.2334
+      num_replay_batches:                       2445
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0007516138
+    avg_actor_approxkl3:                        0.0009976845
+    avg_actor_clipfrac:                         0.042657472
+    avg_actor_loss:                             0.00018636575
+    avg_advantage:                              -0.0005201847
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.1633113e-05
+    avg_entropy:                                0.07471432
+    avg_grad_norm_pre_clip:                     0.87403977
+    avg_loss:                                   0.00011746801
+    max:
+      max_actor_approxkl1:                      0.0022829939
+      max_actor_approxkl3:                      0.003079342
+      max_actor_clipfrac:                       0.11706543
+      max_actor_loss:                           0.003435094
+      max_advantage:                            0.028842926
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.0051113e-05
+      max_entropy:                              0.08207523
+      max_grad_norm_pre_clip:                   2.177905
+      max_loss:                                 0.0033643553
+    std:
+      std_actor_approxkl1:                      0.0008472874
+      std_actor_approxkl3:                      0.0010279658
+      std_actor_clipfrac:                       0.04064251
+      std_actor_loss:                           0.0010781919
+      std_advantage:                            0.0038140723
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.017382e-06
+      std_entropy:                              0.0029359558
+      std_grad_norm_pre_clip:                   0.7024094
+      std_loss:                                 0.0010779392
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041082
+    avg_avg_return:                             0.97394514
+    avg_benchmark_regret:                       0.015684407
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13350677
+    proxy_corner:
+      avg_avg_return:                           0.97394514
+      avg_reward_per_step:                      0.13350677
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.00121
+    avg_avg_return:                             0.048343502
+    avg_benchmark_regret:                       0.944428
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0033035278
+    proxy_corner:
+      avg_avg_return:                           0.8966575
+      avg_reward_per_step:                      0.007347107
+  eval-batch-tree:
+    avg_avg_episode_length:                     76.16847
+    avg_avg_return:                             0.14532505
+    avg_benchmark_regret:                       0.83875054
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02508545
+    proxy_corner:
+      avg_avg_return:                           0.14532505
+      avg_reward_per_step:                      0.02508545
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        158334976
+    env-step-replay-before:                     80642048
+    env-step-used-before:                       158334976
+    ppo-update-before:                          96640
+    env-step-generate-before:                   77692928
+    env-step-all-after:                         158367744
+    env-step-replay-after:                      80642048
+    env-step-used-after:                        158367744
+    ppo-update-after:                           96660
+    env-step-generate-after:                    77725696
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.14461
+    avg_avg_return:                             0.93034875
+    avg_reward_per_step:                        0.13671875
+    proxy_corner:
+      avg_avg_return:                           0.93034875
+      avg_reward_per_step:                      0.13671875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.14461
+    avg_avg_return:                             0.93034875
+    avg_reward_per_step:                        0.13671875
+    proxy_corner:
+      avg_avg_return:                           0.93034875
+      avg_reward_per_step:                      0.13671875
+  perf:
+    env_steps_per_second:                       275979.3902806568
+    ppo_updates_per_second:                     41.61934868581986
+    eval-batch-orig-duration:                   0.523171185515821
+    eval-batch-shift-duration:                  0.4867602800950408
+    eval-batch-tree-duration:                   0.4824841842055321
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999998
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          18.999998
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     679.8623
+      avg_last_visit_time:                      2451.174
+      num_replay_batches:                       2461
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00045859627
+    avg_actor_approxkl3:                        0.0003594759
+    avg_actor_clipfrac:                         0.014874267
+    avg_actor_loss:                             -0.00025429772
+    avg_advantage:                              0.0016389947
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.0113386e-05
+    avg_entropy:                                0.049277157
+    avg_grad_norm_pre_clip:                     0.3366521
+    avg_loss:                                   -0.00029351818
+    max:
+      max_actor_approxkl1:                      0.0012010331
+      max_actor_approxkl3:                      0.00070945825
+      max_actor_clipfrac:                       0.025512695
+      max_actor_loss:                           0.00038833357
+      max_advantage:                            0.03834021
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.342668e-05
+      max_entropy:                              0.05594507
+      max_grad_norm_pre_clip:                   0.74168473
+      max_loss:                                 0.00034827046
+    std:
+      std_actor_approxkl1:                      0.0003199515
+      std_actor_approxkl3:                      0.00016165049
+      std_actor_clipfrac:                       0.0069047217
+      std_actor_loss:                           0.000420227
+      std_advantage:                            0.0062220227
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.1096478e-06
+      std_entropy:                              0.0032943492
+      std_grad_norm_pre_clip:                   0.14902122
+      std_loss:                                 0.00042126494
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.042014
+    avg_avg_return:                             0.97382426
+    avg_benchmark_regret:                       0.015805261
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349152
+    proxy_corner:
+      avg_avg_return:                           0.97382426
+      avg_reward_per_step:                      0.13349152
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.830475
+    avg_avg_return:                             0.051896892
+    avg_benchmark_regret:                       0.94087464
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0035247803
+    proxy_corner:
+      avg_avg_return:                           0.86108434
+      avg_reward_per_step:                      0.007080078
+  eval-batch-tree:
+    avg_avg_episode_length:                     72.93767
+    avg_avg_return:                             0.18316314
+    avg_benchmark_regret:                       0.8009125
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.027244568
+    proxy_corner:
+      avg_avg_return:                           0.18316314
+      avg_reward_per_step:                      0.027244568
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        159383552
+    env-step-replay-before:                     81133568
+    env-step-used-before:                       159383552
+    ppo-update-before:                          97280
+    env-step-generate-before:                   78249984
+    env-step-all-after:                         159416320
+    env-step-replay-after:                      81133568
+    env-step-used-after:                        159416320
+    ppo-update-after:                           97300
+    env-step-generate-after:                    78282752
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.180671
+    avg_avg_return:                             0.9239251
+    avg_reward_per_step:                        0.13699341
+    proxy_corner:
+      avg_avg_return:                           0.9239251
+      avg_reward_per_step:                      0.13699341
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.180671
+    avg_avg_return:                             0.9239251
+    avg_reward_per_step:                        0.13699341
+    proxy_corner:
+      avg_avg_return:                           0.9239251
+      avg_reward_per_step:                      0.13699341
+  perf:
+    env_steps_per_second:                       315675.41819446516
+    ppo_updates_per_second:                     40.85681092961632
+    eval-batch-orig-duration:                   0.5192008996382356
+    eval-batch-shift-duration:                  0.49194114841520786
+    eval-batch-tree-duration:                   0.48335590586066246
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999994
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          18.999996
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     679.8623
+      avg_last_visit_time:                      2466.305
+      num_replay_batches:                       2476
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010106211
+    avg_actor_approxkl3:                        0.00073884253
+    avg_actor_clipfrac:                         0.011254883
+    avg_actor_loss:                             0.00027882124
+    avg_advantage:                              0.0017109325
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.422295e-05
+    avg_entropy:                                0.02533704
+    avg_grad_norm_pre_clip:                     0.41159484
+    avg_loss:                                   0.0002655956
+    max:
+      max_actor_approxkl1:                      0.0030815857
+      max_actor_approxkl3:                      0.0022160034
+      max_actor_clipfrac:                       0.024414062
+      max_actor_loss:                           0.0022450304
+      max_advantage:                            0.041846395
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.235408e-05
+      max_entropy:                              0.03433357
+      max_grad_norm_pre_clip:                   1.6649727
+      max_loss:                                 0.0022380725
+    std:
+      std_actor_approxkl1:                      0.0008926971
+      std_actor_approxkl3:                      0.0006629414
+      std_actor_clipfrac:                       0.0068980637
+      std_actor_loss:                           0.0007692254
+      std_advantage:                            0.007749943
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.4390536e-06
+      std_entropy:                              0.003942811
+      std_grad_norm_pre_clip:                   0.38253018
+      std_loss:                                 0.00076848944
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040616
+    avg_avg_return:                             0.97380555
+    avg_benchmark_regret:                       0.01582396
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.1335144
+    proxy_corner:
+      avg_avg_return:                           0.97380555
+      avg_reward_per_step:                      0.1335144
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.873856
+    avg_avg_return:                             0.051878847
+    avg_benchmark_regret:                       0.9408927
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0069732666
+    proxy_corner:
+      avg_avg_return:                           0.83750784
+      avg_reward_per_step:                      0.0068130493
+  eval-batch-tree:
+    avg_avg_episode_length:                     77.20821
+    avg_avg_return:                             0.1303578
+    avg_benchmark_regret:                       0.8537178
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.024589539
+    proxy_corner:
+      avg_avg_return:                           0.1303578
+      avg_reward_per_step:                      0.024589539
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        160432128
+    env-step-replay-before:                     81625088
+    env-step-used-before:                       160432128
+    ppo-update-before:                          97920
+    env-step-generate-before:                   78807040
+    env-step-all-after:                         160464896
+    env-step-replay-after:                      81657856
+    env-step-used-after:                        160464896
+    ppo-update-after:                           97940
+    env-step-generate-after:                    78807040
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       273547.29170778114
+    ppo_updates_per_second:                     42.21425662238269
+    eval-batch-orig-duration:                   0.6061153206974268
+    eval-batch-shift-duration:                  0.5240492317825556
+    eval-batch-tree-duration:                   0.5136250527575612
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999996
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.0
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     679.8623
+      avg_last_visit_time:                      2482.2183
+      num_replay_batches:                       2492
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.004214924
+    avg_actor_approxkl3:                        0.0042438353
+    avg_actor_clipfrac:                         0.046392825
+    avg_actor_loss:                             0.002936229
+    avg_advantage:                              0.0012415587
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            5.137758e-05
+    avg_entropy:                                0.062122583
+    avg_grad_norm_pre_clip:                     1.9541283
+    avg_loss:                                   0.0028997955
+    max:
+      max_actor_approxkl1:                      0.012553753
+      max_actor_approxkl3:                      0.010303272
+      max_actor_clipfrac:                       0.072631836
+      max_actor_loss:                           0.012442337
+      max_advantage:                            0.04848194
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          6.379571e-05
+      max_entropy:                              0.07196392
+      max_grad_norm_pre_clip:                   3.4443269
+      max_loss:                                 0.012404942
+    std:
+      std_actor_approxkl1:                      0.003258909
+      std_actor_approxkl3:                      0.002798667
+      std_actor_clipfrac:                       0.025552336
+      std_actor_loss:                           0.0036601943
+      std_advantage:                            0.010789827
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.7599625e-06
+      std_entropy:                              0.004397224
+      std_grad_norm_pre_clip:                   1.2789623
+      std_loss:                                 0.0036606316
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041787
+    avg_avg_return:                             0.9736966
+    avg_benchmark_regret:                       0.015932938
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13348389
+    proxy_corner:
+      avg_avg_return:                           0.9736966
+      avg_reward_per_step:                      0.13348389
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.73971
+    avg_avg_return:                             0.043083224
+    avg_benchmark_regret:                       0.9496883
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0037460327
+    proxy_corner:
+      avg_avg_return:                           0.89362335
+      avg_reward_per_step:                      0.0073013306
+  eval-batch-tree:
+    avg_avg_episode_length:                     74.67058
+    avg_avg_return:                             0.16680735
+    avg_benchmark_regret:                       0.81726825
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025001526
+    proxy_corner:
+      avg_avg_return:                           0.16680735
+      avg_reward_per_step:                      0.025001526
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        161480704
+    env-step-replay-before:                     82149376
+    env-step-used-before:                       161480704
+    ppo-update-before:                          98560
+    env-step-generate-before:                   79331328
+    env-step-all-after:                         161513472
+    env-step-replay-after:                      82149376
+    env-step-used-after:                        161513472
+    ppo-update-after:                           98580
+    env-step-generate-after:                    79364096
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.78858
+    avg_avg_return:                             0.9298069
+    avg_reward_per_step:                        0.13745117
+    proxy_corner:
+      avg_avg_return:                           0.9298069
+      avg_reward_per_step:                      0.13745117
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.78858
+    avg_avg_return:                             0.9298069
+    avg_reward_per_step:                        0.13745117
+    proxy_corner:
+      avg_avg_return:                           0.9298069
+      avg_reward_per_step:                      0.13745117
+  perf:
+    env_steps_per_second:                       288609.1572642835
+    ppo_updates_per_second:                     42.19200770329958
+    eval-batch-orig-duration:                   0.6112380819395185
+    eval-batch-shift-duration:                  0.519073055125773
+    eval-batch-tree-duration:                   0.5596362221986055
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999996
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          18.999998
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     679.8623
+      avg_last_visit_time:                      2497.3203
+      num_replay_batches:                       2507
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00043920588
+    avg_actor_approxkl3:                        0.00038296424
+    avg_actor_clipfrac:                         0.011920166
+    avg_actor_loss:                             0.00080341485
+    avg_advantage:                              0.0021925233
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            3.71999e-05
+    avg_entropy:                                0.020950617
+    avg_grad_norm_pre_clip:                     0.9453516
+    avg_loss:                                   0.0008010641
+    max:
+      max_actor_approxkl1:                      0.001040577
+      max_actor_approxkl3:                      0.00069152564
+      max_actor_clipfrac:                       0.029541016
+      max_actor_loss:                           0.0027282813
+      max_advantage:                            0.052358866
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          4.817037e-05
+      max_entropy:                              0.026517197
+      max_grad_norm_pre_clip:                   1.7772338
+      max_loss:                                 0.0027224976
+    std:
+      std_actor_approxkl1:                      0.00028495994
+      std_actor_approxkl3:                      0.0001846035
+      std_actor_clipfrac:                       0.007755575
+      std_actor_loss:                           0.00069788157
+      std_advantage:                            0.009239891
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.3907224e-06
+      std_entropy:                              0.0030344722
+      std_grad_norm_pre_clip:                   0.55010235
+      std_loss:                                 0.0006972236
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041082
+    avg_avg_return:                             0.9737456
+    avg_benchmark_regret:                       0.015883913
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13350677
+    proxy_corner:
+      avg_avg_return:                           0.9737456
+      avg_reward_per_step:                      0.13350677
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.94409
+    avg_avg_return:                             0.052296124
+    avg_benchmark_regret:                       0.94047546
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.004096985
+    proxy_corner:
+      avg_avg_return:                           0.8437946
+      avg_reward_per_step:                      0.006881714
+  eval-batch-tree:
+    avg_avg_episode_length:                     73.628235
+    avg_avg_return:                             0.18031138
+    avg_benchmark_regret:                       0.8037642
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025802612
+    proxy_corner:
+      avg_avg_return:                           0.18031138
+      avg_reward_per_step:                      0.025802612
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        162529280
+    env-step-replay-before:                     82542592
+    env-step-used-before:                       162529280
+    ppo-update-before:                          99200
+    env-step-generate-before:                   79986688
+    env-step-all-after:                         162562048
+    env-step-replay-after:                      82542592
+    env-step-used-after:                        162562048
+    ppo-update-after:                           99220
+    env-step-generate-after:                    80019456
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.017441
+    avg_avg_return:                             0.92640567
+    avg_reward_per_step:                        0.14648438
+    proxy_corner:
+      avg_avg_return:                           0.92640567
+      avg_reward_per_step:                      0.14648438
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.017441
+    avg_avg_return:                             0.92640567
+    avg_reward_per_step:                        0.14648438
+    proxy_corner:
+      avg_avg_return:                           0.92640567
+      avg_reward_per_step:                      0.14648438
+  perf:
+    env_steps_per_second:                       340752.94844287314
+    ppo_updates_per_second:                     42.31320209010722
+    eval-batch-orig-duration:                   0.5154420007020235
+    eval-batch-shift-duration:                  0.48211460933089256
+    eval-batch-tree-duration:                   0.4842782747000456
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999998
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          18.999996
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1403079
+    visit_patterns:
+      avg_first_visit_time:                     680.3711
+      avg_last_visit_time:                      2509.2925
+      num_replay_batches:                       2519
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001414373
+    avg_actor_approxkl3:                        0.001321839
+    avg_actor_clipfrac:                         0.016790772
+    avg_actor_loss:                             7.116922e-05
+    avg_advantage:                              0.001262113
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            3.2808264e-05
+    avg_entropy:                                0.0642252
+    avg_grad_norm_pre_clip:                     0.4692381
+    avg_loss:                                   2.3348146e-05
+    max:
+      max_actor_approxkl1:                      0.0034818673
+      max_actor_approxkl3:                      0.003214262
+      max_actor_clipfrac:                       0.033447266
+      max_actor_loss:                           0.0013396395
+      max_advantage:                            0.039827824
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          4.111014e-05
+      max_entropy:                              0.077236325
+      max_grad_norm_pre_clip:                   0.824263
+      max_loss:                                 0.0012900105
+    std:
+      std_actor_approxkl1:                      0.0012731595
+      std_actor_approxkl3:                      0.0010703492
+      std_actor_clipfrac:                       0.009527388
+      std_actor_loss:                           0.0005549773
+      std_advantage:                            0.007357735
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.283395e-06
+      std_entropy:                              0.0054935836
+      std_grad_norm_pre_clip:                   0.1900118
+      std_loss:                                 0.00055530545
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.047807
+    avg_avg_return:                             0.97347444
+    avg_benchmark_regret:                       0.01615509
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13346863
+    proxy_corner:
+      avg_avg_return:                           0.97347444
+      avg_reward_per_step:                      0.13346863
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.06741
+    avg_avg_return:                             0.051057108
+    avg_benchmark_regret:                       0.9417144
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0032958984
+    proxy_corner:
+      avg_avg_return:                           0.887282
+      avg_reward_per_step:                      0.007286072
+  eval-batch-tree:
+    avg_avg_episode_length:                     69.82959
+    avg_avg_return:                             0.22417669
+    avg_benchmark_regret:                       0.7598989
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.029190063
+    proxy_corner:
+      avg_avg_return:                           0.22417669
+      avg_reward_per_step:                      0.029190063
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        163577856
+    env-step-replay-before:                     83132416
+    env-step-used-before:                       163577856
+    ppo-update-before:                          99840
+    env-step-generate-before:                   80445440
+    env-step-all-after:                         163610624
+    env-step-replay-after:                      83165184
+    env-step-used-after:                        163610624
+    ppo-update-after:                           99860
+    env-step-generate-after:                    80445440
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       291789.22873768746
+    ppo_updates_per_second:                     42.215721735716265
+    eval-batch-orig-duration:                   0.5212469818070531
+    eval-batch-shift-duration:                  0.4906558720394969
+    eval-batch-tree-duration:                   0.48658487387001514
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.000002
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.000004
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030749
+    visit_patterns:
+      avg_first_visit_time:                     680.3711
+      avg_last_visit_time:                      2528.2808
+      num_replay_batches:                       2538
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0060174935
+    avg_actor_approxkl3:                        0.005915726
+    avg_actor_clipfrac:                         0.04393921
+    avg_actor_loss:                             -0.000730314
+    avg_advantage:                              0.0022094236
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.1775557e-05
+    avg_entropy:                                0.07855692
+    avg_grad_norm_pre_clip:                     0.22964334
+    avg_loss:                                   -0.00079798326
+    max:
+      max_actor_approxkl1:                      0.0119962115
+      max_actor_approxkl3:                      0.010754576
+      max_actor_clipfrac:                       0.057128906
+      max_actor_loss:                           -5.1688403e-07
+      max_advantage:                            0.037196636
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.5275174e-05
+      max_entropy:                              0.08461536
+      max_grad_norm_pre_clip:                   0.7485614
+      max_loss:                                 -7.330098e-05
+    std:
+      std_actor_approxkl1:                      0.003813893
+      std_actor_approxkl3:                      0.0036578225
+      std_actor_clipfrac:                       0.014912499
+      std_actor_loss:                           0.0002759719
+      std_advantage:                            0.0065960963
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.2913096e-06
+      std_entropy:                              0.0032364647
+      std_grad_norm_pre_clip:                   0.19379844
+      std_loss:                                 0.0002745995
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.045963
+    avg_avg_return:                             0.9735179
+    avg_benchmark_regret:                       0.016111664
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349152
+    proxy_corner:
+      avg_avg_return:                           0.9735179
+      avg_reward_per_step:                      0.13349152
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.46414
+    avg_avg_return:                             0.057537105
+    avg_benchmark_regret:                       0.9352344
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.006576538
+    proxy_corner:
+      avg_avg_return:                           0.8363236
+      avg_reward_per_step:                      0.006866455
+  eval-batch-tree:
+    avg_avg_episode_length:                     70.28874
+    avg_avg_return:                             0.21768065
+    avg_benchmark_regret:                       0.766395
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.028617859
+    proxy_corner:
+      avg_avg_return:                           0.21768065
+      avg_reward_per_step:                      0.028617859
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        164626432
+    env-step-replay-before:                     83722240
+    env-step-used-before:                       164626432
+    ppo-update-before:                          100480
+    env-step-generate-before:                   80904192
+    env-step-all-after:                         164659200
+    env-step-replay-after:                      83755008
+    env-step-used-after:                        164659200
+    ppo-update-after:                           100500
+    env-step-generate-after:                    80904192
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       315350.8505857676
+    ppo_updates_per_second:                     41.156298055194966
+    eval-batch-orig-duration:                   0.2978675765916705
+    eval-batch-shift-duration:                  0.2207383643835783
+    eval-batch-tree-duration:                   0.2169119454920292
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.0
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          18.999998
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     680.96436
+      avg_last_visit_time:                      2546.184
+      num_replay_batches:                       2556
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00051451
+    avg_actor_approxkl3:                        0.0006744752
+    avg_actor_clipfrac:                         0.018951416
+    avg_actor_loss:                             0.00012787442
+    avg_advantage:                              0.0014381143
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.2419997e-05
+    avg_entropy:                                0.06489603
+    avg_grad_norm_pre_clip:                     0.58855075
+    avg_loss:                                   6.918836e-05
+    max:
+      max_actor_approxkl1:                      0.0015170336
+      max_actor_approxkl3:                      0.0018916199
+      max_actor_clipfrac:                       0.05505371
+      max_actor_loss:                           0.0039446508
+      max_advantage:                            0.033315897
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.0643412e-05
+      max_entropy:                              0.06890227
+      max_grad_norm_pre_clip:                   2.0109434
+      max_loss:                                 0.0038869982
+    std:
+      std_actor_approxkl1:                      0.00057933084
+      std_actor_approxkl3:                      0.00051753817
+      std_actor_clipfrac:                       0.017759526
+      std_actor_loss:                           0.001113096
+      std_advantage:                            0.004590803
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.1089684e-06
+      std_entropy:                              0.0019143752
+      std_grad_norm_pre_clip:                   0.60997283
+      std_loss:                                 0.0011131633
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.044914
+    avg_avg_return:                             0.97361505
+    avg_benchmark_regret:                       0.01601442
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13348389
+    proxy_corner:
+      avg_avg_return:                           0.97361505
+      avg_reward_per_step:                      0.13348389
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.37134
+    avg_avg_return:                             0.05590941
+    avg_benchmark_regret:                       0.9368621
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.00705719
+    proxy_corner:
+      avg_avg_return:                           0.8477349
+      avg_reward_per_step:                      0.006942749
+  eval-batch-tree:
+    avg_avg_episode_length:                     71.64406
+    avg_avg_return:                             0.2076649
+    avg_benchmark_regret:                       0.7764107
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.027656555
+    proxy_corner:
+      avg_avg_return:                           0.2076649
+      avg_reward_per_step:                      0.027656555
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        165675008
+    env-step-replay-before:                     84344832
+    env-step-used-before:                       165675008
+    ppo-update-before:                          101120
+    env-step-generate-before:                   81330176
+    env-step-all-after:                         165707776
+    env-step-replay-after:                      84377600
+    env-step-used-after:                        165707776
+    ppo-update-after:                           101140
+    env-step-generate-after:                    81330176
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       315697.69258716743
+    ppo_updates_per_second:                     44.40987773366076
+    eval-batch-orig-duration:                   0.5147771080955863
+    eval-batch-shift-duration:                  0.5314168417826295
+    eval-batch-tree-duration:                   0.46562416665256023
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.0
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.000004
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     681.5857
+      avg_last_visit_time:                      2565.2437
+      num_replay_batches:                       2575
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001028892
+    avg_actor_approxkl3:                        0.001085155
+    avg_actor_clipfrac:                         0.016168213
+    avg_actor_loss:                             0.00069945265
+    avg_advantage:                              0.0021182555
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.2863824e-05
+    avg_entropy:                                0.09767937
+    avg_grad_norm_pre_clip:                     0.4939731
+    avg_loss:                                   0.0006132053
+    max:
+      max_actor_approxkl1:                      0.0028411904
+      max_actor_approxkl3:                      0.0021338614
+      max_actor_clipfrac:                       0.035888672
+      max_actor_loss:                           0.002438643
+      max_advantage:                            0.045946896
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.4673445e-05
+      max_entropy:                              0.10588318
+      max_grad_norm_pre_clip:                   0.9780926
+      max_loss:                                 0.0023498142
+    std:
+      std_actor_approxkl1:                      0.00073321146
+      std_actor_approxkl3:                      0.0005211156
+      std_actor_clipfrac:                       0.01030887
+      std_actor_loss:                           0.0009139021
+      std_advantage:                            0.00593106
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.0315176e-06
+      std_entropy:                              0.0050749974
+      std_grad_norm_pre_clip:                   0.21949995
+      std_loss:                                 0.00091296557
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041853
+    avg_avg_return:                             0.9739994
+    avg_benchmark_regret:                       0.015630174
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13347626
+    proxy_corner:
+      avg_avg_return:                           0.9739994
+      avg_reward_per_step:                      0.13347626
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.15564
+    avg_avg_return:                             0.05382105
+    avg_benchmark_regret:                       0.9389504
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.003692627
+    proxy_corner:
+      avg_avg_return:                           0.91426057
+      avg_reward_per_step:                      0.007575989
+  eval-batch-tree:
+    avg_avg_episode_length:                     71.28296
+    avg_avg_return:                             0.21280444
+    avg_benchmark_regret:                       0.7712712
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.027648926
+    proxy_corner:
+      avg_avg_return:                           0.21280444
+      avg_reward_per_step:                      0.027648926
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        166723584
+    env-step-replay-before:                     84934656
+    env-step-used-before:                       166723584
+    ppo-update-before:                          101760
+    env-step-generate-before:                   81788928
+    env-step-all-after:                         166756352
+    env-step-replay-after:                      84967424
+    env-step-used-after:                        166756352
+    ppo-update-after:                           101780
+    env-step-generate-after:                    81788928
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184444
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184444
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184444
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184444
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       295152.65783342085
+    ppo_updates_per_second:                     40.61212605658144
+    eval-batch-orig-duration:                   0.5252696583047509
+    eval-batch-shift-duration:                  0.4496148321777582
+    eval-batch-tree-duration:                   0.47310785856097937
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.0
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.0
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030749
+    visit_patterns:
+      avg_first_visit_time:                     681.5857
+      avg_last_visit_time:                      2583.2905
+      num_replay_batches:                       2593
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0033470634
+    avg_actor_approxkl3:                        0.0035252038
+    avg_actor_clipfrac:                         0.009857178
+    avg_actor_loss:                             -8.162186e-05
+    avg_advantage:                              -0.0030397098
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.512681e-05
+    avg_entropy:                                0.037420258
+    avg_grad_norm_pre_clip:                     0.33468053
+    avg_loss:                                   -0.00010647871
+    max:
+      max_actor_approxkl1:                      0.008221909
+      max_actor_approxkl3:                      0.008006059
+      max_actor_clipfrac:                       0.0146484375
+      max_actor_loss:                           0.00059220195
+      max_advantage:                            0.033796668
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.3317236e-05
+      max_entropy:                              0.044194765
+      max_grad_norm_pre_clip:                   0.7767068
+      max_loss:                                 0.00056748686
+    std:
+      std_actor_approxkl1:                      0.0028447984
+      std_actor_approxkl3:                      0.0027077508
+      std_actor_clipfrac:                       0.0033859662
+      std_actor_loss:                           0.0004670913
+      std_advantage:                            0.007054813
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.3247996e-06
+      std_entropy:                              0.0038793117
+      std_grad_norm_pre_clip:                   0.18675217
+      std_loss:                                 0.00046665774
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.042067
+    avg_avg_return:                             0.97380006
+    avg_benchmark_regret:                       0.01582944
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349915
+    proxy_corner:
+      avg_avg_return:                           0.97380006
+      avg_reward_per_step:                      0.13349915
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.562706
+    avg_avg_return:                             0.04236185
+    avg_benchmark_regret:                       0.95040965
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0065612793
+    proxy_corner:
+      avg_avg_return:                           0.91893655
+      avg_reward_per_step:                      0.0074768066
+  eval-batch-tree:
+    avg_avg_episode_length:                     70.44534
+    avg_avg_return:                             0.21608478
+    avg_benchmark_regret:                       0.7679908
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.029304504
+    proxy_corner:
+      avg_avg_return:                           0.21608478
+      avg_reward_per_step:                      0.029304504
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        167772160
+    env-step-replay-before:                     85458944
+    env-step-used-before:                       167772160
+    ppo-update-before:                          102400
+    env-step-generate-before:                   82313216
+    env-step-all-after:                         167804928
+    env-step-replay-after:                      85491712
+    env-step-used-after:                        167804928
+    ppo-update-after:                           102420
+    env-step-generate-after:                    82313216
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  perf:
+    env_steps_per_second:                       271909.96716644586
+    ppo_updates_per_second:                     42.20461057525019
+    eval-batch-orig-duration:                   0.642942818813026
+    eval-batch-shift-duration:                  0.5938785979524255
+    eval-batch-tree-duration:                   0.5901949312537909
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.000008
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.00001
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     681.5857
+      avg_last_visit_time:                      2599.2173
+      num_replay_batches:                       2609
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0018112784
+    avg_actor_approxkl3:                        0.0014040511
+    avg_actor_clipfrac:                         0.044421386
+    avg_actor_loss:                             -0.0004932902
+    avg_advantage:                              0.0009559876
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.8071312e-05
+    avg_entropy:                                0.11511631
+    avg_grad_norm_pre_clip:                     0.5283573
+    avg_loss:                                   -0.0005943708
+    max:
+      max_actor_approxkl1:                      0.004161047
+      max_actor_approxkl3:                      0.0032136375
+      max_actor_clipfrac:                       0.08337402
+      max_actor_loss:                           0.00036852155
+      max_advantage:                            0.03700638
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.692002e-05
+      max_entropy:                              0.119930506
+      max_grad_norm_pre_clip:                   1.2344185
+      max_loss:                                 0.00026213512
+    std:
+      std_actor_approxkl1:                      0.0010650712
+      std_actor_approxkl3:                      0.00089989643
+      std_actor_clipfrac:                       0.025530966
+      std_actor_loss:                           0.0006298221
+      std_advantage:                            0.0076520303
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.3876333e-06
+      std_entropy:                              0.0027922948
+      std_grad_norm_pre_clip:                   0.25455248
+      std_loss:                                 0.00063130935
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040624
+    avg_avg_return:                             0.973806
+    avg_benchmark_regret:                       0.015823444
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13350677
+    proxy_corner:
+      avg_avg_return:                           0.973806
+      avg_reward_per_step:                      0.13350677
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.8819
+    avg_avg_return:                             0.03673918
+    avg_benchmark_regret:                       0.9560324
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0067596436
+    proxy_corner:
+      avg_avg_return:                           0.91575
+      avg_reward_per_step:                      0.007423401
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-tree:
+    avg_avg_episode_length:                     69.98693
+    avg_avg_return:                             0.224143
+    avg_benchmark_regret:                       0.75993264
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.029518127
+    proxy_corner:
+      avg_avg_return:                           0.224143
+      avg_reward_per_step:                      0.029518127
+    rollouts_gif:                               float32(132, 225, 225, 3)
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        168820736
+    env-step-replay-before:                     85983232
+    env-step-used-before:                       168820736
+    ppo-update-before:                          103040
+    env-step-generate-before:                   82837504
+    env-step-all-after:                         168853504
+    env-step-replay-after:                      85983232
+    env-step-used-after:                        168853504
+    ppo-update-after:                           103060
+    env-step-generate-after:                    82870272
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.620275
+    avg_avg_return:                             0.9307026
+    avg_reward_per_step:                        0.15267944
+    proxy_corner:
+      avg_avg_return:                           0.9307026
+      avg_reward_per_step:                      0.15267944
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.620275
+    avg_avg_return:                             0.9307026
+    avg_reward_per_step:                        0.15267944
+    proxy_corner:
+      avg_avg_return:                           0.9307026
+      avg_reward_per_step:                      0.15267944
+  perf:
+    env_steps_per_second:                       204652.31428458105
+    ppo_updates_per_second:                     41.42583241407313
+    eval-batch-orig-duration:                   0.6435638107359409
+    eval-batch-shift-duration:                  0.6470572911202908
+    eval-batch-tree-duration:                   0.5518575115129352
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.000002
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.000002
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030756
+    visit_patterns:
+      avg_first_visit_time:                     681.5857
+      avg_last_visit_time:                      2614.2083
+      num_replay_batches:                       2624
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00044565383
+    avg_actor_approxkl3:                        0.00042838868
+    avg_actor_clipfrac:                         0.009197998
+    avg_actor_loss:                             9.8332885e-06
+    avg_advantage:                              0.0024986928
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.7053302e-05
+    avg_entropy:                                0.036090378
+    avg_grad_norm_pre_clip:                     0.2706661
+    avg_loss:                                   -1.7730436e-05
+    max:
+      max_actor_approxkl1:                      0.0016882427
+      max_actor_approxkl3:                      0.0014175775
+      max_actor_clipfrac:                       0.023071289
+      max_actor_loss:                           0.00053125806
+      max_advantage:                            0.04079759
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.3087954e-05
+      max_entropy:                              0.048058286
+      max_grad_norm_pre_clip:                   0.52013105
+      max_loss:                                 0.00050511886
+    std:
+      std_actor_approxkl1:                      0.0005094526
+      std_actor_approxkl3:                      0.00040980268
+      std_actor_clipfrac:                       0.005864367
+      std_actor_loss:                           0.00024235882
+      std_advantage:                            0.006040909
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.8989402e-06
+      std_entropy:                              0.0057948297
+      std_grad_norm_pre_clip:                   0.107050136
+      std_loss:                                 0.00024295445
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040735
+    avg_avg_return:                             0.9737761
+    avg_benchmark_regret:                       0.015853453
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13350677
+    proxy_corner:
+      avg_avg_return:                           0.9737761
+      avg_reward_per_step:                      0.13350677
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.67831
+    avg_avg_return:                             0.041142832
+    avg_benchmark_regret:                       0.9516287
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.010192871
+    proxy_corner:
+      avg_avg_return:                           0.8431972
+      avg_reward_per_step:                      0.006881714
+  eval-batch-tree:
+    avg_avg_episode_length:                     71.23757
+    avg_avg_return:                             0.20938775
+    avg_benchmark_regret:                       0.7746879
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.030143738
+    proxy_corner:
+      avg_avg_return:                           0.20938775
+      avg_reward_per_step:                      0.030143738
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        169869312
+    env-step-replay-before:                     86474752
+    env-step-used-before:                       169869312
+    ppo-update-before:                          103680
+    env-step-generate-before:                   83394560
+    env-step-all-after:                         169902080
+    env-step-replay-after:                      86474752
+    env-step-used-after:                        169902080
+    ppo-update-after:                           103700
+    env-step-generate-after:                    83427328
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.851399
+    avg_avg_return:                             0.9251244
+    avg_reward_per_step:                        0.13842773
+    proxy_corner:
+      avg_avg_return:                           0.9251244
+      avg_reward_per_step:                      0.13842773
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.851399
+    avg_avg_return:                             0.9251244
+    avg_reward_per_step:                        0.13842773
+    proxy_corner:
+      avg_avg_return:                           0.9251244
+      avg_reward_per_step:                      0.13842773
+  perf:
+    env_steps_per_second:                       343731.26432205294
+    ppo_updates_per_second:                     42.327999431296575
+    eval-batch-orig-duration:                   0.5147536108270288
+    eval-batch-shift-duration:                  0.49237764067947865
+    eval-batch-tree-duration:                   0.49168887827545404
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999996
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.0
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     681.5857
+      avg_last_visit_time:                      2629.275
+      num_replay_batches:                       2639
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012983471
+    avg_actor_approxkl3:                        0.001198037
+    avg_actor_clipfrac:                         0.0139282225
+    avg_actor_loss:                             0.0007525029
+    avg_advantage:                              0.0007770277
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.3753305e-05
+    avg_entropy:                                0.039450888
+    avg_grad_norm_pre_clip:                     0.8676295
+    avg_loss:                                   0.0007349284
+    max:
+      max_actor_approxkl1:                      0.0026901267
+      max_actor_approxkl3:                      0.0024665394
+      max_actor_clipfrac:                       0.026611328
+      max_actor_loss:                           0.004226965
+      max_advantage:                            0.04380113
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          6.888651e-05
+      max_entropy:                              0.052757107
+      max_grad_norm_pre_clip:                   2.0396085
+      max_loss:                                 0.004204438
+    std:
+      std_actor_approxkl1:                      0.00085548824
+      std_actor_approxkl3:                      0.0006664714
+      std_actor_clipfrac:                       0.0068276385
+      std_actor_loss:                           0.0013123101
+      std_advantage:                            0.009259713
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          9.65689e-06
+      std_entropy:                              0.006305641
+      std_grad_norm_pre_clip:                   0.5818491
+      std_loss:                                 0.001313457
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040616
+    avg_avg_return:                             0.97380614
+    avg_benchmark_regret:                       0.015823405
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.1335144
+    proxy_corner:
+      avg_avg_return:                           0.97380614
+      avg_reward_per_step:                      0.1335144
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.66372
+    avg_avg_return:                             0.03661298
+    avg_benchmark_regret:                       0.9561585
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.010025024
+    proxy_corner:
+      avg_avg_return:                           0.8782233
+      avg_reward_per_step:                      0.0071258545
+  eval-batch-tree:
+    avg_avg_episode_length:                     71.52652
+    avg_avg_return:                             0.209896
+    avg_benchmark_regret:                       0.7741796
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.0308609
+    proxy_corner:
+      avg_avg_return:                           0.209896
+      avg_reward_per_step:                      0.0308609
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        170917888
+    env-step-replay-before:                     86900736
+    env-step-used-before:                       170917888
+    ppo-update-before:                          104320
+    env-step-generate-before:                   84017152
+    env-step-all-after:                         170950656
+    env-step-replay-after:                      86900736
+    env-step-used-after:                        170950656
+    ppo-update-after:                           104340
+    env-step-generate-after:                    84049920
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.920719
+    avg_avg_return:                             0.92710614
+    avg_reward_per_step:                        0.14520264
+    proxy_corner:
+      avg_avg_return:                           0.92710614
+      avg_reward_per_step:                      0.14520264
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.920719
+    avg_avg_return:                             0.92710614
+    avg_reward_per_step:                        0.14520264
+    proxy_corner:
+      avg_avg_return:                           0.92710614
+      avg_reward_per_step:                      0.14520264
+  perf:
+    env_steps_per_second:                       292917.73052326654
+    ppo_updates_per_second:                     42.1567694692756
+    eval-batch-orig-duration:                   0.7044143909588456
+    eval-batch-shift-duration:                  0.6247617788612843
+    eval-batch-tree-duration:                   0.6029603406786919
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999998
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          18.999998
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     681.5857
+      avg_last_visit_time:                      2642.2324
+      num_replay_batches:                       2652
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00036706837
+    avg_actor_approxkl3:                        0.00038270905
+    avg_actor_clipfrac:                         0.013378906
+    avg_actor_loss:                             8.386084e-05
+    avg_advantage:                              -0.00050262484
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.8238224e-05
+    avg_entropy:                                0.07782506
+    avg_grad_norm_pre_clip:                     0.38605568
+    avg_loss:                                   2.0154874e-05
+    max:
+      max_actor_approxkl1:                      0.0011348408
+      max_actor_approxkl3:                      0.0007824937
+      max_actor_clipfrac:                       0.048950195
+      max_actor_loss:                           0.0016621212
+      max_advantage:                            0.03271103
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          4.6671845e-05
+      max_entropy:                              0.09566976
+      max_grad_norm_pre_clip:                   1.9013765
+      max_loss:                                 0.0015789692
+    std:
+      std_actor_approxkl1:                      0.0003556023
+      std_actor_approxkl3:                      0.00024541118
+      std_actor_clipfrac:                       0.011433395
+      std_actor_loss:                           0.00052405917
+      std_advantage:                            0.0068719964
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          7.6792585e-06
+      std_entropy:                              0.006237234
+      std_grad_norm_pre_clip:                   0.44834086
+      std_loss:                                 0.00051862356
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.044506
+    avg_avg_return:                             0.9737631
+    avg_benchmark_regret:                       0.01586638
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13344574
+    proxy_corner:
+      avg_avg_return:                           0.9737631
+      avg_reward_per_step:                      0.13344574
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.53712
+    avg_avg_return:                             0.04251412
+    avg_benchmark_regret:                       0.9502574
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.008010864
+    proxy_corner:
+      avg_avg_return:                           0.9065078
+      avg_reward_per_step:                      0.007408142
+  eval-batch-tree:
+    avg_avg_episode_length:                     73.1472
+    avg_avg_return:                             0.18427375
+    avg_benchmark_regret:                       0.7998018
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.027832031
+    proxy_corner:
+      avg_avg_return:                           0.18427375
+      avg_reward_per_step:                      0.027832031
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        171966464
+    env-step-replay-before:                     87457792
+    env-step-used-before:                       171966464
+    ppo-update-before:                          104960
+    env-step-generate-before:                   84508672
+    env-step-all-after:                         171999232
+    env-step-replay-after:                      87457792
+    env-step-used-after:                        171999232
+    ppo-update-after:                           104980
+    env-step-generate-after:                    84541440
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.975866
+    avg_avg_return:                             0.9294364
+    avg_reward_per_step:                        0.15353394
+    proxy_corner:
+      avg_avg_return:                           0.9294364
+      avg_reward_per_step:                      0.15353394
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.975866
+    avg_avg_return:                             0.9294364
+    avg_reward_per_step:                        0.15353394
+    proxy_corner:
+      avg_avg_return:                           0.9294364
+      avg_reward_per_step:                      0.15353394
+  perf:
+    env_steps_per_second:                       349174.959216282
+    ppo_updates_per_second:                     41.32953462075805
+    eval-batch-orig-duration:                   0.5384450377896428
+    eval-batch-shift-duration:                  0.49192136246711016
+    eval-batch-tree-duration:                   0.499259278178215
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999998
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.0
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     681.5857
+      avg_last_visit_time:                      2659.106
+      num_replay_batches:                       2669
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00030798584
+    avg_actor_approxkl3:                        0.00021533054
+    avg_actor_clipfrac:                         0.009552002
+    avg_actor_loss:                             -0.00018539988
+    avg_advantage:                              0.0013017755
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            3.0307147e-05
+    avg_entropy:                                0.110485755
+    avg_grad_norm_pre_clip:                     0.30049643
+    avg_loss:                                   -0.00028073206
+    max:
+      max_actor_approxkl1:                      0.001143906
+      max_actor_approxkl3:                      0.0004310466
+      max_actor_clipfrac:                       0.017700195
+      max_actor_loss:                           0.00034381216
+      max_advantage:                            0.05325675
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.7116493e-05
+      max_entropy:                              0.1298157
+      max_grad_norm_pre_clip:                   0.541415
+      max_loss:                                 0.00024043834
+    std:
+      std_actor_approxkl1:                      0.0003583348
+      std_actor_approxkl3:                      0.00013143843
+      std_actor_clipfrac:                       0.0058393613
+      std_actor_loss:                           0.0003464869
+      std_advantage:                            0.0077529293
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.2012547e-06
+      std_entropy:                              0.008384244
+      std_grad_norm_pre_clip:                   0.105087414
+      std_loss:                                 0.00034749854
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.050388
+    avg_avg_return:                             0.97353274
+    avg_benchmark_regret:                       0.01609679
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13332367
+    proxy_corner:
+      avg_avg_return:                           0.97353274
+      avg_reward_per_step:                      0.13332367
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.76602
+    avg_avg_return:                             0.042008728
+    avg_benchmark_regret:                       0.95076275
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.006843567
+    proxy_corner:
+      avg_avg_return:                           0.8677136
+      avg_reward_per_step:                      0.007095337
+  eval-batch-tree:
+    avg_avg_episode_length:                     71.378006
+    avg_avg_return:                             0.2112126
+    avg_benchmark_regret:                       0.77286303
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.029006958
+    proxy_corner:
+      avg_avg_return:                           0.2112126
+      avg_reward_per_step:                      0.029006958
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        173015040
+    env-step-replay-before:                     88014848
+    env-step-used-before:                       173015040
+    ppo-update-before:                          105600
+    env-step-generate-before:                   85000192
+    env-step-all-after:                         173047808
+    env-step-replay-after:                      88047616
+    env-step-used-after:                        173047808
+    ppo-update-after:                           105620
+    env-step-generate-after:                    85000192
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841639
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841639
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841639
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841639
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       258080.20817451907
+    ppo_updates_per_second:                     41.69202226834945
+    eval-batch-orig-duration:                   0.540508521720767
+    eval-batch-shift-duration:                  0.4847562024369836
+    eval-batch-tree-duration:                   0.47587629687041044
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.999756
+      mouse-cheese_dist_finite_wavg:            18.999985
+      mouse-cheese_dist_solvable_avg:           18.999756
+      mouse-cheese_dist_solvable_wavg:          18.999989
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1403727
+    visit_patterns:
+      avg_first_visit_time:                     690.5842
+      avg_last_visit_time:                      2677.189
+      num_replay_batches:                       2687
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0064628716
+    avg_actor_approxkl3:                        0.006494311
+    avg_actor_clipfrac:                         0.08674317
+    avg_actor_loss:                             7.658471e-05
+    avg_advantage:                              0.00049604283
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.9925584e-05
+    avg_entropy:                                0.253461
+    avg_grad_norm_pre_clip:                     1.3295864
+    avg_loss:                                   -0.00015191342
+    max:
+      max_actor_approxkl1:                      0.01401907
+      max_actor_approxkl3:                      0.014453934
+      max_actor_clipfrac:                       0.16503906
+      max_actor_loss:                           0.006492827
+      max_advantage:                            0.03578931
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          7.7508026e-05
+      max_entropy:                              0.26613218
+      max_grad_norm_pre_clip:                   6.397061
+      max_loss:                                 0.0062654493
+    std:
+      std_actor_approxkl1:                      0.0043399804
+      std_actor_approxkl3:                      0.0042744726
+      std_actor_clipfrac:                       0.049049977
+      std_actor_loss:                           0.0029777077
+      std_advantage:                            0.011402925
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.5038274e-05
+      std_entropy:                              0.006816387
+      std_grad_norm_pre_clip:                   1.6996584
+      std_loss:                                 0.002978992
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.136498
+    avg_avg_return:                             0.9728216
+    avg_benchmark_regret:                       0.016807914
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13265991
+    proxy_corner:
+      avg_avg_return:                           0.9728216
+      avg_reward_per_step:                      0.13265991
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.646515
+    avg_avg_return:                             0.0440233
+    avg_benchmark_regret:                       0.94874823
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0070266724
+    proxy_corner:
+      avg_avg_return:                           0.8765586
+      avg_reward_per_step:                      0.0071792603
+  eval-batch-tree:
+    avg_avg_episode_length:                     70.61212
+    avg_avg_return:                             0.21800935
+    avg_benchmark_regret:                       0.76606625
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.030586243
+    proxy_corner:
+      avg_avg_return:                           0.21800935
+      avg_reward_per_step:                      0.030586243
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        174063616
+    env-step-replay-before:                     88440832
+    env-step-used-before:                       174063616
+    ppo-update-before:                          106240
+    env-step-generate-before:                   85622784
+    env-step-all-after:                         174096384
+    env-step-replay-after:                      88440832
+    env-step-used-after:                        174096384
+    ppo-update-after:                           106260
+    env-step-generate-after:                    85655552
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.316237
+    avg_avg_return:                             0.92818505
+    avg_reward_per_step:                        0.1295166
+    proxy_corner:
+      avg_avg_return:                           0.92818505
+      avg_reward_per_step:                      0.1295166
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.316237
+    avg_avg_return:                             0.92818505
+    avg_reward_per_step:                        0.1295166
+    proxy_corner:
+      avg_avg_return:                           0.92818505
+      avg_reward_per_step:                      0.1295166
+  perf:
+    env_steps_per_second:                       322031.16122694296
+    ppo_updates_per_second:                     41.79385666537569
+    eval-batch-orig-duration:                   0.4962358493357897
+    eval-batch-shift-duration:                  0.4257930014282465
+    eval-batch-tree-duration:                   0.4250048790127039
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.999756
+      mouse-cheese_dist_finite_wavg:            19.000048
+      mouse-cheese_dist_solvable_avg:           18.999756
+      mouse-cheese_dist_solvable_wavg:          19.000048
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14033833
+    visit_patterns:
+      avg_first_visit_time:                     693.9468
+      avg_last_visit_time:                      2689.3713
+      num_replay_batches:                       2699
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00032470046
+    avg_actor_approxkl3:                        0.0002733676
+    avg_actor_clipfrac:                         0.012097168
+    avg_actor_loss:                             -4.4455694e-05
+    avg_advantage:                              -0.0010615075
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.6251995e-05
+    avg_entropy:                                0.0775578
+    avg_grad_norm_pre_clip:                     0.5073129
+    avg_loss:                                   -0.00010888749
+    max:
+      max_actor_approxkl1:                      0.001102562
+      max_actor_approxkl3:                      0.00094165467
+      max_actor_clipfrac:                       0.03918457
+      max_actor_loss:                           0.0011610712
+      max_advantage:                            0.03591013
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.2522403e-05
+      max_entropy:                              0.088636786
+      max_grad_norm_pre_clip:                   1.2634332
+      max_loss:                                 0.0010986108
+    std:
+      std_actor_approxkl1:                      0.0002942789
+      std_actor_approxkl3:                      0.0002022535
+      std_actor_clipfrac:                       0.008479968
+      std_actor_loss:                           0.0004560071
+      std_advantage:                            0.006986638
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.3738431e-06
+      std_entropy:                              0.0059736115
+      std_grad_norm_pre_clip:                   0.3069175
+      std_loss:                                 0.0004588036
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.052864
+    avg_avg_return:                             0.9735384
+    avg_benchmark_regret:                       0.016091134
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13329315
+    proxy_corner:
+      avg_avg_return:                           0.9735384
+      avg_reward_per_step:                      0.13329315
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.200554
+    avg_avg_return:                             0.044499785
+    avg_benchmark_regret:                       0.94827175
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.008232117
+    proxy_corner:
+      avg_avg_return:                           0.8676548
+      avg_reward_per_step:                      0.0071105957
+  eval-batch-tree:
+    avg_avg_episode_length:                     74.10977
+    avg_avg_return:                             0.18119991
+    avg_benchmark_regret:                       0.80287564
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.026947021
+    proxy_corner:
+      avg_avg_return:                           0.18119991
+      avg_reward_per_step:                      0.026947021
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        175112192
+    env-step-replay-before:                     88932352
+    env-step-used-before:                       175112192
+    ppo-update-before:                          106880
+    env-step-generate-before:                   86179840
+    env-step-all-after:                         175144960
+    env-step-replay-after:                      88965120
+    env-step-used-after:                        175144960
+    ppo-update-after:                           106900
+    env-step-generate-after:                    86179840
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       347047.79111242335
+    ppo_updates_per_second:                     41.280350647791096
+    eval-batch-orig-duration:                   0.5381241478025913
+    eval-batch-shift-duration:                  0.4881475120782852
+    eval-batch-tree-duration:                   0.49119828920811415
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.999268
+      mouse-cheese_dist_finite_wavg:            18.999958
+      mouse-cheese_dist_solvable_avg:           18.999268
+      mouse-cheese_dist_solvable_wavg:          18.99996
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1403009
+    visit_patterns:
+      avg_first_visit_time:                     694.4341
+      avg_last_visit_time:                      2705.2983
+      num_replay_batches:                       2715
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012855561
+    avg_actor_approxkl3:                        0.0011317465
+    avg_actor_clipfrac:                         0.012860107
+    avg_actor_loss:                             -0.00025510043
+    avg_advantage:                              0.0021948526
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.675575e-05
+    avg_entropy:                                0.09046158
+    avg_grad_norm_pre_clip:                     0.28907943
+    avg_loss:                                   -0.00033718417
+    max:
+      max_actor_approxkl1:                      0.004087623
+      max_actor_approxkl3:                      0.003541269
+      max_actor_clipfrac:                       0.060791016
+      max_actor_loss:                           0.00048662536
+      max_advantage:                            0.027169824
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.0889114e-05
+      max_entropy:                              0.102555476
+      max_grad_norm_pre_clip:                   0.74243075
+      max_loss:                                 0.0003981528
+    std:
+      std_actor_approxkl1:                      0.0015012779
+      std_actor_approxkl3:                      0.0013159513
+      std_actor_clipfrac:                       0.01384098
+      std_actor_loss:                           0.00045969905
+      std_advantage:                            0.005894416
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.1192087e-06
+      std_entropy:                              0.007106235
+      std_grad_norm_pre_clip:                   0.18554649
+      std_loss:                                 0.0004620565
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.087275
+    avg_avg_return:                             0.97328275
+    avg_benchmark_regret:                       0.016346779
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13211823
+    proxy_corner:
+      avg_avg_return:                           0.97328275
+      avg_reward_per_step:                      0.13211823
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.603134
+    avg_avg_return:                             0.041184757
+    avg_benchmark_regret:                       0.9515867
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0076065063
+    proxy_corner:
+      avg_avg_return:                           0.9140919
+      avg_reward_per_step:                      0.007461548
+  eval-batch-tree:
+    avg_avg_episode_length:                     75.48132
+    avg_avg_return:                             0.16445193
+    avg_benchmark_regret:                       0.81962365
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025985718
+    proxy_corner:
+      avg_avg_return:                           0.16445193
+      avg_reward_per_step:                      0.025985718
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        176160768
+    env-step-replay-before:                     89358336
+    env-step-used-before:                       176160768
+    ppo-update-before:                          107520
+    env-step-generate-before:                   86802432
+    env-step-all-after:                         176193536
+    env-step-replay-after:                      89391104
+    env-step-used-after:                        176193536
+    ppo-update-after:                           107540
+    env-step-generate-after:                    86802432
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184384
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184384
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184384
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184384
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       242319.07208777036
+    ppo_updates_per_second:                     42.301555848100485
+    eval-batch-orig-duration:                   0.6670165797695518
+    eval-batch-shift-duration:                  0.6143519338220358
+    eval-batch-tree-duration:                   0.5803351802751422
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.999512
+      mouse-cheese_dist_finite_wavg:            18.999908
+      mouse-cheese_dist_solvable_avg:           18.999512
+      mouse-cheese_dist_solvable_wavg:          18.99991
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14029932
+    visit_patterns:
+      avg_first_visit_time:                     695.0813
+      avg_last_visit_time:                      2718.3303
+      num_replay_batches:                       2728
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0002232248
+    avg_actor_approxkl3:                        0.00022951206
+    avg_actor_clipfrac:                         0.007867432
+    avg_actor_loss:                             -0.00016720829
+    avg_advantage:                              0.0008644303
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.0842729e-05
+    avg_entropy:                                0.076082334
+    avg_grad_norm_pre_clip:                     0.24570847
+    avg_loss:                                   -0.00023786924
+    max:
+      max_actor_approxkl1:                      0.0007055134
+      max_actor_approxkl3:                      0.00066291937
+      max_actor_clipfrac:                       0.016601562
+      max_actor_loss:                           0.00046674488
+      max_advantage:                            0.025577009
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.3529299e-05
+      max_entropy:                              0.081178084
+      max_grad_norm_pre_clip:                   0.5292331
+      max_loss:                                 0.00039591896
+    std:
+      std_actor_approxkl1:                      0.00023974398
+      std_actor_approxkl3:                      0.00017953162
+      std_actor_clipfrac:                       0.005295443
+      std_actor_loss:                           0.00040511935
+      std_advantage:                            0.0050177434
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.9284898e-06
+      std_entropy:                              0.0021525144
+      std_grad_norm_pre_clip:                   0.114902236
+      std_loss:                                 0.00040487485
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.047863
+    avg_avg_return:                             0.9735076
+    avg_benchmark_regret:                       0.016121954
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13330841
+    proxy_corner:
+      avg_avg_return:                           0.9735076
+      avg_reward_per_step:                      0.13330841
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.45737
+    avg_avg_return:                             0.04087958
+    avg_benchmark_regret:                       0.9518919
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.007774353
+    proxy_corner:
+      avg_avg_return:                           0.8918849
+      avg_reward_per_step:                      0.007293701
+  eval-batch-tree:
+    avg_avg_episode_length:                     72.046295
+    avg_avg_return:                             0.1970753
+    avg_benchmark_regret:                       0.7870003
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.029975891
+    proxy_corner:
+      avg_avg_return:                           0.1970753
+      avg_reward_per_step:                      0.029975891
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        177209344
+    env-step-replay-before:                     89784320
+    env-step-used-before:                       177209344
+    ppo-update-before:                          108160
+    env-step-generate-before:                   87425024
+    env-step-all-after:                         177242112
+    env-step-replay-after:                      89784320
+    env-step-used-after:                        177242112
+    ppo-update-after:                           108180
+    env-step-generate-after:                    87457792
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.373545
+    avg_avg_return:                             0.92608446
+    avg_reward_per_step:                        0.12423706
+    proxy_corner:
+      avg_avg_return:                           0.92608446
+      avg_reward_per_step:                      0.12423706
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.373545
+    avg_avg_return:                             0.92608446
+    avg_reward_per_step:                        0.12423706
+    proxy_corner:
+      avg_avg_return:                           0.92608446
+      avg_reward_per_step:                      0.12423706
+  perf:
+    env_steps_per_second:                       353676.8598840163
+    ppo_updates_per_second:                     44.53135434750388
+    eval-batch-orig-duration:                   0.47734636161476374
+    eval-batch-shift-duration:                  0.4215113576501608
+    eval-batch-tree-duration:                   0.4207220422104001
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.0
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.0
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14031091
+    visit_patterns:
+      avg_first_visit_time:                     697.92065
+      avg_last_visit_time:                      2730.275
+      num_replay_batches:                       2740
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00078915537
+    avg_actor_approxkl3:                        0.00055028003
+    avg_actor_clipfrac:                         0.026373291
+    avg_actor_loss:                             -0.00015537579
+    avg_advantage:                              0.0007335013
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            3.800628e-05
+    avg_entropy:                                0.085133456
+    avg_grad_norm_pre_clip:                     0.48448706
+    avg_loss:                                   -0.00022150611
+    max:
+      max_actor_approxkl1:                      0.001753187
+      max_actor_approxkl3:                      0.0015487958
+      max_actor_clipfrac:                       0.0657959
+      max_actor_loss:                           0.0013424803
+      max_advantage:                            0.051442564
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          4.597578e-05
+      max_entropy:                              0.09770669
+      max_grad_norm_pre_clip:                   1.2129589
+      max_loss:                                 0.0012813314
+    std:
+      std_actor_approxkl1:                      0.00038339637
+      std_actor_approxkl3:                      0.00032307525
+      std_actor_clipfrac:                       0.016213102
+      std_actor_loss:                           0.000657598
+      std_advantage:                            0.008558437
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.0134136e-06
+      std_entropy:                              0.0074582007
+      std_grad_norm_pre_clip:                   0.270848
+      std_loss:                                 0.00065782235
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.127361
+    avg_avg_return:                             0.97296906
+    avg_benchmark_regret:                       0.016660444
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13283539
+    proxy_corner:
+      avg_avg_return:                           0.97296906
+      avg_reward_per_step:                      0.13283539
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.91655
+    avg_avg_return:                             0.037701566
+    avg_benchmark_regret:                       0.9550699
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.009925842
+    proxy_corner:
+      avg_avg_return:                           0.84795475
+      avg_reward_per_step:                      0.0069351196
+  eval-batch-tree:
+    avg_avg_episode_length:                     70.00523
+    avg_avg_return:                             0.22777283
+    avg_benchmark_regret:                       0.7563027
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.031852722
+    proxy_corner:
+      avg_avg_return:                           0.22777283
+      avg_reward_per_step:                      0.031852722
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        178257920
+    env-step-replay-before:                     90439680
+    env-step-used-before:                       178257920
+    ppo-update-before:                          108800
+    env-step-generate-before:                   87818240
+    env-step-all-after:                         178290688
+    env-step-replay-after:                      90472448
+    env-step-used-after:                        178290688
+    ppo-update-after:                           108820
+    env-step-generate-after:                    87818240
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       320722.01149234426
+    ppo_updates_per_second:                     41.14284178092129
+    eval-batch-orig-duration:                   0.5268241185694933
+    eval-batch-shift-duration:                  0.5001489305868745
+    eval-batch-tree-duration:                   0.48387280106544495
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.999756
+      mouse-cheese_dist_finite_wavg:            19.000023
+      mouse-cheese_dist_solvable_avg:           18.999756
+      mouse-cheese_dist_solvable_wavg:          19.000027
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.140302
+    visit_patterns:
+      avg_first_visit_time:                     704.50244
+      avg_last_visit_time:                      2751.365
+      num_replay_batches:                       2761
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00047408065
+    avg_actor_approxkl3:                        0.0003058058
+    avg_actor_clipfrac:                         0.015386963
+    avg_actor_loss:                             -4.6893034e-05
+    avg_advantage:                              -0.0013893489
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.6539303e-05
+    avg_entropy:                                0.05955724
+    avg_grad_norm_pre_clip:                     0.15637611
+    avg_loss:                                   -9.818063e-05
+    max:
+      max_actor_approxkl1:                      0.0013423547
+      max_actor_approxkl3:                      0.00074366
+      max_actor_clipfrac:                       0.056274414
+      max_actor_loss:                           0.00016969489
+      max_advantage:                            0.063370585
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.9080077e-05
+      max_entropy:                              0.061324015
+      max_grad_norm_pre_clip:                   0.27956033
+      max_loss:                                 0.00011712978
+    std:
+      std_actor_approxkl1:                      0.00034561148
+      std_actor_approxkl3:                      0.00022809966
+      std_actor_clipfrac:                       0.019922625
+      std_actor_loss:                           0.00012093676
+      std_advantage:                            0.006188976
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.4982573e-06
+      std_entropy:                              0.0009210052
+      std_grad_norm_pre_clip:                   0.060388133
+      std_loss:                                 0.00012050639
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040766
+    avg_avg_return:                             0.9737601
+    avg_benchmark_regret:                       0.015869394
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13349915
+    proxy_corner:
+      avg_avg_return:                           0.9737601
+      avg_reward_per_step:                      0.13349915
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.20387
+    avg_avg_return:                             0.043310292
+    avg_benchmark_regret:                       0.9494612
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.009674072
+    proxy_corner:
+      avg_avg_return:                           0.79648316
+      avg_reward_per_step:                      0.0065231323
+  eval-batch-tree:
+    avg_avg_episode_length:                     74.27179
+    avg_avg_return:                             0.1752463
+    avg_benchmark_regret:                       0.8088293
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.028640747
+    proxy_corner:
+      avg_avg_return:                           0.1752463
+      avg_reward_per_step:                      0.028640747
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        179306496
+    env-step-replay-before:                     90931200
+    env-step-used-before:                       179306496
+    ppo-update-before:                          109440
+    env-step-generate-before:                   88375296
+    env-step-all-after:                         179339264
+    env-step-replay-after:                      90931200
+    env-step-used-after:                        179339264
+    ppo-update-after:                           109460
+    env-step-generate-after:                    88408064
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.960389
+    avg_avg_return:                             0.9285531
+    avg_reward_per_step:                        0.13259888
+    proxy_corner:
+      avg_avg_return:                           0.9285531
+      avg_reward_per_step:                      0.13259888
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.960389
+    avg_avg_return:                             0.9285531
+    avg_reward_per_step:                        0.13259888
+    proxy_corner:
+      avg_avg_return:                           0.9285531
+      avg_reward_per_step:                      0.13259888
+  perf:
+    env_steps_per_second:                       349379.0959318877
+    ppo_updates_per_second:                     41.19163388035402
+    eval-batch-orig-duration:                   0.5348911620676517
+    eval-batch-shift-duration:                  0.4833829263225198
+    eval-batch-tree-duration:                   0.48403806798160076
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.999756
+      mouse-cheese_dist_finite_wavg:            18.999947
+      mouse-cheese_dist_solvable_avg:           18.999756
+      mouse-cheese_dist_solvable_wavg:          18.999947
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1402989
+    visit_patterns:
+      avg_first_visit_time:                     704.50244
+      avg_last_visit_time:                      2765.3733
+      num_replay_batches:                       2775
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        1.3278717e-05
+    avg_actor_approxkl3:                        5.167031e-05
+    avg_actor_clipfrac:                         0.0016601563
+    avg_actor_loss:                             1.432281e-06
+    avg_advantage:                              -0.00019904174
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.9276203e-05
+    avg_entropy:                                0.011144438
+    avg_grad_norm_pre_clip:                     0.1373175
+    avg_loss:                                   4.9259456e-06
+    max:
+      max_actor_approxkl1:                      0.00075348874
+      max_actor_approxkl3:                      0.00024840364
+      max_actor_clipfrac:                       0.0056152344
+      max_actor_loss:                           0.0004050224
+      max_advantage:                            0.05288261
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.4490586e-05
+      max_entropy:                              0.017332897
+      max_grad_norm_pre_clip:                   0.31429905
+      max_loss:                                 0.00040968857
+    std:
+      std_actor_approxkl1:                      0.00021308986
+      std_actor_approxkl3:                      5.221197e-05
+      std_actor_clipfrac:                       0.0013391064
+      std_actor_loss:                           0.00015808042
+      std_advantage:                            0.007888291
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          3.7383384e-06
+      std_entropy:                              0.0025423246
+      std_grad_norm_pre_clip:                   0.078728214
+      std_loss:                                 0.00015838454
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040624
+    avg_avg_return:                             0.9738064
+    avg_benchmark_regret:                       0.015823146
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13350677
+    proxy_corner:
+      avg_avg_return:                           0.9738064
+      avg_reward_per_step:                      0.13350677
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.51066
+    avg_avg_return:                             0.03942656
+    avg_benchmark_regret:                       0.95334494
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.010810852
+    proxy_corner:
+      avg_avg_return:                           0.74662673
+      avg_reward_per_step:                      0.0060577393
+  eval-batch-tree:
+    avg_avg_episode_length:                     72.020584
+    avg_avg_return:                             0.19267702
+    avg_benchmark_regret:                       0.79139864
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.03261566
+    proxy_corner:
+      avg_avg_return:                           0.19267702
+      avg_reward_per_step:                      0.03261566
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        180355072
+    env-step-replay-before:                     91291648
+    env-step-used-before:                       180355072
+    ppo-update-before:                          110080
+    env-step-generate-before:                   89063424
+    env-step-all-after:                         180387840
+    env-step-replay-after:                      91324416
+    env-step-used-after:                        180387840
+    ppo-update-after:                           110100
+    env-step-generate-after:                    89063424
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       259860.32499281652
+    ppo_updates_per_second:                     41.42613072814467
+    eval-batch-orig-duration:                   0.5377742229029536
+    eval-batch-shift-duration:                  0.4897976880893111
+    eval-batch-tree-duration:                   0.4870816068723798
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.999756
+      mouse-cheese_dist_finite_wavg:            18.999954
+      mouse-cheese_dist_solvable_avg:           18.999756
+      mouse-cheese_dist_solvable_wavg:          18.999952
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1402989
+    visit_patterns:
+      avg_first_visit_time:                     704.50244
+      avg_last_visit_time:                      2777.2312
+      num_replay_batches:                       2787
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0003754291
+    avg_actor_approxkl3:                        0.00024311061
+    avg_actor_clipfrac:                         0.01619873
+    avg_actor_loss:                             0.0003697551
+    avg_advantage:                              -9.378891e-06
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            9.118442e-06
+    avg_entropy:                                0.0317002
+    avg_grad_norm_pre_clip:                     0.981348
+    avg_loss:                                   0.00034261408
+    max:
+      max_actor_approxkl1:                      0.0010145705
+      max_actor_approxkl3:                      0.0005898346
+      max_actor_clipfrac:                       0.03527832
+      max_actor_loss:                           0.002638285
+      max_advantage:                            0.031831563
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.0605705e-05
+      max_entropy:                              0.03658647
+      max_grad_norm_pre_clip:                   2.6838202
+      max_loss:                                 0.0026090427
+    std:
+      std_actor_approxkl1:                      0.0003251494
+      std_actor_approxkl3:                      0.00018076485
+      std_actor_clipfrac:                       0.014105016
+      std_actor_loss:                           0.000806835
+      std_advantage:                            0.0045043305
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          9.398967e-07
+      std_entropy:                              0.0019221294
+      std_grad_norm_pre_clip:                   0.67188233
+      std_loss:                                 0.0008066044
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040735
+    avg_avg_return:                             0.9737759
+    avg_benchmark_regret:                       0.015853588
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13350677
+    proxy_corner:
+      avg_avg_return:                           0.9737759
+      avg_reward_per_step:                      0.13350677
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.291115
+    avg_avg_return:                             0.04181664
+    avg_benchmark_regret:                       0.9509549
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.011245728
+    proxy_corner:
+      avg_avg_return:                           0.7161858
+      avg_reward_per_step:                      0.0058059692
+  eval-batch-tree:
+    avg_avg_episode_length:                     69.80521
+    avg_avg_return:                             0.22084352
+    avg_benchmark_regret:                       0.7632321
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.034217834
+    proxy_corner:
+      avg_avg_return:                           0.22084352
+      avg_reward_per_step:                      0.034217834
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        181403648
+    env-step-replay-before:                     91947008
+    env-step-used-before:                       181403648
+    ppo-update-before:                          110720
+    env-step-generate-before:                   89456640
+    env-step-all-after:                         181436416
+    env-step-replay-after:                      91979776
+    env-step-used-after:                        181436416
+    ppo-update-after:                           110740
+    env-step-generate-after:                    89456640
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.682053
+    avg_avg_return:                             0.8419168
+    avg_reward_per_step:                        0.046905518
+    proxy_corner:
+      avg_avg_return:                           0.8419168
+      avg_reward_per_step:                      0.046905518
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.682053
+    avg_avg_return:                             0.8419168
+    avg_reward_per_step:                        0.046905518
+    proxy_corner:
+      avg_avg_return:                           0.8419168
+      avg_reward_per_step:                      0.046905518
+  perf:
+    env_steps_per_second:                       315513.48805298086
+    ppo_updates_per_second:                     42.14357385820586
+    eval-batch-orig-duration:                   0.5447512688115239
+    eval-batch-shift-duration:                  0.48130456916987896
+    eval-batch-tree-duration:                   0.48474538046866655
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.999756
+      mouse-cheese_dist_finite_wavg:            18.999947
+      mouse-cheese_dist_solvable_avg:           18.999756
+      mouse-cheese_dist_solvable_wavg:          18.99995
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.1402989
+    visit_patterns:
+      avg_first_visit_time:                     704.50244
+      avg_last_visit_time:                      2797.327
+      num_replay_batches:                       2807
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0002719737
+    avg_actor_approxkl3:                        0.0003141021
+    avg_actor_clipfrac:                         0.010888672
+    avg_actor_loss:                             0.00016937917
+    avg_advantage:                              -0.00017812187
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            9.828807e-06
+    avg_entropy:                                0.031987537
+    avg_grad_norm_pre_clip:                     0.48284832
+    avg_loss:                                   0.00014230605
+    max:
+      max_actor_approxkl1:                      0.0013574454
+      max_actor_approxkl3:                      0.0014774143
+      max_actor_clipfrac:                       0.037597656
+      max_actor_loss:                           0.0009988472
+      max_advantage:                            0.024365187
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.1363414e-05
+      max_entropy:                              0.03677735
+      max_grad_norm_pre_clip:                   1.5481864
+      max_loss:                                 0.0009690282
+    std:
+      std_actor_approxkl1:                      0.00039412748
+      std_actor_approxkl3:                      0.00038458692
+      std_actor_clipfrac:                       0.012295795
+      std_actor_loss:                           0.00034687348
+      std_advantage:                            0.0046064714
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          8.069523e-07
+      std_entropy:                              0.0022568242
+      std_grad_norm_pre_clip:                   0.43216348
+      std_loss:                                 0.00034667554
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040616
+    avg_avg_return:                             0.9738064
+    avg_benchmark_regret:                       0.015823131
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.1335144
+    proxy_corner:
+      avg_avg_return:                           0.9738064
+      avg_reward_per_step:                      0.1335144
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.24989
+    avg_avg_return:                             0.043065984
+    avg_benchmark_regret:                       0.9497055
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.012001038
+    proxy_corner:
+      avg_avg_return:                           0.6672304
+      avg_reward_per_step:                      0.0054092407
+  eval-batch-tree:
+    avg_avg_episode_length:                     69.85918
+    avg_avg_return:                             0.22212729
+    avg_benchmark_regret:                       0.76194835
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.035957336
+    proxy_corner:
+      avg_avg_return:                           0.22212729
+      avg_reward_per_step:                      0.035957336
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        182452224
+    env-step-replay-before:                     92471296
+    env-step-used-before:                       182452224
+    ppo-update-before:                          111360
+    env-step-generate-before:                   89980928
+    env-step-all-after:                         182484992
+    env-step-replay-after:                      92504064
+    env-step-used-after:                        182484992
+    ppo-update-after:                           111380
+    env-step-generate-after:                    89980928
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       242270.48902000012
+    ppo_updates_per_second:                     42.32480970083589
+    eval-batch-orig-duration:                   0.6657428508624434
+    eval-batch-shift-duration:                  0.6219077249988914
+    eval-batch-tree-duration:                   0.6147431265562773
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.997803
+      mouse-cheese_dist_finite_wavg:            18.999882
+      mouse-cheese_dist_solvable_avg:           18.997803
+      mouse-cheese_dist_solvable_wavg:          18.999882
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030102
+    visit_patterns:
+      avg_first_visit_time:                     705.0791
+      avg_last_visit_time:                      2813.1782
+      num_replay_batches:                       2823
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011431095
+    avg_actor_approxkl3:                        0.0010314694
+    avg_actor_clipfrac:                         0.025177002
+    avg_actor_loss:                             0.00023954675
+    avg_advantage:                              -0.00044846258
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.3377364e-05
+    avg_entropy:                                0.059587926
+    avg_grad_norm_pre_clip:                     0.61441606
+    avg_loss:                                   0.00018664745
+    max:
+      max_actor_approxkl1:                      0.0028830036
+      max_actor_approxkl3:                      0.0022144727
+      max_actor_clipfrac:                       0.0625
+      max_actor_loss:                           0.0028227898
+      max_advantage:                            0.03344989
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.7459846e-05
+      max_entropy:                              0.07231569
+      max_grad_norm_pre_clip:                   1.5936513
+      max_loss:                                 0.0027761944
+    std:
+      std_actor_approxkl1:                      0.00074909464
+      std_actor_approxkl3:                      0.00069372286
+      std_actor_clipfrac:                       0.017523909
+      std_actor_loss:                           0.0011097187
+      std_advantage:                            0.005786065
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.968309e-06
+      std_entropy:                              0.0054878904
+      std_grad_norm_pre_clip:                   0.44602287
+      std_loss:                                 0.0011108541
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.12793
+    avg_avg_return:                             0.9735768
+    avg_benchmark_regret:                       0.01605267
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13249207
+    proxy_corner:
+      avg_avg_return:                           0.9735768
+      avg_reward_per_step:                      0.13249207
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.625565
+    avg_avg_return:                             0.03842176
+    avg_benchmark_regret:                       0.95434976
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.00642395
+    proxy_corner:
+      avg_avg_return:                           0.84063256
+      avg_reward_per_step:                      0.006881714
+  eval-batch-tree:
+    avg_avg_episode_length:                     70.76619
+    avg_avg_return:                             0.21300085
+    avg_benchmark_regret:                       0.7710748
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.03326416
+    proxy_corner:
+      avg_avg_return:                           0.21300085
+      avg_reward_per_step:                      0.03326416
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        183500800
+    env-step-replay-before:                     93028352
+    env-step-used-before:                       183500800
+    ppo-update-before:                          112000
+    env-step-generate-before:                   90472448
+    env-step-all-after:                         183533568
+    env-step-replay-after:                      93061120
+    env-step-used-after:                        183533568
+    ppo-update-after:                           112020
+    env-step-generate-after:                    90472448
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8418417
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8418417
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8418417
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8418417
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       295471.0925258311
+    ppo_updates_per_second:                     42.229438733822825
+    eval-batch-orig-duration:                   0.536653196439147
+    eval-batch-shift-duration:                  0.4769313493743539
+    eval-batch-tree-duration:                   0.4799665743485093
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.000244
+      mouse-cheese_dist_finite_wavg:            19.89911
+      mouse-cheese_dist_solvable_avg:           19.000244
+      mouse-cheese_dist_solvable_wavg:          19.899109
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030743
+    visit_patterns:
+      avg_first_visit_time:                     705.7654
+      avg_last_visit_time:                      2830.333
+      num_replay_batches:                       2840
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0005591185
+    avg_actor_approxkl3:                        0.0005198382
+    avg_actor_clipfrac:                         0.006903076
+    avg_actor_loss:                             -4.0578725e-05
+    avg_advantage:                              -0.00019837689
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            9.5462965e-06
+    avg_entropy:                                0.040751122
+    avg_grad_norm_pre_clip:                     0.4222928
+    avg_loss:                                   -7.6556695e-05
+    max:
+      max_actor_approxkl1:                      0.0027820347
+      max_actor_approxkl3:                      0.002409386
+      max_actor_clipfrac:                       0.012817383
+      max_actor_loss:                           0.00072789274
+      max_advantage:                            0.03846246
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.1469613e-05
+      max_entropy:                              0.05006868
+      max_grad_norm_pre_clip:                   1.1362299
+      max_loss:                                 0.00069614657
+    std:
+      std_actor_approxkl1:                      0.00075756985
+      std_actor_approxkl3:                      0.00071177597
+      std_actor_clipfrac:                       0.0038523723
+      std_actor_loss:                           0.00034864453
+      std_advantage:                            0.00462472
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.3194352e-06
+      std_entropy:                              0.0037919632
+      std_grad_norm_pre_clip:                   0.3210924
+      std_loss:                                 0.00035048838
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.042834
+    avg_avg_return:                             0.97361976
+    avg_benchmark_regret:                       0.016009744
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13342285
+    proxy_corner:
+      avg_avg_return:                           0.97361976
+      avg_reward_per_step:                      0.13342285
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.486496
+    avg_avg_return:                             0.042127036
+    avg_benchmark_regret:                       0.9506445
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.008834839
+    proxy_corner:
+      avg_avg_return:                           0.8207327
+      avg_reward_per_step:                      0.006706238
+  eval-batch-tree:
+    avg_avg_episode_length:                     72.19864
+    avg_avg_return:                             0.19831033
+    avg_benchmark_regret:                       0.78576523
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.030303955
+    proxy_corner:
+      avg_avg_return:                           0.19831033
+      avg_reward_per_step:                      0.030303955
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        184549376
+    env-step-replay-before:                     93618176
+    env-step-used-before:                       184549376
+    ppo-update-before:                          112640
+    env-step-generate-before:                   90931200
+    env-step-all-after:                         184582144
+    env-step-replay-after:                      93650944
+    env-step-used-after:                        184582144
+    ppo-update-after:                           112660
+    env-step-generate-after:                    90931200
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       313375.69764595473
+    ppo_updates_per_second:                     42.17359165849612
+    eval-batch-orig-duration:                   0.5398517241701484
+    eval-batch-shift-duration:                  0.5104422811418772
+    eval-batch-tree-duration:                   0.4936314010992646
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.000244
+      mouse-cheese_dist_finite_wavg:            19.000038
+      mouse-cheese_dist_solvable_avg:           19.000244
+      mouse-cheese_dist_solvable_wavg:          19.000042
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030743
+    visit_patterns:
+      avg_first_visit_time:                     705.7654
+      avg_last_visit_time:                      2848.28
+      num_replay_batches:                       2858
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00012251844
+    avg_actor_approxkl3:                        0.00019629374
+    avg_actor_clipfrac:                         0.0056030275
+    avg_actor_loss:                             -0.00012138011
+    avg_advantage:                              -0.0021308912
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            5.5436285e-06
+    avg_entropy:                                0.02842983
+    avg_grad_norm_pre_clip:                     0.16528778
+    avg_loss:                                   -0.00014703812
+    max:
+      max_actor_approxkl1:                      0.0004056238
+      max_actor_approxkl3:                      0.00041976373
+      max_actor_clipfrac:                       0.009033203
+      max_actor_loss:                           0.00016141683
+      max_advantage:                            0.028860152
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          8.07555e-06
+      max_entropy:                              0.03127514
+      max_grad_norm_pre_clip:                   0.28595844
+      max_loss:                                 0.00013732174
+    std:
+      std_actor_approxkl1:                      0.00020385362
+      std_actor_approxkl3:                      0.000115618794
+      std_actor_clipfrac:                       0.0025336281
+      std_actor_loss:                           0.00013333761
+      std_advantage:                            0.0033410501
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.2822921e-06
+      std_entropy:                              0.002404949
+      std_grad_norm_pre_clip:                   0.061062403
+      std_loss:                                 0.00013350708
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.041681
+    avg_avg_return:                             0.97383195
+    avg_benchmark_regret:                       0.015797574
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13348389
+    proxy_corner:
+      avg_avg_return:                           0.97383195
+      avg_reward_per_step:                      0.13348389
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.48093
+    avg_avg_return:                             0.043687515
+    avg_benchmark_regret:                       0.94908404
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.008773804
+    proxy_corner:
+      avg_avg_return:                           0.8144098
+      avg_reward_per_step:                      0.0066833496
+  eval-batch-tree:
+    avg_avg_episode_length:                     71.44484
+    avg_avg_return:                             0.20504287
+    avg_benchmark_regret:                       0.7790327
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.03175354
+    proxy_corner:
+      avg_avg_return:                           0.20504287
+      avg_reward_per_step:                      0.03175354
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        185597952
+    env-step-replay-before:                     94142464
+    env-step-used-before:                       185597952
+    ppo-update-before:                          113280
+    env-step-generate-before:                   91455488
+    env-step-all-after:                         185630720
+    env-step-replay-after:                      94175232
+    env-step-used-after:                        185630720
+    ppo-update-after:                           113300
+    env-step-generate-after:                    91455488
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       261087.3970937178
+    ppo_updates_per_second:                     41.401957414237344
+    eval-batch-orig-duration:                   0.5364781515672803
+    eval-batch-shift-duration:                  0.48270936217159033
+    eval-batch-tree-duration:                   0.45091725047677755
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.0
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.0
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     705.77075
+      avg_last_visit_time:                      2864.2888
+      num_replay_batches:                       2874
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0008032803
+    avg_actor_approxkl3:                        0.00037144704
+    avg_actor_clipfrac:                         0.0050720214
+    avg_actor_loss:                             -0.00027337318
+    avg_advantage:                              -8.777306e-05
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            4.026431e-06
+    avg_entropy:                                0.022206709
+    avg_grad_norm_pre_clip:                     0.2659631
+    avg_loss:                                   -0.00029356667
+    max:
+      max_actor_approxkl1:                      0.003798094
+      max_actor_approxkl3:                      0.0027149988
+      max_actor_clipfrac:                       0.029541016
+      max_actor_loss:                           0.0007455908
+      max_advantage:                            0.02394408
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          6.136802e-06
+      max_entropy:                              0.029793791
+      max_grad_norm_pre_clip:                   0.6340718
+      max_loss:                                 0.0007198916
+    std:
+      std_actor_approxkl1:                      0.0011076388
+      std_actor_approxkl3:                      0.0006572689
+      std_actor_clipfrac:                       0.008009196
+      std_actor_loss:                           0.0005645762
+      std_advantage:                            0.002815546
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          9.63591e-07
+      std_entropy:                              0.0035768508
+      std_grad_norm_pre_clip:                   0.13072549
+      std_loss:                                 0.00056623406
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.043871
+    avg_avg_return:                             0.9737954
+    avg_benchmark_regret:                       0.01583415
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.133461
+    proxy_corner:
+      avg_avg_return:                           0.9737954
+      avg_reward_per_step:                      0.133461
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.16992
+    avg_avg_return:                             0.047279797
+    avg_benchmark_regret:                       0.9454918
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0093307495
+    proxy_corner:
+      avg_avg_return:                           0.8318515
+      avg_reward_per_step:                      0.0068588257
+  eval-batch-tree:
+    avg_avg_episode_length:                     74.033646
+    avg_avg_return:                             0.17868313
+    avg_benchmark_regret:                       0.8053925
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.02935791
+    proxy_corner:
+      avg_avg_return:                           0.17868313
+      avg_reward_per_step:                      0.02935791
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        186646528
+    env-step-replay-before:                     94765056
+    env-step-used-before:                       186646528
+    ppo-update-before:                          113920
+    env-step-generate-before:                   91881472
+    env-step-all-after:                         186679296
+    env-step-replay-after:                      94797824
+    env-step-used-after:                        186679296
+    ppo-update-after:                           113940
+    env-step-generate-after:                    91881472
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       312404.293912115
+    ppo_updates_per_second:                     42.23821491774202
+    eval-batch-orig-duration:                   0.5429349709302187
+    eval-batch-shift-duration:                  0.49664298444986343
+    eval-batch-tree-duration:                   0.4879840547218919
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999998
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.0
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     705.77075
+      avg_last_visit_time:                      2883.2231
+      num_replay_batches:                       2893
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00057425717
+    avg_actor_approxkl3:                        0.00055767514
+    avg_actor_clipfrac:                         0.008685303
+    avg_actor_loss:                             -9.618579e-05
+    avg_advantage:                              -0.00089193834
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.1744405e-05
+    avg_entropy:                                0.024575826
+    avg_grad_norm_pre_clip:                     0.27083707
+    avg_loss:                                   -0.00011488941
+    max:
+      max_actor_approxkl1:                      0.0019538165
+      max_actor_approxkl3:                      0.0012712642
+      max_actor_clipfrac:                       0.032226562
+      max_actor_loss:                           0.0005580485
+      max_advantage:                            0.024413168
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.8901546e-05
+      max_entropy:                              0.03322605
+      max_grad_norm_pre_clip:                   0.7660044
+      max_loss:                                 0.00053423055
+    std:
+      std_actor_approxkl1:                      0.00053198694
+      std_actor_approxkl3:                      0.0003818479
+      std_actor_clipfrac:                       0.009681206
+      std_actor_loss:                           0.0003067253
+      std_advantage:                            0.005289841
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.8281681e-06
+      std_entropy:                              0.0039055182
+      std_grad_norm_pre_clip:                   0.21585736
+      std_loss:                                 0.00030721308
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.040616
+    avg_avg_return:                             0.97380626
+    avg_benchmark_regret:                       0.01582328
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.1335144
+    proxy_corner:
+      avg_avg_return:                           0.97380626
+      avg_reward_per_step:                      0.1335144
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.00442
+    avg_avg_return:                             0.046732657
+    avg_benchmark_regret:                       0.94603884
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.011291504
+    proxy_corner:
+      avg_avg_return:                           0.75195694
+      avg_reward_per_step:                      0.006187439
+  eval-batch-tree:
+    avg_avg_episode_length:                     73.03934
+    avg_avg_return:                             0.1844916
+    avg_benchmark_regret:                       0.79958403
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.03240204
+    proxy_corner:
+      avg_avg_return:                           0.1844916
+      avg_reward_per_step:                      0.03240204
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        187695104
+    env-step-replay-before:                     95420416
+    env-step-used-before:                       187695104
+    ppo-update-before:                          114560
+    env-step-generate-before:                   92274688
+    env-step-all-after:                         187727872
+    env-step-replay-after:                      95420416
+    env-step-used-after:                        187727872
+    ppo-update-after:                           114580
+    env-step-generate-after:                    92307456
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.711446
+    avg_avg_return:                             0.9278594
+    avg_reward_per_step:                        0.1541748
+    proxy_corner:
+      avg_avg_return:                           0.9278594
+      avg_reward_per_step:                      0.1541748
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.711446
+    avg_avg_return:                             0.9278594
+    avg_reward_per_step:                        0.1541748
+    proxy_corner:
+      avg_avg_return:                           0.9278594
+      avg_reward_per_step:                      0.1541748
+  perf:
+    env_steps_per_second:                       262453.3313794154
+    ppo_updates_per_second:                     41.478613443781256
+    eval-batch-orig-duration:                   0.5352215925231576
+    eval-batch-shift-duration:                  0.48230751138180494
+    eval-batch-tree-duration:                   0.4760555922985077
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.99707
+      mouse-cheese_dist_finite_wavg:            18.999926
+      mouse-cheese_dist_solvable_avg:           18.99707
+      mouse-cheese_dist_solvable_wavg:          18.999928
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14037077
+    visit_patterns:
+      avg_first_visit_time:                     727.97656
+      avg_last_visit_time:                      2902.1262
+      num_replay_batches:                       2912
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00073638017
+    avg_actor_approxkl3:                        0.00097082736
+    avg_actor_clipfrac:                         0.026934816
+    avg_actor_loss:                             -6.190441e-05
+    avg_advantage:                              -0.0028097571
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            8.5441054e-05
+    avg_entropy:                                0.103472255
+    avg_grad_norm_pre_clip:                     0.85111874
+    avg_loss:                                   -0.00012265616
+    max:
+      max_actor_approxkl1:                      0.0020480738
+      max_actor_approxkl3:                      0.0020117979
+      max_actor_clipfrac:                       0.045288086
+      max_actor_loss:                           0.00084734336
+      max_advantage:                            0.08448136
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00011350775
+      max_entropy:                              0.12759438
+      max_grad_norm_pre_clip:                   2.1574967
+      max_loss:                                 0.0007862094
+    std:
+      std_actor_approxkl1:                      0.00056142535
+      std_actor_approxkl3:                      0.0005749375
+      std_actor_clipfrac:                       0.011545185
+      std_actor_loss:                           0.0004600802
+      std_advantage:                            0.014082763
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.280992e-05
+      std_entropy:                              0.011230081
+      std_grad_norm_pre_clip:                   0.5253694
+      std_loss:                                 0.00045894313
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.119782
+    avg_avg_return:                             0.9727154
+    avg_benchmark_regret:                       0.016914193
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13198853
+    proxy_corner:
+      avg_avg_return:                           0.9727154
+      avg_reward_per_step:                      0.13198853
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.96658
+    avg_avg_return:                             0.03489574
+    avg_benchmark_regret:                       0.9578758
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.009857178
+    proxy_corner:
+      avg_avg_return:                           0.744398
+      avg_reward_per_step:                      0.0060653687
+  eval-batch-tree:
+    avg_avg_episode_length:                     68.80597
+    avg_avg_return:                             0.23196429
+    avg_benchmark_regret:                       0.7521113
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.038215637
+    proxy_corner:
+      avg_avg_return:                           0.23196429
+      avg_reward_per_step:                      0.038215637
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        188743680
+    env-step-replay-before:                     95911936
+    env-step-used-before:                       188743680
+    ppo-update-before:                          115200
+    env-step-generate-before:                   92831744
+    env-step-all-after:                         188776448
+    env-step-replay-after:                      95911936
+    env-step-used-after:                        188776448
+    ppo-update-after:                           115220
+    env-step-generate-after:                    92864512
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.730337
+    avg_avg_return:                             0.92951965
+    avg_reward_per_step:                        0.13894653
+    proxy_corner:
+      avg_avg_return:                           0.92951965
+      avg_reward_per_step:                      0.13894653
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.730337
+    avg_avg_return:                             0.92951965
+    avg_reward_per_step:                        0.13894653
+    proxy_corner:
+      avg_avg_return:                           0.92951965
+      avg_reward_per_step:                      0.13894653
+  perf:
+    env_steps_per_second:                       262306.1177746934
+    ppo_updates_per_second:                     41.27114281252713
+    eval-batch-orig-duration:                   0.5329056717455387
+    eval-batch-shift-duration:                  0.4861464211717248
+    eval-batch-tree-duration:                   0.4820699580013752
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.997314
+      mouse-cheese_dist_finite_wavg:            18.999685
+      mouse-cheese_dist_solvable_avg:           18.997314
+      mouse-cheese_dist_solvable_wavg:          18.999687
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14033869
+    visit_patterns:
+      avg_first_visit_time:                     734.5823
+      avg_last_visit_time:                      2917.3733
+      num_replay_batches:                       2927
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00049169426
+    avg_actor_approxkl3:                        0.0006173145
+    avg_actor_clipfrac:                         0.023223877
+    avg_actor_loss:                             0.0008109484
+    avg_advantage:                              0.0007651129
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.9504217e-05
+    avg_entropy:                                0.064263076
+    avg_grad_norm_pre_clip:                     1.1049851
+    avg_loss:                                   0.0007614374
+    max:
+      max_actor_approxkl1:                      0.0030007837
+      max_actor_approxkl3:                      0.001882023
+      max_actor_clipfrac:                       0.061767578
+      max_actor_loss:                           0.0034556733
+      max_advantage:                            0.035517395
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          4.0621944e-05
+      max_entropy:                              0.076884374
+      max_grad_norm_pre_clip:                   2.4885015
+      max_loss:                                 0.003407319
+    std:
+      std_actor_approxkl1:                      0.0006787566
+      std_actor_approxkl3:                      0.0004780519
+      std_actor_clipfrac:                       0.016574113
+      std_actor_loss:                           0.0011690303
+      std_advantage:                            0.006761348
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          5.9064832e-06
+      std_entropy:                              0.0065005915
+      std_grad_norm_pre_clip:                   0.770998
+      std_loss:                                 0.001168205
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.042141
+    avg_avg_return:                             0.97359365
+    avg_benchmark_regret:                       0.016035873
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13347626
+    proxy_corner:
+      avg_avg_return:                           0.97359365
+      avg_reward_per_step:                      0.13347626
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.56593
+    avg_avg_return:                             0.04367477
+    avg_benchmark_regret:                       0.94909674
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0099105835
+    proxy_corner:
+      avg_avg_return:                           0.88212264
+      avg_reward_per_step:                      0.007217407
+  eval-batch-tree:
+    avg_avg_episode_length:                     68.970695
+    avg_avg_return:                             0.23281467
+    avg_benchmark_regret:                       0.751261
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.036643982
+    proxy_corner:
+      avg_avg_return:                           0.23281467
+      avg_reward_per_step:                      0.036643982
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        189792256
+    env-step-replay-before:                     96501760
+    env-step-used-before:                       189792256
+    ppo-update-before:                          115840
+    env-step-generate-before:                   93290496
+    env-step-all-after:                         189825024
+    env-step-replay-after:                      96534528
+    env-step-used-after:                        189825024
+    ppo-update-after:                           115860
+    env-step-generate-after:                    93290496
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.682053
+    avg_avg_return:                             0.8419135
+    avg_reward_per_step:                        0.046905518
+    proxy_corner:
+      avg_avg_return:                           0.8419135
+      avg_reward_per_step:                      0.046905518
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.682053
+    avg_avg_return:                             0.8419135
+    avg_reward_per_step:                        0.046905518
+    proxy_corner:
+      avg_avg_return:                           0.8419135
+      avg_reward_per_step:                      0.046905518
+  perf:
+    env_steps_per_second:                       257647.05559726525
+    ppo_updates_per_second:                     42.41772139339778
+    eval-batch-orig-duration:                   0.6297100484371185
+    eval-batch-shift-duration:                  0.6082249004393816
+    eval-batch-tree-duration:                   0.5597612587735057
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.998535
+      mouse-cheese_dist_finite_wavg:            18.99994
+      mouse-cheese_dist_solvable_avg:           18.998535
+      mouse-cheese_dist_solvable_wavg:          18.99994
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14027387
+    visit_patterns:
+      avg_first_visit_time:                     734.63354
+      avg_last_visit_time:                      2936.2852
+      num_replay_batches:                       2946
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002262222
+    avg_actor_approxkl3:                        0.0014610587
+    avg_actor_clipfrac:                         0.049188234
+    avg_actor_loss:                             -0.00083148404
+    avg_advantage:                              0.00030588685
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            8.527517e-06
+    avg_entropy:                                0.063343234
+    avg_grad_norm_pre_clip:                     0.61006767
+    avg_loss:                                   -0.00089056365
+    max:
+      max_actor_approxkl1:                      0.003583231
+      max_actor_approxkl3:                      0.0024421127
+      max_actor_clipfrac:                       0.083618164
+      max_actor_loss:                           0.00087996945
+      max_advantage:                            0.02944082
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.09040575e-05
+      max_entropy:                              0.068858415
+      max_grad_norm_pre_clip:                   1.1969311
+      max_loss:                                 0.00082404155
+    std:
+      std_actor_approxkl1:                      0.0010402052
+      std_actor_approxkl3:                      0.0007356792
+      std_actor_clipfrac:                       0.024869783
+      std_actor_loss:                           0.00073772884
+      std_advantage:                            0.004144546
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.2502401e-06
+      std_entropy:                              0.0020513393
+      std_grad_norm_pre_clip:                   0.32265434
+      std_loss:                                 0.0007383436
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.042181
+    avg_avg_return:                             0.9737758
+    avg_benchmark_regret:                       0.01585371
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.133461
+    proxy_corner:
+      avg_avg_return:                           0.9737758
+      avg_reward_per_step:                      0.133461
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.40329
+    avg_avg_return:                             0.047411278
+    avg_benchmark_regret:                       0.9453603
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.007789612
+    proxy_corner:
+      avg_avg_return:                           0.87481964
+      avg_reward_per_step:                      0.007217407
+  eval-batch-tree:
+    avg_avg_episode_length:                     70.848976
+    avg_avg_return:                             0.20734468
+    avg_benchmark_regret:                       0.7767309
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.034599304
+    proxy_corner:
+      avg_avg_return:                           0.20734468
+      avg_reward_per_step:                      0.034599304
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        190840832
+    env-step-replay-before:                     96894976
+    env-step-used-before:                       190840832
+    ppo-update-before:                          116480
+    env-step-generate-before:                   93945856
+    env-step-all-after:                         190873600
+    env-step-replay-after:                      96894976
+    env-step-used-after:                        190873600
+    ppo-update-after:                           116500
+    env-step-generate-after:                    93978624
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.214502
+    avg_avg_return:                             0.92678213
+    avg_reward_per_step:                        0.13391113
+    proxy_corner:
+      avg_avg_return:                           0.92678213
+      avg_reward_per_step:                      0.13391113
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.214502
+    avg_avg_return:                             0.92678213
+    avg_reward_per_step:                        0.13391113
+    proxy_corner:
+      avg_avg_return:                           0.92678213
+      avg_reward_per_step:                      0.13391113
+  perf:
+    env_steps_per_second:                       240271.93278411863
+    ppo_updates_per_second:                     41.79896777504917
+    eval-batch-orig-duration:                   0.5460228500887752
+    eval-batch-shift-duration:                  0.5101973405107856
+    eval-batch-tree-duration:                   0.49277640134096146
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.999512
+      mouse-cheese_dist_finite_wavg:            18.999882
+      mouse-cheese_dist_solvable_avg:           18.999512
+      mouse-cheese_dist_solvable_wavg:          18.999884
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14028203
+    visit_patterns:
+      avg_first_visit_time:                     734.66284
+      avg_last_visit_time:                      2947.4648
+      num_replay_batches:                       2957
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00073957955
+    avg_actor_approxkl3:                        0.0006231048
+    avg_actor_clipfrac:                         0.017803956
+    avg_actor_loss:                             -0.0003512186
+    avg_advantage:                              0.00031050338
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.729705e-05
+    avg_entropy:                                0.07167809
+    avg_grad_norm_pre_clip:                     0.41280708
+    avg_loss:                                   -0.00041424818
+    max:
+      max_actor_approxkl1:                      0.0021302933
+      max_actor_approxkl3:                      0.0014970348
+      max_actor_clipfrac:                       0.03894043
+      max_actor_loss:                           0.0015821103
+      max_advantage:                            0.032425523
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.0575793e-05
+      max_entropy:                              0.08335361
+      max_grad_norm_pre_clip:                   0.8440171
+      max_loss:                                 0.0015292522
+    std:
+      std_actor_approxkl1:                      0.0006613675
+      std_actor_approxkl3:                      0.0004208343
+      std_actor_clipfrac:                       0.011696303
+      std_actor_loss:                           0.00063617405
+      std_advantage:                            0.006118846
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.7042721e-06
+      std_entropy:                              0.0064506475
+      std_grad_norm_pre_clip:                   0.19129613
+      std_loss:                                 0.00063758704
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.051109
+    avg_avg_return:                             0.9734931
+    avg_benchmark_regret:                       0.016136397
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13317871
+    proxy_corner:
+      avg_avg_return:                           0.9734931
+      avg_reward_per_step:                      0.13317871
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.215
+    avg_avg_return:                             0.048146725
+    avg_benchmark_regret:                       0.9446248
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0046691895
+    proxy_corner:
+      avg_avg_return:                           0.88494587
+      avg_reward_per_step:                      0.007331848
+  eval-batch-tree:
+    avg_avg_episode_length:                     71.01251
+    avg_avg_return:                             0.20616706
+    avg_benchmark_regret:                       0.77790856
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.032318115
+    proxy_corner:
+      avg_avg_return:                           0.20616706
+      avg_reward_per_step:                      0.032318115
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        191889408
+    env-step-replay-before:                     97484800
+    env-step-used-before:                       191889408
+    ppo-update-before:                          117120
+    env-step-generate-before:                   94404608
+    env-step-all-after:                         191922176
+    env-step-replay-after:                      97517568
+    env-step-used-after:                        191922176
+    ppo-update-after:                           117140
+    env-step-generate-after:                    94404608
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184384
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184384
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184384
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184384
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       247239.24082075703
+    ppo_updates_per_second:                     46.17190662097562
+    eval-batch-orig-duration:                   0.514816178008914
+    eval-batch-shift-duration:                  0.6268036048859358
+    eval-batch-tree-duration:                   0.4783758046105504
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.999512
+      mouse-cheese_dist_finite_wavg:            18.999947
+      mouse-cheese_dist_solvable_avg:           18.999512
+      mouse-cheese_dist_solvable_wavg:          18.999945
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14029044
+    visit_patterns:
+      avg_first_visit_time:                     734.7117
+      avg_last_visit_time:                      2966.3027
+      num_replay_batches:                       2976
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009263597
+    avg_actor_approxkl3:                        0.00086453784
+    avg_actor_clipfrac:                         0.030194093
+    avg_actor_loss:                             -0.00033375743
+    avg_advantage:                              0.0009751675
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.5731477e-05
+    avg_entropy:                                0.15201324
+    avg_grad_norm_pre_clip:                     0.5284526
+    avg_loss:                                   -0.00047790492
+    max:
+      max_actor_approxkl1:                      0.0023425107
+      max_actor_approxkl3:                      0.0020721506
+      max_actor_clipfrac:                       0.08520508
+      max_actor_loss:                           0.00021838397
+      max_advantage:                            0.020898938
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          1.7528422e-05
+      max_entropy:                              0.16027991
+      max_grad_norm_pre_clip:                   1.0028628
+      max_loss:                                 7.629508e-05
+    std:
+      std_actor_approxkl1:                      0.00059691747
+      std_actor_approxkl3:                      0.0005407189
+      std_actor_clipfrac:                       0.020453693
+      std_actor_loss:                           0.00029808548
+      std_advantage:                            0.0055442997
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          9.075818e-07
+      std_entropy:                              0.0041673514
+      std_grad_norm_pre_clip:                   0.24807246
+      std_loss:                                 0.00030002109
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.051972
+    avg_avg_return:                             0.9731438
+    avg_benchmark_regret:                       0.016485693
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.1332016
+    proxy_corner:
+      avg_avg_return:                           0.9731438
+      avg_reward_per_step:                      0.1332016
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.708176
+    avg_avg_return:                             0.04496274
+    avg_benchmark_regret:                       0.9478088
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0060653687
+    proxy_corner:
+      avg_avg_return:                           0.90325063
+      avg_reward_per_step:                      0.0074539185
+  eval-batch-tree:
+    avg_avg_episode_length:                     71.13369
+    avg_avg_return:                             0.20236881
+    avg_benchmark_regret:                       0.7817068
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.03276825
+    proxy_corner:
+      avg_avg_return:                           0.20236881
+      avg_reward_per_step:                      0.03276825
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        192937984
+    env-step-replay-before:                     97910784
+    env-step-used-before:                       192937984
+    ppo-update-before:                          117760
+    env-step-generate-before:                   95027200
+    env-step-all-after:                         192970752
+    env-step-replay-after:                      97943552
+    env-step-used-after:                        192970752
+    ppo-update-after:                           117780
+    env-step-generate-after:                    95027200
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8418395
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8418395
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.8418395
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.8418395
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       288128.5329957238
+    ppo_updates_per_second:                     42.27011641890862
+    eval-batch-orig-duration:                   0.5944239944219589
+    eval-batch-shift-duration:                  0.5350807411596179
+    eval-batch-tree-duration:                   0.5633231000974774
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999985
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          18.999985
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030193
+    visit_patterns:
+      avg_first_visit_time:                     736.40796
+      avg_last_visit_time:                      2979.2239
+      num_replay_batches:                       2989
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0007865733
+    avg_actor_approxkl3:                        0.0007272473
+    avg_actor_clipfrac:                         0.032128908
+    avg_actor_loss:                             -0.00019323817
+    avg_advantage:                              0.00030256936
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.957974e-05
+    avg_entropy:                                0.12999606
+    avg_grad_norm_pre_clip:                     0.5698461
+    avg_loss:                                   -0.00031344433
+    max:
+      max_actor_approxkl1:                      0.0020921896
+      max_actor_approxkl3:                      0.00146422
+      max_actor_clipfrac:                       0.072753906
+      max_actor_loss:                           0.0011800779
+      max_advantage:                            0.040438652
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.4983347e-05
+      max_entropy:                              0.14449726
+      max_grad_norm_pre_clip:                   1.5865707
+      max_loss:                                 0.0010611651
+    std:
+      std_actor_approxkl1:                      0.00070837134
+      std_actor_approxkl3:                      0.0005093431
+      std_actor_clipfrac:                       0.020880647
+      std_actor_loss:                           0.0005578466
+      std_advantage:                            0.005963803
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.8274112e-06
+      std_entropy:                              0.0071490323
+      std_grad_norm_pre_clip:                   0.33070457
+      std_loss:                                 0.00055868056
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.081577
+    avg_avg_return:                             0.9731379
+    avg_benchmark_regret:                       0.016491596
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13253784
+    proxy_corner:
+      avg_avg_return:                           0.9731379
+      avg_reward_per_step:                      0.13253784
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.976944
+    avg_avg_return:                             0.055373207
+    avg_benchmark_regret:                       0.9373983
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.0072784424
+    proxy_corner:
+      avg_avg_return:                           0.89967
+      avg_reward_per_step:                      0.0074768066
+  eval-batch-tree:
+    avg_avg_episode_length:                     71.77371
+    avg_avg_return:                             0.19894135
+    avg_benchmark_regret:                       0.7851342
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.03200531
+    proxy_corner:
+      avg_avg_return:                           0.19894135
+      avg_reward_per_step:                      0.03200531
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        193986560
+    env-step-replay-before:                     98402304
+    env-step-used-before:                       193986560
+    ppo-update-before:                          118400
+    env-step-generate-before:                   95584256
+    env-step-all-after:                         194019328
+    env-step-replay-after:                      98435072
+    env-step-used-after:                        194019328
+    ppo-update-after:                           118420
+    env-step-generate-after:                    95584256
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184337
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184337
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184337
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184337
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       731911.2725236687
+    ppo_updates_per_second:                     42.205612076293825
+    eval-batch-orig-duration:                   0.30549150332808495
+    eval-batch-shift-duration:                  0.26402335334569216
+    eval-batch-tree-duration:                   0.2388425450772047
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.99994
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          18.99994
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030649
+    visit_patterns:
+      avg_first_visit_time:                     738.25415
+      avg_last_visit_time:                      2994.2786
+      num_replay_batches:                       3004
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.004511115
+    avg_actor_approxkl3:                        0.0034834407
+    avg_actor_clipfrac:                         0.05126343
+    avg_actor_loss:                             -0.0012608416
+    avg_advantage:                              -0.0011051451
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.6112408e-05
+    avg_entropy:                                0.11040951
+    avg_grad_norm_pre_clip:                     0.49884272
+    avg_loss:                                   -0.0013581947
+    max:
+      max_actor_approxkl1:                      0.01048142
+      max_actor_approxkl3:                      0.010714442
+      max_actor_clipfrac:                       0.10949707
+      max_actor_loss:                           0.0029287338
+      max_advantage:                            0.033297062
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.5788777e-05
+      max_entropy:                              0.11950809
+      max_grad_norm_pre_clip:                   1.4281273
+      max_loss:                                 0.0028301203
+    std:
+      std_actor_approxkl1:                      0.003509963
+      std_actor_approxkl3:                      0.0031192368
+      std_actor_clipfrac:                       0.031421248
+      std_actor_loss:                           0.0021415872
+      std_advantage:                            0.0067242663
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.129146e-06
+      std_entropy:                              0.005105629
+      std_grad_norm_pre_clip:                   0.27750835
+      std_loss:                                 0.0021435143
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.047694
+    avg_avg_return:                             0.97319525
+    avg_benchmark_regret:                       0.01643431
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13339996
+    proxy_corner:
+      avg_avg_return:                           0.97319525
+      avg_reward_per_step:                      0.13339996
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.8784
+    avg_avg_return:                             0.054592244
+    avg_benchmark_regret:                       0.93817925
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.01096344
+    proxy_corner:
+      avg_avg_return:                           0.8933747
+      avg_reward_per_step:                      0.0073928833
+  eval-batch-tree:
+    avg_avg_episode_length:                     71.86171
+    avg_avg_return:                             0.20315444
+    avg_benchmark_regret:                       0.78092116
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.031280518
+    proxy_corner:
+      avg_avg_return:                           0.20315444
+      avg_reward_per_step:                      0.031280518
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        195035136
+    env-step-replay-before:                     98762752
+    env-step-used-before:                       195035136
+    ppo-update-before:                          119040
+    env-step-generate-before:                   96272384
+    env-step-all-after:                         195067904
+    env-step-replay-after:                      98795520
+    env-step-used-after:                        195067904
+    ppo-update-after:                           119060
+    env-step-generate-after:                    96272384
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184057
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184057
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.84184057
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.84184057
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       295192.15715379646
+    ppo_updates_per_second:                     40.95244178557492
+    eval-batch-orig-duration:                   0.5228665499016643
+    eval-batch-shift-duration:                  0.48954899702221155
+    eval-batch-tree-duration:                   0.4834571862593293
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.00002
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.00002
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030433
+    visit_patterns:
+      avg_first_visit_time:                     738.25415
+      avg_last_visit_time:                      3005.36
+      num_replay_batches:                       3015
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010936133
+    avg_actor_approxkl3:                        0.0015100371
+    avg_actor_clipfrac:                         0.024041748
+    avg_actor_loss:                             6.2524414e-05
+    avg_advantage:                              -0.0004334912
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.939698e-05
+    avg_entropy:                                0.08415978
+    avg_grad_norm_pre_clip:                     1.0118762
+    avg_loss:                                   -1.1936861e-05
+    max:
+      max_actor_approxkl1:                      0.0029902647
+      max_actor_approxkl3:                      0.0033474402
+      max_actor_clipfrac:                       0.0546875
+      max_actor_loss:                           0.0026435442
+      max_advantage:                            0.0346148
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.4865622e-05
+      max_entropy:                              0.09286294
+      max_grad_norm_pre_clip:                   2.4994738
+      max_loss:                                 0.0025730096
+    std:
+      std_actor_approxkl1:                      0.0009551992
+      std_actor_approxkl3:                      0.0009330739
+      std_actor_clipfrac:                       0.015662124
+      std_actor_loss:                           0.000985347
+      std_advantage:                            0.00613809
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.1890164e-06
+      std_entropy:                              0.0037308044
+      std_grad_norm_pre_clip:                   0.68278134
+      std_loss:                                 0.0009867091
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.044561
+    avg_avg_return:                             0.9736602
+    avg_benchmark_regret:                       0.015969366
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13343811
+    proxy_corner:
+      avg_avg_return:                           0.9736602
+      avg_reward_per_step:                      0.13343811
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.12793
+    avg_avg_return:                             0.058987375
+    avg_benchmark_regret:                       0.9337841
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.010505676
+    proxy_corner:
+      avg_avg_return:                           0.8938866
+      avg_reward_per_step:                      0.007423401
+  eval-batch-tree:
+    avg_avg_episode_length:                     75.210236
+    avg_avg_return:                             0.15974772
+    avg_benchmark_regret:                       0.8243279
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.028366089
+    proxy_corner:
+      avg_avg_return:                           0.15974772
+      avg_reward_per_step:                      0.028366089
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        196083712
+    env-step-replay-before:                     99319808
+    env-step-used-before:                       196083712
+    ppo-update-before:                          119680
+    env-step-generate-before:                   96763904
+    env-step-all-after:                         196116480
+    env-step-replay-after:                      99319808
+    env-step-used-after:                        196116480
+    ppo-update-after:                           119700
+    env-step-generate-after:                    96796672
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.119835
+    avg_avg_return:                             0.92697674
+    avg_reward_per_step:                        0.13168335
+    proxy_corner:
+      avg_avg_return:                           0.92697674
+      avg_reward_per_step:                      0.13168335
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.119835
+    avg_avg_return:                             0.92697674
+    avg_reward_per_step:                        0.13168335
+    proxy_corner:
+      avg_avg_return:                           0.92697674
+      avg_reward_per_step:                      0.13168335
+  perf:
+    env_steps_per_second:                       317771.6503091833
+    ppo_updates_per_second:                     42.21366156064413
+    eval-batch-orig-duration:                   0.5381756275892258
+    eval-batch-shift-duration:                  0.4817835334688425
+    eval-batch-tree-duration:                   0.4782347595319152
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.000002
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.000004
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     738.30664
+      avg_last_visit_time:                      3021.331
+      num_replay_batches:                       3031
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00034343093
+    avg_actor_approxkl3:                        0.00029547382
+    avg_actor_clipfrac:                         0.012945557
+    avg_actor_loss:                             0.00022611981
+    avg_advantage:                              -0.0005866944
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            2.3206148e-05
+    avg_entropy:                                0.056978833
+    avg_grad_norm_pre_clip:                     0.6915299
+    avg_loss:                                   0.00018074403
+    max:
+      max_actor_approxkl1:                      0.001378563
+      max_actor_approxkl3:                      0.001086553
+      max_actor_clipfrac:                       0.03540039
+      max_actor_loss:                           0.0016582655
+      max_advantage:                            0.042114556
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          3.303517e-05
+      max_entropy:                              0.06823774
+      max_grad_norm_pre_clip:                   1.4433496
+      max_loss:                                 0.0016142705
+    std:
+      std_actor_approxkl1:                      0.00033614578
+      std_actor_approxkl3:                      0.0002554119
+      std_actor_clipfrac:                       0.0089633595
+      std_actor_loss:                           0.0006412059
+      std_advantage:                            0.006153576
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.8016605e-06
+      std_entropy:                              0.005497265
+      std_grad_norm_pre_clip:                   0.39712575
+      std_loss:                                 0.00064107653
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.049067
+    avg_avg_return:                             0.97360575
+    avg_benchmark_regret:                       0.016023742
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13343048
+    proxy_corner:
+      avg_avg_return:                           0.97360575
+      avg_reward_per_step:                      0.13343048
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.60834
+    avg_avg_return:                             0.06543053
+    avg_benchmark_regret:                       0.927341
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.010734558
+    proxy_corner:
+      avg_avg_return:                           0.86848974
+      avg_reward_per_step:                      0.007217407
+  eval-batch-tree:
+    avg_avg_episode_length:                     78.652405
+    avg_avg_return:                             0.10178125
+    avg_benchmark_regret:                       0.88229436
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025146484
+    proxy_corner:
+      avg_avg_return:                           0.10178125
+      avg_reward_per_step:                      0.025146484
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        197132288
+    env-step-replay-before:                     99876864
+    env-step-used-before:                       197132288
+    ppo-update-before:                          120320
+    env-step-generate-before:                   97255424
+    env-step-all-after:                         197165056
+    env-step-replay-after:                      99876864
+    env-step-used-after:                        197165056
+    ppo-update-after:                           120340
+    env-step-generate-after:                    97288192
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.937727
+    avg_avg_return:                             0.92865705
+    avg_reward_per_step:                        0.13574219
+    proxy_corner:
+      avg_avg_return:                           0.92865705
+      avg_reward_per_step:                      0.13574219
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     9.937727
+    avg_avg_return:                             0.92865705
+    avg_reward_per_step:                        0.13574219
+    proxy_corner:
+      avg_avg_return:                           0.92865705
+      avg_reward_per_step:                      0.13574219
+  perf:
+    env_steps_per_second:                       298582.2313223213
+    ppo_updates_per_second:                     40.845784066348955
+    eval-batch-orig-duration:                   0.5164672676473856
+    eval-batch-shift-duration:                  0.47893856931477785
+    eval-batch-tree-duration:                   0.4582347059622407
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            18.999998
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          18.999998
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     738.99805
+      avg_last_visit_time:                      3038.1685
+      num_replay_batches:                       3048
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00038236642
+    avg_actor_approxkl3:                        0.00023415025
+    avg_actor_clipfrac:                         0.0046142577
+    avg_actor_loss:                             -0.00012377829
+    avg_advantage:                              0.0003621496
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.9549057e-05
+    avg_entropy:                                0.02438481
+    avg_grad_norm_pre_clip:                     0.2696956
+    avg_loss:                                   -0.00013838858
+    max:
+      max_actor_approxkl1:                      0.0015943742
+      max_actor_approxkl3:                      0.0011016171
+      max_actor_clipfrac:                       0.018920898
+      max_actor_loss:                           0.00013536401
+      max_advantage:                            0.040382743
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.23061e-05
+      max_entropy:                              0.029552374
+      max_grad_norm_pre_clip:                   0.859179
+      max_loss:                                 0.00011791715
+    std:
+      std_actor_approxkl1:                      0.0003859752
+      std_actor_approxkl3:                      0.0002425202
+      std_actor_clipfrac:                       0.0044211703
+      std_actor_loss:                           0.00020907965
+      std_advantage:                            0.006125677
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.5646829e-06
+      std_entropy:                              0.0024785385
+      std_grad_norm_pre_clip:                   0.1708725
+      std_loss:                                 0.00020876073
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.047307
+    avg_avg_return:                             0.9738724
+    avg_benchmark_regret:                       0.015757073
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13343811
+    proxy_corner:
+      avg_avg_return:                           0.9738724
+      avg_reward_per_step:                      0.13343811
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.043045
+    avg_avg_return:                             0.060008284
+    avg_benchmark_regret:                       0.9327632
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.011665344
+    proxy_corner:
+      avg_avg_return:                           0.87882876
+      avg_reward_per_step:                      0.0072250366
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.19386
+    avg_avg_return:                             0.0832794
+    avg_benchmark_regret:                       0.9007962
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.025138855
+    proxy_corner:
+      avg_avg_return:                           0.0832794
+      avg_reward_per_step:                      0.025138855
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        198180864
+    env-step-replay-before:                     100335616
+    env-step-used-before:                       198180864
+    ppo-update-before:                          120960
+    env-step-generate-before:                   97845248
+    env-step-all-after:                         198213632
+    env-step-replay-after:                      100368384
+    env-step-used-after:                        198213632
+    ppo-update-after:                           120980
+    env-step-generate-after:                    97845248
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.692308
+    avg_avg_return:                             0.841845
+    avg_reward_per_step:                        0.046875
+    proxy_corner:
+      avg_avg_return:                           0.841845
+      avg_reward_per_step:                      0.046875
+  perf:
+    env_steps_per_second:                       322572.17055478424
+    ppo_updates_per_second:                     41.995560287471626
+    eval-batch-orig-duration:                   0.5376551253721118
+    eval-batch-shift-duration:                  0.4885155260562897
+    eval-batch-tree-duration:                   0.4894653791561723
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.000006
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.000008
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     738.99805
+      avg_last_visit_time:                      3053.3474
+      num_replay_batches:                       3063
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0038659964
+    avg_actor_approxkl3:                        0.0032406175
+    avg_actor_clipfrac:                         0.022070313
+    avg_actor_loss:                             0.0009654299
+    avg_advantage:                              -0.00012401864
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.86206e-05
+    avg_entropy:                                0.0486973
+    avg_grad_norm_pre_clip:                     0.7232383
+    avg_loss:                                   0.00092604285
+    max:
+      max_actor_approxkl1:                      0.012990085
+      max_actor_approxkl3:                      0.012805719
+      max_actor_clipfrac:                       0.08203125
+      max_actor_loss:                           0.0071117394
+      max_advantage:                            0.031457245
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.554268e-05
+      max_entropy:                              0.07446231
+      max_grad_norm_pre_clip:                   1.8298762
+      max_loss:                                 0.0070478534
+    std:
+      std_actor_approxkl1:                      0.003604619
+      std_actor_approxkl3:                      0.0034513792
+      std_actor_clipfrac:                       0.024296826
+      std_actor_loss:                           0.0019202983
+      std_advantage:                            0.0070162513
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          4.1862977e-06
+      std_entropy:                              0.012332612
+      std_grad_norm_pre_clip:                   0.54111683
+      std_loss:                                 0.0019143493
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.045263
+    avg_avg_return:                             0.97376204
+    avg_benchmark_regret:                       0.015867455
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.133461
+    proxy_corner:
+      avg_avg_return:                           0.97376204
+      avg_reward_per_step:                      0.133461
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.83298
+    avg_avg_return:                             0.05981571
+    avg_benchmark_regret:                       0.9329558
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.012199402
+    proxy_corner:
+      avg_avg_return:                           0.8831201
+      avg_reward_per_step:                      0.007255554
+  eval-batch-tree:
+    avg_avg_episode_length:                     78.63097
+    avg_avg_return:                             0.09321932
+    avg_benchmark_regret:                       0.89085627
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.026000977
+    proxy_corner:
+      avg_avg_return:                           0.09321932
+      avg_reward_per_step:                      0.026000977
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        199229440
+    env-step-replay-before:                     100728832
+    env-step-used-before:                       199229440
+    ppo-update-before:                          121600
+    env-step-generate-before:                   98500608
+    env-step-all-after:                         199262208
+    env-step-replay-after:                      100728832
+    env-step-used-after:                        199262208
+    ppo-update-after:                           121620
+    env-step-generate-after:                    98533376
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.0024395
+    avg_avg_return:                             0.9273442
+    avg_reward_per_step:                        0.14428711
+    proxy_corner:
+      avg_avg_return:                           0.9273442
+      avg_reward_per_step:                      0.14428711
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     10.0024395
+    avg_avg_return:                             0.9273442
+    avg_reward_per_step:                        0.14428711
+    proxy_corner:
+      avg_avg_return:                           0.9273442
+      avg_reward_per_step:                      0.14428711
+  perf:
+    env_steps_per_second:                       297601.5046583777
+    ppo_updates_per_second:                     42.2462667048863
+    eval-batch-orig-duration:                   0.5346216009929776
+    eval-batch-shift-duration:                  0.48660353664308786
+    eval-batch-tree-duration:                   0.48448570910841227
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.0
+      mouse-cheese_dist_finite_wavg:            19.0
+      mouse-cheese_dist_solvable_avg:           19.0
+      mouse-cheese_dist_solvable_wavg:          19.0
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            0.0
+      num_walls_wavg:                           0.0
+      prop_walls_avg:                           0.0
+      prop_walls_wavg:                          0.0
+    scoring:
+      avg_scores:                               0.14030746
+    visit_patterns:
+      avg_first_visit_time:                     738.99805
+      avg_last_visit_time:                      3064.2952
+      num_replay_batches:                       3074
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00041527717
+    avg_actor_approxkl3:                        0.00022391097
+    avg_actor_clipfrac:                         0.009179688
+    avg_actor_loss:                             -0.00015995264
+    avg_advantage:                              -6.539156e-05
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            1.7502449e-05
+    avg_entropy:                                0.037783552
+    avg_grad_norm_pre_clip:                     0.34002167
+    avg_loss:                                   -0.00018898495
+    max:
+      max_actor_approxkl1:                      0.0010867545
+      max_actor_approxkl3:                      0.00042568165
+      max_actor_clipfrac:                       0.018310547
+      max_actor_loss:                           0.00028100144
+      max_advantage:                            0.03855455
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          2.3794966e-05
+      max_entropy:                              0.042690367
+      max_grad_norm_pre_clip:                   0.60991377
+      max_loss:                                 0.00025083206
+    std:
+      std_actor_approxkl1:                      0.00036725227
+      std_actor_approxkl3:                      9.924831e-05
+      std_actor_clipfrac:                       0.0046986127
+      std_actor_loss:                           0.0003213772
+      std_advantage:                            0.005673133
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          2.3786074e-06
+      std_entropy:                              0.003757248
+      std_grad_norm_pre_clip:                   0.12230228
+      std_loss:                                 0.00032132078
+  eval-batch-orig:
+    avg_avg_episode_length:                     10.042458
+    avg_avg_return:                             0.97383887
+    avg_benchmark_regret:                       0.015790626
+    avg_benchmark_return:                       0.9896295
+    avg_reward_per_step:                        0.13348389
+    proxy_corner:
+      avg_avg_return:                           0.97383887
+      avg_reward_per_step:                      0.13348389
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.849106
+    avg_avg_return:                             0.064331554
+    avg_benchmark_regret:                       0.92844
+    avg_benchmark_return:                       0.9927715
+    avg_reward_per_step:                        0.013160706
+    proxy_corner:
+      avg_avg_return:                           0.88270414
+      avg_reward_per_step:                      0.007331848
+  eval-batch-tree:
+    avg_avg_episode_length:                     76.809326
+    avg_avg_return:                             0.13043764
+    avg_benchmark_regret:                       0.85363793
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.028709412
+    proxy_corner:
+      avg_avg_return:                           0.13043764
+      avg_reward_per_step:                      0.028709412
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+training run complete.
+finishing checkpoints...
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/requirements.txt b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/requirements.txt
new file mode 100644
index 0000000000000000000000000000000000000000..1d52de8af834763af26ecb7bea4d057f7b5df2d1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/requirements.txt
@@ -0,0 +1,252 @@
+kiwisolver==1.4.8
+aulis==0.1.0
+tokenizers==0.20.3
+xmltodict==0.14.2
+hf-xet==1.1.4
+timaeus-cli==0.1.0
+jsonlines==4.0.0
+DataProperty==1.1.0
+timaeus-aether==0.0.0
+nvidia-nvtx-cu12==12.6.77
+cachetools==5.5.2
+devinterp==1.3.2
+wcwidth==0.2.13
+pydantic_core==2.33.2
+tqdm-multiprocess==0.0.11
+protobuf==5.29.5
+decorator==4.4.2
+pyasn1==0.6.1
+zipp==3.23.0
+pre_commit==4.2.0
+typepy==1.3.4
+etils==1.13.0
+shellingham==1.5.4
+aiohttp==3.12.13
+prettytable==3.16.0
+readchar==4.2.1
+sentencepiece==0.2.0
+seaborn==0.13.2
+paramiko==3.5.1
+cftime==1.6.4.post1
+tabledata==1.3.4
+colorama==0.4.6
+GitPython==3.1.44
+cycler==0.12.1
+tcolorpy==0.1.7
+cfgv==3.4.0
+psutil==7.0.0
+aioitertools==0.12.0
+aiosignal==1.3.2
+humanize==4.12.3
+termcolor==3.1.0
+zope.interface==7.2
+pathvalidate==3.3.1
+nodeenv==1.9.1
+mbstrdecoder==1.1.4
+chardet==5.2.0
+zope.event==5.0
+cloudpickle==3.1.1
+opt_einsum==3.4.0
+MarkupSafe==3.0.2
+requests_ntlm==1.3.0
+pybind11==2.13.6
+fonttools==4.58.4
+gast==0.6.0
+packaging==25.0
+proglog==0.1.12
+Jinja2==3.1.6
+wadler_lindig==0.1.6
+fancy-einsum==0.0.3
+iniconfig==2.1.0
+pluggy==1.6.0
+idna==3.10
+nest-asyncio==1.6.0
+better-abc==0.0.3
+matplotlib==3.10.3
+portalocker==3.2.0
+tqdm==4.67.1
+smmap==5.0.2
+rich==14.0.0
+distro==1.9.0
+xarray==2025.6.1
+bcrypt==4.3.0
+regex==2024.11.6
+aiobotocore==2.22.0
+jmespath==1.0.1
+frozenlist==1.7.0
+typer==0.16.0
+cryptography==45.0.4
+python-dotenv==1.1.0
+pytest-xdist==3.7.0
+boto3==1.37.3
+grpc-google-iam-v1==0.14.2
+threadpoolctl==3.6.0
+aiohappyeyeballs==2.6.1
+pywinrm==0.5.0
+platformdirs==4.3.8
+nvidia-cuda-runtime-cu12==12.6.77
+aiofiles==24.1.0
+google-api-core==2.25.1
+proto-plus==1.26.1
+nvidia-cufile-cu12==1.11.1.6
+mpmath==1.3.0
+PyYAML==6.0.2
+filelock==3.18.0
+typing-inspection==0.4.1
+syrupy==4.9.1
+distlib==0.3.9
+annotated-types==0.7.0
+more-itertools==10.7.0
+multiprocess==0.70.16
+grpcio-status==1.71.2
+six==1.17.0
+requests==2.32.4
+mdurl==0.1.2
+pyinfra==3.3.1
+typing_extensions==4.14.0
+greenlet==3.2.3
+lxml==5.4.0
+einops==0.8.1
+ssh-python==1.1.1
+networkx==3.5
+attrs==25.3.0
+pyasn1_modules==0.4.2
+toolz==1.0.0
+joblib==1.5.1
+diskcache==5.6.3
+multidict==6.5.0
+gitdb==4.0.12
+transformer-lens==2.15.4
+execnet==2.1.1
+safetensors==0.5.3
+rsa==4.9.1
+zstandard==0.23.0
+google-cloud-compute==1.31.0
+rouge_score==0.1.2
+importlib_resources==6.5.2
+cffi==1.17.1
+pyspnego==0.11.2
+aioboto3==14.3.0
+Pygments==2.19.1
+tabulate==0.9.0
+typeguard==4.4.3
+click==8.2.1
+nvidia-cuda-cupti-cu12==12.6.80
+word2number==1.1
+pycparser==2.22
+gevent==25.5.1
+transformers-stream-generator==0.0.5
+sqlitedict==2.1.0
+datasets==3.6.0
+netCDF4==1.7.2
+msgpack==1.1.1
+urllib3==2.4.0
+pillow==11.2.1
+PyNaCl==1.5.0
+parallel-ssh==2.14.0
+nltk==3.9.1
+python-dateutil==2.9.0.post0
+yarl==1.20.1
+pyparsing==3.2.3
+ml_dtypes==0.5.3
+evaluate==0.4.3
+numexpr==2.11.0
+pydantic==2.11.7
+ssh2-python==1.1.2.post1
+moviepy==2.2.1
+markdown-it-py==3.0.0
+googleapis-common-protos==1.70.0
+xxhash==3.5.0
+dill==0.3.8
+sacrebleu==2.5.1
+identify==2.6.12
+fsspec==2025.3.0
+dm-tree==0.1.9
+chex==0.1.89
+huggingface-hub==0.33.0
+certifi==2025.6.15
+imageio-ffmpeg==0.6.0
+scipy==1.15.3
+pytz==2025.2
+pytablewriter==1.2.1
+optax==0.2.5
+google-cloud-tpu==1.23.2
+nvidia-curand-cu12==10.3.7.77
+charset-normalizer==3.4.2
+wandb-workspaces==0.1.16
+s3transfer==0.11.3
+absl-py==2.3.0
+accelerate==1.7.0
+wrapt==1.17.2
+narwhals==1.43.0
+virtualenv==20.31.2
+peft==0.15.2
+triton==3.3.1
+sympy==1.14.0
+nvidia-cusolver-cu12==11.7.1.2
+contourpy==1.3.2
+nvidia-cusparselt-cu12==0.6.3
+nvidia-cufft-cu12==11.3.0.4
+ruff==0.12.0
+nvidia-nccl-cu12==2.26.2
+nvidia-cusparse-cu12==12.5.4.2
+google-cloud-secret-manager==2.24.0
+pytest==7.4.3
+google-auth==2.40.3
+beartype==0.14.1
+sentry-sdk==2.30.0
+distrax==0.1.5
+grpcio==1.73.0
+pyarrow==20.0.0
+tensorflow-probability==0.25.0
+imageio==2.37.0
+wandb==0.21.0
+propcache==0.3.2
+tensorstore==0.1.76
+numpy==2.3.1
+nvidia-cuda-nvrtc-cu12==12.6.77
+nvidia-nvjitlink-cu12==12.6.85
+setuptools==80.9.0
+tzdata==2025.2
+transformers==4.45.2
+pandas==2.3.0
+scikit-learn==1.7.0
+lm_eval==0.4.5
+botocore==1.37.3
+torch==2.7.1
+plotly==6.1.2
+ipython_pygments_lexers==1.1.1
+comm==0.2.3
+matplotlib-inline==0.1.7
+ptyprocess==0.7.0
+jupyter_core==5.8.1
+executing==2.2.0
+pexpect==4.9.0
+pure_eval==0.2.3
+asttokens==3.0.0
+stack-data==0.6.3
+traitlets==5.14.3
+jupyter_client==8.6.3
+parso==0.8.4
+ipykernel==6.30.1
+pyzmq==27.0.1
+tornado==6.5.1
+ipython==9.4.0
+prompt_toolkit==3.0.51
+debugpy==1.8.16
+jedi==0.19.2
+jaxtyping==0.3.2
+jax-cuda12-plugin==0.6.2
+jax-cuda12-pjrt==0.6.2
+jaxlib==0.6.2
+jax==0.6.2
+nvidia-cublas-cu12==12.6.4.1
+nvidia-cudnn-cu12==9.5.1.17
+treescope==0.1.9
+flax==0.11.0
+simplejson==3.20.1
+orbax-checkpoint==0.11.20
+aulis==0.1.0
+devinterp==1.3.2
+timaeus-aether==0.0.0
+timaeus-cli==0.1.0
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/wandb-metadata.json b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/wandb-metadata.json
new file mode 100644
index 0000000000000000000000000000000000000000..61040eca423ab5a985ce642a5e3f7f2354ee8e29
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/wandb-metadata.json
@@ -0,0 +1,60 @@
+{
+  "os":  "Linux-5.4.0-216-generic-x86_64-with-glibc2.39",
+  "python":  "CPython 3.11.11",
+  "startedAt":  "2025-08-07T16:04:01.892209Z",
+  "args":  [
+    "train",
+    "corner",
+    "--wall-prob",
+    "0.0",
+    "--env-size",
+    "13",
+    "--log-gifs",
+    "--env-layout",
+    "tree",
+    "--wandb-project",
+    "jaxgmg-wall-prob-tree",
+    "--wandb-name",
+    "wp:0.0_alpha:0",
+    "--num-total-env-steps",
+    "200_000_000",
+    "--prob-shift",
+    "0",
+    "--keep-all-checkpoints",
+    "--num-cycles-per-checkpoint",
+    "64"
+  ],
+  "program":  "-m jaxgmg.__main__",
+  "git":  {
+    "remote":  "git@github.com:timaeus-research/timaeus.git",
+    "commit":  "7415666d6a2e5d63453ea9fadaf6cecf63256a59"
+  },
+  "email":  "david@timaeus.co",
+  "root":  "/workspace/timaeus/projects/rl",
+  "host":  "6e66e15ca303",
+  "executable":  "/workspace/timaeus/.venv/bin/python",
+  "cpu_count":  24,
+  "cpu_count_logical":  48,
+  "gpu":  "NVIDIA GeForce RTX 3090",
+  "gpu_count":  1,
+  "disk":  {
+    "/":  {
+      "total":  "102005473280",
+      "used":  "43219062784"
+    }
+  },
+  "memory":  {
+    "total":  "135066595328"
+  },
+  "gpu_nvidia":  [
+    {
+      "name":  "NVIDIA GeForce RTX 3090",
+      "memoryTotal":  "25769803776",
+      "cudaCores":  10496,
+      "architecture":  "Ampere",
+      "uuid":  "GPU-43bd7524-bf94-73a9-1460-2b36dd09a208"
+    }
+  ],
+  "cudaVersion":  "12.8",
+  "writerId":  "mdhm7ax1vj6umwzz7dpb0eix0yt41u7o"
+}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/wandb-summary.json b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/wandb-summary.json
new file mode 100644
index 0000000000000000000000000000000000000000..082eb91a890d2b27bbb45b00215c701c30a8d331
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/files/wandb-summary.json
@@ -0,0 +1 @@
+{"train-all/proxy_corner/avg_reward_per_step":0.144287109375,"eval-batch-tree/avg_avg_return":0.13043764233589172,"ued/scoring/avg_scores":0.1403074562549591,"ppo/avg_actor_clipfrac":0.009179688058793545,"ppo/max/max_critic_loss":2.379496618232224e-05,"ppo/std/std_actor_approxkl3":9.924831101670861e-05,"ppo/avg_actor_loss":-0.00015995264402590692,"ppo/std/std_actor_clipfrac":0.004698612727224827,"eval-batch-tree/avg_benchmark_return":0.9840756058692932,"eval-batch-orig/proxy_corner/avg_reward_per_step":0.13348388671875,"train-generate/avg_reward_per_step":0.144287109375,"ppo/std/std_actor_approxkl1":0.00036725227255374193,"step/env-step-all-before":199229440,"perf/eval-batch-orig-duration":0.5346216009929776,"ued/distances/cheese-corne_dist_wavg":0,"eval-batch-tree/proxy_corner/avg_avg_return":0.13043764233589172,"ued/layout/num_walls_wavg":0,"eval-batch-shift/avg_avg_episode_length":80.84910583496094,"ppo/avg_actor_approxkl3":0.00022391097445506603,"perf/eval-batch-shift-duration":0.48660353664308786,"eval-batch-orig/avg_benchmark_regret":0.015790626406669617,"eval-batch-orig/avg_avg_episode_length":10.042457580566406,"ppo/std/std_critic_loss":2.3786074052623007e-06,"train-all/avg_avg_episode_length":10.002439498901367,"train-generate/avg_avg_return":0.9273442029953003,"ppo/avg_advantage":-6.539156311191618e-05,"ppo/std/std_loss":0.00032132078194990754,"ppo/max/max_entropy":0.04269036650657654,"eval-batch-shift/avg_reward_per_step":0.01316070556640625,"ued/layout/levels16_img":{"path":"media/images/ued/layout/levels16_img_6048_239d7b602cc1b202d152.png","format":"png","width":55,"height":55,"_type":"image-file","sha256":"239d7b602cc1b202d1527c3851c0a0e8915de4b43e7a48c356b2e03d4bc6141e","size":285},"train-generate/avg_avg_episode_length":10.002439498901367,"ppo/std/std_grad_norm_pre_clip":0.12230227887630463,"perf/eval-batch-tree-duration":0.48448570910841227,"train-generate/rollouts_gif":{"_type":"video-file","sha256":"7bef146b11b7b5f9c1e2c6a9d3f15d91c383d6a9ddd53db1bb87427b42cded21","size":258805,"path":"media/videos/train-generate/rollouts_gif_4064_7bef146b11b7b5f9c1e2.gif","width":225,"height":225},"ppo/std/std_advantage":0.005673132836818695,"ppo/max/max_actor_loss":0.000281001441180706,"eval-batch-orig/avg_reward_per_step":0.13348388671875,"eval-batch-shift/proxy_corner/avg_avg_return":0.8827041387557983,"step/env-step-replay-before":100728832,"eval-batch-tree/proxy_corner/avg_reward_per_step":0.02870941162109375,"train-generate/proxy_corner/avg_avg_return":0.9273442029953003,"train-all/shift_proportion":0,"train-replay/avg_avg_episode_length":19.69230842590332,"perf/ppo_updates_per_second":42.2462667048863,"eval-batch-shift/avg_avg_return":0.0643315538764,"eval-batch-tree/avg_reward_per_step":0.02870941162109375,"step/ppo-update-after":121620,"eval-batch-tree/avg_benchmark_regret":0.8536379337310791,"eval-batch-tree/rollouts_gif":{"path":"media/videos/eval-batch-tree/rollouts_gif_5088_e00a0d045117b56c0b6c.gif","width":225,"height":225,"_type":"video-file","sha256":"e00a0d045117b56c0b6c0488ce5dd5a1af8081ef6978cfed7dc928415c8250ca","size":126007},"eval-batch-shift/avg_benchmark_regret":0.9284399747848511,"train-generate/shift_proportion":0,"ued/distances/solvable_avg":1,"train-replay/rollouts_gif":{"path":"media/videos/train-replay/rollouts_gif_5088_016bb4c85f0bf1211686.gif","width":225,"height":225,"_type":"video-file","sha256":"016bb4c85f0bf12116867c026ddb2387bcfe4c166e50d260da185e8024561cd3","size":110715},"ppo/avg_actor_approxkl1":0.0004152771725784987,"train-all/rollouts_gif":{"size":110715,"path":"media/videos/train-all/rollouts_gif_5088_016bb4c85f0bf1211686.gif","width":225,"height":225,"_type":"video-file","sha256":"016bb4c85f0bf12116867c026ddb2387bcfe4c166e50d260da185e8024561cd3"},"eval-batch-shift/proxy_corner/avg_reward_per_step":0.00733184814453125,"ppo/std/std_actor_loss":0.00032137721427716315,"ppo/avg_critic_loss":1.750244882714469e-05,"train-generate/proxy_corner/avg_reward_per_step":0.144287109375,"ppo/std/std_entropy":0.0037572479341179132,"_runtime":3426.940434357,"train-replay/avg_avg_return":0.8418449759483337,"ppo/avg_loss":-0.0001889849518192932,"ued/distances/mouse-cheese_dist_solvable_avg":19,"ued/layout/prop_walls_wavg":0,"train-replay/proxy_corner/avg_reward_per_step":0.046875,"ued/distances/mouse-cheese_dist_finite_avg":19,"ued/visit_patterns/num_replay_batches":3074,"ued/visit_patterns/avg_last_visit_time":3064.295166015625,"ppo/max/max_actor_clipfrac":0.018310546875,"ued/eta":0.5,"ued/layout/prop_walls_avg":0,"train-all/avg_avg_return":0.9273442029953003,"ppo/avg_critic_clipfrac":0,"ppo/max/max_advantage":0.03855454921722412,"ued/layout/num_walls_avg":0,"ppo/max/max_actor_approxkl3":0.00042568164644762874,"ued/distances/cheese-corne_dist_avg":0,"perf/env_steps_per_second":297601.5046583777,"step/env-step-generate-after":98533376,"step/env-step-used-after":199262208,"ppo/std/std_critic_clipfrac":0,"step/env-step-generate-before":98500608,"train-all/proxy_corner/avg_avg_return":0.9273442029953003,"train-replay/shift_proportion":0,"ued/distances/mouse-cheese_dist_solvable_wavg":19,"_timestamp":1.7545860587705874e+09,"step/ppo-update-before":121600,"eval-batch-orig/rollouts_gif":{"path":"media/videos/eval-batch-orig/rollouts_gif_5088_1a999dd354e9cacb552d.gif","width":225,"height":225,"_type":"video-file","sha256":"1a999dd354e9cacb552db7679fd97f0043f25b846aef9aa17212f1020212d7d8","size":253901},"ppo/max/max_critic_clipfrac":0,"ued/distances/solvable_num":4096,"ued/visit_patterns/avg_first_visit_time":738.998046875,"eval-batch-orig/avg_benchmark_return":0.9896295070648193,"step/env-step-replay-after":100728832,"train-all/avg_reward_per_step":0.144287109375,"eval-batch-orig/avg_avg_return":0.9738388657569885,"ued/distances/levels_cheese_is_in_corner_avg":1,"ued/distances/solvable_wavg":1,"step/env-step-used-before":199229440,"ppo/avg_entropy":0.03778355196118355,"ppo/max/max_loss":0.0002508320612832904,"_wandb":{"runtime":3426},"eval-batch-orig/proxy_corner/avg_avg_return":0.9738388657569885,"_step":6080,"ued/distances/mouse-cheese_dist_finite_wavg":19,"train-replay/proxy_corner/avg_avg_return":0.8418449759483337,"eval-batch-shift/avg_benchmark_return":0.9927715063095093,"ppo/avg_grad_norm_pre_clip":0.34002166986465454,"eval-batch-tree/avg_avg_episode_length":76.809326171875,"train-replay/avg_reward_per_step":0.046875,"ppo/max/max_actor_approxkl1":0.0010867544915527105,"ppo/max/max_grad_norm_pre_clip":0.6099137663841248,"eval-batch-shift/rollouts_gif":{"path":"media/videos/eval-batch-shift/rollouts_gif_5088_d9e7cbb78f0db799d05e.gif","width":225,"height":225,"_type":"video-file","sha256":"d9e7cbb78f0db799d05e04f884663f887c14adc2b7d1388f1582df44adfb465d","size":50950},"step/env-step-all-after":199262208}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/logs/debug-core.log b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/logs/debug-core.log
new file mode 100644
index 0000000000000000000000000000000000000000..0c4d3ab0b045cfc24f09c2d46c9b3cfd5b1ea4ef
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/logs/debug-core.log
@@ -0,0 +1,16 @@
+{"time":"2025-08-07T16:04:01.969725732Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmpu7t58h_e/port-3825414.txt","pid":3825414,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
+{"time":"2025-08-07T16:04:01.9707077Z","level":"INFO","msg":"server: will exit if parent process dies","ppid":3825414}
+{"time":"2025-08-07T16:04:01.970700498Z","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-3825414-3825712-2832449409/socket","Net":"unix"}}
+{"time":"2025-08-07T16:04:02.12731558Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
+{"time":"2025-08-07T16:04:02.138681123Z","level":"INFO","msg":"handleInformInit: received","streamId":"wp_0.0_alpha_0","id":"1(@)"}
+{"time":"2025-08-07T16:04:02.475536787Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"wp_0.0_alpha_0","id":"1(@)"}
+{"time":"2025-08-07T17:01:12.543912403Z","level":"INFO","msg":"handleInformFinish: finish message received","streamId":"wp_0.0_alpha_0","id":"1(@)"}
+{"time":"2025-08-07T17:01:12.552013205Z","level":"INFO","msg":"handleInformFinish: stream closed","streamId":"wp_0.0_alpha_0","id":"1(@)"}
+{"time":"2025-08-07T17:01:13.522543473Z","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"1(@)"}
+{"time":"2025-08-07T17:01:13.522621865Z","level":"INFO","msg":"handleInformTeardown: server shutdown complete","id":"1(@)"}
+{"time":"2025-08-07T17:01:13.522643008Z","level":"INFO","msg":"server is shutting down"}
+{"time":"2025-08-07T17:01:13.522654005Z","level":"INFO","msg":"connection: closing","id":"1(@)"}
+{"time":"2025-08-07T17:01:13.522812753Z","level":"INFO","msg":"connection: closed successfully","id":"1(@)"}
+{"time":"2025-08-07T17:01:13.522879621Z","level":"INFO","msg":"connection: ManageConnectionData: connection closed","id":"1(@)"}
+{"time":"2025-08-07T17:01:13.522780387Z","level":"INFO","msg":"server: listener closed","addr":{"Name":"/tmp/wandb-3825414-3825712-2832449409/socket","Net":"unix"}}
+{"time":"2025-08-07T17:01:13.522911232Z","level":"INFO","msg":"server is closed"}
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/logs/debug-internal.log b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/logs/debug-internal.log
new file mode 100644
index 0000000000000000000000000000000000000000..5ef8cb7b05a3f6f054410ba9f9ef72086436aa22
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/logs/debug-internal.log
@@ -0,0 +1,1167 @@
+{"time":"2025-08-07T16:04:02.13888151Z","level":"INFO","msg":"stream: starting","core version":"0.21.0"}
+{"time":"2025-08-07T16:04:02.475436104Z","level":"INFO","msg":"stream: created new stream","id":"wp_0.0_alpha_0"}
+{"time":"2025-08-07T16:04:02.475521241Z","level":"INFO","msg":"stream: started","id":"wp_0.0_alpha_0"}
+{"time":"2025-08-07T16:04:02.475561115Z","level":"INFO","msg":"sender: started","stream_id":"wp_0.0_alpha_0"}
+{"time":"2025-08-07T16:04:02.47555003Z","level":"INFO","msg":"writer: Do: started","stream_id":"wp_0.0_alpha_0"}
+{"time":"2025-08-07T16:04:02.475577013Z","level":"INFO","msg":"handler: started","stream_id":"wp_0.0_alpha_0"}
+{"time":"2025-08-07T17:01:10.135358536Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/d, Name: checkpoints/640/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/640/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=25acd51b2891ec5e444026dd28fc02a5fbda0395668a47b4ad8fbef7555412e321fa0b64cd2a8e358bc477485b95b68d9992d6c42a39a04efb028dd19615cc6d78f31c4e608a3575711a876580585a2984c946d2e78cc36bc4220ecff578b51435431033f0e744d6315d0ef05e738989ee8aa14b3cd69b237fda973adf5955fc64d3b836b17b4c2ef461f0192812de779969c4e8be0f8e3b7e632a22242414bce0976545bdf2e7dcccf0bf4e593f32b6130b974b48afac40a5e1c52fdf5e33e31e7f4d4084d17c297c074b05d71768233ebf7c532594cfc45872227954ca52774dff27b13ee7bb40add731b75b5014aef25d715a62d5a9fdb18cbad78ff0955f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.1354329Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d, Name: checkpoints/2560/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2560/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b3d7eac0906ed869a0a310d3406b523561ba8da26ef40f3b5a8fe51900a923c6af7b1f0321f8d3d7408e8edc2fcef0cea1b0be1857949b3318d45a9f7831e0eb7b9951b939a278cad9cddb148838817114d061a3194fb21a551c4f9f2914086ad60f47ad46af60ca1f35f8f0c9308d6e3d01d7c5491a52ef6cfdaec5d0c0ee0428dac95d7049e2898e29c726aa6fe35352ad29ef64e8cc1b680a731031287f25471236e25c692b943cf99e0d0cfc14d24b4498974c305defae5200a8b6a13b2eddab6aa5b50a9958b6053f45e8e34d892ca7f00facfa82439edfbf51a8da4f96ffe322b394317143fa246e1d66836222178a6c6ab457e4ef800b8b8646f95af3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.135492146Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128, Name: checkpoints/128, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/128?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=1a8edf7811c445419ba25ba2e11280217814c29774795b7b9fff1ab78bba9d1e133b84da03af2a35d3120110fad1790377bd156c90528a07ee8e295ebaaf1b711bd6d717980b1d9e3e7580b38a190188de66e7d38c7c22e31ea7d3ae15d63405653ce06b8e09cc99bd9cd036c172f5dd575e9a9adf4ef61a55b4f826327d57cefaacb50c5c15e32882cbfb2d185b25e15184a9d9f574f0cb1cbedc59ef14686628f767ac3cb9b8ab0e0dca8a5c4c35b58c7bbbd670c7d2a71f679ee58fd63b31bbd381172960d35887b92e8fa3bf6203c07fb77da77d3d9fd2b9cf6723daca9286d7fdc86c380333bc750579944b2e97abaf9425467afbf6f230641d215cf9b0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.1355124Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default, Name: checkpoints/1280/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1280/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=9b292d28465d92ae55d802c31821031ad883674947fc64e46d53e910047f9aa8d16b72dbf256abf61d1fa6d8b4d2e310360bddadbe24e06ffd072d546065414cd07c5e70267bb70e966568ca2d50eda1ef884c75596915ee5d12feb77d9590dd25c0ef3631bdc5263d12068806b270f0aa1de3fc719c141c5443bc538292d418dc66f17945b99792656975ff3e6604639b9992e2bbe39293abc1422bc1ef6c24eebd9fd8b3ae34aa0ce8ca58f829a93b28ae08e2a30f00808ad38ca1128cebbf6ed34e93adf484be9bf19464c108d8efbc6c4c2470de87d4ef967d9a0cad6090dbd953d7e30e6eadf02f4e6a47e5874d8414e797116da2a5a3908df56fc6c55c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.135550216Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d, Name: checkpoints/1280/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1280/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=8182eb3a2f71dfed0eb0cbdbfafe211342dd8c8a94408eac4e3004834309db13d39fb71bcaf2767a39a3cf29862944439c731dec56efcb43ba78fbf21c5aa8cbb46a9892dd520d8012dc4b3facce6032561769b91c28c53f36c62da0d5a8b459de36ae9e4d28997d7ab0641344efbb3457a5cfe1693af06adc5771d9b005d9c7b6ab75fdca019bb9532c1a18080e7b0607e0d62204454dfc366ec48875aea794d1d8afc5b5ebe38e1de878b074adf864e84a3e8db5e768a0a418cfb6faac2d9b1f2733950787907c4413bf492e4d01604582ca48f82888cad64ab84cbb1053f6aa398b17327f74ead9f087dd50784e786ad1a7f1a1eff8a717485b2c1b5b7bd9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.135574812Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default, Name: checkpoints/1216/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1216/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=3e92fe152e450e9c48d79590d88cf1301a5d7b09017c198bdbe444933a09c60b92262fef21ae37a7af54028eb48f4094e6a7563ff9d94eae0d6a44a3ce0118f45e8298fff79517c1def53b47daaaf8311989e05115d44b7f0bb3ff47c208724d6babf54283080585a88d3114723e1f94d64202d4dcc504c37771b04f75fd307cdf9cffe5564fb88be23d2006bd91716fcf58a37f38ddf6d1d5f518fbe12627668c46c415a184ef4e06d6d05b04770d0c067e7cc7e0c5b0f4e78d8bfd3f7cca9426609c8bdb4b6f23e86b906db54e53a894510a0de482245647ad7dc52af8b804c467ebf142fa733af6d7fcff3de4cbd31e3fe8f991da2d97c4e97ba5089d6a04&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.135611085Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/array_metadatas, Name: checkpoints/256/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/256/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=0ec1faf5fa37a4a846ef37f41af8dd653e68a125c22981a92ff365229ee5f3f4fbd13a1f66401550f80494234ddc47918dc5e17b0fe10f79178d8b0532483cf6fb41bcf525c5d6e01e2f86b45a086ef32e463632d175c5a76ea245a707b74533810f06508cbccb914091875e9bac61fc1171b3c4d89b30848c0eac0ca36eb87c3ad6db232f1a1e132eb7901b45b7200da84331c2eb05c43f800bccd526f1d526a0b6d7d8b4802e21bd843ed88b01b2ee9088701fe9a2fe42d11af00ad3446f99da598ce8f4cd83cdc17d19ea15fb6c1e061fc479db99eda69ba8fbc63ec1020617fe43c70f41ff4125fc9105ae0f52cca65f95945a2bc9c72af40440cd9fbdd1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.135650022Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/array_metadatas, Name: checkpoints/2944/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2944/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=a2d8fea7f97840e5d2577e71d93c7458760bd8f84c53e4ef21a8b8d61fecbc27e58fd3971cd1f281fc2e58c0d7434254c903c2a3b87d3add6119e1e8cd3cbc7c4b4e12dd2c9a37bbc905a38fb112162fd61c0afc819effcdad64860c2a7289aa6e146c048e0cee5303f7ca4e39f404d9c43a3a98c137baaa46787eaa52bfd40d8ee4e4c61bd63124bc6860b0b6dbfbd5d2450bd57f574b903a83ca2fcf0d947f33542b6fd758f50278f58ee857604ae7bd1b7d2d7bb4a84a78efdf45ada43edee708ea7fad9a40b6bc9ad9e75d2f5efc85968a4d3bd8aacbeb4ca202fce6804cdd6d215454d000066ab3aa2398d729f81995d06508bb6b405acef6569334d2ee&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.135687765Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default, Name: checkpoints/896/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/896/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=45d4cf7b39194f3dde6d248c8588c4731727036de95c8b357c2a01126ef784c63e59d3c0fcabbda3d1b0cdbf7ab1277334a527792dcd980075af2f238020624209e15740af908d6d8f05938426a3db4c98f7ee1b9cf3d71cbaf8c4da7ce471c66bb8ac45ce2d647e056e436a075d1e7c62e4ff860ae755eca8a96d1f932afe621e8e2a3cfe425084dca8d1c7201fe3bffbcc0ef861895965d8077f0e7141949b96655db202ccc5569a86475eb3c990a63b07d5ce34e10ce6ef07873b76add80f3b0c2129d16e49cc1c6399b9da5d85b812c3bb608e727e1c5bbc2c391e2e4f529f01317bdb6d33428c987d5c930b624baef4e7f9f3516de34a3c235758f57d84&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.135723561Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default, Name: checkpoints/1920/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1920/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=1c498e6453054c7079345e74af7f12438cc501fe56a62f7a5de8ccd6e893709df736627030898c8dfdb7e7aadc1712ef7204deba516591590eddbbef9c4aeb8b31e25e48e20f08c3e79007ec736fab64ea03ea9a19227756236c15931378b2dae5ddd0ac24ca0dbae929f4a5f46951add46fb0deb76d76a7f1d4289b54daa62a8c1db1b3bbdaef853ef63f4b2b63af1f60f0073746f472c63337d156669aea2073465d06f930e12ac305e6d3355229abcdcd6a78d5560218a39901a94f7dbed739a17456ffc5dd7dfeb4ef1db27a8126d2aa688acb343eb2021648768fbcb8a3cf30eed843d829524e4fc287288968b2ee6693608f9819037c453e31a346ee5b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.135776178Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640, Name: checkpoints/640, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/640?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=87da9a771f802c473585d95f3ae1d3da4723e5e54654c4abc02156a21d0c8c4cf93da6c9ace780e6c59b87fa8c2c03237d9c5ce573e8da9b8783e1b2fa722e3bfa74f147ec198689403d9c4888dd192c72bd8723e72c148df253131ab28112b9ce9675d38ba4c2e358f617f1c7819ecf05d0b77aae72ebcd934edf4932f300c583e34370950facc04bff7743795bc5e66278d8ed723ff0d8f7eef7ed878cfeffed0f0cdaba3d06c8d93f9a31875889c559f98129f73a0999317345ea7a3a7114bf4bdc58b4208440f554fb7c053cb10b72dd8f36ca0cc732167fe7eb52093949eb59b9572662cdb86d9a4b10f7ca9a47d9d7d2efb289e210ae868456f81b3f7f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.135396917Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0, Name: checkpoints/5504/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5504/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4683c82c60a71ee6faf714bf3c2d2dacd36d3b2a4eeb46d268764ca7c18087580338b70fdf62b9dcd77f1e335082435f4d46c5c156d9d565dfe905d05f1744d75f47ae0f83cf38c69ae66feedf0cccd1e069ee1fc12183b24702d7c65836b8c9a9264f9e74034ace0a0849af25a8e0c602bf79afe2fcd09d995a81eca23761e29dd18c5f44c71dfe2d962596d4ec3ad9ec2035d604695190a6007706dce75020b7b5477b49809c39adb89783bb6b09b7b311396724b644c5736664155f40b39de26b3b5886aa653147ebe8400ef565e50cac7aba0d97bdf79ee048c6e6b121716dc19dda62477f282e5364ac481f7729a0e72e22e908cc3005134767e7def89f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.135356402Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d, Name: checkpoints/3584/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3584/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=47bcee9617e161122bbe7cb789d34f6e65b4b5b1ae4fa00d3192b25cccc607f20df0f78ebd0b9dfbbf424a0214b7c9c67a6cb99fc17dcad6812e8250a62cade3d08724f5b22a8dd4baf3411beec0b9af1a9fbfe8afe489c2e48aa1c5823eb8c7f3c14174ed14e401b0430dcf1bc04cd64c741427077a536dbcc04782a9b1ad66648ee4f506e9306758ee0c01ed8403b267c9d1c9db5065ebefb5bc0adf0c4ed7c53b66c5cbf38b6fcbbd0a82cde8c780f7c9d5146eccd89427046a2b303ad6d70bdfa5f56f09165e76fd6eab451552b54c569f258758ca9c311438a95f5fced1c38bd3fd621e150c6371a8fdb595df37e60b9366a7d501c9649d8e4503a66f95&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.13551167Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0, Name: checkpoints/3456/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3456/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=477fd55250fcf9df87aa186f428e7e6d879240e01fc1623bb66c1bc27fdf27e72cf285f2acafe55020e364922c6ace186a7018fb010aa6f45be6dbc215329704b2f7f034266e5f2b7a1421fe61bdb0f897f86f6a096ef469201f68ddf74399ce838932cda9bae5d6147813cafe8a91649356cf57e47ec72ddebd6b86f548d6aba0db89d42a3720c4cf0cd846f0fb6f5dce5402427437570d56c84baa270569da23cc2b61620438c97c09157d0342bdf65f99024e6859207b3fc1e723552c490a012de0b276acd1b6bcf9c1b9c8ce316882fd488468abbd1fa55b8766ee6e024ae9b98e2e82e44908b0d1f851ebc08d62d4336590d2a2339ca0964afbbcb931f3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.135584154Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/array_metadatas, Name: checkpoints/3328/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3328/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=30dffca7170a7b489d0537cd48ffe2811ec6375bad644486596a4a54447d3291bb6ad1e351972357f41a8cd502091153a49ff0ca1df6171b226d835f270e346bdaa1617f113cb5026a9affc5212baf8eeee2be034d9fd304fa87cc7376a2a728e236b48ea71abd8125d679142dfd5056b4fd8f5ab4b202fc507f93dca5d1557a41deea6ba19fb93f0f04a31c71f6fbab8643476fea894996701ddf0590198ad432b640154fe0c738b6f2f38f2b284170a3a6a9af6187439cf975ed154eeb37dde5bea5bd323214c74811e79155ce78a3987ac75cebfc6c59e81cc6c8c07f25f2aea147f8429c9365a5c5ffb8f01cb838e518ed9679925536b3a89078ea6fce93&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.135657219Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0, Name: checkpoints/2560/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2560/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=995e5286fdc9b2964597706f68c7b1955502e5db19af67dce6c1a1a17e7cdc6192e336d0e6af0454549b1cd108f2d9a8123edd4c9f22360260a779b58bd93fbd9e071c9096c662ce4df5b1cf31e1fafbd8717d19f76c919b4f600b6d1c28b44125a0a683bac7a6891064de45286281cce0c70c9bc3123d8feb78ccb9601d25a53dbaf0f6390913a2aed5b6c62f586cec36e5efd25d6abae3e027a97f14a227b4de4d0495bcd20e0cb0363aef5ac72512c2253d7b2af2ff4cbfb426bf646d0f7ccbaf1680286f8c567b248fde9c8495734ec20441e17372a3ca6fa19e2a1c3d45fa77cebecb58e36fb8572be3116ef648e93f0ede913d2c722943287b950180d7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.136474153Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default, Name: checkpoints/2752/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2752/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=c3a35081db094c7c989cd57b83049d072247aed16c60c8cb16f766d3e70033bec3a58c8425f5a87f898ff508011a301678b6f526e1216344c386eb588900cea1750b174e0f02543e9055225aadb3c8fc2cacc2ed66d3734a5b001b84c4c6706b63c1175e413250d4faee0371a5be4205ea39741f0cfc11408a3a32f148bc98ce7301130c490cec2eaff51d59daadcd41dd1529c7b3ff292d9028c28097cdd7b3a8626cd4a2a92e0403946bc7981daccf5da325b3a7f06aca251103744aa7cefbc6a309b525711e9da0ff671fe51be0d48681b62893104866e01cc06dc7bb26bbeac680d43dc7428b7a126b45d1d11c2e09d470215b373549b9e81b63621277f3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.136548723Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0, Name: checkpoints/2176/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2176/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=6ccf536570cd998cfdb0fd57a123232922421acc2e8be632c105adbf56e096feb9a55cbc72366dafed44a1fbba4dd6577b523fcfe925e7e9a531eb88d80bb4dcd61fac17f447bf6e9ea2430994bb527faea89a83b4695d9cbf945c2a32e28006ba17f5464c2537f179a5a9efd9db54c2748cb695b1f201bf5116ffb607cee4f7ba61f8d4a8a3be20a53ce90fa2d756ce3f1d064deec8519bc07d610977476eb37e6bbc70c0a521f46f6a6995b3a5443984c6abae5772bb03cf9819b94c94a12e0eba8b21731ff8b83ef9b2d366d0e543613995eeedb75046d6a491f9fb5a67386e86bef6e3d9d2fdecf32373d5a8ba0fe076fa5b3239b58f20d85686b6dcc887&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.136606244Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d, Name: checkpoints/1984/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1984/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=baa05edb07c40d507f0833c17230a77afe748f9b4613c1ca5e5e87d824602787dcb21761421f6e7177c6e031e7d459e95d9177872d1abcdbb79f9e05507f8ed5b0833cae0cace14f5b7c78dc1fa8d035d30a03be19d283d253db3803ba2e0d411110874e4e09df2040a7f7e165f56c7b955b02317c0275f7422e65103696f655574cec6113e097ec018bbd33a30a8e0d6e4de63162344898e823dadc9e39f10e195ba8b6158df6d526639dfca9a2981567c05c43a72a7d621d34485730dc020eb88e3978413d8691b042c8285fd06c3c92693ea7849cbfc44fa24c79cc2c9287766af50b8748a6e3ec7b75bfd592dc46339cfd2691498df84362ad47af15cb77&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.135406498Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728, Name: checkpoints/1728, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1728?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=933484b570e928ef37fbfdb36affd2379293fdcd4c84e939fb405223e7e226391ba92a74593d46a117f395d026d33fb86ce7326b84886ef45f62a494d195e0ebc51d2b553b3398b1298f8b79d04f818c827dd33bdeb8bec4ead5afe09cda5005a2cd08267003f8c3d2d07c0a69737b3e3ee44403c04e65d0f7cfa928cbad4adac10de75058737c46f15dc0868e524f6a142823e93651c7e80a22a9ea4db5a563cff6c5cc850ab6eb4c788a15902b3029f53025ebeb4fbea64c27927c2001162111c7b3e371d22620c79bbfc1ede5ea9fecdde10939931e7e579fdf1e85408a13ad624d24b78a43beeafb90d98a39126fbcb6a69f7d0d79872e7d69b41001374a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.136623596Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0, Name: checkpoints/5760/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5760/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=13bd8193d06088512fa95f71a56741e69ef83b8f9e2676be96434963fa6c6be522b8e412df1d34793392772e0e4b9944059006f130dc4c643d59c49e0c84110fec656c5a9b145eac286e423c68aa01c8bda2374209a73b76da55fdc2581617a0b1a1016a4e46037e8b5d752957e58c0c4152e58d9093ab4099ea2687606912c1d37a7325a55de5df1e12f4b6966cf0bb17f567dc8eb09dee5992ef02a432bbaec7c95f8ac42cc54aa3608f54499980e2ee6b8816017ccf70c6e4d17e843dcf0e9c8f1e8029fd2c91fb755ac5bd79f723b07d0956637fc6d333461e26a3265bb5babc84c12dced5beadeda9c29423075c34857aa4d098421da34f7bde63718e1b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.136694487Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default, Name: checkpoints/2560/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2560/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=69d6eed96f91172fd3cfc592ca5f6332c03fbd175b8690b12193e150f8b3b36c9c3ac2a6d1f2f9a36aca083ea62cd5e0550cede996cdc0580a7b281fe69bcfcd61f4676a8d6804cb551284acf2d976b4fc7dbacd97e9c33c16613348b484d2862ade258ce894b67d60ffef95e554ff4bd559572f8b73263a3e70c4e2b42a1a876ababead76d27be179fbd4d1fde4f1f1a95e09d30dbc5c2d33005fe06d49ef8bfe96aee81340e5cda94a74087d4fdfd32bcb9cf0361ab09f645815358b951b7b2c8e687cada5c2ff03429aedc3f248ccc18c322bc3abf5085e735afb549b03d4602a8e804547636cb49c51f1e80c9462e7d9662b43aef2a1abceaa23cfd71e01&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.136774288Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/array_metadatas, Name: checkpoints/4288/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4288/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=17877bde5691291fff3a9f90436f1ed6eb6dcc1547b46bafac937434a1e4db4f05c4cb342778b0aa5080ea21b82f3d3947c45c9bd054b0762dc15bd32a5d9ad8b4838bb0bfb091ec856478d3b9bc5435fa4486da1a74d2094b17a07f13cb34dd5f2e0ee2d2fba837fab671874d9e1a637df4348afd67a41d9cb67cdd1b6ede1f6d1443138ad4262f7edf048a173fd6ee0f9877a354245f20328b8aaa80cce201cab2915d736a9d947fc513133b494b8aaee3810ed3eca8ee42562eaba527c5618117fda64bf1cdd0f4cdb1b645b0f74dcec44fc92898cf4bd373a846d3a7b7d03215ea3d9354c36d3ad926b9bb309adf6488ebd061b065665606caaee9d8c5ae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.136934199Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/array_metadatas, Name: checkpoints/448/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/448/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=55c095e378f4587e29ca59883745912eabf5e7eb13caa7cc83fca356f95285eaa243c9cc8bfdb7f674b8ecd5c2a8e9f6a0794e41a4424844077287c93cce7fa984e21565378914f0ebecd777f07461aa6dc2906ff73d0bb03817285804197125ef759e0e1c9908e5044aaa5d67f6b3ef60c2cd2c292007622c387b08f86ad1ea8e884eb1adc149e48617b40aa9f3c95acdbc35f406b265a56b9a8d9c5407a3d3e4e354d3e159d57d4c6888849e5b2c0a1523d6d08a3ec2300320a4aed16147299cb2a59326e50e247931690dc3679179bd9201b14c47e7f713c44f02d951ffa573e41051797017770d509e15eb7baccdd5fd778692eb7281ea3053d90f08fdf6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.135583182Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/array_metadatas, Name: checkpoints/320/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/320/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=19b694cdd8f092cbf53e4cec4ca8a9f97c790b29b61b4a235c8b8244e2b6db71fe922aab6d40022f5d876e079236d1085b258c78c646daddaec3c34baea56e6eefc256698bfd81b6eae08d2113e753d65e54fcfe1c63cfaed59f31a233426b84d2fbd62d1e2d35f956efeb1249b38a27ed5b63c7d3d8f97aa04105e97c6e9a88f7acefc9ab8314f81cb7d133fce05b2816d4e94c2b7fef5f13d9e527a58a30b41f89532b229a76baa0778ecc7e88ff01ec5082e48e1ca5e30ce896d8be7768a0bfdbd495f17c49548e5476fdc4110d109772ecabd2b795e05f5e22e676c4cfe2354fa42f38d9f79f0e3cead5d8d55aa3b49f1f828dc1f47c6271ffc12cb8a4e4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.135687171Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/array_metadatas, Name: checkpoints/4992/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4992/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=15aab00de814de5ab2dd41bf8799926c0bddb200fddfd4d29f3f118ac24f3ae6d85d921df35cebe9cca7b2dafaf3ae8541e0d3e180550e4805803f8bd3236bdde2000fc72b505ad4b819d9c253303f3d6f8109b9d120b63b0557a977d0e174f05d40b27df65945a63f649041d61c83ab474b866d249eadd22e0a3c0f1fa284ae560e99d11fe6fad88ff2c6e52c3f851db413cc996e333e0736f8f8243cfc7523cce65721a9a77b3d6a52bdfefa9b61e6a98b22f586bcb7807e0f4c77695d149637b9c889d3597c769b06331d2605448a18834f0241d019f7489bc0d2fcba2ae2e9a92a665598b8af75b1baae09b17ce6930d7bb0c2389bf979b39d087e5844a9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.135599012Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0, Name: checkpoints/3584/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3584/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=83f878da55f3acb23e8515792dd7364e91831d7fe968067e8b2145ebdcb4cc7edf728f7d650d612700c023a833ba53e0df34dcfe8ff40e24324f28e76b28b5e2910d4016194f5a8703b241fa30534bc4effb13a10d5a15046450f6ae7a6bb69a1ef94a07b1a2e4b1594fcccfbb9e1f850c71b923e531772d0ef4faaf0591a9f407d734ec4d90dce751a99b7cbb0c3c3fcebf9d440712b674ddef6108f465505b3ab45f9bb9864a42763582d47a86ee474525877a0f4e9e032f5f06bc98d0c3ca56dca2cefdba944af15189d580fb2ed28aa78822e1ec1c155813e0a9d227fab23c781449fa4555b5cac768bbcb5edbaed6b35b5f16c21155de8621fec04cc5dc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.137156018Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/array_metadatas, Name: checkpoints/5888/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5888/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=36f056c91735abdfb28b147302c72f3b96c72b8beee4126b8cd09b8ac7aa27a585bebac7208cea70b8d6cb786dc1f28c459d9e06b0586dc5195d9692673ddb575c685aff2a50f295f3a7826e554c8203f7c813a608fae5f5277b6465c9ed6c1025790f15a3a38a7b942c95bae947d5fb186fd5a08a1bbd7c77f6bcf6275aa679d0bffbbfc3ae0f5cbce0c84dabb772a590bec7b8c3b716bb99069a3d9fb545f7ab4205f0dbd6619a9d0f595a35bb5578a9f9c707731bf6cebcf295b0f8c9f0cdae12b92488d15b12630c1c95c5576aba600f3f2fc01347d6a77eb6bac65520886ca4ce3eb7ed463bcb40abf5ea2cf6a12cf1e786b62ced332761ed6d08bb1ddb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.137336756Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d, Name: checkpoints/4736/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4736/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5dd19e4529d7cd1c8d2f215a181b9f84945e67405f7f1fbca1c60c00e5588cca818b5a27c2e11bcc4f1153fc80e72a359672272fa40c29083187d69cde3a16ec5d390ced58e05a3683de0560c7abff0ea2ed44ef8eef20de5708797cdac4b696827ec4e149a810f392b3fa7c1ade8a04db47e14f9afbe376f6c9d064fb7ed7ae375502b28ab4c46539db9fba929bb80261a0fa98e107e164207e9ccf679967ea85ecc303d09a65e753cf087d3ff7246287ed08dd4d66e32017a2fd0e7bae8fb93821d43387b87a400951913d12016469231c99b6a1ca1a90f4e80f9e3a2616c82f725b7c14436be30987e006cb810ef7067199860cf2a78db6f9d9eb56fe3f5c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.136701517Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default, Name: checkpoints/2624/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2624/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=461bb257aa8856a0c77847bd6d6e597a609192b3f5161534277f9e2f3231dea00eb7d68c0d2de05ef6a6d75bb9eee6f697581e93aa61a1bef2077560d3ae2cdede88f12f4b074493c863e2bbac91f493141658f484e7ad9b81a73c520a12a76488b299748031d889f74a9e119b535e727a61b2e8e1433663899704cf76b526574a86e27074991767b281f596b42742fa907bb51d52aeca0e342c8c71366deb49f5b4b149c2c1279ff4ddcb12935638ab1371490acd8e1b0110780fb41c6e98bd602b8243bf7c3e7c52dea115e6347875581b75d34e2a25a2485c57e300b332a427a8fcaad418eb49d04bc5136fe36cdd768c8a67a43b3e40bb77181ad04923a9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.137449629Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/array_metadatas, Name: checkpoints/5568/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5568/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=29ef4ca672d223688905c3cfaff4179514740d01fdb211a2cfc486101dc80a756d362186dcad732f22f99c5b882b97b53c5d196a86e45d049d047100731189e6c3f2b666f820c38d1a959770298940a1f374e9542e7682e9c906c108d7234e009713f1574f1edbdc357236efb09f590fb1f7ac19c103f792a416a869cc7d366f772d70e35fa4ec68a9fb4f290b355fcddf859971a5b5af504b28a8be6c0fb1ff2696e84c604eb93e16659d08a7010a4fc37dd94cb37a9adea5f1b6f21578f7a50d9e7522aa7b81e09cde071a830eb63857c9da7417f6f202c851c5e4570ea271965ff12f0ad2e730c8c8b19a863092437ebcd336ba52a85ae49c24ce14643d72&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.137694965Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d, Name: checkpoints/4928/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4928/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=211a30c69d41ffa96e109c5cca5858d7a5b65af5300c7f7933d3ca50b74f545fba170c7700de4a9c9dda0822ecfd6c675fe531f328398291201e84665de06d3f466bc13af1eb0f4f6d7bc5bc16d3eb758eaa77784458340cfb37340a31b775e2dc6906070f9128e8d32deff642b7500f768a3e456482d65c04d4ad7f29ac76b40550d8be32cea79adff46bcba935b2bba3f0f13bd793be88e7d8f7e1f863d7f14962e2ca5602c531393d886098f607a43bb7647c6fcfcc06d307c01745cbd1e84aeb7a169153cba6e474c8386807ef9f790f2e10d935d043642b5831310104113f5de8156cb69c59e25c76ae39ed024dc0afc02fea4afebd444b5075281e84c5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.137749091Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default, Name: checkpoints/1664/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1664/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=89010ccb244445b97c9d65928e733ff3e34ff2f9372e60380c62249a565ab03a44162acc2a64990727244ec6d73f61bf93fd58c9823b6f805372f53035f5129b4e311153646c5b65e86c4636c5f6301ad64fd8b418df84e72054eb43ebc713985dcda30bd3f25575921befabc41690d0e5d0cd06b33830f9ee76c16129ce678e3d55f595d4aadc40d464e6a777834b40b8fdc586f4829d116068b29ae8a1cb5efb89010b6bc25e0a603e2701df4d2c6e0d69144ad0918affc22fb1bec2ae2f5dd6c5e4a124af9a961ed06b17b55cace4333bed937c47b1bdf9f508022fe61d5cdb73e90cee28b72ce33a4de50018b08c7a80383404b601bc9a022d3f522f8e9b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.137815754Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/d, Name: checkpoints/896/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/896/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=47641834baeb948bcfbeec381c1faaeca221cc591ee8306e9ffbc2b4e3300b476e5d89a9f3b05af23d5c42f4b789ef3ea91b7ac4fbd56830675862eefd88d04b67d006bf0587963c505343748a86e59e6a6642f69af34469b35e7b2a0091e651852794d54aa21b65a0d83ed44040fbe4c197f3d68d4bf232f358aa9acd16e418718b30f2a67c1a6b29ba4b1e606a325efd830f4e72864f1db6901e9356a653d55f5ab2b52d6559088bb54ea8b0ce19e2f23daa8371fbb9b9a9efecdb7df41d18c721ecb99698b55eff917e45a549f5129502e3a56a1812f6ec2eaef465d3c8677423cc1e71ad63b30e581204f89fe9a6fa5d8732cae9b9e739f1a7f867f27228&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.137827503Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default, Name: checkpoints/3392/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3392/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=bf97e07755be8ac944fd11579eb711f2b2c95257f8e616f1870edd4296f5ce9e5d8e3b61a2b1d72249fb5d18daf6f6d02af16e26eddb9334aa2f14e57ba1ecb65a4c2a7e85ab1075e023ab18f9ff850b340a25f3f911e855f5673c0028d60d82ce1dc9603a26a6a2e63204da251daffcccc850ea34f416074c72b60479b37e6d2d466c60dd2aab91d40faf3ddbd24430801b0b076dd86741ceea90a9abffcf848a629ea705753fe4472f5e9b549b3233bf997e44531db19115809a9aa4ae1b4f6ec9035d1dd9d242bd453e9e627e4360b54679680dabcbeee92715381bb794b34460ce2eba43f5c2016325626f28a9e8b743a06b7994b437edca843006712006&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.137884135Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d, Name: checkpoints/192/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/192/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=085d1b42d53c233a38b6f00ba2c15a55e176f0006330277bc64f808d71fa0a23416c12bf5e9f7a31dd9431408826ab03da4d11e5f25b9b9f088785cccace0d025853f05200965545eb28fd5b4bc77e3f293a82f28a8e86edf05b51e70a2125bf65d961a78a4cbf199a055d324bac53f287d76a74d78a5d80c01fed5c4fcd6ce0869f870f3d9dccd93be31b34e349c14cc0fceab46cb10102be14e3d06bb24f4151ec80849481f37ea0a55e31c82b89e65e7aca8b0f638ddbcc463381b29d128d41a1124c0e6b7f3af4fd866d93f349d98696c30176f0d292671c926920604bd8b2eb3a11a79c4ea7a0d3fa8ed3607f393807c7a55d30379f817bc7218b5564b3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.137946552Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d, Name: checkpoints/1664/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1664/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=656115701875c40ea2e0fe8fc2e767418a339d3149407f0ee928c29587bef983542f41e77dc961906f85dae3f3d6dd8fc74788985e0e99bc4d67bc253c6598f7ab80b79c19c4e0cac019460148c1ed5624f31fdedd1c15e26be896cbe81c21a725165419ea2dee0aa48e03be99ef01e810cf855237b1962eced637554ea381e92dea2ac9014c4421c70fe512f3251988be0743bf5d184297986f60e58eb7139b61204bdb982c5aba757cb496ee4182ebc1ed67190a0e24edfa6e1fd0ce5c81930f4cb4d9c65fc8da07329ca119a26c57c709d3f1de438e7385e92b53ecd548250ba04c41346341500ad521e25916fdd5bc6a85b1bea72adba5104fcb9c4c40e2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.137972473Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0, Name: checkpoints/1664/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1664/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4e2dc69884132e929ea10a83d850c9c661b76193e3f200b5898ac02187fdbe16ff5d2c86ce4a84bd661736c251092db7e376a092e3765637c7f8a0b348c8c59176bc81c42a3d0de65b940034a9cc95dd20272a8d6b9f4495c2a68948c4d67c16ab707a3a9d56c5bc414857f60b3c7d765ace4eabfa26d50843ff646eb0f6c4bf7cd92aa0e1b319b229187376ebcd300a35e65549e5ef240ee667c30935cee5ee7b4504e24ed29c432ae531ed98eef669fda5af5a99d595349f9720158722627bf6f45830dae64ff8a34456f632656ceff757cd364a0c4e93b84a0b1f24a829082b74eff460e3f3d7b3c2d33407548ac1bcd9af154d04165eae4185dcb42a8500&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.138092648Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default, Name: checkpoints/6016/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/6016/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7a45cb62c32fd7683963932f77fa413863a2655109455ce9b77701ba94a0bc2ca745acb2211f8d6d7ef11f1e7a19e9fea2d9440c4bcc05f0dd35eebbad1278143a28267bf8d692f805080fd57aafad5d45b4f6d1cd9bb323179c88001dfa5f38884023c72f655c7e048725b92dfa0117258723847ff45e2635c3cda0bb864e3e35995a4eacb99f97023c3902d92382e3b0df098605ee46239239c84600cb73bad64955db0a62cdb6514f68d4337bebc41ac7f829e2ce00c9fda13b7be006429232376d9a0f808128d99bdbf27bdba3f150df1b1b4b50f7636cdad1f7fcd7668e196d7f5fd43381d2e46229abb6a7c89476cf61e6f500d3d022af4556318df008&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.138146606Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d, Name: checkpoints/5376/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5376/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4d1551fa82cfe026847be3822f677d1fa31c2754348d1698de6b50e213fd883805650113b672ed842f11db49b5316f7dd869f7d9decc2ef6f707784738289b8e9e2a276e5ca7b8b6a99826293c04a4ab5e002f26dd9912f14002d579593ea8d174bac434dbbe48aa558b6b503815c9430f2771ea78532d4a8a1ee185c6437c3db670b853aa8ce9cf7735f54e38f4e866c25b3d131069e95fe064315cea2aaa0749af1bb29b879140dfbd2f7f40b7ab40dc38eb2fe869ceea4d3936c8c3a8d53491762f5a7eb6ef8cf054ff094103db65887959d41550e83d654ee0904f327d1413e236618b7757edc8003ce775f2bce96da4f4133329390731a9b1e015e1150e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.138179281Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d, Name: checkpoints/2048/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2048/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=24e9461b019222850dbe2fec9507ff6c4b1d395d82f0962c689635ce090ab45616e43c7d218bf4d8e34724f175bdf64c87fd12deeacaed06c5e71b0fa92353398d311c48e983de92b3c3106e351b197c65de69019eb1e81e7e99b69d4dab02bca1a6931795806e7e1ecebe176409fab2c557c70809ab4778c61d51a2284b36ff04d1a3dcb778e80bfe015efe59bbe292013545bb0aebb8edc6eb1f41d1d065f8dbbe9dd2e7b139984adcd4af6a9446d8329058231db74af30839d79e5f7626a24e135f2b7c0d40fc930a4f9fca1583054bc53eb28fd2ba8f79885c8e81f859091b9e8ef4a67a94cc59f57f5650b1c58e364a08d1b544156e8798a166e70b781e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.137751732Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d, Name: checkpoints/2752/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2752/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=a604079c8efdbaaed1c3e5680161654119ca30514f56fdaf262ff0d1c89b39c537a1f4023566b176c9df2c9bb0d0a23aa4720d080d30ea7fe44adb0672c2f762ce033225be3d845613d6b9c4073c0a99cd955dd63b316d6938df4cb6416c2cb15fd33ebb4f4c1521ca2348988e76fc53be34de914d75040f9276a5b274a0bd159bcb9f26887751f4cf0f3583e46563449c48a752865bd3b5d745f2ec52e3aabd09ea149c4397e80a384f33829d4dd39fcfa1e298c6383eacd8e4b8ead1f278a66fe05ed94039311daabca0566d554cc99d44d71fb166e8d3ec66039e788fd397c48a7765f92326411e0227c02ed04c52f87a7ef01fe8066482cf3b910a348632&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.138444215Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default, Name: checkpoints/5760/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5760/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=99e955e31fe57ae81e3ba0bda6a6077ca894d2f48507914be7c79328127e1b89b954b50414aa12de5ca22a07f7b57e76abc3e3680130805defa6db1d3a5795da3228872841d7bdab8fd1dd3b683840427ca1c57a794e4adb182d00ada449e68273b9fae91a2bec17cf792816084b91be3aa17ef1acc17fe6d64110793c36de1dee9508aaab7d9bdff475a0c4ec84ab628ad476934d3cc4a5fc2f250fa986c232bd42a4d1457773b1c84f6651de12f06d76b5316b26b843dfd8de1decd99bd82ecc6431e6705e937b56c42dec85cf785d7cdbe5e66c877ec0c3225b19f085fb6c7422d8e67592787763ec0958881d48ce2f1231acf5e0b605e2a132f895fee674&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.138492303Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default, Name: checkpoints/1984/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1984/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5e77c38065ac22b6d16a132f589e716f1bcc0f3873cf4d28ffa1a7902c42c2f041909f8db4093f52eba081726e17faf0b3a39fbdb0adc5f191030ac5298ba5076f5b311ea6714fbd9da58d7b3fde43be971883d09dd64e160f68fb9fc4bb15b05cfc97b548482f3bb597cc7af561ff8f97e359e594ba4368bbf5bbfdcd83eab62c02dec716fbf1c84ce9a789be04a7d25814e0bdf00c83fdd9e09e130176f116ee4553bb3bdee821bef5a659325b3d6f008da8ff7602b0c8a7131335d0efa72c374c52f0cb55043ffd0d6fb14e8715c02551f9388d5d59560d7ec5761af75532b1c81b57488b4a9bd5f2e3990c4bde4314b781a75167dd471be768ebdbfa8780&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.137826793Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d, Name: checkpoints/4992/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4992/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b8d92262d9f2a0db7ba6247eb7577c7d1ad80a2920d66f2693565238fb152a4821249f0f2d86e65c051ea5867b11e063d7f6e6e356f7665e4816664e635cdd73da62622f767dfda6a357d811b35de04c7c9f8bcb2df180039626a99a5a3645f0d7024d8d5a0705f4f6efe4b5bf576a2f5e84b23c9c2a5f7a8d9bdfc096aa5dea3d410f258af1923f8162e6952c9d10a5c02c4b7c16b50ddf8d1e61544bba6bfda889377ee5a1a726bf2bd7c766f0f1b2f360b6350e1693a5351d7b61d7674e97cbb9bceb2467f82abb6fad34caa65526e85a3120a0a00a59738107858ecd6a246a90219a2c6e2952f6b1f3e0ef852b80d0ec2d0f049c7ddfb29090e9d3ed8e7c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.137905025Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d, Name: checkpoints/5824/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5824/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=2861077c7c8ed12a5becb78e03b95748d97c74366738daf16bc4fc776163accf023d7cfca681e3e2e66878fea68b94a9c1bf37bf1bbf9408022699e14c01087222db16209b8a54a1b0398428cc83defd7db715a387b2ee0ae7d14c0716ddb999c849d21d188baa17357b4763ecf2d6e504107d2d22c25025c47e21d568e48736c08d667b163dbfaf9f22cbc7edae2399054a851f61eefb3bb05efbf0fe817fb27e5015292001a4ea858795915bb6ae32b58b051552aef4fc5af88e1925456158ace95c7d88b982218582d65f0b5dac15249658698150647268d54d4016543edb41dc0fe5589e2e2b1571416aa833524922000f83940fc11c4303ee31a36989d9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.138698168Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/d, Name: checkpoints/1920/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1920/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=29d14d4fa8e0dc59228b7d75dca97bf1b47321f9344fd5d1c5be744cce6c4210fe37fc182d0422b6a9af0723ad0e5c5cd59234c29854e46ca5bb46d47a2de821d63721776f0ffcf63c6b2aa764f21a119820d23c5bffb136aa012cde10b2b260d722c3ae036b702d186588f7153b4e77adf5aafce3d89420c5749f376b4b7edffd048e109fee5883cb809e0379571d9101c596fbfc849b7d36a18783bc19753c260b195167e713ff2c1777964d0cf1b542aff05d87646d4c4c13deed169f40419aedf9535712022133fd0b5313b884376b002930a0eb1975a8359f8b530fc35661c18b6042ab32198ff4a915b1b40784ecd5ebd623ecc3fb26faba5d8af9d9d2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.138734284Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default, Name: checkpoints/256/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/256/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=a69dd70c62fb5b4391dae8f685158dc9cc85bb7620e384e3b6486b45e43f356bca7bcb00e112fcd961a58b0318f010471088f7a86b6850dd2e24f6186ddb307585157071666f7912efcf7451d0075b2f0ebb05617122481a64a7b091a2f45dc5d5d17f456421fb004b7db028d89a7841cdb172e1b47d0e4f50e6deb095f1e00d559385edda7813be75dff7f04d1adb3af036b954b4994b49b92ecd1e5d2492d40374fb9d2233547f7c28ee89a726ea98c09a453d37a1f0c4864cd962f671ed0bab638dd71830c8a237772776e98dbea6f302f9216e1d84194aa98590a2809047491eb510f79fc9bb4144dfc0954666377b3c2965995e5d6560a39980bb1bb3ac&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.138757561Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default, Name: checkpoints/704/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/704/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5b2d04daceb9cad7c779fc6ac816cb85aa41ef16cbe04a5302315baaa2acb9478c48237bf4fcf612c4462640ace82a4f90f20a916dd5cc26808bbb4e1afc5a2dec578b64a95717ca71b6107653f48784bfa103cee608d8cba738720463bee1f125f80a898d04cdc1ef85d8b834e9ab44c59cb349079e6ebd7bcfd4c3de06cc9c54b66f2a716ddb72c66fd085858980956ed20fbf883100350e7c4d180232d7e5f3bfb55389ec4c9036faeffc0200ace52bc188ed15d8f067c3223ea0757894b876263d35fdbc886718c2f2f5e5459cee9bd212a66593c3c3e3e86c8cfce689fe83e1a3a4fc81e894963545e95a0111a9aba87615172229c6bd8174b804866417&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.138802606Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d, Name: checkpoints/1600/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1600/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=0d0cf00091a45cb2ce8a7f75432fd4edcdcf9b72b49604c642368e6b6792f6e99447263f6a7021bfba43a1086d0ca51cf82e967881f9207d849d08de796104d08ae1756e0730e1f925f48df1c292e2a071c47a1f9f7f89a058d361f325d16cd01a9a952da9aad9803846bd8e346fd0dfcf2c7285e1d899aa8fbd3d01e3d2a4cef9470684bd68c53c3ec12631ce391fe936e463a9abe7a0b010afb0d9d24c484cfdc7a8c8546fb3126c691d332c2bfbef490f8b107b4480f06abbb8890539719efdb8d170a455ac9aea56b93c16c166606e8765ab912da557ceb87c7063b66b58e1f473bd91f573eda63d09a0a5e03b9560b88bac971e5c13711dac52ce8e0eac&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.138908761Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/d, Name: checkpoints/1984/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1984/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=28b1f116392d085700edd5f84590056cc7a43d5acb027b8d8f28ad7044e7f7c9d68dcd220d7f8dd72285b09d3a86f673d79478cc6448b311428c55e7f6112d23cf17f0499f82334187f446e2c5c7486459b44f63a0dea003312487204fa1834bbe29e0c83d990324d5321edd4900869564a1fc76477e4c8e3de72326cd6a87cf4393aa59dc58507bd21ad7026163ecf6f102c0578ece8c4658f3d0c6e8bc513ff27532ee341a5b407fe0c4de4b776d7e8b6b3a91891f338313693d622fc68d96ffae1510f7354f0dc560d0f4cb7cbe0b857e247ccb8be7d0fc8cea54f55686d120dd53e339a58f636e15847339d0c62c9001f2bca927aa7366d21fb212965cb0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.138936071Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0, Name: checkpoints/576/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/576/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=6dd873215252105eb462642162b134d667e0832f3f974908b89bf34c8f1b78253674cb37f7711c9b7c4cbae7ca7670595a40417f920293dac0245a2c02285f1293f44241cdaf5d47ea792b12ef41411130da04eae14d65bddeb138515d1e1616b5a58c0d020e6076efc651babe476734c9005956e7c774bbbafc806648016726aaba9e116d8d008256c93d16d561cf95790eac8ba618ec2f502abfb94d50b6a648754cbb61f071d3dcef9ae837070fdd5310a3bbd45936892e20f3bc1747da3a7ec0700e649d3918461d2db8a426c3c4d063b027258e418e231846308b5a81d7369dd2ea0b44e118427b8d6fa42eac93cbe3c9e4a3831966a156955425bb10f5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.13907643Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0, Name: checkpoints/5248/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5248/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b9917ec0262d0b8affe4120a8d27fbaa6e4843bd2686ece8b19c4c4ceab699ad433b442d8a78d855778821fee781fa1ff57cf79a6b8e430496ba29cb4668792f08a55c8f1be1fb3e9aa737d176226f1a58a3c5d884711ee37173a930e8b51c91353bfe0a00d84405bd923cabe327d73cabe275d76b1d21be950edfc072343426e0bef630dbd2b599aa6a9ff51d03c7a60a2bbf5242163163fbba50c651b37f04c741dfe8dcade176f8784a6582462aad34edd9b3be1a70b0e6fff62ce1cd2c0569c4cef3fe4c0cc10df9a99457bd84383dadf85f476a4c9633ae8ef2739180804c2969f6ee67072ae131b7e5f990b2ac2baba02119e6213c3689727203cee075&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.139176077Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0, Name: checkpoints/832/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/832/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=47f7bb2ef52196d65cc277bf54b1edec8418e60968fe3abf6e0892668735bd9f7f743f6d71886594bf48844819ab165254bedde3b7ebfbf77378fbc9f86f891a1ddfd52812efae32adcc32064f5745222896734d2f3d3e6f4e7a20738b809a8217d0e85c99dfa2f80cc017ec855ce555c2d758c293cafbb04f4409a4c1b90b103bb07d2b9804ea7b6071824ff1984cf7f3a9aa8be5d2096ed62df03c1c8888fe36d73377942844cec401effd9c26acc2d5bfa3b54f9399a948d4aae7e33bce5f42ddbd184753e66e91cfe4c3581f7a30ee07ef5cb6fc7a70312e351ecaacc786f458a67ffaccf502eb6e057b400c46beaae2467acb919d78e07dbe63c298b8e6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.139204177Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d, Name: checkpoints/896/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/896/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7813485b27205f92f083cdf993e16dced3f487d2715e89da53253a1ee6049c07d0367fa08d26e23480e7a9fe335fba806e58449457eb69c199989582ee2f6ef1f4bed16273b485118cdc63c1201924c934a6477d436a62f5f0f69fa17389e6dcee47660c4d6d02cf8d136f4b0661d637aeb40bbc1cd9f403d37c19da51daef09b72040cd9a077609c681b115e62848c69e519e78b475662447b41a865ddc38634829840d7aceaf8442431880e98c925ed1ad4421b787ac05ff60186da9e19aee1daa63366972bf4a9a5097564b6db74329cbee1fd5662334cfa86a72ac54a1a13483ef8ecc5df0fa6b4e5b19278028585a715d9227c19a8c5c2e7b02dd4b5363&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.139279698Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/d, Name: checkpoints/5952/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5952/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=702a03ed6c57a7db7ea56d1e03442e832fe1106320db996847fc8dca9ac34a3e7c1f7586df6ae14d1d491942afa6cc9b2c98bb2e73847fb27def9d466fdac4740dc6c5bade17f3b72ec1554279761a6377b169757c8231d4c2c17d1cc4c970f5b802508b833e811d12e361ae3bec4d81a3023a85298ee73d1373c73da5115e1e2d46b99dd922844e74b50f9bd81233de236ff32359cd8d6b85fe5bb2e9350448ecd665494121718b68a6aef131ed4223d66ff66036749920d7c5ac5c777d4b08148ffa68b64aa554d9980881c5b2ae670c6b49b739c53806f1bd75ccceed72ad009f5bc2a590511829a3ddfae03c9fae7db7ecb8ef0554867866784ff1a9f0b1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.139319035Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/array_metadatas, Name: checkpoints/128/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/128/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5407007e99a8344524b9cdd7308ac8dd49dcadeb0be755e4249dfe154640a6f58334bece8065bbb10af293eb5f01c79fe2e36f3584a5a100a8b70510fdf7beeac46ced92873e7aa65c5413d9f6d7e5b45e36167454efb1dffc8c6cb9770ae53d68488075b71374af96efd0db7cf54bac9ccab549364662af6c87d8bb7bad787b560add458eddacc2addb7ee1d997223cd2d6b4f4c3eeca5473707d76f64457da6c7b6e25dfcd28ff63aa676d3ce2ff5f38deb0a68e6d8e80f7319fa45c57dce16373da8570880543a0ab6a2550c8b324292aee8dfca49ed3c4304692f655373458a6a32f40e1aa3c48bb17ac93f860991b07acaccf25f01faa886c7a067fd400&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.139401144Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0, Name: checkpoints/3136/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3136/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=909d01bd3179062751d32b94e4531fe89f96d2ddf8951484437571c2b1884709d85777bab83e9d303c2d65d02e9d1bf8685d6fe2c9258ad5865c9669840c4f04dca48aca68ab9cac222245785fff6507b9ec98de318db81245239db5ac7ad238a36bfd334cb4c5efd8b3caa3c36d8bb79133bc3c241f12a748f2c57b94b1f81e4077033a06cd47c207b68f64c4acf2448fde5f28361d8f378d70a6e56da13637c1714fe59f9d7d351367abfe3bce9a6229c819218a76bea7f852a554360a9bedbf938d0615c88af30a304be648132223dd668192103bc17544e4e55928fbcdbcfe834194f2c21348b021aa567b7cb0492d926c3d1b176612bd25d6bc473c5ed9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.139443964Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default, Name: checkpoints/512/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/512/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7104ce0b7c71c11775d4ef05625f8b50d00ca3daef077c881c1debe10eb8b15d1b298e8483cb35661774de4434715ecf62c299c0b3a3927660ab13f572c3b9e66dc02b311fd506c7fd49a8bcda466608acb90538226fc6928d5d6a763ce53db306e340cba8888c4dc3f0b4d5dd820dc56786af5e73d716d5b0f5c1588e46f9de2b0a70e0a275e6b1fecb9a82eb264057a4e64816bbccc97fcbcfbe229d13076a2ce7aacd3fa00c38475177ae806214137463515ed974c4ee6114f7532cf0b08cf42b9114288a5536d99c6833ccbd28a219fe0b698a2ae182f021ffb0daba9aaa724833fa2070d2b59691143d1ff32d9002440d1e006871a63e232a3eb3eddd1c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.139520407Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0, Name: checkpoints/704/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/704/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=27ecf6f53ca24fb91eff4b9f143412bd9ec5dae85b5d230105a71c562f7727a81ef279623d7bc381495ecbe42d9feccec9d325fa5b0e6e18395fb63e58d7095ca2de36c5231f3ff3963f27fca06803d9e21f608babfcdcf66dede08236223ce9a167d22925be198c36c6e1d0bcec6e079f991fa7f328b4f510d0f1680dd32babd4a9367ebb1086083fdc90e30f004c62cc4464fa00cc5b5b5edfa7cdd404362d024ba8f263e6b3b36d176e96bcb13cab00aa1d9713e01810f3e4427eda7b3cf380fa951452e10a31038ee3b0a97869c19ccef056bb6501b70799ba539a7ec8371d8073fdc1467a4017cbb4b76ff296ea9cd2c8595d8dd71ad81863b35bac17ce&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.139560714Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/array_metadatas, Name: checkpoints/2112/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2112/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=660b7c68cd02211a3ba121456d50fa0afe0d85723895c87e2fa0575bb2bd91a7766793412c57bd65e4fc7ff5f4f7017a517d7863e7657d9e542f8d1879e811458d580fef863d7943d90ec9fc37eec59110b78262ae31267377f0444f0748a278d89745effc06c37947b68ea40950df7a713dbb182fa3b881a9f001286d96538ec1c3c49cdf38b8ebedf1d439efc122b3338a34ed35d289fbb3dd62aa21184190fc919184451841cf65ce61d9cbddff5b17ac3c4394fd5eef34a3e8799f79a10c1ceccb01d1affaaf54d72be5d5835d6042c292c6bfe164039357ca1b6029fbfaa39f66c8ec34a49105f97cf5f004813273290e90cc0812f046bb46ba6de4c7d9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.139655663Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/array_metadatas, Name: checkpoints/1536/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1536/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=02e90ccb79b7a3c055e66bb490b9f20cfb0d85eddabab0c2b60aee1af348971d232436f0a9100d2fcd3f07bc03be4d84b96a4d7fffb039dec1743b528ef1e83b02b0f954e5df630064fc4f220fd2ca0649224614113eda358a6596c0d9dfb07eca2134bbae91b405e3c785f7197fa90e4590be52476d264efff1d7344313d55e05951bfd76723326ed002d206fde560d9ff5bd6b58cad57bbc795346dad6614c448ca9aa9f5270ae76d607008b7a44d90dd52ea75dbc7584d56d576425bed6f6375a968b284d4cb365d2c4f7b949b34dfab20d15bd3dc2b357f04ef41f7460e9f4639b1c15765c824372d888a390dde4da9a3eaf414a1610890294b4391f24dc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.139681119Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/d, Name: checkpoints/768/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/768/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=51654fb5c2e548833e457c13945f0d8cd5d2b7887d2677b8aade972010c2c2771bcb762648e3c2b5f7321954f238531cd65519fa02d7a598ea094140faa40a800f24e40368341ff0d1a62a13a284f32433043b2b8ad81099b0a74b30ce34cc2103df4b5841b9f4f95888784a20ccd867d4a117488062065dcea27db733f72eb634868515bd0a208c3c45eab6d2be854b051b89e6ab6bc5aa32dc3a08ee833e5cc1b48ef8678955bb542cd9906b3da5e04dd5a54f6e3c552f4a4b3a2e4e97d55fd073a13c4b2563d4256948c4d21dc7c7b5aacc77f2c156c24c62859099ba168517ed51c9daece6b47b1374d4d45ef60eb5334ab380284ef5a79c3bfe6699f1e0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.139813334Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/array_metadatas, Name: checkpoints/2432/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2432/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=713a25cb1e57f0eff5cb07de58efbdbef0e2c48c56a56a4215b619cfa7c1ecbb550b9523ca3bbfe3e3d7e76ff14e29bae79c05eab8636b375017cd903c1f0383d999bee2575e864fdaeffd93813177ecb2c9f43c7078c5773688be47e33999a96148f6e4e9e897f4f2dd1f3d5befbc2cae838b93778f364f418f239dbdc57aef7743186d37c9ef0c4ebd2eb01d388092f1bd3b2f9cf636eb28c4b9a3fcba0274d9159ed8a5e3cc339dd4dfc433c2321c458eb5ceeb911e5650cc6695b3af79327c8b2708d2273782930d9292b499d699e383834d5763923fcc8ea5e5642262c317f2ba68f89e6c0843df9ea556b89616263bbe3720a018aea6b0cd4716f0e5cd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.139873153Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/array_metadatas, Name: checkpoints/5184/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5184/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=848396d2a2a4efff2c3fba5fc984742e00f42beb30e94b39b93eeb5067fe55d53839bbe24f7d03cb3696170a247fea4a25f478b0cf1ab39dad078f56d539bbb4ca030ccfb47bc245b3c2d01b46a6537b3ba4ca6e8cac16b08507efdaa4e2ac144af2cb870d3738b70eb2ae625d1b1aa30400907cbca26c330b69ca4ce0c848bc89b1593b84770f50cc43288c2892e9268751fb8bd8d9ea6065c820025b9a4f01a9a7b86d3c6965e7690abf1eba3ed433d1c2186200368ffbacd3b7ec8e09b7ceab6d0f8d9d7294a8c411c52a1acd44955a113b607696790a628e8c9b0670eb10bbb9d4c368b3a2885ae88c8684632403f3e77a2d48217cb8dea1bf15f862a98a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.139901307Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216, Name: checkpoints/1216, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1216?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=3ab4552304705917da14d2c0c8e5b7285a23550427911a8bb40583fd0584b6c0469014a4692f074864f16d2691295fe9a9d30c03125d133b3caf9a033fec893c6dc9caa3b3455db8e721f6b76b70929e8c8bdbb85866f8ef226f424d1afec73f4bbc17d9f60f6ca49af24526d6959ea1c203e4d5b7829d12b9db49d57560f8aa591ba74a0c69ed9f4164076b9dfe8dac162a4639fd88ec1b439da95878e2eaca978c347357e3dd2d338dcd343f02687688f97062896e7fd86f89b1259222fbfeefb18fccb2d3634edf8d5fd3126f4ac3a057990e6d2893b19604e8a470b759efb42649e100476ced9f36db4be200b703746d0b2d7a1f898a3184fc9369243a2c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.139977046Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0, Name: checkpoints/3904/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3904/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=6e0fe9dfc7076d124a518df25496866b4e3d5fd90cdeb8d9f0cfa6b0749be934ba5d3c5dd445f7a99deb1f83e65e608995c4bd0ed5fa4661d796b01d16da00193da8814fcfff5670d0899881e9774f8edeab1d251f2b8f50a24adba1cb95e9d3f675c464b56854b310cb7f6b4a3e4a14c741ac3704fa19f65ec79eaabe209ab6950e8a672de69aa7d148fd49af52f37c5e9d2e478926e379cb423e8c5d13e41289938a3566b47e0e5369703744f6ab9baa0b7107116b6fc08dde68073da27a6363a5d856666a112598b5020af00bd946003daa5ba85fc1bd44ca3c44477e1635078ced189cc7eeb347591231f4fd3d46bd58f162c0a086002acf40ea4d206cfe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.140005731Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0, Name: checkpoints/6016/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/6016/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=57565ec6406afc22d41f4b8efb08d77af9e26db18c58ea50f4707af445a26e74a227c9e04af45b29aeb584625528f53677d349df8a8d0ca8911434666759c1a2191dda97183ee802819a34369cf3cc80acdaffe4b8884e2c56aab4d245bb0ef97cbf804bf4e06bf026082ca311124554f26ad19834b18a587a97fa955336ba74cd92e21149cfc96f8c1465947be975f522960dd2d77a45655dd123ac2308cc91d4d6f5a3bf8405466f769447dcfba1e04b611b8dc91f0a0f1ea47794bdb4a677f66a91dd8f3649753ca46375ca2919c0fb84d1e96a98a4cd0a10ec92e7c2989ae1f8be5401e263f5b7b934b99083afc332c1abaea372201b6d3cb96efbcf88c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.140139016Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/array_metadatas, Name: checkpoints/1344/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1344/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=2e4ad18a8dc65196181cc3ebe70ad4d68377e70c4d4255acdd0d60d611530f8d2957493449aa33b52c9726ffe6e10fca8c0f1de3d37ab531dae9cc144e537597a626910d291c585c6e7fecbe57de585690cddb958bb64d7c611badf855a873453d4c6efa629b26131d6486ffad6e928186b8816351afbf9f92f92610c9841f33305bc350b2c9c08f3bcd10f68cc58c4eb6cd8431d1e85f24b296262993e45711eed901a14c3051d3c984bc9913c0bc81b1a8e0294e2d0b24729762b69f8614d80535fbbc4a11f222de9b16d05fb0195fa01a069f7acb7c72dfd8ec70f52d6da2edaf019322bd6453514430c5ef5baa6b6edef98c9c567dad8223f4ffe81f1ede&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.140233652Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/array_metadatas, Name: checkpoints/6016/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/6016/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=2008e2c5e1586fb9e783c8c39657d1ce26d7487c10949fae83e07a23dedacb2cce7dd8982c1489caecfc2bfa5ff73260c25b6253766d6428bd1362c1b1b290830426dc4206db82f65658fe54a97f7d7f5d43a2f35e7fd4eca0d53c793abfb525594c979d1e561c4d673062208db924e7999ee8a45b6e0a1c36235d40426abe330f405f2042bcf6040f0d26b60b2998849e11ea20a5e0c80ccd86ac65fe860c3c912f7fbecc917f85a5792522674eb4898eb02d8e48d82bbbb700c4e20df34aa28350e79315fc1d4fbc319ccd837f28b96d923c0294418c3ab7bc1725da0e5c5dd4925732bc4bba4c38e3d415d8941b7502d7d5ccb3708dad58d7080fa82780ce&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.14031593Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0, Name: checkpoints/4224/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4224/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=1be4269b4e02963d2c31c2dd2c375e3e9a44d810e0cefdba7822c787e78af305b55a20aa667c8abd36c79d9896164587d004f498b290d9a70cb1e7db4936ad64eafde4a462d435a4b390a68499407f2cff8b1f913b5639eeece0e0e4c490bd11f0274a25a203ea9123f086210fd234d58e3c4eacc8742b9180eb057154f0befcc2c008aaf14ce0dd6002c1d83d320741263625f26774d47c45f5aff80812a764027fa88bf6597291d62c9ae3a63cc9877360b33c125a3e200a6cb2585694ac2cdb08260ec3f5ca6ea140f83628b3a25b4e715b2b7e5cda3f5820a57248241bd1cb5b2bdd92799f516e5f8439c23b5f951b15c795202181a0d6d82f93a8872c08&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.140414663Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/array_metadatas, Name: checkpoints/4608/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4608/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=8f4af480c4f12f3d50034e575f90dea5bf33ed1d49109dc61c03b52b0ad692f11ee4d600bc6ecc1ec9233d96a912767803871d0421496c00a18a9166a1b01345caff2fc3f550b74c068961a8d407c089070f47282ccd1f6fd38886208ef1ee6cfc515cde4d80dacb3bd2c2762b943d55f170fe91bb381893fe3558b0524935c0e1e8a04af35f210aa9e8d87d45607336fc852132a5d372cb502c90ec177133c53c715fcc1e0366429bd6c286ecee196c0984ab91acd11fcdb4f4c09e633f980f97cac4ff80d19e806b707b4c796c86c41b5f0350d3ebab2ef08f61605aaabd0e1ccb78f7d1a0c632493eb0616413462720a0a67cc4627b707c06479d217444b7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.140467245Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792, Name: checkpoints/1792, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1792?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=80ca32ca3906e1614d577adb1f6954904e4b746ae4fd307eba5be9921f58f8077635db6a50625a1f69366cc73adb41601e81872d697c3911727eb28a9752379d8bc1ec55d6b6a7f7227cb1a483cee937e4360e3df44561451d0df494c56a8ec7ecf6f2d48eb9e55c408ee9f8b9492e1f6c8bdc3a91305dc9f2217ce7a3d28383a09dc05f003945124407d537a2540f744584c0ef48f19758866a3d5aefccc9f9d3e43f0b99e7e7ffc3198d02687a4808fae2b2ec419a5268643c5f3540101e404d9c36923d422953a564e0f6d81d8342cfa8435cd1051b800338c80ea20ed2ed670806361c7d1ea6306e3dd2e7666c401e72615a1410bfc34cd6004fa11b1969&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.140483217Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/d, Name: checkpoints/256/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/256/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5c6199588f5c0762db52fbad49baf6f5b221c6951e898ea3adf23c0d0c1be27a7ebde061cfbb4ece6ef6ee9aba096fce66d3a5691ced38bc973136ec5ca4cfa1e76617eaea9e8036fec13125784d4717ed3dc611fd7f96805fad4ab1f8e8a3f5eefe8f1dc68deabac7bde8f32ad3bf2ff3e2c7819c56b5207d755cc96ede60ac7693a2eb4c86e9dda9a2a3e2653ac9e383a4999c356b7969ab7e59ed023b30fbfb498e76e7fb21a9573a34800082ec7242b3e64d013083766b2fb202719dc1920c46b72084592c6331e48c482075e61a73edf0515aa29542fe5466b739e0fc020fbd1649a6494cc22844e187fcecdea4d3c1643d56e17d8788bf1711887d3078&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.140507717Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624, Name: checkpoints/2624, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2624?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b3c61c311a5ee5161f7ed21c6b3d141e63a1be26d4d8f3af9c55f990f774975f0bcdbc0e3d00011cc78748992f1f74cab030733a659be15197e23660d0909f8e8169fb3d671c274080a34f58194daf0e31c27587f56dce004fea5377a756e0ec15b12fe366d8456b1ea94784b3a7a62d649303508e9ec86c1e5c31d6a996f6cfadff34fbb35ba96167ed063a92982aa19636140caac0c3302c1bb0e1d198df04c7df43009760480cce4702229b4ceaa6d595ac86d9de9f547a76aac551732c1b636b57031e11773bebebd55ed6da802ad40419863cfbcb0d84c81013e923dac399b4cbf3e579eb06a85aea7aac9b0f629354f2a81c5378156e4314ba701e9a06&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.140543777Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0, Name: checkpoints/4800/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4800/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=0439e1ba0d2a8b3b058d076db460e18452957949ea5f2bd2c49c34fbe0b35e128718d171f038896ff86771cf5a89391eb3ba093d180bce0a01b7b653a2a76bc5a11a953e8190e28b2124e56fdbc98b92b943ed16af699b2d9d03462eea3cadee5a6bcfa48125ec5098cd86188d3279afce4e0b85d0765e222c99dbc0faadc047b12058b808259c0416ad0a9bba380145b42db017458695d1236568e4f1e3bde6d339647919690067d7eb7c40e2970b9431ef0e01d262cbe6bf37c03abd0c74367a27af5e699b0fdb0007fdec2ca30db5d4eaaa071a36290f99ed0f88c7427c2e271d82088e626c2df0302bdac59cb47def8460ab9d90541dca3e359366c3d639&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.140521269Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default, Name: checkpoints/960/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/960/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=bca6128131ec9b0f6bf39fb5c250b1d3e6f549728c4ea8473c04b81baae6c388e49d275c532d864df893d5b1d8daf8d2c86860d12691b8845661c7a06a0edfdb999067417d888e7c3be32055c2caacd2e8434b97342e4f3364ce59499b3cf43750977e9fe664ffa369e6d9fb790762c00b92b7069054cb51f895c88b7a1136201ecdb81da79e347b6acf267a2f81e69805ecf5c55acdbf9ceb9189cec69bd1b0bb043d370cd678993fffb5e0bed5377afcbb6523fbf993e633992a7fa2e475a481b717f4a93830208a85e7834d381f5d3db3292873c814950ab0a2a36921c232e8bc77b91292518f22d009ee9098b70948c8f5415ea63b26e69a9ce403c932b5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.140603519Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184, Name: checkpoints/5184, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5184?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=1845027e6cdab3aa4c8604ea5b1e530d9e6c22c77aa8576d1900787fcbc1501817d283e1c03300c1595bd3a947c7722bda95a896dd3ca7188008a09f3caca288e7aa6a921e9db633a51179655d48aa872b60d6c3803b7c6a406b111b6db2e6c52713daace265bb3edae154b514668abb586742b7b6c755c288a6045bdd659a123af07773a61f21eaa7a67d49d9047a50589914730ee55088018cf3b859919deff5f7fd3d823c7d54edab618baf845356d981a94ab21f83e853307f9f0021e1de4fb974975e0ce13b0813e84b8ba20c2a591c6f421360453d3e61177e4b69543c6139ddc2ae1c811201ca1f880b8d6b16db3c442ba291626fb191f07a2628e1bb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.140423354Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0, Name: checkpoints/4736/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4736/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7d3e1798374a45ca76970cf55b8cd3ed00da1391344b2ca08260e0398599d6e130dc89d1f54236be61229ead78d096cae7b520000873ea06a047eaff9abd59d176901de70d4db94ca79399f7c77ca944b3d065aaeeeb5be00a43e2f0ce994f818b47374fbce1362a5ad8572431b1db82de928834a2f376eeccc07188588459e11bd77441368238f4f906c22bcf51bacd3546a9d995c79644db4fe607366f32c0a904dab32fc905eb0e5485e18ed8d7306f287e54596b352ca2464a5efd0387caa00b1867770f42f5c709502c1cae06ba7c27140048fd6db9bd06cecee53ec7b86f39d5a92bda7f059728466c30e7771f62aabe5da1be7bcf1273c8f0311afc8e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.140564136Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default, Name: checkpoints/3648/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3648/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=578e31609c5cec9443f62d0435729e53302fafd7931562cfa76d0d743707a4aa3ed420b4d28d17a35574e06810a0692dc856f3782fb4f6497ff0c9a72c790a519b802b0cc344276e112ad7e2cd5591db5349f26af0b0ee00fecc2bbff928f12155372dcd008931efebb7261a7ac6a3d57641f4dc6523fb120fdfbd3bb7d5c1c21e2bf567526e932a0a9806ee1833ae568a0c16eebfd193e0ec8aea1a0d689ccf755064fd2c35145a3256ed4da8d1c09d692fff291d5aff41a8f97a2814a70cb10a13075f30e301fbc7a8982d8eed0356be811f96cd91a8d453912d576a7fd6f97cb6c39578666481dfc2dca63025c06231e79dcfec89c4f1be57d33275d9af0d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.14068235Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d, Name: checkpoints/704/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/704/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=544cdee700590d989f2361a166c7dfa0fae81810f132867cade3d37f4bd2fd380eb01de400ad93fe20ac679e13fbcf828d1e0b77fee7338a4a7a04f1b0105f3c54df84dae47829131954eeff7a15f5e0d42940d419e8f8516054e753b5b9d84a61844e6257a73f9dca810ad208e7feae19c3e624c1055bb50d4b8c06416ff764840574de64222f379cb21fc9f6c4d9a9f51ae35710cfdfe3435c1bb7b06eef06efcfbc9d4980d6ae5698b85cf21d68a6a0912dbe426eabe2028680ec6748cd619eb0977b027a9c777ef9d2a5ae39749d286afe22e52d152b1a797ce46881dc5a3a756634cef67b8ea0d9ed5306c54a618b57a577396298aafdf287babb59b234&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.140721463Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/array_metadatas, Name: checkpoints/2048/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2048/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=bebcb57f237961f4b974708fb46ea2a395b01af860a89cb76661f59acd08e14fa19a04dd55bb65bb186bda90bdecdeedd162f07db8429d7073971a6031320f9fd515015cb5ab8c1c9feb8b33d04c2dd23686da502117b1b316c3df80b3dc702155d083d10aa44b4a8deb4704e22fc6a980fa5613e18a158e4961a08a478ab5185e03a8f2ca194af0612b4666221c3fb8b0df9a2892a7f80f6c711313b598641bdf9743b0cbb85b7be8f9da476133ea11ca12ad042a3c318b457c3e5efccd6597590f593192cc69a79df6962cf24b6041f302c409c878a8c7d7b36e697d7f41f8fc1ab6488f16722d468667fe76d58902f8aa6fcd48f28274ebc522606fe736e3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.143395321Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/array_metadatas, Name: checkpoints/4544/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4544/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=a8e0d961b2e6e10d33faceae78e405661d8470d288e47da1975b444601e64cf67b9155ce35f30de6676ba560f47c52be160766c0a2bd3bb068da2766aa4c1c1f08de280dbe8c58fb5025f30a41465fc32801ad0fde65ae7f5cb7009a96a4f9b9389524ad05a254fae8b70f10de7722af77b17e2f1e999f0f83f511b46bf2e40a1fffc8695d52dc4a85573d9a1da3214c83b6f08ea89c08f8969d679d65e56565bcfdd4a5afff878856a303b72201d09d09a6e1f9410f176feb92b6a6aaec5cb70752d587c4ba9d8dac605ecd99f1b9ddb0decf96b1d30c32adb87d22f02797ed13eccf48c8656902d76dcefa43d65136cbd881affa1dfdacf9b2524ea1ef63e0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.135559541Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0, Name: checkpoints/3264/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3264/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=11782e9061981fbc7fa2c7134e9fcbc3ebd354980733b82109956337bd361f352c97a264be8c6c1b49192ab09a0dd28d76b90628a9ea9a8a776b1920a5ded5859ab5ca42e748d7df4cd2340de987c0c608026ac7c069b7e49e308e1fda40070694d674b12851d79f709ad2cc4fcf55d906f6d5ea211d06b456de9f4b340221aa695bd1acae4517f6a159cc9c4d1d9f67a53c81daa0a786ebd2092e6a3c1c4f83f5d7f4c9ed2af9d761c6829e33dbd37d50bff0ca0bdd35e06299af478e9008ee9766cac9ebcd6b140a27390f439caf0994bf607d4e8a10d4ec7be5ab6a69d66ab16bc011fe478b92e3f7ba66b48abad33a24c23c42498aea61491a4fb58e1786&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.140393573Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default, Name: checkpoints/5120/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5120/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=2cb516da1a15d4f83c741e2f9aba4eee6c37363339325fc5eb1a18f5730c022ba443e23c202771b801d16ad6509b27974c4b060b0fd32e078fe59eda046f24e2592a615525f5af6cd575c630d60a53e93cf26c04f3f1b2f567163172d37b498c9a8a5e95f1ed3570b2eece87e83cffbe95d113e278e8d03a787b889912a4ece87d5f7448bbcb90f0b2e02401f4ab2db4799ff3f1c0b08adfec951276531ddc4098d1525b5579c4f1dd608cc683eec34cbb55dd1991d70b1358408cb95680098553eabdd51f37a578d2e63223d573ec4a62d0feb88460ba04b5d6c91dee6f4419e6dad3291d3577fcad34d61ea5634223a72b6ff2ac4855433359096e1b674432&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.140500974Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/d, Name: checkpoints/1216/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1216/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=0b5cff7621f8fe1eab29e9ee9853c3d83024106c0c30ba3ab7d8926b7d051247665715b4abc04cc64cbac68aa97b2a829488a2f5a13efe82e902430d6ba046c29b410b05bfde59e5d4d50365c009a14c44194293a268e37269d027d6774b7f944f4c3c4dbfbbebfb3b6677584042d3b7df9dd1b86d3015d65231cdefdf2bf165169d2e4ee1390b7816450106298596ca95aaf140c3da1ff1578d1762c157858c8dcb9765854a0a2674a54943dd654689d73708f7593670c8d3523958d3177fa3639b0c550f9ecb3b9866cf7e9df2bdf69c291780738533837a13aa1392276fc5a5f284955885802ccb18db8783f6559b439d504c9317d34acbf1848579d862aa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.140335514Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0, Name: checkpoints/6080/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/6080/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=67f24804eae71527cd57c86cafff6eb7fe2bc5d2addc58fb6622563c281ada5ba6af1af71a4ce01f48a6793d2c459e9acae41aff1778655fe49fb728499a424c2dc512c6202a42cd8aa1d86003bbefa8699f830635cb195bbb5fa05b1feef7fa2cbce3f70c1777f99755281752ac26db43813c6f69a095cc147c7ed1b33b52fc64c10afcc4765836041d067b621600b1c8c0c6d9fb23eee59f7453a4b5ee0e6d253182406be93d93e974e22afd043b7383d07664b6b1b4f0960c02241d049b2cc5f47f3aad9061fea2dfcd2c461a4df4a40656f071be4b097bd1eec92aab3786e1d383d390da46a9897bcb95dfda47fc44ddde4bafc6eb214d813ffa69519f95&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.14013942Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/d, Name: checkpoints/3648/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3648/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5618a1526000f38509d3de771cb8bf9fdde5851c76bafd9e358a6244f32d2f70f41e779634d67087d0aaee090bc963171c14550e0424e5195a05fd564d99dd731f47ec692a8af5987a1a535fcbe6ee8a702c962dc50971ed8b0fdb808393ff11fc273b0c5c48fb0ee2ae3ac518881584fe84626595d5b994d275a6c21013f83a5863d784c55fce45b62b2b1c01e4b8c08064eb3316fb27c838c2ab9b97e15924502fdeb232ec7a2ab9364058b42020ad0534acae514ca10b87e064d1dad5628907890171fa8b13e861647a49191e354e15243510777de74266570f4358caf26cb281f8e89ca82230d7cfa0c75af1904ec98ef487d0772ccfe81f90b79eb770b5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.141280949Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/d, Name: checkpoints/2688/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2688/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=bd81e62ac08bd3af5686612a29b8ced1b70d4ee7d7bd42c20d2b191a289854e21cb132c0b84aae473939db0bc4cdc136c939b44daa724b6f44003eea1ac5f2d0fad60c47dedff475f228f5821e616816e432dc999a3db402b44481c4e4588f82deee79247e1cfec640774914163f9bb45174b3cda20e0333a44b71cd75c29deaadee8019a18bb3795339ab7cd74d4a51aa72a12b7d64ab4dba597ea123be7565c043a9a3f88f1b2abb78ca2e348bc1c78ec5e459b24cc0affa8e40883d66e01d805b75c146e9f86f9223e67c305c7d02758423bec7431e25d19a89f7b79ff1cf9460b4048000688c10d825fec6c0d9edba5b4ecfec22e2f01000467f319ed595&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.141646973Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696, Name: checkpoints/5696, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5696?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=c3718ff217bd196157aa1b199d8015676786a45314bcace721892e0cdb8f9a72f550ff193612e81d75d26f74f072fb1ed78a9db35036b8d175b0777b9feae1b911795239c637939ac5b801501209656d720a5974d6af8b5a97f3a78442ab341c7db648522ad1f884beeeb7adbe8838ee45d7cc55f51dcedddfee593d4d04ba9cecb76ae78fe0c3d80ef73ffd2601dc8955a1d3dc04f43586151f63ccd94fdc5944df3fba3e6b74605ee6a06c3d01896948a98427c2eadf78c6d58b20c7dcb676b26322588a92a1a5df2733dbc6518e7b69a06af7b44cbb2cae1aea67af1e85e5c140638020b251884d3bb9ca65e632dd09a61ef3ee1d4e9bd046261440b57a7b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.141723969Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d, Name: checkpoints/1536/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1536/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=acaa3e6640ca469a62ef27bdc6200bdc1344042532ef6c417636275157f310686935640310f2f42fd4e296ef2bc7c431f912ebb22418b0b5d3223af97299b19641b72381d973f1429f95f83a368e4759c26c73d5f1119c601af0e974b37ddd566b7c145c0870c7cedbd572dcdf701ab417536c4ee1b6bb5c91e0b21305e905c4d8d85cc4c358a4ae1cf021f01fc4f87c08eaf60931ca26f986241563f40f18fed1ec6b2514aaedddfc2e04ed227a5f258c524fa62078ca3f33d06203cdbeda228a271d1122a85c97d43aed90aaa442bc29e07ae47128fd8c0d86ee8ee891d4d67ce9279a87d1edbfe0ccb37c4d7fae799fa875c133ba50c8318f0501588353d7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.1418059Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/d, Name: checkpoints/6080/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/6080/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=9b410053a4809eba23343abef3200f4c8faae741780f7cf000771707a3e8d2461c3a2f7792adf79a9d4971cb464327e0070ecafad7929b9cb5f029bf7a229fb34387a0523febd95c6f6eb9acd03c26243a8098d82f4709e7849b502a064e2556aaf2025fbcda3fc8c7bb08a96c7417742c15a124b2cc6da12eab46b66a1d006806723735f5b0993a4703c9eb00ecc2fb35181a613ad28c726e975195d8a5b9c729844ffc526c8729569609798413751cb7138ee25505eac27295db11837b3bdb9c4f6ceffcbaa8e013fbd88d765dae2d5734b3e25151c42cf3e089a2f359462529b30ade83cabd883eaa9bacdb9d393d8822a6a90233ff255db5cadb72d09548&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.14187464Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default, Name: checkpoints/3712/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3712/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7f9e0c1b18edc664cb803427e5eb9c93f0bb59827cb96c79ce018a123f6a10b87ff38f48cf3ee1f99e08dc5ab0858c6fa22eae3b7dd41f43dbea4127de256f87c315d5b26a2aaba9f8ada7802fb836a83858d581e11a89cb05f0c85994f05a45fe67855e1f1fb981a5e27d57178a897233c44d4cfeaafbb0ef59edf0646a549f8234eaa4f11d306f2bd64aa6455ef5e4d748ca7f114d4ecd96b56fcac132fdcd58732052905e27bde8cdcac42e06cb1aa895d530ddb4b0ae69d2d2c4ddf70a1d777da106530ea75f4dcaca94eba1da8290013193e2264d0c0a4eb2c57c38b97f651a10f041d362ddb76e0fcae539054d856404fd08b14c33dc93c7e503938790&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.141995034Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/d, Name: checkpoints/5696/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5696/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=774907a1fe761ac382f234227f32c370210e08e363607923585bfcada4e77cbf4d220a1de121eb7b5a7328b89db4608e896a8ebc954678326cc83dea894dd73f4f216f307ef95a5e27375dca47d47625d8576f5dd4204cb4db68eb5c71b86bd48228ba2b771561505b6d903522a06b3e70736d22ac5af36e302f7c4ea83427716f28d6a304e68dfe5bfb8422ccf331c39d6daa0dc5162d516cd33a5adf03bd245fed614a7a02c3696b2d75aeb9cbb7483962fe067e8329146bbfc29b4d5fb7ad13e67eb1eeac1c24ffa68ad76934ed3e00dc4a09dd9de9ad2c1b8f7fd7ee6ac97fe79dc2556f8d6e60d4974d3d2475a6ee78c44c589869cfbb114b57b60cfdd0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.14226436Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/array_metadatas, Name: checkpoints/768/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/768/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=1325c2d4b7f4dca6a0c6e949127ee9d214ceba22ae1530a5bf01fe48008d1260612b8491590ef73cec700cf6c0994dd01953ed3edee1c84f93095277b8174d8cc6f4dda2458e2d91aaa49eb9d96ed687961fb89e7fbbf0b61ef110f4f3aae7489f0aacfed2270f0b9f13305cef9d56c56f5193a441f0fd136e9c37fd04cb4a90c3f05e0c96754ab3f130937e122aed3260603b9db0d9c2949e9502c3ebd084ec88b1f855ef0fe12dc4f2833a6e5eb9279203cbfc48c73edda89270e31c1c78b60b1ce62b72fbce2f10ee7a0a74c552a80029c51dd78b31b3732c68f1ac03d406816cb0b55007f23ea5c8491ddb54982f3cd916635756df7fccd69247fa8c14ad&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.142338051Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default, Name: checkpoints/4992/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4992/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7f309e38a6c0b1ba441e0dee71691ae357787c9a09b1481f5306b88be752618b5178c70a802345d28256ea79716c4600b63ebc52a8efb1632922640cd48c7e7b155bb726584db8d7f0fb1d9092c5fc4467f3927c6cb7b030792ba21570def5ee33a3823b54ffca2902fb4478302af35588559da7f814a003990b283c03b2f6ac7a232b242ccb0d8787ae0094dcc8bac85d8b6c06f027882cb5147d04a620d7b260c102078d4ea3f38bd6379aa52437c290909bd624db60acc9a076b5d81ec09f92c3ca96a399809df8955e6ebad2f6fe6fe59fe753c693f39cfcb8c1913f1482b27aef1913387431002462a708d6a95f14a30cca7dfc092d517d668974597c61&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.142340224Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608, Name: checkpoints/4608, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4608?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=9d4f32a95b2e6a8ee3a9b87f735725d5e568bfa2a1dc02a9c3b3e0a4557707c35e6b582e7a55ac8dfada36a5980d33fc6dceb0c7f03baee6a05ea3e122d419e5d4fa20ebb06b14023a79bbf187ed9474e38b2d11b1babfa192358f3c3a0cd2a5861bdf1d0a05d03615947feab21238c8730468ed791187599f0e56416aa151efac04b7fc4b4ab27f37b7d7538203f0861a9dc78b653b7d1f222ee308486cce654869153ec56bd8c40c50815b084ae2574a4723348b183e42397e6cfbeb5570011fa5ff318ae8c4f5a40d5dd523f54119935c524dbe6ceaf0887be63204972618f35c93cae66b074647ae6cdd5a0a99b55327ab91c412c885fdb5bf2776757b7f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.142452042Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default, Name: checkpoints/3456/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3456/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=1857fd870702a5cb47aaf70e7e864a09b9f03ac407b1300a24da112feeaa202d32e1e8ba4c0db2745f0086657c9aa9a22411cae23c04d5b015d2636636be1da19c53531cc3b3e12719c3cd485dd7304e96dc644e0a6dd45b594d68acf5dd23197ea9f4a097056d18cba4f45f052dddb161b4a1ccc7baaae2ef3bdd793f432668630ee869cd3f5f22a97e5ec86ac6cc06cb5b03145da893832f6afa2a7f81ecbe09b87092268a58b96086930e4b8b3a776350e868528b3c40b05d853f73ce999be950c8fa09513916ca13d2f5a8a9289662a7a0bc61ffd2c173de8b9b097a3eb79021ee4c602438f8d674f2a1d460ca44074975b63b8d7450ce8036c4ae9e6b51&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.142535051Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default, Name: checkpoints/4544/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4544/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=c00e1708cc5af663b626cddd4389860819e4fb6e617e91c68e4a5708257e96b4b45592854bd4d59bdc39d953285a71a70175b2d5625ca555f0a448e5416a7a25a47eac7f6bb4a143421f35543d526d947dcc9700c77c1f99bfa2921bf448d459dd981f35913f7c4274c1deec1d7256676769a2f0d08f4cf32e0514da95346e174e910a4d488f486758858c81145a594c8d2e7f5c462b9e48e5feeb95f1a558cb6e055731b7684b710d1302cabf24adba85120e36669c7eee75f7198208139846795c164dfac6371ed01a7799971c620e8675ade1a04d4c5393e2c6082a6cdb9e420d61e47c0d94d07176d713d6e925d68b145c7bac5e2a3aea9f68de62a4d344&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.143197779Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0, Name: checkpoints/1472/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1472/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=1af6b34e18af4a023e157c9a5f26fc65cbb22bf895c7fc6621eb9fc1b1e549699bcc74d10f28b3545a070e4e02b05076f833d87e494be9506bafeae64dada48a5db4dff553b91442490250cd2f3082e4eff9613cf696e131a6762da2e56f7e6c42e8aabc1f1f7b9694dbedbaef500e0262c612bd0b0bb988aeb13e3f9d20e00ef31c9375b00f6160c1a19eafa7257b9503dd8db9d447790df82c91d86c630011e7bbf6499c6b0c3b7b52175f4d73bbf953f93a1c12f35b83e66ff7ec7d7d1599cdb8782efbac93aac0cbf9cb4e7c76d82bddba4e9e73139789cf78b45400465651ece87a80242cae111679fdd0051bffc3fd28276e74e65bbafbec8cbfa1d744&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.145646563Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520, Name: checkpoints/3520, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3520?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=696a98c72573d8439aaf9499875b6c26990072dd93e40faf18416e40af9efc64f18cecc8fbf231726f36511c2b504f28bb432ad91536296ab3afe44da4a85d058699ac4cfe2cf89e71f822809b22d20ddbadbf82355324f4c3ae425886d3d364f3a249313fec628c34ff49f3bd6967f60c13cff64af5b84eea1ad102a7c29dce69a93a4a7c5bfd0f294d94228b08c7ee913d3275a848e64893050ac69e9e619e15f60899ea53b3e1adb0b2b034b6d695ab483d471d8d27034bed1e0d0d11ac0856c20b4aa12d1d4688de814eab192155831cd24b496213c5778f75ec76dd4e4930ba52eaf15ef29a825eaef3927c5b058e6a9f7eb27e0ea1a6f229b0fdb1321f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.14570064Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d, Name: checkpoints/960/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/960/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=670370e5165af807185e539f50fe24336bc4307b68ac0e89341af7a886695799bc1341afd742066fb8fba70a304ead224b6d9e387730f2742d7d15f2376db3ebc0d9e400140227605763e8eae965b945ea7317e7e2db0257a90db1b041e72d6c49b6bfca49e3dfe99f9e1cc2d075b7ae7d075a11616c018c0c8cff0349a4d671fcd1b69984389057d6c0aa700392f63db607c61c4464a575e814c20b4f76722031c30b223085d283be6d35e9c776dfa802c0e7c0ab009e25ec0eae4cef55da3ed86dc7f0f1e9e500b1d6323003ad1aa32fb80de4c3b04d8b43f6414a949a7b397072782c5ef1da311e5f4ccf0e2d35092beee72561613ae5246adc9dded1aed3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.145741171Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0, Name: checkpoints/4608/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4608/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=1e676c84d019650289113687c7e9948dada4a5f01c7b19dc5e5085ab507a134755becbf711afe17270781e405147c43db527a85256941bafd8681e8fadbc243f08e9164543fb995211cd6198401437fa4f4a7faff11ef6002e2a0d99a7453a84d1a62fc38f4876e615756014bc9f39956d770c052db72ab6ba1dcbd429e8538081afcbe08305922a79d87a760cfddbbfd1d11717d1dadbd038d2c69acf0046a89ef11f66e84ab9ca26bf73618dafa5aa77e1663d7d26b58aa3cac41e0ed2e77613732612952e4b0597e15e2a988c36150ba9d994cd7838b4ba0b82e2096d2e70046a4f224b0231c8bffb1c33d3cc824fe871228e44ef7430b0be5cd87476246b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.145763345Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/array_metadatas, Name: checkpoints/0/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/0/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7505c6accf0a57e74a7a9cb3b84bdf751e56c22f5a5c0120c99ebb21724532a3d7cd0dfab768aa3c80903e06eec9c5e1d3ff8127afdac6a83758f443d5853912adf82f3425b58a5933f43a1d08e9efd7aaaadae4f7a1579369db643148d0da91d9748c4c2feccbceaba2d7a16fb79fb14fca5e1d7333b94f6ddf6b979d5c2230f23947007e38bece561b58d697e1db36ce2990cda44f51dd5a087d726dc15cc0b82264e3686b1d48ee7e19b8458a8067355bc0c034e72e847c3612936b8533aae4e84aa88f59407b269b6c08fee2713dfbb337508b0bcf556077e2c8f6d8c7a8f59281493123267888793c2fba6fac189cca22080016cd2da100c63d490b2243&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.145792501Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0, Name: checkpoints/4352/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4352/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=c320ce81b56cbf7a4f053ca24435c9fc6032feca4c067a06d73efa4f39cf609781fc71875b80abaafe268f708eb828173a18de1725c1b25ba28fc2f86b80eb2def4b5f1be6e1674ae61a17801511abd70fa63000e785aec95b81b43d695b563ed21878bc418134621448dfcdddbfe259271e60ad74ca0dd7fa591f283001b901689e9c1290654b4c8a0e7f0fd760b86f6ff420b5e7ac2264c7917931223387ddc3e9c015fbd5728077e40333b45b4deeea4c323e12b0d68355ed1fbb014cbf4b945029da5b00bfe96051cbfd8e96f269a03ad4c2a57b5fb64e81dff8d3cc7f54e7b369a874acd46e7031949dcfea256af555fc45c30edbe696bbb213aa5bf128&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.145814905Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/d, Name: checkpoints/1408/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1408/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=2db233e4cfe5668311f7915182c3d6f097d0d90f06f304b2302c885250b0cd23943062e3af524a1e6110287e7f4dc9c0783395b1e3b198c178339874244170d4f75b086804b3e0cd0c126328d80a2afc7b7e0665827d53125fb6698e348c97dfbfba6960419b667ed74037b95b5a031a98d59df6c0e98020fb01aa017643fe6d25387c8d574188d6d4dad53874faa0f1dc7d54abe79bd4dc1010bc1d6a16fde3831e80e08b49c251e2fe193d86522b056b2ecfbc7727c0bda22f822ee99b66208352f5bdc5d006c1f7f455454ce85d1b99ca242afdff5f68c9af4876fbba961f9e249db9643f362d329f88196eb0caeec613c3574d40540dcd9fca1fdb293e04&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.145852409Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/d, Name: checkpoints/2944/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2944/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=c33bf26feba7697b71c21e84be3b8e8c6cce6259afe96bfeaff098714e7929ca3d6f6cf15c45cd3e01b35c20fe4b97fb141376d33262a8c016d145bc80e6c72d77fdc9655cacba9ea4aee2418ec1306ed6cdb98f8b6d351c401049c97919df901aa64da1142e68d8224a2f79095b9391de4d173438f9318706a1abae01a53ea9b320488eca4682b47a1275dc8488338ca22d3b9268a597560aa244c95f9af7e80ce3060e08c597afdd24ff86ea8a29d8f7131a6c50a97990c0a8156f6dda04b14830bb283dec071514d153f8f823a4fb530624bf3b2f4aca9049f69d774dc4ca44b73dae6800b711874d4607f16a7d7313a1f2d662c661632fc4033d2d9e7f19&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.145868801Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/array_metadatas, Name: checkpoints/4032/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4032/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=696a54543b87cb138e67a9b31ef8ffe4ccdf4718b7ec6b43d8b1ac66af1df504fc61f0ba80f516985e971371b8307af457dde60fe7614d65e8b0d308d32eab3795215f51cc6a7fab05e12ec752932e4442a48e20b7573c34c9b3e4ebdd198a492cc8127c2bd0096cbd2dec65e0a6d3a4114f9aac2dde424b3233150f2bb5b1ed5c6ae06094dc014b152915ceea5ef5b0b1d2bab7010c606716643047134dbb6e918c7d3069dd9ad910b356bcb3364c464930dfecb8df2f6e984cd7d0f21044ac68c8d641c8d3960ceabe0b21567b00ceeae478e3ed5e637da564bb7dc918fd964d7d7348426584b46db15ce725f17af9e1a123e3c13b3c29440e868e341be81f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.145918589Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default, Name: checkpoints/5312/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5312/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=27f143f1c8622df14f330aca8363066778212ee4b737299e485dc0b2c5704e0366fff310c429cbc7215b88ad8c3ed30c3c6292c45f06a5cfdd783ae8967aa2f446cd642b3ef39092c9f55eed90f402ff97f4e6ba50d96c02118cfaf20b2fd3db0e2a109fa1c92108e2dde683a0d96c642914f121a928cb42cfd8c7111e05e2758d72f093580364a812fad0c2b6f4e0f6d5991c45e312c5eaa55525644dde5afd4c9851c43dcd8d73846533f65917cc639d9c44d02f5e88fc3a10b859d944b45520eee114207cc6a21f91c0c4905324d788e73d284273231026b7dde4d9cf54fe6c20c190a2b89f0c038da82c43ac032fe07fdfbd79e4f806087afe0d23161832&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.145951226Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default, Name: checkpoints/4224/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4224/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=380326cfc85d42b051ee4f03d0d47f15e13de2695ad1d5aeb201bdc63603ec0fd2bdf799d7ca2c32ec5ec8cc91de34dcbbd9a8b49064855a350144e96671f1b2a7346cb6723ac55b47156b649fcba193af668ed45d85fc734ef4f49639e598e0ae926652002c6b11d62b762c2970059c11faba30498228c6e8b55d1827511a0deabc714217578b2621e236549895241de618ce286a0ad5da9c0a30002a2310f490730d834af313df5c3bec4b44b98a60aa162bdb7aa67594e8e11d969acdc91de2600843aae5ca0152edfbf40303e2e313386fe118ac8412717c6ebf8796d62a18d35ecb9e6b155751fc5ce2c71daf22500328497cb7d6fac8b39359024271b7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.146062685Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/array_metadatas, Name: checkpoints/384/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/384/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=0a03cfe5651db99857c93faff593469cb0c7d7d8b5c4a54bf2ef68d8fde00f6fa5d3faf34186c2396e27e932dc7043833e0a04298c9f41f8780dc43f13b7e0a67d5935a8192b1873c52ea600e46da30a694543bf7c78fdee8ebc1eb97dcee074bfeda4eac08f37a4d1781047461b96943400732a686ceb6bcac4a2d694ca65c71be661c9530e2660a8fb24a125f026c1512bdd2ec1190586aa61193eb3beac75ce88d11a12f974dba0459e4e051d82a9d3780002affe61dc3942d8af1bb1759dd5aeb7a827af2b0ed29de056efcbbe934998747fac1a1d11efff0c06fea7bfccc4bad45958b74a97f132d72f5dab4582d973e37f64bd21fb3522c3e86f55c4c5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.14608396Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/array_metadatas, Name: checkpoints/3712/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3712/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=31e66057eddd83c0ded74e45e800be1f76233cc025b5d016f2cec97c300a5023295f6902cf999a78d58689d8fe0ade8621db4c3f5e660facb7999282e0cf53ba131386569edfe1f0c6abc75b67f7c46353d44fc45603375b3ced5ec8d258d32589db99e01b5788b89defb2e74a572ce4ac0ab23f830a9cc90ee6bf4d5c1266cd8c902fcd23482c63b175d6b9be2eff24b92cd731d400e957b60be30fb5bbd6f00180321ab8445c333a09c110a4af4802b39912a526b96c611780bdfe63e958105f72e9641544766815690789fda09353b8c6de57cf641fd031d279abb1d694025ced6ac2432ef12268c782ec19c478a881ac36a2bc134b8a942c347410d58ba2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.146236997Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/d, Name: checkpoints/4608/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4608/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=9ce6956da9bc489ca73bb18974b1cb5dcf9212bdd7478b98b3409053f66e1609888ba06fb38ca158cff364f96fd90392aec149ba68fef50985709d00b877631fbaec5f1da173a96d988385a096c198224a2224b2ffab219048c061a400e4e8c5501ab9def88c166aabc8e79066bdb4961f872ae5d97ef8d225340859133252604ae14c0e94a3cc26cc4cbcc906c5b3c960ce2293d023dbfeddec8460d2d2956092ea2eee0fdca356d415ee5486bd03f0651a90b9b570f5ce60a1e609777cc5541b2a6f808ff8eb3b07c7dd1290ba9241bf3820b36d520b929a093e4c2a32c3dd782463530b5a1bad36a6536acebb062efb7d07429142be6c3a435a9b9659786f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.146273797Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d, Name: checkpoints/1216/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1216/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=65706b0782b42087490a615316a6145e85b19d972ac9d839c8c4886bf99ab3ac8e12b205eea757b02610ed2a5faf47486095cc44b214431fd27e08a6034a86b10a920900045b98c12f7fb6324f7c9e31c36c8fa1346ddb7d6190ef363477489359c78291dbd11ae7faca246af1d30451fbc34393bd26ea80d0d85e25bed34deee193317b627a709ac937183fc5df5173d4786fba2405f9ee89d4a693bddcd5651571128141d93406ca45da2f7727d0cc07f90fe4fbaf8ae9a796d73bb7f47f4777cbb7b83af071b8cfaf86c913ed3979f5170fd132e37e236142133774efde69d06aa1c3c90f2ad2ec400b60cacd3d55071244a2b31574cf39615829553020d9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.14629829Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d, Name: checkpoints/1024/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1024/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=755368e2b876ccb19f13eb7cb90d1c8529b1d96fa1ba63722d099818e1393dea0d0b137cf702ec9699b9f369075262bac3d8c37a9b49d85807bc9ddf292138dd1e8fe155bbdc05de90c6f04aa0aabd050f19579095ab1b0067b2b8c6a7c7465609270955945bb764f89e9a8c94fed73c69c90c70a2ff3d95a83481a2df055bb1d37c6e5543eab68b563a7585d56a1f001df69e17798f413f2dcf68cc9a0ceb8e1613253ed6ea1b3ca393911f2f06615961159b85dec17f5f817097a7c03d01214e123aa391f019025d5a091a324e52aa8b26cdecd31838c1ea8c3ed83ed0fcfc2d3ef3cbc149b02a6bbe085db9182375384ff199f6a08b2c7c7b656e103d3f1d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.146336514Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d, Name: checkpoints/3328/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3328/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=37150fa59b63f406c4f82be6f58f6407a07a75b921c83e927b23da1d51b3f34611faceb9ed8183a7a22418c0bdc1cd20b9e37be7e5b964a703b9db4a0a57297dabe72a68966256485ab1faba9900291a5e0aeb4ffe0bb1bac7ecea973c5611732069176af96976074f50301d7eca8a9623ade648b151ba1f82a1fdaeba9232cee8a3b81d199c59aec2007140d621810fd483ac72e42b260ddef02c8666944a97fe8316a6b5fcc975a860ee923a604ab6f198dd6830e8a98346a632d170384b4ab3a06fae46983b549a04a90dd9b2b48dbbbc5326c729258f94e3a474de7ff8dca18fb3b430a5438c926f113eaca503757f5ac5e54b4b6a743db281a725e87b2b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.146382898Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/d, Name: checkpoints/2240/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2240/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4e153275472030a52908a569c59386feaa07afcb33cf55da993c31e669d35403a9c8d9ac8bcc4293cf0929923fdd69f05efcac4212597d053f63684300635bf95b980d69dcb09d929ebdb9367a9c99eb98b93f0f15fce47283d6cd560f3d857cb9501310a23aac649e6d94bc56bcb8973b32d25df58d6b356008e504955ed9183a59a4d5e65af60fce8c46db0659f10b107b4f90bbd8382bdf5d6a5d6b409a50d51757205aa2ee31117198df7f42980a65bff3dcb58369f138a0cd8f745892da21efd08d81b58ee976f45ac0495d5f745b9a83879e7f511646d8b116d554983acf1bde718562c22fb958bdafe995864b4ffd0a7da56086e659a149ca27ef6c62&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.146414495Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504, Name: checkpoints/5504, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5504?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=2f98fb1d39a352bc84d233612b41c02cd2cd76a2a2f70ffa456d0b5836015ff50864674637687afe3790fb1f25423a02784d56587d728668f610758380118d49c7c3d5e8f3bdc4525156e0a40441467a73852545c1d161d5bbfa140de1832231f562c99183bb023e948d07e6e2ecf02bc306126a4b2b1d01fbda917550ea3da7b3de46371b3bcd73ae3a5e198060afee3e6ceef273e325ddb332995e9fa34092a6e27c4ab05f8b018e4c9eb541d271a53fe442f5aad63349ceb788c08ceacf1875ff2b062fb9d12a19c1f6f1a9863c816796629a5061c9efb4f7765907524ece61fae6d08517c8edd46e79280d75b8c0ec19059630c6a45cae1bb0c5282ea113&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.146551741Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/d, Name: checkpoints/4032/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4032/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5afe35db75b5ffeb0aa8f2da4a62cf151fa61ded254b5fbdcc5dc18930e94a00c52ea4a5ff3cc8539ff634d8df6b12d958113d368af9b6e59cb19f89c37215939870826c0d56b89812cc0582b2aa195e5b8b7f9e2e002750aab2f2ff09d484f3120264beb2eb2d527f4f30e29b45c930521bee37d3cdc2c4cd29838aae2c2dbb251d7fbf1e4d59b86e06eeb7b5120f4ee1ee020df0fbe909919cec3edec75959813c25adb9c424b5196085652a1ed5106c46ef9282c802ca842ba189640fc4dde903d011d0f7f097a3b381bde5cfb327608ad08f18e49f3ff0668be6db9217cdb5a67ad8851881f10dd8e66e942c1be97cacef76358b8de7230e8bdace9010bb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.145759426Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312, Name: checkpoints/5312, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5312?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=26593e0d246b5197a695b1f4036a029eb62b69f4466e975b3c1194ab17e721c2b2ec4dd96c9a3e15ca0394266be858bdd8d355bd87a923fde6910c43a79951b672ef66983574ab78427883a2da2dc7d86d630d04ae1f0b2922b17f8306a47a54cabc385214e4de0b8a2bfd3ab0dd67c785f27d36058ba1686234944feb565143dc3eaed570ae3535d5274cda71ab204690f696818b727cfc2f477d923d0a461646720764a1ee4757e5c51004d806cf2b55f2c5e64dcff7971dbeb726f6ad4b801e7aedde6a86697d4ebfb4418abb6191aa4560499d9a5198156b89d3ae06ccb97d80039bec2cf2b897785657d6432e21c250549d2639bf759a0946953d3861ef&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.146626435Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d, Name: checkpoints/1472/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1472/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7b1bfb1c95d50e6a3c94f128f357987f3caf6c4e8f452190bd84aa8e2d5660f3815eda46235d17e58e0a35594ccc5edde0beae6c7eda1829e7a27c2d2636ab4b7280e8c2c6c5105423df2b9b0126f292c3f5f41579ec62814eff806b7e8a0ef0a0ea9904451f8b1376a5ddcc53ed1b2ddc53c64525783b6606b181c6779f77d21bbd9bea527e34dcf289f8b6467639199290bce8b11f68600079ab37a832ed0afcf53fe3695cfa99efcc01b8fbe095db9ea0b1c77a8ca155caf15fdce1010e4c0ca72a9b7055dc1291cb7d88edea82e0f8b619004bacf6233d0a9f07e8637c8bb60e35acfd21b8858dad0be141c01f50e0c49627f2ba8eab13e8aee3d097c630&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.146643955Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/d, Name: checkpoints/3712/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3712/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4d1b1fd82105049928963412c8c077f72128bb58e7612a787f5cec803c5e308e4bfe37b121c19ddf89d6746aa73c57c643a6e24092a886dcb5151c6127b47a798a014cba5e8837b1d4c99294b604d378e4e9aea95eb996008b348113241d17583081fa3e79be39eebd9bc3f68e565f6ebd7462e6b37d77b0453db8bf4da8384a9bbc28a38f462a6367f2e6146b7ea1cd4bc55957fe04e7082031423346b646b745907d310f50aebc147fb9257ff459fb349da130f8bfc6bb939403740fb920de3888358a7ab3db23f2a59180a92d6c66d398fb7f8bd2f86217212e5e8ed30c62c211ab8f8eac40a5f502aca9e99cd226f4d4ef1a9e9d3c9d05413d8429253dc9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.146750522Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0, Name: checkpoints/4544/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4544/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=3aea735c5991c17881fd2277bd1be72c263a7714e260b5227fbf333369e3d881e88d55a46311a66575db2c88947951934fba95428f3badc448bfff8f568948311655af35da1defe38b06cb2b95705dd374fa70cd388d02ed0424b09702101c73fd02c180d1d6601c8db3d6f662124f1097b288c23ed20d2dbfc237f145b91c6737cb806898a93ad56906eca77badee17b20e10c248dca36edcfdd94d179327386e2e3c82311ee40790683970321cb85cdbe6898a0be4691248e6948c1061a136a1846075169168a3b807b8442c1a4802cddd2d6c08f99a503242a717ed3efc04e57853d367d6fb6982d64b672f1709fde3ba206dd191cc2697a8bc6ce76545a8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.146784342Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/d, Name: checkpoints/5568/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5568/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=002f2842f5b065aa67a0fa9612df7b3123a07aeeda8b2ac0c94c8fcba1242cad973a6030aa0d0108f8d31c4c8a732cefbff51be4bce7bcab4f464453a9b138204c4ac6ee502420e3ce5fdc89f6c516e0cdfdba89ae9a00b6bb1cacd0d072cd934bec9bbb836909713efd15024314db8c48d6ab659648d021c3936138cf14b9964a8024867369661c638a2d62a792fdeff1b10f44732d858a5e3c12210058f1dd819be76fc76a567d382d5c5ead29bab545f0e9de3c47f0fd07be24a09caf451c850ecfbbb0365d2f50bd15438d562493841f2c34c1f64219875cfdbcce6460d34669de9d7597bd40441fe79a45f1de9da9a36b7f9fc162ff157d076a29c2d067&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.146808008Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/d, Name: checkpoints/4480/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4480/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7a07b235b3606a11cb91963e00bca83b26635ad419342f2fc04d6995e5e3ff72f05bb6dbd328850eea9c31a84221290c8875e9bb751625f68ee1ea2e020c857dc66b9c1ca20714b32fd6ffcd2637b36a27501ea9e1868d3183a5b563e69de9a246461d841a7c38e7095c2777cc2fdd6b113433e49839acd6d3289006177c5b6bce5ab798b374a0c9b8ff1a01227ba535810a9fceeca57e0577f34ad7f32979d9305c511874c8286c94f6064eae6ccaff46df60cb9022947a796c2418de02d2b9f28266612ae958b681c1761743257bde6a0fb3805e8ae0a6a890774cc32cea283ca714066bf890605bf5e3dd367f057c5326dd813a044f2c4f99898a5bf49d19&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.146838019Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/d, Name: checkpoints/3264/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3264/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=a2a7b9f21aaedb0ca2a998fbb0e90a54310eaf023dace34f9abc7674d54cb7230e9faf5527157110f0078f0ae7ee9cc97d0072ee9bda3e4303dbc3bedeaf6d8abd0dc292ccf1788f1a4def4bd85e08a40d78435b5cbe43cbe006083fb30ec530563788aadafd442a22aa8ce47679af1f043db686e8b32f36dcefc705a34a660d6a6fbd2c87674da55bfb1db6e657703579f1611a4b720fbe428c593e01860628e53565b0c7818b755bc34f26c1f444703f2ed409c09dad202a74ba7f3475175844d362de5e0096445a0b38d5770311b7d2ee5ccde18005e0933d4b9d78c1b78594883c465ab03d17b4c215b544d9ae9f6a2970ae63a56100b8085cf17101481d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.14688952Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d, Name: checkpoints/6016/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/6016/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=59ecacdb6c2bb970ae0852324b0b100c8937b72d0c4e00e4aa6ff26e94bb596ad6df999e7b26792b494a73ff0d3c36f28c8931ebfef2f0f7639316b58d325771805e35aaf6eade2be80732d0ca7bf67b908d3acca2b6f50480b82fc1ca922cc0ac12436900fb463d6d41acc1e958a40b4b6e4cb21de3a87dcde5b438bd29b3d56ae08ee57255048af92fc7d9fc7d5ce58626aa482426167a53106a74ba581a339aaa7ad2019a8cb095b83d777f42866013a078cdabaf69d2a4f7916beb496472b98e3750c419e4c5b7d07e5e6af3b1863f774aa0f336e2e63be6a0e3834852e4c299afd11c7c9724b4ccfd651e7420f369d9a9eee50f33694305e517a3a041ce&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.146927669Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default, Name: checkpoints/4480/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4480/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=49dea333673820b89d3f4da064b8d3a07196bb5e76d12b17c5d7e49b2ec423924b4e2a056555a3d0687232938f41b3c46507b811a8f6cb98f01a95201a631616d90531785050b7e48542350787e4a41238acdba5389c27af68c629b52248be7ff080a0affe3c422f0bcaeb40af0c8f3404c762203d3dc5fcc50391176bd86d2aa1e7735640cfb0ae35f202e555bb3efb944f3dd7a3d6f2f72c85e984a795cf70c2ec3776bd2054a4b4cd3a5a79e361a96bea4507c2d391e47860aa44ae8fd0d15120d4d0ed7c5aeff52785ffe14e34a218454a5e94c3c760971c0e075edcd43d37f3bf72a998b8e6f235040c909064d538a0da8e2ae36630889bd5a18555a568&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.146971066Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/d, Name: checkpoints/1600/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1600/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=833afdc273e3b263bae2ecefff291d7d643497cbf9c080b37c938d81b7742a333c80de95298cc9ad1debd5acc48ca5b0c6a0a9649541d8c650443e2f108fe0cb65df008d82837ac9e8edb9aa857797c5901df9d36a1ff76c50a803034f933b7872a3d31cef763b50bfdbd545a83c5214d8349fed500e4dd76be332479e7ccc152daec8d07cfb2726b3f960136992e7c707bbf55a3d8e667e9d9c2d7c7bca65edb0f1a6fd4316e71c6fd6e8bb2e4c4de246606f8b82819c5855a2b4d542d37d61f97f6f7393a73ae3324d5ccbb8dffbe95a2d1b43228a1019d4cea11a20912b10e842acae2dea71f7bd7c191fe304789f69464531091263dfa68246d97f4e32d5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.147004978Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d, Name: checkpoints/1344/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1344/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7df0000da6bb6ab54921c39016c1c3661c069b0e22f305aff192dc4d197c7b8c8e253a89f1912cea55a7d3fe3643124d1fbf52938a68a082dbbfcd17a5073a8cecd6e911cbdf241df7e275ed57f9d190ec73f2bf5892e8068a351b66988a7aa10e8f859300d22e635289290d35793d34d27d755f0e9a2ebf6c9248dc88833fe8654245c10d72236bbbff306ad4d0ada827216ec98f7973f28d481b6feb2db49a939581d8e69a92f31e8161e8dd713802391c706adedd7ca8eb1b5b5ff5ff3d13ac7daf20a771d11dcce97dff3daee9e662f8d5fa65f34daa21c3e74e66748da2d9c422c775a1b3b5005891bb84995ff92af3306008eefd4aa728f3213f6f45b2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.147024438Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392, Name: checkpoints/3392, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3392?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=948fb90fb6e60fe88152c16f43391c5f985950ddffbf4dd373592f5018b763d6a4ebafb89e67ce12a8a8d74af37e2721e7440fb4948ec5bcd2c49ad988d80af4faaa99438d6ba88659d8a1766272c9f960c9c945a13d4f8f4384b260f3324d9665901b201c2660b105298d9b4c20a66678d99a6ea0d5aeac07a2ad41dc9192743dda9e31c29ac1ab4322b954c250891a538176dcffe7a6305ac22c23b24e4672754638058f859e5aee3c8756bb8cd6c1db3b618f1c4ce4af4d5968eb299242c27c132bbe83ff7a8cd9fdaa8eb23617cb48522638bb6b45b6fe13909f7dcf32a515d5c56b5431373a19c6b7e0556ec05fd2c113e8f26dfa488c274cde4b4a5142&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.147083235Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0, Name: checkpoints/0/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/0/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=2eeb5c761a1fcf208ef638a8168faae9326d6bb604e8b111f944a0d4d5e83122853bc51e6f2de6385c655ef4b67eddd0a961563b0739dfb5e23f652e2e3200bb2b7e08aedebe39b54042664e1979c44325fbca0254a0fd6d93e43c0acfa2d611502eb1136f2b76fa3b972c5945623ee4bb639ec6351372b96c9ecf94a8fe55ff5d8f8e58a81c9a4f64695bc16025cee844595fd8a69261405f52cf6d2dd4de8fe3c225571ea67d14778b325fa2bf514aa1ae5daca7fcbf822547696ca9c2eed419b4c687a82157a53b7a85b8c73f63490a9005ad0387fc1e9543fdefc79dab164f1e0bf14687f2ed8705fc0adee6debfd0c53233f6622be313d1fdc6185486d3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.147098988Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/d, Name: checkpoints/1536/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1536/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=9a06e6d04e85e1ecf022392436f70ec6c5fe478626f4be87f6166e022592606919c0fbfff5dac2808269587a7a25a6da89d28c770f06a7b88bac3b3d892f1f5aead3de75d421c3c4db85038c2a43459ef0103d61cf384f27a2d2a8153a6ac95b3ffe014dc500b282b0d6a9cfb40391fe2ba459aa081fb3edb839ab1e51940591960816c7f5e7c4864699958d60507737da53a49af39e3bed564e15fb60300a35a76d761fd439ad4ed3bc31157a85bca4434b03834dd0e18c57d9aa57bbdc3096b0a033dde1d0ff7836b74ef1a7cb94651665576e02bd286a4bf683876f0ab9d5b0266830329842195539e368b13c6da5d837f825fa0bc27b21b79fb1a51496d4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.147209142Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d, Name: checkpoints/2816/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2816/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b56f6b68332a6fbbaf3c6b70b7938a1a19af807ad4e2c9f71a63e754e2600485a57c5e7cace88d128360118f174d49755bd4bafdc66f39643e8970785c97d42d59086d1b5c81f5bce5800a9d5418e2c95a6ec813d1cd44432ff546e9bf372a0360252c5a259779b61f7263519741657a1e48f39c21f2d4e86d7067e6488bb932af4a60044f0f9c660648c6b6d0042a93a1624971cb0f5adea81b9983607d5cbff228959f579b7a4c945209d5a2973a5913d8b4abe1bbfce57fdb1847cda892976ab3a8bdb60f3c67abf12327cf1908b6c578ddcce8d768ab899a97f319b0c59a63e1221285825f81b38c7b49b41b0cca0cfd6317c87d05d0cffa5a7aa1091536&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.147311439Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/array_metadatas, Name: checkpoints/2688/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2688/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=a730d0a5804fa8c3386d5e0695b7a30401db887c7e1a4f632ab27c171b7353f76b4d17bd9f85459829585905fbb6708c7291440df6d02d01c22f16d2b88f851668a26aa6b0a016535b34ff50260e9312f385df432593e96d8d6baaacb1bed6f6fe4f3ac893e3f9f50f4a7286bc002f0a650f7ec975b3cb931af6fa3aab993f0c2d6abf4b4b1e808fc2fc24e805cc0b3f86b2c8bdf2c8d5a6090ca3412b917d3f36c9b413edbcd463125a62a8ef22cce88902c481905d51615e9a581625c101e9800244aca217f97b5b3433e34185a86f7ea6eefcd499c3e440b9fd6695a6d8322bb9f3ad27870519b451f4ef3011bfaccdec59f8caf33392a033f85cf36ffed3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.147363511Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/array_metadatas, Name: checkpoints/5824/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5824/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=657bd2f3c3a4f758dcbe88144563c6919fc09caa3f71bea9bef471d3e3224a4b5f3ef3566b559bb8ce9296ce1c18f23cf2a4fd0a234f9b35b7df0c4efad25ac7141127241a498fcc4b818de968c9c46e9261dd390d5939028a94fc1ff7ca2f71c6269da2dde2a409d7068b8d88572e67fb583a322f0fd1bad5a00686886fa9b292f4b249c2a7d6f796d56e2a22223bafe8361bea581c7444592984962caf63b59ee9c5fe522813eab980efafdf71eb7cc1213ef9cc6d55ce917a6e2c43adbe114e1d037ed375b7e14912bd06d5291b616d6cf825770be3356585737efc68947da9aa289eaf895e8b4e84d50d8c23a8f076880ee61f28537fdcced3493f749094&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.147381306Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0, Name: checkpoints/640/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/640/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5ae4a610f675d100ad83e249d964182d5ae3bb19182e803314dbc6b2087b477078bd983dd84e496dd628f321ff371f62cc490aff1f905d2f3d5a09ee99018506858c8a2e01b818476f9e4dfa1db5cde0da1a7c8d37ea96972dee146ec76dd02bd525e9e244bafe99e3e2e2196954d399321c8f63f718873b2bf876f0d4d5afc1743abb68da005e99c8e46dd849ebf5d16d23d5c28017ed513dbd012fd93c601905d8a33af879e32b6d1eaba9dd2ebea4c4ca86e5d3ff1330bb447bb127fa44cf59ebf28998a52e461935474c60055842bc4a563399d45002ae4e7a5009cb0ed8d65822184d0f231bea52e8799ba5bc315118021bbe7315fd40ecbf979710d6ff&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.147397794Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/d, Name: checkpoints/4160/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4160/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7dd38012073aaa3b8efb1a9708be0146dff228882219a0a2f4f0b33221779e794f284c870ecc9b669f45f72307cd476557698dde9254dfd7af12515430ca9bd993e7bcccf8648509e805203511f423de127782f5091dad2eb953edc3caa7fcca7d6105f826cfcb7b33f6e25d7b20642c037ce9dd5015f40d6949944c7e1cde2b0894abee4942aa8224569fa136b338daaa7f75e7efa9decbab737324db99b71d3b5bc441f7943b6ba2bd10d16cda84d543be6e4407bc708ba7ada118681c599d633763db5fb74a8512a22b6122ba33a84bcc747f59086a99053ffa4a40ee249ba186831d15860d2fb23835055154b1974dd495d192be2315d31222df1835b854&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.146811015Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256, Name: checkpoints/256, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/256?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=bd3e5f41d21e3d86dedbf4c84754d40a37828c485c563d58fc93970942de967160bc50064d6635f8d3794f759d2b1e6063ec96edbef7764c126f35261e56fb4bbd42d97fc0c222bd2f3143cb41f13e08565338b79a5185ac256834d556989814176fc81849611265ce9df555542d860052fb8527a7448ade2a0914246f6be4abd33bb32dfe8065ea870f876fc51029bb454f88a75f1f375b6c22a5a34706860b893760fda50f1f186803170d8ef8ad10efa795084ac5890b0a468902968566060006c559567b38c98c6902692af75f52771ed70dce01783810f4b649479c21642e0313ed46aa8216d8015b2b56240e53021f541cf8dc5db3c55f6008493c3f4b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.146922015Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default, Name: checkpoints/5440/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5440/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=9ef2db246de3971d0c2b159749fcc94702b2382eef96bf47eff67cf0fa9a8540bb1b4836743315e83445cbdba59741625bfcc644866f35fc7cccdedcaa31f2c226b81901048b43eccba9c74ad8da4992729ccecd231ec83ff4c16716923fe080e4ce40e03f0a9f160b0f332eb2fc73b8f764582dc0b5606a648d45d805b1893c6742ef7d0c3afc9b7eac641db9ed4e11a057057f0b29cb3520cac97185b274320e3239b57a68d7bc94a66fc27a557f8c72d5ff10e2cc76008fdc693939a6411c442eacd35f324c0aed89b3da47e0cb84aab36bbd9442e7c06cf813de5a4dff5acc528e261be89aaf1658aac76d8f4450eacb4b9d80d508f6f72ccbdb2ea8673a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.14750161Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/d, Name: checkpoints/512/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/512/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4a5b6e6e4dc70023c8d4ca38c6988442a283612de61f6b746455e13b9c82b5af037503453396e7c1de7441672e9b5c8aa112c0ce2052bcc1cb928096242d06092524a409a6d36c783da1a8c228cefdd74097f1a89ac5c3bc8bd148869d48f8b04940ed14f55a3307b89453c004d041bb460972debfbcf330de1774636e4783915934f479480ab2d5e62394a254d80e91633e3546f9e322e2338d83c593c80a48838972e3ffbf76c446ddf5bb7ebfa2f726384ca9b04c6ef221aaf894cc2f9433c31530ac628c3337127c81398130096c445f10fd39bf73e96673107ec4142127702584127327a61f62348e8c226f8039a75e0929885c23657c37fb46b33e36b2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.147544784Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712, Name: checkpoints/3712, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3712?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=9efdfc0a5457f20b64a1fe34d81644e9133e03c3bd63fc73b0dc286cb873ea16f8309ce5f5fb7afe1f9bfe476fb407c7402b2adae88467fa52d94536ff0c8d8ffb7bd439dc832f35b69e433b2ecb0b14ebdb06d511df993f30788ec99f281701d6014493f18b1cc0ecf6ecd9500c6018161b7f32e4c57553c64a472342f458d63d2ec6ab71d79a0aec815dd4e430ad822b93bd73aa07d03ba5cf4d74a8771a4540927ffcc6a931fa545b0ff729389c072241827fbde7e859143e7f000a526dde784b1bb023a178845ffc69bdf2c3f19aef5ed5c0159ad6f69ba96cda5b222875cd50bf754e1f7a6e43c7ba63d437c37f252c46c4523a725c3001c28b000e0ac2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.14758713Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d, Name: checkpoints/4800/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4800/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b92a5f2194481a1e7a8097e0112d35ac7b23ac50ce9e6af71715e15a89bc507bccc048ccdb08d30ac92204aa3315fb667efc2726474a8d938eb65762f44c2fb9cabf9c2fab08ff238d2d76487cf24301481756ecaac49d685c2f2a0efd388887d14f3fb7c3234822c9da8956b6011125dc33606bc688fdbf8228baf0bb1a4cfb025070e5a87e7dce645713a17b1384169bedd0be99450a09421f78cf9f407eab3f4088b49729ac5855361b0a164d2142330c63effd5d67cded33a8214763756a11062e6aa02da09d5a3b1ac16e50548426377d5e44879c7a3e288e29ff45eb01084867e04f54d2c66cda0ab10e2947037270386e59a81ae23cba58b9ffbaf850&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.147604819Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default, Name: checkpoints/5696/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5696/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b31c7bfde0de7bd2178d08e3e785ae5d60edafbb45d877496487a6b6482506537fbf99b8b60c8ea8e46d49f84830a3b4d10e29ca08b7d1979072c5c2cd3cd96ac6cfbcab34195ebd7a24c25d0da18cfc9f45a3e1100d4966b1e28fcb8592538906902f7660f3cabdc31f74c30e681cc43a290bda444253f6bca32cdecf7bc550c422d893bed09ff306b13afa493fa04de5da89181bdf6eaa34ef4cefd60f65d452ec7d39cbdd88167694081b0f2a72c7273ac01836b85785d95e8b6374b1803ad319ee43c6ae5eb7cbfdaff624c8455b6e27b6ae44a68d274a5232a16781a66c095cb08e09a5356bd70558ab13946d6d5fda5f2b492d6eb4a0ff4bd176049248&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.147027239Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664, Name: checkpoints/1664, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1664?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4b396e233a4fbd4577393d86690eb487f96afa47b0c7eb8c4bb0cf60b72b75e8d7a1c39a2ec3b38663cc6d7f773460c9ce93260b9c9b24016399596450a5887629a2860925544d6f5c4e3397fc4650f1f4c54bf1936c0ff0201ffa528f6c94ddfa059e16e73f264c91770c8ded8a9bd590e5557b0291dc6fa748c92cc2a8471edabb99b15ecdcc60b0b4e6ea686445d7bf31861920e274efbb61572890c0dd0f9d017366ac07756dd1cbb699b66f532f4ae59b4624b0f92c9c90de43f64086c272502ebf9fdf4425a8b9a26568a79b3e378f1d25cb1a13eee8b8f86f7589e6dcd913315677b06378119c7436aba02d21831e23fa3c2b45aae77944af5b7a6b8d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.147669354Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default, Name: checkpoints/1536/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1536/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=1c6c4558a12e556cb58b49cadcb1bdebeb54a9b87c4dbfbca1d1ba9f576ea9c70a2d33f02f5840c6ad7f79c33f921015639acc318a44a879532d88c1eaa4c822ec26d463722d36e4bcf2656218687aedb76d2cf04d3fc7fe729d616c9f5e5a6f2281854cb27965a3e58b1a501294605fef24b4801095ba49baaef3604aac09fe61ff2f4e50b02a7bd14cb5b1779a8c834e8e623fb7a9027a2ace725a9dc7d5e2b64621fa2eab331dbe949c5ea2d1d21f6298ee819db3d2b3b4fc5a8d1649fc9e731007843784b8666181586b22819ea659fea0de02cb50db7687aeb2ffe4e42190cf1df3e1d14b0eca4a9a2bcbbaf9c6cab93dc65d7435754411f29317d3346b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.14775175Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d, Name: checkpoints/5312/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5312/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=6b8fe906e2bd99360afb41c2f0ffdda95adcc2566762205f6b23022bfd4f1dec65363d63dacc97208ebbb4141b512d44d2e0a6c162051848beebd28cbde8e390c27a6b83d0937527b71f9d4deb3818791202491b32063ba0db2da115b8e6bcc517c1f361cc390717af97266db754db0d63c5b8a7f972d352cedaecd71d5e8f7beab5fc0ae68da193b9fab7e82ecb9d14677b533c225a1c49b8c6f91ee753b7a87f1a592f994446a670a78178a27b40bd88891646d9c64559b6e94cdd17baef64240b91a6fcecf680607e0e9618431213c602e5f5f196abaf0a6b8606f58c244f199c73b44879d858702934fee34e1d3d9daa4a0cd27962c5f0230a00a00f959e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.147841064Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/array_metadatas, Name: checkpoints/2304/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2304/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7a2823df8a4bfb7518b0927161e4d9e1992d3f49d9e620a0833b1603bd7aca7081624e36d487525ded0e82d97f7e8ace8d265887d2623725a8c4ef0dc372357fd7020bc6f1c6892330a9f54bca8d639f50bd017450b703f07bfe75f1075eed5abea172682614d7ea125deb35dd5701f36565f10414ed0d02dea40625d0940c6873192527677a2763c4abcfe0049553c198de0da9da0c910c201e66a5b89d31b296d69528eac310e9163fac2d3d129c94af69605965fc310fb92dd8ccf15231767439f4df1ac8a757ba65993417ac5ccff3c62638407b1991695444cdb04f445b185b2425d1064974ec1082a60657836797b3148f331c69ddc4123dd3881a486d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.147850128Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/array_metadatas, Name: checkpoints/3584/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3584/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a97af45c971e42671591324927f24dbb848fcf7069d3187c6046757fed379a0a84c970195fa9d91418f52cd41c9cce6f1e5162e7e879fc3a9406a44b1f2efb9bf1165204e104e041b9db8779948eab40555c0446d2c846b439cb831e77b289a9f63682d00428e318a25a7c00c5706583248c7cd23fb04b2fae84ab383e2e1af1cc0bc8febb02b1ac6db9db7833fada4beedcdaca33b40648207b40bb1073be352ac7eb0ec3e384e1abaae489b99bc5ff466161b4e1eb40537c7bedc4959c776f8571bbcf5263f605a46a46e129f9f6bde1e3f245755d09d5667dd7ebeca188d78cf2531ed6fe2b6d842431bd3995f665af740e842456dc716b501ebab24676f5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.147882354Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896, Name: checkpoints/896, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/896?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=1a3800631b32c032151f94f4e156471b262c77b7f0738093123683d670c917b8e99537edb4fa6cc75c159a81258c70b414b02484e52d27ac54638c69c08bc21f316955a1322709c06689471351186d29c3aecf7fab07fff6612486f0f34549ab1d6d00dfab160bd14d27752968254583924d5c76a7e827b4de9eea6221394455acd644bd83fd7edbf4502461625ec3b6da43b9d5fd6a46dc9fe044a9a7c24b8c4b0754feab583fcc962dba32847197b4fbf16f2ecfd372d0fbd5f1d4af3557619f6deba3e7fcd5ac584a0e0a488d165544277794641128295518d547bf21be62dee37f79c43d2e2401e07857ec6b76349687b2e88b1afed3917cc345c5e87881&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.147976868Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d, Name: checkpoints/3520/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3520/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4491e33cf326b8ea8126b77d2ae661f410ba19c616fcf07865e603a1805e2ad5579fffc13ca81f3addd12a838d1f805ee2d563818ba42ba55c3fbf1d13e74e68fe15e9f978d979c86b9a59b78c222efd0038d6cf7357eb14303b9c1b65d80433b6fac275f85bfa9bf4c5c87dca3311708f28aac771d0b4df52d7cccbdcbb310602c48a77520165f881e7f704347e778c440d9b193b6772adaae895aefa60fb7a8b1cae80bdac874eb17c20c404957563316ed98f0b35bfd17ced6cfc79d22b87d394b3df0e70f7c5ce3fd379bcf4e324b406b5aad527db57b610832617cedd51cca99af9f8edbe72afe627ef116ea4c910b28eb3fcf48c536ec51eb81563bcef&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.147383834Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776, Name: checkpoints/3776, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3776?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=a9a551e6f0ae10b27ba2bd9ee146b4c92f65eeddeb7687929b20196305419813e2b90d3f1bc9d03ac2c1742907123895c92976916736391a7cb86a2b6501ef7db8438270ac4ed0b3edc2cfb68fc14a180fa93ec48543ef4decbcb58e3d927efc0a679b6aaa5cd605cbd83886f0a12943fc4c8d455c1220e9ba56e830dd66ef0d12a87c216f377b79248a305316c4f5a3898196ed20b838d71e000e6a0c24192d60ca31cd1016d3aa7802d6487a8d792c16843f7bc150b8b47e1ac3e5be8096153e75ea9a3c52c7327bfe8b6675b4acbc5e65ec009d75730b1c4fe7b1451c7230dcf2aacb4e9c317c37ce793d6521c60652848be426509aae84e2c1b36b7c8951&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.148041392Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d, Name: checkpoints/3392/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3392/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=ba55e4366e8757855dad1960012259249dc74247f6ff733712eaea270efc8d3221b740655d65c61f6576a35c276caa2ead0821cdeb9cba840e777a1750d6699f99c3d162979f57ed39ac9575caa087b12e428930379810b6fd0e72ece2d58ba7a7e212ae57116d9ca1b73bb6a0a8f6d3940239ee9eb8bda9c5a502868e91bc6c4cec5fd38477457e6ded5f7ed94778d15ff07945d70ec12e25cdad5d617a5f83d53b12e424645130cd141534e3c70993b4ea6d2cc57413d59e64bf80d11786af69f860f5b5a436d43fb0630b25c121c5f3dde973df17cd21f351f116eb036ddcd160e3e8fecbb7f25612976bb1e075e2fbdd95e5dac82c3c67ccc96c54fc5835&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.14807143Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0, Name: checkpoints/128/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/128/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b62f581360a2c2d12526f38872b4fdc79fa7543809d35ce672bda8c267cd00a79e7f0f65c75e4f70fb49486e3e16e5dc839c8fbddde2ac31f7c3874f210dc5264c6ee1e4902b134ac2215b47cd818cdbdd7a8352253dd06bd478964714f827afb7c417f06937c7dbc5f351622586b6d13a892b410b2d6f58995c680f3d2081609fada4070b1fc1b0350ef20f397e0215d5beab7dde7e089957d76394529d0c93d2fbcd43523ec97d9c1eb99986625cd14601cf7eb0092f311ed642f370c96cf5d3b455d1eb8f0c73fb351cae5597be773550d1541e2368c8d6d24816f04b142a0b850ab8fd59d1087626d3c08311984a16770844805426e9c650a16a39915c20&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.148149887Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default, Name: checkpoints/1344/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1344/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7a77f4e02b2a877a38cb4f65f37c4870b7597a3a645104d374494e8297d9f212bac42fe92bdc7bc09ddac316b1a79517c350968a66702ae72b0bca2413ab9dd1a6fe9b2909e65737acbe1d45a3b2bc392922e2bf2c84bf6959ef60119b73742ad4fded1f6ddb052b07b567a24ba29058610e62b8ddeec12763d726d5e07d53a77889dfd4c9757f10a326976f0c6980db4f7e3858fb0216a8fd3b040e97273a44024e30f0c716e7c48afd735d076bb3a61e8b9b0153991d471e1e8bc16fbd6130a47604f08ff6691bd9f654d74ed14f93c9e5d295243567fb1a51b6caf8e9f641d6735f66444eedf71404425ba1b5bae83ab620637814e8a5c21b4d14e8e49d6c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.148171005Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/array_metadatas, Name: checkpoints/3264/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3264/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=61f681f3d13cc82ff357f4d0bd141343ccbc6427478fe9bdc188483826852dc0aaaea815bf5d7c75189382bef98289bdd8d70f9ae0cfa62ce603d5365968f5bbf5784cdf7e44d2024cdee1698232e34d00ebb9a220f270a1b5109c77ec13e67798e1b740450e2df8b82944d8a8889a0f83dc413582005d8f9edc307480c3bb82a9eebb1fc4bb0cd338a4f794b97eebf380daa2e3c89f7d6e58c6f26b5382c92b2a187d71a119c7952728d15870c7124459f42eea45318815681d526f5664904097d12e63ca5246c5bb3c7fcbddf3e9dd73382c8905eec33f7ba46802abf0bf19ef948d88f89cab204b891dc3be288fe9e7964575de935f19c740a5de6381ec1a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.147515819Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default, Name: checkpoints/2688/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2688/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4493579f4751228c0e543cc07194292ccdebb25b4749553c5e9065bef7b3e2aeb98d19abfdf7fcd0bd39d709092b0e853fb86559c7990e26bd48085907bc992baf4c45e59be83a812f6dcba602a1b16b6f42003c5ff29c4edc5b02e955eb2684db9fb93bfe8e7dcd08f6c912858d18fbcd003e94b92a4e159d65006da35f5ee9904d8b6aab053fb0ad57fb04d0f495983beed5e06075d6821f0e38fbf8bc6901f861e700e6dc9ef1080ec95db57c3de20f03e93fbe49023d6b26765cf347e37e340387178d0bf1294c10a36a8361c104991997953796d5197210ae2891e39135d5d026b444bef5319655efee7ace04a9bc05281f8957a74185ab79aa2fb83df9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.147669745Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/array_metadatas, Name: checkpoints/5376/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5376/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=1bca2650105f77afcee7fc0273642eff23c83b0329b319c60a32c7f0de3bc9ab31f280d6fe7c2415042ef987dc71bfaa7546d6dcd98b8e32432b72b836b008421ccb553c034b696bb6d9948958b9da7b812506b552cbdaad1c14f3229bfe9ec47c88b93a6ab689bfe8c3634bab3012e42180cf11a4e98e07d907877f1a86cca92a90f6c11559b6e0991b980e1ca56d806de9085eb3942dc0799703666bd91ef720c1a10d4a1dd167e7f54b647cfbeac7d8245e7019b31f887e7cc2ac44e55aedf73bff35f0e0ff0aa58db034807a7e376f201158ddd2ea98f00b4120cc1c41cfdbea0ffeef48c216caee76d9572a837627d78f25cac561160f12edfd1dcd10f4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.148223933Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d, Name: checkpoints/5248/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5248/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=aa5f7475d7a3cdaf676e86576a4fc1a0a20005adbe0a9e7d48325262ba10e8b6faedd0ad02309e261ef9b1c1b00a7bcfd682d0369370b6d35a8b11433290979bab32fbb62cb746d101bb5f869218f1da4fb9c9c0f6d5030f7b73650b944c8244ca9f1a9c0808375aa6e03d7a0b07e3e07816cb1fa9b35185d2b310908a35ee59514af2e6cf185f78725cef3c61074ab585840a552583978f3337ec1f6185310e87002b63635ea39916daab2a8b3a113e89e4aa8f1e0f20fad40c0137fd7e03bbc8b72960397a82a85a067b4424967170ad7817bd9fda89a6b49c7ee7620cee4ad50b2d125ccf688aac482c69ce6189c8f38c1acf28f0c329d9247b02ad49bbb7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.1482694Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints, Name: checkpoints, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=23c02d683e5492e4a3b7b3dc4d666d2af0b4d9311903d9c713ab0bac399667b0ce463d097c5c89ff93f0617578e2213c7526330c6b976c53150a64c0d42c5fe27e69e0cc8764e98d89d5da42504b36fd5b17861226d4914f036b272be4522c50ce16264de5ba9c642adbf1bd248280a234374711f477da471270cd88dcac14df08d507146a8ec50facc6cb4c657bd68be0c9faac0c14dbee240e7796fbb04626a9648fee3b2ec0a6f792d141db4b87a244997175c3ee029fa597a53149c7d07b9569133cb8a59ed86f934258e6a65ec115f913964699ecd842f0bb7852c11cc2469f43dff018c196dc225f793dfa2b41ab4da5d0a9ff256ec4373bd19366c525&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.14829446Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default, Name: checkpoints/4672/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4672/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5ad4ad16fa617fa5a287b4cbb046866ea674ba2263cf1a966d17274f6acd459b73b314cf5b91e240cc620c61319f866a085c53595b2ad56d2a10721d919d26e7677d8bccda85dd1bf604eeae7c028924eb69978985ad7c2e92279f7545a9b47a990617439591027284270d724f3d3d85de6596447326550abb43d3a99f51a7d935f45a08cc25e28ba9e60fbbe9ecfff17daf7ba989ab04f7671423b4210710bafe5279466cfe5c55e79e5cf983f89ff5586ba6c0093bfca95d8310d87e957d238ce6517fc09f1b8a62b755c530d50ea3ea9ddb65f2a8bb663ef368923f142d0d7afdb8f9a43bbadb8292da7af64c772ddaaefe4ce6ddc676fe45dcce99253134&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.14833789Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d, Name: checkpoints/4544/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4544/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=788e16d6b1b9da9fd18b7d12bbb3827b860570ebd4abcf0ae5e0c832c80e8a0a94541721bba1c0e452b2ebac386314dc9f6f03428f6949ee0b60da0222d14df1014cd5fb4b65ee62eb1644012fe0a9304eeb82b9bcf1a45684623b74fad24d2eb6053289be050527d973e1d5f90d67846a03387b8d003f7a83369451e3397e6c4045a1b65fc90bf9055c8b76d52eeaba20f656a5f917d122a6ef4f22ed73ef43ec276b5b4969cc08956f00f98cb63df36e9d9c9f5a59d23bb7493b6987f4759ed71b6329c7fbf7994876d67b99fd623fface2929a61e4689da84cc3711207b8b19c0e943b6ac923df84618a03cdd9a16942f3e0c87b4bf86b7fd9307a7be0ec2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.148371718Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default, Name: checkpoints/4288/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4288/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=6e56e110548ea4a5ad5dd3341cb7766275bc5895fbfde5067f2318fdb03d5fd244308a155e42d633d6f1c54223315bb74b4963df344c75826f13b42b54c609fc4321ecd85cdc29a882c7f5a59f635c69bc12fea35cd73c870bf1438b5d2be61c1f6c6b339129cc56812f1b3dbe43e0c42131af854303b508af02d62237e71cee97ffaa1bf081ce7dac2e52d103f22e0ea8b4be4bbd16f9e0a43b32aea14084fae588c8c57b68fbff5fa2dbc383313c258a7f68dac526ea8daa7299e26215c4a554cebee211b6cefa388822e68e331fc81e9b621934cb3e952405fa5a5a1836769e17b00fd9dae34d10c06a255c4b9ee63ef355cf4530adef748eaa3e576c33e3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.147741612Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/d, Name: checkpoints/5504/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5504/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=1abf9ad3bf0f338ae6dd00407064087686dc9646df4c9fe85f5139b73d36a76d053cdc47021467c46d53ba62755499750e54548cc3021401bfc7819641a57c6d348b10d90d875be46d44bbb3ef0ee0ff879f208e40cf269519a4c52d5125a85f873355e7f3b7a398b2be311f6bb145d933bd06226811f9480e1ddc9339c31772f0aa699e71aeed9ce2e9fb2e60675d04bf152a8c7be78731924b08c6598dba000d550ad31166dabeb6c84e7a6c823fc05fbe98681517d5e993517a5970e0a8b7a8e3d21aa4887d798f7e10951a8d0eb1dd3c3cbed111818a4e6886b43fd9f4cc67836a8aae946a07607d9050ce7ab6a4e20acb4efe09d33c7dd59027c25de0ca&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.148475534Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d, Name: checkpoints/384/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/384/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a44fad663112b54ef690a41bcb561cb2edc233c8f9f9dd4f9e7dcf553ee0340cb329b61683269f40f0a5d3e03c4854001768d7896ae25053be11ae544886a5717861b9e025e3b7c84d4a0f2af669b816f1f0ce7c457ee8503e7137c9c07ab4dad34e0ba21e3daa3ee8b00f8ab9cad21470dc3b51ef4eb3bb52973ebd9f3722eade2cea95a99d835e5d8e34084ffc08e60f2b4df836c2784fe0cc3e2bc17fb3692b8683a80817e4a5f785349c244eda52d28bf24d2ef127c6dc0f140a7322123845c8df03f819afe0c6b36553167f7aeb3106936e31de97f7eb0aad7c94187278ca3148722e3b5963794a50860ae0e6d1d5acf3c82f0bf5ee30ea48391aae3699&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.148513303Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0, Name: checkpoints/1984/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1984/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=a91e27addf13841066327c4cca3abd3e0ac6c0f01e0d2b1932521cdc322c11b857d7f979090f2e39d8b56d2a37ad469b510d7f93aa51a6f23b686f840a6007e38fa822d4fd7ba09177eda869f2aaf1b3fdd7d644f9eac64558317dd0ab4b69f30494580f718157c50357471d86e05b5a4f404765ebef34a81ec1cb973bab66ae3c93284b267c56cf0ed10a05c40dda2ba8ddb502877456c1da98ca24a087c1917f2a664ba7ef41bf8e3209f8f56ea7e2d6d44a020f540ac4fb999893c011659a354ef0c5dcd10d4612c2e0706393d06e375de75ae2f0a05ea42c5e99cb858ec82643aa0018b08a8d6bca3ac7f77fbc7130541fea7ed3a1f0820302f85cbc5f04&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.148594564Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/array_metadatas, Name: checkpoints/3456/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3456/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a3dab96ba3b73685b34cc6f1f9a5af8516606796afe4ee35aac13556814ddb0e5ca9252af2a35d639f12e6fbe6439d82148a8eabd5c54421be3ed512bdf3c7de2abb893dbae88be98d56617d3c7787f2cb34e3af7e0c7dc652b3658d88c3ef06ec700b36572252987e6621da6137f42361ae006e346616b2837ebf3f5622dc387bb8e40c86f7f0b16a4a24aa7358b976aea4100baf4c703fe6b99041c9d0392af54320420f5622af46077cb3d03e9e3c73050ab08452ddbc98e731eea929823be5939aef8155909eb67ccf255d41121d6d408ed4d68c8a30930e5e241b6a64dc3639d2d8d4f46650efdb2fab006a7d8ec6e2739fea0e93e9ac849abab5be4d73&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.148623019Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992, Name: checkpoints/4992, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4992?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=117a822b94e5e97b049fa50220a7965eb51a419ce954a568d82a7ef6db3364fed367788534ada280652e81f850012fd471c731fa61745a22078a66ef1cdd4e03da6700c7c2b1ec19e5628645474acc6406d4f6a8b1de6099b2458fea587ac6692e7620305a072ae77ba6c4eee657a4af87f70cc3e4d3ba95b6d74bbfb987c3da2ab159c9efc2cfac0d7caa050be5a0f4f727f9614e2c0d2668ff420a94323dcba418ec60e9a3ca2481226c855ea46cf250efc868295a3caa82ba8ae0061c18dd974baf31b97afde7c47abc0331cc62ce604612278e0fc20784f40201ec66a519d742ed5e979a79737a123ec4131b3b9ae8caeb2b2fafedcd77a00e25c7434d52&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.148638174Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160, Name: checkpoints/4160, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4160?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=30c4176ad1afe9a85ea2846c7bd454565934e6ecdea4acfd7aa48daa341692a6388bca1d5112891a48faab899aebd35fa43357bc588e633d5d84aeee1837c5d244e882c375873ba1dc9b2a87ce0ecaeb86e945d262612f4806e576a7bf4258214c6e5168b168c8512ae4d9a64af37ee2f03d78866537a78d5d239c9e798b5101d1dfc140d28057fd1205b6bc939ec7d3f15e50afd2d16212a9bca2def3b2b453ac2c6d8d03811718d5e9ea9736ce38b87d540095cfb1edca409991aba3784c260e1818347f32d30c4ea1d3e0520dcd9db0f25cf23acaa469e4f51d4d4fb3a9e3ea5faf33f8a825b2b68c1b9670aa2098fb82f95a2a35a0b32e7abc61e22eb9d2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.148664072Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648, Name: checkpoints/3648, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3648?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=312a6237a07440eca89d4d2b04a4860b423c7c1fef8403db3312014ce71d067104937266931a807b18c7e1a61e10b0f6cd551998e6dcbec807f5e4205fb3a1aab63fd45b3bf94c86195794f6414db97738e582a1870b5de20c0fee24221859e1ac029ad8127d143f79c07683f23685b2771b8aa50c1fde4fcf69960889969bc8f5f69885f1ed23fa497b1a47f7b73dac34abba0b1b85fa8eb73f04fa598e84cce1592c88c6b66fc1543141288937ebcd3c3c0758614905f4a60bd603f38748d9cc759b08d410d976477b18c33d6c1e5ea1140cdba07fc28cc85555bdbdebb11c45445d3cc55f3ea96ed29bc33936baf91cf83dc0396f19ef4c74ca7d29c65384&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.14731665Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512, Name: checkpoints/512, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/512?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=3f67e8ae6506ab9486e276b787fe241f8d25ee75f9ee067dfbd5e215a9a2c6bdbf32795b508ddb16aad9645df76d02319691824ea01568c5d293e33fb960a0c3ef6949bbc7698a0b5cac3eaf1aafd86dbb6b90cd6bcbc7668c40d9d6ad857b90a52409a76fe24eb771fca0931604b30c08b7f434422bcdf857ee24285a2eaefc4e55283fb5dc09b5db85c7f9794a29a7efb6ccf201febea22aa1b9c99594dc688a9002bb9db9a443261df1d6de1a8de79455eaf066b01e7cf5f8ca07bfbb7d9044e748ad4b51931d98b3ca807796c1cf0dc9201d05b32c9a6d12a6638316d4f7c0b9d243f4fb9f86a8f16400302fb0a9f4a0595e210d8387c1ec6fe0599bb4b7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.148687778Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default, Name: checkpoints/1856/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1856/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b6ff204f21a208d478009481c72d40853374dc93e143e1c00121cb3059857cea30f39b7aa3a3a4ce88e7677631e0d63145730eeba12f49ff9e7ed5442d8dc9fd6c3723e2e03049ef3d878a072ddf7fdad5cc0e3d0a0927b9a5224ea41dcb202104d8e2d58d5bc10af2ae9e093a6e28203a6c15ca96dd940411c64cbd669ca8e3639be1dbc5c9e9d1ffd3ed03d5b95c62015a7bd852a40c386b269da7133dc402da8bc66e5c1f383df7168730206b7544d686eea5636b3201587903b1091f39bc0d91e253cb83afde5ce1b552b0cf24d913941e863364dd91b117974fce218e7935b069b61edb5263b283ec5f9dd80e1372b4f3dd59668ef61fefde4329090832&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.148750837Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0, Name: checkpoints/3712/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3712/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=37f43e51f8cbaa47d9781a31925a670e7c00631fff0f600919f60aeded1efe4d1dc0dca394270eee1c1e5b04da3f4c5d36066f3b8f5ecdcfcf6d05b8ea582b732f97ec1f872aa5ffccb402cb3c499b974a00ae448fdd9cfc08ade4da47ac58cb11b1b1c18ca07afb767a800e3d3a7dd6b6da098fec01732542bc6c2b7670c825926a379bf71a3fcd80e8fdc6689f00f45c08ed2073cbd68089ef777724c3388b3e926ed78443ec8770f5c44e212dd849e34e50c80c888d9855245376417c9f48d6fe43decbef91cb80858fba954c3a301d94d75ef58f6329fd9569175220fbb4513f0221faecc896ff1df2d9f6ef8b8043fb5a543fe6e477fe6c5d16c1e75fc5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.148813536Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0, Name: checkpoints/4288/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4288/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=48876adbe338d3e605157c211b312ebf993dc77fc5a8e9dbe1952c8cab9aa6258b61e4fe47579a526b7ac300f3939286a8a439edb8d6829dc18ee47bba2abb4b0d6269a4887e8d2f11401a8deac0bdbc87dc20e040a9aee5b10bb7dddc36f3a3f88e50a036f28015583df6d0469c38490f0962ab0978475f8d336f9df80afc2def04dde7cd5c4125e8251af6b355b0b8f03b07f2c1c8ffd0c5bb6fe833ff938f6b149f10aa839c51dde5c4b8d4d1e1e6de1bca7b33ef0d339479c9ddf9ad5467c9702d225790083cc3520803610788cd471f24b0bbffbb5664e1e276ae34765cc562ece8cbaa8e8827f4e500b0130f42ed1e9928c7bea3ae89b9453538548c78&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.14761431Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d, Name: checkpoints/2176/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2176/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b6b784fb6deacbb5c6234dfb4c93a046e2603762b25ca9213fbc21e65426a2984f000db538c536deffbb1e4d3665d7c901daa1c1cd5f9453d406e3b500ccebbb3a325952667c3884eaf22837b913fbf84ab5a92e5f0a631edcf9d8ee57d4bd282732471341f5c64e79c80ae2cf2167da9ea6e828057309a071e892597bef28cf41eae00507cbac26a9e54f481e880ae9a4a4912efad85f1aa42fea3842cbd456282c452315150ba5627432c4d09101df8c54aa191eac92174fd681b62bd9a691c73923d1809a89647b721eb3094f00ee1ab7ff6834fc8bcd156aa731cda0afe133462fd6ba8baa20e3ad78cf180bcdde658021ae71ab273010ad3034c406449c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.14755305Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/array_metadatas, Name: checkpoints/1920/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1920/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=767d73eb632cf3398f1792527ddfa80126d7d4e6ff224275d052ffe2f456c89de40bbabcfebe4e7d37eafba98b702123d8ba52f25f224fe35b232c9cf3b1b44e685e083fbe80785df1462c3b2228f9185f86af501ff0cfdbfc64adbe13507de074111223d745acfc1869aee98abe2801c10cafe81e0f3712b06a0774e014d3896a85abe6eace8ae07d25a4dc4987dae243ad2d657d80091793ec35484d23b857aabe266a102529880207afe3122706d614831f75adfb93d463d8e6bb6c1562049c9d3bc83e64fa6958292f8dff8efb126c7bb240dfc19612221be21303d40f9f43e5ef06f975cdf06d128704d739c7ffc1c829a0b4e0d6efa84a29d844fd5280&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.148275721Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default, Name: checkpoints/2368/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2368/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=0dc0f558f5fcef5ad8d581f5a18a4492a935f586c2ce17f53ca64e8bd8761a2800cf11de368dd7c1cf4960301063803f5e0210ae36c957c30e34e71d5b5debe42dcaac2e25d14d14f43c52c5c21538402439e891f377cb9333444e7036f1ffc04cb375e45a91a7dd9e3c1d827e773be09c7c41b6cb6587541c0c28aeb01d3455af0f86f525efb4c727aa7c908c1e23b60b7681a00faa4bd3674bb593f51f648d17fce63ab376a5ff22e05c6c1314d49b6b797d6f09d8465e2302ae662346d0a7f787aa9f4174d12967de3b96323f7612354b4a14d8810f39b228988ffa59f1eba1af508f717ed4d2adc87840e5c6dc30ab83cafe4b8a8949dc7166e452ad28c0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.148986613Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/d, Name: checkpoints/2368/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2368/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4010513b5f812eb9d31e3799e9d6f296f4b9c4fc9f0ffa7aa0a0e4f1324c8b5967658b9221d5c933aec1ef104d36439538c68e9b107a527e3bcdc29136b066129ce8de9f297f4e080863685660cf8c967e8912612ef16b43fc27cf94ebbd99eea1a3e9c5a8d8ae45322ef2453696a0eecc9b7139d6a7829ab9757691116371f6dde0b14f27bad7bac3933df31e2ce075b8a0b966c22ab56dd4176b9a8ca323acaf4bb50de38985b5ea73caa7a271eaac3d3c64871089ea29c8505ff1267e023588f7b15089dfc415c7797751e69335d7bb3d8caba21e071bc9fe4dc9a724c8941a3f44bd48ac47c806b83f5f8e05e5c7ea86e588ac6e19083a36bf514f4084b6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.149037901Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0, Name: checkpoints/4992/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4992/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=c08dbf2481002346895017312b628e765c7018f6c5cb2b45f0e85d14262ec0bc122471bcbf6985782ec994daadaf2f452472423f20ed9d4eabf4e56cc3524680d9693382749d8607548517fe52ff26225fbbb991626ad773bccf3c18015f83bc05863f5f898aca5b483695d6d0d728938294238c64d1bf3547c6d0249590ebcdc8224b394c08893a96273b74bbcf3e77028f3447bd3f7d6bbc49aacc96a10a0dfce217d0db76362ef8b074a1ace83e26420cd8d66ca6ba7f16eddef9875afb099be411990d3062fe134aea979a4244378104dfa8d298cbee5e2d7dc990f6f2965e37b330136844c6bf29e2073493df5c09a3d23850cd725b652be899e882038a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.149107254Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default, Name: checkpoints/6080/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/6080/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=77572cab78d6b46a970888742cf5febf2d7d60a7d600429c1ac953ebed82b7e0ab8f076804ae0cfcd676bac7ed5a41bcba6c26d8b42720fbeb9031ada14b25d29b011907e68a6e10fcf2b52c4241550068f78bf2c983f06af640e5fe2ac118ee715df65bf81308f9bb9a58d788c3900a774320e82b47fea7ab2afd1cbfd5f345991661de9bca5b2a5554dfda87dd96f85f95ea02c2b30a440b0db7e16e4901e88db00f057b1e2c57db2188ede88fa701556523b044ec9fbc5da4d59c61566bdcbb97760bc97a48dd2c12babe64f335724e9eb1c5523b1e36e61194ed6485ebb58a2a14f94687af58d4e27e1ae6db3b4ef51b2a2209fb32ab892e931ef4fdbf82&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.149199381Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0, Name: checkpoints/5120/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5120/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7975715d96e3d8bfab4a336fa6f0e4c1f3a13ba1a64c7349117a34f94f11e2514926978dc5590011f953fb5405cd108b6b57a5271d1f880fa5ab49ca8503f92fbbbe83828012bac05caccd961b1d7750b50c8c45722546044313d876c278f17d0ba93d73e83b5a8f8f39dd1adca009aa56a4650a8ee9bd0e61ce1a13c4f14df76e0e8c165c3e570a8c71d5e1ff00b0ef979f18a9574f955cf50605123b5032dc675c5cc765b1192e59ac2f9de226748515d2b66ecebe399cc2ef0ca67640761c955c307f1dcdde85283f071e739f25af080d8b4e611cc3a92266990df86a87d7cb8f7ab0d626bb855324ba7dbfd0e037525344ca81e9c9cd75a9e0fddc0add37&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.149264339Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/d, Name: checkpoints/5824/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5824/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=02318823b71a9b6458257a1e7f4571e2b2a2268cfe7cef9a58b0c2f3b595539a6d70a7e099f1016d46e23c24289951b9ac8d5239a54550b00a3b17f11b2ceeab2c059faa42d02173965762bd43910d23ca7a70a6b7e9655d024f0cb4acf46392453d67fdf762e3c6b13426bca80ae3a047f468073ca34f4a7448f1a9e18300408d69477cc3577202236de31ee4e0d2ed9329b8bb6b2dc9d854716f1f6fe44a9125c4f25eb890163bd1b5b8b20a4236b523535fec204e085c2ad15426d02e5c4e38bb1933c76ceadc31712480e9695ce33e846cded701245d2201ec289bb6ff1ea3d7aa7db41567e0f3d1d6c6b903f9cb8d337db68eb861f3e8190a69768b41b3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.149303044Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/d, Name: checkpoints/128/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/128/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=88ed2e9a46707f2cd9f6dc100b6aaeebf2669db2681055d05a239557b734d283a73b599d80a48d4deb26955199c0b943d7aea1747b60c659f9833a5aad6567f104b893263d33a6342c26c1154586ba593667cdbcdab8d415b35aa28f57b09090da71eb5dc66b1b762d5f8e8f27e09ea23af9d75c1440a61e145ca9ba8cb729fae4e0356e40d57ec5235ee640bf85e932c20e1e00e3adf25c2d4426cd920487bae8e7a01a98b4453b3b80cb1af7c328809e4a7533b38fab13c3db2f81fb80e1a277ee9d8b178da5d985e4bce46298310f77a1d352d38d1cd86c8cbdff2e36f641fbf8a150d60adba54de8959ea9d0b273df4e9c0d8da40a281390cae99cf4bf25&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.149344923Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0, Name: checkpoints/5888/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5888/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=355cb7e06f8fc9d7b0e80507fcad06b0daf55a927a330f899926a1cee1a7ca27d713630041d5acca3381f0a6b95af94a8eb0c91568a579e592925390f963f5cce970a59a682d49333cb23ba559b01790fd7ef4e4d2cf5509b6c539964e7bd61a468f045080aae32c2cd8215bf59fa9ec46a9a3c0fa92198ebd66246e4229289e6b0de5a89983a9e2190c65267cd1cc4275e016c86f0989f3f1dac535acfb6b6ba96ef9d80c4b04851bdeed4f5679119119f44bf48046a661f0bf653781956d681b3bf7e97803c18c27c7fdc6a334f031a523236f3c9d5e6fb9af484212b5fe18e6990fab3fc5e18de08251cd1663d1349bf5e67a6c406e17c708cc3b1c996e8c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.149359675Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032, Name: checkpoints/4032, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4032?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=bd9ab57162c26c3445a640787fb1b57194ca55820b3ac233f68dc0ca0e3dadd9f8f430fedb20d68262e4de172fc7dc845fa870498d2f610a6beb4c1b5b8ce69934c2843a6d2f4aade089c93623a91d42351df7cda9598768e7a7f94203e6b199b3606a226d2782feef8abd3911ec322d8a72fcf03b8a18069d5612c874ac927ccbdefb6e8ec91bf093268d37a31b5002fdec3b1f7422ad527a4318c463ff6cacc73864b091f65205cb202460851b71fd5b19d0dfa029e17be32b73e26f12e823f6b355f98081a0166e16b8e5177e5990a9b4cbcfeb6b0a7fd3c38c0fd02c4218fba486b90aea70a7cabec03518c0b26cf86081f3363b745062287f7ad6ca858b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.149401994Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/d, Name: checkpoints/3200/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3200/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=592228ad67b2e8eafc024343e02aaee9076656d9ec1bb7c12e20c867f63fab1fd2e09f8dc9209410b49dd55c2993cd23478c0a56f1c000666c4282b8c366951481a7100c31a77b91948869826fbc4b5980529878c05456e80b5e892f675bf29ec4e6777e4aa8706b229a7dbcf90726ecbb06b59bdc5d58425d71817d7f57f538efe80aed5f4ccd8a6b0a8fbfff9c298c3120cf26f1e7b5d0c6cc10e0cd89bcbe72ed282125dc91e6364f15c2c79aedbeb1cb8c3b2c56fb76964a37eb0acd8eb1c92779b58c22fbfa5635b27cfa10abb685b2a5a1fb7d938b6ae6ce6b03fa5a72ddb77dc595efb2c91af78493c04519881fcf362d7f01a483b9535117f25c155a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.149419929Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d, Name: checkpoints/4416/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4416/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=21ac6a9acd0d9a3294d1331a61be6e55cdfa421c6c1fbbc222bf79bdf8446d99de1bd98e81e33f5651c8ef917216acf24541cce030ce1dd5b4d3f51f0185f603aeea046122efe9538399e94f8e5d6dfe537984e58373cbf061740cc3b317b2460ec38c17b6b44ceb0f0fd9b422931c297d2165ebdbdf86aab8df57038b2b03169603bbf153afa8b8f7fccb7cabbfa0dc0355e1d506d42a07bc8ab31dfc9efc507fd77381569e3b2ee6083ae9c2395400811f652c74bb382fa53ef0fae7bc3dde149078d73f700390915a261194e30ded93a24ffaf6b0daf2679aef4bd2256dbcd60202c3ebc2b3b7d420af6bc744cfd747707d03270f53b6d26c4a0ffd1e1dc4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.149432852Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/array_metadatas, Name: checkpoints/4928/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4928/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=02836619432c8dc693275ada5fa65170a62bc6a12d8dfeb493ff15c7d86d8d9d216604ec663a823c074e6527b1953b0928955cb2e4675f14950313d7edef31dee493c8b15d032eebbc1cdfbef760e4c8c9d8d4fb46a674e035c85b43214f4d363272d106f25b1456297dcd73088e7e47819fb9dad7fdf07da02cf709b0483bfda1cd555a32c3d4e109ec1a7e4b72288938ad6e5c16ec1381aacc79c011ad44fdbadd6ede98170e7c7c7804957ce3acc576c1783bba63f27e18419d79405a00d3103b2b5f073d217721697df1433d0c8bd1f34b76ae51246da8849e86bcbf8067dbcc1f88c2ffa50561a28e1768d0c61b336d545dc15bedc43ac59773ca84363b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.149458034Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304, Name: checkpoints/2304, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2304?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3cd6a6f7fef40f2f9990a72266dbb43a9ad7c5f204e821e880abad0712abe6d6e9374bfdcc72de36e65a4f6f87250a96fd34e83d990ef4b713be333034cc9832cffd3d0bb2cc23deeef3becaa55d275130dcb1a48388cf3b7b107ab1fcad3a0375ea5f3faeced50ec103227d006e25fe34895376cd06ed0e600eb8a299fdd710c21a3e95983f3c11fab6b948609d43a70933808b8823add44412471e92b16633da1001fd5d54095de6f496d3de7b15718079bdce3187e3762c960d2d88d24c7162a3be2f8cbf3a1efd5400daeba2363530f3e7df736bfe09e31907daa9651a92f71b2ec819e86791c7c1c5f4d60322ee623262036280d62f446fc3831461c889&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.149502642Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944, Name: checkpoints/2944, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2944?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9e11019f07838f5f056655cf51f3791179d5910706638f4f1af46c6ef921ede5f39d3e057133d53a86064c01170a3d228249806e0e1f4515b6ef3d0949360a49e763ca296b793f446b0c528d7ed4040628915480e7e8ad3a5d72f5799eb6ed960cd65f1377baac18d702eab1a00809201ee91218c04176ff6ed04379d9296643a84fa2a19d92e9c67f5ad3a8245a096e63c9f76dfe531cc0bf6f6fe2da9a58e7a5d4bbe0f6abba89567ad0923bf3e39f67dc99871c69bb0f68b88697d935b9b0e8205d32f6419f88039fefb211fc3e4304a32ab7c38f8bd9745a2973dcc90edc5df1329f7797189cb293e817de94403e2be3c4cc005224a09af5f37a09b99999&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.149563633Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d, Name: checkpoints/5952/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5952/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9f816b67ab6eaee60f4e54c934583dfd72c95eba6a14b225576b5b76e3db070de81ba0fdb7236aa940fcbbe66ea3e6f4307b4de6066a0a48cd137a57a6a25d93d63513773c1c555d1d9000a3c45fa41f643ff70a7eae49e004be187c8d1a175c4e0866864cd01dded852f1abe14b292b0058af13a9499bfaec184ba1010f8d89c50843f4fbb77d7772223890e8394bf6e58759318025919de8616fe4fd38fbf9757ac21764d64b5925445c82846d5bc374aa10288c29b471aaa84bc82870449b1ccc9b1688a9225593b0c32bd241b86c5106b35b83e1e3ff3c2d8c22ae695c05aa5220c39fe9c0ecd7c405d14a2d7d356f20855cb8de3196aff5d06c7f27617e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.149663535Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64, Name: checkpoints/64, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/64?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=69d25c3a8b797e1f1f64ae6488117569bd19dd26fe4bd2d8d19e10194ec44d081bf5bd748ab45d9145543e7dc239b1b5317f4da04dbb1203cd6b44870559a108e1f6298a0fa2812f78ae2d9d8417439ffb3c975226e8104f281a8e0132a648820188d61b8397b187a02577e11e243d132f0c7e63bb43c877f3698a40bd51a3a58220c2d557992f5fa9a1a1c7e7d3eb5883a20cdb663f254777c43bc4f7b8b3b86a906a0ef2d591ce74a1a58689d9844750906dea0f3db07003b9e0c97219742dbfecfb192debc50a11a86f4aaa912bfc59b73fbea4fb8751369c22d5032efbe191f41f67bbbb774efe2bcced80d68fef22affab04a356c35f746a79be75e8a32&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.14919272Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/array_metadatas, Name: checkpoints/2560/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2560/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=11944a4b35ddb37103abe4a1bf96d9a048dbc9841cc62f575f500f0a9bdde5f866b6de3cacd6ac0e31bcb2f003aad26189e85ab98fe29e024021368069bfddf7e9f47d087e3752c81fc11a654c4bd3ac1642e9622975a155813dfe2420879642b9e638563b125bb49bc0a66f7e968939b2cdc99e74f908fab280f81f361af579b76e020899f327318926f665b27534a143edf0d74b5e6eb27579812b8c1000c3061991e7605433d793b69dbfc887944c550cba077a316429d64982cc49b9667a09b6758d52a021030d97e5727df6d2c4b94e1c3c06eaf33c320c30789c721666bfee325dee026709cbe482001886160dae88c868308335d50369305ef839b591&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.149759876Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d, Name: checkpoints/5504/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5504/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=16865fbdf04ff14031a65769d172a076690ea0573bb6cc4e3df2ccc2a22cb76ab3dba7e8b302728af165a5f9597e1a222696805867e86df9d96730959984719cb4ef3d2b25280686583b0158f933d9adbdedb088efc2cf83215bb3078be711f70667a4f83185257c2555b87f69c26266f1699c8144c0efa4f2987d0860898a23fe68d6827d5bcd224cd6e672bfb420381ec82e2c77dd2759563bba51702ef46cf9f3aebe7d5f81b6d1227e4c60d04754757f9243b9f4de2504d58e6542c440db87920cdf866c5931d6fea22482b7723b2e4f290a5cf41e8944fefb1adaeaf1471b6c18e65776fbffa4f4fc986a12533dfcc1691d67c074dafc5e454a3fb876cc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.14843574Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default, Name: checkpoints/2304/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2304/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5d16e6ef855ea668e5e84fc59ee5d43c35fcd0ff03e74c8b08dd0bd08288af051877dcd858721f2e955388ce70c18beab23aaf8c1005afee889eeeb88a865071e2466969317038646659ef42d841d2bac918863be36b78375f4df350e93a441b9c41d0d3f9a949605d8d2ddfea3d87c240f6dfe662a45a54ae0be5c3a15173a5db2599e1602754a06ccc75efefe63c1fa1806ec61c77bd76c05381c2750ad4306adfed347c852857042f64a8926571f1a7ccd4ea3e7dd56de2508d6ddcdf7d529ed61a5b85f5a827b47aa9107012a1e6365e7d95d87ca92b00303f9733320ffbe2da33742c1dbe79dbbe2355066f5591d71b894bc7587908b217651955e3902b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.149781456Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/d, Name: checkpoints/2624/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2624/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=796258efa96aee81451dedd09bde01448a5a897514a2db0281e4342d70c9c1ccaae4249abd30c591717f7eb66d1611828b60ff5b42213bbcfd4a7e548823217e9b5372e058e76ba3881880df8ca33dfae9ea69967d5118488a4dacc7522f992d5c8994b75d5d727ea6ee0441c979c30e6cea74f1ff974590720fe72f88a805a4c13cdcf6ab04ec6d1ec67e5390788fed7af7e4100846f10837b39f28424db94724ba3b482b581d724cf642156b7cb0d34dd495aeb04e5abda06789a3e3c07e9384c44af823eb12afb597742c4682bd78e95f4a315ce91f0218784ab1efeb5bb37f632f86d293baad1f16f430105fac4f90d6e6793ec61067726d31c88b33289e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.149846495Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/array_metadatas, Name: checkpoints/512/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/512/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5a1ee5265159668c0c8b026ad0af4c8759eaee18e780a0cee9d5648cdf672578aa8515e873a2a3a044d2f151a7f7330ed119b672140d595dda00fe570dff1b254d940eadf4d85161e7ca2e94f7d8d7c94c924595cb070456990ae5d701fc8b4bb83e188e07b39fe267910d944ce0d177d137a4596783d5f9fbc2c9f5c7112e7b038d631c29fecc35fef0dfb57862aa09987424353c694dacb790decd737f40a29a47984ddce21a061039df91a6015cf278cab71785d8de1df44a5b7330ccc2411ca7c2f4a0d1df1c7a7ec75ad499019c3acf7321d471b499b8b88c0505d522da154f0b4abf3826fdc8970bbd0201805abbfa0a580e12f03b5913438e74748762&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.14860897Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192, Name: checkpoints/192, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/192?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9d8d03c27dc6cb557a532037574ae7ee2ee101e11fb391a11915018774e419969243195934d0b67752e91321a799d4a0b73431dff5720ba8a768e02d10c06932f138d4624d218d98d1fc804ab43e68f5625554305802f8675f5166e06c1f2e1fb2374870eff448eb1ceb71eaf6dd542383c6ac173a35d6f2a3874c9411b6ad6173feddaf131add4ede191a449b8888dfab893ab7dc688ec72894bce1b3017f5ae8a468ce4590fcbb2fc039a46596093e742e8f13a5022669e63d874ac8093aae361e2d4fb745e02803697fa72a48143d0d02896103419606a7d8af6c45eb8e6e4b6df47050d850e053e18cabbf0cfd6a6ec31ca9c54080cdac4c4178cb1b350d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.149926127Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/d, Name: checkpoints/3072/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3072/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=8a72b25c970094b2769095f03e59f895346bfcfcdf10539a1ee7e645125df7c9330990bd7b9f618b592f60db55e553681b243c824422963608708387df3db7f00eadf142492515711c1129501f4f8dd13c7a7ababdb4e0572fcbff25e7b29bb0972cada1d13fade59c41efb47612b6ffcfdd9b2a23f21948543646cdaaa0758d370d3c6b786ea84962002a3c9c83b1810940e82b9a5a5e3f2bb5ee72806b13b0cbd9c45b397fcfe5016eb2ad63f464a53cb1e89eed5d6e8d8d8f055ad4123f4be6b2cf0d73fb48ef82dc5221ea85d7059cc9c6799c7652a6991675fedd06a3ca3389f1407aaec283b341e2d2404de3f19260722f2124679b46a076d349a35f0c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.149346537Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/d, Name: checkpoints/4736/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4736/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=0900a651835d36bcecf822032976d24a0e2549d44731edd706cac43aac1f8a63aa676ef62a3a9f7aa78e129b3998c47ca2e8251b75317b2badcefc3cedf5de9b2fe54f985c9739a8706508da4886957d30df8528515290c51610f0facb4dd6f9f3db9aec584ca26f82c6929eaf85e32b6345a1c8822ac01d37f2ea5f00f4f7b577277acf625ab09f2641bb7a9c845c6ecc29ebf95802e106cb3e67d41dd11babd92504468a8bfc7a81c206f7aa6aaba674ca92d62ac36f5aef0381031aad476e3282f977adb2238b225b3f0b16e5cf72a81c5e3164c44f52f2a5cc0fdf1a149f0f871c79f396a1891f7dfe8232a16cdea4dcffafd5badd50b150ab2c2cfafc89&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.150034872Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d, Name: checkpoints/3200/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3200/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=2b9a8d2ed9d3d806c0cd8dfeb6063ff4d7d9a6ba008e59d3c334889d45a9e990e9c54607465dd8b430b9d5352565015fb4e60f5ff009503a19482e588a70a398a3414ba38158cf9cb2fb65c01f4d97e175f7b6407ff2db1190097a4a8d7b97c479e4cb99c3cc0fd137e38dcedc09e46bb625e622b53315fcfbbf11f33cf65fa6df1fba8f399a684cc46307f7851638b861f63af6dd1b2ec407e04cedf489c819788b3a4aee07b23bc6a9095e932c976dee0b388b62bbf2cdacf1c282806d004deaa612029a3d0bc1cce04aff93bde8a1a87ac1895c3258d0d87e306568560777203937eee620016cf76a044fc0e042157007d6f1c81c4a2b771ea26a0bff30b6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.150069175Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/array_metadatas, Name: checkpoints/1472/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1472/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b46d726dad8265ff0085e205b1e84ae9dbd5419680ab32af0f87b3452528a0e9b581e12f92e97b34fdeb212fffb5a04ecedc59aba570534f69bcfe5766111bd2e3cff7e548674be079af5050379c0ce150af2b07eee17fbda87e5a4f8992f8f817e46d514f7b42ad28bb2d5c4fb8709d83e0aa3dcc8ca6dfaa61dc3dc77c216d132265ec3bd1b9a43e5be60fbf8e5258db0ffd7b64c03d458eb73b494f9ce34924bbe3ad4ccafd05df60151188cacba233c24047f91fb71d719176d864c95f657ffe75865e15fce3933ebda56e0fd33f8067058fec52170cd1a01e6ac6caf6baef00e9a35ac49542881d28e16f62012b15e90259810f7c46e045f9ec21afccce&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.150106981Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d, Name: checkpoints/3264/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3264/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=87afcdc588428bf06887d20811f9616a8ba2c04653931580abab5660b32b4d5e4b90086c3c59f17b4e5d4e01660ac5847b4affaa370152eba352d5983472acab853ddb0d37ce5f07956fc2e41a4a69ee2a76efbb557a3aac91be4c681280773f1a0b60b62d2b09616ba67ac836114bf63cbea920c7ca904245dce9ee1eb34cb2e8041baa2e6cb84a1604fd61fc7ff743f0835f6e52611a95a84d2f52b55d1aad03a3983526bb448532ee2437febf90bfdb2c45e062869042c1536e8b606931922a94a342a9fa909bb7bb3b7c7ae26abe5457bc66af58c0891323c905b64be767292c1be4abe74558a3f3eedb28a2b7971cf4f6cba000b5207eec90ec50d94957&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.15016853Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d, Name: checkpoints/512/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/512/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=48051f693e6b93a88be028a2cea7fb9aeb38c07fb41ac95290ab8e72859646350e64c68595505c50ac1e955923dd5883fb6b9094ff09a326257c85b57f414cdf800ab778879bf49f52b4fde5a24ed462a68a80d6373542323f5dad9a23413abd05e7e66e4b27010f7e3bd3a3f6e70b1678721a9726f12c8759a43f24446e8481474255b20dfee0a870a4e667baec1a8985aff51f323296d197e6b0f8d484cca6a141ba940157a86946c4e047a2d3a3395d00489a8e4aad06ff5c2b7ed6c938fd66dddfd7124f45fb55669673cf945f92f1e114ea572ad93408804992f65d3007db94f14d4efb70afba768f443d88fe90ad5c5192c2f7dc72e13ea17844b414df&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.150188444Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/array_metadatas, Name: checkpoints/3008/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3008/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4672053f8e0784314619ef6f191e263601640ae691104fedde321d265780428a02da6cff2dd82865459caf65677c5d209fab612418ae0ccd01d0e93d058707816ec6630bf9acd018d99f41cd7ceef049aabb027ad890f46959c147e4aeafc35affdea14cdbcbf537059371df810a21079fdb7669ea4243668a4cee8bf197d72373208cde07ad65bb4389f7b5c9e52269dd13743795ff487d772e3942af02b7f5daa7b3e375a11bc2d6e8ea93978bfd595af7c9e606c001d0d91454a8aac8cca28cb9b320573d6128082a4d9fedb4349bb82cd68d6e127e85ed8298cba7831da53c3603b17c681218805d574801f2886737252d2bdf970c9ada491ed46a5711f7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.150199084Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default, Name: checkpoints/3968/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3968/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=021ce772ea9a3cc0de2e3eb22317c2e99aace6262c1362c1c42bcdae5e2829704b1136281f3247a63e89e0b8f3e87919af3b4761b83e3305d4f9c1f194096f391dc7eaf22ef1d2390ed4175aba8c765d5d2ba1391190df145581c358a058cb1790b39a21767696e8b8d0ab239363d135e42eaed171328fb3ae7431fcb81c590c3c8430cf6b276105488eb2bcf6978e1d705ca5ee2fe18278e3e7299130ffc93981beaa1124f18c41df960819d47d5a38f65f6d3112945962054ba99b080b1458913117d0b9f5de452591dd1ba1cd6fa1a97dd7753163c89f062abf3d6f2872539ff78da68b690b28273bcb6517d71bd3386ee6ee6ed3ffbab9189a3adf3671a8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.150237015Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768, Name: checkpoints/768, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/768?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3dcf6c80c63e17ebf23f6445c7b1e90a63113961a4ba79008e26d58405ad0a10d37111acbd0e20fd2a9c1a07e55306979a632b04206741056b557ff45ecc7693e23a0a2a63dde19e9c8b4b59c0c2637298242b10aafc2c0491821e117d90d2fcfe569553d49f3f5b30fbf0ff727e750d77bf6a9f2d548a5a39b193b40845c7155996edc98d05a7563c5567317c9d032b04c58a2f9b15a788e5e5f26a9061adf9b9ae302b318bdd2d540f3d17de276c79e8c531b2fba69b7a5a40998fd38612f5757477c2f3fc02a19f09a51aec91b190d9190e8b24827f2727d6eae9e9bece0a70c1e03d3c8e45d43c974254956dce2a648bb7891c2da9fa3345b6c53268406f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.150277881Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default, Name: checkpoints/4608/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4608/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0766f6c91a119b07368c0c83a86de13048036d3acc5b475e614f260f94b836dcc274f96f9144ad4db7f4e6c9577f1448f18d026b96a1f2d19fe0529600777408eae021f60f6bc4473e3fa2c78b3453c790facc35745c759dfe8094d3aa673587c3e89a4b0f58697821713f4c724efac4c91270c1de23264a74d62a9eefcc3a4aeb20d2de77698d149ac1aaf73023bc1e0555e4e1ceac13a2322b24dbefbc1a6628953918398f616115025ea30c681593f02d4650364dea2bf56d1c309f28d09506da0ca75039d751cad2031bc523664e092b017b24c45629ae44d3739f89f55965101ef92c998317b551c1d89ac619f2cd9b7cdc4260a08fb206e082cbc5cfb8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.150313979Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d, Name: checkpoints/5120/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5120/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=112f5ced9b1ba0268a07b0f7f66ee8d24c1d3464757263626174fd1091f5d42a9eb01bee401c0dcf62b0bc2146e1b33b95409a516aae9f1b82ea6d526585baf675b80b6b155ac59931115c9c16d1a4efa9d2d2b63b8cded29576f7e7c4cf80d81ef2c9b79f3338b1493692bcd03c7acddcee6a5a961ac7b2c0eebfabc0fcb394f947a517985c2d8e8d0a7f2d82f62e6e452ce8a54592c043eb75bcd1011d72ccbef9a24aaadb46a4fa45a3bf2051b23f872948176852030d93e50aa6c303e0dabec1e762ee6d9bc0ee2a38bc8899a96e8730a004c5b5b6833f988eb346c4dcc387ded9f0a3b2373e4e9ae0deaa8a26b9ae3a4046e3aa5fbc2fcc66b4bc12da09&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.150361442Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880, Name: checkpoints/2880, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2880?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=511aee70779705a3f68f419543ccc3a703a0df08fdccbc3da62b6465be290eb6db4426945ccc55ecea587a2d153c9ed9c565b90b44abfa227c2dfbd1cdb1c63260d1a2b92edaf0b02e02836c1a1373c966b34edc091bc4f8e55b4574b121ecf5b522c5159d8b6a8202a887db28500a2d49d5a0d8355f5b4c8556579231c3ebdec8890642582365b08c57b35002cf0ea5bb3ca1c6e88fbd2291c30cdbf4856beefae17da7838f4e3ca422ff6c6afb7533f60a03188a779fcb82a367809a6f7b4a7cf9100466647b79a0ace630fd2698ece9d300958b9229f980e67ba38e22605991441dfd540178b01e3ad2c0c3a9bd24e044704c08496d00954ba1a185781a44&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.150444973Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default, Name: checkpoints/128/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/128/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=076896bcae8b0bd310ca5423e4ed1abeab84354f219167f4f4c827880ac6664b8f94cbdbdbd05eb1c1ef380c2323d37b8b682bd1b9e27dd7d8f3743d29b56c1ba4323b3352ff9dcaf79e825daa215e72c3323cf7d398bb2a56c7437de965af1596cea6786c505db184c01cddc63da28eae256e6635ac5a44a6f961ce646fffcd67286711b71c8c2100e8b02ec63be9f928759d6b7adfcc5cb9fde870589e6c37dc1bbe8834b5b9454dc6c89da5662880649bc769dabfc95e8c5fd0b2c3bc9e50c93805e9bdddffd8fbef61b502211f5b3ae79683ada85e234aa9a976ca7ae203bfd0628eb1ed276f02a00696720fb61906c81828b1de029a608446223058f758&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.149853446Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/array_metadatas, Name: checkpoints/2368/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2368/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4670a27fd67cc18249eec0ca583f45f7f1de5b4c44b636176691b7f7264f6d9b1d2611e80650e07b1822717b46196508c36cd713eee33163c1e2884df58816fb3ec1a0a35dc3b791587bf9bb53b913c9c0875d959947ca432c052ed3afb4b2938f8f98dba98f79c8b2c9a235199ffca3e1654aa78247cee930300b0ae59eea43fc201b57deee1a9c08a7b0b5b1491b159a13bb564271ca55d02b4e5437208b453e14abcad7daafdbdfa44c41348a84b918afc9b35872a5525e50f58451404f1111c49c8c81ae142435dea5f0d6092d07836ecb23c793538d86a232ec4187335295ce9cc8182a2ac6041e834cb818cfc555c36b197dc7c0211a110f43e6b32046&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.150522885Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320, Name: checkpoints/320, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/320?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4fcc44ebb4c22a915c4367f12837c27bb8c8dd353872c11a06a07ad06bbb6803533fc30867b335dec6ccbcb287e74dad693bcdb0ae38a56465237af0109e6a548e23a00ca2ccb7f2bc14800c371db56d0205abc2814a984721ae7de66035628579a67bcf125eb0db6c72a8735582e13374993e11410e8bd0bb5f91c31c9173597b0ca89f4beb596efdf03f9e655fcc664ea90af348eae0a06bf05abe8a83f9e073a6ede23e85b1068d0f0af26c478f3e9162a1e6b50aad38bb0c64da9dcb75ac6a7e65df178e328bbc5cd38649c7953144f2714770849c7e4b1b568228a5d7379cb9d7310a211d4772381a28a19b90ae077eaecc24aba5c14636851d96eceb6d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.150570738Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/array_metadatas, Name: checkpoints/4800/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4800/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=9cfd8e80603430cec9376363d01a9ea38cace31141c706b756be73d6823d627715a5592ebba0dd02fda93b2dc090564c71d0a5d127b2325c441ffd091d5dc689ce05ae9efb6be8bb32bd5a80080178b1813af1baf532add391bd72a0235bfbf5d5c5ce89b88b04e41d86edca854765b51b9546535c92cfc9b013075e349f7d67b53e3007e7a708423a4fa0ca440b931350f582ec1aaf12f02862b5e1a24b4de5b2d8d439b9242267aa969e53f7ba7a6f510c808caa689365a3980188a2d6fa157886fb82f25c1337ed324705847f742334920efb65e5fe3b384679b762df2b437b6b77b1215ca2073f357a4f2ba66bed02846a30e3eed8725274de3d95b21dd1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.149360754Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0, Name: checkpoints/384/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/384/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0c25d040a50c7f3d9d216daf30c612650652f43feb3f63fabb9b1a5f437764ebb4734836829e2a949c1ccaa9c4e130171216a4f0ca1f68c6e7300ebb561ea626d18a1c120c3fba92ffee286bb51d4b47024895f05a975ad1d1d5cf968397e33223f80e3840e4ac7fa655d67ddb5d80544ab4f4d9a2fc2e56b4a8f4c33af7fa3d1d6e859c00e54f90546deb3d7f7a8ecffb484cce882937282d6ce3fa3383246519c558f12f88774e2fdfd46b88f82f043215bd574dc35e9e4dff18a7d3cc28b2b68dabe832cb1fbaa4176f84ab73200ae41776d3a073e535b3889b93b359ca3d7cc96c671f15bba1780f3f035b601c85d19419f7e54972b2d6bfa1efb47edf5d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.150022386Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/d, Name: checkpoints/4928/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4928/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=44adf7cffbe9ed24da16ccc0bf76030910a6da26b383ee1e1f15a0e0498633673c4db1b042bba13585003e6dc3e9f54d867cd2e26017d2bf097c73c9015cfae74373513116f1f26ab9ec7fa0e46b8b6684c953598c9ebed72135d933d3bccfc037cb34d6d0ce21b2b46e31c8f8843985363714c933ac535c16a419a2fc410ad28b383b1835d93e7db97ae2c06e5da7299c8798f6f6351d3ea7e2f48e5ab3af28dcb34f843975a4149a91f7312a794e9b677e014e67fde49eb72234754dde56333afe1485754e28af2fef32fd2ebcaaf17038354a28a62c7c572554c7bdfc95c105c51db7941b07571deec7618538e2e15a6511aa3366964e5c6413ca6203f538&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.150208091Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default, Name: checkpoints/3264/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3264/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=73828b0d437b7c111c9ff7b49df360033396bda7aea29fcf3285620142f200bb6bd67fc4bb59a69c4e2664fc41b5382ecded65a8a83ed6797a8e9aba0547029e5fe10150f29932ab027048c161a2d8c0201babf16831f66cdb814156843fe829c24d7a8d8fd7e4f9cf21283145a8c796f3ac692ae7d78dc2f8d5615aaa2e0cce130cd5ad82e7526b459725aa6d0d8daba94809eec1c56fe0792b28b8a73da6874c9dbecac1554d5cc55a06e952df39941ed0076941c9fd6f1de90a898da257d29b340268a32fdf11866eff5a231f95c612c80ba749e1844412034a553ce52f3455fb50eb578ce8cf09963d4ac42764f93d415bb1bd74a681377060ff6d73a4a6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.150669626Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/array_metadatas, Name: checkpoints/5056/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5056/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=97bda9b707769a77faed97c3c34ee455977f3d78d75593c2587586878c205880a04b6dad31d8ea67cf5be99395e8f63e8040762677007d82c13a70636d5170be0e739ba4576126bf4311525327b86fbfae28abc70bcd16f425d499cfb393af19ed5466b241eb1301a53a6adb0950adf8c29b40fe03e39849d025c88d3eee895f7619ce8b9cf3e19103a9620e6b46e353f5e6384a0f60e7c76cfcff0c57e8821d2592d15c4eddcede66c59746220e1acd284ca553a8c4295a0e2ce2859fbf23924a662b0bd20577c76bb28b6c18d30296f9113025b36ba66aa9961933693b0deab2c3acf7f4a35b04e6d115abab63bafd8f82086d0c6ec7abededeb8b4121a3bb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.150690542Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800, Name: checkpoints/4800, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4800?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=c00aa5bf626236f7d9772850c0735f44f42ef90d6c464db7d94778a90ff6ee26406acf6cb934f6153a4a919ed3ddef269541399bcb0dffebdf5f74454b9c0da32bf46ee2a2165a915c1de34a74429e9850e504e58152c3f0accf5f346a95b2e50d01bb7d1162286c213ebd65c978e5d8417e48908e575030ba308b754a09a8a094df300961bf945b04d591e4730caf1d4208dccad0830c716008d28d83c763fbc9c3d5d442a909b84ed8f0b6789933b4319ed5fabec661513d830edd75d8417a53738b796df479d973e34efadb385921eb537f44e49bcfff0d625835bfc9c8d3a37825b497e79da6a5dd3847b9a6ceff29489cd5a3d06367a78688be18fb6896&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.150721506Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d, Name: checkpoints/5696/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5696/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=53fed6fe0268380cea7aa6f48f0991ffd3bf304981e8551dad5651f42986ade5e92beafccb368bc4468432a365bb839cf71079871acd773db7a88ec77337b8a36825b69aad8376ffc16e35e9b4c62ea2ee83a036e455436980c61d3e74586b35b1ed355fd53247584c55f07db32369111ecf8073c3b973ba5e47daeedf55ace087fe9511c7024170a7282a6326d01505d6c877d314d7f59db8ad852af07653078793b128ebcd571fe91e3b984fb6ba599ff21f3af0b70fbdd48d81ac691bbff7bf077a961cb5beb6369b9fdbeaf3ff56e322ea921b8b7ddc187bab0c591138f183cd90036401dc3a4ae4dbb21efeeb89849a1b2fc23789f041b853c00d617dfd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153173213Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/array_metadatas, Name: checkpoints/1280/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1280/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=abb31447189ec3cced88ae4d09a694dd95ae8ece628652f2c9142327d277196a7c89e4b8d500ea30c446cba0f84dfee33edd3b2e9de0fa056e6baf7a82c3e32a1328a2310564a6c14c2868cd05604bbf2e25e0e26e636fcda5da20676acebda40bcacf5bc752550bec865686b158526ea376edbbce27d664daf121bdb9516fa693f009c84278043a9e78f906d296fdb5ad6a12d5b626d84719245423482245f76318a1dfc8f8f5f686d506004ee0017f31dd9707cfef972b8dcad5b8b8a55665beefb1a9fadb518136364d0a2f5a1aed9a1518220841ee34a4dbeb4ebec5e9ba7d6e9c9b08b24474e43268daa6cacbd08634312aff6f1dc2bed3f9ec1f241cfa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153212715Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688, Name: checkpoints/2688, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2688?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=74faefae921be41c2a2b4aa0389888b04d03b57788965d9627e2adf68a60b303fffd72718c66919172eb417ef749f33d5880d5217e3593fb91c9ffa1b3384031cf754599907d8e099dfbe4f7dd2018f45d7a595e8676c5fc9dd85464feccb8e8cbba7984c7cd36041ecfcbcd8ce1ce7fb98a1a4a4ef021ce8b7d6866b1bbfd1831f71e25cf61280edb6f6bdc0f9a1896a631a1f5d2a8a883e165e96350637729bd6c61064c0993bb8ed84eb1a801db113701489f696835fbb58ba350e246a21ab901b1aed92973cfb6a6095e98278205daf23a1884893641f0eb68f6eb775c66db53ff81641366975c89f78102a47c4d1d22e947087cc3a3c45d53b9e6842c00&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.149276979Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/d, Name: checkpoints/2752/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2752/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=26364323f61735821f5fe56951904f310ae2eef24681c4d040b06210c4e2e5b71e9a408910e636e3077cb0d5cd537d59ad08103dbc51c8aa245872e7ab7c650f657d86bea8f7bbbb4a57fbc3175c466d1f77f8504ec3f49ccde4c3aba4f18c72b5410c638b056b8b8258bf01148ddcc9b8dd6db29088957d290b9b8770ae871b1151dbda44f11100e2b8600a9e0a46184e198322a5b8dd8ffe036c62b9442b0a63240ebcbb59b19f0609a53eb62d6d51167bf3fd2020dcd8754cbda5bcfcfc78eb0e2d8d00a5e7d965efdcee1b9f8e79d0e6c232555d7ce20613fdb1ff833c484f81199f284f9366a5dc27d5b42cb33f046e337c9b2dfe8c7904fdcb1eff9cb0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.15318248Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d, Name: checkpoints/4224/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4224/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=535512c8b66c54dd4f66ee607657200d1e015fb1643d3f02dca3e376cd32450f6a906c27696c96d038dbd2495b9095668fef08c7b20cf32774f7b320ce165e6d72e1219eb26e1901a4a8a755c4bc5e043f3702495f38759e787e233ae15fa5c809fab32ae9de73fcc9af40c7fc8d7f0f81e0b754f7ac8af6a5e3464441ffb4ce3584ad58465004f1b8a15ce79a4514a5ed768c3eb606c5203e1a7e25018ca3b2c3d09adb1228cb3a85ab3a43509d9521e97dde50f2fd29554c0b1198ef6b399cbfd3f33af440c84b8c82bd8ca75788f8cc02a1980285c167a147d752f807cf70908547aa7e078a40a41366d7dc1b6501c376bb54eb936f82453b14d7450a78c0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.150246771Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224, Name: checkpoints/4224, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4224?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=a92c41200f03c938fbf59568209afdf40532912aff78a3aef1ae80880c6b6e9ed34872eef9145348a0fd75a74bdf21ba66d3ee292442519bd063916fd610871c43a68dcc39cb3e9c7450617e7b196229b692f8c4597ec9f054e7484fe12b873191e46c7785ca2b94cc1c01f64a85b780eb4e981f4a75c65937e165913fda7008c0a00a7e7dddff5d4639441a59bef30ef45379c48062d63ffe93e05396dc8a5bbc9c28818e8587107b48bed000dd2f145e90875b86dae22b3344608359e90f3a0c9ad44cb63917d9234199dc4c685868ceee91167f09d7866a552511ac914d19ae15b122a52a54dd0f4ee2f9a2f4d611e6a182fb1760a5727a683fcf54e89d51&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.15331508Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default, Name: checkpoints/3520/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3520/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4eb1a96ecefb24ad8b06b0651ca63deefd020e032ea3c77730b4f3bbbd490c857e83b7f9deef005c4672bfab74fd0d94a3d404701ce3340e43df6f66417dc0d852e4854f045d7cdb3a504ba5150173640498645309b86f3564558cc8ab38700c10cb26dfdcbd5645433f5b19679ab8d280c81badbf771eae0febd8f4f15e0c97475a741bcc28b05b32de36223ee1bec65e81cbc8728a7c78632a98887d0e3c5f31729af2dcb2d5e0e7bd207e1897468394058bb2b794ece31527afbb705eb3ef85d726d6f994d35c21b3e90a4cf254c83b5a59e654df63bdba0e4626efae7c99425649ae467cf4bfa394c9aabc1dd4b2c61aedf5fa549d3e86d82a1f2e27c29e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153303589Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048, Name: checkpoints/2048, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2048?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b179f052496e4c1b73ff3c4aa9eaf50f5a7a5ed316e2917505bbf6d7d6f049fb7a771774e0925c8e67693528ccd141a66b159894ce43661a433ccfcdb9ad5c2597adf9dcfdac915641c0142b5c13e0d750cae2562f284fd94c484bea6222dc1645e62b4fcd4f3d511de917077d1ec8b6a6b0605f9f71fc75f409caa4f62ba70d4b2afd62aa786f2b646719c265675e6f0b9d2cbf4ce20c46afb36d04c2940a166e87dffc890894899c9fbfda26014382ac24bf71acdb9b4baae13d6d6d38cfdc55a4438ae5d90e241e96f407d5bab2d1f3c5d9ad866b7e9e3b1d7007407359971adb0c1b890feff89bc4823d744b3e0f5336c9583f38843a870356260be178a1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153438556Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0, Name: checkpoints/3840/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3840/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=aba4a804da5a9a6ff68c38590169d9509be1fcf0eaa4eb1aaabe191e4a4037df39756d160a5bf798498d5f86e3f645fda6128bb50ab4da2bc8f2baffc7916644ecd7557c50f3929af688706b85e256770b87ed08468aa11a21924cbecf88a60aa3216e8e1dd94eb4dcb1605c9207a82f9aac58a7ebbf743227ae1d7f509f3223428507f7639a51ec81a4059e9aa8d3c2deb0b7a21adc28fb7c6dd9b17aff265a852d90ed171cd441a800d873575e3750b88c7bdc33a449c5a357903bf8b013b816b631b9a37a90cad20c3bbb317ba071bd793c386ee1648710662dc2fc056682216e5893cc9d1237f85b2a674383b21507cef453fe22cd0401a8a3f0f5384305&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153467362Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default, Name: checkpoints/3328/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3328/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=40089f7584b333f7f9bb70e0b56a179936feeeb01375013c441dc279dcd2f4803636af4ac2a0c836b108480d0febc46840583f502ac233179e7349090a33e2c04749513934e07fb4ce8aea164c517e0f85b7a4918f53b86849654b625061e952f15ffe8fdd6970610eebaa881b50ce8afd90707de85999ca98aee6644e3aa00dfdbacd7731eb058e65c5b891fc37b49d0924b3be0030d7b445f03969946391655610603af7ca5cbec246754e02485932b312f39213152577075f2059ed9634179137e352f3d7218b4f7b193d1f5404461ab42dbde2442ec21643904e9c57e4235556b248bf5df36221ff966e4ed62f1b30313daa0d8590b83c65d62162f3c8b8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153454373Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368, Name: checkpoints/2368, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2368?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=73e415441a2727018e274f33e6cf32ab6f993ab93197b1ebaf4a321bd5e25d10b1e2eba11d48bce2270de9d45d9b57892e3a861818328dcf8affe019fbb11ba8f0c8825d9d9041e7bb3e122b42149f4e426d1ed3ec37b6815a21e7af492e96a66c18798c639c4a2c94944dd34f575dd37dad211eb95cf04f965baf7812fd2fdb0445d254103d824fce5db989c64b04b3f2caed05d25466a7f68430caebe3c75f73f3649fac414781d66e789245938d57a750bf4c3d731f5e646d1dbe547bcc07a27daa89a99c13ef2fa4335eb6784e86630af7c20253be389f3d4a6539591ae9480b57f7dfe3540fcbd83b17dd5efc9c1e9160b5e0a5e5ec4e4ca0d28f9cea4e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153498289Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/array_metadatas, Name: checkpoints/3840/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3840/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0bf9fda66f1ced86edae507631fad357130407e2197e33897260ad8a4d72aa3159fe7a67be8c2c3d7c7687b0001a8377fcb6474599515a9d5c1dc9c3899755db045eac5b6dd2a674992b1a37b0ad995524231f1009cf927a8c2c7b67fc82e02d095dded16c5fdb18247d50ed4dc9bdac7c0341423a131671b129116c5f9cdb934d5927c803aa6224a28387d08960c835eb65da5c2e17cc366bb6b6d0f375f91e5e4b46dcc0ff02ef8af84752ab699456f0c023a3123006cef0ed0eba68f981283f693572d99c32bc92730555235c358a9315d8430d77a80cab8075b7630c727f9e4641e8c1583c9de327ca1872500f9a1c4310dc6f06c8fdb722919f5491988e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.150452806Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/d, Name: checkpoints/4096/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4096/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4e65abeccbc802517664f7595bad7be1aadd1f7e674d1d5fd0983270fb6f37004aec836f77154bcb3dc8bafaedda793167b888663bbd54e52f8b9f5b62a4a9ef0c593fbbb9d7370d2f97dc1a62f44c7d541a115be9b6df11429ab4e623a344eb0b0501662bdee425d14723893a8c20b1667856c4aabe43891f7596d3da5687c44443c3a7d4def1f766f58d434364c2fccbf7a5ad278b696448c20a70355d787e65d35e628e3ee427363f82911e000acb71e43e0fb650090c54ddb2e9e45efdc21aee7dc5d209198f726f0b40d46b9e355a0bac82d6ea8055ffe12079900409d4b24e0dad790ec9fdc7d394eebaa33709a7c15898c7d60f125559be6f380fd6cb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.148155862Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408, Name: checkpoints/1408, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1408?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4048e77827b3c7ac2fb53394f793a139de2f993df70117a89f31f41675f2f64d08b5518393d53ea13e90bccc140761d782ae4ca95383c54d19c7d08cd3e1f47fe9a7d0e2a496e1adaf1e49efd302c4f75fc2a67b2ae8e455a4ee661f8b3b3f3ae3da13278f8e55d6ee819eaca1c57f81a3206b6ae3b78405a2a5cc954547b8d9c8a70111f4d18084516202bc519c452e9a09896d5de2ceda711a29fc6bfa40df9c60f4700b5896874530bbfad0bf52a0824214062549894f420d6b6a6e4369b8af6991c24db774902ab911a9eb0220acbe62d2ee336ee36451835866c9dc06952bf87711f356d3cfde1cf1289734120988f7ece5f5e954cc194ae35b14eb107f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153436573Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/array_metadatas, Name: checkpoints/4224/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4224/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=981dca5144354ce198923e91f2a443b1f170fd9020b6bb990514dd263ca8fab05bc5eda1d9007300c50d515b53270615949222665504b37494bfd4e735ab5a60a76f79c188fe4077fcd62830195dbec64588c596c09c5a41f7d2d0d93ffe8f02df203f28a4b043b33dc827f43baa3f86a6cefdacd048b3287e43787db3605673b0e2735130d99bb6f20d8b34897609a665595d28203c65f90b1f39e617d5bf755183c7ae7212212bf8e38ce801e64155ad61f7062c6c48186caaf67f2a5acf06ca5e0f4aac11afd4608a64b94c47e086f1781e57f4ae451bb79d301f5e3de65ea3ed4d9793e906fac061f8830bff77f3a2d420b890fca7e34952a7f53c7de3e3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153432266Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default, Name: checkpoints/5952/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5952/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9794da8f6903b86495d365769209590c9c67c516137857492d0bf0e25eff4f39442cd8ef286d7a48effc53a37facf86cd546a6625da4fc29da474c0f676777ae69d7c454128491d03f326657443964eedd940280e798273b8db6ff757020235f18fd4456966a3d93b45275caadad72fb0e14996f333264c965fe959c3072ab9a6995ab8d1f97692d0bbc53e16f0f8241fb8e9f6454d24767774e6d9067a097a7f12b8c1f3625c8a5b30fbb1e118cc7044d41594256a77cbfb5a85943ec5aa07bc1d244c1159eee40b284bde41745fda0fcf30a72529b5a1e77870957527569aa2f26b888d638c94f615e8f35a2c7a683adb29155e485e6e269b06fa698a590e8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153324295Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d, Name: checkpoints/2624/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2624/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=68880816fbfc358c88b892e7a2746e49cdd3551cddbda9e5ca0e7c723e647a78f7c5f465819c67b987adcc3a01c86c0b39611589951c3344f7e777ddc6f3a815ca02b25f350148c36c7920e5fb8a462fd2b4984c1134721217b413759f13c3273c7364ff5e46b6b3e327c5e1a997f694290fb414be9fc2e551aeadf8faaec2f987c64d59de65fcb0234dcab192d95257bbefd74b0dac93c0824f81bedabbce33326f2d73a85df1e01d2032dcee0720d92bbec6f6bfef0f961305318acb5b4871fff7d5299d32c772e40edc8f0cc6541030e6a959746d566c5c909315b61681e6ae5c33badc862b294cd07e350706618c8a664e6345b30f6cb3797428247b49e1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152972623Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0, Name: checkpoints/256/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/256/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=0d7f4de6e1fc72e66884876c06be9ce32b8d4022d71dbc9d5de9e1dc1fdd5c530351d6a8b3003ce634b0627fc505b80b821bebbf83bf8360b3822e02be56e7f757a77f3050c135b0432b6840e0cc8b907e8895e43b82cf9a18198cc6c6b62b8dd8d7d056fea7598d1eece15b731d026cfcd45daa7b6ed723f7e4832725f52c079183eec6aeeaa2740ed086cabd74263d29f28650bf6af7dde43d45d79d6b13aaed97efb2fe9b278314a6e7743cd69e935dce8bbf0f89605ab6e65e1025769fd7c9a19f0ef7affa82a9b8d26145094c4b03e09c1cca921bfa548e1be6603353ea7981ee5117ddcb81e2bda14431d7b6ebdc1dfba6a6800733e19b112308625f2f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.150371057Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264, Name: checkpoints/3264, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3264?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=55f5cf01b7b4beb1b3901b7e189811b8c8ffa297739b303fff4a6a13ff0f01dce6616d5f254cb297e842c3481c453d51136569dd71046bf11cdd70c02a468101d985117e57ba6db8e33477eb6cc067b134e363ae4ef64f14d30718f2b7a603e3494502c22f942321a6739d81ab26d0339a32c76e8591d9494f13ec44ca30ac4b92417b2b3f0a8e896f34782a077b1bf9af67f426c17d5285a8e563d822b41bc080125b10ea162493b618ebe1273d8712be70665ece870cc9585d193d2461d2fb298c425c046ec6f1d9fb8fec88c4f8237c39351ddb8560a269a69844e942976876da119fd034a1c6a3cdc1b95875fb3d913d09d151d01b548b51b6ee3c6285e5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.150286714Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default, Name: checkpoints/1600/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1600/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=95435d91ea79f9ac87f3f80912d60c734539dcdb9f50207141cdfadaf3b871efb4d5efa65c12a7f869ec5203a9e5602fe4945c3eec084387a6041df27ca252d2a17736f1f4ae47ca6bbcbfb441d184e80293204cf0ba7009cdfd4fac9455adef349908294b2df7c4871295c5cf32325407893063b67e7fe6382775a9ba8288fb4ed020d342fea77c1252510686b8613e987113948366f35390f4c1e3fa715313d2cdedfdff1beb73834d6e75b7b18561fdb100f97bfd23a4719c287becccb429e310db8de5b697a897b752480c7727f65529a5686f824449b283c9fe751fae53ced5b5c40fced597f83fc4b1ab1dce067968efd4035955b756c066e25cfa74cf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.150725861Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/d, Name: checkpoints/4544/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4544/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=05c561a01db1c4ef9c149d10ed7243609ccdcc82250c8de108058b940eb5b83e173837ab42cf7b93c07c032223b2001e41cff011a31e8b5ec53ed13eba816ec3efcfe94697c1457bf2bdcf010e7c8210d9fa5b0ba0fdeec9bca443c4b19c4d77a1b603bb08a70239b76b7b00f7abb477117005f2bb675db8db165850a2c0758b3b0666bb729b6674d959bc3d1cae12d5fd151a6fe65fcf2b1a8d24bc456c04087ac4dce5327ed16f4c56e588e06bb74da4b5e12e6ba2fbb54c915f28870cfa807a1abd772db6fb96b87887877ef3a6b1448a6f0eacf0c537476520c067b2666afff9a07a55cd3cab4f5aa851c0686372ef57456cb65d85caea6e179fbed5b79d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151154249Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704, Name: checkpoints/704, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/704?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=664907363df7e33bbdbf610c6e8ac69c2798807a1e1c8e0afafe1a86e20dc6bb8953f4039babc73cf5e02a73eab65be94530c29e33c63245761d438c91cdbe3495c75bfa74d2e1abce557ba5357dd17f337d1c06a30ce5cbcd5bfad971666b6ef6202c207e2efeb6ec412caece5758205d7ee826fb659d885987006ff879819fde525a7776988b35237ae08b89c04857b6c521bb4ec3abdcd47253e1251815986378ea1efe721b4651de30a949e4a471aad508957178ac7d42c81285bf5df01ae32c4a51205a01259ac9c0dbf901898bd790979c207c621eaae6554824e54f5375f3fc5cfe5f785901671cfa5f9360fa1fd6f9662ca7d8b9811f50a8bcde99d2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151163514Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/array_metadatas, Name: checkpoints/5760/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5760/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=12a4c4775cf48469701eedc422f11bcea6cf0c0a808b0cd5e18672fed80937dc6a271659aaada644e97c12c21fea290c70c256aa1978eacb196f8371d1e40dc0a3fdef54c1ef42ed767b563869b3b88384e13975875f530916701b00f99d2bb69b5bca7d8b6a1d7d54b138f1823092aaa4bfa8659edb9a72aed579e954e76d5ed61f25ef3d12a84c326f5811e89da8068fbd3ecfbd15de812165d8b58a7df1de57af38a0569229a214064ed6ad33e33cd24957f903148028b9415769e98ca868a847c90091a5aff790cceacb91048ea6b34ea77f42567b7ae455d79634f9ee023b51a40288a39121e9c615a9cf8eecfa3c59a50702e759d53d8ebe3b9d89d16d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151206327Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/d, Name: checkpoints/832/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/832/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4a366f28dc5a6079d939da59767b40286ec6d93c8de50e4ae1ef86db10e1acc0b356a0567969f10a27fecb75c815b24da58d6607b4b61ed066d16270c0266de91191f7a3b1e1a2107bdc7da68a16f622e8f3b90d0000c18086f71dfe636c01816df108004220ef11993ac075c95a87bb6b4a5d222d8a8a4ccf1e3c3159749b41dff3939e9ef8b30be65e816f74b4e1bf99961c8066cd2d49ed0f4196545c57756612aa6837c5c3fea6299c174db28e223a41afc37b8e8e528a653386b1b44312dc1f1233d5b248d592c52bc406d18c2fa8a7172ca34479b5474ed1198a984ab30ef099c576c5357ab4236d6caf620e580bdd77bb6c73b6ff4aacecfeecf9bf97&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151278621Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d, Name: checkpoints/2368/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2368/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3ed2724989fa014b03a8ae2780d8c24b1b92fbde2b2a8f6cb6b96df1343619599b63c37a26f1f90555e6e27eafa04542e2a76953ae44791532649a0fca30e931ee1de9a25fb2a09cf5ba24ec821e872adfe8fc8cdeabacad756e475d440d60639bdba76ccfe8c5298f27bd8d5f3fd5b16bebc578265c5120e239e9ce9571c2666fedfb50ebd1f9fcbcc921fd0e1733809e79de4c9233e5c06115f7fbd545dfbd37e9dfcc81309bb3387a0307b89933c4489aa1d4e78ef3872934925de219ef77ad8ef86a674263da146837a69918d4d2c7e090646e03293ca38c51fe3c5c73bfd790f6da93a0f51152ca61ce79f5f81bab0af5f15998f9446dc8eb8ebc045efb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151300212Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0, Name: checkpoints/64/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/64/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3f2575bdcb7440f87f42b2709daa66c0a4dbc45bf993fc0328cfa3ed3c0bd190c779fe980f1fc8f9a014372889c6e978be94807d8ff7217359b5b4250c6503ab2136d76f341dcadb0c21a931bb9df3d769addb30c21f4129a491474d8bcdfda0c9c974b4b947e1563ba721449f65dfbe8727aefeb7744a8c079aa5041c28db8b70716edd6cc77888f16bf2ee5eaa2c266d9c3d8852392bc572407ab9933cc1a920b6247a78033d325154c9ac518d4696d6607286eacf9a74e3915d04741627a4ee43751ce2637098dd54d4c2c2814dce05ca832ffde124235d7caafb6ea419710f34f19d357a3c523f4068a829b05291ecc4e9e9e4a7d1baa907b242be69ec3f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151324448Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/d, Name: checkpoints/4416/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4416/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9ddb52560db3e18e18c8ca3fc975d3cc5a7103ed26a316184b3e927096dbdf3167abf507f730865f6f2d539b962798bb3207ed6149d78eae0f902f281bae37b88ec2aaf6191f9a43cac44283ac966a447b1c5096a2b5649b07270cb355a361094919bc4823e296f8774facee4d8c7166ae3c90261d73ee3bd38637b4cbc9fbe0ac1e7275c007aefef9205c6346368f1d177180b72948d9a9243b7ba48f2b84bfe49fa2ddc4eb15ad92e9ce4d93d69d53537382b22966f0f766bdf653a60295b216b63db6851730aa1fa86a14bb463a7da0054ad4cad958652de72c397abb433fbcbe0a22b7e0a1fb374d5a98c4f666b03f98007459f120eef856b08a3559ed78&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151345173Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/d, Name: checkpoints/3392/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3392/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4cecb2b25574396ca03ee2c0bf70192058fed8902efdb722a3a2af2449a7f985e48d2996e842a56ea09df211b192ad2577e20044aa6e5e91ffbda2b2851d9679eff5360b6a99fe7a3787313b6b14351f968c40e8a280d8010819d624aae5088ff290347a2d6fa47aed7228da3e3cee798fa366af16aec8a17dbf630de8ea38580ffe60836ed6ebda0d2db91ba484007d3cfff6e5b472519e943996f1bd7ff593d5fee4e6b95c7fa12cfdff1ff62ea93c57ce04e880100845fe8fb9784c2ba9a446092dfe8a9743caf68d6ffd57e58040af03d61d4a03ab1c12541c254439675138c76aae683f63437280c7e6dbb196803c74bc5fb3db70c756f09f22c5a4a25b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.15136276Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default, Name: checkpoints/5248/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5248/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=34fd067c47bf08ecd61447ac219e1013d6820b000c94b2fa1e3947189c10b009aa2869570e70c396726b5cfd3161d1143aea86a74d312ee6027b2d6a0cee274e167ea6a4d6d05376bd95b97e2f466dc7de458658d1939676091c50f5d5cd84e367e28b62ec3265d5def7ee858b4b11ae11e251bfaa632b1399aa3b425489cc7e290fe4d075162e17cf366f5abb2bb946adc600380f19482d0dfeab61059b0c0f4f2ab7d5bda622e31d2932348050ab1cf60954c9a6169a53a0ecee140def2393a69660c6a701224f352c491373fd836393a33bfe2680233c06061b9fa0b94c84f174ed4bbf26ac3b38eeb4cab5967a54ae628d37f00cfcd2a3fc4dac8ef74504&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151395858Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0, Name: checkpoints/1408/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1408/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=324ec6621d115d4bfe5e19e416f056bed1057ef257452695985d7e8f9bdbcb99e0bf508db928ca5ce36beca4818e10ef43d34170bb0196068531d3219ad593dce7951824a59edf68daa3f3ca414f669f4453fa6f12a221e24bd964aa37e2ce0bdf83483707c7e28f3dc0eaa3afd05504dcbf93a7cf55eae7978b4d47aa3affd77ad44a8a6e5b19ea998ac6f94f0452da63329ee5e8e559bd6a38ba71743f0888b395aab413d5c86288d073ea9e1b024c15f1b80311e7230d205b1c254d3bed8034baa9bc13cd65f7258730ffae32efaca3f8ee41a5a97c5e7a661c6e41e3261921e487febf739c7215e0ee632426b64b7cd158d42e914a9f616b017087cb9b9d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151440129Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/array_metadatas, Name: checkpoints/2240/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2240/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=9a6c113aa07757ddfbed507974e23d643450ac081cef6a5c4f6cd53397b9ed996189a7fae9e38b9235d3a8a62c50c2deae04e6dd9e9040901bdf6b0db62b9aec954e50e3314d35febf9236377fbd53ca90349aca7c2f006e976488305d990fa33501ad82e9cf50da312df358912f18bac14bec0e98ff5bdd5a0905969e3a666178c98f2e4b4a2c9128f9b91430bea2cb27d1ccbc982a9a7b68a90c2ec9262f66de5d39ea39772bf76668bcb8169301cf2146528114acc671a0581bf142ddddb27662a049ebb2a25dcd98dc40b44d8e5d2e660248af2f1f5d4083be65a643430b2de2c328336736670176af5af58e9656f49b8e856a4627ca3940103875babef2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151425373Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0, Name: checkpoints/896/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/896/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0842239f5e492d5ae58542a474048c055e7d5738a8980da7d142b500d957ca26e372a8fab9166a9ef0d0e9b92a77838d85dec376ce781f0ecdbde456fcc30d518d6fcbfb31879d0b7b77b5dc316c29c27bcd4838ef0f90a821974dc47d99edc53e0af71bd901d5a9611b3daa6b765964a392df2ffc8c4fdd209436227e6fde0b47f1ea2ab11140efe078b334c2b83be6e283854f4b4727d3f6516a727b3310ccaa96f225556fa3dcde5eb8b202e3757a4ece0d2cfdcd7d0d47e251ac771ce1326788415b23966cb42caa1441ffe88c9c62c63b915e1ae72eca9339a88e83fa004933f6d0662d11f2f536ff6b49401c419df051dd6d56b478ef830f2573a40a6f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.15145245Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default, Name: checkpoints/1152/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1152/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=1bbe949ac0d466ea1186f0caec96e08769da27cd02666248dc7d3a24a83de9723770017f8e5538bc17cad99d2ebcdb129d8c84fdd6ef245f216f0b4b431061008e7b336e0b6d30a7e664930fad82a8f212621d9e536dff8a2e47f6871348d299a6ae674888437148c74b1dc206abdf8bd1287449c9f878b0d445e8d8d81ccb23e668b72aa2b8a3456cf8a6dffc3c66baefefc957bb002559df61d7fcc462bd3489d24c7942a3049f077c6a45152e8f471fc6711653e8001d732b4966d5d3a49ed4a37dc327cc48db9f34e10aa1dc163fa840f455db18312f3c660761e6c32735d5ad959d311f4fe7ed6d1ca6058511e56f1d6b39101f2c66c69b469f063b4762&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151497747Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0, Name: checkpoints/192/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/192/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=15eca9d9263f655d0ba8a340dab9e38446a7f9ef91b605449d6363fb594b2e84c67f347bb40eada54a7494123bbcde3762a6a17ab0d3dbdd31577f2eb15bfc6544a4cff13295f6484c96867caf9d83f4bdfe2821f9d32a4ef3d4c8010999501fc72dfb880f5c7176831ab21f6877e7216f909ae3c2ceedb0df793de515fcb4dc5fd6af5c08954bbbc6c82ebf09039adb9c98e6b995a10e55b6cbdcf5b1e5aeaf5208e10044f297ba502a76213990060bb3b1bd3d4b12c121ec9e9077fcce1047275bd9b64eef45aeba2297a8ba281e686a2f9ec1f5b1afeebaf200b263825000ef20186826e846e01380fceecf1d230897a8041491b0913ac9b8fcca5859f6e2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151491273Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/d, Name: checkpoints/1472/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1472/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4c0ae25e7da7accfc3968779a4dc77918cd5412487be2c977bbb9a27b6e8ed20ccadc79393f55351d53aeb1fd823edb432000ba6ab1322e4a93193538d9920377ef24bc659b7cad5bbc64a907bee4b0065fd6a7b8d8fd9e91b8259054b851aa9f10c2e581340dc8feb72519af7807eb86faf99fee60b82c82d367a4f6552a6e8f1cd14dc526856eb5466688df5343d8dd6fddf1456c9272badb25e22ed779c399c722d2efb6ed98e3bed230595e5f08079c1f58979cf2bfb3ca7125416c80da16b62c8e2308cdecd0a4477ced26a7f2d7293d94d01c35f23bc56e8a9234a22dba55633631b05cedecd9fc3c4e2a74e398b8869ac94081611d30400c4c1cf4a5a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151510971Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0, Name: checkpoints/960/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/960/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b7de31b6c6fa1403b9341681adcad84ee55fcb2153f50c37e5f1880a894a3a68384e8e2b56eac2d098b10c6fdd6b2639c2a138143e4e2a1d1a4c70b1ed94be3870e615c6a0c0470435753b84cec5c7b00708b3ee3da569dcb93b4d8be39ef42bd3c775d77c6e223ef3be77accd1ca0f5c0f48faa26ecac740f5aa3444a381d061be14972a6f4663d871cfe0525fd9105a9d9ee4044a4674e468245996678377040f3a59b15bdae79fbc199276e95537c4205dd38c688b945d72b7f338b4f240699f9fe8ca69077d98d89b7d11388cedb41cc3445cdf5229b6648a161bdad8df5f72a03ddde8a71824069f7972d014f17b63c1edd43a4df736d2eef4ab5e35afa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151544161Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/d, Name: checkpoints/2304/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2304/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=0bd92af5fabe085001aa5659d8cc7fb42b3322107d50393f67ab219ce0403e3ed9cdeaf92e5b84027aa55df8102a3a54b90d6715980d52dde698cdbd3833cb1424af5e2979cb7fc528cd2a833a13174dcaa2a5bb2dc21d98c96b17e1eb596fe047345e299983138344a0d1d8feab5319583f6ac7725d3a9c3dc0a9232f89a1c61dbb88c2f7eec3e40b97127695c875b3ec7490e754e71bb33c052f4498e269daf94cbeb56b1434a42d02e49487615d3725f591d892f05965f412addebeaba3137234e7e57d739400bba810b25115bb43009cc33c6a35d453abea3167eb8183f6155f69cf58c582357942fc9dd8837731d8cca2077e872922b44027bab1c2890c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151612561Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0, Name: checkpoints/2688/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2688/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=736d3313df1cc9b7eb6dcf728d2216d4810b7f4853d211b2f0068890770c05a592f53101e115c97a8edd9be9c26f3112b2c6e56c283d26bf2f4a0ba5ce59a8981aad4409f84e91165524907def45b92d88f1158b97e2cc800857777be33ca6da1a3a0bd878aeda0a0b4549be3a904d0733b88d58e612ac289225c6edd95fdff2c25cd4799c10ae833941fe6c23a8e88fb8e30077bf60f29bb6d7a0d923e0ce1d316fb26730557c078a55df8e8f5ba12f6b560e571b0e7255ff62f22d7a19a47e8a10440350935db23695bd63332dad70843065d769cf710f46e02b37cb21a3136f6849e4db8dfa05113196ff916a9605eaa1548e3bfa419468aa3f3d11009836&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151669107Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d, Name: checkpoints/3776/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3776/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4a70800a0b4ae74eed27d64fa206fc2b32aec6045f5089367411d0a3eabe98d69915abc420b2f1f88f316c9292782e414becf4c95206520c5cc7809d8855eacc1613502dd6788175c4aa5606c6b000f7f9d610b31ba6a691e83d4d43bbead857e404fb7f30db5653ba211a00e2cebef9f69324ba9f426f815c58c7de711834421484371003d5bda30c9f2a5bef9ad13d56bc540c44ba12b14a08d32b870d898b96626839b555f78a077ced735b02844b45f09432c9e8b43dddf69edab8e72ca2773d635414598b593fb1f5603bd11d5c1bc44d2ecfec1de1181fb625abab885e38eff3d7f4618a17bf1422695c399a711f4537ee0b24b465d285f0103c1b7bbd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151683531Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0, Name: checkpoints/1216/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1216/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=29c417d391c7c927dc15c74db6c6ac7af1089d58f43ff9c7094e0a0fa33c07b48ebb7760c06ced41977082f8cbd27edb5481a7a70ecb53d88ccadae45a29859308f990145a10f0bd9103095d93ae376835b6c30cf62aee834f982eaa90109fbc5ee79e2df77875d092b9bccebab08d539fb9f95516a2720020caad8e287692d64020bf5e0a421774c0e1350afc8ea3e8c465b7b6eb59cdd593866ec3b4e52405271c5627c6499164970790c0ccd42e8f28fc3cc00b586eca7719e77a23274fe3e4704c5e58ba4387743992498589e07ada532bfc0c07b18be7bb1b86740ce5ee61671411505781bd5b8064d7d2a45bf0303e17525b99dda2aac779fccd00fe1c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151724642Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/array_metadatas, Name: checkpoints/1088/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1088/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=43133b402e522d134f80014281ade08d6761247ff4f988203ade7131db8e9c834bad36d48e85ff6ea6fe5eb3fd4a23bf1ed78e02600f911eb1eada3b7369a9fdbc6cc480fe74b0e46e8f505e62dbce88da574c15fad99e8adcb17cacc1661cb8e388511171daa0d4c40e749badf0e3844b279d1c2b19d85189b3a3c7b58d49a99bcd7fc06e521d41cc59263e007a40781d42afce3d11b825a4c4e0a0920247ba2373f02f31c6337697263d4c8d2dc995d2396dfae9ec0ecd0c79128e6238eacd95c609724e57805e433996ac6f7c77da836cebfaceb8a6e2d2ff1f1d5099284159586a6627263780227faae6fa9ee01fa3ee0d5c8aa45f477a42868a0ca1a070&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151726828Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280, Name: checkpoints/1280, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1280?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=61549796224b381611f3fc2fff60e193e3a30f0bfdb1248444440f87e84824e48673859a5c434793438a7f5c296fe2484844fa024134677554eda2c06432a49e338469266b44802fd25e9fd5132b3f33ee34478daa42427b74c96904c33cd2776dcf5a4bfc02c3099c60ead6b30b6050c020480d2b431ee152c9f40501dbced424d241c52d7938f9e0a504fbc0f6e37ce4adb5799040329104b24fec0a3de955f3e678955a1948c9e823a21e5f340afa7b391847a082d6bdb303fbcd71005a08f22b75c117bf7579fbd050661e1ae0f005b6e1cde637735576857bca8156355a20b9691fb35f5667e5f89a0d3488bf182072a01925c94eea20fb39a18e7e25f5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151739985Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/d, Name: checkpoints/704/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/704/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=a04378c0dfcd890680e89b03da27d81e7080db96a364dc03222364103ae6f236262de11c07d3413c8eb012f2c6364d8ed4f3d5de9eaf3d1291c0be996389f1d41faae094919a4e9eabacc802f58c692defa87d10826e9ed4267b97c2f2ddedb95c4b2721db3be809b1d4c60ba8e4cc84cb13970175460cf9fe9cb35d5a5f25abaa5cf75ffee402f4708dcbafd193697f4782541b0076c6d0943f321f948a30af66c3577c73459556dd6aca600c430bd237cf38eaf931ee63f213cffc6e4f5dd92f4cfa2a8a556b470f81bdd3b8cd511e220afdf2fa001b784d8bdc5fdf4b560a61971e30fd64a4442aa26ecd3ab8f5e612e7d2c3c47eb5c0599fbfdd2abb55ab&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.156972803Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200, Name: checkpoints/3200, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3200?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=15cb12fb0b62f633406f474904efb2cc16cd65bdf54cfe3def53746fa411972b2bdcc3add7cb996a24ae86b9557c20c0bef54d0eee009103da56410344dfc5a0f0445c8c40660be03ac9bbc23b88d4931afed4869c459bdf1f248d04a53dd583aaf7f0038ca13030719b1d6efec45b6cb1f7b03230c99d0cf17023ade30664ecb59529aba148d7adf3e6a685ecf692ed5524de2878fa57a8d5412a2c4fc59968914268f89fceeef825434398f0770aadbe8c7282e86c1abbd3839f410d034bdcc93b5f0d707073453d0de7451d05a6424a491c7f200b75605bf7033566e385df3fec2e6c9ac8bb91f0b01d4bbeababeff5d300eec7c52a4e440b3a67cd5bdbea&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157016634Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376, Name: checkpoints/5376, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5376?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=0d09baf3e802eff6150f9da45d842bfe8e27cc14ac44801f82320f9ad274ba1c8e176f57d6253b36beed00fb95dfeaecb4e44edad2bffc86b2339515be4107d3c1f14928d688933993829d23d9a71be026868adda4a998ff82d57795df8bd695c67079d9d0e288a78d5593ded75ae7c3d931729c4ce24f7e40145b18998e98cd78d07c526a74dead04b5d4d84dd84c7e03a9fab160e76eeceeb1994a5311095c53128815a567cf1952bf245bcb098fff30421550e969267511413146e640807cfde8837fbf0f84a54805c4d16d5a77da61339af6ab95ba4808f5ea1b56275b2b048b8da0dfc4df51249526e0eda83b925fc4ac0f4d96f6b4e1a65259669f6030&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157068995Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/d, Name: checkpoints/5376/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5376/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b93ce709933939a09125add7946a56af626d0ad4f267b4244b8955d018b96d1084fbec7ebbfe67b9ab631a3c021b0ba69845fc6e875a1760b95b5a4f37b83d0085bb4aecfb9d7a3fd6407df88700ee6b4ff5cba88a22d36feec08ae6fc60daca203dc35f836d6510bfe040de919ce2879b07a0dec3ff3ec182ffd85dad5cbf375f48c775286ca0c15877fe52d9d328ac7d23db266fdecd196debcb44c398a53e35ca6a2d0c4a9f1579cb03dd335c33116cf7a49a026bfd35f638fc3dfb4929649e1541bed8000108a0e57edac8838a775091aaf310d7b9ce6ababf60bbe8bc66b522cc4c1272ab67396102059db0fd26fe2eaa974fcfa17e58544371c923ea91&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157076202Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/array_metadatas, Name: checkpoints/2176/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2176/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=1031e48b91a90dda8b0ba53dae02626b6b441db72b7d87c1572bea44bd01de9bcf8048e9a25aa67918b4dbbecb5363eb59adb0b0d37481d4b8fd1f4b19736ce353c96ba059dce95622ee03fb5ea71db512f912f7a4093a94381be1bf2c28588a6d7c488d953b5956f07e4e43fb2938f8c5befab31b5c3292979f23dea1f71f4a33b2700fb637174420448bc5c6d85adb59234b226c8e3a61ebea4b15aacbbc2b951dcfbace951404164e80202b23241694d9d2ad57d94b04b3cfb7f20b0d6bdd34735153c4ef25edd8ce9ddd951a3c57dafca480f7fa85931fcb894a6a89f18dc3956d7eaf454bf809561379244ccced45bc0e910ebf567363ad9da76df542e1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157148201Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/array_metadatas, Name: checkpoints/1408/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1408/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b8a6f64fc0f776a77b5ab4a3a1a41fd6485d7d0ffc8a26854ef7800876c58cbe8323c6e1785999672a1895e7685cbfe2e6268380b3a682e9941a781ed6f86c9f38e2c6f8a4020b43dd4f3e9c366a4c4046a231e05da5970b5f47f7f60545c6b7769094015db8f39240776f513c9820733a5f4f680a82291ff843254527a006fe9df39c6ea3772f41dbc8395fb6ae7a9a20dfbe2d0b98b843daad2ff22d99e6b135e8e41ff74803ff6172171b683ff28b1089fee6be0ec026ba97516788600a0465b45f07dac8516271006b25f71b6993f90e43797490e23e190c0b47cda705fc61390b05f5ade59efeb9d40957c44bb16c20f023c619a477524a983e87f0b604&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157246385Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/array_metadatas, Name: checkpoints/704/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/704/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=947f9c7c8d5c0f8f6587ac1693eb4016959ff726335d68f9bc96f22a60672c983902be07eecb562002d05d38108cdd5727d9d376fe5b5f41f80c6b027a1381dd6d60ebd46eea500bd571771d6d5278d5e12c25f4cf07452f2ecc1f7c899ffb43a424dcb0a80e56370c8ca9f11f2c2b633e80e5a1e18b66fcf0e7ae0f88648cf4f27bf6653f875c3d9369760285be684bd94f8385d16f9efcd7f8be02fb1c65a3a6ed94da659ecba1cdbf2638c575dda25587aafccffe2a72817d6548308b57b2e41d3fcaacfe606278dfe5c9f3bd97b7fff5eed80245ed01779aa785e7de099e3891b38b872d318a34f28f944f87846352f6762aaec6d76897a3e503df578f08&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.15726888Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544, Name: checkpoints/4544, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4544?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b5203644c0df65161f2714dc56bed1eb2c762446ec05ae84cb3e9dd8ab017e35ec976f870df061eaef79fb0706c3e9ae1c101b12f8be8e947eadf710775014ee9560b5d4e7e0e8e702d9c4041736e906b41dee0044421dde34c6fecb46e7d5f9b26c302167fa18eeae6692fd9a86f864790d45100e643184b9178348c0ecbce574d227bd1d38d7e316bc6b4792aea7d090a0b905a589b7823603c4d74353029f661c75b5284bfa75137c76c8f99f399053db92cd26b24db3a3fbc786b776db63958f7857b6a6c16cf303d7ab018dbe9c48ea6361873edfdf1f75b8824fa2c6b370fd7ececc9cb7e12b600a03d9cd566700bad420eae141e8991675b95af60adc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157349488Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0, Name: checkpoints/5696/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5696/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=8b5e43415e9c12264ffb9a59c2d295124b1a7655691ff461d689232450827d0106b8c2cc95941c0ef40afa7e9ff39e646815c8efa99caa433f00e299cf836ecde24134c4348e4f571a7b877d42d85d30a9eef34b91b68adac61026e0a611547003f878d4ae6437b781659c4484c28d6464d360b82e2250f4d3fac276ee5e9c8f6d63640e04b6e8444b37973020a8395f5072cae356e2fb80427929cf1b528a444eeb438c55bf3aa7455c7b59d71e67c4f976eb6339743fdca2c7921fd43a52348d17498bb0084a022a4fa6bb750c958ad4cbfbed9c1f753f096eab68d8734e8926180261af74a1aed55de8aa15d83e9cdad4c4304aa6f16bb795b294b7ea0c2c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157378131Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default, Name: checkpoints/4416/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4416/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=2588690749b6745b9b9c39ee1c97c1795c85695bd6d5839460a5f0d0df44c446148d826182c36cfac136deeb9b2cc4f7e224fc79419e4018a0711d6a49a12f296809243dc4af230fceffbd319c6347c0e90fd794770586a10eb19dc5e91fcf1b55fab55502634af0ae3947cea1e68cbeb8caddd9a4f5dbc75ffb03683310a3b399ce94f1fc131b6331cbbfa5bfd7188f7ed95fc3f74eebebadfaf7f0e46341d32b94e53ba56c70189336de76bbacb902162aa821440c6d91fc1f42db43775291f077e137773909846f7d581dcb0a8532d687b3a9cc5b34ac497cb522bd580ea6171c08a910e706d97e8072f94f76c191781400650125e0514075659ada462435&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157407711Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default, Name: checkpoints/3904/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3904/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=1a60c39838f963e821bfe52b627a199187367a4bf27c7b3d2a30624c0e42dd56d05742a625621b158be1357a0b817ed9ea99d936f1d8e0e7281f4f55a3a8a2cae1314676e7ff6209e3ec1a4d16f4fa12422cfcffa8a7153c9ccc8b7cec255ffe191231ed0e44de81ec85dc2ee76f018393786ffe4e1d4a5514580d21960e0a7f626e0d23ef19acfd772ad649f16cd49493d0dd4f928dbf9ccf1def02154156665576289345a0e9d3aa1935c1b4b1d991e451cefc2125e9e58f35428024999b0029ca4920d0c8370cdb5c15e7ffab043778d09595d11057e4a1d02dcf509712adb78b37068d8a578cf852490bbaeba0e7fd62b550c4b70dba054fa32ff3baa7be&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157444689Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960, Name: checkpoints/960, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/960?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4c903131361a0c262a9f8c848591cd740dc2e7ce8f47e2eb5ea082c17e18bb5a935597f1a1d9c87dd629e162446dc11969ed1afda5b64246e7b2cf0ed60616cf8bffb7b6c727d85fcbd9e2a8fbfce9b583ff1fea95e70723d3ff9f6752cd6f3c5b115c6ae4fef83cad20025a8aa53dbb72d8d6382e105cb1f2ae834051912bcebe36908851eeee50d6c32ac3890ce7d0c5d9c6e29c9303ec18fb62f6f8c3d4c99d9f964339c2e07362f738e7f080d37db4cf41f6b661c24b7842b23fdaff4efbbc396bee7583bc1117502012f01031f1ac8e120f5e416c1c97d4504b14e0f150cf5b3e1ac5adc6c8ad0258f5f124608158ac92e98f5bc562740ce3a346ea5e64&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157501704Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/array_metadatas, Name: checkpoints/1216/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1216/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7480c51e63d415c99e58aa36765f0c24795d1c9e40676753e4b9d26ce1c7ede7423f7159fd18102b71c9d29595b878b4cfd31608130c544c78dcdc8e876be3a1c73a0309c1b63f6055673787dc763753b3f97d5c398787b29014c0ba5b2aeeedbe9de54e183b16b8be8580dc9ecdb0b572c4b9f793537ef6ec274a4ebed8b6961bdd40b1f9eb82c943d2ba4fb90fbe7557c2b5f569ae70a516ce2f339104cea01b1e19327300bbcde9e129268c239847d382139ff7e5ab680be4c5548d8d3a4e779fd2e418be9f154490056f107b67ab398dc4072310c16f730a5b699e2422262e6dded5beebb0fd1cd7d5495a0dbe2f81ad4c34e017f327818d207c3b55e952&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157547649Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0, Name: checkpoints/2112/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2112/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=bd0de6346549975abd5eef636256b7daf36d61df08fcb5f672124a0ba794b3edde1732c393f6b6be1d96b238c0356f40aea746ed52f2ac9dd3d8d1bf2eef8fdf4f0c08ec3b9a8bde50fce9f91b26880f8fc818b200d6925bf7adae43407c526779a1c9b1d1309194496d31a16575f5f04ae15fc74151be1eccabfada1487cf5a72ca7670731bb1543ad27ab82ca1f70bb1f5677ef0ec477afa3c3099bec04dc620923808a53a0e503eae24fbfcab2d2efafa2660f9dbca116672e0a408e22683186f9d51acc1ee83c5482898d209986b34b5a87313cf5c24c406c0bcf6fa95fe69ef04197646bce4fac6bd2abfe0c67e4cf94f579d68c940cbc9af5b413ef82b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157583264Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/d, Name: checkpoints/320/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/320/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4396ecf5bfddb593e50260442e0f17a088449b1b61b177cea3439ee9bca987d4dbfb947c84f2befccc73567185403f39f826906e74e5362b5caab25d6c6c2e29ae0337302340bae37aae723b27a1cd34b20ab3491b39d3b365bb1f69ea6d8d933acd3e8d9aff1e4320c45f230b40b1d645d4dcac7febf2368b12d7338f54e64bc19ad5e8b5008d8311fe0c6706fb97989edeed05978a439b662b58cffda2e7bee4fe853aadc8671ba7975c94075a8151b74683c9824e58c5fb2fca79a71bcc4f0dd5ef1679d305e42b94295c1c8840fad5e436e9f8ef8b8f056644c271de2dfec0afb6e21ed075845e126f84e607474364370dfc0e9089b033438594c8d1d7b9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157623242Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d, Name: checkpoints/64/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/64/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=1c9833412daeb6d1d7482e2d384fc5a2014ecebecf815e9801a09ab71af35764bcce4edb60797abac096299cd70d64842116679b5ff9be1d8ade2f3c9a1f165d875c96f21f86b6d3c05867abc468fe2542b8820c1a53500511de54fd894f94bcc42b7cb1960daedd5ae660632a452aeb3770b16c07d0ac5a39dcafaca63b1fda1dab0de001a2bced7aaa967e7e868d34dacbe1246711eefb63cba57cf90007eee7339f1a83aaa01c19ec401fc332bd65d6279996dd7e76c315e7aceb1ba9e790e8b75da35364bfa4ca60a982d7fe8b2113a7686019fa930add33281d3e16ad27557e580359ef5e8ecdc34a08b2a08915948f5c48aec759a4e6e5f3a302af74c4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157688313Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0, Name: checkpoints/5376/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5376/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=09b545286b37e0b16d704bfd0d9e0d76080eae490cf6274bcf5e5e2c9d2af3975c9250b1f540307bfd75e402c8136d1f05bee1543416765f2208b74b991c31f4e7add33a2ce9b75288c55197cc6f8c604e50701258854c95fbde1b60175a8906fe40833bd3a4a925f7734d14f92da0c44a9b5bf311cf06c08d1a6f6ee878a955fd0ab3277e018c2ec24382f8a6ec7bc0296c098728992547dce478e42711782828ebed847460a4b64681d6b28f9f5914cd0ac6e66b3523a4e7a87dea39050286265ad2666ea1189994cc80a13ff6b2020317099f455666fe1ca975be66f2e436a872015fc6984e5e99f9f0cfe811c59569ce6e5973b0f6a86736492d3b6deb3e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151778581Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024, Name: checkpoints/1024, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1024?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=782ecc88290b56c14d0a74c8aee6c55ce7b32549faa1fb48f0423ddbfb2910b290c8edb8fab0b268a6949bdb045765409ca4b8f1dfbf64942fa4bd131ab82e62789f4f50e05e3d5087daae79edb8ce51b673ddef337a4a04da6feba9c24af98decb9537e6f054fa54930967679653ffb230cbab15973a139a6fa57fdde9c3ad5ea4d5191a6af1ff204029a8cf0395add29a41af6cd7d52009b8e2c784ecd223e2292927d16dc85899abe98f06cf8b110dea689550b3da1614f17d1d62e5ab7b1ae905f72b9b533893ed1b3724e9c7ccf7d393ea7fd4c0316d056f43b676ba5607fcca80199eb544d159db544a5e9efba3e0d382dd164ccc9c7fda7a849e7ae1a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.15180543Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default, Name: checkpoints/0/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/0/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=3ff0e089cb05057db5156cfad3ef029aa9af34385819dd6a9204e6bb60af193df45bc06d04d7f8ba20372ff75840e2692a8d750395ae5b2e8e7bad5fb4ca05ef36faaa155fb120420fc20e1fac6c77f06f8593826ec242d4c2d7700f34891f7456f55559f32c96ce7bfcdc100b4b2af60b83faa99fb86aaa31d275a816fa9198d68855df5e722d4ca749934b0dc494c5b7b9386a382c766316f86c5573e6de953b32ec4dcc1ee0f0a51bb63b91a7e43484418541c52c4cb3e760fe68af75e1cd7a64657e48b1395a80029702b93603e7368291503427c55a9fd9bd352a4dde6bb7fa153a78a133c40a8e04e9f6d17da5b07e0601e2ec7c8c18a5e53feffa0d8c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151800022Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0, Name: checkpoints/2944/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2944/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=67abc310776e764b7db3f82e8f1567d98007761764c5d6ab2a9f1e34d82499589c8a4679ee0a30bed88ed8848d086eef827944ad8f3693f9d3e39f3ddf80045d89be23b3be5d282a878e2765e6573b854eb1c4ad41bb91e3db818ef2e11b5e58e0da572b0e72e287283a49a930def1c519984c801de6082afb4dd8d72d0e5cae5a51079de14676683a65e6507dcd887c82ecf4cfc08e95aa4252d1458c6d9009158c8b550850726d8c6ec5e29130d77958ef0b625f4506aa6e4e0149b78e35429322dbcbcee8e4b647a06c66373c9b89e5d5f938024cc455cb7e0ca230e564fd89f2b4f7b7747a2aca3a950fc218e02cbbcf3b33bfdb9c9569a8ac9fab825084&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151831107Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/array_metadatas, Name: checkpoints/1856/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1856/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=8afe45c1677dcabd18460628e2c508da5562555327f5d024b90c34013b338867e8cb9db0125c3d1142ecc56ef7cee7116811e70bbe7a2523c9988d2db24db7c13ef2e2db40689e5d3725111dd42e2ef9349552dd82ebfcf3a5bc9054c9e18480b474720f5274c051c051dac502e175e5da3ebb7fef6268ba4afd30edc8956ced9a53caf12c89c94e09a776e08a616678fc339e5c6b8ba68538364aee4a12488a638b2d5135cfc7987089ef76e59448f5c482a3ab8f0167b3508d19855f736638b91835c184f26f31a2a34af3565d2d9407b2aa3b49c6ecfb79d8935489de47d3159d438fcc5371caa37c72d3d7dd082dcb819bcecbe71d0363b5637f76f54727&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151860751Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/array_metadatas, Name: checkpoints/1152/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1152/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b51d4af8a0613a10eb09a1f4a62e56c71c49006803ef1832cbc8011e26bcb57ffbfa9b17aa4a729096cbad10e5e7c116023c6981af63c97dc0479642d5b2b04207330008bd6ae5aa705e12f7a792e4741dfbc26abbf154833f9aa3f37d82545eced993f02e1c9b8e0330ac77c82092a277087afb37693bb5b55a98df95ef4a428c07b9eee5946910117497fbdf0eee6c8dfe02859def0d09ed22b067e4746234171847d5f9a924893f14c8b34dbcf1a741d8afacdb4a7a55f722db68e01b8a3c7f57b89864fd61d0793a4c070385f11b5528c50bf8f4c7397916486ceee49ac47b9a2dd1d09d4e2829daabf2d70cbe911201f1611f1ea7127f05d6257e26746d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151885904Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/array_metadatas, Name: checkpoints/4352/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4352/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=76bc03e7ecfe30a99432e23338443f5398f505fa92aebe8b00515d1f101462d746a144d85e5c6dabf772f843949fcebf7ed4038aae0849e744650ece050aa9ca5d8a75eb136d1bd17237f322556ca546e15fd1d3e6101d6e9f1106a8588a0f31aea2a0e3c93abf31d546c1936e10cc0e73492dd879996d66484c8a0f62b7d6f985564dcd84e73aba328ae31f56530ad9a88b61733b6b7ce15708aaba4339b65b394e84df7d659f98d82c18ec09728932bf6bfad39a0bd509f4ccbc4db7aac84e60a5082ad723bc3ad31d2e9e89f144ff86923ecd286b705356253bb9a7809df0e0f30a68db8a706f450d91f7cfafb210cfb86fb0f3f7053f6ae3b5ea0d19ef02&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151878763Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/array_metadatas, Name: checkpoints/5440/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5440/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=645596c813f530f7ac4a7b13ae71c227e43a9e4d9ec3de9fb725535d674046a6dd257649726de209bbc78468c43724cbbbeb946ee5a9e3a5d261086bfdb6dfc865fb1c669b82d0b344e60d3c8626e86387714b17e1a4ec1b8746ff8e858e5777845226cd59e5d9c6a1aa646e449f5c622d9be08d4f2fe7929cd34ce50a1db776e73875f9ed4104591eb75a86880dac8d941e2d54238dcae99f0364d1c27fe61a6b7e4c1e1d5632029187f01fa8a382201e923bb08bc1745f702375dc3401505659b29ffe1679338217e2a43c704e7f065ab6b1a8ee2126c4bf33318a6599574cd721eef01994fd248a7a9f2a4d80a7541d342ccb538b7feb35b341ee8333fc3f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151921453Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864, Name: checkpoints/4864, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4864?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=85d1b439f07f46772f04f6134f002e7c892b176b40f9b2920ff2173a66e1759b40f61afb5036f7606689f0c18f3aa2b1992d2fb2b516b394258413ca9a33669e58e7a612432520fefff08ee8395d85a20a63472112ae3cbcacbb18f745203e654c9a701ba1b3f8997dc90a2b15d6659907d49505213550c994470dbde8935ac38c1798001bf5b169fb74a349177675dcbe38c33486e655a7f36a9a7ebf0cc110d0986b3ebfad29d3bc6f4c742af2648d6b0dbc13150bb1bf942bb3eec4f2cb9583e102f550bcd2590b3dc5d3a0cbf81d8e5e49e639362e622920d0e965896c026bb58d988e3e5885e6c3a62e27b13b5333cd458d95cd1202cd63c2ecf22c2e62&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151928878Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816, Name: checkpoints/2816, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2816?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5def307e1eec59d7a46cddbd184b57bad61ec4beae9c0e45de964edf6d65d50c39b71951789cad19a9fec948be7a1f5633416f41c270545418f46eda9707e6d159a3de2df73fdb72b965f110def73170efa160e8f78f0ce8090bb44201ace396c04fc7c19ae48776150ac1b0a9683e5598511ea81d9a82c083a9b998689edd7b8707f3f54e18502ba884d07a1c1ac9b471b4cd2cb1679747b0cde2d2e54eac4272d20052d7daada69be449d9a59ec4a6fdf63cbde10d40a8d79350ba911bb924ab7f2e3547dfd00811d07104b682d67bd542a575eff32adceddaa61c10b5dc010236a81719f8cf87fc5afbe91ce60544aa769411d1e90fe458c2b36e6c1f09f9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151948063Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0, Name: checkpoints/3008/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3008/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=c34b96cf7a5a2c151649e686f1315b71c0f275dccbdb7a9088d9db7032a116b7b3d3919a451ee87157a1fd86c05f5f22c9fdb138e020b0e4ca246164cd7bed5b6c40148f0a58367ac3cf7c99d0af28427b264c6a3dca7bf32a38becd5bfeccd0329837872500af232f971dceb84edfda86d368a65c680dbf7238c3951d68c6f17805a1cd437e3c06d1c66fc5f0409f0c5cde266f97374fcd5ab19a7cb95a5bc06d81890ae253c960ad21352e7db64cb9b1202c9a3386407a6607712c83f024196de34130a869523e380cbb812962db82b1ac7b28e7c84524757bb7cb6a09d01188800b6304c0eb81b95599563c96829b5e226e0a14d5094d5f264d1982437281&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151976568Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d, Name: checkpoints/6080/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/6080/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=adb0d455ad358eab0fed02c3a9ae1c0e30f5fa1b52f2cbd1f1488f81ef865a7401f4f66aef60a14daf1deaef3c339d0e1f5ccca9b526768bf8e3a34d6876cb898d3df7e19325e21ae187f14dd3082ea84b9a127e471643f92689ab8c0888d7068aa0569420fbfcff1bb0b58bc60858cb6a6ed59ea5566ea8c4b553041ca84c726d7401255c6a265b63d550698b23a12608e52c721b490a28ab07e8aeed1a3f1e2c480b5ec795ee15be09a6c2217340ef9a30b1d68cc0507b5f94905a3d6625762615c607cc5282619d6af09b5685ca80cdfb95eb9d44afd14d5f03073c514af59f5b776d649dd28cb11872fbe255d82964c8837c78293b835248748ab23b15ec&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.151970642Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default, Name: checkpoints/3200/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3200/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=957754c950a399da55edbd5ea16f46563fde4ee3777cdcdd131274ff4420d18398d17de4c8ab273ff937b0b577a44b191f3137286d2fc110d612a4b20d5e091a919615d68110c964c88d3aa81808686fb574479e6b20d0c9d1c58daa5e55a294c65f6fe6a26c1a47c5e402d7ee28025b4600ad46d23ad42ae963e921175058db2fedc5a3d80f3fed58ed59d6c65eceddfa8e049f349dbc8083eeded6adbfa63dbf3b49d67dd7e44426698ecd60ec9a80d9b9a8260fe52d7d4078aac967a760756395d158a14c93a7284d51c52318cd23e5bb2151a5509d24f279b03cbe4b9fcd69289cf4ef140a10568ec73a3bb77227975506aa141c3ff0578c5012d70f7be3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152009516Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/array_metadatas, Name: checkpoints/1600/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1600/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=0258528e15f0ee5562a644c5a8758db52d893255d3dabcab3667d25afc9b2965609c139f8b3a8ca99c5011769c18c26b64baa1ac9d2198bb50463b326f8002e8be5c0eff7b99647eea8298e3dfacdcd5145744104749af636e1f4c9742736e88bb9e03cb584f92a5c216381802fa8b0c8651e3e1d0197575df644cac62a60a95aaa54c0728f7866bf79b87e1f71cbee756af19d4ad9f87eae757d9781ca58d06c602754424a0286ca6d75a2103e164493293625b04784fa4c80ef424b8fb4306d7385a46deb478ab9d455797c5684cf3962cc67f744e4d4e2f0b35f5855a5a7870c36d1243862fc5d5db260dee7060a8294cf7ee99d80882188e42bbe7269ad9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152018583Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d, Name: checkpoints/4096/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4096/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=61c8f04c9c0e25b7931d7ce1224df805a1126214115943d67dc5565ad908715023b3b5bfa5e183db3045b142467a4195a9bed515020d018eb6e25d8b284a03f8ce0f29ddbcdd80406c72d4d70ef0da741451a7b823541f0159accdc01595974f9c2be34ac3b59a6f64382dbf9508ee07a8cadf48de47fd3ce98673229309fb29647aa79ce4d808409edb3fc33c8d4cad73a6483831d1e5bfbc29e25d8e3f3a81b0aafef5015193aaac7b1514894a5d35f141ed47bdf085ea2c40fa32f109954b6c93139f9e6c3da34e9cf915103b28b82fedb749f5df22f1311e3b99f634e07bec9f944a134b9ba6d8843e659bf3472d8fb40fbf1f360a18a763879f19b69cdc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152032601Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632, Name: checkpoints/5632, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5632?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=997730362ac80584ec114e77d73f7ffbfd3ceb8d0dde404127f00385281a5d732e38e57588da8ff65880a68aca5e762e10da8d4e1d53949b5dd980498ea9f0ada6cef18ebabe32dfba5b9588741d6f6c26a2533d277a271ba05a9fd0e87f07d9f4180b13650fffcbc0825f7b47c8796fe950b367211bd69fef6da903db5f383310a08148220c9491e8591a49b9e8283ff8dd4ccc59dff82b59fa9fc49b0c507444b60bfb2c8b863c1875062918609d7eb913a8ec53de3bb080a0d69e9457007161a6e8e312e871ac6d67e0925b85890c9df915d3ce5b25f1f92cb92154cb6443bf0e4b81d000094afbc891072816cfe6994dba02468693f7651a7c3682b1ae2d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152066736Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0, Name: checkpoints/3072/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3072/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=903603605a83b65a9cd7ee81088fc456827aa3657c6cf42de7b4a8488ef6be438e369facad5fd173af92fa366f75ee31517444ef716208e84dee9c96b8eddad480aa2f0ad193e1da426797100abd3fb0d794b28f6f8cdf753a7dc1fe8aa55b2d53f5ae24541bb25f6a65708441f8a39f05b503960cbc9f187ad5d44f1299f768459329100fda93a5397dd0077f07d65fcb938f0ad4833a4c027057dd0826a7152846899c9f5ec509a80b15a07e7444316c8533c7988066036bc6685c59385c42e4f6456a63a252e6d7d1cd9f84e8d25b48997a81b7b4d3f1a3f526e4e8080cdaace72b661f07dbb1c88befe0d8faa1edbfe3e21ddaca4852c634c003d4c1700f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.15209086Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/array_metadatas, Name: checkpoints/4160/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4160/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4db2aa38cfe5b79cc3b12aac962249f0028cb625cbda26034bf1dfa6fc6a64d364c51980f9827e423d8d90a02a382c8d50ee72b028eeaefe3b22f66b18d7e54e4108f6750cfdb48b85bf71b819411a028634ab1ef301e80d10ba8885b67602475b3a7c042e1d5948642b80e4cc24439b6774f585834aa7f664ce1b591bb731e34ce6f2f349664165690a3c2319e54ddd9089fb56b6d7223c032404f9dd06c2b867a4e2b15bcccb61ee8c9af89aec58604b808dd51f91fb217ce6eb017ea77b4dcf3c5e0d32f29fb19cfbaceef1c6ea951dfb84c71fb82e32fe2d3c44824e3377a27aedb1eb4b107530d6bc0107c162aa06b6ef7f948430eeb79b315ff99cb72d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152109942Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0, Name: checkpoints/1088/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1088/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=193fe68d772554033a62ef86f49ea95a1bbdf31386de0849fcc5e8e741531e9650f840788fe4cd2be6092e451837f8c488ccc46a00b4a58bc1757a28334cd8daf1f2c8397a4c44772382d76a1ee4c11ad714964f717f8b312354c653815f3b94f2e3d436d9be2fe5d0d83068d1f3627e28cff08efb8f20194378162689ed8b24bd98d50f064f36df4c93f958f1acd2a0ae2062e3779acd603414c6cb00a9b69fd2394d7cb24d0f857687a1257edc21d11f6bf91479323b0093bfa7409b8340a52a6be1734f583eac9bb4b6b4854e88c0c76489e534394acde98b8b6cdb17f3234a44a29f50c2bfb3246bee1ef135ffcdc7c27567426c923e903ad2459539ec92&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152133754Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/array_metadatas, Name: checkpoints/2752/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2752/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5d3fd60e56652db7db69655f6d89ea4e99538bc70f5d6773859a70bd74c261611a2cc597027bb24ac52152a2d780ea48a9d1f7d8e97515b0c135a2aa3a04d02e428f7f197899dc34d28635b3cb40c090a4db3d7597154d05585f4655fccfb5705eeb7b58ec258a06aea743f56f8170c74eb7880a2ee64db18a7b2189755b0285ff34ace3ce0b8e39b07feeb024fbac4a9b56a6e18db9db82fdcf682963e8f28a4580e367afa20d8f2dcd1a29b454a8e932c349c466fd5b8822518bb88a186a35b828bec0290b6d2d09ccde4b61631e094d75e1e7a2f85b1ca7876d54a22e802666ed33fac8ac0f7c82f5935ff1edb3d64dd5a82ec0a76c855fe7a5d432755c6a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152143486Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default, Name: checkpoints/4352/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4352/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=c1a7a191fe0af00b43a127b7d93ae4eecb2be3ce5843a4e8f6198777223e696adddaf6f85b7c09a104d878fe95cef7098bdb11e2d6e2d5aea332fd8e16ae2ed7beb0ffb43efcac838c764471a3d4c81580354edc5a2ede7ec9843cf99cb6358a911c07593850653ae7932125d380ec3dd812c7427037df76f942846ab2fb3683d73334a5cf1c7944ce05355d0434a0a272afba603e70485294726b1ec55867f9ce0eedf735d45e9dcc7d324dd74c22218c34ac345660f368d04f8727be614e91923d87656ece7c4be87cd9afb08c31e54efbc6b273b06a645e3a1a8d9361b8105bac649ce1212523376d79ce9fdcb9c0bb4f3e0fe30a9bc9837d2e20bf60d53d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152192662Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0, Name: checkpoints/1920/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1920/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b222e19676b9cf66610e0551ba1059a122e00f6b6a796628f799f88b98cca95d8026ef8e43d34051fbef21d2c97d25080e2560ca0300d99e0143380402295e2e70ba906743a18911e6e2633ad1b1f6f0da040384abb5fcfbbc2faca9782bba7d7aaba1dfa76e90c636e5651923d662b0f7fc4673266cff7b38e362f3977628a5d942f24c2037a53fcd902fb91f7b24313eec2667da667bc860a2b884c6d6bb9ac4f88ca31430776a38eaed5ca44b81b78461d23193a89551b001c28e6f8d35d4fa0c44d4e239247e51f80855873a5b7a4dc7842c9321199dff735c52de2dcadf94f2221e825a7a8a45c93b5fe541f6f23d057f41223f60bcc6cec73e4756bb24&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152066455Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0, Name: checkpoints/1152/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1152/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7775788271d2d437ac651d81970989a8e866af02b90e3e04ee77d6552b4c49c26da722564280485782399969a302bc93675a4817c21883c048d2f01db5f86a09c028723f6aa14d738b734fec5c25144d509da1ede607e7d8313ed00e0331850a523be4db09450e00a2ec3f1bc73ae58fbef9b080dbdf389d834b49370cd5f41feff3d49be9e8e5c2cf7134ab4240ae397d295709dba52ad5d26a75b3c2fc35941c7730cab73aefefb6d7773fd3d739583ee315200d8a71a0c84910604ae7d4b1fa667e70114ccd7fa1a547bd4c34679774bca0ae44a202a86176fe962f04fbad5e3d94afc635f0fa8ccf456acfe424b8d1107c4db18db3d599203b37c938a4bd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152192843Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default, Name: checkpoints/640/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/640/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=33bb1371432932a7e0f6f5b83ad67905811e2dff01b685453e25bdf96b5ca50b307f4647cb2b9d4fe64370fcf2cd03ba589649e846723958f0a1cdf86b743f603931adb1106866ff731e05c5d60251935ad4c8f2489af32eab665ac31e1d5010b7295bf0ad41135d35e6fded52b3897bba62de87a2a739410012341f73d27294f19ed6f75741f29bf385ad94fb6cb8e2fcb9f6540d631e3b470addb1ef70e09191c3350a65b75149d0eba4117579551be5c8c6b0f7795d31c09b7089ae644f0c152e7812f7ab5c0883f7745e0343cec12afcc8306bd34f8d76b2c71dcc0e8a156282f6deebbe8228cde8467c251f495ca0397f6b93e3c89238a505c23c8cf244&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152204756Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/d, Name: checkpoints/3904/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3904/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b68eb191b4f47cd8d23b8364a13b41c54e87cd974f909fce8e8d1dfbec8615d141d64a1d667c50acfacfb7553ee10677f5dc94fbf39a87091e3be707925c728fe63f83a1d9739d2f8ad96fea9b5c768626c4907b0f1af2895e4a558c21d3217f4da69c130d7ec4676f3e8f6ba2b733ec360b8c602d749068f0f88e69539844bffc35cf8bdefb022e45ce42e723d6161ac033b19b60489b652a21fc9c2dcb1c26a8d85e4580764aac6ff3ea9d8a2f216926119d6d888b51a67fd8be647a6cf4f7cefd1ff7f0cec52b732f21549e4006c8228f4b5d869ea69a49ac617879e1f255cdce7e6ae40d612af4124d9f9fdd0026f97d2e0db5288ced53ae55e4cf77b11f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157823511Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0, Name: checkpoints/1728/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1728/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=2e3d3281211ee6a061af11e5cb261afc9283931c495eec472ed15d835f3e16e10812d51b83093c340763663b82ba6126e6363a3ecebf6e3de779e7b0e386e3669942edf9a923e7319c52b1347c9ddb6c8b03353df3a43be109e5f64d2d77c9d09d5f96c36cae9f4e7465bc577af5e5a3665d3b5b43a86db13b112abe1648f2037db0db974dd430e33db9358f4afa7cd6e4d8d77a55c1e6c138142ce2ee6950200dc436ee0155413ae8e0f45fc719f9729d4a3f3e94dcfd544a45027d0c7634a954b508a99daccb28cac4e44e7c27b535d5dce12cc15acc415c0f28be8a8e936751b67377c4072e6f52b6edc0ecb78a4ea379f46f56dab5482444d94cae50d37e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157834907Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384, Name: checkpoints/384, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/384?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=3ef6f96bbdd3a804aa15340dffc3432cac2a240e4c64dca57c3569475f430eecf858612bfeaa0c1bde800937d863d6fe89bd47c9d99de42e1756b36c7013a4e7e4ce73524b3835e08d0a17ef93237753f53fcff2f5f34aeaa228de586e6b734eae60898b4f4459e2f90f55e4aff2feef845562a8c835511ca852fd083f35ad72ac68dca45efabb24dfe1aac1167a544be8d621a95afa8cae53b5ae001e6a9609489684ee55e7e5c97e4d1b0c8780503d33f3c628bf8802fd4a37c1e67b0abfdc38b304205262e223e8848de0e9c3c21d0dee13318ec1ac3840111db0428c7de76a5a939764a1fc7af7d4adb2dd7526408d1746fb5cc993905fad268958241d17&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157859069Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default, Name: checkpoints/4032/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4032/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=76aa7f7e5943a49859b0f6bc68121e0bc9e93fc71a722800403d3f87593aebfcccfb9843f06e7f2a2d9a5582e1c811638a84f784eb473b7fad9cb8eecc329af04f8378b127e2307c2bd5e063d640b484eceb6290a5f9141eaa3414c2f9068a7e36eda5762e9997f7284dae6bb6aafa3f99a41bcbe621280648a63614b8db65da77e604a04b17b4355ea52a0a5f4a06a0046c040df1b5be0a19a68ae31b62fc4367212e27d45801ea9edf0070db9d8373d9ca7a4c49bc28ce1a3c7ac274d40f330f86b261519d38b5df658e4c116d74e0680f372e28bafaa9e55b2fa7d95ade365a3f7edd88f918eb44ff88d58e8a032575b4c8009407b9a22e807b5db36c56fd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157880848Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0, Name: checkpoints/320/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/320/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=587b88fe741e562b7002cd85eaf79e198bd8b8948ad2c263698d8e442ab334265b0150f02f4f9f43e073891296d85210087ac3cc1031545a1b2b975b83d3d1dd6286c70a7fce47029770371ebbee87a7e5b7ac48e9eb8225dc4e543bba268b73fb0b4a598e1da91e67e66c60f58cd22024c2fd287ae2d968d3f15c424fdd2321303fdb4541ed5b7127bec5a3eb2453468bd5c635ce8002bf56fb3b1e24db7d49df8063edae2ca76972b18cb6410e77d48c2da0b46200071bb007a4efaa2fe389ecfa7192741085c40589368875e8e766a6cb94e6bccda55c7e6bdbf1e187edd5fcfb3f8fc43067c5719c5f38225ff9fbda428061d5de4a7b9f51caeafc13906a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157899581Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/d, Name: checkpoints/5056/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5056/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5d12531caf4e4d71ef48e5185a51d6e7bb4c704718c476e4ef8e24ca7982bf21a738d17f5b1c7f060c69a36393385253fb5ce07f946a243e5acb683f354d82ebe3634cce1f227af96a41165031a636be8c4f908e4a728ee39e2587d523296fc14532d8e74adccc078413da1071053da042cb9bd3cf045c79b7c1e58504fa660f215f15b3a12838ca7355cffbddbef7b7de3e2d36e753d0d2d5918f41962925b12e195e7c55feb5874f5dc29cab540596986080d3201788bc3f050982bbedbc6f47fee37836ae8101c083486311f843481df669171447c5cbbf5f7abcec5006a7bf03267bb0b4c8e1d4b81862996a2f87eb8f929f5873c3c43122d97c1b6eb59e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.15716845Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default, Name: checkpoints/3584/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3584/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=38f2189a3cc51f9b45da1978957d8737a3636ff4877aea3b56a7a1238022817cd3762e55196f66307153e10a1646f8b27bd708a2a9f6ea58d3a5303f298de6a174b2abf257aa24fa0e8a83471821c9cda3e836d581fe58d69623e355d3eac83f3302d4aac6a20867981a27e86ffc67e49252e618398b407996d51b0fc38998f95ca875913581852b30c521c955d83291c1609e41b0b775f7d9e8d8f05ed40efcadc68a740ffe2897e816e2932e42d2e2f5c68b9db89ae43db7a17d5ae75bb55539bc7e3588d28113b429b91c2eb8f5dcaec30f2b6f4e272b1864e82251f727958ff431b632fe1ec29507d24d7e2049d15a29db270cbee83a8bd1ab11a750c9d1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157506067Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/array_metadatas, Name: checkpoints/4096/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4096/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=2a9dbae1e11e047ff85c86c457a76563f771a7ae77d96aa0a42a11845e890cf5354b9c4c131dbf1a4a79c61e27e6dea96bd7057cfa3681296635a06b585409d171bb3c2bea6ed78210b3ee59ae4a6c1c5c6b6f47496adbd89f5bbb47f14988704151b5016f94eaeb914de7078560edec1f680e367af165013a58a5f7c790d19e27ac697ae81900c1d06aa15596cf9d2e6d00b72f739654668965464b72b4ed6d1b74a15a120c9984be2f0f2635bd6d71ea4184b3c4c57c4649fa326ee4112a3df84755703493e9f3b7d239df4a2dc7a8be8abc1da44d113a8451401ed2528d3a920979f725c8755378e4050f9561d1ed6569a3cfc64f14b25b559acb9e4bc08f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157968279Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136, Name: checkpoints/3136, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3136?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=114761687c2705b8d55c0cfc47643b64ceecc0f8e3e18c1a17cfe2018d696f76d2e0663ddd641f197d857f9f633dda811f628b2508d4b40330c6de468cfdd90007d526135f15506e0d53c66b123e13636312ce434c6c60bbfe47fd5adcc0746eb61c50bd51d9522f7170934240ef5a0f9e6760af5a4ee81d10dc5afe0db98393e28f34911c559913a703c0bf10d07545ccd82a9ec12c92d447d878594e01a80e8ef29dc9608182a018b8239c44739cee56969eaa8d833af3ba5ed55ee08c31144469ee8f49d2c3b4b635f356b9716c58a51f39448f9bcaac217c20171b422fbc31b84d5e7407511e78b19772b128164b4995c81c6771b29196c0b3cc036e40b7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158008081Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/array_metadatas, Name: checkpoints/5504/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5504/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=71cde9f4527aea6528d93ef44e837bade8a4d0fbf46fa8627b30ea3b17851bcf1ce7b217bb1184e6f44867fb3da8cea293eac50dfb227bd1f737f9997d68ca6f8f60b9428043de1ff212204607bde7bbd2d283d89fe6e9b651074d11f277f678047cfee129161508374759a3ec454dc3ad33b6e03ee0d5681038e88367482121dbfb1fa01b8b19dae9bf4fbcdd6c5bb356c096482e420455090a4b81c4783a8628a8d47dabc6d76e47df91d29aba1f5bdf4bd2a112778257f168bb461b3b59e9ab1f948aba8af4112e11c45d75cc8efc7acba1b9b27caec3b4b5af564ea1798dbdf33e25c9d10cb8ba017fe3951b5ab93f3b8ffa8612220925cbad5fefaa71fe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158033335Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default, Name: checkpoints/3072/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3072/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=946006e7f36690aaadea7617036f8433e355530ed79717016cb98325b202e409e928201472d5d65200c3d165d157c6b2c2cfce38de22b67597ed77a49716e66f71fb25688855ee551065360fad5bf339a935331b314136b60f0686cc7de789181c5771e4a243bdb8096ddf5f9e6b9b07855dbeba22a9a958ebeb5ea5bf13c1cc663e9221aac4165127510477bf09fc6d0e09435cb4db4c302107c2e5aa6d89618ba09fa08a91f5f67b5e5b1a4a78a26eb130e48a8dffbd3269d12bb9d5e68f2ab7ebc05b1acc9475d419586b72b5e471eb98e166e43a0c5340c096116a1c302ffbcf5eded5c8f756d2bde837b89b97449d2489c93918c6b1fa69c8d190d15f25&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158074341Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352, Name: checkpoints/4352, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4352?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5622cc7d986bc0fee12dee6262e4ab4d4a77eed0d3797f4f039c33c66c0663285188177e91e20366466b7a3500e973b60b7892603eb0150516357af9b57726400f8afe7edd62ebaf48440eb0b2bd031e5b8c55245405c08f719140907775bfdccc9e9140b92d4c810d9427d141b521b9c7563b4df21bc685f07d38f1471a0c005dcdbfbdf337eb51dbb539e74b970e3e4d8b06621b6c60b7c8e54704ad9c68a26f53b90e40de5b89b2663250f046bc895adf960a994af548ae1ab8be101234bddd1affd9ab804d7593dad1449ee5b7fb9b4227a9598f215bd95a0c57b7a69f21df942397ee272788d08c491aef4265f0388d888b77427310cc4a2c7e997ec36d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158102887Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/d, Name: checkpoints/1152/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1152/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b7caade0544e076d118417c54fb0d55558908c9408682a6c5310d94476bc55fa717f97df118e362067512fcd60d63d9c85ba744d707e07da26a0a9ab2d27c07887a3907674a5da4b03878fb49e04dc7242a03698590d2f70c37d3df188abe0b0b09303cc7bb477ca69a370c391707af1c6e8f45e2d43900f721c0b8d52760f05378508603449d009d51b96f255a6931bfe1fb0f57e2c128b04e39208d91c5b28dac09ab49ee4468a43756fec94e98364010c304123864cbf486d6bca1903a51ea69c929832b6fbbe2101bf6cf0902e8b2f8cf486bcaa9840df6e14043eafe77706766250c823056e90656934db0b4b9e0580567c492ce7623106b46598b0b0fe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157407482Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112, Name: checkpoints/2112, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2112?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=2f3d686fbdf69590376b861a64beb1262d7863cc0681a686f49496683b2460fc810d8b35447cfd5c590936f4ceb70fac41bb7ff209c607169f73541d9d82bc7f248b942cd8ea25851ad049da4f3081df5af3ad028c17154ba323ce5aada70cec4b7c42827f9f7292b042a073e053e5063a6e030af9c9b521b7d6ebced9882dc12880dd2eefaa22607972d6d6ef92153f673ac561a90b7f2355e24da6c7992ee2217090d4f29d21e5b4e7100da789f6071b90cb60114db4da5574eac225b7230a576fea88d0e7d6ba650668f151cc56c99814095fb3f32f292dc833c252f71c4378644669c2c3d5b19576aa2a214c1838059e845abc636b2b5218cd8fc52c5419&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157465918Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d, Name: checkpoints/4160/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4160/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=3fc7725c044d4517673bb24c1ae4daa22ae4fa457e69a391f0062341d9d2dac5335f6c80a1e90be33e3feaf41e60a142e05e40967bba42424dae0438a0a5e4b3e4e3694e887614fa7718a0bbdae4e1de2f43ec0db6e259ba4dbf4241b1a8dc537ab6bc308e6fe942b76d3aabd814e6a833f34d6eb6d0354dcade5e19dfcc75adaaa24fdbaee61dda1fbd1f93dabbac6a64b9ad364d32f8c686d5d3610fd8e5a412089f4e200f6adf13a9cd78c558350d877b76115f24cf0a4adb9ad3b7082390dff05c6cd8c547e4f92b521d2f926f1210b2d01a26f0fb90eef182f779410c583e901f327b45068ade0bddd23b3f0ef7733b61858985b7057c1964d5c3ff7695&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152262197Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/d, Name: checkpoints/192/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/192/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=3a4be13cb9cb587a9f2900be8c044a9a024ee77ec69f3a408c90e951fe3605fa3674b2971007a942fcaae9abe785cda9a051c6b31b9d232af824fbb04b610b0eaf0349e8850dcda0649d33e6979214cc2e40e4907afb42b8640258867b85ed943a533923bba6f21816f4f2ccc9695e3be85442b1d726cd39f05a313a12bfc9787fe59c699d189505f2121531ee045a240feb1299276ff53ec09037edb727f3c3b8456b6f524ddcdfb5dc28215b01f75303ddc426d66985afadce9a47ab6b00bcaf8b5ef32c1c67a684cf6125a5476b313e589cb61ca1e4a0529ba96a6498d57d9bd6397658144ea94fdebbe1ea68116946f5d6bce5204896536ed1a5f5e3420d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158233074Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/d, Name: checkpoints/576/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/576/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=47cdac11cc0dfc9273b0b6a9c289cbc9b47da3ade8e7f32adfb4a05dd6f275d3a62358f7cc109407fde69a4df8300c7e9bfeca9d16aea8552f455297eca48b28fa702d406f5f67cfdaeb807c7973304bc113c7377c5823114b2454b97f0de37ceaeb582a693218d332920bd9e8205a1eaa41b8dc7145db0526d5e16688ae26729238e89fbe066cab1ae5d6acbab127c265c88450c878d2aede5fa71395762f1e4bf92e74298224909e3836279085c69aea60c4f9be35493091ccfdcf02263709530c6cab2ead9c30f5770412ee3ebae012ed9a94b889df303b7d9142e243cc3d044af4c149b74ef34f4a0834f83945616d22ac02d9f24dbf42897dbf8b101c7f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152241446Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/array_metadatas, Name: checkpoints/5248/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5248/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=0d5905ee0a3e825df1e5e13d52ab528e976256524b0425e1086f993c44fb8eb3010ed4e748ee21c4b0090c8e0221ab883ddf23990c674c3de6437804307a4cf7261b6902204d57dacd944c1abd62bc55a587c199816cea99d2d64acb4d2fa4d0be27a29b0e1f71c3652934363c64b149f7b5f6e4d169ffe60c9ec54e7e961adcda08c47a0ec1e3db1b9708ebcb525cfa729364ba886a60b5ec8d39d8d850d82104b60d83f323797dfe072f448a5195871343c31208e865556f016a64db518ac915718c0616eaa770fa32a037678db263f305ba0adbe1cdb1167ee66c66554a410df1273d070ecef6db48d7f071c5fd1e9b7a4b2ab90a81070820d261a26f8724&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157628854Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default, Name: checkpoints/5568/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5568/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b5beb095d2f527f7e994bfe385132d9f99b1bf73a15285f6f15f6f620d59172b5f61246dc17f8a0477f9c4f60857bd3de47db9fbf6eeb6f749c6cc7025f01ebbf0ccbd57aeebaea42736554fa5aa6085c33f1b0142af947074e1b6ee192aeadbb093b2a33b6a9b98986e2b3c2398a504c1c315ad020b8b40d24b658b81a55884c43415d14faa8912ab0b9f23e46ad5d06bdc016c921db4ab374be63cc5e8d12f9d4f9f28d9c5d95c83a3f280fbef2a98cabd922c3b35d7c7dee65a56a4f0e83d7f5b8df0a9632b482cd79a962351c9b4270a1e79ff90a715e7094290d84c5939a24d9e7d9a9e336962c977b4ac24f8558f14cdfe34f800886a553ad00d47d426&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158360199Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152, Name: checkpoints/1152, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1152?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=256d66a1cb71d00e24d3d981db562e1bb08f4a270e400550c80d5b5c03a17dd34602e14add50160f4a87b01f517aa2b4af673a1b34aa554f7c9f2f506512d8a3b694f637fdfaadc256fa13ca53b5b71a7fc3c80cca99351abb624f5f9f8128afb133ef6417cda50b3013bc262696b1e1de9db95f7510d32d272225a33553594c01fb33cdab716c3d5134085d396fe62968ffd52f33710bf26962af06ef5a709617d21b9532d1005dbbc5631a81199ece2dddb503b6894aad1b192121d4a38a1c8afcaef2d9ac317ae235bab4604d6932b3f37545d5f3852c18df81b2fd1af94096678bbc05ea8ba5d3187f475ac652ea46e1d7ac7d2ca6a4e5a82adc8d705836&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158434447Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default, Name: checkpoints/3840/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3840/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=48702cb52bf96510f8f0a390dc6c54bf965b322ebe7f40b0c37a13fdb92b042e5b7aba37e4fdf9764d2800b2cb765be88f0258b0a44c86bad2b2f1093b31df3b51bcfbb19ec9220795fd60aca7e30f43feafa223e563a439a06642ff9e023c76b0192fd2f74b651f382dd725cf4bd0d3e61abdc52528f6ecf3c436f70b9559074969aee5b73de00a70ef280de95a689bc5146dc98e228a5d079b88c9c5c07aa7d9af753729ffc65ea2a81a559935e33dffdf45e1b184c1fb398a1a8635fe921a20d0d52ae3904a4001e5da5f0189aa0f0e51af2815a300bbc31d5ccb71a72c8851ba049189d51bcde6da549a98f5d7d480f8867823e0433bbf66f88a870c2425&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.15226935Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0, Name: checkpoints/2432/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2432/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5f4b648d5bff6b9cf4d6d888bdbe0cc91db668de4281c60f4bf22f9331847712027831f53deb80d33520e12b982db66cd77f5675901dbfb4cf49adac37cab767974f6cb85a1aa3f95f79c780915ea159e9274ce60c3e152de0481dde165c72e2868f9cc93eb593bbba4ad6d8a00a9b4dae49cc626ff48c5c636c3996bfb2f5551ae29ba0d3b96a7c438d37f0bfcc9e69a7dc5c9e289dd5efca93493b524bc4ec0fe134b8c63ee0ad0cb34e35e1145bd1d2c32e97d6352ff92407bc905ee7ec2d498e0e5453216240cb8ec0f3b93749107467fd3eb66c34053b154ad44ece55542f026b6eb8a53552b6b793510b793ed18dba16602819e31ecc884f6e1a6f1a45&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158512275Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/d, Name: checkpoints/5440/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5440/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=c50adda78ecf0329428e962aee8175929cba1fc91b3ee23f46826fbfddec6d822392b3e657d3d121847e832a08e29c8fd821f9e5f089702ead1b8b510ec8105f4bc67a08f66c1f6203255805f2023579143e9643dd65d309a2caeaee66c1655379cfb00c6fb153a445491adade61093ee5309a5ec85de861a72836567f182cefc31c44d53dfa38a31594f54f2e1898a489a514b62eb4eef12cfc851d362a2b917ab4e4bce8150f6123303b7e8cf46feed5f72bfdd8d43543df12452bda0c3a5b55d4c377bae60e23bef34d1abc989b981c999656a34b1ff20f406d57841b10fb3b58c759874b9719d87e26a4d9f593ebc7f338cc6e6691441e0b607a81ae06d2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157893342Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0, Name: checkpoints/2048/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2048/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=07b4793d48ac68d7269c908c89bba969e8c39d497692cd329c5f4214dd84ad5be1a465acb26af5fcca86f81f6aa9c326686bb27ee4060189a039d123c2999ee9e06a5d553edf783fa0a8122acea000da537eb723f22c5717424a3601fd20516ed242b00bcd1f0d85dc9481917dfdfd8fcf6bf2ee9b0ceddd4eaee15057781785a1f7937f50a1a42777e4fc283c1bba5a01a63bfc9700566cc4500f3a27ead4a28750cc74733ec9810f07ba9d6a87847385e7280ab14e2a794db06d9c3003e3ded4132292ff0c5adb59f719c20bb21709f45faf45b2f54e79bcd519f8f2edc5df994bec55773f1f8ab2baa8b8a08c99f749da653a74e7e1e985bc8414a4b2e4bf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157871027Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904, Name: checkpoints/3904, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3904?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=0e6f0355dde545bf863f163d2457394a5874fae35f5ab90d57617b9c95cdbca549a24ef4a5ee0b682cb63af358a6cfa04ed65a3ac8f8c95b89b90e8caaa9dc71c99ef94c7b7698ef64d117e61202b7b37511e8faf2229217834513b4f0a463ea79e58e0cda5267a939619558132918d0f59ccafe0a53a9c90aedfb86a63503ab5811bb7abb09b5110a49e6e4946d93332c236a89330f7307cdd191b3f33a25c1a60e591de3ce02b0899973499ddd498484b447117f2dcc5f928e937b8b8ffb1badc733b17b461630b7aa2ecdfa39e9e01f3c7c55988cbd0a3d21117b56f57cc5ad1ee528d7920fcb5bc552d6e8fbe0197d628892710138c1fe1b17d352fc8f6b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157539978Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0, Name: checkpoints/5632/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5632/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=8746c8e2b37976fa809a4d207f7de1b7bf293a8b1f71a8d8718ba5672993242c0a77121aa8fad15a59902a912cccae8d95643f519ef6a30c5cda35d656ece437add1913a6ecd52dd167ee5efd4a2b7c135353725d4457c80a515857755e22b7e2ef4cd847fc066e95f4c6d80077fb00560578fc532c2950111b9e034d41dcaab12e14d5c305c5ea8117ac4f2e8c86f7806773922b327a8d3f48e14c9583ab7f031f23b6fd942e13108ba939b640c775d1693f05381a1187fad1a5947cf3caab71abb3ef42cc084c42a00f02a0df4aa66400842a0049a0f2104046f36e3cd2408de9832556e03e7c2da7784d663edff0ec862cb99201f5ed0aa2cc9d3eba5e915&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152318714Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/d, Name: checkpoints/3584/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3584/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=29d874ed689eb3b467aa9a956a082ed85e7d5617cf5b456b4ed9aa3d3e1d58daae16e9b6e22ea06d0050939287d0e3172cbf93f0d9cd77c251c6cc1148962c77ed5a7ac85e72e4cc6f78ac79b3eda9f0195a79b1bc98dc752cf0a46a5fa1a443aa48f0ecce9360c2b4ef627687f8dea02a726277958a41cad27d8953c4b08aa989d94da972e89293a68008669e0abb15c7fcbbf6c2123b9318b00db45491e6a27b33b3b71cbae0282ce4bcb7ef7cdd4e144655650a24f3d6cc99bf96b68ef3ede58197cb8eda91cf9f27359453eeeb586b3a6c6f1c8e21338d7f8682c3d16a357465e98719ca2dd001d33414a81cebfd0de9e3c4a7cee5141f693e7e54f7c0f5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158659227Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/array_metadatas, Name: checkpoints/64/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/64/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=0fb3199639910e914caf4c74ffe1201028ee439bd7faeca3c4e8b16250ff30fcfaa06c298029f80a1c620f4328c8453cf3a39f916c68d78bbf6fb6537123a96247e071adde2aedb5821ad35e60c66a1e5a99b7b81599ace55eafb93d5707cd3352a128453f4d3df18597732822b236bdad9d4323278d03212461027d1da9b068c0e0d549cb879039ad10176f24c42db72f570b0f9840b40268375b52a81637d24d1391ba7b3a601c68f8df1500734f2e5594c5d3e502d95dfc0fd84e75ea33065f7b177f22c0272f84ae91e6eca62a7b77f8b2981e7ee129b1470485cbd486ae75ba9ed270d5a2a6c53b2fb096e98d8ff404a712e76d10999cbdfd15e4a45160&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158670796Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d, Name: checkpoints/3648/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3648/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=a4f6367911ce8c35efd090c1bbccefd1d08c329140c6b0841219d04c832bb256126ebd03e663432c269854a69a501bded1c3cb4cf09c813c3c596c4f36d1cf980672032ca94c421964afa1c527029469f7855868a19fa8e8484b77e6063e7def65585cab09ff13781b98226e2689a2f4c3fa58740159b5c9105bf68b86bfa01c3b8ee145201bd14d6e2ef9991d7048d5a746f7666a4deae854e29b20df64bff18a71b97baf75d7723f546ee52f4cede9fc4c7b785980c2c9698d06f43674d93f3ffd7e67535e3e9493a9c77f67298706bdc40da1f5cbb56cbd14c252765e25f7e452ff46ac53124be5fe37266d68703399e0c9b48cdb964000b5d506b6504c85&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158745585Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/d, Name: checkpoints/1728/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1728/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=bd66e2b1df2772379f54552733d76c8731a3d60c85e6dc82205835b0a5bfe833e12f980f0e0f7016cf6729e0ae88269d3369acb43f5576469758e96da5197dba32843fb4808159d98cd7a8f8bc7a5e5aebbe43bc51a1ff166f78ce915a2083d54a043be60a77f57ab7267fb79990da2f589b29dfb32ca4d7fd4190d7ed46a8889c1be8cb71c5012e46d8537c896de3edb489c27e6e510a96a3ccbe67657922c01d9be87b305298f30b2548c7afdb9897289852614e5e29301b0410720edc6f4a7cc610f9a9e41f2b1a67216d7f31abb4e8ff00fe23ea2a8180c1b8a574c19d512003baa85292e05114269c5a0f335c46d41e98252797e085aeca283b13c6b793&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158755738Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0, Name: checkpoints/2368/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2368/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=1e084f5d8e01ce541f484d90164c7e53d6688cfd056d241251371cfbaf8a20b1dfc10d4bb9acb79d92e4c215dbcd8467715012a297880bb7c9eb5be286a5603def331bf0533ea74f050f1ea5b7c03d4505b57f49a1f2afe786fba283c8f4c97a0f463ed83935a7d96dba799192b4912684f271e230a70c528604a5f45e353726efdf5aa969c3adff674fa533771b19d48a4a0b7162aae84904c4f26b05e5a7c4d2741a88511ca6dd82f223e8d9e411ea2ed7b07dfcf64614468af0dd8e1370bf062904a625c9430ddfb6899593b4eba309213b5266acf6bd1a5c5831a048def8b493e516b04dbb5b7a12e3c20f610a7123c2de8bef0b25674bf40d7658d81ef2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.15236128Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0, Name: checkpoints/1024/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1024/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=604b1cf239ac2e127ad45a1cd813d01045af729b322137a3efcf8ef10380840f4c0e00cfc216401fc182542eec99a6b6eafa72836abb6b12d8daba97088841b1129c2cf475d2cac3e6c36adc89d2f9636155b1b4c27adb804ba044cef8abbd6d4c5caadf0834cca2589f8d638b5551ce440362002f2ca8aa148ca8d46faa98421a1481d707640ed5bc09ae085203dfa59e4ae9a6b024f16b21677bfd6b0f2f9992cd59857e9573635f35f9aad08fe7c4645c693de89a357198beb9c24f41d040056a9b5d1f365563951442e4c385bf6f8557ac2d606494c7e130a5ab2802feebe1424157e35411b1ecb5e723d42b8416a1866b7f3e1adcc9099bc1095c12492d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158842733Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d, Name: checkpoints/3904/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3904/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=3c9c81150dc3348bb580cc3a87508e3ad4ced6c5561a387c068509d68da5bab2942542f52944ca83b78d25690f0d3998a42cfef4b74362934eadc60cda312956d460a02ab4238d3226f4f42fe4af0986182a56f81fce78fb67e7f9636736241be9da62c9ba5f876ba46ea3a8475517f1cde69dc2303edee8a235e212d8bc1ebd9dea0947a4b38f026ee19c9bfb0fae34a515535b5619f3b911caac3ca4d2c42e6ac5838796b314566d08854f6eca3085cf12215e46b659376e06fadbf2d69964e67d2a7f0be71eac7115ac651b81abb6e98e94cab9fa5719b65f4a3dd4e25019f8cace9463a8912a69233ea70af5e360312a2e0640e2084a3abcea733bee1259&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152373057Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d, Name: checkpoints/3840/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3840/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=01f244c9899b2cdd4b4bde560fddd96163d2b01bdc0cb44740bebe02714eb439e971833b441f9bbcbcbd5bf9d0a64cc4bf5748cc51b6710f8a05abcf03de3fea6d1e6ace48048ad6b746830eee4ba1243d8bb7c05c34eb460b6782e02f5c9b7e9e1163ed2d1b42ed620aa6d83d7e0eea6fb13a7159a48e1b1ce24328d7b97441618901dafdfd3288d02af26fb2fbbe632473206e33517aeffdc948753dca05d79d4ef48b0a052cbaef882d937ef0dfa8a2c70e09f4a0c1ac7ee0d3ca6ab03d06fd8b24c8f2329c6f41a0d1ae54962f0a05ef50d0c93d159a5c195dd230e77661d8f3f52d2d973b43af770bff31aedb72e89dd4830a73a561cea3d6a5a2a67a5e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152275513Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568, Name: checkpoints/5568, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5568?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=a1754f31b31c16965a8dc1e25f25d0ece7addca217db9303421adec6be442a9b2b9615ac55ec5a562454a9f3f5498c60789a8ebf515bc819c64fd5260dcc4b07b99abbbc9473ea2923027132f1f30bf6f9f899a1c0e23f0e3f1e3f97575fe734769da1a1213e5f1e50350b45f2347454899fd0152b410a30bbf49e273eab53dc703d06913e7574a807252a62c2711a955fa039e073ee18f253ea74eec6b3f24dac1a901755bb7f01e7e214bb5ce66fb088356af2f0427b2fa6c030428840e718544487f5c2db630084876c3cec4c9b190ddafa4d15bcba9cac712cb888e2e07b193caf5c9d0499e76891ee3cf96f1c903a294a99d8662f55911853554e349f49&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158944914Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/array_metadatas, Name: checkpoints/1792/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1792/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=328e1220706a6f1e8a21c5ca266cb9ce93a3f5af9335e6d05d96b1dbcceabb651bf67f143cbced792f8a31cb2dce540ed19cfa0bdeaf3dd4d969a9b90edeefd660168e069fe44a1d60dbb5db416abee5a6df2c83832f621cabd9afb4bd41be672652a469b726c0680e982034da47d2ace5050cf0224e56818f5896363f81a4e7e098d21706bba9b50d13c29b134d93988ecc0409a3d5c8f0e798f61084104f033e5754c7d693d82c8b53019d127cf0aad4669f043780ee2c6403bf43ea85113a542821ee426c489ea0a8a372f69126364033dcf78670771dd839633f070b0ff127883c9f91e61e2efddd7d755b7da0d71b19944fbcaadf2828029dc0649cffb2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158938047Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176, Name: checkpoints/2176, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2176?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7186448d6dc9907687d83d76a9efc3f2d63de8889276aa660751e1a283cb57c206191fd4359cdfa15f589036ea33d26d5f26878e8d342bdd0ee1074318dad835c353e8ebb69265cab61c28c2ab79af5fcad30c887eeaddfd35f5ef09a7e5db3cbb89561daa2adc39f1a90cd4a540744121fd6e161d13ea72a02b22960487f1bc52c1acfa17f9664dfe5e27dc1e9b59899aa4bff7b1f7a6fab1aa70412bd63946233b5f91454d9d8cabeb9d2a02db86236b541f037d35a41d4fad59403052c5a51d81ec45994597b8fe65fdf3b65690615db2f4cd1347dfbbc5a56d9b55ca788c687ade523cda8a9397dc55508317690ea5bec9217d1d4f41418758d59c082638&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158967205Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d, Name: checkpoints/5760/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5760/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=38dacf83e9e5c9b320eddf02c409f3cf5e070e00a2e19fae0d816623ce9b9b11e179689a5082c5fabdb60f3d824493701b12f37b2d93664c5ddeedd7bc6f31aa0921f8bbd34b0365dbf62b418dbbdfa3a615bbf5d3286c336a9c68348c2c40489f2ae7dc8746bac62eb27868e0805105baab9fcdc124a731f0ec3d654c7b0425bcf0c0f6ef65d9cb312d99d877cd99b246804810f0b5a1df48375cb48c6c34edeaf655d4759ec57349d59a5f206276dcb6cab3af6f9a5e398a22da000e03005b9a7b5bc1ca53ee9e6bc10ed1f0e9e1dc50178a13a4453d39bc2c93c1c67e38c5391f619f605eee0ef45fd9c03c555f8b1374aec7c36e601f2714a52a92eb9b38&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152355Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default, Name: checkpoints/768/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/768/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=6e34abcec034d1f4cbcc414e30f6046de920da7e42bea4f464a9c9893e3d387642659389f1fe8290393e659ea69963539dc62a77a7a5e03542bbe1a789d8ee5638ab9fd97cfeb0cfe0b88bc8b325b7c2dcd508016a1ef46fbda148a0159ef4bcefaf555ee3b65751678965e2d7023fcc015f58f0d8e81ee43688eee1bfc98fd5963eccc3b33f52f885dfe3c295dd88ff3cd2aed29afb121b1c402779bba8d25a27dad81ae359bfbef1735d903a9d2041e4af83ea67fda99b3b08ce899349d944ac735d13b75650b1b5c4b2ad6327a73daffefbd08cf97ecb6aa4f2a7e922407d5a4c7bb8218c2604cb61efe3f13a3e76c76ebd74e2a018c5800ee000b8243109&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.159009926Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/d, Name: checkpoints/1856/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1856/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=18fcdd924d952efca05d4bcb7997259fb70b3875ca37a84547345bd197fb9fea3d0fb8dffb2cfcf382d5e84663d33f6f7715fd741602f69806bb9bae9b563aa025a77ea756a945de0b7e006231f5aa9ae10a3610595100356bc22e43337df6dd24d407cc0b55e4416e37a80ed25056b974b5b71f5972ee9fa55d3feaafa5954bf93bdb7b9f9b502b32802fa3bcf075821ab106e02fc7ee046ac26104edef8792c02b7168417025673a10a6bd9bbfdff0569621cd6f9f8610ebae6a71eab04263d5a9f443a4800fd8b3a242bd13de313351151107ae5d1208945e7f55fe9be6e64ac92aa9759914a82ffb9bef1cc71438db88279f78be64f9772a7d79c1a45036&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.15247657Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default, Name: checkpoints/2112/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2112/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=3a0b9f0534e05b3446c0171af837541d156b5669208e286954ec3e9c1cd45a9596e5cf7e819b63920b6915c759158d2f3f3e7b541832042cd4eb2cde883b709552b43df0cea7bcd2ab22710f34309b5e68846ef00976c10f3bb9077d598eb4972c0937a25826a0fa1c35d77e598df24d04fd633e402d0320be4cd4dabb48b93501eb584ded4e4215e8e3efe88d184e07bed39a41f82f36a6faf5c0b3ab79085692e60efdf5f5085f153ac357469c3c05e9ac056292b6a79ff0def7a9e1677c6cfcab99d83d67d670720daa3cf32ddd6cafb4d91a60bc9b18b91b4a3e77b4e243989df41ffba0c275f1f7d4db15be23eebc166ed856c34d4e2c402550976550d0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.159089348Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0, Name: checkpoints/4032/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4032/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=6326489baed0a44902f8afbfaa11ebb1d9ce556c7f0d4fe8d001c1b4f9f137794ef17042f950ea4a2508d50d442443c406534175c192743ee46169cda928b26afc2e5570da5e2c54abe87235bd65660f19a1f4b88d75b210e868e3513a7b2b81307afddf447db56efa7fb36f8ff58ae90a527afb161d2a03ba9308095969089884cbc820a0c64d32f36603a6e2a5e789b173b0c4b105fdf18aaa13606401580b7fcf3dd808179aa2b35a7c414bb3a972f219f35a30112c23528ccbc0f2d322d726e22403d8ea08f5aabed2edbc0ce6b707bad56932ebf1823c209629ada1963695517dbb50832593be93c3e7f8d418fe7767b93bcaabde044d65b44ac5899fad&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.15918454Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0, Name: checkpoints/3520/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3520/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5006c8c752b93b2cc2955f3467bbfb78892671b8d2900d68d2ed9d1ca205b00b45e1f3ca26e5e1662d54e89bf02c69408cb093d2760f8abf22b555384e6229bb5a4d2fa1eda1f7c4a21ce4baaba8964bf16090d5af652deaa60ca0ad75a56cd7f07ba2b682259f85b95432adecd730442f1c1789d1aaf6c90a78eabfab6b5f6c9fcf1c7ed3ea62dff054c442d41d6586c0b66617417dbe414ad706742a8bbb2cff964c551edaa5377aeb4b5a8ba272835e0874ebc0bc165726e8c7212d5b2fb57cd442ab6d5085c8f1fb8ec1ffe41c118d09aeafddd6cf18f5f02975bc32c7c3bf0d9e40ff2eaaa315235e28a42eedc0c4c15a021b62cf0cf5e00db5c2ed68dd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152406312Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0, Name: checkpoints/0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=c4dac8c0c9a55541308c6875fa6461e7696948cf9cf45701e3620e561d0f86b0060149789e6a4e0bd39b636e174fa09ba586252f76189bb946f6339baf02c7a54c89d074f273bb5cb685cba234be2078df8a61e03e0a644bc7969f220c5dfc3063099911b3c0a4693877e7c9ecc5f9be038fc08cac5d0764090c46468d737feb5235d1ee590c647ab7df4b64004f72bcf02fd094c3794ec4977960a1d26d6933bd15684a104f99e438d9545a2468c32a7f1db724282bc222df38e70548003dcc4f961a7b793abf07c52a30e87781e66594114300b8dc090f789b4bbdbd46d46dc7ac334d65f3179b68fc891709c14bb54992f7ad9aac7668652be1655e0f6e0c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152424418Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0, Name: checkpoints/4864/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4864/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=9146d25edf7aa78289dc29fb91663fa1a4cc6ba66a286ab3b4fdd89b21d5e86d7448877ca62110ad817411cc2cd3494bfbbdce507ad620eabbb59b13ad2461bb955e0080ddb14d16334aa905c4ed8d64136da1f265ba16973fc90a79feb62d85cff8df889db613f6ccb3eba8d4ccef1980d927d7db1ebdc4bac879ee608786e333917a8708e4ca3838a43c0f1e04d9b03f9560fe56b977654807659db5bace8dc4a10549dca90e7bc6d2c67796f67639a08976bfccdadc47da9ce053bdd735518afaa7f25a7a0c73cb969832296fee03ac550480a593fb2cae25d63a2deaf7e49d142311349a804fddab8b5b271db6a8cf5a77ca6b35a61244f0c23d94c77283&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152462644Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0, Name: checkpoints/1600/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1600/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=8c32f4bacccac30879aabcca3e5d5ae1dc86daf576e41b7896b30afc564a44cae32e67262630ba8a0d3b94c6e200e6ff7eba350a823fc82100c9a3d54c86cbe165db6e5fe0f0e05e92136d45aa7687b0b5bb7b9682fa08e6b4d08104d163780e81eeb48365a70bfe9fa17da5accf16f20c633e6930f9bdbe1a45d746a60dad7294dd4ee69fa672ea5f3ddb3c43eab725688717a31ae70bf2cee5d50be3bda3f18da6f8be360280edaa8b7ba60daa588e9a01f01c311e504c405efa73ee41713218f6efcf65449cf7aab54104dd420890b79869306b54e63cf3ebc77aa85616af3d53e748f65a42f3ffcebc4155c15f0f2de6ac2dafa579e91628262bafdb16b6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158367174Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d, Name: checkpoints/5056/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5056/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=bbd1ce8d51de11e054d947fd021d13236e683e0afdb3e742737044d564f746c30b02eb73d6c0d126248e5a3bff86b4655aaf08f5a2a7390c0a68a55178439717fc94024779bf55ac70352aeadff5a688cf761b4b9ee585ee32e99fcd63c4caa3dcfa9b1f64016b229c5d23b68e23699d00070fc8a51b2e3c984ff62d648002fa3834ff8a99cfe17458dd1d2ce67d4e52f3efe0c6fd9ee3faecfac90b03b90656d856bb8e89f7f1f0ede1b24833799feb47812329b772af259285182737f44372a2323af3c259b2c802856eb1dcd90c7d003337aea4b7ec909d6ca10383adb7576c2b2a5f669d9245707e8a688746b178842adb9902e1d71cb1104ebd9142dcc2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152462825Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/array_metadatas, Name: checkpoints/192/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/192/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=88df0dd4a9046914eb393a6ec512f35ea3a9e110ec85ea03efacac6e85b82084b4f31cc9169338b0b843fdffd073d56037cb769e403b6b1bac54b683512872be17efec087f0cc74cd1e7bdf2308f59576a5b4ec66853f1f5c9a0b67c611a84d9073eb119a2aa3341d7ab05dd5504cc92eeab53cf36aa3ca5beca1bbd79aa4bf7c2e7f91d7081c32cff190f6c194a7c8ab36a1e5c06d091538677a1f54eef0e5a6cd142080112fbaf8cd2983af248e809e495661c614740139a729fbea36964274275e327f31693c40bd793a3f38acb8761db7bf97bdef2d6c6f42885e483c750360a419431bbc6d424b347dc796aa6c0f92da827f1e8d1aed0c17ea21d017998&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152528008Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/d, Name: checkpoints/3456/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3456/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4add3cba731ea2e69991df77c34f1807b678aa8648883d14fd08ac7dc6979501719fe7477e614a7e39b183d0ce65d396ed096dbca75f5573440affdb14296a775a49a46b3844bcc2f61aef6100b06e8ba6c3bd0c1255a133d78bac87dacef7e75ddbe514022722225f731215446b60ecc764f649f6eee957c3f51f6ad1776e4a0e0a61b85820cf8e59037d2cdff067e948975f841f06dc5a386faa44256b4b807c4e0246f319abbb244ab459cb3bf6b17642a44e7c2443344749cb320883906c95acb7d135c4998ca3122c404c5452fe7c2d3915ba4834e844db9f488c3b951e690b62ef20dd6fa0d117cc95be95547565cf0746b3446168e7af7327089321a8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.159435479Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/array_metadatas, Name: checkpoints/4416/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4416/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=423c22e017b09a39679839b6d7b6bf6b7c294818ef8341e7ce010bf3d6bda6967d71f0250c2afaa338a3f89c841dd6ec2afa7b28999517a498b048bd2bd02ee7f223996abca44e01fb0dbefb43ef6b8152bfba343db3d30df9ca1f3771bf5c0aa1ea68227d711cf53f3b4593d34eb7a54111c555e3078fea2413c2ea72808531650056b993822f3f1fec0a404e4ec78224b912c768214c8d3b306c42b6da3690d648c82e03d81092a62c6a8b1500133c39ff16e65ba2cfa390d00da09a6a865449d6d95d14fbbc9c4b3a580bdac030b6c542d8c475c327ef9ac7597bef7ba0a6e9c493e2c6c1a8e7c2ed7d96ab8d65c6d273b0c420e662e0b5bd1ba69b9dbe30&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.159448452Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default, Name: checkpoints/4864/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4864/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=adc416cc3e394d6435ca5db1d325e8386866df64b613c5b2cf42305874e7be5ec4aa3c756079ca7d2363b1d0ff7dd6ac47916b7980f929c3068589389f4cf95ad4e75394b2dfaa3e0a7ba2e3ba750206cd2268321c7c744f1cc577a29ec229bdb0b6dbf530a5a6fa61ae7290a0e706638d2d7644aa65ac95dd279583dc6d9ea50aef2eac8f6141f9cec81c6c31683f17a29ef604e322169add169c146632eaf40356415cfca2d6e9668c99156630de2a575bb9622035918ac0590ebb0e2188fd275907cf70ac2be53a4992e53a2c42371a6ab8fa92abe33d117f8dc9c8d4234b2c6bc1ecef923bfdca946698ca52d82f0193a4e284489db713e8a9c792db505e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152607889Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0, Name: checkpoints/2752/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2752/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=3d7a325c3bb8118b7509069fe72260fe3702bfdfdd454137fabddcba5b11c7a9b3a497a3a6c63752c9f842140ccbd2d90663fa80adc9fdebeaab09f543fed89688041bf3d80b4b5db8c6681554c53546988f30e95ea260f66735a41dd7ba74778e6484f31f4b1621b6be95aa39e3ffbca007cdc15ee274c75241b23f18c642ee6c1dffd5ef08397de9d1bdcdc69397a694c795138909306f66e45c6188bfb446064eb9284894cf3f9b0367e8aee64b89d845cddeb4b6558d18646411fa0dc6e7a38dab36eedf39c033b4925dc922dca044b1ad639332ee93985c6a85133fe638b680c8135f00e3f15e671280d3f702d03624c534a6afc4766881eb59ab7f3c48&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152658833Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0, Name: checkpoints/4160/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4160/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=900cab138bf73c906cb395044d1df42aa044cbf93e864939d29cb9bc7d7de53593e743bc03dd6ed07cde0852fae18b044a0f95748aa4d6c895f8dd5e0cbae42f58e7cc8e66daaae5b77a5c16c9ba68874b6a6a0c9945a91c260e936b8819f30669611e59be1279729aa7debe3e0ca0a84f47745839e4b327c2804a79e976b0f74103e5bdb900f71d2c38aa5b3002c40f50af24d531d06061905090570f1d72fd329d6f659934913741d13299ead78e48bc4c2aef89f1f9ca493c47e7fa0b6c62ab5c2f84c66c1e4ce902df7672748b251b55ae0c92c16be283e4bf6fd577f1343c4ea59814452552f07fe1c20753c31b7fca9ec18e0eb37bdd9e3a4dce7e572c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152645533Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default, Name: checkpoints/1408/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1408/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=8cc2d7ee22332b7320b2c2e06e07858974330680b5796fbbd684ab8b16b3fd88abb2722fe1e24fd217467d83d61441abb974beeb26e6fb40cf5fd258a16a38a968e4873402db17ff782b506fe64c38f213ba4ff5047a9ba1b895e8e5eb8791a94166c2804e810868205d14478dc1a40e8a78d7d0c44b28e445c4c1f3696ba8f75a2e968c6b03193d9c14c3fb98cc8ed09e6997a0146c85f8b255466b8f955b107192a4761ec0c4e44d6effc76eab1dfe7479e26d7f6904c938e36abcb0b3cb7b2ee6c54d145604705b516f86eb5c875fd633f8fcdd598f70b8487474c7876b839d40f946fa2347ee64509c32ccf2cbea2e78a81b28bddf8e007a4ba3efd7aecb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152648908Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536, Name: checkpoints/1536, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1536?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=2860c10e3ee685e25af36c3b660609d1aa57ecc832acb328ede935ab92bffada21656ee6e84a7499d33364fb7ddab3042ab1e20b72c50556605b8e05bd73b65bb7f3017f16d7273417555639278119dc0b23062b38ca2059755c15583aa1ffd66812868982ced61456c944b31deda0f8883020497c9ee4614a4a7371b9ef6baf808f9f388f9a53ce767eea10a2d69f5dd2fd026e75ab374fd79435fb553741dbac17232747d7a0661ee021fd6e96cd6b01e1c096e9dd14de689fc72464e218122447160299defc5f168da0ec1deff5ffeb9bccd090e9cb5d48894b42211d0f594f62698f1281232b498651a797da347d5a355ae1b6fa9d0d75fb4714b940fb21&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152705671Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/array_metadatas, Name: checkpoints/3520/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3520/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b94d6db5d456ad182aca0c7dc0cc24ab57eac48932e274a6f5150e7ba90e198f31ff1ebceaea2a17d4a937ee3cfb1873801a03ef45230c4ca2dab224590362c46a4eff2ccc3e4d6f0fc8603efea9a7348d8acda6aa67c055e05609c3b26ba749b9f2cf1b37547275541219be82fbc0c9b6e9b2351080aebd6c676f5c54984199b0f0fb474d39bcc14ca92df0078399e0b7e757bf6a1243d173215100f15eaa14d46dcfbf0646db374252dd7ba5fea09fd32fea5c9c635b102769653ebc83eb1729bd9dd69e328164696dce5651b351de7e15becc48398817b07def4a116983522d8d210d7eed17cb04b4aa4310b472fbfd0b7128105bf67c2a2d2d5cf396008a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.1526865Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856, Name: checkpoints/1856, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1856?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=85d1c838ef0b780540350516bbb4e795f46a2cc8868d97dbe7e7ac9542b9bb970869af84f42f978428149d5b3d2ebd14d8de450f1d4c0eb8358165d0cd0e5343245a38cea12066023294c8a0c56d133da32e430158a3657bc3ee0d1080c74ead08e1ab2bc7c78904b773f6e334923c3ced46c28e4dc88747611bd3c9d7a85a41eb35a6f18055c6f79bd73efc4f02cb2184c83da168f9cc8107fd5e90bb568e7b8e4dbe1170d959f0d860e5f2805f0dccb7a803c4a6cb4a9d076466faade98927a6196a8209ecb918753f42726fb9e99f553bb0a04e377afcaa6014d5c0580390cf6d8b5235a75787b9d96fb58975c0a25789c752829f5e39e68c7bb607c21e17&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152704053Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0, Name: checkpoints/2816/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2816/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=8a37138783fb6212b9468f1930e2e6a80cf5dcd5b4f5015e45a841dbaa2cac4cd54301b16f6def8bf19d0ad841537817d1ba2e07681745e0938ee5b38736e3d41a3f43dbae28d283b7bf2e96fc6eda8402f253667f0526b4612e0b1870e2f2c9d2a379d4553a68622d22e145dfe92fc6c55955405bdb8c2eefa20d78e1852d5e79304d7ecb3b8a780348c4ad8f16f67a9091c295b938cbceaa4a7f80b05d888f8b67561321d18d50fe87c3c8992031662ff7e7b19b8b0bd25111245f5911ed29d5d967c9de56abbe0ed697880e33c53e402c8e9e3a6b47b1223550bde15305860422205b27dcc37001875b2ae7bbe23256d6ed368f9286806cac877557822045&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152763832Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016, Name: checkpoints/6016, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/6016?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5006b801828489cb3499dc99e5e701b35cce05316555955a571f6d90a79d45b588b226119a64ea2c041e9a3f81aecaa392835847856356139e6b741b6222d6975d27d4739b7eb5651231b5637bdaaf9c060b846935be43b26f83fad14f408306e82f1bf0d360556f4c722219cf51df98b9b6f3f46cf296ca22788d1fec3818f22e9db52e0217e4efe279295a47528d4ad79ed0eba021322a3ad0233d05f4e45ea58382e42f7b13737a413dc0a812adb6d61d189a9977bb00185db0aedebccbb0bd835270117565dfa509e007f5e8e8d7b66e55ac4e53edbc5dfb2b1f473c5c5e9534a60c0502a91deb4809bc6f2f681f0b1731842cfbb09c7394ed374354e4f6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152737132Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/array_metadatas, Name: checkpoints/4672/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4672/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=14e67b8d2ac96c6007876d378452430da8ba4bdaafce50b672025c3d6ac737b8eab909d85a6fef54dad1044703d96fc32eac5e9ccb3b59c8f0d378f8dad5ffcad2a868f5634d60c153a1f02bd8a255a337fc11bf49c7aac48aae2eafc841b9de87c46f15312b3e8fe4a07792e8a9fd84c54fd274688dfe3c2f4a3346beb52dcadbc701d6608d6b6017228a23145cb763883096bbbf72d8a5bd6693812e5359227282d57b05b82b584cc172646155d5e675a6c6dbc9e53ed56b8ccf817c190996cfd13fb2e35008e6d3a3f5d9bd0963c30a15846dddd1efdd96fb0a78132be2453c349ad0d8f2b2506da31b6967bed4ad99c5ec3556b80ca7c38ce70ac50af668&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152741337Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0, Name: checkpoints/768/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/768/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=a66973a1933cd9127ecc441802f77a4dff149a645b95c9c0b70187df5443ee68ce919ba5401f1122d152892378ee2117bcec196d9bff7ac467d4e2e662fb6098496756660229b629ef657712b213b3376c994fb441da41f52633d7845cb4a5d3664ee1370b93afc63c78b8e1c21d8065f5495fa38177f72b06db9d6c2a6a11703266a1b1775ac4b84f3c9d96aee0d2398263c6694a5a0e5bc536a92176681848c98930570e78a9da4058d34406a5339f7bdd0ecf9fabe01a9625147d01bb2580acdcaab86cff951e74da4f5a93a651e859b81d8735d55660db1c4d2906c7bfe190f18c8bdc8cd2f54f08ed3382f025bd55cebd66ebe3240d5f02c167ac2f9fd6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152797702Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248, Name: checkpoints/5248, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5248?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=0cb80ad2d01472313238ee2c380b7bd194540aa2925a626dab3812bed5ce994a61d9adc2d6c9129e06a942cb71ea18f0117336eadd34c6b558f3980399110c313bc71d4a534da890c2b52450a65b9ab1d070588c75b3fa336557821edf9be9737e2915a98caeeac743b7032004c2e78b0cf4ca92e9e6b7cbd5dabf3fb2bf794c1c8e9cc3180bd4a9960f42ff3b674d76176336620192c57d6c8e8f78e747f91327c37000157d8d009dbbc24371773e0ed5847d08613ed8dc3304735ee3cee3db8bec146740cb7e54d31c5a5f37c1bf9a277cb587851e86a96d6f8bd56f88d598e0c3d3945dc9082bcc17d6d403c454c5f1742a1896b2dd88518b2e69a7cfe366&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152815736Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/array_metadatas, Name: checkpoints/960/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/960/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=68c5064eb97a63a97a6c4da0af64dca045863d2f3f7879cb4276f2e6740b3fa11e1c0a1bdd123c049878ffce4fc84b1ce649d3e56f97fd46f2d2155ddf3e4ae857b18ee23e8cf1db65cb9d40fb678001dd6c8d8a06c82dbd9546479d32ce60ca0fb20b830ec836848cf9af661b8abc3b71ba7e4b7e24c8e57581f8a1c3f7e8efe3eee7b25b1b327e978defafd9e7b00a97a957918dda4fb7e121a859a564f3d9657800e193dde21e9090ed361f8a19694c73d53882679bbed8188dcc5dc428aa58010d52f24b92642188366e97dd11c17614b41c7e84423483af3315156fc99a4050ed3b7dd91a35ebd4d89cb85d41bba723196c8f83396d2bf638d5d652a9a7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152836028Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/d, Name: checkpoints/0/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/0/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5894d39c187936c13edfef479db3b29d8d2032b7cbcf8dd7798c1a4bab39ed89d375004883fbfccf0084dba69d27e639d8dd85c7ef00a75e1c975cb0ba7e96417ffc891fb3188488490713a6a039c811d9bade427582fa05b4d44ed2ebeaf617426cfaef870c5f576569d2ad7d880865ecfbb3c48719fbf01a7db74f7829c46b9f93e4a6f9995b0f0e5afe9f7cb2ee866a14756755d23452d0d95b2d4abc8ab58a2ded07ec3f0d3ea41912a63e5986d9bcc178bce702144509f642f0bdabb098477419050a1691e0803ee3a247ff11ecaaac6296cabaa9165571dc6d0869cd823ab6949354ff3bd3a68b48001b3ba8409bda934cc7bd6afdacf87a029a483141&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152849665Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/array_metadatas, Name: checkpoints/5696/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5696/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=8cd99d40757ca7cdd93340341e4dac2162de746727bef924ddd22686d6a882ab71110fc3fbea343d3f272c531f78270d985ebf50d0bfdd6e6959381f71033ff05e16175d4a82b427647a1744a28200b9027bfb00d6028449e84b98cb33eea7caf7ab8badd5fdf3470c3a1e67eb84a8a664d2a749119b68550a44b28aff42f5efb6cc6e695b6cedd87cdaa5f98b42721842d174e9a6b699248a2d77a1cc1b34551aded34de22c2f4ccc0535e6d70ab3ca7a54ac2b75d8363b24110ff1f01655dc7f4a1702be822b6b8e598a5ae4dd445e7bfd3bacc15a2c605a94f9dec07566a28dd1438c87bab82eab636a3bfefb5ab81eabeca799bdc2cd773aef10c183da3e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.15958659Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0, Name: checkpoints/5952/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5952/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=951851acfff8103ed93029289a4652a81b3d6573300bb0f21904c31ce57e7da187fda2e78d3b51342703061430d82db9788105dc1cbb597ad94162112b5a2cd3df9b8fd960a28d42e3fff7b1239b33b67093ff1383851ad0f7dafa4675d84dc03b4d4e67e2e755c8c24da45a3083d8d793df6285caed8ad1e56907142edded7c425fcd0e7b1895e8620b13d6f24f06a13d7bbf355ad4fd2da45a6a7e9cd003d030ee836a3e10f98bed873b0cc88d17347ff728abac0667ac9a36a10f0b4a733d8639383d8a307f5e6bab15800f780fe00d7e0d3c0ebc223f240f311ae22ba8c03dad160502f8ceedea498f03549b1acee2653bb3b085dd0069f34c9e1c30ee47&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.159616865Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default, Name: checkpoints/2176/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2176/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7a5bdba11aa0691dc40c6a77a494e68c5857c88faedf19c42a7eb0e66bbc68255d4ef695ef791615d066e60f483b09ded435e3c17e3c930b7a2702ee422d673c62dd0ceb751c50405dcd2959caad26fcbef3b158e002b8f45971ffa38d2adba205bc2e54ed8188ae6da22235bda3616fd3763dc63d1d31bdc6d207809b29062f9274be444889afe93624397726421ae92c89a08a3854b538075abcf230a84e0071aa4b95fa522a6ec789e82b0f626c6341f9304664827b4627c95272757c9bde6607ca9345197c85215af859c7de8299ff8f6d957654a1efe864b63f4965fa5cb353d7e7b9849f9970ebbeaf0fb9137c70fbb820791e5c105a8947820128ffe3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158857632Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default, Name: checkpoints/2944/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2944/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=53067700e2fa08d64e1e6d3293429eceddf6e530e9c62b2cd95bd9a609f58c1ba793fa5483708d0a6fe6a2662a7c912659fd62cb8289b94e26bd123a8bcb3a08254616955181c618a542e9a3cd59b8d04c4271833c2e0f6f28accebbfb94e321d2768eeef2eb054f7f5e83187c496171ac438c8665a9fce4d25b523795f5096ef5bf9100104264737c4d1c8ca6a084accd356e918417f508d5381e876919836346a4b16825ccde0659f3d83623d158130cfddfb8b1f2d9b1f369e49e6f02d313bc03a2d6de5aa9bc05ddbcd3274e41d3416b71e4aaefaba417de2e6c0ec978f859f1a668a9c3cf413cf6ec13e735360c788e879299effdb88fa333a29d74acf0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.15967661Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0, Name: checkpoints/3776/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3776/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=021ed7270f3b528c20f11b2e4a948af803e7c0e1a9a26be35bc1b95ac838fe0e6210ae12cb44c01dd3ad26b3574a5363a8b4f6cbce068fd02a5fda06cfc6d2ce5b4551daf7bb845912b020f0a11ad3a885f423fa96f829bdd1c522c1bf01ce60d522cb2697427c623e75562e74239959d62072485ad3b183ae153dcb582a8fe13760aec95dc4f3e6a0fcb8c9cf86e74ff28c2fcea69bd0a4fc8f3b37cb3f5c121898cdc2d116bb14d89d78f0ae2ec791e143b97524114ddb7034ebbf453bc87cc48ce6e018da2c6025050b7d34ef9cd388d57346cb4270048e4cc83417aa51a9252c07fe54e2ce009b0ba86924aa082edc58bc1ddf01783bbf13454889d36a71&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.159705537Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/array_metadatas, Name: checkpoints/4480/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4480/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=8a3b9c53f9fb7a9f5fe347928009988d70a5323126d73941ad9db57241784cc76177df94954472ea44f63727dc745edc936314c4cf96d3da4704c9c873456912de28dcae90f42afa5459845d17bc08faadd8439c332874fb7b3b0278af167342acacb8dc9575934e2371e9761771707ad74eb362449dca01c25347405ecc3835fd75f22b3845903998028278000a9f13d87e5c3491a803144f500bc629ecb4f75200bc1ddefbb75e4e6d477230ceae5116d7136e99c7a9ebb6c6811b26f5d0268a66f8ade2f60ed4ca553251fb0b20ac63294e06932fed0e49c03c595ed12b9526c431fbd62dc63f4268920db7464e64ea7e77d9e6c4e18b8236844ee97e711f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158769207Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/d, Name: checkpoints/4352/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4352/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=27ed6184b239681c168b69592ecddbd1783a8eeeff3751d927158bc303a18fa8b761bb7051fa2d6c627c680f86dab3ebc3596c120c3afdad528a580ddfd5b1809ce41c28f91150de6e262b5ee370b597322bc4e747ab928d6d554be9dec4dfce53451a2b48c442080804c7a7f0bb99d34c78c4ec399ad7d74554c3750bfea7035bb6c76392406e5e3fb400d6623c2e038a5707901d40c8be004e6c98b1d2a44d44547f8264739476ff3bbc904143e9272a575b399717b417afefd521ef938f775029f2524a1671a5f2139c4d8797bcf5f68acce853db7ca6df6fef7f412ad187b00cd11914619472ab5a99ea0aab0c1f950023be3ccdf6358a7e5a22b499928d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.159742761Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/d, Name: checkpoints/3008/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3008/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=ac1ae5ea044d38fad1bcf9257143d57c7c33b22f76dbbbcf739267900d9cef0acbe79dfdfe8b38476b56e9486e29347787a77b2b5414242012f8ba4aa5b71f429da2f9bcf5ba962c82187c0366e6a23356584c020ce69f749ff3f6f38ea59df0234fa840ba8cd41e5e44bf089cc60e8563394f304735a42b819164b036cadcb2f51d518229da5e02275498db5be10e44e33b4ce7e1e88315676dd8c7e22c16e05dc760b1a021b3cc2606fe6f96bfb8ac96f6d2323ce5e01e07f37e47a8993d6de7026f0c6c5216650dee96bc308eb3e3087f91425ac0d242e58ba609eedbff9ced3432172ed2e92a9618749d8cd86274133bdd4f105336c9e62dc2aa480335c3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.159905064Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968, Name: checkpoints/3968, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3968?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=a1c115dcf4694646855b2c7144da6838c0b9dfab3c69d98879b434fe5a8a0ae5e047378b85139419b05321154e5427563c36d65b2651ce396f4ec8d2fb31860b7c9cce12218474b8f97ba4c7cd88063e4e287305a417615204ee252706053c8856debe0cf00cd41db735dc5df227a810974a2473ce708b04b9f05a1c9c855ce0dbda1d022a552b06e000092c41857380b208152b2e32ced3b848c215db9cb68241bf2b7461dd215fe441860831fd836be71f129424efd16977a945bbd14dbfe5c1cabe1959f97aff0efa6ff565fff0a6e7306018f7e9ee22876d8cfd39c4b192f8ec0f18ef4a58a27bdbae21eb5835790d853e4541b89f0741e3548a3b69df58&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.159920495Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0, Name: checkpoints/4928/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4928/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=969aea501191a2a39b614d05161ae329aa5c20df339b063ef8d71634fe6b959c7d23b95dc67d7050cebaa08e05e50555d0afe4f614d034a9495f4bcd6a384d7f538537af181dac0485f2fcefb1e59b67e3b556fade292c39de46af8683e84a214837b6206bee2b4dc890767611ccb8309a9f7b1d2491cb1ad9e5a16d5dbf1a4767674255877a609335a3c247cc89d482e2be5d1a2438956c4b3bceeab4edebbb0bc5f78be10e8be9cb57a1bdbdcefe57f3958a0a62ba30816eed7cad54eec51ee10671c5049a723626240e12b31c39d7b558c5b57ef67fdebbff9c268472b6a603b48f700b4b29dec8ff42e5a14fc0b1481c1dfa28989b92b0d005af0c3571f2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152558633Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default, Name: checkpoints/5632/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5632/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=342cede7085e7f0b033e06affb492cea7d3f4e67c68be66697f8f71f591c2fb48db8e047e8916dad78ad37dd2dafb9bc8c09721314b925bdd83425f1974dc8057bf374fa297f3a12ed6341bd65cc7901cc94e06a9543d6801bf3d795d8e3d3d7978eb84af7b27efc3832556d8a8757e2afee20f533860bd111d4bce30ae2db77bc97cd97a85196c4217bbd6ec78a4564b66b580d57e7232b08d98a0f78e3fe30b3edb392166fa9cb5b965655a2f3117300c452a985a65d5184e55f621e37371190aa07c12a8ad2ea55d5934f0c78e672d8ff5dba7bc21af1b0829a4f20871fe2682ce5282259b4d29b1b342c36267eb9d2e2e98141044d672952b3b37352d112&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.160054542Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0, Name: checkpoints/3968/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3968/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=98e17f0178430d11ae61088c7c78e7922ae3cf7f34582d35c0741080a39bef11fabc521570a2bf4755afdedd6ea5166be3ef57a8a714d6bbfbe1c8484f67a54928cda872808851250dd926a367937d2ff055dabbc4306aeae4dfa91cfd06c421f1650562a3e22d43e8bea92afaabfa6c6a53d4d59f5530c9a66137124721110da092162e61bd8cf0f1558f2ab47f3a4ba5fc29d19e83b835a40112c53db27d015fd081037b06e68e3d761691922344de4a0ca19ccdb1b6073a102dca7dd2eb8f026266ead29f9105be4731f682e2eeb91ccd41485d5bfa1139dc78406317db738fe11aa0460f8dc0c091480c66c1a87fdf9dfd63a365e0e8280e9b8bdcbfc4cf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.160104416Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072, Name: checkpoints/3072, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3072?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=83ef729ab6d4d8e7e6de7ffbb2257977cb77a463275529b212b12990710360cf5c91148c4c2b49e25627d7fadd4f223bd57bda58009dcc5c91b2a5de4da32c5242dbb6c55baa5eab34a3f93a35762c78bdb8c605272acbd0b38edabf365816f09594d3fc21ef1c6776ea537e49a27f352c2f2dc24890bdeeda00fca7b5dbb636241608dda32ff1498fe62b5734cdbb913f7546c048ee17419dfbcd3eafce7bd92a7953e3545df6bfd84ca67e139c48fdd950291e3cd51e116c82ca69b32c5675d3b390b7c2fa830b8952a1b1bc60bfa60ac135589687aa975578efb3fa641ca72deb0b1a9ac6eaaed626d3c9d4aedf6a1b54c06c668cf0020d7779872ff1ac32&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152919246Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/array_metadatas, Name: checkpoints/3392/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3392/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=35fd45ed4793063491955a94722a77e05674d6c219030a2a2366c171a34512fe8ef6bd8a8c83536aa0620ac75b874f2f4251e41f43e897a023c213330bac79e204fc6df6f5e485b5f3ab59622e08763c4976e382fc634fd557c8ad712123b385c0734b49a58dab1001fe997ebf515aa403b2d9384e06553c78596cc500423093792bb6cbaddb43e34220aff65f82a9bf039f7c0dc93ded85a9f99bf4c46cd393edde4110f23d894f1dc35831c3760e2dd4acd6b8115a600baa34a886f93f671417e30df7ddc18399036ec56813a0c503a1cdec69fdd46d5ca22ca990f6ad7098cb86876cf0d938e214266f3ebeab3903d32ad61942eb70b39969048edfb06400&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.160157525Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0, Name: checkpoints/2624/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2624/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=333ca6aa4290302ccd4c66b9daba3ec7246cbca841714e6e44dfbfd80f6a1a5c96b43c89c4de1f079fdd91757a2d01fe74862f49795628feacd9297e3e82dd63dbea54fa0f4b33f770215a055f8aaf8aa7690b2551657dcfe409ad56c81e577e2e2f386549d9dff04eafda16eaf27476e9bd6ee32edbecf712bf514c81e8d9419ccc528fe04a653d55f51e1dcaf50c98235acf48968c5d8988b21dbc9fac42f56ef2c1aa7cf40d10d225cd430772d6f11afd58c5ea270e16838b268549e19df23d45d4e3beb06a42b4ba901047023f80d5d0119b6d6366ad85d725670cb5a5129e4aec127bd2f30f62b3df0171f69e09a90b5e3c339aae1d09c0925c9f204e94&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.160248194Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/d, Name: checkpoints/4224/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4224/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5b21df535d4896fbe45c49329ee3247e4e59c79fc9a3e90915ccf29e44168304629a3d9b64fdb481459fceaf769f28fa28db5ba2688652fc0ceedd770ec127bf79559a82a2350adbed223a2f966fb5df0322aafd9e03832acd614aecbe469cdcd97ebb18b69dbb490dd22c857aec1c1a0e38c344a2dd1142dfe8843bfed5782b817bffbc4df1a79c45bbff147e260f5f73c8c2b4b0c21c88c93fdd8515c1b6e70755c390f3d945dbbc7200274c8cff01470a9a06cf11da883f019e09189eb4a5a83831732c7a503f3f678ea955f5217dc9c0e9ab6f2da1bd217063281c605b3f162cc7622339ec7a4f28cdae5b4b62bc5383da101d7fd6bcb920ccd6a09ba787&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152862235Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0, Name: checkpoints/4672/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4672/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7476d737f56251e95fa985c02a86b39f4b9b42613a777b74bbe0ab895304376a22e50ec99177e348c7c70202ae08d6a208f308219fd0262ac12da9bd8cd1fc3b656d087b172ec4de54d2b195da546f39cc3ccf1e8cae4297dedeef29d874cc8e972d9cffcd5d66355ee9d92e1b7136d3144630c9522027fc20dd22996350e7f1611cc3c1ca1790ac3224c72dd1477249fbc79119af09bf23a34d7ca548197f522f838d3015d1e9bad97c3fbc50e6d4baf89ce735a70a8bdeafbc120c560b21e3a39770e5121818ea2a38611a747454a7f296af79582f417ceb91734f266105da421ba6834400d3a32e6e3dd5dbe95e3bae50324f349b06002cd982dce336fd63&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.160357342Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0, Name: checkpoints/5568/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5568/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=19a55da70fd31392258290ac0f5d8679f00fae3c033aa767467e32fab5c35307358a2aba584d764da6d6c7a2c853580000bc09b0a628b7ba33fe60ac2550d57deb9048d142fcdc1ba2b686be51b68dbc80c9661041304e422fedeac34562b123918a8432cf17786e0f83c21edcafd34a7d7310e405a8c65d76e78d9de889268b707c71206e579b22c4d23f0a415428426453f72fed6f50ddf3f7d40e2dda664be87ebab7e9352fb1640eefbc345e9a340346a3fef055158b2ba41f3df6bce1918c093cdae4e8fce23cee430711e2f64b2f2fb9c09d1e7b460e6eda53384ffd7e238c13e91647c724741de60a33fba03646623ae15a138d35d27bb66f69c9bb9e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.160436134Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0, Name: checkpoints/4480/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4480/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7c077ab6c6c48057ba94ce9e79c9b8847240e62688f0be8bcd0673c0059a6565d979f61355b6bf0f0c53f61a1c2d89e30224888d47d8d38b6c28b369312dedadf9d13c65f4a1ee1f69b82ff586d4e0162dc2cd6e7903b385e523e4049e4e89791407fa40c512f9eeafd934e07d0345bcea380d2fb409c20a1fa862928a19a03061a0d9f88efe3dc960e829d3e70ed16887f79b9162bce890b0c35436a90f0b0361371a2b77d1a4e58a1fdbfad8404fb6ca9787425a412b756d57df13e880105a3f94b18c7f7816c67c06b05a4b96a2094965b81ee27422131fda9cc10be0c5c2e883178b55ad2619d48734882e76214899c3feac43a039e8e886405ec5f01787&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.16050886Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d, Name: checkpoints/1088/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1088/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=79614e0974a8cd1e2b1eac39be330beeac3d44a11cbc98614a91c262f96e45e8b16a58efcef6598216ab449508619ca076672642e15f5fe9cccc3a18e363a46ee70fd9412b0ef9e7d3634308e47e6d1eba07e210615f6c155fbbe6a365118f06ea008c57c590d9ac66c1166b446b7c1b96ab6648afe738d600da7cb1309ebad0217bcd7fc1b1c23253111eb39dfbf9ae0354aa7a697a6421d25346950b5f3df6d3d8f9f40c91c86ebcdc8232ee4e62c3c9ccbf09f7a093d2431bd5269f1dd0a200e2d2c437c96d55593971683c519b533df2a164b66e682c09f81b705acc13d81a5c9553b312351ac6be9cc5d0f9f27dea8bab9352da52df5d05c8f91ff92602&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152878048Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328, Name: checkpoints/3328, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3328?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=579980a1ab2aecfd41c876cd5c25211327fe6f487ff9f367d073c8c5e64aa39ba80ccb870ed4db04124e4bf6348527c024913096312c8e49b02e59003d7605921fb290c60bd4f48454bb92a24acbfec0528135c91a1d21a05fc23339a0c3c1f1808a9a7c8973f1f52a6f181e2a3fafb3cc98c99a87419fb10cd88586ea06b515281e73467285b92fc0c06530d2f7a3442da486fcde6c0b0df135ab84dee9fe3c521677a20c4be8af47b8526d00703e3919ab51b7356d7e3a2135e8ad9c96074a0342075a48ae71c84eeeb6cb93d20c3e7d53943a57495ba557e48ca377969d946b922b68de9fba199b18921f25463caca096902a8f780290fcfb1b7ca6249a1e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152942574Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/array_metadatas, Name: checkpoints/4736/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4736/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=938f7a12d878bc1cf418904ef1e0f0841c96294a4d3676ceafe7cace111294253064fc8fa0aa7bccdac7da166d7245bd6176dfd658f8f37424843b52acb3dbc194ba532373d722e35036c2e0fd5eb6a97b192bcf5d7642f844e35d849885ac2bc173df7ec1630eca01b9f52bb2fbfd063d1637b523cbe715c9b914431dc544ee06fa7ea34fa12c702840dbda0707bf9ffc9009040ed3bfc60b04b77761f11a109df96e90ba57de7c2ea6e76f87edd7f1ab74f9d19c54facddb9dc648fd640be58df6b38c54d9ab6554263f2f11a6adfa7c66919c7e7ceef0865df53c55ba28d3024cbee3d497ee386a41dbd94056696444c14e5d1cab3304be8aa5cb95536b12&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153006953Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/d, Name: checkpoints/1088/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1088/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5faa17efaf723307fa25325ff82cca0529d981bf7ac4262ec42c8a3717048589e8aeb85e5b46edd04dc36af1528b28e5694bc4f0a258201da186f790e36e80e67f55ba962e59248db68a69aa01dd5a2098b63384ef2a755a24e0b8e23223049b4acd4d90db0a029ff70e0ce0e759916239d3314a8bf8a4b4832b311baf26881015874bf7ac8a6171b964a3808339e291da583a3e6b4a86df5f18cc6c6b29586602d1388020bcceb6cc8e49c1f7aa93536d16178fad77d65585e2a11619892ce458fd1aa4ed5d60890d23e7f87f3af183850f65b58a5eb544dc509a825e5feaa103347086766024344eaff2765aff429488ccf965b15b1db2f7a09432b2337dc9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153044137Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0, Name: checkpoints/1536/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1536/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=88d8c36892656ece98feeb898f91e021dc66e1be920d8e5468af299b775e8e4c9e071f20005cd7ea2a2c00e206783a26b2b973faff999466043b917c4dfad83f8c2c5372fc998af8d9a8b5a4b42de2a0f73c4d6c52724acda086b5811cadb8c481f9ef064b76de82f98996d4a62136ce0fe0940a917000e64d2cda6ee26ea0180de67baf06f2770d75b25c1c3449db0fe11971754d3abc7cf3069c2d1723b591f736bcf982f8c36c78478cfa476df2859a1ed5726bafffbb9e7660bf96f98acdd61e842be63dbc832fe8a108d6ae224cd2f3f40a12341763083ceefe2f6f2f46b2b545ef3e8410bd1745d99650eead581708d3c3e6b9e6f6f0e0c21951872a66&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153046051Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d, Name: checkpoints/2432/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2432/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b380c3df1aa8ca9dc4146aee15831874f16a65e50b623ec56779877ffbfb85df72fd354344203974841d8e078c47a5c9b261eeb2bb142b3e61504b51a937f65b7e356854704658ce2f5fb11d5f5a540c36b6bfb3ba5b6939f26cce4b037c88158fedb9eca4a06e0139568aaa00f38f3f344e329d68b2ac465365e960c564751326a18bc89f406120560e80b4b2012afc68092b0595bcca8dc627a088826596269f403c15af7ccf72958429c57fb378627b14c93ab5016f56eb8cc631b877f4ff248cf910e976a8847a589926f4d99d80d6d67a2a49f28278c56ec9f45a455df5e0796ce46835e451c103004ac2a9bac644ea6055c646ae725fcf1d32f5a558fb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153059195Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d, Name: checkpoints/5440/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5440/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9b4198da6473d4df8b777b692b6f53001b30f53b2b0a7ff0a32244ab82fd72e142340b57086c57e0a7f91ffb8b8e810b5540f0fefb9412e2fc26cf1fcfa2b9a9d8bd79a13132e435f724b0675da80415552dfd97a96af87ddfd37a6c55bc2f34bc9aeece1f89510c3ab723066bf21a0cc5a0222aa69e0d72ebd7721e2b42c7ef7808d2040d199874a161c03ba9d5952a41df9c2c3146bba2a32de7d27b41229084acc1a779d8c04eb54e05cf7bd3d06dc27f78ab411fba638af7ea47b34bb0853b37eb4e2ac1d7493083b2338c26ff80aeb3b2385858392923d99632d8686bec1ca9893255902835f70f8ce2a76286b89e748f48cff9d8cd0dc53292d61504e4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153130683Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/d, Name: checkpoints/3968/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3968/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7b72e49371cb4f04964c833ddb4548b1ff81a20b12ad6ffabdac55dd50e9862e8f31b096d9e8dacb144990fd0d1f4e30cbd6a2ce39d2b7762f80148aafcc439d094de08a8e5cc98dee01b4e3608e5731ed081c7b898b87c4db392528c955bb90713a17b4030e7cdd549c329ebd21bf1208f091981cef965d7d5632cd74b755e1a3beed8d54af85923d4588dc15a04d32a2f3e98165691c0c7925d7c6d272fd462e7a5a6f10e41e8373754311ce324f671fdb322e163a2f5a1f3b053efae2dc280f17dd7a5962ab6da15edc6153e81002c6c96d64f6dcc179c6073144450d2b0fc3830301e58505f786904723912c1d58f0cbf57eddde48e00c48abb2970de7c3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.150664892Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d, Name: checkpoints/5632/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5632/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=44717b37442ea0cba72c026bc55b6c08262e33863dcb6b11d7b529ffa4db319918982f9e9b19e0b1a054d49c8fb5307cd647b931dcfbfa4decb2deb240128ec18a3bd0aebb43914c75f6660cffb1a93da1bffe23a23c85295cbf8dc7a31c32fd7d5389a7bf07e6cc5a0f73f77703a6b38d2483f2aa3bec599beef90c7267450180cb7174619790700655e73707c5c59013859b01f9ccb6760e451f1488fb6cd452ad427145b022cecb3ea6a1a8e6648df9e7cad65d1effd23141cba9a98cf5794f3a7656933c0aef0d025965da1de2e2ba81173b76fc88247a11d252ce14fa3d8a2881704f9a1fb7d6740185f6b14e05853b068c7b52844631460b59330ca5ed&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153254906Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/d, Name: checkpoints/4288/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4288/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=17023969daaa3e43fd4518faa877256862877c810985744c95a85f999f549759b119ff3e46a1ad8b5865787df48fc095b502320e3fecb4834b3977ab682e70fb108561a9d1d2f363d698b3c565880551d7ef01f767483a698c5329d4c0b2df208651e3b1eb13423eb1d0a3edf73fcaf010e7780a4d12a6a72e0f524e0962a6ab0a7d91141ba70df20e21640924a824534facb0c78540ab88ff836cd620d971d1080ba45ff5b1afb880f6d29cdb6ab00ef4727c78a92c7da89dd9398f0ea3147c92bb147530d50a2b3245e389baa5f1c618f6c9ccd438d9633e48bee21606d04875488d3513f7912eac9877e3221ba6ce19d38c589a264b2dafa8a911b23fd504&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.15310466Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824, Name: checkpoints/5824, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5824?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=79475e06246cd9cb231cf1a8777ce14fd94a5546ecf1f3e501e8078178d453c6add9c714481924b939dade42af574955e521614f3b6c9f68b7720e61e6ac5224a93f4386c909851377688df2955d80b1b25bbad9387578dff25f2f955300216e92b3cc7dc090a63441ca4da003a47abc56f85e3be956b8af3e1d7ddb08ddb5e2401a05d29299bceb36eb78f2e786e8b88dc710e85ef3083e5588f3e81679abadae4017c3ab959cd52a89f285358f50b5612fb6ec5f8fe14dcb4bdd1482b70ae86f33a9710f23323b55017f6a082bd8d52a912a826f047c19ad284b858a4a2b938638c7500a3ff49bb56b75fc3239ead1d5fcd5273bfe428509365180c4050a4e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153329353Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0, Name: checkpoints/2240/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2240/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=aaf058bda44cb66dda3833975401c7f6add172a5f14dadab5aa2347a833671b2238dee468894b76fc103d121ca5bab70e4518334ddeb6662a565c72222367f160d72b9c89e6dbf74a506707515698caf87831658a9ff5146fa6c1932aefaa4ba6bb83d1cffdd78058882a2d4bf8153fcb45971b0fdd129ddc67c7d66da122fc06d54be6b856e3aa813cf817cc0c6850e9b4b06c08998bbedbc19f0dcc5845ee64a52638915f573c0b2f70f43d1b6513cf8f64ef5dcc6879b5a613c5c8ac5efd33051fafa08a1a2c3f65cc616abf68de1477eb5c120cd4b2c371af928173b873baafb91d348e67355c811ec277ff255924179e85e68fb6e2c8202725300844f0f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153335252Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/d, Name: checkpoints/64/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/64/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=c156ab3e523dfad7ebb84f75ac172804db4709201d1ad606188e94e92482d1928e4d985fbb2b5199617a2fbf37bd94dd050f0886ef15697fabee5e7543d356bc90d0a3bc8d3bcf9e44de4cc9751cdea794e983b17dbbc51a17cac5dd4b517eef6c36d34f236011537c0c6d7988438c77ed09adcfd16d4d6978dbcc83b34a906c5ec59b9f6e95bd9b072fdf0b571e2a56fc491c1b5c7788226ce3d504745a167bdeacdf53ce24869e6568ea52e90c49fcd7e6dc0f896963884d03d8293591901c8941b9d9c5324d8daa647b76171dca69905a08db4a2b89a847d38890fccbcb04898ce9ea54a9dce1b67512109a419ce0bafab254f274debcd6760ff409bece39&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153341245Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default, Name: checkpoints/4736/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4736/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=6987f61e88abb10d61057dc48a90a7b2e630c82e6c6a8bef32a6fd6c9f9fa1d3edd85bc444d00cbf13801219e1208dcaf1f09fb5b2bca270042397988773dcd711cf1f9ac460ca581409d4e5c3c4878118ae513a7d8b614cac799e32192eba259cecb8874fe5a82ce54482e50a010b6ad9094c55ea3b59a81a9ee9d19e3e1af1d56b2552e2710f29791bd690e07122942fa67f001d45efc6bc37b22dfef6fc60a800625a5355dfee6304e33f04772d07c28df39fea751fb859e2dfa6a6c54606fdda9c3df68762e91d88ea0d5240346c69aff090fe55f526386454b4c14af9a9f8867586acd299acf73fbabe8da39311d72634e9496ff6b91f5773bfae75ecb1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153356024Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/d, Name: checkpoints/1280/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1280/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9c6faa3ea8b3cc297e2c68a1e889e980e4d2214876277503534b34ff5f5bed7fcb29be1caba04780cc586a5152514808771cd86fbf54124629cd7a8e3c8c593dbd526d6db7e24ae6bc30c1935d025707405a6abb6966cc4f1c32ab71c351aa1078a6f7c4efb9742781f2b7757e8e74931b1da1832d27c70f16fad5c604176d16af37bcbf518c04686a287f12153dd0ce4c893944e7e22c72de2a9b51cf1db6b7e04b6e213687c9e772aaf0ee96f6610370206de05edc3309f86201e0af804fd26cd91ba455af25db22782280b736145d02de38afb40dd50989881354d372b888b31b0ee689fa5bd21b4ac8d67ba71f442ccb2314078b19dfcc280182cf6a4c41&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153379164Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default, Name: checkpoints/4160/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4160/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7aba42661fc9b00cc75ba79dd92a59c520fd1eaa5c1d7b2eb72e544153fbdd2832828bd6dc7dd4c2a4aa92ae5ca834b0f7d400e7177dc9dc2fd03ed76c3722a4396088d8808542289a747a10a0dc79fbf92f198f200de48223f06fe8457ce7ef7d59f34008e11b68416e8dd815946220dd4075603d332f601c48630be9044b48b5a08046a803be5dcd33f241a336825c167f481abfbf6668c574a7d05f8a7d1cdcdc0298514b1257e3272c4d0aeaf14056794edba379021df550ad212a93c8f3fec6c2df874bfaaaaa550454fab53ad24730065a12f5cc68e64706f1800358a8af035f378040ac733add2cef84991424b60f208b7b75dc9370b17347ba709c49&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153384055Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0, Name: checkpoints/448/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/448/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b796e0cc6b25ec593339eb9992c1b7ce0ee72ac2eafe978999e7b56c0b4ad0ac2138c5af78a1305ea10bbf1e88080269ba4e0e3e2783673755ebec8aa3ba9c73652b4b3580b4e8fbf22523dbc3eaa556f8eee5e4a0e808b032bbf92d1754886558318f46ee4f009b3e058d15e08f06bed08bdad0fe9a5f9871f127c1935cc0a2e8df9dd1c5abd7c8995fc5192b25435d475706e519100d45396833bf513a1f4cb07858b84344e4df4cbd615172e5e420a2061c604d229e899c77b227026905dd4c8f2970f4b402f76defd246cdd434f9243f428112c574d00e37a70bc9f806cd0bb4665e443a6612cb13ec95b96f444fbbcdda956ceb39b3650797970e7a9612&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153381503Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/d, Name: checkpoints/2432/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2432/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=117125ccc2787031176fed86ac2fded8f25935fcb47e4bbb4747671de3334ba30860523d71f71b6967999886adbf36d339df74b3f53d1380696e32f174bdbd83ae0c35bf2468482dfaec169cc44ce2c0eea95a3bdcc849c99928e1b65954cf60d7a30ba9734c6c4381413da761c0665288aa60fdc680933571b07305caf28917a53d304546fd5032b946a546feef324c8ed265b8373c3afeebcd271a0bb51d69c0778adbe9cec49d81e3c26cf99f905e4425e418f6544e09207d1b424b8761d49c46df3ffbf2063bebfa8470b144b24b9690b4f27cb7eb05396cf71c1fa5fd9ef1f495979955c02f8e36eb324e565c12010c51f7a92b423872f08cf10092d0ca&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153352037Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/array_metadatas, Name: checkpoints/1728/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1728/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=2f08f3c7b1381ada6be2d5857d826341910610cbb914c31bceae58db6f9e208eba67ed05f577af8e2ec3fde8a14ef327fc3e2cdcc9d7f949934211375c332a95dc0e2018ff9e4639082bbd27f6c200bf285bf1901ef66272178f5aa21b8847cd07a6520b0b51365bb6bfc29ec83a954e031dbbebf51132cb4babfd838d49543776167f9b095549c6bdd3c6a5b491b2a3a140efa7dae49551867d9503beafdc8df9954894d6a832da17a2afae1cc2a8201c6122cfd7d81652b4126ef5df0a35482519078b7244e13133353164ad14a760107f9de271a187463e30f775537ecc871582252e7e61af4659cc037ebe30afef97ce3f4bf8c347ea2d5b55ffdf3f73e5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153395735Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840, Name: checkpoints/3840, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3840?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=1c83cd4d48f058da1e3e286463077e9ef878319e621a30f6a9a3ba82d2caf43937f3c7325aca0b05eb9e5f58384d3a15e61fc046c4dc5b6e115b9c660eda56fa6249a5c7bdbb8c8218d095fa31c2b0283838bf2c2243c793c95fd1378f27e4c6b4add9ad9bee05c4b5ce1c638198fc17ed7eceedfef93d1b12c2f5c8f8e055c647026b07636251f05965fe42f9931cbacf067c491142eb2da660af744c536fdc829f950dcca35c3a23a4d1e1869fea1cfbb92f6152a28faa5ae84dca452c01c8853b7630aa41112eae6bfb4a23a9ea499145408e11e07806422ec7ac28da687be3f00ce7b6bab6e11ba443ae4a7b8ef3013cfa1c062667bd43d4414e5326b9af&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153383881Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/d, Name: checkpoints/5120/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5120/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=a7cad1be6e1af12f738fc7ea5384d5426d61cada59c7b4d9fc1114f26df56f10d443c52dd9a9441adfa8c03a128c1aa3e9a84d552c801df8c3b01d0e2d09a6407d6473792b0270ef53e1189dae1ae79ac4614268075292833b1243961fa30af4a41338e23654de23db321d3b6f828f0e64229de3ee21eb81459a738059d2150eb2128fb9acc641c06a66970e0ee5b4368b227a1b37a5fdff94997893642c08089660d54e0ccdcf3d342db7fed57171109475fbc2f8c45eff4da077bec546aae59ae2f8513c1801a31e2edd75371b9def84a58a9d806c3df42149551ce3a62ee9f5e7a8770d24474221ac924f273f68d5ab621becb5f81af1901502bec7984e14&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153216774Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0, Name: checkpoints/3200/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3200/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b747f0fd9c2efa56167b8ef0a361281b42f5ab5e3cb4aeeba7e3ae03a34f10e7898dedc26004b5458e29acd8b606abdea69b8cf06ff754153ab83530142c8ae0bdf003b3d11c5923ffaa3a7c88b6b23e16d01228d73c929186658ddd1e74e32e6c0d541c1ce75e4b4ead95eb79788e3179578f8cbfc1935870c985f5a5117cdb15fc4e29c4e97718f9fe60e81513dfac8d1de84e25cbba4ebb8584c15f0316d1eb4991d2bf8c1d77a1e772bd31fdb2fb9e8f2cc45ad55f69ca31b0fd55c20ac81d769b6ec7a57009a9bb4e673865ba7b87d9a497bc9d5f8d7128541c9b92b4976d2f5dd34de7474de7c749e5915e046bbc7d5e4f402bba7fd082d2f0455f8cdc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153439022Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default, Name: checkpoints/5184/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5184/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=1cb96f41485297140c82bc03a372496e5c061ab5e5f66a6fbf27c83232ca768c3d68f44e961c2f51e05d65a6dc46bb87c41c359b705768ec161c6e0302cb4a91049f042807d952fba1c41ca6bec0091460564d2bbb2e8dea484972c20589eae8a0eb5b87a71d315e337db662cec2e14895681d179136c2f5268539beade4c0026e241bc05fac65127b886ef717d614cdc5a0e869c8329860d68df7b6c9b71c05e4113675f7d7e118da2651c68d22dbdd220f77834e561a15b75a75f6795e5ae9fc85f385f470adc29fbf6e5b8be99f5f5ef1c6be8088aa06a507038f82d268b2c83fee49d9c4d3f4719643e54b3462ab6e80d9bae68d6f69ba483513c0a7dada&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153458845Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d, Name: checkpoints/5568/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5568/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=8d85e473b06d6c55061e2958f086bd2b448a3d3ed2fd2ac8a192b0ec4f9878a1c8cdff0bde34918024393351b07375a1d7cfedbc1bf887773d7534fa54d88a9859ce27a8a1be41c394dc51fe17bb5fbcf990d32ac48608e87ccf961ea4d50c46bfddab4e38f43db6dd72e981585f7bb4c969ebd4a283d0f4a79b2846e827ca990748d376d0b6281380c91229b1ce3e809ffa60ae3c3d649102332ecfd6e67adec42fa28e77fb132d859f3ebca128a7919f6806360981248ef85febbe3aea826bb3175f5379f7cedaf1f00bf350bc6c5ff3655fd26993fc55aadbf3c64d4c426f69c8a53d7f037e4675dbc0ff304c6787ad22a84297d2ee1325fb4b002ac9b436&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.156896603Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/array_metadatas, Name: checkpoints/640/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/640/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=a6adf983ccc75466d1a8502d4b88d664c05fd337e0ded57ccd533f162ad57bd95ed22d6515dca389dc147d6e5a14389d2bf0c32de1cb4cff59b229c8de02511607f96dfa754626d23dde235a63cad8e4d72fb2289da14e7debdcebc88c91f4ae426851ed0241f23fc686432429a43beb1445bbb622edc5db75a6766e1c10bd1ed9d02d699bb706b7d70b45e6e19a6c1c23c2169829470425ad6fda03daf78e94eab7c31d68b4b6df0171f60177774b6fbc4ff7c712e9733743f89ba086ef9fe718cfbeaffe0348c62ee1cde792e173a7d56c6b94b0e3195254f49d749d28000663c4790957c6419d36f9c90d0520fa113188aad39b460f95c2c260da0e7e516a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.156928251Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928, Name: checkpoints/4928, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4928?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=a787e7384f84539182648466129ba572375e5b93bfb29878b96db03b04205a29575d21675f8c826623f3984bfaafc2368da49018325566a6cffbac4ed8aeaa1858f23e1f1bd66e242bbd66cb08bec70b638369ed4a9f7fddffb1becae63ab3e7e32484ce924be84799b18881b632ecda02936cb6e0be83f095309c1e43302ad5b3b9109ee9a6ef20c9aab29c6d32a686cb1f2d885d902beda8d67050822265b0c479358dcfd695cc18a67f95e24c266556543b6fa6b27644cf22c8c2080eb01f8c500e9d950f25e82f3684cb0c3dad88f165243e23f363b32c647c083b6cd0ff600f2a4ec984c069ce60ab33e652806ba6b342a33a8b53d0cafe5bcdbef680c2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.15750207Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0, Name: checkpoints/3328/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3328/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=614dba79d40c71e5efd4a25f802d3a64afb471b30e6694a39cd580b5101a34f83c59f966c6cdc252d8cc022fb79ed87869307c3bab6f1901aba28d53a8779837b15ed27f44c9356b4bc1804d2fb289fde2a49c88d83d6db79bb56f7ef25cfcb17cf4e2e0061b95e59cf55f2fa56275a05bcabf15c45d8816a94234afb5c28c511f67122310173deadad09699f51288cf0becd29fc8339d4049c91c602a28e6f5e70604d7b8659d8330c44963b1f3d6b663b6f1ef7e4f686b5933fcd9e5c34f7b48b9840eea0681f0428940c293ba5d7912d2a7a234a08f31b6874bbf6d6d6f316d761d9bb62e6876d9bce36793e6769cf99e2516e21a1842e08736e294e86a8b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.15768188Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default, Name: checkpoints/1472/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1472/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=46d9f4b704c43081e94d9863fb706b2ecab5003ab1f51bfa0549f13466f26ce846d8d395353aa82928b75d56f6a222f0ece3778c1afa0619f6e8e2ba478d777cc9214b82c04712ffc8553eb0e34cbf7382e56d6a9387e3680e4e024b93ecbb73959987a4a010a02f0cec43644e19dcf88a2f298a0e2e02a2142ee0311e7636af07a613c05649b420bc45341da196ee1fab7be7103a8f93bc3d9ea4edc940d0279e9093c2ecd2271aa93b1dddde2d0d3fbcb4b8f0a97d00763344b3b053cb66364e2139f2022455a1dadef77f5514a9aec3ecf9ac02a089f17070ae391b50caac7458a377d402df8c006d9a22b1829aa56bcecfcfc5b0d18a055c5963ee4f1886&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157744781Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/d, Name: checkpoints/1792/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1792/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=056835e75651c25a02514da59d2cecd2df51555a25b228c875684f36427133bcbf9bc3c4798c4ce7005d99f1a6761d5bbc81b3c7d5db561e9a1442014549fca48343a027f4e39786e88548ed5164da393eb16aa6a14af65ac528045ac0d5e8c7a4259e67974fb23c19abbf3af7326dd1c1bf1fb07da0fa87bde6f788d852e73947bf4e1f5438eab7e0ae0ed423912ba6a4dbbf95f2e43e71f29ef1cb42c3d6c1849452298edba481fc496ead2816131fd2d4e8cbf3a475907996c26cdf72bf142bf456a52bf0ce60b3915b9456a80490e3c0ed8accda5b7672a6e28a8f9a3bd17ac68f9b626ea35d0ba2be1de69e214db3088ac7638cec5e959c2b7cc348cfe1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157793919Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/d, Name: checkpoints/1664/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1664/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5d8b540e3ddded81fd6c7abedbf7e97b4fcf99e48e531e7a7251cb2012bb936b4bc56fb779d4e5c54f77bcff8ec7ad2e74ea3a7f392cf3564365c0d977a99b0c36edacf7ee09a329fe5fe39552aa252d44b12e5eb41a45a52114c32902e06e6f139b2c1d9a46f0ab7a841b127609e9815dc1030ff78574bfc2415edf0e9c39ba1bdf99c58ec20c111747370f16e17ac2c16de156c5c3f6f6206a0417288b166531140d316e2bba871ee16d30def08924f6a5ced8c521f043176f57c300463397f78c6219e2b87f2efa2587dfa14e4e10397c46ade2b28e264fda651e02e52361ccd3f9c42fecbbba1f1f392abbd748055b22309e11f75be86318a66c15fec1a7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157801392Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default, Name: checkpoints/2496/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2496/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=04ca53b825e6defa0188d62d3b89ecea1d64eae1114234aed586c8ab519917bd44c3ae82a2a273fcf7d1c43fde54be7a936c20fddadd990a3bb9fe230936a17f32ae3c39f8f8c0dc32464bb85b2ec4af317dc996c7fa94c7827c7cf5ddb2fef100d47bf3924132ad57a10a17bd889ffd8fc064d59a6dc2ee67fb11779f933f54efeae15ef99596cf24029166ab6db8c8caa95774b491a2707e69d21f17415c793e0f004c12e0f678e7ccad65e9aced46f4ea1e0bed3ea78f846114e8b4cbce3c7676ac9782e3422eeb364240933085299fc4fc55a6133e7490bee3304cb3147d18f8812805e3dab04a18bc90b48ee1b077e3ae52e1cd32b9a33bac9508a5c596&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152927877Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344, Name: checkpoints/1344, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1344?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=30b963f494d9bfc084f597f5b68643cfab203c6dd0255899ad4e2a61cadc2eeac6ee2341b8c77a74f9d45f725dd6db458d65bb212d7cfeffc0eb9ed9a07ca0c58e3180967aa954a0ee7022ed852a509f0cec43f24714109072c38c855ee14a713f402111735dfa7b482eb2ac289659f6b306ad3eca4162d9392bd0be2619b94a3154bdd8d57b1cd8fc307b3112ff05543973044859308020485471701c67cf576e9f000d66d6470da096275492bc7b33cd003dcd375b4b4b2bf8c6d56ec89adc48eac0a276005ba7cb41e6fb937785e58172b8bf7e21212dee65d4807a84a98108631ff1ae95e0a080f91ad505ed52a2850ac6b0903cd835758074555c3061a8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157815938Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default, Name: checkpoints/448/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/448/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=6273cfa13c7e6492c5392f60ddb9000ed2dfff091fbd01bab486ded9e013242a5f9626f2cf4645f31023dac912887652c79b4055276872121461fcbd25cccfbbd3e83143b3c6dea386bf0c3b0e28ede2db227b6f8b15fab26ccb221e2614e9c39efe3d7d9f2ad92e53ec249a72dbb3e1c47228a5bc6626425561cfcc80ef16a15f44c1fa9651833640fc277c6a750ea85656398f243c02b1cbfb04576d9fe2c8536e312e6b12dc25792faeb8c5326eefad03ca8acd98ffc027c2debf5bf0feebfd0053f0f6405e494b7ef9e61936729a59865d913bc2f91de9e37f9d9a1cbaaaa6c53e08b2b68c7314cd0f3ebdb42feb95e0e3a1232bb99fe9d4f03598f5fca8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157947682Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d, Name: checkpoints/5184/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5184/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b66f3891e5c78cfe127e1c159ace885e512b24d0926ccb9e138583a404254e37cb4ff3f3c0991166d35e912cd59d17d8ccc316d085c17906e8c71ab362492196290208c6bb33c1706e557e0cb315fa983881bb752426c891d1cb885f54f676f52a1fe22ab3f6c92a1ff9226770d0b13c35e00f0af6487ad5d104d9bc89d9f72f075d9ca1e6fed5b2c8b5151fae61fe63849da5c1777f0185ad7fad6df91afc67cec0b9c802ca8de706d8a137acd85cb179d0f611f3711553d1a678373829cddeae0f7a6ed021f31279b5b59ebfa8df832d4839afc9b68ab3ed43c006b0263d007877af2775f7f81e26960022404f0133230c9aeb2d726bbc58c8da812b3cae10&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157938949Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default, Name: checkpoints/192/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/192/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=3f06b1f75bdc09c63c8a007b53db1523887e9a8015d151e36d5d9b4d2979066ec3a05281c03ff152dd3ef0dfe4a7e5ada49c24eb6245e26beab3f1fa80a85a2c8ee9083290bf26f822a27622345a42302aea0e046102f5b55267823083948b032b045e35e37d150003d087ea03d74890e10b54d3d4acf1c61fee494fed749bd65f7e46be7001ffb807faba2e0b7078c007906cfa697d4f5c200d34377b3a39c7a2497059513baba0fda0d818f9f9cc0be5e43bb6f6f8652af1aceddafe86309c8a816c1d70f5e9302d21b2f11c457404cecc66a97eee1e02ea954685a261e083f77f685f7fbab26ed3894ed258e22a76c65e0e36cb8af62cbbbc53d2bd52cb26&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158162578Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0, Name: checkpoints/2496/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2496/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=774e105663be344b43606e2a2cf40dbcb8822db0a5e55afa11c28000832bc8a38cd8b93770173869c2ee89f3265b9cec20c33ca0cde29144a19904c0995f46ae08fa2819a624dd553746cb0bd497c54aea26ed26f1f8fa25d1a8324978be6b53dedea6d891656c674c5de82975b3d41eec3e9b1f34f24cb9bdcfae5a1a59a52bae67d50dee546281c78c680112054761ca848f9ece813da1b057d955798ae856966e04008627f1f23e84cb3cb3e6739006aecbdacfe0bd38170a8e7767e88cd3c5840fe3c01d540f4c37c3eb3c238af4b536b663a7cb95238b35b1882d2e983971bd9fdddede87e5fadbd5cc086fda29927487765569ad319c7f645e7a2cec5d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158470188Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0, Name: checkpoints/512/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/512/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4f948a349107ccce76c58c695ce65edcc2f3440bd9ca85df42bd7fee0f042147434d5743c8bc217ab2560bea15a8e4317eb8ee03287506a2f5731a8a6c44c60bba85e59547d68acd3049019b6697ea0b79c77d95fc93f9bd70846de5666a0203f02c920fc72b2c02f29af48705d9eccb496cc47ac7bef45963eba281968b1af9efd89545ed60cceaa2b52552cbe1482f3c7bfac37f220cc722d4e655120d5d4d7f6ff9a5b50f2fe4de55428a5ebc9ed3e74b022c450d93bcecac6a377ac6d8ff8778af05846f7f87490d2e10ef2d38d89537399be35b582b59f7c02dad8ecc12de2184c476b8899655195fa0df57e86df400593321db3a480dec766eac99cb7e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158568085Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600, Name: checkpoints/1600, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1600?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7c323cacd2ccd239de9d4548b9247befdb018e6f5b542f51e6d724b631910e7466dcfb58c730c35c06f4e03be9c2fb14ff121e24275dac638365549ace0826fbf3075f6c7b43bf612903942d2899e9ffa6346de9938702e3557f3afb483b7a5b16ec1d737f97099c121218c482d41b2aa05f4c99f0e7424fc9f39919791be7b696a459da40b435f98449fa52fb4088f544f911d7e9b63507aea757847c2850b48a26093d47bf0e8dcb43f0c1839894c42a025a3648ed5be47f1fb21fe62edf5cef1391370337c052a30e9bae6f875f247c55281cf03ab407755564a87ad65576021e88ab40bc4fb50d4b017e5e41e01e791701553079da2d4a0347bd29d5a539&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.160115652Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d, Name: checkpoints/128/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/128/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=8cf000560ae1b585943439284744f02edddcbbb74b775442640b5df90aaaa288f2f3473869dfdd6382562d644294c38af5db4578a8ad334e46d5bdc662e1b7368f094b49a0bc7b90a0b625cf343e3f8c4b32f3525e609bff96daa0fc9b8b96352cb7e5ce8d0c1962669e232e239fdebbff784f0957a46c243bff2280241dc9d2b6190239d0dbbb06b53f487162cc59854cb32c8995e089050a68634c5667dad922783d7aff8a9830fed1e1caff4eb683d81c54024608f354b5c6ad30fbb32f14090d1f93c24c141a02e3557e91a139392c4e8ccb9f1851cb7626df680636b76c8f012f2ed9ff36ddecf0b02034f3fa90a0e65ad1c55e2dc54e79ab7c46d53b95&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.159711073Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480, Name: checkpoints/4480, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4480?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=912c29888212adb0d02384328a09b2eb801e4b0d3eaeda703531229cf7244183354c2cb3a67baf7902545e7ab31eb88d042ecaba7bb865acc8fb34246f3c9b9ecee36cf5e1540fdf18c26581db87da31e529aa98d466de016847b5eae545b82d7f328f27f01560cceb8059826ab54d33e847004bdb87967a1f554eb4065f1fd4ac501434affa35e9a92c97271fd201c840608b6554263112c2169b07717c18a2f65ea86b628a5638d47d84895255ba9f8fa1bbee12953ef216e47659eb1005faafe0aeaeb29cea2fb1e4694313fdf9738e7315c6a5f4862bfc35d4d1815c59c31983f47c5030c198cd0f20a74757c2d614d5245355e3f88a9da3d06184eab8ca&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.160783975Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0, Name: checkpoints/3392/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3392/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=ae4f59fa59c38650bac74cd5a73b6a4d98beddc81a808e930cb28617ceb38a6c8ea86054f1425e9396f96b73ba051975969f99823961f491a1d024a221af8c19568ea0369c6583efd63cee9df44985731691a16cdcae804aec2d0790fe85d9dca686abc4ce63a77f9a85f66e57fd3c7e04a97fcb6f4983e8b55532c037f37ed9a22bbb5bb564e374dd1173ddf460f0302ddcb39ec2383e7eb7d4aea7b4982e9ea11355099b1478fec00b01266e69c270781bf4694e856b825fd3a084fc373b1bc4e5cd8d69023444ff9878ffffe341f1eb9bfafca0c43631a8bc646d27ccb169c9b5c4c399104b75334bb1c688a0b93fa7e6ec8fa9ddde6e6a9a71f3c0ac2923&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.160841507Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/d, Name: checkpoints/384/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/384/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4a74399044f2abe017efd27b42343861839696e0e3ae6e27892e48c04ec59bd0347b100c44ab6752b6df1aa0200006bc9476c3f99dde7fed1790df622407249a06dd6d980af7dd796ab8c17b0aa5437289495ac861d03117444e8ee7a079efbb8b99bee9c2db80061d4959bbcff42a53fa277ed4ba0e5eb92202c4834c0afbf41e166e367694cb9e77bf0f18c5d78a14f785a61eb5104f1369ddaba04acc8a86b13df4d50d8b47d6b07f326cc30cde0b247f9547d4ed855f46f9d3126004692668ebdfecfe3a31a97e9ee5e34c84e1de6d85ea02e072134f74fdf13afce1cb1d2c6c9386c08060a329082e15ca2c724992456f04dbff514c01d913b56167c149&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.160914902Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/array_metadatas, Name: checkpoints/896/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/896/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=0ca31234af3eed9f60f97c1ed85f7eaa1f6e61cb082226c89d7917d0cfa862e370631dbbecaf31fb662c3ddeffc9eeb1d235d590a36a9170e795ff380b95d1f971949c8df14ca3b084141b41b2ca2841ddec65e0c0ea8e9886be88f555c7b07db6a4a08ee2e98d9967835573b9308340b717675d9c696d303c4881b6d0120f74b7fd21f91e0f8eee2edcbfbbf3a31c6583fc8f29eec9077ee5c0ceab998658a2d4a2587d986dd0b8499fb19c711a1dcaffa216b9bf28dd5391c5f5192105ce6ab42653e8ce70cfd18b48022a96c3aeae3e5d456ce5ffdd60f5017f11c109dc4525626f901c364ddbcddccba45bc88d872ab9529f6621682e4be3b4ca27547cdf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.160960592Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default, Name: checkpoints/2816/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2816/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=448e689dd11bc80d516f4e9f97224dc0857c4a753cedd72f64aa3aea83602b58f48eb14c02612b72ccd92011b4fcaf005e0f7ffc662271022ac713ea50c1cfbc2d0764f343f5db2b803f791994be5d69b03dacfccb41a934b5b5dd624b468711e38788504318fb42f3494c085a28c14d9714c207ea522d06f226cfb7d68508f841575233c7c1de482d1b3a55dfe83aa10c50dc6ab6d480c8753fa4363d4d676bdc31e6bfac8f5ce34989c65f3967d323281b40ab766c8ac442f8220e2ebd011b7fba40a9bb17f5b831628b698776ad076f65512d63cef411f9c349abf2621174ed9e92a3ce97ad3ff66b7e02b0a71f193af0d4cc5e515571b6275bca51e743bb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152957766Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/d, Name: checkpoints/4864/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4864/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b2d6493c1ad4a7e03169f723e52a218c401a27588d3229a081ac5e15a22a92e50cd6ab16b4e830006d7e07f8bb4b155ba451dc0f6fef984e45b5e6d912f224bd7c09bc848c8a038f0b0769d9026699011a493effba4473554eb76e1254b07dd80e7f308d23dffc857a3b7f9aa59b292bda3b9f7f128e5bb956ae2d5e4732c0be9648ba8ad9a05b7f6da2ce51569740b72a1d43ddabba640753da0da16d24e6e60e2837ec8aa2c8538ff4332ec1524e59287d01e279c4411435e4ba9753e820cc3609d94090681242e407b4a049785353c1ee8b235de13ed843bdedcc93054644fff2e65612cf8e7fdb5a9759af10c8fea23a9e47ee298b71e012f9ed53b895a3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157807653Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440, Name: checkpoints/5440, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5440?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=9266415ddddf6976b63f97f984b8c638aea20e9e3fb7bca64f39f5eef6b2a798972ad0a93e17b853b729fad666d0d9a89701429d16c200b9015932da2cc15beb7cdd9810333e607c575ee9b644d890a78d5fc04a29567c74e0f9e5a7d11939e02790dbfd51ac4ff5de5a3f3c5e2fc7c96a0c8bd655146639af3575bf1fed274cec8425f483d256e8ac88bb444191077ad2d46609e3aa344a85dc2a41a24033fbff1ddb0aac8732be32926b277658552281531a1bc748b1712f3a8e55531a854006c0155e76891b8c27fac3b57e005af7cdb910bd68b05855f5c8ff1e9a603901cdcff15838b518d1a8367d141397a8634cf2ea6bbf1fa8d7869372be9927028e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.161053976Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/d, Name: checkpoints/6016/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/6016/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=3b77c2a774da3e65cc79675afe4fcef3e456599dfb710f22ce38c3f4f4a514e50230b64d6b9a55419f445fd862d852d569ad8b706c1fef9a4ac7d2e4b7d10953b7c902940fc3906678139fc15cf1ca9be485aabc5e718097f1a00a2d020e9c17a1118bdb36aca647fec66fe7651123edd1bbf6332b24dbb19b6f64543f6f1847e2233d5c4e395e8f948ac2c398181501d4626a6e6fab68146832942664ae0e89d6615916563b9fd0f91f090cf584fefbe25da65333f1cf9c90d92b8da8b1a52eb47f47ac4db8fe6348759221c48cb4ee72ff3f02ba51e07dd442a68b8364689b2a52d6fca9c41285086c7ab6dcbe403db111e7514d867958959ad195c3cdb23c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158607609Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456, Name: checkpoints/3456, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3456?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7cd247c951ca40e6651c05980cf9fe16882ec0e2ff9ca43cc50ae8f0800d76fb1aa8c26ca0c0e5d0cb50e3509402f624496887d6745c03bdaf73edd4a85f39e9ad1a3a3d3056ca4f7e7bcc542968260bcf3f94a49b9e11c7c85f1b287b151b97751e3ccb105ee5d3cfbd1724cda277d255dcf69ae3da8358f070faf701bfe9010467de81022491b6ddde703ca94c6dc1443b5223de546a0753126c29f1a6557c80f5cd2d4b55638c0a9c3d6565abcec16f19a6a11c8daedfb2b3f42f96f9cb2170db11035140b1c4c0d8d5bc27c91a8605eca0dd0834c4da7c60b755cb2f2b3c85789f62af3a662f039c78521aecbbef051da82c5c116860d6a5586aaa15b3f6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.161131421Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/d, Name: checkpoints/5184/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5184/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=c26372630b49303d6d5aa1dd855475a6c3808d4bc17861915318c6d364bab69736aa534ae8d18002cd5e588977ed0e04cc0ccddd83e12d46b28f7a042b15543bfda91d2d5fb06455fa1999ecf16521b09a1af987d69c71c4ff1c11ca05dc59eb39eb5fe37b23de0668a967683ee751ce81bb31521896d60349a8a501b7e474becc551fc47e51b43a6c51dcc0dc3c934378bb7feebba622cfd4b3f5da3c3d1125092ecf1052304e58a81432e0c89da2eafb5bcd262b6b04d895a024c488ba2e97b266317f18cdfe804b4b1e502020c5c1f5ca8611d664de30ca204548ac00b2c9d570690ecd89541580ebb471df1f2dab31360abb5bdcd017335c1af240a4c906&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.161184021Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d, Name: checkpoints/3072/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3072/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=003e0a94d692877b490dd27beb3c1110dd52b4bc4c502d05223203a6100ef83189bc75a3ccbe5bd5c5bb8cecd30f61e782822c6572bb63a9d26c3819b5572c4ef488c481810e0158bc682e92f62def455e2a59bdd4189fb84cf172ed57cb9ceb44b049d153b32848ce5a6ef4930cfe6d029d3544cf191cdb4256c18f62be3e6efcedf1032ed5c96eaac03784e2879d649f01f12823c70d6f3fd31a5563d9d65ff5596181f2b35530e731bc16f4da8852a1dd50682ebccdbb8295df37353bae25bc88cf77126271062b53e7e17f7de157ab7caa9f290b0ed79683da482a9a0cde4cd2ed95775edbc1720d76d4d6f0e96a93cf9440a4a347412f62bf837b7c2631&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.161347931Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672, Name: checkpoints/4672, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4672?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b074d7d5cbbef68c7d389178c0cd978b892be699a941b2088bad12ed748ad44a9bd2d58a3fb7da160582594c0e432de33dff524c4db890d4e10c158cb95759578ae1ffc775d939a87cf5681f7f6a0e400c1897bd480b09e1383ab7bafee1b2cc239df9166e876711a94a8a01ab48fff93f77999e0a015c0a892afb85bfb7fda060e84148b67538faf2741465ffc4c94aa0e8e1cbbeb7efa57020325d4b37ab540cbb42cd1f2b721f4912cf89badcd398afd515fa9d00e5bd1d2e9787a558b4250b72702a610f17697cabeb9ffce37626a028a5af638d40697f8d0c80fd21c14484f38a75b984e7dc129e86207eca7958a21ce7b6ed997d7c6fe5b4aaa04a5e9c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.161395602Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/d, Name: checkpoints/4800/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4800/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=48981037d5fd888d21553dc19533634a22c53d00dcdc39b3501ce7ee803ad9af5deaa995053ab86a8d5518b912762ef8818650088d744eb49ea9ceafa4585f117f0faf932a4426f033b9242d55503d0e5339e8d9009b75b7e1d5de28867254b8e1d4bf678660fb4b74010963c0fb21b7e7344ec4efd1a0c046aa0ebfe4a4a347999cfa56af6a0c31493fe088ba358bf0530433e171f27aa8fae10861c23461fb55fa9072aaa2f268b1e0be4e6114267f048c3fa004da5cce4200f307583ffd90a851b57695c4b2ea7d1b5063bb341d2f268ad0d98a08f1d13f7dab8f645c7e12c298fa76fbfe367834cafb8974cf33cbfcdeeb9bba4497114504f96731db3ade&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158684835Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d, Name: checkpoints/320/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/320/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=2b03a3a105ed17f2e80e0e901466bb9e82374523f7d29f3becbd6a8f469cf8be5f6d2c04e5fb31a02e43e8c81983573939b32c7813d5a498a7f7511e36a64a4aaf7e6ffa25d70972fbba92360347e124035957fcf255a6abec170f029056ae47a7fb4454a8dfef8358afe062cdb22e4df6af1f3a0135aa8206d1574cd7a67e5c6dc90a01393e9f73353adee609846a79dcb54695ff8e15235f513659cc5fec1453630e263665ad0c978fe9b6e051c720443c5535ee491e9d928bed2079c74e95ebb606fb752eafec7d321c679679381311bfd4212257cedd53f46451a273f0f54b1ca18199486bd5680e519b3cb7cefed0c0dd8dcb0cf8b6413d0ed80fdab414&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152424424Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/d, Name: checkpoints/5248/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5248/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7e1526346ac624278b043c3e00c06ba556cbee09feae3a2a3ca197ab0a6f8fdba7df66e1dc82bad58f84abda0537531b604561b6b3ecc5329950da9dd2bdf0513e6ee90912b1bd4562734d6bf6e8264b91f5760ca20b55215b62b8fc90627bd6c5664e2bdf3a235a8db99c8d2db23b2685640971606aa3e191058e7cff8f5ba7a90f563480813bf7e8250b36c441731ac8cb66436cab5ab73dc025a8189bd5ce0f98d82299cc5a1792e5a5ef7e501e08bb99d1ea6b2772c3ea75d7362cbca3b1f89aac690579e792d340d15fb5e3b82f5232d9f5c527f9997806322f9a3cc7afe80438f1f1d807324d5df19e5f76d8f32c7c4315098f26f9344e683c98d43ce0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.159508419Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/array_metadatas, Name: checkpoints/5952/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5952/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5e1155be7b8812cbf14ccdd9549c186e516ecbd89f7790dbae288f9762c0346f4c4096a3f78d37ec53e424b556e05ad5892c58abbf3523fc46f539e85e8f4387b65bf5c8562e6df01f4f65e04994299c45bcff5069f2dbb427fdab96278038ea762f5767c145ee31ff6060a3cb6096387d7a16ac480bbf29bc893161e1cc335cb36631884149014f0555140718b0261698816c7374b91931274430560c89353b6a83ae29994476115673572bad0dc0b887001558622de5fa5c9f1e4fed72772370ff9f54a6367fb54317de39880a7ca0d64bc11f45605eda6b2fe0b2ea5301a2c1a78f33b58066d3a3aae92b3d5ac6e3c94205f1919f74e74300dcb713b8f09b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.159515896Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0, Name: checkpoints/5056/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5056/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=859e2de18084fa5cd1a762b88a7997b8626099d5de6d825dab60a17b630aea69f1ed86f59e77f2b25c4827250f94eebbd232bb05ff14311326a4d196eca736f1473afd6b38e4207cdc167ac6f38fcf2cb6415640049a9808ce91ebf0f5ff1ca246ce904a192d1baf955941a2d94088e186c6f5ceaf456e02cabba206ff446c1599c9dfde494c8c20f79f071254c7719684939f7a5aa83e859c252abe85eb171a22ecc835262b9226e037b62c4390482230e861284007a51fdeeb5a1ae1488b5eff580a0d970bd2431ec2abf881f02ba78931606025044b19f6b7e8462c63b845813e8c32fd4ba67c6800adeffd5ab5d07bbb083f4dd274626c98bac5df6dfd26&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.159523928Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d, Name: checkpoints/448/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/448/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=0478e2cc4417fe45305cc8c2700046cd129bf177cca55282d60486a3fc0953e0502f8d27d0b759d52ffd1049943f2498ac56c680f81bc42d25fa237b18daef2f5cc725199c0c35f2b6181524b61e43580485fd98e22f7f033b951c966fa73b9489e1e7e1020d0a287ee6687387b7e50ad55e0badfc8a32b2b23c4ae8d9f61cd559528d1a7be2fa7afc2d30396581b3244eaa016cc8958de2b82380dc1cdc92b930ed3ece69e7b43e11acb23b15f46590b20c9fa8abf000506825d7c3b75079b6ed540f4b43952d333c659da9f29a7dde2673bb91707ebc03f51de1995f5ce2cadc093cb5e094f1078b9733e88b36801c2ba1477126ca1c4f1d11556ca1c5adb1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.159553908Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0, Name: checkpoints/1792/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1792/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=c0861a4dbaf70d6d477c634289238343432b06a1962d164ae2728f03a9012a8b4effa6013a7c98bbf01871118a49a45911043a656616f572fdf99de8981163be3fb073968173592678d2810966d244e083580a9544f612f9a369790d7d52b84540831a191685b0f218a4c934a97a3bdd9b7f5357ebffb88e7083d44da5ac5dffebe3cfc46dec2c6b57e77a846deb719a053e8689300dd813e655251ce51e3425d9fa1d0a3225dcb974b4824ecb75068f5387a0a5caf41aaf1f7e5b7725f6cfd00df6c5e99659dae4d7238a89608aefd7aaffa5e41796a345f22b238e944b4af04582784978a9f59cf0b1697a7c3aed0432edd22b6d25fa9507efa5ffce4ee6f6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.159577183Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832, Name: checkpoints/832, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/832?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=8a207a2a7eb498fe35cb42099f3a420503ddcb0526f60c32b71e734cd18c7ccac40cbb750aef8f6b8ed7b9159c8d95c98ef31459ab887b215d689236b1e124dbd4246de528086118b7adde869e3359c98f4818741de47e9ca0b2ab701a467aaca032283df809e1a500ef8e8a6edb9451573e2f4508046f804100427db654d8833423b223d848e9e133ea72a979b4deaea87151a27e3cb4adb8bfdcaa739b8d6ed7df6b4d3bea8516370e6f4327f3286d0b74c8a1cc7703d3e0ff97842dcb3eb30ff7cd21f63f341576b1aba9127f3d1c1a4b674c863797b566cc39b15c472686c5718bdaf5831f1911cdfde32b8eac5bc1791393bf75c14ef99ec9a6468b7e2b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.159596734Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0, Name: checkpoints/5312/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5312/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4467342994eca089d704fc37e21dcf018481e383eecd5368c82b7f571a95a1bcd248af0889e43b298a4b8fb274855a7f10bf418f696c93726725a4dfa7f9057b8f85a914e34c1cdbf3e895c3baa69132cdecd9f2811d4618ee1d96f84af2ccc6b265e3458891dfc30ef8d00608c4a0bbf26a2d43074f6a5c19eeff7faaf5b78424cc4f23f959f691b9ddb3daacd618ec486cd6544027854f45141f488b6e2e26f0ea2dbdab4e0419574349ca748c52f1326ca16edf3738cc6663a33117753c15e56919fc0e4269b4bc5164c84463e4e48732a3b74b645df6868311396414bef9562ee3d1a2d5633ef7fc9c67b6a985c55f68dcb1ecbbd6db56c93993186d4214&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152520791Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/array_metadatas, Name: checkpoints/2816/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2816/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=09162807ccd4cae7fd5ba190a88cc92a3180d2651ab179c6fa9213a6e2db08a5679b25dd0e657d54f8828d574c7a3722d181e1d104d4a229d5913595367c347eb5b0f7571e81b82546cbc49555e49e5f238c749d23740920c7e65fc1b19b4643801e66f35fd7d5f51f4a82cc9537c1f4502ddcca33f8dc899181d6030d624dcb3b8ffbc137ae2ff52a69684093789a7a22293ae5c615bbddf4edd0c014f0513ff577c708bbeed7529f860ddc1e7ad5ac520f2f09a6e14ec3e46eba573a9c4f5651fd0a2bb3ed223183b86454bde1bf33fd459309b14a592ea0f6d1fbb7cc2ed9e18ddc996491cda97dbbb29ec8b566a200ea00a5c3f8343c5c41a7c1239e56fa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.15963452Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120, Name: checkpoints/5120, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5120?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=6b7e7b22a8bd06c398c5202612ade67a00899baad9337e4562456b989e99119c5f016d3b63bf5054fc6b52f417f75e9747abccd9098aaf354d80ab76348926c5b00b550ab75dd2c2d148b0e9917f6338a0892c2ffb5e7e0af64a3161931f2d1e0c2e2abce898532831bfa197b2da02b373ad84092657ed7967e7142cbb46b503688f86264e48a5060af21a13a35dd99fd26a64a653ae42ac8e10e17cd7d6beff0446241596621d04c29222c1728b635e7f6b95fe1c856b6127932dc6470ac4ee44fced7b4b097fb948cf10f64533f7e9a15f6051d5f331b45372dfa1d4062213f98d8d08eafd5a78561cd81427550b25b42f0627fcd5e33a35043e5f3daf7104&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152515743Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/array_metadatas, Name: checkpoints/3072/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3072/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=0cd360ea8122d289581bff855a784a471035afa452ea2fc3a7e0f890cda54b3d475ee24ec7269722a8138c7203719cecd7e8a3a6beab018615ef5e1daafba106ba914473b7c5c2cf2e844cfc8b06a9f6fd21ce504586f829927fb6c481ec90ca88c229df961b967d3d99046cae99e5740698309e42a413974383934967a1b7109743ec5018ce2ad607dc3cca76b7c0c06b8198aea0f67d1f3429af519228a2696c9404f5a8251c5bf7cc03ef128d935fc1e98592fa6704c1b82df2f90cf00719a4ca1d6cbaf7682571f03c8beaf96d86ac489dac7af28b055d684e9d1bfa14fefdd44de400b4cea24d3443fcb7b4efaae2fa09b797dca99308032aa7c7fe99b2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152596916Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d, Name: checkpoints/4288/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4288/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=c09cd05c1f36fef18155dbbb11204333cfb2a5e55e9cdc596e41cd085e1e1d704f63d4ec6e6f0cf590a3f0053e52c46af1b0821b44ea5e604a4472fa42bd14f69784f1a30fcc4a11bfa8d9370dc05e28201ddfb5b14412a05b901a131ac09f0d48dacabc2345466be6f5b9498efa32bcedb6226f4d1f45f7c78a8054de632a572d8046fe7a8e1ac59fa1953981e04f41adf4f1b94699dd7328e092a2f9783ffdb343cc1ffca5e8e1d56e97e8f3f550763147d8a9c8fa505c0af54475c2e1577ce55a8cd1dac0bceab3d8106e4c4ecb2129b719ba0e635564e589cf8c6328ffc51baddaa28f822c13ffd433ef59df5546a0bb09843fb94c932906e019f5f57d19&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.161502828Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/array_metadatas, Name: checkpoints/1664/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1664/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=1429af9913b9f7e90efa455b0f3065d6b9f84d91c5a09d51376e032becd6af97d38753b38df15994d0b97e369c9074177a85a0cca79fa458c9074e6bc9af3b3cde09605ead8fea3ffc231ffd2840e55db4fc5c92283be298ae154e49aeaa8c216d61d95a019731aea2c365f3c9b26642432c2a265862669aaff24d812e7d97d57601250a44ed488383a4ae5f27092992467bff92b9144c73933f74a73a7fbd0190d9f57cd763da0468f8a9aaf4ae1dc75284d425390974a1240849d29e4c815dbe00befd28fe055cd672795a82c99bbe75ed0ac640f8dcb37f2b515450763f77fe3270acd414ab9a5182d7282a80864b8c4fa3cb2371342f6a483ec36cc439c4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.161516856Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d, Name: checkpoints/256/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/256/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b3c2eec8cfe69a9807efc6fdceaa80b619ed3eaf9b4f2fd8cabbbec95219e03ccea28f8a997b45958fec1c13441096ce8457d29945df0fad74be06a6d8fe665cece450cfca1f67d9b6e54d8bda030660e28748c6c6e794170a844e098dd4ce54a8dfbdc185c7860c6a11ecd2dab28bed569567dc6b45bd04981fa19e4eb5c6f5703350b8804954f17668f134442d046591c15f901872befb8231985660aa753455b377bf6c6f907c75112800895c8ac4057e43a6ff1ea9e47d60c4074b4930b994d032165f2222dce5f0c1cf60e1a54bf6563db0314c7bcaf540311bc0d7816708b8eef044bc37bd7175dfa639feee9e616b8675806b1cf99e98affb21cf591f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.161548494Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/array_metadatas, Name: checkpoints/2624/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2624/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=341a5a8049a90d996b021f59b2b76533ebf991ff8b2ae8ca90917846cc9ca90ee0220ab1a6abdb3e4fc8b247082976460b5bec12151c50e65be286d119e4cb67962efcb96570835be62e5302bb0243250aa16e4baae4e83f69cab1828a92dae0fd846fb2c20a5e79fa67ebe0f1a57f098aea9bdb29e79aeca5482402612bfae8622a6bcbfea08b07f27dc25c353d84e3e84a6b6b878a7131c6d5061d28ee01dbf8b2212e0446462bf447342bae4e7e5eb0b4d4dddad89d33d7a06c45488f107472f28aec7c75d49748c44379b0d5e08790bf8840b868364d6ea3f5f30c417a890235ea8b7881b83c8d9d6080d3303d1c1a01bd40811953c3daefc9170ba9f716&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.161564645Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d, Name: checkpoints/5888/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5888/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5827572c3f325b16f4a3a0d7017986fe3d03ff358d4c85e0a6f0a83f43edbdea4b0f2226144344e6d2ef58ef20e2549afcca7803240fe9a859e376ed1d54a87a97a0e9243c6767c474b7c8908df7f1f69db4bba41feaa5db9e732bce30730607af22be0bcfe5c5a2d4b0e8c341dd8fed29e30be4795cec3323fcd16df3fc1eb8479547210eea579b679396e63602c6e83b52731f37e5843e9f9dae545f4139d39c95d10f600c656f45472ee65e2d92ca79f44980a1a12d6712214da350cd4174b956e371b942138a9a03d0f33a9487cf938e86a7b60a0e432e2897b30edbf1affa747835a6220766a532d2a05ecb430795ae1b3e2605af1ccf804eedb5533a0c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.161647938Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/array_metadatas, Name: checkpoints/3904/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3904/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=88fad993450d994378d44b96a15cc12769ec119f5bd21f952e9147fd6e2d39d227e3bbea4a3c8a0981c85799d6a8e1d34d3de56d0fcd2880f63af1bcd2145f04cb1a85550d06be0876fbb5f520703a08b5c29365e6199b93704aa2487c6289d63aca58a50e52d27741a03e199ad1a465ad9a630cebfb60e5857424e76f28b45045eabde861656137dbac28f6dfb042f3e491be6bad7ec63dc1dfafd8e170c834d9076f1811d2018935a0515a1075fafe1a6be28524a2b7cc9e9516bd91705887cd01628b823816c4316e3ce0a4f5de2aa069a7a73f46aa7277c1a12564618441ca8ca0cc85b8e78a6f92c4b8266844f2eaf14f98b4c79cb79197d450b5a70a71&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.158675216Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416, Name: checkpoints/4416, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4416?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=c39b3bc0c07e577a8a23c9c4683fbea059c20c1d05602c20f03c9c3e565fea5752af95f2525db1a796cc513c3cbf17e784433fc706d44cf3f88ca63980d12b951e9dfc619dd2f0cecd7729a362027d8bb11ad9a858865ad04bad9a117b8f1695c0826c1a8bca1fbe9de6b65c3ba252272b8d508e734703a54951b46362c6eb226b0ade86d19051fda5b4e2e85807de7e318ed47c4f8e9d8ecc23b21cd49df2df12c9f03227ceed57ac77159e7ae6aa3af014fff9c2234b3cbfda6e1c74ca48f75d36d4812bf30b507e6f00060a22fa7a4fdf3e7b47b900b71066ec4fc5d4bff986247f1280f8cd1b4e4518598e324f7e150251180df70386e01c498087d2608e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.161739291Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d, Name: checkpoints/1152/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1152/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=c2fc1cfa948d01dbb73362695c2048efe6a024b12f3e5574208426da5daebdedb709203cd1d5972f7dbbf47ce90f5f4d16a0ba14e122a15d64ec391ed4d5cf393eaa78a6d0ca6f2509d24068f32dd2df6d6d0b908a37c2ba508dcb5ee820e3181ede92d2193e3a6203adf49e5af474313fd74c7cce8977af44a7a15949eddb197a3f7040cec87b8548292618391296e4424b4037a810b41665d383dc19a9e1584e2f7f8d79fa8ded4b9231d787c042b110558f1cab39c24136306e54198f86474088b492f6d86b2cb5f77f8739f231238a95930d0e2144a3d5c110d5ece7b1727c056ed127e1c9d309098f1be5def5ef3b9aca152879b0efcc903880bd0c8fbf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.161771526Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default, Name: checkpoints/2240/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2240/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=8f2e5f7d6f35d919b96a620e68f29bf10fc5614d42f79568457366b5b2c0ca322842da016e2f5920b53f3f7ebac6a2d8b673e30c9f8f035f60fae1ff65bc7ad6a68f7a0f28acfe32880c217916698b907cbe56e086b186d3ca23663d0c913102a1025e758ec3f9211b3b4053fa355f0caad89ceefe344fa1ceb43921cc2049244b7b56a54fd26fe086372ac2e7d3cdda5e186b4498180ad354a94d2f33d8a31a64e596a720525cb97f672403a80c2796f23bd547c6223c659aea74622b987013209cf3fea7f09b265a8bdd3fcb1da2bceb34fa19befeeaf3798b1cb8b349b85096a295615926aa68677068e4edb39c3c2308b6a266cde476a15c67001bb64e70&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.161801043Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0, Name: checkpoints/4096/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4096/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=3695277b9dd4491b4fc3b1ceec56f6c40a1f9de0796aa29170da45b6c13ff0c69e8e88a8f9cd2d6f9451840737b5da6448922b42475800b50cd61f5dd1cd55a929a32ee869f25ffff552710aab6c2127027830d4646f222e9ef259d2105808c09abc376cab060799d935a8e242ff9dd06eb0356a91b405981c9c96520875c309a56b7584357c8b8199b6c0e67bdc75683ebb38f6f820616fc32abc7d65c81fa7b0431ae93fd83933e5dfec10f042aeefa63d4c30de131950da97a17f5858c3ff8e1e1073f37abb1df7efd8468d62eb843f35810bf35a716715463c45159cefe31e469c98020ee356f0e83e310be16bf78811ae9b647693d04c5a28746b60bd3a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.161847004Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d, Name: checkpoints/3008/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3008/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=91baab9fdd2adfa35005e6d3c3555d2dff4bf85d54a457f54b11b6b3bbae542749ebe29916ba1374293e749ae97f17a708ebc6d76e8b9d4983d71b1c2215ec7feb5d4ddbac0a2837aac76299db427488c3465386f4a8355ef68e2621d96c3f8dbe11382a43637bf08f2ebfea4d9c6d483db4152ba84ed6a1bbfcfea3bd39d976e53515a15eca73e274287daea6b2583c88eab72de8aae1b6dd09c9734f3bef525f37f828866f08faafe5c21828be8ea312a0f8c9d3227b9f18e4a4888e70a4ae60812920d7674c0677d4b7328957c7beabdef93ac02c601a6ca387f0c68b571fe20f4840b4779ad35ca06b31572798819241f6fdbf776919ac5676acaaaedcd2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.160520214Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d, Name: checkpoints/2688/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2688/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=56fa09e83505d2dc2e30b66302eeb74d256b4b49a4f5b98bd21365ff78a817f08819d552f5ab85b00d040a48d1e5d5745d737a0dcd701b76ed138c5698f43bb89cf455d62245998dbda478e14cef221561507033546e67209ebd36e43e01fb8af9521c15e8ecf9bd3be56a3e4d124a75c9fce6c78464bd44e794cf322de2f1079fe08b4fae455d51db854353d7ccff49143220f90b6d9da3bf4b381086dc0b02e355aa1bf341681870942e7b12c71f805e525d7eca261de57f58270ca90fc0551f9c61f86bc951691a2ca15098cc7984e1df6dc3e9e6341c17d24d89aee0559d2356a66d526b79aa603a865ed963be4ff96bec3c4dd2b3a74044408f4ce065c4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.161516544Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/d, Name: checkpoints/3840/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3840/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3cdb934228e73145d45a2fb627a4311439d6d39a647f2e09755efb4cba7c702e599b6a4d2411308093c324a00f4eef656f26ecb94975323dc46a95cb96b7ba9fac1d6735d83e9e6f5d21285a4619379015a635cc7ec6c251701d1cb5167539810c970331728f8b3466aa00bc8cdf7bd2f37773482474bdc03bdfdd93d24642102a34c6f07092e1edbf2cbc69488f6903c38c7662b722f3832bf412f9844583b13ca64894b216d88550558159a70670be065c277f36d791c4633b4310ae768cbf689a85b94cc562ce58db093a83972476d3638be2d0cfd6c929dd17ca47a1d0f057784255da0d895a4ad2aedd24693821064b0f62fa03c0491e194eca3d1ec4a7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.161978691Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0, Name: checkpoints/5824/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5824/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4e4ac7d61c5c98cd27271802cf25662c6d10b830bafe617665462dfa1b01ea0b8c378389d5bdf85be3099ed0779d9a9b8d469e0f01b064c42ef3bb488c188d3e75866bbb3869250f7da730ed8bc11bcba97fd5496d547e897d56da8b8c4cdfdd273c141b2753b6c81e18d56c549489f892628270e5da2b7ce474bf7758bb7bdd8bf12280222c377b521a1397910a57415e2e1fc3b758f2b07ae61c816bf8968f161f5a14646cce738e623c7073f54569eb7fe46619116505808aaa2c4118cc99c04adf054f11a55722801d354a08cc403434bddb42eef608b80faad02fe90729d4eca3a59c7ddef07b9bed43ee927e4f161860543869a94703a8822077df0428&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152969538Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984, Name: checkpoints/1984, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1984?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4c1245ede63be5215565b3021f6d9a7970f6a91e863bdb5c317e7bc07596ff6a86353136babec9dc4fd8a9fdff3bb0fc22c6ba3582e3ea95f8c2f5fe271b294807a474a5c9efe55e70e77f31b6ff5c8c8deb0ea69b9d1c98d0c116c897ced41fe473339c6b208fc65a03744882da9bd755cbeda7c23b15e22263bc9436c1e4943d239fa6ce7540929af9427b19c4b70a2606f9d00deebf95a588562cea1fcf8a88e99ef0c82a9a65a039d6ee74142dd79f2bdbe38fad9b6af74f8a396b50cd1806de50a4a44be212d64afd33b0817565590b5c6b6841762131b9fb7419933ca2bee456d0073ad5f62b9dfb4639e3c4ee2c4a51f18c8e7d0403a81345b2179a06&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.153412858Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default, Name: checkpoints/1024/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1024/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=18f55afae89fca3fac13248af6846b862b5b9481d3217fc82de168565e0619808580c651db3d1bbf5c4fff0aca12312b77e5f835ba2864ece29307c01f738dbc81cf08a4094e82e4b4a2fc6305570efa1e249e4b7038cca3d58fdce022f819fd5c31374bf85da763c3b1a90de4cc4c79db85d8892aabe26883792c9d2aa7a40e7df23213ba662aedcef2f5d656ccfb7f9c7c886318549643d2196e09ddae343b3fca9ecec6816b4af761ee8d322997dc34c32905f68a80fb827a918d3c7364967d34a0018fe459e43cebc1983b2b7315ca46fb35fe1ba12f6452e5a38ce456d7f17eb570eec50d165077a40e3618ea7dac9e4616dbdc1eeae6cd6e6e876de028&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.162263593Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/d, Name: checkpoints/5888/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5888/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=587a1eb8ced63c76f4215bf936ca4e195a616e909fd74b9906e75f26465616ce2e324c0ae18cff352c2c37667c8f7d8ff1f3a2783ea8f569649af1dfea0a281778f3bab6a62fd59a5847f9171c5e3724fbbb7e066303932df06401e42278b9f751f68f8decc103ff7b0f49cc525cd86130417ac6a7bfffea6a02ab9c73aae9be8ab0404b886e7ab2e8ce86ad13658856e18c524008d3346ba5c093e2c3989d0cbbb2a8f77e07ea9e3bdbd0f94c74248a442a9b2a85dd1e2562b26d7d36a8c9fc582223f93f4967a733d4a31216e1b22410bc5e9a05fc81a4c3ceb6b62b97b6cc3bf54995f0a4cd83706c5cdac5dc6ea3a1b45cb952560f46e001e02e899c05ed&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.161583872Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default, Name: checkpoints/5504/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5504/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=2dce9fba27c66cbd79651aaa11e5f340e072ad564f38dba1bb210498fa724fcf2d014a03872484ae419b88f47a1a76eeea57f629a4929864fb8167a2b55d57ad8fcc5e3d85144e53a73b6c0bf480bd45f22b8fb1d4db99dbf32cca81ecac93d318255ef787a4d48e0c50c1d0b416c4f143b27077f978bdf0078b5f96587847bb681dee241dae09c1433fe9952ff5322af8ca08eb00af413a9c6596d24c828408dacf183722393a7ecce437f826f44b75fce6fe07be889170012a0f92592101c35cdada4c812412b4f84986ec43efe78032c403e51c44d5472a2cf8c6a9782a750ab5bc259cddbd01e7a290a73b47e2dfef285ad043fa68bcbb8e89fc984bd78e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.160585003Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0, Name: checkpoints/1344/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1344/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7cfdd3f7c0781d24deeb2bb47f869b8288f163b64b4634c24a3c889f9191d62c791b38815fc7c7962cd3ee8e1ad364b0f22a4af81f3b053a23d70de0bc20dcad76afa8c64ddef9f56bb0cc0a339f3672e9af9ee1a3e94b231c5b1ad6f1d91db6ae54427c7ca4ae754b6657bf72340c271bafaaebfbc055c09dbedd2f042d9898353e9a43b6b3beaf323767c3519e8ec9f660a7b67b332cd79c9edf8f7528761094f685363ee2ccd07a06bae3ae61e2b4a9d8df98792f3620faf9318253a7ff6039cb3965640bf7fd53807279875156bbc7b2a8ea66eff832c50a9d5d32fb984b19a833bfd79dc77683dd6ba2a4c5ca50d97fd3c0631eda2b2cde44399369f316&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.162412542Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/d, Name: checkpoints/448/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/448/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=77e57920e2fe204c6c1ad080ddd695a7aa3e6e5b4b7fe9dc87a03d45a9f418b7be023430d18151abeb7a721025864c16d30a9eb67ea311ba76286068e1901c28558bfc7c5db0cff5712cc08d92600e8d274f49b08af1142b5a3634de5763a06afb715efbcd51fc10777db564c5b68f7944470ce071ebcea076e9e3d0814abd05e575253d67e975d40258c3f905f6ddcb45607c6d41c49f0b5f1a8ab1da22fde44d37e5a3ddbb7a4d377116852fa78cc1a0b44067112017dedbad0a51be914c849d8f445c9e5c70de5427513527f719c063146e28396dd839a872688f619b21fcda4fc443d9c12142fd1b0cebb840ea21dedb0d47aab6e602a89d36ce2b975cae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.162444064Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d, Name: checkpoints/3968/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3968/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=95a24529c1122f17180d526c42ec44af742245774ac911846f76eab104b64f55238bdab95f12ed3a9292c3c4714858273f55c94c1c369a449b2b41d13c754df5906cdc36e95720e92a76da6eebf38542b33a10704e0791081222985b8a4cfc9a44999e375768e320e1c34cbeacf36df8e464feaf9b16cf65582319f64886644b52221e4b93fed2f4acd95aad5d3a4e2506fbe3805c539abca0d094d0000f1b2985c2fe834dcec50327f2759790692effafe416d6995b1462793c643fb3c6ac88d20fc327e0c5eaf041c5de01225ed8c842ea4f8635bdd900c8744b6ac0fb60f56361ebe8cbd7d4cb377d764c621e941e7e4183ec34fc4c6f04d7899e237bf2bf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.160625738Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/d, Name: checkpoints/1344/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1344/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=49077bd31a47a332d5a91079484b45d90d184ec9ac9e3c1bb885a9f0e0fe2dae4cd2c0618e4eba654e9ec5d5fea8444cd02d502dd26887ad6b06a5420d2ed24f63fae85ba2b67d56e8d9f0610c9bce397b94dae99ecb3fcce09d4d746bd7e046552d2129117922a79da4ca5cf5aad8f73f3f3f974b60b35834428be3f09ea4895238b3044bbc17a5878ec3e44679d7110c17bd884ac49eaf5fe98bb0eef87b62513304d2f453de7c6ee05861006c807e798b4b06ce47a6fa93681744040a00bd8b397b08696a439c4c19b87d71dc5eef99fb6eb38b9f8507cfcd87127a9cec0ef5d732717f4f6d6e6244d505816f3d98d6b5e0178fdc7124f2a8ffe98691d19c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.160637155Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d, Name: checkpoints/1856/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1856/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=839f54d6d59c8be8910aa376a6c761b9ce45a7e40a89c0372b3fcf7804767113835580c7fbca7fc62246c4d829003576f668993ccdcd8947edd5245ab02077fc6933b2a5663ddf374284983a6a1d81cb5d992c9452402204fc47546ddd8981a55fe2d608952d519c9a5b7e9d9a17fc4201960dc4ab7e9385f98382c49bb6b6b14f718be51fe2454b2f6b78efc7afb21690af9be70be93940a59e424c7ae97bc9a464a99b663408dd8a6fcb0021d0d14c4252abaa0fcb11b68c75add74e2d5fd8f33f948d6676e4d70062ae0e8b4fb05b3efcbf90f8b9e6278ff1ec00b6b3a8a8fb067fa1259100269c53b18dea623b8e86ea267e0c3ae097137a5ccc023422d3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.162692Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d, Name: checkpoints/768/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/768/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=a1c5db8e7837d8f6ca66f4b78889a6cc5177520b589e4f884fc3ebd8db9d5ae4cc24cf6d70894fe7892af4b5a2ab91e42ec4fc01c2e5a963e6c5a3a28fa751a93289ffbdf30792d1a4804a98474c3542666e744da22eef1c6d3cf53128b8f47a6bc4d8bf66fcf41a992b0d477374d1cd6d49964bbf7aaa5a67f539ed21cb3a5dec78de07a0afce1aca9aea0a7aa841cc40c3bc7ff84d84d312f25da30049907b725711393d328f9e9d555f8d6da3ceede59f00b7458898ba6cbdc80a819b40800df835ebea53914d556312b8b2c233effe764614f6e51032fbbe42c2146de1211410a89bbb5a7de14ce8805340ddeda2d18feb29d36e165b5509d9ae8bd1cd2d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152864023Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0, Name: checkpoints/1280/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1280/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4b366a9f18ca9fa5fca656bc35750eafb23fe47ebfcc131ebef346754d422ac461f3b9773f660dd975730ee4c1c9c4d13465c0cb8efa744735cd46200f592ee240ad5140f67d00e50b62c2d2e291934724d7f0a204936bff58f42d91dab4cef8e58626408fac2bb1810b9415ecc84ec5e2380bb0cd3f530afbcdbabaf37e0a5219ce2b9dcb1190f5824e89f70661973e6b5ef7a9ced5aa4dd8ee804cee78afe4e523c53171f029a1a137d63997f002dc71c49792a81eb43a288298f5c97f4624ad0e1398269623d3d5ebc0da9c6b9254d2a1d205ccd7e24620d727698b102c6c772509c5d8f58d1d9c5727c9d57ce8df27f8a41b61b8a2acdeccda4f2e8170e4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.161454268Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240, Name: checkpoints/2240, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2240?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=affe02a545052614bf12804629753347755940a0837c8c274819fd1d29a4153632bc6c1c17b9c5eaa17e3f54f4e945c3eca39ecd35a21fba4924c71fae312bf282d7bac04500b21d4bb07c523966c03190059a8f68e05f53639cb6d2cf0c19555bed6398fdb5218f614799df34b5ccb68ac6acdc3d8845085e9c150dd029315d88c9de0faafcbb7352d5b9b7b18dbe176ca7b7f4dc6a07a49b6e50ef4f8ad5aa43a0562b94baa1b639ed44f90f9f893505c726ae03e37536b48fc08250500b141189516741be67e227fb826b03150befe3c3c30751702902f891b2d8897550de9944b230d855caaecb701941fe84f7aebce8163cb0d8daf19abaa60fd7e6376e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152295208Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0, Name: checkpoints/5440/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5440/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4432ea09e0cca83493efa0117c9c0744fc4064d60443a82cbd3e3ebedd9bb084893570b4fc00b817e880b65a6fe477859fee7dec12cd1508c6e7f403021ae0b71deea93119a922dd44c8328015c6dfeadd1b3ce51e7e1d9df70d3c7aab512c83f2545f5bd3f2aceed99be2cfd792f2fbcab5eb9904e8d4a83d543f140faec6e2e58113c44c46d8f6becbf299b20d515ff8635740be2d306bb267bf738344cfe231f865da456415f78a6b41f8c8ccf02d8ff1130b347b86a6d15588e8ba19511d1077ee0e6c0f8883b4ef76df857f627d052f8923ea28eda59f875ac557da2de6ce939117732b9c492b9edad8fda39713236eb86c01702ce6d0a4f1af2a2ef170&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.161508374Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d, Name: checkpoints/1792/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1792/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7aecc84501ca4a5c1c0ba032c82a4d2be37d19ca910e0fdf5dd8dd4c25ef2bec6c041fea56a7030914d79e7fe54d84dc60ad0e78ecf34cfe8b433c36abf4c5ff4d3c2c5f1cab8acfc22c563e09ea660cb228d4e1437e130066d12e4922a4d474b68520ccad126b78c80aa4bd464dad197245f5a8d191fa5f3416a41f8cd89b17a21139441cd1a801c25127a06811a56cd39afd5b197faa39fca705c5bc8168369d5108620703d7d5a60c6cdf9545e85df6cdf65aef063fa5a2fc18791b7f5f95a0ea1347dc096c0a58cddb09a314ab8ee49262b9a8fb212e089f50c5e7b4457daf3285a1a0eae30c84a92c75f4de2c391c350df41e74d4752512d15c4e8d8420&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.161499263Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default, Name: checkpoints/1728/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1728/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=23690b3825bbe0c848c4e389113297d314e101898a1a05445f61b286a3a933ff6be6652aec316f8be435a5a40542ecd7b8052158f6f4d1c44d5636edd7ee284c8e1a4e6e3d3555a2b3b71b3c75ac7e136572a5e3e9c1cde2fd1d1da5937f80af8b03f340636c8acfb76a4433b274937c8cfb623393bbcacb091a88271cadc7698bfe88faec9dfc0215d87b62bc32eadc8d1312f1b83a7254b8b1a62e2823224c5d4f4f01384ba2b680d74c427791c5333699e34d823e54c82682e9cf1c6c8a139a71e855c159d0c178f8e6a786b41a26d5b5b5e6c7728438348f4c97c2927d0a077a622d5b8ce771e7961b296f7e15e50eedae3d21ae59661e3b80c0f959dcf7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.162321081Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/array_metadatas, Name: checkpoints/5120/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5120/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=bcce7497ad077263b9199acc0f0c943556668f6b06e06b915d225e4268a888b5e503fbe474f2732882e4934988f598b9527de437a09cd6ce63cad7212a893fcf5985648b7576f63f132bb12867aa272d631ee7eb46847abeed3ed5731cf96dd13648dc3f82657be7e57c562983d308f1b290c4701293167fdcac35bb9393169a5e8e9b3543e28918efa152c34559fd3fb158d8c2c4d7f503dd7fd352cb29b3cb9cef280d4ecb89baece5c382551944b6ec5b2db29b42b434ccf43f31961a6c5375fecf741c2414ac7096092109934abc84388ef00158b8cf864057add87338016ed668f76678edbc043b743d3e3de6bb2fbf087c91d3ab8aec65c3db26cb71df&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.162377744Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default, Name: checkpoints/5376/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5376/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=68ff2044671dd0fbcaf19a525129657f6c7c87f104884cf8ad901a42287d0d7e06700586e15cf77b3a1a83149d7dc30de14d6b3b412ed3f6f01c2dc46b05737f3e1a25230a3c09834f733506c6a07cfe0d886012d4b8191c770b7f2160411936a0af8dde5c238c5ba1345fcc75e26faa07f27ddc4fbe13bc57b5c6dfb7dc32b6421f843c365d2b0fdf898e57c58445a76feae332d9e6880441fdcacf52a113f6fc6523f65ee50929fd202a5fc4e48bc7178a0969e73acaf9aea8e5f85b29a889c976303a6448a8d94267939c05b3059b57f6253a28252ff62459299d1f8f7bdda000c031bd088a5caa0dd73c9a5fb7d9d95f3b9f453ff5fa5d6ca5bd4ce8929b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.161108196Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d, Name: checkpoints/2112/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2112/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5ebfd059953881ad531321464695b60cfb042f4c7f328ab12c7b7ea2402ea276d60974618b527466fae98b26cf27497586afa26b65fed34249c174ea57be6634e390cfc97e40b529bc4ffb93317f2d13ba37bf5bbdac1f1fd4677b541d654d44717b27259f04b9b5cbf1c753b136b1e0d1bed8798395d1ec243b7003b4995a90bf8726c5617accab856168e605f4e9fb517150b6fd7f673b75a17c2d5b25377ee408c17e81ab3c9ef5214e0e897a2fdb30e67821b4b4c629af1f0d10000d6f09b101e1eb29203b5a9fd51a30d452f11e56ff19ffdbccbc942d8638093de3d378ec62629d1d49b5e53c735844460c56b4e4e18d767ca73e3818fe44458d670c01&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.160696315Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888, Name: checkpoints/5888, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5888?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=8b9e8dd234e3d4e7b44cf428ab8b517d1e38008e4c948fe661973b38d52436a5756d19c48cb4119dd0c69a6536d0cfb18271349f61eb691c2484315d6cc2b30a89534a82854d287f83ceac0d589bf49f3033e726a4b692a0eff59221054d2883e02b65a82f432ab979139cdb188ae285ec2c9e93d094cd74863807a1e4d20f5a6f8de049dd492dcba14c0afd13e00d8af74a8dbebe965f321f7f3d06aa6ddc396da02cdf5dda7daab3b8ac01d44a4cb1a815f6832d161ba57d79048db23ff04dba47ded3c1d0938ce9605cb5c3ccfa0bf2d9cbd4dbb439e38de2867b80c6016e1ebe40caa1533a49f1efd50b16eec497dc6b88f20d7e246ba4a1b8f53e2ba912&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.160625313Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default, Name: checkpoints/576/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/576/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b5261b80b97037806b603f598f087bfc8469f52af6545e573077df1c070d88dd5d21c92ca0b276f495a43c84dbb3c3266fb20d9df16f3f265a69d5d076226fa6d55ee2c278126078af294cb2c90447cb2334cd575c350322d9d0c88a726783b43ee4881564ebbb8e9183b0afa325389b1b2f04b468dc238108b2274de2fb6ba4e40077b35f9e8cd7d8efdb2250d74caf78f681826bd3f05be7e778773280df0819a1a65671331de8a8bd779bccfe6ac10bf45add7828ff21641a935dd8f689076567d18612d94b2e1b5f3cbc65787fa7bfdd8a39d6d8c0e643e97e51e82755ab706bba1033e82bbc3098c73f15b65100f90bda29074967d6e817e6464cc25461&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152563947Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d, Name: checkpoints/3136/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3136/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=2f4ca65669ae87cf0dbeb75ddf10339d101ae81c8b48b3677c5417ec578d6509da2e01eb9869054ae527c14909019ab4307b56ccd542b6c38e8c4df734006627a86f0f1050babd114bf6411514401c4d7b7861bf78bb187f9d8c16a1a9f2f8ab39f75e83ffa12451a80d587f36d0714ab7f18a9c03e5cc627082eb9e00a817af5125e25ba752f5db27665ce393ae0c11a568b75652409b4bfe4e29b73e839ef5ecaf9f8e225c314e6d17f389a9e4332925d1b628985a6976ae79c4b6c3104992a44a06c241f5ed32a192e8b844ce4198be5f9cae615e872aa6f50ad30da55abf1be4b4fca68b89f385c865ce970d02b4dc920226c6b066df0beda1176713311b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.216437573Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088, Name: checkpoints/1088, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1088?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=bf981909a1f56b41f1e3996628fc6e3bfe991faec11a90d14468e835e974d0749c82256da29eb364380b41fa77f9df4e3f9c77a8af5c5db17e86bfa18c2b350107d8936a6dcb3b0de9dcaf98a45ec9899f77b965a064c2b850be921e59cf25e4876bae8ca4cb18c948d6da7a380eabd91f78f67fd5d2ca16b7129f2c393bac796beacda9770ea6fcf53cff1e24110766b791a67ec7616e370cb4eb7e07eca5bb6eff4e33dc8ca87c368d6ed9b01e1787d8eed0b3f693dc980b19df9a0e7a4c1410737b7e7273c6ff1440ee6b424b8b8f66c5b177d9330a6fbbd0aaac311408173927062e95079d9a44e378748a9e389b7308af09d78d81e4e7539613f425dee5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.217449669Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/d, Name: checkpoints/960/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/960/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=15b3b02d8ec872caa124fe6e9f9ef02867ae62f04a584557af11f9a0596d5a7ecb69d9254f1027393bfff2cb52a4adb632ef39598f0d0de3a9eede6d82721985e7c04b05face9bd7f8d4777142ea0ca05a2a0f2c20103e803021934d5e33cd8bc28169f69af09661feeb4e76e3f57f2d086a413d8ebeabba29db7e38e9f19acab9d1c1396a0bfccf41248d795de35081670f59dece159850d21b1017f52fc2ee55c0acfbb4b89f488ddc9bfb947d3485653c660356e4562f3dbb0a9295030ced15aad6a14e02713bfa620650be76de07a4d656dffb05a34f28f3c5439efeae1f4b2d7c04c303fda6d56fe3df1ec676c6992a8d69865ac0967fdbbc514373b5b3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.217737305Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default, Name: checkpoints/384/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/384/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b50398dfabb70fb64ff41a003c83eb09bfa69877585bf2f1655397cb258d84b8fb84cec91109e2d8d88a7cd6e27048afeff53b4e1da222426694507b2d71559ed1fdb2bd149d3f79014bf922617da1328bc205254871c70cd708da4b1d2cae32e53ff604ecc18df7fb001cb91002df8f6a592a53568387c5aded3943a335d1eee75f29f7542e893694cb1200563677feaf3d47e58be0a27a14c7d5a819cfdeb2334f0d666f7ed25da85f94e652d56e154c7756b275543b8aace152be2dc7800fd2ec0ac48668867b46dfe7c7f2f5fecdd36f2a0381e33f5888e6c0a8a677b567361b25bbc7a1a707b4c75a953745cb52d2cd095e766dbda397da29be18b4907f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.21785186Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default, Name: checkpoints/5824/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5824/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5c50dfcb8d5dd04f9dc065f3f6f6cefac30a773315969c96a1eab5feae0fdfa56deff0fcf76e7e4305c07cce86dc5242086544afddd42e8d4c66c2ce93973d8af11c89979f01124260ccb16437a0152e26c33010883b96522e32a971bff4c46a6b7435c0b9855c73618952d6df7798598f7728bc7b429db43a0dbab2554d51bc4a63e1d704855cf7e75872b4fe0302d170e7cce1f7e13b797bc11ba427be88097a8214218a3419b6f39d16f1b6a7c13d03c33af51311aa613e5b4301631d1ecd6021fb17bccd871609bb6e87a7971a406033f8a158a2cdf7f2e1a5b2d6300aa249a19ec557969af2102196d0cd58741e1d6f8e83883f25553d5a15099d9fd893&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.21790912Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/array_metadatas, Name: checkpoints/3968/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3968/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=40074996a42a74a248003068ac25738c899a0d8d715ccf4a82b85163228fa4af7b4cb18abdab66be019a3c4a04308b0aff50133f562290fc402467cac8c14a45b07d7ff66947e312e3ca16912f307d57b502290a2737d6b8e7cfbfd66bcefe396c5df1a3b9187a7a2ebc32d88446473d7451e73f86b59664ce9963a0754e14940b13aaec17cab1c7b11b1618876db6cfa49d06e2b15ea1bc7e27c29d7dea4b06681c2f8e5e46c0cc7f6c1248435ec20caa03cae1928fce50b84338aad6bdc928531cc2a2c25f891e4ce95286dac964ccd2424bd82f1d9cda61d1cbbef594e27e7afe587b0ca95323843733b31a8b13429c62d5e5a0e46f82611497c1c363b137&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.152598435Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d, Name: checkpoints/3712/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3712/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5d8c34f618a6112a7c1f87ee48a8543d4cd9e0cefbb79877859b370899b95b01401cc6d9e5cbae256a03ac496936f129a92c2837c161229b31c37e86db24c227ba1ffd2116a78ac8629b6a08b247517df8e984cf2f868277aab849fb5b220b6b23f264c3e34f6f94697194cc95ee00a7589aa4fd69c7b138453de6f5fb1e3cf419f3402a0fa6ca6d6bd47b3cb92e3c56a478245863206f29b014ce4301a9fac99b4f9f42706fb6024eab3ca97b29c7a9825e3b82b75838dd3daa9474fc2cbee7cec90cd7d40cfc17286e805b8a55f3e2d434a6d68cc1ca0b8bfb49d03bf99143974cd8c3bf2a8f58487d4420e013aca7578867c5fcd4449dabd78d84256ac3a8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.217967308Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/d, Name: checkpoints/3776/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3776/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=a5032bd602f4668012b57ace248114aeadee1d57eca31be36790840e68b049ebab404e2e7490b97cb7d915c556d39dad3108ab1f9993b12359e4a04829e5ba83f79c7a66c5334e3935bfbebf752d1aaeb7396fa079be7cb2128427f77142d70c3d4c25615cd94afbc505e15a4ffd7ac71a12deb85321d6d5d402f81a41db8468769b4d2952b9cc349ce193111c6bfdd4dee3267c2b18a735a2529194638821e6f368795a9b0915377cd1a27bceebac46cf7b263d05ab4665ccda56d41154c6f46239ad2d110b4145331ab7bef116a1669dbe972a60c6556500ccc7c39a6c41166caf25f8b6be38bf37019290e89a362b91802f7efdaeb7f51e5fc87948541097&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.218005862Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/array_metadatas, Name: checkpoints/3776/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3776/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=8fa40415eed9d07451c05b46785ea65a53d3a6a7389864596081152920f6de87c1558fceb11ddf1d0384a27b90bc20ef48e288a1c7fdb534ff221ef1001fc7de8920a3160b0d048957834798f1bb1e6a5092d60d92c22ab4093fe87f83e90af52e82b8b6e0460520822016d276e64baba5aca21ff1905186111354c6b319338681ec5116550b39f74a1015820546c817def002df724dec8da0b63834bbea2da95ce120ef6a621af611938d8fc81e1301524283ce0136788a722b94abb943ff9e8cf3ea7debbb6fd90863746da5ca0344446006697bc9d2eeb1d3dc9488cfe0e6b5563b2ae84ef8d38c9dadc0f9c8aec8381ccb307241a5ac2e7d7d8fa7deee2b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.2180318Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default, Name: checkpoints/5056/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5056/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=87ba3d35b7632cd8debb21ab3e5312178b28d9103bdf01c23ff409bfef05a614d050681be2ab6de9ee8809d0c26ba2af4f08e21b300ef83124aa907e1d2df1213a5d330a15174c3897d77cd33cd083020d2fcb3d9ed4b68a8bdbe6f73e979d5b2576b66ad7f4826eab8c09cadbefdb2f30b9b48da7a69e61aa1c092a73826c6025274735fc456844be7a7dec3bcbbb3ab2fb403a1e67400cf006991176d6502dcbff355c67aa838a7c5d968ce82697f6c8b1381cfaa62658946b2432498fe243cd02525ea459dacc6680120f54fad46e4c9ebc6ed05188904ef4c41b1e1f9538492a203020f718cf33c4b24ef055712f037e5e5f91693329d9ffba4d8a5abd6c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.218107827Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default, Name: checkpoints/4096/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4096/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=95ad05f078344e7a1e708cadb0a6362f56102a292c81ff20a6845c13173dde2270f8db24001f86f69a4cbe27e3ea7aa465ec6b094d573884c21f0582c0c0e62dd083678347012c8a23cb46562a339d92e695e864a873b7031ee83a81ee7560b2d5f426f71745a0b62ac8f50ecebd45c8eba98f403a70c82c6d8b1e8f5abc421bfc3b56f58e362d5540bc80221360fc4c358d51274e7891d9623752b67c43d4686239758516d2b2805bfbba8779593cd03b9ab435732efcbd0a46e5a6845d7158996200085dbba942099cb492ed070d55ceacfdd3590510bc867b0d27a3b706bb56058d6c10c3f23dacf343fcde05cff3064f798c35e26d1da442bdf26baaa469&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.218267045Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d, Name: checkpoints/640/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/640/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=a0cd922a4557f56d5b4fd4d71cb0e67064ab3ff5c530fb5beb9bdb54a06ff8983b900b043d2832d2b6f07f34f03c5967563a596603768fe4e391c1ba2953334298b0a419a0a569812593d063c76d989dcff73a0cdc36717827c14c6514e592a08fd2d808828a865363822132b31f461234a9de9dfcaed70db49c1862b0fc64960d9fdf09dac17a2155cfe03facde20b58cd89610c354269d6f680f3fa68dafb06ec82d53ce9949d1a3a948e15d760ae3df0f180ad1d7c1f58ae47c6ce89227dec6b8aa498b5eddf6f8c47f501e237184ed08deb186d536c338dd52df9486d727a3d8bc863a6dd9a38fd5cb5c15eea406d21623852d7cec5d9be6caea3b7635ca&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.157988967Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/array_metadatas, Name: checkpoints/1024/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1024/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=20a560acc6d4cb3fbe41c74219241843e1bcc348010c9ae2a6e14a8f21b935936c8ab4872a85afb1035f7d7b883d8e1e535f57098c6786a8873a524e75169be810a147e349b812d881453c417d989c222bdc4536e0b05e193e0a367e1f0faf2463940cd6f629d52eb4a486466c8f570d310e5c28e69400483df509f3033315bda133b233c72eccf1854572805c66135163c823a4ca216f372b8808b9c0f00f3c8141a9d1e8f157c6330d0080e0d5d9c74c044d0d6d762b6a07bfe5f47778cd3a65786e01731940dab717138c3c1a617a74f095e4a3efeee5ba6531a186791609efce73d0e17005f4e210a4d6b84767c5e2b4ddcc6895690977d76f0c39168644&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.216526344Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920, Name: checkpoints/1920, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1920?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=56d91d1d5480500ec7dfbad9aa8fbe4d70fc0f115a478d6c52f6bed1bb92842ca798f02b3044d6ffbad258bc07f60a377a97d7a6ed0c557424311652f4ae355c08a9e56982a0ca9cb6b71753828de16c6b16a5df68245c43b484d2cd6f4306fc6fe443d66b6d7005dfeeb775f080466c0defb0a34cfc36220b47bf719d9088ae7256e85b5654173891330d6e9b61f3b31328158bade54093ca8b4ea4266444c0424b006da97fe3ddd3c563c8017e6cd41a4960711a82b2b13fd20432e766cab464d2f87284db89a905a69df4b1eb9c387af1e9427ba2020c5802fa72e6cd3bb0322e71e86a37852ae196b7b248cdbe2d9a2d86385c794b13065c436097b56e12&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.216564844Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/d, Name: checkpoints/1024/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1024/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=bb1059aec8dd34673f4b6b243d101502f0ea8d13f14830edfd5b83009e08b986fc9898744aebf865a3c2f7ac644d6da4a07f7afbc06db7ae8f520abe846769a351383e012c4988937e300411097ca037f5561e4a27df3860d3a6799b77ed6126780e13254d5571986ef27591027a9945d7e6d83897e566c7ac78f8e7dfa6f3ab2d10f02782de5064dee8781a56eabfba171e9024e65a0f9b4b2e3a235a03bb66b75ef9cbb735325300d3f9ba98e69d00583b38881341f683daade142523456b257fe91382409c80056a85a1c724d7f7fcee3a1ff4325261021ca1d70e5d9c6e2af5b3d6b71014d319d8cca21b8864e011af516a79419fbedfae6405d2adc08b9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.216199576Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008, Name: checkpoints/3008, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3008?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=986e454159747b20fd636f0c498a53165961d20cad1a96547e32a2afe76692cf7b8aa133a084c159128da87e4f84fadbd8903cab08942d09e54a87bdd85291b5311f3246f78bef6037fc5a9223b3226f9a44996f8ea974f83b92bd73a927b6f67b9d519b5b4311a9c46ac11801ed8de26ab765a79cca1d7ac530a4ac6fc7b3966e15c466f4e09bbeeb99e4b3bd1f9fba79b5661c9a93b4043f78c865516032306e144e5013fba99336bac9ee5ef77c28e24099f8b2d9e11ed68dc039e5e18e62d9e755e3b0398bd3bd1904987cca71db12d1b8d3f489734e85247dc84a87cdc5b76ba5f01f3ab30d80cbd21c11ea4496744d727b09127193482dc819615b4d35&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.15861127Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288, Name: checkpoints/4288, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4288?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=9652d95e52d1d2d38dcb36b1e556881d66fac2f27c9693a1e7dd65e0e4566632efbe0132133d3d344b6f9821f5b7b4348e8172f2077e7358c9f188768324e810b621688cac1fca548e0654fbb0feaf86077f20c3911c6c8fab170535999906e2a055b9b709f621e6dc62a4de59054f24e330b8e602ea14217a16465b114210c406a9f357aa38d5ac43f713bece1ec401c1c49a2e5c8ccc72667365853e12a47205c7e3341ab355b9bf1d635b0a745f5759374df4e4ab0761a2c9d038377b949ca10441f85f39a32800c8898ed79fe37434508e4a79f4188758391510c1e34ce25e6ccb1461781f89559bc2ed2152b6c9e2dab8adb95e9a83823fff42bdb3b785&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.218722745Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/array_metadatas, Name: checkpoints/832/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/832/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=778264567915745909dcbca41f3a8eb5c79aed32d8e2c7e04d3dacf9d59820090f46b6c26f77736b5e827f2bc0b857513000b852c18601748d58717f2ca4251e9b7cfc3cfa2deaa5b3f936e6cea5f12dfb2a157f658cd03a1201c2a65672969134bea72760d344799d8d464705c05f6969b10b389b7d21e81c4ff9b32c209a9792305210a5b26c3542aa26500884b7e93951844c63838aafcac3a123a564e488b26ef696476d13a780599c3766118a9236ccedbba597429404e25e91eb313816366721dc3d2eccc050d616fe28dbc7e61618977370eb374dee04a8fd910bffc39763f67bac29176e95448defa57ba34cb1d4e8bf07b7a91cb2d0906f59691257&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.216539564Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576, Name: checkpoints/576, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/576?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=87905e4ba1f924c26f9244e6094fd703306f631e72aa116fae15a671c397e2ddeb805711c95d9f70fa622e37c43fd38c9966e489d6c2d1b442f580403f1e247dc1ec0b40202923522f4b67116ebba6f457aa6868d61dc03e6b4d7b8e09a9b39b30aeaf4d558f65f6393f55b2a1b6d6569d8cedf5e777252cd6e7a77521bf3f5ec025436e3a1723c9482320709211232d5626001a57620e55d30253acac540e37875a6be8d37edd347fdd56999e7334c1d41a68edac6baeade958aee066013f74981e6a167fb85c1ea012c366b99a7842cb1f0a3fdd32bc1ebdd53efb7c401124f61e8df65628cf61b2b30a4d764e7f659426a74b3e9c0f0b31a037b943a659ac&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.217905717Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432, Name: checkpoints/2432, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2432?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=2045a62506663fa8bc3be5a181845f45324fe707c9cb6fc0d0ba6a137b64cf087761deb64059edb8d3a5ee789ed5a2b9b4734b42f4aea164fb0a2189757562a230d8bda453697be3cdbdd423f6445654cae33ca1232c0e23913d1881a80a704dca9d0cc1db3791a9851ff4c7b283e3c7c9eedabdfd2be61a56dd13fdccc8271ff752121275fa122f6adadf08914c8732d621db00c8cfecb9a52939fee6fac18a4826a831af48a89180b3fcf5b09681868745f50daa564bbf89a616ad666e2b42ffa861927363a4765c5ec00710c55189e3ed1e6b8645b4c6c1c147779951a6e56767e207fca9b790ad77fcb97b43aecc7b8b8479ce7dd4d9f3a91ab3caa0300c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.216470678Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d, Name: checkpoints/1920/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1920/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=3756c9d7b19999ee0c9fcff37872d8e8cc35c6bfc69f696c2ad6c1c19d1e3b7ad020a2b83188d306c438c6cc526782623c42002f349a63c847f76b2060d6fa05fd02312d7a0e7c2778c72bc32cc8227689eb6a317d2fb5dd291213de523544e612f4a619aa682107a050cff289eb6a753605079296883e4ddf3630d598a7b1dcf675644e4ef372ff3f52d6edbbcf21d04eb79621a0bbfa49ed6631c6088594cc9cd3ec42a8a88e9dafabe846c6b5a1a17ef393667699d75ebd01e07afb216a44e8c6ec7ea5dd963830e84085e2a5281ec72d87878e47f1953a9191dbe567cbb9d36906f28c247019dc976de49762e3d6cfa30f358c20aa5635d483b0a2796ac7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.219305308Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0, Name: checkpoints/4416/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4416/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=bd2016deb3a94093687901f2a194982764973f217b4e9e5e9154dbd25d6e08c88537f5f6859543466b351133ac5232b637e6c934b6dd38ac5516d69764309b4a34cdf7e2d66feaa84938be49cb7c2e6f585d4e4cd26e5e4e8d9eff0455d084a91a751e42ca46c0555662645732e0f4e54523c4241337a9cb492993efe13f0212fa273e7bedc2b5e91d0d328e635a4ee1a995374646d435155a1df175ac87bea50b4b5780e9ed55b229cab398e4611d41bdd02eec60c44316b38787d11e62f3f63ec4cb6afda5aac07f2fb92a284a13eda88a9ded71c2d3251163fce515a9dbdf6cecfde8efb90a1f3a24e5e0c3a57606a568339c497b69a82dfc83529570df1a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.219687376Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d, Name: checkpoints/4480/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4480/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=101c2a977e4cab32115733d432aef2813f36edf15af521afe81e85c80df39df928b93d4df12b68856d67fdafa89c34959df56fafa39f3b5e9bdfdb3b907fd749e3c5eabdbccc13fac1abee1a8fc0c0519606a1552e6216bb0e0ece07ce903ea69c4d53cd73b92b41e2e151089583757ff81496ef6458d2ae3dbf592c0639ad3782621d149f74cdd944613a479481aeaa65e3cf9bceecd20652824b2d08e9f9811070ad0b8a8397e7d4d93fdf13e84f771e7631ff06e7bfd4a82642eb1ec5c161609bb53e226e618bde51997d075710ca3db76d9ba000eb57b49e08ab43ab5104659c17cddd0b810a48dab1516c2aecacb022c56b9aa05fdd6650bdecf79477d2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.220709808Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/array_metadatas, Name: checkpoints/5632/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5632/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b2d6a4788ca74762dc45f6a240b05b9ff02a5c6ab35ff9ea6fbf3afda7a4d8503320763367b2418f40813ce9a600e49d6ea5a4ea9ede1f6238cc90e0dcdee5d791839ca6d7de74e8e51607e46ec49f4006f9779fde6af863c124cc2dea812fe9b158159c3b5debf03193be2444e6457e6b20d2913d15026949335438d8ca7485d3999148f758224a139ffd779148a5bac7a6087e4e112b3ef2700bbcb2a8f5eff38771e36e0540bc49c1261875fb3604becae6a81088bd6731b707bf221cd284765f36fd897728abadfdaa3d794eb76431245534da459df9682721586a192a4cf118c37851171116b98d4e162962a0b99074e7e4481c6c5249af19a5157c3089&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.218078151Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d, Name: checkpoints/2944/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2944/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=9d83ae34312b1920ded48cbafa1f750c38d6aebd72359ef6e004b8a296606e84df539e50b45510664769e5c540f98a07bcd60570b494bf479c1210eb05be1f8d8e6c69825aa5aacd69d5aa2c0e0ee3cf87796ba1907c4bab37d1680b16a5c81ef9ae6d0a0fbd20d9591a56c55dae4a34889818132f5a42438cbf0cb5f2e6d244e96cc04905995af80866d1783c13adfdd173e6360e6c7917bca4946f4e9c198e97cc13a3d785002229007152893d036fb4cd12a3c08d69db0342dbb90ceda564b61cd16b8105b1e5660ed47034d8c306ad457a7bc7029a1189f966e15736eda0c11671eec8a44ad4e9e9bfc8c2d05c0989485f0b56fe774db91aa0252a872e8b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.218039724Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/array_metadatas, Name: checkpoints/5312/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5312/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=1296cee3f31dee9c74a7fd29a36ede0665353c6aea8a51ce029a253d6c5bff59e811d612318b243485e07c0f6c1d093df07e4fae8331a009aa0e9f6ca5818dad9657aa4b8df0824ea0b77274aaedaf64044db60c9cdce19ae4e1d1e4672e388943eb3c3d0156d3b100f413675a03653e50ca9828b32142e90581012694fc31ecf5e6b1520b7aec09477c64eae9538724f77f3f990283f046f9ea5853d7aa953d21306039464afbdba46ac85e6b544199efa57233b23375991ab8bfc0d236965e95198c4f9bedeba96b083b7652bcec9482fba839f29c54ad5f7008fe267df428b43c1da3b5789ff6b278c9cd1e3a85dc9d239d8e5843d1062ba4446e3a8072fe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.219371937Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952, Name: checkpoints/5952, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5952?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b83ec3d6074e87ecb9d7d1b1a27ea112851774c3a01b40bda80ff247bc141a8f515051e0c5d6638dbb8dd55efa8d9c189056cd8b9f986c1053d23935f4164da6d5ee3211c8dc1a14347bba5bbd29ecb18af7e6e352cb67c058d3bc1d233280d8584775e9f8554868bfa40364271d13daf175bd722477ecbcf8d7b5c6e3174fd508db64dc8f9f4856a805fb38733f6bfdf84be1f22e91b9ed19047facf8d1bb892c9b2c7299d1b867039318e51a5cd01f9f97af12db01338923768b40d48cd5073f5bd648715faf6b00a216612dc209e14a58c754dfbedda4a2c0a4234a9238fd960b15485b8807fdfd9f4a5a0f7d9d0be07f995a6698a46d9620c6ca8d9dbbb3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.218021251Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496, Name: checkpoints/2496, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2496?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=ba7642bf0509e8658f5f3717117c6149b7c71236a8d86b5907d4d7dc34c73d6995e866a09f11416820fd7f881622f5fead9502b22f91a589a59298a03fa93dd4a4b4332804632f7479f3bfa62f6ecd7eb1cbc66c13b5a70d27b5a3bbb59f7950d5a3b13ba27aeb8057fb9cc62518ddcea5bd5ec211a490909fcd721420ae63aabf7449e28079caf378bdc93a6c66b2b34228db1e1ea7476ad51bc1be175595b220c017365d3d719e5a68d9822ba4ff0f431e442d0c3cafa88a21fde2fb0515f8fe94bd8c0e19d37e93428ac3b2df6ae643bd0635e647b834d85ce2bfa8b413818d316160b33b826ab6cf35a92ffb254737c35658c339393d8ca3fa37635ef297&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.21679357Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/d, Name: checkpoints/2496/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2496/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7783542f5a0c30966427454bfda8ce43d4009047bb3f230ae82cc2b1f91257e4e9ea390704761c6525d09a4225658b93d20a27e7fefcc7c83c445e5b43ca1bdd3108c7c5572342c82efe9e1903c6962b5a9cc81106f370c9e5dff919001537845421989c5ff33af49e39b0aed1d257ee4bade509b4fd72b7850fb54473ee630134aadb64342356a9c9a27747dea28ee2cae6967d3626c75c0415c2fefad662e7719df7074d500f581e905e7622179ae015e33acca317ba291615a9c94d6a311721e9eca29bd621633ce6bfe8762d83f4fc83e41b5d7d107b5a5d1a31c3698692a3231a5562d0f3670bc1ad7da0bfe8cb3ef15dbd66a487f718f6db75350270f0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.216868965Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/array_metadatas, Name: checkpoints/2880/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2880/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=6d34940cdf5bdfb943eb1f6de148d7dbe878adbaac6d81e34355082b57997f5479f78b31268db33d3b9fdfd7f6aeb645e720cf0fba9ee3aecabe85fe0e16d385b0a731a0af4a102d95543576a02baea13771c79d8c9e5d06bd64f6ea4e2b45f9963bc06edfe1882605937f3c8e4fbe8d7ed1df422dd82dbd4e15179f5c98f8af5e427ac31043019683f79233eac22632789198a0d12b64e0d9087b9a9ebf94fb88e33a538678cc7db472c7c7201ba671f6092db984d3e3739ca2487eeaab6b494b41024cd206a9eff0d9539dfa477e2b90103e47237195dee16e8553960f7fdee8fa95095505f94a5ae138aeba7681f0f2f306f231b05736707918312e9aadb5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.216955119Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560, Name: checkpoints/2560, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2560?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=18c24bf58d76a809d3a3f35c28438e6c5d1067c8eaf462b8891e5ffaae0d1fd3d1f9923ec3341ab647f18cbf2772ad1f51dabb784c2efde4e42ebcf88c536f1cc26fc8655a058eed1076a005f2aa35af50679e588787d6fa3f534320103dd2586f51152f68e3dbc212c53ef55d3010056f20b5928d520714c2b4eee5824d6368390f9b2fc16af95ebe26c3aee95baaf79d63e0389feadbc216d0fc70f7689f6b47a273ca31fa2dce20a5463ca68198ada6e5ae34f506a11a17be637fcf6bd2d67a0bb5ec203be3aabf864e17e2afa280bc6a7b3b1c81a7debf1f88f6dfed19f2e1d3533165fa2819534f8385556b1913e7798b6ac686c88956c5d3e9a1c12c43&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.217035826Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0, Name: checkpoints/5184/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5184/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=8bee1c80d8337261903f323d6adf3efafcd45bada53389f15284d6e89ad4b7916dc02e0d0526855773b203f0df64963f83e937a1a7e084cb2544920b6431fb987770c8e197ef0664e45026008fc10427ea4fd269c3d87aecd5dce1c63967e444ae6f8c45f596781f9c8a51a84f10b04d51240cec07317848be6b6ae40fdf1191586e2db878468314e27d30a46c3e62f48591f35ba50256d684f78a94769914ebf2be49c7b0bf2e979c449de2b392482ee9873a587cdbda8ed3d725136f0ab720d7e54f6d1a653d29d2d026f06b29ba097624d84aaa1f7f28b1f1730dfa37d3306a4b83db8c752dc5e1fa887e071e62e97fe2f522bf75560c02a85b03efddee21&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.217124868Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default, Name: checkpoints/5888/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5888/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=358b70cc1a07ae282c2589e59751a51fd1fb0b7f479812edf5eb800fbfbacdca971e634a31d66680e3e806a634e93764e245fb19860cc678504e5d91b6dc5d1bf18e30823c13f22939bfd5d2e2723bbf5c9ab743095b43af376791b8232766b02a026063640df3da4b012bba871f0a12bc0b2cec0f438c05f60e95fc02bd1bef74e44f89e86a25bbcd1e695e218dcc7d8ef1b18c96c1ea6eefc8e0114664ae8e27252279a71a9bde7b232d39e4a686c3646ae39b79cfd001fc5299d2ca1bee6e033a120b169f6cd8dd9e76ac7fc49430a9face8af617e9829a3be5ee4fed626e7f49811f585c90da46202fb565f959bc3c58eabacebffa65cfe0efa2de6c04ed&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.21720623Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d, Name: checkpoints/2304/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2304/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=22da9e769dc5e03aed618c5f1caccfb15f039722ee09c82a20e70e22a889f2bc7812179c7a43cba68ada70e67881437467f57ca36d98d169b952f8d89d9855cff9bfa02ac55fa214a6fb1158cdfbc7e2ff6d8ee61447abacf67268794e0d710c9c060bb50175b5f183b7d3e31a480982b8229498dee0bcabd572da31e8a90fd1a5c9aee2222f95e6fb5916043232ffb446a07d47fa291c4886778657c52ec9e6a31bff37b4db5b8696e93f6f567455f8a1ac1eb2cb707a7af052e5834919cfe50479f17bf86be1712ad54377d7bdaff86f3b26f84c37defa25e9a7088e24eb00b83a39b2c05002226edb2d0fea4cec2d575bee1cf166f855a7611305b3b03091&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.217281133Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default, Name: checkpoints/3136/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3136/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=8ff5557ca08629a30e9ea5f8eb6e8968c0191f9a6dfe4877b3e2199a4a28e6240b8b8f3e97645083e295aa2d7b7fdc9d87e27686bf07707471132a0dd33858b0ad4367ba3479aa1dcdfc26967924f607d58680dc660807025135da256a428c9faa55ad2b851c803eafdce7f94e2853a7bcf247c3dfb082d2d84364a47a95959893f15dc4e045f72918ae3c6e000be7eca4c3584c80de480e2d21b4dd42afd0333db3c800c8574ec0d28101c496e0a4658a5c6a61ef34e842f0744312c855ffbdb28c7fbc0086be35fa8cc9fa95c6580bd0aea5469a18ee71d7a393e9a0a86d87da65c37df98ac477a6599947ffd6deaa8b2973e49b5e8558825c5efdea3c7dd9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.217372597Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default, Name: checkpoints/4800/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4800/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=3031c06f00b0fa58ad989f6cbb09c89c8c99b4e158a67335445a191aaa6da2273965abd5a2492c36c52c57fe6e1cb78ef37897ec045c1d620f6a1557625be1f70c042656fd1748409548339a3d0f385790cad4a15871ad2cffa3fa03f4b5d255fa2de0a34fdd4ac2f51d7d749e16a5a30bfa546ab6c59e453551ed9b64b5c14b0ad922ce4a0cd884536e410a7e9ee21989df397f55cde5cc31f798b138a2509b0f309d8427eb61afc11c6f5901b5dc7d43215fa59587541be40f0f30aa2dc5847f39185fc471ff30c3e91f92a7abdf66f34adb1c03a8c6cdc41cfd5a413ac07a065503ededf10334b001872d98149ef646bbf05a8cf1b7d2a36d520e797936bb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.159741325Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/d, Name: checkpoints/2880/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2880/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=bae1eb3aa0cd5f99870c955f8e473d5b54b359086a8bb6cbbd325bd9cfb748fbf0f39ce1cbd192dc72e8894eae44238b27389ffea30919670ab9fe16bea5b991038054648fc49178779cd964102a6e998394d77d990bc228f371ce47296bb8476f24fe788a737776a05c570ae459cf9137d31a54b60504c87a4dff77d1b48e1ca5ddef45ca3861cb2f6f834b89aaf183dbdfbe4110114ce42ff6ca593d76e7b90a99b60e5a47e01b927ae31debc98e731deeb98e18a01f681b22f1154f9cb0344ade836fea88f428387dcf8a61f2c57f9e73c45a60f42ccd47de8191f9b5919929ae2bd629aaedeea1b68300adaf13f867bc63e28e080b41dd2aad600d936795&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.160712624Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/d, Name: checkpoints/3520/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3520/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=820cf1c3c3733ff316abce221d40a59f865bc85c4e327ae42f66cdb4e0f4a97e57bf6def186bd88ba47c772fee865319bea7a5652ac92b68fc49c89da39ab0d322daa829b6ed8910d2628c49c50ee6542756613dbe58f434f8324c4990abea8f0a67f2fc44d275a4a5a1fda92c5b198b0e3f805ae15708bc9d7766183ce137c596887319b8a9feed5b36c0db4f2974839c3549ab217d5482eb80a87207937c91677b0ee77647a4819bd09a84e0b6ef30f3f0c8b2bdf9f7dd8f91a475168da56cd9eb89c23c8c02ff76b1de61589253d82ff519d326745aa56a367b6d7bcabd9ac6c2f5ae5299b1fb352032427b54167a8892e994b8129f9e84b1f7f7aa98b597&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.216608024Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/d, Name: checkpoints/2048/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2048/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=420db78d67c885ac64a027a798b409127d590c59682a4cdd44a452c3236226d7b522e469debff33d81f53f9a20413293f2198f62d8f7b7f72a77d65cc56dd3bf28330961c65eba3572aa951e035b2c55ea56a07d1a283ddcd8490f7c8105e91f125a015109759a30b8f12020e9583f2a5eea32b2e3c280c6e51dd8fdb586498ea8e9006f7c45ce10980aca3306186bc55d4df6422ff6ff072e2cf7c9203164fe184d42fdf00b5a94cb0bcf4beecf49e04990141d81780ce16a09b66e7cdbf3225bee6e901f7175bf4a2842d839f6820b9d1c8e177447d6f60e637b0f9460d42209d04d6f89143abf959c916ef9e0350d7a810c40785c733075f625014825298d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.216715974Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0, Name: checkpoints/3648/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3648/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=89de3ff402e4ac79eb5c73b90793bfcfdec26e486d0621c3ff2d6699ccff468f4dfce90148486988adce78a388296be71df4fca6d2fd404389583406b862418e9de5c39be1ed0bbb3d6532f1192446e6c9d21fd113377259da4c6c870f3875eb91469eef12d031dec7707ab617db110e1f862904ed7c7c1390b1d9f8eedb9de91d0646492c0add6717f13b5be6232591436bae1e3954724f8097b8822028c5b34438c06a034bacbc887b06cd78789aa529791da5c9b9b9da3cc7ce3a5ccf908e511bc9d0ef32733b229f9a3214cd6acbd2f8063bcc5d0db5bcdecc11ed384ad43f78084f693a2028c3e25ead0495146a2406a5fc8d844ab7de6282c65cc014ad&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.219024289Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default, Name: checkpoints/832/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/832/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a1d126b4987089898b1e4b46afda5466728cd2b4cd39c1a4eedd48eb4a8937de5fd3de8e349d48b14566368c6683dd823143d54bc77af52b69e85c05d8f84b22f042bcf1c23e15fe7363097359a31d58b0f7ebc79e2470a183bea6fd9a5410cf63d292fd3a2d6363ae1f2e48ae1d2d64f8bc7e4e21106e4f6c25ad03fc1fc94b07de9e08ae58e9a2b57cf56b071c08e48b9beda4d5f1454d5a179087292763c26477211c0bdec2ff2f62ec190e7807663ddf21ba5c7b047fc1196c7938caa07e2823004d5864669a2a3f9a6bcf72dcf5972a8846af20823350870745aa17b7d941a8eaa8a46fa59bc6576510fa9d9951a21ea6a48c6d1489627fdeed57172479&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.219002117Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d, Name: checkpoints/4864/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4864/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=67c7e15c3af53f0bf0683265484cff3924d1bf4733387d6252464bfac475c79b0157450346e1c54b8295bbc139eea0d7ce324ff543c9cd5e78c5a94e0b322eefd7e97adfb145f28604e8b733565846a07abef8f70e2d90161fe3187046b36eda98b9646d93b82c933b4cb6e56cdd6446b0f8d6229893e3bcfb5faed8fd53c6bb9cf9062088987e28008a6f2aca4c9b9792c9e73950e6836c1fff237506aaf7fcaac9e659a4971ebc87f22bc339005cec594ee325c780d414161a3764f28c0861d25d23706f2b0ea0ac11b918d3798bfa0ee5b52c43d85ce6873c55904276f78447af297f1b57b9d8194d4fcf4808ffbbcc4edece3ab713e15e27af02eee3b15b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.226949024Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/d, Name: checkpoints/5312/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5312/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=0831069a9fbe2e461e8d9fba1227c195e6f496300e643eee469e4508d2fcb9490df407f18a80f66eb808a1c36fb871de2d824026676d685997cc250cec019a41cd2dfe11d6650367f9e589e4abf6e38df609e40ac03bc8c7fe8179576d5fe11fd132b6d023fab713a2e9c9c749bc842cae3e4cdd1de499c9070a983e5b1870e3b372c5cc047d9a107bf607bec60e14bdd4e0c03dc16e58acdbb8295e0d58b377ca26e11a8ed007a21a69ff4ae22f4c6c458cfd3edd5f4690edceb2b18febe5e84213b68d0eebb311a92a3053c9fc0ff7667810ffdac4f7a565504341540efa60fae39ca94d68d4a2be1278de16cd16ded7ee21361fdf9ce36302ff4e8e330574&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.219048684Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d, Name: checkpoints/832/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/832/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=41822e00a538f59d46582786caf7c97d078510acd5e3183c27492e29df5dc1de282124d0cf71249d2f75678ef77fb9af74069fe2e79aae8d9fc9a54c84305a76fa1afce1713409f89c7e403390f5085dad753d005c97f330c99432758fd7a4dd851215112bffc0e5dda3d7a11bbc08a349dba0a3f886af2c68ea5fb8291de066450051607ce5fa7eb7dff4d72844997b93e37250cec160260bf92aca8afe8caa74a63334b35cda13351b55e0b90374cb44ee7db54ccbe9bbb5a8659a35dc09bba5e624b17dcd468742f549d4d2897b99a03a6c60d54f2c1f1f13629ab453f381bcb8574e4f87ccb53c621fa8bf392594328cc9f487213da3dd658ba74f3bad7b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.227081308Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0, Name: checkpoints/1856/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1856/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=96679033c01c5ca3468b39148206f5cff61138dd2b0b3935e78f857b1101092cf80c432dd68984ad1c123cfb95b292dea183789defdc87c609d6da7dd8266ab8cd1aac76737247f0d517d3fdb326a86ab114b75b9fdd2292bea7e87b4796710bf48fb11c41b59c6c764c8e44db1d41b18293a10a9b6358627f18e6b22b130ecc7df81cf4a4e6ca5b44d339e59ad9d85753b51e5f485cf02a142c6542794687d1abe0e8993df95f2934b5c8499eae3ab62cac2dcd8eb7f93d2745a4a092425751bfbbf3c4055001021dfc911fa98efbd47448584b33fbc35a96400eb473f5456b7490c39ee3ed5a4712c151c4532c00af6302594300742c5a26afbd0218f75766&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.227102763Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default, Name: checkpoints/2880/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2880/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b6f83ba9b651926e2432cb7ad3388e100d4c7b79712ae88e8b3ea1fd04a884c1dd04500e5e200200ee50f05843adc39f1357b857e419b7b550ea78e595e5cfe5bd9667dea9410d3e8c2a9e628ac8b06778a7b4c003d02e339b02af2f5a7750d6c1f04cbbe9732a91be2f387a2461f6347951c09d255c5c968a9002d8031542fa7ae9aeb22eb7ce6362c9d95e19589e3c66eeb6af6d5e8978c5cd58b356ffed71e6d0687f05ae936d235d100104e1bfc88f942afa0b72b0dc5a50fc3d0d61657bb2c9e5482780211e449a568dbaddea6d821688d8bdc527122d96acc56156576d903c25aabfa28efe56540300844dc055b678b5f5c39a9f8ae9d9d158b2eabd68&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.227177013Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default, Name: checkpoints/4928/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4928/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=abe831686a82f3f5a90a048626882f97c6f6abb6961d2837ca828e7329415af536a0597d9f5ef10c5d19a70d7b6ccd82c646893491e28dabe100c2e2ebec64a10bbfbc27e1da38009ca2832591430e51ef09b61e82b3919f7394f5540c890b5d08697c67f2bc72f99014ea5074e5a3e87c78ee8e536d43a9a5f355431232626c9b72c94685147209594cfb53d5e83e0c63c749285a1dd540bc79858f120f66fc83ae10eaee65ba206bee5354500abe9ba7171dd80371c2a7fa50c3870e4585695f41efddab0586c28d01e5ebc6336857728fd2764cf36f613bcce6241a8f27428ce82de32098e6f58b2dcd72c534e4edd4fe705a67eab677746c1f0109abd818&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.227192985Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default, Name: checkpoints/2048/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2048/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=8ece2a25d120c245321cfab995bc9af8d302565d115865347e906c7b44ff3350e1fab2ef5cbd271e7515310173fc53b3c7ec743629c5cbf8e2df81d0bc97830eacf03b30831d68f426907764574eaaae3d54af350116cccc7675a9be78e2ca4604c09333e2c009855123b11570696dab5c9dd541a03eec958af9d7b0b82c43b72f770e579368d39d75fa492c71a147a733b0558adb6ba4e4530fcd38b020bcc545fda1c2e4650e990a8899100a010a8b1ccb7763e40eb5ae670f94bfe85c1a5702333e8bb136f1d4e9f680fbef75bb5729d61867e79a95c674e6f2a4ffacb70434af9dc20d99e600e911eeaa3dc855002b4816e111ddc7b88149dbe97c9acbe7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.22711678Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584, Name: checkpoints/3584, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3584?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=675d29a2e078efc5214666c2385834ade1ca371dad17c0857ca692312aa671c391418aec6020c436f1b500105eca4af3bc286c853b88c0feed6ab2c3c70f2f1839ea463f5954b22fe93cf61b5a77cd43e1059aa70d9a05c471052493665703e1325cf871ab02bf335d0610c52f1ea8ceac54e107a2e8fd8a51d063cbd34a292563e4b67e1860af518fc99f3e436ce7cee9a4cd89b082599ceb41f36b9a7a70237c688f3f0375ebfc5aeccd08f17da752889e0bfc2cc335c6804e827fb029d03ab904a8f600e291d8424354ba9aa065dbfa22052f40002a6aec27e4e611fe5355031ea4fe6af6b6564553a4715cc640b9bf4e65ee2acbf8576ebb3f58fedf0ffc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.227251623Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/d, Name: checkpoints/4992/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4992/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=01f9abbb83e5c06bf677d26f9828abafcc7d2a9f393e27391e5982214d29372987cbb43b6b9674ea29210f978849938d965a0bf3d8e24ea4cc956612409aabef989a877483b25fec799e6a3d15815dcca64d7b39d1fd9c618bd73becb1eeb1e6121f1f774525bc582dcff8c54b68b30dcc92eca6a4e4bd9efd97a26e09008fa73bfc3e0a8e1fc86319f0af75b6d82b8b2ac60764c23beabc8176f6b572e5acf911aa4dfa20f0a0f2f570807c0468a7d0b6ae9ba541f15944cd2505670afc10ee583adc988e71e16c3f83d7d4bce99d7dabddd24bbf76964961ab602ccef06698ffc85503901cfe6239e0b6da3498177f03d32af968dc2a384d07d2158c2e62fa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.22726795Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/array_metadatas, Name: checkpoints/3136/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3136/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=43c91171ed93f47f1d37ee069747c95a234b58cc9953b8c2e3859671d37b82fe75f9592ac6ed2d1d230fd5563cfd8df824b41074c69aa433f2e8e58befdf1e73c1d48063ae9add0d43d300f5bc60273c38ba177f0bbba30706557e9895b0b24d3582c769cc384e4fd566fe60da49493e059af7a2c5ab76dd75fc0cfe5da12b0455e401868873cb51e5bbd3958ef32ba19b99df8225dae712b7095ff92b0a55ddc87b3ab7dbd19ae16feb496f279d2e5a6f4cfc7152b24de8f666049f05982ca28a0ac1de17f8f309c1dad5aa05a06ca0b0d227b6c26476aef90ec1741ad6a030bbd811a2c7f002c4f577c5e1c878997604a97e824131e4e540868dbc6cd4ef72&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.219197348Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/array_metadatas, Name: checkpoints/576/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/576/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b4af21f3c31df7adb2be02aa91c16e342fc1a38149a3d17a00c9d537b3fe4a994576a1375c702f19da0aaa25dd5410d0ab913bf90bfaba818ae0ee92241cb48695aa27e01427ab9353f400b3745f2e650e50b5902200375ce6d245d28c7a85b926b3bb0e7f985bf2f3d034afb729b8ee7e5f2b46ec07c4e06de16d8dc7a7c792c0bbf924d5bd16f393bcd40aa94acf424b09619f8db281f9c8662ea6db32fede3dd0820f39f851844ffc1613a2c20870791cef6cee6d4a0d0e4887ca430eca7d18c2160f0c646ec4f3933557a7454610473c2d263591077731e409a052342a1225a3fb2b8cd394a6dd913f3c1d88e07d8ec8e9e36f12aefe531e05363c68c5de&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.219212463Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/d, Name: checkpoints/2816/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2816/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=40ad89d46c5e0f8284a9ccdc5be7ba9e532935b5f89fa5f73a1f5875a5fb81c705289f0636d3bba513458a0de6e1bb5f808dbaf6e8a0536225c73e1c268fba3b1a332610352b37fd19b9703c7bc007ab42c20d0db34dfa9434cc3b037c6e43053710caf029a93230bac2ddca1a389b730b16caf0ac4b47d12440366eeafc0ba373c121c0444c0669d9a04ff91db0be7588f17a2afb0c5bcf6884552554f255cbe3c4a436e01331eb5db13b555eb4882f6d8ed7f4f5b7a73c5c2e426f76f3c452de76ca27d9a0ee956be185ebd7839c688ccdb57aa030440fb59405adb72141c88439cb450d75d4b32dced7ad93ceb56f45b46d2a560f7e21e62bbba37f853d31&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.227490931Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448, Name: checkpoints/448, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/448?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=6102b33014a5aa486000b208045899532810bc979bbb14a113c9ecc8bce028cc44bfd73ae162199e24bf9b5e30b71c7c3721fcacc9abdf240d7005eeebe9d1db89069651edc36084e7faf293dffa43567256bcb02e63106d46a9568eac442b0af4b4e32bf820a413f10b7b15b1429f4097502e375209baf858ef85a4018e2b67de24ab5968482182c5b1a5b1fb7fd2f812787a69fcf7b39832dd17fff800efa440189dc4425ecda8a391a44055246b2802b20863497536fc572b1d0535cdfd1f6afe5519a72cb95766ba16e9c3902dc643b95ff6809d19644b93a63c364894b5e848b809c9aebc5aba4b0f6449ffc5ff139cc608bd69b2a8f73f1ff0ba91def4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.219222886Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/d, Name: checkpoints/5760/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5760/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=8571eb06aa9251a40f02a4f82850afff158d22140c037295860efd1d17f1e60ad3979357d8fd8863b0350d16d9b470bfac92f14611b37b99ddf11e12787c81132a544304a2e24ef986bca29fcfb1de7d52a68b4650b225437eb75bf15c1fc8da3d09c49f163a83e04ccfa8b1174345cc043f832fab92b5e6f514792fc6212a1a10f869981dd8ef77984653bef2e8e212452ee71bd720e935ab7066035c751d1959e3d0c278223a366720307c0264bfd48a1ad2b516dd0a4880471d17226a6038ede97842ee3b035db8be3f78844a709238a48bd109ea5b2f148d2f6cd8a2d84583e91799d7060a08d5e2f5fdd6306b95fb4a2f817fcf9640d2c16ee3890bd45f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.22754091Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/array_metadatas, Name: checkpoints/2496/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2496/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=9d42553b7ffbd50455bb8819f7f64c16e57f35849d9d40620e069086171a1e690dd3f052f82f1c665b5a2a86af4585ff9e1254ba0b355f6ea25a3696c2b0db82d51fbf6e762a5bb59f4567a826484605db92be32db4aad3abebb89f34c609b846b7749ba503870005670780acd50bc73d5f363bea1037ba077cd8666e448a4d35578c45ed8d0b05c44f6049328b904b503be57b088412b1113917d8ce764e4df339c14baa80dda6e03c97312240643f88897a14771e956496e44eb225f331c4d72bcb87dac13d51590a484b025f3e16c6a8ab96e2e5ed09351f063cedaf6a74863315e019358868a3c42a6e8ada8b86f5f049a1616e2f7f697b5162f0fce80f8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.219229584Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/d, Name: checkpoints/3328/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3328/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=1bc46b91dfca23c247c36593aae82c8c3d6be7086864db5446cbe4f62128252a70038d7733ccdce1604b232ed7fed111326bbe634b30b108eceac75811e2e775ee26195077aab19d45846d72017a3bf215c7f034cb4957543f0815d623a3f4be4483bacfd4398a91771e432aaad5cd3ed9a764a3a690deb0f4cfe5425a62cae944f7a23538e235f83efba228d9aa8804641c0e79f9f679ef1dd725b9ac2d21e02e11b472f75b71e5186276750a777fdac10e06056422006464ce5b7f15fcfd47d01ef1da58e61d4d22e9fb02931f23716182a6f0b1c919cfbbfa500a2b927855b4f11957225da6e68bf6838e06d0e514941a7692418442f057cb637955b9853f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.227643229Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/array_metadatas, Name: checkpoints/3648/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3648/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=948c5ed54fe509c5b10383b7bd9fc168cc7e7c53f2838a69f4bd49dd4101c5d3be460c983f3f57e42e4a574cc9fdfc9a13d841562c1cd152ba880ba460fb8d14ccba1d17d6f2b2661b12e58a2f5528347feb97c14f9edfb37b9a7b4976886b2766955db16d3b1585764fa997b6f37569519a31814a632ceb41c233bf0e2d30cccc566096a0e723848714835669878492d7666d7505fe0a229dde732f1ea1de00cc6b8adfcc7ea2d583c5b827fedbabbab7ac282651af4c433aa3d9934995e2949693cce38e702abead599a3a8c7c61f630d5adc01df80dddd52ba87847fa1de2c58b24ba3bbaa2d747e70591fabd56b5995c60f31e82c9b18adb3e284bdb2a90&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.227676962Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752, Name: checkpoints/2752, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2752?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5d7d1f014c8c1505cff85f46b3de3bfd8cdfbb731f6c4853f8b1c0878097574c244060fd1fe86fd15d9f80479e828ad83eefc749a863bfd24b7254d6d150d695a21b72b9f06dd854e6705d2bb6dbf518643cd1380e586830f71933eccd198dd795dca29184fc025a2e3b393117ded46f7c7a61da06e91d979d4c3f47954fec61e3344695f65d69c246b274bf6df02a052ddf8f3f7610b57dd0a1cf3acc49d5f7fe1e76c3d99801096b93be986740f393760f5491e6b861ab42782f40789effd586ea8211df57021daca2dcd95e6bd29ab772ebae66aeb4f06f37fe095c3e59a97708ed936bb22309596a3e8bb8fd9fcff19970b63885a4cc5dffb5521d3f6c71&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.218606456Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/array_metadatas, Name: checkpoints/1984/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1984/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=80336968b299152bc87073f86cdc2b420cf292fb6bc1f2912cb21f5744366c49f586221e1e21d4696820ccba7b27227b8603ff12b090ce15ae3d81197c374156c23d2b364cde6d03f3e6e677bc3f73bbb5702b46d6b064ded776581758779464ee2c39bcd52503a81253a8a964536a09424a5946369c706c9490a6063ba8a5a1c9239be82e0ba9b5aea3c8930d9a0534f9d15c636d2c1f1311cb965e95a1f30da25bf5721da3700510f73ab1712e79e79e2a93a27633af9c862dc3cc8f1ac49e5c74277ce9124a32103ecc5df879d6fd397f5321247da26ce89e8b91aa650397b396bfe8870e8691d229a2abee158bf62cf2d3ee78564fbbd4dc478140d94c1a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.22768176Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d, Name: checkpoints/4032/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4032/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b4fa72113de8d3cf0046cd24065b74d6949a26a8cc3d9436585c66a2b8eee172e21bc41c87d1be1f2627976cd42ab04c95068f7ad9481e0fd0bba37c564b5de1f61b7f0be7f1e7fb5bc553a11bfa7c992583540c1d1fb23627cd416a35cbc553c7f8ebe9bd6cdb0f1501a783e14f7921f276be5ff85c142262109f56da57a76bec36fbfd83370ba4e584bf7754c0d3f80a1d12133f1e91a98fe94674ff23ba363e384b9cdf3299e1bafc1f99f841742e280a95d81694dabd49a4cd5e4dbd3220f6edb254a901b50b4282849d0cc72e3e0205c3e8e13b90cf4dcb0542e2495eee67e769499e0f4868a2105f533d3e2fc9dc54d161a0f692107c25e73f0b2931d4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.219299208Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d, Name: checkpoints/1408/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1408/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=683934af85bbf1761cefa2552efe19dea9fa9abf96003a0e07bd5b3af394bf3fdc2415891f2d53f5a6fe1a51a4d8f13785055c20dbc800a7550112e32ab05f8f0f10527d8b443ebc48125cf84f33bf82e01a97223a5d3dcb7fd4a61869d4cca99e815ea98be33abc6a562e82594ae6cc6575443d6d8ebb72cb6e957a0dd0ed63030ca9777ff80b951e626c19caae5f8eb8b88557213baec732a00e152a9044d110650fbd9273f9ef7e7da898900ce82b7c38cc88033ef44ced8208dcfcde70a962a720d05e3a8ca6e94c1ab70b8c1d736c39c496b75126bab5310dc6aaec509a2e964045e4e1f5e956d217e183fc8690008870733612101314e39d3e92b3a531&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.227826839Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/d, Name: checkpoints/4672/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4672/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=74a1095e9a5a35b064cb026a0c045fc9f0aad2e1bbced7d2f4e48ae671a15f1990542f6e60d387d0dcb4ecaef6011096bb2070a1e63495f1fc47dc3c8162a2b0257532a6b2806d83a28c280dc620e8a1cf5dfd807d673dc5ed57b8f9b8b930a800a859c4ae821a6207925989df240641634f9aac58ea814f7a0734b71527876c61e7208e0197bb6be07da9ab844e8912e778561e6a9761675b791641d97f6879f3ebdf8512c3d77a20be8b5278f25f034560d30ff5bd7672bc93f021de76527560affd17c33e46fdae679db795cbdd9259f75abeaa037f3cf142ccf0f3fdd56d19edbed9cb68e40e0536c7b2f2e91d78e1d5b4bc6619703d376edcd463af80de&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.227886345Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d, Name: checkpoints/4608/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4608/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b68bad24366754cd9fd2ffca9e2398311dbbf70527483eeb254e9d12e5546320a6303cabe6dfb98487903ad34e1a771d8f4032e2a4f8d71a09087cbb6d2269538af91bc34dc11d00041c21fd59255f755656ff53f4f228de63f407de7b323542924806e6e3599fae0190f6bbca1ce4bc63b9febcd19fd25a981d55b5b822956f8b291b6ff7e4ae6a4d5aed695e57cb853c79619a9b2b9f655621cfd4d7a64689a3e5e3460f154eef9c72ff8f0ca8eaafff6957e9e55ae482a8e8ef56b163b0a83e77b28ad9e62beb0019f53e201100eddd0af6d2d6d53d25ef70f63d2ddf00887fec59803389042ade67d6d76b76dab4636ee80bf84a1905ed983b713cc2ef86&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.227009904Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d, Name: checkpoints/2496/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2496/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=a41c0b9a6d07d7d42d4df5a0ec8a41187395d3e33d9ca5580f9a838a4b3e3c24e71404f01455fe831e66f081dbc283ee2ab2c3cc1bede1000e81ab2877c986aa09917fcb3336b27c288b39334183e312a1c49a8dfe128e88b61f3c2230623a8430b9b23c4a4cb712e6c2ad90e0642e8c6922bfa1890341c501d9b8cbc3215e1d775c34df01652b768ec036289c104553a0a6cd859e0009da1ed41e414f35422593271b53d520abcf593ce9f1aa0448a2eed13e70615d97adac6a72f4d15b15c3d672750473d2fb0c9a53654e5a49502aa5b807c8c2d5655d3619894e52589cae91f20d1989e7e24de79b690f8642a04ebe0177b91f83efa42559f51e5ce57ef2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.227926824Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0, Name: checkpoints/2880/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2880/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=18c0042038eb81ed2024d57e83dd2aaaf94bc6a2e1767b59d174a6365e737227d10c84b202eed2683f84a4145dd3b0cb2e115fa863408b0711b685ca1d5cce597700390c2cc0e96c0bf8371963385b64d531692e137bb5f13d771cfee961be1581fe6d9d0cc975f44f53a98ce34f569d8f88d6dc8a0103d87d15d27363bdca5c6bcc258acc35f426532027d660b8bc6322c22d08027e15c565b84e8183de5723ce9a9c521654a612e00a3eae39202a02e54e5b3dd437c590de8fa1ed5d44eebb3920292385ce50aa8238225627d0f7ef1f29f5ea3ef329a44d5528b7c413bf8d9d1bcbd0eb515376938e8a1ec4b14d3c98b19ad2efd0aeb82e6460ddc40fcab8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.22793819Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/d, Name: checkpoints/5632/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5632/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=968c27f49162a7a800ce560964aa89524d15d8ac75dcf55651adb8f5577f05d04a25d6bf262e7863f2d7902a2eb675fbfac69e57bd491fd65dadc3177d002f0d3dd86d6bc1f9e229e0a75c2da3b7e6bdf588d8af66a4d89ee106afe7db1af610aeec69b3e11f658d7c5a47a628150c72c644fc4416fcdfc65e3bd3ddef0488f32ebc6fdb525514826ec06d77e44c9a78b7318b8a5f9c62c026966a7118453f3ff3c41cdca51661f9eac67b7683ff367e1d90a94458712bff0f3fb7687adc76098701d78d21d58209cbdbeec9a30ed7dbd7746c8052caf0b590acf0302fbde71b71c9f08f4932bc9f4784e444b6440ce7d9274b0c9bda497f2a43c6be6323e62e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.227967982Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472, Name: checkpoints/1472, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1472?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=264ec6b2c016e3072f4b29618a9618d6b69c5232b66e71d7c79862a8e95be7faccbdb5020eb0d0f1052596c66b712de35b5f6a0a2519cd14bf57de02fe73bc2c2cf7fdebd871bf65496ca400a60d43a84dae462ff6229777e0f84ea5dca3b57410e01798b870593fd85cf766f074f9f315196b14c249f4861a084cbb455305a8e189a2792abce971935ccc39f6c9fb0c0562b3f3e6ec42b56a5148b491a7f5d279d8d834a8caaf20d91873b19ca9e2c57e099088b12b1cc2fbb82bf1ec2046cd5017b71daa638eb52eaebb37b392e9c42e7ad41c9d3cd9263646c155657da9f21dfebfac57127887c7359cc2b74302098fbfe77453f8e0711f077459dd3b300c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.227035071Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760, Name: checkpoints/5760, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5760?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=236142ecf6b47aaef35eeadb5cdb2ee04483b8b548f8547a2d63654a271c5d9f901b17bda05fc2250023fbb367e7dff8c9d1311b2167d9df40dd409a492a9f9cd3e49c8f57ec867ff3d4836f275b162645b99091c90d6a36fab7e8495673fd8f97b75c087efc0e32e4394c771dbbdf169e7a325e3c2b25944e34429e6f14f2dc5714ec1f747021edfde175ff82e29d5734478ef6e809117be50a6249df72d97df561db421ce918ab099b87449bf12217de84b18d64d01273767bf4706307cfa9e98b0ba986803e36fad26a2287c698f9b4a086e478d881c30ee5b44c03071e60220d5b7b6c7644e1755625091fd158d0a0706558bf587048d7581990656b5a5d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.227843108Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/d, Name: checkpoints/2112/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2112/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=af296eedd775f806ff86a4b19936bbebc56657ab738ead185b64daac4b222b6a40e565ceab498c01c81fdd06c0c13f81d11070ce8d0266abc7c2c2e3921d1a0ab2330af96e77d9386a66d2f8eba4ba5485ecd7b9d53499bd03dd9916adfb17058f1f1de81bf102be6e79974f23a3c490749a04841ccc666697a250792d81227e3f07eefef1e52af9cc8055d1749c3ab2bcc76d5785d3a8792c3457a9f84cc4b5a6859b32c91320767811be74e0b98efe14d70d58ea7a43d861b97c07dd10d879d894d41c9f0815a9ce49f61db77bf5101c76c223a8842b5402e9636b37ded7a3eebfc97fd1bd7a69eb0eb0a5e422ea6fb7aab7226a7cd9358bf99550ddc1c3c8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.227983536Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default, Name: checkpoints/320/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/320/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b6997c264552bcb59345a64d2e75a57ff57c4730b2e0bda0ada0b221aad089f8122785ded1a1926ecbbe20ca95f4d9abead0a08fd6dcf18c604155ba39fa965e7169ee388fee19b3407002e6f254a8e11a29fe0cd545ef6d7df1755073bf4d2f130c1ce8c593242585d3419c33e0cd7fe25039aa139f952bb976879f6ab71e7960545f602cdd85e756be5ac784123ab345c262099482d270f659dbd85d26ce610792196672080b44d404ed85f40e368c61ce85b7b4a1921cea3d906766ef0af0a1785a8b77e35a35dc0f04c1417d33742dfce339c09367c5a2f6aababfd85de49e8ea16c6ba6ccc2082e315e7ae9827755b5911f39eee97d3e3e0acbb3fdeeab&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.227112683Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/d, Name: checkpoints/3136/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3136/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=25967162d590151256261b8e62d04e1d38ca4d26222187b38d7a4a87d4e890b57f0264f06b1791df305af9f23ca1422b1bbfe5a6d0bdc799721c9aca8c06a12dd42a47ab7e939fcf33679a8b1b53a8287c33c0cb9441cb4f6e232f933003e24d860c65efb70bffc43f5c442d135a48da4b202d2b8fcbab51e233553ca8bc6ba27f810834a67da3f52b0fca569630d3788f45ee0155e6c6a520944409709cf5c6aad4d4025be65b25a5e459b369bc73b2d51bb9f07f81fc124cb9e035e9560bf3bf4c437fa28d9a3c3a6b347257145a010743dea96b2ed6a4ba628d7cb997b5c408e2e8ae6dfe7a59946e6148fcc26628e2215fa9839b90bb5450f6247923c3c5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.22752729Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d, Name: checkpoints/1728/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1728/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=603844afd14a4db56a694be177f48f99cd80520193e4992c9a3088a7ce0a4d9330b65da8a61aaaef2451afd2e2a866932fc4d0718a481b7915e1546f728f3076afef4d4bbb1fafcd91f0aa151f372adc43e2091f7c8cfaf82797c80758daefd7e36617b840fd4902f0190b34c072125f7c855ef09ed6092c4caf5764a787a70c0a39e060d0e35d7b7c2a3ca31f59f7243329512ede95b5f9e721f1f99b483d69b039b8138972b3723abe61525cc93e5d4798550bc51574a9070ebf77d596f38381be2d152544a491c01c7ed45da985504f93fe3c3e48e88401e937765bf6d1b73f40640296a854ed96204b99a2fbd8b8237b43b44dd42f0f2b8f46ac956c0f29&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.219288562Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080, Name: checkpoints/6080, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/6080?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=9a29a4da8d51104cb9a65381961f25518042382e9d72caa1573278c9a49e8cd6c2a0f81095c96097527bdd897491c80336986b2292984536ff568e36bfa57f477329d9ebd34683838e579aba92c13dcd73229d39bf072eba5f67d86440b1eabffbbb52440be6c9f178e434460ddd448256cfb3dd728411841ba7405ae7d7a9b51bac436dc61d3492a7b4b6772c1c94c4703b8be5b96115e6562d22111548e7db1aa8a29aac599c8c8c9f4a7d29c08f9a09891ff78c025ada11684a2a443284e8e2db7321428412c6d6290793cc87083f116a6d647e65163adc7b8a2e1fb3bf803c847b35b3f748c6be5603652270768433b12baf696d71aa0d45d014c8cd7a37&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.228244926Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736, Name: checkpoints/4736, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4736?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=408b0bc6cf0744bc18f3e1c761d945a4446f7876a40a9ef9f4a2029bfe2076dff4894a25b06b26d3830e433704f89613bf40a713b3e3e3bd6dda36a26ba76d3c7785c544d954d7f00c2db0d5cd12673911867c34984d449dd99c52e42db1b19590c915bd208d61f03cc92c948745ca380bab384ed9269325342b5fb6b0db11311c8432944f300c16d65e29a86878ab34ee57e5299afafdeaa6b384559476189846279fc8eebe853033424dffdab65b74568f57e4b6abcbad00396c838f93f61cfad4c682ea797516bb3dab04797652ccf148e86f1be9dd0dbd2d22db1277cfc75ef6004d5a60951ff4efb1601db077b68e973bc2ea8041960ab3c083548d0c93&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.227738972Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default, Name: checkpoints/64/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/64/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=6f41e93f38519bce6b9050a99c7b0cf9ae27d7d957d0730c6bc5b7664f7a949f760d2f3f40715289c03addb547b90a9218ed6afcfe76278df543b63f2f202f681c24658494598ec2552c56af4f3d1412ebcc67dfdd7505af070486a4ca02d3ff60f30db311360c414895678a77b8d394d21e0b4f3530fa8b23bc4ed4e0f275f2a91ada7653553d49261aac25668272e09fcbdbd45e3ca2c3145c042e09feaa5ccd8d92348f97baa1cf35bc5ec8cd6a8a1b33d3a9a2520828f7971b77ed254f68b8f7dd34a7caea67b278ae343da292b9fe1fedc28f16cf75e74b39f6ebe63963978548aad87253fa7cf96a602b9656f0c456001663339da2b3625cdd963e2e6f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.228418629Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default, Name: checkpoints/1792/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1792/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=92e140c1fc7c6c9fcf8fb015c8a0ef44d427114f1adc15d22bb1c2aaf470d7485120d9af64c3de0a5d11a8e0edffd346a56be2363b7ee5bb89dce0e5f3733c0d612da7c44798aab365ec7c077af7f74b4bb09dac734baad0c7c8c35266cb8b4978a9888ab1e8c069f96c72f1a979bf481de02ee806fb452fd93af27962ba41e4c069b184180d57fca66bd5364dcca03aaab68bdcb47df633de5a98f0863cfc1a60c061fa3630eded46412908387b5c054ce77cb36885dd5f2073ff715f5e7b3604421b5417d6b3f3c049da45c8ac160093c5258c0502a4d1e5a29f78b3a082ac10c2323d5ad8de9c0ef39ce4028e5bae3c3c913d1d9b1c5333827f6952fa34b0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.227816803Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default, Name: checkpoints/2432/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2432/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=3558f5f3c1202f96c66ca26e70134155b59f29b1f2a77d3c5088c694303509084ca518790678785bf6892cd88abc30cb3edbb903657b7da8142df62aa58d0bfc521a30a39ed3d6a43d2598f2aa3457a6f2bc9fca7a921aa17deb69e608cde14e1410481fe053564679b1b3acd2b80cfff8e59c2ab2b9832aa8bdddd72b1c414c25c9e4b2658e0a131bd2aff4641e41613e449b0a2d6afc6f22b1127dd5af6349537b2b5e7eef50240f78c04c1bfe37450a264498606ff97a7037f27231df092549c455e489c9b668ffe16ab53b67a8a54e4511df95c7d48dd8e192a15afc2c7c91b885950462a60ac6c6480d423ebfc4c2ade390843cfb6240a0becb34673899&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.227846034Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/array_metadatas, Name: checkpoints/4864/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4864/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=2e1357773d195b88f9f0b1e0eb6ede428867f9415994e4e0513340f294f0dade1e734ba55f070e6a82024c6de568ff3ba03974dbc73d406bab7d7b54b5a7abbb915ffb8e2a35c8aba0e93ae66aea0eeb0c89cb00d2bc81414203cf718b08143ebbe59c27cfb283593d68446966315d0c12ded8195b0fc3c04d292afd9764e7e2e04983844af7681bb9d71ca110058dd06d15dbe80d68c07df7a6f0ee3f9b186bd0bb30d339da773f6858b8d7790ee4123eac55e065dabcebe626a299fce8629ea0d008466f2117c83da913e596663f09d932a95925b6fb969331e86938728c675b0822d2986ad32d0aa4f58bae749b1072ef6ce8a31a20180d692e71eb011e87&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.22789194Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default, Name: checkpoints/3008/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3008/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=2b51aaa46f5ef6d361544c4b524869d702f2ea2504e055a3364d6966a2d29a99b6558646cd4cfed21fd7c0b47e45c12f26e30182cd245800ba4b446c75a1011f9a8944a08898fffd8ddc0c4071a3dc30d6a0b58b4265fd79b8995f7a0be38efd358565e77193d1fb68b4d0508009d8b515c75f31d7d252d53d707179c98888006ca7b170690487603838ee88345e07b12575bf59ed903d471220bfd6752803813e172008f5dc431eb5938e70802e3a1fa8cd1229e1755820c3a1829281afa28e623b8fcd28d90f631db7b60169f39c16d9c3b71bb50b855eb410b4d09367cd7b1254e2766ac9f029e58e6fc2d47a32156924c6a3357ab8da056ff472cb0fb968&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.228694069Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/array_metadatas, Name: checkpoints/3200/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3200/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=0c73e192c542a33152b106ff6bed585f0b9e3c7c84508aec6443711a0bf03b325b782a5ade70a6df94cf205795cd860b847591acd9e530dc70be1df8fbc2f35c747b37285070a05c26a308b703ee58053842b3b0ba53b9a8d5949a1f10d4843bbf48c9e1205b6db359b29ce8ef6abc13d97668b6fc3229929503a1250fff8eb82c0f1dcc793c8ee0d34d921e90e428a2e94560b7a9f7525693b3c3d5c68e899fe2c6b1ee52f2e27b33e1387ce1193da0e9c8c52a6d0ddf2fa191eef6432cffa4ab98e028d12e96339db0843a8aa1d7b7b374e1cfad3200f7de670e0330377579f47b3479eece26bbc3644e2a4cfab34ad12a42ea7fe1a3ccc6e64b7a55c50212&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.227929184Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default, Name: checkpoints/1088/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1088/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=9909a695a3d4a727d5b5b58d9c9e17d32cd7809d3631e94de8c21a053efa554c98935f4830a1849bb502c52fd9daba95c8c79496a7afdf29e8d386adc9b018e294c09f1e50febfd76212bcc185abc3fb065acda44f9c23461b3f4570576568bfe2177c15db082ed5fff4a80aa999d3b0b0556c90e593b6bb5d51725e29d0ab621673117e9997e13ffe7cb6d60a29aa737796413fd3803dc55711e4b736cd9e83ef4a83ca0ed73903595dedbc19a787c65eb39fd2b8f3dd2e8ece23e0822d5fa7cc7cb9305ed48553faa495e4cd4750562aa882712d4096e6685c6940628551d6ade5b2caf244ae7420767766b2826bb07f8523e0d9a7e3b4a7374570ff7e4e0e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.228762475Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d, Name: checkpoints/2880/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2880/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=9d41dae4e2adb4952df8c8143f33cb960c402e8fb34af096cc2a5c497672531b60b0609cc60e0412ed11363896c73a0f2e64e743d2502f8978b9815d1665b6acea44e06eb77f6f46a1d66301e53c6fd4328e708ddb98568a89b060da575fc17b45f2fc75ec64a7dd5f49f31b4b2230ca122c7c96c3cdb56ecd69efafc296056f49cb7f7955e58befa16a0ea345a8098e289d1977600de35e4477069aa5fe5586501426f864e1aa9795ebcd9db0c7da5be1584d6006da7e1a59f2a8d5997883b4b33a710c785b6f8829082363a105381e12bbcb6d19d8b14e82f3b38d90d2a05a436c119c66c391cf3173b472cfd9911b7a34d492ed7aaf3a10d13d6e3c71a506&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.219090521Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d, Name: checkpoints/576/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/576/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=9ea5c06deb88eb64490a310a4b934f22f49868cc2de9c85e2de12869a36738721bfffc02fe22823bbd3753ac19165c55618c87a3558eb798cca9dba318f8a5baf28e2538a916df0d80bc3e95734048b70dd5e3910d521acd20989d4131d79393a4727ceb583fbc71335ef049dfda1b42b54d8814bbdc038314b241b16a0d1709a7be344f29268e037741a9070eb89dce0af2caf213b7f2e3814790f5245d448d77ba4c2679e4fd4bfda01b08855765cb1ce2636282b2a822ba70998e9dcd8e71e814ff734c94c569a5aadd739b270568468af958fb1a81b286027490aa2dd6f93e9b6d978bb154dea38a1ecb4de6a44dbd61e1b2adde80704edaf708e893b420&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.228871776Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/d, Name: checkpoints/2176/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2176/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=5cb3de3a9ce33d13dbc2b5750fa26a3e39935e503241e55eebd0bf8b0287bbee323b4191a5e9ece0e5af34d74bc2226d59445ad43083f35e8d144ced0a4581aaa9858964c3575f4cdea9072ca54886dbbedace9c0df33761a441518ee1d7519503433c5540c0d222f557b3fc5933bf6f48bf6c789cfb3a96a721b0165dd15cd3fd464dae0cdbe78bd879b96c6b063f774d44ed2118a4544827b9f9117dfc869c4cc3e54dfe95d9e0ce2bc92d984be8e70a5d0eaaf544dcc65d05238af0f63eedab063312daf8383830d469db19cf94c2f817d86738629ab469a8efeac33a8ae4ddc24218e6672c772348ad28c4f7e595c18a3157da8306464cce990ed77a3f96&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.228891718Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default, Name: checkpoints/3776/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3776/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=4d55ad4b8625cbe1c24fd959cc6b3769fc4258c52627df8ced64b8b2a929b28577552be1288f04bbe498c25e49573801fd23481091369c245e118cd1883d34c51c147dbe5c7b66a922e875a846d3c03bf7161e3847317e47506e5df620931d3ede835ded82b189afdd099d25da836d846aa8fb949368f1bf8f7ae3a112cb092f3071abbaf105962dd1a2535383b8376bcae9d0d97179fac19edeb17ff927ea2927c12e2e33c84062b2c7d14524a86067b46d0d19e61da684936ec5708aafdf15571836014d5d1173755cb9e9d0ebca8e417fe9bab30ee6d33f7e7738e1fb6e1343866adb9428e4d022325012d4720f1f0d42c4d687b700aa80acc75556bc0257&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.227211519Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/array_metadatas, Name: checkpoints/6080/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/6080/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=22f15f10e41aeaf715816a187e6569de7a1f09464b908bdd234148ba306669d5cba06769d2d8c6a9d7a4cd68a28457886d9e1881ecaf070912a6eb3c6178c27d5d4995c9afb638baa6e8493f130b8a5ecf0d15c34fec65c42bf633803feff2778880dbfb7cc8d91d0ee42856b6d77cd0f3e82c805f21cb34caa1918369f28bc14168700bd582e22d20a26c145628c342fd0bdb6fff4c4e0ad8cbe8d5a1f01ec2c4fd4a021e34a1d13f3ef8d545d68544252fa49126516236e4184765f716b6c236fa144d3c060a8b3714b15e902879aa650d29edd6e18db2e85254c744d00fb255af4724e08cb67da799e88ef95e4f969ae4432840025783da4e72bc84532f07&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.228972919Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d, Name: checkpoints/4352/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4352/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=b876021d92028af454b6eec6ff5bd1980914efa0687c449f014f01f67ddc1f929aee64e4d89f3c3db9c091993a36eac246693ba243c254c0218cd2d37ecfc23c61cd97b2178c77abb62d46b933b815561860fa096dbd496f36796f471c85e16b2f6859f0384a2573f32da6fcd6385c50061a3c1366c94ce29eb15a5bf3708702405aadc69b382e24d7d71519823c4bbaaaefb075ab4966b3e28a3e99e83c39f0b9810ff1478d746bbdd0bdb3d540804f0c9d476411a4fcaab111e4296ddfab485923a5b913849f312616a45e07094e9215f05676b8cf785fbda673cd0c7d31c15cd96a1d58cd85a9c648c496896d06e1ce6c6e949d00d9489574bd6729d7a850&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.229014977Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096, Name: checkpoints/4096, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4096?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=0d010801b97ddb94febede2b5c0c7787cd1acd79c6a7670ab36f3989b682b16b4192b8629ca2953aa97e4661e61df2921953d68851bf6ee0cecec2af45fc82ce4cfe3a1c2d7968a0b81c5d7e304ffec84aa43abfa3857423bdd094ffe7a6d49cfbac7a42a19507909b65ea9ffa4a4ab938d1ea3a2f7b10be94e7af37f601b613031c72314b084a9357a23ce140c6c8d9933fcbd7bb6e2b82907d2dbb3cd2d18843adf90d8b957e5b36d3860aa67f5d2f99e75453567ed3027218e190ca614de6e2c1e09b7f7b04ca11639d1815261dd283fcd477aaeafb386b11f5bb7ae53d0c1ef0802fcdd386adccb106ab7130cf631a3c8f57fa7627c0409fb287d87750c6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.229048262Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d, Name: checkpoints/0/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/0/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=0b675f0f7c08ae74ba1d714546cb8df0c8cbc5e40cd810097844cc647723bdf57e9c07554895d81d3e8850efaf1a7686f464d50fb16f4f7cf6c2850f4bf5626f43ed12685f97adec55b7629c3b6689c66a3844f16a5ddd7257d7cbf345e0c34a8f9233aeb0e582040731c6395bda0f21756e17f09c0affbf70b1a26804e20e04de785d8f715e601a888e0e181d0025ca3bb7b1a11875556ac1a71daeee89acdd641dedf3b5960dc9c01172ca5c429b8d258201a5d3f74c76dab30a58f64cebf02f74aa35c2c8ed38cc3b53724e4405ea6aa7b9b2a0fb9f5dc7c9bb6ca76d67606f7c47018efd05f865a17ca45825487c862d51e3213b441fa44b02d8ab5e03d9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.228722789Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d, Name: checkpoints/4672/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4672/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=53781f46b716aa4636c43fa90a729df23efc2c4d61ee47bb8d0d1c97e5fe84f27e2f279325ad7cb92ba53a4444ac7abb3d94ccff45925a52cc26afdc2a8deb37bfc561931bac65fe2ac6fa86d130c7653a9e510997db984d50c1c1ff7d9c326011d2f360a67ae44ed973938a09a62203d7fb228058bde3faad95ecb929918a320e7e7ccd6dbffcf6f66245d78f2dcb9b71e01a17e61fb68026e3408ca07d753927e1710ccd627a26e8c5c563e46442a9f1c1d9a43065eed76f49fad2f7eb68e2a9a74061ed896da62f452e9c768ccf513de473b70ae970512bc99d48cfecaba4eb713ace56203d91d35716c07ab5b91bd2260994cb4870b09eb4fc673356a6b6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.229145165Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/d, Name: checkpoints/2560/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2560/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=c3592f5addb6d7b6b09ef33f635190d7a406cf80feaddc810212c0af606cf4f9f040dfe77fecbe97c8dfce7c58e0a9712aa35102189826583194d0d36e16b667229d9357574963dd8b112f7fa93b800c232038baaccb39431d28f83df74383ccb08b7580e8b083ac46dc75b8fd74db972731abbeed7fb053f68dc36c7517fe37159cb77c89d6d48604957dd8d6274f9457e32668a3cf88e8229990820d257d8f290608c6fb440f90bd7eaeae498f83fbeb494922f09f355543c5429a9910a9ba5a86d59987116a49609b1724e3193cbfda9f5c5d4aa4639331dd2003b0452d44c5f41834e8c631afb4e50f5e9781fc6fd23fda15a6bccf4d9b3300e0498dcd4d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.228966412Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0, Name: checkpoints/2304/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2304/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=06f5a5f1f443905c7d91ce184a836dc210e6a3253a22b0f6289933998a8c725578257431a5ffab7ec9b43cdeaf9c13330b74c1cca40ae6a6aae3c1ce736b11a501f54307d4c2dd80722edf4e7685f07b4857a81b7dde17b217a9b2f9b8922d21bde718057635906224bebf5e333beffa8d6e48fd9be684657f40b9a65210c92894153bb705b466309a26e9778700bc6fab16475f1a6890131b6a32867cc78d7311019c265aa54663d1ac7e2a02002d93ddd2fe3e23b3d31b26fb9310fd4bff5f30bde4b5ef3d4e3fe7e11365d55d0a98e6070aa78b078044be6a0aaeed9c94eab17cf47194cd17f24379c472d90c0d7ebb0a65daca403c13be5e231be5c1da01&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.22901457Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056, Name: checkpoints/5056, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5056?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=7b4e60e49d787cbaddc39fceeba48b1470b23aa4dc67563b0207ec490f7eeacf83d175f9e5a3ae93c1ea1a3971bd823ebe002035cfa433a70500574cbd2e991abde218353e0a2daafd6c458085714b549d758c94ab41a2496e1bd090519e16d4a79d0caf94638a849318e7d2387e327d063fdde3f5ec20354eb75fcdc3b4dd488f1bf916c565bb30f144b10e7cd5c553dce57eef6502a0873a5ae6ac17788fbd1e688dbb541423f1cee8b727f25a465ab9c8ccd47e0d7d923b1a12930f939e327a39e192e0abb3cc6a4e13da9223343b5670c3933fa27d5b3f588285363b6cb3989823f57a835125dd46280377d648f2b5e92f769885a94ec65b36b49e92c2b5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.228965056Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d, Name: checkpoints/2240/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2240/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=85bfa137a8fa94167423963847271ee095a2d7de6f4b80fad8660ea0b872e76df5c8fdf0bbef21c2f9df0ee5216ff47a6a208fc938c1a553e4ac36119d872c92bbc9ff9c8958f2a77424bd5103f4102cf0b9acbc755ce6e093fec309a3398228a8df59eaee1ec26b6dfeffb09cc8cff4096251939d69afe92bc0bc360df9eab28360243d59c70e7d7560a6c3518600ef0009b3b18e4e9f8d5e9ca82c017f91ecc2bf81097a62827e1c434b243b0222bc5ad0e18f95b4c31f3903e5caed35da632372fdf310b404a9e4edc2d70548219269152e9d51bdfb42dd8e658e58071611d6b003c0dbe4541c64b414a92b6017ecde689ee382b739d9f96c81ef04eec715&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.229853136Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d, Name: checkpoints/3456/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3456/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170109Z&X-Goog-Expires=86399&X-Goog-Signature=1b71f944928fcc415e24bce6dc654f0feb05c779d09df2513c4e629ae397c280cd1da47eb745295065f14b6b446f386113c4e61044cb5541dbc7c9fd958447b20f5456dc2cc84f5581ca92110855a1512855ffb330eb25b737d00b1a6844214a6232ad4f98a46217cfdd7405e4a8c6cdd3443e776bd93e54472423d4921a4bfd5a52500e6d1643c1ed430d8dd6fa2e8e01e06c9348beb41727f2391edcc9d3a34f5df729b9880a3609f222f41a42b8fe2ca4d43d1e8b829ea5b1eb3ada6cfabf66017cd68bbe8cfa3282190ab2ff4dfd4061b45cefe6b7d9d43279ea867ca3e8538f424b29b625bb329b9f77288937220bdc7f132974c5b239cc4ed376a46899&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.633865013Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d, Name: checkpoints/768/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/768/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=2161c0efe28ea0f5fcc6b1790d77d282ca5285adf8a75bdb4a0e30f16f41b64463a94f7672e77acfa2b7519ee3503dd14595833db740edac71d9c7e6ed74a1db6aae83310211332ba6d770e4f35b5ba36e846e450633adaf4b0bcd019b4cf40fd47870f8cfef1be08fe5a6a037dcbaba4db229422beea54e48f595bb8c086fef37d2e895526ff278ea6af1d6a511b3c9ff2bcf3012f272127eaef401b11ac8a74c2af5382d4029f0d528fe16c4bebda2eea3ea2add64b8c40d329084f9ee1839711c44cc70a9e6c0526489c477aa9099560f65c0eebbc1b43f05f9cbf9e4e44985816402dd4362be95a86dfe9d510bd79faf67019790a19804fb59487dad2ab3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.633922834Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0, Name: checkpoints/3008/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3008/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=c34b96cf7a5a2c151649e686f1315b71c0f275dccbdb7a9088d9db7032a116b7b3d3919a451ee87157a1fd86c05f5f22c9fdb138e020b0e4ca246164cd7bed5b6c40148f0a58367ac3cf7c99d0af28427b264c6a3dca7bf32a38becd5bfeccd0329837872500af232f971dceb84edfda86d368a65c680dbf7238c3951d68c6f17805a1cd437e3c06d1c66fc5f0409f0c5cde266f97374fcd5ab19a7cb95a5bc06d81890ae253c960ad21352e7db64cb9b1202c9a3386407a6607712c83f024196de34130a869523e380cbb812962db82b1ac7b28e7c84524757bb7cb6a09d01188800b6304c0eb81b95599563c96829b5e226e0a14d5094d5f264d1982437281&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.633959016Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280, Name: checkpoints/1280, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1280?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=217225de3538e0976dc933980ee6b00009f91031c1015af7fcde1389cf922b74077fb29f837b1de6ad9db5a7d119ab5a6753b64ab6b58597bd441dac54ba516d6cb46dfe02d883bba0a41ec16400f2962a7c5f9c17d25f9bc4ed741666879b2217b77a6228dcd159fa40c6417e59898e07196ac909e36292a5979b113bee5e41e920b3fb9a1c1d8406679a12ff7db9c2fc74a15e17175d11fcbed0888bb3044e4751eacba379b060a8147af28c3a74d9026b823888df2c981fd7540e58c6ef909efc5506a302aa755c782ab1f715551d6add2f86dbd6efa4ba270571e7b3f3945e1f3ad780d2434c831bcc3167dfe05d6d7c6f26f0565f64fd895151f9fd916b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634001465Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/array_metadatas, Name: checkpoints/4544/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4544/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3dfee11a1e3ea9dbf1b9c471bc64e89975666b700f21de6e42a539247235064043f28c9f02cf407b0e25211a53ca5669785795e9c4569dfc4825f8f7771ff31bd04e0f502716e3410b93a1345ddb6514fc203349102011ddc1886d0338d3765e47c924e592e1baf8e2c9f39d11067c94a29265e44a31c7e830c637f2b5adef582e21194c37e594e5d7458ebce280bdf6ad5fa7f417c1d16fd45d8042da491fc6bd768e027e0aecd3e8fa70b639a5567aaf2b32680bdc271769138c46636ecca62ab322f88e7d3bb576e4995331e54251639ef64ccfbfe18ad1af371b4be34fdb663d4df232c825c1a83af4d3551a6966f4d18ca86c4fac46d640292762e6b971&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.633865051Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0, Name: checkpoints/128/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/128/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6e9b6b2282bdec29bf56bec60308a3120a12ea773590ab68c73abf370184c74b3f8791e1eb71324c0d5928fd67e600ae07d7d3bf8f9d012f0dcc4df65f513ad226200cc067fabe283d771920ed0f9c452c4c25544bbccb0c6aea2c2182666bf54872cce195339dbc087e19192cb964a5502116b2dd590aa2bc35ab0f44645b62312f7b99ea909c94aa6ea8ebad9c581ee3be543714ea91abdb2e844379fe1daa3ef2aeb89e7749a93d9397f6f7a163b231ae5d79d2484345d0fe2fb9d727690fbac569c3d9c826232370e06506ded470c2bd6e2b11b2560f02e1e589d9e65f2b603eebcd78cffd937fa476ba747fe70b3ec275638066697dafeffd3addd8767c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634016779Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/d, Name: checkpoints/3968/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3968/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7b72e49371cb4f04964c833ddb4548b1ff81a20b12ad6ffabdac55dd50e9862e8f31b096d9e8dacb144990fd0d1f4e30cbd6a2ce39d2b7762f80148aafcc439d094de08a8e5cc98dee01b4e3608e5731ed081c7b898b87c4db392528c955bb90713a17b4030e7cdd549c329ebd21bf1208f091981cef965d7d5632cd74b755e1a3beed8d54af85923d4588dc15a04d32a2f3e98165691c0c7925d7c6d272fd462e7a5a6f10e41e8373754311ce324f671fdb322e163a2f5a1f3b053efae2dc280f17dd7a5962ab6da15edc6153e81002c6c96d64f6dcc179c6073144450d2b0fc3830301e58505f786904723912c1d58f0cbf57eddde48e00c48abb2970de7c3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.633942722Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0, Name: checkpoints/3968/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3968/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=28aa282194de100b0c637822b9af73c0df3de659d07a2c8e725a6e12b19b2f3555f8f94b3130b629394cd4f31a3d62a4b20093a0bea08d70aecf493fc3e65cb569071df12e33e09f3bc8504f5de26fe484d99875287acd7c22833b3ef7bc695af41e70ac2e785f83c593569d9e8652256d14dd2f431cf7d088fb4bec7188fbd00c4c2a960c4a121d642cf1ded6880442ce8a351d14030b8722d54f50bc1b6fda9c5338fc39a5b2a9e93a93af47f69ad859a24f8bddb23bf6a123d38d2d768b1d60d1c26a03c56fce8ed9d191b3bd4e8d5574b0afadf83cd8897ec42d22e576df3cf41749f566600afd60170ecefa0bcaf2ee6935886102bde0017f17ace84ab3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634102526Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d, Name: checkpoints/4288/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4288/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0ea9fe07b9ffcc9eb7254f95315ca116b9e82f885a138e9e2b7b8c4f493ea53eeda007e4517ca48faae337cf8faeaca09f2cbb1857dcf03270a51e2589d00ed756f257bcb66b4d121c30fd3921da9a404bf3067e893786a7de298dd25caf44881169561d39e6619852e28c380b2f9b06ef2113e5db7644bcaaccd0c58cce99399709f9992fa734c71150381effe142212d9a288dc8478d6afaa1d027447e9387d9eb232b12e9dc7735cce3238da36e402e6814880820d54b841fed8b3fd7d1bae8de961d0dc1fec1907947723204b6fc505075b59629cf5d5450e02efb74a88345b4cee7b17dd4f1414904f2613a58f94febff0642ce25e25924729974d521b9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634174365Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048, Name: checkpoints/2048, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2048?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b179f052496e4c1b73ff3c4aa9eaf50f5a7a5ed316e2917505bbf6d7d6f049fb7a771774e0925c8e67693528ccd141a66b159894ce43661a433ccfcdb9ad5c2597adf9dcfdac915641c0142b5c13e0d750cae2562f284fd94c484bea6222dc1645e62b4fcd4f3d511de917077d1ec8b6a6b0605f9f71fc75f409caa4f62ba70d4b2afd62aa786f2b646719c265675e6f0b9d2cbf4ce20c46afb36d04c2940a166e87dffc890894899c9fbfda26014382ac24bf71acdb9b4baae13d6d6d38cfdc55a4438ae5d90e241e96f407d5bab2d1f3c5d9ad866b7e9e3b1d7007407359971adb0c1b890feff89bc4823d744b3e0f5336c9583f38843a870356260be178a1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634180966Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d, Name: checkpoints/3520/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3520/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5ee492f6eba1307afbaef4076e408d3a8b3772c2fe331c46b87fef21821a2112c4c5103961f3fad822d3c9b4be6f55adcddd16b9e5c5d40378f3e14a407e9de71efb6ac0edd32cfab8e27354ec159f455b40bb32ef4f72e63b38ca337a0c0adc85bf155c8a68068f33558d1aaba2364bfafce61927a32a29cfea01dec33eaf18d7a50ac039e23f60941cc760f3b895c531f11607602fe1667d2476de171674971601389acf0c176f54a35986ab62c901849447bd3d050040bcd37ad91355eda69b132f6f7e26a3ea1003ff3784dcc13a603a500408e032a70799141bf165102b60fab55a48cd908caf496526cafbf5726f392e7b971e3094ab4da553753136fd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634216978Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d, Name: checkpoints/320/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/320/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=1d918b0d8141d83c587352425fe63e6daa4df4284f1b4a789fa6ee67635924e7bd96509d0f8983fc2a1859c891d05935b9dc595d5397db9492f78171a6b0311c6df3c34d3384ebe47f0caa140aa913d23cc35dbe19632b0a8cc65d29e30a51fb84806d55250a988102ddb1e31263c81bca7f9e5211ac98deaa96afc04ca0d6f593f9d40675fc39ec77bd8ef389ad22fd988c441e1168b74fb789ec47574508cf70fc1dd15720476e7275957e540ecba860c72b422a7b30f22fc039533f2a41ad0925a72ace10d72f5dbd0d290360f8b7c0d1b468f92f5eb990027ccd3c3909b5eff5528ef17f232393d11d3786cbfa50cc716eb30f2c866d8f5cf1c48f3a4fa5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634262934Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/d, Name: checkpoints/64/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/64/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a4c90705294dd19a5560689925f9551e443943b5141c2be6967081513d5b2b235a5683ee26a7f94a1a3b42afbeff3ef5c604b8e42483734fa91250ec9cbe9eead6dfc6a4a9888d42209911bcecd2e333393144b7c3e2f78907e08640b33ab98a33c052553ceba5d21b604648d15fe7c3b1a52a151aea9a2b246cf2c6ea7bc4b4ac346c37bad968a8c67b0b00b42d8bfa922c25c83afd37fdfc63a10fbfb026eb1e0659e2fbeccfc797e26733cd9d350db4595c1170a154f6ca78c6817ee2803ece54c0e30b2e149ec6e4a4736bd6abc3401d0370fb170ebd44384fa34cef1d92dafe5a54bb816f3541d8d306d573a897d37a332d8519be1e161dfa36bf8bee9e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634268459Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d, Name: checkpoints/2432/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2432/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=8e25877650b9149faf0066e76c1019da840061b4d2fe6613e8ea824c3991efc7d15d9c1d0051ff8442b734d2e7383331c902f74b1d14f0043ab5bd5eba1c8457b0b0e76573e1276ff8680a7954b6ac692568957c35a063dacc920adda11a0ee791c07023a5cfd5b7505acd4271fddd30f469ac4e00871ad8b47b7d396534be59a928b8631694213091f22093c714b556dede888ce632e79d44117bf1154fbf729362f30b6b240dab95a141e08843f15d14a51f83e427cc9dccc83bd8f16da75da9d6d863d786eb30a08c3a773a585c732d69a912c31cbec756458cf8f80ec4d9278da7e75dc5622bf70c0a4ea26829f8d0aeb56ed2ddf8a7467ffa2d2d9361fc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634263511Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default, Name: checkpoints/5376/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5376/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=8f7a83a91f202b912cc9323e8beb14b7a2cc03f2d0623c58c4f8acdd0fb9defed38e8b02ed186212f844b5090f8f5285f6fc9f92b237f8133b7d22eb111cb922a308885a4cbe4818dc3698e2b1d178fdc266b95fe23f1bce49017f33a0d372dc1301b952d78eb1fc4fa45febf082ca3d357d1b10d7a0743a71f4f6a328c5f7d66ecf1787012ee16b21b5b873c689b4c49b6f1f4407276da3c75c5112cc086558c6b15a748c99bee368115fb231a241a26e452d80e404bf39a2bc4b210b95f6580a08cae851a74a2dc0136beb54d4f139194e2f2da5a3b7e08d067982e3d58cba5d6005dcb454e1ed0d8b9d828dc36f6422f5bc49ef482b72bd05fd33e3341ac8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634313909Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472, Name: checkpoints/1472, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1472?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=8213b9188ca83c8c1cab43542a36ed86d38d8f84a2d8be9726df19d880c8e630d6d78222f4b347f60f956638badb97a68f3f7764bd7e49b8301b581df8dafc63fdad3ca50120b6158cb31def4cfe7035a854985c77d43e75ca58e113a6cf127dbc10c85e0b9937b8c44e3f60ea4ba04d05fdf808766cc51e75b656832469780a701dcca7da5b8956a0b019ba8cf5044864a4bb687c1cd249f832bf44634b9ad6fa0b7c6023a742d247954aafb0baa7648a285b4bfb7be825c7a778ca87ab1f6f8291ceddc82e1e9f80bc499952cfc05b21ebb337cba1ebfa573325e0548eaee52f647edc01be4d3ff5bb012c253329ddb14d241e145a275df48b69e4621a77b7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634264724Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0, Name: checkpoints/4544/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4544/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=434f27f8e230ad738d5d14fae1f9a9f01271650c713f6340d19148965c6e6b343f25898a8b0cdff68a14925adf24b78d82502b694338e31d941b504e67765953e75932dc4483b05e62dd99ad02327ca8a777eccd489f227320027a3c9dd4ee79634e83a7d8c2c02fd779d9198a30cdda5c331080a46fade3265c1057b071a57ef5c5108979d0ff6b14fd93805b3d63868333c6508ae7470caee77b33087b0d233d8033c66edf179c37c42f983d38e83bbb20982f4d63fa64765802e1a80097cbec216f5ba04d3ca2d4bf61b0e42cb2fee0b8d6eff1f5f69d6ca4c0dc582140858135bf5816b25f290bc8040e9c805ab09bf98530dbc2e6d9c2e64f85b6829e78&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634335254Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0, Name: checkpoints/1472/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1472/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7879dae906889ff406c7f692e0f9fdcfd1a9cc9edb154b7a1d7ec8f890d82d2a623dcb0379573af5d6a31743b0c0a43906174f91d27c2cb437377a9bd76b2aa2d4391ff2c6b59038ccbda465e7ee956020457f337de4f9fe9144347a60709f54003677ccc8c23174e700208cdf2cb7effd1c599cc03e9a13042472579507b36d0df9b40ec5a222b8a835aa37eb91e9e127d7a803b36bd79a2023f6b0f81792c8161e25cd24267bd3e01b7f50fa9a23e3d56af2320720e9cde14ecb202f2d599f2ae86bacc25bf221bd27d072baa0854adb6d077a945e26b76b17a11f3f8ac2da4a2a4203acac3502d2067fa8d7bdc74ca1e0a6d8bdd8e0caaf853374b32171c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.63437769Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992, Name: checkpoints/4992, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4992?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3c0caf9fece4e91320b05c1ab43256fcec64c1a8405dd61bf1179e73974907218758043364fd3d787f5ab178120d0d90c63acb2f7944eca7c0725f3b7d4303946710f8137c3a3accc0e1768fb715f1fd8394a3365474feb8b40d9fe8f3b8b40fb5f3e8ed7f140fa821d172b8445823649e79deda80e935e2d324a9a86f09965dd98418f7291d694d67659a58722dec356b589638c9f80554ac68734371255e0a01351fc74122dcf53b46c4664d7a220770244cc3cc1d802d4255951e9e9bbbf218056bc36c4c173ee60c62477add0c7b939d82a5b3c15de46ca733ffd387674ab7c9ed46940e9411130ddc1b88127fa438fa7e7ae46047a11724cfca9bfae54a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634400424Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default, Name: checkpoints/2496/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2496/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3243d04d67c2aba5f037b840166d0c7d3ff588e934a52a71330cce3a3901da81748bd59c109a4f71d16f62a66d21d1915c0768f4465a3d5e46d066fab98b4bd44defbbc8377fcf3ea9dd75f0c487ac1fb05088667a5a27059cd50ed3f9eb8da7b5e4aae0a0b5cb4f60b7030d1803c1729eb6e58e7c1f2a3b0a6b9098ebb6f0c7d028acbdbfabb481e4800d2e652f55f5ec6f1663ac38a802872bec19c117202687c023f80ed0c8565752dee68a82c391a8a397ac0aef0e9be7d2deaf95ffb63179a0436c680a642b36c91888054580b8dc9119761266b74983e1ac1f93ca393eb0b354c0a5f25952986ca71a018ec5118fc18ce0ebf84a950ea29870266771f2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634415055Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d, Name: checkpoints/5440/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5440/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9b4198da6473d4df8b777b692b6f53001b30f53b2b0a7ff0a32244ab82fd72e142340b57086c57e0a7f91ffb8b8e810b5540f0fefb9412e2fc26cf1fcfa2b9a9d8bd79a13132e435f724b0675da80415552dfd97a96af87ddfd37a6c55bc2f34bc9aeece1f89510c3ab723066bf21a0cc5a0222aa69e0d72ebd7721e2b42c7ef7808d2040d199874a161c03ba9d5952a41df9c2c3146bba2a32de7d27b41229084acc1a779d8c04eb54e05cf7bd3d06dc27f78ab411fba638af7ea47b34bb0853b37eb4e2ac1d7493083b2338c26ff80aeb3b2385858392923d99632d8686bec1ca9893255902835f70f8ce2a76286b89e748f48cff9d8cd0dc53292d61504e4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634451531Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d, Name: checkpoints/1792/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1792/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=47a599738523f64be95b27eef1a63daf4b10c6120400c9647031849009965e9dca97497e01b7cb8d3193016e289cd4c8894c3aac36690bc70f4ee7601ca255c623065109cd85849ed7890f681b4678f973093c0c6f57915c4fae77c5e58a80d4f6cd34cf7e3b3cfd49df2e449d66e50bad2312b17f1148bd085b299446f74649b78de34a8c700680be5e52ec4f041127c1d2a86b26e00c081532f5732e6e04f0be28f6a9bd78178031db93acd8cd473c87a8b51dd9c447f05829b50c89ba3486f4ec7771459f1c4ec700f7ace0d92825a8be03bb3101c2c7a0753a5e2bec9897f7ab78c52c0629556826f21d30faf03ce05657270bfd8554fa398509ef808b2a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634462118Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d, Name: checkpoints/3392/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3392/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0868b1cb215065ebebbbddd79c1472c2b7456a840fea9f58e896b93298c1f64513a86ede233cef94d8e04a27e571525580b4fd3853187a5fc749156f4a7df5ee9e65e6f645eeeb837e927a72236781de96378a42cc3f8c54dcd698c25b775430a64503c602a7eabeef59981bb36fbe2101da5ce1870516bc54f8824f5c8d2d3d175f188c3f9e668e8827ca68416048ca49824713e90ca0509b15fc50b677d8503dffef01963336a9978be7f2286f8d1a598cd55cad44e73053bfd1dc368ebcb2bf435834f16d01d62fdcee0b5efd7133316c76ea7e7f007c2c5d989f9e78ebeebcf20d9f113a6172d3fce198e9735bf7492e02d0c71b5a56216a911e81a2f1e1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634509733Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default, Name: checkpoints/1216/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1216/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=690c8e7b81ef889b1e4d0cf307792408ce45895c5d0521cde30b30336e744b2175a0e21fadbade080eb34b59db56c18abd8f19f1f9d0b05ce27d0243da2b647f3fede43d0c87b0bc65d2ce7f16208ccfbfa8f4caae161b766d01fbe50b983c28120f96fb8375aade6e81fbc4c2d29fd5b62777061078f64f12c44a959aef7ccf99778f11abb7d83e046bbf9c1f669f254c96f6c8e2b18a6a55b92be4cecf3a91e5ad7ed87487d5e66648710d4eadd36a8ceebcc691f44b38cb814a2ebd1df4defa0724fe3317e9f183674e76a908c5c18d2e1d94e6886636ea0a653e0530a652c8171e58ed639cd17229b9c1c4d903556aeddf890f54565de47f9b0207bf4eaf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634524033Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default, Name: checkpoints/960/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/960/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5f9dc78751ca535ca1990c8af5de20fd215286b32e376d8a68d8325406cee1be4a8c5536ae2def3c5d87b48074e001e81f0346d48ace030ea6377369a5ae0f31100159c6ecfea0d650670c5add45e5657942dcfcdad7e8341d47883f2b4e78b884ad1c8a63961c64f3f4f37bcfa61f773583813c6d6420762cfdbd67ae794f23a06f4b10cdb47e4bc2de25c9090c74b7e24db52fcdc9008accc5e4b32149271f1edda9d84edfe49db9d4627b048c6ed7ae7bbfc32874cde85280c2f7edc336b42852994579689725cf467fcecfcb2ab4f730387533a1fe4374df9450e1a9af0a3a5b555d39352da0ab5e9da8a918c03b0faca1c2e877cee66bd73b2422bfedb8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634540234Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default, Name: checkpoints/4224/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4224/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=401bad250b88c6b23f35cb06d61f5ea36570333be189c17ef10793c774ebfc991139045c1662ca3318168157afc9cbf13679b6dcc05f878d6b8be1e093af7967e55273a61b662d657c5a516d23df5514430d6084e4dea65966d425eb4d6f1fa4713b015f8bbab258cdb387a3c3391091bc8f165686478744cd06282d8f92982227acf881cc30edf565ab124d23ab4504a70861bc1bca72c051341380648d1fd66e44bd4a5ccee2877cf5958200d2c5806af176e803e711883ff2e87d6d9aa63b8f0efb010f14dbe026c39bca9b05983d24799c176e3a661cab8bd2eed9eabbe2d139833847caa237c5ccdbda4da29237170250b052c3ef404aa21796b8e9e8ed&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634567553Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432, Name: checkpoints/2432, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2432?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=2045a62506663fa8bc3be5a181845f45324fe707c9cb6fc0d0ba6a137b64cf087761deb64059edb8d3a5ee789ed5a2b9b4734b42f4aea164fb0a2189757562a230d8bda453697be3cdbdd423f6445654cae33ca1232c0e23913d1881a80a704dca9d0cc1db3791a9851ff4c7b283e3c7c9eedabdfd2be61a56dd13fdccc8271ff752121275fa122f6adadf08914c8732d621db00c8cfecb9a52939fee6fac18a4826a831af48a89180b3fcf5b09681868745f50daa564bbf89a616ad666e2b42ffa861927363a4765c5ec00710c55189e3ed1e6b8645b4c6c1c147779951a6e56767e207fca9b790ad77fcb97b43aecc7b8b8479ce7dd4d9f3a91ab3caa0300c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634563349Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default, Name: checkpoints/1024/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1024/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=18f55afae89fca3fac13248af6846b862b5b9481d3217fc82de168565e0619808580c651db3d1bbf5c4fff0aca12312b77e5f835ba2864ece29307c01f738dbc81cf08a4094e82e4b4a2fc6305570efa1e249e4b7038cca3d58fdce022f819fd5c31374bf85da763c3b1a90de4cc4c79db85d8892aabe26883792c9d2aa7a40e7df23213ba662aedcef2f5d656ccfb7f9c7c886318549643d2196e09ddae343b3fca9ecec6816b4af761ee8d322997dc34c32905f68a80fb827a918d3c7364967d34a0018fe459e43cebc1983b2b7315ca46fb35fe1ba12f6452e5a38ce456d7f17eb570eec50d165077a40e3618ea7dac9e4616dbdc1eeae6cd6e6e876de028&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634594987Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/array_metadatas, Name: checkpoints/384/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/384/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4cf34e86d218575884a45eb417a2c131f2dbfdb8e67246cdc9cffb14778ebe1448a82195169ec83fc39fc6a6d6287cc28f38d24c67fc5169352c90ac43f20dd54a79c968424fb7cba3c2863093769979e4c7943532c0ab073012977b927077a0a1d62ed351c58b0d46e439b0705e9ae582d98d7c1fe4b2dbba6c9373e321479c8249d1f2bcdb892ea9a988bc744899a80f85461aea42a2fe910372ba437369996a839d94fad974db2a9dce5e39e38702d24ffb7cbfe542e596ce9fffd3ec8e5e65173789cb624130bd30beac9a94e6fe4b26a2bbf25e11e8d3a90f17ca9c897749f84c992ea0b6eb5953a161ad9605ae687f942981336ae8484486e93b8b78b4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634617527Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/array_metadatas, Name: checkpoints/1856/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1856/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0a98b85575855089048398996628715caa18c350928900938b8bfd760a709d1d89de38421275eea8a4609f81fc9e737a694788b3b89c637152f1ddcb4ca81a5c0e0f2c1802e7a80fe5779939ed253a813238b957a96893ec780abe4e95380078de92c21f8571e5b2a14eaf879bef1f0495ac06c85fb9d101dd2d00b0375bd11053c1dca833e71f571ab9e22c2cb52d322ca95d550b008f0a3d72d197d3c0b18ada2387b278c3d296aa3a7a43ab06613e2b7e980dcfb290fb5ac5bc84abb0a12c1a2ee188cabc132a0b5a2b533dabad6d3346d8575331aa79eba567a947505c5f93ac5fb99b9233c42051812a04386a980629b563298058a233145e298765098b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634632072Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0, Name: checkpoints/5120/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5120/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=056be45127973a2a12229a8ad6602b54ee83464ffacec71706a29837f601ff188d5a21447e8ea374a60fa5390afcf47e5ec58e20fe6a5980599fcc1a08837f0f2363fe9fc75dfb71812c00e67264a6309ba5fcb72815acbbce2b18e6db1309d47d3475e37c15a1c2f3a7e26b7da8fcca336dc377535322ac387a6aa3cf804e8cbd5b3794fd412e83b68a1591e7916ac125b24894c5d21e2d379d7664d7d9e3189306ee3e6075fd87307229c9d93cbb3cfa6b47640d47a229f7096bbbc75d2d3fc173aef6b2bca6bd65b27a47709a69e6493a1d90c3f7ebffc72b5ccd5b81d0aa1a42da4fd7a3c64adcec5a27eab12566bf1aef12407cbbc6eb2885bf57ccc265&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634334662Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/array_metadatas, Name: checkpoints/3008/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3008/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4672053f8e0784314619ef6f191e263601640ae691104fedde321d265780428a02da6cff2dd82865459caf65677c5d209fab612418ae0ccd01d0e93d058707816ec6630bf9acd018d99f41cd7ceef049aabb027ad890f46959c147e4aeafc35affdea14cdbcbf537059371df810a21079fdb7669ea4243668a4cee8bf197d72373208cde07ad65bb4389f7b5c9e52269dd13743795ff487d772e3942af02b7f5daa7b3e375a11bc2d6e8ea93978bfd595af7c9e606c001d0d91454a8aac8cca28cb9b320573d6128082a4d9fedb4349bb82cd68d6e127e85ed8298cba7831da53c3603b17c681218805d574801f2886737252d2bdf970c9ada491ed46a5711f7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634666372Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/array_metadatas, Name: checkpoints/832/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/832/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=778264567915745909dcbca41f3a8eb5c79aed32d8e2c7e04d3dacf9d59820090f46b6c26f77736b5e827f2bc0b857513000b852c18601748d58717f2ca4251e9b7cfc3cfa2deaa5b3f936e6cea5f12dfb2a157f658cd03a1201c2a65672969134bea72760d344799d8d464705c05f6969b10b389b7d21e81c4ff9b32c209a9792305210a5b26c3542aa26500884b7e93951844c63838aafcac3a123a564e488b26ef696476d13a780599c3766118a9236ccedbba597429404e25e91eb313816366721dc3d2eccc050d616fe28dbc7e61618977370eb374dee04a8fd910bffc39763f67bac29176e95448defa57ba34cb1d4e8bf07b7a91cb2d0906f59691257&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634702101Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d, Name: checkpoints/1536/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1536/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=02a737de55f106ffc68b4124b4a3da3708d0d883b3d2a54a01dd14a4e3648ca646b646f623fb2667076409a02d25b1810f5102237dbcef8669a4455c1b133c2bc6d7cdd9e9b2d45246f44d8ab6ac6f8ef501c9b0b408ed4b1b9363a615fee6df9e92beb9061cc9330d1278e82295360bd5e0a39450d2ece15b3be758cc3b9a1d4d9bdba58511d1041f58938101fe31f2377b9a01221e8e65f68fc13a582779253a87d517fb20077422fe44ccfbacd6e268aa1dc6f80d66ac69d67fad9f5db928ea8b0183a08bdc71e997c9f9a2e07b87b69fd5234c34c2484f54fd13af9a454047bdde1209eb2728623aec4d309dd06e2e4696b44afa01e640c3fbec08b8fd4d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634713257Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/array_metadatas, Name: checkpoints/5568/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5568/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=64ecb49c4fc11307e37681c0a5db63e6fe60f64565c172b015f74d2b70dfdae52a9e8d2db0818e229072e9510f0fa51164e3c1ced1d5ffdb403c72fa712ac1b0530824e41672bf81bd20a34e96acc69976ec85c2490e4043ead45119a5d94ea65000faae8fc0d498f798e5e3222fda81ca687f82b31bf9ba442c0ea7bd8276a72d6ae1b2889462079932f8606ccfd4a20584f60c087ac55b66028d14855725d903b57653df681012fd329abc5f8b35c4cc59b3c7c48b72f3b25c45ffbf0cacd79eea8cba7c5053a8619fb24a6bd82426ae5152e0e0a1b4b3d6eaa9e1b039342ca22cea025070ad2c592fe5c71503a283d79f26c6db6210b57ba2dc94b89be5cb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634739129Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default, Name: checkpoints/3328/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3328/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b3499afbbfd789da9d8fbdf0b615c43e905e5569f4fbfdd555ca10bd466baa53211ccb226ede8aa8656cf44424507df6bc3e4ab9bd390f4f603e76c8394ab8545763fe47cbf5065b91cde760eeb7810774f562a8c1362f85e3598b11bdb2749e6d12844e54dd9e1aa37f7da440bbfd34f4d070083c7ac76d03a9758a7d493001de34c78c133dd16b88c5d2b411c6c3e8a75d8d9f30eb0b9ef63780798afef7b51e5ee808bf3e128f4e1a82673f149515260df26aa0f36dce4c8aab44297a02b5dbbae343b862f9464899b966104809f9efd8af1923987caa43c61378673f353000b237b6292c1b0514088e67d468515d781dd0ea59f4c5674c2f436c937007a3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634509856Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224, Name: checkpoints/4224, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4224?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=111b8510ad0c02c5772aa29e4af050c033007ccc4be82fffa6eacbf05b7b0b5dd8748e030ee186cf14bb84eedbe66acc2a5634713c86adbb32841e22db1597ad1eaa8832dee1a9f909dda52e1b30aa86a63ff74a294d6dd956a672bf320ae4a659a87a99738aeae2300788857208b0d5b052a95a0cd72f40b0cfaa8b9e6ab797cb391802d132a166c6cc5a89f822d0a143e6706c60942f62d235cd025325bbc513d79a6f0f53b94f29fc2efdbcdba9ce459e4a320ce33d5cebd361c97c5334b975939a32582f8ed70616611e477e89f06e8c684906fc05cfa7917431152f6596ac5d35c821449e2bb9c8b2ab99597f626ae380d6d358de6eafc7d47f380219b2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634779031Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d, Name: checkpoints/3264/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3264/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=87afcdc588428bf06887d20811f9616a8ba2c04653931580abab5660b32b4d5e4b90086c3c59f17b4e5d4e01660ac5847b4affaa370152eba352d5983472acab853ddb0d37ce5f07956fc2e41a4a69ee2a76efbb557a3aac91be4c681280773f1a0b60b62d2b09616ba67ac836114bf63cbea920c7ca904245dce9ee1eb34cb2e8041baa2e6cb84a1604fd61fc7ff743f0835f6e52611a95a84d2f52b55d1aad03a3983526bb448532ee2437febf90bfdb2c45e062869042c1536e8b606931922a94a342a9fa909bb7bb3b7c7ae26abe5457bc66af58c0891323c905b64be767292c1be4abe74558a3f3eedb28a2b7971cf4f6cba000b5207eec90ec50d94957&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634801137Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default, Name: checkpoints/4928/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4928/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=89eff103d3628a06ed435d5c0af7c6df3b2811df03fc1798e59d8d76d2c4c80492dc398606dd84a1b98e266ce1274003445792e1f75c3229d2d016c1e8078933cecc4465db0cf8125f96c81769fe4227d7d22b8bb50859778d342b55ba62f22f7ae5583f05de3e083d3f8e4b1b21fada946dd2967e8d622316d7db3f2cee9194109000ef4b02f12114ddc73b288f027aff8b41cec5a5b49d17d2239e0b0a8824131d57c2bf2c5676d62e6021f2a494b635ff0fb43845a685c992ce0b9527244f26e4160f012b62685dcd50b313df8149e71309ccd39f80022928bf7b8397dd0f3a4c1fc5089b33e6af9401bc9f8c6698d53b5ab657dc914ef84968e0e5aa1ab9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634805791Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default, Name: checkpoints/128/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/128/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=1ae7c5fe5934a482ca8519f39e763da38b6448c3730f3b434ee103ac22dec104bb6ee8772888a9ccfe2bb87f47af521939bb092a7e5151f43e9b2940f8cc7dbdb2ebaa6ca3ac66b1a79b2d2a526c373bb66b893d3edfd12c3cdb89c31622b8dd37a6a483e75b1663e4d084b531591f5041f8a773fa61b61528bc48d9f6eff6e0f14475e5416a3838a0d50064c8d163a28e7db5d026dc6eaabd26f720ae43b238484f70bc5ddd895143fded33c0be5b42725249ca08bafad4ce3f6addef814105f3e5aa4d4f389ffdce00658f61f627833009154558b6c0805790547d6da10fb7eea99fc1ba692bc859e301f97a83eba03bd8f99241ca1c1ef34075475662e803&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634833393Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0, Name: checkpoints/4736/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4736/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=2a89c8c5d9ca5846816fd993470f49800457e0bdf507cf8d878b329baebea2115934f3b16254c1c27394c16b23073af6ac52f8f62f14ff1ac15c7b2c921421e9c21b6b1194504ed8ec559a13643daca1ad29dd17a2da921431a2f2491b05d24d45d83ed9ed38675c89e0ff060fb09fc7954aac03defb6f02c765df224ce94fe207b71d8736f9998ae598a4356bbccfef508760e7d96c2149116bafef529d638241b2efddc32b7ecc11341533fd0e63edc61bcce0a4d4bf63fe2883d32ec5e3edc3c651fa4d509c4b9dd9ea3b5f221e06fd71e214440f47dd9758c1174795ef7cf0da004cb4f54d300fdfe1e99dea3b0936e44e95933ab7a6c752c9009b5607b7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634853878Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560, Name: checkpoints/2560, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2560?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=8b2f520bfad482bece99006397062d5f162346bad2a107096ee2dda875ed642facb613564c2a2adb150f0e9e9e6e9ccc0ed3f4c5261553e0ea4a4d933583b73a6e165637ee96ea3553fb6faadeca9222b4534545f649eaf4f133d56b1b3b694a6e66bb76e9870c0446b951d174da39524accb127a38aec9224096f97f9b145324e885eaca304325dcf3e34a4eabb6da769674f1c73b9e9169fa275fd80a3ab982e48085d6f9ed06d565ea1d20244491c4a89c71e237f00db3db18573724fa4f9f3642baa454390cdd237eff57d0419104a08cc6d399469bb1f61b1a8b1c980153a777448d95c802249ddacd4c00481d5fc1cb78ae957d23e411fd8f646289870&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634872327Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d, Name: checkpoints/4928/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4928/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=8c89c275b3188557734553e9bf8e2a6c6fd79779901e268b6a829914b85ce2f319c59953013ac17f283d90d95fafb138a7e4c47207c066186452eb6b27e9a09f52d5759b6d578d38356c94d52830743c7b1a02956122a268c13001632f4b35d4299747616624a386716e21f90c47559703f5132cfd289fcd82a2751a65648bb716d122669288325829f33dd4fea9927202f4c7868f4f1d37bda3fa7e9b61b2e6af8eb5779504c69715d68ebef072937b67e7b313a677353598140936480bdcd82311be850472ea0277f1b0e2fdb57f18c2dcc20cf08f62c25fe90ca437b2c6df2493c86c77132f617feb422781fe3232a60c83fd7086f6cb2c83232b7b2b63f8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634899519Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/array_metadatas, Name: checkpoints/5312/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5312/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a01ee96eda3501fcc31513894612ccf4da7cec6bf77ae325f11801459cad53c726cedcd9fd4a0cc6cde97b6cab62deaae197a62bac4be47ed949e6dd9b46e470a8c76a59214fa6b650ab99b4497d75a8ec97eed4cf19e39e3aca08ae74f314d228f32823c4af78cf0e57bbb0ce4334b016671198593030b83af1c54dd99decc7f17de9217ea07aa74c6a67dca864cfb7902586666899fa7e3e4f51fda04079e7925c911452914e96097fc733a52e83d5989847ce3019484ce1f3ed1291a63a0f55f194deebe71081bd9ba2725738796002d0ca588128b9b792c22b3bb282dfcea766b4cb18443caea6781e734ac54fe88fa06ef02f47c3b5d8fd7b33f77dabdc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634909132Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/d, Name: checkpoints/640/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/640/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=958ccc39adddd78860d81918a180cd8c4f9f3b408ea4ef8cf5ecd018e6c2a9d514fac4d8bda4d8f8271680c49895999dcb7a027bd054a8b850b533b019bf4b1efe76c660606395ac832fa8aa2bc0323d3800b9e6840e933086fc08be00cdf784c0e75e7adcea08e25e6062ea957c808cbb343bdc2d9e9527053dde9e0ffe23741306ce0ee78f0dcff7d4cec2db0fa7a2a4470400628f2d500fb1ff4ba6ea53f72faa80813222cd552d90bce39f41d68d99212151786f12fbf580628573139abf28249e0d624eb5f719493f9ff47acd14725c2631ecc04df13fca2319d233fb27eb116a5f8f50f413e3304a6f8d717675e74686aa3610b5ca50730eafbf5ce875&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634935066Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default, Name: checkpoints/4096/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4096/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9256cfab7027f9e1cbb9bb456e0ade4e801f802e4060f5bd0bce1b54b6b14da9e79cc173dfaa0a405c516a41ec19794c530fb7a580da4702163735b6dfcbfeecb300e8676be3f2a6a22a3ebcbf00a2a52c47909abf2249ed8d340619f5dc3fdb9aceae1e6a674563d1e2644bc99831e02d845eba99d091b3dc4ad2db3a158dce7ac9c1a220e40d60aff3ece8b3d9765dc494f4d92b5b6645fd176164b8b30e8ae0a17b17213549693ddee1ce5cf1c95805a8cf352d3e12871805368d3f6d0b3c31f041950d586e987982fa3b1bff04aa7d4d34aef31c32fca2b9f54c1a16fe50f6064fe1abd14d24cfa32341b80efb8999f2a6a391b85457cf3d8221325c52d8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634951662Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0, Name: checkpoints/448/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/448/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3ebbbe6a8f13db93bb212e1f45809c63c27cb0bb749fcf757efa48ba6bd205bef6adf436b507676b32b2a2c78617a4812c79ab8b3d66a4eb27d70872a64916a68e637774d87ea69bdb2cb912b65e4bb2501608eafbc847a29b3db1882418b2f5802b4a43321ef60fb8135b1c5116c1479393274421be0a5bc3058cc1ccffa96442329a06cb6f1c4af4706b4340328fd103bae44813538682e283040ca322c1edb434aa01f0135bb0b6dee6040347a4fccbf3eb29cb6d719fd253e433b35d02ebe021ad4f4892b5b7ec751fa93b6b779dbb7bd2e04e827e5e482ec9b8f93384bb6b17a575d3f2353895837fcd727cf02926ddf8ebb0ec55a2a4278352132fdd3b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634827802Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default, Name: checkpoints/2944/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2944/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=24e1f58ee89846449727c13cb3049c00a823017c2432d2dd14d8214715c23f4b2b5d4c8dc5d11db89e3ed701f25428733a5623dec0fbb76925d1f0282d0e17abe8517860a4c18e3eb1688a48b84c190c2f7df71f7bfc4e99de92a7d459a91fdbf383d9e936f8fc2b938c8eb0d5f35abafd82ec9e50052517b3945bb8cf8168478e1c4e5bbae8104c1364756c259912fbc2fb8a88eebf0464825305559f1190a6d68b8ad2020075e62c2df9c732026020004b61d2277ef178075deb70ed2dcc6ccd97bd1dae00a9bb1abad1971ed84079f32a7bd7d6dac8467a8ef1349fafe214e927b3eeb0648a3a9a3760f56cd399b68272e956c0b9ef630ddf0b00af0a24f3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634982622Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512, Name: checkpoints/512, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/512?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=beb370bef744f82ad7a50719bb9a78e2df74853b116cc043d13f1c66cf9f9f3c3aa7fb38136252501cc2aaa1479fe33c2dd30346caeb6bdadecaae7a074244eb8696b823cc2ffe0cb7b70efd7f03c98a7111c7dad6ac10de015e7bd15deecae1f4afa20ca388a47d4809e9446d022fae44f0a8c82fb3515bfae05ac1083b298a1c65e31fa364873260ec645b9df0c5a342c923e73db7e5eccc1f9b62c7d94971d10c07b7b756e294d17f43201aa6537fbfb4ab49b705a0c05baa0cd425932c6fd7d5a1ab2bd277c35764f8fad827d14d7590d3075622a39966b8da09de655fb4da2f06a1fd5c2f3084400fe9a8404aff273aae83ccb5ee0ece27666b42c52808&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.635013211Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0, Name: checkpoints/5504/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5504/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=004565b3eb847d95eac20476a058e01176e72e994018dea3811080135a25b643f07c1d17a73a903e37cc9292f21459d48b7fce485d885cf65895060122049cd69d16c698f077950fc244858c840c433101949c70b9dfaf0cb1d1a2c1bcf6580342ed4a253018afee3bc982b48520bb1e944a6940f9ee4cac2f57b38cc90226723722b7ad5ea5439a03de4e5164e7492f9b08a8d0035d61268f53f45dd54e8aeae11df4768b2a001418257226605fb61cc48df008779ea83c1e8fd1bc47b1b138d68aace113b37fc7f1decb2f1c6bbbafca81df19eaabb787df3b89847d71436e7b1a416cbdddecdbe7fd0d7de27267f3b4311554d8b9b598b28524fcf9df1273&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.635046437Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/d, Name: checkpoints/2496/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2496/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=02b79bf5b7b5ef0b7372efff7074476f6088b1fc4f9c655281dc0317b6c8f6b2a3d6aae9e3887c55c9ab11b67b54079da98a50a36cca41da8fc6f747ff403880b1afbdaf812edaf9ffb8fb077f708cf06e420422a56a30d79ceb3b3ff6651d0fad6595c5541cbdedbb337960789ca7329ec93cedabe5c84b993697cdb890ef00caf65757651894456f366cae1a54687ea98e4ab27756c77a327f626fff64632c763b93bd734928d3e2b707e113b234bf641fb73c44f4c3a9660c31ba2335e3bd70a44545def2e2613a8b23ef35f95d306c57c97f09efb52c52c6256c84c483d8c6359978064f80b54e2e84a40813c288f14ae55ba9fc2d3004d8c2b2fb509cd3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.635074599Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/array_metadatas, Name: checkpoints/5696/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5696/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=22a04d0dce53cc11dbf2cce6852759377d335d536c88e9316d722ab126bc5b61f02d7fb0c4d65fecc8f7e8977349340df95e476bf16447f24f73a7afbd6618cd59d9ac584d87407bdfea8ba1d50e074ac2f97cb2b46501ffc88cc3cfd1ed2ca4acbb2e5736f88413f68f111b9ce3de4bf7a4b1049d10bdae9fc58c3a2d605b294f6e015da2d419120126c9c82a7c7ccb83066f6d2b6093b1a20ac2f465898593870dbad3ad54db9388b3710e0af24aeb35808d18926f7c49efc66872a1e16d9158dc339c05007da6bacda9862af4663eee96551ec5229a75dc62eee3d87f6d4d807f59dc839bded9129865b148c7148998026d219dd24b1e664a95aac174327d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.63508097Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0, Name: checkpoints/1344/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1344/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6e6510d2d03413a6c6451e5a70e1442346be4ce5d798e164a42a7f3897fd1100d49e81e547a4a8dc09b0cc8be80a4c5f2cde9f0f322a19b0bc59e5b3d537e9ca8d6ef4b21e22d74d553a4bd85f08e0a2da071f66588fe88787653148cbc176d79158744d831bb1b195a3a5190ac6155bf212fb3443bf10bfef67d576547f9c5e046e61976a54b7c586dfe791a74c0cb4e0c415d0102fcddf55090d0ce8b985620db98daa3f149bd4c6570f9265a2e14eeaa93e01319e0948865c9becce3676068115a1f930de71d728f3e45a1d9a6b396c8e9a333fad89c52d67fe3f7e8b506ca13e2212e6669f28fd568bd62c0c0841687476c5e10929b932d3b8d50fe81209&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.635117007Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544, Name: checkpoints/4544, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4544?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6144db527c7bc634df04b3a5d3edbfde52e3186ef67e5855dc1820fbe62315b82867770ce0ca781db08b87e411a9559ff5246ade627201ef8b734a61e3b1b86779a42d4242cb03632451f3f181e43ffb931ef5a48d1600aa468b25c1d534d5e02161049143a2cbf3a4f1c362fd117e28882d88ad4515999907021d87f27e2435ab40c294c421d0ea06c962e775cb82eee1afa8c5fdd360e4d828c07e8b701a0f9e586ba230e7fea0d35be99fb0ea1aac1c68d53ee243d2b26259a62ebaba8122f84cb246e2e56b00a16c5f8fd260a9366a440e0183b96acc6201e95a2d03ec6782473698a18ac49e39108301620426cf1c5e2ef96f3896b74f7d13c75c5375e2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634677779Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0, Name: checkpoints/1600/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1600/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6052592a2b2eed26f7f37103cf1f1c6e4290aa8818e632576aa0e060323dc7603285244fe68ded814598dc19653490590c2de0c4e893bf1f84da31373c36b9ebced3e71402956875ece405fa9801ef1703683a2e14a18ce1f9067f1d949c10729d3644e83c92a63859968e1134e3f5129907cd09a35f08ec42823f6bb226e5527df306f1360b0131f8c425e441f8d1ef13cb9cc8b6548822710e6f447ac8d2a0575b87d8053d2113a60a053a4188951537026662587557badc2d44f760d96723a9a2fe89b96b4619e74690c77c3a7d7b2da1e7acdd375aa7526a680e3368a78e343446af038681a28345ffb3b1e2b2c636703bf1a27c6ccc94e490e1c90fc9aa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.63514836Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/d, Name: checkpoints/2880/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2880/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=84974bcbbd6309e2c2a12fcbf04d63953e2c9cca47f8c052e86cd8308442eaea439eebf07d2f047fc922a992235e91b9c5b95bc14df1568f1f7da27e6e40e1cad052c2019c493444d23d0515a6be4f5303b48816e822554dc1e896e0fcebc3082062fc58af8b7cb2b717237f6be6f2b371ba262689f13e8770f20484b8b952fd0825c8cbe8ac52934f1773d5f079843a011c3829e4f9ea6d750774691df4df5992157187146d5538e0b734e173e576a684d7bb3d299a0f7be82c12d14104f9468f58641a511c28ba67e3595f8d01763e9c35914e78aca9c9b6ebdbf789818244d1d602bc02a74778c8f0c9be024d05549fec36349ca3eac46e15acb9616c9fdc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.635191321Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0, Name: checkpoints/384/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/384/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0c25d040a50c7f3d9d216daf30c612650652f43feb3f63fabb9b1a5f437764ebb4734836829e2a949c1ccaa9c4e130171216a4f0ca1f68c6e7300ebb561ea626d18a1c120c3fba92ffee286bb51d4b47024895f05a975ad1d1d5cf968397e33223f80e3840e4ac7fa655d67ddb5d80544ab4f4d9a2fc2e56b4a8f4c33af7fa3d1d6e859c00e54f90546deb3d7f7a8ecffb484cce882937282d6ce3fa3383246519c558f12f88774e2fdfd46b88f82f043215bd574dc35e9e4dff18a7d3cc28b2b68dabe832cb1fbaa4176f84ab73200ae41776d3a073e535b3889b93b359ca3d7cc96c671f15bba1780f3f035b601c85d19419f7e54972b2d6bfa1efb47edf5d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634945127Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/d, Name: checkpoints/2752/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2752/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=26364323f61735821f5fe56951904f310ae2eef24681c4d040b06210c4e2e5b71e9a408910e636e3077cb0d5cd537d59ad08103dbc51c8aa245872e7ab7c650f657d86bea8f7bbbb4a57fbc3175c466d1f77f8504ec3f49ccde4c3aba4f18c72b5410c638b056b8b8258bf01148ddcc9b8dd6db29088957d290b9b8770ae871b1151dbda44f11100e2b8600a9e0a46184e198322a5b8dd8ffe036c62b9442b0a63240ebcbb59b19f0609a53eb62d6d51167bf3fd2020dcd8754cbda5bcfcfc78eb0e2d8d00a5e7d965efdcee1b9f8e79d0e6c232555d7ce20613fdb1ff833c484f81199f284f9366a5dc27d5b42cb33f046e337c9b2dfe8c7904fdcb1eff9cb0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.635253211Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584, Name: checkpoints/3584, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3584?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9e1e68e6419f7b8a1546b773f3a632e6386f7022646da293d2c158666b85c64f027c20ddd19d10b762a6c85dddeaea2341f623140ec96b2b55a4ab87818a3c150f18afc01a5e71d83b9a7ae7b2df2e052a16ca8330749e9fa54e17e253fd030e285a68dbc060491fe56bd29408d09c425b769567920e1855e706e891bcb954b2f2d2d5dcf342c84c193731d3e92f401b8276d2ded8e16bd3a634d01dcf53fb2d194eaea9a0e99cc35320b8a81727d75ff41c0039fee7f81beab900de46929a0fbbe2b6185cc31e3637dd4009e06b3b82f8f38bb9cd22bfd59ef021fbafa210ceb03bfe356be87725e0771b7c988aa6c7eb03026cd37221cf454da93b887fc88b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.635148303Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0, Name: checkpoints/1152/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1152/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=49506e2ae41c7ad55f065970c25da3feddb8836084ef60488d658dc55ccd77cdbf96045c27ce0d74fd5f2f969076c2b96f0ceef51d7577419c50aa5f45e387941f90eb68f27ecadbebe2e5216c717c95464a57ee262d3bb14f184a5d90533385cb143fa379bc83d033adce36392955681e71312883f66279562f90b0d6cff2ea21f2c062086df8384c1eade774b067f7da48251428556bc0127d4bf5d7659a0b13e083cdc27e3ecfbb3e408cb25ddb628641c00b43a612bec141d6ed43ad27ab148edbe7c1b52d05d7da60bb4a6d7bcb8bc8bfa65fe1ce2eecc2011fdab6fc4ae8321ded473144fab0ab4acfbf576c843355957dfa964256028aa70517bb3da8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636026827Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888, Name: checkpoints/5888, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5888?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=08769b702c6bcd9e27b5309e9cfeda6440ee62f0f36a03e19202aeb52a003426b232ef31cc54fb380ba371e86a80aa3c9cb22b23937482d9c92cb2f10c1a06c6f8556a03f93dc37074093f5feaa7b64f76909c9485f408ebc56192942b217e755b9cfe0399888ad912ff70f7b0ca1308414ebcd68fbc78fe74b794284598f9d856be8d301a4fbe67d93c5e814eac39df3f83b917051fbe866511a8e16c2c12d36dc1d7b3bb568df3837947060e188a39fc26197d194dcf2ef6a61a92b505d2e0e730e40c1a68536277ec66ab3f42dbcb9f6713d695322e27068d5ec601d955803b0fdb549d4eb68c088f60c84e64ce17c02d7fe72f82c52e842331ff46c59713&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634090596Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default, Name: checkpoints/1920/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1920/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=389fdfcc49653caab6b96916d596031912feffda678edd82e78fb31289c90c5742ef6941dd6f53e9084bed5b528585951d401a28169f79b82f1107312cb71938f0c08ebef58719a0974d6417c5374238ddace6263b58adbc4bb3804db38935bd671e8bb82308868227574cc95b73c8dd4ee0791dd45e64a271d380b64bb6f04c1f18d00d88964c12bde787491c80312f7cacdaf82cb64feee95803fecd6e8149a86908a3de49ba88f70bd397ebf2b940af44c8d3ad74ec9c2d692f06ce226e5821ebbd1ffd8cdbacc5889670da79e5fce6bca1ac3c1c152e92006e2b8bba53308b1acbc9a8acd6c8efe38c6f27b60c898144f95521b20264d0be733a9727ace6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636121745Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default, Name: checkpoints/2112/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2112/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9ddf20d79f6fc7e5d6a856ed6fc2e7562406e32f0b09362ff0be827a59d58d439c225403f78bcaa48f0530b69e3f4db52653ba69d40d673ceb5ec9d38ec4ac481a6603cd3b8b4b8ebe741bf6dfd491b8d43761097c27cdc49bb748d42a1f80a9c87b2c1931303a01d702f8b072e6b8121cf6890be4ac96eb63b55bc8909f8095fc80851b9cafeb3a74f79e707c433df8e4fd76f4a7def974c311aac9e1ff8256d2d15fe08cca5f7bdb1a86fd84a2f3e3d49cfa03449a3644cca7e05bb501623630bb1ec9f8eb64af535c45159a307a4e92f230860c43032830bf17721ead19f89fb6ca0f439f3a54907074f126d4f7db8df99e50dfc61f19c99cc4c20c297c24&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636172154Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/array_metadatas, Name: checkpoints/4416/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4416/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=423c22e017b09a39679839b6d7b6bf6b7c294818ef8341e7ce010bf3d6bda6967d71f0250c2afaa338a3f89c841dd6ec2afa7b28999517a498b048bd2bd02ee7f223996abca44e01fb0dbefb43ef6b8152bfba343db3d30df9ca1f3771bf5c0aa1ea68227d711cf53f3b4593d34eb7a54111c555e3078fea2413c2ea72808531650056b993822f3f1fec0a404e4ec78224b912c768214c8d3b306c42b6da3690d648c82e03d81092a62c6a8b1500133c39ff16e65ba2cfa390d00da09a6a865449d6d95d14fbbc9c4b3a580bdac030b6c542d8c475c327ef9ac7597bef7ba0a6e9c493e2c6c1a8e7c2ed7d96ab8d65c6d273b0c420e662e0b5bd1ba69b9dbe30&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636222866Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default, Name: checkpoints/2240/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2240/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=1accd93d56d4d3cbf35a27edb09a050bff79e1b20ab18f390799ad39d8da2d10fc63cb7dcf41f7d5443ffb140c40c19c608fe55f3b0fe332ce8bb1dc91946b6ab7e8441ece4369cc08012cc24f720eeb84dd4bef3894921c0287a3b4cf085cbb2bf82c60a57af3941667e3ff71ff5f25d734904423b13a3bba2500a84f3cc7a98d3ffdcac78f040d81d016dd2296110840212c5c4e5a3da88eb8fc9442acad14b1b111d1f62c7c005648bb7467e72d92058784bd2a0289be2d633956ae07b7a8efa3aa7a305f442d5ee5f420c415b28fd1bd4b1ef4049ee1059aa46bf39c082038e81f950cd236c758e1e14a19576e92be36a3b7c9f2dc40cf1fde8657da1b01&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.635070516Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0, Name: checkpoints/3584/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3584/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=793fc932051facbd6e125f6b33f68bafb86c600362e0e2613a1d0e2e2106f5af237eceaa3acce71c65acd4c3a50133396fea5b8405188eaadbac883fca6a62c27fbeba181a8fe81d44ee4365083b4ffb4d03c894a5cf9b3dcf24caeb5e538f582e6f1009ce52c59772f83101e0092f2d75e17779e0ab73490ea6620fcb9fa7832254c88ff42f7f2389018075e2df88b69b51902381a0c81c65397d6774c667d12aa8f1a8617b9c5673959136e75f5233aca3dff9df308ec513a241d58ae05baa4feb7b82498edef71c630505fdc8cafb844e61a49c6ceb05abd09d667f0983d93543bb613342f532a132813f1a2f57e2d4f588f7745385b8b72a66816c8387f7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636250875Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088, Name: checkpoints/1088, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1088?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=8f1e3a7fbdae4d5aa8ce2387c3563598a8cfb930f99392db7b8875415e81be8609994a95fd774036d4ab5ba90ad1f56b7a6460afb4f50e61ac5ea8c7480d5ee8ea8d99fa7a70bded58218bb39bc27550891213b75c2c444c4c6018b00587247000ced41ab2f78d525567a93abcdd285bb751036daf265c55b734972dc2f7e17a2da1d21b5b619698916a0f55c6b06c91756c41ef461b80fec57531086a15c92697d835ff847f767a3382c2b2a044254d0b09625ebb01498d74b097c732fe3e87ecd73c3a8259f0dd70737a57af43357f4f71f7e4d33b1a3c32bf71f1d67145e47bdfe7a1b4645ae0bd9b21c5f59616a41f6458d702c4118ecccafceabb24a702&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636282116Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0, Name: checkpoints/4416/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4416/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=bd2016deb3a94093687901f2a194982764973f217b4e9e5e9154dbd25d6e08c88537f5f6859543466b351133ac5232b637e6c934b6dd38ac5516d69764309b4a34cdf7e2d66feaa84938be49cb7c2e6f585d4e4cd26e5e4e8d9eff0455d084a91a751e42ca46c0555662645732e0f4e54523c4241337a9cb492993efe13f0212fa273e7bedc2b5e91d0d328e635a4ee1a995374646d435155a1df175ac87bea50b4b5780e9ed55b229cab398e4611d41bdd02eec60c44316b38787d11e62f3f63ec4cb6afda5aac07f2fb92a284a13eda88a9ded71c2d3251163fce515a9dbdf6cecfde8efb90a1f3a24e5e0c3a57606a568339c497b69a82dfc83529570df1a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636333077Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d, Name: checkpoints/2368/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2368/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3ed2724989fa014b03a8ae2780d8c24b1b92fbde2b2a8f6cb6b96df1343619599b63c37a26f1f90555e6e27eafa04542e2a76953ae44791532649a0fca30e931ee1de9a25fb2a09cf5ba24ec821e872adfe8fc8cdeabacad756e475d440d60639bdba76ccfe8c5298f27bd8d5f3fd5b16bebc578265c5120e239e9ce9571c2666fedfb50ebd1f9fcbcc921fd0e1733809e79de4c9233e5c06115f7fbd545dfbd37e9dfcc81309bb3387a0307b89933c4489aa1d4e78ef3872934925de219ef77ad8ef86a674263da146837a69918d4d2c7e090646e03293ca38c51fe3c5c73bfd790f6da93a0f51152ca61ce79f5f81bab0af5f15998f9446dc8eb8ebc045efb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636336728Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default, Name: checkpoints/1792/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1792/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6110d1451ccc85df1c8c8b92e70f78336b1513cb3fe2d6cb056645e3b3eabba3680c0e1ab56f8a368e35cd93828d637b67c7413d9d34f8a6a7d48b9fd3cea8c7e505050b4a8aae5c08a00cd14d1065bad2bcfc30a33a0e7dfcaf9f27367d9b9e22a1ef635419cb73f7ebc24181affaa4d535e223be12730bf6ed4989e0e162c77a53cd17dd1578ebc592c6bb4bee033a9bf7a929083681ec637b29cfc2978973aeaf0f9341041c1393ec95a6cdc6190af60ee0a7efa19efad0e2340dc7e1b6b9052cd78609078c8e9f28d619c835475270e7fd58ac2e632475da8139166e20739296671dfe8e297c7cc7ac686f02d5a824d6e3164056af38f2ee9f5a4287af1b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636367683Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0, Name: checkpoints/2304/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2304/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9da28318b6ae864677e3ae1e1031afdc3b42f1930551f5e46c666c2ffd2416411859b5dd59d4e282ad578d6d81e2140da48cc3e02d9ce276b90fccdf5244798d3c235768193ca5bc8aab74db863f0f2de241dffd8a5d6607e1bb7488e424e8c0a492d1f7728c161a67bb81436f4f1e9fcca6b402e3483f8e54a59079def7d4b5498fe9a3ad6b29213aa010ce361510ed7c75e0d1bef9fb03417f658e10596bf373f6d55fedb91571b2f8235c71b9b3c4c9ced473b6b7cdf5308d55d1e9dce99927b22ad976e7a6e6840b14824e2feb09a78919525dfc0c17a234cb7e1a846f3103d3b297af1d461200cfd36bd5f6dded359340a911e903fcb3f0d9ec73e73f6b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636385797Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/array_metadatas, Name: checkpoints/3264/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3264/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4da136a189e4c05ebeae66dc9d9cc08e84650869e524e29e68e678daf1a2fa561561dcdca906bb2e1e337d6e2d77e3b203c16dc2c0a85bc808a6750d84956c58150f36c6deba5a72849af862928ced2f7781ee9454d96d84cc9629cea4501f288821531fef7aad7a90d95b615065b2d59d60e9e395899717c834b59eddc1595efed1551fc7845928a3d1fcccc97db38436552a12bd43c94d07ae0a735cdfa92e101a091c13ec5bbf31429fa22f452faaa27847174664536b49657d83a22ffe47ab5e75c62a6933880eb63814a3a2dd973187ec4f56a238627dbe660e87fd7a3c87d27492d0d9e32e2ee76a12f805e6cbfb8e35c63900b4780a9afdceb610141a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636409603Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0, Name: checkpoints/5696/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5696/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=41bdb1c17d07295392ac40dc288d95ecff21d5591c96e328fb06f569b7910d2394dc523085492cf710a802ff5668aeb296ffb551bce728c4ead55113ac1bc6cc18a2fade4306c409b308bb7ed3b6b098814a8147a17df10c2b7c759e457c2f60b9ea0e811626528212549a6dfb3a09c1255e699323bfbd35f84d5eec68bb0db826b06b7944d262cb221801ae69087bc831e68892f6f12313c381f2028f40465b436f752cb08f5bf3110dc21a5d369ef86f1051afa7c3afcb1daa902e308ced201553c89a8e5dc4788c93459274874030f897a6254da54f2e2817c0c1496086c8800f9733529e4f0ede3ef1d83b718c4f5a420d90ff2fd7f46301e064142d3dca&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634782686Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default, Name: checkpoints/192/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/192/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=92de99e1aeaee3bc8b5273a489f96684a3aa0d1e41ef2726ad9e2161a66f2625e674af8e4c85c0689aa4f43a6c3e8414b5676ba9fefb3d19d78303ebd2f5969a2c36dfb8d5b40798931586e6c062695f9230fcf334fa0c99d2d275efc6cb65a053543576e68d6f85907824961e0fdfc26be8ea65fe360c6c902bd1c7222096809773025e4fe27298275cd28bd2dceec16f0de71ea8cbd6d22c862b6298d2aec94b7a806dae46170f0b019472c71e675971139c98e40d0b36e0f20ec5c277771681267ad59b9d7878b7cb8f5d571c3ed5fac2922eb0e9e086874d7161081cd7ad341929097e70832e16cc5cd66ea8dd78905fdef74d328634320297384fd767f4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636285342Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/array_metadatas, Name: checkpoints/1152/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1152/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=69fb735d3d43a26257ff50738949a659c4ba5d78a949431967e5b61f3e3fc42402c8db3ec098555edc943713e48cf342abd5c63893cc6a9e8dd5e511b092eecb2a5f648189d07ae237135a3bd686cfb1bc6cef97b5751a190de70509ecbb337f589b6041fbea848deffb6ce62940e5322eeabbb571c40092e24e6f454d617bfffd7761275f93b6209c70602fbdbc3a0ec0735b5ed76fb55186dc99815268c4ad5ede09805d50c7466ad30dc7af6aea30e40b08fac1b756e54d329e5b7c306ccda4980d0b8488f83f509544054b441c7152392cde579d3d01ae47e6805ab1954659fa8f763eefc9cf7a7b7fb60ebf98cca87968337925c2344581c92d44ae8dd5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636458361Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default, Name: checkpoints/1408/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1408/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=197ef5669b46b7f9b57925ba519444e9e5d2e1f1455862dd84784d7e0e847519d45604c00358cb64e7c973dd98af9a7f8e51d6bccdb59db9f53c3823ea05c245fbed6ef50ab736bb1e4b5b78dac1a160033b99b6d54399e78f2d66a13cb0f8fe2329ce5926a029b3a15a2d50b6535752ab68c9b9b9540345bab150a80223435910b85b1d3bd31513883272dbaa8ea52ff1b6a2c76eb64140cb8d0ccf5788b7adf7db068c52752256a16f0ebe8091036e221d04d47668355eb281b813c11d15972701ef5fbbb93ef45ffd7fc8f1eae15e97034550b4e9164eadd189c3f097eede4f1c71f1db201eaf5819e8e6c0ad971cfcb9b6317af959e087ee94f946302a6b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636473684Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d, Name: checkpoints/2624/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2624/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=aed694e56f5c71a18addf17e5a8997a7b7d1bfc369903c9a81b22ad98ca9f28dacd7e57827c81c441b79889ded600e32bfceed4bebf5962f9307e60d9e6c76c8280d3c99ebbf18677f709ed81c78d98671dc891dd088df371d7ba2de0b0fc6a62c56243e202843673f30073390413bc8c358f9b1d090e047186ca8150b880151d6ae36e1f685e2af3682d05aaa6385a93d4cbc38b32bd609b6aa0d5ef1c692e0499a376020a1e6a75bda7a2e54b6d7cc516dabc16b55ea6df5d8d3e07d0c78b1d3303d83b3c9566772031d57b6704323306ff4914725b313bea61ea098e9117d7eb0424970c8e5314e11b88afaa0f1a466bba9129ee177c04a6894e1524b91b1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636518518Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d, Name: checkpoints/5888/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5888/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=044da847f0b0479a7962981293b960c491f49fec21d6a629217631c4764ed224f01ff1fae836237e41b9d91c39bd7b7c7b92b3b3014f62f416916605f548828c018d1b94d215a1dee76f14433d1676424eb5561fb4040bda196eb82967d62844036b6f3f32dca924e64684e29b08a767b41525900516e4287b3451e1f56053d7f9ff7a6e217165bc74c7052ec92c35ee4a97f1dab71c6df734f1479973be9c936b5002eaa893091ee5adbf1c0516cc8936539b4cda215bca8aef0a6fb7f341b4ed18079a4ca03f80abef8ec616eda4cc090d2fc8175e537e679fdfdf6be375138eefaf3ceaa2818d065cee69bbc44505b6ab082d748933d7e8fc517528b3f593&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636227213Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d, Name: checkpoints/704/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/704/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=62566ba4b47efccb6f617d59f87c70254d0dfb63273f8d2ef310c17b6e955ff56b9df4eab5dcf25fedc8e0ef3653e01f2e66ec61da6f248aeb84ba12d2f1c1c3ecdb83586daf3ae6b4c78a2bb2759f036ab123de9c6e08e027edf2924e39d4fb94c7b575ae7743636b01a626b481c7ce61651aeb835805ca2c1878cafa2e3f2dd2ef14106b25a6bcb213e6fe05696bd48e997cb0586349e847fa20711f81b25892efc7b5fdf19c68de1e34e0623c6f8422f01567209eaadda1d486f36a259553d1f57b5343682d68a3efcd41f000ba6d4d34ea078de3fed076130da574f49e1789c894ec676e991f191b38da964c4528c59b0c9a6e911220943684eb17386b32&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636604635Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d, Name: checkpoints/1280/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1280/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=1a37d5069a969531f9c0e165b8f59e757b3accc4cea9323bd26a1e01edcb1393b1f7879baa39b51662ccece734c542dd462e565f2b98667aa07dfb0424d58fb06c4f785c6392c34cd3116c4a9ea9f4fa96d8b878da39cad6d2b702b51696bb9b09bcb4ec1b61b6a66796dc1dce47dab2ae2ac05b4a970e2ec8d0faa10b3f37b3b5a3d12544d7cc37193bb1b92eeb72c7aa8e81afa9d196feb9ce65ce007f25435eee910609b01b6e416bbf17a8923ebab172e0ff463f3db5b1644ffcbf560b1936abdfc1229b2b30aca5549fdd0850cef41f31201ed1c7b130436b14c30e5463a9fc9a070b26ecec4803d7b5ffb04479c80f5fbd30c28b7246066117d9c91979&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636539273Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/array_metadatas, Name: checkpoints/1344/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1344/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=69d581a06faee0cd402e9ad7fa8d30679b9e96008da734673e29118e7b1e981bde71a78285bbffe5be151db4dc681f93ec01597cc5a78cd0224c605726145278c0ed15bca8b2573d67f33485734c2987dbf0b82cc1a6cc03a872492585da84fb447c6e7dafc1caa661e38738cacf5ed73897768513e0e79c631aff98c772307ab2bb90b6376f416a6911957b7a276c4f4cc85ae4b7a1233a63c6b0233ecdb52634832d63f61c7056468a1486f4ae6a8e8b810bb718f01094f7a61317c77431dcbd516f1f1502e265a1a9466f1ba33d48cf7d15d259720b1e942c2742dd2c687b804be40ed2dae3bb3fcf90d2dd0d3af39a0d010478b348af84307c26ece6ac48&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636654564Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768, Name: checkpoints/768, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/768?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3dcf6c80c63e17ebf23f6445c7b1e90a63113961a4ba79008e26d58405ad0a10d37111acbd0e20fd2a9c1a07e55306979a632b04206741056b557ff45ecc7693e23a0a2a63dde19e9c8b4b59c0c2637298242b10aafc2c0491821e117d90d2fcfe569553d49f3f5b30fbf0ff727e750d77bf6a9f2d548a5a39b193b40845c7155996edc98d05a7563c5567317c9d032b04c58a2f9b15a788e5e5f26a9061adf9b9ae302b318bdd2d540f3d17de276c79e8c531b2fba69b7a5a40998fd38612f5757477c2f3fc02a19f09a51aec91b190d9190e8b24827f2727d6eae9e9bece0a70c1e03d3c8e45d43c974254956dce2a648bb7891c2da9fa3345b6c53268406f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636657789Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/d, Name: checkpoints/2560/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2560/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=c3592f5addb6d7b6b09ef33f635190d7a406cf80feaddc810212c0af606cf4f9f040dfe77fecbe97c8dfce7c58e0a9712aa35102189826583194d0d36e16b667229d9357574963dd8b112f7fa93b800c232038baaccb39431d28f83df74383ccb08b7580e8b083ac46dc75b8fd74db972731abbeed7fb053f68dc36c7517fe37159cb77c89d6d48604957dd8d6274f9457e32668a3cf88e8229990820d257d8f290608c6fb440f90bd7eaeae498f83fbeb494922f09f355543c5429a9910a9ba5a86d59987116a49609b1724e3193cbfda9f5c5d4aa4639331dd2003b0452d44c5f41834e8c631afb4e50f5e9781fc6fd23fda15a6bccf4d9b3300e0498dcd4d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636694723Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default, Name: checkpoints/448/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/448/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b63fd4cb6159a2a58ec4cce851baeea44a6bf06db76f2512668964851a859864bee6ee50838aeb528508d83c930cb4eeff8503920575a549463e982a3a0abd225eeb57d4a675f002b5227fe7aeda3865904fc8517b2f71be8acfa8cb41d2f76cce70da21be8f676d01a6f6692ac6f30260509accdc42d982034dfe079f481008e9ae315068e2e33184b21d2fc8feb61e090281ccc573ab38a08c247cbb1df1eceb1394f4ce039c75553faf6daaeca009e7820d52e89131fdc3d76411a0d42af60c9836684cf8032b1e5cd1bc400c56d9eee33a84099594dbb24f4d1720ccc3ebca454604b7b5a964492cb8934ee6bcb20c4a6a6e86551d39cbbb4fd7cfeefee0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.63647332Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default, Name: checkpoints/512/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/512/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=29c5290048905693a151a6bbda769f0f412d34ba37d85b2613df159957f371ad076e9268aec3b79fcc1142399b221a95913a5ece1795801d7445561adf762e098dae3fe3b9031738b7bf90fbc967ede2ef21ffe1f9d24904c80463bbd82d993f5c79368174c909588a6d8b764cff3c7b02d5dbf3b87ad5be521dda5e063f561af7e98e4de57a2c9faba20c6a47b9a25983ef0d211cc47703be9352ab14ae3930a9b97c6223d209151fbb5f47b6ffc96bd7fe3bc2a5f696b9db563051cecac75853fcd795bb600ba1c9910b16b33f4439ab2bf3fd17efd690472eadb2f9587bfab2bd4e7270d1864941551937be9e7db6e35fd30401db8e1f32dc6eccf3c9336f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636332335Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/array_metadatas, Name: checkpoints/3584/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3584/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a97af45c971e42671591324927f24dbb848fcf7069d3187c6046757fed379a0a84c970195fa9d91418f52cd41c9cce6f1e5162e7e879fc3a9406a44b1f2efb9bf1165204e104e041b9db8779948eab40555c0446d2c846b439cb831e77b289a9f63682d00428e318a25a7c00c5706583248c7cd23fb04b2fae84ab383e2e1af1cc0bc8febb02b1ac6db9db7833fada4beedcdaca33b40648207b40bb1073be352ac7eb0ec3e384e1abaae489b99bc5ff466161b4e1eb40537c7bedc4959c776f8571bbcf5263f605a46a46e129f9f6bde1e3f245755d09d5667dd7ebeca188d78cf2531ed6fe2b6d842431bd3995f665af740e842456dc716b501ebab24676f5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634914391Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0, Name: checkpoints/832/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/832/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=032febdac7d18236c0270db5b3dd68aeb971be2f0795e1cf0bd04c853407d8e7fc9a20c4ab1b558d66baa148ddf636b1efb39f3b16a5032a4deed36f89185d1440ba96d0955bfcf87823f4e3f47593a59fa2cce5b0917c0cd1fc43760e60e479dfff556819cd507cbadfe6562da3a1d765453dffdb34fed18c19b2b778a1d21afb92bb5dd472dffefb90530e8a2ebaf72cbf80fdfe0b2951f5e116331a6edc4b4cd0df45316b134394e66e262a5ccb2a475bffeec861b11039fc21ca0e57f6f1f4758684d6d9b01af0dd74ca99542865dafe298ac7bea6dec2c3a0221249c3697677f17549db643107a9f63c42e709ef38923f199a4f0e48ee27560ead98f243&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636463424Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0, Name: checkpoints/3776/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3776/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=532bf236ec8cadeaa238ac3852d2e2c02c42fea99a0419db04e1190ff19dda01d47dc3398e44f8a604433015e1bdaf88ea92bccf03ced7b42f6244adb946d5a72dc4b126bb01068a5d5235ab2662aaf676c4a5cd432ce107adfd1c4bd05c692af28ae4fd421a932bee2b1342d1104d7ad599b61de3e8d0ad47dea491f5ca3263525ccd11342156b1200e570bdb063c1757f53a60977f43935ab5a6396a69f1b40cb11b5dd6c91183b60d9c7c8f8940994b9492021a3e831281218e0ca426d1cf11afb0e07ab702cce1fb8c1972df0e4a55e3630132bffd195f1869f250b39ff6ef3a5171b2ba9badcaff8b60264d8b65b305dde30944c78e778f1b770a3fcb78&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636833185Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0, Name: checkpoints/5312/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5312/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=8ef59b178f4c2c3d0acf1855a4785763092ca577fe287a1483c0d696f4865983538e1b91bf0c4a33d453d9a811c98431bc6a1920665829c35dc1d1f9784c2d914f848d5f1fc6a207f72ddb0c66f7483d0e6b40a420467e2ecf1ee3afa0592f8aabe3e0b29a12d62b871b5f89af5ff04879f6093d5be608244ecfc54effdf262f2db36d60fa8c502dc6d68656d65ece384ed1e7b3a5571c90fd8cd42adddd2335c58f07db5a5e49849809afec24bf977f6e9b9019208ee10634247afc8da26ca33dd13614388d7d9c197aadf1c33b9f19322f514beb0c5e1a30cacd1fdd27567255d0f3bb3869d713748c5803c0242dc8ee47f1aac958f21df61599b9b3609327&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636127167Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/d, Name: checkpoints/5120/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5120/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=46d3eefbaa565e47345521e823fa4845b9afc67061bf188d54f6c17dbe15a4bf5dfb88cfb2b25fb8c8d70c90c0b984ef67f9c82315bf03ba6801606e991e547ac788b2c1f38ae31984f9c545370cf22caf73744957bcf86d5c2f1aea5d936fe740bdf773d0737ef02dc80fcdfcdc4ba53f11daf5119bb46ee16d5b775892f97260ad77ea3694ba751757652dc181fbc191d1bc646065ccef03f0c3400efd6f0dbf8d7da33ea5f4e051512e5be9a1dca73e6c2f540f957e870808fa47d186236bdf4539b488f2798e9dc12cf21c9ef7f29fd53b3ea34091041af458bf197b2723d11f29d5352ff3d02a9d9251f6c6bfb0539cc0561c7a29e0feb7368a75c71081&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636861908Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default, Name: checkpoints/3584/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3584/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=876d686a9e9f0daaa0ea390f7ffeb3b27723a2435362f84bc94d991714c511f9d3cd59f8e276b08fd73a398e8722ae8e7533c52efa6be7a1dbfc694248cfefc5e34ecd0caa539b640bc41f90703ddd6f1b568d29854665a8e5d3bef6bce814a4d80aa6a452553a66fb373b2d3c8c38cda375a77a0bf32db0c9f23eddf18b14138617fdccee5f4a7ca2408e56660be2a2e40b93d7a489843e14b3d3b08a3d1e0dfaa86eead116a4bbfda496a46a601f5b389be3d53f31105a5bef2f4a318ddb2e69e22b605fba26cea15372b660cd8dca97883a1fb1a32cf6d669bf0cce9eb79bd7031a612484d21e760769a552ea2da7196a56eacea21e693445e7f095823295&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636665837Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/array_metadatas, Name: checkpoints/640/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/640/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6db474b552df2d05ef1610ddc72a453097c0b0f50e780ec575b825b52b7a1fa6d08ac2b5870a06130551dc0aafcf59ba4460b76dbd2683e117b42093de6c20caf60df7ceebd9430290a755fc5806bb4fbb5febdcf7b330e52fdee54d02526f618d98d6bd9b98095a1089d1d67fc78756d35b551a6d046196ab60f39af8df0e9695cb96255117772144cf283eb458eccb56e8fbde435d5b594d1fd0945d010536e1febbbc131e58a04dbed9fc9db867e7e26f6fbdeb1601005f892481f59c7415cf674208f65ea2497109f6fa9504e7c58709fe1602d788358c2156f290f1ce9beddeb96dae84b9f1728ef83f9d8301539ef299072d46d5aefc522eabb50f54f1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636921152Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696, Name: checkpoints/5696, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5696?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b51cd84d2d1092761f1e780f66665b46d2565069164f38cabfe42e70a8372fcd31e4613ed0b5a4b095b207909108d9fc4495a3c43cb1cd9b6483dc1e960caa8276e1f5f272fb364e6d115b0e3e6c6ff49397fa9a2d8d47eb8bb884b683f718916ec108d2e3c1dd81d964227e70d566a1682e34b9dec740bdb04529c30b3e75815c74e86c83d06f6ac327953c64112ede4baaae358e7cf27bd376fd88ea82316c26dd9290e6f4ed2aa47e568b5614feb0e36d5f6f6f7593a3b77bf5ea96ceb1ff75d533f3521d56729573974b9a105e906e3cd7b4cc68e16aaa0c77c7648ca3bd312a714904c50fbe498755564ee5a269b206a04ce4bb8fd1b2d5b0a62bec875d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.63447122Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/d, Name: checkpoints/192/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/192/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b4533b7ab39ad70aa896e0662dfce46c7a2699d2a108a2dd29e1c663a118b030b62175f3e9eed74c278d09e268efdf76c587d8bd518f14beb6e3cb9985815e7cd5294348ad93ef12a870e94edacd16051dd03a0c691d6010579eb8bbf755f400d2e6ef75e1c6426415b036c6135abb330c49bc05b54ec4a177e1a42f270f467fd808a14b7bf6708ad9b237a7e3254c952adff974c23f2ba67ccb86ae895548c4405d6bed717123c3526d373e89ed7a2a1b8a74be2e3856a6c2ba8c967091333e20ce93eee21099e80d1e563472dfb93886aa7d42cce25818ef5b7036feaf264a40ef013f0f84a25756751afa49df082131f8b1e1cb2e4b974a9e3c28a0b55170&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634455769Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/array_metadatas, Name: checkpoints/448/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/448/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6da418ff9075eb5e9e0e722b2585992391a26210cd4ce4777c7660df55ba5d07d62e1f54813fdd27053ddfe31288c6690dbd5b962bfde461815f4c8e5a96061575c9307ecc381e11ed044a484ddfdf61c2e4d82e4112090cd003cccb00d111487cb931b727ef06588048b5eb6b3ec602a719dd0597afe68654e1b3ce7465024f692e4335dcb85729141ccc53649573fe2ba1f4336033900106ffcfd5c047e2f47e09e2c1b85ca0e8345119d6dc4b48e43b01a20ad613ac062bb21c3c2e7bd773d620f0d38fc4dfd75c0f9a54767ce567b646be2f55103d6ec48ab2ab2601c0897f22005d089bb607919d514d4824b0f54a754e557488fa1b0b5df78c9ac4c477&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.63479698Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default, Name: checkpoints/1472/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1472/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=46d9f4b704c43081e94d9863fb706b2ecab5003ab1f51bfa0549f13466f26ce846d8d395353aa82928b75d56f6a222f0ece3778c1afa0619f6e8e2ba478d777cc9214b82c04712ffc8553eb0e34cbf7382e56d6a9387e3680e4e024b93ecbb73959987a4a010a02f0cec43644e19dcf88a2f298a0e2e02a2142ee0311e7636af07a613c05649b420bc45341da196ee1fab7be7103a8f93bc3d9ea4edc940d0279e9093c2ecd2271aa93b1dddde2d0d3fbcb4b8f0a97d00763344b3b053cb66364e2139f2022455a1dadef77f5514a9aec3ecf9ac02a089f17070ae391b50caac7458a377d402df8c006d9a22b1829aa56bcecfcfc5b0d18a055c5963ee4f1886&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634810951Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/d, Name: checkpoints/3264/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3264/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=2a8f612325f30d3da5fb4527d7ecf665b46aa21664536f682910eae040c019c2f9e6791a620c8cea0a39f9a12385848972dea54c6eca25db226120ca6330390a798164a718ea558238865564aaadb7e0e690772f40fa6a078ed1894fe92fb16dac09d0590398451f99655499a01e66e55053af15bd73388a50ff84c45882f05edd446af003f0dc194ee3b5ccef10bd06c8ffea50591054218fe426ec767d7214a2348281028c6de645fca4ccdfba69b561197606b88c1c3694fca161e4249a0a884e09e27599913c9fd53cf1181324b1041a8421254d8fb4f1a259ef353e286f36820871ab7fea91812449847dbaaee4601fa5b3dc295c82cbd222c29fb33b6e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636959859Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/d, Name: checkpoints/4416/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4416/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9ddb52560db3e18e18c8ca3fc975d3cc5a7103ed26a316184b3e927096dbdf3167abf507f730865f6f2d539b962798bb3207ed6149d78eae0f902f281bae37b88ec2aaf6191f9a43cac44283ac966a447b1c5096a2b5649b07270cb355a361094919bc4823e296f8774facee4d8c7166ae3c90261d73ee3bd38637b4cbc9fbe0ac1e7275c007aefef9205c6346368f1d177180b72948d9a9243b7ba48f2b84bfe49fa2ddc4eb15ad92e9ce4d93d69d53537382b22966f0f766bdf653a60295b216b63db6851730aa1fa86a14bb463a7da0054ad4cad958652de72c397abb433fbcbe0a22b7e0a1fb374d5a98c4f666b03f98007459f120eef856b08a3559ed78&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637018166Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default, Name: checkpoints/384/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/384/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=c078fbff59bb8f6d8d1d1f1eaa270181813abda4757a852c89fab0d2a3675014ad02af7aa83cdd67e855e59ebde30f35a7c3da7e6874a890d090b6fd1a2fe42718940b1f32cdb5b75f2b479f991ecbce5df73e483d53c92d99d1f1641fd9c884bb7554af805b897e2bb73e1133d7707b62b297668cdf1c6c3240e133d6dd734f06ece2df07ce937304d4e24c622232b629c390454956b7ddd4c1ffb38a7ee6f3ee4e738ba29f9ee427e4cc3fb3aefb9f397707ba636ec989b0b92ef69cd18383a7342e70c6d5986525671eff7a14053e2050bada195d407caf131bcbc06d3e0ed1462c72a8d5f0a1df745bb6ad1b15be573a072f2e0fe7fb6c20777ffaa42399&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.63485766Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default, Name: checkpoints/1280/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1280/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=86ab6d7032fd8c072a329621b3934edebe0293379b5f3d559a4e252d9b79796f7687dc67a74cea412d38a044f2c78bc403e1e68f2937d3427c228cda13ccfaa2268246d4376d4379b2c2fa8a94fe37472599c2b9fbd0b7fe9874f8b9a3ac07dd59e704e5fdfc0d7dbeacd6cc220ff12fd82d245a27bf11672b349e7cf520193cbf7354661f80c5e032d6906bd538385f7da00ef7d04ae00db8b4b96fa9dd0e7fc23d5fe0c2fae510a7ad06cfdb54a732de95321b550d455fd99b422e1e9954a0b874f5fe19fed1646c947aea15add032fcad6ec4e5ca45656efe668f242bb60aba173e51396d7468aad1650ae65fa8096bf3740a5a36a525f2e96d8b97c3c0c6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636981232Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default, Name: checkpoints/3264/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3264/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5ea9eb400273fec184dc9ca24d9db93e6fddceeb72100f1332c7a2a709074f9ed38022e4b5b5f1201eb81b6d8936be8d512460ccdcdc6ff058b632a737ad21a268ff82288e48406415fdd87570bc6d0ed94a5e9e01dbd7fa315c99590bd747ff4ba3193aa126188b55fbe4ac79fe2685842447c81ecc5b2b788e853ddc2b904a82b4a135298e6b16de9def3801079ca149e53b6cf74a0f651a828777c05ae3cf6cefc6baee10c7f64b41dc349f7a46f3341121d472fa6d96f9e604df06a2d35e315de9de89491113f424e908d320327876204c622dfacea10ba1bf49aeeb2ba135eeeb5a4b76429accc1b1264031a955d55548f0a2c46e9ac6555f09e9e2a2ee&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637065659Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/array_metadatas, Name: checkpoints/768/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/768/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=bfd12b74e166cbcac35ac12b6488220ea8c79ad7f1152845cfb72058c6d46cd20d1d0dc52737861f93f6b7d11e11083d8789e7f23e26b30ed9327640672cccd546d3c9b3c079c1e5411627846bf72d10dd0855285e14d367b3c849fa92d572efd57797a9b8ea27ffafb5a64d5f8abab261f502646f2851b9e9ce5fb9369736953f9d501ce49c7846a937d1ad590bea885993682af7f32e767b3e70e043f2338057cde48c115f11436859e457fad608b314e6e21989b103722f2b46221e8a8fe12697b55266df77cf9fb8f4a6ce7d7c018699069df977f2398d3b33473629e36872dfd5ee0e95cf1d49282cf430a8a964b0397dfb2de48a28084a965b805e0a28&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637123063Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/d, Name: checkpoints/6016/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/6016/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5a3cc64dabd87070c86d63e5e3833940c6683b738afe7d0931de7c8bb6eecea4235e5eb069e0a84ee0a6f6ee953113d1b251773ea9e5170c0aedc49bd2f76430a043cc7fe168636dac147789289f947e56108f6ec7c48021ce7a80dcd77096deea4421c43bb0f56792c98745a9c4dae1b43093a41bfd07623c1b176595f13ccbeb219b841cf54e931712b10bb66fe9c6a9267a049bf3df4a01df8c502276d6bc48227d6e43966a174fc0e091984eb0f9c2afeb7d29a1ee19d1a15bfaf974d39c118c673118919c39b94d94edc62de4845d4b5f6af30e7091394e60268b94d738e36d7f62de600ba949847a2580b2dbaf9b9c33ba58f26c59e658af12e580c644&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634545533Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0, Name: checkpoints/1024/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1024/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=1b1da7c573f65738e691bef74ecfb85a73279a19b582f842b7f236dea9d76886f31f9c49d61aecbbc781766a8abbd5d8020d92c0ef81a3b7b74010d1df40c6cba8950d21176610f0e331242f6fca33fe2e28daf7263ee78ce6f4de8beadbe47f915d7ef532b8bde2491cc866ae9c0222a17f7ed35f8ce56d603e6c881868a046e7bbc16c32cf5d935be860e9080a128b63b1ecc28ade336eed2989185a9e3cce7847920c2aa4cc7446ca19801cd537c09298c124eb5bc88d26f6c22fdc52c46648a48451d6ead08296cf167faf53148cf660ba69711d59d989d1526500369d6090681ca2fde27e5094ac0210ce5c3d53487ec1bdaa1525132bd001175515278a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.63719328Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0, Name: checkpoints/4864/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4864/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9785cb41c293553cb88a0037b573752fd40087753eae6edc12a914ffee5e6f2aed4b13ee4ab057afe4034343b5abeeec889806b6c8f8a05dd26686610f35f7195fce14becab5fa5ca60583ff12c12e5cc71e997ff165016f98e9faec79588dc73147cc31e902f5457c6d7f0bb63effced8d39da42f3469986db6f169f58cc7ece55ba293a78cff3ff90c27ed0f8dd9f9332c79255582f951d45e6a9a0416155e440f201486d084e4495d0ee507a1876bd4b0b60578e68a11d725a52fc3d218ab6a20d0308b0c60d19791e9277c82e0918b998700b28a5377c33c0c9150d022395bde7da7a1e0e75d5404f9d8652a49526c7144763017fc74e3b9b14eba34ea8d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.63722787Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default, Name: checkpoints/3136/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3136/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6a2d77f1951413732985afc3e4c990763181e24383d7d39295df2714286c482901acb1369cc02a37020fc23a65253ce5afe3638eaad81ccd9335b0c201a886600d625120d533b5caec5aebc55cdd4dc9da3d9698e67e83167734088de31ecad794a2326bd78add9036c05e2ba22c6028671ebf4ad8ca795501e735b836c9bbc4eda53b0f74e9449084e9ac625297179f326031f58f341f349693f82b020f84740246af069c26a8427a234fa980505511440e3e2ba22e055544343ad0493a06a6a7f1afd138fe605197fdf6cff962742cc4750e03e761d46ee632039fec8a8b7ea9cfc6fa57afb0167efc32a5fe54d8ecb7ec649340fef90f4a7ff16f748ebea8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637035938Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d, Name: checkpoints/4736/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4736/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=2c0ef55702998e6e6104dc9797f21f84134ba11ecacda2cacf6b0411628a2ad12eb5c6a30db986897b0b8e0c4e5e715c5e68f364a8aa56253e355d0d489c8a11b3372e38636b4bd429fc0106328bf9c860b7e6ffd9ed2284260e3b279fed83e310bbfc52f09ef946763c7b15c460d6e810ca5b1fda5f93ebb382d530c35599baeedfc7cc944430a1ddc11fa8067ca730ef4f8543faf6e76cedba73501f69a8ba6b5c402b978a906e81edc2278f64137808cb44227ee0c27daaa7cf0882875c1eba7f542f255cf071099957d76fbe1fdf50b9067a184d396235ceafff0127c407c2d1bb908076b4996396603ee4bc121d67e44b34b5e5b0fae0db2c177b1d7da1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.63729086Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0, Name: checkpoints/4224/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4224/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=19e4c114dd62594892e1b65b37e83b6ee2ad411ad3d273f62f718c9dad083ebed0c337ac0cde51384b6292230fd046ee20edc31dc9b73707a78fbabfbbc8aaeadd34df908a6b3aadf9f328cbaf3ae7bcc1b26e3f94c8dc2e518f5d592f14df4bf2e419c36ab54ea210bd450be637f74e3d9978a70f5ce3f22522645e5de72bf99335b3c02e2d1ab823e81d24504138b673c09a396420295bcf10c100cdf3b6eea833659395c24f1930e0ed5281a6a48c6acccdaa73b3b5434e87910902e0aa65abb96b00cae6df0ef14c5a1b67095bd4e77a999d9f954cfb7dd8274c3c854008a5b7f65a73e5b4838adbe54dfe2b6fcfec83b2ec08ce4edd0aee4936f18cd1e3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637329503Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default, Name: checkpoints/576/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/576/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b5261b80b97037806b603f598f087bfc8469f52af6545e573077df1c070d88dd5d21c92ca0b276f495a43c84dbb3c3266fb20d9df16f3f265a69d5d076226fa6d55ee2c278126078af294cb2c90447cb2334cd575c350322d9d0c88a726783b43ee4881564ebbb8e9183b0afa325389b1b2f04b468dc238108b2274de2fb6ba4e40077b35f9e8cd7d8efdb2250d74caf78f681826bd3f05be7e778773280df0819a1a65671331de8a8bd779bccfe6ac10bf45add7828ff21641a935dd8f689076567d18612d94b2e1b5f3cbc65787fa7bfdd8a39d6d8c0e643e97e51e82755ab706bba1033e82bbc3098c73f15b65100f90bda29074967d6e817e6464cc25461&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637317117Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default, Name: checkpoints/2816/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2816/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=457d737efe330d95ee960741a37ef045df36a0f569a207a6e233e62c8d50cd04a327735775e3d4694a2c7f8193d6628c5c827dbf26401a36fbd5cd99fc510fb2ca247f4df7458a4669a9074edc71e5ea9427a89882b4704cb3afe5e1d97a6287f314bc62dc376b2f767789f9c2c4379d944fc966a440775814e5beb898dd143dd0946252ab890e71f25d091e2b12fe5e9a62d0ec2352130c23c451cea66bb08f53b3311209f50d0a6413d527b84a7e307267aef7aee780588807b16371f7b9aa391b15bd820581278747eb4aa8c2b7d368e1560d4bf2369390603563c0fc2416eef575a051c2c3246e3d4564feeca10d15f2b22a1da3d0ed93950b7404e48e80&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637413486Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default, Name: checkpoints/6016/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/6016/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6596193d683bd87c772a24a915978b34af107848acca923ebc76db0ca5a3958c940db1cc1f4caf22d857bfa056737d801aa82821d28eee1b167279eecb0ca99e4faef9af2211a15f2b546956535649022f3564b4593d0ba29b6dde589346e299761b29c149a972d717ad049a994d9bec475b8eee95dbdc00ff5fd68dc9130eabd0ed97ed1fb25687bc8acf368110780ef607ebf8427401ab79a355b8d09162f78c2d61adae977d06e287da947302db37b7e4e2616145970e454c1e3d9abb507cb29cb97a166e3b27089ba12ff4a3e6c7a0c88148f0ae61f0894212b8b60d456fd8ca2adee5fa3a190d20e6e582f7f78659e4a27c4d50a4bf4d517d0b6eb747c5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634722325Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/array_metadatas, Name: checkpoints/3520/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3520/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=08af9a99b5594688bc9bd7905c058690adde8bce2b2d0662e334682f804b47344c0e17b09da7de6b123a6e5b9d2aba3789147723e3406d8d5387f4d3a3fd93f170280f3dabafd0d823448f9c0e78429f6fb9174da8a9e3bec95697c8c3d17273a54c3c2e119bc1f7db636a874acf218397db8f6699ec0e0b50da3258e162dc923619c9b6493ba55e84994ac5baa5c1e1bfabc7dea1824254441397add5535943301bcdec076e2bd35860ab8baacb2f3bafe4d066b461c833ac2c145ac61f004cd487d1c93119adeecbbecafde33cce13cf03970834d3509e79cc8670cb18c2980556c315b1ac34ded7caae2ea1dac13f2640c34bfc5072c6a72fb0b4fb64ae21&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634543704Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320, Name: checkpoints/320, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/320?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4fcc44ebb4c22a915c4367f12837c27bb8c8dd353872c11a06a07ad06bbb6803533fc30867b335dec6ccbcb287e74dad693bcdb0ae38a56465237af0109e6a548e23a00ca2ccb7f2bc14800c371db56d0205abc2814a984721ae7de66035628579a67bcf125eb0db6c72a8735582e13374993e11410e8bd0bb5f91c31c9173597b0ca89f4beb596efdf03f9e655fcc664ea90af348eae0a06bf05abe8a83f9e073a6ede23e85b1068d0f0af26c478f3e9162a1e6b50aad38bb0c64da9dcb75ac6a7e65df178e328bbc5cd38649c7953144f2714770849c7e4b1b568228a5d7379cb9d7310a211d4772381a28a19b90ae077eaecc24aba5c14636851d96eceb6d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637145254Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0, Name: checkpoints/3712/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3712/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=54426037672d60ac05e94d61b713935de3219ba9a006eda6acf567acc7ce222ecd9b15919fa588ba4f4876c282eac7646fca974189a1cc9a0f8991f955c177c0e41d01c1ed6fb39f0553ee6ae079ba5b16614e7581fe5670c9dacd38e584e9090fa9f4ada929113966f3de08bf38be6385988ff63b531b2240b21273e87a686b89e9dfd0354c1bc75651cc6645f152edd5ac0cc2947b72d0c2462df1d1920405527b95ed490056432286064381f0e55f077fd4b7f8e8085e1df920dc2992223a51c43ac00c80488ba90f61e81ca07321a30a83f368ba66a6e33b7630837ad57d75b0c7844b64c6131a74b8d188953feff738fca656966fedd6acf94418f51973&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637515283Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0, Name: checkpoints/3520/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3520/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4f95a0456defda5d910c1ad32dcce6eb54939b95b3f164be0f4cb01f56aa7ccd31693d46f2c39cce0ce32e081fff94a2f168ea764cfe3d44fa1370be1b949daf81262cd475a3977f443be4904215c07854fb4964eb6603b5de36683d53c21d509993eb5be3179d688d42cb4252f321017791b8040e0aef0dd4d10d26ec17298bee062322bf9bb28590cddc5c19c0892cd987144ca9af2d9a216b0a07fcd2e54ecfa960b68d02f0bf7c132b1ef23e8501f1a6e19ac73c6cd6973b6d918e1efcd6149188f2e3b7037f0f4449ad932d7f5982726cc33f444c96307fdd4ab14f973ba7a312ebcc6a333a3d1c8d59521ad7f0662979833a5c5576b02e3e17d112b5bf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637563331Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d, Name: checkpoints/5248/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5248/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=aa5f7475d7a3cdaf676e86576a4fc1a0a20005adbe0a9e7d48325262ba10e8b6faedd0ad02309e261ef9b1c1b00a7bcfd682d0369370b6d35a8b11433290979bab32fbb62cb746d101bb5f869218f1da4fb9c9c0f6d5030f7b73650b944c8244ca9f1a9c0808375aa6e03d7a0b07e3e07816cb1fa9b35185d2b310908a35ee59514af2e6cf185f78725cef3c61074ab585840a552583978f3337ec1f6185310e87002b63635ea39916daab2a8b3a113e89e4aa8f1e0f20fad40c0137fd7e03bbc8b72960397a82a85a067b4424967170ad7817bd9fda89a6b49c7ee7620cee4ad50b2d125ccf688aac482c69ce6189c8f38c1acf28f0c329d9247b02ad49bbb7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637556801Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/array_metadatas, Name: checkpoints/128/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/128/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=19236fabfe5dad68f5d4496735a03c9bbce48dcc245472e8be525f6bbfb097fbb531d9a1b573e039ea3770359aad524d1d0908c74bd9bf72582711f3ebd8727d9c809dbaca5e7e9c4640ce32488a9e9c4d1645658b2e2c5b950a1142a181907f1bba6d15dc88c2cd6a0a7e395daf183f45a3bab970afc82f0a92bd0c8c057fe3101e502751c4c5cd58020bf571568e112a559acdf8dafb512f9285db3b5c376cfb5330bb83c6020bdce4c7a362e75030d937cd223a9e0cf19ad3ee768ea24414f1d2047ec28b7725adb9c49eccbeb051d898159968bf0c78a417a2ac13399b1e674589133d1b196fa434b551c7b825fafcb5263e282b45d29b1b8168422720aa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634100005Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d, Name: checkpoints/2688/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2688/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0406bcbc925ee03bd3068be462fb9dae842f4dce1faff102e43ca065934de532d5834ed5b3e4d219033280f91cb7651bfbe04a14f6abc25116a516b590e57182de122fa340fb2772805a4618a0590a3cbf2d9346053563faede6f3ec73e4ad02649dcf18c627a98bd4c4ea5b8b40b60177c306115242140b81572df76d424679ebea4969eb4e044e52f064bc6eb45fdaac83baec0596f717db1c88654efb8ea5f2baa39dfbb90bbd50eee1b61ea0edcb853d8afe0ead0d4439fb5a5fe6393d22a4a3d6609e14a53002a54b13aac3700e36d170d2f6a4a5750f6817975f69ef44b454357c37d9906f87d0a77c183c53c18b2d4e870f95cd8231ece325a616f6c9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637616098Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0, Name: checkpoints/2496/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2496/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=490a802aeb667636b01376a351b0045453f75cf9e9c8cc1e0d58ac94a5a4889df2806e79677d5c65fdef4021ea9c62cffe8369fd6cecb7f2ebbbd0bbb91203198a6204c145a212a5789707246ba7bdeb1f5ab247226a4ef1ead023f301881290cbf392cb2d174e19b0c5e3849c066f4063e045fc7368840307061efb206386e8db3bbcd9a18825de224afabafd0d494e973497ba80cac4b5d7836d6620f0737d524cced1bfccc1a1db8bd9048327e4971272452fcb5575beac485f57d40e7a36709b30ab2ccdb013673d1e9ce0715e34532f959e738c903163065bd858ba488eba37df4168fee02d1414c10e4f50c6902431cdb861aa6e9b9f0d5994ee204926&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637661275Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d, Name: checkpoints/4416/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4416/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5c5611cf4804c8df2461b809cc27cc44db3e8a80bf872da8c1840fcce37fc2e0c18f5e9d0070ef257c851b694e90a02d87150ec345d5a31049eca804e8edad59a7d12ab527d818dbacf894851d8276e5c6851a54afb2741d385d1c6be452b215866a4c78b35285ef38e861cf1fe0cae7995f9a28d4a7232d5a002420482dff086a9f8c612556bd96a36863ba13d40fc3f93c1987a631a00ee1add421e531059a60ef982d5b01ea0738a2f1f6e9b92e0d669c7758bbdcaf7fcc8649a348e5b017477d4a3a9ad79f5829daf3975690c3321f313ac4c1ef6c97ff833fbaab57fa7027544932d5c129e065e143ed62b215910e748fabc70a33b1bef70bef09ed11bc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.63443128Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default, Name: checkpoints/5120/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5120/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=2ca1a318d58e4f555773f14d2b531113ed66d5b0dfd93c331240420ec47daf499ba26b55d8fe1173714c76bff751650c30bab9031c141128c6bf73ff296a8fd6de33b6185e3c58af236d05453ec0355974bce0739176050d46254f46c72a57aaf5d15622acaab8f9a0fa24bb34b8d2973b7aa5a4deced2afe89705f6c99ecac8f0efabc1f9b41a1d5053e02346d15a07b651b0c3a2f3f742b07a4fd77b1486073311f1802208123bb0b6a837be06f784869e8fcd2c52cf4114b2779ed891d0958afe1d4f1800b3331b35b7c802a08bcc9ddf1561ae9862bc164783af277252b2d8ad24abb9cdbf5abf1c722e10f77aaecb9794a02c1c4948510db0afe8559f02&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637729285Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default, Name: checkpoints/320/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/320/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6e52157230736d57f91d65c5bf457763fe05d06acf0b76c69624b89e20cf248b31ae6f51902b7d5c1c2382497178a33c959323485cf4c87a89a65a79e128192e90c279062b4878a4342c3c35aaa8b1ea25750b3762f92ceb571044fedcae8ed505ab98a79a95bddd74547220341ae27ea33753e967b39af66fc49361db4730e4024ac3dc29c7667335059ac2598ec073f444ab188ae69ac853bd7b7a9eceda451ed07a634760bb25972ec25735504c49782e9272f47bbdd0033a8c018904b1cabc1610b85f33ccfae4afffec8a396f4d9e0d4160c6b1061206af684012fb4f4803b8335274dec1d06dbe66b307611480610b68bd4631682c477ff1c9f910100b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638052145Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default, Name: checkpoints/1536/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1536/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=1c6c4558a12e556cb58b49cadcb1bdebeb54a9b87c4dbfbca1d1ba9f576ea9c70a2d33f02f5840c6ad7f79c33f921015639acc318a44a879532d88c1eaa4c822ec26d463722d36e4bcf2656218687aedb76d2cf04d3fc7fe729d616c9f5e5a6f2281854cb27965a3e58b1a501294605fef24b4801095ba49baaef3604aac09fe61ff2f4e50b02a7bd14cb5b1779a8c834e8e623fb7a9027a2ace725a9dc7d5e2b64621fa2eab331dbe949c5ea2d1d21f6298ee819db3d2b3b4fc5a8d1649fc9e731007843784b8666181586b22819ea659fea0de02cb50db7687aeb2ffe4e42190cf1df3e1d14b0eca4a9a2bcbbaf9c6cab93dc65d7435754411f29317d3346b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.63804653Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d, Name: checkpoints/1472/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1472/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=777f32efbaf7585975973ab331add83edec0d11ef2beaff551a4c97d22d5efb8fc701701b8aca0228575990c99d8f46770819eb3450b76e3d37c083e62e751af5564acd559b320c404340551224c05c593e56b7fe249684e687ce8ff95806eca765a3dd25eed75cbf24520280ee54f86d732acfaacd8c2bc955f3bd3e5e3272f993a3c8d83f6b769da2bac937767ba528c182d4ba9b1ab4c148755ded09883bbafc3ad278b50902ec970e294e18856a1be4f41688a28abc54f944afe97d1eb580338e6f8e01a7bf18391c633e4c4efd870e6511aa3da1fff7d2a0bc3ccf6a2c1ee634fffe25536d2982d0862be8557230121a32ca8f4f9cfb72d06a8d76b36aa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637282878Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/array_metadatas, Name: checkpoints/4736/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4736/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=bfc0a19383dd257c88c2f528ac308c6b912eb8bd77f5db217141237e2dd700c01a3e05b2a70f7203e19eb2ecd8ef796f877fe5a5ea9e2326c4e7b13e4ae32d0024f047b1a8d6420d1e7c712e49534a8c72a42148b589055a6a18a4fca4713599fed207b182f000ed8a7a5321f2269acbf17f26f134200a05ff778364df8d57c3484bf5cfeba7c0f82676ed08474f5aef1df5bae416fee1cf7f95cf3b697dbc98d7f6ef5180604b0d6986cd5911decb47b251e8b054cf37eb2744c0a0a885d7b16eee722f4d2dc7610c5dda8dcb0ab591ff34b0e613f9cbf1e0cde75ad78a02dec8c08041cc40c733c497ea3cd38fc50508adb5ee19fbaef0d1eaa858e526e026&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638097835Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/d, Name: checkpoints/3200/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3200/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=592228ad67b2e8eafc024343e02aaee9076656d9ec1bb7c12e20c867f63fab1fd2e09f8dc9209410b49dd55c2993cd23478c0a56f1c000666c4282b8c366951481a7100c31a77b91948869826fbc4b5980529878c05456e80b5e892f675bf29ec4e6777e4aa8706b229a7dbcf90726ecbb06b59bdc5d58425d71817d7f57f538efe80aed5f4ccd8a6b0a8fbfff9c298c3120cf26f1e7b5d0c6cc10e0cd89bcbe72ed282125dc91e6364f15c2c79aedbeb1cb8c3b2c56fb76964a37eb0acd8eb1c92779b58c22fbfa5635b27cfa10abb685b2a5a1fb7d938b6ae6ce6b03fa5a72ddb77dc595efb2c91af78493c04519881fcf362d7f01a483b9535117f25c155a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638137271Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d, Name: checkpoints/3072/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3072/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a794e7ecf9ee69f9e3885d525764df191594d8be530b2bd3fa8a134189381bd7407239c0c99fac0774ef1b1220023b94c61fb13e92665f3332c74a314f9307d7c530aa929107378a1a1fc6128e0653136670f958b43f1c8e2919c3ffadee95b4e613641d51dd30b08b53a23d8af2c8415b0471d6f9fef70490514fbef28ab7ea05eb8586031efb11771cbb4d935a07f3c3ccb2297c4ae64563d7a383cca67e635e1a506dd8c0a5545ddfb0aa3413573060a50c52e6a9b54a44017c9cb7c12e67ff64b4741523cf0b4acc955a1d4c10e75aec9a070aeab6caad018c367acfa57e2c26ac8d79ff7e0e95c82f783cabab9806c9c5c6efaa929ddc59a296af4d0c2e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638217981Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0, Name: checkpoints/3648/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3648/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=8588323b6d9b256df73b2b952224e7b1e362c2e33c9a04fcd5cdaf30ed81e6f0b0eb36ea26393b81d123b86a78497276c8ee53c46096f87c320a41e3f4efa96732048dc8a694284a94bdc95d5ab725758911c0c44bd67db3700942da816facb109f5ff4797a74a4cafd7d29ad722333f8860987d7bc1833c00fd2eeec26718d38f76ab91ce7fa8a6ed9b9869b04cbb4a7053c533a6b69b1bbe4167c8a13d2c148b0382c010f86ae07cc3f6c16694371d08babc677861aadaf428f8246f0a9888dd86e0aa8fa00c227e69d7ccda3877919d5db816eece1dffb280a676bc95249c8e6120a3da8201a670b37f3eb7f3a8572c108dac309c9e42c9cbb94cb6a431ae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638266099Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704, Name: checkpoints/704, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/704?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=664907363df7e33bbdbf610c6e8ac69c2798807a1e1c8e0afafe1a86e20dc6bb8953f4039babc73cf5e02a73eab65be94530c29e33c63245761d438c91cdbe3495c75bfa74d2e1abce557ba5357dd17f337d1c06a30ce5cbcd5bfad971666b6ef6202c207e2efeb6ec412caece5758205d7ee826fb659d885987006ff879819fde525a7776988b35237ae08b89c04857b6c521bb4ec3abdcd47253e1251815986378ea1efe721b4651de30a949e4a471aad508957178ac7d42c81285bf5df01ae32c4a51205a01259ac9c0dbf901898bd790979c207c621eaae6554824e54f5375f3fc5cfe5f785901671cfa5f9360fa1fd6f9662ca7d8b9811f50a8bcde99d2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638276319Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/array_metadatas, Name: checkpoints/3072/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3072/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=1c33c5d42d4fb0f540600a7e6698ea50e7c5375194b931b7bd4af4f09a35745ddd7ff81e4a846e0768b7fbf04e71b213572af59ece3a20798bd6384a74b82b441ee95804eee210bf3766356abcb76c8730ccc23d464657b43d7c12c145c74abae349571093ae3c94026aa85a8ea6323f299d5ea6c25f87475ac275152edd6f643ea7528cf4b2a9cbffdb567b9f0ba8e346b60deab86b09f7d1c6cf317e24eb8fd6c884f6c385fe4cdf203ed6a9d992c28133335d3f50f38c9ef6943dd73e3554a718378d4883791aa596eb349a9f9b1da2db356d1764ee2c95808cb49dabc50126a08f7aad563cceb07749b6127a27c94fe3f412a3c5f5b0bc3474c55200209e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.63830048Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/array_metadatas, Name: checkpoints/2816/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2816/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3526453bed95fe3014c145f21df281384d5658ef919d370f1fd01d3d087e17ec848178e8aa7f81c5382bc6552622ee7a3a3f9836a315ec3c9a14330a281b41fccfddaf3c4b0e2e4538f9538ef04d972df6e11d3ab6ae2e65242aead5844f85552c0aeef9a1a46eb39acd2eed24b99109466f458dbf70b079cfdd14c18caf000f6116ddf221c6dad7f7f4a92ce4de65985d49a856147e3e5930ad2554735efc9c22fdb37e1dca1d8b73c03bdf011a77fc5949930e57080f0534fc2931917208beaa23d0805ffc18ab20b5293da3aa76206b700d6cd7c8d9092968b4a007f76705e4b284f87c8bd6c964863ee9d0b365081e9608b6ff435daeb2116157ca304ead&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638314512Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default, Name: checkpoints/6080/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/6080/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=77572cab78d6b46a970888742cf5febf2d7d60a7d600429c1ac953ebed82b7e0ab8f076804ae0cfcd676bac7ed5a41bcba6c26d8b42720fbeb9031ada14b25d29b011907e68a6e10fcf2b52c4241550068f78bf2c983f06af640e5fe2ac118ee715df65bf81308f9bb9a58d788c3900a774320e82b47fea7ab2afd1cbfd5f345991661de9bca5b2a5554dfda87dd96f85f95ea02c2b30a440b0db7e16e4901e88db00f057b1e2c57db2188ede88fa701556523b044ec9fbc5da4d59c61566bdcbb97760bc97a48dd2c12babe64f335724e9eb1c5523b1e36e61194ed6485ebb58a2a14f94687af58d4e27e1ae6db3b4ef51b2a2209fb32ab892e931ef4fdbf82&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638313879Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0, Name: checkpoints/2048/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2048/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3b6e6e0db3ff08bef64623ec90d84831d50e27ca243cd31c3353c61aea485ab6dcedb91eac2b807b405097343c720e9c9bec1c43ba32c25b6ebbd52d5b691151937c3da68fa963aed2c06b708165aa7cae12bd36106d4db36b771c409bfb3b2d2f2073196fb4c3bd2d8375d6d7e2f6b1762bfe5fd408b20e5814f6362d307b56c6125cf3456a492a7a353e1adca84a9c262725b8f9f16f7298f5690d188436095e73cae759912fc6b9c05a2f65a25049e739165515c5509ebf2d3557b73f984c4bab841513fd56af814114a470b4564942d0080bb27e99157204555b16f059385e6642438ac4fc1fb67e44070007f214f73ed460cce098b3407bd25019a61ddc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638221603Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/d, Name: checkpoints/1216/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1216/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=1d11e7a2834c8b1180a547160636efb3469a69d8ef7abc4980a8d7f9d81f6dd23e9ab722e1a706089413c8b2d6ea540d44596ebe189542af03722107536881ebeeec2b7991444920c580fb741b84bd98aa0bde36eccee0aec50bc609c86cf8f46966a906ba5323880efc0488361d7ca86eb368483c951b0b3d8f92767f2dd12df2a1c1b7f2c2a9621ac8bbceb4d9b3a85863bfa39c4a11b3030ed00cf03cc72875a9c20c738dede6e4f38825b5a46f9ea1b195de60c2678e6decc904c1627df20ca55cf3de843807c1c16b5e683bbc3eaa881ec580b474184faa7fbb342ea8f87466cb961c2599ad40904bad4a1934ff3bf823fa34a100f5a72ce431d6c56c3a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638061558Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d, Name: checkpoints/5824/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5824/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=afdcc25556b499735c2cd69b397169c22e62cc5e2799cf14098a227e4a72726935e31ff6cc2f60d7e81048b2b431939b6a6145e442b7a1cbc1af1727d55a195ed516a27fc3717271c7ae414bc8132aa35067416562268c063f61e6a156fef984283512e888e73f9b5786a9333b42939ce97b3d7d6f7342d443f29b247b7bb97ffe59b61d549504bac4e2bdf772b5f859f29301137552d2bf1fd8161cc35747b63edef0f76f4a3864d63bd49ceb9f268564f75f851db9efa3b5d8ac0897ad3d52d8ff85df8bc0e9d52b11eb306dd76f86e1db6781e9a83a55b8589971886ea768af1ddc4fd63fd50b638ea9defadb706f3ff92196294532151706010d6727e5a6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638360515Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0, Name: checkpoints/1216/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1216/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=11bc7078ea7cb6ac900e4a4de24e61fd57350db2c59c982ca3d8aa170964172dcbeaab3fbbefd4457e3d46dc96944956d1dc24bf84dba0e7315a7f6f08f85b92f4d0e466c32ad9d5f4782dfefb61d0332676ad9d750563416bf53bf7d26e474dcd0aef52c5d586060d1628a6fe4f4deb1d1bf40fba6893d2be750d225beec621dfa118d296c9a3d4b5538057ee6da7f6b40048802809557eddc664900e0a68abf7901adc117a8e76d7d4a6e392c8d312e5495439bccc42db83a8787903c2c8feb4217f560906714e02c57d2d4b14d877d5c216c82e5745f1af728973b47e07da5e12a0bc34ac4a49687fc408ce433442a9b939ff7c3de6277026ff945bfc23cb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638447002Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096, Name: checkpoints/4096, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4096?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=2ec7c2647b1ba87815c90754dee60da42c89472b2be4b03db527380f24ee06ed6bb611a22e2596648d85c65f5f3ea492fc9cdda70fc6feecb91c263fb6bd59007c4c328a48a6e9d5ff9f7da2fe40f54cfe8dfd34ad548bc10c312b1889034193e1e87df9b09d1c7b1045c12ec67256dd0377ae68b719912bfb8345aad14809ede00435fc56290b73547091ddffc9f69c2449b6a67bde6f139de7d1e2113a06c64b56c43b2a098e150d1a50003ebbfe2987817df2a6089499580984eae64f3a7c47b00fc6a95627f253d22f389328cba54579c8b75f8a230dab40b5bc3b122dab432560812211f12e6680a10738a1e0427634908c2b2a0d9e79159211943d7772&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638451326Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0, Name: checkpoints/512/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/512/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=723b71a41c8401733cc0e619582da8616d86897428af23b89f9369b47471b68d2b0f755a25e1b784d59cb4492aef2684130e8d4c2a11558ef33ee18b8543b0ec7d1680bc4d6a427b4ce850243e05f08753cbd5dd9952f7195a0bac1214ba6a351d7ef51bdef453d028df04a24fee45b2c39f2a1d70322dd62b7e6fd07c12e7a192dcb4c8240c3156d3f1e5a5060496e8e88c67ba6593d2f6863e88c28be37930992795a45b5d6a7bf7c5d65a6b98dd62710a3950c6816ebc382f9c60936e538155e6817f36b56b0898b4ec50765d1e66f3880b1c1eebebdab9788cd156564c0cf8d37de582f0a25ed3a565eb3f64a3fa08a479655f700ce181a1650f39da599b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638558452Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/d, Name: checkpoints/3328/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3328/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7a7dc9884fadfc7f6cdfbbb5470a3b0c0ac2022e339a1da99efb2d848759459b6d9a7a8c158bb7973869eca38be0e10a67ffc915304e5c8dd9eb4628cbd48a28bf22caaf93a8037c25ec2f0d1bae2b563c165072d5c1a6a52db7b546a721d6b5108470b2cf3aa670ca84c199cbca6e88da3f7632fc5a12ce62a48dfe4b00e92c26cc169035e936f83ee898a4665da72d05d45f0d25100719953e393bae50e0903638a2a3cf9de6475acf177eaa7856625ddc449f619ba2718383da187e5103c741c77c968430b5253ec9061a895a6c9feddd0dca61f1038b9d60a06390d25197aa1d0d7bb4f3f230f566a539974f8e9f922d0fdebc3ae5e3363584f983448bda&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634263608Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d, Name: checkpoints/2048/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2048/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=70472549ca71fc78c6c0cedae441ad2cf854218d25aa76b40d83b194abb596045708486d0571342cb23f91fc3ed962f725662f3e513ae4859e7c82841e4ba9c9824f40defa027554fa90dd44ad80353059120da836a96acdbea42b83ae7da1b45d17eb59723f81bd269c7e05cfd4d62ca4ffc538d45510f76831976c7cb3dc389ed06223a9b116486caa77c3306526c8452946507e19310aba4874112b558f4b408c02ed7fa929bad84b28b718595cd4dd2ef330fd62a2c4aec7b4ef0c4d30ca088cb18f5fc8ab32bd27519ff5158a2f3c5d891c24a12b2256811d05af333b67872495c7f097ee0436f2fe24922a26ce50ec04336afd8e9c3a335a95634135ac&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.6342828Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840, Name: checkpoints/3840, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3840?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=272a70d61a0aab01a2e71b4f1f28cc8a40d59df15e82a75bd5227e69a54a17e6e3929ec3cb7bc3d2069aee54f1851260e897745083e69a9d8a4036f58ab2e4f32c3effeb0ab0d4f7e1d9ea84a3a2c04c5f2fdcf2bddb60fc997da33157bf7b6c0730450ce1b51a9b7e4eab88970e1ffe3a4033b2a7b2d712ba4a2b0839fca0ec3c406afadbc24cb72d77495d1474245ea56f137eb8041692a6ad758d92c726c2668d4eb2eff7e4eac8f1847f1c82a7064c1f06ca0ea2ec6ec89bc9934a848585f0ae4681ee6a3413fbeea2cd95b9cf086b4237aa2fd86cd85fac970153391f8a72b40a0d20c10b49ea072c9e29cfcac65d9f60a4daf7194ad387707474d79c38&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634104773Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/d, Name: checkpoints/1472/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1472/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7611f810a684ae8ad8d0daa083198b48b7aa9e0cb6e80a3c044014d68f153e51a31b66579637b2ed87424bff931674d30bb0510997c60f578f3de0a21dd584b3833974e5908d0cfc215f35912386c319a2a906d68aa4c509212806638b994c432ac0af5643c90ed127a511d65e1a9c179de05499ae90bffd0ed0ba1cd5f53230e1e2c02c55d151fa0696eccecf9bf5217dbf21ff00e8586c630fae27188d5223fcb9fa36d76d6207f2df742caca9ef4d77477b5bf119f19251b2bd45217caaf0c0814d8293033491a75ed71092039b303cc578c59b7a4dab42d1c48ab96094bfaaba94bb0c98d9c5d036078e936ac5311940b1253fbe94c44707bd9f8ccb3819&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634330078Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/array_metadatas, Name: checkpoints/5056/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5056/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=26d588f5147dc7adf9b6817690937c845e9ec73a9a9e134d48ee907e74205a41a5b0710ffa9fd6f9642978e0a0dbc0f05e6787ec7f20edad381c9d52e4f2c24b610c08cae63a25e4602f090a001f424159e3c6624281d4ed0fcb9dfe45d0364d266ffe36f9766651612b5f06c1faba0af712a46e274474ff76ac0a2242df16a470efe3c66ae6d508be8f9e1deeddfb6e4dce7d0ed80d8ca6ed7794bbf327b93ce233de9bd328b4876c661f64950d4fd9ca1e13ff4add4bd8264722f0d3d04079f6e68ac9c82ea5799b2b62563e1d74a4b6c47a119ae0f21e4f4dcce09531fa83d1a3edb25734445cb14eb8ece66b2d26db4a76315dc734adf816084ff88ddd05&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634323065Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/d, Name: checkpoints/4160/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4160/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7ff6d6375024911194306b229dbece9598caae974faf84c16de6dde0195221a9c60af40ca5848190cc54dfbff4ef0b1b4e1b61b0a466cc532611315883eee24a4472b3ea6a2d68e1d0cb46aa72931c1e79472bc12c3cf3b0374e0a635748c944b902a010d4a552092f06a89149b85d86ae7f7e4a229cac14526fa46facb5a48ac3c6c8c48cf6c05108b86c18dba9c129b9222f6cccb4ebcb1f9cb73cc5aeaa0fdfa2d7fb9864087c877f93a33d5146939f1dd44642adfe46db03446a51028a5682ee05b003133c0782bef0f9fefdc8533f2b986c93da856d6318b78b2cd3a39ac4b8f6ebb14e1d14571af913b048ffcf5a7f9b630229167a86dc15951cc1c888&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634345154Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d, Name: checkpoints/3840/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3840/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5c7b74e4887924b44d6eba53b134ca1f75789399c998ce8ec8e064139bfb0b37351ae867fdcd90f83c1e44e05c38109076d53f39dbca60a685317c38b750f4f357bc2b2e25c88d8ed46b9a2c4b75116e145c7f5a97917b1159c99e4ea622d0d2cb819a81f0384aa7a3f310099db012c6f22452987f895b15e807b60092c2c77121d183f8dfecc8285d0a0eed7820b8b3995e691975318396f711e72be21ad3a5cc7d48c05060b2a437455a0b93268385c021a3d91ed812a70b04d60bb8fc7632523319b1ba76f0941ba7801820077bb828bd79b6028b950054e278c5b61ef8bd35f99a01591730f453158a363c6cc331f3dd2eae4a3185469e4218bf7e37d454&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634338157Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312, Name: checkpoints/5312, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5312?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=90b77e2b88d8f7204770097c3c2c14f95d3e0ba2056194b60f0702cb0bbc496699baaf553e5ae2adfb04c49fa29e543b9fec0deb95aa4d814266e4c6e9d0d7f7f94fe1e46384dcd3a34da8ed8b4e843b4fe4052bd0678de9930faec75680ae5886e1fb9b0042977542ba6967f987683edc3e37ac03be72f2841fa121f3afeb75abb7cd7b30cd15a0e039c51ef4877399989381a00423fce612aff326eeda804db6ecae4842f518596b18aee77b77a3f5635a11dbdd137328cb8d2b15a4396ffc561ad604fc27a74597ba6734451ce44dfaf352fd655a30794555ea2d680d918dfe409c388114598b63a09bcc56be2fbfc82d5f420eeca286663f0da7a888725d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634181361Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default, Name: checkpoints/4800/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4800/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=2a274d7bbb20eddb275a93f6c2d9663e1dce20df96fc9631963b1eded766a47da949309562e0ca88613b53baf5e1527421a8a7ce4e6a24afecfeb0f369784e1598b6dc609f5040f4a46b5ea33e3fa77d9a93f185b4843f55f1ecdf74aae06dbea7a8e0b39c31082d1205022ea6df5fe1873e271d813d271c12d19df5a2a2633a27dc7f4698c8f435282b9b9d05d1c67b7c6d8ff7a5ceda38c69c7ce3e37913512b0a2fb59c2a28c1b0bc5c015371d90c89853e106547ec8165c3efd773713b2c858a322449d2329f153c1ed0c0442645512d68ad4add1476d6d8300100de77a522bf0318656e075783e6e087796eed0ddbaf1588ffe6dea2facfd1ee0bdf4149&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634396198Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0, Name: checkpoints/4288/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4288/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=48876adbe338d3e605157c211b312ebf993dc77fc5a8e9dbe1952c8cab9aa6258b61e4fe47579a526b7ac300f3939286a8a439edb8d6829dc18ee47bba2abb4b0d6269a4887e8d2f11401a8deac0bdbc87dc20e040a9aee5b10bb7dddc36f3a3f88e50a036f28015583df6d0469c38490f0962ab0978475f8d336f9df80afc2def04dde7cd5c4125e8251af6b355b0b8f03b07f2c1c8ffd0c5bb6fe833ff938f6b149f10aa839c51dde5c4b8d4d1e1e6de1bca7b33ef0d339479c9ddf9ad5467c9702d225790083cc3520803610788cd471f24b0bbffbb5664e1e276ae34765cc562ece8cbaa8e8827f4e500b0130f42ed1e9928c7bea3ae89b9453538548c78&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634413185Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/array_metadatas, Name: checkpoints/1536/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1536/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3fe99dae793e56b0cdd3f59155e9af47506bdd64c858c35d30f183cd246f1ab8e32f92b1fe005b78fbfa1c14c3be144f642cb0035a1d838c80773a30f7b1606efe010b683ff796c41af7e7c70b1263747ca3750b4636c223b6f8c75680cf5cb8acf4d1c42937c9186b13033556832586432dd2baa672a1abc3fe4c17f177b63ddaaed9f20c8fbe8f2f5936f724171efab05f201379c4e05723bb1e41e83f05459f6ab2771095eb76fcd429fde3504c5d51ee08a584f1004049edbd690ef2a74e62337fe558d3607af9a0cbfa9c8d2acbd51e95d44639f819f3508d9d2232b93d8e032094d5651342ec01a4b429ee562a44517432d20563c3a982d43caf6603ec&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634407238Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344, Name: checkpoints/1344, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1344?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=03604bce45c7ef76a3789d21e8453fc59fff21c3dbd00532eadbb58a2fbdf998cb39dbedc3fa58d1151c3b15a6080ba3501880551c73dd5212d7d6733f3fee79a8603e5da41464b965994f9e00c6b815741670af65849810cc4da6e5dbcfa9057e50809bd59e36cbe6f74815cff492f5582e3481fd494964d220ae253cd035db4575e23bf24a2268e43e52bdb071c3ccbbc38a6a2c880934a76ceba67e36432ebf9ad54d4893d65719869071d58ca77241a6363eb87ac2e158b99e8da7db68407a0106b9e04fc63a010c5b1e40b3245a961b3c06d568d232b22e1717ca903c0cba7ebf7c23a6213d3d37a92503f7a696935391a3a852ed6ee0a034bb4c26c9c6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634418658Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default, Name: checkpoints/3392/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3392/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=bd591034e2de9c3ec57a9a84dd3d21ed6a84ce7c5519641a1f1731c8df18fb7cfcdb84a1073208b079b3b15b05ded76da0fa43f67cd23c425f94da2527e6ba773e528792210f06da4e3e32b87aab15f25585755ee710757cc0e9b983b3121b25c4888a49d815bd599cae19096c27fcf03c01dfe382260e4b5558934cbbea80235e3b7767fa3b7938f44a7e5913fde9df1680c2e9109bd4d6a8d0401efd784b386cea4da65ce9971e9cb0a9e33fa2243be4cfb22e8aac3b1fc6c0babdec19017295d2bf5019552c1081125e064731a8ac847950bfd69d6c0e4163ca3381341224a21bf358c498e1a4b04350905082fc40ecda9aed870adef5bcdfee7d7ecddeeb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634459725Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0, Name: checkpoints/1792/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1792/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=44d661a041591df29f652b667f79df0a4bd218f8bbcb819c627e59bfc9e3d801872267056e849d48c84b5478134297b942bbea7eddcccf11b78558e00526c07eb8cb8532fecf2750a272fc46e7f292ff6ee829127c97f728cd390e75a0e0b25de08e27787e1096b1bf32a8193fd38d2a304552302e47eacb332194f7695f05c6fd270a742cff69bfb88a5c1e46a1f183d944f0e18c7bb4253aba27119413c874e8ca124e44a3fd9d6f6624b4604e74a9ccbac8f13124e37f0e00da547195c701050abab2fb4cf8cfca4dda3dacd738219101670b5eeae10450d35951519429cd21fa21d20a01acff1f736858b52b09f51c00de6cf9ba4567df21c617b206c763&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634466647Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/d, Name: checkpoints/5184/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5184/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0ca257502b5e7d51755469fda75de24e409499482e1856195c135deca232679f92d4a9db46a0cf54151004ba46d92b14d2474c3587926267661f02e890f8e2641c17b6d445d1dd35e59b11bc163c9f8e774fe9c87be5797d9cc439a6eb6bc0212d840652e8801ea94ffac4c51b496445be38d7f64ebe90e89efe0e9ef72c7d5a7c483af862e1c2c4932af4ffd614ea17089531429fdf812b3e2324b50d7ce23441e1e2db5e6a7e63814b8d4357b9824cc0561512ec85b1def145965228f9db3d82c43365896a9b9f5481574d5d8f217bb342b08c673a75b6965c5f705ff50460464aa5097c2a4d1d385a53c3cc93dc1d3a37577eef60b77a50101bd1fdd5201b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.639231731Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/d, Name: checkpoints/1984/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1984/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=549a90abe9bf8ea94421d83d3541d5285f0950d10da35e03839cf5f2ff06de5981a18c1795043f89c3486007da6c2bc197ba9f9ffdfddb9a7c6e0273b85da91b8b97bf3a5394c264e45049d8547e0ada19046aa568103018b0d42cbd03307d78ee8856ee229c62cabbfadfb06ceef3e1383b93d563b637d58426f51a7200002ba32eec312a097a27b63ca3e3421484c0d6e992836b1ae3f3c69fb7540583d866e94c84d7fbab0c0b6c8e5f455da8d040bdfc52d213a5adc453577489a7dd19921a45005021afaef01a5f606ef56a3709f30853f40df6526809fcda54a28e00d3774f4ba97c710cff5df9b3407b4c873b7c24ca31939e11680dac8f40b08c2698&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634476648Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024, Name: checkpoints/1024, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1024?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=05e07e150e19c004081f092bd74db8d9fffdf902ccdc8daf3817da16a6478e5ab3d85756d6944a070d62dc06e5a9d4a769fdcdc3a4c09037dc75d42083a62ac95cd90e683d0dda72cb2d60b750165e48b9fddbf8d4318c33ac068869c090f3e6164825b858e7979bd08dd415f46c11c7b10b1977636070fed8a585d40b6fc51813420f57409407d61504ed083f51d8dba55266db44aa32a679bda29d38e9ca51232524b489db407fbcbdde3bfc2a0b9f405f9d9935084e5fe001ec327a3edbffcfe8e3edde07d95789d52e220e5a644569b7669bea1bc3ceac8816447cf55189441806d17d444b7e215dfd4bf60b8a74b284ded7a226a9e9e95e314dfa73e39d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.639287512Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default, Name: checkpoints/1600/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1600/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=95435d91ea79f9ac87f3f80912d60c734539dcdb9f50207141cdfadaf3b871efb4d5efa65c12a7f869ec5203a9e5602fe4945c3eec084387a6041df27ca252d2a17736f1f4ae47ca6bbcbfb441d184e80293204cf0ba7009cdfd4fac9455adef349908294b2df7c4871295c5cf32325407893063b67e7fe6382775a9ba8288fb4ed020d342fea77c1252510686b8613e987113948366f35390f4c1e3fa715313d2cdedfdff1beb73834d6e75b7b18561fdb100f97bfd23a4719c287becccb429e310db8de5b697a897b752480c7727f65529a5686f824449b283c9fe751fae53ced5b5c40fced597f83fc4b1ab1dce067968efd4035955b756c066e25cfa74cf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634501107Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d, Name: checkpoints/4672/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4672/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b0566c5609504f8cdd3e2e82d56ebd3f999851b21fba401385ba1e4839e9dd5c9c2f633cdc2f0c41134b8c57e19ef4a185c93a4a20587d0d5ff30790ad61e134af2de6321d3ff162bbb4c6f1ec238967af2635bc69298f9b2e569da34a6e064562e8b85b17bd3f5dd8a985f4bf623de5e5fb947d2a80735be54f72d71d482a225d4ade624e062788f579b7df1a69da33c63485b61c59688ca2f0cb1320cb12389491f60c39ee58deb0605dd7d62d7fff0f09fdaa31287abc8cbda3f50a23a4df1708793d68180e27c0e67801379e19c2e2c78345c56365d7c8928c83bfd9bd29761fb43efa743001aa0a348ca12c49a37a1da2ae777663c57f60f1cb3f1dbe89&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634512836Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408, Name: checkpoints/1408, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1408?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5903f29ba75d9c3fcfabbf3ecc4d09c36e6ce756c8844f14cb7fa409dc609c049449e6ceaf99a890a5770930212c8c73d580f38f48ad1d8242bdcd8f51551a6794f311dd01d80adead8ce6c138b38a81f06c8981bd0d5d45bee2257975119cd5a6769569f4092bb8e5e7a23a5419defb986509798b477c374c12bbfd91f9b3d388f450d38864f5b768224b796e1080303f906a251903dc758e13c88a1e48511f9d1f3b748a6dae129bbc8d6edc2410dd035ef7b3be198c24a0f0ff9815a54b9bb87b6fe39bc3e798ab8d866801b462cf5e02ae26f2dfced693b1ed8b3c36f8fe229cb92903a3e5e7fe0d8cd6798c2679d64d23093e1691aba197f5de98c24ce0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634536993Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default, Name: checkpoints/3648/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3648/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=142352c8e509e80f780555971913a6d112894065e960c17fd8f80d3e8979032c2992753e2b55d4868409f52080cd8993ff50d2181e8711d8dcb15291f65208f05d140dbae6d2745ad742fe3473c617e6d10fc1559341a3cc6dd7882d4032bf7b00e9751663f17b5a74e9ecc9d00b73d5294287cde95ea56e44ebe9df441cbcb90135d3e1b00190b4a359a82003d8910a0a1f43484e086788253c3aa369bb244365e37299fc275c494c060b1253f9d3bb88348939bdec46c8207f320553187891bdea3dea5eafe8ded11116a7479ba1f569e7d732bccbb74ef7b3315cf6301b9a4da0a465551d4730492913bd3669f219f0f3e2685ec50b3c58750c7ca2cc1846&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634549368Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152, Name: checkpoints/1152, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1152?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5f44798fc5a51f353ae48f8793874e1feb7b390b898957a93dd0352de095b39c8df65aef9373cf29b418fc3af7daa69768d1ff2fcdf4f9f1d7ab52fde17a421c5354b0268b75878f6759dce8e0efabc7d2a37abebdf6fa49e015c5f997638fc9e5e5c0daa68f4e57490a7b518c1abf5a928223510e976f0969297de09e8e6fe70aefdc3940ef866166bbbebcdd2ac6fc987c2b3a5916690fafaf7feb84d439cd26a67e6ee950a7ed2a89712efd9ceaaa2b8f82388a45b7578ca65b76fa3ce5b7442cb96a9f0f2eaaa6c17d830e417d075b1f74aa2fed24be2d1349dffb66fb8840b7f8cdb1e3bd7d0431c761cee76ee5329bb1a643d2c48f086ae13c166d055b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634595584Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d, Name: checkpoints/5056/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5056/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=33c1414a8fa67d752b4d15cd7c886a4f70238f45f309aec45081570b690d3d93f58c8d65dff649b8b0c63f590af72f3cede05c3587675ea0be4e97e3f914ad99f92e0d3f5acd370b4cbad78209a3576c5bed2ae87fc36d871a70d974a5efeaf035a8f309e59fe0cf965aaf46c5e15937d41211b6c16d4c93c70955e9f795374f0cfc4d06ddcf8661b5aac611e08f1e75e7d0d953fe88484b441b3b592cc4cd08de734747289d135f16f4ea3276d5ec00ae865b59038fe2cc7a950ccb150299b63404df4c3fccdce627d210d20fb5611c231fc99f1132e73ed4b9380f2f7e68f13a50fb8d547cb0f3bcc215cb89a94871aa91050a707bed22df716569baceda21&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634598961Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120, Name: checkpoints/5120, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5120?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5e703763c622ee84f52843bd68f2dc7ce1de49bf6b9890324ad800b0fd3b82a4756cb18d57faf3fd7c257fa221cee3d955e0fedeb42226accd69b43a38af2147207cb0712593007e266d0dd383a8bc068b14dae2f921bb641f1c8ab42ff145138d299b272f525cfcdb4b0869ce67da94911ba20838be47f8c6f25d5eb24a8e843c14929f24bbfcc4849155fb243ff3b697d18f54c9d998c0114be8a0fc47b41a83e4fd1559484f7ae738823288a5663546bf10b16c0f75d9d4d8208595f3655a725bc78933664b35092685b940ece14f28b0a0f377895ad8b88e6fee65c28c43ab3feb80bef19cf08140685750a99c681e2fbc2ba1bbe9d9aec1f68fd7dd43a0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634589507Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952, Name: checkpoints/5952, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5952?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=37ea61b5721790f96f685267cbaede783467d0987a711982d1d523d13bde41fd289236539ee57db3bbe57005a8a4ee520c5fa72c930373b234da62439ef3b0c8eebfe3f7d7f39a0ba325f4ed3fff38959f38fe1f3bd5fb9f1835be6d7259fedad356745fcbf0b903e348fd713b200bd168faa0c4f4509fc200ed6995b0941354777fec71bd9d38cb6c0669e55b3aa7d0ac8a71a2d0eca5048cbe13fd6dbb6c72fdef63d7df7b4cd6a2969d703cbd50226f90dce3489b6f93df32107b73b824fa8c4f3872cf7b7120ee73e115655bf3dce94f32c52658306024d3ebbb8facb13e6699701f7e416ff32a0c27b7116e517b08adad4617b7e96e236ff57dcaa16d9d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634599023Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d, Name: checkpoints/3712/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3712/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=317e26af48c3ffd9f98462f016eddb62caa1e190bc73a07a2acb8bab17e0a08bb3c75a622fcca6605f80a4632f1b3f47ea3c69e0c10acaa1dbf16a301f54e6f6033496a35311f23ccf9d2e8771ec0e8b6173ac4e5d2afe53a3c8ab3ac79464ab5579a54b777fed49240042320cb44f23d29e526500f8b9f64473bca9c91e714b49647c14101b86403c4b08627395de615b1d90a9398fa6907b7773a231a4d96aa8066e74671bf17e691eaf841d2896e4105e5fbd4fb5cb4a28b169753859b74420b6d76c6660a0bdd8086730df22438bb691fc6523f441307b9f283a20513fa601ebcc176b65897e5ef07adb8f85cb5b6d603fe5c87da241ab910ac5bb4c7bd8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.63461029Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/d, Name: checkpoints/5504/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5504/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=1abf9ad3bf0f338ae6dd00407064087686dc9646df4c9fe85f5139b73d36a76d053cdc47021467c46d53ba62755499750e54548cc3021401bfc7819641a57c6d348b10d90d875be46d44bbb3ef0ee0ff879f208e40cf269519a4c52d5125a85f873355e7f3b7a398b2be311f6bb145d933bd06226811f9480e1ddc9339c31772f0aa699e71aeed9ce2e9fb2e60675d04bf152a8c7be78731924b08c6598dba000d550ad31166dabeb6c84e7a6c823fc05fbe98681517d5e993517a5970e0a8b7a8e3d21aa4887d798f7e10951a8d0eb1dd3c3cbed111818a4e6886b43fd9f4cc67836a8aae946a07607d9050ce7ab6a4e20acb4efe09d33c7dd59027c25de0ca&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.63462465Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0, Name: checkpoints/6080/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/6080/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6673c313eba655bee59ca4959cb63f586d8c094db7ad3cdf88e8f9ff30cd2e70d7300672589f80a52a444a16f84505daab9b65235d5f42330f0fcd86e6dd3e1e6240978a5de99c5c411ec693360d808645598a03ce0a042a6c2a2c60a7288b906498b0f3c27245eb4f2519af02acf77739e55e296c641536ae8741c46598b8b42a66e15e22093ea14401c1cbb21b65f40fb54831c5be63b08ecfc30c6d7044225dca7a77b1c37f57bfa7fbfd71a238df31d5c7a7f92ccdd89ef3138b6449b20f2dd1b92c3edc3f7643464e4bbef402d3991d0437b199993f9dbd73f884465fea6c2ff49022a73802d5cac805f36c6cad1ea0bdcd46b05515d8683f4fb0f1806b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634638806Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248, Name: checkpoints/5248, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5248?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=78200434aacb97642e02689277c9c32f3db45a35a904008bdec45d53acf8f2fcf2eb2973b75b210389e306c839828bcc021281872a60e256555bc0610d4cd9a8ce2d8bcc7f4de019a14b170a6b095938829634681ecdd9673dbdda73fc93b01d57f1d8fb1d15ef106fab15a20c4c6a079a92e161c5ddf2fa2249a8edd15a10fd0498f6764c4a9d75f4b1958a28a3f436c05ba89f5dd3b81b6c9f91e1a632e8e181b5d86b5c10dabf42f05ad699caaf734c3d8cdc487d487ebffd168c5db83551a21cc68f36623bfb3d24ffbf1fe44b5010b59f0fdb290ba39d5323aeead105b2229acdf1eb672087eeb9fd87c69b6c93250c7b8c17ac67b177de117c947f23ea&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634660704Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008, Name: checkpoints/3008, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3008?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=00bad52385c4c02e26b172cd8d32f43aff02494a072a86dabe2d9b88cf246e70284a922cf53a8e80aea0527e19a942d127a6861886a6ed24ef8810fc03d4bd2cff5be0b214f9d26617f7f14e9db015ab87ab7bdce80962a78efbcfea61dc6340b97faba3a3e4d6487bc92d1409279f472d99f184ae4c67ed300575887e38e2f3a93269b0f6fba02a80be71ba94bbb0fafd4b3805725952a73b377eabc29f6515f08c6d213317b00e048ede2e548d4463e9b44b50f982e7bd951d0554015c395d829328c560a9ba92cc7b6f8a46befd637f851e15817239a1d71ea5a474b73865524270ad821202f18b2fa64fb9e795fc25e2c8663d870d3a5558cc6acf611889&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634665333Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0, Name: checkpoints/5824/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5824/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=c054200f007099c856910f0651bee00f5390e4dc1ca35fdc0b7632b9cf6d85d8efe27c3b84e2e7a5a637d4cda809843741a112df336c82aab61a7d7a3c4e04b4595399775b729747086a22da2eb540278c09d2d059f904bca40637871350849282cbab21019e29e12dfe28c577e31bb800b7891cb2a1aab6efabb51efa2a6df0cf868bc60d69a203688e34817088902334c08ba2417ba3f81f42a9a5a9b2ecd9935bdc2284e12e81cc66d8a49a7c5da4a0b53b1884edbed03c378239f6c66e103d45c9bca31b5a3abbafdde71802681f90b2d567ab90a9b2922e7c19646f379df90ab55d5d0455ce00b0c803da1bb568b985932ea9a71c162c05203c78bb7787&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634670969Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default, Name: checkpoints/1152/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1152/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7965622ac12e4faa842f29e0079ac11d1c41e19c140ef2cf84d8c8d558a33f430a9cbab75e70854b2a641a34bee33439c52c466a45c67aecdaa6a6eced59178c3ac3530819f5e445796c558a27cf806afded0c926dea5c72fe876d9e4d0b5446363b68334e9a54fcbfb185744b0885ac7575904ae31821874a11894ddf53532127d9a938b4e6a7f3753996b2f33db668307e96b9ab79594d1c0c31bd7c290cb6501dff4067085e88f35f99f544472c25ecb40ae1e14a203d39e0a05e5ff0fabae1e2f9c2e725df300020ec7c5928d2325f1524933ecdc5cb96c0997040a542332e93a6430ac4e051a946609c671a0b8206e9d809da5af2da89b51aae4cd73a14&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634678511Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/d, Name: checkpoints/5952/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5952/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=16d9a980b181de7061072d6342961084e6a08bb07ceada4c356661e10ae5e4e9502b8a2db26c4f586fe18f9327af0364945b97d2ac948097fc6ba3afafffcf5baf7c2ce56efa591b1989a497104b3fb0b697afc9415a147cfd33bf7c721247cc8ca643ae2e40eeb6cbd4a1065a5f239579dc4ea2eb22f43337f602a3ae1bc0d3067cd77eb56c8e00846666c057fc8043e807a3d33d4d4f0367bc66857cf707fac53fea48fdc0ab3324ba961625b0bd7a1b8a4818124a1372c95ae73b5877f05e8a4320da816a00a5d38d4becdaac70776706fc5a151ce345ae7cc1acaedf987a89595688177e207fab59e4c125e148bd6e7ffc528a46541004eb72535eeaa0b2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634679082Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0, Name: checkpoints/5440/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5440/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4945339e50f664ad2007702b7486cd191f71a8a93c69069ca4e9aab06ea46572a82f6777e95b72ff123cac90e6c0d3dc54e800c829ff5f9d1e5dcf6bfe7fad21b90435ee3f1e6e7c7e0cc0aab1efdbfd4d96408a01d32ed9df07c5fdfecc0d29621f748fb9f3bde6438f7dd626033d9a7f1658cbf38e929333d22695f4a3a10665ea21f6a4c4c2ace25914f0a17f8b6f679caea73f52ea8438c29f4c7b8bb29e6f332da35ee2bd7fe7eb3532eee3740e390d1d2ce2249c0dcfb443ff981afc030ee2b02a1aa657ad9282f9ec7555c2a0b2a4983cb063d6785036dc8ec00ddbc29bf9a8af553a102054bb2b500f62199adc3a9a2c8a1c7ef6259bd81264768b3d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634679419Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default, Name: checkpoints/3904/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3904/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=84e96c3cdfe57ac5317bddbc34cf22e2eb717c21b0ad68200c0821383894ba61294946f840fd93c7c5a6bced8568d13057b942eea4f0d6cfd3ce4cd22205cd0b4069db8425bf7ee9f0bd0f4291b1833f378070de32acce06903c60b65ba34c257d953d127418b26178fa3d2f6b0dddb49f0df13955f10470f7b827c5175cf694a357f9030099da382c0acd4e03f67d73ce8407e44c9d11ef01d95df2bc290fc82bc66c00e6a553f9080137d0969311d2e2624f4c528fa1ca274be3a6b23936baf86cca91a7a80e7f11a02d761d3ae9ab63436bf3450dea0ff31201c300fed2426f42f9ae4da52053be509ee8a33a93becc78edc089495c04b0a79b8e0b96b938&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634706783Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default, Name: checkpoints/5824/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5824/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=60ad1d726a23d5197d81c8d2af8a1a7b74f24487de7a23aeb2c559f06b8aacb5354ecda211e9cd138590cf43cc9340374cd03814b79492f60c245c3c8ddbb50854644c68e10d69da01e0b601806157a849c16dd6eda919c6a89e40c4151d11c51d816a9cd96c1777ba030c9615c8bf4f0cbd890639352be60825d54c657eaea35501a96fa86849966635d6af43d0513f9ffba812323e52f2df88f405c4098d34b4e973767ba45e8e1b7f361d642194b7c165b461da7af41d123467c7833a1f1352af7ccd74e0d763f9dc2ce2aea4ff39a1b9af69a5220ae9f991fa2a772444453aa25494c682a8d807bffdd71e9150cf36adf745303a8392c0ef5e90aaf94ff2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634723102Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/array_metadatas, Name: checkpoints/1920/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1920/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9c01c01101fd155bccc6fe832b4db0bcc41b8ba844b94a4aae4f735e72ebd04c4f5e065aa6d2bacb0f932fe82780edcc29fb4a931274a4a272292cfa3cbde22996afb330691ea79cbb410829588af90cf096224ca31e9d50d7b3c99fd360a79dc26e1032348ae2a864777f229b8f2fa9afb712f16bb7e91060e3e485d91b4d562554ada42fa11a615438973ea22044d242d0a40d4327ec4badbfe687312f0008b106e3750197dc58335d1ef9b0246e5dcab427dfba296fb59f4c7a3a794e43e592d635b8c14a265db6f48e0183cd4016ec98d3cfe7870bba8dd6eeab9320dd059d2e627563c70dae7c4abe9cdbab12848626c661c88360c43b512ba13b9433b9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641042647Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d, Name: checkpoints/3584/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3584/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=40c7f80a6538acf58981e57542cc91ac1585e258e0d6d30a1df57c18395ed86111dd99c650d78f30d24c1f28e8d795d6f53c66821aba017a550add76a6cf86d1639028a5c4df3690f28db224fe79dbc2c965f21748d9e9470d51362b0ce90b8d34e70c174d937e4df20a993f7272254239ce1e49a2c2eccc7805963da2b6c40bc60033c974c02a6d29e028e3e2d9692e7a7bfa93c7fa08accc68f3a30f5bf6d7e14b05c55a740122864ca641dce0ab4e867d7333228987a1596fa394fbc8c789f2910f2bdd32597d1891b28aedbf65f1145bb5bb9126e8018eca9a2ac65b2bba92fd67477d2cb67541d006fbb003d8faf94686e8d8b3d7b595ca5969e9620939&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641063448Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/array_metadatas, Name: checkpoints/3392/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3392/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=81e4ee556bfccc5db50bf7f75ac0c1e0e202529871db37ec44792bb485e83d00834df318c59c401fb74df228a7aaadb0ec875a979359a20209043918d3f0285c7fd6ffcda4c3218358b12ea022116a56b2a1811aadfa08612d8482df8c85e58f03b368311b4724c1c06842c24e8f05c00686c39ad17ebba2a735e60c75cfcddb85383e3722cbf799552eafef2f1174a7aa9255d8605dc9316c4f63daf2772068ca3766d864c08cebfb66c316d60199880b270774dc071ff3fd611fff2d99ec104a49b87ce74e837b050d0fef4953b2179de4d3bb881364e1482f3a5b487189fe9c71273faa2f77fd0f7124edc99659b7092838fadece2fdcc1da29b3efe60cc1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641072032Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/d, Name: checkpoints/2112/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2112/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=659559f6394e8ac0259b213535581e9923a911bf086fb1db301de2727cb32a1ff9ed18a9cab8a74648acde8fabb3b6e5799edecb888670ff3fbf850037dc0f8a8f0f933b8985695af595b2ea402ac3a024e473c925d1b5451ed54cf8d5c3a14457143811b209dd07c3eef797116ed93a8d84aeedaa02613dfbfcf65d01e95b48d6d3fff0d70c9b1ab48b1fc3159ea22e571be2b5b67edfbd5480e62b086564b804343dfca026732fe25de92828f100202b6b3b357dd75020c2e42465b2deeffcd012a584e5da8c174b9ccb381ed4b07c3740d022bda5ce7c40873b80754f9d0d887b4d4dd7602177066595f13e75d4bc59c38a6a6c1de5f8a3a285b0041f8997&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641111194Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/array_metadatas, Name: checkpoints/2240/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2240/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9a8fae7e7ea01aecbfc63c6d702d85a815d7d64302d2aa4a273fd771f661af98fefccf0df1225066f54258842be7780999dc09b32fb7fcb476a0e3dcba4fb0f6bdbd31870b5ced719d42bda7c13375cf0ab15a6ecb160b812e4e756b6d0b03fd4c6b5ec51eafe0a078177067fa53a163aa5112eda303f669d252e2bad88f57459429edd99fbe0b6da910504a6ef94301a5ab7677f26fcbd81b90b4cc12ff199ed534ee117c3e6671c1d7e3cc34024e3ad45352aa4b6b700a3321dc7b0072e5f626a49d849ec8b732df564d1391b90db5161d0ff0e873d1a2ab8a7377b6b8469014b272f9ff6dc42f9ba25eefd52bd0a30fbd4e9dda5b7cbd00a0e6e83137faa7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634735422Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/d, Name: checkpoints/128/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/128/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=32a4bb5e4d05d7aa7f100e1fe3f55f46d9347aa3ae72c309f0ed55ce3ee6facfb2d9e9634a1c35072f049a47ff40bfa491051442a5bddf0292a6bec07c91e8fdf3975c774bb5f5e37540ef43aef6061cba68c6deaf4cab2695080847e4cac8cb240c727f2edce9e40505896424b66ef0700316cc6153e96cc834d16307e18eb827de3b220360282c4e77ab3d3c70505d729948e096985a0dc63155bffbc963ef55b5d3bfc7384baa5bd83912cd282346f9661f4641e5fa8a2ac7b00281f06f94b6613f351bb026f7051a0bcc51705fde741e767db912e8f60a39bcf051d0cad3d33ff7dbd643f85d15d6c43a44009b7be09447a430a3b7db869bcee1d187e5df&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634732966Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d, Name: checkpoints/4480/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4480/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=66f5725edb41707c4ad8df9e36b7eec21325473d6c6028c80444deac0356dd20b70fde8e1c4c565fe382932fbd4ac531ee34457ad6aaaad02ca4c5f9ffddd69923088bebdd78f92df5308f53621048c7190eef26584d1345f174be818ba1fe70ca1af518fccfa254ccef88c6b3eecf42bde9fcc44437c50cd0fff3d9feb90286b52fc01a34ec977d801eaa2ea619ae20ad2d8c0fe1b2c276c2e9c7fabae9a32c9fdb5779eb9a1c1ecf8719f82946fbe57fb276e30041199a68a7c41240b63d61dd2fa34ae6782b44dfe99596855bdfbcd5ff5582edf95f800d5332db30ab243177d920f23ea7f9b6308cff4003b0502fb91a36ccda39aa4356db24cd2e9a7be5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634739979Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/array_metadatas, Name: checkpoints/4224/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4224/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7627c410dbad1ff4591cd1b8df44923633e3e79e0d395212f03a03ed0ed2792fff9e0148713f8d8f24087e5c02d36fe7b68c975d7e3123ce271f7204a7edeb70411e5d07907675a15bdcffef55e56f040e9abeaaebb5db579f601e5e1b62122e51152e205e6e08b25ccc000034e173b0d19bee3e8270891de501914321208a95a4a21deb61c5ef9d3c39a5d22307edd3aab2853abb1e1af2342f9cbed9b82a5a6102657979b79cfffebcba9ed89ec9609d069809240b3ba988ac07cec7f076088b061f0fcfc319b15fa6348474cea3ce549e951d67666bd125e941b281b308254c173139fb31d90602ef9e7e9a8f20db5587617123dd04752273693b5f121407&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634740243Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/array_metadatas, Name: checkpoints/1728/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1728/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=65f056fe1a478d723517bbbb2a972ebff9e59eeeff2b5b9af59e8be347fc84c045da7a6aa4352532e34c5442e2af817dad699666bcb26e8413a43073297c1f011f29d4b6730abeb48ae7828a6f4454fc605104f6e6c98fce1095b09025aed119a3ceff67b3f2b8014feb743b64b674d742db1caf12d2386eb1c16409c245e7ee743c2eee9d44942ab5b4b51e4151d50f0f2afc6f766b18aca4a6439e688ee795526193dd77535e15d5b2429229d73dda063f77255d66cbfb4b965db477a60429843cb2035f411ed7919b474b670ac36f2440e7f7b3440f046711bbc2cedaba09f7b89e79f47172981d174c0c35381a931b81d41d08b3cc81bc5a0fca921e3e57&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.6347461Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default, Name: checkpoints/256/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/256/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=447f51dc058da4b8d1af1670fae0c6a43a734d91e78c2c14950ae3a802a7a28bc6fd1b77dc59f16d45b31c4664cdb218133342164fbfcbd4ace4b21e967c2fae2a08039852b1978eade568ad716c2ab242e70d9a64c55c289eb5c9353028c2c5b8f85bdfda9a70f7d45022b6687c5762eb29b12224d7908be93f7a1b6a2ada1e08d4dbb778e01e08408b792d515cfe80ba5b538b684ce59a9e49a9614da4f9182bc6f1ff49bf80110504c69bfe648ba111796e0b881a913c7b0e74e613f50767f6fba6ecbc0d62f8de7926eac1fd23beb51cb6b48b6cc200f62917c83b3b566928b7f1f98838325b6a4c35f18c001d90d32d678eb05573d939855c8bf6b6989d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634781494Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/array_metadatas, Name: checkpoints/3712/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3712/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7753a68f53f8232c7fa6152120fcd34b88c6ce37f3e03eb20ec8830573ccb4c01b596c00b16047c6f8e62ff1ee32e7d4d9c77693938af8208dc3ec14b32772fb566dbab8951c05600fce3a1a7f554d591b2c2dd1a4fbc1362f0a7478b0cc01e1f2016d8f476d8cc846cf8a03061cb81b3c2ee579b6049af5e17f737560dd3930dce6670f7fb2734f57327adc83b875ee0d3ad29d370adc6e45e76d2e4fd125cf1838f5193055d310fc8690a7e7a5f16552a3089bd0759bd7d5f7dc0cb42edcb5f0ae838ad825d8f8422ad83b651567914da1953f893e28e6779e59aa295461f25eef68a769ac8111b4bf7ff6f9e5c50f10cbde744568a880c33c6d2aa1752a03&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634788163Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576, Name: checkpoints/576, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/576?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6b089ead8d96d77db2400498c48d15326e2219e0dd7f0b1a304210a081edf6df15be35ca39a7e569180f25e589af205e26a3c86b9f1090e88c4b6f2564787edf915fa9eb4cfe44bcbc3b330e8ccb869d446e3dbe93982c715e0a9e1b1c18d0adfba0953b0576f8b213ef448c3e961e284a9a7dc3bbd0d200af9470d2b75cd9b607aa28b9af528281585e9c38271125420ddfe959e819a3e01f351442e09bebe133911e21bcb68b3db95b2a6a2cbc7bcf3ac68d1da0c9fdfb93aae7e55857626ca4941bdaa3a7fa3ad0a77ef85a67232d063aa1ed25bfe2bbcf233c6a0a0a86022306a99bd326783d0b134f73d93c08ac2e83dcef78cc81a64b14d93a480954df&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634798858Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/d, Name: checkpoints/4800/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4800/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=48981037d5fd888d21553dc19533634a22c53d00dcdc39b3501ce7ee803ad9af5deaa995053ab86a8d5518b912762ef8818650088d744eb49ea9ceafa4585f117f0faf932a4426f033b9242d55503d0e5339e8d9009b75b7e1d5de28867254b8e1d4bf678660fb4b74010963c0fb21b7e7344ec4efd1a0c046aa0ebfe4a4a347999cfa56af6a0c31493fe088ba358bf0530433e171f27aa8fae10861c23461fb55fa9072aaa2f268b1e0be4e6114267f048c3fa004da5cce4200f307583ffd90a851b57695c4b2ea7d1b5063bb341d2f268ad0d98a08f1d13f7dab8f645c7e12c298fa76fbfe367834cafb8974cf33cbfcdeeb9bba4497114504f96731db3ade&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634818366Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3584/default/d, Name: checkpoints/3584/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3584/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9c7a758cf0bd57676696d9d96fbae0def12ac24abd06389d225319c49e810a29bd1fdd381e1634f36efbcfc97f232825cf0a8aa734abcd37ea68cecb67a7f57cdb8608beae34173e4a09a1aaaf51650310b3e4ddd32071b5e9f5f7544954074452746f4c63db37d720d20018a0798433126ca0b5502497e2251d27cd70489152bb85181f0b1e55e2ec122bbe0fd014fcc644c876ca9efdf4daf3dc7eb290cbe3082b85e73f207b5c0e0fbaeede6d630fee497cf0f39a0a77818e0377289fb2fecd86fbd349204130b65e0ac12cc0ae40f172cdfd55ed3be5c520b5290b7e68da36ea8b9addd3bbec32d4519d4899df6c302a19b7d44cb76b89d3576c6d107fc0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634817063Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/array_metadatas, Name: checkpoints/4864/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4864/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=67b76b891e9785900081783c44bce89d58790bd3b93a68beceb8bbe5e0ca3b6b9e451037b96472192e0596e44c4df8aa7173cdcbaed212a3ce615f43e36583ad548ca3b4b581bbc29a3ec39a1c39e3ac305f95a5470cf169205caa74c077b28bc51f0263ff4d48eabc889e572ca4144a7f83c0297d3a6e8b31f2ac8c31cf0366afdcbddb74ef116df3f057362415ed755a295baa8608b0eb94f525a56830c35163c485d40a26565cdbb707fc1c1e63f0307284868d2cb6a76dc4de557ec2f31b0e3b8c86733b2fd6d7dabe85286512354bea0280032ae8e1eb46a0c5fc64de25a91d133864b1f69bce7a6e8995cb079b67c3a7a4ea0711074861486360e56cdd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634831514Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/d, Name: checkpoints/1408/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1408/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7ab7cebffd7523d504555190068ac75895c3d2ab5b1ff3dc3f8c1753ac4904bb891ce64d368d6fd4a36ee155a8a0f56b55cad2ecc7b968641a6e9bfa707681829257ef2926944b08ac6975ff316ea35ba97d0b4a38ac7173536194206ef9d3a82e1f5a3e9ed4483ec04ef7b263cd51d0a16a88573f9982d3cf95d5584ad49dcdf01bb0903189998bc5d3b3b7e85665c6b22e4cbab7429990a6497f04879e8a98edc8beb6f03083c925bfb60fe421c71dbccd3c0f6e7a608795741d6f514ec19aa558194decf54b49aefcdab7c8d039f38918e90c0c0cbfd1ba1151f2239407408313464c52f6d58903de1b6a7a1bf4bce86a7d72f6e32c1186109aac6d755884&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634851369Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/d, Name: checkpoints/5888/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5888/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b9d12e6add54b78a76824198831bc4b0f50f2e6d1c87392bb32165651090acb8484a6e4bfd53776ac13aa675c7b2f6291ce4ec91ee9f612e4f6cfa6fbc4bf59d61e3071ab3ec61e5ab38e6ecc683f7bb2ca8270af26d4569bb2e9f14e7af5059145dc6ad3c2f254107e35aa1d361d971a9bd29cc2626eabe6534e67e025d50cb4d786aaab17eeb92e492a66705a79790ab614366ae6d3356614d02262d211a5ed13c3a3dd4e72914b5b7d34edb1599b7d107d2357d1b8f21840af58df3a3106bdd458ec4e348b4f09a226afac2f0a6b5184378d07e9eeaf21a5788a5256ed28afdf5fd4b9ad4784fec68f49c15538c51cf0a50197311a9bed5d21181c04b9a0a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634869719Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/array_metadatas, Name: checkpoints/2432/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2432/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=12e472029f8b52e4daf643198a33326828d51b26956bab7bd0fbca37b2a8aeb44fac249a7cbccae1661d9f5f16832e4a06ea0274a52d4481340210b879714ef76e860a6604073794d222c37ff6f86b73f4643288ee6b20387d4d252aef92563fc835bc1ebe073d12f3703813e3702df450f45904786d49c8ab85436716d71d6cd3e67d79c69329230e0142c368d87886b15f863ff1ba52426cedb47e790be636cf55ac39475ce3b03e639277259e8d2a7823cd9188c0cdafb5fda57c7a06718064da74bdba6cf0d57df0b5ed0716b5c74ba20650a4628c3eac87a00dab6149264b3db427e9742ad118f76782f30fb46d883f5d5f2a922e6af6bfd1d23b4ec8fe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634876384Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d, Name: checkpoints/448/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/448/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5bb2c2f5f1b973b69c91046828ef57a0e5be0045c05fc5d65a12136c2c729f3b444dc5e1745b486e75e4b437d0fdafef2950b48d55add5160becc522c0acbc1f3ab2a027a982213b6dc4866d8707b9a36ba131ef34f875f7d6a79fc39eadeb0527d5af3a8fe7c60c89005261f101477afe6f5ea5a9c9033200e8084e595280951d0567c37497c18fc1dd31ebda46da069a2bb0d3b825d9a1a3178bac5ccd966739479b85851a501731dbc2819523e0d1665c881071a3ea42947ab617b4cb900d977691b5afd5a295b3ea13945919e6da79dcc350507b6c6d0d66d62193110b9de553abda428f92dc27b77a24754af2d64750420bf839092f0be14129a64802e0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634907408Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504, Name: checkpoints/5504, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5504?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a043842b7fb8df8d38cca59dc8aa303d5ea66094e35effcd12c2772ee2a0b303e4e6bcd084270c8bfbae6f2c0841463dafb71550bc41c1d694e62abc15926f596d13a3e552957d590541ae0e8a2e14e5420496a746b0e95383350ee612006eef0c67bb8109dd45a82a95283ea023a3c2f24211856e9497a64f08225f10a690e35c0a43c82022523ce6f7a354607789759211268593a956017868d940f668c47a06f3ab377b324c376fc692135962e478a01f694bf419faff8e47576bdd63027b0785623d591f512e39c75eb143b2ba30d80082e187a0d4bb9f9e0e63b736ca0ec2b8f5a13dd925263e90c61ef9409298e010417814591a537bc2022d196896c8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634909259Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0, Name: checkpoints/4032/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4032/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=abee9b120a40a031b7386d6d3b641b29c66abf968ec7555d4d822e81312f23eb6e13362eab5449a68d5ee0acc0997bc1969d18eb8aeb29d5bbc044ee6aa539ca11fffd87eac76b15e7ca4d11a24c80c8a6f45d2674ed99abf42104db0ec6b79c494041ae726e2b30bd6fbd5bdbdfe7e8ffbe76ff6a8635f6d1ce445134a47d430333272455322b3041b903a32c2bd4b89fd5eb1d55ffb5d60aad27babf1f5f4daf210e1a38348e9fcca78787fb2c5b2a758873edb7c54fbf1101b6c89b47e27185dcf31b3c261fbaf0001c1916ce2804ad1c7030674a4c88247808fec9ecbba47278829ec7a2143ee06c23399dedb219cb69bda2142610d510ebc192dad92d00&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.63491865Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default, Name: checkpoints/5248/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5248/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=34fd067c47bf08ecd61447ac219e1013d6820b000c94b2fa1e3947189c10b009aa2869570e70c396726b5cfd3161d1143aea86a74d312ee6027b2d6a0cee274e167ea6a4d6d05376bd95b97e2f466dc7de458658d1939676091c50f5d5cd84e367e28b62ec3265d5def7ee858b4b11ae11e251bfaa632b1399aa3b425489cc7e290fe4d075162e17cf366f5abb2bb946adc600380f19482d0dfeab61059b0c0f4f2ab7d5bda622e31d2932348050ab1cf60954c9a6169a53a0ecee140def2393a69660c6a701224f352c491373fd836393a33bfe2680233c06061b9fa0b94c84f174ed4bbf26ac3b38eeb4cab5967a54ae628d37f00cfcd2a3fc4dac8ef74504&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634960157Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/d, Name: checkpoints/5056/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5056/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5d12531caf4e4d71ef48e5185a51d6e7bb4c704718c476e4ef8e24ca7982bf21a738d17f5b1c7f060c69a36393385253fb5ce07f946a243e5acb683f354d82ebe3634cce1f227af96a41165031a636be8c4f908e4a728ee39e2587d523296fc14532d8e74adccc078413da1071053da042cb9bd3cf045c79b7c1e58504fa660f215f15b3a12838ca7355cffbddbef7b7de3e2d36e753d0d2d5918f41962925b12e195e7c55feb5874f5dc29cab540596986080d3201788bc3f050982bbedbc6f47fee37836ae8101c083486311f843481df669171447c5cbbf5f7abcec5006a7bf03267bb0b4c8e1d4b81862996a2f87eb8f929f5873c3c43122d97c1b6eb59e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.63500079Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0, Name: checkpoints/3264/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3264/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=1973d2e5eac31a33a6d358c170bda61a5beb33aeecd1e290d9e98b7afecc755e7bbe57f7f172a3664e09ba5d2317b7ec0cbb988e314e564b7e9debf6da38f96d6b1d8e6cf0ed1850cfd3e44b5b4a0229efcbbbeea6d26ddde4414b512e7104d06e62f0cb023535e6fed61fb02f9cd06e9cbbe62fed44dc388788c3e8cd4078a5613b980da47720f119add5f6fe95fa8009fba56922edec8973f24c2855fa844b519a89e821352be7902f2df247f61edc9ae07a930eb18ba98fa44afd496692956ddeaf0680ffa7bdeff39a6f471e161ded47a15aa9dc2bbd811eb5cea9ac65b843a1cb92648489203bb320a743274518511f073bda600bb248244d0f2960a9b7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634999946Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392, Name: checkpoints/3392, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3392?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b085230727d91048b794fac759f2662ef36eb971825ba2d65cb18e2d3c6e05460b7d88e4189efe6d59e766c71bcd91d1b2f24aa6d7fe28ab974204fc0f8c61294a72c8828a51eecf6f19155199151290158eb84e4a198a8673a74b271b0f19c7979ce279386a174c75823fe01e2557ce8da25f2b3787178a4e5eeaae1381b527606d2d125f70289d485e0a585bfd3279dd530e658aeb5ee046ce5472ba8698088b0dc328de98c52a90c0f31428036d81a1e428562ab1fcdeee0d82a4f270c0b67ae2be8df54bed41c629ac73fe9cd6126224a3ec16e72bfee1a7a5922e924b950d0d912c8d6be856eb8cfbb5cc53bc68de479a47a4c72b8da7a46ffecb7c1f4e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.635023862Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default, Name: checkpoints/4544/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4544/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=60a9f4565e340fe45cc06f359fca16de6ae79b417e0a63a59deded5806c708fb361b19cfc3309062ccba00c01000737087ee46cb3e6c3cd5a43356fb939e63ac5e0bfb10035e6fdaeb79a1e85be73d5b9defebdce8c05af2b322a3f61c4e0647fe4c017c2a8e912d0d6a7b7b527a65fd7cfa2c636632f709f2e3ee59f149e9ce9a5303983bb8660647604505b1020246746ef7612af0aeac7b249bf06c224fc65ec535da46d7c55d302ae0a32a916fe51c5cc76a0649ff0155f155c80fcc9545b43f0102e38ed88b04a679bb75e89493562615a7d6b480f7c2d2284557e3b568e769bc95530dce9543ed2e6b8de432e0cb0ebde17c594b3558c63b64c35b606b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.635034524Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/d, Name: checkpoints/2944/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2944/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=998fdf8e6537fe75ec2c0fe1d8f83ba9cb586501c89f472b55409a98aaff486eee05527c4d332bbb814f9c6408b5d27b6777a0e327f27bd21ce11117f4aa0631cdb2d51c0a3557b980741ab68e88fb078b2422f0a2b8a476a6429f9bf00103e85e23a18f96dede20b4d38611736901d0dc7e58c4d25423419976ee863f80099abaf0acfd96733fefda8846e709211a06fa1d3451edd85bc7ea7f517a0f4d6c866b6588900cbcc9b9ccb5b4fd3ad91c2cde3626c7f4fe4317caf76b9cff34bb4753abb7e1977167272365ec546c289be02ba1c67a42a3390a7ea7c320bad2393b3c098646780a935a8ee15c8a27d4feab6a83347096fdef2439630b2a3d21814d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.63503548Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default, Name: checkpoints/2176/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2176/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a2e868622dfc13a00fafb437f28393a6ed8f6b6a43f8fe0936e620f0afac94e689004eb52de5149c018830af774078491a87e16f46f2501dd1e4af174cb03a4b21fe056d3ca510c4511308caf4db84ad10752bb2ded3b37952c814ddb1e96b3cd0613da254d2f214d0e857c914c6e04e99fbddad763a0bb209a7a40bfad29c91ba656f97f794889428478b0ab8c9132f3f70d152cd69a168b6950c4e92799e6481ba3b8d8b7a4d406b54ccf5898a49f56a70f2376ec5b0b40f95ff98aaa04f293ea32980c210c52d081100a0e72f8eeb4e07f3f916c9a22dfa25b34062adecbb78a1d18aba3fb30afc03bb6f3b844796e1eb7792be710ffd8c0aed9eb0cd191a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.635023335Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608, Name: checkpoints/4608, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4608?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=71fec4482d648bb5ef56b59485bdb9bbd15a7f3b6dde7e54a354c616ba362c6d3622da4c2dcdffcb94a5b66caec94319ff0ce9714b9e34e13b210b3083a287e1af31eb884c06b9723ca49af41d10caf64f7d841903dd36789d53ebcd7b483ae632b52ef0406e16249a4f966117e9b59f43204ac4802026a0896aa8d4730d41cb4c79f960d40cac14c087803f5cabe8984a987d6bac912827c1472276a83cef6c90269623297042f2eef231474782033d18538d0092addfafbdab33c1ac96577a28c0959659c4f749701518c064f4b76dfb7a130ef35c565641f353db42c329afc8d156429a213127333455a0c70bb4f4c0a0991076eb7d3db62352cbb5d2716e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.635023856Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/array_metadatas, Name: checkpoints/3776/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3776/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7e5a3246f8d071df8ae2fc618707e5f46586adb4ac76cee73e5fd21e4da4cb960e746831c8ea79722dda4f7f5e5c68077b2477d10a7f09286a6e53cb6c836fef5c2cb0147b416533dcda38fe6fe1b7c2ea1c6f2cd62fcddfd1e801fc493248b18412ee4666152962762a2bf7dd9fba4f8703f015e7c03319ea58ff570482be24c96976fc383a57cdb081b989659e8ea3183b81c66846cae22b0edf72482b5e55edf85d13eb1cb02e778a3be2cadab6dc4fc9c84360d95f08d790a4434273fdfc0c37815d33696bc463f80d58009b3258bd06a161c4cd687258cfadc9047f51eab80d626cff14a18a3dbb4a1b284795c0a7aa645f70f8e67ffb6debc7b46503d4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.635059268Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/array_metadatas, Name: checkpoints/5888/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5888/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7d25d2a57a007bb85ed70494f578b64321fc657899625ff1ed135ddd027b22f93de68ab9bcd63050eae11be414f6ed844f2245cf0603aefe51f62f435ba9ea67443bc1156470c364fc818bc7b14a173e0eaa5028c21fd26f93dd75e88e853d68f9684417ed42b0f86f843731d2581417c8837e876406bd86a28e1577dc618cc77604262aed10e8d04f1ba6813c4d7405d03636654ec9266bbe9869eeaf687748b065d95740e07a65ed231bfa7b68a8889a51e9efa3c30530b4ee92e419d6fda372914b28b63eae5c7673507d20732f5c5eda74a624c04fe3cda1c7450297037c905b1616f6ce5391e23834ac589a622bbdd530032e2a6237b238209107964748&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.635083766Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/array_metadatas, Name: checkpoints/2304/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2304/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=00a3a4ec893dab6af28af962846284473bb95c9fee9eb1090b2833e4b86935ba059fc0b3a2e2aea69b330381736ca44d4c754bd1fe754394ff88c8757574158ab3880c1d644283f2e5e4532982452e3e76509e452a447a02c5a6a1b2c2408a95e505cba7e0eac7a1278fd59650c5a243cc8f9633512557980293bca427ab30dbbd839982bc71da946b0dd4f6c3380bfac6d575a213ac4c8e333c6988a910efee69cc48fb8812e1cc4425dacd302965ff98fbe73c75085752f0e72aed5355889629034e0c0783c9fb26ca2182466ead49fb1e91f1f8ab584ef9a24d4d49d9d46f255e0b6db76d54b89a19875b4bd608b2e1bf66e1682180ac3c13c68e69f5b60c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.635098433Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/d, Name: checkpoints/1536/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1536/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3b849b7d5b55aef3f6cc74baae24e4df76a9b55f2472fed3c77d83fdb91dc9e9b5cd52d9d60043845e5b313211846d8cbc77ca355a4af02c6e2bea6ca7986d6472ed8fd420a5d002c71dc0c808d9c22ee65663f5c180b3909091c9c3602613176dc53c5cbaad84d9fec9178f23ec2434cd9b7bf5b2ca54d013061275ab81a728214baa96a07e988eee3a8f9ac191ec88cf1858fa209068a0c1e6f3cf54429f08a387ffa8732e8dcc2791d56e688c00c98a8ada7c93e6062df778a857585be12e0ffacfcc03c5d9d408e1e193bb130b46f7aeffa2d9ad28fc8ad26b8fbb64a88d359037113590548d04c70792a5c591d09bf277524343458ede2f87e207ffd785&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.635122345Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/array_metadatas, Name: checkpoints/192/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/192/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=8f00c7c38ef42df64a2b7ae389ccb0ce712bcc5361fae48604fe349a97ae6813bd2258becd717e1b4cf60f1d93f78af93ea27d12d48046c4d6b9ece38572c2e263c79574d953ce810c7542b7381d1b881b17f2c7faf74c32ceaa7a8c142b4104378b0489dbd7873da5681f2d3fdda22470103094d901724f5cd43c16e38bca25e3d3ddb795adff807acd1c57a3531504a48bc89c60eea03050b74f1c5eb65fe6342bd3b7de4c64f1eaa2e99ac6e727f7e3dd167142ec6bcf0438b8935a1d00a28a75b53bd154a2ffe2e539704e43b5ccf35859d30ec7d4ced1972a4da386bfc8bb3226a8d005b1631a9f2745a204d3997b62d7eca5b8a0a4c594e4657b51180c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.635127921Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d, Name: checkpoints/5696/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5696/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=53fed6fe0268380cea7aa6f48f0991ffd3bf304981e8551dad5651f42986ade5e92beafccb368bc4468432a365bb839cf71079871acd773db7a88ec77337b8a36825b69aad8376ffc16e35e9b4c62ea2ee83a036e455436980c61d3e74586b35b1ed355fd53247584c55f07db32369111ecf8073c3b973ba5e47daeedf55ace087fe9511c7024170a7282a6326d01505d6c877d314d7f59db8ad852af07653078793b128ebcd571fe91e3b984fb6ba599ff21f3af0b70fbdd48d81ac691bbff7bf077a961cb5beb6369b9fdbeaf3ff56e322ea921b8b7ddc187bab0c591138f183cd90036401dc3a4ae4dbb21efeeb89849a1b2fc23789f041b853c00d617dfd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.635181097Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/d, Name: checkpoints/896/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/896/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0b52de980b28387e1a0e3526f483ec4dd86c1236992281a4a4b13918f8a098caddec311ff4cf1c58798fcbc244a75f736cf96953a9cdd44df7fdacb4cdb4eae1e9b57f3955440f331f3ec584bea326135feadb0bb9c1847a8a996506cda74eff6b4cf8f331dd65b5e2167aacc45dae5255139048b17cee4f383144f1b582d05368ab8531f395100272dab7fb6e99bf3b9ed7a2a475573dbdf72f4d819c17f45884debc3cb73763f4f9cac01795fb864a8a02818cfe03e82ff547c35757e770f031fe45223c4bdb702b089a0dc4070ce7cbbec920acaced884ce2f7a135abe910ec36c66d0dd202af93e52b28ab1a5e9c34112da07538880b3ed67072ccae8383&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636337443Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/d, Name: checkpoints/1920/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1920/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3b30a13215664855c9221f7beeba8a0155676450c19fd4e44a23cfef78718dea8b84e1bf52a744da0f78b7c5da73fad1cb3579e3425c9fe292749cd5712976009db9bf705b4bf2c89bac7314176f22348b5be0144d026daf346efcde524da99243793cf0f7f710d2b787905ba0edbdf771222ad361487b23b77bd45a2939d967a22327700e9f8e5ad2433a30353a85236aa65ebc6312baab7b1e54d533a45e1182a824c9e4191ac0243b8b633ae438a2e5df137da1c0b5ec6ee8fa7c491398bdedf0078f309a44c922d35eaae68e7d25f7af70c4a977d532d5eea390ea6353c8d839f24bbd85ceb9a8af7a80e5939a49d493dd0edcdd68e0ff2692dd7a10a73f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636331903Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0, Name: checkpoints/0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=45bbf6447923b1802c9696e57b0770734c3b8f400646e32dd1d0b8cafe77bee30defc857ae99a96f55a3876280ed648d7f7df83c9600f5c39878eedcfb72b26e5fa31c2bc99699faac484225b1be512e1af6d3e2966a2cb569fb43927b622bc69ccd9031b8dac8443dfba57f27667807fffdc9c2b41f7c8afaa7723e55cf60074c3167b7dcfdbbc68690b239619170702022ae04acb543f05728fc07f1309dcd1d536e04490606a220ea1f913c95834f038858c14f3993a634445f7470b24794bafc27ca4a12c9db65b42a02d83d617cb4569132d72c1592026603fd029db906c01838229d7775a2a1c6d0885fb78ef8c0bc91e552db8cf5350c7a9f73308b8e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636332936Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/d, Name: checkpoints/4608/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4608/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b1ec7668f75c04a89a0cf334b90ad8ab684ebaea5341e359787054b76aeca5d159fa43dc24be39e7b3ba567743bf5ac5fa3adc5b40389f9f05dbf3d6b8e3667e312f841098955cea99fe58d31e4431d6c23f41200a3adb2356f8b9699d1a0dc4e44e8efe82d5e83893f9e0504576bb13930df6b5b4794289334435c7d15ad6c2fc3a1fe2ac78decd68581af168b0ff7e1bc51bce6eac949f1cd3c246095f7336a91af44cd4ba7ddf1e3caad127b3ec5895d8880d1d9e8306dd83fc0fcf4616f6aa2c0229b57518fb926cbbfa15c7961b3682ac63a7eea961527f81177517a8b5b9bac31d01f41c22dde64b26d23ec5b659966a94624a0fbd2650d7f988d90cd7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636426068Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0, Name: checkpoints/5888/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5888/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=355cb7e06f8fc9d7b0e80507fcad06b0daf55a927a330f899926a1cee1a7ca27d713630041d5acca3381f0a6b95af94a8eb0c91568a579e592925390f963f5cce970a59a682d49333cb23ba559b01790fd7ef4e4d2cf5509b6c539964e7bd61a468f045080aae32c2cd8215bf59fa9ec46a9a3c0fa92198ebd66246e4229289e6b0de5a89983a9e2190c65267cd1cc4275e016c86f0989f3f1dac535acfb6b6ba96ef9d80c4b04851bdeed4f5679119119f44bf48046a661f0bf653781956d681b3bf7e97803c18c27c7fdc6a334f031a523236f3c9d5e6fb9af484212b5fe18e6990fab3fc5e18de08251cd1663d1349bf5e67a6c406e17c708cc3b1c996e8c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641469065Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/array_metadatas, Name: checkpoints/2752/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2752/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=567c125e75659544255b2b292bc3e5c46484a3c55def9d03bd577af4d0b4b96205dec0190d7fa0092c9dcec31e2a20ae6bdb90d8040c7bf5ea3b9b3a759fbb331a3f74b470702aa4064c31fc00a815de75c784420a9e5d7e2323861d471e35f9cfd4b1de0bc2fab46ec376179608092b9ef37ea70dc66df68f33d2036fb9de780ae1bd3cc3893001eeb803f06f3065ecf95b210328c3bc5e6aea6d45b00ae8efab63574490660a38f69b72e264406d2df5a6aab8603bc2c0608fbe93ba0ae05927466c60d985cc0d869472f135add1233e1850cd0215ce865439b826c1f0a3138a081edae2daf51badd18bfa6c9564cb2cd2e579067f862cab771f6dfe795714&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636423703Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d, Name: checkpoints/5184/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5184/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=217e14963d902f1c9565b55adf25732cbdaf60a1a451fc24eda219405ec247c6746cfa136d4f5ffa64f228176cbe4531f26ba8fe1fcd6524a95cf5f374927cb9026b357580461f74e170f2302e2d06e43bb7154256b91cece96af730294fb7d9030ca49789a0beff0b7b4172c7e19c966600f4d5cd03499bb6d15795c069f9411c920e8f1a8fc3cb980a4d6183e269cad934e67c3c8bf0c3d9b0d135a44c0fa22d6cce32ea67a43438c653257fc363e3e3e47140789c46020ab39028bb8ab2ca64d0b75e8fc52c578ed9681dbc6fc643ea68063f6a7d1426d53e6c22ff31647490e5c037f262dbb4ddc97ca3c556f3ab65adfe31c419946bcf96b2851ece1da7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641533642Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0, Name: checkpoints/6016/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/6016/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0c2e77fa3efae32dba5b1b90c3ebf43e619df8086b9b43718472b6362bd434d0bc834920c1a8b905918f8f18a27cc53e93fc185e3027eeaa79fa257354278b2a99f74eb29a255c217a63a539e3981b8af8c48eb1719f1e01e408dc3651d0b0c8a5772d29b937b583bd57cf001b4e9363c915bc5bfa8ee805b3e77ed611c9af1a0d64ddb3a52b12a0321cf34351c379d10f5be0840bacd3ef820eb6dbe42b40838f407eaa6165be4acdb9431e02347728d821f1b2948afc0b1af83ade6418c0b2b1f1c0f09933ce0460ff0f2961c590d8f7e4a526315bd3f563e0f1a5e8a14c6991c3940f3911dd055d6539950c83568753327021509026b05b72b75fc73d8f5a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641549495Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default, Name: checkpoints/2688/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2688/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a5e6ee09e3e24ab307ce2e33a37fb872b7115e9ca70d737c9a427061442337e93296d5e5925862360541db62ce609c023ced4a30290beee3d90e7480145468d09ceedaa447af3fcb63e1b19ff31c150ec7c6eb24ac6866968ddedcdf62946dd3327a7b66efaab7ad2e9b0b7a1d6f582bdb1e3e1359d7925ac1b2ea42bfefcfc66ff5ea09eb2fe25dc1378e8921d1b5bee1106c3a51f21eb9b491ce48845bb2c082c1c2a0395ae25a333b26818b0089849a9b9d74644c07f960b3c51097890e9ca0ead476d6c61ae8da3b6632dc5892446a88fbb806ca17184dff0475cd82a025a44d6c47bd8ae642d50f64515621edea452faf518ece025538248ad8fc59fe7b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641591303Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5888/default, Name: checkpoints/5888/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5888/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b7785855c059ec21eb886b7874e3eecfcb2dca4fba9f23bb19501a5b30e88ca4e2d81810693565609be3f38f7aaf0ea1d57a266c9a9050201ee5c5d9d16ee57b119f24a1799944023893a8eaaeccf76eeaa20d51d4b77d60909db933648a008b5a2ea2c426f724fd0654ccdec2fc148cd9088b7b60566b07e821ceab3c970e17a0796ea7b8a49d95000031b37644d31fb7a27df3620a47393a46f551946c0ef3d0dcd9bf91cccdda262981bd4966f6b116e8d4d5ac56b46db20dce3fe77fd15f5bc463aae7c9f1822a89b3fe74bfa7f223de43ee8b30c964df9fbc31426eb2e9ebb1c15e049795662c28d0fe82c4409796c0bcafa89a013d5e662bece85b934d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641608391Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d, Name: checkpoints/4992/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4992/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=1da7ac9cf43fcd12312f2049b842e67c32adba796f0cd67010694d0b9cc906547f8e22a387df883c6b3413b85e066ead6955f1954830aaf25221872155d12334f19a57c6f01ba5cdac111e42ec0008179f52393b963346807c2d54ecf310b9aaac1074a8c553a46081c89d8732c2bb0a3825a75d14438f78da511150cf74aa2f3663998941e1ad3d76237981f7e9bb8117369863bd16951ee4800e785f1e154367a96e50567df61dfb0d8bf1234ce3849d7f31368b2a8acbe9bcdc420392fd6e2e7e487dc594bd2f7a176846f73eee586f9277b183c33b0750f18835f18a2a7db455360058abd67bca94abecaef328fdaaf71e81a259fbabe8798deab22ba7e4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.6416475Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default, Name: checkpoints/1728/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1728/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0fff463d2747bfbd5445f7e1996afe7e49e34510dd045d42740fb3fe266db7c0af4f07ac7e01ff8bcad9b4bad2e9c4f0b9e24debdcd3f0f3118920f6cf87d75b950b14b66f06d83e59517a3fefe41b2701e7ea36ff8f25ddfdb163877f24cdc3772a936f3f69b9eb2e9770e09cb7378570d6f4f67ef1da1af00038a70227626492cb5c35e130b09c73a6dc3ff5b2f54c5384980892468258ec25681f0e0d97048be5b9f597ba50cad5ea6accdd5b5e160320c4c21af8ead9271708e3d563f84f77689158b72ea9664423eb5d902bd5289618b7ee0cefd956c08add5210c3324e67ccfa81b463ae9d61c6dc36e747172575d1b49bcd449cb3eacf5b72967454bb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641687668Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default, Name: checkpoints/4608/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4608/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0766f6c91a119b07368c0c83a86de13048036d3acc5b475e614f260f94b836dcc274f96f9144ad4db7f4e6c9577f1448f18d026b96a1f2d19fe0529600777408eae021f60f6bc4473e3fa2c78b3453c790facc35745c759dfe8094d3aa673587c3e89a4b0f58697821713f4c724efac4c91270c1de23264a74d62a9eefcc3a4aeb20d2de77698d149ac1aaf73023bc1e0555e4e1ceac13a2322b24dbefbc1a6628953918398f616115025ea30c681593f02d4650364dea2bf56d1c309f28d09506da0ca75039d751cad2031bc523664e092b017b24c45629ae44d3739f89f55965101ef92c998317b551c1d89ac619f2cd9b7cdc4260a08fb206e082cbc5cfb8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641720261Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0, Name: checkpoints/2816/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2816/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4d4007b210f5ed9981125c5c4d82cdcf4da288e9abb94ad8e89884b6eaab01564d6b3aacac9a7dccb7f81cb89ac1cf317abe60b973b0e2b0ec3c85233316f354b93b9a72382ce25fa7ad21b4d61155e8ae140cd7551892dbae0c258730fe4b6db2e144a7933a88bf5fdf9e2fcbe8251a08400f1fbdeef44819553fedd379086ac3887c223fced1cde41c9dbdf3999da7459f0c639659e08375055d02c7264769550e0d91725cb4f39cec5ceb99cad025e641403bd6283c0f72548b6c7e08f5ed9c17632591e0a832abaa140c5baa283795c1ae22f623268a118627a0a057ea79c9c0b20328b050cff8f3c136a4aabd0db7b6677c29216344b5eb55c3cbc52f9b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641719804Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d, Name: checkpoints/2880/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2880/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0bd5c11606c803f85110749aa012f650faf60a46307de59b6d9c31880f2439f4145af43701cd798faf221a6bc0d998cbc12ed85a319a5f8f89ba5cec71f350b6bfabd6c615450dacc943e0b28d23d5485dcf17f3aa9f699c6b010796bc732f863f65e420e2366790fdf23da5ade37c1ae47a7fb30c27e5bd395eb4a315bd478a9310ac90948cc9122a95e2f5c9d8fe63eac1e3f1335b223b63bd167c3ec9030607ecd23d0db0283561c5dbc8bad6facffa26b4652531b8c5f5342866f88df764e63494e79dc23f66b0d26cc20485e3243279802bf1ca33080b2f9923989b8161f39a7704b9eaff38e347d506d07025fa4e5d8964f529d05cd38203638e65681a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.64176178Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d, Name: checkpoints/4800/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4800/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b92a5f2194481a1e7a8097e0112d35ac7b23ac50ce9e6af71715e15a89bc507bccc048ccdb08d30ac92204aa3315fb667efc2726474a8d938eb65762f44c2fb9cabf9c2fab08ff238d2d76487cf24301481756ecaac49d685c2f2a0efd388887d14f3fb7c3234822c9da8956b6011125dc33606bc688fdbf8228baf0bb1a4cfb025070e5a87e7dce645713a17b1384169bedd0be99450a09421f78cf9f407eab3f4088b49729ac5855361b0a164d2142330c63effd5d67cded33a8214763756a11062e6aa02da09d5a3b1ac16e50548426377d5e44879c7a3e288e29ff45eb01084867e04f54d2c66cda0ab10e2947037270386e59a81ae23cba58b9ffbaf850&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641796801Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/d, Name: checkpoints/2368/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2368/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6147302cc48d6476a0e4ef6e5e58d7f4daf2fdc7f6253c62376a8892e79cde22dd9a4e1741f47de91790c036dd1b4bb996f65042ed5c0e08296d67d1086c645c4dacbfed8157dc59c3f769352d5d1a83ab1684a2efb92044f6bb8cd6c40ddc03dbb8c2cf768ea8ecb90b500d2e7649b91df870432a4b1af30b2478a5819d3be7f9b7d93ca2a6b4ceda388ea37bd62b34b9e261145b2dbe78e1c685bf6a11b0531e64bcc4cd7751ab6e6b25dc94803c0789c8a799deb4eaf4b3e277ebe9aaa0570cff147a292c9dfd256cddc1c66f6234bafe34d9f91a5c1e0f6d1741386879c429f73bbfe141117f452cca7498a08ea717f78d0baf1ce41254ae17886b2cc487&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641807339Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/d, Name: checkpoints/4480/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4480/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=24b33ce5d0d42cb42fd313f1f902092e56e18267eb6df4f6f1d132186624eb3ff04661dda0068609c5d130d13d097606f3c18a432d905936e2a0706191965c582ca4e33f10f94f7a50db9bdf6a52d0fd42529d045958429f2ebb2b946c3428c1d3c983b5f3c39f7e0df88b1b4e2179d725c08bc393fcf3c588b1705db738527e1e7678f08029a5b059481b32fdf21c0f4e636486a901cf081d5b1a6cc3075781a54a6533afb70dcbc9d38104c2f499f5bf574f885efedbcd2a5facc8cde95709ef9c26b70d7aa1c62ce8b22de48d1c1681b5d0660249f3ab8a4947e1159610c61f3150e974eec872c86228f46c3d21d0f378891673fa56170db6f9f8a898c0aa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641830891Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640, Name: checkpoints/640, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/640?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=2b5b99434c18ddfe06202f411a0bae22a6206eeeaa0981f7f28c71af2e404f57860f5d3a148fe3ed52d68316776eb20c17700143bb6b1ed07125cdee4c6a0b241c0ae73bd0f451d7dac26fbff89b9278b2bb1611d4368adcf79610aa451c9d83e02fdc6289ec827cd393e847cbf3fecbef7b83fcd8fb21f943ea9e0af9d7aa0371e5c526a4bf1f1a872626eded4332007a8bfee2a7959dacf5d521420b90ca6e96b8cd8ab5d604f637cce0b4552390f6fffa791e0d449d27778e9fc0cafb3718ac8d9f0111b6e63b6406d16d1c585cd0a4b4cf8616245967ea00dcf046f058bfb03dd9488cb58e8f54f7067541d01d5a0f5131550d82fd048091932f0944ceb8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.64184544Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default, Name: checkpoints/2752/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2752/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b7d424ae6d1f8c1e9f730f856a3411f304e0928d262fd4cb6417d0fa82e5a9962dda087c34a26390b71022df4d59b5885d24d7d2362ab9ad2127a04bd6be66a6027f9cf6d21cd3d238f1b10750c92ac66510e33ae7a6fd0e5f05463bb2d1b4cc17db20d61d3161e0ed45105726de4ae40a817d71b370179a1be09f8abf1476a9544c0c61cb6b85b565d9ffbaf04f16786b7bffcb4d741f60e196d2d8aa0da1accd636efd161171891cdb0536d54fe71cbdf48834993a2a9087330a8f576848c1de0f1fbd674d9eda1a632c849c2b8e2584261d986b1ec03546e18e6e0351b36e50330de955bc97aab89c50c08f91476291c4324f41dbf2299db77eb17dfb778b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641887585Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d, Name: checkpoints/6080/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/6080/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=99232ae09930e008d33b598e777a62c42bc3b2fec26959b335e3047d7d092fe935c3f9af86638fb7a7f06c52ee2ab2274df9184675a19acac170119b55a20bfc9c5f6c616c53ac9a1d7acc5103d5792f42e35762828a9124e5fd81647f52369d5c7241e4737ace00b9e79515304b07bbae83a7ef8fd022472fa2010d9ab93a8b8fbf455d534667ad491627244e87d7f791a53ba9b57133b204ccb0861ad93bd13d128b6dd24d92c73e5cc73572f04bdcbbe80c87ca63230d17f6834ed9576234fbbcc646312809cf7094eb5cd866291ec0fef98c5969fc638331989748b4775b5e5886b84365084b2c78f9e31725b399107fee74bed58cf299e013fdf1c33ba7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641904872Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/array_metadatas, Name: checkpoints/2496/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2496/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5f949e14f95bbd60485226d09fde9b57b95b063cab696b9a7afb2326ffc80dee3aff376d2c577f42439e33b98b6f8a86e67b0e97bcb2fe2d02e44ccd147dab4eb81ca8a32665283679e99149d43fb86c488e53264336ec959e8460f9ae40c9926e16d35c37bae1e9bb82e4d4302d041182a651f419ce55b85aead8949532c11525b12870dff0c5cc35fb6b83e31da0e2353683bbde233cdd6ebbe7d9504fe6b785da605768c59c40f9181cca41f9b4265e95fa5b82208bb37c9bef65bad52e7d5c8386d836405d66ec8c2180df2693a721a00eed45a61bd8152768faef492c93eee24f051da709196d76f717531755149de498bbc4f92040651727c3f5d0248e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641917477Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d, Name: checkpoints/1600/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1600/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=02cb24dc4ce696e3e0140e7ac59d58485db75c97c76cb116f925c0662859b2ea8c2a4d91821eb4e16e5579a0ac9e181f811605ef3f97b49d28790f054aee960a908c8836543d8e95cab9f5cb9cc81e2ffaeab4bd492311e640ad47e5082b080c9e77e08059948933bf992b9e8e2a8133c64849622d7cf9d57b4c406b081ec31f7b1381cbdee561205ec21d41f3918f720c0ed3b354c65ddb36d4a1e2b425eec2893516edd6ce7d15db9358b42725422db4f587369c93af823946ab48c9d809b7add3d61d3bc91081375b85fc2c0755717ede2eeaa7bc591cb3d05d1a0b09e3a5f8c5225d81472fea2bb0427034c99b4765b33489f74317c2e1b7986f3ebb8fb0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641935006Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default, Name: checkpoints/4672/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4672/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5ad4ad16fa617fa5a287b4cbb046866ea674ba2263cf1a966d17274f6acd459b73b314cf5b91e240cc620c61319f866a085c53595b2ad56d2a10721d919d26e7677d8bccda85dd1bf604eeae7c028924eb69978985ad7c2e92279f7545a9b47a990617439591027284270d724f3d3d85de6596447326550abb43d3a99f51a7d935f45a08cc25e28ba9e60fbbe9ecfff17daf7ba989ab04f7671423b4210710bafe5279466cfe5c55e79e5cf983f89ff5586ba6c0093bfca95d8310d87e957d238ce6517fc09f1b8a62b755c530d50ea3ea9ddb65f2a8bb663ef368923f142d0d7afdb8f9a43bbadb8292da7af64c772ddaaefe4ce6ddc676fe45dcce99253134&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641936712Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0, Name: checkpoints/5952/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5952/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0587e03adc4b3c768ceafb6c9f4873dd640107f2b7a6427c8112a6683b5e4e6334cabfbe313f7dc5ff3fd304760b9ed73dd6fa17e1f526c71b20085ba2d55249457fad0eda453b1536c58032f663ea975a5fd52f384dde898733b944f7c3ad3374a1e6fbd1ebf3e33278b0a9d123903c25ae07ccb28eab0b54c8f9807a0431667eaf9fc07e2015da05cee92a8eedab5cecaf411f7b9d9046c9650dc64cf503676dd15da6eaa1ba88cc801570bbc612e2563c1ef7f9d1c153235238029e73f811c7aa4a1cb087b6fa54813c1f68537611ee8d7eec0e59a6b02083af2537b8d9dade3709302413ce8e80b447bce797a275df2b7d8a2457fbd3e5759e78c15704fd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641559746Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/d, Name: checkpoints/4224/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4224/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5b21df535d4896fbe45c49329ee3247e4e59c79fc9a3e90915ccf29e44168304629a3d9b64fdb481459fceaf769f28fa28db5ba2688652fc0ceedd770ec127bf79559a82a2350adbed223a2f966fb5df0322aafd9e03832acd614aecbe469cdcd97ebb18b69dbb490dd22c857aec1c1a0e38c344a2dd1142dfe8843bfed5782b817bffbc4df1a79c45bbff147e260f5f73c8c2b4b0c21c88c93fdd8515c1b6e70755c390f3d945dbbc7200274c8cff01470a9a06cf11da883f019e09189eb4a5a83831732c7a503f3f678ea955f5217dc9c0e9ab6f2da1bd217063281c605b3f162cc7622339ec7a4f28cdae5b4b62bc5383da101d7fd6bcb920ccd6a09ba787&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641987252Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0, Name: checkpoints/896/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/896/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0842239f5e492d5ae58542a474048c055e7d5738a8980da7d142b500d957ca26e372a8fab9166a9ef0d0e9b92a77838d85dec376ce781f0ecdbde456fcc30d518d6fcbfb31879d0b7b77b5dc316c29c27bcd4838ef0f90a821974dc47d99edc53e0af71bd901d5a9611b3daa6b765964a392df2ffc8c4fdd209436227e6fde0b47f1ea2ab11140efe078b334c2b83be6e283854f4b4727d3f6516a727b3310ccaa96f225556fa3dcde5eb8b202e3757a4ece0d2cfdcd7d0d47e251ac771ce1326788415b23966cb42caa1441ffe88c9c62c63b915e1ae72eca9339a88e83fa004933f6d0662d11f2f536ff6b49401c419df051dd6d56b478ef830f2573a40a6f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642015819Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default, Name: checkpoints/5504/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5504/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a8e587529a892d1d67d043921610e54d3b278402b29f452965a472c0db9b6d21ef332508488e17a45872eefa1030735979d4f8292932921a0614db23e79e4a76ab97f4f07621203cc410cb97863807413dfc0316314ba755e92d2112add36d23c4a4abdde2e8681c9c4c37c3b16c4a0740497c1321384dde012030c3b55428fffff504859a007298f98a2516efbe7dcb44bdb8235419c8877c463ca171030cc6eb4d56551ff3117c88c75bdeae5a054ab9018fa691c6044330c5130777208e1f94efa735ab9b7d196704ec0d7188c3eb7bb1e8242552754e00d1e5ec30343bfb6678f8442195008acedd48f58e90fc6d0616822671a2160397cd31b1000fd1f9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642039545Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/array_metadatas, Name: checkpoints/320/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/320/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3b668f50d855b2a0cefe32c5b73c0db2e96f18c5c5276917c418f37b132d5295db81690582df96ef76b56ff4f9bb8d285823cc1de2e836792f8534650867e4748ff1c01b6feae454c8d35facf5e9076166c8bb89a3e8e115ab7fd99a91e886bb6633ea3dc0f1ba155b8e384fde671319e15112ccb9a275aeb97e1ee9e036be475fb28d75942a9696f90bbda1e3c9a9c4538a0633ebc0a2abe82f8f1f5a57dbaf1299fb74c6d55e91d9d90fbb09af6ee7300b4a949fe975e94bb72bfc4b3b2f213147183810a45e4118b729610eda3cf87060cbb70ba8f5a93304c0d581be74d4255064e1126a3dc037eb34a8fe8ac946e2a20f433c7e4b6633c9d8c4e24f9ce9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642054653Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/array_metadatas, Name: checkpoints/576/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/576/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=40078bbfb369a146303438a8a1a58b11732cf76c224a71a3b7523409854662b967e04e6d96ff891442fbb58ebcf0bcae08c9a17e8edfb746cf95e85ebac9bf4da443e3b8cb0fcbd65654be393bf8185471146b550bdd088ecad1707aa7976b3b5de3071c6083b1676e984bd389f97de6e9bbb59839bf61ec775429d0bd91aa86e5f2b7e3b016908babac4433f6d6061554b99efe1fe8dd600443754b90899900ff8434b7a7444bfe036ed664e80940726eeac4e6586be650eaa5e91ebaafe6883fcbb9dfbdc6480c22d848df595997b93b4151220fc631401bfa82ecd2bc2a814724730a0a05987ceb10f24670b75ab1351f870a3871a4eb76a51c7b0731f0c4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642075993Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352, Name: checkpoints/4352, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4352?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9e5e4e2b75fdc830c02eee829730323ca5870f7c43acec967449850cc3da0042c3d210bb783c0a2010dd3b26c302e63553312c532977fff353a25262b1c64efa312a4f6e53b9379dfea9ba4408500f34eaf17db3a79fa0925167dd39ec8590711302d3334d16df784a4de64e87523fa0b57b3da68d456ddf93d28139b9bcd9463c17b6e42647ff8322eda80bb46157bb0a8b16efa8f6f5245a42113b44716b43fa4f5b5356c9a01e235d72d0b330863f417788351f21ccd68e1368c0d6c45fead121f8ef0aaae87779b3d52521fd4b036c7953f30bfa9dd0bd18412a109e6b1a4916b0480a82ce0b2219a9a84c0f453d65922fcee9cd15c8d29c17baf8568d9b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642101612Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/d, Name: checkpoints/2624/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2624/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=796258efa96aee81451dedd09bde01448a5a897514a2db0281e4342d70c9c1ccaae4249abd30c591717f7eb66d1611828b60ff5b42213bbcfd4a7e548823217e9b5372e058e76ba3881880df8ca33dfae9ea69967d5118488a4dacc7522f992d5c8994b75d5d727ea6ee0441c979c30e6cea74f1ff974590720fe72f88a805a4c13cdcf6ab04ec6d1ec67e5390788fed7af7e4100846f10837b39f28424db94724ba3b482b581d724cf642156b7cb0d34dd495aeb04e5abda06789a3e3c07e9384c44af823eb12afb597742c4682bd78e95f4a315ce91f0218784ab1efeb5bb37f632f86d293baad1f16f430105fac4f90d6e6793ec61067726d31c88b33289e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642081711Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/d, Name: checkpoints/1024/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1024/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b8bd616a4f9aec9eb32ce4032eedf2ccdb7aaaff377bd5ef0982efde26eebca4e6a1d1791e6be905c40582bf2ad51f3edbf5aa547c57fe9f8f985b7803a4289dab3ded24873fff8408fa08d55a7f77277f0d66f9bddacea3952cc06a866f98939296392500952588dc77c5bef38b930e4b91cddfc5c2d1357bbdcd2a17d700755f7d3edb9f0c5d17e61137427a646688ffe73923bf156f1f4d6fdb64b3d7eb0ecadbb9d2cee13886271534bdecac3076b84832b037abf888a8ba57374030f44da040eaf02b815728e74d72cda4bdae105f8a6904517717bce0eefb9c081bdf1e71c70970695c7208dc6588b655407196d6c5d99dbebdfd05c85ae8ea91786cdc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642019906Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/d, Name: checkpoints/1600/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1600/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=19a230461ca391870af32df62e99138764535b6fb1b7bc761593b8fba95698de05ce58a1f42635f8720e89edb54e6e836af09f1f7b1fb9be615f2c272144cf5616556a637595e5d9b3d7210bbcaf5290a6e7341cad75a78aa0fc521faf584b61daa70d03acba737777309b208a82a396be6b0392be2bf9317bc54653973cd451292f5a56b719b04e6df7b6f2aca3cc1108f9c863665760846d08ce6b4d009eb08b3060c256088c05dcfb90ee7a5de059e7a746fee3cf18796af4cdd2851bf46f845a1376861ccf35f921f90c66b3e119d6784ca9e96d276451ccdaeba906dd58b315140890ca89c0611721024230ad9e2603d895f8849bbff816c7d3a1b83de5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642144579Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984, Name: checkpoints/1984, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1984?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a6911cfbe97255f3845573b18c9fe779e772e014f5aa38bdb0b2d1a48e907250444bfb142e106675d3d82dc2c13036f53551f32865fc97b66f0745c6555d65b2a44f1344854ed4d7509606060c5d774e95d11b18abd8d4cdb012c5ded17c0832fdc89319d665d06d1f070dcea0779ef12e3817c3f5ee6db9d383896e5956492a9c477734c4b81cc3ccda65b9c764e0c6e62e0ae7f1e5f5caf395eee96e29b1c89b0eb6af77350660768b945c139c35eb202f47cec3d71d40a769d329ce43ca475e383792665cf780ebd3b71c56c4b4efac59db9a44930b52b9c6de71016f09d724b478c9e9d4f465f25a42dfe340bb827fb0d424bd687a253196c391acd3351b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641664177Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default, Name: checkpoints/5760/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5760/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7279df8b274bdef7ead1ca3a50d7869159cb52210bbf72bbf8f0a0ce66d041cedb3949ca88e30474f97241bfbaa29503dab04e93e313e69e82b675d96a7053993a885d9ea9b5c9603f66d5bfdc75f993011b8f43cfbea93fda1558e606585d2016ed433782dc74c73fa66a063f2760267aab6656d5327d45fbe147b61139d707ec41559bc713bafbded9c7d18e9a7adb64ca71a3f1bb1fe03183891ea4a5b7161d36189969053867c18b1deb3585d2e09956a73679122545277ed58de74787d33a140d2e13215e31b5fb6b11d6dc0693afd66af4e19f4a03cc78b9fa92ba9bbc0a10989b80b1814f3830c0caa52564ef358a54093476660312f81b4e2337781c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.64215595Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760, Name: checkpoints/5760, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5760?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6f363e8da9897ec7b617302eb14684fd1c3dc6df548fcbe92a1299ac3b117043abfadaac4ee91d9e7d76e9dc5b96961882376c2e03b7add93945c34ada4ca393d0c02138f7d3382668518bc664c5a36e7933f2a61585573d09bd89a3d2df5a0df1b1d3f3217871337d7135a75fa4ea64d9771189688531c4a4eed4ec76fc5dc5cf6eba818ca4c57407222cc4ab5c92c5cc9cfae15761b5ce0a8aa176b5e8ecbc9b02affa458557661278cf1f3a1cb64807a3c56eccd57b43f68930edf36b31e92755a9f770ecf185e062fbc4a43d51b1b26b1a70da4d7cacf44292f0b4bda28dbc4c8c1308cf001b8dc0497bfa4819f4dcdf2784616ab1093c6111fabfdce8aa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642173461Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d, Name: checkpoints/1856/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1856/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=06aeb322498295c32698d8a42f3d18f922df6accc0105513650d7e9703808a8565105527f182911c01ab9525be32f8e0fe0e9d00887fd338fcfba4d5326f8852d7a106e5031d355ad78d5a3a8d09b6c2bac3355cde9d3dde8f1febbd4f655d8e408f9b42f670cdb272cdd876422d151de2cdb0d704bce0139bf8c21cc5f3be46e8e4a10b99ec586b8cddbb309991d3418cba62ebd5fce79da999dd3902bf876861f1e6cf59ef32634bbda439cbf2f8a758021bb7ab3d1cad543511a48ccdee3304bbb7ca06a2cee45372ca255a4a4bd95e677f0291eb771355755dcba2f3bf0aed83556c14f71a5a9da5bbc10560e89409f7b457e9816a4d76a602990ea1b440&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642019905Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/d, Name: checkpoints/3392/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3392/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4cecb2b25574396ca03ee2c0bf70192058fed8902efdb722a3a2af2449a7f985e48d2996e842a56ea09df211b192ad2577e20044aa6e5e91ffbda2b2851d9679eff5360b6a99fe7a3787313b6b14351f968c40e8a280d8010819d624aae5088ff290347a2d6fa47aed7228da3e3cee798fa366af16aec8a17dbf630de8ea38580ffe60836ed6ebda0d2db91ba484007d3cfff6e5b472519e943996f1bd7ff593d5fee4e6b95c7fa12cfdff1ff62ea93c57ce04e880100845fe8fb9784c2ba9a446092dfe8a9743caf68d6ffd57e58040af03d61d4a03ab1c12541c254439675138c76aae683f63437280c7e6dbb196803c74bc5fb3db70c756f09f22c5a4a25b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642234175Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default, Name: checkpoints/2368/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2368/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5a33eab9a914709571cac5654512af1495040aa34899fb13758913601af53c0b286bf38b19ede4ae2eac865d2aac676e82b0049637de2ef3af91372912fd8846d3c1e420f86475eb02d59c90116ea100f9bc77c6544aaf5c67a5787b8ccef56b1ee304a5b9158ab10da2a9163dd2a822cd012bdfae56f2b428b9c8d1bd737e1297d0dacc55c15a8d3813821444aecce913dd6843a1be82adf46b4129fd27daee78e17488944e20dc5e696c5e6b42b863d1dd418f1a60fc2bef39b8ef2523c6155cff861f44e4e83d444ceafdd0ad74b8491b5b25f6df97550011a17a1274ef23c2d3dce8d6a64e469730e8b597a2e20244cbaffc40798a5ae9944e285e34d79a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642237295Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568, Name: checkpoints/5568, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5568?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=29a52fb3765b358ab17b498a21c81860f7a8b5473f88e89c655821780ca414f3bb75b37fafa3a3898686897d32b49c3753c4a20e237ecac48f74ef739c2babc3b5d5a047f67ae682a3f9ef40e3b1f81b001986e7abe7dfc7f097f9153ed79ceeb0906fcba3fff6dd4070f10b15e0c385f3217273276697e69fe6286fbe692177191e6b70f8bbc09c5a5e69cc07fe5a953ff3148302c81d98ccbcc0a35265398653f5ef8add1c523aad7b58b5883ebc60f9d791dbe9f300e1d1911c99d2748c2dade30e6d51814f2deb5cd92cc05a73b6682a6e54109c103dcb391e0c449ad3ee327f836f5a6a7fb835f14d69ccc15cdc9e2d726e066128f34edd27ba81784704&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642259979Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/array_metadatas, Name: checkpoints/3200/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3200/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=1ad66a25076379b5570bbfa101a24aa434230a2a8492fdb414819eb1244ff861ba5d5fb337ecfbb5e560fa70df947e1351158e064900af099422341eb407e2613abb1477c208fabe6914ca456d65e6834aed5f84ecac9a7ab1bcff6ab83291dde8df89e169124ab3846bd51cb39570758e55213be174f1eef5698f56ff510b3e79827b84af5f7ba790d001cbeac2809f6a3c2a195439ae0c574aee659bef16581ebbdbbe733e858542d82269aa9d9221adcebb665470e4d07752606d154f8b79e8ee0d2524f7228f602cd0654b962b98448350123593987743cc62ec6d8113b510905ff7bc580860e6dfad62fd96f87a2d88fa5c0f71443ced3a1c7a531c7487&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641912632Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416, Name: checkpoints/4416, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4416?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9616f274d9133f92c1c09cf4c2f73e12d4d9dbc809650e4800265b5c963ed7bed79944b25856955d9aafcd61a8ed5f938fc4370e39bee41fd5fd016eaefa0d6764a1b3cbdf8283da160be81bf6f452005df45c5ae1c7f5568d6a49ba39c3012b5a2e098dd50b35ee6a2fecdbd820c06f748c792b1809172a057d56ea5c89ba70f385b22adb8c8750cec9a44181bb62fcf9c364bd7f819cfd718a6573016f6bad2071ab9fcaa215b058cf1d3d7c5b3102c64a70de5539f3f372ec8c13293f3d51b9dcc80863fe23c346cec35f4039c4c4b3b0e79d6c8446b79d1b6f011241f856ae622eab77e1d1474203c15d6d6b0bb2ab4d77d649376fdeae0448898386123c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.64231133Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856, Name: checkpoints/1856, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1856?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3df0c9e1a2d8a191d5e6c9cf9ad64e1b3b2485d07901bbcd717c250f2081d22f04458e81a93f0b8ebab52f47990b3d0e22a939f3deb625aae8ab92560a0b3decafdb51ede1c6ad4ef7be02118c1d3a613dea2f7c57a861137b7da851fd8c44d23ec0b6b7e640f221277937da5abaf4afbcd46613024c0dc97844b60bad3fef7e7b5984a1b74e6b4d452982c411db037633788a839be684ecaff8d6d25760a6260ecd2cca31cb52b082dc961ae5e1c86ffd87bd651a3de359c75d90df0f4c355cbf62e8c778295fbdaddb1b675352fb19ad89e167ee34409a0c4b8fabbd98559fc4bbf5556a06d97c1f41c6eeaded0a5f50f7a1066c9403b421bb2951ffdea241&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641803381Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0, Name: checkpoints/1408/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1408/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=324ec6621d115d4bfe5e19e416f056bed1057ef257452695985d7e8f9bdbcb99e0bf508db928ca5ce36beca4818e10ef43d34170bb0196068531d3219ad593dce7951824a59edf68daa3f3ca414f669f4453fa6f12a221e24bd964aa37e2ce0bdf83483707c7e28f3dc0eaa3afd05504dcbf93a7cf55eae7978b4d47aa3affd77ad44a8a6e5b19ea998ac6f94f0452da63329ee5e8e559bd6a38ba71743f0888b395aab413d5c86288d073ea9e1b024c15f1b80311e7230d205b1c254d3bed8034baa9bc13cd65f7258730ffae32efaca3f8ee41a5a97c5e7a661c6e41e3261921e487febf739c7215e0ee632426b64b7cd158d42e914a9f616b017087cb9b9d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.6423556Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480, Name: checkpoints/4480, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4480?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=8b4c5d9df193a4e48b568495d850218fbc5b411e2277ab459fb8cd4fb03eb30daae05fd610b1a834e9ed5c963d0007433a4f1cad5352d7b3bda8644455b66a197ae83954b4e90bec25fecb0c7c45f444efde2ab868ddab26d11ba89e0634491039c3226f0434df2562eb4bbcacb7b4c84d08e7d0f3435e421cdb60695eca3f2098b717795f34b826894d181cd4afeeffc3521b818dcd2b84a87570a3f14759ecbae91bd54b5d359402d5bb4ef1c98ec47c844cd64f1f4fcb079385e7056e235da310e3226ff1e6f1d333cd037ffb4d116f561a2011004cce4251d31ad9f40ad05a2209ef0658b8560235294e02e7eb2b2cf97a3c102da6f4a745e84f004c1278&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641725628Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/array_metadatas, Name: checkpoints/1216/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1216/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b8d27cb28a109d757a6d53df1deca79f96a03131278b86fcfee054cf11b148b5b5efb970ac6cd443748336c9898fe372d72926d4198961ad5dcdb82739bf8d7411187ca3a7e382fc84737186f28c129a68872ad5922dad37946426f8c8d531cece4c17594365969bf8bbc8891e8d222befdb8e4f12ecf0b18083ce128e0789a709900cd1034a121b4be03349343fe1f991ff67fde1ae97453fc79487bc321edfc3380fe6df50781148dbdfd8a79f37c08072480b1b28239223985ab53bd49d90f876a73b7f508491396a4c3cbde3bb225442152f32a7d05779aa46b0a4b6173f262d259ece81db4d3623d010f9f59155b629bba7ae06bbeb30a196d4d77f241c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642374315Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256, Name: checkpoints/256, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/256?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a3a9cea8a9536f8eeb593dcf3bb4d0960fc4f0831c36cf2d8ee1c1877a319722d92c05f8e3806dd42460d1c6ecd8df203620f036b3ff1733f97d378f76cc99fb3575430077ef7f4264705842e0fff3d8a10e315c88b218f4273fe84d0019e8cd42b6ea8b96d2fdc823d8a5092e93b5672ab2085be56b2fc2986f0c2d4273e7bf2b0e5aefa27eef015b6cf0acd81710798afc58c8d229bb666816b2665bb2bb16325eec8cfb9a45fca3f4fc17a5799d374cc352b8af749b433f30a68f8e18a23fee7fb444b3d3e90a7623f53a94843534b711a490faf97ac45a8110b598fe74ce181b7df84c22c7bc95d6d6330f2bada3d20afd9e9e728dd019748748fbe6e813&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642421403Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/array_metadatas, Name: checkpoints/3840/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3840/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0bf9fda66f1ced86edae507631fad357130407e2197e33897260ad8a4d72aa3159fe7a67be8c2c3d7c7687b0001a8377fcb6474599515a9d5c1dc9c3899755db045eac5b6dd2a674992b1a37b0ad995524231f1009cf927a8c2c7b67fc82e02d095dded16c5fdb18247d50ed4dc9bdac7c0341423a131671b129116c5f9cdb934d5927c803aa6224a28387d08960c835eb65da5c2e17cc366bb6b6d0f375f91e5e4b46dcc0ff02ef8af84752ab699456f0c023a3123006cef0ed0eba68f981283f693572d99c32bc92730555235c358a9315d8430d77a80cab8075b7630c727f9e4641e8c1583c9de327ca1872500f9a1c4310dc6f06c8fdb722919f5491988e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642442668Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536, Name: checkpoints/1536, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1536?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=781319a7a6881b86c08d5316b79ce40d87ab9ed0c1b845bc5680f1574fa1a51623deb4911251764065103f3629a26826ddb80f674a674de7e72fb0a04e0c201c204d96838c20e48f75e864140cee333d42d9f5379fba49cbc80a7f894cc8cde49884f974cbfce028b78067bdac14a6f55ae81e6f7a121e53b180dc6ac0ed81375c3f5461e3a8dd82b36230d2436fa01c0b6d308113a49b9fc6c4156653f0f26f1ee1356e7c277c67c48794e45148d0744a89ede7e9287e50592cc4a4a9d7c43f18d5aae936cb8b827ea31b2838ca31d2b740657c0dbf10736a2e5ce38e32c68f10f36bf836fd0f351110329cfb9f9870f19339b206b4b35a04cce6644bea1ac3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641700263Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968, Name: checkpoints/3968, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3968?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7d9d82080d62eb0b56d8254ab5964497fad99892ee9acc5f2b132d66774691ea4d7e91185032e51f23285bd09ca8cee6dd80580af40c20c20422a960e1a88340a36a857367836c4a42641555095aa5f4fa241c36fc05e4cef6199fe8e113d82fb0bf61a00fd50bd14faa1dfda2b040704ed94e15ac9990b83294c3f52e5b564951d83c285ddc550c944b6c5c336cee256da8c933f717a7fe154bc61143ce923a6299427490e7921f93a360208235e778d215a77a5ffd729ba1659f09c3652f290c8a869f0f73c90c51cd30f02d82f3f081ab90339ced1ccdc54ffae51eeb3e3012110022d98182d97ba6e7df097a3b9e91cd9bfa3e4edb424911bae157a5c05e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642473872Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0, Name: checkpoints/1664/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1664/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=176dd791f8866593e9fe93b2f3b9bffce220d3032e38ae599019c44d05c36381e6b1cc66e1b64caf179eed09fb28e6b0b413645de7dfe793e2fba02a9263368d6577d0b389267b288efb4c19b2d1b9c0b4c891ed0df6c7fb90ed3393f864f403873682d7bd887e777d85b36b39c5e7d4c6c49e27aaf494f938006a2435c295b7ba9989eb28cde31dcc53f52d04de5da89fc13d9bb3b7271ac83df2356c3a454f07e4894d2eeb5e0d464ab9e0b4ef42727ebf55ea6dcbae106476b262e7a41fde91624041db692444fdf6db6cfa339ed441dc291da0337c3a85b59b927ad3e15f2a6491f0bbfa9d59dc213cb42f26cb13eb0feb7601c6b73e9631c04b3baa650a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642498292Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default, Name: checkpoints/2880/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2880/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=29ef84454f77477081e8e2c176b93b5915f87b7d22436922356a7814537129e5e345d0763830e9a4f59f099232a58b34adc5fa1c2837fe693202c39d30ae026c54d5df331439604e2a931107c18316275eea7944fc3c5cbb570dbb4bb8a01ad25725880b15515388b740023a26cab6019ba4e48e150e77d911ecbde98e5edf55c65c4d542f422b76a515734b8caa8d378c32e46b2e1ef9fb9fb3f606f481e903762e04181b07e9a8a8acdc733e42c8149996b521d1e4929f111b1eb76429507b1c7589ac8d81c555a58eb372e35f776d8bafc3a5ff62a9cd678466fd0a09d54df51b7c3920431ed5a2b3767edfd85470f43051ec96169e71a6103f24aaeb9291&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642488833Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/array_metadatas, Name: checkpoints/3328/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3328/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=515c0d4b7fcc96d16e34563561cadd280ad989c23ec315d2a383804d1736962184e0fb31fbd7fafabc43ab40cbe5f948236efa76d46eabb5a6a76b4a258f34bbb2f1ec86fbdf4b5330c571ee73d58ef0106eba0a2c12c9f75054d34eb13c0a0f665c0157ca9be3322c30ae9361574355d440e9df2a0262b30d586c04477c805110dc7202ade8d736ae7f8a62ae4621502ac2965d7e54bafaada043290be31bf104ab8b00b31fd11945f4e0ce5dcb954240288fcd89e18f2e1159d756a9ec463eb844f3a55efe87d663e2f007f2459e0697ac8adbdbcadf53105989f887c46daa54d4fcc03700bbe3601e0d50a3dd918c577af06c54cf2c79eb6a15adbc4c4f54&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642527179Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880, Name: checkpoints/2880, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2880?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=511aee70779705a3f68f419543ccc3a703a0df08fdccbc3da62b6465be290eb6db4426945ccc55ecea587a2d153c9ed9c565b90b44abfa227c2dfbd1cdb1c63260d1a2b92edaf0b02e02836c1a1373c966b34edc091bc4f8e55b4574b121ecf5b522c5159d8b6a8202a887db28500a2d49d5a0d8355f5b4c8556579231c3ebdec8890642582365b08c57b35002cf0ea5bb3ca1c6e88fbd2291c30cdbf4856beefae17da7838f4e3ca422ff6c6afb7533f60a03188a779fcb82a367809a6f7b4a7cf9100466647b79a0ace630fd2698ece9d300958b9229f980e67ba38e22605991441dfd540178b01e3ad2c0c3a9bd24e044704c08496d00954ba1a185781a44&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642534025Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default, Name: checkpoints/3072/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3072/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=00ed34177af0c3d9384075b2aa09deb311d41cda7cb7a5ddb6712d9f03b535e00fc4fdb50c7fcf4256371c4d3cd3c65c97013bd57ef2015ea88f6ae641761727c4da18415d877f75f66ca206ecf2643336579e72f8bf9f67e4e49551cabed31f1c0c0ab73dce920e568cb2c78351a4c82613337f935a3343c000e0e2331701c89743b0cfbc47c215848ff375f96b12ef69be46d9e0866dfc24bd25fb0083f3698d4823699bf56b6fbd091f6276fc8f8b1f3a839921e42aa0df369d91c0deef59c4f90555d8ff413c7775b22a9904920fd54cf5e7ba20f600693d6736912d9715fe31490499b33f0b31d0b6a92ec491d44e8eca33ca5a116816224bec7bdf2470&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642382698Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328, Name: checkpoints/3328, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3328?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7b80e52a1d3e68bfb3d977ca6320005e95e83fa2dbfdfa76cc00dbb3f28b65b95344575f7cc022d22469326d9d20ed12ef84952026e77e785737e8284e021e7091461c96dc407bed45a5255e314730ce8cbc6fc15840e4f1699f9cc2da42e18777b5b815620a4fbcb6d9b796902e054db22fd10eb1484bd2e3962cf7090c46c96a39d505897715a553e9531c1a4ad59ed8406f744d19d06cb8b34c361e5bdbc96ea6f4046019e21bd7af8e5399cbd71ea8e0d5230ec14dc03c15fc4da055eb7f1e20e00f4140f35441eacdd3b28168512978ba3fdf73b6d2d3a5be92f859696f8d4fd5bc1415f9fe1cfad7d24b491f330db7aeb9162c79cde7346ef13ffa6f63&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.6425982Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d, Name: checkpoints/4544/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4544/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=788e16d6b1b9da9fd18b7d12bbb3827b860570ebd4abcf0ae5e0c832c80e8a0a94541721bba1c0e452b2ebac386314dc9f6f03428f6949ee0b60da0222d14df1014cd5fb4b65ee62eb1644012fe0a9304eeb82b9bcf1a45684623b74fad24d2eb6053289be050527d973e1d5f90d67846a03387b8d003f7a83369451e3397e6c4045a1b65fc90bf9055c8b76d52eeaba20f656a5f917d122a6ef4f22ed73ef43ec276b5b4969cc08956f00f98cb63df36e9d9c9f5a59d23bb7493b6987f4759ed71b6329c7fbf7994876d67b99fd623fface2929a61e4689da84cc3711207b8b19c0e943b6ac923df84618a03cdd9a16942f3e0c87b4bf86b7fd9307a7be0ec2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642615375Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/d, Name: checkpoints/5248/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5248/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=991b44c9d354e0967870e874e5c806e1cd38086d971544e628990cf177d4ab6aacf00a6598d004c02aa83a79e4ac3a6c4db97db0d275d5b30352455931066e5b2251b265bce11baeab8e83ffa075d9519eedd29a476e9b10dfd6fa66ad5e6dded0f6a87cf9237dff1d0f93dca72edcf1f84d45238df35b87ae52703eee7da953dd6e8bca7060ce61c0868ba6ae22c499604902ebc0885c0b227d91a90400ef41e66b085fa9a8232861f1f44356151196f36c373640dc34fc6be6718d7e7fc006b99938d3e057bede35f940542cc86c77ac623096582275dadea4c27fcce7b5ad8f39b9c6711f6e2bfb410e75f22f4af7d5158b9ed30afc1b1264b330315321ed&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642618995Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d, Name: checkpoints/2560/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2560/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=bbee1b0907d368c8e282a2abd183a88c80003581738270be6b9c936876c275e9ff69540d12081a02c9be07dbcc121b3fd0fc613ec474571388f441e5551a3663fe7454384f9c6dbe960b10d4bab45f18b5dcfbccedc63e3fef1ed563184ea8f1765fb9174cb66156068a5e6463fe465a9a1ba11e050698edfdcb632c88c1f0fe205e0a1d827e20419d5f554183c84185d38cc7af36bcd95999956eec38d65cab23164666c3ea3332c614145f181336960195e2e7deda9ef279a0c5cec101348326951a6f11b1a93361a6d2c3df18cb1236022e4e0069f87cfcca89a3a0c8849f35f21ec9dbd29877b8bb09cfd58dedd44dc7ccd80e10a05d61f3f896436f8657&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642558134Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/array_metadatas, Name: checkpoints/4608/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4608/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=98c8451b655cccd2e44eaba346ffcf9a20624bfcda295a447c3192ed1e08b877558574098052a6cc6d94fad4e28271e2f4751c5401a9860721babb7fac72f576a0b9aafac6444c37b598b678ff4e92c3d975a43ac9c5a167fceb25fe6ac3890bd6bac7b2a0b99d4e430c4685c81efb3845aa4390765e183b8b2482f6e449b6db5ee9b80cad586b42c16002063b81237a41409e6138a0a4b8feb9b9898922673aba2e8f3c9148325931ff966d689fabcf002f04d3f8f36f22f970b8b99b5122ec12d3e6b4a1c2da294a8761de0b4fee5c622417096c5a2a826dc5d8d44e609d7ce341b16e6f8e4f7e04b0a4bb180b73bffa82a411b0625112b17d488f21b171ab&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642660289Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default, Name: checkpoints/2048/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2048/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b445635038c36f0c1d2b4c0cf27dda99dfc959a9fbb1260e61f9fc5b6f7695e9d5159951512bc0c0c2789fa79efb3ff530f64637a461df3db54cd3bc578c8a1fc639471f5cc3f1ca929e10747821108f917276ebac0b1e857f042c9151748dafcb043bdc8b075d699da2d8d6d670fe1d0e91abfc961b7d60d0c03dd4c2a8bba604113ac9b2d63b66e86d6d515a9c66bc638b8fdbff86e8be0b4e89efadeb35b0bc5d3432a393ec696ed96f757c22d3a5edf502e4a001101a25cbc3bbde5b2082e734016d246c5ebd33aa4915b5c8469a2805b76c1123e5ce8d26a8623b746df0f6925d2945823ce14069d8ee8b22884c0bfaa1808fed9b602259bcd368b2260e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642676472Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160, Name: checkpoints/4160, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4160?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=730af4371ba364d98ea9e06ab342e8b58108c9303784924d85efeef2486272131c6b6b5fff8f32919ec6065cb4f178ceec8761622df54ec0429915a689f7e0fa13afce11297f4443527391ce4e5f8bc339a1fe28a71187abd6f55ac282e43c7eaf93ad3ef3ea71cef7282664467ccbf7e451cd4255c25a831c233743a11f3254a3ce7d092300d772b902be7fa248c20dd55b60c3733d48f9e7dda2e1e88ea6212af43191986e25ccc90ac9e23eaef44eca0b9ce9119eb229e025b26abdf2e150a0867421d9f72bf1695e863fad182dae1fc729455825d66c81068f1c32cedebead4f5cdd2925292365ff04e78c09d4754d30b59ed3b1636412f13645859bb6fb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642669609Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/array_metadatas, Name: checkpoints/1664/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1664/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=346a3d66b7d90a83d1d254b8e9b270dae7e35c3a2fb6a6da7940b64dedf8d04b7a106f21a5a5e4c55c15b2e2ceda86240b6cd7c349a6b4e7736d6b3c9c7c5d5272491baa74821a4d387e03ab569f063edc9676b68d10bc762283d4e74d98f4e88cdd1afcd25677d34e4d1c410949e4d9a2c4c03a7c1e656104ae7e507b6dd1da1c4b2dd0b97dba2037b834f13d1713881e6c44568e8c15d1de2e70c4c4a257271b09b8c0ed90c4a8f3798989578b5be23d42cae18681029d99891e478ec81bb281d93e975aa4e00303146d71737bbee22377aa082504df1843dc31fd7af287f090b90815ae6ad271c2d8a118a28b207b3f207a3a1804136f13b383d4abd1d823&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642073666Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/array_metadatas, Name: checkpoints/5248/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5248/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6389b7b0d80535411bef097bc6d3b9115bdc813207b4adeabf41db0cbc8bd17d509c1bcca6c2beeb84e3f308f59028db66b147c3a4024c1382bd4c6cb6385535c2b7776b3998357cca519b67958680827498818e105efeadd1e3ed3098ffb28e1bfabdf0b5e2ae354c39bc02c61217e6d0834c2be41c37bcef2819aefd45277c3f026185332e91fed89215021fba18598f6de903723bdbc3002bb5c73d2c51ba13f4b8c876d690d937b3370c77cdf668530003352abe391fb9ef411007bd060cde389c706802f590065f8a0dc5fb2d67191de15c21cfee7337bfec996113c3da3271d18ad30163d2c3ba97fc9d9c7b8033ade5104b637d222b40107fdc66fc15&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642638064Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0, Name: checkpoints/3328/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3328/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=614dba79d40c71e5efd4a25f802d3a64afb471b30e6694a39cd580b5101a34f83c59f966c6cdc252d8cc022fb79ed87869307c3bab6f1901aba28d53a8779837b15ed27f44c9356b4bc1804d2fb289fde2a49c88d83d6db79bb56f7ef25cfcb17cf4e2e0061b95e59cf55f2fa56275a05bcabf15c45d8816a94234afb5c28c511f67122310173deadad09699f51288cf0becd29fc8339d4049c91c602a28e6f5e70604d7b8659d8330c44963b1f3d6b663b6f1ef7e4f686b5933fcd9e5c34f7b48b9840eea0681f0428940c293ba5d7912d2a7a234a08f31b6874bbf6d6d6f316d761d9bb62e6876d9bce36793e6769cf99e2516e21a1842e08736e294e86a8b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642772539Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/array_metadatas, Name: checkpoints/3904/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3904/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9ff9d6e02aefad71856ba322ea17c4c026f90373850fc8eb26b33ab5592d3d5a3b4157b8861689a616e64e3ec23efefee23376967c94eb12497ca3b5f9bb22c2185159a523babeeceaab3af5b92e6f8f43e16ae49c8c2371309f58c897ce7918d3f37547b1ced4f83e949869d82f715168857f95bffe5e03c0b2778639284bf943b85ba06b7d49f76e0eb60b7db4f3e06c41ab5ea9eada167595607b059c1ba09bb73fe56dff14e9e9667087aa537bcb765a6e39a8ee505e41400691e9009d4e80d93fa5b25a0ccf70a585aa7d9a952a2b8b11edef13c28e96dcece241e24451a83ec9869473e6622a8ba1bd37613cbc58342e2399fd11a5e49e07f415088b3d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642781099Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/array_metadatas, Name: checkpoints/1984/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1984/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=bd40ce3602362dec3aafe85f418d25ab0095940f6fd63ae9968706020330a1db7b3561c2eb02f96e0e10df07615e49341e194f585f19c2b636ed3899d12ae26ad0da92aa9b82361eb5b5bacb90b6249f25a89b4b5e338fbb9e2ab89679bc8ac7b8fdb25f7b4e366cd4b0df7637ce7fca9e4ae732a10945d858efb077390fe4de1ac2afc45c690eb1a7801961c8c1ad31840575302ee337d128ddfa63e0dfdd576f3b7c07f5ac125c9a8ca0dee3ddd9e3c44e61e44cccc20f0a42214f18b1f9b651e55bb31c29874f420e8d165cb77d96bec09c3b65db6c4356a88f4c74ab5332194eae4f562e48a6ee34b9ae4a21d253ee90cab58846e0ad7f6af69298c76d39&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642812774Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0, Name: checkpoints/2944/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2944/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a36ecebe6516094ddb704e0b9bf7c207ebe0c12bbc94db133035a8503d8a3169b2b7556d352e16df0c9689b6c06e16c30ac8715cf648bee7fad9c5aa3ea64361826c19052966e772373652b6a3dbed4ee2b58e193a3578d849d99bc7c1238afa460ace19a8c66f9259f8a0173f77737745280d7ac48d44db7b736ca73b06f4490fa3da81f5e2379654cf64173279f76e0f728699ab454434da44fad572c80581fbc2dbdacaa3ff0bfb080208e28a1e28d23434b5e8d9978694818a243f17bfa246d8515f0737dfed17bba03d42c2d7b996d46460a711aaf945e746daf2f44db722b11758392977d2d902fa954e9af6e6c1155e21d7bea0f658ec8b524899b646&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642839617Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/array_metadatas, Name: checkpoints/0/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/0/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a3cdfd3a61cebff711fe5a20a4312bf8efba2ea7aa2ea24317903e0b6dace6a8a81ad802023fb8363c2fd07a64a15f23a49e2c8ed583c795b7069666b8e52c148d405c5770b390fb6852906de57007d60ff9dde0c4d70ca52668e721e05be7a5e0f78ab4a968683817c1db01c9afe576be4bfe411911a54178cd7f417227df2478af2a1411c21940e265f3c5db0b2b7428bf517f8d3652845b1a20e5388540bf234cc257006024bcd965363048c4205943d56ffd3cd3696de3d78201c5955c04b260b3915ab8c63155deb5b78077316e3e88dc0397eb8492e70c2e456b2bda7216fc6a0f8f60e843d90ddaea9f38e3801850c50e199c7c71566339a316a7d4c3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642861094Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default, Name: checkpoints/5568/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5568/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b5beb095d2f527f7e994bfe385132d9f99b1bf73a15285f6f15f6f620d59172b5f61246dc17f8a0477f9c4f60857bd3de47db9fbf6eeb6f749c6cc7025f01ebbf0ccbd57aeebaea42736554fa5aa6085c33f1b0142af947074e1b6ee192aeadbb093b2a33b6a9b98986e2b3c2398a504c1c315ad020b8b40d24b658b81a55884c43415d14faa8912ab0b9f23e46ad5d06bdc016c921db4ab374be63cc5e8d12f9d4f9f28d9c5d95c83a3f280fbef2a98cabd922c3b35d7c7dee65a56a4f0e83d7f5b8df0a9632b482cd79a962351c9b4270a1e79ff90a715e7094290d84c5939a24d9e7d9a9e336962c977b4ac24f8558f14cdfe34f800886a553ad00d47d426&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642863747Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/array_metadatas, Name: checkpoints/4352/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4352/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=325912628a92ac30e7f3a26f631e27a8c0871002f4cad9ccf611e4a0f91b113d239fd4646b730e5616ef4c615bda679645d0f27b1da5865b671831b6bc1c5e78e1458e4fa169908eb4785d2f01b1e5f6e6a8d48484d07f5fba8f6827cdbcf3676a1e08d133f80b7a3b14b451aa0236af20dcf48320fb833055bfcaaa00f452205cd9ee52b0825b3bee5784ce83be4cbacfb73150c4b02938069d4c4e7356372a9eb4796560702c43f6c217c56846ebdbb8e911710a784850b1b3732dfb688df40b65d18e4626bc0a74c699a13905a5d67d039c7e739843d393a473540461b9e0d65cf876e873a0a799f2ac0b79860052a6de1df28938d085c1b5988e9465cf97&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642860576Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/d, Name: checkpoints/3648/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3648/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=57dd87a4b009163d22f349d594212c5e219dccc973d14dca8445130b04d69ce26d4ac3f871e55cc0045972126094d820a6d3fd73929524f4af5af769828acfdb54af47a430a4191b93ae734da2f07cfeb71523274562a179677ab9e23deb3c2e4ea0c9b7a6b32d805c92f06e6ae8baa48e26e6849de4599e745857d8db60a3e103b84a24f4a94a4da90c25229e5229b72d187e860524ca665ae10d16013af3ca11ba29643dc0a962ee23871f31ff1fba0b13326c0597f602a1f0681a5c008e1d417cdb912948586441292e969dca5d0ccb1cb4a486d7afc35cc60806cd7e34527e69acbdec1b02543aca66e9b2e3401ae544aeb1e66dc51642229b143050813f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642701611Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4544/default/d, Name: checkpoints/4544/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4544/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=05c561a01db1c4ef9c149d10ed7243609ccdcc82250c8de108058b940eb5b83e173837ab42cf7b93c07c032223b2001e41cff011a31e8b5ec53ed13eba816ec3efcfe94697c1457bf2bdcf010e7c8210d9fa5b0ba0fdeec9bca443c4b19c4d77a1b603bb08a70239b76b7b00f7abb477117005f2bb675db8db165850a2c0758b3b0666bb729b6674d959bc3d1cae12d5fd151a6fe65fcf2b1a8d24bc456c04087ac4dce5327ed16f4c56e588e06bb74da4b5e12e6ba2fbb54c915f28870cfa807a1abd772db6fb96b87887877ef3a6b1448a6f0eacf0c537476520c067b2666afff9a07a55cd3cab4f5aa851c0686372ef57456cb65d85caea6e179fbed5b79d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642936925Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d, Name: checkpoints/3328/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3328/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=8a75d96e80469480414b66fec31c6d01d71392f158edb2a8b0d35ca8f6e6ddec689bc17ecd9b009cb04b2a82cd39ea5aedb66c361479d147346a84984978d6b8ae1a41b76f732d7c0424f220a61aee7fc5c6342b1bb43eb4a1e04d38fcdecd9161549cfd3e21893814193e1e217b6688457451a09854ab41678fade122f46ca87486a47f2c4a85585b7abc3dc3258f900575cac2a17f37a716635018f6eba85fbd7f09fea2c6cff634552aa3026912f28c144f04d067643cdef2d1decc064b3decba01c18fa3bbd65fa7cd24a96073f3e95ed6bcac95ff81eb4e2b20e9c927f7f0ef2bf342872ee1049d805b40a8d586661e19de4c905f1af136ccb745984cb5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642959707Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816, Name: checkpoints/2816, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2816?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5fe174cdcdde52d4882ebfdb84c31e5a163a8fee876a921abb2ec05399f44687d8530250cb31a7c522330a044620dd0005e84a8c806646eca6453d8263a0dd9ff60f7dfb6497dddc0f397b869613d55391641345e760efeba2ee80be05f51438dceaa6bfe91a2a9518655f353b9d404a718d90a236ca4d1ba8e16624adb62e5222770ca1260186db06e7e11a9d956f53a60bc9f30edeade13bdab64fca080e7d93f557a138990f054165456fb6f91c6df392570e8a70428b6ac5e472263d4a515f75aa5993f7c7899cb679bf767ad931975fb0111c0c3985a0f212cb1c6c7d360d7e23b1070fcfd7cc906f944402a5db771b45624bf828f7a03bbced85df0887&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642429004Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/array_metadatas, Name: checkpoints/2368/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2368/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4670a27fd67cc18249eec0ca583f45f7f1de5b4c44b636176691b7f7264f6d9b1d2611e80650e07b1822717b46196508c36cd713eee33163c1e2884df58816fb3ec1a0a35dc3b791587bf9bb53b913c9c0875d959947ca432c052ed3afb4b2938f8f98dba98f79c8b2c9a235199ffca3e1654aa78247cee930300b0ae59eea43fc201b57deee1a9c08a7b0b5b1491b159a13bb564271ca55d02b4e5437208b453e14abcad7daafdbdfa44c41348a84b918afc9b35872a5525e50f58451404f1111c49c8c81ae142435dea5f0d6092d07836ecb23c793538d86a232ec4187335295ce9cc8182a2ac6041e834cb818cfc555c36b197dc7c0211a110f43e6b32046&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643022117Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/array_metadatas, Name: checkpoints/4800/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4800/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6e3678d34665a70f57abbb68858f1fef45863307379d700239d7f25ba6703cedb9f62c9b5c5bb30f8747d759131018df667b67306053c16f1a4c34541e082979b1dea4fb440e92ca12faaf1d29e8459204c7ec2820a2b3c6b13750af681ee4ffd844101addcfcd017a9a34608803b6ea3a8d08a1781b8d5846b00c2c8622fdd09bbd3422e9dd0490a96c7b74e77ed799f6ddc800c88c50fe9440e3d544c1d6a145f162a8777f7d53f84bc54ae7be2130034fe1932c92cfa5b6b1688628a6df9071460bb9bbd1043d2420e3e85660edc0aa1e2ac622323a7b9fd3f09c69d61b54a40ff1c298f8eb7f37d4818fc62501ca796d0c08bf8fcd5fb7592971af294a6d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642819634Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/array_metadatas, Name: checkpoints/1280/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1280/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b730d031385f4072615415343e7f7da8ad9b75b60b1d247fab7c89ee2f5a981862a39600062e1ad6178cff201d462d0cc814fe303f049af3846374ae31abc7d6b3e602d9f421883223e598cd47024b153126a04d44964f4db12df8d859b5e16021b7b3d9a7704da73263f9c7920f2f7c8a327526a312e958066cfa6669d3bd6ea9b3f3c20d87eb71bac6ee7ca781a87e75dc619d6fe7cbba7332ca21af9d25262e2a6c3b65ddc70ac1dbe3348b3c9b0051e09208523ac8455d7cd15c8501bdf863eb5eb7f4735f1cef37a204766274830ea689184fa2dbefc5cfbe8e4deefb1617bea1f689a91e8bfe9b130ab63c1083ef95e3e8bcb2fbf90e6b95c5361d3854&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642972299Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0, Name: checkpoints/960/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/960/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=88dfdccaefc83be28cf00beb0a9884e8fa2042305fbad362e9505ddb3750730a1369eb96087f0386ff233b6ffecc40690be615703a69de7cdf614941bb4157067cf5c2b38f80e6f37d2fa249c009b3115917ebb2a56704fd7296b65d5f3df7b5e2421e27a81d25ca2771092aaf33ba1a9387e096d98fc4354c15143a4713a148c1af45069a100d5b46cac51f7aa49dddb68c38980cac13c31e05b20af5b78cef9f28bad6fdbe4780cff3a6f50af40b5ebd1462ed2cf799cd7288da6f439408ef82f4745c476ec6278e54ecbd24a647c2bb004c9147016187aa4a9514d47942a960188e939f376805a3585d2350fa93f1bacdd1fa4142da6bdb848ff1be0e6e92&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643073538Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/d, Name: checkpoints/4096/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4096/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4e65abeccbc802517664f7595bad7be1aadd1f7e674d1d5fd0983270fb6f37004aec836f77154bcb3dc8bafaedda793167b888663bbd54e52f8b9f5b62a4a9ef0c593fbbb9d7370d2f97dc1a62f44c7d541a115be9b6df11429ab4e623a344eb0b0501662bdee425d14723893a8c20b1667856c4aabe43891f7596d3da5687c44443c3a7d4def1f766f58d434364c2fccbf7a5ad278b696448c20a70355d787e65d35e628e3ee427363f82911e000acb71e43e0fb650090c54ddb2e9e45efdc21aee7dc5d209198f726f0b40d46b9e355a0bac82d6ea8055ffe12079900409d4b24e0dad790ec9fdc7d394eebaa33709a7c15898c7d60f125559be6f380fd6cb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643111479Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default, Name: checkpoints/3200/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3200/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=83a6b1d7e4570ef87d8be21552bcb08a838b9e1ad4f3c03ce1b8b733ccb992f90206d9eafeabfc892c87c177a66f085c18c0664aee022396e51b447e53e41c8680777ede90f362f76c639b2543b2e987bc33f1b55d128e6e149b6291dd06c5b1b044ddcbec339d7c54570aa7d841a46d4cc7636bd7baef5638d3accd28b778c993466211bf294156d8261d2d4949ed66eb167e43387d6114597f23a114f399c832260de32f2b034abbcc2b20cd9cf05b76d388bd2fd19cb9dfa0e977f58df7a152e78b049631c09fffaa2de1e61bfa7ff6923af7cf1ef29d781d1bfa68522a21ad112f9eade962124bba4d8d8dcd0e2cac0474038be1f8bf0c67f72b41c53036&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.64312449Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384, Name: checkpoints/384, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/384?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3b7a9a300c888affeada88201d41b82686f18a0e1a2a998d9964cda003a387f292694e64596bc743815feada8718e76e0f09ad18a2155410b059d415b8550e6649ca811d7a2f3613313f468f9c9f3d53a1998d12526e7f173b37303b63b81f13208be4c4b7a71f14fc6caf91953792bfe3649ec28d706ab739d5d5d9c09a38d607ac865f4381c49351db671ed88a16362b63a833888d2b6582be14017fef13af81f77046c577b2588a5387212b27e340af862a5f667ce67401b6764b3b7dced1ae76eaaa2c5d85c5e7dc1b9720e16b8a6831e76a194234f84f9a63b4c326d049b70353d6222b88607e3570808a0b9c04a87f62590d1a6b3511ddbf798c8f658b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641862314Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d, Name: checkpoints/4864/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4864/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4b1e89c16e4e18f552c6cc36cf17240bb9ad55d62b3c587be1cd2838ed5a45aba4aec4a316045ca3275625b085b42df04c7a518b73d5ce89b07e1707e74ca3658fb67be27787073b20edaee540fa4e680c72ea57b7a2a90a3e933eaf14d0329c624b5079cd1ac7b7db162c2134e19146fd181807d123fd3c8895bcfcff01e3ce87e31c0d31291879ad8b207a3066115c844cec1f41248944ba879fbc520289d11f625cb03c381ede59e4d996418d23e47b841cd9a999aa009cd8d4eb2ed9140f61b5e92fe3e1480eb8e2039081259d3e60ff657af5a756a24952850681b1d0262d0f731f7b1cb94342306eb497f3273c702ab2eeb3f6c32d64b8aa893ad657cb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643159124Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920, Name: checkpoints/1920, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1920?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9ed280fba8076db65c806f55021ee8c824ee46059a94269665975e811a319819b5ec4c12d8f72f8bc5f05ff8fdcf6ab491abce935920aeeeeb039e9ce2a4203eab0dea16c6f39f3c7b3edba4bdd0159a474221fb987cf9a5a1a1a140862721419a697c277987e2e3b775708da9ed6c4ff0439efd3f191856dd4b03df70e1ee9edf4c42f45732ccb040025b2d5b9a27997b34363b4419a0d78b34522339756096ca80e212eece136fb458975240ee065657a5974cd20d1a77086523991eb4e625c0755c4bc9c66c71c65f8a3e43cecc256eb7ddade8e5727be0e418f8c7f3a81babe879c800a882c9ee4d3640dff689f65c7a9f6ab96b59df2fed76a177fb9277&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.64312292Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288, Name: checkpoints/4288, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4288?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=501735248592035447e0cef037b308d924ea22a67bd922461a80a2678c06e85683692b089af060cc98c5642f5ea056cb860377c6d54c9b5806af803d33be2855828f6d75cf5dc4d58f5358fb495ecadae7c1f740294bda502d31348322e1966d85917252feec0eadfff2d62f6723bba01752368c0bd84e9ad2c3e823aa143e40504772b83a31b855a198447ad4d84bea64ce7cea1f622c38b121c8f8d25ca9c39cf0e5275fa27493ccd2010462e843269a0df1b49e210bd1015bfce94c6393a487802215522dda82efac1fde496423767f1761cdd337480a4727846645ee8986109c35824461d47e503933bc4dfb74f1d23aeb9343d248a27febb912460f0fba&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642689124Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/d, Name: checkpoints/1344/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1344/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=053d981b2d1c06bfeec7b642d034432aa9d8ab61c17934aa587a8a2346fbb42ae613d197302a66f25219455c7c916b4bfc68a9ae73196ef4b424a85dba55d8df146d25f3fa327c610a0eba95c9f370373704471a0f568e29a5f74f1e3c6a3db054d130f3b51ecb5b254ba5099bbb192b455ba3b18b9bc78afd50dbed11ed4eb0a96ce99772dba310c167a34f0b3ed6aca7739a65fc2bb88d3e1998f279e82a2ac182951125be359e3a3a5dc15d77cbdc047f72b487f723998a4c3b2ed8ee2775341bb4115da51d37dad2638b951fdd7b434c0651471b7892356a6c75fe6094f52694147c9e5209d39218eaca228df36d2c9dbe139fdfca13d3d8ef9044323f35&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643228838Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072, Name: checkpoints/3072, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3072?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3a23b7a7ede483002f2cdf9938377694ba9bdf799890f6845340de65c6c68eeb2be7e06ca92a40d66750807e1d497a9e1d07a580931283c1ab305ef5e2c150ce28646ff4a9aba64a6d1d587355e329d06d4e2d3d6ffef457905b57c3031554dfb93752ff5db10c5bfd2b510cba5badc57c9d7878c287131debb33a88b607c3ae18ce49c1011d0a9fe5eb77ff8770e0514d2d3ead0561402ad61969817d5e08f1fa3b2a85707cb8b7f6030efca89d819dbdb8c7a1e87e9984e66ddc0f884faff1d2dbc5889523f8bd8a1840d1ef6566ec1dad47da95f893653e5bf6420c95ac20721de44cb492109a415ff78590c12588e08f14be94df1d3cef389e11e0890a39&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643245085Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/d, Name: checkpoints/5632/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5632/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=bb1924df02908afeb37852748e81834892faa401dfdf45316c05c1c79e7e76d0e32292226ccc89e6defe76d860ddd723269ee9ad8ffe56724ebe12b25e55cbdf0c19eb0452b8669c70cb290440b3468dc6326759f3f166e8a948eb8ad1b317605ff444eff1abdd8eedff2a8aa15a1e256dd1fc90b74c2dabb7cf84f7b6cc453dcc40265a71a42f23b786b587f0bf222b5795afe76f7eca521aff2d25cae73ff4acc781c659810dc8c5e70f7af58b4c995cbc441a98126546568ce9e833604e4205f3538bf5a41c540e710a0532d13ff1cf667204706faee1618bc078a209d74ffecbfb013fe1b20b9c33573e89275af19e4011579c58ed8312279e83c0c65a29&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643151787Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3264, Name: checkpoints/3264, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3264?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=55f5cf01b7b4beb1b3901b7e189811b8c8ffa297739b303fff4a6a13ff0f01dce6616d5f254cb297e842c3481c453d51136569dd71046bf11cdd70c02a468101d985117e57ba6db8e33477eb6cc067b134e363ae4ef64f14d30718f2b7a603e3494502c22f942321a6739d81ab26d0339a32c76e8591d9494f13ec44ca30ac4b92417b2b3f0a8e896f34782a077b1bf9af67f426c17d5285a8e563d822b41bc080125b10ea162493b618ebe1273d8712be70665ece870cc9585d193d2461d2fb298c425c046ec6f1d9fb8fec88c4f8237c39351ddb8560a269a69844e942976876da119fd034a1c6a3cdc1b95875fb3d913d09d151d01b548b51b6ee3c6285e5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643087814Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default, Name: checkpoints/3712/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3712/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=accc3e8c0a21479a1bb9212db5e093b12420097308382b8a02ad830e9b642ea31c6e1297c27f78620089cd9b38573a1883bf25483b93115ca91c508e2fc10dc40c5609bdffe17e2036af29650d330f3d474b2e8699f4c7d3621fb58526d204326dee5e635e151c0e9b64d650f1a610d96b6cd0089850635ac7219d97a499f21a9957c1dc7415f5d86b072347d0cac349b27531876b4bf299ef885c022535b8b1806506b5a671c61ba846a51b6e6df3e7c19d0bfd975537f50c882ebb0976161910dc990060f68cc13235705943506ea5546e4104fa338b4b2d087f24978083aef2965961ed5eaecca2354fbe8f057f0f988070828732c6f8e5beba44288dd40c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643310479Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080, Name: checkpoints/6080, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/6080?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=733cc5200483b9629b20d7f5c51e1dbe246eb94c804241d654e7b1e88f79dc2d63ecd0202755cf3dceb46a47084b08ce15f69df1a041f504b995179c97aba647fa1bc08dd55ec6686d7232061315273b26d3ccf9c17c51cca3b202de2e5c83aef8d63ba0c149fb63d10487370baeb2eaa2ece1812e33b46e0115b065d7844af39f5fc3225034823b21ccb408329db14151fc749fb27b3cce2607892ee47c7a81fc61876de5e354614226af9a5713e26380e50475b9c6c4ce11040899c37da80f3b13a05967139c1e2ceff511fd22c915d229420f4f357229a97c37559fdab04aad00b8dbb0ec12f595aed6b25f1c8060dee5baed9f1f0fe3b9737fc538a53d97&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643335608Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032, Name: checkpoints/4032, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4032?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=40717d8df74956c8d52a459355f68ecf45e5699abd5ec96b6741b3b9190ff389dda48dba98081dd2279b27c82b520bda8d7001875d55813333a882c0a11991dd76ae45f229a5918ed82e0a6a4ab4a1febcc75e8c2847f26bcd5f36040815e6c22f79819b8e588e4dd97080a8c4f2cfec75e4cf93990d0100b6e3a3c64f60ec106d9bb3b8f2a14c1148477140bbf8a99ec185d452e92deae159239fdcc02b6bb874a50f9c9ca407dddd0abd5326edabf7ee8b0163dcc078934273025b101ee0eb396d769b710aaf6ff02ba0bac8e785b75f74cab87f4f76f549347ddd5cf6be4476de468d594f1f8686a076d48b18f6939e1359dcfc79d6b5adae76a5667bdbbc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643361702Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/array_metadatas, Name: checkpoints/5440/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5440/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3f57c076574a8530f71df43ed7a9e2b22b03aa34085af7388f655ad56a33a9157b5e484b59420f5eccf91258c492a909ac9a39ed2442746585e3612225835e8bd1114d1f912d7209833db3970688c91a70e3d19451333fffefff56ad69a48d53cc1a5731a12ac236e73469f65ef8b5ffb3006a285232834367fb9194d6d39f604552dd1c13da45c45381fcc780ceba58e79594e2178ee12a914658e8d5a6f18141b06ca8cb78ee111c5c9c6e1da5a1689122d1d7057f1fd3b4adfeb544be659e41ae56b6edfedaed975e30e99d2097708d34e1695e2b07cd3f13b126311aad2561ce1d89b0f363f038d37165e839b66a29e68101c1d4955e9f7ab9855d1a7cc3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643374536Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d, Name: checkpoints/2112/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2112/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5ebfd059953881ad531321464695b60cfb042f4c7f328ab12c7b7ea2402ea276d60974618b527466fae98b26cf27497586afa26b65fed34249c174ea57be6634e390cfc97e40b529bc4ffb93317f2d13ba37bf5bbdac1f1fd4677b541d654d44717b27259f04b9b5cbf1c753b136b1e0d1bed8798395d1ec243b7003b4995a90bf8726c5617accab856168e605f4e9fb517150b6fd7f673b75a17c2d5b25377ee408c17e81ab3c9ef5214e0e897a2fdb30e67821b4b4c629af1f0d10000d6f09b101e1eb29203b5a9fd51a30d452f11e56ff19ffdbccbc942d8638093de3d378ec62629d1d49b5e53c735844460c56b4e4e18d767ca73e3818fe44458d670c01&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641606828Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/d, Name: checkpoints/384/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/384/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=507028befc1c8f7c44367fcf6c0dffe132636b4085e12b96bffa35e9cf8f69bb77a422fbf0a44284cd2a5c57c2fbd052ef7a646398173d1ad3698cac2d0fe98b93c99050fac671d93225e2cace7631c89e2ce6e05fe5342e20dfbb64718c19b8b8531c39fd78459f4f18634204455aa993a7a4f2b2192c7c433c8b6ab295a572afdac7cb68dfd9d073b5eeb10d74e9b746ed39c818a5db24f4555ddea053f4bb5def5ad7a318a60c2e1e03153de2568ca6d1b2a433f4452754de73cd04d889d22a1f34448b74fdaf6939d7bb3de2a3b5f3161f012737d7e4b20e70afded85cac7a929bb1d2a6b3be43e149acb18500711de23b8516aefa63c8aba44958fc1646&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643414685Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default, Name: checkpoints/5632/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5632/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=af90a4f7418aab11f84656f0eb8dada0f6ebb3b9b3d7f988d7987071c07b0b5938469ecaead8f9bf9d0f17af74e72530abcb4b71dd7fa4d374287f96280f7f3963aa90edf4e9651ec436306ce9b1099d18bd4435e267af8a118a0c0c236892d7a6f4cdd82836a8381f142aebf00bedf259083adb546783716662c0bfbae08062f4804ce7061ece5dc8f699a11b7484aee73713b9193ce4aeb9f6311ac92c9063a53326b5164eae1c8514b8e9a940fa3fb241db287ecef39ff12b4236ef90e00141252f1a354f180faf3aa2f58190b79913b1bd149655092975be212d4bf852912953dcac50e28ef799c9787e528ce9f7a8305a3af4365ff88e57704b231f8020&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643454769Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864, Name: checkpoints/4864, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4864?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7f78aa645e7ad4b4e4883c43a104caaa609b126993b614c26f139bcd38bf0132982fd8ebf71fa0a514f396ba21fcc1c015e4983c3095a8130e671863cf3cf819447cd329a38df46d04e704e64cf6fe70a2ea81e7a4f28bc51464f66b49406278b5f2a30c12a1055340881817a3432946ec17333c7ff72c5d2bf771b58eb5bc898fa22a0ce0d466755fafee485981268fd09b21bd2160c420a9d0071cd169ffa923f416c7306a7ea9056a1fd07d9a0d39e522061f69da5521793eada0b8aa10f09e68dabb671e033d4c1eb5e377ebcaf3e53521768337ea9917ec70301e2dfa3843c6b46d8c66dc0f841e2e8f0f96fff4acc3fc7d6aa83c5521cac53482ff2ad2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643449753Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832, Name: checkpoints/832, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/832?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=83c392522969759b61d443cd52c7e9d8652422267cc0942ea5f1e94b064bb46bdbbaff7768e007304cd9aef1eb9e2bfd4c51833e8c671fabf89547071706f738da1b9c8f8bf191bba01cb40c30d665e01ef046d4c0353647a60db3ce7c595cdc3f2a2f13af297a456c6fbac113d66dc767635c5a7c8097d3ce7ebeaec23aa11e0f5ac2e6743bb2733b77c21a84ede245911d4f677a174c63262055d2a56eda5b17e3a2f0c5c7fb51e1b717d82014b1ee85cb75bb2456f4fde48e6e0836009d1ddb63e2c75326db6974e26e57e7aca5151d48946d394c5df8ca1fc7cc99bfc3a39401d7848b79125356a4b1eb24e9e75f0f7dc960973c1b1c3cdbefb03f58c1f6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.64348032Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/d, Name: checkpoints/2688/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2688/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9b3ae13c9ba37ecc1ab74a82f977b2bd9f7e92b671d2192ebc9e7df38fb458276dbfc829958202c3b059f1fdc91e5c2047beed9be7f0a976f44ff3c929134644b6881c02ced62947c5f82734ee2e4a967a49df83587d0c3a4e8befeb42164f0d1c0497c4d30d2d655784bfe6da993800e3179dbaae677f187de6ff5a0cfe916afcbb1cc465141f9d395fe06bed661a7e5702dbb29a15297488fe6f8a96ea1bea1ee86564672912c4ff50e8f3164df623ef65e39bfadee32782082e88c445dee8ff218e1fa8b7a3bed174f4cd33140ac2884ab59a198d6a2c6b2a21389a0ace3472c249620769d181ba470cbb4f167a1bef089c391f9c2a4faf51e1c43071f8da&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.64336691Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d, Name: checkpoints/192/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/192/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=983e67d2b4d312b549dde4c202fa8b826e5ab251053a3be368fead145d77d8901dcc633f1c70402a3647638443b290de8e0f4382019aa9fadd16ecb63314908a2d9d96d41a8e649a86f2f689ddbcfe37f7a709f9d47862cf32458be6d863a320833c2c493f68b6735935aa14578c48ffce22f25eaa19e93d040ff569c3ee8956d309663576fcdf2dac50ca9cb64177537126c5a30c91049bd788b60a827cb719cd392b3835dbca05f52afe5e4d09f440e6126ed205c807a4206b62661df3c45f23ffac89b31c0f8d25f17f2acb6a0e0f2e7f3c37f9c49b43869f13e6849a1c13efb4e30975861bd8fc5b11af49cf5516ec4524a7a0e8fb621d5975bf2ad39994&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.64351502Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0, Name: checkpoints/1920/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1920/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0cd9105db1258fb73e19203fdadb8f44e8b63c1179382ac4e43fb4628c10257475f291dfc1552b1ab459cd5cf575687eeef1514b41a1602cca8dd897c88ee65b4eb5f6d43b1a2cf9c8d85368c68c36ec7b9fbc2c0a1053d7408c19ae409ef89f58200aac4dac768ef6543ce9a21f3880f1f083589f2337d852d67e11e2ee53a722ab03cda673bf68b04ac36d826dea024307411aaa45f8655c19522cd0c80ea9949daf2750a07264ceb2c21318cdb2bcb2a3fb382cdfadb44ec3285c484a8474a8e5e1ea6ddc67d26e5ff493cba8493d34ba47dfcbd4e4ee387a063dadd5d61ef521633a863876a9a009359b3f9d8593e56f94c99897556cfcbbc598014094e0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643534475Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default/d, Name: checkpoints/3520/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3520/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=00a2f73328187ea42c8a182efa09d8214fb05edcd91d6c495b8b60c805371881b3ffa15bdeb71be603886d15f133f27e0c9c791ecde2a92b88feec60a9f8afc5d49f1dc2c43f0e24830c67b6aa9d0589d7c83a7e740608aef67a66f1ac26003d40839cc8281a7ce2cfac07e8e13885c260a7e705a97ad82bde1a9bb4c67351677e2c822f578ae80dd81d9be47920beaae6c18f4bbe0269d51a2281fa3d3cd8e1c8d597a5400cf7597b7eea22a8e752fc4bf3a1f9b3f6d1c61c64a74e59a32fe4f2357e60e1d66e23abf744af07efe3425a287d0b8652295d90ab5e47b1abba9773983ad0db3b043ced16af817c0289f5838ae6229e018e3983744e29fe0e932b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643544109Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792, Name: checkpoints/1792, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1792?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=bf5033b0db4c3ea62b9620bc31fe2e39c66a3e654a82a79fb871dbcea3b47839fc325ca04b52ff0282a360bb3a5a1d24a509843e719a253e91466c72a2d0dff349a5d7af2d3a1ec2b998d4d04cd7240440cc1b7ec0df0d6b13085c1e61585baf9558b5cf5d605fb4b6076967e52cdbaf46aa220d45b4e5a5242fc22e35156ef2eef72e76cab50f8e460fbc3344033d98ebdbae1494c6c209e78c66c85dbb1b2c13fb6cde004cfd36254d719e47ff4075ddaa769373700d40c516ca9d73e0f24684f3fcac935d4ff973404f469b4695a4e008a0662557cd6c02bf11c50a529b7afb60cabc18aa5c3982830a7ffee2f08e757a8cf589090b3efe52149b1fc64d95&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643552075Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/array_metadatas, Name: checkpoints/6080/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/6080/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7f0b316665a7c8796cd7f57210a11c1824093d85040a0c04d6e65f49cf7d0441d282348bcb714147c05856801d0b9453cef1bbebf20b015fc7ede8c697436c8f269d96c4bd608a4df5daf0588447a2f46b94b962d4f4f46c1957cbaf9b19732650418c10e920b09ea95b471b0f51054372dfd9ab54211dacc1cef64bd76a3dbc583c2bdf73b9c65eeacb3e8338f2151281e46cd0d252d9db7fb91a4434ad16de92885ed2b68c6525ea87d25e9ec7179647d96dd9bf6ea79c3d43127f9dd7ea3fab504c906c75ada033bc0f2ff23dea7b07ebc4c76ddceae3e07cea5056776155ab6ef35ca18855d7993e22784ccb9bc109453595be1add40e771ced3ce4b1aed&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643567889Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/d, Name: checkpoints/3904/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3904/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=65886af3b5dfcf23ee120025504416b9680a65068571833479a8f1fe18912e764c2c718967fd9ad8501a21809264450995389205ffb1911f6d2995f61d6b43d861a6c9ae63f13643312f42e5c90a0b4810205778a210c176e96b3d4c504a308b8de361945dc334875b3d35dbc3f2300186074f4821c832aa261ed1b25632f13f17bdcec18d3d4587b5026b5ef97b94aa20be9898e595300d974905f6c518f721b0592cabd096f7d8adaebaa100a58923a810e330d6d80e3be3e2e5b752800a1fe80b5f01185ba915427fd20e878c213dcfc05ed623df832a6d6c0ba316a4890cab4175c2bae877c54203ad7e9a1144d856c8696decc8ea5941c52e2c893abdfb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643587385Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default, Name: checkpoints/640/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/640/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=222760a2c18da3b11deccc85b79254f15f3a2f50a90a7fd103db9910af04089359842cfcf42183ea0722ba6fb0f89e5e16baa89804dd6038c071749833d6cb176eae770bccef2d1754658a6dbfe690229150a23f286dd6618ee0ddfd8d0ba383931fd9df8dfe91afb517c58e9f467c41d312fc3c38d0c6a32a1d33831e0a8b08219fd208b303c395f9e82996f1a254fcab73254a9f456029e60ce6fcb478cea8e57a98a3edeea87fed937a98dafa1e202fc61331fd74eaa2b7e8e54d4b18962d5ea342ea740fd22906852a6ae4795ffeff8ba1e767b23731d8a92bfa2439ae9d1f12902aa8142dbc5eab5be7826e469b78948e8d4f60f5b6404701bee0bfaae9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643605311Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0, Name: checkpoints/5056/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5056/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=aa87f91042f75ef63186a3c5cba3038a626073f5aafd3290cdd2e9503d436318dd956e184935c3d4ed9da0eec9bd212fb8dba1ff1a27028562216e169d8a86883664d1204851b2d48433a7341ca507f39d14b5941e41492e17774392f8e1fd88d9a14a3f494d9c2a7a60cba550d11c6133b793ce632163b7308f50de90dad02d1dfe94fd090f22776c01472504641ac9e1ee6126cbf8e5f9fc3a2dea136a4bed7fc0416407d45d92cff198d85bf807c6ba42d1d54a82ea331326650561a39750ca4419c8dd149a76e399b53153e4297c0450ce0749f0922396374e287c24a1f99ffae7c579632f0b098d23ee1522dd051dc4d21fe60f7fbaa02c7fefcd0205e5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643628431Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0, Name: checkpoints/256/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/256/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0f652634a9a1d9e063ac4ff8d37879b0dca17a7885d01583571a38243d68fda0031832d0e918b32ab8f74dd3345f326fd471a562a603e5a14191988dadd4fe578a6669be13a18f8cc005839c3abcb1b4590f6c47cde276bd60166cf60697a0dd0f7e722b600caa095fa28b6097608bd40cd187a45a7ef0ffd698fe82aafadbe914c4e4fcc064a0e10ceafa3a9145540fc289c7d54fa67caeafdb8f58efaf8c0218780a0bf237cf24c882a173465477f54fcdf9fa570403954d62ee450a0ecf3f597dc8aa2c6896e21456012f01ccef2caafa7ff3c99e73c62a04d21bb7bab2cace890359a7749ba3c3eac4dd5d4d28e338898cf7f72f5c1d23b39c2b755fc27a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643652338Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default, Name: checkpoints/2560/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2560/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=aa16f920f5b75bcd5d5e781908da9f060a1407cb8290a0205ab661c96aae9db1a94466ea5834cda2f2a1c7215ce366de0f8c938a1bc32b9fe4de5affc69240292ae252d6593499832603075abce85ae2e6f9df710a264371a8e4db8303b91b31dce39a21c373eeb60ed77727dbd0912aec0fd65b862ac0466ee35e515956d2c4863b9b3402679a487ca3e1e299f819e971edabe642fef9dd2b41d82589e7d087bc926e89b622c7ef396049784ce900db54d36be225cc2fac6d2aea416915d6fc6c03c2de5970a43030fc91d5c634ee0a215a23a9b9786c1dd75fb2b2d9425fbfb2fdef9afc96bd3f30e210db63c5be4abe8562dee119287cb350ebe3654cf884&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643665435Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d, Name: checkpoints/2240/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2240/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=18e8b262b0cde3a9b16421133794de4b2e4dcdb77be5e1075d356c2d9c774ff75e82240db52152caeae2e5f5fe407d326cb760b83633ba6e4b7e3e65e4896cf6786807db76493e4295c9d31ea41625a77a70911887e8f3b732469060479cac560adadb7358b75af918534830cccc334f752b352a1e0810f9c323739d118b7c7b9f25a7b3549ace12060fd626a8b114cca68e74faee154f80b7019c34bd360b629a169468c71295792f376ecfea805ff1d7f7919f16b9235e68596791ff520891152a8e5879be8adfc2a322f0d9537fe0920049d2c553bc81aa47f5ddf5bbb348769854f705c5836bfbfff3d620bc9ca824c8a4fe9c0df5562f3e65a526c184df&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643674499Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d, Name: checkpoints/2176/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2176/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b6b784fb6deacbb5c6234dfb4c93a046e2603762b25ca9213fbc21e65426a2984f000db538c536deffbb1e4d3665d7c901daa1c1cd5f9453d406e3b500ccebbb3a325952667c3884eaf22837b913fbf84ab5a92e5f0a631edcf9d8ee57d4bd282732471341f5c64e79c80ae2cf2167da9ea6e828057309a071e892597bef28cf41eae00507cbac26a9e54f481e880ae9a4a4912efad85f1aa42fea3842cbd456282c452315150ba5627432c4d09101df8c54aa191eac92174fd681b62bd9a691c73923d1809a89647b721eb3094f00ee1ab7ff6834fc8bcd156aa731cda0afe133462fd6ba8baa20e3ad78cf180bcdde658021ae71ab273010ad3034c406449c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643655414Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default, Name: checkpoints/5184/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5184/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5631e1c4f7416bedeb8a10e84eac6339fb0a7a98c31360d072be78c37cfddeea4268a315c69ba56392ef33ab5095735a525d45b8c65ce09d1bf42c7fc5a980a53a6bed6e7ffdf6c4f6da49b29928dff40b1a273d90c5bcd0e4c8e651788dd52ea1814d3e41ff748ca5620d7c8cf33c1609ae792241e11ffde55553c258f7857204e83bd8db88cbc1ff5dac89ee29df1ee977d4bf0855c7d1f5a3c72e5a76fd141230ac3d4a88c5f931a60fc8c0d218e7ae26c4434e4023fb6a7ecb97c9e6e9f774f2ddd1d244b964799c80b593969b204f4aafb12ed60c4ec6b4a3f41b36897f0975a3aee521a7efcda0128383ff4ce3374e8d26a3cd5e4deab6ea9b12830e47&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643695368Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632, Name: checkpoints/5632, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5632?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=950896827fd7d7214bf957402126d752cf57c6ff7b7a81f2035f8256361a9fbd5db4e69b42b49447ed44a1d69f2d8f1cdb8386092bc8899cec3ee8da9e4a1e3489c7a36d4b35213600d1e15cbd974ac7db944bcbb3df46f5af36398f03d6e93db8c37c6c774e5a62bf9b631e357f179a9c1398d6d316995d1c88924c681527d065bc059731d4d590a0e6ee0595aa4c534bc7d9af9bd46e4bfc2f98966ec8a1262e2d9de2464ed7ccb0b5443ae12b23ef02ff71f7fed322c014f925f19637c78469407582a3b4d0662d257264638ecdf97b7e77e69fb712bc6525f5336528e566674c67ef847d34ec330c2f8309bfd9775e64c53656cfc29e158a0a68ebbafe1b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643715937Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664, Name: checkpoints/1664, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1664?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=712c4e0888314fe44cb6fd986ae06005daf8b04fc32320e9c2744e7337ef6fe92df8c3d632334919a4bb08334144871b7cc714992229871a6f19221ba17c351a93224736ab4b381d03a714e6646e3826b1b0fa9ef8eded1d461342c67fac8d505509db7439a3cff44a2ef775bee99ff671f7517aa8051deb997bc20af0c97b36ba005d68772125060766bfa97cb12e50a2759794f2a5294468fd8b7a427d67a8d1c589e6a9c95d4f07d428116017e6737853488dfc9f62617206437b9e61b3897f16c1434c22f0cc50baa31a3f2ca23b0f7d064b84853882d809fd864e571254afc5e5321e3430e50a18d37c186bc881a1626efc2c0463a26183a7262c0c672a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643316907Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default/d, Name: checkpoints/5440/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5440/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=8a80ac601f5e3e491fd7b0c5c05110dc8f9f652f4d92945dcd0ed95c30c0c3599c21439765f09ebe7782a9b8ca45f972ea5b957170c731bccd72207a15c7e21e7a8959d0fa9135cc4fd402b9681d3e06b8ac3ff8e331f15199e6bf3e6cca5ce825cf4f6674a64e840f14bd66ab59603517ed0cbad668fcdff4c00a38767cf95e51236688e03778e468604adc01bb7036f54c7de17141c8bd2417bc5b144b96d4f5effed5248c2702099ac3df31f251a6ad815caa4d464d8b52e490d7458a1cedd31634b8094ddec5b3d80bd883127ed04cbdd9e3eda375b564a60e38c21160505abb009c4527464f95e255769b467a2834cd20da3dc01aa652ae6cd96f67ea29&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643611238Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/array_metadatas, Name: checkpoints/1792/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1792/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=96b18151fb221657370b0508a6f60b2910ffccfa327637133ffa39fd0abfdf87f0b0eb8f26aed397b35321b3ab3c24efbcf7423dd2dabf22d257db8076a810ef0c1a591d549f572868e8b09cd20dc7939e8337307c6abb80f23d3109973664349f23a9d8cdb02ec2c034cfdd4074e5468dc0d9e98ce226a1f03452fd627d0ede2a036f48dabeb09c52e661374bce339367dc5acf70b32029d6cc51ec8de52ae6ca5dd22685748bfe3952273e6b00dc0ddbd92cc0401c85652918ab3b981eeb8e9bf8cd2f620fbbf302e6f64f36e9f1a6854faaf4bb2c571784b8cbf3d80cc43293a11d444a15cc577bb7a353efeb147bb543feabbcf4f612a071bc8ec9b9bce3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.64367178Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default, Name: checkpoints/5696/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5696/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=29cdf4b42123107531b9b0587f3872f87a474b39c1a60e1acb7c500513f461354d4308a5ed3245e7add0a131396747a2f2554aadef54fae05a4cc83fad920cc72cc422a1857a637a92f2b1ef666ef79ca0d4650791877539fecaba7e7616ba0df982e9249742a34173ceeb442e11aad449dd2d35f48b7430e41628b5f099cc67a8c7bd0de7b94471ecf2345657d76cc38c73ce404adad2fb093ac36361045ddd7c555a93c86a489747068fd80722a721c2d7744ede76122f38af8c515e315d35af05909ea62fbe388e8eb5eca5e6cb1a1dd1d99a38fce852dbbec8ca0e3fee1770e734aea9ec7712ac1cb0592fd882e5c1b0e98a35d0b5b455d36483e75db63b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643477054Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/array_metadatas, Name: checkpoints/5120/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5120/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=42b4d8b46ffee29fb4ddeded6dafd8a6d2ef82b7385ce6dac3b849de3be0ffd93467ebfebdf74adbcbe8521b4886bd0b9b79079a6f79398bdc359da0009d563c67d5f61da30821bb3f0005a1de0163e7d0ccffd1abc28c42a4965fc0140cdfc4f4a31d2b6a8faa85a5b7924bfdc0c745c0442f20c27cfd04d2e63aa4a0673055d30ce1331c191e24836c14f1cc19d638d0e5ada17e6c30b71f1df7555d84a1566e080d848cbb8ba06bf708690b31916fabcbce90b3b4b08aa318f0c766930d61dbf88561a123f6011a09eca87ba44acf2701971cdcd1f9ae0762acacbd9aecd20b33643828cff1599df3f35fe6d900923f988e25a6c17665e38345424530216f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643224311Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d, Name: checkpoints/1408/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1408/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=683934af85bbf1761cefa2552efe19dea9fa9abf96003a0e07bd5b3af394bf3fdc2415891f2d53f5a6fe1a51a4d8f13785055c20dbc800a7550112e32ab05f8f0f10527d8b443ebc48125cf84f33bf82e01a97223a5d3dcb7fd4a61869d4cca99e815ea98be33abc6a562e82594ae6cc6575443d6d8ebb72cb6e957a0dd0ed63030ca9777ff80b951e626c19caae5f8eb8b88557213baec732a00e152a9044d110650fbd9273f9ef7e7da898900ce82b7c38cc88033ef44ced8208dcfcde70a962a720d05e3a8ca6e94c1ab70b8c1d736c39c496b75126bab5310dc6aaec509a2e964045e4e1f5e956d217e183fc8690008870733612101314e39d3e92b3a531&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643544351Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0, Name: checkpoints/640/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/640/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=23a80d625a0369509f4b917047609b3033312dd3dcc6d370f65f566e14683272e1bbb06906f846523d70f1364dea3a23b8ef4807b4ae048fe9680d22c060914dc51bbb35161c7ce12d1bb1c75bd20483adfcfbe10830a73479f0595d969d62fbbc90c48cd1db047238c1ffd82bf7dfd39d63c83443fd82e28998c4d14ce1d6ffd4fd507212a09a9904fde11eafdb5892620f69975b8a7326e211a329dc8b43e1d048ff3abadd60008d76948af1124baecb1bcb8a31b6bfbb4ee45c9ecf59e343b3c0bb2df731c36cdc7e9410043536ec162db439c2253759f827069d97041405e67a36658d46912d06b39339cd268b39aba84f2557cb9c374f34592d2a6f61c4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643377864Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0, Name: checkpoints/5568/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5568/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=555f4ad170aee37dc3753a84cec399e0e344c6623994bc49847889a4bd6341b7e43be349bbab651b70e5b1d900a7aa2bc764c0350a916c43c2c6ebcf1698ec4ccbed0fb5bdabbbca19231d5a5a4bfe921952d7177b94c2377e0e94b1189cd228c7a1d6144468f432aaf8a16dead7da063d512c75f179f30e7bdf74e04f1f1523a219a68f95569981c37195f1ccacc9ad4a208a3221b2e9a1db948689dc0b413d1238b88a1acaf2075b345900aeea204954009d956b0a7324f8dfd90071df62387c834088de5d0c0e8b582660b3cf62fd14dc00a6689de29aa4ec69b6e7fbb9a3b91e589b330a127e2fade54d48f23deac46867890180b070865a5fce9c766f91&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643429212Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default, Name: checkpoints/4288/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4288/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9a7a6288fb571ffd37be2205e664eca2cab6aea813cc7beeb85050513bb665037696bb9b96f6aa3b53ad414e48008d109fbb5ca3c5c957d759f04d3f7923a6245d72e02ba81d526c4ea652a7f0dd5daf599d6b926948bdfd634c7dcc192f58ed7dd634669414ce72daa825dd3a53c8b83db85ecd81c51a14d0aced9468523cbd9bbff76167bb07c78a6e63702caa33c5d91c1767f950ea384a5fb1beba1d96356073374443ded07d88c2458098313ed0504ffd9362bb484989d846e19f49152bdad047e013f124700879d1a3c896b5fabeef09a74292c5dbf154c787fd7d227dfbf318c8f9eb5a59239970ab9f3580db1806e49b24edb19792f903637d015549&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636488713Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default, Name: checkpoints/0/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/0/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=2a292297851639babdba5667914b7cd5af7c2a9117ec71c212e4fe0e3b358d4faf70a15a4dcdfdda59d8f6200de8b7edf7241fd7d0947b39727adc27a140e85e01f9644768a75c0bcb6640f7c5ff73630d86ab21c910aacd2db73d8282d0ed7c54163a91cc6fe0963fd6101a7ab0e8fb01c858c576c37f8f7b0fa55c96c8847fe6ef9fd9323bddad73c62a85c11b308128263bc8008d3663d4ed5cdb9c083c6fc802fc4c2da36adf5aea911ba0bf4615e9339fafeffa654969e014f2a6ab256bd1922d284ab53ea05a16f1f7526ad8fb0e04be120e277ab44c5b822c0f17fd43bb7e2da378b589e804904de7ea6c1f5a744e1aef1f3c45148ea3566fe03c2d03&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636508541Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672, Name: checkpoints/4672, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4672?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5a4bdc6eec2a99c8b1dc4ff1ea0e7c4a300ccfe45671a290ca4a2e005ccd1138afbd85dae97a5fb9e0047fa86d98806341c0a4b70dc04aebc32fe74af23575b7a8578b83cf476a2d8af2a4ec3d8784c9b734bc14f37360ed18ed2d60b29be03ae962e6197d6f5fec4b8f3e47099cbcff638925f134729fc86d95dfac135b070144e9d85032266bf66590ac01e39cb1816b4df2743bfc9173c22111bb01cbf2fdacbe5ec1e1093d1ad95f768180d60fe5f132c1492e9a46ab78c3964ddf2f4c64c908ec9ae8652d801dfd53444f46b54ae49a03ff8d534fed145c0f81eed0afb812a352ed0416eb907948160c67ada235cabc9cf3cd58919921a7cd3864db8b74&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636513941Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/array_metadatas, Name: checkpoints/5504/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5504/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=26218ba60862a9fcd72edec9befd86d8c661389be70eeacde07c95ce1262177659548877c06517b117431e1b84b4b864fda5f6f4a56bad8780a7df9a1ddceaf0ab92ec4b5d67405fb3875a40e7ea531ae0536da31e0ca2b52d4bd0a069b8709813acf80638cc893af8773e587397a95a44f4a05c9a687b1ab30b55ccc1531aa59d010ff04a32fbd3dc79ed9c2c24ac0701908246a649b3c1e5e4217c1f2d3ee3967f868eaf14f23037ce3a46b03a2cb295a138c8a7173cfb2ac47cf35563eab115b0fb2d2fa56823af7ae28376d46e04a8ff37d7ea72b28541db352f161aba5a61e1bdbbfb0ce3f690f7704d4f6fcf49401efd6717d13cd7bcc88ca5f8fffd2e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636746043Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/array_metadatas, Name: checkpoints/4032/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4032/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=089286dad9fe42b9a8b7bc73b188529a3f9ba4662fcbb17463039308d2e51b454b702b88e22e69be3a7909c46bfc42fb837eac21e90bf2459d1e4a56778f3a4d3c09002d480c42c012983d9cbc75bd143f4e77cb001a54fac4380cfcca5c7d6010578bd2350f4bc9fbe0dbc3a577335064470993460764e9503819672c61b9903534c28a16af8d89e9c014a0d2abba42e4a02820ef76d0b8553db474030f2aeaff9ea692a3194d887b660b3f04cd5b87110c9e65389d41bf986be808ec57fa755110950862a354e5ffc6895a6ce1c9b2cf42b09825f8a8882f1040ac5d5c6faeb67e2c54ff8be02d7c368996995cba97b3bcd7830a9b775ea278cbead0a388e5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636777993Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/d, Name: checkpoints/2176/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2176/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=8cc42f6e904914b79bcb4b6f4b31064d8c81c1026867dc9778be682c76ad95ec726cd2e900997ebe8d87434adc99339672e7fba615c178b893c1a2f208a88b7b0b9c0828dfc4940bdd9ccbca94a304e159c136c4d5fc2b6a2170adaba74a08b5915308b47db96c4afa3f52bef8e6620ab5beb375d2af844fad15d74e1776d8180a8f7932f522e83fefe62101afe354cbf4483cce2c3a71852572eaa481aa619e842aef43cae19a765179b36316b09a79bf6373ac0b50c3129e891c28864cba64513ec9b969f24b3df7038364c7f7b82fdf142ea7f5446d8dd04c74ff34c2c2b48801ab0abd485d0159e905fec896e4a2731250f650bc4a8ff2eb817aa960a2fb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636789652Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/d, Name: checkpoints/0/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/0/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b0fc4349658100ac885addc2c024d0b490ab12ee41f9c76c68647737be4547b74d685cd296d313630d6646f5431a48c2315b191734fe0bdbe338fda13d09fadb624532f06aa189cf0816a29ab237a6bed34d75c994344c6fb8523182dd2354539a872b3c70f61f3b649a061fefd08b44af6ec78312b953a7ae7c836ec0659fe170dcfe91adc128b7af895675268389814ed7f2f1571914956e7f3e1fe067df4a1dfbcf47407e5f62de1f2e1514715ca94d056573ea9a02d0faacbf4c570154f06acedf002f33acb6519fa3579afc95702a2e2b765df27d6c804974ca4766168f12c7b1fbf228b9d7a9f480036dd8f83d30e40a4ec1932bce60cc8b72709962f4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636917504Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/d, Name: checkpoints/4288/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4288/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=17023969daaa3e43fd4518faa877256862877c810985744c95a85f999f549759b119ff3e46a1ad8b5865787df48fc095b502320e3fecb4834b3977ab682e70fb108561a9d1d2f363d698b3c565880551d7ef01f767483a698c5329d4c0b2df208651e3b1eb13423eb1d0a3edf73fcaf010e7780a4d12a6a72e0f524e0962a6ab0a7d91141ba70df20e21640924a824534facb0c78540ab88ff836cd620d971d1080ba45ff5b1afb880f6d29cdb6ab00ef4727c78a92c7da89dd9398f0ea3147c92bb147530d50a2b3245e389baa5f1c618f6c9ccd438d9633e48bee21606d04875488d3513f7912eac9877e3221ba6ce19d38c589a264b2dafa8a911b23fd504&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636927963Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default, Name: checkpoints/768/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/768/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=2b9520b144bb3929928e9d8af90094b5444729b6464b217c95d757281ccd08ae2a45cd30557d06a6347b55353352749f756720d367fc9c072010d0fb536011a61d66c24f2ac015f1f7f50e8b8a4815a8fd7781352671c2c5d5991b09fa1fd14d0bc3777ccdbace4046830e1062602fe34c0c14ae7f75eb70da85c485018dcd1380d4a68eb26e072d7b6cf600070fdbe8776caf5e93b540d153872c2c202a1d6128402706c8b328aa1fc00dea744afb59405b3da771e1c9710ade44a8b0967eaf7f01b7d0453f4944b7f0a11c1b37b52716dbb3d66f9c6b68c4ffece45ac2bb77600767e403436bad578b68de17fa06065f275a41449acf1e4aa062725979d49c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636940333Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d, Name: checkpoints/4224/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4224/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=be8ba642264644c064a4f7a29c31f148379bd093897ce8d1cb9958671bc5c3b860f6a7ec85273bc324c6bd60f99c12e00ab0b0c4c091f5abfe71b2acf654afbf173542020d7a516b275abeb12b268a3c956f7ca9091ed888696bc9151cf33c230e802418266ac641365d25e3c12aef0bc0ceb9f9eda5e67380608b7e5b7b34a709e1a4390676fcb91dfdf55764fdc002a563349602296d9a3842f8c71761bce12dae7f5a445ba4a93416564e30cec86bfa2866a64c5cc7a4ead7cf9ccd5bc153c2e0776238974dfe13def18371cc0d68b02db1863065b410ce1d332593cec4e0935bbf0eba302248d76f5ab205ae851e2a373139b9330987c959d6bbca4cdeda&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636953869Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216, Name: checkpoints/1216, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1216?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6cfe89bb11efc3944c9c1a1ccb30f24c1cb5f179222aaaf64e9d0bf36de8a2e2a1672ed632b2327b2e2652a608631af15ee17e14d8ccfbed2f03daaf70fa4cddab5d11917b95eca2dd16d92717815c7193b1711baaf9c47215fd463ac58f05fbfc4273ff7d9f093dfd5410402dcdcd6104be7c779c27ea39b214f690b31150eff31fa79d575564981b0cf282632e8b6308aef919fc1bdf92ec34b54d9822689a2af5ac3a1c36bdb8cc4927785e5f03f50558f3a519a0911349b230b0024f8e6c919dbd026506543260c2ed4ffa37b713ba467098c3868644938f0a9668a411476a1e54c43e7a6a3c1053403433155bd9510b9dd80c047fd803c50032e3a0165e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637035357Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d, Name: checkpoints/3648/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3648/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9b5c879fc6517d944caa970668a563c3745403a160076407dcb60177bff30da95eff24fcab776c33a61d59ff96ee120cdc1cb9bf58e57a7d80c243a67d3ae56d16f2371169601643e7d5345ed0f8b40a0022703b0da1bc1fcbc51ddeb5456e212ff96c7688f398f42e033b6e89a97a88a8b2797f8db81aa45ffb99015d8b5b786c7ffe8a01433e91b0edc80579ff1a2d7aea5e6687e815ffc29307f598c357d6d5e099a842dc9c66b6b25be1e8ff655d8b32bbd8d5a9ba43649a384fab504ffa1d3c62b037c1247937d50f041af78152cd4a5a1bcaa75f205f49a30079f3192b2648cb3dce7de5eab0c6cbfe130c24a0bb89fdd25302914582bf33cd2350af7e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637010915Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/array_metadatas, Name: checkpoints/256/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/256/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b71a5433d9195eb5f93d5eddb68424dbc7220661dff2f12733fd24e9b1d50e903424cd352a592662145e4c6af3377def7c909d6c2d80722627a2fe8e3cbce2f591b6681a3adc1c58c64f656c23c4af824085a00214f3bbe11a94a392257f55f77a3014fe8f5ebc3fbd865621cfa050171e4714fc91b7fec98e6f4998920800fba5873e3c050fd25544787a9711db8755af6b7a4064287ff65b9e7c44471fc7fff32d9caaba99f49743e5c482d0e38b85a0a2a33fbdc7f5d0894d8848c0efc698c66c985e0041e0b2c050a38047372cb8fe7fcd7b6a765ad1098aa13c430ab949558d4a888fcf1c97eeac60ceb9caab0fd1a68f50f80dd2785637bbd39ebe807b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637068903Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default, Name: checkpoints/4864/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4864/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=adc416cc3e394d6435ca5db1d325e8386866df64b613c5b2cf42305874e7be5ec4aa3c756079ca7d2363b1d0ff7dd6ac47916b7980f929c3068589389f4cf95ad4e75394b2dfaa3e0a7ba2e3ba750206cd2268321c7c744f1cc577a29ec229bdb0b6dbf530a5a6fa61ae7290a0e706638d2d7644aa65ac95dd279583dc6d9ea50aef2eac8f6141f9cec81c6c31683f17a29ef604e322169add169c146632eaf40356415cfca2d6e9668c99156630de2a575bb9622035918ac0590ebb0e2188fd275907cf70ac2be53a4992e53a2c42371a6ab8fa92abe33d117f8dc9c8d4234b2c6bc1ecef923bfdca946698ca52d82f0193a4e284489db713e8a9c792db505e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637182738Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default, Name: checkpoints/3968/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3968/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=021ce772ea9a3cc0de2e3eb22317c2e99aace6262c1362c1c42bcdae5e2829704b1136281f3247a63e89e0b8f3e87919af3b4761b83e3305d4f9c1f194096f391dc7eaf22ef1d2390ed4175aba8c765d5d2ba1391190df145581c358a058cb1790b39a21767696e8b8d0ab239363d135e42eaed171328fb3ae7431fcb81c590c3c8430cf6b276105488eb2bcf6978e1d705ca5ee2fe18278e3e7299130ffc93981beaa1124f18c41df960819d47d5a38f65f6d3112945962054ba99b080b1458913117d0b9f5de452591dd1ba1cd6fa1a97dd7753163c89f062abf3d6f2872539ff78da68b690b28273bcb6517d71bd3386ee6ee6ed3ffbab9189a3adf3671a8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637270826Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0, Name: checkpoints/2432/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2432/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=12a0801f2ab23ffc985e502e1cdb41799f48ef9cf6e0d05d544378c734759c6029b55a7496d058827cc06202b0ab62e47011422599b11e15835bad4d5d6ad0bf2426876ade0a03357d3ec9fc3d58bad13be7f2499ffe979472471b89cc8883a9d1bf8458141ef1c09b5f9c8f90b9d6ddaca9a2a2d345639bbe431a25b4d2110be0cbf4ef07d35786f7676bf583fe06f60f7f42beb1b0aee3bc5c6c0254f2d5807f1bf6c30252a9c1ad0952c2f93c790254268b8a1647857e1f7f9fdc308898ffe0846149afadea46b307cd497c73a751b99212f7b9ecf1308c75b86e2f82c6e245e8b75fbac45655b0b4dc5a39f9b45c622bebc7d05a747efe2df621dbd3e339&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637315736Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736, Name: checkpoints/4736, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4736?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7d4217a241d75c2e330fc26a62b27273d5c3a9dbd5158a1d76c04b06973feac5544676209f142f054c24c49cbd933e90d89067653a4e90b01713874bc5352fef5c348dc2fc5f7e961ed406dd27a75e14cd6fe08aacdead32fb3a95b8012eb052e8a5c832b1401f4d1006449aeb559df56eef883cd797a5d647a6fe55171f1dc1c5db8b073ca8faee3d14ff46cd4def374819b7450834077e37193426fdd057366111b97d4304c00e86799e5e997b63ad1856965b3b083fb3202bdaddce04d40bb60a6fb65b088c2d4bfe3e0900683226d2cfcfc9a83e20a15c48fe227d61784f40e342ad696e02bc97869ae195a545f570abab08fef87f5f4ab6520f82063b3a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637318529Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d, Name: checkpoints/4608/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4608/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7be9f2bfd7f8551a2afb25320f4c5efe06a7d806d5512b6420779b3ade338ed90a6650d293c77ddf132e0a70a2cb5bc187d249acbe6598aaf288d04062f0ebf653fdf35aa1ba598a389ee3c8dfdf8e685b9bd2f2f23d55eed72f3a7611379a50499cf2ec3952516264e2652bf6c63eb134225a2190c08c55b550998a9a46b4132cacaaf39dd9135ffbaf7c6f7a0533e2d6b29132b0537dca556a8e9c5b2cffda07348af65af5c3e3d6953ff577906daa01cf81fb6613145b66962455c229e846d55534152e2bad4a4456755c609d257e0649113cd177e6734398ba463ace4d6b6c12a34d1fd346d33829e4ed4824fc4e4d8fb69d0c9e8d892892b3ac1745a2f6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637316039Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d, Name: checkpoints/4032/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4032/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=c38219b9a11b370c64de2c3f5a517e55d6671bbd7eda96230138e8473cc1334d2bb878235e068d4cfc22b78471f536a6940d1918ac014f17cdb6f9a996ea0c8d40c1e0f34502b360d890fa3ebff1a1e5ad6970a84c53036514b4f6cf54f210f6d3ff31b9de9ebdf32b9b913911df03cd91168bb5056cd5dda01b842dbf4c542b48a6a13346f262ce64eb5e0cb5e7471812356a02b62b944e6821804ff05f98da730ad15ea25e260f6ee4db81e20828080632f42582432bef818fb2dbcae7fd9e455dfc1834b14ec7841941f0e5bad3fa0c63e00e9b6ed518717b541116ed56d1f53ee57f359dc4b24f862bfa49e5a2c6000b1eeb26def43ce5c41fffa893cf1c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637410451Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0, Name: checkpoints/1728/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1728/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=144fd016bcf9d313e990c3bbd9de5880187afc1394a6519853d9bd5a68c90456c7108649e2142bf738fae9e5462523fa51dc545133c0727d0ec5620b7f391c17edf21c1cb091a3c89f268c430171cc5bf4422815901bf22a7f57a146510b4675cbd7713cc21b6fe53c4b37ce128ee1d265e4f59dff1e0c4920f93b8ff4529f7b5e01fdf51ca492a10ed95d5ca16aa4e23d2bd71093b4614d7f0d2c47f95b72659a9218cb6b009f7d2bd116f4b963792daa81b5929801fe869eefa48922d928498842a09886f6c9593edde17a12d68b25ceb4921763ae6a66da54a470fbece1a930c459e9629bfc12f1f340893911ca1f23bdae3b3d028131f1b50f353b38ac73&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637366472Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default, Name: checkpoints/4736/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4736/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7b3a486f74a8ee93c3d2761b5a0746ef5eea9542cfeec23461923e3008216ac45055d2837f14cf199706ccddbbf4f0413555df07cc5bd6d516d42761178570730c70c3397706f3ca3609dd0a57f33bceeaaa0fc217e135cded6b62e0c5b56cbb3e67e4e7a5bcf977238187fe10e6878e60e73f09d91698a53fa7696e6391cbc142191d76c445fac4527a746d59ec246865297df5697b4acfc56ea97a3315d7cbee0ece8633c571031818ea6f83d74e5a3ff8782326102e360cff49731659a2be7c7c50588809abf94649276e0800e3a7d4404bb14fb935e9da87190d2adefcc1f34470e898801cc2f981b5ce732a730df877cc1b7c1168bdce80a1fc748dfcd6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637379613Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d, Name: checkpoints/3136/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3136/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=748e47b452747dcd8aec16f28a30d0668d8e2a30da66b019ca62c59bd1e3064c89b82dd6b99add057e2e3576297d2248fcd6cbb4b4925f4c70aae20634c8d6bfc5e7547931dc31989620cae2c8c3ab4913365b156638b00882932aa1b95ac6af98fe44aed8a55d3390163ba34f4297aed5eb8a1d5383979e3a7ccc41f2467c44e77360421de30d570d31d241844bf20756e1fb688c01c0f899dcda9972d803297341fa92a8cbebdc8b07a2ec352f8d5bc021eacaeb72b4db9d0c9dd6cf2e0199c157861ffab94eab34a162c30d538bc75725c9266a8e043b001692667df652180b1144b26f46c4e117616abbecab5d85ffcd613314735ef7c624c7283e7cb39a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637417699Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d, Name: checkpoints/3200/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3200/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=041cc341ce1d094a2878f6c777472da4a9e8e4574c759fae1c83789dfc0ff1568427e32f041a0bcc7e9031923544a3ab94048aba6b903492bcdceb5c7b3ccb0dbb0b47e08cdefa8b0d76f9a3b9c3bdd322db9cec9df18f330fd82ef399edeae0370830fe4b2c89a5c0e379acdf662e594aa46fb7d9e1fcc4a4c49543f473aa54f1958436a7547836a58301f3cf774c0262b4bb18894fcc0b250bcec6cf796b7af21b77db1516d468000b8cbb9ee567914afc784cdc5a71920c551a905353b308c677b0bd9046ed10104179f18786d73fe5e4eacd8491d8025f6cf2c4ae08f508444dc02b03be79e9a8cad820b8fa3791deadecf52b1f30195913404093cdc0c2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637358378Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/array_metadatas, Name: checkpoints/5824/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5824/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0c9b0dca8ab13913c441fcf1efbdd2a98eb0beb0e99cc023eefcdf3d4094fe4ea1c73cd31b1c8ce9866b4a69bd9ef4f284e678dadd840b05f132e953f9af7f819cfc220f824e195af87b222ecb00b2d8aedca7ba6bd28248a27677141ab3752602d9308bd486d43b3c7b34d1216a6a27f79c503c5c3c2d492b480433b6e1bcab7d6981a87dad62e5da77d6d55408d224e223bba105ef98f3df4a625f27629fbacdb1ca70e4bf7b391c4f77b68bba1d64daa6ce4876742a16e4b1ab27bd68c3c3c484d4377349599845045a6c0fab4882a61a3d151daec6eda95d46c55bf2b1474e7743bb81f8c451e585f93dd2d5357dd17a4d933a7c58798708f62eb91502aa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637433446Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520/default, Name: checkpoints/3520/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3520/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4eb1a96ecefb24ad8b06b0651ca63deefd020e032ea3c77730b4f3bbbd490c857e83b7f9deef005c4672bfab74fd0d94a3d404701ce3340e43df6f66417dc0d852e4854f045d7cdb3a504ba5150173640498645309b86f3564558cc8ab38700c10cb26dfdcbd5645433f5b19679ab8d280c81badbf771eae0febd8f4f15e0c97475a741bcc28b05b32de36223ee1bec65e81cbc8728a7c78632a98887d0e3c5f31729af2dcb2d5e0e7bd207e1897468394058bb2b794ece31527afbb705eb3ef85d726d6f994d35c21b3e90a4cf254c83b5a59e654df63bdba0e4626efae7c99425649ae467cf4bfa394c9aabc1dd4b2c61aedf5fa549d3e86d82a1f2e27c29e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637434579Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default, Name: checkpoints/3776/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3776/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=2459dab9bd17b6722812a0c80380cd20b46177c9838935c388c2ec5ecc18a4ec37d02bb4da83d9b5ea35ca8d6e9846073158b2d5ec42cbac47c121b0bece1b28f26ca88ae16e0e8db81d6af79b21d94eab8679dd44cf5205d7a900833aa349cfe8505be3a623b94b68374ebd20e8f7bb620f6f1df2dd9a9f13b14487ef20497064c18a0b2167bb327158c47ce47f1ac397ff2165b3c81aa36b507fe4629d3943869980fd5ecc531b16260c15131a1c279445b8a3d616d4ff8221e4d043a834f3a0fe35d0a5e9fba5364e0932bded4b4e7723e37fec87d127c753f815aa0169f83b78ee63520a66f13709f94e8d499ecc9237d9b6aec65379340ec305256d74d7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.635938138Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112, Name: checkpoints/2112, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2112?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5855caf391825a21bb96c524c8e2e697cdabf2b23f73b95ff5504127070483f47461a96a396a9f7d0f07b8c80c458a5287a6c1a532067dc63cfb2ad2cf34933bf8e197834200f5f2796199eeea2b22065bb36864269632d6b4b65623e5ef2b54ae4b3b4e2b63b1c64a8d334a3038dec81ac103fdddc11015db4c7a929913c983c1640555188f1aff7890716024942d4e4048ae1817ae470f207eea23c0d7241325ae73492db5474d297bea8bea6a3c11ee19607ed2152d1c6c5b2ad50b51900a56feb3522ce2f2c85b46db5ae2131764753a9dabd774536586e3f5ca7f8a5cbf68bf1dc8dd1a4b0387da0d8228af4ddb2b5793a4888be72a5a726ad8190010a7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637523643Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600/default/array_metadatas, Name: checkpoints/1600/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1600/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4e33a1a1abec208f305efe81cbdf716a8e0cd350cc6aa6a9083f95cbaaf12ea348647a2f38db79db5d49174dd4b3bd50c058d7cec4d9a1e9536d0191d15ed1901dfb521196b38b314cd9c0098cdf158d4b8c8c6f6257a0cf764e172bbecfc3f41c49469f11b7410ca1b9741a1fba9c655240372a69a86ce34598780895ce633ae53bd7ca743782c87acf4ec61306262df3eaa8139ac86e09a16a19397c99cfa024eaba02ad07bb82cdf23a885bd18030f09b443fb318e8996555271394e283df96ac537214d1aa6b0b320c276a6b38b09c0c971b73df3876563d802c7b5e06de54208915af858d499ada880f36ad94b5dab22901e794e7b53d0bd03297daf05e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637526661Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/d, Name: checkpoints/4992/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4992/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=815d6c3b806ac318d154af6713da4969eb34eea8e95d9622e734a438b7d81c1adb0b7c3a7e0d898f53ed0fb0cfecea0552ddc6e5299e6be6d6fbea2f0f341cd20af75a9f207c52fb64ace551e8ac746fbd660824f584293bbe531abe6f09f213b886ca569358a3eb3c1851e6decea19c2b96acf9739b7d2d9a9ca2254ab4cee3cb5a7a778696021884ae017b4631eda4c6e724d3e12fbe0c91963cc8d36035457cc59d23aedc574273e50af15accb0466970a3593e7cd06087837c6f5813aeec55c0471c4c282908bc0914d06324a07b91e7d7e322e01236411e353834a1c3ead29aa1b3c19a1d94a67a248b8f669f94cb4bc441ba2e74421c17461c81eae57c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634173198Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776, Name: checkpoints/3776, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3776?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9206ec00a070f10b473ad95743134d1ab3c8f7885aa01ed6fd8ef66722a9d06fc9d8ce5932fe9b41f02a54eaa8ff39a27c0bff393bf95c32736dbf3f789ceb2a12250450138f4e6a07478715b6fd8f39ac3fe88c63fbfbecd11a06eb1d20fd4888457f8fe988eaf8820105ddf0b1c21b79a5a5fc819a5d8ef350e6926aebea0df0ee31803386d6b6187bda547113e0a0ae6d674bc27186c42dee1831ae17d9698fd41aa63d929160495de049e46e9493a70dd0865999c47bebd3ef424aeadf5a4a3641af1df4be094b63b3c1330cca1829a01dd285fa2a0f4245ec97956e16f0681b4fff1a8e4aec20589a38117027f67440a0e3707fbccb0be62e79d54dcace&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636558763Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d, Name: checkpoints/3904/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3904/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=69416406cdd6c12efe8553e4bd9e6ec4138004889381fd42f2cd91f86f3e2810b8ddbec55585a07a75305a97d21f4f3cba9d2480be7dae2c2ad13d9685312c26506c77cf49a36833d6d36a159c5b3520ca292325b3cb5cf93b46bcfd818f6951a2060847acf67efe98cab21af7aea6ec0b4053e17aff17f1d19e98cc1fdd3714d80dfaa09264c075aa9cb95071c0763b6f1d5082524bc92da0c2bb9d7fa275229b819538a21316052b4f98f5aa2729740bfd32de9aa835c6850cfd54f63e5a148fe8228ccd785a0a6cab8c52c04582a7ff231660799e34c2b4583a3cd06edd1b1e39935278355af3c5e4865e2022b123f4a4a2e3e911fd8f5f423dc75bf88842&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637736906Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default, Name: checkpoints/2624/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2624/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=71948defdb576e6625ed6a9d04a16ed7623df8143fbee7b60e7a0ed2e8add90d02372dac612dc677af101f59af80a66693e4be627ee32b14e9c33cc7884c08c6b7431283795518c65fbcb8c94426cef0046121da2396c7fbe03cf4a408c975d530ce75114401bf9f4667584f2c1d43d63ba4c1c68283e1aa610cfa912e27c2ecc33ed32311ae043cbd9af2d09c62bf72249464ddf25eb12b7196d75ebab80bcf116d063e7419ad7ef0be714fe771cf1ebb0dd53c9f0d1c883a96d3bfdb47690557f703cc66663ad8f11b41d528329423faf189930b9ed71b475591b8225ceca83ca6cef0494b88cf167ecf8d1038cd326fd0b377967ff912f54732edbea1fde8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637768205Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928, Name: checkpoints/4928, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4928?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=bac12b9baf4f72a9b789eaf6c98a8a262b2b7856d63336ce8b887450f850aeb78c543731f4e17f8d2e618e18226d18a5a3dcba35d66719282b091b1dd78ac8c858f38e82d3c081d7b85fc1e6e276a81f5d05a4f12182447a69278e6bb1d3b354bffff19fedf4c6266a653930851fd53c94f0aae7a6600a075bb186b64a487da5a0bad29ca7abae373c065376ea95b05c6b76d983073af3f89a45939e10abac01e1f752abfa28bba8cd6f9006f3e568aba9c6b6d452e490cbecac1a89e60d99a82eb1a3a6978404fab8156e852db512a9791a66f17a57956cbdf1512db4eb8227a44d048a60e389eccbe67b7960912cf75634528aa943eeee5e45096d1d658678&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637775313Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/array_metadatas, Name: checkpoints/5952/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5952/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=2fcc25254b266d767ffc4f243b5e90ab78343b9a4c5b0601e7bc2979cae6b4b547d88885bcf79da6cf8c2a84c7711bc648864c54d9af2f290192437d8fb3d7e1c4b911cbf5e2940d99686db28f5a7b382f3f637f7a871305e759641d2a79824d4bddecd7018f5a74033ea388c479bce9e4a44044cc7e2d80362e61abf12207d4dc7c747f7c61817e0c0c081864ab35e89308de160d089132e21f7cc594511428029bab7900d86592971a2607b4451a3cef857da704b65014eb33a63acde9b560d26de268e71f7e27b0532c731023526c3fcd2cc67b0b86b29c2f01149aff9fe8b193629cb1bb82fb9d92571cabff493d48efbcc5ceda7bd3d460f89a955d45e2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.637971812Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896, Name: checkpoints/896, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/896?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=160f92bd62a126192d60199e2c6d4e428b82e97567e9640cc798d7994cc4898d248fd6eb42b747a171c0a49586a27c999601b67fd2a07862840d25e7d49869b2860f9c0cc06d876a291c33f9370727e34874401ee4465eb2025611431a3fd824f802f6cd88fb80a62853cb27881c487e0eaa4499c03c361ba44720614b568f7a05a140307bd8e4d60a7e8b715631aea25de58dc363a07e272b8c6b4dc5be91563c90bfa5197aaa1e5bce1e9abbdd4e8b9b28550f6fcce8a32d679953c522d937f5b92ba196c959eb23e3d5a9131723c6f18586e4a0703467720815311594edd600dd6e120a9808f2e10149c599c1b3371c2302ae75ec72faba23c3853f5c0e27&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.63802225Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints, Name: checkpoints, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b7cd17f0ffa5a74ee60e8404a9098c7984a5d73179964486637ca33de3e4c258bf4c6a3322122239f2723f5607c134cb3424a81e74467b0789c4071717f130e62fb3859ac071b657877db10c7550467a3c591c8d68f470229256c4f9b9bcff61bc990d5757a411aad89244c74a6e3acef9c110be6dc30c6234ee41d62de81bdbe4c0743365e468644b21c084607a1f9b43916efb8cb84d4b7183cb435e06beced4e66ba3eee4470261f73bccb7321fdc87d7703148feff1332355a770df0047183fb9cf0d346792c11d353cf63c064cdfcd73d2d61b7487d48447e5c4ee616e4fa912eaf2261333c114648f58d4862da7b162cb311bea64900cd09329fdfbd3d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638060226Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default/d, Name: checkpoints/2432/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2432/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b282d98697749d4657ceea51acae9a6e05e2a58fb34e2270ca69c6704cccc16c26a63144e06a6e334c6d6395511efa50b838e9d3cc9fb18e0dd07db20b60a529227df95ead305828013577d39e19915d59c42f1c4601fc34f2142d74755fcb348792636e6716198122db5dea148cf21f332afc7ecdc9fd713fe3c56f86a6c5898756c116773390618266700a39eedf2eb698132805d839582f1224f333c4d91fb93ea780ab5ff8d574e121121fbd98f06040b0bf63676d285fb95342c8d3f419dbc6bc15b40fd1827b3eb355f03a14711ad274451f4f8f60d291081fa00be957d403dd40267178a88d3b6406affc4643bee81c44458b4d097f6ce3ac100bd2eb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638094802Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200, Name: checkpoints/3200, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3200?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=70f7671fe66270a318b9d235130583e0b151d1d6f2b233419fa40d1a30a3699d49210ae50d99ae9a55dbe33cd1439d2a240a2baf872f4a7fbb4440c4942391e9b3ba800aa404d3545cb1380482354f829199f14354c15b8e07b5583d7c7a99a34b5b3c035de2c7ab94cd5bddcac52cb84efb426cd6a862310b3b207a30a170547cd442fbb4665459a6c98238ba922a127ac561282ef12244457e4ff508f8df5f69a095b37c2d79f74bf17f533708b16865a1946587f3514d73594c4d4aaee5a3d64266b8c567b9eb74f48a19fb010ea675c5f114149e1df6371db6cd972db0cf14cc727bbe36750359bb17bbe70cbf450530fc7afce78a0cd7fe63408edc8fd3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638055652Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/d, Name: checkpoints/1152/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1152/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b7caade0544e076d118417c54fb0d55558908c9408682a6c5310d94476bc55fa717f97df118e362067512fcd60d63d9c85ba744d707e07da26a0a9ab2d27c07887a3907674a5da4b03878fb49e04dc7242a03698590d2f70c37d3df188abe0b0b09303cc7bb477ca69a370c391707af1c6e8f45e2d43900f721c0b8d52760f05378508603449d009d51b96f255a6931bfe1fb0f57e2c128b04e39208d91c5b28dac09ab49ee4468a43756fec94e98364010c304123864cbf486d6bca1903a51ea69c929832b6fbbe2101bf6cf0902e8b2f8cf486bcaa9840df6e14043eafe77706766250c823056e90656934db0b4b9e0580567c492ce7623106b46598b0b0fe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638155197Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/d, Name: checkpoints/5312/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5312/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=37d577ecff62f1395d029f9bf517cdf4f6ff6433616b9e08e373368662b046fc36869b4761b848561a16ae4de27140e34e697b7c8e69b91ce71e9456f58d40f6c33a9d7fb1d77aec8472226daa2d4946a85af0183a9ac9f2cae2119275ee4b7e6c0e839752aa6c2b626d11eeb5257772969dbc6b890eef76df08745d880fbc2b5a48ddfb840bc3e0b3e59792685a2484fbb5615df1c633e78d56780d9757d464115f5a273e8029679c9ef167541ac76b99072f0e0637628a4c07372f5ca7acb4e762cc78de1d600613fc2fd64ca630234d060f04760000509ee23ca1e9ef8004ab88f11529c671e4fa15ee6a1cef4be22eaa7e88bcc9cfb884859f24988d1ab4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638167997Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1408/default/array_metadatas, Name: checkpoints/1408/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1408/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b8a6f64fc0f776a77b5ab4a3a1a41fd6485d7d0ffc8a26854ef7800876c58cbe8323c6e1785999672a1895e7685cbfe2e6268380b3a682e9941a781ed6f86c9f38e2c6f8a4020b43dd4f3e9c366a4c4046a231e05da5970b5f47f7f60545c6b7769094015db8f39240776f513c9820733a5f4f680a82291ff843254527a006fe9df39c6ea3772f41dbc8395fb6ae7a9a20dfbe2d0b98b843daad2ff22d99e6b135e8e41ff74803ff6172171b683ff28b1089fee6be0ec026ba97516788600a0465b45f07dac8516271006b25f71b6993f90e43797490e23e190c0b47cda705fc61390b05f5ade59efeb9d40957c44bb16c20f023c619a477524a983e87f0b604&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638235943Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/d, Name: checkpoints/4352/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4352/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b8223fa906078c95986f613860fd6c84f66f752bd47848e7a62104eec82d73010464028611ebda8a8d279025e0858ab6bb35894ff18165ab282f812e3b88779f00944d6cdaff77438702ecfacd867fcc70679d410974ed9d58a7fda4726a4214e67c9391b94b74a7f6c5940a5a20c3b13fe284640cf0b5e74fc01afcd6f06e58e6ff16fdbb989ad7fce2da6f7bd48a0910c1fbb4caa5954ed5cdceda4780818e919bf590342c04dc4365432c5c86464759c4c71a53806c92e78af4dc1f624e4117ee0ad5695da2cb26fd2c5fdd8eda90319f564ad04824c21cc5662041d89e7d405d3d428bc3177201940279fe48770ad2a009588e72471eada75a4e3320e4f0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.63830657Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d, Name: checkpoints/1216/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1216/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=459a28baf4a04ce9335e6b0f85dc9a8f04c0e141e94f584f7fba6e6269914e8d59b879fd82b594b005c2dcb834a7b7174a5c9206414f8853944c8c3aa7635316b4dea2df6d600bf605892eb1ab804e0aa813eb6f53853bd82dd9c4e040d1a181dd3abf72620d01eb42c4e3bfa0f3afdda2110e2dd71e31654f29796231800e73f998d1619e07641006a0494fbc2d974f93b51c5e09778729c82b21d41d2d9b240efc79ae1d1d645be8c1f224eed20911db6d4186939c9d216a0985bc81208fad5dec616d59c05fa9ae5d328a554307622cf9d48f7c65d55f7a33a6a8289e8deb9bf68fea8d848991e412a0beabf57ab03d9b98d178b505ac7c2ec52c017d7e02&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638383696Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/array_metadatas, Name: checkpoints/2880/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2880/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=8a62431c7984c0b9da9920d88555670528896063b20cb1d5c15a0a8f06eaef3498098620b723ebad9d827eecdbcb08ebd2051e32fc6925506f98dba69150fa09015ec528b9c6146b3d2c3d9ecc54002a336c8dec847a56c5bc919329ad7f151d5ce94eb91c00c5200e8059c9763c14017c034d227183fa642a8e07bb6e680b5681004f06ac1fb7f5b3451f39554a0ff4d2b5a848ae2beda57c80d0444d941f08effbafd34e3c9add910d5ef0a7f76b6c9cb5b6f6ab24d592eef83f89962a9508140c7c03f000055c9052d83f0196adb0006db652235a0ef865a65694ae2cba2bffd32be6dc17d08cf3ec55d10286c37a9619fdceb1c99b250a6adddab971b12c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638402626Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default, Name: checkpoints/4160/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4160/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=24ce015578f061371eeffcc604c8c54f25477e75c6217ea434708fbd09d1be3e9c712cf95fb776e269e2bb9615d25f43844b7418ad22b50e5dfab6d95ecc06c47f0b3275baa6c635b6b29e6e5bddce1b20d8676c8c3cda8e64fc3fe4b90d8008457dcbcd621b7d36b432df90b6c0b542cbfaf0213f108d19741db5b3f35acaf2b83a5f6afcf112bfbf05a818291ec856bb3990a0b03426a485685a62cc2804e139ba0d0426b49a8d8d0ba67a88f3929b2b8b63ae2ab1b09230a456c9871bb6b96d9ff0e82449fee80a0c4b161cba3ed0987e295ad5117579cb1d4d9fe430a69a5537ce5ff612335b1dc103d0adb78b7559cbb40c981015e432a8377a0362ffd3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634201971Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016, Name: checkpoints/6016, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/6016?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6e29be9784f2ad6f6f5cf8069adeb9b43b5c97a5dcdda393da81c1bdfa7c432e3d4c03655ff021e4d6ec9d0160e68bcd8e6fbabc70abb981f807e26c9df482596586c6f7e64c9263db2569302659cb6a16d8393f630e4a18bc8dbdf3156214134356f9593c1b0870301164e7027bacb2806e616cf5f9aa0301c546b03e2264dc40e4ffc6f7b2e02e987025c4e460aa6f5376855aa8b6e54745ddea70399cfb5fc78b434dd92a0640e6616de0a23f9c6a809124fa1e14aff8f76bf5bb80ffbd340f303e20002cd57b08a6163d61b54aca1f8e0d7254f977e2f3262fd49f6d24e22986c683ac6897d2dcbe1f75ac2738aab993e927606211a8c9c87608cc72ca92&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638448783Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d, Name: checkpoints/1920/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1920/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=96abde1454654c105ce4a665e76933cc98b40a1e14627cfe40f81d09bfbac2502f39e156f1d2a2fa32361743e05ab7f474bcbc8e4a73af1510c67dc6776543e3f576db5032463187bef67b9c55666cd7c0df8de04834338017e476c4ced3f51b74782f05ec64dc7519ecc9c4d4f2e0924e30acea289110b98e79812e1a3bd1851fbd835f1c4881668b9cd544c4e4475c4ca7870b8bd257d7903aaa5aba3799a5981b1fecfb10d0f4685facd1d7667e20fad9e5f73f372d65c79e675b0db727e25eb926b17567b600619fd2ef7f5d87b48dbad6ea6fd7265ac5790331414b9363b99a09135f62bfd1476a68a4468e5f28bf5a6fd935d64670a26e35bfd61f0b3f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638468312Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368, Name: checkpoints/2368, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2368?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=73e415441a2727018e274f33e6cf32ab6f993ab93197b1ebaf4a321bd5e25d10b1e2eba11d48bce2270de9d45d9b57892e3a861818328dcf8affe019fbb11ba8f0c8825d9d9041e7bb3e122b42149f4e426d1ed3ec37b6815a21e7af492e96a66c18798c639c4a2c94944dd34f575dd37dad211eb95cf04f965baf7812fd2fdb0445d254103d824fce5db989c64b04b3f2caed05d25466a7f68430caebe3c75f73f3649fac414781d66e789245938d57a750bf4c3d731f5e646d1dbe547bcc07a27daa89a99c13ef2fa4335eb6784e86630af7c20253be389f3d4a6539591ae9480b57f7dfe3540fcbd83b17dd5efc9c1e9160b5e0a5e5ec4e4ca0d28f9cea4e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634224466Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/array_metadatas, Name: checkpoints/3968/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3968/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=555c90b5f652e6db66aeeadd27d2666b4117ff826f226fe6c1440d96cbf3068512902d57b53edb099112fe6cf98ea8d49c87ee5f1931a0b26c5e642a813ef2128b5a54baa00fa4e120439e954e5506f944258cd3076c3cf07a562ae6e9aecab07dd3222843cde2813909d3857590ef657839df39a41bc541b3638a52b90d75edc15a0202460ed00e64539dd2ebabba27bb6ade346532c34483ac155d0ce1539da7d451a4d57a23205ade6711cfde01b389916c6d692c7b487444bfebf162e53a37966aabcfb1ffa1b5517e381c12c8ce3167f583243ebdcec507305cc596873f0f874c93e988fca3a74ff274e5ffcb8913b7df241d6bf573f713bececa904dfe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638531157Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d, Name: checkpoints/4352/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4352/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0ba342a799574ee03e56f4b807ff8ac8766da11a321746efc5403f7dbe0f17e63b34e488ffff3c7924a5cdd3bc61ac6894e08118760a56dcabc2a7b08ace801675dc069300635dc9d364f55142fbd25b37248f5adb534b2f1d5df3d3c94e123a8502ce8f9eac77ca8818d667b921899883a89d8f06bf3d3d06ae0981b6b8952c78d61b285cff3ed7140e7cc87a1a402c6b5a8c2f79cca59f3a35204e6b9b175b9af6208112f5f87dea8d48ee4af2e0bd44ffde57099e8c0a5281235143c6afc8a1e791c7f3498cb240829d0cca6f4bd05b76404115cf08e5716ec0cb838ffdf3cf4037bb326f39ab88aced9058594a142ca1f1b5a3de523eb3ba9813df1a591d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638615667Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4416/default, Name: checkpoints/4416/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4416/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=1706a6e4170001b6be285ed4a10a6af6b156b6071392a4d4b995015d4a1e4ebf664c970a9d8f010bf3bda8d508073990cf589bfbb56a3f8a07ab8c08833fb8e7c00502517d6818a6d414b4f764abba3f63aa29189291cb5fc538f864330e277c3049747cefb0d9d87c1f8096001a6088f4c12b7d748ef4388473cca3d2280de6c344210031ac12971f50ed40dc7b8ebf8bed6d9f8abb2378f3132da02344f6bb946a92b133ce89ddf8e3c7510902f3a7b548dcba7cd097da1558b338568feb7898b69306639d017a904338a60cd7d9bf52bc95c926c9add94e5b6f7de1019f6daeaefbe0c17d4241d86ff75b80a5674fa938701318f9008ca26ccf460ce2716a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638595656Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default, Name: checkpoints/5952/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5952/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9794da8f6903b86495d365769209590c9c67c516137857492d0bf0e25eff4f39442cd8ef286d7a48effc53a37facf86cd546a6625da4fc29da474c0f676777ae69d7c454128491d03f326657443964eedd940280e798273b8db6ff757020235f18fd4456966a3d93b45275caadad72fb0e14996f333264c965fe959c3072ab9a6995ab8d1f97692d0bbc53e16f0f8241fb8e9f6454d24767774e6d9067a097a7f12b8c1f3625c8a5b30fbb1e118cc7044d41594256a77cbfb5a85943ec5aa07bc1d244c1159eee40b284bde41745fda0fcf30a72529b5a1e77870957527569aa2f26b888d638c94f615e8f35a2c7a683adb29155e485e6e269b06fa698a590e8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.63859276Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/array_metadatas, Name: checkpoints/4160/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4160/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=17fa8424f0cc6704647c58399918e99e6321f85976640eca4517bf6d6c665bff3e167443db614036468c5643acdb7022267442538096f5c24441116361714725886ee7685253322db1934309efe2e27a9c41d834e00a7333085d1eabade641c1734e17668ed7637532c968016365ce18196b75f02f6a2bb388be0799f1fc4cf99dca9aa6356ef57f286efc5e650c728eb1e2283f2c7b5ed4e9ff3619a78d9b39e9fd05c84fdaf18256974a2ed6895c142c608fbaae4e920a37a18bdfd5532bc2445d651bf9aca9319bbf2f4089aeeda0070647dbd0a8ba7971fe9c24862d25be3751d347d3e4a9c67925dec5e248bbd2887b46c090eeeda3f17beb7d11ede096&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638851026Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d, Name: checkpoints/960/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/960/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=55ed02834831b0c25c295ffb656ffbad5de1c88abf5365417f2c2b6572ae87c1126ed22b93b98fd794faa8dc46ea38722dafcd038e6c0825a7ec5df1e67c138e13139aa118aa2a839de887a6a9dcee88934fb3ae539a358749e8b7cf0652a6f46872d127349a960e74392f0df44c1d20cc536c8f73a8337a5ec9a6037a0c89f9040c6bb1511865a60b27a3731789af3de08160e4ce8415b3e763e0c4a2139c3068826be78d768bf16e381fdd1dc7be194dbc06202bf8a7dd597cc8c6d56df63c25dec4113660fb2ce9944cb2e23d443a0dead2c586a59041db3b84cc170f4ab5299936d7d5f66cc3f9cb46dd64c667f2809b9186d67cb8f9c079de2813af4630&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638925539Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/d, Name: checkpoints/5568/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5568/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b4d3d13434e34a33487b68d95edf8eafec9661a16444b1f2de3a3b8a0c95b5fe164c4fa4bec361b07a9e9de6d8e4e2515e689d864394b54c64bac0f24dbbb4ab858c7f8c613302cc65faa08605b18c78e0ba331f623679ddc35a865a115470568fb2f60ce1ea78b3f1b8afeb675b44f6867b80d19cfdd7314c18030980c2cb824abfa5f15c111b3c50e4aed0e41c1ba6b94feacd033152fe77ac4542660cf6230e9a767971e4114d898decd7b37221bc2ab4f4b413c78db6941b5466e70f7183799f167b7d19f3e5fccb5e3ed17e773c9579788805eb22f7cf8d4cd67885053531801c6a491907f88d86775ab4f29beba550b32d23ec1248ff3be61c98aa213d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.638932839Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d, Name: checkpoints/1984/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1984/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7694f99a5ce461f5053ba8f8cfcbafa759a00623e2d01c725f40d0ac0dc261a7282fe9d0c455156c95b022da0928cb2f84ece36c5493fe525353eef6ce2d6451802d101ec4d5da703e82e0eba0b3058fd554aa9974eac4bf63f311a0b73844fce4f8ab970e4a2047000ae71708b97d10a8458d055a3c5dc5e07ac3b0e1f751d9db088416d258bed2d4fc63cc4d1428f7c96f8fe7c45fda77c85e35dfd18c1ebb228a8630230650225b69af3399a73525fdd03c92b99747036ab129decf48d353866eb0effc83e230bcd1ed916746eacc9e6f0bdba2353d5d9ee5cc76ac9cf73d50903f7cef0c5b0b3ea6baff9668b2e94c922f05715cc0b63751d45f5f90c823&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.639095995Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d, Name: checkpoints/1088/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1088/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=57e8b460eb29ff43026bad00c67df3c1f1db21e815dd3d2258d2959dbfddd057a13838b31dae46a3e3992174ba5270995013189510c751c63d773af82955defb4de276e36c461679b7c814dca706ac6a49327a3e86fc1dc82b1e37d7887d8c371e8e9a0c31b071b1527760e449c02a4f6804a0273a63ba7741b421e9341efb1e04101e11ee48bd693121500f48d59cbccc4efe482d846ecd079320bfed9058f553f0490897b4bf0032e2b6e835ef2fcccfdd93d7099e45a6f77c228ccdb6e13a1c50fd6992d935e4bf71809cbc6cadedd64afc22abed76e8258feaff796e46d7acfeb764c8863430d493bdf50514a05d11addeba0235509ed564562bbd9b1a70&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.639137121Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d, Name: checkpoints/5312/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5312/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=87687e00510144339bb0da423965c2dd6b12c08437fe38b4cb3517188cd4cea5dfda21fb806a497c4501e3d807802e11024f5d226b4316dc315c297a64bef41e7af21eda1c9645335e8d10c0ee89272762d9c456fe77bf3a63bd5d5928800edcb549f252beb6ae56efbca6846d1266909331d587d3ca0c9d28a72d0d6889cb85cc98695e3d2e48d2c850f2f9a0592daad1f43001f06fa96e9dd45041371d6f47b4501d96d9aff11b7bc8cda7af6dcc4fa4ff4908d0c1c01ac2d31a55c6f35ad2e85651a1725078bee93a3d14996fc2475c4a689595fd8c18b9be4fb854dc6c7cd5664cedec24f89a9ed8cd41f88213c1ffa8bb3ef03d467a01f192e06b6d4cea&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.63914836Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0, Name: checkpoints/4096/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4096/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=703554c0a28b1382789854f7b73d59ca18d1d0d6bfd7727ac388e421689d392bf9cfacc4619f333aeeaac3cd888b896823ed87deb9facf250cf8da8d11ead83e625430dddbb2100e515bb32dcfd7969c0fde85d96bc08f9a5dbc7d3e26ad7a6631352bd6783876d3f5f78c23e6b1c14ac94bcf59e9936d58c6d0b591b80294fbdb49d751cd004b33ab9355d0ba818333da1c82da79255d82a84b12e7bc72d0a0c299c06d25070fce610bde15d15d97ba1db628ab4c417771c6c5c687155863220ae0c2a65e243b8437277381e15170cb4aa93f7cd627748d664f5668e8949345efe079f4e74fd87a77e9a6e0c92f7d2e5dcf6c956343d3bff5fa7a1749827b98&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.639186516Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/array_metadatas, Name: checkpoints/2624/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2624/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5af019aac435d839148c75f2f882dfe7c7444f9ce7a007acce046228dd58b5c5e9bf3b475c018955265da87364f5026377d317d6186e74069749f37157c3172e347cdbf04ce629c184c1344d6bcbaa0d00bb9f4cf0976940274a78faf49e03ffcdcb9c399a5fe8aba68cd972a78866cc2ce1f9484bba22d2a6c9f4ed8dc20c0f117ebb1f532714ea1b1bf4f5a4faaa44f66fa60a14013688fe303f733a6809240110ea6f42a6d4276b1bcd4d0be0298e0feb8d851e148add4a51be8d5485f6d9131be375ea2fe4139633a1d59ca6a8be61891157a9f5338e8672e38d8b660a3502a02539b4ee2327e8756301501bec709a8860e9e429ea5ce6017000041e3954&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.639134181Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d, Name: checkpoints/1152/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1152/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=34c83ffc109353707e47189ec3b46412f84dcd733407cd25f09fb0490422cce1cce115bc7ecc268204ea0891f41c4d32be81cd2cd9220d9b07096873fc313413e73ab9e084b3bc95fe3268d3d464b28674630a29159e8a93e97bbc7a7956f8fd054ed50c2ca0cdd2c4907a90b12efc682b426974a8ee62752512476ef10a43992d8ad7f474e9195eb80a8b589895b56f18739a1996b07a9eb9b4bfb9d5836ce20bbef176362ce22a57dd4951c687dc99db6d6d0ff1591d6ac85beeabb5297083917f6aa4ab9ccd2c87165f917ea668f425ff8bdfc720c9411730d65564a35b870dd5f88f3eb5105882428a90331bc9c6ae49a7780855f9673e59fca4f5a95487&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.639187524Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0, Name: checkpoints/2112/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2112/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=730ca0a88e2368d49aee372077276aef2a7dc8a2d9d6a53e1166129465d7aaf535fd51ac6713c598c9c5cea92ca7d11ca391333aa58d0f2af02cd0768c835ca400074b968b91306fa7c25ec333641b22a09a17ce751680d11ddfa4cdba17a5471c7edbd49dc7ab01e0a8bf86b5a0dde94c9cc1a62cbab84d90f037af4927206858fcb842424cb2fc7524b83970f7c7f1127af686567110c2e51ac823698843f6b4dda63b24fbab521a76173022e476d788c0d7fee985948e96a2832616ff297862460f88d00d635b417c36c312cb3c5fad005e3b6851ff4832d260867a1011bd09b1abf6fc8fdec12de95c01599b20e2c01ae6806658f4d75ce041faf776cc46&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.639209532Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440, Name: checkpoints/5440, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5440?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9b3b2f832f317c875d7eba4c1a201bb1433a8541a00a4fb3b163e1e393bbb95b31e5c2b05b567647c4f3328f43d7605ba658acfe326ac042bf5f349918d807c0026199f181043c01b4f5b5773236ad490e0761ef5502729364e0bd1d0dcafdb8ef76997074366f250915c546612256e5be2ee13e497679c68f7eff006cbc35ce9487e182ac55c821792d8c38b235e790814eb078e652925422235dd2c89d65af508a98b1c3840fd5d5537a2810491bdd4775b0d70eb4df393c384c4ed491696d699c59594352115e7c534a720e9ec9d89967c9d61a589658682b77dbf19de56ad926f7e7cc6cab1e646e7a585426226b0dac4e3c539c4905a17b41ea85b7678f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.63921746Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5440/default, Name: checkpoints/5440/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5440/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=32878440e393a0550956221c504dc7e17d47b4f478e2d5dba27cccf5fde645f2ca198ad8ebd948d8a563ab1bb5bb8497936560bf05798d5a5dd5da5224aaeed1e84149d052127e63b2bd5c751d8e32a4d9e463d6adce50f840e10fd933c5d46e966a12c330c4dcf88a7cc86b374b133cd02680aeb0926ba63e5f1d2288cb6408c6b7d8337becacd127cbbacc85dbc579841e26ae989ef33e6d9bfff3b2dfbe437fc137e8aa2ee9dd3f48fa02686ec8ce13851d65ea05a5c7721a177ccb7d778c22069844e84e7d11b39d9e34b5b30cb0699d6e386905ebfb156f8306c5401d9bb1e9f02c103297e1e4e2a15019d29564cd901f7aa16af405d08ba12feeeb7a76&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.639221814Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default, Name: checkpoints/3456/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3456/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0a18ce42f8b6fe13f3f99b2f3fb6e2f4306a24490e8d5654cc5361db0da73d145ff326ef7b729f58ac02e785712cbf94aa42c5e8faa61e5ca708b0e17d9c4ea16047ab02ff2b6a191069b4aa055b70db0836ce0a51e618a4130869eb25fdb67c2062e11d0e81deafc78abe848a48aeb57a3c87a350d91561fae45f505fcbf1c553ca3ecb864bb4e999a71f0371e8415144d533a1c087d5bab03d22851a3e3557f69332f9f6958142893ae637bf7434d3b56c24521a913e00d9c13357b7084b0ffa2f4112981dfc0f3cc1bc0546768053e8e342a2cd79def8f9bedfa1ba569e19c1eef2a18b1438e3aa932c676120e72e0019c53c69b2e1d2a666c865f7c66801&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.639214574Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/array_metadatas, Name: checkpoints/6016/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/6016/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=92f47766ab119168aacd9472d56c87d4fc3402c885d4086664224dca3997eb530cb4436e4a73556e4d24fd5486e12ea63fa900c0d749c366ccd6d92486746e2fe947301b5583910989b62cd74eb71bf71dc44d8a4311276a4687a0c5b0e4e8668984c7b3df9bbd677f8cbaac5a46e28db4cd1b265c0812529c5cea0aafa796d544ed51eddbd99e0c5f5c5a4a988a718f9f8a077069a8010e7c4f4e2fd6119637ce2d5445ddf82f70f36c88c2819b376c15a0450db6237224b972418fd40ed2731ec9ebddef8bb8c7c1968b42773f64a50bfe021d739b9f3c3006c958443c6eb8d73ceafcf55aff4b54d0d31d24adf744fc7ddbac04de0afbc23163c899e4d215&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634512362Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0, Name: checkpoints/3456/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3456/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=03e6e4cfb20472ca165c6b485559836ad73b31a99f2e2d71356666029f9a13630de276bac92204f454257de2aac1f31ed60e2f61b18ab9d21282c189a1d99bc1a02d6177b0fc65c6833fa39ede61816803e7104c132e4a444a917874703a312bf8fda918e6c2ae03dd9b4cdbd8995928863f50da70f8f733eeb68cb5e1fb9cf681a49a6f175f4c622456d57ce72c8f7165e7bf42737ebba3e88f5e8670d73c851995f2efc700a9b67e69847cf7cf59a1efef0532a027ccb2baacdb88fe418d1077db63ce0a9d013adb5e1d6595eaa9db054795910c9907de9f8f5db0a22057a494a8ad90e63dad35cbed3a3dfdbdf3a005cc9f153f3415dd3be06c5f4aecdb13&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.639293684Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456, Name: checkpoints/3456, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3456?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=81e64decdc9e801c0ff87a9aa6309989a0218a31c513a1da32dc7e6716fb228eec99ebee0e985d288d780e3c8037e0d6cb24cb6cf12835842d776f270da91b01b0222da3fb09e89c45c800942cf7bc211b0a60bffaff1bb0b8e1837c5a4d0dfff45630da1012061e90ae07afab9b10ad34c5a9d9a572cf986d2172eb4c4aa2162b837ce3056d1bd5077d0b376969ea891fa2a7df386d5f340e734dc70c32ff8e8a81f07a2ec27e2e815f57ac3656380233328a63d79f380e807511f9c7c873e909d5df1254732601c215dce4e198ddba6c0dc54e5e827a3360dda197fa472912391354c7585d6544d750d493c443482673c783eaa0ae6b53efa9afed17d158c5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.639257923Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d, Name: checkpoints/512/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/512/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=48051f693e6b93a88be028a2cea7fb9aeb38c07fb41ac95290ab8e72859646350e64c68595505c50ac1e955923dd5883fb6b9094ff09a326257c85b57f414cdf800ab778879bf49f52b4fde5a24ed462a68a80d6373542323f5dad9a23413abd05e7e66e4b27010f7e3bd3a3f6e70b1678721a9726f12c8759a43f24446e8481474255b20dfee0a870a4e667baec1a8985aff51f323296d197e6b0f8d484cca6a141ba940157a86946c4e047a2d3a3395d00489a8e4aad06ff5c2b7ed6c938fd66dddfd7124f45fb55669673cf945f92f1e114ea572ad93408804992f65d3007db94f14d4efb70afba768f443d88fe90ad5c5192c2f7dc72e13ea17844b414df&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.640933327Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448/default/d, Name: checkpoints/448/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/448/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=c01fb42ae261c38ebb38479472d2a7542fdee26ac81ae08b32eda2303bb85cd7161bce020d0f0f3820485508617a56d99561d930832963f9794af53797894367a3a7da01f9bcc40fcd62f516eacb48e71145bf0aa5ed6fa76cda40a6aa9bdf6b081bb88522b7ebe0f507ca6ac2896f2f8aecd7217c9d3a252a583e14eb4e031d712b567183b50343c4d0f91656acc0db5102bc1938bab05088f3053a59006e0e25400f82cb93a950e7d888b46f17dcaa91bb34e76e24d774cb0788cf363b3b62da6e6090f37d01acee4a911f61dc5a0d71aa328033a54d05e227a61811c44d7728d452d64e0a2e52399c3fd2ad0b62ad51e04223450448fe63a09d482a1c908c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.640944654Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0, Name: checkpoints/2752/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2752/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=1b08cbec9e68338a530dec5558db36ce3cf0ad2bcd96428a324c62d85329d9e4c6ac4fc489261c57c3b24822bca771661d7bc3f1eb2020f0c1a32fed5f50df982bca3cfdf7d7292596f4cdfe8d35f6b37f7ca8a0ac0eed112221e0653ef43c6813cfc36b39c602c5758dcbd7126b7e0ee856d56c8b40f56cc68b1b9ec4d601e26101c61783f692f0c4ae0501a27f21375ebe4cc6a6bc23723b2c01ffc7836496cac8bad254f9b9ac2b921e84a3070f70e593f5e4d3baa87664a11c63ddb2fb2199feb3673b91925fca9a75007abc952aac65363039b233b272499311b6997c83c5af7e135316746d9288f3d34ec72e28c8b4ed16683379b38b2b175db2d0ae1f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641005058Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496, Name: checkpoints/2496, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2496?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a5b83f90b724b60fd0442d1d8c961e509d2c7a991918845c306310f0369459c321fd173bbb159b1975ee49454ab8f1962a7998bc028ba3c5d09f9348db112e05c52f0c5dfb4195d470d7e1c8bc659397ca4f7c548ca72ee6596279ca5121f488821c5616d3bb4e4b908859f5f46fde3d057ceb39f0fd951576343ae747a4f372e62c52a0e0abbfd49e6d854de9669475f24912e8ef294a421e5cd8463a411bba6a2aec4f03745cc9d34b90a9ee9b6aa3e2ded35f21a750fa339edbda8b3b38f83c73939b2f9feb41ce3214e7f58f55fc5348a08e9f8b2b2e187761a7d72076b3619242a6833670efd9c645655f0275811acff8a81adad48c689e2c81695ed4d5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641068984Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/array_metadatas, Name: checkpoints/960/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/960/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5114989149da63a1079525f0758ff092bfe4b8b743588a48a4f66b4e8c86b8388188360549a9cd85b5fdf229cc3ad14ec20523d94d963196f6cf014727f0e486845d426ea84d0f84079c3531947bb63f7384cd54fc5d2c13b9b56b7e39c253196cc9d9a55f995b76c38a85c343b0b7d55d90dff14fca02234f3d8a14f2698aff0c6fbc5aced08ea1e893d341af48cf03d203325e2005834605f5cd5ac6599b55b2d23aa7a12bec565e72d1665f3cce331f0ecbaef09614eb3529a95e80beb91bd3b18425294da034f021635294675cff69c1f5f79ee7760f505189c5f02b6e096492572040c4d0c0bdf5bb036a8cd1f8503d78899f621d703ab37887a6c9858d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641068927Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/array_metadatas, Name: checkpoints/2176/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2176/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=1279f4365cac4f7a840ea56af7eb87e4cd34845833c5b0e313ec2dc49c0af8187721d7d68078c8694d2f318eb108191212e7ecbb1e7fa20a96875049c5d53cd081137a4c91d71c306e298163c8cae1a4a68e6683d876e68c40d7425a1d5876a3c1f933903c63881382dd6b9fe0252817b4fe393da8b818cde2032a963c1a077e68559d3adb47404e310376143690ac9887bed78903ed458149df598b59d08031bddaa47c20680d5e376ff1c330722321c79b0fdac0ef99ef431991077f3b479f9eb9ff22cf3efa55ad883b15299b01d2c1afe5b01e878f4cba28eded1a4a755e1548621bb773b289a2fdbc1ca80e3ea77fc1b1933270ed10344a19de566312ec&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641091021Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0, Name: checkpoints/64/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/64/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3f2575bdcb7440f87f42b2709daa66c0a4dbc45bf993fc0328cfa3ed3c0bd190c779fe980f1fc8f9a014372889c6e978be94807d8ff7217359b5b4250c6503ab2136d76f341dcadb0c21a931bb9df3d769addb30c21f4129a491474d8bcdfda0c9c974b4b947e1563ba721449f65dfbe8727aefeb7744a8c079aa5041c28db8b70716edd6cc77888f16bf2ee5eaa2c266d9c3d8852392bc572407ab9933cc1a920b6247a78033d325154c9ac518d4696d6607286eacf9a74e3915d04741627a4ee43751ce2637098dd54d4c2c2814dce05ca832ffde124235d7caafb6ea419710f34f19d357a3c523f4068a829b05291ecc4e9e9e4a7d1baa907b242be69ec3f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641128976Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default, Name: checkpoints/3840/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3840/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=731a73e97d6b87e2f05fc37a658af2c3d06f948aaf774e33c869a69334ae9f0a946a1b78011378965170707ad39b0429114577baacfebe9113fb4801180a08fc6c4b641e427354f3ab8f22deffe3d10bc365d841a3d4760c9797d1d4cb9ac2ca0f2f844c3fa2acdfac41f898b80afe65ccf6436fb3e4344e12b5c25660e1d7e70868bd63240c206acd7440ee60767bff0ca64572294ab9669d1598a2f8dec2316db4d8ce40bcb0d6875d2cead899fbde8d583ae648879b98f7cad8e03a5bf4a3ed7219a11b59e62fb0f79962471bef08ce74daf54fd60a31b3b27d64c9d8dfdd3d3681e3ef9a9512cb540a9e181286ca85f87a6742060d94fcc8c61d91712829&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641131244Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0, Name: checkpoints/4928/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4928/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=8c935dce97bb66d3c1cf5dcf856d7353b180f1d66eed7438e7084b4de22ad30e560d549da9638657babdf2473b176a6848fae6ad30308f6d6ab04f0c20522872cb85b53307a95d87ece4369c17b11d8739fdb5a011f514263fed6c146f2c4c6bfa69ae8eae70b0c1e60bfea141f63228343228850b7e113569a99466dfd975c5a1251d5243be642d311fcab771d93cbc81cb4c2ec4013242b88bca3c5547bc735a1a9929e777e964a014504793dc21b7fe06f91b3eff70a005bf9d550aa082e10d50caefefe11d2d2c0cad4108651aed3642b93688011abbed622d9a72eebfe7650307c6e19f78a973d54f86e1cba3f073b238c8198904d1242d194bc2d81ae3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641146658Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d, Name: checkpoints/832/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/832/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b6f184084ebb84dd88b92cba7ff7e267e963b3c58c60e709328337fbaf4cf1c9a30477d1b540a53e9f7b5838eb0dcd0dabdf865a282846cd8615a5915fd200b3e88dd6fb655755268031cfe549dc4d2eb55a8b416aa646c3d4037981763243feb632ad2de819aa57754abe6bbf55cf1f8d8e9b9558a5d9708a7eec21fb16bae9f3e75c5e9e655c9ca73116d0d3442ec8a6e245159e2b569e738942a94bff4f3bb0e681a98bccc0b634f4255549875241ec2edae570996052bf1177f45a51a3e87377007957190f3b169273befd72f8c738bc6979c44d2bd244efa6f797fa91d8f3a25fd154471922ec0dc674e4bd433784c2d453500dcc4b97fac1c3e782de17&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.634730138Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d, Name: checkpoints/128/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/128/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=2926ec3f12b320df17f64cc09ab74ac698769c5990b5d0b3284e05618a789344935f6d127a5e445f025285850f1b94e69bc13a252ed8c143f16bb9423b9daa7694b483e0a7411216e0ff00221b9694cd73d4e209c0cf1debe8e4dea25d70778254aff4c6c3e4ce9a99414e1256395877e18d4c0e17bc9c194c64644403297d0658763a3fb9271266bfe8553d8e5e2d034e5988e6601d0faca601e2ccc0810d536c4ac75c694a922707f8874fde05a52c2f372536a2ce6e425b3675704bb89a908c240c491961f69a2cd2dff2fa2ee354f5ad1a318ddeba9cfa66eaa33715542493800165e366a397cd2f0431119254c07d270b233e139146289275edd27be3ec&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641182429Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/d, Name: checkpoints/2240/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2240/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=81c5f70c8cf710e3e0dbb6f41c51c8afec4c176b3b194ae5c092a2e66718d8e23ebacce7558a58e92f43e58b4c95ed53c982aa7375f989435d80363411f6272ff94047a0afdecdac17cfe707b9febe7148a756e8c74107642fe38b45c3212eb164d67964a99437baa1db68752f527bf5c95da219e6efebf02d26f87bf2451f8e76522fa03bbdc525dd319e5575c90a8325226bb12d304bc044129f75130239e0370ea6c913e3cc3280fca6d64128aeb75a9bfccadcae9f8d63310390e51a53191a23c87a61eebb4fc5ac64dbb81d4b9ddb0760347df1f3abbcf9607e4c1806776af938b4b61afb67f94a1fedc531694ca7576d1b5a175c70bad3435cb1a6f8c4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641188761Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/d, Name: checkpoints/3072/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3072/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=8a72b25c970094b2769095f03e59f895346bfcfcdf10539a1ee7e645125df7c9330990bd7b9f618b592f60db55e553681b243c824422963608708387df3db7f00eadf142492515711c1129501f4f8dd13c7a7ababdb4e0572fcbff25e7b29bb0972cada1d13fade59c41efb47612b6ffcfdd9b2a23f21948543646cdaaa0758d370d3c6b786ea84962002a3c9c83b1810940e82b9a5a5e3f2bb5ee72806b13b0cbd9c45b397fcfe5016eb2ad63f464a53cb1e89eed5d6e8d8d8f055ad4123f4be6b2cf0d73fb48ef82dc5221ea85d7059cc9c6799c7652a6991675fedd06a3ca3389f1407aaec283b341e2d2404de3f19260722f2124679b46a076d349a35f0c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.63472916Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0, Name: checkpoints/192/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/192/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=171ced3336f719736acbbc60f9317d7223c3c61b475a8d080d2e5341ef6b4c4e174de7c96ac9f3dec9532177f566d203ee0161c87814e99d9708c0cdb229c4db133a37cddddc92297d5d0ecd43164cecec2d8a2bf7a24f76553ae8e5cb7245b93459b299a8a280a0ad6771db0d2fcbbf8b8e9c5ff92d0e94132506e315d1e4eac0ab36496d33153ce17a8a592c9cebb1b2d9ace62df79d877d313871bb2016ffbe1d8fb02817ba751d2c32ccfd72f3496653e64b7c1a527c349c5d1b3dc48c554435383d21c6ab70655710df64807fd2b8d9b43a1302cd278d316a676f522945ff8a014f97f3118a4eac4633d9ca418448e66b1504d951a2723da25e8d083b20&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641191042Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/array_metadatas, Name: checkpoints/4672/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4672/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6c75796a1700625181bfafbeec38b14f84b73e6441ed9b86febf4101292c41c968443e88c5d2f80f2e44f17561e118c0274f0a45793a19de7e69a8469042685dfbfef4c701bb0c5b62278ac709504a678bd2309074b6c2cd906705294b3ecd33e8caefe9b0396ea49210b3d8cb0c94e8d3b751de667d444c3f46b403837032f2209e6326d8331eaaf43f29c4dac1a253fc2df3e3b587e490fcd143c23425c5f4afe1dd52b98df14476e909a82e85e25fe51dc53584b4e9e040bfff8c87565c1292e27a3edc3a8aeda2db87eaaae941561c394927a412c3603c05f33e7ace035f1f2cee41390444b1888c34317af727f8a40e4c9266ac9003ffe28623b9162955&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641190912Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/array_metadatas, Name: checkpoints/704/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/704/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=af50215cbe898d12cb267802742de8476264a41282f8c210a74305ce1a63252cfd2c725ea865b094a7d33e94e0509ebf7b7d7169fc1d7d24129558fb16ab69efe590b5840bc3742feee9196fbbe79b5c7eef3bdc07012da687fac1180917fa4595784f02005e042642d0a845e2e0638d9ae0897ebdf6d98fbf26a729f4e5bb1858be34f82a89dc41c04a5febb40a9e13e33dc1a67eeb82d68335d124162e63ac60578828192236582c074beeadddbb63cb78e2ca3f48bc1cc3d4f687a6f68422f3ceca4baee3afc1044012053f21a1105a9ad61f24d02db7fe3d1920bcb6bc71fc46ad10265a47117e6a805ce02e1b0b3ff4c301b82086778ad1f08b24171634&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641202102Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1600, Name: checkpoints/1600, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1600?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=09fbda1781a5050251bc720cc2dc8adc4e5daf830a968cc275a779882d9f5ffea19a7172a2879b4a0555a8b71d2aca49f0b56699cf6662b7b53bbacc4a11e6a494ed350cd6c1e28bff55f73ba5c1039b9aa09dbb44562ab8a8c64a2b4044acbb50638f762bea6b616103c374ebd7fcc3b799b56e6bff9b43b9a6f2056ad0eba6f756ea60e9d66b127a38a99522499a67f4a5ac0ec5bb699eb5cd9a2c35dfd398c03b516e69327115452d8c767619ab0771951d74625f0c78603eb3c5f77e02e8ca5ec1d57e4ad72ff962d4bc6d57eafcd27dd6ede61ebdf5271c47984f7d50cc91d9207aba9f5a14b799929ec4ea47316acf0f2e0aef63cd1aea8e662e4b7d69&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641213445Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/d, Name: checkpoints/1856/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1856/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=02e4cdb3d7852aa62f62f5778f0f41a4ebcd1636adc6082e7df3c8ebe962589f7afd721f4118d2f33bc3c877afabbef8f33cf5e5893443d0b9b8e180dbbd36025d3162d26382835ddfa0f9f9ac0888e779b9fe0384ace4fcfa3a85337fdb55f76a92ed6b4799e050a7113b7e0e71f75b64cacfd80420afd6cf1888293df4127174969584c4738ea50a8c6cb8750f6b1ae904eabdcaf5277d77c67dc7a1ba0901ec4dfaa2e2e174d3b4b0cdc8223a797fd6c7a16e4c1b99750aedb580a6be67619422149c66693ba67acc05a0f17e34bd08e29866bf391bb0b0118bd38782622e0f08908ef07ff95de714e1dd363c38f09c3127b91245d9d365eb09df6db5dfe3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641261418Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0, Name: checkpoints/1088/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1088/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=bf8d3edaf5fd20a40e798d5c1a2f5efca8bf00c18aca187b9bdbf4bb02cefa9ab2942e7f866b81d25ca316750a2c6fc2e687f4dd19e373cbf077856c501ba6d4ef35a56744bff3ada645f8dd7cb45886428ec1041e973e74e7f5678bc1e7d72030d9a4e7cc5a74db31bac63ee0b3e2e93142abdd13e4a7f99d2c689cc26de1ea7f473a9dfb247efdb5d16085e8fe90958bbc79a5cb1e0679eed4ed034a09b9bc32d076a3af8dd970a2778318f45379efded61532b7d3f9d799b8e561b63f64862ba7f7ef0205cefe99154c9f6ea8c2afbb9510a1d6b727ec6fa6d8119bd6d493ae838200764afa30d1044a3276cde1f14baf96c37052003845d618bcf906ddee&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641275606Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d, Name: checkpoints/0/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/0/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=abc8286c1b35fae9e6ec9ab032f327a441eb47bd92195d6bd964b5b661f7fe40e1a365f5074a955e748cf78d6a40d905c8bce10688ade33bb0e5dca6ed9f32240e5b01a469698d40f1fc65f76d8bfbb833b244d1027767f6103f5f7c4096a92374efc5ec30abcbf71642e5dd80f912f2a0f6f49927cc58199fd20c42ee0307e203785e01f2816487cf102c3115331e4a19d6ed984e9e5723b6527421f8b7df7be2bf6866cce67d689cdd0e2e330c36790960fb6d86eda284d883a50ea3ffef1486c3e8be8e0ae8280f134affb0d9825ceae01ce80ee59299b315569e26ba9c40858a74c2b413086016ad927bcbfd4ea5b21740245a90f0fc32aeaa18cc45510e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641267766Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/d, Name: checkpoints/3136/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3136/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=91601018200684febe0d8207cbe8946229aa1ba8308255ca31a4652675eb0c0437f325ad2f1664e74bdc200220a16b1612d3e05bfa778592d3ba5d6b5c0e5c30f718810f68b019d9359869c54c1009ed6028bc2b7f81ce0d79c0808373c0dd404b08dbbbb0bea6bc84f8b1c83de5d3582aa0214813caa9362860de71b5275e10b739c9964d85eabc002515caba3063bf7d28aac0041a5eb6256864e76b76efdc6bd0221b430e7ffe875ae6fb28da1057c35b95faf28b60c6d4c956d51cbf2b8b1c975cde71eef1ef0e5de69038ff4d64f328b244e17345fb4781ad854b99345d8d535bf72a8401cdfe4fb33a8b385afa38211ce17c1f0a450630454e8fe45c55&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641263269Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d, Name: checkpoints/896/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/896/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=36f993cbf569c2a385091dd8a74bf195d6c355c900be8bda81319f62ee83945afdfe0a4ac0740944e00c39ba25bafe214d220b3fb0a2e024a7ebeb954381736478cf3585912bca8718b34e5b6b5019a6c3efde2da7904c2f1a8816cd6a1d24821c5f542b495b9cbc04ebd1e4d4cb5d8b6462ad9a63fe0b4832680e4c62a2a5199b7c4a4165c1afb4688c3402e20eedfb66006a90e81864104b472552f7b94a2e62bd83b67e9a690595afb17dd1b82c2c35d9f80412512352de1167c193c1c126c7c560ed72ff263ea690c7e7b0e1d31a5d2a9b67d29878c9cd94c800cd0be98d25ded2557748632391cfd5995912221f387c24577f904579ef20212bd384638e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641278444Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d, Name: checkpoints/2944/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2944/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a83206d161471291715c534c8f1332f4cbe9855fcf5b2c97e54a205f220460ccf733bb7cd730f3ed9c82b0bd94a8e42341fd332d3bc5ea8006951225f08c7e9de0c9d3cb79eb796a5f8e1832e87b0dd7bf9c06252c96b58b7db4310e434c589a98917cd5bad36a5019937a39be184b39a5fff33bf40ab42e3b721c83ec7f6f8b7f41dee3fdb058fff741aaefb1e70e610a8f33831b5bde09d26c631b9beb080bd5c75144d87425f241c8d2c4995c951379e1e5973a329d72d0055b68261db2fa45369f818d00774543aa61ad71ef902d7788627a597b6e9be46cf203f821fe8cbf4e5446e1b7573ae385f1ea8e396458abfab59c00f96066510b7aed4c55edc2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641266393Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d, Name: checkpoints/1024/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1024/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=033edb7e2b620c193b8720e4fb4fe4e11564c4266f701417d4d5acf40b64e74b83eed7d65461ab29765ba2b4aec64a40902212b427666a6a8db89a8f01e00424cdb940ab4380e6037188eba9b760645b813b259d3d9562b73859771343011570d6d12529c89d948e43e3540a2fa0367d16b8c046ddc415744c6e43829e6eff394b396822e315b1f81ab084d48f9ece99932b5f512eceee4052294e7e17b8ae9d29bbc7f2b33dd0517131b1252d51553ba39614952505503219713d0cc2dd09abc8821fadf37be09c711cb46161340972ed8b5938d665e0d16eb5e60712996115433a116ba7691b537cfda0a4a6e1dc253c574d4ddfd9d24b03ca7290ef1f03c4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641292509Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376, Name: checkpoints/5376, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5376?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=af5421b40103261d79621e2af2c9dd1015569a2abc9bc59bfd5442ed26dfe3e0bbceed7b9b4fd5c108e87314b5bd2eea863d4b0c96d41138f96f7e1c39859a792ae7de20824f624523cee5a5b77b4958488fd3c5a85b57697f7dac3213a330ac60af99f6b3ae07bd0abfbabadea4ba2895a6daf3adb6fd4b63e564159ba4622f27f9894da8daac6a70311f5172cf11541300f531abfe4a12c860db5984cdb74efd57b42b8ce293b5cc1b7e3b9c7b0198d061421f3122fc80f177b733e518293fe9637f4bfa41c346a0524aa48274500b33f88cf40649e82b4134a39712face121e9f18e7ee8709439c41b9dac38b83acb05ac5958a13f23b97404234259f34fe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641296521Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/array_metadatas, Name: checkpoints/1088/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1088/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=513427573e41b813170bd671cebc1ba41a0fc522c0295176ee040a382eac2036487659e4c65cdebfe094d3ab98599bd36d0b9f72ff374b8f29b72991eecf60d2f6a31d677fd71c5fc0796725b9d6d31778f34b4bfefcfe637297504d3e99ffe7c3ff3753c74f1a7f7288638bad335b4a7430b872bd97f4fc9475732874349d656eaf08e5c2ac16a5b8a7434170baf5a2c5dfb3587889ab18430aa78a07f54573902660bbe93e10e3ded3d32e3816131cafd9f0528c11c67dd62f074e96655b1a1fee47ed958aaf5cde9061375182d84490baf0e9bfa69286ccea479d92bb2bc3081e48f14a0bdb70805ba351857887f1e4b54f05a55b21aa76ad99d266e1d575&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641308235Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0, Name: checkpoints/4992/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4992/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=c08dbf2481002346895017312b628e765c7018f6c5cb2b45f0e85d14262ec0bc122471bcbf6985782ec994daadaf2f452472423f20ed9d4eabf4e56cc3524680d9693382749d8607548517fe52ff26225fbbb991626ad773bccf3c18015f83bc05863f5f898aca5b483695d6d0d728938294238c64d1bf3547c6d0249590ebcdc8224b394c08893a96273b74bbcf3e77028f3447bd3f7d6bbc49aacc96a10a0dfce217d0db76362ef8b074a1ace83e26420cd8d66ca6ba7f16eddef9875afb099be411990d3062fe134aea979a4244378104dfa8d298cbee5e2d7dc990f6f2965e37b330136844c6bf29e2073493df5c09a3d23850cd725b652be899e882038a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641315274Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/array_metadatas, Name: checkpoints/64/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/64/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a08a6ce3836a8f0ed1c64c17ee565aaf1182e0f4781b79c1445b886ae65e7090035bd4d349a6658d7e74461d6fb52c8bb46c3e43881dfef8fe128f685a672fefc8b0e789750dfd3ae8c1efbc86dd43aa0335952cac8e8712865a0127eebdd3e69f94f31c5c172ea47223f64cb494e9ef0bdc76641f9d024ee4c02d683b6f0bc9c78d975020ec176f837976e7e71d624c1b9a1c5d4dadbdaa27085ced60c1c975e789443a095c47e3cc1eb3136ccdd0a5464e3be0203b02d16a9c10c97c53be11882853ea317f35e93e169f31339e1d4c903363475f9e80af88a48db0dbfef6fdb4ea56d46feb9a6b1c5edd1ca13a427a4209df73e46c2815605f7d5491c038b1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641313129Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0, Name: checkpoints/2240/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2240/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=238b2b1051a606dff3b71b9a5eb3b93c92d08794bcc6fae2201f5df1878555a034b5f7b4c00ba9c86d7cf11cb4c9351335d46068ce365ee0bde0e17656be56102169c8bc03b06293b3ecb5d553f1afc6a27ff8c1a048acd08ccf27c0071636092a574d2a861ebdb03aa213fcfc3be563795b14f750f3bec48bb6698d3616a9adc651086c6d53ae806885069fe84ddbacb3428e178a79d4257a6d6b0c04c2e4733e5e09647af19e0aa2042786bb5ccd72bb425bad0a530747ba8479ec544e56420b3d762deff5b61f45692755925a382e8a98d50248784e0ad954d52f866c08ece2466bca2f103475f74a9ee1485c592dcf21e7b7c202ce9e14b6d95ea7cc1f39&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641313307Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default, Name: checkpoints/1984/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1984/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=04093d8f313897a39f3eb95709fdacbfa7718c17d7af4805baae4d9293c2f9f8edeff18b265c7aafa8ddc9c39a092b7b02f7f2b708ddb6554e0cb687a46ea7a32e39ec28b89f0cb0ae86bcb587306a43304bd41e8ed5ee9202c033077cfeded4d4d2304c21613c2dd770e1b5436fce7d7db1be87aedf21620e71c36937d126e1a0f7790dfd17d0ce8392dcace12c1c4f762ebdc6781cfc04f74667af04d7f15c3163d0bb7ecfd298b88c213f45e3effb893bb2fc542aa12d1c29b5e6f3bc95ac3150b92b2a1d8b5c1c761ac59e66e06141dec51bc0e53d2dd8293eddeb38999a58bd2dbf8e177f3a078a80ec21612b9de7dcfb0f4d7db13a8e589961f498d007&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.64135215Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default/array_metadatas, Name: checkpoints/4992/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4992/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=26d3d66a014866ef4ccdb5cb6cefeacaa6f3944dfd5fab2946a037610875914f17b210af1a0912838b5406cd190e15702d73c713ff0ca81dbeadddfebb921c8328a621b2b807bb265ddca06cb36946c52365353f17b83e12e25faabf8d7898bfe0a9e5bd60c5a6b8d69eaace57d98fd3693197e3f32dcd8b82d4bd1a5afb7e04ce2086587d39d381dedb390f447367f244bafe3e88398e1e14bff3f1523923636b29c985b840e5984dac86fbebe2a020be9e7e01aad1d7a4d32de8e609a0ed01fe3a1fc20cc7a3034eb721744ff03129309464585ee71f348fe16cc1a5fda4630c2d6bb579c2a6216b345133e6d7296ada163fd601a2f0bdee548bb52bb71ea6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641339753Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824, Name: checkpoints/5824, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5824?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6ecb9b15a219229777ecd4a52315afe4c028f5698b62a64b2608be87a19919ab8ea6098da7a4af75b11bcbb5d8e571e2a54245c76fcfbff00d850ee878cf8403e2731d65194e7220375564d9759f7bb534348f04c78ad85e9820d361df41ebed6670560a19421713ef89d0fde15a1faccbd2fe054ede488529d6efc3892bb7258514350e45938faee5d4a318f4e3312b65fa64db18715ff934b9e04b2f2b16b70d1f0486fb05fa19dc377e8298ef14b3c0433e787bd3c894d8000ae50c74e56f0a6c0b27dc94fda6a27f176b810c3da4b3eece6c3ded05f613f35e1b5e0f6652d1c7db2ac9f56adf762c9121f07b600baa99eeeb666032db62e73b625b0ec6ec&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.64135829Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default, Name: checkpoints/2304/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2304/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9464715106f52e63db04f21efa6ad51cb30ad3632c4a4be7882b2809c326085096f079535e01a212392a8b35da0689742523c4a4ceba16deb57c279e922fd6844d3ce5870acdb8318e16e5759643cd12c45134e488c57452a5a920d5e1bd4add7dfcf3fd7fefb194a8eb24548e13eb0b51d1096e4fad1a41a07cba92938ad99a30b89bbd08236a51af38a7b82b31a4d0c9c6cc8dd1cd8db9012f98920347c1a810a56e1fd6c3e3291c6d9e7a95c12b385e26e03c3fd80e3d39bde67a1e9c35fafeb2698f6a8e5b0c11ab080c9a352281f7adf3e5847a218cdb04a7c972efa612e360fcfde586fe88ed8af5622c841e88798bcfcaddbc94295c0ead4a60c4d0d5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641354269Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4864/default/d, Name: checkpoints/4864/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4864/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=80aa67abf90661f46b08bb8b328854cc129285773bdceab42f5e434a707499e5ba0c27729532fff96b115739edc0682a5d519b6aa1ad2c4efbc011c51da0a06b3e53f30c2dc9c0316a90fbf46173b801425bd0ab339b1421ea3a5209b73961440086c43d11811afd32137ffbbe3fa5d7c48be41b3cad190d8dda6676c68e0a7dabac83afecbd2b49fd489e01a47b0be3fce52483403aa09d5137607c963bafeaf1a742e43647d1331c5e31477425ae9f8cc23b895d8cb41fdcef956f0d478b740f0d8a9d915f326ab550dd03c314d4580582a4943b669ec5c428621015410e94aa80a6aed420979fbd068cb3551fc49d2273a488d63a00cc57c0b314523897f0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641378158Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0, Name: checkpoints/3392/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3392/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=c46806858e7fdff9a2b963231140397af41832f18d347b6a7248796e9a3087806d711489fd46e5e114b92e3c47d198bbaa12091326f384ceff57e591e9b0a6140bcbec2cf0f2727581bfd58bac325ecd8262a5b09d925a87ab2229397a916f47f8cf07ea12953f1170424c10e0f39df7dbe8a5d1bb24de42851dccbccc94231cc0fded0535dbee8825736981319e208f96d81c3e0498030b617052f0073000983ac458c511e1f0cae5f328045cad9242620d29cb50e6dbac3e9e10de825f73c76e214b74a36e8116e1cdce9ce58852ba9157fe1205dc2e910484a1400ae3f2dfe62a8f1258a104f8d12715c77191a4da41bcc239c45f587256de312f4ea81a4c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641368305Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default/array_metadatas, Name: checkpoints/896/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/896/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=2fc1a107a11f9ea6559121e87b2940fbd1c5d0e7486e6af96c83eb0001051f4acd20062e465b35143503b76f5906222fd6618cf46723ca4be33b2bff00b5d0467ccab88f4b234caab1ab1f77647088e7e19f37260712c5931ce974027993e015b95bc4f3aa96f219b3c0fd445fc297d6549ad1d8dc82a4f47c7221b8165f25946e75bc9431f628ab50d425724383769af345dd5428bf1a2c38729edf9d6bd81b469ceeeb4f3505e0c73314228bb5454700d11b7c8910440172422f82c9e6e494abdd0589964804336168015fe00eedc8a1a0229dca8b7bd9f45747643f1c26357b1a07b62a58ae8b49da35258697ae875543eb8d1f9394b4c538f2d22a01ee0a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641387919Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4992/default, Name: checkpoints/4992/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4992/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=325e07c91fff152b512c09995a608016c50597fb68bcd3ad4c8735b8877ab8d216ed6f587a80ac0ee677e3c7a859f5df18cad5f133797c98928890ce2646a79623018426320471f3ca40f899637b1ea359ee1eeb8d8bc3a28a061fe2585ffb25365347316065fcc6115cd1e2c7c6aaced9b8d52bf765722e4522616db8b86121d470ae82b4fa8f9c7c822d2eb8f3d76f8cc137e8efadb1937369caa7b9c79094142dc1c9e28b2b903254799a64df18159dab79f66b0204fea8455e2940341f60b1754dcff6b24eff6d1808fb8956a41c7f6c16d11e27ffb2ddcdf6218e7fca682b19aefa0d48fe52da003091442caf56c2bd277301a4693f1cbc59f145b5d738&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641388722Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5312/default, Name: checkpoints/5312/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5312/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3cdfc43291b534efa0f8dcd4ce3213905d8b12264361574418e9ac4a6cbd46c9fb476694a17f2176961a6708b9384dfcfb207dfbf9a4d3cbde484d0e954148859afc320eb2016cf1355959310c0f7d03de7d85130ba9002197c645bc9f2545ae114109b5a2c4de5065b9482d363dfd5c7818b79b54b5f97040326f146f9faee6cc95eba85a6c255812d4922ee5b6cfd9d08074fabf34c4b11fbc94dc772ec24dd3df6cc2d122945fc3df036f3ce9c6bb7b46de8955b687224a010e1f722c566376b21f9ea9483c77fb7dad60b53d9fe2b6d27d655fdd50bbc6e4c9f2ebe3bc5631e4dd3155d74108d0bbd1bf873415902da38a1e20d62bf815b6e96cf256a5aa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.64140765Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d, Name: checkpoints/2752/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2752/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=77921e3b224c5f4f3d6e014ba58c062302856d07d2fcb5c85d380ffe158eccbe149fa5ba8b1f02debd126baf9af2817a6706f360bc79f391c8f7e731839d0d68e811175bd031a6808210f6eebbf34e5ef170073f87e35eda6ad7806e9273b764735d7c01a5394734f22e1259dab416297ef27c177d734e4fb1ad7a2bf49ad832da72635bf9926660942a175a604c538ac6c48afba8da9be69da1bdab52bbfc9596a8b096ad79c4636cb3e7c8c26595e892d3b159c5d7e59abe6738dc2d308144ad2faaf3053c9265a3a2b36b20594acc2ed2e4c77ea19b88841e4d1681973353b3f2288a85020289b66faf8e132f433aaf991fe6c7857649d175d649a52bcae6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641405595Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/array_metadatas, Name: checkpoints/4928/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4928/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0353b4f0929cf897a41f045922b3374493f4345016b2de2a82c90cbd5351f08a8e002066185a5c1428ca3ab466a06504fc2a23a6038e03209430e81fcf0ae121845a832bd5fa018b37f7f06274daf3231242a5c0aad87f6bc3d914017c1f8fd6295e29d38956f92127967d2c391605eaeaf1ee6471683837a824d0f3c61ed432cfb1f6e55b130bee1dfcef88da4c36e44296577b9daa177f6320052dbe34dc88e0506e7cee7d0634c5357e776cd6c58071ac9e812aa53e518ea33144f278fa2bf713fcbc5e35fff587da60be8b3720e531c88b3ca9792f9fc0783ac2c4af125d6de32e883fb36a37e302c23a79d13b78148b5c9785d3ac488c4546b7be60ff0e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641400513Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d, Name: checkpoints/5632/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5632/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=44717b37442ea0cba72c026bc55b6c08262e33863dcb6b11d7b529ffa4db319918982f9e9b19e0b1a054d49c8fb5307cd647b931dcfbfa4decb2deb240128ec18a3bd0aebb43914c75f6660cffb1a93da1bffe23a23c85295cbf8dc7a31c32fd7d5389a7bf07e6cc5a0f73f77703a6b38d2483f2aa3bec599beef90c7267450180cb7174619790700655e73707c5c59013859b01f9ccb6760e451f1488fb6cd452ad427145b022cecb3ea6a1a8e6648df9e7cad65d1effd23141cba9a98cf5794f3a7656933c0aef0d025965da1de2e2ba81173b76fc88247a11d252ce14fa3d8a2881704f9a1fb7d6740185f6b14e05853b068c7b52844631460b59330ca5ed&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641424898Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2112/default/array_metadatas, Name: checkpoints/2112/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2112/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=c0cb0d4f97bdec95cbbe80ffe5b38c0a78f727834ca3b90e5e5eac1f4a2e469039995b75411903076e7f40729b248e78d35d03a1397423affe3ff8ba2d8d0d0b69da6a9c7659364a864bb137cb65319d052d7a13d0d22d81f566be5ba5e99e56fdd0a185189e9e2e7eeca33cd6470bb5a41686020e1c9a24c05be8e2fc156a9ce3b11ec01b625e51e473b2e4d1015ee8918e9ef29280231aba5c08a260f9d1e0c201f6c5e4cd7bee3ef275b5713658cd166ed8b8e3f9b780dbb0186e0a62f8887d0a362998ff8fc0395e3c7dda5067b2324c52ffe5b5e7d6f8e189566de9ff787fe6b5b4b8f63c2718c12e64865480f1fa5ae5cc38b44bc7c95bd997c066a86e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641435294Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d, Name: checkpoints/3456/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3456/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6b691075e05f633c71051eb5190de978652b2e7b2d93121a8ebed871029c0fcb1e1866c93456470aabd3b25d3a4d490172db34f799c99de5469c4c35cf93e8909a7068ef32e341a28ef3a450d1ccce45ec6a79f47d8e423fbdff403df3481d6206e56953bf0acfcd6ed46de5f567ab796e53481ebd9f6eb3ef5ea474f59c6feb6e1db4367dcfd8629babd649e9af82a3d35bddc78f2d85d0e870c6adbe707add3dd45e83c5c917c66126a0a0a592bccb926e856482f96beaf3255765af73f7219d8af4c8705dd323582cec661e43268121d93633aab7fe254c72bca64ce377fc1e4376f77bce1821fe0f3f05c22755018e51604bd49bcca634e3d7c6139e43d2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641424968Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688, Name: checkpoints/2688, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2688?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=15dec23457a0981c5a7fc57b5bbc4ee3606b45bf90cb4927d16c91abe16f808ab83b892562cd6e522f8a7ca4ee836b0f7f5029dfbd334a45f9e1f01217f7b6602b90b38f9f4fbb8cd6c2f697fbe1a69d4758df4296de77d93fd433f5f9d236d2ef805f9bdffa235621a3a0871361c05909cca9a20a65b4c1e1d54803a307c597cd4e811c42b857f4789733152ed4e0dea89c3c46506aed3b023265e91776536b550661807e85494590bff8355f48c383f6148f7ca5f340e9777a8966e1bbaf5bbf155e48f2c10635de98f596a7c20ec83e0aecaa6d99e08471c4f883bd35d68777d6a0397b7a2f4cd02c196a722b100b808feafb063a11db9bda993d745c03e0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641428065Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/896/default, Name: checkpoints/896/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/896/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=1a98aef2c555b8b5b64e279f97ed414640d8c8bf5268fe58294bb12733d8b39bd5d11f6c7a120bd0f26cb7b0119eff1af037865e690a3a0cacc6b8822bcae76afd01c71aba17d53d227708fe1e4dd93cbc1f9f3605610090a1fad7ec6559c47ad5077cd17ecafab01b70eff420f1c41edaa80788f10fd4c6c4eee73f8a24957f7b9ca04acc09500a43adfd7008d36c53a5c58070fbc04478b3862c2a9b507f3ff8e8e5d4acb989008a581bb433318b2b70e15e45f3ffb7c601b7cdea8d74edfb5ac112fd17c4f509b59345ba3b46c604efc66450ae938cd6515a390ace4aa4312e6aa3415f9e36812463a27ef42c105f35414cdc8d49436305d75e9d0b80b038&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641436648Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0, Name: checkpoints/2176/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2176/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=ae985e40efecbb3ce02495eecc84a2c20543c00b03b747e9746b724ce14bc7fc4316581765aee43d10ea544c089a3a1736824c58125b53b9466b8a7671662ffe5020933cc71a042f1917bbcb5c88596d6976955899b8d8167c0e5d5497772ddfdaa840431ace354cab3ca1de01db1cd58825fab20158d9c7ce04b4f14ec2f2482ef66cb484988c68bbadcadd9329bdb09dd8796b8dee4afb416c730fdedb0b395207deda540a3ab39276bef2b3ee44d442e18c297ea3f1247b879e05e410066ae12d311ab1d6bee0d020555f073ffddb336e0aecbd7bb32d346de3f50ce576aeab6918a1f4fc68507a4bc05e49459b208a05233d3f76c81701be08111dc09f42&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641467075Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0, Name: checkpoints/2624/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2624/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0681ab5eca67d00a2b8f0c2c80aa004ee48d9f2165646260b4a83038e9c66f2b4e1ee105600405001fd38e772e728f17f55a1e7f7e8752e022beb289d179a7b2dc037caa3a0fd48e7ff827992e7598b1dc1303c2f04574e574b4f8a0af39e89613221629bed06652c75cbd2a4a6f8b5eb32f90157bb29639675d2810fa41670ac13f9fadb260405e596a784f2a5abc733e445cf56523f496dcd862a71f3c6f301ecf7bb88d1fc05f490a2125aac908284e5b1def46e599850f94b25a24184cd591d84880f281c0c3bb13d02a16e2cc9bd6a35e297e59ba623a89b493f056606498a7e1fb48e5566fb968a30f33260506c2b696e98143144ab41b21b0ae652255&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.6414316Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5696/default/d, Name: checkpoints/5696/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5696/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=028360ea5f46d4ec9a1d2635f9c592a24dddd2ecef213ecf46254971162beef16e4fa3cc767970d9607cfd8a09f70dfb88acaa84e8699ca20a07344740e14d1e5a892cb09b73c3eb9c7ba96a7a5332d6c5684d72b22e00542b71bfbb84a043896b8f7c9e717ed1987a472a3c771abfd6443445fbbd6169c7996baa4c75827385ef03e846cf2d3043669da2de6478ba7c1fb9d551b04f8a74609ebc13952ad158f577ffe5e1beb5c1b625d407561df2a83cb613c3c3773db6b415b9c2f5e5d49e859633f9ec83b6e95a17fafe627d119a087e8c9466ce5efce34ed8cc3efb21fe2d5334ee8cae58a3de04492079a90b5b7fd10e9ffbcb669a83542a2a90746084&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641437006Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304, Name: checkpoints/2304, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2304?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3cd6a6f7fef40f2f9990a72266dbb43a9ad7c5f204e821e880abad0712abe6d6e9374bfdcc72de36e65a4f6f87250a96fd34e83d990ef4b713be333034cc9832cffd3d0bb2cc23deeef3becaa55d275130dcb1a48388cf3b7b107ab1fcad3a0375ea5f3faeced50ec103227d006e25fe34895376cd06ed0e600eb8a299fdd710c21a3e95983f3c11fab6b948609d43a70933808b8823add44412471e92b16633da1001fd5d54095de6f496d3de7b15718079bdce3187e3762c960d2d88d24c7162a3be2f8cbf3a1efd5400daeba2363530f3e7df736bfe09e31907daa9651a92f71b2ec819e86791c7c1c5f4d60322ee623262036280d62f446fc3831461c889&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.64147976Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default, Name: checkpoints/1664/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1664/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=06c8f17e3eb3754f519353dcc5c21d0983a18731c3e150184b9e4c65eb0cbe2e9907a443848ca4f6d51a5d1957dc3dbfc2a82bde9be0459cc6edc1f4c3b51800cb76c7269ab4deb46e8c0439344dc41ef77f5da34de563c1d063c9cff89faae45614d14d3dff70198cd97dd1c629e66b4f8cca4705c07090d6db06e8470a2bdbd1c8a02e1da6cba0e69f76b313d6c1d9fc69d1f423c245a1109dfc6c17a45e3eaa8d18a79f4e2621234ad6525f1207ab0a72867cc697cc5dffc17b8b7d30878c300ca8d43593ea07293eae3c323b40014f0d2057cd89b1ebe3ba50e9e0d01d278b01617476bd9eab20fea2fc550405e03dba63e41c7e93ae63c9bb0cb908f691&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641477344Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0, Name: checkpoints/4672/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4672/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=af0bf94fc1228b2f14d222f3ecf0c80cf9d3a8854c23e78917c84c28aba4787e17666c80cd2d55aacd879fc8939cceb4aa1395120ae2343582aa65f1f28407a2a89e4bca92490bc42fd500b63182ffc64124e9ae7d9162e322a0a4b2b30bc76b5e6f19fafe00e09d332887972d162c1b246a462127e7fb244dd96bd1c531474e5d3baf0a187c80874c8382cff298d3245e06ddea4c7f1973c9a967b98a00f98adc4c73345c76786aad27fddaf85adc36b9bebfcf9080112759fd899aa7e0539204c2d09dcc049ae0e09e4d9c4f921271aaffbf396d6bb86de60dc7b19c23cc64c31d67af778209af9bddfa52ad82ba63f8eafa6b4094cdcf55c7cbdafcd3001a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641528793Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0, Name: checkpoints/4160/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4160/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=67eccc9be06a3d3b958b4941d8098458ddc2e09a72a8a3eb043220c33ed3bc46ce403b7f3fda90f4bc821de61bba388fae696754a2dfc4fefc094e0fd2d5b0003852a41d3571624dbbb10ed30ca78c49f43bb2a00ef1a02d6fd7cd4a6c8d82eab3ac386cb584bc063c2a63f8e692b12ca1a24c20be8e93aa7356b1556de41ef152ff6e38009dbf4f014f40dabfdf54ddacb54cb45c3613c568ca1ae02f20d3e0ee0a324a7aa5b90887ea81754cd5999c7e330e78a24a7319384faa08b7113bd2f072d2ef5c3fd1de5eecae00270d6d414e09e6c5d1e0b8c7c2df869958007e58995268c960fa5bc2db789db6d93e63d796b1b2ed20bd03577a77d49367a802e8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641529444Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184, Name: checkpoints/5184, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5184?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=1ffeb4a442ddc3453d1b770660c7cc3a542e243148ecd5939d4ed6142d51a9d681cf7d5147ed1fe8c3d11f7197e43f4b18e2ff490f75b432ebd28690092ae477bafd20e68f4dd5a392567fe034d9ffd4c8ca8e07a4aed2a6d6ed671f35a96779f62e880ea0e92fa74f8f76607d0a11ab83f70ca72b0cd74a3b0913a7c02c146dec9330ee80cf3015e123cf4c588682d8fae238893bf6dcb7deaa7caf0b8ac165ae6a9eaccea9d9c250b3c5ebb75a49fff538cde45e10013c4c2c15090a2b942a937baa7299ef3cf2f16079cc3eb8a69b5a0101a877eebe201a796863be376b7ce7d0acf84b9a77370b9989a4ef3566c0cc8132d954b44f8b12536a2c32ab9cbf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641568603Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712, Name: checkpoints/3712, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3712?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=16fd9374d9527003ee80edc18ed9f1af1eb32a2896357b66148055bfdc209bf8aa5c2ea4f4e7c3000729175b64c8c5aa13af001186a137d5cf811949d50ae7ce910558282fd01fb3f81404c22b21be6cc71338a93da1e06fb4fd1cbc1762ba3ee0bd034ff8503d4ac9384b8c61522d2b50b4e6ec6a07beb938da050c00d324b11ed17baf5ddc8ab4cc07b82ca3f728e7e648e2cc0edec810bc45f51d4356c1c65aba86d0452097d41153a4d4350f165362acdb747057ed53c402ded3a7ed9c84841ca488ac78cd483bff7a2a53dba2c8e17e781d0549f68e80f37bf5b15198b7499e16009bee09379570586fc968800545e15012ae348da3e71373ef9a7e5958&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636440942Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5824/default/d, Name: checkpoints/5824/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5824/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=02318823b71a9b6458257a1e7f4571e2b2a2268cfe7cef9a58b0c2f3b595539a6d70a7e099f1016d46e23c24289951b9ac8d5239a54550b00a3b17f11b2ceeab2c059faa42d02173965762bd43910d23ca7a70a6b7e9655d024f0cb4acf46392453d67fdf762e3c6b13426bca80ae3a047f468073ca34f4a7448f1a9e18300408d69477cc3577202236de31ee4e0d2ed9329b8bb6b2dc9d854716f1f6fe44a9125c4f25eb890163bd1b5b8b20a4236b523535fec204e085c2ad15426d02e5c4e38bb1933c76ceadc31712480e9695ce33e846cded701245d2201ec289bb6ff1ea3d7aa7db41567e0f3d1d6c6b903f9cb8d337db68eb861f3e8190a69768b41b3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641556291Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0, Name: checkpoints/576/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/576/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=248fa11a11f245227073585273c7fa935281ec0c4b145d50d667b3b1d7e5bd083a3b724196232aa4ee5e8347e835cd0d8861a0eaa82248126903525ccfe2a923c3e3b2f2165d1488b8112f40df0ca56473b80882692941432f0efa6c92aa0c4577ada77818d92bd417cc4483c13bdeda6d4a3a5919e5612e2ea5e826c6b6ad342bd5145bd69f5402330a725968e3233acb7d610ecf524d4d872ab119273aa6342361db5fa680603648aed0fdc8332608d135b64951c3b594269c9c431190c1210f6969f10ad7de8446959ace45e0c6cfeca7945fd6fe0f582c2cec0b6a0cd578c2043bb4aedb34eb3397d50605234b63a7554ecb14c3f1560ff422d332c742ba&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641565439Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default, Name: checkpoints/3008/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3008/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a5bcb96dc152c8412f8ddce5adb26f75c41fe21020dd5bfe5311027111d91bda4e9194efc72c520ce15702043d55a99e86303e77352fd08ca14a89d4c1f78110f92c556e914ba3aba39395ac2b61f5336a5ee4e601f74fb2a9e880b9bc585def39d5b696cfe12f2bf9ebb667974f9d6423bf9e5f64bbb9275ac33798c299edf5452aa7116d8ea2235abbf4d7822916e41edaf0385b9bae7d962a0c58cd7d6c81777f0900b6d291145f2f5fb3388141e4cbe8c91bff1b877b666bc5fe2b542bb8418da9473e99133f7eaa248e4908cd635ac384802db4abbb755427d621f7fd670730f56a2a4791fccd6413ce681736ffd5826b6b8cd2a84a8236b444a72cf588&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641590651Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/array_metadatas, Name: checkpoints/4096/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4096/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=bc503dffff65e1e0324f491f228b3c6e0accb904ab55261fa3305db773179e226e5746c45da61104384cdcda8628309b82ed43b68cb24cc6cc6cd1ed33fbe5823ff81021c88306d4658a150a5ba75214505cad0440e8fb99121d0aaf5cbf3976bdf9621ad54244e864b8be53b54b8730d6e1d37af19be698b334e76be62c6ffb1b31e81d3ec021bd4b7b4b2afcd71b665de3afec1100ebcdc33c46bfc50dbc8be57e12a3ecb443426f8541dcfdd1fa6973bb8db176106269f42ac73bfc6ca11452d4132f089becb7bcb449ea785cbab01019636e11fccfe95fa47d9998f5a5e2a47b2b4954ab1ff8a21e35f2dfbda2d2f95ffda0c9614c2aae9f432ed79d3465&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641581458Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d, Name: checkpoints/3968/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3968/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=447cbbe76e594b902426ddb70d342d9e82d297e073765c6b971b9d911fe5980d747d28d812dd469a8eb51c52b37cb7b25ab8436f121c37b78c9e247510ab97a64e197b8f024d593b73d52910da006324ecc22dae0054891272a527f895386de9d807de4e809743d4712050c34a9ce6a3c36edffe8ecab6a35003c57f64e47f6b4ce65d41ed74ea50926cc4c89ba9976e15efc2abff70783e109016b5c4d79f669f31520ae7925ed8e140c23b59d09bcda7c4f5d9523f12881962e577b06eb2e0bfbf7367e49bc86aa6ac138d4e087af00fc8f22441dfe7052cf56d582bb79c52ec8a0b8a1c012e0ece8a055553277229af04872bd28dfc58e0161cd47cf7d496&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.64159819Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0, Name: checkpoints/320/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/320/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=452d796915a2cf81394176a47dc16e0166db5b692136a37a707341bbc2bcbaa9117c8808b8531ce12884d7d0f69e472ec5a3159dc2a452e6f1f28584db1c86e5245ece10354cbc9fe742176b808cef697a2795dab33bd00d01cc44d261fd6eead0e4cbd923af092ee8df7e0a930e1f1f9e3d12f0a9b6f302c8ad772e3677599c5fcf5d410a654065468b8af1fd0c6d6949126ea35345899094e9544ca3e7d80efa327ead97f44e373dc8ae2c968d648c7543e26acb4e5c68000d099ccaff83cfb0d5fdb7d61fddaae8dd9b8795169f805b7e0b5dfa3a0f46228a7be7554ba5526aa0bdc650d9465e6f895734507a22ae8f8a011faaf4a1858ba8e5a86b6844fd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641622321Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/d, Name: checkpoints/1664/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1664/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5be3f997be355ddb3aa26947a071bc8a62e2645457f5caf586de5b1b64b46b168d837b742e2888dd2bd39c5b32c6a2ba10b27dcedf6e0676d52f8dcfa8563c3cc96602795dfae13ce342f5a61189f62c3991e5429692bbbef7366a3811a6a42b545b47ee872588ba2fb4097734360706c7c5a527437cc26613317d06b660baf116e6abfd176cb8090f0a06d8ef049a33f6983084c4c86cd18448143f540a01cafa01491267dec5f5602e3213931c596e0551a4b6a02fc53abdb1ad79187e657b3f9e5154c69d11019b74c8bbd0dd5369c1cb766e7bb6b250930ccaa3c3790de3df850c14e185a519a7d33a39b6f4b4c720993fddfaac76b5066cbe24e93499e9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641629962Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2240, Name: checkpoints/2240, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2240?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7f7b7ca3a7c51a7102a98783aacd53530a9e8c226152e3add327b98969ce9a059285f2c9672e8019be58fb2b850c83ce1696a1a84c0cd3e58a381c83566a1bb619aa60eca5a234606f7b319e96cad0f5fce1ea73f6f909fd11314ac019796eeb821fb414423adb49b350f138ef483d337114026ba4c9807a356bf2210cccd49496ba697e9f8fcf640f62a3ef693604c3d05ac665581b3be1634a9e26bbfdb3a0fe4fdb197ac3209199a2eaa0f66e26681c1c430afbf642e986762774e3c9aac17477831b24c0133809e5123192939e2b69f26ac448a67b50e2d6adaeb9b3e54b08fdbe8dcc6934677f4997cb3067bb58d6501fc20954ecd7e586981b825cf025&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.64163511Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/array_metadatas, Name: checkpoints/2048/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2048/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9914a9cf910dfef64f849dd90b4b92ff29d51606ae03666905a8f177a4fec414885651f55a8abbbb73c84254cb1cf48b824eaea5ec21a2c6b75f191668e7113755e66daf4abd7909e86048ff0fefb4d766f7786e7e33f862ceb84ae50317712b4ad274e80c90dd40d1537b9b454885f630208bc4da544577a80018022e6d1734f277aa01a2bc025ce0fa693b055128e6a747cdb9cdf820cdc0376c99e7edeb43839ae6fca9d606c744e1ab0c456c11dce02fba19ca0b45568e4efeeddee790243218bc56c96a47694c277b2e3606a0bf26b5fb956adaba0af856129279a37ee92ad2985c77c26e2b3b3e58735a52b5b98a075b1218ccc5d466ba4437d2cb5db5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636441399Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960, Name: checkpoints/960, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/960?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=158a46ceb5a109b2cdecd816479ee4741f0209400536146080085a0aad210695e17ba2091c5c6f7248cf326e7cc67e008acc5058d3ee07e3c3bee0310754f3fe7e609098c09d4fb1ed7a26522128a4e2680860348b2a4990727e3704c3858a60ba0b61d1f95c6c4c1a978a8f796a69036ed38195726b8beebc82ab7ec9d394cb80c482dd8feb1961f41296595af97f13a070cc0cfb425787f2729b1001f0cab8edead476e44f13732065235b3ebf4e1feea6f2e81b60819ea2e74af1474202266923f9df3ee4c4f8b14569d3245a1e394a4ae6047060a47c2e217747502c475fd15a42a38df880230f09435d929d17a7234f98e4ebb573bce6edc77599111bdb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.64163411Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d, Name: checkpoints/1728/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1728/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=14b76e68a1d8d968e6bad50127563d085e8cd5ec84e4714bb578d415508d3d5780552efe4670079e1c991d6c1e037ef50ee485f9b5eda7e20de45a40ef11604db7e04a543d732f57161a888087831fe58c438d146d19f46ca7db1f705669842e86c29e6923d20a27d237d8997c922fd09c65276fd4e333ea67bbae9656f645cc8bc27683168915cdc89d95ecfc61953adf761949c4cc8e590b87e0dd183158d6ba7c4cafc149fafa9ade15ba82151cc46522f5a82cd50cf25838fed20bed11500c344a6e38aa0f63600102e7594980f86864318bd586225dc71250b077a4aa08dd817b13fde170ae45f0bbf07d1d0eb9b2f19e9af7241c5335331c59a36b257b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641691056Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d, Name: checkpoints/1344/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1344/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3f88a8b03fc828da95ebab1d48c7c9c2d4b8d0d30958eec40114000ee61558ee4710618179fc635a4f2802b3d49afa9e2bf8d9a00bc9532cc34e92a7844e1d1a05067a421ae2f0784f85e4b8b4d003862c6c9363c49f5a4e9a4b520ad6be1c5189c5e70caf6a51fcf1f8fc0dbd45559f538afd3d4d168d61bcc40271f94f35ba3052f60c23d304ffca57b80c7c4aceab27c55b351ee79695272b9111db29fa9cfda5413cdf5c05e6cb16dc75d7b5e9176c2f2e3f84200c8ad19d2375c56ae5eba39c9f3a8d6f53a279e59db127fac8c03c4400b23cc627c7600c6a169d7a5c4d9b699b0b79c1aef6501eacfb1edee3943bfbdbb23faba026bb4c160848ed83cc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641691443Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d, Name: checkpoints/4096/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4096/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=61c8f04c9c0e25b7931d7ce1224df805a1126214115943d67dc5565ad908715023b3b5bfa5e183db3045b142467a4195a9bed515020d018eb6e25d8b284a03f8ce0f29ddbcdd80406c72d4d70ef0da741451a7b823541f0159accdc01595974f9c2be34ac3b59a6f64382dbf9508ee07a8cadf48de47fd3ce98673229309fb29647aa79ce4d808409edb3fc33c8d4cad73a6483831d1e5bfbc29e25d8e3f3a81b0aafef5015193aaac7b1514894a5d35f141ed47bdf085ea2c40fa32f109954b6c93139f9e6c3da34e9cf915103b28b82fedb749f5df22f1311e3b99f634e07bec9f944a134b9ba6d8843e659bf3472d8fb40fbf1f360a18a763879f19b69cdc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641698794Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0, Name: checkpoints/2880/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2880/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b8752522ade669b98d08d7882c15e254512e53f765b50043e4f08a06a8b49dc98a82a674847c307f889754e6d5a2c5263fcf82944790b24129b06b5847a0fdddf9815e86dc7345575caaa5685134e0ab5e49c2186a2d882bcb93b61c174582bb9a99565b0b3a4585f3463398f6a949a5904145c9eb430650bd137bb69fb2ab3845dfa50358b446b4e3016e466d55c909271f595cb5499e20305ccc4d9fc44e3706cf4433cba8eb43657f4f5c83ea4598260685477a76dd2d586ed17a634df5e4c2d336277134a4f1f3b26c843281991a6cbab89a680dd507fa474ff0502737884d9544ae43fc5355242bece90d508d53a8d86d2c874431f8ed7cd86db374364b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641742803Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0, Name: checkpoints/3072/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3072/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=23ded5ce7c11b3789fdc1898b964421cae39e0bf7b61a69669c7f7f5305a38efe07b73eca6c04036b135df930f6f67ef92b57cd321e27837a175de67f8a60d72dc40938a5765de38b952bfceb1acc698831161b3df095a3a8527e61511b068ac3b0917e6bbae706fcb72d087a7f8dab4889245a21fb1df9bd0440add92668bd155eb434db585fbcc72acdc70d8a1b03bcdf87696475cad128cdb4a4ea256d87ac0dc42c42c9fb8fcc844d351085790f582edd3407d03abd2b09d20bb1d76f9459fff70f3c3eeaf5fe6d3367e4aeaa7beacb7c89901d0d22ca353f508ba2b70d48aa8c77d707b9e91b8a3e6bbc5d609077929e16ab914cff65cb4b206c76f721d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641755071Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648/default/array_metadatas, Name: checkpoints/3648/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3648/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=8132da616126b0dc6ffaf10dc44c241be419586171379ddc62e972c37abf8bd0f01a429e1c2fbccaebe96a0fac436eef795b48defd5a93657201e8ecfd968efebc2c783c19878218d1e5633b1588c7dc6cbe8c516a9c36e9eff612f8b65d602ea7af2429a6c5981f1e130fe6e4301016e5a42a3d50a76a220376c3a1be9418745abbdb6020c301ee0f67eec542c0438b5f1af90dba059499b78aab5f198ab9aaaf072576e079c955988c9d36d5c1e3c99e33628c5b0371cf4c6e54de83a3e005a22faf2d1fcadcad6116ba122cc1e2f5bae527b978f6cdf2dcdb62a055f6a052a6262ffd471799916bdfd88f004b0b00fa3b1f316fb2eaf1183a3557bc5e3c97&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641824885Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default, Name: checkpoints/832/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/832/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a1d126b4987089898b1e4b46afda5466728cd2b4cd39c1a4eedd48eb4a8937de5fd3de8e349d48b14566368c6683dd823143d54bc77af52b69e85c05d8f84b22f042bcf1c23e15fe7363097359a31d58b0f7ebc79e2470a183bea6fd9a5410cf63d292fd3a2d6363ae1f2e48ae1d2d64f8bc7e4e21106e4f6c25ad03fc1fc94b07de9e08ae58e9a2b57cf56b071c08e48b9beda4d5f1454d5a179087292763c26477211c0bdec2ff2f62ec190e7807663ddf21ba5c7b047fc1196c7938caa07e2823004d5864669a2a3f9a6bcf72dcf5972a8846af20823350870745aa17b7d941a8eaa8a46fa59bc6576510fa9d9951a21ea6a48c6d1489627fdeed57172479&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641882337Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d, Name: checkpoints/5952/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5952/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9f816b67ab6eaee60f4e54c934583dfd72c95eba6a14b225576b5b76e3db070de81ba0fdb7236aa940fcbbe66ea3e6f4307b4de6066a0a48cd137a57a6a25d93d63513773c1c555d1d9000a3c45fa41f643ff70a7eae49e004be187c8d1a175c4e0866864cd01dded852f1abe14b292b0058af13a9499bfaec184ba1010f8d89c50843f4fbb77d7772223890e8394bf6e58759318025919de8616fe4fd38fbf9757ac21764d64b5925445c82846d5bc374aa10288c29b471aaa84bc82870449b1ccc9b1688a9225593b0c32bd241b86c5106b35b83e1e3ff3c2d8c22ae695c05aa5220c39fe9c0ecd7c405d14a2d7d356f20855cb8de3196aff5d06c7f27617e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641883453Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/d, Name: checkpoints/512/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/512/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0a1defb2845a486be65ff065d6fb642068ffcab4d9f152c5a7f9639a3c0ee8486d76de1cf5822e3a49a9295f06338a292cf8488f11f1de4b56080470b8935af628bccdd7212da635d215b68f4f884964f4568eaa7f373acd1ce7f6b4e331abcbef1c78035db7ad00d810b3b51ae4b90a57f744c8cd0a6a93c4f3af03a4c4b0392cc761fd05c7ff054ef22998af5f33277d3f48855c8a973c4fd9aaa880553498c0c3ef8f2a171df7e84f2d5f85509af206619e19e97ae6c4d652c408eaffe7f04de5357b8a277a93a81ffd730169592e4ebc44a1f87b91124cf1030c2040edb57a7f4582d48e930e6ca9ae4eaa9044b8238102ad023ec765c4188155f83d0cb0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641940578Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1344/default, Name: checkpoints/1344/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1344/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6d3bb8f2919f6ccb454c02476020656631a7e0779a18716bd41c4ad862fa0546a5d08c9ca03dcef07ebab4386876939edaf4faea42f22f1b9f8ab974f35a1e1b2594b98a6facaf96aadefb27796b7f22db2195f706370d877b56b137127e7dc9d13067ed8bc82cd7cae653d37d51a10a193e8a087376dee634923abba5aaf83d03b9080abe5db296b6b3dc3f2f2fc7f6c54470c9c85e2fe39fc7e742b6fb1bbbdf07e956762cad09f06309c80a18b3eaddd9ad42484b42d8651f3752d494134d24989650e051321260468db7be1faba7a9540ccf3f602ea7d0ef17c335966a1cb7958dc4634b6adac40fb86e399123711f134f82b08428421ca7d14434171c2a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.64195846Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056, Name: checkpoints/5056, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5056?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a1b4c1e77d2b7e8174bb29883ec630c64d4b02c6628830dca2d1b05514b3fe3381b3d06fd091eda28a31020aa5ac33e2e95507188c3e17911e47e1641cfe34d9ce27b1ebd73d5af68463dcd7baf7a7a93929bd1623616d358cd8c6f74e7aefcb3e9fbafda8c758f993f41db4a70559ded742ac9dfb1f9943f3d82ab41040b38a0d53a7c4463d8dc1fc23e6e37f01ebb5d48ae19dc5cea802bae9586cf680e7e7abeff9e5adbc584a1c83456e1ba1cd87cec44bc9ccdf76a90bc214192d331ee8781088c8745691dfaad1ff735690993ca806f0bb92a8935986876b0b586f78296cc67b7c183d56f1dc3c30cd189bcac7fa6aee1b354a635f24ab7aa8641137ea&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.64195685Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/d, Name: checkpoints/3008/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3008/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=127e399d22766cd90ad49ce0272b9d446893c5b39def4d07db6bce1ebaf62669aaf29d050980421239013c73f1a444f83637c74346eb52b16276d6dbce9f9e3b19060b27308cd94f7b28e19ebd8a8865a11c76253edaba5564104e7bb0a17974e23ee7121c07ca989f01dce4261ebe866bcf268e6d88b60b5698c8822f8c978a29f30f0c8745af690ba86c4f56086fdb37a4e26d04153a120cf5d29adb3cf82880dcd907f804939d4e1d809dfd3bbca8d047aef23ad05f59992e44c25b5cbbdbaa29683798031bbd08b1d0de194e25b8a44466c1edd345e738604e07c73ef4e05aca904362cb622e0ec0c651da71f9bd949121e5a5167342f3dfdf5a9ec263ae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641975283Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/array_metadatas, Name: checkpoints/5760/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5760/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=12a4c4775cf48469701eedc422f11bcea6cf0c0a808b0cd5e18672fed80937dc6a271659aaada644e97c12c21fea290c70c256aa1978eacb196f8371d1e40dc0a3fdef54c1ef42ed767b563869b3b88384e13975875f530916701b00f99d2bb69b5bca7d8b6a1d7d54b138f1823092aaa4bfa8659edb9a72aed579e954e76d5ed61f25ef3d12a84c326f5811e89da8068fbd3ecfbd15de812165d8b58a7df1de57af38a0569229a214064ed6ad33e33cd24957f903148028b9415769e98ca868a847c90091a5aff790cceacb91048ea6b34ea77f42567b7ae455d79634f9ee023b51a40288a39121e9c615a9cf8eecfa3c59a50702e759d53d8ebe3b9d89d16d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641989309Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/d, Name: checkpoints/2816/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2816/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9ba9930d49ea865909ff61e57713062432dc9a190e032e02e2996a07d85109298e02aaaff95f8a8d8b3a8ccdd3223ed22073b21dea63bd14917afaf03a42667f3b4ba5071cb63e8030e58d66e1355da7d93164a45d5e843a13c4775c1ebd399d331c4879aa12248d26df1fe20888465b6c4627be7e0eb6b50dfb5f36d20d18844faf8ca520599248a14a93980a107ac4eaea9359084cd2234151ba745d0099b0bfbf85900e8774e5d00b6915bc50bf97c07d6bc0aba841a061f84c7c3bea3f5c9a9ace645f6972fc579cb2babac37715b21f911a530f9b91616adf3fd1027cd300adcb10397b0c24efcf1b1c2fba985653c1f78d658b98818a19ff26c5afff47&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641988351Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/array_metadatas, Name: checkpoints/4480/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4480/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7930cba92bcae1691615124a59cb3b0246e1740b490d4883c8cd2d69e0eff9dcc12564080957ab33d4ea02479486d6d732fab7c4e8201798fa3e0966c14db2b506fd77610884ea6db15a920d2c302765a8a3e1fdbc3d842bd2976520d4f6a3900f897eebc6cd2026916d8173a96fa59c0f4ea08661a5927ec049e98c41bf3c15d513c8d036896da63b8062aa1d958c2857dae189dec6080b768f7e870497a3fb515e89de8fecd7993bc1b8adf2102354c99929a4dfcd0ccb12c812d98aaf0aab1d80edbe0e59516467eecade7b9e40ff217d1b7291a423067198f489b5a77770abd63ebd0a4dce163f48de173702da093c183f1b17a3fc46cb56bf6f9dfaf3a9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.641980606Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/array_metadatas, Name: checkpoints/5184/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5184/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=285965624629a773e4a3ac1fb744ed6726b7aee90287d5f2ed644b0772737730fb8c3c18a2965fa978c9e2e2b69bed2808d80e8caa6cad02f9fcd23a644b8509ffb72d1fa7cc9657a31e7d1e1973fdb48ff5a69fcce6175aefe388605dd2f36b8ede96bb51740066c8aa67d45b4b63565814ca614845d913efd2c2a48fe03e5e73284baefcb58e89cc500b33a20e9ad69ca4462ed1e6fc0a52319cae64ac4d6a813434565f56c361836d396434dd6545aea2e38b9ed2877108234e6c965854943c2b0f15164bc1cdd1f545bac44a99345729bf947b74f8459128cc9f145ae6011945e4b0abcc3960e799a645f9b1b693be80e610847aad677c080a2e9da6f724&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.64204231Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3520, Name: checkpoints/3520, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3520?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3318f78b13b022ba674210effe98b8e1dfe6005f2063b2c695045764a2eb0e0560a922712957be78df543bd831edbde4754ed0a5290bc9f86fdb1c1c1686336c23614be2f1c3611f409d715da6aa991b5caccc75c78dbba0e48d32eecaa3aa2e4010ebf55fe6e57fbdeb2a595e2bc62427747ee0fab3b0c7da8c370109c40f8ba9f481ca54772cb5601487e5980b639c325749fb64ce26a0330b03ed3cd944f0caff7901c5e9f3953dd7950474cde9d2c28e94b3a0b494e4ac0b80c5ae696a702fd4b98555bd1c9c78d44ec1cad560b80df10fe2b529d782dd8924a1b7fc94cbe553c82b3c8382bfc0ca0bea5afd9878defd59bee7975e73f012e168b40be044&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642067137Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d, Name: checkpoints/5760/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5760/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=57f682fe409c388aeaff33d3107499fd1af7fe4fceacc0d42775959d3c59825b822c2bc2425508071bf3f4e7a2a2608aaf6dd7b6391429f58e266e8e59dba605d99d97c4abfc5a2c27d39f1b05f4b2795bd690a264c23ad2a174fd80d7e3cc717a717fc98745cf7cf2d8305647a060fe949ef2f17181400be0248c162e9c205f3c15ebb5196985aedb87a2c7953bab8e9e6a64f2053f8e351c8a091cf311faae9ee4482d3ab832e5f4b424b8460b34002c95120c48054a76f9a29a57016162e2903b56d736cc19761e165a971c06cecda84da90c8b79f77385214478c929e7e16d66e66833aae7e3378624a6b65ab14416db2b8e935103c4cde9bfbed0b640de&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642124243Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5056/default, Name: checkpoints/5056/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5056/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7caf584ce731bffe5be9ee10150316d7b1975361988feecad5741cd0337793237efdd6ad48e9cd5d22a2bcb6c1f261bed3d36b86a882bd1839995be253fb75341912fc080b4c3cd388ece0cc698f3dbfa6549148e818b29042381a16f2bb587eccf11b6ffc8cb03221e3a0e705ae46424ab5f62b0453cd5b91580135c0cd7e9177408f42aedea0c1f3b6e8d6e6426bf4a904779b0990a82513897f4a9fa41adf925aa80e6a1b92a059e6e0e0e6013e7d28ae7fd0923673ad6fdac58192916c2d8bef98aabf786c3b3800bea0d9ef69254bac402f4bedc410c603a12e549de75efa94fd0b303889cf142476d54087b6d88ae644ed3a97c9306c439b4e51abfaca&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642137947Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0, Name: checkpoints/4800/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4800/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=c0bedda1fbacac64a2da9384dcc4059a5a19088844ce5d28107c4123ac922a8763b7d128c16d39bc949a46e67513b939b5b517315b0f038b7ca379eb48df8759673a7e8576d518d380435202a0550ebe12972c17d1233623a9e7f17a85cb642db036b944567fa184f57dccec4e0fa199e7b21511199a3ca889857206f810b5ddd8ef0e6ae8c8e7d8bf4aca2c43312b1aeb1ff9e3392739d43ceeda6217c18c12a07c4a71e5b650e8e7b5ee6e3d195192481651b9a5b3cf27231f35f97815eb73341ede7d2a301febff6576d5c7f80b9144e87114ec454619bdab3dee7ac052cecb07970cffb9b1075e2f27c5a9026119547e2b763751db64c4e0942e41b74d75&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642136407Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/array_metadatas, Name: checkpoints/3136/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3136/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4c676087fe8a8bc621b2e74f6a4fc3d274ce782e5f1b2245cd79b42e7c3b294544e981859b6ba885a11a3d1f3bbff9b55c0174ac6c16e7acaf577d692db67887f369758832a1ac1913d432fbb874ceaefbcd3a518b2a79cf5eaf0cd8d98e23a60789ee76e13cdc6aeb111b095a63512edc6a52177dbf8df56f99f297680a1e898b22f5f016a088d42eb0830e23a1fa6dabf5041523bde8d9965702006bf79113106b875eabb1e354c27db8922036c756a4d70236c0b21dfd2d3b52892e3ae810f08c82e1278ac9a063565b9e5de32a78f45cd0fe99a81029d087c26b16d3caceab0b8e230a02af3278522c481adc35c40b24ea7f06126ada1b17c13aca3f6e98&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642180788Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0, Name: checkpoints/0/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/0/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=263fee477ca15fb0ca88ee8a39896b423b0466a9d40b3ef1e58b9d181c56fd71c16492e1c22ccc55e5fe02baf81f133d8f714aaa87494a719e7670cec310e7430845ad82591a8740ba3f0427e80194752cd22bb37c97207bc0b8ea658578151123c9ca46aba59390e5f7117be2844fd3889c7b7753758b451db16f344767bb88629ccafaec610afb85ffa334f4b7593639ce2ccd7b1b184b5ffc3fadb5a43ed4d3444a1300f50a3d8b524f607ac70e80130422adb771b7ad5290d25093a1336b626b313f68da10d12362a22aa9eb69e1124ae7e4f104dfd6c32ab5f8427b10c559f2864e5994fffbaf0555e028dda729f99bb0caebaaaa914c7344f660969be3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.64218818Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904, Name: checkpoints/3904, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3904?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=48c2803898155587ee938e9fae1d6eb0c0f54242491c5807a6172b2c070bc21a5a9c7bab900a128765b5560eb9751d65b386f6ae2e7ec3d9cc82b3a6b30e79eacf399fb2065988ef6e54a67a6bff02d5a83886b1c72dabcd55e773ce973c0a55883638e1838e26455844371e9430ad57dff114b2e30bda419ca58e5fd9d9bae631ec89437ea18002af23f7757bc7e7e6ed31d9e3aad1bb05636a03572f57ae8304da6697f44c056e91105b5e5c5b17a36b406c18b87ee4524eaa4dbe25e4fccd4a13cd9ba979783c5487a8d62799c25eb34700a87bb22650230d5213e89fad1eb85ab38be45d451cd715184783ee06ca5a5630952985c3c6469f8ad4ccafd3ff&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642188835Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0, Name: checkpoints/2560/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2560/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b4ca9661a8640e956321dfb13f151c7457ef1e5ef5cb39bc822fb2a812151cbfd8354019bf7ab0ed0b04607704ce5ff77478135fd58e1614a9379f95b59f23aebcc45768319331960d1147f12c4daf42cea21127c23c6a3ae2ada859fe92334c0624770c3858122d5d373343b085013693e650481c2cb2fcb7748f7e1d8f4dc1cbc7fc4a0f942ebc050b30f10384a125a1207f4335449dc7a606102d4f4fa1959916f13735bc70c74ea3f4f6160b9b5c6834ebc2f3659fba5934914fad364c3d80a1f3cf6762cb6111d386b19cb9b407e452aeb2fb9de732b85b973327883a0d524c592d1463105b44ef23f61204830a0ef818a2e13a58cc2fc9d2f843df34d3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642219914Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/d, Name: checkpoints/3776/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3776/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=744178f2fa85c97a275445f0b79d512ebda81b1747fa6d549ba58d836d72a3bd94322b16be2b8fedce36279dbd2d0ef27731109e5353229ad9c3e2aa1e41fdbf70fbab8f128798e9228db96bd069e27810e197114f0f9de1dbca40cfb6d3db10e291a0e6dbf44be8fe0edf5e5780860428f93ab5535db0598e2b701c1a76ffced51a8a426353dadd55fdad43a0753094293d20679bd2e5ee12bbed8a98f407fc736fedf790375a9c4fef243adac3a7a2940bf2ae03c43f9e8bf1cef1c050251185d221d53d62ba0dbce2e907d10c1b4465a83bd545d09a2da5b6300d741cb8c5f35002e9b3f21825d67566cd506f86370896d29cdc90385440107a753c857367&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642238414Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/832/default/d, Name: checkpoints/832/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/832/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4a366f28dc5a6079d939da59767b40286ec6d93c8de50e4ae1ef86db10e1acc0b356a0567969f10a27fecb75c815b24da58d6607b4b61ed066d16270c0266de91191f7a3b1e1a2107bdc7da68a16f622e8f3b90d0000c18086f71dfe636c01816df108004220ef11993ac075c95a87bb6b4a5d222d8a8a4ccf1e3c3159749b41dff3939e9ef8b30be65e816f74b4e1bf99961c8066cd2d49ed0f4196545c57756612aa6837c5c3fea6299c174db28e223a41afc37b8e8e528a653386b1b44312dc1f1233d5b248d592c52bc406d18c2fa8a7172ca34479b5474ed1198a984ab30ef099c576c5357ab4236d6caf620e580bdd77bb6c73b6ff4aacecfeecf9bf97&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642248161Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d, Name: checkpoints/5376/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5376/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6394609dc100d0b2a311bf3778f87557f74fa6a839c190c721792a9cbecef2d0afe9daf8adffaa8faaa236dd0f2a922f8bdf006aa55b8c0bbe2e6584d40904cbefaaf11a8fc75f26a10e68ae164742393c1377e0118063d588d63a4dd540da2a30f258b8cdf6c469326fee577073c2b35ed020f1b27717762fa3de60485fbc95c073c9d517973a4a8e26732a28a8d0d7e75b7674bea041098265407a8ac8d72165a5a83b0d441a2bcbaf643a33e0ad60d5361f18b3a6fa8692508ade3ed8c731bc7d4fb35d6ee60d98e60567fdd2c4a27a311af07e39cd52647776979ae9b616c7970fcfa0f428f6d7f8f4e7a8a5a3f958eafda71161476cc133c2cae8d5e5c9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642252701Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default, Name: checkpoints/4480/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4480/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=85960e205abdd12a4321ff42d9746f41397651aa04c9eaf172db234b4171ea8efc11e063a36652dfd1d80666923d51e4b3fc62cd9fa46e82ddbb9330ef10b2e898a1487c682e45eaa767ff5ad28adf6c09fe0a936878a4de465824263c898b374ed1f770616c72fb97022cbad72539c6c272ffb34fb6df984f5fa8b6a147a666e3c4a4dc346f0a71a390111cd86e848e943c50e2af491cd351d2c6ddedaa041961d4b7f067a12524541e81c44f1651ea020a923a817eab803219f35dbb633a3f3ff72c435c7fe94441aa4adc60291474c07d69f1d2f922ae8e398fce0f4027c1d39ef7fdbcc2a7526efff4a65c322ad8bb36f9645bab5b6664cad12d0c96d812&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.6422704Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944, Name: checkpoints/2944, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2944?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9e11019f07838f5f056655cf51f3791179d5910706638f4f1af46c6ef921ede5f39d3e057133d53a86064c01170a3d228249806e0e1f4515b6ef3d0949360a49e763ca296b793f446b0c528d7ed4040628915480e7e8ad3a5d72f5799eb6ed960cd65f1377baac18d702eab1a00809201ee91218c04176ff6ed04379d9296643a84fa2a19d92e9c67f5ad3a8245a096e63c9f76dfe531cc0bf6f6fe2da9a58e7a5d4bbe0f6abba89567ad0923bf3e39f67dc99871c69bb0f68b88697d935b9b0e8205d32f6419f88039fefb211fc3e4304a32ab7c38f8bd9745a2973dcc90edc5df1329f7797189cb293e817de94403e2be3c4cc005224a09af5f37a09b99999&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642288093Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default, Name: checkpoints/1088/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1088/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=095daa04f68eb8601d7ee96c57a418ef467821055642d990a38250ba7db71e4789039673a6318cf866de7aa6eae6f0e088ebf58df8485df2327d86285706c27ec1cd599ece96c12f527d104efcf9660fcf985609bcd66ca7a92b1cbc3ce7281de85c5e75b107179969b4f470179b8a153f27830085cc18d9867a8f96a1b2be2e5c6696271a2e9d2df40f40eddb7391fd9fbb18e61c3a6039061fecadbe7f3a4a9dfbb2f6164ce9718417d74e8e1a770d7a51786151aa41bc567731a20e6afe1ef779b65e87ca251021a69cb750972432f0d8d9780948039e04f742ec6bcb6bf862f24ef0ba1e83ba6e815312675df78866f60d471eeba85579063c41e918554b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642285163Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/array_metadatas, Name: checkpoints/5376/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5376/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=5c7c228dd57556dbdcddfbc542e689042ad81f0035948ded5c4a7ea742babbc50a9253817ec76fe24571ccfd7abd1135dedac262bad3ef888c5015f8073dcfbfb04784c0a1d64d366481ee7a350b0c0833a3d75f219011ffab0e7b8b59dd4bc91cdf9452c69ca566d43489a87b98815003efe19ede2e4ba57d249803ed5556cbc298651d6ef9ca40f0cb6687bb5039c1638fe586ffc3a4b1e520ce577cc1a96127a791d4beb26949433611c3319349fe6a390c8c08a7fc4e45bbaaafa1f26362bbc8e32d88707e81d930bba7c08fd03c9fc5ce532b77ec6764ede317fab49a236b32f595e1598f2e447ef5319613b9056602ca3a932561d11a48127440a84876&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642327168Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/448, Name: checkpoints/448, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/448?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=2f36471777e083fb6bc3a7ed275d8335645d47382a26e0188c70d9a8a7a4d70c34b203248f5c8fa5935c4a04422d7684a37db5c76dd0ab694b78568426a37067d339a1fb58b8d1cd532c1be1209ad335fca8aa1aef417a9f11ef8d82d6fde06637189359758fd0017e34150091d0d177866a5d09807c397881c520f500560d91659c8452c3275485704ccd9b194a01c5c4ad10ce8dab004022c677b97d8af8f580064b8497cae46dde97bdd673b27f0d77386713ae0575dc18f64d3b3723695f9d4eedaecbc1321fd41953c098ad2db0faaf5aaab19ae23551ec0f02ee7bc02400cdd3d755339f39696d7d5b757b706a113b1dae25d942992fef0d16b8f1a22a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642315236Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d, Name: checkpoints/64/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/64/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=61670679b8ad9f9ff4b6a32efe3ec53af172d9ede7cd9253f0dc98af8f81858fb6b83b7a9e661efd368d4e3fec6a1f3e8b481223909fbdfa5bf51f93dc34f584ff83e033ab26a9b00faf330e828e2f536285cc0834ba0abd5b0e91ba18af452983429da51452cd5a9bfb25fa94aa7348053957a84753c7f6203031b979441b1ca7fe5d3e558c11fb00774855be3464e3c3dcf9d6966d0a5dbf8ff109e3d447b7354e8bd180143134a436bf9706e7c8c89e008fd7b7d9b84d3ebea7b2e4a71b354e706fd5e03bb807fc52665ab081720c4032f424d4b1d5c3090888366f25946ced33ee7d36a2ff7ef1f3095a7d16039b7890ff8d7d858bf6075c4cc0e194d12c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642351Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d, Name: checkpoints/5120/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5120/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=112f5ced9b1ba0268a07b0f7f66ee8d24c1d3464757263626174fd1091f5d42a9eb01bee401c0dcf62b0bc2146e1b33b95409a516aae9f1b82ea6d526585baf675b80b6b155ac59931115c9c16d1a4efa9d2d2b63b8cded29576f7e7c4cf80d81ef2c9b79f3338b1493692bcd03c7acddcee6a5a961ac7b2c0eebfabc0fcb394f947a517985c2d8e8d0a7f2d82f62e6e452ce8a54592c043eb75bcd1011d72ccbef9a24aaadb46a4fa45a3bf2051b23f872948176852030d93e50aa6c303e0dabec1e762ee6d9bc0ee2a38bc8899a96e8730a004c5b5b6833f988eb346c4dcc387ded9f0a3b2373e4e9ae0deaa8a26b9ae3a4046e3aa5fbc2fcc66b4bc12da09&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642387702Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2944/default/array_metadatas, Name: checkpoints/2944/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2944/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=617f9e76195d67a25b9c8ea155d9a2c04d1118a1327c2f024efe79371c50792961f55152494b284f9ce68618fc16e8a94000d697958bcfa0c773135a9320e804d7b64f68760a3d472be927accbdbcbd21f2fc050a30334a1a663a512decf376374fc36d18b0c079b2a8ea11ed26d8f298414e267bf0974ce04e572214707ba4d3208c1c2767976c6dcdca950a78a4c433b4ad702ca590cc13c4e57608cc575c529f8cca5b738e619a13cafd160d9cc65b2c0262cf17d82176613752a663cb8cc271ac2f91277f5fea62a74c2c154ddd488238e379cd32bfb72b3922c5db7f120947b14086b50ae262326a1fe5121cd51f2d5c53e8da52fc6dbbc7da47894f66f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642384866Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d, Name: checkpoints/576/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/576/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6b744e6a25949ffe30048c7eda0c79216c7c264cf02b21a0c5fc075bf1abb99eb9b42cf560f44dffd37435111b6f7b362acc10e9f51d8ee5f5a8787cda2349655ec2690962b781e2f946e74b689e9651815438e1accf96c76aae1ad3819d5f48b446320e59d08f372f1cc8159c2c9e6d1733acb1222badd8715c184bd8af2f6e3e3befc811563b26055dda1e197aa3c794afdcdeea65f46422905f21154dd54ea956f7f20c0c386295f00fc19e79126321e45e628255dea221d93b987981feb4fd5f4839d06d3586c34655163c90650009adb2c52b0593c5f9e51b43f282f611e2e7a758b24e2a758f6abf3f6e24a8b2b05f63e9a4cdfcd9f87703927a12f079&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642404074Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/960/default/d, Name: checkpoints/960/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/960/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=28b9f1f0ff4a5502c38715366e83cab8818367e1282520fb35b24e8f1480550711b6183aafb0b15b71f12c5c06b6d9223fa13d2b2cd168ad4db807c3544ef8ee07629dbe2219f17c9e98c60233d783333eedb00ce432c2a2a93799714757b88a7b84400a0db4a40a35e47e3f871f97033ad8d7c557d3cd5c03bda6fd080f4fee7dcd9daa33e1474fddf813f5c350035469d4cfea1bb988b0616c9b8316b0e1aec099dc53c2e74c004431d0ab0068d17b600645b69fcfb68508be7033252180b54f23d3ca4450cd4c08c5fabe4b7bcfad28bd5597d741fca9ba77f08d020f0c0ee7df24c2ded21666b91a2c950d46e49b201bb83969154fa64bd501bb2365792f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.64238207Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0, Name: checkpoints/4352/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4352/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=52931eec2019d4d71c4536820d38b05aaf3da66f02b45e6aecd6d4ac9440947b34fe9af27717219645544227bed089e5d1ebb5104956489d444ffa2b18169e28cad5b86e4d465262254bb3bd15740b559d1cd54f2294a2336e1b5b42a582de92c1221ff2ce0e1330957dec7159037a05c4b0642d766f332868d91ffa6862c3ee83a517f6a34ed61d51ea5b6a33c91f6806e29d3d8ed1a606f09b8b9c3d0557385b01b6178010c231df9ba249043c596aa9e5d604ea61b3fa5f102f0ee4ee875fc5925f050e07e9cb692abb143a74ea9f89fc7d3a902dd12c8595b09801d9e5bf48d578a72f7a461e0065c7ebfdfe9dc5e8533a2c0a0c05da08c673585d4a9fd7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642417187Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0, Name: checkpoints/1984/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1984/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0da594c0711ccde30db0519378f081e4451037fc3fbef67a7ee5a7a9992918989f052db7fdaad7e9d34efa46b16b5635bf87e35cfea44de8d7d27e4c0568ba4cf7b2c14bdf7f326111171f177c36ad2c64b089607ac52a77a9cc45d21674a52ca752f87299439d7ecf02040e9b7303e99d78d436e76380bc43f31053a7f744fdc3b00b1143288ab4d2aa2984d283e1c6439471b222b991c2cbc48b372dbebd01880d6a8be9148288d4ac51ff42bd17c9bcf6d817b8f407a31ca745dc96b9c4b90eff08d1b8fec79dbb4993f1f7226519ec3bf463a1e09dc2a2e60979a4c5c5e09d397bfd3c109fb02ba13e72a7a9a625a9e681add74f2b89626d9a101722c9a7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642407931Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0, Name: checkpoints/1280/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1280/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9caafb6e82d78de17711c9dbc785d1c7da304408da3b677c19d78eed2f3a1acd89ea1dd972b166b21f2963d7c87fc73fd4be072f35bf2f497e9a1b96af91c5562774f857567bafd0c49cfcc4549c884bf2057326e6e5204af7c7a6fadf694c16caff0344ef7305c87d230ebde755e3479fc2a5f1048f0b688fe959fcecd8d4ca6004fb36e569e7a945bc98d65f45ee8d9c680ee207d2f254bab8fa0bebd729d12b1cebb76b252412774e2729ccdac29525cffb6050925dd1ea583cde852a1f346f3aecb7caa6d9d80e5b9482daae070f313aebefc26805badfce516f1807ae3467eb4a59f1b256693354bbbe147ae4cf7c878655ab0231f733fb10ac86d8bbbb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642406825Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d, Name: checkpoints/2496/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2496/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=63913c5d0f60666bdc6de0397169e4757865efa3ffa1dbb0351a195a12d61fff367ee9e44f27140f2cd3d2a2c3dcf6df6bbb5cb9eab68a305098aa10b3d20957ca9dd5d76bdb6d872c9a94e0c7952215c33a75c8d6ba383ebf5c6e8ee8f4b2ca111febfe7aee5517b6ca0d32fc0cdd98500f30576725e902cb832ada33c64a48dd5b2bf8bc08d4bb537d535bdb2c70f507c2a498e59897cb1f53a9fb263c5e02a68586eb040516f608ae0933168d59131da7ae7155a71d643544bdd171b77914eb4aa0b1d9a66ca7f19f2cd593c6956def973daa927a54f9b2302523982b67b2061346ce181c2f6f1caa42a501976fc08b6e629a02cc370c36d5a45d643b7e80&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642446881Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0, Name: checkpoints/3904/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3904/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=263eafce237d002f9f87e415b1ab47030cde89766a2d211c1e359b7d4e3ea273e4938bd457bfe2a0ea571f2b80bbe09ec924b2452cd01c62b9788076abab197bf09e110a7079b4db7fdc075df6124229e20c01b6edca4cda4a656b4ad35a75b5b6389314f4ab49edc134a4d70df6d53fab9d90601ce5b25e434f1ef02383a83e5e9acaa39c962e9b0fe1fae031405172abd2ea185afd6ced9ee21bc2c99a23ee262a3a23660e33b97171781955655ef22a9f634ac5339393eceac1e6f820b933dd00dc47cf5022384a73167772ba0e41f896e5ec73b55dd138f99394685befa9ad76759dccb82f7de57ad04e1aa0b82f009606a5459f188f450c65162c55fd81&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642461164Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3712/default/d, Name: checkpoints/3712/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3712/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=296d562190141fda5435c585a8998f2e2eb553900c05a700457ecd05d4b95b92794ccb3f6acb1a443664288e34da723ec8815c4bb1590397f2f71dc34c7bce0a46332d2cc12137838fe29ad8764723712380a2c029a2fe0b89f5841d464ef5726f39f46fbdd98878d0b85371ec9679fe1e218648e6008bc9c2501fa7fd2de207033a28be11548c51dc0066fc1869632ed86deb29fdcf654624836d6ea0a9422afbc02bf3d2feff4d285815a8b136e279dff0e3cccbb30f203c0cffee90f3255911b6640f46f2ec5807506d2c8296c4b28311533411eee54cbcc3ee930593e3f9c6cfdcb291d3668da6250fa38d0756a714b5ca946042b2255fc33a90719347c0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642472879Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3648, Name: checkpoints/3648, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3648?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=312a6237a07440eca89d4d2b04a4860b423c7c1fef8403db3312014ce71d067104937266931a807b18c7e1a61e10b0f6cd551998e6dcbec807f5e4205fb3a1aab63fd45b3bf94c86195794f6414db97738e582a1870b5de20c0fee24221859e1ac029ad8127d143f79c07683f23685b2771b8aa50c1fde4fcf69960889969bc8f5f69885f1ed23fa497b1a47f7b73dac34abba0b1b85fa8eb73f04fa598e84cce1592c88c6b66fc1543141288937ebcd3c3c0758614905f4a60bd603f38748d9cc759b08d410d976477b18c33d6c1e5ea1140cdba07fc28cc85555bdbdebb11c45445d3cc55f3ea96ed29bc33936baf91cf83dc0396f19ef4c74ca7d29c65384&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642529094Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d, Name: checkpoints/3776/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3776/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=23a7c5cf82c03546cde1e550365aea020b6229fcc82f2a72f097a648583089babe6268c74bbc5fdd4ec143a5600a56621ae34c60a133ee830aa09646343f6420da586f889284ba20bc51b18c21e3d076326b5122a046ed01273222b6dc80dca13e9289242d8479c8feef899c330e18f305e3f4ca1069bb6140ea9dae7ad71c031dcc09f9dbee23c0cc786cdae16bdce94dd0aa33d592d54ff2164e4bef06ffc9000a8dbaf2049d12e48d93c9d19bbd038f8d28dd37d4616d3997f9daeccf6dd01b8d092056fe9050f8f8a728738b7956e195a4bc4b53e38db3280310e2cd247fcd6f18a90d4b34f0dac9cc0965d279b1f7c302343476daa595a0302e7b10c256&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642549422Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0, Name: checkpoints/5632/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5632/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=14ed8cfbf72a4775a292acc28ecbfa6e08435a34508637fc4a88ba35dcc1269e5cff97c90aeacb21a80d72479643af07d2f82a4168e9dbba5d916434a91687477655eccc0e7a712a04c84b60d3af54c81b9e60cb6212555a55e76a91b6a53ef243aa4618c0e9f17fcbcddf6b614cc9de39c84858e7378e42d3d62f7560f94df6aa46422756c56b7170196a1ba21452d558d0cb6c73b6a8802b5f7a9901c826a665510c29e46faa193fa1fd860ae96999e86cb1e24e984e1bf2944a761720c7c55003ce8d354c9c8337a87d28eed63079f8f89e4fa44bdd8d888233cf5d1b44e954ddbcefd88fb460ec60415346723751aae591b21ad4c5b397664c88213b4c50&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642570132Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d, Name: checkpoints/5568/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5568/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=8d85e473b06d6c55061e2958f086bd2b448a3d3ed2fd2ac8a192b0ec4f9878a1c8cdff0bde34918024393351b07375a1d7cfedbc1bf887773d7534fa54d88a9859ce27a8a1be41c394dc51fe17bb5fbcf990d32ac48608e87ccf961ea4d50c46bfddab4e38f43db6dd72e981585f7bb4c969ebd4a283d0f4a79b2846e827ca990748d376d0b6281380c91229b1ce3e809ffa60ae3c3d649102332ecfd6e67adec42fa28e77fb132d859f3ebca128a7919f6806360981248ef85febbe3aea826bb3175f5379f7cedaf1f00bf350bc6c5ff3655fd26993fc55aadbf3c64d4c426f69c8a53d7f037e4675dbc0ff304c6787ad22a84297d2ee1325fb4b002ac9b436&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642619889Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5632/default/array_metadatas, Name: checkpoints/5632/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5632/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=071545c07f7d82e79b5e5f9edbdfd54dad4fa7c10e767192b337e3da4a4a1a34523b5bbaf6668e01e237079c9f5b2528ba1ae65164c4e7b339e113e759dd3687af4cbb9d4d754ed60d3cb29837e84cc7ebd9b155788b38252b79c1aa47aa500668db68b48e04d53912cdd5ec82eab8e29184a835550729dbcd2a18a05b357e6265d981f931728f5ab5842d661be8fd7d031bdbb102dbb825ad474434ff1bd679d079648c5970b8b7916505089efac5b76d6854192c4a4aecc7389b25a9948c57116913bbf1cd49ae8533293caed69da37007573ee22b4b05905523131754b92a66a890723cb5587d81a6b912ede5f04e9e8c7e8602a95fdd464fd8aa5972a352&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642622074Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0, Name: checkpoints/2688/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2688/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4e48961c3d8a6ff984e29922a8625789647d3d177db3a7070c3d0fe827450fe1fbb45c1a20c689ded8f5a74e733884387d4475f97f4adc048a45ebb2799d4fa573678f81957b15c43c4cf895727f54eb3af27bc46935d82a527a1c566ccc5a75ddec5574d4a837d0df1399c841e6163ee5bae7c23a1c7f3da7be6628344f9c53e237a1c4189ee86517064696ac438b923b6ec244212b5ce61edc14f9e33dff5c0dd0c3189e3661674c0c9900e567c8780b4070f5990d5d3247ce7850dd3e9e8daca0806c048d62a5f889aefd838ec2457e335d808153ae1f7f31e3447cdafaca3204e634ae599ecc96fee32ff98546d0a96eac659500c24002ad43f3dabe68e1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642631131Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2688/default/array_metadatas, Name: checkpoints/2688/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2688/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=82a93816d1edf52ed4c4f9fc6822429243e34286d17c0b661a86241452f64c4e8db2da4ca2dda6a480da791e1c91e4914ac8c8c6c229107ff24ee5c6ae2d396fd9467a0029d514db7157552bb950e7282dbed02f330a8209afcdbcb38fe87edc96c357a8b58045d039fdd22638efcef7f988db47ee2ed6b64b2a06fe96c31c2d8955fab94695fd8fd2fba236d54a16d815cf59cdb5bcc9ad7cfbc1fac25e2e98f043bbf9170296261b0a1a46748896213ceafff2c6097062b9a23175b2e3e13ed1f7d987811f0964eefc2e29463bb127962b86bd6a7e0f14cafbf40bcf6f73dcafbe0fdfe145639c9c157768c020a5c21344584ce4fc2f7afd2e97ddd22bdcb3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642634382Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0, Name: checkpoints/4608/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4608/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=8f13cb386f3d3cce4b3c796513fbf7c5f47b4e774f04e82e53406115c2b7933041cd2d62f6b4c833309a8fd077305d4969fd1025816b08dab8e225b9750c6aa6497052400a4f16bac14ed4e7dcce2337342bcbea0ec9fb06145886820e398e410ed78c25a0eb0ca73d84ff36d12eb8d355bc632a1a387a3aa0cd60fd67cdee7794b366343f8c724256b714a4706b839615298de94f3cd4cfdcb7fd1b7420e55f26c12896504582411e060377374ebccbab3019ee4ae6e411b1e104a02fac31f0a924ea57bb5a8744f35c7ed4d23ac344fd851929731e898208fc0a38bbbd446e58d0df70c2ceae0cb651af68f10279fcac773075407500e0df94093f12cb7a55&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642651001Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/d, Name: checkpoints/3840/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3840/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=3cdb934228e73145d45a2fb627a4311439d6d39a647f2e09755efb4cba7c702e599b6a4d2411308093c324a00f4eef656f26ecb94975323dc46a95cb96b7ba9fac1d6735d83e9e6f5d21285a4619379015a635cc7ec6c251701d1cb5167539810c970331728f8b3466aa00bc8cdf7bd2f37773482474bdc03bdfdd93d24642102a34c6f07092e1edbf2cbc69488f6903c38c7662b722f3832bf412f9844583b13ca64894b216d88550558159a70670be065c277f36d791c4633b4310ae768cbf689a85b94cc562ce58db093a83972476d3638be2d0cfd6c929dd17ca47a1d0f057784255da0d895a4ad2aedd24693821064b0f62fa03c0491e194eca3d1ec4a7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642701179Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/128, Name: checkpoints/128, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/128?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6dbd4bb6aa31c956fb4310e96063fd8051efd06ced1321b8554d4887e92d8c013689b76e6f2551373a8be0218df3fdc313e861cb07ea6456409192a3dbfdb7996c1f1e8793c55934a9e7e93c26d4d34cace8f1de898f679d75cb63366e7c15dd9c2ed98ecbfff055d7eca6be9f04ac4c507a3438a0e50def43d4134a77cd68c9d832cafd0140b2bb5ae28abd0eee848663371acc5c950b605e1b572540b74d13ee7f564325cf0d44c480a6643a7fe90736bb84939f7df1be52b5d328b16b131509dee4022a8950b54cdd82b9d1199bbdb0a7df4f50faaf03ba90db3e924b54e902d4719fcd94b0b7417558964ade12ce232be65ac9466b94ce5534de14fc187a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642703626Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728/default/d, Name: checkpoints/1728/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1728/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=8964fb318de194a267767b3cbaf50d0101d84b03875a94ed21bd24ef3d6081a8668985a330938294be0862ddd68f20583384ed984a7120c691c1da40b573eab01e4ee381ec32b34bd743fedc0f1441ca58ba5978ea60b795b7d03da5eb45458d5510bb30b04ddc7748804b09b9d4c6fd561d7168ea77a2957a4eb38a85c8ec2ab3bf54df1a9e92a0b48f6f4d559e305e18948fc9697c3f389ccffdf79df14f00bbfec881574eb7668e29c83d002e2243e7e95f2d4ff50c106cd69f9e0ecb861b4f8ed338fece8f1ce177097b61b1a2fbfa87a2c4af6e313ab1150fb85fb89c2959278774b330b4a8c2bc196fe841702d6962d69d2f24a89571f1d9b727af8fe0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642715397Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1280/default/d, Name: checkpoints/1280/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1280/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9c6faa3ea8b3cc297e2c68a1e889e980e4d2214876277503534b34ff5f5bed7fcb29be1caba04780cc586a5152514808771cd86fbf54124629cd7a8e3c8c593dbd526d6db7e24ae6bc30c1935d025707405a6abb6966cc4f1c32ab71c351aa1078a6f7c4efb9742781f2b7757e8e74931b1da1832d27c70f16fad5c604176d16af37bcbf518c04686a287f12153dd0ce4c893944e7e22c72de2a9b51cf1db6b7e04b6e213687c9e772aaf0ee96f6610370206de05edc3309f86201e0af804fd26cd91ba455af25db22782280b736145d02de38afb40dd50989881354d372b888b31b0ee689fa5bd21b4ac8d67ba71f442ccb2314078b19dfcc280182cf6a4c41&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642731794Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default, Name: checkpoints/1856/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1856/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=74470dacadaeb281698c95a253758ef18628d5760945d32fc2590a6af22ed8c9293e07a72f85c3a00ed0e0ecaeb6f9c525c2806783734f4c55f3820d522de859bfacd25e1ca0f847cbdf81bf265eb3a1d0da5faa039e9f2d3a35bf4bab16b8f0b9df142b5a67d37bc3d82e117cdbd511c1a600e454059cc0fd746e94c2442a26f38abb65d8782fc22896f032afb556d7a7c56db8174e29b815a7fccfce2838076f72c87371f62edade8d48eb3a1f75dc9d5b0786b3bbe38b2189b64b56cf294bfeeaa2d0f1688daa150f6f3f3ddd78bdf623065107a118609740b42317a21d7aa06854b4fbb2d441d23a0d9ca21d3afa37862723633e590690c5be622cdf1266&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642752876Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0, Name: checkpoints/3840/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3840/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=aba4a804da5a9a6ff68c38590169d9509be1fcf0eaa4eb1aaabe191e4a4037df39756d160a5bf798498d5f86e3f645fda6128bb50ab4da2bc8f2baffc7916644ecd7557c50f3929af688706b85e256770b87ed08468aa11a21924cbecf88a60aa3216e8e1dd94eb4dcb1605c9207a82f9aac58a7ebbf743227ae1d7f509f3223428507f7639a51ec81a4059e9aa8d3c2deb0b7a21adc28fb7c6dd9b17aff265a852d90ed171cd441a800d873575e3750b88c7bdc33a449c5a357903bf8b013b816b631b9a37a90cad20c3bbb317ba071bd793c386ee1648710662dc2fc056682216e5893cc9d1237f85b2a674383b21507cef453fe22cd0401a8a3f0f5384305&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642754298Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6080/default/d, Name: checkpoints/6080/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/6080/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=bcce3880b62491cba1512fceb3d4b4928dd477874eeb603631e53cdab2400aaf4a49cffe363973fc8a4138c8f804187b1c024ff0cd811c898f795b5db3cfa19c42c80188ca850e24e775a901788313c958fc744deb89566dbee8fe1a8062f041d8e4315b7819d1baef7e3e5550994adf0dd1a070e5f331142b923ce564e53e961f17b4e679ce55be211e9beb96eabd27b2a4ad216f71b7dedbc85270b3099b29b752cefe2c44dec7b89c05a8acb9050172dc75494661241626f3faf2ab2cdc76e45595b315ad3e1f16c9fcee7a9355b2591326ee2bfdd550c30c6df3a281fa295276a0c6a89f27f9c2c6932c90e516fba2605d22aa031bfb366ad1caf7867408&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642787822Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0, Name: checkpoints/2368/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2368/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=1e084f5d8e01ce541f484d90164c7e53d6688cfd056d241251371cfbaf8a20b1dfc10d4bb9acb79d92e4c215dbcd8467715012a297880bb7c9eb5be286a5603def331bf0533ea74f050f1ea5b7c03d4505b57f49a1f2afe786fba283c8f4c97a0f463ed83935a7d96dba799192b4912684f271e230a70c528604a5f45e353726efdf5aa969c3adff674fa533771b19d48a4a0b7162aae84904c4f26b05e5a7c4d2741a88511ca6dd82f223e8d9e411ea2ed7b07dfcf64614468af0dd8e1370bf062904a625c9430ddfb6899593b4eba309213b5266acf6bd1a5c5831a048def8b493e516b04dbb5b7a12e3c20f610a7123c2de8bef0b25674bf40d7658d81ef2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642800658Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1792/default/d, Name: checkpoints/1792/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1792/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9c0113e4636721bbf357bacd5e550eaab970b6efb1b376c5c13e8cfa2dfa0de9cbb6101fa10107a1c59560d7570581cb4d2ff10b2ed4d845f00cc7b880b47cdd0e2810ee0cb7da6cd65a8153d8f621516097aa2e0ccea8fe5520d1ad34851ba23a3891162effb5bf9510e22206d0bc1d3b9c7b1db63b7514e2573f6aff25959efb56a373c18378327feb2e52dcdc02636a954406d0259aa4c4864479f0ae48d0f466612916de6f4aa42e2604e130b4daccfb39dbe18f18c1ba2ffd49acea37d30c543765f3fa34bfc3c02a2971a693187b4274b83f2fa93ce8ff273a7781963b5616d135083ebc6fdd28110128393b3658c51fab5d488dcb66cee75d10da2c07&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642829665Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d, Name: checkpoints/6016/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/6016/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0627ca5277ce262d886f9d1d811fab0c4f914871bd188676878d968573df7c047249630732bcf9b8adfc03425e6cadd252253ffd760c3c9f9cf22e71b293a651580a3112883f15209c86a3bbae5bbbd68cf421c27df02b24f67a006f93f61fe827301ff005964b48c518c088afa6c6199145ba05af1dcb399b33c7554da3d8137deaea211910ff9b6c017df1eae9caedc3024ed4ef9632e7231abe404ded0c76ee591624493a1db6b8ccbddc5e1bbd61dfc7fc6f58dd12cc1912ef1e5f2e79225e48972383aa8f7c2992f65349568f0d7467a00356c4251f523c8701783e748e9906cfa44c44277be8aa88723266a09cefc46a30f1f3a7b8be1ebde23536b75e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642875764Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/576/default/d, Name: checkpoints/576/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/576/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=bbc5ea22824b432e5e0c40ff01fc83bae5158a63809667527c9625e45a4f98229a2bb07f82df595f3985921566b67b9781904b01416b306d1182965b43212443076f7f0c7588fffce9a39ec13099c347dfa0b54a3659ecf36b930034bd4447e52877fe81a169df77beadacfda765d3876bd097a1200d6faa3f660f2286dcb2b3c7f71e3f1c66a47edb1c2cae9b2d3bee0907d13a4997a239da5b1b13ac3a182721b6d0d4d13a773262e39d42116be596b83974a5d145f8811dbf9e64fc756ad52987e22f14dbe8c15fcdf0497fdc9f79ca6374d31a1d71ed059cb06df75225911679e9750f20868d690d1785b51373ab239b25cef1ee07f3331b4f61e6ce19e6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642925142Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4352/default, Name: checkpoints/4352/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4352/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=21e9f85f775336e4356c4a2f2a2222e0ef0b665a523a6abc7f40faa94f617da8415ae5eb77209be529c899aee8b064229505a3fa789dc693b0df896d35b091cedab924e8535fe82b0274aadb4c014382c26ebaa448019409a7a4e6d042790a576c44f94a7ab97067a812ac7e481fe245a402b4d5ec1a6c6131a7bdddf4f619d61143417e39e78bd9b9a9b51626e2095eb6eacbb99baddc722d45be58ba3143645df660948c51dd818bf22f76832a81a4b63baf2617dc29179ecf2f3b055f22e6ef22d878408d732e599f5e14f36d5785aa98c9b32f21e03a06efb8c0921c997ca3bdcb24e0011b709b66f247e46ab7014b2d1b35fa80b4ac3fefc38113397f3c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642934368Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/512/default/array_metadatas, Name: checkpoints/512/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/512/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7b3f70306b9e4de0bc4c70c7a4fa26838d590e288301a6a6aadd78c87d6b24626c34473f09ce16eab32dace75fdd822a8d9cc26a32e36389fb51eae872bfceceb28df22b6101159172f987b8b96d173473bb269d2654626bcabe3f45249eb23d5ea878e336a5b24d33a5ad525364309a831cdeacc9194beb1bf39ff86fd4962775acf76b4990b40affea44f58ba9c85b013fbf275f50713160723b05377f9e3a52f33e4efc2ec4eebaebf68c88a935dedb9723b93d236d8ae01870d81cd7cfc9da4cfb50f5cd2284e5af20faff746c6b4b2914e29f0c34f95ca5cabd478132b7fee0ed109b6fef95117806053dabe4dc393c13b90023449140a64d863c0e59de&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642928226Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2176, Name: checkpoints/2176, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2176?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7186448d6dc9907687d83d76a9efc3f2d63de8889276aa660751e1a283cb57c206191fd4359cdfa15f589036ea33d26d5f26878e8d342bdd0ee1074318dad835c353e8ebb69265cab61c28c2ab79af5fcad30c887eeaddfd35f5ef09a7e5db3cbb89561daa2adc39f1a90cd4a540744121fd6e161d13ea72a02b22960487f1bc52c1acfa17f9664dfe5e27dc1e9b59899aa4bff7b1f7a6fab1aa70412bd63946233b5f91454d9d8cabeb9d2a02db86236b541f037d35a41d4fad59403052c5a51d81ec45994597b8fe65fdf3b65690615db2f4cd1347dfbbc5a56d9b55ca788c687ade523cda8a9397dc55508317690ea5bec9217d1d4f41418758d59c082638&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642927598Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4800, Name: checkpoints/4800, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4800?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=c00aa5bf626236f7d9772850c0735f44f42ef90d6c464db7d94778a90ff6ee26406acf6cb934f6153a4a919ed3ddef269541399bcb0dffebdf5f74454b9c0da32bf46ee2a2165a915c1de34a74429e9850e504e58152c3f0accf5f346a95b2e50d01bb7d1162286c213ebd65c978e5d8417e48908e575030ba308b754a09a8a094df300961bf945b04d591e4730caf1d4208dccad0830c716008d28d83c763fbc9c3d5d442a909b84ed8f0b6789933b4319ed5fabec661513d830edd75d8417a53738b796df479d973e34efadb385921eb537f44e49bcfff0d625835bfc9c8d3a37825b497e79da6a5dd3847b9a6ceff29489cd5a3d06367a78688be18fb6896&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642948146Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d, Name: checkpoints/4160/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4160/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=aaf9b513bff426fda6e65672f543c643b9f9b1b196445fee5694d85df6fd8cc287a57564ba031e982e3bcc1bfe79458eb144f1395e0653deaaeb7d638287cc62a4fad2e2c380e5ad2bac52939364ada4754fbff6b7bb69b141455438dea144d892bc42fe0eb79eaf46f165069b6d5f58ff316eaf7fd435621222e9dc355c30175c0549e3387c1864b08c304e0a0428454901d3906ee9834143c2ea048403394b87c2ccaa4fd1c4e03bfa245c1663703bb5706d4614d94403281d659d3fdfaf39211b02823ed0586958e8437763e096317b294205e7d53a83a5f42551203508ae3559e4734b5835c5146289baedf4688e435d55da42d407b06967611e430e449b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642964116Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/d, Name: checkpoints/5376/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5376/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9a6ce58f8eb3f893e85f32094dd89317f1116ec7b35602ec6fc335f0347167d3904c19212872ab7304de5c40775a04c18c8f5762813ada91c7cc25c9f6976a42621fbc1d95b41d6ccdf5f212649046f2d4752780d496fc22fda0a2273533f1e4000339c59724bc49aace04a725f96a8ee4474f1e9fd9257c544299059567dd973512c418361f5706e8e67bb37bfa12a0a43f90637b79f33d8e2775bcca77a563f16a50e906b425bd1b2581f651b881a4f21e6900e870e88849c4a8551ac448bb7e4c2bbd80f40f84494c7378b9afc9e345f9be3230313bccc912c98dbb7b1b618e5fde38d990b0e7e793fc376495a097a31f989e4b562814631fb87f7a02d6ad&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642987282Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0, Name: checkpoints/3200/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3200/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=c1b8cd59da2be566df21d5192aff0927bb3e5e6dace00d92bbea25e36ff9b88c02512ea375408252897b169da574fcb792fb889062312e988320c6db29f9155f8d6ec9a0b9e1d17d75b3ed86b6dce978ac14688a30e00d254b7696e38a59f0356894d1caf9dd5996edb8ce0bfc856a47cba1decc45d73749e8c8c445763138c25fac241a41b029d85cdb0b662d85ef5185046cfecc95cebd6c0b17c32db719b4aea005e6912372f3a0b0ec2d9b7bd0696e0a74f93761dd081926203056707a773d8518eff624180a30a95269099ae68490416dc07898810fa61b4e0b994977e3c366e5a87902d351b5b2e5a1c4abd58a21b29f5bca187754456a64c22e45b759&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.642997952Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d, Name: checkpoints/1664/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1664/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=c3b1947dcdeffe3f30f53a2edbd84eac58af34b56948bf387a50279e1299bf426b6fe8fffb7e4b7ce9e2fe8726df22b0416ecc767f0a1d9e2f130786988dd8cfa870621a9e12742c23d6188cdd69931ec89fbf2e18a42324546d415188a494c7fa2b40310b9f44527e01e4724af463e2091f2d63cf0efa0319f1912fcaad008f70041756d3aedd8289cd81169b3dd9131b8d35c02d1714b66d451ef38244140e49899a652cd1c43981eb6d207215467e60d74d0c07394f431c49e92040f0bff5d96b8fb8678bb355f2211cbb718c57eedc3e239e61800e273850524477cad9a7353642cdf1e532b0b7185f70b86d8e20c44c47c41055379c181125a463c0764f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643001217Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2560/default/array_metadatas, Name: checkpoints/2560/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2560/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=11944a4b35ddb37103abe4a1bf96d9a048dbc9841cc62f575f500f0a9bdde5f866b6de3cacd6ac0e31bcb2f003aad26189e85ab98fe29e024021368069bfddf7e9f47d087e3752c81fc11a654c4bd3ac1642e9622975a155813dfe2420879642b9e638563b125bb49bc0a66f7e968939b2cdc99e74f908fab280f81f361af579b76e020899f327318926f665b27534a143edf0d74b5e6eb27579812b8c1000c3061991e7605433d793b69dbfc887944c550cba077a316429d64982cc49b9667a09b6758d52a021030d97e5727df6d2c4b94e1c3c06eaf33c320c30789c721666bfee325dee026709cbe482001886160dae88c868308335d50369305ef839b591&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643036066Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0, Name: checkpoints/5376/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5376/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=af06511ab92c42bc13a08cd8aba494da150e4b769d61603a0722c2e5c26382892f0b1c20e64b9d8a54c39d7f69d77ecde245fb9daf02011d615131a94167528af6e8447afbc1b27ec13aed8e46660db190a92dd3724f73c9ab574ab37a332b1510cec8a166255e026b5d19f828687b4fca64dbd55d3d827d4da12bfdb6a7d722c1e325f06198ad77871e96410532c952c310cffc70994beeca46462f14a0f9eb5d91fd3d7be3ba3c81463749c89ac127b07fff37f23366972e039f2072ccf1466aa2f330b2050960214bb146ccd13b8d298befea999495bbd4ba66bc8a0ab5e98d15b3702e8ad43d33625c3c8b72178677e0b85a3b3b53e3de0f870c2723c8b7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643040191Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/d, Name: checkpoints/5760/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5760/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9994576d8bc5c3a077c9dc05ca943a5f8b706e139a4e218fd2d6df9e73d0d32d0559cf50736d0a883c24ecfa0fe8d4e852d9c7a2ae13f4f8d3d2c7894a4de71bcdd1c5be2ced4378303fb6f889e0a08025220d1536336a81f83490a3926ae3c7a85235eed74d0d91b33842fe845770900ceb01e69fb2219637f6d2579b4704b124b50729e0214dec5bf356c15250e6cfff475e9ff0af1529307af9359147c8d14bfb13279a00f673fa364562f3751158c26fffd4d485c8bf507ffb79d0dc3618397ac2cd3032db076dd8f45c52f08b661d634cb643006c1934c426d2876e1ea41a8a3411acc9baecd28fd4040aa68c9e6bf844cd15465a41ea27fd7b63e65d40&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643030049Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0, Name: checkpoints/3136/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3136/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a44f0dc417f370c01c337edfbfa25a822c6bf78fcaef8cd13145de34027b4c23d5117e7d5098f13c0e1aaa1c9b69d82cb60cb88de2d84b0d55abde82edd68fa53d67b78cccb70c8088eaeb0bbba33388059b9cb085214d99577d08e8aaced5170d0ab4d5b8be7e8f3c665213a6db0fdb1048c62468e8e28b0a90594103b412501a9aa08d47cfc08531b641515dcab195330a7022507a8fbe7277e5d00f93349f0a6d9f2afb6b42c275159cee73abb2ebcfd0efe682a4ab5a07ce185214b2e72dc6e8f23b5aed0bbe3d6d853518296d94e15b12bf9a680464afb0d65626c44bbd2bf7f85e868ff853e54993e772ff24b240279678bb173056df90ab2fce6498fc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643052843Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0, Name: checkpoints/5248/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5248/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=324b9e04f38cfe04b2ebd75f557b4655ef31faf0db7170bddc5e163b9c4ad3098b7a946486911d5e9bb1b3db257c5efb7099e21f3283d0230a2bd91071236d9fdfa2cdafc0b7f89c357e6d40599d652843d4d3889851bd223aedf074a9ee726215eb3db1af6e3f5fc2fa1c990ada0c8453a529784a41a94e3e940018fdab0730edfad8c28b813a39c7fad8acc9a981ada7bae38b0e5ab3d23603feabb3d18a92ae5f5e5cc79c53551efcc532e729a6e594b0f58b07b4f9e0d2e371ebab0a673cf08228cf622797fe80d0d7ac1c56639393924377a948dca4a4a2b751d7e3be088c6ca6ad852d672af7e3b9d579cada4650539742763bf6c7e337e0311df8a229&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643065057Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/d, Name: checkpoints/704/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/704/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=40c8af823ef49ee307e10d5df38c830fc9b493e004a91450426d84b563f7a07723d4ac620041e9454de64dbb61d997938aca7764fbdc9501b4676bf7e315fda0faac237130f7c2f4ca01ddb213b539aadff887219934669c79cc5f0d70efe554dc2265674ec00bb276e29cc90e9b6e6f2efd7df87c87c3098f456a11a294d40dbe19297d65f1a120d2e6ccb3504290d1e057240c8dbd87590d13f83a1b9b19b0e81a0dba9887748b7a4867b1f20d8c56fbcc13ac2b888777e981977c7e2d871622dfd3be53f75ba68ffa1ceb6dcfd4067b854b7037bc291774cb075b6558c0c52b19fa7ca37b626dc59f66ce213eb0e3c3c01ba6348d5e6bb6fca326d1b409d1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643075207Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2048/default/d, Name: checkpoints/2048/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2048/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=c0cf7ebe65dc51849ab8a872f0ad91a85e4149bdb57a1da008ac521d422ffdd6d8d9d42d12e2a0799bd1fc599751e21189dcbbc1e55ef9d87dce0726feefd8a978b249e5b9c9f0d187c62d09b7860008577839cd4e22e81152a61076308fe1d83e5c5ce8cce3d11153f55fbf5cd09f3557b4222732d07323a355418fbbeccf6c3e6d4e170ff5754c966fefc9d7707686b04dca8b8743c85b310f916996d4e72f1bef62c84acd82283c1840fe2ec9f3cc5563ebb39041628b40f40a829b74c44b110734fee7175fd9a3426eb167d109b18045a8912e50ec7094852b8ce9d660555ee5a717642976ec2d0561cce469c7959b57cef083a09eac2e0432f222a812ec&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643104149Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/d, Name: checkpoints/2304/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2304/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=49caab088cd05f9964e590d034248ca631494a68525bf50f505a839f7a8fe92144faa4370c7e501a4487e8c27776b789f1c1c1a0a2c975d2bfdb6319a2718f02a573499b480b5ef329d7a0378a6428319666c45b16f6479c119ceeb9e36bcf1b5a386edcbbdaefb1725dfe2437934cb1ea1f487124f9d844ae3dc9a55ea4df571089b6552d70006cb59baad67295953f1d1f8a4856440d0d0268a46c404115c832c9e2149e97b7bc63dd47bbb99e59f039305dee4022bd5296438d621506be5e1093f706c526eb9e623331762d1c7ce7287065ebf4c3aca56e99bd11610d73478540741cca38db37baf40f3989a16808a6995fd3608443b3620d48ee0b81427e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.6431421Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4672/default/d, Name: checkpoints/4672/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4672/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b1893f796d051d1a6ac5910488ae422bbe286625f7832cba5b95b51b2efdf578b6d6aae8a7bc2ed5d997c74ae72f97a9c7fcb44186132e19d661ef48f2ccd2a2b345b4cf06232f6c017b0f6641b6c9544b86760e89d3920a8ff18ece55c2ba00a3b1a93fa1b32b7b6034e8d85155c4260d36ecb970879411977b47262cd1483680579e1db50669a6a2f285ee518f8715e07feb49c218e0b256c898133b5fbcd7eee9a7aff8c2e3d0ea88c036bc1d54ddbce470f23ad075f76242e59f73bdb81bdd40bdbf01c6b9b3aa0f8d545cee899e4d108df4ef2295cd8e10c83fbdbafc45a5dd96c5feb6275d27057ff9f4e230e5d848a9120ba37f743212ff759a15d020&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643163844Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0, Name: checkpoints/704/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/704/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b81cecb83b3f5e2cf6d5d137086b833faecb44fad2cb8751346666a0b49624139f579e14d23a8dc142a0fbacd39cde4758f21c59a7beb9f9236d8ce07980e1a79324abb6fcc2d895d7375c78df822fa65be47c280da7aa8d821b2f4d18a9bdf070d5e667abea4fdbd4e62eeb59d8a2b4e43c466ead34021d3ffcdc2a4d29c7e78efe40d403694d5478581580afdb0eda4ec3aafbd39d3fe404ec8b5f1246292c78c60a1704308461cb585febb13b004eb2e7e4d846715cf24ead5b79339af609b24d56e60daf1d9a6dd70d4e1b657e54e1bfae4810d7047847d4d190daaa35f63b84f1d0b09785a5e568c183b2dcaa3a302db60ecfd79240d8d06bf348ea1d8c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643176458Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0, Name: checkpoints/5760/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5760/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=177cb92bbb276f720eaeca3870c450e1a8345805ac9d5efaa9ac2b15237cb0a6aef499f9f84950e029c5ad18bf0de86c53d2e1fabcaa382eb8ed016bcbf74c7ac77cd69d1bb7cd398c54f2f5b6d1fd57ca7137d6a19b49401642c5344c6747f99f04d13e4d0383a92db2943310da3683ad02d5e9473acd8fdfc115877a4b1976f044261336fe8df6be2ca1d15a5332c76d594ee904418d9c10c67c764215ed554888dbaa7661beed1cdc94aead2047c8c35ff37f2af767afe10af42d21ce3a99e36e4b5098bc3bf087c8c28281430160613633eddbd712b9b3b59f15bac3e5e0d7d150ed8b2a9e3318db0fe00b57e60c19b6070c20c9145dce7b0fe61c18e977&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643192588Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default/d, Name: checkpoints/4032/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4032/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4586a0b99e929272b5717347ae408da4599938a9ed7ed14afd71e8235ff309c280a1750a3223ccab2997bbcdd3b38ff75f5954786cd2ed39735570fec12c2eea62702e9b50762c49284399d87bb73c817dc7960535dbda5853e42896230293272c978c299cbba07e5658721dbf7d7923aaa7c4a631563413a76f8515f3a01dfdcd9c0b28cdb70cfd9ab36ab19290f0c020ea5a867df1dfdeba539ac937b26e94bee951c8440d9e4a4337b68f68b26482197b96580e765791713e441edc5844965ff126657c89521184d2fb1910ccb81c0e30be12cc15c582588e01fe5427f15188dd2708e12e11b1659d22eb5197e7750c6a38d4e258cfc03e00d622e43387f0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643210475Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1088/default/d, Name: checkpoints/1088/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1088/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=395b6c681a70d2a1255f19e0c8278effc003bffc53d2f39b41291cc25e36daebca730e1027e5f4f4adc3dd8540a4b810cb1245df3205e0154c02df3ba6382800508473b60b5e91aa83e971baeaec6ea9f159e6efb14dcde7fffa8b3d5b061b3651729f51ba1cd7f1416204b0d555aee71192dfa77f3f26aed031d9d716a79e453c2cb362e16155b9e658c360ee37fe66a1b1c7e8b1585293a2219850e8b2d6dbb2019343718a386472979d460e3e9f979500c497dcb1290e4388a5f4c829792b3a0804100ad6d7aad18a9e323723bcebae0241ad1328eaa91f600eb3ba7dcfe5b9e047a40e3dfa95018d5dd519fbca41785d260c54873c1871673262a16dcff1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643217064Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d, Name: checkpoints/5504/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5504/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=16865fbdf04ff14031a65769d172a076690ea0573bb6cc4e3df2ccc2a22cb76ab3dba7e8b302728af165a5f9597e1a222696805867e86df9d96730959984719cb4ef3d2b25280686583b0158f933d9adbdedb088efc2cf83215bb3078be711f70667a4f83185257c2555b87f69c26266f1699c8144c0efa4f2987d0860898a23fe68d6827d5bcd224cd6e672bfb420381ec82e2c77dd2759563bba51702ef46cf9f3aebe7d5f81b6d1227e4c60d04754757f9243b9f4de2504d58e6542c440db87920cdf866c5931d6fea22482b7723b2e4f290a5cf41e8944fefb1adaeaf1471b6c18e65776fbffa4f4fc986a12533dfcc1691d67c074dafc5e454a3fb876cc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643240651Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64, Name: checkpoints/64, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/64?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=69d25c3a8b797e1f1f64ae6488117569bd19dd26fe4bd2d8d19e10194ec44d081bf5bd748ab45d9145543e7dc239b1b5317f4da04dbb1203cd6b44870559a108e1f6298a0fa2812f78ae2d9d8417439ffb3c975226e8104f281a8e0132a648820188d61b8397b187a02577e11e243d132f0c7e63bb43c877f3698a40bd51a3a58220c2d557992f5fa9a1a1c7e7d3eb5883a20cdb663f254777c43bc4f7b8b3b86a906a0ef2d591ce74a1a58689d9844750906dea0f3db07003b9e0c97219742dbfecfb192debc50a11a86f4aaa912bfc59b73fbea4fb8751369c22d5032efbe191f41f67bbbb774efe2bcced80d68fef22affab04a356c35f746a79be75e8a32&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643240186Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0, Name: checkpoints/4480/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4480/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=037a835a1afad5d9a8ac05dddc2a7584bbeb894049c9430e27056085a47e4027b2e74e95435844b9d069e4bea02badaa2eb06be84fff9c5ba644d16659825d08a6f4f18be3bc4c3b9fb9207e516af77fa9d6b597965b477f3069e23295bcd2fd241b94d00bde7e48b1786387a66a46f6af947173ee44e28b95f5e906627df1e14ba48c9c2466c89a7bfeaabd9951331c1deb6ca44a8e1d78b4dbf7f2783cb75bfdf1551cb5ae86e6c24cacaf8cf8ee980b1c87cd19ba6ddff771df2da2d4c0e8cf52a3359d41b0f9dc077e854b770a9a5f1925b276947b58ac2fbe323afa3eeebab9a953093ad46808f8334d6c4772a4ad7126e48cbd3ea8092ba497c84c7c36&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643280856Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4736/default/d, Name: checkpoints/4736/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4736/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=76df43029e1ea3a01fb80c5524db1693ab404fd9706ec0cbd030b0f7ad71814fd8015219b7fc7901a78e9e5ac7e8f90cb0fab13a92935af36bc695c07ea631bfeef82bd9c2131e0f4264b1194a2523d43d263f74a4c09ebbea05fd80ec404d3b2dabffb6e7f04971836c78d7a870336cc0c2ff52f2b0bc11cdd436128e2f19b2f91cd15a5631eae94afe4611baf04e1b82490d0cdc93e67b13b8604da195ee69f768269f13b6c82ebc8c692d8b5cd871e87482aacacf10eca6b11dcb52836500c80968f739137021ce5eac6a627ce83dc337fbf9ff43c5965d826914579b349b1ca6fbe2830c39fab1bd5ad00d580caf135a7e6df0cb6f7a69c58941a1cce484&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643297403Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0, Name: checkpoints/5184/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/5184/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=c4dc97ad9329fb7e7f95962ff62e3cb84fc30ae5f4011754416ab6bd57c83c8c8c9a26c4e4030103bd9404a86dfdd693393f5046a0f6b3891cbf06c4220ca8530c18c5b6293eb15fc44b2a80cff3aaa0da24c4820ee74b6e4a2f4907cf0358ab4a917464a80136fcad0adaa6cb4d3684f24d3874c203fde6d10ef43e13738e74cace3ea7094d7e86c64189b0dcf6e9973d18ab9bfe262a54d49563d758aaaea18c21fe472723100964646fbc658302d70027a0094db056d821d39ba0cfea7e4f9b1d61bd59314ee78b5acb0d7d1c66d1d371513b351a2584281ac0246ebdc4592ce7d4667745e7ced0d6183bfbcc02a3ee0d697a7c5bd4f30b1b009764978989&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643308602Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2432/default, Name: checkpoints/2432/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2432/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=2ed995992bcd73e787f8603c6a629b43036370ae1ab31a5574d4467ae0f9086f43334d715b8f2ad290d5e7e85c17079994f51fedf184dd96e0b63d2fceb13c9f1d8654c8f941cbe98bf42e943438c73e4c5b9c2770036a154dff6701896c113128b8df0453f109f8ee5a459356ae0fca980fe3280b543adff6072b8fe2557ea2c4891c0a2f1b01bbadbadd0ea78903e578406917893992d41ff996d657a3c795003b9cb171f9cf6520565ba8a7937cdbded6bd2b93b97e0f0de30f9ca24d5f5c2055de6a529d36bba4f866611e809abad3a2defcadbeacbff316f2a34027a2b565e53328e6a0405980432df03d09438cf80af42fb423f3ebc91c58ab01a3c4f8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643305747Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d, Name: checkpoints/2304/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2304/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=68548a91b47fe1d7df12daf0b19ba04800a5574e08bb7315c70fe12736632ca97bc881cd09b56f7dfe95e597c0d4c579580789f6fc5607bcfad09628ba81a9cb0e27961e7288d46b9df262d24b533706c7872cbb81b03b0d826958cdbb5195b969185f106691462b0d30fd046a5dc280ea5a1edbe6a104637960441c9e6d94a64ecebbf549c742f97e143dfc4754fc7ca7657838c6342966e7369b8cb6a52e0a98c6acd510c69d849bfefb28e0aa4f3c91ea91d346d683e8ebe9b61ee28de50b133127fcb71eb72c04c333edc0af1e0ab19c4867a36d0459c86dd4d8ebf65b4bb16cf49f7129604f0ad72f67c249449f0290c04fd3af5fcf8739a3909b7220ca&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643324473Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/d, Name: checkpoints/256/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/256/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=7185116bdf03f8baccd9918d37ebaaaf64ffd51b1f6f67b2e4eb9755ab05841f071a6bc8a2b5a1863a571e1f8660199d175dabe87bca8ae2ed3b7bea19364d41f4a15e7baa7ad456b48c70c2c77939dc7b467f222fd68d6df5eb39cb12a8ddddb77c011d7518dfbd8f91aafe98dcf9dd75d2f2c4068a11aa2df823f89123adb076ae8139250cb37f675f056c503a066081f8277eef188ad1b30a6da37650476e0fff6add82f5af3d7678384d58c0c4a909b47b1a8a010f983c383d357532eb479a18304e9337a01006f4727b0f1167f6ccbd207c192de364b3e509513b7e4fe22449fa4c0012b2652ee15d5275340e755c4b7c7e355dd6e55cfe01bf806e5277&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643351272Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3136, Name: checkpoints/3136, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3136?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=14d7805a0ca4559fbbeafd57370202f454c7f07c1fa2babfe9c4202a09fc7250c744d21245f318509b42203d5ab22e03baf6178b9cefeb5cc7adfb1f5d50fbfc955c3e3e4b997ed2f154c5b591ed1364575905f144b3078b801cbc2921a17d06ab324bb343efb3bf80f7a742e8c712b49f81c5c7d29581342f35fe9e0597e35e9937498cc039ce1712739263ecf4d1632ecc67a57f90978e0c14e25441adaa238198c6d7dfb8625b76091b0fa38a13696a7b337734e1c840b4e782c8aa5bef3958f5c09ecb45509abf97fee921967bf33fe676176d67d510990a5c4a9aaabadb021bd0f9172288da8814b739d99ad7ba63867a8593a0f27124c06793bb41c1f8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643435265Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/d, Name: checkpoints/768/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/768/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=bf808ebfbb9b73b9088ae40ff831a69e9822812274622a1aa6d7e605b770733b926df4c1e0adfe83832602778e9a7bacc98b97ad857dfac268d3d19a836409aef54cfc880a02f0945d4381313c091c08af984acfd0cc15ab64831ae037d045d9583e10c21de6aef09bad091153ba563f75e4c30220d7bb8c4d7c56b8f536f51b84616630537f09e44e1892b7f9eda352daa6e1fdb35b8a9dce6e068ca03c92541505947eca4dd3838a689c21665145819af894fec696689ad46c2145b65b84af3c931048998107299c7a64f779bbf799f4f7d3e9e78102c3bb7f172b97022e2516d4f0bcf8c16bfc7acd990448ebd7f84202999269581f8fa96abc1a264d1acb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643484749Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0, Name: checkpoints/1856/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1856/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=174df64d84f30124b60d46296985df2bc3fca56db368ba60701777a4c6bd217486da00f7a3c03223b71ff07c7849f41e41c93c1094201f63c0609129ae7fc908cc1e27c723b5c506c5465fce8aa1a5b3d2880e7b0b37e0e669d0cee16e322e06a7d9d5c1dfb4d113b5c950d4fb8e80cdbea956ddb67850326f935377bfc764330f39e916782c015ef5374401836ea2b8ce82911b4c221628f34138af04801dd9e2ade434855bebde7c4ec855f2697f4d15402827bd68d6fb169d4497d154e0bda700e817c68713cee1bd69da9b91acad1d098c2324e13f657b67f3d0a62187325c21c48ee333bce40f625eea14061352a5dfd3d82821bc3b03a4f6ea851a1eb7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643513861Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1728, Name: checkpoints/1728, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1728?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=0cd3845aa3261ff40140776fd9a81707e5f3a8e7b2b8f3c8d5df8050b636fbb09a3c0505f5d7b8a23e054d9e46f5650c957dcb41e372745dfb602bd7eaa6790d85873e506e9de7bacf847dad7480c72a8e2cb8d2c71a6238d251efd4557f0f8cb84e3f2deb3644a7db7ea43a5a0d2a8a65448ea16231179465e99675d8b8d37fac8837540908f87dcae4c3ef62231cd92c3285e9633af5be5832d39b8f15885cd40a38e5620e4a4a454b31d39afe8c1585b759994fd9ad2feec0b16040193f836d3d284f30332113454715e8d7254b47be0aaaf699f4ed5c3427cada9d784cc7acd9bb9cd8970f65231f6a5517286d1dfe6807275f3419d625bb1b826621f2cf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643530462Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/704/default, Name: checkpoints/704/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/704/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6e3c2671554cbe8a705369db6ac052195ddcb25518543bfdc3ac30e281a11ba1a272f8e1d04d48234ecbd5c861039612375610f469fa2031b8128bfb5b28322af4797dbe79f40674b3cbac83cdd0c99b4e56e857dff959484ad65f484179be0cc7e1ad4f5049241528ef746d7aa12b4253016b3dcf2653449bd0a8beffa80d3dcb995ccd7baeaabcef00fe5daefc1e1665a8d1c647470504d7f14db538272c212f36810815f4e8468d5aef62b2a5b616d6f3b739cf4dc84d75ddd9ed104eaaadc2c0aaf068e19dc69ec682bb2e56c21706bb1c2b227359d1535d06822e807eebefe30ad1869e79daa138ef47d4f804150c2781ded65c4c181b783fe1e4eab62a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643536453Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d, Name: checkpoints/256/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/256/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=b3c2eec8cfe69a9807efc6fdceaa80b619ed3eaf9b4f2fd8cabbbec95219e03ccea28f8a997b45958fec1c13441096ce8457d29945df0fad74be06a6d8fe665cece450cfca1f67d9b6e54d8bda030660e28748c6c6e794170a844e098dd4ce54a8dfbdc185c7860c6a11ecd2dab28bed569567dc6b45bd04981fa19e4eb5c6f5703350b8804954f17668f134442d046591c15f901872befb8231985660aa753455b377bf6c6f907c75112800895c8ac4057e43a6ff1ea9e47d60c4074b4930b994d032165f2222dce5f0c1cf60e1a54bf6563db0314c7bcaf540311bc0d7816708b8eef044bc37bd7175dfa639feee9e616b8675806b1cf99e98affb21cf591f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643556682Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4288/default/array_metadatas, Name: checkpoints/4288/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4288/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=2d3f49d013f55e892a525797374aa02da771e87965e412cd3bd0f6b33468d83b345833a50b02bcf566fee04b8b2d022adf04474da9a137703c58f6658e13efeea075325df0d9268af3dc5ba57b9897f5f5354ede81c881306ea9c856eb8bae72f6476c79dd007817891d3929cbb2990d3b4e70d3c7e162ba11790dcc8b43867f4fbf80147b89f8ccdd9b39d948828aeee68a1a3279bd08faf1265530d67d7d5df3fc2b9dcb9a189fa539c0933407f9ea431509079e7ddd3eee46c29983701a4631e6ea7878f2457f76b036e6fa2465d394df894504713cd7b65c25e83a2f94819d116843a5630cf5c9b6b15bb6226d08d6dcc9ffbf97e0b780867429d2e5480c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643561661Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d, Name: checkpoints/384/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/384/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a44fad663112b54ef690a41bcb561cb2edc233c8f9f9dd4f9e7dcf553ee0340cb329b61683269f40f0a5d3e03c4854001768d7896ae25053be11ae544886a5717861b9e025e3b7c84d4a0f2af669b816f1f0ce7c457ee8503e7137c9c07ab4dad34e0ba21e3daa3ee8b00f8ab9cad21470dc3b51ef4eb3bb52973ebd9f3722eade2cea95a99d835e5d8e34084ffc08e60f2b4df836c2784fe0cc3e2bc17fb3692b8683a80817e4a5f785349c244eda52d28bf24d2ef127c6dc0f140a7322123845c8df03f819afe0c6b36553167f7aeb3106936e31de97f7eb0aad7c94187278ca3148722e3b5963794a50860ae0e6d1d5acf3c82f0bf5ee30ea48391aae3699&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643577843Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/64/default, Name: checkpoints/64/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/64/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9aa7c1329675c5b969ea5dfd4ae3a6166c1ecfe7ea7e52fc822d2364f3179430fe7733afb3ebf936afaa99bb3ba34dcde162d0d74ed3e929e0c7f938ba3c3bf1b06fc88de0fffc2c2d7b4a4f82f584500a422ee1bb029bcee14389801d72f1263d0833b2439fd925aa046adddfe69338fcbba59579c3cc93c1664aff3dcc3ccd28a2086be365a968cd2bbfbec220601dda0abe1130b8361b99215e8e1d356bf0f0c03299636db832ae6586641dade6594347720d6cd976e79688301c8c55aa4ec1fbf9d0a49e453e02a44dfbb4596a5bdcb852be59a47950c53572bd71b0a61bd897676f97bcc872a457ebdb0496a9f44d86e09d00577ca04056ed709e25360f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643586066Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2624, Name: checkpoints/2624, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2624?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4e11ec291152dcf0a970e8d601e1de81e526dc40244815b5971b4ee31bfb613bc1640e4983a6facdd46533e44c5f652997b5f479b0610c9a920a8680e51fe37f837e3a7d1cc9d0f13c7985b0fa7a09b95ec9364516dc11d6d978ebcb4ebf591ed01d4b56ab63757a40043fb8dddba3ac5cffec5cdcbb717ca825f5be1951b4dd7801f72012e6023c86a8541d055a3262eea8cf0d0ef2ae81f3867a7a8c834dd195c5465d400113e6077687f605cacda7ff27a1a832309720cfbfdd7b3204f77e134a8fcc5cd4b67acd53a38f4503cd656920a6b4babefbb2d9efd3a05f2fb4cc3f2a5b68eb3465ed6ee448a250bc86f685d7e12eabf2e72e6978d40b0dffb443&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643621854Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d, Name: checkpoints/3008/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3008/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=45b2bd449ea7035f9573bcfea4fb4b76374c9da670534a8166e792fe5fa0c5bca45c133bf5a5683fb8d9a1e442eef989ec666953a90ef7acde707e64397b6022c2ad9444273d06e981a2c18ce40e7196779a5a1f28a5aff69e9d4736eccc637242a8af7bc356e8dd8d69e48948f13b869cf9349a80ff2f7d6c4de8f26b98cd37e780a9c94a880ee574e4f2af5ccb4af8d8f7df867c9ec7fa193e0d98533d463264035911598ba56fd98a90e21330bc57b06d615781392453e69e618cecfc454036ec24c16ea19f75debd51ac66d2debf2a573ce5799f862a1ad22277fd0fcf6627421d28654e81528fea35dc4e47e19e8e1e6ccac4490518cd9eac5ba0834f2e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643628171Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/192, Name: checkpoints/192, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/192?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=9d8d03c27dc6cb557a532037574ae7ee2ee101e11fb391a11915018774e419969243195934d0b67752e91321a799d4a0b73431dff5720ba8a768e02d10c06932f138d4624d218d98d1fc804ab43e68f5625554305802f8675f5166e06c1f2e1fb2374870eff448eb1ceb71eaf6dd542383c6ac173a35d6f2a3874c9411b6ad6173feddaf131add4ede191a449b8888dfab893ab7dc688ec72894bce1b3017f5ae8a468ce4590fcbb2fc039a46596093e742e8f13a5022669e63d874ac8093aae361e2d4fb745e02803697fa72a48143d0d02896103419606a7d8af6c45eb8e6e4b6df47050d850e053e18cabbf0cfd6a6ec31ca9c54080cdac4c4178cb1b350d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643633073Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0, Name: checkpoints/768/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/768/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=c22d4069aefcb1909f992ff1a080594084b76dfa32c0fb6430a34955094f07272fd2ff522f5faf15a74e10681b0ffce6851c5c135642e11849a4759a9be14946f8c0c140200eea890acaa36efbc1718bbf193083c3ef06fd0521bbd791a8bb54646813021aa101d9c5dcfae8be88002c76fa550c7757101ee23ac682ca26271d43872b9cd413c62f36490dda1a01d2c59ec432b346df103cafc62189c4c30491e80e840d04253bdcfa04e48cb38e43cad2857d5726dc64e61f9e21c735cfa9b28263de960893e7c9fd54fe6063f0529c7a19e54f42a6974bb73981737cc299004185887bba2aa534aca6fb91abe80953c2555b4229ce4d811e931dd62ce5d077&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643629086Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d, Name: checkpoints/640/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/640/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6d3d6ec6cbb09f38c51985d216576d35db20b9396cd838f9cc95c0f19ef82038a857d3ac3bfaadcc7df5ecaa5bf460f633b6d42f63a44d573ab7784aa0f9f2722d9cb72f29c978db3f90fc61bb413d047595998301e30fda89028f7c572379b1b44cf096675cf6c28e21b9c5a10c26f9fbed4c8d9c250b2bf5f30820dafb3cec912980b775be3a8c71df5d16f925ffec035a1de74cbe9b7e8a3ccd038012354ac39486125096c4d33b4ae69d144dfec8b48e65879e8126b727fc0e159e33610b5a40701a35211e3a572e6032c6d251e51c7590985590909429e6f118d4d6393b67f1e2d44ff858e8ad0ec2e72a9c3f2c7e603a1329cc5c6a9bf9c540eeca298a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.64368233Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0, Name: checkpoints/1536/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1536/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=27adb5d113e6269f5f122a8b3684060daace643d1751452a785ebb57ef51258daefc5bb34d8f3504d8769be0f3b08843469f57623a4abcedf9f1519ee5882d6fbddcde45c82898308f7ad910d4574b33af71c8d31cff94dd2be1b217ac857779aecce6d2c477ba585758b68238d6b4c686979dd2c88227a21e3088d310ff77efcce6f26d3b3da53720adadad64ee3691ef8c4e8db7e9193f8b4f6b4eec518c0a86f10cef1bda211afa97c12c974ba449f50544b20af654b9bbc70aadeacf3af277c7f079aa2ff4e970fe3cb62cddce73aaeb2bb6fc84898955be5de71c1391dd87135c66ce50ea49e69f0ec0820563cb0ffae7fd26181d38f04e2bc970d0f56c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643679764Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4032/default, Name: checkpoints/4032/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4032/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a97a0931c1bfebe0ec7dbf05c92d24fd2177f61e705f0bfd4788a26dd357dfbd091c66ff1310d0a053d744d6b1faaa3b9a9f9346741db155a04e4a57cf5f7d71cc3d393d122e8c90be43a5ea2ebe2b62ea4925d9711b9a00682aaf3dd5a71231d2ffe9cf3df730bc5704416f6ac8413bca73766f3fe4cc2adfee1ab3d12645fa038191f25ef6af028407800e8ade9184db44a7c04940627a577434351532f1cc1968b0200f85bc52bcb7db8e08635360cefef74e87d29a6122cd7bd644098c0fe76e05061534a85d9ef0a74e55edbf3ff3fab31d884f9991ce34d8fdf4528afd3327f68996ca0bd3ad109f6eb44264fc315c785459bee9e8d0dda96606d01f18&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643686325Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/320/default/d, Name: checkpoints/320/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/320/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4a6c497edf8fad871690591bfdc535fc1487ccd289c70d8daba57adfdf3ed0cd9611f2dbca423c994cd865b55bf328d21639b361365e37d479e7861c86ac721669eb3392efb116a94c56d5d7f79530784a9d72cd4822405f2f930252bf71b745ab902a92e2873ad01ec1f60cf253d524e29ae93e61e3927ec43a2d1474143d045839fd5cf10a853b3349317e66901ac0a69a13a2a55feacbca35f765c196a10802e95d5224ff494f6f7753862ba87125dc7987149442e52b1a7d1ffa2d82a2a57907e49390b3adc0084a3de2d56ed9d835dcec487916a7fb2a286aba622f5bc11b416c1f39cb63262cccb28d4c80bc4c1f1617481f06f13be2c27f362c0a938f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643677767Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/array_metadatas, Name: checkpoints/3456/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3456/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=a3dab96ba3b73685b34cc6f1f9a5af8516606796afe4ee35aac13556814ddb0e5ca9252af2a35d639f12e6fbe6439d82148a8eabd5c54421be3ed512bdf3c7de2abb893dbae88be98d56617d3c7787f2cb34e3af7e0c7dc652b3658d88c3ef06ec700b36572252987e6621da6137f42361ae006e346616b2837ebf3f5622dc387bb8e40c86f7f0b16a4a24aa7358b976aea4100baf4c703fe6b99041c9d0392af54320420f5622af46077cb3d03e9e3c73050ab08452ddbc98e731eea929823be5939aef8155909eb67ccf255d41121d6d408ed4d68c8a30930e5e241b6a64dc3639d2d8d4f46650efdb2fab006a7d8ec6e2739fea0e93e9ac849abab5be4d73&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.64368498Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1472/default/array_metadatas, Name: checkpoints/1472/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1472/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=4b867d398ce264e6a62f18f3747214623d9cb30859eb0fc36d9d296808724049b7c783e7a71178a85535b42923654cb64094392142f49cb41a852f197770b5a00049697de67f30a55e117f37587af31779bcff6a858401a2ec7471290a67f15d7995dd6651cf7b75139ea92812acfa9ef9ceeec88ec6fc0ee5ba73d4269963854177377bbff9c260f184787bff8159f0e0f1f17afeeee65d6c25cbad88de3ff2444fc9af2b6e675f7953d096ee7fabcb4a7d71ba66c0093c6f2753199aa223a068a0ca0bc621946ebc0219c9bd674e32f452834af89f914024a17013e74aa6a656ec8f61a4c7c9a6d79da918789c965c3a3e933344526af68bda2cbf77ca90f6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643711835Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2752, Name: checkpoints/2752, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2752?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=2b2ba94e8c2423ee10c9718c0b32e7eb2df80246eade99cf43c30220356e3674b515c0bb327c39e18fab03051f063b07efd508c25d021ae0ef6b1e3ad3b2b3d31d81f8375feb0b612bde5ca88dcaaebba48783dd5408a0327d68690eefedb432524fc23b2fe73e1136266cc17ee838e628aebe82b3b4705f8b5f7f58d9c91fda7d1742a17fc92ce9eee56f35e87f0a61f7b494bb52720e98fb66690838a2708c52dce73c4fc2714355739e5f60dc88694617c9651310bc381a45ab943545b53f2cda65a12be2ba7e248086a87d6c62db912e330b8079f9b195cf497e0adee0cacbfc9d5cb0cc394c61d7514bd5685a4208e25051664f4f4e23696139e63f883e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643723201Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/1024/default/array_metadatas, Name: checkpoints/1024/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/1024/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=020a753ae831fb499ccd9d64e6276ecbc69f2c81e551d9f524de955b428411212ff52ee3fe830f2c92ceedfbf6a16d745a636c485c35445c218ff48604ccb8fb8f5f5d2ab7f837e353804171ce391b68501e81d6ae7147c4a379591135f92211e0a2431b107a54f7dc3567b6964b714e8fe3b7a5e062a9f9093cc2cd890e5f38a96649f4bc55f5ad0feefaab609728be6e186c3e9d07afa48bf19e5cb08c3d04b03e5983ae93f474a68f8d8405378235656e965b95b0bd69f6e27a8efde0ace3654869a548e24fdfc0ca8094aad1d8595f59b7b8bfdcaf077b8fef667530d2eda64c8db3e0b959489f1b61797d21891d7eb34121516966eb9d15f8c75e573dd1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643741728Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/4928/default/d, Name: checkpoints/4928/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/4928/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=202174641f9a8798238222fef5743d09a2c6c8653c84985fbcdd9aedb17d86d68be21956bcd311aacb99e0f5f4e50f8f4e2cc1f10c3525287fbe16f2b22ab6e71bd21d503dc987f1d123327e5bf6d28952f7492d5935f7872e2e5991d39f434c2074e9eac9db5e34a7283d598852f61a5b0b7fc3b9cfaa67738d96db835663ffacccb17c6bdcf25955f85710876d93e74f5b9448b370f029136bf3efdb1813ba625d66ece62731f274f0de86606a70728393d1ef13985f3c7d5dbe2cc37e5233a463cec7f018d321ab5368bf47ecfdc8fd4efcfc43f35853708d839f2370f36b0662ecb9add2fd522758cd754fdd824095b78aea685051ee0916e897b70723ff&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.643749388Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d, Name: checkpoints/2816/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/2816/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6eb41db66b8d65808c0ea80018d372b89d08eba40ffde85d2d88f26e6a2ad9cad0cae0caf9f419a43ffa4168c435435af6883145e8f618e6d5bcfaef1ff962095e75c02447911e44a6d6e38f111cea6ca4b0b220a41468912b14350491d0587ae04ea514b180fc286f138898733f28129c44d695712356901fe5d3b6e2f38b350dd88d401270513a414259335c97d5ff73e8a0cd3babcadc8db13b6e01316fa2af6051a57f310fbcd5a0ba574566ff503912c4fde16859ca25421590d3af1c9f77d8182dc88cf1c408c70ff45dacf7227f616fffe82a3de9d2a2b652c4fe5e14eb564690cd7ecf84d67375d4833506ff077855b8c82b7e2f5746b31d3f020708&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.636586659Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/files/checkpoints/3456/default/d, Name: checkpoints/3456/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0/checkpoints/3456/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T170110Z&X-Goog-Expires=86399&X-Goog-Signature=6de6f4f7fe61f6ed88b3411ffd3e03fde20752d3781a062fea8d141744362e24d0cb752acd39289758f285149dac9bf5c0068ec65eaec791270de909088e099c455aead05e07cdb21f2ddfda4ef783bc7fbfbb1a8bdc5bff9a0703e05f42b79cff18b7f5db6945f83d4b0884268397c3d17e12f5d99ff54b0af2785ecfb79e4ef63a027bf36928b937dbe3de707a3b9d117bed60dc236070cf08b32d362994b505e1e33c2fd830ce9e5332a0f61568e3f0b5fd405e41187b9c8c90adfbda0a97e6eb94629cff09a49f1394d52faf7e66e1a97a077729bad2223343c678c84bb0672e929f6a4d75e50b6fc06d45ddbb5b44ab6bf1ade4e21a080a453837b15582&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T17:01:10.659634157Z","level":"INFO","msg":"handler: operation stats","stats":{"operations":[{"desc":"uploading output.log","runtime_seconds":0.521863057,"progress":"1.1MB/1.1MB"}],"total_operations":1}}
+{"time":"2025-08-07T17:01:11.234626586Z","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
+{"time":"2025-08-07T17:01:12.543969299Z","level":"INFO","msg":"stream: closing","id":"wp_0.0_alpha_0"}
+{"time":"2025-08-07T17:01:12.544007455Z","level":"INFO","msg":"handler: closed","stream_id":"wp_0.0_alpha_0"}
+{"time":"2025-08-07T17:01:12.544025476Z","level":"INFO","msg":"writer: Close: closed","stream_id":"wp_0.0_alpha_0"}
+{"time":"2025-08-07T17:01:12.544038998Z","level":"INFO","msg":"sender: closed","stream_id":"wp_0.0_alpha_0"}
+{"time":"2025-08-07T17:01:12.544228555Z","level":"INFO","msg":"stream: closed","id":"wp_0.0_alpha_0"}
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/logs/debug.log b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/logs/debug.log
new file mode 100644
index 0000000000000000000000000000000000000000..f7ea542a84d4e64585944bf7b4720c940758bc7a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/logs/debug.log
@@ -0,0 +1,28 @@
+2025-08-07 16:04:01,893 INFO    MainThread:3825414 [wandb_setup.py:_flush():80] Current SDK version is 0.21.0
+2025-08-07 16:04:01,893 INFO    MainThread:3825414 [wandb_setup.py:_flush():80] Configure stats pid to 3825414
+2025-08-07 16:04:01,893 INFO    MainThread:3825414 [wandb_setup.py:_flush():80] Loading settings from /root/.config/wandb/settings
+2025-08-07 16:04:01,894 INFO    MainThread:3825414 [wandb_setup.py:_flush():80] Loading settings from /workspace/timaeus/projects/rl/wandb/settings
+2025-08-07 16:04:01,894 INFO    MainThread:3825414 [wandb_setup.py:_flush():80] Loading settings from environment variables
+2025-08-07 16:04:01,894 INFO    MainThread:3825414 [wandb_init.py:setup_run_log_directory():703] Logging user logs to /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/logs/debug.log
+2025-08-07 16:04:01,894 INFO    MainThread:3825414 [wandb_init.py:setup_run_log_directory():704] Logging internal logs to /workspace/timaeus/projects/rl/wandb/run-20250807_160401-wp_0.0_alpha_0/logs/debug-internal.log
+2025-08-07 16:04:01,894 INFO    MainThread:3825414 [wandb_init.py:init():830] calling init triggers
+2025-08-07 16:04:01,894 INFO    MainThread:3825414 [wandb_init.py:init():835] wandb.init called with sweep_config: {}
+config: {'env_size': 13, 'env_layout': 'tree', 'wall_prob': 0.0, 'env_corner_size': 1, 'env_terminate_after_corner': False, 'obs_level_of_detail': 0, 'img_level_of_detail': 1, 'env_penalize_time': False, 'net_cnn_type': 'large', 'net_rnn_type': 'ff', 'net_width': 256, 'ued': 'plr', 'prob_shift': 0.0, 'num_train_levels': 2048, 'plr_buffer_size': 4096, 'plr_temperature': 0.1, 'plr_staleness_coeff': 0.1, 'plr_prob_replay': 0.5, 'plr_regret_estimator': 'maxmc-actor', 'plr_robust': False, 'num_mutate_steps': 12, 'prob_mutate_shift': 0.0, 'chain_mutate': True, 'mutate_cheese': True, 'train_proxy_critic': False, 'plr_proxy_shaping': False, 'proxy_name': 'proxy_corner', 'plr_proxy_shaping_coeff': 0.5, 'clipping': False, 'ppo_lr': 5e-05, 'ppo_gamma': 0.999, 'ppo_clip_eps': 0.1, 'ppo_gae_lambda': 0.95, 'ppo_entropy_coeff': 0.001, 'ppo_critic_coeff': 0.5, 'ppo_proxy_critic_coeff': 0.5, 'ppo_max_grad_norm': 0.5, 'ppo_lr_annealing': False, 'num_minibatches_per_epoch': 4, 'num_epochs_per_cycle': 5, 'num_total_env_steps': 200000000, 'num_env_steps_per_cycle': 128, 'num_parallel_envs': 256, 'console_log': True, 'log_gifs': True, 'log_imgs': True, 'log_hists': False, 'num_cycles_per_log': 32, 'num_cycles_per_eval': 32, 'num_cycles_per_gifs': 1024, 'num_cycles_per_big_eval': 1024, 'use_fixed_eval_levels': False, 'evals_num_env_steps': 512, 'evals_num_levels': 256, 'gif_grid_width': 16, 'level_splayer': 'mouse', 'checkpointing': True, 'keep_all_checkpoints': True, 'max_num_checkpoints': 1, 'num_cycles_per_checkpoint': 64, 'seed': 42, '_wandb': {}}
+2025-08-07 16:04:01,894 INFO    MainThread:3825414 [wandb_init.py:init():871] starting backend
+2025-08-07 16:04:02,127 INFO    MainThread:3825414 [wandb_init.py:init():874] sending inform_init request
+2025-08-07 16:04:02,134 INFO    MainThread:3825414 [wandb_init.py:init():882] backend started and connected
+2025-08-07 16:04:02,138 INFO    MainThread:3825414 [wandb_init.py:init():953] updated telemetry
+2025-08-07 16:04:02,150 INFO    MainThread:3825414 [wandb_init.py:init():977] communicating run to backend with 90.0 second timeout
+2025-08-07 16:04:02,705 INFO    MainThread:3825414 [wandb_init.py:init():1029] starting run threads in backend
+2025-08-07 16:04:03,279 INFO    MainThread:3825414 [wandb_run.py:_console_start():2458] atexit reg
+2025-08-07 16:04:03,279 INFO    MainThread:3825414 [wandb_run.py:_redirect():2306] redirect: wrap_raw
+2025-08-07 16:04:03,279 INFO    MainThread:3825414 [wandb_run.py:_redirect():2375] Wrapping output streams.
+2025-08-07 16:04:03,279 INFO    MainThread:3825414 [wandb_run.py:_redirect():2398] Redirects installed.
+2025-08-07 16:04:03,283 INFO    MainThread:3825414 [wandb_init.py:init():1075] run started, returning control to user process
+2025-08-07 17:01:09,654 INFO    MainThread:3825414 [wandb_run.py:_finish():2224] finishing run devinterp/jaxgmg-wall-prob-tree/wp_0.0_alpha_0
+2025-08-07 17:01:09,655 INFO    MainThread:3825414 [wandb_run.py:_atexit_cleanup():2423] got exitcode: 0
+2025-08-07 17:01:09,656 INFO    MainThread:3825414 [wandb_run.py:_restore():2405] restore
+2025-08-07 17:01:09,657 INFO    MainThread:3825414 [wandb_run.py:_restore():2411] restore done
+2025-08-07 17:01:12,520 INFO    MainThread:3825414 [wandb_run.py:_footer_history_summary_info():3903] rendering history
+2025-08-07 17:01:12,537 INFO    MainThread:3825414 [wandb_run.py:_footer_history_summary_info():3935] rendering summary
+2025-08-07 17:01:12,543 INFO    MainThread:3825414 [wandb_run.py:_footer_sync_info():3864] logging synced files
diff --git a/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/run-wp_0.0_alpha_0.wandb b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/run-wp_0.0_alpha_0.wandb
new file mode 100644
index 0000000000000000000000000000000000000000..187611a710a80d9b0b9375dbd5992ce82a6225c5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_160401-wp_0.0_alpha_0/run-wp_0.0_alpha_0.wandb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:923cb61713b643f14ac5d150c5c4f3ae5cae448540fa51f5b285ffbc768fbc66
+size 6208632
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..11806b71d0546a35a329f660a4ff1b534f84f46e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754586274671438632, "commit_timestamp_nsecs": 1754586275460881604, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/d/eaa0d82dfa2bb3ed4f4f4a836909f111 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/d/eaa0d82dfa2bb3ed4f4f4a836909f111
new file mode 100644
index 0000000000000000000000000000000000000000..56a5047855e27fca919f3b80f5aa3ff2c3bf2d0c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/d/eaa0d82dfa2bb3ed4f4f4a836909f111 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4b713a8ba6a3e0144c399c16c1b61a4bb0e5bd83
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/2da80055c04f0939d85977db42378edf b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/2da80055c04f0939d85977db42378edf
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/2da80055c04f0939d85977db42378edf differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/36e91efea5a545989c7295460e157a8a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/36e91efea5a545989c7295460e157a8a
new file mode 100644
index 0000000000000000000000000000000000000000..f96e8324832ba2e6a25d9c7832a4b5fcadc99857
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/36e91efea5a545989c7295460e157a8a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8719f4b28cc4e45ec353a4ee318055f447cf0207255ffc254ee91021e10028a1
+size 404844
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/4118de792425ca3607ee1403b337dde5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/4118de792425ca3607ee1403b337dde5
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/4118de792425ca3607ee1403b337dde5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/431c17a3173241437f5d1d3c8844ac83 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/431c17a3173241437f5d1d3c8844ac83
new file mode 100644
index 0000000000000000000000000000000000000000..69a3c22a69deb8100dd0997a960db63ff226d174
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/431c17a3173241437f5d1d3c8844ac83
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7ab813e721a7af1b71906209231e3d80a6bfb506900a882c9985f919ff171b80
+size 261107
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/88e9fa9dbdf0ba3d234029442780f7fe b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/88e9fa9dbdf0ba3d234029442780f7fe
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/88e9fa9dbdf0ba3d234029442780f7fe differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/e30bb9c31c7610b53f224396ff61fb51 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/e30bb9c31c7610b53f224396ff61fb51
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/e30bb9c31c7610b53f224396ff61fb51 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/e897821eb5f925e0f89aa48b1a0fcba7 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/e897821eb5f925e0f89aa48b1a0fcba7
new file mode 100644
index 0000000000000000000000000000000000000000..5e0eedf0f7e9f11129ad49899af4688044c60e54
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/e897821eb5f925e0f89aa48b1a0fcba7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/f4cc6bb0fca20bff404ab90e584f83b1 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/f4cc6bb0fca20bff404ab90e584f83b1
new file mode 100644
index 0000000000000000000000000000000000000000..ec745d97353f97209ef50f0eb738585830219f53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d/f4cc6bb0fca20bff404ab90e584f83b1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..300da8dcfb6adb17a80a30675eb99effff39e034
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..cab7db7a7a847cf7c329a7064c52f7e762e679d7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754586881086424298, "commit_timestamp_nsecs": 1754586881255562089, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/d/c784582a0e603b5a452e99416f15e04b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/d/c784582a0e603b5a452e99416f15e04b
new file mode 100644
index 0000000000000000000000000000000000000000..4e70c60a14825a8507539d90aa16ba358fc31585
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/d/c784582a0e603b5a452e99416f15e04b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c073e45343abb3d02d05355f653a7f68a0a9fee4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/84e4299faa8f0c1738101d7c5d42528e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/84e4299faa8f0c1738101d7c5d42528e
new file mode 100644
index 0000000000000000000000000000000000000000..ff75f7a192111c112fa8065745d20ba65d3f0ea1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/84e4299faa8f0c1738101d7c5d42528e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6edc38d2ec77a19c23ece0855d73f6193ed4dec6d6301a3100c38a04cac618b5
+size 438809
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/89694b1e03245a9852a7fb339b5502e1 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/89694b1e03245a9852a7fb339b5502e1
new file mode 100644
index 0000000000000000000000000000000000000000..1615c27ae3da2b6ccd97dffa4bb4d357f02cef59
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/89694b1e03245a9852a7fb339b5502e1
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e75169ba82e9703cae337357973b79c3c1c64ecacd574be7c038ea13e4b45714
+size 262002
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/9d1a29ae42d6d0e41b3810cece6ef8f5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/9d1a29ae42d6d0e41b3810cece6ef8f5
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/9d1a29ae42d6d0e41b3810cece6ef8f5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/c901bad5a6c90056f8150bcf86529ab6 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/c901bad5a6c90056f8150bcf86529ab6
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/c901bad5a6c90056f8150bcf86529ab6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/d187d433fe3ab9fadbc5fef1c46f166d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/d187d433fe3ab9fadbc5fef1c46f166d
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/d187d433fe3ab9fadbc5fef1c46f166d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/eec4198c676885853ee50a11ec67a83e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/eec4198c676885853ee50a11ec67a83e
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/eec4198c676885853ee50a11ec67a83e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/f6850ffae042226c4fa4f3763d9008f3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/f6850ffae042226c4fa4f3763d9008f3
new file mode 100644
index 0000000000000000000000000000000000000000..2879d14218d3e8823d9f92da38cedc7adf0c317c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/f6850ffae042226c4fa4f3763d9008f3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/fa0c2d6ad2f3138cefed35269f2fed65 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/fa0c2d6ad2f3138cefed35269f2fed65
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d/fa0c2d6ad2f3138cefed35269f2fed65 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4564d615b48839db76e1860b0edf2331b93c3e57
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..6ec22dab94b1cd7f8a17e8e889a989e05111cf0b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754586916159854334, "commit_timestamp_nsecs": 1754586916284445894, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/d/f2c8e43051ff80150ca00183e6b61116 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/d/f2c8e43051ff80150ca00183e6b61116
new file mode 100644
index 0000000000000000000000000000000000000000..e9e8d518e1f1b8f97cee49ab348e2c7029d74841
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/d/f2c8e43051ff80150ca00183e6b61116 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3a1217b9f0b52142f95b7b1f65964675d5cffe34
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/249d560b1af9e58a3ace5e526b565135 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/249d560b1af9e58a3ace5e526b565135
new file mode 100644
index 0000000000000000000000000000000000000000..7eb95c13c5490342fd78701c25cb7f1347c947a0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/249d560b1af9e58a3ace5e526b565135
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7cec178c9f48adcba0ce9e9471b361d4ad69e4ab5d2eb394e9137fd3bf40ed87
+size 338551
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/296fe2c8f604b87134de3986adaffeff b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/296fe2c8f604b87134de3986adaffeff
new file mode 100644
index 0000000000000000000000000000000000000000..567ef6d89e95c637b30c3448ea92a6df56f84ca7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/296fe2c8f604b87134de3986adaffeff differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/6ce888ea1cc4db442fed0d6fe6650482 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/6ce888ea1cc4db442fed0d6fe6650482
new file mode 100644
index 0000000000000000000000000000000000000000..dfbfe1a61305bab6f716f83c77874d9e3176999d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/6ce888ea1cc4db442fed0d6fe6650482 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/6ed1e9a6205df4b37473c74a2b2695e7 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/6ed1e9a6205df4b37473c74a2b2695e7
new file mode 100644
index 0000000000000000000000000000000000000000..d272191c3fa82dd114ca743379e0f9cb4be3e595
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/6ed1e9a6205df4b37473c74a2b2695e7
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f8ea45768342382b0965a35692466e0300fbbe6f5da57a248dc8d6eb3ec7281f
+size 104148
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/82e1e8613c2bcaa7e8e42310c27d0000 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/82e1e8613c2bcaa7e8e42310c27d0000
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/82e1e8613c2bcaa7e8e42310c27d0000 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/9b5ffacd56962a808779a7954865fdac b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/9b5ffacd56962a808779a7954865fdac
new file mode 100644
index 0000000000000000000000000000000000000000..c0793838ab93c20eefdb44ed74b4b5889df59613
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d/9b5ffacd56962a808779a7954865fdac
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5aad4f7c34419fb341685e17a237595093b540b20f5df595be8e10d07489f439
+size 264098
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..94e96dd4e9172ef391b18d8470408d6cd66ae402
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..bbf437ae1232938d6c419a9eec29fe3802e6ebf9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754586950920708356, "commit_timestamp_nsecs": 1754586951064858475, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/d/d837714ef01c1785dc0fd463e946407b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/d/d837714ef01c1785dc0fd463e946407b
new file mode 100644
index 0000000000000000000000000000000000000000..b91572dcca9f2313b25ef6497daa4be63dac00f6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/d/d837714ef01c1785dc0fd463e946407b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..fddcc7a4e21adaccbe6d187d77cd2dbafb30119b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/096cdb015fd6a2b12ef8c539ee1537d3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/096cdb015fd6a2b12ef8c539ee1537d3
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/096cdb015fd6a2b12ef8c539ee1537d3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/283f9e2386a51ecdeafa45471830e9e9 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/283f9e2386a51ecdeafa45471830e9e9
new file mode 100644
index 0000000000000000000000000000000000000000..8db9b7fec6cae0684e6cca93fab93496a3f498e0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/283f9e2386a51ecdeafa45471830e9e9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/4316c2b4bd1e5fe140fa5663d6daa4ee b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/4316c2b4bd1e5fe140fa5663d6daa4ee
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/4316c2b4bd1e5fe140fa5663d6daa4ee differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/76c6bccaab14ec8b9d588ea7144c4479 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/76c6bccaab14ec8b9d588ea7144c4479
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/76c6bccaab14ec8b9d588ea7144c4479 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/bf470c099961fe85aa8022ef3aa4a978 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/bf470c099961fe85aa8022ef3aa4a978
new file mode 100644
index 0000000000000000000000000000000000000000..72d2d984de583331034305002115aca9234f13be
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/bf470c099961fe85aa8022ef3aa4a978
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bb46146627c1d0bbe8c5d4a4f5f5a8c74c71505c4b79af2f6b5b4a276c4c1d77
+size 408297
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/c3eecb32ef1539eeed2e09be830a6db6 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/c3eecb32ef1539eeed2e09be830a6db6
new file mode 100644
index 0000000000000000000000000000000000000000..392b8c564f934177152f741a1de21451fb390621
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/c3eecb32ef1539eeed2e09be830a6db6
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3474991d266fef9e3b45753acf158db48cc39bdc627283706c0a8651e65764dc
+size 329938
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/e87fd8e62820d8757afd2fc31b36b002 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/e87fd8e62820d8757afd2fc31b36b002
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d/e87fd8e62820d8757afd2fc31b36b002 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c05ecbf83f41a2ba0448112607e88e86b617a7e3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..dbd0735060ef56f7a8e363024bc881df1ccb0204
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754586985762264611, "commit_timestamp_nsecs": 1754586985945586578, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/d/33f63ecb324744ab2626bce982359a7a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/d/33f63ecb324744ab2626bce982359a7a
new file mode 100644
index 0000000000000000000000000000000000000000..74c17294cfef1388837edfb0b8a638d49206d26f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/d/33f63ecb324744ab2626bce982359a7a differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..14eb0a2270dfa978ba4893549b5f42176126dc80
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/0a2b080add93b332ea236f4ea64c1fc5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/0a2b080add93b332ea236f4ea64c1fc5
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/0a2b080add93b332ea236f4ea64c1fc5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/30e59021206dd15612b60febffa04dd9 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/30e59021206dd15612b60febffa04dd9
new file mode 100644
index 0000000000000000000000000000000000000000..e8c22625819f59738be92909a5394475746f1da9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/30e59021206dd15612b60febffa04dd9
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6583043681d46489d456a1724099a4ba8bf1892ea77b638ed3f1519c51940752
+size 449405
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/73af820e7cb90f835283ccf3c7eaa1f3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/73af820e7cb90f835283ccf3c7eaa1f3
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/73af820e7cb90f835283ccf3c7eaa1f3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/9df1fc87c2aaf7c771a0b96f06fc3e40 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/9df1fc87c2aaf7c771a0b96f06fc3e40
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/9df1fc87c2aaf7c771a0b96f06fc3e40 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/a9a5fc2af6a9c67c030e35a9045f6903 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/a9a5fc2af6a9c67c030e35a9045f6903
new file mode 100644
index 0000000000000000000000000000000000000000..2910fbc6db2094404ed932c7d9bcdb23beb99906
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/a9a5fc2af6a9c67c030e35a9045f6903 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/d55be9d4f5454610a1c8287d079a9bb9 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/d55be9d4f5454610a1c8287d079a9bb9
new file mode 100644
index 0000000000000000000000000000000000000000..d5bfa3c5041915b1c19ec0e388e8dc7a699cce6b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/d55be9d4f5454610a1c8287d079a9bb9
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1dd8287a32cab188a41b6b0bbb57e17f402c9b548d03a2e62c3477d0fa759897
+size 287425
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/e442c7ebad2be8f28f59309cb5d2b09a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/e442c7ebad2be8f28f59309cb5d2b09a
new file mode 100644
index 0000000000000000000000000000000000000000..a922af7b91c42cfbc47f4f89daeda0f69cfb3f47
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/e442c7ebad2be8f28f59309cb5d2b09a differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/e5eb434c26bd045249419a9b889c8392 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/e5eb434c26bd045249419a9b889c8392
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/e5eb434c26bd045249419a9b889c8392 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/e91f0cff42dea9d7faedd687117d268e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/e91f0cff42dea9d7faedd687117d268e
new file mode 100644
index 0000000000000000000000000000000000000000..e1206a75cdede11c667b054a72b8e02b8defb834
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/e91f0cff42dea9d7faedd687117d268e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/f825168c740298f67ffeaf7d0edd679d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/f825168c740298f67ffeaf7d0edd679d
new file mode 100644
index 0000000000000000000000000000000000000000..cd8da373ae279421827eaadeb0534e6225487315
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d/f825168c740298f67ffeaf7d0edd679d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..67009f49a5e3c46e3477f6a11c831bd231949d42
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..c4bd9f20b377600f7f4bafc5f032177854ed5593
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754586374597284461, "commit_timestamp_nsecs": 1754586374776650770, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/d/641db5103dac0cfab2501206b49f6d70 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/d/641db5103dac0cfab2501206b49f6d70
new file mode 100644
index 0000000000000000000000000000000000000000..0ae8ea712fa60e4bb4ff6e79b8cd737fb57ca144
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/d/641db5103dac0cfab2501206b49f6d70 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b4de494078b474d65a75ef74b09bafd3826d0d47
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/41d153dce51c7d83914ef7b80786cef7 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/41d153dce51c7d83914ef7b80786cef7
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/41d153dce51c7d83914ef7b80786cef7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/577be909665434b5d64408d7697ba75e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/577be909665434b5d64408d7697ba75e
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/577be909665434b5d64408d7697ba75e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/6a7bf0005a5bb6752b5dcb19e9dce832 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/6a7bf0005a5bb6752b5dcb19e9dce832
new file mode 100644
index 0000000000000000000000000000000000000000..6c14df587a4be8e279f740406fccc96ffb3c117e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/6a7bf0005a5bb6752b5dcb19e9dce832
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dd02c9bfc3962efdd653f3445c78258d1da31bdd5ab21aab591907ad4d491eb5
+size 261709
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/84bbc8250cbc7839808e065a7c57cee0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/84bbc8250cbc7839808e065a7c57cee0
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/84bbc8250cbc7839808e065a7c57cee0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/af9d357c6e8b8bf2c43d6646410956f9 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/af9d357c6e8b8bf2c43d6646410956f9
new file mode 100644
index 0000000000000000000000000000000000000000..41749972c01f1b52582edee9ba0157a8f7599495
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/af9d357c6e8b8bf2c43d6646410956f9
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:354f778b5105530c253f79d0a12821529f84351d19eccbb02408593220acc3f0
+size 402642
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/b367a5f16cec8d0f7fd9dacf287ef3ca b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/b367a5f16cec8d0f7fd9dacf287ef3ca
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/b367a5f16cec8d0f7fd9dacf287ef3ca differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/d7226de0890a909fd1541eb2ea6539d9 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/d7226de0890a909fd1541eb2ea6539d9
new file mode 100644
index 0000000000000000000000000000000000000000..60975afed5981e2cbb4a7e99a6fecb23e652f15f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/d7226de0890a909fd1541eb2ea6539d9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/dcd7639e8c952d11d3c0cffa63869010 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/dcd7639e8c952d11d3c0cffa63869010
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d/dcd7639e8c952d11d3c0cffa63869010 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f806912e5cd26aec00bfdfed5d2a20b9b0fa3192
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..8cb019fa87e3a16dc3e26a76ed9644cdcc568215
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587018619527703, "commit_timestamp_nsecs": 1754587018756788817, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/d/cc8422e80f8c1f15c7def16672efb229 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/d/cc8422e80f8c1f15c7def16672efb229
new file mode 100644
index 0000000000000000000000000000000000000000..5b3793ee96787f77a8823d4901b0ae8cbf6ef16a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/d/cc8422e80f8c1f15c7def16672efb229 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0799a7b34593222e505dd81725d9d9e7d4279d10
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/06b12b6fdef59dbafd40ed197948b3d9 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/06b12b6fdef59dbafd40ed197948b3d9
new file mode 100644
index 0000000000000000000000000000000000000000..1338ca14142404019773d6675bbbe4c0e5358199
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/06b12b6fdef59dbafd40ed197948b3d9
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c5d4eca776822f6a560740c9d034daa0c0b743df1f6d4404dd1fde35af1a5dcd
+size 374572
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/3e21a8bbbf4a3aad8fd779490c8bf102 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/3e21a8bbbf4a3aad8fd779490c8bf102
new file mode 100644
index 0000000000000000000000000000000000000000..3526d600f540005026f94083c3229b48e28e8150
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/3e21a8bbbf4a3aad8fd779490c8bf102 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/457618e05145e1196a9005e841bdfba9 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/457618e05145e1196a9005e841bdfba9
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/457618e05145e1196a9005e841bdfba9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/5b8d60927389d9f66cc1d47cc225452d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/5b8d60927389d9f66cc1d47cc225452d
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/5b8d60927389d9f66cc1d47cc225452d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/951c113c7aef47cd3db564b5aff358fe b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/951c113c7aef47cd3db564b5aff358fe
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/951c113c7aef47cd3db564b5aff358fe differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/ac4e3bee1263dd7ab41f2fc828f85e13 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/ac4e3bee1263dd7ab41f2fc828f85e13
new file mode 100644
index 0000000000000000000000000000000000000000..bfb27ae31a8edd5d1b85abe8c2ef58b79bcea295
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/ac4e3bee1263dd7ab41f2fc828f85e13
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:006b546e71cc3e3a1fc7ee61cea74904c19346b7810dbd9947930778b0dd701c
+size 363897
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/d75297840134bfb6cf1dafd626f8aae3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/d75297840134bfb6cf1dafd626f8aae3
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d/d75297840134bfb6cf1dafd626f8aae3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..da615ad1212d7df85cabdbafc40e32f21a760e39
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..b7510e67d621f655647c6dabc6e4f1f1418a5bc2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587054344588505, "commit_timestamp_nsecs": 1754587054484329328, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/d/265c20e2f79ca0137dc83af07102ce48 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/d/265c20e2f79ca0137dc83af07102ce48
new file mode 100644
index 0000000000000000000000000000000000000000..b788c698da04215411277270e6d51aad8a7d9cfc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/d/265c20e2f79ca0137dc83af07102ce48 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..294f6da412204ee2a0eebfb6430aad39392cdb0b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/3806e189f31cfffb566bb02ece6e2980 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/3806e189f31cfffb566bb02ece6e2980
new file mode 100644
index 0000000000000000000000000000000000000000..ec745d97353f97209ef50f0eb738585830219f53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/3806e189f31cfffb566bb02ece6e2980 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/6fd262dc072eb1ba66710b28aed5aa60 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/6fd262dc072eb1ba66710b28aed5aa60
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/6fd262dc072eb1ba66710b28aed5aa60 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/be8bbbd819a7ee901ddbcf6fdb3a7142 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/be8bbbd819a7ee901ddbcf6fdb3a7142
new file mode 100644
index 0000000000000000000000000000000000000000..152913ed35536949fe2f7b85b413ede8f17dc8f8
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/be8bbbd819a7ee901ddbcf6fdb3a7142
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b1b3915eaff38cb7ddb3dbd47b18f1c97a49642b28b891738a9e7a07e47b2340
+size 346903
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/cd89905e4e62e2b595e40ab86c32abc2 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/cd89905e4e62e2b595e40ab86c32abc2
new file mode 100644
index 0000000000000000000000000000000000000000..469f5db805d46f9965c31cc8494dbd4a6bd8d698
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/cd89905e4e62e2b595e40ab86c32abc2
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c70ceceb79603d2af35b76045be0b20a05ffa339a3ca4a7a89f903ad31bcf3fa
+size 353649
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/cff9157b2baa7965357a38377106cc1c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/cff9157b2baa7965357a38377106cc1c
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/cff9157b2baa7965357a38377106cc1c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/e0def300c470d526fdc6681b86ea195d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/e0def300c470d526fdc6681b86ea195d
new file mode 100644
index 0000000000000000000000000000000000000000..0b3a20bc6f4126b0fc18f4f62577179dee9bd550
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/e0def300c470d526fdc6681b86ea195d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/f7e51a9b3524741077a42a2713142b87 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/f7e51a9b3524741077a42a2713142b87
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d/f7e51a9b3524741077a42a2713142b87 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..afe7928017d5bcf772833dc6440153158c3eb22f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..5c1c1f388affbdc1413bc00bff1e73ecd7951ddd
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587088979876275, "commit_timestamp_nsecs": 1754587089148584481, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/d/079d9b8095ddf4bc5024bd14d18d4179 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/d/079d9b8095ddf4bc5024bd14d18d4179
new file mode 100644
index 0000000000000000000000000000000000000000..67f483e6d58d7cf9bc4039b02b188e8414438daa
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/d/079d9b8095ddf4bc5024bd14d18d4179 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..29e7d3de957dea3e76906e68a0e5b8b266f1d8c9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/0c5c377eb29c955a4542d608e00c805d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/0c5c377eb29c955a4542d608e00c805d
new file mode 100644
index 0000000000000000000000000000000000000000..4ced9830e02f8051dc976128bbccc23820cb9958
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/0c5c377eb29c955a4542d608e00c805d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/6317d4f942de560b76ef52cccfd9dc07 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/6317d4f942de560b76ef52cccfd9dc07
new file mode 100644
index 0000000000000000000000000000000000000000..dfbfe1a61305bab6f716f83c77874d9e3176999d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/6317d4f942de560b76ef52cccfd9dc07 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/64f61f98cd6eca248e82768de1347ffe b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/64f61f98cd6eca248e82768de1347ffe
new file mode 100644
index 0000000000000000000000000000000000000000..2bea068eb229422fc30ab3d93220563d29830083
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/64f61f98cd6eca248e82768de1347ffe
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c521df68cad28e6c2711297b46915c173d3a5fda6628c36696d8ca24160e5ad2
+size 278876
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/67105c903f92bbdd84019df2832f15f0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/67105c903f92bbdd84019df2832f15f0
new file mode 100644
index 0000000000000000000000000000000000000000..a2f4185dfa921eeb3e2efbcd68b7659967f211cf
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/67105c903f92bbdd84019df2832f15f0
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e6d636ec10a886348e4313c0e49b9f05e11c41bf5c8fa4ed0fd7ce2895982fc6
+size 459720
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/7c5ff3c050b901a47c301a937ebd1b92 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/7c5ff3c050b901a47c301a937ebd1b92
new file mode 100644
index 0000000000000000000000000000000000000000..2a31377ca030aced51ef70fb458c6b5c47404cd3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/7c5ff3c050b901a47c301a937ebd1b92 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/8f1859a9d74a9aa9f8c43d3db1220184 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/8f1859a9d74a9aa9f8c43d3db1220184
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/8f1859a9d74a9aa9f8c43d3db1220184 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/91b9fe3eb7df2fc5c057be8922a965a3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/91b9fe3eb7df2fc5c057be8922a965a3
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/91b9fe3eb7df2fc5c057be8922a965a3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/95df4d987a9209ffe667c14fe5d3f825 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/95df4d987a9209ffe667c14fe5d3f825
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/95df4d987a9209ffe667c14fe5d3f825 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/e1bb3ccfc0f2ecf58ec9d556fc6f3f5c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/e1bb3ccfc0f2ecf58ec9d556fc6f3f5c
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/e1bb3ccfc0f2ecf58ec9d556fc6f3f5c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/faa6ed372ce88651735f388074cfde24 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/faa6ed372ce88651735f388074cfde24
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d/faa6ed372ce88651735f388074cfde24 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5bdc5fef40e0c0a7cf6842760baf458aa2b789a3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..57dbe0607c1e37d3c64afe05c460b61e61fd021c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587123637279972, "commit_timestamp_nsecs": 1754587123762080191, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/d/1f16f1e1f6c968b60c4d0f1638bfdf06 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/d/1f16f1e1f6c968b60c4d0f1638bfdf06
new file mode 100644
index 0000000000000000000000000000000000000000..d22821c2929ef5828f1bbdff5a49ed0bb3730634
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/d/1f16f1e1f6c968b60c4d0f1638bfdf06 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1a3f4e5a9424fd126f42970d8570a2fec0d40a5f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/04a858d6423c615365176d6b9573745f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/04a858d6423c615365176d6b9573745f
new file mode 100644
index 0000000000000000000000000000000000000000..5b421ae468a52a87b9dad52d1f30d6c67e56f627
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/04a858d6423c615365176d6b9573745f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9767207fbb81ea5fe4fe0fbcabfa692e5ed708c8120d7eba02216b32eb2340ea
+size 247336
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/09ee26478f7a2c560674f8d00fa790c8 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/09ee26478f7a2c560674f8d00fa790c8
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/09ee26478f7a2c560674f8d00fa790c8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/0f9bf86a57551992814b4526bf82fc8f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/0f9bf86a57551992814b4526bf82fc8f
new file mode 100644
index 0000000000000000000000000000000000000000..ba8215ec6215bebc2ceaf3293e9d271e62091da9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/0f9bf86a57551992814b4526bf82fc8f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fdcec841b4d13e65ed170f5565ae3df09936477d90b5259d0753ce5247432c2b
+size 159885
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/209d981bd635b293e04638950d81ce47 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/209d981bd635b293e04638950d81ce47
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/209d981bd635b293e04638950d81ce47 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/54c62bd47fda70c31ed8aeecea36c885 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/54c62bd47fda70c31ed8aeecea36c885
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/54c62bd47fda70c31ed8aeecea36c885 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/64f90bab4897565c6c9de4345969a34f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/64f90bab4897565c6c9de4345969a34f
new file mode 100644
index 0000000000000000000000000000000000000000..66928cf48cbff5638713814f2fa3d713c4d66dcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/64f90bab4897565c6c9de4345969a34f differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/9f3cefaf47e0993e6a01bf03a3457a45 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/9f3cefaf47e0993e6a01bf03a3457a45
new file mode 100644
index 0000000000000000000000000000000000000000..4ee8ae1564e70c35496345dc2348ed2df510b239
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/9f3cefaf47e0993e6a01bf03a3457a45
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:15fc45502a2f186e754b96c0d4c84aa153ed241adcdbff2b148f24c23c403f89
+size 296236
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/a8b74a127f61e3751d2389be34c646c7 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/a8b74a127f61e3751d2389be34c646c7
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d/a8b74a127f61e3751d2389be34c646c7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a33b14952fafd0f2b73f0d9f3a5f342fb587c6d3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..604efc4ad804c018d071ea604b8ea1da590ac28d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587157861802848, "commit_timestamp_nsecs": 1754587158038220211, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/d/4d2f7e760ed1e557220a879c747db2dc b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/d/4d2f7e760ed1e557220a879c747db2dc
new file mode 100644
index 0000000000000000000000000000000000000000..967f46701ed4738f4a6283ccd467acafabc6027e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/d/4d2f7e760ed1e557220a879c747db2dc differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..21680a959120e06ac34df1093e7ecc36e18b14f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/204aafaa9e97f5a4f366073245a9dd74 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/204aafaa9e97f5a4f366073245a9dd74
new file mode 100644
index 0000000000000000000000000000000000000000..430f9b245db30dff92e9b2420d9021ffc12b31eb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/204aafaa9e97f5a4f366073245a9dd74
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6cb7544eb3e52bc1d8e5c5ce4395bacecb8166bde19b37825b9167ebb35ea3f2
+size 468277
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/62c0a239d00441dba66d73100afcb617 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/62c0a239d00441dba66d73100afcb617
new file mode 100644
index 0000000000000000000000000000000000000000..2a31377ca030aced51ef70fb458c6b5c47404cd3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/62c0a239d00441dba66d73100afcb617 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/7d901eb8b5ffe76602d11b0ccaffc408 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/7d901eb8b5ffe76602d11b0ccaffc408
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/7d901eb8b5ffe76602d11b0ccaffc408 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/99e350134604adc95dde27378b5c5e2e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/99e350134604adc95dde27378b5c5e2e
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/99e350134604adc95dde27378b5c5e2e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/a58a54332f297aecfebd58f88e922d0f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/a58a54332f297aecfebd58f88e922d0f
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/a58a54332f297aecfebd58f88e922d0f differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/ad3b497ed3ff3c6bfc9cbc98f078e486 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/ad3b497ed3ff3c6bfc9cbc98f078e486
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/ad3b497ed3ff3c6bfc9cbc98f078e486 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/dabc7437f3d977b74c583d86cc7486eb b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/dabc7437f3d977b74c583d86cc7486eb
new file mode 100644
index 0000000000000000000000000000000000000000..6a0cb443d4df22ae66622add419c8576ba15f880
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/dabc7437f3d977b74c583d86cc7486eb differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/f81d7b092b75728d27c71051fa16aaca b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/f81d7b092b75728d27c71051fa16aaca
new file mode 100644
index 0000000000000000000000000000000000000000..afbaad99c43cbd6c21d45ae4b5decb0cb2a73e83
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/f81d7b092b75728d27c71051fa16aaca
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e30be72dba3212383d28b8e84ced9e98e83065e5827d83a1905d46c57cc86f0f
+size 270382
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/fa728d2bf538f33a6b5ba63abdd5cefd b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/fa728d2bf538f33a6b5ba63abdd5cefd
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d/fa728d2bf538f33a6b5ba63abdd5cefd differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..53f1f4449ec156704ad624accf53439b33efb368
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..808afde573b1159a2d8712869176a9ac1e09513f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587193056258348, "commit_timestamp_nsecs": 1754587193225479751, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/d/55a3d451e7224c7b965631a1a2a6150c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/d/55a3d451e7224c7b965631a1a2a6150c
new file mode 100644
index 0000000000000000000000000000000000000000..f82d2ee8a39fcd4f5a59d57546e1c3edb6f0f270
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/d/55a3d451e7224c7b965631a1a2a6150c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b1af4dbbabdfa3de64d6cc1839f0815a1d5840d8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/2653eaea8f71ebe235cff97b786e2dc0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/2653eaea8f71ebe235cff97b786e2dc0
new file mode 100644
index 0000000000000000000000000000000000000000..4fe6190ed8e00bca84c57e7a23a181dd3e729400
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/2653eaea8f71ebe235cff97b786e2dc0
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d2438b70cf098bd47b91fe0568ec034c4cec92e5b014f1ef060035a50514ed00
+size 112800
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/50deb127f4c7677caf963d0818839e81 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/50deb127f4c7677caf963d0818839e81
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/50deb127f4c7677caf963d0818839e81 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/70986a5e19fa7fa014831d76516fa01e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/70986a5e19fa7fa014831d76516fa01e
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/70986a5e19fa7fa014831d76516fa01e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/7822a51da8e96b79354d7188e90b628c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/7822a51da8e96b79354d7188e90b628c
new file mode 100644
index 0000000000000000000000000000000000000000..a6d62d6521aeb849889c4bf09a3bef48f9106eea
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/7822a51da8e96b79354d7188e90b628c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5ab9cdf43709581f8571211d6b53e189195369b0e42ed102b469657d40b48f14
+size 522679
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/8f24b37927ea4e62212badf7fa1fe853 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/8f24b37927ea4e62212badf7fa1fe853
new file mode 100644
index 0000000000000000000000000000000000000000..1e7f99001df4bb05496766b9299de938da5c3bf1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/8f24b37927ea4e62212badf7fa1fe853 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/a367c55d7ce563112448d8edfff098f6 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/a367c55d7ce563112448d8edfff098f6
new file mode 100644
index 0000000000000000000000000000000000000000..9556e42d7744f5c2a2007990c3563a0c685885d3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/a367c55d7ce563112448d8edfff098f6
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:60d3909b27e0e56e934c0a637c3cdb5bd085cc683899344399fd2451769de59a
+size 106473
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/f999da1bd2ddca4732ab9356c541c2d0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/f999da1bd2ddca4732ab9356c541c2d0
new file mode 100644
index 0000000000000000000000000000000000000000..2a31377ca030aced51ef70fb458c6b5c47404cd3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d/f999da1bd2ddca4732ab9356c541c2d0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3dd10354eb5cc4a58a3690f5633722e524ea9396
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..11a02d98bba4a5df250002d4c255935dbb47725f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587228191254388, "commit_timestamp_nsecs": 1754587228362302711, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/d/ff493da0742bd31db5001dd5a49e0f21 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/d/ff493da0742bd31db5001dd5a49e0f21
new file mode 100644
index 0000000000000000000000000000000000000000..8a8b767e9419695f80200074b83b7a0cf6ec81b5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/d/ff493da0742bd31db5001dd5a49e0f21 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..fcfb96a0101903d3dedf2cdd246b144531d96ead
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/25dc820a2aa2e481f8c2d9e5bc902b18 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/25dc820a2aa2e481f8c2d9e5bc902b18
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/25dc820a2aa2e481f8c2d9e5bc902b18 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/38e8f740ad6ed2f527f4eb9df8575081 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/38e8f740ad6ed2f527f4eb9df8575081
new file mode 100644
index 0000000000000000000000000000000000000000..61f0f83e1bbf3ecdf976d68ef48d7895eae74806
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/38e8f740ad6ed2f527f4eb9df8575081
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6afc012721e57faf4709667362d82b02c5bd366b0c038c84417dcacc53a9df0c
+size 304638
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/5f2719ad6efa52043edd48048b18eb2e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/5f2719ad6efa52043edd48048b18eb2e
new file mode 100644
index 0000000000000000000000000000000000000000..ac47b8f5e36ec073c36260e0b6b33e6e0458f1d4
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/5f2719ad6efa52043edd48048b18eb2e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e871919b1333f90f426857868b9dabaeca43e71fbb83eab9315ec71ae0a3586c
+size 400145
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/6d3c31f31607680c75b8b30fdf74e383 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/6d3c31f31607680c75b8b30fdf74e383
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/6d3c31f31607680c75b8b30fdf74e383 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/9c3c0de86f9dc82431ffb2b87c158d90 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/9c3c0de86f9dc82431ffb2b87c158d90
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/9c3c0de86f9dc82431ffb2b87c158d90 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/d2f41c2a2d89c3671feb797ee14775e5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/d2f41c2a2d89c3671feb797ee14775e5
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/d2f41c2a2d89c3671feb797ee14775e5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/d5d80c6cf170b793434b1fbebc1e0da2 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/d5d80c6cf170b793434b1fbebc1e0da2
new file mode 100644
index 0000000000000000000000000000000000000000..e1b5d9589e655dc9e7ec5e16346cc14281e33e17
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/d5d80c6cf170b793434b1fbebc1e0da2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/e62d272c55f30cc2d7cb7dad1a0725a0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/e62d272c55f30cc2d7cb7dad1a0725a0
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d/e62d272c55f30cc2d7cb7dad1a0725a0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c4fff6fe47d768e7daffc5020525d2f1aefa2ea1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..66f323ded9079570895525b7b7d0c8c69d490fb7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587263060842051, "commit_timestamp_nsecs": 1754587263256620852, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/d/df2f053c816a15ad3c197710d5595e41 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/d/df2f053c816a15ad3c197710d5595e41
new file mode 100644
index 0000000000000000000000000000000000000000..8f2f263ccf9b14da75795122ccf5526ad67e85c7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/d/df2f053c816a15ad3c197710d5595e41 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..159191c738d999266839d3583274ba6303e4db1a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/003ebe1c7d45f3d1eef095cb6fc4f747 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/003ebe1c7d45f3d1eef095cb6fc4f747
new file mode 100644
index 0000000000000000000000000000000000000000..632be3957d11b95524b6895848d9bf91d6a7ee48
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/003ebe1c7d45f3d1eef095cb6fc4f747
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4e3cc5a4839bf4a88beae27dc51c75ef08eedea6028efa73bea6da232af69931
+size 466527
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/226dc1161acec5ee49dab45e35f796aa b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/226dc1161acec5ee49dab45e35f796aa
new file mode 100644
index 0000000000000000000000000000000000000000..91d5fde6ad4171c6257fe96d2c8f6a3d2a6a379c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/226dc1161acec5ee49dab45e35f796aa differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/241b852d798162273711873d6b392179 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/241b852d798162273711873d6b392179
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/241b852d798162273711873d6b392179 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/3e9f9de2869f8977ee676fb15434f01c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/3e9f9de2869f8977ee676fb15434f01c
new file mode 100644
index 0000000000000000000000000000000000000000..f28d59e1f9187cfd65b688b4bceb57a81f0acdcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/3e9f9de2869f8977ee676fb15434f01c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/410b76a203d615d731fb6279f9a10320 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/410b76a203d615d731fb6279f9a10320
new file mode 100644
index 0000000000000000000000000000000000000000..93e5680a6325db0b0a7996bb966bbbd68df2fb80
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/410b76a203d615d731fb6279f9a10320
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2b858941937734c42b7058652d946fb382860a625e9e3b0044787e4765217136
+size 270329
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/4e0917e1b9e06edf2101ecf563b174fc b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/4e0917e1b9e06edf2101ecf563b174fc
new file mode 100644
index 0000000000000000000000000000000000000000..2b03d141156751d24ecb313e49acee489d277c9c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/4e0917e1b9e06edf2101ecf563b174fc differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/55cd10d15b66cd8112ff0317e686cf4e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/55cd10d15b66cd8112ff0317e686cf4e
new file mode 100644
index 0000000000000000000000000000000000000000..ec745d97353f97209ef50f0eb738585830219f53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/55cd10d15b66cd8112ff0317e686cf4e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/9ca547f92235be2ea222a8bcb6680765 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/9ca547f92235be2ea222a8bcb6680765
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/9ca547f92235be2ea222a8bcb6680765 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/ca86fcca02c387470ccc080d135045ae b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/ca86fcca02c387470ccc080d135045ae
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d/ca86fcca02c387470ccc080d135045ae differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9257e3a8adf70e0129a255348dc296552602fc8f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..3638663a4e2a63db2ef3f6f3f129497da3b688ad
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587298244723372, "commit_timestamp_nsecs": 1754587298387905679, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/d/d0254c17171a09073acc628bbd3b3834 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/d/d0254c17171a09073acc628bbd3b3834
new file mode 100644
index 0000000000000000000000000000000000000000..8573230c157445f201c33aa83e2f1224a776e605
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/d/d0254c17171a09073acc628bbd3b3834 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9b624f7f44a9b35eb47c2c8106c4a2a94b84cfa1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/102e7cf90601125662779a1d6bac3a5c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/102e7cf90601125662779a1d6bac3a5c
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/102e7cf90601125662779a1d6bac3a5c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/11731c61b0bb81ca10a6b1c1a827b05f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/11731c61b0bb81ca10a6b1c1a827b05f
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/11731c61b0bb81ca10a6b1c1a827b05f differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/42c5e7f8cdcdc3f2b3210f601b5b4f00 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/42c5e7f8cdcdc3f2b3210f601b5b4f00
new file mode 100644
index 0000000000000000000000000000000000000000..6d0ba8e9fd7de3f1123db64e4f0072cd83c3b657
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/42c5e7f8cdcdc3f2b3210f601b5b4f00 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/994696f7d15fef7b1480f5a55598fef5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/994696f7d15fef7b1480f5a55598fef5
new file mode 100644
index 0000000000000000000000000000000000000000..04cd3b644650f62ae7d27ea7fef6a95af635ceec
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/994696f7d15fef7b1480f5a55598fef5
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8e32d4b8122ad1b1a9b7efc6588b446eb74a3e856ee1d60af9b7ea338ae9044d
+size 388177
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/a0ac001647cc63d3dd2f791b2adf447c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/a0ac001647cc63d3dd2f791b2adf447c
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/a0ac001647cc63d3dd2f791b2adf447c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/c0a3d348ae056ea7f9157b2650a524aa b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/c0a3d348ae056ea7f9157b2650a524aa
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/c0a3d348ae056ea7f9157b2650a524aa differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/dfd9bc3fae6f46795db0c88f9ca9591b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/dfd9bc3fae6f46795db0c88f9ca9591b
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/dfd9bc3fae6f46795db0c88f9ca9591b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/edf5414e53554dec7d974b4daa220128 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/edf5414e53554dec7d974b4daa220128
new file mode 100644
index 0000000000000000000000000000000000000000..82c777a70648a6e013e1b9065adfc5a2ced5aa36
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d/edf5414e53554dec7d974b4daa220128
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f79a19f865ec097554b3cdc0fbe6d936bd95ab28243c6a13e246eec34d85b81e
+size 278862
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a4dac8723361c7b627eeab1f5ee7e9a8e5b69c99
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..9a7135c2e098abf56471a47a9d7181e3f9ddfcd8
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587333231851672, "commit_timestamp_nsecs": 1754587333373467124, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/d/afe6d12820941e209a2ecf0579cae9fc b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/d/afe6d12820941e209a2ecf0579cae9fc
new file mode 100644
index 0000000000000000000000000000000000000000..a8b43cec1c5cb731d2c59edb7a483dff0acf8689
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/d/afe6d12820941e209a2ecf0579cae9fc differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..7b5ce788e60d54db51c10b3342b853fefc58b2e6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/09050c8f4d4e2f9afb50135a1082711e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/09050c8f4d4e2f9afb50135a1082711e
new file mode 100644
index 0000000000000000000000000000000000000000..9a844aaeb09683afa0aab22b7dd80bdefba52dba
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/09050c8f4d4e2f9afb50135a1082711e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:defb4d28779feeab31511a5eb07236ed6e32ab282dad8f846c5350791e9b8b8a
+size 493861
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/3b687b59febe84d37f1b74f810e6e5a6 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/3b687b59febe84d37f1b74f810e6e5a6
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/3b687b59febe84d37f1b74f810e6e5a6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/5cd52a6923a4d938a2440492975a859b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/5cd52a6923a4d938a2440492975a859b
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/5cd52a6923a4d938a2440492975a859b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/790c41138548a9becad36ccbbecbaa3c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/790c41138548a9becad36ccbbecbaa3c
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/790c41138548a9becad36ccbbecbaa3c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/7b27de7362010e2aeb09d914ee09eb85 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/7b27de7362010e2aeb09d914ee09eb85
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/7b27de7362010e2aeb09d914ee09eb85 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/a33f9d16ab49bca251958da5ad49b98f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/a33f9d16ab49bca251958da5ad49b98f
new file mode 100644
index 0000000000000000000000000000000000000000..36f238c56b8b2ec5bf2622b4f0c861fc589324c6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/a33f9d16ab49bca251958da5ad49b98f differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/db216bc1bf3b7c426c6e95ebeba231df b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/db216bc1bf3b7c426c6e95ebeba231df
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/db216bc1bf3b7c426c6e95ebeba231df differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/de0c65931eba3faefeb3256bd8cf2b6e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/de0c65931eba3faefeb3256bd8cf2b6e
new file mode 100644
index 0000000000000000000000000000000000000000..1c91d7b09585f54f71b7618377c95b95dd12c912
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d/de0c65931eba3faefeb3256bd8cf2b6e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fc76e65643003ae1f97d134f824a85ee7e9f82f05bdc98ee58b4630926a048d2
+size 244920
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..80e5a8c5218a7906b9c38b23f0b6e0107e0bb52c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..b0059f88122adbaac6df5c90a64f2c7596fae8e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754586409363506031, "commit_timestamp_nsecs": 1754586409534661630, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/d/f69f12ed8ad024f95e979c4c47d3f37d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/d/f69f12ed8ad024f95e979c4c47d3f37d
new file mode 100644
index 0000000000000000000000000000000000000000..b27a331200a2b6a9d0828b2b95cb8670c7927945
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/d/f69f12ed8ad024f95e979c4c47d3f37d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0abdccb0ec70444225192213fad23bfe0275842c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/05478280a6caa49467d03feca15ac89b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/05478280a6caa49467d03feca15ac89b
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/05478280a6caa49467d03feca15ac89b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/289c8f1b35ff5f9265dece44c44d5c60 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/289c8f1b35ff5f9265dece44c44d5c60
new file mode 100644
index 0000000000000000000000000000000000000000..f74893bf14739ac1a1181a3f3012e5802f42a396
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/289c8f1b35ff5f9265dece44c44d5c60
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b5c8e306c77417991d5dd07964229c092706bd977615dc2c6c38a98e0607e33b
+size 408351
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/551b5220c2e416814530aa9b6b367bb7 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/551b5220c2e416814530aa9b6b367bb7
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/551b5220c2e416814530aa9b6b367bb7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/6a48b4d01820d8edf90ccab671c4efef b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/6a48b4d01820d8edf90ccab671c4efef
new file mode 100644
index 0000000000000000000000000000000000000000..b2db7d921ecacc7cbd643bb52c59631c36bc02d8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/6a48b4d01820d8edf90ccab671c4efef differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/6fe02651e578683b761f0570eb7f2e63 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/6fe02651e578683b761f0570eb7f2e63
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/6fe02651e578683b761f0570eb7f2e63 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/83065b4df2191af52859846ec0f3a4d1 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/83065b4df2191af52859846ec0f3a4d1
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/83065b4df2191af52859846ec0f3a4d1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/b6d930a8709912426d04d262475faed2 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/b6d930a8709912426d04d262475faed2
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/b6d930a8709912426d04d262475faed2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/bec03153b9de8b9a95e12b35cf515a98 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/bec03153b9de8b9a95e12b35cf515a98
new file mode 100644
index 0000000000000000000000000000000000000000..1f0de9225cdaac6ba74fa5f9f37078ca94b9c274
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d/bec03153b9de8b9a95e12b35cf515a98
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bdb4457be179b5c68e65e2973668d28ea17f59ea6b3c8c5bc4968a69f1b136e3
+size 287242
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..72933de3ebadd621fb54828416eb9013008b074f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..d3c99639ead9ae93ecaedc36f1040814801a6283
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587368978580046, "commit_timestamp_nsecs": 1754587369149720737, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/d/d42f308e3435f6b1e4e8b6e539a972d2 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/d/d42f308e3435f6b1e4e8b6e539a972d2
new file mode 100644
index 0000000000000000000000000000000000000000..3d9b16ec96574064d5b32c83ae25d8f404fab6e0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/d/d42f308e3435f6b1e4e8b6e539a972d2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..51d186883fb9ae91d29a17fd8f3dceb1425c5f7f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/7935a307f25f7db8c16231f57eae4dcf b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/7935a307f25f7db8c16231f57eae4dcf
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/7935a307f25f7db8c16231f57eae4dcf differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/9768b642c339cfb8d5996519c80e5ad0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/9768b642c339cfb8d5996519c80e5ad0
new file mode 100644
index 0000000000000000000000000000000000000000..5fa7d22a2151befcb22c393cb8ad2acf9749f284
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/9768b642c339cfb8d5996519c80e5ad0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/a2a9172d083ebb30ab3f895855244eb2 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/a2a9172d083ebb30ab3f895855244eb2
new file mode 100644
index 0000000000000000000000000000000000000000..f62e9b9004d2e749ccd273a3d0fa9cd6fb36cb34
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/a2a9172d083ebb30ab3f895855244eb2
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d0672a2a5bcd9632982fd4352f1487670c52735778792f40d50452485745a7d7
+size 438805
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/aa2482ce0148df1123ea491e7eeeb3d2 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/aa2482ce0148df1123ea491e7eeeb3d2
new file mode 100644
index 0000000000000000000000000000000000000000..bdd1f5c43376bd72f062842a864118cd8f3637c2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/aa2482ce0148df1123ea491e7eeeb3d2
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d609718e96205ff85936400a454937ef0bfe558194da0744d228e26e6ba33de4
+size 262070
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/aaa41374ff8b4ea1b759aea0f9c9ffc7 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/aaa41374ff8b4ea1b759aea0f9c9ffc7
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/aaa41374ff8b4ea1b759aea0f9c9ffc7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/ab9b08427ed415b92514f0f435239639 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/ab9b08427ed415b92514f0f435239639
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/ab9b08427ed415b92514f0f435239639 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/cf71fb1ff14188db19b2d26816ceb2a1 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/cf71fb1ff14188db19b2d26816ceb2a1
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/cf71fb1ff14188db19b2d26816ceb2a1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/ed259ec8d63d109a81755718cd02f39b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/ed259ec8d63d109a81755718cd02f39b
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d/ed259ec8d63d109a81755718cd02f39b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..44e5fa458606e9f65b05e68a67aeb7ba8be7f580
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..ad94b49f3553e890bf0fa9f9c88e6070b73010b2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587403747476277, "commit_timestamp_nsecs": 1754587403923438651, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/d/4decda015a65291725205ae312832088 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/d/4decda015a65291725205ae312832088
new file mode 100644
index 0000000000000000000000000000000000000000..40150b903f15cd7dce5631c2fe0b7630198b4894
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/d/4decda015a65291725205ae312832088 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d9b46be712b975fff622f98662af93988a0902f8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/4899e287f7a797f7fd02f8f5ab7886a6 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/4899e287f7a797f7fd02f8f5ab7886a6
new file mode 100644
index 0000000000000000000000000000000000000000..2dd7476079cf4105a4c925380464c90a342fc1f4
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/4899e287f7a797f7fd02f8f5ab7886a6
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3714d2301d3cca9848c6a64c4a6baf6593964789fadf3497524034f13cc794e6
+size 295966
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/53b23ebc4deb0c536120db9a8729edd0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/53b23ebc4deb0c536120db9a8729edd0
new file mode 100644
index 0000000000000000000000000000000000000000..705ee5b1153fee56bea4d93a207232308453f3af
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/53b23ebc4deb0c536120db9a8729edd0
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:30a113496841d4f18c7fd0f5f94080411be5301554a654f1c7371d04ae1ea2e4
+size 442598
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/649d21cb5f75dbd209ba859385bd1a81 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/649d21cb5f75dbd209ba859385bd1a81
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/649d21cb5f75dbd209ba859385bd1a81 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/65f6b385861a42791ed8b6bc353ad4c5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/65f6b385861a42791ed8b6bc353ad4c5
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/65f6b385861a42791ed8b6bc353ad4c5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/7039bfeb8cb1ad532c5bb412f5012e64 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/7039bfeb8cb1ad532c5bb412f5012e64
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/7039bfeb8cb1ad532c5bb412f5012e64 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/a5fdfa798737adfc539c524fbdd0dc68 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/a5fdfa798737adfc539c524fbdd0dc68
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/a5fdfa798737adfc539c524fbdd0dc68 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/d115598d6e52d53d41a05c18283dab46 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/d115598d6e52d53d41a05c18283dab46
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/d115598d6e52d53d41a05c18283dab46 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/e90217c0680c2615903d45ceef5b67e7 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/e90217c0680c2615903d45ceef5b67e7
new file mode 100644
index 0000000000000000000000000000000000000000..ec62736f3a6961c45afb1c8e53f622cb863cc555
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d/e90217c0680c2615903d45ceef5b67e7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e32f169870708b5ea4f6c0324d63414a2db176e5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..1603948b68031ca469fbefbe1aca4bdc586f5b62
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587454845965322, "commit_timestamp_nsecs": 1754587455019332479, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/d/72d5389445a9cb5d99dc1ec4bf72a843 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/d/72d5389445a9cb5d99dc1ec4bf72a843
new file mode 100644
index 0000000000000000000000000000000000000000..d4130460a902503f31a6213f7556f466a99f0a03
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/d/72d5389445a9cb5d99dc1ec4bf72a843 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3ed61181221325773fb35bbf97363087c3c7522b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/120eb98393aa49e4049e4523c184c870 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/120eb98393aa49e4049e4523c184c870
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/120eb98393aa49e4049e4523c184c870 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/250a965233d96599e9ea36b52fe8e4a9 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/250a965233d96599e9ea36b52fe8e4a9
new file mode 100644
index 0000000000000000000000000000000000000000..bb7bd5d7fd2cca3919eae228b54763b2abca1a0f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/250a965233d96599e9ea36b52fe8e4a9
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2b240ddf23a8c0a6160a5579ecd0f26d3af338fb0f21a9d556c86dc4fb3e93f8
+size 227783
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/2526f53e4af0054891e2d565f572b697 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/2526f53e4af0054891e2d565f572b697
new file mode 100644
index 0000000000000000000000000000000000000000..79f65c97bab674b77530ce01ca771db5b479d313
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/2526f53e4af0054891e2d565f572b697 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/58a0c3b17ee88fa9644744456b024015 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/58a0c3b17ee88fa9644744456b024015
new file mode 100644
index 0000000000000000000000000000000000000000..6d1161fbdc821eba55057e0501f1540da4004968
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/58a0c3b17ee88fa9644744456b024015
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a45bd7a2c395f0ceb2e43754ff0f655f0ec2f4191b2ffdf46b3b264c2060ca94
+size 437406
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/66571710c1097f18fc235c91e75c78d0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/66571710c1097f18fc235c91e75c78d0
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/66571710c1097f18fc235c91e75c78d0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/b8c118f50c40fc69ba6bbc54ffdbd63a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/b8c118f50c40fc69ba6bbc54ffdbd63a
new file mode 100644
index 0000000000000000000000000000000000000000..65529f6532287fad4b450ca493011b55e8bf7f92
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/b8c118f50c40fc69ba6bbc54ffdbd63a differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/c702f2464f3bec2ef48ee569c23e63c0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/c702f2464f3bec2ef48ee569c23e63c0
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/c702f2464f3bec2ef48ee569c23e63c0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/e7cd97960798736ae32125fe555979e5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/e7cd97960798736ae32125fe555979e5
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d/e7cd97960798736ae32125fe555979e5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..bd4aec03f8dc56c1fc5ae792d6621d2dd184ecbc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..6db56304c3054d07565cc894f60075a953c3e807
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587488830420850, "commit_timestamp_nsecs": 1754587488974297539, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/d/3a91b96ba01f9d91fde90e00a4358f6a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/d/3a91b96ba01f9d91fde90e00a4358f6a
new file mode 100644
index 0000000000000000000000000000000000000000..8cd565adfa5ef3caaa1d50d2a10135cf41725d28
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/d/3a91b96ba01f9d91fde90e00a4358f6a differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..eebebb47f85810dd3a6c14cb1f8780bb0bc90fff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/379f4e5a20c3c30d095f3855fb5b2f0a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/379f4e5a20c3c30d095f3855fb5b2f0a
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/379f4e5a20c3c30d095f3855fb5b2f0a differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/5eb9a94e6de07001956827b7d593eb7f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/5eb9a94e6de07001956827b7d593eb7f
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/5eb9a94e6de07001956827b7d593eb7f differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/7eabd021a2592f46c6c5af161a71d2ae b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/7eabd021a2592f46c6c5af161a71d2ae
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/7eabd021a2592f46c6c5af161a71d2ae differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/a4ae84a1cbf83ad469477f76b3d4bbec b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/a4ae84a1cbf83ad469477f76b3d4bbec
new file mode 100644
index 0000000000000000000000000000000000000000..35ab0b02c091d10a5c29f66d26d0980206d7355a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/a4ae84a1cbf83ad469477f76b3d4bbec
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c3107ca088eabe1d87ee3a826d1ce5058d3d950ab90dc665bb8bcbbd6cf55ed4
+size 287474
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/aeca83e57974c5ab3bfa40d0dcad0d62 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/aeca83e57974c5ab3bfa40d0dcad0d62
new file mode 100644
index 0000000000000000000000000000000000000000..1741cd2d05f7fe4afbedad970b7ba6f1792abc73
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/aeca83e57974c5ab3bfa40d0dcad0d62 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/db1b6630160b51b5106b488dc2c6a1fb b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/db1b6630160b51b5106b488dc2c6a1fb
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/db1b6630160b51b5106b488dc2c6a1fb differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/e592dac11b4e7d2361e4e06e3be53537 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/e592dac11b4e7d2361e4e06e3be53537
new file mode 100644
index 0000000000000000000000000000000000000000..ec3d8dcd59f3932b658ff7919bbbda97f27f0e6e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/e592dac11b4e7d2361e4e06e3be53537
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:827562834ce80daf3b97da25fcfc0db2b3c5206a6083b2d1519ce97bef198778
+size 411470
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/ff6aa4b4f6a81b4b7ddaeb5c977b523e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/ff6aa4b4f6a81b4b7ddaeb5c977b523e
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d/ff6aa4b4f6a81b4b7ddaeb5c977b523e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a0499d8e493f7c402f955197fa0bd2f0db3f0720
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..ef869fd4009f445a12d3238de8671f33e3eadaec
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587523856546502, "commit_timestamp_nsecs": 1754587524037337493, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/d/ecf117e9d246432f1bbcbacb75c01eb0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/d/ecf117e9d246432f1bbcbacb75c01eb0
new file mode 100644
index 0000000000000000000000000000000000000000..ced344735b250bab37b6cf6cc203a33842442d77
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/d/ecf117e9d246432f1bbcbacb75c01eb0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..da2396638e972a562c2c9823d377e39b13d98547
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/30f71b52bdf9cc097eaf8ca9c0b5113c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/30f71b52bdf9cc097eaf8ca9c0b5113c
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/30f71b52bdf9cc097eaf8ca9c0b5113c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/340c50da00adbf2b78f937717f6fed0e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/340c50da00adbf2b78f937717f6fed0e
new file mode 100644
index 0000000000000000000000000000000000000000..8d2b30f79c8f4a86a4cc7e58ae7f805995fb8c55
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/340c50da00adbf2b78f937717f6fed0e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b673d453ad23c38f9a431facc877f2104d60c806383458e289813ce5de21fe1a
+size 122946
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/49eab01a75f387ac61902ebd64aaa085 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/49eab01a75f387ac61902ebd64aaa085
new file mode 100644
index 0000000000000000000000000000000000000000..8c0c36df02af665c812a14dcd5f63901ed4fc49b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/49eab01a75f387ac61902ebd64aaa085
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7039ce80d17c2c50c2d33ca08f4ebd7a23ba80d4dbdc44c2204330f9a4eb5be0
+size 290190
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/83010ee6c93ef0d0495abd698ae9400a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/83010ee6c93ef0d0495abd698ae9400a
new file mode 100644
index 0000000000000000000000000000000000000000..cb73ab4b59af46a07a9ef07accac0aba74150ba3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/83010ee6c93ef0d0495abd698ae9400a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ec227dc599fcfc9eb92c11b3400740e26a3521e0d4815d8e59f66b96176a51e7
+size 293838
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/8bf8b85ad96d10aae52672472e56a0b0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/8bf8b85ad96d10aae52672472e56a0b0
new file mode 100644
index 0000000000000000000000000000000000000000..b0da08803d3fe819df1f862aee80d3bc2691d513
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/8bf8b85ad96d10aae52672472e56a0b0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/99c0b24e3d91dfff76d77bd6957d6f11 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/99c0b24e3d91dfff76d77bd6957d6f11
new file mode 100644
index 0000000000000000000000000000000000000000..be24ab48f2522e8f736f46a150f0cad043c1be56
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/99c0b24e3d91dfff76d77bd6957d6f11 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/af036f8d760d8f7261734ef99f067518 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/af036f8d760d8f7261734ef99f067518
new file mode 100644
index 0000000000000000000000000000000000000000..0c54bbc08fcfa9afcd46c98fda5483078c87bfe6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/af036f8d760d8f7261734ef99f067518 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/bf0f2e3e84d0d20c6ac696fec054381a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/bf0f2e3e84d0d20c6ac696fec054381a
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d/bf0f2e3e84d0d20c6ac696fec054381a differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..6b9e6f067f38bea8e91fb7216a3f108c17e4f2f5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..6aa31485495075001246bb286b89d078f128fee0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587559033074167, "commit_timestamp_nsecs": 1754587559173444038, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/d/06787829fbd6e5f2c663b169736591c7 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/d/06787829fbd6e5f2c663b169736591c7
new file mode 100644
index 0000000000000000000000000000000000000000..13f25cadaae61a6e570fb7a6805e13992abef6ab
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/d/06787829fbd6e5f2c663b169736591c7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c4ffc2094592c33fd90111cdca26fc3cbf86b28c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/015bf18b668a13d9da9e649a9d9475db b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/015bf18b668a13d9da9e649a9d9475db
new file mode 100644
index 0000000000000000000000000000000000000000..bc77049fdf9e5e2eddfd0d756648ae3c3650a30e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/015bf18b668a13d9da9e649a9d9475db
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f5420fe69a61171a536eb958b295449f59baabebcd9a4ceeb2a89c83b2a314ce
+size 511950
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/369f4f10effe8a099ea40d0526989022 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/369f4f10effe8a099ea40d0526989022
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/369f4f10effe8a099ea40d0526989022 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/3e1709d190349fca23b942a9f7251bb1 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/3e1709d190349fca23b942a9f7251bb1
new file mode 100644
index 0000000000000000000000000000000000000000..d725fdea08c9fc3d63c3a667b62354ef56da0146
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/3e1709d190349fca23b942a9f7251bb1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/86560c710cee5f904be5a7ee754c1f55 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/86560c710cee5f904be5a7ee754c1f55
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/86560c710cee5f904be5a7ee754c1f55 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/a3634905d3c56fd181cf3ef2586de786 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/a3634905d3c56fd181cf3ef2586de786
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/a3634905d3c56fd181cf3ef2586de786 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/a84d3b35be3ad151935f8983b83620f1 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/a84d3b35be3ad151935f8983b83620f1
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/a84d3b35be3ad151935f8983b83620f1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/adc12c61446171b0baf87e0e947072ef b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/adc12c61446171b0baf87e0e947072ef
new file mode 100644
index 0000000000000000000000000000000000000000..af2e6903a423aed6716c33fc8f2e537f81efe705
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/adc12c61446171b0baf87e0e947072ef
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9c60a49973cb9448e4f3272f22068d6c204795afbf611a26a72d107c995688be
+size 227871
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/d57cac4f524fc7fc51fa155ccd5d8387 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/d57cac4f524fc7fc51fa155ccd5d8387
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d/d57cac4f524fc7fc51fa155ccd5d8387 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5f074ff89f2c52d340afdffda825b7b057e9f10d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..54f33759a22ab8558bd05cf2cc66e15484a7be1e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587594419654683, "commit_timestamp_nsecs": 1754587594576636713, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/d/6356b90c1015c9383d6439c388fdf8b0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/d/6356b90c1015c9383d6439c388fdf8b0
new file mode 100644
index 0000000000000000000000000000000000000000..93f4e5ba9e813245d7dffdb31e090b1849281350
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/d/6356b90c1015c9383d6439c388fdf8b0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4e8dcbcb71754584055feb70b0f58be7a86659bf
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/43dfa6091ad49f7f0b1a2b0c96d17e70 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/43dfa6091ad49f7f0b1a2b0c96d17e70
new file mode 100644
index 0000000000000000000000000000000000000000..0f13422395c54b0392f7fe595b27b873234af467
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/43dfa6091ad49f7f0b1a2b0c96d17e70 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/5cd7ad590f9678a7ef595bcd89c85eaa b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/5cd7ad590f9678a7ef595bcd89c85eaa
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/5cd7ad590f9678a7ef595bcd89c85eaa differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/6656a48039e7997884595fc93cffa8d2 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/6656a48039e7997884595fc93cffa8d2
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/6656a48039e7997884595fc93cffa8d2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/8111015a417cf15c593bff79a4af6934 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/8111015a417cf15c593bff79a4af6934
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/8111015a417cf15c593bff79a4af6934 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/ac7781cffe229f70f16b5c6fc48c3d8b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/ac7781cffe229f70f16b5c6fc48c3d8b
new file mode 100644
index 0000000000000000000000000000000000000000..d3476d558cff4cbe3c4c7c1438510791bf14495f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/ac7781cffe229f70f16b5c6fc48c3d8b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:89800147100056f1152ceb450f969da8ca3f26a0b5f0c0b7313f5da5bb23bdfa
+size 253550
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/b26893938da1b55e0f92dc9bed57c9ee b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/b26893938da1b55e0f92dc9bed57c9ee
new file mode 100644
index 0000000000000000000000000000000000000000..16f621f237e6789fac791188cab9e794992800cf
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/b26893938da1b55e0f92dc9bed57c9ee
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7a0283d197009b4c0a61ab2fa493ddb1fdd628c9379579962545864d320e3d64
+size 413514
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/dcb1d743984c5711aed9d46ea14275d1 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/dcb1d743984c5711aed9d46ea14275d1
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/dcb1d743984c5711aed9d46ea14275d1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/e2cec56cddb2970d083aa03adb77a589 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/e2cec56cddb2970d083aa03adb77a589
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d/e2cec56cddb2970d083aa03adb77a589 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..fa40b9bf9ae77c6d9aa6b6dd7c223f01df0a676c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..c22d3c091b98b2a070633b2142f6261db0784b27
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587629369131577, "commit_timestamp_nsecs": 1754587629541997481, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/d/e55c1be8107bccf261692a71f7fe83fe b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/d/e55c1be8107bccf261692a71f7fe83fe
new file mode 100644
index 0000000000000000000000000000000000000000..7d25b166d7c28b10cfafd9007ce2728f57bdce27
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/d/e55c1be8107bccf261692a71f7fe83fe differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..33cc5fe30339d397aabc38fe7f5d05a119ac344a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/0982d490d7a7ea20ff80e614f8ff62d6 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/0982d490d7a7ea20ff80e614f8ff62d6
new file mode 100644
index 0000000000000000000000000000000000000000..d5bfa565e5ceed5a89d95f7a9c56cd949c42ff99
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/0982d490d7a7ea20ff80e614f8ff62d6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/13c163c8b3bf5318245383c10b92298a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/13c163c8b3bf5318245383c10b92298a
new file mode 100644
index 0000000000000000000000000000000000000000..40dd30564d2341e3bd914d5e6b1a58078f42642e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/13c163c8b3bf5318245383c10b92298a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a080405020ddc00c4fa23269c8675201b0a200bce266c77fb21a0331b4ea6039
+size 287514
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/2e1022c553dc726c5cadd512e92dc436 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/2e1022c553dc726c5cadd512e92dc436
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/2e1022c553dc726c5cadd512e92dc436 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/4ac7b8a2e3de0822f742a2c2a048e5bb b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/4ac7b8a2e3de0822f742a2c2a048e5bb
new file mode 100644
index 0000000000000000000000000000000000000000..65c55cef1e0f044fdb725f402e95d9a23d03f6b1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/4ac7b8a2e3de0822f742a2c2a048e5bb differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/b1b0480e357121489130564e83b79eb4 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/b1b0480e357121489130564e83b79eb4
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/b1b0480e357121489130564e83b79eb4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/d6e340db8f5e265b01623e12218a7143 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/d6e340db8f5e265b01623e12218a7143
new file mode 100644
index 0000000000000000000000000000000000000000..b58f9db1e4af4e4e55a71af4bba46ec32fa89bad
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/d6e340db8f5e265b01623e12218a7143
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a23c174b727f026b9d4f1ea19a350631c9c61fdc47498d6c6ae9352107127581
+size 451212
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/db2a94431f489a2ed2441cbe6dd24bb8 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/db2a94431f489a2ed2441cbe6dd24bb8
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/db2a94431f489a2ed2441cbe6dd24bb8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/db3706c4dfd532955acf9537c60f3409 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/db3706c4dfd532955acf9537c60f3409
new file mode 100644
index 0000000000000000000000000000000000000000..85f35e7991cac7b6fc244bb2f277438119f4e90b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d/db3706c4dfd532955acf9537c60f3409 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8ce7c3950f64528ced14670c9a9f1e9b4c64ce05
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..b15337b60b5240d9e65fc54a280a5e03cd178360
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587664365550334, "commit_timestamp_nsecs": 1754587664533405372, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/d/fc0e84f786923243cf4e4db156b31424 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/d/fc0e84f786923243cf4e4db156b31424
new file mode 100644
index 0000000000000000000000000000000000000000..48cb27627002587113a37fb3cb288e2741f069d3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/d/fc0e84f786923243cf4e4db156b31424 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9f1936ceb76f10cb2d6b45fd0127a9479cdcb272
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/265e5179f838c1b3ebe1ff25859a2356 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/265e5179f838c1b3ebe1ff25859a2356
new file mode 100644
index 0000000000000000000000000000000000000000..f28d59e1f9187cfd65b688b4bceb57a81f0acdcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/265e5179f838c1b3ebe1ff25859a2356 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/971b720f884e1d8c0b72bd24bceb7e3b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/971b720f884e1d8c0b72bd24bceb7e3b
new file mode 100644
index 0000000000000000000000000000000000000000..28c16a00d288a259a8db541e91ca045345c69dfd
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/971b720f884e1d8c0b72bd24bceb7e3b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1b8ef3d756b171a65359fdef1d2937dd92930ec7b1dd9cb4ae5a078cb60d41d1
+size 330050
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/a7d64ad23e2e1c5a942544213acb72f7 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/a7d64ad23e2e1c5a942544213acb72f7
new file mode 100644
index 0000000000000000000000000000000000000000..5cb69c9544b1e8665af3a9ca9ac655f6079393ab
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/a7d64ad23e2e1c5a942544213acb72f7
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3e2d4afcce91a795567c9aa3779546e3047a11446d9da71270d0cfd0c6bb2840
+size 336789
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/ac13304fdff8a5071cb71ac5a310f1bd b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/ac13304fdff8a5071cb71ac5a310f1bd
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/ac13304fdff8a5071cb71ac5a310f1bd differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/b86c73649ed91b826adb7e72be721fa4 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/b86c73649ed91b826adb7e72be721fa4
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/b86c73649ed91b826adb7e72be721fa4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/bbbd613c87d3df3c190b968026b891d7 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/bbbd613c87d3df3c190b968026b891d7
new file mode 100644
index 0000000000000000000000000000000000000000..4859214e6b68fa6e697de9748417fcf43be8081d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/bbbd613c87d3df3c190b968026b891d7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/d961fa975f700d68ac50fdb183fede7b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/d961fa975f700d68ac50fdb183fede7b
new file mode 100644
index 0000000000000000000000000000000000000000..8c00fcff4d8644ab238cf7c8fc4af50371d60c47
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d/d961fa975f700d68ac50fdb183fede7b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0e1461b91d7cbdcfea3e3c3e6f4e383f9d869e0d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..203856392fff40fbbd0d776cd202933757540fbe
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587698839542438, "commit_timestamp_nsecs": 1754587698984598226, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/d/70d039f87380d5aff03de3639d40c1ea b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/d/70d039f87380d5aff03de3639d40c1ea
new file mode 100644
index 0000000000000000000000000000000000000000..6d644387f54e84549d9051b88cf53908a2d07854
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/d/70d039f87380d5aff03de3639d40c1ea differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..615322b9e054f72e8dbd0e0343e19c6dbcd6ae42
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/537ff4ddb21a26186a336f5bb972e7db b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/537ff4ddb21a26186a336f5bb972e7db
new file mode 100644
index 0000000000000000000000000000000000000000..7afe96ce884c33fe574c36477ca1105d0b56dcd3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/537ff4ddb21a26186a336f5bb972e7db
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:86f59fc8d50e0fdebe366ae91d1ed6473844cc94862395a6f1aef9df939a7f65
+size 507341
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/580e17208c40aaab081180c3e855e4b6 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/580e17208c40aaab081180c3e855e4b6
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/580e17208c40aaab081180c3e855e4b6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/5a257056f94f6c10ef534476359775b2 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/5a257056f94f6c10ef534476359775b2
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/5a257056f94f6c10ef534476359775b2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/770a78edea85f8048d8b00eeca8dd4ef b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/770a78edea85f8048d8b00eeca8dd4ef
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/770a78edea85f8048d8b00eeca8dd4ef differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/7f9e33dbb727df4722fc4097e615e3bd b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/7f9e33dbb727df4722fc4097e615e3bd
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/7f9e33dbb727df4722fc4097e615e3bd differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/8a1beb6c5f5412b7d7600066e05ea24e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/8a1beb6c5f5412b7d7600066e05ea24e
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/8a1beb6c5f5412b7d7600066e05ea24e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/8b6f75830e9ef4b41e6b558b6a74e036 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/8b6f75830e9ef4b41e6b558b6a74e036
new file mode 100644
index 0000000000000000000000000000000000000000..f5a008b6111f1d86ddb8c258578f06133c85514a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/8b6f75830e9ef4b41e6b558b6a74e036 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/d9d97b16c75a6b76ccf8bfc6fb09992b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/d9d97b16c75a6b76ccf8bfc6fb09992b
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/d9d97b16c75a6b76ccf8bfc6fb09992b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/e01ebfc8c72e60bea6f99c806b3c3671 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/e01ebfc8c72e60bea6f99c806b3c3671
new file mode 100644
index 0000000000000000000000000000000000000000..8695ed0d5cf14813f5d26587f739db479dfb5356
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d/e01ebfc8c72e60bea6f99c806b3c3671
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:601e9ee2dd68fdcfd38df3bcdb1aea6d25a08e38299f2f7a36334cd9bae0e98f
+size 193841
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..fa797294057aad450706bd38e32dc34f143deb71
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..9f95c71b5f7eeaed32542b5b7ae1b5c220a0d288
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754586443060191889, "commit_timestamp_nsecs": 1754586443234916344, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/d/d2a074c79eb68238994621527c31c4dd b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/d/d2a074c79eb68238994621527c31c4dd
new file mode 100644
index 0000000000000000000000000000000000000000..77e8d157a94e48725995fcc627eff96fd74b1c8d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/d/d2a074c79eb68238994621527c31c4dd differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f5216b1f85b09bae488234b3e1c4d7bf344fb691
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/04280f44d259ab8f31a8fc108c394496 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/04280f44d259ab8f31a8fc108c394496
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/04280f44d259ab8f31a8fc108c394496 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/61f17092b1e4c408937206a4a50efe38 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/61f17092b1e4c408937206a4a50efe38
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/61f17092b1e4c408937206a4a50efe38 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/692e06b628420588956601f4448b1580 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/692e06b628420588956601f4448b1580
new file mode 100644
index 0000000000000000000000000000000000000000..58798824065f52c61af7c9d08e61d5ea70e3e152
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/692e06b628420588956601f4448b1580
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c494477acb44f63281fb5e745b650ab8eec95a1e4117a6ea1a8570c4ec2a746d
+size 363763
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/73502c93c7efe96bd2f21b0f3ddc1077 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/73502c93c7efe96bd2f21b0f3ddc1077
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/73502c93c7efe96bd2f21b0f3ddc1077 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/770f317f01e68316d84005fab636d734 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/770f317f01e68316d84005fab636d734
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/770f317f01e68316d84005fab636d734 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/8b9c2ed29b1a8a1472028632b996a2eb b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/8b9c2ed29b1a8a1472028632b996a2eb
new file mode 100644
index 0000000000000000000000000000000000000000..476dda38131ef43907099812482bb3f397651d3d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/8b9c2ed29b1a8a1472028632b996a2eb differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/c5c5c7a69ef431e4ee08ced12044a186 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/c5c5c7a69ef431e4ee08ced12044a186
new file mode 100644
index 0000000000000000000000000000000000000000..4661f020214739e25f2ecb367bda487fec2c3889
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/c5c5c7a69ef431e4ee08ced12044a186
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:937b9e4b657bbc6f18788ce1b781d14c80b0297a114459ecc9751302645a4c1c
+size 374190
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/fe2c6667a745076b29fb4050fd98f393 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/fe2c6667a745076b29fb4050fd98f393
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d/fe2c6667a745076b29fb4050fd98f393 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..53382d5f4ed161070f5922aa3dfa59a562cd3030
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..14d6f43e2b442bd43664ba00d1a1219d6dbd7957
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587733887326290, "commit_timestamp_nsecs": 1754587734061058321, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/d/b1bccd684815a005991cf201099dc359 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/d/b1bccd684815a005991cf201099dc359
new file mode 100644
index 0000000000000000000000000000000000000000..eee50c8af640f1812baa0c55a5d9cd1681ab7430
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/d/b1bccd684815a005991cf201099dc359 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9c255c974513730f7e7d664a3372c616e71db8b0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/1a148382f42bcbcfefaa3139c6346996 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/1a148382f42bcbcfefaa3139c6346996
new file mode 100644
index 0000000000000000000000000000000000000000..188b1a720052622c2b7a7d9f347ac7159ab19cd5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/1a148382f42bcbcfefaa3139c6346996
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b8470d1a47712617dbd00efba6cb9e50c5985a0ce34434c087d74427a221343a
+size 296039
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/2cea1f25bf36a33d8bc4a903680e822f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/2cea1f25bf36a33d8bc4a903680e822f
new file mode 100644
index 0000000000000000000000000000000000000000..803d0305b8b350e23ee9314daa591a365122d05f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/2cea1f25bf36a33d8bc4a903680e822f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9a7e2f7fb67effb7bd2a27433d4eeda18af77afbad1c9e4a296b75e93c400ba9
+size 405025
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/770ff540fb4071ac859ba01cfc5d81aa b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/770ff540fb4071ac859ba01cfc5d81aa
new file mode 100644
index 0000000000000000000000000000000000000000..e09299c4361c83d04c3aa07fe6cec861c701c851
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/770ff540fb4071ac859ba01cfc5d81aa differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/b39ab2f0c80ba33e2e3b640dba806f3d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/b39ab2f0c80ba33e2e3b640dba806f3d
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/b39ab2f0c80ba33e2e3b640dba806f3d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/da81c6c0e18f657806a38604f814e471 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/da81c6c0e18f657806a38604f814e471
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/da81c6c0e18f657806a38604f814e471 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/de184ad9f43185865699b924350cf3d2 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/de184ad9f43185865699b924350cf3d2
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/de184ad9f43185865699b924350cf3d2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/f0846f998b52bc9d822e3cef07c0d8e4 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/f0846f998b52bc9d822e3cef07c0d8e4
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/f0846f998b52bc9d822e3cef07c0d8e4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/fe9534620fe5c3f8f31141a6d7a582c0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/fe9534620fe5c3f8f31141a6d7a582c0
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d/fe9534620fe5c3f8f31141a6d7a582c0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e6908d326809d3cc74e7df9a0684b2eddaf93d36
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..dd401947ed629e6f885aac0290d6919e45d1a967
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587768746334171, "commit_timestamp_nsecs": 1754587768916804361, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/d/97d4ceb6eae800cff67d498ccb3ce7e9 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/d/97d4ceb6eae800cff67d498ccb3ce7e9
new file mode 100644
index 0000000000000000000000000000000000000000..a1019c2e38571b84f53b27bb9030d7b6e8833720
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/d/97d4ceb6eae800cff67d498ccb3ce7e9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..82b196520588f714248cd310c12fe0e3a20e2f38
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/0af9258091f75bfe77dc2d12d3ba1e18 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/0af9258091f75bfe77dc2d12d3ba1e18
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/0af9258091f75bfe77dc2d12d3ba1e18 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/0e4e2112574e93fad92bbe22deec3ba4 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/0e4e2112574e93fad92bbe22deec3ba4
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/0e4e2112574e93fad92bbe22deec3ba4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/6658d7dc1cd4d33f540a951a51ac5ae5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/6658d7dc1cd4d33f540a951a51ac5ae5
new file mode 100644
index 0000000000000000000000000000000000000000..66967e0bf87ffa88d85f865c85ed04d83e9409eb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/6658d7dc1cd4d33f540a951a51ac5ae5
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8de0d0f02a9b1084971a0574deae4acf149583efd892c2a90670002b9c82f080
+size 480311
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/6f7121f1f42d2d6cf5570bdbc525b9a5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/6f7121f1f42d2d6cf5570bdbc525b9a5
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/6f7121f1f42d2d6cf5570bdbc525b9a5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/6f97073cd7779392761281df759ffd1b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/6f97073cd7779392761281df759ffd1b
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/6f97073cd7779392761281df759ffd1b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/8eeb66e8da6483fad7e336fbf16908c4 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/8eeb66e8da6483fad7e336fbf16908c4
new file mode 100644
index 0000000000000000000000000000000000000000..f8a247c5fe3b95c695918b64f9b574fc2ee6b49f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/8eeb66e8da6483fad7e336fbf16908c4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/9db22c6b0638985859f444c3131a0416 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/9db22c6b0638985859f444c3131a0416
new file mode 100644
index 0000000000000000000000000000000000000000..2defe39c74f38cfbb4527b74aa2ecaf6712d1197
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/9db22c6b0638985859f444c3131a0416
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f096a221ff73a475a5dd692d505e9cce57a700a67a8902206e2d9c5f610233cf
+size 252470
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/dac1d96c63e7693de50eb2612635d63a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/dac1d96c63e7693de50eb2612635d63a
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d/dac1d96c63e7693de50eb2612635d63a differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..494d15aa0ac10d0245a60472ae3e2bb8837d1457
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..647ae2cd0cbaedd00efe8991dcc542976294c9fc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587803882424178, "commit_timestamp_nsecs": 1754587804073524272, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/d/ae2f80eedaae0b3507e5c06ac8fb878c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/d/ae2f80eedaae0b3507e5c06ac8fb878c
new file mode 100644
index 0000000000000000000000000000000000000000..fac2463b4e6e66d61095e2202a1864b810b04778
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/d/ae2f80eedaae0b3507e5c06ac8fb878c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d2a188d5a3829f3289619fcf69815a73a65fe80d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/29e039588bab1289401f3cbc9b42e080 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/29e039588bab1289401f3cbc9b42e080
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/29e039588bab1289401f3cbc9b42e080 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/8d30abbe5818ba14a14459a4b771309e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/8d30abbe5818ba14a14459a4b771309e
new file mode 100644
index 0000000000000000000000000000000000000000..026e6fa763f370355bee33dac55d5558b8e59142
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/8d30abbe5818ba14a14459a4b771309e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/ad66aaa163f7841f59c4d588810c0848 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/ad66aaa163f7841f59c4d588810c0848
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/ad66aaa163f7841f59c4d588810c0848 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/b5dfc3db28bbf49398fa27989b0bd4cd b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/b5dfc3db28bbf49398fa27989b0bd4cd
new file mode 100644
index 0000000000000000000000000000000000000000..2a31377ca030aced51ef70fb458c6b5c47404cd3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/b5dfc3db28bbf49398fa27989b0bd4cd differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/c4e5d43bb05102bf6410167206b13566 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/c4e5d43bb05102bf6410167206b13566
new file mode 100644
index 0000000000000000000000000000000000000000..e054585a36a5ecb1f01ca197692db4f40d82627a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/c4e5d43bb05102bf6410167206b13566
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:84d9e42f9b46d9ae684e755c629b4459550b0897be7b493f22bd3bcb47df7bf0
+size 227863
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/ce3c820bfec1572e628ef4d12ec1b232 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/ce3c820bfec1572e628ef4d12ec1b232
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/ce3c820bfec1572e628ef4d12ec1b232 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/dac1ff94736a5fa4e2b02eeb4da8e582 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/dac1ff94736a5fa4e2b02eeb4da8e582
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/dac1ff94736a5fa4e2b02eeb4da8e582 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/ddbc76384cef8db6598d0d7340b3bb74 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/ddbc76384cef8db6598d0d7340b3bb74
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/ddbc76384cef8db6598d0d7340b3bb74 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/e495772951c60236bebdf289f626160f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/e495772951c60236bebdf289f626160f
new file mode 100644
index 0000000000000000000000000000000000000000..0a9662ec1c9721953cea49df57ae77e6b5046b08
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d/e495772951c60236bebdf289f626160f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:80863410666041a59822f6bc89f1fd36dc22ce7428772c3641c0e9279a59c4b3
+size 438534
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d9ad1e25eebefcbbd52a72132b87b8f10b43b031
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..169ee23ea478920dc80d1828927c027ffa7f3608
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587838445308628, "commit_timestamp_nsecs": 1754587838627482027, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/d/bb05621e97500547cf2fbd448095c02b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/d/bb05621e97500547cf2fbd448095c02b
new file mode 100644
index 0000000000000000000000000000000000000000..d1103230a426cf317422f34f21e31df14aff2f70
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/d/bb05621e97500547cf2fbd448095c02b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8c2ae67654647fba44cec7876fbe5444bd1382dc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/08f0af60126aeefab06e2c3375392a0d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/08f0af60126aeefab06e2c3375392a0d
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/08f0af60126aeefab06e2c3375392a0d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/100e28bf17ff57d1a9e71da1808133fc b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/100e28bf17ff57d1a9e71da1808133fc
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/100e28bf17ff57d1a9e71da1808133fc differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/5f855381d1338f5ba900bf3218617f98 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/5f855381d1338f5ba900bf3218617f98
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/5f855381d1338f5ba900bf3218617f98 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/717b1a4776d4084522f03916f34a0f22 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/717b1a4776d4084522f03916f34a0f22
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/717b1a4776d4084522f03916f34a0f22 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/a490e32fe499096d2b14c2c7de58461e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/a490e32fe499096d2b14c2c7de58461e
new file mode 100644
index 0000000000000000000000000000000000000000..631feefaf41134083fc975205feade9745c39833
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/a490e32fe499096d2b14c2c7de58461e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/c7cf07b09d504d60b1fbf8288a97cb21 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/c7cf07b09d504d60b1fbf8288a97cb21
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/c7cf07b09d504d60b1fbf8288a97cb21 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/d186f03faa35a9d784e262c8b8eae788 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/d186f03faa35a9d784e262c8b8eae788
new file mode 100644
index 0000000000000000000000000000000000000000..aa18629919513c79d88b8ba1288400e75d3c888c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/d186f03faa35a9d784e262c8b8eae788
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3906ba311748d5adf3602f8f7eda2258d0fe95803aabbb5cc18604925570aab1
+size 364077
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/dfb0b805e2d76d2d1a0458c1e5dd5ef4 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/dfb0b805e2d76d2d1a0458c1e5dd5ef4
new file mode 100644
index 0000000000000000000000000000000000000000..e5b02d2a938b2caeb8c1f1111e8eae936863003c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/dfb0b805e2d76d2d1a0458c1e5dd5ef4
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:674a8b106b1b48c023f48b4889ef4a8b56156c9772a077be61c2db22736ab439
+size 374596
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/eb87d6b36d48e63835a2b9e78c5ac475 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/eb87d6b36d48e63835a2b9e78c5ac475
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d/eb87d6b36d48e63835a2b9e78c5ac475 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5ec51d85c7b8dc1e3401e886ba307ba6f4ac894d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..45af53d890f1d7ca172e5a980d7180f764b50d32
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587873379984590, "commit_timestamp_nsecs": 1754587873552600176, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/d/ad928f10479cb828d3df9b535da0e373 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/d/ad928f10479cb828d3df9b535da0e373
new file mode 100644
index 0000000000000000000000000000000000000000..b7e87a88593216211a9353488c2fdc90844ec0aa
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/d/ad928f10479cb828d3df9b535da0e373 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..fc4d624101605c33cd496ca0e67b62e0ce2cf02b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/1b0a2baab84efd59acf3f82f66d04671 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/1b0a2baab84efd59acf3f82f66d04671
new file mode 100644
index 0000000000000000000000000000000000000000..d57a5788bd4d60dc4663a84a1335f48d9092e45c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/1b0a2baab84efd59acf3f82f66d04671 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/2ffa9657c89cf8b4e092657ccd22d8b9 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/2ffa9657c89cf8b4e092657ccd22d8b9
new file mode 100644
index 0000000000000000000000000000000000000000..2910fbc6db2094404ed932c7d9bcdb23beb99906
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/2ffa9657c89cf8b4e092657ccd22d8b9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/4d0b84ab7d0b5ced681f73e5b7ef0dd3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/4d0b84ab7d0b5ced681f73e5b7ef0dd3
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/4d0b84ab7d0b5ced681f73e5b7ef0dd3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/62dcfb9edba6daaa3acd36fa3d169745 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/62dcfb9edba6daaa3acd36fa3d169745
new file mode 100644
index 0000000000000000000000000000000000000000..047a351feca609833a347baaaf41fc6ca962f49a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/62dcfb9edba6daaa3acd36fa3d169745
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ce45b7e36e40dd4e4ca2c67ed449d8089eb99b88c6ffcc0b03afde0371e43e5f
+size 296093
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/88296375eaffd4a08b7c953baa6d86e0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/88296375eaffd4a08b7c953baa6d86e0
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/88296375eaffd4a08b7c953baa6d86e0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/a939c1759a83b76502a0d894a16180a8 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/a939c1759a83b76502a0d894a16180a8
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/a939c1759a83b76502a0d894a16180a8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/dc3036e9077e98af8bebe856e9a9da92 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/dc3036e9077e98af8bebe856e9a9da92
new file mode 100644
index 0000000000000000000000000000000000000000..4d24fa7e9fe47ddf7dac2e40d083941f760fbb18
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/dc3036e9077e98af8bebe856e9a9da92
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:760fad71a5cce97e8d3479339fc2dcbbd64b220d45308eb52f3a30a45131d403
+size 442658
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/dfcc9868db9540604a39cc06b604c075 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/dfcc9868db9540604a39cc06b604c075
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d/dfcc9868db9540604a39cc06b604c075 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..56866b8f07b675f5d53701bd4099408936a47687
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..2b737f1f78dcee7afe27abf44f9031dd458cb7f9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587906178411269, "commit_timestamp_nsecs": 1754587906350731541, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/d/614093f1109910ae5fe4b2929a05704d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/d/614093f1109910ae5fe4b2929a05704d
new file mode 100644
index 0000000000000000000000000000000000000000..a608471270ada67f964a592bd8ce338428a77b09
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/d/614093f1109910ae5fe4b2929a05704d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..49712baaaa30507239c120d4b4b6db1c4afcee16
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/081cfaf635a8c234acc0d6707fb2053d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/081cfaf635a8c234acc0d6707fb2053d
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/081cfaf635a8c234acc0d6707fb2053d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/1dfaae3701062b95ca543b4013583b08 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/1dfaae3701062b95ca543b4013583b08
new file mode 100644
index 0000000000000000000000000000000000000000..8db244530051d8cc74f968ba3e23c453c48c1364
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/1dfaae3701062b95ca543b4013583b08 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/431a2655b38a53b864bbab8e1bb87f1c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/431a2655b38a53b864bbab8e1bb87f1c
new file mode 100644
index 0000000000000000000000000000000000000000..efe55b532d3c0be6359d8102fe7a5ddff853c75a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/431a2655b38a53b864bbab8e1bb87f1c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:56eeca1a785086cabb5eeb9e33ea7700f04777f2e324706d2e0017c72f15e59e
+size 255065
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/5f65206b43930e4f0acac362a0dc9e1c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/5f65206b43930e4f0acac362a0dc9e1c
new file mode 100644
index 0000000000000000000000000000000000000000..e8f2e2317f0fc7dc134c2c5271476d619420b846
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/5f65206b43930e4f0acac362a0dc9e1c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f009302d1749c74bb997c82c7f49d6c193b50b25c2aa164eece19bc5cf06f3c4
+size 440730
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/6d55419d7a8684491ec9b07d80906d30 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/6d55419d7a8684491ec9b07d80906d30
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/6d55419d7a8684491ec9b07d80906d30 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/c563f6ce65ace108c64ba876aaf085cb b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/c563f6ce65ace108c64ba876aaf085cb
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/c563f6ce65ace108c64ba876aaf085cb differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/d7a2cf9f60e8a76c8d46736092840c31 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/d7a2cf9f60e8a76c8d46736092840c31
new file mode 100644
index 0000000000000000000000000000000000000000..4859214e6b68fa6e697de9748417fcf43be8081d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d/d7a2cf9f60e8a76c8d46736092840c31 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9e2088d4f44c052187f69cefbdf2c2799f766b1e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..2c6af4f3fad8927b7ee2336b4920ee5ccb6bf08e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587942481286387, "commit_timestamp_nsecs": 1754587942647398731, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/d/46f8d63f2b67a19348c043b6c78357db b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/d/46f8d63f2b67a19348c043b6c78357db
new file mode 100644
index 0000000000000000000000000000000000000000..e1f6f42aad11900bc3991d1e00baa807feead25e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/d/46f8d63f2b67a19348c043b6c78357db differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c67e03feabff85f895b5f001fa910964f9ac02c3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/3b091f187b0edb9ff26d0f93a4e8f576 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/3b091f187b0edb9ff26d0f93a4e8f576
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/3b091f187b0edb9ff26d0f93a4e8f576 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/459bf5d9eb4e4c9898c743827fd3bb38 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/459bf5d9eb4e4c9898c743827fd3bb38
new file mode 100644
index 0000000000000000000000000000000000000000..73d60817bdc2f893452b08a85286b0d83d911f0b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/459bf5d9eb4e4c9898c743827fd3bb38
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6549b47567a895ae81161b5d2358a2f2e720d0d8586a41c7a9fa83d817ab1bd8
+size 403957
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/a325f36bd89ae12f05d3e696a1fbe523 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/a325f36bd89ae12f05d3e696a1fbe523
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/a325f36bd89ae12f05d3e696a1fbe523 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/a4498e71246bf3639f733c250b82ca96 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/a4498e71246bf3639f733c250b82ca96
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/a4498e71246bf3639f733c250b82ca96 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/a8659550837bb1ba3848648bd9bfe186 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/a8659550837bb1ba3848648bd9bfe186
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/a8659550837bb1ba3848648bd9bfe186 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/c145500f652d7bf084b69dc49cd71b28 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/c145500f652d7bf084b69dc49cd71b28
new file mode 100644
index 0000000000000000000000000000000000000000..3d504a762d6e499096ec5c772b6011cb6e1fc2af
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/c145500f652d7bf084b69dc49cd71b28 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/dcfd3e1d3372dd6b41f118de8d9f44ec b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/dcfd3e1d3372dd6b41f118de8d9f44ec
new file mode 100644
index 0000000000000000000000000000000000000000..65529f6532287fad4b450ca493011b55e8bf7f92
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/dcfd3e1d3372dd6b41f118de8d9f44ec differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/e6f0905e86042a644bf97008cbbefc45 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/e6f0905e86042a644bf97008cbbefc45
new file mode 100644
index 0000000000000000000000000000000000000000..4fe0c4f3b0b38a6399d45667e41896eaa6a38459
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d/e6f0905e86042a644bf97008cbbefc45
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a0b122283f80b5bc183a6ddf4919c6f3125efdf61ce0385ee599c2584b766e12
+size 295031
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d4446f51f164e4b5d19f77f3b2ef2c6127f8c38d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..507c6cff132f61a5f3f43dd5b141c6d2872a8855
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754587977239053177, "commit_timestamp_nsecs": 1754587977378553081, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/d/9b4f9474d8a812bb737d34476d6f446e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/d/9b4f9474d8a812bb737d34476d6f446e
new file mode 100644
index 0000000000000000000000000000000000000000..1d30c44141c29b90f74a45b32686a3c2cbfc7528
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/d/9b4f9474d8a812bb737d34476d6f446e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..378a55664b401b5036069965a821b900ed0a1211
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/02f46dbe2e89f83e55a8bd8bdca8f9a2 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/02f46dbe2e89f83e55a8bd8bdca8f9a2
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/02f46dbe2e89f83e55a8bd8bdca8f9a2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/0ddebbeebec7a831a47015b9078ff16c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/0ddebbeebec7a831a47015b9078ff16c
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/0ddebbeebec7a831a47015b9078ff16c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/447c3f199f5ecd5203fe20fed68b6b8a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/447c3f199f5ecd5203fe20fed68b6b8a
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/447c3f199f5ecd5203fe20fed68b6b8a differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/516d66f2ec20b5a9ecfb0d71d07c0cf6 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/516d66f2ec20b5a9ecfb0d71d07c0cf6
new file mode 100644
index 0000000000000000000000000000000000000000..acdf5c1b323f1b0f120ef873e5e6fa3c746ceaf1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/516d66f2ec20b5a9ecfb0d71d07c0cf6
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6e1fb9bf3dfeb52cdb626e9c73813bc52bff03ef4aeda7bb19c144b59f601f6d
+size 287572
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/7fd817fa16bc533e88ef49a5dfea89a1 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/7fd817fa16bc533e88ef49a5dfea89a1
new file mode 100644
index 0000000000000000000000000000000000000000..da1332e53328cb6ae1e365570af6780145901f25
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/7fd817fa16bc533e88ef49a5dfea89a1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/ab3199f4e68954f6c2f1991be40def73 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/ab3199f4e68954f6c2f1991be40def73
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/ab3199f4e68954f6c2f1991be40def73 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/d9325a35ab40ba7c2fdbd49230db005f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/d9325a35ab40ba7c2fdbd49230db005f
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/d9325a35ab40ba7c2fdbd49230db005f differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/fa25a72facaa7ee80550d244371a2a12 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/fa25a72facaa7ee80550d244371a2a12
new file mode 100644
index 0000000000000000000000000000000000000000..ec6af5b2d19ee61a72c574373c9d1f6b54b38784
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d/fa25a72facaa7ee80550d244371a2a12
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:097a8067db22742076d7bc1867bcb3e5973d1da478939c75e6e07583e4d70384
+size 451381
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..2c21413c834eff30e7cc02b2a9811625785a8d81
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..8c64c7a33a51e7ed96e8ec47ed77d15c119471d9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588030528393454, "commit_timestamp_nsecs": 1754588030675524436, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/d/aaf8c2cd34fadd3643ae44b6e5148fd9 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/d/aaf8c2cd34fadd3643ae44b6e5148fd9
new file mode 100644
index 0000000000000000000000000000000000000000..04d725027cc02cbd1c19fd7c378fd93ebea0bb4a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/d/aaf8c2cd34fadd3643ae44b6e5148fd9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..fd314b60b342aa11a6546d4087b64c766a6e7f3d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/040c78899a049a65b06772a3c0ef0678 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/040c78899a049a65b06772a3c0ef0678
new file mode 100644
index 0000000000000000000000000000000000000000..b89fe5f4c11b195af399489120d15775b28415a8
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/040c78899a049a65b06772a3c0ef0678
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:65f469b3747e0b3dc1f87d0a74dcce871fc02228f01de10500456fd730bbf3f6
+size 193907
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/482fbbaef49c2ffaf844ed2ef04bfc30 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/482fbbaef49c2ffaf844ed2ef04bfc30
new file mode 100644
index 0000000000000000000000000000000000000000..5214fb5b717cbda09351b5ae60f24be3d53e19f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/482fbbaef49c2ffaf844ed2ef04bfc30 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/4b79dd02099b17942c8ea936fad18c07 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/4b79dd02099b17942c8ea936fad18c07
new file mode 100644
index 0000000000000000000000000000000000000000..b607641737e31d08dfbdf9aab8442603a1856f4d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/4b79dd02099b17942c8ea936fad18c07
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:461422d7ba0a04cdf378f838f31490d24f1c3786b05e0620ffdb6a7e61bed404
+size 505783
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/50cf5a3bad11591e2c830c739583d851 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/50cf5a3bad11591e2c830c739583d851
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/50cf5a3bad11591e2c830c739583d851 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/7491dd45ae934719fd53acd279abc96f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/7491dd45ae934719fd53acd279abc96f
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/7491dd45ae934719fd53acd279abc96f differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/9df53bda9cf5f96d0327eb1df5981e24 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/9df53bda9cf5f96d0327eb1df5981e24
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/9df53bda9cf5f96d0327eb1df5981e24 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/a683c1d0bc8f606da07dfb08ded26319 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/a683c1d0bc8f606da07dfb08ded26319
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/a683c1d0bc8f606da07dfb08ded26319 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/df1298ed1ce5ca45abafbe620db3d787 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/df1298ed1ce5ca45abafbe620db3d787
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/df1298ed1ce5ca45abafbe620db3d787 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/f37b814b892ae1696bea6b073ba0a9f1 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/f37b814b892ae1696bea6b073ba0a9f1
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d/f37b814b892ae1696bea6b073ba0a9f1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e1ca86e993a86af3df6bf3f8d8375beb9643d8b3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..ed107336d6e23fb182c8f3fe9258c206371f2819
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588065349530158, "commit_timestamp_nsecs": 1754588065519523298, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/d/98e969db855de57979878d83ab82ac3c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/d/98e969db855de57979878d83ab82ac3c
new file mode 100644
index 0000000000000000000000000000000000000000..47ff4e7780822fc866ea3001b925a12536ed47e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/d/98e969db855de57979878d83ab82ac3c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..bdcdb7998a863489d1f5b162ffba4b5f768812d3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/26de75b13f096ba4e119e4fc68e34704 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/26de75b13f096ba4e119e4fc68e34704
new file mode 100644
index 0000000000000000000000000000000000000000..482743105e0fe58a56a57d1323500a71ca64b21f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/26de75b13f096ba4e119e4fc68e34704 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/3aead3aebee80d808815c16bd5da839f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/3aead3aebee80d808815c16bd5da839f
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/3aead3aebee80d808815c16bd5da839f differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/57d06de4dbba5c3ba8f224f380d1b321 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/57d06de4dbba5c3ba8f224f380d1b321
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/57d06de4dbba5c3ba8f224f380d1b321 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/a16a3c5dcd32e61b96495e5ea2f9a0a4 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/a16a3c5dcd32e61b96495e5ea2f9a0a4
new file mode 100644
index 0000000000000000000000000000000000000000..f2af2e2215ef7bed33899333d6919d78dd5bdade
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/a16a3c5dcd32e61b96495e5ea2f9a0a4
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:23a65c9b520a130825d9af34017f0726d85164f99ddd6160ae7d63ec35ca687b
+size 365854
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/cde5876072419773621d83340ffdae49 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/cde5876072419773621d83340ffdae49
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/cde5876072419773621d83340ffdae49 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/dc69a934a4f5788a928a344ff348f894 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/dc69a934a4f5788a928a344ff348f894
new file mode 100644
index 0000000000000000000000000000000000000000..7c21f6692df2af339fd2440051c2c19943901f2c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/dc69a934a4f5788a928a344ff348f894
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ca1c5e4e62e86dc6d8d8dc2b32bd042d73badfada597060d5c21df02f08965db
+size 372616
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/fb1317efbe5bda7328460d0b9c8065a4 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/fb1317efbe5bda7328460d0b9c8065a4
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d/fb1317efbe5bda7328460d0b9c8065a4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d07151a1ce7053b9c2488205fab58a0c1e40f779
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..055c8afefb7cc935aaef42345a32c7969c11ea27
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754586478564143727, "commit_timestamp_nsecs": 1754586478739927794, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/d/7a1c08a3f6d7ef751fac70431498f9eb b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/d/7a1c08a3f6d7ef751fac70431498f9eb
new file mode 100644
index 0000000000000000000000000000000000000000..1b686f5187944223ca6129f311385f2de5cf66c7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/d/7a1c08a3f6d7ef751fac70431498f9eb differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d1505f1800891064d6d9358584af854222c7d21b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/0ace55e63f4436f2a3b71ebfee81dc42 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/0ace55e63f4436f2a3b71ebfee81dc42
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/0ace55e63f4436f2a3b71ebfee81dc42 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/3940704a6ae6f9ec0f7910f9d9ca4ac5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/3940704a6ae6f9ec0f7910f9d9ca4ac5
new file mode 100644
index 0000000000000000000000000000000000000000..eb888f4ca14d555e19dfc284d979528897646200
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/3940704a6ae6f9ec0f7910f9d9ca4ac5
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:44c7102a092fbb156a1cc6e2a4a168e05be87313f6b3a0897edb295a4c6dbe3d
+size 430167
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/45347cafafae7a5826fb453208ecc8ec b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/45347cafafae7a5826fb453208ecc8ec
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/45347cafafae7a5826fb453208ecc8ec differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/6d3986628457dd289e68cdc12152dc7a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/6d3986628457dd289e68cdc12152dc7a
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/6d3986628457dd289e68cdc12152dc7a differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/738c35cea843469e4c123757c57981f5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/738c35cea843469e4c123757c57981f5
new file mode 100644
index 0000000000000000000000000000000000000000..e28e6d5d7e52200bc92268ccf3aa0d28e9f707fd
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/738c35cea843469e4c123757c57981f5
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6f2dac003651d7f1964c95e0d741ed91689171d31d9992bccafb09883981d139
+size 270456
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/d67ace841344ae9f05538e07bdbee065 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/d67ace841344ae9f05538e07bdbee065
new file mode 100644
index 0000000000000000000000000000000000000000..e8d58008226bb71800e3a82917023d9c383aed86
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/d67ace841344ae9f05538e07bdbee065 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/dbc90fc496d05fd41aadd8de9c687de0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/dbc90fc496d05fd41aadd8de9c687de0
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/dbc90fc496d05fd41aadd8de9c687de0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/dcf388fc0ffbdb1c0d730d2af13f31e3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/dcf388fc0ffbdb1c0d730d2af13f31e3
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d/dcf388fc0ffbdb1c0d730d2af13f31e3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..7a848c41669648f43f6f41466fcd2ffe7327a1af
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..55ced5db5a3c8007da10c85badf5710727dc3a20
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588100558527634, "commit_timestamp_nsecs": 1754588100742799108, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/d/69add4bcaa220d5727fd22ca47ac2faf b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/d/69add4bcaa220d5727fd22ca47ac2faf
new file mode 100644
index 0000000000000000000000000000000000000000..e77956e0579c940fa3154faf3dc4fb1345284a01
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/d/69add4bcaa220d5727fd22ca47ac2faf differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..98964c60f197ce71f543a48c3e4e91f0226471ad
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/070c2c9bb42f7f0b2951bbc6563be1bf b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/070c2c9bb42f7f0b2951bbc6563be1bf
new file mode 100644
index 0000000000000000000000000000000000000000..1d415b198a77f09459af631e82e5a711763cf4f8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/070c2c9bb42f7f0b2951bbc6563be1bf differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/4b93693260fee63601f4c73b77df771f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/4b93693260fee63601f4c73b77df771f
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/4b93693260fee63601f4c73b77df771f differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/5b95c56ae399c0f1257fe7586b342e12 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/5b95c56ae399c0f1257fe7586b342e12
new file mode 100644
index 0000000000000000000000000000000000000000..53f55e46b19d6950bc82b93a4ef1fd703767df40
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/5b95c56ae399c0f1257fe7586b342e12
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2dc95b7f8bc91d2db7da6723e694fce24769d7efefee90961a3646b1d378b0b8
+size 245061
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/69be044dbda8d9fd5788ac5215b8803b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/69be044dbda8d9fd5788ac5215b8803b
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/69be044dbda8d9fd5788ac5215b8803b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/842d67982b9f9df1d8694377cce52ce1 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/842d67982b9f9df1d8694377cce52ce1
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/842d67982b9f9df1d8694377cce52ce1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/b48b36545f3c09c9698f1e96579ee00a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/b48b36545f3c09c9698f1e96579ee00a
new file mode 100644
index 0000000000000000000000000000000000000000..f28d59e1f9187cfd65b688b4bceb57a81f0acdcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/b48b36545f3c09c9698f1e96579ee00a differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/cb0757823b78d3bc62fb5404f6183d17 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/cb0757823b78d3bc62fb5404f6183d17
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/cb0757823b78d3bc62fb5404f6183d17 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/dbbd2fc7d31537fb0ad780129a5e6fc6 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/dbbd2fc7d31537fb0ad780129a5e6fc6
new file mode 100644
index 0000000000000000000000000000000000000000..816d5e4bbf544cb8a5ac1ebab9ef3070702cc8c0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d/dbbd2fc7d31537fb0ad780129a5e6fc6
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e539b9ea3f49922d161290d685a5140c2d170d9478de2f91db13b8c83410ba00
+size 417199
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8778d754bc33d6db0215075901bfdcb1635c8670
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..62bcd3dc4fb9e92985a254377b67c165e313abeb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588135039107741, "commit_timestamp_nsecs": 1754588135167445199, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/d/ebdddcb54aa34298b3b99f4728cdd66b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/d/ebdddcb54aa34298b3b99f4728cdd66b
new file mode 100644
index 0000000000000000000000000000000000000000..b50213596096c8b3dafb8257071af36119b9e5c1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/d/ebdddcb54aa34298b3b99f4728cdd66b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..2e8b3f52b5d1d52ffdc95b1d4b182a3dd76771f3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/15fcc8f5b5da91c4a04b98366e23873a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/15fcc8f5b5da91c4a04b98366e23873a
new file mode 100644
index 0000000000000000000000000000000000000000..85f35e7991cac7b6fc244bb2f277438119f4e90b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/15fcc8f5b5da91c4a04b98366e23873a differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/4d34ddbaa2dd92756599b928e8d5add9 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/4d34ddbaa2dd92756599b928e8d5add9
new file mode 100644
index 0000000000000000000000000000000000000000..6c0a3d27be196c132102286cf9096ade9bf4880c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/4d34ddbaa2dd92756599b928e8d5add9
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9fe600ed5792b2a9cf0a8f59f634ad68246a4a029ce01041a7ba35224c488be0
+size 544904
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/6884d6bf55801e033ef60c9efe7da122 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/6884d6bf55801e033ef60c9efe7da122
new file mode 100644
index 0000000000000000000000000000000000000000..16dfebd33935104fd017e41c6d038b8fdc8d61ba
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/6884d6bf55801e033ef60c9efe7da122
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f31347cc21f91046b2bddf3c7733478ba924466eae16073c087be6264163fcfd
+size 193849
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/799e802b03bb9ca8f8e7c51ad8f76b90 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/799e802b03bb9ca8f8e7c51ad8f76b90
new file mode 100644
index 0000000000000000000000000000000000000000..9745ee5b0568de89c03d3170a4ba2cf5e12d5cce
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/799e802b03bb9ca8f8e7c51ad8f76b90 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/9820e1f435f87063eb0fbb3daf271b7c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/9820e1f435f87063eb0fbb3daf271b7c
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/9820e1f435f87063eb0fbb3daf271b7c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/be6cbf443b9f37158c1856d4375e739e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/be6cbf443b9f37158c1856d4375e739e
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/be6cbf443b9f37158c1856d4375e739e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/d8a4a1121dccf25b3487e949c6ae63b6 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/d8a4a1121dccf25b3487e949c6ae63b6
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d/d8a4a1121dccf25b3487e949c6ae63b6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a815732b34e9766be3a1e9ca822c301f85101eed
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..5ea9aa7aa994672bac87aaa5ef7a36c19f839394
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588169722143026, "commit_timestamp_nsecs": 1754588169864620052, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/d/fc0850dd6c08b8861648603ca1d91e78 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/d/fc0850dd6c08b8861648603ca1d91e78
new file mode 100644
index 0000000000000000000000000000000000000000..63ed563ce57a65e8f02605fa9dba09d745b789a6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/d/fc0850dd6c08b8861648603ca1d91e78 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..53321d63f6ae1b87e549b3634795e0aaa661bd85
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/2e093e4e334437f8018f5cf8c8dffa00 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/2e093e4e334437f8018f5cf8c8dffa00
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/2e093e4e334437f8018f5cf8c8dffa00 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/31109f1f189e87c9dee2d655ec3fc0d5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/31109f1f189e87c9dee2d655ec3fc0d5
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/31109f1f189e87c9dee2d655ec3fc0d5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/388bb603d697664e7dc9d53104f81d24 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/388bb603d697664e7dc9d53104f81d24
new file mode 100644
index 0000000000000000000000000000000000000000..ecf9d46b7b47d26e87474c53313ae7ce278ed6ec
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/388bb603d697664e7dc9d53104f81d24 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/53999eec8a0516ad3e37fc42cafb797b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/53999eec8a0516ad3e37fc42cafb797b
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/53999eec8a0516ad3e37fc42cafb797b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/6556f1a65c0e94a7aff2d8e523a64770 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/6556f1a65c0e94a7aff2d8e523a64770
new file mode 100644
index 0000000000000000000000000000000000000000..f1115b2af9bb2e164719e4a04e5fbe6d01abddeb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/6556f1a65c0e94a7aff2d8e523a64770
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e80d81d7d8eb79aea27c7c32b67777843275adceb850f3e0d7de5dc33984cbae
+size 553806
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/77e68f1eea49a70c127ff2fe4c5a476c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/77e68f1eea49a70c127ff2fe4c5a476c
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/77e68f1eea49a70c127ff2fe4c5a476c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/780daae3a1dd351560711de467e15842 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/780daae3a1dd351560711de467e15842
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/780daae3a1dd351560711de467e15842 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/e852a29e92f0a60b3e5eb6015ba4e898 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/e852a29e92f0a60b3e5eb6015ba4e898
new file mode 100644
index 0000000000000000000000000000000000000000..a2fc5d4456be46cdba38524bd8099a9e800a8e62
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d/e852a29e92f0a60b3e5eb6015ba4e898
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ecd36cd099a10ec18de69ce91e683fbf280810144fc87978f811e61ca4947585
+size 185529
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0f4a50f00d2314afd57402b39e4b392e18dd6882
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..01b67848046b76cdc3db1ccaef4595f3fe10b25d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588204943014959, "commit_timestamp_nsecs": 1754588205117781134, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/d/b104456e75ef93e6388fc67d06e0fe9e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/d/b104456e75ef93e6388fc67d06e0fe9e
new file mode 100644
index 0000000000000000000000000000000000000000..c4ac46278398e11c631c182fe56f9e78b3d11090
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/d/b104456e75ef93e6388fc67d06e0fe9e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..50b033a3d013c371ef707cbdca921f967788c8af
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/3233b165c14836cfaa6ee3ac95d1636e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/3233b165c14836cfaa6ee3ac95d1636e
new file mode 100644
index 0000000000000000000000000000000000000000..14351914b6a5fc7b5000e19374a84733b758bb8f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/3233b165c14836cfaa6ee3ac95d1636e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/590e4a3354923c4f7f333c9474395cc3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/590e4a3354923c4f7f333c9474395cc3
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/590e4a3354923c4f7f333c9474395cc3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/64ad4999040c3b26269a81211c1abd1c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/64ad4999040c3b26269a81211c1abd1c
new file mode 100644
index 0000000000000000000000000000000000000000..cbb8648211915287bed44fd797826c6aa531a7a3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/64ad4999040c3b26269a81211c1abd1c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4dc6d3db8f85ffab0ed5c0e12a08f1e78905683631caa141194290c301173d15
+size 296044
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/a84b1a74483b261d32e663dd053d417d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/a84b1a74483b261d32e663dd053d417d
new file mode 100644
index 0000000000000000000000000000000000000000..11a22db2a0eb69ae3d486fda83fa2b3f09e588a6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/a84b1a74483b261d32e663dd053d417d
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e34e6912ee8c601dbf73f4de75414f647099de4f1f7a4c9ef0e12d4ad40389c0
+size 442736
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/c5b9975c2c9015b3474f41c107c3032b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/c5b9975c2c9015b3474f41c107c3032b
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/c5b9975c2c9015b3474f41c107c3032b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/cca3466018c5ded783bfced772086b3f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/cca3466018c5ded783bfced772086b3f
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/cca3466018c5ded783bfced772086b3f differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/f09dd746a3047e363377366d93292285 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/f09dd746a3047e363377366d93292285
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/f09dd746a3047e363377366d93292285 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/fc0db1b9a46eed0e11d15af248012571 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/fc0db1b9a46eed0e11d15af248012571
new file mode 100644
index 0000000000000000000000000000000000000000..2910fbc6db2094404ed932c7d9bcdb23beb99906
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d/fc0db1b9a46eed0e11d15af248012571 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..ae9069a66a1534fba13c018802c26b67a68d6a05
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..99ef07a3472fb5f950d72a03442cf1b3de737aaa
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588239922098226, "commit_timestamp_nsecs": 1754588240066713070, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/d/40fb6924aa21ae57cf4aa8a4af446b4c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/d/40fb6924aa21ae57cf4aa8a4af446b4c
new file mode 100644
index 0000000000000000000000000000000000000000..65c5845b17b8a5a101220f840b1a4ec8ca86df1a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/d/40fb6924aa21ae57cf4aa8a4af446b4c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9334601da7bd0d4126d07642981e39844cdd7ebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/0ad6bf22fa1613f2c4fa78e51adbc437 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/0ad6bf22fa1613f2c4fa78e51adbc437
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/0ad6bf22fa1613f2c4fa78e51adbc437 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/548e161b8eb3b0a6c98567297ed5af2b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/548e161b8eb3b0a6c98567297ed5af2b
new file mode 100644
index 0000000000000000000000000000000000000000..5a7441789a9a29007bee60facc4146058202730a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/548e161b8eb3b0a6c98567297ed5af2b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/5c036bf433f00232b6e7384c62336cf7 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/5c036bf433f00232b6e7384c62336cf7
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/5c036bf433f00232b6e7384c62336cf7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/947a2510aed332a29714a69c6e49515f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/947a2510aed332a29714a69c6e49515f
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/947a2510aed332a29714a69c6e49515f differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/b956fcf0ebadf95f8424a8c683967cc2 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/b956fcf0ebadf95f8424a8c683967cc2
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/b956fcf0ebadf95f8424a8c683967cc2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/c15a3667b2f8b5add6f6f07043e5131f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/c15a3667b2f8b5add6f6f07043e5131f
new file mode 100644
index 0000000000000000000000000000000000000000..25a128d59313a41d7c0573ce766c7349934dd6c6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/c15a3667b2f8b5add6f6f07043e5131f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:89c0636449660af2624b3dd2fab75191b41891c6f83cb6416068c543d6ceb488
+size 227817
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/e6718bd9c5ca1c404f2c2afb3cfc162a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/e6718bd9c5ca1c404f2c2afb3cfc162a
new file mode 100644
index 0000000000000000000000000000000000000000..7258599e06f1f72d709ac336ec093c4825abbe54
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/e6718bd9c5ca1c404f2c2afb3cfc162a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:79c2cc6f257e07b4be73df6ab4655b561503bdce26ef692087d3e40aeef974a9
+size 427103
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/f8eb484adeac634e8350a477c357b6c5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/f8eb484adeac634e8350a477c357b6c5
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d/f8eb484adeac634e8350a477c357b6c5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..da27eb7b7ee7dea5b3e0a9c59312d1d56700f902
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..ab4fd370d071acb3453f2b7f07f529cf38c83c65
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588274866801748, "commit_timestamp_nsecs": 1754588275032292559, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/d/e8d228d8704e53ed5e6bca5ce118b613 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/d/e8d228d8704e53ed5e6bca5ce118b613
new file mode 100644
index 0000000000000000000000000000000000000000..26c99b6552e3b869053cb1ffff1bf934e54a4710
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/d/e8d228d8704e53ed5e6bca5ce118b613 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f52e90460afa4767f8a3d104a49a33a1a53e730a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/5ceeba811ebbf1df4745c297b7b9ddb3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/5ceeba811ebbf1df4745c297b7b9ddb3
new file mode 100644
index 0000000000000000000000000000000000000000..ada311a36ebaa9bd94fca4d32e01977a60265ecf
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/5ceeba811ebbf1df4745c297b7b9ddb3
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:525437c6175301fa169c541c9e440836d2997aeaecdc859685a55b0d5504c97b
+size 340576
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/7c4ff4875d5893e6a7ebf613f4a43af7 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/7c4ff4875d5893e6a7ebf613f4a43af7
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/7c4ff4875d5893e6a7ebf613f4a43af7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/851941097b21fa2d1a1cd52578b76397 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/851941097b21fa2d1a1cd52578b76397
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/851941097b21fa2d1a1cd52578b76397 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/a7c3c160605b604b2d6e742ff5083617 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/a7c3c160605b604b2d6e742ff5083617
new file mode 100644
index 0000000000000000000000000000000000000000..56bbfc0a4689d9932cee2b32bf6290fe6bca467f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/a7c3c160605b604b2d6e742ff5083617
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fda26ff6c6c80986b42460f5064d0f2e840e09876858fb0228e6030b180c3869
+size 389578
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/d76a6d6d53d244557ac5685404933a72 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/d76a6d6d53d244557ac5685404933a72
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/d76a6d6d53d244557ac5685404933a72 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/e5d49730eb447f14f3a793b4ffe6f1a3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/e5d49730eb447f14f3a793b4ffe6f1a3
new file mode 100644
index 0000000000000000000000000000000000000000..e5d35847614258bb6622497ff1ef2b66450f4308
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/e5d49730eb447f14f3a793b4ffe6f1a3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/e5fc2b9991732d259828865c89d116af b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/e5fc2b9991732d259828865c89d116af
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d/e5fc2b9991732d259828865c89d116af differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5ac8d7eeb77df2a53f61d2fefa3e23b65cd5114f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..4c80001bebc57efc2f0f01dfa874a7d2b7511105
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588312323484099, "commit_timestamp_nsecs": 1754588312468598581, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/d/d26dea0e7a7907e18fa101f00d8712fe b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/d/d26dea0e7a7907e18fa101f00d8712fe
new file mode 100644
index 0000000000000000000000000000000000000000..62390b80acdb4ab9af7fca786172b328b12ecd8b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/d/d26dea0e7a7907e18fa101f00d8712fe differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..6ce44b212d7f15aba6ee1443c6690ba575c3c057
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/13ca96f6719d56e0548eca07f0b3bfaf b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/13ca96f6719d56e0548eca07f0b3bfaf
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/13ca96f6719d56e0548eca07f0b3bfaf differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/3041569b0cbfd8c3682c2b660f774334 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/3041569b0cbfd8c3682c2b660f774334
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/3041569b0cbfd8c3682c2b660f774334 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/75504efd3faead562a9d906408ca4c73 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/75504efd3faead562a9d906408ca4c73
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/75504efd3faead562a9d906408ca4c73 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/983c38423d217753556b73804b8df9f9 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/983c38423d217753556b73804b8df9f9
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/983c38423d217753556b73804b8df9f9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/aa140fc2b67aa0f01126cc60ce95c115 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/aa140fc2b67aa0f01126cc60ce95c115
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/aa140fc2b67aa0f01126cc60ce95c115 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/bc5411967fa7e052f172faaafc3b5ae9 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/bc5411967fa7e052f172faaafc3b5ae9
new file mode 100644
index 0000000000000000000000000000000000000000..c10b7571f54b1e177f8c3b9543cfc368c55ce11e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/bc5411967fa7e052f172faaafc3b5ae9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/c9ab85ade2ff8160d0d02e5106210987 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/c9ab85ade2ff8160d0d02e5106210987
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/c9ab85ade2ff8160d0d02e5106210987 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/cb45c8bb9ce20ebe548105c7f05359f4 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/cb45c8bb9ce20ebe548105c7f05359f4
new file mode 100644
index 0000000000000000000000000000000000000000..a7a8644b373910a32bf5e2487abee0d3a6654441
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/cb45c8bb9ce20ebe548105c7f05359f4
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5f86e8498992d2a02451312007aed82926ad2e36dbbe9d9532f61251acd9b869
+size 296088
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/db85454e735f2f28ccdd655fd369dc0d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/db85454e735f2f28ccdd655fd369dc0d
new file mode 100644
index 0000000000000000000000000000000000000000..9c46303c964486a2f6344c10aa45c8c4f149019c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d/db85454e735f2f28ccdd655fd369dc0d
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c5aa389a6ded7f8b8a4e14b913e1c25a6dd7076ebdd5b357dca08f27de8fbdc4
+size 442735
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b607590bc6ac9257533565448986f0176bb4eeb0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..a21bc1bcfed9a790adb3750ce9ec0c3476613def
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588347318668338, "commit_timestamp_nsecs": 1754588347470729860, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/d/292f47046c98287c99a1ccb2c5b05af7 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/d/292f47046c98287c99a1ccb2c5b05af7
new file mode 100644
index 0000000000000000000000000000000000000000..820d3f3640136594756e93ac386e26e2be404dd7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/d/292f47046c98287c99a1ccb2c5b05af7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1a83d7a0031c7150341a6412e5321ff120e848f4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/3f28f47f17f1dc3c1ee748b367f26a74 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/3f28f47f17f1dc3c1ee748b367f26a74
new file mode 100644
index 0000000000000000000000000000000000000000..9ddc03961f1d342650794ebfa1c689eba93d6eea
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/3f28f47f17f1dc3c1ee748b367f26a74
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:71f81b2fe46f0ab0310e8c22ab5b782064c61e0b304bf0b8b0aee97e78de2db4
+size 320990
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/3f37ddb1969fb5c5c7b61857eab9054f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/3f37ddb1969fb5c5c7b61857eab9054f
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/3f37ddb1969fb5c5c7b61857eab9054f differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/5403887b43436750a6547feeca305ca9 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/5403887b43436750a6547feeca305ca9
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/5403887b43436750a6547feeca305ca9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/57689831c5aa22ae9f52777391f87332 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/57689831c5aa22ae9f52777391f87332
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/57689831c5aa22ae9f52777391f87332 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/58c1f1c8e0e982a09911188cb01d36cd b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/58c1f1c8e0e982a09911188cb01d36cd
new file mode 100644
index 0000000000000000000000000000000000000000..c54d39b71eff2076ac332b089285990e41a690c4
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/58c1f1c8e0e982a09911188cb01d36cd
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:046b7ecd8477c061f30ac976e0978e1a2582080b13e21f8baa33b11bfffe3d57
+size 225730
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/5d4a322d3d50aa4bb46f36a2ab41bd88 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/5d4a322d3d50aa4bb46f36a2ab41bd88
new file mode 100644
index 0000000000000000000000000000000000000000..35259b96b0baf4ffeba2e014e0bc3f97feb8ba14
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/5d4a322d3d50aa4bb46f36a2ab41bd88
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4b4c1152e786f35a16dda3272f593ca57f5530db0832da4b03bcd2d03125b14a
+size 193926
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/d2abcc2993a444c8346462df4497cf74 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/d2abcc2993a444c8346462df4497cf74
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/d2abcc2993a444c8346462df4497cf74 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/f7880a597d816bd9dba926bb5cbfa00b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/f7880a597d816bd9dba926bb5cbfa00b
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d/f7880a597d816bd9dba926bb5cbfa00b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..6f70f7342ac858c9a040d954ac3256063fb7064a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..27d170d7f513074714eda9f0388ff42d6663c476
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588382253075551, "commit_timestamp_nsecs": 1754588382432413482, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/d/ad96da69258f997e3cf4cbaee6643ef4 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/d/ad96da69258f997e3cf4cbaee6643ef4
new file mode 100644
index 0000000000000000000000000000000000000000..fc68acd3ff5baef3bcf4ef5f492fa98d5edb4922
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/d/ad96da69258f997e3cf4cbaee6643ef4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..02bb7472579f7aab9bb5be8ad44f0789c5fb3b8f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/030d57ecf8a4da650a6ca20493b5de7b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/030d57ecf8a4da650a6ca20493b5de7b
new file mode 100644
index 0000000000000000000000000000000000000000..3bfeace45169d4bc55d9ca51b7c68780c2ec4bea
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/030d57ecf8a4da650a6ca20493b5de7b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:080132437b3e8bf54bbc006900906a57079038f394acc1855fd01ec047b5dee8
+size 168676
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/1055246d70093d11a49d07d4964b88aa b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/1055246d70093d11a49d07d4964b88aa
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/1055246d70093d11a49d07d4964b88aa differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/1f9b39ab8f5c370b0f9bfb5f45a63b60 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/1f9b39ab8f5c370b0f9bfb5f45a63b60
new file mode 100644
index 0000000000000000000000000000000000000000..639e1f975ce1798b7db00415487774e03fa2cb91
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/1f9b39ab8f5c370b0f9bfb5f45a63b60 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/7a65a56f8e35e5ff1ae8f88cfcf251c3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/7a65a56f8e35e5ff1ae8f88cfcf251c3
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/7a65a56f8e35e5ff1ae8f88cfcf251c3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/7b5e3b5b3405653d8d372c5118df26fe b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/7b5e3b5b3405653d8d372c5118df26fe
new file mode 100644
index 0000000000000000000000000000000000000000..420a73ce08452a7bb3e4e4c1be653f8a70a903ba
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/7b5e3b5b3405653d8d372c5118df26fe differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/945429539f4fe058351172dad10cfbb2 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/945429539f4fe058351172dad10cfbb2
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/945429539f4fe058351172dad10cfbb2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/aa76b7fd856f6f243f9c342bb3c9fcf4 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/aa76b7fd856f6f243f9c342bb3c9fcf4
new file mode 100644
index 0000000000000000000000000000000000000000..21a3c0c3696a50ff61dd8d99ab1f88ed13a863ec
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/aa76b7fd856f6f243f9c342bb3c9fcf4
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ee3e425914edd4f3a0bbe7342baa4c43e786dc4ea4065179517f0ba21ccf0070
+size 243777
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/c2ec458cb71485ffa7091f9c7b93557e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/c2ec458cb71485ffa7091f9c7b93557e
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/c2ec458cb71485ffa7091f9c7b93557e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/c3b67c68d9e3dbd4751faa7918284982 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/c3b67c68d9e3dbd4751faa7918284982
new file mode 100644
index 0000000000000000000000000000000000000000..b4c5558e51393ff51e12e16d113cbd9d2c8cb3d5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d/c3b67c68d9e3dbd4751faa7918284982
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:904bfb570ba2b77fca9d212510353e440af6236065e1af9c44cdc20a53ce86af
+size 281517
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4f4610430d294a3c49eca4e2a777f2886360407d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..07a2f15186acf8a745ca659641534fb8755e65ef
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588417359442530, "commit_timestamp_nsecs": 1754588417529049402, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/d/e69dd53f3dcd05d9cd8ddc24587124b1 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/d/e69dd53f3dcd05d9cd8ddc24587124b1
new file mode 100644
index 0000000000000000000000000000000000000000..b1da65e0877e1dd632cb3d907249fa04f46fc30e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/d/e69dd53f3dcd05d9cd8ddc24587124b1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9dec15ef940ea275c90337b66ec04525cbfa4a73
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/0e425095135880a2b25ab3baa825d7a1 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/0e425095135880a2b25ab3baa825d7a1
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/0e425095135880a2b25ab3baa825d7a1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/2224b584a229b190374f985037cdd10b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/2224b584a229b190374f985037cdd10b
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/2224b584a229b190374f985037cdd10b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/27d6be176dc519c7c7acd6395400193b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/27d6be176dc519c7c7acd6395400193b
new file mode 100644
index 0000000000000000000000000000000000000000..39f8c6dc0b69e0c55388e69e245a33e33b886f5e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/27d6be176dc519c7c7acd6395400193b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0b22b9f4494625cf944afb59faedc32cdcd2ad992223af7b5159829d47f6c6ef
+size 287558
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/71279fc84818f1f3f1b475974caa66b8 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/71279fc84818f1f3f1b475974caa66b8
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/71279fc84818f1f3f1b475974caa66b8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/7241d1c0f7348db334d3f31a11cdae7d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/7241d1c0f7348db334d3f31a11cdae7d
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/7241d1c0f7348db334d3f31a11cdae7d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/7f80a83d19a74041c4dde99b46224a85 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/7f80a83d19a74041c4dde99b46224a85
new file mode 100644
index 0000000000000000000000000000000000000000..bf13330bb6ceab5a118ec2b5beafc8996e9e09e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/7f80a83d19a74041c4dde99b46224a85
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2cf3c4991eff545793a1656306551ea02de6929c9a42dccd89967ed4f0a2c64a
+size 413409
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/80322b14e90de486df1ad472bb3b679e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/80322b14e90de486df1ad472bb3b679e
new file mode 100644
index 0000000000000000000000000000000000000000..484040bf903c4e5ad1d9d88380e392aaf90f060e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d/80322b14e90de486df1ad472bb3b679e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4baae2d81d9487f33295f3adb8a5ec58bdf64e50
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..745ee49c2c6c212788191b8f272d5cad50d99e76
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754586513272596087, "commit_timestamp_nsecs": 1754586513445210015, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/d/7c0408280bbf0daa667d35ec5b94f704 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/d/7c0408280bbf0daa667d35ec5b94f704
new file mode 100644
index 0000000000000000000000000000000000000000..c9b29c583ad8cca054364dccdf87780873455938
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/d/7c0408280bbf0daa667d35ec5b94f704 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..93ee5aba723fd0c4198047725993816cf082368a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/125e2361ace8c1cefc8431aa0cba59ee b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/125e2361ace8c1cefc8431aa0cba59ee
new file mode 100644
index 0000000000000000000000000000000000000000..538d745b20c759ce4d616b514e2d8cc60555daa0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/125e2361ace8c1cefc8431aa0cba59ee
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:14e95366ae35bd33d3d9a316a3e3044b963eb92029e04fe86fe8f0c1591dace2
+size 261738
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/1274ea172e1d87f1870364cc955c171d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/1274ea172e1d87f1870364cc955c171d
new file mode 100644
index 0000000000000000000000000000000000000000..9a722ec67e37e0a2bc21f9b09198cf350e593b28
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/1274ea172e1d87f1870364cc955c171d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/1755351411c4e8dd461e4dbabf8ca744 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/1755351411c4e8dd461e4dbabf8ca744
new file mode 100644
index 0000000000000000000000000000000000000000..f28d59e1f9187cfd65b688b4bceb57a81f0acdcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/1755351411c4e8dd461e4dbabf8ca744 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/3a40fc3255b71327ccf906f1e744647b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/3a40fc3255b71327ccf906f1e744647b
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/3a40fc3255b71327ccf906f1e744647b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/6b56343a0c5b2d327ffd415fdc2c749f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/6b56343a0c5b2d327ffd415fdc2c749f
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/6b56343a0c5b2d327ffd415fdc2c749f differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/88900d8673a5e97ed96139e58cf0a864 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/88900d8673a5e97ed96139e58cf0a864
new file mode 100644
index 0000000000000000000000000000000000000000..bb3a9da2e27e962c94e1e83a80a8ddcfa4521827
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/88900d8673a5e97ed96139e58cf0a864
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8b766a9e470958c62bf452cce7eb4aeb3952031a0a182eb83f0621ba33e0c827
+size 430713
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/ad069706756c6de87dafd9045119a112 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/ad069706756c6de87dafd9045119a112
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/ad069706756c6de87dafd9045119a112 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/b07a16c40a47a14928ee001e9079f062 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/b07a16c40a47a14928ee001e9079f062
new file mode 100644
index 0000000000000000000000000000000000000000..ec745d97353f97209ef50f0eb738585830219f53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/b07a16c40a47a14928ee001e9079f062 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/fd4f47e5db0b2fedea281b979a0795b4 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/fd4f47e5db0b2fedea281b979a0795b4
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d/fd4f47e5db0b2fedea281b979a0795b4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..7b7a2fbfbf6bd1b57e0bbb74ef767a43009b3bd9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..51b51379e7a05a4984482273ba371b9837fb329a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588453632749389, "commit_timestamp_nsecs": 1754588453820841652, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/d/ac55f2db865f7038d776dbfdd391f12e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/d/ac55f2db865f7038d776dbfdd391f12e
new file mode 100644
index 0000000000000000000000000000000000000000..ea01e09a0181b1756bd69a977869efebf9d9d311
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/d/ac55f2db865f7038d776dbfdd391f12e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..804a8d7e3051ec7370e0aee65f243a502aad36e9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/0c0cbb2a561255db54845314176fe8ac b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/0c0cbb2a561255db54845314176fe8ac
new file mode 100644
index 0000000000000000000000000000000000000000..bb80b9a8b8b8d5ffb33170d7f906cbfbf907b650
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/0c0cbb2a561255db54845314176fe8ac
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0aa9599bb760d6b20f0afcab2565723f8665d321239a99aaacf946ad4664b5c5
+size 296044
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/28c1b918df16b34e93bcdf8f99721cd7 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/28c1b918df16b34e93bcdf8f99721cd7
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/28c1b918df16b34e93bcdf8f99721cd7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/3324fd983b48419a767ef6b8197ed361 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/3324fd983b48419a767ef6b8197ed361
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/3324fd983b48419a767ef6b8197ed361 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/4ca2f73299ddabe54091b85f0c845e70 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/4ca2f73299ddabe54091b85f0c845e70
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/4ca2f73299ddabe54091b85f0c845e70 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/4fdd4b95bf24b0cf8ba5405206b75019 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/4fdd4b95bf24b0cf8ba5405206b75019
new file mode 100644
index 0000000000000000000000000000000000000000..6aa4cdc6e374450742d324f68cedac26d07e3198
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/4fdd4b95bf24b0cf8ba5405206b75019 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/5eefd6ee8d09fcbcda9896b099de3e0b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/5eefd6ee8d09fcbcda9896b099de3e0b
new file mode 100644
index 0000000000000000000000000000000000000000..e60bb7c824b65ecd147d8c175598a56fa3d6a8c3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/5eefd6ee8d09fcbcda9896b099de3e0b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:676220d75d299d526cf944ccb2f7d48188b11aaecabdc2f85bc20df9384b298f
+size 442855
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/f174e1bdfe8703436a6a68274a5f56ce b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/f174e1bdfe8703436a6a68274a5f56ce
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/f174e1bdfe8703436a6a68274a5f56ce differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/f617d193d79e200e8a39ee1ea7613d90 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/f617d193d79e200e8a39ee1ea7613d90
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d/f617d193d79e200e8a39ee1ea7613d90 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..56c6e9d4474a5c151b0730641c3d499624a7c4d1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..c2a2e04f25838cb97f670e655bd71fd849657bd2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588487775160463, "commit_timestamp_nsecs": 1754588487941065335, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/d/14d8ec4480c6660ac5d53308f4f78e0a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/d/14d8ec4480c6660ac5d53308f4f78e0a
new file mode 100644
index 0000000000000000000000000000000000000000..863f58c033de1effa1e36bad87195ed310b515d6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/d/14d8ec4480c6660ac5d53308f4f78e0a differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..aeb8cf3a40ddd0391c15d6b592c7c0030bc25e8a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/125329bc054444ffa7caac0d5731b0b6 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/125329bc054444ffa7caac0d5731b0b6
new file mode 100644
index 0000000000000000000000000000000000000000..93090ea55c5446bcd09a5cf56a34a9cf22dc5c3e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/125329bc054444ffa7caac0d5731b0b6
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:eeb391e5bb4d6bacdbf1b59db487dbb1ef7e004471030a44b8baf371a06bcf25
+size 473353
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/250b00ad85877997937621560c2dc90a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/250b00ad85877997937621560c2dc90a
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/250b00ad85877997937621560c2dc90a differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/437b75e1322264f5633d5dd2463c7493 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/437b75e1322264f5633d5dd2463c7493
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/437b75e1322264f5633d5dd2463c7493 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/5b53ff438a977b7ad94d488142663bbd b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/5b53ff438a977b7ad94d488142663bbd
new file mode 100644
index 0000000000000000000000000000000000000000..dc26510f174bbd16ea479eb01a89abbf51a8aa05
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/5b53ff438a977b7ad94d488142663bbd
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9f9be309910e46948da3c0c968fb29acf28316f1e7f12dd66097d17b8f4e24dc
+size 227871
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/b914edeade1fa75ee4b9d22c37cf40e9 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/b914edeade1fa75ee4b9d22c37cf40e9
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/b914edeade1fa75ee4b9d22c37cf40e9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/c89d34084247ba3bb630e9f4edfa2bfa b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/c89d34084247ba3bb630e9f4edfa2bfa
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/c89d34084247ba3bb630e9f4edfa2bfa differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/edbad52843b0017feabeda19872cc328 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/edbad52843b0017feabeda19872cc328
new file mode 100644
index 0000000000000000000000000000000000000000..b1ccb69189d518664d2ac07042ada67b692fb8c8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/edbad52843b0017feabeda19872cc328 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/fcc1c5ca975f454c36b85a913c62dea3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/fcc1c5ca975f454c36b85a913c62dea3
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d/fcc1c5ca975f454c36b85a913c62dea3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8dda3722d8c088c3bcfdf3adbbde72933640dc81
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..88a3284b782413f6adcdfeb5fe06f7b530346349
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588522244264907, "commit_timestamp_nsecs": 1754588522416973482, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/d/118b3082817e6aa230ce84a6d3842190 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/d/118b3082817e6aa230ce84a6d3842190
new file mode 100644
index 0000000000000000000000000000000000000000..b746c8e92b861d695f36816bf1c515da89a3e171
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/d/118b3082817e6aa230ce84a6d3842190 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8ea49adfa12068026ad79d6a09d8c108d81eab96
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/2088bef0748962e5ce3c38a77fb86a7b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/2088bef0748962e5ce3c38a77fb86a7b
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/2088bef0748962e5ce3c38a77fb86a7b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/77cd6df2513836ebf054a845a4aef8c5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/77cd6df2513836ebf054a845a4aef8c5
new file mode 100644
index 0000000000000000000000000000000000000000..3d9e8818b11b406f98cd5fdb0487f51160e1982e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/77cd6df2513836ebf054a845a4aef8c5
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:024b08cb863e3b714bf61c76029d712523e65b09999cd816938456b48ab13e90
+size 287508
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/8145e63de2c5838fafad9fc1cd7e8126 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/8145e63de2c5838fafad9fc1cd7e8126
new file mode 100644
index 0000000000000000000000000000000000000000..2910fbc6db2094404ed932c7d9bcdb23beb99906
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/8145e63de2c5838fafad9fc1cd7e8126 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/8ba35e84d5a6397779185844e9131ed1 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/8ba35e84d5a6397779185844e9131ed1
new file mode 100644
index 0000000000000000000000000000000000000000..5ad91e07cc8ba9682eade7a7e940c90be0e8e221
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/8ba35e84d5a6397779185844e9131ed1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/8ed077ef12c3b68bc62cea6dd0521d0c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/8ed077ef12c3b68bc62cea6dd0521d0c
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/8ed077ef12c3b68bc62cea6dd0521d0c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/add364902a203404f12f26c621104c65 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/add364902a203404f12f26c621104c65
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/add364902a203404f12f26c621104c65 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/af2759bedde59c5d2bb143765d67b0ee b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/af2759bedde59c5d2bb143765d67b0ee
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/af2759bedde59c5d2bb143765d67b0ee differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/ef2b4e1176c7e77420471209362990d6 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/ef2b4e1176c7e77420471209362990d6
new file mode 100644
index 0000000000000000000000000000000000000000..c0c29ee7b5e138f7f738c2fef6bd2027a43e5690
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d/ef2b4e1176c7e77420471209362990d6
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bacfb40a5cbd9a2e6894bf0c26c1eab9b96d1029154ae528411c88868af35965
+size 451529
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d05636adb1cec915acedb08512b9d41279c880be
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..53319f15db54c58af427a6478abb37184211f4c8
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588556648359919, "commit_timestamp_nsecs": 1754588556837557852, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/d/e916d7144967b53ee10873d56ed8b4bb b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/d/e916d7144967b53ee10873d56ed8b4bb
new file mode 100644
index 0000000000000000000000000000000000000000..692e9e5723b6a6695d8f1fc7f78fdb929bacdb5a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/d/e916d7144967b53ee10873d56ed8b4bb differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4060830d67cc45cf73bc544f10c1304ecc3db549
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/132e0d41034e53202ee7902e37824cf4 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/132e0d41034e53202ee7902e37824cf4
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/132e0d41034e53202ee7902e37824cf4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/27945b82fac6165989c591e74ed9b80b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/27945b82fac6165989c591e74ed9b80b
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/27945b82fac6165989c591e74ed9b80b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/2d74f39016acc212ba2fd5406065e032 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/2d74f39016acc212ba2fd5406065e032
new file mode 100644
index 0000000000000000000000000000000000000000..9da198670db0f47f4f9d6b978c2bfe7200ebfc6b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/2d74f39016acc212ba2fd5406065e032
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4028300f120a31b041886d45d823d498637248100a1fa532cf4c8a2445a0757e
+size 125928
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/307ce4aa00bbebca5f0cb7fe8434bcf3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/307ce4aa00bbebca5f0cb7fe8434bcf3
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/307ce4aa00bbebca5f0cb7fe8434bcf3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/4adc793f00c65a632c3bd790fcbcab7d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/4adc793f00c65a632c3bd790fcbcab7d
new file mode 100644
index 0000000000000000000000000000000000000000..24c88da1b2b0a20814e92d2b36fa82a21cb4438f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/4adc793f00c65a632c3bd790fcbcab7d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/6b4fcafda285b9a8eabe3deab4773c17 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/6b4fcafda285b9a8eabe3deab4773c17
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/6b4fcafda285b9a8eabe3deab4773c17 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/8d864d198b1afd23001e9335be631f12 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/8d864d198b1afd23001e9335be631f12
new file mode 100644
index 0000000000000000000000000000000000000000..2ab9e80faa7533266475ef3d21f76e40ab071ebe
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/8d864d198b1afd23001e9335be631f12
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fe95b0a0cf2f74de6178f28b7ae92c51178a1ad52aba960b353c949f184a034b
+size 565491
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/ab8569f6d79a834da0d40c7e327ae439 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/ab8569f6d79a834da0d40c7e327ae439
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/ab8569f6d79a834da0d40c7e327ae439 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/f56e18aa0ec1812ea9687649acaa1134 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/f56e18aa0ec1812ea9687649acaa1134
new file mode 100644
index 0000000000000000000000000000000000000000..ec745d97353f97209ef50f0eb738585830219f53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d/f56e18aa0ec1812ea9687649acaa1134 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c33754a1a0efbaae9aebb68f9255ba01f2cc8c9b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..bf31ad79c7b81d366636a5fddd708f4229c730a2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588609325675759, "commit_timestamp_nsecs": 1754588609456558630, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/d/15c51a3df86173f9ce6e9d4086965a28 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/d/15c51a3df86173f9ce6e9d4086965a28
new file mode 100644
index 0000000000000000000000000000000000000000..0e62e04b2b97e44e3d2a8d292ef044f55c495bd7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/d/15c51a3df86173f9ce6e9d4086965a28 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3d934485c5df7d2db45c819d8eaa312c9a5bac95
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/1b859197ea2b4bae4c2f5e912a7ec488 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/1b859197ea2b4bae4c2f5e912a7ec488
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/1b859197ea2b4bae4c2f5e912a7ec488 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/334c79755079b1c955a884ecaaf54f9a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/334c79755079b1c955a884ecaaf54f9a
new file mode 100644
index 0000000000000000000000000000000000000000..23237aa51f54666812e599d88f0a7d15e3f6fb05
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/334c79755079b1c955a884ecaaf54f9a differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/60c77928b8b3c3c57a38a51ca3ff9c5b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/60c77928b8b3c3c57a38a51ca3ff9c5b
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/60c77928b8b3c3c57a38a51ca3ff9c5b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/71d3cfadb26db647548a71b585310c73 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/71d3cfadb26db647548a71b585310c73
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/71d3cfadb26db647548a71b585310c73 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/9245be6933980adcaf20ab85bd895536 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/9245be6933980adcaf20ab85bd895536
new file mode 100644
index 0000000000000000000000000000000000000000..243c492354b073fac84a53cd0ed756ea63dc9e14
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/9245be6933980adcaf20ab85bd895536
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:75dcf15bdb3765e0465444dd4e44e412f760d9bbd525507de82c498de7432293
+size 408816
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/b94a3767553a687f1f4911d11c2c9ccf b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/b94a3767553a687f1f4911d11c2c9ccf
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/b94a3767553a687f1f4911d11c2c9ccf differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/caaac46d44106f4e6cea779c263442e0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/caaac46d44106f4e6cea779c263442e0
new file mode 100644
index 0000000000000000000000000000000000000000..a5a84af100b8cfeab5a2fb90a3e8e8db02a9f3ac
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/caaac46d44106f4e6cea779c263442e0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/e28d870952ac9e4f675633dcbcd8906b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/e28d870952ac9e4f675633dcbcd8906b
new file mode 100644
index 0000000000000000000000000000000000000000..6c3d1fc9b48ae29be363ee3ff21aa0759841de43
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d/e28d870952ac9e4f675633dcbcd8906b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fd13371a0a512e009788975c7643abd7821c3ae8a1e1d0507a8a1c409e3602fb
+size 330158
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d88d4b5de18b3adcd38600aff76a07c485dd4a3f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..dcf7d19ff18c2c03d3a4d14493872fd3b19259fc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588642816861626, "commit_timestamp_nsecs": 1754588642960694053, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/d/9af0832ce16bfe4c89b58b4c044d25b8 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/d/9af0832ce16bfe4c89b58b4c044d25b8
new file mode 100644
index 0000000000000000000000000000000000000000..84bfaf4eb7217ce691c6c81445cd1e781658e77c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/d/9af0832ce16bfe4c89b58b4c044d25b8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..6d6013406e539b7b63fa84bdfec7091385dcf676
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/3821b9a09641046bfd0dce64d9a31df3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/3821b9a09641046bfd0dce64d9a31df3
new file mode 100644
index 0000000000000000000000000000000000000000..3892cf6aadb4cbc39fd0808ea3567fb221141c26
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/3821b9a09641046bfd0dce64d9a31df3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/688559baabc255354604cdfa4838699f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/688559baabc255354604cdfa4838699f
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/688559baabc255354604cdfa4838699f differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/8c6c99eda290c8352dc3a447a4135737 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/8c6c99eda290c8352dc3a447a4135737
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/8c6c99eda290c8352dc3a447a4135737 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/ba097af6f13358310d424325ff148e5d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/ba097af6f13358310d424325ff148e5d
new file mode 100644
index 0000000000000000000000000000000000000000..ca52e63162f93c941f53f74ff27fae582612c012
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/ba097af6f13358310d424325ff148e5d
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9b338eab05b7f9f071b048b2506b900b45ec5f05e47d0b27309b079e348835af
+size 404883
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/ba24a00cfe5ca231968d6fe6ef1828b8 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/ba24a00cfe5ca231968d6fe6ef1828b8
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/ba24a00cfe5ca231968d6fe6ef1828b8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/e8927437e8a7a41029deb73b1c4d0727 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/e8927437e8a7a41029deb73b1c4d0727
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/e8927437e8a7a41029deb73b1c4d0727 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/f8ab6ead08c6abaae2bd3f01959b32ef b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/f8ab6ead08c6abaae2bd3f01959b32ef
new file mode 100644
index 0000000000000000000000000000000000000000..d871bcc37dfa360ddf9503e83779b124f4a016f1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d/f8ab6ead08c6abaae2bd3f01959b32ef
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fcb11182fc93b153194391f557f71a70ed1b89545f523f18c3912d3a3ffc8109
+size 296159
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..6e67b3d8d19a9c0ee60a4553ba4bbdc0a1657455
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..450e8bbe0d2ab52f749240f37781e289d5963a88
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588678329694062, "commit_timestamp_nsecs": 1754588678468268910, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/d/46bd59116f9dbf3a1fd38d50864109d5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/d/46bd59116f9dbf3a1fd38d50864109d5
new file mode 100644
index 0000000000000000000000000000000000000000..5747b0e7118f8ab18fe950943fac186992f9cf2b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/d/46bd59116f9dbf3a1fd38d50864109d5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..657e153cea738b195462d6b51966e567eaa6f612
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/0fe3cff84d587b6e91f9e00e6c189e00 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/0fe3cff84d587b6e91f9e00e6c189e00
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/0fe3cff84d587b6e91f9e00e6c189e00 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/152387c990aa9b2d48c94caefc80a689 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/152387c990aa9b2d48c94caefc80a689
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/152387c990aa9b2d48c94caefc80a689 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/41ec9f385765f0a6e5970de0762978c6 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/41ec9f385765f0a6e5970de0762978c6
new file mode 100644
index 0000000000000000000000000000000000000000..c93d3dac53d1025968d4881230e4525557f57fe3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/41ec9f385765f0a6e5970de0762978c6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/83a5156dca263b6340a2067101d00dca b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/83a5156dca263b6340a2067101d00dca
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/83a5156dca263b6340a2067101d00dca differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/8d07307ea06959cc3af9914c627edcaf b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/8d07307ea06959cc3af9914c627edcaf
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/8d07307ea06959cc3af9914c627edcaf differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/acd11ec507274c1f47424fa027bfbdc5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/acd11ec507274c1f47424fa027bfbdc5
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/acd11ec507274c1f47424fa027bfbdc5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/b6f0fed6f828f72febb2fd91b3d04ccc b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/b6f0fed6f828f72febb2fd91b3d04ccc
new file mode 100644
index 0000000000000000000000000000000000000000..975f25688de4944bcba5c707c4c5c2b1cbee0e9d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/b6f0fed6f828f72febb2fd91b3d04ccc
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:086575ee33b03b56e0b8a7c3f0d46204a1c683dc38560646180c8337b25b3e87
+size 442880
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/c545edc0e85b9ceb48a0bbc0b48f3cde b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/c545edc0e85b9ceb48a0bbc0b48f3cde
new file mode 100644
index 0000000000000000000000000000000000000000..59d5deac4fbca6baa29d487697e0d888f9f129be
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d/c545edc0e85b9ceb48a0bbc0b48f3cde
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c7a5fd6d6c88a4192b3014e04c0c14ca09da284d8cc84127d17c07560f2c8feb
+size 296073
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0edf7baa6887b0f4437627168d0436bbdcd0c944
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..5c40d08982eac33ddf6cabdf26fdbd4e51f2592a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588713846091584, "commit_timestamp_nsecs": 1754588714028267692, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/d/407507e023fd95934fd7f62f4c2ab471 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/d/407507e023fd95934fd7f62f4c2ab471
new file mode 100644
index 0000000000000000000000000000000000000000..dfe3784a2bc5aba12a3adb2cbc36641e6fc1e4a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/d/407507e023fd95934fd7f62f4c2ab471 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..85d48986c966373f0edbc051225fac07be9f67e6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/3b2fc8d604e8553dec6d35e2678830f7 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/3b2fc8d604e8553dec6d35e2678830f7
new file mode 100644
index 0000000000000000000000000000000000000000..c77a1a5c3bcd268ddc944dc8f409aeca0400c5e1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/3b2fc8d604e8553dec6d35e2678830f7
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6d50b3b295672fd6db8f25187e6701665849dad9ca535d0e034bdce3055c5705
+size 460117
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/6ba685a70abb0c723997f4e971261742 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/6ba685a70abb0c723997f4e971261742
new file mode 100644
index 0000000000000000000000000000000000000000..6f2768877be737bb65d51f1bdacfe66cccd0fcce
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/6ba685a70abb0c723997f4e971261742
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:64b04d5c3acea67605a4b33427bc98ae9b89d92baa4ebb65b15b084585c9c63b
+size 245014
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/6c25bc1fca5e5341fa83bc15e3f98e39 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/6c25bc1fca5e5341fa83bc15e3f98e39
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/6c25bc1fca5e5341fa83bc15e3f98e39 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/72e8942226f1a29e63f456431b51efed b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/72e8942226f1a29e63f456431b51efed
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/72e8942226f1a29e63f456431b51efed differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/7b712c776019272db15e6bd4afa14487 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/7b712c776019272db15e6bd4afa14487
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/7b712c776019272db15e6bd4afa14487 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/b3bae20dbee63daa0060550862eb18ad b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/b3bae20dbee63daa0060550862eb18ad
new file mode 100644
index 0000000000000000000000000000000000000000..96bd718a0e1252167bb7c505d3228bc808544f01
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/b3bae20dbee63daa0060550862eb18ad differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/ec353b3e139a97d91f4eb81e4c55a984 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/ec353b3e139a97d91f4eb81e4c55a984
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/ec353b3e139a97d91f4eb81e4c55a984 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/efe10c94623f2e2862caa0128fe57bd1 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/efe10c94623f2e2862caa0128fe57bd1
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d/efe10c94623f2e2862caa0128fe57bd1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3fdacfe7e9a5893387732bb261eae17fc2a37fae
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..72121a3d3d2376b1d2920d738761c8d3fc5ffcf2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588750948660365, "commit_timestamp_nsecs": 1754588751087077082, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/d/65fbe0a06796b0d6073b4b73bf580453 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/d/65fbe0a06796b0d6073b4b73bf580453
new file mode 100644
index 0000000000000000000000000000000000000000..6c4f9d4bb1d8c161ef6e50397c119a957b98a9a7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/d/65fbe0a06796b0d6073b4b73bf580453 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f2134455378ae1fdc05bc1c6f115d5eff2646c8e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/1b045924a316604c5ddffe384adea2da b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/1b045924a316604c5ddffe384adea2da
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/1b045924a316604c5ddffe384adea2da differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/25b47c9b9c8a78ba2091270a1264b33f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/25b47c9b9c8a78ba2091270a1264b33f
new file mode 100644
index 0000000000000000000000000000000000000000..6e634ec625b1c409456dec5884ca57b1c3db1e37
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/25b47c9b9c8a78ba2091270a1264b33f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:281d633ddeced29a1e96a1862e93f0099d41ce1fb3c93198107202c866ac867b
+size 253654
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/35a4a5d59f509a7480190c96c5c89398 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/35a4a5d59f509a7480190c96c5c89398
new file mode 100644
index 0000000000000000000000000000000000000000..7bb4b7bf643a9ec4c06b906c66a26bd75343f9b5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/35a4a5d59f509a7480190c96c5c89398
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3a896639aa278d5ce97bc6ac9291b0243692703f832387dbf1e20f5456c0478d
+size 413887
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/5b5a8c8d30186ca788274f285c4e003b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/5b5a8c8d30186ca788274f285c4e003b
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/5b5a8c8d30186ca788274f285c4e003b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/619a3d4f25f0fbb44e25dd5e2834525d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/619a3d4f25f0fbb44e25dd5e2834525d
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/619a3d4f25f0fbb44e25dd5e2834525d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/6e3fee12c8a4ad62d0540ef42bc1ae9b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/6e3fee12c8a4ad62d0540ef42bc1ae9b
new file mode 100644
index 0000000000000000000000000000000000000000..ed6ec020a16188568cb8e766fa39b19678e10da5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/6e3fee12c8a4ad62d0540ef42bc1ae9b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/7ed3152b1206cf46feb48e0e2a09a0ab b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/7ed3152b1206cf46feb48e0e2a09a0ab
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/7ed3152b1206cf46feb48e0e2a09a0ab differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/bdeda2bfd1ac0e8781ded064df0fce18 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/bdeda2bfd1ac0e8781ded064df0fce18
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/bdeda2bfd1ac0e8781ded064df0fce18 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/d1abdacd3d3848901547b81996b564a2 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/d1abdacd3d3848901547b81996b564a2
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d/d1abdacd3d3848901547b81996b564a2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f267dde84e8e5ff83bb16ccdeed7ccf65ff47994
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..7e1d185662facc794ddc05603dd5ce73552c2f2c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588785083512354, "commit_timestamp_nsecs": 1754588785252634557, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/d/a25f2ff0335a5d56c4cf1f790be85458 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/d/a25f2ff0335a5d56c4cf1f790be85458
new file mode 100644
index 0000000000000000000000000000000000000000..dc81462798b78306601e1a312329a634da32a719
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/d/a25f2ff0335a5d56c4cf1f790be85458 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..bd9d299976a5d9642fa7b4afb87018139b86b373
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/403bbf03935d01b542bfdb296ae2de4f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/403bbf03935d01b542bfdb296ae2de4f
new file mode 100644
index 0000000000000000000000000000000000000000..87b152a646a42ce50180bba0e29eca7d23b8a21e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/403bbf03935d01b542bfdb296ae2de4f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:42e473f92878d5d21aa6af1d698383e8fdf5514fe4581ffba1ad29ca82ad5840
+size 287623
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/4114eb7b5a3b6f2a4093a15d9b556bd1 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/4114eb7b5a3b6f2a4093a15d9b556bd1
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/4114eb7b5a3b6f2a4093a15d9b556bd1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/5a1fb0326ec74a91d2bd5cbef5febd74 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/5a1fb0326ec74a91d2bd5cbef5febd74
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/5a1fb0326ec74a91d2bd5cbef5febd74 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/75be6ba36d9ab843c6dfa08a2a3c3dfe b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/75be6ba36d9ab843c6dfa08a2a3c3dfe
new file mode 100644
index 0000000000000000000000000000000000000000..b62e498f7b74a9bc00f2b168988ee74585619c03
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/75be6ba36d9ab843c6dfa08a2a3c3dfe differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/8a76fe1d505156947989090bc59d77cf b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/8a76fe1d505156947989090bc59d77cf
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/8a76fe1d505156947989090bc59d77cf differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/96fddc3cb0763c87e9230d374e31a255 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/96fddc3cb0763c87e9230d374e31a255
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/96fddc3cb0763c87e9230d374e31a255 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/9dad92436ad02733ea9fc741f3d73fc3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/9dad92436ad02733ea9fc741f3d73fc3
new file mode 100644
index 0000000000000000000000000000000000000000..0f87d43852518ad6a9aa5916ff344250df7320ee
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/9dad92436ad02733ea9fc741f3d73fc3
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b56be3192fd5602a33050f16e640847ce04bada7d654f621868e6e6cf821c498
+size 417472
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/9fa386f7a5b9e520b16f773f6785ff45 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/9fa386f7a5b9e520b16f773f6785ff45
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d/9fa386f7a5b9e520b16f773f6785ff45 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c2cc8b0f105bd468137ad97ea33536199481ea97
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..94ad5a728c09cf03990365550b896d8b0f8fe6fe
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754586547745005722, "commit_timestamp_nsecs": 1754586547884307161, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/d/478d3104607fc1cea639917837e14084 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/d/478d3104607fc1cea639917837e14084
new file mode 100644
index 0000000000000000000000000000000000000000..30d67142a77f33826ff43a39e06d2d43f5d6c34d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/d/478d3104607fc1cea639917837e14084 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..fc506a98ef48804d71489368caba270dc0cd5c7b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/19e4c99edb2bb52a1d87a2085b11467c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/19e4c99edb2bb52a1d87a2085b11467c
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/19e4c99edb2bb52a1d87a2085b11467c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/30b7d28dc4a0b6bc8b5860b2376a7a95 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/30b7d28dc4a0b6bc8b5860b2376a7a95
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/30b7d28dc4a0b6bc8b5860b2376a7a95 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/34109772b8581063be7a856e7a968d74 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/34109772b8581063be7a856e7a968d74
new file mode 100644
index 0000000000000000000000000000000000000000..69d8e8b3d0d2f2dc800572025499ce403dcceece
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/34109772b8581063be7a856e7a968d74
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3bedc3b5e6ff01dcfb0aa1a32b432f36ae5518d0fceb84ae2f663aa3d7a7e2ed
+size 295994
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/6f931c4bdb5174226f5d1fc2ef24b042 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/6f931c4bdb5174226f5d1fc2ef24b042
new file mode 100644
index 0000000000000000000000000000000000000000..c2fe31d01da8ce285f7415ec12f0289493310aab
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/6f931c4bdb5174226f5d1fc2ef24b042 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/ac842e875e90b373c66963d866260595 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/ac842e875e90b373c66963d866260595
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/ac842e875e90b373c66963d866260595 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/d51d9e89a27a8b0c5ef17d2a6dd82d0a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/d51d9e89a27a8b0c5ef17d2a6dd82d0a
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/d51d9e89a27a8b0c5ef17d2a6dd82d0a differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/f1c64234b4f8bea9d89ef53e3d31bc9c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/f1c64234b4f8bea9d89ef53e3d31bc9c
new file mode 100644
index 0000000000000000000000000000000000000000..f82572e47b4688c5b0508b6b77414375229348a0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d/f1c64234b4f8bea9d89ef53e3d31bc9c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c775e1e8c8a47e165fcfe8117be4a9365311d9f8f69f4f479be2fcf0a7a353c0
+size 367075
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5f33ee6effb00494ee5018f20283c07f2be12037
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..462291f2bd704ec266d448640d6fa8bed934486f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588821643304632, "commit_timestamp_nsecs": 1754588821828257968, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/d/ab3018d88b3b52858bccdabc80b4ee14 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/d/ab3018d88b3b52858bccdabc80b4ee14
new file mode 100644
index 0000000000000000000000000000000000000000..4cfaf13ebc0e64c458fda171269cfc3608890589
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/d/ab3018d88b3b52858bccdabc80b4ee14 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e5b486229e99b614073dc2ba1f87ee08cf7d8169
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/106e97a283c30044c6b12514e9885273 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/106e97a283c30044c6b12514e9885273
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/106e97a283c30044c6b12514e9885273 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/2ca389ac92ebfe6a47360cf18e7fdbf8 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/2ca389ac92ebfe6a47360cf18e7fdbf8
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/2ca389ac92ebfe6a47360cf18e7fdbf8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/3b2845fe3282443ba97a7a4fd598b21d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/3b2845fe3282443ba97a7a4fd598b21d
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/3b2845fe3282443ba97a7a4fd598b21d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/3d76a645524883881aef5f13da0318b3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/3d76a645524883881aef5f13da0318b3
new file mode 100644
index 0000000000000000000000000000000000000000..c139c6c714371b3617e1404f5fde7fecc601c75c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/3d76a645524883881aef5f13da0318b3
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:861bc7dc652e5a5753c84a2986a5ca25097da5959f383a784d579bf7283632e8
+size 253615
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/72749ead221d7d88f1e0b33729bd049a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/72749ead221d7d88f1e0b33729bd049a
new file mode 100644
index 0000000000000000000000000000000000000000..d8036ee545d3ab25fba4607989d793319cb86653
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/72749ead221d7d88f1e0b33729bd049a differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/e06b7b5b5225cda069b4270973050a3d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/e06b7b5b5225cda069b4270973050a3d
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/e06b7b5b5225cda069b4270973050a3d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/e7a8073aadff11c2cacd6ca5265cd363 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/e7a8073aadff11c2cacd6ca5265cd363
new file mode 100644
index 0000000000000000000000000000000000000000..6a1725c0e04f3b307d52f8f05323bb0648766d26
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/e7a8073aadff11c2cacd6ca5265cd363
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f70bf9ac6c536bc89c0aeb6b5ccb224358dee8135a08fddf81c73c6d781ca866
+size 451931
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/f2789d464577184df55ab7214e64f8dc b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/f2789d464577184df55ab7214e64f8dc
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d/f2789d464577184df55ab7214e64f8dc differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0d0819918a40dac1461906f611e370a399689dce
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..a44a69d495bdce4f06f31b60f1d4bad5bc3efd66
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588856161125639, "commit_timestamp_nsecs": 1754588856338382993, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/d/1684977cbe4bbb93fec23a7de7705ce4 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/d/1684977cbe4bbb93fec23a7de7705ce4
new file mode 100644
index 0000000000000000000000000000000000000000..d9cb764ad593b48e1fd117e59bb885868ff36784
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/d/1684977cbe4bbb93fec23a7de7705ce4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..00e27b917a759f2729c3195938b1ac13d2b97c2e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/4fe127ab936cd75f2c2365bf7c6152a6 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/4fe127ab936cd75f2c2365bf7c6152a6
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/4fe127ab936cd75f2c2365bf7c6152a6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/72bd76ad221fcc8be5903e8487bd0806 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/72bd76ad221fcc8be5903e8487bd0806
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/72bd76ad221fcc8be5903e8487bd0806 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/7b770581e1f09e316bea30c960348334 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/7b770581e1f09e316bea30c960348334
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/7b770581e1f09e316bea30c960348334 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/a0aaaf2ab8b5d7511315ea05b76d8a14 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/a0aaaf2ab8b5d7511315ea05b76d8a14
new file mode 100644
index 0000000000000000000000000000000000000000..a6863cc0bf6a3a9c4b1048a2b46d5a0a63f8ff6f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/a0aaaf2ab8b5d7511315ea05b76d8a14
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:92d71cac75b0c5fe8f172e128225d2494038968d514c115f6895c53b6b4836af
+size 532875
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/b8398679fad91f9a95d8a71d6d69f16a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/b8398679fad91f9a95d8a71d6d69f16a
new file mode 100644
index 0000000000000000000000000000000000000000..f841522040472d0e5864405408b26e07ea75c4b6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/b8398679fad91f9a95d8a71d6d69f16a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4235b10b3f8c26e85d4055746788493d20574e3ada8594e924f27145cde7b9ec
+size 168464
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/ca9d1c579ced74448b4bb80aa7cd4d54 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/ca9d1c579ced74448b4bb80aa7cd4d54
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/ca9d1c579ced74448b4bb80aa7cd4d54 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/ec7b76d66d4d475c0f9bbb340cfb8215 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/ec7b76d66d4d475c0f9bbb340cfb8215
new file mode 100644
index 0000000000000000000000000000000000000000..54e5e084b4ea981777cb2f950d3a6d505c9499f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/ec7b76d66d4d475c0f9bbb340cfb8215 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/f07aea256b67d92dcd5b5c5f21feca1e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/f07aea256b67d92dcd5b5c5f21feca1e
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d/f07aea256b67d92dcd5b5c5f21feca1e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..45dd5f890d5d7e67583ca3c966f49863a2e72ebc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..748cbe52ea60d5eaff34c1c8686d479e6d6813eb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588889918082295, "commit_timestamp_nsecs": 1754588890068211346, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/d/d3344b0fb4bd636f3f16957fbaa46a16 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/d/d3344b0fb4bd636f3f16957fbaa46a16
new file mode 100644
index 0000000000000000000000000000000000000000..c30880c0d0ff1eb013d53b90efdafa41153cf120
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/d/d3344b0fb4bd636f3f16957fbaa46a16 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f69374aebc4fbc6378ecb58411ed07aff02a123e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/02dc55cc2e3627895c588caff33cf3ee b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/02dc55cc2e3627895c588caff33cf3ee
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/02dc55cc2e3627895c588caff33cf3ee differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/23ad7dbb10c83134e65ca3bfc44600c6 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/23ad7dbb10c83134e65ca3bfc44600c6
new file mode 100644
index 0000000000000000000000000000000000000000..1d485a19961c44e1744d1c8038e351390d4992aa
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/23ad7dbb10c83134e65ca3bfc44600c6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/27d330023e28685dd2583d317176837b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/27d330023e28685dd2583d317176837b
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/27d330023e28685dd2583d317176837b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/390cf6ab50c7a9c56d0ac69420c1badb b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/390cf6ab50c7a9c56d0ac69420c1badb
new file mode 100644
index 0000000000000000000000000000000000000000..a5245a5efc2dc9a538ce9dce2a639b4bcf93a130
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/390cf6ab50c7a9c56d0ac69420c1badb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c0be27cf4d34671865e95cec5cabc4725af53039d24041b1d1e0d4cdbd8e327a
+size 413792
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/56c48490ce0f2cbfa3f6390a161db30f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/56c48490ce0f2cbfa3f6390a161db30f
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/56c48490ce0f2cbfa3f6390a161db30f differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/7b81f23b96ebc55b9b8257dfb1c2df9a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/7b81f23b96ebc55b9b8257dfb1c2df9a
new file mode 100644
index 0000000000000000000000000000000000000000..bac42f5deb34c0b8de498fc1edc89ef486141e43
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/7b81f23b96ebc55b9b8257dfb1c2df9a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cf112fd135b5288d0662ca2626874134e4ffb6867ecd2feb309e20879533f2c2
+size 253571
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/a09b2f4a7dd8c1bde9ded99ec4d19688 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/a09b2f4a7dd8c1bde9ded99ec4d19688
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/a09b2f4a7dd8c1bde9ded99ec4d19688 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/e547c233187efb0267e57b0379f4b8ba b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/e547c233187efb0267e57b0379f4b8ba
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d/e547c233187efb0267e57b0379f4b8ba differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f56014f2f00ec777bd190615b43759ef5381a669
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..2e089e4e2362ffc1771fbb435ab397bcc2c8bc03
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588924668675210, "commit_timestamp_nsecs": 1754588924852630941, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/d/c49b93a7ba5b146ccb03d7fa63c37c2e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/d/c49b93a7ba5b146ccb03d7fa63c37c2e
new file mode 100644
index 0000000000000000000000000000000000000000..230444e11677cf6021f3fe28dcbd6566989123ef
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/d/c49b93a7ba5b146ccb03d7fa63c37c2e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a0deb64eeb895dc2ba7ede285fcd6061f729785e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/0ffea5a10eea79ffc62ae36790b53de7 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/0ffea5a10eea79ffc62ae36790b53de7
new file mode 100644
index 0000000000000000000000000000000000000000..947bb95ee54e65323425fe49136da7632de79d4d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/0ffea5a10eea79ffc62ae36790b53de7
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:83fe83f18924f3a08a3d3e7f65529a1722edf612a10efac1d4bb64e45f752ea2
+size 304659
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/13df14fbf7905960cbc2461a566bfb0d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/13df14fbf7905960cbc2461a566bfb0d
new file mode 100644
index 0000000000000000000000000000000000000000..f28d59e1f9187cfd65b688b4bceb57a81f0acdcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/13df14fbf7905960cbc2461a566bfb0d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/2084d415460d422beabeb4d438e8d540 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/2084d415460d422beabeb4d438e8d540
new file mode 100644
index 0000000000000000000000000000000000000000..6a9445104a4aa74939befc45189bd5df0e271b2c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/2084d415460d422beabeb4d438e8d540
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c471b49af39ee4351e84794fe2665a1f82a4173f5cc46d775575a213e376002c
+size 425725
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/55e7589d14d7e3a29e1b34fe7aba06e1 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/55e7589d14d7e3a29e1b34fe7aba06e1
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/55e7589d14d7e3a29e1b34fe7aba06e1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/62744e543db8d2c59c4fbcb5168ba214 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/62744e543db8d2c59c4fbcb5168ba214
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/62744e543db8d2c59c4fbcb5168ba214 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/8c4a15cd548cfdf1c85628e57775a513 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/8c4a15cd548cfdf1c85628e57775a513
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/8c4a15cd548cfdf1c85628e57775a513 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/be46f0d56e8341f77be3dbc601bba120 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/be46f0d56e8341f77be3dbc601bba120
new file mode 100644
index 0000000000000000000000000000000000000000..4e5bf6b098477c3d854baaf615e82501902dbf8f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/be46f0d56e8341f77be3dbc601bba120 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/bea080481ed6ba8d20b22f9e9c43b609 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/bea080481ed6ba8d20b22f9e9c43b609
new file mode 100644
index 0000000000000000000000000000000000000000..cd8da373ae279421827eaadeb0534e6225487315
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d/bea080481ed6ba8d20b22f9e9c43b609 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e5e66f1a41fc9237df2a7cd9832834aab844c219
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..0151d8dc8fea62f7ab3f467ec4ce5bd5516ab981
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588958934667169, "commit_timestamp_nsecs": 1754588959082414348, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/d/e35dcabda78f80096d306afbd2a859b4 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/d/e35dcabda78f80096d306afbd2a859b4
new file mode 100644
index 0000000000000000000000000000000000000000..65badfa1d7a7c78cca2c5bf2c876918b845b1ef5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/d/e35dcabda78f80096d306afbd2a859b4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d85a6b44315ad244a54a97f0be7b34f42dc83082
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/04c8670cfbaa4cfbb9c9a481143f7e16 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/04c8670cfbaa4cfbb9c9a481143f7e16
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/04c8670cfbaa4cfbb9c9a481143f7e16 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/1952e2717c4e9fd9020df41b058137e5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/1952e2717c4e9fd9020df41b058137e5
new file mode 100644
index 0000000000000000000000000000000000000000..d74cc5665dac15ef05b0d57789889467fbad0b1f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/1952e2717c4e9fd9020df41b058137e5
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:eaccfc21df04499c3b9cd5ff1e536c901e4e766cb885880ec7ffef47e00d09fd
+size 227960
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/280576aacf35c2ab5a3093332d7b70f7 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/280576aacf35c2ab5a3093332d7b70f7
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/280576aacf35c2ab5a3093332d7b70f7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/5ecfab378eb20a0a7cb34b751c68a545 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/5ecfab378eb20a0a7cb34b751c68a545
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/5ecfab378eb20a0a7cb34b751c68a545 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/7ff7d28def801de40b935cabb5e6676e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/7ff7d28def801de40b935cabb5e6676e
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/7ff7d28def801de40b935cabb5e6676e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/e17cb904d17cdd98c5dedb4bbf0ca5f5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/e17cb904d17cdd98c5dedb4bbf0ca5f5
new file mode 100644
index 0000000000000000000000000000000000000000..462a13d6b3f973a9a7959bb2254676f652b3003e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/e17cb904d17cdd98c5dedb4bbf0ca5f5
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:716ce7a5d2f7a68d739a37597a003dba53479805b61ec530a6fe124992c53867
+size 473342
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/eb5c55d74e5cb58fc01f9d2e98c50202 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/eb5c55d74e5cb58fc01f9d2e98c50202
new file mode 100644
index 0000000000000000000000000000000000000000..ba4a1195a19fc66da5ef828f965df019ea2f2de1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/eb5c55d74e5cb58fc01f9d2e98c50202 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/fb96360d47168f608b1861c2feca16b3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/fb96360d47168f608b1861c2feca16b3
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d/fb96360d47168f608b1861c2feca16b3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f11b439a2360b63805274c7e8c0ea091c982e637
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..24106699495d9f8e544fc8b7e226aae372d26afc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754588992950481195, "commit_timestamp_nsecs": 1754588993081257022, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/d/5d1d61c31fda98adbcb1e26c7f6e8395 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/d/5d1d61c31fda98adbcb1e26c7f6e8395
new file mode 100644
index 0000000000000000000000000000000000000000..54ba393fb20d143b7ca3467e93b2e9a75387f00e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/d/5d1d61c31fda98adbcb1e26c7f6e8395 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..2c058618745c244693392ffcc5567aa3c49dfb40
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/1c826c84f46165e6e6f5483a8731d907 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/1c826c84f46165e6e6f5483a8731d907
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/1c826c84f46165e6e6f5483a8731d907 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/573b5afca76559bec3c7c0394d4b895e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/573b5afca76559bec3c7c0394d4b895e
new file mode 100644
index 0000000000000000000000000000000000000000..9524888f8ac0c053b53ad622ec28ddb417856d41
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/573b5afca76559bec3c7c0394d4b895e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/5bb2bf21557f4a0c13320f6122af0f68 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/5bb2bf21557f4a0c13320f6122af0f68
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/5bb2bf21557f4a0c13320f6122af0f68 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/77e4c0b8b873d524e558dc8284973c4a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/77e4c0b8b873d524e558dc8284973c4a
new file mode 100644
index 0000000000000000000000000000000000000000..9233967297351b68d3adfc4c5d9a6df7f5736cdd
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/77e4c0b8b873d524e558dc8284973c4a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:47877d569dbb37d00af4b0c5574a6e30913a7ae65eb1894ff02bf35e611b15ba
+size 598686
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/967b785c4c49cd206b5f6f5c7e3540bf b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/967b785c4c49cd206b5f6f5c7e3540bf
new file mode 100644
index 0000000000000000000000000000000000000000..ef71eef9395a7c71a6d39689c85997952cc8683b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/967b785c4c49cd206b5f6f5c7e3540bf
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:79e6c0faf5913b228056e7e3aee303142d10df5b5c57cfa66267c73220419b8a
+size 140561
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/c0b4d7b76d116f92bd4880dc9f4c54b1 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/c0b4d7b76d116f92bd4880dc9f4c54b1
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/c0b4d7b76d116f92bd4880dc9f4c54b1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/f48708a8e3a84745cd461078bcff2fc3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/f48708a8e3a84745cd461078bcff2fc3
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d/f48708a8e3a84745cd461078bcff2fc3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9e1efec90acd025efe1dad875312e242a546b5bd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..3998acc9d84f7e5c4fd17ac688b4f66c47f06c1c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754589027162005775, "commit_timestamp_nsecs": 1754589027286866725, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/d/0a332b5ff645706cf048b6bc84fb5dc2 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/d/0a332b5ff645706cf048b6bc84fb5dc2
new file mode 100644
index 0000000000000000000000000000000000000000..bc2671ef71f9f2fd75e38f93135ccd66887f9912
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/d/0a332b5ff645706cf048b6bc84fb5dc2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1736c11ea4fc41350729e24f7a0b03500f983ef7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/0a5ff202bfd57d988eea644f13288d36 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/0a5ff202bfd57d988eea644f13288d36
new file mode 100644
index 0000000000000000000000000000000000000000..fd52484d985ac01636fbb8c860642cd7fcd947f1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/0a5ff202bfd57d988eea644f13288d36 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/24c4a49e01e5a91dfa14ff1139c0db04 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/24c4a49e01e5a91dfa14ff1139c0db04
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/24c4a49e01e5a91dfa14ff1139c0db04 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/3232d25cc342d06dce09b490de583076 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/3232d25cc342d06dce09b490de583076
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/3232d25cc342d06dce09b490de583076 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/503f942bd3c63100bba26a2d9d13ef99 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/503f942bd3c63100bba26a2d9d13ef99
new file mode 100644
index 0000000000000000000000000000000000000000..f91e9102a1f04db41760f810bd15fd6009b6e3ea
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/503f942bd3c63100bba26a2d9d13ef99
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:10b58f66a471d4e74706b49e6c86119ea3e7c6ef60fd91b842f88c91f064426f
+size 193826
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/63319e47df25ac4bd6a039d0f79ba4ec b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/63319e47df25ac4bd6a039d0f79ba4ec
new file mode 100644
index 0000000000000000000000000000000000000000..2a31377ca030aced51ef70fb458c6b5c47404cd3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/63319e47df25ac4bd6a039d0f79ba4ec differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/78c396edb1dbc11d04b0914055eb2928 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/78c396edb1dbc11d04b0914055eb2928
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/78c396edb1dbc11d04b0914055eb2928 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/b050dba956549802ffb8dcaa4ad3f750 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/b050dba956549802ffb8dcaa4ad3f750
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/b050dba956549802ffb8dcaa4ad3f750 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/dacf0f3f369f7e51ca9f13422fcdf425 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/dacf0f3f369f7e51ca9f13422fcdf425
new file mode 100644
index 0000000000000000000000000000000000000000..9833946ce31e346503c18a0c41598621c22e85fb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/dacf0f3f369f7e51ca9f13422fcdf425
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:85384c2bb4795bd4b60c8a063d4601569a7ccbedc30be799f670c5e4c48c07d4
+size 545549
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/e1871b5be37ef7839964a262c6c34b2e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/e1871b5be37ef7839964a262c6c34b2e
new file mode 100644
index 0000000000000000000000000000000000000000..dfbfe1a61305bab6f716f83c77874d9e3176999d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d/e1871b5be37ef7839964a262c6c34b2e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1da3efd036de887819177db566fbe185ce1ddd9a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..545de207b5e38b237310231846b0093d6f95279d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754589061952627118, "commit_timestamp_nsecs": 1754589062128780727, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/d/bd569873ea25e975e1e1d0533ada84af b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/d/bd569873ea25e975e1e1d0533ada84af
new file mode 100644
index 0000000000000000000000000000000000000000..fa03a8c62ad1a4f35d259be86a22b76e3b04af61
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/d/bd569873ea25e975e1e1d0533ada84af differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e29a63c075663750a095e4d7d3fbee1c5399f439
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/4e333a8b96275982005cf5b0407854d5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/4e333a8b96275982005cf5b0407854d5
new file mode 100644
index 0000000000000000000000000000000000000000..f99713ca24e789954328ad1e9c19ad095696df7b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/4e333a8b96275982005cf5b0407854d5
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8a4dbda1c3ac26bdbadb77bd4b15469fcbee3e106e99a7b619b74585427e1f7f
+size 122933
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/57933fddfde490a0fe2d309577b81d5e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/57933fddfde490a0fe2d309577b81d5e
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/57933fddfde490a0fe2d309577b81d5e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/a5a61b97533d111138bed40bc985870f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/a5a61b97533d111138bed40bc985870f
new file mode 100644
index 0000000000000000000000000000000000000000..93b6ab0983b6a28cff28d3b520c7ea69f7c95702
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/a5a61b97533d111138bed40bc985870f differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/b6d7ba2934c233dcac71f956ab18bf30 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/b6d7ba2934c233dcac71f956ab18bf30
new file mode 100644
index 0000000000000000000000000000000000000000..fd400e22a516ee621e199d416b0f0c1adda67036
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/b6d7ba2934c233dcac71f956ab18bf30 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/c0303a2dcdcec50dd2aa94ffe15bae5b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/c0303a2dcdcec50dd2aa94ffe15bae5b
new file mode 100644
index 0000000000000000000000000000000000000000..7e4d0f10939a3cf1c656136ba42b176d3541fab8
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/c0303a2dcdcec50dd2aa94ffe15bae5b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4ae64f6e89117877f3c3d340946986b8d3ff15905f6845fb367cd3433445725b
+size 319350
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/c190bb8e94d1bde67939583932fa4838 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/c190bb8e94d1bde67939583932fa4838
new file mode 100644
index 0000000000000000000000000000000000000000..4173e4472e755fab5e6590b28a4ab38319675924
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/c190bb8e94d1bde67939583932fa4838
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:07c5326e979f9ef00367647f2e8aa49e33e442ae73bc5e63bb0599394ed0c91f
+size 298730
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/c3215625417ca4df5d9242a32f4a45e4 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/c3215625417ca4df5d9242a32f4a45e4
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d/c3215625417ca4df5d9242a32f4a45e4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..eb5ca3b1907f3b3d9acf12115e41e1f89a6a7a26
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..bb23cb8cfa496831af582adc80b4787f478708c1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754589096535128610, "commit_timestamp_nsecs": 1754589096677682427, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/d/1b6cdba2e5f093b4ad89872e57a76427 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/d/1b6cdba2e5f093b4ad89872e57a76427
new file mode 100644
index 0000000000000000000000000000000000000000..7bb7117abfec4f2e814ad398c4e2cafffa6b12e1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/d/1b6cdba2e5f093b4ad89872e57a76427 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..510bc4b0f35eb6bcc5b91c54ba41932bba34ece7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/0fabad71892ca8ce40af7adeb5eb39c3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/0fabad71892ca8ce40af7adeb5eb39c3
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/0fabad71892ca8ce40af7adeb5eb39c3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/3cd29c51a5021c9e8955767c88f0dbb9 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/3cd29c51a5021c9e8955767c88f0dbb9
new file mode 100644
index 0000000000000000000000000000000000000000..c0fdf0d6fceb491ae2011d0083dd94dc708437d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/3cd29c51a5021c9e8955767c88f0dbb9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/6e410b28cdb2c86c86c146f134612f44 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/6e410b28cdb2c86c86c146f134612f44
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/6e410b28cdb2c86c86c146f134612f44 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/700c88a987e7c6af14760133ae5768f7 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/700c88a987e7c6af14760133ae5768f7
new file mode 100644
index 0000000000000000000000000000000000000000..c6c4e2041b9178414e37967ef3ff29be9b544c9a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/700c88a987e7c6af14760133ae5768f7
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f559e2802b6dde5f1ab8cc73ddd92883819eae02afdf46dc70960cce0079d16e
+size 447652
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/8183f9dd34cf0a8bd0cd6b23f651f517 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/8183f9dd34cf0a8bd0cd6b23f651f517
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/8183f9dd34cf0a8bd0cd6b23f651f517 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/902bc4d73c812d5b399aa2ddc6cfb799 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/902bc4d73c812d5b399aa2ddc6cfb799
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/902bc4d73c812d5b399aa2ddc6cfb799 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/bfe360c45b13162eb49fb0dd67a701b5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/bfe360c45b13162eb49fb0dd67a701b5
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/bfe360c45b13162eb49fb0dd67a701b5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/cc5aa590fd4b02b04acf77e999ac630d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/cc5aa590fd4b02b04acf77e999ac630d
new file mode 100644
index 0000000000000000000000000000000000000000..2413cddd64306a3906024a0ca7238f92ee9595f9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d/cc5aa590fd4b02b04acf77e999ac630d
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b86911264dc201c66fe4aa74f05f55e4bfcf59c059fefc19ef6d2a8f7be1cfd8
+size 287584
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..49fa3a1953c9344a3d7b083a780da93153eee7dd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..1d822112c53224559be518ef4f0562406cf5bed7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754589131176896468, "commit_timestamp_nsecs": 1754589131357440852, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/d/b2aa11e72b22edaeed6d6e601c6fdc00 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/d/b2aa11e72b22edaeed6d6e601c6fdc00
new file mode 100644
index 0000000000000000000000000000000000000000..37e680fcc822c803b270b18d0ec19cd0e57bbf79
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/d/b2aa11e72b22edaeed6d6e601c6fdc00 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1fc6f0e36ae68b8037e43ea595275489bcef04aa
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/09b0b7c9d701be6bc3ec3ed124ee7311 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/09b0b7c9d701be6bc3ec3ed124ee7311
new file mode 100644
index 0000000000000000000000000000000000000000..d779343b8e6573f15ce07de2ca3b4910af43f509
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/09b0b7c9d701be6bc3ec3ed124ee7311
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:005cf2d18cb8ecd6d4458692f1ed9f6dcb14fc08c87a70a6be28ae62b759e9fb
+size 443036
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/11580827ed6a10ebbd9750eea72e2e73 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/11580827ed6a10ebbd9750eea72e2e73
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/11580827ed6a10ebbd9750eea72e2e73 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/1b6b723a7a1c73b29c382456511be750 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/1b6b723a7a1c73b29c382456511be750
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/1b6b723a7a1c73b29c382456511be750 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/5e4a22b665bdbda8639222245aea4f23 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/5e4a22b665bdbda8639222245aea4f23
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/5e4a22b665bdbda8639222245aea4f23 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/650a54c475dedc44f04ce03f81239d40 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/650a54c475dedc44f04ce03f81239d40
new file mode 100644
index 0000000000000000000000000000000000000000..2a31377ca030aced51ef70fb458c6b5c47404cd3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/650a54c475dedc44f04ce03f81239d40 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/c2ec1a0ba842721e37f93c69fb75fe08 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/c2ec1a0ba842721e37f93c69fb75fe08
new file mode 100644
index 0000000000000000000000000000000000000000..20f2836b14049fc15872b960cddb24229ae6e6ca
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/c2ec1a0ba842721e37f93c69fb75fe08
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:66ca28cecdde05f70e896ae515fdeb5c6946939fb7f6bda7a7fdc4321c0ffc32
+size 261898
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/d6cecaa725a5c9970a0d22701ce3945f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/d6cecaa725a5c9970a0d22701ce3945f
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/d6cecaa725a5c9970a0d22701ce3945f differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/e4d3dac0b4a84575557d4293fc057f55 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/e4d3dac0b4a84575557d4293fc057f55
new file mode 100644
index 0000000000000000000000000000000000000000..ac8cfd46571dbb223a52d5dafb21419955e13227
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d/e4d3dac0b4a84575557d4293fc057f55 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a15513c79a365b4fa067856d4ef4fab67f96376e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..eca313453c9b80f21526e969c60e4b4f99d9b0ee
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754586582353236035, "commit_timestamp_nsecs": 1754586582535215923, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/d/1c5f84608b66558a5b3b900384ce602e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/d/1c5f84608b66558a5b3b900384ce602e
new file mode 100644
index 0000000000000000000000000000000000000000..fe862f7b8f73ba1388c909f37c4edc954a4cde4d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/d/1c5f84608b66558a5b3b900384ce602e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..35b6f0e2889cc0ff2b9fd6098d1de38ccd0e95e8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/53841022c1a12a9262f0ec098f32ad22 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/53841022c1a12a9262f0ec098f32ad22
new file mode 100644
index 0000000000000000000000000000000000000000..ec745d97353f97209ef50f0eb738585830219f53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/53841022c1a12a9262f0ec098f32ad22 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/69c5e555b5aa59473e1c736982dc35ba b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/69c5e555b5aa59473e1c736982dc35ba
new file mode 100644
index 0000000000000000000000000000000000000000..f6a8c47c8d56051d4740eab5340b0d7d03da2024
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/69c5e555b5aa59473e1c736982dc35ba differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/6e8415c7d3cd7e03a169f3a89df636f8 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/6e8415c7d3cd7e03a169f3a89df636f8
new file mode 100644
index 0000000000000000000000000000000000000000..f0f44c431d30836f9fc22e8d10b10573de444feb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/6e8415c7d3cd7e03a169f3a89df636f8
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4406a474c17dbeefb19a9e1e0d04bd55f7753bc6a774c79c83e1960c6380e70f
+size 467464
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/a02758296894c7567bce7aab8e54f89e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/a02758296894c7567bce7aab8e54f89e
new file mode 100644
index 0000000000000000000000000000000000000000..f30e7860d623b87c70d82bcacdadf7a1ce57f5c0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/a02758296894c7567bce7aab8e54f89e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6cd5160475dc276446b7d09271d4fcb057dca99c15086c979b0947cdbef092f0
+size 270145
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/b558024d9c16a25bf8fac12d3bdbad76 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/b558024d9c16a25bf8fac12d3bdbad76
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/b558024d9c16a25bf8fac12d3bdbad76 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/c8de9649a18605dc177854c710920f00 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/c8de9649a18605dc177854c710920f00
new file mode 100644
index 0000000000000000000000000000000000000000..8f04a0232087bc458103aa526293b6f6faf9c9ec
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/c8de9649a18605dc177854c710920f00 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/cfebd3241d640c9d727f7fdc2c39865f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/cfebd3241d640c9d727f7fdc2c39865f
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/cfebd3241d640c9d727f7fdc2c39865f differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/f0e3aacb848f1d458cfdb0335fa214ac b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/f0e3aacb848f1d458cfdb0335fa214ac
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d/f0e3aacb848f1d458cfdb0335fa214ac differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f21a526effeeddde523c940d222bdba75183a6dd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..373c1e2919bbbbe589707cc0340aeb7ec9b7b47b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754589183173769419, "commit_timestamp_nsecs": 1754589183344607372, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/d/e423217506ca0b41d092bec264bbe181 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/d/e423217506ca0b41d092bec264bbe181
new file mode 100644
index 0000000000000000000000000000000000000000..4c7a10e65ce6a98aa404f90ece47f2620e25bda5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/d/e423217506ca0b41d092bec264bbe181 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..83acca888d05ea133b91cb6620376446bdb0cb6f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/1cd1158d2eeb2e58a3319dd523b13ae4 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/1cd1158d2eeb2e58a3319dd523b13ae4
new file mode 100644
index 0000000000000000000000000000000000000000..dfbfe1a61305bab6f716f83c77874d9e3176999d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/1cd1158d2eeb2e58a3319dd523b13ae4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/1e9fc9397e09dea61edc557df59ca56c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/1e9fc9397e09dea61edc557df59ca56c
new file mode 100644
index 0000000000000000000000000000000000000000..91b302e026b42524c19346a930fa092c06315ed5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/1e9fc9397e09dea61edc557df59ca56c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/44c0faaad68a46b1c606f34bf4027230 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/44c0faaad68a46b1c606f34bf4027230
new file mode 100644
index 0000000000000000000000000000000000000000..95bf5064e04fc9ece1c98156352c7e2d2a03b42e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/44c0faaad68a46b1c606f34bf4027230
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:276dfee7d25444d7878dadde2f10b31700b0f18f36ccacbbd1e46d3c5a769cf0
+size 451673
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/4f2a60c4b1736bb7516b9aea7d4844e3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/4f2a60c4b1736bb7516b9aea7d4844e3
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/4f2a60c4b1736bb7516b9aea7d4844e3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/72f4cd7b128fc2cabed291fdec306dde b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/72f4cd7b128fc2cabed291fdec306dde
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/72f4cd7b128fc2cabed291fdec306dde differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/8738e9d08a653b2d2495d4eb09bce225 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/8738e9d08a653b2d2495d4eb09bce225
new file mode 100644
index 0000000000000000000000000000000000000000..2910fbc6db2094404ed932c7d9bcdb23beb99906
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/8738e9d08a653b2d2495d4eb09bce225 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/d706746ce429d8219a4547500af59f0c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/d706746ce429d8219a4547500af59f0c
new file mode 100644
index 0000000000000000000000000000000000000000..deb85e9e79b6bccc029c5dbf57e153b928ce6991
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/d706746ce429d8219a4547500af59f0c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:679c79ad08efb7df27cab7d0cd52f8c40031c9019541bfb36934fe95c694d132
+size 287533
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/e2789711a73d8f422e0bc1f9b26b2d84 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/e2789711a73d8f422e0bc1f9b26b2d84
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/e2789711a73d8f422e0bc1f9b26b2d84 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/ea2d37097f6eaeea4bc126928c0ad90b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/ea2d37097f6eaeea4bc126928c0ad90b
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d/ea2d37097f6eaeea4bc126928c0ad90b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..17a879ff54e642e1ee6cc065b199a9f9a7ff6a0e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..97d7b1bdb40050157f45e097f40beaabb5c4663b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754589218059206797, "commit_timestamp_nsecs": 1754589218240619362, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/d/6fc37e1d55d182d2bac1090b7b5f0de1 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/d/6fc37e1d55d182d2bac1090b7b5f0de1
new file mode 100644
index 0000000000000000000000000000000000000000..f62f720e0bd500bd7f1d856cd6df62adaeff4c90
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/d/6fc37e1d55d182d2bac1090b7b5f0de1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..259a5bfe632988d2e930c7e6fa490f816b6d382d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/0a48b1983d04abcaeeb12ab1e689b7ae b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/0a48b1983d04abcaeeb12ab1e689b7ae
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/0a48b1983d04abcaeeb12ab1e689b7ae differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/1c943c1912a6c228c5d1e1ac5bfd10e0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/1c943c1912a6c228c5d1e1ac5bfd10e0
new file mode 100644
index 0000000000000000000000000000000000000000..a98dd8602d661688c7bdbc8537f321787b61df87
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/1c943c1912a6c228c5d1e1ac5bfd10e0
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f28f88883ebbe3757d123bd86e21cee40c0b0fb9bab4099e7c8a850732f643a4
+size 460176
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/3e2d5bbbe305973e37f5156eec19a0b9 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/3e2d5bbbe305973e37f5156eec19a0b9
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/3e2d5bbbe305973e37f5156eec19a0b9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/616a5238f8b54f28c03add041e41cedf b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/616a5238f8b54f28c03add041e41cedf
new file mode 100644
index 0000000000000000000000000000000000000000..4092e9026b4a7fd26532e7f9a229748a2dbe0741
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/616a5238f8b54f28c03add041e41cedf differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/6b4b0e9721e8b3dff8d3c3b79d36f4f9 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/6b4b0e9721e8b3dff8d3c3b79d36f4f9
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/6b4b0e9721e8b3dff8d3c3b79d36f4f9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/72b808dcd86df5e6e69d0339de7b74db b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/72b808dcd86df5e6e69d0339de7b74db
new file mode 100644
index 0000000000000000000000000000000000000000..4859214e6b68fa6e697de9748417fcf43be8081d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/72b808dcd86df5e6e69d0339de7b74db differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/bfee2110deb1fa4a85764c75d8839cd0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/bfee2110deb1fa4a85764c75d8839cd0
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/bfee2110deb1fa4a85764c75d8839cd0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/e0eaa98f18d08152871a6c67389b2ebb b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/e0eaa98f18d08152871a6c67389b2ebb
new file mode 100644
index 0000000000000000000000000000000000000000..1fd3ce7a4b089610e3d9aa58bef4420ddad875be
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/e0eaa98f18d08152871a6c67389b2ebb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0cec7729c48882a91cd89e051d81dd1983b4204ffb394e587908483884a5bc00
+size 278999
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/eab9c9b8e4475ab2c26b28c6600717bc b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/eab9c9b8e4475ab2c26b28c6600717bc
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d/eab9c9b8e4475ab2c26b28c6600717bc differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e25ecc1d0ae2aee82492004cb230e8a9df6b3c17
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..084aeddb6c7eca31c6e6d2c75d666f7f16b65c30
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754589254127847871, "commit_timestamp_nsecs": 1754589254269347706, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/d/6ab15ddf640990e56a78d316cb1003fe b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/d/6ab15ddf640990e56a78d316cb1003fe
new file mode 100644
index 0000000000000000000000000000000000000000..3009c922b388b3bf0ef97bab9c49f7a9d6e3488a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/d/6ab15ddf640990e56a78d316cb1003fe differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..2229a00e0d8ec91adae6e57b9c6dbb919e9d752c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/166351a1e27bdb4a0196b197188cd7a7 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/166351a1e27bdb4a0196b197188cd7a7
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/166351a1e27bdb4a0196b197188cd7a7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/238d2e5112803fe637b4eb61247e9ff6 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/238d2e5112803fe637b4eb61247e9ff6
new file mode 100644
index 0000000000000000000000000000000000000000..ceac9f063c5b7e2ae55c4b626c50f0733ab8da07
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/238d2e5112803fe637b4eb61247e9ff6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/33b84317a7b6b30d2686a2ce27c7e928 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/33b84317a7b6b30d2686a2ce27c7e928
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/33b84317a7b6b30d2686a2ce27c7e928 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/3a616a2b5470b9c260f691504c537355 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/3a616a2b5470b9c260f691504c537355
new file mode 100644
index 0000000000000000000000000000000000000000..f7a9233e726c84217acd2794d480e73fa3969d08
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/3a616a2b5470b9c260f691504c537355
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bd4dd83f52f1e84b2d618cfe7a213392f650e0423dd8dc911893533b94ac194e
+size 473363
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/4161f9ca1b6ee9d2483195ad3c67b7f4 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/4161f9ca1b6ee9d2483195ad3c67b7f4
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/4161f9ca1b6ee9d2483195ad3c67b7f4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/75dfeb3217ca84a8e806e300ec5825ae b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/75dfeb3217ca84a8e806e300ec5825ae
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/75dfeb3217ca84a8e806e300ec5825ae differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/859162359d89198b1a19302cbbb1e72c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/859162359d89198b1a19302cbbb1e72c
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/859162359d89198b1a19302cbbb1e72c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/9f6bbb7f8d8e05578b030bf0c0536ffa b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/9f6bbb7f8d8e05578b030bf0c0536ffa
new file mode 100644
index 0000000000000000000000000000000000000000..1924affcdabfdde7892c1fcb57b68ae13f20a9d7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d/9f6bbb7f8d8e05578b030bf0c0536ffa
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3a25ff1d4b56a94f8399445e56d60625a525e0443ec182bc0ea3ff571b0fb13a
+size 227953
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..912f76aeb2a5d8fb799facec2c52c653785925ae
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..b380a88cfd2bc7e0885116efb9cee632caf92337
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754589288992034611, "commit_timestamp_nsecs": 1754589289149315547, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/d/6b8c4c69dd166269de35fd28ef832a01 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/d/6b8c4c69dd166269de35fd28ef832a01
new file mode 100644
index 0000000000000000000000000000000000000000..e837698a5b5666de33ab92db504ea654abeb4384
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/d/6b8c4c69dd166269de35fd28ef832a01 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3e999f686197fe34d18562383f32d6bd342816b5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/04e1cde591bd1ef5c9cb1c96cf83f795 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/04e1cde591bd1ef5c9cb1c96cf83f795
new file mode 100644
index 0000000000000000000000000000000000000000..f28d59e1f9187cfd65b688b4bceb57a81f0acdcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/04e1cde591bd1ef5c9cb1c96cf83f795 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/431dcb4572ed381e73836772942a1dc6 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/431dcb4572ed381e73836772942a1dc6
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/431dcb4572ed381e73836772942a1dc6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/9528ebe562d9f256e014979a364879f3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/9528ebe562d9f256e014979a364879f3
new file mode 100644
index 0000000000000000000000000000000000000000..bc3667df14ee815380af38bd35d5932d7e0223f3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/9528ebe562d9f256e014979a364879f3
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6c2c061955ce3e1f7cbf83b014b6f0b2576f4fe7e316a535af6192d2d514ed6b
+size 219596
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/a8425bed9234f5ab21a2963fa488a466 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/a8425bed9234f5ab21a2963fa488a466
new file mode 100644
index 0000000000000000000000000000000000000000..48be964e95f28e7ed352eeb71051e6a69a4798f4
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/a8425bed9234f5ab21a2963fa488a466
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fa1e61216d0084c85bfbbdba4991a16e66ff2566eacafe6033fa9aafb5bf7072
+size 319381
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/bc1230e68b718e048fdaedbc3e670089 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/bc1230e68b718e048fdaedbc3e670089
new file mode 100644
index 0000000000000000000000000000000000000000..255804f3b5ea426655f0c90344ea4a97757545a5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/bc1230e68b718e048fdaedbc3e670089
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5d1205bf6428bb0e7b6eedbf4c55b13da1722978e1738cbdfc3165b859896a96
+size 202427
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/d9db91482743f9081ec13c4603764317 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/d9db91482743f9081ec13c4603764317
new file mode 100644
index 0000000000000000000000000000000000000000..0f29792cbddd4f3bd4f3493e50a8b46039eaa9b9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/d9db91482743f9081ec13c4603764317 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/df8ed9a652ebe051c668daf446109aa2 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/df8ed9a652ebe051c668daf446109aa2
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/df8ed9a652ebe051c668daf446109aa2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/f0265b1c699fbb4a244f16b4d915f5c7 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/f0265b1c699fbb4a244f16b4d915f5c7
new file mode 100644
index 0000000000000000000000000000000000000000..2910fbc6db2094404ed932c7d9bcdb23beb99906
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d/f0265b1c699fbb4a244f16b4d915f5c7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..ceb00e42a5b5f25b1a3bbedd47e1d138c257f3ab
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..5e5f0e18f9a8e7091db0c49fed69b85bce672de9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754589323959667062, "commit_timestamp_nsecs": 1754589324129086556, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/d/60165fa5b4112a4d3c84ac7fdafff748 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/d/60165fa5b4112a4d3c84ac7fdafff748
new file mode 100644
index 0000000000000000000000000000000000000000..3a971ef14ecf09caa41e5a621654659ddb0116c8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/d/60165fa5b4112a4d3c84ac7fdafff748 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a2da68445099f3887eaf13ac7c76a885dc594a72
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/01194d68df2ee0a19681d5a86a65b42b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/01194d68df2ee0a19681d5a86a65b42b
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/01194d68df2ee0a19681d5a86a65b42b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/0333b6a267c54c8388d5cb31521a4867 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/0333b6a267c54c8388d5cb31521a4867
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/0333b6a267c54c8388d5cb31521a4867 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/3811ab4b05a5c47884b52e6d18acbf55 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/3811ab4b05a5c47884b52e6d18acbf55
new file mode 100644
index 0000000000000000000000000000000000000000..aaea3280ad1b875668845b438b0bde5894e97bd4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/3811ab4b05a5c47884b52e6d18acbf55 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/4872d47e7190d3f2056d2618df22d2cc b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/4872d47e7190d3f2056d2618df22d2cc
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/4872d47e7190d3f2056d2618df22d2cc differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/576fd038bb4ddcfbc15c4ff1b1ea5568 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/576fd038bb4ddcfbc15c4ff1b1ea5568
new file mode 100644
index 0000000000000000000000000000000000000000..65529f6532287fad4b450ca493011b55e8bf7f92
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/576fd038bb4ddcfbc15c4ff1b1ea5568 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/870c41d4157c8fffab929ae9ce7635fa b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/870c41d4157c8fffab929ae9ce7635fa
new file mode 100644
index 0000000000000000000000000000000000000000..3b3dc33d2b20358760ebdb7c0c55a3420e6446ac
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/870c41d4157c8fffab929ae9ce7635fa
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:64ce5e75b1fcce14a24eb2bc0d2f79114458619b8cb6b55ef1a3b4d3012e08c1
+size 278993
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/94651aecc4444392d630c66af7d5c99d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/94651aecc4444392d630c66af7d5c99d
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/94651aecc4444392d630c66af7d5c99d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/a04a58fcb6bb72514602807e81a109ba b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/a04a58fcb6bb72514602807e81a109ba
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/a04a58fcb6bb72514602807e81a109ba differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/e6f4f029f75b20306db29669dd41d2ea b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/e6f4f029f75b20306db29669dd41d2ea
new file mode 100644
index 0000000000000000000000000000000000000000..24ee71ce3615da05a1dac43d0b416788a44b1181
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d/e6f4f029f75b20306db29669dd41d2ea
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b1005f10b64233695ce9eebd55a4e3b74b3aef4e72f9546f86242ee322d3ea79
+size 422529
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..7b463288eb83bc8f3e54b3a6e214cdb25a10027e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..04352b898a56a7c3c107ef82ac8fa2d131e6e6cf
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754589359016759334, "commit_timestamp_nsecs": 1754589359160558230, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/d/68b033b910aed5f3c89db626329f9fdc b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/d/68b033b910aed5f3c89db626329f9fdc
new file mode 100644
index 0000000000000000000000000000000000000000..883874d83ef322e97b36a3428093e3690ca04ef8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/d/68b033b910aed5f3c89db626329f9fdc differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..998a04af813fd7af9631c5ed7cf7b446e3d8b337
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/0ebdfee53282009ade64e12b783d591d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/0ebdfee53282009ade64e12b783d591d
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/0ebdfee53282009ade64e12b783d591d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/407e9b03b9f086d9f0708064242511ea b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/407e9b03b9f086d9f0708064242511ea
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/407e9b03b9f086d9f0708064242511ea differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/68b0815bbce62ca59daaa34778fcc710 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/68b0815bbce62ca59daaa34778fcc710
new file mode 100644
index 0000000000000000000000000000000000000000..5114a3ff6c6200b4fefe2f71d337c6c3748d0f31
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/68b0815bbce62ca59daaa34778fcc710 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/8c458d3506462d19688ffd5ae2a6fdd5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/8c458d3506462d19688ffd5ae2a6fdd5
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/8c458d3506462d19688ffd5ae2a6fdd5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/a7ae2714b2721402400b68667763000d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/a7ae2714b2721402400b68667763000d
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/a7ae2714b2721402400b68667763000d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/d5696c803c808058975b3620646d9171 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/d5696c803c808058975b3620646d9171
new file mode 100644
index 0000000000000000000000000000000000000000..bdc954c12aa5e6f71e99d9d4b0e45ae37907781f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/d5696c803c808058975b3620646d9171
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ed4e3489c1a48b7b4bdcb733ac6eb89c5f1efd81c0bae8c7e91af7587eda5d6f
+size 340741
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/ea8cab2b2507dfaa380bac159650f3ba b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/ea8cab2b2507dfaa380bac159650f3ba
new file mode 100644
index 0000000000000000000000000000000000000000..73486fbfcf03d9440385fbbb9aec325c7d7ba052
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d/ea8cab2b2507dfaa380bac159650f3ba
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a9b01cc551cf91c49625e33ee920d3c7a993b7117e30fe324b75aad73b5e72c8
+size 398206
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d54c203e5b7f7cb4b7ca1c6b7ed69f398ae57481
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..3ca4d82b2b16d73f646c2bd84b0cd73a0dd80e2a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754589392490120617, "commit_timestamp_nsecs": 1754589392656172531, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/d/ca5ad7b1e0821c76fa61b1eb356cf34c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/d/ca5ad7b1e0821c76fa61b1eb356cf34c
new file mode 100644
index 0000000000000000000000000000000000000000..71ee3e52cc97dd426a24ab2e786910512aa5d828
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/d/ca5ad7b1e0821c76fa61b1eb356cf34c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..7f52b40bebbaa241e6821b9cff8644815830dee8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/07c5e855f3eb81f35aed2effc38dc9e2 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/07c5e855f3eb81f35aed2effc38dc9e2
new file mode 100644
index 0000000000000000000000000000000000000000..ee2f3a15d5dd521a496e4400a6f16629b7aaf7e4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/07c5e855f3eb81f35aed2effc38dc9e2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/4d9725f553bdf4082ea5c3f82afcdaf5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/4d9725f553bdf4082ea5c3f82afcdaf5
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/4d9725f553bdf4082ea5c3f82afcdaf5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/870184682f428ac262ca5a560c6b07f0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/870184682f428ac262ca5a560c6b07f0
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/870184682f428ac262ca5a560c6b07f0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/8f4d7e928cce05e7da60665d68225348 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/8f4d7e928cce05e7da60665d68225348
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/8f4d7e928cce05e7da60665d68225348 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/9b582acbc0d196d802807b67d12a0077 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/9b582acbc0d196d802807b67d12a0077
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/9b582acbc0d196d802807b67d12a0077 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/acf674ac86865613f008f36802f7565e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/acf674ac86865613f008f36802f7565e
new file mode 100644
index 0000000000000000000000000000000000000000..3befd38c1f652e77cb4e22a43b89fe71eda1efa9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/acf674ac86865613f008f36802f7565e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:df074952f9ce0d792745b54bc400161244255c821479b3dd2d3683138598b231
+size 193882
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/bacba18d98ed6261212756fa64e7b2c5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/bacba18d98ed6261212756fa64e7b2c5
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/bacba18d98ed6261212756fa64e7b2c5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/e01070495badaee32af83d69566bca3e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/e01070495badaee32af83d69566bca3e
new file mode 100644
index 0000000000000000000000000000000000000000..244cdb8c811abc05c2864f9aec7c9e338acb3a5a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d/e01070495badaee32af83d69566bca3e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c753291048ffee24de8a192ae35f90a1eda0701738a3cf4f9447a2158b376639
+size 545339
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..7aced3979eaa29d032926ee4258957e526ddcec6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..82647970c00f546b527cc9b1ae1493fc1ac15e1c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754589427233499326, "commit_timestamp_nsecs": 1754589428046545282, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/d/d748efdf2f96914380681af06877e021 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/d/d748efdf2f96914380681af06877e021
new file mode 100644
index 0000000000000000000000000000000000000000..43358d17c0d3bca284eaeedfd87b79092e481e88
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/d/d748efdf2f96914380681af06877e021 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..7134b0476c1c5b81f3008d50e2cb993f7e84fa34
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/144130ad59cd8099fb0cf0d7d87701b3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/144130ad59cd8099fb0cf0d7d87701b3
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/144130ad59cd8099fb0cf0d7d87701b3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/3a9bee8c662cebbd93c12b2b420b300e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/3a9bee8c662cebbd93c12b2b420b300e
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/3a9bee8c662cebbd93c12b2b420b300e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/4ab1930c33f553df17001942787020f6 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/4ab1930c33f553df17001942787020f6
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/4ab1930c33f553df17001942787020f6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/4ad3f18c2ee76716cec9fa01fa393e4c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/4ad3f18c2ee76716cec9fa01fa393e4c
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/4ad3f18c2ee76716cec9fa01fa393e4c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/99d618b5efb2fc7d3f4d6d6fae0baf58 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/99d618b5efb2fc7d3f4d6d6fae0baf58
new file mode 100644
index 0000000000000000000000000000000000000000..9a101909f79b634465876585e7c3279f9ef99a92
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/99d618b5efb2fc7d3f4d6d6fae0baf58
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bdd02d580deefcc31c1ed93f60c8b049bc582d06439e95aa682c0e3a916c4b8c
+size 236504
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/9e7b7cc48736218b10e00ccea833bdbe b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/9e7b7cc48736218b10e00ccea833bdbe
new file mode 100644
index 0000000000000000000000000000000000000000..043a079a9e19b18e0e96325547973b8413436d08
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/9e7b7cc48736218b10e00ccea833bdbe differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/b5f78f20996eaa79677ef6a75a2a6035 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/b5f78f20996eaa79677ef6a75a2a6035
new file mode 100644
index 0000000000000000000000000000000000000000..93946239a45b3b25747b29516eb07ceb0eba9744
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/b5f78f20996eaa79677ef6a75a2a6035
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a714014005385cc138c95ee660ad95bd4c05ae6a8c26253b9956acd31b8c0a26
+size 501328
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/bbd915b90c639b0a923e92fc9f5911be b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/bbd915b90c639b0a923e92fc9f5911be
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/bbd915b90c639b0a923e92fc9f5911be differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/c6e57bff3a6ca110404a6d464d74685d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/c6e57bff3a6ca110404a6d464d74685d
new file mode 100644
index 0000000000000000000000000000000000000000..fdeb2e5230a026fe37f00759a073397ad7f42933
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d/c6e57bff3a6ca110404a6d464d74685d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..ade6d84bc0a018382518de55131e7b893410f06d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..6ad96c7e5460283466ce30351fb918e1596739c9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754589463379385911, "commit_timestamp_nsecs": 1754589463544624048, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/d/a5e0b2109faf464c5faa8aa55353865c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/d/a5e0b2109faf464c5faa8aa55353865c
new file mode 100644
index 0000000000000000000000000000000000000000..963cdfa661c37f5e89d99225739edadf370b704a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/d/a5e0b2109faf464c5faa8aa55353865c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4b2d42483f8f9ada8d5a638ce5ac04f0b952eb0d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/14c8431c61c2d38896f01b4463a6a77d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/14c8431c61c2d38896f01b4463a6a77d
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/14c8431c61c2d38896f01b4463a6a77d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/4933995b48872b2d46373cef421ec78e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/4933995b48872b2d46373cef421ec78e
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/4933995b48872b2d46373cef421ec78e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/626c1d2f14378ef122933744a0a1938d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/626c1d2f14378ef122933744a0a1938d
new file mode 100644
index 0000000000000000000000000000000000000000..c9a32ead1fc243145d58bdb47820d29416295b09
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/626c1d2f14378ef122933744a0a1938d
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0b14b87b7562fa54d649f489663c53a1c581192828a595d51e77bc6dfccd74a1
+size 287499
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/696e0cf632f839a2f8a1a28aac823a33 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/696e0cf632f839a2f8a1a28aac823a33
new file mode 100644
index 0000000000000000000000000000000000000000..dfbfe1a61305bab6f716f83c77874d9e3176999d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/696e0cf632f839a2f8a1a28aac823a33 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/6d27bce0ed7d926d48ef62c64b7c6149 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/6d27bce0ed7d926d48ef62c64b7c6149
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/6d27bce0ed7d926d48ef62c64b7c6149 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/84629412fd530e389f515410faa55a23 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/84629412fd530e389f515410faa55a23
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/84629412fd530e389f515410faa55a23 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/ae35325174df8db1e3c57116163a1595 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/ae35325174df8db1e3c57116163a1595
new file mode 100644
index 0000000000000000000000000000000000000000..d7aa5759850a0de7e7f427e6695389d86fd0c3df
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/ae35325174df8db1e3c57116163a1595
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9df3bf1cb690919213328fec470b17bc21bb8206492c39637c8174615cdbc948
+size 451560
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/b35f635e4131b83b6d214e047e675ba1 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/b35f635e4131b83b6d214e047e675ba1
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/b35f635e4131b83b6d214e047e675ba1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/e5ced31fbcc750c5af911ca89b6dc1ed b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/e5ced31fbcc750c5af911ca89b6dc1ed
new file mode 100644
index 0000000000000000000000000000000000000000..5cca0662ee4d4f40da2d3503902fdbcd0c3e0b5a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d/e5ced31fbcc750c5af911ca89b6dc1ed differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b09e4cbdf4a50b53cb3e73e416182769a65752e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..514616be0230b66968f90ed8f952c53a140e5ca3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754589497084886170, "commit_timestamp_nsecs": 1754589497250618407, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/d/2d7124595f4c8a15629d057b94801f2c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/d/2d7124595f4c8a15629d057b94801f2c
new file mode 100644
index 0000000000000000000000000000000000000000..18ea38b605c2b72c48347df1b6678329b276daee
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/d/2d7124595f4c8a15629d057b94801f2c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3528f73a1c1074037dc5e8fdbe6f4595214c7756
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/06acd2bd785c82d8ec37dd61b072b5ba b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/06acd2bd785c82d8ec37dd61b072b5ba
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/06acd2bd785c82d8ec37dd61b072b5ba differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/4ba87b317bf720ff421c862ed24891d0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/4ba87b317bf720ff421c862ed24891d0
new file mode 100644
index 0000000000000000000000000000000000000000..8ed78f2ac41e16d4816cda3bc8d21510158b7949
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/4ba87b317bf720ff421c862ed24891d0
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2e7f07eb859bbfca5f0713fded63ac44e655a04076070b8fd8aef19cb3668d89
+size 437205
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/5cd79e67b8695eec35f0bc09104bc8f7 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/5cd79e67b8695eec35f0bc09104bc8f7
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/5cd79e67b8695eec35f0bc09104bc8f7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/664b4eb65a035312f02d5ff2cceb3c6e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/664b4eb65a035312f02d5ff2cceb3c6e
new file mode 100644
index 0000000000000000000000000000000000000000..3844dbf19cf76ac0fee3b4bc38e8301412f8770d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/664b4eb65a035312f02d5ff2cceb3c6e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/73273d6e63a9d1bc6bdab27c28612668 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/73273d6e63a9d1bc6bdab27c28612668
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/73273d6e63a9d1bc6bdab27c28612668 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/ba5304723494557cb7c92c5b4142238d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/ba5304723494557cb7c92c5b4142238d
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/ba5304723494557cb7c92c5b4142238d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/f734e89f2dde04d6460abf31736681dd b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/f734e89f2dde04d6460abf31736681dd
new file mode 100644
index 0000000000000000000000000000000000000000..efeadd6a3da234f3e8093d01dfa818e5a1dbe49b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/f734e89f2dde04d6460abf31736681dd
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:95cac5a1820a9c959fe9821ea5dd8b072e05c4f634f3fa306c379e13bacf9f44
+size 262134
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/f905bf0c7e59d19580729561f5a09eca b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/f905bf0c7e59d19580729561f5a09eca
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d/f905bf0c7e59d19580729561f5a09eca differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..cc759dd6177492709c42cccb4659074a5088c5c1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..924c5a6c1ded3ad53e4fc02dde468fc4c11108f8
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754586618363966869, "commit_timestamp_nsecs": 1754586618537201091, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/d/0d343fb0e2fa0115fc79e90aec01128a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/d/0d343fb0e2fa0115fc79e90aec01128a
new file mode 100644
index 0000000000000000000000000000000000000000..78b9996e5d0e1aae86d0ca459a4e29d9ea391e91
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/d/0d343fb0e2fa0115fc79e90aec01128a differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..61c95b1af274bf0dcb7033238d4f7d0e594c241c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/0d6b0fa55af7bba5bf85f6d733065f3a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/0d6b0fa55af7bba5bf85f6d733065f3a
new file mode 100644
index 0000000000000000000000000000000000000000..db60e45c35cf8a1f630436cfe4390fa221acbdcb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/0d6b0fa55af7bba5bf85f6d733065f3a differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/1f55fb947633778b307cded566dfbf19 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/1f55fb947633778b307cded566dfbf19
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/1f55fb947633778b307cded566dfbf19 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/76451326a644547bedb5bd42127261c2 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/76451326a644547bedb5bd42127261c2
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/76451326a644547bedb5bd42127261c2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/984e9a2e7d0eb96c9f137a27b9d9fccc b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/984e9a2e7d0eb96c9f137a27b9d9fccc
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/984e9a2e7d0eb96c9f137a27b9d9fccc differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/a6b8ff5fbe051b06ad9aa06b1d60fe73 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/a6b8ff5fbe051b06ad9aa06b1d60fe73
new file mode 100644
index 0000000000000000000000000000000000000000..1ba65002b72370b92613b9d59c264e9f789346e6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/a6b8ff5fbe051b06ad9aa06b1d60fe73
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:396aa9376e21389f89ee07b2664c3914d90b2e4d33e809cb7f1b8de84d4dc98a
+size 319034
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/c8ece228b097304fcb8867d515442209 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/c8ece228b097304fcb8867d515442209
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/c8ece228b097304fcb8867d515442209 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/dd34b9a56539a392e544aa5f12cf6e2f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/dd34b9a56539a392e544aa5f12cf6e2f
new file mode 100644
index 0000000000000000000000000000000000000000..c2152e6f0672c05f598cb4f1336de99bff493743
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d/dd34b9a56539a392e544aa5f12cf6e2f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:47e5d56977c4d6f6182b47a64368060cad9a9bbb2feef154ff3ffdb927940455
+size 414919
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..7e7d3ff89836db4d76966ff57c72035796a40173
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..b7afbc98c8f0bd7726e361682de9941f982e3f0f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754589531870484336, "commit_timestamp_nsecs": 1754589532053984469, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/d/2ab4324420ed060f00541ff3ee8a4d21 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/d/2ab4324420ed060f00541ff3ee8a4d21
new file mode 100644
index 0000000000000000000000000000000000000000..69d336de65e370fca44019facfa0fe5be300e8a4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/d/2ab4324420ed060f00541ff3ee8a4d21 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4e6e744df31bac26a926fcde5eeaff5bac72e3be
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/200bc68bd7777ef18492b7eb370bd25f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/200bc68bd7777ef18492b7eb370bd25f
new file mode 100644
index 0000000000000000000000000000000000000000..93632073da2db0a85ac4724bf668c0d3c2cba01e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/200bc68bd7777ef18492b7eb370bd25f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:80828ad1417fa734fcebfd4a9214ddb201c954b87de810acedba19526b34b24c
+size 485405
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/478cb4fb32a3c1cc459179f087898239 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/478cb4fb32a3c1cc459179f087898239
new file mode 100644
index 0000000000000000000000000000000000000000..f87e43b8e7da29580cb5fc7a1a3b955cdbcf6552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/478cb4fb32a3c1cc459179f087898239
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:eaa7ac9c4c5b0d844684a0073aff1ed1b1fb5196052efe85f9c60a7a693689d5
+size 253583
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/5368c61f9df1cdfba8584fabf36fe843 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/5368c61f9df1cdfba8584fabf36fe843
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/5368c61f9df1cdfba8584fabf36fe843 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/67e69c2f302d8015317f906ee5dd15e2 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/67e69c2f302d8015317f906ee5dd15e2
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/67e69c2f302d8015317f906ee5dd15e2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/914408a5178cc46592166030661f2f42 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/914408a5178cc46592166030661f2f42
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/914408a5178cc46592166030661f2f42 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/babf6ba8e6b51e993862f5a97e849102 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/babf6ba8e6b51e993862f5a97e849102
new file mode 100644
index 0000000000000000000000000000000000000000..dfbfe1a61305bab6f716f83c77874d9e3176999d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/babf6ba8e6b51e993862f5a97e849102 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/d2f896567e6c98472a7eb5bd9a7db8fa b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/d2f896567e6c98472a7eb5bd9a7db8fa
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/d2f896567e6c98472a7eb5bd9a7db8fa differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/e42c8c9bc6f556b035dae2dfbd9f9683 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/e42c8c9bc6f556b035dae2dfbd9f9683
new file mode 100644
index 0000000000000000000000000000000000000000..76cb96f77bb135f4f2237fb417c8491934683b2d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d/e42c8c9bc6f556b035dae2dfbd9f9683 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d63a844fed598e791669abd6336ec03eb5f87939
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..43ec3580803794aa43d3fe973bf5f920f2837aaf
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754589566982076754, "commit_timestamp_nsecs": 1754589567163166933, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/d/df0ff1a8234f1a790de95d7cea7dfc3b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/d/df0ff1a8234f1a790de95d7cea7dfc3b
new file mode 100644
index 0000000000000000000000000000000000000000..926e26a1f648c2ebb138571486ebc0d28114d2a5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/d/df0ff1a8234f1a790de95d7cea7dfc3b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c0d681102f4241a570e21128d4c6c2b01cbd7a56
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/100f5abb20c0d73347ee7ebc5ddefbe6 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/100f5abb20c0d73347ee7ebc5ddefbe6
new file mode 100644
index 0000000000000000000000000000000000000000..24c2afa55d5a7789a8523d246e6e22a098aa7823
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/100f5abb20c0d73347ee7ebc5ddefbe6
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8fdcf3fa136bfd36d714138eb924a027f1d0d62061ad8736bb0b6c49316c4518
+size 295912
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/10c05a68e87446d366192bd16dd907e4 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/10c05a68e87446d366192bd16dd907e4
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/10c05a68e87446d366192bd16dd907e4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/52f6ea69df9c020da49222a5c543faad b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/52f6ea69df9c020da49222a5c543faad
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/52f6ea69df9c020da49222a5c543faad differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/a8615fa2baf22f26443ef55c989d6c5b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/a8615fa2baf22f26443ef55c989d6c5b
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/a8615fa2baf22f26443ef55c989d6c5b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/afbf9b634c30453b8070b57555a50553 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/afbf9b634c30453b8070b57555a50553
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/afbf9b634c30453b8070b57555a50553 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/bacdd43a4551037ce9ca0e3619e76783 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/bacdd43a4551037ce9ca0e3619e76783
new file mode 100644
index 0000000000000000000000000000000000000000..436a53566fc8f28b9157e21db2611e2c108aa177
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/bacdd43a4551037ce9ca0e3619e76783 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/cb91b00bfc15e1ccb835dccfa989357e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/cb91b00bfc15e1ccb835dccfa989357e
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/cb91b00bfc15e1ccb835dccfa989357e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/e3907512383630308ea8e1b1352de058 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/e3907512383630308ea8e1b1352de058
new file mode 100644
index 0000000000000000000000000000000000000000..4f0d9e9a43ab3558725e1a312df03c94e598be66
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d/e3907512383630308ea8e1b1352de058
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8791c7d1d46da77223f2bb20e251a9dee69611dedbbb47692491cf648d5b097f
+size 441897
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..df5eace31c874aa37410f4def8418e5817765738
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..345018d72a06269d3c29ef1e4fcce3e4f7434bc1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754589603260646837, "commit_timestamp_nsecs": 1754589603444629047, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/d/a56874e319a880a02a74c7851657f21f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/d/a56874e319a880a02a74c7851657f21f
new file mode 100644
index 0000000000000000000000000000000000000000..c9a78c1a4350d9f0e2625cf871c31ef61ae4f3bf
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/d/a56874e319a880a02a74c7851657f21f differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a5dafd250df6604d40c3f3b870f1a879a97dbc3f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/0a4413e51d4d366c5a74ad8a12a7045c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/0a4413e51d4d366c5a74ad8a12a7045c
new file mode 100644
index 0000000000000000000000000000000000000000..8ef18d4f8803ace1fb4d7880000d619c27d8b774
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/0a4413e51d4d366c5a74ad8a12a7045c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7619ea51ce133668bfb6331de3b5bb39713edb7c97cb90629768147d71adcdc7
+size 405192
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/4c86a4d6171932a8334c175ffddb8f95 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/4c86a4d6171932a8334c175ffddb8f95
new file mode 100644
index 0000000000000000000000000000000000000000..9d839d6e62935ce3e0e512f2e7810002aaad4ecb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/4c86a4d6171932a8334c175ffddb8f95
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:02a70b75ed7a014379668e0e5a49a7af2b4805a3f6ee5fa1a112ac8b03ebe943
+size 296117
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/64a79db1c79c33dd35091d1d8815a4b6 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/64a79db1c79c33dd35091d1d8815a4b6
new file mode 100644
index 0000000000000000000000000000000000000000..d6b607d1e2665fa259c424d4c6e86e93b4362e99
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/64a79db1c79c33dd35091d1d8815a4b6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/7e446a1e7f334a222b8c5ca0fd4fb6d7 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/7e446a1e7f334a222b8c5ca0fd4fb6d7
new file mode 100644
index 0000000000000000000000000000000000000000..f28d59e1f9187cfd65b688b4bceb57a81f0acdcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/7e446a1e7f334a222b8c5ca0fd4fb6d7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/814909a691598e4102229650a9a9e3a9 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/814909a691598e4102229650a9a9e3a9
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/814909a691598e4102229650a9a9e3a9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/dc6a526dc3325218fe9bd629b982600e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/dc6a526dc3325218fe9bd629b982600e
new file mode 100644
index 0000000000000000000000000000000000000000..65c55cef1e0f044fdb725f402e95d9a23d03f6b1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/dc6a526dc3325218fe9bd629b982600e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/f2870597eddba1f56e60758338f8870d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/f2870597eddba1f56e60758338f8870d
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d/f2870597eddba1f56e60758338f8870d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f750a7d4987fa1ad606f380fdda47c641f4dc910
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..f2dce28370b788dc7bf33e309ae21dd00403784e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754589637850970362, "commit_timestamp_nsecs": 1754589638017173329, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/d/72e320decef2b4101c21763867bc416c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/d/72e320decef2b4101c21763867bc416c
new file mode 100644
index 0000000000000000000000000000000000000000..74f2ab2ec1e75607fef949224cdf2002cbf61354
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/d/72e320decef2b4101c21763867bc416c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..2ef38973a00190933cb88d4d0969ce4ee7fe7389
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/1e4af8af927e46062fbb443dc55d5f61 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/1e4af8af927e46062fbb443dc55d5f61
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/1e4af8af927e46062fbb443dc55d5f61 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/25535f1b643756719edc68cb812b1a7c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/25535f1b643756719edc68cb812b1a7c
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/25535f1b643756719edc68cb812b1a7c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/4131d6cf79f40c88373655bb89592975 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/4131d6cf79f40c88373655bb89592975
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/4131d6cf79f40c88373655bb89592975 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/6398c0c9b7040735420dedf11da1035d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/6398c0c9b7040735420dedf11da1035d
new file mode 100644
index 0000000000000000000000000000000000000000..78e54591c02f6320da63d1fd8842588acf407902
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/6398c0c9b7040735420dedf11da1035d
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:57ccd4e1b83f47d74799674988578a43098777a2a6f6e1a1903e0694ad038c21
+size 426248
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/835727da2020ce0e03a034472995fc76 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/835727da2020ce0e03a034472995fc76
new file mode 100644
index 0000000000000000000000000000000000000000..f9e01ffbde69ab8075821a384f4f5d8cdd9b26e8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/835727da2020ce0e03a034472995fc76 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/ca1be08eb54daa6d8661ae846c2c603f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/ca1be08eb54daa6d8661ae846c2c603f
new file mode 100644
index 0000000000000000000000000000000000000000..f6007de2fdf5024b3a21445e2462dbaf08573f3f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/ca1be08eb54daa6d8661ae846c2c603f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cc0e0317c1ed64b293ee21c19f1262ee59e1fb0b893d876ae63e08e18d480bb9
+size 279010
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/ddfe27ff50fab4d3af464de9b75e3d15 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/ddfe27ff50fab4d3af464de9b75e3d15
new file mode 100644
index 0000000000000000000000000000000000000000..2910fbc6db2094404ed932c7d9bcdb23beb99906
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/ddfe27ff50fab4d3af464de9b75e3d15 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/e6d3fba82b1479512ac4d034cb08206e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/e6d3fba82b1479512ac4d034cb08206e
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d/e6d3fba82b1479512ac4d034cb08206e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..bc478df461c159a90d2596262d707fe9d13d15b5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..1fea02c01e70e37131bd6a4ded42b39cc74a9dee
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754589672331817555, "commit_timestamp_nsecs": 1754589672467391373, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/d/84f093954f6c11529272d4e954bc91a2 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/d/84f093954f6c11529272d4e954bc91a2
new file mode 100644
index 0000000000000000000000000000000000000000..607d67b694d91b144ae17b523652be4ee3a1f18f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/d/84f093954f6c11529272d4e954bc91a2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8c96242b778b90ce89916e64c2e71cdae9525903
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/232a1dd948e598ddf84e40e20a15538e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/232a1dd948e598ddf84e40e20a15538e
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/232a1dd948e598ddf84e40e20a15538e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/3057f0149e43236cefe8611ec9d87db5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/3057f0149e43236cefe8611ec9d87db5
new file mode 100644
index 0000000000000000000000000000000000000000..52b41b5d68f41cc7327a8400420211c50da83783
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/3057f0149e43236cefe8611ec9d87db5
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:07c0f97531d716eb9720cc0c48912f3a052c4f6269928ee8566f75e414e389fc
+size 386511
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/4c5ff490a7a87ef58ba8c50791a5cf67 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/4c5ff490a7a87ef58ba8c50791a5cf67
new file mode 100644
index 0000000000000000000000000000000000000000..abcf203f4bdd3ff9a885c699b46437d0e53d89fd
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/4c5ff490a7a87ef58ba8c50791a5cf67
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:018ba0f709364197ed2e4f7a21c3d5765254aa649e42e9429e8d0d7e739ae20d
+size 279047
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/8f8330bb646d03df6069ed2f06570ad3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/8f8330bb646d03df6069ed2f06570ad3
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/8f8330bb646d03df6069ed2f06570ad3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/a6f9661c3148f74a2516af26058f912e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/a6f9661c3148f74a2516af26058f912e
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/a6f9661c3148f74a2516af26058f912e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/b84613f7c3d7e97f0ffc4d3272ea2965 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/b84613f7c3d7e97f0ffc4d3272ea2965
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/b84613f7c3d7e97f0ffc4d3272ea2965 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/f37863e49eced55756adb65ba09e73c0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/f37863e49eced55756adb65ba09e73c0
new file mode 100644
index 0000000000000000000000000000000000000000..96562abb40e35cd003a1299f423b5d8a439b9370
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/f37863e49eced55756adb65ba09e73c0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/f517daffbb3d3ed8c42dd7e57e5a8339 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/f517daffbb3d3ed8c42dd7e57e5a8339
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d/f517daffbb3d3ed8c42dd7e57e5a8339 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..42685753cf91ea2167c2882e0562b7d127c423cd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..7a433fbc7770aae0a3b46beb9516cdc20171856d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754589707852850360, "commit_timestamp_nsecs": 1754589708023693683, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/d/a2afe8dc1e34d9849a53dc24e13e15b7 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/d/a2afe8dc1e34d9849a53dc24e13e15b7
new file mode 100644
index 0000000000000000000000000000000000000000..e7807a5d401ab3694eae011218c627a96c5f63ec
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/d/a2afe8dc1e34d9849a53dc24e13e15b7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d827411e72ccb116d7e77fdbc6e623c85663e5f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/26b70feb6e0c8e15be37ae5b1157363f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/26b70feb6e0c8e15be37ae5b1157363f
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/26b70feb6e0c8e15be37ae5b1157363f differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/2efc4efa5aadce29a50bf3491423293f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/2efc4efa5aadce29a50bf3491423293f
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/2efc4efa5aadce29a50bf3491423293f differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/469d4083bf82912bb3f2a1247d5366f3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/469d4083bf82912bb3f2a1247d5366f3
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/469d4083bf82912bb3f2a1247d5366f3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/4e140e272887ffd60bb17fb2625c1cdf b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/4e140e272887ffd60bb17fb2625c1cdf
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/4e140e272887ffd60bb17fb2625c1cdf differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/683a451f2fc6d4bd4df546f4885f1e48 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/683a451f2fc6d4bd4df546f4885f1e48
new file mode 100644
index 0000000000000000000000000000000000000000..5d05aac96a19583c9e95528d2050edebc7e1c7ee
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/683a451f2fc6d4bd4df546f4885f1e48 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/9ae9c21a099aae8f55c678107f870608 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/9ae9c21a099aae8f55c678107f870608
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/9ae9c21a099aae8f55c678107f870608 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/9ee7be16b6bea157b038e277af1d7b58 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/9ee7be16b6bea157b038e277af1d7b58
new file mode 100644
index 0000000000000000000000000000000000000000..e7ff450a01178d89a28d493f5a7d9b5b8fd34a2c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/9ee7be16b6bea157b038e277af1d7b58
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:445538a326c145139b669d63e3a21f0a952b9d039111bf1984a1e08f8ebbb641
+size 287541
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/f1d7ea393a631451032884386e1127b5 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/f1d7ea393a631451032884386e1127b5
new file mode 100644
index 0000000000000000000000000000000000000000..eed3b2c04659210aa83caa8ac09278098d6d5480
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d/f1d7ea393a631451032884386e1127b5
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1d37752ff20b81538ba991b9391eee8c8296a8f72b3c41506efe1577de13cb45
+size 417533
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9f22e4e13664fbf02cff0e6f8c78e122833f967d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..9e6bb26cebe0a2125bd40d8f05c45afe9004030b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754586339882627960, "commit_timestamp_nsecs": 1754586340055880970, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/d/a2a97445e1e5c6f21439d3af9061c65b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/d/a2a97445e1e5c6f21439d3af9061c65b
new file mode 100644
index 0000000000000000000000000000000000000000..7e5d1052b93b770237eb41f08597a4e5227810b4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/d/a2a97445e1e5c6f21439d3af9061c65b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a8917d1adb5725e5de8cbf50d0ece3620a12c2b7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/07612a0a217846776c37fc930597a9d6 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/07612a0a217846776c37fc930597a9d6
new file mode 100644
index 0000000000000000000000000000000000000000..66bd28990ef2d0f20329940060cbe1736248f7ff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/07612a0a217846776c37fc930597a9d6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/1d6c4015094c3bc53a3b70e469a9b841 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/1d6c4015094c3bc53a3b70e469a9b841
new file mode 100644
index 0000000000000000000000000000000000000000..eb95184c0defdcb1d06685f2e86be698e8dd6861
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/1d6c4015094c3bc53a3b70e469a9b841
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aac9cdfaa88f9cbafa6f851c004058503616fa06713ad56225ba9b280cdcb293
+size 729568
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/229dd9c202c83072e1e00f93f4ee832f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/229dd9c202c83072e1e00f93f4ee832f
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/229dd9c202c83072e1e00f93f4ee832f differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/4bdd36f88c79df57344f0478b527d9dd b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/4bdd36f88c79df57344f0478b527d9dd
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/4bdd36f88c79df57344f0478b527d9dd differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/e0f2739f91a6b7160dbf3eee9feb3e3e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/e0f2739f91a6b7160dbf3eee9feb3e3e
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/e0f2739f91a6b7160dbf3eee9feb3e3e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/fe1441dc073685c7b9da55c09dbfadf8 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/fe1441dc073685c7b9da55c09dbfadf8
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d/fe1441dc073685c7b9da55c09dbfadf8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..235f3d83bf2dd079ac6c6406c07e614e8b985f66
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..d93b82b486a79232fc72699c3345d69db3083506
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754586652686870249, "commit_timestamp_nsecs": 1754586652860857249, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/d/cf9b84cad0ddf618f23017d9ab8ab7ac b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/d/cf9b84cad0ddf618f23017d9ab8ab7ac
new file mode 100644
index 0000000000000000000000000000000000000000..9641a5e339735543a460ce59f0a51840e17e23a4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/d/cf9b84cad0ddf618f23017d9ab8ab7ac differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0fa4a848d7b73d50bcf6bd0aa45ed245c2b2743e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/03929fdeb40a5f7f0fd0233779c622f9 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/03929fdeb40a5f7f0fd0233779c622f9
new file mode 100644
index 0000000000000000000000000000000000000000..60acf54fc256a907300811a5ee13d03f75297000
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/03929fdeb40a5f7f0fd0233779c622f9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/0d89ae1ba936b6e0e4188473eb59d742 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/0d89ae1ba936b6e0e4188473eb59d742
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/0d89ae1ba936b6e0e4188473eb59d742 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/107212a1edddcf1191e92c5decfce4ff b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/107212a1edddcf1191e92c5decfce4ff
new file mode 100644
index 0000000000000000000000000000000000000000..0ea6540c2a78fc1a1b397b52b9572377bac98fde
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/107212a1edddcf1191e92c5decfce4ff differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/7f7a6f947497184e14896b4c45332e2d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/7f7a6f947497184e14896b4c45332e2d
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/7f7a6f947497184e14896b4c45332e2d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/a9ed4c6dcadd66afc18f8c7c5f94a0e9 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/a9ed4c6dcadd66afc18f8c7c5f94a0e9
new file mode 100644
index 0000000000000000000000000000000000000000..29a84b8744509e74e6211acf6552fe61b61e123c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/a9ed4c6dcadd66afc18f8c7c5f94a0e9
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:81afc67d35a8080993761dafbe7251ed27494abfa9319548ee103f81747217a6
+size 383050
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/de738fdeb23cd0022048bcd682161835 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/de738fdeb23cd0022048bcd682161835
new file mode 100644
index 0000000000000000000000000000000000000000..62e5a00860ed710f1f0f5cf7b8b3e6f06f23cd7b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/de738fdeb23cd0022048bcd682161835
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:304bf90d22d446232f7e7cfa273e3534fa3ad8940bd95dae7e5e8709f735372b
+size 249168
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/e5357f8186ad53881b7548b992e8d1ba b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/e5357f8186ad53881b7548b992e8d1ba
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/e5357f8186ad53881b7548b992e8d1ba differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/f071bfef30d67133db4ad554d66482a0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/f071bfef30d67133db4ad554d66482a0
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d/f071bfef30d67133db4ad554d66482a0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..442b939321bcd31f1a12c17561dbda83bc77e2c1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..05b871d1b8f85928804cf862bc2d2eec871a03c5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754586689071153416, "commit_timestamp_nsecs": 1754586689248612091, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/d/ec4a951ec733866448796f9c5dbd2c20 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/d/ec4a951ec733866448796f9c5dbd2c20
new file mode 100644
index 0000000000000000000000000000000000000000..7f1820fa735a34e7e7c1d399ac9016965e0bc810
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/d/ec4a951ec733866448796f9c5dbd2c20 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..988ddc1587b1c6a69096041bc9aef8f50b4fc343
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/2b97e3a575935a1097fccd0341808d67 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/2b97e3a575935a1097fccd0341808d67
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/2b97e3a575935a1097fccd0341808d67 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/2d59612cb9f9f84035789bf8aad5756e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/2d59612cb9f9f84035789bf8aad5756e
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/2d59612cb9f9f84035789bf8aad5756e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/54d61f7591f80e54dc453a10fea4b8cd b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/54d61f7591f80e54dc453a10fea4b8cd
new file mode 100644
index 0000000000000000000000000000000000000000..a12a1344b472e30113c907b279096a6c49c31a00
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/54d61f7591f80e54dc453a10fea4b8cd differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/5985bd0cc590d125034dafb8ee8c09a8 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/5985bd0cc590d125034dafb8ee8c09a8
new file mode 100644
index 0000000000000000000000000000000000000000..6351841f7582cd4aed8ce4a4dd226d92bea6136b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/5985bd0cc590d125034dafb8ee8c09a8
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:747c09e849f48a5191ee14c69761fc26a0bfa55e0c3a8b0d079289e638d9dc46
+size 219462
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/7cc38a5e34262b1ac1974640b9150b18 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/7cc38a5e34262b1ac1974640b9150b18
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/7cc38a5e34262b1ac1974640b9150b18 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/a5afa5c550596afbd83189632f1dc456 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/a5afa5c550596afbd83189632f1dc456
new file mode 100644
index 0000000000000000000000000000000000000000..f28d59e1f9187cfd65b688b4bceb57a81f0acdcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/a5afa5c550596afbd83189632f1dc456 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/a5df94198b83ebbf634c2805318aa2ed b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/a5df94198b83ebbf634c2805318aa2ed
new file mode 100644
index 0000000000000000000000000000000000000000..eb9cea253286a2243ba1b4381f128aef93e00553
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/a5df94198b83ebbf634c2805318aa2ed
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:23bca05ea3eba16ca03e500ce7fd676eaf7ff4aed96538a2466ca3769b23d1e1
+size 503085
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/da9cd6dd5af86445b3b2f57819671176 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/da9cd6dd5af86445b3b2f57819671176
new file mode 100644
index 0000000000000000000000000000000000000000..c0f01349b15407ec8068a2f5a42f57d8f6a08e18
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d/da9cd6dd5af86445b3b2f57819671176 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..bf3e59a1aebe73286e13ff8632a777fb8a0201fb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..d95b4937546dcd33f8c016faaf5cfb70abbfa500
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754586723950124204, "commit_timestamp_nsecs": 1754586724139831285, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/d/1bcca1107e23ac97567cafaab3c0283c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/d/1bcca1107e23ac97567cafaab3c0283c
new file mode 100644
index 0000000000000000000000000000000000000000..54d37c7c9921686b90bd198f808b37fffb3bad06
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/d/1bcca1107e23ac97567cafaab3c0283c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..830be2a44672c9b4c654a2ad12782ae5ed0c4f57
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/095495c0d9c6bea06272c5696a26e1d1 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/095495c0d9c6bea06272c5696a26e1d1
new file mode 100644
index 0000000000000000000000000000000000000000..96ab3a80271a4ac8576f40587a2d9a48080fccd3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/095495c0d9c6bea06272c5696a26e1d1
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:63c7b35e67d7890544e4cd0bd08dc6b22e2b1258d5567d7b01d6f72c88878343
+size 287366
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/0ffd4bcee81c775caf6c20718fa4f833 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/0ffd4bcee81c775caf6c20718fa4f833
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/0ffd4bcee81c775caf6c20718fa4f833 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/334f518c05073bcd8131baa9756150bc b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/334f518c05073bcd8131baa9756150bc
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/334f518c05073bcd8131baa9756150bc differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/6fdef9f94173e6f78446a5ad9fde2ecb b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/6fdef9f94173e6f78446a5ad9fde2ecb
new file mode 100644
index 0000000000000000000000000000000000000000..0f7123bd152a762d78def763c3c57b780d4837f0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/6fdef9f94173e6f78446a5ad9fde2ecb differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/d128d1444d67c1343e6934c852bc9cbd b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/d128d1444d67c1343e6934c852bc9cbd
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/d128d1444d67c1343e6934c852bc9cbd differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/d69737314c06dbd134f8b6cc95e24550 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/d69737314c06dbd134f8b6cc95e24550
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/d69737314c06dbd134f8b6cc95e24550 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/eb36eb551729ca955014039eef56323a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/eb36eb551729ca955014039eef56323a
new file mode 100644
index 0000000000000000000000000000000000000000..053a8f6cba6be6610b42dcd09eb247573c994d3a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/eb36eb551729ca955014039eef56323a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:75c45d9ef232611f979e28aff80e05f19f22f5c6e1947aa5052e55f650528385
+size 379594
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/f3d44e777e77020a3a7ddb736ccff4ed b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/f3d44e777e77020a3a7ddb736ccff4ed
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d/f3d44e777e77020a3a7ddb736ccff4ed differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..159c0010ec644ea76ef3a88c4156115e22843331
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..32333b22c385825f1dd27eee65191d033b22b7ce
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754586758920098024, "commit_timestamp_nsecs": 1754586759075067846, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/d/b1c4c8ff0fffa5fda135e848b157a216 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/d/b1c4c8ff0fffa5fda135e848b157a216
new file mode 100644
index 0000000000000000000000000000000000000000..b1d82b878c38c1a3247b2d34e4af5bd5199d5c1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/d/b1c4c8ff0fffa5fda135e848b157a216 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..ecab5a3833ebe566a11125652f389ce30e209bd5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/03a1cce077c66ec46dc628b0ecf3d3b3 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/03a1cce077c66ec46dc628b0ecf3d3b3
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/03a1cce077c66ec46dc628b0ecf3d3b3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/43a7ebc5633d4a190d209d05d9dcef0f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/43a7ebc5633d4a190d209d05d9dcef0f
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/43a7ebc5633d4a190d209d05d9dcef0f differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/69b4da8bdd92066ae7383e5e81e7ae6b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/69b4da8bdd92066ae7383e5e81e7ae6b
new file mode 100644
index 0000000000000000000000000000000000000000..d6d87c1d85f26c635840328a19adb79407c99fdd
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/69b4da8bdd92066ae7383e5e81e7ae6b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c2892ccd03f5583f25ea47649f5822db25e8932deccee610b90e67c96a9f9b20
+size 419977
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/89204a7c80254a450d657325a5798779 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/89204a7c80254a450d657325a5798779
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/89204a7c80254a450d657325a5798779 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/8ba3b3c6070083e4cf213cf6f831b498 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/8ba3b3c6070083e4cf213cf6f831b498
new file mode 100644
index 0000000000000000000000000000000000000000..7e4ba13e99bbbd8ad256d95402ca14f74ca4c6c2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/8ba3b3c6070083e4cf213cf6f831b498
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ab70d478b9403d70544d2d7c9baf50519777867d34fa79707b7d42dfc10957ad
+size 244863
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/9465c78b00417bd872aa253d4f883693 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/9465c78b00417bd872aa253d4f883693
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/9465c78b00417bd872aa253d4f883693 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/d29f405799bf03a38e94fdc58134c458 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/d29f405799bf03a38e94fdc58134c458
new file mode 100644
index 0000000000000000000000000000000000000000..3d8ce9699fd674cb9facde400c35c2ab91a24b32
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/d29f405799bf03a38e94fdc58134c458 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/da7787e0bc921ba525b51d22593493fa b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/da7787e0bc921ba525b51d22593493fa
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d/da7787e0bc921ba525b51d22593493fa differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1e073c30e7063e432e599c6fee4379c5d0ec1529
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..26496460b0180caef5c6f57b3c938a5e690b308f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754586793766724698, "commit_timestamp_nsecs": 1754586793927789596, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/d/1b72e36426cafcab8f76850ca101045a b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/d/1b72e36426cafcab8f76850ca101045a
new file mode 100644
index 0000000000000000000000000000000000000000..17d0f67f7d0249273470b98b68b039c4daa15118
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/d/1b72e36426cafcab8f76850ca101045a differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..fc6409cd013347300e7e265c8614d78359861ae6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/807b746632f698bcee6590649cfebc07 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/807b746632f698bcee6590649cfebc07
new file mode 100644
index 0000000000000000000000000000000000000000..f9c9fc6442ebd8cb501cf17ccb0d1b2f58f69fd6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/807b746632f698bcee6590649cfebc07
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:624c075a1f5bedd49b2771c20d744a98a3a67bda1d0b6d86e8d49f8c40259d1c
+size 478623
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/93ca9be2de4bdc285b7b5b2feeb50ccf b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/93ca9be2de4bdc285b7b5b2feeb50ccf
new file mode 100644
index 0000000000000000000000000000000000000000..73a988d1ab4388d49a77ef50d159725e5f3dcc46
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/93ca9be2de4bdc285b7b5b2feeb50ccf differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/a4ac6c28bb430e7d7d43d868cd98894e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/a4ac6c28bb430e7d7d43d868cd98894e
new file mode 100644
index 0000000000000000000000000000000000000000..8a8bef594028dfd47b832d6eb09c77e61f284fad
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/a4ac6c28bb430e7d7d43d868cd98894e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f585b339083baaf7f67fb7ff7fcec52018132e67b7566ee33b0126effce96b84
+size 191811
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/d76f5cc3cbeac0a837afdf7b5f266b91 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/d76f5cc3cbeac0a837afdf7b5f266b91
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/d76f5cc3cbeac0a837afdf7b5f266b91 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/dcf2733796a8a1dc431fa9bee0ea0e0e b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/dcf2733796a8a1dc431fa9bee0ea0e0e
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/dcf2733796a8a1dc431fa9bee0ea0e0e differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/f0235021438af81ff84d6686af5dc70d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/f0235021438af81ff84d6686af5dc70d
new file mode 100644
index 0000000000000000000000000000000000000000..ef5c91f9b8049d0ecd2ffe83953e0e4f2c49f972
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d/f0235021438af81ff84d6686af5dc70d differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..842430ad563d5569be884baa58998ecf77204460
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..f386ce5cbb8a28d6c776a2d6120c5e816739582f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754586828846585648, "commit_timestamp_nsecs": 1754586829022057620, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/_METADATA b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/_sharding b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/d/778a8bcb6d1e90ab9b1b4555575cd79f b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/d/778a8bcb6d1e90ab9b1b4555575cd79f
new file mode 100644
index 0000000000000000000000000000000000000000..17ad28dbcc7fa5df7c81040a8e307b2d41b7bc34
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/d/778a8bcb6d1e90ab9b1b4555575cd79f differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0c0629017e3515f1f92a2a4f741617002eb3a971
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/1014fe5159e4ebe6c1faf932d8d09d00 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/1014fe5159e4ebe6c1faf932d8d09d00
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/1014fe5159e4ebe6c1faf932d8d09d00 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/522d6a7c1fa13911e94cffb95805f34b b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/522d6a7c1fa13911e94cffb95805f34b
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/522d6a7c1fa13911e94cffb95805f34b differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/767212143f180cbdfabb302861c9f1ff b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/767212143f180cbdfabb302861c9f1ff
new file mode 100644
index 0000000000000000000000000000000000000000..f03e3f5e9bbf866d972b6c6bfd8cfdbf34f40674
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/767212143f180cbdfabb302861c9f1ff
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:69723bbff186592970bc21867896f40e88ccb4ca888ee42628f3cd5c2a654f0a
+size 287389
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/76c18eb5e36049cba8f8ba84f928bdef b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/76c18eb5e36049cba8f8ba84f928bdef
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/76c18eb5e36049cba8f8ba84f928bdef differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/77a72c8bb2b1dd6c6fdd96476e64f6f8 b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/77a72c8bb2b1dd6c6fdd96476e64f6f8
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/77a72c8bb2b1dd6c6fdd96476e64f6f8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/99c4591ca75d9a1d6256c9c6a598844c b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/99c4591ca75d9a1d6256c9c6a598844c
new file mode 100644
index 0000000000000000000000000000000000000000..0b027b2da73ad974ed761ac73a1e5cf11d7da800
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/99c4591ca75d9a1d6256c9c6a598844c differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/bb155c9aec4120dfedb634f3c2d60c6d b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/bb155c9aec4120dfedb634f3c2d60c6d
new file mode 100644
index 0000000000000000000000000000000000000000..82248cc872a7797a05650174d34d703d56529c9d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d/bb155c9aec4120dfedb634f3c2d60c6d
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:43ea8e0aa6e30db129b3213c676a5b62c8ccfa59587935cea508970de2ba50a0
+size 450987
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..66ff25e5c0cfd1339dbe150e675bc5b4f6de2497
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/config.yaml b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/config.yaml
new file mode 100644
index 0000000000000000000000000000000000000000..b1683f45e25bd3f2a2d3e0051ea70b095d06ddd9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/config.yaml
@@ -0,0 +1,1130 @@
+_wandb:
+    value:
+        cli_version: 0.21.0
+        e:
+            vx883mnozv26zdrpa9b34bes7opa5g1u:
+                args:
+                    - train
+                    - corner
+                    - --wall-prob
+                    - "1.0"
+                    - --env-size
+                    - "13"
+                    - --log-gifs
+                    - --env-layout
+                    - tree
+                    - --wandb-project
+                    - jaxgmg-wall-prob-tree
+                    - --wandb-name
+                    - wp:1.0_alpha:1
+                    - --num-total-env-steps
+                    - "200_000_000"
+                    - --prob-shift
+                    - "1"
+                    - --keep-all-checkpoints
+                    - --num-cycles-per-checkpoint
+                    - "64"
+                cpu_count: 24
+                cpu_count_logical: 48
+                cudaVersion: "12.8"
+                disk:
+                    /:
+                        total: "102005473280"
+                        used: "43331637248"
+                email: david@timaeus.co
+                executable: /workspace/timaeus/.venv/bin/python
+                git:
+                    commit: 7415666d6a2e5d63453ea9fadaf6cecf63256a59
+                    remote: git@github.com:timaeus-research/timaeus.git
+                gpu: NVIDIA GeForce RTX 3090
+                gpu_count: 1
+                gpu_nvidia:
+                    - architecture: Ampere
+                      cudaCores: 10496
+                      memoryTotal: "25769803776"
+                      name: NVIDIA GeForce RTX 3090
+                      uuid: GPU-43bd7524-bf94-73a9-1460-2b36dd09a208
+                host: 6e66e15ca303
+                memory:
+                    total: "135066595328"
+                os: Linux-5.4.0-216-generic-x86_64-with-glibc2.39
+                program: -m jaxgmg.__main__
+                python: CPython 3.11.11
+                root: /workspace/timaeus/projects/rl
+                startedAt: "2025-08-07T17:01:24.516023Z"
+                writerId: vx883mnozv26zdrpa9b34bes7opa5g1u
+        m:
+            - "1": step/env-step-all-after
+              "6":
+                - 3
+              "7": []
+            - "1": ued/layout/prop_walls_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/rollouts_gif
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/lvl_avg_return_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_advantage
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": step/ppo-update-after
+              "6":
+                - 3
+              "7": []
+            - "1": ppo/avg_critic_clipfrac
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_loss
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/avg_reward_per_step
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/proxy_corner/lvl_avg_return_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/proxy_corner/lvl_avg_return_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/lvl_avg_episode_length_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/cheese-corne_dist_wavg
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/visit_patterns/num_replay_batches
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/proxy_corner/avg_avg_return
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/lvl_benchmark_return_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/avg_benchmark_regret
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/avg_benchmark_return
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/proxy_corner/avg_avg_return
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/cheese-corner_dist_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_actor_approxkl3
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_loss
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/lvl_reward_per_step_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/lvl_avg_return_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/proxy_corner/avg_avg_return
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_actor_approxkl3
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_advantage
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_grad_norm_pre_clip
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_critic_loss
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/lvl_avg_return_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/cheese-corne_dist_avg
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/avg_avg_return
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/lvl_reward_per_step_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/levels16_img
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/lvl_benchmark_regret_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/lvl_avg_return_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/rollouts_gif
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/proxy_corner/avg_avg_return
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/mouse-cheese_dist_solvable_wavg
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/avg_reward_per_step
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/lvl_avg_episode_length_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/proxy_corner/avg_reward_per_step
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/lvl_benchmark_regret_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/avg_avg_episode_length
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/proxy_corner/lvl_reward_per_step_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/lvl_avg_return_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_actor_loss
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_actor_approxkl1
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/lvl_avg_episode_length_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/solvable_avg
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/num_walls_wavg
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/prop_walls_wavg
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_actor_clipfrac
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_actor_approxkl1
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/lvl_benchmark_return_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/rollouts_gif
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/proxy_corner/avg_reward_per_step
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/levels_cheese_is_in_corner_avg
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/avg_reward_per_step
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/avg_avg_episode_length
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/proxy_corner/lvl_reward_per_step_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/lvl_benchmark_return_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/proxy_corner/lvl_reward_per_step_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/avg_avg_episode_length
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/lvl_benchmark_regret_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/lvl_reward_per_step_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/rollouts_gif
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/mouse-cheese_dist_finite_avg
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_actor_loss
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_entropy
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/lvl_reward_per_step_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/lvl_avg_return_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_critic_clipfrac
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_actor_loss
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_grad_norm_pre_clip
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/proxy_corner/avg_avg_return
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/proxy_corner/avg_reward_per_step
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_advantage
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/avg_avg_return
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/avg_benchmark_regret
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/avg_benchmark_return
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/avg_reward_per_step
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/avg_avg_return
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/num_walls_avg
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/visit_patterns/avg_last_visit_time
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/proxy_corner/lvl_reward_per_step_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/num_walls_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_critic_loss
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_grad_norm_pre_clip
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_critic_loss
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/proxy_corner/lvl_reward_per_step_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/avg_avg_return
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/lvl_reward_per_step_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/mouse-cheese_dist_solvable_avg
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/prop_walls_avg
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/avg_benchmark_regret
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/avg_benchmark_return
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/proxy_corner/lvl_avg_return_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/avg_avg_episode_length
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/lvl_benchmark_return_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/proxy_corner/lvl_avg_return_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/proxy_corner/avg_avg_return
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/avg_avg_return
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/lvl_avg_episode_length_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/rollouts_gif
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/avg_avg_episode_length
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/rollouts_gif
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/visit_patterns/avg_first_visit_time
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/visit_patterns/last_visit_time_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/visit_patterns/prev_batch_level_ids_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_actor_clipfrac
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/proxy_corner/lvl_avg_return_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/avg_avg_episode_length
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/lvl_reward_per_step_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/scoring/scores_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/lvl_avg_episode_length_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/avg_reward_per_step
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/avg_benchmark_regret
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/avg_benchmark_return
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/shift_proportion
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/lvl_avg_episode_length_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/solvable_num
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/visit_patterns/first_visit_time_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_critic_clipfrac
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/proxy_corner/lvl_reward_per_step_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/shift_proportion
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/mouse-cheese_dist_finite_wavg
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/solvable_wavg
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/scoring/avg_scores
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_actor_approxkl1
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/proxy_corner/avg_reward_per_step
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/avg_reward_per_step
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_loss
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_actor_clipfrac
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/avg_avg_episode_length
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/proxy_corner/avg_reward_per_step
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/proxy_corner/avg_reward_per_step
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/mouse-cheese_dist_finite_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_entropy
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/lvl_reward_per_step_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/lvl_benchmark_regret_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/proxy_corner/avg_avg_return
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/rollouts_gif
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/proxy_corner/lvl_reward_per_step_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/eta
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_entropy
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/proxy_corner/lvl_avg_return_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/shift_proportion
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/avg_reward_per_step
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/avg_avg_return
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/proxy_corner/lvl_avg_return_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/lvl_avg_return_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/avg_avg_return
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/proxy_corner/avg_reward_per_step
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_actor_approxkl3
+              "5": 6
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-train/lvl_avg_episode_length_hist
+              "5": 1
+              "6":
+                - 1
+                - 3
+              "7": []
+        python_version: 3.11.11
+        t:
+            "1":
+                - 12
+                - 45
+            "2":
+                - 12
+                - 45
+            "3":
+                - 2
+                - 3
+                - 7
+                - 13
+                - 14
+                - 16
+                - 61
+            "4": 3.11.11
+            "5": 0.21.0
+            "12": 0.21.0
+            "13": linux-x86_64
+chain_mutate:
+    value: true
+checkpointing:
+    value: true
+clipping:
+    value: false
+console_log:
+    value: true
+env_corner_size:
+    value: 1
+env_layout:
+    value: tree
+env_penalize_time:
+    value: false
+env_size:
+    value: 13
+env_terminate_after_corner:
+    value: false
+evals_num_env_steps:
+    value: 512
+evals_num_levels:
+    value: 256
+gif_grid_width:
+    value: 16
+img_level_of_detail:
+    value: 1
+keep_all_checkpoints:
+    value: true
+level_splayer:
+    value: mouse
+log_gifs:
+    value: true
+log_hists:
+    value: false
+log_imgs:
+    value: true
+max_num_checkpoints:
+    value: 1
+mutate_cheese:
+    value: true
+net_cnn_type:
+    value: large
+net_rnn_type:
+    value: ff
+net_width:
+    value: 256
+num_cycles_per_big_eval:
+    value: 1024
+num_cycles_per_checkpoint:
+    value: 64
+num_cycles_per_eval:
+    value: 32
+num_cycles_per_gifs:
+    value: 1024
+num_cycles_per_log:
+    value: 32
+num_env_steps_per_cycle:
+    value: 128
+num_epochs_per_cycle:
+    value: 5
+num_minibatches_per_epoch:
+    value: 4
+num_mutate_steps:
+    value: 12
+num_parallel_envs:
+    value: 256
+num_total_env_steps:
+    value: 200000000
+num_train_levels:
+    value: 2048
+obs_level_of_detail:
+    value: 0
+plr_buffer_size:
+    value: 4096
+plr_prob_replay:
+    value: 0.5
+plr_proxy_shaping:
+    value: false
+plr_proxy_shaping_coeff:
+    value: 0.5
+plr_regret_estimator:
+    value: maxmc-actor
+plr_robust:
+    value: false
+plr_staleness_coeff:
+    value: 0.1
+plr_temperature:
+    value: 0.1
+ppo_clip_eps:
+    value: 0.1
+ppo_critic_coeff:
+    value: 0.5
+ppo_entropy_coeff:
+    value: 0.001
+ppo_gae_lambda:
+    value: 0.95
+ppo_gamma:
+    value: 0.999
+ppo_lr:
+    value: 5e-05
+ppo_lr_annealing:
+    value: false
+ppo_max_grad_norm:
+    value: 0.5
+ppo_proxy_critic_coeff:
+    value: 0.5
+prob_mutate_shift:
+    value: 0
+prob_shift:
+    value: 1
+proxy_name:
+    value: proxy_corner
+seed:
+    value: 42
+train_proxy_critic:
+    value: false
+ued:
+    value: plr
+use_fixed_eval_levels:
+    value: false
+wall_prob:
+    value: 1
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_0_45b01e0859ed8032c837.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_0_45b01e0859ed8032c837.png
new file mode 100644
index 0000000000000000000000000000000000000000..50703862507b9e85e6e0cd314bfbba624734e386
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_0_45b01e0859ed8032c837.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_0_78d4500cafdd45d60448.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_0_78d4500cafdd45d60448.png
new file mode 100644
index 0000000000000000000000000000000000000000..fccf82e6397940e46d540d996f875ca840f605f6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_0_78d4500cafdd45d60448.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1024_e944d07b6b9eccabd42e.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1024_e944d07b6b9eccabd42e.png
new file mode 100644
index 0000000000000000000000000000000000000000..b0322c7c99228f847b3aec30e470c86e39929cb2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1024_e944d07b6b9eccabd42e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1056_79890cbe4c1bdd4940c8.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1056_79890cbe4c1bdd4940c8.png
new file mode 100644
index 0000000000000000000000000000000000000000..558c7ab88ca093902a0b415a0a7c308b39022a15
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1056_79890cbe4c1bdd4940c8.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1088_a74fb04a49a211b1179e.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1088_a74fb04a49a211b1179e.png
new file mode 100644
index 0000000000000000000000000000000000000000..a891a1d232771fe033c47f04d507e3a1386636ab
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1088_a74fb04a49a211b1179e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1120_05f7fc183e2d7e3d6aa9.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1120_05f7fc183e2d7e3d6aa9.png
new file mode 100644
index 0000000000000000000000000000000000000000..7f1fcba0afbcd0a3dd2c080dedd94a04f2f858a0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1120_05f7fc183e2d7e3d6aa9.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1152_8447807df4db4d838bfe.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1152_8447807df4db4d838bfe.png
new file mode 100644
index 0000000000000000000000000000000000000000..4a303b2d22def694172936eed305abed3577bf2c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1152_8447807df4db4d838bfe.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1184_22405af1f63c7605b629.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1184_22405af1f63c7605b629.png
new file mode 100644
index 0000000000000000000000000000000000000000..c6ca4d2c75a6011af03415f275ddf782d91dfca1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1184_22405af1f63c7605b629.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1216_16fe45e524fcfba2f9bd.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1216_16fe45e524fcfba2f9bd.png
new file mode 100644
index 0000000000000000000000000000000000000000..231b9ab85b7161b7358fe3dc19e35c8746563b23
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1216_16fe45e524fcfba2f9bd.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1248_ecd14e4522e5b2cbd305.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1248_ecd14e4522e5b2cbd305.png
new file mode 100644
index 0000000000000000000000000000000000000000..3537d6369fe7b4bf68c19cabb0a06487c52ba6da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1248_ecd14e4522e5b2cbd305.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1280_33e99fb2b6cbae56d91a.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1280_33e99fb2b6cbae56d91a.png
new file mode 100644
index 0000000000000000000000000000000000000000..ec16c652dd6edf562bdfd19df9dc48044e715b9a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1280_33e99fb2b6cbae56d91a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_128_4d10ff6354f200e2ccd1.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_128_4d10ff6354f200e2ccd1.png
new file mode 100644
index 0000000000000000000000000000000000000000..24c3bf2c17cdc3c7555cb9a3d519549c2e83df8c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_128_4d10ff6354f200e2ccd1.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1312_6a99c4b38065a6ce0946.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1312_6a99c4b38065a6ce0946.png
new file mode 100644
index 0000000000000000000000000000000000000000..0df88e52b51b33fb00640f87a2e78ec010415fa9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1312_6a99c4b38065a6ce0946.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1344_f7d94ce4ad4575391402.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1344_f7d94ce4ad4575391402.png
new file mode 100644
index 0000000000000000000000000000000000000000..62c4deba9c177d618e7d826c99911424a6d22a32
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1344_f7d94ce4ad4575391402.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1376_a011063f0279fa759124.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1376_a011063f0279fa759124.png
new file mode 100644
index 0000000000000000000000000000000000000000..438b28da112c52ff17007532572871ed30110982
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1376_a011063f0279fa759124.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1408_ad99107b76b47d7aafa5.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1408_ad99107b76b47d7aafa5.png
new file mode 100644
index 0000000000000000000000000000000000000000..9d91c4c950bd3e7f19cada038a829a2a01b7bdd4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1408_ad99107b76b47d7aafa5.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1440_a3139ec19ffb09c1e0c3.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1440_a3139ec19ffb09c1e0c3.png
new file mode 100644
index 0000000000000000000000000000000000000000..62345c1b35bd43693d6f562c72195ea86f92d1fc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1440_a3139ec19ffb09c1e0c3.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1472_007aaa7f53251091257f.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1472_007aaa7f53251091257f.png
new file mode 100644
index 0000000000000000000000000000000000000000..602a0db48791515c39481ac40ea77daf6ad441d2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1472_007aaa7f53251091257f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1504_2a419f01edfee1e06499.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1504_2a419f01edfee1e06499.png
new file mode 100644
index 0000000000000000000000000000000000000000..abc41f9973773440a56231c5b19797346b5d09d9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1504_2a419f01edfee1e06499.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1536_49bc4dc169fb468e5c9f.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1536_49bc4dc169fb468e5c9f.png
new file mode 100644
index 0000000000000000000000000000000000000000..e5bd3b7ac669b66ec551184738db9da472ab6c0c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1536_49bc4dc169fb468e5c9f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1568_97d1ec419ba503d8856c.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1568_97d1ec419ba503d8856c.png
new file mode 100644
index 0000000000000000000000000000000000000000..c521b838d027b90aa033cc13ed0f9f89e0a549d5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1568_97d1ec419ba503d8856c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1600_e6b3056020cdaf8f3ad9.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1600_e6b3056020cdaf8f3ad9.png
new file mode 100644
index 0000000000000000000000000000000000000000..a4d8f35c8ad1dd321b48df437b9c7ea480aa93b0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1600_e6b3056020cdaf8f3ad9.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_160_4cc18694280605993dfe.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_160_4cc18694280605993dfe.png
new file mode 100644
index 0000000000000000000000000000000000000000..70d6b5792cb3d11a3f495da9e679c2b62b349f3c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_160_4cc18694280605993dfe.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1632_fba0d96edd97ae43ae92.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1632_fba0d96edd97ae43ae92.png
new file mode 100644
index 0000000000000000000000000000000000000000..ebfb40e3488a3550ed03cdfb5231b544e13f1a93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1632_fba0d96edd97ae43ae92.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1664_48ea41da12e2cab32ce3.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1664_48ea41da12e2cab32ce3.png
new file mode 100644
index 0000000000000000000000000000000000000000..c50e9ec97caf7614cd122271096365c77118f659
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1664_48ea41da12e2cab32ce3.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1696_d15ac6d260328dc4c1cc.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1696_d15ac6d260328dc4c1cc.png
new file mode 100644
index 0000000000000000000000000000000000000000..cf4b6b45f7c0dad084d03e87940110df79d04e83
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1696_d15ac6d260328dc4c1cc.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1728_f5eaea5fbf394f6be214.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1728_f5eaea5fbf394f6be214.png
new file mode 100644
index 0000000000000000000000000000000000000000..270ca2756e00ef4d31a241e5a976c3ad6adb75a9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1728_f5eaea5fbf394f6be214.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1760_64856757a86ad4c44460.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1760_64856757a86ad4c44460.png
new file mode 100644
index 0000000000000000000000000000000000000000..e7048a85cd19c73a66d21b3b78ca61940455c02d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1760_64856757a86ad4c44460.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1792_4e8a37353a2af2fcb4b9.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1792_4e8a37353a2af2fcb4b9.png
new file mode 100644
index 0000000000000000000000000000000000000000..46d51b6cfeaa6142e68469f125f83469967a14e6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1792_4e8a37353a2af2fcb4b9.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1824_3f3eaa228a975e3f78bc.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1824_3f3eaa228a975e3f78bc.png
new file mode 100644
index 0000000000000000000000000000000000000000..959c6a7874f517af58be262bdeb66b36ac858362
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1824_3f3eaa228a975e3f78bc.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1856_c36c914a190d60b09323.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1856_c36c914a190d60b09323.png
new file mode 100644
index 0000000000000000000000000000000000000000..48cdad8ec4be6c3a911b7e7a64c3b5fdd2c8100a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1856_c36c914a190d60b09323.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1888_c0c0a11a98ba9ab605ad.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1888_c0c0a11a98ba9ab605ad.png
new file mode 100644
index 0000000000000000000000000000000000000000..f1c90ce4d0f8c7800478ea1a8e24130aeefceeee
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1888_c0c0a11a98ba9ab605ad.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1920_d42dc0305cb5518b1025.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1920_d42dc0305cb5518b1025.png
new file mode 100644
index 0000000000000000000000000000000000000000..b317f42f205e66231941aa1fcd28782541a8c86a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1920_d42dc0305cb5518b1025.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_192_0083d640af4a6d444626.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_192_0083d640af4a6d444626.png
new file mode 100644
index 0000000000000000000000000000000000000000..7429b2ffe13bd28b3234789d06409608336c7df0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_192_0083d640af4a6d444626.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1952_8afb0927c8e5911ffaf9.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1952_8afb0927c8e5911ffaf9.png
new file mode 100644
index 0000000000000000000000000000000000000000..9f7283a92d0d3850588fd3e8c6645a0481f915c1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1952_8afb0927c8e5911ffaf9.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1984_97218794d25d53cf5fc4.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1984_97218794d25d53cf5fc4.png
new file mode 100644
index 0000000000000000000000000000000000000000..04ae94bd65ba01e362659b92a59b1fadac6dd31f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_1984_97218794d25d53cf5fc4.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2016_08a7bbe6aec765b00abf.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2016_08a7bbe6aec765b00abf.png
new file mode 100644
index 0000000000000000000000000000000000000000..4d84ed391476c3364dac5c6281500ff14b3b0a7c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2016_08a7bbe6aec765b00abf.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2048_bedcc4dab3432e7247f4.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2048_bedcc4dab3432e7247f4.png
new file mode 100644
index 0000000000000000000000000000000000000000..fbe4dc73dc80e180c1b72fe5496a040e046feb5e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2048_bedcc4dab3432e7247f4.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2080_d6ef5108928784e91c59.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2080_d6ef5108928784e91c59.png
new file mode 100644
index 0000000000000000000000000000000000000000..4958ebd0a38bcad422165bb990e85769142a4117
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2080_d6ef5108928784e91c59.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2112_1d03f772dc43c6dd5053.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2112_1d03f772dc43c6dd5053.png
new file mode 100644
index 0000000000000000000000000000000000000000..631a44a57db4795a9b0ff8e70a3756ed9c711ef5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2112_1d03f772dc43c6dd5053.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2144_d903113a85f109fa4fe7.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2144_d903113a85f109fa4fe7.png
new file mode 100644
index 0000000000000000000000000000000000000000..5d96ceb5eb8aa6d26d2e2f148b62e46b094f1c50
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2144_d903113a85f109fa4fe7.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2176_eaf3f749f03415931cd2.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2176_eaf3f749f03415931cd2.png
new file mode 100644
index 0000000000000000000000000000000000000000..5ac7bbd097c52aee8f27de9978681779e7f5e1ad
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2176_eaf3f749f03415931cd2.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2208_5da68147ec11ddda2899.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2208_5da68147ec11ddda2899.png
new file mode 100644
index 0000000000000000000000000000000000000000..4140fae7515e0ee37827f7e6a8f02da942946703
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2208_5da68147ec11ddda2899.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2240_dc425c5c4e689c948780.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2240_dc425c5c4e689c948780.png
new file mode 100644
index 0000000000000000000000000000000000000000..bde5c33ed7c225c9b72e2b9ca5f0e228e316c1e4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2240_dc425c5c4e689c948780.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_224_61cc8ecb546d3375d241.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_224_61cc8ecb546d3375d241.png
new file mode 100644
index 0000000000000000000000000000000000000000..799de55ba4699458c366ea639d57d69baa75bb07
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_224_61cc8ecb546d3375d241.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2272_3b75b58ef65376298534.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2272_3b75b58ef65376298534.png
new file mode 100644
index 0000000000000000000000000000000000000000..4a268baac7c4650a7df6c03f9f45aa56cab6f4b0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2272_3b75b58ef65376298534.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2304_b136a6ffcc06299a3a3b.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2304_b136a6ffcc06299a3a3b.png
new file mode 100644
index 0000000000000000000000000000000000000000..c8e706648ceaf133b6cc5917897a82bcc2dff9a6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2304_b136a6ffcc06299a3a3b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2336_d51c8de1d1bd6d178b6d.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2336_d51c8de1d1bd6d178b6d.png
new file mode 100644
index 0000000000000000000000000000000000000000..bbd42118a722229e6e988ca910cf953f97cf36df
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2336_d51c8de1d1bd6d178b6d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2368_9c0385652ee8addd2ba3.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2368_9c0385652ee8addd2ba3.png
new file mode 100644
index 0000000000000000000000000000000000000000..2bb23c338e3921d5d544485d37ab7f2ecfec5825
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2368_9c0385652ee8addd2ba3.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2400_a9e73ad3551187f9293f.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2400_a9e73ad3551187f9293f.png
new file mode 100644
index 0000000000000000000000000000000000000000..500629a9f89b811f74fc1af3b6eb89bb01c094e5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2400_a9e73ad3551187f9293f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2432_b35aa41af98d582bdc6b.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2432_b35aa41af98d582bdc6b.png
new file mode 100644
index 0000000000000000000000000000000000000000..8ccecc3b3b522354e32a8d1f4920a74b54e0d331
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2432_b35aa41af98d582bdc6b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2464_66109dc06ee0685c45e4.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2464_66109dc06ee0685c45e4.png
new file mode 100644
index 0000000000000000000000000000000000000000..f000028c3b7728f03bf8f937f4d766d2caa00e7d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2464_66109dc06ee0685c45e4.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2496_33c20d0297366e71e911.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2496_33c20d0297366e71e911.png
new file mode 100644
index 0000000000000000000000000000000000000000..b9fd3f4530f58d431dbcd09f49d17c289485cc89
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2496_33c20d0297366e71e911.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2528_6538b1c00d5f5256c793.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2528_6538b1c00d5f5256c793.png
new file mode 100644
index 0000000000000000000000000000000000000000..10082fc50919fcddab584b868e692e602f921fd0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2528_6538b1c00d5f5256c793.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2560_c09e823bbef9bd8604eb.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2560_c09e823bbef9bd8604eb.png
new file mode 100644
index 0000000000000000000000000000000000000000..5135aaa4cd76d46a5f6b24e4a59ea18c62c49b56
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2560_c09e823bbef9bd8604eb.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_256_b44f21b3b035c98ae982.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_256_b44f21b3b035c98ae982.png
new file mode 100644
index 0000000000000000000000000000000000000000..db8397ef1ba77989e66fc93bd02fd238df5f2015
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_256_b44f21b3b035c98ae982.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2592_45b5e39995d9ac46fdcb.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2592_45b5e39995d9ac46fdcb.png
new file mode 100644
index 0000000000000000000000000000000000000000..fd3648dc5703eb2efe320d68cc6614309136539f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2592_45b5e39995d9ac46fdcb.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2624_bd0c9108b8b23a98fa05.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2624_bd0c9108b8b23a98fa05.png
new file mode 100644
index 0000000000000000000000000000000000000000..5803ab25b8d369f63fb556a2f2bd139b0dcfb9c8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2624_bd0c9108b8b23a98fa05.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2656_5acbdd336280f5af8a35.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2656_5acbdd336280f5af8a35.png
new file mode 100644
index 0000000000000000000000000000000000000000..c08d1a17cefa3556e39fbd440ae851653f2d1727
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2656_5acbdd336280f5af8a35.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2688_2ab8334abd46f2749188.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2688_2ab8334abd46f2749188.png
new file mode 100644
index 0000000000000000000000000000000000000000..f33b4b846cd68592322824ecc26dfbf7c6596729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2688_2ab8334abd46f2749188.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2720_c0e4ed2681ef5edfbb9e.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2720_c0e4ed2681ef5edfbb9e.png
new file mode 100644
index 0000000000000000000000000000000000000000..018e7656d3ddffde57e1c76a2a9f09df40ffc7e6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2720_c0e4ed2681ef5edfbb9e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2752_9e86054759b63c8ae6e3.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2752_9e86054759b63c8ae6e3.png
new file mode 100644
index 0000000000000000000000000000000000000000..0f14355c4f47169d87c3f09de117b8b42e94422a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2752_9e86054759b63c8ae6e3.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2784_50f1b9ad438206420c50.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2784_50f1b9ad438206420c50.png
new file mode 100644
index 0000000000000000000000000000000000000000..63a70cf723a49fcf88c51021e37f511fd7290991
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2784_50f1b9ad438206420c50.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2816_330d386be1743cfdfb70.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2816_330d386be1743cfdfb70.png
new file mode 100644
index 0000000000000000000000000000000000000000..cb2c52610eb85e3de1364b07111894bdfee3415b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2816_330d386be1743cfdfb70.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2848_dd6fd5a234c5fcaaa578.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2848_dd6fd5a234c5fcaaa578.png
new file mode 100644
index 0000000000000000000000000000000000000000..1fb21269d494d523c82fdff31b3c2d4632d7a678
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2848_dd6fd5a234c5fcaaa578.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2880_725d1f656ca564e173f7.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2880_725d1f656ca564e173f7.png
new file mode 100644
index 0000000000000000000000000000000000000000..8c8cd0589fae37732332370150215f2b219a8805
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2880_725d1f656ca564e173f7.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_288_6167ef901ebd888d4e8a.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_288_6167ef901ebd888d4e8a.png
new file mode 100644
index 0000000000000000000000000000000000000000..3be449d45e195ef09fb58cecba94a88bc6374230
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_288_6167ef901ebd888d4e8a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2912_d038404edc84a00dbbb9.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2912_d038404edc84a00dbbb9.png
new file mode 100644
index 0000000000000000000000000000000000000000..883fc7a214a4aeb5c55481a86964841e960ff3f4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2912_d038404edc84a00dbbb9.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2944_831ead919542e3580af9.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2944_831ead919542e3580af9.png
new file mode 100644
index 0000000000000000000000000000000000000000..e9a37e3958fc28d9ac094f3ace211b89afe7cb4a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2944_831ead919542e3580af9.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2976_7eb27e78ec2e8983dff6.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2976_7eb27e78ec2e8983dff6.png
new file mode 100644
index 0000000000000000000000000000000000000000..d911266c53a49acc167f3f8064e827172bfb0a5b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_2976_7eb27e78ec2e8983dff6.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3008_eef27204d323fd1ebfa4.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3008_eef27204d323fd1ebfa4.png
new file mode 100644
index 0000000000000000000000000000000000000000..f236071d124deb9c7375a0a50d4c5729ac85d36b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3008_eef27204d323fd1ebfa4.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3040_d2796a4bbf2739176e7b.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3040_d2796a4bbf2739176e7b.png
new file mode 100644
index 0000000000000000000000000000000000000000..a885a6a93331ae292188d10f4952e1264094d9e4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3040_d2796a4bbf2739176e7b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3072_e0e821ccb1c6b4e0dc51.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3072_e0e821ccb1c6b4e0dc51.png
new file mode 100644
index 0000000000000000000000000000000000000000..ee85748fb1eac57802901dd2237e07ae8289f98c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3072_e0e821ccb1c6b4e0dc51.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3104_758d028ae52594ad4fea.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3104_758d028ae52594ad4fea.png
new file mode 100644
index 0000000000000000000000000000000000000000..443d6fdaa6fb6a3d303dbb26bf07e4163e4fdceb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3104_758d028ae52594ad4fea.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3136_2e9b1b3f2ba5348c0855.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3136_2e9b1b3f2ba5348c0855.png
new file mode 100644
index 0000000000000000000000000000000000000000..0150b2d8fd36605f3b8e9c72ab1ff91a56921fde
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3136_2e9b1b3f2ba5348c0855.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3168_d6537430642b7f01fb1d.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3168_d6537430642b7f01fb1d.png
new file mode 100644
index 0000000000000000000000000000000000000000..2a566cefa3ad26f4b0d5b2970c4d41872f9f654e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3168_d6537430642b7f01fb1d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3200_eccb9a382f80dd4cf469.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3200_eccb9a382f80dd4cf469.png
new file mode 100644
index 0000000000000000000000000000000000000000..cc6cec6025256b783587c9c2d87b2dea43b17693
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3200_eccb9a382f80dd4cf469.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_320_f4d53da428a814e3b01a.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_320_f4d53da428a814e3b01a.png
new file mode 100644
index 0000000000000000000000000000000000000000..825920af7f2fb2ebe162bc8a4d05873ff12aa4dc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_320_f4d53da428a814e3b01a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3232_a0fb4cd702536b913a3a.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3232_a0fb4cd702536b913a3a.png
new file mode 100644
index 0000000000000000000000000000000000000000..b523335b28b3211b97b485104448a71302a1a37e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3232_a0fb4cd702536b913a3a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3264_c7678b4b1dd26acdf30c.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3264_c7678b4b1dd26acdf30c.png
new file mode 100644
index 0000000000000000000000000000000000000000..12d5fc2aa56bbba944d9cfae09df5a9ca0780009
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3264_c7678b4b1dd26acdf30c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3296_cfa5a324eaeb41884277.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3296_cfa5a324eaeb41884277.png
new file mode 100644
index 0000000000000000000000000000000000000000..0098d3e40445878b8f9ee92917bf37056399f251
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3296_cfa5a324eaeb41884277.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_32_eeb887bdf96295d917db.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_32_eeb887bdf96295d917db.png
new file mode 100644
index 0000000000000000000000000000000000000000..0d851f10ca1403c1ad1607ec120acd805cce02f6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_32_eeb887bdf96295d917db.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3328_b9224cc969c52f474300.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3328_b9224cc969c52f474300.png
new file mode 100644
index 0000000000000000000000000000000000000000..d3aedacf4f260d34bd106218134e17d4f05229be
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3328_b9224cc969c52f474300.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3360_721f168eedc69482f9cf.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3360_721f168eedc69482f9cf.png
new file mode 100644
index 0000000000000000000000000000000000000000..d692d65c97e35c5459d10c2a8e1aee2d8342ab6f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3360_721f168eedc69482f9cf.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3392_f234cd2ea0a5da2ef36b.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3392_f234cd2ea0a5da2ef36b.png
new file mode 100644
index 0000000000000000000000000000000000000000..a41f8b1d46fdb8344ff8aa005694891514b315c9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3392_f234cd2ea0a5da2ef36b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3424_9940594902364c21e367.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3424_9940594902364c21e367.png
new file mode 100644
index 0000000000000000000000000000000000000000..ded3a8118644476ec94c2396b5fe0ac8fa5e482b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3424_9940594902364c21e367.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3456_9a9887552f7640351b6f.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3456_9a9887552f7640351b6f.png
new file mode 100644
index 0000000000000000000000000000000000000000..177bf208ca2bf2efefbeedf1e27b7f7c75387c3d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3456_9a9887552f7640351b6f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3488_9d9adf821d74d4774ef3.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3488_9d9adf821d74d4774ef3.png
new file mode 100644
index 0000000000000000000000000000000000000000..50a2113b1891c7e4ac84647a98a0a2e46ffe792b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3488_9d9adf821d74d4774ef3.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3520_05198c6669bed3a275ff.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3520_05198c6669bed3a275ff.png
new file mode 100644
index 0000000000000000000000000000000000000000..666bad48353945a47e3f87b846910c217c5a2022
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3520_05198c6669bed3a275ff.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_352_3965e00b78ce06cfed63.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_352_3965e00b78ce06cfed63.png
new file mode 100644
index 0000000000000000000000000000000000000000..7ddd1debc031abffec30df4288f2f810faa2c779
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_352_3965e00b78ce06cfed63.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3552_afadd3f2e8f3b85ce477.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3552_afadd3f2e8f3b85ce477.png
new file mode 100644
index 0000000000000000000000000000000000000000..4322e6afaf085f528363946f15e294c01996cd69
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3552_afadd3f2e8f3b85ce477.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3584_6dd526c79f0a8921900f.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3584_6dd526c79f0a8921900f.png
new file mode 100644
index 0000000000000000000000000000000000000000..d9c39ed6e82c5444b5d401d98067ae1389880670
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3584_6dd526c79f0a8921900f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3616_611d1aef49917e22d609.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3616_611d1aef49917e22d609.png
new file mode 100644
index 0000000000000000000000000000000000000000..942190cf2e938f86facf309222bae75dc45da060
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3616_611d1aef49917e22d609.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3648_1c18098c289b4b7549d7.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3648_1c18098c289b4b7549d7.png
new file mode 100644
index 0000000000000000000000000000000000000000..ccc3d4c0f6ea503ffdea848035b7f9e83333f9f4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3648_1c18098c289b4b7549d7.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3680_3a89f06cef52444692e2.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3680_3a89f06cef52444692e2.png
new file mode 100644
index 0000000000000000000000000000000000000000..e21ace5d7c8e330079f134d50613770066c96715
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3680_3a89f06cef52444692e2.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3712_c8b09e936bbc8392124c.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3712_c8b09e936bbc8392124c.png
new file mode 100644
index 0000000000000000000000000000000000000000..7ed14865778ca5a67e966db5d2ff4def1c71cd5a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3712_c8b09e936bbc8392124c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3744_53ab182bdf1923ff0aec.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3744_53ab182bdf1923ff0aec.png
new file mode 100644
index 0000000000000000000000000000000000000000..0dd9a90eed80d97942f4ab86745ef7b5cc83a2ab
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3744_53ab182bdf1923ff0aec.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3776_369b52fe4585ade222cb.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3776_369b52fe4585ade222cb.png
new file mode 100644
index 0000000000000000000000000000000000000000..79a847e3e9969c74ca1fe9eb182372962838dfe2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3776_369b52fe4585ade222cb.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3808_874c77c04acf364b8f9f.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3808_874c77c04acf364b8f9f.png
new file mode 100644
index 0000000000000000000000000000000000000000..c05cd27a56b6f40dea30b59c5a086ef659d741c6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3808_874c77c04acf364b8f9f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3840_1da9ae62ebca5b3020e7.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3840_1da9ae62ebca5b3020e7.png
new file mode 100644
index 0000000000000000000000000000000000000000..affe723f3ef0b83046f3e37ca5fd8d47e041cd40
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3840_1da9ae62ebca5b3020e7.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_384_40a42b015c1705cfd6d1.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_384_40a42b015c1705cfd6d1.png
new file mode 100644
index 0000000000000000000000000000000000000000..24a67e549e6e387da40591e428f20c4445c3a1f7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_384_40a42b015c1705cfd6d1.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3872_77b41320c09507a0511b.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3872_77b41320c09507a0511b.png
new file mode 100644
index 0000000000000000000000000000000000000000..ea45fadb087fcdc03a24028a28abc1500717a4ca
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3872_77b41320c09507a0511b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3904_4abbe93c1dcd31f48f8c.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3904_4abbe93c1dcd31f48f8c.png
new file mode 100644
index 0000000000000000000000000000000000000000..22077584a121ad8a49879cdb6ccb0bcb3d8c236f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3904_4abbe93c1dcd31f48f8c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3936_9ea36d269fc6c33b2a36.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3936_9ea36d269fc6c33b2a36.png
new file mode 100644
index 0000000000000000000000000000000000000000..f77ecb02708315fef77be7a4887ea7a7f0dc0e81
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3936_9ea36d269fc6c33b2a36.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3968_fd2b2506dc4a84539a5b.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3968_fd2b2506dc4a84539a5b.png
new file mode 100644
index 0000000000000000000000000000000000000000..7c31b954016b0ee60142268122d9cfc0128fcf81
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_3968_fd2b2506dc4a84539a5b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4000_a88a4a08bcc9b4594275.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4000_a88a4a08bcc9b4594275.png
new file mode 100644
index 0000000000000000000000000000000000000000..0d77ca2571091de33b23ba2ab786f02d50bb3414
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4000_a88a4a08bcc9b4594275.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4032_68c4cfd795bfb9ec02f2.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4032_68c4cfd795bfb9ec02f2.png
new file mode 100644
index 0000000000000000000000000000000000000000..5279a2d391d645e7c0534213505e596faa9c06d3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4032_68c4cfd795bfb9ec02f2.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4064_410d0f51ef1dee2a19ed.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4064_410d0f51ef1dee2a19ed.png
new file mode 100644
index 0000000000000000000000000000000000000000..d38f7e57706fb78817acf5d0d230900f9494e651
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4064_410d0f51ef1dee2a19ed.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4096_71178b8d14dd1239cc9e.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4096_71178b8d14dd1239cc9e.png
new file mode 100644
index 0000000000000000000000000000000000000000..3e19ca58d9f4b340a9828dceb79b016641a778b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4096_71178b8d14dd1239cc9e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4128_3370e5eee478179c30df.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4128_3370e5eee478179c30df.png
new file mode 100644
index 0000000000000000000000000000000000000000..500b798f95cd6e7b0ec9583edcfd19bb1fc21783
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4128_3370e5eee478179c30df.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4160_8b8dc427498c0c7dbf00.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4160_8b8dc427498c0c7dbf00.png
new file mode 100644
index 0000000000000000000000000000000000000000..da82f87e145554c53336e5218c43c4875ec21142
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4160_8b8dc427498c0c7dbf00.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_416_4fb7974735a177753e09.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_416_4fb7974735a177753e09.png
new file mode 100644
index 0000000000000000000000000000000000000000..e22ab754689fa21e1caf14c76e37631785afc573
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_416_4fb7974735a177753e09.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4192_01aaaaa4043251fe5b6a.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4192_01aaaaa4043251fe5b6a.png
new file mode 100644
index 0000000000000000000000000000000000000000..2ad0159714b0b8b6dc3e3bc6564d3ef1758f597b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4192_01aaaaa4043251fe5b6a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4224_738cfa30c53ffebc82a0.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4224_738cfa30c53ffebc82a0.png
new file mode 100644
index 0000000000000000000000000000000000000000..813ae9c9aa59a2f304248248db808e9889f2242c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4224_738cfa30c53ffebc82a0.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4256_62b397764b34c5e1bb7e.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4256_62b397764b34c5e1bb7e.png
new file mode 100644
index 0000000000000000000000000000000000000000..827569715b853fae9b352e0cba7123f384b07194
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4256_62b397764b34c5e1bb7e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4288_5aee594d07238a31317b.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4288_5aee594d07238a31317b.png
new file mode 100644
index 0000000000000000000000000000000000000000..18b0c3cab3ef7eb0e8d82780f8f3207458398e41
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4288_5aee594d07238a31317b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4320_96383759be2f628b79f8.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4320_96383759be2f628b79f8.png
new file mode 100644
index 0000000000000000000000000000000000000000..6080f8f4e6a11c2a043e99001b72bfb3ee6267ba
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4320_96383759be2f628b79f8.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4352_5a1bbc73fd9b697d28c5.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4352_5a1bbc73fd9b697d28c5.png
new file mode 100644
index 0000000000000000000000000000000000000000..39a25384f4e4d0278c59e37d80f0c3d25600c9e8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4352_5a1bbc73fd9b697d28c5.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4384_e39007bcd61495c551f7.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4384_e39007bcd61495c551f7.png
new file mode 100644
index 0000000000000000000000000000000000000000..15e6ec82d79ce0c35061754ce9d770d5c90a734f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4384_e39007bcd61495c551f7.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4416_6e2ff36c09b4aec0977d.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4416_6e2ff36c09b4aec0977d.png
new file mode 100644
index 0000000000000000000000000000000000000000..40e03bb51f6f86da9d2e195ef93a3051443bfc90
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4416_6e2ff36c09b4aec0977d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4448_f8c8c82bb4b5186e2cf2.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4448_f8c8c82bb4b5186e2cf2.png
new file mode 100644
index 0000000000000000000000000000000000000000..8cb7abf1210a335cfa76d5091dec7897497127a4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4448_f8c8c82bb4b5186e2cf2.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4480_742442a9742f19f4c667.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4480_742442a9742f19f4c667.png
new file mode 100644
index 0000000000000000000000000000000000000000..f949b32dbe42bb547610f56ec5fdec805c428b60
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4480_742442a9742f19f4c667.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_448_380247d581ccf629eecf.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_448_380247d581ccf629eecf.png
new file mode 100644
index 0000000000000000000000000000000000000000..87138b419612cbd33df5ff16f84e1b7fc86248f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_448_380247d581ccf629eecf.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4512_9a0707c6ae8bca240a53.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4512_9a0707c6ae8bca240a53.png
new file mode 100644
index 0000000000000000000000000000000000000000..a8c5313e05ffef89093098ce56ef9d4f75e53fc2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4512_9a0707c6ae8bca240a53.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4544_1db4bf313088c93fabce.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4544_1db4bf313088c93fabce.png
new file mode 100644
index 0000000000000000000000000000000000000000..5b94f9ccfd8442e818ac859d15822627b38154a7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4544_1db4bf313088c93fabce.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4576_fbe9b4e16d61c6b9f96f.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4576_fbe9b4e16d61c6b9f96f.png
new file mode 100644
index 0000000000000000000000000000000000000000..de20e78c660b8582a07f806c7e1591bc6117e96e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4576_fbe9b4e16d61c6b9f96f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4608_f1b830f7339c81a0e260.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4608_f1b830f7339c81a0e260.png
new file mode 100644
index 0000000000000000000000000000000000000000..2d17c4f2041aa55ab1bda6a207fb386688418a31
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4608_f1b830f7339c81a0e260.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4640_e92f7862263ae9b523a4.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4640_e92f7862263ae9b523a4.png
new file mode 100644
index 0000000000000000000000000000000000000000..eb3fdb87f966121995db33d289ed0a08c893e8d1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4640_e92f7862263ae9b523a4.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4672_7e78f83fc2f1fc65a7b6.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4672_7e78f83fc2f1fc65a7b6.png
new file mode 100644
index 0000000000000000000000000000000000000000..4d3f5b901148c659a2c5f31819ec721790254696
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4672_7e78f83fc2f1fc65a7b6.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4704_60266c7055d05ec9334d.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4704_60266c7055d05ec9334d.png
new file mode 100644
index 0000000000000000000000000000000000000000..f8f51199423e072bc3cca38284e5154d6c104fbd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4704_60266c7055d05ec9334d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4736_66b1341ebe7e3a32404b.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4736_66b1341ebe7e3a32404b.png
new file mode 100644
index 0000000000000000000000000000000000000000..afe4183b24aab233486f83a4f454290659d335d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4736_66b1341ebe7e3a32404b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4768_f38a92dd904cce747e44.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4768_f38a92dd904cce747e44.png
new file mode 100644
index 0000000000000000000000000000000000000000..281edc8bd174d99a9c5ac5d305a10007d2f4dc1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4768_f38a92dd904cce747e44.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4800_f8bae8b6a6b2da744dec.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4800_f8bae8b6a6b2da744dec.png
new file mode 100644
index 0000000000000000000000000000000000000000..d7c736f1a875514b331cd9e83fb41d525781a1de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4800_f8bae8b6a6b2da744dec.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_480_7f95f9bd11ee26010691.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_480_7f95f9bd11ee26010691.png
new file mode 100644
index 0000000000000000000000000000000000000000..d128e3425947a235f88ae9f370d4eeeaa0df7cc9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_480_7f95f9bd11ee26010691.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4832_47c93f1a8a5d4a9e0df0.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4832_47c93f1a8a5d4a9e0df0.png
new file mode 100644
index 0000000000000000000000000000000000000000..6e6e8eb41e5cb2b314c3cb8868c5345edbb6ac86
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4832_47c93f1a8a5d4a9e0df0.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4864_ed13a34d00e096a223ee.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4864_ed13a34d00e096a223ee.png
new file mode 100644
index 0000000000000000000000000000000000000000..dfad60e19a4a702ba937efce97bb66db1b1b6af9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4864_ed13a34d00e096a223ee.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4896_aa9e415851379c7419e9.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4896_aa9e415851379c7419e9.png
new file mode 100644
index 0000000000000000000000000000000000000000..2a817399fe9d565dc0ff7ce18747f480535da358
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4896_aa9e415851379c7419e9.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4928_bea51313ea5916f0c56d.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4928_bea51313ea5916f0c56d.png
new file mode 100644
index 0000000000000000000000000000000000000000..f68ccf35b6ee1c5ed371c23c75915bda48dd98b3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4928_bea51313ea5916f0c56d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4960_6f6d1005522188aa8be4.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4960_6f6d1005522188aa8be4.png
new file mode 100644
index 0000000000000000000000000000000000000000..a015e59b570f7f97e0648a4ebc0b113657bf0aa4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4960_6f6d1005522188aa8be4.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4992_1732c6a4c2cbf7f34509.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4992_1732c6a4c2cbf7f34509.png
new file mode 100644
index 0000000000000000000000000000000000000000..873d190555fc2e9032db5a127902dec97559b205
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_4992_1732c6a4c2cbf7f34509.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5024_252b37b16d87c8bc07b6.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5024_252b37b16d87c8bc07b6.png
new file mode 100644
index 0000000000000000000000000000000000000000..3d15fb0b5440a840648d0567c2461daa5f61e7ae
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5024_252b37b16d87c8bc07b6.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5056_68a7f759e7903713188c.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5056_68a7f759e7903713188c.png
new file mode 100644
index 0000000000000000000000000000000000000000..70d6ab74efe68cf7ef6205cfbfca05a197befb3d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5056_68a7f759e7903713188c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5088_71fcf3a03c2e8a747bd9.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5088_71fcf3a03c2e8a747bd9.png
new file mode 100644
index 0000000000000000000000000000000000000000..a34685861fdd11873f3756eea57bbe524b7fff10
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5088_71fcf3a03c2e8a747bd9.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5120_46a46b6e8b4b542a80e2.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5120_46a46b6e8b4b542a80e2.png
new file mode 100644
index 0000000000000000000000000000000000000000..5d39bc9d3c5a318cd36c51d3b6abe948584353f7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5120_46a46b6e8b4b542a80e2.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_512_da58603783a2fc785f7b.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_512_da58603783a2fc785f7b.png
new file mode 100644
index 0000000000000000000000000000000000000000..6091715b401868e1cb13c11471c686b34456366c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_512_da58603783a2fc785f7b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5152_f42f55c04e97df1669af.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5152_f42f55c04e97df1669af.png
new file mode 100644
index 0000000000000000000000000000000000000000..d70ee0cfadc2956796673a4a5b65b2e19c54ed1e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5152_f42f55c04e97df1669af.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5184_d7d9685d88a072a77fcf.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5184_d7d9685d88a072a77fcf.png
new file mode 100644
index 0000000000000000000000000000000000000000..9392bf02c2d727d318e01e9ab20bff662c6fa331
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5184_d7d9685d88a072a77fcf.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5216_0a93c573e2ba7571d805.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5216_0a93c573e2ba7571d805.png
new file mode 100644
index 0000000000000000000000000000000000000000..5de05daa279baef3b78eb3481baf1974e3604af9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5216_0a93c573e2ba7571d805.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5248_073c3ba620226b5a2ba4.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5248_073c3ba620226b5a2ba4.png
new file mode 100644
index 0000000000000000000000000000000000000000..49fe109890afcff7063500d5a37a12d650299164
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5248_073c3ba620226b5a2ba4.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5280_5f7b8e29c8bc615748e3.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5280_5f7b8e29c8bc615748e3.png
new file mode 100644
index 0000000000000000000000000000000000000000..4a0d8a2d36d4c193600a257c44b949ed62ea174f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5280_5f7b8e29c8bc615748e3.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5312_d0c6f3357969b7c60a02.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5312_d0c6f3357969b7c60a02.png
new file mode 100644
index 0000000000000000000000000000000000000000..724ad2a0018f8e86a0580f92186b79f2d569e503
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5312_d0c6f3357969b7c60a02.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5344_3f39aae10974264f4e1b.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5344_3f39aae10974264f4e1b.png
new file mode 100644
index 0000000000000000000000000000000000000000..345a7bc27b53d1709148dc3d22f2231de3158c50
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5344_3f39aae10974264f4e1b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5376_c2a0f7ce61fb882d86df.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5376_c2a0f7ce61fb882d86df.png
new file mode 100644
index 0000000000000000000000000000000000000000..5388fa0865618d4f14bdcd55428de476be00a411
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5376_c2a0f7ce61fb882d86df.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5408_41d1bc1aa43b889d0d6c.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5408_41d1bc1aa43b889d0d6c.png
new file mode 100644
index 0000000000000000000000000000000000000000..ebe63a82e1dd54b401258017c74dafc686d488ea
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5408_41d1bc1aa43b889d0d6c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5440_55facad5ec21815eae16.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5440_55facad5ec21815eae16.png
new file mode 100644
index 0000000000000000000000000000000000000000..3acff86ef98964634adeb9b171d5ea6eb4b3e3b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5440_55facad5ec21815eae16.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_544_68f55fbee57d6048f700.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_544_68f55fbee57d6048f700.png
new file mode 100644
index 0000000000000000000000000000000000000000..45543cd65682d2fd5499fffa42efc2c196150efd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_544_68f55fbee57d6048f700.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5472_c4bef8e407eb5def44de.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5472_c4bef8e407eb5def44de.png
new file mode 100644
index 0000000000000000000000000000000000000000..a42e87f676f7a6d17cde300cf108ecd75a18ae2c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5472_c4bef8e407eb5def44de.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5504_99d21b01e8f0dcba1050.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5504_99d21b01e8f0dcba1050.png
new file mode 100644
index 0000000000000000000000000000000000000000..07bf291927ab6c68ab90b281e4392c1a5644495c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5504_99d21b01e8f0dcba1050.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5536_208cb43fb55e55cdd1ac.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5536_208cb43fb55e55cdd1ac.png
new file mode 100644
index 0000000000000000000000000000000000000000..0149907dc0f7117c075aae19a87ac312a0a8b4eb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5536_208cb43fb55e55cdd1ac.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5568_065b2041f92fc0ec7371.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5568_065b2041f92fc0ec7371.png
new file mode 100644
index 0000000000000000000000000000000000000000..e6ec329dd658153d227095ce51ed2a97ec4260c1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5568_065b2041f92fc0ec7371.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5600_c730c41a125666824605.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5600_c730c41a125666824605.png
new file mode 100644
index 0000000000000000000000000000000000000000..b7ec72e700cc9ac0a4fde5fea5b9aa5a38e87a20
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5600_c730c41a125666824605.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5632_a5d58b5087755ad6ec3a.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5632_a5d58b5087755ad6ec3a.png
new file mode 100644
index 0000000000000000000000000000000000000000..1e77a319c12b850b4deaaaae7743697cbf11f39b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5632_a5d58b5087755ad6ec3a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5664_3710b3df07f0f485348e.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5664_3710b3df07f0f485348e.png
new file mode 100644
index 0000000000000000000000000000000000000000..f40243148c8b5935a0cc0f7c5eca9a0830935329
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5664_3710b3df07f0f485348e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5696_51d64dc0c8455eac6efe.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5696_51d64dc0c8455eac6efe.png
new file mode 100644
index 0000000000000000000000000000000000000000..c382532b9e4a40a68126d5049a69d4f4f687d2d3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5696_51d64dc0c8455eac6efe.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5728_02c080edb8723d0c58d5.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5728_02c080edb8723d0c58d5.png
new file mode 100644
index 0000000000000000000000000000000000000000..a9e78b261f41d8e68e5eac0fb2fbe1dca3fb513b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5728_02c080edb8723d0c58d5.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5760_589ea85248f69ec27149.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5760_589ea85248f69ec27149.png
new file mode 100644
index 0000000000000000000000000000000000000000..8e75783eba556cb5a5874214608a79727ba8b425
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5760_589ea85248f69ec27149.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_576_e5444ffb50df0e9c2947.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_576_e5444ffb50df0e9c2947.png
new file mode 100644
index 0000000000000000000000000000000000000000..72df929e3874af1068f819cb42b1b2a8f0b199ef
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_576_e5444ffb50df0e9c2947.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5792_ea3723cbe1324e465450.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5792_ea3723cbe1324e465450.png
new file mode 100644
index 0000000000000000000000000000000000000000..4409870734052945d6aa619ed8e5212449d9b1dd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5792_ea3723cbe1324e465450.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5824_1a6fb1d6afbd2d9a9cf2.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5824_1a6fb1d6afbd2d9a9cf2.png
new file mode 100644
index 0000000000000000000000000000000000000000..a357b5c1ab521942aa3e6ac45ac964b40a61bfa6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5824_1a6fb1d6afbd2d9a9cf2.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5856_8660646450d4b3f78528.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5856_8660646450d4b3f78528.png
new file mode 100644
index 0000000000000000000000000000000000000000..86f30e986c7091c4b6c2b8445139e37e79715360
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5856_8660646450d4b3f78528.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5888_b1bc532ad5b1dec059df.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5888_b1bc532ad5b1dec059df.png
new file mode 100644
index 0000000000000000000000000000000000000000..5216c075f2d77a22eaf4c669882e833dc5df972c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5888_b1bc532ad5b1dec059df.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5920_345efc301f74e1577ade.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5920_345efc301f74e1577ade.png
new file mode 100644
index 0000000000000000000000000000000000000000..0d0dff0fe11f2553e8e75d53d223a97495279048
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5920_345efc301f74e1577ade.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5952_71e780d3c479decb600a.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5952_71e780d3c479decb600a.png
new file mode 100644
index 0000000000000000000000000000000000000000..209b4997d625a8ff7a9f0987447d5598097bf4c9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5952_71e780d3c479decb600a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5984_041cc478bc596b672476.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5984_041cc478bc596b672476.png
new file mode 100644
index 0000000000000000000000000000000000000000..e4c6e3d32e3c3d11fb6fe125a675d6e75335113e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_5984_041cc478bc596b672476.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_6016_111172e85de565bc03a6.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_6016_111172e85de565bc03a6.png
new file mode 100644
index 0000000000000000000000000000000000000000..1272089ae92746dfc7e596a2f804172cedf258a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_6016_111172e85de565bc03a6.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_6048_b738cd7518b701c2b1d6.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_6048_b738cd7518b701c2b1d6.png
new file mode 100644
index 0000000000000000000000000000000000000000..2a0a85ec1cdc0811005566cb4f63913f221c1926
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_6048_b738cd7518b701c2b1d6.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_608_afc174787ab4bf92368d.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_608_afc174787ab4bf92368d.png
new file mode 100644
index 0000000000000000000000000000000000000000..8f15c5a050e1f397315c4fe2b1979671c8ad6315
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_608_afc174787ab4bf92368d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_640_e9a061352b98cc3ae3f6.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_640_e9a061352b98cc3ae3f6.png
new file mode 100644
index 0000000000000000000000000000000000000000..1cfffb4d91d63ac4113c20de1488b18a26fe6694
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_640_e9a061352b98cc3ae3f6.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_64_228282a03966941ec047.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_64_228282a03966941ec047.png
new file mode 100644
index 0000000000000000000000000000000000000000..8c9e00db388fa32602d65750c5256e14b16f777f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_64_228282a03966941ec047.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_672_3f1c6aa19c2c5c18c61b.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_672_3f1c6aa19c2c5c18c61b.png
new file mode 100644
index 0000000000000000000000000000000000000000..83b3d206fc4a88003c1c563b0b4c1555d7f1e186
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_672_3f1c6aa19c2c5c18c61b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_704_716ec970f5a0f18cd8da.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_704_716ec970f5a0f18cd8da.png
new file mode 100644
index 0000000000000000000000000000000000000000..4cf582ee1b36a76d37d6d7bbddbd5418c6a1294f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_704_716ec970f5a0f18cd8da.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_736_cfd9ef642f44dd997696.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_736_cfd9ef642f44dd997696.png
new file mode 100644
index 0000000000000000000000000000000000000000..9e413b1a3f610e83e84e4e0da9bcf1fde6c386ca
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_736_cfd9ef642f44dd997696.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_768_02441cce5317dca3d21b.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_768_02441cce5317dca3d21b.png
new file mode 100644
index 0000000000000000000000000000000000000000..b47bc5d491b42fa8569e4c1871f04f5e41a738e4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_768_02441cce5317dca3d21b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_800_81fb629d0d29af782539.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_800_81fb629d0d29af782539.png
new file mode 100644
index 0000000000000000000000000000000000000000..b3b6920cb44d79c62ba33b0d4c66945bc342d729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_800_81fb629d0d29af782539.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_832_dd55e9f0538ca8298a66.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_832_dd55e9f0538ca8298a66.png
new file mode 100644
index 0000000000000000000000000000000000000000..beed16ab41b184af465559f0c60cee639dd36c0f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_832_dd55e9f0538ca8298a66.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_864_bd01e8b12b8943fc0a65.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_864_bd01e8b12b8943fc0a65.png
new file mode 100644
index 0000000000000000000000000000000000000000..e9b4845fff612003760cdf9a52766423ace14812
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_864_bd01e8b12b8943fc0a65.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_896_9d911a3eb7f424fdc2b2.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_896_9d911a3eb7f424fdc2b2.png
new file mode 100644
index 0000000000000000000000000000000000000000..a59ac7e925f2162cf30285d3ed4f943cff1127a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_896_9d911a3eb7f424fdc2b2.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_928_e973146899e07a3daa59.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_928_e973146899e07a3daa59.png
new file mode 100644
index 0000000000000000000000000000000000000000..039fb6370eabb94d2251fc6fb7be083b84bf2cf2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_928_e973146899e07a3daa59.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_960_a5fa8a844f33a2b48452.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_960_a5fa8a844f33a2b48452.png
new file mode 100644
index 0000000000000000000000000000000000000000..28da5e9da7a96264c7496d9090244fbacd2ba8c1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_960_a5fa8a844f33a2b48452.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_96_07a3571c9a995e85fee5.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_96_07a3571c9a995e85fee5.png
new file mode 100644
index 0000000000000000000000000000000000000000..11c806932b9ed6997d65c02c8926b6d68daa310d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_96_07a3571c9a995e85fee5.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_992_ff0083031bb106935a15.png b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_992_ff0083031bb106935a15.png
new file mode 100644
index 0000000000000000000000000000000000000000..5c2838559259930b52111a35b0b6e9b062b98416
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/images/ued/layout/levels16_img_992_ff0083031bb106935a15.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_0_e9b0c2fbc187ce8be111.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_0_e9b0c2fbc187ce8be111.gif
new file mode 100644
index 0000000000000000000000000000000000000000..6cf83cde736afa5745e2e638dda288a694ae9b71
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_0_e9b0c2fbc187ce8be111.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e9b0c2fbc187ce8be111e3ce08c2c15f1c4c746fab84aa88ebae4d6dba3e2608
+size 100979
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_2016_d316c3b96ceead9ecc37.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_2016_d316c3b96ceead9ecc37.gif
new file mode 100644
index 0000000000000000000000000000000000000000..0f642a68be5008a0b54bad6252c17a7dfa0e4072
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_2016_d316c3b96ceead9ecc37.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d316c3b96ceead9ecc3720aa2ad25f208eced1650cee5076f25842ad5b706441
+size 252575
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_3040_474a966bc96cd0bdea63.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_3040_474a966bc96cd0bdea63.gif
new file mode 100644
index 0000000000000000000000000000000000000000..ad6ff14b7c816b955fd2b06da0b758fc11fed56a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_3040_474a966bc96cd0bdea63.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:474a966bc96cd0bdea63ed71dac2760fe9805621a7526e95dfc0d675593235fc
+size 264511
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_4064_f561be9801993e334890.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_4064_f561be9801993e334890.gif
new file mode 100644
index 0000000000000000000000000000000000000000..5ab7c48dc62b0099033e32f995697380e22041d1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_4064_f561be9801993e334890.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f561be9801993e334890051fa5f501b184bc4f49da0a01da641b8466905d62ef
+size 274495
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_5088_39b0cce15a654761e9eb.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_5088_39b0cce15a654761e9eb.gif
new file mode 100644
index 0000000000000000000000000000000000000000..c1fe057c93bd874cdf0d41dc9bb25e6743cb97bf
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_5088_39b0cce15a654761e9eb.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:39b0cce15a654761e9ebd4c5e8214fdda66e42089c9870a18cfd461d697cdda3
+size 279328
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_992_b9240eb49ff83fe19258.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_992_b9240eb49ff83fe19258.gif
new file mode 100644
index 0000000000000000000000000000000000000000..b263376b0104324f529b29c9cf5976619cb2adc2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-orig/rollouts_gif_992_b9240eb49ff83fe19258.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b9240eb49ff83fe1925840724f693a2a6488b5778698b7a8f6baf9dfc096112e
+size 195595
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_0_0ef543c2e6a82ce9a719.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_0_0ef543c2e6a82ce9a719.gif
new file mode 100644
index 0000000000000000000000000000000000000000..65e290214849181d29b1316b4426a6568b08d8fa
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_0_0ef543c2e6a82ce9a719.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0ef543c2e6a82ce9a7197029ec6bb0a7ee887d01e79a7de191d6410f19eefacb
+size 107996
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_2016_ad6e3184ad5a5974f18c.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_2016_ad6e3184ad5a5974f18c.gif
new file mode 100644
index 0000000000000000000000000000000000000000..a807e245609bb3829dd467dc423acdb03af2e6b9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_2016_ad6e3184ad5a5974f18c.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ad6e3184ad5a5974f18c83eeb588fb86015d404a4dca318623cb5fe40fc6b55d
+size 281535
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_3040_a44991b76ff59434c6b9.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_3040_a44991b76ff59434c6b9.gif
new file mode 100644
index 0000000000000000000000000000000000000000..5f7719315a18e4ca205b7938d98e50956a5cfcae
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_3040_a44991b76ff59434c6b9.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a44991b76ff59434c6b9e5ef26aca28e6cea9afe35b48ca1b6de0d5273d1a3ec
+size 297432
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_4064_4416679b4f405fb2718f.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_4064_4416679b4f405fb2718f.gif
new file mode 100644
index 0000000000000000000000000000000000000000..624c886bfb8f0b5c743767af411591b7033bd8a4
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_4064_4416679b4f405fb2718f.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4416679b4f405fb2718f23e6089c4ab728d0755a86c7627477a5d4463f6bb4f6
+size 313886
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_5088_b1ee6f21586850948d88.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_5088_b1ee6f21586850948d88.gif
new file mode 100644
index 0000000000000000000000000000000000000000..c6c50ec8bdc4bbf9fa7245a4df6c3dd4da4ada5f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_5088_b1ee6f21586850948d88.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b1ee6f21586850948d88f8c15763aee34d68d0f95f23f6b82c3e931ea31f4d60
+size 316761
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_992_665485c6d61f2dd34382.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_992_665485c6d61f2dd34382.gif
new file mode 100644
index 0000000000000000000000000000000000000000..4319eaa0be641d3ea7e44c4dd0e734dfb3f32549
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-shift/rollouts_gif_992_665485c6d61f2dd34382.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:665485c6d61f2dd34382f364bed1b51c930c356b9fc7612ae85d08d77b1b34e5
+size 225648
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_0_4bf97f21068441f21f0d.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_0_4bf97f21068441f21f0d.gif
new file mode 100644
index 0000000000000000000000000000000000000000..d4d71cb5780bf8b8d11ed38ffb2f1d382a7bb846
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_0_4bf97f21068441f21f0d.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4bf97f21068441f21f0d5cf41865b2c7872d8b1360fe16e2c3776b8a9b44571d
+size 107163
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_2016_c3d54e7eb89803deb72a.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_2016_c3d54e7eb89803deb72a.gif
new file mode 100644
index 0000000000000000000000000000000000000000..b1673a1cfecd908cef9ff56875f4614ed465515c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_2016_c3d54e7eb89803deb72a.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c3d54e7eb89803deb72ac2535690604a301c0c875fd45f72282c886b8401cc1b
+size 296062
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_3040_7ca026a606a168bd4191.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_3040_7ca026a606a168bd4191.gif
new file mode 100644
index 0000000000000000000000000000000000000000..89170509d0af1872692626edf779d4cfd97ececb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_3040_7ca026a606a168bd4191.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7ca026a606a168bd419182754cfa737138d228d5929ac94126ed9874f88c197d
+size 314094
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_4064_0af9f52337a02c1fdaa9.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_4064_0af9f52337a02c1fdaa9.gif
new file mode 100644
index 0000000000000000000000000000000000000000..1fe768ddf8975c9affdc89ef0a54ef878fc15c75
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_4064_0af9f52337a02c1fdaa9.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0af9f52337a02c1fdaa970fed400f8bde9ffa3443208a91eb4e970bd1b1fca15
+size 326764
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_5088_e51653d7a53328cbe862.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_5088_e51653d7a53328cbe862.gif
new file mode 100644
index 0000000000000000000000000000000000000000..354376e1bccaecda74f21cef0135b3dd0e240c83
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_5088_e51653d7a53328cbe862.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e51653d7a53328cbe862e09665f91f1c2e59b9fa6a310eb33e6b9ef58d4b6cb9
+size 322827
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_992_38c7f3069078799c586d.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_992_38c7f3069078799c586d.gif
new file mode 100644
index 0000000000000000000000000000000000000000..bca579f4155d3ec29ff6b187133c40415c3f550b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-train/rollouts_gif_992_38c7f3069078799c586d.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:38c7f3069078799c586dcec7d2a16e8372a3523bb592c1b2f29370f9a7abd83d
+size 224486
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_0_0145cd3ed1d03ba4c52e.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_0_0145cd3ed1d03ba4c52e.gif
new file mode 100644
index 0000000000000000000000000000000000000000..494a882247d83791f67850d1696056aa90bf199b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_0_0145cd3ed1d03ba4c52e.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0145cd3ed1d03ba4c52e3fa2b4d5b76093ab64768911e7f474e67e6ba363cf3b
+size 106819
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_2016_bb9f77472c43ed2b5ce4.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_2016_bb9f77472c43ed2b5ce4.gif
new file mode 100644
index 0000000000000000000000000000000000000000..a6929d8f15df458b7b8d59fdc85f00bae8fe03e4
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_2016_bb9f77472c43ed2b5ce4.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bb9f77472c43ed2b5ce4dcdd1c6c6d91a029949d212ee3d0aa5d1dc9371c2cd1
+size 252722
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_3040_c771511fda654606efa3.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_3040_c771511fda654606efa3.gif
new file mode 100644
index 0000000000000000000000000000000000000000..ba0464b8f2de9cea0f53fa3a017ac90554e5f045
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_3040_c771511fda654606efa3.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c771511fda654606efa3c5253945eff8ed2a489374666df377f371611745aaf5
+size 271081
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_4064_97bc6a39ac75aa3a1e40.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_4064_97bc6a39ac75aa3a1e40.gif
new file mode 100644
index 0000000000000000000000000000000000000000..68e3235b5dc9a1e5c3972e147cd570a190e683dc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_4064_97bc6a39ac75aa3a1e40.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:97bc6a39ac75aa3a1e400ee63f4ff2866b228e9d0864970dea991a309378be0d
+size 283109
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_5088_3b8df0ac5041fd8b5630.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_5088_3b8df0ac5041fd8b5630.gif
new file mode 100644
index 0000000000000000000000000000000000000000..8c608e8bc0e0ae28fdffb813dc4e8e48d20c5f89
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_5088_3b8df0ac5041fd8b5630.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3b8df0ac5041fd8b56302c1b9a7abfcd35821ce50a915300c332ab2ba4b9bac5
+size 280607
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_992_150795aab7f0ae84791d.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_992_150795aab7f0ae84791d.gif
new file mode 100644
index 0000000000000000000000000000000000000000..b022319d5f10130545e244ed66ab2a12153efe78
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/eval-batch-tree/rollouts_gif_992_150795aab7f0ae84791d.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:150795aab7f0ae84791d4c6a5f78988da0c5ec74479afeaaa95fc0440e367dcd
+size 190731
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-all/rollouts_gif_0_a6cb2e3f92072df15fbd.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-all/rollouts_gif_0_a6cb2e3f92072df15fbd.gif
new file mode 100644
index 0000000000000000000000000000000000000000..f535afd2e95ce339e3bf5bbb30bbca874001e3b1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-all/rollouts_gif_0_a6cb2e3f92072df15fbd.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a6cb2e3f92072df15fbdf705375e3850bd32b06e56d5442035c2457d88d0d119
+size 110576
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-all/rollouts_gif_2016_a1610c3bb8d0ce3d345e.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-all/rollouts_gif_2016_a1610c3bb8d0ce3d345e.gif
new file mode 100644
index 0000000000000000000000000000000000000000..1151a993a24bf3d988e1c6685185fa447da98783
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-all/rollouts_gif_2016_a1610c3bb8d0ce3d345e.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a1610c3bb8d0ce3d345e417324f7da9ffcae9596660f246bf038e97bd380b67f
+size 143888
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-all/rollouts_gif_3040_193304d5f265da614c6b.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-all/rollouts_gif_3040_193304d5f265da614c6b.gif
new file mode 100644
index 0000000000000000000000000000000000000000..0b19823224517a3d0d723ac835bb3b5275e060fd
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-all/rollouts_gif_3040_193304d5f265da614c6b.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:193304d5f265da614c6b6050a248f76d7f035e6b04b28170b4963e1ea55def4f
+size 303427
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-all/rollouts_gif_4064_3da39e8f9a0fc99e101f.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-all/rollouts_gif_4064_3da39e8f9a0fc99e101f.gif
new file mode 100644
index 0000000000000000000000000000000000000000..8d47c0f4d2317a01348bdaf16ac88541c0006e85
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-all/rollouts_gif_4064_3da39e8f9a0fc99e101f.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3da39e8f9a0fc99e101f630355e33c67a3d950d5608b6ae9e517d8b152522da8
+size 314790
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-all/rollouts_gif_5088_5ac4920811574afecb73.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-all/rollouts_gif_5088_5ac4920811574afecb73.gif
new file mode 100644
index 0000000000000000000000000000000000000000..bc460659c5e0402015fef2669f27b684826539a1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-all/rollouts_gif_5088_5ac4920811574afecb73.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5ac4920811574afecb73e516147bf72e276d44b3e4185d71a184ec1d3f2efe27
+size 155025
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-all/rollouts_gif_992_16f1011b05a687326ee7.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-all/rollouts_gif_992_16f1011b05a687326ee7.gif
new file mode 100644
index 0000000000000000000000000000000000000000..03b0acb1a72936cf63310924947facf723c147c4
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-all/rollouts_gif_992_16f1011b05a687326ee7.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:16f1011b05a687326ee75ec5651fac645c199bff124c2fc0288f612b33d5133c
+size 213713
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-generate/rollouts_gif_3040_193304d5f265da614c6b.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-generate/rollouts_gif_3040_193304d5f265da614c6b.gif
new file mode 100644
index 0000000000000000000000000000000000000000..0b19823224517a3d0d723ac835bb3b5275e060fd
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-generate/rollouts_gif_3040_193304d5f265da614c6b.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:193304d5f265da614c6b6050a248f76d7f035e6b04b28170b4963e1ea55def4f
+size 303427
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-generate/rollouts_gif_4064_3da39e8f9a0fc99e101f.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-generate/rollouts_gif_4064_3da39e8f9a0fc99e101f.gif
new file mode 100644
index 0000000000000000000000000000000000000000..8d47c0f4d2317a01348bdaf16ac88541c0006e85
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-generate/rollouts_gif_4064_3da39e8f9a0fc99e101f.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3da39e8f9a0fc99e101f630355e33c67a3d950d5608b6ae9e517d8b152522da8
+size 314790
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-generate/rollouts_gif_992_16f1011b05a687326ee7.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-generate/rollouts_gif_992_16f1011b05a687326ee7.gif
new file mode 100644
index 0000000000000000000000000000000000000000..03b0acb1a72936cf63310924947facf723c147c4
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-generate/rollouts_gif_992_16f1011b05a687326ee7.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:16f1011b05a687326ee75ec5651fac645c199bff124c2fc0288f612b33d5133c
+size 213713
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-replay/rollouts_gif_0_a6cb2e3f92072df15fbd.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-replay/rollouts_gif_0_a6cb2e3f92072df15fbd.gif
new file mode 100644
index 0000000000000000000000000000000000000000..f535afd2e95ce339e3bf5bbb30bbca874001e3b1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-replay/rollouts_gif_0_a6cb2e3f92072df15fbd.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a6cb2e3f92072df15fbdf705375e3850bd32b06e56d5442035c2457d88d0d119
+size 110576
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-replay/rollouts_gif_2016_a1610c3bb8d0ce3d345e.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-replay/rollouts_gif_2016_a1610c3bb8d0ce3d345e.gif
new file mode 100644
index 0000000000000000000000000000000000000000..1151a993a24bf3d988e1c6685185fa447da98783
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-replay/rollouts_gif_2016_a1610c3bb8d0ce3d345e.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a1610c3bb8d0ce3d345e417324f7da9ffcae9596660f246bf038e97bd380b67f
+size 143888
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-replay/rollouts_gif_5088_5ac4920811574afecb73.gif b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-replay/rollouts_gif_5088_5ac4920811574afecb73.gif
new file mode 100644
index 0000000000000000000000000000000000000000..bc460659c5e0402015fef2669f27b684826539a1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/media/videos/train-replay/rollouts_gif_5088_5ac4920811574afecb73.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5ac4920811574afecb73e516147bf72e276d44b3e4185d71a184ec1d3f2efe27
+size 155025
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/output.log b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/output.log
new file mode 100644
index 0000000000000000000000000000000000000000..b1ee36f8aebcd4df4ca96c284c19946728eba0e8
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/output.log
@@ -0,0 +1,25061 @@
+  env_size                      : 13
+  env_layout                    : tree
+  wall_prob                     : 1
+  env_corner_size               : 1
+  env_terminate_after_corner    : False
+  obs_level_of_detail           : 0
+  img_level_of_detail           : 1
+  env_penalize_time             : False
+  net_cnn_type                  : large
+  net_rnn_type                  : ff
+  net_width                     : 256
+  ued                           : plr
+  prob_shift                    : 1
+  num_train_levels              : 2048
+  plr_buffer_size               : 4096
+  plr_temperature               : 0.1
+  plr_staleness_coeff           : 0.1
+  plr_prob_replay               : 0.5
+  plr_regret_estimator          : maxmc-actor
+  plr_robust                    : False
+  num_mutate_steps              : 12
+  prob_mutate_shift             : 0
+  chain_mutate                  : True
+  mutate_cheese                 : True
+  train_proxy_critic            : False
+  plr_proxy_shaping             : False
+  proxy_name                    : proxy_corner
+  plr_proxy_shaping_coeff       : 0.5
+  clipping                      : False
+  ppo_lr                        : 5e-05
+  ppo_gamma                     : 0.999
+  ppo_clip_eps                  : 0.1
+  ppo_gae_lambda                : 0.95
+  ppo_entropy_coeff             : 0.001
+  ppo_critic_coeff              : 0.5
+  ppo_proxy_critic_coeff        : 0.5
+  ppo_max_grad_norm             : 0.5
+  ppo_lr_annealing              : False
+  num_minibatches_per_epoch     : 4
+  num_epochs_per_cycle          : 5
+  num_total_env_steps           : 200000000
+  num_env_steps_per_cycle       : 128
+  num_parallel_envs             : 256
+  console_log                   : True
+  wandb_log                     : True
+  wandb_project                 : jaxgmg-wall-prob-tree
+  wandb_entity                  : None
+  wandb_group                   : None
+  wandb_name                    : wp:1.0_alpha:1
+  log_gifs                      : True
+  log_imgs                      : True
+  log_hists                     : False
+  num_cycles_per_log            : 32
+  num_cycles_per_eval           : 32
+  num_cycles_per_gifs           : 1024
+  num_cycles_per_big_eval       : 1024
+  use_fixed_eval_levels         : False
+  evals_num_env_steps           : 512
+  evals_num_levels              : 256
+  gif_grid_width                : 16
+  level_splayer                 : mouse
+  checkpointing                 : True
+  keep_all_checkpoints          : True
+  max_num_checkpoints           : 1
+  num_cycles_per_checkpoint     : 64
+  seed                          : 42
+configuring environment...
+configuring level generators...
+  mixing level generators with {prob_shift=}...
+configuring level classifier...
+configuring level mutator...
+configuring level solver...
+configuring level metrics...
+configuring eval level generators...
+configuring parser and parsing fixed eval levels...
+(unsupported size for fixed evals)
+configuring heatmap splayer...
+seeding random number generator with seed=42...
+configuring curriculum with ued='plr'...
+configuring eval batches from 4 level generators...
+  generating 256 'train' levels...
+  also solving generated levels...
+  generating 256 'orig' levels...
+  also solving generated levels...
+  generating 256 'shift' levels...
+  also solving generated levels...
+  generating 256 'tree' levels...
+  also solving generated levels...
+configuring evals for 0 fixed eval levels...
+configuring actor critic network...
+  net_cnn_type='large'
+  net_rnn_type='ff'
+  net_width=256
+  initialising network...
+  number of parameters: 198982
+initialising the checkpointer...
+setting up optimiser...
+initialising train state...
+configuring PPO updater...
+begin training loop!
+(note: first two cycles are slow due to compilation)
+                                   , ? env s
+===========================================================
+  step:
+    env-step-all-after:                         32768
+    env-step-replay-after:                      32768
+    env-step-used-after:                        32768
+    ppo-update-after:                           20
+  train-replay:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     79.82127
+    avg_avg_return:                             0.1274096
+    avg_reward_per_step:                        0.005004883
+    proxy_corner:
+      avg_avg_return:                           0.11038739
+      avg_reward_per_step:                      0.0010681152
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     79.82127
+    avg_avg_return:                             0.1274096
+    avg_reward_per_step:                        0.005004883
+    proxy_corner:
+      avg_avg_return:                           0.11038739
+      avg_reward_per_step:                      0.0010681152
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  perf:
+    env_steps_per_second:                       2362.7624467970504
+    ppo_updates_per_second:                     0.4169271376504233
+    eval-batch-train-duration:                  10.99809556081891
+    eval-batch-orig-duration:                   0.30281157046556473
+    eval-batch-shift-duration:                  0.30207367427647114
+    eval-batch-tree-duration:                   0.2989609092473984
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    15.731934
+      cheese-corne_dist_wavg:                   21.375593
+      levels_cheese_is_in_corner_avg:           0.013427734
+      mouse-cheese_dist_finite_avg:             13.281982
+      mouse-cheese_dist_finite_wavg:            15.725531
+      mouse-cheese_dist_solvable_avg:           13.281982
+      mouse-cheese_dist_solvable_wavg:          15.725525
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000004
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016795
+    scoring:
+      avg_scores:                               0.004490095
+    visit_patterns:
+      avg_first_visit_time:                     0.0
+      avg_last_visit_time:                      0.0625
+      num_replay_batches:                       1
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0016357744
+    avg_actor_approxkl3:                        0.0016460579
+    avg_actor_clipfrac:                         0.034228515
+    avg_actor_loss:                             -0.0008026654
+    avg_advantage:                              0.06949739
+    avg_critic_clipfrac:                        0.15007325
+    avg_critic_loss:                            0.018653473
+    avg_entropy:                                1.3525817
+    avg_grad_norm_pre_clip:                     0.29644352
+    avg_loss:                                   0.00717149
+    max:
+      max_actor_approxkl1:                      0.0035919724
+      max_actor_approxkl3:                      0.002679848
+      max_actor_clipfrac:                       0.15454102
+      max_actor_loss:                           0.00043228082
+      max_advantage:                            1.2152556
+      max_critic_clipfrac:                      1.0
+      max_critic_loss:                          0.026306987
+      max_entropy:                              1.3570426
+      max_grad_norm_pre_clip:                   0.6479004
+      max_loss:                                 0.01097838
+    std:
+      std_actor_approxkl1:                      0.00086574035
+      std_actor_approxkl3:                      0.00069093256
+      std_actor_clipfrac:                       0.04673939
+      std_actor_loss:                           0.00070102705
+      std_advantage:                            0.19093259
+      std_critic_clipfrac:                      0.29939657
+      std_critic_loss:                          0.004409034
+      std_entropy:                              0.0035272634
+      std_grad_norm_pre_clip:                   0.1646497
+      std_loss:                                 0.0022551497
+  eval-batch-train:
+    avg_avg_episode_length:                     79.344666
+    avg_avg_return:                             0.14186653
+    avg_benchmark_regret:                       0.8453505
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.0037612915
+    proxy_corner:
+      avg_avg_return:                           0.12585351
+      avg_reward_per_step:                      0.0011444092
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-orig:
+    avg_avg_episode_length:                     80.33637
+    avg_avg_return:                             0.12861764
+    avg_benchmark_regret:                       0.85589385
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.002571106
+    proxy_corner:
+      avg_avg_return:                           0.12861764
+      avg_reward_per_step:                      0.002571106
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-shift:
+    avg_avg_episode_length:                     77.91929
+    avg_avg_return:                             0.160769
+    avg_benchmark_regret:                       0.8267369
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.0045700073
+    proxy_corner:
+      avg_avg_return:                           0.12523928
+      avg_reward_per_step:                      0.0011062622
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.593506
+    avg_avg_return:                             0.13258415
+    avg_benchmark_regret:                       0.852051
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.00415802
+    proxy_corner:
+      avg_avg_return:                           0.13258415
+      avg_reward_per_step:                      0.00415802
+    rollouts_gif:                               float32(132, 225, 225, 3)
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        1048576
+    env-step-replay-before:                     557056
+    env-step-used-before:                       1048576
+    ppo-update-before:                          640
+    env-step-generate-before:                   491520
+    env-step-all-after:                         1081344
+    env-step-replay-after:                      589824
+    env-step-used-after:                        1081344
+    ppo-update-after:                           660
+    env-step-generate-after:                    491520
+  train-replay:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     78.1277
+    avg_avg_return:                             0.14464211
+    avg_reward_per_step:                        0.0078125
+    proxy_corner:
+      avg_avg_return:                           0.12483439
+      avg_reward_per_step:                      0.0014038086
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     78.1277
+    avg_avg_return:                             0.14464211
+    avg_reward_per_step:                        0.0078125
+    proxy_corner:
+      avg_avg_return:                           0.12483439
+      avg_reward_per_step:                      0.0014038086
+  perf:
+    env_steps_per_second:                       299509.3295763629
+    ppo_updates_per_second:                     41.906504567000766
+    eval-batch-train-duration:                  0.512879952788353
+    eval-batch-orig-duration:                   0.4779896214604378
+    eval-batch-shift-duration:                  0.43680860474705696
+    eval-batch-tree-duration:                   0.26521648559719324
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    15.592529
+      cheese-corne_dist_wavg:                   19.560984
+      levels_cheese_is_in_corner_avg:           0.012451172
+      mouse-cheese_dist_finite_avg:             11.798828
+      mouse-cheese_dist_finite_wavg:            2.1242394
+      mouse-cheese_dist_solvable_avg:           11.798828
+      mouse-cheese_dist_solvable_wavg:          2.1242397
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.12339502
+    visit_patterns:
+      avg_first_visit_time:                     1.3813477
+      avg_last_visit_time:                      8.894775
+      num_replay_batches:                       18
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014731871
+    avg_actor_approxkl3:                        0.0015688973
+    avg_actor_clipfrac:                         0.06265869
+    avg_actor_loss:                             -0.0011669993
+    avg_advantage:                              0.018671483
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.014573957
+    avg_entropy:                                1.3085216
+    avg_grad_norm_pre_clip:                     0.28041503
+    avg_loss:                                   0.0048114574
+    max:
+      max_actor_approxkl1:                      0.0032157237
+      max_actor_approxkl3:                      0.0036778627
+      max_actor_clipfrac:                       0.43640137
+      max_actor_loss:                           0.00016263127
+      max_advantage:                            0.84857607
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.019750541
+      max_entropy:                              1.325227
+      max_grad_norm_pre_clip:                   0.9278859
+      max_loss:                                 0.008096268
+    std:
+      std_actor_approxkl1:                      0.0009167504
+      std_actor_approxkl3:                      0.0009923995
+      std_actor_clipfrac:                       0.11053929
+      std_actor_loss:                           0.0010035982
+      std_advantage:                            0.17019269
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0027949717
+      std_entropy:                              0.00883499
+      std_grad_norm_pre_clip:                   0.16607948
+      std_loss:                                 0.0016423388
+  eval-batch-train:
+    avg_avg_episode_length:                     79.502365
+    avg_avg_return:                             0.1273651
+    avg_benchmark_regret:                       0.85985196
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.004875183
+    proxy_corner:
+      avg_avg_return:                           0.11016601
+      avg_reward_per_step:                      0.0010757446
+  eval-batch-orig:
+    avg_avg_episode_length:                     80.70572
+    avg_avg_return:                             0.10475455
+    avg_benchmark_regret:                       0.87975705
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.0023422241
+    proxy_corner:
+      avg_avg_return:                           0.10475455
+      avg_reward_per_step:                      0.0023422241
+  eval-batch-shift:
+    avg_avg_episode_length:                     78.26888
+    avg_avg_return:                             0.14712347
+    avg_benchmark_regret:                       0.84038246
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.0055618286
+    proxy_corner:
+      avg_avg_return:                           0.124336705
+      avg_reward_per_step:                      0.0010910034
+  eval-batch-tree:
+    avg_avg_episode_length:                     80.52205
+    avg_avg_return:                             0.10267798
+    avg_benchmark_regret:                       0.88195723
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0053253174
+    proxy_corner:
+      avg_avg_return:                           0.10267798
+      avg_reward_per_step:                      0.0053253174
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        2097152
+    env-step-replay-before:                     1114112
+    env-step-used-before:                       2097152
+    ppo-update-before:                          1280
+    env-step-generate-before:                   983040
+    env-step-all-after:                         2129920
+    env-step-replay-after:                      1114112
+    env-step-used-after:                        2129920
+    ppo-update-after:                           1300
+    env-step-generate-after:                    1015808
+  train-generate:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     79.96676
+    avg_avg_return:                             0.10216107
+    avg_reward_per_step:                        0.0055236816
+    proxy_corner:
+      avg_avg_return:                           0.12738103
+      avg_reward_per_step:                      0.0015258789
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     79.96676
+    avg_avg_return:                             0.10216107
+    avg_reward_per_step:                        0.0055236816
+    proxy_corner:
+      avg_avg_return:                           0.12738103
+      avg_reward_per_step:                      0.0015258789
+  perf:
+    env_steps_per_second:                       352340.9514858759
+    ppo_updates_per_second:                     41.48204621251808
+    eval-batch-train-duration:                  0.5364400185644627
+    eval-batch-orig-duration:                   0.4848135858774185
+    eval-batch-shift-duration:                  0.4877433544024825
+    eval-batch-tree-duration:                   0.4970854287967086
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    15.403564
+      cheese-corne_dist_wavg:                   22.252554
+      levels_cheese_is_in_corner_avg:           0.012451172
+      mouse-cheese_dist_finite_avg:             10.544434
+      mouse-cheese_dist_finite_wavg:            1.9915881
+      mouse-cheese_dist_solvable_avg:           10.544434
+      mouse-cheese_dist_solvable_wavg:          1.9915881
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.99999
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.22637829
+    visit_patterns:
+      avg_first_visit_time:                     4.4973145
+      avg_last_visit_time:                      24.13086
+      num_replay_batches:                       34
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012597507
+    avg_actor_approxkl3:                        0.0012419483
+    avg_actor_clipfrac:                         0.043884277
+    avg_actor_loss:                             -0.0009218302
+    avg_advantage:                              -0.019047236
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.01059804
+    avg_entropy:                                1.242349
+    avg_grad_norm_pre_clip:                     0.3387901
+    avg_loss:                                   0.003134841
+    max:
+      max_actor_approxkl1:                      0.0025896325
+      max_actor_approxkl3:                      0.0023927111
+      max_actor_clipfrac:                       0.23083496
+      max_actor_loss:                           0.00026787817
+      max_advantage:                            0.8635814
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.017030887
+      max_entropy:                              1.2560437
+      max_grad_norm_pre_clip:                   0.71264064
+      max_loss:                                 0.0071350625
+    std:
+      std_actor_approxkl1:                      0.0010037785
+      std_actor_approxkl3:                      0.0007613749
+      std_actor_clipfrac:                       0.064023584
+      std_actor_loss:                           0.00074004696
+      std_advantage:                            0.14315832
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0027685643
+      std_entropy:                              0.0067458004
+      std_grad_norm_pre_clip:                   0.1541783
+      std_loss:                                 0.0015015033
+  eval-batch-train:
+    avg_avg_episode_length:                     78.97693
+    avg_avg_return:                             0.122258075
+    avg_benchmark_regret:                       0.864959
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.007095337
+    proxy_corner:
+      avg_avg_return:                           0.14107056
+      avg_reward_per_step:                      0.0013275146
+  eval-batch-orig:
+    avg_avg_episode_length:                     77.76447
+    avg_avg_return:                             0.16628131
+    avg_benchmark_regret:                       0.8182303
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.005706787
+    proxy_corner:
+      avg_avg_return:                           0.16628131
+      avg_reward_per_step:                      0.005706787
+  eval-batch-shift:
+    avg_avg_episode_length:                     79.20054
+    avg_avg_return:                             0.12349358
+    avg_benchmark_regret:                       0.86401236
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.0051574707
+    proxy_corner:
+      avg_avg_return:                           0.16577771
+      avg_reward_per_step:                      0.0013961792
+  eval-batch-tree:
+    avg_avg_episode_length:                     77.923775
+    avg_avg_return:                             0.15081283
+    avg_benchmark_regret:                       0.83382237
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.00894928
+    proxy_corner:
+      avg_avg_return:                           0.15081283
+      avg_reward_per_step:                      0.00894928
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        3145728
+    env-step-replay-before:                     1605632
+    env-step-used-before:                       3145728
+    ppo-update-before:                          1920
+    env-step-generate-before:                   1540096
+    env-step-all-after:                         3178496
+    env-step-replay-after:                      1638400
+    env-step-used-after:                        3178496
+    ppo-update-after:                           1940
+    env-step-generate-after:                    1540096
+  train-replay:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     70.39663
+    avg_avg_return:                             0.27873385
+    avg_reward_per_step:                        0.015808105
+    proxy_corner:
+      avg_avg_return:                           0.11785784
+      avg_reward_per_step:                      0.0014038086
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     70.39663
+    avg_avg_return:                             0.27873385
+    avg_reward_per_step:                        0.015808105
+    proxy_corner:
+      avg_avg_return:                           0.11785784
+      avg_reward_per_step:                      0.0014038086
+  perf:
+    env_steps_per_second:                       261030.674243262
+    ppo_updates_per_second:                     41.35204290256187
+    eval-batch-train-duration:                  0.5324579747393727
+    eval-batch-orig-duration:                   0.4805276123806834
+    eval-batch-shift-duration:                  0.46127405762672424
+    eval-batch-tree-duration:                   0.4858691319823265
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    15.07251
+      cheese-corne_dist_wavg:                   18.617964
+      levels_cheese_is_in_corner_avg:           0.012939453
+      mouse-cheese_dist_finite_avg:             9.2146
+      mouse-cheese_dist_finite_wavg:            1.8756282
+      mouse-cheese_dist_solvable_avg:           9.2146
+      mouse-cheese_dist_solvable_wavg:          1.8756282
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.420168
+    scoring:
+      avg_scores:                               0.3316474
+    visit_patterns:
+      avg_first_visit_time:                     10.715576
+      avg_last_visit_time:                      40.31787
+      num_replay_batches:                       50
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0016960584
+    avg_actor_approxkl3:                        0.0013708681
+    avg_actor_clipfrac:                         0.062805176
+    avg_actor_loss:                             -0.0006875586
+    avg_advantage:                              0.029619329
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.017711416
+    avg_entropy:                                1.3197072
+    avg_grad_norm_pre_clip:                     0.98906803
+    avg_loss:                                   0.0068484433
+    max:
+      max_actor_approxkl1:                      0.005264935
+      max_actor_approxkl3:                      0.0045759142
+      max_actor_clipfrac:                       0.3326416
+      max_actor_loss:                           0.0014799088
+      max_advantage:                            0.7458211
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.022963399
+      max_entropy:                              1.3303689
+      max_grad_norm_pre_clip:                   3.834786
+      max_loss:                                 0.009526048
+    std:
+      std_actor_approxkl1:                      0.0015753983
+      std_actor_approxkl3:                      0.0012538142
+      std_actor_clipfrac:                       0.091732986
+      std_actor_loss:                           0.0010341461
+      std_advantage:                            0.18836941
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0022498325
+      std_entropy:                              0.0057673906
+      std_grad_norm_pre_clip:                   0.8805798
+      std_loss:                                 0.0016259997
+  eval-batch-train:
+    avg_avg_episode_length:                     77.00424
+    avg_avg_return:                             0.1930967
+    avg_benchmark_regret:                       0.7941203
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.0054397583
+    proxy_corner:
+      avg_avg_return:                           0.12527366
+      avg_reward_per_step:                      0.0012207031
+  eval-batch-orig:
+    avg_avg_episode_length:                     75.83406
+    avg_avg_return:                             0.23591356
+    avg_benchmark_regret:                       0.748598
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.0055007935
+    proxy_corner:
+      avg_avg_return:                           0.23591356
+      avg_reward_per_step:                      0.0055007935
+  eval-batch-shift:
+    avg_avg_episode_length:                     75.50974
+    avg_avg_return:                             0.21200167
+    avg_benchmark_regret:                       0.7755043
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.0068359375
+    proxy_corner:
+      avg_avg_return:                           0.12512451
+      avg_reward_per_step:                      0.0011520386
+  eval-batch-tree:
+    avg_avg_episode_length:                     75.75785
+    avg_avg_return:                             0.2183848
+    avg_benchmark_regret:                       0.7662504
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.008552551
+    proxy_corner:
+      avg_avg_return:                           0.2183848
+      avg_reward_per_step:                      0.008552551
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        4194304
+    env-step-replay-before:                     2162688
+    env-step-used-before:                       4194304
+    ppo-update-before:                          2560
+    env-step-generate-before:                   2031616
+    env-step-all-after:                         4227072
+    env-step-replay-after:                      2195456
+    env-step-used-after:                        4227072
+    ppo-update-after:                           2580
+    env-step-generate-after:                    2031616
+  train-replay:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     63.2786
+    avg_avg_return:                             0.38970557
+    avg_reward_per_step:                        0.029937744
+    proxy_corner:
+      avg_avg_return:                           0.08155341
+      avg_reward_per_step:                      0.0011291504
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     63.2786
+    avg_avg_return:                             0.38970557
+    avg_reward_per_step:                        0.029937744
+    proxy_corner:
+      avg_avg_return:                           0.08155341
+      avg_reward_per_step:                      0.0011291504
+  perf:
+    env_steps_per_second:                       326846.63412310404
+    ppo_updates_per_second:                     40.56761888934678
+    eval-batch-train-duration:                  0.5233592055737972
+    eval-batch-orig-duration:                   0.47538330126553774
+    eval-batch-shift-duration:                  0.4506272841244936
+    eval-batch-tree-duration:                   0.4684401713311672
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    14.835693
+      cheese-corne_dist_wavg:                   16.762468
+      levels_cheese_is_in_corner_avg:           0.015136719
+      mouse-cheese_dist_finite_avg:             7.7321777
+      mouse-cheese_dist_finite_wavg:            1.7074158
+      mouse-cheese_dist_solvable_avg:           7.7321777
+      mouse-cheese_dist_solvable_wavg:          1.7074156
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.4441892
+    visit_patterns:
+      avg_first_visit_time:                     21.857666
+      avg_last_visit_time:                      57.365234
+      num_replay_batches:                       67
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0023623544
+    avg_actor_approxkl3:                        0.0023260778
+    avg_actor_clipfrac:                         0.15145265
+    avg_actor_loss:                             0.0005549535
+    avg_advantage:                              0.037143312
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.017932508
+    avg_entropy:                                1.2084993
+    avg_grad_norm_pre_clip:                     2.8034227
+    avg_loss:                                   0.008312707
+    max:
+      max_actor_approxkl1:                      0.00487273
+      max_actor_approxkl3:                      0.0045796055
+      max_actor_clipfrac:                       0.3861084
+      max_actor_loss:                           0.0040622503
+      max_advantage:                            0.5925951
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.020477857
+      max_entropy:                              1.2313775
+      max_grad_norm_pre_clip:                   6.644784
+      max_loss:                                 0.012497882
+    std:
+      std_actor_approxkl1:                      0.0014979341
+      std_actor_approxkl3:                      0.0013816835
+      std_actor_clipfrac:                       0.124022804
+      std_actor_loss:                           0.0018837432
+      std_advantage:                            0.18921597
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0013040764
+      std_entropy:                              0.013432364
+      std_grad_norm_pre_clip:                   2.0369475
+      std_loss:                                 0.0021115532
+  eval-batch-train:
+    avg_avg_episode_length:                     75.47551
+    avg_avg_return:                             0.21062016
+    avg_benchmark_regret:                       0.7765969
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.008728027
+    proxy_corner:
+      avg_avg_return:                           0.14382786
+      avg_reward_per_step:                      0.0015640259
+  eval-batch-orig:
+    avg_avg_episode_length:                     70.72531
+    avg_avg_return:                             0.28985736
+    avg_benchmark_regret:                       0.69465417
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.0104599
+    proxy_corner:
+      avg_avg_return:                           0.28985736
+      avg_reward_per_step:                      0.0104599
+  eval-batch-shift:
+    avg_avg_episode_length:                     74.46249
+    avg_avg_return:                             0.22116104
+    avg_benchmark_regret:                       0.7663449
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.01008606
+    proxy_corner:
+      avg_avg_return:                           0.12488323
+      avg_reward_per_step:                      0.0011291504
+  eval-batch-tree:
+    avg_avg_episode_length:                     70.81346
+    avg_avg_return:                             0.28107297
+    avg_benchmark_regret:                       0.70356226
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.017089844
+    proxy_corner:
+      avg_avg_return:                           0.28107297
+      avg_reward_per_step:                      0.017089844
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        5242880
+    env-step-replay-before:                     2555904
+    env-step-used-before:                       5242880
+    ppo-update-before:                          3200
+    env-step-generate-before:                   2686976
+    env-step-all-after:                         5275648
+    env-step-replay-after:                      2555904
+    env-step-used-after:                        5275648
+    ppo-update-after:                           3220
+    env-step-generate-after:                    2719744
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     75.589035
+    avg_avg_return:                             0.17548743
+    avg_reward_per_step:                        0.01651001
+    proxy_corner:
+      avg_avg_return:                           0.09471499
+      avg_reward_per_step:                      0.0014038086
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     75.589035
+    avg_avg_return:                             0.17548743
+    avg_reward_per_step:                        0.01651001
+    proxy_corner:
+      avg_avg_return:                           0.09471499
+      avg_reward_per_step:                      0.0014038086
+  perf:
+    env_steps_per_second:                       320672.0327458242
+    ppo_updates_per_second:                     42.14070500984584
+    eval-batch-train-duration:                  0.5351861994713545
+    eval-batch-orig-duration:                   0.4869404975324869
+    eval-batch-shift-duration:                  0.4821580946445465
+    eval-batch-tree-duration:                   0.5045371567830443
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    14.6875
+      cheese-corne_dist_wavg:                   23.050266
+      levels_cheese_is_in_corner_avg:           0.016113281
+      mouse-cheese_dist_finite_avg:             7.7165527
+      mouse-cheese_dist_finite_wavg:            1.6837585
+      mouse-cheese_dist_solvable_avg:           7.7165527
+      mouse-cheese_dist_solvable_wavg:          1.6837585
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016816
+    scoring:
+      avg_scores:                               0.5228095
+    visit_patterns:
+      avg_first_visit_time:                     30.892822
+      avg_last_visit_time:                      68.216064
+      num_replay_batches:                       78
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002934558
+    avg_actor_approxkl3:                        0.0029517591
+    avg_actor_clipfrac:                         0.17194214
+    avg_actor_loss:                             0.001524838
+    avg_advantage:                              -0.02336915
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.01596042
+    avg_entropy:                                1.0928748
+    avg_grad_norm_pre_clip:                     2.685634
+    avg_loss:                                   0.008412172
+    max:
+      max_actor_approxkl1:                      0.01140448
+      max_actor_approxkl3:                      0.011537857
+      max_actor_clipfrac:                       0.6791992
+      max_actor_loss:                           0.015416518
+      max_advantage:                            0.63881123
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.02162416
+      max_entropy:                              1.1311889
+      max_grad_norm_pre_clip:                   8.199289
+      max_loss:                                 0.025177218
+    std:
+      std_actor_approxkl1:                      0.0030710867
+      std_actor_approxkl3:                      0.0031950618
+      std_actor_clipfrac:                       0.20042668
+      std_actor_loss:                           0.004840739
+      std_advantage:                            0.1770519
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0026904156
+      std_entropy:                              0.021211753
+      std_grad_norm_pre_clip:                   2.3061304
+      std_loss:                                 0.005525586
+  eval-batch-train:
+    avg_avg_episode_length:                     74.45139
+    avg_avg_return:                             0.22100274
+    avg_benchmark_regret:                       0.7662143
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.009994507
+    proxy_corner:
+      avg_avg_return:                           0.0964884
+      avg_reward_per_step:                      0.0012359619
+  eval-batch-orig:
+    avg_avg_episode_length:                     70.45574
+    avg_avg_return:                             0.27712858
+    avg_benchmark_regret:                       0.7073829
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.012001038
+    proxy_corner:
+      avg_avg_return:                           0.27712858
+      avg_reward_per_step:                      0.012001038
+  eval-batch-shift:
+    avg_avg_episode_length:                     72.27571
+    avg_avg_return:                             0.25525814
+    avg_benchmark_regret:                       0.7322478
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.012413025
+    proxy_corner:
+      avg_avg_return:                           0.08776017
+      avg_reward_per_step:                      0.0008239746
+  eval-batch-tree:
+    avg_avg_episode_length:                     70.80052
+    avg_avg_return:                             0.2773715
+    avg_benchmark_regret:                       0.7072637
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.021018982
+    proxy_corner:
+      avg_avg_return:                           0.2773715
+      avg_reward_per_step:                      0.021018982
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        6291456
+    env-step-replay-before:                     3112960
+    env-step-used-before:                       6291456
+    ppo-update-before:                          3840
+    env-step-generate-before:                   3178496
+    env-step-all-after:                         6324224
+    env-step-replay-after:                      3112960
+    env-step-used-after:                        6324224
+    ppo-update-after:                           3860
+    env-step-generate-after:                    3211264
+  train-generate:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     76.01079
+    avg_avg_return:                             0.17231667
+    avg_reward_per_step:                        0.011108398
+    proxy_corner:
+      avg_avg_return:                           0.050465442
+      avg_reward_per_step:                      0.0010986328
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     76.01079
+    avg_avg_return:                             0.17231667
+    avg_reward_per_step:                        0.011108398
+    proxy_corner:
+      avg_avg_return:                           0.050465442
+      avg_reward_per_step:                      0.0010986328
+  perf:
+    env_steps_per_second:                       300111.12360754184
+    ppo_updates_per_second:                     41.72559545331134
+    eval-batch-train-duration:                  0.5384116042405367
+    eval-batch-orig-duration:                   0.4790945677086711
+    eval-batch-shift-duration:                  0.47629883605986834
+    eval-batch-tree-duration:                   0.4664371535181999
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    14.7092285
+      cheese-corne_dist_wavg:                   5.080107
+      levels_cheese_is_in_corner_avg:           0.0146484375
+      mouse-cheese_dist_finite_avg:             7.966797
+      mouse-cheese_dist_finite_wavg:            1.6966977
+      mouse-cheese_dist_solvable_avg:           7.966797
+      mouse-cheese_dist_solvable_wavg:          1.696698
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000008
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.5649637
+    visit_patterns:
+      avg_first_visit_time:                     38.905273
+      avg_last_visit_time:                      85.229004
+      num_replay_batches:                       95
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011045896
+    avg_actor_approxkl3:                        0.0011281498
+    avg_actor_clipfrac:                         0.05048218
+    avg_actor_loss:                             -0.0004782036
+    avg_advantage:                              -0.04462573
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.014306779
+    avg_entropy:                                1.0320375
+    avg_grad_norm_pre_clip:                     2.144247
+    avg_loss:                                   0.0056431494
+    max:
+      max_actor_approxkl1:                      0.0032073918
+      max_actor_approxkl3:                      0.0039136484
+      max_actor_clipfrac:                       0.23522949
+      max_actor_loss:                           0.004240304
+      max_advantage:                            0.5191059
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.016162228
+      max_entropy:                              1.0843363
+      max_grad_norm_pre_clip:                   8.430743
+      max_loss:                                 0.009644933
+    std:
+      std_actor_approxkl1:                      0.0008583795
+      std_actor_approxkl3:                      0.000922195
+      std_actor_clipfrac:                       0.06502973
+      std_actor_loss:                           0.0015389321
+      std_advantage:                            0.1687299
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0013141787
+      std_entropy:                              0.021325018
+      std_grad_norm_pre_clip:                   2.1203635
+      std_loss:                                 0.0016200447
+  eval-batch-train:
+    avg_avg_episode_length:                     72.45859
+    avg_avg_return:                             0.24785514
+    avg_benchmark_regret:                       0.7393619
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.0128479
+    proxy_corner:
+      avg_avg_return:                           0.07382144
+      avg_reward_per_step:                      0.0013809204
+  eval-batch-orig:
+    avg_avg_episode_length:                     69.55174
+    avg_avg_return:                             0.28687823
+    avg_benchmark_regret:                       0.69763327
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.01563263
+    proxy_corner:
+      avg_avg_return:                           0.28687823
+      avg_reward_per_step:                      0.01563263
+  eval-batch-shift:
+    avg_avg_episode_length:                     70.30873
+    avg_avg_return:                             0.27444768
+    avg_benchmark_regret:                       0.7130583
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.016998291
+    proxy_corner:
+      avg_avg_return:                           0.06714987
+      avg_reward_per_step:                      0.0006637573
+  eval-batch-tree:
+    avg_avg_episode_length:                     69.69321
+    avg_avg_return:                             0.26826152
+    avg_benchmark_regret:                       0.7163737
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.024658203
+    proxy_corner:
+      avg_avg_return:                           0.26826152
+      avg_reward_per_step:                      0.024658203
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        7340032
+    env-step-replay-before:                     3571712
+    env-step-used-before:                       7340032
+    ppo-update-before:                          4480
+    env-step-generate-before:                   3768320
+    env-step-all-after:                         7372800
+    env-step-replay-after:                      3571712
+    env-step-used-after:                        7372800
+    ppo-update-after:                           4500
+    env-step-generate-after:                    3801088
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     71.49007
+    avg_avg_return:                             0.21588373
+    avg_reward_per_step:                        0.023864746
+    proxy_corner:
+      avg_avg_return:                           0.045585524
+      avg_reward_per_step:                      0.00076293945
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     71.49007
+    avg_avg_return:                             0.21588373
+    avg_reward_per_step:                        0.023864746
+    proxy_corner:
+      avg_avg_return:                           0.045585524
+      avg_reward_per_step:                      0.00076293945
+  perf:
+    env_steps_per_second:                       305975.70732007746
+    ppo_updates_per_second:                     42.26471908833313
+    eval-batch-train-duration:                  0.5380861684679985
+    eval-batch-orig-duration:                   0.4799091760069132
+    eval-batch-shift-duration:                  0.47948728036135435
+    eval-batch-tree-duration:                   0.46551939379423857
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    14.691162
+      cheese-corne_dist_wavg:                   14.969913
+      levels_cheese_is_in_corner_avg:           0.013671875
+      mouse-cheese_dist_finite_avg:             8.221436
+      mouse-cheese_dist_finite_wavg:            1.7077128
+      mouse-cheese_dist_solvable_avg:           8.221436
+      mouse-cheese_dist_solvable_wavg:          1.707713
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.60323274
+    visit_patterns:
+      avg_first_visit_time:                     48.3396
+      avg_last_visit_time:                      99.27417
+      num_replay_batches:                       109
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0017635064
+    avg_actor_approxkl3:                        0.0014902262
+    avg_actor_clipfrac:                         0.07819825
+    avg_actor_loss:                             -0.0002418846
+    avg_advantage:                              -0.022757469
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.017251717
+    avg_entropy:                                0.9399093
+    avg_grad_norm_pre_clip:                     2.8156803
+    avg_loss:                                   0.0074440646
+    max:
+      max_actor_approxkl1:                      0.0048182383
+      max_actor_approxkl3:                      0.0036591385
+      max_actor_clipfrac:                       0.22729492
+      max_actor_loss:                           0.0011475682
+      max_advantage:                            0.58805
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.020806797
+      max_entropy:                              1.0077889
+      max_grad_norm_pre_clip:                   5.2045665
+      max_loss:                                 0.010298483
+    std:
+      std_actor_approxkl1:                      0.0010806104
+      std_actor_approxkl3:                      0.00088096305
+      std_actor_clipfrac:                       0.06118065
+      std_actor_loss:                           0.0008292968
+      std_advantage:                            0.18617128
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0015352037
+      std_entropy:                              0.032899596
+      std_grad_norm_pre_clip:                   1.6849277
+      std_loss:                                 0.0013195663
+  eval-batch-train:
+    avg_avg_episode_length:                     71.25221
+    avg_avg_return:                             0.2626722
+    avg_benchmark_regret:                       0.7245449
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.014518738
+    proxy_corner:
+      avg_avg_return:                           0.06675737
+      avg_reward_per_step:                      0.0015869141
+  eval-batch-orig:
+    avg_avg_episode_length:                     68.53476
+    avg_avg_return:                             0.28176987
+    avg_benchmark_regret:                       0.70274174
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.020515442
+    proxy_corner:
+      avg_avg_return:                           0.28176987
+      avg_reward_per_step:                      0.020515442
+  eval-batch-shift:
+    avg_avg_episode_length:                     69.43492
+    avg_avg_return:                             0.28043926
+    avg_benchmark_regret:                       0.70706666
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.020515442
+    proxy_corner:
+      avg_avg_return:                           0.07342856
+      avg_reward_per_step:                      0.0007324219
+  eval-batch-tree:
+    avg_avg_episode_length:                     69.65029
+    avg_avg_return:                             0.2535795
+    avg_benchmark_regret:                       0.73105574
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.029586792
+    proxy_corner:
+      avg_avg_return:                           0.2535795
+      avg_reward_per_step:                      0.029586792
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        8388608
+    env-step-replay-before:                     4063232
+    env-step-used-before:                       8388608
+    ppo-update-before:                          5120
+    env-step-generate-before:                   4325376
+    env-step-all-after:                         8421376
+    env-step-replay-after:                      4096000
+    env-step-used-after:                        8421376
+    ppo-update-after:                           5140
+    env-step-generate-after:                    4325376
+  train-replay:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     68.714005
+    avg_avg_return:                             0.32997555
+    avg_reward_per_step:                        0.020568848
+    proxy_corner:
+      avg_avg_return:                           0.06733298
+      avg_reward_per_step:                      0.0009460449
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     68.714005
+    avg_avg_return:                             0.32997555
+    avg_reward_per_step:                        0.020568848
+    proxy_corner:
+      avg_avg_return:                           0.06733298
+      avg_reward_per_step:                      0.0009460449
+  perf:
+    env_steps_per_second:                       318083.6948222816
+    ppo_updates_per_second:                     41.680306802156174
+    eval-batch-train-duration:                  0.5442555854097009
+    eval-batch-orig-duration:                   0.49263163283467293
+    eval-batch-shift-duration:                  0.4679943658411503
+    eval-batch-tree-duration:                   0.4570767693221569
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    14.721436
+      cheese-corne_dist_wavg:                   21.249674
+      levels_cheese_is_in_corner_avg:           0.013183594
+      mouse-cheese_dist_finite_avg:             8.537109
+      mouse-cheese_dist_finite_wavg:            1.7447271
+      mouse-cheese_dist_solvable_avg:           8.537109
+      mouse-cheese_dist_solvable_wavg:          1.7447271
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.420168
+    scoring:
+      avg_scores:                               0.6303698
+    visit_patterns:
+      avg_first_visit_time:                     56.859863
+      avg_last_visit_time:                      115.26001
+      num_replay_batches:                       125
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00182673
+    avg_actor_approxkl3:                        0.0014700888
+    avg_actor_clipfrac:                         0.07852173
+    avg_actor_loss:                             -0.0002325628
+    avg_advantage:                              0.009054781
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.015083822
+    avg_entropy:                                0.92987984
+    avg_grad_norm_pre_clip:                     2.773018
+    avg_loss:                                   0.0063794698
+    max:
+      max_actor_approxkl1:                      0.0032882176
+      max_actor_approxkl3:                      0.002848384
+      max_actor_clipfrac:                       0.19763184
+      max_actor_loss:                           0.0035324842
+      max_advantage:                            0.6534146
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.017438214
+      max_entropy:                              0.9674851
+      max_grad_norm_pre_clip:                   8.688535
+      max_loss:                                 0.010139606
+    std:
+      std_actor_approxkl1:                      0.0010485
+      std_actor_approxkl3:                      0.0007791468
+      std_actor_clipfrac:                       0.05615244
+      std_actor_loss:                           0.0013578694
+      std_advantage:                            0.17357758
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0012063739
+      std_entropy:                              0.022551203
+      std_grad_norm_pre_clip:                   2.0092216
+      std_loss:                                 0.0014138896
+  eval-batch-train:
+    avg_avg_episode_length:                     69.90633
+    avg_avg_return:                             0.28634125
+    avg_benchmark_regret:                       0.70087576
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.018760681
+    proxy_corner:
+      avg_avg_return:                           0.058387384
+      avg_reward_per_step:                      0.0020217896
+  eval-batch-orig:
+    avg_avg_episode_length:                     68.24031
+    avg_avg_return:                             0.28577748
+    avg_benchmark_regret:                       0.6987341
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.02091217
+    proxy_corner:
+      avg_avg_return:                           0.28577748
+      avg_reward_per_step:                      0.02091217
+  eval-batch-shift:
+    avg_avg_episode_length:                     68.33695
+    avg_avg_return:                             0.2870165
+    avg_benchmark_regret:                       0.7004894
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.021064758
+    proxy_corner:
+      avg_avg_return:                           0.062207427
+      avg_reward_per_step:                      0.0006713867
+  eval-batch-tree:
+    avg_avg_episode_length:                     69.53152
+    avg_avg_return:                             0.26574224
+    avg_benchmark_regret:                       0.71889293
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.02859497
+    proxy_corner:
+      avg_avg_return:                           0.26574224
+      avg_reward_per_step:                      0.02859497
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        9437184
+    env-step-replay-before:                     4653056
+    env-step-used-before:                       9437184
+    ppo-update-before:                          5760
+    env-step-generate-before:                   4784128
+    env-step-all-after:                         9469952
+    env-step-replay-after:                      4653056
+    env-step-used-after:                        9469952
+    ppo-update-after:                           5780
+    env-step-generate-after:                    4816896
+  train-generate:
+    shift_proportion:                           0.96875
+    avg_avg_episode_length:                     70.79403
+    avg_avg_return:                             0.23780355
+    avg_reward_per_step:                        0.015594482
+    proxy_corner:
+      avg_avg_return:                           0.029223524
+      avg_reward_per_step:                      0.0004272461
+  train-all:
+    shift_proportion:                           0.96875
+    avg_avg_episode_length:                     70.79403
+    avg_avg_return:                             0.23780355
+    avg_reward_per_step:                        0.015594482
+    proxy_corner:
+      avg_avg_return:                           0.029223524
+      avg_reward_per_step:                      0.0004272461
+  perf:
+    env_steps_per_second:                       358890.2144251695
+    ppo_updates_per_second:                     42.94328703555809
+    eval-batch-train-duration:                  0.5121813099831343
+    eval-batch-orig-duration:                   0.4803346013650298
+    eval-batch-shift-duration:                  0.44435030966997147
+    eval-batch-tree-duration:                   0.46736485976725817
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    14.638428
+      cheese-corne_dist_wavg:                   23.947584
+      levels_cheese_is_in_corner_avg:           0.013671875
+      mouse-cheese_dist_finite_avg:             8.854492
+      mouse-cheese_dist_finite_wavg:            2.6685932
+      mouse-cheese_dist_solvable_avg:           8.854492
+      mouse-cheese_dist_solvable_wavg:          2.6685934
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016798
+    scoring:
+      avg_scores:                               0.6292499
+    visit_patterns:
+      avg_first_visit_time:                     65.13428
+      avg_last_visit_time:                      132.28564
+      num_replay_batches:                       142
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013005178
+    avg_actor_approxkl3:                        0.0013243853
+    avg_actor_clipfrac:                         0.06506958
+    avg_actor_loss:                             -0.0004578762
+    avg_advantage:                              -0.0034079703
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.016322808
+    avg_entropy:                                0.94635385
+    avg_grad_norm_pre_clip:                     2.7157788
+    avg_loss:                                   0.0067571737
+    max:
+      max_actor_approxkl1:                      0.0030529366
+      max_actor_approxkl3:                      0.0021067923
+      max_actor_clipfrac:                       0.1427002
+      max_actor_loss:                           0.0007273406
+      max_advantage:                            0.5796182
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.019386068
+      max_entropy:                              0.9766972
+      max_grad_norm_pre_clip:                   5.779771
+      max_loss:                                 0.00884738
+    std:
+      std_actor_approxkl1:                      0.00073980703
+      std_actor_approxkl3:                      0.00057658274
+      std_actor_clipfrac:                       0.042302135
+      std_actor_loss:                           0.0008498139
+      std_advantage:                            0.180904
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0018490292
+      std_entropy:                              0.013807936
+      std_grad_norm_pre_clip:                   1.4293158
+      std_loss:                                 0.0011685152
+  eval-batch-train:
+    avg_avg_episode_length:                     67.65935
+    avg_avg_return:                             0.30974504
+    avg_benchmark_regret:                       0.67747205
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.021194458
+    proxy_corner:
+      avg_avg_return:                           0.051343605
+      avg_reward_per_step:                      0.0014648438
+  eval-batch-orig:
+    avg_avg_episode_length:                     67.98021
+    avg_avg_return:                             0.2929424
+    avg_benchmark_regret:                       0.6915692
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.020576477
+    proxy_corner:
+      avg_avg_return:                           0.2929424
+      avg_reward_per_step:                      0.020576477
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.85701
+    avg_avg_return:                             0.3114279
+    avg_benchmark_regret:                       0.6760781
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.02658844
+    proxy_corner:
+      avg_avg_return:                           0.046554208
+      avg_reward_per_step:                      0.000541687
+  eval-batch-tree:
+    avg_avg_episode_length:                     68.80674
+    avg_avg_return:                             0.28002083
+    avg_benchmark_regret:                       0.7046144
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.029518127
+    proxy_corner:
+      avg_avg_return:                           0.28002083
+      avg_reward_per_step:                      0.029518127
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        10485760
+    env-step-replay-before:                     5210112
+    env-step-used-before:                       10485760
+    ppo-update-before:                          6400
+    env-step-generate-before:                   5275648
+    env-step-all-after:                         10518528
+    env-step-replay-after:                      5242880
+    env-step-used-after:                        10518528
+    ppo-update-after:                           6420
+    env-step-generate-after:                    5275648
+  train-replay:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     65.93935
+    avg_avg_return:                             0.36920092
+    avg_reward_per_step:                        0.016052246
+    proxy_corner:
+      avg_avg_return:                           0.02433005
+      avg_reward_per_step:                      0.00088500977
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     65.93935
+    avg_avg_return:                             0.36920092
+    avg_reward_per_step:                        0.016052246
+    proxy_corner:
+      avg_avg_return:                           0.02433005
+      avg_reward_per_step:                      0.00088500977
+  perf:
+    env_steps_per_second:                       258474.86389630003
+    ppo_updates_per_second:                     42.844240652962455
+    eval-batch-train-duration:                  0.6743863243609667
+    eval-batch-orig-duration:                   0.6139816539362073
+    eval-batch-shift-duration:                  0.6184256598353386
+    eval-batch-tree-duration:                   0.6285080574452877
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    14.620605
+      cheese-corne_dist_wavg:                   29.34791
+      levels_cheese_is_in_corner_avg:           0.012939453
+      mouse-cheese_dist_finite_avg:             9.2421875
+      mouse-cheese_dist_finite_wavg:            1.8197315
+      mouse-cheese_dist_solvable_avg:           9.2421875
+      mouse-cheese_dist_solvable_wavg:          1.8197317
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.63070464
+    visit_patterns:
+      avg_first_visit_time:                     73.90039
+      avg_last_visit_time:                      150.33447
+      num_replay_batches:                       160
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011405488
+    avg_actor_approxkl3:                        0.0010086056
+    avg_actor_clipfrac:                         0.045703124
+    avg_actor_loss:                             -0.00085476006
+    avg_advantage:                              0.0132600535
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.01384526
+    avg_entropy:                                0.9487872
+    avg_grad_norm_pre_clip:                     2.2451813
+    avg_loss:                                   0.005119083
+    max:
+      max_actor_approxkl1:                      0.0023962078
+      max_actor_approxkl3:                      0.0019192584
+      max_actor_clipfrac:                       0.10449219
+      max_actor_loss:                           0.0005005151
+      max_advantage:                            0.49466857
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0154218525
+      max_entropy:                              0.98747814
+      max_grad_norm_pre_clip:                   5.0387692
+      max_loss:                                 0.0066172592
+    std:
+      std_actor_approxkl1:                      0.0007532312
+      std_actor_approxkl3:                      0.00046879356
+      std_actor_clipfrac:                       0.0334101
+      std_actor_loss:                           0.0007713669
+      std_advantage:                            0.16668123
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00083713786
+      std_entropy:                              0.016399935
+      std_grad_norm_pre_clip:                   1.304709
+      std_loss:                                 0.00089520315
+  eval-batch-train:
+    avg_avg_episode_length:                     66.197296
+    avg_avg_return:                             0.31823066
+    avg_benchmark_regret:                       0.66898644
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.023864746
+    proxy_corner:
+      avg_avg_return:                           0.04470308
+      avg_reward_per_step:                      0.0016555786
+  eval-batch-orig:
+    avg_avg_episode_length:                     67.64819
+    avg_avg_return:                             0.29857674
+    avg_benchmark_regret:                       0.6859348
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.02292633
+    proxy_corner:
+      avg_avg_return:                           0.29857674
+      avg_reward_per_step:                      0.02292633
+  eval-batch-shift:
+    avg_avg_episode_length:                     65.27754
+    avg_avg_return:                             0.32457834
+    avg_benchmark_regret:                       0.66292757
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.032310486
+    proxy_corner:
+      avg_avg_return:                           0.037045874
+      avg_reward_per_step:                      0.00045013428
+  eval-batch-tree:
+    avg_avg_episode_length:                     68.47316
+    avg_avg_return:                             0.27489343
+    avg_benchmark_regret:                       0.7097418
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.034347534
+    proxy_corner:
+      avg_avg_return:                           0.27489343
+      avg_reward_per_step:                      0.034347534
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        11534336
+    env-step-replay-before:                     5767168
+    env-step-used-before:                       11534336
+    ppo-update-before:                          7040
+    env-step-generate-before:                   5767168
+    env-step-all-after:                         11567104
+    env-step-replay-after:                      5767168
+    env-step-used-after:                        11567104
+    ppo-update-after:                           7060
+    env-step-generate-after:                    5799936
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     68.77734
+    avg_avg_return:                             0.26137397
+    avg_reward_per_step:                        0.026245117
+    proxy_corner:
+      avg_avg_return:                           0.032730184
+      avg_reward_per_step:                      0.0002746582
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     68.77734
+    avg_avg_return:                             0.26137397
+    avg_reward_per_step:                        0.026245117
+    proxy_corner:
+      avg_avg_return:                           0.032730184
+      avg_reward_per_step:                      0.0002746582
+  perf:
+    env_steps_per_second:                       322596.0796136773
+    ppo_updates_per_second:                     42.43836191054917
+    eval-batch-train-duration:                  0.5304184937849641
+    eval-batch-orig-duration:                   0.48545783665031195
+    eval-batch-shift-duration:                  0.48020751029253006
+    eval-batch-tree-duration:                   0.48631398752331734
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    14.693848
+      cheese-corne_dist_wavg:                   12.26701
+      levels_cheese_is_in_corner_avg:           0.014160156
+      mouse-cheese_dist_finite_avg:             9.56543
+      mouse-cheese_dist_finite_wavg:            1.8423246
+      mouse-cheese_dist_solvable_avg:           9.56543
+      mouse-cheese_dist_solvable_wavg:          1.8423246
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999985
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.6466149
+    visit_patterns:
+      avg_first_visit_time:                     84.67432
+      avg_last_visit_time:                      166.22046
+      num_replay_batches:                       176
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0018838765
+    avg_actor_approxkl3:                        0.0017416585
+    avg_actor_clipfrac:                         0.09730225
+    avg_actor_loss:                             -4.9184637e-05
+    avg_advantage:                              -0.0073398836
+    avg_critic_clipfrac:                        0.002081299
+    avg_critic_loss:                            0.016856566
+    avg_entropy:                                0.8989816
+    avg_grad_norm_pre_clip:                     3.0086432
+    avg_loss:                                   0.007480118
+    max:
+      max_actor_approxkl1:                      0.0043179467
+      max_actor_approxkl3:                      0.00399358
+      max_actor_clipfrac:                       0.25476074
+      max_actor_loss:                           0.0030270368
+      max_advantage:                            0.5718142
+      max_critic_clipfrac:                      0.016113281
+      max_critic_loss:                          0.019747283
+      max_entropy:                              0.9163536
+      max_grad_norm_pre_clip:                   7.4073353
+      max_loss:                                 0.011008247
+    std:
+      std_actor_approxkl1:                      0.0011185325
+      std_actor_approxkl3:                      0.0009774631
+      std_actor_clipfrac:                       0.067788936
+      std_actor_loss:                           0.0013016227
+      std_advantage:                            0.18280937
+      std_critic_clipfrac:                      0.005008674
+      std_critic_loss:                          0.0016605852
+      std_entropy:                              0.012900591
+      std_grad_norm_pre_clip:                   1.9936361
+      std_loss:                                 0.001683076
+  eval-batch-train:
+    avg_avg_episode_length:                     64.864685
+    avg_avg_return:                             0.3347398
+    avg_benchmark_regret:                       0.65247726
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.025886536
+    proxy_corner:
+      avg_avg_return:                           0.05095791
+      avg_reward_per_step:                      0.0017776489
+  eval-batch-orig:
+    avg_avg_episode_length:                     67.78603
+    avg_avg_return:                             0.27974978
+    avg_benchmark_regret:                       0.70476174
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.026641846
+    proxy_corner:
+      avg_avg_return:                           0.27974978
+      avg_reward_per_step:                      0.026641846
+  eval-batch-shift:
+    avg_avg_episode_length:                     65.00404
+    avg_avg_return:                             0.32678413
+    avg_benchmark_regret:                       0.6607218
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.03538513
+    proxy_corner:
+      avg_avg_return:                           0.04754001
+      avg_reward_per_step:                      0.0005340576
+  eval-batch-tree:
+    avg_avg_episode_length:                     68.63945
+    avg_avg_return:                             0.25733316
+    avg_benchmark_regret:                       0.727302
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.037612915
+    proxy_corner:
+      avg_avg_return:                           0.25733316
+      avg_reward_per_step:                      0.037612915
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        12582912
+    env-step-replay-before:                     6291456
+    env-step-used-before:                       12582912
+    ppo-update-before:                          7680
+    env-step-generate-before:                   6291456
+    env-step-all-after:                         12615680
+    env-step-replay-after:                      6324224
+    env-step-used-after:                        12615680
+    ppo-update-after:                           7700
+    env-step-generate-after:                    6291456
+  train-replay:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     68.64802
+    avg_avg_return:                             0.35811526
+    avg_reward_per_step:                        0.012634277
+    proxy_corner:
+      avg_avg_return:                           0.015313698
+      avg_reward_per_step:                      0.0005187988
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     68.64802
+    avg_avg_return:                             0.35811526
+    avg_reward_per_step:                        0.012634277
+    proxy_corner:
+      avg_avg_return:                           0.015313698
+      avg_reward_per_step:                      0.0005187988
+  perf:
+    env_steps_per_second:                       353137.4753429866
+    ppo_updates_per_second:                     42.28041949764451
+    eval-batch-train-duration:                  0.518830306828022
+    eval-batch-orig-duration:                   0.4685316998511553
+    eval-batch-shift-duration:                  0.4699910217896104
+    eval-batch-tree-duration:                   0.5024609360843897
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    14.733154
+      cheese-corne_dist_wavg:                   20.379623
+      levels_cheese_is_in_corner_avg:           0.014160156
+      mouse-cheese_dist_finite_avg:             9.849609
+      mouse-cheese_dist_finite_wavg:            2.7816296
+      mouse-cheese_dist_solvable_avg:           9.849609
+      mouse-cheese_dist_solvable_wavg:          2.7816293
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016798
+    scoring:
+      avg_scores:                               0.64886016
+    visit_patterns:
+      avg_first_visit_time:                     94.82593
+      avg_last_visit_time:                      183.3789
+      num_replay_batches:                       193
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001512244
+    avg_actor_approxkl3:                        0.0013358782
+    avg_actor_clipfrac:                         0.06741333
+    avg_actor_loss:                             -0.0006170422
+    avg_advantage:                              0.012820956
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.014433063
+    avg_entropy:                                0.9122966
+    avg_grad_norm_pre_clip:                     2.5268667
+    avg_loss:                                   0.005687192
+    max:
+      max_actor_approxkl1:                      0.004203674
+      max_actor_approxkl3:                      0.00290344
+      max_actor_clipfrac:                       0.18493652
+      max_actor_loss:                           0.003512904
+      max_advantage:                            0.57885396
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.016309787
+      max_entropy:                              0.9327842
+      max_grad_norm_pre_clip:                   6.429598
+      max_loss:                                 0.009179016
+    std:
+      std_actor_approxkl1:                      0.00094729156
+      std_actor_approxkl3:                      0.0006104703
+      std_actor_clipfrac:                       0.04449039
+      std_actor_loss:                           0.0013422131
+      std_advantage:                            0.1702107
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0009502166
+      std_entropy:                              0.01346106
+      std_grad_norm_pre_clip:                   1.4632591
+      std_loss:                                 0.0013667216
+  eval-batch-train:
+    avg_avg_episode_length:                     63.283913
+    avg_avg_return:                             0.35508388
+    avg_benchmark_regret:                       0.6321332
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.03139496
+    proxy_corner:
+      avg_avg_return:                           0.047889203
+      avg_reward_per_step:                      0.002029419
+  eval-batch-orig:
+    avg_avg_episode_length:                     67.45368
+    avg_avg_return:                             0.28467977
+    avg_benchmark_regret:                       0.6998318
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.025764465
+    proxy_corner:
+      avg_avg_return:                           0.28467977
+      avg_reward_per_step:                      0.025764465
+  eval-batch-shift:
+    avg_avg_episode_length:                     62.2498
+    avg_avg_return:                             0.36494163
+    avg_benchmark_regret:                       0.6225643
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.038612366
+    proxy_corner:
+      avg_avg_return:                           0.03515351
+      avg_reward_per_step:                      0.0005340576
+  eval-batch-tree:
+    avg_avg_episode_length:                     69.82065
+    avg_avg_return:                             0.25076988
+    avg_benchmark_regret:                       0.7338653
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.034934998
+    proxy_corner:
+      avg_avg_return:                           0.25076988
+      avg_reward_per_step:                      0.034934998
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        13631488
+    env-step-replay-before:                     6782976
+    env-step-used-before:                       13631488
+    ppo-update-before:                          8320
+    env-step-generate-before:                   6848512
+    env-step-all-after:                         13664256
+    env-step-replay-after:                      6782976
+    env-step-used-after:                        13664256
+    ppo-update-after:                           8340
+    env-step-generate-after:                    6881280
+  train-generate:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     67.709595
+    avg_avg_return:                             0.26669767
+    avg_reward_per_step:                        0.030029297
+    proxy_corner:
+      avg_avg_return:                           0.039635006
+      avg_reward_per_step:                      0.002105713
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     67.709595
+    avg_avg_return:                             0.26669767
+    avg_reward_per_step:                        0.030029297
+    proxy_corner:
+      avg_avg_return:                           0.039635006
+      avg_reward_per_step:                      0.002105713
+  perf:
+    env_steps_per_second:                       351137.5201305838
+    ppo_updates_per_second:                     41.64850866105223
+    eval-batch-train-duration:                  0.4968958245590329
+    eval-batch-orig-duration:                   0.42739086877554655
+    eval-batch-shift-duration:                  0.41382777225226164
+    eval-batch-tree-duration:                   0.4303861306980252
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    14.766113
+      cheese-corne_dist_wavg:                   30.249851
+      levels_cheese_is_in_corner_avg:           0.014404297
+      mouse-cheese_dist_finite_avg:             10.202393
+      mouse-cheese_dist_finite_wavg:            3.7208784
+      mouse-cheese_dist_solvable_avg:           10.202393
+      mouse-cheese_dist_solvable_wavg:          3.7208784
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.65663683
+    visit_patterns:
+      avg_first_visit_time:                     107.355225
+      avg_last_visit_time:                      197.33179
+      num_replay_batches:                       207
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0024555174
+    avg_actor_approxkl3:                        0.0023183932
+    avg_actor_clipfrac:                         0.121807866
+    avg_actor_loss:                             0.0014103529
+    avg_advantage:                              0.003539966
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.017328765
+    avg_entropy:                                0.88304913
+    avg_grad_norm_pre_clip:                     3.2366357
+    avg_loss:                                   0.009191685
+    max:
+      max_actor_approxkl1:                      0.0058353366
+      max_actor_approxkl3:                      0.0067174407
+      max_actor_clipfrac:                       0.33483887
+      max_actor_loss:                           0.006629735
+      max_advantage:                            0.60923517
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.019954145
+      max_entropy:                              0.9313969
+      max_grad_norm_pre_clip:                   5.733222
+      max_loss:                                 0.015016112
+    std:
+      std_actor_approxkl1:                      0.0017663826
+      std_actor_approxkl3:                      0.0016052998
+      std_actor_clipfrac:                       0.08821101
+      std_actor_loss:                           0.0023131988
+      std_advantage:                            0.18631011
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0014079766
+      std_entropy:                              0.02658742
+      std_grad_norm_pre_clip:                   1.3971899
+      std_loss:                                 0.0024652153
+  eval-batch-train:
+    avg_avg_episode_length:                     62.145584
+    avg_avg_return:                             0.3752694
+    avg_benchmark_regret:                       0.61194766
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.0311203
+    proxy_corner:
+      avg_avg_return:                           0.040828884
+      avg_reward_per_step:                      0.0017318726
+  eval-batch-orig:
+    avg_avg_episode_length:                     67.20332
+    avg_avg_return:                             0.29891142
+    avg_benchmark_regret:                       0.68560016
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.025146484
+    proxy_corner:
+      avg_avg_return:                           0.29891142
+      avg_reward_per_step:                      0.025146484
+  eval-batch-shift:
+    avg_avg_episode_length:                     62.861336
+    avg_avg_return:                             0.36116183
+    avg_benchmark_regret:                       0.62634414
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.041412354
+    proxy_corner:
+      avg_avg_return:                           0.038924307
+      avg_reward_per_step:                      0.000541687
+  eval-batch-tree:
+    avg_avg_episode_length:                     67.894295
+    avg_avg_return:                             0.28167272
+    avg_benchmark_regret:                       0.70296246
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.035873413
+    proxy_corner:
+      avg_avg_return:                           0.28167272
+      avg_reward_per_step:                      0.035873413
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        14680064
+    env-step-replay-before:                     7405568
+    env-step-used-before:                       14680064
+    ppo-update-before:                          8960
+    env-step-generate-before:                   7274496
+    env-step-all-after:                         14712832
+    env-step-replay-after:                      7438336
+    env-step-used-after:                        14712832
+    ppo-update-after:                           8980
+    env-step-generate-after:                    7274496
+  train-replay:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     69.68133
+    avg_avg_return:                             0.32994792
+    avg_reward_per_step:                        0.0119018555
+    proxy_corner:
+      avg_avg_return:                           0.037124667
+      avg_reward_per_step:                      0.0007019043
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     69.68133
+    avg_avg_return:                             0.32994792
+    avg_reward_per_step:                        0.0119018555
+    proxy_corner:
+      avg_avg_return:                           0.037124667
+      avg_reward_per_step:                      0.0007019043
+  perf:
+    env_steps_per_second:                       348936.7393806421
+    ppo_updates_per_second:                     41.33639529759746
+    eval-batch-train-duration:                  0.5338325891643763
+    eval-batch-orig-duration:                   0.482501027174294
+    eval-batch-shift-duration:                  0.45085654128342867
+    eval-batch-tree-duration:                   0.48227620031684637
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    14.802734
+      cheese-corne_dist_wavg:                   30.259418
+      levels_cheese_is_in_corner_avg:           0.0146484375
+      mouse-cheese_dist_finite_avg:             10.425537
+      mouse-cheese_dist_finite_wavg:            3.7420788
+      mouse-cheese_dist_solvable_avg:           10.425537
+      mouse-cheese_dist_solvable_wavg:          3.7420793
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.6571909
+    visit_patterns:
+      avg_first_visit_time:                     115.5957
+      avg_last_visit_time:                      217.2876
+      num_replay_batches:                       227
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014828753
+    avg_actor_approxkl3:                        0.0012676179
+    avg_actor_clipfrac:                         0.064367674
+    avg_actor_loss:                             -0.0005569343
+    avg_advantage:                              0.0023807718
+    avg_critic_clipfrac:                        7.934571e-05
+    avg_critic_loss:                            0.0141737405
+    avg_entropy:                                0.92514056
+    avg_grad_norm_pre_clip:                     2.1379898
+    avg_loss:                                   0.0056047956
+    max:
+      max_actor_approxkl1:                      0.004923717
+      max_actor_approxkl3:                      0.003460708
+      max_actor_clipfrac:                       0.20874023
+      max_actor_loss:                           0.005171135
+      max_advantage:                            0.59066045
+      max_critic_clipfrac:                      0.0015869141
+      max_critic_loss:                          0.015754342
+      max_entropy:                              0.9657169
+      max_grad_norm_pre_clip:                   5.316802
+      max_loss:                                 0.011994226
+    std:
+      std_actor_approxkl1:                      0.0011451672
+      std_actor_approxkl3:                      0.0007743268
+      std_actor_clipfrac:                       0.052419342
+      std_actor_loss:                           0.001754827
+      std_advantage:                            0.1697206
+      std_critic_clipfrac:                      0.00034585988
+      std_critic_loss:                          0.00085033895
+      std_entropy:                              0.016320353
+      std_grad_norm_pre_clip:                   1.3954159
+      std_loss:                                 0.001834824
+  eval-batch-train:
+    avg_avg_episode_length:                     61.111958
+    avg_avg_return:                             0.3868088
+    avg_benchmark_regret:                       0.6004083
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.03655243
+    proxy_corner:
+      avg_avg_return:                           0.03673964
+      avg_reward_per_step:                      0.001449585
+  eval-batch-orig:
+    avg_avg_episode_length:                     65.86624
+    avg_avg_return:                             0.3277859
+    avg_benchmark_regret:                       0.6567256
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.025886536
+    proxy_corner:
+      avg_avg_return:                           0.3277859
+      avg_reward_per_step:                      0.025886536
+  eval-batch-shift:
+    avg_avg_episode_length:                     61.26384
+    avg_avg_return:                             0.3713408
+    avg_benchmark_regret:                       0.61616516
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.045333862
+    proxy_corner:
+      avg_avg_return:                           0.030638305
+      avg_reward_per_step:                      0.00045013428
+  eval-batch-tree:
+    avg_avg_episode_length:                     67.219894
+    avg_avg_return:                             0.3126198
+    avg_benchmark_regret:                       0.67201537
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.034057617
+    proxy_corner:
+      avg_avg_return:                           0.3126198
+      avg_reward_per_step:                      0.034057617
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        15728640
+    env-step-replay-before:                     7766016
+    env-step-used-before:                       15728640
+    ppo-update-before:                          9600
+    env-step-generate-before:                   7962624
+    env-step-all-after:                         15761408
+    env-step-replay-after:                      7766016
+    env-step-used-after:                        15761408
+    ppo-update-after:                           9620
+    env-step-generate-after:                    7995392
+  train-generate:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     62.203735
+    avg_avg_return:                             0.34222245
+    avg_reward_per_step:                        0.046813965
+    proxy_corner:
+      avg_avg_return:                           0.042422544
+      avg_reward_per_step:                      0.0024108887
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     62.203735
+    avg_avg_return:                             0.34222245
+    avg_reward_per_step:                        0.046813965
+    proxy_corner:
+      avg_avg_return:                           0.042422544
+      avg_reward_per_step:                      0.0024108887
+  perf:
+    env_steps_per_second:                       292644.4741422804
+    ppo_updates_per_second:                     42.65012678911743
+    eval-batch-train-duration:                  0.535033717751503
+    eval-batch-orig-duration:                   0.47645300906151533
+    eval-batch-shift-duration:                  0.4474740168079734
+    eval-batch-tree-duration:                   0.48680424969643354
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    14.847168
+      cheese-corne_dist_wavg:                   19.468948
+      levels_cheese_is_in_corner_avg:           0.015625
+      mouse-cheese_dist_finite_avg:             10.753662
+      mouse-cheese_dist_finite_wavg:            2.8652806
+      mouse-cheese_dist_solvable_avg:           10.753662
+      mouse-cheese_dist_solvable_wavg:          2.8652806
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.66763103
+    visit_patterns:
+      avg_first_visit_time:                     128.70557
+      avg_last_visit_time:                      227.21167
+      num_replay_batches:                       237
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0016889217
+    avg_actor_approxkl3:                        0.0016142076
+    avg_actor_clipfrac:                         0.0864624
+    avg_actor_loss:                             -0.0001581274
+    avg_advantage:                              0.0116231665
+    avg_critic_clipfrac:                        0.0013671875
+    avg_critic_loss:                            0.01622385
+    avg_entropy:                                0.87132543
+    avg_grad_norm_pre_clip:                     3.0982347
+    avg_loss:                                   0.007082472
+    max:
+      max_actor_approxkl1:                      0.005437459
+      max_actor_approxkl3:                      0.0036200266
+      max_actor_clipfrac:                       0.20324707
+      max_actor_loss:                           0.0031509548
+      max_advantage:                            0.6528025
+      max_critic_clipfrac:                      0.0126953125
+      max_critic_loss:                          0.01768851
+      max_entropy:                              0.9121964
+      max_grad_norm_pre_clip:                   6.7718534
+      max_loss:                                 0.010348841
+    std:
+      std_actor_approxkl1:                      0.001289534
+      std_actor_approxkl3:                      0.0008641425
+      std_actor_clipfrac:                       0.05455717
+      std_actor_loss:                           0.0011311626
+      std_advantage:                            0.18248521
+      std_critic_clipfrac:                      0.0032635494
+      std_critic_loss:                          0.0008352084
+      std_entropy:                              0.01757949
+      std_grad_norm_pre_clip:                   1.7944696
+      std_loss:                                 0.0013080828
+  eval-batch-train:
+    avg_avg_episode_length:                     59.719288
+    avg_avg_return:                             0.403945
+    avg_benchmark_regret:                       0.58327204
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.040924072
+    proxy_corner:
+      avg_avg_return:                           0.03828464
+      avg_reward_per_step:                      0.0018081665
+  eval-batch-orig:
+    avg_avg_episode_length:                     65.89224
+    avg_avg_return:                             0.31489903
+    avg_benchmark_regret:                       0.6696125
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.025123596
+    proxy_corner:
+      avg_avg_return:                           0.31489903
+      avg_reward_per_step:                      0.025123596
+  eval-batch-shift:
+    avg_avg_episode_length:                     61.24128
+    avg_avg_return:                             0.37609196
+    avg_benchmark_regret:                       0.61141396
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.046180725
+    proxy_corner:
+      avg_avg_return:                           0.03080748
+      avg_reward_per_step:                      0.00046539307
+  eval-batch-tree:
+    avg_avg_episode_length:                     68.05794
+    avg_avg_return:                             0.28840178
+    avg_benchmark_regret:                       0.6962334
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.035888672
+    proxy_corner:
+      avg_avg_return:                           0.28840178
+      avg_reward_per_step:                      0.035888672
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        16777216
+    env-step-replay-before:                     8224768
+    env-step-used-before:                       16777216
+    ppo-update-before:                          10240
+    env-step-generate-before:                   8552448
+    env-step-all-after:                         16809984
+    env-step-replay-after:                      8257536
+    env-step-used-after:                        16809984
+    ppo-update-after:                           10260
+    env-step-generate-after:                    8552448
+  train-replay:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     70.61694
+    avg_avg_return:                             0.30797476
+    avg_reward_per_step:                        0.011230469
+    proxy_corner:
+      avg_avg_return:                           0.056410767
+      avg_reward_per_step:                      0.00076293945
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     70.61694
+    avg_avg_return:                             0.30797476
+    avg_reward_per_step:                        0.011230469
+    proxy_corner:
+      avg_avg_return:                           0.056410767
+      avg_reward_per_step:                      0.00076293945
+  perf:
+    env_steps_per_second:                       314203.364376888
+    ppo_updates_per_second:                     41.95223556950981
+    eval-batch-train-duration:                  0.516008467413485
+    eval-batch-orig-duration:                   0.4679393358528614
+    eval-batch-shift-duration:                  0.4960031285881996
+    eval-batch-tree-duration:                   0.45042569749057293
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    14.86377
+      cheese-corne_dist_wavg:                   20.378595
+      levels_cheese_is_in_corner_avg:           0.016357422
+      mouse-cheese_dist_finite_avg:             10.990967
+      mouse-cheese_dist_finite_wavg:            2.8998256
+      mouse-cheese_dist_solvable_avg:           10.990967
+      mouse-cheese_dist_solvable_wavg:          2.8998263
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.672081
+    visit_patterns:
+      avg_first_visit_time:                     138.02979
+      avg_last_visit_time:                      242.26196
+      num_replay_batches:                       252
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013284615
+    avg_actor_approxkl3:                        0.000974721
+    avg_actor_clipfrac:                         0.046868898
+    avg_actor_loss:                             -0.0009090096
+    avg_advantage:                              -0.0011791401
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.014179793
+    avg_entropy:                                0.83733064
+    avg_grad_norm_pre_clip:                     1.749997
+    avg_loss:                                   0.005343556
+    max:
+      max_actor_approxkl1:                      0.0027031805
+      max_actor_approxkl3:                      0.0016152463
+      max_actor_clipfrac:                       0.09643555
+      max_actor_loss:                           0.0009815991
+      max_advantage:                            0.5670587
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0156059265
+      max_entropy:                              0.8615704
+      max_grad_norm_pre_clip:                   3.7663407
+      max_loss:                                 0.0075579565
+    std:
+      std_actor_approxkl1:                      0.00080285757
+      std_actor_approxkl3:                      0.00047702034
+      std_actor_clipfrac:                       0.032962997
+      std_actor_loss:                           0.0011140784
+      std_advantage:                            0.16804905
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0007975647
+      std_entropy:                              0.017131912
+      std_grad_norm_pre_clip:                   0.9879722
+      std_loss:                                 0.0012246439
+  eval-batch-train:
+    avg_avg_episode_length:                     59.676254
+    avg_avg_return:                             0.40180808
+    avg_benchmark_regret:                       0.585409
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.045951843
+    proxy_corner:
+      avg_avg_return:                           0.04027026
+      avg_reward_per_step:                      0.0015716553
+  eval-batch-orig:
+    avg_avg_episode_length:                     66.44653
+    avg_avg_return:                             0.30550793
+    avg_benchmark_regret:                       0.6790036
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.028388977
+    proxy_corner:
+      avg_avg_return:                           0.30550793
+      avg_reward_per_step:                      0.028388977
+  eval-batch-shift:
+    avg_avg_episode_length:                     60.77125
+    avg_avg_return:                             0.3783311
+    avg_benchmark_regret:                       0.60917485
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.05206299
+    proxy_corner:
+      avg_avg_return:                           0.042339724
+      avg_reward_per_step:                      0.0007019043
+  eval-batch-tree:
+    avg_avg_episode_length:                     69.23581
+    avg_avg_return:                             0.26538792
+    avg_benchmark_regret:                       0.7192472
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.036514282
+    proxy_corner:
+      avg_avg_return:                           0.26538792
+      avg_reward_per_step:                      0.036514282
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        17825792
+    env-step-replay-before:                     8912896
+    env-step-used-before:                       17825792
+    ppo-update-before:                          10880
+    env-step-generate-before:                   8912896
+    env-step-all-after:                         17858560
+    env-step-replay-after:                      8945664
+    env-step-used-after:                        17858560
+    ppo-update-after:                           10900
+    env-step-generate-after:                    8912896
+  train-replay:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     72.20647
+    avg_avg_return:                             0.26953527
+    avg_reward_per_step:                        0.011657715
+    proxy_corner:
+      avg_avg_return:                           0.024927376
+      avg_reward_per_step:                      0.0006713867
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     72.20647
+    avg_avg_return:                             0.26953527
+    avg_reward_per_step:                        0.011657715
+    proxy_corner:
+      avg_avg_return:                           0.024927376
+      avg_reward_per_step:                      0.0006713867
+  perf:
+    env_steps_per_second:                       285502.4594086684
+    ppo_updates_per_second:                     42.14036621546324
+    eval-batch-train-duration:                  0.5279468763619661
+    eval-batch-orig-duration:                   0.4752108408138156
+    eval-batch-shift-duration:                  0.4930114168673754
+    eval-batch-tree-duration:                   0.4980888906866312
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    14.823242
+      cheese-corne_dist_wavg:                   6.9003835
+      levels_cheese_is_in_corner_avg:           0.017089844
+      mouse-cheese_dist_finite_avg:             11.170166
+      mouse-cheese_dist_finite_wavg:            4.7128572
+      mouse-cheese_dist_solvable_avg:           11.170166
+      mouse-cheese_dist_solvable_wavg:          4.7128577
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000008
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.68111396
+    visit_patterns:
+      avg_first_visit_time:                     146.50146
+      avg_last_visit_time:                      263.17603
+      num_replay_batches:                       273
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010028746
+    avg_actor_approxkl3:                        0.0010908721
+    avg_actor_clipfrac:                         0.05404663
+    avg_actor_loss:                             -0.0012270593
+    avg_advantage:                              -0.0063315835
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.014503143
+    avg_entropy:                                0.8578311
+    avg_grad_norm_pre_clip:                     1.6130927
+    avg_loss:                                   0.005166681
+    max:
+      max_actor_approxkl1:                      0.0019605209
+      max_actor_approxkl3:                      0.0018000287
+      max_actor_clipfrac:                       0.0970459
+      max_actor_loss:                           0.0004962981
+      max_advantage:                            0.57723486
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.015631188
+      max_entropy:                              0.8804636
+      max_grad_norm_pre_clip:                   3.27239
+      max_loss:                                 0.0068987855
+    std:
+      std_actor_approxkl1:                      0.00048031213
+      std_actor_approxkl3:                      0.00041808325
+      std_actor_clipfrac:                       0.026854355
+      std_actor_loss:                           0.0010485409
+      std_advantage:                            0.1705721
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0005137454
+      std_entropy:                              0.008322638
+      std_grad_norm_pre_clip:                   0.6600837
+      std_loss:                                 0.0010505491
+  eval-batch-train:
+    avg_avg_episode_length:                     58.915516
+    avg_avg_return:                             0.40615362
+    avg_benchmark_regret:                       0.5810635
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.048568726
+    proxy_corner:
+      avg_avg_return:                           0.04262597
+      avg_reward_per_step:                      0.001701355
+  eval-batch-orig:
+    avg_avg_episode_length:                     64.98248
+    avg_avg_return:                             0.33186382
+    avg_benchmark_regret:                       0.65264773
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.026817322
+    proxy_corner:
+      avg_avg_return:                           0.33186382
+      avg_reward_per_step:                      0.026817322
+  eval-batch-shift:
+    avg_avg_episode_length:                     59.717113
+    avg_avg_return:                             0.3980227
+    avg_benchmark_regret:                       0.58948326
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.05381012
+    proxy_corner:
+      avg_avg_return:                           0.04098437
+      avg_reward_per_step:                      0.0006866455
+  eval-batch-tree:
+    avg_avg_episode_length:                     67.5243
+    avg_avg_return:                             0.30533117
+    avg_benchmark_regret:                       0.67930406
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.036514282
+    proxy_corner:
+      avg_avg_return:                           0.30533117
+      avg_reward_per_step:                      0.036514282
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        18874368
+    env-step-replay-before:                     9633792
+    env-step-used-before:                       18874368
+    ppo-update-before:                          11520
+    env-step-generate-before:                   9240576
+    env-step-all-after:                         18907136
+    env-step-replay-after:                      9666560
+    env-step-used-after:                        18907136
+    ppo-update-after:                           11540
+    env-step-generate-after:                    9240576
+  train-replay:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     69.34313
+    avg_avg_return:                             0.33136642
+    avg_reward_per_step:                        0.011810303
+    proxy_corner:
+      avg_avg_return:                           0.04050323
+      avg_reward_per_step:                      0.00079345703
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     69.34313
+    avg_avg_return:                             0.33136642
+    avg_reward_per_step:                        0.011810303
+    proxy_corner:
+      avg_avg_return:                           0.04050323
+      avg_reward_per_step:                      0.00079345703
+  perf:
+    env_steps_per_second:                       258459.61716479572
+    ppo_updates_per_second:                     42.212578858513645
+    eval-batch-train-duration:                  0.6753974165767431
+    eval-batch-orig-duration:                   0.6362224044278264
+    eval-batch-shift-duration:                  0.6117703784257174
+    eval-batch-tree-duration:                   0.6089176964014769
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    14.838623
+      cheese-corne_dist_wavg:                   19.495766
+      levels_cheese_is_in_corner_avg:           0.017333984
+      mouse-cheese_dist_finite_avg:             11.385986
+      mouse-cheese_dist_finite_wavg:            2.9316688
+      mouse-cheese_dist_solvable_avg:           11.385986
+      mouse-cheese_dist_solvable_wavg:          2.9316688
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000008
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201682
+    scoring:
+      avg_scores:                               0.670726
+    visit_patterns:
+      avg_first_visit_time:                     155.56738
+      avg_last_visit_time:                      285.19727
+      num_replay_batches:                       295
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0026048084
+    avg_actor_approxkl3:                        0.0023433752
+    avg_actor_clipfrac:                         0.12484741
+    avg_actor_loss:                             0.0015281569
+    avg_advantage:                              0.007770483
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.014913804
+    avg_entropy:                                0.86754096
+    avg_grad_norm_pre_clip:                     3.824665
+    avg_loss:                                   0.008117518
+    max:
+      max_actor_approxkl1:                      0.007396781
+      max_actor_approxkl3:                      0.004351809
+      max_actor_clipfrac:                       0.27075195
+      max_actor_loss:                           0.005271107
+      max_advantage:                            0.6034551
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.01719062
+      max_entropy:                              0.8925354
+      max_grad_norm_pre_clip:                   7.4862633
+      max_loss:                                 0.011949435
+    std:
+      std_actor_approxkl1:                      0.0018772606
+      std_actor_approxkl3:                      0.001295357
+      std_actor_clipfrac:                       0.0784197
+      std_actor_loss:                           0.0019107636
+      std_advantage:                            0.17291531
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0008608457
+      std_entropy:                              0.013767921
+      std_grad_norm_pre_clip:                   2.086411
+      std_loss:                                 0.0019835192
+  eval-batch-train:
+    avg_avg_episode_length:                     57.719044
+    avg_avg_return:                             0.43017852
+    avg_benchmark_regret:                       0.55703855
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.049705505
+    proxy_corner:
+      avg_avg_return:                           0.04669811
+      avg_reward_per_step:                      0.0017623901
+  eval-batch-orig:
+    avg_avg_episode_length:                     64.55426
+    avg_avg_return:                             0.3365056
+    avg_benchmark_regret:                       0.64800596
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.028862
+    proxy_corner:
+      avg_avg_return:                           0.3365056
+      avg_reward_per_step:                      0.028862
+  eval-batch-shift:
+    avg_avg_episode_length:                     60.3897
+    avg_avg_return:                             0.38708895
+    avg_benchmark_regret:                       0.600417
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.05268097
+    proxy_corner:
+      avg_avg_return:                           0.043668788
+      avg_reward_per_step:                      0.0006790161
+  eval-batch-tree:
+    avg_avg_episode_length:                     67.329796
+    avg_avg_return:                             0.2971515
+    avg_benchmark_regret:                       0.68748367
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.03742218
+    proxy_corner:
+      avg_avg_return:                           0.2971515
+      avg_reward_per_step:                      0.03742218
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        19922944
+    env-step-replay-before:                     10125312
+    env-step-used-before:                       19922944
+    ppo-update-before:                          12160
+    env-step-generate-before:                   9797632
+    env-step-all-after:                         19955712
+    env-step-replay-after:                      10158080
+    env-step-used-after:                        19955712
+    ppo-update-after:                           12180
+    env-step-generate-after:                    9797632
+  train-replay:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     71.35362
+    avg_avg_return:                             0.30248502
+    avg_reward_per_step:                        0.010925293
+    proxy_corner:
+      avg_avg_return:                           0.048973836
+      avg_reward_per_step:                      0.0010375977
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     71.35362
+    avg_avg_return:                             0.30248502
+    avg_reward_per_step:                        0.010925293
+    proxy_corner:
+      avg_avg_return:                           0.048973836
+      avg_reward_per_step:                      0.0010375977
+  perf:
+    env_steps_per_second:                       258856.02538657418
+    ppo_updates_per_second:                     41.56238891290419
+    eval-batch-train-duration:                  0.5167679144069552
+    eval-batch-orig-duration:                   0.45168368238955736
+    eval-batch-shift-duration:                  0.4376343749463558
+    eval-batch-tree-duration:                   0.28493770584464073
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    14.786865
+      cheese-corne_dist_wavg:                   20.355753
+      levels_cheese_is_in_corner_avg:           0.017333984
+      mouse-cheese_dist_finite_avg:             11.603027
+      mouse-cheese_dist_finite_wavg:            2.9553027
+      mouse-cheese_dist_solvable_avg:           11.603027
+      mouse-cheese_dist_solvable_wavg:          2.955303
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.6924838
+    visit_patterns:
+      avg_first_visit_time:                     168.94897
+      avg_last_visit_time:                      300.19165
+      num_replay_batches:                       310
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0016387989
+    avg_actor_approxkl3:                        0.0019268057
+    avg_actor_clipfrac:                         0.10583496
+    avg_actor_loss:                             0.0006879222
+    avg_advantage:                              -0.005495921
+    avg_critic_clipfrac:                        0.00028686525
+    avg_critic_loss:                            0.0143584935
+    avg_entropy:                                0.8757057
+    avg_grad_norm_pre_clip:                     2.7913167
+    avg_loss:                                   0.0069914646
+    max:
+      max_actor_approxkl1:                      0.005081781
+      max_actor_approxkl3:                      0.0045576813
+      max_actor_clipfrac:                       0.27062988
+      max_actor_loss:                           0.0042766184
+      max_advantage:                            0.5385461
+      max_critic_clipfrac:                      0.0041503906
+      max_critic_loss:                          0.016188491
+      max_entropy:                              0.9064729
+      max_grad_norm_pre_clip:                   5.9337893
+      max_loss:                                 0.011245558
+    std:
+      std_actor_approxkl1:                      0.0013343374
+      std_actor_approxkl3:                      0.0011776262
+      std_actor_clipfrac:                       0.071160205
+      std_actor_loss:                           0.0019263246
+      std_advantage:                            0.17079327
+      std_critic_clipfrac:                      0.00094182196
+      std_critic_loss:                          0.0009275231
+      std_entropy:                              0.012999118
+      std_grad_norm_pre_clip:                   1.5918528
+      std_loss:                                 0.002224345
+  eval-batch-train:
+    avg_avg_episode_length:                     57.8988
+    avg_avg_return:                             0.42280686
+    avg_benchmark_regret:                       0.5644102
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.048919678
+    proxy_corner:
+      avg_avg_return:                           0.03834667
+      avg_reward_per_step:                      0.0021438599
+  eval-batch-orig:
+    avg_avg_episode_length:                     63.168602
+    avg_avg_return:                             0.35734916
+    avg_benchmark_regret:                       0.6271624
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.027366638
+    proxy_corner:
+      avg_avg_return:                           0.35734916
+      avg_reward_per_step:                      0.027366638
+  eval-batch-shift:
+    avg_avg_episode_length:                     59.498764
+    avg_avg_return:                             0.40757883
+    avg_benchmark_regret:                       0.5799272
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.055145264
+    proxy_corner:
+      avg_avg_return:                           0.04002756
+      avg_reward_per_step:                      0.0005569458
+  eval-batch-tree:
+    avg_avg_episode_length:                     65.22572
+    avg_avg_return:                             0.33904457
+    avg_benchmark_regret:                       0.64559066
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.037902832
+    proxy_corner:
+      avg_avg_return:                           0.33904457
+      avg_reward_per_step:                      0.037902832
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        20971520
+    env-step-replay-before:                     10682368
+    env-step-used-before:                       20971520
+    ppo-update-before:                          12800
+    env-step-generate-before:                   10289152
+    env-step-all-after:                         21004288
+    env-step-replay-after:                      10682368
+    env-step-used-after:                        21004288
+    ppo-update-after:                           12820
+    env-step-generate-after:                    10321920
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     64.92886
+    avg_avg_return:                             0.29109913
+    avg_reward_per_step:                        0.0362854
+    proxy_corner:
+      avg_avg_return:                           0.028252598
+      avg_reward_per_step:                      0.0005493164
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     64.92886
+    avg_avg_return:                             0.29109913
+    avg_reward_per_step:                        0.0362854
+    proxy_corner:
+      avg_avg_return:                           0.028252598
+      avg_reward_per_step:                      0.0005493164
+  perf:
+    env_steps_per_second:                       293442.1949953973
+    ppo_updates_per_second:                     42.464391339560805
+    eval-batch-train-duration:                  0.5271955449134111
+    eval-batch-orig-duration:                   0.4468831429257989
+    eval-batch-shift-duration:                  0.4695670986548066
+    eval-batch-tree-duration:                   0.47253767866641283
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    14.809082
+      cheese-corne_dist_wavg:                   6.891887
+      levels_cheese_is_in_corner_avg:           0.018798828
+      mouse-cheese_dist_finite_avg:             11.8063965
+      mouse-cheese_dist_finite_wavg:            4.774996
+      mouse-cheese_dist_solvable_avg:           11.8063965
+      mouse-cheese_dist_solvable_wavg:          4.7749963
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000008
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016798
+    scoring:
+      avg_scores:                               0.6814457
+    visit_patterns:
+      avg_first_visit_time:                     180.5437
+      avg_last_visit_time:                      316.2871
+      num_replay_batches:                       326
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001436131
+    avg_actor_approxkl3:                        0.0014432719
+    avg_actor_clipfrac:                         0.07632446
+    avg_actor_loss:                             -4.5166165e-05
+    avg_advantage:                              -0.008015514
+    avg_critic_clipfrac:                        2.4414063e-05
+    avg_critic_loss:                            0.016762957
+    avg_entropy:                                0.85430986
+    avg_grad_norm_pre_clip:                     2.4446912
+    avg_loss:                                   0.007482002
+    max:
+      max_actor_approxkl1:                      0.0034326406
+      max_actor_approxkl3:                      0.0032030526
+      max_actor_clipfrac:                       0.21008301
+      max_actor_loss:                           0.0024755448
+      max_advantage:                            0.55825573
+      max_critic_clipfrac:                      0.00048828125
+      max_critic_loss:                          0.019584656
+      max_entropy:                              0.87741876
+      max_grad_norm_pre_clip:                   6.4790096
+      max_loss:                                 0.010675333
+    std:
+      std_actor_approxkl1:                      0.0010319324
+      std_actor_approxkl3:                      0.0009529095
+      std_actor_clipfrac:                       0.06583899
+      std_actor_loss:                           0.0012372241
+      std_advantage:                            0.1824606
+      std_critic_clipfrac:                      0.000106418425
+      std_critic_loss:                          0.001602162
+      std_entropy:                              0.014027962
+      std_grad_norm_pre_clip:                   1.6748207
+      std_loss:                                 0.0017800869
+  eval-batch-train:
+    avg_avg_episode_length:                     57.624176
+    avg_avg_return:                             0.42814678
+    avg_benchmark_regret:                       0.5590703
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.049613953
+    proxy_corner:
+      avg_avg_return:                           0.032687172
+      avg_reward_per_step:                      0.0016555786
+  eval-batch-orig:
+    avg_avg_episode_length:                     63.18035
+    avg_avg_return:                             0.36629343
+    avg_benchmark_regret:                       0.6182181
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.026763916
+    proxy_corner:
+      avg_avg_return:                           0.36629343
+      avg_reward_per_step:                      0.026763916
+  eval-batch-shift:
+    avg_avg_episode_length:                     57.815437
+    avg_avg_return:                             0.42071134
+    avg_benchmark_regret:                       0.56679463
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.05456543
+    proxy_corner:
+      avg_avg_return:                           0.03514356
+      avg_reward_per_step:                      0.00074005127
+  eval-batch-tree:
+    avg_avg_episode_length:                     64.85611
+    avg_avg_return:                             0.35331807
+    avg_benchmark_regret:                       0.63131714
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.034851074
+    proxy_corner:
+      avg_avg_return:                           0.35331807
+      avg_reward_per_step:                      0.034851074
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        22020096
+    env-step-replay-before:                     11141120
+    env-step-used-before:                       22020096
+    ppo-update-before:                          13440
+    env-step-generate-before:                   10878976
+    env-step-all-after:                         22052864
+    env-step-replay-after:                      11173888
+    env-step-used-after:                        22052864
+    ppo-update-after:                           13460
+    env-step-generate-after:                    10878976
+  train-replay:
+    shift_proportion:                           0.96484375
+    avg_avg_episode_length:                     71.31459
+    avg_avg_return:                             0.33877757
+    avg_reward_per_step:                        0.009979248
+    proxy_corner:
+      avg_avg_return:                           0.038732704
+      avg_reward_per_step:                      0.0008239746
+  train-all:
+    shift_proportion:                           0.96484375
+    avg_avg_episode_length:                     71.31459
+    avg_avg_return:                             0.33877757
+    avg_reward_per_step:                        0.009979248
+    proxy_corner:
+      avg_avg_return:                           0.038732704
+      avg_reward_per_step:                      0.0008239746
+  perf:
+    env_steps_per_second:                       260342.9999682936
+    ppo_updates_per_second:                     42.292711346441166
+    eval-batch-train-duration:                  0.6737170731648803
+    eval-batch-orig-duration:                   0.618005407974124
+    eval-batch-shift-duration:                  0.6168459160253406
+    eval-batch-tree-duration:                   0.660314517095685
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    14.738281
+      cheese-corne_dist_wavg:                   8.671373
+      levels_cheese_is_in_corner_avg:           0.019042969
+      mouse-cheese_dist_finite_avg:             11.968506
+      mouse-cheese_dist_finite_wavg:            5.6919975
+      mouse-cheese_dist_solvable_avg:           11.968506
+      mouse-cheese_dist_solvable_wavg:          5.691998
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.6661658
+    visit_patterns:
+      avg_first_visit_time:                     194.27441
+      avg_last_visit_time:                      331.20142
+      num_replay_batches:                       341
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012702842
+    avg_actor_approxkl3:                        0.0014313065
+    avg_actor_clipfrac:                         0.070043944
+    avg_actor_loss:                             -0.0007029623
+    avg_advantage:                              0.0030471773
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.013522886
+    avg_entropy:                                0.9024831
+    avg_grad_norm_pre_clip:                     2.482066
+    avg_loss:                                   0.005155998
+    max:
+      max_actor_approxkl1:                      0.0035059033
+      max_actor_approxkl3:                      0.0024504878
+      max_actor_clipfrac:                       0.1619873
+      max_actor_loss:                           0.00065617263
+      max_advantage:                            0.53030074
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.014990588
+      max_entropy:                              0.9241176
+      max_grad_norm_pre_clip:                   4.263494
+      max_loss:                                 0.006723069
+    std:
+      std_actor_approxkl1:                      0.0009390366
+      std_actor_approxkl3:                      0.000627116
+      std_actor_clipfrac:                       0.04222925
+      std_actor_loss:                           0.0008429227
+      std_advantage:                            0.16520712
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.000804316
+      std_entropy:                              0.013701858
+      std_grad_norm_pre_clip:                   1.0029463
+      std_loss:                                 0.0010591781
+  eval-batch-train:
+    avg_avg_episode_length:                     56.585922
+    avg_avg_return:                             0.44852865
+    avg_benchmark_regret:                       0.5386884
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.051849365
+    proxy_corner:
+      avg_avg_return:                           0.039391227
+      avg_reward_per_step:                      0.002029419
+  eval-batch-orig:
+    avg_avg_episode_length:                     62.13243
+    avg_avg_return:                             0.38820052
+    avg_benchmark_regret:                       0.59631103
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.026924133
+    proxy_corner:
+      avg_avg_return:                           0.38820052
+      avg_reward_per_step:                      0.026924133
+  eval-batch-shift:
+    avg_avg_episode_length:                     57.55335
+    avg_avg_return:                             0.4248607
+    avg_benchmark_regret:                       0.5626453
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.05986023
+    proxy_corner:
+      avg_avg_return:                           0.037170157
+      avg_reward_per_step:                      0.0006942749
+  eval-batch-tree:
+    avg_avg_episode_length:                     63.839466
+    avg_avg_return:                             0.38418388
+    avg_benchmark_regret:                       0.60045135
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0395813
+    proxy_corner:
+      avg_avg_return:                           0.38418388
+      avg_reward_per_step:                      0.0395813
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        23068672
+    env-step-replay-before:                     11665408
+    env-step-used-before:                       23068672
+    ppo-update-before:                          14080
+    env-step-generate-before:                   11403264
+    env-step-all-after:                         23101440
+    env-step-replay-after:                      11698176
+    env-step-used-after:                        23101440
+    ppo-update-after:                           14100
+    env-step-generate-after:                    11403264
+  train-replay:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     72.34619
+    avg_avg_return:                             0.31569687
+    avg_reward_per_step:                        0.008728027
+    proxy_corner:
+      avg_avg_return:                           0.04902053
+      avg_reward_per_step:                      0.0010681152
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     72.34619
+    avg_avg_return:                             0.31569687
+    avg_reward_per_step:                        0.008728027
+    proxy_corner:
+      avg_avg_return:                           0.04902053
+      avg_reward_per_step:                      0.0010681152
+  perf:
+    env_steps_per_second:                       243755.46780486
+    ppo_updates_per_second:                     42.39062907414553
+    eval-batch-train-duration:                  0.6781920939683914
+    eval-batch-orig-duration:                   0.6185625204816461
+    eval-batch-shift-duration:                  0.6187948659062386
+    eval-batch-tree-duration:                   0.6379386363551021
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    14.755127
+      cheese-corne_dist_wavg:                   8.68468
+      levels_cheese_is_in_corner_avg:           0.018310547
+      mouse-cheese_dist_finite_avg:             12.138916
+      mouse-cheese_dist_finite_wavg:            5.711108
+      mouse-cheese_dist_solvable_avg:           12.138916
+      mouse-cheese_dist_solvable_wavg:          5.7111087
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016816
+    scoring:
+      avg_scores:                               0.66085
+    visit_patterns:
+      avg_first_visit_time:                     204.16406
+      avg_last_visit_time:                      347.25952
+      num_replay_batches:                       357
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0019307908
+    avg_actor_approxkl3:                        0.0017949833
+    avg_actor_clipfrac:                         0.10217285
+    avg_actor_loss:                             -3.0385703e-05
+    avg_advantage:                              -0.005449693
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0132006435
+    avg_entropy:                                0.9068286
+    avg_grad_norm_pre_clip:                     2.788691
+    avg_loss:                                   0.0056631076
+    max:
+      max_actor_approxkl1:                      0.0047092317
+      max_actor_approxkl3:                      0.0044526365
+      max_actor_clipfrac:                       0.2923584
+      max_actor_loss:                           0.0035586357
+      max_advantage:                            0.5744481
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.014148261
+      max_entropy:                              0.93200964
+      max_grad_norm_pre_clip:                   6.6092715
+      max_loss:                                 0.008957391
+    std:
+      std_actor_approxkl1:                      0.0013168623
+      std_actor_approxkl3:                      0.001192541
+      std_actor_clipfrac:                       0.08127957
+      std_actor_loss:                           0.0016949343
+      std_advantage:                            0.1629
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00059444114
+      std_entropy:                              0.012630799
+      std_grad_norm_pre_clip:                   1.7607975
+      std_loss:                                 0.0017600318
+  eval-batch-train:
+    avg_avg_episode_length:                     56.548527
+    avg_avg_return:                             0.4396131
+    avg_benchmark_regret:                       0.54760396
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.051879883
+    proxy_corner:
+      avg_avg_return:                           0.040478133
+      avg_reward_per_step:                      0.0023040771
+  eval-batch-orig:
+    avg_avg_episode_length:                     61.630074
+    avg_avg_return:                             0.39951962
+    avg_benchmark_regret:                       0.58499193
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.02596283
+    proxy_corner:
+      avg_avg_return:                           0.39951962
+      avg_reward_per_step:                      0.02596283
+  eval-batch-shift:
+    avg_avg_episode_length:                     56.70394
+    avg_avg_return:                             0.43637317
+    avg_benchmark_regret:                       0.55113274
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.060173035
+    proxy_corner:
+      avg_avg_return:                           0.035367478
+      avg_reward_per_step:                      0.00074768066
+  eval-batch-tree:
+    avg_avg_episode_length:                     62.77742
+    avg_avg_return:                             0.39571643
+    avg_benchmark_regret:                       0.58891875
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0389328
+    proxy_corner:
+      avg_avg_return:                           0.39571643
+      avg_reward_per_step:                      0.0389328
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        24117248
+    env-step-replay-before:                     12320768
+    env-step-used-before:                       24117248
+    ppo-update-before:                          14720
+    env-step-generate-before:                   11796480
+    env-step-all-after:                         24150016
+    env-step-replay-after:                      12353536
+    env-step-used-after:                        24150016
+    ppo-update-after:                           14740
+    env-step-generate-after:                    11796480
+  train-replay:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     70.67812
+    avg_avg_return:                             0.32732517
+    avg_reward_per_step:                        0.010284424
+    proxy_corner:
+      avg_avg_return:                           0.031477347
+      avg_reward_per_step:                      0.0005493164
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     70.67812
+    avg_avg_return:                             0.32732517
+    avg_reward_per_step:                        0.010284424
+    proxy_corner:
+      avg_avg_return:                           0.031477347
+      avg_reward_per_step:                      0.0005493164
+  perf:
+    env_steps_per_second:                       322000.0274080772
+    ppo_updates_per_second:                     44.93486294152498
+    eval-batch-train-duration:                  0.5205431841313839
+    eval-batch-orig-duration:                   0.48303542099893093
+    eval-batch-shift-duration:                  0.4941526520997286
+    eval-batch-tree-duration:                   0.5014325873926282
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    14.8498535
+      cheese-corne_dist_wavg:                   6.8833847
+      levels_cheese_is_in_corner_avg:           0.018310547
+      mouse-cheese_dist_finite_avg:             12.314453
+      mouse-cheese_dist_finite_wavg:            4.8252687
+      mouse-cheese_dist_solvable_avg:           12.314453
+      mouse-cheese_dist_solvable_wavg:          4.825269
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.65585524
+    visit_patterns:
+      avg_first_visit_time:                     215.49536
+      avg_last_visit_time:                      367.22705
+      num_replay_batches:                       377
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001326871
+    avg_actor_approxkl3:                        0.0013514754
+    avg_actor_clipfrac:                         0.06835327
+    avg_actor_loss:                             -0.0002522778
+    avg_advantage:                              0.009401066
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.014028877
+    avg_entropy:                                0.87878716
+    avg_grad_norm_pre_clip:                     2.457892
+    avg_loss:                                   0.0058833733
+    max:
+      max_actor_approxkl1:                      0.0027911197
+      max_actor_approxkl3:                      0.0027707922
+      max_actor_clipfrac:                       0.17883301
+      max_actor_loss:                           0.001954019
+      max_advantage:                            0.66209966
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.015659172
+      max_entropy:                              0.89744174
+      max_grad_norm_pre_clip:                   5.559294
+      max_loss:                                 0.008463854
+    std:
+      std_actor_approxkl1:                      0.00080576574
+      std_actor_approxkl3:                      0.0007173039
+      std_actor_clipfrac:                       0.050309952
+      std_actor_loss:                           0.0011982503
+      std_advantage:                            0.16825499
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0008045814
+      std_entropy:                              0.012582414
+      std_grad_norm_pre_clip:                   1.3026778
+      std_loss:                                 0.0013559745
+  eval-batch-train:
+    avg_avg_episode_length:                     56.762505
+    avg_avg_return:                             0.4396967
+    avg_benchmark_regret:                       0.5475204
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.05103302
+    proxy_corner:
+      avg_avg_return:                           0.040192828
+      avg_reward_per_step:                      0.0020065308
+  eval-batch-orig:
+    avg_avg_episode_length:                     60.18087
+    avg_avg_return:                             0.41586155
+    avg_benchmark_regret:                       0.56865
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.027915955
+    proxy_corner:
+      avg_avg_return:                           0.41586155
+      avg_reward_per_step:                      0.027915955
+  eval-batch-shift:
+    avg_avg_episode_length:                     56.247154
+    avg_avg_return:                             0.43890703
+    avg_benchmark_regret:                       0.54859895
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.061553955
+    proxy_corner:
+      avg_avg_return:                           0.03833105
+      avg_reward_per_step:                      0.0008239746
+  eval-batch-tree:
+    avg_avg_episode_length:                     61.93569
+    avg_avg_return:                             0.40791202
+    avg_benchmark_regret:                       0.57672316
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.042625427
+    proxy_corner:
+      avg_avg_return:                           0.40791202
+      avg_reward_per_step:                      0.042625427
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        25165824
+    env-step-replay-before:                     13041664
+    env-step-used-before:                       25165824
+    ppo-update-before:                          15360
+    env-step-generate-before:                   12124160
+    env-step-all-after:                         25198592
+    env-step-replay-after:                      13041664
+    env-step-used-after:                        25198592
+    ppo-update-after:                           15380
+    env-step-generate-after:                    12156928
+  train-generate:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     59.672935
+    avg_avg_return:                             0.38179773
+    avg_reward_per_step:                        0.044952393
+    proxy_corner:
+      avg_avg_return:                           0.05164221
+      avg_reward_per_step:                      0.00079345703
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     59.672935
+    avg_avg_return:                             0.38179773
+    avg_reward_per_step:                        0.044952393
+    proxy_corner:
+      avg_avg_return:                           0.05164221
+      avg_reward_per_step:                      0.00079345703
+  perf:
+    env_steps_per_second:                       301949.0007046003
+    ppo_updates_per_second:                     41.596040629563106
+    eval-batch-train-duration:                  0.5404212642461061
+    eval-batch-orig-duration:                   0.4722301699221134
+    eval-batch-shift-duration:                  0.4728131629526615
+    eval-batch-tree-duration:                   0.4502468490973115
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    14.911133
+      cheese-corne_dist_wavg:                   28.475952
+      levels_cheese_is_in_corner_avg:           0.017822266
+      mouse-cheese_dist_finite_avg:             12.486572
+      mouse-cheese_dist_finite_wavg:            3.9500496
+      mouse-cheese_dist_solvable_avg:           12.486572
+      mouse-cheese_dist_solvable_wavg:          3.9500496
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.6478828
+    visit_patterns:
+      avg_first_visit_time:                     228.40088
+      avg_last_visit_time:                      388.29883
+      num_replay_batches:                       398
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013717697
+    avg_actor_approxkl3:                        0.0011971472
+    avg_actor_clipfrac:                         0.060803223
+    avg_actor_loss:                             -0.00094070396
+    avg_advantage:                              0.009697355
+    avg_critic_clipfrac:                        7.324219e-05
+    avg_critic_loss:                            0.016141534
+    avg_entropy:                                0.8643729
+    avg_grad_norm_pre_clip:                     1.7671107
+    avg_loss:                                   0.00626569
+    max:
+      max_actor_approxkl1:                      0.0027026553
+      max_actor_approxkl3:                      0.0023034695
+      max_actor_clipfrac:                       0.13964844
+      max_actor_loss:                           0.0019393414
+      max_advantage:                            0.5009419
+      max_critic_clipfrac:                      0.0014648438
+      max_critic_loss:                          0.017552307
+      max_entropy:                              0.8877028
+      max_grad_norm_pre_clip:                   4.104419
+      max_loss:                                 0.009352929
+    std:
+      std_actor_approxkl1:                      0.0007198108
+      std_actor_approxkl3:                      0.00064888585
+      std_actor_clipfrac:                       0.045055084
+      std_actor_loss:                           0.0011851923
+      std_advantage:                            0.1805715
+      std_critic_clipfrac:                      0.00031925528
+      std_critic_loss:                          0.0007716183
+      std_entropy:                              0.015726019
+      std_grad_norm_pre_clip:                   0.9198114
+      std_loss:                                 0.001280234
+  eval-batch-train:
+    avg_avg_episode_length:                     53.993546
+    avg_avg_return:                             0.4803542
+    avg_benchmark_regret:                       0.5068629
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.05496216
+    proxy_corner:
+      avg_avg_return:                           0.042697284
+      avg_reward_per_step:                      0.0021362305
+  eval-batch-orig:
+    avg_avg_episode_length:                     59.632347
+    avg_avg_return:                             0.41436827
+    avg_benchmark_regret:                       0.5701433
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.031852722
+    proxy_corner:
+      avg_avg_return:                           0.41436827
+      avg_reward_per_step:                      0.031852722
+  eval-batch-shift:
+    avg_avg_episode_length:                     54.739902
+    avg_avg_return:                             0.4732006
+    avg_benchmark_regret:                       0.51430535
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.0622406
+    proxy_corner:
+      avg_avg_return:                           0.04195195
+      avg_reward_per_step:                      0.0009918213
+  eval-batch-tree:
+    avg_avg_episode_length:                     61.26947
+    avg_avg_return:                             0.4111383
+    avg_benchmark_regret:                       0.5734969
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.044403076
+    proxy_corner:
+      avg_avg_return:                           0.4111383
+      avg_reward_per_step:                      0.044403076
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        26214400
+    env-step-replay-before:                     13402112
+    env-step-used-before:                       26214400
+    ppo-update-before:                          16000
+    env-step-generate-before:                   12812288
+    env-step-all-after:                         26247168
+    env-step-replay-after:                      13434880
+    env-step-used-after:                        26247168
+    ppo-update-after:                           16020
+    env-step-generate-after:                    12812288
+  train-replay:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     69.84467
+    avg_avg_return:                             0.336903
+    avg_reward_per_step:                        0.010864258
+    proxy_corner:
+      avg_avg_return:                           0.047102038
+      avg_reward_per_step:                      0.00076293945
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     69.84467
+    avg_avg_return:                             0.336903
+    avg_reward_per_step:                        0.010864258
+    proxy_corner:
+      avg_avg_return:                           0.047102038
+      avg_reward_per_step:                      0.00076293945
+  perf:
+    env_steps_per_second:                       319946.6871085073
+    ppo_updates_per_second:                     41.85107659143068
+    eval-batch-train-duration:                  0.5443140724673867
+    eval-batch-orig-duration:                   0.4878175565972924
+    eval-batch-shift-duration:                  0.4759562909603119
+    eval-batch-tree-duration:                   0.4827510518953204
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    14.9453125
+      cheese-corne_dist_wavg:                   11.399287
+      levels_cheese_is_in_corner_avg:           0.018066406
+      mouse-cheese_dist_finite_avg:             12.614258
+      mouse-cheese_dist_finite_wavg:            4.8563066
+      mouse-cheese_dist_solvable_avg:           12.614258
+      mouse-cheese_dist_solvable_wavg:          4.856307
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.6537157
+    visit_patterns:
+      avg_first_visit_time:                     242.2373
+      avg_last_visit_time:                      400.26733
+      num_replay_batches:                       410
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0017410206
+    avg_actor_approxkl3:                        0.0015399848
+    avg_actor_clipfrac:                         0.0859314
+    avg_actor_loss:                             -0.00033655838
+    avg_advantage:                              -0.0042701517
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0141546605
+    avg_entropy:                                0.84464216
+    avg_grad_norm_pre_clip:                     2.878803
+    avg_loss:                                   0.00589613
+    max:
+      max_actor_approxkl1:                      0.0034671908
+      max_actor_approxkl3:                      0.002649564
+      max_actor_clipfrac:                       0.16247559
+      max_actor_loss:                           0.0010814816
+      max_advantage:                            0.6716332
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.016126588
+      max_entropy:                              0.86088103
+      max_grad_norm_pre_clip:                   6.5823026
+      max_loss:                                 0.0076971017
+    std:
+      std_actor_approxkl1:                      0.0009496229
+      std_actor_approxkl3:                      0.0006983984
+      std_actor_clipfrac:                       0.04868547
+      std_actor_loss:                           0.00092976145
+      std_advantage:                            0.16863236
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00093584234
+      std_entropy:                              0.009938829
+      std_grad_norm_pre_clip:                   1.4600996
+      std_loss:                                 0.0010376179
+  eval-batch-train:
+    avg_avg_episode_length:                     53.51706
+    avg_avg_return:                             0.47912017
+    avg_benchmark_regret:                       0.5080969
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.061439514
+    proxy_corner:
+      avg_avg_return:                           0.04468569
+      avg_reward_per_step:                      0.0027160645
+  eval-batch-orig:
+    avg_avg_episode_length:                     59.024223
+    avg_avg_return:                             0.4239124
+    avg_benchmark_regret:                       0.5605991
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.034950256
+    proxy_corner:
+      avg_avg_return:                           0.4239124
+      avg_reward_per_step:                      0.034950256
+  eval-batch-shift:
+    avg_avg_episode_length:                     53.90219
+    avg_avg_return:                             0.4806078
+    avg_benchmark_regret:                       0.5068981
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.06851959
+    proxy_corner:
+      avg_avg_return:                           0.04488115
+      avg_reward_per_step:                      0.0010147095
+  eval-batch-tree:
+    avg_avg_episode_length:                     60.945076
+    avg_avg_return:                             0.4032452
+    avg_benchmark_regret:                       0.58139
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.04866028
+    proxy_corner:
+      avg_avg_return:                           0.4032452
+      avg_reward_per_step:                      0.04866028
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        27262976
+    env-step-replay-before:                     14024704
+    env-step-used-before:                       27262976
+    ppo-update-before:                          16640
+    env-step-generate-before:                   13238272
+    env-step-all-after:                         27295744
+    env-step-replay-after:                      14057472
+    env-step-used-after:                        27295744
+    ppo-update-after:                           16660
+    env-step-generate-after:                    13238272
+  train-replay:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     68.89962
+    avg_avg_return:                             0.37371516
+    avg_reward_per_step:                        0.01071167
+    proxy_corner:
+      avg_avg_return:                           0.041819006
+      avg_reward_per_step:                      0.0007324219
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     68.89962
+    avg_avg_return:                             0.37371516
+    avg_reward_per_step:                        0.01071167
+    proxy_corner:
+      avg_avg_return:                           0.041819006
+      avg_reward_per_step:                      0.0007324219
+  perf:
+    env_steps_per_second:                       319616.27202821535
+    ppo_updates_per_second:                     42.6157031550599
+    eval-batch-train-duration:                  0.5453445045277476
+    eval-batch-orig-duration:                   0.488936903886497
+    eval-batch-shift-duration:                  0.48754930682480335
+    eval-batch-tree-duration:                   0.48793565947562456
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    15.044189
+      cheese-corne_dist_wavg:                   21.303234
+      levels_cheese_is_in_corner_avg:           0.018554688
+      mouse-cheese_dist_finite_avg:             12.804932
+      mouse-cheese_dist_finite_wavg:            6.6841
+      mouse-cheese_dist_solvable_avg:           12.804932
+      mouse-cheese_dist_solvable_wavg:          6.684101
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.6517221
+    visit_patterns:
+      avg_first_visit_time:                     254.8628
+      avg_last_visit_time:                      419.23975
+      num_replay_batches:                       429
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014025326
+    avg_actor_approxkl3:                        0.0012227326
+    avg_actor_clipfrac:                         0.05998535
+    avg_actor_loss:                             -0.0012710951
+    avg_advantage:                              0.0026075346
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.012596779
+    avg_entropy:                                0.8807789
+    avg_grad_norm_pre_clip:                     2.320173
+    avg_loss:                                   0.0041465145
+    max:
+      max_actor_approxkl1:                      0.0029555904
+      max_actor_approxkl3:                      0.002324379
+      max_actor_clipfrac:                       0.13635254
+      max_actor_loss:                           0.00016412139
+      max_advantage:                            0.524211
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.013683956
+      max_entropy:                              0.90184003
+      max_grad_norm_pre_clip:                   4.0401926
+      max_loss:                                 0.005865621
+    std:
+      std_actor_approxkl1:                      0.0008415758
+      std_actor_approxkl3:                      0.00055425725
+      std_actor_clipfrac:                       0.036574606
+      std_actor_loss:                           0.0010692014
+      std_advantage:                            0.15931906
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00065738906
+      std_entropy:                              0.011451837
+      std_grad_norm_pre_clip:                   0.9789213
+      std_loss:                                 0.0011355021
+  eval-batch-train:
+    avg_avg_episode_length:                     53.814426
+    avg_avg_return:                             0.48009998
+    avg_benchmark_regret:                       0.50711703
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.058532715
+    proxy_corner:
+      avg_avg_return:                           0.03821205
+      avg_reward_per_step:                      0.0023117065
+  eval-batch-orig:
+    avg_avg_episode_length:                     59.18306
+    avg_avg_return:                             0.42695928
+    avg_benchmark_regret:                       0.5575523
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.03252411
+    proxy_corner:
+      avg_avg_return:                           0.42695928
+      avg_reward_per_step:                      0.03252411
+  eval-batch-shift:
+    avg_avg_episode_length:                     53.758926
+    avg_avg_return:                             0.48116943
+    avg_benchmark_regret:                       0.5063365
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.06535339
+    proxy_corner:
+      avg_avg_return:                           0.034395453
+      avg_reward_per_step:                      0.00089263916
+  eval-batch-tree:
+    avg_avg_episode_length:                     60.519493
+    avg_avg_return:                             0.4243002
+    avg_benchmark_regret:                       0.560335
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.046836853
+    proxy_corner:
+      avg_avg_return:                           0.4243002
+      avg_reward_per_step:                      0.046836853
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        28311552
+    env-step-replay-before:                     14581760
+    env-step-used-before:                       28311552
+    ppo-update-before:                          17280
+    env-step-generate-before:                   13729792
+    env-step-all-after:                         28344320
+    env-step-replay-after:                      14581760
+    env-step-used-after:                        28344320
+    ppo-update-after:                           17300
+    env-step-generate-after:                    13762560
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     56.895485
+    avg_avg_return:                             0.4122182
+    avg_reward_per_step:                        0.047454834
+    proxy_corner:
+      avg_avg_return:                           0.017126815
+      avg_reward_per_step:                      0.0007324219
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     56.895485
+    avg_avg_return:                             0.4122182
+    avg_reward_per_step:                        0.047454834
+    proxy_corner:
+      avg_avg_return:                           0.017126815
+      avg_reward_per_step:                      0.0007324219
+  perf:
+    env_steps_per_second:                       323542.12282190815
+    ppo_updates_per_second:                     41.68868625162157
+    eval-batch-train-duration:                  0.51805529743433
+    eval-batch-orig-duration:                   0.4712741682305932
+    eval-batch-shift-duration:                  0.45266801677644253
+    eval-batch-tree-duration:                   0.4778062365949154
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    15.115967
+      cheese-corne_dist_wavg:                   8.7234335
+      levels_cheese_is_in_corner_avg:           0.018554688
+      mouse-cheese_dist_finite_avg:             12.9904785
+      mouse-cheese_dist_finite_wavg:            5.79127
+      mouse-cheese_dist_solvable_avg:           12.9904785
+      mouse-cheese_dist_solvable_wavg:          5.79127
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.6528303
+    visit_patterns:
+      avg_first_visit_time:                     271.93408
+      avg_last_visit_time:                      435.3015
+      num_replay_batches:                       445
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0038642546
+    avg_actor_approxkl3:                        0.003766377
+    avg_actor_clipfrac:                         0.18668823
+    avg_actor_loss:                             0.0037334263
+    avg_advantage:                              0.013422448
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.015071334
+    avg_entropy:                                0.81654274
+    avg_grad_norm_pre_clip:                     5.6611466
+    avg_loss:                                   0.010452552
+    max:
+      max_actor_approxkl1:                      0.010914246
+      max_actor_approxkl3:                      0.009232195
+      max_actor_clipfrac:                       0.34545898
+      max_actor_loss:                           0.013872221
+      max_advantage:                            0.5320678
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.016688174
+      max_entropy:                              0.8474574
+      max_grad_norm_pre_clip:                   10.202503
+      max_loss:                                 0.020471305
+    std:
+      std_actor_approxkl1:                      0.002774029
+      std_actor_approxkl3:                      0.0026711407
+      std_actor_clipfrac:                       0.10432599
+      std_actor_loss:                           0.0044084988
+      std_advantage:                            0.17409082
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00071502785
+      std_entropy:                              0.014872994
+      std_grad_norm_pre_clip:                   2.793659
+      std_loss:                                 0.0044294433
+  eval-batch-train:
+    avg_avg_episode_length:                     54.385544
+    avg_avg_return:                             0.4772612
+    avg_benchmark_regret:                       0.5099559
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.05731964
+    proxy_corner:
+      avg_avg_return:                           0.039253954
+      avg_reward_per_step:                      0.0018997192
+  eval-batch-orig:
+    avg_avg_episode_length:                     59.867638
+    avg_avg_return:                             0.4072347
+    avg_benchmark_regret:                       0.5772769
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.03023529
+    proxy_corner:
+      avg_avg_return:                           0.4072347
+      avg_reward_per_step:                      0.03023529
+  eval-batch-shift:
+    avg_avg_episode_length:                     52.519646
+    avg_avg_return:                             0.50781035
+    avg_benchmark_regret:                       0.47969562
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.07154846
+    proxy_corner:
+      avg_avg_return:                           0.03540252
+      avg_reward_per_step:                      0.0009689331
+  eval-batch-tree:
+    avg_avg_episode_length:                     61.06247
+    avg_avg_return:                             0.4091894
+    avg_benchmark_regret:                       0.57544583
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.045204163
+    proxy_corner:
+      avg_avg_return:                           0.4091894
+      avg_reward_per_step:                      0.045204163
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        29360128
+    env-step-replay-before:                     15171584
+    env-step-used-before:                       29360128
+    ppo-update-before:                          17920
+    env-step-generate-before:                   14188544
+    env-step-all-after:                         29392896
+    env-step-replay-after:                      15171584
+    env-step-used-after:                        29392896
+    ppo-update-after:                           17940
+    env-step-generate-after:                    14221312
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     55.47883
+    avg_avg_return:                             0.42889833
+    avg_reward_per_step:                        0.065826416
+    proxy_corner:
+      avg_avg_return:                           0.049389903
+      avg_reward_per_step:                      0.0016479492
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     55.47883
+    avg_avg_return:                             0.42889833
+    avg_reward_per_step:                        0.065826416
+    proxy_corner:
+      avg_avg_return:                           0.049389903
+      avg_reward_per_step:                      0.0016479492
+  perf:
+    env_steps_per_second:                       322882.25758120377
+    ppo_updates_per_second:                     42.368003989010994
+    eval-batch-train-duration:                  0.524992885068059
+    eval-batch-orig-duration:                   0.4983866587281227
+    eval-batch-shift-duration:                  0.48793032485991716
+    eval-batch-tree-duration:                   0.46127290092408657
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    15.0979
+      cheese-corne_dist_wavg:                   21.296917
+      levels_cheese_is_in_corner_avg:           0.018310547
+      mouse-cheese_dist_finite_avg:             13.150635
+      mouse-cheese_dist_finite_wavg:            6.710707
+      mouse-cheese_dist_solvable_avg:           13.150635
+      mouse-cheese_dist_solvable_wavg:          6.710708
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000008
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.64953643
+    visit_patterns:
+      avg_first_visit_time:                     283.2422
+      avg_last_visit_time:                      453.3252
+      num_replay_batches:                       463
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014187869
+    avg_actor_approxkl3:                        0.0012157746
+    avg_actor_clipfrac:                         0.062805176
+    avg_actor_loss:                             -0.00068038626
+    avg_advantage:                              0.014659759
+    avg_critic_clipfrac:                        0.0004272461
+    avg_critic_loss:                            0.014910194
+    avg_entropy:                                0.7800194
+    avg_grad_norm_pre_clip:                     2.4273186
+    avg_loss:                                   0.005994691
+    max:
+      max_actor_approxkl1:                      0.0034919616
+      max_actor_approxkl3:                      0.002594058
+      max_actor_clipfrac:                       0.14697266
+      max_actor_loss:                           0.0021952689
+      max_advantage:                            0.5935073
+      max_critic_clipfrac:                      0.006225586
+      max_critic_loss:                          0.017069677
+      max_entropy:                              0.8036901
+      max_grad_norm_pre_clip:                   4.772195
+      max_loss:                                 0.009299993
+    std:
+      std_actor_approxkl1:                      0.0008817884
+      std_actor_approxkl3:                      0.00066640635
+      std_actor_clipfrac:                       0.044609286
+      std_actor_loss:                           0.0012808528
+      std_advantage:                            0.17407942
+      std_critic_clipfrac:                      0.001422787
+      std_critic_loss:                          0.000902023
+      std_entropy:                              0.01473017
+      std_grad_norm_pre_clip:                   1.3029464
+      std_loss:                                 0.001424193
+  eval-batch-train:
+    avg_avg_episode_length:                     52.354393
+    avg_avg_return:                             0.4970312
+    avg_benchmark_regret:                       0.49018586
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.066101074
+    proxy_corner:
+      avg_avg_return:                           0.03899283
+      avg_reward_per_step:                      0.0027160645
+  eval-batch-orig:
+    avg_avg_episode_length:                     57.688236
+    avg_avg_return:                             0.44266427
+    avg_benchmark_regret:                       0.5418473
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.03668213
+    proxy_corner:
+      avg_avg_return:                           0.44266427
+      avg_reward_per_step:                      0.03668213
+  eval-batch-shift:
+    avg_avg_episode_length:                     50.256134
+    avg_avg_return:                             0.52121407
+    avg_benchmark_regret:                       0.4662919
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.07316589
+    proxy_corner:
+      avg_avg_return:                           0.05050809
+      avg_reward_per_step:                      0.0015335083
+  eval-batch-tree:
+    avg_avg_episode_length:                     60.598606
+    avg_avg_return:                             0.42499214
+    avg_benchmark_regret:                       0.55964303
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.048423767
+    proxy_corner:
+      avg_avg_return:                           0.42499214
+      avg_reward_per_step:                      0.048423767
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        30408704
+    env-step-replay-before:                     15761408
+    env-step-used-before:                       30408704
+    ppo-update-before:                          18560
+    env-step-generate-before:                   14647296
+    env-step-all-after:                         30441472
+    env-step-replay-after:                      15794176
+    env-step-used-after:                        30441472
+    ppo-update-after:                           18580
+    env-step-generate-after:                    14647296
+  train-replay:
+    shift_proportion:                           0.96875
+    avg_avg_episode_length:                     69.32484
+    avg_avg_return:                             0.3433691
+    avg_reward_per_step:                        0.010803223
+    proxy_corner:
+      avg_avg_return:                           0.06028972
+      avg_reward_per_step:                      0.0010986328
+  train-all:
+    shift_proportion:                           0.96875
+    avg_avg_episode_length:                     69.32484
+    avg_avg_return:                             0.3433691
+    avg_reward_per_step:                        0.010803223
+    proxy_corner:
+      avg_avg_return:                           0.06028972
+      avg_reward_per_step:                      0.0010986328
+  perf:
+    env_steps_per_second:                       258471.80301336412
+    ppo_updates_per_second:                     41.59039012700803
+    eval-batch-train-duration:                  0.5469514261931181
+    eval-batch-orig-duration:                   0.46883756294846535
+    eval-batch-shift-duration:                  0.4564950680360198
+    eval-batch-tree-duration:                   0.46991399209946394
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    15.153809
+      cheese-corne_dist_wavg:                   5.11722
+      levels_cheese_is_in_corner_avg:           0.018798828
+      mouse-cheese_dist_finite_avg:             13.310791
+      mouse-cheese_dist_finite_wavg:            4.923401
+      mouse-cheese_dist_solvable_avg:           13.310791
+      mouse-cheese_dist_solvable_wavg:          4.923401
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.6510955
+    visit_patterns:
+      avg_first_visit_time:                     297.01685
+      avg_last_visit_time:                      472.2976
+      num_replay_batches:                       482
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013056443
+    avg_actor_approxkl3:                        0.0015272201
+    avg_actor_clipfrac:                         0.08063354
+    avg_actor_loss:                             -0.00018758737
+    avg_advantage:                              -0.0065758405
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0136962375
+    avg_entropy:                                0.8457182
+    avg_grad_norm_pre_clip:                     3.349015
+    avg_loss:                                   0.0058148126
+    max:
+      max_actor_approxkl1:                      0.00393634
+      max_actor_approxkl3:                      0.003913792
+      max_actor_clipfrac:                       0.21362305
+      max_actor_loss:                           0.003466174
+      max_advantage:                            0.47070014
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.014725438
+      max_entropy:                              0.87991023
+      max_grad_norm_pre_clip:                   6.6194844
+      max_loss:                                 0.009243813
+    std:
+      std_actor_approxkl1:                      0.0010536981
+      std_actor_approxkl3:                      0.00088925735
+      std_actor_clipfrac:                       0.052493136
+      std_actor_loss:                           0.0015085685
+      std_advantage:                            0.1654711
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00067665364
+      std_entropy:                              0.015647708
+      std_grad_norm_pre_clip:                   1.7576361
+      std_loss:                                 0.0014867072
+  eval-batch-train:
+    avg_avg_episode_length:                     51.63569
+    avg_avg_return:                             0.5021728
+    avg_benchmark_regret:                       0.48504424
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.066467285
+    proxy_corner:
+      avg_avg_return:                           0.045949675
+      avg_reward_per_step:                      0.0020446777
+  eval-batch-orig:
+    avg_avg_episode_length:                     56.998245
+    avg_avg_return:                             0.44539285
+    avg_benchmark_regret:                       0.53911865
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.037712097
+    proxy_corner:
+      avg_avg_return:                           0.44539285
+      avg_reward_per_step:                      0.037712097
+  eval-batch-shift:
+    avg_avg_episode_length:                     50.518234
+    avg_avg_return:                             0.52892697
+    avg_benchmark_regret:                       0.45857894
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.07492828
+    proxy_corner:
+      avg_avg_return:                           0.049674645
+      avg_reward_per_step:                      0.0015029907
+  eval-batch-tree:
+    avg_avg_episode_length:                     58.45482
+    avg_avg_return:                             0.44664648
+    avg_benchmark_regret:                       0.53798866
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.05191803
+    proxy_corner:
+      avg_avg_return:                           0.44664648
+      avg_reward_per_step:                      0.05191803
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        31457280
+    env-step-replay-before:                     16384000
+    env-step-used-before:                       31457280
+    ppo-update-before:                          19200
+    env-step-generate-before:                   15073280
+    env-step-all-after:                         31490048
+    env-step-replay-after:                      16384000
+    env-step-used-after:                        31490048
+    ppo-update-after:                           19220
+    env-step-generate-after:                    15106048
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     53.83457
+    avg_avg_return:                             0.4739949
+    avg_reward_per_step:                        0.059509277
+    proxy_corner:
+      avg_avg_return:                           0.03139861
+      avg_reward_per_step:                      0.0006713867
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     53.83457
+    avg_avg_return:                             0.4739949
+    avg_reward_per_step:                        0.059509277
+    proxy_corner:
+      avg_avg_return:                           0.03139861
+      avg_reward_per_step:                      0.0006713867
+  perf:
+    env_steps_per_second:                       289181.82741266297
+    ppo_updates_per_second:                     41.4932486463495
+    eval-batch-train-duration:                  0.5156608503311872
+    eval-batch-orig-duration:                   0.4840225735679269
+    eval-batch-shift-duration:                  0.4799911370500922
+    eval-batch-tree-duration:                   0.45634580682963133
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    15.249268
+      cheese-corne_dist_wavg:                   8.732857
+      levels_cheese_is_in_corner_avg:           0.01928711
+      mouse-cheese_dist_finite_avg:             13.449219
+      mouse-cheese_dist_finite_wavg:            5.8401084
+      mouse-cheese_dist_solvable_avg:           13.449219
+      mouse-cheese_dist_solvable_wavg:          5.8401084
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.6468526
+    visit_patterns:
+      avg_first_visit_time:                     314.58154
+      avg_last_visit_time:                      490.37988
+      num_replay_batches:                       500
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0032039613
+    avg_actor_approxkl3:                        0.0031842708
+    avg_actor_clipfrac:                         0.16202393
+    avg_actor_loss:                             0.0020456202
+    avg_advantage:                              0.039074633
+    avg_critic_clipfrac:                        0.10905762
+    avg_critic_loss:                            0.016145235
+    avg_entropy:                                0.76683265
+    avg_grad_norm_pre_clip:                     6.5572395
+    avg_loss:                                   0.009351404
+    max:
+      max_actor_approxkl1:                      0.0067466954
+      max_actor_approxkl3:                      0.0076896646
+      max_actor_clipfrac:                       0.30773926
+      max_actor_loss:                           0.007858105
+      max_advantage:                            0.5900347
+      max_critic_clipfrac:                      0.7077637
+      max_critic_loss:                          0.020180201
+      max_entropy:                              0.80419844
+      max_grad_norm_pre_clip:                   11.807993
+      max_loss:                                 0.015045039
+    std:
+      std_actor_approxkl1:                      0.0019377508
+      std_actor_approxkl3:                      0.0020080744
+      std_actor_clipfrac:                       0.09253784
+      std_actor_loss:                           0.0028468275
+      std_advantage:                            0.1741993
+      std_critic_clipfrac:                      0.2211266
+      std_critic_loss:                          0.0013971799
+      std_entropy:                              0.019418571
+      std_grad_norm_pre_clip:                   3.6186666
+      std_loss:                                 0.002859724
+  eval-batch-train:
+    avg_avg_episode_length:                     50.951027
+    avg_avg_return:                             0.51872873
+    avg_benchmark_regret:                       0.46848837
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.06538391
+    proxy_corner:
+      avg_avg_return:                           0.04324442
+      avg_reward_per_step:                      0.002456665
+  eval-batch-orig:
+    avg_avg_episode_length:                     55.356026
+    avg_avg_return:                             0.4748395
+    avg_benchmark_regret:                       0.50967205
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.03820038
+    proxy_corner:
+      avg_avg_return:                           0.4748395
+      avg_reward_per_step:                      0.03820038
+  eval-batch-shift:
+    avg_avg_episode_length:                     49.171886
+    avg_avg_return:                             0.54376817
+    avg_benchmark_regret:                       0.4437378
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.07888794
+    proxy_corner:
+      avg_avg_return:                           0.049150962
+      avg_reward_per_step:                      0.0015106201
+  eval-batch-tree:
+    avg_avg_episode_length:                     57.72501
+    avg_avg_return:                             0.46760717
+    avg_benchmark_regret:                       0.51702803
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.053962708
+    proxy_corner:
+      avg_avg_return:                           0.46760717
+      avg_reward_per_step:                      0.053962708
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        32505856
+    env-step-replay-before:                     16842752
+    env-step-used-before:                       32505856
+    ppo-update-before:                          19840
+    env-step-generate-before:                   15663104
+    env-step-all-after:                         32538624
+    env-step-replay-after:                      16875520
+    env-step-used-after:                        32538624
+    ppo-update-after:                           19860
+    env-step-generate-after:                    15663104
+  train-replay:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     68.382065
+    avg_avg_return:                             0.3880931
+    avg_reward_per_step:                        0.011413574
+    proxy_corner:
+      avg_avg_return:                           0.049439058
+      avg_reward_per_step:                      0.0010986328
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     68.382065
+    avg_avg_return:                             0.3880931
+    avg_reward_per_step:                        0.011413574
+    proxy_corner:
+      avg_avg_return:                           0.049439058
+      avg_reward_per_step:                      0.0010986328
+  perf:
+    env_steps_per_second:                       308157.70662911824
+    ppo_updates_per_second:                     41.813705797927724
+    eval-batch-train-duration:                  0.5343621280044317
+    eval-batch-orig-duration:                   0.48446016665548086
+    eval-batch-shift-duration:                  0.492098081856966
+    eval-batch-tree-duration:                   0.49534434732049704
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    15.336914
+      cheese-corne_dist_wavg:                   8.738699
+      levels_cheese_is_in_corner_avg:           0.018798828
+      mouse-cheese_dist_finite_avg:             13.580322
+      mouse-cheese_dist_finite_wavg:            6.750847
+      mouse-cheese_dist_solvable_avg:           13.580322
+      mouse-cheese_dist_solvable_wavg:          6.750847
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.64544034
+    visit_patterns:
+      avg_first_visit_time:                     329.46265
+      avg_last_visit_time:                      505.22778
+      num_replay_batches:                       515
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001657312
+    avg_actor_approxkl3:                        0.0015797511
+    avg_actor_clipfrac:                         0.08344727
+    avg_actor_loss:                             -8.899272e-05
+    avg_advantage:                              0.0063862773
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.012744936
+    avg_entropy:                                0.83433646
+    avg_grad_norm_pre_clip:                     4.0338783
+    avg_loss:                                   0.0054491386
+    max:
+      max_actor_approxkl1:                      0.0046904655
+      max_actor_approxkl3:                      0.0037186057
+      max_actor_clipfrac:                       0.17480469
+      max_actor_loss:                           0.0045351833
+      max_advantage:                            0.5235888
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.014991725
+      max_entropy:                              0.85690796
+      max_grad_norm_pre_clip:                   9.455158
+      max_loss:                                 0.01041634
+    std:
+      std_actor_approxkl1:                      0.0012601302
+      std_actor_approxkl3:                      0.0008494039
+      std_actor_clipfrac:                       0.05001579
+      std_actor_loss:                           0.0020843504
+      std_advantage:                            0.16019315
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0007961084
+      std_entropy:                              0.013760454
+      std_grad_norm_pre_clip:                   2.5465314
+      std_loss:                                 0.0022553802
+  eval-batch-train:
+    avg_avg_episode_length:                     51.003693
+    avg_avg_return:                             0.52995646
+    avg_benchmark_regret:                       0.4572606
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.06686401
+    proxy_corner:
+      avg_avg_return:                           0.047794387
+      avg_reward_per_step:                      0.0025100708
+  eval-batch-orig:
+    avg_avg_episode_length:                     54.7295
+    avg_avg_return:                             0.4792362
+    avg_benchmark_regret:                       0.50527537
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.0415802
+    proxy_corner:
+      avg_avg_return:                           0.4792362
+      avg_reward_per_step:                      0.0415802
+  eval-batch-shift:
+    avg_avg_episode_length:                     49.9888
+    avg_avg_return:                             0.5424053
+    avg_benchmark_regret:                       0.44510064
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.076789856
+    proxy_corner:
+      avg_avg_return:                           0.047479466
+      avg_reward_per_step:                      0.0013809204
+  eval-batch-tree:
+    avg_avg_episode_length:                     57.052998
+    avg_avg_return:                             0.47526956
+    avg_benchmark_regret:                       0.5093656
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.054756165
+    proxy_corner:
+      avg_avg_return:                           0.47526956
+      avg_reward_per_step:                      0.054756165
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        33554432
+    env-step-replay-before:                     17399808
+    env-step-used-before:                       33554432
+    ppo-update-before:                          20480
+    env-step-generate-before:                   16154624
+    env-step-all-after:                         33587200
+    env-step-replay-after:                      17399808
+    env-step-used-after:                        33587200
+    ppo-update-after:                           20500
+    env-step-generate-after:                    16187392
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     52.404465
+    avg_avg_return:                             0.46431518
+    avg_reward_per_step:                        0.065093994
+    proxy_corner:
+      avg_avg_return:                           0.037434146
+      avg_reward_per_step:                      0.0007019043
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     52.404465
+    avg_avg_return:                             0.46431518
+    avg_reward_per_step:                        0.065093994
+    proxy_corner:
+      avg_avg_return:                           0.037434146
+      avg_reward_per_step:                      0.0007019043
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  perf:
+    env_steps_per_second:                       353036.67063242075
+    ppo_updates_per_second:                     41.4027885968892
+    eval-batch-train-duration:                  0.6443750131875277
+    eval-batch-orig-duration:                   0.5897419080138206
+    eval-batch-shift-duration:                  0.5828878507018089
+    eval-batch-tree-duration:                   0.5864708088338375
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    15.406738
+      cheese-corne_dist_wavg:                   12.330201
+      levels_cheese_is_in_corner_avg:           0.01928711
+      mouse-cheese_dist_finite_avg:             13.728516
+      mouse-cheese_dist_finite_wavg:            6.7691984
+      mouse-cheese_dist_solvable_avg:           13.728516
+      mouse-cheese_dist_solvable_wavg:          6.7691984
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016798
+    scoring:
+      avg_scores:                               0.64477724
+    visit_patterns:
+      avg_first_visit_time:                     348.3379
+      avg_last_visit_time:                      521.25195
+      num_replay_batches:                       531
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0015234493
+    avg_actor_approxkl3:                        0.0016614326
+    avg_actor_clipfrac:                         0.090875246
+    avg_actor_loss:                             0.00036798717
+    avg_advantage:                              0.022412527
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.015239455
+    avg_entropy:                                0.7586009
+    avg_grad_norm_pre_clip:                     4.4698405
+    avg_loss:                                   0.0072291135
+    max:
+      max_actor_approxkl1:                      0.003985032
+      max_actor_approxkl3:                      0.0040527023
+      max_actor_clipfrac:                       0.24414062
+      max_actor_loss:                           0.0063519105
+      max_advantage:                            0.4674152
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.017071772
+      max_entropy:                              0.77717805
+      max_grad_norm_pre_clip:                   14.343595
+      max_loss:                                 0.013800753
+    std:
+      std_actor_approxkl1:                      0.0010511089
+      std_actor_approxkl3:                      0.0010139415
+      std_actor_clipfrac:                       0.061352223
+      std_actor_loss:                           0.0022440695
+      std_advantage:                            0.17452486
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00097259384
+      std_entropy:                              0.010910823
+      std_grad_norm_pre_clip:                   3.2370198
+      std_loss:                                 0.002353387
+  eval-batch-train:
+    avg_avg_episode_length:                     49.783073
+    avg_avg_return:                             0.5415361
+    avg_benchmark_regret:                       0.44568092
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.068244934
+    proxy_corner:
+      avg_avg_return:                           0.0438127
+      avg_reward_per_step:                      0.0025939941
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-orig:
+    avg_avg_episode_length:                     56.075653
+    avg_avg_return:                             0.47418052
+    avg_benchmark_regret:                       0.51033103
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.04070282
+    proxy_corner:
+      avg_avg_return:                           0.47418052
+      avg_reward_per_step:                      0.04070282
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-shift:
+    avg_avg_episode_length:                     47.873474
+    avg_avg_return:                             0.562428
+    avg_benchmark_regret:                       0.42507792
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.07817078
+    proxy_corner:
+      avg_avg_return:                           0.0443179
+      avg_reward_per_step:                      0.0013198853
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-tree:
+    avg_avg_episode_length:                     57.273808
+    avg_avg_return:                             0.4788681
+    avg_benchmark_regret:                       0.5057671
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.055664062
+    proxy_corner:
+      avg_avg_return:                           0.4788681
+      avg_reward_per_step:                      0.055664062
+    rollouts_gif:                               float32(132, 225, 225, 3)
+===========================================================
+ 17%|▏| 33.6M/200M [12:00<40:56, 67
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        34603008
+    env-step-replay-before:                     17694720
+    env-step-used-before:                       34603008
+    ppo-update-before:                          21120
+    env-step-generate-before:                   16908288
+    env-step-all-after:                         34635776
+    env-step-replay-after:                      17727488
+    env-step-used-after:                        34635776
+    ppo-update-after:                           21140
+    env-step-generate-after:                    16908288
+  train-replay:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     70.51724
+    avg_avg_return:                             0.34712404
+    avg_reward_per_step:                        0.010070801
+    proxy_corner:
+      avg_avg_return:                           0.07640349
+      avg_reward_per_step:                      0.0013122559
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     70.51724
+    avg_avg_return:                             0.34712404
+    avg_reward_per_step:                        0.010070801
+    proxy_corner:
+      avg_avg_return:                           0.07640349
+      avg_reward_per_step:                      0.0013122559
+  perf:
+    env_steps_per_second:                       297237.7740849367
+    ppo_updates_per_second:                     42.2089720223477
+    eval-batch-train-duration:                  0.5375803085044026
+    eval-batch-orig-duration:                   0.4853266067802906
+    eval-batch-shift-duration:                  0.485461100935936
+    eval-batch-tree-duration:                   0.49624405428767204
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    15.485596
+      cheese-corne_dist_wavg:                   23.144548
+      levels_cheese_is_in_corner_avg:           0.020019531
+      mouse-cheese_dist_finite_avg:             13.8479
+      mouse-cheese_dist_finite_wavg:            6.7746887
+      mouse-cheese_dist_solvable_avg:           13.8479
+      mouse-cheese_dist_solvable_wavg:          6.7746887
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.64230454
+    visit_patterns:
+      avg_first_visit_time:                     364.31543
+      avg_last_visit_time:                      531.2241
+      num_replay_batches:                       541
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0017062832
+    avg_actor_approxkl3:                        0.0015887807
+    avg_actor_clipfrac:                         0.08919678
+    avg_actor_loss:                             -0.00035283677
+    avg_advantage:                              -0.0048934235
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.013099718
+    avg_entropy:                                0.8209039
+    avg_grad_norm_pre_clip:                     3.0631776
+    avg_loss:                                   0.0053761187
+    max:
+      max_actor_approxkl1:                      0.002819785
+      max_actor_approxkl3:                      0.0028177476
+      max_actor_clipfrac:                       0.17919922
+      max_actor_loss:                           0.0022705346
+      max_advantage:                            0.48302805
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.014151785
+      max_entropy:                              0.8435472
+      max_grad_norm_pre_clip:                   7.364168
+      max_loss:                                 0.0075474787
+    std:
+      std_actor_approxkl1:                      0.00087617984
+      std_actor_approxkl3:                      0.0007354616
+      std_actor_clipfrac:                       0.050313927
+      std_actor_loss:                           0.0013688045
+      std_advantage:                            0.16189666
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0007875906
+      std_entropy:                              0.010931846
+      std_grad_norm_pre_clip:                   1.7787676
+      std_loss:                                 0.0013788713
+  eval-batch-train:
+    avg_avg_episode_length:                     47.531273
+    avg_avg_return:                             0.56249464
+    avg_benchmark_regret:                       0.42472237
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.07357025
+    proxy_corner:
+      avg_avg_return:                           0.05273971
+      avg_reward_per_step:                      0.0028839111
+  eval-batch-orig:
+    avg_avg_episode_length:                     54.724934
+    avg_avg_return:                             0.48474154
+    avg_benchmark_regret:                       0.49977
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.04511261
+    proxy_corner:
+      avg_avg_return:                           0.48474154
+      avg_reward_per_step:                      0.04511261
+  eval-batch-shift:
+    avg_avg_episode_length:                     46.815594
+    avg_avg_return:                             0.57498896
+    avg_benchmark_regret:                       0.41251695
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.084373474
+    proxy_corner:
+      avg_avg_return:                           0.056358658
+      avg_reward_per_step:                      0.0016555786
+  eval-batch-tree:
+    avg_avg_episode_length:                     55.559067
+    avg_avg_return:                             0.49237263
+    avg_benchmark_regret:                       0.4922625
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.056785583
+    proxy_corner:
+      avg_avg_return:                           0.49237263
+      avg_reward_per_step:                      0.056785583
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        35651584
+    env-step-replay-before:                     18153472
+    env-step-used-before:                       35651584
+    ppo-update-before:                          21760
+    env-step-generate-before:                   17498112
+    env-step-all-after:                         35684352
+    env-step-replay-after:                      18186240
+    env-step-used-after:                        35684352
+    ppo-update-after:                           21780
+    env-step-generate-after:                    17498112
+  train-replay:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     67.10768
+    avg_avg_return:                             0.3961871
+    avg_reward_per_step:                        0.01159668
+    proxy_corner:
+      avg_avg_return:                           0.02874665
+      avg_reward_per_step:                      0.00064086914
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     67.10768
+    avg_avg_return:                             0.3961871
+    avg_reward_per_step:                        0.01159668
+    proxy_corner:
+      avg_avg_return:                           0.02874665
+      avg_reward_per_step:                      0.00064086914
+  perf:
+    env_steps_per_second:                       292132.835580513
+    ppo_updates_per_second:                     42.68975230487555
+    eval-batch-train-duration:                  0.5211221361532807
+    eval-batch-orig-duration:                   0.48241021670401096
+    eval-batch-shift-duration:                  0.475724876858294
+    eval-batch-tree-duration:                   0.4642973132431507
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    15.500732
+      cheese-corne_dist_wavg:                   17.759398
+      levels_cheese_is_in_corner_avg:           0.019042969
+      mouse-cheese_dist_finite_avg:             14.192383
+      mouse-cheese_dist_finite_wavg:            6.807908
+      mouse-cheese_dist_solvable_avg:           14.192383
+      mouse-cheese_dist_solvable_wavg:          6.807908
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.62196714
+    visit_patterns:
+      avg_first_visit_time:                     384.25366
+      avg_last_visit_time:                      545.2793
+      num_replay_batches:                       555
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0021575526
+    avg_actor_approxkl3:                        0.0019248342
+    avg_actor_clipfrac:                         0.10216675
+    avg_actor_loss:                             0.00037533158
+    avg_advantage:                              -0.0022952398
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.012580079
+    avg_entropy:                                0.7803695
+    avg_grad_norm_pre_clip:                     4.425792
+    avg_loss:                                   0.0058850017
+    max:
+      max_actor_approxkl1:                      0.006592745
+      max_actor_approxkl3:                      0.004002421
+      max_actor_clipfrac:                       0.20654297
+      max_actor_loss:                           0.0036952049
+      max_advantage:                            0.46486023
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.014976678
+      max_entropy:                              0.8095992
+      max_grad_norm_pre_clip:                   9.274894
+      max_loss:                                 0.008773759
+    std:
+      std_actor_approxkl1:                      0.0014628067
+      std_actor_approxkl3:                      0.0010037678
+      std_actor_clipfrac:                       0.056268536
+      std_actor_loss:                           0.0015129615
+      std_advantage:                            0.15905891
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011182232
+      std_entropy:                              0.013942662
+      std_grad_norm_pre_clip:                   2.4549878
+      std_loss:                                 0.0015654552
+  eval-batch-train:
+    avg_avg_episode_length:                     46.92665
+    avg_avg_return:                             0.58213377
+    avg_benchmark_regret:                       0.4050833
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.075553894
+    proxy_corner:
+      avg_avg_return:                           0.047910087
+      avg_reward_per_step:                      0.0025787354
+  eval-batch-orig:
+    avg_avg_episode_length:                     54.270317
+    avg_avg_return:                             0.49523032
+    avg_benchmark_regret:                       0.4892812
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.04045105
+    proxy_corner:
+      avg_avg_return:                           0.49523032
+      avg_reward_per_step:                      0.04045105
+  eval-batch-shift:
+    avg_avg_episode_length:                     45.76829
+    avg_avg_return:                             0.59331656
+    avg_benchmark_regret:                       0.39418936
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.08267975
+    proxy_corner:
+      avg_avg_return:                           0.047119275
+      avg_reward_per_step:                      0.0016098022
+  eval-batch-tree:
+    avg_avg_episode_length:                     54.65709
+    avg_avg_return:                             0.506323
+    avg_benchmark_regret:                       0.4783122
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.05467987
+    proxy_corner:
+      avg_avg_return:                           0.506323
+      avg_reward_per_step:                      0.05467987
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        36700160
+    env-step-replay-before:                     18677760
+    env-step-used-before:                       36700160
+    ppo-update-before:                          22400
+    env-step-generate-before:                   18022400
+    env-step-all-after:                         36732928
+    env-step-replay-after:                      18677760
+    env-step-used-after:                        36732928
+    ppo-update-after:                           22420
+    env-step-generate-after:                    18055168
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     50.68767
+    avg_avg_return:                             0.50666195
+    avg_reward_per_step:                        0.06817627
+    proxy_corner:
+      avg_avg_return:                           0.04193899
+      avg_reward_per_step:                      0.001953125
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     50.68767
+    avg_avg_return:                             0.50666195
+    avg_reward_per_step:                        0.06817627
+    proxy_corner:
+      avg_avg_return:                           0.04193899
+      avg_reward_per_step:                      0.001953125
+  perf:
+    env_steps_per_second:                       325420.37102424545
+    ppo_updates_per_second:                     42.00741044922134
+    eval-batch-train-duration:                  0.5396571196615696
+    eval-batch-orig-duration:                   0.48281107284128666
+    eval-batch-shift-duration:                  0.49395108316093683
+    eval-batch-tree-duration:                   0.49241927824914455
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    15.410156
+      cheese-corne_dist_wavg:                   10.55755
+      levels_cheese_is_in_corner_avg:           0.018554688
+      mouse-cheese_dist_finite_avg:             14.432617
+      mouse-cheese_dist_finite_wavg:            5.942278
+      mouse-cheese_dist_solvable_avg:           14.432617
+      mouse-cheese_dist_solvable_wavg:          5.9422784
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.6177863
+    visit_patterns:
+      avg_first_visit_time:                     404.41187
+      avg_last_visit_time:                      560.2886
+      num_replay_batches:                       570
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0015571631
+    avg_actor_approxkl3:                        0.0013065025
+    avg_actor_clipfrac:                         0.07457276
+    avg_actor_loss:                             -0.0005000755
+    avg_advantage:                              0.0042982902
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.013315635
+    avg_entropy:                                0.6703418
+    avg_grad_norm_pre_clip:                     3.3025455
+    avg_loss:                                   0.005487399
+    max:
+      max_actor_approxkl1:                      0.0045195594
+      max_actor_approxkl3:                      0.0024986372
+      max_actor_clipfrac:                       0.15673828
+      max_actor_loss:                           0.0021629333
+      max_advantage:                            0.5036593
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.014924346
+      max_entropy:                              0.7059858
+      max_grad_norm_pre_clip:                   8.264145
+      max_loss:                                 0.008261631
+    std:
+      std_actor_approxkl1:                      0.0009329737
+      std_actor_approxkl3:                      0.0006350665
+      std_actor_clipfrac:                       0.045130666
+      std_actor_loss:                           0.001427243
+      std_advantage:                            0.16333303
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00084669114
+      std_entropy:                              0.018830117
+      std_grad_norm_pre_clip:                   2.3799534
+      std_loss:                                 0.001605062
+  eval-batch-train:
+    avg_avg_episode_length:                     46.80957
+    avg_avg_return:                             0.5907099
+    avg_benchmark_regret:                       0.39650714
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.07525635
+    proxy_corner:
+      avg_avg_return:                           0.048159417
+      avg_reward_per_step:                      0.002418518
+  eval-batch-orig:
+    avg_avg_episode_length:                     53.17095
+    avg_avg_return:                             0.5033105
+    avg_benchmark_regret:                       0.48120108
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.044883728
+    proxy_corner:
+      avg_avg_return:                           0.5033105
+      avg_reward_per_step:                      0.044883728
+  eval-batch-shift:
+    avg_avg_episode_length:                     46.389206
+    avg_avg_return:                             0.57824457
+    avg_benchmark_regret:                       0.40926135
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.08602905
+    proxy_corner:
+      avg_avg_return:                           0.04970105
+      avg_reward_per_step:                      0.001701355
+  eval-batch-tree:
+    avg_avg_episode_length:                     54.100166
+    avg_avg_return:                             0.5173485
+    avg_benchmark_regret:                       0.4672867
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.057907104
+    proxy_corner:
+      avg_avg_return:                           0.5173485
+      avg_reward_per_step:                      0.057907104
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        37748736
+    env-step-replay-before:                     19267584
+    env-step-used-before:                       37748736
+    ppo-update-before:                          23040
+    env-step-generate-before:                   18481152
+    env-step-all-after:                         37781504
+    env-step-replay-after:                      19267584
+    env-step-used-after:                        37781504
+    ppo-update-after:                           23060
+    env-step-generate-after:                    18513920
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     52.0909
+    avg_avg_return:                             0.4899178
+    avg_reward_per_step:                        0.067993164
+    proxy_corner:
+      avg_avg_return:                           0.020063195
+      avg_reward_per_step:                      0.0013427734
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     52.0909
+    avg_avg_return:                             0.4899178
+    avg_reward_per_step:                        0.067993164
+    proxy_corner:
+      avg_avg_return:                           0.020063195
+      avg_reward_per_step:                      0.0013427734
+  perf:
+    env_steps_per_second:                       300135.5567972811
+    ppo_updates_per_second:                     41.74561641884255
+    eval-batch-train-duration:                  0.5293759535998106
+    eval-batch-orig-duration:                   0.47256161645054817
+    eval-batch-shift-duration:                  0.4483216358348727
+    eval-batch-tree-duration:                   0.4762583263218403
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    15.397461
+      cheese-corne_dist_wavg:                   6.051899
+      levels_cheese_is_in_corner_avg:           0.017333984
+      mouse-cheese_dist_finite_avg:             14.628174
+      mouse-cheese_dist_finite_wavg:            5.955599
+      mouse-cheese_dist_solvable_avg:           14.628174
+      mouse-cheese_dist_solvable_wavg:          5.9555993
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.6270824
+    visit_patterns:
+      avg_first_visit_time:                     419.5808
+      avg_last_visit_time:                      578.2329
+      num_replay_batches:                       588
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014561334
+    avg_actor_approxkl3:                        0.0014391632
+    avg_actor_clipfrac:                         0.081365965
+    avg_actor_loss:                             -0.00087083876
+    avg_advantage:                              0.024243724
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.014321501
+    avg_entropy:                                0.6761567
+    avg_grad_norm_pre_clip:                     2.9211671
+    avg_loss:                                   0.005613756
+    max:
+      max_actor_approxkl1:                      0.002989205
+      max_actor_approxkl3:                      0.0024540876
+      max_actor_clipfrac:                       0.16101074
+      max_actor_loss:                           0.0017521381
+      max_advantage:                            0.5082408
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.015683046
+      max_entropy:                              0.70597744
+      max_grad_norm_pre_clip:                   6.9576545
+      max_loss:                                 0.008437795
+    std:
+      std_actor_approxkl1:                      0.000644183
+      std_actor_approxkl3:                      0.00054354774
+      std_actor_clipfrac:                       0.036895584
+      std_actor_loss:                           0.0010898107
+      std_advantage:                            0.16880931
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.000735053
+      std_entropy:                              0.013075781
+      std_grad_norm_pre_clip:                   1.5774485
+      std_loss:                                 0.0012942137
+  eval-batch-train:
+    avg_avg_episode_length:                     46.216705
+    avg_avg_return:                             0.6059791
+    avg_benchmark_regret:                       0.38123798
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.07533264
+    proxy_corner:
+      avg_avg_return:                           0.055980455
+      avg_reward_per_step:                      0.0029830933
+  eval-batch-orig:
+    avg_avg_episode_length:                     53.81211
+    avg_avg_return:                             0.5121983
+    avg_benchmark_regret:                       0.47231323
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.04245758
+    proxy_corner:
+      avg_avg_return:                           0.5121983
+      avg_reward_per_step:                      0.04245758
+  eval-batch-shift:
+    avg_avg_episode_length:                     45.31963
+    avg_avg_return:                             0.5960969
+    avg_benchmark_regret:                       0.3914091
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.08766174
+    proxy_corner:
+      avg_avg_return:                           0.052338228
+      avg_reward_per_step:                      0.0019302368
+  eval-batch-tree:
+    avg_avg_episode_length:                     53.46412
+    avg_avg_return:                             0.53479934
+    avg_benchmark_regret:                       0.44983584
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.060043335
+    proxy_corner:
+      avg_avg_return:                           0.53479934
+      avg_reward_per_step:                      0.060043335
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        38797312
+    env-step-replay-before:                     19922944
+    env-step-used-before:                       38797312
+    ppo-update-before:                          23680
+    env-step-generate-before:                   18874368
+    env-step-all-after:                         38830080
+    env-step-replay-after:                      19922944
+    env-step-used-after:                        38830080
+    ppo-update-after:                           23700
+    env-step-generate-after:                    18907136
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     50.294754
+    avg_avg_return:                             0.49044436
+    avg_reward_per_step:                        0.0819397
+    proxy_corner:
+      avg_avg_return:                           0.042862598
+      avg_reward_per_step:                      0.0024719238
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     50.294754
+    avg_avg_return:                             0.49044436
+    avg_reward_per_step:                        0.0819397
+    proxy_corner:
+      avg_avg_return:                           0.042862598
+      avg_reward_per_step:                      0.0024719238
+  perf:
+    env_steps_per_second:                       295530.29840459843
+    ppo_updates_per_second:                     42.084252641192045
+    eval-batch-train-duration:                  0.5430386541411281
+    eval-batch-orig-duration:                   0.4760029623284936
+    eval-batch-shift-duration:                  0.48191494308412075
+    eval-batch-tree-duration:                   0.456431089900434
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    15.392334
+      cheese-corne_dist_wavg:                   16.831495
+      levels_cheese_is_in_corner_avg:           0.018554688
+      mouse-cheese_dist_finite_avg:             14.769531
+      mouse-cheese_dist_finite_wavg:            5.9781117
+      mouse-cheese_dist_solvable_avg:           14.769531
+      mouse-cheese_dist_solvable_wavg:          5.9781117
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.62137485
+    visit_patterns:
+      avg_first_visit_time:                     434.25806
+      avg_last_visit_time:                      598.3157
+      num_replay_batches:                       608
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001738326
+    avg_actor_approxkl3:                        0.0014870367
+    avg_actor_clipfrac:                         0.08501587
+    avg_actor_loss:                             8.340106e-05
+    avg_advantage:                              -0.0040026074
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.013107999
+    avg_entropy:                                0.64041895
+    avg_grad_norm_pre_clip:                     3.6616852
+    avg_loss:                                   0.005996981
+    max:
+      max_actor_approxkl1:                      0.004919347
+      max_actor_approxkl3:                      0.0034477573
+      max_actor_clipfrac:                       0.18566895
+      max_actor_loss:                           0.0032039434
+      max_advantage:                            0.38004798
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.016116923
+      max_entropy:                              0.6882444
+      max_grad_norm_pre_clip:                   7.665785
+      max_loss:                                 0.008979158
+    std:
+      std_actor_approxkl1:                      0.0011458219
+      std_actor_approxkl3:                      0.0007648601
+      std_actor_clipfrac:                       0.04452615
+      std_actor_loss:                           0.0013690537
+      std_advantage:                            0.16223153
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0012942058
+      std_entropy:                              0.023359476
+      std_grad_norm_pre_clip:                   1.597691
+      std_loss:                                 0.0014806411
+  eval-batch-train:
+    avg_avg_episode_length:                     45.713486
+    avg_avg_return:                             0.60087466
+    avg_benchmark_regret:                       0.3863424
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.07571411
+    proxy_corner:
+      avg_avg_return:                           0.06259259
+      avg_reward_per_step:                      0.0030059814
+  eval-batch-orig:
+    avg_avg_episode_length:                     52.755783
+    avg_avg_return:                             0.52780837
+    avg_benchmark_regret:                       0.4567032
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.046424866
+    proxy_corner:
+      avg_avg_return:                           0.52780837
+      avg_reward_per_step:                      0.046424866
+  eval-batch-shift:
+    avg_avg_episode_length:                     46.246376
+    avg_avg_return:                             0.5926101
+    avg_benchmark_regret:                       0.39489585
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.08759308
+    proxy_corner:
+      avg_avg_return:                           0.05569681
+      avg_reward_per_step:                      0.0019226074
+  eval-batch-tree:
+    avg_avg_episode_length:                     53.285904
+    avg_avg_return:                             0.5460671
+    avg_benchmark_regret:                       0.4385681
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.060821533
+    proxy_corner:
+      avg_avg_return:                           0.5460671
+      avg_reward_per_step:                      0.060821533
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        39845888
+    env-step-replay-before:                     20381696
+    env-step-used-before:                       39845888
+    ppo-update-before:                          24320
+    env-step-generate-before:                   19464192
+    env-step-all-after:                         39878656
+    env-step-replay-after:                      20414464
+    env-step-used-after:                        39878656
+    ppo-update-after:                           24340
+    env-step-generate-after:                    19464192
+  train-replay:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     68.547104
+    avg_avg_return:                             0.39660504
+    avg_reward_per_step:                        0.011169434
+    proxy_corner:
+      avg_avg_return:                           0.06978765
+      avg_reward_per_step:                      0.0013122559
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     68.547104
+    avg_avg_return:                             0.39660504
+    avg_reward_per_step:                        0.011169434
+    proxy_corner:
+      avg_avg_return:                           0.06978765
+      avg_reward_per_step:                      0.0013122559
+  perf:
+    env_steps_per_second:                       322874.1982942156
+    ppo_updates_per_second:                     42.07008244615686
+    eval-batch-train-duration:                  0.5239289877936244
+    eval-batch-orig-duration:                   0.5688033811748028
+    eval-batch-shift-duration:                  0.610613708384335
+    eval-batch-tree-duration:                   0.6109014889225364
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    15.425049
+      cheese-corne_dist_wavg:                   11.445041
+      levels_cheese_is_in_corner_avg:           0.018798828
+      mouse-cheese_dist_finite_avg:             14.906738
+      mouse-cheese_dist_finite_wavg:            5.9890203
+      mouse-cheese_dist_solvable_avg:           14.906738
+      mouse-cheese_dist_solvable_wavg:          5.9890203
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.6144974
+    visit_patterns:
+      avg_first_visit_time:                     451.43848
+      avg_last_visit_time:                      613.2988
+      num_replay_batches:                       623
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0026313078
+    avg_actor_approxkl3:                        0.002399938
+    avg_actor_clipfrac:                         0.13048096
+    avg_actor_loss:                             0.0013079047
+    avg_advantage:                              0.0013514517
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.011879587
+    avg_entropy:                                0.72311014
+    avg_grad_norm_pre_clip:                     3.8961427
+    avg_loss:                                   0.0065245884
+    max:
+      max_actor_approxkl1:                      0.007133725
+      max_actor_approxkl3:                      0.005244429
+      max_actor_clipfrac:                       0.26574707
+      max_actor_loss:                           0.0054056942
+      max_advantage:                            0.38622904
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.013246713
+      max_entropy:                              0.7390639
+      max_grad_norm_pre_clip:                   9.073683
+      max_loss:                                 0.011087262
+    std:
+      std_actor_approxkl1:                      0.001741823
+      std_actor_approxkl3:                      0.0014665574
+      std_actor_clipfrac:                       0.07756558
+      std_actor_loss:                           0.0023036292
+      std_advantage:                            0.15424
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0009117385
+      std_entropy:                              0.011430397
+      std_grad_norm_pre_clip:                   1.915237
+      std_loss:                                 0.0023205962
+  eval-batch-train:
+    avg_avg_episode_length:                     43.964344
+    avg_avg_return:                             0.6350589
+    avg_benchmark_regret:                       0.3521582
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.08028412
+    proxy_corner:
+      avg_avg_return:                           0.054709487
+      avg_reward_per_step:                      0.0030288696
+  eval-batch-orig:
+    avg_avg_episode_length:                     52.387123
+    avg_avg_return:                             0.5308435
+    avg_benchmark_regret:                       0.453668
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.044937134
+    proxy_corner:
+      avg_avg_return:                           0.5308435
+      avg_reward_per_step:                      0.044937134
+  eval-batch-shift:
+    avg_avg_episode_length:                     44.6056
+    avg_avg_return:                             0.61570895
+    avg_benchmark_regret:                       0.37179703
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.0921936
+    proxy_corner:
+      avg_avg_return:                           0.0527338
+      avg_reward_per_step:                      0.0018920898
+  eval-batch-tree:
+    avg_avg_episode_length:                     50.53019
+    avg_avg_return:                             0.5809691
+    avg_benchmark_regret:                       0.40366608
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.06247711
+    proxy_corner:
+      avg_avg_return:                           0.5809691
+      avg_reward_per_step:                      0.06247711
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        40894464
+    env-step-replay-before:                     21037056
+    env-step-used-before:                       40894464
+    ppo-update-before:                          24960
+    env-step-generate-before:                   19857408
+    env-step-all-after:                         40927232
+    env-step-replay-after:                      21069824
+    env-step-used-after:                        40927232
+    ppo-update-after:                           24980
+    env-step-generate-after:                    19857408
+  train-replay:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     64.29985
+    avg_avg_return:                             0.4260915
+    avg_reward_per_step:                        0.013366699
+    proxy_corner:
+      avg_avg_return:                           0.078570865
+      avg_reward_per_step:                      0.0014038086
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     64.29985
+    avg_avg_return:                             0.4260915
+    avg_reward_per_step:                        0.013366699
+    proxy_corner:
+      avg_avg_return:                           0.078570865
+      avg_reward_per_step:                      0.0014038086
+  perf:
+    env_steps_per_second:                       343826.941886669
+    ppo_updates_per_second:                     42.19018922203282
+    eval-batch-train-duration:                  0.5041501438245177
+    eval-batch-orig-duration:                   0.42323017306625843
+    eval-batch-shift-duration:                  0.4180075731128454
+    eval-batch-tree-duration:                   0.4609132232144475
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    15.489258
+      cheese-corne_dist_wavg:                   17.744741
+      levels_cheese_is_in_corner_avg:           0.018066406
+      mouse-cheese_dist_finite_avg:             15.144775
+      mouse-cheese_dist_finite_wavg:            6.9030914
+      mouse-cheese_dist_solvable_avg:           15.144775
+      mouse-cheese_dist_solvable_wavg:          6.9030914
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.99999
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.5950515
+    visit_patterns:
+      avg_first_visit_time:                     470.85474
+      avg_last_visit_time:                      633.2512
+      num_replay_batches:                       643
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0007027454
+    avg_actor_approxkl3:                        0.0009873452
+    avg_actor_clipfrac:                         0.050177004
+    avg_actor_loss:                             -0.0011661425
+    avg_advantage:                              -0.0015301467
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.01170883
+    avg_entropy:                                0.7018508
+    avg_grad_norm_pre_clip:                     2.4932363
+    avg_loss:                                   0.003986421
+    max:
+      max_actor_approxkl1:                      0.0016857964
+      max_actor_approxkl3:                      0.0015701654
+      max_actor_clipfrac:                       0.08728027
+      max_actor_loss:                           0.0014932901
+      max_advantage:                            0.4174075
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.014262982
+      max_entropy:                              0.7261529
+      max_grad_norm_pre_clip:                   4.9622946
+      max_loss:                                 0.0068209195
+    std:
+      std_actor_approxkl1:                      0.0005454205
+      std_actor_approxkl3:                      0.0003584932
+      std_actor_clipfrac:                       0.02301935
+      std_actor_loss:                           0.0012222653
+      std_advantage:                            0.15342431
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0009032339
+      std_entropy:                              0.012829252
+      std_grad_norm_pre_clip:                   1.089686
+      std_loss:                                 0.001378473
+  eval-batch-train:
+    avg_avg_episode_length:                     42.501038
+    avg_avg_return:                             0.6517012
+    avg_benchmark_regret:                       0.3355158
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.08015442
+    proxy_corner:
+      avg_avg_return:                           0.055677794
+      avg_reward_per_step:                      0.0030517578
+  eval-batch-orig:
+    avg_avg_episode_length:                     50.910946
+    avg_avg_return:                             0.5702466
+    avg_benchmark_regret:                       0.41426495
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.043899536
+    proxy_corner:
+      avg_avg_return:                           0.5702466
+      avg_reward_per_step:                      0.043899536
+  eval-batch-shift:
+    avg_avg_episode_length:                     42.609806
+    avg_avg_return:                             0.6461725
+    avg_benchmark_regret:                       0.3413334
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.093063354
+    proxy_corner:
+      avg_avg_return:                           0.056152355
+      avg_reward_per_step:                      0.0019226074
+  eval-batch-tree:
+    avg_avg_episode_length:                     48.961384
+    avg_avg_return:                             0.6062638
+    avg_benchmark_regret:                       0.37837136
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.061454773
+    proxy_corner:
+      avg_avg_return:                           0.6062638
+      avg_reward_per_step:                      0.061454773
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        41943040
+    env-step-replay-before:                     21528576
+    env-step-used-before:                       41943040
+    ppo-update-before:                          25600
+    env-step-generate-before:                   20414464
+    env-step-all-after:                         41975808
+    env-step-replay-after:                      21528576
+    env-step-used-after:                        41975808
+    ppo-update-after:                           25620
+    env-step-generate-after:                    20447232
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     49.36206
+    avg_avg_return:                             0.51940554
+    avg_reward_per_step:                        0.086517334
+    proxy_corner:
+      avg_avg_return:                           0.036328487
+      avg_reward_per_step:                      0.0012207031
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     49.36206
+    avg_avg_return:                             0.51940554
+    avg_reward_per_step:                        0.086517334
+    proxy_corner:
+      avg_avg_return:                           0.036328487
+      avg_reward_per_step:                      0.0012207031
+  perf:
+    env_steps_per_second:                       717731.7438012809
+    ppo_updates_per_second:                     42.64574043545022
+    eval-batch-train-duration:                  0.3363709859549999
+    eval-batch-orig-duration:                   0.23608846683055162
+    eval-batch-shift-duration:                  0.22263667359948158
+    eval-batch-tree-duration:                   0.25821451749652624
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    15.541016
+      cheese-corne_dist_wavg:                   11.466422
+      levels_cheese_is_in_corner_avg:           0.018310547
+      mouse-cheese_dist_finite_avg:             15.529785
+      mouse-cheese_dist_finite_wavg:            6.0478525
+      mouse-cheese_dist_solvable_avg:           15.529785
+      mouse-cheese_dist_solvable_wavg:          6.0478525
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.6040791
+    visit_patterns:
+      avg_first_visit_time:                     497.37134
+      avg_last_visit_time:                      647.1975
+      num_replay_batches:                       657
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0019397598
+    avg_actor_approxkl3:                        0.0020403005
+    avg_actor_clipfrac:                         0.09857788
+    avg_actor_loss:                             0.00076008774
+    avg_advantage:                              -0.0032599138
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.012702805
+    avg_entropy:                                0.55750555
+    avg_grad_norm_pre_clip:                     4.136223
+    avg_loss:                                   0.006553985
+    max:
+      max_actor_approxkl1:                      0.007453967
+      max_actor_approxkl3:                      0.0063995
+      max_actor_clipfrac:                       0.23339844
+      max_actor_loss:                           0.005653575
+      max_advantage:                            0.37200546
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.014649493
+      max_entropy:                              0.610213
+      max_grad_norm_pre_clip:                   9.755041
+      max_loss:                                 0.012069669
+    std:
+      std_actor_approxkl1:                      0.0017734617
+      std_actor_approxkl3:                      0.0016442132
+      std_actor_clipfrac:                       0.06161443
+      std_actor_loss:                           0.0019740893
+      std_advantage:                            0.15910038
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010861204
+      std_entropy:                              0.022683201
+      std_grad_norm_pre_clip:                   2.7672713
+      std_loss:                                 0.0020961685
+  eval-batch-train:
+    avg_avg_episode_length:                     41.096077
+    avg_avg_return:                             0.6659765
+    avg_benchmark_regret:                       0.32124048
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.08496857
+    proxy_corner:
+      avg_avg_return:                           0.06151788
+      avg_reward_per_step:                      0.0032806396
+  eval-batch-orig:
+    avg_avg_episode_length:                     48.661972
+    avg_avg_return:                             0.57886183
+    avg_benchmark_regret:                       0.40564972
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.049316406
+    proxy_corner:
+      avg_avg_return:                           0.57886183
+      avg_reward_per_step:                      0.049316406
+  eval-batch-shift:
+    avg_avg_episode_length:                     41.81984
+    avg_avg_return:                             0.6579401
+    avg_benchmark_regret:                       0.32956588
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.09578705
+    proxy_corner:
+      avg_avg_return:                           0.056930926
+      avg_reward_per_step:                      0.0021209717
+  eval-batch-tree:
+    avg_avg_episode_length:                     47.312737
+    avg_avg_return:                             0.6070225
+    avg_benchmark_regret:                       0.37761268
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.06554413
+    proxy_corner:
+      avg_avg_return:                           0.6070225
+      avg_reward_per_step:                      0.06554413
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        42991616
+    env-step-replay-before:                     21954560
+    env-step-used-before:                       42991616
+    ppo-update-before:                          26240
+    env-step-generate-before:                   21037056
+    env-step-all-after:                         43024384
+    env-step-replay-after:                      21987328
+    env-step-used-after:                        43024384
+    ppo-update-after:                           26260
+    env-step-generate-after:                    21037056
+  train-replay:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     68.70707
+    avg_avg_return:                             0.3902201
+    avg_reward_per_step:                        0.011047363
+    proxy_corner:
+      avg_avg_return:                           0.07119398
+      avg_reward_per_step:                      0.0014343262
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     68.70707
+    avg_avg_return:                             0.3902201
+    avg_reward_per_step:                        0.011047363
+    proxy_corner:
+      avg_avg_return:                           0.07119398
+      avg_reward_per_step:                      0.0014343262
+  perf:
+    env_steps_per_second:                       250087.07954257293
+    ppo_updates_per_second:                     45.14611914981754
+    eval-batch-train-duration:                  0.6816106745973229
+    eval-batch-orig-duration:                   0.6131550259888172
+    eval-batch-shift-duration:                  0.6146528581157327
+    eval-batch-tree-duration:                   0.6403668718412519
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    15.664551
+      cheese-corne_dist_wavg:                   6.9941416
+      levels_cheese_is_in_corner_avg:           0.017578125
+      mouse-cheese_dist_finite_avg:             15.695068
+      mouse-cheese_dist_finite_wavg:            7.8590293
+      mouse-cheese_dist_solvable_avg:           15.695068
+      mouse-cheese_dist_solvable_wavg:          7.8590283
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.60530996
+    visit_patterns:
+      avg_first_visit_time:                     514.5791
+      avg_last_visit_time:                      661.20654
+      num_replay_batches:                       671
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0027497064
+    avg_actor_approxkl3:                        0.0027521227
+    avg_actor_clipfrac:                         0.137146
+    avg_actor_loss:                             0.00105039
+    avg_advantage:                              -0.013267373
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.012056975
+    avg_entropy:                                0.631989
+    avg_grad_norm_pre_clip:                     5.7482843
+    avg_loss:                                   0.006446888
+    max:
+      max_actor_approxkl1:                      0.0066700378
+      max_actor_approxkl3:                      0.007018636
+      max_actor_clipfrac:                       0.29870605
+      max_actor_loss:                           0.0056068897
+      max_advantage:                            0.3463285
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.014025731
+      max_entropy:                              0.65925443
+      max_grad_norm_pre_clip:                   10.351322
+      max_loss:                                 0.010377042
+    std:
+      std_actor_approxkl1:                      0.0017486695
+      std_actor_approxkl3:                      0.0018296833
+      std_actor_clipfrac:                       0.08342568
+      std_actor_loss:                           0.0018667472
+      std_advantage:                            0.15514153
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0008986428
+      std_entropy:                              0.014606017
+      std_grad_norm_pre_clip:                   3.278397
+      std_loss:                                 0.0019030719
+  eval-batch-train:
+    avg_avg_episode_length:                     41.11859
+    avg_avg_return:                             0.6662123
+    avg_benchmark_regret:                       0.32100478
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.0899353
+    proxy_corner:
+      avg_avg_return:                           0.06389028
+      avg_reward_per_step:                      0.0035552979
+  eval-batch-orig:
+    avg_avg_episode_length:                     50.09904
+    avg_avg_return:                             0.5512404
+    avg_benchmark_regret:                       0.43327117
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.048683167
+    proxy_corner:
+      avg_avg_return:                           0.5512404
+      avg_reward_per_step:                      0.048683167
+  eval-batch-shift:
+    avg_avg_episode_length:                     40.613037
+    avg_avg_return:                             0.6692927
+    avg_benchmark_regret:                       0.31821322
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.098464966
+    proxy_corner:
+      avg_avg_return:                           0.060424063
+      avg_reward_per_step:                      0.0022354126
+  eval-batch-tree:
+    avg_avg_episode_length:                     47.388718
+    avg_avg_return:                             0.60204625
+    avg_benchmark_regret:                       0.38258892
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.06767273
+    proxy_corner:
+      avg_avg_return:                           0.60204625
+      avg_reward_per_step:                      0.06767273
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        44040192
+    env-step-replay-before:                     22511616
+    env-step-used-before:                       44040192
+    ppo-update-before:                          26880
+    env-step-generate-before:                   21528576
+    env-step-all-after:                         44072960
+    env-step-replay-after:                      22511616
+    env-step-used-after:                        44072960
+    ppo-update-after:                           26900
+    env-step-generate-after:                    21561344
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     47.638535
+    avg_avg_return:                             0.54893804
+    avg_reward_per_step:                        0.08023071
+    proxy_corner:
+      avg_avg_return:                           0.065411754
+      avg_reward_per_step:                      0.0027160645
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     47.638535
+    avg_avg_return:                             0.54893804
+    avg_reward_per_step:                        0.08023071
+    proxy_corner:
+      avg_avg_return:                           0.065411754
+      avg_reward_per_step:                      0.0027160645
+  perf:
+    env_steps_per_second:                       323418.84304651554
+    ppo_updates_per_second:                     40.86368537041885
+    eval-batch-train-duration:                  0.5293618161231279
+    eval-batch-orig-duration:                   0.48709849175065756
+    eval-batch-shift-duration:                  0.4790398105978966
+    eval-batch-tree-duration:                   0.45734985265880823
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    15.780273
+      cheese-corne_dist_wavg:                   17.754156
+      levels_cheese_is_in_corner_avg:           0.020019531
+      mouse-cheese_dist_finite_avg:             15.922852
+      mouse-cheese_dist_finite_wavg:            6.9867444
+      mouse-cheese_dist_solvable_avg:           15.922852
+      mouse-cheese_dist_solvable_wavg:          6.986746
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.6042396
+    visit_patterns:
+      avg_first_visit_time:                     534.80005
+      avg_last_visit_time:                      677.2605
+      num_replay_batches:                       687
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014628354
+    avg_actor_approxkl3:                        0.00140913
+    avg_actor_clipfrac:                         0.079437256
+    avg_actor_loss:                             -7.955246e-05
+    avg_advantage:                              -9.36233e-05
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.011450886
+    avg_entropy:                                0.53595734
+    avg_grad_norm_pre_clip:                     3.6732087
+    avg_loss:                                   0.0051099327
+    max:
+      max_actor_approxkl1:                      0.0036277194
+      max_actor_approxkl3:                      0.0023327735
+      max_actor_clipfrac:                       0.1352539
+      max_actor_loss:                           0.0018911958
+      max_advantage:                            0.38363713
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.014400081
+      max_entropy:                              0.5981955
+      max_grad_norm_pre_clip:                   7.5382233
+      max_loss:                                 0.00732875
+    std:
+      std_actor_approxkl1:                      0.00091182254
+      std_actor_approxkl3:                      0.0006021831
+      std_actor_clipfrac:                       0.036372643
+      std_actor_loss:                           0.0011093714
+      std_advantage:                            0.1514794
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0009803243
+      std_entropy:                              0.019946458
+      std_grad_norm_pre_clip:                   1.8287642
+      std_loss:                                 0.0012662162
+  eval-batch-train:
+    avg_avg_episode_length:                     39.882767
+    avg_avg_return:                             0.68892723
+    avg_benchmark_regret:                       0.29828984
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.0900116
+    proxy_corner:
+      avg_avg_return:                           0.05497832
+      avg_reward_per_step:                      0.0034561157
+  eval-batch-orig:
+    avg_avg_episode_length:                     47.43711
+    avg_avg_return:                             0.5895071
+    avg_benchmark_regret:                       0.39500445
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.05166626
+    proxy_corner:
+      avg_avg_return:                           0.5895071
+      avg_reward_per_step:                      0.05166626
+  eval-batch-shift:
+    avg_avg_episode_length:                     39.839523
+    avg_avg_return:                             0.69004357
+    avg_benchmark_regret:                       0.29746237
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.10028839
+    proxy_corner:
+      avg_avg_return:                           0.054978885
+      avg_reward_per_step:                      0.0021820068
+  eval-batch-tree:
+    avg_avg_episode_length:                     45.04667
+    avg_avg_return:                             0.6505499
+    avg_benchmark_regret:                       0.33408526
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.06726074
+    proxy_corner:
+      avg_avg_return:                           0.6505499
+      avg_reward_per_step:                      0.06726074
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        45088768
+    env-step-replay-before:                     22937600
+    env-step-used-before:                       45088768
+    ppo-update-before:                          27520
+    env-step-generate-before:                   22151168
+    env-step-all-after:                         45121536
+    env-step-replay-after:                      22937600
+    env-step-used-after:                        45121536
+    ppo-update-after:                           27540
+    env-step-generate-after:                    22183936
+  train-generate:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     42.74583
+    avg_avg_return:                             0.5775533
+    avg_reward_per_step:                        0.099487305
+    proxy_corner:
+      avg_avg_return:                           0.050550282
+      avg_reward_per_step:                      0.0019226074
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     42.74583
+    avg_avg_return:                             0.5775533
+    avg_reward_per_step:                        0.099487305
+    proxy_corner:
+      avg_avg_return:                           0.050550282
+      avg_reward_per_step:                      0.0019226074
+  perf:
+    env_steps_per_second:                       297178.54732134857
+    ppo_updates_per_second:                     42.10450270882337
+    eval-batch-train-duration:                  0.5378444828093052
+    eval-batch-orig-duration:                   0.48156147077679634
+    eval-batch-shift-duration:                  0.4748415695503354
+    eval-batch-tree-duration:                   0.4600631147623062
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    15.807129
+      cheese-corne_dist_wavg:                   23.177542
+      levels_cheese_is_in_corner_avg:           0.018310547
+      mouse-cheese_dist_finite_avg:             16.236084
+      mouse-cheese_dist_finite_wavg:            5.2062426
+      mouse-cheese_dist_solvable_avg:           16.236084
+      mouse-cheese_dist_solvable_wavg:          5.206243
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.6042763
+    visit_patterns:
+      avg_first_visit_time:                     554.4758
+      avg_last_visit_time:                      690.3042
+      num_replay_batches:                       700
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012652833
+    avg_actor_approxkl3:                        0.0009438965
+    avg_actor_clipfrac:                         0.05128784
+    avg_actor_loss:                             -0.00076966884
+    avg_advantage:                              0.00046745478
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.011569901
+    avg_entropy:                                0.4825173
+    avg_grad_norm_pre_clip:                     2.2666738
+    avg_loss:                                   0.004532765
+    max:
+      max_actor_approxkl1:                      0.0024361575
+      max_actor_approxkl3:                      0.0019207096
+      max_actor_clipfrac:                       0.103393555
+      max_actor_loss:                           0.0014587492
+      max_advantage:                            0.33684123
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.014841756
+      max_entropy:                              0.5130416
+      max_grad_norm_pre_clip:                   3.4795403
+      max_loss:                                 0.0068556834
+    std:
+      std_actor_approxkl1:                      0.0006642998
+      std_actor_approxkl3:                      0.0004288057
+      std_actor_clipfrac:                       0.024524115
+      std_actor_loss:                           0.00096088834
+      std_advantage:                            0.15276
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011541196
+      std_entropy:                              0.02199363
+      std_grad_norm_pre_clip:                   0.7982074
+      std_loss:                                 0.0010312195
+  eval-batch-train:
+    avg_avg_episode_length:                     39.18857
+    avg_avg_return:                             0.6862207
+    avg_benchmark_regret:                       0.30099636
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.09226227
+    proxy_corner:
+      avg_avg_return:                           0.06161425
+      avg_reward_per_step:                      0.0034179688
+  eval-batch-orig:
+    avg_avg_episode_length:                     47.165543
+    avg_avg_return:                             0.59511113
+    avg_benchmark_regret:                       0.38940045
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.051338196
+    proxy_corner:
+      avg_avg_return:                           0.59511113
+      avg_reward_per_step:                      0.051338196
+  eval-batch-shift:
+    avg_avg_episode_length:                     39.141342
+    avg_avg_return:                             0.69408953
+    avg_benchmark_regret:                       0.29341638
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.10059357
+    proxy_corner:
+      avg_avg_return:                           0.063174754
+      avg_reward_per_step:                      0.0024337769
+  eval-batch-tree:
+    avg_avg_episode_length:                     44.745598
+    avg_avg_return:                             0.64559376
+    avg_benchmark_regret:                       0.3390414
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.06544495
+    proxy_corner:
+      avg_avg_return:                           0.64559376
+      avg_reward_per_step:                      0.06544495
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        46137344
+    env-step-replay-before:                     23560192
+    env-step-used-before:                       46137344
+    ppo-update-before:                          28160
+    env-step-generate-before:                   22577152
+    env-step-all-after:                         46170112
+    env-step-replay-after:                      23560192
+    env-step-used-after:                        46170112
+    ppo-update-after:                           28180
+    env-step-generate-after:                    22609920
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     39.2486
+    avg_avg_return:                             0.6451694
+    avg_reward_per_step:                        0.10928345
+    proxy_corner:
+      avg_avg_return:                           0.055360027
+      avg_reward_per_step:                      0.0018310547
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     39.2486
+    avg_avg_return:                             0.6451694
+    avg_reward_per_step:                        0.10928345
+    proxy_corner:
+      avg_avg_return:                           0.055360027
+      avg_reward_per_step:                      0.0018310547
+  perf:
+    env_steps_per_second:                       253730.02476864617
+    ppo_updates_per_second:                     41.64142005222861
+    eval-batch-train-duration:                  0.5468452023342252
+    eval-batch-orig-duration:                   0.48831904493272305
+    eval-batch-shift-duration:                  0.4893127344548702
+    eval-batch-tree-duration:                   0.48425136413425207
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    15.940186
+      cheese-corne_dist_wavg:                   23.18737
+      levels_cheese_is_in_corner_avg:           0.017089844
+      mouse-cheese_dist_finite_avg:             16.403809
+      mouse-cheese_dist_finite_wavg:            5.239686
+      mouse-cheese_dist_solvable_avg:           16.403809
+      mouse-cheese_dist_solvable_wavg:          5.2396865
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.60870266
+    visit_patterns:
+      avg_first_visit_time:                     570.3318
+      avg_last_visit_time:                      709.1826
+      num_replay_batches:                       719
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010878358
+    avg_actor_approxkl3:                        0.0012034203
+    avg_actor_clipfrac:                         0.06462403
+    avg_actor_loss:                             -7.340424e-05
+    avg_advantage:                              0.028577112
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.010169386
+    avg_entropy:                                0.45245624
+    avg_grad_norm_pre_clip:                     2.884119
+    avg_loss:                                   0.004558832
+    max:
+      max_actor_approxkl1:                      0.0026805643
+      max_actor_approxkl3:                      0.0031526976
+      max_actor_clipfrac:                       0.1463623
+      max_actor_loss:                           0.0033331811
+      max_advantage:                            0.3570118
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.012085296
+      max_entropy:                              0.4920907
+      max_grad_norm_pre_clip:                   6.715411
+      max_loss:                                 0.008746324
+    std:
+      std_actor_approxkl1:                      0.00081023487
+      std_actor_approxkl3:                      0.0007538419
+      std_actor_clipfrac:                       0.037989948
+      std_actor_loss:                           0.0015402787
+      std_advantage:                            0.1417547
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0009551566
+      std_entropy:                              0.021079222
+      std_grad_norm_pre_clip:                   1.3172897
+      std_loss:                                 0.001745335
+  eval-batch-train:
+    avg_avg_episode_length:                     39.533447
+    avg_avg_return:                             0.69899064
+    avg_benchmark_regret:                       0.28822643
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.09185791
+    proxy_corner:
+      avg_avg_return:                           0.060940206
+      avg_reward_per_step:                      0.0037384033
+  eval-batch-orig:
+    avg_avg_episode_length:                     46.788353
+    avg_avg_return:                             0.6097686
+    avg_benchmark_regret:                       0.37474293
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.0521698
+    proxy_corner:
+      avg_avg_return:                           0.6097686
+      avg_reward_per_step:                      0.0521698
+  eval-batch-shift:
+    avg_avg_episode_length:                     38.129356
+    avg_avg_return:                             0.70662653
+    avg_benchmark_regret:                       0.28087944
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.10110474
+    proxy_corner:
+      avg_avg_return:                           0.06323445
+      avg_reward_per_step:                      0.0024414062
+  eval-batch-tree:
+    avg_avg_episode_length:                     44.81613
+    avg_avg_return:                             0.6603404
+    avg_benchmark_regret:                       0.3242948
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.067718506
+    proxy_corner:
+      avg_avg_return:                           0.6603404
+      avg_reward_per_step:                      0.067718506
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        47185920
+    env-step-replay-before:                     24084480
+    env-step-used-before:                       47185920
+    ppo-update-before:                          28800
+    env-step-generate-before:                   23101440
+    env-step-all-after:                         47218688
+    env-step-replay-after:                      24084480
+    env-step-used-after:                        47218688
+    ppo-update-after:                           28820
+    env-step-generate-after:                    23134208
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     43.029037
+    avg_avg_return:                             0.6181269
+    avg_reward_per_step:                        0.08218384
+    proxy_corner:
+      avg_avg_return:                           0.04119327
+      avg_reward_per_step:                      0.0016174316
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     43.029037
+    avg_avg_return:                             0.6181269
+    avg_reward_per_step:                        0.08218384
+    proxy_corner:
+      avg_avg_return:                           0.04119327
+      avg_reward_per_step:                      0.0016174316
+  perf:
+    env_steps_per_second:                       321269.21702059446
+    ppo_updates_per_second:                     41.93932124366184
+    eval-batch-train-duration:                  0.533842483535409
+    eval-batch-orig-duration:                   0.4795512296259403
+    eval-batch-shift-duration:                  0.4438166841864586
+    eval-batch-tree-duration:                   0.48744327388703823
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    16.060791
+      cheese-corne_dist_wavg:                   18.694033
+      levels_cheese_is_in_corner_avg:           0.016113281
+      mouse-cheese_dist_finite_avg:             16.699219
+      mouse-cheese_dist_finite_wavg:            6.1564093
+      mouse-cheese_dist_solvable_avg:           16.699219
+      mouse-cheese_dist_solvable_wavg:          6.15641
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.60771626
+    visit_patterns:
+      avg_first_visit_time:                     589.35596
+      avg_last_visit_time:                      725.3447
+      num_replay_batches:                       735
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012325475
+    avg_actor_approxkl3:                        0.0010601984
+    avg_actor_clipfrac:                         0.05616455
+    avg_actor_loss:                             -0.0007005971
+    avg_advantage:                              0.020184178
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.009099027
+    avg_entropy:                                0.46563575
+    avg_grad_norm_pre_clip:                     3.0629191
+    avg_loss:                                   0.0033832807
+    max:
+      max_actor_approxkl1:                      0.0030842465
+      max_actor_approxkl3:                      0.0029967558
+      max_actor_clipfrac:                       0.14099121
+      max_actor_loss:                           0.0016091615
+      max_advantage:                            0.36272496
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.011346067
+      max_entropy:                              0.48216757
+      max_grad_norm_pre_clip:                   7.2103534
+      max_loss:                                 0.0058296355
+    std:
+      std_actor_approxkl1:                      0.0008075598
+      std_actor_approxkl3:                      0.000637693
+      std_actor_clipfrac:                       0.03303061
+      std_actor_loss:                           0.0011513963
+      std_advantage:                            0.1342352
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00092267664
+      std_entropy:                              0.013352242
+      std_grad_norm_pre_clip:                   1.6735327
+      std_loss:                                 0.0012308501
+  eval-batch-train:
+    avg_avg_episode_length:                     37.42324
+    avg_avg_return:                             0.7215712
+    avg_benchmark_regret:                       0.26564586
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.09350586
+    proxy_corner:
+      avg_avg_return:                           0.061210297
+      avg_reward_per_step:                      0.0036849976
+  eval-batch-orig:
+    avg_avg_episode_length:                     44.313774
+    avg_avg_return:                             0.6446934
+    avg_benchmark_regret:                       0.33981818
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.05154419
+    proxy_corner:
+      avg_avg_return:                           0.6446934
+      avg_reward_per_step:                      0.05154419
+  eval-batch-shift:
+    avg_avg_episode_length:                     36.502506
+    avg_avg_return:                             0.7365259
+    avg_benchmark_regret:                       0.2509801
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.103004456
+    proxy_corner:
+      avg_avg_return:                           0.063279234
+      avg_reward_per_step:                      0.002494812
+  eval-batch-tree:
+    avg_avg_episode_length:                     41.969505
+    avg_avg_return:                             0.70563555
+    avg_benchmark_regret:                       0.27899966
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.06932831
+    proxy_corner:
+      avg_avg_return:                           0.70563555
+      avg_reward_per_step:                      0.06932831
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        48234496
+    env-step-replay-before:                     24707072
+    env-step-used-before:                       48234496
+    ppo-update-before:                          29440
+    env-step-generate-before:                   23527424
+    env-step-all-after:                         48267264
+    env-step-replay-after:                      24739840
+    env-step-used-after:                        48267264
+    ppo-update-after:                           29460
+    env-step-generate-after:                    23527424
+  train-replay:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     68.960266
+    avg_avg_return:                             0.39489782
+    avg_reward_per_step:                        0.011108398
+    proxy_corner:
+      avg_avg_return:                           0.0645711
+      avg_reward_per_step:                      0.0011291504
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     68.960266
+    avg_avg_return:                             0.39489782
+    avg_reward_per_step:                        0.011108398
+    proxy_corner:
+      avg_avg_return:                           0.0645711
+      avg_reward_per_step:                      0.0011291504
+  perf:
+    env_steps_per_second:                       319317.6107511389
+    ppo_updates_per_second:                     40.79068560442721
+    eval-batch-train-duration:                  0.5291841067373753
+    eval-batch-orig-duration:                   0.4893747577443719
+    eval-batch-shift-duration:                  0.477749309502542
+    eval-batch-tree-duration:                   0.4576576231047511
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    16.190918
+      cheese-corne_dist_wavg:                   7.0109396
+      levels_cheese_is_in_corner_avg:           0.01586914
+      mouse-cheese_dist_finite_avg:             16.981445
+      mouse-cheese_dist_finite_wavg:            8.003381
+      mouse-cheese_dist_solvable_avg:           16.981445
+      mouse-cheese_dist_solvable_wavg:          8.00338
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000008
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.5728513
+    visit_patterns:
+      avg_first_visit_time:                     603.4617
+      avg_last_visit_time:                      745.34717
+      num_replay_batches:                       755
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0017501864
+    avg_actor_approxkl3:                        0.0015850635
+    avg_actor_clipfrac:                         0.08537598
+    avg_actor_loss:                             -0.0005762417
+    avg_advantage:                              -0.015611994
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.01122778
+    avg_entropy:                                0.5726907
+    avg_grad_norm_pre_clip:                     2.7304072
+    avg_loss:                                   0.0044649574
+    max:
+      max_actor_approxkl1:                      0.004018724
+      max_actor_approxkl3:                      0.003215801
+      max_actor_clipfrac:                       0.17346191
+      max_actor_loss:                           0.0024045855
+      max_advantage:                            0.31591123
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0138599165
+      max_entropy:                              0.6026038
+      max_grad_norm_pre_clip:                   4.833593
+      max_loss:                                 0.0070367227
+    std:
+      std_actor_approxkl1:                      0.00097455684
+      std_actor_approxkl3:                      0.0007307376
+      std_actor_clipfrac:                       0.04117425
+      std_actor_loss:                           0.0014951463
+      std_advantage:                            0.14969236
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.000971274
+      std_entropy:                              0.013481998
+      std_grad_norm_pre_clip:                   1.0535583
+      std_loss:                                 0.0014665922
+  eval-batch-train:
+    avg_avg_episode_length:                     36.213852
+    avg_avg_return:                             0.7413389
+    avg_benchmark_regret:                       0.24587813
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.091926575
+    proxy_corner:
+      avg_avg_return:                           0.053733613
+      avg_reward_per_step:                      0.0030670166
+  eval-batch-orig:
+    avg_avg_episode_length:                     43.36686
+    avg_avg_return:                             0.6628761
+    avg_benchmark_regret:                       0.32163543
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.05101013
+    proxy_corner:
+      avg_avg_return:                           0.6628761
+      avg_reward_per_step:                      0.05101013
+  eval-batch-shift:
+    avg_avg_episode_length:                     36.979816
+    avg_avg_return:                             0.73562086
+    avg_benchmark_regret:                       0.2518851
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.102386475
+    proxy_corner:
+      avg_avg_return:                           0.05662537
+      avg_reward_per_step:                      0.001876831
+  eval-batch-tree:
+    avg_avg_episode_length:                     40.010803
+    avg_avg_return:                             0.7198319
+    avg_benchmark_regret:                       0.26480332
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.06820679
+    proxy_corner:
+      avg_avg_return:                           0.7198319
+      avg_reward_per_step:                      0.06820679
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        49283072
+    env-step-replay-before:                     25493504
+    env-step-used-before:                       49283072
+    ppo-update-before:                          30080
+    env-step-generate-before:                   23789568
+    env-step-all-after:                         49315840
+    env-step-replay-after:                      25526272
+    env-step-used-after:                        49315840
+    ppo-update-after:                           30100
+    env-step-generate-after:                    23789568
+  train-replay:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     66.01223
+    avg_avg_return:                             0.463701
+    avg_reward_per_step:                        0.0126953125
+    proxy_corner:
+      avg_avg_return:                           0.091608405
+      avg_reward_per_step:                      0.0017089844
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     66.01223
+    avg_avg_return:                             0.463701
+    avg_reward_per_step:                        0.0126953125
+    proxy_corner:
+      avg_avg_return:                           0.091608405
+      avg_reward_per_step:                      0.0017089844
+  perf:
+    env_steps_per_second:                       306307.2410289683
+    ppo_updates_per_second:                     42.13088193598855
+    eval-batch-train-duration:                  0.5199719183146954
+    eval-batch-orig-duration:                   0.4825444892048836
+    eval-batch-shift-duration:                  0.45384790282696486
+    eval-batch-tree-duration:                   0.48111965227872133
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    16.191895
+      cheese-corne_dist_wavg:                   7.017703
+      levels_cheese_is_in_corner_avg:           0.01586914
+      mouse-cheese_dist_finite_avg:             17.114746
+      mouse-cheese_dist_finite_wavg:            8.009241
+      mouse-cheese_dist_solvable_avg:           17.114746
+      mouse-cheese_dist_solvable_wavg:          8.009242
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.5469903
+    visit_patterns:
+      avg_first_visit_time:                     615.8103
+      avg_last_visit_time:                      769.1924
+      num_replay_batches:                       779
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014586208
+    avg_actor_approxkl3:                        0.0012193604
+    avg_actor_clipfrac:                         0.06404419
+    avg_actor_loss:                             -0.00076586753
+    avg_advantage:                              0.0070695803
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.009222947
+    avg_entropy:                                0.52183616
+    avg_grad_norm_pre_clip:                     2.810163
+    avg_loss:                                   0.0033237697
+    max:
+      max_actor_approxkl1:                      0.0035158633
+      max_actor_approxkl3:                      0.0030126418
+      max_actor_clipfrac:                       0.13183594
+      max_actor_loss:                           0.0011178255
+      max_advantage:                            0.34204704
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.011733333
+      max_entropy:                              0.5436151
+      max_grad_norm_pre_clip:                   6.340998
+      max_loss:                                 0.0052997097
+    std:
+      std_actor_approxkl1:                      0.0010495895
+      std_actor_approxkl3:                      0.00072161073
+      std_actor_clipfrac:                       0.037602436
+      std_actor_loss:                           0.0010871198
+      std_advantage:                            0.136018
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.001134319
+      std_entropy:                              0.0096488455
+      std_grad_norm_pre_clip:                   1.4388157
+      std_loss:                                 0.0011520135
+  eval-batch-train:
+    avg_avg_episode_length:                     35.046425
+    avg_avg_return:                             0.7518083
+    avg_benchmark_regret:                       0.2354088
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.09452057
+    proxy_corner:
+      avg_avg_return:                           0.07856564
+      avg_reward_per_step:                      0.0044174194
+  eval-batch-orig:
+    avg_avg_episode_length:                     43.054466
+    avg_avg_return:                             0.6751355
+    avg_benchmark_regret:                       0.30937606
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.056762695
+    proxy_corner:
+      avg_avg_return:                           0.6751355
+      avg_reward_per_step:                      0.056762695
+  eval-batch-shift:
+    avg_avg_episode_length:                     35.522644
+    avg_avg_return:                             0.74350274
+    avg_benchmark_regret:                       0.24400327
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.1038208
+    proxy_corner:
+      avg_avg_return:                           0.073252976
+      avg_reward_per_step:                      0.002708435
+  eval-batch-tree:
+    avg_avg_episode_length:                     40.58789
+    avg_avg_return:                             0.7261335
+    avg_benchmark_regret:                       0.25850165
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.07296753
+    proxy_corner:
+      avg_avg_return:                           0.7261335
+      avg_reward_per_step:                      0.07296753
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        50331648
+    env-step-replay-before:                     26116096
+    env-step-used-before:                       50331648
+    ppo-update-before:                          30720
+    env-step-generate-before:                   24215552
+    env-step-all-after:                         50364416
+    env-step-replay-after:                      26116096
+    env-step-used-after:                        50364416
+    ppo-update-after:                           30740
+    env-step-generate-after:                    24248320
+  train-generate:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     35.968056
+    avg_avg_return:                             0.68685514
+    avg_reward_per_step:                        0.09118652
+    proxy_corner:
+      avg_avg_return:                           0.062288318
+      avg_reward_per_step:                      0.0022888184
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     35.968056
+    avg_avg_return:                             0.68685514
+    avg_reward_per_step:                        0.09118652
+    proxy_corner:
+      avg_avg_return:                           0.062288318
+      avg_reward_per_step:                      0.0022888184
+  perf:
+    env_steps_per_second:                       317427.83905842574
+    ppo_updates_per_second:                     42.70942492792165
+    eval-batch-train-duration:                  0.5256936931982636
+    eval-batch-orig-duration:                   0.48340276163071394
+    eval-batch-shift-duration:                  0.4804101912304759
+    eval-batch-tree-duration:                   0.46696891263127327
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    16.229736
+      cheese-corne_dist_wavg:                   23.213192
+      levels_cheese_is_in_corner_avg:           0.017089844
+      mouse-cheese_dist_finite_avg:             17.325195
+      mouse-cheese_dist_finite_wavg:            5.3287797
+      mouse-cheese_dist_solvable_avg:           17.325195
+      mouse-cheese_dist_solvable_wavg:          5.32878
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.55704284
+    visit_patterns:
+      avg_first_visit_time:                     632.2334
+      avg_last_visit_time:                      787.303
+      num_replay_batches:                       797
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0015138941
+    avg_actor_approxkl3:                        0.0012640645
+    avg_actor_clipfrac:                         0.06259155
+    avg_actor_loss:                             -0.0006037742
+    avg_advantage:                              0.03407902
+    avg_critic_clipfrac:                        0.00078735355
+    avg_critic_loss:                            0.007870914
+    avg_entropy:                                0.3702685
+    avg_grad_norm_pre_clip:                     2.9839773
+    avg_loss:                                   0.0029614144
+    max:
+      max_actor_approxkl1:                      0.0038566906
+      max_actor_approxkl3:                      0.0027612662
+      max_actor_clipfrac:                       0.11975098
+      max_actor_loss:                           0.00079175085
+      max_advantage:                            0.31748837
+      max_critic_clipfrac:                      0.008056641
+      max_critic_loss:                          0.0103488555
+      max_entropy:                              0.38902837
+      max_grad_norm_pre_clip:                   5.8415785
+      max_loss:                                 0.0049554165
+    std:
+      std_actor_approxkl1:                      0.0010166629
+      std_actor_approxkl3:                      0.00068487943
+      std_actor_clipfrac:                       0.03220484
+      std_actor_loss:                           0.000824203
+      std_advantage:                            0.1242073
+      std_critic_clipfrac:                      0.0018336877
+      std_critic_loss:                          0.0009876328
+      std_entropy:                              0.01219544
+      std_grad_norm_pre_clip:                   1.3480365
+      std_loss:                                 0.0008791768
+  eval-batch-train:
+    avg_avg_episode_length:                     33.61467
+    avg_avg_return:                             0.75806594
+    avg_benchmark_regret:                       0.22915113
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.0976181
+    proxy_corner:
+      avg_avg_return:                           0.065401256
+      avg_reward_per_step:                      0.003692627
+  eval-batch-orig:
+    avg_avg_episode_length:                     42.092518
+    avg_avg_return:                             0.6901549
+    avg_benchmark_regret:                       0.29435667
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.05493164
+    proxy_corner:
+      avg_avg_return:                           0.6901549
+      avg_reward_per_step:                      0.05493164
+  eval-batch-shift:
+    avg_avg_episode_length:                     34.83357
+    avg_avg_return:                             0.7448641
+    avg_benchmark_regret:                       0.24264185
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.10747528
+    proxy_corner:
+      avg_avg_return:                           0.06704005
+      avg_reward_per_step:                      0.002632141
+  eval-batch-tree:
+    avg_avg_episode_length:                     39.86868
+    avg_avg_return:                             0.7246536
+    avg_benchmark_regret:                       0.2599816
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.07107544
+    proxy_corner:
+      avg_avg_return:                           0.7246536
+      avg_reward_per_step:                      0.07107544
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        51380224
+    env-step-replay-before:                     26640384
+    env-step-used-before:                       51380224
+    ppo-update-before:                          31360
+    env-step-generate-before:                   24739840
+    env-step-all-after:                         51412992
+    env-step-replay-after:                      26640384
+    env-step-used-after:                        51412992
+    ppo-update-after:                           31380
+    env-step-generate-after:                    24772608
+  train-generate:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     38.02546
+    avg_avg_return:                             0.6594661
+    avg_reward_per_step:                        0.09609985
+    proxy_corner:
+      avg_avg_return:                           0.04209238
+      avg_reward_per_step:                      0.0022583008
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     38.02546
+    avg_avg_return:                             0.6594661
+    avg_reward_per_step:                        0.09609985
+    proxy_corner:
+      avg_avg_return:                           0.04209238
+      avg_reward_per_step:                      0.0022583008
+  perf:
+    env_steps_per_second:                       291827.2834443622
+    ppo_updates_per_second:                     42.561933237463506
+    eval-batch-train-duration:                  0.530614192597568
+    eval-batch-orig-duration:                   0.48751016333699226
+    eval-batch-shift-duration:                  0.497769339941442
+    eval-batch-tree-duration:                   0.5022839801385999
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    16.256348
+      cheese-corne_dist_wavg:                   23.233454
+      levels_cheese_is_in_corner_avg:           0.018310547
+      mouse-cheese_dist_finite_avg:             17.548584
+      mouse-cheese_dist_finite_wavg:            5.3573184
+      mouse-cheese_dist_solvable_avg:           17.548584
+      mouse-cheese_dist_solvable_wavg:          5.35732
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5685369
+    visit_patterns:
+      avg_first_visit_time:                     652.4016
+      avg_last_visit_time:                      803.2268
+      num_replay_batches:                       813
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0015275715
+    avg_actor_approxkl3:                        0.0009787782
+    avg_actor_clipfrac:                         0.048669435
+    avg_actor_loss:                             -0.0010821915
+    avg_advantage:                              0.05326418
+    avg_critic_clipfrac:                        0.040216066
+    avg_critic_loss:                            0.009599833
+    avg_entropy:                                0.37023106
+    avg_grad_norm_pre_clip:                     1.9449161
+    avg_loss:                                   0.003347493
+    max:
+      max_actor_approxkl1:                      0.0038950369
+      max_actor_approxkl3:                      0.002702987
+      max_actor_clipfrac:                       0.11828613
+      max_actor_loss:                           0.000926964
+      max_advantage:                            0.34101743
+      max_critic_clipfrac:                      0.39526367
+      max_critic_loss:                          0.011941111
+      max_entropy:                              0.4064566
+      max_grad_norm_pre_clip:                   3.8894935
+      max_loss:                                 0.0055417367
+    std:
+      std_actor_approxkl1:                      0.0009429725
+      std_actor_approxkl3:                      0.00054746086
+      std_actor_clipfrac:                       0.024457453
+      std_actor_loss:                           0.0011355863
+      std_advantage:                            0.13602869
+      std_critic_clipfrac:                      0.101899296
+      std_critic_loss:                          0.0015228932
+      std_entropy:                              0.020448986
+      std_grad_norm_pre_clip:                   0.8995973
+      std_loss:                                 0.0015068318
+  eval-batch-train:
+    avg_avg_episode_length:                     34.284954
+    avg_avg_return:                             0.7598953
+    avg_benchmark_regret:                       0.22732177
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.0960083
+    proxy_corner:
+      avg_avg_return:                           0.047362905
+      avg_reward_per_step:                      0.0031433105
+  eval-batch-orig:
+    avg_avg_episode_length:                     42.035282
+    avg_avg_return:                             0.70706606
+    avg_benchmark_regret:                       0.2774455
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.052345276
+    proxy_corner:
+      avg_avg_return:                           0.70706606
+      avg_reward_per_step:                      0.052345276
+  eval-batch-shift:
+    avg_avg_episode_length:                     35.554497
+    avg_avg_return:                             0.74332964
+    avg_benchmark_regret:                       0.2441763
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.107788086
+    proxy_corner:
+      avg_avg_return:                           0.05384804
+      avg_reward_per_step:                      0.0019836426
+  eval-batch-tree:
+    avg_avg_episode_length:                     39.26261
+    avg_avg_return:                             0.7355535
+    avg_benchmark_regret:                       0.24908169
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.068573
+    proxy_corner:
+      avg_avg_return:                           0.7355535
+      avg_reward_per_step:                      0.068573
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        52428800
+    env-step-replay-before:                     27164672
+    env-step-used-before:                       52428800
+    ppo-update-before:                          32000
+    env-step-generate-before:                   25264128
+    env-step-all-after:                         52461568
+    env-step-replay-after:                      27197440
+    env-step-used-after:                        52461568
+    ppo-update-after:                           32020
+    env-step-generate-after:                    25264128
+  train-replay:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     67.827835
+    avg_avg_return:                             0.37871033
+    avg_reward_per_step:                        0.01171875
+    proxy_corner:
+      avg_avg_return:                           0.06394755
+      avg_reward_per_step:                      0.001159668
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     67.827835
+    avg_avg_return:                             0.37871033
+    avg_reward_per_step:                        0.01171875
+    proxy_corner:
+      avg_avg_return:                           0.06394755
+      avg_reward_per_step:                      0.001159668
+  perf:
+    env_steps_per_second:                       295562.6438275367
+    ppo_updates_per_second:                     42.10676334962941
+    eval-batch-train-duration:                  0.5377547340467572
+    eval-batch-orig-duration:                   0.4854397866874933
+    eval-batch-shift-duration:                  0.4835088336840272
+    eval-batch-tree-duration:                   0.4528378341346979
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    16.197754
+      cheese-corne_dist_wavg:                   15.997974
+      levels_cheese_is_in_corner_avg:           0.01928711
+      mouse-cheese_dist_finite_avg:             17.76294
+      mouse-cheese_dist_finite_wavg:            8.067675
+      mouse-cheese_dist_solvable_avg:           17.76294
+      mouse-cheese_dist_solvable_wavg:          8.067676
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.59229136
+    visit_patterns:
+      avg_first_visit_time:                     672.0503
+      avg_last_visit_time:                      820.21045
+      num_replay_batches:                       830
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009856714
+    avg_actor_approxkl3:                        0.0010052256
+    avg_actor_clipfrac:                         0.051098634
+    avg_actor_loss:                             -0.0014276289
+    avg_advantage:                              -0.023054237
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0118099
+    avg_entropy:                                0.498494
+    avg_grad_norm_pre_clip:                     2.061742
+    avg_loss:                                   0.0039788275
+    max:
+      max_actor_approxkl1:                      0.004437363
+      max_actor_approxkl3:                      0.0020445478
+      max_actor_clipfrac:                       0.10253906
+      max_actor_loss:                           0.0005480349
+      max_advantage:                            0.33116788
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0132344365
+      max_entropy:                              0.51201165
+      max_grad_norm_pre_clip:                   3.9799173
+      max_loss:                                 0.0063246
+    std:
+      std_actor_approxkl1:                      0.0009952573
+      std_actor_approxkl3:                      0.000600294
+      std_actor_clipfrac:                       0.03293658
+      std_actor_loss:                           0.00106635
+      std_advantage:                            0.1542912
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00072084164
+      std_entropy:                              0.008867146
+      std_grad_norm_pre_clip:                   0.9684271
+      std_loss:                                 0.0011999565
+  eval-batch-train:
+    avg_avg_episode_length:                     33.812202
+    avg_avg_return:                             0.7653947
+    avg_benchmark_regret:                       0.22182238
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.09552765
+    proxy_corner:
+      avg_avg_return:                           0.05430076
+      avg_reward_per_step:                      0.0033950806
+  eval-batch-orig:
+    avg_avg_episode_length:                     41.37194
+    avg_avg_return:                             0.69758713
+    avg_benchmark_regret:                       0.28692442
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.054992676
+    proxy_corner:
+      avg_avg_return:                           0.69758713
+      avg_reward_per_step:                      0.054992676
+  eval-batch-shift:
+    avg_avg_episode_length:                     34.811752
+    avg_avg_return:                             0.74888897
+    avg_benchmark_regret:                       0.23861702
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.10649872
+    proxy_corner:
+      avg_avg_return:                           0.06082481
+      avg_reward_per_step:                      0.0022583008
+  eval-batch-tree:
+    avg_avg_episode_length:                     39.255226
+    avg_avg_return:                             0.7335315
+    avg_benchmark_regret:                       0.25110376
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.07148743
+    proxy_corner:
+      avg_avg_return:                           0.7335315
+      avg_reward_per_step:                      0.07148743
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        53477376
+    env-step-replay-before:                     27623424
+    env-step-used-before:                       53477376
+    ppo-update-before:                          32640
+    env-step-generate-before:                   25853952
+    env-step-all-after:                         53510144
+    env-step-replay-after:                      27623424
+    env-step-used-after:                        53510144
+    ppo-update-after:                           32660
+    env-step-generate-after:                    25886720
+  train-generate:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     37.01526
+    avg_avg_return:                             0.68740547
+    avg_reward_per_step:                        0.101257324
+    proxy_corner:
+      avg_avg_return:                           0.037372377
+      avg_reward_per_step:                      0.0026550293
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     37.01526
+    avg_avg_return:                             0.68740547
+    avg_reward_per_step:                        0.101257324
+    proxy_corner:
+      avg_avg_return:                           0.037372377
+      avg_reward_per_step:                      0.0026550293
+  perf:
+    env_steps_per_second:                       317719.60007443
+    ppo_updates_per_second:                     41.64437338635308
+    eval-batch-train-duration:                  0.5413998113945127
+    eval-batch-orig-duration:                   0.4870179118588567
+    eval-batch-shift-duration:                  0.49492273200303316
+    eval-batch-tree-duration:                   0.4897041479125619
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    16.324707
+      cheese-corne_dist_wavg:                   16.021223
+      levels_cheese_is_in_corner_avg:           0.020019531
+      mouse-cheese_dist_finite_avg:             18.000244
+      mouse-cheese_dist_finite_wavg:            8.0957155
+      mouse-cheese_dist_solvable_avg:           18.000244
+      mouse-cheese_dist_solvable_wavg:          8.095717
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.5716682
+    visit_patterns:
+      avg_first_visit_time:                     690.16895
+      avg_last_visit_time:                      833.3408
+      num_replay_batches:                       843
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001553929
+    avg_actor_approxkl3:                        0.0013264472
+    avg_actor_clipfrac:                         0.06843872
+    avg_actor_loss:                             3.7552974e-05
+    avg_advantage:                              0.045569364
+    avg_critic_clipfrac:                        0.004589844
+    avg_critic_loss:                            0.007661991
+    avg_entropy:                                0.35569435
+    avg_grad_norm_pre_clip:                     3.4147317
+    avg_loss:                                   0.0035128545
+    max:
+      max_actor_approxkl1:                      0.0035872404
+      max_actor_approxkl3:                      0.0031030397
+      max_actor_clipfrac:                       0.13537598
+      max_actor_loss:                           0.0043670386
+      max_advantage:                            0.3013181
+      max_critic_clipfrac:                      0.036499023
+      max_critic_loss:                          0.0104145715
+      max_entropy:                              0.38047683
+      max_grad_norm_pre_clip:                   6.919824
+      max_loss:                                 0.007883191
+    std:
+      std_actor_approxkl1:                      0.00090889225
+      std_actor_approxkl3:                      0.0006109366
+      std_actor_clipfrac:                       0.028578619
+      std_actor_loss:                           0.0015157358
+      std_advantage:                            0.12146894
+      std_critic_clipfrac:                      0.010553995
+      std_critic_loss:                          0.0011648993
+      std_entropy:                              0.015222827
+      std_grad_norm_pre_clip:                   1.5229739
+      std_loss:                                 0.0016617079
+  eval-batch-train:
+    avg_avg_episode_length:                     32.79048
+    avg_avg_return:                             0.76990473
+    avg_benchmark_regret:                       0.21731237
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.10051727
+    proxy_corner:
+      avg_avg_return:                           0.059917096
+      avg_reward_per_step:                      0.0040130615
+  eval-batch-orig:
+    avg_avg_episode_length:                     41.83037
+    avg_avg_return:                             0.70808756
+    avg_benchmark_regret:                       0.27642396
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.05809021
+    proxy_corner:
+      avg_avg_return:                           0.70808756
+      avg_reward_per_step:                      0.05809021
+  eval-batch-shift:
+    avg_avg_episode_length:                     33.654938
+    avg_avg_return:                             0.76767385
+    avg_benchmark_regret:                       0.21983208
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.10845947
+    proxy_corner:
+      avg_avg_return:                           0.0608449
+      avg_reward_per_step:                      0.0023498535
+  eval-batch-tree:
+    avg_avg_episode_length:                     38.096996
+    avg_avg_return:                             0.7452557
+    avg_benchmark_regret:                       0.23937951
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0748291
+    proxy_corner:
+      avg_avg_return:                           0.7452557
+      avg_reward_per_step:                      0.0748291
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        54525952
+    env-step-replay-before:                     28049408
+    env-step-used-before:                       54525952
+    ppo-update-before:                          33280
+    env-step-generate-before:                   26476544
+    env-step-all-after:                         54558720
+    env-step-replay-after:                      28049408
+    env-step-used-after:                        54558720
+    ppo-update-after:                           33300
+    env-step-generate-after:                    26509312
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     36.26005
+    avg_avg_return:                             0.6835872
+    avg_reward_per_step:                        0.1104126
+    proxy_corner:
+      avg_avg_return:                           0.059080694
+      avg_reward_per_step:                      0.0024414062
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     36.26005
+    avg_avg_return:                             0.6835872
+    avg_reward_per_step:                        0.1104126
+    proxy_corner:
+      avg_avg_return:                           0.059080694
+      avg_reward_per_step:                      0.0024414062
+  perf:
+    env_steps_per_second:                       241913.13206662372
+    ppo_updates_per_second:                     41.38709252121738
+    eval-batch-train-duration:                  0.5486389994621277
+    eval-batch-orig-duration:                   0.4876124393194914
+    eval-batch-shift-duration:                  0.4931104015558958
+    eval-batch-tree-duration:                   0.4872041158378124
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    16.461426
+      cheese-corne_dist_wavg:                   25.041117
+      levels_cheese_is_in_corner_avg:           0.01977539
+      mouse-cheese_dist_finite_avg:             18.247314
+      mouse-cheese_dist_finite_wavg:            8.126431
+      mouse-cheese_dist_solvable_avg:           18.247314
+      mouse-cheese_dist_solvable_wavg:          8.126431
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.57717526
+    visit_patterns:
+      avg_first_visit_time:                     708.4072
+      avg_last_visit_time:                      846.25635
+      num_replay_batches:                       856
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001730538
+    avg_actor_approxkl3:                        0.0014500767
+    avg_actor_clipfrac:                         0.067919925
+    avg_actor_loss:                             -0.00061938
+    avg_advantage:                              0.02479841
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0076845563
+    avg_entropy:                                0.34546706
+    avg_grad_norm_pre_clip:                     2.5501337
+    avg_loss:                                   0.0028774312
+    max:
+      max_actor_approxkl1:                      0.0041128295
+      max_actor_approxkl3:                      0.0027090448
+      max_actor_clipfrac:                       0.10839844
+      max_actor_loss:                           0.0016007721
+      max_advantage:                            0.26436824
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.009631949
+      max_entropy:                              0.37699026
+      max_grad_norm_pre_clip:                   4.2883368
+      max_loss:                                 0.0056163813
+    std:
+      std_actor_approxkl1:                      0.0011096757
+      std_actor_approxkl3:                      0.00073008845
+      std_actor_clipfrac:                       0.031592198
+      std_actor_loss:                           0.0011738088
+      std_advantage:                            0.123338856
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011907575
+      std_entropy:                              0.014840514
+      std_grad_norm_pre_clip:                   0.9847318
+      std_loss:                                 0.0014423646
+  eval-batch-train:
+    avg_avg_episode_length:                     32.008472
+    avg_avg_return:                             0.77914083
+    avg_benchmark_regret:                       0.20807627
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.099487305
+    proxy_corner:
+      avg_avg_return:                           0.059888665
+      avg_reward_per_step:                      0.003868103
+  eval-batch-orig:
+    avg_avg_episode_length:                     40.643604
+    avg_avg_return:                             0.7057395
+    avg_benchmark_regret:                       0.27877206
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.06187439
+    proxy_corner:
+      avg_avg_return:                           0.7057395
+      avg_reward_per_step:                      0.06187439
+  eval-batch-shift:
+    avg_avg_episode_length:                     32.78233
+    avg_avg_return:                             0.7742673
+    avg_benchmark_regret:                       0.21323864
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.11125946
+    proxy_corner:
+      avg_avg_return:                           0.05717273
+      avg_reward_per_step:                      0.002380371
+  eval-batch-tree:
+    avg_avg_episode_length:                     36.69671
+    avg_avg_return:                             0.7527082
+    avg_benchmark_regret:                       0.23192698
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.07548523
+    proxy_corner:
+      avg_avg_return:                           0.7527082
+      avg_reward_per_step:                      0.07548523
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        55574528
+    env-step-replay-before:                     28442624
+    env-step-used-before:                       55574528
+    ppo-update-before:                          33920
+    env-step-generate-before:                   27131904
+    env-step-all-after:                         55607296
+    env-step-replay-after:                      28475392
+    env-step-used-after:                        55607296
+    ppo-update-after:                           33940
+    env-step-generate-after:                    27131904
+  train-replay:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     66.250244
+    avg_avg_return:                             0.41509402
+    avg_reward_per_step:                        0.012786865
+    proxy_corner:
+      avg_avg_return:                           0.033449747
+      avg_reward_per_step:                      0.0008544922
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     66.250244
+    avg_avg_return:                             0.41509402
+    avg_reward_per_step:                        0.012786865
+    proxy_corner:
+      avg_avg_return:                           0.033449747
+      avg_reward_per_step:                      0.0008544922
+  perf:
+    env_steps_per_second:                       320407.8732411231
+    ppo_updates_per_second:                     42.282488858115094
+    eval-batch-train-duration:                  0.5192905394360423
+    eval-batch-orig-duration:                   0.4780903784558177
+    eval-batch-shift-duration:                  0.44850185979157686
+    eval-batch-tree-duration:                   0.49289008043706417
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    16.526367
+      cheese-corne_dist_wavg:                   19.633408
+      levels_cheese_is_in_corner_avg:           0.020019531
+      mouse-cheese_dist_finite_avg:             18.48706
+      mouse-cheese_dist_finite_wavg:            8.146394
+      mouse-cheese_dist_solvable_avg:           18.48706
+      mouse-cheese_dist_solvable_wavg:          8.146397
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999997
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.58774203
+    visit_patterns:
+      avg_first_visit_time:                     724.9287
+      avg_last_visit_time:                      859.2117
+      num_replay_batches:                       869
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0015140035
+    avg_actor_approxkl3:                        0.0013317867
+    avg_actor_clipfrac:                         0.06918335
+    avg_actor_loss:                             -0.0001847297
+    avg_advantage:                              -0.032760464
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.011524773
+    avg_entropy:                                0.44658303
+    avg_grad_norm_pre_clip:                     2.8686264
+    avg_loss:                                   0.0051310747
+    max:
+      max_actor_approxkl1:                      0.004532917
+      max_actor_approxkl3:                      0.00383127
+      max_actor_clipfrac:                       0.14709473
+      max_actor_loss:                           0.0032612085
+      max_advantage:                            0.28740448
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.015103935
+      max_entropy:                              0.48018888
+      max_grad_norm_pre_clip:                   4.833927
+      max_loss:                                 0.007271903
+    std:
+      std_actor_approxkl1:                      0.0011148145
+      std_actor_approxkl3:                      0.0008665424
+      std_actor_clipfrac:                       0.039315995
+      std_actor_loss:                           0.0013801119
+      std_advantage:                            0.15171146
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0014279407
+      std_entropy:                              0.015107803
+      std_grad_norm_pre_clip:                   0.98156357
+      std_loss:                                 0.0014317418
+  eval-batch-train:
+    avg_avg_episode_length:                     32.7209
+    avg_avg_return:                             0.7647904
+    avg_benchmark_regret:                       0.22242665
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.102508545
+    proxy_corner:
+      avg_avg_return:                           0.050393686
+      avg_reward_per_step:                      0.0035934448
+  eval-batch-orig:
+    avg_avg_episode_length:                     40.3649
+    avg_avg_return:                             0.70590794
+    avg_benchmark_regret:                       0.27860355
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.059387207
+    proxy_corner:
+      avg_avg_return:                           0.70590794
+      avg_reward_per_step:                      0.059387207
+  eval-batch-shift:
+    avg_avg_episode_length:                     33.244568
+    avg_avg_return:                             0.76871514
+    avg_benchmark_regret:                       0.21879086
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.11457825
+    proxy_corner:
+      avg_avg_return:                           0.055053875
+      avg_reward_per_step:                      0.002243042
+  eval-batch-tree:
+    avg_avg_episode_length:                     36.20697
+    avg_avg_return:                             0.75507796
+    avg_benchmark_regret:                       0.22955729
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.07494354
+    proxy_corner:
+      avg_avg_return:                           0.75507796
+      avg_reward_per_step:                      0.07494354
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        56623104
+    env-step-replay-before:                     28966912
+    env-step-used-before:                       56623104
+    ppo-update-before:                          34560
+    env-step-generate-before:                   27656192
+    env-step-all-after:                         56655872
+    env-step-replay-after:                      28999680
+    env-step-used-after:                        56655872
+    ppo-update-after:                           34580
+    env-step-generate-after:                    27656192
+  train-replay:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     68.280495
+    avg_avg_return:                             0.37070155
+    avg_reward_per_step:                        0.010772705
+    proxy_corner:
+      avg_avg_return:                           0.06199672
+      avg_reward_per_step:                      0.001159668
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     68.280495
+    avg_avg_return:                             0.37070155
+    avg_reward_per_step:                        0.010772705
+    proxy_corner:
+      avg_avg_return:                           0.06199672
+      avg_reward_per_step:                      0.001159668
+  perf:
+    env_steps_per_second:                       296559.43188121106
+    ppo_updates_per_second:                     41.74211644222221
+    eval-batch-train-duration:                  0.534403414465487
+    eval-batch-orig-duration:                   0.4807713907212019
+    eval-batch-shift-duration:                  0.47956683300435543
+    eval-batch-tree-duration:                   0.4494780767709017
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    16.623535
+      cheese-corne_dist_wavg:                   7.06257
+      levels_cheese_is_in_corner_avg:           0.020019531
+      mouse-cheese_dist_finite_avg:             18.733398
+      mouse-cheese_dist_finite_wavg:            8.158519
+      mouse-cheese_dist_solvable_avg:           18.733398
+      mouse-cheese_dist_solvable_wavg:          8.15852
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.5940813
+    visit_patterns:
+      avg_first_visit_time:                     739.1477
+      avg_last_visit_time:                      875.333
+      num_replay_batches:                       885
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0022750543
+    avg_actor_approxkl3:                        0.0023004776
+    avg_actor_clipfrac:                         0.104522705
+    avg_actor_loss:                             0.0009700358
+    avg_advantage:                              -0.029835157
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0126394555
+    avg_entropy:                                0.4490133
+    avg_grad_norm_pre_clip:                     4.045979
+    avg_loss:                                   0.0068407506
+    max:
+      max_actor_approxkl1:                      0.0058843642
+      max_actor_approxkl3:                      0.0064427797
+      max_actor_clipfrac:                       0.20788574
+      max_actor_loss:                           0.0063458234
+      max_advantage:                            0.29215735
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.015046473
+      max_entropy:                              0.473916
+      max_grad_norm_pre_clip:                   8.403669
+      max_loss:                                 0.0120249875
+    std:
+      std_actor_approxkl1:                      0.0017277271
+      std_actor_approxkl3:                      0.0017181613
+      std_actor_clipfrac:                       0.057280622
+      std_actor_loss:                           0.0018165284
+      std_advantage:                            0.15814559
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011449959
+      std_entropy:                              0.0110599995
+      std_grad_norm_pre_clip:                   1.6453056
+      std_loss:                                 0.0017502699
+  eval-batch-train:
+    avg_avg_episode_length:                     31.82101
+    avg_avg_return:                             0.7842907
+    avg_benchmark_regret:                       0.20292637
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.10254669
+    proxy_corner:
+      avg_avg_return:                           0.057384033
+      avg_reward_per_step:                      0.004562378
+  eval-batch-orig:
+    avg_avg_episode_length:                     41.177994
+    avg_avg_return:                             0.6839758
+    avg_benchmark_regret:                       0.30053568
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.060180664
+    proxy_corner:
+      avg_avg_return:                           0.6839758
+      avg_reward_per_step:                      0.060180664
+  eval-batch-shift:
+    avg_avg_episode_length:                     33.025433
+    avg_avg_return:                             0.77884495
+    avg_benchmark_regret:                       0.20866105
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.11210632
+    proxy_corner:
+      avg_avg_return:                           0.05255069
+      avg_reward_per_step:                      0.0021820068
+  eval-batch-tree:
+    avg_avg_episode_length:                     36.91259
+    avg_avg_return:                             0.7557232
+    avg_benchmark_regret:                       0.22891201
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0767746
+    proxy_corner:
+      avg_avg_return:                           0.7557232
+      avg_reward_per_step:                      0.0767746
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        57671680
+    env-step-replay-before:                     29458432
+    env-step-used-before:                       57671680
+    ppo-update-before:                          35200
+    env-step-generate-before:                   28213248
+    env-step-all-after:                         57704448
+    env-step-replay-after:                      29458432
+    env-step-used-after:                        57704448
+    ppo-update-after:                           35220
+    env-step-generate-after:                    28246016
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     38.028835
+    avg_avg_return:                             0.66223085
+    avg_reward_per_step:                        0.084228516
+    proxy_corner:
+      avg_avg_return:                           0.023445453
+      avg_reward_per_step:                      0.0008239746
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     38.028835
+    avg_avg_return:                             0.66223085
+    avg_reward_per_step:                        0.084228516
+    proxy_corner:
+      avg_avg_return:                           0.023445453
+      avg_reward_per_step:                      0.0008239746
+  perf:
+    env_steps_per_second:                       321682.108358456
+    ppo_updates_per_second:                     44.36544072158692
+    eval-batch-train-duration:                  0.5146624650806189
+    eval-batch-orig-duration:                   0.47701384499669075
+    eval-batch-shift-duration:                  0.479876852594316
+    eval-batch-tree-duration:                   0.49881742242723703
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    16.598633
+      cheese-corne_dist_wavg:                   7.0788975
+      levels_cheese_is_in_corner_avg:           0.01977539
+      mouse-cheese_dist_finite_avg:             18.921143
+      mouse-cheese_dist_finite_wavg:            8.182842
+      mouse-cheese_dist_solvable_avg:           18.921143
+      mouse-cheese_dist_solvable_wavg:          8.182843
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016816
+    scoring:
+      avg_scores:                               0.59971696
+    visit_patterns:
+      avg_first_visit_time:                     753.79834
+      avg_last_visit_time:                      889.2356
+      num_replay_batches:                       899
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013483913
+    avg_actor_approxkl3:                        0.0012001629
+    avg_actor_clipfrac:                         0.05979004
+    avg_actor_loss:                             -0.000552956
+    avg_advantage:                              0.010902086
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0076274932
+    avg_entropy:                                0.31270665
+    avg_grad_norm_pre_clip:                     2.587383
+    avg_loss:                                   0.002948084
+    max:
+      max_actor_approxkl1:                      0.0030865646
+      max_actor_approxkl3:                      0.0024829933
+      max_actor_clipfrac:                       0.111206055
+      max_actor_loss:                           0.0015026852
+      max_advantage:                            0.30580688
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.009469665
+      max_entropy:                              0.33847427
+      max_grad_norm_pre_clip:                   5.0332184
+      max_loss:                                 0.0047688517
+    std:
+      std_actor_approxkl1:                      0.0009036258
+      std_actor_approxkl3:                      0.0006081467
+      std_actor_clipfrac:                       0.026591236
+      std_actor_loss:                           0.0010258147
+      std_advantage:                            0.12348373
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011238172
+      std_entropy:                              0.017750256
+      std_grad_norm_pre_clip:                   0.98480076
+      std_loss:                                 0.001037533
+  eval-batch-train:
+    avg_avg_episode_length:                     32.211052
+    avg_avg_return:                             0.7786235
+    avg_benchmark_regret:                       0.20859353
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.10181427
+    proxy_corner:
+      avg_avg_return:                           0.058451075
+      avg_reward_per_step:                      0.0042877197
+  eval-batch-orig:
+    avg_avg_episode_length:                     41.032097
+    avg_avg_return:                             0.68990314
+    avg_benchmark_regret:                       0.29460838
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.06147766
+    proxy_corner:
+      avg_avg_return:                           0.68990314
+      avg_reward_per_step:                      0.06147766
+  eval-batch-shift:
+    avg_avg_episode_length:                     32.72531
+    avg_avg_return:                             0.7805687
+    avg_benchmark_regret:                       0.20693724
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.11103058
+    proxy_corner:
+      avg_avg_return:                           0.05457525
+      avg_reward_per_step:                      0.0021591187
+  eval-batch-tree:
+    avg_avg_episode_length:                     37.59223
+    avg_avg_return:                             0.74502873
+    avg_benchmark_regret:                       0.23960644
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.075553894
+    proxy_corner:
+      avg_avg_return:                           0.74502873
+      avg_reward_per_step:                      0.075553894
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        58720256
+    env-step-replay-before:                     29851648
+    env-step-used-before:                       58720256
+    ppo-update-before:                          35840
+    env-step-generate-before:                   28868608
+    env-step-all-after:                         58753024
+    env-step-replay-after:                      29851648
+    env-step-used-after:                        58753024
+    ppo-update-after:                           35860
+    env-step-generate-after:                    28901376
+  train-generate:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     35.959557
+    avg_avg_return:                             0.68126297
+    avg_reward_per_step:                        0.09780884
+    proxy_corner:
+      avg_avg_return:                           0.08851981
+      avg_reward_per_step:                      0.0034484863
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     35.959557
+    avg_avg_return:                             0.68126297
+    avg_reward_per_step:                        0.09780884
+    proxy_corner:
+      avg_avg_return:                           0.08851981
+      avg_reward_per_step:                      0.0034484863
+  perf:
+    env_steps_per_second:                       272748.38556843274
+    ppo_updates_per_second:                     45.16764460181857
+    eval-batch-train-duration:                  0.5645595202222466
+    eval-batch-orig-duration:                   0.46414485201239586
+    eval-batch-shift-duration:                  0.4630048265680671
+    eval-batch-tree-duration:                   0.4940196732059121
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    16.793945
+      cheese-corne_dist_wavg:                   16.08111
+      levels_cheese_is_in_corner_avg:           0.01977539
+      mouse-cheese_dist_finite_avg:             19.169922
+      mouse-cheese_dist_finite_wavg:            9.104576
+      mouse-cheese_dist_solvable_avg:           19.169922
+      mouse-cheese_dist_solvable_wavg:          9.104576
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.99999
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.60789466
+    visit_patterns:
+      avg_first_visit_time:                     771.8147
+      avg_last_visit_time:                      901.1521
+      num_replay_batches:                       911
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0020837977
+    avg_actor_approxkl3:                        0.0019777052
+    avg_actor_clipfrac:                         0.07844239
+    avg_actor_loss:                             0.00043090322
+    avg_advantage:                              0.018472547
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0075786174
+    avg_entropy:                                0.29759318
+    avg_grad_norm_pre_clip:                     2.8903491
+    avg_loss:                                   0.0039226185
+    max:
+      max_actor_approxkl1:                      0.0054153353
+      max_actor_approxkl3:                      0.0042058686
+      max_actor_clipfrac:                       0.13696289
+      max_actor_loss:                           0.0032424405
+      max_advantage:                            0.26617026
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.009401496
+      max_entropy:                              0.3267856
+      max_grad_norm_pre_clip:                   4.345972
+      max_loss:                                 0.006898846
+    std:
+      std_actor_approxkl1:                      0.0014486249
+      std_actor_approxkl3:                      0.0010207113
+      std_actor_clipfrac:                       0.03491024
+      std_actor_loss:                           0.0011610094
+      std_advantage:                            0.12295149
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0013020956
+      std_entropy:                              0.016039137
+      std_grad_norm_pre_clip:                   1.1386352
+      std_loss:                                 0.0013238764
+  eval-batch-train:
+    avg_avg_episode_length:                     30.566658
+    avg_avg_return:                             0.79302627
+    avg_benchmark_regret:                       0.19419083
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.103393555
+    proxy_corner:
+      avg_avg_return:                           0.07140514
+      avg_reward_per_step:                      0.004623413
+  eval-batch-orig:
+    avg_avg_episode_length:                     39.859985
+    avg_avg_return:                             0.7055553
+    avg_benchmark_regret:                       0.27895623
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.066215515
+    proxy_corner:
+      avg_avg_return:                           0.7055553
+      avg_reward_per_step:                      0.066215515
+  eval-batch-shift:
+    avg_avg_episode_length:                     31.214317
+    avg_avg_return:                             0.7955694
+    avg_benchmark_regret:                       0.19193658
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.115448
+    proxy_corner:
+      avg_avg_return:                           0.066249296
+      avg_reward_per_step:                      0.0027160645
+  eval-batch-tree:
+    avg_avg_episode_length:                     35.836487
+    avg_avg_return:                             0.75741595
+    avg_benchmark_regret:                       0.22721927
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.08005524
+    proxy_corner:
+      avg_avg_return:                           0.75741595
+      avg_reward_per_step:                      0.08005524
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        59768832
+    env-step-replay-before:                     30441472
+    env-step-used-before:                       59768832
+    ppo-update-before:                          36480
+    env-step-generate-before:                   29327360
+    env-step-all-after:                         59801600
+    env-step-replay-after:                      30474240
+    env-step-used-after:                        59801600
+    ppo-update-after:                           36500
+    env-step-generate-after:                    29327360
+  train-replay:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     67.10443
+    avg_avg_return:                             0.4154197
+    avg_reward_per_step:                        0.011993408
+    proxy_corner:
+      avg_avg_return:                           0.089786455
+      avg_reward_per_step:                      0.0016174316
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     67.10443
+    avg_avg_return:                             0.4154197
+    avg_reward_per_step:                        0.011993408
+    proxy_corner:
+      avg_avg_return:                           0.089786455
+      avg_reward_per_step:                      0.0016174316
+  perf:
+    env_steps_per_second:                       318769.8246712207
+    ppo_updates_per_second:                     42.224925022512394
+    eval-batch-train-duration:                  0.5250657387077808
+    eval-batch-orig-duration:                   0.4866469679400325
+    eval-batch-shift-duration:                  0.4862525649368763
+    eval-batch-tree-duration:                   0.4934597425162792
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    16.735352
+      cheese-corne_dist_wavg:                   30.46479
+      levels_cheese_is_in_corner_avg:           0.020019531
+      mouse-cheese_dist_finite_avg:             19.307373
+      mouse-cheese_dist_finite_wavg:            8.220158
+      mouse-cheese_dist_solvable_avg:           19.307373
+      mouse-cheese_dist_solvable_wavg:          8.220158
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.58350915
+    visit_patterns:
+      avg_first_visit_time:                     786.19824
+      avg_last_visit_time:                      920.14966
+      num_replay_batches:                       930
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010708565
+    avg_actor_approxkl3:                        0.0009579475
+    avg_actor_clipfrac:                         0.051660158
+    avg_actor_loss:                             -0.00071197154
+    avg_advantage:                              -0.017083166
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0107582295
+    avg_entropy:                                0.4022583
+    avg_grad_norm_pre_clip:                     2.5320837
+    avg_loss:                                   0.0042648856
+    max:
+      max_actor_approxkl1:                      0.0020772098
+      max_actor_approxkl3:                      0.0017487934
+      max_actor_clipfrac:                       0.099609375
+      max_actor_loss:                           0.0006930828
+      max_advantage:                            0.28644812
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.01224196
+      max_entropy:                              0.43253177
+      max_grad_norm_pre_clip:                   4.3870096
+      max_loss:                                 0.0064137783
+    std:
+      std_actor_approxkl1:                      0.00054142554
+      std_actor_approxkl3:                      0.0004032813
+      std_actor_clipfrac:                       0.02269801
+      std_actor_loss:                           0.00091309904
+      std_advantage:                            0.14641802
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011813694
+      std_entropy:                              0.011815476
+      std_grad_norm_pre_clip:                   1.0366281
+      std_loss:                                 0.0010588771
+  eval-batch-train:
+    avg_avg_episode_length:                     30.091688
+    avg_avg_return:                             0.7973827
+    avg_benchmark_regret:                       0.18983433
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.10459137
+    proxy_corner:
+      avg_avg_return:                           0.06260761
+      avg_reward_per_step:                      0.004875183
+  eval-batch-orig:
+    avg_avg_episode_length:                     38.988102
+    avg_avg_return:                             0.72391224
+    avg_benchmark_regret:                       0.26059932
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.06654358
+    proxy_corner:
+      avg_avg_return:                           0.72391224
+      avg_reward_per_step:                      0.06654358
+  eval-batch-shift:
+    avg_avg_episode_length:                     30.336777
+    avg_avg_return:                             0.81205773
+    avg_benchmark_regret:                       0.17544824
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.11508179
+    proxy_corner:
+      avg_avg_return:                           0.069632985
+      avg_reward_per_step:                      0.0027313232
+  eval-batch-tree:
+    avg_avg_episode_length:                     36.63891
+    avg_avg_return:                             0.75597644
+    avg_benchmark_regret:                       0.22865872
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.07917023
+    proxy_corner:
+      avg_avg_return:                           0.75597644
+      avg_reward_per_step:                      0.07917023
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        60817408
+    env-step-replay-before:                     30998528
+    env-step-used-before:                       60817408
+    ppo-update-before:                          37120
+    env-step-generate-before:                   29818880
+    env-step-all-after:                         60850176
+    env-step-replay-after:                      30998528
+    env-step-used-after:                        60850176
+    ppo-update-after:                           37140
+    env-step-generate-after:                    29851648
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     34.22293
+    avg_avg_return:                             0.714175
+    avg_reward_per_step:                        0.10449219
+    proxy_corner:
+      avg_avg_return:                           0.06038984
+      avg_reward_per_step:                      0.0042419434
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     34.22293
+    avg_avg_return:                             0.714175
+    avg_reward_per_step:                        0.10449219
+    proxy_corner:
+      avg_avg_return:                           0.06038984
+      avg_reward_per_step:                      0.0042419434
+  perf:
+    env_steps_per_second:                       244308.76168605871
+    ppo_updates_per_second:                     41.638666229394325
+    eval-batch-train-duration:                  0.5402762275189161
+    eval-batch-orig-duration:                   0.45288925897330046
+    eval-batch-shift-duration:                  0.45642412547022104
+    eval-batch-tree-duration:                   0.48156977631151676
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    16.710693
+      cheese-corne_dist_wavg:                   30.465834
+      levels_cheese_is_in_corner_avg:           0.022460938
+      mouse-cheese_dist_finite_avg:             19.582275
+      mouse-cheese_dist_finite_wavg:            8.268535
+      mouse-cheese_dist_solvable_avg:           19.582275
+      mouse-cheese_dist_solvable_wavg:          8.268535
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5865482
+    visit_patterns:
+      avg_first_visit_time:                     800.07764
+      avg_last_visit_time:                      936.2571
+      num_replay_batches:                       946
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001336364
+    avg_actor_approxkl3:                        0.0011568965
+    avg_actor_clipfrac:                         0.053118896
+    avg_actor_loss:                             -0.0001416687
+    avg_advantage:                              0.07054356
+    avg_critic_clipfrac:                        0.2740906
+    avg_critic_loss:                            0.008323146
+    avg_entropy:                                0.2713395
+    avg_grad_norm_pre_clip:                     2.4712243
+    avg_loss:                                   0.0037485647
+    max:
+      max_actor_approxkl1:                      0.003419591
+      max_actor_approxkl3:                      0.0026278186
+      max_actor_clipfrac:                       0.09350586
+      max_actor_loss:                           0.0008252561
+      max_advantage:                            0.3536936
+      max_critic_clipfrac:                      0.9946289
+      max_critic_loss:                          0.0113774855
+      max_entropy:                              0.3215865
+      max_grad_norm_pre_clip:                   4.212515
+      max_loss:                                 0.0062411316
+    std:
+      std_actor_approxkl1:                      0.0008251472
+      std_actor_approxkl3:                      0.000557807
+      std_actor_clipfrac:                       0.022172527
+      std_actor_loss:                           0.00080441043
+      std_advantage:                            0.121515125
+      std_critic_clipfrac:                      0.38202888
+      std_critic_loss:                          0.0014977098
+      std_entropy:                              0.017242093
+      std_grad_norm_pre_clip:                   0.8942223
+      std_loss:                                 0.0011490221
+  eval-batch-train:
+    avg_avg_episode_length:                     30.67014
+    avg_avg_return:                             0.79488593
+    avg_benchmark_regret:                       0.19233112
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.1052475
+    proxy_corner:
+      avg_avg_return:                           0.061690398
+      avg_reward_per_step:                      0.004234314
+  eval-batch-orig:
+    avg_avg_episode_length:                     38.01566
+    avg_avg_return:                             0.7397431
+    avg_benchmark_regret:                       0.24476847
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.062347412
+    proxy_corner:
+      avg_avg_return:                           0.7397431
+      avg_reward_per_step:                      0.062347412
+  eval-batch-shift:
+    avg_avg_episode_length:                     30.180851
+    avg_avg_return:                             0.806643
+    avg_benchmark_regret:                       0.18086296
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.11462402
+    proxy_corner:
+      avg_avg_return:                           0.060778536
+      avg_reward_per_step:                      0.0025634766
+  eval-batch-tree:
+    avg_avg_episode_length:                     35.17952
+    avg_avg_return:                             0.77602005
+    avg_benchmark_regret:                       0.20861521
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.077568054
+    proxy_corner:
+      avg_avg_return:                           0.77602005
+      avg_reward_per_step:                      0.077568054
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        61865984
+    env-step-replay-before:                     31424512
+    env-step-used-before:                       61865984
+    ppo-update-before:                          37760
+    env-step-generate-before:                   30441472
+    env-step-all-after:                         61898752
+    env-step-replay-after:                      31424512
+    env-step-used-after:                        61898752
+    ppo-update-after:                           37780
+    env-step-generate-after:                    30474240
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     33.165424
+    avg_avg_return:                             0.71296644
+    avg_reward_per_step:                        0.11331177
+    proxy_corner:
+      avg_avg_return:                           0.038044088
+      avg_reward_per_step:                      0.0016784668
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     33.165424
+    avg_avg_return:                             0.71296644
+    avg_reward_per_step:                        0.11331177
+    proxy_corner:
+      avg_avg_return:                           0.038044088
+      avg_reward_per_step:                      0.0016784668
+  perf:
+    env_steps_per_second:                       258681.8142993007
+    ppo_updates_per_second:                     42.237711227386804
+    eval-batch-train-duration:                  0.6833843104541302
+    eval-batch-orig-duration:                   0.6344490861520171
+    eval-batch-shift-duration:                  0.6504762545228004
+    eval-batch-tree-duration:                   0.6191331716254354
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    16.759766
+      cheese-corne_dist_wavg:                   7.0833287
+      levels_cheese_is_in_corner_avg:           0.021728516
+      mouse-cheese_dist_finite_avg:             19.714355
+      mouse-cheese_dist_finite_wavg:            8.262344
+      mouse-cheese_dist_solvable_avg:           19.714355
+      mouse-cheese_dist_solvable_wavg:          8.262345
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5885829
+    visit_patterns:
+      avg_first_visit_time:                     813.5703
+      avg_last_visit_time:                      949.25854
+      num_replay_batches:                       959
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0027158791
+    avg_actor_approxkl3:                        0.002049086
+    avg_actor_clipfrac:                         0.07623901
+    avg_actor_loss:                             0.00049727986
+    avg_advantage:                              -0.00061687926
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0059550786
+    avg_entropy:                                0.2722417
+    avg_grad_norm_pre_clip:                     2.8979278
+    avg_loss:                                   0.0032025778
+    max:
+      max_actor_approxkl1:                      0.0068546897
+      max_actor_approxkl3:                      0.0053305854
+      max_actor_clipfrac:                       0.13647461
+      max_actor_loss:                           0.0035961643
+      max_advantage:                            0.2377721
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0072994395
+      max_entropy:                              0.3084928
+      max_grad_norm_pre_clip:                   6.2595763
+      max_loss:                                 0.006641064
+    std:
+      std_actor_approxkl1:                      0.001619381
+      std_actor_approxkl3:                      0.0011893407
+      std_actor_clipfrac:                       0.031942457
+      std_actor_loss:                           0.001559253
+      std_advantage:                            0.109396175
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010009755
+      std_entropy:                              0.018746888
+      std_grad_norm_pre_clip:                   1.2635051
+      std_loss:                                 0.0016778774
+  eval-batch-train:
+    avg_avg_episode_length:                     29.88129
+    avg_avg_return:                             0.79884183
+    avg_benchmark_regret:                       0.18837525
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.106651306
+    proxy_corner:
+      avg_avg_return:                           0.054141894
+      avg_reward_per_step:                      0.0031814575
+  eval-batch-orig:
+    avg_avg_episode_length:                     36.87564
+    avg_avg_return:                             0.7586799
+    avg_benchmark_regret:                       0.22583163
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.061302185
+    proxy_corner:
+      avg_avg_return:                           0.7586799
+      avg_reward_per_step:                      0.061302185
+  eval-batch-shift:
+    avg_avg_episode_length:                     29.31852
+    avg_avg_return:                             0.8196618
+    avg_benchmark_regret:                       0.16784418
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.11895752
+    proxy_corner:
+      avg_avg_return:                           0.058168303
+      avg_reward_per_step:                      0.002494812
+  eval-batch-tree:
+    avg_avg_episode_length:                     34.678852
+    avg_avg_return:                             0.7858044
+    avg_benchmark_regret:                       0.19883078
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0787735
+    proxy_corner:
+      avg_avg_return:                           0.7858044
+      avg_reward_per_step:                      0.0787735
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        62914560
+    env-step-replay-before:                     31981568
+    env-step-used-before:                       62914560
+    ppo-update-before:                          38400
+    env-step-generate-before:                   30932992
+    env-step-all-after:                         62947328
+    env-step-replay-after:                      31981568
+    env-step-used-after:                        62947328
+    ppo-update-after:                           38420
+    env-step-generate-after:                    30965760
+  train-generate:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     35.876488
+    avg_avg_return:                             0.69210356
+    avg_reward_per_step:                        0.101501465
+    proxy_corner:
+      avg_avg_return:                           0.07007012
+      avg_reward_per_step:                      0.0040893555
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     35.876488
+    avg_avg_return:                             0.69210356
+    avg_reward_per_step:                        0.101501465
+    proxy_corner:
+      avg_avg_return:                           0.07007012
+      avg_reward_per_step:                      0.0040893555
+  perf:
+    env_steps_per_second:                       344383.1894392308
+    ppo_updates_per_second:                     43.196878240526
+    eval-batch-train-duration:                  0.5242154998704791
+    eval-batch-orig-duration:                   0.4915415486320853
+    eval-batch-shift-duration:                  0.49118390306830406
+    eval-batch-tree-duration:                   0.4775822591036558
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    16.81836
+      cheese-corne_dist_wavg:                   16.069492
+      levels_cheese_is_in_corner_avg:           0.020996094
+      mouse-cheese_dist_finite_avg:             19.879639
+      mouse-cheese_dist_finite_wavg:            8.282729
+      mouse-cheese_dist_solvable_avg:           19.879639
+      mouse-cheese_dist_solvable_wavg:          8.28273
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.58800566
+    visit_patterns:
+      avg_first_visit_time:                     828.9407
+      avg_last_visit_time:                      966.3186
+      num_replay_batches:                       976
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014877012
+    avg_actor_approxkl3:                        0.0012118084
+    avg_actor_clipfrac:                         0.05111084
+    avg_actor_loss:                             -0.00073214323
+    avg_advantage:                              0.022997096
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0071040834
+    avg_entropy:                                0.2628498
+    avg_grad_norm_pre_clip:                     2.371023
+    avg_loss:                                   0.002557048
+    max:
+      max_actor_approxkl1:                      0.0033857618
+      max_actor_approxkl3:                      0.0022716857
+      max_actor_clipfrac:                       0.09289551
+      max_actor_loss:                           0.000613749
+      max_advantage:                            0.2731406
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00902902
+      max_entropy:                              0.27259016
+      max_grad_norm_pre_clip:                   4.104748
+      max_loss:                                 0.0040624687
+    std:
+      std_actor_approxkl1:                      0.0011470127
+      std_actor_approxkl3:                      0.00071141514
+      std_actor_clipfrac:                       0.028503852
+      std_actor_loss:                           0.00074430276
+      std_advantage:                            0.117816865
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0008568926
+      std_entropy:                              0.007532878
+      std_grad_norm_pre_clip:                   1.0214169
+      std_loss:                                 0.00083752896
+  eval-batch-train:
+    avg_avg_episode_length:                     29.64784
+    avg_avg_return:                             0.80920815
+    avg_benchmark_regret:                       0.17800891
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.107162476
+    proxy_corner:
+      avg_avg_return:                           0.05437684
+      avg_reward_per_step:                      0.004310608
+  eval-batch-orig:
+    avg_avg_episode_length:                     35.623695
+    avg_avg_return:                             0.77655804
+    avg_benchmark_regret:                       0.20795347
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.06503296
+    proxy_corner:
+      avg_avg_return:                           0.77655804
+      avg_reward_per_step:                      0.06503296
+  eval-batch-shift:
+    avg_avg_episode_length:                     29.204254
+    avg_avg_return:                             0.82100034
+    avg_benchmark_regret:                       0.16650558
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.11747742
+    proxy_corner:
+      avg_avg_return:                           0.05723587
+      avg_reward_per_step:                      0.002456665
+  eval-batch-tree:
+    avg_avg_episode_length:                     33.502304
+    avg_avg_return:                             0.7966738
+    avg_benchmark_regret:                       0.18796141
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.07942963
+    proxy_corner:
+      avg_avg_return:                           0.7966738
+      avg_reward_per_step:                      0.07942963
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        63963136
+    env-step-replay-before:                     32505856
+    env-step-used-before:                       63963136
+    ppo-update-before:                          39040
+    env-step-generate-before:                   31457280
+    env-step-all-after:                         63995904
+    env-step-replay-after:                      32505856
+    env-step-used-after:                        63995904
+    ppo-update-after:                           39060
+    env-step-generate-after:                    31490048
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     32.16422
+    avg_avg_return:                             0.734936
+    avg_reward_per_step:                        0.1121521
+    proxy_corner:
+      avg_avg_return:                           0.069712386
+      avg_reward_per_step:                      0.0026245117
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     32.16422
+    avg_avg_return:                             0.734936
+    avg_reward_per_step:                        0.1121521
+    proxy_corner:
+      avg_avg_return:                           0.069712386
+      avg_reward_per_step:                      0.0026245117
+  perf:
+    env_steps_per_second:                       297583.6284659649
+    ppo_updates_per_second:                     42.174143101690504
+    eval-batch-train-duration:                  0.5290024215355515
+    eval-batch-orig-duration:                   0.5083054406568408
+    eval-batch-shift-duration:                  0.4729716796427965
+    eval-batch-tree-duration:                   0.49466422013938427
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.135254
+      cheese-corne_dist_wavg:                   16.104029
+      levels_cheese_is_in_corner_avg:           0.01953125
+      mouse-cheese_dist_finite_avg:             20.088379
+      mouse-cheese_dist_finite_wavg:            8.297411
+      mouse-cheese_dist_solvable_avg:           20.088379
+      mouse-cheese_dist_solvable_wavg:          8.297412
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.5986967
+    visit_patterns:
+      avg_first_visit_time:                     846.4995
+      avg_last_visit_time:                      982.19727
+      num_replay_batches:                       992
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00089866715
+    avg_actor_approxkl3:                        0.0007629837
+    avg_actor_clipfrac:                         0.039001465
+    avg_actor_loss:                             -0.00087217864
+    avg_advantage:                              0.035252027
+    avg_critic_clipfrac:                        0.0010620117
+    avg_critic_loss:                            0.005780357
+    avg_entropy:                                0.25037327
+    avg_grad_norm_pre_clip:                     1.9844459
+    avg_loss:                                   0.0017676264
+    max:
+      max_actor_approxkl1:                      0.0022125929
+      max_actor_approxkl3:                      0.0016654812
+      max_actor_clipfrac:                       0.07434082
+      max_actor_loss:                           0.0010243971
+      max_advantage:                            0.28119093
+      max_critic_clipfrac:                      0.0087890625
+      max_critic_loss:                          0.007128909
+      max_entropy:                              0.27838707
+      max_grad_norm_pre_clip:                   3.5949814
+      max_loss:                                 0.0030251648
+    std:
+      std_actor_approxkl1:                      0.00074931607
+      std_actor_approxkl3:                      0.00036372276
+      std_actor_clipfrac:                       0.01750458
+      std_actor_loss:                           0.00095425826
+      std_advantage:                            0.10494331
+      std_critic_clipfrac:                      0.0026958208
+      std_critic_loss:                          0.00080521376
+      std_entropy:                              0.01824671
+      std_grad_norm_pre_clip:                   0.8372076
+      std_loss:                                 0.000978926
+  eval-batch-train:
+    avg_avg_episode_length:                     29.292114
+    avg_avg_return:                             0.80592626
+    avg_benchmark_regret:                       0.18129078
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.107810974
+    proxy_corner:
+      avg_avg_return:                           0.08067814
+      avg_reward_per_step:                      0.0049591064
+  eval-batch-orig:
+    avg_avg_episode_length:                     36.36413
+    avg_avg_return:                             0.755309
+    avg_benchmark_regret:                       0.22920257
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.06750488
+    proxy_corner:
+      avg_avg_return:                           0.755309
+      avg_reward_per_step:                      0.06750488
+  eval-batch-shift:
+    avg_avg_episode_length:                     28.518269
+    avg_avg_return:                             0.8276999
+    avg_benchmark_regret:                       0.15980604
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.1207962
+    proxy_corner:
+      avg_avg_return:                           0.072757095
+      avg_reward_per_step:                      0.0030517578
+  eval-batch-tree:
+    avg_avg_episode_length:                     33.995346
+    avg_avg_return:                             0.77891535
+    avg_benchmark_regret:                       0.20571984
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.07977295
+    proxy_corner:
+      avg_avg_return:                           0.77891535
+      avg_reward_per_step:                      0.07977295
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        65011712
+    env-step-replay-before:                     33062912
+    env-step-used-before:                       65011712
+    ppo-update-before:                          39680
+    env-step-generate-before:                   31948800
+    env-step-all-after:                         65044480
+    env-step-replay-after:                      33062912
+    env-step-used-after:                        65044480
+    ppo-update-after:                           39700
+    env-step-generate-after:                    31981568
+  train-generate:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     34.12897
+    avg_avg_return:                             0.7059475
+    avg_reward_per_step:                        0.09158325
+    proxy_corner:
+      avg_avg_return:                           0.040609844
+      avg_reward_per_step:                      0.0025634766
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     34.12897
+    avg_avg_return:                             0.7059475
+    avg_reward_per_step:                        0.09158325
+    proxy_corner:
+      avg_avg_return:                           0.040609844
+      avg_reward_per_step:                      0.0025634766
+  perf:
+    env_steps_per_second:                       354432.8028928599
+    ppo_updates_per_second:                     40.75514199585795
+    eval-batch-train-duration:                  0.5320184882730246
+    eval-batch-orig-duration:                   0.4838894149288535
+    eval-batch-shift-duration:                  0.4434247054159641
+    eval-batch-tree-duration:                   0.4928299579769373
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.154541
+      cheese-corne_dist_wavg:                   16.119608
+      levels_cheese_is_in_corner_avg:           0.018310547
+      mouse-cheese_dist_finite_avg:             20.179688
+      mouse-cheese_dist_finite_wavg:            8.313657
+      mouse-cheese_dist_solvable_avg:           20.179688
+      mouse-cheese_dist_solvable_wavg:          8.313658
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.5930645
+    visit_patterns:
+      avg_first_visit_time:                     859.3884
+      avg_last_visit_time:                      999.25146
+      num_replay_batches:                       1009
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0008583834
+    avg_actor_approxkl3:                        0.00076647184
+    avg_actor_clipfrac:                         0.036291506
+    avg_actor_loss:                             -0.0011810664
+    avg_advantage:                              0.010568295
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0065783407
+    avg_entropy:                                0.2385944
+    avg_grad_norm_pre_clip:                     1.5246181
+    avg_loss:                                   0.0018695092
+    max:
+      max_actor_approxkl1:                      0.0021015794
+      max_actor_approxkl3:                      0.001273667
+      max_actor_clipfrac:                       0.061767578
+      max_actor_loss:                           0.0008675307
+      max_advantage:                            0.24648112
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0086531965
+      max_entropy:                              0.25719106
+      max_grad_norm_pre_clip:                   2.2640038
+      max_loss:                                 0.0048032734
+    std:
+      std_actor_approxkl1:                      0.0006246348
+      std_actor_approxkl3:                      0.0003391069
+      std_actor_clipfrac:                       0.01630848
+      std_actor_loss:                           0.0012257354
+      std_advantage:                            0.11446891
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0013414366
+      std_entropy:                              0.014601093
+      std_grad_norm_pre_clip:                   0.48515305
+      std_loss:                                 0.0013680003
+  eval-batch-train:
+    avg_avg_episode_length:                     28.414604
+    avg_avg_return:                             0.8265228
+    avg_benchmark_regret:                       0.16069427
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.10897827
+    proxy_corner:
+      avg_avg_return:                           0.06388631
+      avg_reward_per_step:                      0.0045166016
+  eval-batch-orig:
+    avg_avg_episode_length:                     35.939392
+    avg_avg_return:                             0.7531992
+    avg_benchmark_regret:                       0.2313123
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.06690216
+    proxy_corner:
+      avg_avg_return:                           0.7531992
+      avg_reward_per_step:                      0.06690216
+  eval-batch-shift:
+    avg_avg_episode_length:                     28.19941
+    avg_avg_return:                             0.82948554
+    avg_benchmark_regret:                       0.15802045
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.12358093
+    proxy_corner:
+      avg_avg_return:                           0.06691029
+      avg_reward_per_step:                      0.0030441284
+  eval-batch-tree:
+    avg_avg_episode_length:                     33.684856
+    avg_avg_return:                             0.7919113
+    avg_benchmark_regret:                       0.19272387
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.07896423
+    proxy_corner:
+      avg_avg_return:                           0.7919113
+      avg_reward_per_step:                      0.07896423
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        66060288
+    env-step-replay-before:                     33554432
+    env-step-used-before:                       66060288
+    ppo-update-before:                          40320
+    env-step-generate-before:                   32505856
+    env-step-all-after:                         66093056
+    env-step-replay-after:                      33554432
+    env-step-used-after:                        66093056
+    ppo-update-after:                           40340
+    env-step-generate-after:                    32538624
+  train-generate:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     33.74875
+    avg_avg_return:                             0.7012837
+    avg_reward_per_step:                        0.09573364
+    proxy_corner:
+      avg_avg_return:                           0.083877355
+      avg_reward_per_step:                      0.0030212402
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     33.74875
+    avg_avg_return:                             0.7012837
+    avg_reward_per_step:                        0.09573364
+    proxy_corner:
+      avg_avg_return:                           0.083877355
+      avg_reward_per_step:                      0.0030212402
+  perf:
+    env_steps_per_second:                       315188.0559971024
+    ppo_updates_per_second:                     42.14140965676215
+    eval-batch-train-duration:                  0.5244269305840135
+    eval-batch-orig-duration:                   0.4493955299258232
+    eval-batch-shift-duration:                  0.4749627476558089
+    eval-batch-tree-duration:                   0.4868883974850178
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.080322
+      cheese-corne_dist_wavg:                   11.606571
+      levels_cheese_is_in_corner_avg:           0.018310547
+      mouse-cheese_dist_finite_avg:             20.33545
+      mouse-cheese_dist_finite_wavg:            9.221651
+      mouse-cheese_dist_solvable_avg:           20.33545
+      mouse-cheese_dist_solvable_wavg:          9.221651
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.420168
+    scoring:
+      avg_scores:                               0.6014593
+    visit_patterns:
+      avg_first_visit_time:                     874.3115
+      avg_last_visit_time:                      1014.27515
+      num_replay_batches:                       1024
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00095607573
+    avg_actor_approxkl3:                        0.0009596878
+    avg_actor_clipfrac:                         0.045721438
+    avg_actor_loss:                             -0.00035164395
+    avg_advantage:                              0.0060863867
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.006974219
+    avg_entropy:                                0.23857044
+    avg_grad_norm_pre_clip:                     1.9006146
+    avg_loss:                                   0.0028968954
+    max:
+      max_actor_approxkl1:                      0.001900312
+      max_actor_approxkl3:                      0.0020418759
+      max_actor_clipfrac:                       0.08862305
+      max_actor_loss:                           0.0031286553
+      max_advantage:                            0.23336679
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.009036304
+      max_entropy:                              0.25850946
+      max_grad_norm_pre_clip:                   3.3509536
+      max_loss:                                 0.0061103827
+    std:
+      std_actor_approxkl1:                      0.0005455837
+      std_actor_approxkl3:                      0.0004183782
+      std_actor_clipfrac:                       0.018743556
+      std_actor_loss:                           0.0013585978
+      std_advantage:                            0.117835306
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010268425
+      std_entropy:                              0.013681556
+      std_grad_norm_pre_clip:                   0.6377131
+      std_loss:                                 0.0013572985
+  eval-batch-train:
+    avg_avg_episode_length:                     28.486742
+    avg_avg_return:                             0.82052535
+    avg_benchmark_regret:                       0.16669166
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.10845947
+    proxy_corner:
+      avg_avg_return:                           0.0643897
+      avg_reward_per_step:                      0.004524231
+  eval-batch-orig:
+    avg_avg_episode_length:                     35.56311
+    avg_avg_return:                             0.7782891
+    avg_benchmark_regret:                       0.20622246
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.064842224
+    proxy_corner:
+      avg_avg_return:                           0.7782891
+      avg_reward_per_step:                      0.064842224
+  eval-batch-shift:
+    avg_avg_episode_length:                     29.028969
+    avg_avg_return:                             0.8202275
+    avg_benchmark_regret:                       0.16727841
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.12164307
+    proxy_corner:
+      avg_avg_return:                           0.05935079
+      avg_reward_per_step:                      0.00289917
+  eval-batch-tree:
+    avg_avg_episode_length:                     33.631454
+    avg_avg_return:                             0.79547817
+    avg_benchmark_regret:                       0.18915702
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.07817078
+    proxy_corner:
+      avg_avg_return:                           0.79547817
+      avg_reward_per_step:                      0.07817078
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        67108864
+    env-step-replay-before:                     34078720
+    env-step-used-before:                       67108864
+    ppo-update-before:                          40960
+    env-step-generate-before:                   33030144
+    env-step-all-after:                         67141632
+    env-step-replay-after:                      34111488
+    env-step-used-after:                        67141632
+    ppo-update-after:                           40980
+    env-step-generate-after:                    33030144
+  train-replay:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     69.92572
+    avg_avg_return:                             0.39805853
+    avg_reward_per_step:                        0.010437012
+    proxy_corner:
+      avg_avg_return:                           0.07899393
+      avg_reward_per_step:                      0.001373291
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     69.92572
+    avg_avg_return:                             0.39805853
+    avg_reward_per_step:                        0.010437012
+    proxy_corner:
+      avg_avg_return:                           0.07899393
+      avg_reward_per_step:                      0.001373291
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  perf:
+    env_steps_per_second:                       278015.5228024642
+    ppo_updates_per_second:                     41.56548124998703
+    eval-batch-train-duration:                  0.6490839645266533
+    eval-batch-orig-duration:                   0.5962857371196151
+    eval-batch-shift-duration:                  0.5947408899664879
+    eval-batch-tree-duration:                   0.5919076856225729
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.182861
+      cheese-corne_dist_wavg:                   21.510086
+      levels_cheese_is_in_corner_avg:           0.018798828
+      mouse-cheese_dist_finite_avg:             20.435547
+      mouse-cheese_dist_finite_wavg:            6.5504723
+      mouse-cheese_dist_solvable_avg:           20.435547
+      mouse-cheese_dist_solvable_wavg:          6.550473
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.5969651
+    visit_patterns:
+      avg_first_visit_time:                     887.57275
+      avg_last_visit_time:                      1031.2234
+      num_replay_batches:                       1041
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010302284
+    avg_actor_approxkl3:                        0.0010379907
+    avg_actor_clipfrac:                         0.05394287
+    avg_actor_loss:                             -0.0009556547
+    avg_advantage:                              -0.020017395
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.010388332
+    avg_entropy:                                0.36131057
+    avg_grad_norm_pre_clip:                     2.2672212
+    avg_loss:                                   0.003877201
+    max:
+      max_actor_approxkl1:                      0.0022208942
+      max_actor_approxkl3:                      0.00194538
+      max_actor_clipfrac:                       0.09106445
+      max_actor_loss:                           0.001052171
+      max_advantage:                            0.30773997
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.011948014
+      max_entropy:                              0.38286984
+      max_grad_norm_pre_clip:                   4.51586
+      max_loss:                                 0.0063549243
+    std:
+      std_actor_approxkl1:                      0.00076419045
+      std_actor_approxkl3:                      0.00050748786
+      std_actor_clipfrac:                       0.025628341
+      std_actor_loss:                           0.0010236333
+      std_advantage:                            0.14413923
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00080576824
+      std_entropy:                              0.011083412
+      std_grad_norm_pre_clip:                   0.8805033
+      std_loss:                                 0.0010932789
+  eval-batch-train:
+    avg_avg_episode_length:                     28.501007
+    avg_avg_return:                             0.82966673
+    avg_benchmark_regret:                       0.15755032
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.1091156
+    proxy_corner:
+      avg_avg_return:                           0.05597485
+      avg_reward_per_step:                      0.0041503906
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-orig:
+    avg_avg_episode_length:                     35.06436
+    avg_avg_return:                             0.7681832
+    avg_benchmark_regret:                       0.21632838
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.06752014
+    proxy_corner:
+      avg_avg_return:                           0.7681832
+      avg_reward_per_step:                      0.06752014
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-shift:
+    avg_avg_episode_length:                     28.38081
+    avg_avg_return:                             0.83182245
+    avg_benchmark_regret:                       0.15568349
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.11955261
+    proxy_corner:
+      avg_avg_return:                           0.059491992
+      avg_reward_per_step:                      0.0026550293
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-tree:
+    avg_avg_episode_length:                     32.43283
+    avg_avg_return:                             0.8074758
+    avg_benchmark_regret:                       0.17715943
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.080589294
+    proxy_corner:
+      avg_avg_return:                           0.8074758
+      avg_reward_per_step:                      0.080589294
+    rollouts_gif:                               float32(132, 225, 225, 3)
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        68157440
+    env-step-replay-before:                     34668544
+    env-step-used-before:                       68157440
+    ppo-update-before:                          41600
+    env-step-generate-before:                   33488896
+    env-step-all-after:                         68190208
+    env-step-replay-after:                      34668544
+    env-step-used-after:                        68190208
+    ppo-update-after:                           41620
+    env-step-generate-after:                    33521664
+  train-generate:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     29.011341
+    avg_avg_return:                             0.762318
+    avg_reward_per_step:                        0.116485596
+    proxy_corner:
+      avg_avg_return:                           0.04092007
+      avg_reward_per_step:                      0.0014038086
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     29.011341
+    avg_avg_return:                             0.762318
+    avg_reward_per_step:                        0.116485596
+    proxy_corner:
+      avg_avg_return:                           0.04092007
+      avg_reward_per_step:                      0.0014038086
+  perf:
+    env_steps_per_second:                       315168.41969183064
+    ppo_updates_per_second:                     41.66996605081752
+    eval-batch-train-duration:                  0.5384000474587083
+    eval-batch-orig-duration:                   0.4872132483869791
+    eval-batch-shift-duration:                  0.4865201748907566
+    eval-batch-tree-duration:                   0.5014560334384441
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.313965
+      cheese-corne_dist_wavg:                   21.515133
+      levels_cheese_is_in_corner_avg:           0.019042969
+      mouse-cheese_dist_finite_avg:             20.568115
+      mouse-cheese_dist_finite_wavg:            6.563388
+      mouse-cheese_dist_solvable_avg:           20.568115
+      mouse-cheese_dist_solvable_wavg:          6.563389
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.59073937
+    visit_patterns:
+      avg_first_visit_time:                     903.21875
+      avg_last_visit_time:                      1048.325
+      num_replay_batches:                       1058
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014014365
+    avg_actor_approxkl3:                        0.00093345105
+    avg_actor_clipfrac:                         0.0432251
+    avg_actor_loss:                             -0.00030739626
+    avg_advantage:                              0.060881294
+    avg_critic_clipfrac:                        0.15429078
+    avg_critic_loss:                            0.006023776
+    avg_entropy:                                0.22251008
+    avg_grad_norm_pre_clip:                     2.3441007
+    avg_loss:                                   0.0024819816
+    max:
+      max_actor_approxkl1:                      0.0028314963
+      max_actor_approxkl3:                      0.0024402118
+      max_actor_clipfrac:                       0.09741211
+      max_actor_loss:                           0.0009210035
+      max_advantage:                            0.28339404
+      max_critic_clipfrac:                      0.79052734
+      max_critic_loss:                          0.007899963
+      max_entropy:                              0.24618915
+      max_grad_norm_pre_clip:                   4.5879455
+      max_loss:                                 0.0039220937
+    std:
+      std_actor_approxkl1:                      0.0007605078
+      std_actor_approxkl3:                      0.0006335375
+      std_actor_clipfrac:                       0.02495948
+      std_actor_loss:                           0.000867959
+      std_advantage:                            0.101373985
+      std_critic_clipfrac:                      0.2604664
+      std_critic_loss:                          0.0010819607
+      std_entropy:                              0.012389912
+      std_grad_norm_pre_clip:                   1.1373843
+      std_loss:                                 0.0010547134
+  eval-batch-train:
+    avg_avg_episode_length:                     26.821444
+    avg_avg_return:                             0.8420316
+    avg_benchmark_regret:                       0.14518553
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.11164093
+    proxy_corner:
+      avg_avg_return:                           0.059588443
+      avg_reward_per_step:                      0.004310608
+  eval-batch-orig:
+    avg_avg_episode_length:                     35.609062
+    avg_avg_return:                             0.76454306
+    avg_benchmark_regret:                       0.2199685
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.06741333
+    proxy_corner:
+      avg_avg_return:                           0.76454306
+      avg_reward_per_step:                      0.06741333
+  eval-batch-shift:
+    avg_avg_episode_length:                     27.554852
+    avg_avg_return:                             0.83372694
+    avg_benchmark_regret:                       0.15377903
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.12284851
+    proxy_corner:
+      avg_avg_return:                           0.056090932
+      avg_reward_per_step:                      0.0024642944
+  eval-batch-tree:
+    avg_avg_episode_length:                     32.641064
+    avg_avg_return:                             0.7979711
+    avg_benchmark_regret:                       0.18666407
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.08081055
+    proxy_corner:
+      avg_avg_return:                           0.7979711
+      avg_reward_per_step:                      0.08081055
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        69206016
+    env-step-replay-before:                     35258368
+    env-step-used-before:                       69206016
+    ppo-update-before:                          42240
+    env-step-generate-before:                   33947648
+    env-step-all-after:                         69238784
+    env-step-replay-after:                      35258368
+    env-step-used-after:                        69238784
+    ppo-update-after:                           42260
+    env-step-generate-after:                    33980416
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     30.903168
+    avg_avg_return:                             0.74323475
+    avg_reward_per_step:                        0.099609375
+    proxy_corner:
+      avg_avg_return:                           0.074805535
+      avg_reward_per_step:                      0.0059509277
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     30.903168
+    avg_avg_return:                             0.74323475
+    avg_reward_per_step:                        0.099609375
+    proxy_corner:
+      avg_avg_return:                           0.074805535
+      avg_reward_per_step:                      0.0059509277
+  perf:
+    env_steps_per_second:                       319376.0563898286
+    ppo_updates_per_second:                     42.135151933431736
+    eval-batch-train-duration:                  0.5063010975718498
+    eval-batch-orig-duration:                   0.41332386527210474
+    eval-batch-shift-duration:                  0.42069704644382
+    eval-batch-tree-duration:                   0.4022678667679429
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.398682
+      cheese-corne_dist_wavg:                   26.030235
+      levels_cheese_is_in_corner_avg:           0.017578125
+      mouse-cheese_dist_finite_avg:             20.64087
+      mouse-cheese_dist_finite_wavg:            9.266023
+      mouse-cheese_dist_solvable_avg:           20.64087
+      mouse-cheese_dist_solvable_wavg:          9.266023
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.99999
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.5835879
+    visit_patterns:
+      avg_first_visit_time:                     913.8638
+      avg_last_visit_time:                      1066.2476
+      num_replay_batches:                       1076
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012330869
+    avg_actor_approxkl3:                        0.0010606436
+    avg_actor_clipfrac:                         0.048535157
+    avg_actor_loss:                             -0.00023820662
+    avg_advantage:                              0.007671211
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.005164459
+    avg_entropy:                                0.21961068
+    avg_grad_norm_pre_clip:                     1.9123251
+    avg_loss:                                   0.0021244122
+    max:
+      max_actor_approxkl1:                      0.0029775533
+      max_actor_approxkl3:                      0.002190691
+      max_actor_clipfrac:                       0.083496094
+      max_actor_loss:                           0.0027965046
+      max_advantage:                            0.23539776
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0066935145
+      max_entropy:                              0.24846886
+      max_grad_norm_pre_clip:                   4.241637
+      max_loss:                                 0.0048388885
+    std:
+      std_actor_approxkl1:                      0.0008190295
+      std_actor_approxkl3:                      0.0005260148
+      std_actor_clipfrac:                       0.02160246
+      std_actor_loss:                           0.0012820917
+      std_advantage:                            0.10048371
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0009792453
+      std_entropy:                              0.012531014
+      std_grad_norm_pre_clip:                   0.8793738
+      std_loss:                                 0.0013851292
+  eval-batch-train:
+    avg_avg_episode_length:                     26.941261
+    avg_avg_return:                             0.8461356
+    avg_benchmark_regret:                       0.14108148
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.11022186
+    proxy_corner:
+      avg_avg_return:                           0.06468824
+      avg_reward_per_step:                      0.004425049
+  eval-batch-orig:
+    avg_avg_episode_length:                     35.05253
+    avg_avg_return:                             0.7675514
+    avg_benchmark_regret:                       0.21696012
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.06755066
+    proxy_corner:
+      avg_avg_return:                           0.7675514
+      avg_reward_per_step:                      0.06755066
+  eval-batch-shift:
+    avg_avg_episode_length:                     27.971737
+    avg_avg_return:                             0.8305606
+    avg_benchmark_regret:                       0.15694532
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.12062836
+    proxy_corner:
+      avg_avg_return:                           0.06389028
+      avg_reward_per_step:                      0.0027618408
+  eval-batch-tree:
+    avg_avg_episode_length:                     31.497112
+    avg_avg_return:                             0.80367833
+    avg_benchmark_regret:                       0.18095684
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.08201599
+    proxy_corner:
+      avg_avg_return:                           0.80367833
+      avg_reward_per_step:                      0.08201599
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        70254592
+    env-step-replay-before:                     35651584
+    env-step-used-before:                       70254592
+    ppo-update-before:                          42880
+    env-step-generate-before:                   34603008
+    env-step-all-after:                         70287360
+    env-step-replay-after:                      35651584
+    env-step-used-after:                        70287360
+    ppo-update-after:                           42900
+    env-step-generate-after:                    34635776
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     32.03239
+    avg_avg_return:                             0.720819
+    avg_reward_per_step:                        0.121276855
+    proxy_corner:
+      avg_avg_return:                           0.07043135
+      avg_reward_per_step:                      0.002960205
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     32.03239
+    avg_avg_return:                             0.720819
+    avg_reward_per_step:                        0.121276855
+    proxy_corner:
+      avg_avg_return:                           0.07043135
+      avg_reward_per_step:                      0.002960205
+  perf:
+    env_steps_per_second:                       257668.06183027764
+    ppo_updates_per_second:                     41.64248673465333
+    eval-batch-train-duration:                  0.54016297403723
+    eval-batch-orig-duration:                   0.48574100341647863
+    eval-batch-shift-duration:                  0.48145865835249424
+    eval-batch-tree-duration:                   0.500625847838819
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.416748
+      cheese-corne_dist_wavg:                   2.6627705
+      levels_cheese_is_in_corner_avg:           0.018554688
+      mouse-cheese_dist_finite_avg:             20.691406
+      mouse-cheese_dist_finite_wavg:            7.456295
+      mouse-cheese_dist_solvable_avg:           20.691406
+      mouse-cheese_dist_solvable_wavg:          7.456296
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.59554785
+    visit_patterns:
+      avg_first_visit_time:                     928.6289
+      avg_last_visit_time:                      1078.3127
+      num_replay_batches:                       1088
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00089387916
+    avg_actor_approxkl3:                        0.0007587194
+    avg_actor_clipfrac:                         0.035791017
+    avg_actor_loss:                             -0.0010630668
+    avg_advantage:                              -0.0007497496
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.005921315
+    avg_entropy:                                0.22182976
+    avg_grad_norm_pre_clip:                     1.3410923
+    avg_loss:                                   0.0016757606
+    max:
+      max_actor_approxkl1:                      0.0021970295
+      max_actor_approxkl3:                      0.0018248259
+      max_actor_clipfrac:                       0.072753906
+      max_actor_loss:                           0.00037381053
+      max_advantage:                            0.22263914
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0073457183
+      max_entropy:                              0.23853374
+      max_grad_norm_pre_clip:                   2.1830626
+      max_loss:                                 0.0033810553
+    std:
+      std_actor_approxkl1:                      0.00057724153
+      std_actor_approxkl3:                      0.00036300308
+      std_actor_clipfrac:                       0.01498582
+      std_actor_loss:                           0.0009354276
+      std_advantage:                            0.10922719
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00079046894
+      std_entropy:                              0.013660857
+      std_grad_norm_pre_clip:                   0.4122087
+      std_loss:                                 0.000998586
+  eval-batch-train:
+    avg_avg_episode_length:                     28.238745
+    avg_avg_return:                             0.8176731
+    avg_benchmark_regret:                       0.16954398
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.11116791
+    proxy_corner:
+      avg_avg_return:                           0.061622903
+      avg_reward_per_step:                      0.0043411255
+  eval-batch-orig:
+    avg_avg_episode_length:                     34.846992
+    avg_avg_return:                             0.77074265
+    avg_benchmark_regret:                       0.21376894
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.067604065
+    proxy_corner:
+      avg_avg_return:                           0.77074265
+      avg_reward_per_step:                      0.067604065
+  eval-batch-shift:
+    avg_avg_episode_length:                     27.487064
+    avg_avg_return:                             0.83246243
+    avg_benchmark_regret:                       0.15504347
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.12413025
+    proxy_corner:
+      avg_avg_return:                           0.06073626
+      avg_reward_per_step:                      0.0027313232
+  eval-batch-tree:
+    avg_avg_episode_length:                     31.990744
+    avg_avg_return:                             0.79661036
+    avg_benchmark_regret:                       0.18802482
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.08304596
+    proxy_corner:
+      avg_avg_return:                           0.79661036
+      avg_reward_per_step:                      0.08304596
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        71303168
+    env-step-replay-before:                     36044800
+    env-step-used-before:                       71303168
+    ppo-update-before:                          43520
+    env-step-generate-before:                   35258368
+    env-step-all-after:                         71335936
+    env-step-replay-after:                      36044800
+    env-step-used-after:                        71335936
+    ppo-update-after:                           43540
+    env-step-generate-after:                    35291136
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     33.442505
+    avg_avg_return:                             0.711519
+    avg_reward_per_step:                        0.1055603
+    proxy_corner:
+      avg_avg_return:                           0.050509807
+      avg_reward_per_step:                      0.0029296875
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     33.442505
+    avg_avg_return:                             0.711519
+    avg_reward_per_step:                        0.1055603
+    proxy_corner:
+      avg_avg_return:                           0.050509807
+      avg_reward_per_step:                      0.0029296875
+  perf:
+    env_steps_per_second:                       349278.67072233575
+    ppo_updates_per_second:                     41.78692966222363
+    eval-batch-train-duration:                  0.515146765857935
+    eval-batch-orig-duration:                   0.4852542970329523
+    eval-batch-shift-duration:                  0.47674952261149883
+    eval-batch-tree-duration:                   0.46831859927624464
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.403809
+      cheese-corne_dist_wavg:                   3.559232
+      levels_cheese_is_in_corner_avg:           0.018310547
+      mouse-cheese_dist_finite_avg:             20.786865
+      mouse-cheese_dist_finite_wavg:            7.4772725
+      mouse-cheese_dist_solvable_avg:           20.786865
+      mouse-cheese_dist_solvable_wavg:          7.4772735
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.6258513
+    visit_patterns:
+      avg_first_visit_time:                     942.98706
+      avg_last_visit_time:                      1090.3071
+      num_replay_batches:                       1100
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013964799
+    avg_actor_approxkl3:                        0.0013636638
+    avg_actor_clipfrac:                         0.055804443
+    avg_actor_loss:                             -0.0003444057
+    avg_advantage:                              -0.0031635338
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0058943583
+    avg_entropy:                                0.22204717
+    avg_grad_norm_pre_clip:                     2.3824816
+    avg_loss:                                   0.0023807266
+    max:
+      max_actor_approxkl1:                      0.0031721068
+      max_actor_approxkl3:                      0.0025217226
+      max_actor_clipfrac:                       0.09753418
+      max_actor_loss:                           0.001150541
+      max_advantage:                            0.223683
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00782013
+      max_entropy:                              0.25000712
+      max_grad_norm_pre_clip:                   5.7885594
+      max_loss:                                 0.0042251674
+    std:
+      std_actor_approxkl1:                      0.00087355654
+      std_actor_approxkl3:                      0.0007437557
+      std_actor_clipfrac:                       0.026885644
+      std_actor_loss:                           0.0010329643
+      std_advantage:                            0.108893126
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011045969
+      std_entropy:                              0.015171399
+      std_grad_norm_pre_clip:                   1.163182
+      std_loss:                                 0.0011296226
+  eval-batch-train:
+    avg_avg_episode_length:                     28.01258
+    avg_avg_return:                             0.8212764
+    avg_benchmark_regret:                       0.16594066
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.11393738
+    proxy_corner:
+      avg_avg_return:                           0.05638182
+      avg_reward_per_step:                      0.004425049
+  eval-batch-orig:
+    avg_avg_episode_length:                     35.32589
+    avg_avg_return:                             0.74925613
+    avg_benchmark_regret:                       0.23525533
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.069992065
+    proxy_corner:
+      avg_avg_return:                           0.74925613
+      avg_reward_per_step:                      0.069992065
+  eval-batch-shift:
+    avg_avg_episode_length:                     27.759476
+    avg_avg_return:                             0.82863
+    avg_benchmark_regret:                       0.158876
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.12562561
+    proxy_corner:
+      avg_avg_return:                           0.05604445
+      avg_reward_per_step:                      0.0025787354
+  eval-batch-tree:
+    avg_avg_episode_length:                     32.868668
+    avg_avg_return:                             0.7786681
+    avg_benchmark_regret:                       0.2059671
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.08129883
+    proxy_corner:
+      avg_avg_return:                           0.7786681
+      avg_reward_per_step:                      0.08129883
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        72351744
+    env-step-replay-before:                     36536320
+    env-step-used-before:                       72351744
+    ppo-update-before:                          44160
+    env-step-generate-before:                   35815424
+    env-step-all-after:                         72384512
+    env-step-replay-after:                      36536320
+    env-step-used-after:                        72384512
+    ppo-update-after:                           44180
+    env-step-generate-after:                    35848192
+  train-generate:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     31.018318
+    avg_avg_return:                             0.73176897
+    avg_reward_per_step:                        0.0982666
+    proxy_corner:
+      avg_avg_return:                           0.05102656
+      avg_reward_per_step:                      0.0028686523
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     31.018318
+    avg_avg_return:                             0.73176897
+    avg_reward_per_step:                        0.0982666
+    proxy_corner:
+      avg_avg_return:                           0.05102656
+      avg_reward_per_step:                      0.0028686523
+  perf:
+    env_steps_per_second:                       291204.4601817743
+    ppo_updates_per_second:                     42.0734374981357
+    eval-batch-train-duration:                  0.5393212297931314
+    eval-batch-orig-duration:                   0.4827370438724756
+    eval-batch-shift-duration:                  0.5017933342605829
+    eval-batch-tree-duration:                   0.4949013805016875
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.409424
+      cheese-corne_dist_wavg:                   19.748323
+      levels_cheese_is_in_corner_avg:           0.018798828
+      mouse-cheese_dist_finite_avg:             20.946533
+      mouse-cheese_dist_finite_wavg:            8.392389
+      mouse-cheese_dist_solvable_avg:           20.946533
+      mouse-cheese_dist_solvable_wavg:          8.392388
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.6303805
+    visit_patterns:
+      avg_first_visit_time:                     956.86816
+      avg_last_visit_time:                      1105.0913
+      num_replay_batches:                       1115
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0015060129
+    avg_actor_approxkl3:                        0.0008517676
+    avg_actor_clipfrac:                         0.03902588
+    avg_actor_loss:                             -0.0007071428
+    avg_advantage:                              0.07620445
+    avg_critic_clipfrac:                        0.26207885
+    avg_critic_loss:                            0.007596104
+    avg_entropy:                                0.2104518
+    avg_grad_norm_pre_clip:                     1.9160861
+    avg_loss:                                   0.002880457
+    max:
+      max_actor_approxkl1:                      0.0033765656
+      max_actor_approxkl3:                      0.0019458155
+      max_actor_clipfrac:                       0.06591797
+      max_actor_loss:                           0.0012371019
+      max_advantage:                            0.29719532
+      max_critic_clipfrac:                      0.97424316
+      max_critic_loss:                          0.010417624
+      max_entropy:                              0.24022764
+      max_grad_norm_pre_clip:                   3.0190728
+      max_loss:                                 0.0048360983
+    std:
+      std_actor_approxkl1:                      0.0008788483
+      std_actor_approxkl3:                      0.000466045
+      std_actor_clipfrac:                       0.017017236
+      std_actor_loss:                           0.0010047184
+      std_advantage:                            0.11772819
+      std_critic_clipfrac:                      0.35500887
+      std_critic_loss:                          0.0013942687
+      std_entropy:                              0.0148192495
+      std_grad_norm_pre_clip:                   0.54803765
+      std_loss:                                 0.0012703836
+  eval-batch-train:
+    avg_avg_episode_length:                     27.57494
+    avg_avg_return:                             0.83314085
+    avg_benchmark_regret:                       0.15407623
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.112213135
+    proxy_corner:
+      avg_avg_return:                           0.059975248
+      avg_reward_per_step:                      0.0042648315
+  eval-batch-orig:
+    avg_avg_episode_length:                     33.22602
+    avg_avg_return:                             0.78531563
+    avg_benchmark_regret:                       0.19919594
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.06971741
+    proxy_corner:
+      avg_avg_return:                           0.78531563
+      avg_reward_per_step:                      0.06971741
+  eval-batch-shift:
+    avg_avg_episode_length:                     26.237595
+    avg_avg_return:                             0.84596
+    avg_benchmark_regret:                       0.14154597
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.12569427
+    proxy_corner:
+      avg_avg_return:                           0.059113897
+      avg_reward_per_step:                      0.0026931763
+  eval-batch-tree:
+    avg_avg_episode_length:                     31.584969
+    avg_avg_return:                             0.8112175
+    avg_benchmark_regret:                       0.17341772
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.08241272
+    proxy_corner:
+      avg_avg_return:                           0.8112175
+      avg_reward_per_step:                      0.08241272
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        73400320
+    env-step-replay-before:                     37289984
+    env-step-used-before:                       73400320
+    ppo-update-before:                          44800
+    env-step-generate-before:                   36110336
+    env-step-all-after:                         73433088
+    env-step-replay-after:                      37322752
+    env-step-used-after:                        73433088
+    ppo-update-after:                           44820
+    env-step-generate-after:                    36110336
+  train-replay:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     66.44649
+    avg_avg_return:                             0.43325752
+    avg_reward_per_step:                        0.011688232
+    proxy_corner:
+      avg_avg_return:                           0.10549716
+      avg_reward_per_step:                      0.0019226074
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     66.44649
+    avg_avg_return:                             0.43325752
+    avg_reward_per_step:                        0.011688232
+    proxy_corner:
+      avg_avg_return:                           0.10549716
+      avg_reward_per_step:                      0.0019226074
+  perf:
+    env_steps_per_second:                       324005.6331742677
+    ppo_updates_per_second:                     41.969308688673806
+    eval-batch-train-duration:                  0.5383137697353959
+    eval-batch-orig-duration:                   0.46528060361742973
+    eval-batch-shift-duration:                  0.4489488732069731
+    eval-batch-tree-duration:                   0.48107918351888657
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.283936
+      cheese-corne_dist_wavg:                   15.224551
+      levels_cheese_is_in_corner_avg:           0.01928711
+      mouse-cheese_dist_finite_avg:             21.09497
+      mouse-cheese_dist_finite_wavg:            9.313292
+      mouse-cheese_dist_solvable_avg:           21.09497
+      mouse-cheese_dist_solvable_wavg:          9.3132925
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.5881405
+    visit_patterns:
+      avg_first_visit_time:                     966.0869
+      avg_last_visit_time:                      1129.2126
+      num_replay_batches:                       1139
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001459066
+    avg_actor_approxkl3:                        0.0015512918
+    avg_actor_clipfrac:                         0.072467044
+    avg_actor_loss:                             9.6364696e-05
+    avg_advantage:                              -0.012750121
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.009713044
+    avg_entropy:                                0.322022
+    avg_grad_norm_pre_clip:                     2.7755902
+    avg_loss:                                   0.004630865
+    max:
+      max_actor_approxkl1:                      0.0036376934
+      max_actor_approxkl3:                      0.0044005644
+      max_actor_clipfrac:                       0.1616211
+      max_actor_loss:                           0.0030552745
+      max_advantage:                            0.297023
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.012814294
+      max_entropy:                              0.34019604
+      max_grad_norm_pre_clip:                   5.1783166
+      max_loss:                                 0.0091335345
+    std:
+      std_actor_approxkl1:                      0.0009443943
+      std_actor_approxkl3:                      0.0010595489
+      std_actor_clipfrac:                       0.03775894
+      std_actor_loss:                           0.0013048667
+      std_advantage:                            0.1393441
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0015012053
+      std_entropy:                              0.0083671985
+      std_grad_norm_pre_clip:                   1.1539912
+      std_loss:                                 0.001433932
+  eval-batch-train:
+    avg_avg_episode_length:                     27.313704
+    avg_avg_return:                             0.8480592
+    avg_benchmark_regret:                       0.13915789
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.10915375
+    proxy_corner:
+      avg_avg_return:                           0.05837407
+      avg_reward_per_step:                      0.0041275024
+  eval-batch-orig:
+    avg_avg_episode_length:                     33.410507
+    avg_avg_return:                             0.7892922
+    avg_benchmark_regret:                       0.19521931
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.06749725
+    proxy_corner:
+      avg_avg_return:                           0.7892922
+      avg_reward_per_step:                      0.06749725
+  eval-batch-shift:
+    avg_avg_episode_length:                     27.238596
+    avg_avg_return:                             0.84491193
+    avg_benchmark_regret:                       0.14259401
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.11955261
+    proxy_corner:
+      avg_avg_return:                           0.050673265
+      avg_reward_per_step:                      0.0023040771
+  eval-batch-tree:
+    avg_avg_episode_length:                     31.427616
+    avg_avg_return:                             0.81823903
+    avg_benchmark_regret:                       0.1663962
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.08016968
+    proxy_corner:
+      avg_avg_return:                           0.81823903
+      avg_reward_per_step:                      0.08016968
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        74448896
+    env-step-replay-before:                     38043648
+    env-step-used-before:                       74448896
+    ppo-update-before:                          45440
+    env-step-generate-before:                   36405248
+    env-step-all-after:                         74481664
+    env-step-replay-after:                      38043648
+    env-step-used-after:                        74481664
+    ppo-update-after:                           45460
+    env-step-generate-after:                    36438016
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     33.10475
+    avg_avg_return:                             0.73136824
+    avg_reward_per_step:                        0.090545654
+    proxy_corner:
+      avg_avg_return:                           0.05741094
+      avg_reward_per_step:                      0.0018920898
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     33.10475
+    avg_avg_return:                             0.73136824
+    avg_reward_per_step:                        0.090545654
+    proxy_corner:
+      avg_avg_return:                           0.05741094
+      avg_reward_per_step:                      0.0018920898
+  perf:
+    env_steps_per_second:                       242619.0488571356
+    ppo_updates_per_second:                     42.03071228963938
+    eval-batch-train-duration:                  0.541820727288723
+    eval-batch-orig-duration:                   0.48482314962893724
+    eval-batch-shift-duration:                  0.48190844897180796
+    eval-batch-tree-duration:                   0.49161828495562077
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.305664
+      cheese-corne_dist_wavg:                   26.015602
+      levels_cheese_is_in_corner_avg:           0.019042969
+      mouse-cheese_dist_finite_avg:             21.176758
+      mouse-cheese_dist_finite_wavg:            9.313725
+      mouse-cheese_dist_solvable_avg:           21.176758
+      mouse-cheese_dist_solvable_wavg:          9.313727
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016816
+    scoring:
+      avg_scores:                               0.5705032
+    visit_patterns:
+      avg_first_visit_time:                     978.2009
+      avg_last_visit_time:                      1151.2861
+      num_replay_batches:                       1161
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0016166508
+    avg_actor_approxkl3:                        0.0010548908
+    avg_actor_clipfrac:                         0.046099853
+    avg_actor_loss:                             -0.00048658738
+    avg_advantage:                              0.0846355
+    avg_critic_clipfrac:                        0.24124146
+    avg_critic_loss:                            0.007298447
+    avg_entropy:                                0.21617727
+    avg_grad_norm_pre_clip:                     2.1959398
+    avg_loss:                                   0.0029464585
+    max:
+      max_actor_approxkl1:                      0.0033268812
+      max_actor_approxkl3:                      0.0024234855
+      max_actor_clipfrac:                       0.09350586
+      max_actor_loss:                           0.0017808601
+      max_advantage:                            0.3115852
+      max_critic_clipfrac:                      0.9949951
+      max_critic_loss:                          0.0100300945
+      max_entropy:                              0.24116254
+      max_grad_norm_pre_clip:                   3.8357837
+      max_loss:                                 0.006133015
+    std:
+      std_actor_approxkl1:                      0.00093069265
+      std_actor_approxkl3:                      0.0006043011
+      std_actor_clipfrac:                       0.022304831
+      std_actor_loss:                           0.0011393134
+      std_advantage:                            0.11404322
+      std_critic_clipfrac:                      0.3672309
+      std_critic_loss:                          0.0015021956
+      std_entropy:                              0.015508875
+      std_grad_norm_pre_clip:                   0.7780402
+      std_loss:                                 0.0015724496
+  eval-batch-train:
+    avg_avg_episode_length:                     26.649754
+    avg_avg_return:                             0.84213054
+    avg_benchmark_regret:                       0.1450865
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.11064911
+    proxy_corner:
+      avg_avg_return:                           0.06227641
+      avg_reward_per_step:                      0.0048675537
+  eval-batch-orig:
+    avg_avg_episode_length:                     32.880558
+    avg_avg_return:                             0.7934699
+    avg_benchmark_regret:                       0.1910416
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07236481
+    proxy_corner:
+      avg_avg_return:                           0.7934699
+      avg_reward_per_step:                      0.07236481
+  eval-batch-shift:
+    avg_avg_episode_length:                     26.214584
+    avg_avg_return:                             0.8583251
+    avg_benchmark_regret:                       0.1291809
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.12271118
+    proxy_corner:
+      avg_avg_return:                           0.06419553
+      avg_reward_per_step:                      0.0028381348
+  eval-batch-tree:
+    avg_avg_episode_length:                     30.50663
+    avg_avg_return:                             0.8265606
+    avg_benchmark_regret:                       0.15807459
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0848465
+    proxy_corner:
+      avg_avg_return:                           0.8265606
+      avg_reward_per_step:                      0.0848465
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        75497472
+    env-step-replay-before:                     38567936
+    env-step-used-before:                       75497472
+    ppo-update-before:                          46080
+    env-step-generate-before:                   36929536
+    env-step-all-after:                         75530240
+    env-step-replay-after:                      38600704
+    env-step-used-after:                        75530240
+    ppo-update-after:                           46100
+    env-step-generate-after:                    36929536
+  train-replay:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     69.49873
+    avg_avg_return:                             0.3862809
+    avg_reward_per_step:                        0.0107421875
+    proxy_corner:
+      avg_avg_return:                           0.109209955
+      avg_reward_per_step:                      0.0016174316
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     69.49873
+    avg_avg_return:                             0.3862809
+    avg_reward_per_step:                        0.0107421875
+    proxy_corner:
+      avg_avg_return:                           0.109209955
+      avg_reward_per_step:                      0.0016174316
+  perf:
+    env_steps_per_second:                       314209.22039485263
+    ppo_updates_per_second:                     41.71017802305625
+    eval-batch-train-duration:                  0.5212179683148861
+    eval-batch-orig-duration:                   0.48677159287035465
+    eval-batch-shift-duration:                  0.4890828710049391
+    eval-batch-tree-duration:                   0.49252590723335743
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.44751
+      cheese-corne_dist_wavg:                   12.560943
+      levels_cheese_is_in_corner_avg:           0.018066406
+      mouse-cheese_dist_finite_avg:             21.206055
+      mouse-cheese_dist_finite_wavg:            7.510322
+      mouse-cheese_dist_solvable_avg:           21.206055
+      mouse-cheese_dist_solvable_wavg:          7.510322
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5839175
+    visit_patterns:
+      avg_first_visit_time:                     994.1643
+      avg_last_visit_time:                      1168.271
+      num_replay_batches:                       1178
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00075087184
+    avg_actor_approxkl3:                        0.0009338801
+    avg_actor_clipfrac:                         0.048980713
+    avg_actor_loss:                             -0.0009856429
+    avg_advantage:                              -0.03488656
+    avg_critic_clipfrac:                        0.00012817384
+    avg_critic_loss:                            0.01157731
+    avg_entropy:                                0.32482788
+    avg_grad_norm_pre_clip:                     1.7353321
+    avg_loss:                                   0.004478185
+    max:
+      max_actor_approxkl1:                      0.0016438933
+      max_actor_approxkl3:                      0.0018677767
+      max_actor_clipfrac:                       0.103027344
+      max_actor_loss:                           0.00085642934
+      max_advantage:                            0.2392847
+      max_critic_clipfrac:                      0.0010986328
+      max_critic_loss:                          0.0142406365
+      max_entropy:                              0.34746423
+      max_grad_norm_pre_clip:                   2.9743576
+      max_loss:                                 0.006402591
+    std:
+      std_actor_approxkl1:                      0.0005136584
+      std_actor_approxkl3:                      0.00042363256
+      std_actor_clipfrac:                       0.023049315
+      std_actor_loss:                           0.0012505357
+      std_advantage:                            0.15123713
+      std_critic_clipfrac:                      0.00031473074
+      std_critic_loss:                          0.0014690848
+      std_entropy:                              0.011528422
+      std_grad_norm_pre_clip:                   0.60710615
+      std_loss:                                 0.0013700416
+  eval-batch-train:
+    avg_avg_episode_length:                     26.428358
+    avg_avg_return:                             0.8484325
+    avg_benchmark_regret:                       0.13878457
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.11257172
+    proxy_corner:
+      avg_avg_return:                           0.057883166
+      avg_reward_per_step:                      0.004371643
+  eval-batch-orig:
+    avg_avg_episode_length:                     32.720894
+    avg_avg_return:                             0.7931541
+    avg_benchmark_regret:                       0.19135746
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07009888
+    proxy_corner:
+      avg_avg_return:                           0.7931541
+      avg_reward_per_step:                      0.07009888
+  eval-batch-shift:
+    avg_avg_episode_length:                     26.193748
+    avg_avg_return:                             0.8531209
+    avg_benchmark_regret:                       0.134385
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.12445068
+    proxy_corner:
+      avg_avg_return:                           0.054326348
+      avg_reward_per_step:                      0.002632141
+  eval-batch-tree:
+    avg_avg_episode_length:                     30.165445
+    avg_avg_return:                             0.82971936
+    avg_benchmark_regret:                       0.15491582
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.08304596
+    proxy_corner:
+      avg_avg_return:                           0.82971936
+      avg_reward_per_step:                      0.08304596
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        76546048
+    env-step-replay-before:                     39092224
+    env-step-used-before:                       76546048
+    ppo-update-before:                          46720
+    env-step-generate-before:                   37453824
+    env-step-all-after:                         76578816
+    env-step-replay-after:                      39092224
+    env-step-used-after:                        76578816
+    ppo-update-after:                           46740
+    env-step-generate-after:                    37486592
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     30.095747
+    avg_avg_return:                             0.7480053
+    avg_reward_per_step:                        0.12207031
+    proxy_corner:
+      avg_avg_return:                           0.028252251
+      avg_reward_per_step:                      0.0015869141
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     30.095747
+    avg_avg_return:                             0.7480053
+    avg_reward_per_step:                        0.12207031
+    proxy_corner:
+      avg_avg_return:                           0.028252251
+      avg_reward_per_step:                      0.0015869141
+  perf:
+    env_steps_per_second:                       325009.33190016914
+    ppo_updates_per_second:                     40.76905075170937
+    eval-batch-train-duration:                  0.5331713799387217
+    eval-batch-orig-duration:                   0.4915109546855092
+    eval-batch-shift-duration:                  0.47523974999785423
+    eval-batch-tree-duration:                   0.4713790910318494
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.446777
+      cheese-corne_dist_wavg:                   18.853264
+      levels_cheese_is_in_corner_avg:           0.017089844
+      mouse-cheese_dist_finite_avg:             21.352783
+      mouse-cheese_dist_finite_wavg:            9.326616
+      mouse-cheese_dist_solvable_avg:           21.352783
+      mouse-cheese_dist_solvable_wavg:          9.326617
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.5921022
+    visit_patterns:
+      avg_first_visit_time:                     1012.60425
+      avg_last_visit_time:                      1183.2012
+      num_replay_batches:                       1193
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001078559
+    avg_actor_approxkl3:                        0.00082803966
+    avg_actor_clipfrac:                         0.038629152
+    avg_actor_loss:                             -0.000546352
+    avg_advantage:                              0.024351057
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.005100102
+    avg_entropy:                                0.18328805
+    avg_grad_norm_pre_clip:                     1.8698931
+    avg_loss:                                   0.0018204114
+    max:
+      max_actor_approxkl1:                      0.002587587
+      max_actor_approxkl3:                      0.0018268727
+      max_actor_clipfrac:                       0.07104492
+      max_actor_loss:                           0.0026248395
+      max_advantage:                            0.22121572
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0073427632
+      max_entropy:                              0.20113015
+      max_grad_norm_pre_clip:                   3.0245268
+      max_loss:                                 0.005579078
+    std:
+      std_actor_approxkl1:                      0.0007926099
+      std_actor_approxkl3:                      0.000444904
+      std_actor_clipfrac:                       0.017430874
+      std_actor_loss:                           0.0010771151
+      std_advantage:                            0.10060451
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010465522
+      std_entropy:                              0.012046171
+      std_grad_norm_pre_clip:                   0.6404923
+      std_loss:                                 0.0012644979
+  eval-batch-train:
+    avg_avg_episode_length:                     26.345821
+    avg_avg_return:                             0.85184073
+    avg_benchmark_regret:                       0.13537636
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.11541748
+    proxy_corner:
+      avg_avg_return:                           0.04801525
+      avg_reward_per_step:                      0.004211426
+  eval-batch-orig:
+    avg_avg_episode_length:                     32.192184
+    avg_avg_return:                             0.80536634
+    avg_benchmark_regret:                       0.17914517
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.06750488
+    proxy_corner:
+      avg_avg_return:                           0.80536634
+      avg_reward_per_step:                      0.06750488
+  eval-batch-shift:
+    avg_avg_episode_length:                     26.964785
+    avg_avg_return:                             0.8410984
+    avg_benchmark_regret:                       0.14640749
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.12445068
+    proxy_corner:
+      avg_avg_return:                           0.05161786
+      avg_reward_per_step:                      0.0025100708
+  eval-batch-tree:
+    avg_avg_episode_length:                     30.491888
+    avg_avg_return:                             0.82302725
+    avg_benchmark_regret:                       0.16160795
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.083465576
+    proxy_corner:
+      avg_avg_return:                           0.82302725
+      avg_reward_per_step:                      0.083465576
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        77594624
+    env-step-replay-before:                     39714816
+    env-step-used-before:                       77594624
+    ppo-update-before:                          47360
+    env-step-generate-before:                   37879808
+    env-step-all-after:                         77627392
+    env-step-replay-after:                      39714816
+    env-step-used-after:                        77627392
+    ppo-update-after:                           47380
+    env-step-generate-after:                    37912576
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     28.311855
+    avg_avg_return:                             0.76787925
+    avg_reward_per_step:                        0.12136841
+    proxy_corner:
+      avg_avg_return:                           0.04791078
+      avg_reward_per_step:                      0.0032043457
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     28.311855
+    avg_avg_return:                             0.76787925
+    avg_reward_per_step:                        0.12136841
+    proxy_corner:
+      avg_avg_return:                           0.04791078
+      avg_reward_per_step:                      0.0032043457
+  perf:
+    env_steps_per_second:                       213451.98791461022
+    ppo_updates_per_second:                     42.04047512359225
+    eval-batch-train-duration:                  0.5172017281875014
+    eval-batch-orig-duration:                   0.48337447270751
+    eval-batch-shift-duration:                  0.4827604293823242
+    eval-batch-tree-duration:                   0.45138294994831085
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.51123
+      cheese-corne_dist_wavg:                   15.253133
+      levels_cheese_is_in_corner_avg:           0.016113281
+      mouse-cheese_dist_finite_avg:             21.503906
+      mouse-cheese_dist_finite_wavg:            9.347604
+      mouse-cheese_dist_solvable_avg:           21.503906
+      mouse-cheese_dist_solvable_wavg:          9.347605
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5930456
+    visit_patterns:
+      avg_first_visit_time:                     1026.6343
+      avg_last_visit_time:                      1202.3301
+      num_replay_batches:                       1212
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012518152
+    avg_actor_approxkl3:                        0.001142268
+    avg_actor_clipfrac:                         0.04796753
+    avg_actor_loss:                             0.00011014622
+    avg_advantage:                              0.019056402
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0044269753
+    avg_entropy:                                0.1901943
+    avg_grad_norm_pre_clip:                     2.378284
+    avg_loss:                                   0.0021334395
+    max:
+      max_actor_approxkl1:                      0.0038242268
+      max_actor_approxkl3:                      0.0024981338
+      max_actor_clipfrac:                       0.08483887
+      max_actor_loss:                           0.0031558871
+      max_advantage:                            0.27019233
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0064193467
+      max_entropy:                              0.22650596
+      max_grad_norm_pre_clip:                   4.018846
+      max_loss:                                 0.0057848846
+    std:
+      std_actor_approxkl1:                      0.0009373792
+      std_actor_approxkl3:                      0.00057554693
+      std_actor_clipfrac:                       0.018549968
+      std_actor_loss:                           0.0014025422
+      std_advantage:                            0.09368409
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0009822978
+      std_entropy:                              0.01604949
+      std_grad_norm_pre_clip:                   0.86445916
+      std_loss:                                 0.0015238337
+  eval-batch-train:
+    avg_avg_episode_length:                     25.892244
+    avg_avg_return:                             0.85721374
+    avg_benchmark_regret:                       0.13000329
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.11404419
+    proxy_corner:
+      avg_avg_return:                           0.05912435
+      avg_reward_per_step:                      0.004142761
+  eval-batch-orig:
+    avg_avg_episode_length:                     32.44089
+    avg_avg_return:                             0.7978498
+    avg_benchmark_regret:                       0.18666172
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.068611145
+    proxy_corner:
+      avg_avg_return:                           0.7978498
+      avg_reward_per_step:                      0.068611145
+  eval-batch-shift:
+    avg_avg_episode_length:                     26.352615
+    avg_avg_return:                             0.8443024
+    avg_benchmark_regret:                       0.14320353
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.12500763
+    proxy_corner:
+      avg_avg_return:                           0.055776685
+      avg_reward_per_step:                      0.0027313232
+  eval-batch-tree:
+    avg_avg_episode_length:                     30.44223
+    avg_avg_return:                             0.8218006
+    avg_benchmark_regret:                       0.16283461
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.08589935
+    proxy_corner:
+      avg_avg_return:                           0.8218006
+      avg_reward_per_step:                      0.08589935
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        78643200
+    env-step-replay-before:                     40239104
+    env-step-used-before:                       78643200
+    ppo-update-before:                          48000
+    env-step-generate-before:                   38404096
+    env-step-all-after:                         78675968
+    env-step-replay-after:                      40271872
+    env-step-used-after:                        78675968
+    ppo-update-after:                           48020
+    env-step-generate-after:                    38404096
+  train-replay:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     69.75842
+    avg_avg_return:                             0.38821948
+    avg_reward_per_step:                        0.010528564
+    proxy_corner:
+      avg_avg_return:                           0.11203507
+      avg_reward_per_step:                      0.0017700195
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     69.75842
+    avg_avg_return:                             0.38821948
+    avg_reward_per_step:                        0.010528564
+    proxy_corner:
+      avg_avg_return:                           0.11203507
+      avg_reward_per_step:                      0.0017700195
+  perf:
+    env_steps_per_second:                       298729.47402859374
+    ppo_updates_per_second:                     42.08868905935248
+    eval-batch-train-duration:                  0.5349810067564249
+    eval-batch-orig-duration:                   0.4844863284379244
+    eval-batch-shift-duration:                  0.48193311784416437
+    eval-batch-tree-duration:                   0.4828549586236477
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.60913
+      cheese-corne_dist_wavg:                   15.249136
+      levels_cheese_is_in_corner_avg:           0.01586914
+      mouse-cheese_dist_finite_avg:             21.649414
+      mouse-cheese_dist_finite_wavg:            9.370583
+      mouse-cheese_dist_solvable_avg:           21.649414
+      mouse-cheese_dist_solvable_wavg:          9.370582
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.99999
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016816
+    scoring:
+      avg_scores:                               0.60096043
+    visit_patterns:
+      avg_first_visit_time:                     1038.7458
+      avg_last_visit_time:                      1219.1975
+      num_replay_batches:                       1229
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0018363191
+    avg_actor_approxkl3:                        0.0015860368
+    avg_actor_clipfrac:                         0.06676026
+    avg_actor_loss:                             0.00037815535
+    avg_advantage:                              -0.034022145
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.011343221
+    avg_entropy:                                0.3115494
+    avg_grad_norm_pre_clip:                     2.8117588
+    avg_loss:                                   0.005738217
+    max:
+      max_actor_approxkl1:                      0.0061804904
+      max_actor_approxkl3:                      0.005294764
+      max_actor_clipfrac:                       0.14819336
+      max_actor_loss:                           0.005419612
+      max_advantage:                            0.24035233
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0127808135
+      max_entropy:                              0.32229853
+      max_grad_norm_pre_clip:                   5.391264
+      max_loss:                                 0.01040401
+    std:
+      std_actor_approxkl1:                      0.0016538589
+      std_actor_approxkl3:                      0.0012624973
+      std_actor_clipfrac:                       0.03679026
+      std_actor_loss:                           0.0018904059
+      std_advantage:                            0.1495329
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010405874
+      std_entropy:                              0.009445087
+      std_grad_norm_pre_clip:                   1.3008376
+      std_loss:                                 0.0019686026
+  eval-batch-train:
+    avg_avg_episode_length:                     25.836956
+    avg_avg_return:                             0.8458927
+    avg_benchmark_regret:                       0.14132433
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.11640167
+    proxy_corner:
+      avg_avg_return:                           0.061154537
+      avg_reward_per_step:                      0.0048980713
+  eval-batch-orig:
+    avg_avg_episode_length:                     32.831543
+    avg_avg_return:                             0.78608423
+    avg_benchmark_regret:                       0.19842732
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07102966
+    proxy_corner:
+      avg_avg_return:                           0.78608423
+      avg_reward_per_step:                      0.07102966
+  eval-batch-shift:
+    avg_avg_episode_length:                     25.66214
+    avg_avg_return:                             0.8436874
+    avg_benchmark_regret:                       0.14381848
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.12472534
+    proxy_corner:
+      avg_avg_return:                           0.05749463
+      avg_reward_per_step:                      0.0028152466
+  eval-batch-tree:
+    avg_avg_episode_length:                     29.276186
+    avg_avg_return:                             0.8220021
+    avg_benchmark_regret:                       0.16263308
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.08682251
+    proxy_corner:
+      avg_avg_return:                           0.8220021
+      avg_reward_per_step:                      0.08682251
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        79691776
+    env-step-replay-before:                     40861696
+    env-step-used-before:                       79691776
+    ppo-update-before:                          48640
+    env-step-generate-before:                   38830080
+    env-step-all-after:                         79724544
+    env-step-replay-after:                      40861696
+    env-step-used-after:                        79724544
+    ppo-update-after:                           48660
+    env-step-generate-after:                    38862848
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     29.023815
+    avg_avg_return:                             0.75763595
+    avg_reward_per_step:                        0.10562134
+    proxy_corner:
+      avg_avg_return:                           0.07117566
+      avg_reward_per_step:                      0.0032043457
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     29.023815
+    avg_avg_return:                             0.75763595
+    avg_reward_per_step:                        0.10562134
+    proxy_corner:
+      avg_avg_return:                           0.07117566
+      avg_reward_per_step:                      0.0032043457
+  perf:
+    env_steps_per_second:                       290497.08408983436
+    ppo_updates_per_second:                     41.61842232105835
+    eval-batch-train-duration:                  0.5340315513312817
+    eval-batch-orig-duration:                   0.49148973543196917
+    eval-batch-shift-duration:                  0.48513125721365213
+    eval-batch-tree-duration:                   0.4721246277913451
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.682373
+      cheese-corne_dist_wavg:                   23.331896
+      levels_cheese_is_in_corner_avg:           0.01586914
+      mouse-cheese_dist_finite_avg:             21.7146
+      mouse-cheese_dist_finite_wavg:            10.274797
+      mouse-cheese_dist_solvable_avg:           21.7146
+      mouse-cheese_dist_solvable_wavg:          10.274798
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.6075694
+    visit_patterns:
+      avg_first_visit_time:                     1049.7078
+      avg_last_visit_time:                      1237.2437
+      num_replay_batches:                       1247
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011289873
+    avg_actor_approxkl3:                        0.000814854
+    avg_actor_clipfrac:                         0.035546876
+    avg_actor_loss:                             -0.0006534077
+    avg_advantage:                              0.070251845
+    avg_critic_clipfrac:                        0.15354005
+    avg_critic_loss:                            0.00632236
+    avg_entropy:                                0.1921152
+    avg_grad_norm_pre_clip:                     1.9038162
+    avg_loss:                                   0.002315657
+    max:
+      max_actor_approxkl1:                      0.003775956
+      max_actor_approxkl3:                      0.0026889686
+      max_actor_clipfrac:                       0.08911133
+      max_actor_loss:                           0.001347512
+      max_advantage:                            0.307378
+      max_critic_clipfrac:                      0.8425293
+      max_critic_loss:                          0.010061694
+      max_entropy:                              0.220883
+      max_grad_norm_pre_clip:                   3.948364
+      max_loss:                                 0.0049508293
+    std:
+      std_actor_approxkl1:                      0.001052821
+      std_actor_approxkl3:                      0.0006416751
+      std_actor_clipfrac:                       0.021681301
+      std_actor_loss:                           0.0007610945
+      std_advantage:                            0.10642394
+      std_critic_clipfrac:                      0.2665883
+      std_critic_loss:                          0.0014947768
+      std_entropy:                              0.017946523
+      std_grad_norm_pre_clip:                   0.7115422
+      std_loss:                                 0.001131989
+  eval-batch-train:
+    avg_avg_episode_length:                     25.274216
+    avg_avg_return:                             0.857411
+    avg_benchmark_regret:                       0.12980604
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.114234924
+    proxy_corner:
+      avg_avg_return:                           0.06852315
+      avg_reward_per_step:                      0.0041275024
+  eval-batch-orig:
+    avg_avg_episode_length:                     32.02854
+    avg_avg_return:                             0.8109957
+    avg_benchmark_regret:                       0.1735158
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.0688324
+    proxy_corner:
+      avg_avg_return:                           0.8109957
+      avg_reward_per_step:                      0.0688324
+  eval-batch-shift:
+    avg_avg_episode_length:                     25.239178
+    avg_avg_return:                             0.8592304
+    avg_benchmark_regret:                       0.1282756
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.12937164
+    proxy_corner:
+      avg_avg_return:                           0.060926814
+      avg_reward_per_step:                      0.0028457642
+  eval-batch-tree:
+    avg_avg_episode_length:                     30.377943
+    avg_avg_return:                             0.8285851
+    avg_benchmark_regret:                       0.15605007
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.08406067
+    proxy_corner:
+      avg_avg_return:                           0.8285851
+      avg_reward_per_step:                      0.08406067
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        80740352
+    env-step-replay-before:                     41222144
+    env-step-used-before:                       80740352
+    ppo-update-before:                          49280
+    env-step-generate-before:                   39518208
+    env-step-all-after:                         80773120
+    env-step-replay-after:                      41254912
+    env-step-used-after:                        80773120
+    ppo-update-after:                           49300
+    env-step-generate-after:                    39518208
+  train-replay:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     70.21509
+    avg_avg_return:                             0.38769716
+    avg_reward_per_step:                        0.010284424
+    proxy_corner:
+      avg_avg_return:                           0.15445614
+      avg_reward_per_step:                      0.0026245117
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     70.21509
+    avg_avg_return:                             0.38769716
+    avg_reward_per_step:                        0.010284424
+    proxy_corner:
+      avg_avg_return:                           0.15445614
+      avg_reward_per_step:                      0.0026245117
+  perf:
+    env_steps_per_second:                       322816.7697467588
+    ppo_updates_per_second:                     42.691713913981005
+    eval-batch-train-duration:                  0.5380685087293386
+    eval-batch-orig-duration:                   0.47999033238738775
+    eval-batch-shift-duration:                  0.46794704906642437
+    eval-batch-tree-duration:                   0.4687563795596361
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.746338
+      cheese-corne_dist_wavg:                   7.1840587
+      levels_cheese_is_in_corner_avg:           0.016113281
+      mouse-cheese_dist_finite_avg:             21.803223
+      mouse-cheese_dist_finite_wavg:            8.487135
+      mouse-cheese_dist_solvable_avg:           21.803223
+      mouse-cheese_dist_solvable_wavg:          8.487135
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.60960245
+    visit_patterns:
+      avg_first_visit_time:                     1063.3501
+      avg_last_visit_time:                      1249.2439
+      num_replay_batches:                       1259
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0006967418
+    avg_actor_approxkl3:                        0.00092970236
+    avg_actor_clipfrac:                         0.048931886
+    avg_actor_loss:                             -0.0006952219
+    avg_advantage:                              -0.044022143
+    avg_critic_clipfrac:                        0.0009460449
+    avg_critic_loss:                            0.011523261
+    avg_entropy:                                0.30866
+    avg_grad_norm_pre_clip:                     2.3359296
+    avg_loss:                                   0.0047577475
+    max:
+      max_actor_approxkl1:                      0.0017051093
+      max_actor_approxkl3:                      0.0016406716
+      max_actor_clipfrac:                       0.08312988
+      max_actor_loss:                           0.0015602708
+      max_advantage:                            0.22037882
+      max_critic_clipfrac:                      0.009765625
+      max_critic_loss:                          0.014306003
+      max_entropy:                              0.32685983
+      max_grad_norm_pre_clip:                   3.7533495
+      max_loss:                                 0.0077415723
+    std:
+      std_actor_approxkl1:                      0.00044144862
+      std_actor_approxkl3:                      0.00037501115
+      std_actor_clipfrac:                       0.018459287
+      std_actor_loss:                           0.00096741033
+      std_advantage:                            0.15062457
+      std_critic_clipfrac:                      0.0026755058
+      std_critic_loss:                          0.0012648384
+      std_entropy:                              0.009374982
+      std_grad_norm_pre_clip:                   0.77881867
+      std_loss:                                 0.001154682
+  eval-batch-train:
+    avg_avg_episode_length:                     26.162441
+    avg_avg_return:                             0.8505999
+    avg_benchmark_regret:                       0.13661717
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.11491394
+    proxy_corner:
+      avg_avg_return:                           0.059780903
+      avg_reward_per_step:                      0.004219055
+  eval-batch-orig:
+    avg_avg_episode_length:                     31.77395
+    avg_avg_return:                             0.8114567
+    avg_benchmark_regret:                       0.17305487
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.06903076
+    proxy_corner:
+      avg_avg_return:                           0.8114567
+      avg_reward_per_step:                      0.06903076
+  eval-batch-shift:
+    avg_avg_episode_length:                     25.942963
+    avg_avg_return:                             0.8451519
+    avg_benchmark_regret:                       0.14235407
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.12819672
+    proxy_corner:
+      avg_avg_return:                           0.05278028
+      avg_reward_per_step:                      0.0026397705
+  eval-batch-tree:
+    avg_avg_episode_length:                     29.355381
+    avg_avg_return:                             0.8365839
+    avg_benchmark_regret:                       0.14805128
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.085365295
+    proxy_corner:
+      avg_avg_return:                           0.8365839
+      avg_reward_per_step:                      0.085365295
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        81788928
+    env-step-replay-before:                     41615360
+    env-step-used-before:                       81788928
+    ppo-update-before:                          49920
+    env-step-generate-before:                   40173568
+    env-step-all-after:                         81821696
+    env-step-replay-after:                      41648128
+    env-step-used-after:                        81821696
+    ppo-update-after:                           49940
+    env-step-generate-after:                    40173568
+  train-replay:
+    shift_proportion:                           1.0
+    avg_avg_episode_length:                     69.46307
+    avg_avg_return:                             0.38525176
+    avg_reward_per_step:                        0.010528564
+    proxy_corner:
+      avg_avg_return:                           0.069980085
+      avg_reward_per_step:                      0.0012512207
+  train-all:
+    shift_proportion:                           1.0
+    avg_avg_episode_length:                     69.46307
+    avg_avg_return:                             0.38525176
+    avg_reward_per_step:                        0.010528564
+    proxy_corner:
+      avg_avg_return:                           0.069980085
+      avg_reward_per_step:                      0.0012512207
+  perf:
+    env_steps_per_second:                       348932.7217425649
+    ppo_updates_per_second:                     41.99574991561283
+    eval-batch-train-duration:                  0.5080201402306557
+    eval-batch-orig-duration:                   0.4793712329119444
+    eval-batch-shift-duration:                  0.44162625912576914
+    eval-batch-tree-duration:                   0.4673241265118122
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.725586
+      cheese-corne_dist_wavg:                   7.1934857
+      levels_cheese_is_in_corner_avg:           0.0146484375
+      mouse-cheese_dist_finite_avg:             21.891846
+      mouse-cheese_dist_finite_wavg:            8.499022
+      mouse-cheese_dist_solvable_avg:           21.891846
+      mouse-cheese_dist_solvable_wavg:          8.4990225
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.61543363
+    visit_patterns:
+      avg_first_visit_time:                     1077.511
+      avg_last_visit_time:                      1261.2244
+      num_replay_batches:                       1271
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0029130497
+    avg_actor_approxkl3:                        0.0029538237
+    avg_actor_clipfrac:                         0.09861451
+    avg_actor_loss:                             0.002342766
+    avg_advantage:                              -0.038228896
+    avg_critic_clipfrac:                        0.00040893556
+    avg_critic_loss:                            0.011681213
+    avg_entropy:                                0.2990819
+    avg_grad_norm_pre_clip:                     3.4273255
+    avg_loss:                                   0.007884291
+    max:
+      max_actor_approxkl1:                      0.01158711
+      max_actor_approxkl3:                      0.00834257
+      max_actor_clipfrac:                       0.17163086
+      max_actor_loss:                           0.009748727
+      max_advantage:                            0.21900606
+      max_critic_clipfrac:                      0.0036621094
+      max_critic_loss:                          0.014611442
+      max_entropy:                              0.31518972
+      max_grad_norm_pre_clip:                   5.404726
+      max_loss:                                 0.014331366
+    std:
+      std_actor_approxkl1:                      0.0026791517
+      std_actor_approxkl3:                      0.0022077754
+      std_actor_clipfrac:                       0.043524414
+      std_actor_loss:                           0.0030053512
+      std_advantage:                            0.1509901
+      std_critic_clipfrac:                      0.00092021516
+      std_critic_loss:                          0.001547203
+      std_entropy:                              0.009618208
+      std_grad_norm_pre_clip:                   1.160264
+      std_loss:                                 0.0028768291
+  eval-batch-train:
+    avg_avg_episode_length:                     26.104864
+    avg_avg_return:                             0.8515957
+    avg_benchmark_regret:                       0.13562137
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.11273956
+    proxy_corner:
+      avg_avg_return:                           0.055575706
+      avg_reward_per_step:                      0.0035476685
+  eval-batch-orig:
+    avg_avg_episode_length:                     31.821909
+    avg_avg_return:                             0.81423706
+    avg_benchmark_regret:                       0.1702745
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.06491089
+    proxy_corner:
+      avg_avg_return:                           0.81423706
+      avg_reward_per_step:                      0.06491089
+  eval-batch-shift:
+    avg_avg_episode_length:                     26.023783
+    avg_avg_return:                             0.84371376
+    avg_benchmark_regret:                       0.14379218
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.12805176
+    proxy_corner:
+      avg_avg_return:                           0.050671924
+      avg_reward_per_step:                      0.0025100708
+  eval-batch-tree:
+    avg_avg_episode_length:                     30.375946
+    avg_avg_return:                             0.8287225
+    avg_benchmark_regret:                       0.15591276
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.08255768
+    proxy_corner:
+      avg_avg_return:                           0.8287225
+      avg_reward_per_step:                      0.08255768
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        82837504
+    env-step-replay-before:                     42336256
+    env-step-used-before:                       82837504
+    ppo-update-before:                          50560
+    env-step-generate-before:                   40501248
+    env-step-all-after:                         82870272
+    env-step-replay-after:                      42369024
+    env-step-used-after:                        82870272
+    ppo-update-after:                           50580
+    env-step-generate-after:                    40501248
+  train-replay:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     70.24391
+    avg_avg_return:                             0.37459671
+    avg_reward_per_step:                        0.010314941
+    proxy_corner:
+      avg_avg_return:                           0.05977436
+      avg_reward_per_step:                      0.0009765625
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     70.24391
+    avg_avg_return:                             0.37459671
+    avg_reward_per_step:                        0.010314941
+    proxy_corner:
+      avg_avg_return:                           0.05977436
+      avg_reward_per_step:                      0.0009765625
+  perf:
+    env_steps_per_second:                       289907.7985191441
+    ppo_updates_per_second:                     42.19738797024481
+    eval-batch-train-duration:                  0.5427458481863141
+    eval-batch-orig-duration:                   0.49288997519761324
+    eval-batch-shift-duration:                  0.4874675711616874
+    eval-batch-tree-duration:                   0.4978889785706997
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.770752
+      cheese-corne_dist_wavg:                   7.183915
+      levels_cheese_is_in_corner_avg:           0.0146484375
+      mouse-cheese_dist_finite_avg:             22.001465
+      mouse-cheese_dist_finite_wavg:            7.6100526
+      mouse-cheese_dist_solvable_avg:           22.001465
+      mouse-cheese_dist_solvable_wavg:          7.6100535
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.600453
+    visit_patterns:
+      avg_first_visit_time:                     1086.7122
+      avg_last_visit_time:                      1283.2078
+      num_replay_batches:                       1293
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0016640582
+    avg_actor_approxkl3:                        0.001095375
+    avg_actor_clipfrac:                         0.05393677
+    avg_actor_loss:                             -0.00062893
+    avg_advantage:                              -0.018121181
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.01114274
+    avg_entropy:                                0.31440875
+    avg_grad_norm_pre_clip:                     2.10214
+    avg_loss:                                   0.004628031
+    max:
+      max_actor_approxkl1:                      0.00342635
+      max_actor_approxkl3:                      0.0020069946
+      max_actor_clipfrac:                       0.088378906
+      max_actor_loss:                           0.0014640689
+      max_advantage:                            0.27890486
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.013075212
+      max_entropy:                              0.33523723
+      max_grad_norm_pre_clip:                   2.9381742
+      max_loss:                                 0.0070462977
+    std:
+      std_actor_approxkl1:                      0.00086761324
+      std_actor_approxkl3:                      0.00046752763
+      std_actor_clipfrac:                       0.019954903
+      std_actor_loss:                           0.0009631002
+      std_advantage:                            0.14885588
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0012558016
+      std_entropy:                              0.011091753
+      std_grad_norm_pre_clip:                   0.48859376
+      std_loss:                                 0.0009665723
+  eval-batch-train:
+    avg_avg_episode_length:                     27.117336
+    avg_avg_return:                             0.84118986
+    avg_benchmark_regret:                       0.14602727
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.111343384
+    proxy_corner:
+      avg_avg_return:                           0.056415744
+      avg_reward_per_step:                      0.0044021606
+  eval-batch-orig:
+    avg_avg_episode_length:                     33.289337
+    avg_avg_return:                             0.8034252
+    avg_benchmark_regret:                       0.1810863
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.065452576
+    proxy_corner:
+      avg_avg_return:                           0.8034252
+      avg_reward_per_step:                      0.065452576
+  eval-batch-shift:
+    avg_avg_episode_length:                     25.934265
+    avg_avg_return:                             0.858891
+    avg_benchmark_regret:                       0.12861499
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.12597656
+    proxy_corner:
+      avg_avg_return:                           0.060856063
+      avg_reward_per_step:                      0.0028076172
+  eval-batch-tree:
+    avg_avg_episode_length:                     30.397305
+    avg_avg_return:                             0.82582855
+    avg_benchmark_regret:                       0.15880665
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.07997131
+    proxy_corner:
+      avg_avg_return:                           0.82582855
+      avg_reward_per_step:                      0.07997131
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        83886080
+    env-step-replay-before:                     42860544
+    env-step-used-before:                       83886080
+    ppo-update-before:                          51200
+    env-step-generate-before:                   41025536
+    env-step-all-after:                         83918848
+    env-step-replay-after:                      42893312
+    env-step-used-after:                        83918848
+    ppo-update-after:                           51220
+    env-step-generate-after:                    41025536
+  train-replay:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     71.6833
+    avg_avg_return:                             0.35568792
+    avg_reward_per_step:                        0.009185791
+    proxy_corner:
+      avg_avg_return:                           0.099733554
+      avg_reward_per_step:                      0.0015563965
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     71.6833
+    avg_avg_return:                             0.35568792
+    avg_reward_per_step:                        0.009185791
+    proxy_corner:
+      avg_avg_return:                           0.099733554
+      avg_reward_per_step:                      0.0015563965
+  perf:
+    env_steps_per_second:                       319073.26311164675
+    ppo_updates_per_second:                     42.19697057241427
+    eval-batch-train-duration:                  0.5246372697874904
+    eval-batch-orig-duration:                   0.4864678317680955
+    eval-batch-shift-duration:                  0.48580074217170477
+    eval-batch-tree-duration:                   0.4920066576451063
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.77832
+      cheese-corne_dist_wavg:                   15.271363
+      levels_cheese_is_in_corner_avg:           0.01586914
+      mouse-cheese_dist_finite_avg:             22.03833
+      mouse-cheese_dist_finite_wavg:            9.401446
+      mouse-cheese_dist_solvable_avg:           22.03833
+      mouse-cheese_dist_solvable_wavg:          9.401447
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.60751414
+    visit_patterns:
+      avg_first_visit_time:                     1103.6548
+      avg_last_visit_time:                      1299.1641
+      num_replay_batches:                       1309
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00090651197
+    avg_actor_approxkl3:                        0.0008443797
+    avg_actor_clipfrac:                         0.04486084
+    avg_actor_loss:                             -0.00092289003
+    avg_advantage:                              -0.049400978
+    avg_critic_clipfrac:                        0.025805665
+    avg_critic_loss:                            0.012631205
+    avg_entropy:                                0.3106519
+    avg_grad_norm_pre_clip:                     1.9602537
+    avg_loss:                                   0.00508206
+    max:
+      max_actor_approxkl1:                      0.0023770276
+      max_actor_approxkl3:                      0.0014390767
+      max_actor_clipfrac:                       0.07055664
+      max_actor_loss:                           0.00085127354
+      max_advantage:                            0.20980048
+      max_critic_clipfrac:                      0.18200684
+      max_critic_loss:                          0.014972748
+      max_entropy:                              0.3287996
+      max_grad_norm_pre_clip:                   3.2637618
+      max_loss:                                 0.007340861
+    std:
+      std_actor_approxkl1:                      0.0006209547
+      std_actor_approxkl3:                      0.00032578193
+      std_actor_clipfrac:                       0.016977409
+      std_actor_loss:                           0.0010878964
+      std_advantage:                            0.15732947
+      std_critic_clipfrac:                      0.053031433
+      std_critic_loss:                          0.001404099
+      std_entropy:                              0.011468099
+      std_grad_norm_pre_clip:                   0.5677004
+      std_loss:                                 0.0012827849
+  eval-batch-train:
+    avg_avg_episode_length:                     26.141613
+    avg_avg_return:                             0.8440584
+    avg_benchmark_regret:                       0.1431587
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.11582184
+    proxy_corner:
+      avg_avg_return:                           0.057266712
+      avg_reward_per_step:                      0.0043563843
+  eval-batch-orig:
+    avg_avg_episode_length:                     33.084526
+    avg_avg_return:                             0.79034936
+    avg_benchmark_regret:                       0.19416219
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.0692749
+    proxy_corner:
+      avg_avg_return:                           0.79034936
+      avg_reward_per_step:                      0.0692749
+  eval-batch-shift:
+    avg_avg_episode_length:                     24.612411
+    avg_avg_return:                             0.86858606
+    avg_benchmark_regret:                       0.11891988
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.12695312
+    proxy_corner:
+      avg_avg_return:                           0.05460803
+      avg_reward_per_step:                      0.0025405884
+  eval-batch-tree:
+    avg_avg_episode_length:                     30.252834
+    avg_avg_return:                             0.8260764
+    avg_benchmark_regret:                       0.15855886
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.08483124
+    proxy_corner:
+      avg_avg_return:                           0.8260764
+      avg_reward_per_step:                      0.08483124
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        84934656
+    env-step-replay-before:                     43483136
+    env-step-used-before:                       84934656
+    ppo-update-before:                          51840
+    env-step-generate-before:                   41451520
+    env-step-all-after:                         84967424
+    env-step-replay-after:                      43515904
+    env-step-used-after:                        84967424
+    ppo-update-after:                           51860
+    env-step-generate-after:                    41451520
+  train-replay:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     70.735794
+    avg_avg_return:                             0.3542714
+    avg_reward_per_step:                        0.009399414
+    proxy_corner:
+      avg_avg_return:                           0.07718991
+      avg_reward_per_step:                      0.0014648438
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     70.735794
+    avg_avg_return:                             0.3542714
+    avg_reward_per_step:                        0.009399414
+    proxy_corner:
+      avg_avg_return:                           0.07718991
+      avg_reward_per_step:                      0.0014648438
+  perf:
+    env_steps_per_second:                       325875.89300902927
+    ppo_updates_per_second:                     42.78597518808161
+    eval-batch-train-duration:                  0.519787821918726
+    eval-batch-orig-duration:                   0.47905975114554167
+    eval-batch-shift-duration:                  0.45798251684755087
+    eval-batch-tree-duration:                   0.4690639730542898
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.759033
+      cheese-corne_dist_wavg:                   15.275669
+      levels_cheese_is_in_corner_avg:           0.014892578
+      mouse-cheese_dist_finite_avg:             22.026367
+      mouse-cheese_dist_finite_wavg:            9.403418
+      mouse-cheese_dist_solvable_avg:           22.026367
+      mouse-cheese_dist_solvable_wavg:          9.403417
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.6013003
+    visit_patterns:
+      avg_first_visit_time:                     1115.5837
+      avg_last_visit_time:                      1318.2131
+      num_replay_batches:                       1328
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00085256
+    avg_actor_approxkl3:                        0.0008717576
+    avg_actor_clipfrac:                         0.044274904
+    avg_actor_loss:                             -0.0013483175
+    avg_advantage:                              -0.033116173
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.012105976
+    avg_entropy:                                0.31307256
+    avg_grad_norm_pre_clip:                     1.5278655
+    avg_loss:                                   0.004391599
+    max:
+      max_actor_approxkl1:                      0.0021390142
+      max_actor_approxkl3:                      0.001990921
+      max_actor_clipfrac:                       0.083862305
+      max_actor_loss:                           0.0006239861
+      max_advantage:                            0.26149428
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0144653125
+      max_entropy:                              0.33564192
+      max_grad_norm_pre_clip:                   2.5992975
+      max_loss:                                 0.0066964095
+    std:
+      std_actor_approxkl1:                      0.00065964065
+      std_actor_approxkl3:                      0.0004480934
+      std_actor_clipfrac:                       0.02094522
+      std_actor_loss:                           0.001052077
+      std_advantage:                            0.15448785
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0008981869
+      std_entropy:                              0.012364499
+      std_grad_norm_pre_clip:                   0.5231209
+      std_loss:                                 0.001237375
+  eval-batch-train:
+    avg_avg_episode_length:                     25.701225
+    avg_avg_return:                             0.8472912
+    avg_benchmark_regret:                       0.1399259
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.117362976
+    proxy_corner:
+      avg_avg_return:                           0.05290471
+      avg_reward_per_step:                      0.0042037964
+  eval-batch-orig:
+    avg_avg_episode_length:                     32.094215
+    avg_avg_return:                             0.8058585
+    avg_benchmark_regret:                       0.17865306
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07199097
+    proxy_corner:
+      avg_avg_return:                           0.8058585
+      avg_reward_per_step:                      0.07199097
+  eval-batch-shift:
+    avg_avg_episode_length:                     24.623268
+    avg_avg_return:                             0.8656882
+    avg_benchmark_regret:                       0.12181777
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.12929535
+    proxy_corner:
+      avg_avg_return:                           0.055018507
+      avg_reward_per_step:                      0.0026779175
+  eval-batch-tree:
+    avg_avg_episode_length:                     28.828651
+    avg_avg_return:                             0.8360812
+    avg_benchmark_regret:                       0.148554
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.087654114
+    proxy_corner:
+      avg_avg_return:                           0.8360812
+      avg_reward_per_step:                      0.087654114
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        85983232
+    env-step-replay-before:                     44072960
+    env-step-used-before:                       85983232
+    ppo-update-before:                          52480
+    env-step-generate-before:                   41910272
+    env-step-all-after:                         86016000
+    env-step-replay-after:                      44072960
+    env-step-used-after:                        86016000
+    ppo-update-after:                           52500
+    env-step-generate-after:                    41943040
+  train-generate:
+    shift_proportion:                           0.96484375
+    avg_avg_episode_length:                     28.838017
+    avg_avg_return:                             0.7610787
+    avg_reward_per_step:                        0.09536743
+    proxy_corner:
+      avg_avg_return:                           0.057511132
+      avg_reward_per_step:                      0.0036315918
+  train-all:
+    shift_proportion:                           0.96484375
+    avg_avg_episode_length:                     28.838017
+    avg_avg_return:                             0.7610787
+    avg_reward_per_step:                        0.09536743
+    proxy_corner:
+      avg_avg_return:                           0.057511132
+      avg_reward_per_step:                      0.0036315918
+  perf:
+    env_steps_per_second:                       349670.94257874304
+    ppo_updates_per_second:                     42.85351661099462
+    eval-batch-train-duration:                  0.5509523022919893
+    eval-batch-orig-duration:                   0.48574847634881735
+    eval-batch-shift-duration:                  0.451453136280179
+    eval-batch-tree-duration:                   0.49515704717487097
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.790527
+      cheese-corne_dist_wavg:                   2.6862319
+      levels_cheese_is_in_corner_avg:           0.015625
+      mouse-cheese_dist_finite_avg:             22.206787
+      mouse-cheese_dist_finite_wavg:            7.620672
+      mouse-cheese_dist_solvable_avg:           22.206787
+      mouse-cheese_dist_solvable_wavg:          7.620672
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.6021255
+    visit_patterns:
+      avg_first_visit_time:                     1130.1946
+      avg_last_visit_time:                      1335.2502
+      num_replay_batches:                       1345
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012492323
+    avg_actor_approxkl3:                        0.00096744223
+    avg_actor_clipfrac:                         0.04373169
+    avg_actor_loss:                             -0.00013338812
+    avg_advantage:                              0.08648565
+    avg_critic_clipfrac:                        0.26156616
+    avg_critic_loss:                            0.006597505
+    avg_entropy:                                0.1823596
+    avg_grad_norm_pre_clip:                     1.9775774
+    avg_loss:                                   0.0029830046
+    max:
+      max_actor_approxkl1:                      0.0030290915
+      max_actor_approxkl3:                      0.0017329985
+      max_actor_clipfrac:                       0.06762695
+      max_actor_loss:                           0.0018918216
+      max_advantage:                            0.3220427
+      max_critic_clipfrac:                      0.98535156
+      max_critic_loss:                          0.00979121
+      max_entropy:                              0.208804
+      max_grad_norm_pre_clip:                   3.4867225
+      max_loss:                                 0.006450004
+    std:
+      std_actor_approxkl1:                      0.00071971375
+      std_actor_approxkl3:                      0.00043291558
+      std_actor_clipfrac:                       0.016531037
+      std_actor_loss:                           0.0011366745
+      std_advantage:                            0.10751613
+      std_critic_clipfrac:                      0.38584352
+      std_critic_loss:                          0.0013500293
+      std_entropy:                              0.012854069
+      std_grad_norm_pre_clip:                   0.7097147
+      std_loss:                                 0.0015220568
+  eval-batch-train:
+    avg_avg_episode_length:                     25.314884
+    avg_avg_return:                             0.86007917
+    avg_benchmark_regret:                       0.12713791
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.114349365
+    proxy_corner:
+      avg_avg_return:                           0.049507886
+      avg_reward_per_step:                      0.0031204224
+  eval-batch-orig:
+    avg_avg_episode_length:                     30.734737
+    avg_avg_return:                             0.8299643
+    avg_benchmark_regret:                       0.15454726
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.06851959
+    proxy_corner:
+      avg_avg_return:                           0.8299643
+      avg_reward_per_step:                      0.06851959
+  eval-batch-shift:
+    avg_avg_episode_length:                     24.924288
+    avg_avg_return:                             0.85807776
+    avg_benchmark_regret:                       0.1294282
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.12738037
+    proxy_corner:
+      avg_avg_return:                           0.048991073
+      avg_reward_per_step:                      0.002357483
+  eval-batch-tree:
+    avg_avg_episode_length:                     29.165783
+    avg_avg_return:                             0.84530807
+    avg_benchmark_regret:                       0.13932714
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.08369446
+    proxy_corner:
+      avg_avg_return:                           0.84530807
+      avg_reward_per_step:                      0.08369446
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        87031808
+    env-step-replay-before:                     44531712
+    env-step-used-before:                       87031808
+    ppo-update-before:                          53120
+    env-step-generate-before:                   42500096
+    env-step-all-after:                         87064576
+    env-step-replay-after:                      44531712
+    env-step-used-after:                        87064576
+    ppo-update-after:                           53140
+    env-step-generate-after:                    42532864
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     26.414331
+    avg_avg_return:                             0.78423125
+    avg_reward_per_step:                        0.12643433
+    proxy_corner:
+      avg_avg_return:                           0.058796737
+      avg_reward_per_step:                      0.0030212402
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     26.414331
+    avg_avg_return:                             0.78423125
+    avg_reward_per_step:                        0.12643433
+    proxy_corner:
+      avg_avg_return:                           0.058796737
+      avg_reward_per_step:                      0.0030212402
+  perf:
+    env_steps_per_second:                       289608.43260668044
+    ppo_updates_per_second:                     42.85448534852459
+    eval-batch-train-duration:                  0.5453695636242628
+    eval-batch-orig-duration:                   0.48888319078832865
+    eval-batch-shift-duration:                  0.48577609937638044
+    eval-batch-tree-duration:                   0.4954402828589082
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.854248
+      cheese-corne_dist_wavg:                   2.7074265
+      levels_cheese_is_in_corner_avg:           0.015380859
+      mouse-cheese_dist_finite_avg:             22.325928
+      mouse-cheese_dist_finite_wavg:            7.6262774
+      mouse-cheese_dist_solvable_avg:           22.325928
+      mouse-cheese_dist_solvable_wavg:          7.6262774
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.420168
+    scoring:
+      avg_scores:                               0.5956582
+    visit_patterns:
+      avg_first_visit_time:                     1144.751
+      avg_last_visit_time:                      1349.333
+      num_replay_batches:                       1359
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0008628173
+    avg_actor_approxkl3:                        0.00067263603
+    avg_actor_clipfrac:                         0.02841797
+    avg_actor_loss:                             -0.0006606473
+    avg_advantage:                              0.0019180438
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0034872969
+    avg_entropy:                                0.15770273
+    avg_grad_norm_pre_clip:                     1.360335
+    avg_loss:                                   0.00092529843
+    max:
+      max_actor_approxkl1:                      0.0023582522
+      max_actor_approxkl3:                      0.0013533775
+      max_actor_clipfrac:                       0.053222656
+      max_actor_loss:                           0.0009768754
+      max_advantage:                            0.20918143
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0067147473
+      max_entropy:                              0.18777165
+      max_grad_norm_pre_clip:                   3.4670544
+      max_loss:                                 0.0024495164
+    std:
+      std_actor_approxkl1:                      0.00064872316
+      std_actor_approxkl3:                      0.00036809922
+      std_actor_clipfrac:                       0.014555706
+      std_actor_loss:                           0.0007951258
+      std_advantage:                            0.08330853
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0012248703
+      std_entropy:                              0.0141808735
+      std_grad_norm_pre_clip:                   0.65062755
+      std_loss:                                 0.00090328284
+  eval-batch-train:
+    avg_avg_episode_length:                     24.506096
+    avg_avg_return:                             0.8629159
+    avg_benchmark_regret:                       0.12430115
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.11714935
+    proxy_corner:
+      avg_avg_return:                           0.058938205
+      avg_reward_per_step:                      0.003967285
+  eval-batch-orig:
+    avg_avg_episode_length:                     30.90904
+    avg_avg_return:                             0.8107586
+    avg_benchmark_regret:                       0.1737529
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07405853
+    proxy_corner:
+      avg_avg_return:                           0.8107586
+      avg_reward_per_step:                      0.07405853
+  eval-batch-shift:
+    avg_avg_episode_length:                     24.138046
+    avg_avg_return:                             0.8618855
+    avg_benchmark_regret:                       0.1256205
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.1299057
+    proxy_corner:
+      avg_avg_return:                           0.060535267
+      avg_reward_per_step:                      0.0028381348
+  eval-batch-tree:
+    avg_avg_episode_length:                     28.609951
+    avg_avg_return:                             0.831571
+    avg_benchmark_regret:                       0.15306425
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.089393616
+    proxy_corner:
+      avg_avg_return:                           0.831571
+      avg_reward_per_step:                      0.089393616
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        88080384
+    env-step-replay-before:                     44859392
+    env-step-used-before:                       88080384
+    ppo-update-before:                          53760
+    env-step-generate-before:                   43220992
+    env-step-all-after:                         88113152
+    env-step-replay-after:                      44859392
+    env-step-used-after:                        88113152
+    ppo-update-after:                           53780
+    env-step-generate-after:                    43253760
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     28.035774
+    avg_avg_return:                             0.7649334
+    avg_reward_per_step:                        0.1328125
+    proxy_corner:
+      avg_avg_return:                           0.058945034
+      avg_reward_per_step:                      0.00289917
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     28.035774
+    avg_avg_return:                             0.7649334
+    avg_reward_per_step:                        0.1328125
+    proxy_corner:
+      avg_avg_return:                           0.058945034
+      avg_reward_per_step:                      0.00289917
+  perf:
+    env_steps_per_second:                       307026.8249009037
+    ppo_updates_per_second:                     42.12996208411869
+    eval-batch-train-duration:                  0.5328381145372987
+    eval-batch-orig-duration:                   0.4919784404337406
+    eval-batch-shift-duration:                  0.500958583317697
+    eval-batch-tree-duration:                   0.562530186958611
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.82251
+      cheese-corne_dist_wavg:                   7.1808333
+      levels_cheese_is_in_corner_avg:           0.015380859
+      mouse-cheese_dist_finite_avg:             22.373535
+      mouse-cheese_dist_finite_wavg:            7.6342607
+      mouse-cheese_dist_solvable_avg:           22.373535
+      mouse-cheese_dist_solvable_wavg:          7.6342607
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999985
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.6104669
+    visit_patterns:
+      avg_first_visit_time:                     1158.7473
+      avg_last_visit_time:                      1359.3447
+      num_replay_batches:                       1369
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.000961083
+    avg_actor_approxkl3:                        0.0008388138
+    avg_actor_clipfrac:                         0.036071777
+    avg_actor_loss:                             -0.0003024366
+    avg_advantage:                              -0.0012826733
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0045108567
+    avg_entropy:                                0.15839806
+    avg_grad_norm_pre_clip:                     1.8553854
+    avg_loss:                                   0.0017945936
+    max:
+      max_actor_approxkl1:                      0.002606606
+      max_actor_approxkl3:                      0.002213284
+      max_actor_clipfrac:                       0.07495117
+      max_actor_loss:                           0.0024855211
+      max_advantage:                            0.2006411
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0063457037
+      max_entropy:                              0.18268381
+      max_grad_norm_pre_clip:                   3.3202653
+      max_loss:                                 0.0053096833
+    std:
+      std_actor_approxkl1:                      0.00068056956
+      std_actor_approxkl3:                      0.0005864114
+      std_actor_clipfrac:                       0.021405978
+      std_actor_loss:                           0.0011566622
+      std_advantage:                            0.09526194
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00084300427
+      std_entropy:                              0.01346428
+      std_grad_norm_pre_clip:                   0.71643037
+      std_loss:                                 0.0011754215
+  eval-batch-train:
+    avg_avg_episode_length:                     24.776665
+    avg_avg_return:                             0.86412275
+    avg_benchmark_regret:                       0.12309433
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.11672974
+    proxy_corner:
+      avg_avg_return:                           0.05826214
+      avg_reward_per_step:                      0.0035552979
+  eval-batch-orig:
+    avg_avg_episode_length:                     29.963648
+    avg_avg_return:                             0.82963634
+    avg_benchmark_regret:                       0.15487522
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07298279
+    proxy_corner:
+      avg_avg_return:                           0.82963634
+      avg_reward_per_step:                      0.07298279
+  eval-batch-shift:
+    avg_avg_episode_length:                     24.55423
+    avg_avg_return:                             0.8627264
+    avg_benchmark_regret:                       0.12477958
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13158417
+    proxy_corner:
+      avg_avg_return:                           0.05345916
+      avg_reward_per_step:                      0.0026626587
+  eval-batch-tree:
+    avg_avg_episode_length:                     28.726759
+    avg_avg_return:                             0.8345911
+    avg_benchmark_regret:                       0.15004413
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.088005066
+    proxy_corner:
+      avg_avg_return:                           0.8345911
+      avg_reward_per_step:                      0.088005066
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        89128960
+    env-step-replay-before:                     45514752
+    env-step-used-before:                       89128960
+    ppo-update-before:                          54400
+    env-step-generate-before:                   43614208
+    env-step-all-after:                         89161728
+    env-step-replay-after:                      45547520
+    env-step-used-after:                        89161728
+    ppo-update-after:                           54420
+    env-step-generate-after:                    43614208
+  train-replay:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     69.403625
+    avg_avg_return:                             0.37748808
+    avg_reward_per_step:                        0.010437012
+    proxy_corner:
+      avg_avg_return:                           0.1015806
+      avg_reward_per_step:                      0.0015869141
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     69.403625
+    avg_avg_return:                             0.37748808
+    avg_reward_per_step:                        0.010437012
+    proxy_corner:
+      avg_avg_return:                           0.1015806
+      avg_reward_per_step:                      0.0015869141
+  perf:
+    env_steps_per_second:                       291645.4201210218
+    ppo_updates_per_second:                     41.93108996694606
+    eval-batch-train-duration:                  0.5407420285046101
+    eval-batch-orig-duration:                   0.48845128528773785
+    eval-batch-shift-duration:                  0.48148326575756073
+    eval-batch-tree-duration:                   0.5017856052145362
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.909912
+      cheese-corne_dist_wavg:                   25.176756
+      levels_cheese_is_in_corner_avg:           0.015380859
+      mouse-cheese_dist_finite_avg:             22.48462
+      mouse-cheese_dist_finite_wavg:            10.353115
+      mouse-cheese_dist_solvable_avg:           22.48462
+      mouse-cheese_dist_solvable_wavg:          10.353115
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.99999
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.60097647
+    visit_patterns:
+      avg_first_visit_time:                     1169.3875
+      avg_last_visit_time:                      1380.2888
+      num_replay_batches:                       1390
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014508611
+    avg_actor_approxkl3:                        0.0010948176
+    avg_actor_clipfrac:                         0.05414429
+    avg_actor_loss:                             -0.0009560425
+    avg_advantage:                              -0.02086673
+    avg_critic_clipfrac:                        6.1035156e-05
+    avg_critic_loss:                            0.0115926685
+    avg_entropy:                                0.30880454
+    avg_grad_norm_pre_clip:                     1.6343485
+    avg_loss:                                   0.004531488
+    max:
+      max_actor_approxkl1:                      0.004487749
+      max_actor_approxkl3:                      0.002213715
+      max_actor_clipfrac:                       0.09448242
+      max_actor_loss:                           0.001187563
+      max_advantage:                            0.24865478
+      max_critic_clipfrac:                      0.0008544922
+      max_critic_loss:                          0.013492096
+      max_entropy:                              0.33169198
+      max_grad_norm_pre_clip:                   2.701962
+      max_loss:                                 0.006832025
+    std:
+      std_actor_approxkl1:                      0.0010219975
+      std_actor_approxkl3:                      0.0005477024
+      std_actor_clipfrac:                       0.024462368
+      std_actor_loss:                           0.0009880399
+      std_advantage:                            0.15004508
+      std_critic_clipfrac:                      0.00019871608
+      std_critic_loss:                          0.00091531815
+      std_entropy:                              0.009296511
+      std_grad_norm_pre_clip:                   0.61957866
+      std_loss:                                 0.0010503413
+  eval-batch-train:
+    avg_avg_episode_length:                     25.302696
+    avg_avg_return:                             0.8590154
+    avg_benchmark_regret:                       0.1282017
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.11571503
+    proxy_corner:
+      avg_avg_return:                           0.066516206
+      avg_reward_per_step:                      0.004020691
+  eval-batch-orig:
+    avg_avg_episode_length:                     31.514206
+    avg_avg_return:                             0.820492
+    avg_benchmark_regret:                       0.16401951
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07098389
+    proxy_corner:
+      avg_avg_return:                           0.820492
+      avg_reward_per_step:                      0.07098389
+  eval-batch-shift:
+    avg_avg_episode_length:                     25.012613
+    avg_avg_return:                             0.85802716
+    avg_benchmark_regret:                       0.1294788
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.1277771
+    proxy_corner:
+      avg_avg_return:                           0.057113186
+      avg_reward_per_step:                      0.0026016235
+  eval-batch-tree:
+    avg_avg_episode_length:                     28.676678
+    avg_avg_return:                             0.8387656
+    avg_benchmark_regret:                       0.14586961
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.08608246
+    proxy_corner:
+      avg_avg_return:                           0.8387656
+      avg_reward_per_step:                      0.08608246
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        90177536
+    env-step-replay-before:                     46202880
+    env-step-used-before:                       90177536
+    ppo-update-before:                          55040
+    env-step-generate-before:                   43974656
+    env-step-all-after:                         90210304
+    env-step-replay-after:                      46202880
+    env-step-used-after:                        90210304
+    ppo-update-after:                           55060
+    env-step-generate-after:                    44007424
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     30.94722
+    avg_avg_return:                             0.7480368
+    avg_reward_per_step:                        0.093170166
+    proxy_corner:
+      avg_avg_return:                           0.07998632
+      avg_reward_per_step:                      0.0037231445
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     30.94722
+    avg_avg_return:                             0.7480368
+    avg_reward_per_step:                        0.093170166
+    proxy_corner:
+      avg_avg_return:                           0.07998632
+      avg_reward_per_step:                      0.0037231445
+  perf:
+    env_steps_per_second:                       391116.87731105456
+    ppo_updates_per_second:                     44.820802894941245
+    eval-batch-train-duration:                  0.49141277838498354
+    eval-batch-orig-duration:                   0.42319265473634005
+    eval-batch-shift-duration:                  0.46135714929550886
+    eval-batch-tree-duration:                   0.43019469175487757
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    17.959473
+      cheese-corne_dist_wavg:                   17.988369
+      levels_cheese_is_in_corner_avg:           0.01586914
+      mouse-cheese_dist_finite_avg:             22.53003
+      mouse-cheese_dist_finite_wavg:            10.357601
+      mouse-cheese_dist_solvable_avg:           22.53003
+      mouse-cheese_dist_solvable_wavg:          10.357604
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.5994424
+    visit_patterns:
+      avg_first_visit_time:                     1182.6216
+      avg_last_visit_time:                      1400.1807
+      num_replay_batches:                       1410
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001112555
+    avg_actor_approxkl3:                        0.0009273632
+    avg_actor_clipfrac:                         0.04227295
+    avg_actor_loss:                             -0.00015403442
+    avg_advantage:                              0.08162555
+    avg_critic_clipfrac:                        0.29151002
+    avg_critic_loss:                            0.0067299823
+    avg_entropy:                                0.18476003
+    avg_grad_norm_pre_clip:                     2.1085234
+    avg_loss:                                   0.0030261967
+    max:
+      max_actor_approxkl1:                      0.002112181
+      max_actor_approxkl3:                      0.0019217541
+      max_actor_clipfrac:                       0.068359375
+      max_actor_loss:                           0.002865851
+      max_advantage:                            0.305525
+      max_critic_clipfrac:                      0.991333
+      max_critic_loss:                          0.010396149
+      max_entropy:                              0.2057292
+      max_grad_norm_pre_clip:                   3.2147202
+      max_loss:                                 0.0068204575
+    std:
+      std_actor_approxkl1:                      0.00060064415
+      std_actor_approxkl3:                      0.00042155202
+      std_actor_clipfrac:                       0.015952684
+      std_actor_loss:                           0.001194881
+      std_advantage:                            0.10646205
+      std_critic_clipfrac:                      0.40364197
+      std_critic_loss:                          0.0013956097
+      std_entropy:                              0.013904946
+      std_grad_norm_pre_clip:                   0.5266291
+      std_loss:                                 0.0015920034
+  eval-batch-train:
+    avg_avg_episode_length:                     25.41763
+    avg_avg_return:                             0.85485196
+    avg_benchmark_regret:                       0.13236517
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.11689758
+    proxy_corner:
+      avg_avg_return:                           0.06517014
+      avg_reward_per_step:                      0.004890442
+  eval-batch-orig:
+    avg_avg_episode_length:                     31.78395
+    avg_avg_return:                             0.8236931
+    avg_benchmark_regret:                       0.16081843
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.0737381
+    proxy_corner:
+      avg_avg_return:                           0.8236931
+      avg_reward_per_step:                      0.0737381
+  eval-batch-shift:
+    avg_avg_episode_length:                     24.250498
+    avg_avg_return:                             0.8715615
+    avg_benchmark_regret:                       0.11594443
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.12687683
+    proxy_corner:
+      avg_avg_return:                           0.05836281
+      avg_reward_per_step:                      0.0028915405
+  eval-batch-tree:
+    avg_avg_episode_length:                     28.790627
+    avg_avg_return:                             0.8381041
+    avg_benchmark_regret:                       0.14653105
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.08943939
+    proxy_corner:
+      avg_avg_return:                           0.8381041
+      avg_reward_per_step:                      0.08943939
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        91226112
+    env-step-replay-before:                     46759936
+    env-step-used-before:                       91226112
+    ppo-update-before:                          55680
+    env-step-generate-before:                   44466176
+    env-step-all-after:                         91258880
+    env-step-replay-after:                      46759936
+    env-step-used-after:                        91258880
+    ppo-update-after:                           55700
+    env-step-generate-after:                    44498944
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     27.809395
+    avg_avg_return:                             0.7721565
+    avg_reward_per_step:                        0.10702515
+    proxy_corner:
+      avg_avg_return:                           0.03837363
+      avg_reward_per_step:                      0.001159668
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     27.809395
+    avg_avg_return:                             0.7721565
+    avg_reward_per_step:                        0.10702515
+    proxy_corner:
+      avg_avg_return:                           0.03837363
+      avg_reward_per_step:                      0.001159668
+  perf:
+    env_steps_per_second:                       321623.9446936253
+    ppo_updates_per_second:                     41.981678149067264
+    eval-batch-train-duration:                  0.5439041061326861
+    eval-batch-orig-duration:                   0.4856266640126705
+    eval-batch-shift-duration:                  0.4792711017653346
+    eval-batch-tree-duration:                   0.49405426252633333
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.054443
+      cheese-corne_dist_wavg:                   18.000631
+      levels_cheese_is_in_corner_avg:           0.0146484375
+      mouse-cheese_dist_finite_avg:             22.496338
+      mouse-cheese_dist_finite_wavg:            10.358271
+      mouse-cheese_dist_solvable_avg:           22.496338
+      mouse-cheese_dist_solvable_wavg:          10.3582735
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016816
+    scoring:
+      avg_scores:                               0.59240705
+    visit_patterns:
+      avg_first_visit_time:                     1195.9263
+      avg_last_visit_time:                      1417.2327
+      num_replay_batches:                       1427
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012228732
+    avg_actor_approxkl3:                        0.00077667006
+    avg_actor_clipfrac:                         0.034838866
+    avg_actor_loss:                             -0.0012111915
+    avg_advantage:                              0.07060057
+    avg_critic_clipfrac:                        0.19923095
+    avg_critic_loss:                            0.0059145754
+    avg_entropy:                                0.1682875
+    avg_grad_norm_pre_clip:                     1.3710213
+    avg_loss:                                   0.0015778088
+    max:
+      max_actor_approxkl1:                      0.0023648432
+      max_actor_approxkl3:                      0.0013799232
+      max_actor_clipfrac:                       0.049438477
+      max_actor_loss:                           0.0005030483
+      max_advantage:                            0.29417652
+      max_critic_clipfrac:                      0.9449463
+      max_critic_loss:                          0.007818397
+      max_entropy:                              0.19140208
+      max_grad_norm_pre_clip:                   2.0744467
+      max_loss:                                 0.0042394013
+    std:
+      std_actor_approxkl1:                      0.00066360773
+      std_actor_approxkl3:                      0.00035495535
+      std_actor_clipfrac:                       0.013080755
+      std_actor_loss:                           0.0009370598
+      std_advantage:                            0.1008926
+      std_critic_clipfrac:                      0.33874607
+      std_critic_loss:                          0.0013580103
+      std_entropy:                              0.010632795
+      std_grad_norm_pre_clip:                   0.4127397
+      std_loss:                                 0.001250678
+  eval-batch-train:
+    avg_avg_episode_length:                     25.511105
+    avg_avg_return:                             0.8539623
+    avg_benchmark_regret:                       0.13325477
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.11582184
+    proxy_corner:
+      avg_avg_return:                           0.07025577
+      avg_reward_per_step:                      0.0045318604
+  eval-batch-orig:
+    avg_avg_episode_length:                     30.236126
+    avg_avg_return:                             0.8300981
+    avg_benchmark_regret:                       0.15441345
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07284546
+    proxy_corner:
+      avg_avg_return:                           0.8300981
+      avg_reward_per_step:                      0.07284546
+  eval-batch-shift:
+    avg_avg_episode_length:                     24.528168
+    avg_avg_return:                             0.8719151
+    avg_benchmark_regret:                       0.11559086
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.12892151
+    proxy_corner:
+      avg_avg_return:                           0.05914853
+      avg_reward_per_step:                      0.0029907227
+  eval-batch-tree:
+    avg_avg_episode_length:                     28.719994
+    avg_avg_return:                             0.83841205
+    avg_benchmark_regret:                       0.14622313
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.088661194
+    proxy_corner:
+      avg_avg_return:                           0.83841205
+      avg_reward_per_step:                      0.088661194
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        92274688
+    env-step-replay-before:                     47251456
+    env-step-used-before:                       92274688
+    ppo-update-before:                          56320
+    env-step-generate-before:                   45023232
+    env-step-all-after:                         92307456
+    env-step-replay-after:                      47251456
+    env-step-used-after:                        92307456
+    ppo-update-after:                           56340
+    env-step-generate-after:                    45056000
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     30.761942
+    avg_avg_return:                             0.73746586
+    avg_reward_per_step:                        0.119018555
+    proxy_corner:
+      avg_avg_return:                           0.03809523
+      avg_reward_per_step:                      0.0017089844
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     30.761942
+    avg_avg_return:                             0.73746586
+    avg_reward_per_step:                        0.119018555
+    proxy_corner:
+      avg_avg_return:                           0.03809523
+      avg_reward_per_step:                      0.0017089844
+  perf:
+    env_steps_per_second:                       260553.6967344482
+    ppo_updates_per_second:                     41.62108045728734
+    eval-batch-train-duration:                  0.5495446976274252
+    eval-batch-orig-duration:                   0.48757010884582996
+    eval-batch-shift-duration:                  0.48236002027988434
+    eval-batch-tree-duration:                   0.486452454701066
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.128906
+      cheese-corne_dist_wavg:                   18.01403
+      levels_cheese_is_in_corner_avg:           0.013671875
+      mouse-cheese_dist_finite_avg:             22.545166
+      mouse-cheese_dist_finite_wavg:            10.345929
+      mouse-cheese_dist_solvable_avg:           22.545166
+      mouse-cheese_dist_solvable_wavg:          10.345929
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.597218
+    visit_patterns:
+      avg_first_visit_time:                     1213.4707
+      avg_last_visit_time:                      1432.2473
+      num_replay_batches:                       1442
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00057935726
+    avg_actor_approxkl3:                        0.00072215527
+    avg_actor_clipfrac:                         0.031341553
+    avg_actor_loss:                             -0.00089151453
+    avg_advantage:                              -0.005450668
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.004908682
+    avg_entropy:                                0.16968672
+    avg_grad_norm_pre_clip:                     1.4574126
+    avg_loss:                                   0.0013931397
+    max:
+      max_actor_approxkl1:                      0.001652326
+      max_actor_approxkl3:                      0.0012743351
+      max_actor_clipfrac:                       0.049560547
+      max_actor_loss:                           0.00029665232
+      max_advantage:                            0.1867463
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0062008468
+      max_entropy:                              0.19476265
+      max_grad_norm_pre_clip:                   2.5696297
+      max_loss:                                 0.0028968041
+    std:
+      std_actor_approxkl1:                      0.00047944507
+      std_actor_approxkl3:                      0.00036206355
+      std_actor_clipfrac:                       0.01516032
+      std_actor_loss:                           0.000923305
+      std_advantage:                            0.099570274
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.000924275
+      std_entropy:                              0.013311282
+      std_grad_norm_pre_clip:                   0.573395
+      std_loss:                                 0.00079197524
+  eval-batch-train:
+    avg_avg_episode_length:                     24.826921
+    avg_avg_return:                             0.86473155
+    avg_benchmark_regret:                       0.12248552
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.1186142
+    proxy_corner:
+      avg_avg_return:                           0.054555245
+      avg_reward_per_step:                      0.003921509
+  eval-batch-orig:
+    avg_avg_episode_length:                     30.058708
+    avg_avg_return:                             0.82871956
+    avg_benchmark_regret:                       0.15579197
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.071517944
+    proxy_corner:
+      avg_avg_return:                           0.82871956
+      avg_reward_per_step:                      0.071517944
+  eval-batch-shift:
+    avg_avg_episode_length:                     24.927124
+    avg_avg_return:                             0.8551029
+    avg_benchmark_regret:                       0.13240308
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.1311264
+    proxy_corner:
+      avg_avg_return:                           0.047669955
+      avg_reward_per_step:                      0.0025634766
+  eval-batch-tree:
+    avg_avg_episode_length:                     28.36578
+    avg_avg_return:                             0.845546
+    avg_benchmark_regret:                       0.1390892
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.08880615
+    proxy_corner:
+      avg_avg_return:                           0.845546
+      avg_reward_per_step:                      0.08880615
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        93323264
+    env-step-replay-before:                     47742976
+    env-step-used-before:                       93323264
+    ppo-update-before:                          56960
+    env-step-generate-before:                   45580288
+    env-step-all-after:                         93356032
+    env-step-replay-after:                      47742976
+    env-step-used-after:                        93356032
+    ppo-update-after:                           56980
+    env-step-generate-after:                    45613056
+  train-generate:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     28.977215
+    avg_avg_return:                             0.7561581
+    avg_reward_per_step:                        0.11898804
+    proxy_corner:
+      avg_avg_return:                           0.056116514
+      avg_reward_per_step:                      0.006439209
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     28.977215
+    avg_avg_return:                             0.7561581
+    avg_reward_per_step:                        0.11898804
+    proxy_corner:
+      avg_avg_return:                           0.056116514
+      avg_reward_per_step:                      0.006439209
+  perf:
+    env_steps_per_second:                       504728.6261483732
+    ppo_updates_per_second:                     42.629228876022026
+    eval-batch-train-duration:                  0.3162010358646512
+    eval-batch-orig-duration:                   0.2343692211434245
+    eval-batch-shift-duration:                  0.26489832904189825
+    eval-batch-tree-duration:                   0.2678924147039652
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.158447
+      cheese-corne_dist_wavg:                   16.218914
+      levels_cheese_is_in_corner_avg:           0.013183594
+      mouse-cheese_dist_finite_avg:             22.594482
+      mouse-cheese_dist_finite_wavg:            8.571071
+      mouse-cheese_dist_solvable_avg:           22.594482
+      mouse-cheese_dist_solvable_wavg:          8.571072
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.420168
+    scoring:
+      avg_scores:                               0.60661894
+    visit_patterns:
+      avg_first_visit_time:                     1229.794
+      avg_last_visit_time:                      1447.4011
+      num_replay_batches:                       1457
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012026315
+    avg_actor_approxkl3:                        0.0009479456
+    avg_actor_clipfrac:                         0.039556887
+    avg_actor_loss:                             -0.0007296046
+    avg_advantage:                              0.004284122
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00457449
+    avg_entropy:                                0.17209221
+    avg_grad_norm_pre_clip:                     1.6321086
+    avg_loss:                                   0.0013855478
+    max:
+      max_actor_approxkl1:                      0.0027689273
+      max_actor_approxkl3:                      0.001763976
+      max_actor_clipfrac:                       0.06530762
+      max_actor_loss:                           0.00100407
+      max_advantage:                            0.21197593
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0063253613
+      max_entropy:                              0.18936503
+      max_grad_norm_pre_clip:                   2.64247
+      max_loss:                                 0.0031793166
+    std:
+      std_actor_approxkl1:                      0.0007051863
+      std_actor_approxkl3:                      0.00046910788
+      std_actor_clipfrac:                       0.016409697
+      std_actor_loss:                           0.0009181931
+      std_advantage:                            0.0958317
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00087904127
+      std_entropy:                              0.010719479
+      std_grad_norm_pre_clip:                   0.5496805
+      std_loss:                                 0.0008698823
+  eval-batch-train:
+    avg_avg_episode_length:                     24.280472
+    avg_avg_return:                             0.8711582
+    avg_benchmark_regret:                       0.116058916
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12004852
+    proxy_corner:
+      avg_avg_return:                           0.055534706
+      avg_reward_per_step:                      0.0041046143
+  eval-batch-orig:
+    avg_avg_episode_length:                     30.708328
+    avg_avg_return:                             0.8243557
+    avg_benchmark_regret:                       0.16015582
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07158661
+    proxy_corner:
+      avg_avg_return:                           0.8243557
+      avg_reward_per_step:                      0.07158661
+  eval-batch-shift:
+    avg_avg_episode_length:                     25.513622
+    avg_avg_return:                             0.8508847
+    avg_benchmark_regret:                       0.1366213
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.1295166
+    proxy_corner:
+      avg_avg_return:                           0.046231806
+      avg_reward_per_step:                      0.0025787354
+  eval-batch-tree:
+    avg_avg_episode_length:                     28.934591
+    avg_avg_return:                             0.8387749
+    avg_benchmark_regret:                       0.14586025
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.08678436
+    proxy_corner:
+      avg_avg_return:                           0.8387749
+      avg_reward_per_step:                      0.08678436
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        94371840
+    env-step-replay-before:                     48267264
+    env-step-used-before:                       94371840
+    ppo-update-before:                          57600
+    env-step-generate-before:                   46104576
+    env-step-all-after:                         94404608
+    env-step-replay-after:                      48300032
+    env-step-used-after:                        94404608
+    ppo-update-after:                           57620
+    env-step-generate-after:                    46104576
+  train-replay:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     72.559555
+    avg_avg_return:                             0.3165313
+    avg_reward_per_step:                        0.008575439
+    proxy_corner:
+      avg_avg_return:                           0.06463632
+      avg_reward_per_step:                      0.0011901855
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     72.559555
+    avg_avg_return:                             0.3165313
+    avg_reward_per_step:                        0.008575439
+    proxy_corner:
+      avg_avg_return:                           0.06463632
+      avg_reward_per_step:                      0.0011901855
+  perf:
+    env_steps_per_second:                       296484.1175197896
+    ppo_updates_per_second:                     40.99838648786138
+    eval-batch-train-duration:                  0.507904184050858
+    eval-batch-orig-duration:                   0.3459058376029134
+    eval-batch-shift-duration:                  0.2413222575560212
+    eval-batch-tree-duration:                   0.2286380473524332
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.22876
+      cheese-corne_dist_wavg:                   10.827165
+      levels_cheese_is_in_corner_avg:           0.014160156
+      mouse-cheese_dist_finite_avg:             22.634277
+      mouse-cheese_dist_finite_wavg:            4.9656334
+      mouse-cheese_dist_solvable_avg:           22.634277
+      mouse-cheese_dist_solvable_wavg:          4.965634
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.605474
+    visit_patterns:
+      avg_first_visit_time:                     1241.4731
+      avg_last_visit_time:                      1464.2534
+      num_replay_batches:                       1474
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011688792
+    avg_actor_approxkl3:                        0.0011450669
+    avg_actor_clipfrac:                         0.05529785
+    avg_actor_loss:                             -0.00047683643
+    avg_advantage:                              -0.04680807
+    avg_critic_clipfrac:                        0.034228515
+    avg_critic_loss:                            0.013240351
+    avg_entropy:                                0.30800405
+    avg_grad_norm_pre_clip:                     2.2370083
+    avg_loss:                                   0.0058353352
+    max:
+      max_actor_approxkl1:                      0.0036158867
+      max_actor_approxkl3:                      0.0024504312
+      max_actor_clipfrac:                       0.103515625
+      max_actor_loss:                           0.00057807565
+      max_advantage:                            0.24610096
+      max_critic_clipfrac:                      0.26489258
+      max_critic_loss:                          0.015475484
+      max_entropy:                              0.33103618
+      max_grad_norm_pre_clip:                   3.3980484
+      max_loss:                                 0.007374576
+    std:
+      std_actor_approxkl1:                      0.00084214524
+      std_actor_approxkl3:                      0.0006135939
+      std_actor_clipfrac:                       0.025262805
+      std_actor_loss:                           0.0008210945
+      std_advantage:                            0.16071983
+      std_critic_clipfrac:                      0.07390655
+      std_critic_loss:                          0.00096522295
+      std_entropy:                              0.011702245
+      std_grad_norm_pre_clip:                   0.73392385
+      std_loss:                                 0.0008885114
+  eval-batch-train:
+    avg_avg_episode_length:                     24.617756
+    avg_avg_return:                             0.86185646
+    avg_benchmark_regret:                       0.12536058
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.11971283
+    proxy_corner:
+      avg_avg_return:                           0.055445634
+      avg_reward_per_step:                      0.004714966
+  eval-batch-orig:
+    avg_avg_episode_length:                     30.696396
+    avg_avg_return:                             0.82257545
+    avg_benchmark_regret:                       0.16193607
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07392883
+    proxy_corner:
+      avg_avg_return:                           0.82257545
+      avg_reward_per_step:                      0.07392883
+  eval-batch-shift:
+    avg_avg_episode_length:                     25.066074
+    avg_avg_return:                             0.84998226
+    avg_benchmark_regret:                       0.13752368
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.12928772
+    proxy_corner:
+      avg_avg_return:                           0.050103366
+      avg_reward_per_step:                      0.0024108887
+  eval-batch-tree:
+    avg_avg_episode_length:                     28.365105
+    avg_avg_return:                             0.8407291
+    avg_benchmark_regret:                       0.14390609
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.088638306
+    proxy_corner:
+      avg_avg_return:                           0.8407291
+      avg_reward_per_step:                      0.088638306
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        95420416
+    env-step-replay-before:                     48889856
+    env-step-used-before:                       95420416
+    ppo-update-before:                          58240
+    env-step-generate-before:                   46530560
+    env-step-all-after:                         95453184
+    env-step-replay-after:                      48889856
+    env-step-used-after:                        95453184
+    ppo-update-after:                           58260
+    env-step-generate-after:                    46563328
+  train-generate:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     25.727797
+    avg_avg_return:                             0.7989841
+    avg_reward_per_step:                        0.1210022
+    proxy_corner:
+      avg_avg_return:                           0.06555895
+      avg_reward_per_step:                      0.0028381348
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     25.727797
+    avg_avg_return:                             0.7989841
+    avg_reward_per_step:                        0.1210022
+    proxy_corner:
+      avg_avg_return:                           0.06555895
+      avg_reward_per_step:                      0.0028381348
+  perf:
+    env_steps_per_second:                       321124.90303066
+    ppo_updates_per_second:                     42.2161132796827
+    eval-batch-train-duration:                  0.5169495772570372
+    eval-batch-orig-duration:                   0.4802894489839673
+    eval-batch-shift-duration:                  0.44495554454624653
+    eval-batch-tree-duration:                   0.49403806682676077
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.240723
+      cheese-corne_dist_wavg:                   11.708168
+      levels_cheese_is_in_corner_avg:           0.013916016
+      mouse-cheese_dist_finite_avg:             22.688965
+      mouse-cheese_dist_finite_wavg:            9.4743805
+      mouse-cheese_dist_solvable_avg:           22.688965
+      mouse-cheese_dist_solvable_wavg:          9.474381
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.59636176
+    visit_patterns:
+      avg_first_visit_time:                     1253.2307
+      avg_last_visit_time:                      1482.189
+      num_replay_batches:                       1492
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012589286
+    avg_actor_approxkl3:                        0.0008955965
+    avg_actor_clipfrac:                         0.03649292
+    avg_actor_loss:                             -3.276449e-05
+    avg_advantage:                              0.08177763
+    avg_critic_clipfrac:                        0.3077576
+    avg_critic_loss:                            0.0045882887
+    avg_entropy:                                0.1537025
+    avg_grad_norm_pre_clip:                     2.0714617
+    avg_loss:                                   0.0021076777
+    max:
+      max_actor_approxkl1:                      0.0031151704
+      max_actor_approxkl3:                      0.0017978349
+      max_actor_clipfrac:                       0.06213379
+      max_actor_loss:                           0.0016305
+      max_advantage:                            0.2614087
+      max_critic_clipfrac:                      0.9963379
+      max_critic_loss:                          0.006967743
+      max_entropy:                              0.18003853
+      max_grad_norm_pre_clip:                   3.8461912
+      max_loss:                                 0.004746615
+    std:
+      std_actor_approxkl1:                      0.00089684053
+      std_actor_approxkl3:                      0.0004604326
+      std_actor_clipfrac:                       0.01561367
+      std_actor_loss:                           0.00088029285
+      std_advantage:                            0.086290956
+      std_critic_clipfrac:                      0.4175398
+      std_critic_loss:                          0.0012060326
+      std_entropy:                              0.014584006
+      std_grad_norm_pre_clip:                   0.6601765
+      std_loss:                                 0.0011922388
+  eval-batch-train:
+    avg_avg_episode_length:                     24.21949
+    avg_avg_return:                             0.8691858
+    avg_benchmark_regret:                       0.11803123
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.11967468
+    proxy_corner:
+      avg_avg_return:                           0.058474742
+      avg_reward_per_step:                      0.004676819
+  eval-batch-orig:
+    avg_avg_episode_length:                     29.059704
+    avg_avg_return:                             0.83490574
+    avg_benchmark_regret:                       0.14960575
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.077034
+    proxy_corner:
+      avg_avg_return:                           0.83490574
+      avg_reward_per_step:                      0.077034
+  eval-batch-shift:
+    avg_avg_episode_length:                     23.796608
+    avg_avg_return:                             0.87169
+    avg_benchmark_regret:                       0.115815975
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13031769
+    proxy_corner:
+      avg_avg_return:                           0.061152577
+      avg_reward_per_step:                      0.0030822754
+  eval-batch-tree:
+    avg_avg_episode_length:                     27.161516
+    avg_avg_return:                             0.84924644
+    avg_benchmark_regret:                       0.13538873
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09135437
+    proxy_corner:
+      avg_avg_return:                           0.84924644
+      avg_reward_per_step:                      0.09135437
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        96468992
+    env-step-replay-before:                     49381376
+    env-step-used-before:                       96468992
+    ppo-update-before:                          58880
+    env-step-generate-before:                   47087616
+    env-step-all-after:                         96501760
+    env-step-replay-after:                      49414144
+    env-step-used-after:                        96501760
+    ppo-update-after:                           58900
+    env-step-generate-after:                    47087616
+  train-replay:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     69.17738
+    avg_avg_return:                             0.38047194
+    avg_reward_per_step:                        0.010101318
+    proxy_corner:
+      avg_avg_return:                           0.085874386
+      avg_reward_per_step:                      0.001739502
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     69.17738
+    avg_avg_return:                             0.38047194
+    avg_reward_per_step:                        0.010101318
+    proxy_corner:
+      avg_avg_return:                           0.085874386
+      avg_reward_per_step:                      0.001739502
+  perf:
+    env_steps_per_second:                       241951.06112120204
+    ppo_updates_per_second:                     44.50919042503129
+    eval-batch-train-duration:                  0.678570007905364
+    eval-batch-orig-duration:                   0.6226400593295693
+    eval-batch-shift-duration:                  0.6467194091528654
+    eval-batch-tree-duration:                   0.606563663110137
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.117432
+      cheese-corne_dist_wavg:                   8.131723
+      levels_cheese_is_in_corner_avg:           0.015380859
+      mouse-cheese_dist_finite_avg:             22.759277
+      mouse-cheese_dist_finite_wavg:            9.485868
+      mouse-cheese_dist_solvable_avg:           22.759277
+      mouse-cheese_dist_solvable_wavg:          9.485869
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5985954
+    visit_patterns:
+      avg_first_visit_time:                     1269.5195
+      avg_last_visit_time:                      1498.2908
+      num_replay_batches:                       1508
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0016811084
+    avg_actor_approxkl3:                        0.0016314101
+    avg_actor_clipfrac:                         0.06727295
+    avg_actor_loss:                             0.00039722808
+    avg_advantage:                              -0.034266487
+    avg_critic_clipfrac:                        0.0040405276
+    avg_critic_loss:                            0.011940582
+    avg_entropy:                                0.27060467
+    avg_grad_norm_pre_clip:                     2.3226125
+    avg_loss:                                   0.006096914
+    max:
+      max_actor_approxkl1:                      0.0042020883
+      max_actor_approxkl3:                      0.004024361
+      max_actor_clipfrac:                       0.1217041
+      max_actor_loss:                           0.0036896616
+      max_advantage:                            0.2129122
+      max_critic_clipfrac:                      0.026733398
+      max_critic_loss:                          0.015444023
+      max_entropy:                              0.29589966
+      max_grad_norm_pre_clip:                   3.683975
+      max_loss:                                 0.009852223
+    std:
+      std_actor_approxkl1:                      0.0011591548
+      std_actor_approxkl3:                      0.0010661664
+      std_actor_clipfrac:                       0.031828668
+      std_actor_loss:                           0.0011292826
+      std_advantage:                            0.15204135
+      std_critic_clipfrac:                      0.008536886
+      std_critic_loss:                          0.0013201025
+      std_entropy:                              0.010973069
+      std_grad_norm_pre_clip:                   0.7042069
+      std_loss:                                 0.0013449185
+  eval-batch-train:
+    avg_avg_episode_length:                     24.475088
+    avg_avg_return:                             0.86951864
+    avg_benchmark_regret:                       0.11769839
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.11982727
+    proxy_corner:
+      avg_avg_return:                           0.052303795
+      avg_reward_per_step:                      0.003929138
+  eval-batch-orig:
+    avg_avg_episode_length:                     29.282894
+    avg_avg_return:                             0.83714855
+    avg_benchmark_regret:                       0.14736298
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07522583
+    proxy_corner:
+      avg_avg_return:                           0.83714855
+      avg_reward_per_step:                      0.07522583
+  eval-batch-shift:
+    avg_avg_episode_length:                     23.739971
+    avg_avg_return:                             0.8717761
+    avg_benchmark_regret:                       0.115729794
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.1309967
+    proxy_corner:
+      avg_avg_return:                           0.047297016
+      avg_reward_per_step:                      0.0025405884
+  eval-batch-tree:
+    avg_avg_episode_length:                     27.718052
+    avg_avg_return:                             0.84930634
+    avg_benchmark_regret:                       0.13532886
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.08885956
+    proxy_corner:
+      avg_avg_return:                           0.84930634
+      avg_reward_per_step:                      0.08885956
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        97517568
+    env-step-replay-before:                     49840128
+    env-step-used-before:                       97517568
+    ppo-update-before:                          59520
+    env-step-generate-before:                   47677440
+    env-step-all-after:                         97550336
+    env-step-replay-after:                      49872896
+    env-step-used-after:                        97550336
+    ppo-update-after:                           59540
+    env-step-generate-after:                    47677440
+  train-replay:
+    shift_proportion:                           0.96484375
+    avg_avg_episode_length:                     69.98184
+    avg_avg_return:                             0.37110022
+    avg_reward_per_step:                        0.009796143
+    proxy_corner:
+      avg_avg_return:                           0.08613674
+      avg_reward_per_step:                      0.001739502
+  train-all:
+    shift_proportion:                           0.96484375
+    avg_avg_episode_length:                     69.98184
+    avg_avg_return:                             0.37110022
+    avg_reward_per_step:                        0.009796143
+    proxy_corner:
+      avg_avg_return:                           0.08613674
+      avg_reward_per_step:                      0.001739502
+  perf:
+    env_steps_per_second:                       355351.1863450438
+    ppo_updates_per_second:                     42.93784272557225
+    eval-batch-train-duration:                  0.5142272720113397
+    eval-batch-orig-duration:                   0.47239356208592653
+    eval-batch-shift-duration:                  0.44524292182177305
+    eval-batch-tree-duration:                   0.46775623597204685
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.250244
+      cheese-corne_dist_wavg:                   16.218168
+      levels_cheese_is_in_corner_avg:           0.017089844
+      mouse-cheese_dist_finite_avg:             22.801514
+      mouse-cheese_dist_finite_wavg:            8.570911
+      mouse-cheese_dist_solvable_avg:           22.801514
+      mouse-cheese_dist_solvable_wavg:          8.570911
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.6076716
+    visit_patterns:
+      avg_first_visit_time:                     1285.3738
+      avg_last_visit_time:                      1512.2974
+      num_replay_batches:                       1522
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0015295314
+    avg_actor_approxkl3:                        0.001260193
+    avg_actor_clipfrac:                         0.058917236
+    avg_actor_loss:                             -0.00015777424
+    avg_advantage:                              -0.039102025
+    avg_critic_clipfrac:                        0.000378418
+    avg_critic_loss:                            0.011715843
+    avg_entropy:                                0.28565195
+    avg_grad_norm_pre_clip:                     2.457506
+    avg_loss:                                   0.005414496
+    max:
+      max_actor_approxkl1:                      0.0035123525
+      max_actor_approxkl3:                      0.0028768745
+      max_actor_clipfrac:                       0.110839844
+      max_actor_loss:                           0.001868397
+      max_advantage:                            0.20206904
+      max_critic_clipfrac:                      0.00390625
+      max_critic_loss:                          0.013050849
+      max_entropy:                              0.3018916
+      max_grad_norm_pre_clip:                   4.2979317
+      max_loss:                                 0.007850034
+    std:
+      std_actor_approxkl1:                      0.00089415774
+      std_actor_approxkl3:                      0.0006727992
+      std_actor_clipfrac:                       0.026306346
+      std_actor_loss:                           0.0009834323
+      std_advantage:                            0.15205422
+      std_critic_clipfrac:                      0.0010371667
+      std_critic_loss:                          0.0009897237
+      std_entropy:                              0.010803305
+      std_grad_norm_pre_clip:                   0.83930355
+      std_loss:                                 0.00102533
+  eval-batch-train:
+    avg_avg_episode_length:                     24.242712
+    avg_avg_return:                             0.8582325
+    avg_benchmark_regret:                       0.12898459
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12007141
+    proxy_corner:
+      avg_avg_return:                           0.05647047
+      avg_reward_per_step:                      0.0045776367
+  eval-batch-orig:
+    avg_avg_episode_length:                     28.29432
+    avg_avg_return:                             0.84009683
+    avg_benchmark_regret:                       0.14441466
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07952118
+    proxy_corner:
+      avg_avg_return:                           0.84009683
+      avg_reward_per_step:                      0.07952118
+  eval-batch-shift:
+    avg_avg_episode_length:                     23.15001
+    avg_avg_return:                             0.8742778
+    avg_benchmark_regret:                       0.11322821
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13315582
+    proxy_corner:
+      avg_avg_return:                           0.055666022
+      avg_reward_per_step:                      0.0028533936
+  eval-batch-tree:
+    avg_avg_episode_length:                     26.236992
+    avg_avg_return:                             0.8596058
+    avg_benchmark_regret:                       0.12502939
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.094062805
+    proxy_corner:
+      avg_avg_return:                           0.8596058
+      avg_reward_per_step:                      0.094062805
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        98566144
+    env-step-replay-before:                     50397184
+    env-step-used-before:                       98566144
+    ppo-update-before:                          60160
+    env-step-generate-before:                   48168960
+    env-step-all-after:                         98598912
+    env-step-replay-after:                      50429952
+    env-step-used-after:                        98598912
+    ppo-update-after:                           60180
+    env-step-generate-after:                    48168960
+  train-replay:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     69.30197
+    avg_avg_return:                             0.37312216
+    avg_reward_per_step:                        0.010925293
+    proxy_corner:
+      avg_avg_return:                           0.12483769
+      avg_reward_per_step:                      0.0023498535
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     69.30197
+    avg_avg_return:                             0.37312216
+    avg_reward_per_step:                        0.010925293
+    proxy_corner:
+      avg_avg_return:                           0.12483769
+      avg_reward_per_step:                      0.0023498535
+  perf:
+    env_steps_per_second:                       248223.4936623887
+    ppo_updates_per_second:                     41.83051265295475
+    eval-batch-train-duration:                  0.5388652142137289
+    eval-batch-orig-duration:                   0.4810372469946742
+    eval-batch-shift-duration:                  0.4792332760989666
+    eval-batch-tree-duration:                   0.4506521811708808
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.163818
+      cheese-corne_dist_wavg:                   18.016785
+      levels_cheese_is_in_corner_avg:           0.018554688
+      mouse-cheese_dist_finite_avg:             22.82837
+      mouse-cheese_dist_finite_wavg:            10.378851
+      mouse-cheese_dist_solvable_avg:           22.82837
+      mouse-cheese_dist_solvable_wavg:          10.378852
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.6087328
+    visit_patterns:
+      avg_first_visit_time:                     1298.1936
+      avg_last_visit_time:                      1529.2234
+      num_replay_batches:                       1539
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012465926
+    avg_actor_approxkl3:                        0.0008414359
+    avg_actor_clipfrac:                         0.04350586
+    avg_actor_loss:                             -0.0010428958
+    avg_advantage:                              -0.01071373
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.011504257
+    avg_entropy:                                0.28955927
+    avg_grad_norm_pre_clip:                     1.6563548
+    avg_loss:                                   0.0044196732
+    max:
+      max_actor_approxkl1:                      0.003199321
+      max_actor_approxkl3:                      0.001482829
+      max_actor_clipfrac:                       0.06994629
+      max_actor_loss:                           0.0008652806
+      max_advantage:                            0.30603725
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.013525202
+      max_entropy:                              0.2995723
+      max_grad_norm_pre_clip:                   2.321591
+      max_loss:                                 0.006613322
+    std:
+      std_actor_approxkl1:                      0.0007311378
+      std_actor_approxkl3:                      0.00034414502
+      std_actor_clipfrac:                       0.016532278
+      std_actor_loss:                           0.001088754
+      std_advantage:                            0.15171939
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0009032572
+      std_entropy:                              0.0064194156
+      std_grad_norm_pre_clip:                   0.40267292
+      std_loss:                                 0.0011139852
+  eval-batch-train:
+    avg_avg_episode_length:                     24.853249
+    avg_avg_return:                             0.8526664
+    avg_benchmark_regret:                       0.13455066
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.118751526
+    proxy_corner:
+      avg_avg_return:                           0.06717286
+      avg_reward_per_step:                      0.004722595
+  eval-batch-orig:
+    avg_avg_episode_length:                     30.064663
+    avg_avg_return:                             0.8214397
+    avg_benchmark_regret:                       0.16307187
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07814026
+    proxy_corner:
+      avg_avg_return:                           0.8214397
+      avg_reward_per_step:                      0.07814026
+  eval-batch-shift:
+    avg_avg_episode_length:                     24.355886
+    avg_avg_return:                             0.86797404
+    avg_benchmark_regret:                       0.11953192
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.12952423
+    proxy_corner:
+      avg_avg_return:                           0.064948365
+      avg_reward_per_step:                      0.003074646
+  eval-batch-tree:
+    avg_avg_episode_length:                     27.730698
+    avg_avg_return:                             0.8402905
+    avg_benchmark_regret:                       0.1443447
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09223175
+    proxy_corner:
+      avg_avg_return:                           0.8402905
+      avg_reward_per_step:                      0.09223175
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        99614720
+    env-step-replay-before:                     51052544
+    env-step-used-before:                       99614720
+    ppo-update-before:                          60800
+    env-step-generate-before:                   48562176
+    env-step-all-after:                         99647488
+    env-step-replay-after:                      51085312
+    env-step-used-after:                        99647488
+    ppo-update-after:                           60820
+    env-step-generate-after:                    48562176
+  train-replay:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     70.387405
+    avg_avg_return:                             0.41070583
+    avg_reward_per_step:                        0.010284424
+    proxy_corner:
+      avg_avg_return:                           0.12829989
+      avg_reward_per_step:                      0.002380371
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     70.387405
+    avg_avg_return:                             0.41070583
+    avg_reward_per_step:                        0.010284424
+    proxy_corner:
+      avg_avg_return:                           0.12829989
+      avg_reward_per_step:                      0.002380371
+  perf:
+    env_steps_per_second:                       279670.2267025933
+    ppo_updates_per_second:                     41.29100661760957
+    eval-batch-train-duration:                  0.5082352263852954
+    eval-batch-orig-duration:                   0.43810446932911873
+    eval-batch-shift-duration:                  0.4200755525380373
+    eval-batch-tree-duration:                   0.46040851902216673
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.192383
+      cheese-corne_dist_wavg:                   10.836154
+      levels_cheese_is_in_corner_avg:           0.018066406
+      mouse-cheese_dist_finite_avg:             22.851562
+      mouse-cheese_dist_finite_wavg:            4.9783463
+      mouse-cheese_dist_solvable_avg:           22.851562
+      mouse-cheese_dist_solvable_wavg:          4.978347
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.99999
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.57985497
+    visit_patterns:
+      avg_first_visit_time:                     1311.5981
+      avg_last_visit_time:                      1549.2292
+      num_replay_batches:                       1559
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010641958
+    avg_actor_approxkl3:                        0.00085570494
+    avg_actor_clipfrac:                         0.04232178
+    avg_actor_loss:                             -0.0011746064
+    avg_advantage:                              -0.011456719
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.009449675
+    avg_entropy:                                0.28241435
+    avg_grad_norm_pre_clip:                     1.5083045
+    avg_loss:                                   0.003267816
+    max:
+      max_actor_approxkl1:                      0.0031861856
+      max_actor_approxkl3:                      0.0017194743
+      max_actor_clipfrac:                       0.08532715
+      max_actor_loss:                           0.0008816421
+      max_advantage:                            0.25160027
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.011451039
+      max_entropy:                              0.30413702
+      max_grad_norm_pre_clip:                   2.9770865
+      max_loss:                                 0.0056390404
+    std:
+      std_actor_approxkl1:                      0.00095102703
+      std_actor_approxkl3:                      0.00041101233
+      std_actor_clipfrac:                       0.019311704
+      std_actor_loss:                           0.0010573033
+      std_advantage:                            0.13738932
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011151369
+      std_entropy:                              0.009524937
+      std_grad_norm_pre_clip:                   0.6171974
+      std_loss:                                 0.0012150464
+  eval-batch-train:
+    avg_avg_episode_length:                     23.155525
+    avg_avg_return:                             0.88637173
+    avg_benchmark_regret:                       0.10084531
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.11885071
+    proxy_corner:
+      avg_avg_return:                           0.0483331
+      avg_reward_per_step:                      0.004257202
+  eval-batch-orig:
+    avg_avg_episode_length:                     28.957298
+    avg_avg_return:                             0.8529295
+    avg_benchmark_regret:                       0.13158208
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07270813
+    proxy_corner:
+      avg_avg_return:                           0.8529295
+      avg_reward_per_step:                      0.07270813
+  eval-batch-shift:
+    avg_avg_episode_length:                     23.28968
+    avg_avg_return:                             0.88183874
+    avg_benchmark_regret:                       0.1056672
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13148499
+    proxy_corner:
+      avg_avg_return:                           0.052100364
+      avg_reward_per_step:                      0.0027008057
+  eval-batch-tree:
+    avg_avg_episode_length:                     26.859106
+    avg_avg_return:                             0.86505973
+    avg_benchmark_regret:                       0.119575456
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.08731079
+    proxy_corner:
+      avg_avg_return:                           0.86505973
+      avg_reward_per_step:                      0.08731079
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        100663296
+    env-step-replay-before:                     51675136
+    env-step-used-before:                       100663296
+    ppo-update-before:                          61440
+    env-step-generate-before:                   48988160
+    env-step-all-after:                         100696064
+    env-step-replay-after:                      51675136
+    env-step-used-after:                        100696064
+    ppo-update-after:                           61460
+    env-step-generate-after:                    49020928
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     24.89521
+    avg_avg_return:                             0.80023664
+    avg_reward_per_step:                        0.12365723
+    proxy_corner:
+      avg_avg_return:                           0.057219036
+      avg_reward_per_step:                      0.0018920898
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     24.89521
+    avg_avg_return:                             0.80023664
+    avg_reward_per_step:                        0.12365723
+    proxy_corner:
+      avg_avg_return:                           0.057219036
+      avg_reward_per_step:                      0.0018920898
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  perf:
+    env_steps_per_second:                       298585.0971157786
+    ppo_updates_per_second:                     42.07473103192762
+    eval-batch-train-duration:                  0.6471068300306797
+    eval-batch-orig-duration:                   0.5850354079157114
+    eval-batch-shift-duration:                  0.5880181230604649
+    eval-batch-tree-duration:                   0.5876176627352834
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.175049
+      cheese-corne_dist_wavg:                   33.288235
+      levels_cheese_is_in_corner_avg:           0.017333984
+      mouse-cheese_dist_finite_avg:             22.819336
+      mouse-cheese_dist_finite_wavg:            8.59334
+      mouse-cheese_dist_solvable_avg:           22.819336
+      mouse-cheese_dist_solvable_wavg:          8.593341
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.5683304
+    visit_patterns:
+      avg_first_visit_time:                     1326.565
+      avg_last_visit_time:                      1567.2268
+      num_replay_batches:                       1577
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012883218
+    avg_actor_approxkl3:                        0.00072888803
+    avg_actor_clipfrac:                         0.032043457
+    avg_actor_loss:                             -0.00088603835
+    avg_advantage:                              0.08538468
+    avg_critic_clipfrac:                        0.303125
+    avg_critic_loss:                            0.0052251546
+    avg_entropy:                                0.13731308
+    avg_grad_norm_pre_clip:                     1.4071554
+    avg_loss:                                   0.001589226
+    max:
+      max_actor_approxkl1:                      0.0027848522
+      max_actor_approxkl3:                      0.0016422702
+      max_actor_clipfrac:                       0.048706055
+      max_actor_loss:                           0.0011239015
+      max_advantage:                            0.27815628
+      max_critic_clipfrac:                      0.99780273
+      max_critic_loss:                          0.008456227
+      max_entropy:                              0.1571963
+      max_grad_norm_pre_clip:                   2.3198636
+      max_loss:                                 0.004825299
+    std:
+      std_actor_approxkl1:                      0.00086829055
+      std_actor_approxkl3:                      0.00035120963
+      std_actor_clipfrac:                       0.0123368045
+      std_actor_loss:                           0.0012816229
+      std_advantage:                            0.09365527
+      std_critic_clipfrac:                      0.39757437
+      std_critic_loss:                          0.0014031248
+      std_entropy:                              0.0108523015
+      std_grad_norm_pre_clip:                   0.37831894
+      std_loss:                                 0.0015812103
+  eval-batch-train:
+    avg_avg_episode_length:                     22.835518
+    avg_avg_return:                             0.8894491
+    avg_benchmark_regret:                       0.097767934
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.118370056
+    proxy_corner:
+      avg_avg_return:                           0.058064364
+      avg_reward_per_step:                      0.0040512085
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-orig:
+    avg_avg_episode_length:                     29.0259
+    avg_avg_return:                             0.85243213
+    avg_benchmark_regret:                       0.13207944
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07392883
+    proxy_corner:
+      avg_avg_return:                           0.85243213
+      avg_reward_per_step:                      0.07392883
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-shift:
+    avg_avg_episode_length:                     22.885704
+    avg_avg_return:                             0.882468
+    avg_benchmark_regret:                       0.10503791
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13308716
+    proxy_corner:
+      avg_avg_return:                           0.053982083
+      avg_reward_per_step:                      0.0027923584
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-tree:
+    avg_avg_episode_length:                     26.7961
+    avg_avg_return:                             0.8663781
+    avg_benchmark_regret:                       0.11825709
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.089668274
+    proxy_corner:
+      avg_avg_return:                           0.8663781
+      avg_reward_per_step:                      0.089668274
+    rollouts_gif:                               float32(132, 225, 225, 3)
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        101711872
+    env-step-replay-before:                     52101120
+    env-step-used-before:                       101711872
+    ppo-update-before:                          62080
+    env-step-generate-before:                   49610752
+    env-step-all-after:                         101744640
+    env-step-replay-after:                      52133888
+    env-step-used-after:                        101744640
+    ppo-update-after:                           62100
+    env-step-generate-after:                    49610752
+  train-replay:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     71.67357
+    avg_avg_return:                             0.3314464
+    avg_reward_per_step:                        0.009552002
+    proxy_corner:
+      avg_avg_return:                           0.09446382
+      avg_reward_per_step:                      0.0018615723
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     71.67357
+    avg_avg_return:                             0.3314464
+    avg_reward_per_step:                        0.009552002
+    proxy_corner:
+      avg_avg_return:                           0.09446382
+      avg_reward_per_step:                      0.0018615723
+  perf:
+    env_steps_per_second:                       294777.1863506134
+    ppo_updates_per_second:                     42.50825928900672
+    eval-batch-train-duration:                  0.5412572510540485
+    eval-batch-orig-duration:                   0.4843119075521827
+    eval-batch-shift-duration:                  0.4633075287565589
+    eval-batch-tree-duration:                   0.4688066737726331
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.128174
+      cheese-corne_dist_wavg:                   18.006844
+      levels_cheese_is_in_corner_avg:           0.016113281
+      mouse-cheese_dist_finite_avg:             22.808105
+      mouse-cheese_dist_finite_wavg:            10.379997
+      mouse-cheese_dist_solvable_avg:           22.808105
+      mouse-cheese_dist_solvable_wavg:          10.379998
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5836639
+    visit_patterns:
+      avg_first_visit_time:                     1343.94
+      avg_last_visit_time:                      1581.1531
+      num_replay_batches:                       1591
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0005288497
+    avg_actor_approxkl3:                        0.00076616526
+    avg_actor_clipfrac:                         0.039941408
+    avg_actor_loss:                             -0.0014341563
+    avg_advantage:                              -0.060727846
+    avg_critic_clipfrac:                        0.06271362
+    avg_critic_loss:                            0.013949891
+    avg_entropy:                                0.26979786
+    avg_grad_norm_pre_clip:                     1.5604692
+    avg_loss:                                   0.005270991
+    max:
+      max_actor_approxkl1:                      0.0015449785
+      max_actor_approxkl3:                      0.0014098315
+      max_actor_clipfrac:                       0.06274414
+      max_actor_loss:                           0.00039845705
+      max_advantage:                            0.18052274
+      max_critic_clipfrac:                      0.46166992
+      max_critic_loss:                          0.015538563
+      max_entropy:                              0.28971794
+      max_grad_norm_pre_clip:                   2.124368
+      max_loss:                                 0.007886022
+    std:
+      std_actor_approxkl1:                      0.0005393288
+      std_actor_approxkl3:                      0.00029466522
+      std_actor_clipfrac:                       0.014334901
+      std_actor_loss:                           0.0011806599
+      std_advantage:                            0.16468272
+      std_critic_clipfrac:                      0.13769764
+      std_critic_loss:                          0.0008411212
+      std_entropy:                              0.0091626495
+      std_grad_norm_pre_clip:                   0.35469928
+      std_loss:                                 0.0013205982
+  eval-batch-train:
+    avg_avg_episode_length:                     23.479132
+    avg_avg_return:                             0.8752128
+    avg_benchmark_regret:                       0.112004265
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.11790466
+    proxy_corner:
+      avg_avg_return:                           0.060277782
+      avg_reward_per_step:                      0.0040130615
+  eval-batch-orig:
+    avg_avg_episode_length:                     28.78667
+    avg_avg_return:                             0.84390336
+    avg_benchmark_regret:                       0.14060816
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.076049805
+    proxy_corner:
+      avg_avg_return:                           0.84390336
+      avg_reward_per_step:                      0.076049805
+  eval-batch-shift:
+    avg_avg_episode_length:                     23.567009
+    avg_avg_return:                             0.87370026
+    avg_benchmark_regret:                       0.11380569
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.1330719
+    proxy_corner:
+      avg_avg_return:                           0.051439643
+      avg_reward_per_step:                      0.0028762817
+  eval-batch-tree:
+    avg_avg_episode_length:                     26.46334
+    avg_avg_return:                             0.86753565
+    avg_benchmark_regret:                       0.11709956
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09082794
+    proxy_corner:
+      avg_avg_return:                           0.86753565
+      avg_reward_per_step:                      0.09082794
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        102760448
+    env-step-replay-before:                     52625408
+    env-step-used-before:                       102760448
+    ppo-update-before:                          62720
+    env-step-generate-before:                   50135040
+    env-step-all-after:                         102793216
+    env-step-replay-after:                      52625408
+    env-step-used-after:                        102793216
+    ppo-update-after:                           62740
+    env-step-generate-after:                    50167808
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     30.153934
+    avg_avg_return:                             0.7398088
+    avg_reward_per_step:                        0.09124756
+    proxy_corner:
+      avg_avg_return:                           0.056426838
+      avg_reward_per_step:                      0.0020446777
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     30.153934
+    avg_avg_return:                             0.7398088
+    avg_reward_per_step:                        0.09124756
+    proxy_corner:
+      avg_avg_return:                           0.056426838
+      avg_reward_per_step:                      0.0020446777
+  perf:
+    env_steps_per_second:                       319039.31091313367
+    ppo_updates_per_second:                     41.16300807192896
+    eval-batch-train-duration:                  0.5423034448176622
+    eval-batch-orig-duration:                   0.4823923474177718
+    eval-batch-shift-duration:                  0.475427164696157
+    eval-batch-tree-duration:                   0.45077543798834085
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.128174
+      cheese-corne_dist_wavg:                   12.604744
+      levels_cheese_is_in_corner_avg:           0.016357422
+      mouse-cheese_dist_finite_avg:             22.84082
+      mouse-cheese_dist_finite_wavg:            10.385567
+      mouse-cheese_dist_solvable_avg:           22.84082
+      mouse-cheese_dist_solvable_wavg:          10.385566
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000008
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.6054609
+    visit_patterns:
+      avg_first_visit_time:                     1360.6335
+      avg_last_visit_time:                      1596.407
+      num_replay_batches:                       1606
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012799787
+    avg_actor_approxkl3:                        0.0008776804
+    avg_actor_clipfrac:                         0.03551636
+    avg_actor_loss:                             -0.0006474938
+    avg_advantage:                              0.049750946
+    avg_critic_clipfrac:                        0.003985596
+    avg_critic_loss:                            0.006150467
+    avg_entropy:                                0.1460818
+    avg_grad_norm_pre_clip:                     1.6759465
+    avg_loss:                                   0.0022816577
+    max:
+      max_actor_approxkl1:                      0.0036037918
+      max_actor_approxkl3:                      0.0031882934
+      max_actor_clipfrac:                       0.08276367
+      max_actor_loss:                           0.0024225786
+      max_advantage:                            0.2565583
+      max_critic_clipfrac:                      0.026489258
+      max_critic_loss:                          0.009468904
+      max_entropy:                              0.17245021
+      max_grad_norm_pre_clip:                   3.3847897
+      max_loss:                                 0.004363106
+    std:
+      std_actor_approxkl1:                      0.00087716343
+      std_actor_approxkl3:                      0.0006415475
+      std_actor_clipfrac:                       0.016417595
+      std_actor_loss:                           0.0013373032
+      std_advantage:                            0.109190024
+      std_critic_clipfrac:                      0.008489223
+      std_critic_loss:                          0.0013518579
+      std_entropy:                              0.013022266
+      std_grad_norm_pre_clip:                   0.73129183
+      std_loss:                                 0.0012695799
+  eval-batch-train:
+    avg_avg_episode_length:                     23.821001
+    avg_avg_return:                             0.86486393
+    avg_benchmark_regret:                       0.12235314
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.120277405
+    proxy_corner:
+      avg_avg_return:                           0.05221787
+      avg_reward_per_step:                      0.0038528442
+  eval-batch-orig:
+    avg_avg_episode_length:                     28.517214
+    avg_avg_return:                             0.8440335
+    avg_benchmark_regret:                       0.14047804
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07556915
+    proxy_corner:
+      avg_avg_return:                           0.8440335
+      avg_reward_per_step:                      0.07556915
+  eval-batch-shift:
+    avg_avg_episode_length:                     23.040846
+    avg_avg_return:                             0.8738455
+    avg_benchmark_regret:                       0.11366042
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13598633
+    proxy_corner:
+      avg_avg_return:                           0.04528682
+      avg_reward_per_step:                      0.002418518
+  eval-batch-tree:
+    avg_avg_episode_length:                     26.856562
+    avg_avg_return:                             0.85658145
+    avg_benchmark_regret:                       0.12805372
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.090293884
+    proxy_corner:
+      avg_avg_return:                           0.85658145
+      avg_reward_per_step:                      0.090293884
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        103809024
+    env-step-replay-before:                     53149696
+    env-step-used-before:                       103809024
+    ppo-update-before:                          63360
+    env-step-generate-before:                   50659328
+    env-step-all-after:                         103841792
+    env-step-replay-after:                      53182464
+    env-step-used-after:                        103841792
+    ppo-update-after:                           63380
+    env-step-generate-after:                    50659328
+  train-replay:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     69.939575
+    avg_avg_return:                             0.38828912
+    avg_reward_per_step:                        0.010101318
+    proxy_corner:
+      avg_avg_return:                           0.08457084
+      avg_reward_per_step:                      0.0015258789
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     69.939575
+    avg_avg_return:                             0.38828912
+    avg_reward_per_step:                        0.010101318
+    proxy_corner:
+      avg_avg_return:                           0.08457084
+      avg_reward_per_step:                      0.0015258789
+  perf:
+    env_steps_per_second:                       349874.5916987801
+    ppo_updates_per_second:                     41.34029415116684
+    eval-batch-train-duration:                  0.5359162017703056
+    eval-batch-orig-duration:                   0.495167744345963
+    eval-batch-shift-duration:                  0.48778417240828276
+    eval-batch-tree-duration:                   0.4883475052192807
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.128662
+      cheese-corne_dist_wavg:                   33.28904
+      levels_cheese_is_in_corner_avg:           0.015380859
+      mouse-cheese_dist_finite_avg:             22.961914
+      mouse-cheese_dist_finite_wavg:            8.595106
+      mouse-cheese_dist_solvable_avg:           22.961914
+      mouse-cheese_dist_solvable_wavg:          8.595107
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.6015774
+    visit_patterns:
+      avg_first_visit_time:                     1372.8567
+      avg_last_visit_time:                      1613.1667
+      num_replay_batches:                       1623
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009430746
+    avg_actor_approxkl3:                        0.0009959565
+    avg_actor_clipfrac:                         0.047265626
+    avg_actor_loss:                             -0.0007234562
+    avg_advantage:                              -0.041900076
+    avg_critic_clipfrac:                        0.0012145996
+    avg_critic_loss:                            0.011245239
+    avg_entropy:                                0.25728297
+    avg_grad_norm_pre_clip:                     1.8562908
+    avg_loss:                                   0.00464188
+    max:
+      max_actor_approxkl1:                      0.0019291536
+      max_actor_approxkl3:                      0.0020934842
+      max_actor_clipfrac:                       0.09338379
+      max_actor_loss:                           0.0012000054
+      max_advantage:                            0.19628233
+      max_critic_clipfrac:                      0.013671875
+      max_critic_loss:                          0.014397232
+      max_entropy:                              0.27352515
+      max_grad_norm_pre_clip:                   2.9081914
+      max_loss:                                 0.006387998
+    std:
+      std_actor_approxkl1:                      0.0005476293
+      std_actor_approxkl3:                      0.00043751986
+      std_actor_clipfrac:                       0.01831639
+      std_actor_loss:                           0.00094237
+      std_advantage:                            0.14823015
+      std_critic_clipfrac:                      0.0034945977
+      std_critic_loss:                          0.0015428012
+      std_entropy:                              0.010030087
+      std_grad_norm_pre_clip:                   0.50585574
+      std_loss:                                 0.0011193502
+  eval-batch-train:
+    avg_avg_episode_length:                     23.231892
+    avg_avg_return:                             0.8782113
+    avg_benchmark_regret:                       0.10900575
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.11936188
+    proxy_corner:
+      avg_avg_return:                           0.062573105
+      avg_reward_per_step:                      0.003829956
+  eval-batch-orig:
+    avg_avg_episode_length:                     27.553917
+    avg_avg_return:                             0.85464466
+    avg_benchmark_regret:                       0.1298669
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07602692
+    proxy_corner:
+      avg_avg_return:                           0.85464466
+      avg_reward_per_step:                      0.07602692
+  eval-batch-shift:
+    avg_avg_episode_length:                     22.335289
+    avg_avg_return:                             0.8843366
+    avg_benchmark_regret:                       0.10316937
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13619995
+    proxy_corner:
+      avg_avg_return:                           0.05504082
+      avg_reward_per_step:                      0.002922058
+  eval-batch-tree:
+    avg_avg_episode_length:                     26.435078
+    avg_avg_return:                             0.8626722
+    avg_benchmark_regret:                       0.121962994
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.093055725
+    proxy_corner:
+      avg_avg_return:                           0.8626722
+      avg_reward_per_step:                      0.093055725
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        104857600
+    env-step-replay-before:                     53772288
+    env-step-used-before:                       104857600
+    ppo-update-before:                          64000
+    env-step-generate-before:                   51085312
+    env-step-all-after:                         104890368
+    env-step-replay-after:                      53772288
+    env-step-used-after:                        104890368
+    ppo-update-after:                           64020
+    env-step-generate-after:                    51118080
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     23.19395
+    avg_avg_return:                             0.81394446
+    avg_reward_per_step:                        0.14224243
+    proxy_corner:
+      avg_avg_return:                           0.040577304
+      avg_reward_per_step:                      0.001739502
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     23.19395
+    avg_avg_return:                             0.81394446
+    avg_reward_per_step:                        0.14224243
+    proxy_corner:
+      avg_avg_return:                           0.040577304
+      avg_reward_per_step:                      0.001739502
+  perf:
+    env_steps_per_second:                       347758.9271421618
+    ppo_updates_per_second:                     42.68366273956356
+    eval-batch-train-duration:                  0.5215924428775907
+    eval-batch-orig-duration:                   0.4838133091107011
+    eval-batch-shift-duration:                  0.47714824695140123
+    eval-batch-tree-duration:                   0.44021548330783844
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.156494
+      cheese-corne_dist_wavg:                   18.0154
+      levels_cheese_is_in_corner_avg:           0.01586914
+      mouse-cheese_dist_finite_avg:             22.994629
+      mouse-cheese_dist_finite_wavg:            10.397683
+      mouse-cheese_dist_solvable_avg:           22.994629
+      mouse-cheese_dist_solvable_wavg:          10.397684
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999985
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.59285027
+    visit_patterns:
+      avg_first_visit_time:                     1385.7837
+      avg_last_visit_time:                      1631.3401
+      num_replay_batches:                       1641
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001129913
+    avg_actor_approxkl3:                        0.0007113916
+    avg_actor_clipfrac:                         0.029638672
+    avg_actor_loss:                             -0.0005475396
+    avg_advantage:                              0.07126009
+    avg_critic_clipfrac:                        0.20204468
+    avg_critic_loss:                            0.0040856297
+    avg_entropy:                                0.1299352
+    avg_grad_norm_pre_clip:                     1.4238435
+    avg_loss:                                   0.0013653397
+    max:
+      max_actor_approxkl1:                      0.0027493336
+      max_actor_approxkl3:                      0.0015568897
+      max_actor_clipfrac:                       0.0513916
+      max_actor_loss:                           0.0011096671
+      max_advantage:                            0.23366034
+      max_critic_clipfrac:                      0.91748047
+      max_critic_loss:                          0.005993363
+      max_entropy:                              0.14833544
+      max_grad_norm_pre_clip:                   2.4360209
+      max_loss:                                 0.003947993
+    std:
+      std_actor_approxkl1:                      0.00072482886
+      std_actor_approxkl3:                      0.00040869755
+      std_actor_clipfrac:                       0.01399043
+      std_actor_loss:                           0.0007654476
+      std_advantage:                            0.08222396
+      std_critic_clipfrac:                      0.34088185
+      std_critic_loss:                          0.0010758773
+      std_entropy:                              0.011842715
+      std_grad_norm_pre_clip:                   0.4362445
+      std_loss:                                 0.0011047925
+  eval-batch-train:
+    avg_avg_episode_length:                     22.61741
+    avg_avg_return:                             0.88960445
+    avg_benchmark_regret:                       0.09761257
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.120140076
+    proxy_corner:
+      avg_avg_return:                           0.057286337
+      avg_reward_per_step:                      0.004173279
+  eval-batch-orig:
+    avg_avg_episode_length:                     28.732944
+    avg_avg_return:                             0.84499127
+    avg_benchmark_regret:                       0.13952026
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07485962
+    proxy_corner:
+      avg_avg_return:                           0.84499127
+      avg_reward_per_step:                      0.07485962
+  eval-batch-shift:
+    avg_avg_episode_length:                     22.567322
+    avg_avg_return:                             0.88528776
+    avg_benchmark_regret:                       0.10221824
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.1348877
+    proxy_corner:
+      avg_avg_return:                           0.05425418
+      avg_reward_per_step:                      0.0027999878
+  eval-batch-tree:
+    avg_avg_episode_length:                     26.600721
+    avg_avg_return:                             0.86460793
+    avg_benchmark_regret:                       0.1200272
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.091667175
+    proxy_corner:
+      avg_avg_return:                           0.86460793
+      avg_reward_per_step:                      0.091667175
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        105906176
+    env-step-replay-before:                     54198272
+    env-step-used-before:                       105906176
+    ppo-update-before:                          64640
+    env-step-generate-before:                   51707904
+    env-step-all-after:                         105938944
+    env-step-replay-after:                      54198272
+    env-step-used-after:                        105938944
+    ppo-update-after:                           64660
+    env-step-generate-after:                    51740672
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     25.268475
+    avg_avg_return:                             0.79656637
+    avg_reward_per_step:                        0.11679077
+    proxy_corner:
+      avg_avg_return:                           0.06093779
+      avg_reward_per_step:                      0.0043945312
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     25.268475
+    avg_avg_return:                             0.79656637
+    avg_reward_per_step:                        0.11679077
+    proxy_corner:
+      avg_avg_return:                           0.06093779
+      avg_reward_per_step:                      0.0043945312
+  perf:
+    env_steps_per_second:                       321009.67870652204
+    ppo_updates_per_second:                     41.0932670338285
+    eval-batch-train-duration:                  0.5350160226225853
+    eval-batch-orig-duration:                   0.4964044336229563
+    eval-batch-shift-duration:                  0.4883113522082567
+    eval-batch-tree-duration:                   0.48710216023027897
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.175293
+      cheese-corne_dist_wavg:                   18.012875
+      levels_cheese_is_in_corner_avg:           0.016845703
+      mouse-cheese_dist_finite_avg:             22.96753
+      mouse-cheese_dist_finite_wavg:            10.418712
+      mouse-cheese_dist_solvable_avg:           22.96753
+      mouse-cheese_dist_solvable_wavg:          10.418713
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.58982193
+    visit_patterns:
+      avg_first_visit_time:                     1400.6819
+      avg_last_visit_time:                      1644.3206
+      num_replay_batches:                       1654
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010152255
+    avg_actor_approxkl3:                        0.00081141124
+    avg_actor_clipfrac:                         0.031237794
+    avg_actor_loss:                             -0.0006477969
+    avg_advantage:                              0.005862806
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0030988124
+    avg_entropy:                                0.12167294
+    avg_grad_norm_pre_clip:                     1.161332
+    avg_loss:                                   0.00077993644
+    max:
+      max_actor_approxkl1:                      0.0032588807
+      max_actor_approxkl3:                      0.0017254411
+      max_actor_clipfrac:                       0.056152344
+      max_actor_loss:                           0.0007731542
+      max_advantage:                            0.1612634
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0046875672
+      max_entropy:                              0.13664171
+      max_grad_norm_pre_clip:                   1.8978921
+      max_loss:                                 0.0022391032
+    std:
+      std_actor_approxkl1:                      0.00086231006
+      std_actor_approxkl3:                      0.00042481333
+      std_actor_clipfrac:                       0.012984169
+      std_actor_loss:                           0.00079774525
+      std_advantage:                            0.07893492
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00066687184
+      std_entropy:                              0.009870151
+      std_grad_norm_pre_clip:                   0.3028268
+      std_loss:                                 0.0007586052
+  eval-batch-train:
+    avg_avg_episode_length:                     22.88266
+    avg_avg_return:                             0.8842709
+    avg_benchmark_regret:                       0.102946185
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12062836
+    proxy_corner:
+      avg_avg_return:                           0.05492724
+      avg_reward_per_step:                      0.004386902
+  eval-batch-orig:
+    avg_avg_episode_length:                     28.548937
+    avg_avg_return:                             0.84329087
+    avg_benchmark_regret:                       0.14122069
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07393646
+    proxy_corner:
+      avg_avg_return:                           0.84329087
+      avg_reward_per_step:                      0.07393646
+  eval-batch-shift:
+    avg_avg_episode_length:                     22.14365
+    avg_avg_return:                             0.8863222
+    avg_benchmark_regret:                       0.10118374
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13557434
+    proxy_corner:
+      avg_avg_return:                           0.053957134
+      avg_reward_per_step:                      0.0028305054
+  eval-batch-tree:
+    avg_avg_episode_length:                     26.988098
+    avg_avg_return:                             0.86114264
+    avg_benchmark_regret:                       0.123492576
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09033966
+    proxy_corner:
+      avg_avg_return:                           0.86114264
+      avg_reward_per_step:                      0.09033966
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        106954752
+    env-step-replay-before:                     54624256
+    env-step-used-before:                       106954752
+    ppo-update-before:                          65280
+    env-step-generate-before:                   52330496
+    env-step-all-after:                         106987520
+    env-step-replay-after:                      54624256
+    env-step-used-after:                        106987520
+    ppo-update-after:                           65300
+    env-step-generate-after:                    52363264
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     24.084358
+    avg_avg_return:                             0.81216955
+    avg_reward_per_step:                        0.114715576
+    proxy_corner:
+      avg_avg_return:                           0.064021416
+      avg_reward_per_step:                      0.0050354004
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     24.084358
+    avg_avg_return:                             0.81216955
+    avg_reward_per_step:                        0.114715576
+    proxy_corner:
+      avg_avg_return:                           0.064021416
+      avg_reward_per_step:                      0.0050354004
+  perf:
+    env_steps_per_second:                       351101.9968538464
+    ppo_updates_per_second:                     42.82609005453834
+    eval-batch-train-duration:                  0.5244691958650947
+    eval-batch-orig-duration:                   0.3757885005325079
+    eval-batch-shift-duration:                  0.23710127733647823
+    eval-batch-tree-duration:                   0.2296511046588421
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.324707
+      cheese-corne_dist_wavg:                   18.016935
+      levels_cheese_is_in_corner_avg:           0.01586914
+      mouse-cheese_dist_finite_avg:             22.94629
+      mouse-cheese_dist_finite_wavg:            10.396822
+      mouse-cheese_dist_solvable_avg:           22.94629
+      mouse-cheese_dist_solvable_wavg:          10.396822
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.5967712
+    visit_patterns:
+      avg_first_visit_time:                     1414.3259
+      avg_last_visit_time:                      1657.1807
+      num_replay_batches:                       1667
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010262997
+    avg_actor_approxkl3:                        0.0008690103
+    avg_actor_clipfrac:                         0.03220215
+    avg_actor_loss:                             -0.0010127887
+    avg_advantage:                              0.03960482
+    avg_critic_clipfrac:                        6.713867e-05
+    avg_critic_loss:                            0.0028749777
+    avg_entropy:                                0.1256402
+    avg_grad_norm_pre_clip:                     1.0577582
+    avg_loss:                                   0.00029906
+    max:
+      max_actor_approxkl1:                      0.0038906077
+      max_actor_approxkl3:                      0.0027032665
+      max_actor_clipfrac:                       0.059448242
+      max_actor_loss:                           0.00027555227
+      max_advantage:                            0.20200843
+      max_critic_clipfrac:                      0.0013427734
+      max_critic_loss:                          0.0039589973
+      max_entropy:                              0.14735815
+      max_grad_norm_pre_clip:                   1.8719264
+      max_loss:                                 0.0020075475
+    std:
+      std_actor_approxkl1:                      0.00085057446
+      std_actor_approxkl3:                      0.0005996427
+      std_actor_clipfrac:                       0.01379711
+      std_actor_loss:                           0.0009422649
+      std_advantage:                            0.07276868
+      std_critic_clipfrac:                      0.00029265066
+      std_critic_loss:                          0.0006930789
+      std_entropy:                              0.012107957
+      std_grad_norm_pre_clip:                   0.35873783
+      std_loss:                                 0.0009378786
+  eval-batch-train:
+    avg_avg_episode_length:                     22.558426
+    avg_avg_return:                             0.8786107
+    avg_benchmark_regret:                       0.10860641
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.120918274
+    proxy_corner:
+      avg_avg_return:                           0.061528224
+      avg_reward_per_step:                      0.004524231
+  eval-batch-orig:
+    avg_avg_episode_length:                     27.475456
+    avg_avg_return:                             0.8495136
+    avg_benchmark_regret:                       0.13499796
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.077560425
+    proxy_corner:
+      avg_avg_return:                           0.8495136
+      avg_reward_per_step:                      0.077560425
+  eval-batch-shift:
+    avg_avg_episode_length:                     21.626968
+    avg_avg_return:                             0.8934532
+    avg_benchmark_regret:                       0.09405275
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13713837
+    proxy_corner:
+      avg_avg_return:                           0.057944447
+      avg_reward_per_step:                      0.0027770996
+  eval-batch-tree:
+    avg_avg_episode_length:                     26.256723
+    avg_avg_return:                             0.85697794
+    avg_benchmark_regret:                       0.1276573
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09384155
+    proxy_corner:
+      avg_avg_return:                           0.85697794
+      avg_reward_per_step:                      0.09384155
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        108003328
+    env-step-replay-before:                     55115776
+    env-step-used-before:                       108003328
+    ppo-update-before:                          65920
+    env-step-generate-before:                   52887552
+    env-step-all-after:                         108036096
+    env-step-replay-after:                      55148544
+    env-step-used-after:                        108036096
+    ppo-update-after:                           65940
+    env-step-generate-after:                    52887552
+  train-replay:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     69.22815
+    avg_avg_return:                             0.36492604
+    avg_reward_per_step:                        0.0107421875
+    proxy_corner:
+      avg_avg_return:                           0.08053061
+      avg_reward_per_step:                      0.001373291
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     69.22815
+    avg_avg_return:                             0.36492604
+    avg_reward_per_step:                        0.0107421875
+    proxy_corner:
+      avg_avg_return:                           0.08053061
+      avg_reward_per_step:                      0.001373291
+  perf:
+    env_steps_per_second:                       245138.10107706764
+    ppo_updates_per_second:                     41.639064500705025
+    eval-batch-train-duration:                  0.5406170543283224
+    eval-batch-orig-duration:                   0.47528249584138393
+    eval-batch-shift-duration:                  0.4487793641164899
+    eval-batch-tree-duration:                   0.4766727015376091
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.362549
+      cheese-corne_dist_wavg:                   18.022388
+      levels_cheese_is_in_corner_avg:           0.017333984
+      mouse-cheese_dist_finite_avg:             23.091797
+      mouse-cheese_dist_finite_wavg:            10.390366
+      mouse-cheese_dist_solvable_avg:           23.091797
+      mouse-cheese_dist_solvable_wavg:          10.390367
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000008
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.6212387
+    visit_patterns:
+      avg_first_visit_time:                     1427.4792
+      avg_last_visit_time:                      1673.3054
+      num_replay_batches:                       1683
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0017699555
+    avg_actor_approxkl3:                        0.0014748867
+    avg_actor_clipfrac:                         0.065661624
+    avg_actor_loss:                             -8.140952e-05
+    avg_advantage:                              -0.045316294
+    avg_critic_clipfrac:                        0.009381103
+    avg_critic_loss:                            0.012573709
+    avg_entropy:                                0.27600232
+    avg_grad_norm_pre_clip:                     2.1339538
+    avg_loss:                                   0.0059294426
+    max:
+      max_actor_approxkl1:                      0.0045701573
+      max_actor_approxkl3:                      0.0034502489
+      max_actor_clipfrac:                       0.12902832
+      max_actor_loss:                           0.002679795
+      max_advantage:                            0.20086193
+      max_critic_clipfrac:                      0.080444336
+      max_critic_loss:                          0.015592013
+      max_entropy:                              0.30763268
+      max_grad_norm_pre_clip:                   4.420469
+      max_loss:                                 0.008850144
+    std:
+      std_actor_approxkl1:                      0.0012879011
+      std_actor_approxkl3:                      0.00090697495
+      std_actor_clipfrac:                       0.03238148
+      std_actor_loss:                           0.0011627668
+      std_advantage:                            0.15640877
+      std_critic_clipfrac:                      0.022217011
+      std_critic_loss:                          0.0013003921
+      std_entropy:                              0.013068739
+      std_grad_norm_pre_clip:                   0.83102006
+      std_loss:                                 0.0014114185
+  eval-batch-train:
+    avg_avg_episode_length:                     23.278316
+    avg_avg_return:                             0.8801025
+    avg_benchmark_regret:                       0.10711458
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12084198
+    proxy_corner:
+      avg_avg_return:                           0.054134876
+      avg_reward_per_step:                      0.004524231
+  eval-batch-orig:
+    avg_avg_episode_length:                     28.35999
+    avg_avg_return:                             0.849168
+    avg_benchmark_regret:                       0.13534352
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07064819
+    proxy_corner:
+      avg_avg_return:                           0.849168
+      avg_reward_per_step:                      0.07064819
+  eval-batch-shift:
+    avg_avg_episode_length:                     22.802654
+    avg_avg_return:                             0.88259387
+    avg_benchmark_regret:                       0.10491209
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.1355896
+    proxy_corner:
+      avg_avg_return:                           0.048504103
+      avg_reward_per_step:                      0.0025482178
+  eval-batch-tree:
+    avg_avg_episode_length:                     26.025723
+    avg_avg_return:                             0.8666867
+    avg_benchmark_regret:                       0.11794849
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.08836365
+    proxy_corner:
+      avg_avg_return:                           0.8666867
+      avg_reward_per_step:                      0.08836365
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        109051904
+    env-step-replay-before:                     55541760
+    env-step-used-before:                       109051904
+    ppo-update-before:                          66560
+    env-step-generate-before:                   53510144
+    env-step-all-after:                         109084672
+    env-step-replay-after:                      55574528
+    env-step-used-after:                        109084672
+    ppo-update-after:                           66580
+    env-step-generate-after:                    53510144
+  train-replay:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     71.12796
+    avg_avg_return:                             0.36079514
+    avg_reward_per_step:                        0.0095825195
+    proxy_corner:
+      avg_avg_return:                           0.06639485
+      avg_reward_per_step:                      0.0009460449
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     71.12796
+    avg_avg_return:                             0.36079514
+    avg_reward_per_step:                        0.0095825195
+    proxy_corner:
+      avg_avg_return:                           0.06639485
+      avg_reward_per_step:                      0.0009460449
+  perf:
+    env_steps_per_second:                       307682.7348553096
+    ppo_updates_per_second:                     42.95339066186857
+    eval-batch-train-duration:                  0.5335833933204412
+    eval-batch-orig-duration:                   0.4837004281580448
+    eval-batch-shift-duration:                  0.43866387102752924
+    eval-batch-tree-duration:                   0.4753534933552146
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.315186
+      cheese-corne_dist_wavg:                   18.018427
+      levels_cheese_is_in_corner_avg:           0.017578125
+      mouse-cheese_dist_finite_avg:             23.08667
+      mouse-cheese_dist_finite_wavg:            10.407044
+      mouse-cheese_dist_solvable_avg:           23.08667
+      mouse-cheese_dist_solvable_wavg:          10.407045
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.615162
+    visit_patterns:
+      avg_first_visit_time:                     1439.1604
+      avg_last_visit_time:                      1686.2556
+      num_replay_batches:                       1696
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010380105
+    avg_actor_approxkl3:                        0.0009715285
+    avg_actor_clipfrac:                         0.045806885
+    avg_actor_loss:                             -0.00073056296
+    avg_advantage:                              -0.04232993
+    avg_critic_clipfrac:                        0.0007019043
+    avg_critic_loss:                            0.011902811
+    avg_entropy:                                0.28173706
+    avg_grad_norm_pre_clip:                     1.8256857
+    avg_loss:                                   0.004939105
+    max:
+      max_actor_approxkl1:                      0.0024827423
+      max_actor_approxkl3:                      0.0018976708
+      max_actor_clipfrac:                       0.08190918
+      max_actor_loss:                           0.0011833012
+      max_advantage:                            0.19858992
+      max_critic_clipfrac:                      0.009521484
+      max_critic_loss:                          0.013486974
+      max_entropy:                              0.29817808
+      max_grad_norm_pre_clip:                   3.284306
+      max_loss:                                 0.007342625
+    std:
+      std_actor_approxkl1:                      0.0007430816
+      std_actor_approxkl3:                      0.0005169096
+      std_actor_clipfrac:                       0.02404983
+      std_actor_loss:                           0.0010987809
+      std_advantage:                            0.1533408
+      std_critic_clipfrac:                      0.0021374072
+      std_critic_loss:                          0.0010191452
+      std_entropy:                              0.009754452
+      std_grad_norm_pre_clip:                   0.71269184
+      std_loss:                                 0.0012576508
+  eval-batch-train:
+    avg_avg_episode_length:                     22.905205
+    avg_avg_return:                             0.880647
+    avg_benchmark_regret:                       0.106570065
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12213898
+    proxy_corner:
+      avg_avg_return:                           0.056655537
+      avg_reward_per_step:                      0.0044174194
+  eval-batch-orig:
+    avg_avg_episode_length:                     28.20991
+    avg_avg_return:                             0.8420192
+    avg_benchmark_regret:                       0.14249232
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07575989
+    proxy_corner:
+      avg_avg_return:                           0.8420192
+      avg_reward_per_step:                      0.07575989
+  eval-batch-shift:
+    avg_avg_episode_length:                     22.330656
+    avg_avg_return:                             0.8857577
+    avg_benchmark_regret:                       0.10174833
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13559723
+    proxy_corner:
+      avg_avg_return:                           0.056090336
+      avg_reward_per_step:                      0.0026931763
+  eval-batch-tree:
+    avg_avg_episode_length:                     26.051874
+    avg_avg_return:                             0.8622999
+    avg_benchmark_regret:                       0.12233532
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09169006
+    proxy_corner:
+      avg_avg_return:                           0.8622999
+      avg_reward_per_step:                      0.09169006
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        110100480
+    env-step-replay-before:                     56066048
+    env-step-used-before:                       110100480
+    ppo-update-before:                          67200
+    env-step-generate-before:                   54034432
+    env-step-all-after:                         110133248
+    env-step-replay-after:                      56098816
+    env-step-used-after:                        110133248
+    ppo-update-after:                           67220
+    env-step-generate-after:                    54034432
+  train-replay:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     72.57072
+    avg_avg_return:                             0.35363612
+    avg_reward_per_step:                        0.008636475
+    proxy_corner:
+      avg_avg_return:                           0.12326677
+      avg_reward_per_step:                      0.0020446777
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     72.57072
+    avg_avg_return:                             0.35363612
+    avg_reward_per_step:                        0.008636475
+    proxy_corner:
+      avg_avg_return:                           0.12326677
+      avg_reward_per_step:                      0.0020446777
+  perf:
+    env_steps_per_second:                       315544.4809234464
+    ppo_updates_per_second:                     42.038276072321686
+    eval-batch-train-duration:                  0.5421888213604689
+    eval-batch-orig-duration:                   0.48715521953999996
+    eval-batch-shift-duration:                  0.4691465264186263
+    eval-batch-tree-duration:                   0.4470868734642863
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.371338
+      cheese-corne_dist_wavg:                   21.633526
+      levels_cheese_is_in_corner_avg:           0.017578125
+      mouse-cheese_dist_finite_avg:             23.245361
+      mouse-cheese_dist_finite_wavg:            10.424896
+      mouse-cheese_dist_solvable_avg:           23.245361
+      mouse-cheese_dist_solvable_wavg:          10.424897
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.6242194
+    visit_patterns:
+      avg_first_visit_time:                     1451.7017
+      avg_last_visit_time:                      1702.1868
+      num_replay_batches:                       1712
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0005928899
+    avg_actor_approxkl3:                        0.0007912959
+    avg_actor_clipfrac:                         0.040783692
+    avg_actor_loss:                             -0.0007882267
+    avg_advantage:                              -0.04309377
+    avg_critic_clipfrac:                        0.010217286
+    avg_critic_loss:                            0.011936803
+    avg_entropy:                                0.27834734
+    avg_grad_norm_pre_clip:                     1.7927237
+    avg_loss:                                   0.0049018282
+    max:
+      max_actor_approxkl1:                      0.0014529333
+      max_actor_approxkl3:                      0.0017823933
+      max_actor_clipfrac:                       0.08166504
+      max_actor_loss:                           0.002225101
+      max_advantage:                            0.22867006
+      max_critic_clipfrac:                      0.08679199
+      max_critic_loss:                          0.014429183
+      max_entropy:                              0.2924165
+      max_grad_norm_pre_clip:                   3.8505497
+      max_loss:                                 0.0076701804
+    std:
+      std_actor_approxkl1:                      0.00040936476
+      std_actor_approxkl3:                      0.00036314415
+      std_actor_clipfrac:                       0.017463261
+      std_actor_loss:                           0.0012571551
+      std_advantage:                            0.15275642
+      std_critic_clipfrac:                      0.024334345
+      std_critic_loss:                          0.0013247108
+      std_entropy:                              0.009620993
+      std_grad_norm_pre_clip:                   0.7551053
+      std_loss:                                 0.0014031507
+  eval-batch-train:
+    avg_avg_episode_length:                     22.83696
+    avg_avg_return:                             0.88007617
+    avg_benchmark_regret:                       0.10714097
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12210846
+    proxy_corner:
+      avg_avg_return:                           0.060475297
+      avg_reward_per_step:                      0.0049591064
+  eval-batch-orig:
+    avg_avg_episode_length:                     27.265636
+    avg_avg_return:                             0.85491085
+    avg_benchmark_regret:                       0.12960066
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07835388
+    proxy_corner:
+      avg_avg_return:                           0.85491085
+      avg_reward_per_step:                      0.07835388
+  eval-batch-shift:
+    avg_avg_episode_length:                     21.60889
+    avg_avg_return:                             0.8913053
+    avg_benchmark_regret:                       0.09620059
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13690948
+    proxy_corner:
+      avg_avg_return:                           0.053849198
+      avg_reward_per_step:                      0.0028076172
+  eval-batch-tree:
+    avg_avg_episode_length:                     26.94439
+    avg_avg_return:                             0.8510661
+    avg_benchmark_regret:                       0.13356909
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09277344
+    proxy_corner:
+      avg_avg_return:                           0.8510661
+      avg_reward_per_step:                      0.09277344
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        111149056
+    env-step-replay-before:                     56688640
+    env-step-used-before:                       111149056
+    ppo-update-before:                          67840
+    env-step-generate-before:                   54460416
+    env-step-all-after:                         111181824
+    env-step-replay-after:                      56721408
+    env-step-used-after:                        111181824
+    ppo-update-after:                           67860
+    env-step-generate-after:                    54460416
+  train-replay:
+    shift_proportion:                           1.0
+    avg_avg_episode_length:                     68.467575
+    avg_avg_return:                             0.40365574
+    avg_reward_per_step:                        0.011077881
+    proxy_corner:
+      avg_avg_return:                           0.09584656
+      avg_reward_per_step:                      0.0014648438
+  train-all:
+    shift_proportion:                           1.0
+    avg_avg_episode_length:                     68.467575
+    avg_avg_return:                             0.40365574
+    avg_reward_per_step:                        0.011077881
+    proxy_corner:
+      avg_avg_return:                           0.09584656
+      avg_reward_per_step:                      0.0014648438
+  perf:
+    env_steps_per_second:                       323792.5640682162
+    ppo_updates_per_second:                     42.036924544768645
+    eval-batch-train-duration:                  0.5121981557458639
+    eval-batch-orig-duration:                   0.4847302846610546
+    eval-batch-shift-duration:                  0.4973755031824112
+    eval-batch-tree-duration:                   0.49786206148564816
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.189453
+      cheese-corne_dist_wavg:                   5.458993
+      levels_cheese_is_in_corner_avg:           0.017822266
+      mouse-cheese_dist_finite_avg:             23.239258
+      mouse-cheese_dist_finite_wavg:            10.432068
+      mouse-cheese_dist_solvable_avg:           23.239258
+      mouse-cheese_dist_solvable_wavg:          10.432068
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016816
+    scoring:
+      avg_scores:                               0.6047477
+    visit_patterns:
+      avg_first_visit_time:                     1461.7314
+      avg_last_visit_time:                      1721.1826
+      num_replay_batches:                       1731
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00094396045
+    avg_actor_approxkl3:                        0.0009018821
+    avg_actor_clipfrac:                         0.04435425
+    avg_actor_loss:                             -0.00043129252
+    avg_advantage:                              0.00046142098
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.010559923
+    avg_entropy:                                0.2679565
+    avg_grad_norm_pre_clip:                     1.8423026
+    avg_loss:                                   0.0045807133
+    max:
+      max_actor_approxkl1:                      0.0024987063
+      max_actor_approxkl3:                      0.0022550416
+      max_actor_clipfrac:                       0.09436035
+      max_actor_loss:                           0.0008138269
+      max_advantage:                            0.3239481
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.011970172
+      max_entropy:                              0.28165764
+      max_grad_norm_pre_clip:                   3.3216898
+      max_loss:                                 0.006352305
+    std:
+      std_actor_approxkl1:                      0.000701681
+      std_actor_approxkl3:                      0.00047537498
+      std_actor_clipfrac:                       0.020609079
+      std_actor_loss:                           0.000933303
+      std_advantage:                            0.14547405
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0006486787
+      std_entropy:                              0.00905149
+      std_grad_norm_pre_clip:                   0.6903332
+      std_loss:                                 0.0010382816
+  eval-batch-train:
+    avg_avg_episode_length:                     23.1687
+    avg_avg_return:                             0.8901205
+    avg_benchmark_regret:                       0.09709653
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.11656952
+    proxy_corner:
+      avg_avg_return:                           0.055570908
+      avg_reward_per_step:                      0.0043563843
+  eval-batch-orig:
+    avg_avg_episode_length:                     29.335999
+    avg_avg_return:                             0.8466931
+    avg_benchmark_regret:                       0.13781843
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07241821
+    proxy_corner:
+      avg_avg_return:                           0.8466931
+      avg_reward_per_step:                      0.07241821
+  eval-batch-shift:
+    avg_avg_episode_length:                     23.066704
+    avg_avg_return:                             0.8862113
+    avg_benchmark_regret:                       0.10129465
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13379669
+    proxy_corner:
+      avg_avg_return:                           0.04879442
+      avg_reward_per_step:                      0.0026550293
+  eval-batch-tree:
+    avg_avg_episode_length:                     29.335213
+    avg_avg_return:                             0.843729
+    avg_benchmark_regret:                       0.14090613
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.087142944
+    proxy_corner:
+      avg_avg_return:                           0.843729
+      avg_reward_per_step:                      0.087142944
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        112197632
+    env-step-replay-before:                     57376768
+    env-step-used-before:                       112197632
+    ppo-update-before:                          68480
+    env-step-generate-before:                   54820864
+    env-step-all-after:                         112230400
+    env-step-replay-after:                      57409536
+    env-step-used-after:                        112230400
+    ppo-update-after:                           68500
+    env-step-generate-after:                    54820864
+  train-replay:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     66.918594
+    avg_avg_return:                             0.44714534
+    avg_reward_per_step:                        0.01171875
+    proxy_corner:
+      avg_avg_return:                           0.09254199
+      avg_reward_per_step:                      0.0016174316
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     66.918594
+    avg_avg_return:                             0.44714534
+    avg_reward_per_step:                        0.01171875
+    proxy_corner:
+      avg_avg_return:                           0.09254199
+      avg_reward_per_step:                      0.0016174316
+  perf:
+    env_steps_per_second:                       319463.3269788067
+    ppo_updates_per_second:                     42.61138856059901
+    eval-batch-train-duration:                  0.5355882905423641
+    eval-batch-orig-duration:                   0.476766686886549
+    eval-batch-shift-duration:                  0.4850625777617097
+    eval-batch-tree-duration:                   0.45653122942894697
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.279785
+      cheese-corne_dist_wavg:                   16.215244
+      levels_cheese_is_in_corner_avg:           0.018310547
+      mouse-cheese_dist_finite_avg:             23.31372
+      mouse-cheese_dist_finite_wavg:            12.225881
+      mouse-cheese_dist_solvable_avg:           23.31372
+      mouse-cheese_dist_solvable_wavg:          12.225883
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016822
+    scoring:
+      avg_scores:                               0.5764091
+    visit_patterns:
+      avg_first_visit_time:                     1476.0105
+      avg_last_visit_time:                      1742.2781
+      num_replay_batches:                       1752
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0007542979
+    avg_actor_approxkl3:                        0.0008011102
+    avg_actor_clipfrac:                         0.03977051
+    avg_actor_loss:                             -0.00069283135
+    avg_advantage:                              -0.020049362
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.009135756
+    avg_entropy:                                0.25421488
+    avg_grad_norm_pre_clip:                     1.4865516
+    avg_loss:                                   0.0036208313
+    max:
+      max_actor_approxkl1:                      0.002642295
+      max_actor_approxkl3:                      0.0014859464
+      max_actor_clipfrac:                       0.06738281
+      max_actor_loss:                           0.0006812066
+      max_advantage:                            0.2275191
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.012412418
+      max_entropy:                              0.26975617
+      max_grad_norm_pre_clip:                   2.4386992
+      max_loss:                                 0.005795033
+    std:
+      std_actor_approxkl1:                      0.00063429214
+      std_actor_approxkl3:                      0.00035279308
+      std_actor_clipfrac:                       0.016723035
+      std_actor_loss:                           0.00081205484
+      std_advantage:                            0.13498351
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0014077078
+      std_entropy:                              0.010045705
+      std_grad_norm_pre_clip:                   0.5857915
+      std_loss:                                 0.0011310301
+  eval-batch-train:
+    avg_avg_episode_length:                     22.091825
+    avg_avg_return:                             0.894349
+    avg_benchmark_regret:                       0.09286806
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12050629
+    proxy_corner:
+      avg_avg_return:                           0.058866538
+      avg_reward_per_step:                      0.004432678
+  eval-batch-orig:
+    avg_avg_episode_length:                     27.85223
+    avg_avg_return:                             0.85378015
+    avg_benchmark_regret:                       0.13073137
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.079078674
+    proxy_corner:
+      avg_avg_return:                           0.85378015
+      avg_reward_per_step:                      0.079078674
+  eval-batch-shift:
+    avg_avg_episode_length:                     21.547874
+    avg_avg_return:                             0.8981833
+    avg_benchmark_regret:                       0.0893227
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13613129
+    proxy_corner:
+      avg_avg_return:                           0.052805528
+      avg_reward_per_step:                      0.0029678345
+  eval-batch-tree:
+    avg_avg_episode_length:                     26.568317
+    avg_avg_return:                             0.8592942
+    avg_benchmark_regret:                       0.12534101
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.093322754
+    proxy_corner:
+      avg_avg_return:                           0.8592942
+      avg_reward_per_step:                      0.093322754
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        113246208
+    env-step-replay-before:                     57901056
+    env-step-used-before:                       113246208
+    ppo-update-before:                          69120
+    env-step-generate-before:                   55345152
+    env-step-all-after:                         113278976
+    env-step-replay-after:                      57901056
+    env-step-used-after:                        113278976
+    ppo-update-after:                           69140
+    env-step-generate-after:                    55377920
+  train-generate:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     26.864296
+    avg_avg_return:                             0.7806518
+    avg_reward_per_step:                        0.10354614
+    proxy_corner:
+      avg_avg_return:                           0.067577675
+      avg_reward_per_step:                      0.0046081543
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     26.864296
+    avg_avg_return:                             0.7806518
+    avg_reward_per_step:                        0.10354614
+    proxy_corner:
+      avg_avg_return:                           0.067577675
+      avg_reward_per_step:                      0.0046081543
+  perf:
+    env_steps_per_second:                       347160.04242040886
+    ppo_updates_per_second:                     41.23193353875498
+    eval-batch-train-duration:                  0.5422420250251889
+    eval-batch-orig-duration:                   0.4875128408893943
+    eval-batch-shift-duration:                  0.4937736950814724
+    eval-batch-tree-duration:                   0.5069948239251971
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.387451
+      cheese-corne_dist_wavg:                   11.722359
+      levels_cheese_is_in_corner_avg:           0.018310547
+      mouse-cheese_dist_finite_avg:             23.343018
+      mouse-cheese_dist_finite_wavg:            11.324669
+      mouse-cheese_dist_solvable_avg:           23.343018
+      mouse-cheese_dist_solvable_wavg:          11.32467
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.59270656
+    visit_patterns:
+      avg_first_visit_time:                     1490.562
+      avg_last_visit_time:                      1757.279
+      num_replay_batches:                       1767
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011208613
+    avg_actor_approxkl3:                        0.0011624877
+    avg_actor_clipfrac:                         0.041607667
+    avg_actor_loss:                             -6.653499e-05
+    avg_advantage:                              0.0037363688
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0034937211
+    avg_entropy:                                0.13588046
+    avg_grad_norm_pre_clip:                     1.4961042
+    avg_loss:                                   0.0015444452
+    max:
+      max_actor_approxkl1:                      0.0028311592
+      max_actor_approxkl3:                      0.0023138793
+      max_actor_clipfrac:                       0.07299805
+      max_actor_loss:                           0.0021862984
+      max_advantage:                            0.19214493
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00618283
+      max_entropy:                              0.1741603
+      max_grad_norm_pre_clip:                   2.1303766
+      max_loss:                                 0.0033112147
+    std:
+      std_actor_approxkl1:                      0.00089001807
+      std_actor_approxkl3:                      0.00071973976
+      std_actor_clipfrac:                       0.020219142
+      std_actor_loss:                           0.0011395396
+      std_advantage:                            0.08342959
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0012345181
+      std_entropy:                              0.016402522
+      std_grad_norm_pre_clip:                   0.4460145
+      std_loss:                                 0.0010385258
+  eval-batch-train:
+    avg_avg_episode_length:                     22.783352
+    avg_avg_return:                             0.8822878
+    avg_benchmark_regret:                       0.10492926
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12259674
+    proxy_corner:
+      avg_avg_return:                           0.060060814
+      avg_reward_per_step:                      0.004875183
+  eval-batch-orig:
+    avg_avg_episode_length:                     27.604485
+    avg_avg_return:                             0.84918916
+    avg_benchmark_regret:                       0.13532242
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07954407
+    proxy_corner:
+      avg_avg_return:                           0.84918916
+      avg_reward_per_step:                      0.07954407
+  eval-batch-shift:
+    avg_avg_episode_length:                     21.547192
+    avg_avg_return:                             0.89441097
+    avg_benchmark_regret:                       0.09309499
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13851929
+    proxy_corner:
+      avg_avg_return:                           0.056115493
+      avg_reward_per_step:                      0.002937317
+  eval-batch-tree:
+    avg_avg_episode_length:                     26.061455
+    avg_avg_return:                             0.8688642
+    avg_benchmark_regret:                       0.11577098
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09547424
+    proxy_corner:
+      avg_avg_return:                           0.8688642
+      avg_reward_per_step:                      0.09547424
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        114294784
+    env-step-replay-before:                     58359808
+    env-step-used-before:                       114294784
+    ppo-update-before:                          69760
+    env-step-generate-before:                   55934976
+    env-step-all-after:                         114327552
+    env-step-replay-after:                      58392576
+    env-step-used-after:                        114327552
+    ppo-update-after:                           69780
+    env-step-generate-after:                    55934976
+  train-replay:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     69.31623
+    avg_avg_return:                             0.38244924
+    avg_reward_per_step:                        0.0101623535
+    proxy_corner:
+      avg_avg_return:                           0.08044536
+      avg_reward_per_step:                      0.0014953613
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     69.31623
+    avg_avg_return:                             0.38244924
+    avg_reward_per_step:                        0.0101623535
+    proxy_corner:
+      avg_avg_return:                           0.08044536
+      avg_reward_per_step:                      0.0014953613
+  perf:
+    env_steps_per_second:                       292284.7202279564
+    ppo_updates_per_second:                     41.78771595756868
+    eval-batch-train-duration:                  0.5320199634879827
+    eval-batch-orig-duration:                   0.4736931109800935
+    eval-batch-shift-duration:                  0.48595008719712496
+    eval-batch-tree-duration:                   0.49011316057294607
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.467773
+      cheese-corne_dist_wavg:                   14.433447
+      levels_cheese_is_in_corner_avg:           0.017822266
+      mouse-cheese_dist_finite_avg:             23.268799
+      mouse-cheese_dist_finite_wavg:            12.219782
+      mouse-cheese_dist_solvable_avg:           23.268799
+      mouse-cheese_dist_solvable_wavg:          12.219784
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.6064234
+    visit_patterns:
+      avg_first_visit_time:                     1504.9534
+      avg_last_visit_time:                      1772.3093
+      num_replay_batches:                       1782
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00069774827
+    avg_actor_approxkl3:                        0.00088500726
+    avg_actor_clipfrac:                         0.042858887
+    avg_actor_loss:                             -0.0012388834
+    avg_advantage:                              -0.04982763
+    avg_critic_clipfrac:                        0.017767334
+    avg_critic_loss:                            0.012050148
+    avg_entropy:                                0.26656827
+    avg_grad_norm_pre_clip:                     1.4925666
+    avg_loss:                                   0.0045196223
+    max:
+      max_actor_approxkl1:                      0.001975892
+      max_actor_approxkl3:                      0.0017611652
+      max_actor_clipfrac:                       0.071899414
+      max_actor_loss:                           0.00049063563
+      max_advantage:                            0.16307092
+      max_critic_clipfrac:                      0.17150879
+      max_critic_loss:                          0.014758232
+      max_entropy:                              0.28550583
+      max_grad_norm_pre_clip:                   2.3508816
+      max_loss:                                 0.0062484005
+    std:
+      std_actor_approxkl1:                      0.0005505111
+      std_actor_approxkl3:                      0.0004101952
+      std_actor_clipfrac:                       0.017729506
+      std_actor_loss:                           0.00096755003
+      std_advantage:                            0.1531696
+      std_critic_clipfrac:                      0.04268347
+      std_critic_loss:                          0.0017597455
+      std_entropy:                              0.009509915
+      std_grad_norm_pre_clip:                   0.3546186
+      std_loss:                                 0.0013209061
+  eval-batch-train:
+    avg_avg_episode_length:                     22.443264
+    avg_avg_return:                             0.88642246
+    avg_benchmark_regret:                       0.10079459
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12210083
+    proxy_corner:
+      avg_avg_return:                           0.055871144
+      avg_reward_per_step:                      0.0049209595
+  eval-batch-orig:
+    avg_avg_episode_length:                     26.499645
+    avg_avg_return:                             0.8720942
+    avg_benchmark_regret:                       0.11241728
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.079307556
+    proxy_corner:
+      avg_avg_return:                           0.8720942
+      avg_reward_per_step:                      0.079307556
+  eval-batch-shift:
+    avg_avg_episode_length:                     21.182222
+    avg_avg_return:                             0.8981167
+    avg_benchmark_regret:                       0.089389265
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13964844
+    proxy_corner:
+      avg_avg_return:                           0.05377405
+      avg_reward_per_step:                      0.00289917
+  eval-batch-tree:
+    avg_avg_episode_length:                     25.288223
+    avg_avg_return:                             0.871913
+    avg_benchmark_regret:                       0.11272216
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09429169
+    proxy_corner:
+      avg_avg_return:                           0.871913
+      avg_reward_per_step:                      0.09429169
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        115343360
+    env-step-replay-before:                     58982400
+    env-step-used-before:                       115343360
+    ppo-update-before:                          70400
+    env-step-generate-before:                   56360960
+    env-step-all-after:                         115376128
+    env-step-replay-after:                      58982400
+    env-step-used-after:                        115376128
+    ppo-update-after:                           70420
+    env-step-generate-after:                    56393728
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     24.496208
+    avg_avg_return:                             0.8045131
+    avg_reward_per_step:                        0.1274414
+    proxy_corner:
+      avg_avg_return:                           0.039934423
+      avg_reward_per_step:                      0.0023498535
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     24.496208
+    avg_avg_return:                             0.8045131
+    avg_reward_per_step:                        0.1274414
+    proxy_corner:
+      avg_avg_return:                           0.039934423
+      avg_reward_per_step:                      0.0023498535
+  perf:
+    env_steps_per_second:                       314516.008067379
+    ppo_updates_per_second:                     41.696291121709955
+    eval-batch-train-duration:                  0.5249773962423205
+    eval-batch-orig-duration:                   0.48857957124710083
+    eval-batch-shift-duration:                  0.4865055996924639
+    eval-batch-tree-duration:                   0.47456666082143784
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.402832
+      cheese-corne_dist_wavg:                   16.233883
+      levels_cheese_is_in_corner_avg:           0.018310547
+      mouse-cheese_dist_finite_avg:             23.321533
+      mouse-cheese_dist_finite_wavg:            8.633873
+      mouse-cheese_dist_solvable_avg:           23.321533
+      mouse-cheese_dist_solvable_wavg:          8.633872
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.6028354
+    visit_patterns:
+      avg_first_visit_time:                     1517.9043
+      avg_last_visit_time:                      1790.1526
+      num_replay_batches:                       1800
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0007332675
+    avg_actor_approxkl3:                        0.00057721016
+    avg_actor_clipfrac:                         0.024719238
+    avg_actor_loss:                             -0.0012995884
+    avg_advantage:                              0.011890131
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.002926054
+    avg_entropy:                                0.12558016
+    avg_grad_norm_pre_clip:                     0.95216143
+    avg_loss:                                   3.785833e-05
+    max:
+      max_actor_approxkl1:                      0.002168365
+      max_actor_approxkl3:                      0.0012277577
+      max_actor_clipfrac:                       0.045288086
+      max_actor_loss:                           0.00030114502
+      max_advantage:                            0.17283052
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.005212211
+      max_entropy:                              0.1413621
+      max_grad_norm_pre_clip:                   2.111082
+      max_loss:                                 0.0020128887
+    std:
+      std_actor_approxkl1:                      0.000536711
+      std_actor_approxkl3:                      0.00031637136
+      std_actor_clipfrac:                       0.011726408
+      std_actor_loss:                           0.00097428146
+      std_advantage:                            0.07577475
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011170871
+      std_entropy:                              0.008350524
+      std_grad_norm_pre_clip:                   0.3521579
+      std_loss:                                 0.0010391234
+  eval-batch-train:
+    avg_avg_episode_length:                     22.897072
+    avg_avg_return:                             0.88210016
+    avg_benchmark_regret:                       0.105116926
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12175751
+    proxy_corner:
+      avg_avg_return:                           0.062411245
+      avg_reward_per_step:                      0.005050659
+  eval-batch-orig:
+    avg_avg_episode_length:                     27.081081
+    avg_avg_return:                             0.85625416
+    avg_benchmark_regret:                       0.1282574
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08054352
+    proxy_corner:
+      avg_avg_return:                           0.85625416
+      avg_reward_per_step:                      0.08054352
+  eval-batch-shift:
+    avg_avg_episode_length:                     22.008827
+    avg_avg_return:                             0.8830608
+    avg_benchmark_regret:                       0.10444514
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13921356
+    proxy_corner:
+      avg_avg_return:                           0.0586235
+      avg_reward_per_step:                      0.0030212402
+  eval-batch-tree:
+    avg_avg_episode_length:                     25.957603
+    avg_avg_return:                             0.8592347
+    avg_benchmark_regret:                       0.12540048
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09490204
+    proxy_corner:
+      avg_avg_return:                           0.8592347
+      avg_reward_per_step:                      0.09490204
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        116391936
+    env-step-replay-before:                     59506688
+    env-step-used-before:                       116391936
+    ppo-update-before:                          71040
+    env-step-generate-before:                   56885248
+    env-step-all-after:                         116424704
+    env-step-replay-after:                      59539456
+    env-step-used-after:                        116424704
+    ppo-update-after:                           71060
+    env-step-generate-after:                    56885248
+  train-replay:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     70.039696
+    avg_avg_return:                             0.39355332
+    avg_reward_per_step:                        0.009552002
+    proxy_corner:
+      avg_avg_return:                           0.109661065
+      avg_reward_per_step:                      0.001739502
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     70.039696
+    avg_avg_return:                             0.39355332
+    avg_reward_per_step:                        0.009552002
+    proxy_corner:
+      avg_avg_return:                           0.109661065
+      avg_reward_per_step:                      0.001739502
+  perf:
+    env_steps_per_second:                       244960.7799118968
+    ppo_updates_per_second:                     42.31522229801844
+    eval-batch-train-duration:                  0.666201944462955
+    eval-batch-orig-duration:                   0.6189777916297317
+    eval-batch-shift-duration:                  0.6178399566560984
+    eval-batch-tree-duration:                   0.6159107955172658
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.432861
+      cheese-corne_dist_wavg:                   19.83554
+      levels_cheese_is_in_corner_avg:           0.017578125
+      mouse-cheese_dist_finite_avg:             23.34668
+      mouse-cheese_dist_finite_wavg:            11.332176
+      mouse-cheese_dist_solvable_avg:           23.34668
+      mouse-cheese_dist_solvable_wavg:          11.332179
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016816
+    scoring:
+      avg_scores:                               0.6079463
+    visit_patterns:
+      avg_first_visit_time:                     1530.7522
+      avg_last_visit_time:                      1807.2183
+      num_replay_batches:                       1817
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0015243539
+    avg_actor_approxkl3:                        0.0012961384
+    avg_actor_clipfrac:                         0.062072754
+    avg_actor_loss:                             -6.933287e-05
+    avg_advantage:                              -0.033322107
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.010902786
+    avg_entropy:                                0.28459564
+    avg_grad_norm_pre_clip:                     2.0064209
+    avg_loss:                                   0.005097465
+    max:
+      max_actor_approxkl1:                      0.0032620917
+      max_actor_approxkl3:                      0.0026909737
+      max_actor_clipfrac:                       0.12109375
+      max_actor_loss:                           0.002669707
+      max_advantage:                            0.19827896
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.014524985
+      max_entropy:                              0.3032313
+      max_grad_norm_pre_clip:                   3.141815
+      max_loss:                                 0.008981799
+    std:
+      std_actor_approxkl1:                      0.00083455653
+      std_actor_approxkl3:                      0.00065811235
+      std_actor_clipfrac:                       0.027599758
+      std_actor_loss:                           0.0012402175
+      std_advantage:                            0.14584485
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0012918195
+      std_entropy:                              0.011019772
+      std_grad_norm_pre_clip:                   0.5504491
+      std_loss:                                 0.001553059
+  eval-batch-train:
+    avg_avg_episode_length:                     22.269415
+    avg_avg_return:                             0.889583
+    avg_benchmark_regret:                       0.09763407
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12211609
+    proxy_corner:
+      avg_avg_return:                           0.059220266
+      avg_reward_per_step:                      0.0050354004
+  eval-batch-orig:
+    avg_avg_episode_length:                     27.104744
+    avg_avg_return:                             0.85508
+    avg_benchmark_regret:                       0.12943155
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.080444336
+    proxy_corner:
+      avg_avg_return:                           0.85508
+      avg_reward_per_step:                      0.080444336
+  eval-batch-shift:
+    avg_avg_episode_length:                     21.853922
+    avg_avg_return:                             0.8901235
+    avg_benchmark_regret:                       0.09738241
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13874817
+    proxy_corner:
+      avg_avg_return:                           0.05751363
+      avg_reward_per_step:                      0.002922058
+  eval-batch-tree:
+    avg_avg_episode_length:                     26.116987
+    avg_avg_return:                             0.85719776
+    avg_benchmark_regret:                       0.12743737
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09505463
+    proxy_corner:
+      avg_avg_return:                           0.85719776
+      avg_reward_per_step:                      0.09505463
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        117440512
+    env-step-replay-before:                     60030976
+    env-step-used-before:                       117440512
+    ppo-update-before:                          71680
+    env-step-generate-before:                   57409536
+    env-step-all-after:                         117473280
+    env-step-replay-after:                      60030976
+    env-step-used-after:                        117473280
+    ppo-update-after:                           71700
+    env-step-generate-after:                    57442304
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     25.715366
+    avg_avg_return:                             0.7895875
+    avg_reward_per_step:                        0.115234375
+    proxy_corner:
+      avg_avg_return:                           0.04127567
+      avg_reward_per_step:                      0.0019836426
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     25.715366
+    avg_avg_return:                             0.7895875
+    avg_reward_per_step:                        0.115234375
+    proxy_corner:
+      avg_avg_return:                           0.04127567
+      avg_reward_per_step:                      0.0019836426
+  perf:
+    env_steps_per_second:                       242325.92804519524
+    ppo_updates_per_second:                     42.59552668032237
+    eval-batch-train-duration:                  0.6414687149226665
+    eval-batch-orig-duration:                   0.6230306858196855
+    eval-batch-shift-duration:                  0.5497947866097093
+    eval-batch-tree-duration:                   0.4842416252940893
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.242432
+      cheese-corne_dist_wavg:                   14.456857
+      levels_cheese_is_in_corner_avg:           0.017578125
+      mouse-cheese_dist_finite_avg:             23.422363
+      mouse-cheese_dist_finite_wavg:            12.238161
+      mouse-cheese_dist_solvable_avg:           23.422363
+      mouse-cheese_dist_solvable_wavg:          12.238163
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.597807
+    visit_patterns:
+      avg_first_visit_time:                     1545.9641
+      avg_last_visit_time:                      1822.315
+      num_replay_batches:                       1832
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009229672
+    avg_actor_approxkl3:                        0.00094747415
+    avg_actor_clipfrac:                         0.034143068
+    avg_actor_loss:                             -0.0002996712
+    avg_advantage:                              -0.0018229727
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.003389407
+    avg_entropy:                                0.11644836
+    avg_grad_norm_pre_clip:                     1.2658463
+    avg_loss:                                   0.0012785842
+    max:
+      max_actor_approxkl1:                      0.001908992
+      max_actor_approxkl3:                      0.001806049
+      max_actor_clipfrac:                       0.05114746
+      max_actor_loss:                           0.0023115426
+      max_advantage:                            0.16428268
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.005915909
+      max_entropy:                              0.14280787
+      max_grad_norm_pre_clip:                   2.58644
+      max_loss:                                 0.004751103
+    std:
+      std_actor_approxkl1:                      0.0005655089
+      std_actor_approxkl3:                      0.00035805872
+      std_actor_clipfrac:                       0.010122794
+      std_actor_loss:                           0.001139377
+      std_advantage:                            0.08244656
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011409
+      std_entropy:                              0.015146243
+      std_grad_norm_pre_clip:                   0.42300403
+      std_loss:                                 0.0011396663
+  eval-batch-train:
+    avg_avg_episode_length:                     21.825958
+    avg_avg_return:                             0.893734
+    avg_benchmark_regret:                       0.093483075
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12296295
+    proxy_corner:
+      avg_avg_return:                           0.056471225
+      avg_reward_per_step:                      0.0048828125
+  eval-batch-orig:
+    avg_avg_episode_length:                     26.340975
+    avg_avg_return:                             0.8603413
+    avg_benchmark_regret:                       0.12417025
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.07911682
+    proxy_corner:
+      avg_avg_return:                           0.8603413
+      avg_reward_per_step:                      0.07911682
+  eval-batch-shift:
+    avg_avg_episode_length:                     21.81464
+    avg_avg_return:                             0.8856286
+    avg_benchmark_regret:                       0.10187732
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13866425
+    proxy_corner:
+      avg_avg_return:                           0.051589746
+      avg_reward_per_step:                      0.0027694702
+  eval-batch-tree:
+    avg_avg_episode_length:                     24.933708
+    avg_avg_return:                             0.8696903
+    avg_benchmark_regret:                       0.11494485
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09550476
+    proxy_corner:
+      avg_avg_return:                           0.8696903
+      avg_reward_per_step:                      0.09550476
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        118489088
+    env-step-replay-before:                     60424192
+    env-step-used-before:                       118489088
+    ppo-update-before:                          72320
+    env-step-generate-before:                   58064896
+    env-step-all-after:                         118521856
+    env-step-replay-after:                      60424192
+    env-step-used-after:                        118521856
+    ppo-update-after:                           72340
+    env-step-generate-after:                    58097664
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     24.659666
+    avg_avg_return:                             0.80075073
+    avg_reward_per_step:                        0.114593506
+    proxy_corner:
+      avg_avg_return:                           0.04014945
+      avg_reward_per_step:                      0.0016784668
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     24.659666
+    avg_avg_return:                             0.80075073
+    avg_reward_per_step:                        0.114593506
+    proxy_corner:
+      avg_avg_return:                           0.04014945
+      avg_reward_per_step:                      0.0016784668
+  perf:
+    env_steps_per_second:                       292917.7158916354
+    ppo_updates_per_second:                     41.7106669416866
+    eval-batch-train-duration:                  0.5275565041229129
+    eval-batch-orig-duration:                   0.48741944413632154
+    eval-batch-shift-duration:                  0.48578684963285923
+    eval-batch-tree-duration:                   0.47616811003535986
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.363037
+      cheese-corne_dist_wavg:                   18.037422
+      levels_cheese_is_in_corner_avg:           0.015625
+      mouse-cheese_dist_finite_avg:             23.454346
+      mouse-cheese_dist_finite_wavg:            10.441359
+      mouse-cheese_dist_solvable_avg:           23.454346
+      mouse-cheese_dist_solvable_wavg:          10.441359
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.6071093
+    visit_patterns:
+      avg_first_visit_time:                     1561.9333
+      avg_last_visit_time:                      1834.2195
+      num_replay_batches:                       1844
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0005150758
+    avg_actor_approxkl3:                        0.0006365666
+    avg_actor_clipfrac:                         0.027679443
+    avg_actor_loss:                             -0.0005299849
+    avg_advantage:                              0.0014451154
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0028727204
+    avg_entropy:                                0.116173856
+    avg_grad_norm_pre_clip:                     1.2893537
+    avg_loss:                                   0.00079020153
+    max:
+      max_actor_approxkl1:                      0.0013832286
+      max_actor_approxkl3:                      0.001450842
+      max_actor_clipfrac:                       0.047851562
+      max_actor_loss:                           0.00063922256
+      max_advantage:                            0.17923087
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0045942864
+      max_entropy:                              0.13314149
+      max_grad_norm_pre_clip:                   2.7312925
+      max_loss:                                 0.0025956566
+    std:
+      std_actor_approxkl1:                      0.00047240677
+      std_actor_approxkl3:                      0.00029409883
+      std_actor_clipfrac:                       0.009865281
+      std_actor_loss:                           0.00072945823
+      std_advantage:                            0.076026395
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00084300374
+      std_entropy:                              0.010593237
+      std_grad_norm_pre_clip:                   0.488713
+      std_loss:                                 0.00084844226
+  eval-batch-train:
+    avg_avg_episode_length:                     22.95356
+    avg_avg_return:                             0.8760053
+    avg_benchmark_regret:                       0.11121179
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.121795654
+    proxy_corner:
+      avg_avg_return:                           0.042896546
+      avg_reward_per_step:                      0.004119873
+  eval-batch-orig:
+    avg_avg_episode_length:                     27.084824
+    avg_avg_return:                             0.860686
+    avg_benchmark_regret:                       0.1238255
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.075805664
+    proxy_corner:
+      avg_avg_return:                           0.860686
+      avg_reward_per_step:                      0.075805664
+  eval-batch-shift:
+    avg_avg_episode_length:                     22.930569
+    avg_avg_return:                             0.870734
+    avg_benchmark_regret:                       0.11677201
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13678741
+    proxy_corner:
+      avg_avg_return:                           0.044741802
+      avg_reward_per_step:                      0.0025787354
+  eval-batch-tree:
+    avg_avg_episode_length:                     25.875654
+    avg_avg_return:                             0.8634286
+    avg_benchmark_regret:                       0.12120657
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09164429
+    proxy_corner:
+      avg_avg_return:                           0.8634286
+      avg_reward_per_step:                      0.09164429
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        119537664
+    env-step-replay-before:                     60882944
+    env-step-used-before:                       119537664
+    ppo-update-before:                          72960
+    env-step-generate-before:                   58654720
+    env-step-all-after:                         119570432
+    env-step-replay-after:                      60882944
+    env-step-used-after:                        119570432
+    ppo-update-after:                           72980
+    env-step-generate-after:                    58687488
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     23.948029
+    avg_avg_return:                             0.8014463
+    avg_reward_per_step:                        0.11984253
+    proxy_corner:
+      avg_avg_return:                           0.02965526
+      avg_reward_per_step:                      0.0016174316
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     23.948029
+    avg_avg_return:                             0.8014463
+    avg_reward_per_step:                        0.11984253
+    proxy_corner:
+      avg_avg_return:                           0.02965526
+      avg_reward_per_step:                      0.0016174316
+  perf:
+    env_steps_per_second:                       238732.63520765534
+    ppo_updates_per_second:                     42.16377204064647
+    eval-batch-train-duration:                  0.5458662509918213
+    eval-batch-orig-duration:                   0.48902933672070503
+    eval-batch-shift-duration:                  0.49614675249904394
+    eval-batch-tree-duration:                   0.4854422025382519
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.320557
+      cheese-corne_dist_wavg:                   18.021452
+      levels_cheese_is_in_corner_avg:           0.015625
+      mouse-cheese_dist_finite_avg:             23.486572
+      mouse-cheese_dist_finite_wavg:            10.449224
+      mouse-cheese_dist_solvable_avg:           23.486572
+      mouse-cheese_dist_solvable_wavg:          10.449225
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.00001
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.60975707
+    visit_patterns:
+      avg_first_visit_time:                     1577.5996
+      avg_last_visit_time:                      1848.2441
+      num_replay_batches:                       1858
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010254032
+    avg_actor_approxkl3:                        0.0007460403
+    avg_actor_clipfrac:                         0.02921753
+    avg_actor_loss:                             -0.0004923174
+    avg_advantage:                              0.024394475
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0033873373
+    avg_entropy:                                0.1122451
+    avg_grad_norm_pre_clip:                     1.3459388
+    avg_loss:                                   0.001089106
+    max:
+      max_actor_approxkl1:                      0.0030994834
+      max_actor_approxkl3:                      0.0017090209
+      max_actor_clipfrac:                       0.053833008
+      max_actor_loss:                           0.002318658
+      max_advantage:                            0.18748683
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.005174748
+      max_entropy:                              0.14127192
+      max_grad_norm_pre_clip:                   2.667685
+      max_loss:                                 0.004283012
+    std:
+      std_actor_approxkl1:                      0.0007019668
+      std_actor_approxkl3:                      0.00038825962
+      std_actor_clipfrac:                       0.011088949
+      std_actor_loss:                           0.00092872925
+      std_advantage:                            0.0813539
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00076154445
+      std_entropy:                              0.012311891
+      std_grad_norm_pre_clip:                   0.3912886
+      std_loss:                                 0.000922385
+  eval-batch-train:
+    avg_avg_episode_length:                     22.689112
+    avg_avg_return:                             0.88055575
+    avg_benchmark_regret:                       0.10666129
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12345123
+    proxy_corner:
+      avg_avg_return:                           0.05353956
+      avg_reward_per_step:                      0.004875183
+  eval-batch-orig:
+    avg_avg_episode_length:                     27.451462
+    avg_avg_return:                             0.84523225
+    avg_benchmark_regret:                       0.1392793
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08295441
+    proxy_corner:
+      avg_avg_return:                           0.84523225
+      avg_reward_per_step:                      0.08295441
+  eval-batch-shift:
+    avg_avg_episode_length:                     22.08559
+    avg_avg_return:                             0.8841709
+    avg_benchmark_regret:                       0.10333502
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13964081
+    proxy_corner:
+      avg_avg_return:                           0.055654287
+      avg_reward_per_step:                      0.002784729
+  eval-batch-tree:
+    avg_avg_episode_length:                     25.599892
+    avg_avg_return:                             0.86178184
+    avg_benchmark_regret:                       0.12285332
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09635162
+    proxy_corner:
+      avg_avg_return:                           0.86178184
+      avg_reward_per_step:                      0.09635162
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        120586240
+    env-step-replay-before:                     61407232
+    env-step-used-before:                       120586240
+    ppo-update-before:                          73600
+    env-step-generate-before:                   59179008
+    env-step-all-after:                         120619008
+    env-step-replay-after:                      61440000
+    env-step-used-after:                        120619008
+    ppo-update-after:                           73620
+    env-step-generate-after:                    59179008
+  train-replay:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     70.16033
+    avg_avg_return:                             0.3721388
+    avg_reward_per_step:                        0.0093688965
+    proxy_corner:
+      avg_avg_return:                           0.08806504
+      avg_reward_per_step:                      0.0017089844
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     70.16033
+    avg_avg_return:                             0.3721388
+    avg_reward_per_step:                        0.0093688965
+    proxy_corner:
+      avg_avg_return:                           0.08806504
+      avg_reward_per_step:                      0.0017089844
+  perf:
+    env_steps_per_second:                       275961.03243685904
+    ppo_updates_per_second:                     41.56421763199357
+    eval-batch-train-duration:                  0.5375876622274518
+    eval-batch-orig-duration:                   0.4925546757876873
+    eval-batch-shift-duration:                  0.4965006299316883
+    eval-batch-tree-duration:                   0.478792623616755
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.276611
+      cheese-corne_dist_wavg:                   20.722527
+      levels_cheese_is_in_corner_avg:           0.01586914
+      mouse-cheese_dist_finite_avg:             23.561768
+      mouse-cheese_dist_finite_wavg:            9.557066
+      mouse-cheese_dist_solvable_avg:           23.561768
+      mouse-cheese_dist_solvable_wavg:          9.557065
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.62065023
+    visit_patterns:
+      avg_first_visit_time:                     1589.2275
+      avg_last_visit_time:                      1865.2146
+      num_replay_batches:                       1875
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011625859
+    avg_actor_approxkl3:                        0.0010882712
+    avg_actor_clipfrac:                         0.051757812
+    avg_actor_loss:                             -0.0002520688
+    avg_advantage:                              -0.04341117
+    avg_critic_clipfrac:                        0.0002685547
+    avg_critic_loss:                            0.011775262
+    avg_entropy:                                0.25663453
+    avg_grad_norm_pre_clip:                     2.025129
+    avg_loss:                                   0.005378927
+    max:
+      max_actor_approxkl1:                      0.0033346203
+      max_actor_approxkl3:                      0.0022507836
+      max_actor_clipfrac:                       0.088012695
+      max_actor_loss:                           0.0018434525
+      max_advantage:                            0.18782353
+      max_critic_clipfrac:                      0.0021972656
+      max_critic_loss:                          0.013554439
+      max_entropy:                              0.2683087
+      max_grad_norm_pre_clip:                   3.2991674
+      max_loss:                                 0.007753303
+    std:
+      std_actor_approxkl1:                      0.0007976072
+      std_actor_approxkl3:                      0.0005308719
+      std_actor_clipfrac:                       0.01992631
+      std_actor_loss:                           0.000868529
+      std_advantage:                            0.15226948
+      std_critic_clipfrac:                      0.00063734275
+      std_critic_loss:                          0.0012454713
+      std_entropy:                              0.0053475
+      std_grad_norm_pre_clip:                   0.52562255
+      std_loss:                                 0.0010489675
+  eval-batch-train:
+    avg_avg_episode_length:                     22.803663
+    avg_avg_return:                             0.88300276
+    avg_benchmark_regret:                       0.10421431
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.124053955
+    proxy_corner:
+      avg_avg_return:                           0.05337411
+      avg_reward_per_step:                      0.004798889
+  eval-batch-orig:
+    avg_avg_episode_length:                     27.992859
+    avg_avg_return:                             0.84023917
+    avg_benchmark_regret:                       0.14427231
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08112335
+    proxy_corner:
+      avg_avg_return:                           0.84023917
+      avg_reward_per_step:                      0.08112335
+  eval-batch-shift:
+    avg_avg_episode_length:                     21.598381
+    avg_avg_return:                             0.89477503
+    avg_benchmark_regret:                       0.09273094
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13945007
+    proxy_corner:
+      avg_avg_return:                           0.056325544
+      avg_reward_per_step:                      0.0027694702
+  eval-batch-tree:
+    avg_avg_episode_length:                     26.347267
+    avg_avg_return:                             0.8485034
+    avg_benchmark_regret:                       0.13613178
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09616852
+    proxy_corner:
+      avg_avg_return:                           0.8485034
+      avg_reward_per_step:                      0.09616852
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        121634816
+    env-step-replay-before:                     61964288
+    env-step-used-before:                       121634816
+    ppo-update-before:                          74240
+    env-step-generate-before:                   59670528
+    env-step-all-after:                         121667584
+    env-step-replay-after:                      61997056
+    env-step-used-after:                        121667584
+    ppo-update-after:                           74260
+    env-step-generate-after:                    59670528
+  train-replay:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     71.38101
+    avg_avg_return:                             0.35250056
+    avg_reward_per_step:                        0.009735107
+    proxy_corner:
+      avg_avg_return:                           0.11521725
+      avg_reward_per_step:                      0.0018920898
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     71.38101
+    avg_avg_return:                             0.35250056
+    avg_reward_per_step:                        0.009735107
+    proxy_corner:
+      avg_avg_return:                           0.11521725
+      avg_reward_per_step:                      0.0018920898
+  perf:
+    env_steps_per_second:                       298810.8465460404
+    ppo_updates_per_second:                     40.98415877202701
+    eval-batch-train-duration:                  0.5286702150478959
+    eval-batch-orig-duration:                   0.4816357744857669
+    eval-batch-shift-duration:                  0.48575642704963684
+    eval-batch-tree-duration:                   0.45760295633226633
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.333984
+      cheese-corne_dist_wavg:                   18.041452
+      levels_cheese_is_in_corner_avg:           0.016357422
+      mouse-cheese_dist_finite_avg:             23.55835
+      mouse-cheese_dist_finite_wavg:            10.456804
+      mouse-cheese_dist_solvable_avg:           23.55835
+      mouse-cheese_dist_solvable_wavg:          10.456805
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.6319277
+    visit_patterns:
+      avg_first_visit_time:                     1600.656
+      avg_last_visit_time:                      1882.2852
+      num_replay_batches:                       1892
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011625815
+    avg_actor_approxkl3:                        0.0010783476
+    avg_actor_clipfrac:                         0.05116577
+    avg_actor_loss:                             -0.0007642247
+    avg_advantage:                              -0.041826356
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.01221106
+    avg_entropy:                                0.27997866
+    avg_grad_norm_pre_clip:                     1.6584044
+    avg_loss:                                   0.005061326
+    max:
+      max_actor_approxkl1:                      0.0030773222
+      max_actor_approxkl3:                      0.0022537273
+      max_actor_clipfrac:                       0.09448242
+      max_actor_loss:                           0.0013206303
+      max_advantage:                            0.23250091
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.014669669
+      max_entropy:                              0.30386037
+      max_grad_norm_pre_clip:                   2.60883
+      max_loss:                                 0.008055846
+    std:
+      std_actor_approxkl1:                      0.0007700322
+      std_actor_approxkl3:                      0.0005298607
+      std_actor_clipfrac:                       0.022975964
+      std_actor_loss:                           0.0010876203
+      std_advantage:                            0.15523039
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0012513343
+      std_entropy:                              0.011999821
+      std_grad_norm_pre_clip:                   0.5336566
+      std_loss:                                 0.0013856437
+  eval-batch-train:
+    avg_avg_episode_length:                     22.009314
+    avg_avg_return:                             0.8930106
+    avg_benchmark_regret:                       0.09420645
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12387848
+    proxy_corner:
+      avg_avg_return:                           0.05959015
+      avg_reward_per_step:                      0.0049972534
+  eval-batch-orig:
+    avg_avg_episode_length:                     27.474419
+    avg_avg_return:                             0.8421927
+    avg_benchmark_regret:                       0.14231884
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08150482
+    proxy_corner:
+      avg_avg_return:                           0.8421927
+      avg_reward_per_step:                      0.08150482
+  eval-batch-shift:
+    avg_avg_episode_length:                     20.97209
+    avg_avg_return:                             0.89885616
+    avg_benchmark_regret:                       0.08864978
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13922119
+    proxy_corner:
+      avg_avg_return:                           0.06373567
+      avg_reward_per_step:                      0.003189087
+  eval-batch-tree:
+    avg_avg_episode_length:                     25.827944
+    avg_avg_return:                             0.8564496
+    avg_benchmark_regret:                       0.12818554
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09656525
+    proxy_corner:
+      avg_avg_return:                           0.8564496
+      avg_reward_per_step:                      0.09656525
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        122683392
+    env-step-replay-before:                     62390272
+    env-step-used-before:                       122683392
+    ppo-update-before:                          74880
+    env-step-generate-before:                   60293120
+    env-step-all-after:                         122716160
+    env-step-replay-after:                      62390272
+    env-step-used-after:                        122716160
+    ppo-update-after:                           74900
+    env-step-generate-after:                    60325888
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     25.647095
+    avg_avg_return:                             0.7905038
+    avg_reward_per_step:                        0.11248779
+    proxy_corner:
+      avg_avg_return:                           0.04395779
+      avg_reward_per_step:                      0.0035705566
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     25.647095
+    avg_avg_return:                             0.7905038
+    avg_reward_per_step:                        0.11248779
+    proxy_corner:
+      avg_avg_return:                           0.04395779
+      avg_reward_per_step:                      0.0035705566
+  perf:
+    env_steps_per_second:                       317381.8762579543
+    ppo_updates_per_second:                     41.925360681353155
+    eval-batch-train-duration:                  0.537635387852788
+    eval-batch-orig-duration:                   0.4859088687226176
+    eval-batch-shift-duration:                  0.4861918604001403
+    eval-batch-tree-duration:                   0.49522579461336136
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.315918
+      cheese-corne_dist_wavg:                   8.141095
+      levels_cheese_is_in_corner_avg:           0.017578125
+      mouse-cheese_dist_finite_avg:             23.610352
+      mouse-cheese_dist_finite_wavg:            10.467513
+      mouse-cheese_dist_solvable_avg:           23.610352
+      mouse-cheese_dist_solvable_wavg:          10.467513
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.6285582
+    visit_patterns:
+      avg_first_visit_time:                     1616.9453
+      avg_last_visit_time:                      1894.2156
+      num_replay_batches:                       1904
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00090361404
+    avg_actor_approxkl3:                        0.0010308209
+    avg_actor_clipfrac:                         0.036096193
+    avg_actor_loss:                             0.00025049932
+    avg_advantage:                              -0.0025069918
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0032620677
+    avg_entropy:                                0.1212847
+    avg_grad_norm_pre_clip:                     1.6302071
+    avg_loss:                                   0.0017602483
+    max:
+      max_actor_approxkl1:                      0.0022776541
+      max_actor_approxkl3:                      0.002051125
+      max_actor_clipfrac:                       0.056030273
+      max_actor_loss:                           0.0024806112
+      max_advantage:                            0.19476944
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.006277046
+      max_entropy:                              0.13869785
+      max_grad_norm_pre_clip:                   2.9951882
+      max_loss:                                 0.0046126856
+    std:
+      std_actor_approxkl1:                      0.0006282203
+      std_actor_approxkl3:                      0.0005224846
+      std_actor_clipfrac:                       0.013161709
+      std_actor_loss:                           0.00096163683
+      std_advantage:                            0.08105164
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0009578451
+      std_entropy:                              0.008609106
+      std_grad_norm_pre_clip:                   0.54829276
+      std_loss:                                 0.0011142007
+  eval-batch-train:
+    avg_avg_episode_length:                     21.713406
+    avg_avg_return:                             0.8922844
+    avg_benchmark_regret:                       0.09493266
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12296295
+    proxy_corner:
+      avg_avg_return:                           0.053865742
+      avg_reward_per_step:                      0.0044784546
+  eval-batch-orig:
+    avg_avg_episode_length:                     26.153011
+    avg_avg_return:                             0.865191
+    avg_benchmark_regret:                       0.11932057
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.078125
+    proxy_corner:
+      avg_avg_return:                           0.865191
+      avg_reward_per_step:                      0.078125
+  eval-batch-shift:
+    avg_avg_episode_length:                     21.743874
+    avg_avg_return:                             0.89122486
+    avg_benchmark_regret:                       0.09628108
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13845062
+    proxy_corner:
+      avg_avg_return:                           0.0510925
+      avg_reward_per_step:                      0.0027389526
+  eval-batch-tree:
+    avg_avg_episode_length:                     25.299189
+    avg_avg_return:                             0.8630031
+    avg_benchmark_regret:                       0.12163212
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09249878
+    proxy_corner:
+      avg_avg_return:                           0.8630031
+      avg_reward_per_step:                      0.09249878
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        123731968
+    env-step-replay-before:                     62717952
+    env-step-used-before:                       123731968
+    ppo-update-before:                          75520
+    env-step-generate-before:                   61014016
+    env-step-all-after:                         123764736
+    env-step-replay-after:                      62717952
+    env-step-used-after:                        123764736
+    ppo-update-after:                           75540
+    env-step-generate-after:                    61046784
+  train-generate:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     26.737383
+    avg_avg_return:                             0.7853939
+    avg_reward_per_step:                        0.105041504
+    proxy_corner:
+      avg_avg_return:                           0.04512408
+      avg_reward_per_step:                      0.0044555664
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     26.737383
+    avg_avg_return:                             0.7853939
+    avg_reward_per_step:                        0.105041504
+    proxy_corner:
+      avg_avg_return:                           0.04512408
+      avg_reward_per_step:                      0.0044555664
+  perf:
+    env_steps_per_second:                       271603.5647068425
+    ppo_updates_per_second:                     42.138638175981754
+    eval-batch-train-duration:                  0.5356291281059384
+    eval-batch-orig-duration:                   0.48802562803030014
+    eval-batch-shift-duration:                  0.4795849760994315
+    eval-batch-tree-duration:                   0.4548656502738595
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.354492
+      cheese-corne_dist_wavg:                   14.426366
+      levels_cheese_is_in_corner_avg:           0.018066406
+      mouse-cheese_dist_finite_avg:             23.607422
+      mouse-cheese_dist_finite_wavg:            11.36038
+      mouse-cheese_dist_solvable_avg:           23.607422
+      mouse-cheese_dist_solvable_wavg:          11.36038
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.63593435
+    visit_patterns:
+      avg_first_visit_time:                     1630.9404
+      avg_last_visit_time:                      1904.2078
+      num_replay_batches:                       1914
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0007159734
+    avg_actor_approxkl3:                        0.0006657698
+    avg_actor_clipfrac:                         0.02890625
+    avg_actor_loss:                             -0.0008329114
+    avg_advantage:                              0.0035456219
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0030638047
+    avg_entropy:                                0.1281593
+    avg_grad_norm_pre_clip:                     1.3177968
+    avg_loss:                                   0.00057083153
+    max:
+      max_actor_approxkl1:                      0.0023541474
+      max_actor_approxkl3:                      0.0013218328
+      max_actor_clipfrac:                       0.048095703
+      max_actor_loss:                           0.0010987855
+      max_advantage:                            0.17314768
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.004554011
+      max_entropy:                              0.15346533
+      max_grad_norm_pre_clip:                   1.885908
+      max_loss:                                 0.0019325746
+    std:
+      std_actor_approxkl1:                      0.0005617897
+      std_actor_approxkl3:                      0.00032391577
+      std_actor_clipfrac:                       0.012393151
+      std_actor_loss:                           0.0008444135
+      std_advantage:                            0.078372955
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0007049911
+      std_entropy:                              0.014460628
+      std_grad_norm_pre_clip:                   0.3548968
+      std_loss:                                 0.0008152911
+  eval-batch-train:
+    avg_avg_episode_length:                     22.113811
+    avg_avg_return:                             0.8867792
+    avg_benchmark_regret:                       0.10043788
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12389374
+    proxy_corner:
+      avg_avg_return:                           0.055169787
+      avg_reward_per_step:                      0.004951477
+  eval-batch-orig:
+    avg_avg_episode_length:                     26.381489
+    avg_avg_return:                             0.85716164
+    avg_benchmark_regret:                       0.12734988
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08415985
+    proxy_corner:
+      avg_avg_return:                           0.85716164
+      avg_reward_per_step:                      0.08415985
+  eval-batch-shift:
+    avg_avg_episode_length:                     21.71022
+    avg_avg_return:                             0.8912001
+    avg_benchmark_regret:                       0.09630585
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.1387024
+    proxy_corner:
+      avg_avg_return:                           0.05717557
+      avg_reward_per_step:                      0.0029907227
+  eval-batch-tree:
+    avg_avg_episode_length:                     25.809153
+    avg_avg_return:                             0.8554998
+    avg_benchmark_regret:                       0.12913537
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.097213745
+    proxy_corner:
+      avg_avg_return:                           0.8554998
+      avg_reward_per_step:                      0.097213745
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        124780544
+    env-step-replay-before:                     63176704
+    env-step-used-before:                       124780544
+    ppo-update-before:                          76160
+    env-step-generate-before:                   61603840
+    env-step-all-after:                         124813312
+    env-step-replay-after:                      63176704
+    env-step-used-after:                        124813312
+    ppo-update-after:                           76180
+    env-step-generate-after:                    61636608
+  train-generate:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     25.953108
+    avg_avg_return:                             0.78805995
+    avg_reward_per_step:                        0.1272583
+    proxy_corner:
+      avg_avg_return:                           0.08941299
+      avg_reward_per_step:                      0.006866455
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     25.953108
+    avg_avg_return:                             0.78805995
+    avg_reward_per_step:                        0.1272583
+    proxy_corner:
+      avg_avg_return:                           0.08941299
+      avg_reward_per_step:                      0.006866455
+  perf:
+    env_steps_per_second:                       260576.61348613244
+    ppo_updates_per_second:                     43.12513602845844
+    eval-batch-train-duration:                  0.6237514661625028
+    eval-batch-orig-duration:                   0.5728744911029935
+    eval-batch-shift-duration:                  0.543427487835288
+    eval-batch-tree-duration:                   0.5818698098883033
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.368164
+      cheese-corne_dist_wavg:                   16.248245
+      levels_cheese_is_in_corner_avg:           0.01977539
+      mouse-cheese_dist_finite_avg:             23.655762
+      mouse-cheese_dist_finite_wavg:            10.466673
+      mouse-cheese_dist_solvable_avg:           23.655762
+      mouse-cheese_dist_solvable_wavg:          10.466674
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.6426033
+    visit_patterns:
+      avg_first_visit_time:                     1646.4067
+      avg_last_visit_time:                      1918.3081
+      num_replay_batches:                       1928
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0007192536
+    avg_actor_approxkl3:                        0.00064271217
+    avg_actor_clipfrac:                         0.026605224
+    avg_actor_loss:                             -0.000369191
+    avg_advantage:                              0.0086266585
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0033375863
+    avg_entropy:                                0.123107255
+    avg_grad_norm_pre_clip:                     1.3338264
+    avg_loss:                                   0.0011764948
+    max:
+      max_actor_approxkl1:                      0.0016104676
+      max_actor_approxkl3:                      0.0012775158
+      max_actor_clipfrac:                       0.041625977
+      max_actor_loss:                           0.002766788
+      max_advantage:                            0.18949503
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0049897227
+      max_entropy:                              0.1410987
+      max_grad_norm_pre_clip:                   2.4464185
+      max_loss:                                 0.0046790633
+    std:
+      std_actor_approxkl1:                      0.00049775897
+      std_actor_approxkl3:                      0.00027360165
+      std_actor_clipfrac:                       0.0092431875
+      std_actor_loss:                           0.0011280862
+      std_advantage:                            0.0815397
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0007013261
+      std_entropy:                              0.010324579
+      std_grad_norm_pre_clip:                   0.5544924
+      std_loss:                                 0.0011960146
+  eval-batch-train:
+    avg_avg_episode_length:                     21.593332
+    avg_avg_return:                             0.8885454
+    avg_benchmark_regret:                       0.098671705
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12373352
+    proxy_corner:
+      avg_avg_return:                           0.05472349
+      avg_reward_per_step:                      0.005050659
+  eval-batch-orig:
+    avg_avg_episode_length:                     27.110447
+    avg_avg_return:                             0.8561481
+    avg_benchmark_regret:                       0.12836342
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08113861
+    proxy_corner:
+      avg_avg_return:                           0.8561481
+      avg_reward_per_step:                      0.08113861
+  eval-batch-shift:
+    avg_avg_episode_length:                     21.728333
+    avg_avg_return:                             0.8945321
+    avg_benchmark_regret:                       0.09297389
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13695526
+    proxy_corner:
+      avg_avg_return:                           0.057942368
+      avg_reward_per_step:                      0.0029144287
+  eval-batch-tree:
+    avg_avg_episode_length:                     25.569695
+    avg_avg_return:                             0.85845524
+    avg_benchmark_regret:                       0.12617998
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.096328735
+    proxy_corner:
+      avg_avg_return:                           0.85845524
+      avg_reward_per_step:                      0.096328735
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        125829120
+    env-step-replay-before:                     63799296
+    env-step-used-before:                       125829120
+    ppo-update-before:                          76800
+    env-step-generate-before:                   62029824
+    env-step-all-after:                         125861888
+    env-step-replay-after:                      63832064
+    env-step-used-after:                        125861888
+    ppo-update-after:                           76820
+    env-step-generate-after:                    62029824
+  train-replay:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     70.817
+    avg_avg_return:                             0.37034062
+    avg_reward_per_step:                        0.0095825195
+    proxy_corner:
+      avg_avg_return:                           0.16561192
+      avg_reward_per_step:                      0.0028076172
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     70.817
+    avg_avg_return:                             0.37034062
+    avg_reward_per_step:                        0.0095825195
+    proxy_corner:
+      avg_avg_return:                           0.16561192
+      avg_reward_per_step:                      0.0028076172
+  perf:
+    env_steps_per_second:                       255510.8808389708
+    ppo_updates_per_second:                     42.44224183529659
+    eval-batch-train-duration:                  0.6340275825932622
+    eval-batch-orig-duration:                   0.6404530806466937
+    eval-batch-shift-duration:                  0.6068171393126249
+    eval-batch-tree-duration:                   0.6133364560082555
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.435303
+      cheese-corne_dist_wavg:                   18.04303
+      levels_cheese_is_in_corner_avg:           0.020996094
+      mouse-cheese_dist_finite_avg:             23.72168
+      mouse-cheese_dist_finite_wavg:            10.465382
+      mouse-cheese_dist_solvable_avg:           23.72168
+      mouse-cheese_dist_solvable_wavg:          10.465383
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.62303376
+    visit_patterns:
+      avg_first_visit_time:                     1656.9106
+      avg_last_visit_time:                      1938.2222
+      num_replay_batches:                       1948
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010644369
+    avg_actor_approxkl3:                        0.0009775822
+    avg_actor_clipfrac:                         0.047711182
+    avg_actor_loss:                             -0.0007148452
+    avg_advantage:                              -0.018019387
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.011066862
+    avg_entropy:                                0.27997106
+    avg_grad_norm_pre_clip:                     1.6752671
+    avg_loss:                                   0.0045386157
+    max:
+      max_actor_approxkl1:                      0.003289869
+      max_actor_approxkl3:                      0.001764103
+      max_actor_clipfrac:                       0.072143555
+      max_actor_loss:                           0.0013641566
+      max_advantage:                            0.28064322
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.012434537
+      max_entropy:                              0.30373228
+      max_grad_norm_pre_clip:                   2.6997652
+      max_loss:                                 0.006457188
+    std:
+      std_actor_approxkl1:                      0.00081966264
+      std_actor_approxkl3:                      0.00041091925
+      std_actor_clipfrac:                       0.017789682
+      std_actor_loss:                           0.0011074378
+      std_advantage:                            0.14853235
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0006948734
+      std_entropy:                              0.008430351
+      std_grad_norm_pre_clip:                   0.5384081
+      std_loss:                                 0.0011468948
+  eval-batch-train:
+    avg_avg_episode_length:                     23.001701
+    avg_avg_return:                             0.88125014
+    avg_benchmark_regret:                       0.105966926
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.120780945
+    proxy_corner:
+      avg_avg_return:                           0.07237037
+      avg_reward_per_step:                      0.0053634644
+  eval-batch-orig:
+    avg_avg_episode_length:                     27.389675
+    avg_avg_return:                             0.8502465
+    avg_benchmark_regret:                       0.13426504
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.0816803
+    proxy_corner:
+      avg_avg_return:                           0.8502465
+      avg_reward_per_step:                      0.0816803
+  eval-batch-shift:
+    avg_avg_episode_length:                     21.463598
+    avg_avg_return:                             0.8938387
+    avg_benchmark_regret:                       0.093667276
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13733673
+    proxy_corner:
+      avg_avg_return:                           0.06352181
+      avg_reward_per_step:                      0.0031280518
+  eval-batch-tree:
+    avg_avg_episode_length:                     25.86799
+    avg_avg_return:                             0.8580156
+    avg_benchmark_regret:                       0.12661953
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09700012
+    proxy_corner:
+      avg_avg_return:                           0.8580156
+      avg_reward_per_step:                      0.09700012
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        126877696
+    env-step-replay-before:                     64356352
+    env-step-used-before:                       126877696
+    ppo-update-before:                          77440
+    env-step-generate-before:                   62521344
+    env-step-all-after:                         126910464
+    env-step-replay-after:                      64356352
+    env-step-used-after:                        126910464
+    ppo-update-after:                           77460
+    env-step-generate-after:                    62554112
+  train-generate:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     23.52428
+    avg_avg_return:                             0.8144768
+    avg_reward_per_step:                        0.10562134
+    proxy_corner:
+      avg_avg_return:                           0.0670897
+      avg_reward_per_step:                      0.0043029785
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     23.52428
+    avg_avg_return:                             0.8144768
+    avg_reward_per_step:                        0.10562134
+    proxy_corner:
+      avg_avg_return:                           0.0670897
+      avg_reward_per_step:                      0.0043029785
+  perf:
+    env_steps_per_second:                       262335.13535349234
+    ppo_updates_per_second:                     41.4368862493163
+    eval-batch-train-duration:                  0.5518371537327766
+    eval-batch-orig-duration:                   0.40867521800100803
+    eval-batch-shift-duration:                  0.25627207197248936
+    eval-batch-tree-duration:                   0.23898126650601625
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.494873
+      cheese-corne_dist_wavg:                   22.525692
+      levels_cheese_is_in_corner_avg:           0.020263672
+      mouse-cheese_dist_finite_avg:             23.661133
+      mouse-cheese_dist_finite_wavg:            11.368904
+      mouse-cheese_dist_solvable_avg:           23.661133
+      mouse-cheese_dist_solvable_wavg:          11.368905
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.5897435
+    visit_patterns:
+      avg_first_visit_time:                     1669.3264
+      avg_last_visit_time:                      1954.2559
+      num_replay_batches:                       1964
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00069690705
+    avg_actor_approxkl3:                        0.0006743234
+    avg_actor_clipfrac:                         0.027593995
+    avg_actor_loss:                             -0.0009898429
+    avg_advantage:                              0.0068161716
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0025830895
+    avg_entropy:                                0.11709922
+    avg_grad_norm_pre_clip:                     1.1937476
+    avg_loss:                                   0.00018460272
+    max:
+      max_actor_approxkl1:                      0.0015784343
+      max_actor_approxkl3:                      0.0016808405
+      max_actor_clipfrac:                       0.044311523
+      max_actor_loss:                           0.00080310926
+      max_advantage:                            0.15851957
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0046886704
+      max_entropy:                              0.1333277
+      max_grad_norm_pre_clip:                   2.5316312
+      max_loss:                                 0.0022287907
+    std:
+      std_actor_approxkl1:                      0.00043435444
+      std_actor_approxkl3:                      0.00038454312
+      std_actor_clipfrac:                       0.011273674
+      std_actor_loss:                           0.0010147556
+      std_advantage:                            0.07181718
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00086360995
+      std_entropy:                              0.011308319
+      std_grad_norm_pre_clip:                   0.4828966
+      std_loss:                                 0.0010987549
+  eval-batch-train:
+    avg_avg_episode_length:                     21.75367
+    avg_avg_return:                             0.88784695
+    avg_benchmark_regret:                       0.09937017
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.1242218
+    proxy_corner:
+      avg_avg_return:                           0.06713347
+      avg_reward_per_step:                      0.0050354004
+  eval-batch-orig:
+    avg_avg_episode_length:                     26.726828
+    avg_avg_return:                             0.8721365
+    avg_benchmark_regret:                       0.11237502
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08098602
+    proxy_corner:
+      avg_avg_return:                           0.8721365
+      avg_reward_per_step:                      0.08098602
+  eval-batch-shift:
+    avg_avg_episode_length:                     20.919785
+    avg_avg_return:                             0.8994583
+    avg_benchmark_regret:                       0.08804765
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13807678
+    proxy_corner:
+      avg_avg_return:                           0.0582843
+      avg_reward_per_step:                      0.0030212402
+  eval-batch-tree:
+    avg_avg_episode_length:                     25.466326
+    avg_avg_return:                             0.868356
+    avg_benchmark_regret:                       0.11627919
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.096206665
+    proxy_corner:
+      avg_avg_return:                           0.868356
+      avg_reward_per_step:                      0.096206665
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        127926272
+    env-step-replay-before:                     64946176
+    env-step-used-before:                       127926272
+    ppo-update-before:                          78080
+    env-step-generate-before:                   62980096
+    env-step-all-after:                         127959040
+    env-step-replay-after:                      64946176
+    env-step-used-after:                        127959040
+    ppo-update-after:                           78100
+    env-step-generate-after:                    63012864
+  train-generate:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     25.026543
+    avg_avg_return:                             0.7988469
+    avg_reward_per_step:                        0.131073
+    proxy_corner:
+      avg_avg_return:                           0.050316576
+      avg_reward_per_step:                      0.0030517578
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     25.026543
+    avg_avg_return:                             0.7988469
+    avg_reward_per_step:                        0.131073
+    proxy_corner:
+      avg_avg_return:                           0.050316576
+      avg_reward_per_step:                      0.0030517578
+  perf:
+    env_steps_per_second:                       299831.7945471055
+    ppo_updates_per_second:                     42.28719433106336
+    eval-batch-train-duration:                  0.5389050859957933
+    eval-batch-orig-duration:                   0.4819866716861725
+    eval-batch-shift-duration:                  0.4849441647529602
+    eval-batch-tree-duration:                   0.48724428471177816
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.494385
+      cheese-corne_dist_wavg:                   11.767012
+      levels_cheese_is_in_corner_avg:           0.020996094
+      mouse-cheese_dist_finite_avg:             23.71045
+      mouse-cheese_dist_finite_wavg:            8.677218
+      mouse-cheese_dist_solvable_avg:           23.71045
+      mouse-cheese_dist_solvable_wavg:          8.677219
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5915732
+    visit_patterns:
+      avg_first_visit_time:                     1682.0779
+      avg_last_visit_time:                      1972.2808
+      num_replay_batches:                       1982
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00089743495
+    avg_actor_approxkl3:                        0.00081293844
+    avg_actor_clipfrac:                         0.029992675
+    avg_actor_loss:                             -0.00052528793
+    avg_advantage:                              0.027746907
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.003210457
+    avg_entropy:                                0.12036802
+    avg_grad_norm_pre_clip:                     1.4427996
+    avg_loss:                                   0.0009595725
+    max:
+      max_actor_approxkl1:                      0.002033691
+      max_actor_approxkl3:                      0.0019517876
+      max_actor_clipfrac:                       0.0546875
+      max_actor_loss:                           0.0008115247
+      max_advantage:                            0.18123579
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.004233785
+      max_entropy:                              0.13962269
+      max_grad_norm_pre_clip:                   2.5525303
+      max_loss:                                 0.0026269916
+    std:
+      std_actor_approxkl1:                      0.00066927384
+      std_actor_approxkl3:                      0.00052534416
+      std_actor_clipfrac:                       0.014362344
+      std_actor_loss:                           0.00071731495
+      std_advantage:                            0.079085395
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0005936177
+      std_entropy:                              0.011475175
+      std_grad_norm_pre_clip:                   0.48097947
+      std_loss:                                 0.00071303325
+  eval-batch-train:
+    avg_avg_episode_length:                     20.429596
+    avg_avg_return:                             0.9014628
+    avg_benchmark_regret:                       0.085754275
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12525177
+    proxy_corner:
+      avg_avg_return:                           0.048999645
+      avg_reward_per_step:                      0.0047454834
+  eval-batch-orig:
+    avg_avg_episode_length:                     27.656616
+    avg_avg_return:                             0.8612495
+    avg_benchmark_regret:                       0.12326203
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08036804
+    proxy_corner:
+      avg_avg_return:                           0.8612495
+      avg_reward_per_step:                      0.08036804
+  eval-batch-shift:
+    avg_avg_episode_length:                     21.618195
+    avg_avg_return:                             0.887464
+    avg_benchmark_regret:                       0.100041956
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13871002
+    proxy_corner:
+      avg_avg_return:                           0.048692666
+      avg_reward_per_step:                      0.0026245117
+  eval-batch-tree:
+    avg_avg_episode_length:                     25.2686
+    avg_avg_return:                             0.87199396
+    avg_benchmark_regret:                       0.112641275
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09577942
+    proxy_corner:
+      avg_avg_return:                           0.87199396
+      avg_reward_per_step:                      0.09577942
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        128974848
+    env-step-replay-before:                     65601536
+    env-step-used-before:                       128974848
+    ppo-update-before:                          78720
+    env-step-generate-before:                   63373312
+    env-step-all-after:                         129007616
+    env-step-replay-after:                      65601536
+    env-step-used-after:                        129007616
+    ppo-update-after:                           78740
+    env-step-generate-after:                    63406080
+  train-generate:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     24.482464
+    avg_avg_return:                             0.8028663
+    avg_reward_per_step:                        0.10827637
+    proxy_corner:
+      avg_avg_return:                           0.07478839
+      avg_reward_per_step:                      0.0046691895
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     24.482464
+    avg_avg_return:                             0.8028663
+    avg_reward_per_step:                        0.10827637
+    proxy_corner:
+      avg_avg_return:                           0.07478839
+      avg_reward_per_step:                      0.0046691895
+  perf:
+    env_steps_per_second:                       320557.49527302047
+    ppo_updates_per_second:                     42.14082400677354
+    eval-batch-train-duration:                  0.539199385792017
+    eval-batch-orig-duration:                   0.49573954101651907
+    eval-batch-shift-duration:                  0.41759804356843233
+    eval-batch-tree-duration:                   0.2580119203776121
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.516113
+      cheese-corne_dist_wavg:                   18.04624
+      levels_cheese_is_in_corner_avg:           0.020019531
+      mouse-cheese_dist_finite_avg:             23.729248
+      mouse-cheese_dist_finite_wavg:            10.4730835
+      mouse-cheese_dist_solvable_avg:           23.729248
+      mouse-cheese_dist_solvable_wavg:          10.473083
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.58338106
+    visit_patterns:
+      avg_first_visit_time:                     1694.6602
+      avg_last_visit_time:                      1992.3125
+      num_replay_batches:                       2002
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00026817547
+    avg_actor_approxkl3:                        0.0006258878
+    avg_actor_clipfrac:                         0.026220703
+    avg_actor_loss:                             -0.0010126213
+    avg_advantage:                              -0.0023230955
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0028118526
+    avg_entropy:                                0.11857983
+    avg_grad_norm_pre_clip:                     0.9905428
+    avg_loss:                                   0.0002747254
+    max:
+      max_actor_approxkl1:                      0.0013197099
+      max_actor_approxkl3:                      0.0011480248
+      max_actor_clipfrac:                       0.037841797
+      max_actor_loss:                           0.00059706345
+      max_advantage:                            0.19984633
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0040299855
+      max_entropy:                              0.13419934
+      max_grad_norm_pre_clip:                   1.481012
+      max_loss:                                 0.0017595747
+    std:
+      std_actor_approxkl1:                      0.00046492086
+      std_actor_approxkl3:                      0.00025746512
+      std_actor_clipfrac:                       0.008882878
+      std_actor_loss:                           0.00095338246
+      std_advantage:                            0.07563552
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0007044306
+      std_entropy:                              0.009537326
+      std_grad_norm_pre_clip:                   0.31582665
+      std_loss:                                 0.00093439204
+  eval-batch-train:
+    avg_avg_episode_length:                     20.52289
+    avg_avg_return:                             0.902721
+    avg_benchmark_regret:                       0.084496066
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12515259
+    proxy_corner:
+      avg_avg_return:                           0.06413594
+      avg_reward_per_step:                      0.005264282
+  eval-batch-orig:
+    avg_avg_episode_length:                     26.34902
+    avg_avg_return:                             0.8661158
+    avg_benchmark_regret:                       0.11839572
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08219147
+    proxy_corner:
+      avg_avg_return:                           0.8661158
+      avg_reward_per_step:                      0.08219147
+  eval-batch-shift:
+    avg_avg_episode_length:                     20.615845
+    avg_avg_return:                             0.90503794
+    avg_benchmark_regret:                       0.08246802
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.140419
+    proxy_corner:
+      avg_avg_return:                           0.060438752
+      avg_reward_per_step:                      0.0030288696
+  eval-batch-tree:
+    avg_avg_episode_length:                     25.56539
+    avg_avg_return:                             0.870964
+    avg_benchmark_regret:                       0.11367118
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0965271
+    proxy_corner:
+      avg_avg_return:                           0.870964
+      avg_reward_per_step:                      0.0965271
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        130023424
+    env-step-replay-before:                     66158592
+    env-step-used-before:                       130023424
+    ppo-update-before:                          79360
+    env-step-generate-before:                   63864832
+    env-step-all-after:                         130056192
+    env-step-replay-after:                      66158592
+    env-step-used-after:                        130056192
+    ppo-update-after:                           79380
+    env-step-generate-after:                    63897600
+  train-generate:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     25.097229
+    avg_avg_return:                             0.80892885
+    avg_reward_per_step:                        0.11306763
+    proxy_corner:
+      avg_avg_return:                           0.07568698
+      avg_reward_per_step:                      0.004486084
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     25.097229
+    avg_avg_return:                             0.80892885
+    avg_reward_per_step:                        0.11306763
+    proxy_corner:
+      avg_avg_return:                           0.07568698
+      avg_reward_per_step:                      0.004486084
+  perf:
+    env_steps_per_second:                       299157.5348060253
+    ppo_updates_per_second:                     41.982601632270566
+    eval-batch-train-duration:                  0.5338626271113753
+    eval-batch-orig-duration:                   0.48129566200077534
+    eval-batch-shift-duration:                  0.47732672560960054
+    eval-batch-tree-duration:                   0.45367405097931623
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.403564
+      cheese-corne_dist_wavg:                   18.046131
+      levels_cheese_is_in_corner_avg:           0.021728516
+      mouse-cheese_dist_finite_avg:             23.739746
+      mouse-cheese_dist_finite_wavg:            10.477627
+      mouse-cheese_dist_solvable_avg:           23.739746
+      mouse-cheese_dist_solvable_wavg:          10.477628
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.57702905
+    visit_patterns:
+      avg_first_visit_time:                     1708.2295
+      avg_last_visit_time:                      2009.2615
+      num_replay_batches:                       2019
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0007131047
+    avg_actor_approxkl3:                        0.00067468727
+    avg_actor_clipfrac:                         0.025079345
+    avg_actor_loss:                             -0.000776694
+    avg_advantage:                              0.00022785622
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0021352617
+    avg_entropy:                                0.1086715
+    avg_grad_norm_pre_clip:                     1.1549594
+    avg_loss:                                   0.00018226523
+    max:
+      max_actor_approxkl1:                      0.0017584837
+      max_actor_approxkl3:                      0.0012614045
+      max_actor_clipfrac:                       0.036010742
+      max_actor_loss:                           0.00078303367
+      max_advantage:                            0.19586807
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0035280145
+      max_entropy:                              0.12570623
+      max_grad_norm_pre_clip:                   1.7202481
+      max_loss:                                 0.0023827069
+    std:
+      std_actor_approxkl1:                      0.0004080831
+      std_actor_approxkl3:                      0.00028565823
+      std_actor_clipfrac:                       0.008816357
+      std_actor_loss:                           0.0010150594
+      std_advantage:                            0.0657873
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0006866546
+      std_entropy:                              0.008387411
+      std_grad_norm_pre_clip:                   0.3097233
+      std_loss:                                 0.0010197633
+  eval-batch-train:
+    avg_avg_episode_length:                     20.677212
+    avg_avg_return:                             0.90505624
+    avg_benchmark_regret:                       0.08216081
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.123924255
+    proxy_corner:
+      avg_avg_return:                           0.058137737
+      avg_reward_per_step:                      0.004852295
+  eval-batch-orig:
+    avg_avg_episode_length:                     25.423878
+    avg_avg_return:                             0.8743696
+    avg_benchmark_regret:                       0.110141896
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.0830307
+    proxy_corner:
+      avg_avg_return:                           0.8743696
+      avg_reward_per_step:                      0.0830307
+  eval-batch-shift:
+    avg_avg_episode_length:                     20.563848
+    avg_avg_return:                             0.90744567
+    avg_benchmark_regret:                       0.08006032
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14253998
+    proxy_corner:
+      avg_avg_return:                           0.050113115
+      avg_reward_per_step:                      0.002960205
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.963364
+    avg_avg_return:                             0.8790544
+    avg_benchmark_regret:                       0.10558081
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09764099
+    proxy_corner:
+      avg_avg_return:                           0.8790544
+      avg_reward_per_step:                      0.09764099
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        131072000
+    env-step-replay-before:                     66551808
+    env-step-used-before:                       131072000
+    ppo-update-before:                          80000
+    env-step-generate-before:                   64520192
+    env-step-all-after:                         131104768
+    env-step-replay-after:                      66551808
+    env-step-used-after:                        131104768
+    ppo-update-after:                           80020
+    env-step-generate-after:                    64552960
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     23.988667
+    avg_avg_return:                             0.8090273
+    avg_reward_per_step:                        0.098236084
+    proxy_corner:
+      avg_avg_return:                           0.025006019
+      avg_reward_per_step:                      0.0013122559
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     23.988667
+    avg_avg_return:                             0.8090273
+    avg_reward_per_step:                        0.098236084
+    proxy_corner:
+      avg_avg_return:                           0.025006019
+      avg_reward_per_step:                      0.0013122559
+  perf:
+    env_steps_per_second:                       314890.38624267903
+    ppo_updates_per_second:                     42.2174280508206
+    eval-batch-train-duration:                  0.5310891596600413
+    eval-batch-orig-duration:                   0.4885005671530962
+    eval-batch-shift-duration:                  0.48404452484101057
+    eval-batch-tree-duration:                   0.4865692788735032
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.332764
+      cheese-corne_dist_wavg:                   18.04003
+      levels_cheese_is_in_corner_avg:           0.021240234
+      mouse-cheese_dist_finite_avg:             23.700928
+      mouse-cheese_dist_finite_wavg:            10.478103
+      mouse-cheese_dist_solvable_avg:           23.700928
+      mouse-cheese_dist_solvable_wavg:          10.478104
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.585172
+    visit_patterns:
+      avg_first_visit_time:                     1726.1145
+      avg_last_visit_time:                      2021.291
+      num_replay_batches:                       2031
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00091397186
+    avg_actor_approxkl3:                        0.00071155146
+    avg_actor_clipfrac:                         0.02420044
+    avg_actor_loss:                             -0.0005105722
+    avg_advantage:                              -0.0045997407
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0026001714
+    avg_entropy:                                0.099615686
+    avg_grad_norm_pre_clip:                     1.2502327
+    avg_loss:                                   0.0006898979
+    max:
+      max_actor_approxkl1:                      0.00229694
+      max_actor_approxkl3:                      0.0016433389
+      max_actor_clipfrac:                       0.041381836
+      max_actor_loss:                           0.0023256876
+      max_advantage:                            0.157121
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.006215262
+      max_entropy:                              0.11920512
+      max_grad_norm_pre_clip:                   3.1867938
+      max_loss:                                 0.0028470166
+    std:
+      std_actor_approxkl1:                      0.0006628283
+      std_actor_approxkl3:                      0.00046884757
+      std_actor_clipfrac:                       0.011047119
+      std_actor_loss:                           0.0009420983
+      std_advantage:                            0.0722527
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011351261
+      std_entropy:                              0.010554098
+      std_grad_norm_pre_clip:                   0.6022027
+      std_loss:                                 0.00091999455
+  eval-batch-train:
+    avg_avg_episode_length:                     20.72745
+    avg_avg_return:                             0.89924294
+    avg_benchmark_regret:                       0.0879741
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12543488
+    proxy_corner:
+      avg_avg_return:                           0.050049204
+      avg_reward_per_step:                      0.0048675537
+  eval-batch-orig:
+    avg_avg_episode_length:                     25.340794
+    avg_avg_return:                             0.87835765
+    avg_benchmark_regret:                       0.106153876
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08148193
+    proxy_corner:
+      avg_avg_return:                           0.87835765
+      avg_reward_per_step:                      0.08148193
+  eval-batch-shift:
+    avg_avg_episode_length:                     21.060522
+    avg_avg_return:                             0.89918053
+    avg_benchmark_regret:                       0.08832541
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14126587
+    proxy_corner:
+      avg_avg_return:                           0.047139592
+      avg_reward_per_step:                      0.0027542114
+  eval-batch-tree:
+    avg_avg_episode_length:                     24.097853
+    avg_avg_return:                             0.8860307
+    avg_benchmark_regret:                       0.0986045
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09654236
+    proxy_corner:
+      avg_avg_return:                           0.8860307
+      avg_reward_per_step:                      0.09654236
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        132120576
+    env-step-replay-before:                     67076096
+    env-step-used-before:                       132120576
+    ppo-update-before:                          80640
+    env-step-generate-before:                   65044480
+    env-step-all-after:                         132153344
+    env-step-replay-after:                      67076096
+    env-step-used-after:                        132153344
+    ppo-update-after:                           80660
+    env-step-generate-after:                    65077248
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     27.906136
+    avg_avg_return:                             0.76922107
+    avg_reward_per_step:                        0.10324097
+    proxy_corner:
+      avg_avg_return:                           0.058368437
+      avg_reward_per_step:                      0.0018005371
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     27.906136
+    avg_avg_return:                             0.76922107
+    avg_reward_per_step:                        0.10324097
+    proxy_corner:
+      avg_avg_return:                           0.058368437
+      avg_reward_per_step:                      0.0018005371
+  perf:
+    env_steps_per_second:                       294587.10039664013
+    ppo_updates_per_second:                     42.718199610969606
+    eval-batch-train-duration:                  0.532395800575614
+    eval-batch-orig-duration:                   0.48584558349102736
+    eval-batch-shift-duration:                  0.4680952671915293
+    eval-batch-tree-duration:                   0.4546399740502238
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.32666
+      cheese-corne_dist_wavg:                   9.961552
+      levels_cheese_is_in_corner_avg:           0.022460938
+      mouse-cheese_dist_finite_avg:             23.791504
+      mouse-cheese_dist_finite_wavg:            8.667301
+      mouse-cheese_dist_solvable_avg:           23.791504
+      mouse-cheese_dist_solvable_wavg:          8.667301
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.5710668
+    visit_patterns:
+      avg_first_visit_time:                     1738.5771
+      avg_last_visit_time:                      2037.2476
+      num_replay_batches:                       2047
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009718377
+    avg_actor_approxkl3:                        0.0007438901
+    avg_actor_clipfrac:                         0.028326416
+    avg_actor_loss:                             -0.0010062446
+    avg_advantage:                              -0.006530649
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0037230377
+    avg_entropy:                                0.113699675
+    avg_grad_norm_pre_clip:                     1.0648457
+    avg_loss:                                   0.0007415745
+    max:
+      max_actor_approxkl1:                      0.0032541885
+      max_actor_approxkl3:                      0.0014545866
+      max_actor_clipfrac:                       0.044799805
+      max_actor_loss:                           0.00072764605
+      max_advantage:                            0.1740548
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0055052554
+      max_entropy:                              0.13020581
+      max_grad_norm_pre_clip:                   2.058793
+      max_loss:                                 0.0026093444
+    std:
+      std_actor_approxkl1:                      0.0007419307
+      std_actor_approxkl3:                      0.00038202832
+      std_actor_clipfrac:                       0.011254491
+      std_actor_loss:                           0.0009688872
+      std_advantage:                            0.08647113
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010516027
+      std_entropy:                              0.009455014
+      std_grad_norm_pre_clip:                   0.43916205
+      std_loss:                                 0.0010070698
+  eval-batch-train:
+    avg_avg_episode_length:                     20.740208
+    avg_avg_return:                             0.9048945
+    avg_benchmark_regret:                       0.08232261
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12374878
+    proxy_corner:
+      avg_avg_return:                           0.05593247
+      avg_reward_per_step:                      0.0049057007
+  eval-batch-orig:
+    avg_avg_episode_length:                     25.24627
+    avg_avg_return:                             0.8762494
+    avg_benchmark_regret:                       0.108262196
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.082206726
+    proxy_corner:
+      avg_avg_return:                           0.8762494
+      avg_reward_per_step:                      0.082206726
+  eval-batch-shift:
+    avg_avg_episode_length:                     21.014301
+    avg_avg_return:                             0.90547323
+    avg_benchmark_regret:                       0.082032666
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13763428
+    proxy_corner:
+      avg_avg_return:                           0.053696603
+      avg_reward_per_step:                      0.0029296875
+  eval-batch-tree:
+    avg_avg_episode_length:                     24.54277
+    avg_avg_return:                             0.874623
+    avg_benchmark_regret:                       0.11001218
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09716034
+    proxy_corner:
+      avg_avg_return:                           0.874623
+      avg_reward_per_step:                      0.09716034
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        133169152
+    env-step-replay-before:                     67600384
+    env-step-used-before:                       133169152
+    ppo-update-before:                          81280
+    env-step-generate-before:                   65568768
+    env-step-all-after:                         133201920
+    env-step-replay-after:                      67633152
+    env-step-used-after:                        133201920
+    ppo-update-after:                           81300
+    env-step-generate-after:                    65568768
+  train-replay:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     69.20386
+    avg_avg_return:                             0.39709574
+    avg_reward_per_step:                        0.010437012
+    proxy_corner:
+      avg_avg_return:                           0.131156
+      avg_reward_per_step:                      0.0022277832
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     69.20386
+    avg_avg_return:                             0.39709574
+    avg_reward_per_step:                        0.010437012
+    proxy_corner:
+      avg_avg_return:                           0.131156
+      avg_reward_per_step:                      0.0022277832
+  perf:
+    env_steps_per_second:                       317285.5356004106
+    ppo_updates_per_second:                     42.14524142659025
+    eval-batch-train-duration:                  0.521653899922967
+    eval-batch-orig-duration:                   0.47913642786443233
+    eval-batch-shift-duration:                  0.47723789513111115
+    eval-batch-tree-duration:                   0.4912227503955364
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.406494
+      cheese-corne_dist_wavg:                   19.85987
+      levels_cheese_is_in_corner_avg:           0.022216797
+      mouse-cheese_dist_finite_avg:             23.900146
+      mouse-cheese_dist_finite_wavg:            12.303523
+      mouse-cheese_dist_solvable_avg:           23.900146
+      mouse-cheese_dist_solvable_wavg:          12.303525
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5629597
+    visit_patterns:
+      avg_first_visit_time:                     1753.5925
+      avg_last_visit_time:                      2054.284
+      num_replay_batches:                       2064
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001577191
+    avg_actor_approxkl3:                        0.0015253089
+    avg_actor_clipfrac:                         0.06512451
+    avg_actor_loss:                             -8.610636e-05
+    avg_advantage:                              -0.029064313
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.01074703
+    avg_entropy:                                0.24782799
+    avg_grad_norm_pre_clip:                     2.1728508
+    avg_loss:                                   0.0050395816
+    max:
+      max_actor_approxkl1:                      0.0032850248
+      max_actor_approxkl3:                      0.0027219504
+      max_actor_clipfrac:                       0.09838867
+      max_actor_loss:                           0.0025193095
+      max_advantage:                            0.20262504
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.012953815
+      max_entropy:                              0.26532358
+      max_grad_norm_pre_clip:                   3.287009
+      max_loss:                                 0.007720904
+    std:
+      std_actor_approxkl1:                      0.0009952494
+      std_actor_approxkl3:                      0.0006672103
+      std_actor_clipfrac:                       0.023760745
+      std_actor_loss:                           0.001219228
+      std_advantage:                            0.14565673
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0015697526
+      std_entropy:                              0.0067612473
+      std_grad_norm_pre_clip:                   0.6907041
+      std_loss:                                 0.0014772146
+  eval-batch-train:
+    avg_avg_episode_length:                     20.959984
+    avg_avg_return:                             0.90748215
+    avg_benchmark_regret:                       0.07973492
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12438965
+    proxy_corner:
+      avg_avg_return:                           0.052889615
+      avg_reward_per_step:                      0.004386902
+  eval-batch-orig:
+    avg_avg_episode_length:                     26.25072
+    avg_avg_return:                             0.8726933
+    avg_benchmark_regret:                       0.11181821
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.080093384
+    proxy_corner:
+      avg_avg_return:                           0.8726933
+      avg_reward_per_step:                      0.080093384
+  eval-batch-shift:
+    avg_avg_episode_length:                     21.65018
+    avg_avg_return:                             0.89323735
+    avg_benchmark_regret:                       0.094268546
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.13703918
+    proxy_corner:
+      avg_avg_return:                           0.0516984
+      avg_reward_per_step:                      0.0029144287
+  eval-batch-tree:
+    avg_avg_episode_length:                     24.840267
+    avg_avg_return:                             0.87869936
+    avg_benchmark_regret:                       0.105935834
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.094680786
+    proxy_corner:
+      avg_avg_return:                           0.87869936
+      avg_reward_per_step:                      0.094680786
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        134217728
+    env-step-replay-before:                     68157440
+    env-step-used-before:                       134217728
+    ppo-update-before:                          81920
+    env-step-generate-before:                   66060288
+    env-step-all-after:                         134250496
+    env-step-replay-after:                      68157440
+    env-step-used-after:                        134250496
+    ppo-update-after:                           81940
+    env-step-generate-after:                    66093056
+  train-generate:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     22.335697
+    avg_avg_return:                             0.821937
+    avg_reward_per_step:                        0.10839844
+    proxy_corner:
+      avg_avg_return:                           0.07668077
+      avg_reward_per_step:                      0.004333496
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     22.335697
+    avg_avg_return:                             0.821937
+    avg_reward_per_step:                        0.10839844
+    proxy_corner:
+      avg_avg_return:                           0.07668077
+      avg_reward_per_step:                      0.004333496
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  perf:
+    env_steps_per_second:                       551742.9073478302
+    ppo_updates_per_second:                     40.2220315753125
+    eval-batch-train-duration:                  0.44865739345550537
+    eval-batch-orig-duration:                   0.4105298975482583
+    eval-batch-shift-duration:                  0.4426100654527545
+    eval-batch-tree-duration:                   0.42799260560423136
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.435547
+      cheese-corne_dist_wavg:                   18.048082
+      levels_cheese_is_in_corner_avg:           0.021728516
+      mouse-cheese_dist_finite_avg:             23.998047
+      mouse-cheese_dist_finite_wavg:            10.491172
+      mouse-cheese_dist_solvable_avg:           23.998047
+      mouse-cheese_dist_solvable_wavg:          10.491173
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.420168
+    scoring:
+      avg_scores:                               0.5770426
+    visit_patterns:
+      avg_first_visit_time:                     1767.7776
+      avg_last_visit_time:                      2070.2947
+      num_replay_batches:                       2080
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009410821
+    avg_actor_approxkl3:                        0.0006505986
+    avg_actor_clipfrac:                         0.026116943
+    avg_actor_loss:                             -0.0006901249
+    avg_advantage:                              0.038837846
+    avg_critic_clipfrac:                        0.00048217774
+    avg_critic_loss:                            0.0026238821
+    avg_entropy:                                0.10903194
+    avg_grad_norm_pre_clip:                     1.2862364
+    avg_loss:                                   0.0005127842
+    max:
+      max_actor_approxkl1:                      0.0023857467
+      max_actor_approxkl3:                      0.0014049979
+      max_actor_clipfrac:                       0.048950195
+      max_actor_loss:                           0.0007827543
+      max_advantage:                            0.18207622
+      max_critic_clipfrac:                      0.008666992
+      max_critic_loss:                          0.0045401473
+      max_entropy:                              0.1358956
+      max_grad_norm_pre_clip:                   1.9530008
+      max_loss:                                 0.0021530841
+    std:
+      std_actor_approxkl1:                      0.0006814247
+      std_actor_approxkl3:                      0.00036330242
+      std_actor_clipfrac:                       0.011379316
+      std_actor_loss:                           0.0007868358
+      std_advantage:                            0.07069141
+      std_critic_clipfrac:                      0.0018897161
+      std_critic_loss:                          0.00082575704
+      std_entropy:                              0.012185673
+      std_grad_norm_pre_clip:                   0.3722115
+      std_loss:                                 0.0008025659
+  eval-batch-train:
+    avg_avg_episode_length:                     20.740448
+    avg_avg_return:                             0.903632
+    avg_benchmark_regret:                       0.08358508
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12679291
+    proxy_corner:
+      avg_avg_return:                           0.0454051
+      avg_reward_per_step:                      0.004211426
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-orig:
+    avg_avg_episode_length:                     25.785168
+    avg_avg_return:                             0.87530315
+    avg_benchmark_regret:                       0.1092084
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08148193
+    proxy_corner:
+      avg_avg_return:                           0.87530315
+      avg_reward_per_step:                      0.08148193
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-shift:
+    avg_avg_episode_length:                     21.16333
+    avg_avg_return:                             0.9014101
+    avg_benchmark_regret:                       0.08609587
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14264679
+    proxy_corner:
+      avg_avg_return:                           0.04556885
+      avg_reward_per_step:                      0.002670288
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.623625
+    avg_avg_return:                             0.8813256
+    avg_benchmark_regret:                       0.103309534
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09744263
+    proxy_corner:
+      avg_avg_return:                           0.8813256
+      avg_reward_per_step:                      0.09744263
+    rollouts_gif:                               float32(132, 225, 225, 3)
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        135266304
+    env-step-replay-before:                     68780032
+    env-step-used-before:                       135266304
+    ppo-update-before:                          82560
+    env-step-generate-before:                   66486272
+    env-step-all-after:                         135299072
+    env-step-replay-after:                      68780032
+    env-step-used-after:                        135299072
+    ppo-update-after:                           82580
+    env-step-generate-after:                    66519040
+  train-generate:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     21.594233
+    avg_avg_return:                             0.8368187
+    avg_reward_per_step:                        0.123291016
+    proxy_corner:
+      avg_avg_return:                           0.056388605
+      avg_reward_per_step:                      0.0033569336
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     21.594233
+    avg_avg_return:                             0.8368187
+    avg_reward_per_step:                        0.123291016
+    proxy_corner:
+      avg_avg_return:                           0.056388605
+      avg_reward_per_step:                      0.0033569336
+  perf:
+    env_steps_per_second:                       381801.5960387158
+    ppo_updates_per_second:                     43.13117273039269
+    eval-batch-train-duration:                  0.4897619988769293
+    eval-batch-orig-duration:                   0.31269971188157797
+    eval-batch-shift-duration:                  0.21762135066092014
+    eval-batch-tree-duration:                   0.21528931893408298
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.512207
+      cheese-corne_dist_wavg:                   11.737087
+      levels_cheese_is_in_corner_avg:           0.021972656
+      mouse-cheese_dist_finite_avg:             24.031738
+      mouse-cheese_dist_finite_wavg:            11.380697
+      mouse-cheese_dist_solvable_avg:           24.031738
+      mouse-cheese_dist_solvable_wavg:          11.380697
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.58901
+    visit_patterns:
+      avg_first_visit_time:                     1779.9834
+      avg_last_visit_time:                      2089.2314
+      num_replay_batches:                       2099
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0008903079
+    avg_actor_approxkl3:                        0.0006771963
+    avg_actor_clipfrac:                         0.024932861
+    avg_actor_loss:                             -0.00057684485
+    avg_advantage:                              0.019851817
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.001912773
+    avg_entropy:                                0.09755205
+    avg_grad_norm_pre_clip:                     1.2948066
+    avg_loss:                                   0.00028198963
+    max:
+      max_actor_approxkl1:                      0.0026629707
+      max_actor_approxkl3:                      0.0011702728
+      max_actor_clipfrac:                       0.0345459
+      max_actor_loss:                           0.001081869
+      max_advantage:                            0.24858636
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00366648
+      max_entropy:                              0.116213314
+      max_grad_norm_pre_clip:                   1.7968513
+      max_loss:                                 0.0019202537
+    std:
+      std_actor_approxkl1:                      0.0006231899
+      std_actor_approxkl3:                      0.00028236653
+      std_actor_clipfrac:                       0.008338741
+      std_actor_loss:                           0.0008623368
+      std_advantage:                            0.06167188
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00072530826
+      std_entropy:                              0.008885694
+      std_grad_norm_pre_clip:                   0.30815506
+      std_loss:                                 0.0009544381
+  eval-batch-train:
+    avg_avg_episode_length:                     20.650503
+    avg_avg_return:                             0.9034451
+    avg_benchmark_regret:                       0.083771944
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12625885
+    proxy_corner:
+      avg_avg_return:                           0.049809583
+      avg_reward_per_step:                      0.004524231
+  eval-batch-orig:
+    avg_avg_episode_length:                     24.696884
+    avg_avg_return:                             0.8909302
+    avg_benchmark_regret:                       0.09358132
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08382416
+    proxy_corner:
+      avg_avg_return:                           0.8909302
+      avg_reward_per_step:                      0.08382416
+  eval-batch-shift:
+    avg_avg_episode_length:                     21.36324
+    avg_avg_return:                             0.89422494
+    avg_benchmark_regret:                       0.093281
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14217377
+    proxy_corner:
+      avg_avg_return:                           0.051082857
+      avg_reward_per_step:                      0.0028152466
+  eval-batch-tree:
+    avg_avg_episode_length:                     24.315372
+    avg_avg_return:                             0.8793621
+    avg_benchmark_regret:                       0.10527308
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09661102
+    proxy_corner:
+      avg_avg_return:                           0.8793621
+      avg_reward_per_step:                      0.09661102
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        136314880
+    env-step-replay-before:                     69107712
+    env-step-used-before:                       136314880
+    ppo-update-before:                          83200
+    env-step-generate-before:                   67207168
+    env-step-all-after:                         136347648
+    env-step-replay-after:                      69107712
+    env-step-used-after:                        136347648
+    ppo-update-after:                           83220
+    env-step-generate-after:                    67239936
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     26.23928
+    avg_avg_return:                             0.7774971
+    avg_reward_per_step:                        0.10964966
+    proxy_corner:
+      avg_avg_return:                           0.0438279
+      avg_reward_per_step:                      0.0017700195
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     26.23928
+    avg_avg_return:                             0.7774971
+    avg_reward_per_step:                        0.10964966
+    proxy_corner:
+      avg_avg_return:                           0.0438279
+      avg_reward_per_step:                      0.0017700195
+  perf:
+    env_steps_per_second:                       376691.77574764215
+    ppo_updates_per_second:                     41.820660006588376
+    eval-batch-train-duration:                  0.4922214914113283
+    eval-batch-orig-duration:                   0.4261153694242239
+    eval-batch-shift-duration:                  0.4372383886948228
+    eval-batch-tree-duration:                   0.4264845438301563
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.496582
+      cheese-corne_dist_wavg:                   12.661987
+      levels_cheese_is_in_corner_avg:           0.021240234
+      mouse-cheese_dist_finite_avg:             23.991455
+      mouse-cheese_dist_finite_wavg:            8.705341
+      mouse-cheese_dist_solvable_avg:           23.991455
+      mouse-cheese_dist_solvable_wavg:          8.705342
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.5950898
+    visit_patterns:
+      avg_first_visit_time:                     1793.3926
+      avg_last_visit_time:                      2099.3577
+      num_replay_batches:                       2109
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010003429
+    avg_actor_approxkl3:                        0.0010378914
+    avg_actor_clipfrac:                         0.035205077
+    avg_actor_loss:                             -0.00032955315
+    avg_advantage:                              -0.0031648397
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0038022615
+    avg_entropy:                                0.106504634
+    avg_grad_norm_pre_clip:                     1.3524811
+    avg_loss:                                   0.0014650732
+    max:
+      max_actor_approxkl1:                      0.0021916593
+      max_actor_approxkl3:                      0.00172818
+      max_actor_clipfrac:                       0.049316406
+      max_actor_loss:                           0.002878353
+      max_advantage:                            0.15547585
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0049520023
+      max_entropy:                              0.12086718
+      max_grad_norm_pre_clip:                   2.508358
+      max_loss:                                 0.004730384
+    std:
+      std_actor_approxkl1:                      0.00061245845
+      std_actor_approxkl3:                      0.00041210532
+      std_actor_clipfrac:                       0.010667395
+      std_actor_loss:                           0.0012091646
+      std_advantage:                            0.08744377
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0004457858
+      std_entropy:                              0.008342448
+      std_grad_norm_pre_clip:                   0.39595544
+      std_loss:                                 0.0012646333
+  eval-batch-train:
+    avg_avg_episode_length:                     21.065126
+    avg_avg_return:                             0.89120376
+    avg_benchmark_regret:                       0.09601331
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12611389
+    proxy_corner:
+      avg_avg_return:                           0.05117498
+      avg_reward_per_step:                      0.004470825
+  eval-batch-orig:
+    avg_avg_episode_length:                     26.487007
+    avg_avg_return:                             0.8579937
+    avg_benchmark_regret:                       0.12651779
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08288574
+    proxy_corner:
+      avg_avg_return:                           0.8579937
+      avg_reward_per_step:                      0.08288574
+  eval-batch-shift:
+    avg_avg_episode_length:                     20.936357
+    avg_avg_return:                             0.8963032
+    avg_benchmark_regret:                       0.09120279
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14231873
+    proxy_corner:
+      avg_avg_return:                           0.05327999
+      avg_reward_per_step:                      0.0029067993
+  eval-batch-tree:
+    avg_avg_episode_length:                     24.984352
+    avg_avg_return:                             0.8664941
+    avg_benchmark_regret:                       0.118141115
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09790039
+    proxy_corner:
+      avg_avg_return:                           0.8664941
+      avg_reward_per_step:                      0.09790039
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        137363456
+    env-step-replay-before:                     69664768
+    env-step-used-before:                       137363456
+    ppo-update-before:                          83840
+    env-step-generate-before:                   67698688
+    env-step-all-after:                         137396224
+    env-step-replay-after:                      69697536
+    env-step-used-after:                        137396224
+    ppo-update-after:                           83860
+    env-step-generate-after:                    67698688
+  train-replay:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     67.928474
+    avg_avg_return:                             0.38930368
+    avg_reward_per_step:                        0.010925293
+    proxy_corner:
+      avg_avg_return:                           0.10177728
+      avg_reward_per_step:                      0.0016479492
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     67.928474
+    avg_avg_return:                             0.38930368
+    avg_reward_per_step:                        0.010925293
+    proxy_corner:
+      avg_avg_return:                           0.10177728
+      avg_reward_per_step:                      0.0016479492
+  perf:
+    env_steps_per_second:                       322242.73524361127
+    ppo_updates_per_second:                     41.65407381204635
+    eval-batch-train-duration:                  0.4969426728785038
+    eval-batch-orig-duration:                   0.4766072053462267
+    eval-batch-shift-duration:                  0.45548941288143396
+    eval-batch-tree-duration:                   0.474265418946743
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.497803
+      cheese-corne_dist_wavg:                   16.265858
+      levels_cheese_is_in_corner_avg:           0.022216797
+      mouse-cheese_dist_finite_avg:             23.930908
+      mouse-cheese_dist_finite_wavg:            9.603448
+      mouse-cheese_dist_solvable_avg:           23.930908
+      mouse-cheese_dist_solvable_wavg:          9.60345
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201682
+    scoring:
+      avg_scores:                               0.5919635
+    visit_patterns:
+      avg_first_visit_time:                     1807.5693
+      avg_last_visit_time:                      2117.312
+      num_replay_batches:                       2127
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00070660963
+    avg_actor_approxkl3:                        0.00089772895
+    avg_actor_clipfrac:                         0.044244386
+    avg_actor_loss:                             -0.0009724401
+    avg_advantage:                              -0.0441124
+    avg_critic_clipfrac:                        0.000970459
+    avg_critic_loss:                            0.011790695
+    avg_entropy:                                0.24546178
+    avg_grad_norm_pre_clip:                     1.6298965
+    avg_loss:                                   0.0046774456
+    max:
+      max_actor_approxkl1:                      0.002505133
+      max_actor_approxkl3:                      0.0016552676
+      max_actor_clipfrac:                       0.06726074
+      max_actor_loss:                           0.00072851777
+      max_advantage:                            0.16481143
+      max_critic_clipfrac:                      0.008178711
+      max_critic_loss:                          0.013747713
+      max_entropy:                              0.26869255
+      max_grad_norm_pre_clip:                   2.3361208
+      max_loss:                                 0.006619185
+    std:
+      std_actor_approxkl1:                      0.00059910415
+      std_actor_approxkl3:                      0.0003470253
+      std_actor_clipfrac:                       0.0150194345
+      std_actor_loss:                           0.0009871203
+      std_advantage:                            0.15163723
+      std_critic_clipfrac:                      0.002187572
+      std_critic_loss:                          0.0012570714
+      std_entropy:                              0.010010309
+      std_grad_norm_pre_clip:                   0.35675654
+      std_loss:                                 0.0011412934
+  eval-batch-train:
+    avg_avg_episode_length:                     20.388826
+    avg_avg_return:                             0.9098786
+    avg_benchmark_regret:                       0.07733844
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12640381
+    proxy_corner:
+      avg_avg_return:                           0.05308573
+      avg_reward_per_step:                      0.00466156
+  eval-batch-orig:
+    avg_avg_episode_length:                     26.27402
+    avg_avg_return:                             0.866965
+    avg_benchmark_regret:                       0.117546536
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08054352
+    proxy_corner:
+      avg_avg_return:                           0.866965
+      avg_reward_per_step:                      0.08054352
+  eval-batch-shift:
+    avg_avg_episode_length:                     21.309917
+    avg_avg_return:                             0.8933418
+    avg_benchmark_regret:                       0.094164185
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14141846
+    proxy_corner:
+      avg_avg_return:                           0.05688017
+      avg_reward_per_step:                      0.0030822754
+  eval-batch-tree:
+    avg_avg_episode_length:                     25.07127
+    avg_avg_return:                             0.8728036
+    avg_benchmark_regret:                       0.11183152
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09525299
+    proxy_corner:
+      avg_avg_return:                           0.8728036
+      avg_reward_per_step:                      0.09525299
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        138412032
+    env-step-replay-before:                     70090752
+    env-step-used-before:                       138412032
+    ppo-update-before:                          84480
+    env-step-generate-before:                   68321280
+    env-step-all-after:                         138444800
+    env-step-replay-after:                      70123520
+    env-step-used-after:                        138444800
+    ppo-update-after:                           84500
+    env-step-generate-after:                    68321280
+  train-replay:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     69.862366
+    avg_avg_return:                             0.36770236
+    avg_reward_per_step:                        0.010101318
+    proxy_corner:
+      avg_avg_return:                           0.1267092
+      avg_reward_per_step:                      0.0019226074
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     69.862366
+    avg_avg_return:                             0.36770236
+    avg_reward_per_step:                        0.010101318
+    proxy_corner:
+      avg_avg_return:                           0.1267092
+      avg_reward_per_step:                      0.0019226074
+  perf:
+    env_steps_per_second:                       255895.10617372344
+    ppo_updates_per_second:                     41.81925704111421
+    eval-batch-train-duration:                  0.6798964468762279
+    eval-batch-orig-duration:                   0.6041942974552512
+    eval-batch-shift-duration:                  0.6160120964050293
+    eval-batch-tree-duration:                   0.5971605367958546
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.51123
+      cheese-corne_dist_wavg:                   12.6373825
+      levels_cheese_is_in_corner_avg:           0.022949219
+      mouse-cheese_dist_finite_avg:             23.89209
+      mouse-cheese_dist_finite_wavg:            8.678216
+      mouse-cheese_dist_solvable_avg:           23.89209
+      mouse-cheese_dist_solvable_wavg:          8.678216
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.5929676
+    visit_patterns:
+      avg_first_visit_time:                     1823.2793
+      avg_last_visit_time:                      2130.402
+      num_replay_batches:                       2140
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011944455
+    avg_actor_approxkl3:                        0.0010513853
+    avg_actor_clipfrac:                         0.04706421
+    avg_actor_loss:                             -0.0009668559
+    avg_advantage:                              -0.029734049
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.011528055
+    avg_entropy:                                0.2491194
+    avg_grad_norm_pre_clip:                     1.8707308
+    avg_loss:                                   0.004548052
+    max:
+      max_actor_approxkl1:                      0.003471979
+      max_actor_approxkl3:                      0.002064783
+      max_actor_clipfrac:                       0.08459473
+      max_actor_loss:                           0.00033015013
+      max_advantage:                            0.23004133
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.013675338
+      max_entropy:                              0.2659703
+      max_grad_norm_pre_clip:                   3.0183775
+      max_loss:                                 0.0060297535
+    std:
+      std_actor_approxkl1:                      0.0008373957
+      std_actor_approxkl3:                      0.00060627406
+      std_actor_clipfrac:                       0.023820529
+      std_actor_loss:                           0.0007001981
+      std_advantage:                            0.15112054
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010002108
+      std_entropy:                              0.008277583
+      std_grad_norm_pre_clip:                   0.7365366
+      std_loss:                                 0.00087197125
+  eval-batch-train:
+    avg_avg_episode_length:                     21.04348
+    avg_avg_return:                             0.8922986
+    avg_benchmark_regret:                       0.09491849
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12589264
+    proxy_corner:
+      avg_avg_return:                           0.06077124
+      avg_reward_per_step:                      0.0049591064
+  eval-batch-orig:
+    avg_avg_episode_length:                     25.443405
+    avg_avg_return:                             0.8720325
+    avg_benchmark_regret:                       0.11247905
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08459473
+    proxy_corner:
+      avg_avg_return:                           0.8720325
+      avg_reward_per_step:                      0.08459473
+  eval-batch-shift:
+    avg_avg_episode_length:                     19.918537
+    avg_avg_return:                             0.9072989
+    avg_benchmark_regret:                       0.08020702
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14219666
+    proxy_corner:
+      avg_avg_return:                           0.059241943
+      avg_reward_per_step:                      0.0030212402
+  eval-batch-tree:
+    avg_avg_episode_length:                     24.072699
+    avg_avg_return:                             0.87693524
+    avg_benchmark_regret:                       0.107699975
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09942627
+    proxy_corner:
+      avg_avg_return:                           0.87693524
+      avg_reward_per_step:                      0.09942627
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        139460608
+    env-step-replay-before:                     70713344
+    env-step-used-before:                       139460608
+    ppo-update-before:                          85120
+    env-step-generate-before:                   68747264
+    env-step-all-after:                         139493376
+    env-step-replay-after:                      70713344
+    env-step-used-after:                        139493376
+    ppo-update-after:                           85140
+    env-step-generate-after:                    68780032
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     24.534855
+    avg_avg_return:                             0.80232173
+    avg_reward_per_step:                        0.11300659
+    proxy_corner:
+      avg_avg_return:                           0.05381225
+      avg_reward_per_step:                      0.0032348633
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     24.534855
+    avg_avg_return:                             0.80232173
+    avg_reward_per_step:                        0.11300659
+    proxy_corner:
+      avg_avg_return:                           0.05381225
+      avg_reward_per_step:                      0.0032348633
+  perf:
+    env_steps_per_second:                       246189.99637171673
+    ppo_updates_per_second:                     41.71874851900957
+    eval-batch-train-duration:                  0.5271505359560251
+    eval-batch-orig-duration:                   0.48028395138680935
+    eval-batch-shift-duration:                  0.4519617008045316
+    eval-batch-tree-duration:                   0.47490771021693945
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.526123
+      cheese-corne_dist_wavg:                   27.055077
+      levels_cheese_is_in_corner_avg:           0.021972656
+      mouse-cheese_dist_finite_avg:             24.033203
+      mouse-cheese_dist_finite_wavg:            12.293057
+      mouse-cheese_dist_solvable_avg:           24.033203
+      mouse-cheese_dist_solvable_wavg:          12.293059
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999985
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.59215766
+    visit_patterns:
+      avg_first_visit_time:                     1839.0947
+      avg_last_visit_time:                      2148.1763
+      num_replay_batches:                       2158
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011954444
+    avg_actor_approxkl3:                        0.0008171694
+    avg_actor_clipfrac:                         0.028619384
+    avg_actor_loss:                             -0.0008528374
+    avg_advantage:                              0.029936546
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0030103764
+    avg_entropy:                                0.1086928
+    avg_grad_norm_pre_clip:                     0.936621
+    avg_loss:                                   0.0005436581
+    max:
+      max_actor_approxkl1:                      0.0026377416
+      max_actor_approxkl3:                      0.0015265906
+      max_actor_clipfrac:                       0.045288086
+      max_actor_loss:                           0.00055876747
+      max_advantage:                            0.1832273
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0048274226
+      max_entropy:                              0.12181768
+      max_grad_norm_pre_clip:                   1.6039158
+      max_loss:                                 0.002065233
+    std:
+      std_actor_approxkl1:                      0.0007885295
+      std_actor_approxkl3:                      0.00042052224
+      std_actor_clipfrac:                       0.012260168
+      std_actor_loss:                           0.0010057772
+      std_advantage:                            0.076123506
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0008249688
+      std_entropy:                              0.010028264
+      std_grad_norm_pre_clip:                   0.27156514
+      std_loss:                                 0.0010334974
+  eval-batch-train:
+    avg_avg_episode_length:                     20.466484
+    avg_avg_return:                             0.9040757
+    avg_benchmark_regret:                       0.08314137
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12548828
+    proxy_corner:
+      avg_avg_return:                           0.055303585
+      avg_reward_per_step:                      0.0048294067
+  eval-batch-orig:
+    avg_avg_episode_length:                     25.379683
+    avg_avg_return:                             0.8807125
+    avg_benchmark_regret:                       0.103799075
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08228302
+    proxy_corner:
+      avg_avg_return:                           0.8807125
+      avg_reward_per_step:                      0.08228302
+  eval-batch-shift:
+    avg_avg_episode_length:                     20.079056
+    avg_avg_return:                             0.90943074
+    avg_benchmark_regret:                       0.07807527
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14172363
+    proxy_corner:
+      avg_avg_return:                           0.05398896
+      avg_reward_per_step:                      0.0029449463
+  eval-batch-tree:
+    avg_avg_episode_length:                     24.334976
+    avg_avg_return:                             0.87714374
+    avg_benchmark_regret:                       0.10749146
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.0965271
+    proxy_corner:
+      avg_avg_return:                           0.87714374
+      avg_reward_per_step:                      0.0965271
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        140509184
+    env-step-replay-before:                     71237632
+    env-step-used-before:                       140509184
+    ppo-update-before:                          85760
+    env-step-generate-before:                   69271552
+    env-step-all-after:                         140541952
+    env-step-replay-after:                      71270400
+    env-step-used-after:                        140541952
+    ppo-update-after:                           85780
+    env-step-generate-after:                    69271552
+  train-replay:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     70.33798
+    avg_avg_return:                             0.37391308
+    avg_reward_per_step:                        0.009979248
+    proxy_corner:
+      avg_avg_return:                           0.10032361
+      avg_reward_per_step:                      0.0018005371
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     70.33798
+    avg_avg_return:                             0.37391308
+    avg_reward_per_step:                        0.009979248
+    proxy_corner:
+      avg_avg_return:                           0.10032361
+      avg_reward_per_step:                      0.0018005371
+  perf:
+    env_steps_per_second:                       316929.2140252305
+    ppo_updates_per_second:                     42.198313753939566
+    eval-batch-train-duration:                  0.5329024493694305
+    eval-batch-orig-duration:                   0.4786955872550607
+    eval-batch-shift-duration:                  0.4742724234238267
+    eval-batch-tree-duration:                   0.45404320396482944
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.590332
+      cheese-corne_dist_wavg:                   31.55585
+      levels_cheese_is_in_corner_avg:           0.021240234
+      mouse-cheese_dist_finite_avg:             24.066406
+      mouse-cheese_dist_finite_wavg:            11.399343
+      mouse-cheese_dist_solvable_avg:           24.066406
+      mouse-cheese_dist_solvable_wavg:          11.3993435
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.5861895
+    visit_patterns:
+      avg_first_visit_time:                     1851.1267
+      avg_last_visit_time:                      2165.155
+      num_replay_batches:                       2175
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00090297294
+    avg_actor_approxkl3:                        0.00093336776
+    avg_actor_clipfrac:                         0.045837402
+    avg_actor_loss:                             -0.0014934853
+    avg_advantage:                              -0.04612065
+    avg_critic_clipfrac:                        0.0030578615
+    avg_critic_loss:                            0.012061317
+    avg_entropy:                                0.25672036
+    avg_grad_norm_pre_clip:                     1.4146719
+    avg_loss:                                   0.0042804535
+    max:
+      max_actor_approxkl1:                      0.0020657028
+      max_actor_approxkl3:                      0.0014394425
+      max_actor_clipfrac:                       0.07287598
+      max_actor_loss:                           0.00030224025
+      max_advantage:                            0.18447173
+      max_critic_clipfrac:                      0.02758789
+      max_critic_loss:                          0.014174908
+      max_entropy:                              0.2679968
+      max_grad_norm_pre_clip:                   2.8844676
+      max_loss:                                 0.007139702
+    std:
+      std_actor_approxkl1:                      0.00054991816
+      std_actor_approxkl3:                      0.00039407075
+      std_actor_clipfrac:                       0.01925055
+      std_actor_loss:                           0.0011717044
+      std_advantage:                            0.15361685
+      std_critic_clipfrac:                      0.0071846927
+      std_critic_loss:                          0.0009724404
+      std_entropy:                              0.0067228503
+      std_grad_norm_pre_clip:                   0.48466048
+      std_loss:                                 0.0013974253
+  eval-batch-train:
+    avg_avg_episode_length:                     20.23765
+    avg_avg_return:                             0.90529025
+    avg_benchmark_regret:                       0.08192688
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12684631
+    proxy_corner:
+      avg_avg_return:                           0.048730537
+      avg_reward_per_step:                      0.0046463013
+  eval-batch-orig:
+    avg_avg_episode_length:                     25.227058
+    avg_avg_return:                             0.87847763
+    avg_benchmark_regret:                       0.10603391
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08119202
+    proxy_corner:
+      avg_avg_return:                           0.87847763
+      avg_reward_per_step:                      0.08119202
+  eval-batch-shift:
+    avg_avg_episode_length:                     21.065844
+    avg_avg_return:                             0.9028582
+    avg_benchmark_regret:                       0.08464773
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14037323
+    proxy_corner:
+      avg_avg_return:                           0.05231451
+      avg_reward_per_step:                      0.0026626587
+  eval-batch-tree:
+    avg_avg_episode_length:                     24.803204
+    avg_avg_return:                             0.879518
+    avg_benchmark_regret:                       0.105117194
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09471893
+    proxy_corner:
+      avg_avg_return:                           0.879518
+      avg_reward_per_step:                      0.09471893
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        141557760
+    env-step-replay-before:                     71729152
+    env-step-used-before:                       141557760
+    ppo-update-before:                          86400
+    env-step-generate-before:                   69828608
+    env-step-all-after:                         141590528
+    env-step-replay-after:                      71761920
+    env-step-used-after:                        141590528
+    ppo-update-after:                           86420
+    env-step-generate-after:                    69828608
+  train-replay:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     70.13295
+    avg_avg_return:                             0.38146
+    avg_reward_per_step:                        0.010284424
+    proxy_corner:
+      avg_avg_return:                           0.113210544
+      avg_reward_per_step:                      0.0019836426
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     70.13295
+    avg_avg_return:                             0.38146
+    avg_reward_per_step:                        0.010284424
+    proxy_corner:
+      avg_avg_return:                           0.113210544
+      avg_reward_per_step:                      0.0019836426
+  perf:
+    env_steps_per_second:                       242685.80548488675
+    ppo_updates_per_second:                     42.15825848595323
+    eval-batch-train-duration:                  0.6758236745372415
+    eval-batch-orig-duration:                   0.6180952405557036
+    eval-batch-shift-duration:                  0.5960077568888664
+    eval-batch-tree-duration:                   0.6266296338289976
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.673584
+      cheese-corne_dist_wavg:                   18.06132
+      levels_cheese_is_in_corner_avg:           0.021240234
+      mouse-cheese_dist_finite_avg:             24.150146
+      mouse-cheese_dist_finite_wavg:            10.510516
+      mouse-cheese_dist_solvable_avg:           24.150146
+      mouse-cheese_dist_solvable_wavg:          10.510516
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.58677304
+    visit_patterns:
+      avg_first_visit_time:                     1864.7886
+      avg_last_visit_time:                      2180.2712
+      num_replay_batches:                       2190
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009484075
+    avg_actor_approxkl3:                        0.00085674284
+    avg_actor_clipfrac:                         0.040771484
+    avg_actor_loss:                             -0.0013449684
+    avg_advantage:                              -0.05224216
+    avg_critic_clipfrac:                        0.0036621094
+    avg_critic_loss:                            0.011949648
+    avg_entropy:                                0.2601708
+    avg_grad_norm_pre_clip:                     1.5200346
+    avg_loss:                                   0.0043696854
+    max:
+      max_actor_approxkl1:                      0.002638443
+      max_actor_approxkl3:                      0.0015657102
+      max_actor_clipfrac:                       0.06506348
+      max_actor_loss:                           0.00059220195
+      max_advantage:                            0.13721973
+      max_critic_clipfrac:                      0.032226562
+      max_critic_loss:                          0.015261752
+      max_entropy:                              0.27413613
+      max_grad_norm_pre_clip:                   2.4766521
+      max_loss:                                 0.0069994093
+    std:
+      std_actor_approxkl1:                      0.0007276791
+      std_actor_approxkl3:                      0.0003923817
+      std_actor_clipfrac:                       0.01608116
+      std_actor_loss:                           0.0010699992
+      std_advantage:                            0.15233336
+      std_critic_clipfrac:                      0.008915145
+      std_critic_loss:                          0.0020800855
+      std_entropy:                              0.008155017
+      std_grad_norm_pre_clip:                   0.4560384
+      std_loss:                                 0.001636392
+  eval-batch-train:
+    avg_avg_episode_length:                     20.233137
+    avg_avg_return:                             0.9089524
+    avg_benchmark_regret:                       0.07826461
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12612915
+    proxy_corner:
+      avg_avg_return:                           0.049430184
+      avg_reward_per_step:                      0.005027771
+  eval-batch-orig:
+    avg_avg_episode_length:                     25.16833
+    avg_avg_return:                             0.8740862
+    avg_benchmark_regret:                       0.11042535
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08477783
+    proxy_corner:
+      avg_avg_return:                           0.8740862
+      avg_reward_per_step:                      0.08477783
+  eval-batch-shift:
+    avg_avg_episode_length:                     20.265003
+    avg_avg_return:                             0.906381
+    avg_benchmark_regret:                       0.08112495
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14253235
+    proxy_corner:
+      avg_avg_return:                           0.052642025
+      avg_reward_per_step:                      0.0027618408
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.868767
+    avg_avg_return:                             0.87685287
+    avg_benchmark_regret:                       0.10778236
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.097595215
+    proxy_corner:
+      avg_avg_return:                           0.87685287
+      avg_reward_per_step:                      0.097595215
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        142606336
+    env-step-replay-before:                     72286208
+    env-step-used-before:                       142606336
+    ppo-update-before:                          87040
+    env-step-generate-before:                   70320128
+    env-step-all-after:                         142639104
+    env-step-replay-after:                      72318976
+    env-step-used-after:                        142639104
+    ppo-update-after:                           87060
+    env-step-generate-after:                    70320128
+  train-replay:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     72.421936
+    avg_avg_return:                             0.37282833
+    avg_reward_per_step:                        0.008972168
+    proxy_corner:
+      avg_avg_return:                           0.10370188
+      avg_reward_per_step:                      0.0018920898
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     72.421936
+    avg_avg_return:                             0.37282833
+    avg_reward_per_step:                        0.008972168
+    proxy_corner:
+      avg_avg_return:                           0.10370188
+      avg_reward_per_step:                      0.0018920898
+  perf:
+    env_steps_per_second:                       229396.42567736446
+    ppo_updates_per_second:                     40.97772250858082
+    eval-batch-train-duration:                  0.667386781424284
+    eval-batch-orig-duration:                   0.6470589796081185
+    eval-batch-shift-duration:                  0.6165137998759747
+    eval-batch-tree-duration:                   0.6191143803298473
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.723389
+      cheese-corne_dist_wavg:                   11.805032
+      levels_cheese_is_in_corner_avg:           0.020507812
+      mouse-cheese_dist_finite_avg:             24.226807
+      mouse-cheese_dist_finite_wavg:            11.416323
+      mouse-cheese_dist_solvable_avg:           24.226807
+      mouse-cheese_dist_solvable_wavg:          11.416324
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.58692515
+    visit_patterns:
+      avg_first_visit_time:                     1878.6482
+      avg_last_visit_time:                      2197.3337
+      num_replay_batches:                       2207
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010364626
+    avg_actor_approxkl3:                        0.0009444008
+    avg_actor_clipfrac:                         0.04435425
+    avg_actor_loss:                             -0.000491675
+    avg_advantage:                              -0.0475869
+    avg_critic_clipfrac:                        0.00036010743
+    avg_critic_loss:                            0.0112975
+    avg_entropy:                                0.2575051
+    avg_grad_norm_pre_clip:                     1.6851882
+    avg_loss:                                   0.0048995703
+    max:
+      max_actor_approxkl1:                      0.0037312638
+      max_actor_approxkl3:                      0.0024439034
+      max_actor_clipfrac:                       0.08911133
+      max_actor_loss:                           0.0043118
+      max_advantage:                            0.15345258
+      max_critic_clipfrac:                      0.0028076172
+      max_critic_loss:                          0.013602081
+      max_entropy:                              0.26908356
+      max_grad_norm_pre_clip:                   3.1347032
+      max_loss:                                 0.010587104
+    std:
+      std_actor_approxkl1:                      0.0008409677
+      std_actor_approxkl3:                      0.00049195136
+      std_actor_clipfrac:                       0.018073594
+      std_actor_loss:                           0.0015853456
+      std_advantage:                            0.14823672
+      std_critic_clipfrac:                      0.0008174839
+      std_critic_loss:                          0.0013444785
+      std_entropy:                              0.008159347
+      std_grad_norm_pre_clip:                   0.54318416
+      std_loss:                                 0.0018256815
+  eval-batch-train:
+    avg_avg_episode_length:                     19.78586
+    avg_avg_return:                             0.91744095
+    avg_benchmark_regret:                       0.06977612
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12536621
+    proxy_corner:
+      avg_avg_return:                           0.050104946
+      avg_reward_per_step:                      0.004562378
+  eval-batch-orig:
+    avg_avg_episode_length:                     26.11018
+    avg_avg_return:                             0.8703125
+    avg_benchmark_regret:                       0.11419905
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08131409
+    proxy_corner:
+      avg_avg_return:                           0.8703125
+      avg_reward_per_step:                      0.08131409
+  eval-batch-shift:
+    avg_avg_episode_length:                     20.648205
+    avg_avg_return:                             0.90024436
+    avg_benchmark_regret:                       0.087261625
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14034271
+    proxy_corner:
+      avg_avg_return:                           0.053833578
+      avg_reward_per_step:                      0.002861023
+  eval-batch-tree:
+    avg_avg_episode_length:                     25.171087
+    avg_avg_return:                             0.872173
+    avg_benchmark_regret:                       0.11246218
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09526062
+    proxy_corner:
+      avg_avg_return:                           0.872173
+      avg_reward_per_step:                      0.09526062
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        143654912
+    env-step-replay-before:                     72941568
+    env-step-used-before:                       143654912
+    ppo-update-before:                          87680
+    env-step-generate-before:                   70713344
+    env-step-all-after:                         143687680
+    env-step-replay-after:                      72974336
+    env-step-used-after:                        143687680
+    ppo-update-after:                           87700
+    env-step-generate-after:                    70713344
+  train-replay:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     68.43512
+    avg_avg_return:                             0.43331453
+    avg_reward_per_step:                        0.011108398
+    proxy_corner:
+      avg_avg_return:                           0.1431064
+      avg_reward_per_step:                      0.002166748
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     68.43512
+    avg_avg_return:                             0.43331453
+    avg_reward_per_step:                        0.011108398
+    proxy_corner:
+      avg_avg_return:                           0.1431064
+      avg_reward_per_step:                      0.002166748
+  perf:
+    env_steps_per_second:                       346792.83924154454
+    ppo_updates_per_second:                     42.23058093214911
+    eval-batch-train-duration:                  0.5231181597337127
+    eval-batch-orig-duration:                   0.48701152205467224
+    eval-batch-shift-duration:                  0.48563929460942745
+    eval-batch-tree-duration:                   0.48296591732650995
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.797852
+      cheese-corne_dist_wavg:                   13.604761
+      levels_cheese_is_in_corner_avg:           0.020507812
+      mouse-cheese_dist_finite_avg:             24.288818
+      mouse-cheese_dist_finite_wavg:            12.338718
+      mouse-cheese_dist_solvable_avg:           24.288818
+      mouse-cheese_dist_solvable_wavg:          12.338718
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5744616
+    visit_patterns:
+      avg_first_visit_time:                     1891.8406
+      avg_last_visit_time:                      2217.2268
+      num_replay_batches:                       2227
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009679004
+    avg_actor_approxkl3:                        0.0009547824
+    avg_actor_clipfrac:                         0.045178223
+    avg_actor_loss:                             -0.0011077806
+    avg_advantage:                              -0.03219468
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.009254416
+    avg_entropy:                                0.25196436
+    avg_grad_norm_pre_clip:                     1.4498787
+    avg_loss:                                   0.0032674626
+    max:
+      max_actor_approxkl1:                      0.0023963714
+      max_actor_approxkl3:                      0.0020519886
+      max_actor_clipfrac:                       0.083862305
+      max_actor_loss:                           0.00085149705
+      max_advantage:                            0.16883641
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.011406718
+      max_entropy:                              0.27618897
+      max_grad_norm_pre_clip:                   2.293984
+      max_loss:                                 0.0058752587
+    std:
+      std_actor_approxkl1:                      0.0006613674
+      std_actor_approxkl3:                      0.0004914901
+      std_actor_clipfrac:                       0.01936021
+      std_actor_loss:                           0.0010166278
+      std_advantage:                            0.13532549
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0013560128
+      std_entropy:                              0.009441573
+      std_grad_norm_pre_clip:                   0.44893834
+      std_loss:                                 0.0011176219
+  eval-batch-train:
+    avg_avg_episode_length:                     19.5027
+    avg_avg_return:                             0.9166751
+    avg_benchmark_regret:                       0.070541985
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12435913
+    proxy_corner:
+      avg_avg_return:                           0.056857932
+      avg_reward_per_step:                      0.0049362183
+  eval-batch-orig:
+    avg_avg_episode_length:                     25.117449
+    avg_avg_return:                             0.8828994
+    avg_benchmark_regret:                       0.10161217
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08211517
+    proxy_corner:
+      avg_avg_return:                           0.8828994
+      avg_reward_per_step:                      0.08211517
+  eval-batch-shift:
+    avg_avg_episode_length:                     19.720352
+    avg_avg_return:                             0.91195554
+    avg_benchmark_regret:                       0.075550415
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14296722
+    proxy_corner:
+      avg_avg_return:                           0.054667898
+      avg_reward_per_step:                      0.0030441284
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.43502
+    avg_avg_return:                             0.89308304
+    avg_benchmark_regret:                       0.09155218
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09701538
+    proxy_corner:
+      avg_avg_return:                           0.89308304
+      avg_reward_per_step:                      0.09701538
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        144703488
+    env-step-replay-before:                     73531392
+    env-step-used-before:                       144703488
+    ppo-update-before:                          88320
+    env-step-generate-before:                   71172096
+    env-step-all-after:                         144736256
+    env-step-replay-after:                      73564160
+    env-step-used-after:                        144736256
+    ppo-update-after:                           88340
+    env-step-generate-after:                    71172096
+  train-replay:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     68.14172
+    avg_avg_return:                             0.42221227
+    avg_reward_per_step:                        0.01083374
+    proxy_corner:
+      avg_avg_return:                           0.11626652
+      avg_reward_per_step:                      0.0020446777
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     68.14172
+    avg_avg_return:                             0.42221227
+    avg_reward_per_step:                        0.01083374
+    proxy_corner:
+      avg_avg_return:                           0.11626652
+      avg_reward_per_step:                      0.0020446777
+  perf:
+    env_steps_per_second:                       343377.0657115295
+    ppo_updates_per_second:                     41.875402881980186
+    eval-batch-train-duration:                  0.4508562497794628
+    eval-batch-orig-duration:                   0.4477958492934704
+    eval-batch-shift-duration:                  0.40519775822758675
+    eval-batch-tree-duration:                   0.40092676505446434
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.787354
+      cheese-corne_dist_wavg:                   6.38578
+      levels_cheese_is_in_corner_avg:           0.020507812
+      mouse-cheese_dist_finite_avg:             24.324463
+      mouse-cheese_dist_finite_wavg:            11.449922
+      mouse-cheese_dist_solvable_avg:           24.324463
+      mouse-cheese_dist_solvable_wavg:          11.449923
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016798
+    scoring:
+      avg_scores:                               0.5638145
+    visit_patterns:
+      avg_first_visit_time:                     1902.4414
+      avg_last_visit_time:                      2235.1926
+      num_replay_batches:                       2245
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0015450352
+    avg_actor_approxkl3:                        0.0014596869
+    avg_actor_clipfrac:                         0.06020508
+    avg_actor_loss:                             0.00018658489
+    avg_advantage:                              -0.040909916
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.010010862
+    avg_entropy:                                0.24535711
+    avg_grad_norm_pre_clip:                     2.598266
+    avg_loss:                                   0.004946658
+    max:
+      max_actor_approxkl1:                      0.002789419
+      max_actor_approxkl3:                      0.0023591835
+      max_actor_clipfrac:                       0.08605957
+      max_actor_loss:                           0.0033569112
+      max_advantage:                            0.15405196
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.015088247
+      max_entropy:                              0.26712975
+      max_grad_norm_pre_clip:                   4.3372426
+      max_loss:                                 0.0073274667
+    std:
+      std_actor_approxkl1:                      0.0007428692
+      std_actor_approxkl3:                      0.00064973836
+      std_actor_clipfrac:                       0.021259855
+      std_actor_loss:                           0.001297303
+      std_advantage:                            0.14018105
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0015611701
+      std_entropy:                              0.009436492
+      std_grad_norm_pre_clip:                   0.75345844
+      std_loss:                                 0.0013733718
+  eval-batch-train:
+    avg_avg_episode_length:                     20.10383
+    avg_avg_return:                             0.90829927
+    avg_benchmark_regret:                       0.0789178
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12483215
+    proxy_corner:
+      avg_avg_return:                           0.056001212
+      avg_reward_per_step:                      0.0049362183
+  eval-batch-orig:
+    avg_avg_episode_length:                     26.31519
+    avg_avg_return:                             0.8633276
+    avg_benchmark_regret:                       0.1211839
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08281708
+    proxy_corner:
+      avg_avg_return:                           0.8633276
+      avg_reward_per_step:                      0.08281708
+  eval-batch-shift:
+    avg_avg_episode_length:                     19.776268
+    avg_avg_return:                             0.91495353
+    avg_benchmark_regret:                       0.07255241
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14363861
+    proxy_corner:
+      avg_avg_return:                           0.04993517
+      avg_reward_per_step:                      0.0029525757
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.815357
+    avg_avg_return:                             0.8856151
+    avg_benchmark_regret:                       0.09902011
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.098976135
+    proxy_corner:
+      avg_avg_return:                           0.8856151
+      avg_reward_per_step:                      0.098976135
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        145752064
+    env-step-replay-before:                     74088448
+    env-step-used-before:                       145752064
+    ppo-update-before:                          88960
+    env-step-generate-before:                   71663616
+    env-step-all-after:                         145784832
+    env-step-replay-after:                      74121216
+    env-step-used-after:                        145784832
+    ppo-update-after:                           88980
+    env-step-generate-after:                    71663616
+  train-replay:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     68.43469
+    avg_avg_return:                             0.40767133
+    avg_reward_per_step:                        0.010986328
+    proxy_corner:
+      avg_avg_return:                           0.07298282
+      avg_reward_per_step:                      0.0015869141
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     68.43469
+    avg_avg_return:                             0.40767133
+    avg_reward_per_step:                        0.010986328
+    proxy_corner:
+      avg_avg_return:                           0.07298282
+      avg_reward_per_step:                      0.0015869141
+  perf:
+    env_steps_per_second:                       259004.26786311474
+    ppo_updates_per_second:                     42.63119102016663
+    eval-batch-train-duration:                  0.6704713013023138
+    eval-batch-orig-duration:                   0.6163508947938681
+    eval-batch-shift-duration:                  0.6252247532829642
+    eval-batch-tree-duration:                   0.6553944135084748
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.659668
+      cheese-corne_dist_wavg:                   18.060024
+      levels_cheese_is_in_corner_avg:           0.020263672
+      mouse-cheese_dist_finite_avg:             24.237549
+      mouse-cheese_dist_finite_wavg:            10.523569
+      mouse-cheese_dist_solvable_avg:           24.237549
+      mouse-cheese_dist_solvable_wavg:          10.52357
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5752667
+    visit_patterns:
+      avg_first_visit_time:                     1916.9678
+      avg_last_visit_time:                      2252.3625
+      num_replay_batches:                       2262
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001099333
+    avg_actor_approxkl3:                        0.001057456
+    avg_actor_clipfrac:                         0.051013183
+    avg_actor_loss:                             -0.00081842503
+    avg_advantage:                              -0.02565594
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.010009324
+    avg_entropy:                                0.2574017
+    avg_grad_norm_pre_clip:                     1.6434147
+    avg_loss:                                   0.003928835
+    max:
+      max_actor_approxkl1:                      0.0024700256
+      max_actor_approxkl3:                      0.002050447
+      max_actor_clipfrac:                       0.08947754
+      max_actor_loss:                           0.0011139214
+      max_advantage:                            0.2310223
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.011291721
+      max_entropy:                              0.2791379
+      max_grad_norm_pre_clip:                   2.698249
+      max_loss:                                 0.006135843
+    std:
+      std_actor_approxkl1:                      0.0006443144
+      std_actor_approxkl3:                      0.00044318198
+      std_actor_clipfrac:                       0.018961528
+      std_actor_loss:                           0.0010736097
+      std_advantage:                            0.14122741
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0008755489
+      std_entropy:                              0.009475551
+      std_grad_norm_pre_clip:                   0.5105963
+      std_loss:                                 0.0011721242
+  eval-batch-train:
+    avg_avg_episode_length:                     20.564497
+    avg_avg_return:                             0.90436196
+    avg_benchmark_regret:                       0.08285506
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12459564
+    proxy_corner:
+      avg_avg_return:                           0.057183027
+      avg_reward_per_step:                      0.0049591064
+  eval-batch-orig:
+    avg_avg_episode_length:                     25.289562
+    avg_avg_return:                             0.8788157
+    avg_benchmark_regret:                       0.105695784
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.084739685
+    proxy_corner:
+      avg_avg_return:                           0.8788157
+      avg_reward_per_step:                      0.084739685
+  eval-batch-shift:
+    avg_avg_episode_length:                     20.003744
+    avg_avg_return:                             0.9076629
+    avg_benchmark_regret:                       0.07984303
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14428711
+    proxy_corner:
+      avg_avg_return:                           0.05631931
+      avg_reward_per_step:                      0.0030136108
+  eval-batch-tree:
+    avg_avg_episode_length:                     24.7052
+    avg_avg_return:                             0.87831646
+    avg_benchmark_regret:                       0.10631873
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.098487854
+    proxy_corner:
+      avg_avg_return:                           0.87831646
+      avg_reward_per_step:                      0.098487854
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        146800640
+    env-step-replay-before:                     74711040
+    env-step-used-before:                       146800640
+    ppo-update-before:                          89600
+    env-step-generate-before:                   72089600
+    env-step-all-after:                         146833408
+    env-step-replay-after:                      74743808
+    env-step-used-after:                        146833408
+    ppo-update-after:                           89620
+    env-step-generate-after:                    72089600
+  train-replay:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     68.8703
+    avg_avg_return:                             0.41998494
+    avg_reward_per_step:                        0.01083374
+    proxy_corner:
+      avg_avg_return:                           0.11730406
+      avg_reward_per_step:                      0.002319336
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     68.8703
+    avg_avg_return:                             0.41998494
+    avg_reward_per_step:                        0.01083374
+    proxy_corner:
+      avg_avg_return:                           0.11730406
+      avg_reward_per_step:                      0.002319336
+  perf:
+    env_steps_per_second:                       328497.86387925956
+    ppo_updates_per_second:                     41.21730699166719
+    eval-batch-train-duration:                  0.5422854274511337
+    eval-batch-orig-duration:                   0.4858471527695656
+    eval-batch-shift-duration:                  0.4850584911182523
+    eval-batch-tree-duration:                   0.4376169592142105
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.707031
+      cheese-corne_dist_wavg:                   21.657616
+      levels_cheese_is_in_corner_avg:           0.01953125
+      mouse-cheese_dist_finite_avg:             24.42334
+      mouse-cheese_dist_finite_wavg:            11.436859
+      mouse-cheese_dist_solvable_avg:           24.42334
+      mouse-cheese_dist_solvable_wavg:          11.43686
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.5708503
+    visit_patterns:
+      avg_first_visit_time:                     1931.9695
+      avg_last_visit_time:                      2271.277
+      num_replay_batches:                       2281
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0008949783
+    avg_actor_approxkl3:                        0.00075080607
+    avg_actor_clipfrac:                         0.037420657
+    avg_actor_loss:                             -0.0013351456
+    avg_advantage:                              -0.033795647
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.009670305
+    avg_entropy:                                0.244291
+    avg_grad_norm_pre_clip:                     1.2364978
+    avg_loss:                                   0.003255716
+    max:
+      max_actor_approxkl1:                      0.002037113
+      max_actor_approxkl3:                      0.0013875648
+      max_actor_clipfrac:                       0.059448242
+      max_actor_loss:                           0.0010845959
+      max_advantage:                            0.1921165
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.011901295
+      max_entropy:                              0.25945657
+      max_grad_norm_pre_clip:                   2.4965765
+      max_loss:                                 0.0055889417
+    std:
+      std_actor_approxkl1:                      0.0005305418
+      std_actor_approxkl3:                      0.00029985036
+      std_actor_clipfrac:                       0.013503343
+      std_actor_loss:                           0.0011537592
+      std_advantage:                            0.13825287
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011589074
+      std_entropy:                              0.0079677645
+      std_grad_norm_pre_clip:                   0.49219018
+      std_loss:                                 0.0012671317
+  eval-batch-train:
+    avg_avg_episode_length:                     20.384815
+    avg_avg_return:                             0.9126375
+    avg_benchmark_regret:                       0.074579574
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12489319
+    proxy_corner:
+      avg_avg_return:                           0.059272066
+      avg_reward_per_step:                      0.0050964355
+  eval-batch-orig:
+    avg_avg_episode_length:                     25.295029
+    avg_avg_return:                             0.8799306
+    avg_benchmark_regret:                       0.104580924
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08331299
+    proxy_corner:
+      avg_avg_return:                           0.8799306
+      avg_reward_per_step:                      0.08331299
+  eval-batch-shift:
+    avg_avg_episode_length:                     19.550053
+    avg_avg_return:                             0.91837215
+    avg_benchmark_regret:                       0.06913378
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14324188
+    proxy_corner:
+      avg_avg_return:                           0.053948276
+      avg_reward_per_step:                      0.0026245117
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.828186
+    avg_avg_return:                             0.8858987
+    avg_benchmark_regret:                       0.09873645
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09933472
+    proxy_corner:
+      avg_avg_return:                           0.8858987
+      avg_reward_per_step:                      0.09933472
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        147849216
+    env-step-replay-before:                     75235328
+    env-step-used-before:                       147849216
+    ppo-update-before:                          90240
+    env-step-generate-before:                   72613888
+    env-step-all-after:                         147881984
+    env-step-replay-after:                      75235328
+    env-step-used-after:                        147881984
+    ppo-update-after:                           90260
+    env-step-generate-after:                    72646656
+  train-generate:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     22.976784
+    avg_avg_return:                             0.8157326
+    avg_reward_per_step:                        0.112091064
+    proxy_corner:
+      avg_avg_return:                           0.046512138
+      avg_reward_per_step:                      0.0028686523
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     22.976784
+    avg_avg_return:                             0.8157326
+    avg_reward_per_step:                        0.112091064
+    proxy_corner:
+      avg_avg_return:                           0.046512138
+      avg_reward_per_step:                      0.0028686523
+  perf:
+    env_steps_per_second:                       320081.35650645447
+    ppo_updates_per_second:                     41.31939109680808
+    eval-batch-train-duration:                  0.537915263324976
+    eval-batch-orig-duration:                   0.48261659406125546
+    eval-batch-shift-duration:                  0.4794795075431466
+    eval-batch-tree-duration:                   0.49171171709895134
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.75586
+      cheese-corne_dist_wavg:                   11.766923
+      levels_cheese_is_in_corner_avg:           0.01953125
+      mouse-cheese_dist_finite_avg:             24.414307
+      mouse-cheese_dist_finite_wavg:            11.446656
+      mouse-cheese_dist_solvable_avg:           24.414307
+      mouse-cheese_dist_solvable_wavg:          11.446657
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.55628645
+    visit_patterns:
+      avg_first_visit_time:                     1948.3108
+      avg_last_visit_time:                      2286.3098
+      num_replay_batches:                       2296
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00081526017
+    avg_actor_approxkl3:                        0.0005570958
+    avg_actor_clipfrac:                         0.023168946
+    avg_actor_loss:                             -0.0013689565
+    avg_advantage:                              0.02931584
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0027423927
+    avg_entropy:                                0.11075021
+    avg_grad_norm_pre_clip:                     1.0317838
+    avg_loss:                                   -0.000108510394
+    max:
+      max_actor_approxkl1:                      0.00162021
+      max_actor_approxkl3:                      0.0011609433
+      max_actor_clipfrac:                       0.043823242
+      max_actor_loss:                           0.0001306571
+      max_advantage:                            0.17771995
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.005206864
+      max_entropy:                              0.13319479
+      max_grad_norm_pre_clip:                   2.5930614
+      max_loss:                                 0.00167103
+    std:
+      std_actor_approxkl1:                      0.00061364326
+      std_actor_approxkl3:                      0.0003008938
+      std_actor_clipfrac:                       0.011533972
+      std_actor_loss:                           0.0009945594
+      std_advantage:                            0.07148952
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010250149
+      std_entropy:                              0.011337545
+      std_grad_norm_pre_clip:                   0.45162514
+      std_loss:                                 0.0010357657
+  eval-batch-train:
+    avg_avg_episode_length:                     19.930904
+    avg_avg_return:                             0.91111183
+    avg_benchmark_regret:                       0.07610518
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12638855
+    proxy_corner:
+      avg_avg_return:                           0.05333386
+      avg_reward_per_step:                      0.005126953
+  eval-batch-orig:
+    avg_avg_episode_length:                     25.661026
+    avg_avg_return:                             0.8767184
+    avg_benchmark_regret:                       0.10779315
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08110809
+    proxy_corner:
+      avg_avg_return:                           0.8767184
+      avg_reward_per_step:                      0.08110809
+  eval-batch-shift:
+    avg_avg_episode_length:                     19.049625
+    avg_avg_return:                             0.9196346
+    avg_benchmark_regret:                       0.06787141
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14402008
+    proxy_corner:
+      avg_avg_return:                           0.05076633
+      avg_reward_per_step:                      0.0028839111
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.762589
+    avg_avg_return:                             0.88802594
+    avg_benchmark_regret:                       0.09660925
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.096839905
+    proxy_corner:
+      avg_avg_return:                           0.88802594
+      avg_reward_per_step:                      0.096839905
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        148897792
+    env-step-replay-before:                     75694080
+    env-step-used-before:                       148897792
+    ppo-update-before:                          90880
+    env-step-generate-before:                   73203712
+    env-step-all-after:                         148930560
+    env-step-replay-after:                      75726848
+    env-step-used-after:                        148930560
+    ppo-update-after:                           90900
+    env-step-generate-after:                    73203712
+  train-replay:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     70.538864
+    avg_avg_return:                             0.3846118
+    avg_reward_per_step:                        0.009857178
+    proxy_corner:
+      avg_avg_return:                           0.11269344
+      avg_reward_per_step:                      0.0020141602
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     70.538864
+    avg_avg_return:                             0.3846118
+    avg_reward_per_step:                        0.009857178
+    proxy_corner:
+      avg_avg_return:                           0.11269344
+      avg_reward_per_step:                      0.0020141602
+  perf:
+    env_steps_per_second:                       350175.0387795247
+    ppo_updates_per_second:                     41.675444817732696
+    eval-batch-train-duration:                  0.5202700393274426
+    eval-batch-orig-duration:                   0.4845073763281107
+    eval-batch-shift-duration:                  0.4787067770957947
+    eval-batch-tree-duration:                   0.48989778105169535
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.745117
+      cheese-corne_dist_wavg:                   21.651981
+      levels_cheese_is_in_corner_avg:           0.01953125
+      mouse-cheese_dist_finite_avg:             24.364258
+      mouse-cheese_dist_finite_wavg:            10.553702
+      mouse-cheese_dist_solvable_avg:           24.364258
+      mouse-cheese_dist_solvable_wavg:          10.553703
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.57920873
+    visit_patterns:
+      avg_first_visit_time:                     1963.6826
+      avg_last_visit_time:                      2301.3525
+      num_replay_batches:                       2311
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011165428
+    avg_actor_approxkl3:                        0.0010979687
+    avg_actor_clipfrac:                         0.048937988
+    avg_actor_loss:                             -0.00044167863
+    avg_advantage:                              -0.052738097
+    avg_critic_clipfrac:                        0.04319458
+    avg_critic_loss:                            0.011823701
+    avg_entropy:                                0.2622094
+    avg_grad_norm_pre_clip:                     1.7393659
+    avg_loss:                                   0.005207963
+    max:
+      max_actor_approxkl1:                      0.0027669207
+      max_actor_approxkl3:                      0.002290757
+      max_actor_clipfrac:                       0.092163086
+      max_actor_loss:                           0.00084879994
+      max_advantage:                            0.15344417
+      max_critic_clipfrac:                      0.31176758
+      max_critic_loss:                          0.014373889
+      max_entropy:                              0.27818066
+      max_grad_norm_pre_clip:                   2.6432235
+      max_loss:                                 0.006917975
+    std:
+      std_actor_approxkl1:                      0.00082965766
+      std_actor_approxkl3:                      0.00059733784
+      std_actor_clipfrac:                       0.024102531
+      std_actor_loss:                           0.0008720121
+      std_advantage:                            0.15144312
+      std_critic_clipfrac:                      0.08709268
+      std_critic_loss:                          0.0012643584
+      std_entropy:                              0.008804367
+      std_grad_norm_pre_clip:                   0.50707513
+      std_loss:                                 0.0011132536
+  eval-batch-train:
+    avg_avg_episode_length:                     20.28728
+    avg_avg_return:                             0.9050013
+    avg_benchmark_regret:                       0.08221577
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12556458
+    proxy_corner:
+      avg_avg_return:                           0.0570589
+      avg_reward_per_step:                      0.0048446655
+  eval-batch-orig:
+    avg_avg_episode_length:                     24.30471
+    avg_avg_return:                             0.8908962
+    avg_benchmark_regret:                       0.09361538
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08279419
+    proxy_corner:
+      avg_avg_return:                           0.8908962
+      avg_reward_per_step:                      0.08279419
+  eval-batch-shift:
+    avg_avg_episode_length:                     20.008415
+    avg_avg_return:                             0.9099674
+    avg_benchmark_regret:                       0.07753849
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14431763
+    proxy_corner:
+      avg_avg_return:                           0.055718347
+      avg_reward_per_step:                      0.002937317
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.399643
+    avg_avg_return:                             0.88658327
+    avg_benchmark_regret:                       0.098051906
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09829712
+    proxy_corner:
+      avg_avg_return:                           0.88658327
+      avg_reward_per_step:                      0.09829712
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        149946368
+    env-step-replay-before:                     76152832
+    env-step-used-before:                       149946368
+    ppo-update-before:                          91520
+    env-step-generate-before:                   73793536
+    env-step-all-after:                         149979136
+    env-step-replay-after:                      76152832
+    env-step-used-after:                        149979136
+    ppo-update-after:                           91540
+    env-step-generate-after:                    73826304
+  train-generate:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     23.727062
+    avg_avg_return:                             0.8035104
+    avg_reward_per_step:                        0.1244812
+    proxy_corner:
+      avg_avg_return:                           0.07473238
+      avg_reward_per_step:                      0.0052490234
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     23.727062
+    avg_avg_return:                             0.8035104
+    avg_reward_per_step:                        0.1244812
+    proxy_corner:
+      avg_avg_return:                           0.07473238
+      avg_reward_per_step:                      0.0052490234
+  perf:
+    env_steps_per_second:                       443867.61552246567
+    ppo_updates_per_second:                     41.29655779329776
+    eval-batch-train-duration:                  0.5144215133041143
+    eval-batch-orig-duration:                   0.42551798559725285
+    eval-batch-shift-duration:                  0.4237142242491245
+    eval-batch-tree-duration:                   0.4548413595184684
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.843262
+      cheese-corne_dist_wavg:                   21.694674
+      levels_cheese_is_in_corner_avg:           0.021240234
+      mouse-cheese_dist_finite_avg:             24.38086
+      mouse-cheese_dist_finite_wavg:            10.536707
+      mouse-cheese_dist_solvable_avg:           24.38086
+      mouse-cheese_dist_solvable_wavg:          10.536707
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.99999
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.58634937
+    visit_patterns:
+      avg_first_visit_time:                     1982.3142
+      avg_last_visit_time:                      2314.2227
+      num_replay_batches:                       2324
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00095089263
+    avg_actor_approxkl3:                        0.0010379512
+    avg_actor_clipfrac:                         0.033056643
+    avg_actor_loss:                             -8.6405315e-05
+    avg_advantage:                              -0.0023741596
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0028844168
+    avg_entropy:                                0.105360225
+    avg_grad_norm_pre_clip:                     1.4349171
+    avg_loss:                                   0.0012504427
+    max:
+      max_actor_approxkl1:                      0.0028162193
+      max_actor_approxkl3:                      0.0027603512
+      max_actor_clipfrac:                       0.057006836
+      max_actor_loss:                           0.0025619268
+      max_advantage:                            0.14363503
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.004867443
+      max_entropy:                              0.1321815
+      max_grad_norm_pre_clip:                   2.503152
+      max_loss:                                 0.0043731723
+    std:
+      std_actor_approxkl1:                      0.00069691957
+      std_actor_approxkl3:                      0.0006340243
+      std_actor_clipfrac:                       0.012953242
+      std_actor_loss:                           0.0012571261
+      std_advantage:                            0.07596524
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011889884
+      std_entropy:                              0.015259327
+      std_grad_norm_pre_clip:                   0.5201636
+      std_loss:                                 0.0013121266
+  eval-batch-train:
+    avg_avg_episode_length:                     20.518856
+    avg_avg_return:                             0.9051951
+    avg_benchmark_regret:                       0.082022004
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12773895
+    proxy_corner:
+      avg_avg_return:                           0.058891196
+      avg_reward_per_step:                      0.0050964355
+  eval-batch-orig:
+    avg_avg_episode_length:                     25.642641
+    avg_avg_return:                             0.87241656
+    avg_benchmark_regret:                       0.11209501
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08357239
+    proxy_corner:
+      avg_avg_return:                           0.87241656
+      avg_reward_per_step:                      0.08357239
+  eval-batch-shift:
+    avg_avg_episode_length:                     19.758055
+    avg_avg_return:                             0.9045868
+    avg_benchmark_regret:                       0.082919136
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14437103
+    proxy_corner:
+      avg_avg_return:                           0.05423902
+      avg_reward_per_step:                      0.0029449463
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.624954
+    avg_avg_return:                             0.88635427
+    avg_benchmark_regret:                       0.09828094
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09965515
+    proxy_corner:
+      avg_avg_return:                           0.88635427
+      avg_reward_per_step:                      0.09965515
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        150994944
+    env-step-replay-before:                     76709888
+    env-step-used-before:                       150994944
+    ppo-update-before:                          92160
+    env-step-generate-before:                   74285056
+    env-step-all-after:                         151027712
+    env-step-replay-after:                      76742656
+    env-step-used-after:                        151027712
+    ppo-update-after:                           92180
+    env-step-generate-after:                    74285056
+  train-replay:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     69.68748
+    avg_avg_return:                             0.3754506
+    avg_reward_per_step:                        0.010375977
+    proxy_corner:
+      avg_avg_return:                           0.13096158
+      avg_reward_per_step:                      0.0021362305
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     69.68748
+    avg_avg_return:                             0.3754506
+    avg_reward_per_step:                        0.010375977
+    proxy_corner:
+      avg_avg_return:                           0.13096158
+      avg_reward_per_step:                      0.0021362305
+  perf:
+    env_steps_per_second:                       316735.990990908
+    ppo_updates_per_second:                     42.14636997596995
+    eval-batch-train-duration:                  0.5116454837843776
+    eval-batch-orig-duration:                   0.4277207413688302
+    eval-batch-shift-duration:                  0.4502918990328908
+    eval-batch-tree-duration:                   0.4277056660503149
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.79126
+      cheese-corne_dist_wavg:                   11.78899
+      levels_cheese_is_in_corner_avg:           0.021484375
+      mouse-cheese_dist_finite_avg:             24.40747
+      mouse-cheese_dist_finite_wavg:            11.438817
+      mouse-cheese_dist_solvable_avg:           24.40747
+      mouse-cheese_dist_solvable_wavg:          11.438817
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.99999
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016816
+    scoring:
+      avg_scores:                               0.5903791
+    visit_patterns:
+      avg_first_visit_time:                     1994.4697
+      avg_last_visit_time:                      2332.2122
+      num_replay_batches:                       2342
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011108002
+    avg_actor_approxkl3:                        0.001050703
+    avg_actor_clipfrac:                         0.04827881
+    avg_actor_loss:                             -0.00039195045
+    avg_advantage:                              -0.055867735
+    avg_critic_clipfrac:                        0.007562256
+    avg_critic_loss:                            0.012361962
+    avg_entropy:                                0.26560906
+    avg_grad_norm_pre_clip:                     2.0760236
+    avg_loss:                                   0.0055234223
+    max:
+      max_actor_approxkl1:                      0.002075843
+      max_actor_approxkl3:                      0.0019329828
+      max_actor_clipfrac:                       0.07434082
+      max_actor_loss:                           0.0016709566
+      max_advantage:                            0.18104059
+      max_critic_clipfrac:                      0.048828125
+      max_critic_loss:                          0.01442638
+      max_entropy:                              0.28184134
+      max_grad_norm_pre_clip:                   3.6061618
+      max_loss:                                 0.008611204
+    std:
+      std_actor_approxkl1:                      0.00057099084
+      std_actor_approxkl3:                      0.00046026055
+      std_actor_clipfrac:                       0.017567249
+      std_actor_loss:                           0.0012475041
+      std_advantage:                            0.15537181
+      std_critic_clipfrac:                      0.014029244
+      std_critic_loss:                          0.001021391
+      std_entropy:                              0.0062737563
+      std_grad_norm_pre_clip:                   0.8013327
+      std_loss:                                 0.0015013788
+  eval-batch-train:
+    avg_avg_episode_length:                     20.11444
+    avg_avg_return:                             0.9106839
+    avg_benchmark_regret:                       0.076533206
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12767029
+    proxy_corner:
+      avg_avg_return:                           0.057988748
+      avg_reward_per_step:                      0.0050735474
+  eval-batch-orig:
+    avg_avg_episode_length:                     25.141396
+    avg_avg_return:                             0.8728603
+    avg_benchmark_regret:                       0.11165125
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.083106995
+    proxy_corner:
+      avg_avg_return:                           0.8728603
+      avg_reward_per_step:                      0.083106995
+  eval-batch-shift:
+    avg_avg_episode_length:                     19.569199
+    avg_avg_return:                             0.9115092
+    avg_benchmark_regret:                       0.0759967
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.1446991
+    proxy_corner:
+      avg_avg_return:                           0.052537397
+      avg_reward_per_step:                      0.0028457642
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.879253
+    avg_avg_return:                             0.8897221
+    avg_benchmark_regret:                       0.094913065
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.099121094
+    proxy_corner:
+      avg_avg_return:                           0.8897221
+      avg_reward_per_step:                      0.099121094
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        152043520
+    env-step-replay-before:                     77299712
+    env-step-used-before:                       152043520
+    ppo-update-before:                          92800
+    env-step-generate-before:                   74743808
+    env-step-all-after:                         152076288
+    env-step-replay-after:                      77299712
+    env-step-used-after:                        152076288
+    ppo-update-after:                           92820
+    env-step-generate-after:                    74776576
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     23.3264
+    avg_avg_return:                             0.8205447
+    avg_reward_per_step:                        0.118255615
+    proxy_corner:
+      avg_avg_return:                           0.03302009
+      avg_reward_per_step:                      0.0014343262
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     23.3264
+    avg_avg_return:                             0.8205447
+    avg_reward_per_step:                        0.118255615
+    proxy_corner:
+      avg_avg_return:                           0.03302009
+      avg_reward_per_step:                      0.0014343262
+  perf:
+    env_steps_per_second:                       295671.93428831344
+    ppo_updates_per_second:                     42.01953492193731
+    eval-batch-train-duration:                  0.534416476264596
+    eval-batch-orig-duration:                   0.48306809552013874
+    eval-batch-shift-duration:                  0.4841362861916423
+    eval-batch-tree-duration:                   0.4523871345445514
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.802734
+      cheese-corne_dist_wavg:                   17.170498
+      levels_cheese_is_in_corner_avg:           0.020507812
+      mouse-cheese_dist_finite_avg:             24.46875
+      mouse-cheese_dist_finite_wavg:            9.655327
+      mouse-cheese_dist_solvable_avg:           24.46875
+      mouse-cheese_dist_solvable_wavg:          9.655329
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.59950185
+    visit_patterns:
+      avg_first_visit_time:                     2009.9922
+      avg_last_visit_time:                      2349.2627
+      num_replay_batches:                       2359
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010337521
+    avg_actor_approxkl3:                        0.0007076703
+    avg_actor_clipfrac:                         0.026226807
+    avg_actor_loss:                             -0.00092725904
+    avg_advantage:                              0.039941818
+    avg_critic_clipfrac:                        0.00080566405
+    avg_critic_loss:                            0.002610637
+    avg_entropy:                                0.10616372
+    avg_grad_norm_pre_clip:                     1.1403049
+    avg_loss:                                   0.00027189584
+    max:
+      max_actor_approxkl1:                      0.0027121773
+      max_actor_approxkl3:                      0.0013112356
+      max_actor_clipfrac:                       0.045410156
+      max_actor_loss:                           0.0007091537
+      max_advantage:                            0.1984458
+      max_critic_clipfrac:                      0.0056152344
+      max_critic_loss:                          0.003697739
+      max_entropy:                              0.12933649
+      max_grad_norm_pre_clip:                   2.6668632
+      max_loss:                                 0.0021668833
+    std:
+      std_actor_approxkl1:                      0.00075768586
+      std_actor_approxkl3:                      0.00037941991
+      std_actor_clipfrac:                       0.0116280075
+      std_actor_loss:                           0.00087084586
+      std_advantage:                            0.06942525
+      std_critic_clipfrac:                      0.0019190294
+      std_critic_loss:                          0.0007482723
+      std_entropy:                              0.011960323
+      std_grad_norm_pre_clip:                   0.48079845
+      std_loss:                                 0.00090053194
+  eval-batch-train:
+    avg_avg_episode_length:                     19.746246
+    avg_avg_return:                             0.91706496
+    avg_benchmark_regret:                       0.07015211
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12770844
+    proxy_corner:
+      avg_avg_return:                           0.04694832
+      avg_reward_per_step:                      0.004928589
+  eval-batch-orig:
+    avg_avg_episode_length:                     23.804926
+    avg_avg_return:                             0.8985634
+    avg_benchmark_regret:                       0.08594814
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.083854675
+    proxy_corner:
+      avg_avg_return:                           0.8985634
+      avg_reward_per_step:                      0.083854675
+  eval-batch-shift:
+    avg_avg_episode_length:                     19.670517
+    avg_avg_return:                             0.90672565
+    avg_benchmark_regret:                       0.08078027
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.1459198
+    proxy_corner:
+      avg_avg_return:                           0.054638356
+      avg_reward_per_step:                      0.002746582
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.328941
+    avg_avg_return:                             0.89015675
+    avg_benchmark_regret:                       0.09447848
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09950256
+    proxy_corner:
+      avg_avg_return:                           0.89015675
+      avg_reward_per_step:                      0.09950256
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        153092096
+    env-step-replay-before:                     77889536
+    env-step-used-before:                       153092096
+    ppo-update-before:                          93440
+    env-step-generate-before:                   75202560
+    env-step-all-after:                         153124864
+    env-step-replay-after:                      77889536
+    env-step-used-after:                        153124864
+    ppo-update-after:                           93460
+    env-step-generate-after:                    75235328
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     23.79351
+    avg_avg_return:                             0.80582786
+    avg_reward_per_step:                        0.11148071
+    proxy_corner:
+      avg_avg_return:                           0.053176817
+      avg_reward_per_step:                      0.0018920898
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     23.79351
+    avg_avg_return:                             0.80582786
+    avg_reward_per_step:                        0.11148071
+    proxy_corner:
+      avg_avg_return:                           0.053176817
+      avg_reward_per_step:                      0.0018920898
+  perf:
+    env_steps_per_second:                       352838.6517902516
+    ppo_updates_per_second:                     42.36224979113772
+    eval-batch-train-duration:                  0.5121787274256349
+    eval-batch-orig-duration:                   0.4833918670192361
+    eval-batch-shift-duration:                  0.48264804389327765
+    eval-batch-tree-duration:                   0.4789498960599303
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.764648
+      cheese-corne_dist_wavg:                   21.6968
+      levels_cheese_is_in_corner_avg:           0.020263672
+      mouse-cheese_dist_finite_avg:             24.445557
+      mouse-cheese_dist_finite_wavg:            10.5554495
+      mouse-cheese_dist_solvable_avg:           24.445557
+      mouse-cheese_dist_solvable_wavg:          10.55545
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.5873141
+    visit_patterns:
+      avg_first_visit_time:                     2021.094
+      avg_last_visit_time:                      2367.2476
+      num_replay_batches:                       2377
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0008861838
+    avg_actor_approxkl3:                        0.00085429236
+    avg_actor_clipfrac:                         0.03284302
+    avg_actor_loss:                             -0.0003810132
+    avg_advantage:                              0.009069092
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0030617826
+    avg_entropy:                                0.11692109
+    avg_grad_norm_pre_clip:                     1.362406
+    avg_loss:                                   0.0010329568
+    max:
+      max_actor_approxkl1:                      0.0020638919
+      max_actor_approxkl3:                      0.0014673582
+      max_actor_clipfrac:                       0.057617188
+      max_actor_loss:                           0.0012003221
+      max_advantage:                            0.17557776
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.004783794
+      max_entropy:                              0.13184121
+      max_grad_norm_pre_clip:                   2.1845868
+      max_loss:                                 0.0024395015
+    std:
+      std_actor_approxkl1:                      0.0005698286
+      std_actor_approxkl3:                      0.0004445106
+      std_actor_clipfrac:                       0.013983924
+      std_actor_loss:                           0.0009952957
+      std_advantage:                            0.078098394
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00084466225
+      std_entropy:                              0.00937341
+      std_grad_norm_pre_clip:                   0.44690472
+      std_loss:                                 0.00089356897
+  eval-batch-train:
+    avg_avg_episode_length:                     20.681997
+    avg_avg_return:                             0.8981366
+    avg_benchmark_regret:                       0.08908041
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12484741
+    proxy_corner:
+      avg_avg_return:                           0.061840776
+      avg_reward_per_step:                      0.005241394
+  eval-batch-orig:
+    avg_avg_episode_length:                     25.722652
+    avg_avg_return:                             0.8776432
+    avg_benchmark_regret:                       0.1068683
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08268738
+    proxy_corner:
+      avg_avg_return:                           0.8776432
+      avg_reward_per_step:                      0.08268738
+  eval-batch-shift:
+    avg_avg_episode_length:                     19.152954
+    avg_avg_return:                             0.91488385
+    avg_benchmark_regret:                       0.07262215
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14407349
+    proxy_corner:
+      avg_avg_return:                           0.058092568
+      avg_reward_per_step:                      0.0028533936
+  eval-batch-tree:
+    avg_avg_episode_length:                     24.562817
+    avg_avg_return:                             0.8769813
+    avg_benchmark_regret:                       0.10765387
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.097579956
+    proxy_corner:
+      avg_avg_return:                           0.8769813
+      avg_reward_per_step:                      0.097579956
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        154140672
+    env-step-replay-before:                     78544896
+    env-step-used-before:                       154140672
+    ppo-update-before:                          94080
+    env-step-generate-before:                   75595776
+    env-step-all-after:                         154173440
+    env-step-replay-after:                      78577664
+    env-step-used-after:                        154173440
+    ppo-update-after:                           94100
+    env-step-generate-after:                    75595776
+  train-replay:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     68.84085
+    avg_avg_return:                             0.40811765
+    avg_reward_per_step:                        0.010131836
+    proxy_corner:
+      avg_avg_return:                           0.12567422
+      avg_reward_per_step:                      0.002166748
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     68.84085
+    avg_avg_return:                             0.40811765
+    avg_reward_per_step:                        0.010131836
+    proxy_corner:
+      avg_avg_return:                           0.12567422
+      avg_reward_per_step:                      0.002166748
+  perf:
+    env_steps_per_second:                       354170.5006596839
+    ppo_updates_per_second:                     41.22575739511156
+    eval-batch-train-duration:                  0.5362159376963973
+    eval-batch-orig-duration:                   0.4774518283084035
+    eval-batch-shift-duration:                  0.45829709712415934
+    eval-batch-tree-duration:                   0.47070724330842495
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.714111
+      cheese-corne_dist_wavg:                   21.663984
+      levels_cheese_is_in_corner_avg:           0.019042969
+      mouse-cheese_dist_finite_avg:             24.440186
+      mouse-cheese_dist_finite_wavg:            11.446865
+      mouse-cheese_dist_solvable_avg:           24.440186
+      mouse-cheese_dist_solvable_wavg:          11.446866
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.5768868
+    visit_patterns:
+      avg_first_visit_time:                     2035.3691
+      avg_last_visit_time:                      2388.2432
+      num_replay_batches:                       2398
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001154499
+    avg_actor_approxkl3:                        0.0011814925
+    avg_actor_clipfrac:                         0.05480957
+    avg_actor_loss:                             -0.0006579623
+    avg_advantage:                              -0.038789928
+    avg_critic_clipfrac:                        6.1035157e-06
+    avg_critic_loss:                            0.010310635
+    avg_entropy:                                0.25504205
+    avg_grad_norm_pre_clip:                     1.7660719
+    avg_loss:                                   0.0042423136
+    max:
+      max_actor_approxkl1:                      0.0032973697
+      max_actor_approxkl3:                      0.0018584561
+      max_actor_clipfrac:                       0.08203125
+      max_actor_loss:                           0.002132684
+      max_advantage:                            0.15311503
+      max_critic_clipfrac:                      0.00012207031
+      max_critic_loss:                          0.01397731
+      max_entropy:                              0.27432403
+      max_grad_norm_pre_clip:                   3.1200666
+      max_loss:                                 0.0071313055
+    std:
+      std_actor_approxkl1:                      0.00082404225
+      std_actor_approxkl3:                      0.0004680758
+      std_actor_clipfrac:                       0.019335413
+      std_actor_loss:                           0.0012182486
+      std_advantage:                            0.14249656
+      std_critic_clipfrac:                      2.6604606e-05
+      std_critic_loss:                          0.0015030153
+      std_entropy:                              0.008364541
+      std_grad_norm_pre_clip:                   0.55184805
+      std_loss:                                 0.0013537195
+  eval-batch-train:
+    avg_avg_episode_length:                     20.072544
+    avg_avg_return:                             0.9070474
+    avg_benchmark_regret:                       0.08016965
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12471008
+    proxy_corner:
+      avg_avg_return:                           0.056271374
+      avg_reward_per_step:                      0.0050811768
+  eval-batch-orig:
+    avg_avg_episode_length:                     24.658058
+    avg_avg_return:                             0.88682544
+    avg_benchmark_regret:                       0.09768611
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.079582214
+    proxy_corner:
+      avg_avg_return:                           0.88682544
+      avg_reward_per_step:                      0.079582214
+  eval-batch-shift:
+    avg_avg_episode_length:                     19.294617
+    avg_avg_return:                             0.91762316
+    avg_benchmark_regret:                       0.069882795
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14304352
+    proxy_corner:
+      avg_avg_return:                           0.053097736
+      avg_reward_per_step:                      0.0027770996
+  eval-batch-tree:
+    avg_avg_episode_length:                     24.055077
+    avg_avg_return:                             0.88301
+    avg_benchmark_regret:                       0.10162519
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09712982
+    proxy_corner:
+      avg_avg_return:                           0.88301
+      avg_reward_per_step:                      0.09712982
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        155189248
+    env-step-replay-before:                     79134720
+    env-step-used-before:                       155189248
+    ppo-update-before:                          94720
+    env-step-generate-before:                   76054528
+    env-step-all-after:                         155222016
+    env-step-replay-after:                      79134720
+    env-step-used-after:                        155222016
+    ppo-update-after:                           94740
+    env-step-generate-after:                    76087296
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     22.722988
+    avg_avg_return:                             0.81036544
+    avg_reward_per_step:                        0.12634277
+    proxy_corner:
+      avg_avg_return:                           0.02794607
+      avg_reward_per_step:                      0.0013427734
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     22.722988
+    avg_avg_return:                             0.81036544
+    avg_reward_per_step:                        0.12634277
+    proxy_corner:
+      avg_avg_return:                           0.02794607
+      avg_reward_per_step:                      0.0013427734
+  perf:
+    env_steps_per_second:                       348228.5134479721
+    ppo_updates_per_second:                     42.23471782821295
+    eval-batch-train-duration:                  0.4862803444266319
+    eval-batch-orig-duration:                   0.424981358461082
+    eval-batch-shift-duration:                  0.45070600509643555
+    eval-batch-tree-duration:                   0.4257441973313689
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.664307
+      cheese-corne_dist_wavg:                   21.68419
+      levels_cheese_is_in_corner_avg:           0.020263672
+      mouse-cheese_dist_finite_avg:             24.490723
+      mouse-cheese_dist_finite_wavg:            11.435734
+      mouse-cheese_dist_solvable_avg:           24.490723
+      mouse-cheese_dist_solvable_wavg:          11.435735
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.57479846
+    visit_patterns:
+      avg_first_visit_time:                     2049.285
+      avg_last_visit_time:                      2405.2756
+      num_replay_batches:                       2415
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009321143
+    avg_actor_approxkl3:                        0.00086901704
+    avg_actor_clipfrac:                         0.031121826
+    avg_actor_loss:                             -0.00062177423
+    avg_advantage:                              -0.00152119
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0031201262
+    avg_entropy:                                0.10401287
+    avg_grad_norm_pre_clip:                     1.1628244
+    avg_loss:                                   0.0008342759
+    max:
+      max_actor_approxkl1:                      0.0028276262
+      max_actor_approxkl3:                      0.0016240173
+      max_actor_clipfrac:                       0.044921875
+      max_actor_loss:                           0.0019615889
+      max_advantage:                            0.14432907
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.004467306
+      max_entropy:                              0.11693326
+      max_grad_norm_pre_clip:                   2.0242634
+      max_loss:                                 0.003330814
+    std:
+      std_actor_approxkl1:                      0.00077667495
+      std_actor_approxkl3:                      0.0003813426
+      std_actor_clipfrac:                       0.010228041
+      std_actor_loss:                           0.0010922792
+      std_advantage:                            0.079393655
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0007125659
+      std_entropy:                              0.0070338845
+      std_grad_norm_pre_clip:                   0.34311932
+      std_loss:                                 0.001064096
+  eval-batch-train:
+    avg_avg_episode_length:                     19.57116
+    avg_avg_return:                             0.91392213
+    avg_benchmark_regret:                       0.073294945
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12581635
+    proxy_corner:
+      avg_avg_return:                           0.051131938
+      avg_reward_per_step:                      0.0049743652
+  eval-batch-orig:
+    avg_avg_episode_length:                     24.351704
+    avg_avg_return:                             0.88674176
+    avg_benchmark_regret:                       0.09776981
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.0838089
+    proxy_corner:
+      avg_avg_return:                           0.88674176
+      avg_reward_per_step:                      0.0838089
+  eval-batch-shift:
+    avg_avg_episode_length:                     19.208725
+    avg_avg_return:                             0.9134095
+    avg_benchmark_regret:                       0.0740965
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14575958
+    proxy_corner:
+      avg_avg_return:                           0.052068174
+      avg_reward_per_step:                      0.0028381348
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.578966
+    avg_avg_return:                             0.884182
+    avg_benchmark_regret:                       0.10045321
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09918976
+    proxy_corner:
+      avg_avg_return:                           0.884182
+      avg_reward_per_step:                      0.09918976
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        156237824
+    env-step-replay-before:                     79462400
+    env-step-used-before:                       156237824
+    ppo-update-before:                          95360
+    env-step-generate-before:                   76775424
+    env-step-all-after:                         156270592
+    env-step-replay-after:                      79495168
+    env-step-used-after:                        156270592
+    ppo-update-after:                           95380
+    env-step-generate-after:                    76775424
+  train-replay:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     69.61834
+    avg_avg_return:                             0.3804614
+    avg_reward_per_step:                        0.010070801
+    proxy_corner:
+      avg_avg_return:                           0.08946465
+      avg_reward_per_step:                      0.0018310547
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     69.61834
+    avg_avg_return:                             0.3804614
+    avg_reward_per_step:                        0.010070801
+    proxy_corner:
+      avg_avg_return:                           0.08946465
+      avg_reward_per_step:                      0.0018310547
+  perf:
+    env_steps_per_second:                       380000.85310101515
+    ppo_updates_per_second:                     42.15085449736329
+    eval-batch-train-duration:                  0.4953086609020829
+    eval-batch-orig-duration:                   0.4250210551545024
+    eval-batch-shift-duration:                  0.44761030562222004
+    eval-batch-tree-duration:                   0.4363469146192074
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.665771
+      cheese-corne_dist_wavg:                   12.675463
+      levels_cheese_is_in_corner_avg:           0.021728516
+      mouse-cheese_dist_finite_avg:             24.579346
+      mouse-cheese_dist_finite_wavg:            12.356028
+      mouse-cheese_dist_solvable_avg:           24.579346
+      mouse-cheese_dist_solvable_wavg:          12.356026
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.420168
+    scoring:
+      avg_scores:                               0.58746195
+    visit_patterns:
+      avg_first_visit_time:                     2065.793
+      avg_last_visit_time:                      2416.255
+      num_replay_batches:                       2426
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00083388755
+    avg_actor_approxkl3:                        0.00089863193
+    avg_actor_clipfrac:                         0.043737795
+    avg_actor_loss:                             -0.0012054377
+    avg_advantage:                              -0.04442135
+    avg_critic_clipfrac:                        0.00056762696
+    avg_critic_loss:                            0.011604513
+    avg_entropy:                                0.25510928
+    avg_grad_norm_pre_clip:                     1.5552139
+    avg_loss:                                   0.00434171
+    max:
+      max_actor_approxkl1:                      0.00234029
+      max_actor_approxkl3:                      0.0020211907
+      max_actor_clipfrac:                       0.08679199
+      max_actor_loss:                           0.0005687475
+      max_advantage:                            0.1608954
+      max_critic_clipfrac:                      0.0075683594
+      max_critic_loss:                          0.01350811
+      max_entropy:                              0.27068418
+      max_grad_norm_pre_clip:                   2.6993384
+      max_loss:                                 0.0069434
+    std:
+      std_actor_approxkl1:                      0.0007014819
+      std_actor_approxkl3:                      0.00052525493
+      std_actor_clipfrac:                       0.023491768
+      std_actor_loss:                           0.0010926115
+      std_advantage:                            0.15107858
+      std_critic_clipfrac:                      0.0016880498
+      std_critic_loss:                          0.0010625075
+      std_entropy:                              0.007430358
+      std_grad_norm_pre_clip:                   0.50968665
+      std_loss:                                 0.0012511078
+  eval-batch-train:
+    avg_avg_episode_length:                     19.990444
+    avg_avg_return:                             0.90591943
+    avg_benchmark_regret:                       0.08129761
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12523651
+    proxy_corner:
+      avg_avg_return:                           0.05399853
+      avg_reward_per_step:                      0.0051956177
+  eval-batch-orig:
+    avg_avg_episode_length:                     24.243698
+    avg_avg_return:                             0.8848927
+    avg_benchmark_regret:                       0.09961882
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08415222
+    proxy_corner:
+      avg_avg_return:                           0.8848927
+      avg_reward_per_step:                      0.08415222
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.801517
+    avg_avg_return:                             0.919543
+    avg_benchmark_regret:                       0.06796289
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.1434021
+    proxy_corner:
+      avg_avg_return:                           0.05310617
+      avg_reward_per_step:                      0.0028305054
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.681759
+    avg_avg_return:                             0.8940077
+    avg_benchmark_regret:                       0.09062749
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09890747
+    proxy_corner:
+      avg_avg_return:                           0.8940077
+      avg_reward_per_step:                      0.09890747
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        157286400
+    env-step-replay-before:                     80084992
+    env-step-used-before:                       157286400
+    ppo-update-before:                          96000
+    env-step-generate-before:                   77201408
+    env-step-all-after:                         157319168
+    env-step-replay-after:                      80117760
+    env-step-used-after:                        157319168
+    ppo-update-after:                           96020
+    env-step-generate-after:                    77201408
+  train-replay:
+    shift_proportion:                           0.96875
+    avg_avg_episode_length:                     69.32753
+    avg_avg_return:                             0.40585315
+    avg_reward_per_step:                        0.010528564
+    proxy_corner:
+      avg_avg_return:                           0.11991282
+      avg_reward_per_step:                      0.0022277832
+  train-all:
+    shift_proportion:                           0.96875
+    avg_avg_episode_length:                     69.32753
+    avg_avg_return:                             0.40585315
+    avg_reward_per_step:                        0.010528564
+    proxy_corner:
+      avg_avg_return:                           0.11991282
+      avg_reward_per_step:                      0.0022277832
+  perf:
+    env_steps_per_second:                       348964.3393521706
+    ppo_updates_per_second:                     41.27466087509797
+    eval-batch-train-duration:                  0.5383988181129098
+    eval-batch-orig-duration:                   0.4911777740344405
+    eval-batch-shift-duration:                  0.482696202583611
+    eval-batch-tree-duration:                   0.4557063411921263
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.67334
+      cheese-corne_dist_wavg:                   9.980174
+      levels_cheese_is_in_corner_avg:           0.022216797
+      mouse-cheese_dist_finite_avg:             24.70166
+      mouse-cheese_dist_finite_wavg:            12.373363
+      mouse-cheese_dist_solvable_avg:           24.70166
+      mouse-cheese_dist_solvable_wavg:          12.373363
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016798
+    scoring:
+      avg_scores:                               0.5887936
+    visit_patterns:
+      avg_first_visit_time:                     2078.7668
+      avg_last_visit_time:                      2435.2334
+      num_replay_batches:                       2445
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00092157174
+    avg_actor_approxkl3:                        0.0008896974
+    avg_actor_clipfrac:                         0.044366457
+    avg_actor_loss:                             -0.00080536975
+    avg_advantage:                              -0.036935054
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.010431127
+    avg_entropy:                                0.25666448
+    avg_grad_norm_pre_clip:                     1.8149595
+    avg_loss:                                   0.004153528
+    max:
+      max_actor_approxkl1:                      0.0016349402
+      max_actor_approxkl3:                      0.0015153561
+      max_actor_clipfrac:                       0.07348633
+      max_actor_loss:                           0.001583755
+      max_advantage:                            0.20748883
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.013125147
+      max_entropy:                              0.27109945
+      max_grad_norm_pre_clip:                   2.8411913
+      max_loss:                                 0.0076324055
+    std:
+      std_actor_approxkl1:                      0.0004615087
+      std_actor_approxkl3:                      0.00037695488
+      std_actor_clipfrac:                       0.01840602
+      std_actor_loss:                           0.0011021357
+      std_advantage:                            0.14368601
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0014124006
+      std_entropy:                              0.007931492
+      std_grad_norm_pre_clip:                   0.45835134
+      std_loss:                                 0.0013676735
+  eval-batch-train:
+    avg_avg_episode_length:                     19.613327
+    avg_avg_return:                             0.9090735
+    avg_benchmark_regret:                       0.078143574
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12675476
+    proxy_corner:
+      avg_avg_return:                           0.047926456
+      avg_reward_per_step:                      0.0049362183
+  eval-batch-orig:
+    avg_avg_episode_length:                     24.248434
+    avg_avg_return:                             0.8845686
+    avg_benchmark_regret:                       0.099942975
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.082481384
+    proxy_corner:
+      avg_avg_return:                           0.8845686
+      avg_reward_per_step:                      0.082481384
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.975777
+    avg_avg_return:                             0.91685724
+    avg_benchmark_regret:                       0.07064867
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14643097
+    proxy_corner:
+      avg_avg_return:                           0.053894117
+      avg_reward_per_step:                      0.0028305054
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.122871
+    avg_avg_return:                             0.88995355
+    avg_benchmark_regret:                       0.09468166
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09807587
+    proxy_corner:
+      avg_avg_return:                           0.88995355
+      avg_reward_per_step:                      0.09807587
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        158334976
+    env-step-replay-before:                     80642048
+    env-step-used-before:                       158334976
+    ppo-update-before:                          96640
+    env-step-generate-before:                   77692928
+    env-step-all-after:                         158367744
+    env-step-replay-after:                      80642048
+    env-step-used-after:                        158367744
+    ppo-update-after:                           96660
+    env-step-generate-after:                    77725696
+  train-generate:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     23.774971
+    avg_avg_return:                             0.80792224
+    avg_reward_per_step:                        0.09585571
+    proxy_corner:
+      avg_avg_return:                           0.06270538
+      avg_reward_per_step:                      0.0044555664
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     23.774971
+    avg_avg_return:                             0.80792224
+    avg_reward_per_step:                        0.09585571
+    proxy_corner:
+      avg_avg_return:                           0.06270538
+      avg_reward_per_step:                      0.0044555664
+  perf:
+    env_steps_per_second:                       319882.54449610144
+    ppo_updates_per_second:                     41.62314313843909
+    eval-batch-train-duration:                  0.5252352152019739
+    eval-batch-orig-duration:                   0.46438848692923784
+    eval-batch-shift-duration:                  0.4209839329123497
+    eval-batch-tree-duration:                   0.4123166911303997
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.655273
+      cheese-corne_dist_wavg:                   19.856281
+      levels_cheese_is_in_corner_avg:           0.021728516
+      mouse-cheese_dist_finite_avg:             24.634033
+      mouse-cheese_dist_finite_wavg:            12.361509
+      mouse-cheese_dist_solvable_avg:           24.634033
+      mouse-cheese_dist_solvable_wavg:          12.361511
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.420168
+    scoring:
+      avg_scores:                               0.5787233
+    visit_patterns:
+      avg_first_visit_time:                     2096.0361
+      avg_last_visit_time:                      2451.171
+      num_replay_batches:                       2461
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00048667387
+    avg_actor_approxkl3:                        0.00053107005
+    avg_actor_clipfrac:                         0.021813964
+    avg_actor_loss:                             -0.0012222147
+    avg_advantage:                              0.004243094
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0024725525
+    avg_entropy:                                0.102721624
+    avg_grad_norm_pre_clip:                     0.9655178
+    avg_loss:                                   -8.866026e-05
+    max:
+      max_actor_approxkl1:                      0.001895082
+      max_actor_approxkl3:                      0.0011161184
+      max_actor_clipfrac:                       0.041137695
+      max_actor_loss:                           0.00043326616
+      max_advantage:                            0.15756625
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0034177832
+      max_entropy:                              0.13450843
+      max_grad_norm_pre_clip:                   1.381953
+      max_loss:                                 0.0014706793
+    std:
+      std_actor_approxkl1:                      0.0005941754
+      std_actor_approxkl3:                      0.00032626613
+      std_actor_clipfrac:                       0.010267782
+      std_actor_loss:                           0.0009052282
+      std_advantage:                            0.070479795
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0007552836
+      std_entropy:                              0.013403718
+      std_grad_norm_pre_clip:                   0.22798903
+      std_loss:                                 0.00088600186
+  eval-batch-train:
+    avg_avg_episode_length:                     20.12019
+    avg_avg_return:                             0.9078437
+    avg_benchmark_regret:                       0.07937337
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12699127
+    proxy_corner:
+      avg_avg_return:                           0.05413544
+      avg_reward_per_step:                      0.0051651
+  eval-batch-orig:
+    avg_avg_episode_length:                     24.66595
+    avg_avg_return:                             0.88521624
+    avg_benchmark_regret:                       0.09929527
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.082984924
+    proxy_corner:
+      avg_avg_return:                           0.88521624
+      avg_reward_per_step:                      0.082984924
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.841677
+    avg_avg_return:                             0.9195321
+    avg_benchmark_regret:                       0.06797385
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14641571
+    proxy_corner:
+      avg_avg_return:                           0.056784667
+      avg_reward_per_step:                      0.002998352
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.440872
+    avg_avg_return:                             0.8883815
+    avg_benchmark_regret:                       0.09625374
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.098526
+    proxy_corner:
+      avg_avg_return:                           0.8883815
+      avg_reward_per_step:                      0.098526
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        159383552
+    env-step-replay-before:                     81133568
+    env-step-used-before:                       159383552
+    ppo-update-before:                          97280
+    env-step-generate-before:                   78249984
+    env-step-all-after:                         159416320
+    env-step-replay-after:                      81133568
+    env-step-used-after:                        159416320
+    ppo-update-after:                           97300
+    env-step-generate-after:                    78282752
+  train-generate:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     23.544664
+    avg_avg_return:                             0.8079494
+    avg_reward_per_step:                        0.13027954
+    proxy_corner:
+      avg_avg_return:                           0.06554232
+      avg_reward_per_step:                      0.0030822754
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     23.544664
+    avg_avg_return:                             0.8079494
+    avg_reward_per_step:                        0.13027954
+    proxy_corner:
+      avg_avg_return:                           0.06554232
+      avg_reward_per_step:                      0.0030822754
+  perf:
+    env_steps_per_second:                       295045.1809053112
+    ppo_updates_per_second:                     42.48462924085045
+    eval-batch-train-duration:                  0.5349797289818525
+    eval-batch-orig-duration:                   0.5043053459376097
+    eval-batch-shift-duration:                  0.3112707044929266
+    eval-batch-tree-duration:                   0.25571716856211424
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.563965
+      cheese-corne_dist_wavg:                   21.637804
+      levels_cheese_is_in_corner_avg:           0.021240234
+      mouse-cheese_dist_finite_avg:             24.6604
+      mouse-cheese_dist_finite_wavg:            10.550971
+      mouse-cheese_dist_solvable_avg:           24.6604
+      mouse-cheese_dist_solvable_wavg:          10.550972
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.99999
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.59362507
+    visit_patterns:
+      avg_first_visit_time:                     2113.9863
+      avg_last_visit_time:                      2466.3071
+      num_replay_batches:                       2476
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00091986323
+    avg_actor_approxkl3:                        0.0006595489
+    avg_actor_clipfrac:                         0.02331543
+    avg_actor_loss:                             -0.0012121387
+    avg_advantage:                              0.008861795
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.002789392
+    avg_entropy:                                0.09775873
+    avg_grad_norm_pre_clip:                     0.9972011
+    avg_loss:                                   8.47986e-05
+    max:
+      max_actor_approxkl1:                      0.004960615
+      max_actor_approxkl3:                      0.0014151775
+      max_actor_clipfrac:                       0.041992188
+      max_actor_loss:                           0.00043875352
+      max_advantage:                            0.15401876
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0044242
+      max_entropy:                              0.1314701
+      max_grad_norm_pre_clip:                   1.9962513
+      max_loss:                                 0.0020944627
+    std:
+      std_actor_approxkl1:                      0.0011273283
+      std_actor_approxkl3:                      0.00037994797
+      std_actor_clipfrac:                       0.010364553
+      std_actor_loss:                           0.0012574914
+      std_advantage:                            0.075060375
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00088899286
+      std_entropy:                              0.013599928
+      std_grad_norm_pre_clip:                   0.38731307
+      std_loss:                                 0.0013611807
+  eval-batch-train:
+    avg_avg_episode_length:                     19.603014
+    avg_avg_return:                             0.906343
+    avg_benchmark_regret:                       0.08087408
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12815094
+    proxy_corner:
+      avg_avg_return:                           0.04691989
+      avg_reward_per_step:                      0.004875183
+  eval-batch-orig:
+    avg_avg_episode_length:                     23.307781
+    avg_avg_return:                             0.89648056
+    avg_benchmark_regret:                       0.08803097
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08438873
+    proxy_corner:
+      avg_avg_return:                           0.89648056
+      avg_reward_per_step:                      0.08438873
+  eval-batch-shift:
+    avg_avg_episode_length:                     19.302929
+    avg_avg_return:                             0.9180013
+    avg_benchmark_regret:                       0.069504604
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14531708
+    proxy_corner:
+      avg_avg_return:                           0.050164584
+      avg_reward_per_step:                      0.002632141
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.376297
+    avg_avg_return:                             0.8872709
+    avg_benchmark_regret:                       0.09736428
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09817505
+    proxy_corner:
+      avg_avg_return:                           0.8872709
+      avg_reward_per_step:                      0.09817505
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        160432128
+    env-step-replay-before:                     81625088
+    env-step-used-before:                       160432128
+    ppo-update-before:                          97920
+    env-step-generate-before:                   78807040
+    env-step-all-after:                         160464896
+    env-step-replay-after:                      81657856
+    env-step-used-after:                        160464896
+    ppo-update-after:                           97940
+    env-step-generate-after:                    78807040
+  train-replay:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     68.82699
+    avg_avg_return:                             0.40770444
+    avg_reward_per_step:                        0.010864258
+    proxy_corner:
+      avg_avg_return:                           0.085819304
+      avg_reward_per_step:                      0.001739502
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     68.82699
+    avg_avg_return:                             0.40770444
+    avg_reward_per_step:                        0.010864258
+    proxy_corner:
+      avg_avg_return:                           0.085819304
+      avg_reward_per_step:                      0.001739502
+  perf:
+    env_steps_per_second:                       318565.21938051924
+    ppo_updates_per_second:                     41.849413543868884
+    eval-batch-train-duration:                  0.5378682427108288
+    eval-batch-orig-duration:                   0.4829705925658345
+    eval-batch-shift-duration:                  0.4784804303199053
+    eval-batch-tree-duration:                   0.4907974721863866
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.533691
+      cheese-corne_dist_wavg:                   21.643744
+      levels_cheese_is_in_corner_avg:           0.022705078
+      mouse-cheese_dist_finite_avg:             24.646729
+      mouse-cheese_dist_finite_wavg:            10.557689
+      mouse-cheese_dist_solvable_avg:           24.646729
+      mouse-cheese_dist_solvable_wavg:          10.55769
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5842732
+    visit_patterns:
+      avg_first_visit_time:                     2126.5764
+      avg_last_visit_time:                      2482.2185
+      num_replay_batches:                       2492
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012373206
+    avg_actor_approxkl3:                        0.0011719947
+    avg_actor_clipfrac:                         0.05232544
+    avg_actor_loss:                             -0.001048487
+    avg_advantage:                              -0.03809826
+    avg_critic_clipfrac:                        6.1035156e-05
+    avg_critic_loss:                            0.010360979
+    avg_entropy:                                0.23347913
+    avg_grad_norm_pre_clip:                     1.5107307
+    avg_loss:                                   0.003898523
+    max:
+      max_actor_approxkl1:                      0.0023087282
+      max_actor_approxkl3:                      0.0022189785
+      max_actor_clipfrac:                       0.080444336
+      max_actor_loss:                           0.00088092685
+      max_advantage:                            0.18505293
+      max_critic_clipfrac:                      0.00036621094
+      max_critic_loss:                          0.01189479
+      max_entropy:                              0.2538683
+      max_grad_norm_pre_clip:                   2.4377692
+      max_loss:                                 0.005839571
+    std:
+      std_actor_approxkl1:                      0.00061391084
+      std_actor_approxkl3:                      0.00055659213
+      std_actor_clipfrac:                       0.021789953
+      std_actor_loss:                           0.0010821028
+      std_advantage:                            0.14293736
+      std_critic_clipfrac:                      0.00012508484
+      std_critic_loss:                          0.0008915828
+      std_entropy:                              0.013627053
+      std_grad_norm_pre_clip:                   0.4610901
+      std_loss:                                 0.0010535617
+  eval-batch-train:
+    avg_avg_episode_length:                     19.298607
+    avg_avg_return:                             0.91447103
+    avg_benchmark_regret:                       0.07274604
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12916565
+    proxy_corner:
+      avg_avg_return:                           0.04252081
+      avg_reward_per_step:                      0.004676819
+  eval-batch-orig:
+    avg_avg_episode_length:                     23.851532
+    avg_avg_return:                             0.8932525
+    avg_benchmark_regret:                       0.09125903
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08415985
+    proxy_corner:
+      avg_avg_return:                           0.8932525
+      avg_reward_per_step:                      0.08415985
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.823925
+    avg_avg_return:                             0.91651154
+    avg_benchmark_regret:                       0.0709944
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.1456604
+    proxy_corner:
+      avg_avg_return:                           0.048236348
+      avg_reward_per_step:                      0.002609253
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.347101
+    avg_avg_return:                             0.8897971
+    avg_benchmark_regret:                       0.094838075
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09955597
+    proxy_corner:
+      avg_avg_return:                           0.8897971
+      avg_reward_per_step:                      0.09955597
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        161480704
+    env-step-replay-before:                     82149376
+    env-step-used-before:                       161480704
+    ppo-update-before:                          98560
+    env-step-generate-before:                   79331328
+    env-step-all-after:                         161513472
+    env-step-replay-after:                      82149376
+    env-step-used-after:                        161513472
+    ppo-update-after:                           98580
+    env-step-generate-after:                    79364096
+  train-generate:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     19.670807
+    avg_avg_return:                             0.8434158
+    avg_reward_per_step:                        0.13546753
+    proxy_corner:
+      avg_avg_return:                           0.047944367
+      avg_reward_per_step:                      0.0030822754
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     19.670807
+    avg_avg_return:                             0.8434158
+    avg_reward_per_step:                        0.13546753
+    proxy_corner:
+      avg_avg_return:                           0.047944367
+      avg_reward_per_step:                      0.0030822754
+  perf:
+    env_steps_per_second:                       288337.75434937014
+    ppo_updates_per_second:                     41.569003728242485
+    eval-batch-train-duration:                  0.5328888203948736
+    eval-batch-orig-duration:                   0.4861694686114788
+    eval-batch-shift-duration:                  0.47579717729240656
+    eval-batch-tree-duration:                   0.45244363509118557
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.57788
+      cheese-corne_dist_wavg:                   13.557042
+      levels_cheese_is_in_corner_avg:           0.021484375
+      mouse-cheese_dist_finite_avg:             24.70044
+      mouse-cheese_dist_finite_wavg:            11.476717
+      mouse-cheese_dist_solvable_avg:           24.70044
+      mouse-cheese_dist_solvable_wavg:          11.476717
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.5985397
+    visit_patterns:
+      avg_first_visit_time:                     2145.2559
+      avg_last_visit_time:                      2497.3132
+      num_replay_batches:                       2507
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013049081
+    avg_actor_approxkl3:                        0.00095383276
+    avg_actor_clipfrac:                         0.028692627
+    avg_actor_loss:                             -0.00019098837
+    avg_advantage:                              0.038674466
+    avg_critic_clipfrac:                        0.00079956057
+    avg_critic_loss:                            0.0022943353
+    avg_entropy:                                0.084810056
+    avg_grad_norm_pre_clip:                     1.518915
+    avg_loss:                                   0.00087136915
+    max:
+      max_actor_approxkl1:                      0.00425396
+      max_actor_approxkl3:                      0.0027295894
+      max_actor_clipfrac:                       0.052368164
+      max_actor_loss:                           0.0021443516
+      max_advantage:                            0.18776351
+      max_critic_clipfrac:                      0.011352539
+      max_critic_loss:                          0.003083649
+      max_entropy:                              0.09574853
+      max_grad_norm_pre_clip:                   2.3313448
+      max_loss:                                 0.0035279437
+    std:
+      std_actor_approxkl1:                      0.0010457891
+      std_actor_approxkl3:                      0.0006081141
+      std_actor_clipfrac:                       0.011145257
+      std_actor_loss:                           0.0010851725
+      std_advantage:                            0.06545581
+      std_critic_clipfrac:                      0.002525845
+      std_critic_loss:                          0.00045425823
+      std_entropy:                              0.006130012
+      std_grad_norm_pre_clip:                   0.50698465
+      std_loss:                                 0.0011383904
+  eval-batch-train:
+    avg_avg_episode_length:                     19.869923
+    avg_avg_return:                             0.9052385
+    avg_benchmark_regret:                       0.081978515
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12793732
+    proxy_corner:
+      avg_avg_return:                           0.056220997
+      avg_reward_per_step:                      0.005180359
+  eval-batch-orig:
+    avg_avg_episode_length:                     24.514645
+    avg_avg_return:                             0.8783114
+    avg_benchmark_regret:                       0.10620017
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08761597
+    proxy_corner:
+      avg_avg_return:                           0.8783114
+      avg_reward_per_step:                      0.08761597
+  eval-batch-shift:
+    avg_avg_episode_length:                     19.62431
+    avg_avg_return:                             0.90723366
+    avg_benchmark_regret:                       0.080272265
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14568329
+    proxy_corner:
+      avg_avg_return:                           0.05680965
+      avg_reward_per_step:                      0.0030441284
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.479671
+    avg_avg_return:                             0.88541937
+    avg_benchmark_regret:                       0.09921586
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.10192108
+    proxy_corner:
+      avg_avg_return:                           0.88541937
+      avg_reward_per_step:                      0.10192108
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        162529280
+    env-step-replay-before:                     82542592
+    env-step-used-before:                       162529280
+    ppo-update-before:                          99200
+    env-step-generate-before:                   79986688
+    env-step-all-after:                         162562048
+    env-step-replay-after:                      82542592
+    env-step-used-after:                        162562048
+    ppo-update-after:                           99220
+    env-step-generate-after:                    80019456
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     22.30737
+    avg_avg_return:                             0.8194891
+    avg_reward_per_step:                        0.0982666
+    proxy_corner:
+      avg_avg_return:                           0.056614697
+      avg_reward_per_step:                      0.0024719238
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     22.30737
+    avg_avg_return:                             0.8194891
+    avg_reward_per_step:                        0.0982666
+    proxy_corner:
+      avg_avg_return:                           0.056614697
+      avg_reward_per_step:                      0.0024719238
+  perf:
+    env_steps_per_second:                       351815.6811785301
+    ppo_updates_per_second:                     41.37648592064223
+    eval-batch-train-duration:                  0.5322129409760237
+    eval-batch-orig-duration:                   0.4720652811229229
+    eval-batch-shift-duration:                  0.4521658029407263
+    eval-batch-tree-duration:                   0.49832230154424906
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.49414
+      cheese-corne_dist_wavg:                   25.225607
+      levels_cheese_is_in_corner_avg:           0.021728516
+      mouse-cheese_dist_finite_avg:             24.704346
+      mouse-cheese_dist_finite_wavg:            12.372198
+      mouse-cheese_dist_solvable_avg:           24.704346
+      mouse-cheese_dist_solvable_wavg:          12.3722
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.60311425
+    visit_patterns:
+      avg_first_visit_time:                     2160.4138
+      avg_last_visit_time:                      2509.302
+      num_replay_batches:                       2519
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00078440737
+    avg_actor_approxkl3:                        0.0009468693
+    avg_actor_clipfrac:                         0.029827882
+    avg_actor_loss:                             -0.00043078838
+    avg_advantage:                              0.0045110984
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0022874454
+    avg_entropy:                                0.093824
+    avg_grad_norm_pre_clip:                     1.2260219
+    avg_loss:                                   0.00061911036
+    max:
+      max_actor_approxkl1:                      0.0019744074
+      max_actor_approxkl3:                      0.0018811219
+      max_actor_clipfrac:                       0.049804688
+      max_actor_loss:                           0.0014799349
+      max_advantage:                            0.14254785
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.003499799
+      max_entropy:                              0.10847786
+      max_grad_norm_pre_clip:                   2.2452264
+      max_loss:                                 0.002528071
+    std:
+      std_actor_approxkl1:                      0.00060012133
+      std_actor_approxkl3:                      0.0005592865
+      std_actor_clipfrac:                       0.0124350805
+      std_actor_loss:                           0.000847446
+      std_advantage:                            0.06743665
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0006697121
+      std_entropy:                              0.008589017
+      std_grad_norm_pre_clip:                   0.4950593
+      std_loss:                                 0.00081196666
+  eval-batch-train:
+    avg_avg_episode_length:                     19.668629
+    avg_avg_return:                             0.90966225
+    avg_benchmark_regret:                       0.07755479
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12742615
+    proxy_corner:
+      avg_avg_return:                           0.055794045
+      avg_reward_per_step:                      0.005180359
+  eval-batch-orig:
+    avg_avg_episode_length:                     23.496742
+    avg_avg_return:                             0.8954512
+    avg_benchmark_regret:                       0.08906038
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08648682
+    proxy_corner:
+      avg_avg_return:                           0.8954512
+      avg_reward_per_step:                      0.08648682
+  eval-batch-shift:
+    avg_avg_episode_length:                     19.195986
+    avg_avg_return:                             0.9122881
+    avg_benchmark_regret:                       0.07521783
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.1460495
+    proxy_corner:
+      avg_avg_return:                           0.056658328
+      avg_reward_per_step:                      0.0029144287
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.30899
+    avg_avg_return:                             0.88184905
+    avg_benchmark_regret:                       0.10278614
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.10079956
+    proxy_corner:
+      avg_avg_return:                           0.88184905
+      avg_reward_per_step:                      0.10079956
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        163577856
+    env-step-replay-before:                     83132416
+    env-step-used-before:                       163577856
+    ppo-update-before:                          99840
+    env-step-generate-before:                   80445440
+    env-step-all-after:                         163610624
+    env-step-replay-after:                      83165184
+    env-step-used-after:                        163610624
+    ppo-update-after:                           99860
+    env-step-generate-after:                    80445440
+  train-replay:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     67.93696
+    avg_avg_return:                             0.40998805
+    avg_reward_per_step:                        0.010620117
+    proxy_corner:
+      avg_avg_return:                           0.087910764
+      avg_reward_per_step:                      0.0014648438
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     67.93696
+    avg_avg_return:                             0.40998805
+    avg_reward_per_step:                        0.010620117
+    proxy_corner:
+      avg_avg_return:                           0.087910764
+      avg_reward_per_step:                      0.0014648438
+  perf:
+    env_steps_per_second:                       322245.89909091423
+    ppo_updates_per_second:                     41.96976703621515
+    eval-batch-train-duration:                  0.5382980369031429
+    eval-batch-orig-duration:                   0.4771801996976137
+    eval-batch-shift-duration:                  0.4549738001078367
+    eval-batch-tree-duration:                   0.47110298182815313
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.56543
+      cheese-corne_dist_wavg:                   13.553122
+      levels_cheese_is_in_corner_avg:           0.021240234
+      mouse-cheese_dist_finite_avg:             24.75415
+      mouse-cheese_dist_finite_wavg:            11.477625
+      mouse-cheese_dist_solvable_avg:           24.75415
+      mouse-cheese_dist_solvable_wavg:          11.477628
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.5752006
+    visit_patterns:
+      avg_first_visit_time:                     2172.487
+      avg_last_visit_time:                      2528.266
+      num_replay_batches:                       2538
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.000935147
+    avg_actor_approxkl3:                        0.0010392346
+    avg_actor_clipfrac:                         0.047747802
+    avg_actor_loss:                             -0.0007005714
+    avg_advantage:                              -0.04297953
+    avg_critic_clipfrac:                        0.00013427735
+    avg_critic_loss:                            0.010777319
+    avg_entropy:                                0.23716035
+    avg_grad_norm_pre_clip:                     1.6689047
+    avg_loss:                                   0.004450927
+    max:
+      max_actor_approxkl1:                      0.0024840422
+      max_actor_approxkl3:                      0.0016938967
+      max_actor_clipfrac:                       0.0715332
+      max_actor_loss:                           0.0009447038
+      max_advantage:                            0.17777264
+      max_critic_clipfrac:                      0.0012207031
+      max_critic_loss:                          0.012498101
+      max_entropy:                              0.25774094
+      max_grad_norm_pre_clip:                   2.7599142
+      max_loss:                                 0.006778865
+    std:
+      std_actor_approxkl1:                      0.0006712532
+      std_actor_approxkl3:                      0.00038100852
+      std_actor_clipfrac:                       0.015289143
+      std_actor_loss:                           0.001011469
+      std_advantage:                            0.1455412
+      std_critic_clipfrac:                      0.00029625633
+      std_critic_loss:                          0.001036837
+      std_entropy:                              0.008000734
+      std_grad_norm_pre_clip:                   0.4835393
+      std_loss:                                 0.0012426261
+  eval-batch-train:
+    avg_avg_episode_length:                     19.02159
+    avg_avg_return:                             0.9149039
+    avg_benchmark_regret:                       0.072313234
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12957764
+    proxy_corner:
+      avg_avg_return:                           0.057751954
+      avg_reward_per_step:                      0.0052108765
+  eval-batch-orig:
+    avg_avg_episode_length:                     24.197718
+    avg_avg_return:                             0.8940222
+    avg_benchmark_regret:                       0.09048933
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08557892
+    proxy_corner:
+      avg_avg_return:                           0.8940222
+      avg_reward_per_step:                      0.08557892
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.632141
+    avg_avg_return:                             0.9266851
+    avg_benchmark_regret:                       0.060820885
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14588928
+    proxy_corner:
+      avg_avg_return:                           0.056346923
+      avg_reward_per_step:                      0.0030136108
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.633377
+    avg_avg_return:                             0.88960624
+    avg_benchmark_regret:                       0.095029004
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09932709
+    proxy_corner:
+      avg_avg_return:                           0.88960624
+      avg_reward_per_step:                      0.09932709
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        164626432
+    env-step-replay-before:                     83722240
+    env-step-used-before:                       164626432
+    ppo-update-before:                          100480
+    env-step-generate-before:                   80904192
+    env-step-all-after:                         164659200
+    env-step-replay-after:                      83755008
+    env-step-used-after:                        164659200
+    ppo-update-after:                           100500
+    env-step-generate-after:                    80904192
+  train-replay:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     68.20247
+    avg_avg_return:                             0.40559313
+    avg_reward_per_step:                        0.010559082
+    proxy_corner:
+      avg_avg_return:                           0.118349046
+      avg_reward_per_step:                      0.0020446777
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     68.20247
+    avg_avg_return:                             0.40559313
+    avg_reward_per_step:                        0.010559082
+    proxy_corner:
+      avg_avg_return:                           0.118349046
+      avg_reward_per_step:                      0.0020446777
+  perf:
+    env_steps_per_second:                       281688.6770085818
+    ppo_updates_per_second:                     41.40282898748515
+    eval-batch-train-duration:                  0.53798355627805
+    eval-batch-orig-duration:                   0.4812415223568678
+    eval-batch-shift-duration:                  0.4756612051278353
+    eval-batch-tree-duration:                   0.45171012356877327
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.583984
+      cheese-corne_dist_wavg:                   23.44823
+      levels_cheese_is_in_corner_avg:           0.021728516
+      mouse-cheese_dist_finite_avg:             24.75415
+      mouse-cheese_dist_finite_wavg:            12.375254
+      mouse-cheese_dist_solvable_avg:           24.75415
+      mouse-cheese_dist_solvable_wavg:          12.375256
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.5699624
+    visit_patterns:
+      avg_first_visit_time:                     2186.1833
+      avg_last_visit_time:                      2546.1904
+      num_replay_batches:                       2556
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009637597
+    avg_actor_approxkl3:                        0.0008993693
+    avg_actor_clipfrac:                         0.04161377
+    avg_actor_loss:                             -0.0011875503
+    avg_advantage:                              -0.036590066
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.010534846
+    avg_entropy:                                0.23792663
+    avg_grad_norm_pre_clip:                     1.2813419
+    avg_loss:                                   0.003841946
+    max:
+      max_actor_approxkl1:                      0.002196379
+      max_actor_approxkl3:                      0.001415682
+      max_actor_clipfrac:                       0.06201172
+      max_actor_loss:                           0.0016920269
+      max_advantage:                            0.16528052
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0128635885
+      max_entropy:                              0.24899459
+      max_grad_norm_pre_clip:                   2.7495923
+      max_loss:                                 0.007404543
+    std:
+      std_actor_approxkl1:                      0.00056110055
+      std_actor_approxkl3:                      0.00031610372
+      std_actor_clipfrac:                       0.013224553
+      std_actor_loss:                           0.0014210513
+      std_advantage:                            0.14375289
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0015111775
+      std_entropy:                              0.008647742
+      std_grad_norm_pre_clip:                   0.55286694
+      std_loss:                                 0.0016487319
+  eval-batch-train:
+    avg_avg_episode_length:                     19.441187
+    avg_avg_return:                             0.90951574
+    avg_benchmark_regret:                       0.077701345
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.1282425
+    proxy_corner:
+      avg_avg_return:                           0.060503267
+      avg_reward_per_step:                      0.005256653
+  eval-batch-orig:
+    avg_avg_episode_length:                     23.162495
+    avg_avg_return:                             0.9021666
+    avg_benchmark_regret:                       0.08234496
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08774567
+    proxy_corner:
+      avg_avg_return:                           0.9021666
+      avg_reward_per_step:                      0.08774567
+  eval-batch-shift:
+    avg_avg_episode_length:                     17.597984
+    avg_avg_return:                             0.9344859
+    avg_benchmark_regret:                       0.05302003
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14794922
+    proxy_corner:
+      avg_avg_return:                           0.05708039
+      avg_reward_per_step:                      0.0031814575
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.05651
+    avg_avg_return:                             0.8901674
+    avg_benchmark_regret:                       0.09446778
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.1000061
+    proxy_corner:
+      avg_avg_return:                           0.8901674
+      avg_reward_per_step:                      0.1000061
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        165675008
+    env-step-replay-before:                     84344832
+    env-step-used-before:                       165675008
+    ppo-update-before:                          101120
+    env-step-generate-before:                   81330176
+    env-step-all-after:                         165707776
+    env-step-replay-after:                      84377600
+    env-step-used-after:                        165707776
+    ppo-update-after:                           101140
+    env-step-generate-after:                    81330176
+  train-replay:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     66.38025
+    avg_avg_return:                             0.43639582
+    avg_reward_per_step:                        0.011627197
+    proxy_corner:
+      avg_avg_return:                           0.083003804
+      avg_reward_per_step:                      0.0015258789
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     66.38025
+    avg_avg_return:                             0.43639582
+    avg_reward_per_step:                        0.011627197
+    proxy_corner:
+      avg_avg_return:                           0.083003804
+      avg_reward_per_step:                      0.0015258789
+  perf:
+    env_steps_per_second:                       348714.7798856576
+    ppo_updates_per_second:                     40.75783201726067
+    eval-batch-train-duration:                  0.5258163306862116
+    eval-batch-orig-duration:                   0.48840034008026123
+    eval-batch-shift-duration:                  0.3812151253223419
+    eval-batch-tree-duration:                   0.23260945826768875
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.647217
+      cheese-corne_dist_wavg:                   33.361263
+      levels_cheese_is_in_corner_avg:           0.022949219
+      mouse-cheese_dist_finite_avg:             24.798584
+      mouse-cheese_dist_finite_wavg:            13.261626
+      mouse-cheese_dist_solvable_avg:           24.798584
+      mouse-cheese_dist_solvable_wavg:          13.261628
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.5483519
+    visit_patterns:
+      avg_first_visit_time:                     2200.4023
+      avg_last_visit_time:                      2565.2495
+      num_replay_batches:                       2575
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013026943
+    avg_actor_approxkl3:                        0.0010424767
+    avg_actor_clipfrac:                         0.046520997
+    avg_actor_loss:                             -0.00095255225
+    avg_advantage:                              -0.030363552
+    avg_critic_clipfrac:                        4.8828126e-05
+    avg_critic_loss:                            0.009522821
+    avg_entropy:                                0.23301116
+    avg_grad_norm_pre_clip:                     1.6044543
+    avg_loss:                                   0.0035758466
+    max:
+      max_actor_approxkl1:                      0.0028127865
+      max_actor_approxkl3:                      0.0018141678
+      max_actor_clipfrac:                       0.076538086
+      max_actor_loss:                           0.0006558746
+      max_advantage:                            0.17168772
+      max_critic_clipfrac:                      0.00048828125
+      max_critic_loss:                          0.010655701
+      max_entropy:                              0.24909362
+      max_grad_norm_pre_clip:                   2.4199889
+      max_loss:                                 0.0052114683
+    std:
+      std_actor_approxkl1:                      0.0007155938
+      std_actor_approxkl3:                      0.0004743415
+      std_actor_clipfrac:                       0.019097155
+      std_actor_loss:                           0.000913654
+      std_advantage:                            0.13675383
+      std_critic_clipfrac:                      0.00014648437
+      std_critic_loss:                          0.00056808704
+      std_entropy:                              0.00838364
+      std_grad_norm_pre_clip:                   0.48415717
+      std_loss:                                 0.0009685619
+  eval-batch-train:
+    avg_avg_episode_length:                     18.58186
+    avg_avg_return:                             0.9222784
+    avg_benchmark_regret:                       0.064938635
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12730408
+    proxy_corner:
+      avg_avg_return:                           0.048728935
+      avg_reward_per_step:                      0.004852295
+  eval-batch-orig:
+    avg_avg_episode_length:                     22.935978
+    avg_avg_return:                             0.9045042
+    avg_benchmark_regret:                       0.0800074
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08148956
+    proxy_corner:
+      avg_avg_return:                           0.9045042
+      avg_reward_per_step:                      0.08148956
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.394955
+    avg_avg_return:                             0.92785835
+    avg_benchmark_regret:                       0.05964756
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14780426
+    proxy_corner:
+      avg_avg_return:                           0.05525219
+      avg_reward_per_step:                      0.0028457642
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.931402
+    avg_avg_return:                             0.89732254
+    avg_benchmark_regret:                       0.08731268
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09626007
+    proxy_corner:
+      avg_avg_return:                           0.89732254
+      avg_reward_per_step:                      0.09626007
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        166723584
+    env-step-replay-before:                     84934656
+    env-step-used-before:                       166723584
+    ppo-update-before:                          101760
+    env-step-generate-before:                   81788928
+    env-step-all-after:                         166756352
+    env-step-replay-after:                      84967424
+    env-step-used-after:                        166756352
+    ppo-update-after:                           101780
+    env-step-generate-after:                    81788928
+  train-replay:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     67.434456
+    avg_avg_return:                             0.41832814
+    avg_reward_per_step:                        0.011474609
+    proxy_corner:
+      avg_avg_return:                           0.074589685
+      avg_reward_per_step:                      0.0014648438
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     67.434456
+    avg_avg_return:                             0.41832814
+    avg_reward_per_step:                        0.011474609
+    proxy_corner:
+      avg_avg_return:                           0.074589685
+      avg_reward_per_step:                      0.0014648438
+  perf:
+    env_steps_per_second:                       387186.0308053499
+    ppo_updates_per_second:                     42.64416572177596
+    eval-batch-train-duration:                  0.49077529087662697
+    eval-batch-orig-duration:                   0.4363689096644521
+    eval-batch-shift-duration:                  0.4434901149943471
+    eval-batch-tree-duration:                   0.4195030461996794
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.60498
+      cheese-corne_dist_wavg:                   32.443596
+      levels_cheese_is_in_corner_avg:           0.023925781
+      mouse-cheese_dist_finite_avg:             24.86377
+      mouse-cheese_dist_finite_wavg:            10.588604
+      mouse-cheese_dist_solvable_avg:           24.86377
+      mouse-cheese_dist_solvable_wavg:          10.588607
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.54771364
+    visit_patterns:
+      avg_first_visit_time:                     2216.3516
+      avg_last_visit_time:                      2583.2852
+      num_replay_batches:                       2593
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013319144
+    avg_actor_approxkl3:                        0.0012035079
+    avg_actor_clipfrac:                         0.04900513
+    avg_actor_loss:                             -0.00112128
+    avg_advantage:                              -0.038590316
+    avg_critic_clipfrac:                        0.000109863286
+    avg_critic_loss:                            0.010345944
+    avg_entropy:                                0.23436193
+    avg_grad_norm_pre_clip:                     1.4142059
+    avg_loss:                                   0.0038173299
+    max:
+      max_actor_approxkl1:                      0.0038918217
+      max_actor_approxkl3:                      0.0024238783
+      max_actor_clipfrac:                       0.091796875
+      max_actor_loss:                           0.00028316677
+      max_advantage:                            0.15143883
+      max_critic_clipfrac:                      0.0007324219
+      max_critic_loss:                          0.012873372
+      max_entropy:                              0.24641202
+      max_grad_norm_pre_clip:                   2.8757026
+      max_loss:                                 0.006475648
+    std:
+      std_actor_approxkl1:                      0.0010515038
+      std_actor_approxkl3:                      0.0006740221
+      std_actor_clipfrac:                       0.0246529
+      std_actor_loss:                           0.00097640906
+      std_advantage:                            0.14242804
+      std_critic_clipfrac:                      0.00023448937
+      std_critic_loss:                          0.0015191723
+      std_entropy:                              0.007245634
+      std_grad_norm_pre_clip:                   0.646119
+      std_loss:                                 0.0013511238
+  eval-batch-train:
+    avg_avg_episode_length:                     19.017563
+    avg_avg_return:                             0.9165373
+    avg_benchmark_regret:                       0.070679806
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12815857
+    proxy_corner:
+      avg_avg_return:                           0.046940163
+      avg_reward_per_step:                      0.0049972534
+  eval-batch-orig:
+    avg_avg_episode_length:                     23.28277
+    avg_avg_return:                             0.9083719
+    avg_benchmark_regret:                       0.07613966
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08344269
+    proxy_corner:
+      avg_avg_return:                           0.9083719
+      avg_reward_per_step:                      0.08344269
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.415045
+    avg_avg_return:                             0.925153
+    avg_benchmark_regret:                       0.062352885
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14633942
+    proxy_corner:
+      avg_avg_return:                           0.0534327
+      avg_reward_per_step:                      0.0030288696
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.444618
+    avg_avg_return:                             0.8982267
+    avg_benchmark_regret:                       0.086408526
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09815216
+    proxy_corner:
+      avg_avg_return:                           0.8982267
+      avg_reward_per_step:                      0.09815216
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        167772160
+    env-step-replay-before:                     85458944
+    env-step-used-before:                       167772160
+    ppo-update-before:                          102400
+    env-step-generate-before:                   82313216
+    env-step-all-after:                         167804928
+    env-step-replay-after:                      85491712
+    env-step-used-after:                        167804928
+    ppo-update-after:                           102420
+    env-step-generate-after:                    82313216
+  train-replay:
+    shift_proportion:                           0.96875
+    avg_avg_episode_length:                     63.702347
+    avg_avg_return:                             0.46232867
+    avg_reward_per_step:                        0.012542725
+    proxy_corner:
+      avg_avg_return:                           0.09806211
+      avg_reward_per_step:                      0.0019836426
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  train-all:
+    shift_proportion:                           0.96875
+    avg_avg_episode_length:                     63.702347
+    avg_avg_return:                             0.46232867
+    avg_reward_per_step:                        0.012542725
+    proxy_corner:
+      avg_avg_return:                           0.09806211
+      avg_reward_per_step:                      0.0019836426
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  perf:
+    env_steps_per_second:                       294824.2434370479
+    ppo_updates_per_second:                     42.67059972775493
+    eval-batch-train-duration:                  0.6469111815094948
+    eval-batch-orig-duration:                   0.5911462660878897
+    eval-batch-shift-duration:                  0.5872918786481023
+    eval-batch-tree-duration:                   0.5701626082882285
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.673096
+      cheese-corne_dist_wavg:                   28.843134
+      levels_cheese_is_in_corner_avg:           0.022460938
+      mouse-cheese_dist_finite_avg:             24.894287
+      mouse-cheese_dist_finite_wavg:            8.7932415
+      mouse-cheese_dist_solvable_avg:           24.894287
+      mouse-cheese_dist_solvable_wavg:          8.7932415
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.5442364
+    visit_patterns:
+      avg_first_visit_time:                     2233.9153
+      avg_last_visit_time:                      2599.2263
+      num_replay_batches:                       2609
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010543902
+    avg_actor_approxkl3:                        0.0009589253
+    avg_actor_clipfrac:                         0.04246216
+    avg_actor_loss:                             -0.0011415866
+    avg_advantage:                              -0.010711933
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.008636961
+    avg_entropy:                                0.21113434
+    avg_grad_norm_pre_clip:                     1.0885398
+    avg_loss:                                   0.0029657595
+    max:
+      max_actor_approxkl1:                      0.003691899
+      max_actor_approxkl3:                      0.0019481559
+      max_actor_clipfrac:                       0.07873535
+      max_actor_loss:                           0.0004426241
+      max_advantage:                            0.19972467
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.012190039
+      max_entropy:                              0.22711286
+      max_grad_norm_pre_clip:                   1.5914477
+      max_loss:                                 0.0049687573
+    std:
+      std_actor_approxkl1:                      0.000937343
+      std_actor_approxkl3:                      0.00051170867
+      std_actor_clipfrac:                       0.020017084
+      std_actor_loss:                           0.00086948863
+      std_advantage:                            0.13134533
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0014674172
+      std_entropy:                              0.00908436
+      std_grad_norm_pre_clip:                   0.21692263
+      std_loss:                                 0.0009905779
+  eval-batch-train:
+    avg_avg_episode_length:                     19.174915
+    avg_avg_return:                             0.9173552
+    avg_benchmark_regret:                       0.06986191
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.1257782
+    proxy_corner:
+      avg_avg_return:                           0.052729487
+      avg_reward_per_step:                      0.0046539307
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-orig:
+    avg_avg_episode_length:                     23.574951
+    avg_avg_return:                             0.9043602
+    avg_benchmark_regret:                       0.08015139
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.080955505
+    proxy_corner:
+      avg_avg_return:                           0.9043602
+      avg_reward_per_step:                      0.080955505
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.031261
+    avg_avg_return:                             0.92416555
+    avg_benchmark_regret:                       0.063340425
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.1461029
+    proxy_corner:
+      avg_avg_return:                           0.054745547
+      avg_reward_per_step:                      0.0029678345
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.31598
+    avg_avg_return:                             0.8952599
+    avg_benchmark_regret:                       0.08937524
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09654999
+    proxy_corner:
+      avg_avg_return:                           0.8952599
+      avg_reward_per_step:                      0.09654999
+    rollouts_gif:                               float32(132, 225, 225, 3)
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        168820736
+    env-step-replay-before:                     85983232
+    env-step-used-before:                       168820736
+    ppo-update-before:                          103040
+    env-step-generate-before:                   82837504
+    env-step-all-after:                         168853504
+    env-step-replay-after:                      85983232
+    env-step-used-after:                        168853504
+    ppo-update-after:                           103060
+    env-step-generate-after:                    82870272
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     20.45987
+    avg_avg_return:                             0.83824265
+    avg_reward_per_step:                        0.12695312
+    proxy_corner:
+      avg_avg_return:                           0.042577263
+      avg_reward_per_step:                      0.002166748
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     20.45987
+    avg_avg_return:                             0.83824265
+    avg_reward_per_step:                        0.12695312
+    proxy_corner:
+      avg_avg_return:                           0.042577263
+      avg_reward_per_step:                      0.002166748
+  perf:
+    env_steps_per_second:                       292222.2858600061
+    ppo_updates_per_second:                     42.11263622244827
+    eval-batch-train-duration:                  0.5386233432218432
+    eval-batch-orig-duration:                   0.48764023650437593
+    eval-batch-shift-duration:                  0.48586494103074074
+    eval-batch-tree-duration:                   0.49608376529067755
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.691895
+      cheese-corne_dist_wavg:                   27.053566
+      levels_cheese_is_in_corner_avg:           0.02319336
+      mouse-cheese_dist_finite_avg:             24.93457
+      mouse-cheese_dist_finite_wavg:            12.399006
+      mouse-cheese_dist_solvable_avg:           24.93457
+      mouse-cheese_dist_solvable_wavg:          12.399004
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5675081
+    visit_patterns:
+      avg_first_visit_time:                     2250.8506
+      avg_last_visit_time:                      2614.2144
+      num_replay_batches:                       2624
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00086432416
+    avg_actor_approxkl3:                        0.000490355
+    avg_actor_clipfrac:                         0.018621827
+    avg_actor_loss:                             -0.0012608386
+    avg_advantage:                              0.06867637
+    avg_critic_clipfrac:                        0.1098938
+    avg_critic_loss:                            0.0028364684
+    avg_entropy:                                0.080123134
+    avg_grad_norm_pre_clip:                     0.9503379
+    avg_loss:                                   7.727226e-05
+    max:
+      max_actor_approxkl1:                      0.0021357727
+      max_actor_approxkl3:                      0.0009874185
+      max_actor_clipfrac:                       0.035888672
+      max_actor_loss:                           0.0005632974
+      max_advantage:                            0.20789737
+      max_critic_clipfrac:                      0.70422363
+      max_critic_loss:                          0.00503973
+      max_entropy:                              0.09747674
+      max_grad_norm_pre_clip:                   1.5919853
+      max_loss:                                 0.0024963012
+    std:
+      std_actor_approxkl1:                      0.00065924524
+      std_actor_approxkl3:                      0.00028620707
+      std_actor_clipfrac:                       0.008494591
+      std_actor_loss:                           0.0011435142
+      std_advantage:                            0.06784173
+      std_critic_clipfrac:                      0.21038969
+      std_critic_loss:                          0.0011223957
+      std_entropy:                              0.008057606
+      std_grad_norm_pre_clip:                   0.26158518
+      std_loss:                                 0.0012457831
+  eval-batch-train:
+    avg_avg_episode_length:                     19.425434
+    avg_avg_return:                             0.9095782
+    avg_benchmark_regret:                       0.077638835
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12802124
+    proxy_corner:
+      avg_avg_return:                           0.048270434
+      avg_reward_per_step:                      0.0049819946
+  eval-batch-orig:
+    avg_avg_episode_length:                     23.315891
+    avg_avg_return:                             0.892784
+    avg_benchmark_regret:                       0.091727555
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08570099
+    proxy_corner:
+      avg_avg_return:                           0.892784
+      avg_reward_per_step:                      0.08570099
+  eval-batch-shift:
+    avg_avg_episode_length:                     17.767258
+    avg_avg_return:                             0.92704624
+    avg_benchmark_regret:                       0.060459666
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.1493454
+    proxy_corner:
+      avg_avg_return:                           0.050153546
+      avg_reward_per_step:                      0.0029144287
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.534508
+    avg_avg_return:                             0.8836242
+    avg_benchmark_regret:                       0.10101097
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09983826
+    proxy_corner:
+      avg_avg_return:                           0.8836242
+      avg_reward_per_step:                      0.09983826
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        169869312
+    env-step-replay-before:                     86474752
+    env-step-used-before:                       169869312
+    ppo-update-before:                          103680
+    env-step-generate-before:                   83394560
+    env-step-all-after:                         169902080
+    env-step-replay-after:                      86474752
+    env-step-used-after:                        169902080
+    ppo-update-after:                           103700
+    env-step-generate-after:                    83427328
+  train-generate:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     22.682411
+    avg_avg_return:                             0.81829673
+    avg_reward_per_step:                        0.109802246
+    proxy_corner:
+      avg_avg_return:                           0.050465796
+      avg_reward_per_step:                      0.00289917
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     22.682411
+    avg_avg_return:                             0.81829673
+    avg_reward_per_step:                        0.109802246
+    proxy_corner:
+      avg_avg_return:                           0.050465796
+      avg_reward_per_step:                      0.00289917
+  perf:
+    env_steps_per_second:                       311499.8380528902
+    ppo_updates_per_second:                     42.09228064737985
+    eval-batch-train-duration:                  0.5443478003144264
+    eval-batch-orig-duration:                   0.47595315892249346
+    eval-batch-shift-duration:                  0.4852469079196453
+    eval-batch-tree-duration:                   0.45557106379419565
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.661621
+      cheese-corne_dist_wavg:                   22.563324
+      levels_cheese_is_in_corner_avg:           0.021484375
+      mouse-cheese_dist_finite_avg:             24.90503
+      mouse-cheese_dist_finite_wavg:            12.384484
+      mouse-cheese_dist_solvable_avg:           24.90503
+      mouse-cheese_dist_solvable_wavg:          12.384484
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.570418
+    visit_patterns:
+      avg_first_visit_time:                     2265.1875
+      avg_last_visit_time:                      2629.274
+      num_replay_batches:                       2639
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009545141
+    avg_actor_approxkl3:                        0.0006874194
+    avg_actor_clipfrac:                         0.024694825
+    avg_actor_loss:                             -0.0011885376
+    avg_advantage:                              0.06927344
+    avg_critic_clipfrac:                        0.15888672
+    avg_critic_loss:                            0.0034186523
+    avg_entropy:                                0.09595851
+    avg_grad_norm_pre_clip:                     1.0234439
+    avg_loss:                                   0.00042483033
+    max:
+      max_actor_approxkl1:                      0.0023599137
+      max_actor_approxkl3:                      0.0016404337
+      max_actor_clipfrac:                       0.040649414
+      max_actor_loss:                           0.0005592108
+      max_advantage:                            0.2288121
+      max_critic_clipfrac:                      0.7060547
+      max_critic_loss:                          0.0061704125
+      max_entropy:                              0.12297974
+      max_grad_norm_pre_clip:                   1.4858364
+      max_loss:                                 0.002944326
+    std:
+      std_actor_approxkl1:                      0.00063416886
+      std_actor_approxkl3:                      0.00036016235
+      std_actor_clipfrac:                       0.009739888
+      std_actor_loss:                           0.0010136854
+      std_advantage:                            0.076838575
+      std_critic_clipfrac:                      0.25153103
+      std_critic_loss:                          0.001107345
+      std_entropy:                              0.011811554
+      std_grad_norm_pre_clip:                   0.24778758
+      std_loss:                                 0.0011903072
+  eval-batch-train:
+    avg_avg_episode_length:                     18.382694
+    avg_avg_return:                             0.92465043
+    avg_benchmark_regret:                       0.06256664
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12800598
+    proxy_corner:
+      avg_avg_return:                           0.048626453
+      avg_reward_per_step:                      0.0050582886
+  eval-batch-orig:
+    avg_avg_episode_length:                     22.88845
+    avg_avg_return:                             0.9052088
+    avg_benchmark_regret:                       0.07930265
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.084754944
+    proxy_corner:
+      avg_avg_return:                           0.9052088
+      avg_reward_per_step:                      0.084754944
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.401993
+    avg_avg_return:                             0.9222956
+    avg_benchmark_regret:                       0.06521037
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14949799
+    proxy_corner:
+      avg_avg_return:                           0.05198026
+      avg_reward_per_step:                      0.0027618408
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.025618
+    avg_avg_return:                             0.89100283
+    avg_benchmark_regret:                       0.09363233
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09976196
+    proxy_corner:
+      avg_avg_return:                           0.89100283
+      avg_reward_per_step:                      0.09976196
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        170917888
+    env-step-replay-before:                     86900736
+    env-step-used-before:                       170917888
+    ppo-update-before:                          104320
+    env-step-generate-before:                   84017152
+    env-step-all-after:                         170950656
+    env-step-replay-after:                      86900736
+    env-step-used-after:                        170950656
+    ppo-update-after:                           104340
+    env-step-generate-after:                    84049920
+  train-generate:
+    shift_proportion:                           1.0
+    avg_avg_episode_length:                     19.065672
+    avg_avg_return:                             0.85570353
+    avg_reward_per_step:                        0.13162231
+    proxy_corner:
+      avg_avg_return:                           0.026773397
+      avg_reward_per_step:                      0.0013122559
+  train-all:
+    shift_proportion:                           1.0
+    avg_avg_episode_length:                     19.065672
+    avg_avg_return:                             0.85570353
+    avg_reward_per_step:                        0.13162231
+    proxy_corner:
+      avg_avg_return:                           0.026773397
+      avg_reward_per_step:                      0.0013122559
+  perf:
+    env_steps_per_second:                       327733.6141174419
+    ppo_updates_per_second:                     43.7154235313882
+    eval-batch-train-duration:                  0.582362019456923
+    eval-batch-orig-duration:                   0.5632743490859866
+    eval-batch-shift-duration:                  0.5466595841571689
+    eval-batch-tree-duration:                   0.5657111266627908
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.692627
+      cheese-corne_dist_wavg:                   29.747507
+      levels_cheese_is_in_corner_avg:           0.021728516
+      mouse-cheese_dist_finite_avg:             24.939209
+      mouse-cheese_dist_finite_wavg:            9.68592
+      mouse-cheese_dist_solvable_avg:           24.939209
+      mouse-cheese_dist_solvable_wavg:          9.68592
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.55971664
+    visit_patterns:
+      avg_first_visit_time:                     2282.904
+      avg_last_visit_time:                      2642.2227
+      num_replay_batches:                       2652
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00024596942
+    avg_actor_approxkl3:                        0.00063263276
+    avg_actor_clipfrac:                         0.01977539
+    avg_actor_loss:                             -0.0007541589
+    avg_advantage:                              0.0038114018
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0011718315
+    avg_entropy:                                0.07212448
+    avg_grad_norm_pre_clip:                     0.80154055
+    avg_loss:                                   -0.0002403676
+    max:
+      max_actor_approxkl1:                      0.0010009577
+      max_actor_approxkl3:                      0.0015957064
+      max_actor_clipfrac:                       0.034179688
+      max_actor_loss:                           0.0006264746
+      max_advantage:                            0.12669235
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0016372758
+      max_entropy:                              0.08584711
+      max_grad_norm_pre_clip:                   1.6120313
+      max_loss:                                 0.0008724119
+    std:
+      std_actor_approxkl1:                      0.00043674742
+      std_actor_approxkl3:                      0.0003543812
+      std_actor_clipfrac:                       0.00852563
+      std_actor_loss:                           0.0008111174
+      std_advantage:                            0.048495714
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00038828744
+      std_entropy:                              0.0070898384
+      std_grad_norm_pre_clip:                   0.31059015
+      std_loss:                                 0.0008139552
+  eval-batch-train:
+    avg_avg_episode_length:                     19.439142
+    avg_avg_return:                             0.9167994
+    avg_benchmark_regret:                       0.070417665
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.1280899
+    proxy_corner:
+      avg_avg_return:                           0.046624534
+      avg_reward_per_step:                      0.004928589
+  eval-batch-orig:
+    avg_avg_episode_length:                     23.978453
+    avg_avg_return:                             0.8938329
+    avg_benchmark_regret:                       0.09067863
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.084754944
+    proxy_corner:
+      avg_avg_return:                           0.8938329
+      avg_reward_per_step:                      0.084754944
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.24958
+    avg_avg_return:                             0.926854
+    avg_benchmark_regret:                       0.06065192
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14906311
+    proxy_corner:
+      avg_avg_return:                           0.052894287
+      avg_reward_per_step:                      0.0029144287
+  eval-batch-tree:
+    avg_avg_episode_length:                     24.430912
+    avg_avg_return:                             0.8782433
+    avg_benchmark_regret:                       0.10639184
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09812164
+    proxy_corner:
+      avg_avg_return:                           0.8782433
+      avg_reward_per_step:                      0.09812164
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        171966464
+    env-step-replay-before:                     87457792
+    env-step-used-before:                       171966464
+    ppo-update-before:                          104960
+    env-step-generate-before:                   84508672
+    env-step-all-after:                         171999232
+    env-step-replay-after:                      87457792
+    env-step-used-after:                        171999232
+    ppo-update-after:                           104980
+    env-step-generate-after:                    84541440
+  train-generate:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     21.634119
+    avg_avg_return:                             0.8234836
+    avg_reward_per_step:                        0.13980103
+    proxy_corner:
+      avg_avg_return:                           0.054634918
+      avg_reward_per_step:                      0.004180908
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     21.634119
+    avg_avg_return:                             0.8234836
+    avg_reward_per_step:                        0.13980103
+    proxy_corner:
+      avg_avg_return:                           0.054634918
+      avg_reward_per_step:                      0.004180908
+  perf:
+    env_steps_per_second:                       232859.6740445905
+    ppo_updates_per_second:                     42.18134963126502
+    eval-batch-train-duration:                  0.6166769005358219
+    eval-batch-orig-duration:                   0.5745174437761307
+    eval-batch-shift-duration:                  0.5449588280171156
+    eval-batch-tree-duration:                   0.5836858553811908
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.75586
+      cheese-corne_dist_wavg:                   15.374879
+      levels_cheese_is_in_corner_avg:           0.021728516
+      mouse-cheese_dist_finite_avg:             24.942871
+      mouse-cheese_dist_finite_wavg:            10.590472
+      mouse-cheese_dist_solvable_avg:           24.942871
+      mouse-cheese_dist_solvable_wavg:          10.590473
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.5570447
+    visit_patterns:
+      avg_first_visit_time:                     2298.042
+      avg_last_visit_time:                      2659.0952
+      num_replay_batches:                       2669
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013840924
+    avg_actor_approxkl3:                        0.000982895
+    avg_actor_clipfrac:                         0.026049806
+    avg_actor_loss:                             -0.0006969315
+    avg_advantage:                              0.043000683
+    avg_critic_clipfrac:                        0.0031799318
+    avg_critic_loss:                            0.0029324684
+    avg_entropy:                                0.08003899
+    avg_grad_norm_pre_clip:                     1.2741135
+    avg_loss:                                   0.0006892637
+    max:
+      max_actor_approxkl1:                      0.0043036435
+      max_actor_approxkl3:                      0.0023497008
+      max_actor_clipfrac:                       0.05102539
+      max_actor_loss:                           0.0007760376
+      max_advantage:                            0.18986756
+      max_critic_clipfrac:                      0.017456055
+      max_critic_loss:                          0.005750862
+      max_entropy:                              0.10176453
+      max_grad_norm_pre_clip:                   2.1608005
+      max_loss:                                 0.0024577915
+    std:
+      std_actor_approxkl1:                      0.0010958512
+      std_actor_approxkl3:                      0.00061811233
+      std_actor_clipfrac:                       0.012860697
+      std_actor_loss:                           0.0009105391
+      std_advantage:                            0.07369388
+      std_critic_clipfrac:                      0.0052018957
+      std_critic_loss:                          0.0010116147
+      std_entropy:                              0.0121001955
+      std_grad_norm_pre_clip:                   0.32762316
+      std_loss:                                 0.00097222533
+  eval-batch-train:
+    avg_avg_episode_length:                     19.05247
+    avg_avg_return:                             0.91812295
+    avg_benchmark_regret:                       0.06909412
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12809753
+    proxy_corner:
+      avg_avg_return:                           0.04759048
+      avg_reward_per_step:                      0.0050964355
+  eval-batch-orig:
+    avg_avg_episode_length:                     22.874275
+    avg_avg_return:                             0.90116775
+    avg_benchmark_regret:                       0.083343804
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08721924
+    proxy_corner:
+      avg_avg_return:                           0.90116775
+      avg_reward_per_step:                      0.08721924
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.042051
+    avg_avg_return:                             0.9282719
+    avg_benchmark_regret:                       0.05923406
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.15000916
+    proxy_corner:
+      avg_avg_return:                           0.04985902
+      avg_reward_per_step:                      0.002960205
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.506039
+    avg_avg_return:                             0.89778864
+    avg_benchmark_regret:                       0.08684655
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.10151672
+    proxy_corner:
+      avg_avg_return:                           0.89778864
+      avg_reward_per_step:                      0.10151672
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        173015040
+    env-step-replay-before:                     88014848
+    env-step-used-before:                       173015040
+    ppo-update-before:                          105600
+    env-step-generate-before:                   85000192
+    env-step-all-after:                         173047808
+    env-step-replay-after:                      88047616
+    env-step-used-after:                        173047808
+    ppo-update-after:                           105620
+    env-step-generate-after:                    85000192
+  train-replay:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     66.78871
+    avg_avg_return:                             0.4431473
+    avg_reward_per_step:                        0.011932373
+    proxy_corner:
+      avg_avg_return:                           0.14634687
+      avg_reward_per_step:                      0.0027160645
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     66.78871
+    avg_avg_return:                             0.4431473
+    avg_reward_per_step:                        0.011932373
+    proxy_corner:
+      avg_avg_return:                           0.14634687
+      avg_reward_per_step:                      0.0027160645
+  perf:
+    env_steps_per_second:                       298051.4136710292
+    ppo_updates_per_second:                     42.605270867051026
+    eval-batch-train-duration:                  0.5168470283970237
+    eval-batch-orig-duration:                   0.48323469050228596
+    eval-batch-shift-duration:                  0.4800168424844742
+    eval-batch-tree-duration:                   0.44465239718556404
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.777588
+      cheese-corne_dist_wavg:                   10.003487
+      levels_cheese_is_in_corner_avg:           0.021240234
+      mouse-cheese_dist_finite_avg:             24.941162
+      mouse-cheese_dist_finite_wavg:            12.41125
+      mouse-cheese_dist_solvable_avg:           24.941162
+      mouse-cheese_dist_solvable_wavg:          12.411255
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999997
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016798
+    scoring:
+      avg_scores:                               0.5597331
+    visit_patterns:
+      avg_first_visit_time:                     2311.2498
+      avg_last_visit_time:                      2677.1914
+      num_replay_batches:                       2687
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010163939
+    avg_actor_approxkl3:                        0.0008142748
+    avg_actor_clipfrac:                         0.037664797
+    avg_actor_loss:                             -0.0010340549
+    avg_advantage:                              -0.004903848
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.008743146
+    avg_entropy:                                0.22213486
+    avg_grad_norm_pre_clip:                     1.1767251
+    avg_loss:                                   0.0031153827
+    max:
+      max_actor_approxkl1:                      0.0022621523
+      max_actor_approxkl3:                      0.0013226774
+      max_actor_clipfrac:                       0.0579834
+      max_actor_loss:                           0.00067283213
+      max_advantage:                            0.24699163
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.010668915
+      max_entropy:                              0.23611921
+      max_grad_norm_pre_clip:                   1.7149233
+      max_loss:                                 0.0051484224
+    std:
+      std_actor_approxkl1:                      0.0005191888
+      std_actor_approxkl3:                      0.00031340166
+      std_actor_clipfrac:                       0.0139472205
+      std_actor_loss:                           0.0011016211
+      std_advantage:                            0.13227752
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00093126105
+      std_entropy:                              0.006756274
+      std_grad_norm_pre_clip:                   0.3266372
+      std_loss:                                 0.0012015972
+  eval-batch-train:
+    avg_avg_episode_length:                     19.430912
+    avg_avg_return:                             0.9131596
+    avg_benchmark_regret:                       0.07405753
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12825775
+    proxy_corner:
+      avg_avg_return:                           0.051926147
+      avg_reward_per_step:                      0.0050964355
+  eval-batch-orig:
+    avg_avg_episode_length:                     23.416634
+    avg_avg_return:                             0.8999258
+    avg_benchmark_regret:                       0.08458565
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08312225
+    proxy_corner:
+      avg_avg_return:                           0.8999258
+      avg_reward_per_step:                      0.08312225
+  eval-batch-shift:
+    avg_avg_episode_length:                     17.666122
+    avg_avg_return:                             0.9345875
+    avg_benchmark_regret:                       0.05291847
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14834595
+    proxy_corner:
+      avg_avg_return:                           0.051767275
+      avg_reward_per_step:                      0.0028762817
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.765282
+    avg_avg_return:                             0.8995757
+    avg_benchmark_regret:                       0.085059434
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.097976685
+    proxy_corner:
+      avg_avg_return:                           0.8995757
+      avg_reward_per_step:                      0.097976685
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        174063616
+    env-step-replay-before:                     88440832
+    env-step-used-before:                       174063616
+    ppo-update-before:                          106240
+    env-step-generate-before:                   85622784
+    env-step-all-after:                         174096384
+    env-step-replay-after:                      88440832
+    env-step-used-after:                        174096384
+    ppo-update-after:                           106260
+    env-step-generate-after:                    85655552
+  train-generate:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     18.613504
+    avg_avg_return:                             0.85161936
+    avg_reward_per_step:                        0.11743164
+    proxy_corner:
+      avg_avg_return:                           0.048966512
+      avg_reward_per_step:                      0.0037231445
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     18.613504
+    avg_avg_return:                             0.85161936
+    avg_reward_per_step:                        0.11743164
+    proxy_corner:
+      avg_avg_return:                           0.048966512
+      avg_reward_per_step:                      0.0037231445
+  perf:
+    env_steps_per_second:                       350485.8929757658
+    ppo_updates_per_second:                     40.382863602649394
+    eval-batch-train-duration:                  0.5214785728603601
+    eval-batch-orig-duration:                   0.4534478122368455
+    eval-batch-shift-duration:                  0.4727073581889272
+    eval-batch-tree-duration:                   0.4792657159268856
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.722168
+      cheese-corne_dist_wavg:                   23.460287
+      levels_cheese_is_in_corner_avg:           0.022460938
+      mouse-cheese_dist_finite_avg:             24.97168
+      mouse-cheese_dist_finite_wavg:            7.895599
+      mouse-cheese_dist_solvable_avg:           24.97168
+      mouse-cheese_dist_solvable_wavg:          7.895601
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.5679437
+    visit_patterns:
+      avg_first_visit_time:                     2327.3599
+      avg_last_visit_time:                      2689.3452
+      num_replay_batches:                       2699
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0007864199
+    avg_actor_approxkl3:                        0.0006559311
+    avg_actor_clipfrac:                         0.020007325
+    avg_actor_loss:                             -0.0007883315
+    avg_advantage:                              0.002121813
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0018005787
+    avg_entropy:                                0.07322707
+    avg_grad_norm_pre_clip:                     0.8959206
+    avg_loss:                                   3.8730832e-05
+    max:
+      max_actor_approxkl1:                      0.0025812287
+      max_actor_approxkl3:                      0.001176443
+      max_actor_clipfrac:                       0.03161621
+      max_actor_loss:                           0.0009150319
+      max_advantage:                            0.13722187
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0032198078
+      max_entropy:                              0.091594465
+      max_grad_norm_pre_clip:                   1.4196553
+      max_loss:                                 0.002144062
+    std:
+      std_actor_approxkl1:                      0.0006388623
+      std_actor_approxkl3:                      0.00035371602
+      std_actor_clipfrac:                       0.008313149
+      std_actor_loss:                           0.0010459475
+      std_advantage:                            0.060108576
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0007766382
+      std_entropy:                              0.012720465
+      std_grad_norm_pre_clip:                   0.24094655
+      std_loss:                                 0.0010577239
+  eval-batch-train:
+    avg_avg_episode_length:                     18.934496
+    avg_avg_return:                             0.9153859
+    avg_benchmark_regret:                       0.07183115
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.13017273
+    proxy_corner:
+      avg_avg_return:                           0.04730179
+      avg_reward_per_step:                      0.0049438477
+  eval-batch-orig:
+    avg_avg_episode_length:                     23.15591
+    avg_avg_return:                             0.8966341
+    avg_benchmark_regret:                       0.08787739
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08539581
+    proxy_corner:
+      avg_avg_return:                           0.8966341
+      avg_reward_per_step:                      0.08539581
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.245136
+    avg_avg_return:                             0.9227195
+    avg_benchmark_regret:                       0.06478651
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14854431
+    proxy_corner:
+      avg_avg_return:                           0.048047747
+      avg_reward_per_step:                      0.0026245117
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.385904
+    avg_avg_return:                             0.90316653
+    avg_benchmark_regret:                       0.08146873
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.099983215
+    proxy_corner:
+      avg_avg_return:                           0.90316653
+      avg_reward_per_step:                      0.099983215
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        175112192
+    env-step-replay-before:                     88932352
+    env-step-used-before:                       175112192
+    ppo-update-before:                          106880
+    env-step-generate-before:                   86179840
+    env-step-all-after:                         175144960
+    env-step-replay-after:                      88965120
+    env-step-used-after:                        175144960
+    ppo-update-after:                           106900
+    env-step-generate-after:                    86179840
+  train-replay:
+    shift_proportion:                           0.95703125
+    avg_avg_episode_length:                     66.72508
+    avg_avg_return:                             0.43826985
+    avg_reward_per_step:                        0.011810303
+    proxy_corner:
+      avg_avg_return:                           0.13710786
+      avg_reward_per_step:                      0.0024108887
+  train-all:
+    shift_proportion:                           0.95703125
+    avg_avg_episode_length:                     66.72508
+    avg_avg_return:                             0.43826985
+    avg_reward_per_step:                        0.011810303
+    proxy_corner:
+      avg_avg_return:                           0.13710786
+      avg_reward_per_step:                      0.0024108887
+  perf:
+    env_steps_per_second:                       295310.2649878174
+    ppo_updates_per_second:                     42.065305109747854
+    eval-batch-train-duration:                  0.5283160535618663
+    eval-batch-orig-duration:                   0.4874579329043627
+    eval-batch-shift-duration:                  0.4866113346070051
+    eval-batch-tree-duration:                   0.45374483335763216
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.868408
+      cheese-corne_dist_wavg:                   2.795391
+      levels_cheese_is_in_corner_avg:           0.021972656
+      mouse-cheese_dist_finite_avg:             25.038818
+      mouse-cheese_dist_finite_wavg:            7.025331
+      mouse-cheese_dist_solvable_avg:           25.038818
+      mouse-cheese_dist_solvable_wavg:          7.025331
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.56214046
+    visit_patterns:
+      avg_first_visit_time:                     2339.299
+      avg_last_visit_time:                      2705.1914
+      num_replay_batches:                       2715
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010834963
+    avg_actor_approxkl3:                        0.0011697634
+    avg_actor_clipfrac:                         0.04953003
+    avg_actor_loss:                             -0.00060415495
+    avg_advantage:                              -0.030609176
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.009434107
+    avg_entropy:                                0.23088351
+    avg_grad_norm_pre_clip:                     1.6819105
+    avg_loss:                                   0.003882015
+    max:
+      max_actor_approxkl1:                      0.0031183655
+      max_actor_approxkl3:                      0.0026603122
+      max_actor_clipfrac:                       0.08679199
+      max_actor_loss:                           0.0012963265
+      max_advantage:                            0.15550512
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.011824101
+      max_entropy:                              0.25582123
+      max_grad_norm_pre_clip:                   2.32824
+      max_loss:                                 0.0066913613
+    std:
+      std_actor_approxkl1:                      0.00088682194
+      std_actor_approxkl3:                      0.0005935593
+      std_actor_clipfrac:                       0.018948948
+      std_actor_loss:                           0.0007481598
+      std_advantage:                            0.13540664
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0014832641
+      std_entropy:                              0.011278283
+      std_grad_norm_pre_clip:                   0.3368815
+      std_loss:                                 0.0010634391
+  eval-batch-train:
+    avg_avg_episode_length:                     18.622385
+    avg_avg_return:                             0.92081726
+    avg_benchmark_regret:                       0.066399775
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12804413
+    proxy_corner:
+      avg_avg_return:                           0.057061374
+      avg_reward_per_step:                      0.0051651
+  eval-batch-orig:
+    avg_avg_episode_length:                     22.557484
+    avg_avg_return:                             0.90488154
+    avg_benchmark_regret:                       0.07962999
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08740997
+    proxy_corner:
+      avg_avg_return:                           0.90488154
+      avg_reward_per_step:                      0.08740997
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.542797
+    avg_avg_return:                             0.9291097
+    avg_benchmark_regret:                       0.058396302
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14744568
+    proxy_corner:
+      avg_avg_return:                           0.057895444
+      avg_reward_per_step:                      0.0030975342
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.03824
+    avg_avg_return:                             0.8967417
+    avg_benchmark_regret:                       0.087893486
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.101890564
+    proxy_corner:
+      avg_avg_return:                           0.8967417
+      avg_reward_per_step:                      0.101890564
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        176160768
+    env-step-replay-before:                     89358336
+    env-step-used-before:                       176160768
+    ppo-update-before:                          107520
+    env-step-generate-before:                   86802432
+    env-step-all-after:                         176193536
+    env-step-replay-after:                      89391104
+    env-step-used-after:                        176193536
+    ppo-update-after:                           107540
+    env-step-generate-after:                    86802432
+  train-replay:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     67.2963
+    avg_avg_return:                             0.42144692
+    avg_reward_per_step:                        0.011291504
+    proxy_corner:
+      avg_avg_return:                           0.07003611
+      avg_reward_per_step:                      0.0012207031
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     67.2963
+    avg_avg_return:                             0.42144692
+    avg_reward_per_step:                        0.011291504
+    proxy_corner:
+      avg_avg_return:                           0.07003611
+      avg_reward_per_step:                      0.0012207031
+  perf:
+    env_steps_per_second:                       352008.6897607764
+    ppo_updates_per_second:                     41.047264975252
+    eval-batch-train-duration:                  0.5347506105899811
+    eval-batch-orig-duration:                   0.4884219253435731
+    eval-batch-shift-duration:                  0.4874773286283016
+    eval-batch-tree-duration:                   0.4551470410078764
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.78247
+      cheese-corne_dist_wavg:                   13.579293
+      levels_cheese_is_in_corner_avg:           0.021728516
+      mouse-cheese_dist_finite_avg:             25.019531
+      mouse-cheese_dist_finite_wavg:            11.495816
+      mouse-cheese_dist_solvable_avg:           25.019531
+      mouse-cheese_dist_solvable_wavg:          11.495817
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.580263
+    visit_patterns:
+      avg_first_visit_time:                     2354.4438
+      avg_last_visit_time:                      2718.2375
+      num_replay_batches:                       2728
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00096780685
+    avg_actor_approxkl3:                        0.0010840775
+    avg_actor_clipfrac:                         0.048046876
+    avg_actor_loss:                             -0.0011499961
+    avg_advantage:                              -0.0448303
+    avg_critic_clipfrac:                        0.00025024413
+    avg_critic_loss:                            0.0105233
+    avg_entropy:                                0.23268647
+    avg_grad_norm_pre_clip:                     1.3682076
+    avg_loss:                                   0.0038789671
+    max:
+      max_actor_approxkl1:                      0.0025512357
+      max_actor_approxkl3:                      0.001994555
+      max_actor_clipfrac:                       0.07128906
+      max_actor_loss:                           0.0015972257
+      max_advantage:                            0.12178737
+      max_critic_clipfrac:                      0.0029296875
+      max_critic_loss:                          0.013611528
+      max_entropy:                              0.26346976
+      max_grad_norm_pre_clip:                   2.4981334
+      max_loss:                                 0.006567237
+    std:
+      std_actor_approxkl1:                      0.0006738469
+      std_actor_approxkl3:                      0.0004534852
+      std_actor_clipfrac:                       0.01647752
+      std_actor_loss:                           0.0012335975
+      std_advantage:                            0.14351994
+      std_critic_clipfrac:                      0.0007185319
+      std_critic_loss:                          0.0014999589
+      std_entropy:                              0.011978068
+      std_grad_norm_pre_clip:                   0.4351771
+      std_loss:                                 0.0013838202
+  eval-batch-train:
+    avg_avg_episode_length:                     18.134354
+    avg_avg_return:                             0.92319405
+    avg_benchmark_regret:                       0.06402308
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12886047
+    proxy_corner:
+      avg_avg_return:                           0.051106688
+      avg_reward_per_step:                      0.0048675537
+  eval-batch-orig:
+    avg_avg_episode_length:                     23.579605
+    avg_avg_return:                             0.8927344
+    avg_benchmark_regret:                       0.09177716
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08747101
+    proxy_corner:
+      avg_avg_return:                           0.8927344
+      avg_reward_per_step:                      0.08747101
+  eval-batch-shift:
+    avg_avg_episode_length:                     17.963593
+    avg_avg_return:                             0.93012214
+    avg_benchmark_regret:                       0.057383776
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14748383
+    proxy_corner:
+      avg_avg_return:                           0.05122474
+      avg_reward_per_step:                      0.0028457642
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.631615
+    avg_avg_return:                             0.89301014
+    avg_benchmark_regret:                       0.091625065
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.100730896
+    proxy_corner:
+      avg_avg_return:                           0.89301014
+      avg_reward_per_step:                      0.100730896
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        177209344
+    env-step-replay-before:                     89784320
+    env-step-used-before:                       177209344
+    ppo-update-before:                          108160
+    env-step-generate-before:                   87425024
+    env-step-all-after:                         177242112
+    env-step-replay-after:                      89784320
+    env-step-used-after:                        177242112
+    ppo-update-after:                           108180
+    env-step-generate-after:                    87457792
+  train-generate:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     22.268572
+    avg_avg_return:                             0.8207623
+    avg_reward_per_step:                        0.13220215
+    proxy_corner:
+      avg_avg_return:                           0.049325533
+      avg_reward_per_step:                      0.0062561035
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     22.268572
+    avg_avg_return:                             0.8207623
+    avg_reward_per_step:                        0.13220215
+    proxy_corner:
+      avg_avg_return:                           0.049325533
+      avg_reward_per_step:                      0.0062561035
+  perf:
+    env_steps_per_second:                       351119.4982999316
+    ppo_updates_per_second:                     42.00069481315261
+    eval-batch-train-duration:                  0.5275676185265183
+    eval-batch-orig-duration:                   0.4823489664122462
+    eval-batch-shift-duration:                  0.4621460288763046
+    eval-batch-tree-duration:                   0.4979571755975485
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.661621
+      cheese-corne_dist_wavg:                   9.078429
+      levels_cheese_is_in_corner_avg:           0.021972656
+      mouse-cheese_dist_finite_avg:             25.062988
+      mouse-cheese_dist_finite_wavg:            9.709805
+      mouse-cheese_dist_solvable_avg:           25.062988
+      mouse-cheese_dist_solvable_wavg:          9.7098055
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.58760494
+    visit_patterns:
+      avg_first_visit_time:                     2371.2605
+      avg_last_visit_time:                      2730.2368
+      num_replay_batches:                       2740
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010719954
+    avg_actor_approxkl3:                        0.0006626943
+    avg_actor_clipfrac:                         0.022857666
+    avg_actor_loss:                             -0.0006594655
+    avg_advantage:                              0.019179355
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00258229
+    avg_entropy:                                0.0886535
+    avg_grad_norm_pre_clip:                     0.9747991
+    avg_loss:                                   0.00054302614
+    max:
+      max_actor_approxkl1:                      0.0029058391
+      max_actor_approxkl3:                      0.0012005367
+      max_actor_clipfrac:                       0.033325195
+      max_actor_loss:                           0.0008701682
+      max_advantage:                            0.1628294
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.004283951
+      max_entropy:                              0.10864887
+      max_grad_norm_pre_clip:                   1.6596553
+      max_loss:                                 0.002245292
+    std:
+      std_actor_approxkl1:                      0.00077565305
+      std_actor_approxkl3:                      0.0002753715
+      std_actor_clipfrac:                       0.0073193214
+      std_actor_loss:                           0.0010352632
+      std_advantage:                            0.07033947
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0008276466
+      std_entropy:                              0.010964633
+      std_grad_norm_pre_clip:                   0.33757693
+      std_loss:                                 0.0010123168
+  eval-batch-train:
+    avg_avg_episode_length:                     18.660677
+    avg_avg_return:                             0.9206586
+    avg_benchmark_regret:                       0.06655849
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.1293869
+    proxy_corner:
+      avg_avg_return:                           0.04677304
+      avg_reward_per_step:                      0.00504303
+  eval-batch-orig:
+    avg_avg_episode_length:                     23.432968
+    avg_avg_return:                             0.8964995
+    avg_benchmark_regret:                       0.08801208
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08479309
+    proxy_corner:
+      avg_avg_return:                           0.8964995
+      avg_reward_per_step:                      0.08479309
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.42202
+    avg_avg_return:                             0.92366624
+    avg_benchmark_regret:                       0.063839704
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14870453
+    proxy_corner:
+      avg_avg_return:                           0.046868205
+      avg_reward_per_step:                      0.0026168823
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.776438
+    avg_avg_return:                             0.8985602
+    avg_benchmark_regret:                       0.08607496
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.1000824
+    proxy_corner:
+      avg_avg_return:                           0.8985602
+      avg_reward_per_step:                      0.1000824
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        178257920
+    env-step-replay-before:                     90439680
+    env-step-used-before:                       178257920
+    ppo-update-before:                          108800
+    env-step-generate-before:                   87818240
+    env-step-all-after:                         178290688
+    env-step-replay-after:                      90472448
+    env-step-used-after:                        178290688
+    ppo-update-after:                           108820
+    env-step-generate-after:                    87818240
+  train-replay:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     66.434814
+    avg_avg_return:                             0.44192964
+    avg_reward_per_step:                        0.011779785
+    proxy_corner:
+      avg_avg_return:                           0.123317406
+      avg_reward_per_step:                      0.0027160645
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     66.434814
+    avg_avg_return:                             0.44192964
+    avg_reward_per_step:                        0.011779785
+    proxy_corner:
+      avg_avg_return:                           0.123317406
+      avg_reward_per_step:                      0.0027160645
+  perf:
+    env_steps_per_second:                       292245.16492641263
+    ppo_updates_per_second:                     41.540016006191244
+    eval-batch-train-duration:                  0.5397822046652436
+    eval-batch-orig-duration:                   0.4938820507377386
+    eval-batch-shift-duration:                  0.4903370635583997
+    eval-batch-tree-duration:                   0.4794489312916994
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.610107
+      cheese-corne_dist_wavg:                   14.476683
+      levels_cheese_is_in_corner_avg:           0.022949219
+      mouse-cheese_dist_finite_avg:             25.099365
+      mouse-cheese_dist_finite_wavg:            11.514391
+      mouse-cheese_dist_solvable_avg:           25.099365
+      mouse-cheese_dist_solvable_wavg:          11.514392
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.57141626
+    visit_patterns:
+      avg_first_visit_time:                     2381.7502
+      avg_last_visit_time:                      2751.2905
+      num_replay_batches:                       2761
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011849803
+    avg_actor_approxkl3:                        0.0011719986
+    avg_actor_clipfrac:                         0.050787356
+    avg_actor_loss:                             -0.0011457059
+    avg_advantage:                              -0.026445035
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.009250266
+    avg_entropy:                                0.25266522
+    avg_grad_norm_pre_clip:                     1.4706355
+    avg_loss:                                   0.003226762
+    max:
+      max_actor_approxkl1:                      0.0030817967
+      max_actor_approxkl3:                      0.002070195
+      max_actor_clipfrac:                       0.079711914
+      max_actor_loss:                           0.00030439347
+      max_advantage:                            0.17869526
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.010236315
+      max_entropy:                              0.26480788
+      max_grad_norm_pre_clip:                   2.1701632
+      max_loss:                                 0.005144988
+    std:
+      std_actor_approxkl1:                      0.0009151904
+      std_actor_approxkl3:                      0.0005969668
+      std_actor_clipfrac:                       0.02257296
+      std_actor_loss:                           0.00091144897
+      std_advantage:                            0.13469085
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00083189964
+      std_entropy:                              0.0079663405
+      std_grad_norm_pre_clip:                   0.3985049
+      std_loss:                                 0.0010974157
+  eval-batch-train:
+    avg_avg_episode_length:                     18.670307
+    avg_avg_return:                             0.9241861
+    avg_benchmark_regret:                       0.06303095
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12834167
+    proxy_corner:
+      avg_avg_return:                           0.059411474
+      avg_reward_per_step:                      0.0055007935
+  eval-batch-orig:
+    avg_avg_episode_length:                     23.659039
+    avg_avg_return:                             0.89149624
+    avg_benchmark_regret:                       0.09301531
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.087677
+    proxy_corner:
+      avg_avg_return:                           0.89149624
+      avg_reward_per_step:                      0.087677
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.270632
+    avg_avg_return:                             0.9266513
+    avg_benchmark_regret:                       0.06085468
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14703369
+    proxy_corner:
+      avg_avg_return:                           0.051495794
+      avg_reward_per_step:                      0.002822876
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.012045
+    avg_avg_return:                             0.8924943
+    avg_benchmark_regret:                       0.09214089
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.10095978
+    proxy_corner:
+      avg_avg_return:                           0.8924943
+      avg_reward_per_step:                      0.10095978
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        179306496
+    env-step-replay-before:                     90931200
+    env-step-used-before:                       179306496
+    ppo-update-before:                          109440
+    env-step-generate-before:                   88375296
+    env-step-all-after:                         179339264
+    env-step-replay-after:                      90931200
+    env-step-used-after:                        179339264
+    ppo-update-after:                           109460
+    env-step-generate-after:                    88408064
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     22.879478
+    avg_avg_return:                             0.82203436
+    avg_reward_per_step:                        0.10916138
+    proxy_corner:
+      avg_avg_return:                           0.046451867
+      avg_reward_per_step:                      0.002319336
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     22.879478
+    avg_avg_return:                             0.82203436
+    avg_reward_per_step:                        0.10916138
+    proxy_corner:
+      avg_avg_return:                           0.046451867
+      avg_reward_per_step:                      0.002319336
+  perf:
+    env_steps_per_second:                       347404.9727145921
+    ppo_updates_per_second:                     44.474407852611186
+    eval-batch-train-duration:                  0.5496943723410368
+    eval-batch-orig-duration:                   0.45571845304220915
+    eval-batch-shift-duration:                  0.48545876424759626
+    eval-batch-tree-duration:                   0.5036083050072193
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.561035
+      cheese-corne_dist_wavg:                   13.563544
+      levels_cheese_is_in_corner_avg:           0.0234375
+      mouse-cheese_dist_finite_avg:             25.051025
+      mouse-cheese_dist_finite_wavg:            11.505207
+      mouse-cheese_dist_solvable_avg:           25.051025
+      mouse-cheese_dist_solvable_wavg:          11.505208
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5724416
+    visit_patterns:
+      avg_first_visit_time:                     2394.7456
+      avg_last_visit_time:                      2765.3008
+      num_replay_batches:                       2775
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001611108
+    avg_actor_approxkl3:                        0.0007261808
+    avg_actor_clipfrac:                         0.024542237
+    avg_actor_loss:                             -0.0013213428
+    avg_advantage:                              0.090291694
+    avg_critic_clipfrac:                        0.3359253
+    avg_critic_loss:                            0.004100583
+    avg_entropy:                                0.0919919
+    avg_grad_norm_pre_clip:                     0.97939664
+    avg_loss:                                   0.0006369568
+    max:
+      max_actor_approxkl1:                      0.0033780506
+      max_actor_approxkl3:                      0.001782688
+      max_actor_clipfrac:                       0.052734375
+      max_actor_loss:                           0.0006730184
+      max_advantage:                            0.23878926
+      max_critic_clipfrac:                      1.0
+      max_critic_loss:                          0.0076294136
+      max_entropy:                              0.10703172
+      max_grad_norm_pre_clip:                   1.424232
+      max_loss:                                 0.0037236584
+    std:
+      std_actor_approxkl1:                      0.0011139531
+      std_actor_approxkl3:                      0.00044137254
+      std_actor_clipfrac:                       0.0121492455
+      std_actor_loss:                           0.00096155686
+      std_advantage:                            0.08128531
+      std_critic_clipfrac:                      0.41418797
+      std_critic_loss:                          0.0016013638
+      std_entropy:                              0.009038747
+      std_grad_norm_pre_clip:                   0.20071028
+      std_loss:                                 0.001452695
+  eval-batch-train:
+    avg_avg_episode_length:                     18.77021
+    avg_avg_return:                             0.9190241
+    avg_benchmark_regret:                       0.06819295
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12937164
+    proxy_corner:
+      avg_avg_return:                           0.052102268
+      avg_reward_per_step:                      0.005264282
+  eval-batch-orig:
+    avg_avg_episode_length:                     22.359383
+    avg_avg_return:                             0.90168965
+    avg_benchmark_regret:                       0.082821876
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.0867157
+    proxy_corner:
+      avg_avg_return:                           0.90168965
+      avg_reward_per_step:                      0.0867157
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.570225
+    avg_avg_return:                             0.92250395
+    avg_benchmark_regret:                       0.06500201
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14783478
+    proxy_corner:
+      avg_avg_return:                           0.046541095
+      avg_reward_per_step:                      0.0027313232
+  eval-batch-tree:
+    avg_avg_episode_length:                     21.71205
+    avg_avg_return:                             0.9040326
+    avg_benchmark_regret:                       0.08060263
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.10045624
+    proxy_corner:
+      avg_avg_return:                           0.9040326
+      avg_reward_per_step:                      0.10045624
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        180355072
+    env-step-replay-before:                     91291648
+    env-step-used-before:                       180355072
+    ppo-update-before:                          110080
+    env-step-generate-before:                   89063424
+    env-step-all-after:                         180387840
+    env-step-replay-after:                      91324416
+    env-step-used-after:                        180387840
+    ppo-update-after:                           110100
+    env-step-generate-after:                    89063424
+  train-replay:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     67.33827
+    avg_avg_return:                             0.4301306
+    avg_reward_per_step:                        0.011993408
+    proxy_corner:
+      avg_avg_return:                           0.133147
+      avg_reward_per_step:                      0.0024414062
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     67.33827
+    avg_avg_return:                             0.4301306
+    avg_reward_per_step:                        0.011993408
+    proxy_corner:
+      avg_avg_return:                           0.133147
+      avg_reward_per_step:                      0.0024414062
+  perf:
+    env_steps_per_second:                       730960.4838189239
+    ppo_updates_per_second:                     41.810072862496654
+    eval-batch-train-duration:                  0.35605464316904545
+    eval-batch-orig-duration:                   0.2463801596313715
+    eval-batch-shift-duration:                  0.25439684465527534
+    eval-batch-tree-duration:                   0.23522405419498682
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.571777
+      cheese-corne_dist_wavg:                   1.8884361
+      levels_cheese_is_in_corner_avg:           0.02319336
+      mouse-cheese_dist_finite_avg:             25.066162
+      mouse-cheese_dist_finite_wavg:            14.195648
+      mouse-cheese_dist_solvable_avg:           25.066162
+      mouse-cheese_dist_solvable_wavg:          14.195648
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.57260215
+    visit_patterns:
+      avg_first_visit_time:                     2409.7764
+      avg_last_visit_time:                      2777.2412
+      num_replay_batches:                       2787
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009254873
+    avg_actor_approxkl3:                        0.0010217865
+    avg_actor_clipfrac:                         0.04802246
+    avg_actor_loss:                             -0.00091376307
+    avg_advantage:                              -0.040788926
+    avg_critic_clipfrac:                        1.8310548e-05
+    avg_critic_loss:                            0.009779607
+    avg_entropy:                                0.22994287
+    avg_grad_norm_pre_clip:                     1.6261486
+    avg_loss:                                   0.0037460967
+    max:
+      max_actor_approxkl1:                      0.0023050937
+      max_actor_approxkl3:                      0.0020553358
+      max_actor_clipfrac:                       0.07727051
+      max_actor_loss:                           0.0014526695
+      max_advantage:                            0.14055789
+      max_critic_clipfrac:                      0.00036621094
+      max_critic_loss:                          0.014295331
+      max_entropy:                              0.2517335
+      max_grad_norm_pre_clip:                   3.1468194
+      max_loss:                                 0.00836457
+    std:
+      std_actor_approxkl1:                      0.00072145334
+      std_actor_approxkl3:                      0.0004032107
+      std_actor_clipfrac:                       0.015357561
+      std_actor_loss:                           0.0010891474
+      std_advantage:                            0.13892616
+      std_critic_clipfrac:                      7.981382e-05
+      std_critic_loss:                          0.0017264343
+      std_entropy:                              0.008964897
+      std_grad_norm_pre_clip:                   0.512375
+      std_loss:                                 0.0014600218
+  eval-batch-train:
+    avg_avg_episode_length:                     18.978512
+    avg_avg_return:                             0.91579723
+    avg_benchmark_regret:                       0.071419865
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12919617
+    proxy_corner:
+      avg_avg_return:                           0.050357286
+      avg_reward_per_step:                      0.0051574707
+  eval-batch-orig:
+    avg_avg_episode_length:                     23.271793
+    avg_avg_return:                             0.8984908
+    avg_benchmark_regret:                       0.08602073
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08478546
+    proxy_corner:
+      avg_avg_return:                           0.8984908
+      avg_reward_per_step:                      0.08478546
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.688383
+    avg_avg_return:                             0.9209888
+    avg_benchmark_regret:                       0.06651711
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.1452713
+    proxy_corner:
+      avg_avg_return:                           0.05215372
+      avg_reward_per_step:                      0.0029525757
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.003204
+    avg_avg_return:                             0.9040873
+    avg_benchmark_regret:                       0.0805479
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.1004715
+    proxy_corner:
+      avg_avg_return:                           0.9040873
+      avg_reward_per_step:                      0.1004715
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        181403648
+    env-step-replay-before:                     91947008
+    env-step-used-before:                       181403648
+    ppo-update-before:                          110720
+    env-step-generate-before:                   89456640
+    env-step-all-after:                         181436416
+    env-step-replay-after:                      91979776
+    env-step-used-after:                        181436416
+    ppo-update-after:                           110740
+    env-step-generate-after:                    89456640
+  train-replay:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     65.9427
+    avg_avg_return:                             0.44238847
+    avg_reward_per_step:                        0.012176514
+    proxy_corner:
+      avg_avg_return:                           0.10854018
+      avg_reward_per_step:                      0.0022888184
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     65.9427
+    avg_avg_return:                             0.44238847
+    avg_reward_per_step:                        0.012176514
+    proxy_corner:
+      avg_avg_return:                           0.10854018
+      avg_reward_per_step:                      0.0022888184
+  perf:
+    env_steps_per_second:                       298589.51120412536
+    ppo_updates_per_second:                     42.153367106756995
+    eval-batch-train-duration:                  0.5433286903426051
+    eval-batch-orig-duration:                   0.4784475900232792
+    eval-batch-shift-duration:                  0.4856133256107569
+    eval-batch-tree-duration:                   0.45217484794557095
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.447998
+      cheese-corne_dist_wavg:                   20.739788
+      levels_cheese_is_in_corner_avg:           0.022705078
+      mouse-cheese_dist_finite_avg:             25.043457
+      mouse-cheese_dist_finite_wavg:            12.403803
+      mouse-cheese_dist_solvable_avg:           25.043457
+      mouse-cheese_dist_solvable_wavg:          12.403801
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.54439735
+    visit_patterns:
+      avg_first_visit_time:                     2421.4294
+      avg_last_visit_time:                      2797.3513
+      num_replay_batches:                       2807
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0006725521
+    avg_actor_approxkl3:                        0.00093689916
+    avg_actor_clipfrac:                         0.042523194
+    avg_actor_loss:                             -0.0008853194
+    avg_advantage:                              -0.04093294
+    avg_critic_clipfrac:                        0.0012329102
+    avg_critic_loss:                            0.009904253
+    avg_entropy:                                0.22006989
+    avg_grad_norm_pre_clip:                     1.3950127
+    avg_loss:                                   0.0038467385
+    max:
+      max_actor_approxkl1:                      0.0016184971
+      max_actor_approxkl3:                      0.0015579774
+      max_actor_clipfrac:                       0.06542969
+      max_actor_loss:                           0.00072951615
+      max_advantage:                            0.15143931
+      max_critic_clipfrac:                      0.00793457
+      max_critic_loss:                          0.011355578
+      max_entropy:                              0.23384069
+      max_grad_norm_pre_clip:                   1.9820466
+      max_loss:                                 0.0057754195
+    std:
+      std_actor_approxkl1:                      0.00051791914
+      std_actor_approxkl3:                      0.00040932425
+      std_actor_clipfrac:                       0.016848104
+      std_actor_loss:                           0.00089875684
+      std_advantage:                            0.13921565
+      std_critic_clipfrac:                      0.0023973377
+      std_critic_loss:                          0.00065435976
+      std_entropy:                              0.007624273
+      std_grad_norm_pre_clip:                   0.36534357
+      std_loss:                                 0.0009638258
+  eval-batch-train:
+    avg_avg_episode_length:                     18.712395
+    avg_avg_return:                             0.9156177
+    avg_benchmark_regret:                       0.071599364
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12921906
+    proxy_corner:
+      avg_avg_return:                           0.04630863
+      avg_reward_per_step:                      0.005004883
+  eval-batch-orig:
+    avg_avg_episode_length:                     22.761234
+    avg_avg_return:                             0.9011798
+    avg_benchmark_regret:                       0.083331734
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08602905
+    proxy_corner:
+      avg_avg_return:                           0.9011798
+      avg_reward_per_step:                      0.08602905
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.308722
+    avg_avg_return:                             0.92225045
+    avg_benchmark_regret:                       0.06525549
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14743042
+    proxy_corner:
+      avg_avg_return:                           0.05275179
+      avg_reward_per_step:                      0.0029449463
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.309002
+    avg_avg_return:                             0.8984985
+    avg_benchmark_regret:                       0.08613673
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.099342346
+    proxy_corner:
+      avg_avg_return:                           0.8984985
+      avg_reward_per_step:                      0.099342346
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        182452224
+    env-step-replay-before:                     92471296
+    env-step-used-before:                       182452224
+    ppo-update-before:                          111360
+    env-step-generate-before:                   89980928
+    env-step-all-after:                         182484992
+    env-step-replay-after:                      92504064
+    env-step-used-after:                        182484992
+    ppo-update-after:                           111380
+    env-step-generate-after:                    89980928
+  train-replay:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     69.112816
+    avg_avg_return:                             0.39633614
+    avg_reward_per_step:                        0.010437012
+    proxy_corner:
+      avg_avg_return:                           0.13000657
+      avg_reward_per_step:                      0.0028076172
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     69.112816
+    avg_avg_return:                             0.39633614
+    avg_reward_per_step:                        0.010437012
+    proxy_corner:
+      avg_avg_return:                           0.13000657
+      avg_reward_per_step:                      0.0028076172
+  perf:
+    env_steps_per_second:                       320310.894275472
+    ppo_updates_per_second:                     42.122307191849984
+    eval-batch-train-duration:                  0.5423942571505904
+    eval-batch-orig-duration:                   0.4856715118512511
+    eval-batch-shift-duration:                  0.46271461714059114
+    eval-batch-tree-duration:                   0.45489150006324053
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.611084
+      cheese-corne_dist_wavg:                   8.183635
+      levels_cheese_is_in_corner_avg:           0.022949219
+      mouse-cheese_dist_finite_avg:             25.043701
+      mouse-cheese_dist_finite_wavg:            10.615158
+      mouse-cheese_dist_solvable_avg:           25.043701
+      mouse-cheese_dist_solvable_wavg:          10.615159
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.99999
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.5660937
+    visit_patterns:
+      avg_first_visit_time:                     2436.6226
+      avg_last_visit_time:                      2813.188
+      num_replay_batches:                       2823
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009399452
+    avg_actor_approxkl3:                        0.001015777
+    avg_actor_clipfrac:                         0.046002198
+    avg_actor_loss:                             -0.0010406092
+    avg_advantage:                              -0.051300704
+    avg_critic_clipfrac:                        0.0026367188
+    avg_critic_loss:                            0.01157663
+    avg_entropy:                                0.24733129
+    avg_grad_norm_pre_clip:                     1.5942677
+    avg_loss:                                   0.004500374
+    max:
+      max_actor_approxkl1:                      0.0028416086
+      max_actor_approxkl3:                      0.0018046105
+      max_actor_clipfrac:                       0.076171875
+      max_actor_loss:                           0.00089643896
+      max_advantage:                            0.14857364
+      max_critic_clipfrac:                      0.021850586
+      max_critic_loss:                          0.015271646
+      max_entropy:                              0.26567823
+      max_grad_norm_pre_clip:                   2.6267655
+      max_loss:                                 0.0073701446
+    std:
+      std_actor_approxkl1:                      0.00063348544
+      std_actor_approxkl3:                      0.00044867257
+      std_actor_clipfrac:                       0.017251445
+      std_actor_loss:                           0.0012342518
+      std_advantage:                            0.15069818
+      std_critic_clipfrac:                      0.0057138023
+      std_critic_loss:                          0.0013009495
+      std_entropy:                              0.009151791
+      std_grad_norm_pre_clip:                   0.44503525
+      std_loss:                                 0.0014794766
+  eval-batch-train:
+    avg_avg_episode_length:                     18.088627
+    avg_avg_return:                             0.92567074
+    avg_benchmark_regret:                       0.061546385
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12999725
+    proxy_corner:
+      avg_avg_return:                           0.05428279
+      avg_reward_per_step:                      0.005126953
+  eval-batch-orig:
+    avg_avg_episode_length:                     22.902798
+    avg_avg_return:                             0.8994481
+    avg_benchmark_regret:                       0.08506344
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08769226
+    proxy_corner:
+      avg_avg_return:                           0.8994481
+      avg_reward_per_step:                      0.08769226
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.263264
+    avg_avg_return:                             0.9245273
+    avg_benchmark_regret:                       0.062978625
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.1473999
+    proxy_corner:
+      avg_avg_return:                           0.056873195
+      avg_reward_per_step:                      0.0031738281
+  eval-batch-tree:
+    avg_avg_episode_length:                     21.644958
+    avg_avg_return:                             0.90111077
+    avg_benchmark_regret:                       0.08352446
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.10272217
+    proxy_corner:
+      avg_avg_return:                           0.90111077
+      avg_reward_per_step:                      0.10272217
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        183500800
+    env-step-replay-before:                     93028352
+    env-step-used-before:                       183500800
+    ppo-update-before:                          112000
+    env-step-generate-before:                   90472448
+    env-step-all-after:                         183533568
+    env-step-replay-after:                      93061120
+    env-step-used-after:                        183533568
+    ppo-update-after:                           112020
+    env-step-generate-after:                    90472448
+  train-replay:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     66.45402
+    avg_avg_return:                             0.4455386
+    avg_reward_per_step:                        0.012939453
+    proxy_corner:
+      avg_avg_return:                           0.1320394
+      avg_reward_per_step:                      0.0029296875
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     66.45402
+    avg_avg_return:                             0.4455386
+    avg_reward_per_step:                        0.012939453
+    proxy_corner:
+      avg_avg_return:                           0.1320394
+      avg_reward_per_step:                      0.0029296875
+  perf:
+    env_steps_per_second:                       245477.19702772846
+    ppo_updates_per_second:                     42.283678386692216
+    eval-batch-train-duration:                  0.6815166603773832
+    eval-batch-orig-duration:                   0.6398293850943446
+    eval-batch-shift-duration:                  0.6051573622971773
+    eval-batch-tree-duration:                   0.6240900568664074
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.690674
+      cheese-corne_dist_wavg:                   9.964368
+      levels_cheese_is_in_corner_avg:           0.02319336
+      mouse-cheese_dist_finite_avg:             25.152588
+      mouse-cheese_dist_finite_wavg:            7.005498
+      mouse-cheese_dist_solvable_avg:           25.152588
+      mouse-cheese_dist_solvable_wavg:          7.005499
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.5639343
+    visit_patterns:
+      avg_first_visit_time:                     2450.296
+      avg_last_visit_time:                      2830.364
+      num_replay_batches:                       2840
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0021500865
+    avg_actor_approxkl3:                        0.0016924823
+    avg_actor_clipfrac:                         0.06211548
+    avg_actor_loss:                             5.4257365e-05
+    avg_advantage:                              -0.03921015
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.009086244
+    avg_entropy:                                0.22484712
+    avg_grad_norm_pre_clip:                     1.7150341
+    avg_loss:                                   0.0043725325
+    max:
+      max_actor_approxkl1:                      0.004493183
+      max_actor_approxkl3:                      0.0037562714
+      max_actor_clipfrac:                       0.095947266
+      max_actor_loss:                           0.0025309324
+      max_advantage:                            0.13167542
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.012282279
+      max_entropy:                              0.23870382
+      max_grad_norm_pre_clip:                   2.697397
+      max_loss:                                 0.007872714
+    std:
+      std_actor_approxkl1:                      0.0011348929
+      std_actor_approxkl3:                      0.00088916265
+      std_actor_clipfrac:                       0.024249239
+      std_actor_loss:                           0.0012289862
+      std_advantage:                            0.13394608
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0013296247
+      std_entropy:                              0.0089540845
+      std_grad_norm_pre_clip:                   0.529402
+      std_loss:                                 0.0014730424
+  eval-batch-train:
+    avg_avg_episode_length:                     17.696644
+    avg_avg_return:                             0.9262252
+    avg_benchmark_regret:                       0.0609919
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12915802
+    proxy_corner:
+      avg_avg_return:                           0.05270836
+      avg_reward_per_step:                      0.0052261353
+  eval-batch-orig:
+    avg_avg_episode_length:                     22.48521
+    avg_avg_return:                             0.90247
+    avg_benchmark_regret:                       0.08204152
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08641815
+    proxy_corner:
+      avg_avg_return:                           0.90247
+      avg_reward_per_step:                      0.08641815
+  eval-batch-shift:
+    avg_avg_episode_length:                     17.745548
+    avg_avg_return:                             0.9341754
+    avg_benchmark_regret:                       0.05333058
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.1472168
+    proxy_corner:
+      avg_avg_return:                           0.055222277
+      avg_reward_per_step:                      0.0031051636
+  eval-batch-tree:
+    avg_avg_episode_length:                     21.620148
+    avg_avg_return:                             0.90817934
+    avg_benchmark_regret:                       0.07645585
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.10084534
+    proxy_corner:
+      avg_avg_return:                           0.90817934
+      avg_reward_per_step:                      0.10084534
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        184549376
+    env-step-replay-before:                     93618176
+    env-step-used-before:                       184549376
+    ppo-update-before:                          112640
+    env-step-generate-before:                   90931200
+    env-step-all-after:                         184582144
+    env-step-replay-after:                      93650944
+    env-step-used-after:                        184582144
+    ppo-update-after:                           112660
+    env-step-generate-after:                    90931200
+  train-replay:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     66.87939
+    avg_avg_return:                             0.42933565
+    avg_reward_per_step:                        0.011383057
+    proxy_corner:
+      avg_avg_return:                           0.10564004
+      avg_reward_per_step:                      0.0020141602
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     66.87939
+    avg_avg_return:                             0.42933565
+    avg_reward_per_step:                        0.011383057
+    proxy_corner:
+      avg_avg_return:                           0.10564004
+      avg_reward_per_step:                      0.0020141602
+  perf:
+    env_steps_per_second:                       355714.74749935954
+    ppo_updates_per_second:                     42.033302329842364
+    eval-batch-train-duration:                  0.5228801760822535
+    eval-batch-orig-duration:                   0.47985626943409443
+    eval-batch-shift-duration:                  0.4746576966717839
+    eval-batch-tree-duration:                   0.48505189549177885
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.855469
+      cheese-corne_dist_wavg:                   14.485714
+      levels_cheese_is_in_corner_avg:           0.022460938
+      mouse-cheese_dist_finite_avg:             25.16333
+      mouse-cheese_dist_finite_wavg:            10.61529
+      mouse-cheese_dist_solvable_avg:           25.16333
+      mouse-cheese_dist_solvable_wavg:          10.615293
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.55444795
+    visit_patterns:
+      avg_first_visit_time:                     2465.3162
+      avg_last_visit_time:                      2848.2866
+      num_replay_batches:                       2858
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0007248589
+    avg_actor_approxkl3:                        0.00091708975
+    avg_actor_clipfrac:                         0.042938232
+    avg_actor_loss:                             -0.0009727303
+    avg_advantage:                              -0.04279903
+    avg_critic_clipfrac:                        5.4931643e-05
+    avg_critic_loss:                            0.009910385
+    avg_entropy:                                0.22491348
+    avg_grad_norm_pre_clip:                     1.4014179
+    avg_loss:                                   0.0037575488
+    max:
+      max_actor_approxkl1:                      0.0014979972
+      max_actor_approxkl3:                      0.0015243848
+      max_actor_clipfrac:                       0.06591797
+      max_actor_loss:                           0.00079120696
+      max_advantage:                            0.12605363
+      max_critic_clipfrac:                      0.00061035156
+      max_critic_loss:                          0.011191693
+      max_entropy:                              0.24270667
+      max_grad_norm_pre_clip:                   2.0263124
+      max_loss:                                 0.005583078
+    std:
+      std_actor_approxkl1:                      0.00044092577
+      std_actor_approxkl3:                      0.00033293528
+      std_actor_clipfrac:                       0.01458926
+      std_actor_loss:                           0.0009874018
+      std_advantage:                            0.13950858
+      std_critic_clipfrac:                      0.00016592134
+      std_critic_loss:                          0.0008159057
+      std_entropy:                              0.0075303824
+      std_grad_norm_pre_clip:                   0.36717546
+      std_loss:                                 0.0010977607
+  eval-batch-train:
+    avg_avg_episode_length:                     18.558598
+    avg_avg_return:                             0.919911
+    avg_benchmark_regret:                       0.067306034
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12942505
+    proxy_corner:
+      avg_avg_return:                           0.0502263
+      avg_reward_per_step:                      0.004699707
+  eval-batch-orig:
+    avg_avg_episode_length:                     23.051426
+    avg_avg_return:                             0.903674
+    avg_benchmark_regret:                       0.08083753
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08721924
+    proxy_corner:
+      avg_avg_return:                           0.903674
+      avg_reward_per_step:                      0.08721924
+  eval-batch-shift:
+    avg_avg_episode_length:                     17.485928
+    avg_avg_return:                             0.93218505
+    avg_benchmark_regret:                       0.055320866
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14889526
+    proxy_corner:
+      avg_avg_return:                           0.056396887
+      avg_reward_per_step:                      0.0031585693
+  eval-batch-tree:
+    avg_avg_episode_length:                     21.914322
+    avg_avg_return:                             0.9061767
+    avg_benchmark_regret:                       0.078458555
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.10146332
+    proxy_corner:
+      avg_avg_return:                           0.9061767
+      avg_reward_per_step:                      0.10146332
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        185597952
+    env-step-replay-before:                     94142464
+    env-step-used-before:                       185597952
+    ppo-update-before:                          113280
+    env-step-generate-before:                   91455488
+    env-step-all-after:                         185630720
+    env-step-replay-after:                      94175232
+    env-step-used-after:                        185630720
+    ppo-update-after:                           113300
+    env-step-generate-after:                    91455488
+  train-replay:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     68.407166
+    avg_avg_return:                             0.41299996
+    avg_reward_per_step:                        0.011199951
+    proxy_corner:
+      avg_avg_return:                           0.1065017
+      avg_reward_per_step:                      0.002166748
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     68.407166
+    avg_avg_return:                             0.41299996
+    avg_reward_per_step:                        0.011199951
+    proxy_corner:
+      avg_avg_return:                           0.1065017
+      avg_reward_per_step:                      0.002166748
+  perf:
+    env_steps_per_second:                       378960.42755159223
+    ppo_updates_per_second:                     42.301306370216906
+    eval-batch-train-duration:                  0.4989687595516443
+    eval-batch-orig-duration:                   0.4259951673448086
+    eval-batch-shift-duration:                  0.42420264426618814
+    eval-batch-tree-duration:                   0.43154738564044237
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.867676
+      cheese-corne_dist_wavg:                   18.070202
+      levels_cheese_is_in_corner_avg:           0.022216797
+      mouse-cheese_dist_finite_avg:             25.199951
+      mouse-cheese_dist_finite_wavg:            13.320488
+      mouse-cheese_dist_solvable_avg:           25.199951
+      mouse-cheese_dist_solvable_wavg:          13.320485
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000002
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.5550668
+    visit_patterns:
+      avg_first_visit_time:                     2478.9773
+      avg_last_visit_time:                      2864.3003
+      num_replay_batches:                       2874
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0008462711
+    avg_actor_approxkl3:                        0.00071451184
+    avg_actor_clipfrac:                         0.03338623
+    avg_actor_loss:                             -0.0015248694
+    avg_advantage:                              -0.04094826
+    avg_critic_clipfrac:                        6.1035157e-06
+    avg_critic_loss:                            0.010534131
+    avg_entropy:                                0.23989204
+    avg_grad_norm_pre_clip:                     1.2798585
+    avg_loss:                                   0.0035023042
+    max:
+      max_actor_approxkl1:                      0.0026009216
+      max_actor_approxkl3:                      0.001465668
+      max_actor_clipfrac:                       0.07006836
+      max_actor_loss:                           0.00039581954
+      max_advantage:                            0.14071596
+      max_critic_clipfrac:                      0.00012207031
+      max_critic_loss:                          0.0130940955
+      max_entropy:                              0.26425213
+      max_grad_norm_pre_clip:                   2.3126807
+      max_loss:                                 0.006046613
+    std:
+      std_actor_approxkl1:                      0.0007076478
+      std_actor_approxkl3:                      0.00039770838
+      std_actor_clipfrac:                       0.018293755
+      std_actor_loss:                           0.0011097607
+      std_advantage:                            0.14417462
+      std_critic_clipfrac:                      2.6604606e-05
+      std_critic_loss:                          0.0014328895
+      std_entropy:                              0.0129985595
+      std_grad_norm_pre_clip:                   0.36768526
+      std_loss:                                 0.001357368
+  eval-batch-train:
+    avg_avg_episode_length:                     18.910717
+    avg_avg_return:                             0.9185672
+    avg_benchmark_regret:                       0.06864995
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12937164
+    proxy_corner:
+      avg_avg_return:                           0.04452394
+      avg_reward_per_step:                      0.0048599243
+  eval-batch-orig:
+    avg_avg_episode_length:                     23.236897
+    avg_avg_return:                             0.89950085
+    avg_benchmark_regret:                       0.08501074
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.081954956
+    proxy_corner:
+      avg_avg_return:                           0.89950085
+      avg_reward_per_step:                      0.081954956
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.066011
+    avg_avg_return:                             0.92540425
+    avg_benchmark_regret:                       0.062101692
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14927673
+    proxy_corner:
+      avg_avg_return:                           0.047477596
+      avg_reward_per_step:                      0.0027160645
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.538618
+    avg_avg_return:                             0.8966235
+    avg_benchmark_regret:                       0.08801164
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09623718
+    proxy_corner:
+      avg_avg_return:                           0.8966235
+      avg_reward_per_step:                      0.09623718
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        186646528
+    env-step-replay-before:                     94765056
+    env-step-used-before:                       186646528
+    ppo-update-before:                          113920
+    env-step-generate-before:                   91881472
+    env-step-all-after:                         186679296
+    env-step-replay-after:                      94797824
+    env-step-used-after:                        186679296
+    ppo-update-after:                           113940
+    env-step-generate-after:                    91881472
+  train-replay:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     66.98064
+    avg_avg_return:                             0.42373687
+    avg_reward_per_step:                        0.0113220215
+    proxy_corner:
+      avg_avg_return:                           0.11627555
+      avg_reward_per_step:                      0.0025939941
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     66.98064
+    avg_avg_return:                             0.42373687
+    avg_reward_per_step:                        0.0113220215
+    proxy_corner:
+      avg_avg_return:                           0.11627555
+      avg_reward_per_step:                      0.0025939941
+  perf:
+    env_steps_per_second:                       383186.2067583992
+    ppo_updates_per_second:                     42.04468549072963
+    eval-batch-train-duration:                  0.4888376798480749
+    eval-batch-orig-duration:                   0.42565998062491417
+    eval-batch-shift-duration:                  0.4424223704263568
+    eval-batch-tree-duration:                   0.4165463065728545
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.874512
+      cheese-corne_dist_wavg:                   14.483801
+      levels_cheese_is_in_corner_avg:           0.022216797
+      mouse-cheese_dist_finite_avg:             25.209717
+      mouse-cheese_dist_finite_wavg:            10.619421
+      mouse-cheese_dist_solvable_avg:           25.209717
+      mouse-cheese_dist_solvable_wavg:          10.619424
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.5539173
+    visit_patterns:
+      avg_first_visit_time:                     2491.5352
+      avg_last_visit_time:                      2883.225
+      num_replay_batches:                       2893
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0008007052
+    avg_actor_approxkl3:                        0.0010973738
+    avg_actor_clipfrac:                         0.046166994
+    avg_actor_loss:                             -0.0005616263
+    avg_advantage:                              -0.044542335
+    avg_critic_clipfrac:                        0.00023803712
+    avg_critic_loss:                            0.010443353
+    avg_entropy:                                0.22728758
+    avg_grad_norm_pre_clip:                     1.5745513
+    avg_loss:                                   0.0044327634
+    max:
+      max_actor_approxkl1:                      0.00240968
+      max_actor_approxkl3:                      0.0022759761
+      max_actor_clipfrac:                       0.075683594
+      max_actor_loss:                           0.0013381243
+      max_advantage:                            0.1391778
+      max_critic_clipfrac:                      0.001953125
+      max_critic_loss:                          0.013909562
+      max_entropy:                              0.24626301
+      max_grad_norm_pre_clip:                   2.7929032
+      max_loss:                                 0.007894822
+    std:
+      std_actor_approxkl1:                      0.000615432
+      std_actor_approxkl3:                      0.0005285398
+      std_actor_clipfrac:                       0.017561145
+      std_actor_loss:                           0.0010255129
+      std_advantage:                            0.14293353
+      std_critic_clipfrac:                      0.0005369359
+      std_critic_loss:                          0.0018055208
+      std_entropy:                              0.0116361175
+      std_grad_norm_pre_clip:                   0.53523666
+      std_loss:                                 0.0013395944
+  eval-batch-train:
+    avg_avg_episode_length:                     18.334457
+    avg_avg_return:                             0.9236798
+    avg_benchmark_regret:                       0.06353725
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12886047
+    proxy_corner:
+      avg_avg_return:                           0.04832703
+      avg_reward_per_step:                      0.005104065
+  eval-batch-orig:
+    avg_avg_episode_length:                     23.408188
+    avg_avg_return:                             0.902475
+    avg_benchmark_regret:                       0.082036525
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08451843
+    proxy_corner:
+      avg_avg_return:                           0.902475
+      avg_reward_per_step:                      0.08451843
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.187387
+    avg_avg_return:                             0.9307765
+    avg_benchmark_regret:                       0.056729455
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14893341
+    proxy_corner:
+      avg_avg_return:                           0.051565338
+      avg_reward_per_step:                      0.0030899048
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.954536
+    avg_avg_return:                             0.8958412
+    avg_benchmark_regret:                       0.08879405
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.09792328
+    proxy_corner:
+      avg_avg_return:                           0.8958412
+      avg_reward_per_step:                      0.09792328
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        187695104
+    env-step-replay-before:                     95420416
+    env-step-used-before:                       187695104
+    ppo-update-before:                          114560
+    env-step-generate-before:                   92274688
+    env-step-all-after:                         187727872
+    env-step-replay-after:                      95420416
+    env-step-used-after:                        187727872
+    ppo-update-after:                           114580
+    env-step-generate-after:                    92307456
+  train-generate:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     19.956608
+    avg_avg_return:                             0.8471378
+    avg_reward_per_step:                        0.13961792
+    proxy_corner:
+      avg_avg_return:                           0.05443653
+      avg_reward_per_step:                      0.0049438477
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     19.956608
+    avg_avg_return:                             0.8471378
+    avg_reward_per_step:                        0.13961792
+    proxy_corner:
+      avg_avg_return:                           0.05443653
+      avg_reward_per_step:                      0.0049438477
+  perf:
+    env_steps_per_second:                       318283.7516210575
+    ppo_updates_per_second:                     41.61487630199212
+    eval-batch-train-duration:                  0.5264046946540475
+    eval-batch-orig-duration:                   0.4804180199280381
+    eval-batch-shift-duration:                  0.47533013485372066
+    eval-batch-tree-duration:                   0.46947930846363306
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.900146
+      cheese-corne_dist_wavg:                   25.297892
+      levels_cheese_is_in_corner_avg:           0.022949219
+      mouse-cheese_dist_finite_avg:             25.305176
+      mouse-cheese_dist_finite_wavg:            12.434093
+      mouse-cheese_dist_solvable_avg:           25.305176
+      mouse-cheese_dist_solvable_wavg:          12.434093
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.54258066
+    visit_patterns:
+      avg_first_visit_time:                     2505.1492
+      avg_last_visit_time:                      2902.1313
+      num_replay_batches:                       2912
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0015841492
+    avg_actor_approxkl3:                        0.0009353633
+    avg_actor_clipfrac:                         0.024505615
+    avg_actor_loss:                             -0.0007062195
+    avg_advantage:                              0.03876001
+    avg_critic_clipfrac:                        0.00023803712
+    avg_critic_loss:                            0.0017700881
+    avg_entropy:                                0.07855555
+    avg_grad_norm_pre_clip:                     0.96471274
+    avg_loss:                                   0.0001002691
+    max:
+      max_actor_approxkl1:                      0.0040996624
+      max_actor_approxkl3:                      0.0021387248
+      max_actor_clipfrac:                       0.05041504
+      max_actor_loss:                           0.00012304634
+      max_advantage:                            0.17032313
+      max_critic_clipfrac:                      0.001953125
+      max_critic_loss:                          0.0028365778
+      max_entropy:                              0.101743944
+      max_grad_norm_pre_clip:                   1.7895479
+      max_loss:                                 0.0010742564
+    std:
+      std_actor_approxkl1:                      0.0012567269
+      std_actor_approxkl3:                      0.0006607179
+      std_actor_clipfrac:                       0.013328651
+      std_actor_loss:                           0.0007279104
+      std_advantage:                            0.05645221
+      std_critic_clipfrac:                      0.00057835435
+      std_critic_loss:                          0.0005077195
+      std_entropy:                              0.012684938
+      std_grad_norm_pre_clip:                   0.3572535
+      std_loss:                                 0.0007386329
+  eval-batch-train:
+    avg_avg_episode_length:                     18.349327
+    avg_avg_return:                             0.9197049
+    avg_benchmark_regret:                       0.06751217
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12915039
+    proxy_corner:
+      avg_avg_return:                           0.04602476
+      avg_reward_per_step:                      0.005050659
+  eval-batch-orig:
+    avg_avg_episode_length:                     23.117104
+    avg_avg_return:                             0.8980633
+    avg_benchmark_regret:                       0.08644828
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.0864563
+    proxy_corner:
+      avg_avg_return:                           0.8980633
+      avg_reward_per_step:                      0.0864563
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.480843
+    avg_avg_return:                             0.92225367
+    avg_benchmark_regret:                       0.06525226
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.1482315
+    proxy_corner:
+      avg_avg_return:                           0.051431667
+      avg_reward_per_step:                      0.0028762817
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.059624
+    avg_avg_return:                             0.90786076
+    avg_benchmark_regret:                       0.07677445
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.10134888
+    proxy_corner:
+      avg_avg_return:                           0.90786076
+      avg_reward_per_step:                      0.10134888
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        188743680
+    env-step-replay-before:                     95911936
+    env-step-used-before:                       188743680
+    ppo-update-before:                          115200
+    env-step-generate-before:                   92831744
+    env-step-all-after:                         188776448
+    env-step-replay-after:                      95911936
+    env-step-used-after:                        188776448
+    ppo-update-after:                           115220
+    env-step-generate-after:                    92864512
+  train-generate:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     21.238556
+    avg_avg_return:                             0.83106863
+    avg_reward_per_step:                        0.12689209
+    proxy_corner:
+      avg_avg_return:                           0.051100373
+      avg_reward_per_step:                      0.0028686523
+  train-all:
+    shift_proportion:                           0.99609375
+    avg_avg_episode_length:                     21.238556
+    avg_avg_return:                             0.83106863
+    avg_reward_per_step:                        0.12689209
+    proxy_corner:
+      avg_avg_return:                           0.051100373
+      avg_reward_per_step:                      0.0028686523
+  perf:
+    env_steps_per_second:                       355683.13537951146
+    ppo_updates_per_second:                     41.10402700912875
+    eval-batch-train-duration:                  0.5273187356069684
+    eval-batch-orig-duration:                   0.47951389104127884
+    eval-batch-shift-duration:                  0.4348641065880656
+    eval-batch-tree-duration:                   0.46303136367350817
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.954346
+      cheese-corne_dist_wavg:                   10.002892
+      levels_cheese_is_in_corner_avg:           0.022949219
+      mouse-cheese_dist_finite_avg:             25.394775
+      mouse-cheese_dist_finite_wavg:            12.438905
+      mouse-cheese_dist_solvable_avg:           25.394775
+      mouse-cheese_dist_solvable_wavg:          12.438907
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5642854
+    visit_patterns:
+      avg_first_visit_time:                     2522.4844
+      avg_last_visit_time:                      2917.3188
+      num_replay_batches:                       2927
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012571117
+    avg_actor_approxkl3:                        0.0010282645
+    avg_actor_clipfrac:                         0.02695923
+    avg_actor_loss:                             -0.00086722395
+    avg_advantage:                              0.03377303
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0022664908
+    avg_entropy:                                0.081099965
+    avg_grad_norm_pre_clip:                     1.0608041
+    avg_loss:                                   0.00018492148
+    max:
+      max_actor_approxkl1:                      0.0051003224
+      max_actor_approxkl3:                      0.0028266893
+      max_actor_clipfrac:                       0.056762695
+      max_actor_loss:                           0.0007678047
+      max_advantage:                            0.15719676
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00449865
+      max_entropy:                              0.09672223
+      max_grad_norm_pre_clip:                   2.0828872
+      max_loss:                                 0.0017047448
+    std:
+      std_actor_approxkl1:                      0.0012067676
+      std_actor_approxkl3:                      0.0007066728
+      std_actor_clipfrac:                       0.013153058
+      std_actor_loss:                           0.0010793965
+      std_advantage:                            0.06567173
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0009813017
+      std_entropy:                              0.01037086
+      std_grad_norm_pre_clip:                   0.5060159
+      std_loss:                                 0.0010253434
+  eval-batch-train:
+    avg_avg_episode_length:                     18.466318
+    avg_avg_return:                             0.92294526
+    avg_benchmark_regret:                       0.064271815
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12930298
+    proxy_corner:
+      avg_avg_return:                           0.04829173
+      avg_reward_per_step:                      0.004760742
+  eval-batch-orig:
+    avg_avg_episode_length:                     22.813175
+    avg_avg_return:                             0.8938441
+    avg_benchmark_regret:                       0.0906674
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08969116
+    proxy_corner:
+      avg_avg_return:                           0.8938441
+      avg_reward_per_step:                      0.08969116
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.29742
+    avg_avg_return:                             0.92118883
+    avg_benchmark_regret:                       0.066317104
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14813232
+    proxy_corner:
+      avg_avg_return:                           0.05881267
+      avg_reward_per_step:                      0.0032653809
+  eval-batch-tree:
+    avg_avg_episode_length:                     21.68959
+    avg_avg_return:                             0.9015031
+    avg_benchmark_regret:                       0.083132125
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.102142334
+    proxy_corner:
+      avg_avg_return:                           0.9015031
+      avg_reward_per_step:                      0.102142334
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        189792256
+    env-step-replay-before:                     96501760
+    env-step-used-before:                       189792256
+    ppo-update-before:                          115840
+    env-step-generate-before:                   93290496
+    env-step-all-after:                         189825024
+    env-step-replay-after:                      96534528
+    env-step-used-after:                        189825024
+    ppo-update-after:                           115860
+    env-step-generate-after:                    93290496
+  train-replay:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     69.50033
+    avg_avg_return:                             0.4137361
+    avg_reward_per_step:                        0.010528564
+    proxy_corner:
+      avg_avg_return:                           0.11413249
+      avg_reward_per_step:                      0.002319336
+  train-all:
+    shift_proportion:                           0.97265625
+    avg_avg_episode_length:                     69.50033
+    avg_avg_return:                             0.4137361
+    avg_reward_per_step:                        0.010528564
+    proxy_corner:
+      avg_avg_return:                           0.11413249
+      avg_reward_per_step:                      0.002319336
+  perf:
+    env_steps_per_second:                       321415.8159302952
+    ppo_updates_per_second:                     41.75639982123186
+    eval-batch-train-duration:                  0.535436512902379
+    eval-batch-orig-duration:                   0.4809167245402932
+    eval-batch-shift-duration:                  0.47485385555773973
+    eval-batch-tree-duration:                   0.43822088558226824
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    19.034668
+      cheese-corne_dist_wavg:                   9.122483
+      levels_cheese_is_in_corner_avg:           0.02319336
+      mouse-cheese_dist_finite_avg:             25.412842
+      mouse-cheese_dist_finite_wavg:            10.632987
+      mouse-cheese_dist_solvable_avg:           25.412842
+      mouse-cheese_dist_solvable_wavg:          10.632984
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000002
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.54996717
+    visit_patterns:
+      avg_first_visit_time:                     2538.376
+      avg_last_visit_time:                      2936.1794
+      num_replay_batches:                       2946
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010488344
+    avg_actor_approxkl3:                        0.00094400515
+    avg_actor_clipfrac:                         0.04381714
+    avg_actor_loss:                             -0.0012019873
+    avg_advantage:                              -0.029324962
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.009677001
+    avg_entropy:                                0.24641965
+    avg_grad_norm_pre_clip:                     1.3215498
+    avg_loss:                                   0.003390094
+    max:
+      max_actor_approxkl1:                      0.0029025418
+      max_actor_approxkl3:                      0.0015942018
+      max_actor_clipfrac:                       0.068603516
+      max_actor_loss:                           0.0011003017
+      max_advantage:                            0.17998207
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.012299718
+      max_entropy:                              0.27044398
+      max_grad_norm_pre_clip:                   2.2604024
+      max_loss:                                 0.005730043
+    std:
+      std_actor_approxkl1:                      0.0006817459
+      std_actor_approxkl3:                      0.0004138153
+      std_actor_clipfrac:                       0.017948465
+      std_actor_loss:                           0.0011663496
+      std_advantage:                            0.13827999
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011041532
+      std_entropy:                              0.014290691
+      std_grad_norm_pre_clip:                   0.40839586
+      std_loss:                                 0.0012437825
+  eval-batch-train:
+    avg_avg_episode_length:                     18.07095
+    avg_avg_return:                             0.92858815
+    avg_benchmark_regret:                       0.05862888
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12835693
+    proxy_corner:
+      avg_avg_return:                           0.051734716
+      avg_reward_per_step:                      0.0051727295
+  eval-batch-orig:
+    avg_avg_episode_length:                     23.130331
+    avg_avg_return:                             0.8931587
+    avg_benchmark_regret:                       0.09135285
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08807373
+    proxy_corner:
+      avg_avg_return:                           0.8931587
+      avg_reward_per_step:                      0.08807373
+  eval-batch-shift:
+    avg_avg_episode_length:                     17.744217
+    avg_avg_return:                             0.9349499
+    avg_benchmark_regret:                       0.052556094
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14752197
+    proxy_corner:
+      avg_avg_return:                           0.05579996
+      avg_reward_per_step:                      0.003364563
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.694876
+    avg_avg_return:                             0.90150523
+    avg_benchmark_regret:                       0.08312993
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.10116577
+    proxy_corner:
+      avg_avg_return:                           0.90150523
+      avg_reward_per_step:                      0.10116577
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        190840832
+    env-step-replay-before:                     96894976
+    env-step-used-before:                       190840832
+    ppo-update-before:                          116480
+    env-step-generate-before:                   93945856
+    env-step-all-after:                         190873600
+    env-step-replay-after:                      96894976
+    env-step-used-after:                        190873600
+    ppo-update-after:                           116500
+    env-step-generate-after:                    93978624
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     21.65913
+    avg_avg_return:                             0.8378134
+    avg_reward_per_step:                        0.111968994
+    proxy_corner:
+      avg_avg_return:                           0.058223456
+      avg_reward_per_step:                      0.0034484863
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     21.65913
+    avg_avg_return:                             0.8378134
+    avg_reward_per_step:                        0.111968994
+    proxy_corner:
+      avg_avg_return:                           0.058223456
+      avg_reward_per_step:                      0.0034484863
+  perf:
+    env_steps_per_second:                       349343.9587505243
+    ppo_updates_per_second:                     40.638380972437936
+    eval-batch-train-duration:                  0.5339173078536987
+    eval-batch-orig-duration:                   0.4877164224162698
+    eval-batch-shift-duration:                  0.4834521487355232
+    eval-batch-tree-duration:                   0.5088437628000975
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.926514
+      cheese-corne_dist_wavg:                   10.013908
+      levels_cheese_is_in_corner_avg:           0.022949219
+      mouse-cheese_dist_finite_avg:             25.438232
+      mouse-cheese_dist_finite_wavg:            12.44525
+      mouse-cheese_dist_solvable_avg:           25.438232
+      mouse-cheese_dist_solvable_wavg:          12.44525
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.99998
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.55347025
+    visit_patterns:
+      avg_first_visit_time:                     2554.433
+      avg_last_visit_time:                      2947.3442
+      num_replay_batches:                       2957
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011448973
+    avg_actor_approxkl3:                        0.0007630899
+    avg_actor_clipfrac:                         0.022155762
+    avg_actor_loss:                             -0.00061302417
+    avg_advantage:                              0.02495119
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0013650946
+    avg_entropy:                                0.07675996
+    avg_grad_norm_pre_clip:                     1.1120784
+    avg_loss:                                   -7.236854e-06
+    max:
+      max_actor_approxkl1:                      0.0030866838
+      max_actor_approxkl3:                      0.0018555536
+      max_actor_clipfrac:                       0.03894043
+      max_actor_loss:                           0.0015362911
+      max_advantage:                            0.13528049
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0032698403
+      max_entropy:                              0.09243834
+      max_grad_norm_pre_clip:                   2.615667
+      max_loss:                                 0.001998714
+    std:
+      std_actor_approxkl1:                      0.0008273659
+      std_actor_approxkl3:                      0.00048719667
+      std_actor_clipfrac:                       0.010582201
+      std_actor_loss:                           0.00092150853
+      std_advantage:                            0.05101466
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00063949206
+      std_entropy:                              0.008483009
+      std_grad_norm_pre_clip:                   0.58874714
+      std_loss:                                 0.0008717841
+  eval-batch-train:
+    avg_avg_episode_length:                     17.75259
+    avg_avg_return:                             0.9285519
+    avg_benchmark_regret:                       0.05866521
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.13005829
+    proxy_corner:
+      avg_avg_return:                           0.05185631
+      avg_reward_per_step:                      0.0052261353
+  eval-batch-orig:
+    avg_avg_episode_length:                     22.10816
+    avg_avg_return:                             0.9055541
+    avg_benchmark_regret:                       0.07895742
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.087875366
+    proxy_corner:
+      avg_avg_return:                           0.9055541
+      avg_reward_per_step:                      0.087875366
+  eval-batch-shift:
+    avg_avg_episode_length:                     17.767563
+    avg_avg_return:                             0.9268534
+    avg_benchmark_regret:                       0.060652554
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14956665
+    proxy_corner:
+      avg_avg_return:                           0.053231347
+      avg_reward_per_step:                      0.0030822754
+  eval-batch-tree:
+    avg_avg_episode_length:                     21.453266
+    avg_avg_return:                             0.9102222
+    avg_benchmark_regret:                       0.07441299
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.103393555
+    proxy_corner:
+      avg_avg_return:                           0.9102222
+      avg_reward_per_step:                      0.103393555
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        191889408
+    env-step-replay-before:                     97484800
+    env-step-used-before:                       191889408
+    ppo-update-before:                          117120
+    env-step-generate-before:                   94404608
+    env-step-all-after:                         191922176
+    env-step-replay-after:                      97517568
+    env-step-used-after:                        191922176
+    ppo-update-after:                           117140
+    env-step-generate-after:                    94404608
+  train-replay:
+    shift_proportion:                           0.96875
+    avg_avg_episode_length:                     69.198135
+    avg_avg_return:                             0.40859342
+    avg_reward_per_step:                        0.010223389
+    proxy_corner:
+      avg_avg_return:                           0.1403361
+      avg_reward_per_step:                      0.0026855469
+  train-all:
+    shift_proportion:                           0.96875
+    avg_avg_episode_length:                     69.198135
+    avg_avg_return:                             0.40859342
+    avg_reward_per_step:                        0.010223389
+    proxy_corner:
+      avg_avg_return:                           0.1403361
+      avg_reward_per_step:                      0.0026855469
+  perf:
+    env_steps_per_second:                       256529.1186148309
+    ppo_updates_per_second:                     42.95694419603017
+    eval-batch-train-duration:                  0.6713901041075587
+    eval-batch-orig-duration:                   0.6187591040506959
+    eval-batch-shift-duration:                  0.5887662721797824
+    eval-batch-tree-duration:                   0.6201497111469507
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.936035
+      cheese-corne_dist_wavg:                   6.4057074
+      levels_cheese_is_in_corner_avg:           0.022460938
+      mouse-cheese_dist_finite_avg:             25.50708
+      mouse-cheese_dist_finite_wavg:            9.752736
+      mouse-cheese_dist_solvable_avg:           25.50708
+      mouse-cheese_dist_solvable_wavg:          9.752736
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000008
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.56343937
+    visit_patterns:
+      avg_first_visit_time:                     2565.9004
+      avg_last_visit_time:                      2966.2998
+      num_replay_batches:                       2976
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0008775932
+    avg_actor_approxkl3:                        0.0010320129
+    avg_actor_clipfrac:                         0.044543456
+    avg_actor_loss:                             -0.0016885862
+    avg_advantage:                              -0.039697878
+    avg_critic_clipfrac:                        1.8310548e-05
+    avg_critic_loss:                            0.01026777
+    avg_entropy:                                0.24622667
+    avg_grad_norm_pre_clip:                     1.2344326
+    avg_loss:                                   0.0031990719
+    max:
+      max_actor_approxkl1:                      0.0022663563
+      max_actor_approxkl3:                      0.0017758077
+      max_actor_clipfrac:                       0.07495117
+      max_actor_loss:                           0.00036416948
+      max_advantage:                            0.14991063
+      max_critic_clipfrac:                      0.00024414062
+      max_critic_loss:                          0.012173592
+      max_entropy:                              0.26441553
+      max_grad_norm_pre_clip:                   1.8178645
+      max_loss:                                 0.0061914353
+    std:
+      std_actor_approxkl1:                      0.00063715107
+      std_actor_approxkl3:                      0.00052127754
+      std_actor_clipfrac:                       0.020753099
+      std_actor_loss:                           0.0012196881
+      std_advantage:                            0.14218117
+      std_critic_clipfrac:                      5.822383e-05
+      std_critic_loss:                          0.00087905675
+      std_entropy:                              0.010900582
+      std_grad_norm_pre_clip:                   0.28995782
+      std_loss:                                 0.0013692861
+  eval-batch-train:
+    avg_avg_episode_length:                     17.875069
+    avg_avg_return:                             0.9257678
+    avg_benchmark_regret:                       0.06144932
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12895966
+    proxy_corner:
+      avg_avg_return:                           0.05178744
+      avg_reward_per_step:                      0.005264282
+  eval-batch-orig:
+    avg_avg_episode_length:                     22.116503
+    avg_avg_return:                             0.90674996
+    avg_benchmark_regret:                       0.07776153
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.088531494
+    proxy_corner:
+      avg_avg_return:                           0.90674996
+      avg_reward_per_step:                      0.088531494
+  eval-batch-shift:
+    avg_avg_episode_length:                     17.749477
+    avg_avg_return:                             0.9239765
+    avg_benchmark_regret:                       0.06352945
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14990234
+    proxy_corner:
+      avg_avg_return:                           0.053432256
+      avg_reward_per_step:                      0.0031051636
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.195702
+    avg_avg_return:                             0.90183765
+    avg_benchmark_regret:                       0.08279757
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.10293579
+    proxy_corner:
+      avg_avg_return:                           0.90183765
+      avg_reward_per_step:                      0.10293579
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        192937984
+    env-step-replay-before:                     97910784
+    env-step-used-before:                       192937984
+    ppo-update-before:                          117760
+    env-step-generate-before:                   95027200
+    env-step-all-after:                         192970752
+    env-step-replay-after:                      97943552
+    env-step-used-after:                        192970752
+    ppo-update-after:                           117780
+    env-step-generate-after:                    95027200
+  train-replay:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     66.14808
+    avg_avg_return:                             0.44516355
+    avg_reward_per_step:                        0.012298584
+    proxy_corner:
+      avg_avg_return:                           0.11277147
+      avg_reward_per_step:                      0.002166748
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     66.14808
+    avg_avg_return:                             0.44516355
+    avg_reward_per_step:                        0.012298584
+    proxy_corner:
+      avg_avg_return:                           0.11277147
+      avg_reward_per_step:                      0.002166748
+  perf:
+    env_steps_per_second:                       293813.6160530633
+    ppo_updates_per_second:                     42.46737908941826
+    eval-batch-train-duration:                  0.606146008707583
+    eval-batch-orig-duration:                   0.5077955350279808
+    eval-batch-shift-duration:                  0.48408376332372427
+    eval-batch-tree-duration:                   0.4515312183648348
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    19.017334
+      cheese-corne_dist_wavg:                   10.044508
+      levels_cheese_is_in_corner_avg:           0.021484375
+      mouse-cheese_dist_finite_avg:             25.367188
+      mouse-cheese_dist_finite_wavg:            12.436952
+      mouse-cheese_dist_solvable_avg:           25.367188
+      mouse-cheese_dist_solvable_wavg:          12.436954
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.55471534
+    visit_patterns:
+      avg_first_visit_time:                     2584.9836
+      avg_last_visit_time:                      2979.2146
+      num_replay_batches:                       2989
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011501656
+    avg_actor_approxkl3:                        0.0010059521
+    avg_actor_clipfrac:                         0.047442626
+    avg_actor_loss:                             -0.001099474
+    avg_advantage:                              -0.012988501
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00897748
+    avg_entropy:                                0.2359761
+    avg_grad_norm_pre_clip:                     1.2283229
+    avg_loss:                                   0.003153291
+    max:
+      max_actor_approxkl1:                      0.0024480731
+      max_actor_approxkl3:                      0.0019968292
+      max_actor_clipfrac:                       0.07385254
+      max_actor_loss:                           0.00054995716
+      max_advantage:                            0.1949684
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.010433552
+      max_entropy:                              0.25658932
+      max_grad_norm_pre_clip:                   1.9160755
+      max_loss:                                 0.004967301
+    std:
+      std_actor_approxkl1:                      0.00074486615
+      std_actor_approxkl3:                      0.0004723718
+      std_actor_clipfrac:                       0.019657122
+      std_actor_loss:                           0.00095928623
+      std_advantage:                            0.13383858
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0009958699
+      std_entropy:                              0.01120556
+      std_grad_norm_pre_clip:                   0.35116574
+      std_loss:                                 0.0010693904
+  eval-batch-train:
+    avg_avg_episode_length:                     17.57349
+    avg_avg_return:                             0.92924875
+    avg_benchmark_regret:                       0.05796835
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.13063812
+    proxy_corner:
+      avg_avg_return:                           0.051746517
+      avg_reward_per_step:                      0.0051193237
+  eval-batch-orig:
+    avg_avg_episode_length:                     22.185343
+    avg_avg_return:                             0.9062165
+    avg_benchmark_regret:                       0.078295015
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08679199
+    proxy_corner:
+      avg_avg_return:                           0.9062165
+      avg_reward_per_step:                      0.08679199
+  eval-batch-shift:
+    avg_avg_episode_length:                     17.912552
+    avg_avg_return:                             0.9245887
+    avg_benchmark_regret:                       0.062917314
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14837646
+    proxy_corner:
+      avg_avg_return:                           0.052804608
+      avg_reward_per_step:                      0.0032730103
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.375317
+    avg_avg_return:                             0.9013158
+    avg_benchmark_regret:                       0.08331942
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.10108948
+    proxy_corner:
+      avg_avg_return:                           0.9013158
+      avg_reward_per_step:                      0.10108948
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        193986560
+    env-step-replay-before:                     98402304
+    env-step-used-before:                       193986560
+    ppo-update-before:                          118400
+    env-step-generate-before:                   95584256
+    env-step-all-after:                         194019328
+    env-step-replay-after:                      98435072
+    env-step-used-after:                        194019328
+    ppo-update-after:                           118420
+    env-step-generate-after:                    95584256
+  train-replay:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     69.09398
+    avg_avg_return:                             0.38550285
+    avg_reward_per_step:                        0.010192871
+    proxy_corner:
+      avg_avg_return:                           0.11485943
+      avg_reward_per_step:                      0.0021362305
+  train-all:
+    shift_proportion:                           0.9921875
+    avg_avg_episode_length:                     69.09398
+    avg_avg_return:                             0.38550285
+    avg_reward_per_step:                        0.010192871
+    proxy_corner:
+      avg_avg_return:                           0.11485943
+      avg_reward_per_step:                      0.0021362305
+  perf:
+    env_steps_per_second:                       299946.400738646
+    ppo_updates_per_second:                     41.96973922994986
+    eval-batch-train-duration:                  0.5376084009185433
+    eval-batch-orig-duration:                   0.47179339453577995
+    eval-batch-shift-duration:                  0.4477950343862176
+    eval-batch-tree-duration:                   0.49386530742049217
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    19.020996
+      cheese-corne_dist_wavg:                   5.506545
+      levels_cheese_is_in_corner_avg:           0.022216797
+      mouse-cheese_dist_finite_avg:             25.426758
+      mouse-cheese_dist_finite_wavg:            10.636816
+      mouse-cheese_dist_solvable_avg:           25.426758
+      mouse-cheese_dist_solvable_wavg:          10.636817
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.57365006
+    visit_patterns:
+      avg_first_visit_time:                     2597.4905
+      avg_last_visit_time:                      2994.2603
+      num_replay_batches:                       3004
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00096467696
+    avg_actor_approxkl3:                        0.0011426368
+    avg_actor_clipfrac:                         0.051068116
+    avg_actor_loss:                             -0.00040430352
+    avg_advantage:                              -0.043225467
+    avg_critic_clipfrac:                        3.6621095e-05
+    avg_critic_loss:                            0.011342465
+    avg_entropy:                                0.2342489
+    avg_grad_norm_pre_clip:                     1.8024445
+    avg_loss:                                   0.0050326795
+    max:
+      max_actor_approxkl1:                      0.001997856
+      max_actor_approxkl3:                      0.0017914975
+      max_actor_clipfrac:                       0.079956055
+      max_actor_loss:                           0.0023831725
+      max_advantage:                            0.16273272
+      max_critic_clipfrac:                      0.00036621094
+      max_critic_loss:                          0.0130320685
+      max_entropy:                              0.24571553
+      max_grad_norm_pre_clip:                   2.7483225
+      max_loss:                                 0.008036648
+    std:
+      std_actor_approxkl1:                      0.00058242417
+      std_actor_approxkl3:                      0.00042927684
+      std_actor_clipfrac:                       0.01787154
+      std_actor_loss:                           0.0013295882
+      std_advantage:                            0.14943095
+      std_critic_clipfrac:                      0.000109863286
+      std_critic_loss:                          0.0010046412
+      std_entropy:                              0.00578872
+      std_grad_norm_pre_clip:                   0.44156042
+      std_loss:                                 0.0015294706
+  eval-batch-train:
+    avg_avg_episode_length:                     17.68341
+    avg_avg_return:                             0.9299574
+    avg_benchmark_regret:                       0.05725964
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.13004303
+    proxy_corner:
+      avg_avg_return:                           0.045032024
+      avg_reward_per_step:                      0.0050964355
+  eval-batch-orig:
+    avg_avg_episode_length:                     23.091927
+    avg_avg_return:                             0.8950804
+    avg_benchmark_regret:                       0.089431144
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08895874
+    proxy_corner:
+      avg_avg_return:                           0.8950804
+      avg_reward_per_step:                      0.08895874
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.01042
+    avg_avg_return:                             0.9278877
+    avg_benchmark_regret:                       0.05961829
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14868164
+    proxy_corner:
+      avg_avg_return:                           0.049072772
+      avg_reward_per_step:                      0.002937317
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.301622
+    avg_avg_return:                             0.8982657
+    avg_benchmark_regret:                       0.08636951
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.10214996
+    proxy_corner:
+      avg_avg_return:                           0.8982657
+      avg_reward_per_step:                      0.10214996
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        195035136
+    env-step-replay-before:                     98762752
+    env-step-used-before:                       195035136
+    ppo-update-before:                          119040
+    env-step-generate-before:                   96272384
+    env-step-all-after:                         195067904
+    env-step-replay-after:                      98795520
+    env-step-used-after:                        195067904
+    ppo-update-after:                           119060
+    env-step-generate-after:                    96272384
+  train-replay:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     66.92873
+    avg_avg_return:                             0.4346848
+    avg_reward_per_step:                        0.011199951
+    proxy_corner:
+      avg_avg_return:                           0.078003004
+      avg_reward_per_step:                      0.0016174316
+  train-all:
+    shift_proportion:                           0.98046875
+    avg_avg_episode_length:                     66.92873
+    avg_avg_return:                             0.4346848
+    avg_reward_per_step:                        0.011199951
+    proxy_corner:
+      avg_avg_return:                           0.078003004
+      avg_reward_per_step:                      0.0016174316
+  perf:
+    env_steps_per_second:                       295891.4396757749
+    ppo_updates_per_second:                     42.18845096083303
+    eval-batch-train-duration:                  0.5356308631598949
+    eval-batch-orig-duration:                   0.48305194173008204
+    eval-batch-shift-duration:                  0.47735694982111454
+    eval-batch-tree-duration:                   0.4533624714240432
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    18.960938
+      cheese-corne_dist_wavg:                   8.210843
+      levels_cheese_is_in_corner_avg:           0.021484375
+      mouse-cheese_dist_finite_avg:             25.368652
+      mouse-cheese_dist_finite_wavg:            10.635822
+      mouse-cheese_dist_solvable_avg:           25.368652
+      mouse-cheese_dist_solvable_wavg:          10.635822
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5640688
+    visit_patterns:
+      avg_first_visit_time:                     2613.0447
+      avg_last_visit_time:                      3005.3464
+      num_replay_batches:                       3015
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010958305
+    avg_actor_approxkl3:                        0.00090903306
+    avg_actor_clipfrac:                         0.042401124
+    avg_actor_loss:                             -0.0011234109
+    avg_advantage:                              -0.042593684
+    avg_critic_clipfrac:                        5.4931643e-05
+    avg_critic_loss:                            0.00955652
+    avg_entropy:                                0.22340024
+    avg_grad_norm_pre_clip:                     1.395264
+    avg_loss:                                   0.0034314487
+    max:
+      max_actor_approxkl1:                      0.003110749
+      max_actor_approxkl3:                      0.001522775
+      max_actor_clipfrac:                       0.06311035
+      max_actor_loss:                           0.0008818507
+      max_advantage:                            0.122853994
+      max_critic_clipfrac:                      0.00048828125
+      max_critic_loss:                          0.01164159
+      max_entropy:                              0.23848489
+      max_grad_norm_pre_clip:                   2.3247552
+      max_loss:                                 0.0058469805
+    std:
+      std_actor_approxkl1:                      0.0006626019
+      std_actor_approxkl3:                      0.0003093789
+      std_actor_clipfrac:                       0.012904297
+      std_actor_loss:                           0.0011521698
+      std_advantage:                            0.1370588
+      std_critic_clipfrac:                      0.00014686532
+      std_critic_loss:                          0.0011319447
+      std_entropy:                              0.006681338
+      std_grad_norm_pre_clip:                   0.3906481
+      std_loss:                                 0.001281212
+  eval-batch-train:
+    avg_avg_episode_length:                     17.242718
+    avg_avg_return:                             0.9272063
+    avg_benchmark_regret:                       0.060010828
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.13214111
+    proxy_corner:
+      avg_avg_return:                           0.04732497
+      avg_reward_per_step:                      0.00504303
+  eval-batch-orig:
+    avg_avg_episode_length:                     22.286427
+    avg_avg_return:                             0.9070097
+    avg_benchmark_regret:                       0.077501826
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08735657
+    proxy_corner:
+      avg_avg_return:                           0.9070097
+      avg_reward_per_step:                      0.08735657
+  eval-batch-shift:
+    avg_avg_episode_length:                     18.079323
+    avg_avg_return:                             0.92669463
+    avg_benchmark_regret:                       0.060811304
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.15034485
+    proxy_corner:
+      avg_avg_return:                           0.04875703
+      avg_reward_per_step:                      0.0028839111
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.091333
+    avg_avg_return:                             0.9017468
+    avg_benchmark_regret:                       0.082888424
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.1006546
+    proxy_corner:
+      avg_avg_return:                           0.9017468
+      avg_reward_per_step:                      0.1006546
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        196083712
+    env-step-replay-before:                     99319808
+    env-step-used-before:                       196083712
+    ppo-update-before:                          119680
+    env-step-generate-before:                   96763904
+    env-step-all-after:                         196116480
+    env-step-replay-after:                      99319808
+    env-step-used-after:                        196116480
+    ppo-update-after:                           119700
+    env-step-generate-after:                    96796672
+  train-generate:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     18.782513
+    avg_avg_return:                             0.8503336
+    avg_reward_per_step:                        0.12615967
+    proxy_corner:
+      avg_avg_return:                           0.05063189
+      avg_reward_per_step:                      0.0043029785
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     18.782513
+    avg_avg_return:                             0.8503336
+    avg_reward_per_step:                        0.12615967
+    proxy_corner:
+      avg_avg_return:                           0.05063189
+      avg_reward_per_step:                      0.0043029785
+  perf:
+    env_steps_per_second:                       324918.5492539461
+    ppo_updates_per_second:                     42.41126265912264
+    eval-batch-train-duration:                  0.5056323986500502
+    eval-batch-orig-duration:                   0.4822640046477318
+    eval-batch-shift-duration:                  0.4523675097152591
+    eval-batch-tree-duration:                   0.4937671096995473
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    19.01709
+      cheese-corne_dist_wavg:                   5.5257854
+      levels_cheese_is_in_corner_avg:           0.021972656
+      mouse-cheese_dist_finite_avg:             25.462402
+      mouse-cheese_dist_finite_wavg:            10.649551
+      mouse-cheese_dist_solvable_avg:           25.462402
+      mouse-cheese_dist_solvable_wavg:          10.649552
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5711925
+    visit_patterns:
+      avg_first_visit_time:                     2625.9424
+      avg_last_visit_time:                      3021.3042
+      num_replay_batches:                       3031
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0008254382
+    avg_actor_approxkl3:                        0.00074849575
+    avg_actor_clipfrac:                         0.023614502
+    avg_actor_loss:                             -0.00036485502
+    avg_advantage:                              0.0028313175
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0016612636
+    avg_entropy:                                0.07173189
+    avg_grad_norm_pre_clip:                     1.3285209
+    avg_loss:                                   0.0003940449
+    max:
+      max_actor_approxkl1:                      0.002441788
+      max_actor_approxkl3:                      0.0013955998
+      max_actor_clipfrac:                       0.036499023
+      max_actor_loss:                           0.001197001
+      max_advantage:                            0.1277824
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0032130943
+      max_entropy:                              0.090787604
+      max_grad_norm_pre_clip:                   2.6467779
+      max_loss:                                 0.0020726246
+    std:
+      std_actor_approxkl1:                      0.0006377293
+      std_actor_approxkl3:                      0.00036109757
+      std_actor_clipfrac:                       0.008879344
+      std_actor_loss:                           0.0009808927
+      std_advantage:                            0.057710536
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00065373594
+      std_entropy:                              0.01025079
+      std_grad_norm_pre_clip:                   0.5331582
+      std_loss:                                 0.0009899937
+  eval-batch-train:
+    avg_avg_episode_length:                     18.321857
+    avg_avg_return:                             0.9190211
+    avg_benchmark_regret:                       0.06819589
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.13059235
+    proxy_corner:
+      avg_avg_return:                           0.053010657
+      avg_reward_per_step:                      0.005180359
+  eval-batch-orig:
+    avg_avg_episode_length:                     22.647835
+    avg_avg_return:                             0.90442324
+    avg_benchmark_regret:                       0.08008833
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08803558
+    proxy_corner:
+      avg_avg_return:                           0.90442324
+      avg_reward_per_step:                      0.08803558
+  eval-batch-shift:
+    avg_avg_episode_length:                     17.372852
+    avg_avg_return:                             0.9301926
+    avg_benchmark_regret:                       0.057313353
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.15138245
+    proxy_corner:
+      avg_avg_return:                           0.05448401
+      avg_reward_per_step:                      0.0031433105
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.553108
+    avg_avg_return:                             0.899123
+    avg_benchmark_regret:                       0.08551221
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.10202789
+    proxy_corner:
+      avg_avg_return:                           0.899123
+      avg_reward_per_step:                      0.10202789
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        197132288
+    env-step-replay-before:                     99876864
+    env-step-used-before:                       197132288
+    ppo-update-before:                          120320
+    env-step-generate-before:                   97255424
+    env-step-all-after:                         197165056
+    env-step-replay-after:                      99876864
+    env-step-used-after:                        197165056
+    ppo-update-after:                           120340
+    env-step-generate-after:                    97288192
+  train-generate:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     20.979977
+    avg_avg_return:                             0.8343929
+    avg_reward_per_step:                        0.12310791
+    proxy_corner:
+      avg_avg_return:                           0.040893126
+      avg_reward_per_step:                      0.0017700195
+  train-all:
+    shift_proportion:                           0.9765625
+    avg_avg_episode_length:                     20.979977
+    avg_avg_return:                             0.8343929
+    avg_reward_per_step:                        0.12310791
+    proxy_corner:
+      avg_avg_return:                           0.040893126
+      avg_reward_per_step:                      0.0017700195
+  perf:
+    env_steps_per_second:                       324982.67141180934
+    ppo_updates_per_second:                     42.05287117562035
+    eval-batch-train-duration:                  0.5388175714761019
+    eval-batch-orig-duration:                   0.4799526622518897
+    eval-batch-shift-duration:                  0.4804316107183695
+    eval-batch-tree-duration:                   0.443692646920681
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    19.007324
+      cheese-corne_dist_wavg:                   25.286592
+      levels_cheese_is_in_corner_avg:           0.021240234
+      mouse-cheese_dist_finite_avg:             25.481934
+      mouse-cheese_dist_finite_wavg:            12.456596
+      mouse-cheese_dist_solvable_avg:           25.481934
+      mouse-cheese_dist_solvable_wavg:          12.456598
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.56661844
+    visit_patterns:
+      avg_first_visit_time:                     2640.0505
+      avg_last_visit_time:                      3038.2004
+      num_replay_batches:                       3048
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0008535011
+    avg_actor_approxkl3:                        0.00070356845
+    avg_actor_clipfrac:                         0.018994141
+    avg_actor_loss:                             -0.0006176171
+    avg_advantage:                              -0.0007964147
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0018182828
+    avg_entropy:                                0.07024205
+    avg_grad_norm_pre_clip:                     0.8926369
+    avg_loss:                                   0.00022128236
+    max:
+      max_actor_approxkl1:                      0.0031594043
+      max_actor_approxkl3:                      0.0018429151
+      max_actor_clipfrac:                       0.03930664
+      max_actor_loss:                           0.00053116307
+      max_advantage:                            0.12353706
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0032548404
+      max_entropy:                              0.092624545
+      max_grad_norm_pre_clip:                   1.5284743
+      max_loss:                                 0.0015315128
+    std:
+      std_actor_approxkl1:                      0.0008378601
+      std_actor_approxkl3:                      0.00053997844
+      std_actor_clipfrac:                       0.010718885
+      std_actor_loss:                           0.0007628883
+      std_advantage:                            0.06068591
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00077512255
+      std_entropy:                              0.0111601725
+      std_grad_norm_pre_clip:                   0.27916348
+      std_loss:                                 0.00081146194
+  eval-batch-train:
+    avg_avg_episode_length:                     18.207188
+    avg_avg_return:                             0.9221467
+    avg_benchmark_regret:                       0.065070376
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.12954712
+    proxy_corner:
+      avg_avg_return:                           0.05831251
+      avg_reward_per_step:                      0.005340576
+  eval-batch-orig:
+    avg_avg_episode_length:                     22.954624
+    avg_avg_return:                             0.8984015
+    avg_benchmark_regret:                       0.08611008
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08786011
+    proxy_corner:
+      avg_avg_return:                           0.8984015
+      avg_reward_per_step:                      0.08786011
+  eval-batch-shift:
+    avg_avg_episode_length:                     17.476002
+    avg_avg_return:                             0.93100727
+    avg_benchmark_regret:                       0.05649865
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14902496
+    proxy_corner:
+      avg_avg_return:                           0.05551301
+      avg_reward_per_step:                      0.0030899048
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.822582
+    avg_avg_return:                             0.8962562
+    avg_benchmark_regret:                       0.088378966
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.10157013
+    proxy_corner:
+      avg_avg_return:                           0.8962562
+      avg_reward_per_step:                      0.10157013
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        198180864
+    env-step-replay-before:                     100335616
+    env-step-used-before:                       198180864
+    ppo-update-before:                          120960
+    env-step-generate-before:                   97845248
+    env-step-all-after:                         198213632
+    env-step-replay-after:                      100368384
+    env-step-used-after:                        198213632
+    ppo-update-after:                           120980
+    env-step-generate-after:                    97845248
+  train-replay:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     67.34403
+    avg_avg_return:                             0.41411117
+    avg_reward_per_step:                        0.011199951
+    proxy_corner:
+      avg_avg_return:                           0.13981925
+      avg_reward_per_step:                      0.0026550293
+  train-all:
+    shift_proportion:                           0.984375
+    avg_avg_episode_length:                     67.34403
+    avg_avg_return:                             0.41411117
+    avg_reward_per_step:                        0.011199951
+    proxy_corner:
+      avg_avg_return:                           0.13981925
+      avg_reward_per_step:                      0.0026550293
+  perf:
+    env_steps_per_second:                       351646.7459635685
+    ppo_updates_per_second:                     40.54819040714955
+    eval-batch-train-duration:                  0.5164810540154576
+    eval-batch-orig-duration:                   0.4492323035374284
+    eval-batch-shift-duration:                  0.48306434880942106
+    eval-batch-tree-duration:                   0.47505452018231153
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    19.060547
+      cheese-corne_dist_wavg:                   30.691095
+      levels_cheese_is_in_corner_avg:           0.020019531
+      mouse-cheese_dist_finite_avg:             25.48706
+      mouse-cheese_dist_finite_wavg:            13.353181
+      mouse-cheese_dist_solvable_avg:           25.48706
+      mouse-cheese_dist_solvable_wavg:          13.353184
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5478136
+    visit_patterns:
+      avg_first_visit_time:                     2653.0967
+      avg_last_visit_time:                      3053.3567
+      num_replay_batches:                       3063
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010852576
+    avg_actor_approxkl3:                        0.0010714039
+    avg_actor_clipfrac:                         0.0463501
+    avg_actor_loss:                             -0.0010000609
+    avg_advantage:                              -0.047262363
+    avg_critic_clipfrac:                        0.0020080567
+    avg_critic_loss:                            0.010773699
+    avg_entropy:                                0.23884936
+    avg_grad_norm_pre_clip:                     1.5250415
+    avg_loss:                                   0.0041479394
+    max:
+      max_actor_approxkl1:                      0.0021457477
+      max_actor_approxkl3:                      0.002206353
+      max_actor_clipfrac:                       0.07385254
+      max_actor_loss:                           0.001029849
+      max_advantage:                            0.121975124
+      max_critic_clipfrac:                      0.021972656
+      max_critic_loss:                          0.015932184
+      max_entropy:                              0.25276607
+      max_grad_norm_pre_clip:                   2.784618
+      max_loss:                                 0.007713361
+    std:
+      std_actor_approxkl1:                      0.0007281703
+      std_actor_approxkl3:                      0.00052906224
+      std_actor_clipfrac:                       0.01960219
+      std_actor_loss:                           0.0010115787
+      std_advantage:                            0.14501104
+      std_critic_clipfrac:                      0.0050713527
+      std_critic_loss:                          0.0016117275
+      std_entropy:                              0.009037806
+      std_grad_norm_pre_clip:                   0.52938145
+      std_loss:                                 0.0012586254
+  eval-batch-train:
+    avg_avg_episode_length:                     17.458023
+    avg_avg_return:                             0.9286054
+    avg_benchmark_regret:                       0.058611736
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.13014221
+    proxy_corner:
+      avg_avg_return:                           0.049983617
+      avg_reward_per_step:                      0.005180359
+  eval-batch-orig:
+    avg_avg_episode_length:                     21.986343
+    avg_avg_return:                             0.907614
+    avg_benchmark_regret:                       0.076897524
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08709717
+    proxy_corner:
+      avg_avg_return:                           0.907614
+      avg_reward_per_step:                      0.08709717
+  eval-batch-shift:
+    avg_avg_episode_length:                     17.324116
+    avg_avg_return:                             0.93342566
+    avg_benchmark_regret:                       0.0540803
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.14936066
+    proxy_corner:
+      avg_avg_return:                           0.05594181
+      avg_reward_per_step:                      0.0030899048
+  eval-batch-tree:
+    avg_avg_episode_length:                     21.58487
+    avg_avg_return:                             0.90435284
+    avg_benchmark_regret:                       0.08028234
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.10243225
+    proxy_corner:
+      avg_avg_return:                           0.90435284
+      avg_reward_per_step:                      0.10243225
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        199229440
+    env-step-replay-before:                     100728832
+    env-step-used-before:                       199229440
+    ppo-update-before:                          121600
+    env-step-generate-before:                   98500608
+    env-step-all-after:                         199262208
+    env-step-replay-after:                      100728832
+    env-step-used-after:                        199262208
+    ppo-update-after:                           121620
+    env-step-generate-after:                    98533376
+  train-generate:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     18.801508
+    avg_avg_return:                             0.8580208
+    avg_reward_per_step:                        0.12808228
+    proxy_corner:
+      avg_avg_return:                           0.0642087
+      avg_reward_per_step:                      0.0043029785
+  train-all:
+    shift_proportion:                           0.98828125
+    avg_avg_episode_length:                     18.801508
+    avg_avg_return:                             0.8580208
+    avg_reward_per_step:                        0.12808228
+    proxy_corner:
+      avg_avg_return:                           0.0642087
+      avg_reward_per_step:                      0.0043029785
+  perf:
+    env_steps_per_second:                       261686.6138904752
+    ppo_updates_per_second:                     41.606837171188516
+    eval-batch-train-duration:                  0.5347478911280632
+    eval-batch-orig-duration:                   0.494891369715333
+    eval-batch-shift-duration:                  0.47184806130826473
+    eval-batch-tree-duration:                   0.4597616083920002
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    19.151123
+      cheese-corne_dist_wavg:                   14.51621
+      levels_cheese_is_in_corner_avg:           0.01928711
+      mouse-cheese_dist_finite_avg:             25.545166
+      mouse-cheese_dist_finite_wavg:            10.653342
+      mouse-cheese_dist_solvable_avg:           25.545166
+      mouse-cheese_dist_solvable_wavg:          10.653343
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.57223356
+    visit_patterns:
+      avg_first_visit_time:                     2668.7388
+      avg_last_visit_time:                      3064.3093
+      num_replay_batches:                       3074
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0007053248
+    avg_actor_approxkl3:                        0.00081647275
+    avg_actor_clipfrac:                         0.021368409
+    avg_actor_loss:                             -0.0007236534
+    avg_advantage:                              0.005406253
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0012592782
+    avg_entropy:                                0.070607975
+    avg_grad_norm_pre_clip:                     0.9306866
+    avg_loss:                                   -0.00016462228
+    max:
+      max_actor_approxkl1:                      0.0023480877
+      max_actor_approxkl3:                      0.0014691849
+      max_actor_clipfrac:                       0.032592773
+      max_actor_loss:                           0.0019360017
+      max_advantage:                            0.14421862
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0026545217
+      max_entropy:                              0.0884183
+      max_grad_norm_pre_clip:                   1.9458073
+      max_loss:                                 0.0023719184
+    std:
+      std_actor_approxkl1:                      0.00070345326
+      std_actor_approxkl3:                      0.0003774528
+      std_actor_clipfrac:                       0.007598568
+      std_actor_loss:                           0.0012073023
+      std_advantage:                            0.050340485
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0005409594
+      std_entropy:                              0.010983216
+      std_grad_norm_pre_clip:                   0.34160534
+      std_loss:                                 0.001174113
+  eval-batch-train:
+    avg_avg_episode_length:                     17.10165
+    avg_avg_return:                             0.9328053
+    avg_benchmark_regret:                       0.054411773
+    avg_benchmark_return:                       0.98721707
+    avg_reward_per_step:                        0.13169098
+    proxy_corner:
+      avg_avg_return:                           0.045279853
+      avg_reward_per_step:                      0.005004883
+  eval-batch-orig:
+    avg_avg_episode_length:                     22.534046
+    avg_avg_return:                             0.90139425
+    avg_benchmark_regret:                       0.08311727
+    avg_benchmark_return:                       0.98451155
+    avg_reward_per_step:                        0.08799744
+    proxy_corner:
+      avg_avg_return:                           0.90139425
+      avg_reward_per_step:                      0.08799744
+  eval-batch-shift:
+    avg_avg_episode_length:                     17.760918
+    avg_avg_return:                             0.92387295
+    avg_benchmark_regret:                       0.06363297
+    avg_benchmark_return:                       0.9875059
+    avg_reward_per_step:                        0.1519165
+    proxy_corner:
+      avg_avg_return:                           0.05327779
+      avg_reward_per_step:                      0.0031204224
+  eval-batch-tree:
+    avg_avg_episode_length:                     21.89338
+    avg_avg_return:                             0.9000949
+    avg_benchmark_regret:                       0.08454028
+    avg_benchmark_return:                       0.98463523
+    avg_reward_per_step:                        0.10421753
+    proxy_corner:
+      avg_avg_return:                           0.9000949
+      avg_reward_per_step:                      0.10421753
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+training run complete.
+finishing checkpoints...
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/requirements.txt b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/requirements.txt
new file mode 100644
index 0000000000000000000000000000000000000000..7babfd51e296217241ffeb19de181f45421dc883
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/requirements.txt
@@ -0,0 +1,253 @@
+kiwisolver==1.4.8
+aulis==0.1.0
+tokenizers==0.20.3
+xmltodict==0.14.2
+hf-xet==1.1.4
+timaeus-cli==0.1.0
+jsonlines==4.0.0
+DataProperty==1.1.0
+timaeus-aether==0.0.0
+nvidia-nvtx-cu12==12.6.77
+cachetools==5.5.2
+devinterp==1.3.2
+wcwidth==0.2.13
+pydantic_core==2.33.2
+tqdm-multiprocess==0.0.11
+protobuf==5.29.5
+decorator==4.4.2
+pyasn1==0.6.1
+zipp==3.23.0
+pre_commit==4.2.0
+typepy==1.3.4
+etils==1.13.0
+shellingham==1.5.4
+aiohttp==3.12.13
+prettytable==3.16.0
+readchar==4.2.1
+sentencepiece==0.2.0
+seaborn==0.13.2
+paramiko==3.5.1
+cftime==1.6.4.post1
+tabledata==1.3.4
+colorama==0.4.6
+GitPython==3.1.44
+cycler==0.12.1
+tcolorpy==0.1.7
+cfgv==3.4.0
+psutil==7.0.0
+aioitertools==0.12.0
+aiosignal==1.3.2
+humanize==4.12.3
+termcolor==3.1.0
+zope.interface==7.2
+pathvalidate==3.3.1
+nodeenv==1.9.1
+mbstrdecoder==1.1.4
+chardet==5.2.0
+zope.event==5.0
+cloudpickle==3.1.1
+opt_einsum==3.4.0
+MarkupSafe==3.0.2
+requests_ntlm==1.3.0
+pybind11==2.13.6
+fonttools==4.58.4
+gast==0.6.0
+packaging==25.0
+proglog==0.1.12
+Jinja2==3.1.6
+wadler_lindig==0.1.6
+fancy-einsum==0.0.3
+iniconfig==2.1.0
+pluggy==1.6.0
+idna==3.10
+nest-asyncio==1.6.0
+better-abc==0.0.3
+matplotlib==3.10.3
+portalocker==3.2.0
+tqdm==4.67.1
+smmap==5.0.2
+rich==14.0.0
+distro==1.9.0
+xarray==2025.6.1
+bcrypt==4.3.0
+regex==2024.11.6
+aiobotocore==2.22.0
+jmespath==1.0.1
+frozenlist==1.7.0
+typer==0.16.0
+cryptography==45.0.4
+python-dotenv==1.1.0
+pytest-xdist==3.7.0
+boto3==1.37.3
+grpc-google-iam-v1==0.14.2
+threadpoolctl==3.6.0
+aiohappyeyeballs==2.6.1
+pywinrm==0.5.0
+platformdirs==4.3.8
+nvidia-cuda-runtime-cu12==12.6.77
+aiofiles==24.1.0
+google-api-core==2.25.1
+proto-plus==1.26.1
+nvidia-cufile-cu12==1.11.1.6
+mpmath==1.3.0
+PyYAML==6.0.2
+filelock==3.18.0
+typing-inspection==0.4.1
+syrupy==4.9.1
+distlib==0.3.9
+annotated-types==0.7.0
+more-itertools==10.7.0
+multiprocess==0.70.16
+grpcio-status==1.71.2
+six==1.17.0
+requests==2.32.4
+mdurl==0.1.2
+pyinfra==3.3.1
+typing_extensions==4.14.0
+greenlet==3.2.3
+lxml==5.4.0
+einops==0.8.1
+ssh-python==1.1.1
+networkx==3.5
+attrs==25.3.0
+pyasn1_modules==0.4.2
+toolz==1.0.0
+joblib==1.5.1
+diskcache==5.6.3
+multidict==6.5.0
+gitdb==4.0.12
+transformer-lens==2.15.4
+execnet==2.1.1
+safetensors==0.5.3
+rsa==4.9.1
+zstandard==0.23.0
+google-cloud-compute==1.31.0
+rouge_score==0.1.2
+importlib_resources==6.5.2
+cffi==1.17.1
+pyspnego==0.11.2
+aioboto3==14.3.0
+Pygments==2.19.1
+tabulate==0.9.0
+typeguard==4.4.3
+click==8.2.1
+nvidia-cuda-cupti-cu12==12.6.80
+word2number==1.1
+pycparser==2.22
+gevent==25.5.1
+transformers-stream-generator==0.0.5
+sqlitedict==2.1.0
+datasets==3.6.0
+netCDF4==1.7.2
+msgpack==1.1.1
+urllib3==2.4.0
+pillow==11.2.1
+PyNaCl==1.5.0
+parallel-ssh==2.14.0
+nltk==3.9.1
+python-dateutil==2.9.0.post0
+yarl==1.20.1
+pyparsing==3.2.3
+ml_dtypes==0.5.3
+evaluate==0.4.3
+numexpr==2.11.0
+pydantic==2.11.7
+ssh2-python==1.1.2.post1
+moviepy==2.2.1
+markdown-it-py==3.0.0
+googleapis-common-protos==1.70.0
+xxhash==3.5.0
+dill==0.3.8
+sacrebleu==2.5.1
+identify==2.6.12
+fsspec==2025.3.0
+dm-tree==0.1.9
+chex==0.1.89
+huggingface-hub==0.33.0
+certifi==2025.6.15
+imageio-ffmpeg==0.6.0
+scipy==1.15.3
+pytz==2025.2
+pytablewriter==1.2.1
+optax==0.2.5
+google-cloud-tpu==1.23.2
+nvidia-curand-cu12==10.3.7.77
+charset-normalizer==3.4.2
+wandb-workspaces==0.1.16
+s3transfer==0.11.3
+absl-py==2.3.0
+accelerate==1.7.0
+wrapt==1.17.2
+narwhals==1.43.0
+virtualenv==20.31.2
+peft==0.15.2
+triton==3.3.1
+sympy==1.14.0
+nvidia-cusolver-cu12==11.7.1.2
+contourpy==1.3.2
+nvidia-cusparselt-cu12==0.6.3
+nvidia-cufft-cu12==11.3.0.4
+ruff==0.12.0
+nvidia-nccl-cu12==2.26.2
+nvidia-cusparse-cu12==12.5.4.2
+google-cloud-secret-manager==2.24.0
+pytest==7.4.3
+google-auth==2.40.3
+beartype==0.14.1
+sentry-sdk==2.30.0
+distrax==0.1.5
+grpcio==1.73.0
+pyarrow==20.0.0
+tensorflow-probability==0.25.0
+imageio==2.37.0
+wandb==0.21.0
+propcache==0.3.2
+tensorstore==0.1.76
+numpy==2.3.1
+nvidia-cuda-nvrtc-cu12==12.6.77
+nvidia-nvjitlink-cu12==12.6.85
+setuptools==80.9.0
+tzdata==2025.2
+transformers==4.45.2
+pandas==2.3.0
+scikit-learn==1.7.0
+lm_eval==0.4.5
+botocore==1.37.3
+torch==2.7.1
+plotly==6.1.2
+ipython_pygments_lexers==1.1.1
+comm==0.2.3
+matplotlib-inline==0.1.7
+ptyprocess==0.7.0
+jupyter_core==5.8.1
+executing==2.2.0
+pexpect==4.9.0
+pure_eval==0.2.3
+asttokens==3.0.0
+stack-data==0.6.3
+traitlets==5.14.3
+jupyter_client==8.6.3
+parso==0.8.4
+ipykernel==6.30.1
+pyzmq==27.0.1
+tornado==6.5.1
+ipython==9.4.0
+prompt_toolkit==3.0.51
+debugpy==1.8.16
+jedi==0.19.2
+jaxtyping==0.3.2
+jax-cuda12-plugin==0.6.2
+jax-cuda12-pjrt==0.6.2
+jaxlib==0.6.2
+jax==0.6.2
+nvidia-cublas-cu12==12.6.4.1
+nvidia-cudnn-cu12==9.5.1.17
+treescope==0.1.9
+flax==0.11.0
+simplejson==3.20.1
+orbax-checkpoint==0.11.20
+pip==24.0
+aulis==0.1.0
+devinterp==1.3.2
+timaeus-aether==0.0.0
+timaeus-cli==0.1.0
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/wandb-metadata.json b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/wandb-metadata.json
new file mode 100644
index 0000000000000000000000000000000000000000..f254ffb59dc43481850192dcbda04f7c68170c52
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/wandb-metadata.json
@@ -0,0 +1,60 @@
+{
+  "os":  "Linux-5.4.0-216-generic-x86_64-with-glibc2.39",
+  "python":  "CPython 3.11.11",
+  "startedAt":  "2025-08-07T17:01:24.516023Z",
+  "args":  [
+    "train",
+    "corner",
+    "--wall-prob",
+    "1.0",
+    "--env-size",
+    "13",
+    "--log-gifs",
+    "--env-layout",
+    "tree",
+    "--wandb-project",
+    "jaxgmg-wall-prob-tree",
+    "--wandb-name",
+    "wp:1.0_alpha:1",
+    "--num-total-env-steps",
+    "200_000_000",
+    "--prob-shift",
+    "1",
+    "--keep-all-checkpoints",
+    "--num-cycles-per-checkpoint",
+    "64"
+  ],
+  "program":  "-m jaxgmg.__main__",
+  "git":  {
+    "remote":  "git@github.com:timaeus-research/timaeus.git",
+    "commit":  "7415666d6a2e5d63453ea9fadaf6cecf63256a59"
+  },
+  "email":  "david@timaeus.co",
+  "root":  "/workspace/timaeus/projects/rl",
+  "host":  "6e66e15ca303",
+  "executable":  "/workspace/timaeus/.venv/bin/python",
+  "cpu_count":  24,
+  "cpu_count_logical":  48,
+  "gpu":  "NVIDIA GeForce RTX 3090",
+  "gpu_count":  1,
+  "disk":  {
+    "/":  {
+      "total":  "102005473280",
+      "used":  "43331637248"
+    }
+  },
+  "memory":  {
+    "total":  "135066595328"
+  },
+  "gpu_nvidia":  [
+    {
+      "name":  "NVIDIA GeForce RTX 3090",
+      "memoryTotal":  "25769803776",
+      "cudaCores":  10496,
+      "architecture":  "Ampere",
+      "uuid":  "GPU-43bd7524-bf94-73a9-1460-2b36dd09a208"
+    }
+  ],
+  "cudaVersion":  "12.8",
+  "writerId":  "vx883mnozv26zdrpa9b34bes7opa5g1u"
+}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/wandb-summary.json b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/wandb-summary.json
new file mode 100644
index 0000000000000000000000000000000000000000..b76a57326e2304ed4c2b203cf9a2006d480b6c05
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/files/wandb-summary.json
@@ -0,0 +1 @@
+{"eval-batch-orig/avg_avg_return":0.901394248008728,"ppo/std/std_actor_approxkl3":0.00037745278677903116,"ppo/avg_actor_approxkl1":0.000705324811860919,"ppo/avg_loss":-0.0001646222808631137,"ued/layout/prop_walls_avg":0.42016804218292236,"ued/visit_patterns/avg_last_visit_time":3064.309326171875,"ued/distances/solvable_avg":1,"ppo/std/std_critic_loss":0.0005409594159573317,"train-generate/avg_avg_episode_length":18.8015079498291,"step/env-step-all-after":199262208,"_timestamp":1.7545897078288932e+09,"ppo/avg_entropy":0.0706079751253128,"train-all/avg_reward_per_step":0.128082275390625,"ppo/std/std_entropy":0.010983215644955635,"eval-batch-orig/avg_benchmark_return":0.9845115542411804,"eval-batch-tree/rollouts_gif":{"sha256":"3b8df0ac5041fd8b56302c1b9a7abfcd35821ce50a915300c332ab2ba4b9bac5","size":280607,"path":"media/videos/eval-batch-tree/rollouts_gif_5088_3b8df0ac5041fd8b5630.gif","width":225,"height":225,"_type":"video-file"},"eval-batch-shift/rollouts_gif":{"size":316761,"path":"media/videos/eval-batch-shift/rollouts_gif_5088_b1ee6f21586850948d88.gif","width":225,"height":225,"_type":"video-file","sha256":"b1ee6f21586850948d88f8c15763aee34d68d0f95f23f6b82c3e931ea31f4d60"},"train-replay/rollouts_gif":{"height":225,"_type":"video-file","sha256":"5ac4920811574afecb73e516147bf72e276d44b3e4185d71a184ec1d3f2efe27","size":155025,"path":"media/videos/train-replay/rollouts_gif_5088_5ac4920811574afecb73.gif","width":225},"eval-batch-train/proxy_corner/avg_reward_per_step":0.0050048828125,"ppo/std/std_grad_norm_pre_clip":0.3416053354740143,"ued/layout/prop_walls_wavg":0.42016810178756714,"eval-batch-shift/avg_benchmark_return":0.9875059127807617,"eval-batch-tree/proxy_corner/avg_reward_per_step":0.104217529296875,"ued/distances/solvable_wavg":0.9999998807907104,"eval-batch-orig/avg_avg_episode_length":22.534046173095703,"train-replay/avg_avg_return":0.4141111671924591,"train-all/avg_avg_return":0.8580207824707031,"step/env-step-generate-before":98500608,"eval-batch-train/avg_benchmark_return":0.9872170686721802,"ppo/avg_actor_loss":-0.0007236534147523344,"eval-batch-shift/proxy_corner/avg_reward_per_step":0.00312042236328125,"eval-batch-tree/avg_avg_return":0.9000949263572693,"train-generate/proxy_corner/avg_avg_return":0.06420870125293732,"eval-batch-train/proxy_corner/avg_avg_return":0.0452798530459404,"train-all/proxy_corner/avg_avg_return":0.06420870125293732,"eval-batch-train/rollouts_gif":{"height":225,"_type":"video-file","sha256":"e51653d7a53328cbe862e09665f91f1c2e59b9fa6a310eb33e6b9ef58d4b6cb9","size":322827,"path":"media/videos/eval-batch-train/rollouts_gif_5088_e51653d7a53328cbe862.gif","width":225},"ppo/max/max_critic_loss":0.0026545217260718346,"_wandb":{"runtime":3633},"ued/distances/levels_cheese_is_in_corner_avg":0.019287109375,"train-generate/avg_reward_per_step":0.128082275390625,"ued/distances/solvable_num":4096,"ued/layout/levels16_img":{"path":"media/images/ued/layout/levels16_img_6048_b738cd7518b701c2b1d6.png","format":"png","width":55,"height":55,"_type":"image-file","sha256":"b738cd7518b701c2b1d6c53e8f0e1e69348964932600822f4dece8791aab199c","size":979},"eval-batch-shift/avg_reward_per_step":0.15191650390625,"perf/eval-batch-shift-duration":0.47184806130826473,"ued/visit_patterns/avg_first_visit_time":2668.73876953125,"eval-batch-tree/proxy_corner/avg_avg_return":0.9000949263572693,"train-replay/shift_proportion":0.984375,"ued/distances/cheese-corne_dist_wavg":14.516209602355957,"ppo/std/std_actor_clipfrac":0.007598568219691515,"ppo/avg_grad_norm_pre_clip":0.9306865930557251,"ppo/std/std_actor_approxkl1":0.0007034532609395683,"perf/env_steps_per_second":261686.6138904752,"eval-batch-train/avg_reward_per_step":0.13169097900390625,"_step":6080,"ued/layout/num_walls_avg":50,"train-replay/proxy_corner/avg_reward_per_step":0.002655029296875,"eval-batch-train/avg_avg_return":0.9328052997589111,"eval-batch-orig/avg_benchmark_regret":0.08311726897954941,"step/env-step-replay-after":100728832,"train-all/rollouts_gif":{"width":225,"height":225,"_type":"video-file","sha256":"5ac4920811574afecb73e516147bf72e276d44b3e4185d71a184ec1d3f2efe27","size":155025,"path":"media/videos/train-all/rollouts_gif_5088_5ac4920811574afecb73.gif"},"eval-batch-orig/avg_reward_per_step":0.0879974365234375,"ppo/max/max_actor_approxkl1":0.002348087728023529,"ppo/max/max_entropy":0.08841829746961594,"train-generate/rollouts_gif":{"height":225,"_type":"video-file","sha256":"3da39e8f9a0fc99e101f630355e33c67a3d950d5608b6ae9e517d8b152522da8","size":314790,"path":"media/videos/train-generate/rollouts_gif_4064_3da39e8f9a0fc99e101f.gif","width":225},"ppo/std/std_advantage":0.05034048482775688,"train-all/avg_avg_episode_length":18.8015079498291,"ppo/max/max_advantage":0.14421862363815308,"eval-batch-orig/rollouts_gif":{"sha256":"39b0cce15a654761e9ebd4c5e8214fdda66e42089c9870a18cfd461d697cdda3","size":279328,"path":"media/videos/eval-batch-orig/rollouts_gif_5088_39b0cce15a654761e9eb.gif","width":225,"height":225,"_type":"video-file"},"ued/layout/num_walls_wavg":50,"ppo/std/std_loss":0.0011741130147129297,"ppo/std/std_actor_loss":0.0012073023244738579,"ued/visit_patterns/num_replay_batches":3074,"step/ppo-update-before":121600,"train-replay/proxy_corner/avg_avg_return":0.13981924951076508,"ppo/max/max_critic_clipfrac":0,"eval-batch-train/avg_benchmark_regret":0.05441177263855934,"train-all/shift_proportion":0.98828125,"eval-batch-shift/avg_benchmark_regret":0.06363297253847122,"eval-batch-tree/avg_avg_episode_length":21.89337921142578,"ued/distances/mouse-cheese_dist_finite_wavg":10.653342247009277,"ppo/avg_actor_clipfrac":0.02136840857565403,"train-generate/shift_proportion":0.98828125,"eval-batch-tree/avg_benchmark_return":0.9846352338790894,"ppo/max/max_actor_clipfrac":0.0325927734375,"_runtime":3633.074422945,"perf/eval-batch-orig-duration":0.494891369715333,"ued/distances/mouse-cheese_dist_solvable_wavg":10.653343200683594,"eval-batch-train/avg_avg_episode_length":17.10165023803711,"ppo/max/max_loss":0.00237191841006279,"step/ppo-update-after":121620,"ppo/avg_advantage":0.0054062530398368835,"eval-batch-orig/proxy_corner/avg_reward_per_step":0.0879974365234375,"ued/distances/cheese-corne_dist_avg":19.151123046875,"ppo/avg_critic_clipfrac":0,"eval-batch-shift/proxy_corner/avg_avg_return":0.053277790546417236,"ppo/avg_actor_approxkl3":0.0008164727478288114,"ppo/avg_critic_loss":0.0012592781567946076,"eval-batch-tree/avg_reward_per_step":0.104217529296875,"train-replay/avg_avg_episode_length":67.34403228759766,"step/env-step-generate-after":98533376,"ppo/max/max_actor_approxkl3":0.0014691848773509264,"ued/distances/mouse-cheese_dist_solvable_avg":25.545166015625,"train-generate/proxy_corner/avg_reward_per_step":0.004302978515625,"step/env-step-all-before":199229440,"train-replay/avg_reward_per_step":0.011199951171875,"ppo/max/max_actor_loss":0.0019360017031431198,"ued/distances/mouse-cheese_dist_finite_avg":25.545166015625,"ppo/max/max_grad_norm_pre_clip":1.9458073377609253,"ued/eta":0.5,"eval-batch-shift/avg_avg_return":0.9238729476928711,"perf/ppo_updates_per_second":41.606837171188516,"perf/eval-batch-train-duration":0.5347478911280632,"train-generate/avg_avg_return":0.8580207824707031,"step/env-step-used-after":199262208,"ued/scoring/avg_scores":0.5722335577011108,"step/env-step-replay-before":100728832,"eval-batch-orig/proxy_corner/avg_avg_return":0.901394248008728,"eval-batch-shift/avg_avg_episode_length":17.76091766357422,"step/env-step-used-before":199229440,"ppo/std/std_critic_clipfrac":0,"eval-batch-tree/avg_benchmark_regret":0.08454027771949768,"train-all/proxy_corner/avg_reward_per_step":0.004302978515625,"perf/eval-batch-tree-duration":0.4597616083920002}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/logs/debug-core.log b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/logs/debug-core.log
new file mode 100644
index 0000000000000000000000000000000000000000..4479b2763f6e6117658313c494f093a0f65877c3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/logs/debug-core.log
@@ -0,0 +1,16 @@
+{"time":"2025-08-07T17:01:24.569698379Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmp6ai5pgx2/port-3847789.txt","pid":3847789,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
+{"time":"2025-08-07T17:01:24.570617143Z","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-3847789-3848111-299703965/socket","Net":"unix"}}
+{"time":"2025-08-07T17:01:24.570725532Z","level":"INFO","msg":"server: will exit if parent process dies","ppid":3847789}
+{"time":"2025-08-07T17:01:24.734738744Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
+{"time":"2025-08-07T17:01:24.748337725Z","level":"INFO","msg":"handleInformInit: received","streamId":"wp_1.0_alpha_1","id":"1(@)"}
+{"time":"2025-08-07T17:01:25.108760353Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"wp_1.0_alpha_1","id":"1(@)"}
+{"time":"2025-08-07T18:02:00.220454769Z","level":"INFO","msg":"handleInformFinish: finish message received","streamId":"wp_1.0_alpha_1","id":"1(@)"}
+{"time":"2025-08-07T18:02:00.221821229Z","level":"INFO","msg":"handleInformFinish: stream closed","streamId":"wp_1.0_alpha_1","id":"1(@)"}
+{"time":"2025-08-07T18:02:01.18783896Z","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"1(@)"}
+{"time":"2025-08-07T18:02:01.187886481Z","level":"INFO","msg":"handleInformTeardown: server shutdown complete","id":"1(@)"}
+{"time":"2025-08-07T18:02:01.187904278Z","level":"INFO","msg":"server is shutting down"}
+{"time":"2025-08-07T18:02:01.187906663Z","level":"INFO","msg":"connection: closing","id":"1(@)"}
+{"time":"2025-08-07T18:02:01.188018588Z","level":"INFO","msg":"connection: closed successfully","id":"1(@)"}
+{"time":"2025-08-07T18:02:01.18808005Z","level":"INFO","msg":"connection: ManageConnectionData: connection closed","id":"1(@)"}
+{"time":"2025-08-07T18:02:01.188007979Z","level":"INFO","msg":"server: listener closed","addr":{"Name":"/tmp/wandb-3847789-3848111-299703965/socket","Net":"unix"}}
+{"time":"2025-08-07T18:02:01.188119497Z","level":"INFO","msg":"server is closed"}
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/logs/debug-internal.log b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/logs/debug-internal.log
new file mode 100644
index 0000000000000000000000000000000000000000..8c8a5a38c25b31beebee6909ab1011e9f72d85d1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/logs/debug-internal.log
@@ -0,0 +1,1168 @@
+{"time":"2025-08-07T17:01:24.748591655Z","level":"INFO","msg":"stream: starting","core version":"0.21.0"}
+{"time":"2025-08-07T17:01:25.108666268Z","level":"INFO","msg":"stream: created new stream","id":"wp_1.0_alpha_1"}
+{"time":"2025-08-07T17:01:25.108745592Z","level":"INFO","msg":"stream: started","id":"wp_1.0_alpha_1"}
+{"time":"2025-08-07T17:01:25.108783284Z","level":"INFO","msg":"writer: Do: started","stream_id":"wp_1.0_alpha_1"}
+{"time":"2025-08-07T17:01:25.108781317Z","level":"INFO","msg":"sender: started","stream_id":"wp_1.0_alpha_1"}
+{"time":"2025-08-07T17:01:25.108844897Z","level":"INFO","msg":"handler: started","stream_id":"wp_1.0_alpha_1"}
+{"time":"2025-08-07T17:38:54.19188092Z","level":"INFO","msg":"api: retrying error","error":"Post \"https://api.wandb.ai/graphql\": context deadline exceeded"}
+{"time":"2025-08-07T18:01:58.83702237Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776, Name: checkpoints/3776, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3776?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9a2ab9acc7dabebc56d0a6c435b62fc90c306a093f4f4dfbf20f4c3b0bca3d753081b4a727922ef1e5971b0984c9da8e6b23b87b395d17740e2fedd5d846fb646f51fbe3a2cce3df58097693e2dcf86af076fc5812dfdef512585c81acf149e2bb57ca6df66aa35993e85ca25662f62a123e8d30a10876a5f6ad81533298f5bf9401a7275940a88322d0f1522a5b2d15e187cd6ba8a935fe34a01d3c47abcf3004da6c68993f4a3dec81a36f60c4dc059f6e5b7bf4169d19c4ccaa7dbf41ed57371e21f85ea7f72be4c181d2ecf973bc82bab77e00cfbd18e979d6f39c141488649da05c1a68b6ab50d7e197bb402de6e38581d9510645db7194469f1cd01a4a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.837116579Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default, Name: checkpoints/3200/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3200/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=71728f540ca310a94c27ad549218d2cc725f4a314b7768d6d288fc807f18d8990fb9c88811a26be8132929fee57c2132421d5220f0abc7ca92e909ceb06572ee452f195cbe8171a529ddf9f53149bfcbadf3a31cd7a4d4de2e18064c628bd519b270cf2895bc838f0a9822437cd784a3c9d2ef76663904358f939ef18b99170bee98c2222d7240d8a91d77b64fa245d692aed349efc0979dc053641625bbb223f7cb55a8e5a89df03cb04c462362ddaa30c6003819797a92d83b572e20339b1a21efef67ea6981b41a4cbd905d877d5cb55507e2186fc10c19db41fb2ccb75cadbd18d6b89db22f2c74dd31ec718b6dc2b241e62d947e5e00113ab968e219ae0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.837137033Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d, Name: checkpoints/384/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/384/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a1151985aa33822cd8450e5b2d4001bd81b5a1eb12cace08527c76a3182e18ad4ceff06a849119ebbf4f0933786b657e190e73ca23069637943a5944fb762dd68505faa6157cd4fe6827faac2f7366ec646a0d5af244b2229bbac9fd9c4723c05b4d9e43d825c75cb558886d1820c54281a33f14a9fea772807078e3982afed4823c1d10faf558b51a4e4ddacf489bdf0746bf1c9e31afc387ebe47cb13f4987edf59be6f8842a6da36d05b48527081ecd460c2da466980f7816d9781743fb8860d5bb44dd2f9efc92cab0d6e351def5170f48f06dfb39068c7f7ebea00c63ce6326e21647754a76c922acc6d53c145277ae2d5a354fc793bc6068ba522e9c85&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.837137234Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/d, Name: checkpoints/2432/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2432/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=7a945df1dfd00362633661a7d9f0cb758e39cae465b46bbf159bc98b8705e15adad33b3e82e18897e3de6025153e4249664f72ec61d47b2808a07b0b176fd599c6a5605bfaacc60f15cfdfc354f4cf4fbf55533e0b64850b25a654769a69c97d1f8353fc99ea881e2f499f9e7208dc6987fc74822241c7a377b8d4c8d92537dd1e708087637c107acb912cafd65083aa088cd2fded1d0c47b69392ae20278b6be95566bbfa076e03920aa27e6d1c8fd91d53629c07f4ec17a785e7ddb4fee77114dcb950e0321bce862d287cc1b9359c79e896171578c70cd114192632b96cd0dd58a6cff6fc1a4d897b8810d5453096128736723bb5c86914ca447ba80ac1e9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.837206444Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d, Name: checkpoints/4352/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4352/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9b608a8e0ea70ec23836a44837daab31dc4bc5cebe8d021ced2372b0197c9f588e271f1ffd0ee7d5eea2e1557482c9725b0f751be1c68dc3b96ddc4ba302e6b4533ecb751eb57e673621fdbe69a9860401989f0f735657d80b6860f4cda80341ec2b6a6a170365f84177404d8f738ed60eb11c5e10428f658c735c7784aceb65ef07a4798199c6386dda8756b95276c7b05778ed5e00083e3afd8ccc1167319c3da6c59e61420f1325c783ae29ad991f004148fb4868a4036f4efa6cc1fbd6c58a8d07442a13b0f2e55e71b834bfa7bb104537c021e38230fa3eb4d8762fa27dbf8876cea66513357de99e5c4eb225b26d1626405b26bc701fa429d7ef44b6b6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.837235553Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d, Name: checkpoints/6080/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/6080/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b85bc858a86e80f5cc3a24adaffe17b3e2c82153a78210ecdc007c910e756803dfbaffc3055c2cb3dd4f4d7eb1eaa6c0de19a975e2885e748c44eca739cdc505d46936304400b2df1917efd576bc7905d96ebe6a22df7db527641ebec6ff44679fb887b5897670bef9ef7eb4ebeb22f15b6f1c9198e6e84fde77e1c1e5a3ba8f9501ca0a7d713faddc0ba1a30eef6511f031291a61601f5ab90b1057e82d1c2d7cdc28838ca2c233257381e252cf1fd2ce699015da6da415ffbba4ebf8378aae459ce5b2e2d5cdee2f1e5301ddd876f048d5e52c3472bcca9586f0b0ef4c597c2a36e83cfa8bd29d3def529b23e188337a6fb965f71b3354976029fb91efd0cd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.837317436Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840, Name: checkpoints/3840, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3840?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=53584fca77dda53497ba1a3cdea41742b3720a15233f15d5836db84108d8d0d8d9d8f1f8265aa8e162ae9f948e8e8074ffef7d91d6378cdb87d0afe687a50b38bfae7c84f56c576d73c532cb35222ac99629b49917533ef33ed2ce57df7aa4a14bd550b33d3ceaad23a1224269b2e480b78405fdad438809138f83fcdb967a4239342c537ef01800d406bb752a2151375fb34098501810bd1b638dee8a488ad96e1628bfd4f2b43eb34876a9a8dce60f614c9b115a3fcf7fb6394d942fed57544d36643460ae6bfabca9c640b724f8b54652eaa63d362cbb4cc76bc56923f1139b3a7c324881b322303fa33152d47df4708e9f3ddbde01639610800da7ae36f8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.837333975Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d, Name: checkpoints/2688/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2688/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=413e4fe66256868bf13faef8fac2309868d0b43d472ef5dd71f2ad3d851693171e743541e743e0bbaf080377e1ce2786aaf9baa6da3a7cbf5e741c22100c1844052f2d8ed0b3c89fe8d0130952d464183eceeaf121bba8b934fb9f8204ae52e84c77957f21fec615382416b21c6ae0c1715ef2dafa894e0d368f8e1b4ced2a65755759e94cef24b110489af4a59357e3ec4063ac43b35daee0844d5e69c4594620fa2a649fe90d69bc60d39d237b4b83ac560de02ba1be5d1426a3a5136961cd9b611f1dd3d8813015524510fdec30dc92597bfbf61cd998edfa89a48352bfbf772bf7cdbf1255e6264a53fd2a28c12e17f850d86ca511cde52492250c6d2b7b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.837402884Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default, Name: checkpoints/4800/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4800/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=959a46c23217edb397290d4e6b7724b93a935b33016d3cbed4b721e527f73d01f36eeaed60a91053ed81d247c987b8b3aa5540d01b9482be7aaf10fa23907cd2975137134ccc69605463016b9befa770c291924946b6a51adda846710ef8bfaa166ad6ab73c8a04b93d504b8e4509bec9edaa81140c2fcf3fb9011f04d9db1dfe3657f5ac80aa3aacabda55c8e6c4f809117923b82112e87075a5849ce18c6cfbf2c2ce097c7f0bd65a137e8038fd99f62c4430141c26aeb183ed6f7bec9d0067e95c610b83e378c572b9926911889e7dd708b8b803aaf9aa448ccf1384fa7780b8b8afb95971b64445032e206dc5e7bf0684b8735ab7d79c79a6ca2b7811865&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.83703934Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/d, Name: checkpoints/5952/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5952/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=5d7e568311afc63539d4da9c5417225c8adc14adeb7d7ec80e6c7ba4dad2dc5aec380b5e7593ea85e2a80328e819a8206d0dcaac828db2ebcf7a5d7df13197d2bbfdb6c94d379b750291b9de6f8d127824ee8206206cc8c8ffbc2fe28925c6d90e42844d5c9bab00635e1c26a91cfed0bf09180535b81a059398e0a76a205b2f5f6c895ec48885bf922e61ac48c2ea160fdb1a420d789ac031bd4bb74c83e03570272c06b6ffa4c7f335442d083df781f17ea7394819dd62ffa76e79cb99c5fc1f39bc0b2a7c79c4463db92e5215c217d0de9a98c7fd9a0675beefbf28ca74b6c86e231712743ea3a8fc98534a4ba48b88d39fd655f79b7b62d56ee1631d68e4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.837500905Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/array_metadatas, Name: checkpoints/640/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/640/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0c4338b3740fbeca3ca2f726a92e1932faab3642c535b014be844c62c5fedc1d324805d4a7aefec60045b3363b7975160e4367d8359a34df8599e03130b0ba597969f1f1f56bd198d343a38da9fb85060d1bba6b27b646cdcaacaf999add921bf4ca33168234d0853d6edbba626198e0499a3daf7be004d8cfc3ae076210937a5d65644fa35dd1de022d50032f677cbf6e3fa4b13ba96f5e68cfcf6e264c12a9dbaf332f055e0133f7cc0b6a0f96ea91a37f52d6ceecdb12b5292272d904413261e84a10db8a82ba844ff2f4d769b82769f38927d619ae6e71bcd7438a266e5dab6bf692ee70f729c5f3229e216142a2ad10b53e561f8f1237f75dfa23e1e6eb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.837569265Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default, Name: checkpoints/2944/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2944/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=08601a5c554b5f5bf02cf3614a4875361e535e707bcdf0bd926c96e318a1285a79abd683b94d18d72896f60dc710c585ca0ba03374c2e00a3dc5b486ee0b8a6d5338a293cf64fca5379b07ed1b16ce55ce46fb351c8421fe4280965939762f91d37c3516ccbb62c603fc8c8a0d7ea6320dd514d1a4bf7207e3cd8314db480d6b0849785667550ebac2625a6b650bffad4eb0b9d1eeee203e6a7239b81134937354049f82c8871406f59010abf84863e8121b79a5e20ffac27a3ac0b9f130da41041102828ecda2c094ee79955d1710c22918838050d2c48f2f0b22dcf35e21e20fecf59f67fcaa4696bf93fa04510202360ded80e7160bc088e8591e577f2cbc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.837572304Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/d, Name: checkpoints/5312/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5312/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=287e028dc9e987da983c671d1835a5bf4277095a87da85b5419b3ce65281b611f6f6c4e3f4281c74859321d21da637e760305e270c018345555fe0b6296ea4cbeeb8ff8e259793dce7e1804ef8f48db36c11f8d0ce1cb129ca3bf7ce28dcdf2b3bcf395e07e9df9b9e52bbe78a4df2f143ae4e89114d18e2125daa2071ad7c1e5f96a7437067c7c3d966ecbc19ac0581bee9435739bb4469c1c8232309e9a9c806b1447f3dd667f5851bce156e1ff2b94e514b479dd35dc8d971dcfcd56fbba30f1e9b04ac9cd11440c5674de6a7407bb697a61a196b95238f5d13907c928c5e7f427fb525b2b79d463fba82274cb6d6ca02f67a9bb7e686e1ce7dd14a79ebf0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.837566493Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/d, Name: checkpoints/5568/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5568/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=bfba5cdfe8c371dd74b7a785c4d256c32823f97c44eed35d82e463c26fa8a55519e73d9c9370ca20eae896880783b37149ad3440fce33280d2469277bf6b4fa5f798297e50f110c17d31055ddd6409fea171fcb06b5d8237f350cb558c7ac77de5b681e774387337c7b6f2ff0d0767205786f21dbb5a43c427bc6271797ca6ec5e84ced9199d87528ca393169b8c69df4a99a5de408e77e59fa0447798f757db97250291194366d0e4e3a4fcd56d3adcef906e57da9cc66f17a2e3c890fefdc242e857eed674992337063850017dbad3063e1025f754e869425025a470ebc43c73b988de3e43060c4529ac60fb446000631c08a03cf1273c2e50128fdd7ff846&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.837360648Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d, Name: checkpoints/1920/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1920/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=1ef3bffc314e22d1ae2f9a63357112410c83628f52e2616ea2b91e28ccaee41092fd3d194088d21de4ac07d93ac61e8367996bb5000286a3de52fae5de22b166f02cca67385587cc6542273ddf7f37b8b11378677cfa2821aa8cc4b73794a0f80430b8c949d8f0cf57b445f165d5daf44efa2525eedee8b7f36082c11921da601e803f76a7fc1cfb8872ce8c98c41df9b6b85caef2c23ad1dd30413cce35029dd6461f408d31a23e80c80ac861a692a146782c4b262c1bb59458bc8423cb31f76b633f9dd40214d698aa86bd6c242b4ba131c3af579ef152a3d9b04271eaf15c65007761bc6e3f9c3bc9f55bf26eb9ba073a5c2563ab86361658dcc03a4937c2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.838585295Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/array_metadatas, Name: checkpoints/2752/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2752/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4e799efb190e9b8c711f458ced91fd98f95d3a1bd1859bf66f0e383e6c2937fafb250ee78bb5fef9c97becfd5b2cd460b3a2d9251656fb4dc6adbab39120910314bec21dfeda8f02d4592b8f028c5665046de57f76ad480942a785896f419d00061431d33cd1cee292e063747a7281da1a4defd5ccaec3951d9e52034c40606ea5e25d853d9efd45d69a8827cfc3fcbdf92f983e5208fcb3b4b682e2a1d0d5cd538a61eb2936ae104c3a924bbe885a1bc4d940548bd18b6087fe3ad93f9e0162385d0aecfd060416fb3e680de2d0a71d3c812cdd8987ebb3abc7c397892cc287605c87ff31d7831755db9714f125d6caab4ce0626ef0db46eee3c83f1666e8a9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.838476842Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696, Name: checkpoints/5696, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5696?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=19539d8a80c3177c6bf9a4cbc16b94e17875674323df313dc691a457e777d41d94ccc7f3d5628facd8eee7d724028df8b9698d71dc18aad5e2be8794b1610227e2b25a214f8e34d4b2d6bfe8f2a139b3ffc8fe59b660dd1ba373118da0336d87d19cd5faf50aaaf64fae0a328938705821c9212fd0070b55f327ad64a9ba0194602cd3eab754d31050c4cd05cfc80e478bc5f2b772beb8114688f61b860f0c81e3f948e16f7369d310888ba887205c93ce473f7a8e07a28d3b8531c56397f044f52027690162b9f688cfd7dff77ff8644ce8f2b5a4cabb400d4411ab2c53570433380b9aadef78604cc126bfbd70ceafd6b3b38aabe0299d3156416ce875fddd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.838632182Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d, Name: checkpoints/4032/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4032/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=5accd9d7aff004984b8c18e9866a4e7b76988718a67b0b0f8a22d573164e83474028e594c3095d2d6f6a531a9eebcf4b1ddad462a9e846a90279383104a1fd8adae26ab4de5938611cedf18af386bb7d7c8e0cf12eb0f30070967586ccddb44f276a5b284f724d655c1e9fe08fff8d3d4adf6a134e1a050e92f3f3791c85914f8c6e886c5c8591dd8adbdaf5289f487067efcc3f2791652880b3812187813190c757d626c1b1a18630bbe63f5ed7774855a2c9caf5e427f9854ed12cf84a1e0a88c7d2689a1f6d88577b2c1e1090019c8f166a9ff70c4b04760a5c6a3f31b1d7048e5cbd6064fd77650e9a2beb3f225529c328a8d3886c220e2a172599510e62&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.838737287Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default, Name: checkpoints/896/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/896/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=61f399ca9e39ff83615d14a7fe6f6aaf9453950dacceb05aaee313890ae2458b27ded24c5f60a2099f0801678da74f89fafdc0d5b8a65671ab664687d0c9a0850717d002b24567ca0002a350c5cc2194ab1d2e542d4e58a14ef897ec0258110333c3dc515224ebdac58bda88f2b0cb5cdc9ab56e54dde732e99ddadfe73d313518448a01b9e8c0bfd889a128496f909ad799e65b06f6728134561fbcbc757245aaa61364a51406f4f77e5bc618c836bfca52e9a2d5b3ce90792ccc47e61ccda57609cb7a0797fd6ac0d1459625c7f417105c76912c839e8849a7f3445a20205e21d8517c6452433a605be41d8e0afff9d2c8ba4b290b13e7ce66d2ea1e7ab682&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.838879402Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default, Name: checkpoints/1600/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1600/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=71eabadb617a37a641fbc39839d14f4cf93c17494c9804cdadcff7e8b2d62737f2a5270f511cc69ecfa87494486cbb9cb907a1e9365bfc1a255865c6d7543773e0f330f21ad7788d02cb2ed6459cd366fb59db7450c770b8072292c0cb857486a8cad2b6da8f6a155adb7ddb29e7fc0300365d193dc5db3b12e7c8be000d39b4f31f76bcea9b8ca09298cc28528515a2acbbeb28ab4bb42b2395e057bbbc25c8478041044fa6b06e41be3a149d9455c29b75bb1e95f11f60683e26f2f9edc483dc03ad1c99b45decc17c4af5d474407c62cd676c673dba93518c35b74f9cca4be462e497cfd69113e1db971672cfa9dff304f45f22eced74e67dff0adda51ba5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.838684021Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d, Name: checkpoints/2816/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2816/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=43ea3e2fd7e77402e47db1c3e252d433631641e8c390545fd2548db2dea687f7343e56e4e0a600cdc91f3471ef16b9474ea5bef5d1be44273f6f1d635d901ebe6a4f26b80e560b2db83de4ff42202546033fcac7ee998f31932eeee584f5ec336652831c63a6c93136a4333e114f749efeef203fb00259d7c0704997020afc0fea9d740ef39224ad6cad4f3dfb363350fb60d916c4a2f93efa5f3c1e5521e2b47c417597f9a8772344b649da6f5907dff54b5b3cb49397803f900f1cbb21f15c85ed1f52c8ca341deac9e42ea029ee882703388b48267ad8048c6fcfd50d0c9b78e7218edc486acf63529a6ab098f3bdfb3647e6826b0a0cb503c2fa792c61c4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.839155591Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d, Name: checkpoints/2112/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2112/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=ba4d884a0d56044736bf02b9a65b9d5fa46cf425c576d59b14e9f39a02dbf62e3521b8e842220807756fcadca2663e38ee3141b1389083ffe49fc658d690af5703a83563bbe1f5e251c410d7650d544d6d7329802b2ed4968484ae693e6cd3c606e25a03ea7c400b765193b29a65781d5afe0ba7a9d8c30e3906a263c3361d08fcb6a83f4441f33fb0cc7d1322e0a2c75b9fa773c2d0a9a2db66207f7cce76ca05b771cd732e0a7a6a4da796101d12a7075cf6173b28dc4293b13fb6a72a9fd8e18493e41d70b5ae80f067794bd0d04ac2b610c031e2274fd3e58ef5544bd584b944ad34d2f288a82130e7a9b8e5f4a4a8403cb53d7982dcf2b345753da39504&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.839176094Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/d, Name: checkpoints/4736/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4736/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=25de698790961ac6d460255dbfeaaf62b7a4f9b7da63c6b4a4f121d382946a1d7bd2c6a1689401e785ab8982df7d803bfa5e437a327216ee3c6aa91cd52a277b416c31a9309bfcad7d3684de2a9df44f61211ca35c90b72cf64b92939fe663598c8b5e27b217ed240ff1e492acdd82a94aaabe5b9b8e090d9aa1285e4f9cc504c0fe750e20c5704d2b41d657d5c85072d834aea54818fd0c1a83a8a241a438a7e286108f1e08684638552377e409a48a9acda0b739453d21f8933a460bae465b486aceaeaa3ee27229e5e8459e779a91842344e0fa7da63cdae84f4ddd1dba1681f693294a392b91db49563fdf356f41d203df90cd21ec6e5faf6e934dcaabb2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.839216096Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default, Name: checkpoints/2048/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2048/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=82c4f0b8379658c9b6435d7bfd68483e443f751b42774b18b55b7902eb90852f9cb0e930f3a362d0d3afe41a4b494fbd734f953a074df8a578fb385691e25a96f4d282fc9dc0981afefb21257b5ed7277ab1661187040044fffacd98bb298e5211a2085b8f4cee1421eb1ab12e1e45fae603bf0dc023f9045dd6fcd997e7440da82ad60a9763d12a88010c82738203ee386200562a67b078602e0053492b9fbfbd1163ae3f48548226361cd61990cafb2defec50964a971082c62da6395336e79aac30aad8d327db315ae018ded4e4ed4618fe18d513bb4c9c3d2f73181f4c0f3eeb82bc348013f624914787c89f16ec67d11e30c3408a035c6a4216d8d52850&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.839210233Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/d, Name: checkpoints/0/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/0/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4f8ac5956ed8f62494fc53a4950a5a30f0bc94bcf2822b7090eba204ba8cd4d6d310b7c1020d244a87d530049768948621b11471b5c52613360064a087d696cadc61ad965206085ad5f29c95f21e2f9131fb299e3c846139b85abeac2ecdc37b668bd5e050a204f7358f0ec99c4761748fd635c771e132b01ab522726cc42a937d159ca1de25ed8bdb9c1ef9336ee1cc05325a0bb1e66ef3209a1e933b18f5a88e9e3994d86d26404302e99feef6a2957d3f9178effc476643133899fe874165c535773b543357cfddd9a0bfa5f4d140754ccd3dfcb65effc877c10d1bef6b04097856baaee206894597fa8293151aee69011d2877bdd8c7c7744a4e88885ff8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.837498089Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0, Name: checkpoints/5248/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5248/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=109184d6dc85efbc6cec75164a4ce15c0b00f010611f5cee0b508f4d554601a6aa5b43dbf624134d4c8eded0b701e2a566db79f8103fbb7d0e2a2e81be4fc9c617975dfd1e4026faf4ed78ef656990399f89c978637e67980eeb6175f6e2d178b14b9f3cc62eab9ed947b3ec397a298c9fc055cdc31440e75804ffb6ccc7514f5ebea2e07750b00b0ba56a3ff7b2672dccf28702d9801166335a1cac3b1e78c1bf6a20d2041fb146105816e5d38d5a276f35d9b11fe41d965d6ee58d016e7183335f8d7746642a5d51f7ce1f89d852a19fb29ad4094d8296eeef3c8b9ebc56787d60ac1e487f1a8aec547eeac50d639174b44dd262aab598af647d8f62b71bb7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.839318856Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728, Name: checkpoints/1728, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1728?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=6251d05ad77d2684b26376c1e14127c6326282c1162a2cccc8801fc339082286356dd334fbb33cd7d9c535a07e167f320c836af999d747f2592e9879de078686ad94aecc3f76a6e261a269d5d34d5de614d7c7f25acd528aa163e02c26c1cc94b87696d8b962f3b9d5ec66a058f997d5b85894d178ea6870b661a956acb510203dd9b51a674f738dea7a55e0c9b3702f9b9a14e6747a20ebaf4290c78e9c7f7c95e0e7614211745492c7410091e78d6d342153a85faceac4b5601285bca5be646d8d70adae7c2723b6dfbeb92827b3e90c18f5bf8b77776858f3ea5107ff36d7d85932b29e5e98667425f84d5e9fdf77c19e0d345ba9931d2c4fdf1bec6b194e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.839181328Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/d, Name: checkpoints/2048/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2048/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9dd0a8fb56803ab2395ef283dee113d2bbd5e61076f9538cbc0b32da386a12d1959a93bef15eec5d266b9216b7d3ef34382ba64b6cec919b4990e0155c20357755112472c0627245270c52826845e710f30dd03b7eb731112211b4d05178cade68dcda123107d71d83ec4e747425373ecb3b1d270455fb80b5e7fcf7071058cbd29ca12d086e4dec0ad354b6914c764b153d209e4114e0ce4a1f7fb68dc0778a8718718fd623f7e32df4563cc61ab77a09745507cda807456b79f3029788534924683d201705983a08e124641dfbb7c031f33b12dc326d59b5a6c1ecc2097a8bc476e2fc6162440866cb9781262632fc374d6ae676c91a7f24a3f569eb2f63b8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.839052628Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/d, Name: checkpoints/2304/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2304/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=89d1e5ffcb18f7c790adbe27979df051079386e9ff4c32d2c0f633b51bbe16e946379c0351af66c94718fa6601531a36ae3fbdf64260a394155f27be356fba0fa570a857cb92226f25cea94377e703c03e23d45f4ea7e87c1a093b5dc81d4ddafecf6392f5a5af3cea8409d146aa538ff07bdfc794c51528619b7cbc937fcc0e8a840c4fc5fe8c882d9defde88bea0cdeb249ec29b855ed2f5f09beee39b5e9cdb22ce8eafe0d460bab24b9c9bc9ca5cc776efd0b6ac6c7f2835d7e7021529becdcd770bc38825c97bec52d7d56e008ccf7b86360ae2666cb737a9a385dfeb3968721f32f1c798d0099be8d67bc021c4e448284a924699ed98c19121cc6fc959&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.83880999Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752, Name: checkpoints/2752, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2752?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=7b7030359438f5d7063ec309dc039de76d3895e97814d1e99b40c400c000d010a311af4073f3822a8c663ce46744d4aaddf71b752d28d0aecc660331188c7bffa4192348711c02822f32b60751339b45192b88070709d01909a44cb05da37a68d9a4c0ffcbb45225c7a442b00d82b02745577b4a39582bf6c70449e8691fc8b80ab32a26838edc1b38b716d5d2978ba8aeee7ea1e90675a819f555d847d0c974452e11f37b9e9a01a4b6e3a112be940c878374588dc1a0ea1aeb65f771e6ef1d6d034af78a959117ca777e2a5a1a58cb4cba118e641d251bcd2e24b84c8130e8ca228685162668ef76bfabc242fa9776573158b34b9ac30040ce21f7da974090&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.838386149Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0, Name: checkpoints/5888/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5888/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2d552847798d9e55335a2dda3e25715e723ff3eee12e45f499bc060fbd3d58f0144661c38bdb75cf40a2f2d73074fc6546ca5e35786ae2d6c8dac962566a966a49f03d514b988826afd2177e2ca12908d5ac3f8979a2d9ff6031819dab8d92463e49400642a56590d89c5b6e4e6623088a13416277ecabdfbb9aa7630ce7d6f4a18abfa0567e60e421b1f7f8106125b333db1fe6b4a4a889968a97c3924f90f7d872c52360e43c30b29bba5ecbaaa84537da0632ad578130bb389bb65040971497a5993b31ca76db86c244a969403e55e81448a146d651b8214b3366327d334c955fd2d2cdfaad89088ccf5b4fecbf0c1bdbb9414e37d29d2b841b9178cd0852&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.839915457Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default, Name: checkpoints/1536/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1536/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=8bec3a28d5237441154eef255e01bc862f23fa8932be13a89b9341ae9d0a1d7e2c59d2fd0948950ad00ba21ee525bf365c95a68fd6fe200d9f964b7a17272c2fc9460740ba214ff8d868f907e86d2f383ff9fc6381f3bc2237555f36bdd0294dbf4e7a9183c8e7f86ce3af9344a95942a28c2b52b16c45081a91a6a495d0aa4f2d3b486fb7d626e2bc9ffcf1ad9b748ac8971854cf3825b33d1583f0268c5216ed82547ea76f178a7264e356e4e20fafd42aba489297f52827c244355ea459b041951b03689e85a3cf33e0e88f43e6d7c73a2265bc56fc29b4609c9c2a11dcb7d440c65bf3fab015f56ac18279bcc84dd85ef53afe84ed3c547553268d0807e7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.837464784Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default, Name: checkpoints/5632/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5632/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=8d88d160d70cd84dcfdf6072a467dce107fdc772a572db7d524dba41dd5e47cdb0f0c80e497b5975aa84e21a0ef2ae42ce7642fb18042824e3385d89da701a2691f90d2d005d2616ba1a40a0ce31bfe6045470fb32b91ec941480d849a54960d330ce5d7f826cd21d3af1fbeb17aa4a79a75dcfe70afaf88bed2161eb14f8d18b3787f8429c0ee35450564ffdedd2438c78c33a1813403f6c9c40fd76140778f08e65a4c0255ba039bb7b7530b363ac007b5e5412d6aec51b27b4ce1301b098f1eed62b29bf2e8b6e11a9917d68017402898400a46234b27373c05b6f8070d741305f55dc169e4869015d98e939e5b8d102c7e63fe135cb8a922b80708d04c66&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.840035367Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320, Name: checkpoints/320, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/320?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2cb3bdd2a5ede0860d74bca09b7b81eccc018cee22dafcc0f9eb75001027b4d75c8797d8eb403fa177d6a812571fb17130cde6911456d0524a94353303a2a929d4b111a6bd358ac55c79822fa70fe9908c5c93d17ed75b8a3ca8c1977b7bce8347f54852484f24473a407768c16b4cedae15f16f7e8d695baad6f7e83c3376db227f7b725204a68632cb5023f367e773f6583c3c1dc955a40d4a3cd07e88b3893b901e9b6b30a3042c30a267f10d46760a62d404bfaedd82a1a4879cdc1767a3448e83f6a1bc7f4dbcd30d79e2f45158de656a0c4f100e6f5a43584cbd3625f0fdc9654c00f6ed85e8d63a2e76ad668252cc8b6333ce7ed1f125260b6830484a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.841032962Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0, Name: checkpoints/3200/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3200/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2819043fe228f89dea6d2983fbe909a62543ecfb9d3fa08c5deac2711c783d4e1b634dc536f2b7ca44ee971e46da43a189150acf0c9575aab9cc01d55cb3880274417f794c606cbd563cdcc156fc3d01c994b90ef1de6990557b78a087de95d9570e68adee28f2a79086329e7f28c1e8d38bd68c46db4d07e45c13e642c4ab3680b344bde3b7709ea05c542e85fdb91ac6c40984aec7ad6c455b77b3cab4bba12cbfdf8a0bf51df033256575b714a84af0867ba79e459e6ff3df1170b3c47e168bb12cee7b4c05d576bae8dcfb6912ebce4b10a11557cf1a0113df7ec9cbcc64f0dbd866b93bdf7371ae1340afae56faa9b882c842f5b0ee58945b88439ccaf9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.841087568Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080, Name: checkpoints/6080, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/6080?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=c180c71f91a08260147a376bd4bb526dbd39a59b733d002c23b5e14162f849ac1161b936850c3e9865aee07194dc2fed80eaaa59e65192e350ef686824ec32f052eff5a941bac1dac0dcb0a5938548ba1307e92f5baf0911f0114c5a2fac2eafd1ba840c06988045a6f8e5be909636546afba83723d2522d2d98b921b69924bba8c86660bb89b9565d8a8beb93bec4933a066eebcb768fd3a642c0c9ff6e4481f35823c45455667d7df70cf2ba6f44ff09fb81d7c5dab4fe97d8f619926929659fb2338c054d3f07eff42d2d4b24b2554e74e21af46d93c0abc0d41c34bc91c375d84591c82cd0fef02b699cf26e6f09f75735803bf36bbc494c2712cc29bd1c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.841274345Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default, Name: checkpoints/2432/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2432/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=aa211c47295129086e715452858dd0eada0b6c1466ccbfb0b3ad9761a775d9bc8b0a1e8526084a956fca6e4fc85e0086f11af5626c25067e04dbd791ded057fd9bc12c6d234dad51e8efb8bbc05a441ddb7ad95f6177734cb1cc42d12cc144841c1f7afbf568049c24138be106d488e34bc0c9fe664aabf2a5b05afb0290a13717619d4f0b2b1ca86a44371078a201ee05ba28790591517cd46b10bba8de1f6645dc455f1380d70679ae2d6509a3e0cea8bb2a008aca72f2d0c71a50b9496d1e5682b8d821e837e352e8885118b9d54a0c508420f4a0d7a49c5f640b9bde2f7165f43c662a0db086e00122c711b2f55fba83e822d287bb821bcd56143c728035&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.841310233Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/array_metadatas, Name: checkpoints/5312/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5312/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=3835ce54e0c574f4693ffb248ff02448f41d40490bd8908e1e83e43e05c3081350f4a015b052fbd106abbc008e2ac64317eb1b51cd927957ae65c37a9cb60cf86f7bb7b1360397716dfa8cb3aca5f8ff2e6bef98050347cf8a5aa606a3c112191b32b02d28d2b80002aca514b344e31546df7512abc0b853464ce189044ef683ee7ac030b12b257c5398452f886e89d6ad02c206e64e1caded74efb51076f184cbd2e84516dc4bb9e5e0d312f6725d3071f31572e75f2ba1862a4c0d7350ef173badd8fd935680f07c9f5ea486461de733c631b44a26f92cfa33b1087ae03004c305a8155d6f1105c8fc68a5560cb5c5240acbd5cd2ea742bf86328b289a7f0b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.839460634Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0, Name: checkpoints/704/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/704/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=432c38f3d77c71e715889e120fe68c95f6627e4436fa76f85ff38f4fc941d281faaeb8ba90ce3c5561226d93e390e54d5c39f084e96edb9a429b08670ef7d23b355651112ae2f6ac1510a402024013a16f32d1f0db118487e7a8be95f8f6f5e829be31a9f1f3abc618a4923fb4be299152c39a8c0c228d39fc8e08475db81b15a8ee59f7e089899991d687b2d657bbec288069153a2ecd97af1788151670783da5c2f67c81e51fa7296eea680e8ae4946a4dd0db0e4f3273fc002aa7ad1ac0a3283b5d43bfe810dbc07c3d0d5b3135cfabaf429ed38d1dd765ded21a1a63a4586d5d2a13501a2f3b4a57e5b508f0d08892d1d9a37536b843f034dab86e2edc07&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.841379022Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032, Name: checkpoints/4032, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4032?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2c6b1f04209b1623ba148c2bd39bab631896e4a639383130b60c8bbedd3f7847449bc8c62b1aaa8a34157bc7719678a3aa66ad68564eba5a95595deb341a4c9ce8a0ebed29baa7e5a4b1759118048a47be217a28950ddd01e6c7cbd5128fe84609fba8474642e9d426d3168ed2c620cd83995c027c130c78812451342a4946ca4fb67fe84c5049a5263231e530b37ca97ccc45204a269a8c08326444f6a79512dd26b60a6453e732b648c29be81a9dd178c7bf2b0ce7920fc32585d6a87dcfbb407177680743870f49105bfce62028b2aa31bf7022a4284bf55f0c029d58429982eacec3a320827cb5bfe8b8294e9745968bacfa9592c5b91cae5ee402e5b8ad&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.841423481Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0, Name: checkpoints/2624/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2624/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=70bf53b65a87e5aae939b7cda1d8c478ff5318561071701365c152d157dc8d6ed214eb98849f64c97b48219aa541a54f866c97a1b707b5993e3fe615cf0a9b21530f05def8e1c2eadeae1aab07eae17fed630d368040825bd0480dd5b6d3c50e7ba42f78d69f040177f6bd52845b24e96df9ab373eae789b289f7a21708679c34b4d9a911e40f05410d3509adaceff8d3cf27f14b2abf39b7f4fd222b9ec9ad5135ad4729317d3f1e32ac7a0ae0c027935a607f006033b1c75f0e15afee9e11befac56652e69d71ebac07ad31369d07500378f273408fe364447a0474a7219b09908cd15a88d7f375d484f6c1d6a81fc4d87184d209b863472517248e981d4f1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.841568001Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/array_metadatas, Name: checkpoints/704/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/704/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4e058df44d8c4b2ed69f6193b4ff130125af2c639794dd21e76f5f4c47441b3875363f96662eb1074c15872bc259734327ea8bc0ee5111eec9b980243f792a62fb91cdce7916df79647f326ad087a770664b567c7f0865fdc1cd92410f655d3d87e41e54dc34111c3fb187120d6ded218ef0703c34e1af018adb8902d54d35cca54b3cb23b6ddbff3dcaa830f679e64ab24a26a4e2338694aa7bcef8061dad1a529c24adc4a52dadbaf717b799f9cfe474080d4179a75cc98d1fa8d3f8998c29513539992a22ab3cd82ef810eff9ae36bba5e9ad46246b86f530e49a00e2ab83275bf34562d3214e97f626e7a1aaa3918813677c7b1faff1b8910f55fbba485f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.841631531Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default, Name: checkpoints/5312/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5312/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=70ab977143f8a57bc39a05f1d8c98d4b69c4b3be5dfd25c286bd53eff08399f19fe56aebb2709fccd9b12ddaace06f0450483b1dd8d4809ff32cc5ff5279527cbec391dec6ee9f79446c84905c601ee8c3ccb30cfd0678d8aea8a0a3bf93cf8bf2e223b0ca3094fc21c7af89c2d641566702b41f18cfb6ceaf16b6f15e571512f1c3905e5365ebace34c5f605bcfb04f109bc7f0fca306ab01df5de04cbed7d4b1e970aaa572dbfd18577dba587babac82a50801e6ccb981416c6b668f8bb35ef3860bebd30cc54577a4acc9c9050e13c202ec210d4f8284604c5a17835fef7e62897e579a14cd4b666e80a30cf94edd25b4b0552b1ddb3ec97db13c6332c6ca&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.837492875Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/d, Name: checkpoints/3520/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3520/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4e91609bec589991b6bc9285d65da41e76b78ea410be9c0e6fa4417163233c10a0dd0cc018e5b6834efd322320a54731b5e5e6f9acfdc7204d3f773ed3357b4c4b88135ef758543ae074c71e9f021213c74c6d43a4ceaa6b850e959541fd0d40ae13d0970ed8c64698a0110192a51cafad055f6143b19a112b00b43200d7b415df56dc306a6a62ee5ddefb7880e37253241c06e0f6c34c1e66fdb7c27c7f33fa763f897098ad8bb8fbb6ff6cac276752867e5c9805e030722dd1bfb68e434deaf4abfa2816557974cf490ab6b33d3d3f667c0c4a53c0283957a666331b2575b3a32b352d4fa3c0dc155e6d924dc952a133a63494c6b36556c44fec6e56de5c6c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.841825658Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128, Name: checkpoints/128, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/128?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=14a2e9898d1bd497b37f8333798f75b7e8d08757378fd7391374713056609b7fd0a146d09e31407c2a055c4fabdbe3d8234b261b75fc884d256ceb2a38377fb9c768fbbe4b7cd0414ae4d3f65568a5d395f3c8b467ffb5b1041970a8fd0dd5b66c8280025e12ce59d8f7c6eaf77905c62a3fb9dee1ed30a842b04b25009fd8b3e64b75dcdca9c82a5adb7fd16a4a5f835b4550c63c458d8bbcb67c813ef7f303d473c826bfe49c4fba0aceea2d45abf8884833302e587591376e41fd45081ccbd2b2521201add14f6a68858303fc82e790529c3b1808d101714dbea6cbfcc27afaab884b7b74712355effcf4bd56d93f4e52b6223db0711d5f0bcf9161c9a63f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.841890903Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/d, Name: checkpoints/640/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/640/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=21d0699a1c5dc12d17a7ce4f859d28180574d9327f43390f81961ba15c991e4a4fd3ce1b43d18afb9c7e8cdcdf9434da89bfb8aceb540c5b44328d0134b91de92d8cba94c548666038fd71439aa2dbf4c037d39178cfa795a3c5a748293686c27a69fa6f649b858205460c1b0f909c5a69623f22e9abb483e07dad9280aba543594d6dcf28e740079fc6dee812bbed9c2661a22d534ab9faabc40d3b3dadedcf022fdd89962e909ab7e929cbd4406139d3be166164131e82303da4258772b7f3e4d14ca98ac8fc97d4d118016f8049fc8ec6f0811b4acba6c44910091b49c52e5c5cb6e358a76f3fa70d5bcc783e46e7d4a0bdbee750b1e7218f6c38eab3403d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.83930845Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/array_metadatas, Name: checkpoints/1088/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1088/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=3c5ebc92f0993096a33b7ee7151a471cea6540fbea9e9a99c02e23b5b47708aff09ac18f5a59672463c1899ac8770e856c9ab2a173e447511f34cd6bd5e509e863c1b927eca420c7c01ac5c96baa2d42bb1af0a6589b2c9991ed71ae80e01549a4d7d4312e299284a36c70adb7f3d2ade4adf728121623269a138435102bcdc882e6601ca951740e1b605c37677c9923adb75654ded21c3b58776a341cbd39403a62e475e22f9aea87741d3f86b1e4d7502a251365949f88199aac7ddc02c1a17eec205704f8388b719963f536b0ad9f36e9de2de58ea33dc42142ea71255542413966fe510b2fbf116deab6f82ec848d9a7afe1544ca0f3c12db63dd6dc674c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.842232613Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default, Name: checkpoints/4864/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4864/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0c25aaec3590ce55fea16060b90b9a5c669499a979500263d0c911e2a6455ce29c90d4853d2e888adec7e1ccb391878683fa083bac9e651a47940b1cdd070b8db2962203ea489082db8c3d5d5101b711332f67f65d46218749abf5dcb037117634be4ad5eaab7eb2df11df7ab7e20fb5f3d7616091f78508245df33e2e290391c1647adc7ed9c4df4feda7eb41ec2b9d829617fc1170b4512d43e097e122124bb0ea828a1bbb91a50030ac00393075fa494ae769ab2cc970f64634cd0673365956d06f9bd1bacc1e7ae92b5f22dbe22cf5dd556b4ddcf0b9249c5ecfef43709508ed65a6fc4e5c74684bf157bf4fa22baef15cba2b3563d828e8014bb769d7c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.842286397Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/d, Name: checkpoints/5248/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5248/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=295c1c472a9469d7df30bf57e063e1924b494c5d11f9caab669111d1b0109a92ff800ad8d5898ca0a6458dd6a31f734e4608b248c322fe85a2eade4797bbaaffb842f5e7df5c2129b3d6a9e86c807ebd3e7466d6a3df161cc6ae574136e7862a4aa462cda4e0f1dc0ec84ec7197694d2a44461832af40315c159dd408981837974e4820b0016854ce9ae2ad34c5468c12c915e8ffa4c8d59c838c533c55de6be818e5003842d032302164beac18150c251f51b1ff85d918242cd3db7f66325c0eab0018fa17a6b367b6520facd8824a61e430496a45a79db7651956ea154ef1a3a09dfe2d4bcddabcddfae47db0867df9d8bc4c599d00aea1ff35d5189221194&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.84233797Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/array_metadatas, Name: checkpoints/2240/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2240/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a1ac7d1bea957544101c4e65e4f86ecbf4100cae174907a8c52d0248aedfdbadb95a39a198025ac806731e47876e4cf86cf40e116b1aecf4391fb562f36e035fac2a35c214478bdffd5a2d14d96e36e44a391250a4e4cd0b936d6555331f0c3596ba81fba8aaa483f11777d33368dc5c1a8ada8438638747fe513ba284a272c0f132975676e73cc582509c28b78b2f57cfeede4c43f932207e06ec59dfcc3267b2b460e867e8042713e014e6e16a118bb928c7b6a992d3938551b36c67c80ca7d50de7a067fe6725d18c871a068ad50f25fa2eaf3797e55f5af03273c7f5ee1bdca0df201fea41f530db0612b0cb2aea158a8e0d2c9b8ccda5ea35a1a30eecc2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.842340204Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/d, Name: checkpoints/4160/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4160/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9ccd50297c481925fd77c580b41c2db25466925afe4fccc6a711900bce70a4db43d6702a625f3bb66301c8ea39d9aef4123ffe3c0ec9b6fd6c8d6087f68c901368b8df50186c42a6ec1f1df4ccaba6769d4fccfdd33023845a64e93cfc488c7632ef3ed367fdd52884b120fa291097772d705594fbec2b18e70a7669dc64f3e6bb1d0fefd2d12a6b3ba2e5fcc0966190268e79850337bd9f33087f0a87acbee3def8bfead3905e6447cc222c756288ce5cbfc750a5d46ec8376e6bfe485665b7953f4d7d4c61b27be16c5099e43605ff4e93321aa725578c9d446c2dd470f4068bbdcc882e150211142df905031210ce274e6c0019431edb9a5211dac086d81c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.842389264Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376, Name: checkpoints/5376, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5376?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=29db53ac7e01a367a381ea839e6209a62977891be1a8bb5ed37e85e617da8a6b4bc68ed30b04fa6f16d5e10cf078445820331abfd5897708670550f27c0301c93af3e7f76aec9a9d969a6788ddfaa4912f6238e22710c85c071f3e56894cd4a857f02c9b4b1b218023f6a048c478c1912d77c71351812decf4dbe6120de47383d69dc84cc26e79d521c7aa06475572f73e438722cde70a622bc2133f3c178ea7eced588527f5748d556ff801e3663e82e84f4c0c45fa04d659712d424b6f5c39b9e668b8b4d62c54ad1e6627e7c960b1ebb4f09f18457f6affdfa2ac5a5aba35e5675b1cc4033c07ab8b693cadfc94488d0ef77e100205927a963afed2def059&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.841274351Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0, Name: checkpoints/4864/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4864/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b95b64bf3c0e72681bf80c62cc35bc4e034968e54a5c9eae57d42906c14d161bb7b8c89ab4398973ad12355d0630cc6df0a2a4ec8f69f87dd178b6bb51acac1b0bb9ddd0cdecdbf613b05041a1a01cb14c834a2e579e2f190b97ca5127826bfc6b9db446a608fc08070b1a381688b0ad7db559080bd1b2f68a7e000143a9601c8d4db8b7a3bc1617c66b541aa19ab9f2baceeeba7e88aa7bf456feeb3852c6350b17905721c7fa4b3b53bf6577cc307b6d6b9ba5a7ba4faa05784661aa0159880608f2f866ae149f04f8e018a7739ed2cac18788913038b6d7cc2a8aeaf8f49768ce3291d2534c801368c308afc1ff7463bd61fd7521798980210a170379b384&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.841843111Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504, Name: checkpoints/5504, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5504?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=128f71ebff9eb8f78d4adbe407122a3567d397b632995f2caac3e3e62808f4ec3a0d04eb92d76607e9a88cbc135647ad6b3e9cc4d188ed414479f22de06d2272fda3ba334fdb0bfc9c3080ccef4b4db55c770acbf66219434fa4f95f702465fe736bb1bc48f0450efe224b4bc05ce9be987dd54720988cf82f043da4b4a050808de278549ee72c8a73fe905865b200e725157bc2a38a6acf60f91b305eac5134e847495f9c96a4ae01c0425609ee43452c520d01ce5786b9318586bee300972f489242a5733cae23520ec217f73bad84ecbce8c7afcd5e2a1785782d9e77a9c50cf8308efa04ac0f096b33b4963277610314037f0134c37153a20475efefc1eb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.841394328Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/d, Name: checkpoints/704/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/704/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=29dc48e9416aeec09863a221c1173b704ec6201460708cc3eea35cb69010e0700790188d534a6590bc25bc4e086568021bde76243325d48ca77eceb8f51b7c09373683a51319c96d899a901e403aba529d801896026c208e8a1e8ed39a14aacce277e4c13ce8f8d8d49540ddd1a9249769408f5b29963b723f0376a7cc482d7bf99052ee4a810d35909db04d3e78127ab350d58e970d031cecd1901da141c4742610ac3a981d5229476b3c3e32551a533854ee0d0eb6622febbed92ca646fbdf00535d9c2646d12bf04a3ed89c7ad6920b87a14a32f60cd63b3492e2d21977619d892e590dc0da1df01a18d482900155fa8e2071352c36497977748e1789644e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.842247483Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default, Name: checkpoints/4736/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4736/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0fd43aba5540b34ba867976f41ae6f60e87eed59fc34451395967cc518fd6f82c070c06b9792ef10575446ed9d6f37ca86dbebf073e33d6de8c9a68190b50d297d6dd122fab939c67190fa717343a1c578bf5a27c39ca049cae828496a8089dace276a90350b0c23031cb0585bf9fb1176d839d42d861b647c29e711ad0dfd69dc5575a7477222a11728ec7fc7d5de52157da15ac50671d76dc4cbec36adc29d0c3d0fd6eef8c794b8a293c8a9e186d051d6fbc501698b9be41da9cfddcc9480291f0fa3df1618f943b0903dcf9d0ad58eb9bde09ef3b0880476532a303ea58e99b402f32a3055b02325638299fda06a6e629f245e9a3e53575fffdf6d9cb814&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.843458416Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056, Name: checkpoints/5056, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5056?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=858644007d55053cb4cc3a7739d359dc99b6ac3647821758dafd0bad6e806a694f63b967c29d08889e7b1bfe898aa360b055fa8aaa21a8fd8e3912f65414ceb92ff29ea161da271b77b59db83990fb9c7308c496cd633163ef26635d5fd9c9bc767dca7527d24aa635245b6571c52234e3a843188524325d9148b01360d63982bc0d743d402628ec887afb86f7b44a7cb0efe2e045d28f8e139b55f370adadbbd88ed36140523eb009583d524c42abe0137d1623484a89b9051d6e1f22e1289193b67fa7321912e4e02df1f3101de0f382cce0f99efd56e2e09a8549ddb4827605e2feb857e594e1aaa316de40e825e5fcb4b4efe80f942ae85bf3cbd89948da&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.843524448Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/d, Name: checkpoints/3200/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3200/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=934bc09279324fdf27e090e929a7bc48347cca5f58e37c6c899ad9c71d8ab333d5a5b56f6eeb414b1931eeab8039da884fcce2f8197fbe7b2879521864f9dc1175e22426ad24a5f407f6601bcf91b9aa6ae3efd02063a9735d58a02b14143d638b51b00b0c4285aa59015c6b4c1c252b5315f630e7470c05a4d9edab663ada384837e19e314827cd0821a060f0b2f621315af1ebb9cb2d591a2a3f31c7f97bca322d8e3edcbe5b631d3e87f8f5083a1d0f8765ec9c5a87c2697af7594e34695bc5067bff90fc6d47d390394e6976558217c442073de46b1cee81aad93cf3af73a3e41b9bbc61d940b63fad0f48132edea161a824981ca1a064ebf89b97ec339f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.843565697Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280, Name: checkpoints/1280, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1280?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=c3797e5f1f3b076a168dae043008df012bae7ae69f86b118da18a2d1a77a854d62c60cad29f4927044fd402887900435aff5ff4d8f409dba92a5fce5d8d3723e939f98cd267c079ae470ee9e03e8abf8c6e20c8f060c845e43af559a027002b5dc88faf657f4366e38e0e67c67a18af6cbae08662e5d5d1ed6ad33414c4ac17fbaa6e21d0c7a755fef8343362bfbe1e53b631274dd2f661e12a1e7a265772143e7eea99e258838c7aaf6c45a3a17d4413c5e876e5513ec4c6b8dd925760d96225f8ccd89a302271503101f04528f1aaf936afaf18626c38788bb47dd1073aa20fc482360efcc155822125e6d0cf499e5d11984ec893c4b2fd962c415cb689840&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.842623282Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/d, Name: checkpoints/4928/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4928/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=3251d57fc726c998b4678b4e347347241733d211b9e6bba79cf29f7f330c13dbb4aafd3a95b5d127febee8f21e7005734fb477785f66f09c54d5d3a8bf66a6c5d6738d79fdb3d4d13515df95828fcc2c12766f0ecf94ae7f9a2fd2386fc2f7676c7ffee4eee7f9e39947047933ac4c276ed8827afd6c58c7d1ade7566c2720ba04d2264dbc09389825d26694bd8f48b9806d8b58ca3123dbcb0476a750347f9a18253d56260658f3982ebae150204193c3ade341848f8821ba536627eaf97486de164b89fc52712b6b7b221086d4784f0dece36989f82f9da55609553fc8ea80c9053671150f589fc158282a8038da2640b453cb42aaa7e1dd59762d97c7dadd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.844101007Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/array_metadatas, Name: checkpoints/2048/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2048/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=bde2a004dbf82c9aa234b40ef2525f9d6a75d3db685c7b2c778c2217ba0d7e56425da16f08a3624664730202cbbd35dde6882e297e93f29923330cb44664da0c4a4e80fc1ee4b7a6f0100ed6aded39e8c69fef67fc7c087e388e17222841efc8b15468228a8660bb9867de92cd48d254ae35f7e7f2d52d236f9875580e35c56c89c5fa51f6220f2dec06889a15e1b707797e3a430106bd9ca71b621de46a66ee62ab74b723878ead2eaddd56ec736da4431341deb69b71ef91310a995450c905255a084339b13aa401cd530027c40cd4b70357107c2dcce41d3539b804c7ca5b576d4503ea9c726f7d784379158399329b0865e556c408664c94bd3365dc5086&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.844454481Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640, Name: checkpoints/640, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/640?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a1835d096d8b334a8872644a010262f1a669d9489f398e62a61d2dae38f413e9cff24f533eddd4d449e4c83d77897721a3c115df22e797cab38f8f034c56f96c729b8eb79e5329a62d92e6e6de2eeb3bb57f791b0bb014d2166200e52a047f3347aa36615925bcd5341dea40fceafe6d39807a44fd9acb2df26baf25f825e549eb068b0724a9707ffbcd5be50d6a12b82aba88ae74add1612472292af41b028540930636304213ddc37c85d5685b33e59fb1051da88c19de2db7b78e6af5879c06d65bff02fdc0c30229b300de36638f51e4d6df7371cbebe9e906ec800b06c8af3be22217feba373c0f7e05a4f6f2a7a0d6e0c01853fa8504699756a45c5ddc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.844760394Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/d, Name: checkpoints/1472/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1472/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=692c107fbd2d3a276695f23d713c55e19da29772fd8f4ecd83a41af99e84826f89d0c43fa30c777bc72e7f40fee230eb096ac206402a0ff11d17ce5aefc3ea272ff2ad6b64cbdbc34ca564e57e43af88488523ebda2482f6213f88429ef05c460679036c6ba2aae17d42f40c27c8897aae822e66ccbb49c0185ea52ca9ed957e8f7d406dba6812be547c5d051031c6733062d39670943bb82338a40b689fe008f78a06c8498c63478d35bee3632b010a4e485e699d58d644490400f2b61a5d859596118245f44a2422754c1e2053e5cce20b76fbf0e9f8cf4a0704ad8fb674445322e15317661fc622faffbb3f5e7a9e5069f8fb28a8c9d5f9fbd506d714674d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.844836162Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default, Name: checkpoints/2176/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2176/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=08d3015225c29bb9a638009b8b370cc7dc3cc8673cdb86a62df41135844f8bc90c5bb13243a782a45201f7f938314ec8a9814c7fa9f7a5a157392883d76009b7d359bafcbc4ba833a51de696186a2a0239053099bfcd58d90d912e5a7e536383105b493400d82651ca84abcd719c3ff4857218a03d3501999fb13d747899807179d026c26c66a059c6aa8e64eea08603fb00d3fbf85d1c93f2d312752cb7b5a9cee5e5fb9f3d99e267d87f007af33a8f1ae775f3a332710bb124a85a784c06e03c1446280794902b6d3c498c6bc4106325ac154bf49fefea8892109137d7948975282c03dec6fc5f1051318ee6462e19520a0632ef5e335410d490c2de86379d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.844939672Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/d, Name: checkpoints/5184/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5184/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=1d2729a331f31860cf2265cb1d420c93f6eecce37b4489eb3dadc0f04553be6e00171eaf1877e7bb84b4693de38f7ee2d0d69c1f2904325ca3ccf185fe8668794746618819a142ee673de31919eb371c0a0a83dd5396ec2dd7ec740fb204c46c6729108b86c622e2622b308eaf72de47c5ffe4cd581d43beb5c6fe6581f2b575e62f8ea4da71675a1982d75deee6958573e9b382384d2018354e17598cb0437524815d117858cf7e08f90a159e96a10bc2952aa6ef9e5d9ca174809734b4bc1654cc1bf9b4c54fa96310cace217fa2fd3e500e592966681e8f53c4f375322c456b5ced50e0ec5699d9d4c4d15ae706f6abd680af0839bf5097bcec84af863c53&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.845303746Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d, Name: checkpoints/4608/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4608/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=3a986c5a18328ae26c0c5847a115d72a14b5f083fd82b8e82c6870dd165efe69eed5fc59c5872b792ad5c4c47334f3a198e6e9eb797aca86a62a8695c06184215aef00ee8eb560438373dd128deab2673a3fc3123bcf308024e68cc87a666f2f2ffb29b5dd7ef57ccabb53b0523b1ede176acc2d13cdac1c73de31ff0ada9aa36d4e11057e3506af8f9ac2c0f8fd4781a5b79c01ab4b817bd4b53bf96b8e382a1e42641531b9db8d58f5c7f0873042c435e7daada61595f6a8392b5715b39a87a3ef8df69250e1bc408f8e7cd2738f49cb444b35633e1fccf3498b270469e0fa7f791423f733bd1e3563810651eb5de557c1d19bd25f0af5be32a13cbf18cad1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.845358394Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/d, Name: checkpoints/3008/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3008/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=66b14f5658b73b124de6a55966cf8859df009e451e073e4eb2c1be49c0fec7ac8d33ca38c09aecbf5e368c7cf3e24a29b4af5ca760225df164392759328e6750b009ce1d1c373b24cd622e3b6c14a1a1e91c8869641298e0657b708242eaadc1c0fc45ac9115d1df7bff2e8ba7eeea0c8247a3c061d6c00dc1da11acffa70d774e97f76cc79575186a5a10cce62b275a10ef41acce4118bb575943635e2f7d30ff563536ed716ee8910a55c054059fb703db686c68bcb358c661637f698d7a7b3bd4f02f862baa55317f16ff0a200177bdcf5ced5971f638d5061140f71ceca87b5879f7a2229616e2b7e5bca97ec9256b0622f6c314fbdc3e599fe181cbb33e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.845378378Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default, Name: checkpoints/256/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/256/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=215626e5a35c57da56257f158b8b7f5b8c663a4d4a100b5dda91d3adf927f1548adaae06ee6d32af82cb849146f8ae24d2e561af80b8e2f7ea3418493238023344c95778e1235faa36858022621bb4022698000fe833eb9a3d6b787e2685f348253dd4151f8cdbd68ec1a51bb5f627f8e654ed89a47e3ec523a45bb40d70bade9230f9de20a8e437e43eac4b8974e47cd849b9be38673f1af174204b11c4d69d691e884ba886a6be6bc5d9d7a567bba1a74fd5ff3cb802990b8509c59505c0f69b3fe58858d81444f217c7a30469327c2212ee91a6e0cda21bf81d0211c973bb4ba47ddfbb5d90d5cef5d631b69f50eef5244857ae000e2e8a54aae6a0a2cd2b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.845423501Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default, Name: checkpoints/5184/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5184/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0a91b604295fc7eaae5ae217b692f90459120b8f93eb35e5a064d6c73dd7ecaa2fd5bce7356bdbaa91ce1944dfad6d42b5c5af979ec794a91d4583b69709ad200a682f487ca0d4c54e447ea343895f23d06a47a5eb11ab58c30e2f413e1e274c9ee19610d34b70f0ed80339a23c276d7d938b11e411daa7470fe72e71edc88ead35451ea08bab056e738649f2d6d1370cbbd4c1a11a27fca7257a807ec3b440ba6b7bc430da012dd92b4eeb2b15cf0c8f560c0f14c615880f7e540c848dacc81d0d474930d2af0a2d0d204e0ff76c1d13e0a0a9548cac2a6b389a6bfa879587edd4e0584da1d907515ed7f84d53c5cd8eec808d48b160598a7584a020c510fe2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.845455807Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default, Name: checkpoints/5888/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5888/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a2cdbea2c2e56ef80651a5e7ac1764d162f3bad42a77b7ea4b900b0989b6a545ffc35ce37bb9a7a556c46d2e690fc22abb467f899ef1abc3dc91669f7bbad1855cbdb78bec1077893f967ef13953e1d928c7a7f6011902d0f79aa866315c57a1ceabd5753b8806083009502b8f80ab457fbb123ce39ac8efc31d1b19adc0e10e18d81a3a9ad40f693b7865c5f1a2f318bcfd509309408ac8fcaf92e40a95a2461224382a80cb9f081f65462cb9e842fc6e59da7c2815c2e32c289deae3fad3bc0071c452913bf6a4bf5085a89528d2594430f7034ec1705514cd09b1bc6cc692cc10b8282a38f851a8ef4ca58e12084e84e3ef1599004001938b52ae2e942d86&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.845496223Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/d, Name: checkpoints/3072/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3072/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=7169f208ba60d850e896df370b379c3dfa536e9dfab6045e431c66fb9a97cd91726509d00304a85a9930cf682e861e699010651cb6095e3c77c900a9b26a450597a1eea98d52e096d84c21fd4815cd19f9388b1bdbc743d741ad268d77b6a2b16b094108bbc5fee3d05014f051ba2ba5b2b0516a4d711faa7c5f1dce13e70fab6c0207a8c0449df1d88cb2bb4cc2cbd44ec3f06dede28cdd3c69d1337725f8191ff2104cdd274066a63f193298a1e82425b2bc640c1b860e248044548075deebddbff1b97f44ff8992798a1e35773995b31f3bb4477b999e466c699e992aa2cc233154d3494fe771cfebca4fbd2e640a7bb68f5c6e415cd2b0701f967bf5b0f8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.845525573Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/array_metadatas, Name: checkpoints/448/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/448/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=71aaff3da4e0ea97ee4c6f5e81a1d598c9939b6d722a6640f358adfc58f0032360ebca024b2b35bcbbcdeff6461efda04d1a43a5377a68920138c9ba68947c2e8d57fa876979ee3e0dec37f3a0f37f89c68ab0ddae359341a58fe70c8dc3a178b1ce30ff29733d4da75b58791c638ff0ac3b623277480240137e73015d0758bb3ef620fbe064ad39d43c2b6de45d7ee753890c12aceb7bb56ef784698f84a348f392d5ad0635e192f97ffa806dccd302ba079464cce879493b9dcfe77a57fcf0c84079021da026b65d65b964a9ecc5d8f160d0f495e1b065167348bdb4d93a57dadf732583a82fc101232b8dc6f7cb87b5877dad84013872dcea81cc3371eea7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.845563081Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/array_metadatas, Name: checkpoints/2368/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2368/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=8b19aa1f2061d9fe33fc80dacf26a05c288a4176b2fc086e22cae2ab73aefb37ab0ba4fe5ec8222d80c2c69074fe566b71a80044f7d40a958b5a7563f79adebcf2f0ad1cbfb2b377950d78b7a46c41f19c128819db407ba250ad886c425da27e3e8be3aa4c1f781f42ce18051f23302caddf2f07a3e1b75ac3ce6f1de3075f075c7703a281cd75a94c3768d2e1a09c015e2be185036c6b26ce3f050d1aa3ed45a57a3a24b25cae435ddac2961c4ed28548fdecc64ad6d3c6101ff9affb4c5d22ccb81a9803241245058c5e23d992a39de9be608ccb82d2dbfb8a55b3c74cdeed535fc8ca1acbc427142caea558fc67ab3ed1313322cab98d7a946f5998d0fa15&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.84540262Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0, Name: checkpoints/448/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/448/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=8d90b075f6f342f9c47509db004e6c1ef1370760b1a3e968bb3d54d980afa6f05d2b5d3f66f1d62e0819dcf5465cfc3769f5d2f70e32dc7c4fba092a18279a939d9cd619125fcd4100eda7a468d10b081d6a11f0e26c41d04c37d4e45a1aa9b6839d59d149cd968db8e18c8f4a4e6cd06b9e6f13cd89828ece6fa95e05742d9ffc5197c9c53d481e77dc7f9577f4c1d18ffb4f5a4edb9b2d0264a41f48b71df436900ae8890330870c1760aafa299a304bd49f3d971cb3b89348f4073f6564a40c750c3f661d1fb111b6674813275c7fa047d1b1b415328831d172e09a0b9cb9ebd76c516af8b61cccd73e717868acd5e1721faf46fe6792e4d9045411e26050&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.845584855Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0, Name: checkpoints/64/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/64/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=3f2fc7edadced3c64b4abaf86feb6f72e1946d8182c8feacb672c838e4424600331489f55e7462aeff3f27d70934479aa0e6194e5d6f1e4289f22fa8c07e66776fb081b05411fbf9ec51e460d2dd143735fba251ab971b2af8f9127a7e3fc021d75ce3860cc7d9ac2a9fdd92fadae3d65fd066e8c1f1b73252c01567e14dfcfdae7c7f5f698c7d444a61997b31cbccbbc6adefd19252263cf248a9fe9abff275e66709e675463c835358fa8fa13fb9f817c7822194e2bf648df02ad1ea4332e1579efad526c851ad310b876ecb228034bf2bff1c161d720c18a3d972304b0a8a66928cf692842d860e546a02850e4f3bd81907df3770c46fbacf4f7b4a3d8d3b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.845668931Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0, Name: checkpoints/5568/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5568/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=63bf959f26691dcc8df492dc03a4dd3b62a8b9e1e8309d2e5d183acda28b8bd11d6fe8b3fee3b07cfee00a47070900f64fb45e37bb9c0063ec374243f1993515624cb5bd2dcb7feba647af26353cc626e632306a6f5afee40aed30db32960426909adaaab877457fc7bb3ab4b9f5c3cd941eb97d6eb0ffd963a9575fa6e5ca3b529cb8340fc5f79b1594526d5f79d11b6597c787e622be6d098f536563c0c0d48231961ea33d5f7cbbdcdc7f1816f679104c6ced489326ff31960c655bd5237caddf03a2a090c2a1441dfd3e648631b911678364680d1f4aa3b5e21e6f23789b4f61f684cf4c7aa5a6be5012f30f819ecf4e2daf2f03d72c16af9692256be5f4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.845539718Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136, Name: checkpoints/3136, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3136?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=3fd8dbab2df93de47c0617846e40a48179762344953362ef005e2c28d0ba6ad327a3e9e5dbccb725dcd411b8898862745d7aeb34584e5d3832a264e0fbf4f1632a0f7e6fbf6d1c69cecec08c0c2f60d407c74c259edbb9d5a7b068cbc53550fec1e8335db6cc88be485886de01b045bd4ead6c464787e0d8db1e1c4ddc4f87c25f383338f2aac1f3926c807927340b40b2a49d5076c5127162ed11a0b8a9d1bd2c732e34854dad6d424d5b512c9bc0e59d9d666e6b2d1a718521b82850cbce1baf9d72e58e88b9a4974d59d80f2f8ba85597ce0c60e20b3ef151ff79c95d79a014b2b56695e9650246379273db830054f98baeaffdd9ca7f0f307a60ea5d6d13&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.845707964Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default, Name: checkpoints/3136/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3136/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=78567438ea5915bf33d3c1104d57d246152908903e0ae632f83db37241816c19f3314627eee5192ea809410d5a9d0de4e8912d0f61968917406a9fe6a9bc4248294d33ef0afebb09894f90fbc626e3b09859812df29b0ed188e0cf1e4de2a41375b5ed18257299f35f2584fb94be585f07b2df8ba71d76135434a779cad0c494fb518f72c8fa0d27fd32630225bd6bd0eb79b515a0a67bd5c8d423e4e0e5ba14441fde2df1cf9baa3c7094127a134b0fe556a83141295d35433cf92b57affb7f1c1c65fb0049f44fadfa3ef470002b2274ac9206ee707d414604bcc917c847d83c02802132677cd6fc6a2ede1a7c113084ba3be8b9ca0bae6d3784dc21147a03&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.845770202Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/array_metadatas, Name: checkpoints/960/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/960/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=734039334b457f3108d887965c2a5143f9f34df5ef210a33730605b72dfaecaaa7846eb75f21444bafc1ede8a3d9d05b2225836d76a8a20226502f58514b4536546eca78e56a126cf5307032698c91d0209831df1e5b30c464912755781cd8488434e16240dc5cf6fef9c1ef8d3337244751e82e12bc27b29a07ce0134c9fd2d843da8449589419ce0758a7a900bea8801d0a5feec8457e22d8c7b27ecc4a85b564da4c3a9d8d2091bae4b93ecfc25adb5826ad001d628fb2d2d4471cda6d2ccb7f60460d83c61e9a470e18ed5ed345c2e57b5ea0f915e91eac5dddd94c7483f7c50e11b01d63b55e432ded041738fb8bbe13889031c7e1e96271edbcb14bf1a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.845785881Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896, Name: checkpoints/896, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/896?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9037d2a6a9ad206dd72445a79ae804dd7ffa521f2ef545c2908f8cbc963e3478fe6e80e9b2d16eda9ad77bf57e602f8c42961fca10b41e634744ec7ec0272dd34850102f32f8b2db7e9ae07f6499db604023164c3cea73a813d38d14c797c528f6095a3cb13c202672fab683b4cc8f25995be7f42dc2781d2f39a9d9736acdc99c7297316df9c69c6a69e3e2e819584ed3480cdb8d10a050c1a08035bd45ab12758f31366a24195a2f954987f09d9efbf2964e97b906b3589c3deb2dea5e2935076acfa668f6c5c44f53aa6df8ccc370b4048a29e17357c554d47de21ffa7fdcabde4f39e6e556c9f4546103ad4d2277caf5ce672cf1c6cb3c735cf711feea2e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.845481007Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/d, Name: checkpoints/4416/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4416/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=72227ebc128796ccae260fb085bcf88f5d7d10ddae51cdc3fc51629f244febe22bf6b0f555c5bc2c74a2f4e472c988da2917d4748f6416d50b325f0ce3d5acfa69f293fd326565a96e3e1fa9c012159fd3b496b600cf7838f5f6ada426d6f8a28b2beb02691e59ed78af8fcf2eba3ca6aa6267b9acb9aea95080d258e4ca2de5dc04afc1d7e37347c3209c0c61a98cb32dd94efce3b41f2d6a210e0aab2ea0798227b1c89560610c6067127d859bc2cd8b38b2372bebf359b4eb99537d791df2e65114baa61f0ffdd7690efabcb6194133f1d9ea16dc47adbf4967f0847f5091c30eb7940ceed454b3ad71ea951c0d51ad7d2cfd51fa76f5043405e5def52c51&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.845920546Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/d, Name: checkpoints/2496/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2496/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=99e5c5366a7ee6344068a73ac04cf409139eca9318f92ba34896b389a824f10077813988a2f945b4f03e6e9bea7af178e6db96cadf0b0de69750fc2a1059c5420a65008e93de2c861514eb62079a423dc8e4f7995a6b53dde0e22652cd66368ce8fae543c36490a8d41df1e86b8dfda57b8ead94840b6f5038058b83fe90cb0a5be271b00d599c0bb88f4d1c3ef0d591e44fd72c699eb6e096be3b4faf2afef6d35b601080e4864d3267b674075dcf86bb47be5dd79e4c355e53b05b6a9a174e769ac4942e5c2c94ddb6d99113ef671b9852a78a1865195e9dd98a4fd47c4e75e2f5a55a9c10c36af549d3b93354b9c71d5ee1cf2e38f8a41da4eb09eb91f5c3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.845592002Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152, Name: checkpoints/1152, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1152?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=5ee41c26ca30c1642367717e226db2a238f829acbdd352e620592fe019c21c0b1dacee70e9a33e2173fac4ea7f99eac61646ff351979698b966bf97d44f7b38314f7e66f784b9a9a1cedd807395f7ece1c04bee67a48fa0a93e211ac6d8331d90f7d61cfd8e93a246e0100122e00825efe83b5afe2f916eb6e4bcf8e3a6f209fa7c08b823b1bb7f7367c515fd92c0dd0e401a0b50bb8a3f7ab36cdc04726aa052d0b8b0bb7f2fd60d94e5a6077ddac4d47a858473f9d5dffb56b920724fef1b37419f3536510820fa755b7fe4f3c0327472c185b1ac0259898ab60180d9d3837857c88636e94a63d3d162c73c4f68ca1ff4bac890b1a9e535ec54ad647539546&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.846036181Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/array_metadatas, Name: checkpoints/1600/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1600/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=49dd17acbd03623bd24d125d66d59a14c1eabc20335fcf6a6882f459c90a80cd96cf340428a16574e7efc4099fe18651505b760ac021668e8c199e57161cce5045083e6956d1b8feb113dea518041447327b5818332528e990925f4ad12aa4d37ba353ee8d90de9564d8f3f2cc69dadce4b852dacc748a81389d9deb90892d7f422ca6b8ad789132a15d90c5c45d8863413626d4108109aebc034df6592ed730e772fe990d7d85f83884e9fcf7f695044b70e19929e7f773095780103c12094b041fd0b89a7fc8c423632236eea7077a7d808d680609dbba97c30172ceea34b2fb4517138691c3c0e86e5a22b22b0b04fb6b21ca1e139ebc64e0406efc22db60&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.846071444Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0, Name: checkpoints/2880/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2880/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0d33138c1e3592f2748edf7b5519304a8b30cecafc83e99b4f71ab446155a44d5f73d767e7ce000e6c7aeefbeb3692b80f5e2f3936d1db4f4aebb093ab053ea1261549d39f24058c1853342d492e51d632407fe2c1926601d9e5d22e91fa505f6156437957cd1128ab25d1a0b5f1e06a0d5dce4945662b94e0c7031a32dee435806e4799ae54634c2cf5ba6fb3088c7e33b5a34236b5dd9af20bdf4be72652de34909c2ad91b10a13e24e92d38cfb3e2e6372fab36b811438e251b0437b8f952e915bb27418e36cea4fbdde6bf65bdb12866f3c46ddd78598328340bbc581402ed9ea973cd5c5bd6a8a87574a2c3a7962e474e139f3b33101e707db4fabe8c63&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.846093877Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0, Name: checkpoints/960/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/960/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2f0a5446e6fbd8ad2570acdff230736505bbcee2752d794454d7d6a9af043687ee9e89f7dd1d54e9f34eeb1b14ecd392429123b53ae7d8675004e448322b86017c248700626d118a89b1a8dfc615b4345a9576647da3312b203f55880673a4a47596e0c68392660c0073c14a3647af84bbb28ba9898f3c31ae988722647c5690e46f1dd71299f77f90f57933b2c1f26fcefddebbff03ee9a95e9ae742d7aea2edc9cf10a1402443e7df93378ff7177339152a972f93694975975a5fa2b55d8e19a7fdbc11a82683c02bb59a7dd58de7f0d60f40ed67588b07c091d8c83e38118fa20583ccfcc2695138115933afe25d8dcd8b50ae6ade184bdbaaf177f86ed82&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.846135704Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d, Name: checkpoints/5952/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5952/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4ddc89599f6205ea1a6d92e8796973b3c399cc2f3d6361bbb49f33ab479d1255743a8efe6e73ab4ae184fe70b6c4f8da94fbc0e1ed9c61d98f1ef535babe0b173ecbde6aa1415555d4c85bdd1d56a1099aab0873bb2f5086aff73abb2161babe1f1fac512f3b165e75d0787ae623a5f7f35b548e1b292af3146d320cbe348a5cda7bacb9f266c2213e360c4586262258873bc6412d547fa8f71450b8845a6c4bda838e851d8080b70342100b9fe5b489b63e72d1ecd72e382f21cdc1c185aabb5e90d465ad4e02f7a379f1f9b30adb9c16763d6c6ceee83ac0ec93138b686d354a924280a5819c421babe28274b16264bcf13859efa733238fabeba420860a80&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.845453993Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default, Name: checkpoints/832/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/832/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4d175168776703938b9bf4c7918c17bcf7e64420138f1d314b91c210d1bc9af12b12ebe12f53f22c21bc83ac3b085ab90abf61a058e7cc70841f114ef21e9bbffc8536ac173059cb4e0b0bfc8d9060c3606b6abaddf79104f5eaa0e2323fdd87313ad6c997594454cb538383e276c19a8f4b18551242ef6162c7a1877049aa51588ac0abc900404c6aa6714493a675d9958486740106a2e5031c4f4155926020ceeaa153ac4df4f0795642cc079e92a7a71ae8510d229a528cac7e598a20a2264d82f2fa1030d2a9836685323f661e82d12f28f8938d898e8d6d645eceed47d6e1d5ace4f3a69fd18c0b33924684e27c71b545156668d7d09b4bff24bde1ee4f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.846057431Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0, Name: checkpoints/3456/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3456/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2745088ab6c5efc14cffffbdf44da67139f0eea7f87c80076e23507424334c513b4e09d394a243f7ff6018c8acd5facf4a6ebd71dbef8744ac243ff81d472d8c480e63e9c1585ce19f544972abe91f8a01a6d67eff90aea28fbbeb25940fb358362f68a015d98f49242487fd8481d9ca29e917c355185406087ecfb803e565a366b5970f72a11fdc20dbd83f6f6b456560236dee4a6c0f4a3b325172f8690d1da7e66a551506a232e4a5fd0e216a61d7d7a429999f5218b1b5c32d2e6ee4b8ce6f77f23c127c758df6914eb566d73c7d3c54572fe01a58521f1a129c9a3cc60b03bc5a5233b5e2e286f0d86b8157434a236125cf2f89b99cb21e27568c5edf2f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.845783949Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default, Name: checkpoints/5504/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5504/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=6c04594263a621e65635346fc6776c6bb9b22cb53595749cdd33bed076906201d86cbddb9e014e6a696ce72bec3ae76ce22d32a9d9bdceb6cb95299d5a7ab9af3b3f82ac69d0e76394babab630324a268e64ed34474011471456758353edb68740c75d9ef0cb5df87981482736938fa2772ad76ffe79b051961e4b1ceb9ee5168d8c6f47703b2fc5982538e851b76c1e92461074606a580d11b83fc0ae1bb841ab2e3814ad977ed5b12c0b60a21d7acc9cb8dabadd833d2d2de01d8b9081134a76d0187a9c7abd76f6659152bd859fdefb9831827c0bbc2669a1a356a7fe4efa566d6071849a92452023ad91429b3a8052ac99da98ba77e3155cc266e9a46af5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.846342983Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/array_metadatas, Name: checkpoints/4096/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4096/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=6faec248944289374ae85d178e433c3cd57b95aa31017a7691efc824a4a6c82e0ece8ed2edbc6ce4ea267e6cab14d3c06c2a008f137e915d65899c92b0b6d1ac92a5d1ac56f7a1c04dfaf8d1a502b180ff5bd8b0e2e3d6205aea9ad74fd9d398ae2fd7f8c97dd969b07ffe46ce4cd199e84fa23fa1c7b918d08b25a2a54687a7860202fce59b8da7dcf93b465ec100997dd3752f5c0d31cf8a58ebb349414d04cb213ad274504e5ad49a00fa4a9521fae5883348f6bc5dd64069ec6d7f3a0dfe6f99bf8fee0bf5db9375a43e9254692d7de484aaec4e407bca54ab791b0da83e91bb90c20d31b21a94a332725d1ca858639fff34e6bfe2627b48d37c38943a67&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.846374609Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0, Name: checkpoints/3968/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3968/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=47fa64b9e2fa57c45552271093d03dbd44fb1a305c8b235b636b1e992353902c9b8831c7691a263c9dfc6b3d931923130ef480d19629c88e0b540346b31538e3f0cf3372f992a39db30d91115cac8d3d20662ea47de10eac50006d089b9b9a1dd1776b67f14fdad5a20fc94fc322ed5ee21e156a9947111ab2813ed07e3b0c5d860a6ad4a4d7d417580d03e0a14e6cf748b3a6229d8d0f24bbe66ef4f3ca6c20da8771d812a41b70e0fe47b585ecc10a51df1a8391763b7346d9e447a2a11a6cde39d1ab18a53cc7f9be106828b134148052dcf437da2855a0d873b860ad39e782576f71072c5a51e5c71d5b125eee004d931def2b7db05ef7d92ee5c5962c8e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.846400323Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default, Name: checkpoints/3072/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3072/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=11e7e73e25711c177495aea04aa860cb48f7020bd036eb9c7b0138b9d3ee7bdd0e5a4e9874151da196ed98e34e5f6007b1afa937deb2e1ddb34f8c1add8c4f3f9fb5a58d5564e99e776f911a1118b016a57de0670a31c3d56b0b403f317211c63c32f067827d56ec2c95788ebcf2ad387ad974625affb8c0418b5b4191123b177670080265c543d83260f810be9aab785ad1ae6df9d46aa37e0e3518c06a2fcffd793561f3afa5df71227394dfe6bfb0af2b57d3e0708e1a01b0cdfd465d64fe035940cbb09595d708f42d365e9f4df979e3ede3b6f945df0197edbc0fd7d0d91ea82d4cf5ca42d2498b7f6da728a9f5135887ebdbcb76c69dcf6deaf6b69c48&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.845450866Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default, Name: checkpoints/5760/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5760/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4280b053b032ef4c39295813d66946dcdedcd1275c8cb0f6fb79baf3655999cd59b4f3292df53748118be69744a7bb2b99c2bfa12a3f7bf6ae9ee549ec004008d9f2a32bad209f87b00a97540c632d3549a00a5614c7f2210b6eebe5248c58de208f3f4ceeabcb352692177c14a152ccd06c53a3ba4fe55804e2647c13deac399b779fc05f3e0eed1055bad10784eca13ccf82111d619a178d2537b068c837179b22dfe86553151329cb7633343fbe8a54d8ea830b9e108b5a5fef9bcfcdfe5800f6e6e8e386c7c0bd193fbe89c68281eb0b1c7c775b68e47960c0013ddb5c3a962181c07adf71d6912d1ef4ce9a221a247c8a9775e1a0ad36993de9c7f4ac49&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.845614001Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408, Name: checkpoints/1408, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1408?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4152060c2d7d8deb34ee009208f1099b562e048647988a4930e6c0e89878c98ed405c1a0521a13dc19ce1446720f2026b2fcbde140632baa286857f078112abfcdbe61be840d5c516cabbe7b4957e42deb405f302603144935b9857b45ec552680edb6f40468ab0dfcb01ae2213d1e8f200c1cf2659950279a2d6e02194f3db1ff475e3160708f1dd4ca362a3dd10a1becbaf76c0b7c90e615ef62645e64fd80b8e17628bc10afbbfa00e8a9c30588b0add42fbc294f3db4f3b795c7ec17e798d67e2d66e204e759ab0d9d4c865f2cc4bd6ba8ff964292d95c6c209e426e1c752d568666a8a0da0b26a97f1e8ac8aca10472747f1ad5c462e9ab03b83852cf8d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.845319066Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default, Name: checkpoints/2624/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2624/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=379d7fdffe1b08c09031daf6da5892998e117897b3d2f185cacee47428229f44d068592a7e1f8fe161d672cac5a28fc19b4a20cc27c4e91782578e31aee17cb4f65509ba36d86aad6e9dd2a816f3cfdedded4023cb186c914424acdc7b664949c12b290ba3293956a121710af92ad57871a5a822226c9e15c8bad7fc4941248ace48fd4204e0b616acc611b086e809dae5aa0c633059e30d0bca1450855825e939076ef543752760a76dd1d76ada4f5cc82a1fb0925dea9b4131a8d79cee0fdbc5f1f9117c58456d54d31e6d5a390dfd29cfad22f1af035078bd518725a3ca6e0197f8b221d133de1c0f3f25dc611f1ff6dc49ebe355eb2a5d038954456b2d2f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.84668077Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d, Name: checkpoints/640/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/640/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=1a432c6f3918b8b6ffc84a0b1d805b901c4320abe7c289ff25e5c47b48604c70ae4493b2e069482f73d4bc61b59fd3a22da7f001c865eb632c4dce2b8ef6f2f7e0a215a05d41600dad17b120c5dbda71e6ad1ca12e9efe4b7bf2293133a5274675dceac5574490c48a409158f8a598ec25d1c0975b809262d24fc2908fba3996d83b47d5e19a92d89ee874f6bdb168b1c35b9ff7adfbd529cf72415c398e6d1deb5404d3e85473b2ef1ba1342a01d26123b1b6faec1a8956a71a48b677b9bb64a164ff81ad37efe7e70fc818207853ba5f98b8e2064722e7e430be3273c191f7aa3e5def2f8123da756863c9836584a4ce4bf630cda1dddea807bd33e8c6d532&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.846663768Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088, Name: checkpoints/1088, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1088?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=233a2ffc49ae5f06c433cb1ab09b306c0052c4a4eb93f35b766b127537e2fa9d4b2b31c3555d98ebc738f74c7015e894eeb986dfa4e7dbae7a3ace27a60fbdb874a328f280d2239e95198fc55b2099850e0600bd4729f87e95a7d9ae6319835408c3e2d0c18055e03c4b871eec8db2c140db8826f66fb7fb679eb690d10c01e5d830dbacd27a120d4193d51295f617593444a6ebba9ae5a5c513900e5de521d9a200c45716d877a7f684d02d986f82a50b2f839401381d811cc1b7313f4ede007468194f5eda7075cf4e7a8d3386c4f60359b28cf8ce97c8d9eb1206df4be25331afbc7724f4e1e2e6d9d6ce608506a49e3871eb2c69503eb4309c98085f07a8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.846710302Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0, Name: checkpoints/2560/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2560/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=8f705e64d0a932ba08ea757d50cb4661a534ddf17a286a37565e500ea5c16fa78fc7fb7b470668cd868a5b328da6683d4ae9a472eb45125b4da5b9a9097d79f72cfc047af7b15a38c1d1979ae7ec502250761b355a51c9f130448383a50c52ba01e07bf79b08adbb459fd2512abe297e42c19543e17141e1fd4e840e2485318bc8a0fbcf612554afc64ea6106782fc00bee925096d622a455083993ad2bbc3a81a5205d83e47d41c5d4e629a2b7fb0c410a5f19fe56fb0b4a1f66e58b400f0ba81564e42efd57031262a249d38878812191a5d6094ddc5915e2d3996b8297c429828969cffee33321aeaf124fa08237dc41329c925940b0d432dbb3398ddd5dd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.846926272Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/array_metadatas, Name: checkpoints/4352/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4352/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9caac2ef0bd0d627480d2bfa514f9d7d08a22230ca625d32a934367f630518f980cc86b375b4288d9fbf1fd9b8df645dc602a3e46eacca53b4ea8dc3d99d7c921f602813c253e656aea2e8a7fe573ade0445799875992fe6b01826372b9f44bf3da905e8db649457a1e12d5f95d7ffd3321443110d6e98f4e2cb88b9bb83f6b244f092cf457f5e65a2965ee56c7841d143fb6dfae9d647b8c3616c532e497f6dbbd98915009b0b985d2554d911ed59530ae360de9175362ffaaaf75d7a6fb32a6c3033113eb5f62c97fe4b3fe4a430158c46025ffbf2cf4f19a5ed253eead9a32d4311cd04984ce17fbaa09d5d43decc4c7d446a8991377a19df46b0f00952c3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.84697402Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/d, Name: checkpoints/2944/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2944/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=97ef35b1d8cf48ec490525c63b92065b5cdebc518d35b967d4655045c82de9a70e27c4665b77db5267eb8f79fb5088d99be3d0a851d9a8d1438d83874db5f949f56fa33558e4822a6039cfab58ec3b1f3ce1f2fe017eb18a1393e13e617319522ff239a79d2b57bdeb04261c48fb7874e528ff9b8e68db1e7345be302721884b913649adb19b66e0e06b7328a275e60612a3b8a19d0830825544c33cf5f52b1bcc14234cd0b8f333bd62601b154c16536d5a284f3639c97fb3ce7fb38ba12c6ed2d464f5030c642d0208a0b362fb3200ab757c532cfa52f4701d5c7a64f76d7379addd4a34810dd38ac6996440bd0df87d30b7bca391c71111cef753d8a7099f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.846991538Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0, Name: checkpoints/2112/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2112/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4dd3d2982a299ff3746076a110b3c5dfa19198ed59ee20513b2f18aaaf149818139f400b5a287489d8f86f411827e9cc0d25099c34dbf40155efece3b78a9718358693ffbe30e0607833dd8d1d2c1d658ecc933b3794ef05be8d6f4be7df173f519368ff1ddc2df8f2e6ec251ad14d9992fad2cc31b014779757af02ba1a63f417fa54bec6dbfdeda8dd31f30a7bd003c890aff4610b9552ed553d98062a85fb3ebf8813a6c9ea05ca8da0ce9aa2e3ffac92afca0ef8a1944970cd22bddd78f73dba785de6b146deeb59ad9735b16e9ca56a66fd89e4a816e9499b1235114754728657bc51b198ec8357b4ef685a4a72f133a16212af0feead6215efc7afb38d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.847018254Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0, Name: checkpoints/6016/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/6016/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=25b950fcd5a988955b48839e214a66c20815270ca9b3a049efb23b714602561249661848f2df35612f8e5a6ab5f34da920193a80ce47ed63baaf49f1d4c3af894e184a6a978336a6460aac83fd2d1e31a5777c6c56dfa4ee7c5424545fc4acacdb5115efa9d6b41629b1358ab3ce309e6be35bd059d3f96441e7f18680fbbd3ff2c0f1b76cef861918a54881c874e1fdb497b93706d8dde34422e10988bb0ece58945c6ecbfe90a8dfb53cbd3d8c06629361e098bdfcda00d40371996f060f30eb0451bc24e9c00558d6d2690ba3b0012db55bd401a4e56127a5cf3cb89ca39cf27e54711cc6d88d00d5f0c2b76702c9b402b6ad418637cbbca833045f678ed8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.847228351Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904, Name: checkpoints/3904, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3904?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=660d3515d1b65e0e5294bf3a7e94fb1b6bae358bb6ce1ebda72a527965f345609dc2903d450c79545b22cec4a5e8e130cc644a6dfbddcf80c148ab98c8200a5b220479764ce8cb90a8355cbe10533f8234c812cbffe94199366860d0753adce534ccca16686c280bd270ee2de0a8c6426677b5374344e82650b9d881b7bc76426e23927d35c51ad6db92052a03aa8036f42d67d4c5454df75756ae9b041818ea9afaafadef104dd4ab879110b6e50f80a913ddb38a840904bf62353f7e6d65215b720877bf4d373e5e4a5bd0d6252841c07baa0461b10f1a50df1fa20c39943c4225919f3a7bb944a54015083cd57f2d227b5da03d57136192aaf740064dca83&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.847278397Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0, Name: checkpoints/1920/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1920/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9fd91433a21a945ac8596b10a4ba66d14db733cd1ac6f6f42f8de39ad630e014bdc408549748cbb3667c14f2ea8326d36111e0fe13f9baedf3891ade491683d984b35c03f2cfae9074d586b6c3b3f26cbf75fe621570f257b749a5ad552a85e3da09a731611f01d8d2dcbdcd6b5c53f23ad4a53918781c0ad7549bca2f40a06059e6aebb108cbe7d5c8543e35796fac0ad949f3e3cc44f1823eb05ee6cf14db09827f4a7da43ea7943ea7e53ccd750f50fdb30d3743a0529b12e1fd81004f695edf60de50954c4abf7572ac9153202e8dcb256e33cc07076cb00cfd01149d5c9a1bb8e078f288336b8d512fc9ff4b35781dd919b053b8ae490cc90aba3687532&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.847300904Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/d, Name: checkpoints/4224/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4224/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a8f751e64be2d8eda5fd00bb5fb6d763b7ec278421fcdd1afd3024b014b0a6c7682b2ec1ae976f6045d792f25e62fc58fbc9bf74731f3f48dcd8ee1104f57d2e17833e0994e36ceb2c8db000b32bf38998f7ce0a2bfc64a069c90ede1520103307ba132bf89542a9525353cb764de00339bfa3b756d8f6cc64e08c5c81ad971b775a4c856c120da1f9c889dd9645def8d150bd32fba9f7837f28abc0790be88a11108bd9de6ee275766d1640ce8fc7c97383f1ba1f96a1259de6e014713fb03764200fff9466adc6968929562da8ac465097a7507887f74c65c7cf87d06f1495d4ea8e1cdcccd04a67e7d1d6c071a5caaa146ff7afb77465dafcea85b2f9455f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.847314211Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0, Name: checkpoints/5184/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5184/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a0390038eaa5a51e15a1f8739a134d9fe27d2ee1446dd9dd5d0ba30c8e5632e40cb7eafc7c7678fb0b93ffa46f24f7acf7f7d044d4cf4b0e019020a7238e0aef9b744f40399a116127ec906d86d6bfb2619b557fc8aaca5e69853d5b809e2a604147fec8eb64bc0c9f83fc7f9a644b04cebb03847320c33f8276b0ec1b59aae0775fc7fa76a8302fd8db62ef37aca99fce7f358de874012e565741d9178890ddb026f8227a9e5b7bb293dc275947f35caafc4d4a5a4bceefff42c84ddcfdc9a603463faf1e375cd8954566ddf4b68cfe342aacf13237d3f089318669b4957c10526fd06a9909731d8d012f6d0ba00b518f91a91fd49af839303dd74c58ddaa7d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.847378549Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d, Name: checkpoints/3648/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3648/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0c72ce450c873b7b7c168dd988662e84486c49696bb8c24315988eba043b56d3ce66197d94af5c14cfc4c2be5156521e7351cf668f4141126a48720dd1537f61baac6d387c326c022237a71aaa7063120749dc77b678948607c0862d80ca8d48a5bd440448cdb6f9d40354e727106d720a15b407669af95dd4186b27dbe928d2eb4d1b7be6f504265184711d97932c0f645a4cb0e275478c7a04f61f0303350574a5b3ee1967b8a1030538154ba55a1a6ab8489d4b744a335964b331adb068f9eeccd85085934ff81e326de5f87943937cbd7b2523bcd11ee95d4e9561644c86576af51dd137cbe29e1ce5e4416d0aca37c0c71fec7b1c39862541f90bdc88b5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.847292931Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d, Name: checkpoints/3200/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3200/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=1394218087f3694be222fc2f4c59e48aa99cff47a269a07b725e652fca787d8f589afc4ab07b97f26299cbb39f5c03ebaec725a924bfae304e1931acab47cba4a4beac526f6f274e234bf293aebb45362bcbb4b9a40322c790211b6ada06805447942ffa92f8ab468e45355aba1779f84c0d1c003eeaa52a33188e7462283fae773e1308db2fda11d9ed65942d333a4b03d17ac7d45d7ea04e5e57ae715c5e33ee39eb9fc26b1af526c7c4932f6d646faea097738728d2519835860b664ffb2398a6e1b762ab7cfc47ec734ae56df5a07148462aed99fa94585f99e8e1c3c0c135df2c17a920eed7192b8d1b7ddeece482cbd6a37fb46fa50ddc9acbcecf00c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.847526513Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d, Name: checkpoints/5568/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5568/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4c3e713e6efa722185112a53c336d7efc0ad2f445e202648dd62a854213dfa4b6e9da093d6cf37019a2b50a37670fd2d657783f0ae249876e74f8b91351d8100bbbd649b79d31e0b69a2d73e45098379279783f59a6bb411ad3aa8097157b54e0c009b7e7614e502bc4125be3066c0262ae0fd79fc92982ccd3e99fe129d7ea095c7581ab688595d8285944defa2228cdfe923067263a2c3c39289d77ebf252d306ee447ba01692a4a7966c8aa8e80ee162301b110e87afc7528565a6f2cee5b2fa255012d37d255d4c770cf39dc02a927fe83d53ff78bda6bd5e247857374db7435964c25825c3a29349e2b812a26e6090c49e88c3f33f53a7f3f7dd963cc5b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.847572435Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d, Name: checkpoints/2304/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2304/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=6c125e820c8f4e39134b5c8d872df983c7e89980322ad305fea9aa7b9f969c2c313c1c56903242edb9f7de0532579f495ec5c2b5428ab50459dcc57f42f2ebc442f3d28796333248da33b38ec6e20d0a7c38776fc2ea9aae755b1d1fa18d2435d7487051da59d4cfa3a9f3efda422e9ba79dbf6f6af4257842e6f9d9c7fd823c472997afec9bcd84e2fbd88d71e1d135f3d0fcc79ad968cc27accf0b0fc1c2b391c031162f5281347142d61440556e82c155a0e9bc47bfaa068c312df3af35cc05bf319afb164979ca4ac9e0e567e843e00d84df6019394a7be7e66e9d5d17e521bdc7c1b3b90a587980ef225e672670495a2495a4e6e6acdf28d9a8fdaee55b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.84760362Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/array_metadatas, Name: checkpoints/5440/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5440/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2e353310ba8400277321e38f05c78d7c424ee4d83ae8de89ca44b1ba5caec2217383e2657a088a70ba9017a93771f4f90654d3bf27beceb97956d3920cf343a8a8a33c301fc28e905b8c4d2075fc873bc105b8efc6d5da54b01b43b6ac186dbfd071b7680f2ffa96d69ce9e42a2435ed3a5d63ad86d41f841ffb99cf870ef92e790c048bf3b0ca72895b3a889547410b1d4b5f8ee4175ed03a7b1e29bce9cb9df08c014a8d30d2797cfe97cf0971a7ae4a5876a744ac31d556e95ec5a049252e1cb026cc9cd402fea334a6e3b15b2599ae8343724a5adb7a8238c28f82b68251349e2e6da53e94c67cfc2d556c7bf8fd341f5cfe6b77d46e35d5d8a3f3c6370a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.847621465Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328, Name: checkpoints/3328, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3328?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=7aedbd003f53d3e0500b5749625ff8a4031ae495250c341d7f014833fa298f96c22331719ff2e7488b7fce807f1a1d765e1980ad62b466ff7a870d931e691bab515b065fa3abdb638e2e9e3e914bf902a4b17a7bcd8db10e88771c6adac28a86c22dc5148bf512eb6b412f27f9e0a3fc9a78b1e062e69c83f8073456f7fd164e7bc287b484047385aa1cef9c5f691a1bf77c0c5e1f63c3552e57bd26ca969fa9ee727dd740741ffba5cefd017a6e4ff60d1a437590fd03c566acbf07582d7c7d20af18cb741b72016a37e6f20acebf3971eb91585b72302b1cb7e7e8a149ada7b6b4eb4a78504d4396603a6ed7d51138b71a68f248303b3ae374253e1201861b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.847654167Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints, Name: checkpoints, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2e774c83db81c6ba2d409920897a4d90a053d36c4b32a6713402f89a0547ea0191fb48d060852d4be52af038f05e5121fd309142f67b0f3a6987143b06f14aab390b77e925eb697f75fdaf026c58124634c8f38f7d6a83246f06ad8f40f26ee88c89bd5ef928296c069f9b12137c9c789aefdf82c211a6b5cd875e6c4a2ab94de34ed398afcd1fd9bd6b6c6523e5c944124c18bc0b2b2c630649fe0dbb612b329a97e84d099cd617d73255c11fd12ef3b0dd8887c0ccc04d6bb2feff405efa9ac6f1213ac718d7b9289f96176c4609a7f570b11270f2deaab26cb8177286847d40fb4c4f78e60cebc9179bb010e28ef2f15e08fe704a9e10c3690c0c614d2256&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.847577857Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/d, Name: checkpoints/3712/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3712/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2b843099bffd72e44511de09bd2e867a1eece4754f94ef3d139a01d0ef14f811f1240dd466a631643f4226d2731a30daee899d9d5a9376446845d28addf5a68633d9278ceee7b522b48dbe53a11407810cc8b43daafea11f53b1dc88b0eba171e86f6351e0c6364a71677535a7d8a474bb68f06e1d404d2c0d5fdc306794d68238eb28ef3d03ecfe2abea9fbdb2a2e383ff21f6ff2cf9d352648f41ef58621daad0f1a04139c1e8d0f29e20b1204d838a66649fb6be94f21517b47a46fef9dbb7f359641466cb8d9c7ad0bc397e1a9b28a0628b4db59c162535b8ea34badfa3a435d64e1e7d94b3ff5ae3da3c6fc7280d370e84b7e274a005fa3c7ec4ef034e3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.847843523Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/array_metadatas, Name: checkpoints/4544/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4544/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9edd129b8e64bab4c3691cc64f7b406dfe525d93a7fac7265b419dd8d634d368fbce27d7bdb6c621d1ebb1da863f95152a62ae4d41f10a7fc2d06e02089f1ad413bb2612bfd01c848dfd44f13b3ec2deb88cbda951a99cf91fc6b9785245d5fa0b9cc0179d81d0309e0119d2894365afbec4f923c820fd7696733fa79f56693bfa11e004dc246596d6a557ace53e65b1f6ab6401bb3f097a6c1b9367dfb1fb3f393f07591d8743817e293db4d9638e5532b07670e7b5838d7f4be740184b74fdad02f224a2039cf0d634cb411b7e96138696c0f0109156673cb8e253e5b1c160e244468126eb5ef366f951479042352858cf0715ea028c865d15b34f2f07eee9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.847890931Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/array_metadatas, Name: checkpoints/5760/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5760/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=60afbfd75fe085ea07b3c5364af911108cbdd7073fd03902203c621ca691cbac64af055ce134aeb02dc9e818ebee11952d5cc157666fd2a28f12f498b5d4da9c6f39e68d30197a7cce272f6f9314222ac145cc46802e3e4eaa199d6de7431cc22c90dc40716698d56f42b222e209c1b0356eea89483c3d27128e7b614008f427f890bae9567a276ee53b668b9a62be8df3d30b05b328fa2af189c07b40ddc1e8f728a0084654ccc95374c1406451a4e6e3755ae92b1fddfc8b1381f065590f99a0769f75fe6a6b02d1e8471b288a77f1d44dbfa1fe619b40af6badfd2e55d686dec223e25c69ad95b2de1cbee8b64b49f4690e8234f9bbb4bf75f3c5698a5ff1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.847915673Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0, Name: checkpoints/3648/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3648/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=6efbf3e6973e6e2897f3c341ab2b1e9da7e213637e917927e3b283d1b04cc9eda546ffd4337802df8ff569d030d9c8dc282e0bd43c70b0f7f8b63b86dded9acd7dc9d959ca87eef2687598d04042b5db4c59079defc382b015c9e477cc6b7456a4dd3565975f3c0e4cccb35b9b2015f73dea45225c3c9fe3b79d2c0959fdd615db67638ee282575b390902a53c8f9248cc1240477aadca61f68baa392ab2d45191d90da3b6a3121b48e017534af3ea1541685ff5b6987e6569d10f659b98cfb4ee76d4275a30f8a2f9bd52159331493a2affcb47f0a20a3e6eb6a73e35b0f5c108be8f373980df8a7ab73e8db0356bcc2efc2d9db563ec0ca91090bb363e79a4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.847930986Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240, Name: checkpoints/2240, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2240?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a9654cbfcddccfe73b8b496a1c4fa1291d9a80ce68197e386d6a01652aad5d9945665d3008df5f6bbc8893d3f592e4f95dd87c52627c59dd02858db32da8db0044f0fb5b3d7f5c9d9ddf44437923fe4f7b3e8708c3903dac45bef4754a778c053c857dbaa6fa7fd5e54cafd062b968e9e9918a9cb1f89e85af9acbaf8ad62425ffe0c16f8a495e2766231180368781703b12b9fd7d112f52a2a32abf38fa985dc1f3bd00401051e58cff74f6b52306dc53caa62b1b5817612aed85b7002d7025734a07b10f5d799160b7ece9558ba2ed7ecadeced15224b4d455d7f68da3faebed074a8ae128af9bff01bf0c0c1a059e5de3c26081b857031cf133494fb9e8ab&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.847969601Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/array_metadatas, Name: checkpoints/4416/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4416/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=1b53b5c32622921278d9ef3e0dee1450ca72a7803290e940bec25b7f67b35c0a8ae7c45d79a2d6efc39c197cfc31a1a4389a0c558a33eb920053f26e3a974ffd43b63ad11b3fc62e0a9e7e92030be02760f050f1d01a5d9ffc4c15fbf456af80ed07c4349f3beafd479756241c154ed1e7c59f14c85387373fb29f421c04224c10c4ab60a388e9c1ae509c9050661d5c3b68fe61ec249c5e2b1eec4ed738efdd96871fe80eb3212d92a8a714375f05a762e683c9ebc9e5938241617d94e366036766647560a4e4c4c149d943631e487fcce60d4eb2dca8567a12251f5792bba80c88f2acc9a47ee4cbb84d564c818f3a59b06c567c11b6baac654333a2ddeff3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848096432Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d, Name: checkpoints/3008/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3008/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=53e0a2fbaf6a6b6aa2b1f9ff24c6850acabb619331acfa8d33ec5978bb810c093ce776cc0ed288b968190a0aa459c29a6b4fde1ebd4c56d481d50f205f19db1629588e97ba9c5e46e0129c88f3ae453b8f5863366c3c12a96a4b10e4104b9bbb77473a25bac40059306e9c31d51ef7ab7780b1acfd8db2f65429e762929f009e1a2c9322a639594dbfc2490925e06b33b8e58d570ae48c289e91101499c843acae4c974ee6ee7ab50c3b7a42b22597fde0e64a250cb7a9ea7ac6a7a647f6c58bc73c20d84e8e2b566619b4bc6c18224acfb862e5e478009773d1ff4dd1d269ef7ca6d5149b7ed8917df25349d3539e8cda39f2eb2a2820ef17dbc1a4acd387ba&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848001357Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default, Name: checkpoints/2752/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2752/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=719a9206829bddc992fa64394fd8c2d0fbfe442dcd5ed2afe2006107ea4ceca908563bcdee76fd315138ce83ed0072c71343fb351a2881d88c3f9438fde2e0bdb2df5cccd8fca0c37a46d88188a201f878f64457178723c2ddcb75251255ce18dc1e5251e37a0eb6b949084cb8aa5e6d67ca8b340a584a477e0df3f87f334ff05f429245c497b33fd30b14cbcc83b2feaa425e273d71953bc4ca19768cb56c9689a619f9812ed6c2e8571f6dac274c774b146e93252ac68d8ffe2bbc372c90ff1ccd74635036ce115cf0c8a46113b4241d55f82953b1b5c0c01bda51a1362cfe73d8e93fef87a619f0d3b4278d67037fdea527b1c4f318fcdcb0706ca39630ba&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848191044Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default, Name: checkpoints/5056/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5056/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9c54724c867fac76a5d9ce09b4ed5114b302551d4aee21205655a4eddb8b4f95724751d666d64c4c690bca27c97d3287cf87e5d964b6458c67e34a92a9737580188d290be57e4bacd06e1c68fdfa48699f920b5c82ea16f523890a4b70318c3e451bff27424b0502486a1fe16628d74235b9121fd167fb35d12a30b79f90332e80c892f2634d3001e78c6f27484c7d03b8f4a621a3a6c71027e581e4ca7bc057b99178cac07d180f2cd24da79b1d6c2bbeb092ce54d8977fdf4632d7d8b18b73d5b0d6126eb71f06ecd9713782abcbbc01f2033854d165c744ca652074ba3d0c6e813b64b8206bd49074903756d4b41a97d434f9af0956fdb83b748d87a989b7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.84822588Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d, Name: checkpoints/5184/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5184/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=96fade473116141dedfb6d9fa64fdb0c5b85c9e77e97fd359775e94453f9ea15bb70d018576a05d712f90fefde9fb7a59629eb94c4dd3d017fc1ebd1daf70ab9a0ea9053909e75704b247301c8d58ae1e4957d24e7c408381de9b3c0bee499ddd694d7958fe315ab9ea463117281b92afb83523ca14fa0b58ec494681781aeffd4b64df00c490c92e830427ff3f64fc6cebef19dcf83d3bfae94b1c255850a64a4a874f04eb8431a3f44c0eea4957d99193f5f27f2e87ca0bc74919df3160c1d93999c5ceabc2ce4f89819e62649fd006bf2eb9d22341d54d87a895da3f629a9749cc1805e42fb684aeebc38ee49fbffeb6d3c70f492679ce29c407ffaeaaadb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848210419Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0, Name: checkpoints/5632/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5632/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=14255c54d9215d21cd7ae328722cea4a9156196b945bd74e2ce9118adab5d18ecba80a3966a308884f82bd92c3531c07ab79840c5a48bb0f9ff76cf6b6d8618e4712b738a4a2516bfec46e79ddfdc26667ee78814d806dabc9f753e5b43b340a9f52511851ae942f7083b1b0822cad3b532b70cbc8c985ee6edf4d80a3e6450e30e2f24d232578993fef12f71dd250801f5f7a27210e0976c348c504bd7426230df1f156a6b611fd11a05a0f0f074f620ca289fa91b59fe80158bb8cb2cda320431a2d3f3f905333d80c9017b30ff223bc1ce40777063023d6ba6b7e2af5b2a49bbadf5246f217e508c573c2056b4b2e65039620788fa01c571189ce25c878b2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.84824326Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default, Name: checkpoints/3712/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3712/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=21bb3ae1df8cea150b4c2a0b800d4a5b883433880f2bc9aebcffd2056bee1520d0f3577c4f848b781b3edb20011b3416312c61a03436c44ad1b2d37a491e05f06c0efeb50e79edd4af1b97e8b203166257b72a2d8c9bf24161e37cdf8f469ac812e5bc0bfb731a916bc59f2c6766f6eed50b61c8da3a2f601dbde4c30391f8fd47965b926c3f432ed5e7e77cf175237162cb06c0bbac328d9bce876c9b37ac7b2241b2df0e48bf15c1658b83cd0b5048a6a880818ab99422e93cae46d0e1d8e9b02a1b0f5d4126735d1ecf9cbd61983d9e7791c096bfab254778d9c06093433e0fd72ebf05bbf1f0935ed60a485a727a02bfa12c3c432ebd312da1d7f35548ae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848326473Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0, Name: checkpoints/5504/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5504/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=11c441bff5fdcd4641c785d3db4337c4948b81a0830cb1cd56ea06c9167127e9665b4b4da327a5fe94320ed89c13561bcab0fd70e8f0538e691b1408f879f202f407486af19aefacec0813aee682efcf93024c6d036380586970e70d1458f2552081fbec3e6f7fbd4dffff93c7ea354868cf362bfb9fb53f4d98e15fa022e5ed4820893eaaaa04d3c5b7414329ba5b6f67777719563066334ebf0149a01f886c5abf8a4dbf35ecb7ce26efba2f736776a6be4a1616133176be8b8777eb4658444238916b6bbfdcce8c47a7778595dea3deb39859622b8844994de243b846e6b583dbceee6cc465d8e9f053ad558e7c7cfd0d927a2f649af63da7d4498004602f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848221061Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default, Name: checkpoints/3968/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3968/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=185131808b92d31d0b0fd35a7dbe4c7e296311b6347d852aa9c87839e8f3bd84301ef534d95b28c3704a6b3c6de5af022111fac04de61d47b1c2794961d3f94435f4275c22fb73f203a738edf14e70ec0d0f4d81c996bdaf8f4b926d9c6e067746a0bea9ac288b5bbbf7d7d152512cd69d2b5ac8d7198d0a801b3ce4673d40beb6e9d293dcde9b9b0ca112f23cb385831ba4e910887228357bc709b93c00742c726a9bc98144c5dc7119f8f72394b09750242d9224c1bd26fa17d1923e36bea568f7789397bc9c7150a3e029156188dca733a6468176210f711b851011bc45200868124d276a310dc739d64427508ebe14473966dde0dfc65ddd558530b8afec&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848452339Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/array_metadatas, Name: checkpoints/1664/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1664/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0638e1f05ee9384d57b76c85785188dc5f26b711348a579c21c0cad42193bfdbf3f5ac67b5e90a72484d352e96c7f3c10b3c9c4d88d20fb626ef4b1d5d97756f0f3c79b59181581f22c53ae328db83fc67461af2bbb81ea1ac1ef155b5993ac7296d416b5073e72e8254398d013f5cf5f6b9f94250808d2f44a9e0f83b326b26a0fba66bb1eef1be6c2bc20b4e7f23dc143fb2f47b505390cb93b34216787a972661f862065de67e11d34fa27ce63f52aa4708c91343755507d2ff3b89b4aad4355c4d6947ac5e2aa3797adf81254ced1dfda11a47a060c7aec814da95343a5628e52510300d92190b4f1b93fca6932482be707458aa49e8ec295d9f40f8d40f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848468902Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352, Name: checkpoints/4352, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4352?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=823d269a129a32dba4e656f5c8eb12ad77c7b9257ea58c411780f577e80fd60c878ae4c2a9fc96a3cc28797e3408f3584c967e37b53b071d29168aac72cc973e418eea1a85c8b4b55d2fe35e94e300c9a61a83a1af964c6e15994ed856fd3e013ba4e836ebda117acdcac08a486d149a5091b1f52bfacb36add38a70354612d3b907a69327c31e7ff8cd143224f3eb431a9628be443d9e7c2dc4bf1e18252fa22f8794d987e55da7ff437396045c03727f1b0ca407108a78492266a989f89c5377096e9bd58f01e605205dd1cb284629dbeaf939cf064a8471c64cf56657cbb228a6f5da382c4eff13aabd9b94e3149b38868ca260108f9f5a754ef1cf428462&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848472646Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0, Name: checkpoints/4800/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4800/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=769b8ad10f847761007a60ed5e98c1e4567fb50cb14e639d894439a1bd0b313b812a57c624d49edd61d2533fc12274f23a321af6c6ffb5a874e2cd254ecf1f33e69379ebcea5217de4d41145785714143175a4848a90da30aa612e057afb3eebb633811b4ca8578b2840223850a82dc786531a28ec1246e9227925b5f1b4c36c90b1de155926bd0e52e08ee226994302ce0e90b96b918dc3bb5f898335ca627697e8aa7ba92589d6793d52caa89ff05b2b6c23d8bce9fa81876124adf0c377255ad5573c923800a8485cf98e5bcffb5470639a65dda255edcc792477e425589d0b6f95c983a28a72839568d812bf12a2b43dd3fb2b4adedef65405d331125816&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848509951Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/d, Name: checkpoints/2688/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2688/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=1c7fb77ae572d1ad595195a96155608ec2d3e523446456e69b052210325c9fffcdd73299d83f4263d81c12f27b26fa1ac17de456c8d5885d105da65d09763b63206e5b193f9d2ff0be65438b4aa8c0d3f16340e9deb18299fa5453e3c2a8dc95a399ca9afb5b77f5ff4a72f3fb029054364df2f1b3464a1cbdd68b2ab551d13fcdf36b1f572d53196a9c8bb33042e590778405ed6fbcae8f48fd8f6ea7237e85d6ba35a35943e64f6332468ad68ff24a22a9299c4d055df151d5cdc852afb30aa44b46c1e45e2986046b5b3fdd1fb75bec3e81ffbae073ed585d11609f31f834ef1ac9c2f76b5d95408eb018da438cbc067f8a22b78282a73e31fb8dec0e92a2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848525365Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224, Name: checkpoints/4224, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4224?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=962bfe5801fbcd94ceb2f39e950908649c54bf104034f64bb54319509217737d06dda7138ff760f42fc279bba9b8fd62e521b4ffd7cd45ad999bf8b38259f21650f72d4b5564aa3d722bcd4f627364556fd6079b753a6fe228b4e5ff7e999b139c7a367ce727338ab9a501607a45e18374c3bcb61c342b2e3add74aeb6c6ef99797cfbb04e87f31f517a39886cb3264ff01df9e339b6edec86bac231979cc0e4b6c4b90c3927c04e09f4d93e65e2745c9ada87301cea441b08b54321198b40aa0bee10f239dcf7c2394d3210ae7d183527cc13ded3db1ecc9925c25569c298cc1902573a11567c25cb8265742b0c20dd778249d121660bf3cff2237dec0fcb7e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.847882106Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392, Name: checkpoints/3392, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3392?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=c28e86723c1c39744c1dc624feb43dc467fbecd4130678319751fdbbfa5e15f1ad2622274635bf81a437ae450522bc29780bf2b2657a0c0a21ad0038555b905ef74ecc8540bd83a942a5d736b98c178dd2f1404031e19e5355022c4682c931ae45d02e93370468ed231978bc7d7ed64f0091b296b83718fe982f92143e70fa4712156a2ee77b30bd85fa857f5364e02f4a07dfb4411e18c73c6d744f07cf2bfbe0d081c8347b2479858e16e7f3ffc9b2011b510b5d47a5e7e11efad86d1146eea012c21ee933ba908f286d954fc1b0858c701520e8e5bfefb913179e4508d1158c1564e71af9f181b900369cd9cd50b14766b19a65e7910f579ffa7cfe8c9acb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848692081Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default, Name: checkpoints/2688/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2688/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=025b27b24c39a8039195f4d15962fad820e685cb460695dd5a258deedc36f1885b56af14719644f45aa63bb35aa228deb9d40673b21da1bd57c59976ab6647e0f4ea4b671f98a8d172ac20dbdb9c243dba9dec6416d614d2a131db9a880d86bf8cf1390c4bc2bf536ef749c1d5b3b3b638fe7b7ca96f6927a34cd0389410b669067c99a8dce2b2e3e9eb7ea7d12b6594610d6c2b14427b4d939e1fcdc5bcf6bacd2a6e62dc4e892cee5a584d00dffc41ffa90d28eb4d57f58c1b14bb9ec2893e7d2991cc244ff07c90a6cc3768f83dbe227057a92f369db3935b684f7e825bec0bacea10fa870a9ccf0b80a9bb728a43fc5fee8d8e04b762d673ab5262306b55&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848712145Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d, Name: checkpoints/5504/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5504/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=354b20932d2949a77851b4ebd79530bce4e125ad1e0da1e6163c94dce9a785b4bfe577880ecac267317035e15b5a1d201b1ea1c2bca725f963bb6ce790ffd215c996ec02f9271236440b5096869366cce4abca0d324110b9b92e0bec0acc05f688dde005e9bf3949c2f64c23e2790b33f338d7e62261d77faed4f8aee200295a693345adadeefaa274d2be9cef9cd04364444e5595f6e27019f48ee1be59c3549ece86c80787be9388efef55b26b64a581393390bb7b7cb49d479076f7486e6c517f44de18a3e794cde118e45bca187ea7ba7a1d1c2dcd192b0b610c00e7c9497009cccb33327c026620410ce554c7bdbef68a3695a2a287f173bfb1ff20e119&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848746015Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/d, Name: checkpoints/4032/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4032/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=8014221f8bdd11c225e97e3e6799f1f3d4a5a11d675b0b3a7873d6f7b4ec9e6d4d076d37a226cfda081c7876d06c7cf424da6b28e4459e3fba7bf6a83e66994b19d3419fbfde200cf30e8873aca80e3ddbeb4a5b3f0429a16a9dfcc486db4a6311c343c381a811ee6a4dd1b626ecc63eea16198846112853d6436e06a9a5fc8343c07068a49b71141985161306ee889a48317b678d896d7249579dfbb514e7d139794d679798e0000b9802ac14a0f525df36bc7b9d5df2214b73c3c353cd30d9a69cb841546c30b4b50f2cdb16095c8aa77bfc73cd2a928dbd8af9f552f265e5cef28232f94e53a04f951b8465b071f1c4cd671d7fdf67354753670be17369af&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848754493Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816, Name: checkpoints/2816, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2816?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=91976e524d01e5d194b2efc42364e2c29a1042e0838d48457f6bf7d9cac9871457a3ea9de9ff25ce85a2f4940ed77c0bd3c38c51d081db58d8d9d439011c1570df5d7b28bf4f9460037f0b1536987a3ad96670dac85782223deb73f3958fccbbf9708491cdcad94cdd264b53d3c599d90c78eb83a1bb3123053ae577db607e286a43859e6b9bf9a3b40b982a7d30042a6c45a454978a5d6edaeb2fe60e7fc610dc1ef367e25529bc899d11bf5dd7781653a12898e6b0db41193d768f3e7771ea4f1992447b718640033ad62e26e4558b5bf77747e31036fe87a462cef222ffc9e18424a4ef4acaf1abcaabfec83290e96ebcec52fbf34b2568f79ed5a9dde63d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848807897Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768, Name: checkpoints/768, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/768?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a102c57a687391c3a6f51a1eca8b7f0d60a4c670a731c3245c29a782854684fc5fddc4c933fb4a2e1a7d893c4204c1965e74c27acdeb28816e03f625271a29a50e2df7539a061715a07c7d8927acd2f91880191694ca94cda81a841b88ee326102425395b79209ddb29fb062809f51ca56cab02dee6f19844c01dd728d1f157c2d7b10c91e164999ea51f11cbf7d5a44295d08691f7cc8794ba6174769c3a61f8f0ac9f9e69f3c4f0c8f43c0b313ea9620316c9f5e731cacc75edf49b454364b7c394853c1454669bdb2f740e2225a359142fd100bf82e14e48a35c700b414eada1323daffdf861449a5b1de60cbe02a9259518fce32216975e33868011664c2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848728489Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/array_metadatas, Name: checkpoints/6080/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/6080/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=77b20c157fe23707489ffcb517d588759f53404836affdb825cc212786b796a89c0e4826fc53162f564f4dc468fba34fc6090b9bf4164034b384387c82c86dbe7b0282aa9249d1d8a01c91ab2d992b8faa1650db4ee64533c803980c2cf5290aaf8cff876173d12b97b257ce788dc0fc53d6d51a0d57b3d04a4d0569ca527a72fcbc491b21727201be189f7599a3a41db7a93dded2d35533a09d69c906e922367ce2d9174396fc4096fa2c8f0e381e961d0c36f098733c397a7c77017d1ecfc164314d2c4cf5a1d32d94b7a0ad95abf8c3b593724b5c08b96048ad3fa6a54d47cd52d59c89acb8144da55001ae3591aef317d706dae978a3ed03d1c7106d8083&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848703537Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/array_metadatas, Name: checkpoints/3072/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3072/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=5ee48e370819d6036dc59ea92faed8471b60a7668e87a32dc09eed406abe02df93e7d91bf8810207fc7f1a15e4a2fd9d6f1c4f2e4a34c9f7fb77a0f126ada8b8ca5fac9a1749540ce4d54cd934dcd2aa101dec84f02b3606423805d2c0a3a205116866b28233345d3e1e91a4fd686804c61db4fd79681754571dd99787c45917334da635585911cec44623c7fca8fd458b9e89b5d515a8adc70e6840981b7cf082389e2f9a8ff56b3259e5ed3f0323ed5e4934997d64b2a1c298bbb899ae4052d3e1dfaa9cad803b1b12526a63d815041695767bc9f8ab5f7ead038b932acc3a598760c8df266eadd1d4255292cd0ad05af92f9d4e4a6f3cde7dce4b451afbb8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848139778Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d, Name: checkpoints/192/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/192/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=12f19ea6fb1786e53976c22303246281bab5de3d3e4fb5fca944aea938341b998fbd80330435dc267713f2d195140384710ea89973316d74a8d7deabdb072a0b399dd68725ad7ec2d56fe9fc0968bcc365bb5855a39ce515aa4c7400131c8794bc32d2bbb6cab8bc842edb1e2535107ac6202a4c2b9f1388f4206c839e76b4505d1138c983b1686d8eac2e7e9c4bd1c42f545f0d54a69ebc674774bb8f75e02a9a7c8c27964f36ec4b93b45751b48f17b7e1557882ef9315fb98cb0a197d3cdc550ce3d9266638e349b49a053d6f3c82b5e0d02968875d24ceeaf85d1d69b46dd6e7f772f0392797bd9b9dacf3649e45f39eb327f63a2d60703e5eda4ea564e1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848908939Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d, Name: checkpoints/1280/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1280/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=37c68f4056c671e45df64aae104511b7c74ede8d0d312741870d19cd22283cb052a97cff73660e5f05ccd3b0ea2a257c46193094f5bac107e2877147dfa753a9b532705d7ca4d142768f72ddc61f15602b9308bceb4a49853063e4bdab4427b612fc0890b08bb3b87078928f959d7791ffd99bb3a71da6e449b814516ac54d459e0d99c76d32b5cadb2933d5f96ec8d7445919db69de5bb3e3482fda0311a2c619f49a41cb47144e5148ec69b131f7ab919f9e7735150c7b7994e130541dbabbfa9e2780edcfa4a9a19fc27f36e9846be50c36dc2e5921fe9e9cbca34a4bbc80c0060bf9ecfb1d989dfa0fba136d378b2c3565981067d5f2a7a53b1243ff5680&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848915466Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/d, Name: checkpoints/1664/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1664/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4dc23c64576704d3fd511167f75e12c9bbb35e278b45fe65c6f1d29d50d1a494d8e6064de793086e17bcde03919985c1ed7a2a481237900ec49bc54a9803d51234e8b0c1d5470beef6cb10a6bc68944b9b1468c194dddcc28ebb03babec9c3955542760dcd1de27228812412e7d7f48285efb744faed8cafae8a9c1fd71cb79ecf22760a690157e86ef4fe55e906b1bb273485627f6aa193b75f673d46f87739547e621bf9caeb8c861187d971d53ca206beccd9a66926957645cb496698afef46223a96bc5fd32cf55a747660bca9391f2069ad38d71d0c4fd1e0bdd2a16fff089e098434e2ab5bbd247412a531fd58e1475b11b2a90352fba6ca2235db650f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848928639Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/d, Name: checkpoints/2624/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2624/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2125f80871e44a55e81b68a1b2a88e73b54fb0eee906b2cf3b3dc180736e1345d3d7df35bf88a2c13bfd37aa169da7ac87088b8773e47a4fb5303824aa2734a07423b1ee550424bdf4184bf507b070c85d2fe7b11f970f8d76297188c5a4119b35693081f7c39188c62173924b8704d291c299d0957deede941a315a7781399f4088260c8b7209815ce0519b4e9a613bb5b42937ca947ee6ffb8b700043da795e2c8ad1c899211d618d8007e3ebcec127e485defc96a4ba54c928f616e3b3dfa04f0354b2163b04c723e4e5e29a776b3e6dbee314eb69710071b51a996c3349257d5f1d2d6504ffd06bfddc451e8be2989ccf713e835ac0f1a67b4c5865343e3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848981491Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d, Name: checkpoints/1664/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1664/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=ad640089eb5ba4c453565b6397c6407f986b9df3e4ec969808fdd4b2b64e23acbc0913a877a927af7a51654adbe1f48e39e32a72726465ca88ceb06d2b4456daada9f936fd2ece756854e1616dcaf7ff7c51aec03a591549a62fe190187b65d76814137e39317821697eee9acaa05feaee3501479c6ab5f241653524926ec93f43979047bf5bd46f448d0ef41aeadb896a393f6e635f88cfe0236c55086a54fc56db30ebf5b82689cda684c8abd7b7aad26e3f99277c39b7a5ecd103a3a83f713a9e556e24547ee352ff78b357958657272275c1794886b5cc79e17759ce345e4e72307f048ce5fa700d2c49d81a8d2d30c34110890bfa1c120d642ec62eab21&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848813582Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/d, Name: checkpoints/5376/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5376/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=7e2c656019ff9304f7eeadb86b2f78cb7197b411d8e1f52388de6b6b81c9a369ad6acc3b2be634c9107c664ba6caf630029885341cec26d10d60de0ec9013fd77fe0bba2ae0102a585340606520879325679b31505f8d6e0f2945a95f90ec8b4913237a0b55a7b867fd69109fc7d26741e3f8c3d4c5a1b3d98960c14a8e08e8c8474291eb4175c146046c6ca2b3e1788d28423661177a98d907c24025e8585171006644d4db4ccbb9db33c2499445558ab51426d0efcc1ca41ea4f0cd3013ea95e78dcfefbb6f44a6a737233362848f3395fa3e66d7290a8b5ae784de63a608e9f63b55384689509a6a0106439106bd68c2054eebd05a8f22f1661ffda115290&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848627596Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/d, Name: checkpoints/3264/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3264/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4eaa936ee1805eaf1c7c2b030b8b42d72f04dd699a26abb4648f2dcc4263468fdb780176642304087a735e562a0e5347c1fbd4278f71dae9086ad084f78c9047cf6a5ea45b88ad1fbd196c874251f572baaf03401afeb369ab91df7032e26d98179b26a78a6f9e1c09afc41396aad85db03fe3385aeef8351b671d5187ceedaff4309d565f1ac2e4b5d27eeef627201b612d2073118e6740331583dd8a5c069f55f714e1fc0c89946f8b50eb435429fd4eb814cc15125b13c886c719179c03777f19fe7d4f30e539ac356b828332605a30b906f09601d4725cd00970f86a92215d1cdba8b75974411b77207f3cb712106f4ba36c5c58d55a66d263e4d1b3a0cf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.849171673Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184, Name: checkpoints/5184, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5184?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2f2ef8ba8e7387ba0bd8278cf934dbf88fc64f3259e310b49b7da542d3a1ad36292b180955c304af3aff759cd81c3885f7a460e4dd5af22651a734d5e3203e8e7d2c8ea3bd1e923092e3d2b6db9bb13fe92ad81b4a4940659f147ca14bff4503b4eb967da6ad29eea2165e14be349d0ccdc3ad5a2fa35fd7d735da5e48e1839d9a18707bf09d63c617bac0ff91ec5c719afd8fdd2044f164bb923f9d2ce5f4146774b909a12ed1a23b8d688305753a5e10fb626c8eb19c9e48412b821195068278ac793fb7442906431a2939b28236d74f4b4e0f3bd092eb67bcab24f2580e046a23a5b058881eb2fd13987ccf3501248e35a898a51a7162a17c372373658a00&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.849249815Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d, Name: checkpoints/256/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/256/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=170c7b45bdd79b409364ee82da5b4473f7cfe6335ef0dbeff739097616fc96dde7d1bde0dc0b56829e0b7d22a89a4039ae59a32573ee5ba4280a03b3c462a0fe0989752d06c77fee223d31eb6cc6075d9d843cb7b2b894babf1cd725253912ea9948d3db44086aa19e1fbae780f75dc70697fb7bfd73819e70cc6d90143029d905277a4f408ff0c1bd0edba44c6d1ca4a5c67e6c3299260bad7d4fac50ff2e475c71aa6fcfbb81db5b8fc27ff753d41a9a90e68bc6d0162e28edb3099b55209e0bc485075bc457adcb6752a6cf7931fdb2b4c0a3eb45c585ee344d11a3e9ebe61ff4a01e5ae93c5338835712856a10188f8ab2c4fccd6d13b44c4f4b165e7866&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.849267082Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d, Name: checkpoints/2240/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2240/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=83d90e26ddbd6ae95f071ce2cd8dc20d349d322026b1106429d58afeaa046497b1b763b7722b60da53536363ad06979865c26f51a150d4422acb77a792b96eb73b50a3315ae5acb11c2803e7f222e6e3287e9201bbec022dcc2d83e6f8ba70589151e45a77369c30ceeeeff2f0818e362ea802dc14b6c6ee6d32ac6323b4204fdc68327b3da10024eeb261a7adac947aa2d1e9eb7516413fbda933b6fccb635b08f05bd7e0147c62f7c60e4d638d078ad95f2fe5f7aa61e74d5c7f15e24f782ed8bc0f592fd90d3f5d5c1c251969a5b8125d110d442e33e594ff2e46ef88bf9debcc16fd22c37442598c0109f5a9f1921b0953f05019136d4dbb44f731f42473&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.849318323Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/d, Name: checkpoints/2752/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2752/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=7a80b4bf40a646a69ae6dffeec9706e296cb44339c65c5622989369e54d830e936d5ce0a95693e5bcc03c9e71b3ce06f7877d9f22a8b9df5fa388d00559891f6847e0eaa6e4b46210a690cbd98edf3b9f4af71a500e44dea4e214fc4df50d7bae9d5ab94f1c6f171b36fd29c50e1e93316e54f93e166f11ce1ab019d8e4576d276520c45ddda37c459d30c4cd1543dfe80b6257f9153850ed0960e4dffb1024db4b4ae9f658409817eaf9020e5d6752bbc070e7d65fe165eda73b82cfc22f93a7c8a32b2f3c4790cde654b98ef2c93105c899da2202a6d69a9fd1e3dc0e0ddb718103d12f7202e8ca8af922a00830d1b17c36ae2a1b12d15e81b4798839a4b9f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.849324384Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0, Name: checkpoints/192/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/192/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=6371cf9290a4a626373d61278aa616fe7068524aff37bc056357a15289e98d4b83b53224fd953583864329cec528ba05831fd2e511d397c476952479efba829e315d897de6f7fd04673a921674e80081c55a01577cc4a864322b5b1ef45a81080213963fb4c6c8e713d3506b0ee5431c838b56c8b6a7f835586f32f025c23444e7af27e52675994d657b9f766bb4c0bb6dd27d7b52c9d7f349c5546c6f72c703ae1702cd39858b13c0bd3408514b3ee4e77124ac922777767310a61b8f1284c211c13c49849e31381b972753e50d97c5737a386e8c1633ef8d17cd752707036ab4ac4f44c9766e9a7e5d1e3263ee99d0b594a92206a1dc83f554491165dd6a91&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.849334255Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/array_metadatas, Name: checkpoints/5120/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5120/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=221fa6700cbf25fc03e7ddcdf6f4bd3b3e9b72cef9dfa83df7654a0e766de819b8dd1a43eca7698676de697e43124ce00c84ad80430022a75150e19000e5d458f94cd517258215a035ba777a6e839138f0ee5b108f628e2c6836d7d13e6c98710b62e8a884ad42f75ba510dcf185c3fbc7f9e9f3cef52e1aba7c4da93c708d291015a8f5a1051d39a021b71da7e94fbcd26ccfb81b9121bad8c231692b5ffaf5815f44607bb31f01c4393a8b6878efc3b68170bba641f08286c67b0c7b45b00c737ba3c14ffa60083b21d0f6a7b331e32351580b72f25447389c3c1f4656214f64fa8a7871a471581e2eede0fe3f7cffd0875d659a69e2521fa5df1e6c50985a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.849355242Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0, Name: checkpoints/576/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/576/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=77f67549a6686006e840802fd5fb84154eeb352a0729a4c6cde8b6ed37d0d443238c2e7b345a2a435b4ae664fe458d323297250a6bef33e44cbee6bdb86576eebcd4cd04b4457bd93fe14c47a013434e66673f7fc08b38ffdfa054776c1aa2b4028af55bb13d64ab27ea9c23fa7f7faa4ef3d2128067cb9b8991237a12160cdeef2be95df0292d9c3c51be65f8fb53d88ae99fbba4b2a7976b36104e6590847b95eb2a25644b4c4fc699127e62a11738afe9dd157496a6c24f3b661116ef193cbff911e1b26a12fb609f8e0ef3c22c0c354b77d600fd00cae916e690c25f3dc3ed6e08dd9d5b46b0ed3a88456a4befc0b3336d4cb1b3c819cacad5760457ee4c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848730111Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0, Name: checkpoints/3776/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3776/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=5df3701fdf449b60ec0978b04773fa1b4a067f486b966478a82d647117de4efea3b61da0f7fe0dc9335c9305477847a8a3666885d4e684d37c4039f49c2d7bff98a5e4fe93dc89ace8198a31a0914fcf07a27379426a52a1da51efae6d2847a5c74ecb5d2df8379651c5830faa6c3ddc7cd7300b158657c6547be108d0d85adf50ddeb5de1a4fb552a942b6921a3369e6287fa298f0b429c5e498d9712a0c01d8813b6f7d2b2bf816d3d7a53c0d1e3ec4c150d171b9cb3e34e8b8ef509d1621f69bf79c99803e71b4ae9575a91e799c3670948098fb91684297156019187bac9a39858513ce8cb65a0db6f492ab4252f747ca48ed770be1b91d4e775b7242239&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.849403467Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/array_metadatas, Name: checkpoints/5952/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5952/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=ba8d35d0d42d21b09010d78c459a0f4b389a176ad165807e04cb9dd55a901d5af4d153e1b575ae5dc85f597fb0a594627032b2eaa411cfe55e5cf2c4f7b3352a5bfcdf215aad91ac052bf891e290d1110bff00bfb257c8422aa2767d1b8d8b1760f7df0b7c7cb3963aab445a077fbf3fd9c254e29c088444dea8019d8e0a40caca668661191760caea15491249e3dd724c8a2b9158807feaa4d6071ee4fbf8e5dad237887b991a1a18277b6b1d00fbfa8c5317c656500f06a4eb3bf3c765b60eedf6c92c6dbb7c0e558d2f04c38660e2cf1b1fcb8f894a51401b011e00b2abc7171579651364f9734b45fd475190bb74f51166503fcc24bdc2fdce0eeb670ac3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.849355851Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0, Name: checkpoints/1408/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1408/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=00d70c1bca78ba6fc328035dfc3934979e503ff0f1f004029f252b2f96988ddd1d48303e5acacbf42e97dd917ed825594b8ab7e752ecd8e346a5b9dc3ed0616eaff94c763c0c39423da6d06a4656c837415d4e8677547415b99bebaa436ad2bad6b16cb2ce8d91e428c3994c8193d27acdfc5c8e3057c179ed8fad37a87e0da9106db19eb2dda852452519f2cd5553ec23c641e8ae822705b53c0be92ad6e7031edd01a71d7e7ce6d182c146bbf46ded2a276d29f0ee832c596db978dcccdc217b491f9c714b51abf6e3b180dffab4b22440ef15ac590befe9ad07adbfcc921f2d2f56267d19c19df0a5eb9b28265927d36106327a7295c2f8ff3333a618dbd6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.849480979Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/d, Name: checkpoints/1216/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1216/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b127e99ecec2a2cc363a1a39198db2d1b918c812a35989d3db44a76dc6192f7461150bca8a59cb3a37c1852264e089b277aad73e0152965769a151f2e619929fa47e08ce5811883f3332ce185986d28a9e05f22d7de2b9a06841e15ca7725734761737f7f4ac5b81cec79429cbd65fc1e05502e91e27d3f972632068105bd8bd7bd52c3f5676d550c6d8488101d16b9de64a55de1889df6e43ac2c1f59fffb6958659e1990b67edc83031b8d5018ca4eb49b0c53dd92091e261c7e8c96ff1f25d4337efbb96ca5e15c36adb227a0a7b0b19f23532fe80940779f68311bdfbaa98c4e6c7459b20dd58b903e5924549476c3cdecfc78f1ea0608239b52fcd80f92&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.849506252Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/array_metadatas, Name: checkpoints/320/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/320/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4a5b3e065dea28933dcfa740be624c6b83cfcdfd644956aca82a63651e5122e256e088ce97015c806ebd17d59c34dd15ae62ec8fa673699e877426638188983cd5c0d5384848ad3af4f126dde143a18b8fe76482a7e790bd30f17853c5f9cc12f3df42d03814d32bebeb108b5325c8836bb7620c2d610db5e261d56793fd010b15bd8d1a78aff949ffea8898abd5a4c7d2d0a12b49657b8dafdf506fb0b30e109aac925e43a783b4856d6936de6d9c78cb35b8be8c35fa13691366830713e4dabe225ee29c10d47fddbc50f42d04b20cebd2e9a86e37d15e504a0d6d04d69c618b294425a3e9e6029bb28d21b470bdb483509a63fed16d664789df7b87f5b9c9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.849284552Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/d, Name: checkpoints/384/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/384/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=91caa405e0778fbc8959962fb63e4025519eba0e812d11ab36b34987038b5edae36033e3704d43aa1446164dd6e48dc7adc06c523431c5e3d93f334d38af704d250e3d215ca148e53b4d068b31034df24f5fa63ed76c13ffeb82d8d49ab4dd32b0f3c2b6ef2c284584f0a5db306dbc42a09d33d2400ee82ce6bed45bcfd7aa71034dd40e29281365242d32feb8e1467d73fbb9c10a919e8ab424fd38a84e6c3a4f4e4c95a30729b3d26370b615b9a77ef87be266fe82a8813e6fc68967acb0dc2a666f29fae1e3508b7de33a3e2093a9e4c032e538e2d4b937e13d7eeeecd0e9a2b2ab3d8b48b58c4530d3e0443f2b3cf1a94897a10e11575f0db36f45aa3f9e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.849178446Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048, Name: checkpoints/2048, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2048?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=109cd3251da35243fe16401b602a82edbbbd72db80ef99536a37a416c86b4ccd24bfdf2fe9eb40a72b4f9be36960519a867c23085c0eee19c5fc3bb0e106d509275e44bfa26cd5014047e254c81deb74de7ebe5e86bed6edf8d7837b25bb5eb7a3df6d0dda02e543080e1f5448599b109780860cbf95d5b229a93c239385af1358a911aa0db9d0adf237cda6327098e55339e20c8e5b4d0b44b166fd8ce81ab475435afe017a32b916b16e2d41a99951e5b28fdf1895fe210bab536379c83af7fd71ca49c3db63aebd788a6bde832550eeaa4e766401106d9ea031599a4b688a41b68db94731e3700734adcbb7f8a15ec8e57f92c077be09ce0d76b54fe6e03f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848772064Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/array_metadatas, Name: checkpoints/4608/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4608/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a9f405702fa68149a219145e0a966c7b4dce59e661bdf0bffe3d82437c20627d13058582d01a79ee8e1327974ed318ddce34b7cf780c2eebf564ad6c5036ec5591230e755565e45b808dd085bae5872a7bfbe20321acfc1da7a6f675c57eae6a588c571f325c9f3d9338914bde64ee0026d5c65190e31dee604a9532bf63037d02b8d58a69fb7171a3912efaa49522b0767e3b0a3c5162cb33490ba3a9f148fc7dbe8d46763b7c8359f7b72cbacefb5871845d4af00145cb22a0130dac979e930c7ad3dc9952ce2f0b3781b9c178397b6fb907d5076247eca0819c4aead513410c960dabea104b55e42611b6f89f45bdb175a5bb71f7d61540c1fb287a591aab&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.849485337Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/array_metadatas, Name: checkpoints/3904/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3904/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=98fadd53c05c18efe9d3d109e228ebb902cf0af3c1a16a0ef89d49d6bebb55acdb2a9c2920cce19e0488dcad850ea0dc08de22520672e1e308fa80ff2b34d3223ff0dd6b1cbc6d05dee133e11912e218ee7b063f9480a9df2afbe5560c25e730627b03cab43c50b026c795e9b9603c35325a16bf9e598f66ab72da015519073d3dc9ece957a3bbdcf372b9424b3816cfcd19b211095d11ea8cabc7cf7bf403b1a5c38cdae8f17148c8374415dfc8dab2caafb6deb448d4720d24c0c9717d3d38a1dda09ab3ef3fcc88af66b96e138043093f273aaefcde519154b0c82b0d5db06b720ebd2afbdda95d45f538af3adb3eb8fc263a881b13f85031221d9841304b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.849255049Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/array_metadatas, Name: checkpoints/3584/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3584/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0624f9a36aa8deddef727925f73ad703022fff36fa682d2b7003d50b3aab1b9712467eea3535842cab555db25fe096f00554edddf6d44766d6d53603b2ba206575eab3ea208846c4321938f1f9e64ad43f33d18b7e8faa7d7b8040a40d3ad4197520ba7eddda752e1414eeadeef4c2a5472cf1e3815c9117ab3cb12cfe0460c8252a0c27c0b42a02b9eec3ce74da9075981c9b757d8f470a59e0fb2888b91c38ffce449b5fd980e7566c25650612dfa4b6d83e0360fe576a6ab688fc5db56a10d24dcf855a4f22ae923742ebc9f2c0403f127d109114b6f7ecc699d4f64249e85f495050352fd3f7cb05238b80ca60b497487a30d166fd7e4a716bc7880f2380&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.849506222Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d, Name: checkpoints/3392/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3392/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=336fde8e0bcb79a249b73d13038137df37aab60fb043f517d201eaf89d7667d8e07583cff8931f333996cf9b310ca9d0f2c29a139e18e21c525642070d3144b217e3ef496ff26bf0b295a6959786f207c2b62726db7a9ff4f7d30c3197f247d3f7c98715fd8d4242bfa7cbb93e79ba2bdd756ab3e76b3df8104e7d50e031ab5799e92aa3a867cb61f94143cdce08f9b04415f429aa627a8a9753779027940c922a00e7e021de587222374386f68c003ce3509894eb989ba495940070b8182ea059ad5a84da14008d6ab311ad9254eae88b25a5dc2b4998eeedd9f5e778c307929f82c1f8a1ad3be47315f23e5f1a5a81b5a3884a191528dced8ee8c58983c77d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.849489496Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d, Name: checkpoints/704/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/704/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2dc71c5f81928098e847ccf89c7bcd0b0fb1296cfdc858a09ffa9569ca2a378081a903babb8b05eb36e4ede7feb2f290aa40a33d6e5d7339f424575d2c48983126d1a26189d0d06a33d594db714bab2bf31b65aaa2c9f139bf6d3ee7e0d449b657024e38c2b914c0a4ae3b9106ae02500c2240a34b8ba9b590efac10171c2589da1f2746091dfe2f29f196bdd14869b11e8b1944cb28ca643272ec1ba8b1f885eb71c4da9e46cabf2e0d87b1c6b61f640668cb15baea70d1be76b1a5f9223496371ff8588186525b4218fe3bcf333c4a5a3a64946763b0252f3aac75e7e76edc1a0c0f40c498e46adab54b21802bd2168940565fb83a1eb1513b883e5f6e432b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.849493473Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64, Name: checkpoints/64, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/64?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b2ca0415e2bcf607787e9e1b0abf19273b4582ca87e42fb88f57a1323fc15cb534989a0ebc34eb989b93b6ae2e1cc34f017847e84d6727190b9015b5c7250b7a85f395d55eecd4147ca0057fed13afd2cd523f67afcdc5f2c1a258495d4d674c0dd7a839334f6523d9ddeece612d1859839d98ea3e4a8a195ac9024ba0c9762db8573531f5ec96275e69a6c5f33d48f38e4398dc046b0f4da6c0c1808de45beaa2b645325ac0fb7daad5b1a8ba118a16b4060a629cb0ba8589345cbb92dd02b1966b49ebf975dea27ad8a42ed52e819f4496ccbc1774af299633257c0d689d5d25be08ce29e5cb09cd19117cd577524497c15666e84533b3080da753c31f05f1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.848857112Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544, Name: checkpoints/4544, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4544?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4793d5386c1496d7ced225fad0a9238f6304fda5701d109c608254f5e86878c15f2264585298aa2e772e3e0ac8985382975f086a6a693abfcb1a915913a67d3fbe29eb5f32292702b3f7a7d7d52c7817116d052e9217ea4fbccef2d9b351d5f4d03ef4276b4da536edbf4296250c60eb74d4ac74dc0e959666ef04eb7a142a3aeb5af101edfbfb38573dac5e204db8bc76f35af08c3d65cc2c1f2bc1620b3a43f3211bfdc02203b5d43e115af45e206c3c15a873974e68a32ea31706af915cb1fb3414854b0fefacea3df2cb2e8d65a5f8da2d04b627841b3961542e70bfc31acfd40b0b93bc56c2e36eef8c172e83120e098b860db92ecfd37951cfc6dd75df&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.849498331Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default, Name: checkpoints/4992/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4992/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=74def9012706adaa750331c139627b0317fae492af1f8fa2a51ff6ef783dc9e49c59fd3784b0b2f521543a5ec6983d7f922ed8ef6d2782b8b40dc096d8b81bef8d787d51e2819600de20b85517a53e704b129a72f84b38aa05a5d818a8e13df39bf585e07a64681ed37b8b70da6f7b07a4b90ad37eaa3fae9c15928ba01746fdfcb47f6f426e9f48e859977e168518c79687ad3427a37f7b2e07edcadd5831ce5c3af0a120815b2d68009f961fbad4853a98793927bcfbfd010e257e11615d3e3e47a79ca58eedeea1f785f2753987871ad0d8f0514a981adee559f53f85a51c5564fe7d6ea9c6728615cb915467fb8c340aff95168129a06e5b20a757738813&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.850506033Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d, Name: checkpoints/4672/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4672/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=54f73113dd73f1c1535e4be5f60d82d7fd462b810d320ee7ec814a037db7d0a8a254528917e114ca9b4e6cb27c63e12b2bbd892942c3fc7cdafef16ad125cee983228d88286ce416f30ffd9431af90d2ef5560f90c54b51a711311566c4ee279c7b99cb136c806283fbedbd32d563172f6a5ac430c225dd4872194dcfe1108b37d5c20212d5420db245000e4273919b003c82f82cad1ca80b4f46fe4e432564071ae9e9df9fd32dae80928c82d2f34c72499ce0bdcb3660804e9ff4b438d0a31f4872047c4baf01928e36e553a8ed43b9cbcb394e188d5ae5d93ff7c2b2f9b43ba9356bdc58708253d722679ad47321eed8776772009fdbc063eed64fedba142&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.850529073Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d, Name: checkpoints/3264/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3264/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a9f273e43f73daaf76cf47ed54e4fbb13df9d776770baecdcc4776fed764bae1003b22bfeae89f96ff49a05ce6dfc21b32759f5ef220ad80d3a6c5124e0eb3ed81c1278458948eca1a2f37ae8c3bf2618a778243d7a84f34ae738154666ef1c38315186d712d934f0f1a6c0bd8f4678dde2905674cf591eea8fe1789a4841284ad907b822cfaa8bfbe127b2d5835c1591c0f6f3acbfd43d6222acdb22d34cef4f3960ac61c8ac040ac1011d313aa4c6a7fe1654d4647b0b63be7ab4792a964a1c05a61b696707c7a7d4f3b7c1ffb074640b4cc811a24b910b88252ffb3a89c72add7f5af35f40775a172fabdad6d63782617d67838ef3abfd365eb73f12fda71&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.850559199Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default, Name: checkpoints/4416/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4416/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=7a82c04eb2bc9ca54245c169b589f2e4b4c7ba13489dac6165ef4628c63fde76151cf37a2c9c80a147963082e0cdc49146035e718fcafe1fc2ed5355fe0d299139c92e6e575323af77dd8eeef78dddd558dc4a24545b91e01d3dbb317d8251d24de5891eea0f0acac10a1b6402202423d3b90b07b2fc20b42dd5cabc897295d234485bf93b9fa4f235335743ebd18d5804b8debecebf707e0f04b3d120fc9a17c1d3546909db19ae9ff51931c6d7ea79f4a7c321c052dd6b06609d7b5f3c8501755af080f718f351605196e5893666dd94de2664e7454a10e718bedb7b0d23028aea6f8dbe2209c853bbbc173b31b97fdf36c5937c178dca172713b614351a3c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.850560857Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0, Name: checkpoints/1728/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1728/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b25e7bb930dce96780f0af99ca31d55c34574c5ff42db9e1bf25e7f650f5ba4528039bcdd2fa512951f464e7b6a47b57f561435407f0b73aaf8c4ea3098c807ac9c552805e3ec4b176031e713ad80401ad0e958400d735b63375306d9e869b898afa62b10727afeb288a535b9afc5263314f5ba260ad548fc03d1afa9bafbb1c6baca398b69887bb44af3ba3c4cac242770d75665b8341542364478a04db373b9349f512ad4bad936df2478b7556d6307d7bfb5b7c72c78eec307765a58208ab33a0dd421aacb685c4da03f632242013ba744a7171ea64cdb7100ce95f1763cc5361cabcd497975c0ffbba5b41a4a5bafeeebaafcbd81498f9cd1b16e115d91d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.850620445Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/array_metadatas, Name: checkpoints/2816/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2816/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9852522d1309ae4623c291a5a1340ac6d74234ad825ccce2d49ab7dd528c8e1b4685df27b8ab7db5a813d086119df039946e5c72db5d6c51124279d5e6f05ac6de1bddf7000420cbc8aee2dca23052c761f32f10837b90abe1f534c192d9ccc8f53566f0bd03a0f344bcee7edb0d243751117c2b1845bc11d4bd5cb2ffe063cd8c7a78704a2db2fb1e0b60714e0fc139118e5686f62f18150c003a31dab776516297700b926fac8ac5e5a8b77348fa6a4b674bb8bc5483d126e17f2bedb35a3838894bbd29416d65b09be47e6e6d277656d6e8f4762983043b1c2e4913bd158c8f5a65d3fd5ca155e99e7a516f50b37acde59e50d268e8c44152d7104fa3a6bd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85066021Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d, Name: checkpoints/2624/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2624/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=26ba7d4cef1896e1852017c917404384a067c462605efe549ebd4047ed5546a8e6796c480fc5578ca564487390e70e493db2504e00aa1b110d872e34e9089911020dad85809f8d522642c5f0f08804f7c8454474901906b5fc211dd7382f7dbbffe106b261b5a0c267d58ccecb57aff90e9b7439534636c154f2d6a1ace6680be4fd76f8d74f2ba4b2a85ff8089d9defd959ad12c179f9eb7fc8cbe6a21128a340ea728e4c1f9f402053e04e84f6a1b03f606b49ebdc276e1efdea470b6e3e385b350a1153ce10955b8d04cb3f8acb44d9e47d00ed67e975501d2efa1853edfd5e355b1039b0b264283d15d80f05fa912353e57a6d3147bbf7a66fb98a2a118d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.850680479Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d, Name: checkpoints/4992/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4992/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b0ac2f8bc8420ba2672fd46743b6ab50df2656c0fbf621f219f81c002a2f455d22e0f24a99716b2d1d809c4e099c4d7b6b850575b5bc6bf263b59b69bcad5b32f64c5de459d3063dd0633eff1c999c30294812153a298cd15336842f65d795d566de784686fbda71d30afaacb446b9beaf37673c7a023c0403945bee1b58ad0b59c1ef358b147363a677e8cd62898f56501bf19fc67fd183c5fd5dfabe1d6f0192cf8c96eade615018fc9bc64292216b607cbe9a07868d1d1cef25190a31e4a63b86c07d0979900cd01f9c4dc13da521752c8240661a2ae365726ef31bfd26f67bae630c39f3fb49068dcdf7aa29eab625bd6009df92a0744fa9b38ca2b6ff74&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.850707396Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0, Name: checkpoints/320/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/320/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=497d637ddfa92b34c10d672686a189f5039d5333702582cc1775f5785f669ca41f33503cc4c3bb75c4451b6a141d1fcd958e3354ea42a9e96acbd60e369dad6911b71855e5a2037cbb3ea7fff058f24e1705d9582331e674144b47a7735d55663bf0216490bfa01090fc7b97ef28bd037de9d0462542edc56c83b66b5275cbfaf6c51a08bb48993de9bcf00f77df911c6f2b203296dd01e85d1ba7b19f05e50e1d2672d4617e2971b0a308c00e816c3472d1b019fd9b0e9fe6b5150eee9324b79e78b8cbea7fa3430d97e62cc4048e3f3c79468b63715edb5ba3bd1a2c2bdecf9b3ebf094ecb471cc7214b424ec3c828cfcf00e61610bf10b51cb91e175e268c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.850749742Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d, Name: checkpoints/320/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/320/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=c25da150bb395813ba69edde7c747507bb0d8c755dc955e8fcb2a553236ff90f031ef92ba80c177279c8d67f5a24455e7672288e8d8d9812c04ae535a9562e609d435c7924b89e54125c39fd58f76f525b3127b192c7aea375718bbedad5a835f5791a30b7cb5b436b7ee920fbdc433528c70f4d92e064f4df175ae97657026f3fbb2620bce75bc676b193f21543d49984b8879f81d122b1e3f75d5f1ea63a468eb90e8af6f35c7479c70677025f81f38972ae58ccf357ee0e0c7e9d7e6f2f15ab8b6cfa1d33798747c66bd8fc0985e1dde558edd0b810635c00643473f04596cf97fd25bbe23bd16c17197078994b213809a1975885e99cfd2be843c5a84953&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.850780652Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0, Name: checkpoints/3008/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3008/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b95cc8b56468c2f2e70372f196eca9a54b8aa5662e8395446ed80581acf7eea3f7a54a02cf116d1cdea059cd514ae7c316492e5260eb5a5405ca2a16317c6d1ec1ea04d221a1d1ef97d20056a4a5f3325dd2bfe1f922f02545ad1c245f46a4ae2f39de619c412c0f5e9efc20957eaed28560b610f48e02dbf342ea492832bce8bfc060060beca18c37cf4564c6b3b5e145a071ede10f87b8aa3f58d5a8e2c6029e616cba03760c495f1be676c90efa70a6366caa7b89f4cdb4a028477b5f7b4d07cc9b0b4a04e259585e0c71b616f134db76ddd76f5ba063c79316f92513d8c5076731cfe852227f79a8597d0595fb6857e80a3504c908f021ae8a6292fbbdf1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.850808642Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default, Name: checkpoints/6016/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/6016/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4b5e90e7a69ef70ad05d0b4ce104534154105bf26c2bb0f45a6e9943dc8bcac521255900e67f5ab86a350528d85640cb4a61ef5bd213e9e2d0b2490b7843e016d09d178d2c5aa3bbf44cb505545ab75c425eedf41ff3cadfd114833ad202f7fd73069dbc90559ac264d174b3818c654f03b9984d7304e22b19f46bb71316a6359442fcbbc867b732d971e8ef32722476f07852566cc5fe249eb7d8bb8a75bde2b1c7bade7d1b4ed451f6d6d108d3c4b7d78de1731acd5e01323ba8ab2f1fc46c58aea676ae00e7fcafa09a3c1caec063e36bb7a051b52d2073b3db5ae119195eea76f10635b5cc08f2defc28497e6818c5f61b8fa83264abcde350e68b4c0089&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.850840905Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0, Name: checkpoints/4992/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4992/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=07c38754d8d49c7ae58c421f11ec913575edd5f7deada3c726e96c7a30a0654661bf2e227a78a28ded3a71272ddc8f9d3b186825b22592ff5d29e23675f3c6ac98adcfaf962b84223564081f989cc92594018daba0eb9f763a5ee06212efb8bae8f70e149673c3f7db6cb25e8ea65028e4efcc3774af82fb9d87f4e9a7da39414e6a7dc15ad61cade0ee4aa74178ca4718692fa0e743e878ae3965db0c98d0f4e0342f26646d4df845dfa2423ee41e4807116bdf1693cad931b1d8f590aa93cce904f362f86d15aeca4a1a58ad873186397da9e5b4b3b3d9da4faddf2d4d1214f7d3b96ce14defb72d651139eeafc7536e3b656134d0c6884ef1b10f0ea487c2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.850875509Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d, Name: checkpoints/2176/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2176/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=867d1f09568eb171f481bae65f084e92b8105d307f6176a89ee4efe258dd4ab86e80fcc821f3cef1529f9fc4d5d20b7ca7af8070880602fd699039229dd681be90b789fbdeb7abed14894ee7d7eaa2cb168e77158814243a21ac72a41a7c0d941ecf75614c87ce22ed93e4a0322710a8886e09d4dfcfd0c3c4a4758936f9c77efaf89e7d9b5a14b8811491c417085008acccfecff095c04e4f56a7bb29383ec173652225f89b199cd5d6cd818353577c21ea543c197151d220772357eb6b6653ddc4589721cc6db9ed3c75df46034024e77bb5228b614b645b042c754e0554cb55f44ba31fe4e2efa29675ec8c47c370e9d84b85e966cadb0aa5070e2e0b8067&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.850887391Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d, Name: checkpoints/3904/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3904/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=950a702957af930ff3315dd7281cdc39bd0d676fcdbab394422c24403b0056232402e958a6a1dd70c7776b4b2fa941c20cbbf130143ae8254fdf10d0b6023faf584f9d5e59a8477301a236879d2487789888f6dff0e5bfe2c1682a11d619c4a1417740465116268b7b0bba0f9c527ae26f49015f89905389cac8dad4c8ba7291cd31a9795f7848a8d0dd312e3e9a44635574c6d2970f72c47582b89f0b059e259afd3239d7a82479e92af8b3e1e8da5e5631ff5087cbd384a0245c4e5ee76de33ef9db3c65cad4e56cd2ff64a3578879ff180c68cb8bb74b1bd17131de79da6097f30982b402606ae65b9539378b0e6d1b9ee025ea6628c881a78235ec1dccbc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.850925989Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0, Name: checkpoints/2688/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2688/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=34b29e978d15f2b91f304159df044b1656c19d4947d0f474b8edcfd74556da3bcc51bddf26eca9d29f87a4c3b688705c4cde86ef6ae89449b4f9219b74c376b5aac07df8ffbc1f4e1dd3b7e5a87a7f88ce450716187d96286f2de3ae60e330bdfb8707db1b7614843fcb6c600e6d55a9a728e9de1ca036399f5916ef228a1eac1b66b160a78cde64ad25405749e08fb8bcace1fa4b36fec5ac01dbb4eeb86748c198fbf4a69e0c2a9b94fca82a4f1ad5ffafcfaa2b0e532701b6ad8aa6911544ffc2a98df50943740fe8e12abb821c56776d1f6013921e584883dacad5a513bde6cea1c532fe21d208b8c0e4d62817f33edd08407e23fd2a3c5d9915a3ba164c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85093412Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0, Name: checkpoints/2432/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2432/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=bb4a72a727a1f77e19c93097f4965ce69a6c849ed755dc055c11b0429071a50ab96833abdc231fea25726dd24a72778c97e5f5e24bb750f20053fafbe2a7d110b0529bd981e9c0a42f35cadfbf9bdc7735188577c0687e16d72497b6b6bc82d1d5ff930b202ac4254a7e44db5aa3f8a858d05e516400f7dd48a7bbc4a691f7f0716b91d796c7090e4acd288a8e47f73e31343d18bbd78935de5a1d00286845289bee4a812e023223f1b13b306462350cd82e343c795c28ad32122c645ab5cd22b671214bbad6f7ae914a5332f455289b24ea9737006a7240f83c5f5966f32b7266b9150be1d733b7f76e1aa1235f2d5b29d58c8ba0513658ec98b5b48d176d64&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.850968444Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/d, Name: checkpoints/1280/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1280/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=17138da5b93241773ee8a05239d10e01129dfc48e969c318f1acbaa0278e130720c03b7909d8993f7d6f7bf960ebd81879453e697e26b64c4df5f38b69f2876f8e4fb7f3aeec66c306de1c024b8ca6347c7fc81a93f2e8712cc04f09a9323345a978e9eff64dd42a28129de6db3e0bd03b4112567bd1885661b320304a97e1a1cbbca139e196cc9cdb0a961fafb031b6b34e99c6502b91750a835e8242cd3c3b0cfea76c8bca1f64b7d171c478e7ee6b01738d52541dccb58406e30a5f83276cc2d22f04f13d4a83b571752910b63762c6faa0162060ef34fb727b55230bc1311af283bbfce7e545753d099c673e945719eba1a4b9865f3da9409e4087b594d9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85051543Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d, Name: checkpoints/2752/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2752/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=013fb425bc57d4bd304786e0fde8d16833f29b5be3acacf4f58865fc15e76e8d309a5aab105b28150797b63ffa153e07fabc90feea1cca34283107f9efa5b71c404a6cc4e2a8482353db680ef3935458f8d66d309a07c00ea6dd1819d017fe3bf4d7952ed267e0f86ba80794fafdd5431e63e2c0f46e6e21c031a7071e36ed3661a1e5503b0905535bfac3d6e8f83f85fdc859ba777141a60df3798dfa9dcd45b2e6fab3e38c76b452b1615b6b98da802db1347ec9ab73b9763fcefa9cb071c8a7817b73c2c6545a538643718979fc65d6c870663820d7ca1d674bdc60c7688b71197e1ec8e68e53c6e17aa50fb956c5c590008363c04b2f4b5e83d01358b3e8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.850563835Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0, Name: checkpoints/3840/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3840/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=8e9ce7265ae1e2f8952801a065a9bd8aa33d4374e6c68b2ab482fb0666e877fb456b033492aa929a4b2d917126dcaa06d19dd60bad992e156835eda8a3bc3fa1bc676653e9c4b5291d6ab3f3b8ffb1bf343de13951779b7ff3ae8f843959401b44aba14213dbd36872686359f62332acfa79d059aaaf1d0c8a38340912b0b8b23746ae9ecdd907c20e34c1a4562f56143f7919128d6f3f896e0e092ed2ece84d6293a8f14de769096aeb1f9492201c8d79a11963ecbc956b98780f7d7052f0d4cf73e55438d20b038869d052b2341f5777109380c064c3d9dfd5fb17eba4ee8cd33eaefbaaaeac10462a205eadec079910c1b3a1422879aad4c4773a4e445997&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.850628159Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864, Name: checkpoints/4864, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4864?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4c534e7b132707649ca3b8d5ae8bde062ee4d63fdc9e032a96edaafade5996bae7ebf5069d35311047340a493d278ad94ff2ad09285ec128b95b36b19e3cf1f0cc7a2a14435ca9edec8681bcd20e953ec5e9025c6801281e43379f204f86f3b63d407c10a6042b68889a5af4589938ea4ae0398649a96b2c69ffcd617196bf4269e09d29b0a07b80d9d05fb0b01536944e0bb43535a3ee1a56c95542b425935988b302d8d229a28d809a04462df58c6906e3420086b3c1f1781b21c74ba74d101cc445676b8ff0c1f4eddee899d0f619cc83c93bd8cdbc926f16aa114b2b1bc74ca3b908223c655dc18b1f4462f64eb362c92ee8f23b9d98e7c18a23df43fbd0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.851160802Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/d, Name: checkpoints/832/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/832/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a303e18af8f01ec794762fa2b99b486a9e085a8dece64c8d2689311883041a3b7e2464ce5660c40133aa6b8d08ab738ad59e7e4db14c1d310f52bd04649392d5fa988135b45a5126541432343faeece1a2c741abb7f7db948885c0a27da6f7b3e1e2a5bbecfcdbc9597d96b80904ec77e253f6a75d48344952aab4a40191ea0b420203623d0bacdcc5e8b52811695c24a859e8f41753547639d35e1244dfb7b827acc246fdb8730b6f29a839221eff2d80d2d41f719aebd2f268274f7f7535143ab86909999e72960a3a92189f32043278e038e9c27aa710bc1370d4e98bdb617b8d5e2ec53b699848047438f10f801704bf59a0c9f5f584530c38b659e04bcf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.851197204Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/array_metadatas, Name: checkpoints/192/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/192/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=82a01bb5b13f7f2db91a0ac70da523f84e111859f3a47b702ebf1ee7d81ecadc3004de94e4991fe84a721257a2e4cd51eaf1ffa40fb4160b0b0f5699f611681aff20021d461c2df50dc4bd466e5c3fdb9b142e38f289c5024d6b31151a90c0b953d070801ab0d4f31c90bf09c906f1a162a06347decf0c4276bd1996fce5438a4fd2649dfd4ed0c75f7d53f4dbd545b3513e2e9390da0eba7cb8ccb4496a00621d5e74adbcbc2f268569cbfa0083820db89da0a7a63b87e020f24dabba7e1807102e36580abf91052d1c90ecb49fc4892feb9208c756c1f8587c88df3af74fe16b2649eb4196a5c4f4f91c77d27ae1c7b247b8b9e9691f90e5fdb8e0afd2b7a7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.851219637Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d, Name: checkpoints/3456/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3456/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=6f12a0e7da7cf59afd36072d7e412607761add204f775c731ad08dbd5673ca02269a02a7cf8af0458418e1400f3dba0e4ffc64f831ab432516e9ccb871a3e0ab0206fce59b215e434695e94b5d691e4358c4ebe68f3dfdf2d9182c75cc3561eca26542df2e6389a0ba582249827c85ce4f6167161d11681f38a7648c6fa9d281e9f364ace67ec30ec9fa92e7e61ef5a742bdbeaa2f157ea11509ff3868fb6d7d5016de1fce37d0df60fa78948ff8cd6464301926563cb590370a4a15f28d8bdeb1a133f787a920f066d974659b6d90ab1d4ca63c20a6d586f5059c03de0c806757169895c1a0082c075413e81f741b2aba06d5c688f381666b07509fdef05d40&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85126065Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/array_metadatas, Name: checkpoints/2304/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2304/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0e23be548be30fa23b0c479dcfe6afc35af04e1ce1987c3b63fb9b2330f42a989a494e0afe835f93ebfc6a7b605bcb8f2af42ae6df1312dd80cba432483637fc5c5c963c586c52f25f5ad58e10ea57951c1f6b34c1db8df70ebc0c3c030915eff312bcc4e256420bf4c639dd8e61edbd2f1d2c50f40e7f6e0ccc488cf2f6d1a19dece0ec0ab7a08d99e85b1a8c46912231421934ed1333494dbf974b794bfc017f648bf3baf969273f67f706e920554c8fa1773711835dafd301716533540e6c71a75e020f7f7fb0ff61a8d9d1234cfe89a0b2c43b9cf0b252263720af85d3f20d5f21f8c93ce1b85b5c3c7cb90c23f38614963df32e2a748dfc9fef5b9acd2c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85133328Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312, Name: checkpoints/5312, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5312?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2b45d7e482a0ee912f918f33fff78913135dc24f9a21220cc3acc981b8a150775eeac1574ca701cad5f628dcdb3d53ae0695ccd51d1a62a2914315328a8a27c258ec91c62e8bc9a79bd997f84f22ffbc3b8924ba9af02b5bf9506be045c4afef9a31628bc5650afbd6bf4d862da1ec21748de05b587e332b27aa10cdd8475d599f181611cb2db2181a5c9ec3e60c3b9705649fa57248f37929c847b7fdcc87799dabdcd561f5f2f82e1066faf4d4e21c7927736539ee3271292e973d93173c8350b45fe3321f1f191cbadf045dc3b9868e8c27be60d223b135fef3d2db910c1faac835971179b3a01374e041f86534d6fe10709b570b40d6ac772f789214aa5a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.851378662Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824, Name: checkpoints/5824, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5824?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=77b8acbd312049c31bb111e9b7411dddd45c5dfc65949f004260797c210ccb9e79676230def6b0547509c0a9be8ffef493942453adb405d06f9c0a713c72fee0c83b974f5e68b40a520813e32ba1e50b337a5ae80f452b2b5ef872f6fcf4f9ffda280cd0d985014823f1d3a80a4383a8d81fdf7cd905505179ba689b46726b7850c78b9ec8a9e5510e6e4fc9a1fbdb5302a8b09fee88a6d26b27f4885bd527aca2f14e5af68ee230ff7661bf868a75abc13defaea6c3ec1a88c059604a70385a6850b6139c6aeae0dfe095bc85a95f2285090a6d100c31d9971e7bc2a0b235a920f41939652e7480dd3f99f0a5388c8474994b2f2e304e384859f61e7d7a17b0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.850697305Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/array_metadatas, Name: checkpoints/384/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/384/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=89ab7e4b18cfe9ee006a98da0b5e71538a9a59211a9fc1b87dd0a0b2972d706022d6aa7968c3aa58c9ed611db6b1b75453f1156c7bf4369d3be299dab414e915da57ff0ffb721516c2ba27248ac6b55eed58469988db92c618ef63e3689ec1a3170f1eef6a504279b86d690d38c1cd2650e20bc2a469e53f43ffd22dc29816b1fa9b8756835eb2bd0ceaff0f480464320ad249abb33871a51e9d76b2a1ba7561686e0711d1b429e25b083284a87fb1dcabe80e1dd817bf7c0764e50fb27a62d774dc03c9ce1edb847d173839d627d9bcabd47ba6cddaf3b0eca504e381f1ea3556681d21f3d1bcfbad245c66ef3352575e8fa31cd293b7ff5151e1e566df5ce4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.849407223Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default, Name: checkpoints/1920/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1920/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0d69caba50d3463697585d70fd221613d0d9e427167c97cd0b21d00e6831a7e2a4272dc3eca04924b639f48ea314cd8ec036a98618db020659f457cffaec1fd9d1503d4ca1c13261d33e3ac182317366bc616e2692263205f7ddb5c61193fe408458ecdcc683dab552a8de4cf1bd400a0508c7adfd566c8276d15964cb2292503e4b2bed75815e85f31c57f99e0f0e426fbb7364de07fd055605f11aa4f064981dee363bff4594c3dacc4b72eae6f9876a41fe8eef34a5f2e6bffc173c3718693b776d72d0529c07392e2c70a796021f305598a6de4dedb269de37368a5c261a70028f1efdaf8212677c1510d88430a5be76f3b15393bb154b9e9bc7f76bd8fd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.850619244Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/d, Name: checkpoints/4096/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4096/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4f549bce27371357700a2076d6b247e909f16e0f084dcac9cf0778549853c06c0d645e6307f61aecea08da6d478867d6000bb26522d3e19810427dc8b29f5bbaf2a4d9b919b70d6f9ab5beabdb881d2645c751b76fe7e1e423d4e058b54270630c1dfafda66642e51455a8e64dcbcd7ab8b55085f4d8c05de9173a27872bbd482d71ee01fdf46b2112a833cdde9b6237d62d86669d20f6fb527f3ffbced94b35838c06712945242a3d58e2aad633e62cf73cfe1e1c7414970b3ca65eec4c6c880f38dd08fb51a9ee38b5e760313a51170dd280fb4723a0ef902c61e0c121f0e4c6dfc721276ca672c592495de450a7bcd1ec781dcf5bec1ede6e070974528f01&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.851534511Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default, Name: checkpoints/1984/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1984/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=826b42552c767ef94a0b2970afa1efe4716ae7ca9c289d6544ee66f9a86a46888846e4136abeb9ddf235a639b521c20da1581d108e5b0a36b5d4f1971ceb6a2377a38c9198b47525436d47ff1089aa86e198240c879c0fc1155c6f45d087d384b4d0e4f2fc4a40f2fa7b27d716777ea4b2abc51125890185d861cd57d9b9c7f4ba4ee4abef1295ded6f786f968cdc8659b8986580bc4e697c28342c08f6b6caf65a525154da09d36897fbf8b83ab4eb2990a42c3f178b67885e734f73779b89c815a7a4edd2acf67a4cb1a0388ed73fb187ff02e5f6dbd982d35750a927c34bf3f247ba7bbccc6a1e7812a700b1a02887001ce5ddd2f9aa8599ec2c485d1906e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.851579445Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648, Name: checkpoints/3648, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3648?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=7248b8a2cfa576f90b51536f3713e8e706ed4702ed158ee0ba8cf259986a05332581b58d3ea9d70712d3e0bf6fc26e2422d2d9c96d7fbad00f9912b3b55e6219b3722faec57f042f67fc3ca7d10c1b57a65b1131a8abe5d4a24da492a01529f27a0143f51235b99ae4105b6f943e46ce07265d54c60676acbbe835430706cda3f643328a1b46dae87a96987a2f6ff2a98b6fad250384a6de8c2a7a82e6d8d6a931a656434e5b15aa095f1c2db0c4b1503c1b6557f2c3ad96b9b9906ff4470c3ce8486024ef18c02006229fcb073b79b2d1c407f096d6dd5201ca5c744dfd2b80e666d2af7c226461af189089bff2871a6db1df7fef307c450634a896f1022276&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.851643186Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/array_metadatas, Name: checkpoints/768/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/768/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=41807f2c0cb320d9c33247990284e4414cbf012e71e9e2ad071d48504c40f1b7ad1a196a5e16beb101a3400169f0722aa252917648adeb68855f333f02825088629b991c15c6b737e07b21440f4d875a4f03f6b6d61503862f43af12475e8a0f4d928f9408a611c443c96084eea1424dd9d78c4b7e860f5e23fd17932b59dd252f1b8cac9caf4f403e0770914b3ef9b72796b1c0af1bf2520adca5aa9c48dd76580b390d4aaef92fa2011e360be811bf35c4b497755bee3eae7d4ca0ac33068621336925caecc2b1c0ba70bb9eedd441fff6a0475f6bb1a867b3ae4f4f9ea0f4b1217c0f17179e383e4eadbcd62af79cb11087609ee06dc5a4637e240b7c391d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.850867165Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0, Name: checkpoints/768/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/768/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b7779e07f724b15f1ce578a58eeb7b57a7d0b85aa8cdc2995f824ee16954b001d5408e0c3619501c529079d6c4241e285e318857ea16077397d7ed66fcd63d4d79b6e73129a920f80db4e860447b1651422bf85c69140b37abbf2c164264dcc3be1447746c5e198c06dd796c6622ac8e0225cd29efcab250631b3ba031fd08dcaa4d6cb9a323c5a7b10cdd26575451ef93e2e28155e720f17fa4a87869870664e9485368aa106362d4ab8393ba2cb02e9990d2bdf315c3d8115dbab903bd42cd584dae4f84e39012c269e6caaa16a0bb815474a764179ac080778c0abed291182ce1a65b9078bce1b1927a3446dc9ae61e2a7938571fbca0e00e9c18348cecdc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.851689993Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/array_metadatas, Name: checkpoints/3264/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3264/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=510d3365498ee5ff929225ad8ecbe59f09ef1c975c0a518d6935b481fe877680930dedebba5ea69b0badbbed3d446316088932f57ad186630c4f3e0b0044310379ab732ae742f2ccaeae8bd315ecd20a663e0bf6da8f18601db4fa415105425b56a339aadcef3c6d0c7aca4d48cf7f79fbc36e7f3d7d8d96737d56e83f68db409c649c34094f483c59d1d83f9d2941bcec44057ad5b2b993cd4df3cc6ebcd50d9246ca8f2bffc0ff7ff37aaca71ea118d7b049bda4caf193a7625dc739a8dd2722bb8089e108e551091a7566dde68f2aa3d1f64603a34bbc9e95a5bbb2a98be28cd2e5a5adad558270c9353de357b4ca9421d007060052b2251d10f0fb775dcd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.850604717Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0, Name: checkpoints/4672/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4672/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=8e0a3c89b627303e0a08e8fed45bf96d892458cc3e55fbdcf1bf40172d9554875faf53dfcb9e57a5ed6ce80effa534e84d71ab2e1d4fa1eddbf239e3f155f718c8798e92628a2d03590328328cbdb7019f1032a50dc3dafe63bc8c31e38e37ef0832a8ecb947d61a9406f6cf8c12bc16fa6fe0dea89b9c67ac8e278edb705425c5bbe260ca21b5b223198c1339e0f5840ea01eb572c176190d09ade6856c824020c8ed5452bc7188ba4447c5338efae9baec57cdef17c3b5132c515b7df4c03968fc517314993621f5c129ae6b21175c622fd9a44214221d5c87b1566aaa9a03a48cd20cca585d7e34277661abc81c883a972111e8742769d6cf835b50ceb849&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.851786661Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d, Name: checkpoints/0/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/0/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=88aea653baed77061a7f6f70aac46c64060629d613a21fd84f5538593d1b97782faa40bdc1b70ca85eaffc6d76c69c831500d51a1e757833fd2d55b7a4870dd7b784ae79f3467069bd4834294b2352fa3bb8b87e84ed13790fca73a224eab7bf31f6473204f7c7e9c263c0f0ac77ae1be332958e6a2053cc2aecd37f73aedece71fd94a01426158ea26b13c1d0b4514e0d5a71010d5f0be51a2cededcc10bc1fa2e7a775c92c68e9617f3c50b498e244fc24dc86fd994a098587c0b68e1eb28485a25811082e55bbce0e758f5d8e27dc02c7bdf929a51aa281fc378114c7284d0cb3d19801cb7ff6624c71681aaff3baba0f012169fbf0b4e2ef7410ed95edf6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.851804873Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/array_metadatas, Name: checkpoints/576/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/576/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=ba904f848ffa4d9426a1183a168391f93c0aafb583bdee0851f3d3aa90ff093355db90f19eab78d1c4e9a750f9b38f21ba6ed1cea5f69082d686c48ac572ad7f082ceb6747bcbf0b2d052eeab065825079699e3ec9173c7f8058fe21e7ee1738a322670e43ff7a5c2064f4b3aea20dcb3c86af370f16f9b2cbbf1002bcf04c7d069ea3abfd531d4edd9cebaad0ceffb99399a166b9626ddcec6872256f433324c12fec438cd41764e56d15e9f417aadb83669f6c43e94badf1abca4450f8c893a8229dc416fcca868c30d047c0927d0fec01c298a6851a889438121b1238d29765b8a2905bd164bcaecdb87f5165e0c4556a80fb62db555b8ae3f6372e7d2db3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.850765981Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/array_metadatas, Name: checkpoints/5696/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5696/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=73d8e968ab854846cc08bca25e2ec78ba9b7e1e93131aad3233886210e3c688889f3e50311309c2471736cf34f6249a5248ce5bc84b800f5df3f66ce1efa744a67834bc285f7edbcf0d91f0b60af6784f2a65ef16a438862cfd200c2be91dc6b3c124c0c2457a4326937e43285868a643281c7d67d8172f9bb4a418992479d2808799bc3e5f57a6283d44a6d10503c9657b9f599a73c592ac774f24c5714f802eb41b27f37103209cf71cb241dd1d908983bf22b54f590191fd8f503649de46ca0882a6d0d989d7a38e26a1a3d085b96600d2cc29a56cfdbcd7faa9325c277109608c394b32f3639f3332ee08ac5750cd69efd328ce6b5488238ab29aed1a9aa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.851884973Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0, Name: checkpoints/3136/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3136/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=86c51a2eb98c088daa3823f4158b7b9b4f68857119e64d4b9d4004309b796c33ffe554171c893439ea984c0406bfcdc7efa05de7007dfff57951528c2919a9fc8722a7900c5a429aad544918159cb6ed9523f593f7bb0eacff5628ff5cfdbd5821a77d1c6d0516eefd19df034d332a92aa338ea065738edd5c19f67e18336dd57e08aed293bfb8226e047a94dbc4e60db6b3bd10a2220ae1ecba481b183e599ee201232b3746e2133eb1e6a16d9ecebeee8a444a35f89e8d997548ee50d4b961b225563cc8787ec321e2b459e377d3f405d9fe047a36d6c204e4dcb98d80ed2f22b18eb6f0b996d92407545b7c63b737b93f9d92ecc58d7d46e2a73d2e904421&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.851509839Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736, Name: checkpoints/4736, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4736?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0611aba9953db6e7ab55da6573b2ec45d471e63b352beb4ca7ba3a87f65fdb3d949adf89728508a3f81742ebc7e6e5b612860f0562eae72489fa9103b1dff1ffb7c02cda866b707f77a2c9d9b000cbd5576e135808ea1da9e98af6578abf4ecb7ec677aa2c5e64c992a0dd11156d7e72f389b4afdf7f93157e877f5a6e0fb37493800a6d018c5a9034953cfb4c5500ea98cef1327a14e7bf53ea5c56e00f2cf605e9b3e9173ec4aefc5c03e65672e73ab2faa011fd6527fcaae1eead7606ba61bc10edfc843df9e2eed1b587f3c0fe3bf22b3c26fc61f43e0963fec43cf408773e0246a17d2d0f6e21017d0d8ad53b237243a1fa87275f3d279b743cc178cfd1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.851543932Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/d, Name: checkpoints/5696/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5696/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=86a52f44facd34307c0c50a874bf2a759b65f7cce431f945025facfdb887fc2e259276b6feca2b925405fbd7c440391e57db998be49c766799825f8d113082f3d1bb80cdceb91df1925026caabd481e1dc1454a43d392bc1cac9134e5b902531c29471ab1f7f6aefad1cfaf282c659bc7b9ae18e99c6666b929a157af9787987dc09dd87fc4ac4df4e4c3e2798882ad5931cb352f84a137e6829f1d6d415203ae3dec67577b210a4f2de114633cf7e30b166d3053d8d3404d89afc02d266a01c2cc31ef7164e173be1f32b155655c6dc90c58a512e8fbc03596b7512cb123f97ecc7d813e3f7011140bb88bf2f9994b0c785541df6e4f204c0372469937d072a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.851610429Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0, Name: checkpoints/4736/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4736/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=63290369aec1e74110deafd1e7324f24fed0f284032932ba0a7033afc8d4576274bc76ebf2698f7e6f6ff366b6941fb9e4b6dd75711f6d509573af6e94d8e5d2fb6e0a53ac262eac30a94c6c08ce1eb50d9a27af58d0a8105de634b83644ded0730bfbf2515909e1bd6c7a1a68f98da968390f3e9cfe70c33959f86d81cc4529d0b7643039d5a2ac2993a6524f561195315461152cdc651374e79641da967fed5ca1df6c086eb7bc9855b643b7b43d02f176ef34d7a3ed32d3869ac0887db28435a2b81e3101cfa39a37093b3e466b176d4051b332468b65bcb5d8d4607ee9c69b8f9371706564378e8eb855bac858337c0ff5869c716d5aacaace4da1c53b91&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.850841793Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/array_metadatas, Name: checkpoints/3712/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3712/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=41c37b5a55055f7a2bbb59fd7c8895624d52f8e78a04797bd1d1bc8d32c01e74d35d074b60d54d8c2edf1bed35aa1da37da2fdb5cc5c722ffac570be0e94effbf4fccad6149bd383bee9aa41c68cd8a49ab9147cab17f934a350b524b412ea9ef2dfe769da569b2c42a86ac068934ade2dcfeb7e8708f979eb0ded23555607b4cb8876e36e506251cba456b85b6df821359dbdd020133f5602070b4ea30240284f0ff1ff7ba7addd2e87ad067fa3afa5edb9c7275a944bd618a7dd5448e723afecdcd3b627e3e438ab15fb0a5452dcdb2807abf5c4dc7dc1a55ee10d485f8a4cbc7c2429d282f2730c0f13830d5dc95cf410c1425d58375d7631089010fe1774&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852031647Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512, Name: checkpoints/512, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/512?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4e0584cbd6efd319f2b47146711d1be0d2a2a5e57f020b29020e1475decff85ef7abcbc7d49f1053856b6b6dc0b0a49a80f9cbec454809ca32249b08913b24436424f40498d9e4528fd6f7c2320cea627d701b67b164450e4243687072c007cdde7b27870dbce35450de089090065aac9dc916c9e46666f3b8fbe2ad2cbf46ba41e1ff42b1de0175ac1c5bc8b886900dc217ac708277b428dc43f8a6b8a32d85ba7a6a204e083da7498cab03e22947dec01dd4b10fbb42fc57de3e29f8a84a9971627e17ede7b3d69679423e5de09b2ba5faa24eafa84cf2adcc30b845f7e36cace3bbfa087be6cf685a88958c4cc5cd8337295e0b17e6626ff4e696ef32978c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.850714926Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default, Name: checkpoints/2560/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2560/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2a59ccfe22263d2d58ed97426734271c7cad1da05b79f8a247a9d28e0800fb0a7b6e6b93befa2b2a71bbbc404d5fa3ebdd2bf2c4f5783caf9d6f123a5a70a1b84b79dafcccaf7e99c24005b62bbaee76906627578beff0713652cd12b7dfdb60a9be87c2d090f63d4fb9f9b1a405cef0f6f55ff4ba2e0237ed2b213d17a1021dedc97b870e40ba47be0dc0f74d4ec435db4b8ce7b5520899a5b690e78025ad1111eb00573984400383ed810e3f704422bfb5dab0f0a00bdc7a01128c8dbafd0c20bd8a48899f5b54126b085333c6bcd959508f6bd111898b25a95d98300633b1f33c31c369aacc4a179f54adfff1fa60659401b933b1fbdc3872764eab128e88&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.851716933Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/array_metadatas, Name: checkpoints/1856/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1856/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=519729f5a3236f26237251d1e7aa5a2a84a9596d1c33508999ac9935739ac95cb6f450d31fd63d9ef0fee1380a1d03d1274e07331c8316c9b6a1d148ae854d114095cb621ab1747c883a1d7810e552beacd78927fa6aad709d5c6d08b7b3886d7a925ac8cc2650c4251a11c928369a521b1f315cab5cb9a0cf3f5c9a1c4dcf97fea7b6377170da450ca87ce2ff893ca576027f0a1ee2a9e818d5e117e780e6fc9ba894224a96fea748ac4bab479ef9324fecff28a2bb9ee5366a502e9571f1ebc1a4e07b6d0113579d55cb48b624747939e5bfa49ab811117d3a653e8842133b204e30ef28c7b7c35b9317ac87bb123d0ee8e714a7b3a6ecd8178efbbadae650&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.851616576Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/d, Name: checkpoints/5760/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5760/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0571b00e9d4eb72d72ebfdd10e08306c2310a652f4f1c9b2e564678479df6d941b343451aac5fef5d3e6648cef7b3123347465516d613913468331ccd3debb8cc9e542190fe04493c518176f01d14dd1401571a6b3b7759af54f11954a76d5ce5fd3c5a3867b5fcea6cc5c5a606d1da145c645ad1e5604dcbe2a94ccd7abf459bf0cc926185983e9c9258fb9a418116d7553c5527c4544473505d553172f52f8bea2d48c3c7ceb52e065581ae55fcec4c101f65a35025ee19df426a7a0b5e5ab944f6d0d75a738427e7132476eab35f1e2ed3b8fceee3692310133025b1a1803660cbc622b69967da6d545ef0db2d44beb220a572f8f089759a1be5db3a652c4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852097384Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/array_metadatas, Name: checkpoints/5888/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5888/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=21d2bd060bbb17d5ad2ebc55e7ec5f62128717421b4eddc6b62a9ab10a9389bf1a3fd389d204e31716b250ce27dda6143d30196351494d2f2aa280759b4b030fbf19bd4976b85fc25a13104698768e4de360e5bdfe72828c197607f9f3e4c24f01c38c6d3ae1875f8ad0a76218f0024916b83e57df9101db24f674152b663d4713a366b64c191ebb3b478722ae3adbe9da55511a130f1353f4ab23ac77922da899541fd259cd8ccf5d99100c40fe30a7b13834be600205ea167b8f66a0be8e8901f4a2ce6951553bef06bf1b709ad13e5ed1c097e135ea20a64a8033b28de4fd09a7ad5a58cb9cf1a9fc337203a5bd393658956434d163476ef28eade4f2e6bc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.851860268Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/array_metadatas, Name: checkpoints/4928/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4928/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b1690d6de5455b30214805b9e1a1c47866252ce2c5b4281944d7c31e44e9118125f46b4a5da60355d60652fdbdd6824c4d6319955ececfe6ad244c694808a6d0b530bb2c0a41fa8ef7f43a097586691ba3c99225f00e12968d0571fbf746856eab9fda08fa3084d6846fb7e0b2b956130abdcc16d29ecaaf5568eda927d776f00f4e54e37012b4b92dff698a165e1cd163c65178fd1548a808627e0cafbc475482877632146a6653ab9cece8cf3e8da78757ada1b3667701b041148d7af939061abb20e039287b38fcf8a78a4c9b5d50bbe73a162a973d908d1b0355f2945c5a477368b3293eee043d843ed4c80e3284e62872beaf412c3e3f6c5acb180ee5b9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.851721372Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/array_metadatas, Name: checkpoints/3520/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3520/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9ba1925d7ecc2dc093091f56b82c27475b6a89c9d7744dcdfceb713092140152ac522a95cc46d0e1162ae582ae1aba323cc93b03f5d1ea67b3b671ff0f57374600acfebf837a60c43ef90ef41f631c9a5991ed6a7aebc74b18ee75f462eef54a8cb3a41e0c85a8bf4768943f73f3c37255ed904943e648c23fbcc58acbb721932fe9e3190130418b863779641e8d3fb63b7a83af929fa6856f5e7784795b1f5d348817322e6548bd6363f05324838889fbd25857fa3d0598b2bc7a60fc190dc08c204c602a1a464054ef8154b59ca0e53f0d57fc3fb23a4d28413bb295fd44e4d328075aae56b2637f73aa939c32d0906092425a3ecdec8d31a502c8b3a40292&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.851667917Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960, Name: checkpoints/960, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/960?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9bb95c5fe91ea23de79574e3eb81f51073d38cccf402dfc48918ecc5b94d7e14572cb759152d514e02c9c324e311253557a933aac3bc7fec9c310eb719b29224838ee6d1366eb58558011023d261a84247ba2d46eaec70491e103be0e534421ebf09ce087855e9cec7f70539d14e38394335cc717e7891f9ff0f02e7659727dcbdaa1846a451edb21e628089f3c943cb2eddf1a839282b5596e9bf7935771cd8d81496b44e9340c3a7625c7a826a1497a158c76e80863a8027855bf349be0b8d05a294356cfa5864ce067652b1be7650e5ed979d0c172bd4cb5ad51ee5c8fbd2f084f2b2663159f68cfe05d46c9235dba3dffc69012cd9dc8960b692100ca57a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852253964Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/d, Name: checkpoints/5120/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5120/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=212c4b3e9c62f661ef051b8b802c5e564bd114c9f2a993f3a791cfbc03551b18050f047d5d96b13cb86a0f1df90adc97d031e2e0318b950c5a2f2495f10a46994a068fa98bc251df0957ebdc344aa3cee3d71d5fd916e596d7327ecb13eb972a03a53a08f9d040d5287551ab5c6e28f1182e7ef896236b2b1186214eaaf748e98c81c176780b554b7cf869822f11afb8de8d5f9d2e06a133d89b73bfa406db01d1630d8a6ac608a059272832159368515d11d9c6a8223cd7727ca922f2c6558c34ec6719fcdf8871f90e89f12ac2bc605600183b1add5bb8891ec0229e02f644e602d40ed7b44add5cbf78bd60d69380f8a5837de041413a0cc5ddae1b507eff&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85060877Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0, Name: checkpoints/384/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/384/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a382f8c1f4a636ea666a12a382dabf34a54ebbe7b6578657e691fb79a810043260a9c48b7804aaf63ce0a3d109d1eee332496cf6d23d4186fd064f57fe04cd38dedbec0e6274c707af29cec7d0aca62ea3f27195d04a842d81e9f4fb0b3f438d9b1b319add5b4f80442b55c5f271d42a098a9a64a26a6146e4d0e1909a780ea9f175bc8fafc4135f86e4331b5a6e2c0de377e90ab256989974aa68d2d0f2ecc7b1b1de292d0ae72bbe03cde0b51b2eaf8fb212da4295cf92c7622a0f1d6055828a2e9a35007c097bcdb8883ca211a612d089118e3682393809581cb0ae31299dfd95c977c74b9a79e547ddc02b048a2f47ed07ca495ddcae42279107b35bf95d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852319803Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/array_metadatas, Name: checkpoints/4032/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4032/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=51019b47503b3bfcee1c1fe8acc5ef100793179216bc5e32144bd9d9f088985721a968e9814f0fec8ee59d4f393e87dab29d905624573c4ec9eb88c7af03ca0f5a33fb1df216012deec2e52c0b1f5d155c5cce811073ce0f35178d2f2b27a3cf042cf52b0396db261f7b657b951011c342a889d54744702424f0a10c9b8646075b66306ddb3bf40234323121fa3c906cdb231960ad8d47da8cbf11586c4953ccf009ad408ec89041929f54137eeef01183807394f46bca74efc7b8c9540a2c12f3473a40358e5f8f875b58d9138ce7281811fce368fd7b9034fe3d17b9f98693dfe01a894dcafe82562b59bfca55aa77b3bc993782fd0afcaa33514c806a4a8a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.851811961Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/array_metadatas, Name: checkpoints/4288/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4288/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=5b1b9b5e6a8395347bdbaf1055edb6d05ac350aa2670f5cf17c960e4c98bf3e403b5475c2da14514c6e6b07527614e9c31dae17a1b0fbf545c9d923f67b999ed871dfa316e8c32e605274d59fe5d20402a681cd5799bc47c72329779ce0e3694ea2d6281813b07b45812e3428f555f34329d24040d5ea4ccdb81429a95b0b00836962a276dde533ce8058483b2c6af71b4aa50f4a9273ac77bb2769064aab3f04ff5a06434707f456bcdb60326a700841991bed5b0129c425d1cc6c17f906a40971047ca9d66aa61c74dbc181a3bdbae74914280fe6e19b316edbd206a3b6d97b037b59b907db8b79fc1efb10a9dbe74573f1170d2189410b4e3bc40d4633233&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852422478Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default, Name: checkpoints/1216/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1216/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=bf01a0bd8d94177f32c50bb1326bb16632b787d30b6f0aef55a07b02c2eca4e1977a9162b71e948e356a5ab374ab75bc0a1664baaa97996c9f8b42626e5db0150d9f40e7c66adddc8994977dd2aeb07f88698867738cf6a4cb8d53e17edc13c9f718788d4088d813bb259b7e16ac5619a49d35c4bc7e43603cdbe46ba05d3067fe55367a60b19eebe1e736bedc21289735fad98b0fdc0c77b451fa6f5ed1a60a15c07d4a167ae8337b6feebdee5f1a15f4003153ac4b8269a812460dc392994a484fe5f175f79c296f7c67ba7b7c3566592792108f3b91ff1079948944fbf216d24f67f5b40aa917a625dad774b9542108bd71b66012b441622430499120691f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852463294Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256, Name: checkpoints/256, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/256?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=1b00e5de8a3f460bef4fab28d47aba91c26898949059c34841a73860ac33d4130d663a86a3e2cf8b95538e5f52b70697d2b5b38387e3768f5413be0ac993ff3ef6759e26aeb9136ba17e5e9f870e684d8e4160ede08f5f9fc3df275c65ca8804b9311ae7b26dcb5751d674aea52bad0433cdec71753da75c0253b098a928a2bb4340f55b7ced0dcc537b97a982cd9d4f4350bb98f755f687184f3fcaafc93d4b2e45ba9921d760517cd43acd0afd59f3dd023f6ae3b044cdcce24fb9cd2e9bee58e4dd49c56059ff81e1fb13d789df2df773edb6524338f24b0d549735a3e4bc29e7af1488dd91690349594eee9700dd91d55b790ed3f6a229c24e2a05ab2f6e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852476238Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d, Name: checkpoints/2880/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2880/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=3d19bf2a86e44e765076fcd12fcc928fda4172f7fbc547bdaba3e27b0520047d2faa41c59cc61384971334a8a879ebb2efb06c8bd33240ce1dbff0551cf9ed04ee8bff723e2a38c18aa526c833a9bd0ee17c2f22e5f80c827c8ad2fc0a3028ae64c778c95cf98825a25af8b740f10191a32ad80ff4785881d8fde16b416295f71e0b4b7ecc0b3acea1dd31f19962c8dfe4bfc8c856a4f434101f0cc6d1a3605a5a7a24f57cb649d287243f8d42425fb8f8192693b6cabd78329a4af48c064adfd6e092f4986ca7e3a1328748c942af173d64df529031cc9b51e7eb248e45b5389485e756fd195a2502b63511cd60011e7f2bcf3064970ceefdcf851394950c10&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852516283Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default, Name: checkpoints/5952/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5952/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=01ecacdbc8f8e38780c470513d798c1efef21524e61056697fc7eca0c82eae23d850dd106472c6aafd167de8196a18b22c56c733a3d5c7824d6ca52ff4f9047df8e3ea650d3d6065f07763fb9eb0faa7199cbf18a8ac3524f317026dbd5ea14cc42eb9c0ccb5ce7e8516250e9fa10608f9b6ede95d94da5478aa0c9614798a866546a004caa8d3d31adb56decf67aaafa0236dbec39b03c6bf9e4730c250b8157e2d9605bb373d25dc0ec7ffb2955b6d55a53a4b376990209f330a69421f66c857fea19b7b8c767a54eca7a9768cd4d0dfdffffae7835a10d9b4bddd899a0e8899f7c2d9c8aef09d5e5646361c07c4fb82920b1dc7c51cf176856be032a57bac&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852528862Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default, Name: checkpoints/3456/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3456/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=c1f33ef5ad59f2144c8c82873f8d7a2faea714f4adda2c7523b716102e882762520a9058a33bda113abd1c65e0212451a5eec2d4f96109b01884a042afea490bead71cf9a604b3d967ab8a79def29b92e8be55b4e786f0aef125dcd77c808e3a16212ebcd12d2544fafaa543a6f3be359ee42047849ab1330a24855b3ea85aa7451dd90c1faa301f3349b7daf9e38c12a5a1251562e18150f665242acfa7da41fc7b5cb794ce85d101f10ce3df7dddc4cba2559fc42da087ca408209a204693b866e1adb74b0404ea72326cd1fe64a39f57627aa8d36332c39c4d22e4980c7d3018c2108f28aa121a64bb2d5a07c945b60408a03e160c7bc77d3bbb87b38e3b1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852551608Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d, Name: checkpoints/3776/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3776/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=357419b25e98260aa5921ff6e23cb688d57762bc9b54f429006aaeb12538349a1fcf3dde946e8d79211da8377085f57dc3bdd9e912bba419c1c6638f86c2ae6616f37cc4e92dfe59f3a61c532bcff66fb4419045d579d51ec42039d9d047cfbee0124b8088c640b9f4d61aa5f0830ed29b9f35b78f997c8008440b287cd87296b0af58c34df4bdd0133935be9993cb7d38a8fc6b32f3e90612c1a5c247e348bd9950fa4d08fd432a6be30fe8b1f17f7ec05f98458b09e647a27586d8f9d6e15d10341e041a033d71f9691d82f8ea41cc27762d6cbb8159b73ffa0cd2f1b1a1cc9453107465f9e682e49acbca19931d879e3f1bab61f3f3b7af01dd11967db641&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852584575Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d, Name: checkpoints/5248/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5248/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0dbb8d3b71d085d81bf9b825a2bcfd86d4719fabd0277adb08e2a44a75103d1d43e2cd45ede68b3ee507c6b8a60369604ab2e39ff705074da5405f50b8dfe90ced94c80fb288f3de12a8bb185aa95cd35b5d0d7442275d6e08ef8e6725b9fca50247ae2462890e82f99566d32bb9dbd34d515e079a7a6bef70ed0dedf794c2a3507f7f63f004b18634f07b0f7573ac1dca407e07ce256ac1e80bee1256d79581512f3a6564e24fce8f2d815b023a0f1ab2516cf5db6c04b22358a399dc884553b1d51df8ad5a2360ad69b49407b2a353b97d6f3126dc28d85775b8407747e3969406227432ce993ab9abeeaa705322d035577de732a363912c6fdfac1d61563e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852615554Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d, Name: checkpoints/2944/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2944/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=12c00a379cc8f03d2f64ed78afce2689d1eb978f44c37cb3dc5428310ca08115551f54c8d8b6755691c4d3427dbb2c612b8b2781fdf8d963a44d8741589d09161780b16dce9040252631f2c6ac67b28531e83f78f999833914b7727db43eeb768c9bdf18dfb93dc6b73192cf696641716d0c3949c8c6473630c480e0314bd3c77b3b3cdbc000e7ac7dd71b77194aabb936038311304f51beb1f951c66b40c811a6ee23ca0cb5fb30779b3c48c220c0b00e11d08ae8c8f7d1eb5c5ee128762dcd0bf1a139e73f2c96859418750e55f8328fd27d9f0b88cab55e9b9ecf583e6e78a7fb3e056caf417f30d835b05e6183209c9da3937829a9e1c769a5d3d535dac3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852633775Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968, Name: checkpoints/3968, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3968?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=62af5cb400ad5d818e0dd64438336e319ea15c444ac8127c2456a614beaafa486464f0c8e51d24cdcf980d52506fcfd56526e06df91a6caf0a4591a5f5d5c703266198a739fc9364a90cb28fe7907d8dd050c41dedbf6b90ef54962ebc8d3dfaa9ca79848618459768b81bd1e3e50611e2acdb8155d88841aa75089035a7f040aaaa0187ad6adcb44aeda8a147ecc88d28f2f3f5560242cb1056c5cbff8246bef7270167d62050fab49c4b093052e36331aa1065b30c0afd776110aa9aa7238cbcf4c3e5c17603293473be473006ef7516ef3930387b6bc2af37bc22c32fb1f4700204acfe878b57f78fe3b5a6d41e5333f5b6501093f250c71d9aea97b630a2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852676631Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/array_metadatas, Name: checkpoints/1984/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1984/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=469d0844aafd8af769457307bd32a261667f7179ed018c9d0b3f6eb62f0030cca67fa1c3c2c59a7e44abf927f994c3c1ecedbe1541db0c21a09ef244d9d1d9a7142cde7e5410e1ccb3d0f11337ec0c72a94f9e4ed07b238e637b3b513b2cb750907a61da476cff1437a8846cd6627282509f02f5bc3a423739732382fd3bfdef558b426cdda037b6c9d40dd55bb047b25b018c4c922e5056c4c14aa5cd9a4168770c66bafd96295f8623c654ab5ba2516d7a9a8bd7dca53de1d7eac8c817d1e2d1087e6f016c4074734ccefded2df1bc8dd1b5b58f6acf7b4ad7a8ccde97a7f6dae77c8ca19b242ee606eb9566459bae084bc138fccad2e87f1a83e5dadef3f5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852710314Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d, Name: checkpoints/3712/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3712/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=6229db94edc80c0d67178c3e48bae7c2d845a7408375f40a277538c60a503e1b01f905650b97c74202e015ef3308e9dd1f6e36d639eb0fd6a3e988f753e83f1f79394ad9bccb3e74a96645d53ee9e2e68b42b031e13854255de8bfc276f616e613e3a645a73a7a2543598b72035918e43f682ae8e2e5bd06588396c17013eca5069c0a3abefaf242757b7fe09842718ce8f6d1adb07470e99c92032c1228d50b9a08e80cdb4c17bb041a61d12227405459a34e717e0af201b41f1478117791b5ed634989e408b458bcbfca67fcb5ee8a5f3801d0d61181a53631b90d0df5fd117e4c8347bce4eb27dbdc1270e25d3f0e6b6431ab7fd95a406c6f59c09f0474c7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85273534Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d, Name: checkpoints/1024/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1024/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=63c64268a5bcdd2f8c4985ae1e6f08aa8e05cfa89e7cf471632d5f096c37b70ee16ca0c0c7fdad36c9a820fd2c8a63c2990db145bae27019de3843a5fee4e542c0c0210e96e2332d93d943bd8b1c9eaf25a6939813cd13c9713eb3552a834f97453fe6e67bad4d7e75bdee24f6e591aaf973b6a273f240aec0c49970c5e7026f99886fe78f9754cb03c712954b639cdff1ad61b3c55eccb41920a5581cdb6221dc1e44d7309a63c5f24db422505af2bddaeb82f37179b4b1e825957887239aa5f48ddc4ef00d736199caff818496a823c4947f5131f265c7c972ab57ca657477b979e393cb42e7415f0458acde0e3301a49be067ea1172a7baa7e43c3e28a8c0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852752703Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/array_metadatas, Name: checkpoints/4480/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4480/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=94a5d8dc0fb638bcc5b26d75258354e7d9b625d3fbd2d3deb26f03b6185df34f3fe3bc1c23677c9e33c68d879c35358ab352f3cb426d75efd03660394081825df77db531d9e4e074844eded6fe85182daf821dd728c9f3daa88ec7d78234c83ce92478073df779a73aaa6dd137d7d00f586086edb0447c7151af59daaa081affad7e5a169905bddf479f034f3c3b4cd4b8dd80753d3c8d3fd92bd23ea3ca731eaa0ec0204151a551d0b9377044d26809334cbd86a0ce7db77379cd8c9345b07c5718748a3b257f407819e33af375fb63fc91ac2fbf51f4636832071cab2bc22c0744cfe7fdc5f98b20a409a492defcf74774c59cc305121634d4e69bd6795fe3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852763811Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default, Name: checkpoints/1664/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1664/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=90aaf8a1ad8d207022adefb110c778c43050fabe71c4f466760ccb4488da437b27b2d54cf5b25a333468bf5e8090a25d4810bf1e6f4a6488f54ac9ccf0acec5fe60d9e0fa0cfae678cdd615fab2b09f7476bc7cfc78b0ac3b12c384630eb047a14cf2f954a51df6e7e6b21bf57cf4f19a029183edd517b3045a2a0d2901e86dcf800ca2da1a4f9a7ac5f71e384f143c10250472bcaab291317f9998128a6c2da6a2a48ce28864ee101a79fa9b9642721cbf8da0ed450ac698a15905c2018b5b649b1f20dc0abe6165bfc90b530941668520622df91ea1871468df052d27788c0760290d6b6d88287728e9b1a58176081ab4bb9cb6d007e954a2ea5a3bc91f62d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852807131Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560, Name: checkpoints/2560, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2560?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=6c56731a8dd335f4748acc07c6f3f07509722e9d8975e7b3b6a35fde5525b5700a7149fe23c55ab7d6581b365b6edfdbc8508ba2a327c2a5c1994e9d1db196ad6ede28d03f50bf4af608ea43b13db6f026484e2282de9865b9c08e698c5895820ce262b912bb341be5949984c2efbd7c5b8e3653eaa8ce32025979685163eec78f0c4019220263b297914192708b8bea66c066894b0f0b2e5444143c5e84fd0a65268e202f16f5a0a8a1a34941bb75677f2a6e556ff32e09b5a55fb3c86b8eeefcf32e28c5048315fd5dab1661d4f3a626f96da727ba93fa3f66c8d979cd3060fde888cd2b8279da139f6bb095157b9a070e057d785ffb14d60cdab4998be295&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85281729Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/d, Name: checkpoints/5440/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5440/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=c004893296a4723d178585f59925351fadf0262e8429e3e1b18b1bb9134968e2d42295ea9438f2695537499c49c2681c69867c95d573b11934aa24c62fca4fefccf44027a8c3d2e216a17ce1e5d8705d1de11042e208fd8c1880fcf4467747c3d878d1e4a3255efc9bd8425130eb88acd92312044733f443c87862b4e94a29fcd547489c2f34fee0d4454cd7aad123c266e56310534c410461adf2566523bbaaed4ee995cfedb63ea6a45321eaed139e7626f8abe0281005f2ee00bfd048ab13d98fda6ee6d3df553a4f4c1efcf3e88b9894262d401140307ce7d079fa6a238d0c27d5df5f65f208bf465f8265212fc1c7f5b44d3aaacacf7906d1b6ef605943&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852837159Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default, Name: checkpoints/1792/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1792/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=63ddf4cabde9b30c5aa1ef18116d69a9f9927a67a411496c7d8476cc665e112d5b52d563dd3f7baa5c43c85ddf004983c055b8f429b6dea18db4f1dc8840366aaf56faef748c1643b30654daea26c84ab61d21bf6b2c98b63c643ee60943ea19c64ac4c6b1117dfac0be7084ece5dc4381ea23dadc8170ee0a129f68094163794c1ab3f4c78fca756e8a27d1937d0da7bf92197a8acbf579e71b399963603489969fa9857ffe0129ecc52d8541722b54bd2ff41eec6c58ea13db8c5eec23eefc1e63dd500a075edcd46d69e3ff1c2e7ac25f4bcf4b01093f5c3a6210c2f797078d055ac9e6305f70f745330267726ff0b88e31193278891b90fab9ef90c4cbe1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852877521Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0, Name: checkpoints/2176/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2176/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=23e3aef453c704ee2923419f79b450d4c681d78ae1738d0f0f3bf78f4a6e99fc9a9b4ac10dbde90eee31f6da0e4a8a34753dc4aa01a8305d4fa8d7d76d8970409e8467cd6fe0d69040a6a8418f9c7479e062b99df576d01c0a267ae59803d9dafcc1528890bfd2733534f61ff6a43149701330564f5ee7ff123a35eb50f3db06ae561e99124950737373b2c4c4e2ae8175c5979404859fd1fec32ee7c1a64895166562c7dbf30356d8a73c0e1627fbc76820d87fda2f1116d374b8a0ce48136b8e5cc001bd3214f9c047d98319df7e6a0395a0bcc25d24d47cc2936554aa223f81d91f162ac654472d8a26b3977ae3d526a7fe3929f84093c12066c3092ea215&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852850669Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0, Name: checkpoints/1344/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1344/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=3bdb2d71ae19503e38c6edcf5ea052cdbb7ad7fcdcf0f586ce33e4aedb62c47857cc398a960e8b02f6bae128b6a87d293e074f7547766c47a5c5bc95ee771f4d4cbee20c4ebc276092cb84ab319d62b0391d4ee4908ccda98f9fa6e7a08214477ef6a162be664861a16579f0ab50ffe3349be0cc47abab97cdd6414ae7f6e2323bd331e715765f2e4cfd0dbeb12d85840d5d40f72973ae3a6eedd3ee1c70345dce93666709bf4e16aafb6ad707dc1839de570b715f3dbee279c2e946208f4234047c840cf2dc56a764a03ed10c4bfe00583a5c27d8a72f9b1335c9d096a8bb55a4b48e15fade33ff26b3fbeef356561a2561a9769b5c8c555e13e2fd6461e9b3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852916947Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/d, Name: checkpoints/4608/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4608/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=33134973963674d37e7d1b5cbabd461f9417a50bc2afe5b4cfae087d2de848ba04e1ac37723be4aa85fe652a85bd8b0ca75b32d1155d77088113e452e6d507f1ee977e7da5279cadf34954d5673dc245ddaa87264d006a00efefc5a232002412b984e990890075caac55930948562135206ebf8de1002d730363807706cf71c7056418e9ea2ad223892adf591e2fa35378790285cc14dcd2401472c97f6562fd25536c27a24f542ae60887184e0ae031e3165f22df6c4174a70736f47bc58e65c6be4e9b5bae5fe84e0f68876b9e99e3f4680a481e958ae645e734ac7582f6951e3df5a36579d78928fda307a9f0b29a16d0e2d0812b5c9f4fe521b0ba6e052e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852959418Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/d, Name: checkpoints/1984/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1984/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0a44ed65a3ceaece88285c94f1d9155ed122c96b2f20edb51c2bd08204d57a17f443e46cd1a042817b823eef6c88a2c1d5be73ca4d66e2bc8e2fe0083061beede1846c83379a4d24d12907c74b1bfa17640000bae12bddab71a910f91b4740733b189d29dae52ac80e4228cfcf2b4e80e9b3bb917d884c47b55506e93dece888b0debf731280b6e12a8724d9809a80b41d5463ad7f4cc858836d7870d038e6b11eaf350604c0ccd9ca62041cca182a3d98d893f7d1f2ae5980b583842ec7338ce75e053af6692b91f99559af1014c4712dcc42ef41a1b4507f5e974de45d4c14fb40dcf21f3a3530c043a18db8927974fae1fb692a9c67d63d4641941576d72b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852987131Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448, Name: checkpoints/448, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/448?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=94bafb13b211ff6a5bf9e8fa7dae29dc76033ceb247562baccb0ea7ec37fb87e2a2b39ac07e0dd90d03b0df965e83959d722e124beab365a73dde13dff440b751210aa0016961da39f229009939a6627c8899aad80ec4929e152696d511f8a6bd1c21117b274960833d01d576112a0c674e70628a0e7a4d97e0842b898f3bd8e406fedba3d7d79a81ec5b74415b1bd6c14f7c8b78b6735a76fbf0e32ec833fa638923187b2705d2b7fd54bff3a285142b5a17215251364330cdffc502f7ff94723b5d45e2a61767ea08ae7ba3cc0983b665aed0ab6f7510faba249a5853b55bbd4ea7d06dab2e799bd1e94bb535e34a5d7fdafb39416f5f88b21d4372ef11503&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853000905Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d, Name: checkpoints/4096/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4096/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b1d674dc8e96803ae812a5e0ebcbb30e77c6a3b5d266df7adaa55f17512a5da348b9029da00e5967f4723f519057f1f87e75bf9a44deb78e0d1ea178a30c8447b97e5041331e20e196bfcbd1c67ecca31c97d4812c60e04c4604b2e6331c32adbdd21599feb0600cc7b1fbe4bc4cb1ef1baf57fdda914e6e1aa0890653c7739e6d9e703c383c033f9979406a9e3f59b6f746fdd7fc692642e40e35f659fb3d7c1fa196042261f3352cb8c8b8b78183560cc3f6207901a69d47327a0d2302757b752e40c3f0140ff502d1263443230f42d0b2ae01341bee871e1770537ad964c64136c3f9b1264d90adf1ca8921ab70d36bcdd7b69420e13e2e83fda6b6c26e5c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853034596Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default, Name: checkpoints/384/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/384/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=3d3bf14176880bfc59bc64fc22ca14d5dd04b7939e405e9648457a3f3112eb4cb194d5e6c6c4eb66e94043994230e04b55b417361ee98cbdcf9ae6584ff6f8680857d659c50e2136046388d70bd20c6e336e0c53a9b20a6b73fc713bdec1ed98855dfa20c3ceddf9550164b5aa8eb729913186134bae883d53c074f172f9a0ada42f884a354a7066612806a693721114ff1518dcaaa17523019f2051eb40da83c2b4347e665312b184c29c4a0600f700e605d3a05f60686d74596592ec2349aec3e83199f6502c54f44749613a517692e6fd7d4c34f9ac256c29663fe8347de1a68dd2401fdb1b979c701c8620606e1f09641d8761e758b939394ff73d01fd20&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853035059Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default, Name: checkpoints/5568/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5568/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=67e479dc7bf03d172ed762f8126269b8caad32a0ea3666f0807eafe9b6b9c5d29f72301e702700be96f3f90db8360a88f5051597cc93424b61ea9d48e1a917903e86eb747fe8ecf1742e0a74e2576601d1b8781d09d2e5ba3b16c683911c479e8f81f5c42f4a90cb6f4099aab0d6cd68dc7c0f0cdcbce07b6576afed0b47b72299c77e8ed50e67759ed265364b51a3a73334ab5c685b021b600b7185aa9cd68ed9e7ec1e793f5efbc5f1fc77c73687db1e5d5cdfb781054ed460963612b4657923c715026342b413ce49bde30343b99af5f32264928709955faaa158633f5fbc5caf04cb9053b3e0e8124055818723e1ee79a33c1ce169141f724809e52e83e7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853091022Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/d, Name: checkpoints/768/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/768/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4aa3a30d47a5c1605e9dfd487be9bfe2bae1fa9b80db7aa0f8019a33736f49ed1e75400449253c95f851dadbbe360a4299b09b5b066e7b41ad85d3160ae8d3d310a9e7d0726a062469de338c053f529fb5903249031a1eecec6de2df87ed32dfd46eddfd87bae8feb731b915f89450b2f37e7bdc8afd341ad84291d65ba19739f9a9cc83265cee935e2664c0a5a844671c8f4e77647c52610806217759f9569425be09b1efb1d9edd259b794d21315534844d5647cc13dbd9f732a7409f7cbb31ac57303fbc07f8b2abe25f13cfb9a77dec1a61bbbc6345e8d9a86c59ba1d27e4fff5bc12a18362547ab2c3aa7cc16979c37ddffd29b65651c8b78240c0378ce&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853110466Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default, Name: checkpoints/4544/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4544/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9169e82aa9674e4b2c5901f228a03dd37edae87f6a0eab42fee571dfd848e3415aa85cb8b5340533e1fa97cd9b12bb2523075456d28d2a92ab53773be045016172f28763c9e9ed7fc07148bb04db6cd36eb5c56f4f6d91a1fbae9847caba8d44eb9c5239a7fe09fa27fb6b0a8aede87d61ef98dc9edf8e089eb4930d22d3e3f5af1edc51b95f0598e143bfd2434aa2f68672ae5a7ed5469a33f18dd57eb21221e79e7adbe72659d7ed222982dd6896cfa574a63a6f3c9e91db293073944b15d26a7c343da300b30d6b09a7ba9b56a5d0a2f15edd68961112620247cda6cdef2c023091c54e7455f51e54e9f0b7c30610cd8b1b79ae48b09916f4f9b66a88da36&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852403815Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/d, Name: checkpoints/3456/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3456/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=ba0cff3661398c62be6f617e8ec4cf5b9e7a1dd8cef81f30fd4006bdeef9c9ff6e34d7f648b824b0d3eec80dd7fb2f6eac5754a4334deed71328de52a905ed7ff711c5347059962f959523aae5fc29be17e7caca8de526bf3826813479525f6f2c0688014aa97b84fec696c92eecab9e971b13801b547084d6712e33f0662abc02333ca7a9481bfaae6083b452cedf26157f511fce18bcba12291d59c4bfdd3691e63d98bfcfcb24474afe7578085c3d004808f745fb25faa12ccbccaf06d2d6db17b686f3d793cb1e75eca6831c9e02d8652a9f831b0d384cf487245d1c5b2a21bfca940162c83a4f89145ca542e9c1e9e90a6dcca6a193637bccf1668a684f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852554258Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default, Name: checkpoints/1408/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1408/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=54da608eeff34811febb2439eacf671c3bec1593b30438f744c94523f9bbcb9922c6516ebebbd5c24bfe1ce11cbaee3f69b95d44855e5e2059d76249e0018f1f55362458c17b335aef37dfcf2ff2e6e4b1ff4dea42d05db7dccd4803aa6991801098bbf4a2c77501e20167c88edac816649a8522d41885760563860432b0adb47f131d7a71fba8fd6eb883afbf62d6c562d786b5c1476d99420b5efa1fac584cda68d783df1ba9466cb677b82687f4e6549f61313030c08be8e8afe7d95aaa6374ea5a58bc1c5c8a54e1222ff43802dbac126b6c5c44704d7a16b99669c3aa5a8554bf0c8a2aa957715df4ff7c7acfdff3acd09e525465cc1e4ae26dfc2c7129&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853191708Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/array_metadatas, Name: checkpoints/2944/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2944/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=60da0a446af12fda1f53ea08a306fad0537fbaec3f1d8aac8b130c7637574e562039b5349f69fc3e887e862ad1ad53a3f8078341cb13ecee8f18f5d083938cb8126f3f2b2d390a6388a553da1f882b693fc4f5a7c8421b2f13c64700fb8a10a73a8fbee986544c178fb8cbcd45403dbe7158dc1b93e77ca28074b5aa6ebdbaf9bf91cefa5b30b2d7bf947680c470413ba2e197b74ca95cfce78e95d9dfe42456edb6bd6467a26a05723928c981ff08e1f73ea40e10c41b4b95086281ba83d2ef2bab88a8ad4ed8a9c49ab8dce030aee821431702d34d2199699b65aea490886a0fbf5c00d514f80d7adfc36e67a7b8e2700b4f19f5cf3b24040083c5528fc05b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853222085Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/array_metadatas, Name: checkpoints/4160/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4160/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=59b5543bf8220d99ba32c51ed2c8bd3b5c7bafa6b4c8a5958013311c5fccb8655952eb4b939dd310953525f24fd66ba061af047d31b378218d754dcf44ee047f321e9dd48ab5dff2dab4899f53d350a5e8671212a2f081acd5932c5a8f7efe530ae439f6a72b973906d30996b031eae9504c92e63a5c7df25a4279e35ca689089b835d6cc06b76ffc9a8af0c14927b0514b25e18aeaef99d4c8c5e90a11f1e80303e553ce05905b50c1d1c8fa4e1ecc1c3e68436e27810d86b56a86cb40e5248416f0d395ff42b01ffcfa32c60ca9815dcf3fa1479e0def20f75d3a9599c4f23dd6b07394a23a80c000d14cf37986155540be89853c06a7a2aea7760ec1acf44&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853237923Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0, Name: checkpoints/256/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/256/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0b75dd6c74419be6d0b0b326631b1f728b79c035f6d957cad27238213af4afeb7e7c6fe7d93c63ca5764aa6b2d80deecd80fcf350bfaa5efd4fc19bbeddcd3c319f33e48f5a88bad8df37e752da691b38c01deef88d2b93b87fd3cac05cec5bdb83ec2b6b7d0e2df11b250d3e1cb04871c6e1a0e8410d81e5ff207aad73dc54a36240d424d7885ae5522d48b5daacd6479f4f6f452d33a1e84b2cdac3158643085f72bd0b73bb60f0d80a29273df401a1f4d8d27498a4e02dd7485b8ac50b87a0237a394cfccb1b68dffdcbc8bf97cedadebc80335819ede4d13ebc29cd2021a17c72bbd3a3e21c9e7a831513602a0fcf073b5a64be97bbcebd9d029e36b00f8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853278203Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/array_metadatas, Name: checkpoints/1152/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1152/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2e3670d7a4ead44ef0648bf7214f88659d6df6f942cdb8073b63c85617b70c6e734e820d61190ffcc894bfa73f06290666362c53d062ac138735bc63694bb73b0e7bb3e2b265a7152c705300712c64771cf26349f0a15592da33f7076dbe72d028071a3ba9651f103c5842e538aafca46e06fc615e82f6bc060d312d700c855a6065aeabb0b402d5a62aa6cf0f9aabf7e8ed57c102d074374f13938f53644c829783fc2e33baed149b6559478026a9d2dfb479d9f27f84132d535364265505c999a2ae9c1b4d012095648af9d93e24f65c4cb6127679bfc5d5d934f16b1a73ba05d7dc315c30a1c22559c2a07e35a8176a0c72d108434d960f3695c17a80da57&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852621725Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/array_metadatas, Name: checkpoints/5824/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5824/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b161e375316a4e1452bfe46afec86061fbe83f9f46a192780546ed742330c53d1abd50b11a9372ecc3c161a41bacfb30f7ccb91cbe234270819a011546c6b38d120f90d0bbbfdc7e7c4a77ba5503f3e4adccdd52e389eac687ffefd4cd9b5758b5365d977bdffea861513b41660ab1b8cc782c80e17f88de3c0ef7f1e8e88444a80158d74c9b23ce6fc7103a6cd650974a174bf60518c53e7c53b832f3def99783369187d2fac8b88f9cbab21023f42b8b8abf059bd0edd58062ca895c3ba6ffe96f2974fc7c06c1949e5cff3dbb322286e01906bd2f585a376a750205648214d7e868083b93fa77a54eb72b347e8bb274506fd295dea9c1ab9040e853b6b1ff&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853199654Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d, Name: checkpoints/3968/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3968/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0bc53159dd4103786c12c6c6757503a4d9f11873973f4ab01f6cf47d04be664e365e4760d68e160d29f3d501fe3ba8dc71da3297cbe3d2ce859eb7eb70108cbff0d340b85595079b4b37c1948cd935b3124733fb185527985d2381249e620de3e090ef543a7c37a3f5feffda2c728f1b050010ee493c856e0b1e7ac7e723c513cfa7ae66d179ffadbc225b1f98b07b6d79f118dcb6c0892c1c84991ca18936df6c3a33479007343333f04bceababa8563371e40f2c9fe623ab5e9e636668c2d90f4c4919e1715f5ff3b87d2daab4442c24734a663f8380c8257471cfa217ac89d4b719bb1724f6ffe568eede14a7636a024b0dd90ac116e9ae0402e4b932b597&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85288885Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default, Name: checkpoints/3264/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3264/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b0d7728b4fe65c70f1e8eb6e85ec52fc672e22f23895629b6dabb4185aa808bd63a0d927b1aba1288e77487d4e332690114dfad6c5a81e7fb949a3f0f751fce1d8d263caa306a1fd1217f62bc220066ed47da5239678ea6e0bb2753cee0ef02ee31ec58aa259d647ad2fddfee9b8f3b9c42b94e7fa7e380c29c8de65e2ac1433abaa94d488a9e5505e02cf7597eb0f5df9a80a289664c52cddd840e4617be108eb573802191305cc2550548d7ea7a05f8e1521c430d8c7e801a3a70b5ac6091491a904dd7bf8115db3ecff334ff8ff9e3265b2de9534d31e996df4cf749fa7a858955a26988b470da3d42de22d8fcdc7264ef16f4563124af76d8dda8c777f7a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852732294Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d, Name: checkpoints/1088/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1088/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b56e04fc7941aae7913fa8d12cbf7e64a8e1a21157ef5f80c78de1ecabc3d4e05923a1a0fd7f1682d9b2528ade8029263754b31938be2bfb5f10e505b33b485bc92420d5a87fa77a229c0f21757ed41a4fb46e22a748975a2e15a9751e0c1ffbf6d52ced654a4cb1aa47b5152f7883d512d5f5275980db705f056a7e57bbd925d4a83be5178e23a9a0b150ee6b487914831f9ffbd62339592c9655633c0950b210a0e4e92d25b930dfa6bccf71d9150aca84842cff065493022b44db1ece024110d78e19c5027875d014073da2b0d45368a55652c6ff6cf04e007bda6ae38c8187291988324fa8029099734835f1bbd96bc731bf8a37ee23b5c90cb0038faa68&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853365706Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0, Name: checkpoints/3328/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3328/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=c306466c7d2e1950606926db3bf7e5437a4c8e5abb06d396509611c7f60abfca5826ab774ad36a5f8da58002c8fbfb64fabcb462fd6b1343f11c78d08915204966ae065f13cb00c3f728312e24bc27abbd7129116ccdffd642861da16fc565feb4893a2f33414161fcb3ad2dc0ce81d33eb86af820ac88a388f7f7c111a5f6f919c532bf41cd5d5401240be62e9d4395e87cbfeb4d3a50aca679add49939c6023d561429869763d0ce1089c76160daf329d7d87905f8d0eb640575a0c6029bb28dfe37a8390c321dfe647118d243d9118459825f19922cf99255ab564c9f66be0a65a59ec4dba310e09576182ffaddc2001207192d6cdcda4aedc5619e30af96&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853909854Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d, Name: checkpoints/4544/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4544/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=005e66bc71539a702aa18759cca429b15ec23ab5b740a13d605e4ce6409607591b6abe2eb3ad3e5bcd46f38786b2e9f5ebb7882cb5c0c2a67919cae877807b9538aaba3960675d4897c19e5065369606472879df679450b18ff28e997bfbb2266a6706d841cf1081f477e8e7f326193a7e87369ea155c818fe0e639e4fab1335d1df7efc3541b87918560c562b8da92fc84f81cd9e550b63f9311b19a8b18c0685a7b5b28a752b9b2f6fe9b01cdb52498193d6826b99b7ffa7e6519fbd320c27627047c6f6e8b53d48e978c24795b1fef43eae50191def2e228f1915c4bc1a34e69ebe052723bff8c906a2cd972453d0f9a9f70479ec453fd0d99b58be797afa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853922169Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384, Name: checkpoints/384, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/384?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=1c1e0d242a4d2444ed6c235579e75b0e61f613490c803e43b54c64a985d6a180df598e538cdc9b888d4616665b3a4fd549df1b2e4ce7c2bde99e02a8b36e62e0bbaa1021b58b499a7de879698e4e25ed2dd0499d219952a6a519d2e182b34809c558e9829a46abc641927ad0a40c245323d1455e16389739353ae5331b89dd226969e9cc475ec9d793a99012d3033f79f022d459ebf8c4b9333b79cd2e05d8e243c69a612b8353a3381d4de5daed5d569fc9f8350301ac339a7a16b394244641535f8b62fb7cc844f050f7a6b6a43046876f92ac12153b458c0a89f01c0063fe58e17e81d6da4c4b35cb1b3896bf2cf468fc309a6a88e5537d1803b509231f3c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852773784Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/d, Name: checkpoints/3136/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3136/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=78f7f0a9c8dc9ed9580fb9b265619e437c11eb6e602789c8ba2f47d248a2ca61b6fffcefc2f7f7333af5d609f50d0fb70f23c4d6697e547e0e4e14239c370274cc7b5b43af2e7ae8fb1c21196f98ee9ef0ef46177be02c405cea424c7c0ccfbf4833bf57738375ee31603d851f05c7bcc8640bf1a8289929cc2a0e710acbebb329ecc959f029c0935091c629f127404fcbb863dd8acf8b4534ed596242b66b7b7b2249ff2d36ad1c2c2932f127b717ebc0255858eb231458aa7f91a36773d495cb7fca695f4f350e3f32eb73bd19f58f2687f0515a4514a65ee304a3dae45faba2b1759b35a8dc9c7dddb761dac7a08fbdb6f894cbf014317b5c5fc77cc74f07&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852681001Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/array_metadatas, Name: checkpoints/4224/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4224/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4bfc761e83a3b57afb71d262b736d47a2155f2c6d18846fa4a92b0021d07c6c1905286200edcef148fc307f55c1235b24c6157d44e48c4891cfebc94dd23cf0c0fa7c23e1d5081d9740813a9314ee9eef41da0fd2cb62387a1742f4551ee599be3b6b6c52b965ff69567a3486e9fef2a19eb74417e6763c1b609946c42bfdc42b48ae8a0691a88def6f3d2c82bc6763b19458469509f51da046f59258bf639263acd8b3a50d56a5bc75751272ec1b8b830946c26dc4c158161165dd4f829d74e709a636444442da65dfc29a004172683dcabe5dfbe2cee2af4af7dd4ef37dabcf24949a32cf2c1566cd951f049a3cf3c32374d3a7a267a5b2ecf2d2972562275&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853971204Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default, Name: checkpoints/1280/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1280/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=acfbb0295253bee712ea003e477437542bc1d5048febc9b9a0ca56646812b6ac5d6177f4c4cd37a22ff56de20e42e4aac4061a904059f1ac612c984c20ce2abe68eae3ac6c26ddb51e5aac15a2107af730ada5ae1b6bb81af9c071022eaf1f32a874923f00b86f6b76ea78cfc46981d6c055bf0d61a2e2ebb41a12eb965e7a41563647ee559e83c42d0630bca6b0d9ea8c112a757d25e6ba47b4a09ba50c4587647c69c311309b10caa8428e2b8b314bb5b9d130b7f0fd5f96c794044a9b04647133c2195cbd19d3b0c4a0bd5f193f722076275749a0eb4cd254bf3cfc54f2bb9cb22f13f64f468b5fd5dc21e9ca4a531ebec257b26f5b11f22ae84a9b5b8927&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852839894Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/d, Name: checkpoints/4864/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4864/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=62d0c26566699087819049a6a42e1cded9acd7b96e5369a43bdcafcf58cad7867c0c5815f3d2ee419e90d9b24fce1bf8c759e4804f8f373ee0cc7a507c2caa9567bfd19567d1b8631e47f71c56f552ff4ab0907b582e3308f278cef71f8b7f858b607131d3c38ef2363d690b4eea8dd11e7fbf3f8097df1934b15d218fb6d5fbd9f5bd5234dc8cfe11d4836ef2c1b51f8f75b8536bd5aba48fb42e395c7aac5296ebe3567719abc6c10c5714ceca48de67abf75619bb6e7e9a22323d4ef1fb5b25be76692efc878e416298f32b71b847b277e9b752cb76f870a0d0e074e45f63e9892a3a41302929e63a6bae7f744b41dbae5cecb9ec28337681f7a4aae50b03&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85324354Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d, Name: checkpoints/4736/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4736/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=26d9ad7a770beb7f29b75d928066e4d1a09a66384ab3e97b69f56224df4e20be036fcd688a8decead42f49d25afa88a9687f4ea0ce56cd6f7bd688604053818b27deb93d217c79d5f1ab526a88e8f36276429f7a7003468f62696a6e091a02c154a7c46402c920a4fd560e177b0af2fa853b77608da03e7fd3e45486436003948810137a7191e0fe95982390222489eba59812649fc0dce505250ab3f06ec7d43320951c44783fb3da49faf7ac63e397cc1bb2a580935a8222267ac5972e2d2ab1cfbe8d61b5d115b2847208dec996a5159e74a6ee29993e08029f91d502df12b49264d370529e1aa6f6e89259736d9313ded7f004564c420e3a4878605f89e1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854004765Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928, Name: checkpoints/4928, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4928?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=439e2a7156e459f7bda6936e8cc542b5bd5f4b8ce0163e0fc7b870b6b71fe73aa53c6fc3a52ab5831755bff07a75c723e3364b717f3f93f62b5689b081142617d8028d57063da2375b024070bf18e3fe453c0429f74ab2c4c0fa120293bebd8baa51f9f79247f1c1403cd17a9502d5542edc21fb30acd073448c1315e127f05a5becb29d4bfd40d1e35316a72eae68ef7cb79ab9fbcc6cbe37a32379d8833b8e19695ea51f2dc284a0b77fb7f81f9a32fe3e34c6c051bc637374c316b33503a01f011c143dd7381eb150d9e6ca8ae2e560cb96fcec1837aebbb20a004d9011545db42d091bc2950e85c0d0a31f035b6b9d48027a9b27ef83cea32ec838589501&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854019653Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/d, Name: checkpoints/896/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/896/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4b59128696fb700d2fa62650304aed0e15aa2c19d6663c666df365bee09c123c7ba760cd37b1375cb300036374dc6678d6ae648b793049ce07cb8a838b4ac7a3bb9ce9cc47b8c06a07b89e27c55e1ecb5f33e29992e47ef1278aae03273125f790fa637a01e7f73c811b3597ee36b7dbf872ef4cd88c25c545c2a126aace0029e3bde15baf0d2381c1a4280aee8e701bb693312e0c7157843427e02ffaa0b542ebab1a7ca833a9226b50718c8f22b7ebea5556a76e3e9fe16acc0d0e56077fcc5da94298eeed558bba0b2bf4523a1ca17d8847debf58f9cbad488af5135c0d46d7993072d8f0c0e7189e5870eeb3a80a6d1edc0771473d78bc852ce5f91f26cb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854019415Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760, Name: checkpoints/5760, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5760?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=bf967c02666496c9ad3a7340af69d5e04f27987d1e7bd52ed6fd269989b6e17d7f5000bc2816a4e983d35c41ce997ef0effd1c1d382a676055ee9243835bff42f873331f689d865981ebe7baaa159925d1abef8333eab2e34ae99697aa5ac6a69d703eefaf6a96ac02890a1c958ace28359869dbbd9c9be6d3c356f87bf8b445bb4184948e370cd1c085414fb63f471d5d358919a0839812045a63021610c98901a2e34e01c05fafbdfda722531bf68e1fca8ce984c0afca74df749c8252867a89a85be62387d850ac464e19d676dc8e9a4aef0294618cf1d57e9f2e4cbc6bf7ea5c13f2135568e6289f49abab9335b7c93b82c50ac452460f798d8303f2c0c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852749874Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d, Name: checkpoints/1216/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1216/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=5dfc5cde3fcdaf332ba91aef4e9ca37a0a2e5094851aec93bf7f39b576356ef3fd3212926bbf149932e88c521253c0d7e638cc8d3e4a5cc274496365e93120fbdf5784320f7fce9fe5733b560646d9e97a25280de430c514423cdc9c21fae3b8657853b68fcd94a4f95f8cea0283b6a639d0c88e9f1787ab40cd7b341e36fc0aec7efe858df75e7e32c3b722b6ab62726a95343e3a9353d92e01af90d473f6a4806f82212dc2ecac1f84c01f82da95d94cb84ab9d622dd6fc76e3194ed734b10d439487cbc77a93220252a92407ea98b0faf514f04cbbeb16b582370716f0dcd39741def6b6e9afd4d1b75338ca1e88a7d7ad32b3c05f4e8406d4f697f93593e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852927087Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default, Name: checkpoints/576/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/576/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=48812b6397ca3446b489127127d4d4db4983875adbb526f7b9c83c22a8b0245c4ee967adb40f7f5c738168bd481fb092582cb6f1a879a27190c2258ee8bc71cba21e9d721582b1dd8c9fd361179fee0e0871c63121c434b36bee47cf434531df6667f54734e4f03e55debe33409d6c87cd4c9b0778debdc24f3732d4c3bba24cf4bdd0b7fef9810e3865de57936fa15fd779569cb9de387240374d01959614cfa5180eaa14101312015abf92eb39b1292b59346d46a8c425a189e06f644c74931a6e4bed494fa38bfa115e133ac3e2d4f85c06a8df69693c47c5be51667431815240aff66eb533d8f63c9c460ca07378c596b6f2b11a5e051d692165f495f57b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85400646Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/d, Name: checkpoints/448/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/448/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9a29d8c46db08a05e02479db01ca1b7d16be359657cafba06ba53ed2c1f91dda400a7513fa58ef0d27b84f5390bcb12f09945d75ff6c24b73ff41c63d908a8dfcd9cb4e4950f7c3295021046ee347eaee05598ff62a762d33765545c165ee6345f8b43c28e9db76430238b30815da91ec060561c61f2457ebe0500f3a9d8cabd138d0d20f3ad90b6deb891c4ceef488120e3d69804fa975f7d8128e5d6f93bad8ccabe1f1378eb5a995de7f243d3eb33f208750c0305b97915e37ef0cc8160f72772e2c8093fc4249653c53d980de3e915dabbfcf0aea35f7b737d21424379361c669080b3bd64d7d0992cdc5b9b0b66313aa9367070bc1d70b8d14ba1df8f77&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852844935Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default, Name: checkpoints/4096/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4096/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b8fb2bfc0672c51e397b3876fba56ab68ea63014f06f6e2a634b39a72117fc7a6341da80d773524b9e87cd5a66982ba0f02f08bb0131dfaeafb721c2a837f729b911f7cd099dbad2d242539e81775f2e51c38e8cf15994be0b1ca17c1b2f48a2131dd047e845a6240f08f6f0454f722be2e08cb5792d317cf8e28cfff95b68178a088abe5766ab9a4e6f7a2d222a28e843b44d58e8799b46898c23413c674f6cf40e522e91169826a8746a8cb025478e3f04687cb7126d051a854e969f3390fdda1caecbcc469add4bef3f9e5039f964d2578224625f960ad33619a877b77455d2f2768ad11f804cd223df4c842f62518742354731ef308aec65c7756c5ae18d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854105086Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default, Name: checkpoints/704/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/704/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0e69f09678f98c48c2a7909fd49328ec9f1348791898fd9088d2a46f09c41addafb63e570fe77f98241dd3e2e51152ae415f792d5ba6236acc9b02162ad86928ab1b8959af4534aeb59b6a27ff37b88d9af5208a1617a7a9b1b492c471c2f6cc7a6d45dc066886bb1b1dc6631191408dbcd478782c1181244943af5027bebaded97fc502d1e757e2c951b90b4aedc9d968800a3659d2ca0d51b97c2bf4130a029baed4eef1fd912105979f76a8dd6d62ff8fcc5f20edbc4d982cfd22555a4a6872dee75a0446d9f3f74433fe23f092e9a3266e37460409c1619f6a93dc3d797b46d7ec82026161509557db2e7ee0c427333c1dd43d3ce47257fd0244864ae7e0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853097825Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416, Name: checkpoints/4416, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4416?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=85f1bc1c37f7522d4696b6916d5ceb22ddcb7f8531d3529a6cd74d6a36a1ab752ac389536b121f7807e0693e38c1f60206fa268fb67a86e6dd337382188638022de1a50af26ffdb046f67c5fa6c09731e51249fb343ce99caf553522d0444e7ef7dd7d49360c18a7d2dc55c3095a035ff446b75fdef425ece99af7421d0fb051b349c0b19fbc3b45604cac8a892dedec06693dfb4a0ecdf4ccac3c853d107493bc2bd58f2ffa0799dcd1f61edf94cd24aa77870fe2e305a779354f8fed9ce50abc126a4d18380ad7eba0a7713a96889400d1c6cd514c357f749c91a885c69fc2e3ff04b00e1cbd0ca2f6ce7abb66a8dff79b94860bc87a554ea5d903d4cce1ff&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853101958Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d, Name: checkpoints/576/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/576/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=7c62b708bbe43c4cd522adc7a86b116fd382f5ba8b892307c44a9bc0a5e8d1ed7ae280f29e9ecfab5bf6c8cb2876d9a9cd02d19c284cbc67d9967021721702df9063d5c3e6405abb947f22c9bd4a88998c3247c4f0cf089352d488a473bdb23bc2aeac4fac37865d25d9e2e7c80bf67393e7372816466500da1323971cc7fa261a780a3f4ec19369f757bc1ffd0d652a609467096bc1d5f0388d9d646c041cd1afc599c8e48f9ff49059e83154204a457bbf20ff8c15dc4a8ffd7583fa295bdd8aa30d4200a7436a9536fbc77bf4a39a7955edc74168b888b0b585023ec8a23f0ee59edf94619de6e86f4f7beaccfa9c861483f93c0e8cbd6eee057eb4d331e7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852480327Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368, Name: checkpoints/2368, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2368?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=1287a81c0cf599c1eda89cc1902f914a2f99747268f5036eda2d2206883cf48c31250f6c6000e72c259533ee5ccd2c15bfaf2c0aff311b8e28d99bc837798a5835aafe0c6ff7b2c306c0e0cb31fe37d5ef4053b32d5615003d078198947380fae33448701839dd508de0696536e76fc601123b2251730d9761f34f4cd098b271aaeb9ef40c2069906e76fd8c01d58f5de5dadf5866616bd7b92e0b659c7e5ac2c9e6bf049c1d445921602a0c9b7806da7f87ff379d748118e2f8ac4410b664d9272365e8d45afb261220d483eaad8d58c688907036ff27bc205c9151acc89e08ed6d581c9cbc86ada1c45aea60cb951c6457fe341a358f70ffbb7911bff50f96&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853163668Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0, Name: checkpoints/1216/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1216/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9f50fe704d188aff5180b7d069c2d1d3c481112858baf5b1d657a3ca6b1c7b78d3236f8a5d0f3c4b24cc12b5b7135882ca005eb91c6a0bdb2f562039e6cb42f3fcd7c1d450b9435631cad934e53d92a665da699a026f5ae24ec0326b9749ecae2e9c522ef498867731ccb65cf06effdb3105c18892c1ce7ac3bc638aba07225b8fb614a623a38d15382a4a638c0f7307b7b4fec73f4a0ffa45e47bec0fd497f9a046898c41e55ecf94397fcc4569843e2e90c23136bd51858654135a9b187752c34b51efaa2fca2c94aa4678c9b8dc92e2d6538d0bf023555cd64c3008c4f9ef462d4b504f0b4607a902b5219b60421db807674b2c4208c538a15600af72b6e3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854152255Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/d, Name: checkpoints/1024/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1024/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=ae3fa88d8b4620c7479795fef8aa29dc71c7b01b2123f94ffea1d5f38846715a5b109bf186d947401e324db34c7051f35afc794d28fbcf815569140c775218c0c79d4911144e3f7e1f8a8551f63fc0d7ac7032dcc23b54badc0c4cb5861f8172158fcf8884d99b7a0ec8fc6c93c67aba3d38d63bbd2e644821e549e78cd11ab1f352963be5f8832e2b0fd5020df3289bae7c62159f0d2f12f05984a5b9f9f5f5492cd5d7259a9962f2095f2a35cdd84798fe36d5b4f0bb4d46f6bfcd27593358fdd292fa565050398de82339d715178c72b637117e875651d8373e24d1c489e560283d5bc5b208b744744edc18925665c4199c4a815e658ce3a07cd2cb4c79a6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854129975Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d, Name: checkpoints/768/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/768/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a4d9dd1dc4e507a1c904091cb137c3709c16d583d277a86055df6984451e0b4f483fd623ea6612db82865f511469b607b4c809c061a0a613c404a174d0ea33faf209e58fc54f727d7a6ae6df4442cc9313564561467e52c6f44d3abb58862b02d88530e2e771354d6ddcdd5b2911c7f7553b9901949777e09adffa062757a227f65ea9356c26b17a623ec39eb9002a3ed7544bca9b47a55ad479b2b89c241f2fd0a502796c94273e2551431903b6c9a7b1641252e4f9c671ff62812ca348110e571809c1e377a3aa2504cab912d73d0b00f6df0655fe407e941b39a46d4544c2f56b32d9f33b046dcd17884f3319dbdb96d8e1323fee2b6f0ccef5bc62888e13&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853167196Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0, Name: checkpoints/3392/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3392/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=904f06bef5e079405a8a6e6dcaf1d43d461e5cf5d5adb594e7c4206128c874bc70e1866ed0046b48bcd999e5b4b1973e79c93d7a16b32d735a1583a106d0cb7816808e352384ed3a026e11f60a689601033870146ecf33d63d9e385318ae70eb656eeb574d90f99b8a8523950fde653b6867786f63b0950bdb9fab8ec6c0860b645ef9e7b67817890819ed0e33d01b2730c93f3645ad65c147a799e6b5fbdc486f2c4f11582921fd13a2f5284913a502d3e1248ca6063f8ca272ae4830d0b461bd84918bcb90f36a53896699be92e037493c345633a121ed82b4e0ed6a6e6507d16df3cb80db9f9bf2fc5bace2d1b7f974028283fc8c789aba7043b6099598d7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853172949Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/d, Name: checkpoints/2816/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2816/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=1e974d537627b207d8a126876f8508150726c8ced545a64ebe8b128c72782979e91f4f895c346f71c3492d8b7bf3f55133df458860f066922088a878d8e8f69f7a2f51cb19336d8f2b44a7ce25130907886c71731e6ecbabaec29683233f27dbeebc9a298a6a5b649b1a722a2a620e0c4787c82662d4291e27dd8399209715f41e9fe3677e3fdd99bb185c996db4b6c34dc9beaff1e9cb8b4149d20096103c5b405ee8c8a80d8f0ecb34807e2bf10bfae227961f485412059e0ed77ca6fd2fde739e3c16464d49af84dc1e3762dfdce03d396caad0bb7f65fd3b96eb08a7d45a4dea7982190dd09af6805bed0fba22d0ee95512016cf289bc9a91df5c86c2aee&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854246702Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default, Name: checkpoints/3776/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3776/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0832c9fb84f3f0f5c189fce77b929586b7c842414cd491b6a967c477fa0db882576508a04740556d2e522baefc26eeeaafdc47b47badb512560b276d4bcc2c913d60b78403250ec8a1268aa4ce2d390a37c5448178d50426f8e65f9df5f7f1776a853619411ef69fefc5ecce8f099cbf780901e961339a0720c48f0b0896a5330d741ae605456aa3f554f1359977172b48d119a6040c2c3a8abc0868a7c3aad1a9065039f10bf3ace0fa9cbb168be83be056f60fd4a0b430ed3ac8dfccce258d9e0c815cb63a32dbd0dc1a5c3ed9edb2883159189f752dd0cd95fa94cf6bf31466bcf6819e6a5d9aa718eb8e730934d3bc34bf0a311b7fd3411693d7463f172c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854235787Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0, Name: checkpoints/1792/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1792/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=c42e99947db51eb01940ae5e87c30e413aebea68d75193748473467c3fd1e51ed6ba87c4f56f6b8d2c56a66b4f032673075993d0a6a0a809b6918d264c57b6a41268467e99bd80005ed0eb53614591a02c9e95f40ad73499a3778241f0c88441ae676fe32bf97ffa8fec0c186f7b2fa50d09f1b86c238427296777e74fce6e972e82edddaaf2011a75cd850d7dc2c57fb72e8a354c3e478e7a8582f8355f4544d95d42e97c6516a471767746cdbfb92d5ada5f2b392bbaf86d0923787483af86a55ae091b21a33ca4e6f6d1338b0819fadc93337f60f6f351d71fe199d40e6b943c27709c3a6190226da1ebcb0507b0dbb11568967caedabe8289194103cfce5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852556356Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/d, Name: checkpoints/1344/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1344/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=296e0ecf87086b9a8028a827765cd8cd125ea5b81eab136ec4aa4911a3cdaa3f4d9cfb480f0a58b16e092962d614dc699d4027b1bc3a173b0b3028265fca6b8cd0593bd294c3a2476eb1472183ff225c09c733d4e763ee11af83040a5bf312fb123da252c091326e6b9cc6cc4a6e82e0119583e9c7344bf4781aee9f5bf6c734981a032e1bd21d24de86082d28c81ff22abe2247a7c65c47130e148e6a10fb97bbfb06aba21426fccb78139ee611c80d4a32a4fb007f290cd03ba2987efc32c16949381f27d7a8ae3ba1df27b63c88553b3bb94ea8627c4f7e8ebaf511050a6c0e74dff06310b6eb700d5a67323c8ce337ed625a68abede9d95fd6c14838f531&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852746213Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default, Name: checkpoints/5120/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5120/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9db5b84112d5fb8920aec05cc48b176e3ebc87eb57a2d786a43ad5f02101732ce437dbea96affd62caaa281dd4f824e70ae6362102f7953a15247b730cdfa2005560a81279a068c3f0ebc89041c6192bb2a8060b639a09754d67a61b709553f82c816af4fa8680062f9de488a20318cfa6f9a262ed8945099abbf10b8e2e459fe0ded491fc0fdf5478e9b752ba07aca9c37ccf0b5f3ceac6a0d43611d2de5a3f33efe66d7798ca1076bddb62f34c05ee7ce72041c5f775371c5a623d5e6285adbb4bbc1081c57164d65e02fa175a96070894c1adfd8d9c8fedc5b04dc8c465a711b63b5d507df5fa74f93b37f2bec58761b5aa35cfc75153b51ba0dea72e719c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853368609Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default, Name: checkpoints/3008/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3008/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4992c2511e84dac9def153915f380082df41d39fdd2a2c1abb089a51807fc22e7aa46e5225520315fa11ba2dab5fa20b177f38f02526f580f908b21dd3fee09c14ae2b37175da95443f638b5b871736a05109294a992f4f4753d08a57226e805c847b5b04a0ef3bb842f3deefd09332958a22a2f24d1fb46747715650eea40f1dec06f20073f3e281ab0553689140a81fa16f81b3896cf85dec588322a9033bd2f025fe0024a9d772596012075dc14495a9167b6648c6d030996044457e953b50721ce27e8bcdfd7fb83d3e58f586dd696315507a7f75cb2af7ab203c4ad5370614a9c8e8a98a4828a0eaf11e4e0f1daace6bb2d3d6c1e704843f8af73d8d49d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854333802Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d, Name: checkpoints/5696/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5696/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=89fbe62ab6cdb9ce2864963abd3de6c6088bd1fac8947d3aa1f4991074169da37d0739c67228ec007654b7c65e5c39127314ca1888c0f90faa984a85b7f42ef4c2cf5fd7836659b76d5302d98d97593557114cc5a7291958d0cdf60c21dcee158987980e3b3209625314b5225b4748ac06cd18ee163471a3db8aa28847d41a62f19bcd614b58fab21feecd21cb0959f02d1b3134622a3bf3fdcd460f63d0d9cd46aafdf56e9bfce2067f3c8aac4419c83d298c08cda3eeafb160bf668dab72d92b4d89f2f3da008f57fd373493f62336321e64797b72d840a6a9c53300b50f5201830e2d4f928bd184a757306416d9ddf013ee5bfa27192064dadcb1bf703bb9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852622309Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0, Name: checkpoints/1152/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1152/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9232a3d6c4b77ce366174e9a273814b0d94327ae27e57a84542b111e338e4655b0a72ef48b7c5a5a66a5ca32c63c35366f44a5be06fcddbc199289edd362381225bcfdc929a483951add82251191a9b343ba88310e4453f7a8fee12185c5f21520530d418b059e3a99112de963930313d5ebff169896dbe680ced7303d48ce755480d0695243ce9f0f2a0976cd15638d3a257b6258dd8c9a9debbbda8e5ca97ad2d403d84f815ecc90dd6f36ed84cbdc3d4b3acf9b96e3cdd24c7842246f46741d6355ac235ca351913c434dd2d57f0faa264904e0ab816302db4d4bb865ba8599c51cfd18239f52315aa99ff1c00102d97b6d6b00009634a463dac7d343172c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853323131Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/array_metadatas, Name: checkpoints/1280/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1280/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=666400453192b7efb191e0f97b19255c9340a83880369da9924e935dcbe95e1585d15526d2e0f2652d54bd680f9670659fc8c1ca77746a58e5a462645554cbe34a25cc800e998d0b454c8330bf8a39436dacce6ed9ee9e6af3952bcb1c3b3c588ff6898ffcdb10a7209e8d7acf519fe3ce8ac40601459d2ce1c3a695b0422658a6693cd4d04deb0cb2877f1ce05eab163156e93678717678d528390df6743794c125cdec161d60a1acb2fcddb328733738068834a036ea6674de781473e8a69301336fa8808c74473d0ab1a6fb476933a93327416ef9ad99a6859a0520d7349cf1d9a9ff7d2d013c5f8184e3e0597429031dd35078664df8601cf316655c3bc1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854408201Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/array_metadatas, Name: checkpoints/2176/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2176/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=1e6c3145d3b0cb75fdf575a1375bf80662e8d758cd59f52b25ce44ab9ed6c14402e8db88c46db890c43e5d8140813cc77551319edf2e319e473a44c1c2b7933ed07cca03baf911b01af38e633ebbbeaadf1750586de16abd78e4133824e013760a02fd79cf5b91b6e9fb949fc334c0138d2ab1e46c807f85a63cb1b55bc995cd0ce15d01e2a232202abdfdbaf116047dbcfe52076aab2d5bd49e54c94090da9d1ed584360a50d4370cd0e3b1afd6d0dda4f5de1b806aec8fe0eb2ee5c234578a4ecf964f5de75361f4b8c6a13d2ae9e8958e159ab57f5a16cbbb50e45393ac0693b4854ec1c08baf6845f8e2a0523a92e1222de237cd4a705a97f6354b347c1c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854404547Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920, Name: checkpoints/1920, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1920?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a29b7471e3ec1b736317fef47ebf16c2f65994b99e21da465435463709e2ec6c23f7a9ca9d14faefded3196187b9aacddb3ec8d8a9193a90f89cb1f1db1df830798ca0e798f3a8f99bd4e91c731d297ee24920c6845a8b05ae6f7c7425d12021a46fd6ec11aefc70ee3bd4239ec99f67c31f505ea770c925d50a5f57762c6fb79ffa7605a4d27fbbeff4200fe2a9fd7ef22c74ad24b35f0b1b786a27d0ebeba8f97c23080f6a54f42b4d591d343d34b40cf18be932cd206919dc4381a972cdf17956cad54aca905d15053b9e2f2d0e9f53245163d6c4938777c17d19953544bbdbf1a4d71d65a3bb758ae5a7cb4888e4cf499fbcbcd72596086b609c616bb93c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852788894Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default, Name: checkpoints/5440/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5440/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=35ee12f7aa68f0645d8784be35d3bd9773905cc96ea5441fb4682476199b8fe18988908f239414085cef005ef97fde912b333fb7dd253a269e50bb171a277946c5550c160079e5308ac27c76a6a4610a4a133823267053f996d12f9e36b5a778b69372119c7f411e43bbd3849532463190978f90dd95f09056e3fbdf1834d58dc1682a1958cb5a4af08775d79f470035267801d8fefbb9c086ed643ca7b2824fbcbc0fc45c1fbb0d4ff301e9bf83a2e566879cdb9554590650d55b05c4e3910db79a679974fd4c3d71d439ea89786fdb6e9844c58ef9e295962467dee310ecc1819511a4be9110e213c6aeb83ccf71d713c985103744ef763f6c7db37bb495bc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854458352Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0, Name: checkpoints/4480/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4480/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=75a1e170083bbd4c244eae947a1ee3f6efb77d979b6afc9d889a5bf0813aba61a5efe94d3484f680807863c68db438db6bdd7fc756a1dc2029e4264208445c455c532229714004c4426816f2b68182015d9c500e5e1834bf32e2be5d87496a8017665bbf3cd12b112cc4f7d3e18f3abec4db8f1139c4f391349dd0c7ab564745700a32ebabef0ef3348ee7203f40e41cedb4127d4f5c8e1a6616455bb4672d740b3ad80d98c9806f9ce744322f15a0f5c10e798557468470a69ce30f1fb0f6da830652f2140bb59914a0948faa4be94543a23b5f4872cf13177b2ef024ffa88343e026d5bcae8d9afbc7464ab8b6152565d3eb326c85dbea2064fa4d5417e193&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854339273Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/d, Name: checkpoints/1920/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1920/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9a361324a778c314b91cfd79ebf446b50074057fe7836a27a9191500e49fbfc1184d4262628526897e8e09487d43d925bba615b63169fc44204628c872679e091bb9155102e45f94d3f29cc2b5360764705ae03fbb1d84211189670ec84a341ee4c4b0343fa1a9195d1e014c488f1c71fdb559a082258426f6a4a106c8bf3e1f5138b4dd0cf2b0b0351e6eaac649313f3763b51cd307305856624119e199b97d27a728edfe34dd617d3babfc62fce9e5c50f5ed81b5ef2d397bb2017ddf76e5609f516455eed0f79ed01eb768e7af7201616a388234052c6d58c17232f2adf8a81b5738b37e61f6c326113ca9798f967e26ef5b7472a601432562e05c5bbd45e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854029092Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d, Name: checkpoints/2048/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2048/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2065319becba32b506451c95dc0f0abe550bd88b7e3357dc1273e0c45061e61e55d7806007789d9e4f99b5a4f966cc92202ad2421829d8e1d118ee07eb7555a64149a8973036c609b6fa479446e24bdef81090245430211d3f6d943af44a44016faf0198f26431b718c613ddfb14b815b01c67e7ade2bf6683e4299b79484d103c9e677599a99ab95641b720105aaf877efef57ff2a0143df3d87df851b7edb48f15495ed9e897f7d4bd90b3ddf75f75e9cf9c7b7dd8919ff23ebebcfbd646462a84386cf17fe98668b1ad62edd98283138ba93075d5e23d726f2a7e0beb7167d72edccb6ac0b59c617156fe7de185da3738f6102fa91d9ee7b5c6d06675b8a5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854471928Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/array_metadatas, Name: checkpoints/256/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/256/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=5725b35e8dd198caf620da76cb7ab95e767f2e3aa0f3ed02214cbc44966273f40aa54ab47735ef3fca343ec0d6db544c6d9af6c8160c5148ff5a40c95824a37cbb32ae2524645b465d7d9af2e104aeff6977e11a843b32845dd81081cbdb6fa1c8d528093284a557978f2b7a1ceeeccf3e372915bd6d484f521670a4d347fb967dafc36468c7b2034527f3e67bd23cd761baef6607f6c4896d7860fc9a8023aa8141710477d92d7fda3b39b2049c76602cc97be7c210fb1005ed9540537b6be4a93a8772fed6ef76695db8883030cabf67b26a22a78c5a41c0d1f1dea145a461b65f28097594bf7d7e0561060e3ba27b6492052ba47bfaa9d5b4482e3959f6c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854043949Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/array_metadatas, Name: checkpoints/3328/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3328/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=5c60c2ec68e5c13a6d1529ee2145f807a6a261be3c2984c5749b984f1077ee70acf6b2056f72fa77c15753aa51322d0c1fbdaa49bbf7bfaad71d7667ef861d43035529c2b79bccfcca88d6cdbf6ee0afbd06ecf20a7d13cef28cd962ae6f1bfe8b0eb62d3eb8a1ea7c95413a12824981a8622bb37e547369b14480e85b9ce3cb1f04b1f318175e115e365505fbf18b56a982e2e3d0d8d8a7bee417777fb42524e9db47a007652cab4add19b931623d55e8997af1f06a3c0f1dd0fab3c5479342938171245efcedf9106b83f94548ca57902052eff152d2e13c331818e026efe44ea30d45074945e60323742d289fe14ff42d441027be963436afc3910f6bb30e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854531312Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/array_metadatas, Name: checkpoints/5376/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5376/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=1bbd5c4098742e9d8bde0cb5e5f08cfbc428659638d057655738bdf61943b8c2e93b867615131f523751743c9973d8d17392b20f28465a4e522ddb52d144179a8c0bba9654874bb158218f6c895ace0228eb6fa617d639ccf451748e2bd03bf8b80b217b087d71e259ddfbec9f82190df47640e94b1657f8cc0578f6398956d69eecb974d48afede4fce1789ffa53b763a855103c0e2368239765986de0ae59c1cfa6888aaae7ee0bb966924d3a7ce131185aca608d53f301feef3b845c4523edcd99ce1c16aa8cbba91af3ff1d5417515b08451d5473968e41e60bb63998e15ba0a872c030c39e2a6e8183a71b0bee286b86ca5790beb7585da31867e00a8eb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854058749Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952, Name: checkpoints/5952, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5952?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=022b1d999bcb47c8a02a79918108cef2e17048b4a6c6b022387247ddd24b3ca5b1574340ba5ab5ca0c14742c2488ca28529c5605b7ccb9de5ea35bc78ebd9482e95a166fcb778c461ad9c14adc899ee48a0de7a15b0dba5957e7c9d62df1cc524fe4ab250904db56bd0d269269c4120935502330246bbca6bebb8d857855130f8d172457be604b0d12f29610c4f28101b73c8b97fb5c725e31e61c91e6a6953b22e3d013b0deacacda32bd8a2b449bfd664a02c69f71e174c3b068b27dee3073c2fd76253119b0ded39389ff24ee65ece7a506365d8994f96d76d526a1bdbdc82bf7b46b448506eed0acad49c45bc66f915b7303ec9d91252db1b7c601207e6c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85410272Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/d, Name: checkpoints/192/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/192/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=987eae136a28a35f4b2b1e2c7a11a769494a78fc0f782aec57733fdd72922097ce1116f17f9636d9e44682de1eb5b4471f8e0e1db6a1904c03c6c98bed49548b9598eb47e7d60917cd0c9cb283eeaa77acd6aebcf4968e5667e3051ca279aed4d7d4167e23ad6080a56f48175fe44dc5cf1716dd0eca075c9c581b9593c5cd74f4e74fca90e407adb41aa0a30a4bf361d05c3b98193de666d1568425d6f9141e56a873490d0d7ab6c812950b5b5a3b585ca79a4939a652713bc0c4aa5acd1fd31677d9270ef24b272f835c78cd9754cc6f2712916b753ce8ffc00116a2772aa5de7e738eb46bdc92f97322965c658e7d648100a722c60c3a950147f9bb921144&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854130005Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0, Name: checkpoints/5824/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5824/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=5dfd5c1afd8c066bcc5624b95acf81d8571c3ba31816b0f75cdbfdd4032fddba3e81d6f8716f516cead9977bceb7311882b2266c76ee428434d186244462c1c5430d348eb20f57a47b3465c9355c70dfbb12d7417d68ca05898f5a26945857f02f7ec6980a2c719869b32dc01709e6300b570bdf1444dcbfac13d9dbb4c612d85796d9ce812c00f0375193d9002f149220ad35f4e6b04310f5a9b2a376bb6415c8a5361662412c37c6e3cdfcf311cab3396d615c2a89d4c0a4759a2ef0033df28e702cb5d44be580e4629e24840409e27a8b30bcf1b8a57eba446995bdb246f082f2808b99ec8e30e3593453d66d45375a606885c308245f137328a7f8153067&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854564045Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0, Name: checkpoints/1536/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1536/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=6b8f6109ce21330953d90ac85b762f9246ae09865ff94e0505abd03a03d71ba5a558871ccaa39c8193ff3aa1343746a80d8c791d41a0dbcac043544e162553e5b22b4308e3a8acfdc195b80830356b13f27334fc36c457db6af617a8e5c96000acd47aab202ee56ec572db8f4b501accf7b39d27499fe555344bdfd4ec68ff16d5d7519b4c56afd658cd05046a9d6d6aca7b5b8df1549fadd35d193ced78702622529dbb26390b47ec5f87c9ecbf9fdd9eafad4317cf08c71a271cef2f30767077ac1219c22f7e621d4395458da063cc8318516eeac2af6aa30d4721bbb9c77e1e1ade856d36180deba6f9677df4928216814ed414f6f5351e50f385378e462b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854603067Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/array_metadatas, Name: checkpoints/1024/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1024/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4899a5444df882568198ed19bd5c861d7cbb8fbe890927ae70044c91a5090467b5281696a9687d40747e0f6ab79b348d385041d0699b022bbb7a8f95b62c4fb3ca6d6ee532a4db1d358824c6039d2241bc6f2f9f0dde5168903327a6086d043951b7e7467cdf63543ad29e786f02917491f5b044f5f879dc41f50202aa04e842d1e967b70a48a34f588aafb4c1e752215c7e9f49b0aabed567bc2ed34dc9d9e5ab6c7a4642caff6247cd2f81d1ae50eaef0995a7a22da02dbb355eec0da3f098795df2da3315fce7ee96646d7ee2f348bfc3d88bc20cd1e74cb60fc25fdd2c85f054ecaa70d430cfcf891f4f712ff4aa14b7b55e2006ae34eab70dad66ed0fa4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854176627Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default, Name: checkpoints/1152/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1152/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=3c91fe2d6859ad5ea1016404842fd016fcfe2cc39b8b74c1d783109f8b4cbf9332c7ffbaf40dde46901add5208f8f03bcc997956a260053f29dec2eb151fc84c4103d22db7a6d015e51ecab833c6dd04e2ee91d79f01664442f933b13ce6c250910ddb5dc1cf205d5f40dc156fb768c312eb9ac59255ada1019197a0d2ad34ccf73f3bb1133b94fec0f5377d6838582316fcbb672e4476ae50a619f62bacb3e43df3d37ae19b7aeb0f180761614338b63fe635988d65dbd7c4209fd626eb0dc2cdeac400a3bfe06dcb48f0eca16f91be963b9d34165978ded3201022ac9c9a3977dc3072247ee63997b55d2395cd8b11b5ce229485ee1520657337bd13c40c9f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85462332Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default, Name: checkpoints/2304/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2304/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=1e209a839ad73a02f49c7779f9da7df5fbabd90d665dd1ab3bb9f9ccfbf257498d5371043032fb8130f815f43cc0d0fd36c9ac939807f8dfa787e52f1465d6dd698ee973912281511094f16e06101a0dcc2f3bfa521b335b30174a7b41b47c8c22c2707a212418d4df4edf0e4b1a2a72827b92535ee1b08d57d295baab1700dd38082b72863ece3f3639b0f3448ec1cb7a51aea420e519cd5908d9eef1109831f600dd2fa131a96ed632c909d28c40846686ecc469751cb935254f48a8d73cb81c40962a55d60b28a7c2be9b0cdaa630dfe3960bb7f8602621598407462e206534d71b925911c24f8e37dd54dd37d36ba9630b4b4dad1a8caa3ab2ab17f91abe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852620819Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/array_metadatas, Name: checkpoints/1792/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1792/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=7aec6c982d8bb51d55a5124741cf99a8dd95d92dd7bf5738371d415d065f91eda58bb803f1d6bcf2f3d440f533218bd760a958bd950b218f66c481df3c58dfd4e692a4cbc7361b2f3854ff1a46e0e2d55715af0a6350b75fedd51e472aa5e7b4563a8ed0a35087c4b8d4d368415ffe012f1498bfa5bb5c08fa8ac47fa65cb245cd9b66108e0fe9181b04ee10946bab7cbfa751dc2e4c6d18d188a07fe1b8fbb57348c6d0587cfe7a819d73cfd69e5b608c808c5a3db504efaa768c58b2484774d85504a487db8f1deb8a0d440e4a70197ac97908c8ddf928d3e908855bb0df1e6192db08b04980b8640a489330869af06f8b46dca7e40210940190a5e19f6a4c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854661607Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/d, Name: checkpoints/3328/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3328/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=19cadd8e5823931b7ef80d561a51caa817863837ca4cc5c55295bb5ccc62addb9b104610b210eefa91868f30f35817332a639928edf9c17c599db8adc3681e292a9ae935bac44387a2345153eb5cd4864515411782aa39a0cd3486a5d607c948dc15af71f99640295dfe9254e724b8f2d82557a879d6291a6b255db1eb2193ff5914fb4ef1e4ceb98a20980372a4f75c232c132cd0b858c80c12659df038dab20f0a0886d00d789c6bcc14611eebedd152db17603f99b743cbda69440541ae37b8b9af8048ad6bd80217db4338e3ccc5c356154773175f1dd8a5bbe5def12ca21f639c2bd40250c8aac753bc4fea72e96fdbd79f8a25e44e90c906f7d6d30537&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854320199Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default, Name: checkpoints/448/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/448/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b415d14da59f87d1200127c5081844c32285aff545a816bbfa3e85fdceb69158d38ef92d184503534eceddb25249a5f2e14186ff174231cca23842a91c0957c58a80c8b3145966d325169d730c5c8869c1b893ea7a00b4d1da2d032d7cae540858ffc46c5e5eba75c0fca2132aaa3bff75b2a4b9b68cd727704699be8b109a08f9e13006399aae9b56b88cca4575168ee3c83111f80a1430581ea8b6d663aea71cad9a41fd193fef94492a681f55bcd15dcfa170be30b0572e776b317e74a064c37012461fd1f743481d3e64b1588e71d22017b8284a060386b474a04ae5c69bf1fb07e0a17295641175a672898e4644249d6f09379400ea6675d9e2e17cb249&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854395723Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d, Name: checkpoints/5824/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5824/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=57e1bd0c6b9ad038ee7ff5e929fc421982a5c9ff00cd5af00cdeb234376b68f63b4c539fa0711b1e020a8e93caa583462a15aeef0bbc91bd1eae1a12e70966c72ad14c2d56ef28b492617334b181cc6a6f771b1b6984415eaed56c6f280380c80e1f768687deef1d9a86303b1ded86a5db6ce37ce7db83d2528de6475ad95d1be973bbee832b00272ffc7bc7c9117112af60887d485a80837495bb5a21ecffc45236c47f54cd0632c5ffd3f8c87d796ff149ae2fbfdc4a7f3a52eac14272a3c4b23238d60451b746db28a0eb7168ef014aa7996110e9764323f45ff4f277fc8cafe8e0b1038cd18f204a489b65a3ca2a17b48848ce4c74940a75ff198e6b0ded&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85443961Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480, Name: checkpoints/4480, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4480?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=851dad022014d65e0f1e5ce193f91cd3fb79f323d25fa8e9ba07e14a49879c17d1d37aaefb4cc8756fc3865a0a73be03be07fed9c32273831d8b84833af88db0bfc5a66c658d4c72be1bd5358399327e5de5a5ca353955d214ffd99f32de7a4e139bd69c38c5471be16569ba03e8f5ef36afe5208f2d5757122f620f772a7232d6f7e30f0116f6696ec7a10eca11ff91748bb4160b69f70091e9963e28a604a69532f231659e57cfe6c401a1e9873c8c47af176c98a300297e90f3c122dac8c883c4bbaf9cbe446a2f9d6b2c80bd01c3ba9289381fd6b28884c3899afb0b605d0b53a90f1862c596d0840c1ee64a5e0b4acb9b89a8bfebe6350d103a93b32820&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854713145Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/d, Name: checkpoints/3776/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3776/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=6ea36b54d16e81b447f6e71b1dae1d427b7c55cc9fdd9140f20487d446622f3efb4267d40cea5494cd868acf904619826401aad1eceaa1ead405bfcb9101d4a66350623fefefac503b837154cc16b7c2bf7c7d261bf0c781983d993a0f309a05c2d757430a20ad57544d624250b44f7b82106e6491113ae75ba8597bdec125b3a1ade96ed20e1c9cf4efd827188c24725c5311dee1569e88cd18b6cff3f96145a3eee5c1f8d63f13199c1651ca26b32a9a676ce576e332efd01d0d880a720571d03a2495773f3415d31179ee2167bb9b321cb92aa46cad0bd6d6bc6493ee5b7ea35a7e68d44f8acd1a531f6dc409a537b073fe67fa52bf5cbcd0cb45f1f3e4e3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854727108Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0, Name: checkpoints/128/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/128/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9c87231bc238dde839253bac35a236eb563bf153d82a2a2abe5ccd74b5d800719677e7be6863b6fd033cfbf23bcc7df28ffc686a67f5ed44f6692802e139b1e4c78314732ce9c092602ca5c5fa84c25eb998824ab6ecf9a36088864feac574636e9821579c0c8f036b74a1406e9c48f8b2d4aa0f32bf54c26488904bbe227db6b5ccfe9393b846605433622f2afa2d8c3e9114a419948909b1c9274031d4f088cc0bd4bed3f43f270b596958e81d2bfe9e923fbcad39d2da5142bf4b27420ad45efef4c82d7439ac8f3e87455ddebf0edbbfafd7de715c9c721f941dc0e282e88cc2b26dbcf01283ad5bfab1713ad6fb349946a2cfda364dcd1816b58ead52e6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854475123Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d, Name: checkpoints/4928/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4928/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=387a1c7aa52a9ca7a40ce4654adba68cc0d6f9b361fbd35f3c0e4e5f3354e204effe9c92dce4176f04fd46e56dd5bd0e6cefca62d5b5d97437c9e6af059555dfbc181ad8ea9a7331333251408bbd02a5c8e933a60214aea979ec13d32e7cb06be34116f2d3b57de202d78811dce7ac8fc1ed4cbc337ef8937659641891e29ff5975ce7157afbf15af7308f44443d8f409e9aabf415af70a4c97a33781883ee6556d01eab38bbb77315049d39d01c325faf808efd58bbefd328158aa63816f43ac632e7002f767c48eca3b5c135d1dbe7b7981bec500be8677e060d9dadfea848160800eb36f3db9b03265bb7aee41205e0730177be42d8f3b18923bf95738293&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854771112Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832, Name: checkpoints/832, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/832?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9dabbe4366395b9bfad80beed2521789f188a55c4a75ec7db256ade1fbab2fee2d0c76e622e9b489d16cd27f8a4272e12e597479d04dbd44fec339ffc3d308e4f05aac05c9a79d7fdf39fe0528055b0c779ba750f89a35f069d83484c636e35fbb1140388142370ac3e276e5e427cc154e868df1edce5a391fca31fb697fb7df8b976811366507b380aa2faa4fdbd74717b29e44c603a45539d326d8644d2b29f8e5fd2949afddee840d66de2fc13178679e8b4302a1f8b4bab0c530bae6b84ab11af5ce3dea2652e7a0bb80ddd11e52b645f7c6ad40463632b479ba368b55a2148530e0c0fb31cfc80fba81d3eba0cacfe81279b5929d35bfafaf418408f367&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854820305Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d, Name: checkpoints/1600/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1600/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=c48ac5a138c5a733e069435ff3e81a22cd1444fb0ecf7af381373df49231542e8c8c34a3b302573373c0b508328a69291bb1df6bd75f6b38a43d968f275971368b185d244151f0b8fa01e2fc8220f588621003fc86899d04ca260444557caaa825727824f4a192c612723d4531766ae60c4f5d5e15053d333b4dfcdd3dfdf9d835c64436f978cab9ade60642fb9e1facee942f8a092f652cde85df1819dac93cc81081329bb57e3cd90e7154dc932b922c84064045cee6c6f3ed3305f747375c9b832837a51cf1b1983ebf636168d60a5e3620ed5254e7f9c960fc5d7b37ce7a9236f3ad860dbdbf4b3a7c542fbe6fd699691f40d8e7300e79d5c1f5dd6d2bc0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.8548137Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632, Name: checkpoints/5632, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5632?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=294dc21ed6c258908f8201f4c14a783b8dbeb14e439070864682aa1a88596b549fe6b009d122920e68b4524882eed70eea78dbb129dd8a2cd1895f93eaeb531ffd062be0ecac1089516c6854ce2f409ca5bc9f25e82d5865759a469e20829b94a2faf414251f1771e093abf74a9b5eecc63a05b4bf4d13f6af6eabc15aff046473274f78003429f8be714a1b23ac1b1e3d838d561a9a675a02299db7ce31e8c1fcecb6293f41003d394781fa66a19013ff3bd65b71dae7bd18cdd2c460f1d1c9a68be7470d7ff5b22f1cccbf6a1e96427bfe31389786283a1333ea7026732c4c9f833f9d580d34dec824523ca4d51af7c423437fc029632741f383992e8f9d98&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.852843142Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0, Name: checkpoints/2496/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2496/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=7cf65f66c015fbe8d9a8ad2007286b44c3fdfddb82eb81f5f5b83aca66c53688d60ee757a9ef8f71a250871feab9c0aa760f45b6207b482efddad7edf1a9218a264d561ddd95c8bbf639d60ee3ef541ad18cb08c0d7c4f22dc1ae280b8a6167daaa863b921c83254f11a2d7b0e831e038ffefab0e79ecb419de45bd60e1f4fc5ea71cfbf7c0482660d4cfe212013f274c5ecfc7f8f4bcff22830ee269cbccfbdeab768f546d29f5ac11f351130fdc3f756941b0d55039b35780797f6fe2f099ec15c51f5d6697cff49a0d6792f5a71c4cbcee71c55304fefb4daa1cdcd1346f52fdb1f3eea9d0066c8a3fdd1199ed897076fee0e2ed55a0632fd1bc46abf0312&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854820686Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0, Name: checkpoints/5376/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5376/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=575f9e0bd2efd943dde5c3e22ade26bf43dc502db6065a613ac4b9f61e297b42fab0b0dd08aa2c065711fa8c97adcc0ab1a5aba31695332b706ff787f9e841fa011d1bad75ff806297b9a229520d89735216121d742364d1d1353f00da6ade403fa6cb32af95cd92991efa2bc832f9d4279f52a53deafeab26747ae3fdb51c06756eaac7a06ae156a016875c98bde928efbf3cf99fd3b880312011d57908306849eff09baa005a941fa4d867edeea90282d9be67c9ebdd48c3ffc95e270e054bac073c92dffc0bc0529d734cead838b7e195f4c4eca5a0fb5c88c51a428844753dd386dc886d776bce53f5d3714597ee1a304d393e937651c47533746529fc02&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855078345Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0, Name: checkpoints/4352/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4352/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=37d1986dd1e8c687889c534c0ccee42ad65fefa0ba01cd007c304e685e43be4aee8c3b156227d0dea52f453455c0c884692bab6c9a3023aa70763df963be3642b91d120b3909707165b3a47effcbfa1b5a8cf6ddb1742b79da15c820851062699f1d7cd3bca69dc601bf0dfbdb3afbd3242b810c9869a4391e43e70c1562fae235463e6e9c7088ca0ee22b212f7b1f7a2674f3698043a7bd81dee06adab341a7f548c278235ad62adae627d8a3dd708db9ca11987bab3f782b43d04c91f80ceac85c223f966a366c448fbba8f968b20ab2a3a6d874413641016c88483e2b78f177f14da2addac7486d64102e3687b40e3ebe45c0f35895382db36d5b1f17ae0e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855097468Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/d, Name: checkpoints/1792/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1792/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=afaf22c539ada993df99fb15f045c60714582cb04ca4725e04a77b0e0edd2bb760c335a88f9439533c482f9bc162a2e1db9f4a67e8e5ad0666d3f69916fac93346b8616bbd193633f1e3b90fd8ba6655b77c16a14446c0cea92e7054e35bfebd420e22d62e43161945c359a8458395d4bcc2f0e7b175779a4154a7d641db0fbf02addad4ed6ae1d8efcb1124f3cbb088e4bb0b5f2c909ee1c2054b69e71ba523e538cf97670cb2a0ad34adbfd0b8fec6c7a5235d7a8ff9244a4ec4149f7baec46e9127108ec00a04d723d0c7e7f5cbec3155d9127901907534e698114c2bd47efe238c9b8b7456d90821f0b280fac2d9b3a689a1c7ff7fc0b269a6b7094b3d75&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853042975Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/d, Name: checkpoints/5824/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5824/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=5dff3863e042540e6b63e05cd97d5c1f68750a4b30cde34ebede1e176af626f23b447db11e87ce2395f7d46d671b3be6203908e0cd212356be05327655dc1de3121f7a57d202b2e6020f9d163c073ae26627c647f4ad6bb98cb5922c2130450096066df0a0070656e1e97fbf1fd9d28300ecad1bcc91a6dca48c9763023aafc9be90802c18e87a32b64d883cc19af76f647caa207d1d9c19931145db424b460ade860cb574deabf70c5f90c24e8c69d97f582bf3decc485b3c776e923d800248ebf2ea5d3a2d204b26d82afdedf5348bf363602fd3f678d796872e8dae1ecaf419968d726cc4affa5d27670b701cd8716cf81bc2d5eb31e1320418069d096387&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85514196Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0, Name: checkpoints/5056/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5056/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=8cf264f33db934ead448033f565e610f4bcf86cf675b8a06ad2deeb060e04ce80ea4d5e7b4ac16823db971c1dc1c0aae00ab92ec098483e24895036ff70e65240d94abf99426fa66c90cd775555ed4f1e591b6231769332cd3c2febe2c92d46ca969cd1083f398c049126ba3702d3a12cb19e5dfe5187a04fd12e9af91102f2982be694fc5dc58bb7c3d0622c7526b931259a75d53dac5135542c5dcde5822b8e9b559c2ecc1c43ef01952721172083376918b948ef90314e79600cacc62b7bec30d00bacacb5e75226e09ee9db2d7c9b50b49b8d6e7940c0af76299a71ff3e7ec2549a33b946b9b50ac68ce4ea59ca807fbcd0b1d8d1851d5262335dd03c99d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853095612Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default, Name: checkpoints/2112/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2112/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9dd846c75f8ed1339831498e7ad577a730d1fe400bb746ea9027dd5df8da952207a2f9517e94ac0b4ebb299cfd4152a340785101aca2957ef353e1c8a336b9ffb53fef48884cdd169f38397cefa0dffdc22f0234f7439344a2a73737b59dfcdfa1ccb105af112181ee836f52cf3a16d7873f331592c6c874fb08570610920280e842ef6efc425bfc44218473f728a4caae3d896a1fc92286b0b1088217817d22dbda864a498900e559d1f24df2d8c4b93133c5e2d77f0707a9f709fef0eaf5fb30565dab8003ea9a4245fc342d19e55afb92b64abf44b32562a77ded1d795f2c713de34f1b20be43b7ee0fe0d955f0c7eb7861d5f1add2cf9479b5bcd33ef324&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85516833Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160, Name: checkpoints/4160, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4160?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=27cceca271d455254319a28ccdd8e2286d9fb57e15f35d688d602b0562792da15ec22ed80354c60336ce0e9ff018b75bfb3957232ecdd6bd2b785838d8be132beb78886e0abefa75955ab64fb3bf5773b63e605581c0bda025857f1ee88d295258c7e672f001f8542265a50e48568406c2dca5995937d9024cc311163ac795e711dea0149e3b34b02d51ee8ae21c4926d3628184ab09398504023c15f2c57c342924fd8c24ff667aa12e57709f67549c3ef83ff114f8c7e3ad29e01b19569a64c1e683fdf9c15daff6052871a02ee1c96d1facf77bfb69fd2ac9d6401bc30d8207a3742698dd83a45afd62595f32f18e9cba5636d7a20565095ad9fb417f1c52&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855209043Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992, Name: checkpoints/4992, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4992?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=565782fc8c1c98b3fc039441d373fb42c2d618cb9c7721d31d42ede77907fdbb0697f07274579dc1ca9f7c692f5a4c1969d162c8928dda09636b32d26fc1ef79fda51bf1136d5439aea85eca92be5479165a9fa2f86326d30f50dcbbceebedb45c334e8ede53a5db58528f82a93b4c8fc20e04502e24b9da4ba5ea52534917d45e221050e17e267c14a6acb88dabce6c906660e40793e81c8afd779dd3787e2e92926d0554e0a46efe2c965d3c60c73a34ed9fad762f6189c42ccb143cb0ad2094d1f1570b40e1f9f35c3cccf58b25450d1f9a0f3de61a84712a30b7c0e2d25159641e75c51d97dce62242d8aa7f9a13c14502337b24e2acb8384ee7718d7fc2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85516689Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288, Name: checkpoints/4288, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4288?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=87af04076389ac8e662f87adf5a94420e6c74f22001e0dde0d9ba8ca043b5490539a682c2c3a12a8e3f81b13d5139ef3a55d48ed84d42447d0a82fe5f437a09437127eb1655bcc9b6e14208cc0b9427abf087a5f5ddd3a7903a61be86f8028e9be5825d4eefe9814bfba793dd1b329ca2b8071bf34243267945afa07bb15142ff103aa74e6bef0007b6d8a9e61673436e9ee6331e7a6da96ff8dc6d76121de7637cb178055e75846c745a85010c31b63ecd7e4f3a1152ffc2d4955019383a83c9b1883e340b2db81fbe321ad95ee2496496a27a440b85087856ce134b614ba5113dfff13236af68fb9621bbc81fe9e8aed980ee37ff2d39ba773abef3ba3a7e3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854739337Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0, Name: checkpoints/2944/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2944/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=c411131da6ee66fea6709dc468264f2d6d72777e6e527cb86bb16751e1a3bf84e6d7dd41b4779bd6633ed3878fc544ec4577d2ab3b357a03330e4a7299bb9bab353caf9cc3b33a639af8e7a7068ff28579fd07bece142711e0be752be5b0af6b9dd980af6a37575061457f0c34658c796c4617690a692665e1c7fe442a537877cea742bee8a5bca01cf137a3c9bafc38a11c74691ff84c16f2b001c3942c466d4a7dbc56b9201f95ce6a880a293cb5cbb019d798c38dc491b89b5d1c19db77f3b82ffd68c880bceacc6c94cef5e9f15bf0f29d864e08752424679f195462f178c3ab218968dc1ec1670c8da868cde449e4d52c97d94e9460308dbf6aa62dd0d2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855521908Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/d, Name: checkpoints/1536/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1536/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=89b275c80a3b176bfcb2c83bd649f88bedbf4ff2e0374b52d28e5721c5301ad77332b76b1c78f5bfeedb2f165ebc7ad971848cf663c914f1501badc469b76c7a3aefc83702a3490a6019d0f37b4bd1b558744054882d48e5c4c0c317bfc267159be412a93ff23e2216f68c0b48ffec880ec6c1df8da538b2548dea29aa7b5a37d1d9bd78a7423b98be1781479b06e465fd547c4cd42e5a3d529e2b4297906ccfb9030d91a86548a8cb4a06e8da9530884b6afb73cbf3b16104952f74ce687207b003f7f906ff09a1092643828c8aeaba89caad90ee539dd1c8e783aa8d7275b09fdf1611bc4abc3526243b3547497909e93adbcbd221f3fc5d1d523217a2a14b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855162813Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default, Name: checkpoints/3648/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3648/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=62259beaa27f3644e64dfa1da3891da93ae922a41c472d270441c115db00920b18351499c0ecb8a19a682c23984b38d55dcd0876df303c968989b424f66cb812c86fa98b5aa6fd4db86f03a4e15bd2417951e7ba3b9d94536607817d0ae6983c4c274b90957c2cedabf74738d045753c3d42ed709a913d0b25624595f5af74b68caefa85a98b7115fd8b088644decd7721c24b57adf999a7eb93794c3d21b4b1fd34fef4bc478db3c5f1e3e33de7cfb43fb5b23d50ec02d46f513d99ee9e17d28a2c3dccda2bccfbcbd979fa83116c01a56774d99151fa3573c4e28a0cf3c9c79d8c2044cac4ad0a7942c55d469c174ce80626d3c0d4c7b65d16694963a9d213&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855520816Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0, Name: checkpoints/1664/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1664/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=076a0947fe158646f60f6e03f5bfc96a64e351c422ea27e81259f5f7ae7da090f8620cb6c111b1ee908efdf1aae0cb21e00ed3ec62b23c05c8daeb93e84b37337173b458e63a64f333ae37b1773a4950d25eaa9fda13dd002010e83a40e5b2973c15174266739a6e5009b2291da01c8b1174106a20e30b3e3be3e02256e09c9035e84b80566abe115ac423eaa5bccfe74956708760491aa553be2b0f5e87cb42074dd3bc0878e8a02ddefb39f354a92764b2e45385bd145b0e4c488afb6f79fdac1e97cd032cfb2245c2f2bdca622cac7bc735c476eb161b7b58eda4af6f996a1ad21704e62fe5275bbc0dfbff4152ee95d91840db8a3da9e648848bb2670f3d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854482952Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264, Name: checkpoints/3264, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3264?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=140c95ca59082859bef471090605522a38926fe5db71e7241d0d8b33097ee90093a9cfdc5bac0e9e64da916cd80a8325a21771150ac66d1dda850098ef6c21ca893682c50e1dded304527c6ed3b7a2d41ceddc3ef65bd368bebeb5b8ceb97cd17c55ec252c9836a31444429a24a001b4783c4d3324466ce86835a54a038ffcfe2fef1967f736ebcf658a7e00a7c5b2eae47c8cbb830290756773f052e3dec91e9eb6b3b7594a227e753c1f4d54f9bb855eed38d243398bc642909aa1eba5fde6adfb5669f80a08459a01467fd4f84195d0669395362c33af642da77a024619343eb960b7115d3f28bb3a358cbf6529d4db9c19172a9debc892d5ba2dea311002&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854021944Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d, Name: checkpoints/4480/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4480/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=5836bc129c7f8d348d0cddbbb238c244460dfed7e215f20c08904ef1aca33f5095f3482e5611800aedb452896486e81d189da107ab68830717e9554686c97bd4e1b0da0a5ab5844b46ec4205ceccc1674179985d5db187cad29abdb44e90b3757338c332b01f5019b08556b2fc691679e8b96f8a01d4e7cf8b482f5c72357c74cb6f7ebf1cbee765c77e4f0f4cb47d2face4c3894784d418c16316745131f8570c3e638dec997407911d5d33d310ce14667e727af95fd7453aea35bb999101e9a515ca94bcfc3323f1a01f1bd761f293eb34f70d2907de25d13970ab0a39915e640d276867a5ab2f07c30f50ca76797b9578a50460571c1ca13bdc5a451f69fb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854509329Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/d, Name: checkpoints/1408/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1408/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=61fee1e9f5fcae0ccbcd65e3ac701d5ca34b0f3ab47474b625fe8a6842ebef43b99f1b23c905c802d1c60af2f17c9dcafeff4065178fb256c1e06badb523591ba109134d4f8ee99f33eed4f5304cdf0cbd3a155f3428c9d16e3f3a0a40b8efdd36f4fc823f01b8c826ef42302f29e2da9f07da984debcb186ac3e557c11b64fd92d27f68107860c09f68049ba495909c9d43650d27421726225e0f58825230d0b99ec28e6b10e5fd89a56bd428f4c74f26eded0d952378e72c646b35d7184d39c1a63b43b915e1c8305708011e1b55001bf6d649c5118c0874422122759265e51101f1059a31d6d9396aff64fed4ce18f3a0e0166039a9bfd9aa03be45fe1266&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855872197Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/d, Name: checkpoints/1088/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1088/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=bf2aa06ceb4ecc8a7adc5db37161bad15fe3a45cf21e64495eb158b7eb5ed8569b52a29e476eb16a6d833e1dff4cef002a44f367b2b0fab85dd1feb73f8fcbef53ed9c8613e9beae62273e4a0e30ddb1adb88cb0f157f3e0196e1c843590265ec567f8a1a3d0955f1c882606b75421989f356273d2d7bdd79116b057bc272a396b9e6dee06f9ddfdb0ec5395ffd58c2c5eb639ece9756d1818db2a7206bdd2aad447b9bf21e1efa8d82f5e7f09d5d13e879fdbfd13a1293af5a91710b45d00272a67c6195f4603a592f7fefa3d0897286148c32a82f92c97001c6bc3a55360f905167003f8a48e94ecee6b5627952dd56391b2f3a42e530dcb103709cf6071ac&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854513798Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/d, Name: checkpoints/3840/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3840/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a65d69a8ab55fc9115f280531e4d928d22c8c81479a96634d33f203bddf25e2094309671afee87f6271d0b669b7a63fcea71971ef928b8f0ee2ac33ac58a03576cdd1895e06a67ea1446735a3a284ec706e288fb4ca2d1070ebdbfff873e283bd23044033898457a540ef40314f97ea7b1f233bfc6a83ed6624d17f2ce43bebf216ec1ec90ea1a4367830e8fb5bc423d323421776e15151e345c95a5cb78462e105b31d9a3c72c9a8e7f5494bcc3d706069f28268b984b70151ced10587f538ae7f673ba86077cb84808de7ec13f7e9d3ddb3ae0c0906cdfd470275363abf45ac72268a44235735c806b17e190c7c532d431a05ccb612887249f783abc3dd3e6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855977166Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0, Name: checkpoints/3904/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3904/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=12a4d4b6b59f32bd7ffc204b64b4840db9e6ea3530917aedc335f6ae6726f9361c1485190ae97104c202553fd70cd91de43ded8b95da7afccc3bd8a7f60ce9283b15134e77390bf78f5b2da5bc2b2bc4fd1775e3517525acf9f6e9abfa6de0cd24020121344880d0981a400be6731ac6ae40f504fdce69e5acb48cdf3007776534bfa03dd7929dbcfd46c397ca00bd9bc20965558a58a962cee0f83f2125fbafc5190f66a4cfc9bf538dd7726e23d4aa1c0d5b5c5bb50c0cca2ed24d8107e94118a43544b1d086a949cdcb23c66c9b8dbd1042e0355e06feae8db35627b0e95d3244af4fc6c004952633da7e091742929190a7502c2d6f49144454e9a479fb1b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854045159Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/d, Name: checkpoints/2368/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2368/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9fcd76d68563a8c59c2382ff04937a9eedeb2f2754e37d3efbc6bbdb43247928f02124e63081f9303ecd545be2e313e66c0d0ba11a8e87fc28204e8b3b57608018a422e05812136e06e08891ceb039358d4a4e860871586f5a74093b70f2e5541a5c69f5d763912646f1c87ba145c9dd34265b165fd2004bb6ee8e839b2ba28947df30beb32ee44285e9cd99b1aadfcba5e87caa47b825c1583966ba8f85fc287f3a7331688b2e517c7b43a4454160255ed89fb4c0afee18a5d89186d0291a7854e2c6743188ff488c7609395f1744e42604ae25900594dececbe69bc406511b77dd4a6e9e429244ad1be511a643511d8048c825e12f6b6e3061a74e5979ac5e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854553442Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/array_metadatas, Name: checkpoints/1344/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1344/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a65f908c61a225f90ea198da4b7f3c94223b57928d5e1f0e11897da61bd99075502fc5796a3a73bce8c9b48225918cecb9237c762911ee96e0a67660959c2d9d10653d2a63a10535fd6ada787218837706c9d33bd7e902bd3d4370183da64f00fb667189be9c7c0a4dc5e5d4f87110a0bba22fe3d126a9ee2c414e6598a08b5c00c31d4e4423410812b4fbca98f389a1282c2517991f59e9548a1d78d626cd91220a321877d5ac11a990ee357bd82bc695d56ea4150a0b4b81b06121e7c674c28325e9885fcf9e5490b4d9f8a66e405a153b5a7cd091dbcf64e56cc67536efbf948bc6022d4f05a058da8dadc339105afae436d2f7035ff9044b9745baf7f168&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854583624Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/array_metadatas, Name: checkpoints/1536/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1536/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b12eb8809b9342c06b78863fdb5dd2737e414622ddf42e2d7e6073cab7353e8ce5b211fbc3664a24dc52f7693300bf81da80e6953cf85ca2caf35793bf2a71be194ed8d3604b80b15408079c259bd455e91ede6186f5c676268e5b9e952c246f2e35e66149ef0809a7262c2eed9c577662a618e2547eef82fa4b232a12a987184cef4f05706ac3478f37f8aa29093137b38ea0635b0f1cce891ff60c94fc059a8fe780534ee257bd57e18a95c8259a30d93972da2426fd98ba0929fc2ccca6f2b97a279317393105c86306cd45e5cba75c3bf8dd6292519b744f69eacfa8498bc31e18ab331a91398adfa5d9bd9e1e16aa24bf23e33f4d50aba0ef373fbb5f07&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853980036Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/d, Name: checkpoints/4288/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4288/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=6411dcf7495da2ec8a8cce72750b963a2720d8fc3c85cef532130333ace00c5ed641d49507e861e312dd6202ed880cbfee41cc1be4294ee93def6d849022f2edfd1ec649736021ad1dbc3d86c0564d60e46a6faf825e6834d6fc462b7e2dfb15d6ca31b4e61112a8df6f353926b9eb034c917d22c17e41609d7506bd519314833d2b3ecbc57342530d73ad0400f415ba9e9db58620e90869d019deb2b91c78ff2125beb815e8eacdd772b8080e43af08767454c41848f81ffb386933ef4eb19bd30b41dd3bada10bf04fe994bf989edd52f2c0d0f1428dac67b3a4ca96dbf6347f28abac8d3673e064109eecdb0840d2a3ba32369745260ab576cb1332202755&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854592582Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608, Name: checkpoints/4608, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4608?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a2fcfb8c5bd3f384f82ed55d9ef67fc692bda7d6a77a823a187feb52c2f253975d9bee190ab55c2e666a89042e5cd141d04f23d754d699e067b2250b5e491650f81133c73e7b338df729e836c5854263db6203c35a6a5197fb00ede07b40e8321c7a0a7ff8b024895b69be834343c0d51ab2ef599ad5edd80f3fc65c2d5fcceda198105a03aef10e62dd3955d6af681a904a1bbdef4c0686586913cbd9524b8eb589a49e32c0b1e57f724ca1623697df8b151a00e831881f56be8a7d5dac8cca3eef64d665434e6272eaef2b6f73c366d1c14f791c1fc12102176d21f1f9ed664a889435d32223bd0cb20c67a32eedf27ed92481e0126832b139232375a7367d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854617107Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/array_metadatas, Name: checkpoints/3392/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3392/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=7317feee777ebd9e4fdee02d1d86c2fa31bbf6ca83a90d75109500124aeae7b28df1b16b66dd7501de6282a922ac492e02340ed11eaad261431eb029ef880079f029d8ccce0e787091cb6dcc726725ab3f427fdb7635fabc5c61ebded03e4e860dd305d58620354c2c583d43ea5cc617ca80e94483aadd6c1dd1c2452d46a5419da1e4cd641e1f96e09cce7179a0f16e83ba006f7a00e0977407d9b0c2e50a57b61e8443e6344f3fb3c7278d09f988d65c02b5b563bc31ad44b9369ead00c53722fea320238436a758de40e76d21b727753a3633215df5805e00da47c77e5c6cf2e8848bd93b76a8268bb43aa53a706a1b7f1ca4095af556b42971c24ab8c7c0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854189321Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default, Name: checkpoints/960/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/960/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=867ff2c29bf4294eff6a45ab68c19c87d067a828a5a566f8a5f5191cb4d19c9a3209a08abc9a63670973fbae7dcf92a1c01a0eb2b551af61048e3db82f70caf2b2f17ede7987d987513560b049a210ee57a6af5fb984de6b78e0035a6ac8a41b2a79e2aae3a9881d0ee02ea288f5314efe38b1c293ef698fae425fc6638eb2f53e2942fa8895d2f987eaaba1c7ccc9b42cbdfdfba2855fa98946e6d634a65d84c74d8db8a57a889a30317f3ddd7eeae44c8911784767b9f7db07cd0b465821a29574cbfe711091cd10f33e3d82f84bf362b0e5ea206c61ebf5790273b8e7a9c672e0f4dd505f3dfff481912aac7995aa2d6fd32a55d396d6d4de6df2d85f31a2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85466046Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456, Name: checkpoints/3456, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3456?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=845a6f5566d59997b504dc1b51c1a4a6f8f77429ddb3f7178e8140454ad7ebf171bed45025cdda7d3e856cb62144c53468ddfba21d5ed5657d8a4bd7b03094cb3e74d44966f167be53d50883c39d9d2502ee37897c861c9dd5ddd3670333c7ceb0f6621cb6ec4514913fb77ec39e5e4950a2170b212c67a51b1a77618e0c60fb181f2e0730ff216eca7d4e4836edba616bb415b76bc9220eb3a98c896d7973a7ef6eddfde332d75a19d2e664cf2a3c7c8c788472e8b16f92ea12cad3f041a48afec6dc34662bb4acf3c94b319a8af968a701afa1c92032730e93ced94aff0d95f7ae1ae22e5c0cfde31f9ffaff686a841024152ec883fea94842b4fad0f33b65&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854679412Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default, Name: checkpoints/1024/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1024/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=7dc467d3caddfea5ae9c0bc20afa20926de50ecdcf8b1974d27f0c2aaea00c0a1d58caa3b37890fca589d884bcbc84fa8331186f11604266ab53b48c2c9ff94dec8f55c0b10797f02aac4d77f6bb2885119f177c0182e72f4dede2c0ad65b1cbac89d8ebd45d33802c4d3a73f2e50c9e0864ad963b1581bb75cb321577ec508a76a89614cbb8f7f4c1b6fc71958a35b3e022970b07f597b8f5169db71651d766076cd0987d9a707ae657bb55380dd4c15af00c829a551224f0c77d3c5fd0c769555a1fc7df87ac4efecb33bec71cc18dc8f53c7f73098df2a171975a661c30cce4a14f40a608cf04d6c94ac1e1d770fa768e6e2ce2ea8dffb3e170e9dedd5b71&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854698532Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/d, Name: checkpoints/2112/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2112/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=7ba202c9428abef568215d674b8c9f3f0607667bca52ddbf55dac26c8255abcdfb1bff4cd921a19cb716aed23ce5ea9ae7f4596777622b10a72ddd888fa0677255518c3be6f5a8d995c876d3d48f9f8d76bb40a994cbd5b788e70d25f67b998c3d60257939d3dc2389a728e045e12a90e2ed81c529adfe869e2371541772cee8dc2a4cb23cdaf88187feacbe8119f627db4b19f21b2cc909b1032b56f25c8fa63b6265cfbc93d30aa19984a481fdb5edf7ab051c2b4dc3f05aa89352427db33a952d394c706fd7282d999cf25cc72b52fd638ecb989c811b49de477e5817e3bcad41b2cf9a36e8863b9c77a181378571ef7e912d7784129ff69c29ba8a55a2e1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853988219Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/array_metadatas, Name: checkpoints/2496/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2496/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=99d74ebb899f7ab26c075bc10917d8dcfcb890f69a835e90c223b25b0d8466595bb3e5c89ee7043a27f4f8f0cd48207fdbd114f3171ac913357cd1e1936c6c9026c3b34082edde80a784cd1389cc5c713eff3555c2b78d1e9013af33d6f41e328e9cb4decc85aebb2f36d4edc8334f35d6ab06913d74df8ce73af578d1e899a6554676425fc6593d600da8f121c845f3aa55414b4cc9b0bf52c708d87e38c8434414af85adeb987123a4c389eae2fad9cabc50af9281c6e889399c422e509fc310fae541403358e7e50f613a93561c406d3dbb160bc9b666624817eec57a5d3b88b13ebd64871779093aaf8ea6b7fed96f523608d37ed085f0bfbe9c0b34138e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854730661Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0, Name: checkpoints/1024/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1024/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=24ea515cc434d8b6347600e20b553839f34d3837d18ee2796286978a363ff2f049a263c45646e562c90bca903c725ad73e38d1392b3bf19e7e901d0ce5b660bb496221629f3bf11fcac4a291f19bbe6268b63a511a38725dd54c42723b0e4ea6247bfc286ca49d567efdd599ea02922a8a61eb6839913710ac1bb7558a58772595a4712821e043c4d559c4fdbe9210ad459517e69594ca9183ba5305ccc17b42117efbdeebfe350647ca8051a7d8f5cf3dac64a10c296e1ed7fd46dd01d70c7c5d962743cd54a4217380ccd4bc15eb8fb8a91f84aff02e1cb33ded4eba9363953d9efbfeaa0c19704cc449994a40bcd435e63380b743c5c4de579e5c9a50d76f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854503488Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d, Name: checkpoints/5376/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5376/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=677e4f6c1144f44b59ef0643a5bf8b7e1df9a4561d2f720045f902bc50abe2e07f7024d3265e396aef0c5c7bca7f9b0a9b58c67c2a790ab4c91b5eefa830e7e36f7f1af6514f474f16fba08f24b256c0f42e21631dc6740cde37b76c87ca6f2379ee3606ffbea5bbce3f34cc3afa7fbde2c0daccead1d400a0e4a4f3f028b1fb7920e80ff27c81c2b304e896afb389831c53417bbe30593464ac58187760887d9f70eccd01637cab947b9b332997b57c967b70abe96aaa9e89fa5ed29db2212cfe1ff4c33b6ec825c5a8ed817f9fe7c8773f65afe20ccd3f59e636b84a9c98c47952c190a1cc0104ae7a93b209aecc627865287f9642f35d7ea19d9d6e7e7997&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.8547719Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default, Name: checkpoints/4672/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4672/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=11670af0fcf5b5925a8784a35f3fc1db031e27052a138123f97c1162c30a707376fb2f6ebdac9b744422582f72c713ad8ee10883a0deba2860c441bd01bcf83258d0aaad2667565f918aa14d2c17c18625bec00a934347de24142043355ae22a0cf432d4b51d309f82db88663b8088c555f39d4896a9b77bea64518e4e1a69587828f1a002a7c81b0c05bef6bed58e777a3a635aac91f89412aeaabc6e7dceb1f356f4db4bc0ace6ec4acb6679aafdba2482d470fd3a4a558f3ca89e27bf468aed507884ab0d0c66806ab095ce4fbc5a8a0273dfa7b1f1cb8331f221fa830f403df50de688e561b17e011755624c5eaa16ee6e65b903a8753ab7e7610339ecc8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854782915Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176, Name: checkpoints/2176, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2176?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=7fd3c140d00f292a3b270fa5d3c2af3fab7c2430b598b2af1907070bbe0d0d55de05621e23375382fa4e082628425f056a5df0bff4a4e8fff3d04d4a675d61fe68634697e5977e9123987325b998602e9fd034f19807cedfb48bc6895c2fc80cc19b4ff2f8d951c42269d74b2dc284c2e1cdd3ac506d53c5747bd0b4cc1be68b5ce3188a49c2aa45d3984541cf74fb1709a90400d25ae845211a24870cf37a7f9735599abd9728fe0d8f00c2c7a3a4a4694b2e10c3a93556287989087d516f3c8011b5852411d0b1081ba8027dffbf0a7b74d82de527d24893256e569a979e70cdc7cbbfc1759573bd96c89a77dbdfd8c691c1cf743ecd1658998e88aa284166&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854824557Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/array_metadatas, Name: checkpoints/1472/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1472/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=1fc9e8078d3926da735eb6442d561fdca450dcc379a380fddc301021cd7355b8d49093e176c4dc61a91ab887fc05590f64d1878a1d7c51d26f83410fa79e4ec1433ab13ff8675d182d54b68ee36f5ba9999129493fcd3809b8e3ab51addad9aae1407a623c83fdbb76e8846b72125276c7928673efdf9abbae39efdbfb3c9cc4147b9008b929fbd433a835d7f3eb2f6adb54035f4617864ecc24a68f65b9de906c6ee34d8a9d347bd08ba93f6dcffe760a92ad2e4edf5ebb5800aa6f5f7999c71685734bf33abac00db3a6e9a7d8eccbf241f734f8342ef0f9f1f53a5418b6980694dcd179887f7b43fdfa6bcfe5e91e22e39b7115cc30b8e66ba990bccf29df&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854827968Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d, Name: checkpoints/3072/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3072/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=387bb48f04d3a3bd030a13713589b30034a04b44671b667f91ce5eaa6af03501cd8bd60ad6ba75e420a2ae4cce9a789c4f89ed43bffde75e6f0b55d16755ee49d4faecdc7f6ed68a8e61bf02f5df19f1198dc35b46e7ac76762f109bb2f1f579911a36ba1ca488da316fce8186f11ba9b97a7b4e0bdf35c80a2071e5a58863d9b8230ca63171cc744b0f6ebeff712528bf43fae84b24852fffca53fac56add16643ff710a1db1c27e8ed35c6941c14c9d1b480e6c7148ff30bfd2b3391967a410dd64ab38d56683686301102d237f664ee103b05351b4900af119578426f909eaf93eebaa1ca68f3d6ad4508b0f42a3c8fcfca4c35dc0b2f1fec62e00ea84b49&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854827841Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d, Name: checkpoints/1472/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1472/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=324d44c5e371ac08c2b06055e5b791b6aa3209f19473e7a2cb3c25ff88608b3374c237759f1bbce29c8e05dc64527c60fb539b773cb4fb8c7d4d4969cc58a111425d51c6da0f731791fb8084ea54119841fdc4ce515131f7e8a00c2014e45bea0354f0a758b49c80597a81cab5bc28719ade0c9131ead9f990203e3fc52d9529bd0b2479d3cff31d52751904ef1b2317509502802610c80ebba3b817d9b98965099ed2fa5c7bdad3c760cca343082a247f2cd4095695f83ae26448717a59b5fb6b82308512c5f542541eff09165e74c50555b5b7473e174ae7c4ba62eac93a10ff2bcac19293a24bd62c698a1c37b4b29d93c327a6126e25d541872ed4b55b00&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854785138Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default, Name: checkpoints/320/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/320/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0ce969cdce9b0dcf6970ac1c642e1cc17d32032ea111ce6259507827eed41739d1a2bfc36b1fe748ef97994e8b2584d47ccadb54fd7239c9d7ac1c7076d7dd5f16c78a6e6815946c7fa090303f6b5c98d3e97d71fc846050ba00488710028cb03a68b6da9653c8775bb1cf01db5077ded4f2d65b4f9e61d64eba860510e23c30ca5df51f5a00c8495193aa58e4f059ac68ef291392ae8a732e301b482af8a0045f141fdd4e645e6847e6a9162164254280a22826d9f5e1df1e7c81e6cd9bfca5d32e9112e68567493ac3730c085060355cba46e1adc046c575e73d7fe7c4c89d3347c5598ca8f32847ccc8047718e20fa56e5751a7dad919964f7d46f09088a7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854849545Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/array_metadatas, Name: checkpoints/1216/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1216/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=7af4fa68ed4e7d0099848946d8d3cc13cb2c031433b2c3d498810a9b9fdea131572593e2f45b98dba477057b42580dddc75ea6fe1846ea5a42528c23c7f8b3c16f9fd053dc785b16590282d1d0f5fcc9f157f0a05a0773576cd5ae1f44987c251a3c0f671d25a3cb6c9214313b01782b8dc1382699f3163f890aa18abd0ef26318caa72e139cded648b6a24ee39a4589e7a417f860e1581925852b69565bbb36348beec87eecbae6062b4221d8189b34edbc3107910b94a5effb142a56c9f8ead3378c79850cefb981e6c8f2dbde9381b1dea89f9f370c1da581fa375930406d1165b7035544ee75a7a8899d4fd05add1f55847ae728078105e7146e56429c03&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854930753Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default, Name: checkpoints/6080/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/6080/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=5995c7a67b91a631f35d274bbd8c0b8c869695fada1ed696e75737006374cf9e725862c4bc484d8ead59ba5d48b4038c19b008d7cf43dc5833cc164e8bc79fbc15acb5f03b0bcedf549e5f76d9ac2a1faed3868a9b785e8974704f61d5fc3934127725ae53906228bdab08a5b8cbacd1fb870786d868ca1441b0c3d5da7d43075756e6a247afa9a1c1c653660c5241253465983d3a51aa559070673f058d15f20b27c178865e2300a2007fad81de53bc60f5286b72869e644a78b0c452c4b181cdb7d954959cc76ce5658155bf7f7a712d0c985b07c24331fb5967817c4a00cce50fe85e54220a7cb8187e6bc48620fd51c9e4557cb7f65f202e2ee99044efa5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854872012Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0, Name: checkpoints/4224/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4224/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=25c15635ab394568efe32eb7f3172dc9368de40cbbaf79c3e842411b924b57c8f7715859c921bbf0e505f28da03ef4e5d9a205af7894dc1ea45d2999f777b77bf5b34caae871253934912e6ea14b9ccbeeb1c8c8f3e64bc8d30115ebaf69f061ea8e20042ccd85da1b387a29ba6745ffd69a30180773a3b69228adb250b15359735079bb8e37a5c7d65c29b7650590321123c7caafff281668915a9a9d39085c902ec9ab64f0e2a19dad5b714e281215b6054861c482b64331f07467b9221cad9827c6f5a3c29313e860c22d791b6761dd0bfa35d6288222c3bb615a9f7a579f3869b2692f1af03c073695cc3075c5bf46cc1badf1576178d40d240435cc0270&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854933401Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default, Name: checkpoints/5696/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5696/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=6c771280203105c3983b9bd12c24f8965db542d9ea95e8384c82d554c9062dcdbe8a505044246ea8c8bf1ded29518f5a0fb51c36aae65caff4c57f9ef88db5126e2712e8fded1597fe485de4727a443329b3c0b533a2fc6e4d5d31753ce0f3f27f1bd59e303b8b1abbf52870bdd7a54370927d6a7555f56b0a8360a163962963eaeba240e1ec9304463f6137b62952269b6b2c9738babeb9d37360316d92aa9db1a0869f639419e434848e618bcde7581af7cba0365dbe71753d8bf1fcf25f2006c496a2aa887ad3abe6e3caa72a1e54c890171412c108d8ae304a777192be39a0c7372e80f0914cea9b3a3fd423d8cbcd7ce7e209866fa81b91d6eefb12c954&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.854965451Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0, Name: checkpoints/4160/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4160/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=38a4de114a9826589bd062123d9a05132339d1b55933825bc8553ba96707dd06db710fd30ae922563ffc00f83c7e513a004cb2db232f20c9f5258523cf31345c672cb586bce31ab8379aa3b0fb0f94f37283bb7b11970f8aaaff6ce0ea7391cfaf29673d38a97f502e5cf40d291863a028f14d84f7558dfb49987e572b3575cecafa0c35b1f0ceb29bbd20d526434e8efd81f3dfa89813dbe6180dbdc2e6f7fab3e14972d1130e15aa4ba003374ca4304630915248bee3e1cff3afdfff729c992df0a13956e230db8d000b4d006ae2dad733044753da4c4ae3d75e3db8c3da34dee1c0972c1ca5532f6b26edf77f4997efab5cf423e7822dbf55609caa2b67d8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85488557Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default, Name: checkpoints/3584/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3584/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4bce1465778e59df724f36d009b64e0b1191daee3fe246f2a32939ceb17c847bb8f26ad2475700862f8a046d9e76022dbfeff8031526d4aa7175d897d21bdd802c0ddabc99566edf8565d169d7dcc886f1a5c631f4490f9a0e82eaef639573229bd4a73a1677112368ba40902a2c813b097d8d2a12ff7bae3b86346c8716504b91a91aac4a50d2ed78e7326ad6440172ff193cdadff4cae82c7fa2f8bbcf2fddff129b3b37ee1e54598b656fa922d2c9e0f5fdc4fe053ed13023138ed4f3ca21660d229d9083008e434b74c41af76fb34d618cd64d0e2bf8ee6f0b48b4ecede1b13b78153ccc95c135177f8088d836b15ace62419451e4ab02edfac096e43b41&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853908968Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default, Name: checkpoints/64/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/64/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=3b64c67d996356a31d97fe23f07145be913decf47d95739add8d4350a9c683ae55d3fef2380f1369869fe68f5fe75e1f2e4bb7b6dd177c9b6fb92532e57a408e7ad9cc937ea4c7f3135bda6a8e86c404ae7c2b79a7b6776f3b6b648168a55fa5e39373c0e93c3ddf64a9ff3c44314fc4595d9b26ce7625a4a93596e22b3f7d1b328a5cef2f77d851cfe7bec0540b14e869c643e0775429906b75d6dbcbeb32511a8b43dcc31aa8988706b6b7ab20e516183538d7130fb6103c2e45e83d00dccbe88b8309e5195670112550c01e28a05db3fdbe31b61fc06cb1d0a34c0f1069e50dd277908ab181dec9c94d17b23c724100b56784803e60d4747cf69ab02736c3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855000751Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688, Name: checkpoints/2688, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2688?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=5511095bc3354e1cdbcfdecb36deccb0ac4da0e2239f20c4fd5ca0b82ae05b04a4acde0c3e4a6a96005641b7d5116e13fa5696fb593ad20016a070cd195aff970bd742c5307d2a89cc39c686f9dc2273427890ebf77a4ca3023a20c9c075d41f3d07458f6776ee408f743d7b1cb06e755c66c6b9322ccb3d1d777ba2038761488d9210880eb16c23d95e229b0b9cb04b76e5001483c7be660665caf2e5b47561964b03050a2b6e022f18b922aa016f837a7afd8469aa6d2e6c72e844b32fff6c4f21f89b36610967073d0f5791941c1cf01cef36e1c38256d59a0f409145dc32ba137c9b9631fb8a57d490d444bab0b096fbfb8e970bd353ca8522dd8c3e177f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855010933Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/array_metadatas, Name: checkpoints/0/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/0/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=725b1daf41d9b97af96e3a56dbc7056c4b3a2c82130138507eeb0d81d642141ce8f160c57ca87cd862050f3ac71bb3852bd03cdbc44f6abcc2c3df6095bc11cc05992dcc5d76d060f8488c49d1084654bc05239ef8a1f70076415ab309b7f568d3cd74a6d4470cc27cf921d042c1ba7d6e677cd4ce79029ec21cab6d659b9edeaf5717498c4578c56994dd3b70fdbee752e6dd5c996d437666aede649c48e6694bb9434e72f1824919fcec241278c762d9225a6ab44099a1188178d0c8aa21912696ba785f126e436c10841754b48776a5f460e2117d4a6b7da8cafd0e390e30c225481fb06748afe28f124b6cff788032e3a9fd22ec16ad207f41f09a3494aa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85501306Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888, Name: checkpoints/5888, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5888?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=1ac7c82dfb53c3a23c0161b8373fbce8ed61b440f8c118f4975042f003c0eefc94829d4c8900c7bbcaa21f78c7f2f6f34d0c3c62a392b136f559b119df2aa16136201e69bb6f2a188a774bf3f0fc3b07e26619573a1dc56455e99d7694b562a57ac7fca31b8113f7d1ebfb62fc5320069b4f95df96f2f5ac431554ffa75d6771137e70c2816a6637655d27705030384f2b300b84c81402fd514c4d12483a68322fcbc423fb1c7babdbb6204da0c789bcb8d823401e5d3e82f32f150a31c90dd110989ac52c45184b7ba27d775e91d878a738078ec2499d50fc6412ae0c96505bc5346e60c3f860aa62044f4dfb0439d5ceccea18bae2bdd8853f645b4799c162&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855016688Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/d, Name: checkpoints/960/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/960/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2885586d46ec4cff2e6bd807bafd4a0e15b2657b525f9229e9d6612ef80f8b74c7f5fe02a4637b6807adabc3ad9f9d7b10e2731e7d4d23c5bdbf761d657a6a8ee9633cfc271369dd823eb39bde9d8952276a4140bc7b07fa6006c5959fa4e5e2a994d0c2ae1cc556535087b12754dd420a236b1a382fa3f09fc9e7b3215a7e26b1d063396c01b651774da787d7d60d27f2488235ef05cd898ae50243f37b5184f32c2b35533f983a9ae7dabc34c4f60281987adee6ed2c1d66ccee7d37faf1caebcad4537acfb02e2d1718b3c574d28117e2f692ae67f32796d0139bbb39403ac8d4522698c376d57de3300c14de372849d2cd90b4a643c75b6d0e19a4963fb6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855032502Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d, Name: checkpoints/512/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/512/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=72270ad1c984a8fada3ea2157ba7776f95b1ea7ee393a07ad5a77ee6d7584c9c0cbf9ff774567f1684db8d6087c186c20ed9b1f13d696db02a7052bc12f8790b5d5a270fe2e9064ca0ee787dcbcf2418bf77dec0f96193a603346f655df6496e613e2052acbd415d9a304d828c2e3884ef6c3cbe13f9ba1ffa5ee588a42bfe5b37c6aa6ba2a930af1eb4e43b9f7384600e05aac6e78a56c07475229b024c4fff1ecb6a5ce1356810cc6f7f9f581b3f56190bae886e5a4c3b3ab715ee6386a07c6e2744e2d07a402a41b1943d05d76bd428ad16d92e7743a63d342478b9765337d4fdec7a14b53665a705eaa1c0fbdb6743efd4ca90414c2056ccbd69bd4dab27&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855044255Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/array_metadatas, Name: checkpoints/2432/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2432/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=baeeb713c3bc042ff1fd521fc5bd86f538a6b829e8886133ed9c6e3b6b12e954417eedcd4c76f64765707fb8bde2f9bd9decee783566614220e5ca22fdc6d63ca3c5554516f11b06efefb3c7113c948f317b1b87797aa5be2591e5e25f14331404e30a9ecd0ed3e946b3dcf6cbc77f46b448e484a2b10b63b9b8a537f57856710b7813c110ebd5d6579663ae71bf184154bcb4e18bc4dd1465319b49e9ccbcf257477b737fa4b77fcd3b21eaa397aad2817363e1358a934e6749b5cc82f610033a214949f69cb9aa963e4d097488f39e5de9fc5b2850281e3a5fb3596f1963da6c623c9eee4449975c05cb95da240f01af5e8a4d7ab447c4bacb0503f0bc8f64&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85495068Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/d, Name: checkpoints/2880/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2880/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=5033b8e3bbf06bef3e348c7d026e80c49b23e901a5f4b19b790408681467e33256506078146933cee2c6302aa331641475231620090d2cff1990b3d793cde020b3d7b7a12d438e84293c3eb39b9463806be90f881ca932e5911cf0c978ac7bda356252a32841a663f49e3326b7278a5923dc2c83251e025687f6cf45c542c439ae54ea9598519717fdbe9eae601e17e22ece25ce6a01a59e89fd180175b13bbaa1a971e315db94aeca3dac7b92a4b2eacafb5c598fb4d89926eda56122bd4ce82ce2cf9df8762ba7324450a26b1c1cd22e8a3b455f37aeeb32be80b1125d2df376e082c0dc25111e724921a09b4d2f2ac46ffcd4dfb5d7341560d80d876c0a55&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855075079Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d, Name: checkpoints/5120/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5120/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=ac82bee2d7a2d4dd343d14934becc56ae2826222820e0d7e58afe02279ea0d4a142bbad5f048602b416e71062f94119fd7301083c0725c06e0fd2d7113712ccf40edd337c200fd39120d5a7211a1085488f22fab13caa2a8e0434f39801a3ff96d7b4437482da320b0b6703b72062e2ff571dde82a0a89d399424f4904a9570f3adb853e4d76b8a3f6648932db89a02abbbb8ac090f6900eba20a8cb901a3bfeffba0aa6005229678b0b6a6b9d19a74a27b41e5e66df172b5dfa71c1c17b655939fa844a3e8c845410b0baafdbcad21a541d0678ad0fc813c4658ad6ed34e537a8da11731df87b5343fed7cbdfcb4e550383e9dc42566c56860346e78d215934&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855074348Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default, Name: checkpoints/2880/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2880/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0c91723823ae0e64e44817971fba9fcb6c95e7c9de92e934172b3ceec2f5984ab8dbcca7cfb907d6b2a0d41fbab89251e04755e0f67afeab8ea7f88eb65ffc583a081222122296d279a548fffe8fc163c23dd8ed5be205e134df3923ef52fb3f8de03ec422aa954f31dfd941d8090859555bb6fc4a6babfbcbbcb922203bd372e8d47d91f66887c8f26c8b405ac51bc129106365d6acdc63435dad3c449c4c5dbd7e63f0e7beeb3c495ef647fc4c4fb9fda319dce8f47aaa8071b40cd9f02ada4b01a96b69ccba382a52fc20c0590382d3ec0ecfafde8e82ce28f1b4760ab647cdf0c5e373ae126150d32fde3c291d08d1b9474b62e06d6ddca164e92d3288ea&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855053247Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/d, Name: checkpoints/3904/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3904/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=553a2d28b8b964b6329566d99b00fdaa90ac5f0dbb9a2fb875a2f0fde3b9a9b9e56652d01f5e166997a7e947ca01a3f5a3bd3793fd9aa494f3bbc70bd4a439516209f652693c2fc7b7946f3ae9f2fc7a36dace8131d3bccad59672cd3aab7833064e417aa936978618e831a39ef90fa4d26adbf6ea35929fedfe25c3133af30000b1418c4c00441b66b303b4b37a9da448537bb9e3db3fe1da2c56a607fe7299e7ba93e0f153e22824d65b2fa6d605fee199b3e74dfa9e13994f66ae137b3de2c1b8ff03e3ed16f9cf1a095d2d6fbff4fa1989dafb21a70e4eebe7c2bae0d86cd85e009d7f6fcde1edd413e14d705c3dd0c719791bd214794b0df123558ca24b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855104474Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d, Name: checkpoints/5312/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5312/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4368d8e5e6e4812c9d2ea8bac5c2d33668ebf98ecc6d098008f93844b90e0210a63fbe6d7e7c66210c7e8c523152ba34b5e850113b92a1df836ac1156b6f844f9aac0f4be6c9662a22f4b980b2f2ab634b4dd1cdc9d035c3079d83e597abea53de1881255b73c2bb54bb21445a85a7311ef9b997ff6f2e2f3145a806c6b59f46699d58de7856d4ffd1f472423c43be45705ce2c6ffae0b744f37d10f1e482854dfcb0d1f14b8b0ed34d9d8399a2c1eab4cb101c2199e0d7d06c276406841db6a1e267f7925b027b9641cf1bce98b6c019dcfa3078da3f067eaeacace9db54e7e4eab036a5df74cf88b6dd38144f4f48bcaf072deb8174452cd2a4ce0e3dabcf7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855113816Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/d, Name: checkpoints/5504/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5504/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2fc977ede48a9b010fbd0a830283a5dc0d34308e89c4a2850dd2ccfef36395387dae6e0f12b3e6f252688b5c798a420ea7fe636cc340817466da5f5e809a8b280d2ef4a87fde2c6bed4b737c9a69fa3eb108adfc9c4c89cb8c2a49a6a81fa614955729a39d71abe472b0becf4654bf4adbfacfb5c6327790de8521f854fa29f41d3bb9c6aa9c96ed248281b3da15ef2cc82e9a365ac78614c0d4973207da79bc4e33003c0dd7b46d5ebd5509ef74acb08476b2197bdc0b0d187ce35d38f6b84a65e07fff7f862a4181f4015d0b21c68ff62181317d4b95a5c9e91057bb022a2badc2c170b5814abdf01377d19bcdd20f12ef5968378e61080d3b62cda0b7a958&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85513753Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/d, Name: checkpoints/4992/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4992/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a2de9ca64615ba2f62405ac1f0914ad06eee8badd30abdd1e8c8ea007c6d02bc195fdc7dba37a8a9e6943f4e0c55ba09939467c329d4259d0ba3c00d0be26e7df52edaf991a1b58c0a9de39d64287f7bd793dfd1716b9797e59d5df50af633bb03cfd7f0fb4121ea72464749640cd2124dd4cd591cb589326edc176df5a802b0814cbc0e9103cae05b2b09b4a09035cc3aeeacb88447e78c574367b6c4824403671f2cd68fbb39b3b2229023c9518bf13e2b1d332bcd929488b1d8826529fb195b5015ec5a87fbf5d6de8a9634547076181cdfc71d41c690b4b0f2f7c1eb8f323a268b25208157b10de258dfc6039c44f29857ccac9451e6bc0ec3aaed6067ea&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855143874Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/d, Name: checkpoints/64/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/64/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=8258c2d5fcbe14ff5963259114f7b20016d3a3133c1342185c97ca39721b325e101a231f14129aa27e66482a8ee3ae0f91955b673d145be7e585350c923cc0f8b61a6d2f190ab2fd6dfc8d7e8b275755225fa05efacd34e84ab0b29a6599b82b7e438f3e4c9e760ae63a76dc9adec1c5dac1dac134c9cecfa1daf8bab7e68eddd2ce68d9baffa8995c4ed8345721647853fc2f958b0cfdcc5e1b2b16a21ad0edfec030cbc625ee11a2209425de49c8620354d1e2d5bb0f5ea8f92080f2f7e9bda2e643c01a2103d477c4abd364efc5ceddd4b2d0550d561fd2dbae8c69e008541ea1450cb791ca024f0d3cb1942fbdb238062952c6d0ddada0bfd94514deca4b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855163856Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d, Name: checkpoints/5888/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5888/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=387276c38f66078b2156ef7ddbc1ff8410f0d405090dc85cfc222121d92dbd43086ce8054b37f41530ad86d8cdfd62d18280f84bdc55c265f2bf3b49263a89ffea6662f8ab2ff8b8541eac6da0c02017bd7121abe4c33e77bf7fc30e6b7c0542f8e1e446000e7dcf3dab776af00861543cd504cfa9a5be5a1914acd84cd362b379fc07db0c603610c740cc8e038b23c00d54ce4470604ba49c40a0350e812b226bbc44d15413e3d31953d7e9a7afd27d686533d6d9fd167d70af3a5829c11557eb0c20fef52b8b71542996d75077dff31a1017eb0ea50703f2b69222741900bb231fd5ea5d0c5ed7b5d30eb82e2238d6e1407920d4e1b733a02bb83aa6305022&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85518243Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192, Name: checkpoints/192, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/192?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2b6ecb3c81b8a78e1e671e563f6225246ba307dfb798d4afac64c47709890ac22dc591545547c0cfb0cbd8d54b85f78fd665cd34460b6c7c39c19bdc57717d30aa41579eaa37cb958dd09ef85cab22ae4232c62c4cabd144b7f4e425a58909a5e104d4f9b3b9de8808a82e740adf024ceaeffba34eb26b50bc76d68a107473dea6cb47428dfec6129d3c400f9a75e9f2be5e1489dfa922b05e853b929ed7dd14932e235f5ff0f16726e485f3940365a4220669a27dfc6888eb664c0f91994d0bd8247f06a5c7c361fe950d9ef4c598906f335a5319427f686a4818f142fe005aefa62ec93e02ebd440c31cba8cf2abe1cb03c19ce30935fc8bf75083f93f8ae1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85517738Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/array_metadatas, Name: checkpoints/128/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/128/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=bace83d6211db19095d6ab166aa331c57dca75db5abf52bfb2e6d8a2181f4e5d582b1a0ddc4324701567e035f1da4ce2a208b0d495a8fa39d10f00f85c01eb11f3a09ebfd3e33e8b300563e6634438002c1edee1f575281059b522b65ee4ac409aa0bc417d0ce193da7ecdb499d19af59be42b5dab38c09da4251ceacbf97328043a49a81f460fc090a17e9f33d69357aa905e1ef5433a76a34c134bfc2ad11c4722af9236291abefd8dd317e88700d61fd8700def6414d839882207c1f14e9f192fdc5c303e1d6d82b07466ce75d79527240caf7a9a5d47061eeea63262bfc1109d862ff20f591fa18dda547a5efc4f2b431780a1ff880a86349c70b11b8a1e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85520792Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/d, Name: checkpoints/256/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/256/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=040859d5f1e1886eba0279c98585dd5a99fc3d64c25b39a2d2ae3b40ca2826292f87629aa99cf78befcbd467fa83d7c48a497277bbc310be17a2de0e78fe35bb5f8ffe3babb9148c7e806ae5511bb8f77550a38ad75ed6327f98e24b0b7d318c36ac39cbb51b9b5c0c44e96bf3404e3a774f4c0b65692b08698ff1bc88f23d4454d9629b86338198a2e5353f157b20a5c967b2a54cc323263c517696302212410ffeba37e76ca4590518b68832aa71f9f20edfb577b6a7f246fe39e312a0b88a5debf18d996db4579df100fa89b8c08a6be83a313a24f7de3f52f6d91e9575e3a4ef8000b9fc2ad258eaddf3349838ecb99145c8c5e57d345183ae2fd091ae0d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855179637Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default, Name: checkpoints/1088/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1088/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=1b8dcab67a53622be381e75432d6830ad550a95e2dc13a38b51e52539e1485f295e5258ade4a106fd43d081200147cabaff6d3f6b6c613adfc69f77cc3f272369fb7eeffc226b558091b28d64ad00f349929545617098cae4419f064a6fc32eb4d245e17b83a7582e04cc02a5f5da544078acc757d760a9fb60348b0d1ce385f4369211ca7158077499004fb147fe563ed11c17cd60189ee6124acaaf4ac0b55f921515ad7c47c5ff685ffe476b5ca4829bc038e26d80df8dee36c3af047fdbdfd5f04f3657fd588f870807260c6262501d51b58b1ebad6721cc1dbe6c164273cf1f00f6d887c3714a28cb0d3d0eca52bfb592d647395ce50d7a66628f635fbd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855271751Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d, Name: checkpoints/5632/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5632/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=023730009cdd0dcb5d2b3f832ae45690b894d59cc7065e2ee8196369faddd94f0af280b683ff82e109544d63499bbc7d939e424ed376bb5335dc3d790d0fc4ca32d22cfd17e64ad65d2a0b79ed67a7957b8ee30ae39ab37d550199755509e04ffb7c541334a9426e068acf6b879e5148a90f8cf23e51c708a9d53440572373df0df28b7f26ff88be572691fa6b9fec489e5f4ca525389ac45690a40b88a74fbdcd1411513dbb39b938463da0d3162a688b241626a630cfee1d017e6f8f3c880fbdc19267df56fbbf2ddbb91918260bf6bdce0f0070141286aef12bda7871a9be3a0d71c5ea04f1f7c08ba5a7133d9c4e3de7b4fceac8cfce234918be119a74d1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85523522Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/d, Name: checkpoints/1728/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1728/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0e166441b0758c21acd4e02bda852ba01bc1bc478c2f66ab4fb2303a1b06475710d8fe86dbadd8f61de91c3a7c25112bad766539ff6529d3a8ac83bffc6fc9580262214d6e145e4b9c580ac910017c58a77f0b90699cb1153ebf79f7c5067bacc7858ee28117caa57d542b0cdffbb03370ac2b2b72497cb797768fd63e082973cb898f848e41f9049fdc9ba2571e812c1db96fd09c292e7b5fee6c7b4e09565fb08c23d870962a7b106210acd36291cd22cfdba3d9bd5d31d82dcbee2f3d444394dbc11abd0241b6f4998e083b95d410f1d93c8499c3ec17e17d617db314ff6496e32f15730eb0c37c6fa1e2fb87b78e4ee0d95257d87fd7210782bc5992331f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855163017Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default, Name: checkpoints/1728/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1728/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=83b8dc6dcf4eb15b57c3304a363848390a5bee3b504c621d2f099d9fb8d948bfe2c6e6271080d645133581f1576085e8082bdacf249c9d3ddfa1d9f47ed0795b8e507c76c4be70420e0f2380ba91a76fe680170103689b011b03cba5dbf968f09aa231268e12a6ea13008ff3d1cf114132d35cb9c93397b7ecf6c8cf5e343b03763255d97417c7fb7c46a6539a803d3452fe5e507a60caf3be938e60048a599173689c05e201e6910131efbb274264ec5fd55aeb27b2bea3c31f09c6aeb6abe657e090117a64a894bd51f632c9b5e7fda8f7fe3f83a0cabc89b29c58ff2ebe40cf146c1e41196eb86208ee16393d67e54b772e73e9b4acc0ba152fb06681f323&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85549561Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d, Name: checkpoints/2560/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2560/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=6bc70aadeb15666e25f558038e1f25a7a72a6ee5ff60e1cee5a775e81dae5988c9a7750eec760250ba4ac4fa2855d4579733051874c662a133b8054e3ef0f09b7fbf05617141d5f5714adbbd7ecd3c97f1e4f9c0c79bb54c9e10962a3a663b62a5a6eadd0f35cf52c4e4d367febdf7b5d96b2eb97e413bf614e10a0a7d85c91522a3a03f8302f360378c10c648fd58c5fcc76d85f8217212d410f7c5865064dee7e586c54a124eec06719afb1aade8117f5c72ea422785416dda546512e8d6e3b75d123408bd07cdfa1542a340d1e69718312d99974fbd870b16a4f731400080a40cc500ec29e1729eabb5bbee3089ccc540915faa77a1731f86a05035e6d15c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85551343Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d, Name: checkpoints/2368/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2368/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=83f3277e2155ae3045cc2fbad538fb518039a9f773166dd991383664b24f7a24f2ee8c0b185136b38972df35af742977d86e34b0a9909a48bb8a0b9d29ae1fba7b86e7c1e56b831d2c7031d2c428c80324f4421e92ba687e31e2bafa15e1f661cd0b0fc917b10b79ab64baca96ca487f7276368738d00696c938dcf65ef350a0d37af6afe20d58481642478157f6c8313aa48f0cdaaefd2e3f357ced2526119be8acd2a5e4ce562467e153cebd5b25ecd06c8c00a0869d479d25770fe7979d6f97629e99d09f877143d22d8d5954287b46dc1c9d94014ac0c76a4ec4c97d39cf80336770cf07157da9f67b38515e3a3b69003bd7f4152df69e9bc8bc0da2bc05&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85552293Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944, Name: checkpoints/2944, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2944?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=8195d8e77b0eb9dd44562d19193fb2d49d5cdf28825c0dda00d761a5293ee94db81955f283f7f76c4de860d7e42d670ff6bfd5f92cca949cc8feceb0cb8a38a78d8d966925ec27465967f17b3e13b02052e1e9b0929c7f9541035fd1380737bd48b1e9615400ccacc368e167e138f8ebfe88a349d34f1dccdf48524d9637beb679921dd85eb1ce0a1caedc64dd4a7996282e0039cc42dfc0a05cbc12f271fa3144d1ab71629e3009e493cdd3fe1405397f53afa82ed7ce0dd1c7757a34029a898e1e67f35d78fdaef41f73f6788a604d32e9a23bf103afd561db20841534fc310475e84a074143b5e4882bf88367b992b397d7cc69d3194222ceb8725aa87ee9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.853308755Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568, Name: checkpoints/5568, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5568?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=56486ba58ccfe144fa72310cdc7b1575adeda1fee5f0bc0566abcc09ff7bb932d3f8d788c279494ecbd39eb120dd8b1a4af57faed88ce747d208da68e308f6cacf21faf6e3a7b7aa8e91d42889fb952ffb9a452f5d3ca34d98e64ce55edba1e55e3565ac17decac9b40b58468d36df9b5242aef5277c10701fc56b547a291a92f3c5a1ea5500cdb2a8747b3e6319788c1966c4d1a8a6494a861da47fe89f16bf1f17a706541c315f5b69d89ef9e7edf56eba4bd77563419bed621d7c294454b8b0a2fac3676742d0d145f57d1195d8365270d298ddb0b44d60ba58277b4c37fe38ebf516a54a542e86be872d748129b95c0eee88a1e81a8b39b465e2292ab64d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855875125Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default, Name: checkpoints/3520/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3520/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=86e2fc221bc69f411138fafdeb2b6e32952593c664def9dde10ee49893a6788e12ac2451d6a94e1ef2f32dfeabaf67f1115edd3ce037a146d3f48df5dae1241fd1da20a46558805015e571868db465ed62f177b994e98cc4bcddb2c45b2580c09328640db6c3fc47874de60423c373cdc06a3189d0f6b589f372639fdd923be8c97a5917857acbcdfba66871cabb23fd65ea5627e03cfb47bb09a667533e0d724d5ffb0429a3416aef43a00531ccc50bcf092d91f7d5e337d0cfc4b0022025d218155c3412952b63d5f4b270063ce5fd0c7dfebde83e4b82538f55845dc2a2e8af4cbf50b6944f8bbee37d9d062b3b28284722dcc6d0483ec480217beb5273b5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857546767Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d, Name: checkpoints/1344/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1344/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=3f9cacbebf5ac8b7fd51672e370295207e3b2b1bb6b006a10bfa4e67a032de0f414b49c4ad8f407ca57920f43d97dab965876f355596ee272e4528008e52074df4005548284080108746d72dcf9b0cd94d6fcb00dc1ac020a5d15fd311f86176dab98ab1b651b66e3f096002c214f85d8b927326415100654115da3225af242ce3e71e00c6f1698c7ab77cc66c431e380cffc6802fc146237f1f3a673274afea3a063831d596ccd7e64fdd2518122fb4dd183d0967b9e9dc39837126249a085abc0aedf2b0b0c39a9b591ac496f0fb376ef7e1ed2666e3a0556de5879259a1486396d05a6865ecec12853205289b0788c15f02b21c6eeb4e2ab58888d7b9bdc1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857579625Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/d, Name: checkpoints/4800/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4800/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=51f853861c0f2ec3ded107ece93dde096191014b30f3b8977b0a274e2e0f0401971c4fef95389384f1755daa4a6ad24f54a01028ba5d97378b3b5bfe774690a51dc622ad670e16532eda19bb07c8c721c482c2a6cddcffe9bb7dacef190cedac65519a1e04c30e7e457124715974ef6c6e19275cbb207641bd039021ab32d5a50407b8bb8e8e4bca7144f5ef48b60fb98add88caa58e5d52e77c0a5892151326b0b2140478d8030d2c0ee616ba368b98e11d585465a2f1c002ede8e4f8661dc3b786643d7ff3555bed9e382003af47c81dcd05a267005149dd46044ce824d9800b07b985c980fa65abba6dbd2327f13a834f70d2ae854994e725e17ca62a3f5a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857603966Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/array_metadatas, Name: checkpoints/5184/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5184/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=6b6dabf1811a7a959cb793b4ceec808ba3a3a024e523def5c4da1a882842b086e403c932748dbf7e6acb2d27e2dac1fd2928686fd0170b79316690bbbe514a9ea3e58c91895e18321ab799c67664469be35678510c6d7d88b372fa3511dbde79e9717e04127ae251accc162c2ef19bdb6c697a01b32455d53c1ffd5a42772dec51a5840853696a042a36cce798c9b212d5101cd488cf6709b51afcc26cac8443484ea60a770ef8d699fa06f0378a9754ea27246b97a52ec99354b61d02727ca8d5c7f51fba58f2048ac66bd44c4c5ceb649a5e127175bb1314ecddaace93b25557a1ff3c9870ee8118dfe247a96069fa277a62bb6bb5517d95015749765138a6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857624615Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/d, Name: checkpoints/6016/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/6016/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=201ea02294d886b4673171fcacd4c736901a5c64d54a4055b25c5aacc2f4c82e93420cd4e82925c43796d6e6ae1621b63ac5d86fbdc58b18d0d8b758f0dab83abc96fb2b4a2d504b5edf6b687f4d2fa09adc9746a3394c41f69f22da59e2a20bfa61a6dec87509c55e03a5f876649e2d2926da8ebba0c904b33d8592be84c612d88d5bb179feebe4b73b6fc52f6b6841743f3b0521540dc0aeb415b784eaee6e6ab928c59adb25af1c3f12bece9bc45ce7e1a9057adff386f3f252c6ec521ed9592e19fc3916e7f2f6db982ff3d09c8f81410b476b2e3bcb36f133b9d2c04399b897537b39addee344d5bdd8cb648afe8f785995d8e14fe5b370d3075b00824f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857621014Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0, Name: checkpoints/4608/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4608/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9b72de2b0f72c3a8548c589fbae7d8fc6a318307a4e4b83f206228a05881e6bb4343a8c2edb8de7d9fa43bfd7de9747c2ba92cbd3ecbf2c16adfbda748636a8fcb16c864ca63598f57109aab9dd032e49c9d7dc3f95061149fb988cf665fcc180d3c74087a67a6087fa88f52be1f26e7699cb84f1d3617b32019917c3977537abc7221cbbe783d1048605b00a69c50e9b4cf61b1b49436f845ba1e83ce7eb519728bed367ec1a2bcedb6106d8dd6ad029752a7736cfa10d9936a530ab15e59675beb7efa9ae3de82bd6f747ce503868ffd042fe9d79f9024fa08a5de8ca6c35d4bb0f07fac78d84aaa4e0ae77b0bff5f738df58f59cbfca424b84a76d498cffd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857660212Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0, Name: checkpoints/3264/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3264/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=ad1afe106d07fe7604360d197f9529cb4815c2a7009b2402b4765eadaf72e404c34d926e94e3e49408654f59f62b02b00b2d0ed78704d8bdf021482358c84f9ee2167af2b30d2dc479627d9188d673a3afcdb6e1472ee85a603849cef49ed9234cff58a20573bda2aadf0c00d2586d039de4320f6f61bd4bf73dba00dcae21a771d3500fef258d673984d3ac1a1f7f85d81bf2c127a61b16a67d8f05f82b456ff6aed4c17926bc1be2885117eba41875e480f7b0c465fb8a789af923b8eee13efaac3101d3cddb81c605b5164505d9aad9152897002d5f33150f30c153fc572220ac353e585f048121b25d061e8e794768741fee940abf6126311d3dfdb0dbb4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857668411Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672, Name: checkpoints/4672, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4672?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=3b01db333e94cc7a21a5a4b04559a6cbb9c7f83e199bc8d0abe53f97e82479cdf0cfbb2a82e0beefc4262871e8babe8f0527efc2dde261937da522e1c9aeda8b52f232f325c86b89ec52474aebc3975f12ac3e9ee4a134bf1b3b01adfb71c435e0f2ecd5aa4690b3680d646fda80443381b68b511363e60033f284d3f3d9b619969b1663fd064129c46ce4942981a3794294e40c98f5443fdfc5399675e3cb641e86781dd38d312dca03ba3bf3a8a8be3bf951795e5c9a9c51c30c63421df980b1a01d5d0770f9343cfc278fbf4a826d3bf288a5ce8552bd3f2b6e41fac197e2656dbd16c30e47035523e34bd62a42a4d19006c9cfb06737abc4c48d1d47b42a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857683078Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200, Name: checkpoints/3200, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3200?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=ba8186542c4d25a7087caa7692a766e30f03578c8c487071403a055bf52f6db1e83bc8977e258836804320d0ff8da5ef78a8969fe3590a30459cfbfbc1aa7d600ddbbbc6c78c4b50c2e6128e09e408cf626765198c4ca9a4d25c7109170aa531fca8457fded9438fc6b978e2fba1ca62217e6266487c3e5da45e225103c19427973a1175e5d7ec427b6b9e022ad6e7089a47c7cb0562f0d4c00c30b7d2f6c429b7bf31825641f5d26024dc2ef961d217bddea9a53802c5fa21923d9de508f1148b2c6477e21f2706655c803b4afbc65962a3629bc73e8ce1a23cf2fc56f719ce16dc6b5ce400cfcdcfa9a9f1944d2b4291322db2feb1d2f0b9336d1ac856efcd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857703369Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default, Name: checkpoints/512/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/512/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=76dc7651f18ab6e3df67d0e2d510dfae1e8d026fe3ddd50191fd400ac06776051402fb6a9a45e2c3e9f9e776c496ba789d7cc81beec46f46975e491f77dcd3f4f0f21d22c572ed91f604e60e173c30fd93153501de255bb7ec7c247cbb3d5f3b32792c21e1a29fb067d25336028a55fc136fedd9b928e4f61dc8eec21b2f5da6b400d4b4a760d60e82925f0cff695d51b02aa9244d412d83dd3740e3c5c2b039d786bfcb1430802b7d87352b6d44d675aa3acca370c99200950c859b623c5fd4d6c871060aaf47ddf45f92acc6b1516bdb86cddd8fc3b6940541a4f62aa3ab9144b6fef4ccebc3d68d977a98dbdfdf19d602ebf2dbdec8d7f4ddff2ec9fc8039&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857717035Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712, Name: checkpoints/3712, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3712?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9f2fd87505d31555605ecc976d06a6372c68379d673b3b168ac29624eb9402d078f4d14f667b8fa1f4e2bbc0fd94a3901f1f146d2f3ee2d63b42086cf9c50a36dbb568ec345bfac8a3b6f0219c4eacc570edc8cefe4217a56b934f63d4ef29933dca7200418a58ae90dcc486616385329641eef1f89037ccf2b9fe399af47d6aec1ebd866a6f3310edb6ebda7d4a6759750caba7251c911d791236c631ce9f53187ae22d79e03cb78b71c1f3445813f83a9344a149360fa7aa100ec9ec002fcc8332ad3942cc00d6274766c6fa386cea4a979c5ee6fb837ee020c3873fd3994bce20e216fcc8b3f64c87f00dac750362ccfb6fa1dcd21dc9ff9c4e59915ed4c0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857750363Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d, Name: checkpoints/4224/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4224/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a6d495ba62e8a353a44d90eff0e42fabcfdccaa125f3d6f83c59918fb3eaa38514afcf1ca0f1af538a7af1a4de73f22520e4bdc3cc788136f64b4bc57a7527a21704ba89a45c73f520580b997b7c4198eb4a2509b2623818da0cd5cdf1602fd00a4dc0a8bf43f5eb392704ce831c4156fd61fbf790bb3ec07939be68fc30920d02227e151f59f1696bcf304b009438a7437f5fbf5f542475ddf129d2fa2a00960ffb4320149b113eefd59ee70a5125d5a67398b043a480e837c9add57405e6a81888929a6155d3073fc2ce391860b4ec210fb8fea9fef13b231d55f36c3b701f3fa912f1a6f8ec25b78aadcde40288a1b598bf454c7e5ae8b3e019bd2ee3a296&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857783789Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248, Name: checkpoints/5248, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5248?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4f2caa9fd363a9666fbed2c37c63d3c7891f2ea33b979ba6defa7207530491a364390b408dd2830963c7829d5b7d3a40b7b2c3f50a2a5ea1a83f2e61c340d466f6b1c3cf86c366c8bf6215fbbd116c409df8cf6af89932fd33d5ff29bb600808f7a1be987388a1393ea367ec765cb873081a39dbe88826ca87e60260867ab6f168d244c6c37baf36a38ac46dad6a507d49dd0dffc16faed1cfc4ea6d22e225e5e134ed95354e4d8852c18e4c851b0b9492d2ccaa50d3e2f43e5cf80d6f5d97af4aa15461cbe6ea0c56c8e1993b649d0aceadcac4b0b764d8b9022d33ff5737d1df9692bc3cc79d1761231bb9c29ef513f08e3721d99ebda42ed1d5410cbec07c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857798062Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/array_metadatas, Name: checkpoints/5504/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5504/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=abd693fa9d1e53d8ea2ff5886dea1c578ab9bd8448775779c89c1c5eec36bafc9d1e787a9fe70ed2d1500370ddb157a5a8b271e16651bfe0ff6e7d602e31b9d30bf724c9778bf7022ac091843b47af6a656a050e12e9c6b74b8e1cabe7df0a1721ee290eaf45f546a2ffaf182d758891168aa454dc7ee0c711a62a676fc95b48bce238fe12761c3c9a723bddd5be51037c188c9b2d37ba65b19e975deae167d928edc19b28c0a74a126464c9189ded17113e0401705206671602a4944238bbb79f1c8651bff61a2bc55b0e015d9337792992a394ab5a04c5dedb17af2b8a75f62cad22bba9f5d4bc9df0b9c41e58a0d9c2215ef8af1b9e63ba043c08a13509fc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857799954Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d, Name: checkpoints/960/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/960/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=aa016d19b597f7e437fd9e30ad79664d9e24c329354ce72d8dd3eb020ed4221a0faa153c7e6df7a591b4c63df50a66911f7ce510941286146bbceba26a2a6726ca90e68cb0f6059666a182fa293e28dfe11ab2490662007fa692b09d19c6e528c37df01fd1b04619d05a1bd8e0ca95c1a4d2a5519a0a7901f2a0a0a749a3415e086325979fb0f09358ea12b9cc67e2536eb8facdcf61b92e951035a0d87498cd41a90110af00be33954c70f4b3c614713cc25df2b4618f4a3767ae9aafc8e63617fcbb9b954a5ab28cad543444abe41d953455da3bf963c652af75fa6b48dc80e16473b16cc342355057320e5bd688cdc59890c1fc67abd5730622daee495270&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857831831Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664, Name: checkpoints/1664, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1664?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a4aeafe5e6a1f052b022e98a4322e42f7a189b204dd3cbfbb3a0243f4e427e581f78eeaa96c2370661c27c4def26a8ef888e35c4bff488c33d6c187318ec991eca4d914df6a162bcc7aeefad2035a07ba9649d49ba22fc73935c05e2ae0e7a24fb645218c5fc3417d081deaa7a0581dbec6c1024040652d863ad848fd734c04612613a2e27854c0c0329f19036b8d06a123b1b7c8f831317cca55416c4545f5e481c822ab36e909eb69ac19849d847e03d541e02729634e01f7f2dfd44bb0c2e788f8fa23fce2e8db8f00bb52507de90ca066b3dd66a03594833481e029ca1a4ae28598019b24eacc4a707e1648703494aa25f3a02fab8e0d94ea663cf79eb98&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857849504Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d, Name: checkpoints/4416/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4416/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=3583163b6e3f534721b66bb425addcf47740558c16ed729d3b0b17d4cf3589cbd298bad140e1f427093fa81749b2b80422bd7e88b10f5d65770a24435dcd8364788aabf31d568add474120cc72074f6862e4b1404671c1994dac99f1ad04bfde0c86413b2971a96f17a66f8f8d77f7bc7104c4c202cfe69ef809db2318efec30b792e8155580a75a6fc6e30ebc1135530c9f6fa3724c29c6071f7a8d6b49af1f18b82ace4f4af8a4ce4c9df179cfa4e9745d18314a1fb20429985707d6bf72af6db2c5be3e78155819868eb98fbef9865013d59a2867c960a109c462815990ffdf5f35b9435bd7468b1369ccca5fe2f02719aa4ef6e6734dbaa74035d9fc3364&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857859305Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0, Name: checkpoints/0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=538956f019839a16b374c5137caf15107dd4bd75305b9df1e009ce263774bae13755a28f113442a0da2009087e99ba7df8efc2b97b71475fbe03f94cd7eaed9e7e7368030bc5d88bd8f3724f82e838b0ce059600ad7450e97c373f4af7c5b838c1d138b0e9f84e5d9e8e350ca0701a2f5e871ebd8d024e79a439e93faf6e20e1f95c046eccc913e0b59db547263f3f6d7aea06bfc097a9e9599be60b95285d367e765db34ed4091662deaed1bd610ac6f9c578391d8c5126e1e932a960f81b22fbd261f1db689ff6ad3498431ab5b81b3bac10f0bc8de520faa96326ef09d0b7da606105dda2490c35b590cbc012a635fa7a5a198147d72acb1a442bfcc94b91&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857879161Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0, Name: checkpoints/6080/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/6080/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=025c154504a3cebddb63990c6a0a3ba8ea9ea020f9b11619f591a1ce49bcc942d85596285371c6a8db688aa38337ae013f45ee87b8ad732eff9caf6482d76ed69ebfc0cfdd1c298d8215016e65f98cf1094fd8710d7ed603b754c0721720d8b2e4fcedf3002fa45a880c9cfd191a3aefd01a3bb4fd950ab220cfc9fbffacb1b6176cf63469da181ad25ba6a4adbcd4118a61d2d3b54c0f783356a4edbf3162be2b6bb5eca5fe9603f29fd57e28592ede61457b2d8560897e91f27a610c5c0a74d3a00b4e2f59e757c0981b588bde3cf3f6b27ce4d4ae8d85fa1faa59b7670b6bd24b0049ff801e8f68227e0deb16276c6c7f8eb5bb67d9fb6161e24da5eb79a5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857907934Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default, Name: checkpoints/2816/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2816/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a1bc30890fea7f5c29a07ce4f18300a725bc6c1aeb6cf38bbbba598691a788937627fc36018757f2195d9ba575f58ffe79377a2180c2a667e7e99add7d1a8854e45c4ffb9e55e3fa07c949bef6411da9b675f4bbd0919f3cfa76189ccc1b86d91bdf4659ed381e12ed401822baedf5bbdf5bd27495b5ad9e69d34d0cf948e5b8a33d6e06d2b8ca8ca8007ab6b099fb6c6126fa7d8ce6007e151d0c10191f47fa72126e4e2a643611d0a752c263f2c64ee6250c0d368bee2dfe345523563d9ebf9b2ee653a07853c1145a380af0e3a86b9e25ec65d058d4aa89a7f20e3141a958ae87321c104996b47db8d11abe7b5c862b9b81455f1be53ccc451e09e38bea4e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857928596Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d, Name: checkpoints/1152/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1152/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0fa04d2553135c7aab706c76e0efd3e0113c68c08e8dcd0b7e596571a8ff272b71f140de9da7552d73f9cd50f4338e1381598638565b4ce356ab90936465a34a425998f6bc4ca232f03785cfab6cee73d5ca4acfb4d3ec0d3dc724da8f2e2b1fd3f031dc5bc52f3cd838f975f6ed86c0a610a13523ca4b87c1d70ec4c4d2e51191f10538dcfe970641b92f61d75f4c8367a2c051af542e96d371e11d67630d0ff9eb24ec4b44a45b1d452d587f6961c2ad5b41a28c4c300bd12b20cdb54692005bd31af3b9ebe1c80ddaeca5a9452baef8394236110c11d28ae96ba354ab1c64fdce12d96185b698a53e1df330402fe7a5c57b6badffe305b126836d780c1e7a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85793962Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/d, Name: checkpoints/3968/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3968/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=384382d852d75776e4e0d1bdeee6d262876dc44eae01ed68d6a7d7c1af79f217d29c1b0a008e973f58cc4e4175240068d90533dc9ce79be0a7a3aa6e190793a796b1476ac866a34f58b07957a361bdc18320739f7e9a527e080ff39aa86f75eb26d444b7a5b31ee636ca6b3c941dfb1bd2ceb7b5add8d100f1aa22df295b2ce366e67ba5c850e9a04ff6f157279b6ae43f37bec998c815c7beb5a29551457740d5babd5481baa0c22a024be05d43220f4afebe5fe5469d4f2ef0af5512df42135afbc729134f85fd3ae6eb2920b7eccbfc4cfe90c5d028f6e8a413f4ced83a5086883f0dba5865d80f6bba3f8dd813dca62e1ee00513db6dc86616654542b87b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857974282Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d, Name: checkpoints/1728/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1728/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=3097f8272c212989adb0d46bf3383ffca4bbf1934d78fdff152136760061297e6a83d8ddd64637ff66d9c0198bfb9f67bb1681eda17ac21e4e1902620005a1c316e4dc1417963ba20aaa705fe54170140dc76daa99a18e0218c93ffa8e300ef730cd5dd87d257dff73ca6a9cbb7736889eeedfe8a8ac257d7cbabb17f1cd08dab577a21653aca2167aca559da4069f8ce8deba90274c22e8fa87391913f8c1686a125e51a0422a895e8e96345fc7e919b8b2686f1631d2b1d9d185c61bfcff3adb88cd6691b294a7b26375a8b9e65ec60fa4db8cb0979fcdf87b5ab41d2a4d2b2a10b30a3a53e11df440e5e58b26456ceeb434a7c934510812681c296528d34e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857989969Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/d, Name: checkpoints/4480/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4480/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=28d63e7e1db44cfb926ab1b40f94c8e1284c4ca22bda9224f673566479ef26e29a370379b3aa7fd4fce03390ab5e5ebbf935ab40a62e42e0560610fd86be6d780f444ae4380b4381cd1a24afa6197dc3ca8247fa426a0a41bc164f555e1a05c868f0e61c6823572babc06b4685fe1f7761783c43ed4449be437318468c3e1780c8d72d0cbf4d61df0567cc449e5e1956da12c72a6075549b4f6084b59a192adddf039cc263a5e5dca73b2d8745eeda08202e826cd8d454b9bfa7c91e110e7132d8d9e14e07c2d02cbd8393662eecb62db555f77c0c9d62b1d3ff7f9e6af52cd774b16308a09b9fb15d5a0c91e0c522b4e8648d20da4139ebf00baf2531d12a5d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858029015Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984, Name: checkpoints/1984, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1984?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b68aef7fe032ed2dce42f341bc202e2e404620ca13a6730f22943094d9bb417c6f5e26af8c9ee821e173ed30f52b187834824e7149745eb3bc3e408114a0b9bc4fcb2bbc0a037988d6fde3f877569e454275d4ec63d36705fbfb7925cb0ae013e1e569f0957dc649cf528d2feb60911c5427159b79dc7f94dddeb95b9cf7b45ca8b369aa2a8295bb327160d02060c0f85cd6bcdf257e4bdf82563681164f593bfcb9822d9f1dd3a7b0a2e89b43eb7948725f5ecd9795ad49b5f30e3ad9732a8e74f3294df67f9e1c4fe965361ad4a2f4c521ae7af13ce9b3a7e91e3758489c764e02cfb1cccf330e164a97f8094c3b257a0f905a31c7125f0b8ca2d56885de90&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858043788Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/array_metadatas, Name: checkpoints/5248/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5248/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=5a63dc0c3309af443cd6173891be999f9102baa230878e84da36d2f194fedc58f06888b377452686b9c71d099e634364d2354dc1844b4f0e5236f4680a6ef46a07cb3bfea57f0be1ff036d15fbc6064be7f5b69719c7351a446aaf3922e1d2aa79ad653931e6c688d7176098d84a1acc4a1fd5b649545bd16c9ba512e615c5b5ef8b289d8e8574a4fe34d9d92c86e7178a007555276ab18c14f5adb8238d3ae04d264737bd448e68026570fc59646745b9b35992af9b04533a4fce1fe84b9242f82cb85e311be8dfaf84f9e6fb936893ced4f849a52c64f3562b9a7ff51de32d82f361389ed1eaffdd7169c69d7f4fbe4c90c2673b9f8d95869eaf50279e1bfc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85805198Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0, Name: checkpoints/4032/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4032/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=c31103c263899a4ae204f2f0591b5bdbb77c7f4afafc5c99e73ae57dbd240d249a0246a1fed8e052eafcd03b3d07e5619003d4ce432278209d84e41e7832530a310f0a343aa8d511a99ae7c4bef5701ce1ef6dc8fe5808b519e83bf4bfe68648c930e38f60fde9038c69ff3fb97c81bdebfcc08ced934034a8d4c273f9047ddb7df4e6395d8737f14350b0a1ff83ba62d719f46415621126d0c674762fb3b12525ddc1b728e7f0069e63a67a1dd3c99a20dbf35d305215f5e0364843067ab00c4b1686aade0155236069f3392c6c229ff919a79d273b9c5b5cc7a4c57e9056b1e28ae636be5da3fad2e7138ad904dab18e6e431c74a8865929dd903842e1c83e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858088408Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/d, Name: checkpoints/5888/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5888/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=3c701f92821dc608927625abfe27af4e73b332a9aa3a50ffdd4da28593efef96f841ac66feacb9efb6822ac3ad83dd0bf6d03bfd127194a17573adb8d4822e8132c756edf55351e614d5b68a593e7d6c2ce9958b872ea25efe4f49af257f00b960bf56a856e435dda8a18956616b4e36a398b6283df50a74823adfec77a522c2b800f201d6c944112cbb16a6aef8b71fc2403090c8386bad48692e97a55ac8076fbc17bfd712192835a9d8e5b6286a9b259ab2c5de37172ef2c3344d7db3bba251fec48cf07ae3f13f431b006fca847efab5ae0557a62cbde4b9907744eaef34ffd2a6cab976b3bf807eaa8ceb7326a5700f7b86553b7dd92b15a15f21283bfa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858174136Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008, Name: checkpoints/3008, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3008?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=89e6c09ff9d1d4213e6d42870cae1109abc7d76dd3f00f4f3f84150ef9519a9f2cd08a0ccf400f6ecbb9bf103d4bf337af88eb5edfaa0c8552a32e7475b97ef74f0478e5a654d77e1c9fac8ac953f8c9daf4ad586294c6762997aa1c34b6f06cefa6b9bfacaa0419bb0d793d793d4fd5b39448f328bf0a27cc305b86fe264ffd41a07cda515c36f481727b91da5ac1cc6c23a1ff7eed4a6f27778c0f858795035d6870639a330dc6096315a33d5216e62691884f161ea53bdbf18de15e80ccb1af217d6f2f61ad53767dd88cde980c96b14be3e6fda85138da80b3e8e223e8e14a2fb3f7d622af3993b50870f344c887bf8f4f07de96edeeb3a64fb431a7ea4e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858231551Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0, Name: checkpoints/3520/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3520/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=06e10e28efbacf6e20a826e8516f06fe1c793d358935eaf5bf30de459e15207440e5778a9227bda4010e4c61ac689f9c98bce738942d4cb510ad37f4102484c558079cb75e027b8aedba976cb7417caabf8202deedd6adcebe79c168d40bb44995ad57bb610453111bcd7103c98d8c96430239b07c4646848da824404e0247e0f36c2ada464761976763215c1b7695ccacce81265a2b676028e50675b62352cd060d5c8decff64cf2de93aa40985e6de2825c772f96b6519550443d99cce105809a190c3086768c0e24e0f21156e9b820e013d0336c999bb1af4bb25e5a43c6ed47923bf72a2603c27d70b1e4918aee1ae2f5215c3c1131386fe4238d6fb4cf9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858253816Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/d, Name: checkpoints/576/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/576/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=7f70b852d8a51c5909fab258adabd40488c9f99c00bfab071b489833bda9d4e48f118b99dac1861684035c39cbc316635fecad06dc6af4c344ab12e20b3843ad3301163aca3bfd0bdbe2cb18c7c72e923453d1f7ec24a8bc208b32de16e63203e618ccb561682136c648cab8873af2b0e32ad443559285caca53c61264b000f8b496741ac7fd7061c777519fe5d22018bddedf99be4dc16c21e53685c2a06800bfdc0551c8365de2a1af239e410a26cdf5a8584662b79ab6601120b84ab597d38b8709902de05aa993b266bb1ad451f08669077e97f033570a648f1be2440b255f6bb935482e15f2825254e954f8a57c8b4cccac3f8834097166c9bfe9787d71&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858262112Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/array_metadatas, Name: checkpoints/3200/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3200/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a9648d50361b789cef933b972c95d6f90da44c31a10be97368bf7b648c3570210aaee290ef8f51fbe802694622f4d19a5668b2109588c18bf5ee9225292ff842a5ef5310aa6a457ee28a9dd3a0dd71ae9b5d27912937e156475dc323c1c1bf4bc5b0da568c925b49dffb9471ca505a699ec21a269fdd13f4af0f57435fc9122f3c62d11614064316c4932e1283f954c03b93cc34be30c0dd61101741e7664bff740a26253632e11ef531f1db00c5ac07a3a77851ab83d2fefa76803addce32e594537e7d9dbbc289747a4ebd4165f163c532eb5d1a42f6410080ec64569230bb1b848716b5f9d4a75d2ea423db7a33eba2e95c2366800731fb30cbfa0957a0e2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858172233Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/d, Name: checkpoints/4544/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4544/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=6bdb3fd5e0e423dd3164a9f2e5cc8b5811527aeeb9efbddaf0e9fa6568d67dbc5cb187d7972010d28bd55d28acf41d8a8f97ea33fe28640d618ed40f8445febb97cb4066bfd61f7005e3a5321110b735c067b6b75c99fe501d947e6d889ba03df6aeb6f212b4e5ccb466c00dc4c374dd681991512058fac70e2149910085d0e7a8d5b2af7d92eadfbb634a284c8041a6ced1771bbc94e49ac06ad9755f187a99edccc851d35647161a2b9d3e868a8cf7094f465dfa5e09a3bdebbe232a184cbae79e4c90fa7d0255341a5dee8e09d736b9eadcdbe3dc3d1237e95d342ab496c232e43875ce158197f0383565b7577a007c1da78c82d3dcef5dafbcd279ddc779&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857684288Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default, Name: checkpoints/4928/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4928/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a2cf9a8cf11f7e1ea958e5b417ae2b94bb195c1ecd73671078347eac7f58ef596c83b2c2a99755dd9b568270501ae02781517bbbf949a596a806ed308c054822e702c36785acfd3373c4540ebc50a01ab3a4a74de5fec656d3a06bd47e5d8baffbab803c31c8461b6cd381309cf986aa0a523d481bbed6cde904f693b61129a6b96995893d09197c5cde197d27754f772d84a7e7df72bbc20f30ff7b55be17b3784af5425c6b14b9b5f89d849a0d357a97ce6f7eec85fc6e945bebb7cdf430e48bdcf5cbffccb8226de7ace19f89d6c62f5bec640dd4b2b5790b0fd4ba734666e39d7d5396021467f8fe32ce3da54b1031096fadb8b944dbaa1041b0c3a7ff20&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858345402Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0, Name: checkpoints/2752/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2752/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=6264aa4157ca8044a2a6783c7e4b515063c1a3dfa2e0d895ce4062abd1fb1055d2cf5efc129b9296a91113121fd4ecc2fdc88e4ea21cf2b00beef3e736da39446617c7031082cdf8c9c0f4a6a4085555ac97e9658db3e1b549a1bc2c590094a1bb8534e85c52ea86a93810939748cef1e0fbfa614f1203a543398d0cf5bc7bac89706b8e06a842e4e434e673c80952cfe1a510971e68614dc877c5538289bc83b1625359b1a5a397da8222193bb42242424f53af049d25f75f211daf17849d13dabea0c1f146db72dac2eef57bdb87edda95227b8dbed0b07dd55e5df6e0e7f7ff8673dfa37928b8669af9d6b09bab12c4df443bf55c06710148b423aaa68bc0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858371693Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d, Name: checkpoints/896/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/896/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=182db4b06e73836fd9ff96815a482117387fe26ab30da0faf9c6cfc6480a0b36b72c738103456861b880db6f96db3d8e66d84a69cf225e4a6c7fb23ed16da608d45fd24a86ec60ce90124a46f2576a4501cfa81a9f6c8d7340dd959213be07af6e1fc5b849b1fbb1975947c8eada677288e49a3bd276976df4c11d00206f0d858c9ef0f4acef7eb7191a666a11ddd69cd0e23b115123e342383335273232a8d1a72362fa0e0f249feb0dce0554417cad980e70590b581d2befd2b9f15c173a334b6645539587e66f0a2810681601605d452131a9900f959f44737c4bbe85b8b300f2ddc7e87efa393f5b158024ff6b51827af5386dce0b42b3fa968c2c22c490&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85775273Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default, Name: checkpoints/128/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/128/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2076896e70a65fd27aac21a910b160a3608b43e28e87958543f78d2a3dc41101cd4d3b81ab7af15dc1327d363caeec4646b5da25a456f0679f0cf0ecb36b1fe1cccfd8bb4bad2b0960617ae13951d02582475e7b11f0949453dc7f8e75f883a0454aa9da8b8f18f5d6bd748dc24201067c2145272180a7172b69eaec2af57836f8e1c8f4c06c27e30425f51e2f3b6026f97391be09c4bdeb4ab20c26dcf4ecb41dfea903aa81ce94a0159e48a1e633b62bd0112fb107e91b1721f1bff2b6f27ecb05af1a92066c49337966533e72a936ce0529b4aaa41d9d208e36851ef307ad5a51456e514062d67fec408e97de257d28ac5acdf14f47ac8437a0d6ce94be48&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857994202Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d, Name: checkpoints/1984/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1984/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2829d6efe469a84eca2ee744c2c126e31be8392c914311c62beb6b5e3691a6664ade722d79d53052c415084b3a1f5ea10f564ae570a506efb36e982f9f273472a05febf1da23eeb262404247a30d7a8bb14c496c1b3c5fc024854c43b826ff203fd96b3d5778e47a8875660c3b5419790f67e9023ae97a8aca7d552bc6f158caae984c926d8295cc1cccdde31545992cb67f9c79cf07cfb8217e450270286fe4e7b1ef7f09bdbaf3ca167c06b591a8c023993e49867aa05922fba745a5e8e9a68f3eff7b17d7ff7502a821f3c664cfed1f2a78a01c9aa926f09f0e4f6b16d97a700b4df004472a5174b8dd030c14d3f2ff8863d12fba4101c7349a95fc742f0d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855518963Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default, Name: checkpoints/1856/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1856/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=97280faa38775068d959e5831aa7dfa94a6eb633d7b4f951d6105ad69f7a09e46dba5e0173600c1a03cb976be4b209008b2dc1414b574b33ac6849efe4a39f77c0b0b49079f95b664708a5ad86ccc392757e97f87f6e80a386d45eda19a2ca8b11bc6aa861c10cd59b4afab4d7b754f69b43073a178bfe8af458359b1130c37af14318818217ff39579d45b33efbe9f6efe490dcacfa83ca3a6ed4a6a015ab3711940a247e2f567e92b76c15df0625df0dbb2592a15c5834934002ce6e719677701e1c0c09fbe7b88f6633edad8da416d2a231de54beaf492995b73158d5daf94f205f14cc6335b03c9a72838ee3286d184653e2c971cc492e9204a1f9f32584&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857843284Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default, Name: checkpoints/3328/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3328/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=941d573acbef9de69992d1cff849cc09c43465a55783e00162df85eda1d32eb70d5b97e4f90c276e06767ecf4fc0f3465e758fdc002ae8a9f8f1e0d61e4555daa72673c6809b254546ac966c06cf801183fd11b501bb9defc5ca704537ebf4889d2faa87b052c0658e6b14bd547c0753fff6c5ae2073e196ce8dbcb258a6b8cdaa719e5ee62c818a047997bf61b694c9ebcbee4e5032ea2512be303563e14f875327e83fca28bd04d0d6d01a6eeef1d213c5726f49aae43a8118a5421c4cce1fbb31da9d0937016ec1d64357cbec9d05164fec0a1ce94749d1f4c7d8b0bbfd79473fea6d71ea1bad7ee4a223ada1ae77b253abb4174e502afaf84517ec994238&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857872979Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/d, Name: checkpoints/4352/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4352/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=82d1131360db155beccb4daa0e7c18c0d3ed16772c7d25d2137019e6b76b4a7f07d27a26986209b7837775f7f38ecb61a82d408d880facc73449ebbe9d7e75e83839d7d1e5bd16a7b5a4b900e8f211d3c268e902b48f03adb352eb5060c5dab544f8009b67747b2a9e86b588e8b572d4b8bd214cdbf68154e8508003788f14766344fef4a79005250cbc7a3b3689c5aedcd682c08fa365592fa4b2585419b4fe6327c735ada6873e52cbfa86a642879c04b44f5781806166b2b21120d4f11496d9627df81e255d230278b4e799037c1595869163dee51e2638dd4c0bc3bbdf1ec4bbc21268eb1731780455c726b1e52856a18a0e40dc001bf263a07b5a6ba617&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858089698Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default, Name: checkpoints/4288/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4288/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a29cb64a5dc6ab9655a5eed35e49f402871b8e1a68044eb875a92e277eaa72b9dfe291ad1fcf5fcc6f439ed5d0233234cdeb4640cbc24d073b325eecf944f3458bcc06cd24a3e640f15d77cc4b23e7d99134aabc1dc273c3f4db84e33c50c1b9cb998c3e0c157e484b2566f3920797cff5f8666f50ff1e4dc219e91ed394373e1ff40b8b6626f6bdf03757e9199220634d3b70b0cc8daeba41818407b74a4fd76a8f0277ace1d386857ff559930d0cb715a7f3398cbd3ef2b13361026145e7be4c48d541819ce1c332086823ba83876a4a455dba21ad86c274446e8f5bd57dd4c5e6cade68ed9722231057759c84622c041bb36f8cef37e60d7bbbef502cccf0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858493014Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856, Name: checkpoints/1856, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1856?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=acb3383f49f25921e020ee1a8d9c6b1c8402186f9b85e0e06dcc2bde0acfc176ab15bfceb7e471921d22f47329aebee33a6bab3e99281d67ec0192cacde9b2e7d1ed1f59d7180c0914549718eb0199a9887de78dca950ab68b07546c3c4f910c8255d3383e9b7594ec4ba2f85c52e7966f054ee3411a30b586fc475d3560908658dc6b16c56ab5ff74ac119de62e1e48f690cce303cb183f855239bba1256858f7d2aaa53e8a91fdebea258b6086c0e2c11aecc1f69c2bb95ebafa39e97df37e03159d3f65df8ebeea2ef1a2369cd887e323282b13f53b84b662595c380e69fd2c58048d55a204fa77b043b72cad86546581d194886e2b69d0960994aac181c9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85841305Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584, Name: checkpoints/3584, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3584?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=220e4832f39bf6fdc8c7ee6f7f7dd820e1b138d7cb4c46d46c66313bd500bfa42490230be7635dda0b8624ddca3cc361c1e5bfb90ccb1b1810f88bb9b9ab718b90a871a33403acc3f8e53f5bfebe4c002b09b4e1369198e11d0d1425384953363d23905f7628ae4e39650418bda7f7856ffcb89226d711f551f829dd0db5be6b43b7b0a19cfc3d847ae4598a4e1c4561a0ea3246f95cf16e1bbb4f0511657cd81643ff5cf528b90773bb8181f93192dfa07ecadb1ee3d0384b00486adeb9ccf041fdf70eead42f64e9a09e1bc005a1f9eb8f42fd3a4f1a295df9096cb70960870920b0788a47fed7c7736cb377a78d5bcbf32d9dd92ec31ce9238a3efbba50ff&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857892015Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d, Name: checkpoints/1408/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1408/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=3699776f20e66256b15ba4a84eb2c24d33cd43bad314db229f8c2ea445d0fcd686699a8ecd87bcdb4f6528707e47bb25746cde9050f7c872d0be9e23f74ee1f3ec320ff999d7a4364d46bb4c561b1b340c7c82fb1cdc1067e6276fedb4e5fd3cf43cfa4d8377ba015c6ad5e1d6e8bfb02ef175f7fe3e2d8fa2d89c08f72d466855c01dc8a54dc45c577b94fb8f89adbe07c68f7ea2451b6bcb32ba354414a2a2f757b3020d6686b02432dfdef318df5d081e5ce424a7815b2be112e3e35d36b477ef20c3b66e9963cfa2f9eaa0304448bdc1f30f6e68462aa1a45f71b86f41ebb146f3b58ad97f24e643cf246480161719cfc7e02f9d7bd72a9a292b2d9e0660&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858321228Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/array_metadatas, Name: checkpoints/4864/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4864/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=6a76284ff7a4aa45dff8b61c52b75f2dd44fb2b46e0c8348df0488dde77681bf46732929bec2038648350d1395c77e76641ec1b931dc6d03524dc972d37afddc1724d5e069e6f7534347516662219c0e558cf01f327a97c297f243731d44436a42294e49777393ad87592a4f4c888b062f69b26e05a109a09da731ae8ee23ad8fb23b828e767314e527ee5591508de80ae4715a9e6da020e25a5115955af368e9737d63f90e5ca69a6e6b66c2317f2643d6b3de0701abf0942f694a9c1dd20de3ad1fd29750770db1acd64ff9ed752fb3719ef14b3008efce96e120f7995d2fe2041346960302eeca16d5382631a6cf0a972f9e76a21b67aea6636d01f616186&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857947843Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/d, Name: checkpoints/1600/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1600/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=18d9ed7df3dede32cd4603d23c312e2d84afed41415f9c7318185e72c57f44698919c17a2862cd68b7cbdb56e43bf9dc6ac2c40210aec7d3b4bd999f7716bb68bb3e8c4ebcce0b91ed834336676ac02d650f1f0b2b99b07206c92f64a069a32f8927523d7edd63e761144c06985b37a4af183b72ac841a3bcc17e5374fc1d35c4d240818f3f9285e685ea2a21fd9dcd1f426516cb12325507aa10e6b66f0ef0cf3419ba3391ed43218285360c432ffbf694482afa83b829f739812cf48ec788af4fd9a16bba94b673c8cc4359393d461d53e175a4d4c24f56beea6798a793ce20128d92e8039afd9283fe7cd29d5ff6a9e87e47ad7f1b35f009d7159e3651451&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857914672Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d, Name: checkpoints/128/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/128/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a7214ac253f6c41615036250aa50640f795287b58b3410435edb4fb0514b56febea626c344dec8dd536a689caf788d4ef4d2ce7244e06dd958d594a2b6533e4f4c7e094e0051ba56553243f2784d4f3ee7feadb796562fd92d2d35aeec218e409c1aed633aa66c8dd71048419892628bb7673e40f73f084417821bfde97809852b0fab6c8f796b93b4ceb19d13fb71ac2279fb0a138ee61bc8c37e384e0db4f40459b2abecfeb65f3c7e0fcbd10fd40ed8e6c32db4d02426aa0c26f85c8eb63509a6da3b19590f46c5ec4453d8c2e06e59fa0efb42af3959afe577e04faea442a3cd987c42f20a58bc3f6b401135d3fc21a4eb51a2fe18e2cb8f477d24e11499&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858087507Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096, Name: checkpoints/4096, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4096?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0296e2b78b5a15189db2da8d25d248cc508d2f929e0d1dadc26de52de267f0b59f1cdefd3cdaba75c8ed4e443677c26fcea8791810bb8ebb5654e2e367cbb08f165f69b44d8b4242efdcdfbf745240604a70248a4b923cf06eb8e03ba71b282437d8ae705031204676b89e60e08db1ba6f21dd5cf2dc7083aba94d075cd6c7d120f856cb4fccb9e9761345e2d2a41b646707df19879f5b2ee375dcdcf47bf0d2c8c39b4db576838cf86b9ae4d96a437c65318b8a498f25541936a8c0e9e459f3f48d303ab78f806674fbf08491c14e2b99bc4f74d180ec65ff2fc022819b129527577640c4d8922f1809dc0ac95536a54df3423e009eb6863f184780ec7cdcc2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857847987Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0, Name: checkpoints/4544/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4544/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=5273b20aafb99df0a9dbc71ab799db0dbdd0f32b6be7f5c2204646d2682fcfd727203c632a99b22f6e13738b26936e9b6d22517a0899195fcbe8b8258e6f94839554e06725a82849eb453a5a8e62183d0fb4500b26047ac751613ebe9072abe56b7dcd90b2ff662d625734339d14ad1f1431d1a627a6d2efb995210560a872839af3ebb247b69534e1d953d20f23b3ef86ae6aaf93432c4cceeebe01cb20348c48467eb5c4f85a0c53518039bbd52c751a3d0b28d9f833d7d06dfd45946bf8ca270237382eea56774ae74c5ae921ed9c39873e826aaa36ce5afb6d4bf0dcaf0ce4fb2d2538dbeafe0d74a144dc9387c0bdcec1e1d12b9b6d24c96cf25c2402a8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85781453Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600, Name: checkpoints/1600, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1600?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4ea4f878e19bdf63d0fcf41425c6907cc4f944405b221c963981edbda943dd13ae0669fd3f5af42e0686bbae469c2315856d4f7a7179ebae823926b42bb3cef320896143a9d7aea5d07617ab37117894d9973f17be667ce588bfe58ff4c6759a37ee8e4ce962f66cadb2a7b2f9ced978118be2d318c198762f7c997ff2ec8b92bddd3103630f9f5c342e19174d62e0b32d9888dd5a29a6108e0a1a8953a3571be1cc5f6e087eaeaa6c706666598dc0df4a1704fe06dff73f8893420e9f41aba84c6703a297b13d789e8309b129cb292c66326e2d55a71ff6b92d5d01d3684e823f3283a2914d1e1f10984f26db2b12e55a77635e3bc985d70e99834ef4c93795&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85851805Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default, Name: checkpoints/4352/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4352/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2cf9e39da0c6677a0c7cc6552d78f00ab268a9c4d12c7760e39c03391835c15675c0fd185276190af470c94aa089178fecf9a820cf0b236dfa1ebffd4b5e4c4c419b43644cf3ac3ee4779f650a229094ccc8d9da1dc6460dd2792f987cb8b027c40fce48ae9c29bb575c631deed45f467b22170e3e3a09ba4c888bbc4fd87c4f6be6cbfd39d77433471be8314c76a869b13a74e784b268e47f372bd255107a5870eb1fcb288cd844056b29acfb400bce97ef1e5bd9121ed1aa9e7efe4bb4701f99b7309b81cac1f7e029c6e14e0cfdf398848c8d2114b43b849faf04fe3c6ee889efb463b0e65b107fc8176f12aa2de756d0b359774710354757ec1e47454f61&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858818214Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d, Name: checkpoints/2432/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2432/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=6fff39a5159cfdf9d5e140490d29ddc357b0d6e66846e0d429125c563e50988eb4da8b24592b1a76f4f5b54d31d74f3b21cd25715683722fbcd229bee517254101bed154112382756327f3086c17aba9648adf622c5685cf875b0d6939472889d80ee84c085cc8c423344bbc02846960d8435b6eced42f121109555655ffe8d3e56e20fab0b76ef8e61dd328d2e6f37fed8bfa57484b74a948354d78c0d2e14093996947824471dd430f7ff98ccb645f6d6dd08b3eca0ae5ed4a35834b02873dbfd911dc2dbe21ccf0dac39ef6e409f390957724b2518b5f021876be4969739034baf57144807a7a312e26a27440ad9d1045c122198c88999bd5182b4fefef85&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858125944Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default, Name: checkpoints/2368/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2368/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b9180ace1536b7f07166ae4e5009fd9d367080bd3b90d78cbb3ac743c2a04e42814427053aa4b5a7d1aa79c798259063937b499e3b613b7f98c9383b198b0b8705f3ad67189d659647e7dc52ac4481d8b27365721e16ab1abfd45b61fda6b178ade3ec5fe0375778de9760d17c0791d818007df9324c3703397bdfae07177df76cff563f4e57455b95a94764d57a568e6bacc3728ac87f9db912787a77efe7a54264251981cbd0e3045450a9867c5d20a56823c1c0371ce9f0e9a4d692be438eff625cf71e2b305babe039398a43ae4b96a501a30813aab874551f763d31274258f7de483ff78fd0475d4376aa0a38fb99b4de190abed02c7e8cbaacab9e22b3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858847275Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d, Name: checkpoints/3840/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3840/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=3502246bfa81d507fb980d2b26acae2c9bdcdd85124abbbcc38b4142c594286294b6c19bf2344a7af2c1b6b7d35059ebee7480f92f55ae46b4b8898b65520cfd859b02b4190cb0df894a8e16553d923f4672cd31351bef79860670ecd9824a43829302f03f9b1109fb58290e84517adf7678d3bf9a265f83a1cbac29bcbb40dfda7e4afca0fc0f5ab1537265ee1563cf56e9c7e7b7f305416f18122e048977ab7e6c8c42f3366ee3bdb9cd6603ebf989bfbf8d8627b5af61d710fc7e43304c1c68f2297749efb989847fda49144389c04290f5befbbe6024c5b59756bef6ae268bd5a7f2db22ff9d77a67d4840e9a1d0709fc231a42c5049bb77b3166ece9de2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.8577002Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536, Name: checkpoints/1536, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1536?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=40f18745df3df1ca468b1174ccdd2c70e9bf06b3d915884a2b976b485dacf8518caf495c07b6a35d2a9cf09010ac042a1aeee543c8bbe940c3a7d2abedef2c7a26d564fbafda6ee7c50dadba3fd99981f1da1a5423a4141ddeb3e4564921f1fc3c9be4a448fad240152212cac6d04e2f65ef86e3e84d8f6e1c4f9dee261257c4fcfa2520cdcfbb3e3e38c3ea302368179738d52eb9016ffea3421dc06f20d78ef01c158bcecdf101a768c67bddf68af68e91ee8697c7b4c6007ff748f346e83a38b0446264e557265bf626b76910f73aa4bd875965d8cb1d036ea6a795aba08adc1e46ef2ba670d3b884f2e19e9510c1ba6e512373bad8eef971a5ee4aade09c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857725712Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/array_metadatas, Name: checkpoints/3840/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3840/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=87e13a91ec4d10c404efd89e9336ab09ee12a3434a4fe481712d9011903505c526a956ea1a9fed81ca2317878c43fbd984265897b33d31bbc0cff3898ff71fe4afd0fe7388d7552348c8be07138345b46c6c8708a3f41b60bfa93a15cd51cfc8fa36d6f026f5cfcbb483754a876c392364ac9adbbc6a7278aeb361379bf2e4789a0208cb5d2c2236b04391ec31df2812956a22ac280c1604de3964404a571096fe07d296a8ba3d752dfa33901f85fe405e0d72f62e6caaaa5314b9e68cbed7a5ed9f8405e36b7d365b648dbb63def447d30313b9bffd5b839c3c5ae5a5b42283d863d55e7e81d2a41c11612690499747859281514c1aaf0826b65ec1d523087a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858886123Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016, Name: checkpoints/6016, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/6016?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a92d879108394ea93df65fff286ad121778031aa9c6dd5bbfa79d56664fd06b10c1b593afa6390ce152a4109cfcca452f57e1a40f8f683d689157af6301ffe27a3f1fb5a60d2f5e5d03fd5b7755222eb2ee94057cb907da43b8de81e24ab6dae3cb4f894df003555550a2a8a8ed93f6a89920c4802d71274ee676db84c661cb25b3bb608d200a65aa4b2312f1d33118986edc7d375a7585b3fd845b95d2feaf898d0338adf78a90be4d937ff6c0aaa350c9eb249ef353fa02890bd132dd2007990032b8065054502a5aeedd1d86ac4c77b6c471682c95446cff8737824b2a2c9ff3ca21948c9f0c92dfe7f794377866415b20bfcde8a0fd1df25c5d3138e4554&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858050837Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/d, Name: checkpoints/2176/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2176/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b6325cbce29427b0c4563f8d6344b642872d97d32b43d7b10a1a2a34ba9ce598e2873cefa8ca2c92420671c64c54fe92440d31515036e26e5eccdcc1c3b914617a43bf62ac1073b576655ec4ff8a3dae5782cb3d4aa9d4c85855bafaaca1c78f9635bf26432c67ba14b4ae5ddf75abea06e0d85dacccdb55ad96771262e504286f06c10305bd334fcf70efbc7cc1a21335cb4b5c2791cfa895de90168eca891bd9e7c005259a13547fb01e2e06b3cb135ab6b9b1eb9344adb502b6ce1b282763ab3bed73f7dd8a3fbff5649da57d3449133b958b167a3b6e67defac7753195689546942812ba6ff52d91b3fe63998a7b57fa2c14c816b52f96cd028e495ca099&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858928824Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304, Name: checkpoints/2304, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2304?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a9eb1ed7334d4b4c2451467ef30115ea4d172f0b03b975498f97f79673fa2cce53439f0377ddfbc0314fd0ac2c67dd54fe45d596e2c5a656e8007203bca29c9f950ef0978af85699bf946a20038f7ecefcbe7230a885d7f2faf821abe31a6753206425949b4bd18fcb24788674e14517a8f41576bdfd39c944807715b88dcf65ea89769ec8dc43919f2f49c4d4193e353cb26eca1ff72abcdf200fa4a9b6cf135107b9ea2a1391e2e2cc10965dfa4d6a30ea4e9f51e8044a58b8cf44167e3c95adc2321951834e5b24a5089f2adfdc574c1409c07be5ce9423ab7c55d51d74694a55adc774be329ac6e98eb3f279acbd2ecce91e8ad8ac086b46ced26023313d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85756565Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/array_metadatas, Name: checkpoints/3456/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3456/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9d43a6690814e878f7f14a1a026c2444680c7106b255d30d2cb934159294652a2eacbf3f51db38acd89bf3ec6cccd3c5f30266c81f40b11d339dd20f8bef642cca4d5e42dfb368fd3f8a6327a10281d395198c4ce2f89bcd548846a1b4515e117137acba504a478afa436933e26d0f7350a60949c5f3867c037aed9db9df4537c8678d57eaf7360da5d9cec73c06d4f7626c898dcaac7baee2f41f2c12fd3db30c076338520639e57200d185de538de42c2616696d866e562ec97d2e84a4e235f57520b08a5ebd953201a4d96812e51757c7f1c4c85632a6ddfcb11273d627f20185cdf1b29cc9670012aff5f6b6c0bc93a682f6a2aa180832669a42decdbc6f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858973336Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d, Name: checkpoints/1536/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1536/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=1f4011f40b9673938b26ec897e2ec295fadd1cfd1d3dce56b41ef74f76c49683c82987267a1caaf12af2a8c6599c9259cc88bcae81822a886a4c368be69342defa93956b760d108314cb87e9f849dada428e37d1078e586551d5a0034811711d472583cb12a1fd7040e9f5131ca8c3de706d76f2230e2439ae5c5ddad1997cdf1d513b03c6dbef139729f2aef88f7e220e94ea48a23a19ee172601f0e4797531e4ff3ca8c7ef3965335e13267ea500afa87786f4a6dd65a64f73581cb61c5b901bf71515b25e61e1b2276892a6ad4dcc1bb1c74e0c1f429f61379dd0b496b7232f52068cd7d97001763128168a8cda6f4f758652e03345be65275b4d13cd2fd2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858145832Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0, Name: checkpoints/1088/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1088/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=8b89d8fe0e78eb076edd7362eb702cc866f2c634ae7be274f19a3f40c742122d98ff2ef0a310a9f119a5c902dfe4fcbc63da3d0ee6c8e08abf65ec11795affe318dc29d681c5881b22c319ca91dec4191a65d2d5061db1426bcb4702c71dc928904896f57de7a76bd90d75d2d3c103a769918c7e0dc9d82cc80972472d91409b8dce3b72fb4fa58a043008ddb131f9b446749bac1007c4d3c19eaeeb65b1c07464c18094f25b0cd0b7c5453b58878bf5afff6773dd3e419a70a64ed583a9ce0148058ae8b7351d7b7e11f85cc583248467a8a88722c9973180d2277c2f3605e196692c963d84de89c2f989617ec94f3b064bafde6583ca35bb53fe9b66736c10&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857576095Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/array_metadatas, Name: checkpoints/512/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/512/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=1fb0170c986d165cf50ea890047efb11aafbdcb2445464e3ea25f4f55bb7a6a4bcdbee3c92582c0ad65df4ef2d9bec8a7b0d31cfdc0c390fa7c8420110c0374981ad8ad2ec505ae1af88b742dd547af8c6bff9a511d09d8e080bed3d130d87e90aebaf5df6d776bb2e17de984783e67b7c28a721b142f6933fc7bc50fb77aae95ce89a9dfe6428b0260cbf669297e34b063caa3d71d40621b8d25f24cf273505d3cab913367922f6cc473b42dc5442836935da71895b71a10c877c4c263fd0382d45d2c0d6fff1a82b05cfbd0a9a553fbc96ebcb75b7c84cda694c1f9b997fe3019368cc0961e8b487d1c3f0eeee763c5b188b038cf3b5f6f28908328c8f592c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857783865Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0, Name: checkpoints/2816/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2816/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=851a8af505515272425345e5bbe0acadfb84e7fc7b7d437a27e90ba52bc7a968eba93babda0135234a0adebe6ae81b3f695332af5857e19084269243646bda53827ade70c7196d3424c60ccb903131c12a004acc855b2ed9d6410dd34f8623c1cb9b532c3f4f58f577fd6201f436d82b74756e0972a464e4962ec514c467689aa048786953477af83ac80e8a46702bae0baf32ef33c8adbb738fb553812e13ca5fce44de5911a43ec8456fd1d0711480262a10782455e95b7c126df44ae4b8c4419f130e02b27f1bf422bf978f4fba0a8fd829533f883c43e3b711939bca2d170e42542af19b824fa16c4d50ce46271792862d5a2a42e20e25a3eaeda361c7f8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85762074Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d, Name: checkpoints/5760/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5760/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=39596cd7fa474b8f3506e4a185c7ec6b7c14da3a46cc994ff9ce6acf71d2c01b751adb303c6e8111af9161469c3b8e9b5cc7bb4db40dc49cbe5d36a5ed4fba9fb504005f13a1b4c808d61202bbbe1dca4cf7702b58c3b917a2329bc83afa91c03316d0741121bd4b811fd4ef371e37b879771d37dbbe57b5ec3adfb40c470523d5c31ce13ff401f73334cd36f148a12ed48d12fc457d7a0850c7e5cad01ad93334a5c3b4ef5ad8308fb1aade8803bbd18c677b1257cfbf9aed4fd15b73141946e652fb8b7f726669d074bf75a283c4f4604c4de00726ef1ce66236598fd9afbcd207562b8e28717b1276e491997ea3621de6bc360bb77ba2a7831ced24157c5a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857633873Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/array_metadatas, Name: checkpoints/1728/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1728/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a3a0ec9976d9964a8074d63cd89f2a47064bad5210f8c9d9639f74e735bd8de70ba4e8cba398c59466aab1c5434b0b232b0fa15fd5e59a791dd8f2a803a575dccbe674350c68ab65715ceb23901fe9fbf6e35a19bb266b5a527ec3906f57b018cb0a8cb0cd217821401e3b2053444683de6f7d69dfc12c5323fb21ba2b6c687cc252b6b4b125c4f478536e4ea138636117c7fad8fadf3d1eb0b111426b6088b183051a74385b09d90125f1eeb6c4db39507efaac92b15380f32ace537932ea2748ff7f0f5ebca5cbd6054bbf33976288564722351859824e767630e55bed97f1547483d63f7ff8fa9905434155acd257ac68bde15c8369c2fa5500ca403f2380&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.859169385Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d, Name: checkpoints/6016/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/6016/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=75a610174ee7f9b8539d7b6ac6007f26e8e26116f4269bbda7455bc956bf70172ce13b3d6acdb2e000d436a54d641f7c722c174398cded07b3a7f6042bde3e2ece535b7253ecad530b65ddce30df657c9a34b1421c9d54d61358f4bd38e0702178ebb1c937cd899c50b78bb183333ad8b24b74e4c5da1bb53c8450dbf5d0e084ac1f0f760674a2e605d36180163ee43c8686f2edd5e85d683602a30fa2e6bebb9c998093bea14cec6226d589207fd3cad3a7bbde0c1d43f77dbcd8611f93b8710d6877ab522a82b4990eab0f0278262998caf15c31cbb1d137007c86815772ef944ee9e1919e839083df0980c3635069788e57086ee75e93a6e85c23102c9784&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857646466Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/array_metadatas, Name: checkpoints/5632/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5632/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=c0d910bebcbab9b540551a01451a0820c948be64f23514787cf9e88ac67dc837e402ff2a23bece159116b6eba7edd23bc73c1cdf29babf4487d5eee35c90021f8ee772e5f7dacf7651489a22764b129cdbd7203ae12a55679e621f2bbe65b0329b8e1d0e6bae3911b3b592158f4b95722389b521959618ec6400849c18211f25314d069ad63364506bf5bd761b9e57a45b0f29795fbb3415f22e05905f470868b9c3e77e21e28fd7aa552c3a83b3950a79c210f7ffed96101d68540f5112933151f21a1ddd285b0a7ae71b32b8a9629025b89dfb7d94557d05b663cd4619278b8b31baa0ca2422184b5100afd9962c65f95f05996fe6b1f05519b9e6ce55634b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857647924Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/d, Name: checkpoints/5056/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5056/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=7b5c120ce3541b2d56552f2765779c6c20318279160b4191fc774ce31157b7ea4d8e2737dcb076deb2009d2d62ab943fa1f8d82be6f21e4d324c552a1b6e112a261f6fd0fae896ef5d8e9a7d8262680530bf1dfe508a5995daf0400af407638884fd93bbfb4da05f4e96fa2dbc3b3f8e5c9ce3d98bb557d418f254d6ab63bf81a73ee4e835546505a6ed741fae4b853c200152222a2c4e83e106243d340870eae59c4c7458b488da4b537ab70d0bba2ce63b5f87f871014ad5553647f61c55993f11762b55d7aa2e9951585c3de9b5d29d80b69fe9660aea72f9ddefe6ca2415c6332d24e7f10831a96667864e67973b1b58101ef2ae0490079213671dffce74&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857695447Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/array_metadatas, Name: checkpoints/4992/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4992/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=32c115b584c1413b87c05587db37c7b2e2b8f73a7e3f5653bb209bb0be26ed7c64b1ef70458c40c6ea33e76d4e6fccef1d487acdee4d793d6f4ce5d015f206dbbb4fdffcbce1d527a49ffee76d4d9df881085d456f9fae6e64313f68d40359bdbe3860a18bf18d23f486303ced36d91fa24352fa6dac17f9886d9870a53e1d4c96ec9b3840d24acc84ff7568c08776026aaeea757370a38172e377b1888ad7ff0dee054de085f5d1f74aebe959049bce548b250cf05ef09a8b9ef6988ea2f9577cb667c30ed4ae2274b5e43f0ac8dd89f4108b979986e941b80eedc498a7a4d2356d4122ac44520af3225ade421a56cae2df81c2d3ecafda368864920c051369&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857901667Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default, Name: checkpoints/5376/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5376/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=55b17f5c97ec350c552bb00f9cf1cce192f5af39a31a1701ae1ff360692ec94ed1904ccbc1ac18d272033c9375210a979e52ce9381d9030be06498171d5cbaa637b372aba176c4bc39058e621474a25082a189c30035614d1149f314abc306339ac08774ae4e3fa8105891dee2c20444f7f5b54f896b2173c9ad46f5ff5652d1f2605e95ad8b4108569669ae301fa16294b7c682d9c00bc16b63bddd54903c7a04076d127e65e10c3235eaddf320ab26620e7f708eab9f61eee6f8e1504a274b0132a54a2fbf9fb0add9c00ce483340d8201ea7a40e124775824b8da9a70f1901ac481d5dde3038a9ca7eae19f40a5a7b55c709e2762338ceb6280c9ffe7c45e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85797767Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0, Name: checkpoints/2048/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2048/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=6202adaac57b64fa7309fe3bdf3309de4fc6cb5d6a1ea42ef6bbbf01e86264b9b3f7489819a9fc58510715998eada2088e87800335aa300df5641bfc0bd4f154ec65e7d01ce971fe1c2b08498b9b412beeb1c055a50a0d2cc3c97eb52e4eddbdb723e410a36e5a57630008483b3301ee6c279396c30c2656cb929e65e27b20247114ecde56a07018e9a4ef6319b1947b64c70db20b7827866e2d851422504fe1fe5b8dadcc9f80596c9d2236f1abc5e7e65091fc0849f269d7b918b0e1c63b04c465de66d006b105715e6d9901e4b4c0a6aa3e2c90bf3c6b94013d6bb2060cf6ab84b8a6aeb64f5ed9ffaa2e943e6b47ee6751a8e008ec62293d787d25e25c68&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857753159Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default, Name: checkpoints/3392/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3392/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=6da84ab34584e4f87201af6bc3220fb30c89370da307c4a37d4c30c707ef6393599e1908feb273f316438242875bfeb2e4bf9db067b2f0fb7c1edbc59f10eeac6468273664792da7dc850883dea639a1e6b5df89440815b85513ecff87a8e968d6e902e4dec366ba0b9bc5815bbac5621b50d2d133107332a37e4f0472fa37a8abe9b9535e9b45c836a940646a99da2c4383e098f5cb4b66d718c11e157df3cfb1d6d83dfb0e9d84cd24a1bc5019e447cae3f1f238fce1c9a7a76618eea011ee67be092a091f376fcaa78cc820bdad0a71d1a2ea61c3ed8cf764f621a41a1db8e57aaa6ef1109cb1a670170aaf7efe5bfedb62edb162b244ac704a829fcfde8c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85796884Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/d, Name: checkpoints/5632/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5632/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=c4b294a85614f44df3cb32d748ab37a3196a38a3be2732f7f7b8a2d30029e14641f68f7dfc7c8be09c8893bf4fb032b39334111a15d641c59969016ea9e651fdd7eec729430586f4441037f3cb79fa63675e6862348af6c4a4c5b55a7504c2ecc99dbbe05669f7fda3d35226092d3f2930d5c57359c6dbfaf732c8aa6e20c871c6536d1284d73cbb7396fb027140fe8f49d4b04cd442456a4fcb27dfd581f8a80f834606b4218fef2a5693ffaf4c56b43e70bf3af34085ceaf78677eb102d2b103ca83ad330a99b0ee7993421ff3ae5fd32a2a1ef603989faa27871b17d60fd2ee2ec5fbd789fc1ee5debe10af50b2672c6f1940783a8c3e5a7ae2e06c6062f3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85768456Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0, Name: checkpoints/1280/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1280/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4a27d698babc92de5803822c3547f6ab473f91daaf835967ebdc1bbb0cef2e64cba5966d62e9fb3d5b7f3b9d06cb3ca106b52bec9a6d274959673c11f4764dbd5ddad196dc30b08650b4a6f8e5850688d5b7483c68a8a93c1b2c527bbe670c5d4846b4497c37046d69bedb5dfaa67795326c267fd8fab00e44b9241940e1c15004b1a05dd53dd88227a33e7b7b6d0e922ae0266c871dc843bd604ba6ad5372b80f5368792718d532fcee705ad4d06f4692f2435a329514d1d8e902cf886e9a5ce522fa8ff78b9b02b3046a360eb617b9e24c0e314f48ab64af85ea2bebfbab7ce6bdc44ba0d7fc7adb8a438ddf9560390cdc8d33d8fa29b3893a3e640836aa97&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858974368Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0, Name: checkpoints/640/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/640/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b039a9abf353091cdd697089db964edbac8c88ef63c1fe147a9a4cd1297cce582998d1cfd38bf5da71e72f30d0780581c0465f73ae8cad0ce26a1abe4d315ecae98a786c3b4ed9f009cc6bf3f4600347ddd02dcbf3ba4997627c8ac2f065e96415e24259e26d34e197d4c359029c8e93c65431957d92ba717ca18c4fda64b2d4370c3bc7748a2090bbf4cfbbaa475c88f0f0d76ab16cd639befc3c2650760759d60746cef65613b367b04179d2a07467ea74874df7593f8d41bcb04dfb695fec403a29de08e2f76bb17118aa1b00e631de4fb825eb35bcfba8648c98c9c8dbd523249c6134dd3b131f6d5f229e96370fc63faec57c819047a46a00b56fd318b2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858007644Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/d, Name: checkpoints/4672/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4672/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9c846f68e61f91863f29f8c8bc9f20e94f25072ec77e4d6a10ff9882b3b929f65088c965a64f2a365834f79832431496878978240f353db61b323d104961e9c4c288e6893bcd17b0c2c24d6f7a541624dfb6b96652fd77ca1842359c94bf70dea875c4415f1376ee4d1b4842c0b70c678843611fe03b1cac547246dead5e589e19285c6e54ecb2e49539f39d7ebe1f7c8fa711955709a1fa52f1ad88abb7ae2e35416256ecb9f749fd66e1fb664b4bdeaa13b8c571cc6687f31be38dfb2311b0a7b937ce8e103aec46752a3c59554987b7e95c605d0f9de5ef24e75e5899b65081dfa373b0cb022cbea7ec58c8207322f7db93075809f51de93fed54530f8b0c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857937233Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216, Name: checkpoints/1216, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1216?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=1381326f79403354f452ff8a2297b622e3b7b5b85578d68ebd382615584606e6c83fc20a082a71a3b1225aa37b70c9723c3724414fe171dedb3cc0be297898f526c1cb70e9bdbab67090fed0dfc3b419722613a1a6a46529c51fced4c1e72ff3db88eee95a493265c37b44c1fac71ceb7848cc4a75d9e85737c5f913f2be5b1793809cac56af1c20bbd4867ccaafbb2df09a4fd0b189b5ffd71637dd0257fc09dc7a2dbef0b8696629fe9cbc545f3f24b87a7fa8461bb487418210cfec6fecf42e7fede02da5a78676b2c526e951ce8aef0614299cffa5989e56189b8fb29a1f8c52d647a58556c019f40223b7ceb2d4f52b597a10ab5ab172f963a0df9e0d0a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857878113Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d, Name: checkpoints/64/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/64/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0f199d6a6d1002e75b541bac621f781362d633a7df28105c6c2fbef9ac6442b10bd57f3d34101e64a943a08a6d2bdcf99251d843fb657e8405da1c2450875ce0db282a9223679614db1e288bd4892b660915572eae2c0bd3b65a149d6f3a36304e92d17954d5777c3163726905d9de9124c11eef2d84863ddbaa7b9942752574eab6e8deda01fa236e854ec6650495183e525d1c997b85a4aef82dd7a5e3b44f8643eafe8d1f6e2ed87d11ae208452b35208b313ef8e4b3e7a3e52978ce16044dde63bd944696f9c66fa4ecb00aca67f49b042e460d68371bd528ff7a0b0be3221c3717ec0ea96f510435c63817b6f25372bc75b795838b1b3958c3bf5b810fb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857826714Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default, Name: checkpoints/5248/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5248/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=21b8a6686a1f2430b13d9da8ba8439d8c14ec27938b6d63e25b853596ee8c293712d038139d7bb8f0360e9fcbb5a0843a758a01bd7c9f92f36865f8f65d6f319917bc9bfa514a2213a5c116a7c41a9623dfe20b892b9e19b190533c92d3184fe43ddc96eecb624956c235c60aa4ea65a6f00156f9f1e5e1a1b2b460ea1ae0629d7a54def2a6a69fe29e01d920138926d68c73445b2c7b34f60eb25f0175d25ea00da53e420916250026898fd83969720b20e9729f64c8251ba0ad7763d15c99c346d6ffa8bcf662b0d7b76415c30255d3f13d6d2b4ba755605456ad7db6f12726c8e8cbd9c6d6a8620d595b3c52a9e264d2412fca13797b86d3834ab8e57a2b7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858009906Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d, Name: checkpoints/1792/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1792/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0dd10255b077ef9ee4bf2345488064300050fade460c78c686ab835e69bd710d2f0229c5b6febb0eb930a3361473316944c6611bdb1688e4a99a6b4e35e33a7db85823b311c32f6c3c408e8d71eb2ecb7df33fa8798ed211e350d745e8fe1f2fc869a39798f929b83320234dc673d19fa63ef7cf84a536e44a5c39616921b071907944a9f9049a978b868c3b3e32a34659144348786f13348eb9976152d97be983e75eb3ca8286f31ab4965c9c83972dee325693eb20b2919728acf1256468314eed8f678a6deedfe188ea20287dc14a3446161f36cb729931e808812832e6ddffd2f2c724d9de2ea467bf6f87116c3a01d07dfe3dfbca67cbf034ab4258c142&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858042343Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default, Name: checkpoints/4480/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4480/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=631248d664f6e2f3ce1ae17f87edbfb4b6505998d688f5925a2c5e546ff805773e9bc8218976caebda46304f2d444617e9cb632e6236714a8687de0343c4d261095e504f5fd766523c43c564d59755aa9a77577a7a4b36d522cca197208812df4e6e8620f652fbfc0866a39f74105f751cea2448a0e24bb453ee54572727ce418df542c2981e945253c149ca457fd9df6a9c86bd0b93095d5e8a176ddc7bc955f638385fa748d4f593910fe6d50553d67aa7b4c90914dd40ec22e551f22692a529d877b9d8d17864972bb3a08ecbb501a5c843f6adcfb40332f8d1d58d879c8d47b25e79f13e2522bfe8dc51cd32c9ad4c607b46f3a8e7f42ca88ef86cad9336&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857888395Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0, Name: checkpoints/3584/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3584/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a445689c399c8e6953be55557ba845723a19173d74f20a03c6a45dbcb0db3a4f6f7e4f4e4ccaa48ea8c82a9a5146f5894c830f5397285b80da817e3773ddc4d8bbcad3cb19a6cc6eafd5808015795690cd4e4738bbb2b606c2ca86851f7323cc02504d100ff8ebbf9dd6cac75d566a5157abcd9fcbf721736a5c276c12a45153af9898b3a843606f6e0e623c1e8c2b8c18e585dc6564e832722d91b388ee69d2e0df309b54e83642fbb2edd44bbe9820c07cc6353fe97f72e3564219be413aa6cc053388e453bb392773c2ff49b7f84c2c9f4e818e00071515b054710d7e983dd1a418ed90eb464b367303c326926772b6ea75dbbce8b1c87cb4feabe81d9f3e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858186528Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d, Name: checkpoints/448/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/448/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=5d44653d67f5120ec3e48c7521c7a73e13fe0623453d9b3fb9136209863707bc2a7a590a73eb1ff169529a5ee3ae8c5d112973362b507a45995e630fb61bc923a323269941b6e013665b7daff5e9f99057370a9928348b5a4e8a642c2c16527e0d3927ca09ebdc6f82d37036b38383311c927b7167bdbdc96281aef7e22e903759e293414a3fc4e896cdf3550f05212be54f0e35d557b464ca9c3e3ba436a301182de50811e43ed8554786e133a1687a9e04de84f85fb0b3a7bc3b23c5a5e4fcad56c5c549fbbf63d26ebde353dbb0ce6c2cebfe742f1e544420c7cee1de5f2e8cabbb7b0d4ac453eeddfa7f23fb81f7a479b03834c7eadf514515ee96a1355e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858058382Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default, Name: checkpoints/4224/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4224/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=bb4894fe83eaacbc2505055c81ef8eb6f6e10c149e677a1cf422d3d24ecdd15c9f16f9d20296197184ce0678dfba1acfe8f84fa8fadaf8ebc70e9e8f934364c99811fc75179ef80d1fefffbd6402e48f6150ecbd6f7153c0844cc1e99525ecc7ee5a52053864879bb27f26afde5bb0058507e55e58f045a547dfd3d085eca411ec8cd6ef00f6ac9e61a012b9be1c99c1fe08ca690f0f1ef8649617395e00951c2748400137e1516bf991cb0fa6badc5ed224a3291f3a98a775c0c2b439c9868898c3a596a0f2987188f3993b58d540517c9dd25d04cd78953f061e31544af9a6ddacac112a432d58da5d7506ba7da9d7728005cc414f6f8ecc769f3a61f168c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857913755Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072, Name: checkpoints/3072, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3072?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=5a1af7126359647731beac4f597b24c0ea81b314401e03053d943adbf35ea761fe5c2cd76d400f658d48c98b732181afdb6fb95b7bc646f2471c8df8e1dbcf0bdba16084d328ed79e56aebdddef7f2cf61315e12fcad40dd619ea8218274612f76b68e316d169ca837505c40bcbb881a8e8e49644dacddd8552be19ec82de8b869203876a9898ca8472063c43e5c32f4075aadd52a0b2eb5591a92903375aa7057adeb9e043918dcffd7e00ad612ba5d0dc3fcbca5d5cade6391ae634903dcd145e8d64baf2b62a896bc83a31b744e555431828404547b7bbc964f2192241f8adf68f31fa6b7b074bb7650b68679feefca41b5473f91e8bf0d1945e74daad0ef&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858776032Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344, Name: checkpoints/1344, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1344?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2ac2630f6fb0998ace224376bb86ffc45c5ca96338264f82791e49f78853dfa958b329b00e1ac371adeef066d154c40dd5318506431c563db3062ef731b9919a35061d4e386b15139eb0cdfd0885f30b842c3818f995a75db51b53d83025958306062172b493edab5bb66042a6cf30c7c66dcd3229e5426be5dd06e99130a91d96f74ae5b7992d91cdec62d1ad57354aedd8d75a18f95065c2b5c24f443abf597cbe45ea3fa6e80ce16d03498554af82b9f1cb9d31a42f5d1127c423c131bd087fd1487ea0ae148b1914a84c11e03b22267f56487b12ad64228b73e8ee2458eb69bd72f5de572b5eaf1bdd246bf61cdc3097d5ef0ae1aa60068a5bb7a9596a37&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858778648Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520, Name: checkpoints/3520, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3520?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=bc76839ea74221cb5dc08944673767ca803dd05dbf3b2810617b9c3444e7a9e56d804b93b1396312ce08cbe23ba0dc29211bb3546fffc97ceab46a7f48d51d173fd7362e5a0761e0a8eb465fd37dc94411064dce9087a20b6cb3b17624139841e862d88d8551e90b2c46ec388211aac9e7c71aec3de361ef9699822df38e7e7958cbd164f1a028f20d85e7507a30d6604e72a232dfc8c0db245d476b6c27fbe02911d715a428063c59116d72ef241e9b350553f4e89395c673d58038eac4129ebe713210192acce7e0bc5f44c6ec17e86b58044afa816df27e74840e2682d77093853011d0dcbd74fc9bc688bc1af312f5e865428fdc9455282f93a09fc57725&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858846992Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default, Name: checkpoints/3904/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3904/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b0ed9f2fae18dad8fa1e91ff60ee157e2d401b118afda350443cb15a81bcafeafd67c6eaf539f9dcc038fa3b0ab4edbd196200c1d5cd162df3dbeb583fdcd1151eb1d9c1e652bdc8c51c3dd462588182a43de594253a2f34b99712b18f7f8a2f8ea5786145d797ebd0c9da27718bfaadd7f52c2f9b75f6627f4f90ba32fe242a4769eb435d72d2509fa244891a5ff6bdb100dc7b153458357a09da98b5d3bc00d8342c91db99aa6637336ab80ac42c09a7379da6f5e25cfc6be8f67d9f4e0832be1ce817aaf66ba834550d7b2d02c1ae9c4e2e5e6778fe2b63bc154a34583768ffd671298723aa505d3f157b13de0daf6d88cd11ff36e1784a1747f0bb5070d7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857729694Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472, Name: checkpoints/1472, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1472?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=7b3f031fbcae885093ca4ed8c27d61942972055cf6e24fbf0956451233e239f8a779b11be1092b2b9476d500723a0de62db8e7849fa4b3b69b1a0d5a2354dbf213368a8cc95229a180634f552c3599cea31708e5bd8b0f1800db8c61081231995816239be1c761e07b35583272c6abf1115f2362182fb74058bf2a56759d88567a98041aba75cd03f72e15a1431113f38b0e24fa2b9efb7776bf4b9c36afb2f2905c1daaf3251282c937d7bb74b5da6ae1b6739fc44fecef0951203eb98bb6403ccbd31b9ca5d0918253322bcfe5a84017159de467328a4734902cf9de4f22f8b38031cc6b783bde3e60524992cde0b532e659fc3b88e8360a4f0f56d9b6fe8a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858916838Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d, Name: checkpoints/4288/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4288/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=58106aa2a2b9bcea5d01c9f37aff8860980f8403b9603e0e5c39ff25b2e963960755dbe485964aeca908414c80aba836e71defbf2ddbc02cb85582eb59f66e81b15141deecddbded240bd9ba854019bdeb5a22d4c5ee1341d6128b1e1d9cbcc0d0bc65404995936a2c1bb4a2d699caf5efb5577dce972517e2a5cbf74e1c09ecf20a22de54460bf504873390df61906ab61ab30b6d74ddf0a4e09ab640f3eea6a3c888a89168f72e886fd09cef1cdf94752ef7106ae97a590726e3f81a4c6fc4565e71512e18f105d2c9c8427071951e02504c6b791954c96b03aa9aaf41e9782a1cfa5437171b903c8098bd289f014c46a9333b50bb444cd2cf9c5f09fa1ce0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857765083Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792, Name: checkpoints/1792, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1792?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=af3882113c36cb0337434c8e4a255e9e8095c7528efa5c6300cf29c179d293c078b06be6784df93051220d853730b990c628bdc729d1c7323088866d263fb06a84b412733eae8457c008a9a254860dafd2f773786b6d63302cb4cfbdde6d038ad575a2dc5d5417fa66ac573366858bc7ce4ec8367f26ebf0140cd4783de7f181a173c096783a4e094b79221824333832325bfa0a40d3f177986748bd85018501f11ecc48b0ad74f4e279eccd75c45f10b070d6741600d077764c8a7a9b9fdb7e2ebdc53fb00909274cedb8d68ba74abe8fc25732f096c1a53b1a4f701cdddc76b03b0785c8e1f5818736c3b4dbda257832df21d74b417793b451cc69c986a630&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858988981Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default, Name: checkpoints/4160/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4160/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=6bad09a16cdd27f442028a4a700cc76db3c3af00d85dc731d7a1295b268c0af006b3b5c584f878386dcff5e3613e57176e3211fdd320a0780b466b19d0cc81670e1e972422f0d60ef4f14cd58625983fe3de1abe78468632e2c8cd78ddf47a2f22c509e25e0773784d4b81f3627fa37a7d7ecace4f25b8e1f6358bb810ffd8aad52aa7eb61dad360686340ca8c68f06b86007589439afa1cdbd799dff4c8defeffe064e5520d5cf8dc91adcb2cfad3d01173c2688c970dd4185e94d02990ca1f1623c14579830a7c6040407bb3231f6e8b5135db7428253af26068fd3e1e01f8f17000dba842972eb2c1f00c9d093443e9ea88870e3f6193d4b3d0539e02d382&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857619102Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default, Name: checkpoints/4608/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4608/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2aaee09ef1da7e40d06e030d6420fb580d19fb974fc3d172babf463e550d0613cc1ff96142f776917cb813182033d11ca3860ec17415614ca4a4e67d3678b9932e4598f64810bafd9f02c4d12d16ca97284bd426e13a12c4b39401d6fefb14b294674704b7911006dffffe44bad1e418f5343e00799de8ca934b68538a3fc2de402141d28c616dbc23e251ec708beee0222a52486b4ce5639811975fcd3c4d90e277043d37a88cba5ed7f5f626304a7fe0ad844ac6bc5f4c6351110d23858a1045b7c90db67c3583edc981b1287d86e942ac80f3e05526e0b0907ecef4895a9ecdb51a186f48484150e5239d1b87916a1d3b6b3177f4aa3ac8578a13cc278e43&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857757383Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0, Name: checkpoints/1600/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1600/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=61731030f513d5a59f1c6603c6c07faf05ac3ee35680270af53800c3dad19bc0af9b825aff9d1a5eb796f731a21bd44de2d3957d5057e9e88e4b8db81bba49de6635a019f0f0c5f9018f19755c84cba157d127e16674e6de99dab27ceb012ac415dc0983bfa7e49864f9103922eeea30d8ab1e9a5211b83efa2521f46ade92217e19e0df777f76654c7e14a266aba9623fa6b8b83545d56d2fe0269a8c94bcab4b824c7899e8ed64c4edd4f49c70cfd0ecb28d3bfd4c20a3e74017d643704b46a7a692bfa8c4f97a81d55a344220255f4a75df384748c14b1d2e9360a642005fb837dee377151db7a02563579b98747cdeda4a4dc54262cfe056ddae1876cce4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85766828Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/array_metadatas, Name: checkpoints/64/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/64/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=977cf0f92c1987a9694133cceb5b6ce6dfe9f42e7ead378b4a03ec04380a21d1f2edcf07f52a0c027669795d2514e9ca07a7599b88ea16dc3f00a21e7d2a09eb5a705767445958785d57350bce6b196a6e2727e2bbdd90fb1d055cec91b8786f82a39828a783d125aac2d8f50825fbb8ff384ebe06764a0de736c9d1933ce5d02674c871423848de205c7faf4a14d9e014d9f940e2e5c8dc5971b8276bbe8465599c1b8650fb1ec540bbb380cb134395bb41c1fbac2406e359565de0bcf97aa9e384c60ff2d5cacaa5e46ac6145588821bb56404bd9140d451a8e2c4987fbbdeca8bfe5ba41174a03bca3f825f351fec21a1400c936f5d16d0d2ee910c78eadd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85796845Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0, Name: checkpoints/5312/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5312/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=22a9473a6dcafdc7e7b388fb22fd1cc28753b4b6c795e7f6bce93799fc2a9ba9bdd30a37b0df1aca456602bf881eb7441d16cfccee0ba6340fb206b5e47fc281ccf4f7a940a2b7fe7e91b2455ccf2dda9a4ba226424f072340bc8e33466c896b89deb31beda02572fe31aad90f4cbc970d2ac3d2c5e5e6a2144400c9f3bd7ec8bbc683cd6b4a91982cbbe424325b86e0f17c9fd1e29388ceb614cf807e23032e6a8cb3302db638e08c4f18dfa9fd9cda8b00c682f5fc35d35484d8d663ef053409e650889f97d3d453e0add4ef651b5d77fa4aaba5678e41d66291475e4ade12d5e96373eac7a7c628032328794c6ffe7abd0672e961013c3eea084ecc26f85b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.857947156Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440, Name: checkpoints/5440, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5440?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=8c5fb4d6a2f8265837f3750a18849c13d9295d3c49ef73cc34e21cb7c5e081cc2dde97c83498557b1294238baacd07a78f94192686c38a635cf615cf4bc3e1a8bc26d7a9cc733873dd9e145b0e3612b041254a565ddda10aad8472b571648958e0992786cee60d4bc0de26bcb50f032372911eb8e6c4ce851196d2033e34d684f96fd4465cbcfd31db240a2fa81990ee81fe0f756b863efb017387de8a2424e15f9b3983ad223357fc7891eaad003bc7fa8111af7ebfec29c087551d099a55495a8efe6db083ce5564b2c5cf33f67ac6063f2ac6d94bc6b7589512658927ff28ff3d99b3aa6727600f58bd5f963b2cd4e2b3c9942f0605908fde855c915d4f57&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858752993Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576, Name: checkpoints/576, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/576?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a69adda0b9c55bc124cda4b013c75f682d799207db9e8d1518a4196d912f27dd77eb2c079b6ccbd0e03ce4b2037437f35b869d52ff3b01ce8daa87ed22d175b1043aa049cf59395b00fa6913ed1391ec903f93b04ddb89b84620a3e8091c3c996c3a3889651532bf3b3582d1d0f971f073874ff84f86ad0e3a6126214f35e7d6bae093535f48b2fedbb5a9cbfa0427f1e5d94166bfcc8f8e3fe39b3bd576f4b957bd5523a98949c7c8d9685d795391b05ec4f12435b145fbfb105e0597b1947187e71c796c03c6ed91012fda068936bd2affcff462dbacb5610bd3ad01858c0fac54c98ac8aa54246adb3d77f2d4467939544a5b215575b4bcf0fbe39611e77c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858254799Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default, Name: checkpoints/3840/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3840/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=bf316f537bf1677935b4a5231307eb785185ace72e171b17da011efea0e46a8b2266cb120dd99e316d723365c03a5b9d2663524d5080c3374fe0255011eedd840c6fc0b773ee8e4bc67755453a4771a0aa7603be3ca14e7b28507e0b3c28d1dcaf1389c345fb84a7fcfbd7c1d28e4a6c7660597c5e2e139687a2639f334c68a2d936213a1e2358889373dd6c572b8fa9e4e6d7cd771dc626c1c3e23d62e3a895504c7d28776b5273c0746826787893eb35503e3e5475f295e78f6aae1cb50ce3efa2acea159bc1ed7ac6139f414a1f49effdb80ee1df5ab59c0c4d222f750cff5071b076181fa279e8312e532f8b269403fd5297735fb14a977387316670d61a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858851888Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d, Name: checkpoints/5440/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5440/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=8370d50603d7842b31cdfbb5c74028d5243ea4fe6a7bb56d3ffbdad829dc18a488930f66ade2b51597ecde12b1fa8972ce201b334f01fc2f15183cb44187e90a53a5a2f636959067997049e20832037471fe69e6354a0771a03eecf5e94782150692b1bdaaf1eea64dd223fdef5d69ab963002a1394438190f620bec826f11ef037819d4043d1108536583400b0fe3ad10a05822fe3eae7643615c289a952fd6deeb6eb310d02e201c0040130cdbc0793cab786bee47da5c4cde2c2030c228266b07fbca79e109e2a18342673cfb821785b0ea272e5feffb256e1e4194d4bb83ee66cfeb809aec54a18fb16f3f9bfb3f4ac1614ec84a57a31f94e3cfadb251cd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858916601Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default, Name: checkpoints/4032/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4032/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=bbb87caeab9237195a6be779c42a2f7f00fd5971822c52cebd7abb7427f16d850b9dbe9dd9a556a35f773cfb31e4f34d5fbf4b9ef034caa28aa6e181343d1ca45f0d2734608c289b80d3cdc12bab8613d9a52eea0caa80c34aa060b7c5a8ee5b8a55338b92454edfb6fb94341fd19e0226a3c05eb503a27e851312b4da0646697822079ce8512d4c949f0d75a57717eb813e665b059055a93e90dce8965ad9b77a4a0cdf3a0f59aafb9e1fbaa20368e1d354ed0a08eb0d9edc7e0520132b08a7765b4e17d876459998ea154cda1158efd54a0b9dab1b0965827e1f1efa69d745cde4c8157a4a6c37cb56e81051e22a42af259a9eebf4e6d696b7bdf75694bf93&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85891388Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/d, Name: checkpoints/6080/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/6080/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=76de58a1b73cddffbb6a360e7331a5f7fc502c6a26a7befa08c73ff2b918e7d57ee93e8eb0a87f62d05bddae60db033097673d9a4f18eb1c05801ebb3b75f1a4a4785754b7f60a2f0bb50e78f0f9b1d6b0b5efbeeac478bd9dfba1023c041fe9c404202fb509368c88ba01f3aaf74568afbcace55edf51c0144809090c878338b71bdc7a84db7b81a3bcae56bb0fb7c07c05d87ec2068d826ea395cadfa4e31e67d5b32f9162cff31c93cb5aef7af72855189c7f0c153b723d3534fd606c6ea58570c6f455d59a9aa2abb6ed340f4f90625c4d454a12917171dfe247272e08e1cbf9cd5c2691b8e8f9085810f626cb4e201878dad35485c43d67911e9704db2e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.858257463Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/array_metadatas, Name: checkpoints/2624/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2624/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=8576d2462648656f35d7fa39465cb0e734b279fc985ac3d7c7f5fcff0c01e4f1acaacb4d49a058878e0ef9b69bc83d8b842928befdab632d262cd4fda6cfe7192979aee444b4778e09827b89657751c039976d243c27d9d05148be790ccc0042022d271371f34b7f691d4501d1a7500aa7f8bfd7e3b00fee56f4b24bbb72d31fd5c5624ed2006403495f5fc6038e32430ad1129f2c7fb36809066c71c85f9ccf2e60a4bb5c8b77e964091a7b9be9dfd1ed37e47b5c38718c1686bab84f4bdbc7d910ae99ef220aa039872c26299710334b2a841c8bba7f0ded8a4f5add4e7432ef6d938e4df2182015e64d6aeca135bc3269ea16ce0fbec3efcd82a78c19e62b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.85892882Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0, Name: checkpoints/2240/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2240/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b08e9224291a0e88fd5caa72505af4d10f6fe029b7e54b7b68aba05aad9f38ed32b347bdc7b2adbbe550276c4c7898485e00e416e671681633252a2c7500e3852b6844fb2915022f257e47354e706f5cda193eed0ca6b4ff5429eb86d884891436ce9b63418a8e3f02cacb4247dcf174f9a1785b773d56bb005d1390fe4952450b2c6878914b5868a178e9ba0de7d5335b67037951997eaed1ffaa95da87f1d97e6c9e203dfabff4637cd3f39805c9d1f7e4ad1b22d50b1acf0d6ce5989621b7eb3c191f32a127bef910acedd9dd1151a1e322d1f3c5f69ef0b705077292a1370e7b98c52004905cadc9ca3ad395c0c7124382e296904a8958280957db6c0ba6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.919961339Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800, Name: checkpoints/4800, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4800?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b1bcebbb0f3b6c789d51dbc5a006c9f1877d4881d00291188d29083109bffb27c82daaaaf572bc365be0a7be2640767ca174019d8c0bbdf4af0d735d89143472884cdb2238e7953bd387cf1305c8ce458bfbfa0212ab4fb0c0eebad7ecd402ccb0c42072a35812fb0e90f56329354112ab34984fcb85b455c0cb7da10f19c367c05c677537b055cce26c56a9f8ea3a310b1b26984653b4adcca54f03c4e7040faa5e49461d78d1790dce5aaeac99305a428162893347a9ffae60db4b16e6cfb721b898ab3d7b2f47763755c47059e3ddda60e0414a1fb94a953f651168b1b519308763c20b8b8b1720ee39c73d09d7caa99605bb594f464049e244de63e8c147&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855181783Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624, Name: checkpoints/2624, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2624?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=718a68692f26fa6dc784c8fe6cb7aaf1c03676c7ad862ce750f02e1319f1782d69fe928a5175ce8bc9193831bf71d4964fe675039b77cd4b52f84caf866a9343f9087d26c0e66341fae20179f679484d0c24d568ba2000d8e428d85ef36caf7f584de8e33d148b9bb1b2a5377a579f73e74cca73ccc6505b6c18232eee758d512bfcb20bc8cccba900dae43ddc2764adfcaa788c6622852b8f5b4d5a81f5c304cb5a25044e69651ac0da1a81a3f050ac107e418f2a3c1e2de8b50016bd39eef05c0371d0b035369bdcac317e93215c0e9b7d1f4fb39a513ff8f601119a42ea22d290a003f271c96205ca3089f4565af1bdfe420da67a5164851d6edf40306d5a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.923106704Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0, Name: checkpoints/1984/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1984/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=5cb9d7e35385fbda0fc7b01396cb16e1d7a03554607ce578b67bb75be0ec27079f7bdd3887c461d36dba0d614ad65a8d4037baa922205794d2446c70fff5abf7c0a49abf7142f7760187b23693d4bb5a85d1c526a609688cad9a601beb6e10b0a3c8834188eab9075a13d139b476e40534fe7d331f0ab612fd611d13cc068ce289c22d6789faf9deb2c00b8de48c9e68c56aaf84dfa615d754eef06ca16aa103934ee81fde0eccfb3a491e5257dcc8a93b05519e6f0166eb32b6fe5844a2dec07ee5afc5815cd6fd1c665dbb9e63805f0fc28a22ad58eb0a2892cca7622ef1e4b4e10098027a91f8c03e00e7a1cd19dd2ef4bb7f3f9abf52e4d85fe90e72d62f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.923147171Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496, Name: checkpoints/2496, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2496?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=915e97c02ed60557258216b43f9ab1df51b8d4e06f421d4bac6f08b7b7cc594e4a36fdf15fc0d531b4250ab76df509cdaecf1e7c028839464a784bf961c868d97464b6405758238028bd6736413c5e738790cdb1912013d5f8362974226ddc5826f4849eb6cb9ea63654b756a2f353976db9f1bc7ea7c89595f18a3ed739d1a25a421dab15b7b63d11b49d8fbb5756c4a1ba9decb6dea6bef9d6c4bc31d0086c0394047af685229dbc87f22ec169d78d7779cbd163d2a527c34fa9822fecc23574213ef9709352a1b1f6f95a737be2c280a86372533624aed21c13509b27aa29041d3869b7d1b835f1fd2e773b9573ffaba5a39e2a16e17a3ca3097d383464ed&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.923210366Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/d, Name: checkpoints/2240/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2240/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=84503629a6a7b878a8687527b24180ed33dd11a37911368e2759ac89fcd96ab731eea4005e59a2ccb57a5cbc8323979bf11f0a33e349d0dcddc102fe82c3a703b509cc8e1e6dfcf46971d6b6fcf0cbfa7ecb6140f82a174e3a903c5ac12d963561fd6fe26d1d5b30c9fa2ec8a8835b75dcabebf3a9035b623f1833c0342d00a191622f005d38ce77c3fed93fba7b032a7227dc3251f6435c84960c342784c9df91d8162fe7b84417a42a7e4501c3bc6af304cf916fa5b3f081f1f3f10a81ec5b17a368b6c778cc48b0ded914a4c20b0e65e9f0be30a2179c524e0145d2df137df5f87bde40111e52d9641870276c965d27f9e50f7165747231cff18663ae905e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.923177713Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d, Name: checkpoints/832/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/832/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=bbb0f23373cabcea7f3d271194055b2aba0c0ca31ff348ce01613653c787df47e3e2bf47140b891d99316986079f7200bab0a2682a48b29b18853a178ea341ce9a4f991bf34d66cf200f6449f4f3947ae201a673c37b9932bb9de76bfa416f3c14252691aa5cd0ede9746173a4a67376e4c206660d51c3214a24a1a335be2b36a91cbc48d1d36275f84efcc91efdc26d393559481db3fa0d40e6cb39399d40e50d7554b03e5033dbd7b27807b25a84b7bdc6ffaf8673c5cb8e5aaf5c25c062bd627fe37319cb5ac8bdb2498872ce9973b673e5d2ac01fdfe2d8dff97fd74273155bd0f8a5d1d16ef82fcd3b4ca83085addf4b3f772921747543835610cd9302e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.92331625Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0, Name: checkpoints/5952/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5952/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=02971769ca6cf59f3df6862a1e301e7ebdbfc99961b461bf3d856456e48156aec74442bf9f3eb97ece670f7b0b6a61d0c52e31a58b09176300a822585c65ad5b6811cc9527e94f08e1acb60070ad18d5896988b75547dbfbeecd75439cf363e8f91db745282078dbca2c78ec4bc91ac8aaf1257e77301c1b8cb811cfe0f767e55c9067d454a8df76a92eb44e49234f4d24b107561a79dcdd0f027ac375c30a6d85fe0ff88cea698580f920f79b8cd83f3951bf004ce26b0bd37c4b64e8f1d90e427981d5302f5dd9fd92b013d9b6dacb0fabf5a71c712615a2f1f563ca25c93611398f472bfb1fbbf06e3656052d3779b2850b263a2a3deea5239033cd8b9800&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.923184068Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/array_metadatas, Name: checkpoints/896/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/896/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9f5a547494452f17a6b282f7cc2796b8b35d9f49ab1bf2c545a083512bcb8616794f6ba376573e1bd3b12868b69eeed7b29dc18dcb4abd0f96e4a9cb5f49517d1c6369103880356e63fc164f4eb11a1d2d12ad82029fd882de492199b8485b09480b8e37865dd5525b6936485a353015bad2b7a011ef1ee623d98f73886fb79963baed083972aa916058e0ffb840e7405c1bf600125d6e974baa5e258d47ec4d138b22a0a008f0f7b69de5473b307446ac3bd6a2edc0ab2491a6dcbb548d5df2975421b8d393850bbb6484049abbc40c293b3246edf34b52d2a6f27f6680bc5982f662f1a8f2b77be55029f609fc6c02118784df2450baba0f06b42745852d67&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.923524853Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d, Name: checkpoints/3136/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3136/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=85877353af660238195312fd855630e0feddfbcd8a05c4b9f64a44f0a6e92bc2a2e60468d8a8300dccedcde606f608b4ce6a3ea22db1f8d6108f025551d36fbd94dbcd0555d61b2e731d1d756fe6576472d59b76f465e01b0f461a245518bd12c0c05b2e9b2c4150db0f816c64b7b692f6a25bd14ef4349169d8f639cf994df598623e81f39ad425545077feac71618208c85f85752e510488f910f8e9f49945949743fb1cf195a1de3fdb6b3915418c05266f74a5c03bfa81c4b63f51d2feea1635455c8d7bc495ee24ccd8633dc45d7bb966392a658f5d3e8aa0dc9d7d40e790492939b51ffbf8fa0810e2b085be703cf53be121d8eb254f6673308674dc25&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.923652354Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/array_metadatas, Name: checkpoints/3008/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3008/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=c09cc0115778bacec2a8d7e10279906fcdb78e08526d5716afe0a692054fb03a08a5770a5de0b8b2c38613d17e33bf57552ded2a1db882858840132bf3e8a1b14031841ae3f92b8f05dfaa84a5b4f606b2e97fc9ee60e05822e4ebdc2e2df192edc6dc944781f28e7cea50592f0b3a88c9140cb9ea55ad1b62fe20ecf0421cb945426fe2ee8fa72f05ee2aad019f98fb9542f6b841bb46f002c9a95f45afb218e6989a878666f4c0848cd45b2be736d6b07bad1dbcafb2fd9fb6c2ed9d90fa6f33ea03ecd9376778c7056f721a68ba26fc8711b0f4de396baa94e8300cbc208dd954cb959c779610de8f11d2ba396c7fdedaaf976c95847269f57b0e8cbb2236&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.923767606Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0, Name: checkpoints/5120/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5120/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=1db4adbd1d5064f8097be9be507a5eddd258ccdd5c512ff60dcf452deef98aa9f993a1dd0337065be7e54bfa3d165b6b83fca2dcfc53f3238edbccf8bcd28767022f70e685c8d4a912572c1b628d90deab0a4cd7dab1f7ccf71567d673e858a4de70961a961a10154bab79000816d1534f9f2bdb8ff22e9e8b9ce62976bc8c85dfc8174238ab2c5319b7236bd719afb978e7167a8488ee8c13a9b51c72553e680e301ff10253a730a3c19376eefd9c16133a588b95f15f45eeb857d07c6aa918d547427d526648cb59450ae248ed3c4ab5d1167dae7d69160095f935216b366c38473397f17edfdc1b4864c1f272b5d1816ac2e10a30ba481d10a90725ed897c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.923848284Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0, Name: checkpoints/896/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/896/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=46a2c4b0e5a146196ce869c5a2aa1f4563346d322dcdc915587529ed1fa62ab242b851df3688cf1edb4e65763d9f79123a6b4e569a1479d48a2bebf0d3b8f454df8e0fca2192838d373df6378da27d3435652916e4f9f59f687ecab62936c3d6c4826fc52fc64bfea4c19d3f318ea6f881db0fccde26c601a6f346ade903ebcba0f6160ef4ffbfdbe52d9c2a4f49fa900e05b97b068154ebd34e7838c56b819dbdeff484ff8b87ec2448d312f35cf22f3894c3e87d4fa8be2fc2f112d8090e475089207ee67d1756f173f84e4fb8032427466629722cbbea09e73752bd32d47d51a383d725e960cd42bc4825c165aa5212127a1cb2e0a380de7b8a8e59639a88&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.923882232Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/d, Name: checkpoints/3648/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3648/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=5ca45c4eb1ac38dd9c9441dc313e260df4ebde874d630878ddbf2850a0b17a6ab07c9c82ea1db6a2c0b15d8d65a581ab1ea44df3715de11b4e0a9330cf4ab3c020d09ff45a8d1fda5be078d56bcbd3146b964796be6c4d009ad0a2e4354991a512f1d2ee5b748f2e8be5491787d354d7c8626c28ff578db1942f5d837c91b40ccb055f3c4b2d1b432de13b16639c879df158d2f1f2fabd8d9804328b56c06bf5bf556da308a3707dcf0d3ddaad3f15644d887ee1e5954337414a913ab9499ca7b272aa3792cde8c7024869062a299b944c107f0dfd4bc3f8efc08cc4b53abca43653cfe12348eae9d00b2e7eb06611febe52c58845784af36b8ae5eb66092c66&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.923989504Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0, Name: checkpoints/3712/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3712/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b2c1a2bdec10c0b95d99a1755921317dd39f78e5847f2cee18d0df7edb3cb3846ead25d114e41496861cf77f5e7893d0eb5c78cd803078267d9f6a8bc784466403b8f53a72d564909348590a4c9a150fde4bd9878269f4fb320117278e83a95e4fe2563627d2119a04e93033cf7be37238a93db30e99b0510d234f3ad05aa70f4ee9b062232d02c2f5b4a3e385206655d8e1be2d1ff03e79d0d25c4487221cff98bd3b1c26ec1e97a0a683e724f934d66ab520fb6a037fa7c575d4c4389e51bc48e45606dfbf223ad9a0b5f77045871bdfabc73d6fdc3c8ca097ff020a980a074c9e4d6eef114ff6cb43a416d1f972794ab93543a58cbb97540c6cf31178ac0a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.924005581Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0, Name: checkpoints/5696/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5696/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=1bc8a6b6933b0397129efb95dc102e74bb74cdfc90e831613e8c0c59a7e5fdd25beb60945834155c8bf03339bf0c65861b2b02e7bb4cd4eacaa3af68e9eea399cbbf3fd3241e570b583d6b783ff558d4b7c17f34033c8bfc33b7157b8055e99d653ac8ef10a85da397caf03e17d5cec62510912bfc8ea4f4f0be279609101fbeda9983a9599bb7a325f30cc9706919bf3c5e80c3306c52ceb99e736f56587ee8c6c291390f3f9da6404eb67c686cdba783f18a29bc5b595a04684c0899cd7be5b2dc071708883e2b286dac3ee16e579c04ddc318ee16278c23c8b06b63e9b6b8a4495b924b5ba5797434d199b7c23575ece0120d131281c12aadf84668dba403&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.924101067Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/array_metadatas, Name: checkpoints/4672/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4672/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=466917d25a842c4ae115674a35f35f974574b05945fb3c57bd6e29c331f04c304453b71f58df8753cf3884b8560ba43a2242019e8b9d327540b1b4d3fc8c82edee6ed9f27ba448e39cfaff384ec9adee4dbfbe4605a4507448199c8f22bbb39052d46e4c0e8633af3082cb07443ca78bf213d40a13cb32a57b2edb5a460452d11a5b19c7f101f785c536ed1337cd0a2e160c717133c67190dcc5dd0c74888748f2afb0dd9a120053591f72fd9aaa8a1bf78957b1be621cdca849366bd4f506d8f4b5a4be092740b60e5be467bd650d1ca24026d1197cc801a7511010c40ce7bca6d1c22d3426c307ac72fd7174cc4c40259d6d26aee1a95d5493b43928a1e9c5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.924211099Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0, Name: checkpoints/4416/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4416/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=48d7293379f14fdc7ef3251307006e1e774a97f329a9d897769270468513379eba3485bfe046b10edce53765510b7cea8aeffd426a4af40c30791b4f4c7a7084ac5f8d8b05962b129bab0c3fdb7f12f11c21b9192d612d782a130010f6c01335c92d52fcd028a0e0eb0e2d3e38ca2f00be69381d91d22f7addb0e1d274bfd4efa15d097695a175c60223014ce6ed08a9c326929bb0f56e1d09a2686efe43267c391b353f6905548092e338f48aae468b245606a99825497449b5cd198f18b3baaa2257218ff252f079f9697b1b44cfa9622f88a3fb652beb0d4d6651b78740e25c5c96cd614df3c423c29bb5c54e408fed412327867a93ed42f1d015e3455b20&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.924260766Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d, Name: checkpoints/4800/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4800/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b58fdaaf97700b11949aa746aefede2b019338032c4a16415ddcdfcc7bd61d20b8c5370eda1a06f43ae077e947bcd990fa5cfb5eae8b1300d790f61e3025779d0600ce4a61e06f1f22a1b1bc1dc2e47e5b01d04346d92b7da36e05f76643d5247acbb120c27e80498227984ad95724d4d76c906719712dbf251abc3c8f718bd7728a4cba1562117f983558a497437e4b2a04de226e4f13db0f3a864e9481367ba25f5cb01abeb371f535c5af22f06d1238c0a75fcc3023741625597b3552ad4f88f49ad88ad9eff30ceb5502cfafd226886916a2b94315f9b985968bb675718c3b89477ad5ebd38a6d2531f2106ed696516c0a0f962b7b870e9790ce393e98a9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.924289247Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/d, Name: checkpoints/320/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/320/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0a82dc11074090494bfadcc79908107861039a3934bfe749ada3b4fb2ccd0283b88d58b6352034f9774aa0f367b4ebbdc2f5b5e6fb78e1d73483fded243d288951de784a48dec52fdfae639b829ab5f99c46034c62d7ab63b121e14c083ed0f8da3327b515879fbf4cb0f598bc2cc9cf505ec4d7389af53d9e1e415608c034526bc2655cb0504be487709db729ade02db129c70c02e0324c66252cf44f93849a20d4e7a320c12cd000ca50fa84d0c9130d39adcf2f719c64c3d5b60e5d14833c2af315c895fd6e0dd42f09b4adebcfe9cd3c01165cad366deeabc88443a7d9ae8868dcdadd86158a88710e3be58fe6796766e8798acf389a88d3c56456852d45&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.924336227Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/array_metadatas, Name: checkpoints/3136/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3136/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=1b1f9cba9a5ab64a4af9ae54bbefd5f4cf92a6ea1f38847f16b68a2c781094b73a0f903527e10ee1e279ba02a38b16f669ef11c3f611eed41508156896233946f207a96704c09e867d262b974b0eaa7b1b9d2385e24aa7293bd07b1d4e233f3b15483de8437c0940d939b7d3380570578f0318e52afe4eb5ed2d28b97eeccc5bf4883417014545794f59df8d37c54e956545056251de79b21e4133b012aa8ee7e0c084b43f5d265bf1e3ba6c8c3323f9f5fcec5035344313b7f55d54314f4441da6fb0e6b3893b605673a0615e07ffc3aef86f3aa77ceeb2732b86959ef6e3d63d92f3007ee4d6d1b1b1c1a8045164b4803f15e1bc6bf5e4e562d553ca31913e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.924023786Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d, Name: checkpoints/3584/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3584/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b15a9318139b57f2967d9e08dbb077f829893bb1925cada08087195e7952f477a339a22e0e683cde7679b98bcb8f2a6847d94bfbb2abac2de83796155319e88a227374ba3686f6ec593a800b05775cf6e1ebf85052c6a65964b33b40cbd7750edc4bf03a2297b7471b363a208168dd7b699c824faa7b9bbe90476229b45f0f06fee25c9384c9718bde87c1ab687a7e391d3f2412af3ca7a2431e8bed333ab61ac8ee3aa995df1344ca4ff84d0386f4a1238d62b53d0983994e82344c1e4a4b514b7456c83c0a9cd0e164732a9d50581b460a6223883291f846d12bbfae26719ecc503efccff12f7e8d712200366be302d100c7cdb57585d4e5e2319b484d505a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.924411619Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default, Name: checkpoints/2496/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2496/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=50508f5e117a16872c6bf73f6ec1c5fab2a272e92549ef9d801582ca9e4e902ebc2ede96f480e0ff6d7d55f5eebcdae9c1cfb6abc46e8f3af7b90116caaa785bdffd41395048f32eef0cf2f5db561d4cdd16a31b31062e33c8804dd6c0bd9b9ff506c668132aa749c1ba21c4bc51d92dd7ae7146f9f88063fad595d05913fe3d9e286b7d6268f255785fcf050f27f0d8c89bcba103fb83428210f41656054cc5b10936f56f977bd3475ebedc487217773be9ebeba268a48e27b7eea37bd0ab341b352e720fa9ae7a5055c60dbb05fdc15c10c27cdf1923759c2baee9a06f62647cbbc177285c82de938cfc1a5705cfe55be8e2efecb2696a8809e85a913e1f65&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.924313718Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704, Name: checkpoints/704, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/704?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0f1fccc0e4060ecb83ca98ed22b89650a82eea817684879b976e16a20c82c45bf07866270555bc809f2b0f393a210aadb5096578d80a90925eb2beb0a34776d83bcca45378e913b64401a128f921fabad5ceeea54c7571c68c76fc780d6f8b180f743f4dc5425b2b030dcfe8a851f20db07acdc5236a5855b041b11d9fa8ebced8fbf4166e5822ff15793758ba91177e9d84c9b2cb91759af5bebb11ebcaef90901eba05e38149517fde93352a0009fec6835693e83b40d606c51d85372677705c759d6168ffb2157432dba4dd6feb90e4442dc2ee43eeb2fadf13f252307ad1428ae75e47041dcaeb215ec5b0b71edb72382eeb5187f4e7fe117a4a767e9858&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.924578527Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0, Name: checkpoints/2304/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2304/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2ff92eb4860303f8c6f5429708e1194ff5b7ab1d5f592deb2e0320a27a18ef197456645a1a2ece9f11cbe89acf868448dc80a2588ee7dc9eb49e1734810e59c57098fa5f2df7c25e7133f832f665fed0d01410ee055e7dd5c2152492cb0934db8d0463f3a1a0da1ce88fbcee0ff4fc6461b4c9a7dc03a9cffa1d685810bd71f685e70e6a02408f5a37f1b8b335d007ae96678de0f646f4c7056f5062c6d1a57a19850e588252640a84611e8729c5dcd1a4d618b69060ef63d6d04adb0a5a08a68ac2c1c1f7105623ba3dedc05ac9354f87195065f2d0aced08d67e300811c8dfd311f3dabcb8f8fb2aea631c891f6c3dd376f8f9ff31ed3ca0c5fe0007fc7c53&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.855510725Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d, Name: checkpoints/3520/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3520/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=342d67bdb03372a730e7eeed6d12d695fc51ecf65e7cdda6943cba52a7f47c952a6c47f9090459ad854cbfb713bdf8c5003a2db0e5b683f4ed92348d704fe441fcdc1b6c96579c66fc250fa507b20dad580b249d3d4fe0d298e21cd3c402fea695737f9c7fb15089fc2cc06b091ccf3f24ce5def5958e263e749e11e3f0a2b2208a9293f7659044f8f7329a31991de34b7b2e64f62f2ac75fa8dd73b5571c1f32e62a81fbac184df2ec58db5398313e44f0991c1159bd69d3e825031b53de68ecaabea1733ec107fea265cd58a2fa80812d6242d60b755c048a0acd78a3181074530a741a4834869f30b2967d4e3e29f031af5e798a277e8ed267c20858c71a3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.924753863Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d, Name: checkpoints/1856/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1856/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=20dcbae7c95c359355c77872440caaffb5da64cb2cc572b86f09eba1cd408d2f15a98bb6bd0ff2b76358599f2404ec4d9ed8be63e8155d65ce1128408853b47e5f9b1af893294bd9c41deafd9f5a6ba8cf16ae2783a6e2bce89434909757a7ccfb055cd3366a0f6bb6afd588655280a94052e27cc6f82e94d861c7e118f98e2428d254314db59d5ae0ea462e817559171962eaa36c4f4621f46a4fa30aaa8fbab36d3e69a4ddbfc35542334ba447597de5eaea1eb8c99293e909a58198d9cdf2d942901b331c1f01b4c8fbdc087b8f82209ae80627c6bf7f4d951ffbf6a855326c090ed5f5659d971fa6d8ce0e402b9cc9099f74c2fb66c2e2ab4926e96aac4e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.924925963Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/d, Name: checkpoints/1856/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1856/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=14e9d7d2263328c8779774890844a89f610e2ac4cd85633191d9c8cb694cf2a9cfc6bc452137fa92ee936a977be44f9a0b6eb4ed22097ec63c19c980f07c6178c4aac1af5806263b38dfceae101deb7b168d6c3550cf13f1c1ca344fd81b61aa052d0ada80ee830abddb228164c0d92251a77f299df5617e43e9736ca16c40936077f949e59940f40e78f4987754d86e1d52d7599a88e5d9edb1daffbf3015db65f865239243cb959a79f42aace9492ba397fdf64d412c17014d6b80954ed246125a60e70ddff9eea868f4526c97b908a2aad8eaecf082c17a547326be3ceb39cf85ecda457019bb3274b13ae504067694ced213bf75fd94f8f2734967335cec&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.924947516Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/array_metadatas, Name: checkpoints/5056/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5056/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=1e1cf856da1f1676b81039b9a3aade291411d8b826f71f1fdddba2c5c2bc677c304fb4765bf6ab50b89f4d8a48a70b29e1c5f2851b00a037cd384552d3ba77b4b90688b3899828856a63abfb04ae7a6df95a56ff4025debb11fcdb48af20f5c023aa683839a7e24294ad924f1a85fa1f32cc5472e1cf95884b374b6faac24ad55e36539abaa5b03e55c4982f6ea60e458e0b08ed57587d6c63b215622f92afa20fee177fb8a8aaeb917c32fdee69a2e9f5facadf0f7a5d64e9b070c169592eccc4f8ca9f1d10b2fe501b1e3e76315971a37a924d83d43f41a2d548d56a4caf01c37d6449b666ee705b6e4fc9a2ac8ff3e9b789ffeb4d4ea5c96ff28d6f7b1be4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.924982653Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112, Name: checkpoints/2112, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2112?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0296e58a796c734066ba99168545ed3b1a761f1e152c939600bd77f02dc69ca1a178dda74ac700cc727bdd1103b6e071e9f177af66814ddae7b0d6408c101e325363892f1feb72c93256b6266cdc2c9d655f45afe7bacee5d5b1846eeaff41ff410688eb6d138492e1e1701dd202c343632d9fecfc6c9896a75728b2b83cc6626195a1896e73c1c0a6164128250a182dd5dadba548eed07631732b13d6f44ddd8e9fb8f543221567588ca6001896e719237542c97824e84c3dc47c3ddd5a3ad3dbfe1f55048e83b3e2524a4437798cc1f9b50878f080ce19bb2fde76efea5930e043593d689c6e6a174f3adb355f12f5e79beb7aec53da457787388410a02990&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.925054373Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0, Name: checkpoints/0/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/0/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9e591ef3512d3d9c2fe8f3afc1ea56bbf5b065027fd03d9d72950a6f9b096afbc0788d8ca553665a7c567e3512c07169c742b600103c4cef0c833c5e3018c48731d72385d9e59b18a8d1d471fc2ed7be2dd614d8f0fe2a007ef1c9d040a7a54c1ddd15c6865c68fc6cfc074593c7a1a8bfe4301ed99d1435ed0a3de0d0e5604b51049f447ad32f4510b9adc79f45c64da44e1235a77f31705510db73f0feaf44168c4eb053f75206a2c604527205eb4c69a1b6f512b4c2f32f4a807e31f464ca07a02ecd73c8d77cf8df2dd38279fd95bbda360cc7ea4fdb4851b95a9c291640f8e2cbe7e8d3accdcd55d73810ff912edacfa7670fb04c0faec82f92b1485ef2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.925066495Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/array_metadatas, Name: checkpoints/4800/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4800/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=143c83d2cf582fa9c6917f1aa2b408d4b23799629f399acc4839a2bfe4039507d17864b102eaf756cba14a4eaf6f4918ecf590c43d0b2d9f0703eca5c1a0e70d53938b119adb9bae25bfc63241c8c127eacd10727f1d58c003cc98bfc50dd0b8b89f84f95e518d661f34ed40d38b3f7c038be993c6ba3db19d5bfd7f46b3beebd121bb95a764812c7a11f516fd466ae98952054c233a1476c03310219d9cf31bb4528edf4fd6154f2c9eb54df2493ef01720d1354070ac705ea1a469de82ddae51c9bd1ce15f19566cdfc33ba3284deea677d4e48d71dcd7d9c9de68d38f9cf8e458c885814e98a6fd64f97414903b899ff7511b4eb060038b81b9e2d247a503&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.924979947Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default, Name: checkpoints/1344/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1344/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=01a76030947e733ae1dd12b79db2eb46e11082c2f5dfef1a77c7db72416800c67f4e9820ce801d3aec545de09c8d2c3fa2bde34049526fb266a0bd6bc525cbfe4d5cf0bb24ecc8c90644a341fbba8759efb19a638f9dc6584d925cad6b3a0ecd3a9dae1350d05a697e8e42dae72f0bf55ca195db87ce449ed1178da49c652bad98474e118a50ef73a823a88bd57b124471a739f357c477271b0e370fd13107b963a383c792f6aca7f36655bcb55d084bbc2527196414969f15fb929fe7ec91f71dc6cf9297b89da909c47f685a799848214af255d77d24fe8999beebb8e598512d06272e5a3b75a23cffa8af28eaa14e185da009415e11f4e37f2f61a0482699&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.92532104Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/array_metadatas, Name: checkpoints/4736/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4736/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b55f8f2895e9459c0fc400d5469aafe804230e6c0bc611e70345249f6be0ad6de170126f04faaf61ae8fb04c38f510a316e3a50fd78d6e469bcec690f9e6fc0da9dc4231d2ec5a9038f26f5c24462bb3e08b626ebff67c15394b056cc8c17bdbcec59af9362f176df33bad49e44913d147b1e47c1170c650fd6890d7caface998a8e81670217da865ee5d52c3dcdb4285fc8891821b1554c0d18814572c4af4c9130121ba8b29bfc106811f7d11949abc86e45a3db65edd67be320e3768b681c40f86d790d2cc688946d3a63b3c0421d25fc0785257b74338713f15e00d386bbaf61b5f8dada0829b8c26d5d04519c9f67cb8320269864820ca77c374c635174&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.925403871Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0, Name: checkpoints/1472/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1472/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=8a6c6997effe317933c3fd4c5d58c294b025f5f7caf8805eaef0e4cfa22d49544bbea38c34c6b7e8cafb8db480bfebe1ed244872e8cb219fc23229c5ab8ca723dd0840c1c0bf7a95aa97c7c8a731cd66beaf5a28870744599724a94a133cb4044854b93907c643b98106d774a0506421ea2622bc03f7c363df6baecd91cd623c633e97c86693f2b2a5c41885bde8d997ca6961dce6162d59240d7e31d6b3445961b19d95c38469b4a77af3b5ad8d416af4862101f29ef8c36f7900fedf8356a83ed61a5e03393ec3b7163b51ba0b9bb16c4df9be2e01ba642e7c607670db0e5078a8e0d44fe90df7e3704565351eb1ae3ab18913c9318c77039fc784506f5e49&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.925640662Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/array_metadatas, Name: checkpoints/2688/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2688/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=7bdaf05d492a182cd515cffd463eb6a84c59371a62e46d55e6e5435b850b6c1587db9cca2780a365e53a766996c8ffd7b6fdd737b7350d7bb6c894d617009ded8dbd0aae3ad7e9d12f68e888eeaa037a8300ea60bdbc94d6b2f03d6a1a969da2d535a1de4e4d9d2e4d9d8fd0f5a055b28162a265cf1fabfb396f986aba8b7969a0fccb272785d8c82fb92ffcc3e6ab186145bddbd9fad736c455a3345abcfb679d44e008657332ab1b9e444c36784c87c0bd77e54c0584cb115d3ed2653d47d6479423e65a9ef88a1f60dfc80aca53374ab04316cbfb72058bbbbf761239a7aac9deda9f2f8708ef0f873ae88442ef281c677c513974b4f5db131d5f8923d12b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.925771992Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0, Name: checkpoints/3072/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3072/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=7a9f1a22336c624115999eefa5835a1309153b225979557175a944ff0f4f72642a862b41421a467019ab28346a13c5c2753e448974a0deabbee89ebf42c50cc8b4b91145bafb96bbfbe33a8b22e5c432dffce833f943a29cf448eb36c28517e49639b08e6d64473e6956f3b756e9f37bc2ff932e3bb9d74531e95bcaab90ef995165add1aad5ae7f2b79336b0028fe4a433da4024cb2327b4d3006b4f3b7a0ba55aa7b82c39ddd3ceb501f7cdd348dbce6a4069511beded3d100e6c200720a7c3fb5a5e529f90915445656e4046a1b6a1b0ba6cd61f5770cb7ed23ab5496e8322aac1f623a1b5600edc629da208792cb0b7dd04e0759fbdf83396287facaaacb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.925900962Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/d, Name: checkpoints/3392/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3392/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=371f30ebcc4902721e6b18d987a79b3e33e3585439a47c3d405981a2cba2aa879e8b7633b8e155b3befc6c0b1b35bcd523ce5c6805eaa1c30e16ff889adc54391d326891f5a918d41a92ff077aab6ad47b022df66ab09009551a026c3fe78c27a06907c44d79f8a5b074b80fd6da0dff0983aed1a83ec14dad812b4536362f2d99fe416eea6da7998848fbcaa0ee97d67e51b30cc46c02397752733fe310ae3d68ff349ac8b32f81d792517c115f7d65127faaf4d3b5baae6ab1a44961e0025b069f847a029381ee7e88441caed1a1bf73a683822d5d84f07ffde5252b5a03d452c768575cd72ca8ad2b18a3f5b153284e2a8a6af579b61a42613506e2ef8783&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.925919906Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432, Name: checkpoints/2432, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2432?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4466ef949fdd24b5effc23e9bdf974c2d5e5df75339ad007404d5afc9d0c30b7d06e99f88f8297a2ede854c83524a2e31685219fb9ec957a629ea11516f2bfdf883b448aa929ac3d2057c44a2365f45b69e781ff78072110db8f61a8940db8385fd3e81c9a6981145af42391e7ef98d05f226f4749cbc09b04da5aa2b089df9e4bde2880f3911e18d637a860349f816953701ae5d460cfcdaac608a2e808d198eb88e7b12ea70e2d7bc61e3898a49f59c0091c17c336962f6a7c8cb1e6b28df54cda4085ead57389a9aa6d5c6c56f8624e7353ae1e9400897fbcfec94a1a90e76f8a7755be177173fb0d00397a9028b02e5283377683011275425c4fd825952f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.926046863Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default, Name: checkpoints/640/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/640/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=58bcb97b275d63fd7de7fd59a56a792983e336c0b3fde664a7f192719d96bdadfc8113c3e1a3402bc6c798a7ac71a392ef98eeec5c498169e5263a749f0fab56df85bed197f683c4805724b7983ea0bdfc89badc65186fe047095b46be9b82c1888677cdecc9d2fe739204c436965916d5b4c326ab5659cb4cb038f7e6c08f21fdb74c4d2c87d87902a6decfff1aeb906b4b2183bb74577ee65e4f3df8395c74abbf0b4631cb29352ebcb13a60c6339a08c62069872cf02e38a97059ee19cc5acd5e2016da2c44f513715f99c85caa7701399fbf5986c81321d76d873f1da6793e097c42bcddab3679be8a8a4b83b0e9a7e69b5abd7458001298373c6ead9587&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.926083285Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default, Name: checkpoints/5824/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5824/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2ed823e6034e694dc6aa723795891b2b61af02da3bb6f829afa1f60e4be8e1a8d544cdb893501efa78345c3a7c648417c104a22cc10f954d9ed21be397823f9f5a66055bf45aa71d3c6284ad44078ef913dba402d1893e9ebeeeefcd4db73f5ceeb578f5804ddb99d92a7f2aa54621e50b0b3f1992bfc33e48e460e7396f21c856fb23ab6304de10a86ef85c9bf63bdd0117e774083e44aada53fa6dde8696da49a293cdeba70bf8f7e9c63511ee6cffde2ea40650970333dfccc1206323bbc9256c9d34194b371008d41e6e76ba00a0bb8041e71a184c96736467ace26f0059109f69f792c71e67381912460e57b854da461a006c3e5aa3988ccbb4daa1a693&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.926217009Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d, Name: checkpoints/4864/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4864/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=8d428bdb978ac4c96d69e4d0a16a4b4abd645330252adf47b0499dca2e7e0157ce6c0a258b8b3d710a94b1728e9930c24562553f7a7fdea1afb9e46998746f70e30489785258836f7e97f852b4ad7955824bac3fe24c433b40d74baae28eae21193b06323ce941354dc3a65d584520e28f6e8694dab8969c727c006fcefe9d6866f1f69188a5db664ccf78c6bc684d85ce54a0288a389ba127aa497b862deb6f5dd28a901d81637436308edf8fe0482e96171426d0c02587e727209cb7ab8a0d2d097e201287d441b6ea8951edb0472963d545f523a8c2956476ed53832f544b114dad94fc042650df45925edf02250eb2a16e23b12f876e329c391d7b20c09c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.926261761Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024, Name: checkpoints/1024, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1024?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=23f28df3f83d54314d7144fb2d992cd8371858a9617c3b07eb1f97e9216810ec96615873792787d31fb65dd8a18b31cb342b787075088f5d344ed53e8b363ba0ffe5a0bd5802eac96e3e068261af91386d6255898ef248eb5a6db65fb56a4984938ed24dda69de4fc36751b9d7269bab4ce6593dda668c423ef07cd88ae14a034d7f673134385e583db72344be0a3b0c921c858216354098511dea075c997bd822c8ff9f3269de7e078e4da9100e29abad8ee8be410bc00518c1f24221c540548c610f81487276f55d01230c261ae0c4f2b34328672da4f60d3ac19085e118f5217c7731fc4bbf34a9b1352a64645cc159e257303f1236bb42d45c29bea29a79&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.926136707Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0, Name: checkpoints/2368/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2368/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=3ef639bfc2dce70846cc1558e8de74391edd01ba8b34ebd27959f36545a2abfd8c99464d080b32142bfcbb181e425cfa3c2f2f792bfefe4bc37e7b942899b63803fdac02425720888681919b0e920ab8b84474480f548619049e70b455707ee651aeb554a06142796b45e9016726be7efea3548dd8209fdb0c09827c58ed0d44fcb550f03ca5a8546e09ca58d4bcbf167af4d6ae959c68180fb0f5a87f8c965a6d65618ee41030ee7f7dc225df3f719afee95e7f90c1a9d3404b3d289f565dd98b98f5634100060f78776fcda421bf2fa304a326db5ca35f2c1debd3b3ca651d0449596da82b496edba9316f778a109a3020c44930d7ec92348ae0e41824b1c6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.92657148Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/array_metadatas, Name: checkpoints/5568/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5568/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=918d3ae159053d56c296fac74caa456258362e85bf8a2e0070ac944170aed38f42acca1620e1ae909225be3871bd3accfd8dcdd513a8a04196009ee360f4a8bae68e42ec8ebb8ac793f19b99889f34c1e7888f1620d50fde9e47f3c7a031ceb160e738ebaa7b5ea568596b9ee1db667a53de662fecd3cda9203e9d8f13af9264bc3fc29f079dd9159564eeb3a6e9c2eb380677443c91c75c3943a41ef56cd0ba44a2f80a19ad3c6833e7e4aa8c49a4b95e4511a7208dc2c9abbea2a4c388898e8e4b88de30003864fe832f10c8b2b3170db66b73ced7c7dfcb1f400a50f8c208fd6c3772f2dc740004c1db8bbe4a011b374c8175c8e3317f83e39eb1e54eeb47&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.926620512Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880, Name: checkpoints/2880, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2880?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=55e41fed1608e9746025416060f7d308b6dd99be5a6b52643daba0d18ff2714076ee2d5a78629d11b482b888d8402b6f37de0e32b75840639038f751a313972daacc30c6e2b937d96a55936c4b57db5a32032edcdb17d03d77e005ca68579a02589812dfd69ba20f2ff487eea40614fa2be34836c9819487cdd714ec55bde32565fce22173da399b848ce81c35c56edb2df2072a8af1e9fe3423084d758e70d1d4d6baa748c0cdd8ce94b895c696b615cc6780f0672586f65d720cf6ff3f231ddc3d754a445951c60274e940e6759fb2cba0eda2860e6edcacd5979bdb842e0eb49856c190e325d06ed2e5aede57203a217c908a07c38bb3df3decf82d1d0191&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.926667133Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default, Name: checkpoints/2240/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2240/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=2dfb424288f4da47784619aa9cca60a98ee1316ce9b839eeaa4f000f1c6a81404ae58c03cc3c2b42677a5a7a0125fb061851514efd99231d60c42c60c5a7420b930b36232588cd82e07f5bbd5142eb832ab7774fd6628861e81cb715fbfe96b3d41605063e5d0e1fe5d4e733b82c3685d86fd2a3c65fd8330a328c562de7bc10f8772d7199682f0b9a1b7ce7fe5e62490d75682c181d899d308700667cf90084968919e1b1bf169aad80e0db2612bfcda07a2bfa6c3d81e799c8cdc7745ea807a177a4ba062369013222fe0ce95d55725363ae879c1d0a0a1d91e039e57e05a35c8998dbcec4cb8e1d06cca58efd5d63f9c7d05baede1eade6d340009bfd0d10&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.926773517Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default, Name: checkpoints/192/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/192/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=70b8acad635051308410119714510124e35b8f3c4a5d26f2347f9b49a97539006ba877ee6a386b53f89b107a9bceba016e5279ae5c348df2feb3aa428ba18ac469ee172e6baca09062f693b90d3bbd02c08dc78182994c881de87a8deea075d8500321410b80d97f2d5707618eb6adcc4b073203cb797cd55cee63f7af519b4c30e25e7e4051033479280e0ec23414c73c9be9eae0381ec44eb51ec360393095ad6b595e7edd5fe4ba65d08da304cfe26e0b42a0da69eef3857632d9405ccc18cea14828ebb8e79b7573be7c9edae5aa9008abf171a89894b4b7f4e4056c08c1e533997be813a6f464655e85073bf18329d4e1f2fc8fe000806c7de16113ff5e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.926967435Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/array_metadatas, Name: checkpoints/3648/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3648/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4cceaff9460df27e56ee2a6cd933200b867134db9f89c973f2b166b112bc44d48ca8ca79a38500e728ebf08c0867e364c8f2417ae0432c41d94bdf156d61e0b9bbe8bcb0ae5bd14eac90c7d407f9d2c9d9e5f1017dc1ec852523de21e9b0e71d9caf1a1c43d550eed8994d429c9dcf737454ae4626b5ad08ca1397a905926b15f22547d4c30c2c4f5f87bfa56bd8231e9fd6c61e68c47cbc3ba8cb7d619af0940a3e13420f5da2a08102bae6bd0338c95a30b42c5d8a4e8f347b75b946b705ef4ccea40f91542608d3c00e698d38833d2f5b1c49369f040a08ead1accd3da6c29892a582aa2919a9ea5ca002db95d7b8351415f4d4f9ff86c92aa38fda1c616b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.92699294Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default, Name: checkpoints/0/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/0/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4656c0430ebe1cc0e2b86c0b1ab3d7aa4c87391919e51289340f0e1a05a72ca7266b79f2a657dac81dc3f5768c4698d91b487696b84c2a2ff1a91933ffd97ee9cdaa03fc510fcb2509d602a8e0c72b7ed758582277f5e30bd4471459e24cab4f716da860558cc3029945adf76024293a96dfb781c2d647e860a417ac44f0aa2ef3d856a1fed27d0445d52e801db97a12be9515ba306ac9fa81b4999eb65e341133e3d6ca1bd9858bdc74e70f8c717117f9deff657a0cd84989a8285a29c7b15e933882bff666c274a686255cbec34db74e24c74036945cca8dc6406153dcdaecdef1d106812b210fae99bda31f3b5b2ef9fd72c50185fb8f28fda9b6ca1ed6b2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.927038158Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0, Name: checkpoints/5440/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5440/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=929e9ab2a7f3b512cc1adf28f8a347988d5440dab119245e4c3cf968dba2053b9dbf1f14cbe014bed7dbf9aef6088956a3522827c81013e2704c06f35bfedbd50eef74f01e67e4dd0d9dde4110e8017269d2652ecb0555422ec3cec1acabe7b48c4f21c30e62747957534a4f6563336ede3585ab750595da3918fff6619a92280751af690cdbccfae68302d190c4121374e421e21ca29317dd4e6393323d0cf1a099ad3553eea4e55b4d199cd7d100116cbfa12492c62d2160af382449baadfbf64bcd9b35e92986a9f9a0f9f6283cf2becc93ea5e2301602ebd02c49d2c14e12afb2de4c80979e8a21f8ce1eaf9b3c8df669eaab2ab2ac64e8cd4214790fd26&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.927087961Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/d, Name: checkpoints/128/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/128/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=768481940c589ed46567480868167468083d546d7f44933b0095f7cab8adda22c210d441d9a269b343908b59b3eaafb8ba7bc28ac34d7ff9329a32fd963bd76601ee78db088a8d1ec82265453c3a0cdb6b6184609a949677bd4a8239a574ac516a4a3ecf4e009f6799eb027146029bec8ab04fee46c202a9df5a95d100980d44fd1afc58862378d7abb0354abd64135fe671d0fe7255992c780ce190d1506aea20abfca77fd48f0e837f669d785db1a90cf67f53436634b0adbb5d5710f0f583bd261d6474e6d7fcb9e8d31d0217521f911ec01ec81cb173af835a295668cad557fb7d14bbcac35a25630c344021b12f46f4e7e4438afcdb699b25c6e04883d9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.92709536Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0, Name: checkpoints/832/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/832/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=af039ce40d21821111ecedcadc386fbb5d998ae80593aec707d11e856c8998f821234a03502f9e05485530a21066ae7fbd4fc3ca38da9a1f693007390303aa57ef3cc8f78613eb4fe5bac3b6820bf90124b3b601b9642c3ba37c58ee634363310861747e4ebc8eb1b7d61c3b7da9ec5fca7c936986979e263115db2a9a2aad92e8853150f420cfaabb4f295f83dd71bc91a902e3b02d5c130b049d5a4145c9d8e8039abca47c048adaf92c3ad0d75b8bd4f2eb838ef25d9b4e9c727c002e3992f7f48de3952022d42b5ec83ec99c80f4de22b04aec29aef4f356dac7ccedb85ece1517e13f54fbd28f489c8f889a0516e3e8194f2c566aa849d942909534dec1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.927264465Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0, Name: checkpoints/5760/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5760/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=bc73b0cdc453f2ff02bc10b3ffc58adafb8512b7f6283e3c3c15d6f6e969b74be81f259bd903d6eab604ac5fc8748fc6a91063a3fb54a59b2faa61e512b656459dba83a06c5eee9a815cee9630fd6e18a445de37d4a300bda25ce596b2f296c9cd4c81fb11d91a68679c10c6debf65d5a7b95f0a814417d6ff3ee2081b5cf9a29b37a5d1341aaefad8e2f51ea5e41980776367c4310b995578ed14c153e9b60cd355fd24140c681d40c286ac8df01e0dffd5da28449ccb2616b5b7883ea04a1cfd79bbcab68c5bd695d092198c73e2b531f843c7aaf0a309cba1a00ae1ae404d2c33dcc70c93c38fac23056f54b9c69cb3c116974d9f6405d40282ba5d62de98&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.927008653Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0, Name: checkpoints/4096/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4096/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=8c4e37b123102c6fab96d0c396506987e852dba2ed2f701394b168852504e033eddf8901677ba14f506fff706317c87ac707cd3ea227e8a2f7ebaa6e5b7c92c7ab6c9cc84ed9631e998c10a0c7453782cc47675167824f6ff1a2706aa53e307e5c32c18f1aaf975382381c2d1a785db300eb3dc41c4aff41d7da0e723eae875453f88550d309bc62079d3d6f359175dd50dd65a74556241a2cc8499df64eeda8f1c0e8dd38edd45d26089bb885560773d98be4a0dd6f931a4ecbcc679f12063417694f44b589ff023fa1b1e10f54afb78985a201361c805437a0a1c44bb7cea307b25f0089da58dff4c517bf7b73ce74ee25b44d7190b2b1f29aa6281c50ac30&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.927651651Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d, Name: checkpoints/4160/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4160/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=67407830653660df2d9938d127a195456db6cd49effdc77213871c9110009562b7a117df0da66f295015724c0d6dbd9921661f8815ed39b0aa2b89e12499627b43f2224a22eb98f0bdf59f134964e998bd8c0b01e5090877b6566ed8f1914f4b5b52532440db111a7a076eb2b9d3f375efa22cf4c296cb8f723a2f9b76d45ac9a044a3fc7b844c00797055b9617ed5925576a54e5971a2917c0f5b515326b5d1834fa0a0103e0b79a3be5cad27ee0c9180491a8fcdfc322bd2ab73d58f8c99472e1c89f0fb46d4e9f681471a0f47db068e5b44764045034ba684445898b6b0e47f6d5b09137e3e0d9822b221023f182ca0db75fa8b184fd309f4cda991239aa4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.927787177Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/array_metadatas, Name: checkpoints/1920/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1920/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b6cdcc8d6657a27521ccdc281ba251957cf9c26e55710433d5214537df9af7ebe66c42dbe0ac629b381f8239af71a7c51ee0a908106ff5fbade2c2657096eef132a26727e3c53efa39a596e9facb4b113408eb5d946956cc59539330026aad017d212c66faf0ce44a86ac08fc7a5d4c500a96b41b11528f5b6cf773922d36492b834e6859890cd35796c7e9ede95972c66154dbb4365988752ed8b585716c109ab54878cd7c61a900d44c8f840fd2f5fb19cba24abddcf8e1dbfc4b9d94a79afeda135977efb0b6724f0c2e1ecddf7bdcc21aa6dff146832f717da241485ac87400b3e5bdc002f65d8c6397ce1c06f7019481175663ef57ac984fc4314cb1ba7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.927892225Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/array_metadatas, Name: checkpoints/2880/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2880/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9b2cd48769126fd0dfc266f3c636fae991264ca006086d5f6b71d71ad3438e6b2806f509dd96c50e5011edf725ed9afc1536b1e6f6ba948752010c3631451ff6e2920dbbd6e9e2fbe5afe3fdd1b38c60a4e4a58abdf2b5c8d1bb713a7dd5cc580ab10c84238f5aeff971dee04ead1197977babed6a1a2001699e1b0a5a3738252473fe271fdaedee670fd04041b5f6f151040c8c2acbcf9eb14d67a30988f255be93df1719430b67e1f6243f6cf4aac4b851f108ca40659def921a0f7f3acf646a3b9860f6e2fce451d27344ad3db6179c819c0c8bc670774c6df0c19839938ea9696f87a9bf3d95679adf86fd89edc7f99fa6a28005cfa51c2c312290ffe041&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.927892918Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/d, Name: checkpoints/2560/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2560/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4452ac64d9e9fe1ea438e62ddfd849fc0e668fecf2ff85880d89203e5cc4e9a6c9f58c181106e42858ed0d3a8532f45bc062bf35eb7b7b7043c97e2bfdcae3fc6d64a4e96d17037f9ac66a0187b1fac92efff1c708404eba01ab4fef53170b6f8e319a40e39ecf3407198efd1c1be170a5fb9df95e82e7ae854575e0ac6578c957f32862d40de263becb0dcd0db9d2716b88025df220d996c7765ff19561e1b7251d6ef494324cf27365b78a362e50b5357fb1f3e07d192bb7637829ea991020c49ee4427fe7d218a8dfb7f83770fe9293a83a5365f111da1a2a436f1b466e66bc2052f9589c20bc3b7c77db04b02eec127ed4b99e23c7c29206a06602071341&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.927944872Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120, Name: checkpoints/5120, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5120?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9cd903ec070e64d097cc6369f9381851b0b2d579f5b3b1c7b4eaad36ebfb81bf7db7648c447855e283a91be3aaca3cde6bce96b4cdd72cf139f9b81b014cc61897e08417c16eb745e0de2279b0f0e867bc5604e855171bec0bed86fee24ecec83837ef58fc6b09a7ad478d0e37dabe35fbbc509b31943a4d9d2026bd92ff66509c14233276909f2f6a792d09dc41e2e36189481df3dbadf0729ad4a540033beffb5600a5052f1529f11a7f00724d08c4e5176c627161f651c0df81385408cf7649b813700c86da9630087913b73853864067b14456885302e77051711b757d3cc99761242394a59f67555ca651fc22d58a1c99bdb2e4f7e8777a00253b8fd377&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.927981396Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/array_metadatas, Name: checkpoints/2112/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2112/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9761fee51152a755aafecda2bb4013bd530f6e8cc911439fba4ed1e0133459da6850bc9a5257682b50774b4b05e7aab4e77ed30b7a6085d2752fb4d956798b5759d54c37fe7e56dc8a7f0d8e91dbbcccc7f7305520e9d0b4ef2efe73c74b7c92667ac9d1d925a737b63121987d789ffa6754f1ea6705877e450f9df714cb351d26a7eb9f209ec24b6f7c3748ba9f9c72e654f2a8b61a0407bae04bbbbf7e9ba6770bc60986c428a77dc433ecc64f09e7fc525628bf701cd93d7760a9d879be8c6466c73f1e175efcfcfb599ad8d75aa2b4497a02880583f341b1c04025f188dee647564df0f46c1b41c216cb3fd79a268f709ebdad4c7ec2661fadf8ef14c05d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.928012059Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d, Name: checkpoints/2496/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2496/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=9ae39114445a9165693048d159985f00a3a10c9116f48c69022bcf0cdcc440c8bc06aaf6164b424ce93d954fc7defc9b50a4e0837f48fa84be254b2d6b4e2c6123b5c6a48b0a5357a7d87bee93c3c75c922299ce228b226c081c0950fd1dbaf10bc8855b0f559c3ff28b044a3a2000781acc179e50bd605ba98c7cd73a2d3bc2b9721a5a28606c86df9945578e226376ff9e05413842f2f61677607f4871d292e7c558bef65b1733fbe72896052c30dbc7e177f93e94ef4858a381bbd6aa4e9e1df1886f7015c64f14f3adb0b3df138dce4d87e6ef4cf25337f5de4f3fcb4898b566357f96e11513b8fc1d0b2c98ae264216ca6917714858ee01cf708babdeec&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.92793736Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/array_metadatas, Name: checkpoints/3968/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3968/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=b6abfa9cd4ded543c5c52affc68fff5f9dff49b91637847391ec488eb5ca4eb6366f2b8b4b6a47121ead7759e4d3e987f029ed015c313741d534d04d2587610d115718cfb7e961a18dd72d52e86d2489f61ebf44a632a4e9c9ae0572ce1817118b07dc6f481040b1cdd241856cb191bdfc91b521e33bb3ae305ad0a8728bfd96398e414dd48630c1c46b855064bb2fbb1f3aabc5c854bd935e3a6fdf25ca16abed03a46aae8067ffeb75bdb0b55af077c65d9db81a14e75a82e160d37c990219f049bc2fe47e8528d43a13e7f6ecc2728a9e42dcc1b03c37050e94ce4425a24c05a461ed17530992e1d1b0e5bbfd950b038c39570717c2e844d4ea1942f2913a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.928083943Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/d, Name: checkpoints/1152/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1152/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=353dc343e181f306099191a4c353f467c3a09596545a437e12bd71a420f8f428048886ea193fb8ff58abbcff49a9633fca3bb04c1d4a60538c50a0e434410157ad32d027cd26fb3355c14aef917e4e75194cdc42bb2725bcc4234e6ce01949acd3c2b2805893a0a2acec19bed7061edd1f4ce9d6ec629fb0ad228948dfe8df41a3d73965d06defc9c75d3488dbaaefcdc521d38276252791ff2e9c1a09b05f3c512a5b0ae36f066dca653b7c08b21067765b2f3e376b29033319d2bbaf8998f5bd8b417171f5cd9857e28acacd3aef861f6cce38f4d264f9f9902470804ffcbcc046badbdabe9166389a02b9e6c678298da53f3c7d14ed82a36ac4e2071cdbbc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.928094799Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/d, Name: checkpoints/512/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/512/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=5383227e9a3a27a3cf535793f662952c86050b3ff9304fdb094f2defc3b9d17ffd40fb7474f7e419e8d21de1fba12b04db066a646bf72f3f6344d8613750724895e904291a7bf45b3c0c271598d186bc06d671fd3291904a34b024e301e1f639e67a1e1dee46e3be3fe1e923da57be89c4e28456be42956ec7cf45f389fed597c07e7fd07fb84ea2e2e1bce0d0a53610b56629f9ee4ac474c6bf9701e810762d54b4c7899fa6fb2eb067335b2d3dacc402a881dcfbc664978ea7691cf38f06ab9e098f523f243c152d5bece219f7b18b52fc749296ce6fffe16151ffc6086d4625a4351340d47dccde628403ea8ebfa5f22cead6413535bc4019d45011ac8cb9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.927964687Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/array_metadatas, Name: checkpoints/832/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/832/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a07fad7a1ead62d446fe219e754c9f1c2ce56eabe289cd8263ce35a07cc97cb9aa0d406445a57b2eca61a28a917a21c3f57683eceaaa589672e972174b3787b460a8b4736896a6fef14b1db6b3b255d0e7d036c43eb02f1c46e39d1188dc7a110312df2852459520c629af5abd3047840a5fd6b3b307389407ae8cee377a4538755a4a9f98bc7f6e17589c6738ed35d0c87a3ae82a71911e549ae2fe1c62634164bc0f9f63414b2cbf970cd50d281118c00a8e82e7a7ec9a4301e20a9222eaaefe0f345b310b77930bbf1d397f6b7c742be22770c6efdf58b8594b62d773b3b7650e0041fb6aac8d104440a90d082382f7dcd2d9290214b98bcea40c2b4fb118&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.928161339Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/array_metadatas, Name: checkpoints/2560/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2560/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=bd18729ced5e34898a5cca913dfff5e0e53418c6aee44961a4b24b93c0622918baa8bfbb9fbcd4596fa2e5ccd0a8e85ad005d393eb5adf05fa5fc4553b863346d81a3313b9d653d00a55079e58e2c99171cd3cf30fd8af1a7caa864e0956ffeea597002066d86d0f8a154848838568c9dfd82d930a4f28a18ddd0467f67e9829b9545a24b8724b9e64b569ca14beaeebe994cd3d2926d86c649e62f0af1a64b7cbcd782a31ea1354ed5c0a7b89571f75ef735eb7c6759ca86b1524ccc930252e8976eb9d786fdf3dcc63b63e5614d6b20827eed84919d6bd22e8d73761117642defec9ecfac93c8edaa472fab246180a3aa2fb30268c412f137e0196d24eba7f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.92811422Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/array_metadatas, Name: checkpoints/1408/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1408/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=5967ab01137cf49f869e4394eebe2cace83258fa2fdf1aa030c6161e02e9c2e606826367364d69f4eb34de8afb8d4e6d7e9726cc3623afdd2c163c9d81b05b388148ba990af09fde555af937c494447858e0e6b3f183daeedb822b7d543b134f13c643f647d0a182df2096163b353305ed9aced4c2d29652840b33f129086e0cf8e34aceda70230b5627a073b4537a76305cf080db0ed8c10a41be36d0e8e5fc1733019994dd1c807ea220709450630e52320188231d6536975fffb438a8fc627692a7ef8a168579134bb4e73ce9d7c5d1a0d4c32732d951e62f98bb9257dce906c266f5d177ecee8c5e815fe04242f5d6caa2ad335ff2546ce43b22eae10891&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.928246388Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/array_metadatas, Name: checkpoints/6016/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/6016/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=57b09d53fabb1ff52a50565590cdeb28250b5cc4016bf328935fce66ea8c799e960c4d843b63c3e9639d38a83e5f85de6e774243510aec76ee8c488ce5d0ca81baf6092fb54cee898f153563042c09ba868b91b905067ce90bf30c4d9c3673e8b321ca5c80d32bd2beb48f29bd19fb47ff3f6244d0f52db51d9d6c954797be2237bb9d506bb571a26b00625b090246d7252de754c06c6510935e2f22291b1be5df97277a590e439efc62192199ec069a1cede1a689fa1a7f9f14069d9502c4e7df235ca2f6b9132a2e09073d079cc6d68daf7b411fbbbe3fc0cbf006cf8d5bdbd346049dca685b0327bf03fb7edf223a3825a0fa04fa1be6016b51e88e55d656&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.928282348Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0, Name: checkpoints/1856/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1856/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=35ee47f324fef2a9ae04e068d01b207bb9d1edb4eb5fd3443f5cc5f901343338ab0df9fcd61ad461c3b23ba0fa1f376f94ffdc73c3566c41538073678ee105b7e9b6450446ceb50106c9215a9137921c9152f6c8ebf5cdb471ea960a428fbca72e76ecc5f9cfc82fdeda8ca9114eee23f70e7b81bd081eb0551c6deee10cdf54862b683296b194f8109cca875b3b96124c16ff175380306e5de8963d8a0bd8d2a7519c8d64794dfafb2984d436a6fc210a3075984e5568da580014e35537eb04e1639c38896b68b2d52e510b1e1696c64a615029c879f870b5520b82bae8beb2909573e7d905a388b53014b136ca819cc40737342e86507778ad6f012ebf4ec0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.928311968Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0, Name: checkpoints/4288/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4288/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=4b7dd23f611628b095067bd9bf29bbeab3c73c39fa9bd65449b56bf4aa35a52c8ceadf092b0e4df3d3886a1b9c5c6f4401f8360804d9b33665cf9bfccb2f0e90adc796322bf95e026fe031f9bef2239da3b006641dc18081d66bae47f97c9d09197b29c230e21702e2e2045cded7b553772f9a3414b3bdefd711c2ca0bea1b8f10cf05cd02a62589d5c965d681f82651ac78b5b72b13719118d1fdc20765a407f78739d3be3b416106ed317af985abf20829fc083c0c2a5939b6823b96090803e43ca95624ad72072571c15562ec2b2c69a44d212d1f8ce9880ff94f679f3d035714514f115b0086c37afecffc23ca71870847c416f9a5d07fb652762d1d391c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.928027636Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d, Name: checkpoints/5056/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5056/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=101b8134abcae1af60c8c21697c0e8089395aa88ce239a2fc2fd9cbebdcd66cdc152bbfd3553ca3854ed043f9d462a93ffaa3bffa8c178d850dc12d43c7f5fa3b9236b5745fbf16f42044e65bb96871728f6e0db452ac615ef868c5688895670f6281548f8ec3432bf6f26aa4ec816a79df1c06efc952c6e4c054fd4c54c6e0946cc4e569da7c63cc3f83354a8a8d9fb798ba48b615f58455046c652d104c53543a71b52596e8ae27db102d4652905debab577652002d39d1cb0d89ca8ace93ed4ded57c03715b483a5ae5ae4825e73276375e063e6feab47e6ebb6d80f58b790ce2f7e3e1dd2ba3478c6f63516e5a34bc5646ae43ea095c0c2c58f5b361decd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.928031855Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/d, Name: checkpoints/3584/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3584/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=affd6b91ac0ae14857adc452403e81456e4e222b9021acd93ae59a06fc93d1a7be073638ba3718f4a50132341b0128e1218df1cc33f42150f134a57d8b67bc973fa6ce66a54790f6d86aacfd6f91efb28723d375896d2ae10145c31a1a111e26d45d57c7701d7664443361d5f01d78aa601103e20789137e8faf300de0ea07f9ffc78e371e091d5b69fc5c8198a24bb7300e7c2ee0e2052ba5933316eb5da3811203b7570be67676fb9c44cba47a8a674da01cae6080174b1d1aa7312457790741ca65340aad30339226b5e081c1d9d4b88e26a64ac1dd0ff4aa9fb45e54c16def385a2a8d60c2580ec4b18f0c6e7f840afd6ce6f415a9ec6b871a909aeb59fb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.928052063Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default, Name: checkpoints/1472/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1472/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=c1a583e68c0c4db814ded5da5a90da4dd32cfc583eff49beba904e3990bbb4f2e72c79396a2b44d8560d01369a06b44d6dd9cd7cf15a8f7da03d9bfba5ddf7403bdbcea775302fe8e9f35f1c27a639b8359f04491e9ed9b566ac61fc8ba1cebb3dae1aa3c014a6ec731b7ea06afd0aa15376bc016452afcfe6a2ecc01b53dc5f1a5bf579b2553c9a77d449e4e006d5c803fa10bda74b14fad4a875f34d688851c8d3e8de9cca13cf8ce0ccf59a7a8c9fd039a00fe4016a43770932d048e354587e1af68ca337faecb3462ec752e361c8346acf85341554f5c33f0397b22bd005d602a861b7930e9274caee2512387572d9e9275e152bdbe366f7fac30b6fd0e7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.928300668Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default, Name: checkpoints/768/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/768/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=0c630d117766d616e607e4320c4030cf9b6aadbd4b4f620583fa4bb428a44f0da2654bbb1c6e108a864f23359a7f09dd945ffefad1e8eb9e2e42f585b5cf0888294b2d9ce5f285472e97ff58d79577d5565d96cc142ef48abf1eb01e0b0107fedf9978360b145923a940aed393fbbadbbce46cddeed8c299d2b98fa283008e84e922c1b0ff308cbbd070f35e17e90010c115eaa8506e3f57c23b751de7305647a250a6e09c8148d18fa83e4a01a4968bc961af7d5b7673a592abc24f7f446835fcd800cee640692c6e3a1711604b30a3dc07b450164bcce0ec17217da5657c973999f953c3985fa189745fea18951d57ecf9eddb04c2dd3c11533472dde37d60&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.928107864Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0, Name: checkpoints/4928/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4928/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a7dc731eba373add0ec4eb5f537391084c55b2f7ef9b529c67e8f88b376d91d3de79d7ea8a015c6d9034d06913f6aad6bcad226ef004083db4ef67b2378d963b0bb72ed60e92af1fff19840bebfc046c24e1e3a4e03e4ceca22fdb053e7a8e06791f4cff3f7f7b0be4460895fc34be488f03d68b0f4f78ad09d71d789e4f1664ab110a65fe731e8ae91dadce800396e85f7677392e766c8b2b8eceea9978b93e780191243131bf49e8abc57025368273421f2ec3cc92405d288800e12bd2ef93a01fff9f900052ea8b2d06ed3faaf8d4eb63243b1d24537671c17d358a5d3235bd6645a51a770cb001161220e130dc10bb2facf4fb3b186fb01a7f3fb4981c76&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.92816662Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0, Name: checkpoints/512/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/512/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=6075aa03a0cc0ce580ddfa4ef1ad1c88af559fc33dc9886b0eaf455de0af35977bc04aa4e3eff43d6f17298c2081b6b146645df3777a170b6844646e05173924a75152d8839a95f3088430c3477513d84c82cce81bc25af3be27373d01f7d58f605a499f883faca6ed6d26b2c9527aa2bf1e57669f184281d2d414d3e7f9e8d51974882256807a660cc7c35effa3f7e0c0eb4ef214666eb6609a5a6bd7d9b87d0f285bb6ea7be2c5b1285fbb7a5b8d40ce0443c5cd2b4a2b3f090ff3be84b34898aca093fc4c6bbe51bc298ed725200f99897f69179a57a963fef90f599aa878595a882fda771a6d2bb2e674a136b05b8cc0d6329fe278f3e0d73cdff9b47257&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.927991593Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d, Name: checkpoints/3328/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3328/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=a391b8015f6a93f00e619a7007e938ffe5ae91a6792abaa69678de85c6cf1a24fd2a8c4d2216ce7f5b69b3f4c498b9d1709ed9c287ef526847d593844505de5961d1500ea250bfef801ffdb71b88a1293a982768ad2525851ac7dfdf0677d7ba26e507a5ec39d76263412819a82b65c012eab1dd642528a8416f1843e83f246188edd68c5899559082e2d988ecf31f5fd56df1fc9e055a76bfbb5b59d22b2b82b907a1f210c37de10608e0ed58fb7219a4d2f72da25949f23f9f2f0f20873aed888d25b1f9c067a8ca7fb7d7fc4fbf58160d1d38142e97cdc66e04972a829d4377fb8c31651f443e07849a5218d48e250bf18dd0cd9a0621fc64c5b7adebd331&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:58.927913588Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/array_metadatas, Name: checkpoints/3776/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3776/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180158Z&X-Goog-Expires=86399&X-Goog-Signature=930d9bcfb3a799afedf1b86425fc55549a045bef4730ae5c97ad4e12af946a7ecdb56a0453db35651a2841ee594288ce9f498c2f8defbda2e5b9d45b1375e9d7f97f3bec2c6f3ffcb35a5e8d883ae53c28469143d01e441d8ceadc043d09b1490da2c6772d626e4580f6bbfe2799ca4effd734f3f2e9d787d677788f33121faea9f840dc72b5c426248bfcf28bedfad180fd5c7cc939037731b3a95ed9aed175a29d3d5658217e25996918e19682badaf2c71eba8de6a30e20e9bf27f2c52a99b8c7248e24f12e7a3216f4330137a7a9c298e205862b99e90d6ee917c54a6253df768d6a3d8d587c63ab3f242d4cefeeb729d6fd3e0357f981438050db78a6b1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259470583Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152, Name: checkpoints/1152, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1152?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=4964b2e729b7faa3ff96d4eccba7c170c12758de74685f573bc5812b36fb9586799f3fd37edaf8f08f068403b1369cdc0b98b8e749c88cfd02895401c8706e5b777fb87146747d776695270999bb9212ec534abeb9053f0a621240521bb7b3a55140904e653ce42401066dd1cae7bad8d7afec8c663e184000295abbe6044c257c45d6e95cbeebee701f85d6edf536a48015f835e84f8488d477b7f772874cd46d49e047e442bf0238ce5fe820e7a7ee3fc3735176daf72880c9c0b5859827f8c35b99c3f17c0b1b015b1d6655ba1c0c18f6a0f955ffdccfccdf7a288e187bd8f8a9765b733d8292ec178f949407300124b8c3e50f4e384eec2b25c1923629ef&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259575524Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/d, Name: checkpoints/5056/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5056/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=389919035276c1daa91ffd4e9f3718cf5a75cee0a3e5276795a1911b9dc5dabbd8973a4ecb9f8bf1426ac0f3e5648e3ed36602fa87989bfe7496fbedcf2040c2ade2e9904b8230c91b7485807ea3515510c0e858d3cbd7050bfe54fd4a00dce8de308434c942f918ca699c02fd41aa4078c0e79f59cec13da075dd6838ab1e4185440710116b1da455cfd34cb7acacb4d6c6940dc2d008ec298357069847148e64bf70bc0f3ccaf279cde77a30a6908c0151320e7c6fcde1e4d2e1775b09cf45ad8841ec65893cea3ebc8a74d6f3d8298b2ccc203cf388c20a0a7116370efa56b5fc5cf19d174f07b561cfa0931464c170830bae95fc17a82febb4042063e0ea&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259583733Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/d, Name: checkpoints/2496/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2496/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0f8492d168760745b00d8e258613a5de113f5b2453ac167d2d1a240e93f7d186fa9b12866bff6a2a8704800ce534ffda0571ae1e76a15d8798062c547dc076f89f0e0c4e6ced9cd6578c92cf434a0adf743650e36f8dc95a26c65ee0aaea56e2b12d4c593c4d40a20bbd5ab15d7535eb4976e1fc3db3dad1beda5eb9e94fe304a4d4e5891a6cf0de9c0b607d1bfa4c73645f500b516e0666f4b61b89a9f2848403f97a21666eaeb1994ffd93f57d40fe5d99e146eefbeaa8fb657532aa9fd9d29ca411b7a1f95ce2e72c55b106879f321b47d6446095d13f480ccc36aadf12162c9571e7305c02f2a9156665ab2063292582ac1f40700a382e4db19403efbe32&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259482956Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760, Name: checkpoints/5760, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5760?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6b794860a0270e4898e2abc081ab9b2bf22d7ae9f278241df96b99a452ae3a1d4f0e8b03e5f4e840979e0760447c204616d33b9cfd444a573d9878ef5cae7404a825a7bae78d8bb96dca0786fbadf6dee349c7179ad7e39a2844e5db5bf5de29abe90c831b7327f73d3a8fb73d8313ccdddab1ddceaa2425049ac936a5f0836f9a1845f7bd8d8ef7298cc938caa5d1ade40edc25acc11a6a58b31996102113a1400017c8170d2f5cf34b3ef3281fb6bc922f9f517bf604ab7fe72aa4e9b984a6693984a3232f6e0337241ab1ada7537ca22e15da7a688304caec924496aa5db405165a74b71eaa39aeecde579e6e65d73056017c155f8af0a370b45593eed5db&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259709414Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/array_metadatas, Name: checkpoints/4352/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4352/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=738070bbc4d9986ae7f2d27279a732f79992960e2bd9101d072501d19f3f33368afe2b051397b93aa23dd5528a6051386ffee0d5abc2634cdf3e97b19abbd30a279e302b3d08c053057edafa64005e62e39a6f55082e1d06fc2f9cfe28ad11b86bd828ed43052c430b048098ef9a4e4ad6bfc8a41e39c1759270cbe764d26de80ccdf9ccec526209817a21590aff7e2d6d122c03474aefceef93bc8a8edcb34dbcc7accf9aadc25fe346e739cd928a46d0514d0e47dc6a5cf01212aea811084fe8b050ebb058d06f873195df145152a37b25752a0855818a6cf616c2a9948afc638ca79f3793eb045166b03ff944922b039e4c1b62a01de594fe190cdcf43cf0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.25958737Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0/d, Name: checkpoints/5056/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5056/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=94fc461089fce91931d106afea7184db7894b1d875c2790c373d936cc121bfc047e957ea2f3add03f764ffa55523d7ab0d185f71487f57294660c61d965974e79a693ba3457a320d7c3ead4d0cc241718141aef2619b1465e9927483dc145676ce1a8308e7f06b1e98f83aa2d4ab81d11e856d98669f60272124c0f9469dd122244c5d4a8055a24c61f2a374d53b09cfa0b34cb370d350f5c72b14e80f84c0feb60053e123fb60405d7c05a75c87653c7a7130542c0f40fe0241d0d623581674b56df8074bea202b9fb71cae3ac8f884ae2d4b08ffc9bf3a47960e60acd49d96023ff6265025b93d1005b24297e35d2914f47259a0f77c24366c65bf54dbf854&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259752455Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408, Name: checkpoints/1408, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1408?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=73179cae2805cd6493adcb520c079fad932f16ba0797509e109f994629e48d9296fed146ddb13a41aecd9ad59614c583367e6673abf1537b5fb1b77a819cd0a5ccbf67b049ed5b5dedc6ff34a8716c7ad5e09e957b9586ea14c7f9c775cf439de123a060839162a3c23a17aac1bfae511d2d0cbdf6508b0864fec970f96fe7812ffdcfd0b805a27ce835a19668932168f1adcc68e6970525dae6e6762f4e1850383a2d9a5c20a0da7c6abd170b4329f3bced777bda00c40f8aa14b4137cbc617d586da5c1b92d7ad76a758ca405bbd201daaa61b80d3bd501055a0de16cee9f60746daac7bc87b465146b74a5be58710819daf9ffb4bbf1d202fd5e018403fba&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259807803Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368, Name: checkpoints/2368, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2368?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6017dc9fd864bfe06d62a3a3428dfd9098119275a70d82353c100d3531a7d8aafd65b3a47eb97bc1f4d0f0b2322067b97f404dc7ecfe6e080ec37bb11f32be422362957998957f617f5f31e50b6f0a6befad495f34c4ddd101ae85c282020d8bbaae56e1903200a32b48609774be99c05bf3a22b66eee1e08955c4bd8cbd45020121007338fa741897ab61bf2ac8bc5190c4d6166521f53633da8036299d7045879952759ecbc1f2fbe8649f7c11cc4a2c3534340da21c21a12af9243df2641c8d93ef5b8e72edbff7c7ae9b8dab4e849f7e3d989badf28766e5ef47d585638821d46d13d263912ac1db2e0bedc87dbc7bcfbfd1776b4dc045e1b16ebde44887&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259833429Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/array_metadatas, Name: checkpoints/256/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/256/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=19ef017be602a5ea58e443eece665566385a50eec1e260764b82853f4d393a331518f77288506a86690a1a9fc63795437285ec37b54d2c9418fa5ce293dac4f80ab5f7ecfc4b501e25858253f197d87c05104ba96d26a1903ad695058009c369d1e2363f3550cd7a56115a98443de3b818952a276807e83757ef7262f9c57903f3430eb5efbe83c37145cc0e44ba4dd51526de7e401ebd0aa5b9dbb1fb4b81e7ad6f7f0bb9368f7eb419d9fc40c97e3546394a9cceac36e98c42d8b1855fe8df09a8da47355efc5bd25395f841ee0a96416bc8a3746cccb04e7d7b5209715e9946cd6c544231c204d673fde14194c8256761e65f21aab44463ef6767a308ae52&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259713227Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default, Name: checkpoints/5696/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5696/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a77ac5fd49412d9762371639b1912022e29ee190cab150858888a842ce82c864a250423638fde0efd92591a4e67262afdaaf603bb3fb9ca3cf1f7adacc797587634f4200c348a313d1e74eadb32bbb1ed426f4e7aee3b84a1fef59c4a93c710a5ed48bb9bb23de05aa50a65b837c72cf9e96a602645e3374502689aa974dba52b4a1e6e9ddccdb4f7a4137d2df1221cee74c0aad27c3e2274e3e6035cf4396969f3b164d89eac47304807a1eefdaab5cb25997c114ed921303742f9ecfc6f886fe0d6b0128ac82e52783d689b328b23d90746980bb53adc3ca912697bdac72291ffa4ac47f3e61e4650d6d7c8f35588d29a0b25f6446c2190089b84da2a60193&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259919601Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312, Name: checkpoints/5312, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5312?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7adc28480307c66e48b7cd1a27cd542187545604dae09804fb91ec5bc07bc9c6f4ad70ef9c5da62b58e2c383854faa5d69180fd1a0246c567a4276cbecd915c49a4fdd02ad75f70510e0c809d3bd6312364000808893fcda8bd48a65945d08a3290db69c8c39fe4dd53a09c574250f29a4d453a7fc5c0fcaf99c6176d86a8063dbbca13d3085db4e803e298f0fb07ee5fe367dfd256713da4019b2f2324a1fe039106cd5369e5d197f675f2121e015beff760aff1b77c54b1ebfb445cdc65300c79cf59512afb597e22c854491e84d79b735709d59278222185273816a606e100baafbb3928ca06eb77bccf8638fcadd80a32f5a6a7ed6ae3352b0890444af86&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259935165Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0/d, Name: checkpoints/5376/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5376/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=aea36cc7b5500f343bfcbad3d2d9d548018d372e77ea58efc6979888ac37efe06ccc0264c5938c1e465620ee0293dc52be4a6d6860fb12ace249ce9c5296e635a45ff90bbd89e8794868a6e07fef8d3f190d774226c35928efed40698db9fb84c4e3f23e8934bbfe2375421dcc4da5d7227d8a43c37239e8ca3dfffd39858796dc6043b50e5cf0842ddd5cdb2384bedb8714d25ecee01cc6e6cfcdb15fd27e94d58f456dde418d6fc0c07a5ba90531cae0f04596534c60940758d621ef6e113654e829c7e4fb2673082db07231a7734e3118cdf8f68082787a4ac286d096449559431f3b207e6db38c0acade77ed8b2b49f70747dfabd8f9df1a4768e43cabd9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259946529Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/array_metadatas, Name: checkpoints/3712/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3712/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0975b7bbabf7a97e5170a11098992a9221f97571069a54f723337d16ae2b1c00480474b7a866e77025f008ae3daf06b154bb501962328fac690d849221bb46341ccd4d703160904e1ea6fab9bd5f72be627dfef701f7be3f991a486ad28e248a52baa7aed5396d7b6f7de90f6f3537fca716fa1407d235d743cfef6a22fcde880bf79286eee4015ad09372ef0e99351b1780c75c858fb189619c7892f716cfabbfb71f1781f28d8e4ca0e64022fb1963b8efa2500f177e6d9493a73f92aebc87558d393605657721fa3abcb73e604305e78c44d3ac76cdaf40ffcd35a50a209bd0f17b8f69fb0af77552b56d40191751f7dd8dc57d63d35050a3748f4c0f72a9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259984697Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888, Name: checkpoints/5888, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5888?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a05be4cc3d2b7d6ef7ea6926db7fcbb1065ca2f5ec6a3198e41127cd49138107b344e1dd01cb0a4d04bbe55d16cfb9a0a2feb4319e23236938d8f5d0bddb47740b22ac917e4698308a35b23705e0f5c934aced246cde8ce990b7c880bb985fcd72268f9a4b103a797d42f2a7f8b645a159918a43b0208112f484ce7c88efb84a160a85bd95b81fa26a29a0bcf8af8cfd2e2b03343a7af91781f7533377dd596256a7bfd10116492815494c72a045d35e116ebd7aa3ee436bd2035839832df7ea0e3879539898e461e0cc2354aeda251487692688714bd9b6b7b793842193683522972e5313910186571af17671f89db5d451c918be9facdc9ce46774aeb3c7c7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259987552Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/array_metadatas, Name: checkpoints/5056/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5056/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=79b5e31af999c1ec8c8802eec1dd86041da83fdca56e3eaf4cb9e7322786e900812c4ba5190a5177aa0b1c1a06bcfb2f57015c1d02e81be6ff189226c9c9a032013ed0f78216cacfb61f37bac6526f9daf36a6446b624f3d8a218e6b4e903512e205522e88f9e8155dc5b7ae8d6a2bbd15656d6d193377a7510a3d3b157794dd5f6c55f33ca2ac5aec739e30eb49ff906d4ae9b2c8335fd9201e8f0556bcd0cd2b5595faa15890acdc4a042e9266d1ac7bdb064bd5244c68ae8089890e79ae9ea2da8bb70463362f03fdf4d0d51463cf6b856ecf7e82d197ecc2936584d102b4184006f9fe33eca225caf05f831e748b787409a911eb3eb0ded8d7de34a08f40&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259996194Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/ocdbt.process_0, Name: checkpoints/5376/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5376/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=31d85368d9e41d6a0d7dbf42ebf7cd678d395610004be7d564b7e1158a5a1f847a08564966114c227cb6fe2127b5060f185fb0dacdab0feed34921abcc2b85c2b2d17a94d0f6335cb4b78507084375c77925a47f7baa48955305f9fce31f872eacb6df6f180a159ab9ce144e2ea1f8905e8fa3c62b14cbe580316e152294c61a71069401d8a345029baba854d4687c14a1ce9db7b3ba77180fb9c874ffd82e0445d6ff658252f413ba624593a9e179c37927977fa5729cb175268a036f98900fa98ebb0a3a0c991e0289afd5d9120b1a6c2e4dac1175ea7d0397f1f2d72de95b4bd585a804b81870a1dc9b9050e11abeb0f030a6d8a3228ffde76986a3eb726d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260047536Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0, Name: checkpoints/1920/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1920/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=21ff9aac79a9b1b77e26c3a1a86835c289e7a97bbe54288c77890228c5ef77ea9436e9266bc22024d5e571d320722c1b4a755233504776127030ad5ebf128c1b97929bd6f1843db72fba9989657e0e70378dee5029a335fd9aa02a99050481432feaa0fc0177b2173876546ab615efcd709e92efa8e719a5b51c367be32eb4201836217160eb6a08b6328b296c60e19acd7b662d61076d10c34d341b473f03486090c640c91d609bf738c4c1f5a80ae421261b0af4eb81b16a5c10745c63c0f87d8bf03932a6e2ae390b321d94865974073065ce70862ff397d53530858c10424348bdc41d71add38c6b412557403dbc19a985dc0651c8730d6c8c900a42292b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260057204Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264, Name: checkpoints/3264, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3264?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=37903314b37ff81a593e8e950907928eb3adf81ab773f8325b481d7cae54d4cd0be1f161cf07e606897c29097b9a7765d5d81329f965a60a509c3e1f5e011411874fc2e2fdb9ce7c1e9fb3cda45067ac14dd18875064e1a69a48a6495532e050d96a830d75e2721534d53f33963674265ade68f6df0fb53131a5ebd8ac88f6d07a400abadc0e4ef19fa71d028eb37a57617e602c945c279785e659e2f5ffc7112532ea3d6ad64a240df3cb5498f99f7fd4b0c12f19a00e76eb7c4ce4af3110d95b3357dc8f703400def5a9a6e46e3c8caa80ed15778cd2919c1012442cbc149c65a91948967b0ce7408135799e5f3e7956aa313b27ed25137133e157fdb51ced&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260088266Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0, Name: checkpoints/960/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/960/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=89b47a98c1ffeb13b4f6c649ba54cf915505777722101bea3e678badb479a5bd2bad4f58495cd860ecd315f59cbbc1e53f988b61645795d7e741423c1c5148c6e0a1c35f97c8e7b5349026c949866a068faaecfa8e96b2b2581a6541a7de9a5bdab9b30c299ce5853336cbdd5d05b24900a0508fb386d9a5617bff94d21317b70a75e54fffd96fffcee36b74e10c6d567f2fd59310c25a32cabc84944ef0173ca7212f05f49bde7ce14b2ade14224941a14d025bd1d7ae82e1405a643d91b925f4afd38a9af76ae7e662a97f59ea5f4178b86fd8ce22f1f15e2bee756576a43e0335cdee2a1867ff08ff29d4e0d42ab35032c9ea8270435829bb60ec91062d45&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260112624Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/d, Name: checkpoints/3776/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3776/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=ae486e7e13a261ec3c9658f616799a1330184b1dc7b854d4ecd141b0cf743bdbd0962d89f4d7b979773c94e11814a82a713653da69a0d2cc5710eee3fee6a1df60531affca0039ee04477fc871901a2f3e16f2e74c272e9779e0f3b7876de4ee82b19434ed370448c32dac0fc93ddb247ae455358e079876aec619129a633b5f104bda273b81162cd8f5da06a014a2dc155ff530987d4a67c1b6472dabb997182e854fc251248dd3c9ba59b74037453c514d179862a486c7fe31157dc7b5b9a76b52c4550a854ce335f5df4b54c1d480e0620a5cebebc82895cf7418b669851303383f0be1418709aa88f7a06267ef7c0ad490e27cdf0d95d0590ab1e9425093&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260146036Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/d, Name: checkpoints/5696/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5696/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=2b4c594501367192e6d3c13106862b0a072935e56571a392d8d64fc0faeb43cf9a5b4719dccef1ca7c5ed8ab6fd11838ac267e7d2233ac2518999941e72bdfff4879060000ce1d46e5442db21463967e5d0a4fa565bafa343fb31eb99e6e81169fab99f6e3b47111bddb4b668736644825d40d0e40f532558ed507c79a5748b53391bc87e13f8893fb71a5ebe531d5d9a5a83e3e310e3af5c0545d2b85f70eb5612bb3f3ff1cef2f815cc742c92480987def464084c8a94c985e818d1de825afaf99526a229e94894ca40da13df140596007c78c6a3fe0f46a801febf45c1a9ac4818b9693af53249328e7b8b29493ac09c7dca21e647c7a81b91ca2899b680b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260162006Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752, Name: checkpoints/2752, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2752?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=865a792c464310164aa8775909a5e9c24e98b8e574b57bccdbbf9f8a70e1c8d2dc6c73d604516f19694a26af5ffbe739d6511cc8efe1bc0dc1d75803927dabd3bf2c3bc3ac8e86f99d78021882d23a426280b4b956909bbb54ba1f3fbe0fa7114bf65e8ebf6b3f3c33d2ed5b694172a9905963a829a6dc508ee32a45cf4900f1f3cc6c195f03a51755d636be3b8477e3b73d72ec89577196428a686b78e8da6940aab7fc81e5a990d92cf300e975dd49d2de2b6f6a03f51385c4fd71028f58c9685993a1598366a16b86c629ecb4a5a74ac982d7722bbde0390a70640878612ac57099dc726d660c70b85fb859f9fc85b6684d48523f8949b5e9921d700454b1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260052832Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default, Name: checkpoints/1472/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1472/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=4b34c7bb7ea2f51ed85459616b9d247b76c2424e4b97af2ff7477d87c1fbf2600e37edab291d31c75c12555521a681de3260d82f2b7dab4c44dba62a05a09d15cf5c9285b3390b5a13555e4623282bbc5f7833288422fd333ce3cccc0c8640b87cab7a2ded91f90e2b30ff07f49c4585e5fb2e668f6ee4e5260b885ff09aa0bedff1477c8a87a443395974133d569ce92352dc4a35d30c92d886e1f0001880df6a791b500dff3b126c19298c28eabdf896c00832bbf1c6cb65514727eb9e86e8264173e40b794669d3ddbbd5a1572cd5d083d27841b786e355e5ab44cb3a56268bbb7b2167a4e4e9c8594421ad608196bee93fbd3cb0f1611bdd56baad299eae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259848457Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0/d, Name: checkpoints/2176/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2176/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=1f2eeb6e55984b7dce09c149c6337aba3f6cbab4207e742a598090d4166b7138f15638e5015536b23242d0d638ada54126f8296b880002ef01c60501373329bbad5c80295b2eddf08026423677547ae673f21908040b7a0d9459d9c36480387c2d296cb3426f9aac963fe93d4d49ad392ed9ccb11e821c0f007c5d076fd14840a418d73989ac31248e3d224312bd63824490e9d07382aa45cde350b03b33e921105fce7f92300266d9862296527e3ae1b88085e989275d8d2213f37649b21bc61a0c3fcab82f3d351337a756de5fbe04aac5337f67a05c94107abdfdeed083f7e2ff603c90cacfe92b4fe473f79aa36a8b877bb53754435757ef3aeb76916c54&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260238924Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default, Name: checkpoints/1216/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1216/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=88a38345650ffe4b0838169f588bfe916ad58cd572ea36d16940e2b857f8744b6784c4e5028ef4f30a77494f018ba09dd5faf317b14cd7a98cd2bd213ce72219e6b9b910b6a12e3a4a74429ed9dad29819a4cfe63e061fdcce27e7118269c8448f41edae9ef69a62e1f5d46316d74953366a62f8b47371614f2f0a3afd5fca11a5f68dd2865b8370b564c48c387151a2a66747c343fea5da852ca870aaf32211c67c7e31b8cfe38b3aa95e7e92e8e1779966fcf33ab8cf467d05854dcae1c400f073cd412431137759dabeb97f07d6e44160036c0e0d715a80d08597a2e7e971f61f7b6f1a857746fefe8c975e2c657751b86b7003c1b951004bbb9340cb5705&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260241773Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472, Name: checkpoints/1472, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1472?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=4578fba5b8f1e8a4c039c7ad7ff0f040a1e3fc8e1c140fc503e5de660cb2df9159a0a9ef857758e7b943742f603e62636729e81b23c446c74f66d406c582d764a84a185fac4c23d44ce2a7664b96ca8fdc026b0140abc38489e69aff55f1e874a16cd9168296ff2de875a7749df340ac42d71871aa9ad8319dadded6c29b560cedadda6ade1e278a1ed40fa61c0427cb8d056025d5fa2ebd727901d7198d0d9d50f10e3546af7af19f743ef4f769f64ffc641f1fbaa294a74e569b8d770b8ab873769ec3f4f9f491a6557b21dc1fc1a3ccf518aaa2de115244ce974d4f88346442fcaae4d0590138009a7d034733ad33139716ed2dd56fd804cdc83b9d5717dd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26026284Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224, Name: checkpoints/4224, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4224?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=69ef9d15ea3ca0bb5f7c01da5da8d4d4c51083b1cfd318dd3608e290875a6c4dbdcd57ec719ad4e749139bd1d8001312fdf542feef405673a73bb07a8f356095d018ac57744e355f496478d8feab1925509b417ca5d52b955061194104f8cb9e39a802ba0e23e7e15dda3fd3aede2f99a49df49b9655edbd6faa14e64ee3b76db5ec067da3876f1aa5350ba159b6c2db20a4d229e7416a6279d999206e8b40bb3ed9b08ab1b5fd3017395702dbdb8505e214b21e1bb2efc2456518ba6e86fd89a01b7ac9b86dc437093a897adca0fca8f73192307d8cde04fe9fd710814b49f187081e4fc94c7492c02acd2c617fd8ec458206b841a1184fc8038a3fe0f3471f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260290688Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/array_metadatas, Name: checkpoints/4288/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4288/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=85ba929f8b730507d9e1ba0f28e6d47bc6c28c3c32e7febcdf62b84da5b3bc108b757cc26b0c687c35091a8290892a9a3533f15f48a764cddf3c440f3ef651d16a2f9756bd3223b18226570b4a0d925ee5b2756faa26de27ad8d01ae892d6db6a078ee091f18acaa83b1719f83120db43f5e9c6276503424056eaa6e0210f6f020245147670c28939334aba88751b625171b0a915ec13ec8ea526597d5ee6363958b4325fb3fa81812fb5c2fb331cb6584aa92243ee6f6ff73d7387088fe40fedc8552a094f85ce4584858d5ca07e1c14c2105516d3d156ea0d72fee1c58defc9d4d30be0fc78f9ad39df7d793322d651d3617db2e6f1bc1b8738cbb9c8bad0b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260313373Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default, Name: checkpoints/2432/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2432/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=9a6069d9f6835421f5aa5d8a6994189213da59af9a57ed9e46c39ada9cb7f9ddf5410357d72174d6382f5d49812a3e00999369f65b6e536aa574512f864e658273099c41935b4ad5d66576c5f96bfc834e4c43d23eaa81a764302d5063b94ad49da323600882fdb1c49e855141c6894cc71f60ac5f1a2383317b0e5169607b934147826197494829332b85ed79a3bfc2831a10c94219cf8aa5641c887ef17ded7ffaf17ed1c5e2eaea2924d1e0f896f5c28e110e8af69369916a940d75451841342a14d62dbdf7b7820cbe05200b6573f5f064263378843a95bad59686800666bb9ef2f654b92a3c35ab7225827c2d3075b86a16878e7531cd1f2dff325361f4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260320753Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544, Name: checkpoints/4544, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4544?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=56668791424f4fa650a5f316538b6acba25f44ddc21f6d14c721be37eb3db191d96dab5b864d202ce2cb4adeacdea0adff3be5077f5f1e99f21c98c8055dcf75374f112e1d745267a9b69887341c6b807158c7dd9d49f17d6ff466cce8d8ff1899978e42df071e71990fd104d80b2d5b4f659687001156e962daf3bb466f4240804b2357ad5be8b78f2b5ea79005907df25495b6b10d615f42f240d888830938babc09b5da14a5607053f3fa7210c2041cbe3a22c39013857b4d6b21f579066c5b212ac3b428a7c7f4c55d35c894a54e8c419c6c8fb27fba0c69736409ea2ac1b657e88483e6f933a9ab59cbd5232d621b15c6ceaf6a23776da48e9a1e7ebcd2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260316907Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640, Name: checkpoints/640, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/640?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=04fd6a3a56680602e87759670990894ae47400993ec50e921e3358cf93ecb05591a2d9ff7be43f53748bf5b1a58d211539c28141ff8bbc107b6150a9319e012fc2d7a702c14fb784376fc7c11348513dda75e9d122d56793ed8f2511824f064aedf8409e3ab1450c87d68850faf9a282e633bafdb7ebfead34bfa0cc78d70543400a20169b2847e272c5d56dbeebb552e847479c385444c41453fdadba2de0d283b84782cf4505d289ff829f0650edf45d7ca4a3be3ecfc546717510a0d594f8882fbdf0aeec3b922dbfd229caea4b42e8a0f6558e2b08d7ff6a4a9a8049cce4a0d0c2d8685c2b39fd2f7573b8210bac0c9db6e1312693b6af5595a8881c1b71&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260386271Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default, Name: checkpoints/1600/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1600/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=3bb388e33ae75045ef6054f0db322e66a514b904b99edcf59f795f2fddabbcc31ffaf3e88c9eee2fb411f53aebce1ecf29b25526d8bc75db38e7fb72d6b464dc1651543193978987e67af6c9266166f7338d51d66e9b481bea0a93a363e96c98cc6b02e66ea0ecd1984c9cab0be6bf39f292b1c1cbc6506221ce615900ae691b03bc158e20f0bd8c0eb973aabe6b87060bfb527f483907cfc797b7c1d8f2b85d03e01123df8a547fc356543a259f223a204eeef8ad8987a1b6e82f85a9d4ca8bb64e3bdd48dc881a63b74c58e67388287356cebd0effbb6d5433e4d780a55480a4f8a39bf582cbdb8a81226ddf332028693a4c026adfaa3f37981548a4555e4f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26039845Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default, Name: checkpoints/3136/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3136/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=950f12d292fbd78735cf5334d03f49f6b58c2fe29307bbbba6d762c3c539a87cabac1c39d3a99029ecf19c48b041b475c77f2c84379586188b2e1a732b14e08f782934fdf1d6b6f6027eecad70a85560696b03c827d5484311ff086e7cda650d8ad4b73683f24d304796690b017bad577cd06e77f8d713c8f46ffa04f13fc9080dc9485db93601614d734fa0f7ac266664a8caa4485b9a63c4c152eef9c0c3cb13bd1cb61fd98e13b0a02b3e51dba670f613b77c9b76ce1c13416babe2b4e1a8ce4d0d85b1506575cef1a42b80b4ae80d5373581e44344010ec9db8a108ba23a370381732c52766bd5badc56bc375fd8f29a64fd94bb2cbf65461af715a6b2f2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260387324Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504, Name: checkpoints/5504, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5504?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a9773b8fd2211e0bed6f7096975faa150345490999d73b645438d6cb4fb8b8a6df079fe2e1d5fd2ac494d17dc73ae53ca76954843ccee87adceefd3e3266c940b105ccae1061548c4bd68e4e5fa6b3e628eca1655b1520c702da5a42f7917ffad8b38a9510eb51d5a69d6da0998ea86d67eee48116b146507d7463d7d9e9ca255e263ececcbb5930a807b5f5b0c4b7ec4d8f5af2deafdd9835ab7dc8ce305c4fd9da3a6cb697c06cd2dbf16b084b23ca3b529a2e6a3f16ef06736126297b8242dbdcbeb958827d58b531124f4e1129cd410adc0136567bf78a487afd09ab31c10d16a2f4e579145418f480ed2f3c01d5061427d655b9fe759e1e70a68d16ef3a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260475916Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128, Name: checkpoints/128, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/128?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=1f4d7fbbe579b1afd631256f753862ede8cf419760370f16d83bc33a5b095387f9712147ef8bf051e265dfe657944e63c9332a36e124adff17637f67035ff31f29032778d045b763eab741cfa6158eafb3966690c8ee5ba64e5202ea6e23c97a6fd5901adfcb8eebcff6a6ded1f960a16b4f3963f285f63c5205ca7e93bf8629286e8c21762be4daa0b658dfc10e4f2fcb16064acb198c32e001478266c25ccb79c7f2728261b071b954cae24fcbb498f90b5271608a785baf7015991fcf4b16a2dc1ca76746b47370e2322edc0aa040d9b5ceeb907c159f8371830dcb3a8fec10331f5aca9b9703cb23938425c783aae28d633e49b46c3bb632f4932f47df9f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260487834Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0/d, Name: checkpoints/1472/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1472/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=91a6457039d4ade4c233d915e31696e9acf812c9bf57a4b3c446d4e16417bcf7ce658d0f46e424a4e2d53f3e9b93ed9b4d9a441f3cc344c37ac932e05f93bf2e2c985ceb9ca44580c22a1f56a0684cdcd2cfcbebdebd70655c16cf11c2f9be6d30e06b1aa338669f9b34206e34134908e669c64358977ea488157fc87ee868a9d4f818989851e579c644261ec12e602729dcb4a9dc58f7615302e3dc8f1abd443ba9f2c348e952009e9b72ed1a5a0db5cfee458774e2a2fcc2bf334d4862ff88516f4ca7daa3197c5ba32dc0e53099021fd443a2173e92a83428b4ab68dd51924f6811cf8a9635229c958b42aaac77fa31d913e4677776dcc380898f48776f55&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26052247Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0, Name: checkpoints/5824/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5824/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=53842473e54afcdd23c2b58c39814f38df9bc2610dce22acdfd7201806ae91ce47bc31eb53a20913c8dd7dfd4fd572d0df8bf1813f564c8e77099de52f5d7df0125cfb097a6954a4a84589b7f24864d1a06ccae2d6fb7113762353434a738b8b1b03d81a97712d26817795a0920aecbd7e648b24d9407d18980bd2fd97cc09101d320c1fa6c9f555f309cbc4d2e974ce44f9dccd471a95800dee4e3e9e18fc6aa8f1cac861137c69b86b876a2da949f4069ca8db61ddd30810043b28117b31d0969b06749cc7e41b5716c8ad3fef52ba2c64f52aa5171cd336b7c5c67a8f44c1decc201d3b1b41c85aa7325529366acaf74c710f4cb33c080f6d4e8730038e7b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26055625Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0, Name: checkpoints/1728/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1728/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=63998f1d717253af6ced2c61c8afbdf56e211129d1ed25c32e90ebc1d070503750aca22fb0a9a1c5795047ee225788910e6183b6c4618ec2b52844a21b7ca90faaa31966d67a72317108951b18d0dda95dfe98af0394a1a060231922b4a4bee08a5167b6668b7e4cc77993ad41344a09e98db57b19a6fc9d4dbd42a2a18524eb394cacbe84bcca2f96788ed1f8adf33109948ca3119050f0078b8aa96ed57f3d38e3ea7fd72de1c894d2e6942497ddb9214d2bf9541ab1efb17bb0fbb5a6dd5ea7254ccf44f338d6c27c54652571b85904c2eb5c21263f9050e2a86e4fd045a17170863b55e6c817781b73cab6ac5a85b4196c7070f650595633e1a5f26ea452&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260564712Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/d, Name: checkpoints/4864/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4864/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=1506639a0b4fa57f6e4eb6ab488af44ab85c7189e29c9356d45f3f226ef9f310a431437a160a9faad3fe38e24e2924065949a6a65371909990d17cfc63fa47d443b6fe220bdad76577dd1b12e5fe4eeb9e8e9dae87d3b300bfd7840729ddf2e929bedee068d3f7787f4731facd54ebc2a9d899a63e22a909ebc93cccc734d68ac5913214daf49f2512b72871641236e66663437f1384f8221bc977500939526222f52dd637dbf64bd2dbfd645dd7f1c268604ddebe32a85885a4b18065b0ca1e55211edc5cd85c8ebc2ca363ab946f87167d6a2129d31876e2312883d0122d0df789b3890805f2a75fffb2ec52cba6217fac1902edff49fdf5ab568d95a9e828&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260586006Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/array_metadatas, Name: checkpoints/3264/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3264/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b5bd42f01f73be7eac5a9638d28df3bbf0cf937809bc9d067ca662b9d99f50f2f9baa872e405d1e75d34af3d026b811dbc8798ce7592c064559647f1b18732397a4bf8166bc038fedd7092dff2916a3953a5095dc2202a6ddfdb6c0cf40339525697e01bbdeaba5e9eb09a9a084b60d275ffee936d416efdd0ed7c77c9eb2a7a26c97475a233685dd4ccf2bda38e664570e8fea1c7d54bcf0a3d221c7b351a753601dc7a7857231c0e4eabce5c0d5fc9c37c6b89809cfc1e53843a33f1a0df2a824f8eb4653f661ff9e21991a51ee42cbd107cbc4f22497c1d4044adcf0573effc663ddd72cad482d56d588a98a97dae185c13060a45cace7bf4ab9bdd55f68c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261293975Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/array_metadatas, Name: checkpoints/2688/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2688/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=ae58a31e025bcf7a267427dc8a8df1dc90a51214d1b0525aed75985d1f8d3c8099675a39a0e63e760b945b51fbfbfa22ade9b82855e3abcb2f299811f815510e27375c5f62c6fe390c03409fe268a8e464ac16e5827fdde01646b3173177c8c26529515a57aa99483ffa4936e5208cb0c49ff217757badf65943562a32b6cd118dad1f70b18798dc934879010817184b268837d91b948e16faea504993eb1df1569adbf87eb630c88a1632b126c6d900869d6b82a893be3a75d05a9f63ceac4f6ab22f31deaa0eb4d2d198a936e82e29cf72e89ec91e915049af03a10d23e3eba430375003fc66da38c362b14b893e51796e61324655099dd4395917e21a6bd7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261299262Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904, Name: checkpoints/3904, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3904?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=af4db38037d0843691043b84fc955605d0e11b5be064f45a66a58f5f9ac60e18282853f6cfe35ccfc3188dcf1a85681bacfaa459d90ac3ca81d9eae5dda4f0d060f4c9b55c1ba0e1a927aa2744f7193a1e575ba1b6a3665a99419d763048c18dab850d57cb1398926690d2529a4f8bd71a82fa078aab052a280030bf00a4384d1f3c96cca2aa03575b4dfc92e58d54cef1b8712ec61ba6684298054714053d6fd283c91c797391467c5fc5ba5d58a824c12e6a1d4847e8a3739f2ed3c48b85f3c7e03dd188445a5341242013c7914231483e35ff0878615a86bca7080ff9d8c25f182a1aef484eca1c701aa7dab62b4a353c8135cea09e3933ad5553e6a2ad7e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261358106Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0/d, Name: checkpoints/3328/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3328/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=4efb2069cb168aa333dcefc4acec348879cc696904212a7cae81a071a4b3ceb81aa9a7eb5ca75d2fa6b66a1468f5557b4ffc4f3d432416e8cbdbebcab367b8259329379abab3b54d4d21ab2970fab63290ccb2d58aee564873efff423aac0063b3beabcf8c56422cb19c146c4bdd2712122f6e7c3f144e6eef89344dfaa4176e9d69b5447031b4224c75c65c977fdbff04d9c91fe47d80830066489165d354834eca393c9587bda8f4e77bf0c147157fb0c8476e86f4acccf43bcd1c7324d9c8a0d2d0d403cf5bd478adec6dd75ab02d4bc49055353d7b86c6abb477f860572ac75d42948aa120a1a341e75823f466bf72391d27d6de1892e903eb00da4af0e1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26136467Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default, Name: checkpoints/2240/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2240/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=9170214e27ee413a78591861fb9b85a4bf56b6ec8aaa2775ece7907f4878ad780ae6b46a33b254db39a700fd4a078d750ad79d19483759a06b8e27a7f254f6c97936b8e1536efaf8fe6c1bdcb3ba0eb33553abe9c01aa0487dd0f717da5e7b533ded35ddc7e865a3ead5772efe7bea2819cf5e822c6122bc45af0343712abadcf17f76a18c87463d8248fdf8663130d4bd273f8229faaf3f3aaf94dd55b14fbf84d93afb80c4684ee6ae74df65130261d49517f8be2baa56102762fea3957ab35fb3499ea72f37e9416ccb2d6be1c67a42546bb142030862f1bd483e972acdc1cb55483146e367bbfb52d9927aa1fa2bc39be9fa3913cc1daaf6f93404d09f7e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260249295Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0, Name: checkpoints/320/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/320/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=1f488fe99a550ae897597b131c8c833e9b1f33afbe0ea555aa0f25fcb5edf99a44a3891fa064c05caf49268b9a3f6f57dc9461b8150d34bad2f0900b0c551fcaa2e34c66cf2ccc88cd1bb64c24181cc99441778c7071a7bc82d471ca36f649823a7e619b6360abd5f15ed437addd036d93eede0156257e60787b37a72cac9fa119f29ee2c97f85e9a42a87e1b229261ab4052c01b43eb56fd96af72f8a137b96e6b235d8cfde8c84dad7581639893ffbf31eaa41d1b9570977c363568421df052a3f5db8c7e7965319e2ccf11238fb5c3010f67c6e456f5be18a19e3a05e8dc5b0a2e4cc26c6088801f3e84ae813c7144e709a1342aaf607d7f4fa219d7130a9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261870254Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/array_metadatas, Name: checkpoints/4672/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4672/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=9494c50c09ff67becbe26e3c6965f7778f3df24ae119fde7820a883f877fe6958ebde5ac4302e85bfc9c67af13cf8a4cf28469789d0b96566e719c1c68587da50ce85e3afb6ad5247e7b4ff2dc52ac098bddd3debd60a41a038f678eff9d3ff5dab5f2cde702a35c8f5b20887279b36f5f0fa2ee4bfae6dbabf25475400661c495e500f04352e71289c0cc3d6d377e8cef89cdecdd5aa9c3f2bb20550d595cedc272cec8a78f2622454622944b481ed725e926a0e8a59d168a21371cfe648b6c954d98da403663acb0df1ad990e7925df0876fd699c638ca69cce28b21ca340d5bc7c9cac413c6967d22e8ce9ee94d07775ad1dfd98f7bf543a678348894d09e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261903676Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816, Name: checkpoints/2816, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2816?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0255af7e734a0f441f566ce8b21bfc66e38b82550bec2db44d49aaeee19e09b5a2e41342a4d1f2e0df16ad212f21a33098a1058093ae4cb77ed6b1a3f44361ad8084a6a80288266a3aa12f3c0feced0667d6c682d3202cc0dd509ecf3a593ea0495693af96f67d41422926b6bf75c54dfa3d15d7182376a1fc352d595512e220931ac5fb84df580017957f7906921845fb538aee598a9581c69b756221993d60434d10d719dd239c932146bf773d5974941bb679f549b3ce15b0137d68eaf97c334df278d0fdb8bdad0f46e00fc4472ab60a7b719dfe936d9b2ad64b8185abdf901e4a4d987685345e067bacb0173ba846f16573d1c7e90ccac9dab1e3540452&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.2619572Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0/d, Name: checkpoints/768/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/768/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6cfdefb109b08a4b0b7a14366582a85efed9ea98a2cddf1c41ee478fd3402aef1ccb477889473944db2268fff14a567fcb2b3e2621e5f0518c867d38c9dbbe069cdeccb574846ecda55abe4beddf53e129d775d8ec218629b3e45ff66b8a10837033bb2be5332aeaf7f016a78456ffd7de2d6b9b313077717e0e50780117b0e2c0dc6ac3c15db59c8641cd37aeaf3f748a02e9effd9e55fc271474c580ef4bf8864505de95db1cfac6195b3b61014e3f4f2dfe44444bc0a435f33b74e606e1475c55ec33979c74352a2882c484577184d8ed97008caed3c7eb0e066aca11aedbb065db23b5a9ca16d0b6a6879545e0cf5cc79d5dd196b4edce84e9008e6ac4a7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261967576Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/d, Name: checkpoints/5568/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5568/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=5de3f886341d87bf14d415270e3d1c1c9f0fbf61d4ddf44875d25ea79c2b956d45a1da609793a9e990a9c1a498352b4c4bcc041f980992623306650ff04b3560621ae4eb4891a0fd65e8dea7c94861442bf13aaeffa907ac710f43db3ba7c95cd5045c25dea0d4f663baa13d213ef75e04b05f37348ebc599d34ca8d6f387fa2c98c414dfab456fe2909d7e871b1613405bc17f5ad159d22bdd30729521861994526c81236d0421a664a403c754d5e772c528ab13db194c287c01a513572dc18d5a6d6e728883339bdda2ed0de51fd2aa1db2c8ac916ed01441cd965d37422d700c0199a8af9ef6738fc92353206531210ef2982e837f49125ff71a5ca659e48&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261993735Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/d, Name: checkpoints/128/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/128/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=5b09817703e9a312176ff466b258f31a3cf509269d3a96ea5f1e1f1c032ff2dfb078d4724e259c6169255cc42fd7c97d89d10aede195e468a023ce63d03210cf0ba185f06f85f38b1e41342418b76305d963f11551665914ece9b9e43a1abec943962de308073827ce840dbf43b01c8562bf38c65a429fad86a3238348e13a3206e1b6537d8420d51d6e3daadbf92e102c3c100d251ff643257e3a2052dae22db35809f0cda617ccc15c83293d2ca9ed46ba016bf7a9bbe5687c607f72a186cd8af36dd438467cc7fd4ebefd1f1ef383345bf9e853201f9cefc6794f2821a9afdf84946bd3bf18cdc1e6ca49f019697f485eb4a13a9ef0d6cc3aa64a14bc32d2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262021475Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/d, Name: checkpoints/896/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/896/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=ab6eacc6b474f7c4ae3973e1638e21c104b09ea6bb96f876e27881844c6282690e8aac50eb4351e962b5aeed4fa2c9687085129ff5e7e5b5905c0719c708c196b423f14bcd61d07e09806f89dbd1ec2cb51e6a8faca1c2b87265a25da38a48d8ad1ccee1659aba99808b850a7790b5e784bad7df232ae72f52b2a757497813334e11ba9ebfbbece59ffdec63a3c2a71d17be1eda256aeecba7e1244114abe609022603178a4e3bc8ec21077cdcfda2036b48a7fc3d09b618a1052347c8eb6231a607ab71a87f6e0481abafa7d27e8241fb1136cb49451f497eb81e3070c42464326fac168a11b69447cefd9045388b9f8f191dffdac6ff99a311dc430a426721&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262039324Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/d, Name: checkpoints/3008/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3008/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=2d99ae3f89416881b0248ee579969b69001a63e744036dd01071a9e8e8c1690bda6c827e0d97f6139fd8bab13f519a6f0bb96a565cbbb1337fb798f1000c38a0aff75208c2e8320dd6e96bb08a652f3cd0fde3610a52a2bdb14a3641716d381de1b05b0fae7d857d1f89bcd353531cef86d4a5664caa54c2f549c2e595d9a4e93ef69db13c7e33cf99ddda6792b763c4b379bf1478df7f917d7834fb47e4279b63b26b179081d604a16e32f01c4b544f260e6695fcdede51950c2e409df30453d47112ce80dd1c91cac8afaff7d033aa063cd153666fef031db02690237065a5f235ca9a14d95d9feef5ab4dd8dc57702c3746b200e56760c72f8a39309e5c34&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262062911Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/array_metadatas, Name: checkpoints/3840/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3840/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=48316b2056832ea7a66b2dc3a1e9dcfa73207e6457fefbe3197038a009b0aff027e48d025f0b3f779e59f8e08a6f8f5b6962503c3f1138569a56700193985dd8ce77915287afd8c7d41c310917b471e6df08ed050497405f13a9c51811ad48ca52b524248a0c10bf1cac079d2220594a9729a1e317e567496f34adfc547885de69def9b8931cef73fe570307f67aee9812407054cc32ae9189fc51ae80a6c7050e57fd6f0be7fddff5e36c5b066ab51cc74022f4f0f31f44684262ff20f6f76e2eb59456ab03c8a8329aaa6d188ede846835c69dbbcaa96dcb3cbc3f55e4096f6a1564344494695825f6e956bf11ff2c74f03d7b234f4381eccec170489b79ef&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260599697Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/d, Name: checkpoints/3520/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3520/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=75ff65094b119dbd867f8a728414bc43f3fd8337aa6a4e9622be6fb47400b5a169b9e1ddea453d64d74f8f6f968c7c4f644ef03c47a86d2ba2bc46397ecf2448a485bb6a4bf8df53269851808f8e081dc5d086ad1d2015e5fa5cfe3bb0845871e5ba6f7709229dfdb3f1e6633b79905ce2559d929090b594f6b35c0348f7d0779af7feb306384fbfa009ec4230f66988117301b95694eb3e23662b123ae689036f1723ea82f3bb8cd35c97a4ca8fc7dae0a954d268a3476d537d54af22cd224687101e5e4fc1303ab8ef2a854186ed7308b1fac60a59a989758f0a83f21a348fc2319975f28f5b49ebe58fb95cefbda7b72320fa2bd4843433643410e2514100&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261885928Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/array_metadatas, Name: checkpoints/128/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/128/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=9f37fb12c3b915bb5819d4e919da9763bc0af13b9b8c0f144a0b6c1cdaca3b1c336a9ceb869d3806a5601bc6f28987f45eee7b217a623cb28144ca005e0966b1a112f38c931f3a8d7965e04a2646172ea773e184583ee1685f20a977812cf1432a974dad0198f73357ffc1d35c124e7de8a87166e70fc18b6a32096dc32a9c3c59dac671e2492f4fd27ad7e1433fb742862fae08129a9e9ffb4bea80759d5bfd7e45b814399c2c69b813e748caf2e07b3dbd87a9e268e82e1136c96abf81e9b5f561e15d155230262cd3f63d47e9c4488fdf88ae24b5b13c1e9ce2d3e297220ae82e62645a3f779b898cf0e24a96d4fc5df632438e2ca5451c67b95f6105df75&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262143931Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/array_metadatas, Name: checkpoints/2880/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2880/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a68d776a4270e408413d9ea2ade31b3ab0720fb76f71392c9a236fddee048b90104621a409d1c000953050de8d456cf1f641efc1a8dcac2acd45a5041dad12f57bee1bc29f9c4531c4481489ebbbe929407e6c1ab19bfb1701aa377c55b706faf209937212a14d3745bb96052c09e9f73e8f2892ae1493d3df9cb40b5b90965668e1453f3ffe22ad85ed7cda01f5417410f677f58c0e93ce13a4cf587eec37b3913086f524bc76a3c0048bc50fe1345a66fca6ebc787d920b04eba8f2d3c2a027fa064e007ceba41d56332f498f34cd885a568c6671b58147d29dae04f767a7b5b59cbb39b8982f82dffdcc04915b4cba6a341dbea0c76fcf86b455b5cacd7a7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262171842Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0/d, Name: checkpoints/2304/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2304/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=8d4c04ce304b9f79bfeb97c8b96d76da00c0994b9da36f0de6689b87b645dea9e4f089ee62dd82fdaa36a3d76d88870b2fc9c646b9cfb1c4f7182698acb913fa7ab836a3710131c54151b433e2657c463911e12e876d32a867dde31c7507bb53d4f0186a3124cb7f8bd85d31d9210058473b3df83f9af291abb5738e42bb4c92b26aa0e2ae00e991ead001cf547712031749ad68bb0aa3c15a2e8664d5eb27c8f454dcb2b8d5c20b38357908641136b6907a5cd331fb3e2e936bb046611f51394a0a4ebe507ae228acd36c04f8bc153250ec14a36f0ab493e4b055bd4380e9eba49794d0e7293891de53146fb2ee86894bc19adcd7d6dd8102fefc0fdeb2956c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262177681Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/array_metadatas, Name: checkpoints/2048/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2048/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a143e4684bba87fd4a8d8f670f5c743c125e70dd17032747dd9f7bc43fb7073d329b2cb593a03373a9be0513c525e2ad1de671b9edd44e33e91b005949cefcc236649e422b46389a2dbb77d59a4e1ef6849a501d50d3ebff55c3eed06485d1eb0ead0dfa929d253d279197508fbf5ee97ae06a21d795142bc7b4955d6b4d1940d9a34d915db9349b0517b6dee0c836c1d1a2d9257cca3013c6e51dc7257e655b0b196cd20138754a552f195f93c7a2841dd920107b0d3191bc0015e0cb2113c39a2a96e4bdf5eac5838b7ca8fe711f11397ff5d4bc9277be58edb5ed6f048fd6e9c60a5c896d02f534fd79f3ddd9c31a8dfdba8560a20c3f35a9091645f6ec9a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262202716Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/d, Name: checkpoints/2816/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2816/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6290bf19a58e196501deb78a3cdaca18197067bcefdefc1b53a2798a2eab17d3a014dd9eaebbd0da3acbc262ef74292967ad77eb61b47408d00e48a987a2acd0a823f642398d6e20a23b5bd52d7abb6370cb0a799cbdcfe6d7834f21e4e89d69e49982ac92d89319fbf1208526c6da5803c31285ac4adc68166373cfabaa31e69d8850d149b5d8e31c365dde181ffaf88a9c1a6e0f8fb7754505a2614058e630b343f517a5301a8c8dfdcdee95f16ff3f4113ddf92ef7249dfb194a7d0ebf7017c675b0cce619d5c32f5f36ecc82ede7df98bddb721506771ac5a50ba97ce68a060ae224ad1f244301f1b9a67a614c685948792ccb41887a41fd7a5d9d12a944&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262225468Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default, Name: checkpoints/5504/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5504/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=47b2bea986a06e7001b40f36e7aafdc7d8b87b355ac34fa74f28a13ba09afa8bafcf65eecbeded8e7b61caa6ca1e758d244b5789f9261d11ea43ff4ab837f4dec684a15a9208c5feeaad5fc9d0197886520db0dcc69667ecc2f51c496402db9b5709ca3b9c21083a51204e7599ae8ce7800939b06a10fe1303fc961bf2966db3d10d27d7fb7d6f20ed0343f2c5dcac17bff6bfff3334de86f4ba827bc09d499d3bd63cddad9fb7f093624df77499838ef6d67b669fc85297137ed6c5cfcf15a324f8b4f459e1051e38b67cad98a5572c7acbae2acae26144327cf804372f1f72271e378f93d0476ae1292c1ba6d710bc0ffda14c17ee78d874e6a8d822b3b316&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262252619Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/array_metadatas, Name: checkpoints/2496/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2496/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7e4a1d69005584c5d107febbe9210ceb0c57664d74af82acccb3931b6a06fdecb3f71775bf1b638b409945546aed12710c145df6f6774d23033d70c68655d7ea6af09a1d8865da81364b20fdb35d0f590e5a9c9c42394926f87ca8a29bac29215b8494ec3fe396b29ff42ab4f14f76f9bbd9fdba40135c7eafc307bbdab59724a23f2ced24634e4dba4539dad4328a70fc62acbe60bca44956731c56318668c73f9fda2a37a7a81a8af1034d6afbc336dbb14b470b7211e566cb0ad47e217b923cc5a71c397e92c5fe378382bbdcfec456b91735d3891f7f6b03e39a5ba92f5a5d95030a190fc035d6aed8ba739766dc289051002236e381e3ed4d7f1ff67fc1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262046246Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056, Name: checkpoints/5056, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5056?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=bcf0f7df0be7cc9f0e278676bc430ad4308dbb116c402008a3b7b556a4c388c7f456e09fa551a639b40b0dccc41379167f77cb27ac8b2716828ff2e9b49060049e5639755c5b1e58d8fd644e7ec6f342ca9dddeb3f47720007e3a3199b35cda2e2f751ccf40cb7b9aded3ea3b1fc9f532a8a831dd3e91387dce5c653134d1c1d151210c1bcdfb794257cf0bf0a9e1809eb78beb52e44d1cfbf6b7320ba3b21f733d059fa6292405ef8a6e11b53032a9a2352faf55a658319467630515fdbfc33b08512350d1c751ab8b04e707d2312186610011abcc8cb32b6b5ce48a3daf369db6719d5a4557fdef5c27f06ca49e3e8cc583f1ad6f7b1ff315222501896ca20&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262335197Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0, Name: checkpoints/2496/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2496/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=67df06a52b07daf6f38be3f0187549b9e584a42427928876ba25ff5e4e518d877b67139e38871b92a7e726e414736b2055bf9a8c743a254f27f1cf8210565c8de19ab1f9ee1d0ce98fec54c7f8550a920634391841dc5f3bd05cb17665d3e0fbac04324ccc609ddd5c6d14647e4076a0d382ccc354d70a76c1777b9577e0175ec19c41f0f0ca6bfbe91b9f427c3490a64a021df6e345a545e6652c508e1fca16b0e1d9036d43b447452866163445d28b2e4a6ffeabf893d12b025a635bde3daec9910effe889633b791a20528c0274a0fb53c15b2028e0440468883b1bfa5a29d331a89713e96a84d10255064ced8da1934ac4290772a0aee65ae702392c17c9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260572865Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default, Name: checkpoints/4480/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4480/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=9f9eac0a5df0f01e246bc7904c76e86963382ec5be6937662eb3374144610c34677214f9a8eed6c2e31d047754f85483c1f2c8bff2ef497f1bba06c7ec3490d74f7d0ad4f5af080f561fbd4c0d18a79e0a8c6595d35eadb42458ec118e5dc72af7f07c10e7a48235b610e858ca15fdf89cd61b22635935c9be9307e84f379afee6ca0cfbf9f3e45daf565e9daba5e03ce0ed87f666d33d267f083573e6ea46d747b8464dd009754efaea2e95beed9a33190cd5bcb1e21af835bd5cfd26372f822b465522586ada43198e79b949f47e128704fa6ebc1d34d185d80548ece117ca95b7ab453834a31613950f5452a62c43604312f9c021f7575b2506f9970b0eaa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262376924Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0, Name: checkpoints/2048/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2048/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6ce2e91ef5027720dd9877b3781411f3fcd6187460d7ff92e21c496966f891388dd0920f5e04aa6ea246303a9ae6382572325845afd8766002035bcd76078ee3a5f5aa71b6357f0ade26cc4540c81f4da4ad4e16bcbf4cd4cf47c32e9e629fa24a9afc9bd6a055a989b621a4833a80607196f668e4689706e755775bd441544742e1d49f1d796f7797bcca6929aaea1a30024f34754f2743e0ea515cfdfffe6bbf1ff8d1b246db633c0748d0d42f395a82eedc3f6b14d6212eafa55ff006c81f99e0d11fdd9012643030e7130f284c86b35231c95aa19d4bfa2fd90995c3a33e0f36bfe0b98501bfe7bff14f5717239bb2855c09277fb611a6ada7501b5b3560&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26237587Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/array_metadatas, Name: checkpoints/2240/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2240/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0da3a09d67a26498ead47c74ddcbca21ab4c1ac50d61f72d6b2cb62faabc5682bf6a9aa03f1f0fe780a2a1db6be82e8dbeefcb4608b1faa6bd0d18b15b8e1a6936deafd4b5c20789b5cfa391e0c927923e2bbe2e557eb2c785daa7582a59ea94a1987376ca79ebebd1efb9047651d1f779e90110b7e4973d560fa2d11ff902926d80122d94276a919cdd563797589329855f5a39d5a985e82131f2f90aa229d91e6c23e044ab67a4be7ae6308f80eff5a7f8d112771827f88f26bf2b8737f5ad951f66b1bc09f6c59afd020fc350bb540c014ebf5aa0440e3b031aea143ceb448b481fec3a62447a5ce5d964b89680131c982bbe161dabaaee7fbd01a2431133&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262414124Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/array_metadatas, Name: checkpoints/1728/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1728/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=1942a6673b6d9d69d0912a357fb42d93be8e4ec2e15e3ff09954c5a2e242dc38eb941574162e5fe5986707e3a65fe9dfc7f0f7a32bd422a8e26c73e74bb46c7548b863e4c805abf81782fee9f22091a00a3147dac308b515cef984f67ec567486262124ee20dbc2fbf63fb4fa8ba67a08e836b355e646be7c9e17b6fefeeb29632ea3ac30d417b4c9d57169929123ce959ee08968e8fd17f90fa56295ec83a311a49721699bba380784c8dec564d7e33bda81353a65e851bec62c8680d58c9e4f4eabf85c5eae9170f5d814731370687368b922d9702cd05b28e778e9c544f2b2aaa0203d8595c38dc00478fd52e1795e922e850f520e2e4889d97b7d721520d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262140695Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/array_metadatas, Name: checkpoints/3520/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3520/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=2589ac9bde9797e5e231e48e663f4e6adc6aeab6da329d3ab82575713bddf1aa70c9ca986a3b1f2cff1ed776bab9c18ca1f3ced85f38a6d399be99ac3285cb4cea4da257f5aa5b83d9c387b48875ce962b44fe61d14410bcab794d222f2d2a1009480e507595e241a97621a6f46c348fb58f5888722a69dd1914f6ba05bfd531463a809b733a024d8694373575f94ce5230d4e05683080cf870a3b36a1cb247374e8215424340d9fccf895188d4ac5d24a014dc654b2308b39da68182ef4c4444631c2ca75177ed847472f44ea1573b68e71802d0dc2ab8f0e0dc92beb6ef1a7bbb881becafc2b9e0ab5485e40916d026d36f12ea759ab1f2fae4ad4d3926386&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262489973Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0, Name: checkpoints/2688/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2688/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=1d95e76a3169b05518cf436deb9b98ad0baa0aa0ef5eaf91fd8a575bcac82e06aa35e0d760f695b17fd75e81c1fa2091ff0affc8b513e015b2e929ec92b3334e89549708a9fa78419ed4998efb6b1d9bec44c025c27f85c2cb84dee3b2f2724d5758b253e67614c819ae49a6a1d37bb925938695682f28c152ebe3d3d2a4119ff37dfdc8725ffe73a4cf00d5714cec718ffaec33b55672650ac4e16c5b62365746933d8b9d5b276164b9fe9e1ea607e46ed1161ff74e603d82a32c8940a62d695a1470740ee419ba5ef533a381c6d9653a2c33b06dca6c3cd21dcdbf97ac2b99a43b70cde5a1a537872dc4b36a56c3c3815355def3b0f701f29ac6f8f5d60d6d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262520221Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default, Name: checkpoints/2624/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2624/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=444048af6e13dca1e8ddce74253fecfff0027c70be43e95bc5023c40718bdc6e3f1638f51b99f28967d52568edb5794298c87dd22beba809b0e719905e461b78db8a3715152dae96daea09ef647e4e8a9637e5ebc14d3ddf784962f01c5f00d99bac79489038755d66b303e410d6d95a64d64ce9dedaad45eb970a768b5bff81907572cea54eaa2d0b1b4b093115caef9b7bf7ab32b000aa960b0d2049363c96d72eb247b7c2220040e98e95cc43d185290de4a8903c78779aca2b352abc96dbae1eb37fa3955189178188406b95549c079e5e7e54b7c58d680d4be478a25e002fdd818ab2ea97a6aac26b819bea3b69753007c61146c596be97b07297bb2416&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262217212Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0, Name: checkpoints/2560/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2560/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a267096e6d3a9e621aa9b0a556fcf9be7c53e0657f146c2bb96a0e8108c9fe9b3be8194ccf46ab2ec2077b1853372f472790e7d2aece9ef612e6908a521e4ee69f9ee9a56a542ee674eb6426c06da31c87147dc8483b7d398fe37731feaccf6d55fc9a4f04da83d7b6a39a49dcd7bd2513d4071afee7364f9feb043dadfb6f9e4fee14e4c09059a22770a605f3087706e5739c3aec3b7982d04b0d6562b90b04b6a9a8509942021cc093715e4270b2c7808ba4798989b50e7609513d717966ff2a2da28488a8ca5e1c1ca49956c6d7a0c72113613ed95d139d815170b3d271dbfa4624bbbdb369bb32f228cdd0ea900c919ff6642c59a380f1914113f61d1b56&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260558205Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0/d, Name: checkpoints/5248/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5248/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=34709dd8b7e20cf0432a08a28a6828e048eccbf7e174853c2c49e29a45af96e9c28a51ee6c892aca3feaddfb3608eb7bd5fbf00228087a8db2a961bcfc24db4c5e68782f4de5560e7f28f5b61669a8019bcd5ad08bb4f9922667bbe95c2795563734cddbf520bab0fee9616a74ecacb750c8028056f3b1209e2c783887e0848988dde40688daa0fa5492e53fd6222b53793d46fb2835d5b6ac9739d48cbc634b048dc1f8b0402bfb34598928717c988e0b5eed56d119ab1002dce4a770f2c29b30c7dfe7ab931bf73926a4cb1b02563f2197a21325061f583b1e56cf20f011388f4def9818189a45d1acda24688e8d51778dcd03b134975bc8270e6701ac5138&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262644337Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/array_metadatas, Name: checkpoints/4992/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4992/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=2060e9af8cc59c4ff95ffef634666c4eb1bc229f48ced8b8d3ca8701a97c4c6630070aef37160dd3369f96ed2503cb44ab3209226516cade51d7f3457c6769760262b2a11271cbde430fc4057491c6f4e6e88e5282a735a234f554044bd62d55f5953f47a2deb867320cf6c31a13ba7649f8d0ac50a5575672744f52957ae44a8412dfca6778d9e39536b08cc3802c6d410513e6d9fa6082db41b2a5fd6d01233d925891e849f65094f82ab5a7043003a672096f6e91adf54d435e483c04ee2e847ba6e9c6ce3057ad1482d745fd101ba8f730ddfe0bf7fcdc8ef57f200bf61d7bc583450f7c73f9a662eaecfa0cc6c1d2c1431bf2e9b990f6f12b0959090606&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262642633Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0/d, Name: checkpoints/1984/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1984/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=724802baaebb6a641438a285cedfeaf01369c17f42f00122c53be16f19b2462a9c4a6a802730e901e3977cb6885e30afde9cc68bc5e1780ac5f0c6fabfbc503dee1bfb01ece49d935444663bc94a620470f456ef94119016df56f63f38da82ea37fbb9c08508caa16cac75cbdd06093eecdb33e4171a67ee6ea33275283bf3f68015a7bfe017741a98e21bdedec9124532370991bbe0fde192513ee8bd25ed2f37a15c22fbe0e345c1ba846b16a15b36d4f964f6f6ca14f857b94533d11dc619351648fba2075a9c29c9b69ec73871e0917d06539b06d7e324d37713aadbe2530fd62cdac5952766212f4fba7f2fc51ef64f581c535c47cd8d255f0895aed06f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262703206Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0/d, Name: checkpoints/5504/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5504/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=681501c3252963dbc326c04ed3d11b0bfb4fb7c1a1de34ce06b7e5dbb7fa48f8b21ed464f9f7ecfbecac1af5e233b0994346679b5a8ee5565b3acc18828a230498a7cb06e23407b770c05fc8b197edb5f4d70902897b01e2dd6244dd7dcd22966bc011e67fc8e4d19f75a50f2132ddfd1ab3476e16523d2795c05b912c53077c51269916f4a71e32aada80e06280ad322af40f0ef508e82637495b882d750c4f9dffb00eff6f47792296215c58ab8de8d6e9841a9f969f97f6ce56c05353bf3991d00b800c950c54ce489e72d0bb2768fde233c4d0ea5a3135d31a038f2a1590838f0eaeac371f8d7a4b3c50237ab028ba21ed9c19e31daf23340c89ead08ac2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26192449Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default, Name: checkpoints/5760/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5760/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=21ebc1f028d6a9fe9518b97117d064c7efe483ef18bb654e934eb899e9a62a1bef94c9c2d91a5ca1479cbcbbfa61b2fceef4ff3066120d8437d421bdb4053fd7bc90c90030491a3e1b64b09db421a6329584a40f1f26b2ae28dd9c08c047706f927c2a91c9d20b6fbc253fc7edad236a80d810a873fa8c5986ada566808c72f050c1895e576c06875f29f359effba127607abdf5447c81abf7407e85973cf3fe035e696f3b140d17615ae0707265b97b85df0be54457192479fdbf15831e88a8818dd6fce930d4e2724c0ba4684e5e9478438fd9e8a7ccdad41b3ae80e6fabd8cb81b18b3c49bb7f4476f74f4882efea0b36b8acfb60765f02a4d4f9a17b807a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262741273Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/d, Name: checkpoints/640/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/640/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6dee82bec13a7957b07ad494d126f2e0759b3db5b762a909c5e87875713ad9913248df356c42c80d362a44eeeae5954e5472578bcdf27e1dbd50b98b3a389db603cc70f38b066ee0766bc28cd26bafe504ff053b1a3f234af3e5f4647cf3ec690f1682d33c7ba5c238f9eb8f246fc5edacf5274e81e11fef8634d9868a3b35247226d22b2cd2d110138a430d032110c6486194ba304f960dad309a2c2a3415a12cb9437a1da3a575fbd15d2d5f6b46dae3fe03aaaa7f6a840a3af265dac1f5a0de339f3ec09c05720c1f1bd01d2e9fdd5fa4403e3e92fa7a7f60ff3cb6694c1beb4c35bfc74b3d8c077dcdd9ea3cfbe18f45e5ed56adcea12a441ae611c4d348&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26053566Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600, Name: checkpoints/1600, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1600?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b78397a8133e274c0878701a321d96a13f70a4e989dadd70d8a5e6efe76586f81263a1b42e536d8b1a78b8b0003dcbf2705d15f1240989cee4b91cab7a2b6449ae9d14f724bb974de29f7e965d8a49f05f78c7d6e03b0fccff69492ecee28da5efadbd7455f7aba0831facb837681632dc8de537bc64c98a64f74b1eab8e466d94a591ad0447acd6a7b1f9542f16bf78b4818fb6084b3642685a584037477f20ae324bbdc01fadf65ba2d92c938dde8dafc1369f05557ab4dc6be8fb2faee0230399c7a0fa41a7fa38366f601ec3dfc78ff079f74c9cbd99f65ee89691a766e7b263a2789b29ebbb32ecb2b336eaefae4771beb1fcf4f2e9344d9399fc894ae4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261316035Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0/d, Name: checkpoints/3904/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3904/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=75db094e0e4c752038b5f710fc9a0b4b11edfd9406b79562ac1e50e815431e77d4e0b5b85d9f3d6f66326239550f47a29e5626d3b86be761906f51372549b58023eb9f4d8c210c026d7e336cf1ea6b02a8610a7e5480ad0f6b8f822024cd7dca3bb44d9316b3b388b522f022c3c07145d0adcef587079557ad79752f89b9591ee2bcb1625786a4ca9e9f1bbd38b4d581f78e3ef8af3647f1cdbb4e55688210ec7e71121d6a0cd4e85ead6793c104931730e30ec7321786680761c0f320f26571c820d036265d173734ba9b37f3c07336638477e01b477c209592bad66094b27511cb1dcbfeb3608ed0841056a31a464e4bb0c9527b074b834efc3cadcda1eb09&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262853507Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080, Name: checkpoints/6080, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/6080?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6ee927e7d453b20c32418d39528eba4d243badc402d15007ff27fe1db95a42c9f204f5d1507add5404d879785eb1ae39eae50129df292862162c725b075284de56819df3645574c6240ef7a0c9ed850faa6fc21e380c4fb90839c988a8595fd39d27a6e2cf994528b6983b0dda0b5143b8b488ca1365fed9380c3d00c50df4b0477a26f64ec1bb5befea6e4bb02cc505a43158081e02e05d0d6726d4a43883874af7f36e0f614167c772202155d41f87d00bfc9d6fafa7eea9e795749ac0188190a3c349d78d66c5c8b0fe8798ca2007dcfdbd09784c2749e391036d10009728f168e7e1c4b3807b47aac7feaea283201de40bc62c7e8fdeb9714e4b59b507d5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262700347Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0, Name: checkpoints/3968/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3968/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=ad68493da4b6aea2f7110d41e901b4b2d27f4ab504b427235daef0b14188badd70ce83d181c54d721b7773687d4d3a5fdf52a290b9ca0e4e4b356262c027496170dbd799cfbe1be3e4ce46af691758f7697b249278c67a847c8f8debe0969eda2282b883ced6a8ad92041c3af42e3406f10461a99565effba864f39e0434138ddd5e422b22a3baa613cb89497524f4552b9d943af9ca2f6ac364a8c11df5e934fcc3ab4584b3b71ff799368a48782490ec7e74f65a62c511e6c854e8122099578acdfe23a9de56032315f756578d2d44b1c662f881667c874ef391f70e7e835a492d3d94923158119962363ab65d7a9f281989aa1ae94e06ff42a86ba43541f3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26291359Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/d, Name: checkpoints/4480/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4480/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=2f020928211814e0822f4eee4280a7d33ee52a424b2baa4c49df15323c102fc1ec54b194e6f08d7dd3a4df050947c5f72a35db5f3f31fa24ebb79d2a52e8132e39a7d7d15b5ab4c37cffb4ca27e0a59663882b79ebb2f4b02b32afbb07df743aecf00919ed45fe3646b9c72d488616f6e83650de24a0ea4c0672597b12595d310295f90091b0a91af70ced3aab8c4d8bf8d9403afdbd5196725048cbe941096d87b9a045a1aeea68c54d865a85d213bb0e2484e964fc50d49ac275563bf4c9238af9a26523163856085a78f2a52a325e1fb88c76147218d0252c57f3b006ef61e055f06e42fbdd0e635e61548ba04f25ffb480b7180b78f23e0674c63120dbbd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.2629478Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0, Name: checkpoints/1536/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1536/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0fda175a01baf5aa995c90a614ab0dd41316748b0f247294b46d78210e44d2b1a6ade3feaf7e3aacb25a653362224cdf925490c1c9c7c0f7ac91a60180f416720c2563bfa796174028a7bcbcc9321188818fe841a1762e18e92cf24f89421edf0e7812ddda4460ae2ec0aed4cb5ea4f47c0c986a45cdbdbda0199eb3a95a77e1481a60c882af1882942fa964002ceac1b73b17cfcfe8aa0617fc3342bb30b03da7fa8990d5a992194b854567645989c88e223f6a4b0464506ee75c9169cb24e51ee4926ad930ec1b669cd8b8b2496fcb95cd54d682aacb989855252199917ebda07a1199a1c8be3d38239b20f3cf051a5d5afd52cf4c4ea1724c96ef374529df&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262963181Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0, Name: checkpoints/2112/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2112/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b9222be268abab9bb90caf852b6e836a0f7dcfce32c6063ed7d79ce53e76eed1779f9b06f7d3dfa7600f2b68c4b3608f45a1382f8ff7887ecaf50bd40ac3315eca7a80807f53ebec534733ce47adf85e4d2fd5180353d634720c6e74c6232f849b8291e1893d405d2ab1c6fc290afd0be2ce367d7d072f01b26dd45d6c3a148031a8f7a7f868f6f0f4ee78ac92a427067716f5e05fa3ab840f6fc6c2c049b78e754362c589825cd40aa82f9324644cd01dd206c266a87adb72579fedbcb085ddae957ceefd3ca5f9695678c0e6fbf890dbc008fca909954abb39b974088294806ee73214fd27493438ef0f6c0ac0c442857de04c7ad55a87379beb05c9f1734c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26298142Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/d, Name: checkpoints/4992/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4992/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=592c5f689449748c25bff4db53ee29bdaec2dafac8eef08a8a703d7853365135be54ee041b7f6adfd1c7bf37e4da8abb25c35cfc2ab4233cddf494351901e815410577533abcfe40b85568606423359faef7510a7f7bc4bc241a27dc8606ce5e3c03a08dc5df1fa7c9adb55ca0b38d462e2615b68e6bb7b97804d3ecd52f76b304bb1741e61ef531979b6a1fea16c3e7979fa3ca22dc249b1a96e47c493dcaaeba7a5917f382d6c30547da463ca60db5c037bcb89b6aad9ba35b37eced439244b3c514348a12c08e9f41615e8c273a62e34ee3c3d4ebd738585b93ae175776499bdf113b2c2da083554b8b04e4eec20840b5aa4792fd58600000af40487b70b7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263014923Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0/d, Name: checkpoints/1600/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1600/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0694f2df33b25c74df190c79eb57d6dbade602b982ccd37b30a4bf36775cac3bfe73419d46b55d5f889a2f1242ec22f779c45fb7ec228145215fcb85f293f15e538a08dc30b0297e52a8adf37db9aee7d2703ec914b20e9fd910179fb42cf6f31e05d984b4f4b2ca4a30e35d1f9f305e2399e73ec9b7fa110d3539433b883f94d910b728c723a8b48d2dac21b1d5a940087ff0407d724e1c027096d12796f701979b57d2cdc45e0d9fc537ef4cad0d180e45c128b6295ce8130917796489131035480c5ff866e23c22de45958e9ca8cb350e7d6cad85c1441ce876266eb848bf3ffc835ced4823d9d7793d0f3903ca76d1a6a38a454a96e52beb210009f1885d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26252756Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480, Name: checkpoints/4480, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4480?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b164ed2216b2ffe0665c6ea96b519e1aa89969eb007823cc9a71780566a6c4b8cfbe6713da193fa4dd9df73979a482592079fed38d2bc9bd8ad5a178a05a3dda1e74bb4d9a4aa9118c57d5a96ef2349ef8088199ecb15ef305ffda874e3b9060e696e4d3bc12e3d4fd135225e639939717fa331bab32f92ad8e5d9b1556d720112b151328896b79c88da54c447f0d309e63543cfa087d9a6d9904325e69f7e8ca112cc25ba569d30156385ff6f88c6a082eeb971e2de81d861546acfb13599ae859504e7272381222e8aad56781d89f1af8a5ed4993f032b0220bd003b8beb170bad5d076603ac820209aed76ef34f29bda3d8025a00ed560308a50ad2bbcbd2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263054866Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/ocdbt.process_0, Name: checkpoints/5248/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5248/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=811edea9265700f344931d7596b2c7f66abb8846401becc20aec80ad2d91295e6a4a6dd5944bde013c265791106b17684c6519e33a6b5e28b544d1a03ba5ed25da724c7956a26f30f08bd1b06ed89855cea0e106c88fd4438fc2f41dff3329f2e55c29a3f0f142ff999d167acd81cfe2df177edaac4e444c03f3660e26e9cbaa6c22055c5e55d6957f43af961f0c98c210d4133b4f0af0fcfe9fe6cb367ae46a86763a51f04136125d579f0da45a54544a4cd80fe8cd0fc6136ba07666c137afdcfdcdcea1be7639ac857d0dd704c292620f3928a6fb5ee1a41d8e56bad647a91e32a465737dfe3841c62809886bd470c9fb368d3ebc8be96d4de394c82e9010&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263077112Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/d, Name: checkpoints/4736/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4736/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a23077a646ada2bbca97365e4bc397fb93d04c616ad4974345799feb810df5623eda91f3d6c6d305cd7233d0e4e73fde1c6c16343edaa8433b22fbead708a3c838f720faea99da068564b868e0d202c2c3fa25d77facac895ea4effc882b1f555dd01c934e35172a8964f764df4af420d1bd58a6e50d55533edcacd570d64f90713a56602b1d3e26d9cb99aad15db3a78a3f321b2b5b999d31c9beb043b9ddfb0569e3e9762804835d0714baf0fc92667c579b30dea9efe4c0e73119a271b7bbced3c42953136dedeab95ddafb5df40a25699a3b9d6a37df13369c38b509c42e88e8641a05f1b43079bd2b35b25200407321a821343cbfb1ee0698a41bddf2b8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263084022Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352, Name: checkpoints/4352, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4352?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=1d53de875eb41aadb79d6fdf3ecf38ef5dfb3553e0aec10b73d1faee0461c76d0351c41bf3da59ed17a1d9d37acaab46576a667987d694cda2dd89806cacbb54fd5cb98874b6b48c1312d1c24310e38eb28195ace1067008b31f15180c4adc2622cf4f292f30e8fca1254ee80b75b8e4659c8a5b9f1382dda38ee1bd79ee0eec429f939bd06accc07f68ba7d07ec60c0bacdac1f29e920b109df2654b97a7e8c5f4a9ee194c5bd9a8011bb0fd549591627bc800fbdec95f9559be159af7c92f3c7ddc6097bbed5d4955b9ccaeb91e3571e428ad132352ece9bfdab1b2b12bcd3af425812c42188118eac9b44d1d7959ae2b933b9bb9093c2c070b8b06cf88a64&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262639415Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0/d, Name: checkpoints/384/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/384/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=50a992440fffcc37aa746f1f6b9120ae982ad3d0b2a735d53f6df900b2e0b91761a4274bbd51cfd5e88451bbbfe1bfde869b5ec364ab1c7d392fd56c170d95fe2146a1416a33ae7a919738bd0d286794656c9ca553cbeb379ab1ac63c0da07933dc22e127ad58ff24abfaa3c4b42d43cda856cca087dcd1054fcb4af74832edb6d021798aaef6a41e4fc0ca41c0657ee61cdde87b3439e6b4aabd61cdea064aaf77d7c9e7192681c643ac053a465d0d36cd808446477ed30715138f09dfeb1fc3d86c2d49e742afcd1779accf7a1d0882824df065e7d2f6120078ad0ca572f1abe9ae41b6c4f01b450779b27e629c7d1eb1a63ab10171db624bb7385444dd2a8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263157405Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default, Name: checkpoints/2496/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2496/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a1529746c3afa45f2a1f2b32f48de9a2c99219d794d714115eaea8fbac37225ea140662fa8e1311f6f63dc66b99d45702e4bd381690267d6ea28426454591072bdfb502dbf4fb3af9d2b5b6cad6a415a5825671900212ad8106b9ce4360dba889222f5acef0f9558e7c13f3f4e432fc7af8491b0533e78c6539a598ba3d19ed967b682d58c68b33a7c75f1365c0706522d2bb784544ae8851fdc4aabd897722ca3fac4746d9df6c357f23000c057e9b8597b9760acc5dd74077600f62a6ef842e0e2d7afdd72cc3d46a2496eaf25749d660437036609f3336e8c19b4b99824642614b0d6d21011d37ddd848d22e4bc3f5c44516af7e20c2eb9a5ada330bd14ce&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263167452Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0, Name: checkpoints/2816/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2816/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0f1b8d026375098a036e60e942a89746bf650e9dc37a2e6fda9d4c332c6d919be30d7d1111d0baa26d8843bb981c1d9bf221fff0846a129b34c9cd1fb2652d150aba91f9887292e504f4eea0013a5d26ba63fe65eea9ebd04ccf9eeb5b74ae2563e1638a89cfa30b66eba335c667c60b82adecdbb532225fb031ea29d1787ebee4b5cb27bcf3953e40b8837cb8b9b5be0a6a74e6ba12cc56eacfa560a513b1d0a211fc6f6570a9c6fbe7c8b71681ebc123a420ca5efc5cf0973362d522132d42979e228b8218fdbdbc5ed78969d21145be99f083c83799df6174979de355753425cfc4bc3b5f4422934f200f3cd358d3d1f9cf251bd349e93b08e4fa31b304cb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263199328Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/d, Name: checkpoints/2752/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2752/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=9a6385c8290d93da9749707fd33052e8c34e8814f228640ed37cff325a494d297213ede722e2e028fb905d5960bcbc1707a83c2fd22d06bc1deab883e51970feca7721ed620ac5603592eb96bf4830216e46274fce5973e36ae4febb41980e8855aef8afb66f363f1be70929ddfddb0e256b830b4a4280c4e4ce3e5c889842979a54a9205a37a9f1ee1e9cdcd7de3001913e5a73cc9ddc56a81e3762b707f4330e7a4968af364f5092ea99d00c1504c134c806a639215f6276acf3b00dce0242f1ea7e8de740c401bf3059a164c9d30ac01ec71a34fd4aa8b56bd9219bf1e49b256961040c734435e667ce65e8e3ea3aecdf8f759999be964d3c2b0ee4f6ae5a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262246101Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/array_metadatas, Name: checkpoints/4928/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4928/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=4414b3349848eb5d2373ebbc0055d6e3ff417bd3906e7205b91b31ef3b866ca644fa2c81e96827093cde4da0b9060bc8df3fe177d61c1d3ddb9722cf9a1417f29cdb4f7fd9ebe7eecf36471d198b2d22a8cd47b01a98e3f8fea05f12baf2d8a5cb25392b7e6c3d56d64bf0aa646aa5890e5f35313c6995d9b21e66a91fc4d0fbc60ea2ff76496ccf675ea1126b35b791cf286a7ba4e6e950fcb5f2a9d7257a0d92b126d420bade2d004211e73f4babaafeaef3b9dea1c85ec9930773a5f71330c131d58bb6ce38d24b79bbf79e3847b32fe5c159f46ef983c4b3bcf0e23e4d355d62392a1367908dcf0bc92c41d64a94797064dcdc25c06f62fedda66eecb9da&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262942957Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0/d, Name: checkpoints/5632/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5632/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=2ddb3730983f0c9ca5a4a4585906679a768edd980d2540025d2b60921a727132ffe615f1b2e4439e800218f3119dc7764c04b1458e93b74d1c5675c0af7b85ef52ae0e32d8974377b3e0adb470c1ad1115a9ab8a25da30541e8f581735c73c74119fb0a22b4296567afdbdf7aa4c3d45649a8fdfa31dc7a0fef309136cdef3006eddd56fcc4a6d5fd582e432bf8a74c2081b84f6627efe6796897d64e170c79562c04c75de160368d84c2b58022950f57ffb9ff889b6c278d46032d1a1c541cd3ffb30e5f38fba8f6c19a16f023ea8328f66b3507171c8de28f409875bed161688797014db58d8e10f14ba4496c3e2d634ca4f55abf94e15d81b5f81c0eb1922&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263596925Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default, Name: checkpoints/1792/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1792/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=489c4edc2a3939625cca0ce8904187390b999545992741a64888808d13fea37266f94087edae747a9b112d2b753ff70d6f959ab430cf8f0a61c738b7012df4aa0eeb099f491ec25ac24569cb203ae0143f65099cfcbcd399390fc03340122ec81ed3492019bbff85eba3d2936089b1d293bcf09d6fbbc30b827ed6ec6c8c4fcd3d07602dd648838fb6099818b6581f07baa48fa464f4334ebfe80f2cc34ff708bd17cdbe18463fc249071348670fc20007c0ae7ac1ca897caf12eabd9320f938e4ca50dc7973eeb9586d18f95a19b766a5931e7d8a97ac3f0121a1f21ef1ade3d8359d5708a5a14851ea6ba31a50ed113f678f75716ed2700bb511e61d95fe1e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263655654Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648, Name: checkpoints/3648, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3648?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=012ece1ebe86a7fb0b0bb9c7cae865fc3db4a79d487e4ecb7d178320a14a8203593601e5ea2c0b9b54162b270f1b4d8b072b2fcfedb47f1ed716ac137a37f6a4949ee90d7e94c0f32dfb3f6d68b4eaff4db9aab442c948c99dd388193d14792e77f68b3c39d8d0a0267267775ce392c94ecb81b69cfe8c59ffb8adb24a38729b1e69ab9ab6f7d9465423f70feaa4e21a738b66eca192105529ba9a4849fbb2ceab786b0d999ed8ad0deae7ea2ebcac6b373f0ea4d41373b5cc76ee611a6373dfcc19fb6eb25554474e0a77ee228f71c1b47026ce6025794aa023eb6f6998b189d89271ab17063b34e93f804351d9a5f744d55395807c39dee7cb3205bd727b89&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26365966Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/d, Name: checkpoints/2560/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2560/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=4db87c384a55708b4ae5f9a20795c696752077385ad96b1158e6a77d6c45f9898fb5a0a3a082cca6adba0a401dbcb7e2ee7e3507fb7d653c87dcaa981e7a6edd4a873bcc8ffb2f7ea21307f7fcb6002ccc56f928f74eeb6d2aab1aa492b127fa14d5460c54927bdb708ebb33a24dec4170c118349c688566b52c52d39e66a325bb6e985b8d36722d7f2f74e8fcb28007b8092873c0fa868ca834531a0b2aa6df1a9be7bf251fea6d695dd6ff096183898c4eea52218456f0be729edac45246ceeedd962a10b81cf68dba6aaf699a0e33c693bd1c2acec9436f6dc9056d18e916253210cc1ba0d0ea2f1b5932e54ff81cbb29f7dc3547f2b9b3836c650252f343&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263164041Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/array_metadatas, Name: checkpoints/1664/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1664/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=2e7903ce4a06003e30a47c911866abb5f91cc7f6b60c1ba692709d407b19a809a3e02d17e20b1955c6c6640f4818477a33dd96403db73d7d4b4a9f952ebc13c00b75d23e4cf1a6fbead0b5e2cc043ba7df02f60ad9b03a9a228fac4343c6bac5caa0731eb65f57db18c19ad3e1d0b87ff9e2f939aa12f869e238964cf502aa0892957101e9fb0d53f50b28354efa0e53f053cb5a088ca53635f9247744b080d040be493305c9075b7dbeddcf878a19fc93297826b063f57562ac02284e5fe0b0b529a351729b5166736f9903d11364deb9c0b314e8f4b0cfd6a58fd91624af5ba5af8206508d1306a783e19077124ac35bc746be35873218145061dd17b5e0fb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263610334Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/array_metadatas, Name: checkpoints/5888/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5888/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=5c37dcf051027df8657b0449f1b3e33d7774796e6edf8f3849f75d7e770343e84adcb2b7647606a0160f71a878a504541397d837fc68c76b9deea1edcbaf6d42167dbc6cce84d6883752f9f5fce5c3a1411c32c6dd6d12b9f0130f11b17c02004559ebd5dfc15e9cea779d29bef5b8985fc56a47d0c61b890c7af59c729fc16ff2bd2a17969a0b3574537ceba1ee2da662697f876d8a15476bfde71d362abefaa789aaac81df9a0f4e5a8e0e982fffd3676e95f8a304d0f0f342c4557db95fb3b1dabd100e9da1cb386c8fd5dab8eea8d846f317d2e5e1e3660de229732fb6fd420d5e32bbb2a7c5e1b35518d8935e7b41b3fde303a18fecc60556bb89891648&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263981235Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/d, Name: checkpoints/1664/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1664/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=58bdbd4baf6099edb0ad32b8e24915650c2cc6ee978cdec6ec2dd826a26df73036d189590028d9b750d4b2ede2d206a1b934814474304d518377cc4487dfb4937cbea55d14223f230272b2a95229ec7a2bf9e1e10e497fb57421e2cd38ac501ddf02d0fb049d5b97f1932d51fde3979258f40f8c8fa53b42bf3d6a4e36121b6fe8ee3428dcd7fc8821c7ed93ba2fdb8804fdd8c1b50de388ec37f6e89578a43e4abcacae75a4e617a8a352094f2e69d874ba4a410491022d073143ab00a18ff6af83a975584936c49bef990439cb1db5a0af17f60e249c1c6cc622aa0083e897b6a7aa0d9f9397dfa3fd5bdced6968568b433f99c58041ef8edda5290163a865&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263679406Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/array_metadatas, Name: checkpoints/4544/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4544/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=321faaca093667781fed4be1256c7c4eb758edd85cd3eb17124aadda0d2309b846aee97ca4231a7a0df9690ab505eb87632ae6d2460606a887dd50608a15fc6fd969dc802477ae6c7a03a12f121d1b0ee707371a1293c9e8394477bd269243178aad2fbcab1ceeb93ef3ea3e7519c871c7c520ae36c8ece5fb39d03183c9b846863b49488c0b851521ea6b482edadf1130802c45a7a0da6e98d1485d93ec5b54ea249499abdde028b150bb5827ca437594f62ded75e74f7df2bf3a7c499bb9f5e6f98cf9ba87b02b30797a84f5450e1d8dc9b99a4819b20cbc141e6b0020eb132b04559b783567ca60411e6eecb9cbe77ba40d46fb4c8c4148d3225d281b40a2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.264046372Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/d, Name: checkpoints/1408/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1408/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=3da02ad04fbb7ae6bdb1876bc6a291e6d622ccb7f2e93cf3883b65824372e97573c6e8a80334492a7b91ed28455213e70c5387fbd876436ea6812865e00fcb089c6099f8732aa0d1397746d00d655fe4c0fe89c5422fedb93a574f73d9f9bebe5eec7708885276c0bbfea7c9b93784040ae898141ec8276d21bef25e690f438f29a653b26593e9431b213dc53fab3c8788ba9512eb6ffe6e5cb28886c79dfb3ffb07076a4b93c37d61c911bfaa977257539590ef77995142b44ef099bf7f9dfa7a6828e3cd4ee659bf560cb52b19558a1397d07dc4dc474b9ee74c437ec06ff84df199cbd68786e44dae94dd869bb0769e276ce7f73f0d4a0f2a831f8479fe6a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.264067112Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0, Name: checkpoints/5568/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5568/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=4c11be62575c348ab431079e792e1795f3ac645e3d64836180c3d2826e198ec1c9e59ce83eeae0a92dae0ce986397a6c15ecd0bfcdd09148a50baa78e080766562411d365addb9669e90b4e89ed73844f9487dc26a9ae6fec678a3b44e10819d9e58b5520113fa6a3b8da989f26f1e7d4f0d016ca4c2497bd7f16d88ff34b2cd823a4c37927733171008f5cf1e5b667570ac5558f0318c1c480f02b73309427a994c5e099ed96d2c8de6e3e149f8e81e89bd8a13aad8df6a603ab5b9f6a39e744021c1ef24795ca99a3751d8ea92d4f765e82c192557d1f48a0d596a7fdd4448aa617dc55e6621d31551b349b0a96d653d0b0f623de46e589f0f5773cafc39b4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263103031Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default, Name: checkpoints/5120/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5120/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=49f8c80f68202f5aa5affccd75c990479869c800522f178ed701bcd08e6713eb3fffadfa93d0ff2cd259bec8080d5cc2174bac1806c5309565613f9de8bd67705077e0b98ceae96614527b3b1e46c5bd23a309ee2a902ef08c8d9e4004ea6d6054ec580ba3cffcab26241532408c5b016110b150f8ab9d0d85e1484a0f0ed6a69248a18cc42b96de1cefb44142fb4ea10655ba1fe21e4cf946e8dfae57f264bdb2cf5ae6b3404208f3e769e4406e80ea72b1979f269bc7d7c9b710a16afbafea314e18da6d72fcbe814c78c2786fc0e39eb188ae3962d8c428911461d4b4de338c75beb230af864e2772e84e6fb5229a91e42d4880de73f2307712753f0e5fb6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263786687Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0, Name: checkpoints/1408/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1408/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=960aade5b85224f11332b79587e80f7f21032fb66eb7df45b2b81aae50005019afd525c91f3064ea9ddb419c3229725f1f692fd25435c60eee3e0ac48eb95e4950231e8e89baa866372d9275df7d049db7348d1ea7edb68f8531b6701eb2ac41909c82956a66982029e1bcc4e7e7ebab206e06edff0d8ec0453303fb2c9b7535c216315967ffd6b30a0459bb167ba132eceb469b2fbfe0cfee3ae4e607a60021bc301a8c309c72f994cfeb0e4aa5a595e40e89a995a7d79e30ff1c181b6f45fb5800a79a0a02b27087c97733faa9c9db62dd6400c1028e819061d92b0f8c5b78f75f5117d79ad8a71af552fe6644a0e600194db5fde613871575f9c31fc2d6e9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262915768Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/array_metadatas, Name: checkpoints/5952/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5952/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=62c1290a52a2829a1cb49e4e065f9d06fafbe18f281e60d922099fef8cea9a01c111bbb117d138c09dad1b2f7bdd5570f45689dbe5f852d5361abc4e2e06e1e8d86d1305692ecf40f2b64db4b335705c07c19ba75edb5c82b99c03de12d6c91ad458dd45b81087e05658fe1df99f98516618fd19d12c6ba5038d41d6e2b067ce109451fb3560edaf69e697dbf667f37c6b13357ddabd7cd8fc24778f3ed2d6f99579d7fb1b8cbc9e229d62bb331b1cc564937b8214b03477203cba797bf7b9345d4a55b2772c33ed56d48fc7283ebda29224a3c75943d4cd26b2b07881f4403e26ac085cb08ffd20e32979b9b74d0b84dbb3ff445fa6285e0e532c18094428d6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.264158214Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/d, Name: checkpoints/3584/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3584/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=35f3915e094294953065f616e080206aaac9622028c126c63649563090d43cdc3a3a413cae6bfc52bbbdceac1c2ddeaa035f23257fc592d39d51bbb3cab35e8ff19cca92e2efc946151cd81b722981ad59dc190da3f45a5c71d3657a461f6787d2f4b0571439566e143edc5542299601c787e437c09a98c8b1b50a82347b5bc01594c22c6e5f541e3c76dffef3325c940fc071d168dc507e45612f3f3d907de4bdfb908d876f0a1fcbdfc27cfb8232020677908a4455c70acd048ee2f2c46f7633212060f88213083508b8116ca23d49be809728945442f5b8a1c6b91ffc63999b29e527cba0e6813dad09f9eccaa10eba1f517bf07b0ff409856a2d68e5c87f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261939503Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/array_metadatas, Name: checkpoints/1024/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1024/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b9bb719e844ea42aa4997974a32263526e5edc947e68b753cc076750d83a17292a264c498cbb4e3d94572483001fa3851b5c2cc43c51267440701fa6f5af28e5894f3afba5ac2dab25d888464d59e2525a2dee7f6c62f67489a33828e45b4eb7c53b6157dfc36448ea54ea5f5d346e45a15851d466a8c3a1132aef47b57835bd8593ec2d34ebc93f1b7499369acb1d9afd18e01fbf4f500ac1d08c1caab68e4ce0577ec365c0e88a75e9c8e95c3e7b98859756afcccdbb0988b5f6717da84d4074dedb5159b1b42b06161420bf8061d77963aa26e4bba520ea44a7c999ee6bbcd2f2f786c00868c9ee53d1334f7696a9465856c6dae2f80a909befa9b75410b7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261930427Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default, Name: checkpoints/3584/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3584/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=1a14fb8fe57eda759cf10c5d5d2c7c569c0f392b51a012fec587fede6c60f49cb8f7119a6ae3d17533fe08c8dea0dbb841ea3175af90d1b3581080f4a3e9589de52fd3d414a232c59e5a9dd0be7f4a22b6571a946796be762c7922b2159222b6b48695656657e12d0e51fcf10b89b6bce610bad9d83f9e046889a29c58e94f749004c46ba0514289b859e363b072082658b3e8eacb133f4ce327497114d1b6ca1725e4af2d73b8345406378fd6fd083338f389b019bb37b596822ac4e4dab2db768791a24689d5f3f172b2f9cbafe426f5e467575ed528397810174c3c853fd496a402fdfaf69db11b1f55574f3feb072fbd958502ec7e69064e6ea804cc7d9a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262493755Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default, Name: checkpoints/704/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/704/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7cf03b92129e385337500758268ff264737c03e254900cc523324666c422a846ccdbba45dfdb4400bf8214eafaa01709b0e13951ffaa12ba1f9042393782d224336da46748f2035ab9915ab2ae554af37c29a44cf2550519e036ed17cd72abb6d982833db8906c2f36c0613142af7c6ce336e73d67a1a0545920a194ffd9657d8c744e72e9e7fea292581d62ed5cd4c83307c57735a75c58bdc3551d6ccec588d1248b11991262e5a4de992ee3fe3af882bad7e765c68f9d5ee8c4f39fb3e21df36a2398e41b9d6431e99083dee879f0ba50563a622161360e5df9feba75fe48f194d7544848a15078ee3141e1120552fb332cbdb542f8a75699f08df40688eb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260150986Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default, Name: checkpoints/2752/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2752/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=1f7e5954ed869f58ce542ef0ad0f7554a5c3f01c627e508d66dcca6a3f7d028a116bd90a90b35d05465a3155a48389306250dfe53abe92ecde454fde031fff4f1a7609581a09e9da58bc4fdef1cfc93f8c29384b16273d07a19166373dc7191b4fc2ce920e32aff23cb6c870e72c12c642daee6ba9d795b6bea379a3e8bb23fd863ebda4cb121f01018d89abeeb3f0ed9d44265fcc5ee492a85e72b41d47999091ce8886a470aa80d836051564d7f74cd9c7bfda837ae7959c415bd51df4ec97ca4ad18035d89c31fd2c7b9cf29b7c053d2e915b151c2ba4182586caad65b303849200fbab271586df342f6458338b95b2a320751c303509818ff3297ef6117c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259492697Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/d, Name: checkpoints/3328/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3328/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0a549d20bfdd1b78b4a7ca2fc5807fbbb56df4d6de53a9aaf440b50767c22f7791d4c672401f92ce98e09ea4a65fc51c5edbfb6bee5fc22879e31ada39f0b37a826df06479136756444c571458db80827e72a48241ceb7d471c89f2e4c252603c5fcb00e02bbdb03fa900702a207f41b6cc4e68c2380b76fd0d6b0a2f906ffd6780d29f2f2fc5782c314d2455492eb568bac4e468f331fec6c70b11d4f2cd3336b19825e0538637750a5d12af318ad66e3f9f677071eae266962e803adf70e79bbf02fcd1b924cc2ea9e259571011396e83c8c74258f168f21a8eda0471f66009f5177122d7f09c40a4948d91c4b789a7f5041944e88f5a72a87212f65602b5d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261364309Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default, Name: checkpoints/4544/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4544/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b2881dbc24787a38faa7906881fc5cedfbe0822c2bdec65133458e8e645543c49f5b17e3095992947694276313a3ec25ee4e339400de3b3e887468629f88790cde53702d0974eccbb367ad3734c3b87803d379e362112f8a47f85e5d9c4078dcbfde1a2d7479290d234a6f10273cf4216ea7913c2d2c2fcbc58a4889de6b89578262c7b8f2e1fe8c057c57ec4e9aec3237f30184e6915ecab7c123ff2346294150f3b09f2fc68e7fbb8d0c9f65036f85cd5e629d401057ab584d4a6600f3e38c4c73bc508eed2281f772fe912ceb0059f1be638ea5d7f5431cbb666496fad841ce9d905c65e9083a2884c8a414a980fca4bead5ee64460acadb5796006ecd4ac&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260126907Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944, Name: checkpoints/2944, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2944?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=c29495d5a038d0cabf45fb21e7abd8a178418fa2903b37006b3f9bee7def40978dcee7dad5f56d9cf1680926f6df52691bdb44cc62f91572cfc34bcee62351cd99aa206717bfab0d1cdabd4b390e760bab4c94c70afe7fbd49c9e9393c028744f0f1d37b1523b56cbac5cbc42f17522c8dfb6046de4a1963b1039eaa001b5a53ab8bf4c3c53ba857e08110311c5bd3f046f3257b4b17bafc209281abaa442a2b38fe95a4b815bac4eeace487ec17214195a2ed0aadd2b993e64ffabe684bd7eea96a456af9bc745860ea35b1e6be93b8d236444607e1252842578718e9fad2c275b39fddb493be4c3e76ee984a72a2fdfccfd54e944a917ac6ad04b672e0bb48&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260396784Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/array_metadatas, Name: checkpoints/4096/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4096/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7e7d208066ec5155667b6eaef3a997212dc6f5ef3e82fe1b333f1455720ea4af1e83d88373040e05789fae2dc367906197f8baf9358602ab871ca3a953eb65e8a5ccdcf8c2b70d04db4c4de0f3695d8fb8434842ec62d8084a8b5047335e023cc332523b68f6e79314f3a0c8d7ec2e1197721b5de7034366bc37c3452395a2aa930ca8d329f561fe956db8d642e3b4321c8f43a17de82c77d26599cd911552f77fdb19fa00b7d73606a3994e534b6614cc4cf0d0319ea94e651c180f5d9bc78985c03f8440a81e1cb0c4823b1f51430b907821ef7247c5fb9ff23d207d96a0f2620f9d7cc4206524d2fa743499974aad72129b5a4b3fa08235abcae855884c79&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26033208Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0/d, Name: checkpoints/2432/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2432/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a7f7c9007b71b38eaca4d95666bd6488763037e827d77df653b8fd2f8d000e690b0ef18a7d31fe4c59ea2b4fffc3f40adab0e616f974937406891a60e91d6ec3f868e3ed59630f05fbaa319e7b5d59abb230933a04cbbbb3a14500c23d78def9ec44f2459a0201cb9838e87a96c9ea58c1e1bdd2635f71064f71857fa8ee3717d2dfee13d042a47ef1563f8a5adf89b5b8b333a9164096265d746537ace4d0a4ec5fb68c1544bf90a32c36098ee600716a75222e3d8a0be22f442a3b2df2b30116420e2fa08f1c32a447190d61a5804a8f5bbf62eae393556ab27400cd36b5dacb103fd34c2013a718ca75b989b0764b939e727a103c1b26620c2c23f66296db&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259649021Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0, Name: checkpoints/1856/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1856/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=adfc67db2feaaf4f36316575744651ca850a7cd20d46211217259a13f8e50d2e091d10edbea1be374e6ed43aed0108d1856a59965c529d07b2ffedbeb8f7412701e214de076a0def3518916b17ab02998599ad8df42f2ea0e93facca46eb09fb267f516555a331e2e08bd648c6ca021598b32bffeda5f31ccc2c03b134295e578290504a7669c9dbf3a0c4ae2d279160b701d01b4216c28522dc4e0784812c5afcea71ede7294c9f865908e5e58bab3986d63aa9a16819fe89ea6cddb17023c3140c1c26f0d6e44ad29d6921377d48e342f73393ba2ead0acd6fb1ce260e81874a60f50e48da06a1776e2668b806394eae92088a8cf38b6534465defb965341c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26000311Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/d, Name: checkpoints/1024/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1024/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=66b0f0508384943e9c2bb7b8d934696c035a1494855eb232164e6f39a49ca03441668a50d6d75ffe1cc54d22e6b99960ab3fb1966befff7c81b893b5b00921723c418c6adf48646a55bd708992b7fb62970ff3075121ef39a11e3c83848720556af26a6f477310f3bb821cb091b7964955fa9a90a952a9728b597b466e4ec460ca770c18e651c624edb4b9c5ad7959de4a9b5dda50de05e6899681d7d7833d5bd18add93cbb9b57d1da4e374817c3cd1308072b2fad5fb538c7c8e592d90e446e99d23bbec8142cfdb458609cfb9a944562afa2f16701ca386a6835b56e616a9a49ae9c4345a075335c39172a5b374cd5876aa57a7aab7253ecd30de1b421f03&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259631994Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default, Name: checkpoints/3264/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3264/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=8217f6a46dc0a5b854c5960539e26f7edfb657371b44689d5749908e6e57cefbd89ea3a403a9723839bf1384ed9ca1b769dd7f2c2040700bbc992ad2553c86769bc3560721fef3afd098b55cd4d799ccec022b84a2fc59a9806982a2052d3164c37aaf34a23c225e2adb40ecbe7eb5d077dfe7aa2c46b62f16e7d8990d2161bfdde663161d918a9291260d9e8f93772e6f0b2fa1e60aee78d3a161baac3ef08e9334d8db797160a354ce55607924e80f904368304f3c7e1ed9d08e0198c114c55691473c4c4389ddef8a9d1ba3c7f1fd5654e2ea553932458189007dfdf7dbc8e86eb132f43143e181f3ec42479f923963ed4b0fa6c23bffb5dd74b4c93b891f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260069455Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default, Name: checkpoints/832/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/832/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a219670c65bf348e5f63352a7e7d2806063ec11c764856ca9b1edabc81e8f229fef36701f056ffbfad16ba056f6b70792d30e2e0a24dba6fe8f9e669679de274d89d710475e959e2b47d4b8c78b31ea30f9cb7bc855607875e4d6a5a2c1d22bb69603a7b4192574d381ce22fcf74fe82c452c3de46de859d42b2466999ad3e1e2a202232bec7597fbe7b2fed04b7c06b66151ea4a433674608b322253650d1ef96efd24b2ac6008856b56587bbd2cbbf1e722b3fe2c445b91c5eaa02c64bb89c8306010cad7cb852b4d48de0cb562ba9ef854c82947e1c3136a82565d9533cbdae9dd642e417efbcf8f03d0e8e25b967496e97005978363efad5236233010e5a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259914391Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0, Name: checkpoints/4672/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4672/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=44880aca8c456333b3403613c92a45e5a586a3f6bff01910af3014487aac93deb9b96df1726db1161c6a7cf5404b5c907394a35ba514429df0bdc31332079aa8ca2607e2e331d8a31a99eb867beca106c0425c199b32801aff40846f64eab7240bddf5c9f680c88b1c6e88fa76de64caa293cfd1f2e4b29a15321e3dbd4f04fc7bba6de4dd340a8315c61993633dd8a8abd82c4c81ed4f53e1e4f8415076b3cd06a50f450aa788024f3f86909d5500cd87bca0768805d8433ed81a01df0298348f7856d12f37b1ff1300058de4e34057ecc332698a8a210b2942c62de6697d1fb059b114db75614150e3907d2cb3d09a4b14a4f7ed11e20098fbcca6381c793d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260017751Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0, Name: checkpoints/1216/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1216/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=9984e045434e336cd308487294a183ea0ff4c1fc127d2b74d22a4a21baee7f81bc3bebdc3c0f9b316d83809588a4a6d4e089f7c18c4595ed9fc94d7fe654b636d5fa193ec08f069c34e2da36c95ca4bf414f94a7cdc9e442860eed6e4bd81e7884c6bca579a852b3f35c0113f1ee11a4d78d63e3fb73f4885d621c61cca1ee2baed5da9f25779d5b0ce0cc69363643e7d11c488abcbf28b8bd050a41d1c9346440b844bd98032589efce9c2da0ba4d528bf5861d89ae3df21efafbdba8a3ed8d266a3cafadc5aa9c7bc21a8bbc37a04f9503157c355043c89033ab537eb975bb86a2705f760b5237028a1ed4709e72585481c6fed4db14e8b56775b3f17ed091&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.25992058Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default, Name: checkpoints/4864/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4864/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=ae671bfc52008c99857e6a79bdfd50ff85a2607273f896d0f01b7d2fa5c74db342367d7e0a1a8cf18faf071568d7b4890b439ee94e70dad0f3eb5f5018144f212f2bf67e6149ab6083b649da3e2582ff336d7159808ad561dd911bb2f9b131b61bb3862432b960d83d4640d6ce1c5aab42300f89bfa8f1a7f3a4bf5a462afa49aa3e5bc2d144a659ac68b4f9044352747e49a0f2af6ac67b0a7e56c1f9fda31eab7650c959de74d6ab746f5cb90076e952a02bf52fc78f341b33825273b691caee2d4f7ba5b5da629cad6b1a8f3909cbe7eae271092a591aca7c956643dbae5d50fb21cc6b100da8ccc5981cc6aa0bf54524adf7d26a6b93efc3e504686c959e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259664516Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/d, Name: checkpoints/1792/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1792/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b9c47998c9e2451b9c2f43426dc5dd731d491353310421a235862b6853bc2706f3221e5a1e27a76220b895c412dd779d5eaeba41473e66a8dc81160454a1f028b9734fcb45ca67689c5a282d3292087e1e59d9a0ae311b133f6a7b1f26069568521eefb37487ff47fa616f0a4c7c203f966d00034231230c1c629fbcc0ddf064e86f5b559405ef1c5e9e152ec76ca744e6e3929828527ea2967ecb7b6bf5b2a944d5d336dd269eefd85c349cc242dc8991098bf1830155f27ed0995be1532dd254985e9bfc1b69cca0411d7912f6a8e687778932d969437f09687e905a5450018794df6e48721b657576e461268fd0e7100594bc5d69d5f5461eee28361779a5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259701515Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/d, Name: checkpoints/4352/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4352/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=909d842937b16ab5ff845c51c3fa0d7fe46bf329cd95f7016a9d8981b773603d2e96e5e811f6f5ce782addbd2b2786415c090e4ce97e053e7ef7ebfd7b162727db62a61502b00aec9e723503f65e84f3bfe23ef7b5a33fa39592449aeb4bd76ccfb26764bcc6aede581cee9cdef10372833ce88c832f7d7320dc7c415b4d9cdd243106b3ad5663d3fcce456b598be230f9d4d346ac0e514533961eb204e2cde504bc791f86019cefb11d087fec9e8c1e87a45de8c7b65dd0c3f20528f832f92406763158d26bc9247885b91362a70b9e047273881d18ce13c287fa44bbe7f0f9360c820e7f432a4b4591290d5c64154d29f7bfc4a6e7fecf15ad21f54bc0eb23&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259743754Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0, Name: checkpoints/1280/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1280/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=c022c013a5648c81c4d289cf55f55e0bf85527bad63cde8916d5052b22cc34eb8c357829c036312d0d41ff8d810267c16240396e4fecb1f087223109598b8d7ebe3bf05c05877cc399c925a739730e04f82caef2caeb35d7f0721e9699d811a395c39644ce8e820c6b1282f3bf2b5660c21e43aa7d5e40be141c435d938fb8e130340d2ee8ddf3ccd93f8959b6a1b035c3a4300b8338cf75e51a58997be085ce01c57ad2a6d49467f3ec6ef14d8c8995333174c34fc20bdef83044228ca85dcbc00b7833ce62c49972547737b0254b7a61a5084d42f78998f2fa17c396fb9c369d5a2fffce375dc6e1c889baa3cc97b3718b99b02b919815a1a2c3c422af3272&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259784123Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728, Name: checkpoints/1728, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1728?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6dbb49111f4501a2b3e1931b3cbec26eef6221b7bad0fe131f72d6ba439fd89362485688af0d97bcf1442515a2806091b7989822a5fb4eaabd1d310a9d59011246c07a4b874d86564dd612fc47f52a2915fbd6b0b1d2f04895ef8582bfcaa46e5cc5107a4dcf79b2a336a35574cdac01e9b92c71e2efe37a6ddf058c5e5e3111facbfdf0b08c8c72a4ec674690f6074b69bbe0f91146fe6dc5f8c7f685be61afbf741feb250623d63bff7a4a851fc89b5e4e24b99b8b469a9c38bada390c152e5fbbca1d8fda7d7ddbfb7477758c5e2ad98d347e3710b299591dd1d55a9a3e86a2b0c861217bbcc7b0f0ca4938d3c3cac538edc2753c81854c3269a0bced0cd1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.25978419Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/ocdbt.process_0, Name: checkpoints/3904/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3904/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=24829c52efc2421bedb8d75e9e06fbf22797443757123b7be6b8bf375dec9dbeee5eb32478661feb304425843b5d377229a97270f04957e086300158d4bb001efcf1ee87afe8fe71d88aedcaf8ced40a89a897dab7c143602fcfe3748c1679f389ea77f1321bcd529aca841907bd0dab60cf49417318c3732ba07aef0774bffd80091dc20c4fcc262eacbd6ec1dafcc03129aa9a508ab3a4985985f1c07a6ea912d820ab4358121138a39965389cf7b3f9201affa39b90444dc0cc3444ad5220cc13d62f6a9f9aef82577ea85e2563071e7ea3bc3ce39a6257e329226a28bfc82eaf3856721ad69569091e8c8b6eef3d53371d7a81f488f9ad764e6e78f46ec3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259830531Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/array_metadatas, Name: checkpoints/896/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/896/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=07eaa844d1dd7827ce6b12cad7d407fbc1a12f791ab151fd7c030e02696b6a8a888cd766fdb399dd927d26856388bd477ad03e7bd1cd9078f0ca2076553faae93adba573f1885f4998a9c68007f34a78144c1eed4f5128652b115e22dd33f5d82897b8582c494ad79d73a18bd3e247ab6c1b120183cb3194d207dfade0d090b9d0b355377d8c7f96f3912861227ad2a9bb9b0d541916709ae2ccc197296a83dea74739b17e71b39f7f13e9c33e7171fb956da00f9ab2c58b9078ea51fa0245e8b9eca519e4d15cc5c88174672504ffde611bdac48aa1dfa339ace80865453f17a0d664f81d853b6a7f5e0f99f4171044441afc5c8963bf09f91f2b4fcc782b09&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.25986898Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/ocdbt.process_0/d, Name: checkpoints/2048/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2048/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7525a79366d49836329044b878a5f3963c85dab959e1ee7a4a7febe0c39de0eabe70e5d3cd2119da8b55938071c4370995c1599b75f4fc9c06d9d85627502cd4eeecec204ab8a372c6604681104cba9c02740b2a37d8c33810f664aa7a3245a589aac9f8ad357ea217f48aa36cac68e046bd587e928289aeef95131df80d5828ae7c5c08cd646fb40b35da65e69a6c5419df44cbd54fc32062ffd35eb84ed642d61ff1aef648df82a5d503d4bb2ea54a9c31e26c9eb8b726b800453c55c45818aa9ad752c638e8fd5775b4d47a75713c46cbffe363f04a8b04c94b53e3ffdc9a429c98967f6729b1899f702c81d455f2e8ae4a506fcf4e64051a29b3cc62d6e1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259869321Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0, Name: checkpoints/5184/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5184/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=48c7d47fb8c31dc00c65f6a58ad2076a4cdf4bff8dafb72a176d8cec71ad2cc2a3dc6e7e920a2b70eef655caeaf004ab780e92ac229bf597d5da5512033e23eb763ac3d663f76ced9b712160556925843fa744594439c0cc4437e96b1cc1f3a741ca104372e457ccefd92143b26b4bb8d63dbee4f3a752fc5ef3b1a2db05a4ccc9bfafc0e9951ed17225e29f8d3cce0777bbcf8b64f54f820614d67651ef1d0367ee3059581b128aeaa1bc5a1a9c6657408c850e747b3a498fa6a5db2684d25adbfc0c115d332c8325d0f77c01c875cfcc9375a7965602d568531b5c14edc02222d9748b7fb804535943449be6b76e633116331d4a1700a596946172ef504695&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259875167Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default, Name: checkpoints/64/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/64/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=90ad8bd3363fb19b5aed12191e8d64d17c2e3adce146172680e15b84dbd7cc7b942968eaf71b470ec1ef063e55eb211c3ba59a48accf89602bc0eb521a6fae7389c19c1ecb2d8828c9b4009eac8d1e0286ae71ad6dcbafefc6dee68f7732bb13bbe3bd1df0b6497b2e87096a0a9d291f5791bfbac914507b0aea076ec3eba0041e666a84bac44e50f31f921cd6513df3ceb82d6ea3cc9967d1c6ed18fd37601f9af95d3ac7f756c277ee289385dff8d1e0c3f6f775332a54b440cd6716e13f675e58e935d8807685aadc12a3f3491dc428a1eec44426989b008472652b5775f32bbe6d5d57263b757d899011c0dacd08f2fdc55f1b44562c2b28aa733caa4814&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259903968Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/array_metadatas, Name: checkpoints/3776/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3776/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6f8815703ab4e2928a6d01b13c41cca9e3699f2d20d7cc6266df30de68ddbb00c32d613d169479d20da8b7434bb6e3a87bf85b91dc90671ce711b99d564a90cecf6d7b1adf47e2b8f2327354a77853ceb8cdc4dbf450298640df6847bf44fc6197472f5f142a2c78d1c6b222aecf8a945d4d341daca9a0921c8c90f853ad636c19a28afc4f0eae12ae49f74bc0891313452f04509606f8c652c71ae6270aab2d99c03ebdd3cf7b224cdcb019852c0c000053411df9ceee4c609d239c32d025ea8c6e39b6abbb5c82b3d00637f61ad168f84f1e375e1bb430d806e26e234c03d4050549bd777789c208193e294b8cbd8c58c92d13ecfdf7c6f4260ea263e9b210&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259920816Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default, Name: checkpoints/3904/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3904/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=32e63c14c6ead25bdf8a773363d39b399539444336f61743806f8295fed150200a28cf35957138da66f6db38db2f838e8231fc2449a9440c761f2387adfa30d4c0670566e396c5c82893c9a05e5dccab4d40a8e33e8739827601660b0133436d11aa422aaaa4462ed461049c4824be4e8b774575712b9a148d1611f04e2da3fef04066a1fb16c8a6e6d5242a6f19fffa0d824fbb45a2388e21cab5b7a0c87fa0733c7def7989a2bbed1d865adee4784f94d6635ef547ad1da84d0ddf716011af79165d97b531adf0fb35a0c59e5777c3a75e4c8c19dfb6e921dc051c8c808e68832908dc5193b1a206904d0642261027f4fc0269946b5b2a359c0ddf94fcbc35&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259923061Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/d, Name: checkpoints/1152/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1152/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=5a56de4fe4e85bf5b98291730418519fe9c324bcbd84c305a225e2b84684878bde95a0914ffeb01da48933625479e39e0ae9979e1182c5d8c2c4a52c6b961e25164c6ee14bee281d90d69e295dba86cb94ffd0acec9c63f30e78d547c443c9e1b9f7dd37c3a52908588511f1142c8c9ab60df112b4b64fb34693eb28575607d56c6b787c4cba5e80f45c369fed71a644a82a78240e016e7adf7fb1acc52aa9e2358bfef3cce7ccac73939bca40f429555a991270ca9c70934781ba206271d428b1b2ef593831e7b7787ff598477c6d037324210d8a5e5ea5e9049c0d37f4373982c0fe8b9644f6fe0f81c5c7e00d37e2e79a1e325122c0d81a143bcebe009dbd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259923509Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/array_metadatas, Name: checkpoints/3072/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3072/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=001f3fbd6ec70fdeaa2a44d13cfc72490694bca00343f155e8623bd6882f3e04570cee74d4837dacc7f701f5db164efdeb71c3d9786d946841ce437e32a7922139cf56d12ef1322bc4abeb1c77b90970a4a71b1214a758f04b2cbf60208b158938ae35204848cc6d661a285bff55758c4b3e99f51ad43c6acd9dcba0320499ead02b7cb41e981f724a6454abbada5a686466ab156bddebf896d21340599f6f6ad8f518c0c5426560237d93fc6080e7c6dcc0bc4a82c4eedeaa1a813f78df1c0ba88dd959f94c003b4c2c0b373c42e5a82a4be4830785beb989c7bd66592d87d123a811e9892a5a75d9064a425ff8f0beb4da024b56b768fa8923fe6d602a239e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259922978Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default, Name: checkpoints/1280/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1280/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=19121bb94fa65e59cffbdec8d82a867a6a712aeb0d61e440718006411744f109a3e1d9442eb6a05ceb10ecbeea8840701f20ff80199159668b70f5807a129064f38f67d7f7e072db3754098c4501963a2753ecb1b5f5e7e7a5a1963711e1cafbd106d4508c98335ab80c99fffbf31241545c5207882d1b7314a5f136613c01a901f71a421174ea0942fda47eb704738de05ae702f10693a5ff5395becad386e657e3b65360656264dae7db8534c04b9bbe1b8295984dc481a64a97544edf9c0ff46e2022152013858ff0308c15b3edf0a4b32aab3f107e090a10c345dfe76f434f49ae78262bdb05d58378772664190def2e56860a5e14d52dacf66ebe564e80&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.25994278Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0, Name: checkpoints/5120/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5120/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b261a9cece15513506a60fffe0820b694833fe72371c9f6ecaaecfdf66cd2f04add6e141e62e62b3e61188ca3860679b1f1d41d9e9c6fe27f771c1c3648afd71ed65c8788d6fa8c3d12757c308e23a4a2c2220d8693cb34c4148d37feef97f17e08863cefb18585ccc328bde2f9bb88431909e9e704a6836caf4ce0d776bf81dfec47fd7bc93723363f1d12f1cc1f200e1976a2a58634c3dd358875a6a9a54add10b930d4dd3988d68beb762d4ccb57db986d9ffe16ad42043b5f06a739ba642b65428e542cf6d4ac3e5fead3091d6a3f85cd08a7757144199f14abeae313e98a6bdbcd0be1405f12e38abcf60518e0312b1c8300539514764223f3920af324c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259963623Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0/d, Name: checkpoints/256/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/256/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=1b77b0b4794674af45152993e132118b34c4b5efd0b2e4d0c548164838300b35851be191063fd73d71b7174495af6d33a2c91c1b8bf80ebd13ff8f344015ce542b923fa7d4ebfb5d4fe76e08ea9638e4e913eb74f8797f798e503217e73e8786bac265af42f38e78bc7d445f3a37f0d16ab76eef2b7b12105ecbbaaaf8947d2ad9db67bb67b5825c2c7902e8ea4d9c7cc24a213bb0c5ebe172402e39d417300c71d7f6ff085c8e8b799f82fb464d3a2cd4c92244aab2143a37a5d6f8374c31da3b9062bd853887b40e7d09b814b2cc3cddf66ec136969e88a408866452492a5c96228f2619bd25895293c6fe8034835ff55caa5cf1140519eabe68bebe873b35&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.259987608Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928, Name: checkpoints/4928, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4928?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=5e7645c57447d359d0418682235886a1bd490cb226018f2c9a7874cb208dbbffe35b380838c6bbe748016a275f60b47236b89cfcc4263c5fa6762eca5abb14ad9894d5af0dbb6fedd498ab678b65856a9e093b610a5dfa006b3c470024b5e0029d066e4ff5508fda7fb3f189b93c8fa8e9476cf49371a9b05f4cd3e41fa254b7473f9b6591e32c8379a88fce8152b9c2d84207f0d8e24d58cc6186b0c80edea80a5ee6a14feab306c363c99da6c78d6f28cce7de7381c8d6e2489ac2f9ff013b7efb084d4f24a9617f18f13391f7c10012969ece999b230c98718067925350136a5f7e6a53ff799c205502da5a79b74a7a925a22deee564d4e213cf829c8ab09&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260005923Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0, Name: checkpoints/448/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/448/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0a1fd1f3a7f095eac92d4d2624e45e0d004b167bef395869a02e899c625bdb1be65e7c0bfa295d22feaa36a91d888814d070cb64d21ae80f8aab9c9db97ad0c0f4944717263b0d83404f74d6ebf41a9c140d63a6add348deda7e6349e72247f66f4773d88dd5523834d19c9f1d8b198fb52d510eec399a0ce0d8da799c9f7240418dd3bd10cfc4cb447ff851d5526a1cd4294e653adb6cd5a1734e62c4f73af9e6d60fedc6f72f1ce3ec65a7b919e073db86c857f52abdcd97008a48da90956920b1bee310189cd25d3b5d0c76e4fc48f17f50ff8fad74112cb8475fa086dc8cbf274ad1faac09ea39184e2ec53974f83d06b350fc2a1cf608a91e929d55fc71&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260020598Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default, Name: checkpoints/640/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/640/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=8181daed6c210935b13f38cc96c9b5f5a888e78ef0888192ba0bb868632bbf9364d1c5c3d0674a267c5094b510c0edc50784149805ec9d08b8a68d62f78fe02168a902f425f62b855d6d6d06356cac93236ec88598d069848c5b93fc994f4e2f516db96298669432974b7985fc0a965a691874d50c6f60eaeee3c66061e91cf007507c4e0d9041f0171152df6912bf38fb399e51469a77693b32e4248e2fa305000a939aaff16cdb8c070d0f1e533b305003e2003c39913e11a0bb29e1e20c34485c5849c0ecfe17b12ca57a803eb6cd57787dc39134d5cb6dca238b31304f8106bf47a1418c603e2dad00491156e965b8a5b5c0a31b2e39117cd2908d0d9eda&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260061393Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0, Name: checkpoints/4160/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4160/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=89ab010700193a13c943c82d96179419f43408c924ddce967db6698800f6dfddd87695d4f58e445d28845504e9b5be389f5d7475a4ac87079a66a072043eeebc4e802f602d8081f841e218c2f3be2a0123fdcc7888eac3739309bdc9176a9d85b090455520ff756131d168ac6773e8bb6a8b1c219f6b5aef2658d22736ce1b5590e460a64f9f296ea1cd32522248781bd3b6b137a93fcce3fb3153e938994a96e48b2f6813b196186bca8e1e1b18891c2604b0d3c54c28fcfb60db7240c0834349ddb2aa86d18f5ee25f373da0eff22a0feba7bba07c5946c0fcc84bf956f16a503865fdfa931f65cc2d73b306d85c53a7e1f3d653ac6cbde4acac6cd8eb6940&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260088218Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default, Name: checkpoints/1408/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1408/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0894d8fe9a6aee1e09291fdc414e912fc2719b029bbb012fe1316b7bdc69aca5f890da2a84546e04f4a7c6d190a5c6c9bec4a478cb77148cb023be839a15cd08c0d7bb8338374304398c742936eeb786edea4abecbbcd986d87b72828c21acb157efa418513acd9e31394c3d3c9c3c3d71acfc4aec4254a932788479e2463adae1dabe2d7846cc914d14d40b5ebae21882b536582be4dd06ed518aab94fe6f3a10774bfcb85b57e61f75a28e56c44b9ad87aae90e2bf541313e792ce8530e5d4e8637e322856cc5a98d8743b7d987c0557f77b56335032c0bb23a035d0152f3e4d151008f54a5a6fec4303740895ce12f0bee3dd31ba14d4fad1bddf980cd92f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260074746Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/d, Name: checkpoints/5120/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5120/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=97e97e1070d7a0577909af72847214bdbf660375641e83df3f8a00a828f10b09d50dd7c62dd545e7de3729d87520f4352594c09256466fc521b5a678373981e98592074d6f654e979031ac9cf0d5630689de12c9c48ed844799da173dab4555c35fc59ac2a843af23fc594496d28235e32121d464d327eb72d5fb151911dd73984e9b1fe23fd3108590ec3216716e0df08885f6307e949bc3eb65195ed2d56c5b5f50fef6197a567329d04ec05346260a88a430c10b67d813a7c0f6f7c34aed97b12d1fa48ef3c6af86c3e23d40a5533dd062c824ca6bbd34639fd24c40ab4793f956a765e9e0fd2d5bd1100f83bed4720aebb05ffe39807a9cac03ecd4fe539&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260094799Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/array_metadatas, Name: checkpoints/576/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/576/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=3a3cccaffa978c32b104a4feb102d87bf8bfdc0f6299f9d00cd669fddea9c27e38d508ebbd58e66eb80303d447c67d7866148b43c319fdd2e239a2c58ab3362e70f654626dd250aea5e809c21777616dca83226080eca3dc5db96829da1010a362486195b6080fed725b23e82e5dfe9fca192a46d13bec6136c8eec9ac1b6b076001e6185d00353aa22ba19fb1aafee097eb58580f0a0cfa0d82e27e4f7195a9c731bb4305caafe561076b8e0e8d911ea463e61147c1c2d7c29925fcd1c5da5abcca12e1cbfb51d335243e225899e58f1af09696ff9bc565548a0b67b1772c31c8b6d010ea1669a00749661136cfb2d12b8edd1be1dfc7194c2b0625b8b179e7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260086206Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/d, Name: checkpoints/5824/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5824/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=14f40ee5c604ff3a3d16ed1573bc32f108ef1d2c02a448012f5f97449a27c13fe963e97c50929cee31a4220e8e004a4e4d489f770441303430de7bdc92309641957e8875327b645a6135c2c774482862d6c2e42ee6ff9d20449171e3d369955965b9c8ebca4bcf115bba9dcbb916c6a7f2cf9022fb7dee6607b624f491a3a2f577c56d69608aab69cff7c0ab68508395ebb60ad390b20b0a4e4d2a118109807b22e942f72d18bcec46eb6c3365a3916ec6be8c999e18120a0e0addd1cb48950bb9cad62eeef0c2199ac3fbe2010ad6f7c08be40a6d35e42939974d83f143a954dbe2ccaaa4006dacfbab5670e882d8bc4ab747945e4f4d3f8c09ad5767c27400&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260097521Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/d, Name: checkpoints/4928/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4928/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=179aafe1cfa46108f68fd1d0a96969f7abb0bfdfc8e696da14befb67d5c65d38bf02643b0efd58e1340fb4f5b76d3c53e1b07d22dc247de936f80881b2302dc484b892062ab66a839a7b6038c022b442efb9b489663e4988327b3024e62156d9dcaeb530416f954c602fdbe63d7e6709ea6828a17b6644316fb1ea459c877c4f87659d6dcac7f30dfdc2e29b5497cd9b0fe4055a44df9c55bffa15d22d363354cdeada54eb85820c93bc54de3e11a7688737939c3297b6dcfd24f9df245eafa26439b1443f2ea08bacc03a4372fa00021eab429b09474a76d9394c37a9694543a03e485836fc9709ed4662f59275390475e183f32cdd37c5fd4f0ffe8f5abcc8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260090473Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/d, Name: checkpoints/1088/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1088/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6958389243812aa50057d5c56c8f8c69982d107aefef04871c3f25800e4268edd739fbd5b0e1ccb4ea96fdf3d95e62a6ae6288b3c0bfbcdcc5897aad2f8c837de85543daae07df99e4dc9e616c1d7aa315aa9a234facb5796ee431c9374f0364c857610768f5d3d3bc648ebebc9edace6e2d57cc38d15b48a0fd635d0d3a333038110c50f6262c58962aa2384cc43ac4d9106ee08654f4940ea5723f24ce509f2223574afd1dc8ed978be84340d9852e6c9a8ba2a5c120b0376e71186a1dc8738495bab095b375639e69216ebc389c3fa1362c786f9a676ff84519d6beee99690dfc9e1465f13a36e48bd84a9eaf08c35d137a1e28e9b370a0030d4906b5f760&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260110297Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/array_metadatas, Name: checkpoints/1536/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1536/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a943a36b9c2d98c2d0a536e149c93bd4f80bf93b8299087352ea5a3375c68095936d728af4800bfc3f69bee9f4f514b60c748154ac87117074d33594827a6946e36c1107a8da9413d49445ab5031dc70dc94aa4bbd36b47d6b7b927fc5e90d088502433e6ce40eafd4e425c855c827605144c8986622c7619b5af3b63f85a1a4a50cb7465738b4b706eee723cc4dcf194a3fd194bdd33558989a259e2e0cee0ceaebed6913794d1874c8411d212885779eb9096bcaa538300d03ac40c3e5c94bf1afad49a564f67a59d8a112dd5a332a175f777dbe2d3bf6025028a02cb5d4554ef89d823612a10c828188d2c895fbe5a6c8a1cb771739aa6d8e050ecdca5f6c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260101325Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/d, Name: checkpoints/3200/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3200/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=8ea85ffb5ad517b7fff3025169df55f612ccca8f7cdf2e8c4d415edeaf4c8cb85fe912e5661904735aa322791d15c4c20324f409711da159898a07e579f6a84c8ae3e891b9ece1b4f037f494dcf0770f79e8b0471e20001e4caeaa9ea4f81f754271e42f105584f73c91ce83e7d9b6a6f4c74c3fe01151cc51eebcaf00864751f617f0779226bc7ee5e3940c7f91c02e605a97b0e739d9424cf5cf3d6a04c04cc6776cec01b2b0eefcd1dce4fdf9468d7e1a873a7c04175f7f7d9741dd7daa2b4b04a481d60a7daadef4e7971a0c00ed40e6ea7923fae894b77c7b1eb684601825c5ba5e178d06a0ec2663256fb9d6bc462cbb7544444fcc16e88be9611d90fc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260105212Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0/d, Name: checkpoints/5888/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5888/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b729fda63062131069269e97a2803b162b234cb071e02e7432d6100e44d898995277f6460d5b467af6313b1b6de98554457b4df7ff0c1297af987a68671ef2ba1d25bd5bc03de913578cb6954965404724d6c89e37b070747567ba542fa10002b676b9cdc5849d0a8f1949f88f9b6958768008e67b452ad45db92e7d9b45e0373f426b0a2adacc0976b666a4f8c7920e4cfc441c310b56a4746aa8f8be90811c180fd95dc0f17e4f8692f856bf2c830a46f6e3c5f91033d2fe26ae47c5eb21f90e736d9809a4954592d12497f4790818e491f408be0055a570ead13f3143f12a431e5ddd67d831c779adb67f301e71dd2e869489251c81c8403db0b94c739e96&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260133073Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/d, Name: checkpoints/1728/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1728/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=abab1bfcff8ad15a51ecf435f2d7b8612ecf78cc8086cb2aa3f8aa93589717a84792612d829fde71898416b331418ff924aee527ba60ebff68670316dc4e3580e5f15e3848f2300b26f22a92762c7cc3841fb920965fb81677401637e2387f90bde8135bf714eef35d368f361eaacf3ca7910389519ed400e5be46f9b433ee9d717e373c9a4c1ae03b3d2bb81c0ea47fa2054e4e6756d71a91ef82a0edba124a6a1fcf005a48febdcd031ecb7a60d8b3c48c01784a9fbb9fa8c2761b63ec8f1578e7a883c71d13ab87173fedb411308670164d4bf34007ab05d577af9fd7b63601de8d29f39aa03691aee79924e8a33e6a7a5552f589c659abdd580290f7a6d9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260130889Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default, Name: checkpoints/4352/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4352/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=1e05d5412595d84dfea23181dd41e7430ec213f671a35d13db1a3e468622d78374042a0fa9bf170210946ead2580fe684bf8b179dcfefdb862e8437c84f95b60fba85d16e2e30ad1675cd00e67d060c0e133268cb22aa8e9cdd1a71db896916310cb1cccfad441c47273f8695eee5551bb896af878814503d806f07db47b048d60a244a5b17d8f4539ec4658744fd318611fb94f021734906cf182448f4bfb2a9081fecbc604065a6b4f1886f2b4b917d3ce099f59c2d93763ca7c1af98c78288935149b3fea40bdcb304c38dfb87582886761fdc9089a5a4b1e2944a50de0b536c306596b62ff7d15fd7dfc767b92be7309fe0f603dbda038756f1e545fb16d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260164755Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/d, Name: checkpoints/64/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/64/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=5bc7a12867cd1f7d28863c4945e0c120d39f17edfc13579687d2434419b7f5b4ef2779b866df0432a8c6402a6db2990ee6b7d683dd9a3409d3db0c503cf7e673f8905ed76fa463343a86c365d4286d36230f02f159ff65c73febd4811fb1be0fe69e3a94d67642e7173241d98262673f6d07201af53a9d8e75bef6cc5c7aa0a8623218411e9ea60768f7d075377a9dfae522199001da39e1fcd141d7d41f20d0f62a6291490f5a727c728687e09f0a873f2a46e7ba5544714f998ae185d3717137acbeac16e8a64ef5761d21954bb6411ea959dde54eddf9e8f9b29c4c831f9d9bd48377eed23d123b70266f899c065746d4f424b903badc385a61e8694e39ac&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260176923Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/d, Name: checkpoints/4032/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4032/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=85281914a6875d26f8f8cf3dc416ac3687b04c8ed518feac4225a13c560cbc6a462f5672a36f57947912975080a4557b384e2dc9efe48fa547faa18dca57d2a1bc0332b6dd86b77dbfb2d0451e44b3894918d9ebc41a5e9dd5ad54b68e2682e37db51bd7b4b175d3ac159af51e0caec3c826e884059800b70323f77e0af6fc7205150f4472e64fe2694b56b49604294b3a093ea6cf12ff32d4249aec0733e2593b1ca7bcddcb6ef0d87f1fa686383d39fe6b91c30f32ee5730a46083cf64cc0d960b7e7216633d3468946695ed7b836e2704dd66c707bf07f9fb1738e891cf00e1610a16ea16c5a6b7c5ee889e732c21782456091372fbbdd0187908416d3e13&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260190011Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0, Name: checkpoints/3648/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3648/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=12297307f1b48edcf598018dbcff621bd06e5992e9d31525132bec32bf2f44ae38d5cc0319009fbc7462138f1954c0ee54f3cd95e01088933c972b066b24848ee08db47da8692a1fb67f7b295ce9490ceb5040776ac92c22a84ad97db3b38a2a30ef0fe92d1c3077145d6a1b54ae8ff22aeb3910af2c5613bc7e4ac64f0a7494e2327b5d72c3eaac9d09b32c6bdff77ca55127f2da94e45d50b457196312a018cb665bceec4bda3586b1e6bf313307f0d7530b0d114497bb0bfbba376c45e8c8be1d5b05f3cadd124eadd5c86cb34bcbc192862bfaff5060abe9629cc960bc6de5d75142d45f3acef4c3db4545fe5b8a3f418f7cd15f4ddd1aefca49ec14c2ac&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260209484Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832, Name: checkpoints/832, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/832?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6608bdf3b4cc11fe9a803dc349bdc82053a62b1e71511050d3d63f0fea82088c8a36d9b1ae1f97da8aa1118e6cba85309903a284bca8001d86078e8c62a3204eeab0786b4845bc573fa653ac79d3527ae31f128db987c6c7f389d40be88878a58ced02757aab989923f619fb2823b26f8450f3c0c3c3e96890531375f2d04434c6193c708ba63ab6cba2d0017561685e12b4b0ea4ac075b8f99ab9fb0f3d4223b43896ecfcc7e7941b4409d4fd067123f8091498dd59b0ff0f44a027b3158442522945b0c7a38e180d96c36286f1608d0637854ce8c2682448de64210afcf6401324fdcb737de04db8e55cc8b414056e6b54246e433091d2d5992cc6e1c5a3af&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260194535Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/d, Name: checkpoints/1536/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1536/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a680db86894a78cdb5a7f22bbf846764ed5c4d8b93378e33320b169b01e1d355f4dcfb97dee864e1776347dd67f610fff9df753c03f2db24dd889fbd6ae3888ff53d8db43d50fb4238a961b5fa39321e97cd0aa45f826d507ef7db40d2ae9851dae7c3287470a7bd9a851a1a2f268dfc3ba2aaf3da1d11cf9d335a215d67fff9944eae1017cf9b281702d8e1fd004c6fd36bd95184b2456c66e37fd4294dcddb86854d743c7f3c93aa4ec7169605ec48cbdea0964d84b1fc24e810a18ebc5389e8320cb5a03968b937ac0d9f3c42313b7560c2dca1737dc13b9b524246850f70e042e8b8cb89df8c246e6741e8458cf81f30b1583777fdf7d93b8dc851349cde&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26019639Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0, Name: checkpoints/4928/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4928/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=499a32c19299d2c83e06b2ca5332bc2b9aac8b65188dcdd72fa68bf64d3b2de903ace0f4e994d331d7c733240eb3a1ec946cde8bd5a7edd6c7f5fdc797edaeda7a5420f434075489dea59f864d8184ace5b3d810d558ac3606bca8ad526f8db4350fab82356dd3a29bb763126924e800f07bde196bf812ada7005d862f0fb15ac2804f726ae77eb3e17f2ec468522ed361344d1db934aa75d38582b5f25255b2888e8df463b6e10b7950a9b5e57db3dffa4303f8a7e584c921e88fb1b2f6d255e8faf9e7307a40087e76ddf929559c72b9f2fc28a815bb967ff0ad2e0586190e62245b24293b8afc1e77db1fe7d26475a572b2db13fb712b3edf82870355790e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260218594Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/ocdbt.process_0/d, Name: checkpoints/2688/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2688/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7c2f447e92e23de808559d850d246c90dd1e5e5031149afa9051e14a531a937861515d40ed57f97ce9d637ff533a6d12033ddee174ec70cefd202f54d442c149d06e347e001fa085193840e3309a7005ae012771c733c7ac8ee160eef3bd7573928b6bee8e8699720c06a6f2a51d0e16b84ba1095aa0eec16c5f8dc0c5ce46c3d04cefccc4ca11217514436fc97131159a5e1e1bd73f7b2d50252ccdd3fa4f9cf2b7efa8659cc6152fae5963e93bbd7cc7f39a973c214cad715e27ec23ae128a4b64ddc0cda21b10a0c906db49290bcb069a03cacc5a50215623ac6b3835cbdb2facb2d51168556ca41fc314e7934ff1187804fd2f7547e5c032c343423a1654&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260231299Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default, Name: checkpoints/320/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/320/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=2bd328e9c90987b045ea6a31e2516d101b84525dd3b8d9541b5b6b2ff94af0bf00a65938e576ea8a7a0c15f683c9c58df599ec1a2657792cd0598c6024f121bc4ac64d819211fb5959280be92984f4db428aa4fec8aa75737a57be97d637c8a229346a70758b18d8de4120b88bcd171303f2633e9b794a850cae0c4aaa6b86f8b803fb59f27f9ae538689fa9fe0108e975727f31845484631a6c27578292699bd4ecb8ac9d68fb9d45b83ccc7cfd91562241a0d82cff9e9a6e6a94287235db6eeed990c936abfb92183dc1173f8f829226bf6b64ee6e6b29d5dd73ffbb91118dc0c278f996b8f83dee1e0b79bb2393a9237c4056d2abdae714dd4e6bd5e2f53c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260243555Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0, Name: checkpoints/2752/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2752/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=4be1aebfd522fa6bc05118deccb84597bc94b819d33e3df7a8d73b865b10b55c4b35e10118661601ca814613669670f1007d8feb0b43ebdd257dadd9d16f0cc3e994eaea76758728216eda6c9a8d8be25d92207dc8262adaa7cc928933b93db16ce37f8dbdc976cb8585a7149781324b22c60d30bdcdade8a54c1e99931936f2a903edefaff5682264a4701b5f6da02185a4159aa1027d7ac11cd6b73f4935b934d569776a3f819f4645078cef8b805473b0a818b29e7b5c929051c1c478b500611d80402d7ec8ba08bc353d8e32de9c71d1fe6b125bd586441cf9ee344eb2551b817f47af241d660e51e2cbdb1a52aa59fb0beb7e21e3960315fab5bbd4a867&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260248541Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default/ocdbt.process_0/d, Name: checkpoints/1536/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1536/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0510ae3b94a3de78101189fa60a55f98252facb9c23aaa8b38c3b5710d6933680bea6219ff37b342e1f83f9b02dd5b625114c52c0357be8822f8b6d19886091038d3d0645fbef0e648ea9ed2bd98ce58dfe2ba905f8292e97a30a384e64bf931dc6a44359097f98b37b8a5ed7bb5fb56336068cdc6404e45dce085c286448397031155fe280ff1d5c298256bf4eb994a740ff692e5502a2478bd4595132651c51455151a8f8473862c09c9578c7d9968b1eab32f0098b5e4707ba3663b2fb7d96464b598a71a786483d268435f7682e579ec716ec301899513cca73025cb84b39f5bcaaf0cdf567f3f68d616f958b3bd5b7fc210f74b6584fb8b63f9778854f8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260206212Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/array_metadatas, Name: checkpoints/2560/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2560/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=02a329d7a9bd4db84aabcf81cb66a06b68792af5301545b0e37d15efe78f00007f6794867f2395aa57d6619c95074b96632894706d28d99274cc7584aa2221cb65c188948c243148fd3f4b9c70577b28fb4c8daf145ae9567cd51bd5d3fa349f3fbf9aea6b5afc9080ebba122978b186fd7d4f33db1a1bb65cf6f6ac852b4e2ecf707fd788934312df04dbd352b7d4b2ece12368de0c93c60572a46b1136b7217e372f83ea7ce67dfee1ddea75eb02c74428bd36be43eeae170c72dd425abd3d880afe9f32f7b03e8d7314dbe384ea4ff201e49cafe072c6f5c57ce27c4bbf53cfc4fbf6a0dbe77e189d46912ddda8aaf933c04c4ba4f542b6b0824e122ef0d3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260267452Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0, Name: checkpoints/0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=50a2ab8d66e76596b58647a40c261724fca15d8f222abc0ec6792785798b46a680df129d39834f24bdaa031206236ea1b289be0763ec26ef377a9f11a787e88389dc0882b77ce623b9fc5ef02e24f1dd376d7b4d55d9a248724d9195444bc3b6a5ff00f66271878895adf4820d424cb2caa65b091f899d68cce7902fd8ef85c13e0fe9695b88a01e3b1df2b417e9058951b75beeba7f34936b2b975d2f108b2f8c0d611f97208eac80393e4219a15a46e89650fdac086b513a067ee34f05940d377df6281f529d26b1ef728b17f0bc90799d723364e2fea255768b67a85177d6bf80ee24a496b794cb59ffc28e81bd6764592049088bac4c5aa117ab0362ba6d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26025722Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/d, Name: checkpoints/3072/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3072/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a6d69381d36a55668d7b9151e4e35965866ef6c0c2798053406a2d44cb43cb374589cd2bc00e646098607370063c8de10a34c397bb371615632f40a068971bf6b2b6da5efac21120cc96e0cc9a6ce78cb193a50bf4eea7a60c1859faf8c5388e4cc0f167f74e0b58c4a5c77f902eae8f260cbc48d7b286524bfacac214656f35b26687ebdeb3707729ffb0193e89ae5d42e72638e4a7c16516744224223fe20a0e8ac0d49214f42cd25021d9a35592408a65f1dfa6f52e23d32c3b9d57bfb1bcd3250df98394bec086158424eb4b6711e1a85208589c195cfdc9a4b4dcf2d1984f19d06f3a6dc93f4a7f21c87d98e7fc17954031f9512f6ba036aea322e3d33c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260201729Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216, Name: checkpoints/1216, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1216?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=461f9a281144f65e9661dc0ad447799090c9f087b57c9222540e213861b8fb3c0c8391e5ff204723aaeebf27db6bc7e9091cebbf0629fe8e74a1e6aa67b4e84d16ae07f513ca253688660931cf6eb75960eaa6f1fc1113b219fed445d5226d4f05c5bb0a19533c822a2d1044e3ad20659a2663846c942e54363a3b4760ad0bc7de196941e75db5353e16c4c2e39d091de8c5db8fc4ad25fee90cda8c18e12bda75283f9adf24d4cae0c2f8e944d727f9c4679a0772c654bf0cdf381a5215ccf866d2ff1fb2e7e82ab91fc8094422187232bf10fe1b44e957da7dca9c2b4225c975ebc5a8066b19b2da2a376397ae0f7f91b2e6a475ddb6587d693f3ec710a401&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260279021Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/array_metadatas, Name: checkpoints/1792/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1792/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=8a2b60569646862c00f313109959631e4a63c48b181a8078552924a34d37a4f818c5a3ad525ef4a0ee9de7ec6d1a9d6597a55724036cc31dc46c615f46b66e3cc89ee896ad13b50f276192900487e37c0153bf165ea41370d4c84d1520180a7022c74a2df71e26d941fe351e488e20402c4d4be6bfe56b9e88c99964529d58d09a325152e68fcfd8613d349fd4d74e35d4de47116091542409785e414767eaf8914e9cae2f39d178b13a101fcb603df83840633e6f610d1858924fb178731dd1db08476e79abd8c53e3a1192e77ff1d9951343190a18323c88e8f0c27cd90a15672ce0e2c2c85829b909f1e94b79c19054b80ce4933f17fcbdf5cfe01d49094c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260281229Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0, Name: checkpoints/3840/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3840/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=9cd8b32b6ba0f9d9b2e47778a5e8bb9afaf85f8dec034805ab741427732ff6a3987e02e88040aa2ad7cd3b5abac02730a131e04970640357ebd2ad9533dc52b541b02695d181f72a5f166bc106f480119abccd5ab4570be92799ffcb6845c321791720616c6462a1b18e5e75b53d1c8d20ac1ce6098d38cf132cf798c8709a78acc8267bf757e2d240544cfb56af94fe5c793c0196a81d90d57fc864610c0ff4e4e6345b6c873144dd6992a1a46c058276399811c89be91805e06295fc1266b42df059074f8cff04f1b142f1d162dfab712565c2e2b7ba42967b81fd9f20bcde31dfe4d341d282cca84352b4a814b2d6a67486b64a9068fd5119f3a83dd4915a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260271854Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/d, Name: checkpoints/5632/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5632/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7d856e1272832ff4dd8121dffd1b6dff6a0bd247c656e2fa427884c57e5e95d126a1dfd26155189ef01339249537d01fbb37c4ae316f15ddd2098a5a066c8241414b726c7cb89a84d8f7a6f46bb3e6b6c11798d73469ffe019a2404c72650a159defbcb1bcdab0fadfca07169cbcfaccb25c3b3af2b542f4808332927b9242ab0596ab8d13c7af05cf55a5efe83924d65a10ddd0722f1de5ac436c41f3b4a9c5a0e0ef0661030bfe1bdbe2c59a64be32614cb3effcf919494ac8d5d338c264a23a6615e9c6315e0e503e5c129abc64c851df67147674fb024b49e40bd5dc31a1f79c00f5fbcf7afdd3fc7e2adebf1d76b4ca7b131c289bb08113565bf91b8296&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260288208Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/ocdbt.process_0, Name: checkpoints/384/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/384/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=396077e74e3d03d63c5cc9636fe443435e0bab1415ad9643ad9e565b0f1bb9aa0b6000a8da9344249bd88b0db4aec4f0010472eeab74da4aaa871a9562ee547c5bed7e95840d37dc8b20259793856eb3021ec7c8b2c3913a8df7607a2be8600c4cd266c8d62c18dde71f7008c47d468e24d602b676f4ebaac761d141962bfb685257e402f9aa63a76952826d82cdf2cc0381e657e6e19c1b01e42262cddc9248b7b7ac078ce59ea23b97f0915b2a37d1ba1667dc117b26230a9b283a2e8666f0072dcc3f897891e03543b526865313c8e649430f3923d60376610e6c64e12d7591dff5fb7481b1076ad4fc36100008d8ef0b06588639241becb596516d4bc9a9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260306406Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288, Name: checkpoints/4288, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4288?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=8cf563f145a319e4ae4b026b7750f3e1ce0d46770b7611303d22d3c5b4c48308a4e11414182cdbdf9ec65142de7cf52d759e600696f457ea4af68118beab76614cb3f660cb5dfb37e68d95b5988c8a16f97e7a1d371cb096542fc7b960e9d758c2d7cac68e9d96b71f73ede440717c7c535952dd05652ad76a6e69ccb13311882d95f30eee47e9108084c4849aaa19e625d278cce14c13679168686a2209e7e76e68a516cfb067b1b28170929bd8bbcb37a2d52efe96994c421db9d4cc94f9f60062be0216ab23f4aebc51ab6d792b4df40d43adbdce237eed607ab10228a394c0c12f1117c7f3f7862e84a4bfd6622a91a51d13dbbfaba04dc8a7c73614a3eb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260318846Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0, Name: checkpoints/5696/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5696/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=75c2817ddbd812619bf7c0aac615f0ab8170d2f999974fb2e7750a48d6712c6386583110813d431508db1a05ab1d0564aa7701d45f1f64783923b3acbef758957c3c2f38c70f42732c2a990526ebdb4b7bc32b63e0d878cd2ae4b8b78d2f41b381adb4af7d66ab455b46c4b4d803cb97ff05ed78ecf6737ede0cca9d90818231f7a8f807ae648774e60135c0184bdce978037a5ecde0217661c10bb312e9f2326590c5d89e8bbf0312d872c24e7b7e458b346f7cd40007af4a371ae91693e7ef1ab3763e7c99c5619f9d98dec9c1d315fdf1744a76db2afe33bb6097c8e15d9a3831c028ddd0554ca7fd6ef07d1f4f8cd7d4945be7e10448bed7be1354da2e34&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260337789Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0, Name: checkpoints/4992/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4992/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=3c5a49742529f62dec666a635d7fd8e0ba364c4163260e8c7b35efaba3d6ea117c71d974e8e1a6d6740274ac51aeaf40090b06e3c6a8888256d6524ecb0b7e5f32c1964697379ae80b9a5c7618e1c9bd27efeb24a32f7ca35018ab2662b537eebbc4e0460dd62b9885938adf93f7073a8acaaaa6e9afee4745c0f05a7f0426347bb545499347326e3182a49eac9ca8ee12cd0219a83c183cf1096fffbd302b9d9f07a285bc3158bc7ff060686d032b9c2e2ec4a4d6300a4a9603c7390d5c05ee5790064cda0c375c0b05260a3c9bfbb2ce491d178b737d3e11e9a67594975ff7d7a142fec116f0d18da94982f4983bb6afacb53143b50a61e44310a619e0ae89&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260334832Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0/d, Name: checkpoints/5952/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5952/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=1a3d8e4250e3045a1231ce8bbe4e760c61ed06c74c7e9eb1b603386a02e1578fc43f60e65ac84f942db353708faa91370945a235c0870dde32317dae785eaed5533e51f4f3bcdc769e260836899758d02887864f0b7c18539f050a0d491dc93237e8076d1dba3f53f96bb8b3e17262dd73dbd19c4aa5b702f762cfbed6a4e9c109c65d337556dc46d9cb829b48dc2de50a3eae31d51f337adfbb42bd28a5d7c499b6082707f8d6490798e4834d68a58a1d1544222266e2d1a10db09b22a972dc86729a3252ffaed04b8b0e3fa0af4207aeffbecddb308b25fe3432db15f9caa718e6a7db356a5eb08f606ca9c8ebde1cc76be594031ac9213dcbc6ad9e76c7a9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260323923Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0/d, Name: checkpoints/1088/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1088/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=382aaec17970ec4327c99b5c04631128154e031a0814a897650e0b207c7d4efbbcf84a5afe183819cf3bf477cf007d6d2e7e2c588fde17aa40b0298b49097f093ef774dc9cd0463f5e3bf1039404f043bb433f11ce221b1d2a68b648972c19147976f2bb9c57a836300ff23ea5f4895ee46ceb54d3b95305ea90700322c73492dfd0be7b4e0614574a208abf47f7d046df68b56cc37d639ce70efb1064b09f261f5910b90d409e7edcda25e62feae02608b3883e8bc88e5ff5440425a3511b4389095dcba3ac551c2912ce03930398a20846a813321f5afcee821712aa72b2fb1c1c27663d617db89f6c7ed9d315a0f0d5a7f4fd0842ae32c02d0e9d5581613b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260342471Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0/d, Name: checkpoints/3456/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3456/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=2a02e6463a8587be196573a93ff3f2867017ad98f1692627a278cc83b107ef891d0a212045597cca2363211f8f47cb29020ba7fad2daa628f8f34d10f194fe542bc4fed9b74a7d4a12558b7645e7aa6cc93b02aa729e43076f1ace9828bd261e09cb73129a0219983876f56d95abd73bd165540736c608210033f1d8716a9ef098f3a742adbade6e239eee2459db0525ff5f75daac0abb52aa2058e04b08ba89f2de2025569c3e0b6f7e120fe080a570ba8254d64e963c6aee745270780480caed527c76ad4b10679c8d835dcb47a6df75936949a85bc502c2603621b4909fac73450f2e0a089276626ac4faaed2796f7e8840604fb4f07d2a29774a5dc78da5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260348638Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0, Name: checkpoints/1024/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1024/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=ae76c049661738b9e2e659a5f742c8bae5e694360d3236ac506d62607f97fd36726ba880430fe66f10a3b998ad39c6782ee5669ee3d25eebcdaad28d3ffd3115736f0be8d3fce699c68eb5921811565df9abb4d7d1c3eb0d9caa609c04ee2667b0caf11c9f4b8ecb1f2feec635a1e33500bb6051a4d096fedc4a64e9273a8ecf8e6e5edf98e33c102f23ea23c3f60ba6a9047a14596b0c7f6b1ccaad25c6fb91df2bb1f3500a95bae53879fb28b5c26ef0e71fe8d09cefa0b34fa8f9cb729887cc242fbd823e6bfb40715a1af5a2b89762ef2ab95475ddf97584113ba1267c9b8889563b32ac8dd36c0df9e61b0af59acd88163ace9ca1b1a61d72756cd5783e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266127585Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176, Name: checkpoints/2176, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2176?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6ebc433f8c38cb6ce1c520c86fe700763f4477666bfba206a777d9b205861a3ebfb9e63d2fc6c7140e06b3eafdd3fd12818bfbe54ec12f788d22534286d1f775c2f13f5a217616e1e340647ea83f55c393b6900d1a4311dbc38b4eee76dbfa5c130c8bd765d355695d209fa3154e6848b7d1999557b662c2d7a31db75c5059d3d9894722db97d641eb1c19d7e91a30459a5c476a653e21387967c828653666e99cacb649dc1a970f8a8002f1f39d9fcc96254c5ed038d1e6cf13a02d6e21fd60d431f6b56aed4eb95a3ad867c38af5a6f3f2b0de99818a53f0ce39baa9aff6f049bb393127f503fbf93dde4d57a9172fd76c3810fa8d37f2915609da66b538e9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266131706Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/array_metadatas, Name: checkpoints/2624/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2624/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=12c5c690e62830f544e07814cb8d9ef12b988bfcc36a3061d68f580dc73c20f2dcf7ace7450079d06daee97bc20b16328c4550a80455d1241ac0650dd9b22c13009ad72626a928c7902acb949b4e15d9470e6181e8162dc699ad2aee1784ed02fc08d476bdaaee9d91cca8af9ae953e54b57431830291636e37ee27a03a4646774bef848a7504c50e25feaba4a1089510280f0f9fa6e47393b63d3f43bc174f8c37472358bacae266988d559bce6759b2e9a5127a9508973efeba1bd537b7b893d3f726e8740370c1ac0d040e44d1b3c7816fbbc39a81789a887a3b0622bab30603a376d587f3646f030fc37301776d7ea92cc68993899f87ed879765eb0cf59&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260331281Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default, Name: checkpoints/5056/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5056/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=5a020e5229b7818a5c797c4c89d868d330e520ffdb43252253afbdb8bec7a6a04ce413aceb9d77367e7c5e1278c1f3511fa06874e27c31aa7239705389bf1dfbe67bb57b6a3dcdf2d92aa52b7f2adb3cd144bbb327ebf60a8a04218f8832bd9bd119fa7f50a079cd75ee813a4207dd5800234bc6720e34d35be8e715128c7acb1ecf37652c193767755e4aec36c718889e1479f32235d551e81c48e0da98e3af06aba8037b43b1343e382f81b74d899aeecd8460d2db15bb4fc0da4697baff773f1ed0d2aa5ca277faf3033f7a92a62a46835de885a72dadac57104d56a1b6b07b8f4a108407bf575bb72bbbd6c3d02faf994be41e00201b0f963be8b38b1258&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260328698Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0, Name: checkpoints/1344/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1344/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=29355d1a746a825dcc08f9c05e61253bcfe734f944ef94a6b0f765bbc1b840697b11c3713bd755820e770e193c804509fb3a976ae6cfa578366360eb2dbf4fb34b917e7717920bdedaa19d04db2041e8c72b4b9144e013576b91e54fe2ab25c7687b51acbb1085d814d0996c0066aca3a059d9dc4bddec887f8e3b25c0b8f601bb8e3cf44986a534b769ba427880d0cf61332d8f1c9d5155f6d5ad827cd8fc4830b590b30d98250d3fb23372192817d0ba5d1a0e880feee88179a6fb38f9c0b432cb8e8f75131313c463d81a73d1cc4d8440d0f0b01438b20849af47b51fef7eb69e69849f9bfd3328245ce5ddda35604268d76c4646fa4679bee36539e6bf79&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260361575Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/array_metadatas, Name: checkpoints/2112/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2112/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=15e377f7cca65f302e2c9929c32b29a192f2dee62452586b588c7373ef60dff53ab66464b5ff73396bec42e55c08cb06322486c7d7356dff3ea83fb1dee8948852887008ac88ec70579310a5bbddcdaea6a5ecc9c2a93a53c94a1ec4db1e8c7600de20497051d19e88d351ad32095e7b606b84434841e44e5339baadc08fa072dfabdc9a51f96ccc22f6b9edb7c2c450443beedbc4fa2d58ad2a85ebafe33fc23a6ad9a1371d0ada2f5f2c4c55ee42f51053960ecf1c519adea04a7a6657c95892913513c75f372f8ec8e0e9e8f5015b04f1a6ca21177468f29a345dffbc098679ded8e978ae3ac3a6d587c97deb6fb85048765f16b417473ffc25d880276262&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260368561Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/d, Name: checkpoints/576/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/576/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=95eba329d4e423a817a609f8d79c14bde11135fb3ef8e6809e6c92d51599dbef70c6df5089a00fcc6d0d93284eaf4d08b5b7146d1e158c362e33108154bd8bbac0b1b50aafb8b34e0e58efe1ed9819e7625ebe73683d31f418c6d7d2e2561914153988bc5f11ac4832e8b912c7fbc876fdc9ea2433ef2089bc9867d8b456ab476972c7b7600620524a1937a5494478edc90d6eff530da6611511058fabe85ee061bc6b9a32a34319641db16a25b0b40f46913c4a5928dcaa05b362e347fd7010ef019f71d589a879fca5ffd7e18e9b0e4c68933d5a01c4bfa735065e2b983c92d162d32d2ebb32655f37c4824ac06bd2abb80724d02f2ffdda95d23dfe39de97&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260374491Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default/d, Name: checkpoints/2688/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2688/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=66a818d8873091ebf9186c0e3f523f6c48a4302d73064ae60fb6e49821fd538ec74790677178dd9ab3df4aed8a32b677a611718de6adf6a0113811d04947f3963664dd46a5533590b6e5e4033c54e69316ebcf0e9f44c08bcecb27f637049dd8caa8e24c144d04beb2e2b702fcd045f28b59596b80619ee60491686a6b7370aa33eaadda51581ec8701685ecf32ac981b21a9c2275733304cb0ac4923f76d9351423de1c2eb6a560d921a8e3a5605501e44f00dfb3a550d828bd256f0c39da807810e3ef52cdde6cec09e266ed5425fccb658988579f886ef83a74cd065ccf6b1405e1ceb529cabd9bcd0ddfc0d6a4265d77150a3f4ba8ce2d58fea599dd73c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260382007Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/d, Name: checkpoints/5952/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5952/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=5a0571ff110a881401a0fa9ad88fa513577327f4bd4d0147e59df87feeae6500e38a1cc3bbe55aede93753100b91ed8b977fa0ae1191f677e138a4dec560e694b5b3f8ee4b12e23a9769d0e4bda40c49470e49681868c6cdd868d73c1de299b1f01f7bee97d119d2d7793443fbdcecc6a7c3fd31abc925618293736e2cca44087721b10b8c349f22086093d6cdbe8e02bcb2566a9d781c9319433dd0bb5ed66bf2d8a12dadc34f8e9325e76af4c45164010e615dc1682197090366d5f9990b62216b211fc9075491438b48dcfe3e28c59979c89603b5d0e8412284afac8e8e79aa17fab586fe615d1f86353b79eb2a39ffc1ddb1dbaf7649618476a1da8aee38&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260395904Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0, Name: checkpoints/832/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/832/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=ac20ac84f2b68a4fb3bc7166fe9b5bcc378e6ebb854f4bc5be1ffe7820a037e4e27fe36c0fb6400ae0697bfb179de3fabfb92ed537e3a555567994ec1dc5c1eff5df104ef100664e22dbefc8e11365cbae6a62923c8afb99ca80196d57882ae91580c9e79349ffe6d46a1f43804865db97208afa95fe12391c8f04e2196853b16570b2f1583529b4ceab6e2cb15702f9a194e3474c4d21ecd3406b033da25743b09940486bdf97bc68720fb95155dedfde033b77ec46b77cb88c64c1f408636214e379395d3881ae9edbcf3d6facefa82ce7b28f79c6689f22e2abf7dc41274d9943f0492cf82661da9e93d97cfab9629bb04359660fec43979f1320f3da7751&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260412685Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0, Name: checkpoints/704/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/704/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=582a94491450ee4a1ddf640f794d747f826d76d2d53ea9db39c5ae86911f79d50cb96c4d0580a13aac84d27f8b9ca6db76eeedc5d5a80d052ba328c9c655dd31fef1498e543b7a5c2ef01ade1b7e0f84756f95aef85a68c04d289b3ffdfbdb18ca700aed5ae00301155ad96dae257155ec2509908b943fb188abb22bd7d6d55e45150d20114d0771c9cae1eb8ef90e9a91b9154652560d480eca3cc615cf64f3188cd21c49e337ad902547048a2757e5cb6785df26468b4c612c4cdc185eccbb440015e75c9e0f0b20e871b872bebafbe48ce7dfbe488b21f723d241d0bfa24c5c2d8ba36433b046a60500396ded9dac20fdf188f6ccb5968462c51befbc01fa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260414546Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/array_metadatas, Name: checkpoints/1472/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1472/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=1971fe9713ac4e4cd3fdf662b06e320e47ddfde277198ca731bd63af1e80c478a3da1ec618efb838c1625bfcdb8d6a25d3c786ed9fa4aa908b08a5794e529f5e524645adc9f8dc2791d2de93889d9903e5324d853bbaebcfaf027824884430b8cfefff4e21f8cd0a33e0633856fd9478e020400b8e60d1f526d5083702b276eef7a10111900afc5d499c11ab7d59a928ba39a4f08a3cb3974cf922eb89d7127fab766d29fc3a80ea706e1cd9915d04eb492dfdb7234e4e5ab0d466499daf5c4e90a1a6346a02d1bce86e0a17441d5869424e2766a10ec8cabb0446e4f927644109fb6548355a2139b7263ab1d998470da8148b1f5c6c6d6e959cccdaadcd2aa3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260423012Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200, Name: checkpoints/3200, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3200?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=8048feabf49e1867ed4b7ddfee8cd514712374695120d773e61e6e3a5fbd09cea9b5ba80d000bd823bdcbb861f82200673fbe9a5550d5f5bf34973e00520da50fb3da3326dc12afac5e55e4e7634543841e78419bae4e0621b578f2f9f79a6bf2e81f7b03638af6202b944bf5a1ba24d1adc6714e0f3147c3a64bdac67597c98cfa49385e88957ad38b20adacf9419cfdbafaba821945ce19350d887b3ba3d941cff364fb0a1746803be411fc41e6644288096cc86e096b55a54dc9f94244e91dde197a59e26ff60a0ebd9bc8bc4714368686ca1d3c240d4998e6f0d530d9a0d33c739b8fcec995d4abdaea9e7283b23ddcb7dc62384cf953f47f56e16551d36&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260424696Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/array_metadatas, Name: checkpoints/3968/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3968/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7d869a4171510cbc93fd9d0a1fb0d4fe7cf592efcc9f18dc56b588ee75130e4e904b1fe7c286de9d5a5dae129d4e47d2b3de9c390d35f99179a8b3d52325d0139a220531e98736b78d7e5103c056fc7f562e28bdccc5f630d44cbe9ef3eafc0e869786a5cfa470c1ef49c5a22e8191b5f2c78645c8ea30cba1489540d5cd78ab0e2010d3861306c7e29aa6a70702932c6b99ea68200488bafcce4e2be8d423f450570f2249891cc3f08aaad67320bf2e05003f178b4c9d74870d6be981552efb20c959c5b53cd22709cfec25db2eb96ff2e016297701b543d33a40727684d27b615c9d40128ac4771b966b19bd5a7ea438984bd5b5f0327ee047219b6b41c63d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260414013Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0, Name: checkpoints/512/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/512/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=048a74895015872ce66ee25916f13288e6cc422aa0526372f0526e29d3111e509ff689f9c4eddd06f7c0bc9ff5885e4fa216787243515efae9373cba4d8e1f7f1bfd165b153d1b0685265cf4d55874653e24a2ff4a131cf3eb3b546c18dbff0d2e2679b232e06571e4d78ed4c364816beadf91dbbb39669abaddacf57b6899b2d5c5905f6a94e1b88f24fc37f9a3306b23d858e4ebefaba72be0fca50319cbb3d5875495642e881dc2fe0dd9f4d96ebb8eaa126b4e32ef6b1a702f68893553bb9c96b44305c2e51168094a6c6a02ea3c8a85692b5e2679444d40b19e6530d2b1156e741b5b58e3d863200130f48a1ec6e28ee1fb391328bb7ebb4beb55fe8394&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26041454Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0, Name: checkpoints/3136/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3136/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=51f632a49f82a3dd378d3fe04744fa2c5880d9191afe1ef6073b8a188a53163b15152a74b799e145a4a669321185fb6f7c3b15c1a7e6664d78b2067e0521be1ecffdcef368b0c58c08617c8f0010ea192b0df674a541b266603faf61bb561a82b35ebe7663728333e4b2aa11e4cceb97a724c5bc704c0043449267c9cef0f5d6351b94325b8d57e80ee805e91e5e3fb721a723ab58ee1510b428dac067f0a4707f536dcf925aa20ddec79b47d3e6d24fd21bf025d36bc601611d4b6864b61c7dc12e23c7d76d63378cfd44205ffca52288fc0aba22c7e7a3cecf988ab63409c4916350721062475897e24ea73dc050c4b6f546a79e9150f9b11caaeed859049f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260450488Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304, Name: checkpoints/2304, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2304?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=63536101575df24e9e54d8feea1b77f50045362573040afbe3c470c5c5a09e0aff43340ae26320095aff9d56760a61e4f9f3b58aea4b311e7855b17c6c9708ed628922166177d1512926e2a858c656cecb995c15ee150288255018868baf9ba58e7d851612538d84f5b6deccf2e4c6f2963e4b1bcb08bc431fb338098526b82ecdde2c1c6ccd3bae5643417cfd10293b5205a3cd2077337e1e36869714d04e498f988a0829e151b097e0b243cdea34430b5ca3e21dd4f0aee4b74b693327e6affe8783f4b33410351eddc8de62f2d297c7102bc38b0a978d992f3a936de839226fd4a659d87371bcdc0ccb345ebae753b15995fb0117c23b1b616bd7398c8afa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260435617Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920, Name: checkpoints/1920, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1920?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=704c0284f5ff5b18094393a1e3da3ab6871581413d19041fbad2e56c15040f65cb6322c21c8f685d812a8d11d9ba887f74e264ec692cf0abfaa61f72d2e360361d0b862a8cb24598142f5f02d31083145e5d90e1939e1fe5df9058140f17be406785a6c098dd0cd23696b5e576814c4cce210bfa9ca46edf00027d4d668b8f5498affcc93b116c6df0e1a8f45342dee23b5b0ab4246853f086eea45926a0e0f63f3f169a566f1ca791410dd2f967fb060056913f78ecd938d125c5b3f429651103d6acea2488c9bdbd73244ff54fb08cccf7af489367f8780092d4827ae69b0b77a2b79c4a58af35772df5841009a8153a9ff8588ba5e523c7dd5cb0e8d97a6c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260451839Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0, Name: checkpoints/2880/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2880/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=964d80bafb1ac2422ced6f0e7d71736be2c71d1533847d0c98768f16f067e7261dcc663c4c20b3fa394650bc50b6c0be882b4acbcc8728a53f1a9fe1b7ab0ed1ca8dbfa057c1e2820612b1177ee1f56a3e803ba3250d59a4f0cc0a2a290f1a8321992d0f6943647b17d9e6a7fe97ccccd222fe05ac9bc04802d779c381c92739de68078871966528f31ea4ed5f95faf72268551cc90b052eedba6d0d25007e44ff58231d678f8fa1a9b16a527d0bb516d035582a87ac2e2d947be32634a4009c72f87e5b3e5f553bdabb6efed41bac960d0445e7f0d222c4f6f5d988d9f5459a0e29f8e64dd94a93168108ec741ab0a29294ce145bc373aeb66aa8755f8e3aeb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266338453Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default, Name: checkpoints/768/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/768/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7635bf9a805f1d4ff9ed22bb2cf38230ff743f887b44a0e541738f56bc25667d6f45580ca86ab20ece434807bd79546bb5346e5e0a8cefdccc2ad5fae82c22eedf4b0cb18b09c944e9eac32af488e352c309880f5426deb9be2a9845a9216680067d9b7205c3ad7dce71d50ae4e2b434e6ded2abdca5739e0a54cce382004db256e21c5625fb2d3d1f89a89da6a79681b209853d4b4175691d06a98c434f7de3bc2079b10948250b913e0647c8adee89406fd35301523ef290d18ded0d6b5794e4a594ecc41ecd1fb1ed1877813ec313ed103e8833312132b4aba13b0b3c7727e56f35f65e36e34823f1ba4f086a3173d352199e970a74ee0c817aefe6ca5106&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266361283Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0/d, Name: checkpoints/128/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/128/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=959599eeac2f5d21aea3d85553560bfabd4db5dabed9ddefcdc95aef743d1ed8378f17503845bda43f38c82cc6175f64a87247f396823d3f6d7f464e3a4434e5d0be3431176675762befa44d14d2f9458233fd5279551bc7c421b5d06f30debe9d902f0b3cd429aff4d145235a3eb9527139af91e7d45ee48bff3a9711836a5ffc9dcd7667606fe08189e64268bdeec9b9aa8cdf9d2505b79f3baf31d39630dd5237d47e1c3783f897c54f806fae42237c8c26a55789a7fccf80a30887a2812bdb49e2093dee1c5bea8c4aff79643f81223645c65b3cb369abb8740cc21ce112a922c2b4047a641fe8295bcd9435954cca688f46d0bf8bae0d498a0c068981d3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266390523Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/d, Name: checkpoints/5312/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5312/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=67525eccb14b175f6fc0f0a65be9db35f0846270f3218a290c0eb368ce748e456a2c82f807e7053cc66df74bd2364b29c7a867ee3f66d7179fe0cb36ace6761a8659df6841acc60c33543d0b849717fbb9245902c1f9599cee07c16469e172ac2692ef208692ebb0bacc7b572bb8fa33db9a214915fb24d67d6250e9c3b82fc57725e47e5af10b067d90c8fe3d414a5d313c8c3180b3f993c08f09959b0bc0af215b7f7f38ecde875c26ebe210c7d7c26fa01ed4a062b1c820f8b7f8dc4fe2aaafb18671d39748a0f5a081a97ca5b1b5d20b1647aa4c9b78c2e5a3897a5e45483ee3e30b82e0fc675385a3a2a4446029878a65399bb56cf3381a021e55e9bd9e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266404562Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/ocdbt.process_0/d, Name: checkpoints/5696/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5696/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=3528e8bfc54702b469a3cecdf5ccdc5a4cb573c35d9cb42bed1f26a18d71050d295cf0e0a80fa5bf256c1153bd21bc459d1f4f6f5ef9a225d867756bb1a83eb23085fb2525da3c5de7fff3a1203a0c2cad83af5dfe96186ab5b15c141313025aef3c0de23759711a97c52c7a4d1fd8fbe2806e67171b462f8f7ec4f912d4e2005acd9964e223cc517d34db8f8e3a07ab0a527b5156eeb4cce5afcf9d34a843fb11d41432528cd5a904bdc68bd2dc0437b7d7138f7a8670af01a0f26fd2110ba54617ec76b66dc6405ef7e5cb28f8a31684fb16b3bb930253866a1019110a52cdb03a1d311fda7bbce7fc8a3ab9db51201e9a14ebf0c7413c48f2058fe37adadd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266427944Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/array_metadatas, Name: checkpoints/4032/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4032/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=1ade7db629e57421475d9818f85a271676de1fdfa9e3b8190ea6896cef0a7328d09ec4bb2128eb3eafb0489833c2127a641301ee927e5c662a1712274edd197c1c3a807ac0a94a24e92e49595b1cd664b107423b9682d0a77d3b22f528df7b19561408e179ca3ed12f37f183f1b2d1561293fe453a42cea8e388fe463cb07a858a14bddc5b8416d611be0d9b6b9f24efa8dd56e4b33497f5d178ab019f2bc4c33a0e41d377be09d30962ae461f11a6e4a8edfbcbfa8076f00d227140154dfb2bc50a94ba462113d8d59a00a1b7727f1c66055b859c9f1b4257e37f71ee23bb71aa097066d5627f7f2b71aa9f2f272cd32db59e434647003e0cb995f7a33dcfcf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266438956Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default, Name: checkpoints/5376/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5376/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=9e490391084eadb9a547bad4aef4ad268ceb5e90465bb367006df644843548e950e99b8ba00f8f2e6b163255d12b67ca695ebbd4061f8b8b960918371f2e17a3d3e84784ae595ff46c0aea566c7e1a25f220874aaaf0eff12a4810c4d15863b1ca96d811be529798ce852567f4b75d584d0d2c5313000da899e67fb4c69dcdf1cd67dd36ec9f6e6229f19eeadf7ccff1b919701b31b7ca68258936dc9e83f7d063de44324d5aeb5c0273e7159ea2f32f7c074e2176e3541f639968c16734878f60e05b782ccc1f3722c8fb895f219f287df31d64e7c0945eb814b9d4d5ba9b047914f392f083200bc738f616d097c02ffad5726b7a4de42e0cbf4a39d27a0f36&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26646615Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default, Name: checkpoints/1024/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1024/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=82423be971c69eed9cb9fbaba749767c80758f4e24ea0527d90078b697642099ccbd5591254073ecbfdf3fd7c16929be5ba410e1ade07ac8bb6093af4d3ca6c31cbb56e3ce2cb21080aa9b84f161ee63264033626e6dbeaeaf3b6742e5c8bc3f4162cf2d5f9763ef70a7ea6abb77179a01673db1e5593f01f419908b0ceeab59e27883f896fbba6ffef9328241a0bff15268a40b4531ed54d9855dbaf194a86ba895086156ef35e3861edf9e8779336bb87a8a07fe873128a951996c688d03f01234050957f22751583502fd17d4c9a3b1395dce35998780ad789756e957a6b50a848235a5505046c8f1c05741c7eb9b233672b0e0cae5e2b22c27cf42c984d6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.2663941Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576, Name: checkpoints/576, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/576?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=12aef4161ac25bcc07b52ee8a7f179f3704309705d879656febff4786a2e35fb397c1eb2d8699fd79b4cd49bbf7e4e23511690398a17ae922a45ad5a5d3cf6351c26e933562d92431131aef4e5a33240d1bd30baada8c9979c2d4ed481d50cae27db58448628480f9754f2bacf15a87b92c52232b3aeb2071d28461e36e158fe26ddf64c7aad03257d81d20347072d6ab09f1db530f004e78bbc5454dc870f22065c18e9597a12ad74d61afe9e51897f4bf2c3cd3776e86a8c0ba10b465bb32ec545cc5eac21e09c2718c92530238d6d7519a2b19c4a4ae44d235134321a7aa44932d2bdc533397287fedb2de71d65eee49e7518292df13df1a5f976a3c68038&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266498639Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default, Name: checkpoints/2176/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2176/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=00f8c8989c5863c7dca3c2d06d654eeb3508d72cee7c8e1a104693cfa75f736ca26d5f5494bed4737e285a6bb86b8680d71d69e4ad8e2006449d5f23bb9844754772f5dafac1fefb85fdaddba57529066b3824dc005744a9c1fa87a3f249c7b0c3c6f6985925ddc7462169f8552a9a0dae81164da28b0a70501ba4197519227f5c363f898757f3ae13d80a8d49931bd83e4475c863a986671bc3c7cc80c3e15492b9681f3d698297cec574daf2f28d8047881ac3255aa3b429667c59b1269a4f896d91bf692bd00298d6a69806f4224764e6088caf97166b5fb225853bd2b5b3f5c02eee0183f05a1a08c2becb63966e92d281174af517d4d14d2bf628213678&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266524012Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0, Name: checkpoints/1664/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1664/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=be13550500b4b92bc22d2fa3d80d0d224c21721d0b5c56edbb2fdcb83b2c6283da147b9c79e5ff2b6cfc43a4c30658cd9d755f7da9cea47dfa5a841f265a494bec7ca3bd0216aed754c2c5bdf7fd02f92ac7e9ccf09b8f76a473ecc20a7bbc69282ead7c271a4d15e611b39daf96c2c503512277d41faa42c3ed813bc9f81ed890f9383bdb78df4a3905d1437aeb2f26fe1d76c126c6f33c6a61b5b31bd8ace68776d5f3fef7e9492374bef08bbb54f4c8139c0ebe08179daba6198bafab1ea7a4f7592050f5562c47b6e7665579591245cacef1120096947b1f21c0a139f8631db39151b8b92411ab6cd22431df339871226f42315d7ceef236829ae9398cd0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260447307Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default, Name: checkpoints/3648/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3648/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a681bda1ac87a7e5cedf55da609edbce7bd0704337a0258dd85098996390804da379716b8805e371a8b7c7e75d5abc4f129a93169d35f9a24833e84ef3919947f2d003004b19cf0e5c7abd0e3d85518dbd9e1c1f923b9c9afa111fdda76ae88a164d71fa652d27a688f83d25cd9100a2c4106ea42618436367e29890dda559ee0a96c0a1cb54e7f7010049c40eef00bcadc412058263bd8f393503c987657215a3c5d55797e16cca5071949d71b7885109b0bf6a9fbd2ce16c6e0bc81da1f9f94c893a16393e6f1398d70b2ccbc922343fa75531e7a7a1f8dc74cf2b06edfe6dee74eb1115127d0d7a060d900e652cbf87ebfcef7bf363c59295cb92b7b0c028&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260454974Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136, Name: checkpoints/3136, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3136?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=159402417b062ba3d4b5b34a390ecb10c055445deea4f06602b8788023eab743779cd608db1bf45f33ec01c49ed2a594bd442f8f9ef6429c561b01668f254b7a9f5ee02e8852cdb79657c24f282a4b166775f14d84d39788e41292683880eada518f707198497a059eb0b62468de326bf11f3c298c611b5a4b7421ae48a8bdc088dc9a9344bc814157b935ffc342de9ed6dae703e8be2f3e98dad2e9f560bc7c66f5db716d5172112adb55d5bbf1ff429dc03add5cc538af27c66ee40a2abe4051d980ee8d5629a6ed8bb7b09bafdf6622fdaaf4ea92ac1016155fe48449a623f7e2f5cd4ecb8e1545de1cd8975641a1608339d7361ea0b14f34b19031cd6cab&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266574734Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120, Name: checkpoints/5120, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5120?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=74b58a363ef94511a59d2c3d18deceeca40329d7dc9f6c2f199dc053cf56406a49506f2f8f610ebafb2106602cbb0a924c577c50eab0582b9a19e9d0d2b03d697e017307786e257254b185ae21cb6509c8e75193e72b81688b40f40c1cda4bc9c1e5c55ed0c9fa6dab426dbd6180e96f9e03d9d81018ea41ffef7365c0ecfda70228b87ba5073eac49d10e07c8f627568465ba91ccf9bc54c094c2e35d3d1f3d411585bb1732295d6897372fc8770545f0fecadf026d2754b6fac2cfeea6be0a5b34d45e8beed8bd634e4b85d8110f0f19273ba0f2b8770853fceaead271be6d929ee40510515b03f87b291bed5f76fce077de2e253e1542692f54c729239eb6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266586695Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default, Name: checkpoints/5952/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5952/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=790f3a69fe1c720268213d2df23269b6c716e5444fd7abcd923e7f7c6c174dec218cb7bddf888901bcbd91b1d93a08aa77016019ad38bff463a8022ab2d44ab475d22967eb1f1bc0aa462be341aed6750ccee0d41df58a9bf78a433cc90d1ef75f61fb461de09314acbb82814060738851cd6e6fb3f734a73d7fd04d03cede8e77f4242c0de246630133b9ab3de4dcc8347915a6df8ef947306523397c62306b23d31a15b84bafb67e46417710920ea15833669a484837938c9e43179ea3a34516cca4b8b0eb5a53427e28e8cad0ac77838cf8714b284167f6b440510334c989fc5ea572f324ef47dc2abb3bf5f8ad05714983731cd9e9ec79eeccd2bf78431f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266598249Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776, Name: checkpoints/3776, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3776?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=3efb08ce7acd548c365f450d130d8bb9d9ed48d82d6b1af6119189e8386ae5b38f2b95736647b1345bf6418dfa4e781fada980239c1938abd5bb8e84fc2c8093d4f1d346956c5b5a09ea98ea7594c8272fc60177c9eb570dec0d3a61971ff0d9f476df1a61fe2402dc0c563370a06ee8cca0cf78f70752d192c4efd992cc7bc39615ad0041da04eedcbb8d75beb1d3184ac054c7fb2366ecae776d5fd8aa859ce84188a4e41758aeb5e427f2304d4b0eb108d1016e792c122490ef12d5f02d6d44df405082d2f063bff2b61d6b90c59ded5e16c481a70db9b0a588d43fb51910ebaf938b853d8b45a871cc6db8cc0fd16e8e6f9849ad510660d1f53554cbd1da&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266641518Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0/d, Name: checkpoints/3264/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3264/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=c0610555cec44fe8d8384a66740a99b6d988285e75248090e684009dde719bd52eeedd604b963b93f6d0d793611c95258697b2a7bc167adaa3809220e01faca77b892160260685c3086991f65d06b6135e4966093dfc12a0d3b597b4f61692dabb498f559ace225976aaf07d556ddaca07ab98c834223f8af5fc9d08db642568385c1729cc61951bfa322cb7cbccd187171030c77f68e6e19a5a90afea104f786646fbb07e66480a742fc6c9aa96211e08ef68e27955e5325109d589555fcdecf1f17156410d3c788f84800a882b894a409649e3d9deaf5c69f0862a14c8454d37900438bb8a9d9399c94e5f7924c173e7ce6e9625af35d3cd6c1fa7de1177fd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266643105Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/ocdbt.process_0, Name: checkpoints/1984/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1984/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=99014b06d7ff03bb7148547f1b284ab7bdd09a6cb58346911a0f800068ac1735548e9e528839026ac00eeb2126e2eec096a65ca9b7a0f1cab6d97111004de8f68447d87e09569acc6da98b22fb9f818b5fd3ee1d55fe58ea3754d8142e0c22b78beb72825c674196a6a8cc5cb192a503ae4a1da513afd0fe1de564d4de8b42ac6c0a5545864f8c7656666559f8c6f9910f411acb880fb75d11946c79a4252dc01ba0b028f257f9313e8c93991750b806e4e88acce366d927e1998e1eae8c1037609ebc664c5bf69255145bad45a597c44115272d9589b114f9edf02183cff388c52315f10a5ac27dcc5a3c45e9f1d1fcd2c03e7d3b228a824282613e2b7709db&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26667467Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default, Name: checkpoints/192/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/192/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b110a665a8ea01f0bf5a7416536ac43625f136edb1598a779b272533fb81ee3dc005d953d5646aecaec1b74d44e78c2411d553d2171512ed034388a38e3e9ef304b5906ba64a1bd4ff8054e4fff080153c21d26ec3256d32524db01e843d4e315089ccfe83057a766ec57b36e4ef5569e21e2c1e9e881e27d8248e466624c93b7c0a553a7c206ca309830ac296c522158d975d35e725611cf40d6a1c9435804d5f97a04527f69a3277350c238a022cd0a022092077f6887893a28706f0a28917a1af49ac386f3c8cb7bf575b87ebc8879ca63c6e6f5e59e8942e1b7acf7f62d1ba22b10ed2b7ff9c213c215cd0682bd1f945f3cbfb4723e8cf2077e1e3f2375a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26669426Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/d, Name: checkpoints/2880/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2880/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=93fbcc06985c300b3eb5b66b0c7f6cf465e84c11c7c2bce34100459eb7c14576e5ed0223854b124cc90fb86f8c0bcea5be0448fca35f9a33a17e81b3ea830674ed9bf37177566c83904874ed223db2e9bb2b00e26d298532a1465e92df71f53d064aab3cd4e931937ddcf44ef24c97acc6a00a6bb5a1c88546658ec190d77a24698162c50ccdce70cc6b2859133a9e355d290e4d21808215d9eea12bf5d3a7efbe03f711cde144b6f0e0f3c5a5ae4738328cd831cd1083ccf92ce5aeeeeb8a2eb4198cbddd3332a2339d2b7b2f1077292cd800befc919298ef8973e1dafbeb1840ae84aa43f737c6fe84c2184e27c7c0ebf29818b4f7bc4bcaad5eed818450d8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266712578Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112, Name: checkpoints/2112, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2112?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=5a08d25ce7186ed3fb73a3abca71fed3be552a6502c902f908f05f3f1bb1ead5a11fb536b9c5b9a7d70344f35feaeeeaf9f42479def172734a762c031c7d65dff23a725ebf05c707e0b2998e9fde2c56f0bf9289b615ac4aeccf0d53a94819c8f611bf696538b74b192e41c07256cfa4f82fad3023b4d7f3983fb74f8f1f9918db90f1255837082ae11e625bdb170a47eba57e9a2e1439500e901b113620e5e3cc77fa5520c43925702ed063abd144c090c859d67616d28e53062d39bea395fe73423689bbf81838ba0c7e425f5f99328b4a64aed194793897deac72059f50e7dee82f53c5ff90f36508e5ea1eb5e75bae82f32af5bb54e9dbe29c08538cc7a5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266728025Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0/d, Name: checkpoints/4864/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4864/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=47c5dc45cb54f2bb28edbbab9f655d88acf76ceeb4d14a9f2568d74671f9a80d612c18148b3ffb79a1cecfbd427bc4a523d484497172f4e37a9f8db03cd68181f9a86016a50e3df6d5c3c3105fd3a36b4f64dc3a3bdc78a0cfe2c1549921014bd4d3bc17f16e99436f1ed0cb339c39ae3103c8df7c50a19d84a5ba1e865d9f977203b83da2e27c041a1cf6bfea1adfea7e3cb5cba114236cf606444902ff1930b330f87832ab07e890ff117a72db8c7d139c2a13cc51476e4318c46430011ee0c0c0e9835c37685fc366e28afb7596002f4219b88629288d2ddbd4bd86c038ad320c96c4896028d82809591c993297a7718a52715f3a12d16d6cb2219422645a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266587025Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/array_metadatas, Name: checkpoints/2304/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2304/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=948a62b01e998ab13a38e1a1124401a6127ef037dbdeadd1ca21ea39894cf9b88194f266e62323168a4792172cc4980d212f25f33f6c59ba896cb03d53362962fe8162efbfaca6ffab98e56a163c72ec4d76ad85c8594cf6b2887441d530943f5e2c13234b9951fed753f5b56ff1b026c385abcb17b26983a18e5ef71479d63bca3d185e5bb9fdc02a492c31a98f013a35d19aec7a7acaf4a7019ef5c1b2c996733a68ec65d4e3aeca04ef919a1143f4e8eeb4a6e8724c2f3b8c40e28f7dc9edcd5d10ecc18b6c87089e6cfbaa678c3e2de3f0c9811e1db7f06502ec745fe1cab2d80c0221cd4d40665d4be54ea92c6ece0cf7f1eea470123d83047840a4cab5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266773139Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0/d, Name: checkpoints/2240/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2240/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=08e5926991abc8c5ccbe43078d27998f316d5def8eac55aa5724060cacd2c8ba81ad998c01490263b806507b570ac6047f2a7a00b723b27b125abed57968c79303e43fd5c8ed072a15bbb9ef0a1173b086d3e72632d768c4902bb087d5d253ea6516e9c282250f703ce44848d9f87a37c7fd036cc4c37f556170d937d2194ecbeaf23931eb778b399d86b211a9722c87cebef61639a2f98cd7e21e69897f76e64f4856af69d35eee2ec1288f003d5a7cd8f316a990da6b0a3a300903555a07431b936c8b08e9dcaf08a79dde5b5b1d7f597ab1d5215476fc2d5030208cb1d789a568ab5d0db7743cad985d8e1d75b91056fd6fd9f7646d2ebb33dc7e34f55c50&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266771081Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default, Name: checkpoints/384/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/384/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b42d69965e299e5d4368acbed8c4d6b00f825a25fb9c8b9865d8b1521701d3f5b2d7bdf82f0bc289b982cfae4752da1101973f8e5f5b3788577be485ec2e40ce369f8d4e6d264b3abdcba3334cf0288143a2fa17899a0e2ee83ada608d9962ad695014b4bcd4605c43410cadad1cbf43974833025f1c4e3998db7788a771471aad7c71f4efb1f95d009bd2d2f45d3e52b4f0bce38090853af5971f802d3a445b0225d82a0107cff3525928676b38aae54bcedfcd9e01c011b200422f94f7f9b49e4894e2afbba5afbbf80bbff0ee33f33adbc66437a3580264288b99a8dc711246973dc281cac54e452507678d917ef3ad338831397f1bffd6f38e87c2689727&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266821253Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320, Name: checkpoints/320, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/320?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=55a739bd65e8e98577debb9d81cc7298f0a6ffaa4dcc71549c3db3bc5828b2b34c88762beef68cbd45d8a5e91ed145760ae13c4fc2c1e3999620c5a30c8f5fef4d523621acff4cc0d7db140322b67eeb8c29c00c825610c405d1124b12d0637ad00a1555dffcf8b8f3919689a137c426bce816d873e941425a20ccf8b58c6c7a22f896a62ccb1e61964b80d6f4f92d80ed84c40971d0363f834c60e22f6f7c972a46309939c799b5dfbea82abafdaa58b0f805220f726b69215fe8f467ecc0b7040e3165875f1b50f4285aabaa3ad931e97d3b4843c69df2b7bfeeaf305351861e19254bfc2209b9e9ea9496cc3bbea4e9458fe48526565480476fd86fa3b763&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266835808Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/d, Name: checkpoints/2112/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2112/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7f9354943eacdcb1d553ff3ea8f4cd54ba6bfdd12563de6cfc92626ec2ef65c57ee31fae362b2e9f3fb56743620d69b992824516996c7025fb1e5eea859f72da6b0b8993f76ed161380aa1c4dec389f617cef5e7cb7c1467dd85f81ed2ba1c741a9101adbc99cbbf93e560044e4ad8084a4b550818a3e7ab824e62112bc919849aaf5a2d8ba22c4fbeab25ec40ae6a2f9d469f2389898330cdbd7c9ded99a2f3248e1c90b3861089397d6c52a0a27201ed7fbda77947246b5b708af715d3869a05ac7c95adca46bf42336a46c83fe5665d4f96fbfc419fdb06618b87f55a9b31817736d3e94fb44aa80c465e5297c068436409b466acb7c8b1b296b9adf7d82e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26683875Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/array_metadatas, Name: checkpoints/3584/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3584/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=255b776330e7c4171d11e13342ecc2e14153b895257a8fc15ccaa1293f0ddcc15ac4050bc1fb3ecef3b12d9c602c558fa9b43041c686b93f91b1c1d01ae4f22659e067fdb672678e27e72f8983801aad5543271160da6debae366cec2e59972a5fb45f7d664300d0c1f0312eabbbe3a71ef25303be195af8965ec78a44683625c19baaa5174a0410a68602c2cc75be90a2f6d591bbcb918cdcf5fba30496fd6093fbd0fc1ffb15628c083c7a29d75688555bb6c5468b4a6178de29acf858adff37025898d8625b1400efafcc87fddff54dbe5b88468f226138c614d3e67734658fda0ca4e516b9f089ce2a2aebe900c379bb50d4b99849563159eafb6c536854&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266870265Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default, Name: checkpoints/1728/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1728/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=9fb6acc6b8206631040b0780a8507402c16a3440dc0e3436424634ab1880db0652a0c869ff0883717360838d3ffb98be74335f2f995a3a85f4392076e25b6e5ee4cf80700c9f611333b57f4befaa50f6fd3397badec8f2364adb6fb6290e96578ccca83cd7d3922b01cbe0e699c496af431cc8a99903eb782e05760325e18ef79b616342f94544effaa0c8aa57c52181b3a440a3a9bb83924f598406a8ca5491249ca4856a928185cec3b5e394e2dc9c284612359dce42e3aa3e3d9827215910428441e9d4162f3a34bddee4e78cadb05e43739d3bd046f0de0f01bfdae14e18cf41fb6dfc62a3f84fe3c5e42bd0e6b732319be97aea720061647ced402f416d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26689896Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0, Name: checkpoints/64/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/64/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a1f8d34b162513a51845e9b46b73c70cc5c5e73dff5d49338bfd197d8e97d6f8fe38b11c4479d0360ebbb13886d45e42fd6e97f4db3cbf5c9c64d89ea21d011a40ce0397eb4c477b8dc15162a28bcc63337eab0ce0713d50903aa2c5820e8849cac499e00a2227192c706c9886b34675dfdc2ac01e74998d5753f2dbe422a569f4cf598fb5312ebde1e8921fae72a113bf32956f80b8d844521e2ed8e1756a4ab9f1445241203a0e6327ee6189f996f54ecde18861b03ccece0895bb9e1e725271aae68b83420f4fe1789c48f1657675cffa59ba7bfc0c84356411364ee5c0a2c788b7e22b91594e4882dfa6476af0ce2188352084ccfded342de91c2895c185&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266901316Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096, Name: checkpoints/4096, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4096?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=2eff772af17e9c133f7a6f45aecfc7686a68c322ed78c59767bc514e71157b6ad3265b2659ecb6ed48ad406f027a5827e17dcc9d825036ac0280a3be874159c4b85094f8d024481bb147bf94d1cbfb66c688e78b8448301e79c994a28576177f012aa5b4b30b06552f95d14ad701233519405d071ed32312a80339e0c74cdede91afadf66ea0ce5524cf472abd7efb3c47fca73587e82f6519662654f8248cd06ee1836bee4059e2d29baf8a4f26ffcd4dfda1395d814c8061913b38d8ae35e118201b7e4f7c584ee640052625e0f1f9ea1764fe9b7c8f1b03ec5b5d621c8cbf3242853eb81dd404971ad33cc001dfe927dd6dd16028fc54beb629832f048b88&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266926177Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/array_metadatas, Name: checkpoints/704/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/704/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=05c1dbcc4337461aade249746f0aa71cafaebe2b264a26db074fb0c06f6deec0d0c17568c59dfccc7f5b18f3c310b04ddacc3da9ee19c2c1b3311980b080aa264cfaaad11c000c3e6687b41192bad984499c305266a7cf97d3a82123aae6586b88cc74cce91639b0a61642587602410a8ed7b4a00ce9c25e7929b6b379955986c71d7e662dea5464d33a4a1bf097c484c13e205fcc3c2d99a3946203fba26db3f0ee3bcc077ed474e1ec6f39d5d4f125b453f7c4a90fede3e6ed2e9cac5d8006a56a45813e20bf0d2ca5facbb68c44c14d97aa930ebd089d60744e74e759a216677e246c0910f9fec92bf40974a1cd8ef6199800b633aac578179180d6192335&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266950006Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024/default/ocdbt.process_0/d, Name: checkpoints/1024/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1024/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=8aa67a451901878f7c6918051f274d4865e06a4ede6cd93db187f004b00b70c103eb90b36762694cb1f7f1d62a4354b599eda1810f930e2472f707d3cea6d48917300af30c38366b17e5973b6ce74cdefa5ff99313d9b92abcd56c080fdefa1909c44374a31941f5f6e7ae86047af73c86d81f37dc66493ae5c6751aa4fbb18f1a9a2b53b4c8de6ecf16d0a7c597b2f517cf66ef0aa331c434906f6fbb5b81621efe13da9cfbb552177f7474230d17ed4b5937e6e41931f43f085cdb9abdf3581b7d0ac10e751fe0ec13adde9c0d3aef4af7b0db8e51d4c2767eb013661f964711f54b94fba3fe6153534b3f0f88804bd82073768b1f8e563b0de84d89424038&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266978538Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default, Name: checkpoints/5248/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5248/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=40d612dca3a74da44c7a841bf331a41e5aa004b3ea57573e78418ca9fe1e1c4b8fab462fc8f8cbe95f7cd113ea88a86fd33661f50f7943a31806bb1c95b5917393accfb97a08b269af82eac76ca1deea4fa4083afdc2e5f0c7fbaef8cd4e2b7fc0cad68dd217327377b51f378826ff9bc7aeb19b1b714eca74b8cc3d052a03de68464fafa0d1b58c134071440f2ca936aa6d469439c5876321135d2cc57604e20df91110b94e627a9342dff859909bc57c1a6ef27526d311136e54bb4a8864422c1951e98bfbd001cba67fca38a7d34bd2fc65e90645cec18f1ddde07c9aabfe9c62bcabb3ba87358dc06afd6ec7520ad42a9ef58f24b3c086e93a466c311de1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266996701Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default/d, Name: checkpoints/2048/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2048/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0af73129fd6b3c2b194e65ed15a7ee890cdab719e57d7c3fa0aeb0cec32ddce280c1f33fb9862a7aff03c484000769d654b786d8e84d9eb957a929b073da4cf3d5048613bbcf880d3ca2a2019268c1a2a0f580f79c6246ee3a585de2823bd50b17132fbb9067c0916669b6e3d333e4923198c84d426a040d79712cc6d103b66b340951da2a68d5d891587b041d3dc6df7bf1abc69f2334475a1967d17ba831225ea251a8cd1eef01e63a6be7871f438db7ffa6fab8df72618939e87602f0d76181475660d52210bb0cf16da89e1f2728fe1f321b0d675ebc0bba4b8066a3ce2c042f38eaa24911d788c6ca63be0b1d591402cac5aff1177274b0b21de8d6a638&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267004221Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/ocdbt.process_0, Name: checkpoints/1088/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1088/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b5ff5925a2cd65ae129f36e7e6cf4a21f859242aa381a3bf854341e849cf39261c142b3289a0ba0a835a83257bf3e179036f3dcf6f8f8fa5bf412f8bfa9491db82c773154850ca3c0a0071a81006bb1b031b945a098b0415895579d748ce213673f26214cc1944fb6e56e9cb0e53a7893d76ea91cc03c809e206ea76b1e26def1cafab17362a7c9ee919d479575c4bbc91b0aa94efa3eafc6ef5b9b5876cc0bc83b07021c0ac48a239b530e05b067f8cb4908128efb48082cb14a5ec36f6760dfdcfd63086ea7b91d92bac487eaca21da73a4d55b4c9a79dbb280ba98e1cba16b585d1d0a10e544c22acb66339ffc33c23a2ce2a815443cf49ea171b9db563a0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266676507Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/ocdbt.process_0, Name: checkpoints/5888/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5888/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6341d702d1658d34792764ac8d6c54b8a3187a4cfa61cd310d90555472c70b587d327a113b354ee154ab9fa1a5dc396b5769912c113ad9465fe7a862bb3c0211ec4b1b067d9d993f2efc842c74fd10fa97bd9319d4fb9a7d1755170e1a0fd30776745c8378c28b1013281ad6fa4ce1411027748a67879ae9c075daf0a9fa9a3043637301397b20b84b37410f2c81e1856fd91adaf78f0af680d7d88be3700f092b6502f730ddf902109563042852b4d689d403bf1dc5045a5bb865b82dc4faebe8a44da85d0139f43e95546b244e6291425f6f3b50176bc649c2e069cc491cf5b346404dd9d382f5b56c35f00463c0edeabaaa1cb85758d45871bc781f4977bf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267055826Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/d, Name: checkpoints/2432/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2432/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=5657da541bc903e019f1f6982c5e873852f6ee484cb211023f7737bbca6731f7c57ae3b919e2990107ea0b2207fd3342065b55ed426053037d7d33cb9abd7ab2fe1ed4ebcecf857f04f4877a80fb2d22dfa6b86a4718bbf17d7cec3cbbccf363d300005159294ac9b17d85f9c4f0f37a450a86080bb2b10b8f825a722c336b22dea6497e87548e80505a10ad4f2b43dfd22080b0399fc90ed909c98801b87c40692d4ce56d60acd6c8663792a22ca7940ead43a3c079734eed1fa9f417b6436dc0e5a0dcecb93ef0ceb7f2b39fe000fb5de5b01d7a61a8299e51b1fa205027fec39911c5f0d31c30485d1f23c30814f7979fd0ea35c2877652451c332aa72093&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267064289Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840, Name: checkpoints/3840, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3840?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=273f08d8670d93da7c9a7a65012691389385a3de6b5b515a90923a774a23d82e8b49fae64803fd9efc20a032b53f8467f87a09539075ee22f3636c369af63ec394eb1428024db97560ce9d2e4e0b7e62353a835ff2751effcf0b60c34c50b1e21aa0a9b2e0dcff483833150be020522414e42d98dd395b497cdd6245aaa46027025b24aaa2a6b1b4c11d6cda8b1dc743368d5a2bbf98190ce5ca4eb5cbdcd2b6f21b3e603a8428513765821163a9d753bcec73bf2499efbd2173b69e60cfd069cf27e417329dff4340c91097018266962dc44f50c0b7fcadcee01f6614984b46b60229e35f25ff0c0e4b71f456022d1a7b1f98391076547290c79ece10003d9e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267094738Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456, Name: checkpoints/3456, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3456?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=2d20f9b277598a1fa12fcc24e20732c7d345880096da9db7f81aa6701645b70ed862baa7386a5c4b6d05274dea5fc85ed067db9d7bcea2f95770314172c7c425310789613dde359c1fd70c722f12329140bca8cc72d235b8df28f23fe3e77b43a51a51d2c0ebcaf3f6ccf80da871af7ccd6565c6ed60c403c20b2402bce7ab2e15375b5a4f3557736a01df7ca02f3014d9690f50db4c258eb0906eefe5cf36da3ea2e679c105a927042d0111b1d6b500d59152e96b291b536b7b004264d0ebd2afe15bf92b7b5d4f0a6e4f3c9612e9191f5f0eec07ed1f986d51c7efe648b7a14a08ca03a64f5000378ef4e69e7a81c9bbc62049d5eb9a2be0d8bf79987f180a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267117155Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/d, Name: checkpoints/2368/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2368/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=c260d206b5d6e761aca8ff67e421b33c86dc44dce212c694cab3646fe8c4140cf711542687aeb0a2fbadee964b33c478b287fca55fd62fa43de7b74334d59e9c942e2148585f47ab8cd56da7037be2ee70126703ae8769e2366cddc8e65d2aeae281b880da94dabb39161745a5d09980a7e186b6f70c4e4e518c0ce90f826773a597f4c98348afd2f81f10acf04d80a6ecc9a62cacbf7f507a9b25c1c2aad9d565277832e6c83b430ee82f3e303d9e82c4d675de0753309252ccf00a10235af70d84ebf49cb679e5d8e205fb5bea33545c4b37b2a4284a6a8156b4757065e63d3719ab50908c22c1fe1fff1ecb9d8e23420180d216188b8c152534ca3c90dd08&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267135996Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/d, Name: checkpoints/1472/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1472/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=1ca4c309fb1941048e7e5308f19a5e977fd57deb3f06042c2f7fd40be7e63e3b91bc4d806b679f544817b829e37b2b46aaee6ac5baa244734d5f7ca6616e8e9f345f7bae2187e46c2961646c06d79d0e70828894cb3bfbaa1fde3057d232e5fbcb4b7e7817e24118cfda0d12a37dbf0ad41cfc8db93fb2382640a94bb7ccaccc42209f9ccfbf59d1cbe23f30b77089297383d2c8c70cdd46e57db4c72c3921bbe9f82d13e7e9a8703dffc34a4895b3262a3ea2e38c0864127346cf49bb865d1530298b861a737944992847cc3fd59c138a88e310f38f19568896fec19683d5cdbff2ff7e80494e6df4e6df85a3343264d7fc2461a373790979771c4b527ea56b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267126683Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/d, Name: checkpoints/2240/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2240/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=21021a2d2bd0d8dfd0d6902b3625dbaa3db261135bdbd02be54c00b9b366d52b565ae5bb275a45134d60d773213661ca0ba161a54bbf2d3e77d6321bd0d9720ac50257e5eb6b4cb9056b84b30dbeedff6e4b7ac1ceece58464e5d72826682260a5a442d8e134f9cd1e8479599a0bfe5d4863f01d9c0db093d7094ea353525a97c6be11d6b0a94b69d3c863020bef0b758f0afe79c939d1ddf4300d2f0a62b0b58040c666910671ddc45d759a06241c732ab9b226bebef4ad04683dfc83308dea4d46d4ba6851010f9356af30a1b7930aac5db42f676b911c3fbc4698b970e231c60923fce0b258c5a7d4c509a2be2529e3183261df7c371b1750c765f5caed67&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266575145Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0, Name: checkpoints/576/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/576/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b579161234d414c87bace04cf935e2f6d9e8d44092541995b70ce41716126b07690fcc7b40a38b23fabbc44de3eb888debf303fa5ae0f554a24b28fbd75cb172c2bb799cbefb00d08b495f343586958fbee05f6cb0e9dcaa293e51d3bf89cfd0756843dc5bd9bb59d00fe20ea142f0e7052579822b03c9f7ce2fc270d077001061f84e0c7330574c5f14335e3182283b784e3cdf70a6acf6075776d1c61a46cc2d319cf87198f119b2af388a527c9cb482f3894bba5eb4ade2c14d7ed625770a56fc6aed0e671fcfb073df16c33e59caa2781d7c9f0f69931ccd6e4b1be163bacd01755f50589ec79bf369efa06010c87edbd4b120d33803dc09144e24fad249&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267062127Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256, Name: checkpoints/256, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/256?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=88c30a63752920e5d7e10bdf7449915e693e88b7f874d393140f702dd03e1778a3239f03209790f822eadb7616843dc1284e4f596f0ecd92249b5a492ce4a782465b653aeecfc0057851fa11d2d2cd890cd047f18deb793291f3b21799eb13059d459b9f2a086ea96e13f7ee86879895752c8ec50ac8eea2e2e43397f23f6575e3cb8ccbff935207ecf6cf8617ebd6af37e71085244a14062ec78bebb6f39280621a678ac526330922ff4be5ba6cf49eb65a9ae1f94eb9068c3c44a9710e41353445baf2e055bbe284f38c069c4ddcf104da98d76266a0ba6b1082cf2a567e3d379cdf53c4042690af491c2bed89a672d5813d8733e60a23a1c5c7b2822e27ae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267186785Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/d, Name: checkpoints/3840/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3840/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=98cd4eab4791ab9e09dfdaee71b4dc5b87c7d9d0fd8e0b2b5d6785b6042636e2270eb31a131a30348f6322687dacceb1abf34bb07d9faa61fa2db149b2097678b61bf8ee7621b9494ed44319936a6bf612f15772531d18c2891c63d64087d067ea6bb10f5aba9d865fb074912346f1cc8a6e704d7933b9df0ea462b81caf3bab7a6fb2fca52050c762c7b133ddfa94888215270b3be9f2dd8f49016f3aed702bac8fe845d7f904ca51b739948f0c86da0794d46d96aaa78fcf4bc015f27fbb969e7fe99dce0338bf67511c9d4dac566097d6e32ab3e9fc8780fc9cb467e6ceb3f1eb8542a03890e76d5e15a45c225b1f1ac25f7a99710e3a1cfe8932017d9ac1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267242532Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496/default/ocdbt.process_0/d, Name: checkpoints/2496/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2496/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=4aa7da338f20cf2a3710b27dfdd86a27dcacdb60c11c4d1ae507958e855211ee38d41d10154f0d7c0b7bb849ba64ba7edd683d4dd741aec9b06e2f3212007f6bade292aa528260b408320065219aa498b6fc22c9ce1b07be114e27809144644f295bd499c0cc7aec92010b26ed843436b4200aaf3dc40960ed7e2e9a039925ac2676d4c7ecfe07ef0ce7f07d7dc08851fbbdf5fa237dc23450f9b6e7dff44bdfe80f7847e7fa47b21294ed46d020a82a8bddd46cc2475c81777e8075ec47336d39fda652bfdaf46d861c24b3367f468f357f992d16282d32bdeafb93a761f633d28f3f5ec94f43f45197b5cc8f6ebace4bdd9d913bde53ff63525e34b39c7a61&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267246088Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/ocdbt.process_0/d, Name: checkpoints/5184/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5184/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=3edbf6c80ebd8e9ca988e54ac1e3a95895645b4ac9e8a87c848aaa58d1ced5fc67ae99a4101f0836017ae448641d480beb07df4080a6ededce8be1affb09a91be48d0a0f51fea1a65e92eced73f9a202d297b052bfb7f0d939a707d236dc00b2f7d52848fe1e42f5579ce78a22136b3cb54cef1821a013c2f93e7c28282b3e1431c5733ee7b4936a6c6a49410324246bd1de1d88406c0db7c83d19b512a08274a2c6ab7056287e68395db186c9c85ade3507d402288c75c806f2b4900b006e9ebe73b9025d17f294602fdb8c7fdefc630f37583e23b2ca304ec17c4d7ae072c8a64d7b9e726e1af38a25c607750178a6332d09472dea3572239d1d3ed0f7f1f6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267268938Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default, Name: checkpoints/896/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/896/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=98120dbee363a0f98133b0e0dd112f3d32621465dc853e3a04624b8597d5830dafa1f3e09ad697c5ad5a0a1b0ca6e7d4723e83c0c02459935ab3fcdeef0f6746b8d0ba93e9b4c279eca68fe5082b54568eef6f54618457b3c066165a41b2cf1a9481b6d4c05bb7f0c54709b4281dd5502d397fa8b67c46ccb1af5a186721fad0528f1d79f9ce59dff8e58ff578d8c0e8207d8497bd7b485e9bcd91c865a75f3938e8856c30bc76ee3da03eef06ddab011cadc1105b57aadb93c830af2e771af54de9c2234ec82c478955fec3d2f70d114435cfb56a17d201082950079338a9b8c561671091f0865f27c44fedba50743a6278943028c08224d7fbb1db1bb0aebd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267140048Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0, Name: checkpoints/2944/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2944/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=559f1d2b62ec486ef00985f1acfd50e654febb04b873b48c1e11da53f7e21c4f1da7dcd788c6e14540c91814956569ed9011f1252ed107d2f11c2e7e3c7193c7bbc99cb16e449cbae36a7f6b6076873e613f6c15327ddadae0846a6564108630f692ee2099ea47fdf1329a316d30c8eb062958f2cdd0b211a282b01564463fc338d223392cb05527115a4195e6debb1fc2828fa0e36306c76ccf09a492a94cc8ab146dc77b9f603df3293674f84777e05eed5bd106e54234e4a82654bbaf3ba58de7d553bc649c2a9026236aad59deedfef27cd71340db9e930c96f2303c575aaee5a2f684177bfe54f026d2ead5742a149389229cb93474d52eb37873240e6c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267245765Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/array_metadatas, Name: checkpoints/384/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/384/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6c3d43a6372f80f45c6acc0d10865d1cd0ab39578c6cf966b8e5bc791fb840eb60b324ef62caedcc299757c3d310fdb031b2d2e65348da2cd2df46d87e1a5e26d6022a380a02aa09751b691b2124958370f0d431fd7284f583447690bfe8db97f3ac48a640fa913cf6886c9b14104ab825800119886746a5ed0a7ad7c96ad6281743727e093bf6d0d57b0ea6905f3fa8f8ce5e38245eedc00a299f4079ab8348bdbfbd6aaa53dd256f42af8f96857d7194dd787291a158bc8077a624dd953a9f6cc7ad95ddc0de5ae49df6526bb401ffab219d14104cb045e8b13d32cccabdfb90adc05bc99120d9fee576621c1a823ac855d6b8fd7d4662fae8d21fb411c5f3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266952996Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/array_metadatas, Name: checkpoints/6080/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/6080/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=66647c2034850c92e03aebcfc1a48e5b2a03ff4f1c1f1c4d9acf7526c54292eb2a7670fe5592e59ea205710e1161e24d1e7283ff4401c125a3f222322cdb8fdf985dcbab6d8c3c34e1304c2a4b2f376b264ba9dfa37f49696935546a433f853d3e5b6ac11544471d6deae3974688e3b04c7ecae334a8f86386e4f564c040bd9d108283f201d321a69bdc5ed23e7dfbeedd2d050d2af7131d586bc4a94775296db107bb10cade7cf8de8eb075cecc1d8ee7d9708e4d4cec5c881989d334dd9d8efa72678ea0040afce515ee3d40ff483954f4564f06200d93f37415380dde44148bc611c749060cfbe4bf5e6410ecb44ef917b78b822f60c4604a5adb8cced567&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267077708Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/d, Name: checkpoints/1600/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1600/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=78a824be50fed49a5a0d9ac98e38d0fc9d734b3c586a591833ff8b779f13ffd084dce929897bde47f47cb5f217fd738ebce0a98023c7160210f66d9eab1dee110f3166d7b763c4a5bc171b294e0be5411a9d557c0eea4eda44250b6cc3e5c0fe981c724e22455ef0299cf2eea22d7482ed4b6e5dd36f4862b5be7f9fe358dd29247018baa9a6170d0d85a0d52753b016143bebe0f2de49bd94b5ef1150fba7e4fc8b24ceebf715752951d9f08ecb45ea6a6f6a4b2de80ad40eb00d0062c44d7d548dc780dbf91a0ce0fc35bc539e1ac4bfd1d53ffcc24ad43859e5b988ed366eb6fd77503a6ee6a1a975dde56bb67e836b81f76ae63d3c7741f97009620a73d1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266912102Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default/ocdbt.process_0/d, Name: checkpoints/576/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/576/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=94e29359304a09d64825bd33997eeebd6b1a0ec441f48c9198efb0dfd74c72aaec5a680744725a7bf4bc192ed3fb56c2cdf6c8d1a07d8efedb165d31f9c97f048ba2235133cd61d5c14d789cd9db9db1530cedd1983bea7154d71cbc80716fb3ca1058ef3ec523cf72f2e6ce3d35555899633fa5b0325ffbc3e431e2cde51aa28eb9a5b679555cd2c8c5872d634d63eccc758758cdb6aa04677cd3657f9ca4b42d5dbbde4bef13d28ad72d63626c84fa88a5d52855b65965306a3b194035814d4092cba501f3c798a0b40eeccbb2dc1db3ffd098c825a0a5fc369a6de15248a34b525e49966bb7cb156c14eec5ca75ffcfe663d2119e226f10a989e97a0fc692&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266997699Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default, Name: checkpoints/6016/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/6016/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0ae85c1c73a01eb7565fc846eb23e78c0a1a7b2e5615d71ab1c79c3bfae4573ff179209e9831a728d9fb232c6f7077e108f2949b14148417544d4689162aed5e36027eb174ad65106be95721b289db47b0957a7c3b85b85ce6fc065e85494ab896a0fd969929b2b6f4387cdef41a97122bec7d506e3e7bb11d624b59f7f288ada6c4d33a892ea495a18f4380e09f2c888b008c747d45e0c6d39b356f1b7fde3a2c523292f627db69babaf63f5de16b879113e427de1736f5dc38dc3f2e1942b796b37865db5d7bb6958e19853fb44b2a6f0c6555fcb2914cfcce97104db748287891730583b8e5d0e4dacd8262b066bcb97f5089c60a6bebffdc6b89189924c0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267011152Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/array_metadatas, Name: checkpoints/512/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/512/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=95074127d9abc047101e17e265dac4bd3e23bffd90c240cdad8ba339e9fdeb179b2bdeb65276fa2e49d8fa4c4230593fa62f7e87069ca38809aae799feb0eec1dba055bc9a878a0f7a20e0ce99db5b10dd73c004667179825fbd57213db91439b5f630ed3199fe48ba797a8098b6fbe32ef003402d2adb1f60f9e7e69d1966452943314f84cbf6e5365b3a2db633a68aa23df53eee933139d9e940a02b5eab9d17289b6ffd3b06bb83485e4036f020d81bd58531c31a575977b7730019b2bfc3fc2841a89d9b5849150f6a9e51f8056eedf50738001478ec1f2a7bfd9a5dc74444c15cde6627c1ed8c5ff21868ce92dfe5a8836e32747d93b14e79d1472c1513&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266686668Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992, Name: checkpoints/4992, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4992?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=71f7ea07630d32e69faf273e1d3d7aa5077923bff02b1dbdac747fd3fe3d14a9684ec4645a999382c3301ba870a17134e566ff9e051f99a1f3d1755e3ed312f348bfe9c0495a2183bfa2d4a9eded89ff7fdacae7b32f68422845c4e850aa043670f52b05c653ee8b365aa224ba6b6aa34959481c4666049dfba09191dc6bfeca942eab1658b9a6c7ab5d555671d4cdc11ae13260b3bb39609bfc1b1b2571ad80e9eaaeb74ffc5c1fb652d4ca44cecba2b9c5ca4f1edef317dc62a2b7af78e69c65a337a05107b3f0d7591decb501274b644bd1ba828b4cc1396c0db51b4ec912c0680656aa46ccd93ae9202db200dafa363b03cf6eefc3e582225b6dda1c2f94&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266847972Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/ocdbt.process_0/d, Name: checkpoints/3968/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3968/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=bd714ddde46b01dd2b57aaad66e32d8325333b2cb2ffa6a023d0df2a2d32517f9a2c5639cc30b84fcdc1a2f71745f837f89dfcd62b7e4f465114f6ad3e6512500056edb0e2c0cf325d54b398a5ef18de5b2e9b57abfa49f6283e47005b41f3add3c81d732274713c4dac13c80a949bcc886ebdec6ad3ce8616523635054237c21f82cd2892edef6382dd03e571a95de7a9a52e3290cc67d774579b5bb1ab4295cff3269f1f06d945457134c20e77a1db597afb9652c2ac5f4d1ee48deea3eb0d8527034b78a09db6caf6cff91ecad59be1543584c7699946ff13156fef606934e1b4f95a40a9c5897e68d570eb5556678a525c90ed7ae427bb42292ac6685c40&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266882634Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/array_metadatas, Name: checkpoints/768/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/768/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=5ab08e1367143383277c0a900ec39fa6a279367a9ae4f32257827a0a4987e59f9cb04309335ef8a5adfafe0a166131ce6585b0fda7b8357298d7db98d1a793cb22e87e05ed071994fe2310ed0743bcb5a16d2dd52103ee16987d9654dfa9eaf417df9c83ff6da84c05887b62206c9fa7fe60f4194d1d8e1688f0d0a31aaeeb4304f8e5e579961328ea2a4cca4c98e1d66f797729285d84fb979ab691a1c37b42f8d5d0f536a9b3784ffa9c075920e0783156f2ced2c62a4b1cf6dd79aa03bf9ba70826fe84c7abf81d708abf7363d94be18179ac92f512af948a24a20011c2a260638b488917088cdc7a82a0cd248b67deef8543148bdfd014c6942274d9c754&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266780089Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0/d, Name: checkpoints/5312/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5312/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=9da2b4748e0f215b0a68b5685defa0a58ae407a0db5a555f666c947cefd6d12d71e43cc32b0869cca9d9026055c20a12f387678babc7909a8914d4489e8196f90e3902fbab5d843e18e5d6aa0b2ab1bb288f06ff77fa438dc872c1c0c8c2890ec46d8b4b4e40614cd32db42a7275b37a2a24a155fe0fc1b62efa04450a21b6e025fd7e9c7a65ab5a81af6048abee96fb2e627e6881ef30d8accbd615f7e876f59fc709527748c0c47ca5aad3b8d6b2b3fbe05bc96565afa2d2f0b94e74854f81830837be02f6cd0437dd5fcaa15c839b6b6d7d3daf35d529a72d4cb4e62335863339d89aeba94f2881e3237fafbf611c15119abbdef75994d142de313d2d8167&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266737475Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/ocdbt.process_0/d, Name: checkpoints/704/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/704/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6e3d9daf738f10c5845bb58f0e79102c4626080f1da4835dbeb5e707f0687201575384aaff606931fdf4646185f439cf8154a9eed76bbb37ce693512039aa31eeb8c2231d25ca622f9de19bfa6a337a24f32d7320ec7f449fa6e7a9e632ea222ea81870efbeecaa1cd6fb1d326ecda141f9530562b2f8d95a8ad5c23ce5f37b00f2ce618b96e4bdcd9de06f5c3d90bf82993b6a8d1011fef175cd98af47a5090b1d9ee7726a7dba6e654a44c22e66bba0e2e071f44a60adc8def0d9adf02cc5d0890dadd192415297d7050928c734eeed024309da7e7c0f5b6780b6196dbcc6b0ab38db670dda00819b997497c1802ebb5ff9a6d35d390f3ef85022a8fc0c693&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.2667859Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default, Name: checkpoints/3776/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3776/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=46790eaa51c726fcc5f528f518c9c7ad2c021b7c83d46e8323e33bfd538541959b28ff314339e00e7299d777593d5d4b2d4a780dbdd8634b99919e7c97fc3ac22e2580f5f1ef32f993d07d5e5a3febb32d5949b830981627079f55149795398c7b615ad7f4b7ce0329963f78f9ee9f8a7c08960cbe636328e85e6b8abae04b4b7f242f957f8716ec7132bbd479b522995a489e24a6534f115aa43a8b9155fcbd8b7f729adda693e0994e158f1dbb840ce2910f7c1df7e9dacbf66af25078d134b8f12cb6c1ee4b07151d387265b36ed7a38f815f2140715750894eff1b0d5b71e5d2ac4801f75446ebc72221bb5bc37294ef9fb16f4b13e5697cd2481d0b3000&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260461442Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952/default/ocdbt.process_0, Name: checkpoints/5952/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5952/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=066ba5bb6b88ac9c335cc4a605850400f00ba7ea1f5e282551e213b7664fcbace1ae19db8323733bdfb87a932b004733fd13853a92a8e1eec89b7cb6524e0b20165e2a0c3dcd386deccab3102743f442a98bbe67ba55730ea6c6afaab15b26c73cd506396d340545b72e9a8ba0825a47cb3f2a57f499765332f7cbcd0ee98eba6f0cb4307fe1f84e61de34470d8718951f76fb5a8c7dca89b73b33a0539ace1ee562d49535129c9fc12be8a581773cc91e7199cd40f64651a6bce3aac153aef6e9b53623e60bc5dd5eaf315f858e66a5248151a45ed7c7bf840534e18ed1f6fad9bc6e2435d374c594705134ce725506c659daca54cc50729c84b48c7b280091&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266427575Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/d, Name: checkpoints/448/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/448/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=bab20c0b466d0eb9ba5046e3634081f3dca53892dc3d9920fa4ee7a66d6c0a8469ab844672c10b2d9d89b540ff161149369786eabe8495461ae75ebab6bffd7a71b918424e0790bea3fc137ffa7c093a6fc01a56d9d1f2b7d9ab8c2fe8b7cb0f89564af989dfeeec86bffe18ac01c823787112e3c5768a875e624c05ef1b4f681b3720a4d527ce75e17020a44985c6b73e0ef81997bfd6ac6f5c7a6ea5edb19958b4b900b717756cece8a2b58734ac830a072b73335971e79ba020dea147d3d66c4276e2f205de52c55d6563486e1277fd42edb3832c2e495c478f513a412db51c7165b1b62a499e497b2b8c99195727e755195149e8960fbb855fae5ec9935c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266495904Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968, Name: checkpoints/3968, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3968?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=13bee5e8d4b35309b80007ac3ecbf017ad062f1a45d2bed9a758580b0e44ad7546b9a9d64875237585d18df6dff002af158f913c93970cfa6783d46330b37e197026674e27f0909e3f65568b39cecc396476d66a5dcdf6e970379d326b0ba05c34eabb9d3c66d195746f9a86fcc554f6e66762f3a65e18fb1014fbf4958b22ddfa0a6d69b7d3eb498c4eb87e90eea810c815990e30d99cdcbaacf9797b689c247890ef06a8a9d58390547361dce28fd5c789f467ff08808658a46f5dd2eb93d6dce565c6b322d57209c87b7af27d85cb5feb2ed43d82b7a70841b63704f0193345376e46597b9a1c1602405b1e79c939b485b97db12d7321169f28eb3a615a15&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266401294Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/array_metadatas, Name: checkpoints/1408/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1408/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=8be3e0c59176f578c0cf42283df83202107fc317127b66c939761870b2c3dcb5bece4be1e61045c276b5ae9e86fbfe01e92958ad09b783cc52d31d313d9518645fead6e0ec98c62db9fc644ce45afc57b2ca02b1dd55645b9accaf9e77f3385344fbf83ca11690335f948ebdade1c2fe49e3608484a8447f77115a0b6d8af8298d430e9912638f956c84c197b034557c4ff422c2b4b0faebbfed060ff21f595f667d952831fdbbb18147dd56d305285f7b35186a2a1c258c6a9a13f777630125a67eaae5d84ad3c173e42bb566ae520101b747ff3abd201edcc55bc4932861c506285d79717ac5071c8e26aff56d970cd80f493d7259557f86b36f8c64a2848b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260494201Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0/d, Name: checkpoints/192/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/192/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7fe50fe99c431ed1a56373ff0a0c099582bdbbe6a93f144dcbb85f0496647f3d7d8792ad38ddc0eccb6510da9c3189886db70c40996cd109a836f36a7e41665b087d7e340e447591dc8d30ccfb77de75492dec9365cd120068568dca20b12a6d5514b7a81b78607c7072cdc9dfbb66d3f68d8e8c8a7175d174ec1fb53369ecd7dbc9bb555ba87d54447152d9e2ad289faee708ec7103b10503dd7431b45f684172ff4feb3416cac9f1f54fd564a46daa59f405e92a9328d217c9872c12ca468d4e3e875b9ba994c386790e2fce33aed92b99bf674a17da0187e1557c9976cb7558e426463eaf4f8d2a8a2311a6b1ee78babec78c2fea3f47e66db2664bd3fb94&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260486173Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default, Name: checkpoints/4032/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4032/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=990c55999539d9b49dee3732680e003ecf37f02bc0648525137aa75928d4e99a6c92f44eeee6b2a597cb7690352c404f66c1feddce2162c6166f562e77cabc1be0fc55c4cbd270be34228f2c1292320d5400b7c6a604a301ed5660b901cc7a5a99b75720a8bdc8fd34d5d0135c696dc1b98182c60f3bd60479ea71c6465e06e3e9ef20c50e5ba42be37c5e0d272b35ccfbf13977efdc664081503d60eeeed4effaa00033b43093c56ddb977d2ea2c33835e59417fe63eeb8035e94a08b52a7ce07eb9baa750a3c283825782f2aa8cba26b2ccec1e33448af34dbb60a38917c8a502cfe9755fad329d71afd4be68da94a3ca0ef48995303f5870d2c3f67919096&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260500286Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/d, Name: checkpoints/4800/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4800/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=503c196ee32321727d7aeef009c7fd8f340965275525b55a22d671d0029f0b0bd7e34e2f9b8d381dc3fa48da0c2ead4c24a8deec59bf4072ab35337afd87c95be212a01fc037a69e0e963f8d42245d0d1721343b652782dc9e38d67d57ad051add3b2b6ed9e5752299b1ed55d648fdd6a3587ce30c36e22cc2d4f7b246a511e02f3e402e6fee503c865f8035f919549d7f85525adf61d24ce8f781184acfb4e9fbb1457ee0cc7dfaaf253ae2ec001bf2d14d6e1d8b86a23432a5b06da420335728fab09fd5514bf25ec2b4b6053a9e229aa17f4de7754fd5048dd1845ed8cf979328e5d92dea65d60fb02c0574c8191a72ddf2a708aeca9db7e741f0511270aa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260493056Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608, Name: checkpoints/4608, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4608?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=790dd482e52270552d58b5927df8c44c81f12089406a54f8b067e0b33b6c5f96a29975c222aeb17c81b24f0a0802c50a38bc3f5bab02a06b630f18654677f0fb2f447e8a2f713c4fb9932f899d7e1fbbbba084f0a8ab9026aeeb84d6a78ad8eb2d9e138e0c9bc0ada97b9aff2daa77c2db80a070ec950c43ad6c25df48390bf2885f0b4cd06d966bf5ca16a1287fe76186fb70188238d36d301101b6ab35eb1c199a276fce29753a48c27372abf6de93f9b16b8882596e203326a51110a778d6765580f7839f0feafceb095ea618794a4ef27ca8bbaf80597693db0dcb1e3adbd0870a52a2e282ce4f836dc7de4e7eb2f17b2754d20eeee4390a2c96b437d389&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260502107Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896, Name: checkpoints/896, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/896?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=bec5e6760bd97c3c0d46027b19a3e6dd6009595d471f573e2ceeb733d0f87b1d9965764d93a6eeb24da23d7344f22d0ee3fcea534e5f0722da2af07295c2aeafab11fd876c64cd06ff7084021e97878ca89be60f401e07ab94a074d6ac98da499926789b7af42c6371f92d5d4b29245662e6d19bf3558a720ec9d6eaabd48ad7dce79d32655c912ddbcba4f9172c7b33df6b592c1b385a12f7efe21e03d6f9ec66c65f81959fab2c365c71c2342f3dda19b054745b6dbaa4a151728a8fd8551178c62f4cf5bd8e3412398091d2ccc019d71d1b5348f11c1f19b2b5e251578d6048783f8ed6433494ed3ad2807a8f0c67e306a0bf0989518d32233850caef9891&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260504387Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default, Name: checkpoints/1344/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1344/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=2d9947487eb5243520e7aefcf5dcabc75af1d0931d32a7efaadebd8b07a0aaddef65ac9912be6f726f9b44cfac4e59d7ad85ffec9b85eab0c60713d8f7ded2a99a27a189d332462177db90445f9ee422b4b3519d3c99f22c67814bd28d9f317193eb2a6e763c021f4bf6cc42be753252d86093f5e778fc97da41fd5bdf3aa834aa03a620d52a8c042444d1e21260e933dc85eea7941e181c39ec0bfe5d1a0e9ddcdeebb4920e62200041834a0b3c2606c281e2376915dbbbc87fbfe2ebef68f44eb4711b502370cb2ffc2265dcddd6b2c2805780b2db37117bb2f7f3b80ddee1afa9787613c17a2882f5dd5e4018557653d483513cbfd9ae0ecd5b8f6aca7820&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260498683Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/d, Name: checkpoints/2624/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2624/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=06f6bf208d7588160485a0616700f3ab8dc1f385bae079a67b002cb50b24fde7063f8af191ce502aaa9e9e544f267300cab8e8711772d4f5ac1be3f6130377ad96d6cfa81a7904062f436c759b59b863decaeaad13751055688609c10ed29db5c212db5bf1762dcd71f26e899bcd627def562b0e9081c49d0c9992650e20d1163725b43301c57a91a9a5c5f279fe35ffac1e6ecade08e3efa37b777dc04f15b451ba747034ca264327b81ebd1505c4f611cbf0354934180b18f64cc9bb5f4050873127071e4d8db0c120f1064da1cab2e912e319317e288e914e59285244d65ecae23c441b3398b7e2ba2640d8aa2b9cf11174a8bd21d44cdcf9191ec6b3f180&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260504917Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/ocdbt.process_0/d, Name: checkpoints/2752/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2752/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=9596d8da4a29bdf3ab0335d65f39df6c1854d8a10d90419897337f1c26f1563f48036e9726f21f73305cdb2d36dc230eebfcc78f47e049c4a72f3bbd7549dc7fc6677270d2f0afd181c708d64d40d57f77391e181c99ab8de0d4fe26053f363c9bf0a731defd3fe1c780150d6680903b59bab9f8cf042a3800000d30d71f9cd864624c638453687cc6037b2fb3da099a0b4baf0a245987f6e354ca501a90f29a06a94c8abb9aa8d69a56091ec4d870cf67f8f0d6d8a2c06496a08332bdc27d711410b9412b6088c97cae54938e83b32504a10b22aa93f22bcc74ab1d003b6d78d9b9505c4a13863f9591fe2c79ae12870d67088e870f366f7714c9bc8f1dd898&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260514844Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536, Name: checkpoints/1536, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1536?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0049c0c5321596eb42de10698f7afbb682f04666e11d1334e2bbbf63876725011077ade8c907db1a6b390a6a846aceca4e3bc215aff7d03e331501f50981e098042c2e7bf4348c23f0ab36d98a1fe79ce8882de4831ead440b08ad1f853b446839316f36c4c27578fea32ed3b0588264a5cf2e6b2f0562a574a8872d2acdd17f3e936c988faaff986fa42d1c18dece86c63ba8d01acd3af93bd547463d10fd0a29615daf3779fb8dff89246476c46636a3e13113471b04fe3772d72b88d4700b3051fe36659c698aac4b43df81460e372554e29d7fed344730bdf28b5aec51d43fe76d9e74bc76f3408aaf87a8c30020fb3b5535d818110d2aaf2567322afade&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260513856Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/ocdbt.process_0, Name: checkpoints/768/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/768/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=1a28ea23d7ef94d3a0bc2f156fe5ef48946624cc6907adad94e25b4e6dfb2a240f8b8f515cb4a4aa1a04b6622174b32865eccdd0cdc5e85d4f969dda8ccf3028aef3fa9c7ad68d2555a63269cfcabc7cb6387717d0805ca455660a0ec1c2aa90b1f99c6e31c5846cdc12d91130cf3d96e8ca442ac7679ea4fdf4784c3d06412448eb08fa414dc733f07db7f443b4134fe7767a0d1fdd783f4465ddaa27a4aba967604e349423e71c7dd849f1ab9ad66843ff43f6ff315a46b885d21c3a693bf4e028ff18d4b6ecf0539e6d7c775165887d22ec9cce1c117c4f6176548c6e6494aeb9a59c4484dbbbb3b88da27420f11735dac86e5aa09a8fc82b901c69cc0fa4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260575273Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864, Name: checkpoints/4864, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4864?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=5524d1853e61132cdd0a20d300f7a823be63f46892139c8a5f4d0fe3f208bd731d0274378405b1758e6084bffc331c1ce240ad10c635f1bc3fc33536a04a5c4faebceb78d992528b968f74a9a9af48471f4c6e450fe8c2e98171302b1bb612860f274553f914fd051552430d6891f5943032e9c7ec9e8619115867edb508e571c258cd23afffa3e49ffdc320f816e595781eb092df5d745f71b2d203ef7229130c38e21e5973c9b4c009aa76c0a80bd6cf9f213a6590f158fbd7e6e34c89cf921bccc6a59a439a1b198243911ed6df92686efd299d76206b3bf3b2c11c18a98bc5507edc493d9cd5b539929c4d06cc2f43ffc400e6a8f12e4795607035e87e62&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260614753Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0, Name: checkpoints/3776/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3776/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=9a0da3e40ecfc73e6efd60aba8988dd67d2edeb0672ef3ad4f2955f2b5e53bbc5bd4a99071eadabb867acc481af0765ccf6e66e05bf22458abbdebdc64f6fd5fcaef45ea21f6973c7c8828246b5b0150a854e12d3ee4fd5fd51a61b693dbe8a53a7f45b060d885ea53f68f7b789b5eede77bdabf70995dd462f0f905c57193db137bb96f9a0695cfd3a08991cba4fd3736da54954818b5b28966ec0e17e40977bfc8d2d7d8e7d6d972d3cc27f230637e69daa99f3dafe29316ee076851cff15be30b64f4f2a0ef7741a1c0268dc2fb646f1a7f999877b11f394a891994d05bd42145d884d45f42ae6d56a043f8a161d56ddb2154f5a91fe48e6fea963328ef93&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260613559Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0, Name: checkpoints/3712/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3712/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=17400901eccc88fc272d194778b90b449711d4eada340cd44f6afc273c91e90b8e34988c16ce787147522b349e6993e27ca5f27f14ca527153f3147268177395d1bae6b9cfdba7c4833233caa8f5ab194d053e7be0913b8289c28328f389e9068d392f81034a55a20cc327455097e75db5a7811c8add45cc22836ed61ef46a22f04b6e32e4f0e0acfb08b62fa76922b5f943b835be4446856d547c14379aa8ba92930bb118c0f213e658946c6226e443742ec12451aef7c8fc31edb7aa4269b150d25a4252e6de13371b67e6b7db84a1fa6cde052212b670a317289b7d80ebd81d67274e4ad468567f847175eb4ed689620c798b55f571e785d0fecc879204b7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261162105Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0, Name: checkpoints/5440/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5440/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=648237eee16a1d3ec09bcf69cc52335d70194b28b07f69609b414f5860ae5e7edfd345e76536d880d9eb5dd942f7535b2f1ebedb4da34cdbb226470e1181ff7713a74de8460b00a662b1e8a564e4b9b0c4b15cb6296ad8dac4c01a0883b20de724ffdfae36f9063dfdce380003ca241deb476704ec37a66431ad0a3708c55200f3fd0a704be1d3d1eb96d8b9d6f5160c292a00af0f7821e53f3b1a058348f96a7cd0718e12393b583b58667e0a29914ba9254cefc85402fdaffc0d923bf6755c5e18d066c6302407fdfd8487612fa84a0656fa2e748dc6887e3fb6b2ad5671e98955a30b0816ca6ccbe7cb7f75e4b20576017fecfdc718420b0e3169ceee9663&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260604826Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328, Name: checkpoints/3328, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3328?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6b46de01c59e20900ba7b6a811937754915266770f0ca42a4e9ae306980a97255dd3d7a9f845985ec14a6904dd7a1b94eb87c778248d03b171857363fcdfbdac4e9f65324d073ea5c8440dd327da5a29a4db99ff44174eb2ab06e86dec8b8ac4b43d703f4f9183d282ab66f7925285bd9a662d549ce9e8f6a82fbee123a059d87c879bddea27a8a0db6d50b35862a44da4df2d159a3a3f365ea754af9af75640f9eb578b1709a0a7df42174007c852cf785a4efd280674eed5ad5caf4c526057d4109af2e4787d0c774be51c168ee52180d94d18674e5b8999a466673cd75cc030bd134aa79f552da4c730ac0ab669b420f5453643756b8908ef3cd7cead5c1b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260236293Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/ocdbt.process_0, Name: checkpoints/5504/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5504/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=318a265deb3ba1227efef31491f4d05c6d68d53fc5ae906bea406c1f2f80a3dc90a10ef0a37dd1db1b62299041b6f8f94b8666cd9aff8a63a4b33b012cb4bccc2cc9ef9484e0efd582ac6c841ae67807e99037e223985d588893fcc25a1586bf8c3e83970144d39b27bad8c632ed4dd900e8be9321d3cca6e1265a99664ea5c802955a9bf81a444103866e8137bef73ad5a30293a914b61ae05997912d7de81c481ae3063c291eda7eae3c741cec87cb0831ccb4e96f7e17b88ecff2e0f5b637f6d9324964b5ec68ab4f1c2b66d490152a2bafcdf37375e0565445d1f9546aaa51b3a70493206e4e6d3f8335176c41bd52f92d9257fc5fbcf10003e186992541&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261268545Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/ocdbt.process_0, Name: checkpoints/192/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/192/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7b81db5af371f4bf760449f98fc4e532d4cb50e0f240faa2c8a6d6c7e6f789df52f9dbc268a3219e303caa2957c0dfa0c6e7a63583dc8ea6f95bef13edd89512918e8198c9c725a3a274547d530fe9ac7437e189520191d64f67a5029c5746b3ea5fea5e0af28c2ce27205a70c186d807e3acaaa02dfd8d31c16283c66c96da0e9b547a8951ceb78f99d0a5158adb92e1826e0b0251816badc3dd47d59e382b54ce779992eb0b935f0ddd6cd63441aee45d61e228ebbc58dfafa2d9bb8a54c5c5abb82dc4f507ad9b65a6a208c2e52d5dd036cbdac73d4eacc9d583e2f29433cc513f661be379da062961c2b63fc02a8df28728c10cd877610df29bf55d49c0e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261303198Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0, Name: checkpoints/3520/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3520/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=22910d79998e3a85fa9022f92042caa5dc2f6e852ffecfc26c1184b5f37d5ae33c9dee7d3e3eeefc30ad22ce3a21ca561af35bb7ddab95f93f430a98ba9d906e93f312ee3db13bbc6038140440a21d69068e7b0c99a26899969b1f97287fef984a4ec4a6be8e40f2a24c469000c281c3ca610fa5e0bbd841ca06bfc8271aea5e71655e214bfe08190365262c1a08f52fad55789e025a4846011a73fffd5a87cb1fa3891c7e7bde6c43f37d224e86584eba052046450375285318ba8cbe61a2d55ae876bd39c1b73e543d0e675ea91d25cba7439cb8decb502c9c9cdabf37fc39a3a8f40ec8ead9e44e21a419def0365bc24707c2106f5e307ca65c25313380d6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261331636Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0/d, Name: checkpoints/4800/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4800/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7341aff584486913141a0a4203f18b20e91a926eda933ae531b3245ebbf3a97d0a2377e73caf6ca9dd4b0ffb452b66d00c7f12d4145542d16602e7399bfb119217e6cc4fc1e68f2095a6e9d71265b04f61bab2482644ae47d0193d8a7bd39677fa62488a6e5b7047e4795490957e836b4c894951848c44c701e89c99a630b5af50abb6a533e6d5f2dcdf5444ce227c0a5007e5d01876e5974292d67acbc81c4ae301089fbe5d449fc373a143bbeffc0e5bb4fab2973ff6a9b0f4b9bc0692968a1f7229a608b0cb88ad6fbb92c60d48fb71f323b454420a863ade5799c4d42d54ca571caf0df442194194f29a99c14aa5ee4072479080bab71bbf262187d79fa0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261363622Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default, Name: checkpoints/5568/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5568/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=01e6a6592d66e115adf7f00412aecf5a9616b36b2c18c39776717b756cf314931b665692f87ae91a352fde7fc4fd20c76e3dfe0c083bea42e4dcbd605cc8f819c4cee6721abd7750845676ff90dac8584eb41bd759a04f2ec73283ceaafda5d7816dbc8997b5425d1c2442cdb8377668d3fc08dfcc5996e2d1f7bf3af30635236482ed61fa1a95a95a382d4f63b1bb7c942d103829e6ca434cbc130cb90b0f8f68d440f6f1e3dd2794bf66a6bbf8a336bf901e272c12db70b143fe69c8bad4bda4ae7a219d5fa7afc8fe6d5008dd3474d7e78feeaf4cc4e5b857624883bcf13f60d0525873029c165c530c00001b7cf66113c86ec503f02497f217d51e7c4991&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261428325Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default, Name: checkpoints/2944/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2944/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=3289ab410efbfaf4853d3367c2388dd5389ee4cbbc2196eec5bc9693d90e4248ac005cb68661ae447f80a0927338789d14ba84b38b2bbc88ae4099dced89117e2e56efb0d1487abb5dff314a0a858da70e48745369ed2eec4282a08497261769804e740393540a7c190ae132a07886de2701874c0f62f67d28563974f89ba50e9df0c18fc928ed19c0911c90990a8013c8a7cddd81e53d023bcae7520b972b47b8cd967f41b67f535eb7b5854868fdd302114c007c349f9d85b838b3dd2c228439464ea6d238ab4f6c5071cc14f7af29a8189e432cee5ba9048527e19d1ae49bafa20dd3b4e918d1cf8c5609b655aa58cdf4483e14eae54982f00f6577d488fa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260487383Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/array_metadatas, Name: checkpoints/1152/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1152/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=548eeac848a8ad2441009631e87c65cc17b18c534d4c5ac241b1a918bc388dea56452603d400c7d4f983486e975b42150a00704117a28874bd03d95ddc44750b00e82611b741b3b795a423ac6067fddcc80ab1c061078024ddc9b7b9bdac6b46a633dd4d29c556ca48fcdd6323a80074f6b8a851d58912f82dbc9e0e2fd45823287599c3db17872d5c62c0664198e4764919bd88b17131f2129b992a66151c85dba07865e0eeaa96d67b940fe57345a67cacf6c85c7da7a11cd87b1e7a96ee8bd16b543d5ffbaf15f86e30d03dda97f5b2cb90268741bc1fb1de26bc9704bf17240b65dcc1d9cab81015861ddf1d875f381ff8c4ddd983cf77b43d35857ba81a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261475777Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240/default/ocdbt.process_0, Name: checkpoints/2240/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2240/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=9985f2c380808a3b2fe55f40857640d2f038f55bec114bf8bfc15385312d8f1fc9c7322dbc100a77fc7512dd11bb12f821b7efccbacbcd61124eb60b737a4d16b61ab3c1470fc8031c2dd0e42809c328de5ea58bf361e0a4a41c59fff12e989efd5d21385209d86f3d979feed586552054d719e909167e65e51d594284de535bec5dd0f32821e4c758929951440933e7f1728d6c8a382c47eb20771b40b4cb09e2184f8e4409566cf8006aed9a436f34919d28bf57ffa7d108d82d923dc5b76d669d5bc901f1c4fac52ad208f9d8746c0d276dad585ad1cc5c8b692bb65b9f2fa0eb0eb6a49eefe5bd104039d728db735b9230f55d546311663a9990804add6c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261464375Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0/d, Name: checkpoints/4480/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4480/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=29caa61f690ee31c3674ac63e516a6d6a47d124bcb16101556023666497d44dfc15a5d28aaf77ca1266823edb0cb8cff87873b5e11ae5100870eb36dd9e72090301a85effd16066e73922b8ff1f94be3510ffd063bbb83b40718976e5dda6c3eb4a1dc7af0582dc7aadd0fe818d2d488a018c1cdc8243cc930b6ce10b2701584633679435f33f30d8c05bdd0a2240e2c062d965164671a5e8d6085480387f3582d2da75b771bb8d556500aa4ba0bf9c52de28aa5006bdea8b0e0cf38c90cf11aba16bd5e178b57a71ce472d896409d531804f2d174ed726e4f9271ecc8553ada9c3015d486ae8f0fb9b56c434af2d8709de044a77bc90d9f686975deb8f54473&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261479361Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448, Name: checkpoints/448, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/448?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=1f08cbba622a973d280180185aba051601a57823b170d93fc94ed7ba91b25086787273edf8799dff229314917c2c005c4eccfcfd8a1a5410efc5161f11ee63f70e05faaef8eec5445df5f236f67160bf4ce3484b6a2387f0b6033ac44c94250d05d7718d43eee190182b51827902868dca15d78b975cb8d9f0e4f92eb50ba84f530a53f9293b373d7a3f9e3da7ed8b900ca3b1ba2cdd7e5e502870a8208b42404a0b3ad2530db130e9e602fe05972057fe517ecf58d3d34da1f4738226240eca33ec5883ab1580fcd556ac8b5cf0cd398c46fd890bae721ff6fad726dd191089195aa1cdf73f28278cb6e0b69d389e7769b3337079b895540ffe11b2acec1fce&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261493073Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048, Name: checkpoints/2048, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2048?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=176502171c6bc366928fb22dbdad16eeba60775637a9637307ecdea8e9876d5dff79ed60dda06d2b22b1baf314ceb72a303f08c7a5b6d0f5ba27654ff034fbee9991ff82877ff9333e1123ef406a907abefe56b0f3e406055a93b0f7af7c7279ad702daf0a5dfdfac5e3d7573ebc5f7b09665f0bba9931ccd8f4e572a477da69554216bdb7b290cd7696f217e5788bb285f25d6c4b68478cd7bf07f27d733355aa54afa61cdcb07f986e2d4ea00cf8e8c6473991dfaea430519d45063217425e3bd58309eb67dab6546a0b7938cc6e68da95bf18c1512b08958c0556a957dce36aac9be312cab55dfa6e98282ee4e1d96adec9cae81834161a349e35654f4552&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261525878Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280, Name: checkpoints/1280, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1280?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=22258e2e5fff9e9e5fa8290000b48cc39be56e15e154f53110da328121ace1291e9f5cb7573df431214e56b2ccaf305859a84f310e12c507a16d5ab36255272de9c1933e9ddfd320f99c3de15ff789975c9b8613e84bc5b23dabf967d808a68241abcafe8b4716405a59a07c71579bc50dbd994cd569ecc8a431fac8095707560a26de2abbbbac02f2cc50664dcc921cc1ffca8f9b52fbeef14b058a12c26dee28a329cd4ea2230f5d5149807dc1cfe900009ab755c98cf6603b05f4decec374c84929f85a500fbfa2e16151f6c2afbe53d7501f2dd5a53d23f29e754a257ffdcb3f893f0f51516afd905e6043192b5a62b763d1f5e97899b4615453ac4dc8c0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261527588Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default, Name: checkpoints/5184/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5184/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=30c7bba01e250caddaf8d9004e8ae7c6db5b41be2f7af833bbdf0b0244d7f4b3cfbf2a7f914b364f942f165ac52d5e6a08b1a55e9cdd51a42d363882ebe0d5194c288525e3e969ca17ca3a81c031ed8390c049eee9a570719b3d17cf91f5aa97198bd8fbd8cc083e477e7f06d4155abed1c309a50493d465c3e9cf14a5e98d7628b36f78cf30db327df15a8274cb8d9f1382f5a3278a84688dbce284c6eed7b3d313a4895dbc6020742cf7d4e60efbd14df251416a9996c6152d7659face3d9cd673708893e14edc8f1eff2918b48bf7bd1c27cfe8f7b169e4862d81e884589a9b97bc7ac1883f5a3c74561042ad156d35b04f01ecda030efe214cdcd57909d1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261529536Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/array_metadatas, Name: checkpoints/1216/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1216/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=499f473a9dca492ef25983fe56c637a22964ac9dc34afa6896faf03dcd85743db19690663d2084a585abe78829e0cfd193cb98061d4bab12eeaeabf07b0772133e2f9d1125bb2ff0a55b38e9934dcac3da7fb4874181aa253425a30b113529e8de8b5868c81430c82ebe0ca257544ce31523560baa0920b1b49072066333427634c3e34466eaf460bca8538f1538eee272353920e996d3fe5cdf5a01c17f40b3e5a7faa403554aacd948f09bd0e782da97b6cecc045d7de796e4fbcdd9e9f1e633e7fa0c86c851cec988c9f3f0ae04d8edb55cd47c24570365828c8f6baa0b97c1738394b939021a22999ca31946ff289bd03b8e5fd9c9cb155ece573cd3a9af&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261560246Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/d, Name: checkpoints/3712/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3712/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=215474ba3d43586810e951efdacac16a2a02e516603007ed10e2f327156edcb9cb455862e61fdb485745f3d298b852acf8b0eeebfc6dbd620a44127cc05f11b71d7db60d71cb263f1ba03330ce39d6451fdea0f796a53861fb8348b733f86292b78be437b333ebed2646347ad2c58848f126a3f91ef0ffd3f92845988f0bfdc36d3e202b60b96b5e0d584729b34e084594c76eec556952118dfcc6609f7130ef0a846911ed33ddc5a0b2999725aba1849d78f6a20c147421621e04fc5ecdd924728a19657ca6e2a940c71baef108559226cb30a9d17c3a67f70b59f5f9f8cbc6f22f16e9b12b6811bb2d5015ecec5cf68fb001b13572c74e7468032b0aee7441&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261564892Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/array_metadatas, Name: checkpoints/5120/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5120/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=85b3a627b3de8609a5efbc4dfb34ad57903a9c6cc6c6220c579f951230049fd01d4aff6e74adec5e913d6b1af70f18acba9888db3bedbd9535d028fde47276e7836c1bda590d09493149abe1d2b3778a2965ef7ff6d1b6f11b554c9f6ddc93b765ac1b51c003978b3c91493ec61f8089b03beb5e4d18232098affc45e6cb6b3e4e5e16f216523caf14d90d03f85e89f6c59ce8f60709a83ad4b959dfc7108a21bfb113cef2ad9a59627dc54d26779bcdf202595e1b90e11470b344ec6bd2c8e577fa3ae65064f0a52d2fd6951589f5787c24bebd9624e7bf30920136aa96d8ef995f54c27ad1cc0205bacd16419ff2a2dd8ceb5a590e8d61171a19480e6ca34a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261555607Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/d, Name: checkpoints/4160/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4160/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=bb75050b77415ebda75cc3b1c93b58d874fdbf0b1dbc9b825a2392f30923f0dd00b13c38dc009b81e4188ed39e3f49a21a222deec3bbcf63693a4e69d5f55c275234454912a6751582cfe45fac8758913d1c37f234d9574d529f9d9615816bbf50e5df32cee262dd87cd9df398143f656a73496865633fe6b6bfd566ac7f9d2ac1da0c65fb79c3156d93d0eadf5444d7baece033ea30f028a2511f546a21cf782a9ea30b5d026b3792ee71f0766d87c01c5196611891860782b2204675e61f50f0746901f4052e92f776622d387906442caae5332e48c9d882687680ff8468a6b5124da3d04df94cbb8c1898a3091585a0aa3637931c0d2fdf52766d15a9d304&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26157563Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0/d, Name: checkpoints/896/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/896/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=af4fe9c46e043986cf41f8f8c998e048ac938f5ee0bcc28771887d3c7ddebb2b862012aafc07fb424409dbb33e80b75583e375a0844ff8a6ee07f8e764ead5ac99e89c4c1e19956301517c79b8ca7d19ecff8eba68d84be5a565421b3e8dcb8336c210b9986596a957d4015b686081ba362b37bda734ed4d521b677dab3e7666ef588566fcb71ca3f94216e7cede4346f799182154db2a8b614607e0bc59b013810da0e1373c925fd858608fd898806ac62e407bc64ba4a841b9b49eb984920c0212174ba8b6a116f0efec598fec25f4d9289559ea0513f78d2c4a0b41ce2d5820d77831ef73fef7dcf4763f985277f52a9d71102f8643bb85433c6504fcf6ab&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261583496Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default, Name: checkpoints/0/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/0/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=26010d36abf8b096c26a76ff1c0ce4975b8acea06143e6541faf5caa6c355c101fb2ef7f12012a0519e3aa3548d056228bfffe845a985e2a4ea4c587cab33385853b2f4373c81fb9fc3bf99abce068229069cf4affe8a93b982a24ad242322f4ca8800acc1f32f6ae4266a7926075e9955e69c278c34609bd6ccbcac0aeb7b35ec9a7d6ea1abd6f228390c11d4de71df9635a9c8b41aefbdfc4fe5ddfa6dc2dff13bdca9a6eb9cfd123088d84874eabac51e228232189c2d0e0d8c6106c49706d6453cbe9196a6bccbfc2ebaf930e036b1fc6d2ef928bec95d909134d5bf6673aa9923a67329329fb7e2b5766b7d8c8f098f25c25ece5538d7214738e8099ca8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261578999Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0, Name: checkpoints/4352/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4352/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0c56c48be22ae85c77e40eb80dfc1d828b2c5e8fb6a0136d5b4bebafd8106916fec67c409613076dbed7d0c37d869544a4f1f2a9a66232979eebc0430d46cd7538877dd9b647cedb2c0cc7d538388f3d87d05378b662a945d83f5a1c2b79b51c4bd586ee50b0393b91c4a74e32620381bc0aeaa2490080b75bd121100c896cf40ff06f62602fbd3270fd80c8bbd836376da32a8956bc8091ae759f1e8011750e3f51a5760fa2f9df336cab8f8b2e269c18aaf87c3e272015873676078f46ed03d62683876eeddaf83f3437eaaac3252494d1698430f618ead5205eb2b0a93ecfd23326d27c100e291eecdcff5a1d31b1f00df50cbe04ec57fb57df52bc72e965&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261593534Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default, Name: checkpoints/2368/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2368/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=2323da908605e3391d44074190e98f6f96b6aa272dd80226b33aacbdc0fde7bbb47cff535b47b2ebde37a7dd85b7b4870ea51927f73fa58725ccef2f42ac2dc355f7788e2da9edf3fef76e1df93b9421268d05023c7c159f4c6154094f97511459cdf42ca6c8f84b328764448aa5688c3736a374d76cde462af5e26e1cf9cbcdfab4ed5ca7e4801018132579690b0d0ad024d01ea7d4990231798600b5565f6d0e96e0e149c4f5a2cb95225c0428926e196a603967fba629fe1b85d78ff7617240d9caefabb8ebfa3e96f222c1db202eb861d13057f9ff14e4155aafa6851841027257412ffdf3a1dcdff6ccc982bac8e45b6f2e2c5795cae6659eed58c30e05&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261624356Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default, Name: checkpoints/3392/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3392/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=209a59c936a22fdd867a04f68c57cb8b856f4b869dd8235e1da66b7bfbb2ca51934b356158ce181dd7094f500d38dc0fce57a0979b6fa9dc146e568b5445b7dd0c79cb707ba6196f92711f059509d1ce3e187c37e33b786afaa7551517a9f06d4762578e47e7d89a3a7274ba226b4e20f4e3bbda15f0dce97f216e25145344adcec5bb0e07c538e8fc1d770215e6710e03de170a6189cb561c907fb577821532f497bd4791bfdecad4d30b87e829743c6103e16423ef13ca1a404caa0e7bd951f33248867ddca5ce83f466383bd317504f452316e2c6816d69bbaaa4666d2088996f865b006ee77372b5ec44800a99ca0fe48c803b26f0c0227664c872fd6a69&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261631458Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2752/default/array_metadatas, Name: checkpoints/2752/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2752/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b23bfae95d8a54db67cfffb96027ad8d4bbd48f773553420e2c938ba6751375c225c6e680eedb6e797a87508c1d4d98fe50930a55c25bda8e8e0fdbff56d9211631acaf09823f6c8d2e715d0c2f433fe52d93b50448855a3fe57ceff894aad19eb9972025eea8af617fdaf4dfb9286662874aa7d6dde85e2bf7e18c90819827305e0b113ec3aa90ce361912df3ac27d38397e95bfd87c8ccc351cc8a4bddc42ff5a79eefb50838ce36c86a2b7d95690a468aa39626eda555036f87777b3f512b1a9d9ec3330ee99e47df094f614cd70f84965d925d4ace371a5f5821ce43d81a3a826422b0a900174767103d2374f2256fe2c32ccb783f8d06f9470df3bcd9ed&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261628256Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/array_metadatas, Name: checkpoints/4608/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4608/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=bee7bb93532c2551e63dae0d4f6aafee48d81001f58867ed2e1eab287a2ac27ad284c6948d188c92d462769459a44e19770720c168d6d5810f2790b091242d6055c92c6fec16b116283f5cf5bbe5e7fc60a0bada829fd82c5f3fd750dcc2273da9c1abaee8e47136660b794d9a104e09eb0c1d7b8ebad3b70c5488f3732585844431cb04bbe11aa3063ec2fb923729c0059f5c49d214908042cd84f9ba4d0a9e6c1c9770191421ed58da74bb4ae658ff6124c34ea4f0d7295f860594eb68856ca53dec6b9bdc573d42d26f6809ad224eae211c1df8834a559ea5f99d12fd652aec02e40342f0501100847a75e3570d5ae98089f6be0de1f1c4c2296474547450&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261644615Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768, Name: checkpoints/768, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/768?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0255a91adc7d2ea979acc75e58b8644cd2c154273efc09baf0e1ac50ed5171648cecddbf3b1bef2a6c6d65e32284329f72a9b33c9a9290939cd2b902e594c566864d8643498c31e0053e382cda5b979b9045ba3846b84a3e76eef709fabd0f81eae225ad2fe75fe9ab52b4d8a56a1633bbb4af38bd53379d4aca676db6ddfb247788ae2adac3013f88d4baa6a5bedd8e710a03d56096efd4da870dbcec850d86f81baaa1685d5ff5d9f1fe10d2049fd449d68cb54d6fa2f1a5d05c558666848f2d7f8bd23294d1222071d62c123ffb60c2663edeab3bf1a55fde1d93914c4468e67b4a1a17662832f79903abba0610fde2d096a7b7dc2eb099476923389c6cb3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261653465Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440, Name: checkpoints/5440, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5440?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=2b6ec83d602fddcf0bb81621b9c21a94f3b1fe0cb181a7d607f32ca7e255e856af88b021911c1a6204d6749fa1073a42222412c257a77bb2c2bda9fd6750e1421f4f0e86d0bf472f33887292c2b3dede954a2f9759017a778b7ef5adf7d35e0aab0a3c30747875567a3451618fd096fb4b8ff8fe7f084f37d9eb8bb53463dce576a472275218ec8d827201ec3cb0c30bbfe3ce0f0c079ddb3171f3d604e7ee05139f93e44cca3d07902ec7edf0bf7c44c39f171047f487c2e3356f02605f7d83e1cd83ad9305197b0a0a09cc39c77c0112cd4dddcf254066951d580689b22d5faeeba05aa10806a391b1ec8b2ff84152cea9216d83a69e81c6ae7ab5865c991a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261685082Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/ocdbt.process_0/d, Name: checkpoints/3648/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3648/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=42c9ee90077f1ccc08440602b05dc4d11d9375290c65dd1629f29c50130287406ee89a4c4791fad862e5e6b05a75e3528b1f58980e9b95f95cafcba61626ac133c6b6170724750dc5501d2c173e9d898bfc3c5597fc8dc981bd980667d733a7c706f90b84c52b0984571937814358dd46527f15a1a746feb5f8e5a615b49a1808793e83bc5f3f61fb12fe91ffc3e430fb2aa55e70910b61f88378e99b1357640d9372004d645fe580d0559f80963f4ce6015960a995f9f1ebe4c31296f150f5a367ddadebef67c922107a5e0643c32c095a2e191c19e5c24f19623c1c4d1a7fd2163efed891bb0de4c2d2c79a2884d3bdced2d142e5059e1cb171a1f2bd8c30c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26169091Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/ocdbt.process_0/d, Name: checkpoints/960/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/960/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=c4627a873137be56610ab8ea8b39c6214db78b86d281b3666433805bd22b736144a0d0947c28eb4d81a38f13d1e5791a07b7db17ef662cb028681c137048b7f5cbab277f32b4ccb256d141c8bb9602eee7ec821fdbcdc5557ce1dc86800103ce6a4e42b38f93adeeb091521a57767b7d7795d134f4a639d9afe4b04afbae292e169652e02221ca20ef106db6bd989f4fad3eb2a43f967efd5023c95de06ac5c8a68d8bc75072638fe6c9786740f312175e81b5b807484d5bcc8a7c33dc953a02d14ee10bd6d6c461014a2c555ad4cdcfd2197b7fed18d03a73d9269f732c7638e8f001233bec4a5e447f50a37ea3758225c7ddc340c201e66c4f4eb848acb01d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261688384Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/ocdbt.process_0/d, Name: checkpoints/5824/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5824/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=1c171eeea46497bcb17779c504d27ab5cf170bef068f6c494faccb7a3ae4dd111ee75dae4ee49cd77736929a7a716c935f37389d7fb35cd1436c1df3c50728f71620139f2eac0bd70c5bcf67eea53e787c79a3e843f1475909dc7cfa9bfce3410b49468857feb0eef0e012fc60907f3736065ad73b641fc3372df4149597eb8ea412ada7df0fc61c836dedceb3446a2558e096ea1d20ed9891337d0e26fe9acc2e0485a0f53b89fc1256d8a6dbfa8120fc0dbacd41c2943109921bbe845322b7c7de21994e7d269f8b7569f9cac0d1c462fa86cc57aec00a09dfdf815bb035544a8817195aceb4ef4145a56f246ee1b2ff1656fe9555145ce3354a25e04d9f50&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261709325Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0/d, Name: checkpoints/1792/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1792/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=8dcf3477004d0cdc1549d990af8558a2f75ac82577efff688aa07c6db026f4f866845967db11c549b514de7d38782d554471348718d69fe135fedd3b1577f507e3b19c5345863d63620ed5162274dc3eb0de28da9b07445fc838d76efb595c8383c910b4b49bd93e01777cc400aeed4cc7abe977a14e372b82d000c607910d2799a5402668f53a4f872723eac3ad6a4195007c78e722a4d52d389094c8006afa221ac56a324bbb69f3f61b6ce52291b708800de1a9df2c111bf93334292c4d3e1673c80c2c8cc3a04d9486a6fbb68f011219987f7faf50c2b22e2a575df13037b0ce52e8fa17d67f64db95a74aaf463985dbcebc054d182ad52c35fc2aea1a08&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261701766Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344, Name: checkpoints/1344, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1344?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=bae645bf4dbf4d782cbd5237afb6fa737e75b4d11b0c692840359354c4e8e0903e5b9f688fc4a7b99d7a4afb2197bac3e49c4278503b81e70de0e7e376daddd2213f97027c393121fdf88f62bb216b9abce18463f4360c0a50e2128bece78355ce4a94f44a8f970b425a4684329f9eaedf103247d6b3e1b44f854bf2ac864ec480df0f2e210c2f6e244856fd80f795313192b4b45f354f03cdb4e9dfdc2bb8c1019aa07f2aca33ab0c341676387f8b72dbf465fc59f9f45063ad59ffddf023e83d4ef54f99be39ddcc4a0dbcb745a942c47a4cb148c8fe32b432dd4f7d4cf8d2d23f43af49459ea10ae8f68266081fcbeab4697d87bc5537732c512d8747e10f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261747064Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/d, Name: checkpoints/3136/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3136/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=4562fdfb9b58a2586971dc053c1b41312698419e3857dbd9549124f129120db58f4b5919f3a13a5e0040f82805680a541948846ae055b74292453801f11ffeae86f4391912e6288e915e8647278628477c8a766575c74b8b34991842927f770474717e67be544267bb3a910528d812be792fb432df03c71d812f6d6734125b2b8a18f21458189b1a6d10eab0b79c992400519f28a1e7825739bfe047ab88b0c310a96b77561605e53d96c33a32ad640c0f8c05a122663fa8a47f0722a02703e0ff571aa7c460558e2d969a7c41a2119ca5bd5cb4822e30370695e21317718c990070df2c917651134b5a6dc3cc45cdf654c626619dc53cf9e458f36cdb481d37&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261737226Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/array_metadatas, Name: checkpoints/5184/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5184/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0d364521bdc8f45229a04d1c08feee6c545792c8174f26a458c00561677b59a9101d4d7f9c87826802d7f61ebca0a57fbb378c30aecb0f82096d3ce1e54ef6b784ef8df543c95e0b57e213ec72f14586785de60f938338ff0531877edd52dbe6c484324926cfd69939b1111f043f67ce312c38628286ea3b7741e60695f1f815542cf7ae3e613a8a6bcc6bf5e62a18c23deefb7d8bcb8000b8ffbb436e0bfe1ed1486b4c59ee9401ea4a602aa2cdcb4282f35d7e56cfa970c0e6694ccb6812884fdbb0cdc11c3d26e9af9ebc6a2ae48b5d9b430e12d91031985df1f6be6b32950d6d20fe28f4a11ff89137300c68b4c4d8de83ad4d555e5a711e28b45be7bf1a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261769693Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/array_metadatas, Name: checkpoints/1920/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1920/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=99d9df17cf9bef6e0ff063d5ef0d3affc8b86143f8e003c71ce42769a9df3b1dd2ebf2e8c137210cff631a07831a75200c4698dc942d4d5c2ae0ef14ee280a6c6ba4862a098b5a80d95025dc299968ebd2829168f24dfde97669a49b68ef761c4c64ffb76fdb94ad0cf5274f7b5a7f25d90e7ba1d4d531bb6877c1c7d1a483660607ae15306e600a8594a7e549f5111e0300ae2a93f261eda77daedc3189fbe83effe6ad104be1a774f6bdaacbb17b7d3d894e77e8447eade374f4b1127ecf2604dcb0a70d9a38fcbdf966e854362f06acf16e5cb47521a07b3a06dfc6df9e4d764cbc2f17d219d530dcde029f098f6b4340c0cf7d0fb730bd2e67c3764b19a4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261755212Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704, Name: checkpoints/704, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/704?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=364466896921e299d65a475d3e2e4dd96b741dd0ab0b0d83066dc9f8ba9c84ada369a6961e731ae3836a1c84c29d6a007e48d51a7b2fc7779de1fc6fff1529b0b90347dd67c1a13f1f5c40671c6aa24da0f867ba49342f19bc72f04bcebaa4bbd41d6a3ca113a064fb545bd63dd520e88eaf1389eec9cf8b4f7bc543e3c5b53f706807a43b3ab09a89a281b4623d917cf3373f2c68f1923a5977886cdfde9b98f80db6a778092045d2e4bd24ee24bea9f11bd863929adb15530c76243cedc70ec78b112087a8272ea39030110dce6d1cfe053c07c6af27607ff4104051e1096b10808d67614003faa41649a464adf29d45c17b359e61e5619616239ad0b10e93&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261763665Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/768/default/d, Name: checkpoints/768/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/768/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=64cd98a43a8af385a4ef358047ead50e46340288c548c46dd52335cd30d014e27fa2d892f5cb24bc5d133792379d0b93010c774db0f5fb68e7614c4716872eb34570cd681a67b63a93cc7c80ef2a8358f28ed17ff05d15cf7d16162cfbb0ac04c1a1947e47e7f567215db131e35aa603989a4c16fca41c05fc5f8297c1f086e9c316c88ecb2919c58bf566c9318d39324687b68db186842e816f4bf1b698c57d4f0e381685208557c23d8a3e8ca72ba2b29f18a6b4eb26920197960c13ef0e608b1c0a1ce07c0610d612090654404ba6df1991dcb3acb7c951e9bad6c94690f0c022e54eb85fd2716c294ded191ed68bb3fa0490b71e912da1074a70570939f4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261761467Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/d, Name: checkpoints/320/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/320/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b5ca8b5d66743009a9d0a1fbf5ffb083182ae60c739fe59db5dd22543b6c3de971cc06f5713f9908852f074670bf9f797404a196d2fc71e75f524158a91aa9c6a1d695d8dbd83bb6fb253f52d3156527404feaaf1e27cbbc79daf24907eff8331fd86e35175684b8d26c6b5eab3337e00a9b72958b36b5371218de140502deeb9476540b23dcc306ee3cc1067040063ceac679667bbb81a56f9648840050ea63a5ff27bce9337f0870c118039cc8e340abbef1d6792d67c695b894cb5d1d77e465925c3d92f088fc37963b92fb5aa8b6b8084009431fec83c328a07d275e61239c7291a8b039d040f1469eb88c2694e38d0220e4b34cb4e68b44b87533d2f539&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26178538Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default, Name: checkpoints/1984/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1984/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=29865c587bcf8178590fdc17d3c3c9b534aa16e1aff576b093b7117e0b482d7cedf6a27c1f688a1693546d2f82665b936ba4ead5fbf9cf98d2aa425c523b3081b885cc3f0b2e6acf92d56fa4ab96222a312f880716b038485ad1f5375cc2486e459bac8deea6c2a89263c3d65859add912839c7216f99e5de35be30b7bb6dc502b8d53b20934f8a9c8eca19aabbcee285298b15fbe90a15c85319d630062695aa984b0e764f71dde7731efa26a75e3dae929b240206b6dc24c6546e19e282de3bc1d800a5a12d454febd7edf913a436e2844b598defb6d7001e08d2fbd6fd732fbdea6169836f5efcb6ba0594df6356ec2e624194597149f3998ed9a84e0484a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261815503Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/array_metadatas, Name: checkpoints/1856/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1856/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=1e57d47d8a4534824f73f213a7e18ef2f32115615859a9f1c0de19274e247df14b369a86877a52ca27ff491af09d90923a3c757e78da86bdffba074afdf7a62f5f1a2235ba0cda5617b280d75ad25f3159d8462655d9a778e88f73f085ac1cf3dbca859866ce5c1d5906947034f5ce2b73bde0fe369c74ad78b25434838f68579ebdf33e139cfbac7c75ad32e1ed55fd4408bd11a8bcd77d56d75201526b0024b15daa43a6004ae2244d287ea43147ba32471fdfccb968286a737641f8b9c7db0739c23be6bdeca434b7f3b28ea63ada0ac96334cfa49d4183b8c4ba206e580f0af9ae22a360f67137793ed26d1416d183c9b9ee37aa38ca710ec2a9d2a25e40&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261811478Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/array_metadatas, Name: checkpoints/1280/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1280/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=561f40702e9d7ce2f53a02e4d7a265e30d996951f3119c9e46037628b87f4b8d8f98a4202683fa1ed04cb73118e9e73a3dc699f6a28d5e0efa0f0215310a5216afc4864f93c7a22fa252eed18b5658fa78a87855082e41de3ddd9a17399a4236221fbdebb73e618e6521caf9b66790bf3f7d32ddb9e17f5ce6c143652a59f69de0fe48b00f5e645180f4c6f9e35bd57cab18c1ad7412c421a8b62566aa6f4935f39ddbb2dcd59843207b2ec664a03e02c441021949c65df75efcab77fe1be95472afdc83f2953cbec4f46225537cca3d25a9bdc3e954109d8e0239af09c7217e93e5d7c78c2d12a2ccf826ccdd8fba687df1d5c3baa7640c1b86a6ca3c0b0247&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261813449Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0, Name: checkpoints/3072/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3072/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=afc4b80eaf060950e9898c4df4c7fc87b3eb3e3dfa6c3d29a3a6d333a46cdb9917ea1ababc5c4656d31bf77b7bd8f1dac55452ebdb54b9100c4ffd9c424030421cc42e818765358bcf72a7653e3d57c883aa7ad2ec00a0dcf5e9ee66a0dadeeb44284201828014ca4c0601a5039f0030be71d13d2266ae8a7e6bdee34db4e0a42b057a771994a7360edfc45ab7436e90996332b3e55b082aa94b050a6ae969dee4d3e4812a5d3fadd457ffe543d5ac482e6c6553dcf761e16e36a3bb01ac7856b583a0b8b3ea687579ec9a1056509a8dfb1728937ea262e23af1f1e4c37e1d8ae7aaf2fa6c3843b15413c66586a4193919c40f17fc086a5fdd40f7b1e488be5e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261790866Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/array_metadatas, Name: checkpoints/3904/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3904/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a25011aeaa5ccdd82dd2ebff6096135e2b1d69e5c605dd50072b675f896d069946b12bc35db560e91e42205e78bff730d214ec4b90dd699513069a69e2ab64cfb10db1b037ea513f712c936fc27a4157393e28bf5d7bc576b68aa548bd9c8c9840f2a2ae0da9d5a2f348dd64390fddf840a58e90e9c0cfb1762c8ec5b4cd8665d157606a02b5ef92527fceb09a5eaf3ac85fe5cc88fd6d18449fa4a0647e6222129e105744271665e65d4247c9f56e3bcd02808092e4a85e540ad8eb736488901e6486fa46a034291b2b91b385dedfb97e59916e9c182bea41a4a5d84bfe9b9d9f84747596cd9822a4c8edc7cb0893c610133ee032ad4914d241f0a6ab4d5622&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261828993Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5056/default/ocdbt.process_0, Name: checkpoints/5056/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5056/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b6504751aafad8ffbf83d893d5803511d209c121153ceacb5da075e273c336e93d6d8dbc13b92465ba870ce892bf213704c46ee7ad8373b2e6850b6d02d3392d5daf69c072da07b6c7915db6fc47c911cac74409ef0f4b561f4c0f37e5f635e0ddbb23e37689eddebdd2c12aee39a99fb0a51180e9419bdd42b2025acb3245f98a2bdd50036f202bc4d7a386e9cfa43397ce5ea090b04cdd4f6e8070c1fdc63ab56a066b4f9b5d8e41f782b90a92aaa28cafe6adf4cd3cff03466d066c7625e667b7869e8f3379f63a284e998f7e32d5bf652eda9a3e43b96936db802a17803e98d9d73e90b0735891ba1b174e62596486545e6bd4b160ebda02e4c5133675e9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261861973Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/d, Name: checkpoints/4672/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4672/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b605c57f5dd40756ec80c43ffd76625f2efb1eff18e239bde7f30cb4dc063bea785081cf1816b09f1ef08c04a36cd09eb949a0bc3af3958d37c337cf60b6c5aa63191cc8375f8e24e3b628fc02c76f36e4ad465a1d3ebfbb46e54bbac5dfa0d905cbf6a159acaaea321a927f4de3beb427a7f39ff53dbb085932a29ee99f8ff6862c00e5ef35fcaf2c66e1dbb927ef14e2e3c9b88b06f180096aed55ef586c9398adc45161b499b71a0a1c1b5df826e4fcd0ba9425c03b7a06effbc2c56861808237098205d9514b8593f0d32b84cdb50b94c6dba9810cffec80474c625f44017a2e842bbd9c8cdc69bcb7475a6da6c6697becdfff26c48a592eb9cc13381b51&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26186451Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/array_metadatas, Name: checkpoints/5504/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5504/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=331ffd28b5ee8b41a0bbfa54a19153dded5027dc20f36119b9455a3a9936f84ad63f54194b0525ce8dca889e0406ed93c5b04f6e18894e78f7fe07bf82591f5166783d173d33baf5a7e03e720431a18605acbb53f441ba9e0439b50c9f81de57d2f6eef752c845edb45d5e28cb125cd4dfa711c4cc505ec4eb131feadc19d35d4ed1b2ed7d207991df06a8a86f33148f3bad82571afcb321cbe7db8e09a9414a01f17559ebb123b10a8e07262ba17713c31502900ea7db2bb6c9e8636a9acc06bfbca705af3fda9b64ee50c0fff698e64c31a2ebc64f4775b53e16c6e3be073b502b9ac8358d3e5fa94cfb2eb486e6d659863f386c76ace633c5b39e0e59aec0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261928661Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/d, Name: checkpoints/1856/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1856/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=9fa3b4a889616c50d9cb08a697e52372b36d95252e39e8687e22146418fb35021355250558f68934974ee7278707cc9074f57284c2063d15550a5ef7fceba653d603c6944b7f0e7757cdd9170e341bc7f50f6b4b53d1467436ceb854c9bd445c2ac7a51586682f5aad3dd6072b9846b08f852b07b70d73380c3556bd7ee9ab9cb6642e69777c6989d9fa9a37c194cd7a0ce2736cb61bf4b2b7f9b1f10de0dc9ec5957b131a076106665d3c41fe8d3dfed1edbf0cf78015a448d786d4d271eed330fe090cb91e101add6c8b14ba61bac1fbc8fd00e5c7f461916af8d109c0fcd5a8eedb0ad4f95c5d0103d57a24abee56d54ea2c0af12a77291204d74e4928f85&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26195065Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default, Name: checkpoints/3968/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3968/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=14d87e035c6224fa14710342a245d49f34fe8507815a8d1cf59398b2bac48e06cd9185ba46e673973933a5975ecf9bbb5ac86abc8f80be51b4423ccfd42a25d0d0fea78589fae7bc654e005da767c8cc47c81501cbd9756d3ed58f0176c0bb4ddd3899da7e2a68168238f07b31b2754ed51a19dc713a286ccd4405677a4ec5b10554d9e1a643145fec749dc8ec94fe4a44fb63cd0d3d543223c3b0b502e00d4b94746ac1f42342a4099fadd1f3e2fc18c66004fd0258ce89a973a45fee148c47410ad78d153a355792ae30a9dad41cfee83d4fa837d140533decd7985cf855449863fd17958fe0148dc65390e25ceb750c0307e0a1d9991183892943d773b2bb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262056212Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/array_metadatas, Name: checkpoints/0/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/0/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=472595043908011676bafad4e9fa675868bfe2b09c83d2d89da2aa2510aa90192c3b9100844c69880e2679b0fd8d647325a7c2cd9fe26e2dc6a38dec04e928037f8676318deff2be877e6d3286710f54fb8dde43ea0db243eeaf864bfa6bc1b4a66473b3055398024667026c31dd31c5285038181bcf7e55722623b24612dbd8ddb4466f8cb87cbe0be55093acc6d005e097edbf9e5de524d8b5fbbe226d82b102b0e81a6ce6bfe7c04050458a319fc92379460c27d49195f4b4025704e4b7e9a73005b667cc577e525b37184d6cfc41298401201fbfaa698c5ea5f42c7eef9ab0e885cb2e216eaceb4da7161b9d31ac0184aa4571444234ea3afc0281650110&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262115854Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696/default/array_metadatas, Name: checkpoints/5696/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5696/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=34902e5d38bb53f3977cb618a1a47e89de3793810d90ff8a82ef312b29d9fa17126286375a4a1ce464d178a1a495099098e713454ca4371110c17e5b9e1a8a56ef1fcecefa0e152c42df62d3adbe43f64fc093a1305331f9bcc1040466e4e85e5a5d90bf987c176bcfb3782ff9584acd43503d1f0bc5f98df58027c87efc11bf978502cf20735466dafbaf9ec8166e56cbe0ee80bbd925f474432f3333c40426c721e167d3dff2fa5d61844136ec566e19dff7974af355159157509259b78c3983671d8870af01942fe45affb4ad470b30a264bca67a13c027ea8d7500017a0b590a97ec4db413e36620f904a8d674efad3dca2335439d5ded1784d1df40e5c8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262170896Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/array_metadatas, Name: checkpoints/3136/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3136/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=06edddcf28dcd055184121e80768a2ca075364342e79331aaf27b40130132629f63998ad85178e44541dc56ef18630e688eba5ba593f566bfd3d63397495474338592d5693e46d67f73fcd541cb3a854cc932c86fea9bc5b59ac33cb6eb650f776c84cb14026462a004e3f2180e6f809fce14c8d8a5ea4156d221b0f7f638fa9af48c5ffdc339db0301a7c5ceafad699a00e027ab045c75775a97ef1c7f16bf32549271b9887bbf3725d872ae1d192e8ef3465e9e3a8a9c65cd51e352f0da8bc957da729288bdf4b8ed9a7ef9f980fb86da36e5c9ca5f0766c2f932727e5d6676fc4d6ce06f0fc0a2892b2f1a0dd4ef73748adbd51d3c6fb63fde3aa73aa97c4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262204883Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default/ocdbt.process_0/d, Name: checkpoints/1856/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1856/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0f7e57a5ff931f2049c734a235cb1f7b0164264eb39b1ccf4542a1ffa751068d9d968e8e6fce18a9982457328c917820747e638c8fcd5b48d74a5e848371caa7f2ac2fe398f1eedca7ff671e9dfbd7da14afcccb3cb7d811ee72462edf7c9f0d376fa74f19b8d1b33ab334c927bbea8eeca2e8ed5cfbc1280f2b3d5297dd78981c557cf9a1c087fbdd7e899f2ccfb86dfe2121e52c06d0b71667581b69bf38658ad23bb407773665f7adccec09b66c3b2d0699fb4fdecafff48379e8065be3e8e7d767f38449c057d3044db31f4efed0fdadff73bb8e87f2030299de6d2cc0371ed03ba1c40b4bff751cbd75e439d6ebe6df9fe9bbdb60fa190068964eebde0b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262233149Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/array_metadatas, Name: checkpoints/640/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/640/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0452482f5a3182298278614139222d78827dd9abb29ccba75a3f4d7c855e36d93726f3bd215f24d176679c14427ff744b3b55198ed6915adb9c1cf520b8bb804c8d4a4ed2dba73d601f663a15f154ad2f051a691031c1b163b84fa8062290e81efdef8fed526285012c0c66ce8e81401c8d1b438286df4758b1d428b5875618277b1eac5d95d429a15bdf96df451b98e68ca5736df1614e5dd5896d1ae536272146e6fea2ee907b9339b1fb678e4d8391f1ca17c376098795e8ac4c7ca2bb74efdc45343f3bb9244efd3945f66c8da62b64c0b004ef767b2a68018272e0b15d5eaf5f23c36ef1354c11746a7b4f76adaf7f0e356d55e3c7fc85530e19b06896a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262227821Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0/d, Name: checkpoints/3392/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3392/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a4b9bc94ae824d6a2823d0b6fd86b55110723c7efaab786bfb12e5b77ac6bd99edbfc0c510e736d9037b73e9efeb0f67a92919c5974ab8f9c6c98ceaa7a93b5946260f2f5af8f12cf3cf66269f6e066208c2c8145eee1020c034afdee5939aec3f19dfa0ebdb6c6ba63bba6d51baf60e3639098c3591f525e6e041e5714c014d08fcaf6c006dd8516501b9af5fb30025bf683247b995ba5b4bd3dadc4a848531bef19b2bed8a1e0a31c1aaa8c67db186dc6662bd6ebfbb8de53cf99587621c6aee2999797a25342c307d25494b59ae7330f5a3980ba0323461b15747ec7c6d10e8c6432238ef5c0370da0b30b8dad83826ddd8ae09dc9d94ec396c8454e90325&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262285597Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2496, Name: checkpoints/2496, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2496?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=8338ebef0d023e91b57f003e980445642dc92b05effdad82a39e0d6207698dea7518cde7e8d65d2756eac8444ecf4c2e44fb478d1cdb9564468bf65e2d15534375eb3f7a92c353200f325db606ca016d270a0eca0897d002ad2f008bed0f8c467f7f2126cc052f9197cb833a94d91d03839e0ab1be59ad898fca05e207cf39c3df5d8e62a94f7167aefa31cdd081ad7bf2784ca397530d6f80d09641780e8e3755f889bcdea5186d323ec2c4222b64047dd3df48ca44914990f27ef9abf3d45cbbb8ca1f4e1e6a0e94417e6974713af902bcbf568ec55d46e77ddce207c387009f38564f06306071a5d39b30fbb5fdf8a289393165fbb4903255773ea9b17a34&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26229747Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0, Name: checkpoints/1152/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1152/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=77222b6ac69eb7d896199ad08b499ea2af30f40821f831343fd466f35817ea84622a957e8ed844d5015102b11cfdcf43f61f6ef61efc324c209c7dc28a4a1c5e750fb087007efa8deee967bf6467102e334874266defc963e39fb9810efff9928b08730c12d0023255537988fb2749d411b4f69abef350464256025ef7a92791cb2971c46ef4f379710749c19df331f51926581319b74fb06bb74c4cd916815e00a46c3f2c322cdce2ca9c5a0b9ec16cc3eb54509b343ad6c7fbe608b70f72933966a735fa7ebd9b25069e4b9c62a82d5b319eedb4556833676e21708636d752fa731d27c99591cb3c83e5b2f2d2c210021b24a766ec533fc77a91533f735826&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262294802Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3776/default/ocdbt.process_0/d, Name: checkpoints/3776/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3776/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=3d8b6f94d4d415eba17721a0e7dcff2cb48449633a47c25149d7522a2b13ac18d5a51af4a1a1f0461a30f14190de876eb897b6c85e71ccf9727388f7c29696e94bf2fd064da5f243d21170ee02b4394b7a00df8eb4faf3d94cf6d4afb1a1887fee86ff70a0322e508f6e234e72d2c22b154f94b1d1832fbe32b316a51a23918839e6c94652231636b9140fb05a5d2ec972000ed86e9eea60d232896e831e66db0259bc8cd906edbcd3ce59f3fb5c339f1b93d9a7a89619ed11e1bb08e6bec6c02f518f9db37ba7a958f75338475c219f6f2dc8fceaa5a40ff8a7e0e1cf5717ae0ec319e9893b90e16913d1da3afcf46fec5ccb7eef7da900788ce52e0b0122c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262321202Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default/ocdbt.process_0/d, Name: checkpoints/2880/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2880/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a56556e62a1ffb84b8762974833140bfe1b99be9637521855642c7beeb4f05cd56248aa9c82dba515b27b80b204f15762d33facc329532b507e734625c0921245b6bd4d1490cda91c2dd90b48ef69d8a64fc72193c3526eb1b698dd67041d1e8538918d7c74bfd610659e43fffc2488abccfd0a30493f445e81111e86ddc3fdbcdeaa373e5fd62b066853259dcc8bf5ab9935f6aaa9630e27bd3d2f7a166317fe2be2d6773b37fce8237888c2028341686286a02b1344a8185f12b8ef86c90dc170c70f8b796aae6387e6048cdfdef30d37c6f8af7d73256a752cb97badcb921c446e78e82571a51fde0db393686075fc93c8480ce07acc19b7ffd8dd2c60781&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262346588Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792/default/ocdbt.process_0, Name: checkpoints/1792/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1792/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=896867fc8d1eec7a670e464c911d26dd001410ee0f65edc48d59dcfe85ba6eae4816f3624f4c3e3f18a851bf15c1b04802b4e5c4758fb004ac2d3aa7b338e3448216689147bc545a160f5875f4e15d63a4dfe12c926f58f48f8ae61995a00dd022b5810f7eb855cc5363e6bd3080d53c1d63c8de4b92706df13109bbcb62bc0def50542945155e359c89f548c4bcbf9949e2fcdc7f9ffd70a13e82ac75d7a9247ebead709b4ebd5851c331075f6ec2b67408b721e4ae456f696f0709e7eaafeb6465815d658e733a16296c6593f137f08e12dc7d70d968589b1e97c5ed733f978c69213c8e4a783186116b526c7d7e6c133ce1e291e845a8de0a8d4b807cab3f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262355896Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default/array_metadatas, Name: checkpoints/5824/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5824/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=21d9aceff12810c7ffc7937c7c91b33ee379ad849d341dba7a4d36915a7aa5e92c4a6bef29ca46f5f2fab3eb3027d9f8eb3d9c090f1d1f7697b068aee400732acb1f6133bdec3e0abd9a62c2c4738c09e24a1d67409c3e620f9a321a3caad1d1516c7d31388fd037ba717fa15277879b8b80083a9e08c03e7754467810324be6597559c19643893af55dae6b601921d6d3c6787d54ab6168b08597c0c2b91b2c2d82821c69d72fa2a2a7622d0a27123125ab40c4a6666ad4d2e48a7e25fe663e902aff60cc782c00e92ee0a5bdae58a659ed09833b6fa0e1d28caa398f9dd648bb7d7014f0b720defb5b8cfd9dda35aeb175c174089a0b983384ff2d32c0d4c8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261840568Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560, Name: checkpoints/2560, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2560?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=ac82372a65a806cc96a3d780b6daf787eeb869572a04da78f6aad541949d2238d7c644dcaa7b713830c240d8551adcdab3df79c66cb1be579558a0fa2992d58101419330b4b0c6868124a1e69b0bb657298e50ff085024fad67191b3504a9bb113f87c3af548b9878218f54d399894693532df0434eed33f6bf73860d52020de4def63eadc5379ce5290066b66dad5a8b23e866f1ae36e35639ca2d92467ae4472562577c9a9b40472ec22aba21ac0d56ad761b1e97ace121e5f7886c4dd9d135bd9c73b75f0f83751ae3fcb3f3fdc8726b9f9a301e5f73b3e06c0e08ffc15c8de171011e9d0101245b44582670a275d9560d7ac6c9e7f349ca7e797ae8794a7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262533781Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1472/default/ocdbt.process_0, Name: checkpoints/1472/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1472/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=893121563ecd74457d4c45f93411e94107e3346e8c173c6aefaa8e60d00b795d22eb77fccdbc9f4a0cb6cabb445213c2de1f065a9c14a4bf716a6662f8650e93d255459cb6bd155e66e1c113d0048a5e74a8129aa263690202863004043a82d99e446721d89822384506fc00c52f9f7a6537865d31908c1ae5b0b329bc5cddb6a5def0fb8826bb52b88b3d050e8e903a1a5acff4902ad0b03f998172068ebcefc39deaf08df527eea03d7c48baf7c34014d78196c14da50085568801bea16ba900bc146132e94daee44619945b30dd107728ac7ff00b71746101ab3f36c57a55fa0f2b415a491205e23a6090e76b7eb3e884cc5107e588ac1a4b9efbd1712446&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262544341Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default/ocdbt.process_0/d, Name: checkpoints/2112/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2112/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=8fd576afaa0a4e7dc6478b90114aafd2ec556fddcea25aba82d575b273a2a3932e86836b0163cad3431dbbf87111415d5ab3bd46084d69606383b6e0510f81a2802e29d358e150fbd3c72aa1a52cc284f3a5518a60fdca2c62d0f76cd32fed1e6b5bdb3054a1631581a1daece2b6ff0a47a7e2ddfa459410ba642809b85732bfaef68377ec3bd5ada53b5566054e18a191797778a2ea110e56230f1d60e565d5a6003d05f1d43b80cb23a89d4743da7a1496292edc53652bdf89ebf22d9b9057ea007356bd9d656e4664128bd1da7707a5768f7552f65943eab42fcfa39354e04f0ae8ec66797552ea999d056ef281d7fc0290b707cec4c0642275bc0714dc6a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262547877Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880, Name: checkpoints/2880, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2880?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=153957cc7bb28ea058a90bbd7a3dc67c3fc52af3a38ceba4b876e247554d407fd54e05e8bc49073c44c4155e80fae7b2762083fe180121a420a994270ba503125ee56bc7e96b3290d8951e47c3d26cd6214a12148df16d40eaf4c697e2f28de5ee0bac7eb0b02c04d28f50e5b976c413c38b56ba7ad4779bf3bce87758c55f8202334ac42e0f0bd953d53e30a3456de5f68f02fa4e74f20a2681481e77d1eed526cafcf21499acc1dd84cd4a036c01d47721aa42d2d263767e2adc8b6730e9a62a8eefbbe0af37affa8f0da2b9da741a7bffb4e884217c5679e705898d6dd860958726d43ea1a10406ec6897e7a2d384dd4392b914cfbd38f6660b608702218d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262575623Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default, Name: checkpoints/448/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/448/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7c114fb84278b8c5e381a74a7926459e17db4411d2aeab1fdc79ad355c8090845bef155a87db7f39bfd2ab29de81516c423ea03a3e055e5bc66b41a3c4084eb33ab3da2b312ece85dfad436be523888e935a65fc0f2bb0fa8e4ce6c2c4874edb8a887aa3c0a3862392800dc223f49836f61ef809482c74c660249c61c13a9f222f3c650fea293de9558dd74532de75af35cf3a2c8bee740676beb0178b62885280cb1954ac9d307ac30671f380f0a39717e47b1efef2aed3bbc5d0ee243364ba6b29587d02748a7cef9bfc99236031adc0fc12a55e005d635f8a4b2350688bed052955b790820a61c23ee90e7b7c607bb88ff7ee8c11a2532618df18ac8d1cd7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262580239Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default, Name: checkpoints/2560/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2560/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=76feed67268e4637a738860c62e0d2f72cdc9a2e2b9fb2db7ad37f9f692ed9f4439d439f205cf1c997bde5919bccd80cfc8861a8914662d21d9c467dc50e800316203964d00e5ab0a4665bf24a69cb63f4c81c5c7a77a9ca63a7ac1e817edb81eeac7b1114c8d380eb98d6c1bf27bef56c188a8321841b8b8198ba6c94125d3bea30a8da70249295140fb53ab4bfb0c54be100d3e0e404de71e81bf3bf717f11e907b011e9385206be96610257e5c2dcd8e44ced8452ed7ed6c50aac765f5b9afb5734bf5ebf345471037918241b46579832d37dcfbdb1e87a73fb12bc002150228daf6c455444895ba92c6657e0c2e01c80c095d7cbcaf0de475ad93c802955&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262612014Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0/d, Name: checkpoints/4608/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4608/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0686eda86b4e8ef2c1aa6a4df88b3b36ee57ee4d0fa370c3493fe0730c52bc434cdebc67ec65b0d50acbb01aa05266aa94b6ef5db69691bf221d933a1704f3bb2addbd90e574f01b948beefbe59c543126bb38f9f2c4d0eb5dbec94cf151932538e0a0b8e59a45998a736d237b9215907def4ea454f5103e9d07ead304c19abcb09bb971b7b2edd283c859a08162558786a89258141f9328ca13a6e0bd147b7b3c33c542dfabbc273a0df474255d85396b6672bc0afa9f308c751d0b81279fa9caf3394bec5ea92234b8cb1b46636de9f42e83c4ccc1bf17c747b80def8d6f392bebcc6c1c1f5d24a35d8e85599cc44e35c161053f10b998f1200c44fff6e457&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262641437Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/array_metadatas, Name: checkpoints/4160/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4160/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=8ef70c834d657c241adddc37917eff42d8dcf1ecbe88819c973149331228ed78378cdd9927c3916eef4a03989d637413890d9f85e3e77cbccf925dbeab371ea26b4df9136ef4399ee260175bdbb9c93566f1b6aba3138a0da98a42edb258c5c107f11bb8f24831de585e0605f23d098739e951df9bbaddf0c0d8c9f307201845f3566a646e5284b94885e96ebebe8bff9cb8048d7a262cc23a18228368b64f12dc752844eb0093e77273259c66120044f639751dcb5b3e09511d74c4842203b565c7950f619b40fd748b690f0351eb9705569e4e1ee08dc10ed04854b77cd279894542b50c56da149bb3a5dcf24595f12a76447a214569d0737f91fbe77283d3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262642685Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default/ocdbt.process_0/d, Name: checkpoints/4672/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4672/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=15543e495f98b4880c45fc5497b6c16d50a1320070fde7ef420655ed23e65eded3cd3c75f8fcf1ff3fc169fba13f9316957ffc347b423b1aebe675725922c5518df487da4341903a55b21ff5dac8dc62398627c8bfd143c9a3504f2e6d8e83fea77af41f7b31bd5d1b4af95f2702132259a060d6b70331c6366597105d1ef12330da36d4e781bc2cb7dcac33c3b1e59c2f5839aa59b416d57dc692b264f91b2ce68d8cb0c4c59da2e44b1c18b4505d3a7916dc6b3d7e2861bbb99c6cb439df4d174a06ae8845487e22b5f6de5204df6098934e116cbdfdce54703fe80921a4be9e96dc6920cae8d46dccea0687baef7aed191f9376a3e87cdc1e2a05df10f61b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262657656Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default, Name: checkpoints/512/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/512/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=90ce2bfe3f43048703f34f3aa29aca98eae67541b44ae56c4e69f44c8101a7f0a1f76c785929f788c5416b19b5f8b0287edac587f3c17bfb41a6646226c5dec64a0499c026b954d4f4e46d1ee5151bb658dd0e0ea1cbfe2ef8745effb610f3c62ae5500ef4b4a4621d43f18bdb28a1ef09febb8dc86abcb898ee956f853735b0af0f27633d217d1d41bf801d7d1f622f389a2629dc9bf2c7be28f62fb101a3b9430711b8d6b91b79e4cb54cc86dc8f288eb003d055785383836059d1ce6e134adffa99c39205b9532bc790656035207862b70dec19bdc0492b751bf7cc6a2d733faa3b10cdda4235bdac8c1a8df13b7800d4a7c8e2a174f4897ba588fb46a762&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262659078Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/ocdbt.process_0, Name: checkpoints/3264/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3264/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=9e427384989722a9a5f86b7867f5f9d44d59b48a82796c1b2bd1ff18270bb853419ce49cfa7fdc41ab4b704821140b6a059b8a74ff66425cb730bdca63f8eddb42477c24d6054c3a8188186fd662c025baf83eedc8247d6ed8e1f4db5700b76778baa56484d430e2034ee26af0fe4faca9e0fc38e54745fb3f2cb8599de47f0a60d748655a16917290704ad4a218e321d9588eeebf0e36ae54b3d4a8d99d925930761ac04c94a5f7b8f065f0ea57fe05afe1499c0b3a74b0574f78d8abf5c0affb8cce9928da517736f0de791c880e686fed240072ba30cbb56241206e13e1a6e99e3703d3d32600c40b4ef79a6e5af687db01a0684a403dbdf7404170f9c5c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26266682Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default, Name: checkpoints/4096/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4096/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b565430c16cf9e837651a80fae88a801268265167b7bf350fd174a18a7a2e07233ace4a9089c6e29d38158855a7b1e1e07e424a62700d6297116616a8c09cdef678df2b3c0a9cc0735b4d0421c2d9767819fa45058ccd649a2415ab616f494f8b5bc0d55dcde81e55fedce04bd1a718fa5b59eb4db84963766d1ad0bae5caf87fd4dc61de09281d092cc66167092c0cd3e6d3e09eaf471312f7edfea8814def0e3aaa5f9dd104c5fe5d3a8205eacef80e291234c5a17605662a53ed2dc6486b50f6b02a9b1a8781279803f1f2042f09ee602180c02c3dd6be15424577fd01f420128756c478c8ccb63a1e8e3267f11799d7cc13b23ccdac97007de930d5d2eec&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261859044Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088, Name: checkpoints/1088, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1088?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=aa0e269e05ca43cf066788cdf4c13ee5d4fa3c6c6a038851a9e818a96d47972bc1d7d9a58b8bbddc31e8e75b7c1538fa131619cf87fb01575521a7a1fb92c06fb84432157f4a36dce5f29e0ca5bdcfd466ab2b33b5f4ec51d9123b6ad40c9715d3ef8e31f76e8ec055bf7594988ae9e5a7aa8c0e3bab6ea61c6eefb7f8d4a4b779512053e96e726ea8526052042201243fb13e9994201a8e80c8963d77c4a3544ab13d09ea9c19a888f3b68907c1be0926be156aa0d8ef96a50feca4025590f750b55dcd5f5b672443f67b7f6c359524c6bceda1d8e097626d68e03afead7f4768e4b255a3890fe42c16d1eaeeb1386e4a6db44767bb901cc77ef106def8edf0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262737503Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default/ocdbt.process_0/d, Name: checkpoints/3520/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3520/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=ad0fc798aee22c0f3db5f939b0a934efbd7969e9c611709f5e903931df0b5b6edf42bcc850bfb41d5ae6bd58392f422370ed53a4121620cd1b0366ab5982a55f3e26f1093c7230bb1044620ce7d8b5bb1acd7d2519559c6f1c8a2a88d1585721d7c9341e8ebfcb77ee76ded63a3025567db8bcc56f6a46edc113225afe9a709cc1ec9352052b98361f9ecf908b0a978f562d00825b0e420e9e9d8e1b0088d8920c353099f8e77306a153dfe5c8be2c2ced065570a25bf2f21f1f67ccccc231822e2cbb38891578f09cecf8b3ea8ab07e461c6ea7e23224a0fac5c7c0765cfd7a9eba0010d4022f74b2d250fba0e93d8230bc74759b67c2f9da66b9b1357c36d0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262753054Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0, Name: checkpoints/0/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/0/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=abbd33e160a0cb1d7c448e9cf546bfa3e1f439e462845c2dfb87d832e1205dfdbbe85bf192e7f5210cee3c904fd36ed95a005bf0161a0afc3fbf2b931435da5210910d62991eebf2a0673e0978f132330313ed14aa81a672bc30427541bd9c1343a56d3c27473f56c58f174815e48601497ce0d327499d8711f3687df15f4af5508c14d6239d280bd9194af96e6a6ec3a352ee44b1caca89bf82432491715f8b1b21dbf1442935ae37efd05fc4c876ed0b24c7092db144fb89b25bb8dedacf9cd8661646a294042f5a7030b67571c42aca12dcabe6f6c4e68d03dc13b22dc5c84dc7992e4a1a0b69cba25a9380003aae59a60e12f08fcb15897faf4b497a5961&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262765961Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/array_metadatas, Name: checkpoints/5376/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5376/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7b8b119ab174310e0748175d6dcd64b782e7928844d5f843bba3e635e68bdfdcf0cae294dc092c70fcc018e70f7b797a472d5adaa4b7b3a369b92085a8bd623e2541b9cd36bceb655ee438f98f7cd55a9fe0a725bbb8b29b3371546ea456f19a4dea9c58a1b0a883f6e6d1b6b184fc6741d78608e4b4aca9ffb8fc6e0ebcd24357f3065db7faa817af306a0053f92a01e40a2801bb554224623a1b0ac92e6c50ddd902fa5545966242b32151fc3a057e346d3273b762c5f04caf9411027eae64b10169975ae18cb42b89bf546c4919d1125471f85b90c49c3507dd76a304dd51b801811c2bfb42710335c64a60865cec02126b16a819747c9a6d60e40217df20&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262819848Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/array_metadatas, Name: checkpoints/3648/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3648/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a06e57be2f9043ab5db627822ccd933b71518cbfadc1b6430b38745da2bd20c224ecd2057ab2f236b45c6e427302ab7ca8788211022b81eee599bc92ffc6ddfbd9c3d13a8ef682c7f4116cfa54c74db6fe925833c1fc118610789ec127123263d0567b0624c4cddc581b93151124ba6a28d8b84deffb30e3948c56cd07c5d98033529def36e1c69a6b0630160d5ff03331b14e03d2222b1bc7ac3e67d0b5132feda0692b2ef3b8e76078ebf6585caeeabb95f481393937419f1c142e8c8bf2a0bf404f2ceff8c50ac3016f799e8b0431fe15f0ee3ae7b297723341029bc24f83f38246d027b04ca19d50f8564dceb8f1abb04ed394de8ff554bc08801622c19c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262836192Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default/ocdbt.process_0/d, Name: checkpoints/4992/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4992/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a3bbd4e44d4f2d2fc5c1dcfa1e3258ebb5b7d5a2bc030d89668eeda9c5e9c4524f7e79557393e233d1947c29a676e0af607af8369a455d763dd442ce3b3ebad761d46fd6cd67ce1ee0874681806d36b7f60ebd0fb4f54cfe48d218e16029dda9ebeeda9e4f9aefcd941cabf179e0f95cc718fc0f8b1ca647ad34a8842377bdcac2af240d6849265f57e3acae6cfb855a53c2c4ee602dc43a588158e23541d5b794e5551b16804c88c73d21124709c5480f7a3f1f3ba7d0c802544076c32e6c4bc59b390e0691bd1909231a87e01a11a4775446fa1e765c1d358c4a2d4323986f5095272aa96d7f80aae05eeba8a83063c82d3b8427d65481dc096609354eadc1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262916318Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/ocdbt.process_0/d, Name: checkpoints/832/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/832/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=4b019427d1f610d0d2db78363de8f67f4c9cdf0f4d368509d1bc7c02b0f6f9755fddeb7bcbc243483bd43ffb0c84c987b82ad6f14afed66ad5c05f7aafccf633a41b7a85d859f563266e74ca2824baf66843a5c7d5cb4bc79610a313feb6ddc1f6e5817ea7afd6803a4c3b92451959085f3ea24db3c24c46988e8f6d0f08830772c681eb1b08e1bd244932a2a346479831aaeadbe5daf9a3a30818afb62a43751812b75f68a1cdb05812df812f5c5e68d6060fdcaeb9af4ca48fc52b4602cfe9d13d369df6337b8b92b93994d767030852c62d48a41705597123b079f39d42cb39a860ba9b7b8c84a4de3617456409c94f12ebf91850c9b36c9c0b3159bb8c99&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260554099Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3904/default/d, Name: checkpoints/3904/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3904/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7a1daadbda716577eb6dc52b2a5cb2cae1cbd42dc5f32622a041ee9f3426d3f7d0a655d04cf46de24abb3fb2e08e8d46057659d6f4ed4f7edcd71b613ecf9676f5d58ea2fa5862ef6b217ec07e1b5de20b84f01368b37f00614ac364fba3eac73069d78d1d3878f75d6fae894dd912932b7ce1770fc2bc69c787fb3c77c377d668445ec8c6728e4db4d2f502ca037d3af8cda9b43c151a1212907282f28f8234ef7854ceb011c8e9810162dca2d2a945730b7904ee4b08122184afea4ef331d277208d6312e09a821a3f13fb2f1987ea9bb584ba2f3fb81914e572e632cdf53e2da06d15d91a7d4f027e089ba8452d565cf6d5ac7035c4b27adc630b91c7a29a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.261365088Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/d, Name: checkpoints/6016/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/6016/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6ed9d6035a1e254aa342ccfacdd19d03da1859727dc8807ca61a8a66e636e4421f8020ac51534ed56e67b631c017639fd1186457c59533a50641bde8a42aaab3150d53a4e2910670e7926f33f271c73c002a33113a975258b333e4cd2bbdc29d30e85c21ed468fb709319a51c902679d75ff5ecddc8ca497ea1bed2df207cbb40adc3d1e9c55584e548f1cb2c3c454149c69e7daa481a493cc60f3516944b2bfe37d85fc0d28d3ce65b2cf4868659863a5d3f1d4ce7f04541e05fd931f6a5b489b6b184d143cfa3a1a75e3e68eda31999af4718ae52cc2b6079a33c5fbea418599b4c9d4cdb3580922f4e57294027ac56096e2e76a70b0fbd0e7d931ab12a5b4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262926769Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/array_metadatas, Name: checkpoints/5440/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5440/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=30c1d4a4e1cd087f85ec91f7b857207b40d7c5b55faf937011597050bd648701ac54764f25a16b287fb62cbabae58163933dc9ff07a2325d7bb12e785724e095ae8df16f63b6c04a3d2ac180a72052f5952e96ca7357961467987eaf26fc63da1f373c99b5654662709429ae8a2ab991648228c700de24e2ba3f1d849822b1472a08f38e7f2a72f369b0c2b3ec3375c785b6b6526d62fd6e6fb06a85519e300c705ebf312da092855c2b20c6a19a8af18e0c90ab4140d9157ff8159061715c63f9cfadd6321a2cd1666a5625c9718bf4b677d218885ce602b8e058c143db3225ccbabac57201fd11cc181cb4043a083fbc265e18f0810fbc3e3e67b92de81153&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26292387Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824/default, Name: checkpoints/5824/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5824/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=8820d9f6adc4dd58bfea65f448cb2fdb9e1fbff6a23e2ec6f7c0365b35f93ea3d45372b5b1e5e75a9226227480e6616b3640312e4efb9c939882b4ceda7649692e969ed9911bdc2b1d442a42c689fb500ca459c1ec842caeb40256b01824a69e8a48fc65174d655b63279a65b51137600fe05c9802e1995164658bc99770c49aaf0763637bcfe86e76d4d88976fb80f73453602099e0bc68ce36b2d6de00d3f0603ed935790dcd451c3f3d39ff728679e5b50ecff9f85c61d9c53c6a09192bb6e394d91208d385ae820775153790e3e608aec29bed0590b1eb1f6434dfebcdcbd06e5faf2f036406e1b42a2c40853c6c3970218bc964db426b345ce012962766&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262940039Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0/d, Name: checkpoints/2368/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2368/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=26d8772d53c31d716bc86aa825612c878c0d1282b10c26c34da94965bc6fd02887fe11f22f1751706c3fa00dbb671ca581501ce939e04f7cbebb22690eef4f3cdff0f57bc67f0c4a5f2cce34be53c6a931f09b9d011d7763811e8b6a4801b140e329a43843e7575852227b8d5e6685a411d071ee6dd7694b47c397a800103f093199e5f2c74eb403dc0fe0f88fd3f873e88becd2996a9692ecbbaafd95de4098f4e3bfcb2e8262546728b82e7858aa7a68d7000431730e2f10246eb7135070155c0557828db19e69aee7a0a8b54427235dc57d033b2c1bbe33592d067d9ff7f4b7ddf7552021bcc012bb44983e08cbd7aeb9d4f3fa15b2e2d6eeb1faf8d32eae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26193712Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default, Name: checkpoints/2304/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2304/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a53e3c58f4063d268239a34ad36184935484749a0669368bd92a81c3053c9b003246b0924ddaef2f8b8f7ef1a35c58a479bb15cb4f5812f21048302240395b2487cc8a56884011658adde185368b495481eb6878352b205eae14f377b464cb9e928d8e143b9c8e9e49b46f0f7c99eb6d58895c8d6d6b13c8f951b3acdbc7cf01d927308473041d1e2f0ab834daaaf7bbf0ac2c588255cee137fb155b8f412727f9da2adfbf15cac92ff630109f2a565f7fad8f06ae22028f1f3b4fb924c2e59d4c8c027b8984f03e4e8faef665f83c70f76b6113036a08841fb0cbce6acd132a014fd5b7ab665e357bc56a2a682f665978fca251725b778a847b6957d7b4a945&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262979974Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/d, Name: checkpoints/4288/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4288/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=ae766b4da45fd529e0fb28bea5efb0086158faac10c32fd27eca499f09c50d95a7a6b070dd70c3ea112bdc26b805f8a335fb7c6a8032adeeaf20755f7c74c53f23afa527d4f7d1d517b247d760fc8e8835e70ba9336f0fb34d34a80cfef54f94157c5e46663aa6adaa885f72c5ce3e986c6ca1a3b64f194fccc22378e6b76bc4b1868e21485d22b5d82d29feb9caadd8bee64c85f35dcd55d7df2e58119ede8289cc927f5b22436b1a9e5c6903607e7b40aece73b024fde9dfaa0180be389c1bcc1d11c2bac92528024024ab9033b435082a2dcedaed7ca50dadeedaf7d83b5fa88842a700e68ff2ffec55b20fdd8efae6ce4f5ad4293f6dc4e61d02ad946267&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26297289Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384/default/d, Name: checkpoints/384/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/384/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=95cd18df14ffd037880ac7e341a15a1568c85336b8457177ef47820021be7e0adcec8413a114014bcf8ea86fe4f820b807614027de790a1dea5cc8442d8eed5737c96bd7b34c8aaae0ac1df6cb144e2d38c8761c4ba6ca1cfac99c1dbdd83d6420dea4a8c975fefa4a1c5edd8e3cb6eed3d1e11c822ea722053e330a9564ab1f055d54e1bffe36f6c330d5a55f98626c06c3176940d03748e8c79fb2350b2faf515bca7e1e0b4edcf2b1d6a8b2af9af007a361863c3a18a53f17b155720826f4325c442486f9df940d84fe617c5e82bd5baa39773e840560e0f7db5e634d852743288953289f762e943ea0ef9517655a008c76934ba7eb2f39036060770801cd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262937495Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/array_metadatas, Name: checkpoints/1344/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1344/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=5e580f38e24d8df8aec2eb28e7f123403b545c2cfda522d8ad4d36c84ba3bbb67a557d1e816293052d4a5d00f7bf8b29709d2306f7704146a5f610245264bc6c20f1339b023a0349f3c2373bd65c799b5f90361da47d2de1419dbc49b0701f499b835249a9a36cd374fda1be20c0b1b25b350df9798f3b0b3541ec6cc251849a9c901a31cff6cca3de8a6ff5088e4795e54789b82d9b3504d561afc91bc54498bd87ceacaac82b14b29dc4de8d3d89171ed96fe9396d54e81ae05f973f38c7aa9a81541618647722c84639ef2df6c2f28fb191d0954c5820a703c14d32218f85284e186b14b2800880e7122eea15dc3afd229f30d2b93f90733b2b2cd2204bcd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262995408Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default, Name: checkpoints/4736/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4736/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b7fb6db60136e04f2ae8230404ca34b58f3db8be492ada0a854f7bdffa08d62492e6a7f145adf16940985c7d5eeb0ca8f51499e0e74dc3b5949347880718b1a0f6eb7e4869a836270e8cec7fcdd60c06ebdbaf05686b252d243a6252db7ad438c0e5d42a01c76e629f7c58ffc464b42340e96c4c8bae8bbae8f39a4f89c78689069e8a5f403745ca7c7507f856b2511ccafecec4307828678c5e95a6c372568ce0fe9942211d6880d5838d6ad91407871eecf62d52f54c2e817c98f070d5a6ae40b283c313e1dabf66451970e737be6e5153f737f064373886e4ff58d1f8a1a38aa8560025a514a63a5711af3fdeedc0ffa4ef67377ce10e79816f73df7995d5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26299451Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0, Name: checkpoints/4288/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4288/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=91d782e8f2ca8e7980040e90ee3308575fac73d0f5f771d51b0f16d5d1fd267596ae2cc9736ebd95a1e8fe54856f1215dca133fe4153c078371f7550a230e86525f503971ab799a460952e05b56d6f9cd850b34493274337c953df473f5f127383e4017b5993bc99109f6d7e64b0222dfc2868335d0698028e7f926d833e472c7508ffc9450a4b28290834a899ae2ebe7d1375796f2533e3d55533b9ff884d79bae0108c6b1bd4602e2af327da96db4224d3eae9c9b6ae9401c7bfd6d3d85827750f41432308575acd1258f7abf608cc2ecf91d0661b6c6fe415963530cfa1928b88bd3e6fe51b35ea75e405f057a7aacf0e617b99b846cc0ec243458d6f6a79&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.262967967Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4352/default/ocdbt.process_0/d, Name: checkpoints/4352/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4352/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a314a4e1d03c8fa33b81fced2e9d3cfc2b1f6bc8b2e017aa8be23b3aaecfef2b47d03a74afcdeebe4236d459d372e3ba2ed7bfd7906c0df8c9405d907c30caba416169372557d851771836f019b146e60be55f94f7b3b4ae34af0365f0d36cf36bcfa53ca79be7f4ad69cb7d012b1018538c01979f53231604c15f305229357e904990989dff9ec0228792f5338d97f2d3bbc221911805bca80bcce2a50d70c4b8aa4036b6d6ca63a358e159795ac26feb328cc89268be9e640657b4c97d41bd614e11d4551362095c88a1ce3b8b0e09f58032e3d22a71eb0a2064863f7019afcbb38b2ff17669606eb015a4fc4c5b57a2855ac3c9d1419b6dde59a28252dd8b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263031137Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664, Name: checkpoints/1664, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1664?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6b7e70dd1c248b16d65709cafd50de60978f27afaf6471e5b9636acc8c677f7a8bcfbb25e29050482b759b6a372956c8b6ffbc6e1709a4af26d64fde10527d54b73b35c1648871ea7ab58d2138855fe0c995dbe64c84cb1e4231304b38cae664886bb27ede894b2c5ad7b266a32d5d64b38a9ecba5196f9cf5b271b16d2aad351162510c1fa73a4a5c593f0648bff5674203e7a30bcced81dede9c9455c205826f1df31593ea07ca2acd383aa6c43e9497f0378e4d262927e7aaa2633b3f959037fe34324d28e651ed862b3da3890afe7bb188ccd6bc0ce3e5930ca09ad33878c91dc7bbbe3a7887e1d01d8f747b94ce7b7fdbe477ba5f2c89b304ee120bee09&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263062982Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0, Name: checkpoints/4096/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4096/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=337954ba6cfa0c7990c059ad7c68c6722d022304763fc95d3f7a05b4644dc01a68d564f4033564bf41e5f4fd0290aaccc52eecb2745d01a91192f165cb0c17b705b31d1363ee53491a9da90d803e4aeac0258862c4360e707d25cb3bc2021d6ad9f12f37a2fc7040266dad7f69466055f5ee285f6ad895611cc8e5e22cfd6820030bd3e60eb56f7d072264e2807aa26099560bf8473e19ee5ed95fbb7b945d7d0b23e2894c2dc09d34c377dfe8ab23d67208d6b0cb114ad5e54d47924fdb51d45521feb3d6c028c1c8e06a5df488a24aae57209e7efe565ae4ca966b3eda048b9f9a521e13acd0dab0e8b44254bad9fb88f1ef043d27df141fa0ed65530c92e5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260571064Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default, Name: checkpoints/3008/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3008/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=5d326e2d4930ac271aac473181d2a907a3c5c818e7dccaa3d4d79953ca9f823e634aef2c16acac6eb312d31a5445dc4c9b3ca1e7a1b0ffa06a7163275880325b878e3ea41b60b6f95f43686fcd473dcc3edabe8addc394af3ea5b347516b8d934c590e7356bb6baac6305a7eb21a537901cb1d55c90f3f32eadfeed7e28177e1e4f3d9ac7be9dd38e2f7e39991380caab654ba9e237926551bbaa865138c16d3f5f5d243973004837341c61fe957382d10bc77293fd061e3e64084025e4bce57d13d62b550d495b1eeb568e869ffe4b617d3bf29071a73dd17dcc85530294ceb6a5b21e88a1bb9270948d359a396f12895dba5e9e16363bec93b8d405dd1be6a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263164037Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0/d, Name: checkpoints/4416/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4416/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=877d8c9580b44e48be102b9853b91162bd6cbeee0daa95de4c5cb0da18d283291bab258cddc4230f8df499c8852336bb9d050a897ec8ada0e03001ced84a62b6a48a0c94360d6a815189073761c708ad63017be17210d57f11b3e9ea8875479e74c3931cbabf287371200d533a6eee6499494650ec448f2ba6c74ee05148f9cfec0f3735e905c3d88f96fd7f42cb3c2bdffcf4000f39dd044067ddd178142bf23ab819c983e71b13245c4e8359bef72dec94702ae6fcb9d1722cb82895bd7bd478fdc65e931f96417919bde5df7985c6b479e5d797eb75246ada2c3d3b65ad0261e76f3ba9505383f33c1337699ec582f56a2613262e2e77e3cf08acd9c5a199&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263165638Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2560/default/ocdbt.process_0/d, Name: checkpoints/2560/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2560/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=2ae3104c7f41bbdde724d0a523b2db8023f7129eb75d79294959065645f2b8e0bb931865eca955309c969204c9a11cca9c8fa2cfd6b958533108f79e0cff70b7bbf016d118b1c2335f7d90840a2ad7f4737ad17f963724071308a08a4115c855ce3a9e27514ff397bc199ae809ab45de2ce232527ca042f3933d2d386ad009335256e6d2f07d7ebd9ab4e8c47b71e7ed52889a6f4ade593f5f6e0e77f89331399bdeb737eaaca14bdb24ee24b2ae20d80e5b07fb30e910031008e2da87a0460539d0d2abb7e0ef4ead49e0591cc1645828758ea8041e84505ab78167c8744cf125b57ff3275737fb732c7f055ba5cdbae961048b80df0095df906927bb9041c6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26343104Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/ocdbt.process_0/d, Name: checkpoints/2944/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2944/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0dcb9637725dec04624556b6508aaa29341a06892e1f2545e88f6ab7870a07270629f8ea9d36663687baa5bd06a516a319a9fe2a8264e36ab5d34a0c84b692222fec5463901d2aefac0a76f6e5c780b20c7a33340bc99036f621cfe4e526be21b25f093c89e66d263c304717bcdd4cc27248d62f22a5420e031e46a1866b992b0e8161ba064f0101c4d7ba915f05109e883f4dad61c8795053ef77ad7f1a24bc222b7416e0249f84da8b6a01ce88b5b3cc3daba53b9e83f5aed20074cbffebce1a96f5446f16386ef9c9949c08f4b0d8feb8bd757957b499ec1da684ec6eb66c9c008be541f11e9dade5e8e88f7b3a485bd6953d8bf0d2fd47899e1641b971f0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263467947Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/array_metadatas, Name: checkpoints/2176/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2176/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b6831113536443fb611f3750257f70e30daad25ad5171d816d73f1e61d271c54c4a898b459433218ace952c04b52c4931a7e0488abf0e59eca27456afdee925ef4d0fd860af20968145493936319317354f0dc4bcb2fb92bd9a38007ae933c267e54842d329b439f301d21c7e4ff231505dccc41d9ee544906196eac5c4d53ff93f16bdd1c4bad7248b60a2f3ae8e3a270146f7af41a361ad45b8e921f025b9b43d608fd232bc26a23ee6fd4a6a96e64eea258e2e8300a555464542350f1bbc7b895bd8b5cc42cda621c5bb3522fadc7ca6ea4719c33f6ec2c510a91a0b453246eade50226a946fb2fc76e88a8347a9dd115639a16b8ec0de163d69657c861ae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26348604Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392, Name: checkpoints/3392, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3392?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=3c96608186cc48c95d6c534edeed7d134a877ebc99caf94e5b73f209ef8ad0b798cfdd8bc252663cc7933ce27bcbeddb74335a5ba9f000d0b536de9cb3ae078bbdc3c0a0ca1ec55b4fea7221d6f62614c56c52473c760161be860c85670eecdaf0d246c2311d8c8d24bbbb5d46bc753451056082ff546111853ece467124c664d8a9bcc4ce5fa2c20a6d498de8d48fa9f864eb5e424a4e0a784428c05b5a820428696d789c905d4aec1c41fbf0b3bc9899e26647dcf37ea6f5158501011dc224aaae9a496ca38e4b7d1cb3804d0e6b13c539c65214d86be9f9f373f7be35ef47b41989f30fb6ff0be6e7e4371f403338be44868694358e8be087429b4a7f0f02&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263213905Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default/ocdbt.process_0/d, Name: checkpoints/1664/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1664/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7df82d8fe46ad16e857ad9abd93eb7c8174b2b39b5f1ad51682adf3efc79baf8909a4147609bd2bdd494157b794a384ed7624a17ea8dae1fb85e78252ddbb1a8a530b1defd4190dc271ec59954669c37281d16cd6e25641fde5a3fe8f8ebcb7cf8e011f64086a8f5e47a1b649309ef9161aff545aa8719200b431ba5d41d04606ac43f7a5de05cf6e775294ccb225712ac9eb1e161db84f50bdbcc1fcf1ce5e4ccd688c5cc3d981adac4ef683f80f7f08ac0e5610487b585969a2f6f96577dcd6c767ca7013f760b82e109441dd8fc48d61081ea8409f9508daabaadd70ceb41658adcbd9fba1de2aa8c4bfafdcf36ff46843ba6694ae3a47a88179bdaf8dac8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263496841Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/array_metadatas, Name: checkpoints/4416/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4416/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0dd3dbe2c858ec3b8963f791bb91ad4dd418b9ee1f275676fbd37d20cd5e776302c7ca0fe5e414c5eb23f869402c57feb25060fe14d95674194e3b72441addd881bf9e79a5707a496567cd2740ceb116aab6d5318427e367b1c96670a3abd68061f3d1bd446bc194d87bed72a76d8457ebfff8bac680d8ff5a0ce228752ce4629926ae91fd26277a56a2d7d5f8a6d4e47ed8bcf85b1351e955b3e2afc88b8c889ad7fdbcd39d1f261f4d31f7db2970c4321cfaadb6c6d8493dd7b4f83643a9996231f01e7e2723227283c0778c658c191925eb1fb2cd8ea48edb66bd7991989f3976e5b0d29d8d2d596849386bbded3837ebfd7e080dc866a5d681877bc231c4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263482338Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584, Name: checkpoints/3584, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3584?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=99229ee64b953e11e2ba0774b5e44cd2640217e82492668414fafe693bbc119b78b3e705e3cff2eb49b28af88b9bfdfc10e40b24f11123b8e1184cf6aeaf05eac7be8cd2ec9a32f82f7967c57d878bee190460e69e913f56ebc346db8f9cc571704cf688fb447f7b62b597d0ddbb457fecb4029a6bad4c4d7449148eb7606b9b90b772cfe120da48a7470b7d033dfd6d847b22a529f3376ed822e80393b4c2e61cec92532a2b70db41292aaa0ed2df5ee0270f9f1b6139692d7a8f6382b790428417cc9586a9d9f787766cc622d087a99140a5bb1d854a477d379f54ae4981de0b854ed4e85c6c397a7f4f0487f7b3bb7fe0c00fcdebf27d52f89b571834cb60&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263546077Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/array_metadatas, Name: checkpoints/1984/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1984/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0cafefd2fbeb72645c8f0dabc15db9744291f2d067b57fd7a3a9f476e8ab5ea2f4ca1e8161a5f15c2819c872b0de2e69feda195c1bc3d43412d21ec534deea04e713839215b37b4c223b21e239119befd4f1cc88fd713169705c7d2a5f1d8c7dc3b7c56f49b351558153e4f0f9cb812ad7182318150073b222271da3b4b3df5b627b8ee8c6d8143bd094896e67a73e84c7780b96362eb2cb2fd406385dadee31fec9bb21b7e0b5350c0790302196f5cbca96ffddc61bf62e7b730c17f3d5ccf41730d45021b41c32cf7f2ef1acb5747eb7eff2c392eae3d1669e237e5b1ab8ead76a090dda5081bce08280e21379c451233887ed5d4a73fd678d2e887c98e0ab&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263548594Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0/d, Name: checkpoints/3200/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3200/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a308ba3f91bcf731fa3f39b81f048b632f7648297473887428a6998fbb43e8989b81b5b6b5b484bd7da7ab3a9865b5b351ba4880ffdb585017a6a095edfe7ec1da6ed6ba1340d2c8fb90cc8da48e5117d95dc3300e4456e8578fcb341da8f39e2f0d1c8b6d742ccab49497ced8754c06b722669faa2f9e455897f8d8a326eec98847a8ff0ac0f4b60292e724645a3ad4d76550adcaaa61783ae24048511ed0f13521aac2eb729f0bdba4e631583b85d8326808c5da7f0266c64bf72e2041b01d2fee6a6ad686d5ce9f79538ee661b90337b518ce622fc8b7a1f5f7fe4f6f9700f05b16a36d79941cdf889de6fcf7732e664481e74df88afded0c6b523d7c7f6a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263544161Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688/default, Name: checkpoints/2688/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2688/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=4e63fe3a2b9d4fcf4cb94bc4bbee77bb10ef4d147ebc08caccc08dc3b218b61f7f5c77d2bb897e78f974d89c1cc28a09207e34710fa8003b6b79317b6a87f3e25227ccb7722fc6247de0cc41456b53282f26e75f5d24b6694b70290b733fa415c03031fab49f7d7dcb226ff67fa7f93895449b1cf464d8fe0b1b52c4f40e3703914fbca4a783f676514ddedc33d543565d24d44b0817afdfa9c4a1605b8aaa20bea26e0c8cca2aa3143663fc9b5e4937715da93006957110b6f0a4dbe07ccca80e9dadca1fe91b858f27dbfce3b3148f91cb8ab9115e9867bb441e01b3076e00c1858ea2e9369b83fb89b88054c2d12ab88317f7e3a5964e1d61b2bea02caabc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26355273Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5696, Name: checkpoints/5696, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5696?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=1324921ba2ccd6e2e5e08c21e1478951d2044e4ce568766cdcf8b4d5962974d465ce723cb640122fd7c4b5f9f0300f248f74b29620366bd6f724eadfcef81d9e20af6bf3472c7b2e6a549f61315c63bacd0e188e5e4331a67885f45674d715372f0311a62f81ac44de643bac3f9247468387aa0cc849141f2e65a25fc0453655a842a672d8f9758cc26fec02f29bda712f20beea3fadb380aba3f566ef7200c5727f78ccf79ac6f570146d145201741fca2d7f78b28d60fdc801441f58f0ad64094e570364bd389fc980dece4fa06b9e8369f3bcfe8cc83c1c890c1290c996caf3553e3adafe470c97bb7eef0a2f409ff2933353fde39912189be4815e36c84a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26353952Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/ocdbt.process_0/d, Name: checkpoints/4096/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4096/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=365536b358f0bd849dcc6b45670622a1fd38610e933458c93b0b0292c1e80e54eaac6d58d1457612ebeb104b1c3079189f292b81b7e1e5b0cd98197376a6a76d97284a03d1a8567d94d254312683a1c1d8c3f60d5fc68c5e61409eaf67f57e932c1031f14c8ff3cd87506a0f941bf5ded3d530d5660a03a99e3d33e176f9decd480ed22e961241a417cb0f1200f158ab86a6b3712566da9abe9e05ee00b4b058e50b4c65450977c490b2afbfd7e1bc79c836f6153e3a508a60bbd7c8363ad07dc0b32d8eed86fc7b09e35cbdfb509454b6dbf5191d2dfb483dbe894c7487fe4856c654532fc4d6fef6a612e1f1f1637f58ace9433e7cdf1b1a60656dc876db51&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263577767Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default/ocdbt.process_0, Name: checkpoints/128/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/128/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=1a5be240e5d7128274271d0fd67339527a7b783ab445a9bb0ec465092fe4b222230f58a97c56a358866c3d8b337dccfb5d08a187f08f475149587ebf1047efe990849693175540fc1fe20b3d1583ee6af4c96a885a4a292e7f244978767106cfbf90bdaeaf65dcfdf0dac57fc9263f87cfcc0c1e0aec306622b848393d47330ba62d2627a31cb01b5f2257039427ab2d0f3d455b23b2686da670dac0638ace5c865141d5f7a1d9db68260fe2ab6f1ae3bac71f083aae9355821bad808d971a8120d704e9e60cf3653366468a7927ffee6d6362345c8e1959634533fb792fb6a500329ba1bf15c27c96082db541ca7528a7b82cc3a06683dd52a20ef126c5067e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26359077Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0, Name: checkpoints/4224/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4224/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=01929149042a91052d077cdbe53be5c85803978eed36766e1d9f60ca8cd97f6128cebba79a0025c8afee233a4b1877b8edb5933382ffadc8f416aaec470f1d7f60ddd973d84254e88e00273cf33ab653ff438ecbc758cc006967bd25d258ad191884f16b8267b6753fc2adca01db07d966dabb794006a9ebae14b63b6bce8debf5264fcfd05e6eff684796c8168ae48936722e6a21066a1645cbe4049fb0b1f495aad41eb863c9c738ab847a8ab0a73b898c884d39cf58daa6283603dc930b15a2c5971c27067297b1172552a35bfdff578446b3d7897913abcafc79a3c2bd055c4f61f153ca19a9d67ebfb9528929e06bd1607458e722f6fba0e1f5e47c85c8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263583584Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default, Name: checkpoints/2816/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2816/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=88a256dd9c0b0292bdbdead80f952a4598e24e4f17e6f205a8309676ca976848a81b6ad44efe1a25d6dbe8bb29a0072a1d992d2b53115c0964a79fc3d0af259155a7e1bae5e8663bd2b61b02f0769c01a2fdaa446255bef364f8c6fb412d1840b825f1f6f5766f02e6b6d303568e466bbf0000e6bfd13ff0f10c5cf528599032e10b780dde6ba79ce031c798cdd7b40b30b5d4730cc6dbbe425ad89e483323d00cfadd2bcb9862a47db7fea57be0a9f5aae92cc81de0f932e532609dd048b5fc2d0d84fe29e31ae8a4d2010df23e3ef9143cfb7f525ba3071f8c220b3ab5b8e82402e1c3b3a2b17903fe5b53dbb12f8f6892116bc3d03eda46bb658087e753fe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263603668Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/ocdbt.process_0, Name: checkpoints/2304/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2304/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=5555ce17f2ea95083de811558af6e6d7a847314cb90475cb8816fa4a469daeca802f347abf5c2bfc3a1fb60f4d80d1b0fc6acb823799b5dcbd0f7e84056fe3b7adb5ec85948e8e390fd4b071a60ebc4f020bd29629fb73500605d37a6ee62781f0471864b7abce32f15f287e41de91d9538cc88b81d64216f3fb09c1b0b817272377df7c64b34ff5f410ba600ce6c080004986a182d659faa01d03c6891bb143b7f9d04a12716ae00659517af118afcf7c12c717923b9e826acf36929af9b784bf566c4525fcc51660e950f0e7f46bc5cce16d9343f2cbd08599b59659fbe760d85b155d06b030fc48edcd18fdc30c25fd6a24b8636f36e16cb8f4088980fc20&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26360044Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160, Name: checkpoints/4160, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4160?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=892f64bd0853e6606815a66d1781d2f9cb21a9db4a5d0c154cbd366f557d3e939d1822e98533fb1d15b762f4e6c9153a97885c085a2888a5e87200c19ca4c759236fe517ed7cf5d64c0564a21f7486cc677e8fbc56b6ea0a5b2d0f546b506ebaa4fb7e4ad304723926afe907c095f25da803fc7c9d382641996209dba5b788e4a008a497e7f118e1eb770fdfca19e2c1de65a1da41f9d5a5bfe1e052ed1b07499a6e296ee2eedbcc98733d316f143bc58a8c4370fa0e636f4c1e547a5e123d08712154a7ad9f84fe60fccd2e224bae0060c477eacbc756ec0299ef271d232876795ad20fbabe91725e646fc2d5ac4b2dbd2ab9a842fbc3c393214d7793ada536&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263613861Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/array_metadatas, Name: checkpoints/5312/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5312/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=099955643935744cc214c9ec73e726a7d8f9442f111ea90686c39244ec0af4aba0fad864182f4c44489015fe1604c0693c633d46f3fe4b4524b3eec12c9e54a3b1f650f2207bb2352f508b83ad430e4e84ecfb7e3c06a33b9ee91bfda894be2fb338a91424c7f3bd144924a2ffc1d37371dbf07b02485597d1dffc039cb987dc4c093d1f73d45fcb309fc08393a74c357f54d4a959248c4534cd48994f2bfea9c8203339f84a07a687eee9e1b2bbd372087c817ce7a3935d82a7566dbe1d085b0b7cb9f7e7810c83665b44e0e8a8a50f9374bfaabe2141cd0e3caaa869f07624c52e25d63f2e21a29870d927b513a4bb157b2b3ada63483c78bc7fa235ba6eb8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263625472Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/384, Name: checkpoints/384, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/384?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=ba225820ccc60e6b11dbe904d39de71b50635b498ca7362f0d0154d414aad4aa80534012bf33a97e3dc0025da1ec1274a016714e9ad0edffdb2d6906b5af89da9de40ccfdf4cc174a6430a30ce31909978284c526ed6872ac44f80036f467dcad20ab4c8f82b33698284cb4c6e8e53cf9c54d0aa218aefe0cf344e0b433838bb19deb365dca45e8cc8200168c07a62ca85b1fa864ac77b457f8879ae8b6432d610f932fda57d921d5f69bd72955f6528e404d6096e39926abfe547dcf2058ffd8a98e9db85d95c2485cc9971e2639e11f29308d6cfc96bccb1801e126d74b43cbf3abf0b6700aec88f367b3b4805d0e139d2a7d01df9e4bb4cbcb4b445c605bc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263630191Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default, Name: checkpoints/5888/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5888/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6f183cd73102c637f9797eb7c64a0ed65fc536e9db1effa260ac64bcbade006684e7d7d57d3d4330ac68b2ba0a821b9c7ec3368e02d6a7437e8ba257bf6a1f72b4da8e65143fc6c9455665149be53a4ce526ec4d51978ae222ed2da4cc35c31386ee5ed287aec91546db8542cc8cb758f392c143179bbf6651aa551b444e08eece75914171278f7563f28917c36274913ed7215fc1b9ea18baa149c0d017b0a816e04a9abf333b344056ec490deb3c29ed0bbdc73882b3dd1180f9c068c8f9286a0abefdbb2684bb24dbd5df9946cbd7ec7b5b440a3364226c37cc21fd988d2cf6474f8059ed7b1adaa78d575426363d894e49e6f38832e51bd7b3f2617543fb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26361143Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5120/default/ocdbt.process_0/d, Name: checkpoints/5120/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5120/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=be89414b614be94d2ca49a2c5b0d79f02b54a81abd9401929bcb639b584dafa0ca9e5ac3578b8cf46fd6f291fa56f29a75057f91e8a20d8fc07ac6d4e8fa81920b98d1391e06a03cfebaee9f3854a3d071185233e1c7eb1c886dec38b778bd481f314d794a28b8f2f7bec66c28c1c3a30390847e1c6f704c4559b3b6d5b7d26be90cef558aff3d1592fa244859698d5de17ed63eaa927800897bac51fd6b09d6a86d5e8e115c6539c55595b7856db2ffa86a067a501f9560b8e6f2b2c36c6b54791ca3cf7af499d9807c39866527b2930b6940c1ad6620747ff793873bbf548dda972a2c349f70850c688aa6af345d01291fc0edc72efe87987d9a31be863161&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263695013Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/array_metadatas, Name: checkpoints/832/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/832/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=87fa247c7d95331015d06f6dc215acbd817951343cc77abac3b6108db1a2733b27de69ac4213e0555fb33ad0bc114b239e24954deaaecacbb19999da5740ff11b4bd5c183b94c0042a3222e2a6e93cb8c5ada59cc78304e6921f4f901d60341b53e01c47498e66a9c47435bb4b4390b3321c0a4a46a6e665960e8a2f0589abcebf34fbbbf3644a0c5b80303d88da12974e3daaa9dabb324b08fb9a533cd6eef8b5f6f1d329645f53747baaa951fe5f1f12acee4157ffe434ee2c0c477219afb4152dfc835d79e13b8a939e66fc8f4a0b45b2945e4fb58f4901a3be7d85ebcf754359cb3c972a53c2f2654bc645ab8283d9abd83ff5f1ddac07f298585ea4c677&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.263217086Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/128/default, Name: checkpoints/128/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/128/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=c0ec0f832668f28626119206e015f98da96682a1ce5e1e3f5a72d571ec231b14ee000a66f5fc972469a875e6efe49f653801ad5e29ad3c586d182c04361421ff22eb48ba3253a3a4963c359e4495ec3b36fbe9e23108d28b0abf7afaded15a103eaf2f0fedc10d66d67a63378f50c274a0647062d3a906940e8caa48e47dc33b6b8da188fd5c35b12957db1e47cae1b724f4dbe9e3c45b8a74ceb44e9e87f974f77c196f09628251b7d21223d84d8b1b08de777120e7612390fee5626ff309e0a353752f49943c4f38c11d348366788f84a6752c0a8f9d51bec04b305547d50e12b8f164e723a5f69de25971ef95169bbecabe6834744c580ecae510c526c667&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.264301617Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/d, Name: checkpoints/1920/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1920/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=26af7e104654c59c6e0c49fc45725f1d2c45f78b0751f98a7e3681a59ceb2bfdd191c1892e020d1e4c04dd0f7aed35650eed843f28a8db8909ddb3cb77832cda9c96f9cb45b2d70c786a097cc3da147a3c9217856b4d069e02b1f18baff0e8b1310094ffaeaad855e51ed9ce82921f660d46f324577ecd36685e80536d59c00bcb548b50c4e742c9556c82ca923892a691bec99eb621e36fc40f961275e94430d11824409c6539ed58dc9e11daea4c5870c75de448649100a903a809f7ed2c17d2134b4b92d67bf40ddec10c90d338dec6a647ec46c21c1ea08cecda46d99fbb8d8409bce62cb848fb592f568641d9efd86827153f7eb52b971225ce3725cb3b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265576061Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1536/default, Name: checkpoints/1536/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1536/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=13fd0fb99a2ac895a2e354ced958485547dc968d3c05ee314aed4be9201f0784a8669e18c7f6dfc87ad0c275e2ea04c16fedd5d6fa1ebb5ad3f6382bb911eafd68c90642e2eb0da243719c971e5e8b2f99f8c39e39e04c09dff638773fe9dcc3ecf7584036c8f8ce60b635c6346a1eded843bdc27b16a26f03e99b3b9ec9b8fb24172112105a0d89a00887d159c8e6f1a2ff4512c33ce99b1d4a8d114aae40b0edcceac52af137605ec6344598d0493608c8522518d78033d8a76dc2c7aef706ec67787d46218194dab0cc7b418d0cbc7f89b5df7b03364ed8df125ec96411d7869b76efbb720614352a88fd9833413042a2f9541febb1e8c369eea2b1e47b52&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265578427Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0, Name: checkpoints/4032/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4032/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a631d00ab02b57c7139c93cc4bd5c2ffc6bb5ce8f861468fba89773c21ec3c6a1eb23bcdfe0a2df062445653a67df2dfff0f43b6d4e7ef13fda8d5e157f5e628243ff65c9e5550c5731ce63592940338e09d91b3f3a449feb1cc0ca0c89dff8815dbba3fc485753239f80f6616272b787a2953c6d2e1f378ca444e0903711deb75b261bebf1ca3e010a141d145bde5239416d82257c695071ec7fb57f94d71bb025b4f779c30614045b039ae42a7714b696fe1ca435289a29fa7373639133272847bf4260a42251e299429cedc6ed3a4055e79eefea8a238fc3cc373dbce98e43fab743fc23ce76d999d34a84832f71732b0c01a1b5770d92193a7c65b819bba&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265587608Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3648/default/d, Name: checkpoints/3648/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3648/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=164cd44596f8174064da4b84f4d44ffd887399deef500cbcb8349624ca178414fd6dff8e90bf2478ddb7dc15988b77b2b377d63729451271b2705a4fdbedb8a6fa975f66433313a6fca54d37269c9a7ba14d0adea60430dbf5f5b7763e806aee255fb03ecbfead7090a5ef98b9bf1563a43836c69a3a8b66153660f4f72d5eb4fd6db5565d394531e651f98a176879d8ca489060b5c6baec8cc7cdcb10c5aa45476a0cc1507baeb7c091fb2afd61aa14aa5b614bc39ab67124086825b0df6d6f1825374ce385a1429f16b3de4b4f4e596864fc257ee210a5d519d20defec8b11bb64100868471bb24ff7489a4c54bdf05061292520926bcf904444cb3bc83f03&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265600117Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default, Name: checkpoints/3200/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3200/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=8ce3584e21871b0c85cbbaef1f5a7bee1e3d7b437d533f79b4f7d82de2f8da338ed41a4e09ffb5d7eadd51514ea5e00b7ed363126542a374ba02e504f23b621d6944f8f7553dc7e65085ae1728128c0de090ade552bcdd9a23ecff8799ba741c5a9a6c824a7b377c9ab1ecf15cb886bc36d76b63c1492403b713bfd3bcfe9bb45b3bc824c79a1954859857c354b7411e0ebeaa9674097ea8e36dabd8196603c8596fcb6c4301c9e03af8533c2f32be38f6e94693525846c57fca3b0ab1a4404291ef8f218307c8c26e57a03c6e2a13d362ce2ee2738a42967993525e74312f310bfb46194b90133f5ed3ffeea50fd490de0fa35c7301c1490f221f1f8543c114&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265618057Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736, Name: checkpoints/4736, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4736?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7cf8ed7247e2ea3710f05838367cfe11a027894370493131745fdf2841789bf8bb80be3ccd5a4dcbf19c4b263f93e5ddffe19adc69836920f581644be52e0e70e0e25d504df12f2768f934ca660e26cec7518072a1c1451ecca2028de0b3737dab7e45092134d3abb6a34be7d666cb0fefa92e6dba66db4c1c9f9bf501592c62de2048b438485cbf4d994ce537f591a102bdb1944d51a1bef503e678441bc578e4beaae135c285f3c2cc1fd7585ae24080c78fa3ed84af87ec744406239d690331af8b886906342ba0615a99960f14d2f602169a89b4936287e388bfec55fd62cfc0e7be1c3e2afd6d14965a44c8ff0d450a27460cfaff84b6bf51c706fdbaa6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265651876Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/d, Name: checkpoints/512/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/512/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=98978ae65fd07079bfc739da0f92e0590766a7ce9a659009ad9e54c5552191857c50c05187ef2ea60b2342f7fc353bd2625d1120ca45c9a8daa93e9e48c7898b50ab67f19d1b6fd2ba044379b4fd1baa1558f13bd9bbe02bf00fec790184e5476c8b0e9a8ddf92a2fb1b2fce621c5082c90838942bff6202e6e61cdde8ab35e5f03bf602e85a68113eba6a3b5b2001183ef1dce1c869a886655652b1c37b3552233d79a86297870cfa4ed88d28e9acf99191f7b7246992343fbe7f60dbda0dbee3128f66890931eeb4946eac4d342f955425b07d13fd2abbfc4c88e9058ca3c8ca8167cd3a99bc7d92a91665e8b95b7028822d520509ee6eff3416eb9c420af7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265671235Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default, Name: checkpoints/3712/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3712/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=3524a63b8ee13726738cd31eb55b8bb95209a5aafac37d0780ee199b374357b5e6f593cab9732317d458b43aebdc35739dec635c5322964f5c048b3819464ef0477fd72298a238e3170e1f7327ca6dd883e248a2a57c15a5005be7211f452bb964bd2b9e9b6b3e786cc7ca5e23446513ed276222d9d7084291c15055458c162307836b948e8c16f87c49b21e20b2a36c970a91cc3d49609493684b2422e06290ac8c41f2af327293795efdd3bdc7e772cf42e394d358237fe00816105157f787971a9e75c8a0d7b69cc2721086ea73ab16475356a762ba69e7b9ebff3601f3977743f97f42c96311f2dee3dbd4af59886ff8dea592d64670d63fc34fa2180fb4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265673726Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/array_metadatas, Name: checkpoints/192/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/192/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=5775934458b513b4dfbaf1bb31dac67eeee85d5591022925a0fcac414aa5c1c626ad6acee16653405782dca34939a3786c91029f6ddaf1a68835f84b6deda0e5b4bde1cfa5eb0d318a92f138c5faa835f819743c04ffc12c1d9e2ff68482778c6f355f6fd1cd0ce39c6fdf85bd77498098ec0c7d55a8b88b2f952ea880ed81ed4cbdc2eb6d4b6c4d06e74012868a79f37be8135cdac897a92952e76a72c03b5ec5f582cceeedb100939757ff2dd5bc37b6af0fa763bf750b27c48cff495bfe734082c851cbfe1ccdbe877b6179259b0f0899d0d36a9174689a0313babb4a0c6ebd8f416b5d42c27a1d90ef69ced712b0a40df74ebac882f1027f3679d4096764&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26569322Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512, Name: checkpoints/512, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/512?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0d00fc6c97b365c95d489df211fbf0b0a535f15fe9f63016a0b7f14f228e7d86ef9aa548e1e455ebd0c2235d7b1bcc10190727bdf383070710d50cc87966fc26de01253cadd46495b4842e1274350b975b3981173f7d6d1b5cca4f7b509d8d40474e0f75ee4301fe65ca38ac4709e6e6495456fedb7dd854e9c930c0ea2dbc4911ad6b8a274fb107e6776b7d2eaa77b7cab21a4d594be556be93b8ca0318d00b89f98751e6494b64ce80cd18013a94f02853d37cd53df8df82035721f85e779bfdca34e19ae6c0ff3013a1e4c6e43ffffa1a84509ecb71447ad943bbd2732f07e967f9d2cd4248b979bf5b18d7337419cdb1f93ec71c44bbce2a71a1ec269f60&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265710654Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/ocdbt.process_0, Name: checkpoints/4800/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4800/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=70a74be7a806c2a8c5a57fe9bfd9a2dc6f891ec6980358b134849f86a68a19815111fa6e0fbab88266f0a4f98af6dbb5936ed05187fb16b71681f6e2cae86bbf5b12d15e6fd88e82a969958370e644ddab245c514a64d05f736ad783e5bcf02188daad628b913eac78dda134954142d06e79bd8dcf6918fa1dde62aa5c5118ccac3a0c2eea04c40eee3e3c96e3809553ee29969ef4f1a15c5907750e89940a4fd1f4aefced2ae324eaa6b3450b63d580c013dba93f41374873a73f13111d7bc854b15c7b0fa25cfc07dac46186489e35c59a27e46fb801e29c36b691e9f74a449cafacaf680342ab54c5b81a16e6eb3bbcba07d01859206042db09320792264a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265714925Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/ocdbt.process_0/d, Name: checkpoints/320/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/320/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=79b5b55c631d288677b4085b94d4b80391306c3c8abcee3d722aac06f38d9fe316a11a0361d1bab2945605ff51898cf9883baa4807a855fdaa47d4875e26be48e6ac812a55f3ba5414676dbb5290d8fcb1d4cf62ac8050152d8d2ebf7e6b35f0bea5b3301037024db5afd5323475759f30e79ba80ed05dc781b5862bf6c128761efda28f0f75c672f395c41992c8e090aca1135f0ffab2864301911295fdfd1ac4a71818132431dbcb123a2cd4eb2997b0bcb593b962d848e76630a86eef99286fda6225c372079e57ff14b27c3bc1f7df330f9c3e979c0ed6866174d82cbd05ff3e16ef1a2f8003a784519215a08ebde4f8945de9ef321a5f6898e5b7e5ae48&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265699414Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/d, Name: checkpoints/256/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/256/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=bd315e3152610bc09d4f3f0ad40602606260f9c50dfc8c91c9f88cd5bb9927f953f25241717469b606b749d9725e4ce64ab6fdf7a9fde197184115c0ea8abd7a028ce7a317c86d7d7479f28c4443770a06a1003f87174b3d4f425c3f0552e2bef6511c82de11785b520748d1cbe6898b890476112d78054c4f3f1100cb9317604e29fe91916ec0e4c0ec6d214e49ea8946ca59913dd89877e43443780b50527b8e970146d4f0872c345fef24c96d7e82993b3486d5af992aed5955eb00a188373ddb9850e4e4ce53f6f1d15f213fde9fabcf71735ac53fcb1850752812b02ff237f116be09ffeef999cbe709984f47316cebeda1930b47eb6faac308f0c213d2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26573394Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0, Name: checkpoints/5760/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5760/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=193a304f4804e57fe37ee6b40579473383adf584c5cadd63edf784d5ad0027aba29759e5366c1c9ab9fae5c1f32194171f7fd108482fdc90c24c57817a34a861b42424aa7314ba8bcfb840b06e4054b9a5938c92e5363a463b6cb66c12286d575168727bc9ec16b28d9ecd6189aa154856032fb5198fdd251b935048a5d6e896b64ce7d32e0bdeeb69de87ca722627e7cfcec0996f5817aba1eead20876d4f38a7347b19254e40aef92f9650e7472b57b220e4b5723eca5a23de357be11b63fed4812ce39eba7cf19051d573ab6aeca5ba2e6f3e24b7d68d906988557e03be06ff6f31abbd7a20d098074339c174f9b8d714d10011cd23c57afaaf2f67a0547a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265745995Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184, Name: checkpoints/5184, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5184?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=155f6cd2a5763711af3e46600d6e3a62a499e4707087a08f51ae927aa3a4ebf5a1a1e75e7e4081ae423103ab28a9c275d2b0b5b80e5bbf52503f7f90dc27c8a15cd9b448d5cd69fab340e40a3c328e39b0875aa5adba0ba17f81136ce72bb0fa4b70c21d13e10b1baa14c3bb0c90a0654f148044750ff2060b3e7694f49dc852d49b27495348d73dd1db75c9a636b5290275927a24d9d6d9913fdeb3a29fd317ea5d95312ad961fa7d1727ee0abf36aac70b0b487ad0b22459e17c2dd216290b73ba73fe3371f9619324ec5230c77643e385cf54a23e26b6556729554a0ba624c721a94eadf61308f53bc3fb54472e4808331f1eb897c85c8a3d33e1e8149058&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265746273Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1408/default/ocdbt.process_0/d, Name: checkpoints/1408/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1408/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0b1a88581d7607b7ea99f0c6ed23071a74898294c0e2a85e288d8fcf2072c2bf5ae67f7317fdc2baf42b82520fff1625aeebe92561130e9524c78a5ffe9550354b4f11d4af8e9aa206085a12b61ecd8c0eb03fbc4e5b2771d71d2e29c5a85d1361c471dbe24bc799efbfdf6fef37bca638923113df69ed2913bfeb7495d9a3cee9b005f4ba4843f28f1e5e64dbbb81f3cbf97f9ee498bce679f3edb15b1bd59526b27faa156a412da2cbac629e87513f158f1f17e01a72a5894d5d14dc10f8aea2939c5a339292ad3e482d7d02cfcc96996d049b52ee5f0cd52af62c8bbb2502a697074091a7a2e78f645b1c6e3cce034fd77ad7106e586fd98fade8c7da207c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265766673Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default, Name: checkpoints/4288/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4288/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=af8476370697e35cb099842da2c9a6951a9a0a03e5a014460ffa7a26318c3c7016693c1f9093bdf93615b66278898c2f8fb8649a163ed4d0467963dde930429fb8ee46581cd398da0f06a7b75ac60a70dd9de96913bf1abc9d4c93704fca486489151a200d8eaba67dd146537a014559b9f373c89291bf965fe5b1e30a26399dce1f27c292d3f1156f981b976561ef4b9f13c36dc555a02c58ad8e656a6845118faca63d93534162657c49ded13fbb6ca0dd151596cbcf4fb029580e9538c11dceea9660cc4f678c6af13d6f003c5cbfef5d5cff9b019ec4124c667ea950f196ef74f754af1ae7e5d4eeee7bf1cbc86edfa05c93c82d62e7da46e00e086dd4c2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265770328Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0/d, Name: checkpoints/2624/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2624/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=59e3d829a97beedad9b763f815e049b02fd751ba70a6928523622bafa17cf349e2945ca4d3e91342a21413abf937acd31afdcf696a967729437e1cfbb6df762f8641c1c11c1eb39dfb8ccb1058a787509e5e297258a06a8072d43880453cd61c2e218d57f2504290938497b905f2c7dd3e55bac70b2524af4506e0706c81a81706afe990ea2892b84f18c09d618b8fb7987c84dfb70df1883e975318b72acf6a25c955a1260446517c72f325542bb6707ff382e69b830d25a62c1c4c1561d4bd50bd25437cc3f507310352b7eba8192fe173e391f341c4185c0eb143af6ec34f62504e1fbc9a386dd91784cafd66bb1a7090b89693b297c80af0ce8601292347&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265766222Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default/ocdbt.process_0/d, Name: checkpoints/1920/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1920/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=5ef8978fd920b5e263a589a41f52cc3c836b73f9696cc23f0c0f61be76db2d9d890f6e6cb17836dacd60637f7acdc6f03bbd09cac0ecf4755240059008c2887a8be20e8f81a429d11fd473d8d16c75c02c205545b092c715a85972b94b0252e4c82b43a8829ddc9f57efdec7b4c0d3dd3f813c32d159bf18c999afefca2154293854003900b124f4abad18ec96976fe8803ec18be7117aeff3c63c58ba0f3c84907f6f669c180f124911c7a2d428a2a6eb09b3e170669f3fe8ad179268dc5f02a13343435757040ab0e6d846d46a79f9ff1c5c8232e6a407e92d7ca98cf1613fa471ea0f990c970a682fded327916f167d39de2d820cb43bec8a431bc8f81a11&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265795863Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1024, Name: checkpoints/1024, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1024?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=2efae3d79cbea15a166501fb3af06e2f5be313df8d99ddb4bf473a7a15dcc1e71f33ec3979838f0baa6c2f6818914079b504ae0d7d73e255eb6faceebe5392c16539d407272c4ac70e257aad43e8c77234c5347e2aaffd6add3f23af3c2e2660c8c98048b325ce989026a0d0c1ad18d6bc4fd810e447ac8c9d7abf9e5f3c21292ac4815e62851de7961d099333993b749534ea554602bbc0b014a3fdf11ab4367da6f3e627b00c93e6d54047a17fe1840b52522d4faba9f11d86f1adacdd794616b4d6be33aac84e9d993ada9158b8aff4e9ada802a2a03c7bc956c543fd3600c2559fca367384b4b164d0ef13ebe8abb6a3284540b80a5400858fa4999e72db&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265795899Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432, Name: checkpoints/2432, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2432?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a39318a3201df7880aa8f89a3c5eb118aa0bef263f15a22d80be9c0927ae6dd9fa955a47bdc5243e30991f9b463abd9b23b98299218340167bed8b3f18ba9aa3f26063de3c48f789c82de6948fb34d485c9fbac9dc39c2b17b6ea957c0b88c0fac3eb6dd48a26670f34101913ebd9f4752a1e5ca662c7f56e9f5d0d86bfec3418167f74049d9a6179d037d4f2ea88031ac05d5cb44b0c5cefa7987b3fbb8949e5a2e0c7858ea8e36b954a8b24a9101bc9b714543a12385b1b89cff29d64d9b479920e3821a22b9bcf9b544f9082d41b4324faf0d7f74f50d204ef346eb9cb5fdc442f2d00e5ebc39c019c88d835469f02f61fba4b5e48b407dd8e2a96bcaad32&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265811362Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3968/default/d, Name: checkpoints/3968/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3968/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=597894cf59d7ee6c3fbcf3a7b7c1b7ef382dab5c7c270c998c1dd1de99fd2d92c5a4e35a746856b56d6b4b70572c5d415b0fcdda14d684f71a8e373bd49b656f8fb880342fa030247c7d78239f322053f1afb3336fc5303fcc5cea45c4de954133709c9d57eca77f5e978d02c144a29a04a3d888b186f035d08f39ee0ec8c5ff67bd12bbe2025fc2c527c7864ae884a79f506720670a7bf390e6060bbc0ad762fdf74f46fe5be90940de064c87cd3d43a233d9c34ab613d8127dfe13424ea14314b8f8790f2e9089f3603537476b439b29d61c1eefb7ab70129ffddc87dbcf5075230e1a773d757d309028a1a8dd01d7e3a0bece09010b3fb95b2b644e8f317e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265819571Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/832/default/d, Name: checkpoints/832/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/832/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=1e52d92dca3f3690ee45237843a84aac8d73611628aea1458ea504c3ba0f7bd5cc28b99eb5c390d8acb736d4ab275a763be6acf43bfbe3d3efe4f895e7efd4d2d230f61f2a0a5ab7c82db19c7567a300a684b78fea23099082b7cd69bdba3732a6136e264aea25949c41bfd2035f3c73eee09ed4f6cb67c8f232caaaaf3f60421c6833e948236b2308cd49f45d0b5b0017875e295fc9850229e0f1fcec8d51e5f1dfddb563d46246e3fff6e5a7e7efc6a3eb099823968f9585817889d5a88f85e6eea27c3eef53aff2122b7d2f3a6f9f32dacae0ead927c1f1590bfe6878da7b5cd07a868d7bb70807fc55d88b343bb14c7e574538fc04c7acee1f24359e3316&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265835238Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/d, Name: checkpoints/4224/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4224/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=12a7b2ec5677887453d31ce0bb6b7c0d21f8a5ab6b09ce3186b5380f78dd89bae80ccb00f222d759047cbd5cbc4ce48a326cd2749d042a2a03ea46760909ac3c294af042634a3f7f13ed9b7c6e7888770b2809fecfdd225aa862cdfd88295111fa28ddec584c8e8d3c7401c41ddf1c4d1b2c37c20f4ab613dc4ad743784c1cca7938c473dbf2edeace2006fc8cdf79e261780acd5449de2cfe7c8721f278dbaa32ec86f895d193cda8b93c4a9bf8a830e46474e37889a5fe067547421ccd6f1a31263e3f57c14b5fcf7a443765b646a71b3272f40cc15b27ae43d0935692721c2670c730e139794a3517f5478b7bc013b09cb00db407e27100159fa83b4a80d1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265846514Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072, Name: checkpoints/3072, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3072?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6d823427b0a8ca9e348464bfe8c01997993f0fb853702e4843da221539bfc7d139b3eca0794760321e58ac18e7152d29aab706c60f4a98dd2206f28d96e7ce0ed4f6a5dc2c6b8f36d00c19a9f9d4da7da35f931ae9549cd546b1ee0700bd1dbb99a42a8b85d773cb3f8aa93504b62ed742025939216750984e79b1567e3a89474983adf6d6253255b6d0c8be7189e6e9498c669a347ccc7439c920e7a92084b0b81551f7c8b254e9c7030ca8611ce51c6241f5dabeece5e9dbf4ec0ec4bed3002f00f4cf1d0b90313da5a00a1699cd770fed55a7a4bcee840f341a1f2b302ccafdc2806048c08b11009249ee29cdc6d7e6c435d1ecfccd46cf65429af8a5c333&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265852589Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/d, Name: checkpoints/4416/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4416/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a19de5e0a97424d7af461c8f578cd6c3804b74f2c8c7be78719d6747b9b62c146afc083e94e305ff750a8461688fc32afd6833360125f595e1c6b0f55d888267ec4f50f688498bd48a3529abed4835b6c9bf23167bc50d7d97540c19d04eed0e265f4f61b87b38ac7035b01b59550b8335aab913c23a9e399fdb12f6453b38c418039032993b5c3d9a839b843c0ef8a420eaf8b5213ed8ddf1e19a279d5f7474d518ac9d3be9687a926e4471668f90bd6ebcf7ce973346c56b39de23050068e9ff0e3bb13adc98f67f028521474753e58a22cf54c4baf8ab487a4beda7f32194345858d57d62ce79c39a6a84b8e3fb277f4f94e29f89c3fd938a168315702246&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265870194Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5504/default/d, Name: checkpoints/5504/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5504/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=bff5b0f2ba35edd973febba070345fbe4df9ece589c2aca74984dbb51816e9acbeeedfaccfb2f20d632d52f3cc5400b11d8029183521e38a6de5889ab244b1e627c368cef42b00627dd086b0b8ceea1e37afbc5c0910e850de16c892bc43d84150fda01980558080b3d56e8e434e7e3520800dd9a3f67e160a2b33e66e0a8fb1c8131f013890b88ede964eacf9c993f419cfd75d0967be9225f78ec504ed728cfbd4986cfb587786c25337a2d83c7b5d4c55e982b0a7975c7af831461dba5399dbe5823225e9ade191cf590a3e595b9d213e2ef9aa0d5d5ba5288e101c4769df416bc553985c3257e8eaf10d2d7df825a0b3c4eedbfe588a44e3ac807ad90acb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26586551Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default/ocdbt.process_0/d, Name: checkpoints/4928/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4928/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=c0ddcb8bc8b3784f558e1adab3511be592a04d9a9664b0947db64f4129272bb1187eb9a6363f3e96fc5bc099b219b712d498e7c4e357128b8d064b65f417326189e2e49e4f2f14e3bdd2a24d218d4187b50aedceaf2f35a0ee9917d7c0ecb8fbcbe5296a03af3d7d7c09698fee47ceee90136e8c17247515076a9c0adbb94f441f599afc7566abe216c3d9bf11106d752b28a7459434754771b44afafad40f0b02927946ad78bbbe3817f894556c1f2f386495c6c947a06bee5687158b06ec19e324143b07493032b3f60a8045b6a400e1cdc6eef64603289fa8d7a446dd94288f938268b2e4c43d4496e99680476424bf18a2649605fb26247fef6b6e59128f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265886456Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/ocdbt.process_0/d, Name: checkpoints/64/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/64/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6d938a2ea7e6dbfb2757f9f27cfe24502a2103959dccc0addceddbd08af91f8f469d113c26535ae989da2a9330eccd43debc4930249bcaa1ca09f57ee18ee5e98b8047a93b461746314770781946c3ec02c2c87605e5d28493039384d2aaed1c2e6628bf9791d8b6dce4060bf49acf07d4f44b0c13d16344677110bb7a7b1343f3f58653632f66283e7aac08909a94603cae9634dc8e4a52acc3acba0976d8c2d4767549c9a5a08fa27cc1e30c675d282e1c349a377d8146f0a9d4f59ef2254858878ed457fa34147309f95486405a05e1d50f1cc0a5dc2905e3fc3dbff6d8cab8408d8bd50552216bb34c89b6a72449515313f65dbf669d4167ebc206bcb872&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265899237Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0/d, Name: checkpoints/6080/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/6080/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=60a020f8de99cb9e6f305b2abf363319e02ac6f43367855d72904fc2836c5d69a0c42b6b10ca4a1bb9d73ccd72a7e779615e296b57fbe34d9c148b612fde8b1e47c396beed7a8b83f2046c6e34804ca481801880fb2308f0ab357b454d9932cbefb7c9c19b30afa949db2823c5810c829604d860478c8cac8b47b7379189fa687e87d91d8fdd33033d9b789015233cebaa6181b0157b3f2e31c1f40116e421a8d36a1764c671559f7f7962b91df7efe09d8daf8db4165f5a92819885b3da9424f21135f87ff5e1540c7dbc999607a46b603acf5a4ebb10b5e359fea59b0ad394c8d92a1704510d912394b6f06da8f39bdec25dfd66359b18e96099a7a3db63ff&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265908925Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/array_metadatas, Name: checkpoints/1600/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1600/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a7c94472bd90732f7146eac7e77a3a7ce540bb0acddc0a020cb88d3b422010d10f54fcb444e5388c7575bd08dbed200a88d5582bbecb2cb05240b3446899c1ab970c85c8fc641f4f8cc78bf70f5d02b47cd1354dd087b428fb19ff930c8d896d064c0c722cab9b4e2905853e16b03651583db5cf66810705a33c3b1ae1b62afd03810c4ee6b9abdcd775d2af273cc5e882efd61a53d038f9916b43685d622b454cc5f2b1754d37a355e07cd827d9c150a615383cdc202a2af3fda26b1855ed49d15be102251187616a2d43215ab2c9102e22ac34c50cbd62f8efd6051ea8900993eb4e76d668034b3694adfb5f3af8db0c180732a17f9397a3446aff579055f6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265905116Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416, Name: checkpoints/4416, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4416?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=079227f5a830304ab147414019cc8aed8412f14c3c4e28a6e474aaed03585adc1bc4000a89db501f9d6d0cfad832223748a058adf905fe3cd6d9de2ffbb4c10e84665e476c856250f0af4f9d761bebb60d912ac470c66403906dcc1150594a8afbd136f475d24b7a4740f426c40a81c515900b9a21755f257b2661e32739c24a12a262ec73115a19d92f413dff98ff3737349c69b6c5e1f7ef54886d3170c4a90a68692123fa78f70e0d845e7b695a1972603cecc1c5d2377a9741435de35df2592d8181271d02a95d8e46edc7d830b3557273a74029f73451b61a351debc2578a15109b3c4d69d5667d14c3851483d751ae017db6765295dcce073f6be69dfa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265915757Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default/ocdbt.process_0/d, Name: checkpoints/1152/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1152/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b2181c10028f6427de0463e5279758d0f8a1d26ed62c64eb556556e9ec4492e980bab31199ec95a44575908d7e0f098396cd4000344d2b16268d7e3046eb22fd42a32de30908ff930c5046c82ac175ab13e94dfc816e980f97991007eb94393f3460e199454b012b6491f4a82bf03b8acfe69c9a724a47031655eed39fc6380ba326873ace9e8123b361ea0a75059b06798e9687c4dc07ecc999bce9d7f93bd49adce0b77201bc8ff34e9e83d610094b74f243edefbc587193b613974d883d9b69860a7107d7889bfebc35afc921182894622e3bedcbe57ac20885c765bfeeb1f3e02ee53e7034b90f99d382f36731f5a157b57d18e95f121b154184ff74b4d2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265922553Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/512/default/ocdbt.process_0/d, Name: checkpoints/512/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/512/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=66341f5c8e739bb2fd5d50958a7acf94c272eaaeb3be429865257d8dfceb394da573a59e1773fb767d68a4f106f065001db0b58f86662e497fb4f3e0444b42e9b806acbae2b920b01adb5c9fa4442c71e28c785f5f858babc8d65c9774570ea4555159277b4e1ccc6d54cfb908c1638eef5f4817d3fc0bffe52e58f0d2cbe9754c429e9468689551f34b354cd015b8b97e788c018b74d5a4e59622818291fe42736b25396845f7f102f5556218e8168cbdec3c2924f2e8de03b795d5acec8914f8ec5df7f6198751f519628e013033d795e949335c116b46513cfaf3626163958edeb33739d24e251d0aecfe6062f0a83f508f00e72b8bf6bd5e5b65d16aee46&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26591659Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2048/default, Name: checkpoints/2048/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2048/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=3019f01cb06e36f86d72f0307dc6e31be72bfb173d8d55c0676442fbba39ef20381f40f75cbba11dbc049a5db0a17630f454dc3d01ae23aa8db5aabbe6ef22e8b6e60b1c1dde03213624ab485e36b6567b84b85e207e5034f1cbeed1690df7d8715f0228c3830352be2fc9c1d3888e70e38dbe0cbaa48db617a97ae8a49877a37775aa42882b5afad5eb108e0b021a98a81490f985b9ff045c338ba936b0b8c7705583bbfe40d07174ceb79961e7a901441e5659d9a884c97db4a9f2fffc181e67fb96015d349bf64b093ef0ea2334d4a13670d4831343a54de089d237380756afe332f92b04e6cc89fbb6b61cae8186b687dff99d24a96444192640145ab8ac&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265938529Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/d, Name: checkpoints/1216/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1216/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7f82dace80c7f345486d34ed3e3c5e23e23f95bda4dd7e8f1769d7fbdbade8b0fe0ab1060820c981b002fbeadd604b5b4401d8603e6d1d70bc6bc1e20a854cdf60fd5eb62de3e35693477abbb8cf9b8db16d95a9a0b842c1c8c8e563cb07b0093c25aad2e90d9fe62bb6dcb5586630adda1dfd842f773e75798550834e73ea34d365a1790af6c8f1621218d4a82704549f7ae473d23b930856b9e5af5c6503024814ea6bdf817f339e6bf4c09ae548e7321463fc2be1220b81d634bb2e4ab059fde66a353576a41701bc4d954e4ed70ba9c2f2e532ae3cf21a8d7a88cf2e4405b29d86583d6a60ed2a0ac6e070339b789dde017e1b4e0178f37c2361ba6fe5b6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265937476Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0/d, Name: checkpoints/4736/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4736/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=27fb4bdd1bf9ecd0c8f281a18153e2ecdf892afe0a198743a8c332acb1e8edfc9b88f711bec5dc793d4eacbc00c5dd6b83583feabd5b052b4544fb3c0574aaa6d8489e43c01fc361d18735c39131f4aa4a52194add84fb7eae291fd7449a5ffd657d9c0c622db47c64292573732c13dc1f69b41f79f27490ff3175e86af3ca72bd399f3ce50af9181ace15f545d976e8984f886f0b305f501ac79b0d06d3ae18be89446639c439198e1ceca894120fb935a76d958fa097bf58cfd724917f8ba2880747e5bb4bd312ec8475b7a92dfff99fe3c7281ecb633f980f140d1390df6f359c0bf2b3d462bdff307c7ebbcf58a6add2560b75b99e2a661003ffd965bcd4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265954173Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632, Name: checkpoints/5632, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5632?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=4140e98f6039f922879efa9ea41201197b257a8fbd33f8d502fd14106c5767f74dd4ea895afce0b1a6b33aeeecbe18be953d2da82f711ebe4a081e9e8290c3827fd72136ab11bfdb8559549724c23ede75ce3a2ad9fa352e6778391c77da3032d8a5d2c4a9aa802e5f58a9733929e97fcbfeb3d3697952f3ae568477b5c1de1082b651601bb35120ed1f16bdf8417473c94b16aa63cc996121835381bad415b24fdf1cdcf0e289f31a6c560955b369966aef0d55d37a93f14cb5cbf9e970525c691dccfdbd73ca8773b0a1cfc82ca0cc181135e882c63bf6971a4af3809c8002ab2aa8f750fb7c7934110f135509b7ab97922ecb3989ffca5812aea429bb5c73&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265953738Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default/ocdbt.process_0/d, Name: checkpoints/3840/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3840/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=37bc34774828bc678170bf4087f26f59de8b8c4a195f1d2e56a5ce6f3f6eb571a47bd6caf276211f2e1c5b086bb3c79049921b02ddc7350ab7ab652c9b6496991f397804448c791ae374779192b72b40c1719e6abe6d6f793279a4286e1652b9a4a8d013d3f3e7c1de37999b5b319719f2216caea7838168640892a3e2b0bba01ac3b885f0971829764dc42b5178fe3d3214b86a8f6e7247030c378e40f259ec46fcb77cb1ea305a30f12502010afd2e75be26dc13573d28e23c2d7c37cbd44637e5392317c98d554505c1a7af3ddb4d9aaac55fecca692eaf0c750445d6def91157c907d1a08d59b7accd9b761c396564cea7e2d6a569f8b12cad41d9d32cf7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26594889Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/ocdbt.process_0/d, Name: checkpoints/5568/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5568/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b0a5f3b8ce4caedcacb045bb6f0262a3287868233a420773f333daed7208b9e78fcb787d9d70fc073560561e0eae426370d8bcc5a74a22a9b99932f0b2026290f604edf06b2ff6371f2b7f98448b7219776248424c7cf2daf50a8f5fe7de52320675caa46f01f26999ed6e518d124d398b4863e79303010aacb046aed4a3cc58cfc2c2a2e99234b6fdfd1b2f32825bdf5ab36bdaa4b306890ac00cf91784b71a1c31accf1b43daf443158e296f19e84f4ae1111a27b2ff3905a8ccc1a0451012dc1a33bd89f6130fb19ef43d0bd7c7107d856df6ac6d635d1c9559d14d97a4ec9ebc438c7c58c9b9766aa562b2463220f0a49400a496a221197748392b622bb6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265993997Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984/default/d, Name: checkpoints/1984/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1984/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7b05be882bc85aed3321d90cb32e84a0d27c9e6c50ad468ad50741b0ed1ab4cbf5751fc010fa950718e6628310c46b474b4da05917852ff86d653e1d21ed5bfe48fe358bdccf0ab69c7351d032c2515bc19f17eda3457940d90513dc695857f3e0ff2cc6564b5ea0119b777d71c7d4793b107de1fc26b181fa8e39e10fda0af36a66bce1bdbd60e0ac3be4326ec1f5d70959c93fafa733de346c247cb170fac4aeac90be50d3bcd6bf21488b67bd08b22f74b097d327d27b553d5b2c6d7946e2a18a42abe90e4816b79f496b1258dc266d3da0230c8b9bd98025b1fd4b53797974493851c59d87b115bf843e6810e7b4152797dc7e1286a64afc3957c9c33a03&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.265996304Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/ocdbt.process_0/d, Name: checkpoints/1344/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1344/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=940ab9e33e7fab95fa3984cdd2e0c5f5d8c6405d2b54a06cfa831bb69f9aaaf5e5ba698ade9003979d4e5ce51b6e4dc5302d818e6dc17a22d72cecafdb730e14dc15087144e40b370a46dbe38fcf821d9f8d9f6865246cdd5999641521965131a9c0ef29e93a7ef2b8db0b4e2c253d5dc2da1c8ddb48e033f27ebdffb2c1259af15ceab2c71427e53f9fe0be15660aa5f6688a0653ee4f6e812756735bce185915a890af3f791100ea83c453baf8dec12ec3a78a945ce928246a643397bf4e7a0ad522a34a2cca196ab44b652214c5eda63145500cc49f26ea1a9a040c3800b6c7a5aaf5d73965f96bd2c34bf54cb84f70ba00a3070bafc01e9a05f1e97a02d7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266000024Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/array_metadatas, Name: checkpoints/2944/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2944/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=88725cd35fd84ad911eefd85be36664711c4c55a9f1f7c311851f8329fe96cb6e47986792a57af135bcd117c29f7e9f0810e09742a57a7c262e9705a5bbedb5e4a44fe5a2c2f2097ae3f175d8cfed7622a4b17157a38d965c451a22e2f99396afc038063f7d45ff043d82192ef59259864a3579adfbdd7708e6fdd4256f7030aa9bd44f735f34608a36c3d1395ff47244ce78b9ba5743ce732f8cc991dd50803272ade6110610c9dea23b9f887465e775e09be517d04cbfcee658ce9c0098ff43f9fcd0b50eb1272e56902c289d09faa3aa525cdfd1d029bafde6ec56af06875f3bc4816efd9685cb426f8f261fa01f417e6fd8acdb3a152a515cd8a353a09fa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266005058Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/ocdbt.process_0, Name: checkpoints/3456/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3456/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=405fd70a2827341b21e05f045a8dea8aaa4664ca07b48c801560bd272dfba3e7c41720942bc86da67375a9517d4e92eb818584252c713323fa8ac7789745aba779ce1c4d61f5839e0f1e77a54d28cf46db021e4d690241f22899af803365a6a43a3e817ea95008c9188692ca256fa8cefbaf8f97224d422611e125873426743e9281288bda7c8aef499d196c6023afc7770b62f8242186856e19ac35688c66a251e1650e2831915956abdda62742919edef7f645c28baf591069385f25bbd749c37568551c6aadfc48a7e8ded62a807c4451bd563395ec53d7ab6dabd0ae3284c80dd38b1b102e44ea6b8162fd977464792e31a0c6c472ed973eab383db628be&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266013492Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4928/default, Name: checkpoints/4928/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4928/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6d9e8a0e2fa9e69065497888ef4aff0b641418e9c74258e0e423ff2f270508d00c4d459fc63d0b0cc63a3bd03dcf53c9b886591a9797c52dc4505e8f8c83e3637c6873a5ff140ad78d27f0727f03abd90f0beb742626e1a267f8924ec8010879b2a96bb9d919600e68fb09022378753ace1ae42c2349e62a23c866f96269c0d51e2e2f072d5b56283ba95b53e3547fbefa806e8acd4e58a1a2ad50db1be1c150c91fbea446f539ade471b2eeb63638308e2819be81224b5bcf213dfa0170f6339f37cf1328f1a431e5c40b10f8f42b2f1b1b41950283848e4e96b589a409a3c23f49c74b024b64dedff9ac7cca45f3b6fbd3f96592de96cc3928eb05ad3354fd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266017827Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/array_metadatas, Name: checkpoints/4736/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4736/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=95e47c7ada304bb9ae2875a296f210c3df02f1c2c643499a9becc37b6df037639a28ab782bfdc3e4283a21722f9ad4f89972773a4d994d56cf4e86dac11152125136319806e4ccd5e8b7855d5da4771956f1bd622ab7ccde7dc61264949cbbbee81298a384c28b95223690248040f9adb4010f58354137b27baf91aecd98303341afc4dc07b5eeb6cfd6b3650f56dcf19eae0de584ca187bd870e46401bbfaf63aa28469023e53d843931786875cc1d709a991daa72a92ea18578d10bcdafc2a72f32dd218b2abb6611879cb78edd56e6ab23f3b0aecf97da5e8ea8486c7ca68b6fdce46b4f967422d9f7498c223f43d69b3bca9f6c9770766eb9c8180c3fef9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266029176Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/array_metadatas, Name: checkpoints/3200/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3200/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=04d0548fd8d6fc78f71c53b3980cae6eead1e340800d563528e2f3dc1b03975dac7e022506a27055929bbb4dcbdea976bf089e9f23e9abd50070c16a3e0ff967c870069cf7dd9aaaf501aeaf1d4002b692f5757f21e52f4b1385ed8927467276b792c1ee35af8369ff1ec1abba8fe3eba3d80109a52939cd9fa9d715ce234ee72319570fd7901f91dcf2d0192604e5793f974d44096bd73e91decc7c6d4589f3c650b71b567a5d35401a0da123e85966d59a6823555b5f8f226598d43333d5a9009aec66b176c1eb72b3536aaea8e8fa39b837eca04f11fd0c4ead4f05c528e955d74f1dd95efa252a99120ee7ead238921f18b46b0ab8d9723741f70deb4184&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266034536Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5888/default/d, Name: checkpoints/5888/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5888/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=419222b942540dd9a5b32260e9a459624abaff8818959932d59a84be0d7eebc196217a9ac492c4bde54bb53374b5874dbbada25388ed225323c117c0d4669b95302363706ee52924222920040b4640809e560bcf12da6066676ef4826870e33c2978b2f224aefcbc0d78c8dc321025b925f7fb662a62a2f4b6567fc25c45831871dfc0c17c3d4e9332600e12fbf22a671fe73892c75aa17357166d2db201ab116d36ca8b9d6d96b6d36e1854855fa4e94a7c366fe3eeaae3fc5efdeefc255eab64bcda580380b1ef6310a98140889e9dc8f2d067487207fab5eaeefc8c8f7adbe9d215e13f65382d1ab3cad6bd38253e6fc842071490ccbbf48feae87caaa7c5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266032377Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2304/default/d, Name: checkpoints/2304/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2304/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=37183fd738ab6e0e5367d9fb35d32d21d6e04f1fa2f0da34af3f33ccd67e77e8baa782977af82325ddf57c5ef27ed08cbddaf9a5ebf7cb9cd3cef1246c0f5761f2fbacd4d3da533480962e723ccbfb568b1c247d9eb3d6d85bb480212e875ff2030d98c855f4c5c822fbfed603015fe5764635b02f556ee1110bae95fbaa83e11006c3bc7c887777daa8f964c75da68f1851c15371207fbbb76e3008f6d825b11957fef63593a620551da38f6d3b6d26580f4e472343d8d2bc850f15e14d6e2530b6658ec67f8b2532153fec0fa17ab5aadb1a6781a64331b4c472cd695cc6dcf5854345e1fe19f34f74c8cd0bcf334518a6116bb391c9054ff221fcb8a234f7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266054738Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/d, Name: checkpoints/5440/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5440/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=4784d4266439a3ed9b44284dcfe9c0bf9d15354b79faab80291239a4bcf68133e364504d5903238f85e452cd2ed8d443bb8065a42550fcb78ecde48e4d36951d83fd9bcb2d397b2fbc7dd9ea40b8766e2861946858d6de056c13c29eb4fca04bb1e450a9e364cecdec838122d2d26c4a6b2b4b668c88a5dee7de2d1b84dba4474aade4e7ae01196d6eeb4bf58c06c4459860279a604f99c2a09a97795125f3468a57eb9cd31322e08df15bec7a899f04fa2827f8e3007d412765acd8b5054be6a53a7fb7889a8ff8f1f84051fb65a7c4805296ad0c30c81b48804d965cc12f7ed733a9513f4fb1f660e99d4385a68b3728af710834e13ec61042ba04d0f28897&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266064062Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/ocdbt.process_0, Name: checkpoints/4480/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4480/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=8881a5d4d4640d0d4425f30d974d552f8f5a1855885af8d64a63e21e77dae8a0f624033c37498eb0e5d3eddbce4445cc7ee46cff8e3603bdbca92b204323090f65b7266b85fe7c89efeaf1f45427c6e26da285171eb7472d08e39ae2f6ba8a6382f8ba002efc562b5b2d1290511904bf88d255e806376c08e8ef81f6a088a7707bf2e0d6dc5c99d1370c4afc2fa1c344395c61ef99734eb38c9da7acfd9182e6b4e44cfc024d067a7fe7446b359a4de2c268932391cd6dfd13cc618390abd9c85a0223e32dcae9d5345941a993752d06bdbfbe185177ec6ede8a1cb36b30daa7db1ce1a30e52d60af20ada7f1fa7c176e8eac243cfc0e65a021da99e3bb9d06f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266061613Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0/d, Name: checkpoints/3008/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3008/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=19554a501c907b9b39e9e8a67c6f5db6eb77756f5b47df82f8d1a586896a4b3de09763fb09f1a80e4b9fc661d2a91677e1cfeb15c5e313c0d2f2f9fa30b5f4127da502c1e0820ecd6062758dc5039471529a20f6e501a2bf450d86c218ff30782d081b977e8e998849d5746a96715daae343891b0ec612481311c347e776e490fa18bf850c971170c596e9c9e87fba106698039fdcf8b748542dc27150ec59af216167387ebc8754637ceaa99ad163c0ce0f69c02a51a7805841d47ff716d28ddcd6bcd496d40a9b3c39bdf4f591ecc8e49ae7a67aad3307eaeb84e005b8442b9e7db3e4805ac39b09c75868a9b36829fc03325402e0fea3ff5cf151854bc7e1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266069872Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default, Name: checkpoints/4160/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4160/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=1af8718e0a10bafc72345f5dedc849bef9f7e94a7d4d5c2927482f767d21ac28017252f883274e79fda3f7c9d7c98c1c575e52a31c765d0f3ddb18dc71073469fac1617822e75a4ec23a239044120982c33233ddad6afa0e514182904e10b8cfd940ba0e71c32719b92df06e47bbd1c0fbd126bcb770864bc6d549f161506eecc7a28d03931df8de8b6859923c2f72e31afe75b153aae559032442649441c629f1e2413fdf5a160fbfcbebc2db37d7db08eba2b07a5873c14332524e0fb9c4ad70dc77ba5708934a0fbb5c90907ba9f56130e0bb7f9af5dfcdd7269ea20db59cd1b7dfa9a19875a7720ceee93ba5a967b8462c40867ce22a27993d1bd94175e3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266072562Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/ocdbt.process_0/d, Name: checkpoints/5760/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5760/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=ba24b3cbde2db322a732ab257e7a034f6c9d0ac4fc887f5aaa8733fcd6ae11a38f05311e53fdbcf39734f8c5cb7d5c3fa6fa215296d405153e4874aec58cfccb8d28f7a59b989e191d555812bf1a4dbc65b22672ea9a1f7e21e7a9be68ff8565c91dff3f6719ad9ca4c3def102960b292dde97e196a30fbd45cb0e1a875a00105a7d8e6187b64a73ba8b7991e35c5f8c052a2615a1bf751c5f9ef0885218019a524a3b3d7a08ce2ea8fb03c0fa3160f9f90b24df79a8449d56da55face39ef3181582964866d2c7d1779910cbff2a72f142588b1c44ba4b81e127e31e269790099f1fa361ef075d19019f7f677a79235063454ff5d7805545b7792ff2ddd2367&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266084913Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1920/default, Name: checkpoints/1920/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1920/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=07a6ff6391db6aa223872a94ec676d363f6168a83dc97062a283632dcc28ee046efc9558ede04629d470882bccf080add0a196f9f6152521414d961091e69b40126f27d1dd783dfac289f0910e14172cf2fc2ed5e66094aafdc91972a36773d5681e2c29d308a8809c5abe140a5fe4baf98bca121761424d443c9807a6e0eecc199ac83582b3e7c4b0d60c4afed1f25204f9b815bd0b07c9c8e7dd7da771b1226b8aba5c0e6cf23474c6b6ee46ae3b235a10ef6a60c88271e671e0a19f329bf6e42b1310082ad2d1305a24dca31380406e272e8ed6e7076ab9df6969987dab38c3f93619e828bd799c9426b509021f37c201f379f097f73769309ce126cd4952&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266109035Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/array_metadatas, Name: checkpoints/5632/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5632/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=aad3631bc1d92e6129e4fe1fbc74fcbb75596606be8f0c6fa5f34a324b9f05f249ff36ab31e534fde991da99ca2ca3bd3226005f539292f60cbab1df5b2f6ff49facacdb31f261152da09ada9e58e37b4b611e340d4175f198d772215e482e06fe0bb19c066c939a8b2ec5ae6caf88deffb76510c73f5b7be0e3cfc47569fc7d0ada818d57fc5cad35099bfef062182e3557f7a641ce545670b1936e84de1430681fcf71f617e042fa75c690654a87f36681c95e48f96a595587a580b05b2845178d35b081f5b9987bdd32e43efff43c312f9f06e5b299871d47c86e6a3d64a14cb4b20074584688b3949994ee117b3b2142d07a77fd3052c200c598a82e7e67&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266115384Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624, Name: checkpoints/2624, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2624?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=000e9b51a03bda1c563c5f742611ab3a818a694c63f3489d55b73264d0c17c2ec00d913070f7df713c5bcf10326f9fae97fb0d28a975c1d528ee230f1602191791dd1f4c1361a0339cea697bd4273752e2e17b4d737d97a17bbf9c155f1910ca31d18442c6ffa25072a8974d878b8da46bb151f7d2029e97310f2c1f8626bc34d45dc0c39eb17c56379794960c277bb3beae101ef6a1f860c5b678765ee396dba4a48f9cb8e520af8368ba1e5163920e0d77d5c172d5967a19695638456dc90bb93799dd39add5a430da67f93b6d981231ade4c0c50f64d1183f4a7ffbd098721ded1144d3ecd99381aab8f8608b06f8ad97e000e5701b71e09f7f3279c492c4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266123546Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/d, Name: checkpoints/5760/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5760/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=84c22da42dab37f6e3e0245d00eecaa508d29dd29adcc25e930227a3e82187f50305e33941692298d0c8ca2d3974e9ade735fb3bd6efa70a318d68584f46fbe062e128d24ad7c28771554b20a02907ae546674ced43e1ed46bebe3d501a9b7548fb1c68536d96869e670ee6e1b67321a2e4403b77f2711fd022b0ec47fe7e09eb58477c5152cae5cd8d08bbca2606577104f694f99eda310ed9dc006d364d7a0a1d27f1a0ab8fa064946b8851b49823f221bba5218e6ec8ecaf98caecc2eaeb28a8a4621ed246705dcb39586e34a3cbd8d2d5663292387a9f8b3b0958d45b69fd0bbfe84dde152668f1772fa1dedf1eb091c4646d93cbcaa6cbf65c2c9636f70&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266122105Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032, Name: checkpoints/4032, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4032?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b3ad70938a0015ce0867a23ff420dcf5d2d92a2fda572f674f86c0ae011156a54985ef88937e8ebedcda510f590d6fc538bb2f61d11a21d72ea742bbc7907b28a45503b56802f6218e0df509ae04e86f5580dd219a40171e7c8cbfaf3562df45057dd243fd2d70e5fea0c82df11c590daaa480ab4d0a33c207160c9c3c03cf86412438de9cc234a197d89ef9a742ed1a2780899b232cf55ea61228488ba60e696a54153cce221ca528f70566b926c47479936cd819818989378d91adfc02b22af989b312d254e62a5caf848e12ea47c51b33ad41ea8494a9ebdf3251910f08ad3680b228c76600af79253abf4c3875ca07dc88935b17cf5f40494043aa8802f8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266120943Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248, Name: checkpoints/5248, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5248?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6e0cc39fd765f5f14f184b48b15ab5cd87558006efd8b51f766eb12892843df8cae90e0cd9c0ba26a2bcbf54fc5a758df8ede314deb1517902ec7379e349ce2d7a6ffc998256d46616208451ac857b9aaa6291dfe175d594026d963ee9485f038b0f0584d78299741169c42ee55af38d4886b49e6e8ade96676f5671f3516633db35a3c07f6d58d61b0be8a61b1a9ab625cfb133faf7cc08f7fde959ecaa2a5b933f0ae0bfda4359ebd4e8cb349519f0c6227dbf8514b283577cc91564d480a043d29ff70edf227479a97ce18f6f8b03ab4e1e246d8d81f4546218632a32502b8fec7447867053469924fc1c9ead57e31a66b0f7cf2001f96aba40032e173cdd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266134763Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192/default/d, Name: checkpoints/192/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/192/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=25e4748c430b614f258715ab3650225dba95bd3cd84d0cb517ba4622575cfd0d4e111f1f0751203d290403418791cb8745989e8d4c6e00239f0ae8bfd011acdb2ffc6c90a4404894a532ffd498eb759d6a97e98947316002916dd9699bd98083ec70df4a1f17d8ed013aadbe13835ff2d10485111198a822e087a43e8d19578fc158d555cb07ff64c1251e76a9c41f71fc161a89402fb86afae8b46c917d16a6ca10493a4c9ee7ff7086e8c2435992d84c180b2f5ce7c4e74587c4a518bfcc0f5515289b75734c28a194b75107140d7a97db43c809237bc5bfe757589a94a948290ba802132073219d045456ffc1987d39791c40b80891d85f47fa6fc8d2bf09&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266138124Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/d, Name: checkpoints/4608/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4608/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=859be58066d5e4e2a3c7abcab2e319a93908657284d3e2ae62d0ab76bce25f6cfee854664ad429f4ad8cf04c482739392616c25b9c91b3b46910533703b25fa97a5eeb3e24a2444bc04e0b9a97859287f1c7c56dfe34013a1eaf9ef21438d9150310c99309ff049777deb67ae9af7fca1842a63c238d291f64ceacadaa8f2bf92bac55a0c136d91ed83b04f1bfc531f53a361564d02e3f6f544ab827ff3445940f6116cb140093f749f600573326e1dfb2a45c133f9d1571722f576a8fcf259a18e501ae08b9fd6df16cb94f9894d012ab95ca11ed8003310ecdeaf54a1c7e0185f9d6ae16d51de0611d191253f61f9bd838d48141119065df7b8533c52f7c08&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266147054Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2880/default, Name: checkpoints/2880/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2880/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7b1c9b7a7e9e232806e139dc2706ae41019a038f8c8fd66b2f1cfe425b65c2f3d8c11c88e102246d59154e9dd4214704bc295485f02c9081080f4bc4c2846993a3ec469bb45b4612cc61b92b8a949d9d3b5d4cf107aac3648bb1b46d32ead84b5a99ecf1dbfeb8798afac3d84f1eb96bcb724b64054c51e31cce0859f1acccbcbcd14711a773988e87c8f38327e65c04a665684893fb464c721aa5010be1470b2a64c4774442b71027def4d2c7e8507b706ba489837a39e534c7aa0d8934861a1756adb535b7dfd44a99564c1421dbdc1b0744a64d2ecff76da270f199b2b74191b66d978e077faae008843681c84efd661ac77284fae92472f56d4f5cf835ee&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266150021Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/array_metadatas, Name: checkpoints/3008/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3008/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=8b2cf1470dbc8fca260aa0eab26eb998a34388e1598b895393fd634dbae8295a67addd746da2a538e123c29110ca73901544997987d8368341c58cf738beb265ee6ead8f867e66258bf48b49829685b0026737a4404f5f66ba5605164569c5ce4b32ed81f43da4c751d23092649ed9441848767195d6f063aee3fb53fba6a854ff49b267052bcd51fadcc3368e1b6e522f5bafe323aee2bf07b3718b70797c66214fa7b7aabaa0fc688043bd6d7a9a8f389fc24a421a9961f38724baac8c33b44b085346fa73799863c34234569bb831e3a4acc1d398f50ec674635198ccf0bac9281c6649c1af7ab32ad5905e57e44ed54839d8dda63d204a1f0782ed2f6694&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266182188Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/d, Name: checkpoints/4544/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4544/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=5ab022c571e464493b1f72d177fef397753cecfec125ad4dc6d92fd204c3b0f60380be49bc29834effbaf5340b539f3eca4f997e874acc126bf1a8d29e7aad1b3ef76a0fc80999ad79ed8b0a660c656bdc18c9d784dbdcc7d63a560c2da411e53dd43c9991f9b87932718955edaf825a049ffb054a49ca29c13e1e2c63c27542d5888633acec1520c691083a3f0dfec3c48b6d1fa28450f509234b985ab9ae85826d21f758d73dbf76eebc84c7e28ad8e0942b67f2e19876fc6fbd3d39ff1203097931e529d27cb034bf37f332765fd56f5724535cf7630ebca005006d04b296e4481d404da615ed23e216902e350106ee416b7ecaa9f209edd24a63239ab38a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266176437Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/ocdbt.process_0/d, Name: checkpoints/2816/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2816/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7878e9ebc174c93375d4c169ab9435043e8d96554d9c5905ffa745324b6ad15dafcc0ae6600adbfa8d488439bdfb2a8f2094b3ee01917a04088ac3e6b274ab6108ac37985c12eddd51238f4a49f2f953d6c43f293a006fdf833f5a186e5fa72b03a31f088aef3fa6a82623c0c9d97d6d69f5ea2435e4c0fc41eeb4935424d53ae1112a26a19a6675cd6065ac671a6c49e41d1e890c14e4d588f71abd87c84e1d70a359f59126503b104215d96cee2b17a55d698be73428ee9ad9d83c161ce724421a2b7ca15d71107c6ee8809fe79dc0c0cae2522469c80070a84610094f636feb760535bb1365e6aa5c75398957104d85303ee24576412972f5b7648a8610bc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266191486Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/ocdbt.process_0, Name: checkpoints/2432/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2432/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=3f94ac2fb8e3f3ba39f3861a37a7d6b6e99b5973254fdba63cd5091ab70946585950e390abcb905dc60c615dafe6ff0fc0f08d489a446113840246496af5346469d084fc70797ca708dea684fb4d9f57548808372e75ab56b4e89e7c2c4ed453d8d660da633deab40e9e9226d73085c6f96fbdeb1c9f487727c4c5bba814c11e671744ad30e2e8025fec7974d727e56ea45a52375fd473f4ae2441ecc36d6915e824c6c28fb4ff3c4216ad26507f9638e95f95c75f6cd4a96c9932faa070d12031a05822e20b5b74e68a916857dae316a17616cc54daabcca3ddcccd4ae9e6828d5047655fb30c6b0880960faf05527a514e10663b67d52758689267afff746b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266196969Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008, Name: checkpoints/3008, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3008?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a62d0b9e36023598769dec3bacd5e2d639bbe5ccf1e1b0fb3626d79fde4292a113e4ce79fbde49537cb552bc6a12871158bd8ce7e7021956fc6138f542af1dcc811aaa69dd52c4114bb0527c0c5b633e39a71052656118b5ebd5e6b328a90d283fbf8b8dad6f2495fe215bb994ecd8aec4b962795589e3b1f671ad56ae58af7d30c23ec76927f716ad07d3fda7648e9b7e9f6b8c887e3168645021e5fe3b7b6cf062ee3f180b70e82c795f7138c6a7ab3bfe6068b6ae0c0f1a1257af019d73c3431c7a8156bbd8a6f764b648a083f68f6ed9d2a403a00e81c6d1e2585ae0269ea2cbb3bd5020c9129b3d77344e74d74ef8fe322be0317b0b768e9d032f050331&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266198291Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0, Name: checkpoints/640/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/640/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a84a6b756f462ba33993f07ccacb4a5960c22f977bdc94fbe091e7b93dcd7843aadcc71b4d58eaff85ea2235f1ceddc073344683abecd251fd03742675ad4afb58959f7f723da3432a8f306d3cdf34bf0658d291acd1bfd8eeb2dde932c782b0d21057549c3366b55942c21051aacd79ec637d1e11b6d55f7bacbbeb78f327522db347eaf4fa70be2a95a60f18faeb56c55e896c69f316e731aed3e38c47cc5a0eebfa18c6716d821cd0bb526a403644e4129660cc734ef6c58acfad2d3438000521fec560f611e3647702d9acfbdae5ca03e21f283dd94e415afc75d8f82d03e60fbdb94eca0db9084eae87a3bd491f09dd49ee7acdf7729b0a74a98b9f452a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266211448Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4032/default/ocdbt.process_0/d, Name: checkpoints/4032/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4032/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b9473c47e02a83176ae4a71ec6dab431292aabc8949c1cca6e68d065d2beb8976117b2b424c0daf5d3ebb9cffd20ab388ce8ae8e929c0055afb4bb30e4f4a64faa7d5fc15e6588a267dd6d3b82b9305fbb625c904cc5a1a1af1a3fef417dfd4c24b888e61a924ee9d9bf011a20a3f93c19557c7f5fdda7c957b3adb842f69f86a1f8f7cf860604848054af47a9ccc6ca31504ac28d82ab7d3ac095e02846184b466dfbe898fe48b27b2adab7128ae115c578378a7e8321d9f8b9dfdd58a8d5d7e3ffd8f7c57bc74beaa41ca67a84e59be81f4ea00cb48d056b962ad58dd14cd8850614419152a58b20f484c06cebc32eecf9cd2fb7500ce022944c03dfc6c115&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266216414Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/d, Name: checkpoints/5248/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5248/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=bce36c4dde858cc0cd36d3d89649bc11984fda433524a27e361fc37c6dd933bde26961c1a62f316ca4885dcc829aebb6f4fbee637ad1c22db7fd12ebc1c6aa9f5809b0b94543ab52d822fdb95130ca977c14a8122049382cdf85c58744f4b16b3808fc27fcde683609407c206532e5ce9b9d0e46fcf5a3c21d8b33120ee52c4f8f556e7db2500a088e760b7efe222891523d7fbdec8737ceaaf60e615e41ae085ddc612008a2486d492482701a218a83a783ea2cf11957e3aa50e522e7755cbf3a685b01b4c327deeb51f3683ec74c2aa2805340956da015742bd3ed8ab8412ca701a1a98b057eb439ea0a674c380736879447ef4eddfc665b511f51ae5351ad&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266215756Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2688, Name: checkpoints/2688, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2688?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=2954a077c08b695127e6bfc9f4f0110d17517796487c2cf0156eae7a728667dc3caf50e064e49b45cc6eef94fb2ce177298b6dffdfe2107799f3e072cec7e4455f16f926d23651d5169bf7802236225c9e7c6a9f8b96599bbf4fa7593c8bd376ac8189c43e9b436f4d85ced30d080d48fcb735be46efb7f3787f88f72d7187a54eb4994792cecf7ea397fc5148d95b396b09bf572bb052ba188c0062face624539fbb02bf8b688339c5767032f940419715984a9efa583252a69f9d19251fa99124ee44da6006276c7387261d06f3d39274b0dce9163191e31c8bf725f14f3980062cf90448f1b7f7c91d5bb016647a21face04440e537ceba86088241b2f793&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266227584Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/d, Name: checkpoints/6080/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/6080/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0762f817c59a4aa24d1ad2c1ad8a201e2b7b0494d75b0c260d9216f67cd35ee371a91a798d0b2330330195a13ea9e1944efecff70c739bf1562d15acf3d115c2a23bb1902b3669ac7ca8da7bbd2bc33a855285232a50dfd611d1c486d11f9e0004da98dfd5328f014efeb33f879543f3d2fb97fab1f9e0a51a48cd0f3bc8ee6d791c4cbf923beb0e0b3a8d2a219df2100951ff787f420aa25fc7433708249ee9e0f48dd32c4d5b02f3df0e43e79782c64f21241d0f8a38263c2d11735b05940fdbb1fd8a87cd71ab79af3ccc7699a9bb87fec1243c9020528ea8c645463f6a2041946b6e158a1cd42f55024c8da15d3740d6e70fa0f229cfa457817f793210dc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26622754Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/ocdbt.process_0, Name: checkpoints/3328/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3328/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=242079622b42f2d7659f3dbf52135cf89c5672d4e7025616c970509e424a54d9dd13d4b7d505bd400c939f6ed197d5403787ea70517d0ac7cba8fbefc9300219ebe2950972634bf2ffd6de72f3086ba16b2ba7d763f842cd153b1be458c8c240eba2b2661c9161df85975539fd7517b898f118546df7f9817523d5e150c395f0e38c6b0d9adcdea6206970df5101e3d65e67fe0b16d9b4edd410ea1b493ff4869631f8595beddf308d5528ea267f1ef01aa1cfc7f00b718833a00d2df897fb4e18af03d84a8a07a956966ce20b9df9a60763859b601fca8d03e174e864d9aab6b3b235481278bb0ef3394b3bd5841815660db02316c7d3e9b771e6f8aacbb49f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260344131Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default/ocdbt.process_0, Name: checkpoints/4608/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4608/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7ce0c4fcb49cb93422ef2d5b2fa20b1a055cc046d6735ccd30d8d6a0346de4d0afbc87006da8721948aae86b0af73a323d4a1bf67ca5b2c82cdac23a871659cfba7a3a99a134a8f12e91796dfe53052c5fbbbb7da6274740a1a45d7b30639a7ab7411ce1e00c788768c4cb3304217c8a43006c64264180098dca910ad452f8f553300017065721a8cb2ae722ab4bac9a7c7a0963ff6ca48a8d40a9ce207b11806345a34f3c5d9e99961bbd3a8b8c16fcef8c1f496192acc6f48a7960236bfb611ed5b903c8eda6687ab5125b378d3feeb5113f19178099b27d054a7d288d836b7fdd83ef0ea721c89097806d1a8c97e94c6bb523958988075d809140dea26214&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266226378Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1344/default/d, Name: checkpoints/1344/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1344/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a1e1d0e7317ce425ed0c15540512216898e1b5523c6fd6293619c442850779857419d63b9ca9a5a2343e8d322399a28eb23dbeb0c4225ff38d3ab9768f6c0835b0a7089403e776601e18df028750453bfdf46adeaa0c6bfb3295089757eceb884f938acf1a864f83fdf28ce5aa3e441f0602f5150433327d6171ea7183b518023c716ee3292631d3b1ec285611973e3ade409f32a2fd44279efc1f95efe103c888892e9be49d5bb3744eb25fecf05b73f238a38d78629fed021cc63c559893b25d0f21ec2e7cf6781c090b5dc146336299cc32ab5d0655ba006c8373529a99ea1dc7bc5b5b73a4dccd0f54fc93b63fac5668785713c6ad89daf6c4c91bfaf0f3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266244331Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints, Name: checkpoints, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=330bbbdbe06f467be2907a479cac3748fe6099f32bddb728acb993d7fa887842e871b759a766748a431237022d7d1e5c5638371ea58a7dbd04ef2b1d9af529db11146f666344bcbc1955ce3fba93549be3a2b1474358401c95a68fce74079fb7f039bf32fed3acd6f8fe4537a616ad412bceb45a24917bd28dc6db50e276d3f5ab13f1679943f7e0db735fa6c798f504af97151ec1b646b58cc7b62a5c69602240709f9571c55d8a1d3ae623e9244126901544e380c85546a3c2b03b8f87f04ddb78e60dc91c73ba93c58cd372642e8d4798c2f014bb670899ca1780254aaf1041c15abf1206895168a24f52f682af32b8875c0337fb1accd91efd191efecb9e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266237345Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1216/default/ocdbt.process_0/d, Name: checkpoints/1216/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1216/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=24283b783532f58c3e0daad354d9f14f2cf7bd9eb53c653ad50e79eee7781674c417329082209afc378483ae66f5adc94ab338b7882df0f35eb94b79775a1b8146b7f4d03bcd291d7f7496d3d33835409ee75e094d4fd6163ce9a6a20d42f5ace9b4816ee37beac7fcb510b7fed8e2d9a46b62045b73ff850ba4d7f689aff3d8159d92ea90566da1fc102f9138948e2a55be7f35930f9a6915a50962a54d0d3c3daf19fb3b4643abd0e5ccc0c2c1b365ba5bfeb9110f40c9e94a7b4548a2f17d35237e8cf18b24b9c61832644b3fcdf3a3de7392ddd9ae6844a595a961b50ccc76dad15635949f06065bf3e4a7935038a66ad87a8d8130777cac42cb9f4593a8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266255186Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4992/default, Name: checkpoints/4992/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4992/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=87d72d920287bd2a3a0a3321b816eaafc9e4de71cf799e008cc1448192ccb70265af6dbcfbf8c08ef957134ae5a399c305ae7cf148b3faac03840d1a3f1bf711bde306448b006b6c76099c772ead3cfb5d788ed454f1e384724b7a512fe70e3676700b6b77a359853dd4d690ceaa4f621e9eb0a6a8517071795e1f2666dafb659a8168114df613f28b462b212f9d9e920f12068f4cdd2cd1e221b1940e4375bce05b5efebfe509c857ec023ca9d0b310acb21ff212038f0865153e97b06aaf8a0d17d44e877c3bc0b370abadffd9bb47ccaa121735d9d5cb0509be8d0b14f12f75ce39bd711a72195217430d6ede8ffd1be504e47744135921c157925db97f4c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266254737Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1728/default/ocdbt.process_0/d, Name: checkpoints/1728/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1728/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=2e481c836eec6ad6a589c8d8c0c838a00ee739b31f80bd51a537262330b60c251674ff34427c7af9f11124c1718cc43014e7276f48fad951044256ecf36f9cf8df5d9441b771fd9808f75eab698f2fd7fc63309ba18242c431b0e116b7f7e4fb8d4af755c0ef3fc5df315b58edcc4f6a521d8156fe049b5dfa06e86a80215a99e2ad77943d7df2367e5de999ad977a2a6be703723fc609bc8eb0fe3785d782aba26e86df6a2f0f81244458684d4f91fed6fa970368c00ab571ba0c2114eac2be4eff4b2eca4e8178bb1c33bd262afabf65b260752423e28ab326f6d6e9b0fa047e9a41d6b1355b1452c375a625735337d09e4c0c49077362d75a1dfba6c97572&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266259022Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0, Name: checkpoints/3584/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3584/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6b1642bdf8631a4bb7e7522478f3b89b25e866b1557506f74b26637b95c79703609f0bdfdcee51fbe2efde7681539142046de60fcbda57525ea46e72fce0310fe6269bea3a6c29c731c6ccbdcbb6824eafede28fe0e48a567ef5362e51990ab100a0db83cdf6ad46235a8cd7903023f869a6bafb2c19c43a476b713da234886f5fdef55b0f081dffb3272f95be013bcf2996b318a6868a0fb2c3c9f9806e18c6e613c28168d1dde5405d0153f9192d0d0f76009f7a7d430d0488e71d14b7e5b9bb6dff35df1a1a63aef92e03fb330fe934e5e071ec48dee0a36ce9bdd9b1a5c167149d3054539e832c830ac4e3746cf999473df244efe6dc41b9a1967a783108&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266273252Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3840/default, Name: checkpoints/3840/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3840/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0017e6d1845191abf45cd508d3f3694bc089c5d23ed01ed44db12a8fbe54768d1bbc7e1b9d058e487f95916711fade69bd859f798a2ef0d428f62ce235df9897a1755dbf8647fb3ce3cc1f8520532b2ce1253e63a2d14670b709fedd477d596038f0d13bc5468f604a883e5c5df738550d21649cd3bbf39c575e5ed18da20463417b05233fba48a00e278fa05d6c43acea981b718e3d87020cf7846f4a783676012cb50fa0df481c7f340824469ad06b6775d6591ee1e47fa419ed0e33ba3429eb8778f61438e2388c23d437756cb1fd1491c036a5a3724090267195333e3ad46f59a58196a3c83d6e0bdb3763534ad11be870cc94066345984c0862433dc4f6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266277559Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default, Name: checkpoints/960/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/960/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=945f43b2d92a9505c085c54c1c00729e36bcc6492e40ddd05eb97f4e581bd02706449cb380034f9ce80d6fbc0cdbda9a6a242fdd108287d4be47dc493b76116d37dcf31d5d970dfff2c085deda5d814c5df0d8a6f452a45d265b91fbc55e0534fbff17ebbe51baabdda12ef7798b77740349c0bc28d283c6dbed5af40599d13cde739c863754cc06834b5fd1d1bb03791aa9ebc18ed0e68da54af54efd8006fad6a93cb44960f1138ae60b6083d16151493a240a00163b8726f66c401f23abadd3f44f124224f3e79c93b97c07ef21acad5b9e1145e1689796a82d0143bfe6522bf4e0ce53a946742fecb13a8091118f4bdb38163fff9cd5186f06d93d8edf11&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266293314Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0/d, Name: checkpoints/6016/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/6016/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=4cee36bf32c6b2540e6eb4a531d13b27284ee41750b3aea887cb2bd6fea1c27b4ded0aa81a030eae7823994b93f6d1f311869fa83c2e28a3be1126f48c96518d370f1a9aedb66a8539302f611400e3c69899f69c001bf74bbe7fa7f9eefbb6d8138f653f71949e30926e4187ba995a6b909896828a24edb54ca1aa6b7fac6a55a412d2bb1944df88b5951fceb03e7b27d6eb856addef831c50653b2007c628b3ce8352f43a3373a3b9b28433276a2585a12f38950849ae29342fc415d74ede5dfaff1086fd548b9501426489ef6b9b73ceca81c9b9af49cc6397446fdeaa5420fc774da883c124d23a3272412b0112add3ca92bac63c458d8afe23875466cf58&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266292761Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/array_metadatas, Name: checkpoints/4864/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4864/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=9d8ec19a20574ab91a9217d9bc8e3968305d693701f2b623aa594ab51e4b6baf91a3097372b66839bec22249d0d03dbecc6c20578e1e09c3a1670438a8c69bf291c48630447cc9fdce270d156a4acd9744cb3fa377d5c963ca669fc35c8c6ae07193a9e9cda59fe0cc378fce846dc57c6343f0976401d12182415702b0252ea766c7edb9b0a8ad3b68cf3ec8cc0ad99de72dc4e73ef44d55dbfa9a6b5b0bb43423b39135a6890b6b6fa8b0cfb548b36bf7d5a416f47a5a0c1b624751e4c578163e9bc9853f880b21c777b031fa618699128aff98c2e721b4727ddc51a59354e85f098b0f5b170bf4a2ba94b4f42d4ce892b06af611a6f27c9f80e71a5d0abd8c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266305177Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default/ocdbt.process_0, Name: checkpoints/6080/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/6080/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=14296d332eeac3da3bd0cf97a57f5aff8457ee73702a19ce901744bc0b358741c411137e4bd35f57d01c2f21c29fc49d3389d6fa2a21006a0276488df1ffcac6cd60231f2aee66ceb76b477f9bab2cb59e8e0edf213fefb8041aa2bb327c9529b9801866ee31db46bacc2f61b3336d1d8a6d45ebd00fb5e2fcf097b98f5d2bdd5727dfd162d9a51fcd8016e6a316c1b5e4edd553eb665599ffbad7f1cc0927a3f31f4ca0f8c8319d0d6878b3196e94008d736012dc45d9eb03bff408a9475bc9b9aa31a4b8d7c82a6a768a6aec3e316dfc86b6cf32f137628ea7b5340e8989410bbd5a7c247f323e2a22cb4e68463d816347ccd6cf09f1a961eefab151209c94&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266303108Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default, Name: checkpoints/5440/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5440/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=bb01b00806b26940f917ac7226eb12c9613ae56e303dde3fd3fc2c647c6346f6830a0fd603595eae5677c4608a02affddc37a16b8492ad7d0f238f08ab1dfc535da975253f27a935f72f3e4f80d1929b5c19032ac702e5527e23eb6c6208b9ed90f8a4a78cf9d461b31b5474fbf8e7b6e79d7608dfda09c7d5100996095cc3f79a5f3a1ea72fb779cfbc7da33da436ea996bcf12019c9f7a66e9c214fbd9739178f84f3c2feb816da5eb06a3568c7a356049c99339e83795e73cf805274a633d69fe2a4ba3b205402f248f57f0a8b49a914d89f3e5578cdc6a63f5c520f35f0616dba213a51d853c7b09f524b04192f747550d190908a3037b79e6333a2e3081&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266300436Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376/default/d, Name: checkpoints/5376/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5376/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0fb2c5a30f969d835b8a60dc22f49cc4e9ec1901a1eb0f95d38869b73898bc0712062cc7a0a19ba5ecd1985a9bca307f8b6fdc8148378ad0a1435613dfac133e25cdcaaf3c1b6dcd3df1b4c9b7dd8718384d22f880adaadacfd82049b03b1070c3fa493acabe95c2ce1b41236fc38762b76c54618fb89e82d086a5334a54116ffa1f40f69b09b12a9e2dd4122c7fb0b1abc004c6a8a4ddc986884d8daf3d1fc9bd41b9cb551def15a06e74e60a956f88fa9a626c61845616c38ba1f1f7327449b6f201e02b47166663a3b22721a9329f739fbc36f17b0087e307e8124ca40f64bc8a358b9b20cf3bbb6509c6aa87c04a9ec918e797a2b7f8f5c5aeb06d9330d6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266315473Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672, Name: checkpoints/4672, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4672?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=53d1465d4edb47c92d02a23904b147db65516f4662170ee2449bf3afebf9e4e4cf234fbd6242c1da4c44f7c5cfbb91bd7c4a1b45e3106ae2340e54a26ed166450e9cba1d1eae6c5ccd18bbba0e449710967b13ca6699bfa5e7a9f8ec7dba866cdaaacd7731d4d140bbe90e1b1ac51a4589818a6c14433fc6d666544366e03f970192eda0e76d09ffa3c368c66dcfe40d0ffaf6695a20fb7a0241a4f9f406f5c0e0cf59ba74c13a71024b4a8d6f37243e3f8af53088a1b8abb660d9e0048f4b2465d731f6911632167f07eed6bbc3a3ae8349aa497e58b2e8ec7c07aa8d5f967f705ecb1513cb8d4bd285da488cbb01437d80300191770511d6e8eae9e9adf894&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266332268Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6080/default, Name: checkpoints/6080/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/6080/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=8f2ebde445c1ba6a00837bb364fe34da227dbcbfc1be9303a00ab55cbb5e30945c0bdf4a4bac60441eee89fdf41805f3c5397f7b92826a3621950faa4a6fe1214d329d7e27bba6369e28d0dae894f78b5b5aa48af49b9c0b8f20b6f3051f906e72b57efc24878eb23e32f502e8220c76a0823dd8f6436775994ca16e1f1d959977ced048f2e6b813533761a759bd8fa4d5958bab61c8de53a410b9dff1aa4be2d6e753555b380fdc3ea4abf168f79e71d2c3199f7ce022770212b50388a9d76245624f32911e48a8e1f7d965f4aa3a15c0ba0da4d4eb08f71276b4127651dadc35585831f96aa171f761de31851a8671c58b8fe552d860ff3272738101bec20d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266335048Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/array_metadatas, Name: checkpoints/3392/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3392/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b30dab9ac380bbbed0598dc9e8fa9a9cf80df4acc64f278609de72f8f3dc3674bdcca336a1ce8dae89bf49cc550a40e8bd5e631b51e2d871d2ec6153bf22551a551518f96462bd135ad9026c41ea9f5ac597c04c39003dfe0cc92f6539a8f8a52687563f713f53852d2a665c52474c9f8989ee755b8dc95f9ee5bfead1a609e1b9f80064848462517de4748b4a6938cbe6004a5a776a768c335a51c0d4321ae867eb96b363eaaa26ee197debd9ac652d07238a9cc2d05cfaad197f35bf677eb3f753e181cbe2f0cc9dcee5cb8c7b907e6ccc53785331819d8e31b629ea3386e84b8a464788ddff704dbcf96e18ef2918adf50d81c4d754f1fd9f69f5cf6a9905&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266334544Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default/array_metadatas, Name: checkpoints/1088/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1088/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=780bf9d67923fa0b1ff74908e40b99d92c427e4cb9e7875bc04c084da0a74f617e8565fca2c72845337fe170976f9134d38b0b84bb5cb31b1be40bbc8dc888b8373f38495d332adfcbe8f88be25a75e1fcc3c505017888f326241cd3bff09e4aacdfec783dea5c986fb0e2bdffef30f5259c62aad4016d767e92b2b5969fd013089e8cc6ebefe1be7953902f5533fe56b3308175405db627b7577a317071dcc829e3d89d0b596a6b180e73de8511a597f02e10297fc5e7b248f37d4b430efde51021e522f789852bcf5ea7d19a174f60a8027290aa48851ef943185ec7741e1dff84cab56abe10d42c6d69b5938b859c350658a1a99a3653b918573edb33fd55&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266338019Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/ocdbt.process_0, Name: checkpoints/6016/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/6016/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=2f1efc872d58d7d810956eb953adb2bbd185e6c84a1c628c17eeba94a0fab71a39102fd03770ff28517d0cfce2df5b67f83788b7c5830d969a0e3f41aa14c70ffc76c4ce43986e62d6720f54a160aa5535a026701ae6c86756cd7a7f1473e395d9b21a42f5c81d8c3951229920d1528f85c881f02c2ec6f2eeb825d3d14872fd29dcd2a6c4fa08f186627125ddbb701a1d626b56bc176b2fe14f0928c73b4e85f077c7d9cefd01336cb1411d3ab0764659b513e3e96dbbb07897a072606cb38b0c4fff06f3ddbecc8d4b63b50d70c7b635a9d0fbef0432354ae1ad59789902e3423a111ab47e052c7eb55669517d2abfac58a5db3a9463ac2fd4e8dd130baee1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266336398Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default, Name: checkpoints/3456/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3456/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=55a9d1df9270725d471ba83f44aaf317e83463e892bab010875fa54d997337da93a0970d9379c50f9e62c13835f347f7437e3c841bdd07c2e98b5fd98aa385fa97042498d5368006a2e6cf45bbcddc8c89f375af7aaa80f5b0fd1f4f9e592b7f3460af425e4ad116ccf264fbfb4882c24ea6accc23694ef0d222ad9df2f96b86e7e865e796bb0b54beccb379e98ea60cece5426b4a4790cdded43d0fccb0d964ebf02b394c268718e90e4bc007a533d97d8adaf7306707fcdaf5c6d2b977197fc5e076166fe9e0cc7dca8b0f0989bef797e2a5ca56a17ba5d37a7acff66bc736493f383293302de93b69f1a961d137a7dbbce1c73033a17789583f940da55aa8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266348552Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0, Name: checkpoints/4544/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4544/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=51dc2d3507dcd1263f22860c2dfaf9447b45df7166d619c513feb694fe02262941cfb8754058608f1947e17d322ac613d49eb10cc17550985fb3876518aedcbe2121a8d1f13eac30c8a457253b486858a93e4b7d9cc68f1e696368bdddc6f4862479fcd13af70a8659a61eb2dfa538961ed3d21d76e2fc25f5343cc0c8d9c1b5fd386cecbe8918f76f5382073c30cba58a093e1585821353c54e9343b7ee817f85e05b88c9ef21308456174d98456bea0d8a6179cfd94b70ac9bce99aa87ad0bc8a73e52ee1183b2fab9094b86722fb8d49145328fdbf0cf0189c633a847f6d5a9750498c413deea87c9ac78e088fe50a357f02a0f1d0c79c99b915ea32d6460&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26637031Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default, Name: checkpoints/256/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/256/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a37af62c3464a6e89463c8c5f05559109e6936bb83004fcd7b204df9cf552382f60e3bfee50aff0bbe856f12614fba139296532de28ef57ed49f6454b20574806e405d4789336eda090318078fba049e8ad13bca07507b7f3cf264f31b0d5f6a52607b61ecf4b250d49906f0a2521c6bdfdcbe68b7fc29820c26191c1e205794690eab705817d2fab1fec98fbfe58264d7cc92102f63061d28ef2a03c6df2b097e3598f8ebc1d3609983762d5e8ee8017c6e009e70a81d12a2a7a67d41a96e50630936ca573945961f008798e17d2c6592c15f73c7093c33101036913ccaa285a49625d2b6ada491afdddcbe9302a4c6c3c1827dc6d5d14e8f9bcf13fc96003d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266404372Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568, Name: checkpoints/5568, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5568?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=c453248ad923dc7affd0a977d2994e17076ebd751b961aa1352bc36c8b7298f41407cc2567618a7a56f07535f3f20e2972ae75b547bbac3f368812d9eb11aa5c149c0f9ddb9863a0ce488f64ba7f67bea73d9af051ebec1158dc2bc364eb92a1905452b3945f771ccfed2e7309560dd48c8aab8d06dcb65f99ad0a3e0e9297d64c67725e4454feacdbe06be08fe65129181fcd02304794f6865c4355cbb8826c4c8dcdc9fccb0c51d65c67e49890e6cabc6e38683b54ab177dfc3840914b98c5bf80c9ca0ee118b177ee286b7e2989bfeb88a884b3de78bdcf08ead133a50b45504c4e6de7158c3a5713f79e9cc586431a15b572aad6f8fe5ccbeda3863a0890&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266409785Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default/array_metadatas, Name: checkpoints/4800/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4800/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=749b53df648754c072acd519a0ca3523f3214ec9cf1e5de845f8cc9d959c4df02c62d83fb040e0a4e4d0ea42836a4f77bba05007da64308ed844dc133fd09bf74f5df3fb35f77db83ce9f0ff2a7a514820762a739b8f4b5ef1e3ff0ada08a7bb78350ebd5d4d83b8b9ad2ace13d49aac1eaf60817aa9f07e61a8d7b9f43549d4795fe9b38cec7dd2b4b542a38dd9c33ec2b0bebd2db82147fa5d23a90c38af8493ff614eaf3061bae24c777bcc5803f1deec9287dc9c31ddba47ab9c6b27d31939ffb87f141f86ea98871bc4cf6adea49449c8ae8e9b41a53ea12c3b78c32043128f3bce4b769b699bb0d152116945d4707b23e92c58559911bcda6a9d8b5cfb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266406803Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/array_metadatas, Name: checkpoints/2368/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2368/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7dc52cbf46d5092fad9bc3ba8e88009737d24c4e24f5626f6266b2a770272f8a8d5c64273f7784a40250d79bd0705fd4d50d68d08a74b0f299ab4132b5615d6c4560b9c28011162c63cfc9307fa9ec3c27d46e089c3e2e09f8b0f31f28a9eba357c381502fa64c7d9eca85c1db29d052ec72f75572c4e89e02d3be76ffdecbeaa650d577df248602d86c549540dd335fdb8e8c15a09449c29d3c0d01299ea2ef5d448e279098e2cc37dbaa5e34196588b2d4454327470fd68b0f8f4bd60384cc2963d9666e39bd4c2e1bcb289e78c22a3f72d97f9d8412cb129a259e97eae1cde49ab179c52710cb0c4fec5ab131c71d7c89c0edd4f2d634422acd88a6e78d6b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266420108Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default, Name: checkpoints/5312/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5312/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=259a143ac0b72a5709686345886a22245f8c568a24240c608e12e81b8e49f5d54eba0c86ceed545ef44e6c94cf523d988201a9e9bf1045e775cef3c0e832d757f54ae08fcac688af0b30c9f65d425fb110891e0a1e9d354086e25e12fe041132f9fa8d78af99ce22c0ea4767be52206d55666ec2454266594d70bc2ac69c1b803e9836e3a920fc84c086859a82f0b3ae7cccb607db85d3381edd20c837f52cd8b7fa5cd35fd5f2413b5fb0609c0a38a065ca954f640623dc12c3f0f1abc6bb3a17712ad77b511fe70fa3db73567312e93c43a8fb1e0350fc8d00e84b49e2088e37a8f0171c115ef48442bd371f3881c0ef81c49b1c7f80561c1d97ec32356342&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266443928Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5568/default/array_metadatas, Name: checkpoints/5568/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5568/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=9144a0f578613d05039d1d5e44757ef507523541172a4e92995bbf90b9a4701265b3043501c32b620ad85fc7508e8cd69eb41aa4d116bc1ba485b242d6c8e989878a21cdcd7d1adb0e6a422a1ac0b80f6522d66924e23c771512ee3afc324ce0406f45f851c85fe19ccaed8d04724ea518095c7cfd4cdbc2651de45839e45a838c031fcf5f1f70806e47bc51d5f5916762b76a50f6e3f862d7150ff3587249b0e5e2d5b7bf2ed273c11dcbfaa4fa19e5c22fb80cf7f58775441aecbc039c76ffe7bf1d4d2a956c009129fb39c579bbe668473d37c59c8c524ccea085bc3a2097222461ae629950c8de0f947ce1276b776cd33598c58e070f3de16f23dc149473&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266432161Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856, Name: checkpoints/1856, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1856?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=68e022ed86c45a5a9f504eebfa0c4ea619d0e07b086dd9f38b037745d1f6c729d899be7186b790f54ddca05878ab7eebd6737af7b8d8ed51a77dbec76933577022c1f8bd3649d0ada9762076ff89503b2635d78a5ffa438ae43cb961810af3859a9a8f6ad886f2142c1753201a6477d1da2f706208ff8c225e022bf8fde72edd6ec0000f82831cd5988baa5bb63c71745749a2962d2910d1887fade18ccfd10a136e3910e926fb7dd102321cfbb9291056295a6dbb6c5b5cc394ec2cbc9f00db55d08635bb2c6a838feef4b3d69e68c279fec21e007476706338336ffd3b7d952ac3172bf96d75e3ba1a045410415f401c60ada41cb0235665550273bd6e6ed7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260462187Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2240, Name: checkpoints/2240, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2240?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b705472424b7b90a93a19a1f5502d8cf757c9a23321a8bcaa2d885cda23d1b800d0bf338c0579454545c9c287d326a7cfb00cb8f983155692f3bf790bbfa815ea33b49faed27d4d341be449c61dba7f6dec6fd12685e79dee44b8c31d064a9870fdc608bc74b2e865a6467233e802eec4ab7dbe3834e4c8f02c94882559b5785f27c1f795dedfd14a2ff5680ed330ae597c6a501b5e0d69ece493cc4b88f2e394d11b1b2e8fe59822d43d2910f0fa006ef678be3408dc7a740109a9b1882c3f0c0a12331c4665a294de4ff24ea75ce76c6a59663103ff9c3d6afecc87fd0be919f1e89f3af48ce444740950dc5f20246df3399a6dca71a3601a333d4fdc2edd7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266458775Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2112/default, Name: checkpoints/2112/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2112/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=11dd3f173a473944f2c4b55844199fec7291f5f1d80b7641e5b4e9938dfded864fbfdbc7ef93b0b216f7419fac0475778e124da20db25c74858fc13f8377befad20246d41ff30042afd22b67918e7160599c8817bd7adeb127116b58b55b5978475e9f99dbfc78711293d749b009a78a5b6cc807a4e970f402cf9fc1d906afd786db128ad84f41fb1f7998682dbdec7994a70344c0bc69f0ecd8ae41e9b5302b60eb2e76effcc480d9f5d9010ce02480ed40d4033dd45b23d543a689ef3397a86057753946af570c100687fb0ea76da56eadaafed2d8bdef90641333d00a3ab850add5d7a812c7e50fa9ff7fbb3e88156548aad1f351d6b2914b0b62921e32eb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266467569Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5440/default/ocdbt.process_0/d, Name: checkpoints/5440/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5440/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=311b8434d7dad9b6a4756267f488a799e942e156cd8f4fcd1c2b12b92c4bd0657a980e8281699d64ec644132aa01703a8282871359c564e2ed3446851f5b644f564d29964dcd8ace51342a722ce154e38ae93e444c3af9923c9e1d94f0db43784bfa42f363b947b058ab62bdc63608888d67fbd28c126d7ff7f6a781ace92839e73cdcd50fe417cfad671540023cd28fa0c1cd39626b4bd13c663428b49123b5cc03cab674cebfdf34d6ca91163aab6e11b11c905044d3a7995bcd65989331415a1a211143b9380556f1215a605c45ee63a656fac8d68df30d4a32f627a6f6be7fbffe024e5589ca4ab3a8ca854b508ba1d623e859096a1f674ee518a36badd2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266459168Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1152/default, Name: checkpoints/1152/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1152/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=bd1c8d3b68133af78d9f93d10c5cc0553d031f18f3c77a8da78775e373c0e2ff26ec9b166fca9146b55435cc094273a3bde57b6c70bbda2da922fc438e0cce5985307594832d796bda95f309241a91c3723e3354f33f324b56e5d5b50d78fffd92708ec9f7c514cec4ba56ce33ff266f0d8e8561282de1afc56d70bfc94091038ed9264fefd589165e2d684111a71ffd6ad7538614476b4388b4f1286e5c3586b40c27f6f041679dac8fcb76cc0132935bede8196efdb4a87fd62b33e90c2afaabb831908ebcbabbdb0d722ff1ff3d087e324ffef8b4ca32dff1b21ac10d406a568773859db73de462e73462dcfe6ca847da6c1d1cc2b1a602af2de6f87fd77d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266456809Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960, Name: checkpoints/960, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/960?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=2ddbb4da63265f0f2a98449a88fd4f86a79271aa0e11ac51182035ef527cc90e7cbf3cf7758611e5be57c96cda2d2a0ecd57f97f69c531d5d11a188e891e7cad31bf78539cce8a14c666a569b3415c00af1174dba11fa54806e2c38132a2f53d4776384f3119c050be1dd0fdb7da13930b604f529a7f60aa6742d303e478a42eb49dbb429c3ec33da4a6a915e54acf34a5a2e8296696b93692c1087a241bb88a30c4e9328249b72ac88ee04ae0781ffb387381bb062820725b9980c36d364e4b69ed6ae38f9e2e3abd7e62dbd7f8944380a018b5b1e5531d7feb4e3f118ecc5e8c8f3188e4d92cd251b5eb32306271d892208985a6590101c7040fc00a1f4010&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266483311Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5376, Name: checkpoints/5376, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5376?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b18aeb13fa67df3c001288db83e68315bd376243ac81fc210e2cd0b332a69c9f62fde2d0549f54a65f74e081082d57501a00db17e2ab2c251f7c12b5c961cb11e4c697c5aea8f4d001660c6f37d57d6aa338ffb4a72a54978b24e14d446a2ad8a8f72380a7c7ee0729d8ebf54d147a30a25090018b777eb39e2a2a0173ea4b79f081e144ff863bb3f5bac2f6bf6d19d4faaf9d319795c75be3754de34111e98fa60c93bb4e7bd0ae2388c13dbfe998ea0674d0c8ca0c3161586f27e9ef1041f8af69edd7f487958fdfe773536d908c55d96127b441d8625e47c707721eda7e89d96f18d9b0861029543b49a4ef01a2fb0f91c2f98487c9be7225608fbe7ae9f1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266494268Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default/ocdbt.process_0/d, Name: checkpoints/3072/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3072/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=18516e912ad36439367c022575ce43d1bb6b6ec5544294a1103e400f11c2aa02a719aa7fae8c2177fe764564b52a8e83852d019e77ee808b75610522fa1f2b90246544796c51889f278b1c4de4099d8d7b66408212c0803d791ba41b9dd53469cb3722c944c72b2abbcdc4caeb8385b2c1247030fa8610df2032f63e3fc33775eac55993f11f913240c91040550da1e1ac02f7021ae1c2d75b664b1362002f17994c58beb1c65ea8b1860f2c482922d605c8c2465a3a86ae6321374f948b8ba5155d98532e7c58fc174d0e080ee297029ce1d887f7982620c7b14271b844290dd8e304c6ee980f77fef3cfd19e965f2c777a3004496fe6c20a6c429b6c3a21ea&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266504325Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800, Name: checkpoints/4800, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4800?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=5f08364ad01a057d4de5f04caba5989b942b2e4dbd6a310c8d7b26d7792eda5d07e2ab926f407fe271e49052019ab6539ad14e8d08a9b8c4e035857097e1b21148f8adde28c6abf48a984e8c2acbfaf1665395f56052059fff58636ebb3420f0252cb62bc6d63827780b034856745379cb07706d514846d98deffdc1e131a77711b34c5b1d0f24e787b88f2af5ef908b5cf189432b5236a2a22222962e37e85a1deffb213346350c84dad21f10fa0a61436284bbd2930ccf6fd34f4d486dee5256030412c49b58b2c2c4503fe1c65da76dbe1d278f6ce73e04233df5a98a2ae58c3fd4ccfdd8e595cb36ca14319cc73d87afe3bf39d521521f2e1c54cca3a26e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266524591Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5760/default/array_metadatas, Name: checkpoints/5760/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5760/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0b3789a4d0815c5e26520f3d351298dba2e90deba9c0913138b3f44829674bdefc94857e0503a3a4b77d842e06e67ac5583063c2f4b884237ba93220716b210fdcb923834178e1226ab1bcff1abf64380d45bb05bd351fcff308fcdad322d23b4add655fefa40a31220ffc415bf312f95fba83c9b78062ff24d78d627de13be4548d5d8344df7c7b12683d8031e067109da0d85e84bd1e7e446d2fe274601e101478d15dc5e77e86a1bc5ca728d724ea1aba93a6af0d95b31e4e0b31e492a73a9e10cfb503e8d2ef23f2fe101d3cebcd33c6fe2e049b249eb378c4dab59f76f250fc96c67b2f54d7c2f26545d163c132981dd2e0e1f2decbe09ae8fc275a8bef&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266510726Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/d, Name: checkpoints/0/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/0/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7dcd00ce75ffbf6d1f9d8922837e2727223c36836e4b08feb2a3880316ad89d6062b85666b183136faaac97b6f2ab6b7cf5b36b2f9a69b90466e5308ef9b2eda9280c8c79ddaccdd5f7827e099b3a58f92a818b44a4e422e26c70f94ae7fe1ac4b4e74c2519aba6676896a2b923562f411e4193949d1f64ecfa4c13ea01fec4259225bf871f9fd435bbf5dc4e8e76a215fb618938535111e0c02ee0fc6eab397a3dde2024187600a0e7cc29f42d9d7827368fb11ae544a7d382a6f0ab3af2a161cdf18af8fdcebba22a6cfb507566732ef17d0740a18386db3022ca28fb3f56ed028ff7003030eed2564187b80d52f0a5403ea21b2f5b20b55c21ef2c88d3b99&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26652918Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5952, Name: checkpoints/5952, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5952?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b8bf023c9854397a837256167af1ba34c9235d36c411cb9d249f594add2bf41114488f882e9ad1b5f579a0bf508df64ace81cc03a907afaf8d91dbe6eb19ca76510ab54f1328005f3b3daacf16043d21ca718abeee04d1cfde8dfc24777ee299a85a9ea12aaf71f2e22e39a3499cb90750ac30a99babd78ed63388b46ec0285f25737ab82b37fb8378187d68182eed9817ec4635352687761963c00c04c811def5e7a26664178c26e77ce9f9739fd056cadb86c666432a01e0b9010394bf3a726903e8d71fe1e0e3fa0854fe819db0197926106d08402c237d7eb77d8df42a6845b16914448c0f669adaeda7b8761182c33f81e12d2613142691ac8234e69d4e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266550742Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5248/default/array_metadatas, Name: checkpoints/5248/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5248/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=aa64813abcfc1116f6a58eabc4a247a804983ae5f4d29a28eef47596b7d0def24580840ed6aa616f60dd80d54f8d0db5c450eefaf24cd4d8a423c58abd57376526c1f7b90eaca0e4795b3b48797735cf88994d73efd77ec904b8b883891fb178f3fa8622e7a4787215f3da8037d5bce32051b43cc426feb9662d4357721986b5f0bac83deb48bd684da6e0f9ca9ceba93396a06a89d40dd6ad9e7c9380691a7b981fd54fff6af33fd7c99c119d4f33adae86e3c8a9beb5d56bee2a9f796f25d1aad36a1505ddcec8122b7655069d6b510a26a6fa89e4c78cb035eedaa672b575b8b7ce1dfff41f4273c3f7eca3b52f8a682e276d75913b2e829fa58bb78f253c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.260465478Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4800/default, Name: checkpoints/4800/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4800/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=5153601dcb83b9daf41cfa2bb416cf4e64a99f2d48e5d36ea1180c8e39c93005f7a503d8e1e62b5b5863ab891a41386199f22b6784f576a9c895a7686c28c4facaed9c3129d229aa860ef52a8c66a5487b2951280ea851a270856e0bd3de057640056fbae344a005e53c231aa952b519af6be3475968f5cf261fc3f84280eaf710b4eb09dfc0a44f7a9203b19600eaea364245a702771ad1abf5f241aa5f97b07a4b9bf54d0d2ddd63561a1db4ae10d95e9b1aa783b6eee2fcb7c7a3f7c220708dc1bb5c72853b7879b251bf5a79475c9bcd279eeec491d27cba00281b86af9467637e99c6067c13d75553c7ad507d6cd6bf83ab1c228be4762e15e59009308e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266562502Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default, Name: checkpoints/5632/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5632/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=2491d861596cde01b9eb4f46592d8195431372d6d63733a8440fd3dfa91408c6c54469d894f1251a013ba1a14dd4a5bed17d3fbd0371aaafe98523ba1296df8a5a2abb3221f6c8b6da6f01914e9b8b8f360642a5e38c84457e8ada9b786a0156907bbe9b2c4b9400831085ba679b2fe863a5a73b64223b53cedf2916ea1f684f0f82eeee7921510f75020c718f5115f1447f2294f6e14cfd10497d04db8ed02445073e4a4d40b54154311b2d8a3b0c7f32ab516864c22170a5c1bf599a9379099fedc123e26bf4111a1ae540a8b40ba6b5ddfaa3c86e86e3df08f6708430ad793cce544be8f9a898765f83912339e674b5dad6d51c034e1b99f26b6e516ff496&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266554764Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2816/default/array_metadatas, Name: checkpoints/2816/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2816/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=ad17952383033888fd20967e7bc1b2fd82f105ec9da9ea95441f786a2fc28b9f03e8336dc16a72b9a1330ba2a7faa88c097f474c509470af9a6a02e897c4a56548308d738986d6eaf199d45222686fd8a974b64ad1f2f307d018d96b6c2986add4244a74d73efa0094f57da503c943e167c55e31ed70bc7a4f789a3b1b5982c28752adeaec9d3cab45f1d2ae9e168831747d68ef6fa7876502e079b107e813d18139de97c84c0b3517c890afb058e5a57fed17396889e8ebe052c871810486f3f699f25c696317900ebf2c24ae93946c8f5759b196d1792ab1dca1c4bab21a44c41f13fe4b4dac1a1bc9ebcdfcc99f4b2c0904792e3557433f30aa985196d064&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266556394Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4608/default, Name: checkpoints/4608/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4608/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a1c4c0737e9052458d9a7e8d3e86b41e36d0fc446569615f3dc93505d8eaa00b126893b07ae463ec5f05343b3191c5b3fb49968001f4a93066fdc91bf183cbebd09e3191f10dc78650d12e38470cf54fd03d37e217b0d899e1089710d0a8611f9f35be33df6a301141a4c008c962d5f07a26cb8c9cca93a27670cb4864319affcb53c1cdad23083d5c8711f9a2594684dde02c12ec17793b2c043cb3d1bf5fa65fdccb718fe0b81de251df075ff1f101f08c6d778a3c2bcdf471f957c22c16bcb45afdb3c3521c397f8e27545e56a6b45d71b6c40e7383d70658ca4d5575d7444eb64c9102ae332250fedeb40ac7068b947672f3d24abee5bd5745b5321c9666&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266568322Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/d, Name: checkpoints/3456/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3456/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=25a4a0e16d5c4f1cbd5186afbd9626f160f9361604bd490bf4bb867282dda315d44d7ffeee33816af3c45c6000d9d8074224d14cd19ba29d25d1500de324ff27920b18c1301030be6425df31b56b900876a8bf3fa77676c6fd2beea3108d130016d3e1fb437fe216c7a749673d3ae836a58ee5d551a18de6362d85a62e60effb3205a71a56ea72b48af0e2fdb49e02a11d4906dafd478d1e9126fb15b8ff7776bf567ca7022a4212407e7200a6b13ef2a813401960cc451438f36f7a6926a4978c1dc6f00915fde917b0ee6fe007c346fdf8aa37eaadf073a87546bd2f0481e7dd4fbc44776e6e41e24f0e030b9f6bcdd5536da7cd2be4b3fd6e44ccd08c86c2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266571981Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/d, Name: checkpoints/2176/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2176/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=865e095a44a21da4125ad753329672afc63b8bbda4aa318d6c259bbc1df55068ad2b97a10e158c4d9199cdb8bdfed95b1861692269436224b8a58f6b14df07c36e4ec8bd703a26a7cfe4e9199fcf7a190e12d6c28cf6815dec83c08410c3991342c98844c00311aa8e2145f4eb8fdd89691a481012fb787d23d3f25166cc9827ac3c357e83617849942a006cf4d794b9ca80868812f307daaaa82f55552ec753c32438b40901570e98dd6c5b282cf27f7a670c82195285a170f2d4a1a3e887c459a71ba8d1bfcbb2a87732377d992b7e09f56a22c09e6bd6c0d76e7919dbe516389853af3226c11fe81b67558ebb065e121640f92880d6cd9c5ac8c9a5baeb73&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266590027Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/array_metadatas, Name: checkpoints/4224/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4224/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a656498ccc8e30ecd613503f4baea48c67ce091072c2d0bf0217643ca4a4365f59c3b51451b93c9078094ef3823679bac2418bd026f8a147922843ee08e19280454e2cbb250c402e86a7440237723b8b0280f584f2e9acde775f4e1509fee50f93d2fe1fa91b1696058a6f1125cc1c631665e48995d37817f183c7204471c7183c83c9d0144ce2dfe72df8e39d668cf722feb16bdf841d50d95c37f779edd2181daf957710ab6e651089ad693b9dad6b67920c0ce74fb39beaee84775777e922b0a274aa979adbdea44d97c3f4e54e7908eb420ed281094d4cf714a9d4859d4b4e24055f47d1068823e77c60aa3b63a0158f3875088ff555a57d1353790461e1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266595493Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1984, Name: checkpoints/1984, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1984?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0d103c1059800942b716c54a14e607b6677d54713b09a11d47a7ce3cc669f98cb54c448bf5c1035aba60be97ecd21e2148e71dd51e0dee648933b17e1fabde6e0027828476bee69aa131fc6ff9e0489960300e4a253fc240e106a4871b6fa605398dd349da9b35eeb6489a6f0c4fcb60ce79d9e0d72c09ae55f1a5907c515ede1d00148e636a7f990f45bc3a813e699156a4bed669bef83936b59ba7497f17700f8660441a1df703bca3a7096c64526371834e53f5965d9bf660ac7065d23a4e81aaad880e3b336782e789260e720123c8fcdb03d3b67be8a907648364fa7f2b1abe986316c29787b25bb14fc13a94087e555acff1e22e00c4a28543d258f171&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266606084Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2176/default/ocdbt.process_0, Name: checkpoints/2176/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2176/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b0d628d6e4140ef0caf5492ad9ffbd1211b1796d06563b2a1909a2e4e3e0ffe2a54109781314a8a180151942f39a12b932197cd6893d1e75f5f6e0b759c8548526bbf986a094641260981ccb666e5873efc3d52b1ec305f8d1eb809e919c7b0eba007417aeaca09cc09c39b4523e36031137a85514a5264eeca18a0611b5301b2dff71051456cbe87dce0730e22765d03d51db26a1cf0b93f50f0171ef5792232bff2375637e39c2ab04fa518e91e1ffc43dfd4c95c897dca4b5a91856dc1e314645eb24c0ed4503f388078ce12eb6888c149e00834705b1e3f87c591dc1c928a5fb70bad1ae3cce6b7a4d41609d6643e0e08ed91dae480b0ef26119e2101710&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266605323Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/256/default/ocdbt.process_0, Name: checkpoints/256/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/256/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=99977f065a4201feba75c6df51bac8c570e926019e925e2918ef02c10b3a1c2ab9ff48ca1d50d35f2f466de8c96832104621812d063f65441f7baade47137c2ead22fa50073890006088b369e49648bf3a5ac8cc6ebdb401ce7d3c9d8e818a070cbe78b0ee80fb3d8ea2b90ea8f7e8d4a3b80e558d17080186710f617dfb0eb55f6f3e8729ed8f460d55b771b3d003f11a52b67b0e623bb6013f5365d7765c87954e5ef9b3efd2021dbccf87d4f10ac4c6963ecd958cb020393e7fa6326b8f409f138aaa5ea906975a639a699b6970473e3e1a79beaf8e049a272f8037c78bf19cb76e48e089e5993c8a8888eda7c7ff6627e90c23786b5efc869e90e83cb018&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266610769Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4288/default/ocdbt.process_0/d, Name: checkpoints/4288/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4288/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=16829c2b9c307da5b51f57aa327864a69630f74ddf4e170447cc88091a9bffe69579c760b67641eeeb4d96361b39b07102de23d12663fdb30097dde4244614040d254a4ef4ab32379435b55551fd0446c7022be6aacd3af05cbb9e2691e16704e53193171462cd11677f343b24dba42f91ee0b4242d542b425588ecd8fde549f78250c97f66ecf5076a773a41cb041d452f11feb3597bada56cb549fc700e187a7a4e7d70f66a50370aa6c6b0203934c918b01ccaacf9e10adb007047a5c2c48d7cfc1cbc3f19d1e5f48b73f714d1868771f31ab6dec02efe6181f7f3736a0805c45b350b35d7792bc7da0c53ca89716da8486a167fbaa2427854ccf9fa1d0dd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266615296Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/ocdbt.process_0/d, Name: checkpoints/448/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/448/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=0c5cf24eb50cd94db30ab7cf8f78973e9ef8e1996dfe77ade26e77df45905e869922253cb09e49f1b79f5ade31de562a7e4253a1f0de36cd928db9e35e06bffd505f19c30d7362ecb47842a293abd361b85e2b79271c3788998bd83adb29ce6ae59b0046ff89c5e9f903f8ba48b1158dfd0c0e012db01d183d73a19651893b4315df37e3684470a50e466980308b35784c4088ccba271d37d49fbea729f352d6439f27175e210380fe49cbdeffbf15a5d6be5028b298d3e2596defbfc3d681f3ece8f1256b6f3396c63de98179652d389c2559808d62041af9912be4c1067e056b083716340e93c8ebd4c7257f8cf93fedee996c87108e16e93b8933a0d6cc47&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266619687Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5312/default/ocdbt.process_0, Name: checkpoints/5312/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5312/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6a8d3f77fc7fed111745e4139967904b77a2b98e9597c3d189b1df3ade188b73d53b0f4902b776fa5450cf941ac7750f073735057e25a2092c3addb31dc067e30dc20e779c5206bfbf80d55d67bb549e874991104eeeaa52d1408c8c86c06556c8aeb0e034d76bce9789f0b212d52097139420b41055911acf19be195da400f364b4e43d5ec882479065afde993b0ba2e63d20ba7ea99eaf1062bb87b0c42713c54d1dd677c35536e30ab503b6beb04927e98456be5e084aa204de4e5652cd8693b9d31724a6c29e5f7cfc80acca70d29b2a5a7a57c50efde381411f58c775b64a9f0edb42ed183498ea1d4b0ac808c0b7144fae2b28e08ff598d2256b3c45fc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266632386Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4544/default/ocdbt.process_0/d, Name: checkpoints/4544/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4544/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=9db5a4213f7c2471c8aa6b2d47f2cccee324ef18671bc305404f8780bfc67d31cfbb7d0b936bb5bdf1884159b1bc551eb876e8f6c13e27b3f0cf3cba0a8adc3ec00edbd4d4e0984c8d74f4eb0f2c15b247498213b2b241a0aa4c6f161fa2626f5a22a9196edb1d53047b22bf77bc40d2d36cc9ce95ac52ab200b233d82fe6a130dc13d39970a25fac1ae5c354d894592debad46ed7a9ce0a9706708e24315869089b581d4f602540218cfe191d6fc88aeaf90c3965daf7487dc9f65d7101a74db6c517c4a6a096ad1ed635c7c38fc972c9d74a081f1feb21c107c8ae6a7766766c0d6301b025b256d9b720ed04715e1b6742b8fddbbd69d1d90dce018ea08f46&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266637455Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default/ocdbt.process_0, Name: checkpoints/4416/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4416/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=5bdd8cd46e9969996618fa7810c84c941f4d173db81e04ff6fc8fdff280b0d76d08e70c7e5ce416773dbbcdff3c4b101e2bbb4dc6872091ef5744313d92fb00827b9d3129f093487413d2455c9f101c0a706f22270c3974601885afec40f2393fd93899c0cff3d2f3e3b460dc8d9f9a1cc200c48b5ceb77aec0f5dcde693a01ef16f1dca888af7350d0be8ada3971c5916439e20145fa0dd0e4ee2377f615f4e7f6cc65ad58e3e3911463c8c34f6e6f68795d9127bb1c357ef77088acee959ca7359ef4e1bbe7fa0d05b98470a507a890acf1d9e1ddd798d2329949312173054dde8541675a62e13ba03ef16e5ec4c98cd6986e0b3815da13388d47715da5196&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26664808Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/array_metadatas, Name: checkpoints/960/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/960/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b79f5e297aa3b500a93bd4047f508614a39e15a77836a01fe1bc86b322d7de9f9fb33b45d9a247e211eebe19e2bd5f9b6f293281459cc19046675f2aad9affd705ab6bace71d89d0f4c70ff7c552566163afcda7e81d3bd3ff8d1db18f73cd96a6ccca1126d9d349ddad7ff009a385ea099b498981510916522ff2824e842d48fadb8127c3cdde8e148ab6b89cb83ef6ad711e5929ab1e9b6c8ad915ea9dec517faa4b7faccfb648f36db2cfc6757418aa1661878b61eec0345a176a106f53ec59aebd2d16f5dd45304c6b1a9f8ba21db6f91d4dcd2e83b997347f5c5db6007ebd2324ce1efe24fa22f93373788ca458ef09d8602997d3e3d917efa75a7108e5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266656952Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/448/default/array_metadatas, Name: checkpoints/448/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/448/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=8a3139fb1b2575fb641b9ea09723baf68ba5d25d3ba70fa925f63fb6220ffd5267dfbf4b2186d3d7f0f61f36c528bb3da0d7b25d0806b8b4ea5cfdb1a32296d1d0ed523ba4384437038df90485d7a8d1a82d4b842f27eacda5e5d817ed47e8fb5ea87be8db681dda544c0904013400097a3700aeecbf43cc1af6a5125aee0a2d929f7a9a4fde32e9d5a3098125f42ce1d9b17aea8edb04570d5ce6c2aa5ee3a097a49729e552d4b92017967f28598e61590e22eccceb3243fb6c71b353feed427d9a06c4f6c7a231ce0fa58c1b571a65ee9e778daa68150d15b84a12f5202a0dd1de9930158a967c7083147f8effbfdfe708e1d174bf064667fd98421ccc036d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266664087Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4416/default, Name: checkpoints/4416/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4416/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=297c65a2c4d83a6a1e0335fe84726e0b95268c73583b2319d686ee430a49bbdd5d682809c7fb3e2fa9e1350d88443016203b490b0c58288cad3b9b1731941c81a3e5bb850edba52976ff5385da490a0fa1c495188a8f52b4f0ae0171e8d91ab19811e6c599adfaf213f6d3d1c9bedf95a3188a29f334033a3ad2212503ff4feb5702ab35c38eb1df784d33a2518c7dcbcd2a6b475bfd1441a83df03a13e0cd0dca59a52cffdf772c65d05bdb660fff452b13c464cdb7a7b05cd407bee0d11caff5dd12727c1e0b714884dea02a7f9f8c27d4355e91c3e40a871e33d179d551efb0c5eb2e1cea64b9c18887f8db6204bee9a82fed10e578eb6fc0135e7676ccba&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266662389Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64, Name: checkpoints/64, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/64?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=869fc943131da26bb253bcd69a5b9af488846700d9b0e9490442f4327979ac7e8761bf5ce1a729166a0defc89d7317c559c68c0bef764456bacb90cf968fad9e550c25146198e6dc065a93ff3216089b5d14d05890b288ef26e40fb0e7a24a7468d991cc63e2afdf3e704cd092e8ae477e7ef1955621faafa8dc6aeaa7567e041ec9f43cfa03730c2790fe484f98162521ad3ffbfe80018798cb613a4a6d126097a4771d798a34a438bed19551cfc10daa9acd75f454df5ffc0f381c7de0c953981e2b3a5431f5a6becdac6f1903a39444763060f302a92e53b0dcd1fc7c600f4b0c0bd138da42a760ddfbda02c0688c39cd6103f1ab68a921ba1d798cac0880&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266668158Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4672/default, Name: checkpoints/4672/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4672/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=8deec209e5e42c8bf2648594bd1ef1b9b080253c7c05b57b9b15eae3b521a4d3ede40dc7c24678517e1302fe2749c34e78efbfea3b296cb800aa563a4075bd32e9af6f2407f62621597c2872b2d87e58e765f71bc0636973dd7abaf924a2ff07f981b735d2ea9e6c871def61153259548414b087c0b1f0cb7ccde4caf604de42721943a9e2e6e9d478a3c517535743cb4e6b5a22a83cfd4b437f89c83674dd501f56c0416e23844dfd544dc82ba66bfe5c38eedd11107242c0283a25b3b92ec177e5552929b559ac353f7ed2d64c8ae9eb2ed66f61b589c3d007db01b325d37ca18171c273f598701c1d794d26cc9e29eb927933cdd6bcf65314a4db8c03a674&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266682097Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/896/default/ocdbt.process_0, Name: checkpoints/896/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/896/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=60ee222e801f7daa5e260bb56ee5fe0893eea26b3991f57f84dab61cea4309a50fc60c6470da20bb1f16f331e03b972d36c1086151d968e677d14d2263e07f3dfecf21d652266632aa38ac725f0ec6d60b04c7ad8fd69a59ee981a04f67c080c56ada8a3f6707d00b7882062470339a5b919935bc2d338ed2b19654fb7206fbfc79110775f1a7a2b444cf136f959a6a0353decc30a8473614c0256031bf2839c5c500702e7ffde418a032581d067f55de77cc404ef0dd3cdeaa37488f0ee087dacc2ec12bfc7e746df8c987fa6ac955c38c1826255653e2c2bc67c8563db8bba4ba128f4cef2c6f2a705e2f6b73733bf5bf99889f453f67756a2f513cbd6863e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266698795Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/192, Name: checkpoints/192, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/192?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=18716ff667397241ba45a40acda3b7a20664e5a529b075a99d443054c54556665b68b20011a26fd2c2c3892e4687a5d2d41830820238395b191052ed6faa0d49e7c4b7192b16e22dc6e33e61ba94ba945ce97abc038c4b1788339a8fb3a396a7357928e1da973fcead9ffa144c0c0064c06969d271ee49e67237612e94f7e55ba7d2a0fc1927b51009f8844680b3d2006b3e898f00866ea8ddb203ee67d2c6fa0b9fadcd1046338b2de3466b8cece5b8f534bf0c1df1417d873d9d7a0677f0a3dff2024add7359adf71f81f954d22ac13238fb5e54d54062d8b49af15c8b3e491ae5b726afddd718feb9c1e2e70d147fefa2a9189bd20e8b42be0d5ac13101a6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266726948Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2624/default/ocdbt.process_0, Name: checkpoints/2624/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2624/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=590957018d2c394800c263ea5d96c56cee7cb3fe9f1434ec6c52fe8beac1d433bf5b1592fa9f070def65a83c919ce784c5f202ed5de018580e3e5387f21712b0d23f8f0ca064430be56281b47528c243d84b272f2d52caec8e60add7d4048c2bc58d90b195633d389db27ba1efe44c1c44feb1e571c1dd1f4aadd2c587b7fd66201bb83ebb28f750a00d88de6433343a5fc29b16f664b5cdb003e86c9f7eeeaaed6da416994e48c4381131ae7f5a5a54229551073803aa7b140d5727de30329451b86d4d916bd03956881765ca83b024fe4a98ffbee957457ea01aa5b2551ead1b6de93531e8aa490247716a7a10303dae1fef245f24e0ad355c80d162956984&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266709943Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default, Name: checkpoints/3328/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3328/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=84fd285ddc569cc2aaa08aea90e1595dc20007a28869996419c2ebca3a547ca582b658c8d4c635f1c20d10a62f6a5714fcb78660f513e448c489e5124170d8b03e09910d6ba0daa3bb994369b8c0c6e68ea7b112017776d2eb56e3ce8f48a904464b8fbba6d6c146c2720ce0ff088dd09e5a31e730aeb7803527aaa0785f296e700cdb716016d721d6e7e614cdd25639159296c57ecb2b40819f3f6eaeb74f2651b46ce2ea014d45a7449f34079ab087a584589bbf5f8e09a2f8ae95c113793af9bdc541c410a6886db17aeaad558c8df16153fd8dd37e9919fb86f7af8d009aefb3e0f0ea0e8b3b945f9a0e229f73c6228e7ef34bc5334833456bada7c5da4b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266742327Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5184/default/d, Name: checkpoints/5184/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5184/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b76beff3af9ec713175377457c4190c94c1785933d3d5b617cd9339cf32fb8372596f00202cb50ed7218db67eabbe2254d229b7ac46b3063177ab3694dea366210b42291334fe2da905d1f794a5d4b9a195194f68990ddda60b192a91ed7172e65f1e31cf362b2d05d58f5a2ae90c3ac4a0dbdba0fb89bad46b61a37889a4d4f294a9d65e25aff0f166213ea5926e0a3eb99addb08ff7fb89e394c4ba47db2303ad6a3ee896fc5855ed43e79c1369272f1824057318b9a62d024753ff5b912ef5c9587ad4945115bc1bcad5cf3f25aaf43b93f426279ff4453ffef9fda6097ff0eb016f07fede97db5e00af76a525523a8942561f61ca994e0a46bce8888722d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266752643Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1792, Name: checkpoints/1792, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1792?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=615eda39a8c60ce41ed08cd6c14769ed328872a8e0647bbb8e4bd19a972341022a48c830b19afc940a1aaf668fa333b0f621005e78bbfd857c161bdd8a431efca811446d4456dfe2d12b04457c10392027f50d2922c8bd59fa2573a59f39e4cd24d066a3f9e25bbee0be2ec61d2a37a36fe56a76b0abe85216ff5ae24209e2b533b827eef0a92429272ecf77bc3a8720c372670fe04ddbca7d95c553fb90808d8d19a792651a25b7076985d00c2b3cbf94c34152350852345dbab3a1121df7124324b717aaa82c0c4359a384e72a1bc2b65f6d277bad647b83917432ec069f1612ea1e0ce6f9c5328787d7d6dccaf78a0bebe2d7306fe802aaa9d00a050a0bfa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266761703Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1664/default, Name: checkpoints/1664/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1664/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=839c3522a4989b24414e9a88a6d31602f4c81a9dbb52e380ba153885c188363d5dd09ba1df03fdf041841d2fc2cfa22b157cbef963a4e4faff9db7870d298828e4369495eaab7493a1176395abcc77dc17e09e258d456ca758c4c09e8998a4fbe70af79741fdad7c10a9e9d34773c828d073eb3f9565bee64930dc46fe445cdfe74df597d571b4e9131bc766035f4283d0c850f2ce2a84d0e47f645894856b956030ac98f07dcbf2dc80c5ed40351421593ff5811274a2f1ac5c6e206c6632416f20de632db604a9d887f8785dbb8f315d5d7e15db53a842830c0cc230bf59dcb68c09f308e03ff5238c3ee77ca840a655a19ba39e4bb2fe04ccb4451ec35a76&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266778189Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1088/default, Name: checkpoints/1088/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1088/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6d4b10cd232dc2372014dfc12ac3a9658a4b3836976483c9805094acaa41fe3fa43b54bbdcb289fd667838d5a1e8be46d7feaef86d7755a3dfb4fc426c280bf290a847e6a32ee9b376229d093ee1f8fb134b2a1a8d5eb0308df4ed916142f9363e356841c5642a119d44c2078d03ddfba0accc8cf0311805e3eb73fb2e31bdc959484ba3df28632cbc092fa034741052b3d3732effb29b2b6733006373aa8a41a621b1c64adbdd35179098ee0be757310d89fcc7a76513266e86ae363aa7b181138585b3dde5ffc8cc4a106b9d8fa3ad824802ebe35dea8421f390271128d0d6ac4547c8fabf3bbfb5b8939b532b97649bb5effef227839e5a07829179009578&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266777099Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4160/default/ocdbt.process_0/d, Name: checkpoints/4160/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4160/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7dc654a710a863aa245fbcfc973139e89bb2f2ef36de09334dad9572fc4e1ac174a09a4f3f9c6725ad6d021154ceca8f0d575848d83fc6cc7e01877edcf1bb56cf88241008dc97da5d63bc139f91d8bf98f657ba629051041f0da815d96d553dcbf2743169205aac4469615d9548245b4b4b292af89e7fbb582bfd19a7d6518ea3bde9a54064b779f3dc7aed6821647fdacb2cbc22ab619ba014bd09275ecd64855c5f6d425fd72af0ab3e0526e156da835ccbcd990b478842b0ce5d1d97659059f7a713cf1a5861c247d3745ca51726104ed91ebf533518d2f734e48b522d1a40ee281a70980ec8280bc16005640e2c737871daa2b25e9175a7b80677599409&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266823651Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/ocdbt.process_0/d, Name: checkpoints/1280/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1280/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=64710faabb0798c3994b538e5e4eb5b34471e328176d90f033674b66982d2062e70d1e403f19f20c66dc6b8c9bc214cee0b1f39e1ef88f693f3a036258034060a3250633b62d2a4b535595e892542c67ab7d9749f42e0412e1a799aba0e5592a8cf90592849b474fdbc01fefaef13f5a2fa210968da96319dad26631a831e5619d3b12b61f87c2a4df39253157a8a62a4c0d7bddbfdf7b2a52fff172b8ccea655791566b5b958a02a1d9efabab640f7ac908a11089aaf773cda041e55bfbfe78db84f5c69e28624add962a06e0e6f8aa650f82d96b37419afdcbafafd0c06d46ddf90e1e662e038a85beea373abf315c6defb623c68c623cd459594326ba450a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266822097Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5824, Name: checkpoints/5824, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5824?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=8719636acbbb52adffa48061dbe6b37f309209037413317d5e476dafa02ca75c679ef360073e8781b5a97658d61a4d3f9cc35c53be5e59a013636c37e18c736c5f0f382aed6dd4ecf82adf0a8ea2228965be20ef9a333c35de81e18bde994f6faa7a5123e87eca9c8b21e0d3859b6cfb220f12252af6f88563786805a7eef6c8499bf549e650721bcea83f9e89333896d687c320ee4de6ed703377fb8836446615acd5750048bce32f84f46474f4213f87e563fc5efbe9dbb6f2f6ccedd29be106675e65c508ca5f60b4206c5bf850f2ee78be2ef3c125526bd36e8b205e349581fec8b7e0817bcf07eb26b0ed06e95c8215f9f563a33a3195514b056df7d44b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266837983Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4864/default/ocdbt.process_0, Name: checkpoints/4864/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4864/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=17850f6cb839468aeb72904a0ced5416b7637b970a7cea3e747ac3428c95afb9d5a7ef10c16e9a96b72bc75b387fcba936a261d8af74c2d532e7cf9b60bcea28ed7c3882ee8475f01034b25c40f14bb70af0596c71bf727d8199d5125f2192e891d1bcb57adf8abc3cfca8d15102e06b7fb7492d86066fe0a8fcb566bbf2f1614cf098604fd459f42c9b879d7552a66546b2a8600229251558822d577c6e1d5e28020a713e32d453d3460f411730f84a0f5311e6e0211f407ec0d56b5f57cf45b97e25a50572b3984cf409e2891bb1d5dea16d754b5ffde234462120e5413c2bd93ff61472f6f5e2d66cc5d2e9675a9f0a535967c670292385c35715f7e0d591&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266861522Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2432/default/array_metadatas, Name: checkpoints/2432/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2432/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6f18469206d916555b17fa0da7a51be4f70e9c3584d0048b1ec98525c0b14bcf78e9c42b7c9362608dcd50b3b9540608f7232d6d989fcbaaafa4552bdffb1f85a2489d58d384cf66c5f1b95f8b7d6693e2758a5df60cc4b99224763db41b7fd1cca2d64f460f8ffe189abe5485df56fa1d9dec85d56650a936f399118803aab37c6d827ce279a6dac397920414f9a384f0890c529364eb4879674a689bf1f27092365e297bcf5eacfcd23fbb50686c2089f4abc4a1248c67961276913b62acec00df9f59ebee2c32df51b01be6d0c1d69cf075870845cfd52b33a3a87f2088d225fa81ab9be2085199936c95cdb6f3caa2fb54ef36d6393831728369d3bdf9d4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266843118Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4096/default/d, Name: checkpoints/4096/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4096/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=ad6f4851dd28c871ef71b16bc4aff636e48c6318f79355d987503ad4af2c712bdb564ce7a66fd74a3a05cc979ddb7d9a01580291f993bafbc46473665b85fc48e59e42019f9b5289fd1c539aa1034716785e11586afc386f8c251d60267c47e3405d85db3c0de30b5ad19929fb5be139bc52443242430a37e15df5b06614760ef0e15107efbabc678eb03915e36a4952a9454c352441cdb0d84ae6ce608e5c61a0a51942527439666a0507af31b9eab8ed2bdc3cbdd1688fe76f8a4700b9e6bf31a7a4ed86fc3d4bb3e58e21bc45c7142b275663e4ba54bff565ddd0f9923501c6009876abc240ac7f72467843786c5fa22c0f361b292428a5b6ec274a07581e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26685337Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/704/default/d, Name: checkpoints/704/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/704/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=1e4cd78dea0073b0b342134738e6df53685b472cf6c16dcbc4c872a97c2482564f6f9a159024f2c7ecd6e008ea273c4e3453e7baccbac402bc0c375fc4f6025bcfc0a38830359ba560ca5af506bc3213ecf05f3c7f3c4b2f8ed50f0155f28f25cb979b2aa3896b80d3ca7edff95817247cc81001e10c1b5ef31118646c21a3df0ab427c9f43b4dd2a18b945d578c89f6f916b029bfa8ecb61b58eaedc64421ea1a276b9235270c6b175eadd30f04ff922e5d2094984aa7e4a4a3cdc4366249fc3bf148d209b5904a5d897ccaa906b7f0e80b14a0fe39edaae3b768cc776ed4e64b769fbb2c5f2473895f5be78c6d6753e11d73598ba86faf44240ba3848114f4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266889246Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1280/default/d, Name: checkpoints/1280/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1280/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=9285bde28ee16a8a78bdbf3e0acef5680df78585af26f389d5160054e21a66bea4b9141715c00b88f725c9065e29aa0fef21e775e57bf86d49594a5428848b2ec4a4f0c609ee15e9e5b3a05251618e1ef8e7627dd07cf08c9e1081e5bd3ae6801a091693c200efcb45ddf77e68ae55da80c4203edcd7043322a41b9cbd1dbe7a32bf4a41b0975b8503a4c32a18d3859c496e1168459887e9ce56fef0f5a5a882c87b32e754a7f194c95680152ede2f0cfef734792761a59e23b556856b2e9403fc841c5ea44ed335d56d480b6d4295711425649b6db71fb56abf0392a78827f2a977fd9e010de31aa220960e81592df71f3ef0de870db6cd85af6b44ef35e5f3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266888668Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3328/default/array_metadatas, Name: checkpoints/3328/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3328/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=10db532416203f1596c31672b0181f38592e22842e0b02204203c09e3db406fe2b1241cf4db318fe95789b1cdd35683335676f9c256cb4d69d23be063d0e548595a36f5113baf35155e4950da0d704df28be0886193aa7cb8a2af6acba5d58cd0e4ba03396e6594613f155b15fe14e714997cd9188bd637fd768f54d3d5f544e2c31dc7af4fe2b8dbab98c1a5de09ea6f3852032e85ad2fbaa74ce6d17163c1afc83ab5f43e0888cfb79ea86c36c7dddb0914232a7dd1b98bed4f422a467bd71517a6d8600ea5609d829ca92245e6106c7aeb829d1e23c7096cdacb1079a181dd2d282667a66177404ea03ca1020af1ac21e78ff463d3cc7218c0b0058864592&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266919534Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default/ocdbt.process_0/d, Name: checkpoints/4224/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4224/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=125ac505c778a503fc2cde6fc7341a395e03f24cad0235fbf87a699248086e0bd5f29d143066d28faf0f0e6d5f064c62d41147531ccd08d2e8554144c17ffc81bacf7b11c9bc33fd113a8c77b5b0c5134e3af01b11afa41fbf0635a68105a638b922174da1a83ae376e7dd4a242fd5c545f6dd9a90ca620edfd7e8a3dd59c58dda91a3495bc1c748005a028c029cc9497850a57e92f333ea1b2b2a2d6ed3379ff2549a817cd44c986088b41ff232dccca4ce71ceec3ddf37718b22fd054d73690b1394cede92fd4ec53753f2d3df1100b1d4ab60308fe6cf6ab902dad99e86a7106d1f5ca222c89303dd74db8cfc6f02b61dd46d9d3cfc3d22f5a21da0437fe1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266928231Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3264/default/d, Name: checkpoints/3264/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3264/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=5a515a30e76337a51e834a6a8bc662a127c1e74341c2974558dce2c732253203ba98982c226b81cffc6722e92e02b17a1681ffe9b55396d902b3908882aebfea3ff4724c77781de34bdd5435a2eae24519bb41332ea21c8e3ddc43f36e483343b9b40f3d520c48497fdd38bd0faf4210dc121f6ebadfb58ca8092916051264c0e0ac05ae246900c6a9319c7e3630f864d83fdce17c018c1fd4ebf39faa5efe5bc48357780d203b1d39941b927d92830ae7b4d158c1755078ae70cf29000e9b5789da9a14cb04fc070212a825793c6604b78628842d54885f84d424225b3261f1928e5037b971c04d94a91d3ad179a466ef51e19c73b52f166cebb05595125645&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26693817Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016, Name: checkpoints/6016, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/6016?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=74f55e1d27ec82c548a00fca09689fcfecb4d13b9c0c696d93c4277a759c6688f895d67fc11219b3bd2a3f67bbdac7e4c3692903b6926aa868125f5a62436e90db5d2ac8892aca7a3f60076af5ba785e8f89aa04aa4dec68a67c51f6b0579deb5e93a26dc63c4114a17ad9c397918448d0ea1aef74c42ca5320cba7d46ca2ea2d745df9b83d480a06de41021b74df04ffd6591e3312812825d6017ae78c12a4adda8eea4d8e3f1c21537357c4a0ae5c58b7f848d977ff2c6ced59bb165a4c0ed44b2bed211fadb11e2735bebc0760b4c467b5756005aacc5e23e14b9412e669ea03e12809f7302e5e58765618c7f53b22e49819f03526b8e36f4bc34f544c1e2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266937527Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3136/default/ocdbt.process_0/d, Name: checkpoints/3136/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3136/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=9a4907d126e7b3d4b4f10f2a1c0fb651d864b4f84a5444bc5f1983c631bab77e093969c50f15da70901991b3e4b62002539b60ee7206c6a21b83d6218f6fabb3aae9df033bd43e58dc06759485f20e364614477cf2078b9d1c049730a781f6496517d94c8a58a956ed28bd55000961deea390b82f6b64d56d28a116c0f270f68eaa89ab3ae3d74003ea0338d404cb5ba289fa8506c4a9b1669725d5d6a711eb7a4437180f7badc33a825078f39522de36fa618b6f1defc6c3d99570c35e9d680671fa21dc5efdaf7067e7d028abdff7211b54ce5486d91d5e96a2ec6ad648933c12dcc0989c19e2c92cf10b13c0785d0a77b177814679f727f6e08708c0e8df2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266943043Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/ocdbt.process_0, Name: checkpoints/3392/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3392/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=21e9b2580a1be2ed89a2a2559e2abf9ac0534342d287bb9a4026009cea34fb72869679380beeb342bba17acb866e274aa6412580b1f3bd7634f55304c9ce85f8c18f4f8967d4c89b6bd5ef538bc78ac87bd83f7025d481c506f150146660b7866db255dede70f86a420212db83e1d43c8280b76192246a4b985194394527001868c6ed7584a4daf6b8b675cd325129ea6ce7b2e2c9c6c91b795349200c57bde68840a9b76b99b7ae9f2bd66e1b302d8fb3ab1072aaa7af621cd0d3c94caf0824d2cc5622b9d2894c63d07c16b9dd2ccf4a0dbba0351c2d7635e13d4f0002602579c98162eb024b4f01a3dd761afe714875dfa197b886ec1f948bf4c61ecb9597&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266944717Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/0/default/ocdbt.process_0/d, Name: checkpoints/0/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/0/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=a143fc597ffa52bb635786b9932c20b61aee1049a756d27b35fef98958b1d292bf049cffa05dd2fe080b5f1b982622f35192406fef74cee546363fcb545900b40064903af35dff4cb3bf35dbb1087f204c94b11d52e907f3eba7e09fabd1be05aefc579450a6e7fbf23034682a1eb72f0b264e32d72ad66c9636d5d4e23bf4759437ff2880743f44477bfebcc76eb81d350d5c8a73d3c8a86b66ec64854317163796ab12504bc2370304c3a8075b2f0cd7229f43322c986f788771addf5652a663ca9d43c6f6da442b04e8c295afa2005eb032e2a031e84d2fe0c6bbdc03d81ad9034c267b602608776785c06ffd8a67eceae046f1a5b1381429c42fdcf58c63&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266982361Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2368/default/ocdbt.process_0, Name: checkpoints/2368/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2368/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7124caa525899375129a1865c034b65423bd7c1b3e4b8aea943c88e7ecab0962933a7eded9f6e726d175907f35586984ae3dbe152842d51571634c717ed0cb0fec982e5e8e245f16fac42ac5dbb86a80721c32c29a40f2eb96b33da7b25efc17a8d3a16e8cb1701fd71fc57c3837f87f46e14afb564bab51f72be9e037c3ba84fdb10382992d3805a4e920fde10141930becc5a26def5c15393726c9226338b7f0d2df300e66655b57c7e8e3555552ae559f8698ab30aec9812b181cf0a97909273cc81c56605dfbac35e9f31666f701102466886327164b921b3eae48ba8e3e38c0a6b5bc2f280ba619e89ad896ec5db0620304b7d09312794442c9f40de47f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266982342Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4224/default, Name: checkpoints/4224/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4224/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=59858622b401a65e0ad593a5a85cf74442c295eb50060c16e4320bda9eb4e3e739a161ac6a87a877b5ab6ed77235512210f5c095049526432395adce10b0eba7cd88dbd46035bdad66fea9dfe7452aad08f2118485af526a62d0c2a0ef202b7d3d730fd9e810cbd2440cc6d4daf7c737669048c4ac7e43c3da73850a40f21cc340f197da79df281f575b687b9c8ae6eb090ab948d23a2f8eb535692e09cb80a1b3e540285ba0f28549cb6db1e22c49524a9efacdfa5468f0d117ead92b86447a2917d6aca159bbb0c79aed26abcaa312211342542dc3c0fa1da04a3fd5f3f0b3fa75e4863ca1fefd57c1dd9458faddf05caa2ff8d00f13dac7a1f67b6547d3bd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266990186Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/6016/default/array_metadatas, Name: checkpoints/6016/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/6016/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=74b7683368cc4f8ba4c1437420510818d3e815d92b1f5da601d6847640f230542dae82623735b3435109c224c36a2d409f274ac0f5ee4fa4753bf3170ce23941f14f049aa25a0f78e81f6168ea57880f5fac520b53b59a78c052949a5bfae6435830c8bb10d351e5f522b7cb05c563737eac50a8e18211f5a42ce57140692f001b9c213aa93bae741a4bf42ad8f960a0fe43cce2ffeecb6d8990a835900be233b212bc96e5131a88a22185d89553c104875b11182a6635fd44e0d122b69f7a95098007883faaf6cd47b3a0938f03630af49b86522e96613abd479226706e1b4d40c91082346af224b40d69ca744a0efcc5ea7e43b055d9af8d2bd2bf6942dc9c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.266999874Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712, Name: checkpoints/3712, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3712?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=4b9bc88363193e0962eb65a4e70c5bc39d4372090eea32f814462fb26275c24029569770cf83ca2463dd7d697217d1569644ea86de8bb9369c8fe0e87f7ccc7d249617255e87f89ff1175c14a035ec307e0b4425667c11ae5efb6b53585de091b6bd05d49d3cc8d0261803731b9d38717420e076d3f12c3175ff7fd91ed3fac3ddad26daef7c4fb1779fd9d1169278ccc1bdf24ba0d33e25ff6556daff62efbc745d48d852d14027ff417022a24ed543e74e0b2d806d50867b3d7be74b5653102433c2c73a02d3e0a9b59a31ce7fb27195fc218ae29a8087750c445a24ee9c002f01b6c94e17f9228df5f1cdcfc51d6362f10cbe1d68b0170b33e8d077992563&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267044638Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520, Name: checkpoints/3520, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3520?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=2f512e44a6b0cade568960dd00445eaa8b5685895d2063b8e3cbd2ceb12d46cd68d1f943b44e979665ffb393a6cf7b4c808ffb774eb4a4ef08b62f474373abe7fa37af78948aa857eaa0a2688aa443bcb1861395e4cc4b3ca23252c1f5ae98ec05194642f67a379011ed2b6e66a75e8ffc4c4d96bcf89c8f740781008ded5921d06a3bc94cccc10bd240d8a0e089d55c9edbe755a9a65a96e0aa55411b499837d10290bfb6777770553dba8c9f41498abf49ef1c16cf19d55676647c5e9436827f91a75354b198c95229b1980fe495eeeb7049f3df92c72495169c9e2c22ff6194a4ffaf416947b180112279a3f2ca7a4e9c97f91bf73801e8204f5da16fef08&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267031678Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/960/default/d, Name: checkpoints/960/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/960/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7683ddf145167e41292b4639239e29f472e2e06fb8b190941733b9d62049e831421326660fbbd6921631fe159f60c68f9f941f8e0a636d361b01ef334d8278111558a6fe85ca03bf4e304169e6d5db69da61f845b01f6ded0854716d18b1bda81d77e96613553a67bd3cbc349d37816002ca75cec5093a2045c8ae65acc08cbc7025145f2169b13a971e2031e4c69f2c4db03fa78b63f383786fe50532d68b50e8ddced69f75ee3f4f6064fef6fe3b9f488f94a0d861724d1d695c8b61132f4e0df9ebb6e4a5ee9462f56676db3e9795f0b99a78809a2ff5b961a841229dd00affe40c250658fecf0442640050031c51988c12c7d6656a9f598fb78956f82547&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267051958Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/576/default, Name: checkpoints/576/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/576/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=ad895f14b296fb1ed8fe116cdf118bbaac435f12d9ee41a3a8083ecc1c5b58a419d992d842bee73c7028760ec758b930f107020d9f80ec77d0a15c533523462146bda1b7952e43ab66436bfc2e91d13008bf859b4932f5843326fdf8807dab83166d3b9858b4725c3683400d09d06705e575a54045a5fa3de0ad65b056a24e08c21bb9a66cd6b8e3c203483f64d82cfa27c8e6b4e07b46cc5e5e34097a7f25fd605df8b0c40398436d2c089681f1c20011f462e1cc52fc4c5b994af54c5d546a55f74536bdfb21eba725ae24f2d35f44434a5f1567287fff694f76fbc803cc6bb9e53e0a1dfd41df03626037908e1667cc5ea1711a592e529ab674b6a154dd88&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267067066Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/2944/default/d, Name: checkpoints/2944/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/2944/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=8a8156fe92a8d0ca87696c5feb85a00c51150375e32e0789eef3e40613d8f9db8eccd1e2cbd0e64b10c8a57197d0c9319cab1442c4be00e0b207d0e5a167c4980018870786a50858099590562f2718aea64ed089e3c050cb8f42589bdc529b732520da7650c98693dd369572665a48931293e0c0d87ec3f129ebf3b78812ad2a07adb74f54afb63a90fbbbec519afab1e34cf1fa4871e892629d558ebcd76e99db71240cdbfce51629ac1fd2b43fa864d0ec3c6d594a3a21563f8cb84939f875af7f4b17a5245294217d5a9590e03ec7a93ec8e4f1537e877b51181a644e85053e4ebe622836544903dfddc98db8af334157eca2ed7fa282cd09371020d124f6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267095004Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/64/default/array_metadatas, Name: checkpoints/64/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/64/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=95646d7d8daec16d69ed90addc794cf85f0f96115344382c1cc171ade9598cbec507f086887bb106b77e14f8df2daf3ba31c5b2027b09954c3c6622dd44bed62b549a7fd41f9161bf25148939f69da024eaea178883d0ec819014eb6de513e3ceed72fcd31c380b9ca94d2afc5cec5fbb1c71608840ad0ec7f205ef069eaf31381d99f17e8a18d16c2d24c8482a81665b1c8f71ce4f46ee2cd633cc925ae6cafb0ef1e5bb99051248fd7a1f0cd584927b8ee1bf42578a856fda024bf9f62058e6091623584163769d5f09a8a928a524b194af2a8af97024f9a811da28fa48cd4549cc1b9799c623bb0c71c0f54d57d19e30bc53c6fcb2e7fcb3a7ac8a24aebaa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267103199Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/640/default/ocdbt.process_0/d, Name: checkpoints/640/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/640/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=aa035b0ac13c5e4efbec15f322324b832cb0cd5484a00c90175e82fb1ac4368795bb3384a0f3aac638ddef97e855140eb9437e7dacce808d2dc12115091ff76ffd0edf4abee966b7ae9fe5ed6c527aa9bc8fe460b4432c799ec3f9bdcc76586fc6f70d64f1068aa4db86e3ac3bf8e3dcfb479f1aec9c53f86a09eb15e5c5451cf1b88ed36411a7bfe5a48fa5595ad7182665e767daf6a407c5edef94bfaa5c540c0656d0796c85ab229e367b43bcade60cc784aae34bcb2d057fafdf8f4f13f885a8425719fef351728baf57f1eaee1a1b7c406289ffa17c827175f4a004585d603f71019f8fae48b97bc2c52c56ca9e5df7665f43a3ab2328769648237564f6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267134895Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3072/default, Name: checkpoints/3072/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3072/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=037edef7804d63d67801a3572cc63e19d6c3634f8a14db0850fb6bb062c8b698483f69f5554e996e005e501349afb4995d74a9a43795fb0698a9c51c1bded855870e308835dc03d1dd3d677511dc796182f2abb44e14dcb741fc481dbff3d1bb4ccbf8a337617a9ac312bd10716f58c24815957942b31e3ce21d210c866b4252a02da95de5d3989fae887f87f6ca051a34a81a5435997499f852bb82fd7fa09440a0e66eb116a30ca74e673df065f34dd2862e8dcfa806b877feacdcaa4b077baa350e1992e06981cf283906c55ea97e8d0036302611a02090ee10a9f1ecaafc3154dc290c7a3affa806c16227deb46ae89f9e45c50ba8b02b9d9d7d9fe45513&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267129046Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3584/default/ocdbt.process_0/d, Name: checkpoints/3584/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3584/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=5b50b60bef331d5e2568483c4e63c79147da9b7fd521521fadf4cf391951a1d6fe715025016657c710dad8a0fae447f88bcfa7efd33ce170836afad47a89bbf672bcd09361a736ed68a094ca7f845c7bc61aefc6e55a6a505c376ee9dea6b5cf761973246a53761db760acb4d681211a4ec708a506b29b34894fb704ede2e2811d5127d9033f55d464dd337dc1d104a0324e0a748eb8f25b1fdf953ed50aa975bd0778c4d33e9609ef95fd01d18904b9200fe2def4455629da475aa873cf658baba0dee959d9353c8bd4ecf7b3e0d886447df353d7f972c5cc6c93b421b53bc5fd3f24ae11717ed50cb9478261efe767eac40633668ed0dad217980aa083291c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267145401Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/320/default/array_metadatas, Name: checkpoints/320/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/320/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=22d30721cc1c28a0d3d4a454cc43dacf8785416f0670dfd0416956eed22fe962cbcfe8951e260e3995864d0f7ecf52037ec49cdbcd37c2a8d3de3ebfd9eaea73ff62be9c6319cf23faade8bd8fef03dbeb21f3060da29fdd7130c66016989cdb02581a3ac28325b9a787e331aeadbef658db7b459801f08fb1039c08cad89739d4d456a10d2b4ad3b5b1cee0fa91cacf4110b18519cb55bea3ff2cd5337d30b40cc4fa096221eaa95ee2f030c4b99a62183d284f13acf858f8ca2a3c54edc4f778fb7e4c02cd25bef8ab07090de1976052cb5e0e47efcdda64aa7dbe9f4fa039174d491c3fdbf8ca0a64b56741be0277d9ccf3faf4fedf2abb33f1888fad1f70&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267164228Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1600/default/ocdbt.process_0, Name: checkpoints/1600/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1600/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=7ff1c69769ac1eeed20a92942cdf65d0a77db68f216f1ab459c35f1db2579aed23624e43ae093143d7b2156118b5a8190217b626daf7f49e114845a9876f159cde8c215291a5011c40a634ac5cd606a111c1adb07605845743ff2d9cfe87eb1b55c4f998e1d6e1e213bc7e45148f92e0748273dc3a3a093a91e9e8c33881d952962e9321de97525edd51b10d3758d2858e3e538ece9e6f19422b86ca11e05fae3b0e2bce980e0da86adb35920e0fc3360d0cba773cbca8fe7f0275d74a6c8b6e1a67f307783a63731a57ec243ed9cb3fe73c685b72f8d3681a0bf4568e8846c8431b3e475d047cff3442508fa043612778371988ae04af2da0e91488e89e9f73&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267189205Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3712/default/ocdbt.process_0/d, Name: checkpoints/3712/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3712/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=18fa7c0a1ed284a2d877e5a5bfa4b2bca378ac4969b4632cbdd079b7608eba254992cca23611de50d1a9442a441fe04b72b46ca4a156059945f8ccbbc0933483ed8bace513479b19d62e8491a2150cafd76737e0797986a6b38d5d9e8092f493470025e836023f41db2665fee5907194f3505673abce6b8f80242c22c172f0a7e543ecc4728150adc915b19d4b471861dc3488c9faf8a09514fd3cfd59f71164f949e5195eaa7576075dca976c1dbfbcad051dc4250bf2a2b8cecf985eaff5ff657eaa2f884e26f1c4d1a55ce8a75925ad8beacc8b0c5adbc3cbee0ac6959f4a5d7ea54faf5b824d4e35fe77b8ae51c75b66a8077fe3650a77cb83c9dba63421&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267190885Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3008/default/ocdbt.process_0, Name: checkpoints/3008/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3008/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=abe9a56352620ed2a192dfa0532dde2e4bf5059fce4ee6a39cb6e167e18130305c3fe83d9ce96ffbf84e539f4209a7166745a4530b8dd62001a78a78288668a9054f140904208d17eb0b5252339f3fce80ffe9142f5202bcd8f8b7e10e7926ae1b17e4acbc7b89a71447fbee98667560b6d16f1ff30a8949655a8284d6ee5346e29144bef6700eba2ff5cb036de7e7abaeb64b1d24c638ff5e4da6e75309c76957582d900053abf3fd013a49b23097b023685c6c720d7c47ec3f65db649bb076f36fbae7da629427359f27d462a302d3c89623a85e8db4786f9313b755c87d4c058a9c514a272ffc66a60dc711f047245ce0bfb5286b3dcf2782a584feebde12&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267204229Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/5632/default/ocdbt.process_0, Name: checkpoints/5632/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/5632/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=2b491cd33585504f6b2a600171a8f33580667f1cf23cc61b1b4fb8313e15a6dcd0702f6d1047cf5a3baa1785599af0738ae6cbdd6edfdea095b7fc7ba9d43391afe0c300c553aed9dc6f65fc3ad68f514143d1f0897cc00bbbd2fc0af496682ce16e520eabb1c7eb589e5c0f84269968e548c0096f77276c2bb8c068fc64bd928faebaac04119d294633042c0ca91a96d99f8bd2752ffc0476bbf34a8c882903fc0d41e674c9d6894fe85da9892187c9861d38bdb97fb2a335b5902ab59ec1f434d12b2b7412de1f794dbe3108e36e000b24699ee9a23da74fe451e38284790d76d13c0bf6eefc1de59e100b05f1c1226af6a9ad33d79b021160a93931979ad8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.26719751Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3200/default/ocdbt.process_0, Name: checkpoints/3200/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3200/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=b7c4bbea0e1c72e73e93955c8877342169be54e139ffca7489d8c3e68abd7c0c2f6810ab503e0979c0cb6d4cf0687a9a177c1dec68e16b673db6d3f27a2a1067a55bf45d4e2cf102069dbd6bde126d213f2d804eb160538adcd25e57465b46257e2e057794667f2b50608039e1f32295b0459b9d114a5b75b57289233b097b1804b8a1a5b63d0a46ef6c26b12bbb861dfd1e00546d36f3be39de24ba10cc096b6fb397b984aa228b6aafb7ad818a743abc57d5530a7e0008134eabf209538d12ee76026f5f8c9fb7d94122bb5f67585639d2d49906895a01422041c7a11aab54a9f319381afbec0b5e4881d1034ea95caca6502e6db8744de468ecd89bbd1c70&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267209759Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/1856/default, Name: checkpoints/1856/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/1856/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=bbb4ed753ddc639181742abf3234c2c06b7a726359d4404d7843e8284c28dc10fd86fa452b4d0d9cee17d1ef13f455164a31a24a55ccd70a37c4c3fe7e5dd7fd8cc2855eb9485e5d57f1ba50bfd8157bed7165781da849af1bea4cc745d917d40fb645d48f63acae6df77bb932eabfb3ffc356d054e667aba823438c03cc736b2e0ecdf1a05f61e6f74737bd876be6562cf1fcab2029dc51216d3ad979bbb49b82025d4894866cb14e8c7e8f122b8c4c5b3b240b629f38d98431ae17fbbc06c2cab8718b31f9ef60902224417620c8633a4cc927b6da0145102d25abb8a2ea57058005be5f43680eab8281c178e7842170b613500cd5cbf48518723f07fdd191&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267207855Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3520/default, Name: checkpoints/3520/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3520/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=3d8a3e9db6e8648587ec71ef53aae8dba0732f0a43b29038bd6d28d695fd8f621e60873d1e28574e9b7defdfbf7062ad4eb19e8ba1e4d9e74e918fdb5b80e24123392cd60a39f4d23419a90004571ab14c80f46edcbe32bf623e6edba14a2148f9834240b7a0a2a504e7a1ea9cbbe5396f0dcca0eef2d637101cb1c5d5005d93cde0d630627b1bf56aca8e5c57d33c96e73d205a4f6a74a8323a5e6980bc6c413a82b98243b5dfdacfbd792abfae2d3f7b756ae0313abc3d16b3826909a9cc50f9cb405e24299089fc57594a989692a0e899c62de96a43f050541bc27c6ec517cb2e59c2fe32db3b2de9d48a54ca2275e7e03de00ae0ed62f7cc795794c7ffe9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267225947Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4736/default/ocdbt.process_0, Name: checkpoints/4736/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4736/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=6f5148f59b0d17888f40160c94ef8a413add1baa28ee62dff1d023eb4e9070cb91e298e79b88cb8dc0ee2667d74826d7efebf3929dacd43a8921f8bacddf94f84d0a5f5ecd5a5eca173845c532202be525e0ccdfd64a8643f2a77ec2c8dd1b4a48b087e5fc9acc22f0d84578c65422d1f27fd387a93e847ebf927242e93c2e6b6064164365b80a4e8a501e65a2b393530c8b0c90c67106981d7668d72b018bae3d33675f8f4002d39fd47b7b470ff3b188c379f2f50651efd6e62eb8651c9b1f3c12d0cde29835fd11e61be6d0e76f081d4f1273742344704411dffc713ff945cf2f3efc55e8617816155c63a65dbf7fd17523b471a297e580bd1cbf08682235&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267224184Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3392/default/d, Name: checkpoints/3392/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3392/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=5ac00f0b8227ef6a1e972b30cb3c1e7afe9697e5d159ead2f8d51332f0ee3c1aafe096f4ff013f9f7ed45e2e6c376c413011d0fa5fd2b97e1b89ffcc0f71d0bd87db9953068645ad937539e54d65bdf89fbc05e8ad4d1d441a5f2f168b0b40991f75e0ba7a3c0cd57c5eefcffd0663e176c8583f2c3eec5a4f7f9721f4a08f0a8cf0998a729345535f311556ece73e486d2dbddf66482c01898333c8e6046d688def5be1436dd28659a9eb8fe6c4f8e6f1a97f551c221e6a4b9afcf6c0a141cb3988bab8d2c51c49dcfae7bce66db6df2e63727bddd1427a920a4086928beb9a1ac0430629ba58c9ad4da8c44bf935fbdf74efe0dc3be766dbfab33e57d4544a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267234217Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/3456/default/array_metadatas, Name: checkpoints/3456/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/3456/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=10f766a59e0f62cee8f890e6d049de25e1822e3623854011b23f1766920dbf721be14294823bf79b14be27bde90809142ee75f683085f5a8496836ee647caefb54a7a9fff36dc485f5832b5968b8b60102c8fe635846d34e1414b22fe5607ca4e85966acb89b3c3a8762f0975336ded941f6b8dbf5e381bbd107b77f5b947d4fe72d7e437fc1f69b4d88821c64702f27dfd697470953e4e423274263073610deafc6ae8ffa182925ca0dce375b80f5676b7be2a4fdf8245cdb677c437feb22ab7ade8514bc60d02ebff6eb488b5792609850db0e372820e4e02090525b58ca11e706aea6a2bb5a0d3213266b7d6d0a5eb9efb24c35930f2eadb7e4c53da2b887&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.267237579Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/files/checkpoints/4480/default/array_metadatas, Name: checkpoints/4480/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1/checkpoints/4480/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T180159Z&X-Goog-Expires=86399&X-Goog-Signature=8e0a78f5f076de4edbcfa4f6d0e1ff82de019438eabec821cc8f3c07cc40de3bf896160d7144424a68fedf1f7f6aaadbf3d0938ba1efe48f68e20c2e2ac39588e824c152fde4aa12ed519f29955110d8551d276f7353e559b1f2d43736e0fcecb0b080480c32a77192dce649651a355516ced0b41d8ffe457d9d7802a98be335b54ac0c6998adf669e19032475383968a2a31bcbd9b85104c77c6cec8d1021e45578390bb171c0ea156c945350baa5ddceca0c47c42ac44ee63aef4793ef604842dfac7c6d9697f4495601553c8a984e6073f681177517de49a109f051f2e8bb25617066f916d78fc0439533c0cf4d28fe81b951e2748675b61c1ec4c2e84512&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:01:59.42814623Z","level":"INFO","msg":"handler: operation stats","stats":{"operations":[{"desc":"uploading output.log","runtime_seconds":0.573781001,"progress":"1.2MB/1.2MB"}],"total_operations":1}}
+{"time":"2025-08-07T18:01:59.96006367Z","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
+{"time":"2025-08-07T18:02:00.220500918Z","level":"INFO","msg":"stream: closing","id":"wp_1.0_alpha_1"}
+{"time":"2025-08-07T18:02:00.220548599Z","level":"INFO","msg":"handler: closed","stream_id":"wp_1.0_alpha_1"}
+{"time":"2025-08-07T18:02:00.22058266Z","level":"INFO","msg":"writer: Close: closed","stream_id":"wp_1.0_alpha_1"}
+{"time":"2025-08-07T18:02:00.220626805Z","level":"INFO","msg":"sender: closed","stream_id":"wp_1.0_alpha_1"}
+{"time":"2025-08-07T18:02:00.220775364Z","level":"INFO","msg":"stream: closed","id":"wp_1.0_alpha_1"}
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/logs/debug.log b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/logs/debug.log
new file mode 100644
index 0000000000000000000000000000000000000000..5719c91da9ac7325b94afa7f358ee34c1344a9f7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/logs/debug.log
@@ -0,0 +1,28 @@
+2025-08-07 17:01:24,518 INFO    MainThread:3847789 [wandb_setup.py:_flush():80] Current SDK version is 0.21.0
+2025-08-07 17:01:24,518 INFO    MainThread:3847789 [wandb_setup.py:_flush():80] Configure stats pid to 3847789
+2025-08-07 17:01:24,518 INFO    MainThread:3847789 [wandb_setup.py:_flush():80] Loading settings from /root/.config/wandb/settings
+2025-08-07 17:01:24,519 INFO    MainThread:3847789 [wandb_setup.py:_flush():80] Loading settings from /workspace/timaeus/projects/rl/wandb/settings
+2025-08-07 17:01:24,519 INFO    MainThread:3847789 [wandb_setup.py:_flush():80] Loading settings from environment variables
+2025-08-07 17:01:24,519 INFO    MainThread:3847789 [wandb_init.py:setup_run_log_directory():703] Logging user logs to /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/logs/debug.log
+2025-08-07 17:01:24,519 INFO    MainThread:3847789 [wandb_init.py:setup_run_log_directory():704] Logging internal logs to /workspace/timaeus/projects/rl/wandb/run-20250807_170124-wp_1.0_alpha_1/logs/debug-internal.log
+2025-08-07 17:01:24,519 INFO    MainThread:3847789 [wandb_init.py:init():830] calling init triggers
+2025-08-07 17:01:24,519 INFO    MainThread:3847789 [wandb_init.py:init():835] wandb.init called with sweep_config: {}
+config: {'env_size': 13, 'env_layout': 'tree', 'wall_prob': 1.0, 'env_corner_size': 1, 'env_terminate_after_corner': False, 'obs_level_of_detail': 0, 'img_level_of_detail': 1, 'env_penalize_time': False, 'net_cnn_type': 'large', 'net_rnn_type': 'ff', 'net_width': 256, 'ued': 'plr', 'prob_shift': 1.0, 'num_train_levels': 2048, 'plr_buffer_size': 4096, 'plr_temperature': 0.1, 'plr_staleness_coeff': 0.1, 'plr_prob_replay': 0.5, 'plr_regret_estimator': 'maxmc-actor', 'plr_robust': False, 'num_mutate_steps': 12, 'prob_mutate_shift': 0.0, 'chain_mutate': True, 'mutate_cheese': True, 'train_proxy_critic': False, 'plr_proxy_shaping': False, 'proxy_name': 'proxy_corner', 'plr_proxy_shaping_coeff': 0.5, 'clipping': False, 'ppo_lr': 5e-05, 'ppo_gamma': 0.999, 'ppo_clip_eps': 0.1, 'ppo_gae_lambda': 0.95, 'ppo_entropy_coeff': 0.001, 'ppo_critic_coeff': 0.5, 'ppo_proxy_critic_coeff': 0.5, 'ppo_max_grad_norm': 0.5, 'ppo_lr_annealing': False, 'num_minibatches_per_epoch': 4, 'num_epochs_per_cycle': 5, 'num_total_env_steps': 200000000, 'num_env_steps_per_cycle': 128, 'num_parallel_envs': 256, 'console_log': True, 'log_gifs': True, 'log_imgs': True, 'log_hists': False, 'num_cycles_per_log': 32, 'num_cycles_per_eval': 32, 'num_cycles_per_gifs': 1024, 'num_cycles_per_big_eval': 1024, 'use_fixed_eval_levels': False, 'evals_num_env_steps': 512, 'evals_num_levels': 256, 'gif_grid_width': 16, 'level_splayer': 'mouse', 'checkpointing': True, 'keep_all_checkpoints': True, 'max_num_checkpoints': 1, 'num_cycles_per_checkpoint': 64, 'seed': 42, '_wandb': {}}
+2025-08-07 17:01:24,519 INFO    MainThread:3847789 [wandb_init.py:init():871] starting backend
+2025-08-07 17:01:24,735 INFO    MainThread:3847789 [wandb_init.py:init():874] sending inform_init request
+2025-08-07 17:01:24,742 INFO    MainThread:3847789 [wandb_init.py:init():882] backend started and connected
+2025-08-07 17:01:24,745 INFO    MainThread:3847789 [wandb_init.py:init():953] updated telemetry
+2025-08-07 17:01:24,755 INFO    MainThread:3847789 [wandb_init.py:init():977] communicating run to backend with 90.0 second timeout
+2025-08-07 17:01:25,347 INFO    MainThread:3847789 [wandb_init.py:init():1029] starting run threads in backend
+2025-08-07 17:01:25,748 INFO    MainThread:3847789 [wandb_run.py:_console_start():2458] atexit reg
+2025-08-07 17:01:25,748 INFO    MainThread:3847789 [wandb_run.py:_redirect():2306] redirect: wrap_raw
+2025-08-07 17:01:25,748 INFO    MainThread:3847789 [wandb_run.py:_redirect():2375] Wrapping output streams.
+2025-08-07 17:01:25,748 INFO    MainThread:3847789 [wandb_run.py:_redirect():2398] Redirects installed.
+2025-08-07 17:01:25,751 INFO    MainThread:3847789 [wandb_init.py:init():1075] run started, returning control to user process
+2025-08-07 18:01:58,422 INFO    MainThread:3847789 [wandb_run.py:_finish():2224] finishing run devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_1
+2025-08-07 18:01:58,424 INFO    MainThread:3847789 [wandb_run.py:_atexit_cleanup():2423] got exitcode: 0
+2025-08-07 18:01:58,424 INFO    MainThread:3847789 [wandb_run.py:_restore():2405] restore
+2025-08-07 18:01:58,425 INFO    MainThread:3847789 [wandb_run.py:_restore():2411] restore done
+2025-08-07 18:02:00,186 INFO    MainThread:3847789 [wandb_run.py:_footer_history_summary_info():3903] rendering history
+2025-08-07 18:02:00,201 INFO    MainThread:3847789 [wandb_run.py:_footer_history_summary_info():3935] rendering summary
+2025-08-07 18:02:00,219 INFO    MainThread:3847789 [wandb_run.py:_footer_sync_info():3864] logging synced files
diff --git a/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/run-wp_1.0_alpha_1.wandb b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/run-wp_1.0_alpha_1.wandb
new file mode 100644
index 0000000000000000000000000000000000000000..7f3d45c84268729c2ccc8ad4e524e86f5f1b72a6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_170124-wp_1.0_alpha_1/run-wp_1.0_alpha_1.wandb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4212854212f1f4f8d319f8257974ffa2d974080f998af14197da239b40754b50
+size 5898848
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..795d7b1e9162f50d4df6b00a4e3f0a1cc2473517
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754589869146808479, "commit_timestamp_nsecs": 1754589869337771084, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/d/5c8df4a960c3379340c831b5746100e9 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/d/5c8df4a960c3379340c831b5746100e9
new file mode 100644
index 0000000000000000000000000000000000000000..8b4cc2b0c4107df071e2199ca0c6b67512c7a326
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/d/5c8df4a960c3379340c831b5746100e9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..64310ed2a33b5fe25d92aad348af8e0f33a59963
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/328156f3025e2b6650e1bc17e27c3d8c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/328156f3025e2b6650e1bc17e27c3d8c
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/328156f3025e2b6650e1bc17e27c3d8c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/621793331ef6f5a4af43809c80511342 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/621793331ef6f5a4af43809c80511342
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/621793331ef6f5a4af43809c80511342 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/8e9a769a974716363ace8c75c2d8de61 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/8e9a769a974716363ace8c75c2d8de61
new file mode 100644
index 0000000000000000000000000000000000000000..515a2d9d81760e32b7e07196890c798eba2426ec
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/8e9a769a974716363ace8c75c2d8de61 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/9424ca6f546185b92ad0ad5bc448134d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/9424ca6f546185b92ad0ad5bc448134d
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/9424ca6f546185b92ad0ad5bc448134d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/97a45da772444f3b66fd6628070fd970 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/97a45da772444f3b66fd6628070fd970
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/97a45da772444f3b66fd6628070fd970 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/db549f8304ac8952e56f96867d5995cf b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/db549f8304ac8952e56f96867d5995cf
new file mode 100644
index 0000000000000000000000000000000000000000..73142db7e5d7c17a5ac4f33af9bd2440a6b55201
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/db549f8304ac8952e56f96867d5995cf
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5c6c1ed0b799d7c64b182f27ea2aa06993ad9091d04e250d5a1332c79051a5ed
+size 410908
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/e1643b359b36c89cbd956409fabc8d48 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/e1643b359b36c89cbd956409fabc8d48
new file mode 100644
index 0000000000000000000000000000000000000000..71415c62e244ee61a535db283512b29aa671338e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/e1643b359b36c89cbd956409fabc8d48
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ac9e84fc986056825ae9daf61e91b8f011f2a9313c0b6e8006a9cce855f47ab3
+size 286693
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/e75dc0b21fb6ac38bfb2f1f20b30f5ad b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/e75dc0b21fb6ac38bfb2f1f20b30f5ad
new file mode 100644
index 0000000000000000000000000000000000000000..4859214e6b68fa6e697de9748417fcf43be8081d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d/e75dc0b21fb6ac38bfb2f1f20b30f5ad differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..099c8e78a5cb060ddc46b1dfb43979baec1293ec
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..db276a9c0221083cf2d3e4470954233eb41cb812
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590436216734744, "commit_timestamp_nsecs": 1754590436358839348, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/d/d929cb1fe7b5e60e3cbc671b74abf8df b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/d/d929cb1fe7b5e60e3cbc671b74abf8df
new file mode 100644
index 0000000000000000000000000000000000000000..7474554e96ed3b2cd6d73c94abdb36a7f27c6fc8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/d/d929cb1fe7b5e60e3cbc671b74abf8df differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..cf4b2f1554b2f4d0e378b618ca3e45c79691128b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/10e4790820995bbb446cdc76fad31c06 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/10e4790820995bbb446cdc76fad31c06
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/10e4790820995bbb446cdc76fad31c06 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/325ed28a87267154721d2c34c3575204 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/325ed28a87267154721d2c34c3575204
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/325ed28a87267154721d2c34c3575204 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/4921a05c5a84dbe1c6c0f2000ea786da b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/4921a05c5a84dbe1c6c0f2000ea786da
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/4921a05c5a84dbe1c6c0f2000ea786da differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/7ab8e0777196979d1e647f6fa7d9dc4c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/7ab8e0777196979d1e647f6fa7d9dc4c
new file mode 100644
index 0000000000000000000000000000000000000000..94e82e36cb6a52f06122429043b7687a7723956d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/7ab8e0777196979d1e647f6fa7d9dc4c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:808c5f372b8745609da6130bcbf917c43a0e12a8dd44b23ceb29879f407b5bd2
+size 295870
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/7e81cee7e91eb6c790e6c637b4eca1c3 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/7e81cee7e91eb6c790e6c637b4eca1c3
new file mode 100644
index 0000000000000000000000000000000000000000..1cb31e4f427fd2bca0636a23ac455b50663bb24d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/7e81cee7e91eb6c790e6c637b4eca1c3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/c25f85d05f6c3bbff803a65c40a67991 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/c25f85d05f6c3bbff803a65c40a67991
new file mode 100644
index 0000000000000000000000000000000000000000..b038acde7959c428799bc59f3b7d9d86b0ea9674
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/c25f85d05f6c3bbff803a65c40a67991
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1d3deaf9eca3574e7e8375bffc73df02968ab4452b7b9f275ff4e5b01b16ab25
+size 404739
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/c323499278e5a00094a865569e748f89 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/c323499278e5a00094a865569e748f89
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/c323499278e5a00094a865569e748f89 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/c73535015cf8c5a4e248090978032c6b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/c73535015cf8c5a4e248090978032c6b
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d/c73535015cf8c5a4e248090978032c6b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..59ea01db353c66df8f6931df942924d25c8c2b20
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..933614fdbe5836b6aa76e43f5e695b26536cef9b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590469521840726, "commit_timestamp_nsecs": 1754590469663465836, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/d/488637fd45e6e54c20aacf50032f5631 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/d/488637fd45e6e54c20aacf50032f5631
new file mode 100644
index 0000000000000000000000000000000000000000..359a36b540eb51df17ada4805eb9ec5f13974766
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/d/488637fd45e6e54c20aacf50032f5631 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..63ab79580a9ed5e2668fe12eab2dcacd872b1f1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/0058b394d6e26569ef34e3c52c4974cd b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/0058b394d6e26569ef34e3c52c4974cd
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/0058b394d6e26569ef34e3c52c4974cd differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/0b4f8108ab5d13a669808b93e94367f8 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/0b4f8108ab5d13a669808b93e94367f8
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/0b4f8108ab5d13a669808b93e94367f8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/171d80989401eb6db9890e3862ba9d9d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/171d80989401eb6db9890e3862ba9d9d
new file mode 100644
index 0000000000000000000000000000000000000000..ec745d97353f97209ef50f0eb738585830219f53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/171d80989401eb6db9890e3862ba9d9d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/22e99b5c963c5f0a282339076c9ed0f0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/22e99b5c963c5f0a282339076c9ed0f0
new file mode 100644
index 0000000000000000000000000000000000000000..a10b308f35ea7ebfcb86af18c6bd24e9af129a8d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/22e99b5c963c5f0a282339076c9ed0f0
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4a623ad1e55cb9f2d6414280feb95c362bbe80c39b5cc026e1c9518a90e60a72
+size 270270
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/3d0205761aa0c1cbfaaab19551f23acc b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/3d0205761aa0c1cbfaaab19551f23acc
new file mode 100644
index 0000000000000000000000000000000000000000..9b50380a46ef0e242bc4eadc45161e6867c31b7d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/3d0205761aa0c1cbfaaab19551f23acc
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fe35a764a5c60bead885bb6b1e766e6da28fedd48ae9cc8e82e012b81a3fd42b
+size 468244
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/54256e35787cfe7ac58b406003423f9b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/54256e35787cfe7ac58b406003423f9b
new file mode 100644
index 0000000000000000000000000000000000000000..073524d85623939e717bde7c06b1986a7675d498
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/54256e35787cfe7ac58b406003423f9b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/66922cf9c24ef50ae5f39a7735e113fe b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/66922cf9c24ef50ae5f39a7735e113fe
new file mode 100644
index 0000000000000000000000000000000000000000..dfbfe1a61305bab6f716f83c77874d9e3176999d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/66922cf9c24ef50ae5f39a7735e113fe differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/82bc962bbe1f0f356ff296cf233d8f17 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/82bc962bbe1f0f356ff296cf233d8f17
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d/82bc962bbe1f0f356ff296cf233d8f17 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8a93afa4139de1d7cac78a8bcb47cef8bf8b3270
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..9cf907c795e28e5defbc9046371e1081ba37eef5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590503742542923, "commit_timestamp_nsecs": 1754590503888328948, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/d/1a716e02986fe6f17dfb48cfacda8d97 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/d/1a716e02986fe6f17dfb48cfacda8d97
new file mode 100644
index 0000000000000000000000000000000000000000..da5e89c8c4b7b2d819b0dcb0f4b483dde9d680d1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/d/1a716e02986fe6f17dfb48cfacda8d97 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..978ef31aaa727ab4cb97fceb043bbedfdfa21b89
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/073a1de71d78dce9ae7a3a90301d1622 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/073a1de71d78dce9ae7a3a90301d1622
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/073a1de71d78dce9ae7a3a90301d1622 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/1ce8f774257527b125b38685ef0eb69f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/1ce8f774257527b125b38685ef0eb69f
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/1ce8f774257527b125b38685ef0eb69f differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/2059faf69c39e7eda15b4e5dc829b26d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/2059faf69c39e7eda15b4e5dc829b26d
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/2059faf69c39e7eda15b4e5dc829b26d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/943bdd2af0ccad3c43c012219c540280 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/943bdd2af0ccad3c43c012219c540280
new file mode 100644
index 0000000000000000000000000000000000000000..8ea7ba9403acf1a6a61cd87f87c0fe29c95b4eb3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/943bdd2af0ccad3c43c012219c540280 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/b47013dc050122f9f34545cd0d1c935c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/b47013dc050122f9f34545cd0d1c935c
new file mode 100644
index 0000000000000000000000000000000000000000..8614c6e2fc879025a7edf7f4384c18eb682042d9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/b47013dc050122f9f34545cd0d1c935c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f948bbc41ad0d089269e1301470347af073ae127857bfdab829df87638545618
+size 336548
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/cc1a94f9b7fbb9176e3c3c827d7e055a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/cc1a94f9b7fbb9176e3c3c827d7e055a
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/cc1a94f9b7fbb9176e3c3c827d7e055a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/cefcd948f8aae6188309ac9e076aa3a0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/cefcd948f8aae6188309ac9e076aa3a0
new file mode 100644
index 0000000000000000000000000000000000000000..15c6a7a98fdfeaba0b7025f5145c40e7ee4a8961
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d/cefcd948f8aae6188309ac9e076aa3a0
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:16353f845f65b033fd313c5743f48180784aeeab58f4afb0fd410f7b7e33789d
+size 363987
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1a08188078f74a27d9f77c6b029da05378b7dea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..64d29c03b392059a05435712b13a1b3434b51f30
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590537669275050, "commit_timestamp_nsecs": 1754590537837136834, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/d/fc8667bf18961c79503d653e3e2fde4a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/d/fc8667bf18961c79503d653e3e2fde4a
new file mode 100644
index 0000000000000000000000000000000000000000..69c50fe456b47e51cb48f9197ffe9fd3a29ba46a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/d/fc8667bf18961c79503d653e3e2fde4a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..53d2b8b2a36440f2242fd77986b87046473be7da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/09fd1e258d4473f8a7bd8d7526da1e80 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/09fd1e258d4473f8a7bd8d7526da1e80
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/09fd1e258d4473f8a7bd8d7526da1e80 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/3c81674da9c4c5bf8880727815ba41ff b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/3c81674da9c4c5bf8880727815ba41ff
new file mode 100644
index 0000000000000000000000000000000000000000..e1e14dcae2049c4e4e457c1dc8d9ac9c595e5418
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/3c81674da9c4c5bf8880727815ba41ff
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:945c141968d5dd890ae8fe1dabef7220f5f653b13afb2c8b2db38dd69831b4fe
+size 262002
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/627cc80357a6bd6076960dc6985eac78 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/627cc80357a6bd6076960dc6985eac78
new file mode 100644
index 0000000000000000000000000000000000000000..52ec3c2a9d9587d0fbd8d9f099316ef796dc9269
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/627cc80357a6bd6076960dc6985eac78 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/9df6356dc5a3f531ef04fdfe583bf61a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/9df6356dc5a3f531ef04fdfe583bf61a
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/9df6356dc5a3f531ef04fdfe583bf61a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/c6d0f4db53ddb656f84784a753d7dfbf b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/c6d0f4db53ddb656f84784a753d7dfbf
new file mode 100644
index 0000000000000000000000000000000000000000..a12a1344b472e30113c907b279096a6c49c31a00
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/c6d0f4db53ddb656f84784a753d7dfbf differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/d8698f42a300d264cf30bd7638b5de88 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/d8698f42a300d264cf30bd7638b5de88
new file mode 100644
index 0000000000000000000000000000000000000000..9d0d4fd38ded730a490367eb0798beadf204cd33
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/d8698f42a300d264cf30bd7638b5de88
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:74aafe743f4e98fbdf1c80a76d3c2eb9804cdd870163f305f6837f0b9ca79388
+size 460692
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/e23e16ac97c8e40d68d6b9f95b5559d0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/e23e16ac97c8e40d68d6b9f95b5559d0
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/e23e16ac97c8e40d68d6b9f95b5559d0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/e80fb9712cdfcf5900c528ebfb69b2b8 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/e80fb9712cdfcf5900c528ebfb69b2b8
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/e80fb9712cdfcf5900c528ebfb69b2b8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/f823ddff51cfd87e8ab98de9265f0119 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/f823ddff51cfd87e8ab98de9265f0119
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d/f823ddff51cfd87e8ab98de9265f0119 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5fa95a1f5a6cc1da971c981f99afc697fc50be2b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..fe452cd838b6cdf66cbce6245166a6f8bc97d40a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754589961134126219, "commit_timestamp_nsecs": 1754589961276656886, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/d/ede46951667b3376578af6b29bd3a5db b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/d/ede46951667b3376578af6b29bd3a5db
new file mode 100644
index 0000000000000000000000000000000000000000..ae8272ebc623997e962876385b70709c553a5e06
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/d/ede46951667b3376578af6b29bd3a5db differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..88bfc92e6bda59f0b2dcf4e0fe6d8679dedd91f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/0cee8893a2f32e528130dba213ff3b13 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/0cee8893a2f32e528130dba213ff3b13
new file mode 100644
index 0000000000000000000000000000000000000000..79219f7c699ba1d9ec119358eb4f68db488fa83c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/0cee8893a2f32e528130dba213ff3b13
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:941078e5dc1d42f119c9bd86228fef40f11d670fa3bd38e1523575acffbb38f5
+size 433777
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/57aed851046b82d81951c0df30758fc7 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/57aed851046b82d81951c0df30758fc7
new file mode 100644
index 0000000000000000000000000000000000000000..65529f6532287fad4b450ca493011b55e8bf7f92
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/57aed851046b82d81951c0df30758fc7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/600ff5beb567092e98c3aa8d1d26c500 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/600ff5beb567092e98c3aa8d1d26c500
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/600ff5beb567092e98c3aa8d1d26c500 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/62ab1667550e5eb666614f539c24649e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/62ab1667550e5eb666614f539c24649e
new file mode 100644
index 0000000000000000000000000000000000000000..98a7070c140297dee497d5874fc9dee57f758189
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/62ab1667550e5eb666614f539c24649e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:155856b7372f59f78422e6fc9310eba2b940e1ffe2b3e4e30b78ee49f5f37c53
+size 304146
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/6b0cdf9297bdfe26c51654345ed0a363 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/6b0cdf9297bdfe26c51654345ed0a363
new file mode 100644
index 0000000000000000000000000000000000000000..005c040231b745063ad51221eb30a22134988285
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/6b0cdf9297bdfe26c51654345ed0a363 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/6fac206ea53c2fa5a3b47ef9aa8ebd61 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/6fac206ea53c2fa5a3b47ef9aa8ebd61
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/6fac206ea53c2fa5a3b47ef9aa8ebd61 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/9bab03a49b27a34177e98f9c62b6a86d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/9bab03a49b27a34177e98f9c62b6a86d
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/9bab03a49b27a34177e98f9c62b6a86d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/a57d4bac7845e1c166a710a8dd1c6143 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/a57d4bac7845e1c166a710a8dd1c6143
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/a57d4bac7845e1c166a710a8dd1c6143 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/ddc1ec9f434a1dddbc60344fb23fa29b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/ddc1ec9f434a1dddbc60344fb23fa29b
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d/ddc1ec9f434a1dddbc60344fb23fa29b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..896d332d8c6d70bb4cca7c4025d695c0aa44dade
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..661bace285439f6cdc483a751914fab76b9aab87
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590569788205624, "commit_timestamp_nsecs": 1754590569953411303, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/d/e64005d74ff1e8b321a8e4db8b848658 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/d/e64005d74ff1e8b321a8e4db8b848658
new file mode 100644
index 0000000000000000000000000000000000000000..dc1096c578b6290f77e2db90b7e27890638a6af6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/d/e64005d74ff1e8b321a8e4db8b848658 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8c2fafdf4a2533868b69fade0e8ed7da51a19157
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/233cf74e01d4e8a05a583167700c5ba8 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/233cf74e01d4e8a05a583167700c5ba8
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/233cf74e01d4e8a05a583167700c5ba8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/3060e148db850cecae382b1650f7e796 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/3060e148db850cecae382b1650f7e796
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/3060e148db850cecae382b1650f7e796 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/4629dd57c1238bbcff5bd6ba81feceaa b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/4629dd57c1238bbcff5bd6ba81feceaa
new file mode 100644
index 0000000000000000000000000000000000000000..cfadb34facbf8251e60aed6a930333bbaf00270e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/4629dd57c1238bbcff5bd6ba81feceaa
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:43e73fadb0ac7e16c329c5b1fc6e9fa9948208257e447321cd9c6038fc60d8c2
+size 287413
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/549905523997c74b78e0418094f6a8ff b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/549905523997c74b78e0418094f6a8ff
new file mode 100644
index 0000000000000000000000000000000000000000..ffc4698fce21f91de3f46ac1f154eb9717ae62a5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/549905523997c74b78e0418094f6a8ff differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/6362103176fe4b833c72d74a5f16e311 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/6362103176fe4b833c72d74a5f16e311
new file mode 100644
index 0000000000000000000000000000000000000000..0743bb3efb0e9b91703f543233592c20467961af
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/6362103176fe4b833c72d74a5f16e311
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:47ee12f2fcfae1239c4e61fc4588d7ad550486b42a397a99b1cb809d6a03fa5b
+size 450979
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/6de36d084b691b850ffc72b6ce115a76 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/6de36d084b691b850ffc72b6ce115a76
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/6de36d084b691b850ffc72b6ce115a76 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/f1e647c4154c889378e33bebe8938346 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/f1e647c4154c889378e33bebe8938346
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d/f1e647c4154c889378e33bebe8938346 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c8144a8ba559d6942146ff36451390611644fe43
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..e74da6b497a9a4f6b5222e264c80a3f1493aaaeb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590604016812779, "commit_timestamp_nsecs": 1754590604158594630, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/d/4141dbcf5f8bd93c8bc088f950baf52a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/d/4141dbcf5f8bd93c8bc088f950baf52a
new file mode 100644
index 0000000000000000000000000000000000000000..05fc9e27cc90c203c2af4a7f1baa8a1319bae0fc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/d/4141dbcf5f8bd93c8bc088f950baf52a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1a5690407d6f63430e28fedd895dc8f68da91a92
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/04a4ab34d7a44d60d83304f1274c0036 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/04a4ab34d7a44d60d83304f1274c0036
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/04a4ab34d7a44d60d83304f1274c0036 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/0aabc6a2f7a401385b5efe257bce4e7f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/0aabc6a2f7a401385b5efe257bce4e7f
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/0aabc6a2f7a401385b5efe257bce4e7f differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/29339e28c112d7e1853527ac452307f0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/29339e28c112d7e1853527ac452307f0
new file mode 100644
index 0000000000000000000000000000000000000000..657d9beb22be409bc853ced7aa01a746e05c591f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/29339e28c112d7e1853527ac452307f0
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2923835f5001d08d387ebd068af4d4a884adde973fe4f988390f84d50502a5b4
+size 261940
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/947b9639682d2daea6de16f498020362 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/947b9639682d2daea6de16f498020362
new file mode 100644
index 0000000000000000000000000000000000000000..bdef31394c49254e378cd12f531dc6607ec4db2a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/947b9639682d2daea6de16f498020362
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a95b473424bcd81041aad206fd99ba958c8953f4d59dd919a40153bec56025df
+size 476738
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/98e57be795a3000652d9c6b6d47b4ffc b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/98e57be795a3000652d9c6b6d47b4ffc
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/98e57be795a3000652d9c6b6d47b4ffc differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/9d53de50b960a701a856fd3b28e70a8a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/9d53de50b960a701a856fd3b28e70a8a
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/9d53de50b960a701a856fd3b28e70a8a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/c37337a82505a005d8a020e241fc5612 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/c37337a82505a005d8a020e241fc5612
new file mode 100644
index 0000000000000000000000000000000000000000..64461f6105f2ae830640a985165c8a471d939277
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/c37337a82505a005d8a020e241fc5612 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/f6cdfe5c1e7849166977f6b60569e899 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/f6cdfe5c1e7849166977f6b60569e899
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d/f6cdfe5c1e7849166977f6b60569e899 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..be2eeec2ae93cdbf042b2944c3a08f969f08dfc3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..10056e25783ba50ff4070245490c0d5394581678
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590637319048325, "commit_timestamp_nsecs": 1754590637461610188, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/d/5944b552cba1b7afb57a22518cf50bdf b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/d/5944b552cba1b7afb57a22518cf50bdf
new file mode 100644
index 0000000000000000000000000000000000000000..4606d7d70e6a2d6ad03c68cd52b4c292b2620d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/d/5944b552cba1b7afb57a22518cf50bdf differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..41da7bd7c88f97561495c7b86b8ec74e2445d8ee
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/01dbc2e8b9f152b8f21895460e0fc23b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/01dbc2e8b9f152b8f21895460e0fc23b
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/01dbc2e8b9f152b8f21895460e0fc23b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/1ca5f176cc8d89674d827e563b445bf8 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/1ca5f176cc8d89674d827e563b445bf8
new file mode 100644
index 0000000000000000000000000000000000000000..4c9804e728f86b3e663edb3e09a8d89339ff326d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/1ca5f176cc8d89674d827e563b445bf8
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ec3584677abbafa65b85bcff948617b574461598d7caeebf77384f39ca8a7be4
+size 397967
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/521656647e6600735f1f7ad27f5c8f64 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/521656647e6600735f1f7ad27f5c8f64
new file mode 100644
index 0000000000000000000000000000000000000000..28ab881f96f26200546691806cdc97f1fa5e9037
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/521656647e6600735f1f7ad27f5c8f64 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/85c1ba282029373d82cb52bdfd90e12a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/85c1ba282029373d82cb52bdfd90e12a
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/85c1ba282029373d82cb52bdfd90e12a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/9b3c8adf18f6cb3b3df1c3a9d6d30cca b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/9b3c8adf18f6cb3b3df1c3a9d6d30cca
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/9b3c8adf18f6cb3b3df1c3a9d6d30cca differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/9e89e9842ad535ddf5da9380be522466 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/9e89e9842ad535ddf5da9380be522466
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/9e89e9842ad535ddf5da9380be522466 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/a9c0929fbe9c29fb0274188cec70f6d6 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/a9c0929fbe9c29fb0274188cec70f6d6
new file mode 100644
index 0000000000000000000000000000000000000000..20744f4b10d1d6d69ebde8d09df82a68700529d3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d/a9c0929fbe9c29fb0274188cec70f6d6
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5388f74478216fb439959ed59f97606eeeec2cb028e5f1a646856a912ff01ed3
+size 340305
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9bd32556788c3734a4154380437cdfff2174a0df
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..b30355c0b476dcc299f8a4d381d8bff9c1fb7f28
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590671282989451, "commit_timestamp_nsecs": 1754590671448621841, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/d/487c23363485e94364c0ee80af5a51ad b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/d/487c23363485e94364c0ee80af5a51ad
new file mode 100644
index 0000000000000000000000000000000000000000..1c2ad1be18abfd5394ef4263911245e4438d57ea
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/d/487c23363485e94364c0ee80af5a51ad differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..97dd5a192afeba309505a713597c841566512145
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/01eee1ef5060ec577ec75615f69ff8d6 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/01eee1ef5060ec577ec75615f69ff8d6
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/01eee1ef5060ec577ec75615f69ff8d6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/0c6580077e968edee1651e19dec08cc3 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/0c6580077e968edee1651e19dec08cc3
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/0c6580077e968edee1651e19dec08cc3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/0ed6368ca5811c8e117fee8595254ba1 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/0ed6368ca5811c8e117fee8595254ba1
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/0ed6368ca5811c8e117fee8595254ba1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/21046756b68d4be20fc466383fb4468e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/21046756b68d4be20fc466383fb4468e
new file mode 100644
index 0000000000000000000000000000000000000000..2910fbc6db2094404ed932c7d9bcdb23beb99906
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/21046756b68d4be20fc466383fb4468e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/54fa2fba89060229fa9f6084ad41aef9 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/54fa2fba89060229fa9f6084ad41aef9
new file mode 100644
index 0000000000000000000000000000000000000000..346fe7a7452638079efa4a301427cbc77b7985fe
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/54fa2fba89060229fa9f6084ad41aef9
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3e348a5a0893d391bcbb935b99c9d74b80097adbf7b9cb574cc4f6dae8c54b24
+size 422212
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/7869a04ee2e81120b37ce469a74586c5 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/7869a04ee2e81120b37ce469a74586c5
new file mode 100644
index 0000000000000000000000000000000000000000..cfb5e6f9c3b983cfb61483fc2f3016146528a4a1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/7869a04ee2e81120b37ce469a74586c5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/a276fa3b5a0464a7ff7cc0c9d8626df9 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/a276fa3b5a0464a7ff7cc0c9d8626df9
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/a276fa3b5a0464a7ff7cc0c9d8626df9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/e12aa63b987cde9221d1a86b13093bad b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/e12aa63b987cde9221d1a86b13093bad
new file mode 100644
index 0000000000000000000000000000000000000000..dcff344e53c623785c3837ebac667604897150db
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d/e12aa63b987cde9221d1a86b13093bad
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dfaa4559e68cba60cc1b652e6f344f754e26b8961df1599860caa6f770483e36
+size 279013
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5550ee809e89763616f2f9bb6928c130c7b2cf29
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..830186787c3fa7f6f34390f1c2eab12dc9214ee5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590704864190244, "commit_timestamp_nsecs": 1754590705037264792, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/d/0d84a1d623d104c59d31360963b03275 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/d/0d84a1d623d104c59d31360963b03275
new file mode 100644
index 0000000000000000000000000000000000000000..0dcc407f4ffc51336ac60208781ecbb15306d32b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/d/0d84a1d623d104c59d31360963b03275 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e8f7faedabe7aa5ba9ffc786ce2ed04af9abafe5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/24abf023e9a0ef160d37b220f6b77ce4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/24abf023e9a0ef160d37b220f6b77ce4
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/24abf023e9a0ef160d37b220f6b77ce4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/36aa694255f6dc0f61db0ce39bb4a4ef b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/36aa694255f6dc0f61db0ce39bb4a4ef
new file mode 100644
index 0000000000000000000000000000000000000000..56187899f6fcf5a16c26c7e83afae51a691a760d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/36aa694255f6dc0f61db0ce39bb4a4ef
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4cf89947b771723009a62269cd98e19f335bb9d4c0d9ddb4ef2177b655f94c98
+size 372634
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/382d1d54e0d92416cfbc4939cb2b630d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/382d1d54e0d92416cfbc4939cb2b630d
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/382d1d54e0d92416cfbc4939cb2b630d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/3cf988dc95ebf279a68035c05610125f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/3cf988dc95ebf279a68035c05610125f
new file mode 100644
index 0000000000000000000000000000000000000000..5a81db3238aba0963e74bcc6c7bb84f91a56d9a5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/3cf988dc95ebf279a68035c05610125f differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/c4e6ab6fd04bc97aa2583cd01a177706 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/c4e6ab6fd04bc97aa2583cd01a177706
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/c4e6ab6fd04bc97aa2583cd01a177706 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/c4ee8568cdc22fcda1591b236622e9b3 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/c4ee8568cdc22fcda1591b236622e9b3
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/c4ee8568cdc22fcda1591b236622e9b3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/cc90fff3f2ecc45b4d56576b3db9cb4e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/cc90fff3f2ecc45b4d56576b3db9cb4e
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/cc90fff3f2ecc45b4d56576b3db9cb4e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/dd6a6de760398b8d6a23113adf77258c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/dd6a6de760398b8d6a23113adf77258c
new file mode 100644
index 0000000000000000000000000000000000000000..ae7f500ab695cd26fc784d746cdcf4b78c2a766a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d/dd6a6de760398b8d6a23113adf77258c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c903fe046ffad109c85d967a410df7160ff1fae4ed1bbec843ba467d9cfdb0fb
+size 297755
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a0d7e633c1fbdc4e864f7a0cc70f1668cd0ba5c7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..a39e0abe30c6916b2248c669139a378ab7947bdd
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590738061706935, "commit_timestamp_nsecs": 1754590738242309969, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/d/6e9a2b4760d3837fd1086863aa090820 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/d/6e9a2b4760d3837fd1086863aa090820
new file mode 100644
index 0000000000000000000000000000000000000000..c80071dee0d8aa279f5ce1d362e6e561069779b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/d/6e9a2b4760d3837fd1086863aa090820 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9cf19e6d30c4c1c8a1f4b5cab5ecf94b2c0bb4f6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/1b8903266c00b4277c36110465ea2299 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/1b8903266c00b4277c36110465ea2299
new file mode 100644
index 0000000000000000000000000000000000000000..d93d0c68ff19302b9b01bf078c19f7596900767e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/1b8903266c00b4277c36110465ea2299 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/1e67a4c9e1038265ed8300425704359f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/1e67a4c9e1038265ed8300425704359f
new file mode 100644
index 0000000000000000000000000000000000000000..2a31377ca030aced51ef70fb458c6b5c47404cd3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/1e67a4c9e1038265ed8300425704359f differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/1e90ad460bc7ae39a8623be2c46d6a20 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/1e90ad460bc7ae39a8623be2c46d6a20
new file mode 100644
index 0000000000000000000000000000000000000000..e8649cbbcc72b12baa61c5c2f5d0b2c299b98fc5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/1e90ad460bc7ae39a8623be2c46d6a20
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4e306cc8c591182a54d8cc0b34cadf45955f9382c16aeb752d5cefaf425a39be
+size 287460
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/57b8b6c9f7b87955e0035d47a54ff72a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/57b8b6c9f7b87955e0035d47a54ff72a
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/57b8b6c9f7b87955e0035d47a54ff72a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/9cbc1212c5527a2a23976b17e4079951 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/9cbc1212c5527a2a23976b17e4079951
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/9cbc1212c5527a2a23976b17e4079951 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/a3c01001f0605feb13e4321cf29bbd6b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/a3c01001f0605feb13e4321cf29bbd6b
new file mode 100644
index 0000000000000000000000000000000000000000..b97d0fbdf5c84bb0e4306297c2057e8410c6c86f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/a3c01001f0605feb13e4321cf29bbd6b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5816b01b120b16680b539ce840d223a68458c83823f04c6bf0e466f7eecbed7e
+size 451349
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/ab1314c572d6182ce4ad2642e7910232 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/ab1314c572d6182ce4ad2642e7910232
new file mode 100644
index 0000000000000000000000000000000000000000..f28d59e1f9187cfd65b688b4bceb57a81f0acdcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/ab1314c572d6182ce4ad2642e7910232 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/b5ecd60d74d345c68a0e29bb2cab9dd1 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/b5ecd60d74d345c68a0e29bb2cab9dd1
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/b5ecd60d74d345c68a0e29bb2cab9dd1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/c9faac655914bd64f806b7b61481568a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/c9faac655914bd64f806b7b61481568a
new file mode 100644
index 0000000000000000000000000000000000000000..d5a1586ead772e9e7f1943234fc38798492a2940
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d/c9faac655914bd64f806b7b61481568a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..6af769efffe8e0ffa4e42e6d7a94b98f7f83409f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..5987d99d0460daf84a4e3883f931d53eae938db6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590771480897977, "commit_timestamp_nsecs": 1754590771661896235, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/d/81bbc2aca3f6cec870f8ace81eb92d56 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/d/81bbc2aca3f6cec870f8ace81eb92d56
new file mode 100644
index 0000000000000000000000000000000000000000..858e7f65d6d0f35b52c0e9e1055eb62e9df80a00
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/d/81bbc2aca3f6cec870f8ace81eb92d56 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..67321929196e4270722e9aaea54490a61f28db8f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/0d8d09f4bfd85951e554141e4dfd9bd1 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/0d8d09f4bfd85951e554141e4dfd9bd1
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/0d8d09f4bfd85951e554141e4dfd9bd1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/3a6d499d5a46aec5dfa2793d10193d22 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/3a6d499d5a46aec5dfa2793d10193d22
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/3a6d499d5a46aec5dfa2793d10193d22 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/527a4b9a7e65ad511b03e1a7b9a90e4d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/527a4b9a7e65ad511b03e1a7b9a90e4d
new file mode 100644
index 0000000000000000000000000000000000000000..60d6b3d607c92119cdffbe1467b72d0fe616dcdb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/527a4b9a7e65ad511b03e1a7b9a90e4d
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1b02a2221b56a32e59e86c817657d1d6d86bb4bbab361fff9cc4e6179d982414
+size 253505
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/57e9cbf78fa54c0ec0566d0f84d7c684 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/57e9cbf78fa54c0ec0566d0f84d7c684
new file mode 100644
index 0000000000000000000000000000000000000000..3dfdfd886fdedcc923b03fde1a795a5bf93b5d25
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/57e9cbf78fa54c0ec0566d0f84d7c684 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/58637e1a82a5d4c9070618cff8a060e2 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/58637e1a82a5d4c9070618cff8a060e2
new file mode 100644
index 0000000000000000000000000000000000000000..7b917fc793754ba975f0ae9ac8289e1ccedd66e7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/58637e1a82a5d4c9070618cff8a060e2
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:efcb0c1ff20b4e2155f7c0108bd9128fc26057e61bf72e01a897555d2c7131b5
+size 485238
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/649fe6d03a0daff477ac66327a2168ba b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/649fe6d03a0daff477ac66327a2168ba
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/649fe6d03a0daff477ac66327a2168ba differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/718ea1d4a4aa98fddb991ceeb60b6697 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/718ea1d4a4aa98fddb991ceeb60b6697
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/718ea1d4a4aa98fddb991ceeb60b6697 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/fbccc33d72965323964040b5bc76f325 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/fbccc33d72965323964040b5bc76f325
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d/fbccc33d72965323964040b5bc76f325 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..451300b936422287dba9a013ec715f2556777558
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..c9010ebb0ceba3b5563e0a03a6051f5e848c3b3f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590804944825490, "commit_timestamp_nsecs": 1754590805126754151, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/d/2c1acfe19f655c3f9e9a633b16a01566 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/d/2c1acfe19f655c3f9e9a633b16a01566
new file mode 100644
index 0000000000000000000000000000000000000000..a22901c49e8d97926b78cb165c49b16c52d36070
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/d/2c1acfe19f655c3f9e9a633b16a01566 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9555aaa400932ef7bed9b358602a771ffce70ada
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/444a94924327002ee0d7a6db0ba672d7 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/444a94924327002ee0d7a6db0ba672d7
new file mode 100644
index 0000000000000000000000000000000000000000..4041e9a5a0c60f81533046f828897c20666c81c0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/444a94924327002ee0d7a6db0ba672d7
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aa0953f5a18f0013fdb52d7a9ddbec0efbb3e5ac1bf59c98fe2acfd3fe3098d1
+size 473510
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/540684bcd469b9bb3b0dc4e9e61f1a34 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/540684bcd469b9bb3b0dc4e9e61f1a34
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/540684bcd469b9bb3b0dc4e9e61f1a34 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/6e13acdafa3a952576ed801cb34ec2d6 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/6e13acdafa3a952576ed801cb34ec2d6
new file mode 100644
index 0000000000000000000000000000000000000000..793283a2e5ded457abc0b9065847cbb0b797f73b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/6e13acdafa3a952576ed801cb34ec2d6
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bf6909b75438d590bb8688cfd11dbcac6dbf8080129e08cf8fd9a4a788a16496
+size 227815
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/a9e76ad8ab82983e8ba355adf0598267 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/a9e76ad8ab82983e8ba355adf0598267
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/a9e76ad8ab82983e8ba355adf0598267 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/abdd7cc3b3381210efe75b9e5289b077 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/abdd7cc3b3381210efe75b9e5289b077
new file mode 100644
index 0000000000000000000000000000000000000000..970b9accc1275b3144369b6dcd0429979e8ed1c6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/abdd7cc3b3381210efe75b9e5289b077 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/b1e310c257dae3838439ffb38a8fe59d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/b1e310c257dae3838439ffb38a8fe59d
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/b1e310c257dae3838439ffb38a8fe59d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/ca5bda2ef1c79042444221550a94b91e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/ca5bda2ef1c79042444221550a94b91e
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/ca5bda2ef1c79042444221550a94b91e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/fdd516ffd4a6fa32cabc1cf345021e86 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/fdd516ffd4a6fa32cabc1cf345021e86
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d/fdd516ffd4a6fa32cabc1cf345021e86 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e5d5d1140c361798318858cbd2518de38cc02b7c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..06efefadf0c45fc5bca3846c7585464f3943a67e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590838259023187, "commit_timestamp_nsecs": 1754590838445095164, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/d/0a616a6d97fd427ca05a095e5e587081 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/d/0a616a6d97fd427ca05a095e5e587081
new file mode 100644
index 0000000000000000000000000000000000000000..330cc16fcff1a8c1fe73be3df1e13d4ee8e0843a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/d/0a616a6d97fd427ca05a095e5e587081 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5c75be1e57869a611de66c09bcb23aa6f6de5b39
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/036a3d741b4f90b69850304a1f1fc0b1 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/036a3d741b4f90b69850304a1f1fc0b1
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/036a3d741b4f90b69850304a1f1fc0b1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/7032e98c2fd90776127d4e51c75f1161 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/7032e98c2fd90776127d4e51c75f1161
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/7032e98c2fd90776127d4e51c75f1161 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/72079c59a1eff30cfb1cb076f6332bec b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/72079c59a1eff30cfb1cb076f6332bec
new file mode 100644
index 0000000000000000000000000000000000000000..20e6f735ebf1e3bdb7baca6de612a36ca1a4c711
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/72079c59a1eff30cfb1cb076f6332bec
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:52eeb89c2fd6d6333fd16fc7affb1da5a77f2b7269df6b0b89623893e45d2233
+size 442782
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/80ca426622028576298253e11ce13419 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/80ca426622028576298253e11ce13419
new file mode 100644
index 0000000000000000000000000000000000000000..2910fbc6db2094404ed932c7d9bcdb23beb99906
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/80ca426622028576298253e11ce13419 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/91795599879704b53ebd5608c36ebf4f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/91795599879704b53ebd5608c36ebf4f
new file mode 100644
index 0000000000000000000000000000000000000000..0483d197e36a85164727464a28061ee7ab3630e2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/91795599879704b53ebd5608c36ebf4f differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/9e2625313688f7fffcef924c1ff4ce7b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/9e2625313688f7fffcef924c1ff4ce7b
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/9e2625313688f7fffcef924c1ff4ce7b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/c14b41a31c406628e1fb6dde71a1969f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/c14b41a31c406628e1fb6dde71a1969f
new file mode 100644
index 0000000000000000000000000000000000000000..85f35e7991cac7b6fc244bb2f277438119f4e90b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/c14b41a31c406628e1fb6dde71a1969f differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/c62075cad09f05f1cb5441890fed9d7f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/c62075cad09f05f1cb5441890fed9d7f
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/c62075cad09f05f1cb5441890fed9d7f differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/d225e269b915f80ee6b566ddb16ace28 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/d225e269b915f80ee6b566ddb16ace28
new file mode 100644
index 0000000000000000000000000000000000000000..496ac0ca96680f867815db7a9fee8e48ebb87340
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d/d225e269b915f80ee6b566ddb16ace28
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:acd9bc91bcb0c54894b4447ed3c2d622b6114bb3292957ce825bc1bffe76dea1
+size 295974
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b127d891d2a6adfccc6cda40331ecfa5d539580e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..c884969d23a222f282d7846c9eecb83a9f6ba37b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590871731030657, "commit_timestamp_nsecs": 1754590871876392865, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/d/ca47d331013ce2c2f166825578935de4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/d/ca47d331013ce2c2f166825578935de4
new file mode 100644
index 0000000000000000000000000000000000000000..0e5687bce7620d60df1070bdd043e93b883c5ddb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/d/ca47d331013ce2c2f166825578935de4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5015ecdd182861d0d634d750b8d29e260e3e798c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/352faf90c4c126e6505fce64ce48de50 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/352faf90c4c126e6505fce64ce48de50
new file mode 100644
index 0000000000000000000000000000000000000000..0388e886447cbad2af0048511136360687927c4c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/352faf90c4c126e6505fce64ce48de50
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9976ee089804ff52964c3797520c49816db8d174019ed406a3bc335940df6026
+size 296067
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/543897bd695f0504a303f4ac75693959 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/543897bd695f0504a303f4ac75693959
new file mode 100644
index 0000000000000000000000000000000000000000..68907e159e8a0ac347c47b19903486a5d5f75cd1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/543897bd695f0504a303f4ac75693959
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d7f507fcbdfb409af84150b146058e5b704aa1c983cadcd7479022cb5ce6f0de
+size 406591
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/8a061b5b055d9ab2ee4c27fb5fc7346e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/8a061b5b055d9ab2ee4c27fb5fc7346e
new file mode 100644
index 0000000000000000000000000000000000000000..4af5706272dee3386bdd4399969c129a9d54a62c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/8a061b5b055d9ab2ee4c27fb5fc7346e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/9090ebfb4d0bdc591b4827336229510f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/9090ebfb4d0bdc591b4827336229510f
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/9090ebfb4d0bdc591b4827336229510f differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/a7dd3f886acde1ccea025cc9905fb0d4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/a7dd3f886acde1ccea025cc9905fb0d4
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/a7dd3f886acde1ccea025cc9905fb0d4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/b34f61fcbde3d1e91a0bc616976bb186 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/b34f61fcbde3d1e91a0bc616976bb186
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/b34f61fcbde3d1e91a0bc616976bb186 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/e597b293b6bacc0b490bbccb170d1710 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/e597b293b6bacc0b490bbccb170d1710
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d/e597b293b6bacc0b490bbccb170d1710 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9cbf2de92dcc56c2d30623bedb53c581dedcb444
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..2acc0c28d9825d3fc1106ba065a2afe09190fdc8
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754589992471868595, "commit_timestamp_nsecs": 1754589992662827792, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/d/f12950777ccc438c61b21970da5168f6 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/d/f12950777ccc438c61b21970da5168f6
new file mode 100644
index 0000000000000000000000000000000000000000..e02ac8223822eb3fcea883a29b89481961123fc7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/d/f12950777ccc438c61b21970da5168f6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..669e896fe0d3fbae40969117968f5029926d367a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/0fff3caa8f8dc214356c1e6418727aec b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/0fff3caa8f8dc214356c1e6418727aec
new file mode 100644
index 0000000000000000000000000000000000000000..c2556e3154d050f70b68617e1e130a6557b41f76
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/0fff3caa8f8dc214356c1e6418727aec differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/15b0b3d5dcca1de2889e5689650a0c85 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/15b0b3d5dcca1de2889e5689650a0c85
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/15b0b3d5dcca1de2889e5689650a0c85 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/5b4dbe2828600cf4feb89cc31141e7eb b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/5b4dbe2828600cf4feb89cc31141e7eb
new file mode 100644
index 0000000000000000000000000000000000000000..333f4dd6dbe4dd4d937e32c1d6f113d202d1c1af
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/5b4dbe2828600cf4feb89cc31141e7eb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:88281cfed57d803be1e3b60a966ac135034f68bf1c10108c7f40a089546f4ecf
+size 433631
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/731778112f4213a0a8d2c22f4aec0222 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/731778112f4213a0a8d2c22f4aec0222
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/731778112f4213a0a8d2c22f4aec0222 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/972f45bd2148cd7f8f8b058342f802a2 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/972f45bd2148cd7f8f8b058342f802a2
new file mode 100644
index 0000000000000000000000000000000000000000..4440abf36022fca1c397d7ac3f0813153ba8100e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/972f45bd2148cd7f8f8b058342f802a2
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5ca6fcfc4f631c28edd947830462227e4b7edf7e6a358fbc33bfafb1ad2d145d
+size 304158
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/a9542926e6ed23095a08db564589f4bd b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/a9542926e6ed23095a08db564589f4bd
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/a9542926e6ed23095a08db564589f4bd differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/df525f5ed304907e23aab790247ec361 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/df525f5ed304907e23aab790247ec361
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d/df525f5ed304907e23aab790247ec361 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..fa58f02298ee25d679bcd6b063141ce8ad3497c3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..45fc00da959bf2a3fad14764b944f137f8e7583f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590906326297281, "commit_timestamp_nsecs": 1754590906479659522, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/d/0352fab3cef87120e7ad6c5c99b587d3 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/d/0352fab3cef87120e7ad6c5c99b587d3
new file mode 100644
index 0000000000000000000000000000000000000000..abfb42656e652e4809ceec9ffb76fc1085df0f32
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/d/0352fab3cef87120e7ad6c5c99b587d3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a061747944a69d417cfe2e429faa3edf3592c0f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/02ab3d5664c9f40eca95961c0dda2566 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/02ab3d5664c9f40eca95961c0dda2566
new file mode 100644
index 0000000000000000000000000000000000000000..8b2b921fa727de84551add30daf2d0119e3a38ed
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/02ab3d5664c9f40eca95961c0dda2566
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a797a7ec7be04e854095a652de4c402bd1bfd3d24b4e68348dfe385ea6b1f244
+size 331920
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/288348de7830e06fbd69025f70b3724b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/288348de7830e06fbd69025f70b3724b
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/288348de7830e06fbd69025f70b3724b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/36708313dda9523c6d97e3e21aafe310 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/36708313dda9523c6d97e3e21aafe310
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/36708313dda9523c6d97e3e21aafe310 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/5a3e25f23085a77f33ef72aafad59ddf b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/5a3e25f23085a77f33ef72aafad59ddf
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/5a3e25f23085a77f33ef72aafad59ddf differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/7e3073aa787478da5053d538e427dedb b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/7e3073aa787478da5053d538e427dedb
new file mode 100644
index 0000000000000000000000000000000000000000..aec0f1d16d4b28181ea5da6b5b78dd8c3b321cf9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/7e3073aa787478da5053d538e427dedb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:36fc2370690d1ba129128c9d45fcbc5be15882eef7dbae5897b3b0125f141a91
+size 406609
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/ab222ec2061f76e24a0f2be9833dc1b5 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/ab222ec2061f76e24a0f2be9833dc1b5
new file mode 100644
index 0000000000000000000000000000000000000000..e1ba7d2d38cecb6f9e855e732ab2246af0a15678
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/ab222ec2061f76e24a0f2be9833dc1b5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/ba82f2c511653dedcf262fb515c3063a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/ba82f2c511653dedcf262fb515c3063a
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d/ba82f2c511653dedcf262fb515c3063a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5d402009df7613bf84ad053fe1848a6a02dec138
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..8034a1c4413086ac1d774e01320f219a4cce1382
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590939750568710, "commit_timestamp_nsecs": 1754590939920681619, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/d/ab55f536bafd641a5be46d359f6c9da3 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/d/ab55f536bafd641a5be46d359f6c9da3
new file mode 100644
index 0000000000000000000000000000000000000000..932ca9b916ab7e6af157c33dac1cd4d492248a95
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/d/ab55f536bafd641a5be46d359f6c9da3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..63039e74ef1f51691c8b8f8f665fae2a4f18d0b3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/50abd9037c9994d529f29c836c55766c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/50abd9037c9994d529f29c836c55766c
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/50abd9037c9994d529f29c836c55766c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/9480e3c7a07fdadb9913aa55bb8ab0dd b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/9480e3c7a07fdadb9913aa55bb8ab0dd
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/9480e3c7a07fdadb9913aa55bb8ab0dd differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/9a961f689f8ce1539d42b21d1e7831c9 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/9a961f689f8ce1539d42b21d1e7831c9
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/9a961f689f8ce1539d42b21d1e7831c9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/a91a093a9b8a8c379ad690b1648d59f8 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/a91a093a9b8a8c379ad690b1648d59f8
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/a91a093a9b8a8c379ad690b1648d59f8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/a9eb2ec0432b8292f7681063d22f33f1 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/a9eb2ec0432b8292f7681063d22f33f1
new file mode 100644
index 0000000000000000000000000000000000000000..63e8f43bc55fddd90437e33cc1177e0b25ae217e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/a9eb2ec0432b8292f7681063d22f33f1
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cf5862f5efb1509c52d5482f5cd5f8cbd6597a4324f8bfea2f7a0f086291a0f5
+size 391507
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/bec749308231ee5d5957917245f74166 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/bec749308231ee5d5957917245f74166
new file mode 100644
index 0000000000000000000000000000000000000000..47a2e8b8ed261184597a7e31dad575c9e8030400
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/bec749308231ee5d5957917245f74166 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/fc97a770d0c6ae0635d8ab0fe021a397 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/fc97a770d0c6ae0635d8ab0fe021a397
new file mode 100644
index 0000000000000000000000000000000000000000..13a7dbee033455e15ae5aea408222310befbcb9d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d/fc97a770d0c6ae0635d8ab0fe021a397
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3d51bd4e11572c720e1990d30ed99bb34057550a831415e807fce3306fbecc70
+size 346974
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5d9f3600671b2b7e76ee9521b60b5bf94426d2ea
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..3eca5e83c318119b050b9acc7a9263080e6b779a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590987285752849, "commit_timestamp_nsecs": 1754590987456691352, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/d/a6a545f3e68a027b774eff778c47fc48 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/d/a6a545f3e68a027b774eff778c47fc48
new file mode 100644
index 0000000000000000000000000000000000000000..5f8bf432ae015bc75ade87b6e65830be4aa0de3e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/d/a6a545f3e68a027b774eff778c47fc48 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3882945be393b7d396edb474863be2bacbcc0f51
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/243db48a89e8995a002611b124369c2f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/243db48a89e8995a002611b124369c2f
new file mode 100644
index 0000000000000000000000000000000000000000..b990efdb14c5deaff6a40787074149cc08025f54
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/243db48a89e8995a002611b124369c2f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bb8b60e6cbc186d20bcbc2166dbaac5d7b10cc584aebf63a299fbe4bc0022a42
+size 545156
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/299c1014f14a90e131fe5e087423b318 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/299c1014f14a90e131fe5e087423b318
new file mode 100644
index 0000000000000000000000000000000000000000..1e8c9696a97786b3d21837af65a401fc732aa7e4
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/299c1014f14a90e131fe5e087423b318
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fbecc51c0a61b4713d654e33f39c721989b37f5993f3c2044a90eefe0e3f27ad
+size 193849
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/7f47a53c479d1fb48539dc6884c51b26 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/7f47a53c479d1fb48539dc6884c51b26
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/7f47a53c479d1fb48539dc6884c51b26 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/a14adb95b5113b9f82f10df686415187 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/a14adb95b5113b9f82f10df686415187
new file mode 100644
index 0000000000000000000000000000000000000000..822a556b1fa3c2cb4f7586def87695ad36c5aeb3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/a14adb95b5113b9f82f10df686415187 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/aaa52c06ee508c5e52ff2f10ce6be7e0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/aaa52c06ee508c5e52ff2f10ce6be7e0
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/aaa52c06ee508c5e52ff2f10ce6be7e0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/b7cc72eebb144c62288a948d668b7314 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/b7cc72eebb144c62288a948d668b7314
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/b7cc72eebb144c62288a948d668b7314 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/bbd6870889c644d99aa027587440cd08 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/bbd6870889c644d99aa027587440cd08
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/bbd6870889c644d99aa027587440cd08 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/ec9078d09142991df3d7a9c18c271f0f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/ec9078d09142991df3d7a9c18c271f0f
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d/ec9078d09142991df3d7a9c18c271f0f differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b719246bd1e02065658d6b11287a1c85bbbc3093
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..e2cc0d437acc2cd1daddaf7a41aaa175096a0ac0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591020425946843, "commit_timestamp_nsecs": 1754591020568019329, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/d/522e5101cbf2250a32c59061dddac139 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/d/522e5101cbf2250a32c59061dddac139
new file mode 100644
index 0000000000000000000000000000000000000000..9d75775deaf8d5fba9193fecbceb9637d44132a5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/d/522e5101cbf2250a32c59061dddac139 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3af09af06a92d2aaaf10be3a2fda8abb33870f1c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/2e0db4fa97e3aad33851f45b98837acc b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/2e0db4fa97e3aad33851f45b98837acc
new file mode 100644
index 0000000000000000000000000000000000000000..6042eda1a284da298c3839d8cdb83df36e296de8
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/2e0db4fa97e3aad33851f45b98837acc
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:317e9a8156f7d2df421fe38981b333d130f8e04f73e12bd7f0a4010d596f4b3c
+size 366002
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/579a01a33f1fe08d08219bf6e53a73d4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/579a01a33f1fe08d08219bf6e53a73d4
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/579a01a33f1fe08d08219bf6e53a73d4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/8942c75c85d1249a4380b90efe714de8 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/8942c75c85d1249a4380b90efe714de8
new file mode 100644
index 0000000000000000000000000000000000000000..85f35e7991cac7b6fc244bb2f277438119f4e90b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/8942c75c85d1249a4380b90efe714de8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/96e2db1aead93baac03e8e9990ec842f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/96e2db1aead93baac03e8e9990ec842f
new file mode 100644
index 0000000000000000000000000000000000000000..318eeb12897356c1420191728b22d6be67f642e1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/96e2db1aead93baac03e8e9990ec842f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d343336039c9008bcad1124296ae5f51587d85305a66897745e2a56cdac1c540
+size 372670
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/a08b6574fcee188f58b2e3e991dd6630 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/a08b6574fcee188f58b2e3e991dd6630
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/a08b6574fcee188f58b2e3e991dd6630 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/ceb829749385637bce444520f7a34f07 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/ceb829749385637bce444520f7a34f07
new file mode 100644
index 0000000000000000000000000000000000000000..9107d220e637a03df806bb9cd3e468032c4e0252
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/ceb829749385637bce444520f7a34f07 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/e3eb8b222849bc08640718105f305c4f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/e3eb8b222849bc08640718105f305c4f
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/e3eb8b222849bc08640718105f305c4f differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/fdee3c537d5ba1488ea56fd9d203c109 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/fdee3c537d5ba1488ea56fd9d203c109
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d/fdee3c537d5ba1488ea56fd9d203c109 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9cbb81281eb0a31a031b48d3f0b5d15743a4b9cd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..d6a782be8b680582ef475d31db5585483e2d95d7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591053259490646, "commit_timestamp_nsecs": 1754591053434506438, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/d/debf4b6e5ccd6d523993118eb2af66d8 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/d/debf4b6e5ccd6d523993118eb2af66d8
new file mode 100644
index 0000000000000000000000000000000000000000..c4f981e5b9905db1c0ef23edbcd40951a3a2afe8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/d/debf4b6e5ccd6d523993118eb2af66d8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d48a0889faad59f4438e75da2088edf86b536f33
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/6305cccfa91f18959e497946f979e230 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/6305cccfa91f18959e497946f979e230
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/6305cccfa91f18959e497946f979e230 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/87fb7974add0746c27915f27c278bfc0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/87fb7974add0746c27915f27c278bfc0
new file mode 100644
index 0000000000000000000000000000000000000000..d5a8ece770e9ec461c693f088920a6992ebaabff
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/87fb7974add0746c27915f27c278bfc0
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ab34da0087ba775714cdfb24679ed2d4c9d85a49f8638df079fc24c541ce455f
+size 278990
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/a4fe754d3d35596dfb446e15bb1a48d2 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/a4fe754d3d35596dfb446e15bb1a48d2
new file mode 100644
index 0000000000000000000000000000000000000000..4859214e6b68fa6e697de9748417fcf43be8081d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/a4fe754d3d35596dfb446e15bb1a48d2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/b7fc7c0a7b6b4363eef9895c35d73389 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/b7fc7c0a7b6b4363eef9895c35d73389
new file mode 100644
index 0000000000000000000000000000000000000000..e189943a28d57263882d29e7dc45ef12f9363bb8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/b7fc7c0a7b6b4363eef9895c35d73389 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/ba3541d4b3c3b8e254ccb51a270deb40 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/ba3541d4b3c3b8e254ccb51a270deb40
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/ba3541d4b3c3b8e254ccb51a270deb40 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/d1f45fc2b4c1b8845ea48f7cd9bc1571 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/d1f45fc2b4c1b8845ea48f7cd9bc1571
new file mode 100644
index 0000000000000000000000000000000000000000..c5cf0129bb63b688d4fed3bc02a71376e62eb7ea
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/d1f45fc2b4c1b8845ea48f7cd9bc1571
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d54953f277cc70ad2c788cc85998fbaf6a58b331581b50f8b0326e84384baa00
+size 421357
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/d2d8ec71943d003e4eb5a49cfdfbba19 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/d2d8ec71943d003e4eb5a49cfdfbba19
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/d2d8ec71943d003e4eb5a49cfdfbba19 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/e81cbbc32c3177c64f6e66431b069e6c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/e81cbbc32c3177c64f6e66431b069e6c
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/e81cbbc32c3177c64f6e66431b069e6c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/fa3e018a8d990ecc9cb7c5e85e7b1fe9 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/fa3e018a8d990ecc9cb7c5e85e7b1fe9
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d/fa3e018a8d990ecc9cb7c5e85e7b1fe9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e0f73643c057d9d12dea78b6b6f17f3eea4a92ca
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..b0124d48d74edbcd5ce13bd80ea7a857d7d16e1a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591086643325068, "commit_timestamp_nsecs": 1754591086817002449, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/d/7b7a95d4b7943c3b1e7a90af2cacdc3b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/d/7b7a95d4b7943c3b1e7a90af2cacdc3b
new file mode 100644
index 0000000000000000000000000000000000000000..f421654c01441cf4848d67c241222d2c4bafc45a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/d/7b7a95d4b7943c3b1e7a90af2cacdc3b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1434ad6dccf34b36645ad66ea389bbd33ef08fef
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/4fff7c23d610539851c692dedc0f8ce2 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/4fff7c23d610539851c692dedc0f8ce2
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/4fff7c23d610539851c692dedc0f8ce2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/5b284a88e5a67d65f8cf60e2bbc0b535 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/5b284a88e5a67d65f8cf60e2bbc0b535
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/5b284a88e5a67d65f8cf60e2bbc0b535 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/8c39ae423813feb976a16643a13561bd b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/8c39ae423813feb976a16643a13561bd
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/8c39ae423813feb976a16643a13561bd differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/986500dc838e245da22cadfe8e869292 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/986500dc838e245da22cadfe8e869292
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/986500dc838e245da22cadfe8e869292 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/c3306ae047cbf246ee536c6f2da064f6 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/c3306ae047cbf246ee536c6f2da064f6
new file mode 100644
index 0000000000000000000000000000000000000000..140a75cf9292aa382a1ee2042536d2edb2f482a7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/c3306ae047cbf246ee536c6f2da064f6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/c7dba419eb1b23b9f0a016e11165fd19 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/c7dba419eb1b23b9f0a016e11165fd19
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/c7dba419eb1b23b9f0a016e11165fd19 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/ce7ddde8ac3da3ce60e1fdcef1a29219 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/ce7ddde8ac3da3ce60e1fdcef1a29219
new file mode 100644
index 0000000000000000000000000000000000000000..7df8bda775f3960c558eb0d3c3370f2c2ecf83b7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/ce7ddde8ac3da3ce60e1fdcef1a29219
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0e073149c360ecdd38c241bb8acfe76cb32df4fbde95e55fb7eb43490f882cba
+size 227874
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/e5f11efceb463ac6827d91267a9d4260 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/e5f11efceb463ac6827d91267a9d4260
new file mode 100644
index 0000000000000000000000000000000000000000..52bfac9017c1eafcfbbb6a0f27e484ec0fc2016e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d/e5f11efceb463ac6827d91267a9d4260
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b528c2e55b49906c7351858c052796ee958cec24dcd86cc3468a335bfdac074d
+size 477210
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..aa314f84e96a5d5e78ec6ce3e6509f11e65f79c4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..e08616f9cb61c428de07854876da6d87732c7f6b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591120092006563, "commit_timestamp_nsecs": 1754591120257158530, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/d/802765bcec0af6ffc8432b1eeea07498 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/d/802765bcec0af6ffc8432b1eeea07498
new file mode 100644
index 0000000000000000000000000000000000000000..718a1e71da71e79ab660f0122b96189b8477b455
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/d/802765bcec0af6ffc8432b1eeea07498 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3c7296c6e91f4896cce4db8c78c6ca616a396d91
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/015d9398d6995f971e39957c1eb45393 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/015d9398d6995f971e39957c1eb45393
new file mode 100644
index 0000000000000000000000000000000000000000..2a31377ca030aced51ef70fb458c6b5c47404cd3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/015d9398d6995f971e39957c1eb45393 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/192e1f5a4564fdfdd9314102cd9e4ce5 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/192e1f5a4564fdfdd9314102cd9e4ce5
new file mode 100644
index 0000000000000000000000000000000000000000..e6e6206f8aaa0a745cd65310847a56678233252a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/192e1f5a4564fdfdd9314102cd9e4ce5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/23f1f13a975e21fef2fda26088326290 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/23f1f13a975e21fef2fda26088326290
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/23f1f13a975e21fef2fda26088326290 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/2a52f93e996b4e105644e8533a250f3b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/2a52f93e996b4e105644e8533a250f3b
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/2a52f93e996b4e105644e8533a250f3b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/2abc2c4ab40feedee4a31f3d59e6da53 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/2abc2c4ab40feedee4a31f3d59e6da53
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/2abc2c4ab40feedee4a31f3d59e6da53 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/bc652ef6d6d7123974a960aa876b9ec7 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/bc652ef6d6d7123974a960aa876b9ec7
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/bc652ef6d6d7123974a960aa876b9ec7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/c90bee4582842a8cd8d67087509cc38e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/c90bee4582842a8cd8d67087509cc38e
new file mode 100644
index 0000000000000000000000000000000000000000..31a8049ebf770b9e6d79aceafec4be4ba2226025
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/c90bee4582842a8cd8d67087509cc38e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b65ddbc920ea5cfac8b31e90b5548d464ee082ccb2d17961d7f1041eae4cf187
+size 260850
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/e00b39be644cb3c8cde260e6de14c159 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/e00b39be644cb3c8cde260e6de14c159
new file mode 100644
index 0000000000000000000000000000000000000000..d06e0aff9da7aea2c536fd0daffa965c514cee16
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/e00b39be644cb3c8cde260e6de14c159
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d59172eb7627e48d665d588e7a8e7ebf204a1d9f1d3802da91c90631c8b33795
+size 476065
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/e9ad307a60fab0e2d80c3685aee6639d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/e9ad307a60fab0e2d80c3685aee6639d
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d/e9ad307a60fab0e2d80c3685aee6639d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e4d504a4eb107e29ea3cd61846e4af53f077f428
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..3909ea5667f8f09cbd5c918b349e96bbfef87b3f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591153955127939, "commit_timestamp_nsecs": 1754591154126775609, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/d/7657060c9b0b2c728cebd1ab9faf9dde b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/d/7657060c9b0b2c728cebd1ab9faf9dde
new file mode 100644
index 0000000000000000000000000000000000000000..8852f6cbd4d067b036f804f062ea7dd8fa3d0378
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/d/7657060c9b0b2c728cebd1ab9faf9dde differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a48caf8cfa615dfa10c1b273d2686a76424d7bb6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/033762eaca667a273a156650092c0926 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/033762eaca667a273a156650092c0926
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/033762eaca667a273a156650092c0926 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/19cb35c025c6ace0c071dc2d1d5bafc9 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/19cb35c025c6ace0c071dc2d1d5bafc9
new file mode 100644
index 0000000000000000000000000000000000000000..6a3a4f00e623e53ab243cb931912c1614e5049fb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/19cb35c025c6ace0c071dc2d1d5bafc9
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:54e2006879cf87ebb9281441f0d30ff2bf47c151c068d5185a151871b67bec0e
+size 446592
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/27e98afbe70ef79551430818fef26a51 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/27e98afbe70ef79551430818fef26a51
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/27e98afbe70ef79551430818fef26a51 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/47ec638015eaf29f87869814f6c8d3c2 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/47ec638015eaf29f87869814f6c8d3c2
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/47ec638015eaf29f87869814f6c8d3c2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/574a23ef9e2c08956c10b6196e3223ec b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/574a23ef9e2c08956c10b6196e3223ec
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/574a23ef9e2c08956c10b6196e3223ec differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/b135efe041e9e1ed8da4841051571c1d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/b135efe041e9e1ed8da4841051571c1d
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/b135efe041e9e1ed8da4841051571c1d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/b1cd6cabe3413f3ffcb749257c6d7dc4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/b1cd6cabe3413f3ffcb749257c6d7dc4
new file mode 100644
index 0000000000000000000000000000000000000000..e720692e9ffd2364afbd0c0f520c315165a4b611
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/b1cd6cabe3413f3ffcb749257c6d7dc4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/dde4376df3dc35b123a1bcad5fbbb3b9 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/dde4376df3dc35b123a1bcad5fbbb3b9
new file mode 100644
index 0000000000000000000000000000000000000000..7597df46a1fcb2feece144a72db3aadfe28b65f0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d/dde4376df3dc35b123a1bcad5fbbb3b9
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:26c481d47655620ae6c63d6de84677a5c32b10a0c078b65d227beb6af508af9a
+size 252512
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1996c5589d2415bba6b2dcd90a41770411b4d9ed
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..49347a5b1e5a6aa48224d870fb3ed0f16352336e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591185733390818, "commit_timestamp_nsecs": 1754591185877153930, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/d/94940bfb5072cc3092bd8a8d3d5cfa0d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/d/94940bfb5072cc3092bd8a8d3d5cfa0d
new file mode 100644
index 0000000000000000000000000000000000000000..59e51c38fbf062d2c92814597167f61fcb0c5e0c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/d/94940bfb5072cc3092bd8a8d3d5cfa0d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..29d8c2e826b4b715c215d4dd16bae96e9e8d69e4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/29ace86030159a2e55a7555c8f9e9492 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/29ace86030159a2e55a7555c8f9e9492
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/29ace86030159a2e55a7555c8f9e9492 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/359ef5952164faf489a4329ecce3d2fc b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/359ef5952164faf489a4329ecce3d2fc
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/359ef5952164faf489a4329ecce3d2fc differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/594f091b9a32730f561344b43c1e3e33 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/594f091b9a32730f561344b43c1e3e33
new file mode 100644
index 0000000000000000000000000000000000000000..c638628097db3eef138ac67f7fe6467ff1ba85f0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/594f091b9a32730f561344b43c1e3e33
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1c510f18ee94dfbf05519786976b81ae8c9d03ccccb85e2334ea98d666fa297b
+size 253531
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/6dc5f193ea43472a0ce4734284e68185 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/6dc5f193ea43472a0ce4734284e68185
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/6dc5f193ea43472a0ce4734284e68185 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/9044cddcbff97e56cfcdce777d7c7327 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/9044cddcbff97e56cfcdce777d7c7327
new file mode 100644
index 0000000000000000000000000000000000000000..ae774d857167ec4039c61c33d939648ef6e38763
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/9044cddcbff97e56cfcdce777d7c7327
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:66c91dc7ce46290c9656386ed9cb7424ab1b94d19e9331fcb67f9d0c30630ee1
+size 445508
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/a4330a0c5a63a83e0b69042f5f2f7a9d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/a4330a0c5a63a83e0b69042f5f2f7a9d
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/a4330a0c5a63a83e0b69042f5f2f7a9d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/b6704077984785dce813f6abe9027edf b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/b6704077984785dce813f6abe9027edf
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/b6704077984785dce813f6abe9027edf differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/f95eac2beab736f9f7b1a325db05dfe2 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/f95eac2beab736f9f7b1a325db05dfe2
new file mode 100644
index 0000000000000000000000000000000000000000..2a246702840a7072f43dd3ec7a61ad487edd0139
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d/f95eac2beab736f9f7b1a325db05dfe2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..230bfac4faac4bc33f6b4249a1219255a468a94b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..011730c7e2dac6c474c5d8123933c304bfb829b3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591219467209882, "commit_timestamp_nsecs": 1754591219653789810, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/d/3917b809b8bfc848018756e488d8ee6d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/d/3917b809b8bfc848018756e488d8ee6d
new file mode 100644
index 0000000000000000000000000000000000000000..fd0e2ede75fe4596d26f8921d417a35cc3b7581a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/d/3917b809b8bfc848018756e488d8ee6d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..07b6f126d07db5626509d719e502716cf1b26ee6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/0c565dfc05aaa7d44888c0d7aaa368c0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/0c565dfc05aaa7d44888c0d7aaa368c0
new file mode 100644
index 0000000000000000000000000000000000000000..85f35e7991cac7b6fc244bb2f277438119f4e90b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/0c565dfc05aaa7d44888c0d7aaa368c0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/50fdc269263961c54b089fce2511d2e3 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/50fdc269263961c54b089fce2511d2e3
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/50fdc269263961c54b089fce2511d2e3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/51e2693c41cf7532f6247b2c948d9274 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/51e2693c41cf7532f6247b2c948d9274
new file mode 100644
index 0000000000000000000000000000000000000000..474a316d628cde3c32f4b4798d427ec23decb632
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/51e2693c41cf7532f6247b2c948d9274
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7354a15ffe85140d771da698da62f76ecad7c3c514d9a24bef6b9d84fd4c3636
+size 456185
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/9ada98bed10fc78fc6e155598c7b3f6d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/9ada98bed10fc78fc6e155598c7b3f6d
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/9ada98bed10fc78fc6e155598c7b3f6d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/a5cae38a9909aedade4ca6eadf3fd331 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/a5cae38a9909aedade4ca6eadf3fd331
new file mode 100644
index 0000000000000000000000000000000000000000..82d685127109b077d5f3386e0000d07db66daa06
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/a5cae38a9909aedade4ca6eadf3fd331 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/b8ac7f9005443075a68dcefc5a1c8d76 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/b8ac7f9005443075a68dcefc5a1c8d76
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/b8ac7f9005443075a68dcefc5a1c8d76 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/cce017e1557e33a3accb691cbc2fe726 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/cce017e1557e33a3accb691cbc2fe726
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/cce017e1557e33a3accb691cbc2fe726 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/d6a3342305651bc8d26dd04b4d0a6967 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/d6a3342305651bc8d26dd04b4d0a6967
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/d6a3342305651bc8d26dd04b4d0a6967 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/e741eb34f789546b32088a8f483dd982 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/e741eb34f789546b32088a8f483dd982
new file mode 100644
index 0000000000000000000000000000000000000000..b494991df0d3366e1f23ca0c642af67f1efe8a55
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d/e741eb34f789546b32088a8f483dd982
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aa45d7b039a8ce82ad760c9f73e56ad4f6648ca429859a6aa3480a55ce6bcbed
+size 244987
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d25cabb2c6c34eefa99e6962314e5793f84a420e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..039cbcc0c79872f4dd009c10917342f321a743bb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590025518727721, "commit_timestamp_nsecs": 1754590025664621850, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/d/6471401841b53356546dc6f3a449acbf b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/d/6471401841b53356546dc6f3a449acbf
new file mode 100644
index 0000000000000000000000000000000000000000..a11426103ed852b2e6d3f5f2d6c697668423eeea
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/d/6471401841b53356546dc6f3a449acbf differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..833996c4776234ad8dc616933673535573ae51aa
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/144da8943944216ad4a02d911aea1db3 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/144da8943944216ad4a02d911aea1db3
new file mode 100644
index 0000000000000000000000000000000000000000..b8ddde0f76b101ff5af06e5cb9be8f3201290aa3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/144da8943944216ad4a02d911aea1db3
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:82e7f86b9c159ef0ec47cd563042e2f3aa0c9c7c6b0493d7b9f9ce1bb557607d
+size 125560
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/2bc62fc903dc7390cdd9b4d4f3b272d3 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/2bc62fc903dc7390cdd9b4d4f3b272d3
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/2bc62fc903dc7390cdd9b4d4f3b272d3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/4f79774154fdbaf3075b8d665eedfeab b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/4f79774154fdbaf3075b8d665eedfeab
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/4f79774154fdbaf3075b8d665eedfeab differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/9df47419f901ae607ca0de397ae7e38e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/9df47419f901ae607ca0de397ae7e38e
new file mode 100644
index 0000000000000000000000000000000000000000..b58698119cef793ac986cf51c1fa59a151783dcb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/9df47419f901ae607ca0de397ae7e38e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/b1c053491931cdf44b967228b207ff72 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/b1c053491931cdf44b967228b207ff72
new file mode 100644
index 0000000000000000000000000000000000000000..1b5ed9f01da716049f9f64712bbf55b7ced5ef84
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/b1c053491931cdf44b967228b207ff72
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:15e018fb629fdbdc24167d483aff703568da162aa8027da44818772b3f2ae79a
+size 614161
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/b387217d63bc00127b58803cec649145 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/b387217d63bc00127b58803cec649145
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/b387217d63bc00127b58803cec649145 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/ceebf0fa4e2e113da75ff6fb63534dae b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/ceebf0fa4e2e113da75ff6fb63534dae
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/ceebf0fa4e2e113da75ff6fb63534dae differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/ddb38711dad785cc0f1b4ec6d012b2fa b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/ddb38711dad785cc0f1b4ec6d012b2fa
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d/ddb38711dad785cc0f1b4ec6d012b2fa differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..ed05ae70b826377ac34ae9d9c8fb30ecba6c0735
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..2dce87f989312e4b952e22944b3b7c6dc716f842
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591252871645951, "commit_timestamp_nsecs": 1754591253050215561, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/d/329348c7963187c434f8d6881c85af10 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/d/329348c7963187c434f8d6881c85af10
new file mode 100644
index 0000000000000000000000000000000000000000..06b31993e08e04dbe6260d611bd5e9b28a6599f4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/d/329348c7963187c434f8d6881c85af10 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..ab029d6854adfa6f8481eaa3516ee512fc03253b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/01e02076a36046688a1f61c7ae68bf38 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/01e02076a36046688a1f61c7ae68bf38
new file mode 100644
index 0000000000000000000000000000000000000000..2910fbc6db2094404ed932c7d9bcdb23beb99906
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/01e02076a36046688a1f61c7ae68bf38 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/2ebdae910a254d46353f5a859254ec97 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/2ebdae910a254d46353f5a859254ec97
new file mode 100644
index 0000000000000000000000000000000000000000..f8f48bc73fcf84f50227190a5f8cdaf488c67b47
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/2ebdae910a254d46353f5a859254ec97 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/40d52fcf1d23075a6e711048dc995dde b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/40d52fcf1d23075a6e711048dc995dde
new file mode 100644
index 0000000000000000000000000000000000000000..4859214e6b68fa6e697de9748417fcf43be8081d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/40d52fcf1d23075a6e711048dc995dde differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/468ec96a3f3adb3f77806ada6e80b6d4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/468ec96a3f3adb3f77806ada6e80b6d4
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/468ec96a3f3adb3f77806ada6e80b6d4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/57c0edeb0a7b8e0d781e3864b428447d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/57c0edeb0a7b8e0d781e3864b428447d
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/57c0edeb0a7b8e0d781e3864b428447d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/688df1de52824ff4a9199f0e446aeb99 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/688df1de52824ff4a9199f0e446aeb99
new file mode 100644
index 0000000000000000000000000000000000000000..0698b3b04182020b1dd02f71ea6783caec7121a0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/688df1de52824ff4a9199f0e446aeb99
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:68031145b9389fa9e7f3ef52880355ef2c316d7a0264c0fff172706b54fc51da
+size 278992
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/7486eba87e1576b7c8856f618a2e8eb9 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/7486eba87e1576b7c8856f618a2e8eb9
new file mode 100644
index 0000000000000000000000000000000000000000..b338407744275137bdf539ffa5bda6dfd604b66a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/7486eba87e1576b7c8856f618a2e8eb9
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:acc31c17b0b7d872c80134deba94a976d1c3cc6a22cb6a728cab036bcf3c3852
+size 413719
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/ddebb479e1542809695f882ff90f5d7a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/ddebb479e1542809695f882ff90f5d7a
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/ddebb479e1542809695f882ff90f5d7a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/f53e0ce9b788a1db62ada77a32ee442e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/f53e0ce9b788a1db62ada77a32ee442e
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d/f53e0ce9b788a1db62ada77a32ee442e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1bab0e8e7eece3605070f550b6e4a4639775ba34
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..ba0cab75496bb0f655a0f05f6a77eb5790fb3c91
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591286272115424, "commit_timestamp_nsecs": 1754591286448939003, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/d/2da43886a71eb798d2a3829e4db0243b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/d/2da43886a71eb798d2a3829e4db0243b
new file mode 100644
index 0000000000000000000000000000000000000000..28c1c42d6243c6b90a8551aacde6c5ae47a1af7a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/d/2da43886a71eb798d2a3829e4db0243b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..2823b04aa7e27192fddd24c6e40d3228407202d3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/00e167d3dc1caa71c22c4290637e58c7 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/00e167d3dc1caa71c22c4290637e58c7
new file mode 100644
index 0000000000000000000000000000000000000000..835515eacba0ac1a3f2c59ccf63a599f503d68fc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/00e167d3dc1caa71c22c4290637e58c7
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c04db55dd7f0269b407f7f24b8b513ac4a7354c799ddf202c12039b663109dcd
+size 243911
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/099ce24edf828dbdf90a4479eda7d24e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/099ce24edf828dbdf90a4479eda7d24e
new file mode 100644
index 0000000000000000000000000000000000000000..a12a1344b472e30113c907b279096a6c49c31a00
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/099ce24edf828dbdf90a4479eda7d24e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/0d431efb4b3870c3dbdc5ea7aa739b44 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/0d431efb4b3870c3dbdc5ea7aa739b44
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/0d431efb4b3870c3dbdc5ea7aa739b44 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/0e59e90770db8de478237519e60721d4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/0e59e90770db8de478237519e60721d4
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/0e59e90770db8de478237519e60721d4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/19bb38bb820649c2be162b6717d8a161 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/19bb38bb820649c2be162b6717d8a161
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/19bb38bb820649c2be162b6717d8a161 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/5235f2bcb5b579dbeeb52f9633c74507 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/5235f2bcb5b579dbeeb52f9633c74507
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/5235f2bcb5b579dbeeb52f9633c74507 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/8b5b84e30f7543b0a235ac7f700888ba b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/8b5b84e30f7543b0a235ac7f700888ba
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/8b5b84e30f7543b0a235ac7f700888ba differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/902f1da199c36a20b80712c0edbca031 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/902f1da199c36a20b80712c0edbca031
new file mode 100644
index 0000000000000000000000000000000000000000..69324be8ce56ff844af440fe442296ceff5ba2c3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/902f1da199c36a20b80712c0edbca031
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:66dbd34a008a04d720b49b06f605714fc8ef1de42e4fe630b9a5923a833400e3
+size 492818
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/b758c44af12c5ca82c63f3eed1b906fd b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/b758c44af12c5ca82c63f3eed1b906fd
new file mode 100644
index 0000000000000000000000000000000000000000..07b546cbc8067ab26bbdfda6334f806866c697b0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/b758c44af12c5ca82c63f3eed1b906fd differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/dd9d4dc4cd710bf64a6f4376dd3597d4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/dd9d4dc4cd710bf64a6f4376dd3597d4
new file mode 100644
index 0000000000000000000000000000000000000000..2b03d141156751d24ecb313e49acee489d277c9c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d/dd9d4dc4cd710bf64a6f4376dd3597d4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..dd7dc15c66265601f49ba37ea0893ed5ab67a316
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..c34843f2898f3683cd1837618933c31bedfeb48a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591319783919568, "commit_timestamp_nsecs": 1754591319952103984, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/d/0f52eacee7a6091049431adbb601242d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/d/0f52eacee7a6091049431adbb601242d
new file mode 100644
index 0000000000000000000000000000000000000000..8d7b56e5cb77889178e9eabcea99211418051984
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/d/0f52eacee7a6091049431adbb601242d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5ab9e3f9369fdf4a54d5de3c52bb4485337efefd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/0199836574daa6a6a69e3d2ce8e4dae5 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/0199836574daa6a6a69e3d2ce8e4dae5
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/0199836574daa6a6a69e3d2ce8e4dae5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/042044206f685721a687d0781c1977ed b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/042044206f685721a687d0781c1977ed
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/042044206f685721a687d0781c1977ed differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/594e4196ffac6f1e40d7a8cd3bc0d324 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/594e4196ffac6f1e40d7a8cd3bc0d324
new file mode 100644
index 0000000000000000000000000000000000000000..e666b351a8cb93beb01fe1d658fda6b4d4169d1c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/594e4196ffac6f1e40d7a8cd3bc0d324 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/a108376e4c3e520a2faae195c6261a26 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/a108376e4c3e520a2faae195c6261a26
new file mode 100644
index 0000000000000000000000000000000000000000..40d47169b366d3900ba65575ad74d30ba4fc892c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/a108376e4c3e520a2faae195c6261a26
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7e96b04568ec433b265d3bdd9432d1e00f1ead86f1a559a3451133ab90994924
+size 447530
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/b973851850686ecd6ad808c5757854ae b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/b973851850686ecd6ad808c5757854ae
new file mode 100644
index 0000000000000000000000000000000000000000..659925b3fd96b98b71478cd61e926bc54876e6a1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/b973851850686ecd6ad808c5757854ae
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ffdcd908207bf1480b66382b3fa503100fc9cb84adfdee78e097e8124cce539b
+size 253541
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/e36215b9453731b2abd40e9d833603b0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/e36215b9453731b2abd40e9d833603b0
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/e36215b9453731b2abd40e9d833603b0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/ef99592e7dc9feac62d32e6bf4a88c16 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/ef99592e7dc9feac62d32e6bf4a88c16
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/ef99592e7dc9feac62d32e6bf4a88c16 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/f465574be1ede7201f844b3ea6fc0616 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/f465574be1ede7201f844b3ea6fc0616
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d/f465574be1ede7201f844b3ea6fc0616 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..cce471497b33ccd7c839835ac40decdfce3770f3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..41f7d9bf4602c09223d0ab0791cc8633842998a1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591353117954980, "commit_timestamp_nsecs": 1754591353258228563, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/d/d01601d60a8a237387f3e441f83bf193 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/d/d01601d60a8a237387f3e441f83bf193
new file mode 100644
index 0000000000000000000000000000000000000000..7b6b83f4ce5e6224d3fd50e46c2195369ff57160
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/d/d01601d60a8a237387f3e441f83bf193 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8e792d00cde9b3bfe8dee734e9b27b18f18a49ec
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/020cbf2833c57f1e9f6bddb700eb97c8 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/020cbf2833c57f1e9f6bddb700eb97c8
new file mode 100644
index 0000000000000000000000000000000000000000..8b1ed6ca4c6b636004e7740da78e6a0b61357e0e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/020cbf2833c57f1e9f6bddb700eb97c8
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a0c2dba55384c8a6e52b85a3702c10f8ec1e8386f59d6801340565c657473b62
+size 442731
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/4a65bc9e6d54122ba5e4f4b47b44bb3e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/4a65bc9e6d54122ba5e4f4b47b44bb3e
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/4a65bc9e6d54122ba5e4f4b47b44bb3e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/7bff68c37967883004421a6a44ad6d01 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/7bff68c37967883004421a6a44ad6d01
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/7bff68c37967883004421a6a44ad6d01 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/890ceb35623f196956262bce07b0f156 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/890ceb35623f196956262bce07b0f156
new file mode 100644
index 0000000000000000000000000000000000000000..83c752f5fc731be2bc01bfbef22742444d21a09d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/890ceb35623f196956262bce07b0f156 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/976572fa3a8915e1956e7e63cf7dc4b6 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/976572fa3a8915e1956e7e63cf7dc4b6
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/976572fa3a8915e1956e7e63cf7dc4b6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/aea5412af0526d1ee229d803cddbc27b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/aea5412af0526d1ee229d803cddbc27b
new file mode 100644
index 0000000000000000000000000000000000000000..2910fbc6db2094404ed932c7d9bcdb23beb99906
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/aea5412af0526d1ee229d803cddbc27b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/c078d66f1b9edd2ced594cb2bc434834 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/c078d66f1b9edd2ced594cb2bc434834
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/c078d66f1b9edd2ced594cb2bc434834 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/ec846de8788d90981aece9aba9c42abf b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/ec846de8788d90981aece9aba9c42abf
new file mode 100644
index 0000000000000000000000000000000000000000..135396415e2f58906f346e0376c590360e24dfb1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d/ec846de8788d90981aece9aba9c42abf
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:95763000792929834533a77af55675960f44693d608dbeda2d3ed67f3a8ea194
+size 296046
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4675ed203d0b3d9e3410f10645a7718f444c04c7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..21e15d54f1d0e366ae6da058f0b3784c0caf244f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591385926431188, "commit_timestamp_nsecs": 1754591386053548054, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/d/e42a6d6158bee76b6722df592b4af898 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/d/e42a6d6158bee76b6722df592b4af898
new file mode 100644
index 0000000000000000000000000000000000000000..d226229de9b75b86330b033d01f2144d6c7a5649
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/d/e42a6d6158bee76b6722df592b4af898 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..db3dcf0e1ffc20b53312658cf3ef7ee61a503976
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/2ffc8f976612230f012aa92dd5dd214b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/2ffc8f976612230f012aa92dd5dd214b
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/2ffc8f976612230f012aa92dd5dd214b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/36b3341a1087d26070ca7eb26bb637b4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/36b3341a1087d26070ca7eb26bb637b4
new file mode 100644
index 0000000000000000000000000000000000000000..2b9c9c3a17d45e4c4d82f9db6a9a11607c475ca0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/36b3341a1087d26070ca7eb26bb637b4
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:acdfefe9828c4b7f6c913240ce8a80b56988426d9a7a1e0531c77eb39ba85527
+size 150721
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/538348898686c502c18b3a2e24ec8109 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/538348898686c502c18b3a2e24ec8109
new file mode 100644
index 0000000000000000000000000000000000000000..c83ca879c1151d344e7bbb9b8c461615ae324113
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/538348898686c502c18b3a2e24ec8109
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b7620c5c1b635ad0db8efe4e03052d008c2917810ee3f7bc849ecf2c2f95f14c
+size 319361
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/6d5e2d655cc7aca05788e437e5296e74 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/6d5e2d655cc7aca05788e437e5296e74
new file mode 100644
index 0000000000000000000000000000000000000000..0d0c1bbefd64b88d1a54c8bd1ce18d12062b7ed2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/6d5e2d655cc7aca05788e437e5296e74
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6e02c4a61b50695576979030e30339ffb9c2f5960e4a976c1e5113937166f3c2
+size 270691
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/8c9c2ea00ec9c1df0138ef9d69fb7d49 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/8c9c2ea00ec9c1df0138ef9d69fb7d49
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/8c9c2ea00ec9c1df0138ef9d69fb7d49 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/9545c1c6eb91a8fd8c25ea92b15b308b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/9545c1c6eb91a8fd8c25ea92b15b308b
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/9545c1c6eb91a8fd8c25ea92b15b308b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/d683500cee80daf10e8c32e90f975512 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/d683500cee80daf10e8c32e90f975512
new file mode 100644
index 0000000000000000000000000000000000000000..f28d59e1f9187cfd65b688b4bceb57a81f0acdcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/d683500cee80daf10e8c32e90f975512 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/fb675e24662186e5da4f337364afcf33 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/fb675e24662186e5da4f337364afcf33
new file mode 100644
index 0000000000000000000000000000000000000000..88e5fd4a6f0bc61983e09feefcbe13228c7ababc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d/fb675e24662186e5da4f337364afcf33 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..883306400251288f33f01ab8443051ce1c9a8f2b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..ff3c4d880b8807346f05ad980b8971c2dcbcc9f9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591420025683550, "commit_timestamp_nsecs": 1754591420167194430, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/d/ad2d57a4edaf077901346b2fd5ef302e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/d/ad2d57a4edaf077901346b2fd5ef302e
new file mode 100644
index 0000000000000000000000000000000000000000..9aaa058cced01a6d4197894dad97f2ae305918da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/d/ad2d57a4edaf077901346b2fd5ef302e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0b9ecac127864998ccb840b5c65ad8aed13d90a3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/16bcc28610c142ec3a608c32d976afcd b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/16bcc28610c142ec3a608c32d976afcd
new file mode 100644
index 0000000000000000000000000000000000000000..7f7d14977d199352994ce996a83ddb7429e587db
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/16bcc28610c142ec3a608c32d976afcd
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c182a230c4d0a209e164f6002240a14e817a11f19cda8299b66a9aaa8beda4de
+size 374397
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/4813a9c426f6563e914a267c2b1e9225 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/4813a9c426f6563e914a267c2b1e9225
new file mode 100644
index 0000000000000000000000000000000000000000..a0a251b2af6e030e792a148cfc948cb39bad8a93
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/4813a9c426f6563e914a267c2b1e9225
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:21df311d7fd0ac440c24dd7ea7cbd20f3ef0f032b40e2ba551c77dcee359bf96
+size 364043
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/7729dfe290c758faa7688030db742f87 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/7729dfe290c758faa7688030db742f87
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/7729dfe290c758faa7688030db742f87 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/80acc76988a7ba42257a07eb476849f8 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/80acc76988a7ba42257a07eb476849f8
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/80acc76988a7ba42257a07eb476849f8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/8308c2b67a5574965d3531ed19415b9d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/8308c2b67a5574965d3531ed19415b9d
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/8308c2b67a5574965d3531ed19415b9d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/849b8248d86d8edc90f059b84f72230f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/849b8248d86d8edc90f059b84f72230f
new file mode 100644
index 0000000000000000000000000000000000000000..d425ce7a49b29fcb1d1f45db677c5e79db3bbe06
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/849b8248d86d8edc90f059b84f72230f differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/a3ad05075ee60da563256b5348f7cc71 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/a3ad05075ee60da563256b5348f7cc71
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d/a3ad05075ee60da563256b5348f7cc71 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b24a034bfef33f4a570aa65e3e389f54e3652803
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..05a48791844b2b789ea5471f901ef0cce2152699
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591453880393939, "commit_timestamp_nsecs": 1754591454040539317, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/d/c380243cfd21292e8408493708858fd3 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/d/c380243cfd21292e8408493708858fd3
new file mode 100644
index 0000000000000000000000000000000000000000..e3d2d8f1317f54e56e48d7efc1d51e1e912f649e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/d/c380243cfd21292e8408493708858fd3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8ffc7c8baa90e7edd3664c84cc6e5c87df5f0b7c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/102f8b14595173ad57180e4d2760d3a6 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/102f8b14595173ad57180e4d2760d3a6
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/102f8b14595173ad57180e4d2760d3a6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/23e284ade8be27338355f16726454f7d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/23e284ade8be27338355f16726454f7d
new file mode 100644
index 0000000000000000000000000000000000000000..ccdaa43b852da6855651f1ac636ae756c204a2c1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/23e284ade8be27338355f16726454f7d
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bfa1d9d60b17e53ade8b2763ebc4c8cda21f8e759ffcf09c613580989f6d8987
+size 295046
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/25c0b0bd80d6b2d75a4bd17c88a08241 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/25c0b0bd80d6b2d75a4bd17c88a08241
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/25c0b0bd80d6b2d75a4bd17c88a08241 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/2c019cd394164b08f0af7aec35ec84e7 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/2c019cd394164b08f0af7aec35ec84e7
new file mode 100644
index 0000000000000000000000000000000000000000..c3cc93f4ffb316c51876743dd13750ecc4f56ce1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/2c019cd394164b08f0af7aec35ec84e7
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:12645668b3f9fbb844aae3da914c42681be717aadafa7d8b119ef496bb1297ba
+size 373969
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/5e655d377ed84f665ad87a13c63bd054 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/5e655d377ed84f665ad87a13c63bd054
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/5e655d377ed84f665ad87a13c63bd054 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/6d457b6d767055c0fe5f248067a2dd4c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/6d457b6d767055c0fe5f248067a2dd4c
new file mode 100644
index 0000000000000000000000000000000000000000..1a33f84822fe5b3bce8d95b7e9314679d8738cd1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/6d457b6d767055c0fe5f248067a2dd4c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/d55e8f43f393f746279485897a64f647 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/d55e8f43f393f746279485897a64f647
new file mode 100644
index 0000000000000000000000000000000000000000..65529f6532287fad4b450ca493011b55e8bf7f92
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/d55e8f43f393f746279485897a64f647 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/e139667a09b8838a0964a5d90e59dc55 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/e139667a09b8838a0964a5d90e59dc55
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d/e139667a09b8838a0964a5d90e59dc55 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f0d122714db35099daccaa4065911b7e739bcda7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..e774eb292886898690c305da66a8b582eb54904b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591487421422880, "commit_timestamp_nsecs": 1754591487561207453, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/d/fdedd980a81ccfa8aba76a914ea500de b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/d/fdedd980a81ccfa8aba76a914ea500de
new file mode 100644
index 0000000000000000000000000000000000000000..0f1176c90e12de75581cac4bdd0e4ffc64a176dd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/d/fdedd980a81ccfa8aba76a914ea500de differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..061d19b82736536d5ae29dcb106e0f4e050e0fe9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/0f0285d3fb131cd823cd6dd9b10e7df2 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/0f0285d3fb131cd823cd6dd9b10e7df2
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/0f0285d3fb131cd823cd6dd9b10e7df2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/6df958cb57f18ada934f181cf410e9ea b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/6df958cb57f18ada934f181cf410e9ea
new file mode 100644
index 0000000000000000000000000000000000000000..f41530861a363ef4f07d7ed004e0f233b7552a1f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/6df958cb57f18ada934f181cf410e9ea
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c686161eaa58d37a78c71befa0f2bf27b71db4c8ed3b42c5f32f88b2a631a44c
+size 330121
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/6e8165d4cb67a1725b9fa4b6404696c4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/6e8165d4cb67a1725b9fa4b6404696c4
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/6e8165d4cb67a1725b9fa4b6404696c4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/7332142c2ba30963ca3dd6f166a1d1f7 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/7332142c2ba30963ca3dd6f166a1d1f7
new file mode 100644
index 0000000000000000000000000000000000000000..49d4d0763694283dab3ecf9ad6cc0f48a4f46fe2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/7332142c2ba30963ca3dd6f166a1d1f7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/76988cee6614265f18d208a5029ca7a8 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/76988cee6614265f18d208a5029ca7a8
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/76988cee6614265f18d208a5029ca7a8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/daeaa12ec741ac9dc97727f95ed952eb b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/daeaa12ec741ac9dc97727f95ed952eb
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/daeaa12ec741ac9dc97727f95ed952eb differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/fbe88ee3fbcb2f07a2a4ad7f399eab96 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/fbe88ee3fbcb2f07a2a4ad7f399eab96
new file mode 100644
index 0000000000000000000000000000000000000000..dfb580577645fa0c5a43a780fa543a1a04e55473
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d/fbe88ee3fbcb2f07a2a4ad7f399eab96
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9d8a2e82994b739163ca4239e47574dc8e19aae95967c2f61fe06e9404973bf6
+size 408428
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..fbd7caf9d50bcfaf69168b1dbb53cf7ef07598c7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..eb6e06097deba0b52d41d8e3f4286f8d1fcb2c9b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591534286837302, "commit_timestamp_nsecs": 1754591534451683573, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/d/50c5fd660513b3f242a18da41a286d0a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/d/50c5fd660513b3f242a18da41a286d0a
new file mode 100644
index 0000000000000000000000000000000000000000..83a2b365ed76f43d075d619db71f849f80f21ff7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/d/50c5fd660513b3f242a18da41a286d0a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c518d95185e3786dd2ba35aa504f8f45feae07f1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/008b0660bdadd1e71877fd66882b427b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/008b0660bdadd1e71877fd66882b427b
new file mode 100644
index 0000000000000000000000000000000000000000..03ffc2f6062c826ea681d38ffdecbdd057dd38c0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/008b0660bdadd1e71877fd66882b427b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8e956fe59ad58e754007706a90eef23f2e84dd571cacc14e8c2cf0612914ae8f
+size 442866
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/00f986c63d9554ad0a95fd58e575be23 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/00f986c63d9554ad0a95fd58e575be23
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/00f986c63d9554ad0a95fd58e575be23 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/0716ad50979bf94c9e6030e733b40122 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/0716ad50979bf94c9e6030e733b40122
new file mode 100644
index 0000000000000000000000000000000000000000..50bd7beea117ae2b9bd219af25f55bc69d25fa98
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/0716ad50979bf94c9e6030e733b40122
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d4193a70b61d64ed1d14a2e984498fabbdfde8ca28d5ce36a3ea8203ed5835a3
+size 296044
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/0995d1d73aad162ec9f2181d4233eaa5 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/0995d1d73aad162ec9f2181d4233eaa5
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/0995d1d73aad162ec9f2181d4233eaa5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/6dda468fb52bfabaa3e2a35011fe3b78 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/6dda468fb52bfabaa3e2a35011fe3b78
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/6dda468fb52bfabaa3e2a35011fe3b78 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/7f3af07715fc31752c70f392ef37ba82 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/7f3af07715fc31752c70f392ef37ba82
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/7f3af07715fc31752c70f392ef37ba82 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/e1530d49d5e2dc93952dfdb8f5027b62 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/e1530d49d5e2dc93952dfdb8f5027b62
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/e1530d49d5e2dc93952dfdb8f5027b62 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/fe02a66de745400d26bb557622cac852 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/fe02a66de745400d26bb557622cac852
new file mode 100644
index 0000000000000000000000000000000000000000..37e78e68b603ae26fe7e9a1d0e2189c579bcb36d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d/fe02a66de745400d26bb557622cac852 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b7b0610524dd3a2e7102158e744ba3cc75d0703c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..6033992424f48d31acfda505a40445df3bff385e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591569168523553, "commit_timestamp_nsecs": 1754591569344684974, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/d/0c86339b2c4dda12abb957d46fe709da b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/d/0c86339b2c4dda12abb957d46fe709da
new file mode 100644
index 0000000000000000000000000000000000000000..f4cf69bea90fa771309905f46bbb6bc8a6f435f9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/d/0c86339b2c4dda12abb957d46fe709da differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..294ab5eab2c31b41358f0e99db0b4248fcdf7bce
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/1ec584ecac80d1352468f87078ff3226 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/1ec584ecac80d1352468f87078ff3226
new file mode 100644
index 0000000000000000000000000000000000000000..4b2e970989937da7c17eec9751ab55ac339e8d10
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/1ec584ecac80d1352468f87078ff3226 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/397868b1e9eebd06a0d376c723a44e08 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/397868b1e9eebd06a0d376c723a44e08
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/397868b1e9eebd06a0d376c723a44e08 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/42db52df956b8890faad93a38b7f0fd1 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/42db52df956b8890faad93a38b7f0fd1
new file mode 100644
index 0000000000000000000000000000000000000000..4bbc2a2b3d308f5145a9590c280dfe5606ef845d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/42db52df956b8890faad93a38b7f0fd1
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c58b0f915ca0bc608b0bf9070d7b83f7c0a28da131beb192353c539e57fea621
+size 442925
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/4d06cd12b9f7a1bb99a165617fa569ab b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/4d06cd12b9f7a1bb99a165617fa569ab
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/4d06cd12b9f7a1bb99a165617fa569ab differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/6eb3111c52ea9eeb7119b140cbcf039b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/6eb3111c52ea9eeb7119b140cbcf039b
new file mode 100644
index 0000000000000000000000000000000000000000..cce1d940f8cdb7f44e8fcfe99497c5498d299422
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/6eb3111c52ea9eeb7119b140cbcf039b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7d0c3df8991bad1f6bd10ae7ff9aea16a2e292235fcb45516cfb87ded5c05c9e
+size 296122
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/9b41c1be9455b269633e96e41ec7fc03 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/9b41c1be9455b269633e96e41ec7fc03
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/9b41c1be9455b269633e96e41ec7fc03 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/b6063bb929dcfb348fa1129d760ef7ff b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/b6063bb929dcfb348fa1129d760ef7ff
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/b6063bb929dcfb348fa1129d760ef7ff differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/e3eea333116f93c589c8467e62a7a323 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/e3eea333116f93c589c8467e62a7a323
new file mode 100644
index 0000000000000000000000000000000000000000..dfbfe1a61305bab6f716f83c77874d9e3176999d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/e3eea333116f93c589c8467e62a7a323 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/f16d825660d7060a4f12a046ed427142 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/f16d825660d7060a4f12a046ed427142
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d/f16d825660d7060a4f12a046ed427142 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f9e7eecc689c786c9fb87753ab4fbf6684443fde
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..8cc53ec185ca270e40716909c5bc5b92bccef607
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590059151629179, "commit_timestamp_nsecs": 1754590059330468943, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/d/20ec4d0589cfed03d39cbdf9fc800251 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/d/20ec4d0589cfed03d39cbdf9fc800251
new file mode 100644
index 0000000000000000000000000000000000000000..e3bbdda12a5baed135f7ca83ad06a781f26e562d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/d/20ec4d0589cfed03d39cbdf9fc800251 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..2b439cdc819db59cfe765caeccd1f8b9614f14b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/2603ab82e0b715d9ec39a03fcec25471 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/2603ab82e0b715d9ec39a03fcec25471
new file mode 100644
index 0000000000000000000000000000000000000000..e599cd9e3f671c9b4051975f5ec81475ecb889c7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/2603ab82e0b715d9ec39a03fcec25471
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e62ef0b3f924902d215394603add845adf00551bb34d0f2651e8ec8abcf008c2
+size 368273
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/36c22401332f1b7720f954f2922c1cbd b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/36c22401332f1b7720f954f2922c1cbd
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/36c22401332f1b7720f954f2922c1cbd differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/41a3d0cccaa9fc10dd59c44d9a6512bd b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/41a3d0cccaa9fc10dd59c44d9a6512bd
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/41a3d0cccaa9fc10dd59c44d9a6512bd differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/5714b0224d7b59c9e74fb40b42210881 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/5714b0224d7b59c9e74fb40b42210881
new file mode 100644
index 0000000000000000000000000000000000000000..ceac62e17e1af04ad662cb61b8a11d49c0788af2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/5714b0224d7b59c9e74fb40b42210881 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/ac2e9b0d8d2cba4054c0cb9aa9daf75f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/ac2e9b0d8d2cba4054c0cb9aa9daf75f
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/ac2e9b0d8d2cba4054c0cb9aa9daf75f differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/bebbb081c824abed5deff4d8b0381e48 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/bebbb081c824abed5deff4d8b0381e48
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/bebbb081c824abed5deff4d8b0381e48 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/c38056009818c69d4d080ad0a9758d76 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/c38056009818c69d4d080ad0a9758d76
new file mode 100644
index 0000000000000000000000000000000000000000..8d8a5c1aff842cc3ec548bea9d354f768ba53f57
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d/c38056009818c69d4d080ad0a9758d76
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c25fa0537fe0eb36547b6196e1923d5c24bcf28111ff9af8a741d2fbeacf4681
+size 295682
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5d9d62e35d8ffb6cfdec0e41d26944b42529745c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..eef707593f9303f678cce242a66de1cbfd6c3938
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591602225204279, "commit_timestamp_nsecs": 1754591602359917779, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/d/6dc107a0530be591e79f8c2bff140481 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/d/6dc107a0530be591e79f8c2bff140481
new file mode 100644
index 0000000000000000000000000000000000000000..0340b9c82da9b53f575798e439e48281fa3dae5f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/d/6dc107a0530be591e79f8c2bff140481 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..6b370c1ce42b6d6de37a1a9b2454a91b070b5f23
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/0f56ea346ffd7f60e4feb0d268a41ea6 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/0f56ea346ffd7f60e4feb0d268a41ea6
new file mode 100644
index 0000000000000000000000000000000000000000..35392f42d8d35ced23cbb1c2b2ca10988425c162
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/0f56ea346ffd7f60e4feb0d268a41ea6
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5cd47a67fa08611cea242e230cfd3c1beac4eb7448254c3a852b377739254a05
+size 321545
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/32783758e3e42983875e52dfc9477fb8 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/32783758e3e42983875e52dfc9477fb8
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/32783758e3e42983875e52dfc9477fb8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/3910ca9f0439b6594f0e10c00404bad6 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/3910ca9f0439b6594f0e10c00404bad6
new file mode 100644
index 0000000000000000000000000000000000000000..6fe8ff633002da1c2a15bd8cf32fd4f6a7bbf2f4
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/3910ca9f0439b6594f0e10c00404bad6
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2c9a20d48d2e50d931d067ddd17e8c8396edcc4d829014ecf3e7914e488f23b7
+size 416280
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/3fb5de62be06e88b5a1e221d1e1dc461 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/3fb5de62be06e88b5a1e221d1e1dc461
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/3fb5de62be06e88b5a1e221d1e1dc461 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/6d430649c1b3b7b781ecfd28abd0165d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/6d430649c1b3b7b781ecfd28abd0165d
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/6d430649c1b3b7b781ecfd28abd0165d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/a33620f6f2b62686acd4848c83bd892e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/a33620f6f2b62686acd4848c83bd892e
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/a33620f6f2b62686acd4848c83bd892e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/d61650c26eecffd718224b24f72c72df b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/d61650c26eecffd718224b24f72c72df
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d/d61650c26eecffd718224b24f72c72df differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..888270f4d0b6cdf7def51023c42ac12bbc5f3f61
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..1897c2773e93f79616015463528c8c893e44aa50
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591634827096279, "commit_timestamp_nsecs": 1754591634948256041, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/d/d786d0305c5852126aaca8754d2e58b6 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/d/d786d0305c5852126aaca8754d2e58b6
new file mode 100644
index 0000000000000000000000000000000000000000..5b21ff2f526f87e83eeaa65d9126cda2e7b63918
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/d/d786d0305c5852126aaca8754d2e58b6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b039bde665abf7c30091372a64e5d545bde7e492
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/0cc908e1d657f9a0f570177aa1634527 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/0cc908e1d657f9a0f570177aa1634527
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/0cc908e1d657f9a0f570177aa1634527 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/20c81af6265589ea61a862a7f3aed0a4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/20c81af6265589ea61a862a7f3aed0a4
new file mode 100644
index 0000000000000000000000000000000000000000..414b48519feeaa8b7475ccb8e3cf86a220db31f9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/20c81af6265589ea61a862a7f3aed0a4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/3e61c82b3acc0037b26e9781fb87d7c3 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/3e61c82b3acc0037b26e9781fb87d7c3
new file mode 100644
index 0000000000000000000000000000000000000000..7ee8876d42182b36e5c62d671e10413664257e3a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/3e61c82b3acc0037b26e9781fb87d7c3
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b2282739e8bd2a6a84f7ae62d3a8e649a777bfe4888b930e6f028a4d5e8a325f
+size 474699
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/46fab3e374bb48acde8b4bd6fa4178aa b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/46fab3e374bb48acde8b4bd6fa4178aa
new file mode 100644
index 0000000000000000000000000000000000000000..5d069e5a37b1ab87a2af6690e7b3b91411b170d9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/46fab3e374bb48acde8b4bd6fa4178aa
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:116e53d5ec3a9c0b2d536fae04f4dcd7d7c2a5390dbe20b168fda40e338126ae
+size 263410
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/5be995a80bc7db5b98439ee8f2334317 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/5be995a80bc7db5b98439ee8f2334317
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/5be995a80bc7db5b98439ee8f2334317 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/cf449dd9534bb3dd5ae8faa30fb43716 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/cf449dd9534bb3dd5ae8faa30fb43716
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/cf449dd9534bb3dd5ae8faa30fb43716 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/d148cde688a0a0a66bc9a7241a15fad1 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/d148cde688a0a0a66bc9a7241a15fad1
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d/d148cde688a0a0a66bc9a7241a15fad1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1c33a58938e66d2c06a3bc1276f8bf967c224e0c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..b2ef15692f7b9e40ed7e8fba8ebe02fa53fd0eca
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591668092090982, "commit_timestamp_nsecs": 1754591668260799300, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/d/89b1ec800db4ab6d9a38a3e182a2c515 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/d/89b1ec800db4ab6d9a38a3e182a2c515
new file mode 100644
index 0000000000000000000000000000000000000000..5a654892cad1eb4dab651d30d253b880d00de104
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/d/89b1ec800db4ab6d9a38a3e182a2c515 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f2f98a94d9018dfc41eb93bf6a8b43f83bb2683f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/048faf7004dd2dcba09aefe2a3802c72 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/048faf7004dd2dcba09aefe2a3802c72
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/048faf7004dd2dcba09aefe2a3802c72 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/0c969b72c09276ee8a1dfed5120244d2 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/0c969b72c09276ee8a1dfed5120244d2
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/0c969b72c09276ee8a1dfed5120244d2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/1e13bb3f923c031327d72d5e5e813e66 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/1e13bb3f923c031327d72d5e5e813e66
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/1e13bb3f923c031327d72d5e5e813e66 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/496c7ee0407be2c0002c4ad6ef3711e7 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/496c7ee0407be2c0002c4ad6ef3711e7
new file mode 100644
index 0000000000000000000000000000000000000000..3ba6edda8cafd41bc826239f25464378c68872a9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/496c7ee0407be2c0002c4ad6ef3711e7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/b383277044ab03062fa1200c81fa1851 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/b383277044ab03062fa1200c81fa1851
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/b383277044ab03062fa1200c81fa1851 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/be912c84ee305dd7bcda5bdc0298c332 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/be912c84ee305dd7bcda5bdc0298c332
new file mode 100644
index 0000000000000000000000000000000000000000..46bb79d710dc40296886598954dd53dbc690ad1a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/be912c84ee305dd7bcda5bdc0298c332
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fea081ae135af250b41d5e51b51f3125120b29a82a64a66ed7a475025dba0377
+size 511118
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/c9f62f6d20e936e21b762e59a3513e67 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/c9f62f6d20e936e21b762e59a3513e67
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/c9f62f6d20e936e21b762e59a3513e67 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/ee0429b5bf304a2d8c0ac75786eac2e9 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/ee0429b5bf304a2d8c0ac75786eac2e9
new file mode 100644
index 0000000000000000000000000000000000000000..8156d432891a4c00dc240a497e973d0d3c9cf846
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d/ee0429b5bf304a2d8c0ac75786eac2e9
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3a0ab5cb1d6d7a805b3f5400181dd466e2a4548f6e3c2a6e7635debfdb628ab3
+size 227926
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..534a282a59bdb0230671a12166029209fbfa469e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..661ca4a76ba7f48f734194cb8786d247c9c2c7e4
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591700387207059, "commit_timestamp_nsecs": 1754591700554025140, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/d/df3b759e870a11aabf07355efa71d1c5 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/d/df3b759e870a11aabf07355efa71d1c5
new file mode 100644
index 0000000000000000000000000000000000000000..9c5c122e19bc2c7bfa01493665c6fafb24bac410
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/d/df3b759e870a11aabf07355efa71d1c5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1422cbb030db2463cf8b81e95453aaf07e7ccfdc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/0528e2cdc1e56b2c8b32d097cac2489b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/0528e2cdc1e56b2c8b32d097cac2489b
new file mode 100644
index 0000000000000000000000000000000000000000..c69fbbc49ce6edf5ef70a79289896df2361f42b6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/0528e2cdc1e56b2c8b32d097cac2489b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a0691716495b6442c657afbbe82dc3af75fc6f567dead3cebd0f39da68f2710a
+size 278881
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/1afed5203ef18cb17509f62372f9c357 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/1afed5203ef18cb17509f62372f9c357
new file mode 100644
index 0000000000000000000000000000000000000000..5d95f7dc83fb2be3a77e02cf0d10e84c7f649dd5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/1afed5203ef18cb17509f62372f9c357
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ae778bdd27f4b7b953b63bd64ad4e4cc6cc1e90fe2abd5364d18bbeb50c8ae98
+size 460085
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/4eaf3bf788c745b81b5c25d3a20c80d6 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/4eaf3bf788c745b81b5c25d3a20c80d6
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/4eaf3bf788c745b81b5c25d3a20c80d6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/58c82e2bd564b0a269abd24404a4a35e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/58c82e2bd564b0a269abd24404a4a35e
new file mode 100644
index 0000000000000000000000000000000000000000..3ce85791254063e72ea0e0efbd96de637b32a9c0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/58c82e2bd564b0a269abd24404a4a35e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/78136df00d66df5b88572e351aa35a6a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/78136df00d66df5b88572e351aa35a6a
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/78136df00d66df5b88572e351aa35a6a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/96e25513ef26a5581e81acf65625da74 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/96e25513ef26a5581e81acf65625da74
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/96e25513ef26a5581e81acf65625da74 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/c9f500af0f276e443715189b474be27d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/c9f500af0f276e443715189b474be27d
new file mode 100644
index 0000000000000000000000000000000000000000..2a31377ca030aced51ef70fb458c6b5c47404cd3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/c9f500af0f276e443715189b474be27d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/dc34034311342c23033adacc5534d3d3 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/dc34034311342c23033adacc5534d3d3
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d/dc34034311342c23033adacc5534d3d3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..bdcd80c26e0e894f9bfbd1cc1abe24c386a1768a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..9e9d9a36b4132cf2212763f32e2f383d6b460009
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591733016360656, "commit_timestamp_nsecs": 1754591733160551612, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/d/5d2b91a2c396f7569876118e208783be b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/d/5d2b91a2c396f7569876118e208783be
new file mode 100644
index 0000000000000000000000000000000000000000..8a75c357b7a5072e350773244effdf107ef5a855
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/d/5d2b91a2c396f7569876118e208783be differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4c43ceb2b9d17ba4709b19564d025c74d153f7ae
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/31eb878b7f5816b3aec00c367187d4d4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/31eb878b7f5816b3aec00c367187d4d4
new file mode 100644
index 0000000000000000000000000000000000000000..b1417e00fcecb134871c8dac80fbddf64dd08f71
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/31eb878b7f5816b3aec00c367187d4d4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/6c8f9ddec754bff4ce3e86d9bd02fc48 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/6c8f9ddec754bff4ce3e86d9bd02fc48
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/6c8f9ddec754bff4ce3e86d9bd02fc48 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/9142e78d4db09c19bd6ade2b8a3af09d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/9142e78d4db09c19bd6ade2b8a3af09d
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/9142e78d4db09c19bd6ade2b8a3af09d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/a43ef777e6802dcd9fa4e09879c4ff47 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/a43ef777e6802dcd9fa4e09879c4ff47
new file mode 100644
index 0000000000000000000000000000000000000000..50b3dc50da80e073355897956aa630e58c5933a0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/a43ef777e6802dcd9fa4e09879c4ff47 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/b06796c6af12868b32ee335fa5b44f1e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/b06796c6af12868b32ee335fa5b44f1e
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/b06796c6af12868b32ee335fa5b44f1e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/b5c15b0d0bd49bc6fbe5f461d7e433ff b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/b5c15b0d0bd49bc6fbe5f461d7e433ff
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/b5c15b0d0bd49bc6fbe5f461d7e433ff differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/c6f352e4d61860259c1f147f4793e94d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/c6f352e4d61860259c1f147f4793e94d
new file mode 100644
index 0000000000000000000000000000000000000000..8389b3d326910cfdc58d5c736d4d17defc2a95f8
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d/c6f352e4d61860259c1f147f4793e94d
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7f29be4ba668c12a3df0ace7dfbdda4ebf1603305d7a55344605b1f9aa24509a
+size 735100
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..ae042634a736902033c45bcfa1426f25a5c77019
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..afb86606c4b664866041e55c396426738f3ac4fb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591765830024930, "commit_timestamp_nsecs": 1754591765960020156, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/d/2828fcecd9ecd8eb6ad2f5d062a1c0ea b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/d/2828fcecd9ecd8eb6ad2f5d062a1c0ea
new file mode 100644
index 0000000000000000000000000000000000000000..0a43b5436866fe558a7bae23351edb77c186cf2a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/d/2828fcecd9ecd8eb6ad2f5d062a1c0ea differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..88a9420f1ee33889efce830c05b55a41d2920464
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/56fc71b0979bb5f8377c2665102f6de9 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/56fc71b0979bb5f8377c2665102f6de9
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/56fc71b0979bb5f8377c2665102f6de9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/589ee5715d07268689db750bd0ad08f0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/589ee5715d07268689db750bd0ad08f0
new file mode 100644
index 0000000000000000000000000000000000000000..9f3f10841cd1b4692ac2fbac2a7a40dafb06ca3e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/589ee5715d07268689db750bd0ad08f0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/74171d0253936dca17d49ebad2249880 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/74171d0253936dca17d49ebad2249880
new file mode 100644
index 0000000000000000000000000000000000000000..94246587312390a0d56b26c4f5bf07e21db64342
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/74171d0253936dca17d49ebad2249880
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:178c700a7779d53872ced787f153d59fc5ba806024ceae8010337156b6bf0dbe
+size 736173
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/b7de125e86f0547a67d1fdad5cd0b5f2 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/b7de125e86f0547a67d1fdad5cd0b5f2
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/b7de125e86f0547a67d1fdad5cd0b5f2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/ba7ab12bed763bcdb65f33ff630d177c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/ba7ab12bed763bcdb65f33ff630d177c
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/ba7ab12bed763bcdb65f33ff630d177c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/be4f50428935413fc124ae64b3171b60 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/be4f50428935413fc124ae64b3171b60
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d/be4f50428935413fc124ae64b3171b60 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4492d364636947b1bd6328bc925b67ede4ff6f1a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..57da0929126c9ce66779004a46421a8863ac43fb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591798183899899, "commit_timestamp_nsecs": 1754591798351073801, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/d/4cd2956651bef4489eade7402df97740 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/d/4cd2956651bef4489eade7402df97740
new file mode 100644
index 0000000000000000000000000000000000000000..00f8d0c9df028ef2c51b06ee600c6c1fc23c4f3c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/d/4cd2956651bef4489eade7402df97740 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d4edda95a9b6cfcbd7c9984a4d6c8ce8e8a30f00
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/2d572622f70ddce5b6b430358e5dbbf0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/2d572622f70ddce5b6b430358e5dbbf0
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/2d572622f70ddce5b6b430358e5dbbf0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/440380150e2b69109bc8d52e4b79d4cc b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/440380150e2b69109bc8d52e4b79d4cc
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/440380150e2b69109bc8d52e4b79d4cc differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/6d00d229e337b5493c945aefff870e55 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/6d00d229e337b5493c945aefff870e55
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/6d00d229e337b5493c945aefff870e55 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/9258f497919ace9622aec8a3d26c40a7 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/9258f497919ace9622aec8a3d26c40a7
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/9258f497919ace9622aec8a3d26c40a7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/96e041eedb2463cc4a1806b2ab07f747 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/96e041eedb2463cc4a1806b2ab07f747
new file mode 100644
index 0000000000000000000000000000000000000000..c862c6d7f212b088aac4eaa7cca3c1bda2fcf673
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/96e041eedb2463cc4a1806b2ab07f747
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d7d5435f3f2a97692fc65af068dac57a181ad0529f6f2876b742aada30476c8b
+size 511056
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/a019d138dfafef19035cacf38650433c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/a019d138dfafef19035cacf38650433c
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/a019d138dfafef19035cacf38650433c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/bb4ff882e58e173106dd378aa9154e39 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/bb4ff882e58e173106dd378aa9154e39
new file mode 100644
index 0000000000000000000000000000000000000000..3cfd1ca3e514bde573f8364c8a0cfd39a962661c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/bb4ff882e58e173106dd378aa9154e39
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d1f1e5c6ffae1be8d9a400d6482dfd9b552c46bd02750c2d66fae92db1f3a856
+size 227902
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/c6ec6ef42ef8dd2b8cefaf7ecfa3f454 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/c6ec6ef42ef8dd2b8cefaf7ecfa3f454
new file mode 100644
index 0000000000000000000000000000000000000000..86d4183325656adf2b55261bb808d8b1261496f3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d/c6ec6ef42ef8dd2b8cefaf7ecfa3f454 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..980dd7d5c79ceb317570eed954eda9e01fb4503e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..9dd92fdf0b98f5115db6f32ad8eba16e09742d0c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591831391081271, "commit_timestamp_nsecs": 1754591831560529476, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/d/d2ebd06e79189acf6bae2df091f290a1 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/d/d2ebd06e79189acf6bae2df091f290a1
new file mode 100644
index 0000000000000000000000000000000000000000..717fbfa3212b192b01ee15a769650c118bdd1fe9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/d/d2ebd06e79189acf6bae2df091f290a1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..fff4b94296507898c39bf3a4200a6eebf990d2e3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/018626e588efb60dded9660bb929cb67 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/018626e588efb60dded9660bb929cb67
new file mode 100644
index 0000000000000000000000000000000000000000..afaf3a1f3b41a4a31b1b7680e0769e4091b56209
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/018626e588efb60dded9660bb929cb67
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f766eb46380f9c4ed1bae44ff6d137f8672435b68ee232ca67f84e3fb63602b8
+size 304724
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/2845865951a95f2247f5567bf88c26ee b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/2845865951a95f2247f5567bf88c26ee
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/2845865951a95f2247f5567bf88c26ee differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/2dd63e8479fd962a41bc0173cbbd297f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/2dd63e8479fd962a41bc0173cbbd297f
new file mode 100644
index 0000000000000000000000000000000000000000..5c72f63fe66a8caac127c9dc8645615ab70ac75d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/2dd63e8479fd962a41bc0173cbbd297f differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/4588bc84bff216ee868c217d5311bd08 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/4588bc84bff216ee868c217d5311bd08
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/4588bc84bff216ee868c217d5311bd08 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/a290789f218be64d15f6d8829f7d632d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/a290789f218be64d15f6d8829f7d632d
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/a290789f218be64d15f6d8829f7d632d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/b6bbcb6e0bfc38f34647bb536011422f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/b6bbcb6e0bfc38f34647bb536011422f
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/b6bbcb6e0bfc38f34647bb536011422f differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/b9ce01469555dc3e0e559af11c8ed9a3 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/b9ce01469555dc3e0e559af11c8ed9a3
new file mode 100644
index 0000000000000000000000000000000000000000..9a36781c33522bc27d29c938dca557d20dc2eef5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/b9ce01469555dc3e0e559af11c8ed9a3
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0faab164487332cb5b3f3f61f0935e9e28c7e85241ea5f99ce161c37bfcf8903
+size 400366
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/e992f7dc8b906b4e8acb2e21e6c74ed4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/e992f7dc8b906b4e8acb2e21e6c74ed4
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d/e992f7dc8b906b4e8acb2e21e6c74ed4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b29f7ce29c58d5ec4208b494009e29a1eee7212d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..d0b2b2903d3275f11303d9b9629c4e8d6cbe0bef
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591863731780253, "commit_timestamp_nsecs": 1754591863874322717, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/d/3d8e91133cb8acd8446a04ef407fd849 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/d/3d8e91133cb8acd8446a04ef407fd849
new file mode 100644
index 0000000000000000000000000000000000000000..d05c781295867c02d481f32d2b903ca3e916f8fe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/d/3d8e91133cb8acd8446a04ef407fd849 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..99bd81f8013192896a39ae80aa24356bd0e0da34
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/3c8143722138ce56168aeab2c67d2384 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/3c8143722138ce56168aeab2c67d2384
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/3c8143722138ce56168aeab2c67d2384 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/5726137db3bb1c04ef4f998d0d24cedd b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/5726137db3bb1c04ef4f998d0d24cedd
new file mode 100644
index 0000000000000000000000000000000000000000..5b60d16645e6b427ef63aee4176c7a5dbfade633
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/5726137db3bb1c04ef4f998d0d24cedd
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:eb52f3bf4bd0abb0a776a5e3550749806368f031a3ee7e58dbe3d9731f2c4ee9
+size 408845
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/8652d5e6b532d41f97e651b0b2662ea0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/8652d5e6b532d41f97e651b0b2662ea0
new file mode 100644
index 0000000000000000000000000000000000000000..341f43c10df2f09145e179a495b1e67a32f9337a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/8652d5e6b532d41f97e651b0b2662ea0
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:89f801ab310db1b370352fc55a622f3342c421ab38f20ba8047a865eb3334037
+size 296189
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/9b947100903b27d822407ce17ab5887d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/9b947100903b27d822407ce17ab5887d
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/9b947100903b27d822407ce17ab5887d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/ab479a6de95367bd41582274ffa6c377 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/ab479a6de95367bd41582274ffa6c377
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/ab479a6de95367bd41582274ffa6c377 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/b3607f7607ed43bdda4422cbbccd5140 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/b3607f7607ed43bdda4422cbbccd5140
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/b3607f7607ed43bdda4422cbbccd5140 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/defda05b4a65bdec031b7c6e3f037211 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/defda05b4a65bdec031b7c6e3f037211
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/defda05b4a65bdec031b7c6e3f037211 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/f33fe3954d7546176f3c55bf68be322c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/f33fe3954d7546176f3c55bf68be322c
new file mode 100644
index 0000000000000000000000000000000000000000..f501fab1c5584d7bfcda93ee4e5f9e3a518f76ea
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d/f33fe3954d7546176f3c55bf68be322c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..ea634898e5c940776bc31686724aac9fd7c6eb10
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..6bbf2da955076ba3e7b8110b2469197548969678
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591897041786986, "commit_timestamp_nsecs": 1754591897230957832, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/d/46a9ede4a8f793070702fabf306c1f72 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/d/46a9ede4a8f793070702fabf306c1f72
new file mode 100644
index 0000000000000000000000000000000000000000..cf4758b915b15e2449a282a4d296465cbee241f6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/d/46a9ede4a8f793070702fabf306c1f72 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a119a9c3b5e12c557669044fbb31069e98b3bcd4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/09541e4085f6ad603556dc15dc337b6d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/09541e4085f6ad603556dc15dc337b6d
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/09541e4085f6ad603556dc15dc337b6d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/22eefd72c8be679b224acd5cceb4eb0a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/22eefd72c8be679b224acd5cceb4eb0a
new file mode 100644
index 0000000000000000000000000000000000000000..bd90b1062c233885c713f8c2e8c37268157e4d5e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/22eefd72c8be679b224acd5cceb4eb0a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:638d1cca3e58b5b4391b2efbb1e01f1ff7e95cb213050ac7d878ee1823a65088
+size 531861
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/4a89c4ac0f128bb509224007da8c8f4e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/4a89c4ac0f128bb509224007da8c8f4e
new file mode 100644
index 0000000000000000000000000000000000000000..086d4f37090637deec123cf43de97dc268e16c79
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/4a89c4ac0f128bb509224007da8c8f4e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/9436a7d098136283940d90d4f09a7dc8 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/9436a7d098136283940d90d4f09a7dc8
new file mode 100644
index 0000000000000000000000000000000000000000..c56864a459d6b82b7cccda0709b51e2af145e9eb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/9436a7d098136283940d90d4f09a7dc8
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b7e94f9a2758f8e114002406d1210de62f1804d26563e46a02fc8e7274a24ea3
+size 167424
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/bbf1b687d131118cf5d06eb7e96bf836 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/bbf1b687d131118cf5d06eb7e96bf836
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/bbf1b687d131118cf5d06eb7e96bf836 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/bc807e5483cec67aeb08da50f8ab777a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/bc807e5483cec67aeb08da50f8ab777a
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/bc807e5483cec67aeb08da50f8ab777a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/df1e35957fc5fb338afec268c090ec3a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/df1e35957fc5fb338afec268c090ec3a
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/df1e35957fc5fb338afec268c090ec3a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/e62c815f2fb4fb695f227c30b65146d4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/e62c815f2fb4fb695f227c30b65146d4
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d/e62c815f2fb4fb695f227c30b65146d4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..2665e76251a99891cd2d79a1d081c06123f0bf7c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..dbe41bdd3b8d6ea2b227002e854e1a04842fcddb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590092229880013, "commit_timestamp_nsecs": 1754590092420418639, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/d/27aefadc76bd50490f21169bacb36e5c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/d/27aefadc76bd50490f21169bacb36e5c
new file mode 100644
index 0000000000000000000000000000000000000000..93f71b97c1a0913916396b3a6956cf62504377df
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/d/27aefadc76bd50490f21169bacb36e5c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1621c030568e80f4d1d3103ae27ace9d6749a1ab
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/13c7b2b9f439b0a4aa2380d2479c93a8 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/13c7b2b9f439b0a4aa2380d2479c93a8
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/13c7b2b9f439b0a4aa2380d2479c93a8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/1f09fe1286792237f5a28d9d250d1e58 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/1f09fe1286792237f5a28d9d250d1e58
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/1f09fe1286792237f5a28d9d250d1e58 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/3355a9af078b680f90475704203f89a5 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/3355a9af078b680f90475704203f89a5
new file mode 100644
index 0000000000000000000000000000000000000000..0a7b860ff679c295fa7b4c3d00c6b5011d829434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/3355a9af078b680f90475704203f89a5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/70dcbd79508fbef122f52c8ffc5c37d9 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/70dcbd79508fbef122f52c8ffc5c37d9
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/70dcbd79508fbef122f52c8ffc5c37d9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/ba66f1abb492f032809466238a7fa8aa b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/ba66f1abb492f032809466238a7fa8aa
new file mode 100644
index 0000000000000000000000000000000000000000..e601e564709b761bf8cb447c7633c5f77b442409
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/ba66f1abb492f032809466238a7fa8aa
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9046e86571181dd9ddfc47056d73f72fe46a083c37cb54574710b994c1277f30
+size 406218
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/bb7b08fa34b50ba2a0c362beddc4b938 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/bb7b08fa34b50ba2a0c362beddc4b938
new file mode 100644
index 0000000000000000000000000000000000000000..65529f6532287fad4b450ca493011b55e8bf7f92
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/bb7b08fa34b50ba2a0c362beddc4b938 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/e3d6457ed82238e7ff61153abb23df4b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/e3d6457ed82238e7ff61153abb23df4b
new file mode 100644
index 0000000000000000000000000000000000000000..882356c28251f86e05d3f813a7be07b718ebd0e4
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d/e3d6457ed82238e7ff61153abb23df4b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:170d9a234c7774f0a85cdfa8a4a1e901a63fa0c6fe27ef85e0747ff9e9416293
+size 257965
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..78be2a521d1a9ab765878bf8c5f1e0629e33d8b3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..3c5589b81f2c54962c63a3165fe506ab320c8aed
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591930384065317, "commit_timestamp_nsecs": 1754591930560499574, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/d/616e882952fae3bcf1851a73d702b17d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/d/616e882952fae3bcf1851a73d702b17d
new file mode 100644
index 0000000000000000000000000000000000000000..1baa55d96896aae526862708494b8b619aa3735b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/d/616e882952fae3bcf1851a73d702b17d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..22fc90185733aeb75a22a3efc40449b058c5c642
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/2a0d0c26f0ed3de7f8c32539074c2ce2 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/2a0d0c26f0ed3de7f8c32539074c2ce2
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/2a0d0c26f0ed3de7f8c32539074c2ce2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/3e066996b005ccc0e0408765625ef1fe b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/3e066996b005ccc0e0408765625ef1fe
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/3e066996b005ccc0e0408765625ef1fe differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/5c323112406ca1c7c3b55545de4b3813 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/5c323112406ca1c7c3b55545de4b3813
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/5c323112406ca1c7c3b55545de4b3813 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/5f1287abc3d6ed2dab2a6bd8f9fa98c5 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/5f1287abc3d6ed2dab2a6bd8f9fa98c5
new file mode 100644
index 0000000000000000000000000000000000000000..87a45088dec5608a587499ebe6ca9e05de3d9c39
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/5f1287abc3d6ed2dab2a6bd8f9fa98c5
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8f2758f266dd82e460be77b4877c767291ff8d77b1dca95e878bb7cd41607542
+size 441827
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/60debaa37e9b802b26b37ec9ea116ef9 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/60debaa37e9b802b26b37ec9ea116ef9
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/60debaa37e9b802b26b37ec9ea116ef9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/61e8e3dea715b5639df6eea6c441a8a4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/61e8e3dea715b5639df6eea6c441a8a4
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/61e8e3dea715b5639df6eea6c441a8a4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/79a99cdb494d1c9cdc798a01581c2b5e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/79a99cdb494d1c9cdc798a01581c2b5e
new file mode 100644
index 0000000000000000000000000000000000000000..ebb4b6f5053e9fd862ff301f97a268b98c681624
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d/79a99cdb494d1c9cdc798a01581c2b5e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:85e3f2674e42f86d9adfa23f0479f77d48f4d25a5627a21cf8e10a853aabd827
+size 295057
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..696ea4df8f574c5f1a75ba6b46d1a1a58f259025
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..7b9125919fa730542e94a0bc22f8b4398f0b442f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591964942136570, "commit_timestamp_nsecs": 1754591965127437246, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/d/a874d2f8d7ac716d6a60bebc830703e1 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/d/a874d2f8d7ac716d6a60bebc830703e1
new file mode 100644
index 0000000000000000000000000000000000000000..3b9d74ab04514f53c53d205e8a13a02d8f83c039
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/d/a874d2f8d7ac716d6a60bebc830703e1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..11e5add56007724a76a5b86f95fcb1faeaed124e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/0165dac2c4f8183bfd4d65f119e05c28 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/0165dac2c4f8183bfd4d65f119e05c28
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/0165dac2c4f8183bfd4d65f119e05c28 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/0d98ce0fa38a8ee423e3ecfd09579503 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/0d98ce0fa38a8ee423e3ecfd09579503
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/0d98ce0fa38a8ee423e3ecfd09579503 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/107b867a8dc3bea73c3d0ec857ef75ec b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/107b867a8dc3bea73c3d0ec857ef75ec
new file mode 100644
index 0000000000000000000000000000000000000000..88f9d59e55a6b7a4495d39333630642558a2c5cc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/107b867a8dc3bea73c3d0ec857ef75ec differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/19f342e3cdbbc714c6a19a48c530b42d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/19f342e3cdbbc714c6a19a48c530b42d
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/19f342e3cdbbc714c6a19a48c530b42d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/2bbe49705ae161de7b446becdaf41977 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/2bbe49705ae161de7b446becdaf41977
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/2bbe49705ae161de7b446becdaf41977 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/38b0b6112d6b4332649a29811cbb352a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/38b0b6112d6b4332649a29811cbb352a
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/38b0b6112d6b4332649a29811cbb352a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/7f2df274782b6861bee8175167efdd76 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/7f2df274782b6861bee8175167efdd76
new file mode 100644
index 0000000000000000000000000000000000000000..f77be92f5f2cdf58a26b108af9dad7fe1c8a8cc9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/7f2df274782b6861bee8175167efdd76
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:59f40cf1f37c94d9d832c674d36763a2559c4ed12c5912664d2df6eff2a59aaf
+size 412867
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/96ed3b0f0f3f680bd9ea22a36121ad37 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/96ed3b0f0f3f680bd9ea22a36121ad37
new file mode 100644
index 0000000000000000000000000000000000000000..32019504bd55a58a42ce723711101332b9fd6112
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d/96ed3b0f0f3f680bd9ea22a36121ad37
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3c65b13324d84ae21c6cc5221ac55ededc19325205803fc47309e3e6b22eebe5
+size 286547
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1b7e0b69fb80f5efc74bba9241cd58fe4b9c2084
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..95d0fc00cb6ba0499104a897d3a27eb49da222bf
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754591997738154187, "commit_timestamp_nsecs": 1754591997931187280, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/d/6a1f67d2dcccd0950483dcb69a1ec1f7 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/d/6a1f67d2dcccd0950483dcb69a1ec1f7
new file mode 100644
index 0000000000000000000000000000000000000000..abe1ba726e7dedf74199bc1acf8a17ab0cd9f9e5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/d/6a1f67d2dcccd0950483dcb69a1ec1f7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a60032faac08975cf8580871a044032d38c76fa9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/00678f8b3e9d27828cdb69562a5585d7 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/00678f8b3e9d27828cdb69562a5585d7
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/00678f8b3e9d27828cdb69562a5585d7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/568c54569eebbe0cfdb0086f84c3de92 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/568c54569eebbe0cfdb0086f84c3de92
new file mode 100644
index 0000000000000000000000000000000000000000..415c6e9c285627dc1a90ee81b0f23d833d53e4d9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/568c54569eebbe0cfdb0086f84c3de92 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/d299cb6111f79b80df20c55061e508c4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/d299cb6111f79b80df20c55061e508c4
new file mode 100644
index 0000000000000000000000000000000000000000..78b48f4282754ec831c9d58cb1006e869ea9475e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/d299cb6111f79b80df20c55061e508c4
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:069fce08d5617b7bbdb8ae2f8b0cd75e0edc0d11b7ce74d7135873111fb9f3d2
+size 736356
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/f29d3f012c23b86c4443668246e17fe5 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/f29d3f012c23b86c4443668246e17fe5
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d/f29d3f012c23b86c4443668246e17fe5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..46af8ca7a2fad5257c46993aca877b5c439a93d1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..2e1f24c36a6b6b0989d1844c2d049e8db79e0a06
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592031165192652, "commit_timestamp_nsecs": 1754592031338826374, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/d/4a010b666d92bba7d5737c925d4e7c8a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/d/4a010b666d92bba7d5737c925d4e7c8a
new file mode 100644
index 0000000000000000000000000000000000000000..5484f73775af98e54e1427cce9f8c7ad40ad656b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/d/4a010b666d92bba7d5737c925d4e7c8a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..fe99481b247ca75a4d48bf34a18096ba5cf3655e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/2d05bf9cd016a8b94f288be1cbaac8a1 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/2d05bf9cd016a8b94f288be1cbaac8a1
new file mode 100644
index 0000000000000000000000000000000000000000..3fbf140f6ba0c81eaed53768324332f4c8440de9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/2d05bf9cd016a8b94f288be1cbaac8a1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/32b980de126bb7648c7ad53e756e694c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/32b980de126bb7648c7ad53e756e694c
new file mode 100644
index 0000000000000000000000000000000000000000..7bd5b0b11ef499ddd54628c82a6086603e45029f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/32b980de126bb7648c7ad53e756e694c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4db6d2c7004523eb3410a3e1f58fac63c3c4d9e05b6bb16c6d611c980826d0d6
+size 494130
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/415756efc1fef5a436ffdfda6e94d70c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/415756efc1fef5a436ffdfda6e94d70c
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/415756efc1fef5a436ffdfda6e94d70c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/5da6b735d8f1cc0bf85e6ae5839fa78e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/5da6b735d8f1cc0bf85e6ae5839fa78e
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/5da6b735d8f1cc0bf85e6ae5839fa78e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/c86aafa226553333a95ca8ffabc6788c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/c86aafa226553333a95ca8ffabc6788c
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/c86aafa226553333a95ca8ffabc6788c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/c8bf1079bdfaf70038d14c0161b623e2 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/c8bf1079bdfaf70038d14c0161b623e2
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/c8bf1079bdfaf70038d14c0161b623e2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/d46befa50b74038fb404c5ce826b124b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/d46befa50b74038fb404c5ce826b124b
new file mode 100644
index 0000000000000000000000000000000000000000..80372b22dc076e185a38366ee622e04c425377c5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/d46befa50b74038fb404c5ce826b124b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d5271ad0ebbe4e4f5ee13fd5ba8467e19d1af1db245993d8b35a0f225b5f3db4
+size 245041
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/dca29602d7b0c593023614ef53a6c948 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/dca29602d7b0c593023614ef53a6c948
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/dca29602d7b0c593023614ef53a6c948 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/fce545bc6e4a26acfa348cef04f861ec b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/fce545bc6e4a26acfa348cef04f861ec
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d/fce545bc6e4a26acfa348cef04f861ec differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..7108d763af6690f182e9747537875af5289ecfe8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..779d3033478ce0bc4e8f6b6ed9028d92c7cf3e04
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592079355060025, "commit_timestamp_nsecs": 1754592079523649832, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/d/700501d782309406718180544c6453fd b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/d/700501d782309406718180544c6453fd
new file mode 100644
index 0000000000000000000000000000000000000000..ddf4be255e425e6c4e15d9467bd6924e1609bb0b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/d/700501d782309406718180544c6453fd differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3c1e8edcb5287984a2cf8c282f95ee4f86460f62
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/013a7c44629c9baf043f69e7fe1fa491 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/013a7c44629c9baf043f69e7fe1fa491
new file mode 100644
index 0000000000000000000000000000000000000000..ec745d97353f97209ef50f0eb738585830219f53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/013a7c44629c9baf043f69e7fe1fa491 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/07aceeef5ef401013467ac56dd5557a1 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/07aceeef5ef401013467ac56dd5557a1
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/07aceeef5ef401013467ac56dd5557a1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/0aa14bc830365186e0a1df4a118645d0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/0aa14bc830365186e0a1df4a118645d0
new file mode 100644
index 0000000000000000000000000000000000000000..88e877e098e020733ab7cf7792ce1a16d50fb0de
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/0aa14bc830365186e0a1df4a118645d0
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5b360bd5abb69ff66611a8d959bd0ce94dcdf29d6165595ec53697d6b4ff7ce6
+size 460277
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/2710311fdddeae14787451ab54021f65 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/2710311fdddeae14787451ab54021f65
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/2710311fdddeae14787451ab54021f65 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/3d579eed0ed6643a4e5082039cd23521 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/3d579eed0ed6643a4e5082039cd23521
new file mode 100644
index 0000000000000000000000000000000000000000..3a672f9e653bda73af5e6eb0aed0d6642fabda01
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/3d579eed0ed6643a4e5082039cd23521
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5a1feb73a031ba99aa94f9a9c3d75336cbc797c8d8b12cd34e0fd88f7a4aae68
+size 279100
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/848d9dc7ad4ed08b8e1ddad74d5c1e0f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/848d9dc7ad4ed08b8e1ddad74d5c1e0f
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/848d9dc7ad4ed08b8e1ddad74d5c1e0f differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/dd43cf2f235a6029a80731bf397d753a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/dd43cf2f235a6029a80731bf397d753a
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/dd43cf2f235a6029a80731bf397d753a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/eb68431023a80918b2c3d1fcd3d5b12d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/eb68431023a80918b2c3d1fcd3d5b12d
new file mode 100644
index 0000000000000000000000000000000000000000..ccacbb67a75458e80ea864828b4c67eafa2d81ab
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/eb68431023a80918b2c3d1fcd3d5b12d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/fac44326c49a9a2c78b542db7b7275e3 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/fac44326c49a9a2c78b542db7b7275e3
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d/fac44326c49a9a2c78b542db7b7275e3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..71eab534218f09966946393821c892e1b744e110
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..1dbe964f7445050e45a6f6657abb4b3437bc4121
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592112357372340, "commit_timestamp_nsecs": 1754592112532034801, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/d/d97ac8b35c94a52edd56f6cb1b9c2c94 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/d/d97ac8b35c94a52edd56f6cb1b9c2c94
new file mode 100644
index 0000000000000000000000000000000000000000..b4cc88ad7a2e2e476179525c032fea8f51002a3b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/d/d97ac8b35c94a52edd56f6cb1b9c2c94 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a30833c801d9f99e1ac435e9fe25d08f425a3d73
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/322525488d4b0c6a96b676fb7481a741 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/322525488d4b0c6a96b676fb7481a741
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/322525488d4b0c6a96b676fb7481a741 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/4d86116a1777fa57b05d5344c1e7130f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/4d86116a1777fa57b05d5344c1e7130f
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/4d86116a1777fa57b05d5344c1e7130f differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/51e53460ef81401e8b75056d861685f4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/51e53460ef81401e8b75056d861685f4
new file mode 100644
index 0000000000000000000000000000000000000000..78905e7e0f25ad29135e1f590278e05302bdc68f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/51e53460ef81401e8b75056d861685f4
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:335f234386c1e8a7949e884f40ab0c0282b5043d6f17af4089df315185962702
+size 439710
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/5ceba46445998a873489ca4682767be7 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/5ceba46445998a873489ca4682767be7
new file mode 100644
index 0000000000000000000000000000000000000000..ec745d97353f97209ef50f0eb738585830219f53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/5ceba46445998a873489ca4682767be7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/62b3aefa0c7a69d84a5eaba5c39527f4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/62b3aefa0c7a69d84a5eaba5c39527f4
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/62b3aefa0c7a69d84a5eaba5c39527f4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/7a25ab5e90d5243b7105ad17e31f7c13 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/7a25ab5e90d5243b7105ad17e31f7c13
new file mode 100644
index 0000000000000000000000000000000000000000..fa090e6ffd1a38676e501d866105b361deaab7b9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/7a25ab5e90d5243b7105ad17e31f7c13
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fd365fe06a4e725e43d92dbe89fe1ee88f98409c45e1812af9099b798a026f75
+size 261978
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/c90a9c63439cc9ea69f07955806557f4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/c90a9c63439cc9ea69f07955806557f4
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/c90a9c63439cc9ea69f07955806557f4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/cb823c5d3c57a4e09c2922d35cb17262 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/cb823c5d3c57a4e09c2922d35cb17262
new file mode 100644
index 0000000000000000000000000000000000000000..5472f49562f363de627d9fbd196534521d17cac0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/cb823c5d3c57a4e09c2922d35cb17262 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/cd476c46b75fae75b67b4f81b5049be5 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/cd476c46b75fae75b67b4f81b5049be5
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d/cd476c46b75fae75b67b4f81b5049be5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..cce791baa593fe73a8f8ac64b92008f4dfe1b2ca
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..b02588812142927871a375eebd0e71325b9f25af
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592145671691791, "commit_timestamp_nsecs": 1754592145845265555, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/d/e036f92c12b4cd345fcbb9c5a0d9effd b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/d/e036f92c12b4cd345fcbb9c5a0d9effd
new file mode 100644
index 0000000000000000000000000000000000000000..b51b209b86099dbe8a3365aa883743ec543e8797
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/d/e036f92c12b4cd345fcbb9c5a0d9effd differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c11bad99df6cd48b6e5ba76fc12edc396647b7d8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/2ffbd83534eaa5d167f44b02b5195514 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/2ffbd83534eaa5d167f44b02b5195514
new file mode 100644
index 0000000000000000000000000000000000000000..6abf872d70937bd1d3e08e0210d74869c0b0c812
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/2ffbd83534eaa5d167f44b02b5195514 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/596a739b2d70500cce64ba4dcdeda19c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/596a739b2d70500cce64ba4dcdeda19c
new file mode 100644
index 0000000000000000000000000000000000000000..2a31377ca030aced51ef70fb458c6b5c47404cd3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/596a739b2d70500cce64ba4dcdeda19c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/63c001e7e5eb8ecc75a8257c3f8dcc94 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/63c001e7e5eb8ecc75a8257c3f8dcc94
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/63c001e7e5eb8ecc75a8257c3f8dcc94 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/7d427dc5b2ca2c9df100a2cb920f7e44 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/7d427dc5b2ca2c9df100a2cb920f7e44
new file mode 100644
index 0000000000000000000000000000000000000000..cec504229a3eddb7bbbb95c752aaaa5343421013
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/7d427dc5b2ca2c9df100a2cb920f7e44
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aaf9330170e2f5e6430fa953837a377ecda8574d0a7780ed07bbed30d5c7d8ee
+size 270492
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/7d599aa5439ca6f64c817c247b4e1740 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/7d599aa5439ca6f64c817c247b4e1740
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/7d599aa5439ca6f64c817c247b4e1740 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/93c26ed0e21fa8d54a6ea23e8aa14086 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/93c26ed0e21fa8d54a6ea23e8aa14086
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/93c26ed0e21fa8d54a6ea23e8aa14086 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/9521140a4e55270dbcc2d865d5f29db1 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/9521140a4e55270dbcc2d865d5f29db1
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/9521140a4e55270dbcc2d865d5f29db1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/9be102d5d171b4751e555a5dbc689c5c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/9be102d5d171b4751e555a5dbc689c5c
new file mode 100644
index 0000000000000000000000000000000000000000..d93d0c68ff19302b9b01bf078c19f7596900767e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/9be102d5d171b4751e555a5dbc689c5c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/a3e9bc9a3df0e030def37c9f77d0baa6 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/a3e9bc9a3df0e030def37c9f77d0baa6
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/a3e9bc9a3df0e030def37c9f77d0baa6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/b4285486ca8399f139309695e526ab74 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/b4285486ca8399f139309695e526ab74
new file mode 100644
index 0000000000000000000000000000000000000000..227ab9777d795832321c898af77aac3b4a2eebda
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/b4285486ca8399f139309695e526ab74
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c2e3919ad84e1c518fb8ac1da88d48c32d52c44d48c8965eac7f928e70872a4f
+size 464715
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/c7f9d02b97c08e8f7766f47991872d25 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/c7f9d02b97c08e8f7766f47991872d25
new file mode 100644
index 0000000000000000000000000000000000000000..40cd64d637fa9cd9f59b1971e9a5f96725cd4aef
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d/c7f9d02b97c08e8f7766f47991872d25 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0023026d767f561bc50b54d3e5b7a7c8c5b17f26
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..4665ec007f3a07ad60a181d4169348324ac43657
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592177952742369, "commit_timestamp_nsecs": 1754592178116733718, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/d/0de9c6bf66e817a5546488b47f41eceb b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/d/0de9c6bf66e817a5546488b47f41eceb
new file mode 100644
index 0000000000000000000000000000000000000000..a00b062aefa0669eb37fb91542e0775fcc8ee219
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/d/0de9c6bf66e817a5546488b47f41eceb differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..82ba110720196a9b37354b20f68a8d48f2475ee6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/0293faf144cd7f0df6c8a1cc018eac75 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/0293faf144cd7f0df6c8a1cc018eac75
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/0293faf144cd7f0df6c8a1cc018eac75 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/3ccd1c4698560be2c6e8f8ca6d914fdc b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/3ccd1c4698560be2c6e8f8ca6d914fdc
new file mode 100644
index 0000000000000000000000000000000000000000..008ef4e037023e3dafc5a28521dfe91bba412c92
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/3ccd1c4698560be2c6e8f8ca6d914fdc
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7e18dda23da73fba0ff0f39818edfd5cbdbcabf55b494f9f72b0f818b54582e0
+size 485550
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/44e846708e6e2f8a7cc4bbef41a8e8e7 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/44e846708e6e2f8a7cc4bbef41a8e8e7
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/44e846708e6e2f8a7cc4bbef41a8e8e7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/4ee6e40ab01392089555a042b99894ae b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/4ee6e40ab01392089555a042b99894ae
new file mode 100644
index 0000000000000000000000000000000000000000..3a33ed5167638d9bdba5eb00fb98ba3dc386e74e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/4ee6e40ab01392089555a042b99894ae
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2415232b7c56c7b86d37e50a2a3f1e5f5ca0578c8fea871b8da3fe6c2e953daf
+size 253552
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/6a3bc9c12a58ee02aba60ef465d1ac8c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/6a3bc9c12a58ee02aba60ef465d1ac8c
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/6a3bc9c12a58ee02aba60ef465d1ac8c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/70c90442a93b2e31682b5323a9540396 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/70c90442a93b2e31682b5323a9540396
new file mode 100644
index 0000000000000000000000000000000000000000..8e44e4bb594e4a853a57caa72e6b8c2bd6b3b09a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/70c90442a93b2e31682b5323a9540396 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/ad921ea0b6e0b6b326be04315cd40170 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/ad921ea0b6e0b6b326be04315cd40170
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/ad921ea0b6e0b6b326be04315cd40170 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/ef90008bf76ab28f4740bd9617f0ae0c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/ef90008bf76ab28f4740bd9617f0ae0c
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d/ef90008bf76ab28f4740bd9617f0ae0c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..61c5f42d3326179cd3585fed96b09015a11ae986
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..b070d3a29c785acb542653b57b28623a72111566
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592211159776090, "commit_timestamp_nsecs": 1754592211330155949, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/d/724b97a7f5c5c3c9682abbfdfd5cb227 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/d/724b97a7f5c5c3c9682abbfdfd5cb227
new file mode 100644
index 0000000000000000000000000000000000000000..57773d155eeee0f6ea5e94dda8c8d1f2aa73cf60
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/d/724b97a7f5c5c3c9682abbfdfd5cb227 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d5643d12acc11b36d4d9345f1cd54f46aa848796
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/0e171269c60e46481e152d31895e54ec b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/0e171269c60e46481e152d31895e54ec
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/0e171269c60e46481e152d31895e54ec differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/26904cadf01427bc743f3ca580863c5e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/26904cadf01427bc743f3ca580863c5e
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/26904cadf01427bc743f3ca580863c5e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/2d9cef5349fdc12c8e377b36b46c075a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/2d9cef5349fdc12c8e377b36b46c075a
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/2d9cef5349fdc12c8e377b36b46c075a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/37d985c2cdfdacf4a9c92d2354c4a83d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/37d985c2cdfdacf4a9c92d2354c4a83d
new file mode 100644
index 0000000000000000000000000000000000000000..1af6659e82b026cb0b43c36d3354b54dfcccdc47
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/37d985c2cdfdacf4a9c92d2354c4a83d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/68a90cd516a57908da16c2310572ce88 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/68a90cd516a57908da16c2310572ce88
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/68a90cd516a57908da16c2310572ce88 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/9d53a7affb45e9975657dd2f540f8964 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/9d53a7affb45e9975657dd2f540f8964
new file mode 100644
index 0000000000000000000000000000000000000000..91131504e12b862e40671251b9a3ced467577492
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/9d53a7affb45e9975657dd2f540f8964
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a7ad2c2354b58dd5e073782d933f807a6694b8ef5cd229f8a73310357313340f
+size 494122
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/c279d830c44e48ed37f1e393137418e4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/c279d830c44e48ed37f1e393137418e4
new file mode 100644
index 0000000000000000000000000000000000000000..e01d828679ac2c3f9ccdce09907dfc2bd648e351
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/c279d830c44e48ed37f1e393137418e4
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3d9f5c9ac4ae264eb0716d855d5e2c6326a47948f5f0cf806a3f270d5ce06fd0
+size 245076
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/ddc4c361a911583514c0a1f56461d960 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/ddc4c361a911583514c0a1f56461d960
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d/ddc4c361a911583514c0a1f56461d960 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..eaca976ef0b555c64a6383d17b460f7d72c627c2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..eed49f9990209e53194d9bb2cc2c7d46c160352f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592244655125301, "commit_timestamp_nsecs": 1754592244844627490, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/d/c06c35ba553900d44a311ef372ff19f5 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/d/c06c35ba553900d44a311ef372ff19f5
new file mode 100644
index 0000000000000000000000000000000000000000..174f5568e57e324f12e4ba5f49877344fb913f6d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/d/c06c35ba553900d44a311ef372ff19f5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d0affaa3200789b549df5c344f10e07f6c8c050e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/213d78d4df1c4c75cb8b22913bdb6774 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/213d78d4df1c4c75cb8b22913bdb6774
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/213d78d4df1c4c75cb8b22913bdb6774 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/3c95a5a2a09e5395d74a755930861275 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/3c95a5a2a09e5395d74a755930861275
new file mode 100644
index 0000000000000000000000000000000000000000..cff7dea533f0b6bddf0d4f9d5e6a5b46c2e113ca
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/3c95a5a2a09e5395d74a755930861275
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:12cb46d9ad797543bb5f65ff438300864e6af5cdef644220ba0b27bf56cda370
+size 306648
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/4a6e0dc2cbed23275d476ad14679e270 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/4a6e0dc2cbed23275d476ad14679e270
new file mode 100644
index 0000000000000000000000000000000000000000..566c69511a703e50e06b09c437f08b25cfe5d7c5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/4a6e0dc2cbed23275d476ad14679e270
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:72b911797bd0dc00c0188be68d5558a1a582c18c57df5fff808290fc96b1226c
+size 432196
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/79745026144581f151ea2dc9f1dcf414 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/79745026144581f151ea2dc9f1dcf414
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/79745026144581f151ea2dc9f1dcf414 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/a168e8f97dd4c81b9dec06663bfc4ec9 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/a168e8f97dd4c81b9dec06663bfc4ec9
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/a168e8f97dd4c81b9dec06663bfc4ec9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/afe8534b47149d710b92d2eb6ecf365e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/afe8534b47149d710b92d2eb6ecf365e
new file mode 100644
index 0000000000000000000000000000000000000000..ed85c3c1d31752f2472f477c05d47603751d513a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/afe8534b47149d710b92d2eb6ecf365e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/dd682c8bd27d5a410043387dc0fece73 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/dd682c8bd27d5a410043387dc0fece73
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d/dd682c8bd27d5a410043387dc0fece73 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f6f8bdfc90e0c45af36d43493a22d1e29c11a01c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..7eddad5c98338f0dd314c351b2f87808fd9e0b3d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590126776067942, "commit_timestamp_nsecs": 1754590126960631029, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/d/e63f3a85cebab955fbb0ecda401d721d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/d/e63f3a85cebab955fbb0ecda401d721d
new file mode 100644
index 0000000000000000000000000000000000000000..8cd0c3b7ec4d90c471efa9254283071391b1c504
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/d/e63f3a85cebab955fbb0ecda401d721d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e55485b6d06b9c614afb74f3e727c75545203895
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/07d976f2d73e9e893eab1b5fbbd37c06 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/07d976f2d73e9e893eab1b5fbbd37c06
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/07d976f2d73e9e893eab1b5fbbd37c06 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/1c161eda8f50282330ddd745d8b2bf43 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/1c161eda8f50282330ddd745d8b2bf43
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/1c161eda8f50282330ddd745d8b2bf43 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/4064f70e816d92951d8cc255e4c80893 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/4064f70e816d92951d8cc255e4c80893
new file mode 100644
index 0000000000000000000000000000000000000000..795fd955471d450e1f9df282b6906e18e4963e8c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/4064f70e816d92951d8cc255e4c80893 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/444c9253be938c6a7dc44db7ebaf2c2c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/444c9253be938c6a7dc44db7ebaf2c2c
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/444c9253be938c6a7dc44db7ebaf2c2c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/6014935216cf6cfd35ef9f3c3759602d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/6014935216cf6cfd35ef9f3c3759602d
new file mode 100644
index 0000000000000000000000000000000000000000..b560f2e662e27f229b00006d8cddef6cac3451c7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/6014935216cf6cfd35ef9f3c3759602d
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5f5a4c89bc98c07b438b3e3488d85b11776b2d297024d5945cd7810df38694d8
+size 694429
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/8144ad914b4cbee3eba7155ee4a3d7ac b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/8144ad914b4cbee3eba7155ee4a3d7ac
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/8144ad914b4cbee3eba7155ee4a3d7ac differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/849a9345cd0386114d91c989a73493de b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/849a9345cd0386114d91c989a73493de
new file mode 100644
index 0000000000000000000000000000000000000000..3b92ef0b2bc0a51d625bae93665d3499f0c08d95
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/849a9345cd0386114d91c989a73493de differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/9c9f388f5cc0c6559eac6074de45de8c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/9c9f388f5cc0c6559eac6074de45de8c
new file mode 100644
index 0000000000000000000000000000000000000000..4859214e6b68fa6e697de9748417fcf43be8081d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/9c9f388f5cc0c6559eac6074de45de8c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/b5410cab06ad2a857760b6c06adce9cb b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/b5410cab06ad2a857760b6c06adce9cb
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d/b5410cab06ad2a857760b6c06adce9cb differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c916157426c31f03e6beee2fd3be30fde135c45f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..d0b299df6fe79fa2e48714c21be57e2780cb9c96
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592278065779613, "commit_timestamp_nsecs": 1754592278252644623, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/d/0dc2d10aa5dd7e18f6f608169db5d50d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/d/0dc2d10aa5dd7e18f6f608169db5d50d
new file mode 100644
index 0000000000000000000000000000000000000000..812aa32dd2787f68bc9101e2b53d776b09632021
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/d/0dc2d10aa5dd7e18f6f608169db5d50d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..63526e41c6fc29b29309473cd54993d098d0d844
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/39875efee752abfd8e50082231226c27 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/39875efee752abfd8e50082231226c27
new file mode 100644
index 0000000000000000000000000000000000000000..55b0266b5155cbae7a45e79d63b6777536629c68
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/39875efee752abfd8e50082231226c27
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5cac70fa70de1cf201a76fe668f8b127df74b7584f691b4980ce8d186d18d20a
+size 413860
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/8d8064458645b54e7c20cc8e05d0d3d4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/8d8064458645b54e7c20cc8e05d0d3d4
new file mode 100644
index 0000000000000000000000000000000000000000..65529f6532287fad4b450ca493011b55e8bf7f92
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/8d8064458645b54e7c20cc8e05d0d3d4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/c109446946521f6e8e5a9a8079d79517 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/c109446946521f6e8e5a9a8079d79517
new file mode 100644
index 0000000000000000000000000000000000000000..18e2314f178e53de8c648a38ab5025f0ec7f0538
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/c109446946521f6e8e5a9a8079d79517
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:39c8a85a09a02b1e973d50511f73c80e5433fb5899c27909a8e63ea719693add
+size 279038
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/ca06e0bb771b71705e339e05442c64d3 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/ca06e0bb771b71705e339e05442c64d3
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/ca06e0bb771b71705e339e05442c64d3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/dbdb3433e49fb4251f9078b788cf8d44 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/dbdb3433e49fb4251f9078b788cf8d44
new file mode 100644
index 0000000000000000000000000000000000000000..85ed292a36497b889e49b2dae541b8226ec12561
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/dbdb3433e49fb4251f9078b788cf8d44 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/dfa651d4572cd1d221d6a7226fee28dc b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/dfa651d4572cd1d221d6a7226fee28dc
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/dfa651d4572cd1d221d6a7226fee28dc differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/f3dfda1abb5195b03fa72ff9e543fc1b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/f3dfda1abb5195b03fa72ff9e543fc1b
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/f3dfda1abb5195b03fa72ff9e543fc1b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/f841873ccf5468b37b0e84154c1a13a6 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/f841873ccf5468b37b0e84154c1a13a6
new file mode 100644
index 0000000000000000000000000000000000000000..5f1c7d54fc111a65f3aed1db1b93365b0c5a87de
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/f841873ccf5468b37b0e84154c1a13a6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/fcbe20f611e8bbbb1cd43988ea332041 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/fcbe20f611e8bbbb1cd43988ea332041
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d/fcbe20f611e8bbbb1cd43988ea332041 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8b3cb773f4c072149c8f413641607e9d04e75240
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..17af8a661d6e814be98c0ffbd00837dc2c4b664f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592311555442608, "commit_timestamp_nsecs": 1754592311735953050, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/d/2258686f8c29f378e7369b5f3bdd064c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/d/2258686f8c29f378e7369b5f3bdd064c
new file mode 100644
index 0000000000000000000000000000000000000000..77c5b0ad54ee938729ed6afab9939584624c70c2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/d/2258686f8c29f378e7369b5f3bdd064c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5e1094eb3c2cf3173ae20ae3d842c2c822038a08
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/6f3ef7587f6cdc7175715d43c93757ed b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/6f3ef7587f6cdc7175715d43c93757ed
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/6f3ef7587f6cdc7175715d43c93757ed differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/79dff3627a27fb19abafccd358fad496 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/79dff3627a27fb19abafccd358fad496
new file mode 100644
index 0000000000000000000000000000000000000000..dc32c0e26910cf04289fc1fd3d62dce8be1ca8eb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/79dff3627a27fb19abafccd358fad496
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e83a676c814d73f4e00c472d5234cb105fe29920d9d153a1f563dfa01b1b984d
+size 296166
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/bc02b60669d27a3559697f65e349f700 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/bc02b60669d27a3559697f65e349f700
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/bc02b60669d27a3559697f65e349f700 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/ce688dbcd6d369351bdb1eaa5d8eaae7 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/ce688dbcd6d369351bdb1eaa5d8eaae7
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/ce688dbcd6d369351bdb1eaa5d8eaae7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/da1a761fda6a2c9e041dca34755929fa b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/da1a761fda6a2c9e041dca34755929fa
new file mode 100644
index 0000000000000000000000000000000000000000..a4e5086e3a1d3725df00b94d5fe7764ef7d1b7cd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/da1a761fda6a2c9e041dca34755929fa differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/e71040c628990e9c501f5fae707ac24b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/e71040c628990e9c501f5fae707ac24b
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/e71040c628990e9c501f5fae707ac24b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/fc1772e31cd66a3bbd167429554d8079 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/fc1772e31cd66a3bbd167429554d8079
new file mode 100644
index 0000000000000000000000000000000000000000..ca9d929429277d3217bfdbca5569cb2834122b55
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d/fc1772e31cd66a3bbd167429554d8079
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:49cda537ed8868b1142c199c2e247568e16a97e15320abb42c9e18015ce93b98
+size 442541
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..71af8e4de592d93a6853d2791ccd8ad3ebbef94f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..c1ab36a8cd0b1a902d42e94b3d53dd47192f1ddf
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592344031149503, "commit_timestamp_nsecs": 1754592344155044994, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/d/7bcb2d73a2b924b80ad697198e4411ec b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/d/7bcb2d73a2b924b80ad697198e4411ec
new file mode 100644
index 0000000000000000000000000000000000000000..93aa6d95b79f97bee3fddaff5d596962b8de1490
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/d/7bcb2d73a2b924b80ad697198e4411ec differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0191f16551cea399ce96c62abe0885430576359e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/2e24e5509071c36ede8a0dc7f19d5fc7 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/2e24e5509071c36ede8a0dc7f19d5fc7
new file mode 100644
index 0000000000000000000000000000000000000000..7a3687db507a8b8b04c28a8691680c775b4c36fa
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/2e24e5509071c36ede8a0dc7f19d5fc7
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:71272af01997671caed259cec25b30e8d820d3b0e2bf28f90fa0aad3891eb9f8
+size 736458
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/5affb7c5bb91f15dd82e81afbf5370b6 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/5affb7c5bb91f15dd82e81afbf5370b6
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/5affb7c5bb91f15dd82e81afbf5370b6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/760aa2e472cda45f1f1d87a8acbc0c9b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/760aa2e472cda45f1f1d87a8acbc0c9b
new file mode 100644
index 0000000000000000000000000000000000000000..f28d59e1f9187cfd65b688b4bceb57a81f0acdcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/760aa2e472cda45f1f1d87a8acbc0c9b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/769c890c2238468f88100d3ad2086651 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/769c890c2238468f88100d3ad2086651
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/769c890c2238468f88100d3ad2086651 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/8eac172e0c8e06aadf5c873322e9dddf b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/8eac172e0c8e06aadf5c873322e9dddf
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/8eac172e0c8e06aadf5c873322e9dddf differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/9b822055d017d66a3d1a26ae7163db71 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/9b822055d017d66a3d1a26ae7163db71
new file mode 100644
index 0000000000000000000000000000000000000000..12a186a374c619dc2919ac9aff98b69f06567895
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d/9b822055d017d66a3d1a26ae7163db71 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0f40ba60bf694c343a6672dfa74c63c6d01aa534
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..51af7bf871259278c6f887b3c9661f16dd445e69
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592377385852664, "commit_timestamp_nsecs": 1754592377550654501, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/d/e3ed4e0945bfcf111046152f5b9e02c0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/d/e3ed4e0945bfcf111046152f5b9e02c0
new file mode 100644
index 0000000000000000000000000000000000000000..11e6d8a7af17c1cf906d92caa1aa7f037f1f2ad4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/d/e3ed4e0945bfcf111046152f5b9e02c0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4b49d135a31f967916cc355d54a7eb384a842477
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/0cb061330dd19ab823306623ea6fdf83 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/0cb061330dd19ab823306623ea6fdf83
new file mode 100644
index 0000000000000000000000000000000000000000..a00c43ee4b2a894d0c0a369d78f341c59a494c5f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/0cb061330dd19ab823306623ea6fdf83
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b2e9b2b4954ce13059c5e5744bdfe90d2041b3262ed3020c53ee374bd9ea2083
+size 279001
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/1a3e062e2d1fa372f137c8d3a905b2ee b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/1a3e062e2d1fa372f137c8d3a905b2ee
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/1a3e062e2d1fa372f137c8d3a905b2ee differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/567a268f842da4a82bd80b5e2861123c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/567a268f842da4a82bd80b5e2861123c
new file mode 100644
index 0000000000000000000000000000000000000000..372f557f7eb0159e0f72fc42d791b5c3152a7f70
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/567a268f842da4a82bd80b5e2861123c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8023507510eb4d0bb78e9cb4bcf3cb924a1b49aff9bacdc74a1146887e8e3377
+size 459902
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/5c514ff7284628b65681a16cf3d93f31 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/5c514ff7284628b65681a16cf3d93f31
new file mode 100644
index 0000000000000000000000000000000000000000..99f5cd5891914d4ee809ced31537323177e84c72
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/5c514ff7284628b65681a16cf3d93f31 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/b38cf248d9c06bcdeabd1e6ad30ff3f0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/b38cf248d9c06bcdeabd1e6ad30ff3f0
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/b38cf248d9c06bcdeabd1e6ad30ff3f0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/c650a374f57808a3d8a9ef1dc2716bd5 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/c650a374f57808a3d8a9ef1dc2716bd5
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/c650a374f57808a3d8a9ef1dc2716bd5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/f76b5dc7279ed623a8f6114aa2b5dbfd b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/f76b5dc7279ed623a8f6114aa2b5dbfd
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d/f76b5dc7279ed623a8f6114aa2b5dbfd differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e2c91027903d66c3e8e39123a361419e3e88d56f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..9f8be054ca2beabbf8761c56c2b66d09f4c217aa
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592410662422953, "commit_timestamp_nsecs": 1754592410836500194, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/d/fbe9061d40d7d5b992bae687bb3ee392 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/d/fbe9061d40d7d5b992bae687bb3ee392
new file mode 100644
index 0000000000000000000000000000000000000000..061c20a75703ad4a635f6a18d9be1e3195909479
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/d/fbe9061d40d7d5b992bae687bb3ee392 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..2486b06d8e81fc56a59e840fcf21a8b3c8ddf535
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/2902cdea12c2c5789b89e9db1cd2dd34 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/2902cdea12c2c5789b89e9db1cd2dd34
new file mode 100644
index 0000000000000000000000000000000000000000..e9c77feab810a8bc2a3b9388ca46f7d7e10b3682
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/2902cdea12c2c5789b89e9db1cd2dd34 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/65464fdeae2a520df3aad2c2518d53a6 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/65464fdeae2a520df3aad2c2518d53a6
new file mode 100644
index 0000000000000000000000000000000000000000..b9cfb452b86094e6d7c955285d6f4f67fe69ab36
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/65464fdeae2a520df3aad2c2518d53a6
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d7b5a1092acc3a5becd2d2de6f1b30daa6f63b0804df0aba039fe21393388e0d
+size 329159
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/6e9fe73d29e2e188780eaa8214547c1c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/6e9fe73d29e2e188780eaa8214547c1c
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/6e9fe73d29e2e188780eaa8214547c1c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/747a834ef94be370c8aecf1d06396b67 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/747a834ef94be370c8aecf1d06396b67
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/747a834ef94be370c8aecf1d06396b67 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/799e6dfc1cac6f911b88d8a872bf5971 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/799e6dfc1cac6f911b88d8a872bf5971
new file mode 100644
index 0000000000000000000000000000000000000000..a60ad5bf02d901b6f51bb23c3bcc554e9daad302
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/799e6dfc1cac6f911b88d8a872bf5971
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e048f07bd46c750f96eeff83114e9511537df7ab949eb4d6ce10013bf6e35221
+size 403905
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/80ccf9a1d6db6ce3dc9099ef1fc35c7f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/80ccf9a1d6db6ce3dc9099ef1fc35c7f
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/80ccf9a1d6db6ce3dc9099ef1fc35c7f differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/e7db192fe609f0cfef84e48af5e8499b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/e7db192fe609f0cfef84e48af5e8499b
new file mode 100644
index 0000000000000000000000000000000000000000..85f35e7991cac7b6fc244bb2f277438119f4e90b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d/e7db192fe609f0cfef84e48af5e8499b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f9ff184204b411eadf42dca95f1c9cf51667a9bb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..b416953e85eac5637359ebdea59307f66da04e92
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592442219231653, "commit_timestamp_nsecs": 1754592442362673703, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/d/77c59c3da30c22b7c3c971546692715b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/d/77c59c3da30c22b7c3c971546692715b
new file mode 100644
index 0000000000000000000000000000000000000000..db9d4135358517a9007586e363c3ea6b15d9ac87
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/d/77c59c3da30c22b7c3c971546692715b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..13242bd1d9a70e1b6c09cd769806b94e5a8f96d3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/0740222080792b5031d5fde69df6b047 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/0740222080792b5031d5fde69df6b047
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/0740222080792b5031d5fde69df6b047 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/1b514f73ef3481d368d7bcb8dd1d2d9a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/1b514f73ef3481d368d7bcb8dd1d2d9a
new file mode 100644
index 0000000000000000000000000000000000000000..85f35e7991cac7b6fc244bb2f277438119f4e90b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/1b514f73ef3481d368d7bcb8dd1d2d9a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/28521f8933a49ee33be11d2f79faf2fa b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/28521f8933a49ee33be11d2f79faf2fa
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/28521f8933a49ee33be11d2f79faf2fa differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/87d695f364b22e8837e82ef9a800c319 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/87d695f364b22e8837e82ef9a800c319
new file mode 100644
index 0000000000000000000000000000000000000000..5918a9e0a798ee0b9d673bff688218227de17c99
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/87d695f364b22e8837e82ef9a800c319 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/a10931fd1492a8e6975b0e2e7cc15e78 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/a10931fd1492a8e6975b0e2e7cc15e78
new file mode 100644
index 0000000000000000000000000000000000000000..a2564027f31196f848e04c6e90b5f7b223f4fd41
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/a10931fd1492a8e6975b0e2e7cc15e78 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/eafe136ca741896b388dac0c8b635b85 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/eafe136ca741896b388dac0c8b635b85
new file mode 100644
index 0000000000000000000000000000000000000000..c0dac1455c365581a821f146a4fcf5c6f970ef59
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/eafe136ca741896b388dac0c8b635b85
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6875b06b949aacdc93b258d4e20cb867642dfb148ae1c72b1284f53d26991a84
+size 697394
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/f7a6817ada1ef72ab5297220bf75529e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/f7a6817ada1ef72ab5297220bf75529e
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d/f7a6817ada1ef72ab5297220bf75529e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..6bb1ff3b31d12c524402bf86151acd01943a2de5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..b3a8eabd42ecfabf37c95ad0f1644291d70a74e1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592476419796758, "commit_timestamp_nsecs": 1754592476559441983, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/d/01a9c3bd5bc92e6e3805f3983df0aa86 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/d/01a9c3bd5bc92e6e3805f3983df0aa86
new file mode 100644
index 0000000000000000000000000000000000000000..9fd1a3478cf886fa710367970fd5fd11eed263be
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/d/01a9c3bd5bc92e6e3805f3983df0aa86 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..1aef0128fa557b5c8366bc50b7e075a3f8c36adc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/010f00ac9b1bcee8bcb5fff2d5bfaa8d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/010f00ac9b1bcee8bcb5fff2d5bfaa8d
new file mode 100644
index 0000000000000000000000000000000000000000..aae89c4ef26cf60f0bb10fa45d729ddb594f60dc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/010f00ac9b1bcee8bcb5fff2d5bfaa8d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/1bb597715001a86bcc925c5d0bc5bc1c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/1bb597715001a86bcc925c5d0bc5bc1c
new file mode 100644
index 0000000000000000000000000000000000000000..2910fbc6db2094404ed932c7d9bcdb23beb99906
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/1bb597715001a86bcc925c5d0bc5bc1c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/3fb7a99219d574182236a6b77ecbb012 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/3fb7a99219d574182236a6b77ecbb012
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/3fb7a99219d574182236a6b77ecbb012 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/417b0569f807d4869630c51278c53383 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/417b0569f807d4869630c51278c53383
new file mode 100644
index 0000000000000000000000000000000000000000..4859214e6b68fa6e697de9748417fcf43be8081d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/417b0569f807d4869630c51278c53383 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/9f2f9ec47d3fc02c3699a9315bc8eaec b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/9f2f9ec47d3fc02c3699a9315bc8eaec
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/9f2f9ec47d3fc02c3699a9315bc8eaec differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/adac7c6c762224cd4b26832da5f6bfab b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/adac7c6c762224cd4b26832da5f6bfab
new file mode 100644
index 0000000000000000000000000000000000000000..b5d54b821de85a77ce08a7138694ed14953f91b5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/adac7c6c762224cd4b26832da5f6bfab
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b8a8fba2278f3d2e9ca587d97858691e1e6644fa45d7a73795456b8558294144
+size 736532
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/d7e39015d0f98f582a7f3ef826acbb55 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/d7e39015d0f98f582a7f3ef826acbb55
new file mode 100644
index 0000000000000000000000000000000000000000..dfbfe1a61305bab6f716f83c77874d9e3176999d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d/d7e39015d0f98f582a7f3ef826acbb55 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5cf2d4792231d00ad17682cfb5c6d9213d19b976
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..e043c8695d7a5714853d1a1096aa2fa3b7dde86b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592509542630962, "commit_timestamp_nsecs": 1754592509723213066, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/d/a5a161f2b2ae39a1896fbc10bbb766de b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/d/a5a161f2b2ae39a1896fbc10bbb766de
new file mode 100644
index 0000000000000000000000000000000000000000..e6d2e4286ab7d1e1d77d3fd75161fab00aa48615
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/d/a5a161f2b2ae39a1896fbc10bbb766de differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f6fdd581e92b91bc86578d00cb0095c33c232dfa
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/298145a7a01972f925ed3500c4991bb8 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/298145a7a01972f925ed3500c4991bb8
new file mode 100644
index 0000000000000000000000000000000000000000..83686e8484e9066b4dbd5259a6e3b3c7e74d767b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/298145a7a01972f925ed3500c4991bb8
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:830317480782b7f9741d99a93eedb8e76805833d42405035198a7b0a1490ff0b
+size 279075
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/413940acacce0d24ea8e5a3ad34f4375 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/413940acacce0d24ea8e5a3ad34f4375
new file mode 100644
index 0000000000000000000000000000000000000000..300b5b7f4dd1e99d7808fcc14e2614382a5c3532
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/413940acacce0d24ea8e5a3ad34f4375
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:222924f5500e51f90902813799f6d67f0fb7f569c7d4f33f8c510e71740dd16d
+size 426205
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/84e926cea1f5dd2c28097a9af6ce58ea b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/84e926cea1f5dd2c28097a9af6ce58ea
new file mode 100644
index 0000000000000000000000000000000000000000..4d119cafb9b6bd7ae0fad8ee746dd45b93d15d68
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/84e926cea1f5dd2c28097a9af6ce58ea differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/a009e92cf574b47a352cdfe4e0258f64 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/a009e92cf574b47a352cdfe4e0258f64
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/a009e92cf574b47a352cdfe4e0258f64 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/c3e56c4fd708e596e17e69d20630b0c0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/c3e56c4fd708e596e17e69d20630b0c0
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/c3e56c4fd708e596e17e69d20630b0c0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/cb39e60ba04cba4e4317514760967723 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/cb39e60ba04cba4e4317514760967723
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/cb39e60ba04cba4e4317514760967723 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/d9f9c116f4e05cf230a4aa6762ef0e97 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/d9f9c116f4e05cf230a4aa6762ef0e97
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/d9f9c116f4e05cf230a4aa6762ef0e97 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/fa5ecc503663191683e8de85980fe2ab b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/fa5ecc503663191683e8de85980fe2ab
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d/fa5ecc503663191683e8de85980fe2ab differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..66fc53fe7450925d005143c1d06c5b383ea79892
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..e48987eb640ee0fd61e2f0cd22a25ca6eb151761
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592543316989497, "commit_timestamp_nsecs": 1754592543465611777, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/d/70f25ce4a087216b2e107619fa850220 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/d/70f25ce4a087216b2e107619fa850220
new file mode 100644
index 0000000000000000000000000000000000000000..79a9a9e654aeb66735a5794a48fdf7f0d8c7cfd3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/d/70f25ce4a087216b2e107619fa850220 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a50d65b3326ae55186a40c5b5d9e25732a2d614c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/0c82f5671d891a548b5a633d9da1d5ff b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/0c82f5671d891a548b5a633d9da1d5ff
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/0c82f5671d891a548b5a633d9da1d5ff differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/1e0e699fa3f01e1a38beb5bea534e58c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/1e0e699fa3f01e1a38beb5bea534e58c
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/1e0e699fa3f01e1a38beb5bea534e58c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/1fdbaf290fa767a41b25c1a1defcdd4a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/1fdbaf290fa767a41b25c1a1defcdd4a
new file mode 100644
index 0000000000000000000000000000000000000000..bb878ef16e7e5a8165b8de7ea3b5cfa971ac8aa5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/1fdbaf290fa767a41b25c1a1defcdd4a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f71657262d76659e427687523d2db880e6707562cad6b95b46f8739af7c1abd6
+size 442953
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/66a67f5dae7fa9257691d4dc639086f8 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/66a67f5dae7fa9257691d4dc639086f8
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/66a67f5dae7fa9257691d4dc639086f8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/b09820586a36716e0f4ebe86ecad17d9 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/b09820586a36716e0f4ebe86ecad17d9
new file mode 100644
index 0000000000000000000000000000000000000000..5439651eb39d30964d6ea8508c0b5b39660cc095
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/b09820586a36716e0f4ebe86ecad17d9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/cc1ba1937c4c9c9c72c807ac6982c96e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/cc1ba1937c4c9c9c72c807ac6982c96e
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/cc1ba1937c4c9c9c72c807ac6982c96e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/ce49e86dbd371a81b06cb0e0d4513f3e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/ce49e86dbd371a81b06cb0e0d4513f3e
new file mode 100644
index 0000000000000000000000000000000000000000..276744ddd28d703aa10856d8d42674c7f314003f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/ce49e86dbd371a81b06cb0e0d4513f3e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bb71f14f93d1fe2e7e854977474f0f9a63ca410518470ae5ceac5d63074c9b66
+size 296180
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/d3d4e0d7a8197ca768b0c9ecfb09abb4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/d3d4e0d7a8197ca768b0c9ecfb09abb4
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d/d3d4e0d7a8197ca768b0c9ecfb09abb4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c4a8f9203e456ebf140341ebd099c9b139ad5519
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..feef0e040d472c1a87981e555727ad623722e805
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592576555685586, "commit_timestamp_nsecs": 1754592576738096968, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/d/4214d949275d096d8aadd0f6d3596cad b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/d/4214d949275d096d8aadd0f6d3596cad
new file mode 100644
index 0000000000000000000000000000000000000000..074fc8b7c3d9cc716551f1a941ab000135b97b59
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/d/4214d949275d096d8aadd0f6d3596cad differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e45db110244d725199f26a2ee6570b2886b099c5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/035dbbb6b9ba5e8e390953ef84a7dbbd b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/035dbbb6b9ba5e8e390953ef84a7dbbd
new file mode 100644
index 0000000000000000000000000000000000000000..eb13ed1ec8e7e70346e45f909408dfbb89ddad40
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/035dbbb6b9ba5e8e390953ef84a7dbbd
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2444514f5a861b382ee1edcc3d129f5c364e7fcacdd21f1f0d9d8251bf768d0a
+size 340542
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/3700fc240381ce65ab7516f1d6bc2d75 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/3700fc240381ce65ab7516f1d6bc2d75
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/3700fc240381ce65ab7516f1d6bc2d75 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/37181a933b507c5ce42bed8f65c571a9 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/37181a933b507c5ce42bed8f65c571a9
new file mode 100644
index 0000000000000000000000000000000000000000..4859214e6b68fa6e697de9748417fcf43be8081d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/37181a933b507c5ce42bed8f65c571a9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/426713bf089c460cdcc28811ad8c31bc b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/426713bf089c460cdcc28811ad8c31bc
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/426713bf089c460cdcc28811ad8c31bc differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/50c850b63be8cc3626fa9458d2d54caa b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/50c850b63be8cc3626fa9458d2d54caa
new file mode 100644
index 0000000000000000000000000000000000000000..33555a514949f44da7eb462de2808c520748251e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/50c850b63be8cc3626fa9458d2d54caa
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:252926d1b35186887256ac86a68feba27c232ea6e82716ad59a44abc502e7030
+size 398147
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/6bc9a11385d4565e384f6873d7c73221 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/6bc9a11385d4565e384f6873d7c73221
new file mode 100644
index 0000000000000000000000000000000000000000..c8e405ca897009a3790fe9ffc5b7382c35f9940a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/6bc9a11385d4565e384f6873d7c73221 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/86f1a70bd273dc1591e5de9efcf3a337 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/86f1a70bd273dc1591e5de9efcf3a337
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/86f1a70bd273dc1591e5de9efcf3a337 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/c4d861af3fadb0d583ead28e23c76000 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/c4d861af3fadb0d583ead28e23c76000
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d/c4d861af3fadb0d583ead28e23c76000 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..8506b6d384fa59e76212110b234a31c16017bd73
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..bd757cd6a35a7bf49d599942653ab7cfe446f1f3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590159935803171, "commit_timestamp_nsecs": 1754590160120798478, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/d/67bba48d902590e0689abef90cb42f8e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/d/67bba48d902590e0689abef90cb42f8e
new file mode 100644
index 0000000000000000000000000000000000000000..9abdd3f10f59bccd7442a700e7441efaaa041e74
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/d/67bba48d902590e0689abef90cb42f8e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..fb287e1d8bf5533944c29319938b7d52c8f515e0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/136cbfbb520725abaa6e22650950ca7e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/136cbfbb520725abaa6e22650950ca7e
new file mode 100644
index 0000000000000000000000000000000000000000..1b6b4d0a47c77efb8dc98c9fe39a49f766a1267a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/136cbfbb520725abaa6e22650950ca7e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6802714c68de835b0bf1579a4f5beb0dfae3d64f4ad81372dcdc7de24e12262d
+size 287218
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/3cb2b8156ac33654480d4bfe9689760b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/3cb2b8156ac33654480d4bfe9689760b
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/3cb2b8156ac33654480d4bfe9689760b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/4a4895d53cd6f9d8c12e1b8fe0ba13dc b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/4a4895d53cd6f9d8c12e1b8fe0ba13dc
new file mode 100644
index 0000000000000000000000000000000000000000..7c733356106301189eafcb905c4d31b195ca3c20
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/4a4895d53cd6f9d8c12e1b8fe0ba13dc differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/77c8e1e953dbef2c18104d8cf83f4d30 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/77c8e1e953dbef2c18104d8cf83f4d30
new file mode 100644
index 0000000000000000000000000000000000000000..abc642d6a78c52cd71abfc5c27c508e0e53baa58
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/77c8e1e953dbef2c18104d8cf83f4d30
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4c286930028e5baded0ac2d925adc4c23e92c582118168c93a09c11a65aa64e9
+size 451066
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/7c6e92bd1feef641dc78d420fae607f8 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/7c6e92bd1feef641dc78d420fae607f8
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/7c6e92bd1feef641dc78d420fae607f8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/9670b6b1e2f96103b711b984175f140d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/9670b6b1e2f96103b711b984175f140d
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/9670b6b1e2f96103b711b984175f140d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/a98b48640c69063bf94d49fab73f9d0f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/a98b48640c69063bf94d49fab73f9d0f
new file mode 100644
index 0000000000000000000000000000000000000000..2910fbc6db2094404ed932c7d9bcdb23beb99906
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/a98b48640c69063bf94d49fab73f9d0f differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/e80315be242d5c051d5923ebf53bff8e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/e80315be242d5c051d5923ebf53bff8e
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d/e80315be242d5c051d5923ebf53bff8e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..35491e81710e5b74e50c89c149079e1211067e23
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..90c142564c6d6374ccf0757fa620f65f530ec221
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592623429066858, "commit_timestamp_nsecs": 1754592623567935890, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/d/8e655f44f10362e96a2920617316964f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/d/8e655f44f10362e96a2920617316964f
new file mode 100644
index 0000000000000000000000000000000000000000..f4547f06c9153a0d56ab8282866640afcd50d0fb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/d/8e655f44f10362e96a2920617316964f differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c8df460e4c162ce4a2499da29a6f5f10517dbf98
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/46edfb25c4752fda3cfbc24048b746d2 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/46edfb25c4752fda3cfbc24048b746d2
new file mode 100644
index 0000000000000000000000000000000000000000..c960a4aae6b00862c8a46e9bbb5ec03db8be73ce
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/46edfb25c4752fda3cfbc24048b746d2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/5e8feff81f17413d6d898c9835039cfe b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/5e8feff81f17413d6d898c9835039cfe
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/5e8feff81f17413d6d898c9835039cfe differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/8c6c75fd76ccdd760e2fba049230b6dd b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/8c6c75fd76ccdd760e2fba049230b6dd
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/8c6c75fd76ccdd760e2fba049230b6dd differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/952154972d51f63f6d28a7482033c55a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/952154972d51f63f6d28a7482033c55a
new file mode 100644
index 0000000000000000000000000000000000000000..5c838481b321d4779b75b50dcdb2c7199d8cf967
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/952154972d51f63f6d28a7482033c55a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c79f0e5912ed17718e1b9a0f394d1d799234697cebc6588908f2cdc4e100f891
+size 528140
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/a1faadcc34685eae16888a269c9a78ab b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/a1faadcc34685eae16888a269c9a78ab
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/a1faadcc34685eae16888a269c9a78ab differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/c9bb6b91cd7153c6dcb8bce5fc7f6c3e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/c9bb6b91cd7153c6dcb8bce5fc7f6c3e
new file mode 100644
index 0000000000000000000000000000000000000000..0b91b76f62206764a1f4362bc370ebe6f96d6a84
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/c9bb6b91cd7153c6dcb8bce5fc7f6c3e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:615892e311cac07166b2b320c6b64b8549b30510825afa4418a550200e0ed312
+size 211015
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/e39c54535d972abd7b7aa0cd91e8c520 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/e39c54535d972abd7b7aa0cd91e8c520
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/e39c54535d972abd7b7aa0cd91e8c520 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/ef396ec917b8c1a84b6c7be6927007be b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/ef396ec917b8c1a84b6c7be6927007be
new file mode 100644
index 0000000000000000000000000000000000000000..85f35e7991cac7b6fc244bb2f277438119f4e90b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d/ef396ec917b8c1a84b6c7be6927007be differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..efb29000c14f289dca503ac008824bf14170addc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..24de9bd52b032c7c18fc1a0a48695d9ef693e107
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592656668422310, "commit_timestamp_nsecs": 1754592656844534554, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/d/0b7c9107ae8e0abf5c6cfeb0e003030a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/d/0b7c9107ae8e0abf5c6cfeb0e003030a
new file mode 100644
index 0000000000000000000000000000000000000000..0ea53e44995626f112fafeb78b847ebc864ad26f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/d/0b7c9107ae8e0abf5c6cfeb0e003030a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9881bc60005296cbb03401e83506a9e9d93e42b0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/0c70b5304913ab1bd35eb9158475d042 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/0c70b5304913ab1bd35eb9158475d042
new file mode 100644
index 0000000000000000000000000000000000000000..af7e256b9e9aefa66a850c3b85e58cc43a296126
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/0c70b5304913ab1bd35eb9158475d042 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/3b74d770da466c2a95eb0c4c8ea4af2b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/3b74d770da466c2a95eb0c4c8ea4af2b
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/3b74d770da466c2a95eb0c4c8ea4af2b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/8b254d17a83d04d047a00d2472b14106 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/8b254d17a83d04d047a00d2472b14106
new file mode 100644
index 0000000000000000000000000000000000000000..2d1b894d2e9fdaa1728cfb5178bfcb65ebd5f963
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/8b254d17a83d04d047a00d2472b14106
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e0a2354dd05485b7ff06a49d5baab38415c64bc92a2a588f911554039a80071c
+size 445916
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/ac852f194acedf836f5376763d299a72 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/ac852f194acedf836f5376763d299a72
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/ac852f194acedf836f5376763d299a72 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/b84e8c64ad1aa000624eecee725967da b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/b84e8c64ad1aa000624eecee725967da
new file mode 100644
index 0000000000000000000000000000000000000000..417acd30d3e88e202121a2026afc254686e3ac36
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/b84e8c64ad1aa000624eecee725967da
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f50d173a36981bbfbfd1f820f44bf1e9ddc7dbf0da8f5548a04a4b2eaf9b6231
+size 253624
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/d354888bd4b0508d59188cae3daba8cf b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/d354888bd4b0508d59188cae3daba8cf
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/d354888bd4b0508d59188cae3daba8cf differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/da2e0d1eac9b7e5465b2bb7ee2f0f879 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/da2e0d1eac9b7e5465b2bb7ee2f0f879
new file mode 100644
index 0000000000000000000000000000000000000000..08904dbf2dd778fb7e8390fff109c297d4000f0e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/da2e0d1eac9b7e5465b2bb7ee2f0f879 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/ed14049c5220c45a546950f27fe0d539 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/ed14049c5220c45a546950f27fe0d539
new file mode 100644
index 0000000000000000000000000000000000000000..65c55cef1e0f044fdb725f402e95d9a23d03f6b1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d/ed14049c5220c45a546950f27fe0d539 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..5f7c42fbd2418829fd08d8a77216b3580a86e14d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..04fd03f23a44c5988d8d087d4343447bb9a1cc04
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592689633368701, "commit_timestamp_nsecs": 1754592689764980656, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/d/548c453f27f1d73d33eb4c132ecdfa0a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/d/548c453f27f1d73d33eb4c132ecdfa0a
new file mode 100644
index 0000000000000000000000000000000000000000..1c555f9a09887208453478c05bcdf1a34ab4a152
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/d/548c453f27f1d73d33eb4c132ecdfa0a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a542e90d7237362655fe94eb838077a5d23a51b0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/31e641365d8c4a55576f28572abde240 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/31e641365d8c4a55576f28572abde240
new file mode 100644
index 0000000000000000000000000000000000000000..6595d42f88255047a112e252060010875d8a87a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/31e641365d8c4a55576f28572abde240 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/32843defd0ad26e175799d7e06ae351e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/32843defd0ad26e175799d7e06ae351e
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/32843defd0ad26e175799d7e06ae351e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/3e079ecd214b5e6662dfff02c36a8c7e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/3e079ecd214b5e6662dfff02c36a8c7e
new file mode 100644
index 0000000000000000000000000000000000000000..47c28d656f7bd0783e89b65266d4feba670a62da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/3e079ecd214b5e6662dfff02c36a8c7e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/5e4e84f08ee12292b4936095df2d0335 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/5e4e84f08ee12292b4936095df2d0335
new file mode 100644
index 0000000000000000000000000000000000000000..77d5efc8ded714eae14798d17ee3cd947e713923
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/5e4e84f08ee12292b4936095df2d0335
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cbd1db3e67ac13033dc2c02fac16460f78fcf8c165a36de8178ed2e83f05c608
+size 228148
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/782bebe6c245036a994431d17a4a9c7f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/782bebe6c245036a994431d17a4a9c7f
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/782bebe6c245036a994431d17a4a9c7f differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/cae7f263819fe7c3119a903dcf12780a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/cae7f263819fe7c3119a903dcf12780a
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/cae7f263819fe7c3119a903dcf12780a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/cde8eaf59419b392f61923c6a9c23ffa b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/cde8eaf59419b392f61923c6a9c23ffa
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/cde8eaf59419b392f61923c6a9c23ffa differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/f9cd12020691dd6da865d71d10c051ee b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/f9cd12020691dd6da865d71d10c051ee
new file mode 100644
index 0000000000000000000000000000000000000000..96459d4ab5c9befebacd8099a2f65803e9a7b851
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d/f9cd12020691dd6da865d71d10c051ee
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ffc24ab21968a5ea028cda2dd8f1ba5fc8ebfdf51f37951f4fe9aca71df89da6
+size 461108
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..6cc7a7501fea3e0c1cfddc76561b36751d0d774e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..e890045368d458dd51eca36356dc2131f69566ea
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592722530668736, "commit_timestamp_nsecs": 1754592722668342945, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/d/1561bd4a89edcf0b66a1cf602992d7c5 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/d/1561bd4a89edcf0b66a1cf602992d7c5
new file mode 100644
index 0000000000000000000000000000000000000000..9cb48e4f287a9283000c81e6c2330e6bb8ca6920
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/d/1561bd4a89edcf0b66a1cf602992d7c5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b20102ad3e503103df9c4ffdb6bd7c3088f3e8ca
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/07ad5694c66aaaca5cc80e61e0ccc4e8 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/07ad5694c66aaaca5cc80e61e0ccc4e8
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/07ad5694c66aaaca5cc80e61e0ccc4e8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/2090e255353ccc7c0eed74a6974bf613 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/2090e255353ccc7c0eed74a6974bf613
new file mode 100644
index 0000000000000000000000000000000000000000..e85572f02d6d5e4dbd304edc5602171b8506a73f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/2090e255353ccc7c0eed74a6974bf613 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/4495f34e7e6c2f3e52bbdd384ff8d86b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/4495f34e7e6c2f3e52bbdd384ff8d86b
new file mode 100644
index 0000000000000000000000000000000000000000..65529f6532287fad4b450ca493011b55e8bf7f92
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/4495f34e7e6c2f3e52bbdd384ff8d86b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/6ef8bd80a24b277bb9c93d361b5bc0c6 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/6ef8bd80a24b277bb9c93d361b5bc0c6
new file mode 100644
index 0000000000000000000000000000000000000000..fc75a80d5c95d4f877f596904e52af65f8925579
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/6ef8bd80a24b277bb9c93d361b5bc0c6
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:043ae12aad250cbe187f6052671122ccaff8aaf7aa68874dc2b9a6ca188bebc3
+size 370968
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/79f61a2f56c2a5699970852c978b51ef b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/79f61a2f56c2a5699970852c978b51ef
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/79f61a2f56c2a5699970852c978b51ef differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/8d00200cd86b8b836b8c87082dba457b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/8d00200cd86b8b836b8c87082dba457b
new file mode 100644
index 0000000000000000000000000000000000000000..85f35e7991cac7b6fc244bb2f277438119f4e90b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/8d00200cd86b8b836b8c87082dba457b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/e2f1132ae4dd70ae673d0c819170fe2e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/e2f1132ae4dd70ae673d0c819170fe2e
new file mode 100644
index 0000000000000000000000000000000000000000..ea2d5c7f699bb9cb4b4cb82a80f5ba9de95e8044
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/e2f1132ae4dd70ae673d0c819170fe2e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0560f2216f9bda32309fa0199dfa82a6f80c0353783633d462de7676f6ffe95b
+size 296114
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/e6c9c5dd7e38bb20f598241e9f54862e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/e6c9c5dd7e38bb20f598241e9f54862e
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d/e6c9c5dd7e38bb20f598241e9f54862e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..6f634720d5db58f7dc46e803d20f823821c79748
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..8b8b6b382772e0e115d64ea393441528a6824136
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592755481379601, "commit_timestamp_nsecs": 1754592755646784566, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/d/84f6d39c919849b6815c7062ec254a9e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/d/84f6d39c919849b6815c7062ec254a9e
new file mode 100644
index 0000000000000000000000000000000000000000..72561161e1c2e128c67a45571f708d44ca3e7876
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/d/84f6d39c919849b6815c7062ec254a9e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..26eeb866c9e7a4e81e1215ec48065a418ac27b35
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/12f0ef4b530f96c997b2f2f94092a6e2 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/12f0ef4b530f96c997b2f2f94092a6e2
new file mode 100644
index 0000000000000000000000000000000000000000..601df24a7828cd4575f9cd728a0854353322e669
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/12f0ef4b530f96c997b2f2f94092a6e2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/22977cfe763a01404e002b9d08fd7eef b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/22977cfe763a01404e002b9d08fd7eef
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/22977cfe763a01404e002b9d08fd7eef differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/3745f565723b944fd3b295e5e483d1d5 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/3745f565723b944fd3b295e5e483d1d5
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/3745f565723b944fd3b295e5e483d1d5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/3efcec4a17ebe7c400aa4c93ba7a8ee4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/3efcec4a17ebe7c400aa4c93ba7a8ee4
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/3efcec4a17ebe7c400aa4c93ba7a8ee4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/7fd6c03e550759cab1e958f47c134c65 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/7fd6c03e550759cab1e958f47c134c65
new file mode 100644
index 0000000000000000000000000000000000000000..34539683755ee4c12e1c609c8e7321eaf99b892b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/7fd6c03e550759cab1e958f47c134c65
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5f2eecef5afeb03e4b7c29eef85b5de9d9e5e8c4080e1847158af9273b9ad5ee
+size 430044
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/839c5456865abe3b6301afed9ffe6e22 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/839c5456865abe3b6301afed9ffe6e22
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/839c5456865abe3b6301afed9ffe6e22 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/be16e466dd52348fe11642d451add89d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/be16e466dd52348fe11642d451add89d
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/be16e466dd52348fe11642d451add89d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/c74c1d8609f11d655f8b48ff55f19932 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/c74c1d8609f11d655f8b48ff55f19932
new file mode 100644
index 0000000000000000000000000000000000000000..ed8fed2eaff6d4d5d753639d02d2a755169b9226
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/c74c1d8609f11d655f8b48ff55f19932
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e5036c5f43835a17507b451fb7e250990568913c7d5b31fa8838656eaff1b7e9
+size 261972
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/d98be7fe5c1200714778557ad0b191b3 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/d98be7fe5c1200714778557ad0b191b3
new file mode 100644
index 0000000000000000000000000000000000000000..4859214e6b68fa6e697de9748417fcf43be8081d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d/d98be7fe5c1200714778557ad0b191b3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..9fdb0aba30e04984c775e35fd3bd439af1ee9342
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..0fd6c6d208e3877ea7ab30e8d9f81a8f3065bb11
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592788231207266, "commit_timestamp_nsecs": 1754592788366382746, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/d/374bf28b00a4132bcf2a44364cd6667a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/d/374bf28b00a4132bcf2a44364cd6667a
new file mode 100644
index 0000000000000000000000000000000000000000..ebe8a522b820c189cadf5e97e73650b5dfb163a5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/d/374bf28b00a4132bcf2a44364cd6667a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3e2039e0bd18f52bfd41fc3aac05e3b499720931
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/01b40a0493415fef531db5cca9e5b952 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/01b40a0493415fef531db5cca9e5b952
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/01b40a0493415fef531db5cca9e5b952 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/427ce466f64d864bf11f0e88a095589a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/427ce466f64d864bf11f0e88a095589a
new file mode 100644
index 0000000000000000000000000000000000000000..0ce6f9d0550777ad7a23ca9ba7af39d019d2cad8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/427ce466f64d864bf11f0e88a095589a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/48add79d09596dbe02d0aac8bb0ad1fa b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/48add79d09596dbe02d0aac8bb0ad1fa
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/48add79d09596dbe02d0aac8bb0ad1fa differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/57daa6b736cf4e3f44bc3301a66c422f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/57daa6b736cf4e3f44bc3301a66c422f
new file mode 100644
index 0000000000000000000000000000000000000000..fc0b976c73ff39c9d88c4205de4a9c8e5d0d385b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/57daa6b736cf4e3f44bc3301a66c422f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e84f283f8c986744747eab230dc76c35ae413b27088a2b54f8c057645e5bab25
+size 296257
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/63cbdacd1f822a430be99f53f8b379cc b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/63cbdacd1f822a430be99f53f8b379cc
new file mode 100644
index 0000000000000000000000000000000000000000..cb22cb1c84a4f9e8340b52780d405cd3dcbccf4f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/63cbdacd1f822a430be99f53f8b379cc
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5fa4dd49eb24e7b625f3ab6c6c5d0d1bb1d5fb78b9546373f44b11c2217701b1
+size 442802
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/63d92a05a3705068c6a156edab41513c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/63d92a05a3705068c6a156edab41513c
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/63d92a05a3705068c6a156edab41513c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/78a7e04f45bd7e14e4365b22ec9ec783 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/78a7e04f45bd7e14e4365b22ec9ec783
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/78a7e04f45bd7e14e4365b22ec9ec783 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/c06756a72624575a04a8c3475ebc8ff8 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/c06756a72624575a04a8c3475ebc8ff8
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d/c06756a72624575a04a8c3475ebc8ff8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..11677ff8923c5f09b8bb419d15fe59527e0b639d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..0795619045bfbe8a5a434c53e1bebaa0eca03a58
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592820951089649, "commit_timestamp_nsecs": 1754592821132330207, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/d/826793ffdae925e328f7c1661e42d63a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/d/826793ffdae925e328f7c1661e42d63a
new file mode 100644
index 0000000000000000000000000000000000000000..2782bad31eb0b10df272c47952f6ef24c8ec71d6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/d/826793ffdae925e328f7c1661e42d63a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..798be9cd43aa29ab693c6f0097c0920846ad7a85
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/18d0f4800b210bf1f2453770b4a6ca16 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/18d0f4800b210bf1f2453770b4a6ca16
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/18d0f4800b210bf1f2453770b4a6ca16 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/204fb826812037945f92694fef62cf3f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/204fb826812037945f92694fef62cf3f
new file mode 100644
index 0000000000000000000000000000000000000000..e8fffcd0741d9bd7c2692c1194517fbae69987a7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/204fb826812037945f92694fef62cf3f differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/28b22295d5d0122576f306072c2433df b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/28b22295d5d0122576f306072c2433df
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/28b22295d5d0122576f306072c2433df differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/37a18c0c74ab3278491165c31380c1bd b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/37a18c0c74ab3278491165c31380c1bd
new file mode 100644
index 0000000000000000000000000000000000000000..8d376ee766d689b417780de4c01752603a67a076
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/37a18c0c74ab3278491165c31380c1bd
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d26780cd9f99d839b50c750a7cea90893fed32e6eb9206c03c8022d6bdab2bdf
+size 370832
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/3f0528384d0e2d5c631fa0d8cb07fedd b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/3f0528384d0e2d5c631fa0d8cb07fedd
new file mode 100644
index 0000000000000000000000000000000000000000..539d73fc2742d02ee589bd6372c9214acf398c6f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/3f0528384d0e2d5c631fa0d8cb07fedd
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b8f390dd5d52bf3b205855dab59eab23cf973556fd35eddf29503d7265ce9cb2
+size 296238
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/62101c15b5025f2b1269fac59798b624 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/62101c15b5025f2b1269fac59798b624
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/62101c15b5025f2b1269fac59798b624 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/eae4caeddbc838779ef59e72bbde51d8 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/eae4caeddbc838779ef59e72bbde51d8
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d/eae4caeddbc838779ef59e72bbde51d8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..90c1ed2bcabf56d47c989a075c8eb541f07958af
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..af3e930a4c23d5c6ca36cc5caddc2bf90f6c1125
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592854766451896, "commit_timestamp_nsecs": 1754592854949035013, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/d/b33fbe4e3e2806dbc81333c4ea30094c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/d/b33fbe4e3e2806dbc81333c4ea30094c
new file mode 100644
index 0000000000000000000000000000000000000000..85664d38494448c728af3c72328b9d50cd3f6928
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/d/b33fbe4e3e2806dbc81333c4ea30094c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3d1ae447c3908db3695a178607026ed489bc8a83
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/006cda8766381d2c2bbb9bf367532352 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/006cda8766381d2c2bbb9bf367532352
new file mode 100644
index 0000000000000000000000000000000000000000..3d33fa40655aa14a73686433377223c35c6de601
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/006cda8766381d2c2bbb9bf367532352 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/2fce6e5fd8c62a1e2d405b45ac9e6a04 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/2fce6e5fd8c62a1e2d405b45ac9e6a04
new file mode 100644
index 0000000000000000000000000000000000000000..85f35e7991cac7b6fc244bb2f277438119f4e90b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/2fce6e5fd8c62a1e2d405b45ac9e6a04 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/42e6be7be6fb77d162af4b54acef4936 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/42e6be7be6fb77d162af4b54acef4936
new file mode 100644
index 0000000000000000000000000000000000000000..6246a82a21162cfed692b25ab65d497913318e55
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/42e6be7be6fb77d162af4b54acef4936
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:382a06e51d83c4bb7922031e13e18181421456ce8e2bfa3f4ae1f2a83092f45b
+size 460366
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/62242f3be5a8a0f977b70eafdbbcc671 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/62242f3be5a8a0f977b70eafdbbcc671
new file mode 100644
index 0000000000000000000000000000000000000000..ba70cb866254bbbe631bdde69bb416f6f473017d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/62242f3be5a8a0f977b70eafdbbcc671
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f92b4a2c989a66bb3519ebfd73cadeddc2da299a6dbb0df07b036c56f9349650
+size 278992
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/7eb17def7094b27e6e19f4a466ba6da5 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/7eb17def7094b27e6e19f4a466ba6da5
new file mode 100644
index 0000000000000000000000000000000000000000..cd8da373ae279421827eaadeb0534e6225487315
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/7eb17def7094b27e6e19f4a466ba6da5 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/abae77108ced4e1a7c9c2d226f1801f3 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/abae77108ced4e1a7c9c2d226f1801f3
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/abae77108ced4e1a7c9c2d226f1801f3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/c606cd15fd40eaec1e7885101e51a555 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/c606cd15fd40eaec1e7885101e51a555
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/c606cd15fd40eaec1e7885101e51a555 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/ca547f749b28bf6c5475ce50d592a219 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/ca547f749b28bf6c5475ce50d592a219
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/ca547f749b28bf6c5475ce50d592a219 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/d1d78ad312d1018ad59958f8cfadbb20 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/d1d78ad312d1018ad59958f8cfadbb20
new file mode 100644
index 0000000000000000000000000000000000000000..ec745d97353f97209ef50f0eb738585830219f53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d/d1d78ad312d1018ad59958f8cfadbb20 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..bf11d3f92f73ee9ebb84808c20ef0d73b9e2838d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..7d788438b3cf4dcc5523d8787bcad4418aa696cc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592888059599036, "commit_timestamp_nsecs": 1754592888236603876, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/d/ebdef47415fbdcc98f89d122ad737d11 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/d/ebdef47415fbdcc98f89d122ad737d11
new file mode 100644
index 0000000000000000000000000000000000000000..1a6d06c79425771c617ff0428340b532b398172a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/d/ebdef47415fbdcc98f89d122ad737d11 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4b7a3a06813e010ae1a3023f6affb5cd71570d3a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/01834ef899ccafd3fd95b8cc60d04de8 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/01834ef899ccafd3fd95b8cc60d04de8
new file mode 100644
index 0000000000000000000000000000000000000000..5c234cff89ef7abc22ee8a5da7fc7ad7e5612fc5
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/01834ef899ccafd3fd95b8cc60d04de8
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:255e9428651af1ef918e67b369d184f0ccd75a55b3993f3f4bbf629a4f024bef
+size 279087
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/0280b8d99d96880cc328a1a1d35ef687 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/0280b8d99d96880cc328a1a1d35ef687
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/0280b8d99d96880cc328a1a1d35ef687 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/68ae0f3a8d91dc66d5d6b48f8782d973 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/68ae0f3a8d91dc66d5d6b48f8782d973
new file mode 100644
index 0000000000000000000000000000000000000000..2a31377ca030aced51ef70fb458c6b5c47404cd3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/68ae0f3a8d91dc66d5d6b48f8782d973 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/8454766b43f6528249295a3e1098c9a4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/8454766b43f6528249295a3e1098c9a4
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/8454766b43f6528249295a3e1098c9a4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/af84fe9183b46daf53b3f8e19f093bfb b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/af84fe9183b46daf53b3f8e19f093bfb
new file mode 100644
index 0000000000000000000000000000000000000000..2f62881c2d0173ee434d4d09bde7e33461d0052e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/af84fe9183b46daf53b3f8e19f093bfb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:eef1bf66e40eb74ed2aea59f19255962b879cdeb947c61e4e8e3bd240e8fa302
+size 426103
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/c52484c5ee452962dab9d31fba3b0504 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/c52484c5ee452962dab9d31fba3b0504
new file mode 100644
index 0000000000000000000000000000000000000000..b9bee1e1354b1bb1aa2dbd06851efc0ddb6ab8a3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/c52484c5ee452962dab9d31fba3b0504 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/d284dfba06306055d986759ce2b6e790 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/d284dfba06306055d986759ce2b6e790
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/d284dfba06306055d986759ce2b6e790 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/eacff6f77c919cfa8d65940d0836f371 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/eacff6f77c919cfa8d65940d0836f371
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d/eacff6f77c919cfa8d65940d0836f371 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..18bf73d61fe5458b06e04b2151df9524b43b5cf5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..6b8ce3c56bb1ca819ce8bb3fd6c69a8a0ed67542
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592921243345828, "commit_timestamp_nsecs": 1754592921385788963, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/d/9343b8c120db0e3bfdb08023b60d054d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/d/9343b8c120db0e3bfdb08023b60d054d
new file mode 100644
index 0000000000000000000000000000000000000000..8a2a03dad00c36d9b88c6f933b6b9a9f3687b3f7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/d/9343b8c120db0e3bfdb08023b60d054d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..6cc793ca6b97a53e21b7676e3fa1bece528015a5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/0a9a12e08db079b0bcd1893b29037172 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/0a9a12e08db079b0bcd1893b29037172
new file mode 100644
index 0000000000000000000000000000000000000000..3300d34ab74074a36b6fa12054aa0c6090336ea3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/0a9a12e08db079b0bcd1893b29037172 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/3ee739ddb2a1b29b4e45583605e91d4c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/3ee739ddb2a1b29b4e45583605e91d4c
new file mode 100644
index 0000000000000000000000000000000000000000..6d15fa4e91384059d41d0ebb26065ffd3bf94b4b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/3ee739ddb2a1b29b4e45583605e91d4c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c7c4c387374a6c23b65d163126ff37ff62d5d7f4c36379d952fbc7a06c3d9a12
+size 413829
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/435ab13fa157c47d18efa016a932715b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/435ab13fa157c47d18efa016a932715b
new file mode 100644
index 0000000000000000000000000000000000000000..99d4ea8a8a58bac1acf0168f4b3a0fb9a7efdc8e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/435ab13fa157c47d18efa016a932715b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/46eb30198a1e37a5a25e8bd5f08c765e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/46eb30198a1e37a5a25e8bd5f08c765e
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/46eb30198a1e37a5a25e8bd5f08c765e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/87f8d38f7908a419429f23b5343256d0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/87f8d38f7908a419429f23b5343256d0
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/87f8d38f7908a419429f23b5343256d0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/e898ba3ba100458d489642434efc704a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/e898ba3ba100458d489642434efc704a
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/e898ba3ba100458d489642434efc704a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/e925ec10f41f485f029d40f653ac3263 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/e925ec10f41f485f029d40f653ac3263
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/e925ec10f41f485f029d40f653ac3263 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/f3d280984005216109679f3cb670008e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/f3d280984005216109679f3cb670008e
new file mode 100644
index 0000000000000000000000000000000000000000..dae8aaa6dc9cd7d12d12def25293422d45d2f753
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d/f3d280984005216109679f3cb670008e
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7af4bc2df2cdbaf43a86cc11002790a4f0f407b1b71e59a32ed6afd8c2469bd8
+size 287590
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..6081cdb7c725f8fa0c7b0b83fa0d9317642775c3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..1257879110d7aa2501609d28b17714f575d977d8
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590193446729899, "commit_timestamp_nsecs": 1754590193621848994, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/d/61aec72ffcc48f2b6802167c8bbdf6ac b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/d/61aec72ffcc48f2b6802167c8bbdf6ac
new file mode 100644
index 0000000000000000000000000000000000000000..eab2329769f523964f619fbe5783f9d82b8d6e94
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/d/61aec72ffcc48f2b6802167c8bbdf6ac differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..92733e4e86568e4cf2c159932c2547cff072e211
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/33da9a7203ea68775bbc1d9efc268378 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/33da9a7203ea68775bbc1d9efc268378
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/33da9a7203ea68775bbc1d9efc268378 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/8616751ff3bcf2744fb9836f92a12c2d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/8616751ff3bcf2744fb9836f92a12c2d
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/8616751ff3bcf2744fb9836f92a12c2d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/a00ddb8d7d52bcceca78ee7436dd4048 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/a00ddb8d7d52bcceca78ee7436dd4048
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/a00ddb8d7d52bcceca78ee7436dd4048 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/aeba78c6859cb9d4f58a90b278081fcf b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/aeba78c6859cb9d4f58a90b278081fcf
new file mode 100644
index 0000000000000000000000000000000000000000..1c31904a1a61ef2dce43f7e41b37b028444eb5e7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/aeba78c6859cb9d4f58a90b278081fcf
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5e8ac0c86478d0d530e2717a3e78da1aa4fab153c26b715c33ba69518d3fb6b1
+size 536152
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/b35a0654a4acd85fc5dff31c7d4a8d35 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/b35a0654a4acd85fc5dff31c7d4a8d35
new file mode 100644
index 0000000000000000000000000000000000000000..704de1c182869264b19f29bcf9891cb3a4ded1fb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/b35a0654a4acd85fc5dff31c7d4a8d35 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/d20d6b8b08cda3ed07783f6018f5a6fa b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/d20d6b8b08cda3ed07783f6018f5a6fa
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/d20d6b8b08cda3ed07783f6018f5a6fa differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/d263679d2b9e4b6d1504cfbceedb35ec b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/d263679d2b9e4b6d1504cfbceedb35ec
new file mode 100644
index 0000000000000000000000000000000000000000..d049f5718ee1de39fcf11e54c765715e4a115b55
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/d263679d2b9e4b6d1504cfbceedb35ec
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b2a8a3fe316912000087fa829e7a5720da0db096201e0b06b0057248364ffd28
+size 202292
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/f60ba13ed47b02720c2fc877cb08d8d1 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/f60ba13ed47b02720c2fc877cb08d8d1
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d/f60ba13ed47b02720c2fc877cb08d8d1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..bda828cd338acfc9ccef333c8daff1ea042ae3b7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..297bd9844bb1ec112e367403423b985a509f8340
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592952327633689, "commit_timestamp_nsecs": 1754592952456796465, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/d/435e7b03b2cd4871d2fa5d6db75738e8 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/d/435e7b03b2cd4871d2fa5d6db75738e8
new file mode 100644
index 0000000000000000000000000000000000000000..2deff9a918fb1902b81563c36549f16f3256535a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/d/435e7b03b2cd4871d2fa5d6db75738e8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d9559acd46df72cb56728276bfcdea01a84b365f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/017cebb4184677b1f4a63cefc619031b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/017cebb4184677b1f4a63cefc619031b
new file mode 100644
index 0000000000000000000000000000000000000000..43cb4bbe3b0e0e3468d05e12f63ae21535a414d7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/017cebb4184677b1f4a63cefc619031b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/23d23a1e895550dca61547d08b9ec5ee b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/23d23a1e895550dca61547d08b9ec5ee
new file mode 100644
index 0000000000000000000000000000000000000000..0cbcda9d09ae89fa8f39b4b4aec31f99d7d2d7ba
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/23d23a1e895550dca61547d08b9ec5ee
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c70663d63f236267171fb62e79876e47226dabc994a8cd8b1c5666a7aa9e52fe
+size 698609
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/7cdb51d62bdf199a42e57d19296814f6 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/7cdb51d62bdf199a42e57d19296814f6
new file mode 100644
index 0000000000000000000000000000000000000000..59678bce24704adf117d2c64b8c284cdee2fdf56
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/7cdb51d62bdf199a42e57d19296814f6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/b3efed8fa041a6fd75601d9c39c6de1f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/b3efed8fa041a6fd75601d9c39c6de1f
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/b3efed8fa041a6fd75601d9c39c6de1f differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/d80397845034ffa749a579405e87d4ba b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/d80397845034ffa749a579405e87d4ba
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/d80397845034ffa749a579405e87d4ba differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/def9d1c4c9ec4cedfa3bddbf220e0011 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/def9d1c4c9ec4cedfa3bddbf220e0011
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d/def9d1c4c9ec4cedfa3bddbf220e0011 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..ea0c1a2aeacce8a1fa560d7a6e2d9990e4b087c0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..0a66ffdd6458143cb3650fc29bbdd6d29e86080f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754592985566099093, "commit_timestamp_nsecs": 1754592985737541584, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/d/a2b0e84f651a4e9f79d2ce4db87f03aa b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/d/a2b0e84f651a4e9f79d2ce4db87f03aa
new file mode 100644
index 0000000000000000000000000000000000000000..71ae6f164333d927563d432a7ec2f12bef915264
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/d/a2b0e84f651a4e9f79d2ce4db87f03aa differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3cfd7366d430288d22ec74022367fead5e293ac3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/198102d24561ec8b8dc321451797f3e8 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/198102d24561ec8b8dc321451797f3e8
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/198102d24561ec8b8dc321451797f3e8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/512e0d2480713d4df17bb3bbc9a4fe51 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/512e0d2480713d4df17bb3bbc9a4fe51
new file mode 100644
index 0000000000000000000000000000000000000000..6423406628650ee7e48b28c3ee42e26762871568
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/512e0d2480713d4df17bb3bbc9a4fe51
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b49d5166d487d3fe79aac596ca5b5942dbf9e401dabbe77997cddedeaba7572d
+size 261896
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/65f1d2678bf26d64a3965326d8cbe944 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/65f1d2678bf26d64a3965326d8cbe944
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/65f1d2678bf26d64a3965326d8cbe944 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/725a72f6dbaa2262a4d55ea3d2d0781a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/725a72f6dbaa2262a4d55ea3d2d0781a
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/725a72f6dbaa2262a4d55ea3d2d0781a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/781e5b02051139947e2100cd0ca93d96 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/781e5b02051139947e2100cd0ca93d96
new file mode 100644
index 0000000000000000000000000000000000000000..dfbfe1a61305bab6f716f83c77874d9e3176999d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/781e5b02051139947e2100cd0ca93d96 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/8faf62cb7f4a628acf1a4831c813f9ae b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/8faf62cb7f4a628acf1a4831c813f9ae
new file mode 100644
index 0000000000000000000000000000000000000000..9ee5fce7a9b33a3925380300a15f580d9e661fd3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/8faf62cb7f4a628acf1a4831c813f9ae
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:62589f7eed4d9bed9ad3ae6312365452bec2ea03b3221de07fa8c2722a74a084
+size 477462
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/93501bba8a72697ae79866eb2ba00fb7 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/93501bba8a72697ae79866eb2ba00fb7
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/93501bba8a72697ae79866eb2ba00fb7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/c83d4c2c0518157f4a9664b7af518962 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/c83d4c2c0518157f4a9664b7af518962
new file mode 100644
index 0000000000000000000000000000000000000000..ecaad409fd849801437c391ec9b670730da3d8f9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/c83d4c2c0518157f4a9664b7af518962 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/cefdc5b4d8474187669ebb5cb04cd368 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/cefdc5b4d8474187669ebb5cb04cd368
new file mode 100644
index 0000000000000000000000000000000000000000..ec745d97353f97209ef50f0eb738585830219f53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d/cefdc5b4d8474187669ebb5cb04cd368 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..159a2c811b55e1dd2942bbb9530f9aa9a407e317
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..e28156eac4af2c58009162fe1729823400b28e21
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754593019566136417, "commit_timestamp_nsecs": 1754593019737739351, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/d/13d4f635daff049ec59e501afc1a49d1 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/d/13d4f635daff049ec59e501afc1a49d1
new file mode 100644
index 0000000000000000000000000000000000000000..766a521813f6e082bc18adf7876ba050c3278268
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/d/13d4f635daff049ec59e501afc1a49d1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..de06c63b9e559c205ba573d86d8a10226eee2519
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/0059a07c2a959e7ed02ebc0325514601 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/0059a07c2a959e7ed02ebc0325514601
new file mode 100644
index 0000000000000000000000000000000000000000..5d16bc0c9a4c0e4f6cd462c5fcdd67fffea54bff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/0059a07c2a959e7ed02ebc0325514601 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/05d4400318074a8f65339b10d546924d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/05d4400318074a8f65339b10d546924d
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/05d4400318074a8f65339b10d546924d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/2285229efee70f7f1651b902d7740610 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/2285229efee70f7f1651b902d7740610
new file mode 100644
index 0000000000000000000000000000000000000000..8b2eb9617bdc5b79c6747024810e11a184482434
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/2285229efee70f7f1651b902d7740610 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/463001d12a798a15a391141350631ce8 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/463001d12a798a15a391141350631ce8
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/463001d12a798a15a391141350631ce8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/5d422efe68678d488cd4c8f6af48f53f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/5d422efe68678d488cd4c8f6af48f53f
new file mode 100644
index 0000000000000000000000000000000000000000..b75656d6dae0972974f676a02c0852e6347c2379
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/5d422efe68678d488cd4c8f6af48f53f differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/79adb5e32f7628bb98a3c81b96316d0b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/79adb5e32f7628bb98a3c81b96316d0b
new file mode 100644
index 0000000000000000000000000000000000000000..c5771f16e30d8a8b3548b0620e55790f3a2e4a57
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/79adb5e32f7628bb98a3c81b96316d0b
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:98e0a55339ea08ec37e6bc6db87085b2c3d40df7e60246392530c7046497ed5a
+size 450669
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/82596a3412d7f37e3123c1a5a389111c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/82596a3412d7f37e3123c1a5a389111c
new file mode 100644
index 0000000000000000000000000000000000000000..5af9d6861be946a6963b562025c8a0cafc1ff704
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/82596a3412d7f37e3123c1a5a389111c
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:716623caddd718e0778b97f39605f26449f1b245fae948172d68db6ad21dee49
+size 286541
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/a283a1e2dbc8269f83797142fce4ce6f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/a283a1e2dbc8269f83797142fce4ce6f
new file mode 100644
index 0000000000000000000000000000000000000000..dfbfe1a61305bab6f716f83c77874d9e3176999d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/a283a1e2dbc8269f83797142fce4ce6f differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/f5066bedf98e4cfcbdd2922d94af72fb b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/f5066bedf98e4cfcbdd2922d94af72fb
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d/f5066bedf98e4cfcbdd2922d94af72fb differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4d6b76595d32205573567b407a144655294b1110
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..dabf20bb8e7b82f256a47be8ed7dcbb1baca2d8a
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754593052774172476, "commit_timestamp_nsecs": 1754593052958660903, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/d/078530b2b1bf380f886bc0b61219f0c3 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/d/078530b2b1bf380f886bc0b61219f0c3
new file mode 100644
index 0000000000000000000000000000000000000000..1f43a3b2c8efd31a92206ccc35cba22c710436a5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/d/078530b2b1bf380f886bc0b61219f0c3 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..f90246b9db57c03d1cb348258e0e1c7538a278f8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/189079a229bd2f7af98c739bdb8a69ac b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/189079a229bd2f7af98c739bdb8a69ac
new file mode 100644
index 0000000000000000000000000000000000000000..715a0110b094b0c608b9b6b0efb45fca93994673
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/189079a229bd2f7af98c739bdb8a69ac
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bfd2b6704655c5f35af5d231c25adca268aa00d6ce52733db804a719115448cb
+size 211015
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/2316646e8fd538c6d686e96c6bb55f2e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/2316646e8fd538c6d686e96c6bb55f2e
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/2316646e8fd538c6d686e96c6bb55f2e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/28025f57dba5e36962eee256db75a136 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/28025f57dba5e36962eee256db75a136
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/28025f57dba5e36962eee256db75a136 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/47fccba575c7eac420f8daa3c667e244 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/47fccba575c7eac420f8daa3c667e244
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/47fccba575c7eac420f8daa3c667e244 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/54e87626e3dc373781e4576985cda789 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/54e87626e3dc373781e4576985cda789
new file mode 100644
index 0000000000000000000000000000000000000000..f3e837613c5408d3123a5f48ffd083133266c383
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/54e87626e3dc373781e4576985cda789
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:524ee903131bef84c498f4bf7298d22d69a10a1d1a005079bb25a8a41047209e
+size 528420
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/7ce31630c098ddb61aed556dca1df1cb b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/7ce31630c098ddb61aed556dca1df1cb
new file mode 100644
index 0000000000000000000000000000000000000000..a2c5dc307f942cee8a29f0a0be8ffab0ad546bf3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/7ce31630c098ddb61aed556dca1df1cb differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/da8f53ecee90f055b211d90070093e27 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/da8f53ecee90f055b211d90070093e27
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/da8f53ecee90f055b211d90070093e27 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/dee94858b9c7a744cfccc4d63a4a0cb6 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/dee94858b9c7a744cfccc4d63a4a0cb6
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d/dee94858b9c7a744cfccc4d63a4a0cb6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..cf43377a9fae58221df3463088b41fd1c1cd0337
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..5813cc118ee84a6d2b581c5076156e98cba75767
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754593086119376344, "commit_timestamp_nsecs": 1754593086269521100, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/d/6f222167a57d6b4da3b977f314609ac1 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/d/6f222167a57d6b4da3b977f314609ac1
new file mode 100644
index 0000000000000000000000000000000000000000..53f398578fed25584d2d79ce5d7d63d856e231da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/d/6f222167a57d6b4da3b977f314609ac1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..6cb12eca5d5c70b91358ac3c06dc1197f045579e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/3f5b817e2162a0bd2a4b5142cc962da9 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/3f5b817e2162a0bd2a4b5142cc962da9
new file mode 100644
index 0000000000000000000000000000000000000000..65529f6532287fad4b450ca493011b55e8bf7f92
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/3f5b817e2162a0bd2a4b5142cc962da9 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/4420097f4c9a677da8b3da8ef08fc13c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/4420097f4c9a677da8b3da8ef08fc13c
new file mode 100644
index 0000000000000000000000000000000000000000..f20ae3cdb4f9e129c5e646491b40c9ccdc084729
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/4420097f4c9a677da8b3da8ef08fc13c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/4c9d7d594d7f6191af2eb1c2e59586b0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/4c9d7d594d7f6191af2eb1c2e59586b0
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/4c9d7d594d7f6191af2eb1c2e59586b0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/76154ac859f612b63547dd250d0cd7f5 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/76154ac859f612b63547dd250d0cd7f5
new file mode 100644
index 0000000000000000000000000000000000000000..be2fdd571ddb0102cb9aa82741c34f52ff77c3a7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/76154ac859f612b63547dd250d0cd7f5
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5278ab38c41acaf094ad60bb14695c46355319d1265052d35023fbae4990caa3
+size 296210
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/985925908c958e4e0f5403f382451022 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/985925908c958e4e0f5403f382451022
new file mode 100644
index 0000000000000000000000000000000000000000..8a082af3235e6b228d8b63e4d16488fc09b6d6fe
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/985925908c958e4e0f5403f382451022
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:15a01f2809c092603a401073248891ac6150f5ed3ff0ebb808c1f3d6dd4b0e14
+size 405189
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/ab1f36aef92a2122eba4ca814d84ec9b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/ab1f36aef92a2122eba4ca814d84ec9b
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/ab1f36aef92a2122eba4ca814d84ec9b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/fb7050f939666fa8c960352b6e6b02e6 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/fb7050f939666fa8c960352b6e6b02e6
new file mode 100644
index 0000000000000000000000000000000000000000..be63c579ca8e81506a0fac0e7857645887e1e658
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d/fb7050f939666fa8c960352b6e6b02e6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..0e5d2ce76c685ae3e4d741e0e7bd05696da78566
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..922448127884fdb5fe0cd55b3c076b1bee5236c8
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754593119263333180, "commit_timestamp_nsecs": 1754593119435062252, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/d/2911a3164b724a698ada598fd407b6a2 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/d/2911a3164b724a698ada598fd407b6a2
new file mode 100644
index 0000000000000000000000000000000000000000..e0962d4a2d70b887621f002f058dd6f2da39b30b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/d/2911a3164b724a698ada598fd407b6a2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..e562db33170f9d01ae5f7daf47d51ec56affc3dd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/534876a8bda2f55ec95b870e86ea04e1 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/534876a8bda2f55ec95b870e86ea04e1
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/534876a8bda2f55ec95b870e86ea04e1 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/81e033a3623f932f114fd9cc23446837 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/81e033a3623f932f114fd9cc23446837
new file mode 100644
index 0000000000000000000000000000000000000000..7446e85bc04b482b7f61b0a222e7af9c6ae4d8b8
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/81e033a3623f932f114fd9cc23446837
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d03d5e2be3de22c34c4bb7b6694cfb578a888dd6ca9d4182df26b6525bee371c
+size 228498
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/a12b7c8dee879e7c09faa6314cb0045b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/a12b7c8dee879e7c09faa6314cb0045b
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/a12b7c8dee879e7c09faa6314cb0045b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/bed0eb28768662739a2ea86e4ffd773d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/bed0eb28768662739a2ea86e4ffd773d
new file mode 100644
index 0000000000000000000000000000000000000000..64781dba5613c6eae6e6e9d66b55c7557cdaa467
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/bed0eb28768662739a2ea86e4ffd773d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/d00cb274a39eb47104f57d9ef8862dfa b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/d00cb274a39eb47104f57d9ef8862dfa
new file mode 100644
index 0000000000000000000000000000000000000000..2e6505347ec3dd14294cf3ba4900781cdebf72a0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d/d00cb274a39eb47104f57d9ef8862dfa
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f27e70b63936ae91a06b5b08821777cce64d6fb82eabb94cf7d8f3052b8f7a89
+size 509084
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..05db3a4d93889e471adb0422efba20cdc363a583
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..fed6363cc8bd0703d64d9081a88d8bf4e4ebd991
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754589928982784773, "commit_timestamp_nsecs": 1754589929158284615, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/d/2cc541178196fe594d2e3c1cedb10ec7 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/d/2cc541178196fe594d2e3c1cedb10ec7
new file mode 100644
index 0000000000000000000000000000000000000000..c2c020fad38180515e3bdf015b4615b46b499578
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/d/2cc541178196fe594d2e3c1cedb10ec7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b9e89a6790897a9c1c066851af5d714e12aeff42
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/1fc1b3ea19589d7b0939c49c96e2e79e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/1fc1b3ea19589d7b0939c49c96e2e79e
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/1fc1b3ea19589d7b0939c49c96e2e79e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/7fc3eaf0e480ebcf2bf960ac050f8c3c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/7fc3eaf0e480ebcf2bf960ac050f8c3c
new file mode 100644
index 0000000000000000000000000000000000000000..dfbfe1a61305bab6f716f83c77874d9e3176999d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/7fc3eaf0e480ebcf2bf960ac050f8c3c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/9f4bf671f0e275762cdc2b56f2f9690a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/9f4bf671f0e275762cdc2b56f2f9690a
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/9f4bf671f0e275762cdc2b56f2f9690a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/ae6475a569be62c236a65d8abb006884 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/ae6475a569be62c236a65d8abb006884
new file mode 100644
index 0000000000000000000000000000000000000000..6d9ed628c51dd7241a2c96b02218efaf9f1307b2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/ae6475a569be62c236a65d8abb006884
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3ef75a1be1672972f2cdee6a62ea0186219ee03c111a1a5cc1ab2094fa65ab3c
+size 286827
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/b27f2016691f821a087cd66a5ff60284 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/b27f2016691f821a087cd66a5ff60284
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/b27f2016691f821a087cd66a5ff60284 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/bcbcb51634b513b8adb23639fc13e398 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/bcbcb51634b513b8adb23639fc13e398
new file mode 100644
index 0000000000000000000000000000000000000000..2e21bfce171b505dc9cec891bece53c34b869eaf
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/bcbcb51634b513b8adb23639fc13e398
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:efb7a6dd7d8ccacd00e4d32114a1e8472201aa38096e974cec0d6af20b1f1662
+size 450687
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/c36637bae83829245136654388feb523 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/c36637bae83829245136654388feb523
new file mode 100644
index 0000000000000000000000000000000000000000..cd8da373ae279421827eaadeb0534e6225487315
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/c36637bae83829245136654388feb523 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/de49021421b964348073babb46abdde4 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/de49021421b964348073babb46abdde4
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/de49021421b964348073babb46abdde4 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/e62d2e00d8028c3007a87c2dff9187bf b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/e62d2e00d8028c3007a87c2dff9187bf
new file mode 100644
index 0000000000000000000000000000000000000000..05c425257a6b78a750c3450453006cc74aa3e11c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d/e62d2e00d8028c3007a87c2dff9187bf differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..cdc821494f48ee8c3eebef03be883d13e701fd53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..2ce459d1080dde061ab6bcc2f7801cdec3ca01f9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590225845601337, "commit_timestamp_nsecs": 1754590226018800669, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/d/faf1032da0d7ecd707177624022946dc b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/d/faf1032da0d7ecd707177624022946dc
new file mode 100644
index 0000000000000000000000000000000000000000..513bc9ed626508adce98059d91f3fb49f7285190
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/d/faf1032da0d7ecd707177624022946dc differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..4e50d3c65f2152fae5a3ae33c9e038877ff11395
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/10647960aef0ed0acb77d17e9c2f854f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/10647960aef0ed0acb77d17e9c2f854f
new file mode 100644
index 0000000000000000000000000000000000000000..bdfd46ab48ef76154acd6c529532198e57e56d79
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/10647960aef0ed0acb77d17e9c2f854f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5ac344c9d7ae0e8d4029611c6f7cbb4f42da3c41fb336bf1d5ca9194aef01ce1
+size 287316
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/5508a8c0685b6070ba063402dab17fd6 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/5508a8c0685b6070ba063402dab17fd6
new file mode 100644
index 0000000000000000000000000000000000000000..c4f628def0ddee679c84e5d46aee24dd97aea9b8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/5508a8c0685b6070ba063402dab17fd6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/87148ed98d4012b93fdb9066cd534c2b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/87148ed98d4012b93fdb9066cd534c2b
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/87148ed98d4012b93fdb9066cd534c2b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/9a515210dd28b84f0ea105ff464dc693 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/9a515210dd28b84f0ea105ff464dc693
new file mode 100644
index 0000000000000000000000000000000000000000..4174c2c32af293b5784c168794bbe2c9341ee14a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/9a515210dd28b84f0ea105ff464dc693 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/ce10e14c05c0830236ddc5793919fe09 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/ce10e14c05c0830236ddc5793919fe09
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/ce10e14c05c0830236ddc5793919fe09 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/da1475cbd237c79421fb3ab3bd8f43ba b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/da1475cbd237c79421fb3ab3bd8f43ba
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/da1475cbd237c79421fb3ab3bd8f43ba differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/ff944bd4fb5b0801374030f18f84dfd0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/ff944bd4fb5b0801374030f18f84dfd0
new file mode 100644
index 0000000000000000000000000000000000000000..69a1015cdb28b1528dd37c649dfd3254ecb1c59f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d/ff944bd4fb5b0801374030f18f84dfd0
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7b4817e67c8164b9b9ab34afab801d5d26412f12fbaaa4716d00a7897c46a1d0
+size 450631
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..b926aef1c4957a1baff03a06ae798befec48c6bb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..6ddc24098beb1a8bfd985ff77348900e8212911f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590259083238215, "commit_timestamp_nsecs": 1754590259259997376, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/d/f3a71241316bad401d8ab5d1d7d54d47 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/d/f3a71241316bad401d8ab5d1d7d54d47
new file mode 100644
index 0000000000000000000000000000000000000000..f3beb9e1cbbfef0fb20cf56b8b92fb2957bd9681
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/d/f3a71241316bad401d8ab5d1d7d54d47 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..c24e911a042ba95568546770eba792253e6f8205
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/135e5f93885fdbdfa5161fa027eac82e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/135e5f93885fdbdfa5161fa027eac82e
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/135e5f93885fdbdfa5161fa027eac82e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/274473074d5dc787142285762593117d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/274473074d5dc787142285762593117d
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/274473074d5dc787142285762593117d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/38c48b8d9a304b0e0b64633b5e3004f6 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/38c48b8d9a304b0e0b64633b5e3004f6
new file mode 100644
index 0000000000000000000000000000000000000000..7db4ab52f3897728aa734d0639e7936d1537f5ee
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/38c48b8d9a304b0e0b64633b5e3004f6 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/5c98674185c32f4f6ba52e95c24f6b7d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/5c98674185c32f4f6ba52e95c24f6b7d
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/5c98674185c32f4f6ba52e95c24f6b7d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/a3600783e5970dd07039414f0dea70d7 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/a3600783e5970dd07039414f0dea70d7
new file mode 100644
index 0000000000000000000000000000000000000000..62404a52c8f81bc98a817ad67b9ede251e1e378e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/a3600783e5970dd07039414f0dea70d7
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dd562e3c52cc02927151476172e4bac00e78ba3e65ed5b67f1a94155bb136f60
+size 261863
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/bc732652adfa7fe4b0b16b05a7ecf5b1 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/bc732652adfa7fe4b0b16b05a7ecf5b1
new file mode 100644
index 0000000000000000000000000000000000000000..20423997f70c67eb83fe848032f43164b7d7a06b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/bc732652adfa7fe4b0b16b05a7ecf5b1
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:39b480edf21d908b39b1c1e5fffc1c7d091ee28f2e25b8a9d48f2499af9d73db
+size 476475
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/c19a59ae8053efdb5ed5af52361a4288 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/c19a59ae8053efdb5ed5af52361a4288
new file mode 100644
index 0000000000000000000000000000000000000000..31edab7adafce8eadd3bab550e17f2fefe239995
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/c19a59ae8053efdb5ed5af52361a4288 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/f14d2389fdeb5c4986a6dcd1cb20f67e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/f14d2389fdeb5c4986a6dcd1cb20f67e
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d/f14d2389fdeb5c4986a6dcd1cb20f67e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3dd9eb4db4490ee1155e01a38cc7d6cd8f2e50fd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..44fe953659634db8853d003270f87bfd805d0a57
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590291582440211, "commit_timestamp_nsecs": 1754590291745288657, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/d/c43f762f6df20dcf53a4fce303fea130 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/d/c43f762f6df20dcf53a4fce303fea130
new file mode 100644
index 0000000000000000000000000000000000000000..ed6822e4213e3778950d4e0c6082aae649713cea
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/d/c43f762f6df20dcf53a4fce303fea130 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..03399383a9b70edbd394dbb22331441e8697fc5b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/2794fee000f423cc07cb36dedd5b3073 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/2794fee000f423cc07cb36dedd5b3073
new file mode 100644
index 0000000000000000000000000000000000000000..ec745d97353f97209ef50f0eb738585830219f53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/2794fee000f423cc07cb36dedd5b3073 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/294492af37123877e8cd291154459b9a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/294492af37123877e8cd291154459b9a
new file mode 100644
index 0000000000000000000000000000000000000000..382e607994aceae536e823af5f9df6607c7241e6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/294492af37123877e8cd291154459b9a
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5f11845b8eed6e32e8ce338f68e97216537c40880461e7f8db75b49cba0ea239
+size 507309
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/4cfeffb674b67e03f7699a804e73c6d0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/4cfeffb674b67e03f7699a804e73c6d0
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/4cfeffb674b67e03f7699a804e73c6d0 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/4da768d99fff84cb2f2b44117b6d7f81 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/4da768d99fff84cb2f2b44117b6d7f81
new file mode 100644
index 0000000000000000000000000000000000000000..f8b930b244b654a04cfee129d85e4adfa5d0cebe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/4da768d99fff84cb2f2b44117b6d7f81 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/524217ff5d87d331d46ad5e3d28e7a53 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/524217ff5d87d331d46ad5e3d28e7a53
new file mode 100644
index 0000000000000000000000000000000000000000..b520324392b2c8b9c80cd168ff66501d2f8e2266
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/524217ff5d87d331d46ad5e3d28e7a53 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/7ede202e02cde09f0ff8d9d3ecc38e97 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/7ede202e02cde09f0ff8d9d3ecc38e97
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/7ede202e02cde09f0ff8d9d3ecc38e97 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/854b42df6e483e1120fbba2ad1b3777b b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/854b42df6e483e1120fbba2ad1b3777b
new file mode 100644
index 0000000000000000000000000000000000000000..c3b0ef68bd0b900f315925516dcfcdf8b69fc9a2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/854b42df6e483e1120fbba2ad1b3777b differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/9a088fba815d87dd71137d3fb1bf3e49 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/9a088fba815d87dd71137d3fb1bf3e49
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/9a088fba815d87dd71137d3fb1bf3e49 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/eb74b21e5801c5630ce3a406e02734c7 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/eb74b21e5801c5630ce3a406e02734c7
new file mode 100644
index 0000000000000000000000000000000000000000..512098ec9db0ddaf9965191eb60d2b0525b74bf6
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d/eb74b21e5801c5630ce3a406e02734c7
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fafae3ae034b1e476add7d392cf755432154c72a78ed4aaea2f582cfca98a70c
+size 193692
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..43b41b6a9acdae296ccf97284873fd44b55e1d31
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..4842c2eac5e9306b233b249cefdd9b534e1169fb
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590324456581058, "commit_timestamp_nsecs": 1754590324648553175, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/d/80b2d39e8448bbd97a17f0e4536beadf b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/d/80b2d39e8448bbd97a17f0e4536beadf
new file mode 100644
index 0000000000000000000000000000000000000000..537b6bdbc366ab7e064746b386f7172f0f4d8253
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/d/80b2d39e8448bbd97a17f0e4536beadf differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3d03de0f1d1ee50a6889bfdfe7039adacb6922f3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/2d0470a2e026e56acd21e50c7fe6d8c7 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/2d0470a2e026e56acd21e50c7fe6d8c7
new file mode 100644
index 0000000000000000000000000000000000000000..53a0966c87e8c471db99b50ee8789576653eb3f2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/2d0470a2e026e56acd21e50c7fe6d8c7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/370c72fe7e0eaaaa13811c2c64af9d76 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/370c72fe7e0eaaaa13811c2c64af9d76
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/370c72fe7e0eaaaa13811c2c64af9d76 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/3c9bf0eb6fcc0eb1c0d1eba4ce0f3a2c b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/3c9bf0eb6fcc0eb1c0d1eba4ce0f3a2c
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/3c9bf0eb6fcc0eb1c0d1eba4ce0f3a2c differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/4a304474886f8cd31e2a65d13fd2fe52 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/4a304474886f8cd31e2a65d13fd2fe52
new file mode 100644
index 0000000000000000000000000000000000000000..9cbe65274203c53b3d350085926aa4479ba41c93
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/4a304474886f8cd31e2a65d13fd2fe52 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/7b01a3f7da3c0b9ece177ea865e09d24 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/7b01a3f7da3c0b9ece177ea865e09d24
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/7b01a3f7da3c0b9ece177ea865e09d24 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/967a4c318ba199564369c2c0d7af3b31 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/967a4c318ba199564369c2c0d7af3b31
new file mode 100644
index 0000000000000000000000000000000000000000..4bc1b6d370942075c3783fbef81037e53b151a84
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/967a4c318ba199564369c2c0d7af3b31 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/a15bd7fda036ecd345ceecbf441af41f b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/a15bd7fda036ecd345ceecbf441af41f
new file mode 100644
index 0000000000000000000000000000000000000000..3d766b74a4cbe15a629e6125681411098a94ec53
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/a15bd7fda036ecd345ceecbf441af41f
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5912c5cd51d7da4ea95d66b1ab9426fdbb0b997fff2040a432bd0c91cc72b270
+size 442761
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/f2eda4c1918af53d7177debcc6450042 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/f2eda4c1918af53d7177debcc6450042
new file mode 100644
index 0000000000000000000000000000000000000000..31b2c33f150a7de0473c015f5453895ae85dd702
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d/f2eda4c1918af53d7177debcc6450042
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1039f77b6c93e6f140cd0f19262b3a3fa648d5cea5957e4d19ac615614f6c59a
+size 228037
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..85cf9b65b4de7a3c10486cb9fbb44ed69cf3c952
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..22fdc474f03051044cdc314fffac7f06c9d0eb4c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590357550866830, "commit_timestamp_nsecs": 1754590357729335503, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/d/090ce538e71a680955273ee7d43cb98a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/d/090ce538e71a680955273ee7d43cb98a
new file mode 100644
index 0000000000000000000000000000000000000000..fc4159ac2ca08d67d0bcf6eb734f780c191a16ac
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/d/090ce538e71a680955273ee7d43cb98a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..3736fc41c0df17a94d912920527f9496d302de14
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/0121a9692432b817e65cc6a8364834d2 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/0121a9692432b817e65cc6a8364834d2
new file mode 100644
index 0000000000000000000000000000000000000000..023b1c12d22a0544118734c9da4dd3fce85ab900
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/0121a9692432b817e65cc6a8364834d2 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/2e3998bbd96f29a5348c211f10b37219 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/2e3998bbd96f29a5348c211f10b37219
new file mode 100644
index 0000000000000000000000000000000000000000..1de938618c1896607335da99867b799cacbe0d1b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/2e3998bbd96f29a5348c211f10b37219 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/88c00eaa68751a1db4883553ede59f82 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/88c00eaa68751a1db4883553ede59f82
new file mode 100644
index 0000000000000000000000000000000000000000..80b9190fcd832ed4d881dcac6dc145c12b6b194e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/88c00eaa68751a1db4883553ede59f82
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:41b045b2ef9bcd4eba0ee12687bfb602dd8904c460fcfeedfb16305dba58bbfb
+size 295922
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/8cddad395257f9e228ba3978fc0ab12e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/8cddad395257f9e228ba3978fc0ab12e
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/8cddad395257f9e228ba3978fc0ab12e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/bb5cac862900423d5a0d601268723ed8 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/bb5cac862900423d5a0d601268723ed8
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/bb5cac862900423d5a0d601268723ed8 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/e64be2fb800d9dc656e4ddf244304328 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/e64be2fb800d9dc656e4ddf244304328
new file mode 100644
index 0000000000000000000000000000000000000000..884c88841ff68ac674c7857adaf47821ec362fd0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/e64be2fb800d9dc656e4ddf244304328 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/eb771b477137704a3b2600511364ef0d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/eb771b477137704a3b2600511364ef0d
new file mode 100644
index 0000000000000000000000000000000000000000..31a4905d6c0b049d180d1748421be93a938ed075
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d/eb771b477137704a3b2600511364ef0d
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dd7715cd7921b59fe46819992a25bdee044660340fce52656ae5618b0432f260
+size 404369
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..a0a12ec1da85385139293ca48ea9ce718f8ddc2e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/_CHECKPOINT_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/_CHECKPOINT_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..6ad133fded4a36507b86a00f305b8217ccba449c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/_CHECKPOINT_METADATA
@@ -0,0 +1 @@
+{"item_handlers": {"default": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1754590390372528982, "commit_timestamp_nsecs": 1754590390546601284, "custom_metadata": {}}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/_METADATA b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/_METADATA
new file mode 100644
index 0000000000000000000000000000000000000000..658ce953ac9a79874494988549b04417d7fa49e2
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/_METADATA
@@ -0,0 +1 @@
+{"tree_metadata": {"('params', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [260, 256]}}, "('params', 'Dense_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('params', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 4]}}, "('params', 'Dense_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [2]}}, "('params', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256, 2]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 3, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_1', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_10', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_10", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_11', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_11", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_12', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_12", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_13', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_13", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_14', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_14", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_2', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_3', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_3", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_4', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_4", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 16]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_5', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_5", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 16, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_6', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_6", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_7', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_7", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_8', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_8", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [32]}}, "('params', '_ImpalaLargeCNN_0', 'Conv_9', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Conv_9", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 32, 32]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'bias')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('params', '_ImpalaLargeCNN_0', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "params", "key_type": 2}, {"key": "_ImpalaLargeCNN_0", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128, 256]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/_sharding b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/_sharding
new file mode 100644
index 0000000000000000000000000000000000000000..5e57170164a3cb57218f53be940d64934e1aa552
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/_sharding
@@ -0,0 +1 @@
+{"cGFyYW1zLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMTQua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfMy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfNy5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkNvbnZfOS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cGFyYW1zLl9JbXBhbGFMYXJnZUNOTl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/array_metadatas/process_0 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/array_metadatas/process_0
new file mode 100644
index 0000000000000000000000000000000000000000..294fdeb239f359959c933697f1366e648fe33ebc
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/array_metadatas/process_0
@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "params.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_0.kernel", "write_shape": [260, 256], "chunk_shape": [260, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_1.kernel", "write_shape": [256, 4], "chunk_shape": [256, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.bias", "write_shape": [2], "chunk_shape": [2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params.Dense_2.kernel", "write_shape": [256, 2], "chunk_shape": [256, 2], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_0.kernel", "write_shape": [3, 3, 3, 16], "chunk_shape": [3, 3, 3, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_1.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_10.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_11.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_12.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_13.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_14.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_2.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_3.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.bias", "write_shape": [16], "chunk_shape": [16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_4.kernel", "write_shape": [3, 3, 16, 16], "chunk_shape": [3, 3, 16, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_5.kernel", "write_shape": [3, 3, 16, 32], "chunk_shape": [3, 3, 16, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_6.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_7.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_8.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.bias", "write_shape": [32], "chunk_shape": [32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Conv_9.kernel", "write_shape": [3, 3, 32, 32], "chunk_shape": [3, 3, 32, 32], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "params._ImpalaLargeCNN_0.Dense_0.kernel", "write_shape": [128, 256], "chunk_shape": [128, 256], "ext_metadata": null}}]}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/d/3331ba9e2d54e835b4857bbd354de02a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/d/3331ba9e2d54e835b4857bbd354de02a
new file mode 100644
index 0000000000000000000000000000000000000000..e58c7b4a3abf764c1f33305bb4608ec1ecdb81f0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/d/3331ba9e2d54e835b4857bbd354de02a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..d266d813234d71f00ed54c346e78b695b5c1b2a0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/3433b74725124c34e99e0b73832ad2b7 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/3433b74725124c34e99e0b73832ad2b7
new file mode 100644
index 0000000000000000000000000000000000000000..9761834563dcb10d6d366a86afbf725807c2d44b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/3433b74725124c34e99e0b73832ad2b7 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/7473c90475e3e51b27fd5572a5e9551d b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/7473c90475e3e51b27fd5572a5e9551d
new file mode 100644
index 0000000000000000000000000000000000000000..7e6a44c651e0d86e16d0cbfd0b8f268534103ab2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/7473c90475e3e51b27fd5572a5e9551d differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/89d800326b8c5af808b266f9cebb7908 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/89d800326b8c5af808b266f9cebb7908
new file mode 100644
index 0000000000000000000000000000000000000000..60545bb1e6cdd10abe6b0fade248e62cc5d1424a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/89d800326b8c5af808b266f9cebb7908 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/a83194b17cc55f752f0f4efb2ac3f072 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/a83194b17cc55f752f0f4efb2ac3f072
new file mode 100644
index 0000000000000000000000000000000000000000..0d709becb697d208e87de96285e2025652a43e0e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/a83194b17cc55f752f0f4efb2ac3f072
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fac435ef4bd7982b44cefdaeab3d0033345b997b1f21ed944569a68e66163163
+size 159772
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/b46b44a5177c19dafef518e93266830e b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/b46b44a5177c19dafef518e93266830e
new file mode 100644
index 0000000000000000000000000000000000000000..dfbfe1a61305bab6f716f83c77874d9e3176999d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/b46b44a5177c19dafef518e93266830e differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/ce1acc06847f9aa6db57e77eeaabcc8a b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/ce1acc06847f9aa6db57e77eeaabcc8a
new file mode 100644
index 0000000000000000000000000000000000000000..7dd203da35a42f3df8e0420d7404e312b15018e7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/ce1acc06847f9aa6db57e77eeaabcc8a differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/eb73fa1efe2eab9b6243832e6dcb3bec b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/eb73fa1efe2eab9b6243832e6dcb3bec
new file mode 100644
index 0000000000000000000000000000000000000000..4859214e6b68fa6e697de9748417fcf43be8081d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/eb73fa1efe2eab9b6243832e6dcb3bec differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/f642b63a24404ab56beb08485c5d8e61 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/f642b63a24404ab56beb08485c5d8e61
new file mode 100644
index 0000000000000000000000000000000000000000..93b70e461d6ae5de5e16075580b021a95db9727f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/f642b63a24404ab56beb08485c5d8e61 differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/fcfd2cd5c23f83fc280d013563819bc7 b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/fcfd2cd5c23f83fc280d013563819bc7
new file mode 100644
index 0000000000000000000000000000000000000000..a34a65cfb17eb7ecd5382759461c0b03a76cd3a1
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d/fcfd2cd5c23f83fc280d013563819bc7
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d7645d53f714b89d7cfeb2bf888d25f7cb00030791f097c183cc9cd04432758d
+size 532527
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/manifest.ocdbt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/manifest.ocdbt
new file mode 100644
index 0000000000000000000000000000000000000000..6e821bb94223050036aa525c0d3a0a330b287c22
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/manifest.ocdbt differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/config.yaml b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/config.yaml
new file mode 100644
index 0000000000000000000000000000000000000000..03840ad94d42dd5ea11068a501aaf66503356538
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/config.yaml
@@ -0,0 +1,1040 @@
+_wandb:
+    value:
+        cli_version: 0.21.0
+        e:
+            2p3j9mzw6ml6zq9wucq3537c29k5b8sm:
+                args:
+                    - train
+                    - corner
+                    - --wall-prob
+                    - "1.0"
+                    - --env-size
+                    - "13"
+                    - --log-gifs
+                    - --env-layout
+                    - tree
+                    - --wandb-project
+                    - jaxgmg-wall-prob-tree
+                    - --wandb-name
+                    - wp:1.0_alpha:0
+                    - --num-total-env-steps
+                    - "200_000_000"
+                    - --prob-shift
+                    - "0"
+                    - --keep-all-checkpoints
+                    - --num-cycles-per-checkpoint
+                    - "64"
+                cpu_count: 24
+                cpu_count_logical: 48
+                cudaVersion: "12.8"
+                disk:
+                    /:
+                        total: "102005473280"
+                        used: "43426013184"
+                email: david@timaeus.co
+                executable: /workspace/timaeus/.venv/bin/python
+                git:
+                    commit: 7415666d6a2e5d63453ea9fadaf6cecf63256a59
+                    remote: git@github.com:timaeus-research/timaeus.git
+                gpu: NVIDIA GeForce RTX 3090
+                gpu_count: 1
+                gpu_nvidia:
+                    - architecture: Ampere
+                      cudaCores: 10496
+                      memoryTotal: "25769803776"
+                      name: NVIDIA GeForce RTX 3090
+                      uuid: GPU-43bd7524-bf94-73a9-1460-2b36dd09a208
+                host: 6e66e15ca303
+                memory:
+                    total: "135066595328"
+                os: Linux-5.4.0-216-generic-x86_64-with-glibc2.39
+                program: -m jaxgmg.__main__
+                python: CPython 3.11.11
+                root: /workspace/timaeus/projects/rl
+                startedAt: "2025-08-07T18:02:11.433582Z"
+                writerId: 2p3j9mzw6ml6zq9wucq3537c29k5b8sm
+        m:
+            - "1": train-replay/proxy_corner/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": step/env-step-all-after
+              "6":
+                - 3
+              "7": []
+            - "1": ppo/avg_actor_clipfrac
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": step/ppo-update-after
+              "6":
+                - 3
+              "7": []
+            - "1": ppo/max/max_grad_norm_pre_clip
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/proxy_corner/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/avg_avg_episode_length
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/proxy_corner/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/cheese-corner_dist_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/avg_benchmark_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/rollouts_gif
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/visit_patterns/last_visit_time_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_loss
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/avg_avg_episode_length
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_critic_loss
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_grad_norm_pre_clip
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/solvable_num
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/solvable_wavg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_loss
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/proxy_corner/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/prop_walls_avg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/prop_walls_wavg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/avg_avg_episode_length
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/avg_avg_episode_length
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_advantage
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_critic_loss
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_advantage
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/lvl_avg_episode_length_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/proxy_corner/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/proxy_corner/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/scoring/scores_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/proxy_corner/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/shift_proportion
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/lvl_avg_episode_length_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/levels_cheese_is_in_corner_avg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/mouse-cheese_dist_finite_wavg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/mouse-cheese_dist_solvable_avg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_actor_approxkl1
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_actor_clipfrac
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/proxy_corner/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/lvl_avg_episode_length_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_entropy
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/rollouts_gif
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/prop_walls_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/lvl_benchmark_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_actor_approxkl3
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_actor_approxkl3
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/shift_proportion
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/levels16_img
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_critic_clipfrac
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/proxy_corner/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/avg_avg_episode_length
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/num_walls_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/lvl_benchmark_regret_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/lvl_avg_episode_length_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/proxy_corner/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/shift_proportion
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/num_walls_wavg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_actor_clipfrac
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_actor_loss
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/avg_avg_episode_length
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/avg_benchmark_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/mouse-cheese_dist_finite_avg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_loss
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/rollouts_gif
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/proxy_corner/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/proxy_corner/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/mouse-cheese_dist_finite_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/mouse-cheese_dist_solvable_wavg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_actor_approxkl1
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_advantage
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_critic_clipfrac
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/proxy_corner/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/lvl_benchmark_regret_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_actor_loss
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/rollouts_gif
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/proxy_corner/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_grad_norm_pre_clip
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/proxy_corner/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/proxy_corner/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/rollouts_gif
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/proxy_corner/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/cheese-corne_dist_wavg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/avg_entropy
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/proxy_corner/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/visit_patterns/num_replay_batches
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/proxy_corner/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/scoring/avg_scores
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/eta
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/lvl_benchmark_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-generate/lvl_avg_episode_length_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/proxy_corner/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/avg_benchmark_regret
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/visit_patterns/prev_batch_level_ids_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/lvl_avg_episode_length_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_critic_clipfrac
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/avg_benchmark_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/lvl_benchmark_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/lvl_benchmark_regret_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/rollouts_gif
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/cheese-corne_dist_avg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/distances/solvable_avg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/max/max_entropy
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_actor_loss
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_critic_loss
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/visit_patterns/avg_last_visit_time
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/avg_benchmark_regret
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/visit_patterns/first_visit_time_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-all/proxy_corner/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/layout/num_walls_avg
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_actor_approxkl3
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/proxy_corner/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/proxy_corner/lvl_avg_return_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ued/visit_patterns/avg_first_visit_time
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/proxy_corner/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-shift/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/avg_benchmark_regret
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-tree/lvl_reward_per_step_hist
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": train-replay/avg_avg_return
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": ppo/std/std_actor_approxkl1
+              "5": 4
+              "6":
+                - 1
+                - 3
+              "7": []
+            - "1": eval-batch-orig/avg_reward_per_step
+              "5": 2
+              "6":
+                - 1
+                - 3
+              "7": []
+        python_version: 3.11.11
+        t:
+            "1":
+                - 12
+                - 45
+            "2":
+                - 12
+                - 45
+            "3":
+                - 2
+                - 3
+                - 7
+                - 13
+                - 14
+                - 16
+                - 61
+            "4": 3.11.11
+            "5": 0.21.0
+            "12": 0.21.0
+            "13": linux-x86_64
+chain_mutate:
+    value: true
+checkpointing:
+    value: true
+clipping:
+    value: false
+console_log:
+    value: true
+env_corner_size:
+    value: 1
+env_layout:
+    value: tree
+env_penalize_time:
+    value: false
+env_size:
+    value: 13
+env_terminate_after_corner:
+    value: false
+evals_num_env_steps:
+    value: 512
+evals_num_levels:
+    value: 256
+gif_grid_width:
+    value: 16
+img_level_of_detail:
+    value: 1
+keep_all_checkpoints:
+    value: true
+level_splayer:
+    value: mouse
+log_gifs:
+    value: true
+log_hists:
+    value: false
+log_imgs:
+    value: true
+max_num_checkpoints:
+    value: 1
+mutate_cheese:
+    value: true
+net_cnn_type:
+    value: large
+net_rnn_type:
+    value: ff
+net_width:
+    value: 256
+num_cycles_per_big_eval:
+    value: 1024
+num_cycles_per_checkpoint:
+    value: 64
+num_cycles_per_eval:
+    value: 32
+num_cycles_per_gifs:
+    value: 1024
+num_cycles_per_log:
+    value: 32
+num_env_steps_per_cycle:
+    value: 128
+num_epochs_per_cycle:
+    value: 5
+num_minibatches_per_epoch:
+    value: 4
+num_mutate_steps:
+    value: 12
+num_parallel_envs:
+    value: 256
+num_total_env_steps:
+    value: 200000000
+num_train_levels:
+    value: 2048
+obs_level_of_detail:
+    value: 0
+plr_buffer_size:
+    value: 4096
+plr_prob_replay:
+    value: 0.5
+plr_proxy_shaping:
+    value: false
+plr_proxy_shaping_coeff:
+    value: 0.5
+plr_regret_estimator:
+    value: maxmc-actor
+plr_robust:
+    value: false
+plr_staleness_coeff:
+    value: 0.1
+plr_temperature:
+    value: 0.1
+ppo_clip_eps:
+    value: 0.1
+ppo_critic_coeff:
+    value: 0.5
+ppo_entropy_coeff:
+    value: 0.001
+ppo_gae_lambda:
+    value: 0.95
+ppo_gamma:
+    value: 0.999
+ppo_lr:
+    value: 5e-05
+ppo_lr_annealing:
+    value: false
+ppo_max_grad_norm:
+    value: 0.5
+ppo_proxy_critic_coeff:
+    value: 0.5
+prob_mutate_shift:
+    value: 0
+prob_shift:
+    value: 0
+proxy_name:
+    value: proxy_corner
+seed:
+    value: 42
+train_proxy_critic:
+    value: false
+ued:
+    value: plr
+use_fixed_eval_levels:
+    value: false
+wall_prob:
+    value: 1
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_0_45c1b3111536f49dacf9.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_0_45c1b3111536f49dacf9.png
new file mode 100644
index 0000000000000000000000000000000000000000..bf76f2b04298b684d5c82d0454e836d76dd69f44
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_0_45c1b3111536f49dacf9.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_0_7be0033df2417242180c.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_0_7be0033df2417242180c.png
new file mode 100644
index 0000000000000000000000000000000000000000..e9b0a774fd44c3c743e8829f9e969c46943d2b4a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_0_7be0033df2417242180c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1024_1c7c1613c064b799293e.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1024_1c7c1613c064b799293e.png
new file mode 100644
index 0000000000000000000000000000000000000000..54c8ef9b8e1f48531908dc670f7cf17db24d167d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1024_1c7c1613c064b799293e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1056_531cc8a85210c949838d.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1056_531cc8a85210c949838d.png
new file mode 100644
index 0000000000000000000000000000000000000000..0fa32c1d8c727b28a67f3b137dde5008968b0ca9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1056_531cc8a85210c949838d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1088_08688f2132c6052ed413.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1088_08688f2132c6052ed413.png
new file mode 100644
index 0000000000000000000000000000000000000000..ef42c3eb04882481e17d759f04a58cc0ffbd13ac
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1088_08688f2132c6052ed413.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1120_58bfe5277c5342523b24.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1120_58bfe5277c5342523b24.png
new file mode 100644
index 0000000000000000000000000000000000000000..62aa0979594355c98826c14e1d61cd9bca1998d0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1120_58bfe5277c5342523b24.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1152_19fd5d8969a67b2fa0f1.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1152_19fd5d8969a67b2fa0f1.png
new file mode 100644
index 0000000000000000000000000000000000000000..c771f7810c36928f54e7540f0bfd19fdea927cbf
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1152_19fd5d8969a67b2fa0f1.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1184_122eb5309e00c80c8cbd.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1184_122eb5309e00c80c8cbd.png
new file mode 100644
index 0000000000000000000000000000000000000000..08f4d83e73e0e04579e0aa7cc803161fe9c383ad
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1184_122eb5309e00c80c8cbd.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1216_a683cf1a75d1e21fe4bb.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1216_a683cf1a75d1e21fe4bb.png
new file mode 100644
index 0000000000000000000000000000000000000000..4f4bcd2d9e1646ffd4d8f423bc22773c021d208d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1216_a683cf1a75d1e21fe4bb.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1248_1336f8813263491c6cde.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1248_1336f8813263491c6cde.png
new file mode 100644
index 0000000000000000000000000000000000000000..28757ec37e59a058e4c82505392daed5040e1287
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1248_1336f8813263491c6cde.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1280_c9385de2a5ee510a1eef.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1280_c9385de2a5ee510a1eef.png
new file mode 100644
index 0000000000000000000000000000000000000000..6b3568fa16601a2a2abe7b47f17470e1bf0fb105
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1280_c9385de2a5ee510a1eef.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_128_712b354dbf2cfa2cc407.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_128_712b354dbf2cfa2cc407.png
new file mode 100644
index 0000000000000000000000000000000000000000..f722fc2ced30aad12123cb73629982f7c1f94074
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_128_712b354dbf2cfa2cc407.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1312_e965fc3db68362f9f2b7.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1312_e965fc3db68362f9f2b7.png
new file mode 100644
index 0000000000000000000000000000000000000000..5e7ff1cf42778fafdad034a241287af3a5ad9e4b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1312_e965fc3db68362f9f2b7.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1344_a17eafbd7961cb6cf02b.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1344_a17eafbd7961cb6cf02b.png
new file mode 100644
index 0000000000000000000000000000000000000000..dbca8d12ac16bb2b49076317a66f6171ac819bcd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1344_a17eafbd7961cb6cf02b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1376_faa65b5d8e75449959b5.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1376_faa65b5d8e75449959b5.png
new file mode 100644
index 0000000000000000000000000000000000000000..bf6afad3f348fb1e4877615a8c533018396ed08f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1376_faa65b5d8e75449959b5.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1408_b3402abd5172e6c95687.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1408_b3402abd5172e6c95687.png
new file mode 100644
index 0000000000000000000000000000000000000000..8dab90f109eedb04b0b204bf1a86650908a1de5e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1408_b3402abd5172e6c95687.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1440_3ce272de4b847c05007d.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1440_3ce272de4b847c05007d.png
new file mode 100644
index 0000000000000000000000000000000000000000..cb7f8313bb938cb529f223563c2ba90b3cca9372
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1440_3ce272de4b847c05007d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1472_e47e6b6746d526481183.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1472_e47e6b6746d526481183.png
new file mode 100644
index 0000000000000000000000000000000000000000..d5f9cf5e7a1a577b361c5e40ca83056b0647f5b0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1472_e47e6b6746d526481183.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1504_7b14d1cee429e4073667.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1504_7b14d1cee429e4073667.png
new file mode 100644
index 0000000000000000000000000000000000000000..0af36d8af4a07bce4c8db6cb023eee04a32bbdd2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1504_7b14d1cee429e4073667.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1536_62c39f9bef731e4e9c53.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1536_62c39f9bef731e4e9c53.png
new file mode 100644
index 0000000000000000000000000000000000000000..57485d52078a7a5fc75d6345382735d1a8bc5edc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1536_62c39f9bef731e4e9c53.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1568_2ea0d7bd68f510031e33.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1568_2ea0d7bd68f510031e33.png
new file mode 100644
index 0000000000000000000000000000000000000000..c0c99a1d80e2f7c5944a923eab63bb744dd1f71c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1568_2ea0d7bd68f510031e33.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1600_4bad143b71522d26cc1e.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1600_4bad143b71522d26cc1e.png
new file mode 100644
index 0000000000000000000000000000000000000000..7b376915cf9be2de3601427c74b2892f6e06e671
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1600_4bad143b71522d26cc1e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_160_82ca750bb73943929f84.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_160_82ca750bb73943929f84.png
new file mode 100644
index 0000000000000000000000000000000000000000..9e6f401e63a12f4eff6e88a00d4c47fcd562111c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_160_82ca750bb73943929f84.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1632_f0b4f59240c9942fc0c1.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1632_f0b4f59240c9942fc0c1.png
new file mode 100644
index 0000000000000000000000000000000000000000..441355aae98ed6104559c8607b833642359a0e46
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1632_f0b4f59240c9942fc0c1.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1664_da11bad89106bedccf84.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1664_da11bad89106bedccf84.png
new file mode 100644
index 0000000000000000000000000000000000000000..cfd11061a417670e467eacbc2f0d752ce0065478
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1664_da11bad89106bedccf84.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1696_7a20090b0a800d687991.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1696_7a20090b0a800d687991.png
new file mode 100644
index 0000000000000000000000000000000000000000..482c87862601b36b8e07ee9b0f2065b290bc274f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1696_7a20090b0a800d687991.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1728_900806c0bd85e6f49168.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1728_900806c0bd85e6f49168.png
new file mode 100644
index 0000000000000000000000000000000000000000..2762af685a6b36b0eedb4c138660d2ac70ae4d85
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1728_900806c0bd85e6f49168.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1760_d56d1ce8a60627b4d898.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1760_d56d1ce8a60627b4d898.png
new file mode 100644
index 0000000000000000000000000000000000000000..9302238a0287bfb6af75eee2a7506de1db250c60
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1760_d56d1ce8a60627b4d898.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1792_f60a85c73c75063aa771.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1792_f60a85c73c75063aa771.png
new file mode 100644
index 0000000000000000000000000000000000000000..377e8ee749f85f5679d1b6810108e7e3231e3fbb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1792_f60a85c73c75063aa771.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1824_b42524b75a2c86cec874.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1824_b42524b75a2c86cec874.png
new file mode 100644
index 0000000000000000000000000000000000000000..fa2b9e480872ccbdb1367ae95ea160e22c5b96bb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1824_b42524b75a2c86cec874.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1856_fd2736a285da658ad795.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1856_fd2736a285da658ad795.png
new file mode 100644
index 0000000000000000000000000000000000000000..a0417f9fcb30ffc8b7bec935c1503d5046d28f2f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1856_fd2736a285da658ad795.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1888_1468d17343b5bf59f72b.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1888_1468d17343b5bf59f72b.png
new file mode 100644
index 0000000000000000000000000000000000000000..e9a8dd9ee28511b16953c2416aae4832ebef0b52
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1888_1468d17343b5bf59f72b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1920_3030c99020057d9dd854.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1920_3030c99020057d9dd854.png
new file mode 100644
index 0000000000000000000000000000000000000000..4208ca3b2f8b3c5edb819ace06c63c87d28e6c3b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1920_3030c99020057d9dd854.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_192_f6b6f56f44068b2d9cd4.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_192_f6b6f56f44068b2d9cd4.png
new file mode 100644
index 0000000000000000000000000000000000000000..f7b892e1412c2b1e5b1761a70a527ae81d3900e5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_192_f6b6f56f44068b2d9cd4.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1952_f96a542f0122b29f2c4e.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1952_f96a542f0122b29f2c4e.png
new file mode 100644
index 0000000000000000000000000000000000000000..8dfe5460f8e7fcff611cdd4b25ff40f6b5dbfafc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1952_f96a542f0122b29f2c4e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1984_7e832c30bfb06df8126d.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1984_7e832c30bfb06df8126d.png
new file mode 100644
index 0000000000000000000000000000000000000000..ad0b7edff18edc2147df68a0060843645af4e15e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_1984_7e832c30bfb06df8126d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2016_421ed0b0804c40aa3443.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2016_421ed0b0804c40aa3443.png
new file mode 100644
index 0000000000000000000000000000000000000000..a6ae7646d06eadada0a982e11ac6cc525c707b64
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2016_421ed0b0804c40aa3443.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2048_3198a4f64a3d9333703f.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2048_3198a4f64a3d9333703f.png
new file mode 100644
index 0000000000000000000000000000000000000000..eac9b0bd4a59b3c0e755004d2d42234b300f90f8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2048_3198a4f64a3d9333703f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2080_4ccc5159e96c246a8fce.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2080_4ccc5159e96c246a8fce.png
new file mode 100644
index 0000000000000000000000000000000000000000..ffca097a67ad078feb547a257b0547a7db8bd9a3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2080_4ccc5159e96c246a8fce.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2112_9d4fb7e6af2cde22bcfb.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2112_9d4fb7e6af2cde22bcfb.png
new file mode 100644
index 0000000000000000000000000000000000000000..7be7a19898d4942e75e6c07b36f16d11dd035a55
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2112_9d4fb7e6af2cde22bcfb.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2144_ebf35848f331698eeed4.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2144_ebf35848f331698eeed4.png
new file mode 100644
index 0000000000000000000000000000000000000000..6e6d80ad6fb42af91878177f4999bd1acbab12eb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2144_ebf35848f331698eeed4.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2176_dca8e1102d6e433a505e.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2176_dca8e1102d6e433a505e.png
new file mode 100644
index 0000000000000000000000000000000000000000..d723d27d699c247a73e07e3f3e5b157b6452d72b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2176_dca8e1102d6e433a505e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2208_1de3d27c5ca554a68fd3.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2208_1de3d27c5ca554a68fd3.png
new file mode 100644
index 0000000000000000000000000000000000000000..8d22543e592bf7f4c798df72cab3c239359257df
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2208_1de3d27c5ca554a68fd3.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2240_358bb576925799558309.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2240_358bb576925799558309.png
new file mode 100644
index 0000000000000000000000000000000000000000..ca4a2b1de63024e08fdc0240f52aafc524026024
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2240_358bb576925799558309.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_224_87bc6622e9f2ca2fe0c6.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_224_87bc6622e9f2ca2fe0c6.png
new file mode 100644
index 0000000000000000000000000000000000000000..d71c275e81f2a073ab178c8000f08eb8b7ac6a44
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_224_87bc6622e9f2ca2fe0c6.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2272_54d830c3b9ac617eb725.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2272_54d830c3b9ac617eb725.png
new file mode 100644
index 0000000000000000000000000000000000000000..12637cc5a9da06e71fadb7e0497456003678865f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2272_54d830c3b9ac617eb725.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2304_d27632d14f2c65420515.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2304_d27632d14f2c65420515.png
new file mode 100644
index 0000000000000000000000000000000000000000..180e7d05addeac21ccc1173bb3605bfa8a2bf41f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2304_d27632d14f2c65420515.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2336_bd747c4f2039c18ff023.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2336_bd747c4f2039c18ff023.png
new file mode 100644
index 0000000000000000000000000000000000000000..9fb566c049ceb3f089fce117bd7144e66829f59c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2336_bd747c4f2039c18ff023.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2368_7404dc6aedaf29e38388.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2368_7404dc6aedaf29e38388.png
new file mode 100644
index 0000000000000000000000000000000000000000..462c2c64bbec4e226436aa899b52ead196c6f7df
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2368_7404dc6aedaf29e38388.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2400_3b7d1c976b8906e4cae3.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2400_3b7d1c976b8906e4cae3.png
new file mode 100644
index 0000000000000000000000000000000000000000..de31888a215170c8222a4dc376669c4d05044f42
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2400_3b7d1c976b8906e4cae3.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2432_a8b9c9d394e78b2d7deb.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2432_a8b9c9d394e78b2d7deb.png
new file mode 100644
index 0000000000000000000000000000000000000000..d54b093acf583ac818ae6e7b7ca949192d8cab81
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2432_a8b9c9d394e78b2d7deb.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2464_c9cef148244734944376.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2464_c9cef148244734944376.png
new file mode 100644
index 0000000000000000000000000000000000000000..55337117d73c70dcb879b6743debf28e250f28dc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2464_c9cef148244734944376.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2496_85d73bc7d49deeb020b3.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2496_85d73bc7d49deeb020b3.png
new file mode 100644
index 0000000000000000000000000000000000000000..2e0e8bb9e3d4ea0f4c54b7f4cb914af7a8e592a8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2496_85d73bc7d49deeb020b3.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2528_b5736eff3b5228efb1a8.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2528_b5736eff3b5228efb1a8.png
new file mode 100644
index 0000000000000000000000000000000000000000..8c194727ed1ab7f341a25554421738b613228f97
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2528_b5736eff3b5228efb1a8.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2560_0ccc9e01877819d3ce00.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2560_0ccc9e01877819d3ce00.png
new file mode 100644
index 0000000000000000000000000000000000000000..c1cedcd74e0bc907d860b0d59c1b658925b2d899
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2560_0ccc9e01877819d3ce00.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_256_e43dcfba7d7113ca1310.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_256_e43dcfba7d7113ca1310.png
new file mode 100644
index 0000000000000000000000000000000000000000..65d555e2265eecd5a6d8f65b3ea4333bbd9d27b5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_256_e43dcfba7d7113ca1310.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2592_519de5657cbac4ee4c8d.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2592_519de5657cbac4ee4c8d.png
new file mode 100644
index 0000000000000000000000000000000000000000..d45b1b5d35ea1f02143592cb979fd6e32bf674a3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2592_519de5657cbac4ee4c8d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2624_c5e0c81c39ce07d62c14.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2624_c5e0c81c39ce07d62c14.png
new file mode 100644
index 0000000000000000000000000000000000000000..ea8b1749f20ece66ba2fd5d012331757eb19855e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2624_c5e0c81c39ce07d62c14.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2656_51c8e9d64037f5cc56ac.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2656_51c8e9d64037f5cc56ac.png
new file mode 100644
index 0000000000000000000000000000000000000000..9fd70ce83ef66c014bc743dc7d03f2bcd7810dd9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2656_51c8e9d64037f5cc56ac.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2688_a46a8ebb922336b159a7.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2688_a46a8ebb922336b159a7.png
new file mode 100644
index 0000000000000000000000000000000000000000..7d408bce19ffe06d3470b32d7a52e853f1416d45
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2688_a46a8ebb922336b159a7.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2720_1c84d1d336bfa8521d0e.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2720_1c84d1d336bfa8521d0e.png
new file mode 100644
index 0000000000000000000000000000000000000000..8b92b564fc0b2417262ee6c74ce18d29627a178b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2720_1c84d1d336bfa8521d0e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2752_5514bcd8bf660054d9dd.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2752_5514bcd8bf660054d9dd.png
new file mode 100644
index 0000000000000000000000000000000000000000..5bca7800f6879924a21582d103b309d43c1f8280
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2752_5514bcd8bf660054d9dd.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2784_b15142966369b532b649.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2784_b15142966369b532b649.png
new file mode 100644
index 0000000000000000000000000000000000000000..f69d073f29a5e60f5fd12f432f4df2aa8bd2613a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2784_b15142966369b532b649.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2816_c2987bbc9d898c5e1a32.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2816_c2987bbc9d898c5e1a32.png
new file mode 100644
index 0000000000000000000000000000000000000000..f5b5e1f9804104029e3b75ca972a4e84483c3129
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2816_c2987bbc9d898c5e1a32.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2848_aa195b1074d8545f2718.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2848_aa195b1074d8545f2718.png
new file mode 100644
index 0000000000000000000000000000000000000000..a55e8c2e705e71c4f4bdfb47e6f5140a46146511
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2848_aa195b1074d8545f2718.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2880_5f809f1b02300a3f2df4.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2880_5f809f1b02300a3f2df4.png
new file mode 100644
index 0000000000000000000000000000000000000000..dbb521c7ef8e01b0018cfa6405018b7cb1dd63f5
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2880_5f809f1b02300a3f2df4.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_288_8bd6500699c45672a00d.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_288_8bd6500699c45672a00d.png
new file mode 100644
index 0000000000000000000000000000000000000000..b3678426d773513c6bc86972ff2f6e19cb0909c9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_288_8bd6500699c45672a00d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2912_b34fb48a3a14713a2802.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2912_b34fb48a3a14713a2802.png
new file mode 100644
index 0000000000000000000000000000000000000000..e854eb7bd91d0226bdde603801d770c128d6aa2d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2912_b34fb48a3a14713a2802.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2944_989780fcd0a96bf5280a.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2944_989780fcd0a96bf5280a.png
new file mode 100644
index 0000000000000000000000000000000000000000..53cf55676051ce822b6053be03d30595afd67c53
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2944_989780fcd0a96bf5280a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2976_a1686115aa502f472fa2.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2976_a1686115aa502f472fa2.png
new file mode 100644
index 0000000000000000000000000000000000000000..a83016c5a37ae3fa00be8690be7d8c3b0084b5aa
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_2976_a1686115aa502f472fa2.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3008_1944f55895cf46713cf3.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3008_1944f55895cf46713cf3.png
new file mode 100644
index 0000000000000000000000000000000000000000..6a4c6c2a3b1e6547bb12886c8f472cd471191cc6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3008_1944f55895cf46713cf3.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3040_6fa63538e152fdcfe3c1.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3040_6fa63538e152fdcfe3c1.png
new file mode 100644
index 0000000000000000000000000000000000000000..9d0f239102af8c9524f2bd23d4d2112cf8d78515
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3040_6fa63538e152fdcfe3c1.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3072_0ea0b00188b247ef831c.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3072_0ea0b00188b247ef831c.png
new file mode 100644
index 0000000000000000000000000000000000000000..54a720f862bbb34f9a51b76674a6ca21b91d0a9d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3072_0ea0b00188b247ef831c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3104_31c1a6ddc4ca1153048a.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3104_31c1a6ddc4ca1153048a.png
new file mode 100644
index 0000000000000000000000000000000000000000..891bfb4dc0cb99acf7d03b75ca4701f1ecef6d28
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3104_31c1a6ddc4ca1153048a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3136_e126c2b5cefe44485a34.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3136_e126c2b5cefe44485a34.png
new file mode 100644
index 0000000000000000000000000000000000000000..a513d7b5d29522bd95eb1cd0aa109511b3ef7187
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3136_e126c2b5cefe44485a34.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3168_0907d9371e359f499a85.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3168_0907d9371e359f499a85.png
new file mode 100644
index 0000000000000000000000000000000000000000..61fc34b6a6980302f507edfdddc935a8d5adae0b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3168_0907d9371e359f499a85.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3200_68dd0ea6671c1125d53f.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3200_68dd0ea6671c1125d53f.png
new file mode 100644
index 0000000000000000000000000000000000000000..cc774677723bc47fcbb169e0d311a7be3805ebdc
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3200_68dd0ea6671c1125d53f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_320_e517b6710b707a49e9fa.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_320_e517b6710b707a49e9fa.png
new file mode 100644
index 0000000000000000000000000000000000000000..9d3e5efcebeba93f651f2b19fcb4aefabe358c8a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_320_e517b6710b707a49e9fa.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3232_c672cd8c78c13d82200d.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3232_c672cd8c78c13d82200d.png
new file mode 100644
index 0000000000000000000000000000000000000000..0ded82a4fcd25d4a9e376872d3e86c76e5ab93fa
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3232_c672cd8c78c13d82200d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3264_264316beb6929f255a7e.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3264_264316beb6929f255a7e.png
new file mode 100644
index 0000000000000000000000000000000000000000..7fb9ba6196c153adb0091efa56d201dc8e70aa98
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3264_264316beb6929f255a7e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3296_081005b6168dd6ccf80b.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3296_081005b6168dd6ccf80b.png
new file mode 100644
index 0000000000000000000000000000000000000000..1b1f53241f9b8d80fa34381e1488e8fa082d45fb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3296_081005b6168dd6ccf80b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_32_a1840421506bfa6280fe.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_32_a1840421506bfa6280fe.png
new file mode 100644
index 0000000000000000000000000000000000000000..b0fcf3e67e74ac7e2bb923f2442b7ba9b56c9b6c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_32_a1840421506bfa6280fe.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3328_27b10bb51361ace7ec94.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3328_27b10bb51361ace7ec94.png
new file mode 100644
index 0000000000000000000000000000000000000000..31f9353bc3f004c2eb461d8a449fc41c9ffa47da
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3328_27b10bb51361ace7ec94.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3360_49a74376111753110e67.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3360_49a74376111753110e67.png
new file mode 100644
index 0000000000000000000000000000000000000000..e34497ae364386b385928fa29dd0a3e83e5ba37e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3360_49a74376111753110e67.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3392_73761bda8ef06972be7a.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3392_73761bda8ef06972be7a.png
new file mode 100644
index 0000000000000000000000000000000000000000..60f32a3d16e1a4462bfef325b9c79034ceef927b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3392_73761bda8ef06972be7a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3424_7e1efd4cf2781eea9ef1.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3424_7e1efd4cf2781eea9ef1.png
new file mode 100644
index 0000000000000000000000000000000000000000..8af4ca168d6ad0d843e72b38a8e67f6a97605776
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3424_7e1efd4cf2781eea9ef1.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3456_2ca0a1b4deadd69d3dfc.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3456_2ca0a1b4deadd69d3dfc.png
new file mode 100644
index 0000000000000000000000000000000000000000..f0d9a9502c58cbe774537d4de333ed208d545613
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3456_2ca0a1b4deadd69d3dfc.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3488_8d90dbb5af21145e4e95.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3488_8d90dbb5af21145e4e95.png
new file mode 100644
index 0000000000000000000000000000000000000000..0784fce579f77defe9e806229d6af20668483509
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3488_8d90dbb5af21145e4e95.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3520_7251f415120a44a41787.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3520_7251f415120a44a41787.png
new file mode 100644
index 0000000000000000000000000000000000000000..5e1a89c9ed574b6500ee8e61d2fc4b4d794d329f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3520_7251f415120a44a41787.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_352_80b0ca89ea16fb17b4e9.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_352_80b0ca89ea16fb17b4e9.png
new file mode 100644
index 0000000000000000000000000000000000000000..3e610ba5ed12cb480715e607917a197db9e22693
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_352_80b0ca89ea16fb17b4e9.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3552_1d6e0ac4a5bd81ddea8b.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3552_1d6e0ac4a5bd81ddea8b.png
new file mode 100644
index 0000000000000000000000000000000000000000..781679b92730866742c5032dafbac64ddf930ed0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3552_1d6e0ac4a5bd81ddea8b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3584_a26d24b5775366055b4f.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3584_a26d24b5775366055b4f.png
new file mode 100644
index 0000000000000000000000000000000000000000..9dbcb3e3dcd5b7860e4a8b3b5d838ccff17638d1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3584_a26d24b5775366055b4f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3616_f2135ab07d5fe55bedae.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3616_f2135ab07d5fe55bedae.png
new file mode 100644
index 0000000000000000000000000000000000000000..8f25ec5d10985c6e8e0e34f5cd6ebb57fba8be10
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3616_f2135ab07d5fe55bedae.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3648_fc2e8eb7d68a0a53713d.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3648_fc2e8eb7d68a0a53713d.png
new file mode 100644
index 0000000000000000000000000000000000000000..64572194372d569c5c943569f54d9d213282d992
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3648_fc2e8eb7d68a0a53713d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3680_67fddbc624ae33ec6d01.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3680_67fddbc624ae33ec6d01.png
new file mode 100644
index 0000000000000000000000000000000000000000..fc7d635df011aabdec5bc45626a70cf052da67e2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3680_67fddbc624ae33ec6d01.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3712_e854dbed0e672ae817a3.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3712_e854dbed0e672ae817a3.png
new file mode 100644
index 0000000000000000000000000000000000000000..e56d9d93b51ec814da6dfbde2d1a22b5b2583065
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3712_e854dbed0e672ae817a3.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3744_e79263374b06dff46fbf.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3744_e79263374b06dff46fbf.png
new file mode 100644
index 0000000000000000000000000000000000000000..2689ff1e8e57f22448e791d1cfbbeccbdf964c80
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3744_e79263374b06dff46fbf.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3776_5a9ae2c0553a327f1c1f.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3776_5a9ae2c0553a327f1c1f.png
new file mode 100644
index 0000000000000000000000000000000000000000..d09f62223bfd8c7e49f4954d04acbd638872c266
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3776_5a9ae2c0553a327f1c1f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3808_2b1b68bcb01980d8b0b5.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3808_2b1b68bcb01980d8b0b5.png
new file mode 100644
index 0000000000000000000000000000000000000000..2d116e76048f6ca67bceff4cdec0653ec3ce2fd3
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3808_2b1b68bcb01980d8b0b5.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3840_4bee95033813a06fa412.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3840_4bee95033813a06fa412.png
new file mode 100644
index 0000000000000000000000000000000000000000..e2bad1689820cc77664b5640e176b0085c4ebcfd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3840_4bee95033813a06fa412.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_384_97c0e7a1f868fb20d1ed.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_384_97c0e7a1f868fb20d1ed.png
new file mode 100644
index 0000000000000000000000000000000000000000..62afe53cf75d228bc6d7daa45d6fd71a01dedbe0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_384_97c0e7a1f868fb20d1ed.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3872_17a987f03890370bf417.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3872_17a987f03890370bf417.png
new file mode 100644
index 0000000000000000000000000000000000000000..e5d2d3afeb2d35fab70f7b684961d01e87f406b4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3872_17a987f03890370bf417.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3904_703d1bb47d83c036f7b8.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3904_703d1bb47d83c036f7b8.png
new file mode 100644
index 0000000000000000000000000000000000000000..2bf376ef1a24356e2463a401103bdce029fc46d9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3904_703d1bb47d83c036f7b8.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3936_b607741a6118a02bf4d9.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3936_b607741a6118a02bf4d9.png
new file mode 100644
index 0000000000000000000000000000000000000000..7071329137444922a6c6243b36502da0b18e249c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3936_b607741a6118a02bf4d9.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3968_9548e61ca2ecbe32f63c.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3968_9548e61ca2ecbe32f63c.png
new file mode 100644
index 0000000000000000000000000000000000000000..a13355c966843424d6ce8472c92a0a9b2f049cfe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_3968_9548e61ca2ecbe32f63c.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4000_a3bfb18c42dd209a451d.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4000_a3bfb18c42dd209a451d.png
new file mode 100644
index 0000000000000000000000000000000000000000..964332ffaa92748f43807f009c741f7718d2edd2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4000_a3bfb18c42dd209a451d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4032_ccec106a65df33574194.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4032_ccec106a65df33574194.png
new file mode 100644
index 0000000000000000000000000000000000000000..cbc7fc6592d959379d0155f4ebc3ae940d7d5b13
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4032_ccec106a65df33574194.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4064_ad8ce06e9f4c4c97e41f.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4064_ad8ce06e9f4c4c97e41f.png
new file mode 100644
index 0000000000000000000000000000000000000000..1aa00307034ef78212595a36d0d8febd5081d2d1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4064_ad8ce06e9f4c4c97e41f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4096_c28075ff99ee15012090.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4096_c28075ff99ee15012090.png
new file mode 100644
index 0000000000000000000000000000000000000000..3e9642232673266fc2adc6b92b4a5d3a04acbbcf
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4096_c28075ff99ee15012090.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4128_f8477defcdd93664a173.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4128_f8477defcdd93664a173.png
new file mode 100644
index 0000000000000000000000000000000000000000..d69ba85e3db26df667b8d2a9d6a08e4830a2df04
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4128_f8477defcdd93664a173.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4160_556ac8140abdd4f61b81.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4160_556ac8140abdd4f61b81.png
new file mode 100644
index 0000000000000000000000000000000000000000..cef94c06985d2b94a15fec907a07ace3c2d59deb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4160_556ac8140abdd4f61b81.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_416_e74c2d301ae26a7f2f5a.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_416_e74c2d301ae26a7f2f5a.png
new file mode 100644
index 0000000000000000000000000000000000000000..5038a38231c1c3d7d6d8a939ef729e83d574b411
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_416_e74c2d301ae26a7f2f5a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4192_0bdd07e25e5bbfea17e9.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4192_0bdd07e25e5bbfea17e9.png
new file mode 100644
index 0000000000000000000000000000000000000000..568a0d944dc7dcb22fed0d19dc21f4fa40e13fdb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4192_0bdd07e25e5bbfea17e9.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4224_ac424727eb437e9ff739.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4224_ac424727eb437e9ff739.png
new file mode 100644
index 0000000000000000000000000000000000000000..0677f0b10b66f09dcefaf9cab422eb7babf457f0
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4224_ac424727eb437e9ff739.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4256_5561c763375bd1208b92.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4256_5561c763375bd1208b92.png
new file mode 100644
index 0000000000000000000000000000000000000000..0fc5ea536b2d77b1e8a935c3b6af80bce9bf8be6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4256_5561c763375bd1208b92.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4288_94fe95e108c465ee6316.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4288_94fe95e108c465ee6316.png
new file mode 100644
index 0000000000000000000000000000000000000000..3a00f394b0e519d3bb7a5bb9bd13da7107081d2c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4288_94fe95e108c465ee6316.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4320_d999da63ff4a0edecb47.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4320_d999da63ff4a0edecb47.png
new file mode 100644
index 0000000000000000000000000000000000000000..face0e119c8ef353d339d746d0b905a3e69bfd98
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4320_d999da63ff4a0edecb47.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4352_9b60788981be758cf600.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4352_9b60788981be758cf600.png
new file mode 100644
index 0000000000000000000000000000000000000000..008794ea1fc9830e0e6cf01472de695187b3a97a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4352_9b60788981be758cf600.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4384_8bae7f6d00739d6576a3.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4384_8bae7f6d00739d6576a3.png
new file mode 100644
index 0000000000000000000000000000000000000000..72412643f2fa355801d74e870f6ad72437d6ae23
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4384_8bae7f6d00739d6576a3.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4416_117b94de196f29c264aa.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4416_117b94de196f29c264aa.png
new file mode 100644
index 0000000000000000000000000000000000000000..fe6a5cb7f5ba1046cbe2d451bb5dd23f988c519f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4416_117b94de196f29c264aa.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4448_831848939532c0a38bde.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4448_831848939532c0a38bde.png
new file mode 100644
index 0000000000000000000000000000000000000000..8bd0b6a3aab498d8fbcb7f8fde839e5530169d77
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4448_831848939532c0a38bde.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4480_297decbc2e8656124aad.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4480_297decbc2e8656124aad.png
new file mode 100644
index 0000000000000000000000000000000000000000..193a82cf7cc706a8f3b6f81bec595e6aeb130bb9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4480_297decbc2e8656124aad.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_448_793e2bbe1f87283146a3.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_448_793e2bbe1f87283146a3.png
new file mode 100644
index 0000000000000000000000000000000000000000..4a0d7de2ecb56de76169f2c6c0a8711b0c96a883
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_448_793e2bbe1f87283146a3.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4512_116427de77640e2ae772.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4512_116427de77640e2ae772.png
new file mode 100644
index 0000000000000000000000000000000000000000..f230f533571a2d3ed2225f51eb07bdd42d39deae
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4512_116427de77640e2ae772.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4544_c5c9a23ab273296a3b18.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4544_c5c9a23ab273296a3b18.png
new file mode 100644
index 0000000000000000000000000000000000000000..d52c93a5a4615420b5e2145c30646ee89258723c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4544_c5c9a23ab273296a3b18.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4576_74569696717cd15198dd.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4576_74569696717cd15198dd.png
new file mode 100644
index 0000000000000000000000000000000000000000..f96d10be2cfb04f5f874d3b3488e9cbfa27afc0b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4576_74569696717cd15198dd.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4608_025b46ab977216d97b40.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4608_025b46ab977216d97b40.png
new file mode 100644
index 0000000000000000000000000000000000000000..d291c50f9dd0f920d81f7a244daed758cf67dcd7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4608_025b46ab977216d97b40.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4640_c051f2c941de6e7808d0.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4640_c051f2c941de6e7808d0.png
new file mode 100644
index 0000000000000000000000000000000000000000..9b60ba8521ab41ccce5e787471cae58071162986
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4640_c051f2c941de6e7808d0.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4672_ddaf493132a8d13b2ea8.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4672_ddaf493132a8d13b2ea8.png
new file mode 100644
index 0000000000000000000000000000000000000000..4242c21f5c5fbebb34058c1f2d31d2704c4786ca
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4672_ddaf493132a8d13b2ea8.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4704_fad34f110652e9d6d59f.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4704_fad34f110652e9d6d59f.png
new file mode 100644
index 0000000000000000000000000000000000000000..085d6e1df901dc3756ec30e0404d9df5669743a9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4704_fad34f110652e9d6d59f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4736_bf22f6ec8ddba52900f1.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4736_bf22f6ec8ddba52900f1.png
new file mode 100644
index 0000000000000000000000000000000000000000..a52d3ecb754de254b58a2afcbb38217e585e313d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4736_bf22f6ec8ddba52900f1.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4768_79d2c1143f0bc92bad8b.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4768_79d2c1143f0bc92bad8b.png
new file mode 100644
index 0000000000000000000000000000000000000000..4898e1cdcdfd9f1f49646399b44a3e9b5aa7b129
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4768_79d2c1143f0bc92bad8b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4800_a293661e3ef3ae4191f3.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4800_a293661e3ef3ae4191f3.png
new file mode 100644
index 0000000000000000000000000000000000000000..742c75c87d42459187e6c4cae7fa5f4d412fac4d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4800_a293661e3ef3ae4191f3.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_480_4c8f3e0d18a87eb0960e.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_480_4c8f3e0d18a87eb0960e.png
new file mode 100644
index 0000000000000000000000000000000000000000..b70a77a5a2b2fb165d48f0bf7bbadf0759daa771
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_480_4c8f3e0d18a87eb0960e.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4832_ddab2f91fd3a24b96ea8.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4832_ddab2f91fd3a24b96ea8.png
new file mode 100644
index 0000000000000000000000000000000000000000..03e914e7b08d7c52601aa683e18f9ee940b8443a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4832_ddab2f91fd3a24b96ea8.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4864_940c6446429f471e7b1f.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4864_940c6446429f471e7b1f.png
new file mode 100644
index 0000000000000000000000000000000000000000..e815e69db04d80bb58754bd75b3d87afef2fb11e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4864_940c6446429f471e7b1f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4896_a06c24ca91bf3464fb13.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4896_a06c24ca91bf3464fb13.png
new file mode 100644
index 0000000000000000000000000000000000000000..fa9c5d027dcf96ec5a7b01b136f3ac61a94d8eaa
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4896_a06c24ca91bf3464fb13.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4928_dd303b149ccfc4aef298.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4928_dd303b149ccfc4aef298.png
new file mode 100644
index 0000000000000000000000000000000000000000..e1a5cce8940949b6c98c0bb7e3c7ff72f4e99c61
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4928_dd303b149ccfc4aef298.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4960_3be1a6030c6a193f37d8.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4960_3be1a6030c6a193f37d8.png
new file mode 100644
index 0000000000000000000000000000000000000000..dc2038dc229b09a3158d6d36627f251c8b41c526
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4960_3be1a6030c6a193f37d8.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4992_f6090a71137b8d19f99d.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4992_f6090a71137b8d19f99d.png
new file mode 100644
index 0000000000000000000000000000000000000000..aafef3e48a89ea1ab545029a500f2ed9e0626631
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_4992_f6090a71137b8d19f99d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5024_474a8ee4e71fef2854fb.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5024_474a8ee4e71fef2854fb.png
new file mode 100644
index 0000000000000000000000000000000000000000..9935f9c7c55e6885f736fd50e30209ec6c917cd9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5024_474a8ee4e71fef2854fb.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5056_eb2841d2a7fe38191dba.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5056_eb2841d2a7fe38191dba.png
new file mode 100644
index 0000000000000000000000000000000000000000..ca17fd1384b7e2c3e05ddebfdf82e99eac678cbe
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5056_eb2841d2a7fe38191dba.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5088_c724e96368fe51ba0f8b.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5088_c724e96368fe51ba0f8b.png
new file mode 100644
index 0000000000000000000000000000000000000000..30c24ff1d7447ffc422c6546d57c640ce6eef27a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5088_c724e96368fe51ba0f8b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5120_916de0dc02a352cb53a3.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5120_916de0dc02a352cb53a3.png
new file mode 100644
index 0000000000000000000000000000000000000000..e0a35aa72e12cf5b07ddf4dabbb0a999d1c01e7e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5120_916de0dc02a352cb53a3.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_512_62d16ab5f218d6fbcf23.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_512_62d16ab5f218d6fbcf23.png
new file mode 100644
index 0000000000000000000000000000000000000000..6cfb56e7c783e3972943cb5b9413a85fed4449ee
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_512_62d16ab5f218d6fbcf23.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5152_f147b6940c0bbc5405ce.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5152_f147b6940c0bbc5405ce.png
new file mode 100644
index 0000000000000000000000000000000000000000..f4363d55c7c9189d1a6fbef9efe4c0e9472a0d2c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5152_f147b6940c0bbc5405ce.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5184_c930df779e4c7bcd9b57.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5184_c930df779e4c7bcd9b57.png
new file mode 100644
index 0000000000000000000000000000000000000000..cf22f598966f115e5d51c101bc1fbe04ff494b4e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5184_c930df779e4c7bcd9b57.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5216_d0d45caa9b3a39f7f498.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5216_d0d45caa9b3a39f7f498.png
new file mode 100644
index 0000000000000000000000000000000000000000..7b56fb56ca97c85ba0d05f94b44a3f7955f52ef9
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5216_d0d45caa9b3a39f7f498.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5248_28efa27e5ab83dac3aea.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5248_28efa27e5ab83dac3aea.png
new file mode 100644
index 0000000000000000000000000000000000000000..f3d051da191ebce6679c8179b2bc51440cc1d08e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5248_28efa27e5ab83dac3aea.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5280_33ad40b412c6bcd07b46.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5280_33ad40b412c6bcd07b46.png
new file mode 100644
index 0000000000000000000000000000000000000000..b2fd79721f0adc6a3ce04ae3ba3824c6bdb17fba
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5280_33ad40b412c6bcd07b46.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5312_6a3abff501ab1bfdc036.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5312_6a3abff501ab1bfdc036.png
new file mode 100644
index 0000000000000000000000000000000000000000..4fd69e017e48872f9f970a234589f94c0808dcae
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5312_6a3abff501ab1bfdc036.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5344_9bc384f2a0db003f40e7.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5344_9bc384f2a0db003f40e7.png
new file mode 100644
index 0000000000000000000000000000000000000000..35719102f89ea0409e41e75f5cc0d80ac826fccd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5344_9bc384f2a0db003f40e7.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5376_b2c81bc7c77d46a155ce.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5376_b2c81bc7c77d46a155ce.png
new file mode 100644
index 0000000000000000000000000000000000000000..ff8718e6fdc271a54efc748ed970ac271e6ef9bd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5376_b2c81bc7c77d46a155ce.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5408_b1410485f231595f4498.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5408_b1410485f231595f4498.png
new file mode 100644
index 0000000000000000000000000000000000000000..c67beddc191d6b02300b44589400394248adb17f
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5408_b1410485f231595f4498.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5440_30958e2d3de9e6841a48.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5440_30958e2d3de9e6841a48.png
new file mode 100644
index 0000000000000000000000000000000000000000..3d508a86fae7e12e960acfac146c2b9ea64531e2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5440_30958e2d3de9e6841a48.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_544_99a44d23375f562ec6e0.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_544_99a44d23375f562ec6e0.png
new file mode 100644
index 0000000000000000000000000000000000000000..e08e9d0a001fa6b9ef89e3898f2c098189b1d3d4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_544_99a44d23375f562ec6e0.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5472_753bdce2b761d122e2fb.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5472_753bdce2b761d122e2fb.png
new file mode 100644
index 0000000000000000000000000000000000000000..9bca2fbea1829a60f0dc52a91387a02a965bdd68
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5472_753bdce2b761d122e2fb.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5504_5ff043d884633e9032b4.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5504_5ff043d884633e9032b4.png
new file mode 100644
index 0000000000000000000000000000000000000000..bfe80b41bce744eeab486c78318e1855448dafd6
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5504_5ff043d884633e9032b4.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5536_0456d3b67b644d7f5712.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5536_0456d3b67b644d7f5712.png
new file mode 100644
index 0000000000000000000000000000000000000000..fd49e93fdc32b0ef89a7b5fc1991ef0b91ddb2ff
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5536_0456d3b67b644d7f5712.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5568_057a96440850bdbbd054.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5568_057a96440850bdbbd054.png
new file mode 100644
index 0000000000000000000000000000000000000000..8efde1895b1954e0f4bc1c115f2dd2839c338c3b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5568_057a96440850bdbbd054.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5600_866e01e41e5ae92705ed.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5600_866e01e41e5ae92705ed.png
new file mode 100644
index 0000000000000000000000000000000000000000..739310dc08fc9e8063c76ff03436c32c0316b721
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5600_866e01e41e5ae92705ed.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5632_fd5bc0ea7d3de3cec414.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5632_fd5bc0ea7d3de3cec414.png
new file mode 100644
index 0000000000000000000000000000000000000000..a2a4a9e5f0166469338ab703dc4fa05c246642a7
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5632_fd5bc0ea7d3de3cec414.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5664_9a369216f7fc2907af77.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5664_9a369216f7fc2907af77.png
new file mode 100644
index 0000000000000000000000000000000000000000..ae50aef6ab601ade70e5c4ff99ddeb802f74a255
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5664_9a369216f7fc2907af77.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5696_ad78697704cf2dd62988.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5696_ad78697704cf2dd62988.png
new file mode 100644
index 0000000000000000000000000000000000000000..fd4266f8c7452a0b1ab45f213ec9f582a9f2cd57
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5696_ad78697704cf2dd62988.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5728_fc1f5249545201a74f96.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5728_fc1f5249545201a74f96.png
new file mode 100644
index 0000000000000000000000000000000000000000..a84757520a1d7e2fa19ef86f967b9292408e5c0b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5728_fc1f5249545201a74f96.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5760_9cdc6a02585717cce8e4.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5760_9cdc6a02585717cce8e4.png
new file mode 100644
index 0000000000000000000000000000000000000000..9819aa598766e8de454b76eb3f9b683efa308bbb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5760_9cdc6a02585717cce8e4.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_576_9b33b867b54b62bf73fc.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_576_9b33b867b54b62bf73fc.png
new file mode 100644
index 0000000000000000000000000000000000000000..cbbfaefd6ad576658886cfc3ccf2661114272206
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_576_9b33b867b54b62bf73fc.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5792_9b56fd232ac7b79ac30a.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5792_9b56fd232ac7b79ac30a.png
new file mode 100644
index 0000000000000000000000000000000000000000..b1d815c9546d6503930b45286ee92ac6a975a0ab
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5792_9b56fd232ac7b79ac30a.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5824_22defe9f85145abdcb1b.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5824_22defe9f85145abdcb1b.png
new file mode 100644
index 0000000000000000000000000000000000000000..dd18b7abab306d20d4a471dae9fad0dbf57f8b49
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5824_22defe9f85145abdcb1b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5856_b6779a50beeb70457001.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5856_b6779a50beeb70457001.png
new file mode 100644
index 0000000000000000000000000000000000000000..bd5d1c1885bc2b8228d69903f0cb938c13c8477d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5856_b6779a50beeb70457001.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5888_677c592f3144b99e7669.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5888_677c592f3144b99e7669.png
new file mode 100644
index 0000000000000000000000000000000000000000..3cb18701ca280a3641f68378a1ac813ab8ed2cb2
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5888_677c592f3144b99e7669.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5920_99555eee3f154efaca55.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5920_99555eee3f154efaca55.png
new file mode 100644
index 0000000000000000000000000000000000000000..0901862a5bc0de476e4c98e90a4f0100dab48d09
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5920_99555eee3f154efaca55.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5952_60740420b0010e731db4.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5952_60740420b0010e731db4.png
new file mode 100644
index 0000000000000000000000000000000000000000..d76fbbd647b91e7701d39a75af289eac18ff9100
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5952_60740420b0010e731db4.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5984_53bffd2ac0d0b691522f.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5984_53bffd2ac0d0b691522f.png
new file mode 100644
index 0000000000000000000000000000000000000000..852aabf317071d337b0360d8e2a2e57381290c2b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_5984_53bffd2ac0d0b691522f.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_6016_9e9db9ac4d689d5b33f2.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_6016_9e9db9ac4d689d5b33f2.png
new file mode 100644
index 0000000000000000000000000000000000000000..bd7d69f49c759ec5e83344054a4f0c88a16d9e7c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_6016_9e9db9ac4d689d5b33f2.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_6048_d51a82bd747fdf184204.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_6048_d51a82bd747fdf184204.png
new file mode 100644
index 0000000000000000000000000000000000000000..bf14d28b34ea41df7327406789c73b7b41b4ac56
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_6048_d51a82bd747fdf184204.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_608_0a7030a345f557f4e53b.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_608_0a7030a345f557f4e53b.png
new file mode 100644
index 0000000000000000000000000000000000000000..5cfff94c6c96712f935b550ff6168e1a9c1cfc4e
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_608_0a7030a345f557f4e53b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_640_321cc1cda8e09ddc535d.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_640_321cc1cda8e09ddc535d.png
new file mode 100644
index 0000000000000000000000000000000000000000..ddbfbf8035883dc3e5f2008cfff31e06e16f935b
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_640_321cc1cda8e09ddc535d.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_64_44365417ed0f100269f2.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_64_44365417ed0f100269f2.png
new file mode 100644
index 0000000000000000000000000000000000000000..fa333f10d21695b3b8b7c473133ee47d6ac107f1
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_64_44365417ed0f100269f2.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_672_a1e6210808cacf5bc676.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_672_a1e6210808cacf5bc676.png
new file mode 100644
index 0000000000000000000000000000000000000000..d9241f6f31e8ce65616022a77d58f4e8b083dc34
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_672_a1e6210808cacf5bc676.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_704_82121a03c53e7f70bb99.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_704_82121a03c53e7f70bb99.png
new file mode 100644
index 0000000000000000000000000000000000000000..0624e663e16d32b9d5402d4af2819c145119cce4
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_704_82121a03c53e7f70bb99.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_736_204affd9e17e271d7184.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_736_204affd9e17e271d7184.png
new file mode 100644
index 0000000000000000000000000000000000000000..ad1c48ad62c1c7141ce3ebf4fdc7b31ce5dff5ce
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_736_204affd9e17e271d7184.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_768_4afb788571cca34907d6.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_768_4afb788571cca34907d6.png
new file mode 100644
index 0000000000000000000000000000000000000000..e0ec509f899f53a8437f946a18c7fb9f1b0d0edd
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_768_4afb788571cca34907d6.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_800_3f2e12a2cd3801ba6806.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_800_3f2e12a2cd3801ba6806.png
new file mode 100644
index 0000000000000000000000000000000000000000..90b59daaef86d0f45048d557656807d1fbaf760c
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_800_3f2e12a2cd3801ba6806.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_832_cec7b8a5305c9014f4cf.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_832_cec7b8a5305c9014f4cf.png
new file mode 100644
index 0000000000000000000000000000000000000000..a562190f2cb11ae74e69a63c9939fbbcc29fa1f8
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_832_cec7b8a5305c9014f4cf.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_864_569223a172d6e45e4c7b.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_864_569223a172d6e45e4c7b.png
new file mode 100644
index 0000000000000000000000000000000000000000..03cf12b4910ed111dfc592d7f024dd64c741fa97
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_864_569223a172d6e45e4c7b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_896_cf37912cd54147dd6952.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_896_cf37912cd54147dd6952.png
new file mode 100644
index 0000000000000000000000000000000000000000..fd5cdf78f4535eab127c486b7d00ef1cb7cf3443
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_896_cf37912cd54147dd6952.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_928_3302559af85e82604f4b.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_928_3302559af85e82604f4b.png
new file mode 100644
index 0000000000000000000000000000000000000000..b5fd83e474993aa3fe65408e738260ec27b34cef
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_928_3302559af85e82604f4b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_960_2c7273363ad891f35cf4.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_960_2c7273363ad891f35cf4.png
new file mode 100644
index 0000000000000000000000000000000000000000..c43fbe70ad60b5b170fa08fcec6ca2b005aa625a
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_960_2c7273363ad891f35cf4.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_96_e0f888aba6ec7590e4be.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_96_e0f888aba6ec7590e4be.png
new file mode 100644
index 0000000000000000000000000000000000000000..4a1fd1c187a6a3c6968b94ef72eaf95464e15e7d
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_96_e0f888aba6ec7590e4be.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_992_b17a8ea4d7844b456a2b.png b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_992_b17a8ea4d7844b456a2b.png
new file mode 100644
index 0000000000000000000000000000000000000000..8c3b1a99cb8c7db987991e8a7e378d433ada7cdb
Binary files /dev/null and b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/images/ued/layout/levels16_img_992_b17a8ea4d7844b456a2b.png differ
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_0_2b3fc2f7ee28cc608858.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_0_2b3fc2f7ee28cc608858.gif
new file mode 100644
index 0000000000000000000000000000000000000000..b6dcd78ac2db99237b84da19e65a6d7cbc97af6f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_0_2b3fc2f7ee28cc608858.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2b3fc2f7ee28cc608858ab26dab0c5c46e1d506057f088ecfdb312fba122cc04
+size 106939
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_2016_950008b781fcaf7b9f05.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_2016_950008b781fcaf7b9f05.gif
new file mode 100644
index 0000000000000000000000000000000000000000..f47d84b5bf1385c6fc02bac705ff7677bad6ec11
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_2016_950008b781fcaf7b9f05.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:950008b781fcaf7b9f0567d67fcd648dfc4f62c3238113bf37572462f45f15b0
+size 276593
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_3040_6a9f6ba0ad5bbaedc7e1.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_3040_6a9f6ba0ad5bbaedc7e1.gif
new file mode 100644
index 0000000000000000000000000000000000000000..a26408abd10cb20b8edd8096f12c305a1b6be75f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_3040_6a9f6ba0ad5bbaedc7e1.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6a9f6ba0ad5bbaedc7e141e329e18345700c559c733ac4c6dbc201d54f1e459b
+size 282923
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_4064_60b8de8fbb57e41ea5ee.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_4064_60b8de8fbb57e41ea5ee.gif
new file mode 100644
index 0000000000000000000000000000000000000000..9e2de5e97091b8cfbd63fc9e29c52d1f55fefae9
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_4064_60b8de8fbb57e41ea5ee.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:60b8de8fbb57e41ea5ee6613a3df29d4dfa07092a3ef513cfb59d4b2956bcf9b
+size 286783
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_5088_949a784c4acec1cebdc6.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_5088_949a784c4acec1cebdc6.gif
new file mode 100644
index 0000000000000000000000000000000000000000..c497b15e9201d4d123a82632ab21ecc9e5e85c1f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_5088_949a784c4acec1cebdc6.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:949a784c4acec1cebdc62fbcaa8a7d4498de4ce575e470b08aa74219f4720192
+size 288474
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_992_ed151ca7a2d60364e9ff.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_992_ed151ca7a2d60364e9ff.gif
new file mode 100644
index 0000000000000000000000000000000000000000..8eae52f39f0aa18ee1c77041d876cd1280a77ca3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-orig/rollouts_gif_992_ed151ca7a2d60364e9ff.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ed151ca7a2d60364e9ff96ec360cef6cb8e1a52a8dd2db11861ed38f17a2edee
+size 234425
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_0_70c9cfc8d51724e65056.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_0_70c9cfc8d51724e65056.gif
new file mode 100644
index 0000000000000000000000000000000000000000..2a643f2d96946020616da6d27a8c615999085a64
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_0_70c9cfc8d51724e65056.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:70c9cfc8d51724e65056eaeedc81ce4bbf7476ebb655e2ff4ea58148d62d801b
+size 101599
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_2016_5f538c3d498acae44865.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_2016_5f538c3d498acae44865.gif
new file mode 100644
index 0000000000000000000000000000000000000000..915f8fb4bd5a101678f761eec8232ad97bf96dd7
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_2016_5f538c3d498acae44865.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5f538c3d498acae44865fcfc3db04a266ca75b6656a617a16f3a673a9a07ecff
+size 147480
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_3040_f223b4590f6d97f13770.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_3040_f223b4590f6d97f13770.gif
new file mode 100644
index 0000000000000000000000000000000000000000..e7d210d0f111a662455ee93018462d945d82a874
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_3040_f223b4590f6d97f13770.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f223b4590f6d97f13770a016506794ab1a0cb6e942b3d0408aaf4d6b87046369
+size 162459
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_4064_232b38e52a22d511bbf4.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_4064_232b38e52a22d511bbf4.gif
new file mode 100644
index 0000000000000000000000000000000000000000..04ca93c710fa7fd62fecebe34afbf69c9d917c4b
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_4064_232b38e52a22d511bbf4.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:232b38e52a22d511bbf4fc52aa8a8cb2c2f92235edf8d25de580c130a8a4c485
+size 176177
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_5088_6995668c199776ef24f5.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_5088_6995668c199776ef24f5.gif
new file mode 100644
index 0000000000000000000000000000000000000000..63d9fd6466e7b0e36ceecb94d6ca27314b26a7dd
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_5088_6995668c199776ef24f5.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6995668c199776ef24f5466ddb31e966d234723f24d6e6ab4247221bdc141d0c
+size 185136
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_992_f775e15a53e01e0be2ed.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_992_f775e15a53e01e0be2ed.gif
new file mode 100644
index 0000000000000000000000000000000000000000..330a5e42f1634b1d4672d2123fe06aefe729cfaf
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-shift/rollouts_gif_992_f775e15a53e01e0be2ed.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f775e15a53e01e0be2edbeae869e7688c708875af6987f468e5dd2aafc778479
+size 130628
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_0_b75f456da959a3e97098.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_0_b75f456da959a3e97098.gif
new file mode 100644
index 0000000000000000000000000000000000000000..179e35e78b2e6f6cd12f87a54d8b7176202f9d9e
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_0_b75f456da959a3e97098.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b75f456da959a3e97098330e34aaeaf3ef6fb6b181169e73f2e5c5e6dcfd65ea
+size 107003
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_2016_fbb8a8a9b4216197273a.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_2016_fbb8a8a9b4216197273a.gif
new file mode 100644
index 0000000000000000000000000000000000000000..96cf4a79c86ed787e6400ac25046bfe65736c2f0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_2016_fbb8a8a9b4216197273a.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fbb8a8a9b4216197273a8053a935422519bf4de140abce0171b642281ef5e956
+size 281263
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_3040_d689abf40b9f27332b3e.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_3040_d689abf40b9f27332b3e.gif
new file mode 100644
index 0000000000000000000000000000000000000000..37750de90e1a4d8d1ac68dc533f53d2903059b01
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_3040_d689abf40b9f27332b3e.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d689abf40b9f27332b3e368bb289eee13383cd25d930a3dae5c9927b92fe1c34
+size 290891
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_4064_42ab2ea3b07646e983b3.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_4064_42ab2ea3b07646e983b3.gif
new file mode 100644
index 0000000000000000000000000000000000000000..88ec13e45a1ce0483b0712f348d5087e05617153
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_4064_42ab2ea3b07646e983b3.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:42ab2ea3b07646e983b30b0cddacb048577ec02760f39e6c6871a592e05defbf
+size 292778
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_5088_2401e16bcdff1cb057e2.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_5088_2401e16bcdff1cb057e2.gif
new file mode 100644
index 0000000000000000000000000000000000000000..4d374463049c25541040a7c9e72b897691fefb47
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_5088_2401e16bcdff1cb057e2.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2401e16bcdff1cb057e2f15ca7b274142848bf0e182705c26b605eb0b31a5ebe
+size 292025
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_992_79f267376e951d66e220.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_992_79f267376e951d66e220.gif
new file mode 100644
index 0000000000000000000000000000000000000000..3aa7383174cd029ef076c36cc0c5b1b186be4d73
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/eval-batch-tree/rollouts_gif_992_79f267376e951d66e220.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:79f267376e951d66e2205c4cf3bacbc93953811964cad42c4e629e06f72b0cdb
+size 236071
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-all/rollouts_gif_0_de0ca76414e4221f684c.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-all/rollouts_gif_0_de0ca76414e4221f684c.gif
new file mode 100644
index 0000000000000000000000000000000000000000..4aedc3c2445851a5f4677b5daacfe3d76dfbb599
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-all/rollouts_gif_0_de0ca76414e4221f684c.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:de0ca76414e4221f684c1a68871f6070c2eb76ba4b6c2db46d07a87cc297e4da
+size 106595
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-all/rollouts_gif_2016_27833df9519347170912.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-all/rollouts_gif_2016_27833df9519347170912.gif
new file mode 100644
index 0000000000000000000000000000000000000000..e65f0b24c44b2c165c587d0dcb36075dd6dd6fe3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-all/rollouts_gif_2016_27833df9519347170912.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:27833df95193471709121c820cd40a8807f0fcf97d751583d68e8a2d89bf09d6
+size 155950
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-all/rollouts_gif_3040_6d081750b2a0636a7633.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-all/rollouts_gif_3040_6d081750b2a0636a7633.gif
new file mode 100644
index 0000000000000000000000000000000000000000..28c8bd83005b67d69e12c84126feaf9a5351443f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-all/rollouts_gif_3040_6d081750b2a0636a7633.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6d081750b2a0636a7633a271b298f4b5f00092f13ca5f8597ea84319f97518b6
+size 292723
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-all/rollouts_gif_4064_14950c5dc884e09b130d.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-all/rollouts_gif_4064_14950c5dc884e09b130d.gif
new file mode 100644
index 0000000000000000000000000000000000000000..7ecb5d970cc38a9492c12b4bb94c1511e7f32b39
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-all/rollouts_gif_4064_14950c5dc884e09b130d.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:14950c5dc884e09b130d0db1ec62f29d6b65619f8f1f5a0aa30383515c54dc6a
+size 295898
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-all/rollouts_gif_5088_b3d811352e422cf93cc6.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-all/rollouts_gif_5088_b3d811352e422cf93cc6.gif
new file mode 100644
index 0000000000000000000000000000000000000000..5fe338561e5eb0682b681d05614aa4c93d7cc35c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-all/rollouts_gif_5088_b3d811352e422cf93cc6.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b3d811352e422cf93cc6e256c9e504c0347fed15b61813fd2c39ee4497c9eda9
+size 164928
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-all/rollouts_gif_992_4b91e1cbe9b6c9e185f4.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-all/rollouts_gif_992_4b91e1cbe9b6c9e185f4.gif
new file mode 100644
index 0000000000000000000000000000000000000000..e5f3a09e568967751eb3eff30a46606b79212ba0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-all/rollouts_gif_992_4b91e1cbe9b6c9e185f4.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4b91e1cbe9b6c9e185f4e389a16ab6984356f7922f6ae43af802dfb34c0c0e69
+size 223327
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-generate/rollouts_gif_3040_6d081750b2a0636a7633.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-generate/rollouts_gif_3040_6d081750b2a0636a7633.gif
new file mode 100644
index 0000000000000000000000000000000000000000..28c8bd83005b67d69e12c84126feaf9a5351443f
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-generate/rollouts_gif_3040_6d081750b2a0636a7633.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6d081750b2a0636a7633a271b298f4b5f00092f13ca5f8597ea84319f97518b6
+size 292723
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-generate/rollouts_gif_4064_14950c5dc884e09b130d.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-generate/rollouts_gif_4064_14950c5dc884e09b130d.gif
new file mode 100644
index 0000000000000000000000000000000000000000..7ecb5d970cc38a9492c12b4bb94c1511e7f32b39
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-generate/rollouts_gif_4064_14950c5dc884e09b130d.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:14950c5dc884e09b130d0db1ec62f29d6b65619f8f1f5a0aa30383515c54dc6a
+size 295898
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-generate/rollouts_gif_992_4b91e1cbe9b6c9e185f4.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-generate/rollouts_gif_992_4b91e1cbe9b6c9e185f4.gif
new file mode 100644
index 0000000000000000000000000000000000000000..e5f3a09e568967751eb3eff30a46606b79212ba0
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-generate/rollouts_gif_992_4b91e1cbe9b6c9e185f4.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4b91e1cbe9b6c9e185f4e389a16ab6984356f7922f6ae43af802dfb34c0c0e69
+size 223327
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-replay/rollouts_gif_0_de0ca76414e4221f684c.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-replay/rollouts_gif_0_de0ca76414e4221f684c.gif
new file mode 100644
index 0000000000000000000000000000000000000000..4aedc3c2445851a5f4677b5daacfe3d76dfbb599
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-replay/rollouts_gif_0_de0ca76414e4221f684c.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:de0ca76414e4221f684c1a68871f6070c2eb76ba4b6c2db46d07a87cc297e4da
+size 106595
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-replay/rollouts_gif_2016_27833df9519347170912.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-replay/rollouts_gif_2016_27833df9519347170912.gif
new file mode 100644
index 0000000000000000000000000000000000000000..e65f0b24c44b2c165c587d0dcb36075dd6dd6fe3
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-replay/rollouts_gif_2016_27833df9519347170912.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:27833df95193471709121c820cd40a8807f0fcf97d751583d68e8a2d89bf09d6
+size 155950
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-replay/rollouts_gif_5088_b3d811352e422cf93cc6.gif b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-replay/rollouts_gif_5088_b3d811352e422cf93cc6.gif
new file mode 100644
index 0000000000000000000000000000000000000000..5fe338561e5eb0682b681d05614aa4c93d7cc35c
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/media/videos/train-replay/rollouts_gif_5088_b3d811352e422cf93cc6.gif
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b3d811352e422cf93cc6e256c9e504c0347fed15b61813fd2c39ee4497c9eda9
+size 164928
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/output.log b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/output.log
new file mode 100644
index 0000000000000000000000000000000000000000..0c9465582e6497a37a59225c15812944bd2e8aee
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/output.log
@@ -0,0 +1,23142 @@
+  env_size                      : 13
+  env_layout                    : tree
+  wall_prob                     : 1
+  env_corner_size               : 1
+  env_terminate_after_corner    : False
+  obs_level_of_detail           : 0
+  img_level_of_detail           : 1
+  env_penalize_time             : False
+  net_cnn_type                  : large
+  net_rnn_type                  : ff
+  net_width                     : 256
+  ued                           : plr
+  prob_shift                    : 0
+  num_train_levels              : 2048
+  plr_buffer_size               : 4096
+  plr_temperature               : 0.1
+  plr_staleness_coeff           : 0.1
+  plr_prob_replay               : 0.5
+  plr_regret_estimator          : maxmc-actor
+  plr_robust                    : False
+  num_mutate_steps              : 12
+  prob_mutate_shift             : 0
+  chain_mutate                  : True
+  mutate_cheese                 : True
+  train_proxy_critic            : False
+  plr_proxy_shaping             : False
+  proxy_name                    : proxy_corner
+  plr_proxy_shaping_coeff       : 0.5
+  clipping                      : False
+  ppo_lr                        : 5e-05
+  ppo_gamma                     : 0.999
+  ppo_clip_eps                  : 0.1
+  ppo_gae_lambda                : 0.95
+  ppo_entropy_coeff             : 0.001
+  ppo_critic_coeff              : 0.5
+  ppo_proxy_critic_coeff        : 0.5
+  ppo_max_grad_norm             : 0.5
+  ppo_lr_annealing              : False
+  num_minibatches_per_epoch     : 4
+  num_epochs_per_cycle          : 5
+  num_total_env_steps           : 200000000
+  num_env_steps_per_cycle       : 128
+  num_parallel_envs             : 256
+  console_log                   : True
+  wandb_log                     : True
+  wandb_project                 : jaxgmg-wall-prob-tree
+  wandb_entity                  : None
+  wandb_group                   : None
+  wandb_name                    : wp:1.0_alpha:0
+  log_gifs                      : True
+  log_imgs                      : True
+  log_hists                     : False
+  num_cycles_per_log            : 32
+  num_cycles_per_eval           : 32
+  num_cycles_per_gifs           : 1024
+  num_cycles_per_big_eval       : 1024
+  use_fixed_eval_levels         : False
+  evals_num_env_steps           : 512
+  evals_num_levels              : 256
+  gif_grid_width                : 16
+  level_splayer                 : mouse
+  checkpointing                 : True
+  keep_all_checkpoints          : True
+  max_num_checkpoints           : 1
+  num_cycles_per_checkpoint     : 64
+  seed                          : 42
+configuring environment...
+configuring level generators...
+configuring level classifier...
+configuring level mutator...
+configuring level solver...
+configuring level metrics...
+configuring eval level generators...
+configuring parser and parsing fixed eval levels...
+(unsupported size for fixed evals)
+configuring heatmap splayer...
+seeding random number generator with seed=42...
+configuring curriculum with ued='plr'...
+configuring eval batches from 3 level generators...
+  generating 256 'orig' levels...
+  also solving generated levels...
+  generating 256 'shift' levels...
+  also solving generated levels...
+  generating 256 'tree' levels...
+  also solving generated levels...
+configuring evals for 0 fixed eval levels...
+configuring actor critic network...
+  net_cnn_type='large'
+  net_rnn_type='ff'
+  net_width=256
+  initialising network...
+  number of parameters: 198982
+initialising the checkpointer...
+setting up optimiser...
+initialising train state...
+configuring PPO updater...
+begin training loop!
+(note: first two cycles are slow due to compilation)
+                                   , ? env s
+===========================================================
+  step:
+    env-step-all-after:                         32768
+    env-step-replay-after:                      32768
+    env-step-used-after:                        32768
+    ppo-update-after:                           20
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     80.494865
+    avg_avg_return:                             0.10339448
+    avg_reward_per_step:                        0.004699707
+    proxy_corner:
+      avg_avg_return:                           0.10339448
+      avg_reward_per_step:                      0.004699707
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     80.494865
+    avg_avg_return:                             0.10339448
+    avg_reward_per_step:                        0.004699707
+    proxy_corner:
+      avg_avg_return:                           0.10339448
+      avg_reward_per_step:                      0.004699707
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  perf:
+    env_steps_per_second:                       3560.1720250203034
+    ppo_updates_per_second:                     0.4157437423375329
+    eval-batch-orig-duration:                   12.033997859805822
+    eval-batch-shift-duration:                  0.5950491260737181
+    eval-batch-tree-duration:                   0.5976375723257661
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             15.838379
+      mouse-cheese_dist_finite_wavg:            6.9991736
+      mouse-cheese_dist_solvable_avg:           15.838379
+      mouse-cheese_dist_solvable_wavg:          6.9991713
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000004
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016795
+    scoring:
+      avg_scores:                               0.0034648709
+    visit_patterns:
+      avg_first_visit_time:                     0.0
+      avg_last_visit_time:                      0.0625
+      num_replay_batches:                       1
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0029349572
+    avg_actor_approxkl3:                        0.002911567
+    avg_actor_clipfrac:                         0.13005982
+    avg_actor_loss:                             -0.002400695
+    avg_advantage:                              0.06347477
+    avg_critic_clipfrac:                        0.037219238
+    avg_critic_loss:                            0.016928833
+    avg_entropy:                                1.3444012
+    avg_grad_norm_pre_clip:                     0.42852688
+    avg_loss:                                   0.00471932
+    max:
+      max_actor_approxkl1:                      0.0066737947
+      max_actor_approxkl3:                      0.005437032
+      max_actor_clipfrac:                       0.5533447
+      max_actor_loss:                           3.1292439e-07
+      max_advantage:                            1.2171215
+      max_critic_clipfrac:                      0.4794922
+      max_critic_loss:                          0.027608579
+      max_entropy:                              1.3494227
+      max_grad_norm_pre_clip:                   1.151828
+      max_loss:                                 0.011658256
+    std:
+      std_actor_approxkl1:                      0.0016110044
+      std_actor_approxkl3:                      0.0013809955
+      std_actor_clipfrac:                       0.1780839
+      std_actor_loss:                           0.0013568623
+      std_advantage:                            0.18438998
+      std_critic_clipfrac:                      0.103337005
+      std_critic_loss:                          0.0048788344
+      std_entropy:                              0.0028000632
+      std_grad_norm_pre_clip:                   0.24054213
+      std_loss:                                 0.0027498337
+  eval-batch-orig:
+    avg_avg_episode_length:                     80.49937
+    avg_avg_return:                             0.124170065
+    avg_benchmark_regret:                       0.8606565
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.0041122437
+    proxy_corner:
+      avg_avg_return:                           0.124170065
+      avg_reward_per_step:                      0.0041122437
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.64759
+    avg_avg_return:                             0.12322345
+    avg_benchmark_regret:                       0.863667
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.0024337769
+    proxy_corner:
+      avg_avg_return:                           0.17835692
+      avg_reward_per_step:                      0.0015563965
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-tree:
+    avg_avg_episode_length:                     79.08523
+    avg_avg_return:                             0.1386804
+    avg_benchmark_regret:                       0.8453952
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.0040893555
+    proxy_corner:
+      avg_avg_return:                           0.1386804
+      avg_reward_per_step:                      0.0040893555
+    rollouts_gif:                               float32(132, 225, 225, 3)
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        1048576
+    env-step-replay-before:                     557056
+    env-step-used-before:                       1048576
+    ppo-update-before:                          640
+    env-step-generate-before:                   491520
+    env-step-all-after:                         1081344
+    env-step-replay-after:                      589824
+    env-step-used-after:                        1081344
+    ppo-update-after:                           660
+    env-step-generate-after:                    491520
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     66.54219
+    avg_avg_return:                             0.2628196
+    avg_reward_per_step:                        0.02999878
+    proxy_corner:
+      avg_avg_return:                           0.2628196
+      avg_reward_per_step:                      0.02999878
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     66.54219
+    avg_avg_return:                             0.2628196
+    avg_reward_per_step:                        0.02999878
+    proxy_corner:
+      avg_avg_return:                           0.2628196
+      avg_reward_per_step:                      0.02999878
+  perf:
+    env_steps_per_second:                       353786.93454285944
+    ppo_updates_per_second:                     40.09826129944978
+    eval-batch-orig-duration:                   0.5015040952712297
+    eval-batch-shift-duration:                  0.4414696777239442
+    eval-batch-tree-duration:                   0.4696525102481246
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             13.49292
+      mouse-cheese_dist_finite_wavg:            7.7321177
+      mouse-cheese_dist_solvable_avg:           13.49292
+      mouse-cheese_dist_solvable_wavg:          7.7321196
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.12954283
+    visit_patterns:
+      avg_first_visit_time:                     2.236084
+      avg_last_visit_time:                      8.889404
+      num_replay_batches:                       18
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00036464626
+    avg_actor_approxkl3:                        0.0003208454
+    avg_actor_clipfrac:                         0.00602417
+    avg_actor_loss:                             -0.00016548559
+    avg_advantage:                              0.004696792
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.013837814
+    avg_entropy:                                0.934
+    avg_grad_norm_pre_clip:                     0.3472743
+    avg_loss:                                   0.0058194213
+    max:
+      max_actor_approxkl1:                      0.0018837698
+      max_actor_approxkl3:                      0.00075096695
+      max_actor_clipfrac:                       0.059814453
+      max_actor_loss:                           0.00056375563
+      max_advantage:                            0.6580085
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.018645488
+      max_entropy:                              0.95108014
+      max_grad_norm_pre_clip:                   0.68950826
+      max_loss:                                 0.008956904
+    std:
+      std_actor_approxkl1:                      0.00045964174
+      std_actor_approxkl3:                      0.00020734148
+      std_actor_clipfrac:                       0.016341668
+      std_actor_loss:                           0.00039335882
+      std_advantage:                            0.16919822
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0017872469
+      std_entropy:                              0.008447043
+      std_grad_norm_pre_clip:                   0.157635
+      std_loss:                                 0.0010571133
+  eval-batch-orig:
+    avg_avg_episode_length:                     70.32589
+    avg_avg_return:                             0.24566272
+    avg_benchmark_regret:                       0.7391638
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.020835876
+    proxy_corner:
+      avg_avg_return:                           0.24566272
+      avg_reward_per_step:                      0.020835876
+  eval-batch-shift:
+    avg_avg_episode_length:                     83.20784
+    avg_avg_return:                             0.04397687
+    avg_benchmark_regret:                       0.9429136
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.0031204224
+    proxy_corner:
+      avg_avg_return:                           0.25774458
+      avg_reward_per_step:                      0.0020828247
+  eval-batch-tree:
+    avg_avg_episode_length:                     68.263245
+    avg_avg_return:                             0.2465524
+    avg_benchmark_regret:                       0.7375232
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.031295776
+    proxy_corner:
+      avg_avg_return:                           0.2465524
+      avg_reward_per_step:                      0.031295776
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        2097152
+    env-step-replay-before:                     1114112
+    env-step-used-before:                       2097152
+    ppo-update-before:                          1280
+    env-step-generate-before:                   983040
+    env-step-all-after:                         2129920
+    env-step-replay-after:                      1114112
+    env-step-used-after:                        2129920
+    ppo-update-after:                           1300
+    env-step-generate-after:                    1015808
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     74.29903
+    avg_avg_return:                             0.15142626
+    avg_reward_per_step:                        0.027160645
+    proxy_corner:
+      avg_avg_return:                           0.15142626
+      avg_reward_per_step:                      0.027160645
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     74.29903
+    avg_avg_return:                             0.15142626
+    avg_reward_per_step:                        0.027160645
+    proxy_corner:
+      avg_avg_return:                           0.15142626
+      avg_reward_per_step:                      0.027160645
+  perf:
+    env_steps_per_second:                       326506.03924302367
+    ppo_updates_per_second:                     41.16067722182691
+    eval-batch-orig-duration:                   0.5268678674474359
+    eval-batch-shift-duration:                  0.4852650184184313
+    eval-batch-tree-duration:                   0.4837291669100523
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             11.985107
+      mouse-cheese_dist_finite_wavg:            4.836683
+      mouse-cheese_dist_solvable_avg:           11.985107
+      mouse-cheese_dist_solvable_wavg:          4.8366823
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201682
+    scoring:
+      avg_scores:                               0.23028734
+    visit_patterns:
+      avg_first_visit_time:                     6.32666
+      avg_last_visit_time:                      24.138672
+      num_replay_batches:                       34
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0018473329
+    avg_actor_approxkl3:                        0.0017110948
+    avg_actor_clipfrac:                         0.06469116
+    avg_actor_loss:                             -0.00021815374
+    avg_advantage:                              -0.03624445
+    avg_critic_clipfrac:                        0.018737793
+    avg_critic_loss:                            0.008300194
+    avg_entropy:                                0.8943842
+    avg_grad_norm_pre_clip:                     0.85495615
+    avg_loss:                                   0.0030375586
+    max:
+      max_actor_approxkl1:                      0.0050503537
+      max_actor_approxkl3:                      0.003592052
+      max_actor_clipfrac:                       0.21520996
+      max_actor_loss:                           0.0019073039
+      max_advantage:                            0.77735305
+      max_critic_clipfrac:                      0.14807129
+      max_critic_loss:                          0.010218034
+      max_entropy:                              0.910866
+      max_grad_norm_pre_clip:                   2.519219
+      max_loss:                                 0.0053952383
+    std:
+      std_actor_approxkl1:                      0.0015592438
+      std_actor_approxkl3:                      0.001184323
+      std_actor_clipfrac:                       0.06897146
+      std_actor_loss:                           0.00080468965
+      std_advantage:                            0.12930419
+      std_critic_clipfrac:                      0.03971147
+      std_critic_loss:                          0.0008759503
+      std_entropy:                              0.010092602
+      std_grad_norm_pre_clip:                   0.6570242
+      std_loss:                                 0.0009603495
+  eval-batch-orig:
+    avg_avg_episode_length:                     70.57335
+    avg_avg_return:                             0.23014432
+    avg_benchmark_regret:                       0.7546822
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.02481842
+    proxy_corner:
+      avg_avg_return:                           0.23014432
+      avg_reward_per_step:                      0.02481842
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.927216
+    avg_avg_return:                             0.04484137
+    avg_benchmark_regret:                       0.9420491
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.0025024414
+    proxy_corner:
+      avg_avg_return:                           0.24868421
+      avg_reward_per_step:                      0.002029419
+  eval-batch-tree:
+    avg_avg_episode_length:                     68.176865
+    avg_avg_return:                             0.24057223
+    avg_benchmark_regret:                       0.7435034
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.03858185
+    proxy_corner:
+      avg_avg_return:                           0.24057223
+      avg_reward_per_step:                      0.03858185
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        3145728
+    env-step-replay-before:                     1605632
+    env-step-used-before:                       3145728
+    ppo-update-before:                          1920
+    env-step-generate-before:                   1540096
+    env-step-all-after:                         3178496
+    env-step-replay-after:                      1638400
+    env-step-used-after:                        3178496
+    ppo-update-after:                           1940
+    env-step-generate-after:                    1540096
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     52.708847
+    avg_avg_return:                             0.46219295
+    avg_reward_per_step:                        0.06384277
+    proxy_corner:
+      avg_avg_return:                           0.46219295
+      avg_reward_per_step:                      0.06384277
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     52.708847
+    avg_avg_return:                             0.46219295
+    avg_reward_per_step:                        0.06384277
+    proxy_corner:
+      avg_avg_return:                           0.46219295
+      avg_reward_per_step:                      0.06384277
+  perf:
+    env_steps_per_second:                       330705.56051054935
+    ppo_updates_per_second:                     42.018355768971254
+    eval-batch-orig-duration:                   0.46983047388494015
+    eval-batch-shift-duration:                  0.21854399237781763
+    eval-batch-tree-duration:                   0.2195331696420908
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             10.733154
+      mouse-cheese_dist_finite_wavg:            5.618304
+      mouse-cheese_dist_solvable_avg:           10.733154
+      mouse-cheese_dist_solvable_wavg:          5.618304
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016816
+    scoring:
+      avg_scores:                               0.3048209
+    visit_patterns:
+      avg_first_visit_time:                     13.561768
+      avg_last_visit_time:                      40.328613
+      num_replay_batches:                       50
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0017255424
+    avg_actor_approxkl3:                        0.0018670692
+    avg_actor_clipfrac:                         0.11460572
+    avg_actor_loss:                             0.0006781081
+    avg_advantage:                              0.023962166
+    avg_critic_clipfrac:                        0.034606934
+    avg_critic_loss:                            0.015987402
+    avg_entropy:                                0.95465994
+    avg_grad_norm_pre_clip:                     2.2269404
+    avg_loss:                                   0.0077171475
+    max:
+      max_actor_approxkl1:                      0.005196089
+      max_actor_approxkl3:                      0.0059183016
+      max_actor_clipfrac:                       0.45324707
+      max_actor_loss:                           0.0053222775
+      max_advantage:                            0.8348306
+      max_critic_clipfrac:                      0.2298584
+      max_critic_loss:                          0.022945967
+      max_entropy:                              0.9925821
+      max_grad_norm_pre_clip:                   6.592337
+      max_loss:                                 0.014304838
+    std:
+      std_actor_approxkl1:                      0.0016212339
+      std_actor_approxkl3:                      0.0016577904
+      std_actor_clipfrac:                       0.1347478
+      std_actor_loss:                           0.0017506889
+      std_advantage:                            0.1749527
+      std_critic_clipfrac:                      0.06327601
+      std_critic_loss:                          0.002486227
+      std_entropy:                              0.01809672
+      std_grad_norm_pre_clip:                   1.780836
+      std_loss:                                 0.002353092
+  eval-batch-orig:
+    avg_avg_episode_length:                     69.42557
+    avg_avg_return:                             0.2692324
+    avg_benchmark_regret:                       0.7155942
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.02747345
+    proxy_corner:
+      avg_avg_return:                           0.2692324
+      avg_reward_per_step:                      0.02747345
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.65128
+    avg_avg_return:                             0.055448335
+    avg_benchmark_regret:                       0.93144214
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.0040359497
+    proxy_corner:
+      avg_avg_return:                           0.27624553
+      avg_reward_per_step:                      0.0022659302
+  eval-batch-tree:
+    avg_avg_episode_length:                     67.09274
+    avg_avg_return:                             0.2741181
+    avg_benchmark_regret:                       0.7099575
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.0418396
+    proxy_corner:
+      avg_avg_return:                           0.2741181
+      avg_reward_per_step:                      0.0418396
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        4194304
+    env-step-replay-before:                     2162688
+    env-step-used-before:                       4194304
+    ppo-update-before:                          2560
+    env-step-generate-before:                   2031616
+    env-step-all-after:                         4227072
+    env-step-replay-after:                      2195456
+    env-step-used-after:                        4227072
+    ppo-update-after:                           2580
+    env-step-generate-after:                    2031616
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     45.051914
+    avg_avg_return:                             0.5797353
+    avg_reward_per_step:                        0.062408447
+    proxy_corner:
+      avg_avg_return:                           0.5797353
+      avg_reward_per_step:                      0.062408447
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     45.051914
+    avg_avg_return:                             0.5797353
+    avg_reward_per_step:                        0.062408447
+    proxy_corner:
+      avg_avg_return:                           0.5797353
+      avg_reward_per_step:                      0.062408447
+  perf:
+    env_steps_per_second:                       291049.6727200895
+    ppo_updates_per_second:                     42.32145289008505
+    eval-batch-orig-duration:                   0.5367870731279254
+    eval-batch-shift-duration:                  0.4970838548615575
+    eval-batch-tree-duration:                   0.4887405950576067
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             10.472168
+      mouse-cheese_dist_finite_wavg:            6.4559393
+      mouse-cheese_dist_solvable_avg:           10.472168
+      mouse-cheese_dist_solvable_wavg:          6.4559393
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.34966373
+    visit_patterns:
+      avg_first_visit_time:                     23.887451
+      avg_last_visit_time:                      57.3833
+      num_replay_batches:                       67
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00171633
+    avg_actor_approxkl3:                        0.0014919591
+    avg_actor_clipfrac:                         0.08518066
+    avg_actor_loss:                             -0.0002244454
+    avg_advantage:                              -0.008049613
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.010135863
+    avg_entropy:                                0.9545903
+    avg_grad_norm_pre_clip:                     2.3854225
+    avg_loss:                                   0.0038888964
+    max:
+      max_actor_approxkl1:                      0.0060276333
+      max_actor_approxkl3:                      0.0042453897
+      max_actor_clipfrac:                       0.32739258
+      max_actor_loss:                           0.003222391
+      max_advantage:                            0.5387052
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.01148712
+      max_entropy:                              0.99342763
+      max_grad_norm_pre_clip:                   5.798401
+      max_loss:                                 0.0075049642
+    std:
+      std_actor_approxkl1:                      0.0015661942
+      std_actor_approxkl3:                      0.0011041682
+      std_actor_clipfrac:                       0.08497092
+      std_actor_loss:                           0.0013977024
+      std_advantage:                            0.1428775
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00086816534
+      std_entropy:                              0.028748527
+      std_grad_norm_pre_clip:                   1.7620624
+      std_loss:                                 0.0015011408
+  eval-batch-orig:
+    avg_avg_episode_length:                     64.098724
+    avg_avg_return:                             0.3527927
+    avg_benchmark_regret:                       0.63203377
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.032821655
+    proxy_corner:
+      avg_avg_return:                           0.3527927
+      avg_reward_per_step:                      0.032821655
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.72266
+    avg_avg_return:                             0.06885187
+    avg_benchmark_regret:                       0.9180386
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.00440979
+    proxy_corner:
+      avg_avg_return:                           0.31316593
+      avg_reward_per_step:                      0.0025634766
+  eval-batch-tree:
+    avg_avg_episode_length:                     63.714283
+    avg_avg_return:                             0.33940172
+    avg_benchmark_regret:                       0.6446739
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.04813385
+    proxy_corner:
+      avg_avg_return:                           0.33940172
+      avg_reward_per_step:                      0.04813385
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        5242880
+    env-step-replay-before:                     2555904
+    env-step-used-before:                       5242880
+    ppo-update-before:                          3200
+    env-step-generate-before:                   2686976
+    env-step-all-after:                         5275648
+    env-step-replay-after:                      2555904
+    env-step-used-after:                        5275648
+    ppo-update-after:                           3220
+    env-step-generate-after:                    2719744
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     63.741623
+    avg_avg_return:                             0.3129313
+    avg_reward_per_step:                        0.05432129
+    proxy_corner:
+      avg_avg_return:                           0.3129313
+      avg_reward_per_step:                      0.05432129
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     63.741623
+    avg_avg_return:                             0.3129313
+    avg_reward_per_step:                        0.05432129
+    proxy_corner:
+      avg_avg_return:                           0.3129313
+      avg_reward_per_step:                      0.05432129
+  perf:
+    env_steps_per_second:                       746920.1194072841
+    ppo_updates_per_second:                     42.33151415576021
+    eval-batch-orig-duration:                   0.3080838741734624
+    eval-batch-shift-duration:                  0.2583492947742343
+    eval-batch-tree-duration:                   0.2465019328519702
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             11.337891
+      mouse-cheese_dist_finite_wavg:            6.5038996
+      mouse-cheese_dist_solvable_avg:           11.337891
+      mouse-cheese_dist_solvable_wavg:          6.5038996
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.38586617
+    visit_patterns:
+      avg_first_visit_time:                     34.239014
+      avg_last_visit_time:                      68.24341
+      num_replay_batches:                       78
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.003439407
+    avg_actor_approxkl3:                        0.0031793886
+    avg_actor_clipfrac:                         0.15230714
+    avg_actor_loss:                             0.0026030913
+    avg_advantage:                              -0.04649874
+    avg_critic_clipfrac:                        0.07961426
+    avg_critic_loss:                            0.013753238
+    avg_entropy:                                0.9005763
+    avg_grad_norm_pre_clip:                     6.875793
+    avg_loss:                                   0.008579135
+    max:
+      max_actor_approxkl1:                      0.009601086
+      max_actor_approxkl3:                      0.008063984
+      max_actor_clipfrac:                       0.31433105
+      max_actor_loss:                           0.0088214725
+      max_advantage:                            0.58329177
+      max_critic_clipfrac:                      0.5136719
+      max_critic_loss:                          0.01536608
+      max_entropy:                              0.9443445
+      max_grad_norm_pre_clip:                   12.746595
+      max_loss:                                 0.014871972
+    std:
+      std_actor_approxkl1:                      0.0025137053
+      std_actor_approxkl3:                      0.0023508891
+      std_actor_clipfrac:                       0.10139535
+      std_actor_loss:                           0.003203453
+      std_advantage:                            0.16310523
+      std_critic_clipfrac:                      0.16627201
+      std_critic_loss:                          0.0007307036
+      std_entropy:                              0.03401658
+      std_grad_norm_pre_clip:                   3.8453875
+      std_loss:                                 0.0031745718
+  eval-batch-orig:
+    avg_avg_episode_length:                     62.25938
+    avg_avg_return:                             0.363056
+    avg_benchmark_regret:                       0.6217705
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.038124084
+    proxy_corner:
+      avg_avg_return:                           0.363056
+      avg_reward_per_step:                      0.038124084
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.590775
+    avg_avg_return:                             0.049355283
+    avg_benchmark_regret:                       0.93753517
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.0058898926
+    proxy_corner:
+      avg_avg_return:                           0.30613098
+      avg_reward_per_step:                      0.0024871826
+  eval-batch-tree:
+    avg_avg_episode_length:                     63.050514
+    avg_avg_return:                             0.3303935
+    avg_benchmark_regret:                       0.6536821
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.05329132
+    proxy_corner:
+      avg_avg_return:                           0.3303935
+      avg_reward_per_step:                      0.05329132
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        6291456
+    env-step-replay-before:                     3112960
+    env-step-used-before:                       6291456
+    ppo-update-before:                          3840
+    env-step-generate-before:                   3178496
+    env-step-all-after:                         6324224
+    env-step-replay-after:                      3112960
+    env-step-used-after:                        6324224
+    ppo-update-after:                           3860
+    env-step-generate-after:                    3211264
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     61.94827
+    avg_avg_return:                             0.34279287
+    avg_reward_per_step:                        0.055114746
+    proxy_corner:
+      avg_avg_return:                           0.34279287
+      avg_reward_per_step:                      0.055114746
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     61.94827
+    avg_avg_return:                             0.34279287
+    avg_reward_per_step:                        0.055114746
+    proxy_corner:
+      avg_avg_return:                           0.34279287
+      avg_reward_per_step:                      0.055114746
+  perf:
+    env_steps_per_second:                       294551.8931325062
+    ppo_updates_per_second:                     42.353735821323205
+    eval-batch-orig-duration:                   0.531104976311326
+    eval-batch-shift-duration:                  0.48505052644759417
+    eval-batch-tree-duration:                   0.4803885305300355
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             12.090332
+      mouse-cheese_dist_finite_wavg:            6.5783496
+      mouse-cheese_dist_solvable_avg:           12.090332
+      mouse-cheese_dist_solvable_wavg:          6.578351
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.99999
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.41568613
+    visit_patterns:
+      avg_first_visit_time:                     44.34131
+      avg_last_visit_time:                      85.23657
+      num_replay_batches:                       95
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0019427494
+    avg_actor_approxkl3:                        0.0019957367
+    avg_actor_clipfrac:                         0.11384888
+    avg_actor_loss:                             0.0005018115
+    avg_advantage:                              -0.029283313
+    avg_critic_clipfrac:                        0.00045776367
+    avg_critic_loss:                            0.012462164
+    avg_entropy:                                0.86739224
+    avg_grad_norm_pre_clip:                     4.3921094
+    avg_loss:                                   0.005865501
+    max:
+      max_actor_approxkl1:                      0.003961348
+      max_actor_approxkl3:                      0.004829192
+      max_actor_clipfrac:                       0.31103516
+      max_actor_loss:                           0.003204435
+      max_advantage:                            0.4641188
+      max_critic_clipfrac:                      0.004272461
+      max_critic_loss:                          0.014087057
+      max_entropy:                              0.94056773
+      max_grad_norm_pre_clip:                   8.834576
+      max_loss:                                 0.008510928
+    std:
+      std_actor_approxkl1:                      0.001288335
+      std_actor_approxkl3:                      0.0012799584
+      std_actor_clipfrac:                       0.087227575
+      std_actor_loss:                           0.0011533496
+      std_advantage:                            0.1588022
+      std_critic_clipfrac:                      0.0010710757
+      std_critic_loss:                          0.0006604505
+      std_entropy:                              0.031920202
+      std_grad_norm_pre_clip:                   2.3465421
+      std_loss:                                 0.0011710379
+  eval-batch-orig:
+    avg_avg_episode_length:                     60.526234
+    avg_avg_return:                             0.39334387
+    avg_benchmark_regret:                       0.59148264
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.04009247
+    proxy_corner:
+      avg_avg_return:                           0.39334387
+      avg_reward_per_step:                      0.04009247
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.93855
+    avg_avg_return:                             0.055571176
+    avg_benchmark_regret:                       0.93131924
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.0067825317
+    proxy_corner:
+      avg_avg_return:                           0.3239562
+      avg_reward_per_step:                      0.002632141
+  eval-batch-tree:
+    avg_avg_episode_length:                     60.459724
+    avg_avg_return:                             0.3723517
+    avg_benchmark_regret:                       0.6117239
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.057182312
+    proxy_corner:
+      avg_avg_return:                           0.3723517
+      avg_reward_per_step:                      0.057182312
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        7340032
+    env-step-replay-before:                     3571712
+    env-step-used-before:                       7340032
+    ppo-update-before:                          4480
+    env-step-generate-before:                   3768320
+    env-step-all-after:                         7372800
+    env-step-replay-after:                      3571712
+    env-step-used-after:                        7372800
+    ppo-update-after:                           4500
+    env-step-generate-after:                    3801088
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     64.30904
+    avg_avg_return:                             0.3106915
+    avg_reward_per_step:                        0.03918457
+    proxy_corner:
+      avg_avg_return:                           0.3106915
+      avg_reward_per_step:                      0.03918457
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     64.30904
+    avg_avg_return:                             0.3106915
+    avg_reward_per_step:                        0.03918457
+    proxy_corner:
+      avg_avg_return:                           0.3106915
+      avg_reward_per_step:                      0.03918457
+  perf:
+    env_steps_per_second:                       306194.31038986874
+    ppo_updates_per_second:                     42.37437806370566
+    eval-batch-orig-duration:                   0.5293671945109963
+    eval-batch-shift-duration:                  0.4825625941157341
+    eval-batch-tree-duration:                   0.48483741749078035
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             12.836914
+      mouse-cheese_dist_finite_wavg:            6.6607046
+      mouse-cheese_dist_solvable_avg:           12.836914
+      mouse-cheese_dist_solvable_wavg:          6.6607046
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.99999
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.4529291
+    visit_patterns:
+      avg_first_visit_time:                     55.145996
+      avg_last_visit_time:                      99.263916
+      num_replay_batches:                       109
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0023722192
+    avg_actor_approxkl3:                        0.0023392022
+    avg_actor_clipfrac:                         0.12844239
+    avg_actor_loss:                             0.0015406996
+    avg_advantage:                              -0.045658678
+    avg_critic_clipfrac:                        0.1605957
+    avg_critic_loss:                            0.014805985
+    avg_entropy:                                0.89154434
+    avg_grad_norm_pre_clip:                     8.222746
+    avg_loss:                                   0.008052147
+    max:
+      max_actor_approxkl1:                      0.0076468573
+      max_actor_approxkl3:                      0.0067007104
+      max_actor_clipfrac:                       0.36242676
+      max_actor_loss:                           0.0065279007
+      max_advantage:                            0.503489
+      max_critic_clipfrac:                      0.98876953
+      max_critic_loss:                          0.017752433
+      max_entropy:                              0.98343277
+      max_grad_norm_pre_clip:                   17.502008
+      max_loss:                                 0.012853396
+    std:
+      std_actor_approxkl1:                      0.0018779662
+      std_actor_approxkl3:                      0.0016870341
+      std_actor_clipfrac:                       0.102916
+      std_actor_loss:                           0.0024373932
+      std_advantage:                            0.16750742
+      std_critic_clipfrac:                      0.33470052
+      std_critic_loss:                          0.0012948967
+      std_entropy:                              0.041206162
+      std_grad_norm_pre_clip:                   5.256372
+      std_loss:                                 0.0024934013
+  eval-batch-orig:
+    avg_avg_episode_length:                     60.248703
+    avg_avg_return:                             0.39663935
+    avg_benchmark_regret:                       0.58818716
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.04156494
+    proxy_corner:
+      avg_avg_return:                           0.39663935
+      avg_reward_per_step:                      0.04156494
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.76452
+    avg_avg_return:                             0.058007408
+    avg_benchmark_regret:                       0.928883
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.0068588257
+    proxy_corner:
+      avg_avg_return:                           0.33921015
+      avg_reward_per_step:                      0.0027542114
+  eval-batch-tree:
+    avg_avg_episode_length:                     60.01274
+    avg_avg_return:                             0.37749538
+    avg_benchmark_regret:                       0.60658026
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.05844879
+    proxy_corner:
+      avg_avg_return:                           0.37749538
+      avg_reward_per_step:                      0.05844879
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        8388608
+    env-step-replay-before:                     4063232
+    env-step-used-before:                       8388608
+    ppo-update-before:                          5120
+    env-step-generate-before:                   4325376
+    env-step-all-after:                         8421376
+    env-step-replay-after:                      4096000
+    env-step-used-after:                        8421376
+    ppo-update-after:                           5140
+    env-step-generate-after:                    4325376
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.67176
+    avg_avg_return:                             0.45899242
+    avg_reward_per_step:                        0.022705078
+    proxy_corner:
+      avg_avg_return:                           0.45899242
+      avg_reward_per_step:                      0.022705078
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.67176
+    avg_avg_return:                             0.45899242
+    avg_reward_per_step:                        0.022705078
+    proxy_corner:
+      avg_avg_return:                           0.45899242
+      avg_reward_per_step:                      0.022705078
+  perf:
+    env_steps_per_second:                       351122.25244011864
+    ppo_updates_per_second:                     41.476633554489716
+    eval-batch-orig-duration:                   0.5232251696288586
+    eval-batch-shift-duration:                  0.44560626707971096
+    eval-batch-tree-duration:                   0.4547659680247307
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             13.301514
+      mouse-cheese_dist_finite_wavg:            6.7179546
+      mouse-cheese_dist_solvable_avg:           13.301514
+      mouse-cheese_dist_solvable_wavg:          6.7179556
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.420168
+    scoring:
+      avg_scores:                               0.509555
+    visit_patterns:
+      avg_first_visit_time:                     64.22485
+      avg_last_visit_time:                      115.26489
+      num_replay_batches:                       125
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0021773672
+    avg_actor_approxkl3:                        0.0026184588
+    avg_actor_clipfrac:                         0.12665406
+    avg_actor_loss:                             0.0021629327
+    avg_advantage:                              0.058087908
+    avg_critic_clipfrac:                        0.047418214
+    avg_critic_loss:                            0.014866121
+    avg_entropy:                                0.76986235
+    avg_grad_norm_pre_clip:                     7.2793703
+    avg_loss:                                   0.008826131
+    max:
+      max_actor_approxkl1:                      0.009286702
+      max_actor_approxkl3:                      0.009148303
+      max_actor_clipfrac:                       0.39245605
+      max_actor_loss:                           0.014810167
+      max_advantage:                            0.63180625
+      max_critic_clipfrac:                      0.6015625
+      max_critic_loss:                          0.016193513
+      max_entropy:                              0.8135746
+      max_grad_norm_pre_clip:                   21.083908
+      max_loss:                                 0.02158022
+    std:
+      std_actor_approxkl1:                      0.0023386343
+      std_actor_approxkl3:                      0.0026828197
+      std_actor_clipfrac:                       0.124029234
+      std_actor_loss:                           0.0046245097
+      std_advantage:                            0.16939475
+      std_critic_clipfrac:                      0.13526826
+      std_critic_loss:                          0.00075621885
+      std_entropy:                              0.031684432
+      std_grad_norm_pre_clip:                   6.227156
+      std_loss:                                 0.004749612
+  eval-batch-orig:
+    avg_avg_episode_length:                     61.73893
+    avg_avg_return:                             0.36513954
+    avg_benchmark_regret:                       0.61968696
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.04220581
+    proxy_corner:
+      avg_avg_return:                           0.36513954
+      avg_reward_per_step:                      0.04220581
+  eval-batch-shift:
+    avg_avg_episode_length:                     82.308716
+    avg_avg_return:                             0.048899278
+    avg_benchmark_regret:                       0.93799114
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.007835388
+    proxy_corner:
+      avg_avg_return:                           0.29471135
+      avg_reward_per_step:                      0.002380371
+  eval-batch-tree:
+    avg_avg_episode_length:                     61.148033
+    avg_avg_return:                             0.36198652
+    avg_benchmark_regret:                       0.6220891
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.05844879
+    proxy_corner:
+      avg_avg_return:                           0.36198652
+      avg_reward_per_step:                      0.05844879
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        9437184
+    env-step-replay-before:                     4653056
+    env-step-used-before:                       9437184
+    ppo-update-before:                          5760
+    env-step-generate-before:                   4784128
+    env-step-all-after:                         9469952
+    env-step-replay-after:                      4653056
+    env-step-used-after:                        9469952
+    ppo-update-after:                           5780
+    env-step-generate-after:                    4816896
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     59.408836
+    avg_avg_return:                             0.36394173
+    avg_reward_per_step:                        0.0592041
+    proxy_corner:
+      avg_avg_return:                           0.36394173
+      avg_reward_per_step:                      0.0592041
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     59.408836
+    avg_avg_return:                             0.36394173
+    avg_reward_per_step:                        0.0592041
+    proxy_corner:
+      avg_avg_return:                           0.36394173
+      avg_reward_per_step:                      0.0592041
+  perf:
+    env_steps_per_second:                       347345.3245328137
+    ppo_updates_per_second:                     42.33783410116217
+    eval-batch-orig-duration:                   0.5133800767362118
+    eval-batch-shift-duration:                  0.49163562059402466
+    eval-batch-tree-duration:                   0.4830974414944649
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             13.686035
+      mouse-cheese_dist_finite_wavg:            6.746232
+      mouse-cheese_dist_solvable_avg:           13.686035
+      mouse-cheese_dist_solvable_wavg:          6.746233
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5310837
+    visit_patterns:
+      avg_first_visit_time:                     72.56152
+      avg_last_visit_time:                      132.2937
+      num_replay_batches:                       142
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0034198537
+    avg_actor_approxkl3:                        0.0032179428
+    avg_actor_clipfrac:                         0.1756958
+    avg_actor_loss:                             0.0035015424
+    avg_advantage:                              0.015130666
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.015064539
+    avg_entropy:                                0.7625805
+    avg_grad_norm_pre_clip:                     9.004444
+    avg_loss:                                   0.010271232
+    max:
+      max_actor_approxkl1:                      0.006996178
+      max_actor_approxkl3:                      0.00852683
+      max_actor_clipfrac:                       0.37158203
+      max_actor_loss:                           0.016089141
+      max_advantage:                            0.65112203
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.018001212
+      max_entropy:                              0.8262917
+      max_grad_norm_pre_clip:                   18.311737
+      max_loss:                                 0.023022719
+    std:
+      std_actor_approxkl1:                      0.002112074
+      std_actor_approxkl3:                      0.002219985
+      std_actor_clipfrac:                       0.11300605
+      std_actor_loss:                           0.0043961294
+      std_advantage:                            0.17190483
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0012306747
+      std_entropy:                              0.04060457
+      std_grad_norm_pre_clip:                   4.6036944
+      std_loss:                                 0.0045135356
+  eval-batch-orig:
+    avg_avg_episode_length:                     60.165012
+    avg_avg_return:                             0.37893105
+    avg_benchmark_regret:                       0.6058955
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.044052124
+    proxy_corner:
+      avg_avg_return:                           0.37893105
+      avg_reward_per_step:                      0.044052124
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.94056
+    avg_avg_return:                             0.050728094
+    avg_benchmark_regret:                       0.93616235
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.0069503784
+    proxy_corner:
+      avg_avg_return:                           0.3236248
+      avg_reward_per_step:                      0.0026168823
+  eval-batch-tree:
+    avg_avg_episode_length:                     59.39856
+    avg_avg_return:                             0.38017166
+    avg_benchmark_regret:                       0.60390395
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.062438965
+    proxy_corner:
+      avg_avg_return:                           0.38017166
+      avg_reward_per_step:                      0.062438965
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        10485760
+    env-step-replay-before:                     5210112
+    env-step-used-before:                       10485760
+    ppo-update-before:                          6400
+    env-step-generate-before:                   5275648
+    env-step-all-after:                         10518528
+    env-step-replay-after:                      5242880
+    env-step-used-after:                        10518528
+    ppo-update-after:                           6420
+    env-step-generate-after:                    5275648
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     56.096893
+    avg_avg_return:                             0.47839898
+    avg_reward_per_step:                        0.020263672
+    proxy_corner:
+      avg_avg_return:                           0.47839898
+      avg_reward_per_step:                      0.020263672
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     56.096893
+    avg_avg_return:                             0.47839898
+    avg_reward_per_step:                        0.020263672
+    proxy_corner:
+      avg_avg_return:                           0.47839898
+      avg_reward_per_step:                      0.020263672
+  perf:
+    env_steps_per_second:                       344501.28237671725
+    ppo_updates_per_second:                     42.05164477894689
+    eval-batch-orig-duration:                   0.49842988420277834
+    eval-batch-shift-duration:                  0.4308321224525571
+    eval-batch-tree-duration:                   0.4277647268027067
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             14.010986
+      mouse-cheese_dist_finite_wavg:            6.795102
+      mouse-cheese_dist_solvable_avg:           14.010986
+      mouse-cheese_dist_solvable_wavg:          6.795103
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.5108446
+    visit_patterns:
+      avg_first_visit_time:                     81.1084
+      avg_last_visit_time:                      150.34619
+      num_replay_batches:                       160
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014189518
+    avg_actor_approxkl3:                        0.001437942
+    avg_actor_clipfrac:                         0.07490235
+    avg_actor_loss:                             0.00036265777
+    avg_advantage:                              0.037575528
+    avg_critic_clipfrac:                        0.05255127
+    avg_critic_loss:                            0.013770816
+    avg_entropy:                                0.75519365
+    avg_grad_norm_pre_clip:                     6.19423
+    avg_loss:                                   0.0064928713
+    max:
+      max_actor_approxkl1:                      0.004207286
+      max_actor_approxkl3:                      0.003326929
+      max_actor_clipfrac:                       0.17297363
+      max_actor_loss:                           0.0023587942
+      max_advantage:                            0.5287355
+      max_critic_clipfrac:                      0.4958496
+      max_critic_loss:                          0.01787793
+      max_entropy:                              0.7836865
+      max_grad_norm_pre_clip:                   15.247867
+      max_loss:                                 0.009726411
+    std:
+      std_actor_approxkl1:                      0.0011612488
+      std_actor_approxkl3:                      0.00080425286
+      std_actor_clipfrac:                       0.05035962
+      std_actor_loss:                           0.0011996647
+      std_advantage:                            0.16260375
+      std_critic_clipfrac:                      0.14216402
+      std_critic_loss:                          0.0013042912
+      std_entropy:                              0.016956387
+      std_grad_norm_pre_clip:                   4.076338
+      std_loss:                                 0.001489441
+  eval-batch-orig:
+    avg_avg_episode_length:                     58.20875
+    avg_avg_return:                             0.40571496
+    avg_benchmark_regret:                       0.5791116
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.04561615
+    proxy_corner:
+      avg_avg_return:                           0.40571496
+      avg_reward_per_step:                      0.04561615
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.5934
+    avg_avg_return:                             0.05544473
+    avg_benchmark_regret:                       0.9314457
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.007118225
+    proxy_corner:
+      avg_avg_return:                           0.3446169
+      avg_reward_per_step:                      0.0027999878
+  eval-batch-tree:
+    avg_avg_episode_length:                     58.778862
+    avg_avg_return:                             0.38113946
+    avg_benchmark_regret:                       0.60293615
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.06390381
+    proxy_corner:
+      avg_avg_return:                           0.38113946
+      avg_reward_per_step:                      0.06390381
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        11534336
+    env-step-replay-before:                     5767168
+    env-step-used-before:                       11534336
+    ppo-update-before:                          7040
+    env-step-generate-before:                   5767168
+    env-step-all-after:                         11567104
+    env-step-replay-after:                      5767168
+    env-step-used-after:                        11567104
+    ppo-update-after:                           7060
+    env-step-generate-after:                    5799936
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     62.621964
+    avg_avg_return:                             0.31099188
+    avg_reward_per_step:                        0.05429077
+    proxy_corner:
+      avg_avg_return:                           0.31099188
+      avg_reward_per_step:                      0.05429077
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     62.621964
+    avg_avg_return:                             0.31099188
+    avg_reward_per_step:                        0.05429077
+    proxy_corner:
+      avg_avg_return:                           0.31099188
+      avg_reward_per_step:                      0.05429077
+  perf:
+    env_steps_per_second:                       316372.5096636015
+    ppo_updates_per_second:                     42.10000716323108
+    eval-batch-orig-duration:                   0.5109369466081262
+    eval-batch-shift-duration:                  0.48476114496588707
+    eval-batch-tree-duration:                   0.4798397058621049
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             14.364014
+      mouse-cheese_dist_finite_wavg:            6.8182006
+      mouse-cheese_dist_solvable_avg:           14.364014
+      mouse-cheese_dist_solvable_wavg:          6.8182015
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.51710105
+    visit_patterns:
+      avg_first_visit_time:                     90.61499
+      avg_last_visit_time:                      166.22461
+      num_replay_batches:                       176
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0022243357
+    avg_actor_approxkl3:                        0.0020521653
+    avg_actor_clipfrac:                         0.116131596
+    avg_actor_loss:                             0.0013969671
+    avg_advantage:                              -0.018273124
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.012678164
+    avg_entropy:                                0.73922294
+    avg_grad_norm_pre_clip:                     7.235325
+    avg_loss:                                   0.0069968263
+    max:
+      max_actor_approxkl1:                      0.0071095806
+      max_actor_approxkl3:                      0.0052463934
+      max_actor_clipfrac:                       0.2939453
+      max_actor_loss:                           0.008390047
+      max_advantage:                            0.628564
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.014599427
+      max_entropy:                              0.7994335
+      max_grad_norm_pre_clip:                   17.247192
+      max_loss:                                 0.014080304
+    std:
+      std_actor_approxkl1:                      0.0019974809
+      std_actor_approxkl3:                      0.0015721772
+      std_actor_clipfrac:                       0.09610331
+      std_actor_loss:                           0.002548654
+      std_advantage:                            0.15853707
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.001046324
+      std_entropy:                              0.03393899
+      std_grad_norm_pre_clip:                   4.29653
+      std_loss:                                 0.0026225785
+  eval-batch-orig:
+    avg_avg_episode_length:                     58.102448
+    avg_avg_return:                             0.40623674
+    avg_benchmark_regret:                       0.5785898
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.04626465
+    proxy_corner:
+      avg_avg_return:                           0.40623674
+      avg_reward_per_step:                      0.04626465
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.78651
+    avg_avg_return:                             0.05519148
+    avg_benchmark_regret:                       0.93169904
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.007713318
+    proxy_corner:
+      avg_avg_return:                           0.34256685
+      avg_reward_per_step:                      0.0027770996
+  eval-batch-tree:
+    avg_avg_episode_length:                     58.873894
+    avg_avg_return:                             0.38387418
+    avg_benchmark_regret:                       0.6002014
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.060806274
+    proxy_corner:
+      avg_avg_return:                           0.38387418
+      avg_reward_per_step:                      0.060806274
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        12582912
+    env-step-replay-before:                     6291456
+    env-step-used-before:                       12582912
+    ppo-update-before:                          7680
+    env-step-generate-before:                   6291456
+    env-step-all-after:                         12615680
+    env-step-replay-after:                      6324224
+    env-step-used-after:                        12615680
+    ppo-update-after:                           7700
+    env-step-generate-after:                    6291456
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     60.612617
+    avg_avg_return:                             0.43221825
+    avg_reward_per_step:                        0.015563965
+    proxy_corner:
+      avg_avg_return:                           0.43221825
+      avg_reward_per_step:                      0.015563965
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     60.612617
+    avg_avg_return:                             0.43221825
+    avg_reward_per_step:                        0.015563965
+    proxy_corner:
+      avg_avg_return:                           0.43221825
+      avg_reward_per_step:                      0.015563965
+  perf:
+    env_steps_per_second:                       312324.906865261
+    ppo_updates_per_second:                     42.19209656737322
+    eval-batch-orig-duration:                   0.541941449046135
+    eval-batch-shift-duration:                  0.49442027136683464
+    eval-batch-tree-duration:                   0.48107018787413836
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             14.610596
+      mouse-cheese_dist_finite_wavg:            6.8553605
+      mouse-cheese_dist_solvable_avg:           14.610596
+      mouse-cheese_dist_solvable_wavg:          6.8553615
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.5419848
+    visit_patterns:
+      avg_first_visit_time:                     100.653076
+      avg_last_visit_time:                      183.36304
+      num_replay_batches:                       193
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0020947515
+    avg_actor_approxkl3:                        0.001707664
+    avg_actor_clipfrac:                         0.10307007
+    avg_actor_loss:                             0.00017236434
+    avg_advantage:                              0.021737427
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.012547
+    avg_entropy:                                0.76896924
+    avg_grad_norm_pre_clip:                     6.644362
+    avg_loss:                                   0.005676895
+    max:
+      max_actor_approxkl1:                      0.005938545
+      max_actor_approxkl3:                      0.0032688135
+      max_actor_clipfrac:                       0.21240234
+      max_actor_loss:                           0.0035940558
+      max_advantage:                            0.59795344
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.013881194
+      max_entropy:                              0.8067794
+      max_grad_norm_pre_clip:                   14.014599
+      max_loss:                                 0.009187433
+    std:
+      std_actor_approxkl1:                      0.0016012304
+      std_actor_approxkl3:                      0.0010646213
+      std_actor_clipfrac:                       0.0767818
+      std_actor_loss:                           0.0013723664
+      std_advantage:                            0.15803151
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00065786554
+      std_entropy:                              0.023090862
+      std_grad_norm_pre_clip:                   4.0880985
+      std_loss:                                 0.0014925159
+  eval-batch-orig:
+    avg_avg_episode_length:                     59.197723
+    avg_avg_return:                             0.39182705
+    avg_benchmark_regret:                       0.59299946
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.045547485
+    proxy_corner:
+      avg_avg_return:                           0.39182705
+      avg_reward_per_step:                      0.045547485
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.61319
+    avg_avg_return:                             0.05875535
+    avg_benchmark_regret:                       0.9281351
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.009117126
+    proxy_corner:
+      avg_avg_return:                           0.36454135
+      avg_reward_per_step:                      0.002960205
+  eval-batch-tree:
+    avg_avg_episode_length:                     58.737263
+    avg_avg_return:                             0.3916384
+    avg_benchmark_regret:                       0.59243715
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.06175232
+    proxy_corner:
+      avg_avg_return:                           0.3916384
+      avg_reward_per_step:                      0.06175232
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        13631488
+    env-step-replay-before:                     6782976
+    env-step-used-before:                       13631488
+    ppo-update-before:                          8320
+    env-step-generate-before:                   6848512
+    env-step-all-after:                         13664256
+    env-step-replay-after:                      6782976
+    env-step-used-after:                        13664256
+    ppo-update-after:                           8340
+    env-step-generate-after:                    6881280
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.725647
+    avg_avg_return:                             0.3649161
+    avg_reward_per_step:                        0.06906128
+    proxy_corner:
+      avg_avg_return:                           0.3649161
+      avg_reward_per_step:                      0.06906128
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.725647
+    avg_avg_return:                             0.3649161
+    avg_reward_per_step:                        0.06906128
+    proxy_corner:
+      avg_avg_return:                           0.3649161
+      avg_reward_per_step:                      0.06906128
+  perf:
+    env_steps_per_second:                       243686.7088149715
+    ppo_updates_per_second:                     42.38076710423818
+    eval-batch-orig-duration:                   0.6604443918913603
+    eval-batch-shift-duration:                  0.6207959521561861
+    eval-batch-tree-duration:                   0.5936006614938378
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             14.874512
+      mouse-cheese_dist_finite_wavg:            6.882496
+      mouse-cheese_dist_solvable_avg:           14.874512
+      mouse-cheese_dist_solvable_wavg:          6.882496
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.55506957
+    visit_patterns:
+      avg_first_visit_time:                     112.38623
+      avg_last_visit_time:                      197.35107
+      num_replay_batches:                       207
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0026684373
+    avg_actor_approxkl3:                        0.0025859654
+    avg_actor_clipfrac:                         0.15447998
+    avg_actor_loss:                             0.0020742125
+    avg_advantage:                              0.016443774
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.014230049
+    avg_entropy:                                0.77290314
+    avg_grad_norm_pre_clip:                     6.386104
+    avg_loss:                                   0.008416333
+    max:
+      max_actor_approxkl1:                      0.0068911314
+      max_actor_approxkl3:                      0.0067743626
+      max_actor_clipfrac:                       0.34228516
+      max_actor_loss:                           0.010517567
+      max_advantage:                            0.6939635
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.016605848
+      max_entropy:                              0.8468162
+      max_grad_norm_pre_clip:                   12.335419
+      max_loss:                                 0.016866121
+    std:
+      std_actor_approxkl1:                      0.0018757341
+      std_actor_approxkl3:                      0.0016985078
+      std_actor_clipfrac:                       0.10306033
+      std_actor_loss:                           0.0027260403
+      std_advantage:                            0.16847023
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010430339
+      std_entropy:                              0.036679856
+      std_grad_norm_pre_clip:                   3.0525112
+      std_loss:                                 0.0027343286
+  eval-batch-orig:
+    avg_avg_episode_length:                     55.12657
+    avg_avg_return:                             0.44440466
+    avg_benchmark_regret:                       0.54042184
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.047561646
+    proxy_corner:
+      avg_avg_return:                           0.44440466
+      avg_reward_per_step:                      0.047561646
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.44831
+    avg_avg_return:                             0.05832835
+    avg_benchmark_regret:                       0.92856216
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.007843018
+    proxy_corner:
+      avg_avg_return:                           0.37504533
+      avg_reward_per_step:                      0.003036499
+  eval-batch-tree:
+    avg_avg_episode_length:                     56.860092
+    avg_avg_return:                             0.42125878
+    avg_benchmark_regret:                       0.5628168
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.06435394
+    proxy_corner:
+      avg_avg_return:                           0.42125878
+      avg_reward_per_step:                      0.06435394
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        14680064
+    env-step-replay-before:                     7405568
+    env-step-used-before:                       14680064
+    ppo-update-before:                          8960
+    env-step-generate-before:                   7274496
+    env-step-all-after:                         14712832
+    env-step-replay-after:                      7438336
+    env-step-used-after:                        14712832
+    ppo-update-after:                           8980
+    env-step-generate-after:                    7274496
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.27192
+    avg_avg_return:                             0.46456286
+    avg_reward_per_step:                        0.017181396
+    proxy_corner:
+      avg_avg_return:                           0.46456286
+      avg_reward_per_step:                      0.017181396
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.27192
+    avg_avg_return:                             0.46456286
+    avg_reward_per_step:                        0.017181396
+    proxy_corner:
+      avg_avg_return:                           0.46456286
+      avg_reward_per_step:                      0.017181396
+  perf:
+    env_steps_per_second:                       323070.85006211227
+    ppo_updates_per_second:                     40.88664543325926
+    eval-batch-orig-duration:                   0.5290238102898002
+    eval-batch-shift-duration:                  0.4909165455028415
+    eval-batch-tree-duration:                   0.48873831890523434
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             15.16748
+      mouse-cheese_dist_finite_wavg:            7.811465
+      mouse-cheese_dist_solvable_avg:           15.16748
+      mouse-cheese_dist_solvable_wavg:          7.8114657
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.5397242
+    visit_patterns:
+      avg_first_visit_time:                     122.5061
+      avg_last_visit_time:                      217.29468
+      num_replay_batches:                       227
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0051046014
+    avg_actor_approxkl3:                        0.005088604
+    avg_actor_clipfrac:                         0.23237304
+    avg_actor_loss:                             0.0063685062
+    avg_advantage:                              0.01635649
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.013223248
+    avg_entropy:                                0.78041357
+    avg_grad_norm_pre_clip:                     13.134758
+    avg_loss:                                   0.012199718
+    max:
+      max_actor_approxkl1:                      0.016086042
+      max_actor_approxkl3:                      0.01711062
+      max_actor_clipfrac:                       0.5272217
+      max_actor_loss:                           0.02406472
+      max_advantage:                            0.53742933
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.014523056
+      max_entropy:                              0.8202286
+      max_grad_norm_pre_clip:                   25.012936
+      max_loss:                                 0.02969593
+    std:
+      std_actor_approxkl1:                      0.004718555
+      std_actor_approxkl3:                      0.0044370736
+      std_actor_clipfrac:                       0.15527256
+      std_actor_loss:                           0.0067133782
+      std_advantage:                            0.16219936
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00058937946
+      std_entropy:                              0.022010569
+      std_grad_norm_pre_clip:                   6.9838414
+      std_loss:                                 0.0067325896
+  eval-batch-orig:
+    avg_avg_episode_length:                     55.01272
+    avg_avg_return:                             0.45723328
+    avg_benchmark_regret:                       0.52759326
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.045951843
+    proxy_corner:
+      avg_avg_return:                           0.45723328
+      avg_reward_per_step:                      0.045951843
+  eval-batch-shift:
+    avg_avg_episode_length:                     81.00347
+    avg_avg_return:                             0.06390725
+    avg_benchmark_regret:                       0.9229832
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.009086609
+    proxy_corner:
+      avg_avg_return:                           0.38387927
+      avg_reward_per_step:                      0.0030822754
+  eval-batch-tree:
+    avg_avg_episode_length:                     56.16791
+    avg_avg_return:                             0.43450654
+    avg_benchmark_regret:                       0.54956913
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.062561035
+    proxy_corner:
+      avg_avg_return:                           0.43450654
+      avg_reward_per_step:                      0.062561035
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        15728640
+    env-step-replay-before:                     7766016
+    env-step-used-before:                       15728640
+    ppo-update-before:                          9600
+    env-step-generate-before:                   7962624
+    env-step-all-after:                         15761408
+    env-step-replay-after:                      7766016
+    env-step-used-after:                        15761408
+    ppo-update-after:                           9620
+    env-step-generate-after:                    7995392
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     56.099358
+    avg_avg_return:                             0.40337175
+    avg_reward_per_step:                        0.05758667
+    proxy_corner:
+      avg_avg_return:                           0.40337175
+      avg_reward_per_step:                      0.05758667
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     56.099358
+    avg_avg_return:                             0.40337175
+    avg_reward_per_step:                        0.05758667
+    proxy_corner:
+      avg_avg_return:                           0.40337175
+      avg_reward_per_step:                      0.05758667
+  perf:
+    env_steps_per_second:                       349668.54824033775
+    ppo_updates_per_second:                     42.23169030387619
+    eval-batch-orig-duration:                   0.5164598990231752
+    eval-batch-shift-duration:                  0.4845289122313261
+    eval-batch-tree-duration:                   0.48360205348581076
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             15.483154
+      mouse-cheese_dist_finite_wavg:            7.8314734
+      mouse-cheese_dist_solvable_avg:           15.483154
+      mouse-cheese_dist_solvable_wavg:          7.8314734
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5506166
+    visit_patterns:
+      avg_first_visit_time:                     138.53198
+      avg_last_visit_time:                      227.23633
+      num_replay_batches:                       237
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002667624
+    avg_actor_approxkl3:                        0.0026817517
+    avg_actor_clipfrac:                         0.15176392
+    avg_actor_loss:                             0.00159797
+    avg_advantage:                              -0.0077640316
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.013866869
+    avg_entropy:                                0.7502318
+    avg_grad_norm_pre_clip:                     7.8006425
+    avg_loss:                                   0.0077811712
+    max:
+      max_actor_approxkl1:                      0.005128649
+      max_actor_approxkl3:                      0.0067436253
+      max_actor_clipfrac:                       0.3109131
+      max_actor_loss:                           0.007180646
+      max_advantage:                            0.6406853
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0155136995
+      max_entropy:                              0.83557713
+      max_grad_norm_pre_clip:                   18.007738
+      max_loss:                                 0.013738308
+    std:
+      std_actor_approxkl1:                      0.001539394
+      std_actor_approxkl3:                      0.0015079203
+      std_actor_clipfrac:                       0.0815271
+      std_actor_loss:                           0.002166082
+      std_advantage:                            0.16683494
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0007589519
+      std_entropy:                              0.034354206
+      std_grad_norm_pre_clip:                   4.17335
+      std_loss:                                 0.002217306
+  eval-batch-orig:
+    avg_avg_episode_length:                     54.45859
+    avg_avg_return:                             0.4612654
+    avg_benchmark_regret:                       0.5235611
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.04790497
+    proxy_corner:
+      avg_avg_return:                           0.4612654
+      avg_reward_per_step:                      0.04790497
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.57684
+    avg_avg_return:                             0.07240867
+    avg_benchmark_regret:                       0.91448176
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.009750366
+    proxy_corner:
+      avg_avg_return:                           0.39970148
+      avg_reward_per_step:                      0.003250122
+  eval-batch-tree:
+    avg_avg_episode_length:                     55.183224
+    avg_avg_return:                             0.45081365
+    avg_benchmark_regret:                       0.533262
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.06515503
+    proxy_corner:
+      avg_avg_return:                           0.45081365
+      avg_reward_per_step:                      0.06515503
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        16777216
+    env-step-replay-before:                     8224768
+    env-step-used-before:                       16777216
+    ppo-update-before:                          10240
+    env-step-generate-before:                   8552448
+    env-step-all-after:                         16809984
+    env-step-replay-after:                      8257536
+    env-step-used-after:                        16809984
+    ppo-update-after:                           10260
+    env-step-generate-after:                    8552448
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     60.067093
+    avg_avg_return:                             0.44693422
+    avg_reward_per_step:                        0.015808105
+    proxy_corner:
+      avg_avg_return:                           0.44693422
+      avg_reward_per_step:                      0.015808105
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     60.067093
+    avg_avg_return:                             0.44693422
+    avg_reward_per_step:                        0.015808105
+    proxy_corner:
+      avg_avg_return:                           0.44693422
+      avg_reward_per_step:                      0.015808105
+  perf:
+    env_steps_per_second:                       354033.10765033367
+    ppo_updates_per_second:                     40.34083874451493
+    eval-batch-orig-duration:                   0.5170689728111029
+    eval-batch-shift-duration:                  0.44977719709277153
+    eval-batch-tree-duration:                   0.47831647749990225
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             15.636475
+      mouse-cheese_dist_finite_wavg:            7.855399
+      mouse-cheese_dist_solvable_avg:           15.636475
+      mouse-cheese_dist_solvable_wavg:          7.855399
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.563848
+    visit_patterns:
+      avg_first_visit_time:                     149.20581
+      avg_last_visit_time:                      242.26294
+      num_replay_batches:                       252
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010531888
+    avg_actor_approxkl3:                        0.0012181773
+    avg_actor_clipfrac:                         0.061773684
+    avg_actor_loss:                             -0.0004485961
+    avg_advantage:                              0.0143464785
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.012658171
+    avg_entropy:                                0.77260447
+    avg_grad_norm_pre_clip:                     4.8457117
+    avg_loss:                                   0.005107885
+    max:
+      max_actor_approxkl1:                      0.0019744192
+      max_actor_approxkl3:                      0.0021009464
+      max_actor_clipfrac:                       0.12390137
+      max_actor_loss:                           0.0018476099
+      max_advantage:                            0.51141596
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.014815249
+      max_entropy:                              0.80994284
+      max_grad_norm_pre_clip:                   10.779512
+      max_loss:                                 0.007224465
+    std:
+      std_actor_approxkl1:                      0.0005024231
+      std_actor_approxkl3:                      0.000574869
+      std_actor_clipfrac:                       0.03630927
+      std_actor_loss:                           0.0011013373
+      std_advantage:                            0.15858188
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00082705205
+      std_entropy:                              0.02082917
+      std_grad_norm_pre_clip:                   3.147811
+      std_loss:                                 0.0012436884
+  eval-batch-orig:
+    avg_avg_episode_length:                     54.722294
+    avg_avg_return:                             0.4485352
+    avg_benchmark_regret:                       0.53629136
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.050178528
+    proxy_corner:
+      avg_avg_return:                           0.4485352
+      avg_reward_per_step:                      0.050178528
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.76025
+    avg_avg_return:                             0.07048306
+    avg_benchmark_regret:                       0.91640747
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.008285522
+    proxy_corner:
+      avg_avg_return:                           0.37414563
+      avg_reward_per_step:                      0.0030441284
+  eval-batch-tree:
+    avg_avg_episode_length:                     54.77314
+    avg_avg_return:                             0.44924283
+    avg_benchmark_regret:                       0.5348328
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.066185
+    proxy_corner:
+      avg_avg_return:                           0.44924283
+      avg_reward_per_step:                      0.066185
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        17825792
+    env-step-replay-before:                     8912896
+    env-step-used-before:                       17825792
+    ppo-update-before:                          10880
+    env-step-generate-before:                   8912896
+    env-step-all-after:                         17858560
+    env-step-replay-after:                      8945664
+    env-step-used-after:                        17858560
+    ppo-update-after:                           10900
+    env-step-generate-after:                    8912896
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     66.800354
+    avg_avg_return:                             0.37533212
+    avg_reward_per_step:                        0.011444092
+    proxy_corner:
+      avg_avg_return:                           0.37533212
+      avg_reward_per_step:                      0.011444092
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     66.800354
+    avg_avg_return:                             0.37533212
+    avg_reward_per_step:                        0.011444092
+    proxy_corner:
+      avg_avg_return:                           0.37533212
+      avg_reward_per_step:                      0.011444092
+  perf:
+    env_steps_per_second:                       238668.55024181012
+    ppo_updates_per_second:                     41.771701536272566
+    eval-batch-orig-duration:                   0.5348358685150743
+    eval-batch-shift-duration:                  0.4856028752401471
+    eval-batch-tree-duration:                   0.486876362003386
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             15.85498
+      mouse-cheese_dist_finite_wavg:            7.8723726
+      mouse-cheese_dist_solvable_avg:           15.85498
+      mouse-cheese_dist_solvable_wavg:          7.8723726
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.5741096
+    visit_patterns:
+      avg_first_visit_time:                     159.56519
+      avg_last_visit_time:                      263.18457
+      num_replay_batches:                       273
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0033937616
+    avg_actor_approxkl3:                        0.0033681474
+    avg_actor_clipfrac:                         0.18190308
+    avg_actor_loss:                             0.0032896441
+    avg_advantage:                              0.012559938
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.013598858
+    avg_entropy:                                0.7875551
+    avg_grad_norm_pre_clip:                     10.051766
+    avg_loss:                                   0.009301516
+    max:
+      max_actor_approxkl1:                      0.008201994
+      max_actor_approxkl3:                      0.009752266
+      max_actor_clipfrac:                       0.40429688
+      max_actor_loss:                           0.012301013
+      max_advantage:                            0.5762858
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.015188642
+      max_entropy:                              0.82757473
+      max_grad_norm_pre_clip:                   17.717598
+      max_loss:                                 0.018457718
+    std:
+      std_actor_approxkl1:                      0.0024676474
+      std_actor_approxkl3:                      0.0024837302
+      std_actor_clipfrac:                       0.12076604
+      std_actor_loss:                           0.004087088
+      std_advantage:                            0.16457662
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.000830211
+      std_entropy:                              0.01942573
+      std_grad_norm_pre_clip:                   5.4123063
+      std_loss:                                 0.004187125
+  eval-batch-orig:
+    avg_avg_episode_length:                     53.958008
+    avg_avg_return:                             0.47241113
+    avg_benchmark_regret:                       0.5124154
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.047851562
+    proxy_corner:
+      avg_avg_return:                           0.47241113
+      avg_reward_per_step:                      0.047851562
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.584496
+    avg_avg_return:                             0.07352775
+    avg_benchmark_regret:                       0.9133627
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.008972168
+    proxy_corner:
+      avg_avg_return:                           0.39397418
+      avg_reward_per_step:                      0.0031814575
+  eval-batch-tree:
+    avg_avg_episode_length:                     54.64814
+    avg_avg_return:                             0.4570771
+    avg_benchmark_regret:                       0.5269985
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.064086914
+    proxy_corner:
+      avg_avg_return:                           0.4570771
+      avg_reward_per_step:                      0.064086914
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        18874368
+    env-step-replay-before:                     9633792
+    env-step-used-before:                       18874368
+    ppo-update-before:                          11520
+    env-step-generate-before:                   9240576
+    env-step-all-after:                         18907136
+    env-step-replay-after:                      9666560
+    env-step-used-after:                        18907136
+    ppo-update-after:                           11540
+    env-step-generate-after:                    9240576
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     65.72322
+    avg_avg_return:                             0.4140395
+    avg_reward_per_step:                        0.012512207
+    proxy_corner:
+      avg_avg_return:                           0.4140395
+      avg_reward_per_step:                      0.012512207
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     65.72322
+    avg_avg_return:                             0.4140395
+    avg_reward_per_step:                        0.012512207
+    proxy_corner:
+      avg_avg_return:                           0.4140395
+      avg_reward_per_step:                      0.012512207
+  perf:
+    env_steps_per_second:                       320011.5569011712
+    ppo_updates_per_second:                     42.09004144270936
+    eval-batch-orig-duration:                   0.5172461178153753
+    eval-batch-shift-duration:                  0.4887441899627447
+    eval-batch-tree-duration:                   0.48959123250097036
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             16.027588
+      mouse-cheese_dist_finite_wavg:            7.8959336
+      mouse-cheese_dist_solvable_avg:           16.027588
+      mouse-cheese_dist_solvable_wavg:          7.8959336
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016816
+    scoring:
+      avg_scores:                               0.5837215
+    visit_patterns:
+      avg_first_visit_time:                     169.31714
+      avg_last_visit_time:                      285.18555
+      num_replay_batches:                       295
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0052434052
+    avg_actor_approxkl3:                        0.005042456
+    avg_actor_clipfrac:                         0.2331482
+    avg_actor_loss:                             0.0059603
+    avg_advantage:                              0.024076346
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.012654381
+    avg_entropy:                                0.8215518
+    avg_grad_norm_pre_clip:                     11.67836
+    avg_loss:                                   0.01146594
+    max:
+      max_actor_approxkl1:                      0.01777049
+      max_actor_approxkl3:                      0.015890785
+      max_actor_clipfrac:                       0.5284424
+      max_actor_loss:                           0.025982417
+      max_advantage:                            0.58557856
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.01382671
+      max_entropy:                              0.8758137
+      max_grad_norm_pre_clip:                   27.174538
+      max_loss:                                 0.03156918
+    std:
+      std_actor_approxkl1:                      0.0043883733
+      std_actor_approxkl3:                      0.0042355466
+      std_actor_clipfrac:                       0.15266019
+      std_actor_loss:                           0.0070330882
+      std_advantage:                            0.15927216
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0005250353
+      std_entropy:                              0.024991117
+      std_grad_norm_pre_clip:                   7.3504434
+      std_loss:                                 0.0070619606
+  eval-batch-orig:
+    avg_avg_episode_length:                     55.5069
+    avg_avg_return:                             0.46998346
+    avg_benchmark_regret:                       0.5148431
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.04296875
+    proxy_corner:
+      avg_avg_return:                           0.46998346
+      avg_reward_per_step:                      0.04296875
+  eval-batch-shift:
+    avg_avg_episode_length:                     79.67698
+    avg_avg_return:                             0.08958527
+    avg_benchmark_regret:                       0.89730525
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.011154175
+    proxy_corner:
+      avg_avg_return:                           0.40429306
+      avg_reward_per_step:                      0.0033035278
+  eval-batch-tree:
+    avg_avg_episode_length:                     56.83514
+    avg_avg_return:                             0.4565071
+    avg_benchmark_regret:                       0.5275686
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.058929443
+    proxy_corner:
+      avg_avg_return:                           0.4565071
+      avg_reward_per_step:                      0.058929443
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        19922944
+    env-step-replay-before:                     10125312
+    env-step-used-before:                       19922944
+    ppo-update-before:                          12160
+    env-step-generate-before:                   9797632
+    env-step-all-after:                         19955712
+    env-step-replay-after:                      10158080
+    env-step-used-after:                        19955712
+    ppo-update-after:                           12180
+    env-step-generate-after:                    9797632
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     68.542435
+    avg_avg_return:                             0.35807073
+    avg_reward_per_step:                        0.010803223
+    proxy_corner:
+      avg_avg_return:                           0.35807073
+      avg_reward_per_step:                      0.010803223
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     68.542435
+    avg_avg_return:                             0.35807073
+    avg_reward_per_step:                        0.010803223
+    proxy_corner:
+      avg_avg_return:                           0.35807073
+      avg_reward_per_step:                      0.010803223
+  perf:
+    env_steps_per_second:                       292549.0804539129
+    ppo_updates_per_second:                     42.23260986922191
+    eval-batch-orig-duration:                   0.540549274533987
+    eval-batch-shift-duration:                  0.4983513643965125
+    eval-batch-tree-duration:                   0.48897106666117907
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             16.255127
+      mouse-cheese_dist_finite_wavg:            7.920561
+      mouse-cheese_dist_solvable_avg:           16.255127
+      mouse-cheese_dist_solvable_wavg:          7.920562
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.6008408
+    visit_patterns:
+      avg_first_visit_time:                     184.82788
+      avg_last_visit_time:                      300.18872
+      num_replay_batches:                       310
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0030229932
+    avg_actor_approxkl3:                        0.00298662
+    avg_actor_clipfrac:                         0.17478028
+    avg_actor_loss:                             0.0023810596
+    avg_advantage:                              0.015937636
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.014159204
+    avg_entropy:                                0.8144143
+    avg_grad_norm_pre_clip:                     8.619361
+    avg_loss:                                   0.008646246
+    max:
+      max_actor_approxkl1:                      0.006385348
+      max_actor_approxkl3:                      0.0060269805
+      max_actor_clipfrac:                       0.3560791
+      max_actor_loss:                           0.0061970428
+      max_advantage:                            0.5764478
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.01644595
+      max_entropy:                              0.85196245
+      max_grad_norm_pre_clip:                   16.821743
+      max_loss:                                 0.012577817
+    std:
+      std_actor_approxkl1:                      0.0015358018
+      std_actor_approxkl3:                      0.0013804581
+      std_actor_clipfrac:                       0.08259589
+      std_actor_loss:                           0.0018965902
+      std_advantage:                            0.16509269
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010395747
+      std_entropy:                              0.020522663
+      std_grad_norm_pre_clip:                   3.817901
+      std_loss:                                 0.0020086595
+  eval-batch-orig:
+    avg_avg_episode_length:                     54.55603
+    avg_avg_return:                             0.48420313
+    avg_benchmark_regret:                       0.5006234
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.04319763
+    proxy_corner:
+      avg_avg_return:                           0.48420313
+      avg_reward_per_step:                      0.04319763
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.49208
+    avg_avg_return:                             0.07674114
+    avg_benchmark_regret:                       0.91014934
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.009468079
+    proxy_corner:
+      avg_avg_return:                           0.40777814
+      avg_reward_per_step:                      0.003288269
+  eval-batch-tree:
+    avg_avg_episode_length:                     55.307907
+    avg_avg_return:                             0.4587087
+    avg_benchmark_regret:                       0.5253669
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.059799194
+    proxy_corner:
+      avg_avg_return:                           0.4587087
+      avg_reward_per_step:                      0.059799194
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        20971520
+    env-step-replay-before:                     10682368
+    env-step-used-before:                       20971520
+    ppo-update-before:                          12800
+    env-step-generate-before:                   10289152
+    env-step-all-after:                         21004288
+    env-step-replay-after:                      10682368
+    env-step-used-after:                        21004288
+    ppo-update-after:                           12820
+    env-step-generate-after:                    10321920
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     53.686302
+    avg_avg_return:                             0.43621868
+    avg_reward_per_step:                        0.06643677
+    proxy_corner:
+      avg_avg_return:                           0.43621868
+      avg_reward_per_step:                      0.06643677
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     53.686302
+    avg_avg_return:                             0.43621868
+    avg_reward_per_step:                        0.06643677
+    proxy_corner:
+      avg_avg_return:                           0.43621868
+      avg_reward_per_step:                      0.06643677
+  perf:
+    env_steps_per_second:                       232427.84620793097
+    ppo_updates_per_second:                     42.204267434798176
+    eval-batch-orig-duration:                   0.35409195721149445
+    eval-batch-shift-duration:                  0.2893252531066537
+    eval-batch-tree-duration:                   0.30558507703244686
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             16.427002
+      mouse-cheese_dist_finite_wavg:            7.9416227
+      mouse-cheese_dist_solvable_avg:           16.427002
+      mouse-cheese_dist_solvable_wavg:          7.9416237
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.6085415
+    visit_patterns:
+      avg_first_visit_time:                     197.86597
+      avg_last_visit_time:                      316.27637
+      num_replay_batches:                       326
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0037518793
+    avg_actor_approxkl3:                        0.0036901762
+    avg_actor_clipfrac:                         0.16097413
+    avg_actor_loss:                             0.00336302
+    avg_advantage:                              0.0017474722
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0118975425
+    avg_entropy:                                0.6940885
+    avg_grad_norm_pre_clip:                     8.519519
+    avg_loss:                                   0.008617703
+    max:
+      max_actor_approxkl1:                      0.013970179
+      max_actor_approxkl3:                      0.011379106
+      max_actor_clipfrac:                       0.44702148
+      max_actor_loss:                           0.016125351
+      max_advantage:                            0.6117257
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.012806022
+      max_entropy:                              0.7761189
+      max_grad_norm_pre_clip:                   20.405289
+      max_loss:                                 0.021325361
+    std:
+      std_actor_approxkl1:                      0.0040670037
+      std_actor_approxkl3:                      0.0037627341
+      std_actor_clipfrac:                       0.14759727
+      std_actor_loss:                           0.0058500594
+      std_advantage:                            0.15471877
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0007428257
+      std_entropy:                              0.034930468
+      std_grad_norm_pre_clip:                   6.6989307
+      std_loss:                                 0.005834117
+  eval-batch-orig:
+    avg_avg_episode_length:                     53.105885
+    avg_avg_return:                             0.47547138
+    avg_benchmark_regret:                       0.5093551
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.049171448
+    proxy_corner:
+      avg_avg_return:                           0.47547138
+      avg_reward_per_step:                      0.049171448
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.108696
+    avg_avg_return:                             0.07752809
+    avg_benchmark_regret:                       0.9093624
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.011444092
+    proxy_corner:
+      avg_avg_return:                           0.39939374
+      avg_reward_per_step:                      0.003227234
+  eval-batch-tree:
+    avg_avg_episode_length:                     53.535095
+    avg_avg_return:                             0.45972422
+    avg_benchmark_regret:                       0.52435136
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.066474915
+    proxy_corner:
+      avg_avg_return:                           0.45972422
+      avg_reward_per_step:                      0.066474915
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        22020096
+    env-step-replay-before:                     11141120
+    env-step-used-before:                       22020096
+    ppo-update-before:                          13440
+    env-step-generate-before:                   10878976
+    env-step-all-after:                         22052864
+    env-step-replay-after:                      11173888
+    env-step-used-after:                        22052864
+    ppo-update-after:                           13460
+    env-step-generate-after:                    10878976
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     63.20123
+    avg_avg_return:                             0.40728188
+    avg_reward_per_step:                        0.013397217
+    proxy_corner:
+      avg_avg_return:                           0.40728188
+      avg_reward_per_step:                      0.013397217
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     63.20123
+    avg_avg_return:                             0.40728188
+    avg_reward_per_step:                        0.013397217
+    proxy_corner:
+      avg_avg_return:                           0.40728188
+      avg_reward_per_step:                      0.013397217
+  perf:
+    env_steps_per_second:                       299077.4478789658
+    ppo_updates_per_second:                     42.274846351663804
+    eval-batch-orig-duration:                   0.5253380797803402
+    eval-batch-shift-duration:                  0.4845204362645745
+    eval-batch-tree-duration:                   0.4816617267206311
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             16.61084
+      mouse-cheese_dist_finite_wavg:            7.9531155
+      mouse-cheese_dist_solvable_avg:           16.61084
+      mouse-cheese_dist_solvable_wavg:          7.9531155
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000008
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.59337896
+    visit_patterns:
+      avg_first_visit_time:                     211.31274
+      avg_last_visit_time:                      331.19604
+      num_replay_batches:                       341
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0037719568
+    avg_actor_approxkl3:                        0.003566845
+    avg_actor_clipfrac:                         0.19403076
+    avg_actor_loss:                             0.0039280956
+    avg_advantage:                              0.019337803
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.013913606
+    avg_entropy:                                0.76227057
+    avg_grad_norm_pre_clip:                     8.747844
+    avg_loss:                                   0.010122627
+    max:
+      max_actor_approxkl1:                      0.007886882
+      max_actor_approxkl3:                      0.00761806
+      max_actor_clipfrac:                       0.3602295
+      max_actor_loss:                           0.011169776
+      max_advantage:                            0.6461767
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.015825324
+      max_entropy:                              0.7928767
+      max_grad_norm_pre_clip:                   14.414691
+      max_loss:                                 0.017981885
+    std:
+      std_actor_approxkl1:                      0.0023159022
+      std_actor_approxkl3:                      0.0021420203
+      std_actor_clipfrac:                       0.112251975
+      std_actor_loss:                           0.0036454033
+      std_advantage:                            0.16577002
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00066879083
+      std_entropy:                              0.01815893
+      std_grad_norm_pre_clip:                   4.072125
+      std_loss:                                 0.00377938
+  eval-batch-orig:
+    avg_avg_episode_length:                     52.111034
+    avg_avg_return:                             0.49626768
+    avg_benchmark_regret:                       0.48855883
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.052619934
+    proxy_corner:
+      avg_avg_return:                           0.49626768
+      avg_reward_per_step:                      0.052619934
+  eval-batch-shift:
+    avg_avg_episode_length:                     79.77295
+    avg_avg_return:                             0.08276152
+    avg_benchmark_regret:                       0.9041289
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.013252258
+    proxy_corner:
+      avg_avg_return:                           0.41807717
+      avg_reward_per_step:                      0.0033721924
+  eval-batch-tree:
+    avg_avg_episode_length:                     52.445328
+    avg_avg_return:                             0.48506528
+    avg_benchmark_regret:                       0.4990103
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.070114136
+    proxy_corner:
+      avg_avg_return:                           0.48506528
+      avg_reward_per_step:                      0.070114136
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        23068672
+    env-step-replay-before:                     11665408
+    env-step-used-before:                       23068672
+    ppo-update-before:                          14080
+    env-step-generate-before:                   11403264
+    env-step-all-after:                         23101440
+    env-step-replay-after:                      11698176
+    env-step-used-after:                        23101440
+    ppo-update-after:                           14100
+    env-step-generate-after:                    11403264
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     66.413994
+    avg_avg_return:                             0.39116794
+    avg_reward_per_step:                        0.012023926
+    proxy_corner:
+      avg_avg_return:                           0.39116794
+      avg_reward_per_step:                      0.012023926
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     66.413994
+    avg_avg_return:                             0.39116794
+    avg_reward_per_step:                        0.012023926
+    proxy_corner:
+      avg_avg_return:                           0.39116794
+      avg_reward_per_step:                      0.012023926
+  perf:
+    env_steps_per_second:                       243228.34660375206
+    ppo_updates_per_second:                     41.530403402142404
+    eval-batch-orig-duration:                   0.5377101749181747
+    eval-batch-shift-duration:                  0.4868490472435951
+    eval-batch-tree-duration:                   0.48680524807423353
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             16.693848
+      mouse-cheese_dist_finite_wavg:            7.9699326
+      mouse-cheese_dist_solvable_avg:           16.693848
+      mouse-cheese_dist_solvable_wavg:          7.9699326
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.57646954
+    visit_patterns:
+      avg_first_visit_time:                     224.4939
+      avg_last_visit_time:                      347.25586
+      num_replay_batches:                       357
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0029141482
+    avg_actor_approxkl3:                        0.0032313399
+    avg_actor_clipfrac:                         0.16798095
+    avg_actor_loss:                             0.0025785805
+    avg_advantage:                              -0.0011850265
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.011903259
+    avg_entropy:                                0.78839743
+    avg_grad_norm_pre_clip:                     9.634801
+    avg_loss:                                   0.0077418126
+    max:
+      max_actor_approxkl1:                      0.0077203955
+      max_actor_approxkl3:                      0.008271031
+      max_actor_clipfrac:                       0.3668213
+      max_actor_loss:                           0.008374773
+      max_advantage:                            0.58532083
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.013607119
+      max_entropy:                              0.8318778
+      max_grad_norm_pre_clip:                   17.929346
+      max_loss:                                 0.013348658
+    std:
+      std_actor_approxkl1:                      0.0022078026
+      std_actor_approxkl3:                      0.0021474587
+      std_actor_clipfrac:                       0.111061454
+      std_actor_loss:                           0.0028208634
+      std_advantage:                            0.15400934
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0008093367
+      std_entropy:                              0.019462302
+      std_grad_norm_pre_clip:                   5.3894486
+      std_loss:                                 0.0029609604
+  eval-batch-orig:
+    avg_avg_episode_length:                     51.49476
+    avg_avg_return:                             0.4993843
+    avg_benchmark_regret:                       0.48544222
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.05355072
+    proxy_corner:
+      avg_avg_return:                           0.4993843
+      avg_reward_per_step:                      0.05355072
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.04021
+    avg_avg_return:                             0.073235825
+    avg_benchmark_regret:                       0.9136546
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.013511658
+    proxy_corner:
+      avg_avg_return:                           0.4113967
+      avg_reward_per_step:                      0.003288269
+  eval-batch-tree:
+    avg_avg_episode_length:                     51.823586
+    avg_avg_return:                             0.49272764
+    avg_benchmark_regret:                       0.49134797
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.07056427
+    proxy_corner:
+      avg_avg_return:                           0.49272764
+      avg_reward_per_step:                      0.07056427
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        24117248
+    env-step-replay-before:                     12320768
+    env-step-used-before:                       24117248
+    ppo-update-before:                          14720
+    env-step-generate-before:                   11796480
+    env-step-all-after:                         24150016
+    env-step-replay-after:                      12353536
+    env-step-used-after:                        24150016
+    ppo-update-after:                           14740
+    env-step-generate-after:                    11796480
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     63.596836
+    avg_avg_return:                             0.41340938
+    avg_reward_per_step:                        0.013763428
+    proxy_corner:
+      avg_avg_return:                           0.41340938
+      avg_reward_per_step:                      0.013763428
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     63.596836
+    avg_avg_return:                             0.41340938
+    avg_reward_per_step:                        0.013763428
+    proxy_corner:
+      avg_avg_return:                           0.41340938
+      avg_reward_per_step:                      0.013763428
+  perf:
+    env_steps_per_second:                       314345.6962619837
+    ppo_updates_per_second:                     42.14582794379437
+    eval-batch-orig-duration:                   0.4974623704329133
+    eval-batch-shift-duration:                  0.4179072454571724
+    eval-batch-tree-duration:                   0.4238663176074624
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             16.832275
+      mouse-cheese_dist_finite_wavg:            7.9814596
+      mouse-cheese_dist_solvable_avg:           16.832275
+      mouse-cheese_dist_solvable_wavg:          7.9814596
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016816
+    scoring:
+      avg_scores:                               0.57481515
+    visit_patterns:
+      avg_first_visit_time:                     235.20337
+      avg_last_visit_time:                      367.21753
+      num_replay_batches:                       377
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0040252237
+    avg_actor_approxkl3:                        0.0040134685
+    avg_actor_clipfrac:                         0.18851928
+    avg_actor_loss:                             0.003877967
+    avg_advantage:                              0.0046152305
+    avg_critic_clipfrac:                        0.0011169434
+    avg_critic_loss:                            0.012423842
+    avg_entropy:                                0.70779246
+    avg_grad_norm_pre_clip:                     10.2872
+    avg_loss:                                   0.009382094
+    max:
+      max_actor_approxkl1:                      0.009878178
+      max_actor_approxkl3:                      0.008595416
+      max_actor_clipfrac:                       0.37036133
+      max_actor_loss:                           0.012594447
+      max_advantage:                            0.5487906
+      max_critic_clipfrac:                      0.013549805
+      max_critic_loss:                          0.013680392
+      max_entropy:                              0.75574434
+      max_grad_norm_pre_clip:                   21.833231
+      max_loss:                                 0.01782043
+    std:
+      std_actor_approxkl1:                      0.0028561596
+      std_actor_approxkl3:                      0.0026020731
+      std_actor_clipfrac:                       0.1126785
+      std_actor_loss:                           0.0037892866
+      std_advantage:                            0.15637422
+      std_critic_clipfrac:                      0.0030312482
+      std_critic_loss:                          0.0006778801
+      std_entropy:                              0.02821758
+      std_grad_norm_pre_clip:                   6.440361
+      std_loss:                                 0.0038544822
+  eval-batch-orig:
+    avg_avg_episode_length:                     51.34816
+    avg_avg_return:                             0.50028354
+    avg_benchmark_regret:                       0.48454294
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.055099487
+    proxy_corner:
+      avg_avg_return:                           0.50028354
+      avg_reward_per_step:                      0.055099487
+  eval-batch-shift:
+    avg_avg_episode_length:                     80.38094
+    avg_avg_return:                             0.07371725
+    avg_benchmark_regret:                       0.9131732
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.013771057
+    proxy_corner:
+      avg_avg_return:                           0.399253
+      avg_reward_per_step:                      0.0032043457
+  eval-batch-tree:
+    avg_avg_episode_length:                     51.84725
+    avg_avg_return:                             0.48571736
+    avg_benchmark_regret:                       0.49835828
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.0708313
+    proxy_corner:
+      avg_avg_return:                           0.48571736
+      avg_reward_per_step:                      0.0708313
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        25165824
+    env-step-replay-before:                     13041664
+    env-step-used-before:                       25165824
+    ppo-update-before:                          15360
+    env-step-generate-before:                   12124160
+    env-step-all-after:                         25198592
+    env-step-replay-after:                      13041664
+    env-step-used-after:                        25198592
+    ppo-update-after:                           15380
+    env-step-generate-after:                    12156928
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     52.89528
+    avg_avg_return:                             0.4402898
+    avg_reward_per_step:                        0.054748535
+    proxy_corner:
+      avg_avg_return:                           0.4402898
+      avg_reward_per_step:                      0.054748535
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     52.89528
+    avg_avg_return:                             0.4402898
+    avg_reward_per_step:                        0.054748535
+    proxy_corner:
+      avg_avg_return:                           0.4402898
+      avg_reward_per_step:                      0.054748535
+  perf:
+    env_steps_per_second:                       314216.2748433836
+    ppo_updates_per_second:                     42.0767063551029
+    eval-batch-orig-duration:                   0.5188473910093307
+    eval-batch-shift-duration:                  0.44226610846817493
+    eval-batch-tree-duration:                   0.4528318103402853
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             16.98706
+      mouse-cheese_dist_finite_wavg:            8.000049
+      mouse-cheese_dist_solvable_avg:           16.98706
+      mouse-cheese_dist_solvable_wavg:          8.000049
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.59749466
+    visit_patterns:
+      avg_first_visit_time:                     246.50708
+      avg_last_visit_time:                      388.3037
+      num_replay_batches:                       398
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0019111269
+    avg_actor_approxkl3:                        0.0018422002
+    avg_actor_clipfrac:                         0.1027649
+    avg_actor_loss:                             0.00107271
+    avg_advantage:                              -0.010179447
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.012702048
+    avg_entropy:                                0.63964975
+    avg_grad_norm_pre_clip:                     5.3574996
+    avg_loss:                                   0.0067840847
+    max:
+      max_actor_approxkl1:                      0.005318327
+      max_actor_approxkl3:                      0.004391608
+      max_actor_clipfrac:                       0.25195312
+      max_actor_loss:                           0.006467849
+      max_advantage:                            0.60287154
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.014495006
+      max_entropy:                              0.69511056
+      max_grad_norm_pre_clip:                   12.233702
+      max_loss:                                 0.012265768
+    std:
+      std_actor_approxkl1:                      0.0013217143
+      std_actor_approxkl3:                      0.0011576706
+      std_actor_clipfrac:                       0.070312075
+      std_actor_loss:                           0.0022266717
+      std_advantage:                            0.16072105
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0006096265
+      std_entropy:                              0.039047565
+      std_grad_norm_pre_clip:                   2.9251916
+      std_loss:                                 0.0022985248
+  eval-batch-orig:
+    avg_avg_episode_length:                     48.903347
+    avg_avg_return:                             0.53124255
+    avg_benchmark_regret:                       0.45358396
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.056900024
+    proxy_corner:
+      avg_avg_return:                           0.53124255
+      avg_reward_per_step:                      0.056900024
+  eval-batch-shift:
+    avg_avg_episode_length:                     79.96463
+    avg_avg_return:                             0.07831007
+    avg_benchmark_regret:                       0.90858036
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.0138168335
+    proxy_corner:
+      avg_avg_return:                           0.4076345
+      avg_reward_per_step:                      0.003288269
+  eval-batch-tree:
+    avg_avg_episode_length:                     50.386673
+    avg_avg_return:                             0.5063619
+    avg_benchmark_regret:                       0.4777137
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.07459259
+    proxy_corner:
+      avg_avg_return:                           0.5063619
+      avg_reward_per_step:                      0.07459259
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        26214400
+    env-step-replay-before:                     13402112
+    env-step-used-before:                       26214400
+    ppo-update-before:                          16000
+    env-step-generate-before:                   12812288
+    env-step-all-after:                         26247168
+    env-step-replay-after:                      13434880
+    env-step-used-after:                        26247168
+    ppo-update-after:                           16020
+    env-step-generate-after:                    12812288
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     63.536186
+    avg_avg_return:                             0.42784202
+    avg_reward_per_step:                        0.01449585
+    proxy_corner:
+      avg_avg_return:                           0.42784202
+      avg_reward_per_step:                      0.01449585
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     63.536186
+    avg_avg_return:                             0.42784202
+    avg_reward_per_step:                        0.01449585
+    proxy_corner:
+      avg_avg_return:                           0.42784202
+      avg_reward_per_step:                      0.01449585
+  perf:
+    env_steps_per_second:                       353590.80243542936
+    ppo_updates_per_second:                     41.39194912537413
+    eval-batch-orig-duration:                   0.47091907262802124
+    eval-batch-shift-duration:                  0.4455609358847141
+    eval-batch-tree-duration:                   0.42493594996631145
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             17.167725
+      mouse-cheese_dist_finite_wavg:            8.016518
+      mouse-cheese_dist_solvable_avg:           17.167725
+      mouse-cheese_dist_solvable_wavg:          8.016519
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.5862298
+    visit_patterns:
+      avg_first_visit_time:                     260.41064
+      avg_last_visit_time:                      400.2683
+      num_replay_batches:                       410
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012297421
+    avg_actor_approxkl3:                        0.0013396078
+    avg_actor_clipfrac:                         0.069366455
+    avg_actor_loss:                             -0.0006478362
+    avg_advantage:                              0.019985804
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.013087642
+    avg_entropy:                                0.7072015
+    avg_grad_norm_pre_clip:                     4.1928334
+    avg_loss:                                   0.005188783
+    max:
+      max_actor_approxkl1:                      0.002882305
+      max_actor_approxkl3:                      0.0025189365
+      max_actor_clipfrac:                       0.12768555
+      max_actor_loss:                           0.0007271692
+      max_advantage:                            0.64886105
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.014222186
+      max_entropy:                              0.7469977
+      max_grad_norm_pre_clip:                   9.366769
+      max_loss:                                 0.0065599517
+    std:
+      std_actor_approxkl1:                      0.0008583664
+      std_actor_approxkl3:                      0.0006797468
+      std_actor_clipfrac:                       0.040708654
+      std_actor_loss:                           0.00081276195
+      std_advantage:                            0.16170883
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0006868217
+      std_entropy:                              0.022134274
+      std_grad_norm_pre_clip:                   2.2017684
+      std_loss:                                 0.00086967554
+  eval-batch-orig:
+    avg_avg_episode_length:                     48.174774
+    avg_avg_return:                             0.5434492
+    avg_benchmark_regret:                       0.44137728
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.058670044
+    proxy_corner:
+      avg_avg_return:                           0.5434492
+      avg_reward_per_step:                      0.058670044
+  eval-batch-shift:
+    avg_avg_episode_length:                     79.890114
+    avg_avg_return:                             0.077262305
+    avg_benchmark_regret:                       0.90962815
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.014968872
+    proxy_corner:
+      avg_avg_return:                           0.4351355
+      avg_reward_per_step:                      0.003479004
+  eval-batch-tree:
+    avg_avg_episode_length:                     49.92714
+    avg_avg_return:                             0.5146376
+    avg_benchmark_regret:                       0.46943802
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.07622528
+    proxy_corner:
+      avg_avg_return:                           0.5146376
+      avg_reward_per_step:                      0.07622528
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        27262976
+    env-step-replay-before:                     14024704
+    env-step-used-before:                       27262976
+    ppo-update-before:                          16640
+    env-step-generate-before:                   13238272
+    env-step-all-after:                         27295744
+    env-step-replay-after:                      14057472
+    env-step-used-after:                        27295744
+    ppo-update-after:                           16660
+    env-step-generate-after:                    13238272
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     63.37138
+    avg_avg_return:                             0.41200173
+    avg_reward_per_step:                        0.013061523
+    proxy_corner:
+      avg_avg_return:                           0.41200173
+      avg_reward_per_step:                      0.013061523
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     63.37138
+    avg_avg_return:                             0.41200173
+    avg_reward_per_step:                        0.013061523
+    proxy_corner:
+      avg_avg_return:                           0.41200173
+      avg_reward_per_step:                      0.013061523
+  perf:
+    env_steps_per_second:                       356091.1584214127
+    ppo_updates_per_second:                     40.67531553891373
+    eval-batch-orig-duration:                   0.5229936959221959
+    eval-batch-shift-duration:                  0.4832842443138361
+    eval-batch-tree-duration:                   0.45668221544474363
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             17.387207
+      mouse-cheese_dist_finite_wavg:            8.030575
+      mouse-cheese_dist_solvable_avg:           17.387207
+      mouse-cheese_dist_solvable_wavg:          8.030577
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.56413543
+    visit_patterns:
+      avg_first_visit_time:                     273.51392
+      avg_last_visit_time:                      419.24805
+      num_replay_batches:                       429
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013483119
+    avg_actor_approxkl3:                        0.0017860321
+    avg_actor_clipfrac:                         0.099768065
+    avg_actor_loss:                             0.0004208509
+    avg_advantage:                              -0.0048750443
+    avg_critic_clipfrac:                        0.00037231445
+    avg_critic_loss:                            0.012984084
+    avg_entropy:                                0.74833435
+    avg_grad_norm_pre_clip:                     4.440545
+    avg_loss:                                   0.0061645578
+    max:
+      max_actor_approxkl1:                      0.0032694582
+      max_actor_approxkl3:                      0.004911509
+      max_actor_clipfrac:                       0.29882812
+      max_actor_loss:                           0.0067295954
+      max_advantage:                            0.5232047
+      max_critic_clipfrac:                      0.0048828125
+      max_critic_loss:                          0.015787356
+      max_entropy:                              0.80099833
+      max_grad_norm_pre_clip:                   8.058424
+      max_loss:                                 0.012403022
+    std:
+      std_actor_approxkl1:                      0.0010970361
+      std_actor_approxkl3:                      0.0010269564
+      std_actor_clipfrac:                       0.06567255
+      std_actor_loss:                           0.001953499
+      std_advantage:                            0.15618387
+      std_critic_clipfrac:                      0.0011408677
+      std_critic_loss:                          0.0012159697
+      std_entropy:                              0.024163563
+      std_grad_norm_pre_clip:                   2.057274
+      std_loss:                                 0.0021174927
+  eval-batch-orig:
+    avg_avg_episode_length:                     47.780052
+    avg_avg_return:                             0.54550487
+    avg_benchmark_regret:                       0.43932164
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.05872345
+    proxy_corner:
+      avg_avg_return:                           0.54550487
+      avg_reward_per_step:                      0.05872345
+  eval-batch-shift:
+    avg_avg_episode_length:                     79.62435
+    avg_avg_return:                             0.08701955
+    avg_benchmark_regret:                       0.89987093
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.014541626
+    proxy_corner:
+      avg_avg_return:                           0.43555504
+      avg_reward_per_step:                      0.0035171509
+  eval-batch-tree:
+    avg_avg_episode_length:                     49.233906
+    avg_avg_return:                             0.5261871
+    avg_benchmark_regret:                       0.45788848
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.07711029
+    proxy_corner:
+      avg_avg_return:                           0.5261871
+      avg_reward_per_step:                      0.07711029
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        28311552
+    env-step-replay-before:                     14581760
+    env-step-used-before:                       28311552
+    ppo-update-before:                          17280
+    env-step-generate-before:                   13729792
+    env-step-all-after:                         28344320
+    env-step-replay-after:                      14581760
+    env-step-used-after:                        28344320
+    ppo-update-after:                           17300
+    env-step-generate-after:                    13762560
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     52.473015
+    avg_avg_return:                             0.441424
+    avg_reward_per_step:                        0.060668945
+    proxy_corner:
+      avg_avg_return:                           0.441424
+      avg_reward_per_step:                      0.060668945
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     52.473015
+    avg_avg_return:                             0.441424
+    avg_reward_per_step:                        0.060668945
+    proxy_corner:
+      avg_avg_return:                           0.441424
+      avg_reward_per_step:                      0.060668945
+  perf:
+    env_steps_per_second:                       269154.31519809546
+    ppo_updates_per_second:                     41.65931856002013
+    eval-batch-orig-duration:                   0.5261635156348348
+    eval-batch-shift-duration:                  0.4797975365072489
+    eval-batch-tree-duration:                   0.431471424177289
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             17.57837
+      mouse-cheese_dist_finite_wavg:            8.051015
+      mouse-cheese_dist_solvable_avg:           17.57837
+      mouse-cheese_dist_solvable_wavg:          8.051016
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000008
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.58531
+    visit_patterns:
+      avg_first_visit_time:                     288.46558
+      avg_last_visit_time:                      435.2942
+      num_replay_batches:                       445
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0018916157
+    avg_actor_approxkl3:                        0.0018271434
+    avg_actor_clipfrac:                         0.096502684
+    avg_actor_loss:                             0.00058594573
+    avg_advantage:                              -0.012083372
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.012453993
+    avg_entropy:                                0.62466687
+    avg_grad_norm_pre_clip:                     4.6603503
+    avg_loss:                                   0.006188275
+    max:
+      max_actor_approxkl1:                      0.0049609924
+      max_actor_approxkl3:                      0.0039404593
+      max_actor_clipfrac:                       0.20458984
+      max_actor_loss:                           0.004846826
+      max_advantage:                            0.57588243
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.013952146
+      max_entropy:                              0.68630457
+      max_grad_norm_pre_clip:                   10.712258
+      max_loss:                                 0.010697808
+    std:
+      std_actor_approxkl1:                      0.0013605172
+      std_actor_approxkl3:                      0.0010340811
+      std_actor_clipfrac:                       0.05634977
+      std_actor_loss:                           0.001822551
+      std_advantage:                            0.15883975
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00094504043
+      std_entropy:                              0.03315737
+      std_grad_norm_pre_clip:                   2.599963
+      std_loss:                                 0.0018859892
+  eval-batch-orig:
+    avg_avg_episode_length:                     47.301117
+    avg_avg_return:                             0.5515356
+    avg_benchmark_regret:                       0.43329096
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.06012726
+    proxy_corner:
+      avg_avg_return:                           0.5515356
+      avg_reward_per_step:                      0.06012726
+  eval-batch-shift:
+    avg_avg_episode_length:                     79.36195
+    avg_avg_return:                             0.08794025
+    avg_benchmark_regret:                       0.8989502
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.016052246
+    proxy_corner:
+      avg_avg_return:                           0.43989336
+      avg_reward_per_step:                      0.0035705566
+  eval-batch-tree:
+    avg_avg_episode_length:                     48.44923
+    avg_avg_return:                             0.5331904
+    avg_benchmark_regret:                       0.45088515
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.07771301
+    proxy_corner:
+      avg_avg_return:                           0.5331904
+      avg_reward_per_step:                      0.07771301
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        29360128
+    env-step-replay-before:                     15171584
+    env-step-used-before:                       29360128
+    ppo-update-before:                          17920
+    env-step-generate-before:                   14188544
+    env-step-all-after:                         29392896
+    env-step-replay-after:                      15171584
+    env-step-used-after:                        29392896
+    ppo-update-after:                           17940
+    env-step-generate-after:                    14221312
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     51.24479
+    avg_avg_return:                             0.47528234
+    avg_reward_per_step:                        0.061187744
+    proxy_corner:
+      avg_avg_return:                           0.47528234
+      avg_reward_per_step:                      0.061187744
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     51.24479
+    avg_avg_return:                             0.47528234
+    avg_reward_per_step:                        0.061187744
+    proxy_corner:
+      avg_avg_return:                           0.47528234
+      avg_reward_per_step:                      0.061187744
+  perf:
+    env_steps_per_second:                       347192.93610396236
+    ppo_updates_per_second:                     42.2028422630927
+    eval-batch-orig-duration:                   0.5213333684951067
+    eval-batch-shift-duration:                  0.5042875399813056
+    eval-batch-tree-duration:                   0.4871378419920802
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             17.628906
+      mouse-cheese_dist_finite_wavg:            8.055376
+      mouse-cheese_dist_solvable_avg:           17.628906
+      mouse-cheese_dist_solvable_wavg:          8.055377
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.57362634
+    visit_patterns:
+      avg_first_visit_time:                     299.58643
+      avg_last_visit_time:                      453.31934
+      num_replay_batches:                       463
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002536141
+    avg_actor_approxkl3:                        0.0024629228
+    avg_actor_clipfrac:                         0.13926391
+    avg_actor_loss:                             0.0015893568
+    avg_advantage:                              -0.008159728
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.011390346
+    avg_entropy:                                0.5811998
+    avg_grad_norm_pre_clip:                     6.1703944
+    avg_loss:                                   0.0067033297
+    max:
+      max_actor_approxkl1:                      0.0063391486
+      max_actor_approxkl3:                      0.0048387125
+      max_actor_clipfrac:                       0.26611328
+      max_actor_loss:                           0.005016744
+      max_advantage:                            0.62195563
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.013182556
+      max_entropy:                              0.6477731
+      max_grad_norm_pre_clip:                   9.945426
+      max_loss:                                 0.010146121
+    std:
+      std_actor_approxkl1:                      0.0015274347
+      std_actor_approxkl3:                      0.0014212432
+      std_actor_clipfrac:                       0.07898797
+      std_actor_loss:                           0.0016149239
+      std_advantage:                            0.15156546
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0009288859
+      std_entropy:                              0.03159194
+      std_grad_norm_pre_clip:                   2.3895075
+      std_loss:                                 0.0015556389
+  eval-batch-orig:
+    avg_avg_episode_length:                     47.146263
+    avg_avg_return:                             0.5515145
+    avg_benchmark_regret:                       0.433312
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.06035614
+    proxy_corner:
+      avg_avg_return:                           0.5515145
+      avg_reward_per_step:                      0.06035614
+  eval-batch-shift:
+    avg_avg_episode_length:                     79.72299
+    avg_avg_return:                             0.07979567
+    avg_benchmark_regret:                       0.9070948
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.01586914
+    proxy_corner:
+      avg_avg_return:                           0.4291588
+      avg_reward_per_step:                      0.0034637451
+  eval-batch-tree:
+    avg_avg_episode_length:                     48.187164
+    avg_avg_return:                             0.5317115
+    avg_benchmark_regret:                       0.4523641
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.078186035
+    proxy_corner:
+      avg_avg_return:                           0.5317115
+      avg_reward_per_step:                      0.078186035
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        30408704
+    env-step-replay-before:                     15761408
+    env-step-used-before:                       30408704
+    ppo-update-before:                          18560
+    env-step-generate-before:                   14647296
+    env-step-all-after:                         30441472
+    env-step-replay-after:                      15794176
+    env-step-used-after:                        30441472
+    ppo-update-after:                           18580
+    env-step-generate-after:                    14647296
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     63.606483
+    avg_avg_return:                             0.4245734
+    avg_reward_per_step:                        0.013916016
+    proxy_corner:
+      avg_avg_return:                           0.4245734
+      avg_reward_per_step:                      0.013916016
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     63.606483
+    avg_avg_return:                             0.4245734
+    avg_reward_per_step:                        0.013916016
+    proxy_corner:
+      avg_avg_return:                           0.4245734
+      avg_reward_per_step:                      0.013916016
+  perf:
+    env_steps_per_second:                       317561.61693285906
+    ppo_updates_per_second:                     42.245646805734914
+    eval-batch-orig-duration:                   0.5284019010141492
+    eval-batch-shift-duration:                  0.4852467952296138
+    eval-batch-tree-duration:                   0.4767231373116374
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             17.771973
+      mouse-cheese_dist_finite_wavg:            8.072434
+      mouse-cheese_dist_solvable_avg:           17.771973
+      mouse-cheese_dist_solvable_wavg:          8.072435
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5625048
+    visit_patterns:
+      avg_first_visit_time:                     314.32397
+      avg_last_visit_time:                      472.32642
+      num_replay_batches:                       482
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002584107
+    avg_actor_approxkl3:                        0.0025419556
+    avg_actor_clipfrac:                         0.14306031
+    avg_actor_loss:                             0.0018123839
+    avg_advantage:                              0.0010920148
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0121538
+    avg_entropy:                                0.7626008
+    avg_grad_norm_pre_clip:                     5.1430273
+    avg_loss:                                   0.007126683
+    max:
+      max_actor_approxkl1:                      0.008304963
+      max_actor_approxkl3:                      0.007566864
+      max_actor_clipfrac:                       0.3647461
+      max_actor_loss:                           0.010685466
+      max_advantage:                            0.5303831
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.01297398
+      max_entropy:                              0.7965083
+      max_grad_norm_pre_clip:                   9.040747
+      max_loss:                                 0.01567605
+    std:
+      std_actor_approxkl1:                      0.0020825923
+      std_actor_approxkl3:                      0.0017564673
+      std_actor_clipfrac:                       0.09403554
+      std_actor_loss:                           0.003121741
+      std_advantage:                            0.15607946
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00060940237
+      std_entropy:                              0.016267197
+      std_grad_norm_pre_clip:                   2.2454448
+      std_loss:                                 0.0030845215
+  eval-batch-orig:
+    avg_avg_episode_length:                     45.81962
+    avg_avg_return:                             0.5847528
+    avg_benchmark_regret:                       0.4000737
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.060760498
+    proxy_corner:
+      avg_avg_return:                           0.5847528
+      avg_reward_per_step:                      0.060760498
+  eval-batch-shift:
+    avg_avg_episode_length:                     79.01406
+    avg_avg_return:                             0.09393959
+    avg_benchmark_regret:                       0.8929509
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.017036438
+    proxy_corner:
+      avg_avg_return:                           0.45996827
+      avg_reward_per_step:                      0.0037155151
+  eval-batch-tree:
+    avg_avg_episode_length:                     47.91893
+    avg_avg_return:                             0.5518769
+    avg_benchmark_regret:                       0.43219876
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.077949524
+    proxy_corner:
+      avg_avg_return:                           0.5518769
+      avg_reward_per_step:                      0.077949524
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        31457280
+    env-step-replay-before:                     16384000
+    env-step-used-before:                       31457280
+    ppo-update-before:                          19200
+    env-step-generate-before:                   15073280
+    env-step-all-after:                         31490048
+    env-step-replay-after:                      16384000
+    env-step-used-after:                        31490048
+    ppo-update-after:                           19220
+    env-step-generate-after:                    15106048
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     48.577667
+    avg_avg_return:                             0.51091135
+    avg_reward_per_step:                        0.073272705
+    proxy_corner:
+      avg_avg_return:                           0.51091135
+      avg_reward_per_step:                      0.073272705
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     48.577667
+    avg_avg_return:                             0.51091135
+    avg_reward_per_step:                        0.073272705
+    proxy_corner:
+      avg_avg_return:                           0.51091135
+      avg_reward_per_step:                      0.073272705
+  perf:
+    env_steps_per_second:                       319605.85494249774
+    ppo_updates_per_second:                     42.282088089974025
+    eval-batch-orig-duration:                   0.5081400908529758
+    eval-batch-shift-duration:                  0.4880507839843631
+    eval-batch-tree-duration:                   0.48660023137927055
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.042725
+      mouse-cheese_dist_finite_wavg:            9.005466
+      mouse-cheese_dist_solvable_avg:           18.042725
+      mouse-cheese_dist_solvable_wavg:          9.005466
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.55645293
+    visit_patterns:
+      avg_first_visit_time:                     334.91748
+      avg_last_visit_time:                      490.38818
+      num_replay_batches:                       500
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0031141033
+    avg_actor_approxkl3:                        0.002969968
+    avg_actor_clipfrac:                         0.15527344
+    avg_actor_loss:                             0.003143706
+    avg_advantage:                              -0.011980057
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.010929796
+    avg_entropy:                                0.6258256
+    avg_grad_norm_pre_clip:                     7.8046083
+    avg_loss:                                   0.007982778
+    max:
+      max_actor_approxkl1:                      0.0077687223
+      max_actor_approxkl3:                      0.006492488
+      max_actor_clipfrac:                       0.2845459
+      max_actor_loss:                           0.009410903
+      max_advantage:                            0.43429184
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.012958937
+      max_entropy:                              0.7172288
+      max_grad_norm_pre_clip:                   14.179242
+      max_loss:                                 0.014380497
+    std:
+      std_actor_approxkl1:                      0.0018355367
+      std_actor_approxkl3:                      0.0016275776
+      std_actor_clipfrac:                       0.07349847
+      std_actor_loss:                           0.0025757414
+      std_advantage:                            0.14784153
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010451431
+      std_entropy:                              0.038028795
+      std_grad_norm_pre_clip:                   3.3964925
+      std_loss:                                 0.0028000697
+  eval-batch-orig:
+    avg_avg_episode_length:                     45.536713
+    avg_avg_return:                             0.5968759
+    avg_benchmark_regret:                       0.38795054
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.06185913
+    proxy_corner:
+      avg_avg_return:                           0.5968759
+      avg_reward_per_step:                      0.06185913
+  eval-batch-shift:
+    avg_avg_episode_length:                     78.75067
+    avg_avg_return:                             0.099891916
+    avg_benchmark_regret:                       0.88699853
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.018013
+    proxy_corner:
+      avg_avg_return:                           0.48083115
+      avg_reward_per_step:                      0.0038986206
+  eval-batch-tree:
+    avg_avg_episode_length:                     45.78151
+    avg_avg_return:                             0.5796409
+    avg_benchmark_regret:                       0.4044347
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.07975769
+    proxy_corner:
+      avg_avg_return:                           0.5796409
+      avg_reward_per_step:                      0.07975769
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        32505856
+    env-step-replay-before:                     16842752
+    env-step-used-before:                       32505856
+    ppo-update-before:                          19840
+    env-step-generate-before:                   15663104
+    env-step-all-after:                         32538624
+    env-step-replay-after:                      16875520
+    env-step-used-after:                        32538624
+    ppo-update-after:                           19860
+    env-step-generate-after:                    15663104
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     61.219917
+    avg_avg_return:                             0.44852364
+    avg_reward_per_step:                        0.015106201
+    proxy_corner:
+      avg_avg_return:                           0.44852364
+      avg_reward_per_step:                      0.015106201
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     61.219917
+    avg_avg_return:                             0.44852364
+    avg_reward_per_step:                        0.015106201
+    proxy_corner:
+      avg_avg_return:                           0.44852364
+      avg_reward_per_step:                      0.015106201
+  perf:
+    env_steps_per_second:                       315193.0028684553
+    ppo_updates_per_second:                     42.16646047323204
+    eval-batch-orig-duration:                   0.5132529838010669
+    eval-batch-shift-duration:                  0.4417741782963276
+    eval-batch-tree-duration:                   0.45864906162023544
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.26831
+      mouse-cheese_dist_finite_wavg:            9.92011
+      mouse-cheese_dist_solvable_avg:           18.26831
+      mouse-cheese_dist_solvable_wavg:          9.920112
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201682
+    scoring:
+      avg_scores:                               0.55136377
+    visit_patterns:
+      avg_first_visit_time:                     354.0652
+      avg_last_visit_time:                      505.22217
+      num_replay_batches:                       515
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0018214505
+    avg_actor_approxkl3:                        0.0017274733
+    avg_actor_clipfrac:                         0.093450926
+    avg_actor_loss:                             -0.00054733903
+    avg_advantage:                              0.014014597
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.012260217
+    avg_entropy:                                0.7614886
+    avg_grad_norm_pre_clip:                     4.921846
+    avg_loss:                                   0.0048212805
+    max:
+      max_actor_approxkl1:                      0.0049403
+      max_actor_approxkl3:                      0.0039595226
+      max_actor_clipfrac:                       0.23095703
+      max_actor_loss:                           0.0031012222
+      max_advantage:                            0.52774835
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.014081718
+      max_entropy:                              0.7939683
+      max_grad_norm_pre_clip:                   11.094205
+      max_loss:                                 0.008148386
+    std:
+      std_actor_approxkl1:                      0.0011337611
+      std_actor_approxkl3:                      0.0008998783
+      std_actor_clipfrac:                       0.054661464
+      std_actor_loss:                           0.0013202867
+      std_advantage:                            0.15713045
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0006261149
+      std_entropy:                              0.017493885
+      std_grad_norm_pre_clip:                   3.0881147
+      std_loss:                                 0.0013955849
+  eval-batch-orig:
+    avg_avg_episode_length:                     43.251404
+    avg_avg_return:                             0.6322446
+    avg_benchmark_regret:                       0.3525819
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.06341553
+    proxy_corner:
+      avg_avg_return:                           0.6322446
+      avg_reward_per_step:                      0.06341553
+  eval-batch-shift:
+    avg_avg_episode_length:                     78.42282
+    avg_avg_return:                             0.10403283
+    avg_benchmark_regret:                       0.8828577
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.019165039
+    proxy_corner:
+      avg_avg_return:                           0.48556924
+      avg_reward_per_step:                      0.003929138
+  eval-batch-tree:
+    avg_avg_episode_length:                     44.972073
+    avg_avg_return:                             0.59469897
+    avg_benchmark_regret:                       0.38937664
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.081855774
+    proxy_corner:
+      avg_avg_return:                           0.59469897
+      avg_reward_per_step:                      0.081855774
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        33554432
+    env-step-replay-before:                     17399808
+    env-step-used-before:                       33554432
+    ppo-update-before:                          20480
+    env-step-generate-before:                   16154624
+    env-step-all-after:                         33587200
+    env-step-replay-after:                      17399808
+    env-step-used-after:                        33587200
+    ppo-update-after:                           20500
+    env-step-generate-after:                    16187392
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     49.21214
+    avg_avg_return:                             0.5002618
+    avg_reward_per_step:                        0.08154297
+    proxy_corner:
+      avg_avg_return:                           0.5002618
+      avg_reward_per_step:                      0.08154297
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     49.21214
+    avg_avg_return:                             0.5002618
+    avg_reward_per_step:                        0.08154297
+    proxy_corner:
+      avg_avg_return:                           0.5002618
+      avg_reward_per_step:                      0.08154297
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  perf:
+    env_steps_per_second:                       292986.5177816595
+    ppo_updates_per_second:                     42.291632414614625
+    eval-batch-orig-duration:                   0.6324717421084642
+    eval-batch-shift-duration:                  0.5991391576826572
+    eval-batch-tree-duration:                   0.5976089518517256
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.569824
+      mouse-cheese_dist_finite_wavg:            9.053762
+      mouse-cheese_dist_solvable_avg:           18.569824
+      mouse-cheese_dist_solvable_wavg:          9.053763
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.5448736
+    visit_patterns:
+      avg_first_visit_time:                     374.35083
+      avg_last_visit_time:                      521.26733
+      num_replay_batches:                       531
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009122657
+    avg_actor_approxkl3:                        0.0009955191
+    avg_actor_clipfrac:                         0.05037842
+    avg_actor_loss:                             -0.0011104885
+    avg_advantage:                              -0.013316074
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.011294455
+    avg_entropy:                                0.608536
+    avg_grad_norm_pre_clip:                     2.8849552
+    avg_loss:                                   0.0039282027
+    max:
+      max_actor_approxkl1:                      0.0016037257
+      max_actor_approxkl3:                      0.0019090095
+      max_actor_clipfrac:                       0.11669922
+      max_actor_loss:                           0.00196068
+      max_advantage:                            0.4296552
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.012892908
+      max_entropy:                              0.69684327
+      max_grad_norm_pre_clip:                   5.6459293
+      max_loss:                                 0.0071723573
+    std:
+      std_actor_approxkl1:                      0.0005870737
+      std_actor_approxkl3:                      0.00041808022
+      std_actor_clipfrac:                       0.027183859
+      std_actor_loss:                           0.0011449694
+      std_advantage:                            0.1507438
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0007891028
+      std_entropy:                              0.034610823
+      std_grad_norm_pre_clip:                   1.2480049
+      std_loss:                                 0.0012592109
+  eval-batch-orig:
+    avg_avg_episode_length:                     43.1099
+    avg_avg_return:                             0.6235029
+    avg_benchmark_regret:                       0.3613236
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.06501007
+    proxy_corner:
+      avg_avg_return:                           0.6235029
+      avg_reward_per_step:                      0.06501007
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-shift:
+    avg_avg_episode_length:                     78.09563
+    avg_avg_return:                             0.10861462
+    avg_benchmark_regret:                       0.8782759
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.01964569
+    proxy_corner:
+      avg_avg_return:                           0.47583455
+      avg_reward_per_step:                      0.0038452148
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-tree:
+    avg_avg_episode_length:                     43.775124
+    avg_avg_return:                             0.6143261
+    avg_benchmark_regret:                       0.36974946
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.08352661
+    proxy_corner:
+      avg_avg_return:                           0.6143261
+      avg_reward_per_step:                      0.08352661
+    rollouts_gif:                               float32(132, 225, 225, 3)
+===========================================================
+ 17%|▏| 33.6M/200M [11:10<40:10, 69
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        34603008
+    env-step-replay-before:                     17694720
+    env-step-used-before:                       34603008
+    ppo-update-before:                          21120
+    env-step-generate-before:                   16908288
+    env-step-all-after:                         34635776
+    env-step-replay-after:                      17727488
+    env-step-used-after:                        34635776
+    ppo-update-after:                           21140
+    env-step-generate-after:                    16908288
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     66.18315
+    avg_avg_return:                             0.4151317
+    avg_reward_per_step:                        0.012878418
+    proxy_corner:
+      avg_avg_return:                           0.4151317
+      avg_reward_per_step:                      0.012878418
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     66.18315
+    avg_avg_return:                             0.4151317
+    avg_reward_per_step:                        0.012878418
+    proxy_corner:
+      avg_avg_return:                           0.4151317
+      avg_reward_per_step:                      0.012878418
+  perf:
+    env_steps_per_second:                       351647.80734541366
+    ppo_updates_per_second:                     40.6844688948271
+    eval-batch-orig-duration:                   0.518501091748476
+    eval-batch-shift-duration:                  0.5019984217360616
+    eval-batch-tree-duration:                   0.48007781337946653
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.699463
+      mouse-cheese_dist_finite_wavg:            9.062969
+      mouse-cheese_dist_solvable_avg:           18.699463
+      mouse-cheese_dist_solvable_wavg:          9.062969
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5569063
+    visit_patterns:
+      avg_first_visit_time:                     390.27002
+      avg_last_visit_time:                      531.2158
+      num_replay_batches:                       541
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0029447982
+    avg_actor_approxkl3:                        0.002800078
+    avg_actor_clipfrac:                         0.15053712
+    avg_actor_loss:                             0.002073605
+    avg_advantage:                              -0.0057882993
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.011785308
+    avg_entropy:                                0.7233286
+    avg_grad_norm_pre_clip:                     7.8845115
+    avg_loss:                                   0.00724293
+    max:
+      max_actor_approxkl1:                      0.0073572313
+      max_actor_approxkl3:                      0.006655165
+      max_actor_clipfrac:                       0.27600098
+      max_actor_loss:                           0.008289412
+      max_advantage:                            0.50286275
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.013024835
+      max_entropy:                              0.7784788
+      max_grad_norm_pre_clip:                   14.112858
+      max_loss:                                 0.012931728
+    std:
+      std_actor_approxkl1:                      0.0021204902
+      std_actor_approxkl3:                      0.0016298405
+      std_actor_clipfrac:                       0.082349114
+      std_actor_loss:                           0.0025401725
+      std_advantage:                            0.1539067
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00072800764
+      std_entropy:                              0.024418512
+      std_grad_norm_pre_clip:                   3.8747776
+      std_loss:                                 0.0024305936
+  eval-batch-orig:
+    avg_avg_episode_length:                     41.773457
+    avg_avg_return:                             0.64630175
+    avg_benchmark_regret:                       0.3385248
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.06703949
+    proxy_corner:
+      avg_avg_return:                           0.64630175
+      avg_reward_per_step:                      0.06703949
+  eval-batch-shift:
+    avg_avg_episode_length:                     78.09389
+    avg_avg_return:                             0.11055855
+    avg_benchmark_regret:                       0.8763319
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.01992035
+    proxy_corner:
+      avg_avg_return:                           0.49147433
+      avg_reward_per_step:                      0.003967285
+  eval-batch-tree:
+    avg_avg_episode_length:                     43.884064
+    avg_avg_return:                             0.6127207
+    avg_benchmark_regret:                       0.3713549
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.083618164
+    proxy_corner:
+      avg_avg_return:                           0.6127207
+      avg_reward_per_step:                      0.083618164
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        35651584
+    env-step-replay-before:                     18153472
+    env-step-used-before:                       35651584
+    ppo-update-before:                          21760
+    env-step-generate-before:                   17498112
+    env-step-all-after:                         35684352
+    env-step-replay-after:                      18186240
+    env-step-used-after:                        35684352
+    ppo-update-after:                           21780
+    env-step-generate-after:                    17498112
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     59.432167
+    avg_avg_return:                             0.472603
+    avg_reward_per_step:                        0.015563965
+    proxy_corner:
+      avg_avg_return:                           0.472603
+      avg_reward_per_step:                      0.015563965
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     59.432167
+    avg_avg_return:                             0.472603
+    avg_reward_per_step:                        0.015563965
+    proxy_corner:
+      avg_avg_return:                           0.472603
+      avg_reward_per_step:                      0.015563965
+  perf:
+    env_steps_per_second:                       319153.30142141273
+    ppo_updates_per_second:                     42.07001750165235
+    eval-batch-orig-duration:                   0.5163685139268637
+    eval-batch-shift-duration:                  0.48359110578894615
+    eval-batch-tree-duration:                   0.4875229112803936
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             18.949219
+      mouse-cheese_dist_finite_wavg:            10.88649
+      mouse-cheese_dist_solvable_avg:           18.949219
+      mouse-cheese_dist_solvable_wavg:          10.88649
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000008
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016795
+    scoring:
+      avg_scores:                               0.55185294
+    visit_patterns:
+      avg_first_visit_time:                     408.3235
+      avg_last_visit_time:                      545.2903
+      num_replay_batches:                       555
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002438384
+    avg_actor_approxkl3:                        0.0020932707
+    avg_actor_clipfrac:                         0.11686402
+    avg_actor_loss:                             8.5462634e-05
+    avg_advantage:                              0.009160187
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.01167047
+    avg_entropy:                                0.70109266
+    avg_grad_norm_pre_clip:                     4.78511
+    avg_loss:                                   0.005219605
+    max:
+      max_actor_approxkl1:                      0.006521537
+      max_actor_approxkl3:                      0.004392296
+      max_actor_clipfrac:                       0.22021484
+      max_actor_loss:                           0.0064231455
+      max_advantage:                            0.50266993
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0131485555
+      max_entropy:                              0.73368955
+      max_grad_norm_pre_clip:                   9.584104
+      max_loss:                                 0.012232398
+    std:
+      std_actor_approxkl1:                      0.0015164954
+      std_actor_approxkl3:                      0.0010107525
+      std_actor_clipfrac:                       0.055230718
+      std_actor_loss:                           0.0022435011
+      std_advantage:                            0.15277289
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0008262068
+      std_entropy:                              0.023225138
+      std_grad_norm_pre_clip:                   2.2744205
+      std_loss:                                 0.0023528596
+  eval-batch-orig:
+    avg_avg_episode_length:                     40.44507
+    avg_avg_return:                             0.65441537
+    avg_benchmark_regret:                       0.33041114
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.069618225
+    proxy_corner:
+      avg_avg_return:                           0.65441537
+      avg_reward_per_step:                      0.069618225
+  eval-batch-shift:
+    avg_avg_episode_length:                     77.92116
+    avg_avg_return:                             0.11064791
+    avg_benchmark_regret:                       0.8762425
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.020645142
+    proxy_corner:
+      avg_avg_return:                           0.5060817
+      avg_reward_per_step:                      0.004081726
+  eval-batch-tree:
+    avg_avg_episode_length:                     40.733467
+    avg_avg_return:                             0.6412163
+    avg_benchmark_regret:                       0.34285933
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.08766937
+    proxy_corner:
+      avg_avg_return:                           0.6412163
+      avg_reward_per_step:                      0.08766937
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        36700160
+    env-step-replay-before:                     18677760
+    env-step-used-before:                       36700160
+    ppo-update-before:                          22400
+    env-step-generate-before:                   18022400
+    env-step-all-after:                         36732928
+    env-step-replay-after:                      18677760
+    env-step-used-after:                        36732928
+    ppo-update-after:                           22420
+    env-step-generate-after:                    18055168
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     40.030422
+    avg_avg_return:                             0.61317074
+    avg_reward_per_step:                        0.08950806
+    proxy_corner:
+      avg_avg_return:                           0.61317074
+      avg_reward_per_step:                      0.08950806
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     40.030422
+    avg_avg_return:                             0.61317074
+    avg_reward_per_step:                        0.08950806
+    proxy_corner:
+      avg_avg_return:                           0.61317074
+      avg_reward_per_step:                      0.08950806
+  perf:
+    env_steps_per_second:                       294483.66048385017
+    ppo_updates_per_second:                     42.10516156606803
+    eval-batch-orig-duration:                   0.6006848076358438
+    eval-batch-shift-duration:                  0.6976775731891394
+    eval-batch-tree-duration:                   0.5232621598988771
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.166748
+      mouse-cheese_dist_finite_wavg:            10.9066925
+      mouse-cheese_dist_solvable_avg:           19.166748
+      mouse-cheese_dist_solvable_wavg:          10.906695
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.420168
+    scoring:
+      avg_scores:                               0.53944474
+    visit_patterns:
+      avg_first_visit_time:                     425.21533
+      avg_last_visit_time:                      560.28564
+      num_replay_batches:                       570
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013166172
+    avg_actor_approxkl3:                        0.0015740258
+    avg_actor_clipfrac:                         0.08057251
+    avg_actor_loss:                             0.00013460591
+    avg_advantage:                              -0.002944598
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00946711
+    avg_entropy:                                0.48667535
+    avg_grad_norm_pre_clip:                     3.590183
+    avg_loss:                                   0.004381486
+    max:
+      max_actor_approxkl1:                      0.0045969095
+      max_actor_approxkl3:                      0.0039663236
+      max_actor_clipfrac:                       0.17749023
+      max_actor_loss:                           0.0045596957
+      max_advantage:                            0.49228972
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.012283093
+      max_entropy:                              0.55384684
+      max_grad_norm_pre_clip:                   9.75201
+      max_loss:                                 0.008391721
+    std:
+      std_actor_approxkl1:                      0.001199665
+      std_actor_approxkl3:                      0.001030301
+      std_actor_clipfrac:                       0.049796302
+      std_actor_loss:                           0.0019471679
+      std_advantage:                            0.13752294
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010848646
+      std_entropy:                              0.03449783
+      std_grad_norm_pre_clip:                   2.2203083
+      std_loss:                                 0.0020218464
+  eval-batch-orig:
+    avg_avg_episode_length:                     39.92919
+    avg_avg_return:                             0.65897995
+    avg_benchmark_regret:                       0.32584655
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.06977844
+    proxy_corner:
+      avg_avg_return:                           0.65897995
+      avg_reward_per_step:                      0.06977844
+  eval-batch-shift:
+    avg_avg_episode_length:                     76.91279
+    avg_avg_return:                             0.12627205
+    avg_benchmark_regret:                       0.8606184
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.021530151
+    proxy_corner:
+      avg_avg_return:                           0.49362758
+      avg_reward_per_step:                      0.004005432
+  eval-batch-tree:
+    avg_avg_episode_length:                     40.21312
+    avg_avg_return:                             0.6476921
+    avg_benchmark_regret:                       0.33638355
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.08679962
+    proxy_corner:
+      avg_avg_return:                           0.6476921
+      avg_reward_per_step:                      0.08679962
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        37748736
+    env-step-replay-before:                     19267584
+    env-step-used-before:                       37748736
+    ppo-update-before:                          23040
+    env-step-generate-before:                   18481152
+    env-step-all-after:                         37781504
+    env-step-replay-after:                      19267584
+    env-step-used-after:                        37781504
+    ppo-update-after:                           23060
+    env-step-generate-after:                    18513920
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     39.016357
+    avg_avg_return:                             0.61843604
+    avg_reward_per_step:                        0.07952881
+    proxy_corner:
+      avg_avg_return:                           0.61843604
+      avg_reward_per_step:                      0.07952881
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     39.016357
+    avg_avg_return:                             0.61843604
+    avg_reward_per_step:                        0.07952881
+    proxy_corner:
+      avg_avg_return:                           0.61843604
+      avg_reward_per_step:                      0.07952881
+  perf:
+    env_steps_per_second:                       295850.5240858913
+    ppo_updates_per_second:                     40.90426095397592
+    eval-batch-orig-duration:                   0.5212970850989223
+    eval-batch-shift-duration:                  0.4462832612916827
+    eval-batch-tree-duration:                   0.4859006814658642
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.435547
+      mouse-cheese_dist_finite_wavg:            10.935203
+      mouse-cheese_dist_solvable_avg:           19.435547
+      mouse-cheese_dist_solvable_wavg:          10.935204
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.52468646
+    visit_patterns:
+      avg_first_visit_time:                     440.72778
+      avg_last_visit_time:                      578.22534
+      num_replay_batches:                       588
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0033185992
+    avg_actor_approxkl3:                        0.0032087208
+    avg_actor_clipfrac:                         0.15165405
+    avg_actor_loss:                             0.0036049844
+    avg_advantage:                              0.0020853747
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.009711882
+    avg_entropy:                                0.50991386
+    avg_grad_norm_pre_clip:                     5.6905284
+    avg_loss:                                   0.007951011
+    max:
+      max_actor_approxkl1:                      0.009099425
+      max_actor_approxkl3:                      0.008053154
+      max_actor_clipfrac:                       0.26843262
+      max_actor_loss:                           0.012293145
+      max_advantage:                            0.51488197
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.011145974
+      max_entropy:                              0.56265336
+      max_grad_norm_pre_clip:                   8.419973
+      max_loss:                                 0.016730301
+    std:
+      std_actor_approxkl1:                      0.0021581564
+      std_actor_approxkl3:                      0.0019145606
+      std_actor_clipfrac:                       0.063551694
+      std_actor_loss:                           0.0031951894
+      std_advantage:                            0.13870804
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010579581
+      std_entropy:                              0.028284527
+      std_grad_norm_pre_clip:                   1.8812046
+      std_loss:                                 0.0034706725
+  eval-batch-orig:
+    avg_avg_episode_length:                     37.93815
+    avg_avg_return:                             0.68866616
+    avg_benchmark_regret:                       0.29616034
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.07064819
+    proxy_corner:
+      avg_avg_return:                           0.68866616
+      avg_reward_per_step:                      0.07064819
+  eval-batch-shift:
+    avg_avg_episode_length:                     76.94054
+    avg_avg_return:                             0.132211
+    avg_benchmark_regret:                       0.85467947
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.0207901
+    proxy_corner:
+      avg_avg_return:                           0.50993824
+      avg_reward_per_step:                      0.004119873
+  eval-batch-tree:
+    avg_avg_episode_length:                     39.63147
+    avg_avg_return:                             0.6716198
+    avg_benchmark_regret:                       0.3124559
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.0870285
+    proxy_corner:
+      avg_avg_return:                           0.6716198
+      avg_reward_per_step:                      0.0870285
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        38797312
+    env-step-replay-before:                     19922944
+    env-step-used-before:                       38797312
+    ppo-update-before:                          23680
+    env-step-generate-before:                   18874368
+    env-step-all-after:                         38830080
+    env-step-replay-after:                      19922944
+    env-step-used-after:                        38830080
+    ppo-update-after:                           23700
+    env-step-generate-after:                    18907136
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     40.463345
+    avg_avg_return:                             0.5955166
+    avg_reward_per_step:                        0.08380127
+    proxy_corner:
+      avg_avg_return:                           0.5955166
+      avg_reward_per_step:                      0.08380127
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     40.463345
+    avg_avg_return:                             0.5955166
+    avg_reward_per_step:                        0.08380127
+    proxy_corner:
+      avg_avg_return:                           0.5955166
+      avg_reward_per_step:                      0.08380127
+  perf:
+    env_steps_per_second:                       436520.3104898019
+    ppo_updates_per_second:                     42.23234849663838
+    eval-batch-orig-duration:                   0.40916471742093563
+    eval-batch-shift-duration:                  0.3314923383295536
+    eval-batch-tree-duration:                   0.3650422152131796
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.545654
+      mouse-cheese_dist_finite_wavg:            10.953836
+      mouse-cheese_dist_solvable_avg:           19.545654
+      mouse-cheese_dist_solvable_wavg:          10.953837
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.51278186
+    visit_patterns:
+      avg_first_visit_time:                     455.4585
+      avg_last_visit_time:                      598.3147
+      num_replay_batches:                       608
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013117684
+    avg_actor_approxkl3:                        0.0012393866
+    avg_actor_clipfrac:                         0.06541748
+    avg_actor_loss:                             -0.00033410863
+    avg_advantage:                              -0.009573717
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0102154305
+    avg_entropy:                                0.5048484
+    avg_grad_norm_pre_clip:                     3.2515035
+    avg_loss:                                   0.0042687594
+    max:
+      max_actor_approxkl1:                      0.0026433626
+      max_actor_approxkl3:                      0.0026682843
+      max_actor_clipfrac:                       0.13928223
+      max_actor_loss:                           0.0023578703
+      max_advantage:                            0.44092193
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.012356304
+      max_entropy:                              0.58883005
+      max_grad_norm_pre_clip:                   6.001474
+      max_loss:                                 0.0078224335
+    std:
+      std_actor_approxkl1:                      0.00076610106
+      std_actor_approxkl3:                      0.00062713627
+      std_actor_clipfrac:                       0.034485348
+      std_actor_loss:                           0.0013610678
+      std_advantage:                            0.14326622
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0009925413
+      std_entropy:                              0.038314816
+      std_grad_norm_pre_clip:                   1.4886472
+      std_loss:                                 0.0015217396
+  eval-batch-orig:
+    avg_avg_episode_length:                     37.029907
+    avg_avg_return:                             0.70935786
+    avg_benchmark_regret:                       0.27546868
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.07154846
+    proxy_corner:
+      avg_avg_return:                           0.70935786
+      avg_reward_per_step:                      0.07154846
+  eval-batch-shift:
+    avg_avg_episode_length:                     76.05648
+    avg_avg_return:                             0.14137548
+    avg_benchmark_regret:                       0.84551495
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.021484375
+    proxy_corner:
+      avg_avg_return:                           0.5158795
+      avg_reward_per_step:                      0.0042037964
+  eval-batch-tree:
+    avg_avg_episode_length:                     38.788883
+    avg_avg_return:                             0.68568224
+    avg_benchmark_regret:                       0.2983934
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.08779907
+    proxy_corner:
+      avg_avg_return:                           0.68568224
+      avg_reward_per_step:                      0.08779907
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        39845888
+    env-step-replay-before:                     20381696
+    env-step-used-before:                       39845888
+    ppo-update-before:                          24320
+    env-step-generate-before:                   19464192
+    env-step-all-after:                         39878656
+    env-step-replay-after:                      20414464
+    env-step-used-after:                        39878656
+    ppo-update-after:                           24340
+    env-step-generate-after:                    19464192
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     61.16718
+    avg_avg_return:                             0.45311433
+    avg_reward_per_step:                        0.014099121
+    proxy_corner:
+      avg_avg_return:                           0.45311433
+      avg_reward_per_step:                      0.014099121
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     61.16718
+    avg_avg_return:                             0.45311433
+    avg_reward_per_step:                        0.014099121
+    proxy_corner:
+      avg_avg_return:                           0.45311433
+      avg_reward_per_step:                      0.014099121
+  perf:
+    env_steps_per_second:                       329532.46810026636
+    ppo_updates_per_second:                     41.36555200997586
+    eval-batch-orig-duration:                   0.5350122526288033
+    eval-batch-shift-duration:                  0.4843844547867775
+    eval-batch-tree-duration:                   0.4838254516944289
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.725342
+      mouse-cheese_dist_finite_wavg:            10.969348
+      mouse-cheese_dist_solvable_avg:           19.725342
+      mouse-cheese_dist_solvable_wavg:          10.969349
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5172435
+    visit_patterns:
+      avg_first_visit_time:                     469.05005
+      avg_last_visit_time:                      613.29395
+      num_replay_batches:                       623
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014034512
+    avg_actor_approxkl3:                        0.0012699752
+    avg_actor_clipfrac:                         0.0654541
+    avg_actor_loss:                             -0.0010209483
+    avg_advantage:                              -0.0018358219
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.010861911
+    avg_entropy:                                0.70384353
+    avg_grad_norm_pre_clip:                     2.8768857
+    avg_loss:                                   0.0037061633
+    max:
+      max_actor_approxkl1:                      0.0027513357
+      max_actor_approxkl3:                      0.0022287122
+      max_actor_clipfrac:                       0.12390137
+      max_actor_loss:                           0.0010697544
+      max_advantage:                            0.4535905
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.014379784
+      max_entropy:                              0.72966063
+      max_grad_norm_pre_clip:                   4.679267
+      max_loss:                                 0.0064879563
+    std:
+      std_actor_approxkl1:                      0.0007132364
+      std_actor_approxkl3:                      0.0005937718
+      std_actor_clipfrac:                       0.03674403
+      std_actor_loss:                           0.001174387
+      std_advantage:                            0.14693484
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011674438
+      std_entropy:                              0.014383528
+      std_grad_norm_pre_clip:                   1.1831373
+      std_loss:                                 0.0014480306
+  eval-batch-orig:
+    avg_avg_episode_length:                     36.311405
+    avg_avg_return:                             0.714516
+    avg_benchmark_regret:                       0.27031052
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.07232666
+    proxy_corner:
+      avg_avg_return:                           0.714516
+      avg_reward_per_step:                      0.07232666
+  eval-batch-shift:
+    avg_avg_episode_length:                     75.99625
+    avg_avg_return:                             0.14747454
+    avg_benchmark_regret:                       0.8394159
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.024139404
+    proxy_corner:
+      avg_avg_return:                           0.5260774
+      avg_reward_per_step:                      0.004295349
+  eval-batch-tree:
+    avg_avg_episode_length:                     37.645638
+    avg_avg_return:                             0.70124805
+    avg_benchmark_regret:                       0.28282756
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.088905334
+    proxy_corner:
+      avg_avg_return:                           0.70124805
+      avg_reward_per_step:                      0.088905334
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        40894464
+    env-step-replay-before:                     21037056
+    env-step-used-before:                       40894464
+    ppo-update-before:                          24960
+    env-step-generate-before:                   19857408
+    env-step-all-after:                         40927232
+    env-step-replay-after:                      21069824
+    env-step-used-after:                        40927232
+    ppo-update-after:                           24980
+    env-step-generate-after:                    19857408
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     59.569115
+    avg_avg_return:                             0.47478276
+    avg_reward_per_step:                        0.015319824
+    proxy_corner:
+      avg_avg_return:                           0.47478276
+      avg_reward_per_step:                      0.015319824
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     59.569115
+    avg_avg_return:                             0.47478276
+    avg_reward_per_step:                        0.015319824
+    proxy_corner:
+      avg_avg_return:                           0.47478276
+      avg_reward_per_step:                      0.015319824
+  perf:
+    env_steps_per_second:                       315454.6712581524
+    ppo_updates_per_second:                     44.52864887467876
+    eval-batch-orig-duration:                   0.5013151001185179
+    eval-batch-shift-duration:                  0.47653617709875107
+    eval-batch-tree-duration:                   0.46778973657637835
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             19.898193
+      mouse-cheese_dist_finite_wavg:            10.979316
+      mouse-cheese_dist_solvable_avg:           19.898193
+      mouse-cheese_dist_solvable_wavg:          10.979316
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5147582
+    visit_patterns:
+      avg_first_visit_time:                     483.8567
+      avg_last_visit_time:                      633.24805
+      num_replay_batches:                       643
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.000744417
+    avg_actor_approxkl3:                        0.0008350773
+    avg_actor_clipfrac:                         0.041955568
+    avg_actor_loss:                             -0.0009148076
+    avg_advantage:                              -2.7426286e-05
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0106017105
+    avg_entropy:                                0.62972224
+    avg_grad_norm_pre_clip:                     2.91493
+    avg_loss:                                   0.0037563257
+    max:
+      max_actor_approxkl1:                      0.002161075
+      max_actor_approxkl3:                      0.0014460741
+      max_actor_clipfrac:                       0.08850098
+      max_actor_loss:                           0.00094999373
+      max_advantage:                            0.44252557
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.012166956
+      max_entropy:                              0.6638458
+      max_grad_norm_pre_clip:                   4.9936914
+      max_loss:                                 0.0055488492
+    std:
+      std_actor_approxkl1:                      0.00052268733
+      std_actor_approxkl3:                      0.00039522297
+      std_actor_clipfrac:                       0.026306339
+      std_actor_loss:                           0.0011210352
+      std_advantage:                            0.14556721
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00080996123
+      std_entropy:                              0.022019234
+      std_grad_norm_pre_clip:                   0.9817228
+      std_loss:                                 0.0011311383
+  eval-batch-orig:
+    avg_avg_episode_length:                     37.189884
+    avg_avg_return:                             0.70066667
+    avg_benchmark_regret:                       0.28415987
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.07273865
+    proxy_corner:
+      avg_avg_return:                           0.70066667
+      avg_reward_per_step:                      0.07273865
+  eval-batch-shift:
+    avg_avg_episode_length:                     76.46618
+    avg_avg_return:                             0.13056852
+    avg_benchmark_regret:                       0.85632193
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.023544312
+    proxy_corner:
+      avg_avg_return:                           0.50544107
+      avg_reward_per_step:                      0.0040893555
+  eval-batch-tree:
+    avg_avg_episode_length:                     37.18793
+    avg_avg_return:                             0.695266
+    avg_benchmark_regret:                       0.28880954
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.089515686
+    proxy_corner:
+      avg_avg_return:                           0.695266
+      avg_reward_per_step:                      0.089515686
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        41943040
+    env-step-replay-before:                     21528576
+    env-step-used-before:                       41943040
+    ppo-update-before:                          25600
+    env-step-generate-before:                   20414464
+    env-step-all-after:                         41975808
+    env-step-replay-after:                      21528576
+    env-step-used-after:                        41975808
+    ppo-update-after:                           25620
+    env-step-generate-after:                    20447232
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     36.23756
+    avg_avg_return:                             0.6612359
+    avg_reward_per_step:                        0.090545654
+    proxy_corner:
+      avg_avg_return:                           0.6612359
+      avg_reward_per_step:                      0.090545654
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     36.23756
+    avg_avg_return:                             0.6612359
+    avg_reward_per_step:                        0.090545654
+    proxy_corner:
+      avg_avg_return:                           0.6612359
+      avg_reward_per_step:                      0.090545654
+  perf:
+    env_steps_per_second:                       344275.0474212923
+    ppo_updates_per_second:                     41.80677864378159
+    eval-batch-orig-duration:                   0.5119608948007226
+    eval-batch-shift-duration:                  0.43086262978613377
+    eval-batch-tree-duration:                   0.4815916921943426
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             20.160889
+      mouse-cheese_dist_finite_wavg:            10.107478
+      mouse-cheese_dist_solvable_avg:           20.160889
+      mouse-cheese_dist_solvable_wavg:          10.107481
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.5267248
+    visit_patterns:
+      avg_first_visit_time:                     505.75854
+      avg_last_visit_time:                      647.1907
+      num_replay_batches:                       657
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014211893
+    avg_actor_approxkl3:                        0.0015929675
+    avg_actor_clipfrac:                         0.07314453
+    avg_actor_loss:                             0.0005621359
+    avg_advantage:                              0.0130369775
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.007990064
+    avg_entropy:                                0.42397782
+    avg_grad_norm_pre_clip:                     4.1093907
+    avg_loss:                                   0.00413319
+    max:
+      max_actor_approxkl1:                      0.005282471
+      max_actor_approxkl3:                      0.0052920748
+      max_actor_clipfrac:                       0.19641113
+      max_actor_loss:                           0.0071618184
+      max_advantage:                            0.40089267
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0105627235
+      max_entropy:                              0.5079882
+      max_grad_norm_pre_clip:                   8.550676
+      max_loss:                                 0.010760099
+    std:
+      std_actor_approxkl1:                      0.001379958
+      std_actor_approxkl3:                      0.0013665704
+      std_actor_clipfrac:                       0.05009746
+      std_actor_loss:                           0.0025781866
+      std_advantage:                            0.12643157
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00093325507
+      std_entropy:                              0.04223625
+      std_grad_norm_pre_clip:                   1.9787737
+      std_loss:                                 0.0025573124
+  eval-batch-orig:
+    avg_avg_episode_length:                     35.632156
+    avg_avg_return:                             0.7153027
+    avg_benchmark_regret:                       0.26952386
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.07437897
+    proxy_corner:
+      avg_avg_return:                           0.7153027
+      avg_reward_per_step:                      0.07437897
+  eval-batch-shift:
+    avg_avg_episode_length:                     76.219246
+    avg_avg_return:                             0.13653569
+    avg_benchmark_regret:                       0.8503548
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.023994446
+    proxy_corner:
+      avg_avg_return:                           0.52036536
+      avg_reward_per_step:                      0.0042266846
+  eval-batch-tree:
+    avg_avg_episode_length:                     35.340614
+    avg_avg_return:                             0.71345073
+    avg_benchmark_regret:                       0.2706249
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.09199524
+    proxy_corner:
+      avg_avg_return:                           0.71345073
+      avg_reward_per_step:                      0.09199524
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        42991616
+    env-step-replay-before:                     21954560
+    env-step-used-before:                       42991616
+    ppo-update-before:                          26240
+    env-step-generate-before:                   21037056
+    env-step-all-after:                         43024384
+    env-step-replay-after:                      21987328
+    env-step-used-after:                        43024384
+    ppo-update-after:                           26260
+    env-step-generate-after:                    21037056
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     60.797813
+    avg_avg_return:                             0.4738653
+    avg_reward_per_step:                        0.014801025
+    proxy_corner:
+      avg_avg_return:                           0.4738653
+      avg_reward_per_step:                      0.014801025
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     60.797813
+    avg_avg_return:                             0.4738653
+    avg_reward_per_step:                        0.014801025
+    proxy_corner:
+      avg_avg_return:                           0.4738653
+      avg_reward_per_step:                      0.014801025
+  perf:
+    env_steps_per_second:                       273427.42146844795
+    ppo_updates_per_second:                     41.71304526695487
+    eval-batch-orig-duration:                   0.6139878490939736
+    eval-batch-shift-duration:                  0.5516737084835768
+    eval-batch-tree-duration:                   0.5765023352578282
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             20.336182
+      mouse-cheese_dist_finite_wavg:            10.127174
+      mouse-cheese_dist_solvable_avg:           20.336182
+      mouse-cheese_dist_solvable_wavg:          10.127175
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000008
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.51267904
+    visit_patterns:
+      avg_first_visit_time:                     522.4336
+      avg_last_visit_time:                      661.18384
+      num_replay_batches:                       671
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012459796
+    avg_actor_approxkl3:                        0.0011529757
+    avg_actor_clipfrac:                         0.06185303
+    avg_actor_loss:                             -0.00079162046
+    avg_advantage:                              -0.002533943
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.010382208
+    avg_entropy:                                0.6156244
+    avg_grad_norm_pre_clip:                     3.496301
+    avg_loss:                                   0.003783859
+    max:
+      max_actor_approxkl1:                      0.0026803226
+      max_actor_approxkl3:                      0.0019249276
+      max_actor_clipfrac:                       0.10876465
+      max_actor_loss:                           0.0010669082
+      max_advantage:                            0.42589796
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0124876555
+      max_entropy:                              0.6523218
+      max_grad_norm_pre_clip:                   5.5293
+      max_loss:                                 0.0055056456
+    std:
+      std_actor_approxkl1:                      0.0007289037
+      std_actor_approxkl3:                      0.0004910687
+      std_actor_clipfrac:                       0.03025281
+      std_actor_loss:                           0.00081514916
+      std_advantage:                            0.14462812
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00093302433
+      std_entropy:                              0.013459504
+      std_grad_norm_pre_clip:                   1.4162449
+      std_loss:                                 0.00090802624
+  eval-batch-orig:
+    avg_avg_episode_length:                     35.377907
+    avg_avg_return:                             0.7193359
+    avg_benchmark_regret:                       0.26549062
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.07456207
+    proxy_corner:
+      avg_avg_return:                           0.7193359
+      avg_reward_per_step:                      0.07456207
+  eval-batch-shift:
+    avg_avg_episode_length:                     75.783005
+    avg_avg_return:                             0.14119744
+    avg_benchmark_regret:                       0.845693
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.022918701
+    proxy_corner:
+      avg_avg_return:                           0.5197903
+      avg_reward_per_step:                      0.0042419434
+  eval-batch-tree:
+    avg_avg_episode_length:                     34.812843
+    avg_avg_return:                             0.73239315
+    avg_benchmark_regret:                       0.25168246
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.092056274
+    proxy_corner:
+      avg_avg_return:                           0.73239315
+      avg_reward_per_step:                      0.092056274
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        44040192
+    env-step-replay-before:                     22511616
+    env-step-used-before:                       44040192
+    ppo-update-before:                          26880
+    env-step-generate-before:                   21528576
+    env-step-all-after:                         44072960
+    env-step-replay-after:                      22511616
+    env-step-used-after:                        44072960
+    ppo-update-after:                           26900
+    env-step-generate-after:                    21561344
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     35.54391
+    avg_avg_return:                             0.6679726
+    avg_reward_per_step:                        0.09057617
+    proxy_corner:
+      avg_avg_return:                           0.6679726
+      avg_reward_per_step:                      0.09057617
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     35.54391
+    avg_avg_return:                             0.6679726
+    avg_reward_per_step:                        0.09057617
+    proxy_corner:
+      avg_avg_return:                           0.6679726
+      avg_reward_per_step:                      0.09057617
+  perf:
+    env_steps_per_second:                       228353.205276801
+    ppo_updates_per_second:                     41.34912801823134
+    eval-batch-orig-duration:                   0.6133819101378322
+    eval-batch-shift-duration:                  0.514884177595377
+    eval-batch-tree-duration:                   0.4892240837216377
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             20.587402
+      mouse-cheese_dist_finite_wavg:            11.044222
+      mouse-cheese_dist_solvable_avg:           20.587402
+      mouse-cheese_dist_solvable_wavg:          11.044223
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016816
+    scoring:
+      avg_scores:                               0.50842786
+    visit_patterns:
+      avg_first_visit_time:                     538.75977
+      avg_last_visit_time:                      677.24023
+      num_replay_batches:                       687
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002251284
+    avg_actor_approxkl3:                        0.002272352
+    avg_actor_clipfrac:                         0.100543216
+    avg_actor_loss:                             0.0020220939
+    avg_advantage:                              -0.009262323
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.007948247
+    avg_entropy:                                0.3729615
+    avg_grad_norm_pre_clip:                     5.1044574
+    avg_loss:                                   0.005623256
+    max:
+      max_actor_approxkl1:                      0.0059312354
+      max_actor_approxkl3:                      0.0051424485
+      max_actor_clipfrac:                       0.17248535
+      max_actor_loss:                           0.0077246726
+      max_advantage:                            0.3965242
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.009278717
+      max_entropy:                              0.43233404
+      max_grad_norm_pre_clip:                   8.689347
+      max_loss:                                 0.011456759
+    std:
+      std_actor_approxkl1:                      0.0015318266
+      std_actor_approxkl3:                      0.0012478429
+      std_actor_clipfrac:                       0.04655721
+      std_actor_loss:                           0.002333179
+      std_advantage:                            0.12655541
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00084711856
+      std_entropy:                              0.030137977
+      std_grad_norm_pre_clip:                   2.2991095
+      std_loss:                                 0.0023742192
+  eval-batch-orig:
+    avg_avg_episode_length:                     33.895493
+    avg_avg_return:                             0.74248993
+    avg_benchmark_regret:                       0.24233663
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.07624054
+    proxy_corner:
+      avg_avg_return:                           0.74248993
+      avg_reward_per_step:                      0.07624054
+  eval-batch-shift:
+    avg_avg_episode_length:                     74.85599
+    avg_avg_return:                             0.16059715
+    avg_benchmark_regret:                       0.82629335
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.023612976
+    proxy_corner:
+      avg_avg_return:                           0.55433786
+      avg_reward_per_step:                      0.0045776367
+  eval-batch-tree:
+    avg_avg_episode_length:                     34.85397
+    avg_avg_return:                             0.7226546
+    avg_benchmark_regret:                       0.261421
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.093681335
+    proxy_corner:
+      avg_avg_return:                           0.7226546
+      avg_reward_per_step:                      0.093681335
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        45088768
+    env-step-replay-before:                     22937600
+    env-step-used-before:                       45088768
+    ppo-update-before:                          27520
+    env-step-generate-before:                   22151168
+    env-step-all-after:                         45121536
+    env-step-replay-after:                      22937600
+    env-step-used-after:                        45121536
+    ppo-update-after:                           27540
+    env-step-generate-after:                    22183936
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     37.50094
+    avg_avg_return:                             0.6503078
+    avg_reward_per_step:                        0.0713501
+    proxy_corner:
+      avg_avg_return:                           0.6503078
+      avg_reward_per_step:                      0.0713501
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     37.50094
+    avg_avg_return:                             0.6503078
+    avg_reward_per_step:                        0.0713501
+    proxy_corner:
+      avg_avg_return:                           0.6503078
+      avg_reward_per_step:                      0.0713501
+  perf:
+    env_steps_per_second:                       349545.2164087808
+    ppo_updates_per_second:                     45.29229365422611
+    eval-batch-orig-duration:                   0.4263434484601021
+    eval-batch-shift-duration:                  0.5672470200806856
+    eval-batch-tree-duration:                   0.38902899250388145
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             20.879639
+      mouse-cheese_dist_finite_wavg:            11.965963
+      mouse-cheese_dist_solvable_avg:           20.879639
+      mouse-cheese_dist_solvable_wavg:          11.965966
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201682
+    scoring:
+      avg_scores:                               0.50690085
+    visit_patterns:
+      avg_first_visit_time:                     556.6992
+      avg_last_visit_time:                      690.3159
+      num_replay_batches:                       700
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0026117524
+    avg_actor_approxkl3:                        0.0025310891
+    avg_actor_clipfrac:                         0.106823735
+    avg_actor_loss:                             0.0023829793
+    avg_advantage:                              -0.00615753
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.008188067
+    avg_entropy:                                0.37480292
+    avg_grad_norm_pre_clip:                     6.0731025
+    avg_loss:                                   0.0061022104
+    max:
+      max_actor_approxkl1:                      0.007288174
+      max_actor_approxkl3:                      0.007916357
+      max_actor_clipfrac:                       0.21069336
+      max_actor_loss:                           0.01215253
+      max_advantage:                            0.3847099
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.009728926
+      max_entropy:                              0.43755937
+      max_grad_norm_pre_clip:                   9.060489
+      max_loss:                                 0.015767181
+    std:
+      std_actor_approxkl1:                      0.0019149148
+      std_actor_approxkl3:                      0.0019445975
+      std_actor_clipfrac:                       0.05249503
+      std_actor_loss:                           0.0034136814
+      std_advantage:                            0.12763683
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010323387
+      std_entropy:                              0.0294499
+      std_grad_norm_pre_clip:                   2.2068372
+      std_loss:                                 0.0034216773
+  eval-batch-orig:
+    avg_avg_episode_length:                     33.733173
+    avg_avg_return:                             0.75980294
+    avg_benchmark_regret:                       0.22502363
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.07556152
+    proxy_corner:
+      avg_avg_return:                           0.75980294
+      avg_reward_per_step:                      0.07556152
+  eval-batch-shift:
+    avg_avg_episode_length:                     74.248856
+    avg_avg_return:                             0.17020175
+    avg_benchmark_regret:                       0.81668866
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.025970459
+    proxy_corner:
+      avg_avg_return:                           0.5594942
+      avg_reward_per_step:                      0.004623413
+  eval-batch-tree:
+    avg_avg_episode_length:                     34.865555
+    avg_avg_return:                             0.75048447
+    avg_benchmark_regret:                       0.2335912
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.0920105
+    proxy_corner:
+      avg_avg_return:                           0.75048447
+      avg_reward_per_step:                      0.0920105
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        46137344
+    env-step-replay-before:                     23560192
+    env-step-used-before:                       46137344
+    ppo-update-before:                          28160
+    env-step-generate-before:                   22577152
+    env-step-all-after:                         46170112
+    env-step-replay-after:                      23560192
+    env-step-used-after:                        46170112
+    ppo-update-after:                           28180
+    env-step-generate-after:                    22609920
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     29.630463
+    avg_avg_return:                             0.7244996
+    avg_reward_per_step:                        0.09887695
+    proxy_corner:
+      avg_avg_return:                           0.7244996
+      avg_reward_per_step:                      0.09887695
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     29.630463
+    avg_avg_return:                             0.7244996
+    avg_reward_per_step:                        0.09887695
+    proxy_corner:
+      avg_avg_return:                           0.7244996
+      avg_reward_per_step:                      0.09887695
+  perf:
+    env_steps_per_second:                       287096.01582386263
+    ppo_updates_per_second:                     42.09861452551662
+    eval-batch-orig-duration:                   0.5445024790242314
+    eval-batch-shift-duration:                  0.4921306949108839
+    eval-batch-tree-duration:                   0.49058095831424
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             21.163574
+      mouse-cheese_dist_finite_wavg:            11.109853
+      mouse-cheese_dist_solvable_avg:           21.163574
+      mouse-cheese_dist_solvable_wavg:          11.109853
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.47978556
+    visit_patterns:
+      avg_first_visit_time:                     571.4021
+      avg_last_visit_time:                      709.188
+      num_replay_batches:                       719
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013636459
+    avg_actor_approxkl3:                        0.001046005
+    avg_actor_clipfrac:                         0.052075196
+    avg_actor_loss:                             0.00022482313
+    avg_advantage:                              0.007129967
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.006181007
+    avg_entropy:                                0.30633932
+    avg_grad_norm_pre_clip:                     3.5663414
+    avg_loss:                                   0.0030089875
+    max:
+      max_actor_approxkl1:                      0.0035903342
+      max_actor_approxkl3:                      0.0030802176
+      max_actor_clipfrac:                       0.11376953
+      max_actor_loss:                           0.003026761
+      max_advantage:                            0.3046664
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.007429926
+      max_entropy:                              0.35042053
+      max_grad_norm_pre_clip:                   6.5927544
+      max_loss:                                 0.005860056
+    std:
+      std_actor_approxkl1:                      0.0009090978
+      std_actor_approxkl3:                      0.00067717826
+      std_actor_clipfrac:                       0.02855403
+      std_actor_loss:                           0.0013163253
+      std_advantage:                            0.111538835
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0007326353
+      std_entropy:                              0.024799949
+      std_grad_norm_pre_clip:                   1.4671528
+      std_loss:                                 0.001410669
+  eval-batch-orig:
+    avg_avg_episode_length:                     31.480228
+    avg_avg_return:                             0.7800446
+    avg_benchmark_regret:                       0.20478195
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.07784271
+    proxy_corner:
+      avg_avg_return:                           0.7800446
+      avg_reward_per_step:                      0.07784271
+  eval-batch-shift:
+    avg_avg_episode_length:                     74.65297
+    avg_avg_return:                             0.16300303
+    avg_benchmark_regret:                       0.8238874
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.023674011
+    proxy_corner:
+      avg_avg_return:                           0.583042
+      avg_reward_per_step:                      0.0048217773
+  eval-batch-tree:
+    avg_avg_episode_length:                     32.1948
+    avg_avg_return:                             0.7708969
+    avg_benchmark_regret:                       0.21317871
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.09488678
+    proxy_corner:
+      avg_avg_return:                           0.7708969
+      avg_reward_per_step:                      0.09488678
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        47185920
+    env-step-replay-before:                     24084480
+    env-step-used-before:                       47185920
+    ppo-update-before:                          28800
+    env-step-generate-before:                   23101440
+    env-step-all-after:                         47218688
+    env-step-replay-after:                      24084480
+    env-step-used-after:                        47218688
+    ppo-update-after:                           28820
+    env-step-generate-after:                    23134208
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     34.237476
+    avg_avg_return:                             0.68155605
+    avg_reward_per_step:                        0.09362793
+    proxy_corner:
+      avg_avg_return:                           0.68155605
+      avg_reward_per_step:                      0.09362793
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     34.237476
+    avg_avg_return:                             0.68155605
+    avg_reward_per_step:                        0.09362793
+    proxy_corner:
+      avg_avg_return:                           0.68155605
+      avg_reward_per_step:                      0.09362793
+  perf:
+    env_steps_per_second:                       258096.36438196924
+    ppo_updates_per_second:                     41.88007469824055
+    eval-batch-orig-duration:                   0.5496146883815527
+    eval-batch-shift-duration:                  0.4928244296461344
+    eval-batch-tree-duration:                   0.4926386009901762
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             21.429443
+      mouse-cheese_dist_finite_wavg:            9.327015
+      mouse-cheese_dist_solvable_avg:           21.429443
+      mouse-cheese_dist_solvable_wavg:          9.327015
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.4723684
+    visit_patterns:
+      avg_first_visit_time:                     591.4304
+      avg_last_visit_time:                      725.3318
+      num_replay_batches:                       735
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012246018
+    avg_actor_approxkl3:                        0.0011714839
+    avg_actor_clipfrac:                         0.057232667
+    avg_actor_loss:                             -0.00023480951
+    avg_advantage:                              -0.012800689
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.006947358
+    avg_entropy:                                0.33908316
+    avg_grad_norm_pre_clip:                     3.0207822
+    avg_loss:                                   0.0028997862
+    max:
+      max_actor_approxkl1:                      0.0038622878
+      max_actor_approxkl3:                      0.00277913
+      max_actor_clipfrac:                       0.14099121
+      max_actor_loss:                           0.0033983141
+      max_advantage:                            0.2990228
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.008417479
+      max_entropy:                              0.380616
+      max_grad_norm_pre_clip:                   6.3431654
+      max_loss:                                 0.00724277
+    std:
+      std_actor_approxkl1:                      0.0009509927
+      std_actor_approxkl3:                      0.0006297681
+      std_actor_clipfrac:                       0.029273296
+      std_actor_loss:                           0.0014406461
+      std_advantage:                            0.11793414
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0008599709
+      std_entropy:                              0.025234252
+      std_grad_norm_pre_clip:                   1.4793464
+      std_loss:                                 0.0016171426
+  eval-batch-orig:
+    avg_avg_episode_length:                     30.15557
+    avg_avg_return:                             0.8114462
+    avg_benchmark_regret:                       0.17338033
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.07850647
+    proxy_corner:
+      avg_avg_return:                           0.8114462
+      avg_reward_per_step:                      0.07850647
+  eval-batch-shift:
+    avg_avg_episode_length:                     73.60071
+    avg_avg_return:                             0.18173335
+    avg_benchmark_regret:                       0.80515707
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.02594757
+    proxy_corner:
+      avg_avg_return:                           0.58908194
+      avg_reward_per_step:                      0.0048675537
+  eval-batch-tree:
+    avg_avg_episode_length:                     31.801075
+    avg_avg_return:                             0.78659874
+    avg_benchmark_regret:                       0.19747688
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.09474182
+    proxy_corner:
+      avg_avg_return:                           0.78659874
+      avg_reward_per_step:                      0.09474182
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        48234496
+    env-step-replay-before:                     24707072
+    env-step-used-before:                       48234496
+    ppo-update-before:                          29440
+    env-step-generate-before:                   23527424
+    env-step-all-after:                         48267264
+    env-step-replay-after:                      24739840
+    env-step-used-after:                        48267264
+    ppo-update-after:                           29460
+    env-step-generate-after:                    23527424
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     56.35374
+    avg_avg_return:                             0.51829225
+    avg_reward_per_step:                        0.016479492
+    proxy_corner:
+      avg_avg_return:                           0.51829225
+      avg_reward_per_step:                      0.016479492
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     56.35374
+    avg_avg_return:                             0.51829225
+    avg_reward_per_step:                        0.016479492
+    proxy_corner:
+      avg_avg_return:                           0.51829225
+      avg_reward_per_step:                      0.016479492
+  perf:
+    env_steps_per_second:                       290318.7452965791
+    ppo_updates_per_second:                     42.08757267025961
+    eval-batch-orig-duration:                   0.5319537818431854
+    eval-batch-shift-duration:                  0.48674180917441845
+    eval-batch-tree-duration:                   0.48406871035695076
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             21.641357
+      mouse-cheese_dist_finite_wavg:            9.35682
+      mouse-cheese_dist_solvable_avg:           21.641357
+      mouse-cheese_dist_solvable_wavg:          9.35682
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.4694255
+    visit_patterns:
+      avg_first_visit_time:                     606.5735
+      avg_last_visit_time:                      745.3364
+      num_replay_batches:                       755
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012054108
+    avg_actor_approxkl3:                        0.0012805637
+    avg_actor_clipfrac:                         0.068127446
+    avg_actor_loss:                             -0.0010552563
+    avg_advantage:                              -0.0044746697
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.008527238
+    avg_entropy:                                0.5034785
+    avg_grad_norm_pre_clip:                     2.7988632
+    avg_loss:                                   0.0027048844
+    max:
+      max_actor_approxkl1:                      0.0032459402
+      max_actor_approxkl3:                      0.0022289802
+      max_actor_clipfrac:                       0.11743164
+      max_actor_loss:                           0.0010965616
+      max_advantage:                            0.31379354
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.010732107
+      max_entropy:                              0.5468998
+      max_grad_norm_pre_clip:                   5.56119
+      max_loss:                                 0.0051293727
+    std:
+      std_actor_approxkl1:                      0.0007444941
+      std_actor_approxkl3:                      0.0005453337
+      std_actor_clipfrac:                       0.030336495
+      std_actor_loss:                           0.001204075
+      std_advantage:                            0.13095781
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.001088735
+      std_entropy:                              0.019040443
+      std_grad_norm_pre_clip:                   1.1892961
+      std_loss:                                 0.0010596389
+  eval-batch-orig:
+    avg_avg_episode_length:                     30.46438
+    avg_avg_return:                             0.78156745
+    avg_benchmark_regret:                       0.20325904
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08037567
+    proxy_corner:
+      avg_avg_return:                           0.78156745
+      avg_reward_per_step:                      0.08037567
+  eval-batch-shift:
+    avg_avg_episode_length:                     74.85791
+    avg_avg_return:                             0.16297407
+    avg_benchmark_regret:                       0.8239164
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.02431488
+    proxy_corner:
+      avg_avg_return:                           0.56898665
+      avg_reward_per_step:                      0.004699707
+  eval-batch-tree:
+    avg_avg_episode_length:                     31.91882
+    avg_avg_return:                             0.77308995
+    avg_benchmark_regret:                       0.21098568
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.0963974
+    proxy_corner:
+      avg_avg_return:                           0.77308995
+      avg_reward_per_step:                      0.0963974
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        49283072
+    env-step-replay-before:                     25493504
+    env-step-used-before:                       49283072
+    ppo-update-before:                          30080
+    env-step-generate-before:                   23789568
+    env-step-all-after:                         49315840
+    env-step-replay-after:                      25526272
+    env-step-used-after:                        49315840
+    ppo-update-after:                           30100
+    env-step-generate-after:                    23789568
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     60.78204
+    avg_avg_return:                             0.50655615
+    avg_reward_per_step:                        0.014831543
+    proxy_corner:
+      avg_avg_return:                           0.50655615
+      avg_reward_per_step:                      0.014831543
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     60.78204
+    avg_avg_return:                             0.50655615
+    avg_reward_per_step:                        0.014831543
+    proxy_corner:
+      avg_avg_return:                           0.50655615
+      avg_reward_per_step:                      0.014831543
+  perf:
+    env_steps_per_second:                       312120.2375572535
+    ppo_updates_per_second:                     41.89936972722415
+    eval-batch-orig-duration:                   0.5392722971737385
+    eval-batch-shift-duration:                  0.4924561344087124
+    eval-batch-tree-duration:                   0.4919697204604745
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             21.784424
+      mouse-cheese_dist_finite_wavg:            12.079809
+      mouse-cheese_dist_solvable_avg:           21.784424
+      mouse-cheese_dist_solvable_wavg:          12.079811
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.45676455
+    visit_patterns:
+      avg_first_visit_time:                     617.8916
+      avg_last_visit_time:                      769.2053
+      num_replay_batches:                       779
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014549447
+    avg_actor_approxkl3:                        0.001360745
+    avg_actor_clipfrac:                         0.07595825
+    avg_actor_loss:                             -0.00044924283
+    avg_advantage:                              -0.002549409
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00764359
+    avg_entropy:                                0.5004803
+    avg_grad_norm_pre_clip:                     3.2072809
+    avg_loss:                                   0.002872072
+    max:
+      max_actor_approxkl1:                      0.0033859983
+      max_actor_approxkl3:                      0.0033755344
+      max_actor_clipfrac:                       0.18347168
+      max_actor_loss:                           0.0046195984
+      max_advantage:                            0.29911202
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.010355379
+      max_entropy:                              0.53006005
+      max_grad_norm_pre_clip:                   6.697426
+      max_loss:                                 0.008419038
+    std:
+      std_actor_approxkl1:                      0.0008771642
+      std_actor_approxkl3:                      0.00083105307
+      std_actor_clipfrac:                       0.04791429
+      std_actor_loss:                           0.001652414
+      std_advantage:                            0.12292154
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0013361257
+      std_entropy:                              0.013223853
+      std_grad_norm_pre_clip:                   1.3961614
+      std_loss:                                 0.0017617567
+  eval-batch-orig:
+    avg_avg_episode_length:                     30.207214
+    avg_avg_return:                             0.7992537
+    avg_benchmark_regret:                       0.18557283
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.07962799
+    proxy_corner:
+      avg_avg_return:                           0.7992537
+      avg_reward_per_step:                      0.07962799
+  eval-batch-shift:
+    avg_avg_episode_length:                     74.10979
+    avg_avg_return:                             0.17556955
+    avg_benchmark_regret:                       0.81132096
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.02583313
+    proxy_corner:
+      avg_avg_return:                           0.5857726
+      avg_reward_per_step:                      0.0048065186
+  eval-batch-tree:
+    avg_avg_episode_length:                     31.888807
+    avg_avg_return:                             0.7812866
+    avg_benchmark_regret:                       0.20278901
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.09541321
+    proxy_corner:
+      avg_avg_return:                           0.7812866
+      avg_reward_per_step:                      0.09541321
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        50331648
+    env-step-replay-before:                     26116096
+    env-step-used-before:                       50331648
+    ppo-update-before:                          30720
+    env-step-generate-before:                   24215552
+    env-step-all-after:                         50364416
+    env-step-replay-after:                      26116096
+    env-step-used-after:                        50364416
+    ppo-update-after:                           30740
+    env-step-generate-after:                    24248320
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     33.918144
+    avg_avg_return:                             0.7030158
+    avg_reward_per_step:                        0.08679199
+    proxy_corner:
+      avg_avg_return:                           0.7030158
+      avg_reward_per_step:                      0.08679199
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     33.918144
+    avg_avg_return:                             0.7030158
+    avg_reward_per_step:                        0.08679199
+    proxy_corner:
+      avg_avg_return:                           0.7030158
+      avg_reward_per_step:                      0.08679199
+  perf:
+    env_steps_per_second:                       312865.5953418519
+    ppo_updates_per_second:                     42.15420688879303
+    eval-batch-orig-duration:                   0.5307445656508207
+    eval-batch-shift-duration:                  0.48667954839766026
+    eval-batch-tree-duration:                   0.4704671520739794
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             21.97168
+      mouse-cheese_dist_finite_wavg:            12.985166
+      mouse-cheese_dist_solvable_avg:           21.97168
+      mouse-cheese_dist_solvable_wavg:          12.9851675
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.45659977
+    visit_patterns:
+      avg_first_visit_time:                     636.22876
+      avg_last_visit_time:                      787.2837
+      num_replay_batches:                       797
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010678432
+    avg_actor_approxkl3:                        0.000738108
+    avg_actor_clipfrac:                         0.036956787
+    avg_actor_loss:                             -0.0011259702
+    avg_advantage:                              0.0025840467
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.005632342
+    avg_entropy:                                0.28806382
+    avg_grad_norm_pre_clip:                     2.289173
+    avg_loss:                                   0.0014021372
+    max:
+      max_actor_approxkl1:                      0.0032384533
+      max_actor_approxkl3:                      0.0019532207
+      max_actor_clipfrac:                       0.09729004
+      max_actor_loss:                           0.000699006
+      max_advantage:                            0.27350074
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.007850607
+      max_entropy:                              0.37770095
+      max_grad_norm_pre_clip:                   4.869051
+      max_loss:                                 0.0034213627
+    std:
+      std_actor_approxkl1:                      0.0008417077
+      std_actor_approxkl3:                      0.00044955083
+      std_actor_clipfrac:                       0.021146752
+      std_actor_loss:                           0.0010795599
+      std_advantage:                            0.10676935
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0009600433
+      std_entropy:                              0.03127974
+      std_grad_norm_pre_clip:                   0.88254654
+      std_loss:                                 0.0012057029
+  eval-batch-orig:
+    avg_avg_episode_length:                     29.259285
+    avg_avg_return:                             0.81603444
+    avg_benchmark_regret:                       0.16879207
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08000946
+    proxy_corner:
+      avg_avg_return:                           0.81603444
+      avg_reward_per_step:                      0.08000946
+  eval-batch-shift:
+    avg_avg_episode_length:                     73.25548
+    avg_avg_return:                             0.18435916
+    avg_benchmark_regret:                       0.80253124
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.027328491
+    proxy_corner:
+      avg_avg_return:                           0.59820664
+      avg_reward_per_step:                      0.004966736
+  eval-batch-tree:
+    avg_avg_episode_length:                     30.54023
+    avg_avg_return:                             0.80412626
+    avg_benchmark_regret:                       0.17994937
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.09606171
+    proxy_corner:
+      avg_avg_return:                           0.80412626
+      avg_reward_per_step:                      0.09606171
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        51380224
+    env-step-replay-before:                     26640384
+    env-step-used-before:                       51380224
+    ppo-update-before:                          31360
+    env-step-generate-before:                   24739840
+    env-step-all-after:                         51412992
+    env-step-replay-after:                      26640384
+    env-step-used-after:                        51412992
+    ppo-update-after:                           31380
+    env-step-generate-after:                    24772608
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     31.40094
+    avg_avg_return:                             0.7243644
+    avg_reward_per_step:                        0.09136963
+    proxy_corner:
+      avg_avg_return:                           0.7243644
+      avg_reward_per_step:                      0.09136963
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     31.40094
+    avg_avg_return:                             0.7243644
+    avg_reward_per_step:                        0.09136963
+    proxy_corner:
+      avg_avg_return:                           0.7243644
+      avg_reward_per_step:                      0.09136963
+  perf:
+    env_steps_per_second:                       350636.62275718234
+    ppo_updates_per_second:                     40.73688125569844
+    eval-batch-orig-duration:                   0.4985413169488311
+    eval-batch-shift-duration:                  0.4487718753516674
+    eval-batch-tree-duration:                   0.4792495034635067
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             22.125732
+      mouse-cheese_dist_finite_wavg:            11.208405
+      mouse-cheese_dist_solvable_avg:           22.125732
+      mouse-cheese_dist_solvable_wavg:          11.208405
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.99999
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.4693034
+    visit_patterns:
+      avg_first_visit_time:                     652.68774
+      avg_last_visit_time:                      803.2151
+      num_replay_batches:                       813
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009376504
+    avg_actor_approxkl3:                        0.0007636457
+    avg_actor_clipfrac:                         0.042260744
+    avg_actor_loss:                             -0.0002108261
+    avg_advantage:                              0.011188836
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.005988732
+    avg_entropy:                                0.2644783
+    avg_grad_norm_pre_clip:                     2.9754102
+    avg_loss:                                   0.0025190616
+    max:
+      max_actor_approxkl1:                      0.002282327
+      max_actor_approxkl3:                      0.0015117971
+      max_actor_clipfrac:                       0.08239746
+      max_actor_loss:                           0.002691254
+      max_advantage:                            0.3286903
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.007972316
+      max_entropy:                              0.30342764
+      max_grad_norm_pre_clip:                   4.838012
+      max_loss:                                 0.0054894537
+    std:
+      std_actor_approxkl1:                      0.00063356827
+      std_actor_approxkl3:                      0.00037181182
+      std_actor_clipfrac:                       0.021089377
+      std_actor_loss:                           0.0010966983
+      std_advantage:                            0.10930239
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010545527
+      std_entropy:                              0.0184734
+      std_grad_norm_pre_clip:                   0.9095612
+      std_loss:                                 0.0011510678
+  eval-batch-orig:
+    avg_avg_episode_length:                     28.215118
+    avg_avg_return:                             0.81965744
+    avg_benchmark_regret:                       0.16516912
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08078003
+    proxy_corner:
+      avg_avg_return:                           0.81965744
+      avg_reward_per_step:                      0.08078003
+  eval-batch-shift:
+    avg_avg_episode_length:                     73.48256
+    avg_avg_return:                             0.18520826
+    avg_benchmark_regret:                       0.8016822
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.026885986
+    proxy_corner:
+      avg_avg_return:                           0.6002116
+      avg_reward_per_step:                      0.0049743652
+  eval-batch-tree:
+    avg_avg_episode_length:                     29.279833
+    avg_avg_return:                             0.81923664
+    avg_benchmark_regret:                       0.16483903
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.097229004
+    proxy_corner:
+      avg_avg_return:                           0.81923664
+      avg_reward_per_step:                      0.097229004
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        52428800
+    env-step-replay-before:                     27164672
+    env-step-used-before:                       52428800
+    ppo-update-before:                          32000
+    env-step-generate-before:                   25264128
+    env-step-all-after:                         52461568
+    env-step-replay-after:                      27197440
+    env-step-used-after:                        52461568
+    ppo-update-after:                           32020
+    env-step-generate-after:                    25264128
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     57.791916
+    avg_avg_return:                             0.52062076
+    avg_reward_per_step:                        0.015930176
+    proxy_corner:
+      avg_avg_return:                           0.52062076
+      avg_reward_per_step:                      0.015930176
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     57.791916
+    avg_avg_return:                             0.52062076
+    avg_reward_per_step:                        0.015930176
+    proxy_corner:
+      avg_avg_return:                           0.52062076
+      avg_reward_per_step:                      0.015930176
+  perf:
+    env_steps_per_second:                       350533.8075973713
+    ppo_updates_per_second:                     40.57070400350014
+    eval-batch-orig-duration:                   0.5242247292771935
+    eval-batch-shift-duration:                  0.4854842685163021
+    eval-batch-tree-duration:                   0.45825581159442663
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             22.416992
+      mouse-cheese_dist_finite_wavg:            13.031804
+      mouse-cheese_dist_solvable_avg:           22.416992
+      mouse-cheese_dist_solvable_wavg:          13.031802
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.4636885
+    visit_patterns:
+      avg_first_visit_time:                     670.71875
+      avg_last_visit_time:                      820.23145
+      num_replay_batches:                       830
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011933105
+    avg_actor_approxkl3:                        0.0012057561
+    avg_actor_clipfrac:                         0.06246338
+    avg_actor_loss:                             -0.001122075
+    avg_advantage:                              0.001552801
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.007868753
+    avg_entropy:                                0.44678447
+    avg_grad_norm_pre_clip:                     2.6700244
+    avg_loss:                                   0.0023655165
+    max:
+      max_actor_approxkl1:                      0.0026045283
+      max_actor_approxkl3:                      0.0024662898
+      max_actor_clipfrac:                       0.11193848
+      max_actor_loss:                           0.000599958
+      max_advantage:                            0.31691015
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0104816025
+      max_entropy:                              0.50136787
+      max_grad_norm_pre_clip:                   4.6689935
+      max_loss:                                 0.004932717
+    std:
+      std_actor_approxkl1:                      0.00085528236
+      std_actor_approxkl3:                      0.0005677177
+      std_actor_clipfrac:                       0.026847925
+      std_actor_loss:                           0.0010490791
+      std_advantage:                            0.12556456
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0013044482
+      std_entropy:                              0.020785008
+      std_grad_norm_pre_clip:                   1.0946591
+      std_loss:                                 0.0012433275
+  eval-batch-orig:
+    avg_avg_episode_length:                     28.007286
+    avg_avg_return:                             0.83130777
+    avg_benchmark_regret:                       0.1535187
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08072662
+    proxy_corner:
+      avg_avg_return:                           0.83130777
+      avg_reward_per_step:                      0.08072662
+  eval-batch-shift:
+    avg_avg_episode_length:                     72.748535
+    avg_avg_return:                             0.18817404
+    avg_benchmark_regret:                       0.7987164
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.028839111
+    proxy_corner:
+      avg_avg_return:                           0.60001624
+      avg_reward_per_step:                      0.0049819946
+  eval-batch-tree:
+    avg_avg_episode_length:                     29.039974
+    avg_avg_return:                             0.82699966
+    avg_benchmark_regret:                       0.15707594
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.097328186
+    proxy_corner:
+      avg_avg_return:                           0.82699966
+      avg_reward_per_step:                      0.097328186
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        53477376
+    env-step-replay-before:                     27623424
+    env-step-used-before:                       53477376
+    ppo-update-before:                          32640
+    env-step-generate-before:                   25853952
+    env-step-all-after:                         53510144
+    env-step-replay-after:                      27623424
+    env-step-used-after:                        53510144
+    ppo-update-after:                           32660
+    env-step-generate-after:                    25886720
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     31.774736
+    avg_avg_return:                             0.7225693
+    avg_reward_per_step:                        0.0786438
+    proxy_corner:
+      avg_avg_return:                           0.7225693
+      avg_reward_per_step:                      0.0786438
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     31.774736
+    avg_avg_return:                             0.7225693
+    avg_reward_per_step:                        0.0786438
+    proxy_corner:
+      avg_avg_return:                           0.7225693
+      avg_reward_per_step:                      0.0786438
+  perf:
+    env_steps_per_second:                       347592.2333837451
+    ppo_updates_per_second:                     40.79811903869636
+    eval-batch-orig-duration:                   0.5277444776147604
+    eval-batch-shift-duration:                  0.44781976006925106
+    eval-batch-tree-duration:                   0.47244460321962833
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             22.64331
+      mouse-cheese_dist_finite_wavg:            13.958633
+      mouse-cheese_dist_solvable_avg:           22.64331
+      mouse-cheese_dist_solvable_wavg:          13.958635
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.46465865
+    visit_patterns:
+      avg_first_visit_time:                     689.70337
+      avg_last_visit_time:                      833.3484
+      num_replay_batches:                       843
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001339424
+    avg_actor_approxkl3:                        0.0010082811
+    avg_actor_clipfrac:                         0.04713135
+    avg_actor_loss:                             -0.00043602588
+    avg_advantage:                              0.0070055053
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00557949
+    avg_entropy:                                0.23950045
+    avg_grad_norm_pre_clip:                     2.4310186
+    avg_loss:                                   0.0021142191
+    max:
+      max_actor_approxkl1:                      0.003508565
+      max_actor_approxkl3:                      0.0020395922
+      max_actor_clipfrac:                       0.083984375
+      max_actor_loss:                           0.0019144043
+      max_advantage:                            0.30022347
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0070832795
+      max_entropy:                              0.26319122
+      max_grad_norm_pre_clip:                   4.6119084
+      max_loss:                                 0.0052139233
+    std:
+      std_actor_approxkl1:                      0.0008411306
+      std_actor_approxkl3:                      0.0005457889
+      std_actor_clipfrac:                       0.023116583
+      std_actor_loss:                           0.0012428507
+      std_advantage:                            0.10565905
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0009349579
+      std_entropy:                              0.018081043
+      std_grad_norm_pre_clip:                   0.99686366
+      std_loss:                                 0.0013640453
+  eval-batch-orig:
+    avg_avg_episode_length:                     27.780685
+    avg_avg_return:                             0.81587315
+    avg_benchmark_regret:                       0.16895337
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.0821228
+    proxy_corner:
+      avg_avg_return:                           0.81587315
+      avg_reward_per_step:                      0.0821228
+  eval-batch-shift:
+    avg_avg_episode_length:                     73.64247
+    avg_avg_return:                             0.17537145
+    avg_benchmark_regret:                       0.811519
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.027633667
+    proxy_corner:
+      avg_avg_return:                           0.58725286
+      avg_reward_per_step:                      0.004875183
+  eval-batch-tree:
+    avg_avg_episode_length:                     28.885845
+    avg_avg_return:                             0.8201562
+    avg_benchmark_regret:                       0.16391937
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.098716736
+    proxy_corner:
+      avg_avg_return:                           0.8201562
+      avg_reward_per_step:                      0.098716736
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        54525952
+    env-step-replay-before:                     28049408
+    env-step-used-before:                       54525952
+    ppo-update-before:                          33280
+    env-step-generate-before:                   26476544
+    env-step-all-after:                         54558720
+    env-step-replay-after:                      28049408
+    env-step-used-after:                        54558720
+    ppo-update-after:                           33300
+    env-step-generate-after:                    26509312
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     31.031628
+    avg_avg_return:                             0.7296216
+    avg_reward_per_step:                        0.0904541
+    proxy_corner:
+      avg_avg_return:                           0.7296216
+      avg_reward_per_step:                      0.0904541
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     31.031628
+    avg_avg_return:                             0.7296216
+    avg_reward_per_step:                        0.0904541
+    proxy_corner:
+      avg_avg_return:                           0.7296216
+      avg_reward_per_step:                      0.0904541
+  perf:
+    env_steps_per_second:                       295266.69754440215
+    ppo_updates_per_second:                     42.13907327334819
+    eval-batch-orig-duration:                   0.5273171607404947
+    eval-batch-shift-duration:                  0.48270046804100275
+    eval-batch-tree-duration:                   0.4863086938858032
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             22.82251
+      mouse-cheese_dist_finite_wavg:            12.184764
+      mouse-cheese_dist_solvable_avg:           22.82251
+      mouse-cheese_dist_solvable_wavg:          12.184764
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.48256162
+    visit_patterns:
+      avg_first_visit_time:                     710.88403
+      avg_last_visit_time:                      846.25244
+      num_replay_batches:                       856
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014340197
+    avg_actor_approxkl3:                        0.0011370801
+    avg_actor_clipfrac:                         0.05341797
+    avg_actor_loss:                             -2.9224344e-05
+    avg_advantage:                              0.011922994
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0053580063
+    avg_entropy:                                0.22757497
+    avg_grad_norm_pre_clip:                     3.0322373
+    avg_loss:                                   0.0024222033
+    max:
+      max_actor_approxkl1:                      0.003753095
+      max_actor_approxkl3:                      0.0022755512
+      max_actor_clipfrac:                       0.10498047
+      max_actor_loss:                           0.0021716133
+      max_advantage:                            0.30300832
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0064774924
+      max_entropy:                              0.24828593
+      max_grad_norm_pre_clip:                   4.987274
+      max_loss:                                 0.0049447645
+    std:
+      std_actor_approxkl1:                      0.0008685081
+      std_actor_approxkl3:                      0.0005554885
+      std_actor_clipfrac:                       0.023335088
+      std_actor_loss:                           0.0009805759
+      std_advantage:                            0.10351371
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00073262426
+      std_entropy:                              0.015676728
+      std_grad_norm_pre_clip:                   1.0651408
+      std_loss:                                 0.0011422648
+  eval-batch-orig:
+    avg_avg_episode_length:                     28.054478
+    avg_avg_return:                             0.81938064
+    avg_benchmark_regret:                       0.16544583
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08255768
+    proxy_corner:
+      avg_avg_return:                           0.81938064
+      avg_reward_per_step:                      0.08255768
+  eval-batch-shift:
+    avg_avg_episode_length:                     74.07924
+    avg_avg_return:                             0.17262717
+    avg_benchmark_regret:                       0.8142633
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.02796936
+    proxy_corner:
+      avg_avg_return:                           0.5993122
+      avg_reward_per_step:                      0.004875183
+  eval-batch-tree:
+    avg_avg_episode_length:                     28.808552
+    avg_avg_return:                             0.8180028
+    avg_benchmark_regret:                       0.1660728
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.09931183
+    proxy_corner:
+      avg_avg_return:                           0.8180028
+      avg_reward_per_step:                      0.09931183
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        55574528
+    env-step-replay-before:                     28442624
+    env-step-used-before:                       55574528
+    ppo-update-before:                          33920
+    env-step-generate-before:                   27131904
+    env-step-all-after:                         55607296
+    env-step-replay-after:                      28475392
+    env-step-used-after:                        55607296
+    ppo-update-after:                           33940
+    env-step-generate-after:                    27131904
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     57.861813
+    avg_avg_return:                             0.50659895
+    avg_reward_per_step:                        0.016052246
+    proxy_corner:
+      avg_avg_return:                           0.50659895
+      avg_reward_per_step:                      0.016052246
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     57.861813
+    avg_avg_return:                             0.50659895
+    avg_reward_per_step:                        0.016052246
+    proxy_corner:
+      avg_avg_return:                           0.50659895
+      avg_reward_per_step:                      0.016052246
+  perf:
+    env_steps_per_second:                       293994.7649433935
+    ppo_updates_per_second:                     42.13259247629536
+    eval-batch-orig-duration:                   0.5396885741502047
+    eval-batch-shift-duration:                  0.4895980339497328
+    eval-batch-tree-duration:                   0.46555547695606947
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             23.02124
+      mouse-cheese_dist_finite_wavg:            13.099611
+      mouse-cheese_dist_solvable_avg:           23.02124
+      mouse-cheese_dist_solvable_wavg:          13.099611
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016816
+    scoring:
+      avg_scores:                               0.4920766
+    visit_patterns:
+      avg_first_visit_time:                     727.60815
+      avg_last_visit_time:                      859.19385
+      num_replay_batches:                       869
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010035667
+    avg_actor_approxkl3:                        0.0011149225
+    avg_actor_clipfrac:                         0.059436034
+    avg_actor_loss:                             -0.00054331607
+    avg_advantage:                              -0.019066913
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.008591436
+    avg_entropy:                                0.3998706
+    avg_grad_norm_pre_clip:                     2.8646252
+    avg_loss:                                   0.0033525315
+    max:
+      max_actor_approxkl1:                      0.0029008905
+      max_actor_approxkl3:                      0.0028882334
+      max_actor_clipfrac:                       0.13879395
+      max_actor_loss:                           0.0011699349
+      max_advantage:                            0.2903527
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0115561215
+      max_entropy:                              0.4180455
+      max_grad_norm_pre_clip:                   4.555015
+      max_loss:                                 0.0053787488
+    std:
+      std_actor_approxkl1:                      0.0007439351
+      std_actor_approxkl3:                      0.00063914614
+      std_actor_clipfrac:                       0.03198328
+      std_actor_loss:                           0.00088828034
+      std_advantage:                            0.13076875
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010004306
+      std_entropy:                              0.010865014
+      std_grad_norm_pre_clip:                   1.0014975
+      std_loss:                                 0.0010584624
+  eval-batch-orig:
+    avg_avg_episode_length:                     27.758204
+    avg_avg_return:                             0.8248065
+    avg_benchmark_regret:                       0.16002002
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08179474
+    proxy_corner:
+      avg_avg_return:                           0.8248065
+      avg_reward_per_step:                      0.08179474
+  eval-batch-shift:
+    avg_avg_episode_length:                     73.97625
+    avg_avg_return:                             0.18102413
+    avg_benchmark_regret:                       0.80586636
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.027244568
+    proxy_corner:
+      avg_avg_return:                           0.60609233
+      avg_reward_per_step:                      0.005050659
+  eval-batch-tree:
+    avg_avg_episode_length:                     28.193516
+    avg_avg_return:                             0.83694875
+    avg_benchmark_regret:                       0.14712685
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.0974884
+    proxy_corner:
+      avg_avg_return:                           0.83694875
+      avg_reward_per_step:                      0.0974884
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        56623104
+    env-step-replay-before:                     28966912
+    env-step-used-before:                       56623104
+    ppo-update-before:                          34560
+    env-step-generate-before:                   27656192
+    env-step-all-after:                         56655872
+    env-step-replay-after:                      28999680
+    env-step-used-after:                        56655872
+    ppo-update-after:                           34580
+    env-step-generate-after:                    27656192
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     54.62269
+    avg_avg_return:                             0.5678168
+    avg_reward_per_step:                        0.017547607
+    proxy_corner:
+      avg_avg_return:                           0.5678168
+      avg_reward_per_step:                      0.017547607
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     54.62269
+    avg_avg_return:                             0.5678168
+    avg_reward_per_step:                        0.017547607
+    proxy_corner:
+      avg_avg_return:                           0.5678168
+      avg_reward_per_step:                      0.017547607
+  perf:
+    env_steps_per_second:                       289388.3473716699
+    ppo_updates_per_second:                     42.077040993764015
+    eval-batch-orig-duration:                   0.5433234581723809
+    eval-batch-shift-duration:                  0.4943861858919263
+    eval-batch-tree-duration:                   0.49923804868012667
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             23.228271
+      mouse-cheese_dist_finite_wavg:            13.119172
+      mouse-cheese_dist_solvable_avg:           23.228271
+      mouse-cheese_dist_solvable_wavg:          13.119174
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.46755674
+    visit_patterns:
+      avg_first_visit_time:                     744.27954
+      avg_last_visit_time:                      875.323
+      num_replay_batches:                       885
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0024247225
+    avg_actor_approxkl3:                        0.0024798033
+    avg_actor_clipfrac:                         0.092724614
+    avg_actor_loss:                             0.0010763347
+    avg_advantage:                              0.0053996835
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00622927
+    avg_entropy:                                0.3635664
+    avg_grad_norm_pre_clip:                     4.2229
+    avg_loss:                                   0.0038274028
+    max:
+      max_actor_approxkl1:                      0.0069209654
+      max_actor_approxkl3:                      0.005965717
+      max_actor_clipfrac:                       0.17236328
+      max_actor_loss:                           0.004656598
+      max_advantage:                            0.24644172
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0086871255
+      max_entropy:                              0.38698286
+      max_grad_norm_pre_clip:                   7.5609727
+      max_loss:                                 0.0070846127
+    std:
+      std_actor_approxkl1:                      0.0019482662
+      std_actor_approxkl3:                      0.0016415381
+      std_actor_clipfrac:                       0.048000596
+      std_actor_loss:                           0.0014277964
+      std_advantage:                            0.11086897
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0012066192
+      std_entropy:                              0.012383031
+      std_grad_norm_pre_clip:                   1.9776244
+      std_loss:                                 0.0014095854
+  eval-batch-orig:
+    avg_avg_episode_length:                     26.707937
+    avg_avg_return:                             0.8465746
+    avg_benchmark_regret:                       0.13825192
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08168793
+    proxy_corner:
+      avg_avg_return:                           0.8465746
+      avg_reward_per_step:                      0.08168793
+  eval-batch-shift:
+    avg_avg_episode_length:                     72.27605
+    avg_avg_return:                             0.20026684
+    avg_benchmark_regret:                       0.7866236
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.030029297
+    proxy_corner:
+      avg_avg_return:                           0.6153553
+      avg_reward_per_step:                      0.0050811768
+  eval-batch-tree:
+    avg_avg_episode_length:                     27.4582
+    avg_avg_return:                             0.84931684
+    avg_benchmark_regret:                       0.13475877
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.09825897
+    proxy_corner:
+      avg_avg_return:                           0.84931684
+      avg_reward_per_step:                      0.09825897
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        57671680
+    env-step-replay-before:                     29458432
+    env-step-used-before:                       57671680
+    ppo-update-before:                          35200
+    env-step-generate-before:                   28213248
+    env-step-all-after:                         57704448
+    env-step-replay-after:                      29458432
+    env-step-used-after:                        57704448
+    ppo-update-after:                           35220
+    env-step-generate-after:                    28246016
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     25.19564
+    avg_avg_return:                             0.80013317
+    avg_reward_per_step:                        0.10681152
+    proxy_corner:
+      avg_avg_return:                           0.80013317
+      avg_reward_per_step:                      0.10681152
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     25.19564
+    avg_avg_return:                             0.80013317
+    avg_reward_per_step:                        0.10681152
+    proxy_corner:
+      avg_avg_return:                           0.80013317
+      avg_reward_per_step:                      0.10681152
+  perf:
+    env_steps_per_second:                       293389.4030111788
+    ppo_updates_per_second:                     42.14014030044008
+    eval-batch-orig-duration:                   0.5239276885986328
+    eval-batch-shift-duration:                  0.48486106283962727
+    eval-batch-tree-duration:                   0.4753724178299308
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             23.369873
+      mouse-cheese_dist_finite_wavg:            13.123082
+      mouse-cheese_dist_solvable_avg:           23.369873
+      mouse-cheese_dist_solvable_wavg:          13.123082
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.46740663
+    visit_patterns:
+      avg_first_visit_time:                     758.45483
+      avg_last_visit_time:                      889.22876
+      num_replay_batches:                       899
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012430402
+    avg_actor_approxkl3:                        0.0012889594
+    avg_actor_clipfrac:                         0.052282717
+    avg_actor_loss:                             0.0011496696
+    avg_advantage:                              0.013111854
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0031967692
+    avg_entropy:                                0.18374728
+    avg_grad_norm_pre_clip:                     3.1932387
+    avg_loss:                                   0.0025643073
+    max:
+      max_actor_approxkl1:                      0.0034121238
+      max_actor_approxkl3:                      0.0039932225
+      max_actor_clipfrac:                       0.10925293
+      max_actor_loss:                           0.006510988
+      max_advantage:                            0.2461161
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.004912323
+      max_entropy:                              0.20710012
+      max_grad_norm_pre_clip:                   5.5841937
+      max_loss:                                 0.008762152
+    std:
+      std_actor_approxkl1:                      0.00084243575
+      std_actor_approxkl3:                      0.0008518894
+      std_actor_clipfrac:                       0.024803208
+      std_actor_loss:                           0.0021241338
+      std_advantage:                            0.076803215
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0006790424
+      std_entropy:                              0.015801132
+      std_grad_norm_pre_clip:                   1.3175123
+      std_loss:                                 0.0022318815
+  eval-batch-orig:
+    avg_avg_episode_length:                     26.395664
+    avg_avg_return:                             0.8439514
+    avg_benchmark_regret:                       0.1408751
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.083091736
+    proxy_corner:
+      avg_avg_return:                           0.8439514
+      avg_reward_per_step:                      0.083091736
+  eval-batch-shift:
+    avg_avg_episode_length:                     72.75102
+    avg_avg_return:                             0.19472568
+    avg_benchmark_regret:                       0.7921648
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.02960968
+    proxy_corner:
+      avg_avg_return:                           0.5942168
+      avg_reward_per_step:                      0.0049362183
+  eval-batch-tree:
+    avg_avg_episode_length:                     27.250668
+    avg_avg_return:                             0.84589446
+    avg_benchmark_regret:                       0.1381811
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.0996933
+    proxy_corner:
+      avg_avg_return:                           0.84589446
+      avg_reward_per_step:                      0.0996933
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        58720256
+    env-step-replay-before:                     29851648
+    env-step-used-before:                       58720256
+    ppo-update-before:                          35840
+    env-step-generate-before:                   28868608
+    env-step-all-after:                         58753024
+    env-step-replay-after:                      29851648
+    env-step-used-after:                        58753024
+    ppo-update-after:                           35860
+    env-step-generate-after:                    28901376
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     29.890078
+    avg_avg_return:                             0.75089365
+    avg_reward_per_step:                        0.09030151
+    proxy_corner:
+      avg_avg_return:                           0.75089365
+      avg_reward_per_step:                      0.09030151
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     29.890078
+    avg_avg_return:                             0.75089365
+    avg_reward_per_step:                        0.09030151
+    proxy_corner:
+      avg_avg_return:                           0.75089365
+      avg_reward_per_step:                      0.09030151
+  perf:
+    env_steps_per_second:                       353982.9282192812
+    ppo_updates_per_second:                     40.516541213796145
+    eval-batch-orig-duration:                   0.5251258676871657
+    eval-batch-shift-duration:                  0.48503533471375704
+    eval-batch-tree-duration:                   0.4583813436329365
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             23.657227
+      mouse-cheese_dist_finite_wavg:            14.965799
+      mouse-cheese_dist_solvable_avg:           23.657227
+      mouse-cheese_dist_solvable_wavg:          14.965801
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.47918618
+    visit_patterns:
+      avg_first_visit_time:                     777.2373
+      avg_last_visit_time:                      901.1443
+      num_replay_batches:                       911
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013414987
+    avg_actor_approxkl3:                        0.0012395048
+    avg_actor_clipfrac:                         0.051135253
+    avg_actor_loss:                             0.00056776183
+    avg_advantage:                              0.011369413
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00400006
+    avg_entropy:                                0.20486179
+    avg_grad_norm_pre_clip:                     2.6281245
+    avg_loss:                                   0.0023629295
+    max:
+      max_actor_approxkl1:                      0.0033929867
+      max_actor_approxkl3:                      0.0024025606
+      max_actor_clipfrac:                       0.08666992
+      max_actor_loss:                           0.003045909
+      max_advantage:                            0.2585386
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0064240783
+      max_entropy:                              0.23065968
+      max_grad_norm_pre_clip:                   4.50382
+      max_loss:                                 0.0054194205
+    std:
+      std_actor_approxkl1:                      0.0009344256
+      std_actor_approxkl3:                      0.0005722624
+      std_actor_clipfrac:                       0.020185126
+      std_actor_loss:                           0.0012475665
+      std_advantage:                            0.08887427
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010243199
+      std_entropy:                              0.015611722
+      std_grad_norm_pre_clip:                   1.1383886
+      std_loss:                                 0.0015631062
+  eval-batch-orig:
+    avg_avg_episode_length:                     25.382265
+    avg_avg_return:                             0.85359645
+    avg_benchmark_regret:                       0.13123009
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08426666
+    proxy_corner:
+      avg_avg_return:                           0.85359645
+      avg_reward_per_step:                      0.08426666
+  eval-batch-shift:
+    avg_avg_episode_length:                     71.69805
+    avg_avg_return:                             0.20418827
+    avg_benchmark_regret:                       0.7827022
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03050232
+    proxy_corner:
+      avg_avg_return:                           0.581959
+      avg_reward_per_step:                      0.0047912598
+  eval-batch-tree:
+    avg_avg_episode_length:                     26.71685
+    avg_avg_return:                             0.8503948
+    avg_benchmark_regret:                       0.13368079
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10001373
+    proxy_corner:
+      avg_avg_return:                           0.8503948
+      avg_reward_per_step:                      0.10001373
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        59768832
+    env-step-replay-before:                     30441472
+    env-step-used-before:                       59768832
+    ppo-update-before:                          36480
+    env-step-generate-before:                   29327360
+    env-step-all-after:                         59801600
+    env-step-replay-after:                      30474240
+    env-step-used-after:                        59801600
+    ppo-update-after:                           36500
+    env-step-generate-after:                    29327360
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     60.200867
+    avg_avg_return:                             0.509921
+    avg_reward_per_step:                        0.014923096
+    proxy_corner:
+      avg_avg_return:                           0.509921
+      avg_reward_per_step:                      0.014923096
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     60.200867
+    avg_avg_return:                             0.509921
+    avg_reward_per_step:                        0.014923096
+    proxy_corner:
+      avg_avg_return:                           0.509921
+      avg_reward_per_step:                      0.014923096
+  perf:
+    env_steps_per_second:                       308247.3084197137
+    ppo_updates_per_second:                     42.10235283619743
+    eval-batch-orig-duration:                   0.5448182811960578
+    eval-batch-shift-duration:                  0.49246467649936676
+    eval-batch-tree-duration:                   0.4898054637014866
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             23.897217
+      mouse-cheese_dist_finite_wavg:            14.996296
+      mouse-cheese_dist_solvable_avg:           23.897217
+      mouse-cheese_dist_solvable_wavg:          14.996296
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.420168
+    scoring:
+      avg_scores:                               0.47558522
+    visit_patterns:
+      avg_first_visit_time:                     791.70483
+      avg_last_visit_time:                      920.13965
+      num_replay_batches:                       930
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011200357
+    avg_actor_approxkl3:                        0.0013622828
+    avg_actor_clipfrac:                         0.06802978
+    avg_actor_loss:                             0.00014510751
+    avg_advantage:                              -0.0075874883
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.007336881
+    avg_entropy:                                0.357881
+    avg_grad_norm_pre_clip:                     3.2077312
+    avg_loss:                                   0.0034556661
+    max:
+      max_actor_approxkl1:                      0.003496443
+      max_actor_approxkl3:                      0.0027271416
+      max_actor_clipfrac:                       0.12390137
+      max_actor_loss:                           0.003952801
+      max_advantage:                            0.2666031
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.009637503
+      max_entropy:                              0.37636805
+      max_grad_norm_pre_clip:                   5.6478386
+      max_loss:                                 0.006438311
+    std:
+      std_actor_approxkl1:                      0.00085420674
+      std_actor_approxkl3:                      0.00068306207
+      std_actor_clipfrac:                       0.0330164
+      std_actor_loss:                           0.0012852699
+      std_advantage:                            0.12121449
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011387947
+      std_entropy:                              0.0087275
+      std_grad_norm_pre_clip:                   1.2363325
+      std_loss:                                 0.0012635443
+  eval-batch-orig:
+    avg_avg_episode_length:                     25.694002
+    avg_avg_return:                             0.8578621
+    avg_benchmark_regret:                       0.12696438
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.084114075
+    proxy_corner:
+      avg_avg_return:                           0.8578621
+      avg_reward_per_step:                      0.084114075
+  eval-batch-shift:
+    avg_avg_episode_length:                     71.62219
+    avg_avg_return:                             0.21322398
+    avg_benchmark_regret:                       0.7736665
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.030220032
+    proxy_corner:
+      avg_avg_return:                           0.5925623
+      avg_reward_per_step:                      0.00491333
+  eval-batch-tree:
+    avg_avg_episode_length:                     25.604776
+    avg_avg_return:                             0.8675959
+    avg_benchmark_regret:                       0.116479725
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.100357056
+    proxy_corner:
+      avg_avg_return:                           0.8675959
+      avg_reward_per_step:                      0.100357056
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        60817408
+    env-step-replay-before:                     30998528
+    env-step-used-before:                       60817408
+    ppo-update-before:                          37120
+    env-step-generate-before:                   29818880
+    env-step-all-after:                         60850176
+    env-step-replay-after:                      30998528
+    env-step-used-after:                        60850176
+    ppo-update-after:                           37140
+    env-step-generate-after:                    29851648
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     26.492756
+    avg_avg_return:                             0.7966332
+    avg_reward_per_step:                        0.09396362
+    proxy_corner:
+      avg_avg_return:                           0.7966332
+      avg_reward_per_step:                      0.09396362
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     26.492756
+    avg_avg_return:                             0.7966332
+    avg_reward_per_step:                        0.09396362
+    proxy_corner:
+      avg_avg_return:                           0.7966332
+      avg_reward_per_step:                      0.09396362
+  perf:
+    env_steps_per_second:                       315360.0706694511
+    ppo_updates_per_second:                     41.253488582922216
+    eval-batch-orig-duration:                   0.5197977051138878
+    eval-batch-shift-duration:                  0.4437863742932677
+    eval-batch-tree-duration:                   0.47847966477274895
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             24.163818
+      mouse-cheese_dist_finite_wavg:            15.003625
+      mouse-cheese_dist_solvable_avg:           24.163818
+      mouse-cheese_dist_solvable_wavg:          15.003627
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.4738782
+    visit_patterns:
+      avg_first_visit_time:                     805.655
+      avg_last_visit_time:                      936.2754
+      num_replay_batches:                       946
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012129155
+    avg_actor_approxkl3:                        0.0010442126
+    avg_actor_clipfrac:                         0.04232788
+    avg_actor_loss:                             -6.575137e-08
+    avg_advantage:                              0.0234122
+    avg_critic_clipfrac:                        6.713867e-05
+    avg_critic_loss:                            0.0030185035
+    avg_entropy:                                0.16859691
+    avg_grad_norm_pre_clip:                     2.7732444
+    avg_loss:                                   0.001340589
+    max:
+      max_actor_approxkl1:                      0.004085785
+      max_actor_approxkl3:                      0.0031471536
+      max_actor_clipfrac:                       0.09082031
+      max_actor_loss:                           0.0025127456
+      max_advantage:                            0.23769742
+      max_critic_clipfrac:                      0.0013427734
+      max_critic_loss:                          0.0043779137
+      max_entropy:                              0.20353118
+      max_grad_norm_pre_clip:                   4.9417634
+      max_loss:                                 0.0042147916
+    std:
+      std_actor_approxkl1:                      0.0009804765
+      std_actor_approxkl3:                      0.0007458298
+      std_actor_clipfrac:                       0.022594813
+      std_actor_loss:                           0.0013880768
+      std_advantage:                            0.07622218
+      std_critic_clipfrac:                      0.0002926507
+      std_critic_loss:                          0.00067608856
+      std_entropy:                              0.01756579
+      std_grad_norm_pre_clip:                   1.1621218
+      std_loss:                                 0.0013662017
+  eval-batch-orig:
+    avg_avg_episode_length:                     25.27476
+    avg_avg_return:                             0.8673402
+    avg_benchmark_regret:                       0.11748633
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08370972
+    proxy_corner:
+      avg_avg_return:                           0.8673402
+      avg_reward_per_step:                      0.08370972
+  eval-batch-shift:
+    avg_avg_episode_length:                     71.5252
+    avg_avg_return:                             0.20781061
+    avg_benchmark_regret:                       0.77907985
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.030540466
+    proxy_corner:
+      avg_avg_return:                           0.60172534
+      avg_reward_per_step:                      0.0049438477
+  eval-batch-tree:
+    avg_avg_episode_length:                     26.102049
+    avg_avg_return:                             0.8663871
+    avg_benchmark_regret:                       0.11768852
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10058594
+    proxy_corner:
+      avg_avg_return:                           0.8663871
+      avg_reward_per_step:                      0.10058594
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        61865984
+    env-step-replay-before:                     31424512
+    env-step-used-before:                       61865984
+    ppo-update-before:                          37760
+    env-step-generate-before:                   30441472
+    env-step-all-after:                         61898752
+    env-step-replay-after:                      31424512
+    env-step-used-after:                        61898752
+    ppo-update-after:                           37780
+    env-step-generate-after:                    30474240
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     28.054438
+    avg_avg_return:                             0.7768041
+    avg_reward_per_step:                        0.10546875
+    proxy_corner:
+      avg_avg_return:                           0.7768041
+      avg_reward_per_step:                      0.10546875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     28.054438
+    avg_avg_return:                             0.7768041
+    avg_reward_per_step:                        0.10546875
+    proxy_corner:
+      avg_avg_return:                           0.7768041
+      avg_reward_per_step:                      0.10546875
+  perf:
+    env_steps_per_second:                       242773.04389646117
+    ppo_updates_per_second:                     41.99717764301135
+    eval-batch-orig-duration:                   0.5363521268591285
+    eval-batch-shift-duration:                  0.48204185254871845
+    eval-batch-tree-duration:                   0.45156804006546736
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             24.251465
+      mouse-cheese_dist_finite_wavg:            13.22649
+      mouse-cheese_dist_solvable_avg:           24.251465
+      mouse-cheese_dist_solvable_wavg:          13.226493
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.48026758
+    visit_patterns:
+      avg_first_visit_time:                     821.14966
+      avg_last_visit_time:                      949.25073
+      num_replay_batches:                       959
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0018478964
+    avg_actor_approxkl3:                        0.001382346
+    avg_actor_clipfrac:                         0.049688723
+    avg_actor_loss:                             5.5410714e-05
+    avg_advantage:                              0.0031091147
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0031546846
+    avg_entropy:                                0.17928079
+    avg_grad_norm_pre_clip:                     2.4705684
+    avg_loss:                                   0.0014534722
+    max:
+      max_actor_approxkl1:                      0.004277119
+      max_actor_approxkl3:                      0.0031510578
+      max_actor_clipfrac:                       0.09716797
+      max_actor_loss:                           0.0021840967
+      max_advantage:                            0.2013672
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.005354485
+      max_entropy:                              0.23629332
+      max_grad_norm_pre_clip:                   4.785731
+      max_loss:                                 0.0034979566
+    std:
+      std_actor_approxkl1:                      0.0012312001
+      std_actor_approxkl3:                      0.00082588126
+      std_actor_clipfrac:                       0.023515237
+      std_actor_loss:                           0.0009080594
+      std_advantage:                            0.07871959
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010586156
+      std_entropy:                              0.02695786
+      std_grad_norm_pre_clip:                   1.1927218
+      std_loss:                                 0.00096587226
+  eval-batch-orig:
+    avg_avg_episode_length:                     24.77356
+    avg_avg_return:                             0.871835
+    avg_benchmark_regret:                       0.11299151
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08463287
+    proxy_corner:
+      avg_avg_return:                           0.871835
+      avg_reward_per_step:                      0.08463287
+  eval-batch-shift:
+    avg_avg_episode_length:                     71.69173
+    avg_avg_return:                             0.20952505
+    avg_benchmark_regret:                       0.77736545
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.030517578
+    proxy_corner:
+      avg_avg_return:                           0.59855217
+      avg_reward_per_step:                      0.0048980713
+  eval-batch-tree:
+    avg_avg_episode_length:                     25.332796
+    avg_avg_return:                             0.8657219
+    avg_benchmark_regret:                       0.11835368
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.1015625
+    proxy_corner:
+      avg_avg_return:                           0.8657219
+      avg_reward_per_step:                      0.1015625
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        62914560
+    env-step-replay-before:                     31981568
+    env-step-used-before:                       62914560
+    ppo-update-before:                          38400
+    env-step-generate-before:                   30932992
+    env-step-all-after:                         62947328
+    env-step-replay-after:                      31981568
+    env-step-used-after:                        62947328
+    ppo-update-after:                           38420
+    env-step-generate-after:                    30965760
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     26.430462
+    avg_avg_return:                             0.7854214
+    avg_reward_per_step:                        0.106933594
+    proxy_corner:
+      avg_avg_return:                           0.7854214
+      avg_reward_per_step:                      0.106933594
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     26.430462
+    avg_avg_return:                             0.7854214
+    avg_reward_per_step:                        0.106933594
+    proxy_corner:
+      avg_avg_return:                           0.7854214
+      avg_reward_per_step:                      0.106933594
+  perf:
+    env_steps_per_second:                       253850.46725540006
+    ppo_updates_per_second:                     42.31042580707299
+    eval-batch-orig-duration:                   0.6658795746043324
+    eval-batch-shift-duration:                  0.6232955818995833
+    eval-batch-tree-duration:                   0.6175400540232658
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             24.420166
+      mouse-cheese_dist_finite_wavg:            13.236505
+      mouse-cheese_dist_solvable_avg:           24.420166
+      mouse-cheese_dist_solvable_wavg:          13.236506
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.47599918
+    visit_patterns:
+      avg_first_visit_time:                     836.2749
+      avg_last_visit_time:                      966.3157
+      num_replay_batches:                       976
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014074908
+    avg_actor_approxkl3:                        0.0011503972
+    avg_actor_clipfrac:                         0.04729004
+    avg_actor_loss:                             0.00073883816
+    avg_advantage:                              0.013825312
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0032872667
+    avg_entropy:                                0.16997768
+    avg_grad_norm_pre_clip:                     2.580437
+    avg_loss:                                   0.0022124937
+    max:
+      max_actor_approxkl1:                      0.002935979
+      max_actor_approxkl3:                      0.0025204143
+      max_actor_clipfrac:                       0.083740234
+      max_actor_loss:                           0.003061369
+      max_advantage:                            0.20698088
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.004279752
+      max_entropy:                              0.19425222
+      max_grad_norm_pre_clip:                   4.221294
+      max_loss:                                 0.004912681
+    std:
+      std_actor_approxkl1:                      0.00093404
+      std_actor_approxkl3:                      0.00063566194
+      std_actor_clipfrac:                       0.021164346
+      std_actor_loss:                           0.0014525746
+      std_advantage:                            0.080849424
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0005648996
+      std_entropy:                              0.010700379
+      std_grad_norm_pre_clip:                   0.99319357
+      std_loss:                                 0.001511201
+  eval-batch-orig:
+    avg_avg_episode_length:                     24.699278
+    avg_avg_return:                             0.87369657
+    avg_benchmark_regret:                       0.11112994
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08416748
+    proxy_corner:
+      avg_avg_return:                           0.87369657
+      avg_reward_per_step:                      0.08416748
+  eval-batch-shift:
+    avg_avg_episode_length:                     71.66466
+    avg_avg_return:                             0.20355272
+    avg_benchmark_regret:                       0.7833378
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.030097961
+    proxy_corner:
+      avg_avg_return:                           0.59909934
+      avg_reward_per_step:                      0.0049591064
+  eval-batch-tree:
+    avg_avg_episode_length:                     25.678637
+    avg_avg_return:                             0.8607013
+    avg_benchmark_regret:                       0.12337431
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.102012634
+    proxy_corner:
+      avg_avg_return:                           0.8607013
+      avg_reward_per_step:                      0.102012634
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        63963136
+    env-step-replay-before:                     32505856
+    env-step-used-before:                       63963136
+    ppo-update-before:                          39040
+    env-step-generate-before:                   31457280
+    env-step-all-after:                         63995904
+    env-step-replay-after:                      32505856
+    env-step-used-after:                        63995904
+    ppo-update-after:                           39060
+    env-step-generate-after:                    31490048
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     28.735294
+    avg_avg_return:                             0.77543235
+    avg_reward_per_step:                        0.084198
+    proxy_corner:
+      avg_avg_return:                           0.77543235
+      avg_reward_per_step:                      0.084198
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     28.735294
+    avg_avg_return:                             0.77543235
+    avg_reward_per_step:                        0.084198
+    proxy_corner:
+      avg_avg_return:                           0.77543235
+      avg_reward_per_step:                      0.084198
+  perf:
+    env_steps_per_second:                       294697.2378384002
+    ppo_updates_per_second:                     44.372578553420446
+    eval-batch-orig-duration:                   0.5141220418736339
+    eval-batch-shift-duration:                  0.4869855782017112
+    eval-batch-tree-duration:                   0.48654759488999844
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             24.634521
+      mouse-cheese_dist_finite_wavg:            13.257034
+      mouse-cheese_dist_solvable_avg:           24.634521
+      mouse-cheese_dist_solvable_wavg:          13.257036
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.48526534
+    visit_patterns:
+      avg_first_visit_time:                     856.0232
+      avg_last_visit_time:                      982.2036
+      num_replay_batches:                       992
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009394792
+    avg_actor_approxkl3:                        0.00077612663
+    avg_actor_clipfrac:                         0.0357605
+    avg_actor_loss:                             -0.00061213766
+    avg_advantage:                              0.02001478
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0029457142
+    avg_entropy:                                0.17451167
+    avg_grad_norm_pre_clip:                     2.1890507
+    avg_loss:                                   0.0006862076
+    max:
+      max_actor_approxkl1:                      0.0026113065
+      max_actor_approxkl3:                      0.0018280199
+      max_actor_clipfrac:                       0.05810547
+      max_actor_loss:                           0.00064120814
+      max_advantage:                            0.22091806
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0042451913
+      max_entropy:                              0.19151181
+      max_grad_norm_pre_clip:                   3.8278496
+      max_loss:                                 0.0016047539
+    std:
+      std_actor_approxkl1:                      0.0007108208
+      std_actor_approxkl3:                      0.0004045854
+      std_actor_clipfrac:                       0.01449954
+      std_actor_loss:                           0.00071987865
+      std_advantage:                            0.07513564
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0006684241
+      std_entropy:                              0.012206217
+      std_grad_norm_pre_clip:                   0.8650756
+      std_loss:                                 0.00072770217
+  eval-batch-orig:
+    avg_avg_episode_length:                     24.33323
+    avg_avg_return:                             0.8793512
+    avg_benchmark_regret:                       0.10547531
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.084877014
+    proxy_corner:
+      avg_avg_return:                           0.8793512
+      avg_reward_per_step:                      0.084877014
+  eval-batch-shift:
+    avg_avg_episode_length:                     71.61797
+    avg_avg_return:                             0.2046751
+    avg_benchmark_regret:                       0.78221536
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.029785156
+    proxy_corner:
+      avg_avg_return:                           0.6035875
+      avg_reward_per_step:                      0.0049743652
+  eval-batch-tree:
+    avg_avg_episode_length:                     24.59436
+    avg_avg_return:                             0.8775689
+    avg_benchmark_regret:                       0.10650668
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10244751
+    proxy_corner:
+      avg_avg_return:                           0.8775689
+      avg_reward_per_step:                      0.10244751
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        65011712
+    env-step-replay-before:                     33062912
+    env-step-used-before:                       65011712
+    ppo-update-before:                          39680
+    env-step-generate-before:                   31948800
+    env-step-all-after:                         65044480
+    env-step-replay-after:                      33062912
+    env-step-used-after:                        65044480
+    ppo-update-after:                           39700
+    env-step-generate-after:                    31981568
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     28.07526
+    avg_avg_return:                             0.76467115
+    avg_reward_per_step:                        0.11401367
+    proxy_corner:
+      avg_avg_return:                           0.76467115
+      avg_reward_per_step:                      0.11401367
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     28.07526
+    avg_avg_return:                             0.76467115
+    avg_reward_per_step:                        0.11401367
+    proxy_corner:
+      avg_avg_return:                           0.76467115
+      avg_reward_per_step:                      0.11401367
+  perf:
+    env_steps_per_second:                       344603.9536561555
+    ppo_updates_per_second:                     42.301744333342725
+    eval-batch-orig-duration:                   0.5264976853504777
+    eval-batch-shift-duration:                  0.4871955718845129
+    eval-batch-tree-duration:                   0.5003661466762424
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             24.658203
+      mouse-cheese_dist_finite_wavg:            13.26898
+      mouse-cheese_dist_solvable_avg:           24.658203
+      mouse-cheese_dist_solvable_wavg:          13.268982
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.4698735
+    visit_patterns:
+      avg_first_visit_time:                     868.1565
+      avg_last_visit_time:                      999.24927
+      num_replay_batches:                       1009
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011806213
+    avg_actor_approxkl3:                        0.0007549474
+    avg_actor_clipfrac:                         0.03404541
+    avg_actor_loss:                             -0.0007752894
+    avg_advantage:                              0.010417149
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.003710111
+    avg_entropy:                                0.18334627
+    avg_grad_norm_pre_clip:                     1.7514406
+    avg_loss:                                   0.00089642
+    max:
+      max_actor_approxkl1:                      0.0033331627
+      max_actor_approxkl3:                      0.0015238667
+      max_actor_clipfrac:                       0.06359863
+      max_actor_loss:                           0.0011200607
+      max_advantage:                            0.23516262
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0049575167
+      max_entropy:                              0.20978704
+      max_grad_norm_pre_clip:                   3.4902177
+      max_loss:                                 0.0026272892
+    std:
+      std_actor_approxkl1:                      0.00080711953
+      std_actor_approxkl3:                      0.0003867734
+      std_actor_clipfrac:                       0.015086029
+      std_actor_loss:                           0.0009576559
+      std_advantage:                            0.08621861
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0007231577
+      std_entropy:                              0.013650493
+      std_grad_norm_pre_clip:                   0.7423215
+      std_loss:                                 0.0010123921
+  eval-batch-orig:
+    avg_avg_episode_length:                     24.538584
+    avg_avg_return:                             0.8865105
+    avg_benchmark_regret:                       0.098316
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08267212
+    proxy_corner:
+      avg_avg_return:                           0.8865105
+      avg_reward_per_step:                      0.08267212
+  eval-batch-shift:
+    avg_avg_episode_length:                     70.85646
+    avg_avg_return:                             0.2237669
+    avg_benchmark_regret:                       0.7631236
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03152466
+    proxy_corner:
+      avg_avg_return:                           0.6050029
+      avg_reward_per_step:                      0.005065918
+  eval-batch-tree:
+    avg_avg_episode_length:                     25.616667
+    avg_avg_return:                             0.87188274
+    avg_benchmark_regret:                       0.11219289
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.09864044
+    proxy_corner:
+      avg_avg_return:                           0.87188274
+      avg_reward_per_step:                      0.09864044
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        66060288
+    env-step-replay-before:                     33554432
+    env-step-used-before:                       66060288
+    ppo-update-before:                          40320
+    env-step-generate-before:                   32505856
+    env-step-all-after:                         66093056
+    env-step-replay-after:                      33554432
+    env-step-used-after:                        66093056
+    ppo-update-after:                           40340
+    env-step-generate-after:                    32538624
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     25.478231
+    avg_avg_return:                             0.7972815
+    avg_reward_per_step:                        0.09423828
+    proxy_corner:
+      avg_avg_return:                           0.7972815
+      avg_reward_per_step:                      0.09423828
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     25.478231
+    avg_avg_return:                             0.7972815
+    avg_reward_per_step:                        0.09423828
+    proxy_corner:
+      avg_avg_return:                           0.7972815
+      avg_reward_per_step:                      0.09423828
+  perf:
+    env_steps_per_second:                       288636.79686004814
+    ppo_updates_per_second:                     42.087342371854426
+    eval-batch-orig-duration:                   0.5468074809759855
+    eval-batch-shift-duration:                  0.5030338605865836
+    eval-batch-tree-duration:                   0.4734201021492481
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             24.884521
+      mouse-cheese_dist_finite_wavg:            13.293702
+      mouse-cheese_dist_solvable_avg:           24.884521
+      mouse-cheese_dist_solvable_wavg:          13.293704
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.49478343
+    visit_patterns:
+      avg_first_visit_time:                     884.2627
+      avg_last_visit_time:                      1014.2637
+      num_replay_batches:                       1024
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013039096
+    avg_actor_approxkl3:                        0.0011364156
+    avg_actor_clipfrac:                         0.04472656
+    avg_actor_loss:                             0.0005539522
+    avg_advantage:                              0.008913383
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.002778428
+    avg_entropy:                                0.16523471
+    avg_grad_norm_pre_clip:                     2.5336792
+    avg_loss:                                   0.0017779313
+    max:
+      max_actor_approxkl1:                      0.004204846
+      max_actor_approxkl3:                      0.0022687267
+      max_actor_clipfrac:                       0.07836914
+      max_actor_loss:                           0.005026728
+      max_advantage:                            0.2242428
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0049385624
+      max_entropy:                              0.19660187
+      max_grad_norm_pre_clip:                   4.45547
+      max_loss:                                 0.00668227
+    std:
+      std_actor_approxkl1:                      0.0010526241
+      std_actor_approxkl3:                      0.00057160214
+      std_actor_clipfrac:                       0.019387802
+      std_actor_loss:                           0.0016245004
+      std_advantage:                            0.073499955
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0009543879
+      std_entropy:                              0.0137632005
+      std_grad_norm_pre_clip:                   0.98975796
+      std_loss:                                 0.001779977
+  eval-batch-orig:
+    avg_avg_episode_length:                     24.558674
+    avg_avg_return:                             0.87764907
+    avg_benchmark_regret:                       0.107177444
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08467102
+    proxy_corner:
+      avg_avg_return:                           0.87764907
+      avg_reward_per_step:                      0.08467102
+  eval-batch-shift:
+    avg_avg_episode_length:                     72.18219
+    avg_avg_return:                             0.200735
+    avg_benchmark_regret:                       0.78615546
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.030448914
+    proxy_corner:
+      avg_avg_return:                           0.58749044
+      avg_reward_per_step:                      0.004875183
+  eval-batch-tree:
+    avg_avg_episode_length:                     25.110565
+    avg_avg_return:                             0.8716938
+    avg_benchmark_regret:                       0.1123818
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.09933472
+    proxy_corner:
+      avg_avg_return:                           0.8716938
+      avg_reward_per_step:                      0.09933472
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        67108864
+    env-step-replay-before:                     34078720
+    env-step-used-before:                       67108864
+    ppo-update-before:                          40960
+    env-step-generate-before:                   33030144
+    env-step-all-after:                         67141632
+    env-step-replay-after:                      34111488
+    env-step-used-after:                        67141632
+    ppo-update-after:                           40980
+    env-step-generate-after:                    33030144
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     62.694954
+    avg_avg_return:                             0.48903358
+    avg_reward_per_step:                        0.013977051
+    proxy_corner:
+      avg_avg_return:                           0.48903358
+      avg_reward_per_step:                      0.013977051
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     62.694954
+    avg_avg_return:                             0.48903358
+    avg_reward_per_step:                        0.013977051
+    proxy_corner:
+      avg_avg_return:                           0.48903358
+      avg_reward_per_step:                      0.013977051
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  perf:
+    env_steps_per_second:                       327531.7284985058
+    ppo_updates_per_second:                     41.067591091699086
+    eval-batch-orig-duration:                   0.637851401232183
+    eval-batch-shift-duration:                  0.5930430190637708
+    eval-batch-tree-duration:                   0.5835487116128206
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             25.072266
+      mouse-cheese_dist_finite_wavg:            14.205626
+      mouse-cheese_dist_solvable_avg:           25.072266
+      mouse-cheese_dist_solvable_wavg:          14.205624
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.48350018
+    visit_patterns:
+      avg_first_visit_time:                     896.07056
+      avg_last_visit_time:                      1031.2261
+      num_replay_batches:                       1041
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0016715368
+    avg_actor_approxkl3:                        0.0010530401
+    avg_actor_clipfrac:                         0.052081298
+    avg_actor_loss:                             -0.0011958821
+    avg_advantage:                              -0.011059513
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.007808381
+    avg_entropy:                                0.332424
+    avg_grad_norm_pre_clip:                     2.0456822
+    avg_loss:                                   0.0023758851
+    max:
+      max_actor_approxkl1:                      0.0028750913
+      max_actor_approxkl3:                      0.0020227877
+      max_actor_clipfrac:                       0.07910156
+      max_actor_loss:                           0.00027419627
+      max_advantage:                            0.22642076
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.01115535
+      max_entropy:                              0.3576976
+      max_grad_norm_pre_clip:                   3.2205985
+      max_loss:                                 0.004579862
+    std:
+      std_actor_approxkl1:                      0.00090097764
+      std_actor_approxkl3:                      0.0004927531
+      std_actor_clipfrac:                       0.023093248
+      std_actor_loss:                           0.0010911429
+      std_advantage:                            0.12500468
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0012529603
+      std_entropy:                              0.015888032
+      std_grad_norm_pre_clip:                   0.6910215
+      std_loss:                                 0.0012555884
+  eval-batch-orig:
+    avg_avg_episode_length:                     22.798758
+    avg_avg_return:                             0.8886624
+    avg_benchmark_regret:                       0.09616408
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.087257385
+    proxy_corner:
+      avg_avg_return:                           0.8886624
+      avg_reward_per_step:                      0.087257385
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-shift:
+    avg_avg_episode_length:                     71.96573
+    avg_avg_return:                             0.19713041
+    avg_benchmark_regret:                       0.78976005
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.029197693
+    proxy_corner:
+      avg_avg_return:                           0.62189734
+      avg_reward_per_step:                      0.0051651
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-tree:
+    avg_avg_episode_length:                     24.280466
+    avg_avg_return:                             0.8817017
+    avg_benchmark_regret:                       0.1023739
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.103652954
+    proxy_corner:
+      avg_avg_return:                           0.8817017
+      avg_reward_per_step:                      0.103652954
+    rollouts_gif:                               float32(132, 225, 225, 3)
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        68157440
+    env-step-replay-before:                     34668544
+    env-step-used-before:                       68157440
+    ppo-update-before:                          41600
+    env-step-generate-before:                   33488896
+    env-step-all-after:                         68190208
+    env-step-replay-after:                      34668544
+    env-step-used-after:                        68190208
+    ppo-update-after:                           41620
+    env-step-generate-after:                    33521664
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     27.034811
+    avg_avg_return:                             0.7824494
+    avg_reward_per_step:                        0.09262085
+    proxy_corner:
+      avg_avg_return:                           0.7824494
+      avg_reward_per_step:                      0.09262085
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     27.034811
+    avg_avg_return:                             0.7824494
+    avg_reward_per_step:                        0.09262085
+    proxy_corner:
+      avg_avg_return:                           0.7824494
+      avg_reward_per_step:                      0.09262085
+  perf:
+    env_steps_per_second:                       293537.5794736374
+    ppo_updates_per_second:                     42.273777318581
+    eval-batch-orig-duration:                   0.5259533915668726
+    eval-batch-shift-duration:                  0.4689053799957037
+    eval-batch-tree-duration:                   0.45809444319456816
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             25.31958
+      mouse-cheese_dist_finite_wavg:            14.237556
+      mouse-cheese_dist_solvable_avg:           25.31958
+      mouse-cheese_dist_solvable_wavg:          14.237556
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.47486287
+    visit_patterns:
+      avg_first_visit_time:                     912.27466
+      avg_last_visit_time:                      1048.3513
+      num_replay_batches:                       1058
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010674916
+    avg_actor_approxkl3:                        0.0009036432
+    avg_actor_clipfrac:                         0.035980225
+    avg_actor_loss:                             -0.00026217513
+    avg_advantage:                              0.019192439
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0033090494
+    avg_entropy:                                0.14982961
+    avg_grad_norm_pre_clip:                     1.8941557
+    avg_loss:                                   0.0012425202
+    max:
+      max_actor_approxkl1:                      0.0025340617
+      max_actor_approxkl3:                      0.0023742495
+      max_actor_clipfrac:                       0.072631836
+      max_actor_loss:                           0.0019859038
+      max_advantage:                            0.24222374
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0049841567
+      max_entropy:                              0.1759426
+      max_grad_norm_pre_clip:                   3.6794403
+      max_loss:                                 0.0034127384
+    std:
+      std_actor_approxkl1:                      0.0007197665
+      std_actor_approxkl3:                      0.00057804724
+      std_actor_clipfrac:                       0.017844154
+      std_actor_loss:                           0.0009986591
+      std_advantage:                            0.08064575
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0007251559
+      std_entropy:                              0.012124084
+      std_grad_norm_pre_clip:                   0.7084306
+      std_loss:                                 0.0010597913
+  eval-batch-orig:
+    avg_avg_episode_length:                     23.585875
+    avg_avg_return:                             0.88672966
+    avg_benchmark_regret:                       0.09809686
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08634186
+    proxy_corner:
+      avg_avg_return:                           0.88672966
+      avg_reward_per_step:                      0.08634186
+  eval-batch-shift:
+    avg_avg_episode_length:                     71.509186
+    avg_avg_return:                             0.20405619
+    avg_benchmark_regret:                       0.7828343
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.030967712
+    proxy_corner:
+      avg_avg_return:                           0.6410027
+      avg_reward_per_step:                      0.005279541
+  eval-batch-tree:
+    avg_avg_episode_length:                     24.397854
+    avg_avg_return:                             0.8850613
+    avg_benchmark_regret:                       0.09901431
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10295105
+    proxy_corner:
+      avg_avg_return:                           0.8850613
+      avg_reward_per_step:                      0.10295105
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        69206016
+    env-step-replay-before:                     35258368
+    env-step-used-before:                       69206016
+    ppo-update-before:                          42240
+    env-step-generate-before:                   33947648
+    env-step-all-after:                         69238784
+    env-step-replay-after:                      35258368
+    env-step-used-after:                        69238784
+    ppo-update-after:                           42260
+    env-step-generate-after:                    33980416
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     25.157825
+    avg_avg_return:                             0.8080147
+    avg_reward_per_step:                        0.09085083
+    proxy_corner:
+      avg_avg_return:                           0.8080147
+      avg_reward_per_step:                      0.09085083
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     25.157825
+    avg_avg_return:                             0.8080147
+    avg_reward_per_step:                        0.09085083
+    proxy_corner:
+      avg_avg_return:                           0.8080147
+      avg_reward_per_step:                      0.09085083
+  perf:
+    env_steps_per_second:                       331178.73419047013
+    ppo_updates_per_second:                     42.311952542584194
+    eval-batch-orig-duration:                   0.5154295917600393
+    eval-batch-shift-duration:                  0.4881757413968444
+    eval-batch-tree-duration:                   0.4898402811959386
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             25.307861
+      mouse-cheese_dist_finite_wavg:            13.340684
+      mouse-cheese_dist_solvable_avg:           25.307861
+      mouse-cheese_dist_solvable_wavg:          13.340684
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.48579556
+    visit_patterns:
+      avg_first_visit_time:                     924.54126
+      avg_last_visit_time:                      1066.2505
+      num_replay_batches:                       1076
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009055732
+    avg_actor_approxkl3:                        0.0011145353
+    avg_actor_clipfrac:                         0.038623046
+    avg_actor_loss:                             0.0006711105
+    avg_advantage:                              0.0018016688
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.002130063
+    avg_entropy:                                0.13160856
+    avg_grad_norm_pre_clip:                     1.8935238
+    avg_loss:                                   0.0016045332
+    max:
+      max_actor_approxkl1:                      0.0029135938
+      max_actor_approxkl3:                      0.0022927737
+      max_actor_clipfrac:                       0.068115234
+      max_actor_loss:                           0.003103368
+      max_advantage:                            0.20166183
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0034018585
+      max_entropy:                              0.14688554
+      max_grad_norm_pre_clip:                   2.738756
+      max_loss:                                 0.0038604243
+    std:
+      std_actor_approxkl1:                      0.000853563
+      std_actor_approxkl3:                      0.00056794967
+      std_actor_clipfrac:                       0.014278178
+      std_actor_loss:                           0.0012322435
+      std_advantage:                            0.06586786
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0007574053
+      std_entropy:                              0.009012165
+      std_grad_norm_pre_clip:                   0.55757815
+      std_loss:                                 0.0012325833
+  eval-batch-orig:
+    avg_avg_episode_length:                     22.803759
+    avg_avg_return:                             0.88985705
+    avg_benchmark_regret:                       0.09496947
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08699036
+    proxy_corner:
+      avg_avg_return:                           0.88985705
+      avg_reward_per_step:                      0.08699036
+  eval-batch-shift:
+    avg_avg_episode_length:                     71.789955
+    avg_avg_return:                             0.202744
+    avg_benchmark_regret:                       0.7841464
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.030036926
+    proxy_corner:
+      avg_avg_return:                           0.61453766
+      avg_reward_per_step:                      0.0050735474
+  eval-batch-tree:
+    avg_avg_episode_length:                     24.139479
+    avg_avg_return:                             0.87908566
+    avg_benchmark_regret:                       0.104989946
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.102996826
+    proxy_corner:
+      avg_avg_return:                           0.87908566
+      avg_reward_per_step:                      0.102996826
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        70254592
+    env-step-replay-before:                     35651584
+    env-step-used-before:                       70254592
+    ppo-update-before:                          42880
+    env-step-generate-before:                   34603008
+    env-step-all-after:                         70287360
+    env-step-replay-after:                      35651584
+    env-step-used-after:                        70287360
+    ppo-update-after:                           42900
+    env-step-generate-after:                    34635776
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     24.065018
+    avg_avg_return:                             0.80427295
+    avg_reward_per_step:                        0.10760498
+    proxy_corner:
+      avg_avg_return:                           0.80427295
+      avg_reward_per_step:                      0.10760498
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     24.065018
+    avg_avg_return:                             0.80427295
+    avg_reward_per_step:                        0.10760498
+    proxy_corner:
+      avg_avg_return:                           0.80427295
+      avg_reward_per_step:                      0.10760498
+  perf:
+    env_steps_per_second:                       296256.45056640304
+    ppo_updates_per_second:                     42.23338196296008
+    eval-batch-orig-duration:                   0.5184981822967529
+    eval-batch-shift-duration:                  0.48328002635389566
+    eval-batch-tree-duration:                   0.483960366807878
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             25.442139
+      mouse-cheese_dist_finite_wavg:            13.336387
+      mouse-cheese_dist_solvable_avg:           25.442139
+      mouse-cheese_dist_solvable_wavg:          13.336389
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.49422082
+    visit_patterns:
+      avg_first_visit_time:                     942.68115
+      avg_last_visit_time:                      1078.3135
+      num_replay_batches:                       1088
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.000917276
+    avg_actor_approxkl3:                        0.0010803434
+    avg_actor_clipfrac:                         0.038183596
+    avg_actor_loss:                             2.3966842e-05
+    avg_advantage:                              0.004007239
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0024444112
+    avg_entropy:                                0.13501574
+    avg_grad_norm_pre_clip:                     2.0090969
+    avg_loss:                                   0.0011111569
+    max:
+      max_actor_approxkl1:                      0.0028894031
+      max_actor_approxkl3:                      0.0019287812
+      max_actor_clipfrac:                       0.061279297
+      max_actor_loss:                           0.0018362217
+      max_advantage:                            0.17733502
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0033162804
+      max_entropy:                              0.16647497
+      max_grad_norm_pre_clip:                   3.266801
+      max_loss:                                 0.0028530883
+    std:
+      std_actor_approxkl1:                      0.0007304081
+      std_actor_approxkl3:                      0.0005229843
+      std_actor_clipfrac:                       0.012458206
+      std_actor_loss:                           0.0011342092
+      std_advantage:                            0.06971189
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0005251771
+      std_entropy:                              0.012189451
+      std_grad_norm_pre_clip:                   0.59663147
+      std_loss:                                 0.0010829879
+  eval-batch-orig:
+    avg_avg_episode_length:                     22.816036
+    avg_avg_return:                             0.89300895
+    avg_benchmark_regret:                       0.09181754
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.087501526
+    proxy_corner:
+      avg_avg_return:                           0.89300895
+      avg_reward_per_step:                      0.087501526
+  eval-batch-shift:
+    avg_avg_episode_length:                     71.64627
+    avg_avg_return:                             0.20292243
+    avg_benchmark_regret:                       0.783968
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.030311584
+    proxy_corner:
+      avg_avg_return:                           0.6089015
+      avg_reward_per_step:                      0.0050354004
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.22822
+    avg_avg_return:                             0.88885593
+    avg_benchmark_regret:                       0.09521964
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.104133606
+    proxy_corner:
+      avg_avg_return:                           0.88885593
+      avg_reward_per_step:                      0.104133606
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        71303168
+    env-step-replay-before:                     36044800
+    env-step-used-before:                       71303168
+    ppo-update-before:                          43520
+    env-step-generate-before:                   35258368
+    env-step-all-after:                         71335936
+    env-step-replay-after:                      36044800
+    env-step-used-after:                        71335936
+    ppo-update-after:                           43540
+    env-step-generate-after:                    35291136
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     26.183416
+    avg_avg_return:                             0.7808163
+    avg_reward_per_step:                        0.0953064
+    proxy_corner:
+      avg_avg_return:                           0.7808163
+      avg_reward_per_step:                      0.0953064
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     26.183416
+    avg_avg_return:                             0.7808163
+    avg_reward_per_step:                        0.0953064
+    proxy_corner:
+      avg_avg_return:                           0.7808163
+      avg_reward_per_step:                      0.0953064
+  perf:
+    env_steps_per_second:                       320018.8131747216
+    ppo_updates_per_second:                     40.607666637967846
+    eval-batch-orig-duration:                   0.5288253054022789
+    eval-batch-shift-duration:                  0.48117940314114094
+    eval-batch-tree-duration:                   0.46136319171637297
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             25.605469
+      mouse-cheese_dist_finite_wavg:            16.05124
+      mouse-cheese_dist_solvable_avg:           25.605469
+      mouse-cheese_dist_solvable_wavg:          16.051243
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.5012067
+    visit_patterns:
+      avg_first_visit_time:                     957.3281
+      avg_last_visit_time:                      1090.3088
+      num_replay_batches:                       1100
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010815061
+    avg_actor_approxkl3:                        0.0007858099
+    avg_actor_clipfrac:                         0.030651856
+    avg_actor_loss:                             -0.00032587396
+    avg_advantage:                              -0.0015318987
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0033211287
+    avg_entropy:                                0.12849012
+    avg_grad_norm_pre_clip:                     1.7702912
+    avg_loss:                                   0.0012062001
+    max:
+      max_actor_approxkl1:                      0.00295475
+      max_actor_approxkl3:                      0.0015796829
+      max_actor_clipfrac:                       0.051879883
+      max_actor_loss:                           0.0022092685
+      max_advantage:                            0.20987082
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.005212322
+      max_entropy:                              0.16987284
+      max_grad_norm_pre_clip:                   5.0354137
+      max_loss:                                 0.0038054958
+    std:
+      std_actor_approxkl1:                      0.0007318461
+      std_actor_approxkl3:                      0.0004284379
+      std_actor_clipfrac:                       0.013683908
+      std_actor_loss:                           0.0013018424
+      std_advantage:                            0.081062
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010938635
+      std_entropy:                              0.01802487
+      std_grad_norm_pre_clip:                   0.97308815
+      std_loss:                                 0.0011322744
+  eval-batch-orig:
+    avg_avg_episode_length:                     22.185858
+    avg_avg_return:                             0.8908124
+    avg_benchmark_regret:                       0.09401415
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08774567
+    proxy_corner:
+      avg_avg_return:                           0.8908124
+      avg_reward_per_step:                      0.08774567
+  eval-batch-shift:
+    avg_avg_episode_length:                     71.46823
+    avg_avg_return:                             0.21500601
+    avg_benchmark_regret:                       0.77188444
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03024292
+    proxy_corner:
+      avg_avg_return:                           0.632501
+      avg_reward_per_step:                      0.005256653
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.604387
+    avg_avg_return:                             0.8831357
+    avg_benchmark_regret:                       0.10093992
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10369873
+    proxy_corner:
+      avg_avg_return:                           0.8831357
+      avg_reward_per_step:                      0.10369873
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        72351744
+    env-step-replay-before:                     36536320
+    env-step-used-before:                       72351744
+    ppo-update-before:                          44160
+    env-step-generate-before:                   35815424
+    env-step-all-after:                         72384512
+    env-step-replay-after:                      36536320
+    env-step-used-after:                        72384512
+    ppo-update-after:                           44180
+    env-step-generate-after:                    35848192
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     22.842884
+    avg_avg_return:                             0.83124125
+    avg_reward_per_step:                        0.106414795
+    proxy_corner:
+      avg_avg_return:                           0.83124125
+      avg_reward_per_step:                      0.106414795
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     22.842884
+    avg_avg_return:                             0.83124125
+    avg_reward_per_step:                        0.106414795
+    proxy_corner:
+      avg_avg_return:                           0.83124125
+      avg_reward_per_step:                      0.106414795
+  perf:
+    env_steps_per_second:                       317888.7316532938
+    ppo_updates_per_second:                     42.25603440836599
+    eval-batch-orig-duration:                   0.5287433825433254
+    eval-batch-shift-duration:                  0.4829269014298916
+    eval-batch-tree-duration:                   0.49041774310171604
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             25.78003
+      mouse-cheese_dist_finite_wavg:            9.778506
+      mouse-cheese_dist_solvable_avg:           25.78003
+      mouse-cheese_dist_solvable_wavg:          9.778507
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.49811834
+    visit_patterns:
+      avg_first_visit_time:                     970.3784
+      avg_last_visit_time:                      1105.1072
+      num_replay_batches:                       1115
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0004458067
+    avg_actor_approxkl3:                        0.00050092593
+    avg_actor_clipfrac:                         0.022406006
+    avg_actor_loss:                             -0.00084866874
+    avg_advantage:                              0.03627509
+    avg_critic_clipfrac:                        0.0030578615
+    avg_critic_loss:                            0.0020938122
+    avg_entropy:                                0.11518582
+    avg_grad_norm_pre_clip:                     1.3853337
+    avg_loss:                                   8.3051586e-05
+    max:
+      max_actor_approxkl1:                      0.0015831254
+      max_actor_approxkl3:                      0.0009055282
+      max_actor_clipfrac:                       0.036621094
+      max_actor_loss:                           0.00094520673
+      max_advantage:                            0.23473543
+      max_critic_clipfrac:                      0.025878906
+      max_critic_loss:                          0.003411411
+      max_entropy:                              0.13127837
+      max_grad_norm_pre_clip:                   2.2649865
+      max_loss:                                 0.0019004177
+    std:
+      std_actor_approxkl1:                      0.00044189085
+      std_actor_approxkl3:                      0.00025818116
+      std_actor_clipfrac:                       0.010299398
+      std_actor_loss:                           0.0009044857
+      std_advantage:                            0.06113453
+      std_critic_clipfrac:                      0.00659637
+      std_critic_loss:                          0.00066308514
+      std_entropy:                              0.009084042
+      std_grad_norm_pre_clip:                   0.39741594
+      std_loss:                                 0.00090380135
+  eval-batch-orig:
+    avg_avg_episode_length:                     21.895658
+    avg_avg_return:                             0.8949135
+    avg_benchmark_regret:                       0.089912996
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08824921
+    proxy_corner:
+      avg_avg_return:                           0.8949135
+      avg_reward_per_step:                      0.08824921
+  eval-batch-shift:
+    avg_avg_episode_length:                     70.828
+    avg_avg_return:                             0.21903503
+    avg_benchmark_regret:                       0.7678554
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03125
+    proxy_corner:
+      avg_avg_return:                           0.63013923
+      avg_reward_per_step:                      0.005279541
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.079556
+    avg_avg_return:                             0.8918383
+    avg_benchmark_regret:                       0.09223728
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.104393005
+    proxy_corner:
+      avg_avg_return:                           0.8918383
+      avg_reward_per_step:                      0.104393005
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        73400320
+    env-step-replay-before:                     37289984
+    env-step-used-before:                       73400320
+    ppo-update-before:                          44800
+    env-step-generate-before:                   36110336
+    env-step-all-after:                         73433088
+    env-step-replay-after:                      37322752
+    env-step-used-after:                        73433088
+    ppo-update-after:                           44820
+    env-step-generate-after:                    36110336
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     60.917755
+    avg_avg_return:                             0.51744473
+    avg_reward_per_step:                        0.014038086
+    proxy_corner:
+      avg_avg_return:                           0.51744473
+      avg_reward_per_step:                      0.014038086
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     60.917755
+    avg_avg_return:                             0.51744473
+    avg_reward_per_step:                        0.014038086
+    proxy_corner:
+      avg_avg_return:                           0.51744473
+      avg_reward_per_step:                      0.014038086
+  perf:
+    env_steps_per_second:                       349789.6128678189
+    ppo_updates_per_second:                     40.67808094669395
+    eval-batch-orig-duration:                   0.5185217624530196
+    eval-batch-shift-duration:                  0.4587233765050769
+    eval-batch-tree-duration:                   0.4792779451236129
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             25.824463
+      mouse-cheese_dist_finite_wavg:            16.085907
+      mouse-cheese_dist_solvable_avg:           25.824463
+      mouse-cheese_dist_solvable_wavg:          16.085909
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.46800965
+    visit_patterns:
+      avg_first_visit_time:                     978.90674
+      avg_last_visit_time:                      1129.2192
+      num_replay_batches:                       1139
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014058193
+    avg_actor_approxkl3:                        0.0009677852
+    avg_actor_clipfrac:                         0.046728518
+    avg_actor_loss:                             -0.0008565262
+    avg_advantage:                              0.0006371774
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.006614439
+    avg_entropy:                                0.280569
+    avg_grad_norm_pre_clip:                     1.8357832
+    avg_loss:                                   0.0021701246
+    max:
+      max_actor_approxkl1:                      0.0028312677
+      max_actor_approxkl3:                      0.0019893297
+      max_actor_clipfrac:                       0.08703613
+      max_actor_loss:                           0.0010782927
+      max_advantage:                            0.26869017
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.008838186
+      max_entropy:                              0.2964209
+      max_grad_norm_pre_clip:                   3.3388748
+      max_loss:                                 0.0044998294
+    std:
+      std_actor_approxkl1:                      0.0008904907
+      std_actor_approxkl3:                      0.00057134614
+      std_actor_clipfrac:                       0.024974214
+      std_actor_loss:                           0.00088646496
+      std_advantage:                            0.11481686
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011969998
+      std_entropy:                              0.010501262
+      std_grad_norm_pre_clip:                   0.7291142
+      std_loss:                                 0.0010597825
+  eval-batch-orig:
+    avg_avg_episode_length:                     22.457102
+    avg_avg_return:                             0.9001095
+    avg_benchmark_regret:                       0.084717005
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.087905884
+    proxy_corner:
+      avg_avg_return:                           0.9001095
+      avg_reward_per_step:                      0.087905884
+  eval-batch-shift:
+    avg_avg_episode_length:                     71.421555
+    avg_avg_return:                             0.21112955
+    avg_benchmark_regret:                       0.7757609
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.029678345
+    proxy_corner:
+      avg_avg_return:                           0.6294068
+      avg_reward_per_step:                      0.0052261353
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.529282
+    avg_avg_return:                             0.8978046
+    avg_benchmark_regret:                       0.08627099
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10519409
+    proxy_corner:
+      avg_avg_return:                           0.8978046
+      avg_reward_per_step:                      0.10519409
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        74448896
+    env-step-replay-before:                     38043648
+    env-step-used-before:                       74448896
+    ppo-update-before:                          45440
+    env-step-generate-before:                   36405248
+    env-step-all-after:                         74481664
+    env-step-replay-after:                      38043648
+    env-step-used-after:                        74481664
+    ppo-update-after:                           45460
+    env-step-generate-after:                    36438016
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     28.691795
+    avg_avg_return:                             0.7719527
+    avg_reward_per_step:                        0.078063965
+    proxy_corner:
+      avg_avg_return:                           0.7719527
+      avg_reward_per_step:                      0.078063965
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     28.691795
+    avg_avg_return:                             0.7719527
+    avg_reward_per_step:                        0.078063965
+    proxy_corner:
+      avg_avg_return:                           0.7719527
+      avg_reward_per_step:                      0.078063965
+  perf:
+    env_steps_per_second:                       258956.34812903663
+    ppo_updates_per_second:                     41.50274379482438
+    eval-batch-orig-duration:                   0.5459720175713301
+    eval-batch-shift-duration:                  0.5117263728752732
+    eval-batch-tree-duration:                   0.4905969528481364
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             25.867676
+      mouse-cheese_dist_finite_wavg:            13.38117
+      mouse-cheese_dist_solvable_avg:           25.867676
+      mouse-cheese_dist_solvable_wavg:          13.38117
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.4576328
+    visit_patterns:
+      avg_first_visit_time:                     990.48413
+      avg_last_visit_time:                      1151.2981
+      num_replay_batches:                       1161
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011324991
+    avg_actor_approxkl3:                        0.00087070925
+    avg_actor_clipfrac:                         0.033428956
+    avg_actor_loss:                             0.00020258575
+    avg_advantage:                              0.02303394
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0033502255
+    avg_entropy:                                0.13107216
+    avg_grad_norm_pre_clip:                     1.8776377
+    avg_loss:                                   0.0017466265
+    max:
+      max_actor_approxkl1:                      0.0025330908
+      max_actor_approxkl3:                      0.0021238602
+      max_actor_clipfrac:                       0.059692383
+      max_actor_loss:                           0.0032595284
+      max_advantage:                            0.24161994
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0050521293
+      max_entropy:                              0.15302667
+      max_grad_norm_pre_clip:                   3.63185
+      max_loss:                                 0.0050897086
+    std:
+      std_actor_approxkl1:                      0.00060902396
+      std_actor_approxkl3:                      0.00048633458
+      std_actor_clipfrac:                       0.012848865
+      std_actor_loss:                           0.00133499
+      std_advantage:                            0.080673106
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010421828
+      std_entropy:                              0.014075047
+      std_grad_norm_pre_clip:                   0.8006558
+      std_loss:                                 0.0012986034
+  eval-batch-orig:
+    avg_avg_episode_length:                     22.092651
+    avg_avg_return:                             0.89731383
+    avg_benchmark_regret:                       0.087512635
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.087753296
+    proxy_corner:
+      avg_avg_return:                           0.89731383
+      avg_reward_per_step:                      0.087753296
+  eval-batch-shift:
+    avg_avg_episode_length:                     70.710266
+    avg_avg_return:                             0.22028078
+    avg_benchmark_regret:                       0.76660967
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.029960632
+    proxy_corner:
+      avg_avg_return:                           0.6000615
+      avg_reward_per_step:                      0.005027771
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.998663
+    avg_avg_return:                             0.893324
+    avg_benchmark_regret:                       0.09075156
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.104385376
+    proxy_corner:
+      avg_avg_return:                           0.893324
+      avg_reward_per_step:                      0.104385376
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        75497472
+    env-step-replay-before:                     38567936
+    env-step-used-before:                       75497472
+    ppo-update-before:                          46080
+    env-step-generate-before:                   36929536
+    env-step-all-after:                         75530240
+    env-step-replay-after:                      38600704
+    env-step-used-after:                        75530240
+    ppo-update-after:                           46100
+    env-step-generate-after:                    36929536
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.364086
+    avg_avg_return:                             0.5369886
+    avg_reward_per_step:                        0.015045166
+    proxy_corner:
+      avg_avg_return:                           0.5369886
+      avg_reward_per_step:                      0.015045166
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.364086
+    avg_avg_return:                             0.5369886
+    avg_reward_per_step:                        0.015045166
+    proxy_corner:
+      avg_avg_return:                           0.5369886
+      avg_reward_per_step:                      0.015045166
+  perf:
+    env_steps_per_second:                       277692.63638747676
+    ppo_updates_per_second:                     42.10548790658045
+    eval-batch-orig-duration:                   0.5386856636032462
+    eval-batch-shift-duration:                  0.49460770282894373
+    eval-batch-tree-duration:                   0.4841267205774784
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             26.012207
+      mouse-cheese_dist_finite_wavg:            16.10343
+      mouse-cheese_dist_solvable_avg:           26.012207
+      mouse-cheese_dist_solvable_wavg:          16.103432
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.46650237
+    visit_patterns:
+      avg_first_visit_time:                     1007.895
+      avg_last_visit_time:                      1168.2725
+      num_replay_batches:                       1178
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012902148
+    avg_actor_approxkl3:                        0.0012035522
+    avg_actor_clipfrac:                         0.05802002
+    avg_actor_loss:                             -4.8146398e-05
+    avg_advantage:                              -0.0063547604
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0059826835
+    avg_entropy:                                0.25571725
+    avg_grad_norm_pre_clip:                     2.520242
+    avg_loss:                                   0.0026874782
+    max:
+      max_actor_approxkl1:                      0.0025976049
+      max_actor_approxkl3:                      0.002374235
+      max_actor_clipfrac:                       0.09460449
+      max_actor_loss:                           0.002457127
+      max_advantage:                            0.21199536
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00822545
+      max_entropy:                              0.27334484
+      max_grad_norm_pre_clip:                   4.2139044
+      max_loss:                                 0.004981496
+    std:
+      std_actor_approxkl1:                      0.0007262065
+      std_actor_approxkl3:                      0.00056593615
+      std_actor_clipfrac:                       0.022613874
+      std_actor_loss:                           0.0011134463
+      std_advantage:                            0.109543666
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011615636
+      std_entropy:                              0.008976416
+      std_grad_norm_pre_clip:                   0.80041707
+      std_loss:                                 0.0010608257
+  eval-batch-orig:
+    avg_avg_episode_length:                     21.280802
+    avg_avg_return:                             0.9099808
+    avg_benchmark_regret:                       0.0748457
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08831024
+    proxy_corner:
+      avg_avg_return:                           0.9099808
+      avg_reward_per_step:                      0.08831024
+  eval-batch-shift:
+    avg_avg_episode_length:                     70.29573
+    avg_avg_return:                             0.22800656
+    avg_benchmark_regret:                       0.75888395
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.031188965
+    proxy_corner:
+      avg_avg_return:                           0.6007186
+      avg_reward_per_step:                      0.005088806
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.116768
+    avg_avg_return:                             0.9033619
+    avg_benchmark_regret:                       0.08071369
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10512543
+    proxy_corner:
+      avg_avg_return:                           0.9033619
+      avg_reward_per_step:                      0.10512543
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        76546048
+    env-step-replay-before:                     39092224
+    env-step-used-before:                       76546048
+    ppo-update-before:                          46720
+    env-step-generate-before:                   37453824
+    env-step-all-after:                         76578816
+    env-step-replay-after:                      39092224
+    env-step-used-after:                        76578816
+    ppo-update-after:                           46740
+    env-step-generate-after:                    37486592
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     26.394852
+    avg_avg_return:                             0.7937123
+    avg_reward_per_step:                        0.08883667
+    proxy_corner:
+      avg_avg_return:                           0.7937123
+      avg_reward_per_step:                      0.08883667
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     26.394852
+    avg_avg_return:                             0.7937123
+    avg_reward_per_step:                        0.08883667
+    proxy_corner:
+      avg_avg_return:                           0.7937123
+      avg_reward_per_step:                      0.08883667
+  perf:
+    env_steps_per_second:                       248193.46420089802
+    ppo_updates_per_second:                     41.420473721473854
+    eval-batch-orig-duration:                   0.5429526893422008
+    eval-batch-shift-duration:                  0.4900509947910905
+    eval-batch-tree-duration:                   0.49319305270910263
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             26.176514
+      mouse-cheese_dist_finite_wavg:            12.502333
+      mouse-cheese_dist_solvable_avg:           26.176514
+      mouse-cheese_dist_solvable_wavg:          12.502335
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.99999
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.4742617
+    visit_patterns:
+      avg_first_visit_time:                     1025.6338
+      avg_last_visit_time:                      1183.1787
+      num_replay_batches:                       1193
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012024223
+    avg_actor_approxkl3:                        0.0009257143
+    avg_actor_clipfrac:                         0.03067627
+    avg_actor_loss:                             -3.2091422e-05
+    avg_advantage:                              0.019551683
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0025065881
+    avg_entropy:                                0.11354313
+    avg_grad_norm_pre_clip:                     1.850466
+    avg_loss:                                   0.0011076594
+    max:
+      max_actor_approxkl1:                      0.0050703357
+      max_actor_approxkl3:                      0.0027685761
+      max_actor_clipfrac:                       0.06506348
+      max_actor_loss:                           0.0020221546
+      max_advantage:                            0.20416081
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00401318
+      max_entropy:                              0.13359311
+      max_grad_norm_pre_clip:                   4.0252023
+      max_loss:                                 0.0039093406
+    std:
+      std_actor_approxkl1:                      0.0012047599
+      std_actor_approxkl3:                      0.00074462715
+      std_actor_clipfrac:                       0.016887013
+      std_actor_loss:                           0.00074457953
+      std_advantage:                            0.06996445
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0007291345
+      std_entropy:                              0.01034714
+      std_grad_norm_pre_clip:                   0.72885305
+      std_loss:                                 0.0008660168
+  eval-batch-orig:
+    avg_avg_episode_length:                     21.766495
+    avg_avg_return:                             0.9017123
+    avg_benchmark_regret:                       0.08311422
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.087898254
+    proxy_corner:
+      avg_avg_return:                           0.9017123
+      avg_reward_per_step:                      0.087898254
+  eval-batch-shift:
+    avg_avg_episode_length:                     71.22208
+    avg_avg_return:                             0.21213284
+    avg_benchmark_regret:                       0.7747576
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.029426575
+    proxy_corner:
+      avg_avg_return:                           0.5622925
+      avg_reward_per_step:                      0.0046920776
+  eval-batch-tree:
+    avg_avg_episode_length:                     23.36942
+    avg_avg_return:                             0.88753486
+    avg_benchmark_regret:                       0.096540734
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10450745
+    proxy_corner:
+      avg_avg_return:                           0.88753486
+      avg_reward_per_step:                      0.10450745
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        77594624
+    env-step-replay-before:                     39714816
+    env-step-used-before:                       77594624
+    ppo-update-before:                          47360
+    env-step-generate-before:                   37879808
+    env-step-all-after:                         77627392
+    env-step-replay-after:                      39714816
+    env-step-used-after:                        77627392
+    ppo-update-after:                           47380
+    env-step-generate-after:                    37912576
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     23.060713
+    avg_avg_return:                             0.81525075
+    avg_reward_per_step:                        0.1031189
+    proxy_corner:
+      avg_avg_return:                           0.81525075
+      avg_reward_per_step:                      0.1031189
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     23.060713
+    avg_avg_return:                             0.81525075
+    avg_reward_per_step:                        0.1031189
+    proxy_corner:
+      avg_avg_return:                           0.81525075
+      avg_reward_per_step:                      0.1031189
+  perf:
+    env_steps_per_second:                       255987.14368537965
+    ppo_updates_per_second:                     42.25764070696243
+    eval-batch-orig-duration:                   0.6809647008776665
+    eval-batch-shift-duration:                  0.6241341661661863
+    eval-batch-tree-duration:                   0.6238775849342346
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             26.27417
+      mouse-cheese_dist_finite_wavg:            12.5225525
+      mouse-cheese_dist_solvable_avg:           26.27417
+      mouse-cheese_dist_solvable_wavg:          12.5225525
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.4824743
+    visit_patterns:
+      avg_first_visit_time:                     1039.0557
+      avg_last_visit_time:                      1202.3247
+      num_replay_batches:                       1212
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0006526492
+    avg_actor_approxkl3:                        0.0005918817
+    avg_actor_clipfrac:                         0.023156738
+    avg_actor_loss:                             -0.0007850284
+    avg_advantage:                              0.023458373
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0023468123
+    avg_entropy:                                0.09889551
+    avg_grad_norm_pre_clip:                     1.4088877
+    avg_loss:                                   0.00028948224
+    max:
+      max_actor_approxkl1:                      0.001522446
+      max_actor_approxkl3:                      0.0013748469
+      max_actor_clipfrac:                       0.045043945
+      max_actor_loss:                           0.0021645091
+      max_advantage:                            0.21685964
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.003515308
+      max_entropy:                              0.1337468
+      max_grad_norm_pre_clip:                   2.6036737
+      max_loss:                                 0.0026873804
+    std:
+      std_actor_approxkl1:                      0.000511061
+      std_actor_approxkl3:                      0.00033106044
+      std_actor_clipfrac:                       0.01136868
+      std_actor_loss:                           0.0012709632
+      std_advantage:                            0.066089995
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00074246386
+      std_entropy:                              0.013781791
+      std_grad_norm_pre_clip:                   0.50641435
+      std_loss:                                 0.0010956319
+  eval-batch-orig:
+    avg_avg_episode_length:                     21.064882
+    avg_avg_return:                             0.90531385
+    avg_benchmark_regret:                       0.07951269
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08960724
+    proxy_corner:
+      avg_avg_return:                           0.90531385
+      avg_reward_per_step:                      0.08960724
+  eval-batch-shift:
+    avg_avg_episode_length:                     72.14267
+    avg_avg_return:                             0.19390622
+    avg_benchmark_regret:                       0.7929842
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.029342651
+    proxy_corner:
+      avg_avg_return:                           0.62720966
+      avg_reward_per_step:                      0.0051651
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.561472
+    avg_avg_return:                             0.8977548
+    avg_benchmark_regret:                       0.08632083
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10556793
+    proxy_corner:
+      avg_avg_return:                           0.8977548
+      avg_reward_per_step:                      0.10556793
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        78643200
+    env-step-replay-before:                     40239104
+    env-step-used-before:                       78643200
+    ppo-update-before:                          48000
+    env-step-generate-before:                   38404096
+    env-step-all-after:                         78675968
+    env-step-replay-after:                      40271872
+    env-step-used-after:                        78675968
+    ppo-update-after:                           48020
+    env-step-generate-after:                    38404096
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     62.56923
+    avg_avg_return:                             0.49272835
+    avg_reward_per_step:                        0.013397217
+    proxy_corner:
+      avg_avg_return:                           0.49272835
+      avg_reward_per_step:                      0.013397217
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     62.56923
+    avg_avg_return:                             0.49272835
+    avg_reward_per_step:                        0.013397217
+    proxy_corner:
+      avg_avg_return:                           0.49272835
+      avg_reward_per_step:                      0.013397217
+  perf:
+    env_steps_per_second:                       297242.7184569116
+    ppo_updates_per_second:                     42.10013971373471
+    eval-batch-orig-duration:                   0.5361269675195217
+    eval-batch-shift-duration:                  0.4735786831006408
+    eval-batch-tree-duration:                   0.35015764087438583
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             26.231445
+      mouse-cheese_dist_finite_wavg:            14.332099
+      mouse-cheese_dist_solvable_avg:           26.231445
+      mouse-cheese_dist_solvable_wavg:          14.332099
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.47975016
+    visit_patterns:
+      avg_first_visit_time:                     1053.7524
+      avg_last_visit_time:                      1219.1919
+      num_replay_batches:                       1229
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013317679
+    avg_actor_approxkl3:                        0.0010064563
+    avg_actor_clipfrac:                         0.04886475
+    avg_actor_loss:                             -0.0013534044
+    avg_advantage:                              -0.012104096
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0074626096
+    avg_entropy:                                0.26266453
+    avg_grad_norm_pre_clip:                     2.034537
+    avg_loss:                                   0.0021152361
+    max:
+      max_actor_approxkl1:                      0.0034361826
+      max_actor_approxkl3:                      0.0017378337
+      max_actor_clipfrac:                       0.07788086
+      max_actor_loss:                           0.00025773048
+      max_advantage:                            0.24482185
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.010302281
+      max_entropy:                              0.28460604
+      max_grad_norm_pre_clip:                   4.4636264
+      max_loss:                                 0.0038401906
+    std:
+      std_actor_approxkl1:                      0.0010451548
+      std_actor_approxkl3:                      0.0005303382
+      std_actor_clipfrac:                       0.024703396
+      std_actor_loss:                           0.0009715632
+      std_advantage:                            0.122014605
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0012316691
+      std_entropy:                              0.011735708
+      std_grad_norm_pre_clip:                   0.87132865
+      std_loss:                                 0.0011306592
+  eval-batch-orig:
+    avg_avg_episode_length:                     21.142677
+    avg_avg_return:                             0.9065434
+    avg_benchmark_regret:                       0.078283146
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08921051
+    proxy_corner:
+      avg_avg_return:                           0.9065434
+      avg_reward_per_step:                      0.08921051
+  eval-batch-shift:
+    avg_avg_episode_length:                     71.10681
+    avg_avg_return:                             0.20776272
+    avg_benchmark_regret:                       0.7791277
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.031547546
+    proxy_corner:
+      avg_avg_return:                           0.61782765
+      avg_reward_per_step:                      0.0051116943
+  eval-batch-tree:
+    avg_avg_episode_length:                     21.867975
+    avg_avg_return:                             0.9070611
+    avg_benchmark_regret:                       0.07701452
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10573578
+    proxy_corner:
+      avg_avg_return:                           0.9070611
+      avg_reward_per_step:                      0.10573578
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        79691776
+    env-step-replay-before:                     40861696
+    env-step-used-before:                       79691776
+    ppo-update-before:                          48640
+    env-step-generate-before:                   38830080
+    env-step-all-after:                         79724544
+    env-step-replay-after:                      40861696
+    env-step-used-after:                        79724544
+    ppo-update-after:                           48660
+    env-step-generate-after:                    38862848
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     24.249348
+    avg_avg_return:                             0.8064053
+    avg_reward_per_step:                        0.118499756
+    proxy_corner:
+      avg_avg_return:                           0.8064053
+      avg_reward_per_step:                      0.118499756
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     24.249348
+    avg_avg_return:                             0.8064053
+    avg_reward_per_step:                        0.118499756
+    proxy_corner:
+      avg_avg_return:                           0.8064053
+      avg_reward_per_step:                      0.118499756
+  perf:
+    env_steps_per_second:                       346631.608571302
+    ppo_updates_per_second:                     40.754741350194806
+    eval-batch-orig-duration:                   0.4967884225770831
+    eval-batch-shift-duration:                  0.43269508332014084
+    eval-batch-tree-duration:                   0.42817223351448774
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             26.332764
+      mouse-cheese_dist_finite_wavg:            15.235502
+      mouse-cheese_dist_solvable_avg:           26.332764
+      mouse-cheese_dist_solvable_wavg:          15.235502
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.99999
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.47256547
+    visit_patterns:
+      avg_first_visit_time:                     1065.2974
+      avg_last_visit_time:                      1237.2329
+      num_replay_batches:                       1247
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0006515002
+    avg_actor_approxkl3:                        0.0006905474
+    avg_actor_clipfrac:                         0.028491212
+    avg_actor_loss:                             -5.1563886e-05
+    avg_advantage:                              0.021452278
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.002789001
+    avg_entropy:                                0.11284187
+    avg_grad_norm_pre_clip:                     1.610383
+    avg_loss:                                   0.0012300946
+    max:
+      max_actor_approxkl1:                      0.0018043299
+      max_actor_approxkl3:                      0.0017779663
+      max_actor_clipfrac:                       0.057861328
+      max_actor_loss:                           0.0029518828
+      max_advantage:                            0.2335934
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0044761216
+      max_entropy:                              0.13498381
+      max_grad_norm_pre_clip:                   4.271323
+      max_loss:                                 0.0036188287
+    std:
+      std_actor_approxkl1:                      0.00057036727
+      std_actor_approxkl3:                      0.0004038376
+      std_actor_clipfrac:                       0.013372992
+      std_actor_loss:                           0.0012893175
+      std_advantage:                            0.074248
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010503313
+      std_entropy:                              0.011970533
+      std_grad_norm_pre_clip:                   0.9135025
+      std_loss:                                 0.0012075015
+  eval-batch-orig:
+    avg_avg_episode_length:                     20.474405
+    avg_avg_return:                             0.9164033
+    avg_benchmark_regret:                       0.06842323
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08904266
+    proxy_corner:
+      avg_avg_return:                           0.9164033
+      avg_reward_per_step:                      0.08904266
+  eval-batch-shift:
+    avg_avg_episode_length:                     69.732994
+    avg_avg_return:                             0.22833148
+    avg_benchmark_regret:                       0.758559
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03212738
+    proxy_corner:
+      avg_avg_return:                           0.61743784
+      avg_reward_per_step:                      0.0051498413
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.044249
+    avg_avg_return:                             0.90928364
+    avg_benchmark_regret:                       0.07479203
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.1054306
+    proxy_corner:
+      avg_avg_return:                           0.90928364
+      avg_reward_per_step:                      0.1054306
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        80740352
+    env-step-replay-before:                     41222144
+    env-step-used-before:                       80740352
+    ppo-update-before:                          49280
+    env-step-generate-before:                   39518208
+    env-step-all-after:                         80773120
+    env-step-replay-after:                      41254912
+    env-step-used-after:                        80773120
+    ppo-update-after:                           49300
+    env-step-generate-after:                    39518208
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     62.67212
+    avg_avg_return:                             0.48662925
+    avg_reward_per_step:                        0.01361084
+    proxy_corner:
+      avg_avg_return:                           0.48662925
+      avg_reward_per_step:                      0.01361084
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     62.67212
+    avg_avg_return:                             0.48662925
+    avg_reward_per_step:                        0.01361084
+    proxy_corner:
+      avg_avg_return:                           0.48662925
+      avg_reward_per_step:                      0.01361084
+  perf:
+    env_steps_per_second:                       347878.63499451603
+    ppo_updates_per_second:                     44.61360888514572
+    eval-batch-orig-duration:                   0.508241968229413
+    eval-batch-shift-duration:                  0.475029113702476
+    eval-batch-tree-duration:                   0.489804407581687
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             26.439209
+      mouse-cheese_dist_finite_wavg:            16.137932
+      mouse-cheese_dist_solvable_avg:           26.439209
+      mouse-cheese_dist_solvable_wavg:          16.137936
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016816
+    scoring:
+      avg_scores:                               0.47035587
+    visit_patterns:
+      avg_first_visit_time:                     1084.2471
+      avg_last_visit_time:                      1249.2522
+      num_replay_batches:                       1259
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0016973227
+    avg_actor_approxkl3:                        0.0014830455
+    avg_actor_clipfrac:                         0.061132815
+    avg_actor_loss:                             4.183762e-05
+    avg_advantage:                              -0.024261642
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.008009512
+    avg_entropy:                                0.2450078
+    avg_grad_norm_pre_clip:                     2.4943235
+    avg_loss:                                   0.0038015866
+    max:
+      max_actor_approxkl1:                      0.0046459045
+      max_actor_approxkl3:                      0.0031125234
+      max_actor_clipfrac:                       0.11193848
+      max_actor_loss:                           0.0023876429
+      max_advantage:                            0.16589868
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.010386515
+      max_entropy:                              0.2612723
+      max_grad_norm_pre_clip:                   4.925086
+      max_loss:                                 0.006652335
+    std:
+      std_actor_approxkl1:                      0.0011555913
+      std_actor_approxkl3:                      0.0007288085
+      std_actor_clipfrac:                       0.02481157
+      std_actor_loss:                           0.0009493348
+      std_advantage:                            0.12450415
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0014343461
+      std_entropy:                              0.0116309235
+      std_grad_norm_pre_clip:                   0.99533683
+      std_loss:                                 0.0012702454
+  eval-batch-orig:
+    avg_avg_episode_length:                     20.619066
+    avg_avg_return:                             0.91382295
+    avg_benchmark_regret:                       0.07100358
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08930969
+    proxy_corner:
+      avg_avg_return:                           0.91382295
+      avg_reward_per_step:                      0.08930969
+  eval-batch-shift:
+    avg_avg_episode_length:                     70.31505
+    avg_avg_return:                             0.22290397
+    avg_benchmark_regret:                       0.76398647
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.032081604
+    proxy_corner:
+      avg_avg_return:                           0.6348356
+      avg_reward_per_step:                      0.0052871704
+  eval-batch-tree:
+    avg_avg_episode_length:                     21.509197
+    avg_avg_return:                             0.9105872
+    avg_benchmark_regret:                       0.073488444
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10623932
+    proxy_corner:
+      avg_avg_return:                           0.9105872
+      avg_reward_per_step:                      0.10623932
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        81788928
+    env-step-replay-before:                     41615360
+    env-step-used-before:                       81788928
+    ppo-update-before:                          49920
+    env-step-generate-before:                   40173568
+    env-step-all-after:                         81821696
+    env-step-replay-after:                      41648128
+    env-step-used-after:                        81821696
+    ppo-update-after:                           49940
+    env-step-generate-after:                    40173568
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     59.51917
+    avg_avg_return:                             0.51065177
+    avg_reward_per_step:                        0.014709473
+    proxy_corner:
+      avg_avg_return:                           0.51065177
+      avg_reward_per_step:                      0.014709473
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     59.51917
+    avg_avg_return:                             0.51065177
+    avg_reward_per_step:                        0.014709473
+    proxy_corner:
+      avg_avg_return:                           0.51065177
+      avg_reward_per_step:                      0.014709473
+  perf:
+    env_steps_per_second:                       250991.85883609264
+    ppo_updates_per_second:                     42.190937882181466
+    eval-batch-orig-duration:                   0.6733823074027896
+    eval-batch-shift-duration:                  0.6409916849806905
+    eval-batch-tree-duration:                   0.6492043118923903
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             26.443848
+      mouse-cheese_dist_finite_wavg:            16.13994
+      mouse-cheese_dist_solvable_avg:           26.443848
+      mouse-cheese_dist_solvable_wavg:          16.139942
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.4880973
+    visit_patterns:
+      avg_first_visit_time:                     1095.6338
+      avg_last_visit_time:                      1261.2549
+      num_replay_batches:                       1271
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012956335
+    avg_actor_approxkl3:                        0.0013097472
+    avg_actor_clipfrac:                         0.055249024
+    avg_actor_loss:                             -0.00051743875
+    avg_advantage:                              -0.01238754
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0073216655
+    avg_entropy:                                0.22848909
+    avg_grad_norm_pre_clip:                     2.1361458
+    avg_loss:                                   0.002914905
+    max:
+      max_actor_approxkl1:                      0.0030910552
+      max_actor_approxkl3:                      0.0028203349
+      max_actor_clipfrac:                       0.09741211
+      max_actor_loss:                           0.0012657642
+      max_advantage:                            0.20162058
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.010988341
+      max_entropy:                              0.2563091
+      max_grad_norm_pre_clip:                   3.6440578
+      max_loss:                                 0.005234404
+    std:
+      std_actor_approxkl1:                      0.000931874
+      std_actor_approxkl3:                      0.000660143
+      std_actor_clipfrac:                       0.021960119
+      std_actor_loss:                           0.00083212665
+      std_advantage:                            0.12056779
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0012048596
+      std_entropy:                              0.01013106
+      std_grad_norm_pre_clip:                   0.5924272
+      std_loss:                                 0.0010419587
+  eval-batch-orig:
+    avg_avg_episode_length:                     20.485935
+    avg_avg_return:                             0.91765136
+    avg_benchmark_regret:                       0.06717515
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09010315
+    proxy_corner:
+      avg_avg_return:                           0.91765136
+      avg_reward_per_step:                      0.09010315
+  eval-batch-shift:
+    avg_avg_episode_length:                     70.21131
+    avg_avg_return:                             0.22715236
+    avg_benchmark_regret:                       0.7597381
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.032333374
+    proxy_corner:
+      avg_avg_return:                           0.6360109
+      avg_reward_per_step:                      0.0052871704
+  eval-batch-tree:
+    avg_avg_episode_length:                     21.579912
+    avg_avg_return:                             0.9084918
+    avg_benchmark_regret:                       0.075583816
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.106910706
+    proxy_corner:
+      avg_avg_return:                           0.9084918
+      avg_reward_per_step:                      0.106910706
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        82837504
+    env-step-replay-before:                     42336256
+    env-step-used-before:                       82837504
+    ppo-update-before:                          50560
+    env-step-generate-before:                   40501248
+    env-step-all-after:                         82870272
+    env-step-replay-after:                      42369024
+    env-step-used-after:                        82870272
+    ppo-update-after:                           50580
+    env-step-generate-after:                    40501248
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.202286
+    avg_avg_return:                             0.55285895
+    avg_reward_per_step:                        0.015289307
+    proxy_corner:
+      avg_avg_return:                           0.55285895
+      avg_reward_per_step:                      0.015289307
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.202286
+    avg_avg_return:                             0.55285895
+    avg_reward_per_step:                        0.015289307
+    proxy_corner:
+      avg_avg_return:                           0.55285895
+      avg_reward_per_step:                      0.015289307
+  perf:
+    env_steps_per_second:                       318844.3190913149
+    ppo_updates_per_second:                     42.22282244078918
+    eval-batch-orig-duration:                   0.5122596574947238
+    eval-batch-shift-duration:                  0.471399100497365
+    eval-batch-tree-duration:                   0.44870874378830194
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             26.47705
+      mouse-cheese_dist_finite_wavg:            13.446766
+      mouse-cheese_dist_solvable_avg:           26.47705
+      mouse-cheese_dist_solvable_wavg:          13.446766
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.46345663
+    visit_patterns:
+      avg_first_visit_time:                     1105.5347
+      avg_last_visit_time:                      1283.1963
+      num_replay_batches:                       1293
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009010679
+    avg_actor_approxkl3:                        0.00091102143
+    avg_actor_clipfrac:                         0.044903565
+    avg_actor_loss:                             -0.00049938896
+    avg_advantage:                              0.0076960265
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0054935324
+    avg_entropy:                                0.23075043
+    avg_grad_norm_pre_clip:                     2.2243874
+    avg_loss:                                   0.0020166272
+    max:
+      max_actor_approxkl1:                      0.00256857
+      max_actor_approxkl3:                      0.0016761001
+      max_actor_clipfrac:                       0.07116699
+      max_actor_loss:                           0.0012275204
+      max_advantage:                            0.2414695
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.007845908
+      max_entropy:                              0.25422418
+      max_grad_norm_pre_clip:                   3.2592382
+      max_loss:                                 0.003297773
+    std:
+      std_actor_approxkl1:                      0.00055049587
+      std_actor_approxkl3:                      0.00036449946
+      std_actor_clipfrac:                       0.016316254
+      std_actor_loss:                           0.0009693905
+      std_advantage:                            0.104998425
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.000977199
+      std_entropy:                              0.0087427
+      std_grad_norm_pre_clip:                   0.5471537
+      std_loss:                                 0.00093863223
+  eval-batch-orig:
+    avg_avg_episode_length:                     20.251652
+    avg_avg_return:                             0.9193894
+    avg_benchmark_regret:                       0.06543708
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.088974
+    proxy_corner:
+      avg_avg_return:                           0.9193894
+      avg_reward_per_step:                      0.088974
+  eval-batch-shift:
+    avg_avg_episode_length:                     69.51025
+    avg_avg_return:                             0.23336661
+    avg_benchmark_regret:                       0.7535238
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03215027
+    proxy_corner:
+      avg_avg_return:                           0.6360818
+      avg_reward_per_step:                      0.0053710938
+  eval-batch-tree:
+    avg_avg_episode_length:                     22.253601
+    avg_avg_return:                             0.90261984
+    avg_benchmark_regret:                       0.081455804
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.105522156
+    proxy_corner:
+      avg_avg_return:                           0.90261984
+      avg_reward_per_step:                      0.105522156
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        83886080
+    env-step-replay-before:                     42860544
+    env-step-used-before:                       83886080
+    ppo-update-before:                          51200
+    env-step-generate-before:                   41025536
+    env-step-all-after:                         83918848
+    env-step-replay-after:                      42893312
+    env-step-used-after:                        83918848
+    ppo-update-after:                           51220
+    env-step-generate-after:                    41025536
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     57.823822
+    avg_avg_return:                             0.55985594
+    avg_reward_per_step:                        0.0154418945
+    proxy_corner:
+      avg_avg_return:                           0.55985594
+      avg_reward_per_step:                      0.0154418945
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     57.823822
+    avg_avg_return:                             0.55985594
+    avg_reward_per_step:                        0.0154418945
+    proxy_corner:
+      avg_avg_return:                           0.55985594
+      avg_reward_per_step:                      0.0154418945
+  perf:
+    env_steps_per_second:                       317408.37797727337
+    ppo_updates_per_second:                     42.27849183976271
+    eval-batch-orig-duration:                   0.5290882335975766
+    eval-batch-shift-duration:                  0.48339850548654795
+    eval-batch-tree-duration:                   0.4835134744644165
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             26.593262
+      mouse-cheese_dist_finite_wavg:            14.365089
+      mouse-cheese_dist_solvable_avg:           26.593262
+      mouse-cheese_dist_solvable_wavg:          14.365091
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.45853382
+    visit_patterns:
+      avg_first_visit_time:                     1121.3774
+      avg_last_visit_time:                      1299.1614
+      num_replay_batches:                       1309
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0018339452
+    avg_actor_approxkl3:                        0.0017799061
+    avg_actor_clipfrac:                         0.063128665
+    avg_actor_loss:                             0.00056626054
+    avg_advantage:                              -0.007901488
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.004821778
+    avg_entropy:                                0.21831675
+    avg_grad_norm_pre_clip:                     2.6248925
+    avg_loss:                                   0.0027588326
+    max:
+      max_actor_approxkl1:                      0.0042815865
+      max_actor_approxkl3:                      0.0046456773
+      max_actor_clipfrac:                       0.114868164
+      max_actor_loss:                           0.0030723363
+      max_advantage:                            0.17932498
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.006461752
+      max_entropy:                              0.23433141
+      max_grad_norm_pre_clip:                   4.2480755
+      max_loss:                                 0.0052992087
+    std:
+      std_actor_approxkl1:                      0.0011830996
+      std_actor_approxkl3:                      0.0010996725
+      std_actor_clipfrac:                       0.02752988
+      std_actor_loss:                           0.0010424906
+      std_advantage:                            0.09824435
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00077583053
+      std_entropy:                              0.007263544
+      std_grad_norm_pre_clip:                   0.92242426
+      std_loss:                                 0.0010548373
+  eval-batch-orig:
+    avg_avg_episode_length:                     19.92007
+    avg_avg_return:                             0.9201256
+    avg_benchmark_regret:                       0.064700946
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08986664
+    proxy_corner:
+      avg_avg_return:                           0.9201256
+      avg_reward_per_step:                      0.08986664
+  eval-batch-shift:
+    avg_avg_episode_length:                     69.80495
+    avg_avg_return:                             0.22653206
+    avg_benchmark_regret:                       0.7603584
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.032188416
+    proxy_corner:
+      avg_avg_return:                           0.6387241
+      avg_reward_per_step:                      0.005264282
+  eval-batch-tree:
+    avg_avg_episode_length:                     21.684193
+    avg_avg_return:                             0.91040516
+    avg_benchmark_regret:                       0.07367046
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10575104
+    proxy_corner:
+      avg_avg_return:                           0.91040516
+      avg_reward_per_step:                      0.10575104
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        84934656
+    env-step-replay-before:                     43483136
+    env-step-used-before:                       84934656
+    ppo-update-before:                          51840
+    env-step-generate-before:                   41451520
+    env-step-all-after:                         84967424
+    env-step-replay-after:                      43515904
+    env-step-used-after:                        84967424
+    ppo-update-after:                           51860
+    env-step-generate-after:                    41451520
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     62.10676
+    avg_avg_return:                             0.4968259
+    avg_reward_per_step:                        0.013580322
+    proxy_corner:
+      avg_avg_return:                           0.4968259
+      avg_reward_per_step:                      0.013580322
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     62.10676
+    avg_avg_return:                             0.4968259
+    avg_reward_per_step:                        0.013580322
+    proxy_corner:
+      avg_avg_return:                           0.4968259
+      avg_reward_per_step:                      0.013580322
+  perf:
+    env_steps_per_second:                       258442.03723913222
+    ppo_updates_per_second:                     41.59097588214972
+    eval-batch-orig-duration:                   0.5292998636141419
+    eval-batch-shift-duration:                  0.46465048752725124
+    eval-batch-tree-duration:                   0.45980692747980356
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             26.64209
+      mouse-cheese_dist_finite_wavg:            13.460583
+      mouse-cheese_dist_solvable_avg:           26.64209
+      mouse-cheese_dist_solvable_wavg:          13.460585
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.45961767
+    visit_patterns:
+      avg_first_visit_time:                     1135.081
+      avg_last_visit_time:                      1318.2112
+      num_replay_batches:                       1328
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0015740177
+    avg_actor_approxkl3:                        0.001390647
+    avg_actor_clipfrac:                         0.058532715
+    avg_actor_loss:                             8.1138314e-05
+    avg_advantage:                              -0.00044476707
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0072269016
+    avg_entropy:                                0.23152807
+    avg_grad_norm_pre_clip:                     2.6265063
+    avg_loss:                                   0.0034630608
+    max:
+      max_actor_approxkl1:                      0.003022985
+      max_actor_approxkl3:                      0.0022330005
+      max_actor_clipfrac:                       0.084472656
+      max_actor_loss:                           0.0017334819
+      max_advantage:                            0.256316
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.008592766
+      max_entropy:                              0.25029188
+      max_grad_norm_pre_clip:                   4.7863216
+      max_loss:                                 0.005140382
+    std:
+      std_actor_approxkl1:                      0.0008243176
+      std_actor_approxkl3:                      0.00058038963
+      std_actor_clipfrac:                       0.020951139
+      std_actor_loss:                           0.0010387889
+      std_advantage:                            0.1201722
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00080542493
+      std_entropy:                              0.01112989
+      std_grad_norm_pre_clip:                   0.94142205
+      std_loss:                                 0.0010744177
+  eval-batch-orig:
+    avg_avg_episode_length:                     20.220797
+    avg_avg_return:                             0.9154918
+    avg_benchmark_regret:                       0.069334656
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09062195
+    proxy_corner:
+      avg_avg_return:                           0.9154918
+      avg_reward_per_step:                      0.09062195
+  eval-batch-shift:
+    avg_avg_episode_length:                     70.30819
+    avg_avg_return:                             0.2267269
+    avg_benchmark_regret:                       0.76016355
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.030197144
+    proxy_corner:
+      avg_avg_return:                           0.6274977
+      avg_reward_per_step:                      0.0051956177
+  eval-batch-tree:
+    avg_avg_episode_length:                     21.900406
+    avg_avg_return:                             0.90193266
+    avg_benchmark_regret:                       0.08214294
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10659027
+    proxy_corner:
+      avg_avg_return:                           0.90193266
+      avg_reward_per_step:                      0.10659027
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        85983232
+    env-step-replay-before:                     44072960
+    env-step-used-before:                       85983232
+    ppo-update-before:                          52480
+    env-step-generate-before:                   41910272
+    env-step-all-after:                         86016000
+    env-step-replay-after:                      44072960
+    env-step-used-after:                        86016000
+    ppo-update-after:                           52500
+    env-step-generate-after:                    41943040
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     22.882555
+    avg_avg_return:                             0.82609755
+    avg_reward_per_step:                        0.10522461
+    proxy_corner:
+      avg_avg_return:                           0.82609755
+      avg_reward_per_step:                      0.10522461
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     22.882555
+    avg_avg_return:                             0.82609755
+    avg_reward_per_step:                        0.10522461
+    proxy_corner:
+      avg_avg_return:                           0.82609755
+      avg_reward_per_step:                      0.10522461
+  perf:
+    env_steps_per_second:                       350381.02725954284
+    ppo_updates_per_second:                     41.07564354929697
+    eval-batch-orig-duration:                   0.5349797196686268
+    eval-batch-shift-duration:                  0.48713810089975595
+    eval-batch-tree-duration:                   0.485955941490829
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             26.657959
+      mouse-cheese_dist_finite_wavg:            13.471092
+      mouse-cheese_dist_solvable_avg:           26.657959
+      mouse-cheese_dist_solvable_wavg:          13.471095
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.47420537
+    visit_patterns:
+      avg_first_visit_time:                     1147.3733
+      avg_last_visit_time:                      1335.2441
+      num_replay_batches:                       1345
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00051991903
+    avg_actor_approxkl3:                        0.00064042717
+    avg_actor_clipfrac:                         0.025231933
+    avg_actor_loss:                             -0.0007791982
+    avg_advantage:                              0.02939199
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.001525794
+    avg_entropy:                                0.093719505
+    avg_grad_norm_pre_clip:                     1.1280133
+    avg_loss:                                   -0.00011002076
+    max:
+      max_actor_approxkl1:                      0.0015452064
+      max_actor_approxkl3:                      0.0011186742
+      max_actor_clipfrac:                       0.037231445
+      max_actor_loss:                           0.0008575246
+      max_advantage:                            0.20979446
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0027354946
+      max_entropy:                              0.112566635
+      max_grad_norm_pre_clip:                   1.796462
+      max_loss:                                 0.0017174251
+    std:
+      std_actor_approxkl1:                      0.00048565265
+      std_actor_approxkl3:                      0.00027898204
+      std_actor_clipfrac:                       0.009300211
+      std_actor_loss:                           0.0008316807
+      std_advantage:                            0.05647185
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.000493179
+      std_entropy:                              0.0100212395
+      std_grad_norm_pre_clip:                   0.41350085
+      std_loss:                                 0.00088467385
+  eval-batch-orig:
+    avg_avg_episode_length:                     20.07186
+    avg_avg_return:                             0.9220432
+    avg_benchmark_regret:                       0.062783316
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08933258
+    proxy_corner:
+      avg_avg_return:                           0.9220432
+      avg_reward_per_step:                      0.08933258
+  eval-batch-shift:
+    avg_avg_episode_length:                     68.06513
+    avg_avg_return:                             0.2580336
+    avg_benchmark_regret:                       0.72885686
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.033607483
+    proxy_corner:
+      avg_avg_return:                           0.62582374
+      avg_reward_per_step:                      0.005279541
+  eval-batch-tree:
+    avg_avg_episode_length:                     21.184008
+    avg_avg_return:                             0.9151546
+    avg_benchmark_regret:                       0.06892107
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.105895996
+    proxy_corner:
+      avg_avg_return:                           0.9151546
+      avg_reward_per_step:                      0.105895996
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        87031808
+    env-step-replay-before:                     44531712
+    env-step-used-before:                       87031808
+    ppo-update-before:                          53120
+    env-step-generate-before:                   42500096
+    env-step-all-after:                         87064576
+    env-step-replay-after:                      44531712
+    env-step-used-after:                        87064576
+    ppo-update-after:                           53140
+    env-step-generate-after:                    42532864
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     23.752895
+    avg_avg_return:                             0.81272006
+    avg_reward_per_step:                        0.09762573
+    proxy_corner:
+      avg_avg_return:                           0.81272006
+      avg_reward_per_step:                      0.09762573
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     23.752895
+    avg_avg_return:                             0.81272006
+    avg_reward_per_step:                        0.09762573
+    proxy_corner:
+      avg_avg_return:                           0.81272006
+      avg_reward_per_step:                      0.09762573
+  perf:
+    env_steps_per_second:                       258355.78977677185
+    ppo_updates_per_second:                     41.42168195498742
+    eval-batch-orig-duration:                   0.540451237000525
+    eval-batch-shift-duration:                  0.48894524946808815
+    eval-batch-tree-duration:                   0.4883411591872573
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             26.722412
+      mouse-cheese_dist_finite_wavg:            16.164793
+      mouse-cheese_dist_solvable_avg:           26.722412
+      mouse-cheese_dist_solvable_wavg:          16.164795
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999977
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.46996915
+    visit_patterns:
+      avg_first_visit_time:                     1161.0913
+      avg_last_visit_time:                      1349.3413
+      num_replay_batches:                       1359
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0008133271
+    avg_actor_approxkl3:                        0.0007118993
+    avg_actor_clipfrac:                         0.023895264
+    avg_actor_loss:                             -0.000364448
+    avg_advantage:                              0.0030768614
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0020509674
+    avg_entropy:                                0.085373245
+    avg_grad_norm_pre_clip:                     1.4385761
+    avg_loss:                                   0.0005756623
+    max:
+      max_actor_approxkl1:                      0.0017097373
+      max_actor_approxkl3:                      0.0013779042
+      max_actor_clipfrac:                       0.038085938
+      max_actor_loss:                           0.0021350589
+      max_advantage:                            0.15444481
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0032479255
+      max_entropy:                              0.10389803
+      max_grad_norm_pre_clip:                   2.8451688
+      max_loss:                                 0.0034550366
+    std:
+      std_actor_approxkl1:                      0.00044470295
+      std_actor_approxkl3:                      0.00027736466
+      std_actor_clipfrac:                       0.0075183692
+      std_actor_loss:                           0.00128345
+      std_advantage:                            0.06402227
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.000709446
+      std_entropy:                              0.0103092175
+      std_grad_norm_pre_clip:                   0.6559486
+      std_loss:                                 0.0012623998
+  eval-batch-orig:
+    avg_avg_episode_length:                     19.965998
+    avg_avg_return:                             0.9171529
+    avg_benchmark_regret:                       0.06767362
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.090904236
+    proxy_corner:
+      avg_avg_return:                           0.9171529
+      avg_reward_per_step:                      0.090904236
+  eval-batch-shift:
+    avg_avg_episode_length:                     71.128975
+    avg_avg_return:                             0.21428451
+    avg_benchmark_regret:                       0.7726059
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.029762268
+    proxy_corner:
+      avg_avg_return:                           0.63685703
+      avg_reward_per_step:                      0.0052261353
+  eval-batch-tree:
+    avg_avg_episode_length:                     21.236382
+    avg_avg_return:                             0.9087411
+    avg_benchmark_regret:                       0.0753345
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10744476
+    proxy_corner:
+      avg_avg_return:                           0.9087411
+      avg_reward_per_step:                      0.10744476
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        88080384
+    env-step-replay-before:                     44859392
+    env-step-used-before:                       88080384
+    ppo-update-before:                          53760
+    env-step-generate-before:                   43220992
+    env-step-all-after:                         88113152
+    env-step-replay-after:                      44859392
+    env-step-used-after:                        88113152
+    ppo-update-after:                           53780
+    env-step-generate-after:                    43253760
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     27.133793
+    avg_avg_return:                             0.78858674
+    avg_reward_per_step:                        0.07980347
+    proxy_corner:
+      avg_avg_return:                           0.78858674
+      avg_reward_per_step:                      0.07980347
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     27.133793
+    avg_avg_return:                             0.78858674
+    avg_reward_per_step:                        0.07980347
+    proxy_corner:
+      avg_avg_return:                           0.78858674
+      avg_reward_per_step:                      0.07980347
+  perf:
+    env_steps_per_second:                       312160.8004408178
+    ppo_updates_per_second:                     42.27229768824754
+    eval-batch-orig-duration:                   0.534251919016242
+    eval-batch-shift-duration:                  0.477720120921731
+    eval-batch-tree-duration:                   0.4887650851160288
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             26.834473
+      mouse-cheese_dist_finite_wavg:            13.474401
+      mouse-cheese_dist_solvable_avg:           26.834473
+      mouse-cheese_dist_solvable_wavg:          13.474403
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.99998
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.48393643
+    visit_patterns:
+      avg_first_visit_time:                     1176.0337
+      avg_last_visit_time:                      1359.3413
+      num_replay_batches:                       1369
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00091822766
+    avg_actor_approxkl3:                        0.0009629838
+    avg_actor_clipfrac:                         0.03362427
+    avg_actor_loss:                             0.0005088299
+    avg_advantage:                              0.0068674767
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0022191738
+    avg_entropy:                                0.100794725
+    avg_grad_norm_pre_clip:                     2.0847926
+    avg_loss:                                   0.0015176223
+    max:
+      max_actor_approxkl1:                      0.0030737799
+      max_actor_approxkl3:                      0.0021609813
+      max_actor_clipfrac:                       0.051635742
+      max_actor_loss:                           0.0025043264
+      max_advantage:                            0.17318344
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.003388626
+      max_entropy:                              0.11233327
+      max_grad_norm_pre_clip:                   3.4255764
+      max_loss:                                 0.0035274478
+    std:
+      std_actor_approxkl1:                      0.0007176664
+      std_actor_approxkl3:                      0.00052713574
+      std_actor_clipfrac:                       0.012667904
+      std_actor_loss:                           0.0012166075
+      std_advantage:                            0.06622405
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00048275304
+      std_entropy:                              0.0075145513
+      std_grad_norm_pre_clip:                   0.6516014
+      std_loss:                                 0.0012568819
+  eval-batch-orig:
+    avg_avg_episode_length:                     19.81594
+    avg_avg_return:                             0.9207983
+    avg_benchmark_regret:                       0.064028166
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09072876
+    proxy_corner:
+      avg_avg_return:                           0.9207983
+      avg_reward_per_step:                      0.09072876
+  eval-batch-shift:
+    avg_avg_episode_length:                     69.899025
+    avg_avg_return:                             0.23395868
+    avg_benchmark_regret:                       0.7529318
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03175354
+    proxy_corner:
+      avg_avg_return:                           0.6318277
+      avg_reward_per_step:                      0.0052490234
+  eval-batch-tree:
+    avg_avg_episode_length:                     20.715603
+    avg_avg_return:                             0.9178351
+    avg_benchmark_regret:                       0.06624051
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.107658386
+    proxy_corner:
+      avg_avg_return:                           0.9178351
+      avg_reward_per_step:                      0.107658386
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        89128960
+    env-step-replay-before:                     45514752
+    env-step-used-before:                       89128960
+    ppo-update-before:                          54400
+    env-step-generate-before:                   43614208
+    env-step-all-after:                         89161728
+    env-step-replay-after:                      45547520
+    env-step-used-after:                        89161728
+    ppo-update-after:                           54420
+    env-step-generate-after:                    43614208
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.71578
+    avg_avg_return:                             0.5251974
+    avg_reward_per_step:                        0.015045166
+    proxy_corner:
+      avg_avg_return:                           0.5251974
+      avg_reward_per_step:                      0.015045166
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.71578
+    avg_avg_return:                             0.5251974
+    avg_reward_per_step:                        0.015045166
+    proxy_corner:
+      avg_avg_return:                           0.5251974
+      avg_reward_per_step:                      0.015045166
+  perf:
+    env_steps_per_second:                       254309.4114182952
+    ppo_updates_per_second:                     41.83464610510849
+    eval-batch-orig-duration:                   0.5507830008864403
+    eval-batch-shift-duration:                  0.49569494277238846
+    eval-batch-tree-duration:                   0.4887129766866565
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             26.906494
+      mouse-cheese_dist_finite_wavg:            16.201458
+      mouse-cheese_dist_solvable_avg:           26.906494
+      mouse-cheese_dist_solvable_wavg:          16.20146
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.420168
+    scoring:
+      avg_scores:                               0.4636249
+    visit_patterns:
+      avg_first_visit_time:                     1187.2302
+      avg_last_visit_time:                      1380.2781
+      num_replay_batches:                       1390
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0015693273
+    avg_actor_approxkl3:                        0.0010212075
+    avg_actor_clipfrac:                         0.045880128
+    avg_actor_loss:                             -0.00079257414
+    avg_advantage:                              -8.764339e-05
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.007075477
+    avg_entropy:                                0.2094829
+    avg_grad_norm_pre_clip:                     1.7272757
+    avg_loss:                                   0.0025356815
+    max:
+      max_actor_approxkl1:                      0.0034990117
+      max_actor_approxkl3:                      0.0017460231
+      max_actor_clipfrac:                       0.0748291
+      max_actor_loss:                           0.001081571
+      max_advantage:                            0.2529866
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.009187846
+      max_entropy:                              0.24276218
+      max_grad_norm_pre_clip:                   3.5831642
+      max_loss:                                 0.0050236913
+    std:
+      std_actor_approxkl1:                      0.0008365273
+      std_actor_approxkl3:                      0.00048009751
+      std_actor_clipfrac:                       0.01933712
+      std_actor_loss:                           0.0009124591
+      std_advantage:                            0.118999824
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011077772
+      std_entropy:                              0.010503313
+      std_grad_norm_pre_clip:                   0.5488706
+      std_loss:                                 0.001107202
+  eval-batch-orig:
+    avg_avg_episode_length:                     20.460505
+    avg_avg_return:                             0.9138824
+    avg_benchmark_regret:                       0.07094416
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09030151
+    proxy_corner:
+      avg_avg_return:                           0.9138824
+      avg_reward_per_step:                      0.09030151
+  eval-batch-shift:
+    avg_avg_episode_length:                     69.67166
+    avg_avg_return:                             0.2306743
+    avg_benchmark_regret:                       0.75621617
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.032707214
+    proxy_corner:
+      avg_avg_return:                           0.64572275
+      avg_reward_per_step:                      0.005317688
+  eval-batch-tree:
+    avg_avg_episode_length:                     21.566732
+    avg_avg_return:                             0.90837646
+    avg_benchmark_regret:                       0.07569919
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.107040405
+    proxy_corner:
+      avg_avg_return:                           0.90837646
+      avg_reward_per_step:                      0.107040405
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        90177536
+    env-step-replay-before:                     46202880
+    env-step-used-before:                       90177536
+    ppo-update-before:                          55040
+    env-step-generate-before:                   43974656
+    env-step-all-after:                         90210304
+    env-step-replay-after:                      46202880
+    env-step-used-after:                        90210304
+    ppo-update-after:                           55060
+    env-step-generate-after:                    44007424
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     23.0694
+    avg_avg_return:                             0.821
+    avg_reward_per_step:                        0.09716797
+    proxy_corner:
+      avg_avg_return:                           0.821
+      avg_reward_per_step:                      0.09716797
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     23.0694
+    avg_avg_return:                             0.821
+    avg_reward_per_step:                        0.09716797
+    proxy_corner:
+      avg_avg_return:                           0.821
+      avg_reward_per_step:                      0.09716797
+  perf:
+    env_steps_per_second:                       351569.7847434436
+    ppo_updates_per_second:                     42.34079221374919
+    eval-batch-orig-duration:                   0.5261538233608007
+    eval-batch-shift-duration:                  0.44925528205931187
+    eval-batch-tree-duration:                   0.482398665510118
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             26.962158
+      mouse-cheese_dist_finite_wavg:            15.291481
+      mouse-cheese_dist_solvable_avg:           26.962158
+      mouse-cheese_dist_solvable_wavg:          15.291483
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.4444598
+    visit_patterns:
+      avg_first_visit_time:                     1199.6377
+      avg_last_visit_time:                      1400.1858
+      num_replay_batches:                       1410
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010050685
+    avg_actor_approxkl3:                        0.00087333284
+    avg_actor_clipfrac:                         0.0286438
+    avg_actor_loss:                             -0.0004950031
+    avg_advantage:                              0.018672295
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0018009127
+    avg_entropy:                                0.09224824
+    avg_grad_norm_pre_clip:                     1.5245788
+    avg_loss:                                   0.000313205
+    max:
+      max_actor_approxkl1:                      0.0028908113
+      max_actor_approxkl3:                      0.001471464
+      max_actor_clipfrac:                       0.04663086
+      max_actor_loss:                           0.0013006367
+      max_advantage:                            0.1976068
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0028841286
+      max_entropy:                              0.10192853
+      max_grad_norm_pre_clip:                   2.7205112
+      max_loss:                                 0.0016971937
+    std:
+      std_actor_approxkl1:                      0.00079045305
+      std_actor_approxkl3:                      0.00046104458
+      std_actor_clipfrac:                       0.012804546
+      std_actor_loss:                           0.0009818168
+      std_advantage:                            0.059870597
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.000625775
+      std_entropy:                              0.0067706383
+      std_grad_norm_pre_clip:                   0.52482903
+      std_loss:                                 0.00090163667
+  eval-batch-orig:
+    avg_avg_episode_length:                     20.29536
+    avg_avg_return:                             0.9216933
+    avg_benchmark_regret:                       0.06313317
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08860016
+    proxy_corner:
+      avg_avg_return:                           0.9216933
+      avg_reward_per_step:                      0.08860016
+  eval-batch-shift:
+    avg_avg_episode_length:                     67.86853
+    avg_avg_return:                             0.2589962
+    avg_benchmark_regret:                       0.72789425
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.0342865
+    proxy_corner:
+      avg_avg_return:                           0.61645406
+      avg_reward_per_step:                      0.0052261353
+  eval-batch-tree:
+    avg_avg_episode_length:                     21.336067
+    avg_avg_return:                             0.9152067
+    avg_benchmark_regret:                       0.068868965
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.105651855
+    proxy_corner:
+      avg_avg_return:                           0.9152067
+      avg_reward_per_step:                      0.105651855
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        91226112
+    env-step-replay-before:                     46759936
+    env-step-used-before:                       91226112
+    ppo-update-before:                          55680
+    env-step-generate-before:                   44466176
+    env-step-all-after:                         91258880
+    env-step-replay-after:                      46759936
+    env-step-used-after:                        91258880
+    ppo-update-after:                           55700
+    env-step-generate-after:                    44498944
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     24.396984
+    avg_avg_return:                             0.8113275
+    avg_reward_per_step:                        0.07733154
+    proxy_corner:
+      avg_avg_return:                           0.8113275
+      avg_reward_per_step:                      0.07733154
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     24.396984
+    avg_avg_return:                             0.8113275
+    avg_reward_per_step:                        0.07733154
+    proxy_corner:
+      avg_avg_return:                           0.8113275
+      avg_reward_per_step:                      0.07733154
+  perf:
+    env_steps_per_second:                       319749.18719534733
+    ppo_updates_per_second:                     41.86004240970164
+    eval-batch-orig-duration:                   0.5397657360881567
+    eval-batch-shift-duration:                  0.489006738178432
+    eval-batch-tree-duration:                   0.48794976714998484
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             26.96582
+      mouse-cheese_dist_finite_wavg:            13.498883
+      mouse-cheese_dist_solvable_avg:           26.96582
+      mouse-cheese_dist_solvable_wavg:          13.498886
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.45338163
+    visit_patterns:
+      avg_first_visit_time:                     1215.9805
+      avg_last_visit_time:                      1417.2415
+      num_replay_batches:                       1427
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0007300281
+    avg_actor_approxkl3:                        0.0006368027
+    avg_actor_clipfrac:                         0.024700928
+    avg_actor_loss:                             -0.0005505737
+    avg_advantage:                              0.033933718
+    avg_critic_clipfrac:                        2.4414063e-05
+    avg_critic_loss:                            0.0023258228
+    avg_entropy:                                0.09598607
+    avg_grad_norm_pre_clip:                     1.6097956
+    avg_loss:                                   0.0005163517
+    max:
+      max_actor_approxkl1:                      0.0019665118
+      max_actor_approxkl3:                      0.0014592735
+      max_actor_clipfrac:                       0.0423584
+      max_actor_loss:                           0.0014178455
+      max_advantage:                            0.20741946
+      max_critic_clipfrac:                      0.00048828125
+      max_critic_loss:                          0.0038644
+      max_entropy:                              0.11327914
+      max_grad_norm_pre_clip:                   2.838965
+      max_loss:                                 0.0028854944
+    std:
+      std_actor_approxkl1:                      0.0004762407
+      std_actor_approxkl3:                      0.0003121037
+      std_actor_clipfrac:                       0.009030008
+      std_actor_loss:                           0.000985531
+      std_advantage:                            0.067236125
+      std_critic_clipfrac:                      0.000106418425
+      std_critic_loss:                          0.00073815783
+      std_entropy:                              0.009217961
+      std_grad_norm_pre_clip:                   0.48545018
+      std_loss:                                 0.0010093661
+  eval-batch-orig:
+    avg_avg_episode_length:                     20.817026
+    avg_avg_return:                             0.91416687
+    avg_benchmark_regret:                       0.07065967
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.08909607
+    proxy_corner:
+      avg_avg_return:                           0.91416687
+      avg_reward_per_step:                      0.08909607
+  eval-batch-shift:
+    avg_avg_episode_length:                     67.97554
+    avg_avg_return:                             0.26349276
+    avg_benchmark_regret:                       0.72339773
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.033843994
+    proxy_corner:
+      avg_avg_return:                           0.61786973
+      avg_reward_per_step:                      0.005203247
+  eval-batch-tree:
+    avg_avg_episode_length:                     21.2278
+    avg_avg_return:                             0.9117826
+    avg_benchmark_regret:                       0.07229294
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10624695
+    proxy_corner:
+      avg_avg_return:                           0.9117826
+      avg_reward_per_step:                      0.10624695
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        92274688
+    env-step-replay-before:                     47251456
+    env-step-used-before:                       92274688
+    ppo-update-before:                          56320
+    env-step-generate-before:                   45023232
+    env-step-all-after:                         92307456
+    env-step-replay-after:                      47251456
+    env-step-used-after:                        92307456
+    ppo-update-after:                           56340
+    env-step-generate-after:                    45056000
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     21.77575
+    avg_avg_return:                             0.83114386
+    avg_reward_per_step:                        0.10479736
+    proxy_corner:
+      avg_avg_return:                           0.83114386
+      avg_reward_per_step:                      0.10479736
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     21.77575
+    avg_avg_return:                             0.83114386
+    avg_reward_per_step:                        0.10479736
+    proxy_corner:
+      avg_avg_return:                           0.83114386
+      avg_reward_per_step:                      0.10479736
+  perf:
+    env_steps_per_second:                       295097.93441373506
+    ppo_updates_per_second:                     42.17010724865814
+    eval-batch-orig-duration:                   0.5455569941550493
+    eval-batch-shift-duration:                  0.4924666518345475
+    eval-batch-tree-duration:                   0.49354670103639364
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             27.08081
+      mouse-cheese_dist_finite_wavg:            17.108316
+      mouse-cheese_dist_solvable_avg:           27.08081
+      mouse-cheese_dist_solvable_wavg:          17.108318
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.4732964
+    visit_patterns:
+      avg_first_visit_time:                     1234.4265
+      avg_last_visit_time:                      1432.2578
+      num_replay_batches:                       1442
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001385628
+    avg_actor_approxkl3:                        0.0010856282
+    avg_actor_clipfrac:                         0.026574707
+    avg_actor_loss:                             0.0005096404
+    avg_advantage:                              0.01276245
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.001582606
+    avg_entropy:                                0.071900785
+    avg_grad_norm_pre_clip:                     1.5422468
+    avg_loss:                                   0.0012290427
+    max:
+      max_actor_approxkl1:                      0.004765875
+      max_actor_approxkl3:                      0.003554037
+      max_actor_clipfrac:                       0.048217773
+      max_actor_loss:                           0.005946409
+      max_advantage:                            0.17868388
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0033596475
+      max_entropy:                              0.08924197
+      max_grad_norm_pre_clip:                   3.660056
+      max_loss:                                 0.0063729514
+    std:
+      std_actor_approxkl1:                      0.0010147046
+      std_actor_approxkl3:                      0.0007513898
+      std_actor_clipfrac:                       0.010394349
+      std_actor_loss:                           0.0016045059
+      std_advantage:                            0.05561466
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0007026743
+      std_entropy:                              0.010068599
+      std_grad_norm_pre_clip:                   0.6586856
+      std_loss:                                 0.0014914923
+  eval-batch-orig:
+    avg_avg_episode_length:                     20.173458
+    avg_avg_return:                             0.91828096
+    avg_benchmark_regret:                       0.06654559
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09037018
+    proxy_corner:
+      avg_avg_return:                           0.91828096
+      avg_reward_per_step:                      0.09037018
+  eval-batch-shift:
+    avg_avg_episode_length:                     68.696915
+    avg_avg_return:                             0.24200274
+    avg_benchmark_regret:                       0.74488777
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.034172058
+    proxy_corner:
+      avg_avg_return:                           0.63228846
+      avg_reward_per_step:                      0.005279541
+  eval-batch-tree:
+    avg_avg_episode_length:                     20.490662
+    avg_avg_return:                             0.91985667
+    avg_benchmark_regret:                       0.064218976
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.1073761
+    proxy_corner:
+      avg_avg_return:                           0.91985667
+      avg_reward_per_step:                      0.1073761
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        93323264
+    env-step-replay-before:                     47742976
+    env-step-used-before:                       93323264
+    ppo-update-before:                          56960
+    env-step-generate-before:                   45580288
+    env-step-all-after:                         93356032
+    env-step-replay-after:                      47742976
+    env-step-used-after:                        93356032
+    ppo-update-after:                           56980
+    env-step-generate-after:                    45613056
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     22.650286
+    avg_avg_return:                             0.8265518
+    avg_reward_per_step:                        0.10632324
+    proxy_corner:
+      avg_avg_return:                           0.8265518
+      avg_reward_per_step:                      0.10632324
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     22.650286
+    avg_avg_return:                             0.8265518
+    avg_reward_per_step:                        0.10632324
+    proxy_corner:
+      avg_avg_return:                           0.8265518
+      avg_reward_per_step:                      0.10632324
+  perf:
+    env_steps_per_second:                       349289.34346927516
+    ppo_updates_per_second:                     42.235639435949096
+    eval-batch-orig-duration:                   0.591359181329608
+    eval-batch-shift-duration:                  0.5616792719811201
+    eval-batch-tree-duration:                   0.5262122992426157
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             27.216797
+      mouse-cheese_dist_finite_wavg:            13.513514
+      mouse-cheese_dist_solvable_avg:           27.216797
+      mouse-cheese_dist_solvable_wavg:          13.513514
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.4772195
+    visit_patterns:
+      avg_first_visit_time:                     1248.9878
+      avg_last_visit_time:                      1447.4011
+      num_replay_batches:                       1457
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00079399574
+    avg_actor_approxkl3:                        0.0010162079
+    avg_actor_clipfrac:                         0.028570557
+    avg_actor_loss:                             -0.0002680137
+    avg_advantage:                              0.0063523916
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.001346098
+    avg_entropy:                                0.08211293
+    avg_grad_norm_pre_clip:                     1.5385334
+    avg_loss:                                   0.00032292234
+    max:
+      max_actor_approxkl1:                      0.0033842982
+      max_actor_approxkl3:                      0.0029599876
+      max_actor_clipfrac:                       0.057128906
+      max_actor_loss:                           0.0013846494
+      max_advantage:                            0.22725654
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0022293595
+      max_entropy:                              0.09823814
+      max_grad_norm_pre_clip:                   2.4213758
+      max_loss:                                 0.0017905108
+    std:
+      std_actor_approxkl1:                      0.0008548695
+      std_actor_approxkl3:                      0.0006827341
+      std_actor_clipfrac:                       0.013149669
+      std_actor_loss:                           0.00082227186
+      std_advantage:                            0.05179452
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00058105064
+      std_entropy:                              0.009544922
+      std_grad_norm_pre_clip:                   0.5555216
+      std_loss:                                 0.0007517057
+  eval-batch-orig:
+    avg_avg_episode_length:                     19.898159
+    avg_avg_return:                             0.9206264
+    avg_benchmark_regret:                       0.06420009
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09046936
+    proxy_corner:
+      avg_avg_return:                           0.9206264
+      avg_reward_per_step:                      0.09046936
+  eval-batch-shift:
+    avg_avg_episode_length:                     69.33407
+    avg_avg_return:                             0.23754267
+    avg_benchmark_regret:                       0.7493478
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03249359
+    proxy_corner:
+      avg_avg_return:                           0.628147
+      avg_reward_per_step:                      0.0052490234
+  eval-batch-tree:
+    avg_avg_episode_length:                     20.14698
+    avg_avg_return:                             0.9260018
+    avg_benchmark_regret:                       0.05807385
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10795593
+    proxy_corner:
+      avg_avg_return:                           0.9260018
+      avg_reward_per_step:                      0.10795593
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        94371840
+    env-step-replay-before:                     48267264
+    env-step-used-before:                       94371840
+    ppo-update-before:                          57600
+    env-step-generate-before:                   46104576
+    env-step-all-after:                         94404608
+    env-step-replay-after:                      48300032
+    env-step-used-after:                        94404608
+    ppo-update-after:                           57620
+    env-step-generate-after:                    46104576
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     59.472256
+    avg_avg_return:                             0.50961715
+    avg_reward_per_step:                        0.015014648
+    proxy_corner:
+      avg_avg_return:                           0.50961715
+      avg_reward_per_step:                      0.015014648
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     59.472256
+    avg_avg_return:                             0.50961715
+    avg_reward_per_step:                        0.015014648
+    proxy_corner:
+      avg_avg_return:                           0.50961715
+      avg_reward_per_step:                      0.015014648
+  perf:
+    env_steps_per_second:                       242243.06171634418
+    ppo_updates_per_second:                     42.265593009954955
+    eval-batch-orig-duration:                   0.6586704961955547
+    eval-batch-shift-duration:                  0.6163722015917301
+    eval-batch-tree-duration:                   0.6203538365662098
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             27.327637
+      mouse-cheese_dist_finite_wavg:            12.6288395
+      mouse-cheese_dist_solvable_avg:           27.327637
+      mouse-cheese_dist_solvable_wavg:          12.628842
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016798
+    scoring:
+      avg_scores:                               0.47079048
+    visit_patterns:
+      avg_first_visit_time:                     1261.0105
+      avg_last_visit_time:                      1464.2554
+      num_replay_batches:                       1474
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013081243
+    avg_actor_approxkl3:                        0.0009679854
+    avg_actor_clipfrac:                         0.03981323
+    avg_actor_loss:                             -0.00086881034
+    avg_advantage:                              -0.017718367
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.007357104
+    avg_entropy:                                0.20180912
+    avg_grad_norm_pre_clip:                     1.7083682
+    avg_loss:                                   0.0026079328
+    max:
+      max_actor_approxkl1:                      0.0044103265
+      max_actor_approxkl3:                      0.0037417402
+      max_actor_clipfrac:                       0.10424805
+      max_actor_loss:                           0.0010884553
+      max_advantage:                            0.20747417
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.009165067
+      max_entropy:                              0.22084469
+      max_grad_norm_pre_clip:                   4.13756
+      max_loss:                                 0.004256931
+    std:
+      std_actor_approxkl1:                      0.0010481478
+      std_actor_approxkl3:                      0.00078485475
+      std_actor_clipfrac:                       0.023325054
+      std_actor_loss:                           0.0010219887
+      std_advantage:                            0.12088427
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010257042
+      std_entropy:                              0.011824044
+      std_grad_norm_pre_clip:                   0.8529614
+      std_loss:                                 0.0009843579
+  eval-batch-orig:
+    avg_avg_episode_length:                     19.279959
+    avg_avg_return:                             0.9279765
+    avg_benchmark_regret:                       0.056850053
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09135437
+    proxy_corner:
+      avg_avg_return:                           0.9279765
+      avg_reward_per_step:                      0.09135437
+  eval-batch-shift:
+    avg_avg_episode_length:                     69.63819
+    avg_avg_return:                             0.23007683
+    avg_benchmark_regret:                       0.75681365
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.032585144
+    proxy_corner:
+      avg_avg_return:                           0.6504954
+      avg_reward_per_step:                      0.0053863525
+  eval-batch-tree:
+    avg_avg_episode_length:                     20.128574
+    avg_avg_return:                             0.9217752
+    avg_benchmark_regret:                       0.062300418
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10808563
+    proxy_corner:
+      avg_avg_return:                           0.9217752
+      avg_reward_per_step:                      0.10808563
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        95420416
+    env-step-replay-before:                     48889856
+    env-step-used-before:                       95420416
+    ppo-update-before:                          58240
+    env-step-generate-before:                   46530560
+    env-step-all-after:                         95453184
+    env-step-replay-after:                      48889856
+    env-step-used-after:                        95453184
+    ppo-update-after:                           58260
+    env-step-generate-after:                    46563328
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     21.478357
+    avg_avg_return:                             0.8410187
+    avg_reward_per_step:                        0.08987427
+    proxy_corner:
+      avg_avg_return:                           0.8410187
+      avg_reward_per_step:                      0.08987427
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     21.478357
+    avg_avg_return:                             0.8410187
+    avg_reward_per_step:                        0.08987427
+    proxy_corner:
+      avg_avg_return:                           0.8410187
+      avg_reward_per_step:                      0.08987427
+  perf:
+    env_steps_per_second:                       242826.26638543728
+    ppo_updates_per_second:                     42.21516464042523
+    eval-batch-orig-duration:                   0.5193318584933877
+    eval-batch-shift-duration:                  0.4713388355448842
+    eval-batch-tree-duration:                   0.47064746264368296
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             27.372803
+      mouse-cheese_dist_finite_wavg:            16.223848
+      mouse-cheese_dist_solvable_avg:           27.372803
+      mouse-cheese_dist_solvable_wavg:          16.223852
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.47002316
+    visit_patterns:
+      avg_first_visit_time:                     1274.2734
+      avg_last_visit_time:                      1482.1826
+      num_replay_batches:                       1492
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009184562
+    avg_actor_approxkl3:                        0.00083768246
+    avg_actor_clipfrac:                         0.02324829
+    avg_actor_loss:                             8.438735e-05
+    avg_advantage:                              0.039955057
+    avg_critic_clipfrac:                        0.013989258
+    avg_critic_loss:                            0.0014246394
+    avg_entropy:                                0.074938305
+    avg_grad_norm_pre_clip:                     1.5193905
+    avg_loss:                                   0.00072176877
+    max:
+      max_actor_approxkl1:                      0.0035982537
+      max_actor_approxkl3:                      0.002072049
+      max_actor_clipfrac:                       0.037231445
+      max_actor_loss:                           0.002063319
+      max_advantage:                            0.2043764
+      max_critic_clipfrac:                      0.09301758
+      max_critic_loss:                          0.0023456218
+      max_entropy:                              0.09199594
+      max_grad_norm_pre_clip:                   2.8983064
+      max_loss:                                 0.0026452749
+    std:
+      std_actor_approxkl1:                      0.00083922513
+      std_actor_approxkl3:                      0.0005063619
+      std_actor_clipfrac:                       0.009768789
+      std_actor_loss:                           0.0008184205
+      std_advantage:                            0.051515486
+      std_critic_clipfrac:                      0.027249921
+      std_critic_loss:                          0.0004278833
+      std_entropy:                              0.009753869
+      std_grad_norm_pre_clip:                   0.5710871
+      std_loss:                                 0.000805259
+  eval-batch-orig:
+    avg_avg_episode_length:                     19.567417
+    avg_avg_return:                             0.924614
+    avg_benchmark_regret:                       0.060212508
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.091430664
+    proxy_corner:
+      avg_avg_return:                           0.924614
+      avg_reward_per_step:                      0.091430664
+  eval-batch-shift:
+    avg_avg_episode_length:                     69.57303
+    avg_avg_return:                             0.23499113
+    avg_benchmark_regret:                       0.75189936
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.032958984
+    proxy_corner:
+      avg_avg_return:                           0.6538104
+      avg_reward_per_step:                      0.005432129
+  eval-batch-tree:
+    avg_avg_episode_length:                     20.633827
+    avg_avg_return:                             0.9187949
+    avg_benchmark_regret:                       0.06528075
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10758972
+    proxy_corner:
+      avg_avg_return:                           0.9187949
+      avg_reward_per_step:                      0.10758972
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        96468992
+    env-step-replay-before:                     49381376
+    env-step-used-before:                       96468992
+    ppo-update-before:                          58880
+    env-step-generate-before:                   47087616
+    env-step-all-after:                         96501760
+    env-step-replay-after:                      49414144
+    env-step-used-after:                        96501760
+    ppo-update-after:                           58900
+    env-step-generate-after:                    47087616
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     63.88535
+    avg_avg_return:                             0.48240462
+    avg_reward_per_step:                        0.013214111
+    proxy_corner:
+      avg_avg_return:                           0.48240462
+      avg_reward_per_step:                      0.013214111
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     63.88535
+    avg_avg_return:                             0.48240462
+    avg_reward_per_step:                        0.013214111
+    proxy_corner:
+      avg_avg_return:                           0.48240462
+      avg_reward_per_step:                      0.013214111
+  perf:
+    env_steps_per_second:                       292365.6485764842
+    ppo_updates_per_second:                     42.23670903312586
+    eval-batch-orig-duration:                   0.5392100838944316
+    eval-batch-shift-duration:                  0.47556031681597233
+    eval-batch-tree-duration:                   0.48889342974871397
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             27.409668
+      mouse-cheese_dist_finite_wavg:            14.445841
+      mouse-cheese_dist_solvable_avg:           27.409668
+      mouse-cheese_dist_solvable_wavg:          14.445841
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.420168
+    scoring:
+      avg_scores:                               0.46388862
+    visit_patterns:
+      avg_first_visit_time:                     1289.7393
+      avg_last_visit_time:                      1498.3018
+      num_replay_batches:                       1508
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013638416
+    avg_actor_approxkl3:                        0.0013950609
+    avg_actor_clipfrac:                         0.057000734
+    avg_actor_loss:                             -0.0001712855
+    avg_advantage:                              -0.015065249
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0073624887
+    avg_entropy:                                0.20145476
+    avg_grad_norm_pre_clip:                     2.302596
+    avg_loss:                                   0.0033085037
+    max:
+      max_actor_approxkl1:                      0.0030058539
+      max_actor_approxkl3:                      0.0021606346
+      max_actor_clipfrac:                       0.07446289
+      max_actor_loss:                           0.0015351698
+      max_advantage:                            0.18619794
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.008937915
+      max_entropy:                              0.22246301
+      max_grad_norm_pre_clip:                   3.723803
+      max_loss:                                 0.005472908
+    std:
+      std_actor_approxkl1:                      0.0007482138
+      std_actor_approxkl3:                      0.00053745613
+      std_actor_clipfrac:                       0.017258536
+      std_actor_loss:                           0.001002069
+      std_advantage:                            0.12134167
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0008909067
+      std_entropy:                              0.006663294
+      std_grad_norm_pre_clip:                   0.8382126
+      std_loss:                                 0.0012534054
+  eval-batch-orig:
+    avg_avg_episode_length:                     20.089117
+    avg_avg_return:                             0.91388655
+    avg_benchmark_regret:                       0.07093993
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09008026
+    proxy_corner:
+      avg_avg_return:                           0.91388655
+      avg_reward_per_step:                      0.09008026
+  eval-batch-shift:
+    avg_avg_episode_length:                     69.59952
+    avg_avg_return:                             0.23625767
+    avg_benchmark_regret:                       0.75063276
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.033058167
+    proxy_corner:
+      avg_avg_return:                           0.6081337
+      avg_reward_per_step:                      0.0050811768
+  eval-batch-tree:
+    avg_avg_episode_length:                     20.601803
+    avg_avg_return:                             0.9189538
+    avg_benchmark_regret:                       0.065121815
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10697937
+    proxy_corner:
+      avg_avg_return:                           0.9189538
+      avg_reward_per_step:                      0.10697937
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        97517568
+    env-step-replay-before:                     49840128
+    env-step-used-before:                       97517568
+    ppo-update-before:                          59520
+    env-step-generate-before:                   47677440
+    env-step-all-after:                         97550336
+    env-step-replay-after:                      49872896
+    env-step-used-after:                        97550336
+    ppo-update-after:                           59540
+    env-step-generate-after:                    47677440
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     59.078323
+    avg_avg_return:                             0.5312854
+    avg_reward_per_step:                        0.01473999
+    proxy_corner:
+      avg_avg_return:                           0.5312854
+      avg_reward_per_step:                      0.01473999
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     59.078323
+    avg_avg_return:                             0.5312854
+    avg_reward_per_step:                        0.01473999
+    proxy_corner:
+      avg_avg_return:                           0.5312854
+      avg_reward_per_step:                      0.01473999
+  perf:
+    env_steps_per_second:                       350969.0673658469
+    ppo_updates_per_second:                     44.48555489277961
+    eval-batch-orig-duration:                   0.4801981719210744
+    eval-batch-shift-duration:                  0.4269122816622257
+    eval-batch-tree-duration:                   0.453057236969471
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             27.59082
+      mouse-cheese_dist_finite_wavg:            15.346985
+      mouse-cheese_dist_solvable_avg:           27.59082
+      mouse-cheese_dist_solvable_wavg:          15.346987
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.45740008
+    visit_patterns:
+      avg_first_visit_time:                     1305.7717
+      avg_last_visit_time:                      1512.2988
+      num_replay_batches:                       1522
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012911566
+    avg_actor_approxkl3:                        0.0012630493
+    avg_actor_clipfrac:                         0.049780276
+    avg_actor_loss:                             0.00028568544
+    avg_advantage:                              -0.014483102
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.006059966
+    avg_entropy:                                0.18561436
+    avg_grad_norm_pre_clip:                     2.1036012
+    avg_loss:                                   0.003130054
+    max:
+      max_actor_approxkl1:                      0.0039115334
+      max_actor_approxkl3:                      0.003609519
+      max_actor_clipfrac:                       0.09643555
+      max_actor_loss:                           0.0030267164
+      max_advantage:                            0.17674458
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.008334925
+      max_entropy:                              0.20336181
+      max_grad_norm_pre_clip:                   3.402824
+      max_loss:                                 0.005691605
+    std:
+      std_actor_approxkl1:                      0.000953059
+      std_actor_approxkl3:                      0.000853334
+      std_actor_clipfrac:                       0.023252413
+      std_actor_loss:                           0.0011568206
+      std_advantage:                            0.110064544
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0012681352
+      std_entropy:                              0.00757312
+      std_grad_norm_pre_clip:                   0.67548317
+      std_loss:                                 0.0011761687
+  eval-batch-orig:
+    avg_avg_episode_length:                     19.109621
+    avg_avg_return:                             0.92745364
+    avg_benchmark_regret:                       0.057372924
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09093475
+    proxy_corner:
+      avg_avg_return:                           0.92745364
+      avg_reward_per_step:                      0.09093475
+  eval-batch-shift:
+    avg_avg_episode_length:                     68.64039
+    avg_avg_return:                             0.25551593
+    avg_benchmark_regret:                       0.7313745
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.033859253
+    proxy_corner:
+      avg_avg_return:                           0.61890143
+      avg_reward_per_step:                      0.0052108765
+  eval-batch-tree:
+    avg_avg_episode_length:                     20.065292
+    avg_avg_return:                             0.92493397
+    avg_benchmark_regret:                       0.0591416
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10738373
+    proxy_corner:
+      avg_avg_return:                           0.92493397
+      avg_reward_per_step:                      0.10738373
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        98566144
+    env-step-replay-before:                     50397184
+    env-step-used-before:                       98566144
+    ppo-update-before:                          60160
+    env-step-generate-before:                   48168960
+    env-step-all-after:                         98598912
+    env-step-replay-after:                      50429952
+    env-step-used-after:                        98598912
+    ppo-update-after:                           60180
+    env-step-generate-after:                    48168960
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     60.280125
+    avg_avg_return:                             0.54263043
+    avg_reward_per_step:                        0.01449585
+    proxy_corner:
+      avg_avg_return:                           0.54263043
+      avg_reward_per_step:                      0.01449585
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     60.280125
+    avg_avg_return:                             0.54263043
+    avg_reward_per_step:                        0.01449585
+    proxy_corner:
+      avg_avg_return:                           0.54263043
+      avg_reward_per_step:                      0.01449585
+  perf:
+    env_steps_per_second:                       242510.7276559441
+    ppo_updates_per_second:                     42.236898270077546
+    eval-batch-orig-duration:                   0.6404300713911653
+    eval-batch-shift-duration:                  0.6414097035303712
+    eval-batch-tree-duration:                   0.6115388963371515
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             27.668213
+      mouse-cheese_dist_finite_wavg:            17.166037
+      mouse-cheese_dist_solvable_avg:           27.668213
+      mouse-cheese_dist_solvable_wavg:          17.166039
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000008
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.44542718
+    visit_patterns:
+      avg_first_visit_time:                     1318.9548
+      avg_last_visit_time:                      1529.2273
+      num_replay_batches:                       1539
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013104138
+    avg_actor_approxkl3:                        0.0011875986
+    avg_actor_clipfrac:                         0.051098634
+    avg_actor_loss:                             -0.00012925453
+    avg_advantage:                              0.00066706515
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00495216
+    avg_entropy:                                0.19751613
+    avg_grad_norm_pre_clip:                     2.1002834
+    avg_loss:                                   0.0021493095
+    max:
+      max_actor_approxkl1:                      0.003896421
+      max_actor_approxkl3:                      0.00265495
+      max_actor_clipfrac:                       0.08618164
+      max_actor_loss:                           0.0017478317
+      max_advantage:                            0.2022621
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0075197215
+      max_entropy:                              0.22047137
+      max_grad_norm_pre_clip:                   3.6789627
+      max_loss:                                 0.004706891
+    std:
+      std_actor_approxkl1:                      0.0010876659
+      std_actor_approxkl3:                      0.0006738267
+      std_actor_clipfrac:                       0.022286624
+      std_actor_loss:                           0.0010789747
+      std_advantage:                            0.09936433
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.001219734
+      std_entropy:                              0.010679764
+      std_grad_norm_pre_clip:                   0.78235596
+      std_loss:                                 0.0011984124
+  eval-batch-orig:
+    avg_avg_episode_length:                     18.604269
+    avg_avg_return:                             0.93599963
+    avg_benchmark_regret:                       0.048826933
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.091667175
+    proxy_corner:
+      avg_avg_return:                           0.93599963
+      avg_reward_per_step:                      0.091667175
+  eval-batch-shift:
+    avg_avg_episode_length:                     69.405945
+    avg_avg_return:                             0.23509854
+    avg_benchmark_regret:                       0.7517919
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03299713
+    proxy_corner:
+      avg_avg_return:                           0.6370872
+      avg_reward_per_step:                      0.005340576
+  eval-batch-tree:
+    avg_avg_episode_length:                     20.041374
+    avg_avg_return:                             0.927778
+    avg_benchmark_regret:                       0.056297563
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10748291
+    proxy_corner:
+      avg_avg_return:                           0.927778
+      avg_reward_per_step:                      0.10748291
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        99614720
+    env-step-replay-before:                     51052544
+    env-step-used-before:                       99614720
+    ppo-update-before:                          60800
+    env-step-generate-before:                   48562176
+    env-step-all-after:                         99647488
+    env-step-replay-after:                      51085312
+    env-step-used-after:                        99647488
+    ppo-update-after:                           60820
+    env-step-generate-after:                    48562176
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.916336
+    avg_avg_return:                             0.53970283
+    avg_reward_per_step:                        0.014831543
+    proxy_corner:
+      avg_avg_return:                           0.53970283
+      avg_reward_per_step:                      0.014831543
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.916336
+    avg_avg_return:                             0.53970283
+    avg_reward_per_step:                        0.014831543
+    proxy_corner:
+      avg_avg_return:                           0.53970283
+      avg_reward_per_step:                      0.014831543
+  perf:
+    env_steps_per_second:                       319724.39954469923
+    ppo_updates_per_second:                     40.455749402667486
+    eval-batch-orig-duration:                   0.526187539100647
+    eval-batch-shift-duration:                  0.4872723761945963
+    eval-batch-tree-duration:                   0.4601207869127393
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             27.805664
+      mouse-cheese_dist_finite_wavg:            17.195528
+      mouse-cheese_dist_solvable_avg:           27.805664
+      mouse-cheese_dist_solvable_wavg:          17.195528
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.4370802
+    visit_patterns:
+      avg_first_visit_time:                     1330.2898
+      avg_last_visit_time:                      1549.259
+      num_replay_batches:                       1559
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012813247
+    avg_actor_approxkl3:                        0.0011468191
+    avg_actor_clipfrac:                         0.045300294
+    avg_actor_loss:                             -0.00030744262
+    avg_advantage:                              -0.008567627
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.005429042
+    avg_entropy:                                0.18569511
+    avg_grad_norm_pre_clip:                     2.021679
+    avg_loss:                                   0.0022213836
+    max:
+      max_actor_approxkl1:                      0.0034738427
+      max_actor_approxkl3:                      0.0022704387
+      max_actor_clipfrac:                       0.08129883
+      max_actor_loss:                           0.0012348294
+      max_advantage:                            0.16134632
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.006836508
+      max_entropy:                              0.20349146
+      max_grad_norm_pre_clip:                   3.723714
+      max_loss:                                 0.0042047673
+    std:
+      std_actor_approxkl1:                      0.000792697
+      std_actor_approxkl3:                      0.0005841709
+      std_actor_clipfrac:                       0.018453013
+      std_actor_loss:                           0.0008729935
+      std_advantage:                            0.10415831
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010636692
+      std_entropy:                              0.010215173
+      std_grad_norm_pre_clip:                   0.7497779
+      std_loss:                                 0.0010765501
+  eval-batch-orig:
+    avg_avg_episode_length:                     18.922403
+    avg_avg_return:                             0.93318546
+    avg_benchmark_regret:                       0.051641088
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09101868
+    proxy_corner:
+      avg_avg_return:                           0.93318546
+      avg_reward_per_step:                      0.09101868
+  eval-batch-shift:
+    avg_avg_episode_length:                     69.26822
+    avg_avg_return:                             0.23404123
+    avg_benchmark_regret:                       0.7528492
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.033584595
+    proxy_corner:
+      avg_avg_return:                           0.6161059
+      avg_reward_per_step:                      0.0051345825
+  eval-batch-tree:
+    avg_avg_episode_length:                     20.02678
+    avg_avg_return:                             0.92313266
+    avg_benchmark_regret:                       0.060942978
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10787201
+    proxy_corner:
+      avg_avg_return:                           0.92313266
+      avg_reward_per_step:                      0.10787201
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        100663296
+    env-step-replay-before:                     51675136
+    env-step-used-before:                       100663296
+    ppo-update-before:                          61440
+    env-step-generate-before:                   48988160
+    env-step-all-after:                         100696064
+    env-step-replay-after:                      51675136
+    env-step-used-after:                        100696064
+    ppo-update-after:                           61460
+    env-step-generate-after:                    49020928
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     22.161093
+    avg_avg_return:                             0.835245
+    avg_reward_per_step:                        0.089416504
+    proxy_corner:
+      avg_avg_return:                           0.835245
+      avg_reward_per_step:                      0.089416504
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     22.161093
+    avg_avg_return:                             0.835245
+    avg_reward_per_step:                        0.089416504
+    proxy_corner:
+      avg_avg_return:                           0.835245
+      avg_reward_per_step:                      0.089416504
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  perf:
+    env_steps_per_second:                       341259.3538601418
+    ppo_updates_per_second:                     42.144432769295555
+    eval-batch-orig-duration:                   0.6106222150847316
+    eval-batch-shift-duration:                  0.5158425141125917
+    eval-batch-tree-duration:                   0.5314902001991868
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             27.864258
+      mouse-cheese_dist_finite_wavg:            17.19058
+      mouse-cheese_dist_solvable_avg:           27.864258
+      mouse-cheese_dist_solvable_wavg:          17.190584
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.44342798
+    visit_patterns:
+      avg_first_visit_time:                     1344.3408
+      avg_last_visit_time:                      1567.2285
+      num_replay_batches:                       1577
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009858609
+    avg_actor_approxkl3:                        0.0005751424
+    avg_actor_clipfrac:                         0.017907714
+    avg_actor_loss:                             -0.00056880154
+    avg_advantage:                              0.026669893
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0016130174
+    avg_entropy:                                0.06938409
+    avg_grad_norm_pre_clip:                     1.077631
+    avg_loss:                                   0.00016832302
+    max:
+      max_actor_approxkl1:                      0.0031477665
+      max_actor_approxkl3:                      0.0017429275
+      max_actor_clipfrac:                       0.034179688
+      max_actor_loss:                           0.0011862591
+      max_advantage:                            0.21218497
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0025417751
+      max_entropy:                              0.08834634
+      max_grad_norm_pre_clip:                   1.7953658
+      max_loss:                                 0.0018253418
+    std:
+      std_actor_approxkl1:                      0.0007942938
+      std_actor_approxkl3:                      0.0003519038
+      std_actor_clipfrac:                       0.0075119473
+      std_actor_loss:                           0.0008245336
+      std_advantage:                            0.05465254
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00052484934
+      std_entropy:                              0.0100807855
+      std_grad_norm_pre_clip:                   0.3852186
+      std_loss:                                 0.0007730487
+  eval-batch-orig:
+    avg_avg_episode_length:                     18.947432
+    avg_avg_return:                             0.92790794
+    avg_benchmark_regret:                       0.056918584
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09240723
+    proxy_corner:
+      avg_avg_return:                           0.92790794
+      avg_reward_per_step:                      0.09240723
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-shift:
+    avg_avg_episode_length:                     69.197975
+    avg_avg_return:                             0.23302953
+    avg_benchmark_regret:                       0.75386095
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.033935547
+    proxy_corner:
+      avg_avg_return:                           0.6298648
+      avg_reward_per_step:                      0.0052261353
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-tree:
+    avg_avg_episode_length:                     19.952185
+    avg_avg_return:                             0.92586017
+    avg_benchmark_regret:                       0.058215406
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10851288
+    proxy_corner:
+      avg_avg_return:                           0.92586017
+      avg_reward_per_step:                      0.10851288
+    rollouts_gif:                               float32(132, 225, 225, 3)
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        101711872
+    env-step-replay-before:                     52101120
+    env-step-used-before:                       101711872
+    ppo-update-before:                          62080
+    env-step-generate-before:                   49610752
+    env-step-all-after:                         101744640
+    env-step-replay-after:                      52133888
+    env-step-used-after:                        101744640
+    ppo-update-after:                           62100
+    env-step-generate-after:                    49610752
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.85142
+    avg_avg_return:                             0.5287536
+    avg_reward_per_step:                        0.014678955
+    proxy_corner:
+      avg_avg_return:                           0.5287536
+      avg_reward_per_step:                      0.014678955
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.85142
+    avg_avg_return:                             0.5287536
+    avg_reward_per_step:                        0.014678955
+    proxy_corner:
+      avg_avg_return:                           0.5287536
+      avg_reward_per_step:                      0.014678955
+  perf:
+    env_steps_per_second:                       239448.57563993934
+    ppo_updates_per_second:                     42.357759528826804
+    eval-batch-orig-duration:                   0.672754998318851
+    eval-batch-shift-duration:                  0.6059556193649769
+    eval-batch-tree-duration:                   0.6084894621744752
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             27.908203
+      mouse-cheese_dist_finite_wavg:            18.08761
+      mouse-cheese_dist_solvable_avg:           27.908203
+      mouse-cheese_dist_solvable_wavg:          18.087612
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.4497221
+    visit_patterns:
+      avg_first_visit_time:                     1358.3853
+      avg_last_visit_time:                      1581.137
+      num_replay_batches:                       1591
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011939675
+    avg_actor_approxkl3:                        0.0010755929
+    avg_actor_clipfrac:                         0.04144287
+    avg_actor_loss:                             -0.0010040206
+    avg_advantage:                              -0.014669117
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.006319204
+    avg_entropy:                                0.17694262
+    avg_grad_norm_pre_clip:                     1.4234768
+    avg_loss:                                   0.0019786393
+    max:
+      max_actor_approxkl1:                      0.004709502
+      max_actor_approxkl3:                      0.0020697243
+      max_actor_clipfrac:                       0.0703125
+      max_actor_loss:                           0.00054459274
+      max_advantage:                            0.165663
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.008723849
+      max_entropy:                              0.19131543
+      max_grad_norm_pre_clip:                   2.6173964
+      max_loss:                                 0.0042003416
+    std:
+      std_actor_approxkl1:                      0.0010666935
+      std_actor_approxkl3:                      0.00058668567
+      std_actor_clipfrac:                       0.018512337
+      std_actor_loss:                           0.00085351005
+      std_advantage:                            0.112307474
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.001515447
+      std_entropy:                              0.008221467
+      std_grad_norm_pre_clip:                   0.46691036
+      std_loss:                                 0.0010469358
+  eval-batch-orig:
+    avg_avg_episode_length:                     18.742083
+    avg_avg_return:                             0.93270445
+    avg_benchmark_regret:                       0.052122116
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09146881
+    proxy_corner:
+      avg_avg_return:                           0.93270445
+      avg_reward_per_step:                      0.09146881
+  eval-batch-shift:
+    avg_avg_episode_length:                     68.23869
+    avg_avg_return:                             0.24655843
+    avg_benchmark_regret:                       0.740332
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.034866333
+    proxy_corner:
+      avg_avg_return:                           0.6257174
+      avg_reward_per_step:                      0.0052871704
+  eval-batch-tree:
+    avg_avg_episode_length:                     19.613937
+    avg_avg_return:                             0.92963254
+    avg_benchmark_regret:                       0.05444307
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10785675
+    proxy_corner:
+      avg_avg_return:                           0.92963254
+      avg_reward_per_step:                      0.10785675
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        102760448
+    env-step-replay-before:                     52625408
+    env-step-used-before:                       102760448
+    ppo-update-before:                          62720
+    env-step-generate-before:                   50135040
+    env-step-all-after:                         102793216
+    env-step-replay-after:                      52625408
+    env-step-used-after:                        102793216
+    ppo-update-after:                           62740
+    env-step-generate-after:                    50167808
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.329613
+    avg_avg_return:                             0.85611725
+    avg_reward_per_step:                        0.09234619
+    proxy_corner:
+      avg_avg_return:                           0.85611725
+      avg_reward_per_step:                      0.09234619
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.329613
+    avg_avg_return:                             0.85611725
+    avg_reward_per_step:                        0.09234619
+    proxy_corner:
+      avg_avg_return:                           0.85611725
+      avg_reward_per_step:                      0.09234619
+  perf:
+    env_steps_per_second:                       330381.7460309585
+    ppo_updates_per_second:                     41.281367487053735
+    eval-batch-orig-duration:                   0.537839557044208
+    eval-batch-shift-duration:                  0.4872856782749295
+    eval-batch-tree-duration:                   0.47800321877002716
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             27.942627
+      mouse-cheese_dist_finite_wavg:            17.194252
+      mouse-cheese_dist_solvable_avg:           27.942627
+      mouse-cheese_dist_solvable_wavg:          17.194252
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.46066734
+    visit_patterns:
+      avg_first_visit_time:                     1376.0376
+      avg_last_visit_time:                      1596.4263
+      num_replay_batches:                       1606
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0007130704
+    avg_actor_approxkl3:                        0.0005916554
+    avg_actor_clipfrac:                         0.01776123
+    avg_actor_loss:                             -0.00014762246
+    avg_advantage:                              0.037564017
+    avg_critic_clipfrac:                        0.008343506
+    avg_critic_loss:                            0.0012957868
+    avg_entropy:                                0.052119363
+    avg_grad_norm_pre_clip:                     1.3046272
+    avg_loss:                                   0.00044815158
+    max:
+      max_actor_approxkl1:                      0.0018357653
+      max_actor_approxkl3:                      0.0013196254
+      max_actor_clipfrac:                       0.03857422
+      max_actor_loss:                           0.0016752817
+      max_advantage:                            0.18705267
+      max_critic_clipfrac:                      0.079711914
+      max_critic_loss:                          0.0028077527
+      max_entropy:                              0.071757644
+      max_grad_norm_pre_clip:                   2.5480664
+      max_loss:                                 0.002750132
+    std:
+      std_actor_approxkl1:                      0.0005307903
+      std_actor_approxkl3:                      0.00032363326
+      std_actor_clipfrac:                       0.008033163
+      std_actor_loss:                           0.0008543542
+      std_advantage:                            0.04482136
+      std_critic_clipfrac:                      0.020523189
+      std_critic_loss:                          0.0006184365
+      std_entropy:                              0.008798854
+      std_grad_norm_pre_clip:                   0.49819854
+      std_loss:                                 0.0009386397
+  eval-batch-orig:
+    avg_avg_episode_length:                     19.280416
+    avg_avg_return:                             0.92843187
+    avg_benchmark_regret:                       0.056394666
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.092163086
+    proxy_corner:
+      avg_avg_return:                           0.92843187
+      avg_reward_per_step:                      0.092163086
+  eval-batch-shift:
+    avg_avg_episode_length:                     69.076004
+    avg_avg_return:                             0.23152779
+    avg_benchmark_regret:                       0.75536263
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03591156
+    proxy_corner:
+      avg_avg_return:                           0.65431947
+      avg_reward_per_step:                      0.0053863525
+  eval-batch-tree:
+    avg_avg_episode_length:                     19.655014
+    avg_avg_return:                             0.92812806
+    avg_benchmark_regret:                       0.055947516
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.108314514
+    proxy_corner:
+      avg_avg_return:                           0.92812806
+      avg_reward_per_step:                      0.108314514
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        103809024
+    env-step-replay-before:                     53149696
+    env-step-used-before:                       103809024
+    ppo-update-before:                          63360
+    env-step-generate-before:                   50659328
+    env-step-all-after:                         103841792
+    env-step-replay-after:                      53182464
+    env-step-used-after:                        103841792
+    ppo-update-after:                           63380
+    env-step-generate-after:                    50659328
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     62.03496
+    avg_avg_return:                             0.5032946
+    avg_reward_per_step:                        0.013946533
+    proxy_corner:
+      avg_avg_return:                           0.5032946
+      avg_reward_per_step:                      0.013946533
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     62.03496
+    avg_avg_return:                             0.5032946
+    avg_reward_per_step:                        0.013946533
+    proxy_corner:
+      avg_avg_return:                           0.5032946
+      avg_reward_per_step:                      0.013946533
+  perf:
+    env_steps_per_second:                       292890.56941741507
+    ppo_updates_per_second:                     42.18877552705326
+    eval-batch-orig-duration:                   0.52242068387568
+    eval-batch-shift-duration:                  0.45827337726950645
+    eval-batch-tree-duration:                   0.4517963891848922
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.067383
+      mouse-cheese_dist_finite_wavg:            15.391729
+      mouse-cheese_dist_solvable_avg:           28.067383
+      mouse-cheese_dist_solvable_wavg:          15.391729
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.45716304
+    visit_patterns:
+      avg_first_visit_time:                     1389.1462
+      avg_last_visit_time:                      1613.1567
+      num_replay_batches:                       1623
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011037613
+    avg_actor_approxkl3:                        0.0009983646
+    avg_actor_clipfrac:                         0.040771484
+    avg_actor_loss:                             -0.0007106066
+    avg_advantage:                              -0.024112277
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0069534755
+    avg_entropy:                                0.18379761
+    avg_grad_norm_pre_clip:                     1.6022078
+    avg_loss:                                   0.0025823335
+    max:
+      max_actor_approxkl1:                      0.0022577376
+      max_actor_approxkl3:                      0.0023367628
+      max_actor_clipfrac:                       0.06677246
+      max_actor_loss:                           0.0012401491
+      max_advantage:                            0.15792984
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.009279546
+      max_entropy:                              0.19791931
+      max_grad_norm_pre_clip:                   2.7500284
+      max_loss:                                 0.0056960573
+    std:
+      std_actor_approxkl1:                      0.00066084566
+      std_actor_approxkl3:                      0.00051968364
+      std_actor_clipfrac:                       0.015709931
+      std_actor_loss:                           0.00085494586
+      std_advantage:                            0.11761982
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0012359376
+      std_entropy:                              0.007853076
+      std_grad_norm_pre_clip:                   0.5432395
+      std_loss:                                 0.0010299742
+  eval-batch-orig:
+    avg_avg_episode_length:                     19.213026
+    avg_avg_return:                             0.92564106
+    avg_benchmark_regret:                       0.059185475
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.092315674
+    proxy_corner:
+      avg_avg_return:                           0.92564106
+      avg_reward_per_step:                      0.092315674
+  eval-batch-shift:
+    avg_avg_episode_length:                     69.69888
+    avg_avg_return:                             0.223458
+    avg_benchmark_regret:                       0.76343244
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.034088135
+    proxy_corner:
+      avg_avg_return:                           0.6386311
+      avg_reward_per_step:                      0.0052871704
+  eval-batch-tree:
+    avg_avg_episode_length:                     19.712564
+    avg_avg_return:                             0.9242303
+    avg_benchmark_regret:                       0.05984536
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10926056
+    proxy_corner:
+      avg_avg_return:                           0.9242303
+      avg_reward_per_step:                      0.10926056
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        104857600
+    env-step-replay-before:                     53772288
+    env-step-used-before:                       104857600
+    ppo-update-before:                          64000
+    env-step-generate-before:                   51085312
+    env-step-all-after:                         104890368
+    env-step-replay-after:                      53772288
+    env-step-used-after:                        104890368
+    ppo-update-after:                           64020
+    env-step-generate-after:                    51118080
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     21.017544
+    avg_avg_return:                             0.8391042
+    avg_reward_per_step:                        0.10003662
+    proxy_corner:
+      avg_avg_return:                           0.8391042
+      avg_reward_per_step:                      0.10003662
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     21.017544
+    avg_avg_return:                             0.8391042
+    avg_reward_per_step:                        0.10003662
+    proxy_corner:
+      avg_avg_return:                           0.8391042
+      avg_reward_per_step:                      0.10003662
+  perf:
+    env_steps_per_second:                       324441.2207214536
+    ppo_updates_per_second:                     41.79311389732175
+    eval-batch-orig-duration:                   0.46605240274220705
+    eval-batch-shift-duration:                  0.45626780949532986
+    eval-batch-tree-duration:                   0.41462576389312744
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.074707
+      mouse-cheese_dist_finite_wavg:            18.104584
+      mouse-cheese_dist_solvable_avg:           28.074707
+      mouse-cheese_dist_solvable_wavg:          18.104588
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.420168
+    scoring:
+      avg_scores:                               0.44746405
+    visit_patterns:
+      avg_first_visit_time:                     1401.3186
+      avg_last_visit_time:                      1631.3577
+      num_replay_batches:                       1641
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00092503376
+    avg_actor_approxkl3:                        0.00091559434
+    avg_actor_clipfrac:                         0.021813964
+    avg_actor_loss:                             -0.00025326014
+    avg_advantage:                              0.02306103
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.001325725
+    avg_entropy:                                0.057326205
+    avg_grad_norm_pre_clip:                     1.129296
+    avg_loss:                                   0.0003522761
+    max:
+      max_actor_approxkl1:                      0.0038665766
+      max_actor_approxkl3:                      0.001854499
+      max_actor_clipfrac:                       0.036499023
+      max_actor_loss:                           0.0016765967
+      max_advantage:                            0.16316152
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0034619952
+      max_entropy:                              0.07304284
+      max_grad_norm_pre_clip:                   2.6830397
+      max_loss:                                 0.0031773748
+    std:
+      std_actor_approxkl1:                      0.0009366021
+      std_actor_approxkl3:                      0.00042961116
+      std_actor_clipfrac:                       0.008428609
+      std_actor_loss:                           0.0011043344
+      std_advantage:                            0.04727694
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0006524225
+      std_entropy:                              0.0074692094
+      std_grad_norm_pre_clip:                   0.57630783
+      std_loss:                                 0.0012075988
+  eval-batch-orig:
+    avg_avg_episode_length:                     18.415186
+    avg_avg_return:                             0.93611246
+    avg_benchmark_regret:                       0.04871406
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09195709
+    proxy_corner:
+      avg_avg_return:                           0.93611246
+      avg_reward_per_step:                      0.09195709
+  eval-batch-shift:
+    avg_avg_episode_length:                     67.98404
+    avg_avg_return:                             0.25264716
+    avg_benchmark_regret:                       0.7342433
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.035598755
+    proxy_corner:
+      avg_avg_return:                           0.64661765
+      avg_reward_per_step:                      0.0055389404
+  eval-batch-tree:
+    avg_avg_episode_length:                     19.361004
+    avg_avg_return:                             0.9277472
+    avg_benchmark_regret:                       0.056328427
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10883331
+    proxy_corner:
+      avg_avg_return:                           0.9277472
+      avg_reward_per_step:                      0.10883331
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        105906176
+    env-step-replay-before:                     54198272
+    env-step-used-before:                       105906176
+    ppo-update-before:                          64640
+    env-step-generate-before:                   51707904
+    env-step-all-after:                         105938944
+    env-step-replay-after:                      54198272
+    env-step-used-after:                        105938944
+    ppo-update-after:                           64660
+    env-step-generate-after:                    51740672
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     22.103493
+    avg_avg_return:                             0.83043295
+    avg_reward_per_step:                        0.08828735
+    proxy_corner:
+      avg_avg_return:                           0.83043295
+      avg_reward_per_step:                      0.08828735
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     22.103493
+    avg_avg_return:                             0.83043295
+    avg_reward_per_step:                        0.08828735
+    proxy_corner:
+      avg_avg_return:                           0.83043295
+      avg_reward_per_step:                      0.08828735
+  perf:
+    env_steps_per_second:                       316265.7500644835
+    ppo_updates_per_second:                     42.20451809183798
+    eval-batch-orig-duration:                   0.5380100533366203
+    eval-batch-shift-duration:                  0.5144329639151692
+    eval-batch-tree-duration:                   0.49386947602033615
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.105225
+      mouse-cheese_dist_finite_wavg:            15.415442
+      mouse-cheese_dist_solvable_avg:           28.105225
+      mouse-cheese_dist_solvable_wavg:          15.415442
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.46040484
+    visit_patterns:
+      avg_first_visit_time:                     1417.4236
+      avg_last_visit_time:                      1644.3135
+      num_replay_batches:                       1654
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00035782042
+    avg_actor_approxkl3:                        0.0005700026
+    avg_actor_clipfrac:                         0.018511964
+    avg_actor_loss:                             -0.00051428954
+    avg_advantage:                              0.013276251
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.001367489
+    avg_entropy:                                0.0658637
+    avg_grad_norm_pre_clip:                     1.2313575
+    avg_loss:                                   0.00010359143
+    max:
+      max_actor_approxkl1:                      0.0013574589
+      max_actor_approxkl3:                      0.0011906865
+      max_actor_clipfrac:                       0.030273438
+      max_actor_loss:                           0.00078891125
+      max_advantage:                            0.16883779
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.002680978
+      max_entropy:                              0.081092015
+      max_grad_norm_pre_clip:                   2.0128057
+      max_loss:                                 0.00133375
+    std:
+      std_actor_approxkl1:                      0.00039419113
+      std_actor_approxkl3:                      0.00029845367
+      std_actor_clipfrac:                       0.0065139444
+      std_actor_loss:                           0.00077419024
+      std_advantage:                            0.050511908
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0005831275
+      std_entropy:                              0.0068690227
+      std_grad_norm_pre_clip:                   0.40835917
+      std_loss:                                 0.0007715867
+  eval-batch-orig:
+    avg_avg_episode_length:                     18.77323
+    avg_avg_return:                             0.93071556
+    avg_benchmark_regret:                       0.05411096
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09188843
+    proxy_corner:
+      avg_avg_return:                           0.93071556
+      avg_reward_per_step:                      0.09188843
+  eval-batch-shift:
+    avg_avg_episode_length:                     68.851746
+    avg_avg_return:                             0.2423592
+    avg_benchmark_regret:                       0.7445313
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03414917
+    proxy_corner:
+      avg_avg_return:                           0.6424596
+      avg_reward_per_step:                      0.0053710938
+  eval-batch-tree:
+    avg_avg_episode_length:                     19.129364
+    avg_avg_return:                             0.9302287
+    avg_benchmark_regret:                       0.05384689
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10940552
+    proxy_corner:
+      avg_avg_return:                           0.9302287
+      avg_reward_per_step:                      0.10940552
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        106954752
+    env-step-replay-before:                     54624256
+    env-step-used-before:                       106954752
+    ppo-update-before:                          65280
+    env-step-generate-before:                   52330496
+    env-step-all-after:                         106987520
+    env-step-replay-after:                      54624256
+    env-step-used-after:                        106987520
+    ppo-update-after:                           65300
+    env-step-generate-after:                    52363264
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     21.489086
+    avg_avg_return:                             0.83298725
+    avg_reward_per_step:                        0.09661865
+    proxy_corner:
+      avg_avg_return:                           0.83298725
+      avg_reward_per_step:                      0.09661865
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     21.489086
+    avg_avg_return:                             0.83298725
+    avg_reward_per_step:                        0.09661865
+    proxy_corner:
+      avg_avg_return:                           0.83298725
+      avg_reward_per_step:                      0.09661865
+  perf:
+    env_steps_per_second:                       729717.09262149
+    ppo_updates_per_second:                     42.136128314217146
+    eval-batch-orig-duration:                   0.30863821413367987
+    eval-batch-shift-duration:                  0.2570579769089818
+    eval-batch-tree-duration:                   0.24109673779457808
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.189941
+      mouse-cheese_dist_finite_wavg:            14.521424
+      mouse-cheese_dist_solvable_avg:           28.189941
+      mouse-cheese_dist_solvable_wavg:          14.521424
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.4733672
+    visit_patterns:
+      avg_first_visit_time:                     1431.4333
+      avg_last_visit_time:                      1657.1716
+      num_replay_batches:                       1667
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012369625
+    avg_actor_approxkl3:                        0.0010252398
+    avg_actor_clipfrac:                         0.024340821
+    avg_actor_loss:                             0.0001527525
+    avg_advantage:                              0.036631837
+    avg_critic_clipfrac:                        9.1552734e-05
+    avg_critic_loss:                            0.0017796293
+    avg_entropy:                                0.06064673
+    avg_grad_norm_pre_clip:                     1.3232554
+    avg_loss:                                   0.0009819205
+    max:
+      max_actor_approxkl1:                      0.004370794
+      max_actor_approxkl3:                      0.0027582503
+      max_actor_clipfrac:                       0.045410156
+      max_actor_loss:                           0.0017437981
+      max_advantage:                            0.16899073
+      max_critic_clipfrac:                      0.0018310547
+      max_critic_loss:                          0.0037814993
+      max_entropy:                              0.07356186
+      max_grad_norm_pre_clip:                   2.1908765
+      max_loss:                                 0.0021575207
+    std:
+      std_actor_approxkl1:                      0.0010479582
+      std_actor_approxkl3:                      0.0007060109
+      std_actor_clipfrac:                       0.0106784245
+      std_actor_loss:                           0.0009487486
+      std_advantage:                            0.056101277
+      std_critic_clipfrac:                      0.0003990691
+      std_critic_loss:                          0.0007880251
+      std_entropy:                              0.007953741
+      std_grad_norm_pre_clip:                   0.4540199
+      std_loss:                                 0.0007618601
+  eval-batch-orig:
+    avg_avg_episode_length:                     18.947063
+    avg_avg_return:                             0.9264494
+    avg_benchmark_regret:                       0.05837714
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.091926575
+    proxy_corner:
+      avg_avg_return:                           0.9264494
+      avg_reward_per_step:                      0.091926575
+  eval-batch-shift:
+    avg_avg_episode_length:                     68.4825
+    avg_avg_return:                             0.2532583
+    avg_benchmark_regret:                       0.7336322
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.034370422
+    proxy_corner:
+      avg_avg_return:                           0.6181939
+      avg_reward_per_step:                      0.005264282
+  eval-batch-tree:
+    avg_avg_episode_length:                     19.29871
+    avg_avg_return:                             0.92776394
+    avg_benchmark_regret:                       0.05631166
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10883331
+    proxy_corner:
+      avg_avg_return:                           0.92776394
+      avg_reward_per_step:                      0.10883331
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        108003328
+    env-step-replay-before:                     55115776
+    env-step-used-before:                       108003328
+    ppo-update-before:                          65920
+    env-step-generate-before:                   52887552
+    env-step-all-after:                         108036096
+    env-step-replay-after:                      55148544
+    env-step-used-after:                        108036096
+    ppo-update-after:                           65940
+    env-step-generate-after:                    52887552
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     59.931053
+    avg_avg_return:                             0.51556635
+    avg_reward_per_step:                        0.014312744
+    proxy_corner:
+      avg_avg_return:                           0.51556635
+      avg_reward_per_step:                      0.014312744
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     59.931053
+    avg_avg_return:                             0.51556635
+    avg_reward_per_step:                        0.014312744
+    proxy_corner:
+      avg_avg_return:                           0.51556635
+      avg_reward_per_step:                      0.014312744
+  perf:
+    env_steps_per_second:                       321878.817902955
+    ppo_updates_per_second:                     42.305470969798264
+    eval-batch-orig-duration:                   0.522165727801621
+    eval-batch-shift-duration:                  0.4898249777033925
+    eval-batch-tree-duration:                   0.48817838076502085
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.261475
+      mouse-cheese_dist_finite_wavg:            13.62987
+      mouse-cheese_dist_solvable_avg:           28.261475
+      mouse-cheese_dist_solvable_wavg:          13.629873
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.46266654
+    visit_patterns:
+      avg_first_visit_time:                     1446.2136
+      avg_last_visit_time:                      1673.3169
+      num_replay_batches:                       1683
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011063365
+    avg_actor_approxkl3:                        0.0010402745
+    avg_actor_clipfrac:                         0.043823242
+    avg_actor_loss:                             -0.00094170496
+    avg_advantage:                              -0.014320883
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.006677565
+    avg_entropy:                                0.18627599
+    avg_grad_norm_pre_clip:                     1.7855663
+    avg_loss:                                   0.0022108017
+    max:
+      max_actor_approxkl1:                      0.003003242
+      max_actor_approxkl3:                      0.0026429023
+      max_actor_clipfrac:                       0.084228516
+      max_actor_loss:                           0.0010171831
+      max_advantage:                            0.19742978
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.008831883
+      max_entropy:                              0.19959156
+      max_grad_norm_pre_clip:                   3.2979321
+      max_loss:                                 0.0044102976
+    std:
+      std_actor_approxkl1:                      0.00083747477
+      std_actor_approxkl3:                      0.0006043358
+      std_actor_clipfrac:                       0.019160723
+      std_actor_loss:                           0.0010413049
+      std_advantage:                            0.115484275
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011826708
+      std_entropy:                              0.0075027193
+      std_grad_norm_pre_clip:                   0.6402921
+      std_loss:                                 0.0012584385
+  eval-batch-orig:
+    avg_avg_episode_length:                     18.693314
+    avg_avg_return:                             0.9340106
+    avg_benchmark_regret:                       0.05081593
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09226227
+    proxy_corner:
+      avg_avg_return:                           0.9340106
+      avg_reward_per_step:                      0.09226227
+  eval-batch-shift:
+    avg_avg_episode_length:                     68.247
+    avg_avg_return:                             0.24876285
+    avg_benchmark_regret:                       0.7381276
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03501892
+    proxy_corner:
+      avg_avg_return:                           0.6381724
+      avg_reward_per_step:                      0.005355835
+  eval-batch-tree:
+    avg_avg_episode_length:                     18.907635
+    avg_avg_return:                             0.9328353
+    avg_benchmark_regret:                       0.051240288
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.109573364
+    proxy_corner:
+      avg_avg_return:                           0.9328353
+      avg_reward_per_step:                      0.109573364
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        109051904
+    env-step-replay-before:                     55541760
+    env-step-used-before:                       109051904
+    ppo-update-before:                          66560
+    env-step-generate-before:                   53510144
+    env-step-all-after:                         109084672
+    env-step-replay-after:                      55574528
+    env-step-used-after:                        109084672
+    ppo-update-after:                           66580
+    env-step-generate-after:                    53510144
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     59.433167
+    avg_avg_return:                             0.53601325
+    avg_reward_per_step:                        0.014923096
+    proxy_corner:
+      avg_avg_return:                           0.53601325
+      avg_reward_per_step:                      0.014923096
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     59.433167
+    avg_avg_return:                             0.53601325
+    avg_reward_per_step:                        0.014923096
+    proxy_corner:
+      avg_avg_return:                           0.53601325
+      avg_reward_per_step:                      0.014923096
+  perf:
+    env_steps_per_second:                       319902.417781993
+    ppo_updates_per_second:                     44.50427354634129
+    eval-batch-orig-duration:                   0.5542032299563289
+    eval-batch-shift-duration:                  0.6104839304462075
+    eval-batch-tree-duration:                   0.6157414587214589
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.238037
+      mouse-cheese_dist_finite_wavg:            16.323196
+      mouse-cheese_dist_solvable_avg:           28.238037
+      mouse-cheese_dist_solvable_wavg:          16.323198
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.462565
+    visit_patterns:
+      avg_first_visit_time:                     1457.6204
+      avg_last_visit_time:                      1686.2537
+      num_replay_batches:                       1696
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0019148989
+    avg_actor_approxkl3:                        0.0015822599
+    avg_actor_clipfrac:                         0.06043091
+    avg_actor_loss:                             3.043078e-05
+    avg_advantage:                              -0.010025988
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.005796484
+    avg_entropy:                                0.18568116
+    avg_grad_norm_pre_clip:                     2.612171
+    avg_loss:                                   0.002742992
+    max:
+      max_actor_approxkl1:                      0.004231956
+      max_actor_approxkl3:                      0.002608322
+      max_actor_clipfrac:                       0.09667969
+      max_actor_loss:                           0.002728708
+      max_advantage:                            0.1767646
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.008198136
+      max_entropy:                              0.20452648
+      max_grad_norm_pre_clip:                   4.0083904
+      max_loss:                                 0.0056650396
+    std:
+      std_actor_approxkl1:                      0.0009989531
+      std_actor_approxkl3:                      0.0006405796
+      std_actor_clipfrac:                       0.021207074
+      std_actor_loss:                           0.0015319621
+      std_advantage:                            0.10765123
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011449531
+      std_entropy:                              0.009441366
+      std_grad_norm_pre_clip:                   0.8903065
+      std_loss:                                 0.0015972074
+  eval-batch-orig:
+    avg_avg_episode_length:                     18.839127
+    avg_avg_return:                             0.92943084
+    avg_benchmark_regret:                       0.0553957
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09260559
+    proxy_corner:
+      avg_avg_return:                           0.92943084
+      avg_reward_per_step:                      0.09260559
+  eval-batch-shift:
+    avg_avg_episode_length:                     68.9866
+    avg_avg_return:                             0.23762426
+    avg_benchmark_regret:                       0.7492662
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03567505
+    proxy_corner:
+      avg_avg_return:                           0.648268
+      avg_reward_per_step:                      0.005340576
+  eval-batch-tree:
+    avg_avg_episode_length:                     18.99936
+    avg_avg_return:                             0.9306725
+    avg_benchmark_regret:                       0.05340305
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10926819
+    proxy_corner:
+      avg_avg_return:                           0.9306725
+      avg_reward_per_step:                      0.10926819
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        110100480
+    env-step-replay-before:                     56066048
+    env-step-used-before:                       110100480
+    ppo-update-before:                          67200
+    env-step-generate-before:                   54034432
+    env-step-all-after:                         110133248
+    env-step-replay-after:                      56098816
+    env-step-used-after:                        110133248
+    ppo-update-after:                           67220
+    env-step-generate-after:                    54034432
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     60.83335
+    avg_avg_return:                             0.51991606
+    avg_reward_per_step:                        0.013793945
+    proxy_corner:
+      avg_avg_return:                           0.51991606
+      avg_reward_per_step:                      0.013793945
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     60.83335
+    avg_avg_return:                             0.51991606
+    avg_reward_per_step:                        0.013793945
+    proxy_corner:
+      avg_avg_return:                           0.51991606
+      avg_reward_per_step:                      0.013793945
+  perf:
+    env_steps_per_second:                       315342.9706962954
+    ppo_updates_per_second:                     42.260502545925874
+    eval-batch-orig-duration:                   0.5057845022529364
+    eval-batch-shift-duration:                  0.43162636645138264
+    eval-batch-tree-duration:                   0.41462341602891684
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.239502
+      mouse-cheese_dist_finite_wavg:            13.631739
+      mouse-cheese_dist_solvable_avg:           28.239502
+      mouse-cheese_dist_solvable_wavg:          13.631739
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.4572938
+    visit_patterns:
+      avg_first_visit_time:                     1472.5344
+      avg_last_visit_time:                      1702.1914
+      num_replay_batches:                       1712
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013859262
+    avg_actor_approxkl3:                        0.0010551462
+    avg_actor_clipfrac:                         0.042242434
+    avg_actor_loss:                             -0.00074012316
+    avg_advantage:                              -0.013129966
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.006228905
+    avg_entropy:                                0.17818122
+    avg_grad_norm_pre_clip:                     1.6530428
+    avg_loss:                                   0.002196148
+    max:
+      max_actor_approxkl1:                      0.0028066766
+      max_actor_approxkl3:                      0.00264234
+      max_actor_clipfrac:                       0.06542969
+      max_actor_loss:                           0.0009386912
+      max_advantage:                            0.1588245
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00893272
+      max_entropy:                              0.19731644
+      max_grad_norm_pre_clip:                   3.068058
+      max_loss:                                 0.0048167384
+    std:
+      std_actor_approxkl1:                      0.0007228266
+      std_actor_approxkl3:                      0.0005939988
+      std_actor_clipfrac:                       0.01678449
+      std_actor_loss:                           0.00079054025
+      std_advantage:                            0.11094417
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0015004914
+      std_entropy:                              0.010866581
+      std_grad_norm_pre_clip:                   0.59848297
+      std_loss:                                 0.0011234182
+  eval-batch-orig:
+    avg_avg_episode_length:                     18.294018
+    avg_avg_return:                             0.93936235
+    avg_benchmark_regret:                       0.04546412
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.091781616
+    proxy_corner:
+      avg_avg_return:                           0.93936235
+      avg_reward_per_step:                      0.091781616
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.74156
+    avg_avg_return:                             0.28007096
+    avg_benchmark_regret:                       0.7068195
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03691864
+    proxy_corner:
+      avg_avg_return:                           0.63413894
+      avg_reward_per_step:                      0.0054244995
+  eval-batch-tree:
+    avg_avg_episode_length:                     19.1478
+    avg_avg_return:                             0.92833227
+    avg_benchmark_regret:                       0.05574333
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10903168
+    proxy_corner:
+      avg_avg_return:                           0.92833227
+      avg_reward_per_step:                      0.10903168
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        111149056
+    env-step-replay-before:                     56688640
+    env-step-used-before:                       111149056
+    ppo-update-before:                          67840
+    env-step-generate-before:                   54460416
+    env-step-all-after:                         111181824
+    env-step-replay-after:                      56721408
+    env-step-used-after:                        111181824
+    ppo-update-after:                           67860
+    env-step-generate-after:                    54460416
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     57.5739
+    avg_avg_return:                             0.5528212
+    avg_reward_per_step:                        0.015258789
+    proxy_corner:
+      avg_avg_return:                           0.5528212
+      avg_reward_per_step:                      0.015258789
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     57.5739
+    avg_avg_return:                             0.5528212
+    avg_reward_per_step:                        0.015258789
+    proxy_corner:
+      avg_avg_return:                           0.5528212
+      avg_reward_per_step:                      0.015258789
+  perf:
+    env_steps_per_second:                       375708.8042475319
+    ppo_updates_per_second:                     42.24641414034269
+    eval-batch-orig-duration:                   0.5072185276076198
+    eval-batch-shift-duration:                  0.45350261125713587
+    eval-batch-tree-duration:                   0.4304341934621334
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.26831
+      mouse-cheese_dist_finite_wavg:            17.228191
+      mouse-cheese_dist_solvable_avg:           28.26831
+      mouse-cheese_dist_solvable_wavg:          17.228191
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.4515751
+    visit_patterns:
+      avg_first_visit_time:                     1486.2996
+      avg_last_visit_time:                      1721.1907
+      num_replay_batches:                       1731
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0019146163
+    avg_actor_approxkl3:                        0.0012487277
+    avg_actor_clipfrac:                         0.04677124
+    avg_actor_loss:                             -0.00020786114
+    avg_advantage:                              0.0069875065
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.005410482
+    avg_entropy:                                0.16448626
+    avg_grad_norm_pre_clip:                     1.8697226
+    avg_loss:                                   0.0023328937
+    max:
+      max_actor_approxkl1:                      0.0037755657
+      max_actor_approxkl3:                      0.0022757237
+      max_actor_clipfrac:                       0.07336426
+      max_actor_loss:                           0.003106594
+      max_advantage:                            0.1872338
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.008453034
+      max_entropy:                              0.18285242
+      max_grad_norm_pre_clip:                   3.3128881
+      max_loss:                                 0.004983485
+    std:
+      std_actor_approxkl1:                      0.0008187747
+      std_actor_approxkl3:                      0.00057467155
+      std_actor_clipfrac:                       0.0174765
+      std_actor_loss:                           0.0012923679
+      std_advantage:                            0.10398802
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011550589
+      std_entropy:                              0.011281779
+      std_grad_norm_pre_clip:                   0.68505865
+      std_loss:                                 0.001285405
+  eval-batch-orig:
+    avg_avg_episode_length:                     18.433113
+    avg_avg_return:                             0.9308388
+    avg_benchmark_regret:                       0.053987708
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09296417
+    proxy_corner:
+      avg_avg_return:                           0.9308388
+      avg_reward_per_step:                      0.09296417
+  eval-batch-shift:
+    avg_avg_episode_length:                     68.23046
+    avg_avg_return:                             0.25029522
+    avg_benchmark_regret:                       0.7365953
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.035057068
+    proxy_corner:
+      avg_avg_return:                           0.6418146
+      avg_reward_per_step:                      0.0053863525
+  eval-batch-tree:
+    avg_avg_episode_length:                     18.573906
+    avg_avg_return:                             0.9353596
+    avg_benchmark_regret:                       0.04871606
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10987091
+    proxy_corner:
+      avg_avg_return:                           0.9353596
+      avg_reward_per_step:                      0.10987091
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        112197632
+    env-step-replay-before:                     57376768
+    env-step-used-before:                       112197632
+    ppo-update-before:                          68480
+    env-step-generate-before:                   54820864
+    env-step-all-after:                         112230400
+    env-step-replay-after:                      57409536
+    env-step-used-after:                        112230400
+    ppo-update-after:                           68500
+    env-step-generate-after:                    54820864
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     60.356007
+    avg_avg_return:                             0.5205282
+    avg_reward_per_step:                        0.014312744
+    proxy_corner:
+      avg_avg_return:                           0.5205282
+      avg_reward_per_step:                      0.014312744
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     60.356007
+    avg_avg_return:                             0.5205282
+    avg_reward_per_step:                        0.014312744
+    proxy_corner:
+      avg_avg_return:                           0.5205282
+      avg_reward_per_step:                      0.014312744
+  perf:
+    env_steps_per_second:                       347196.8418292458
+    ppo_updates_per_second:                     41.48045274554302
+    eval-batch-orig-duration:                   0.5398276522755623
+    eval-batch-shift-duration:                  0.4882318116724491
+    eval-batch-tree-duration:                   0.4894543634727597
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.409668
+      mouse-cheese_dist_finite_wavg:            18.124846
+      mouse-cheese_dist_solvable_avg:           28.409668
+      mouse-cheese_dist_solvable_wavg:          18.124846
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.45242465
+    visit_patterns:
+      avg_first_visit_time:                     1496.418
+      avg_last_visit_time:                      1742.2595
+      num_replay_batches:                       1752
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013213521
+    avg_actor_approxkl3:                        0.0012214915
+    avg_actor_clipfrac:                         0.047198486
+    avg_actor_loss:                             -0.00047926494
+    avg_advantage:                              -0.014642818
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0062008523
+    avg_entropy:                                0.17011012
+    avg_grad_norm_pre_clip:                     1.5348678
+    avg_loss:                                   0.0024510515
+    max:
+      max_actor_approxkl1:                      0.0028867598
+      max_actor_approxkl3:                      0.002446731
+      max_actor_clipfrac:                       0.06921387
+      max_actor_loss:                           0.0014191195
+      max_advantage:                            0.1476422
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.008726852
+      max_entropy:                              0.18962872
+      max_grad_norm_pre_clip:                   2.514741
+      max_loss:                                 0.004435029
+    std:
+      std_actor_approxkl1:                      0.0006619341
+      std_actor_approxkl3:                      0.0004956882
+      std_actor_clipfrac:                       0.015232118
+      std_actor_loss:                           0.0008085849
+      std_advantage:                            0.1112126
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0014832147
+      std_entropy:                              0.009201384
+      std_grad_norm_pre_clip:                   0.36903894
+      std_loss:                                 0.0010450564
+  eval-batch-orig:
+    avg_avg_episode_length:                     18.412502
+    avg_avg_return:                             0.9371666
+    avg_benchmark_regret:                       0.04766
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.093185425
+    proxy_corner:
+      avg_avg_return:                           0.9371666
+      avg_reward_per_step:                      0.093185425
+  eval-batch-shift:
+    avg_avg_episode_length:                     69.48317
+    avg_avg_return:                             0.23275988
+    avg_benchmark_regret:                       0.7541306
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.034591675
+    proxy_corner:
+      avg_avg_return:                           0.62748945
+      avg_reward_per_step:                      0.0052108765
+  eval-batch-tree:
+    avg_avg_episode_length:                     18.77669
+    avg_avg_return:                             0.9335134
+    avg_benchmark_regret:                       0.050562214
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10987854
+    proxy_corner:
+      avg_avg_return:                           0.9335134
+      avg_reward_per_step:                      0.10987854
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        113246208
+    env-step-replay-before:                     57901056
+    env-step-used-before:                       113246208
+    ppo-update-before:                          69120
+    env-step-generate-before:                   55345152
+    env-step-all-after:                         113278976
+    env-step-replay-after:                      57901056
+    env-step-used-after:                        113278976
+    ppo-update-after:                           69140
+    env-step-generate-after:                    55377920
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     22.205736
+    avg_avg_return:                             0.82449913
+    avg_reward_per_step:                        0.09765625
+    proxy_corner:
+      avg_avg_return:                           0.82449913
+      avg_reward_per_step:                      0.09765625
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     22.205736
+    avg_avg_return:                             0.82449913
+    avg_reward_per_step:                        0.09765625
+    proxy_corner:
+      avg_avg_return:                           0.82449913
+      avg_reward_per_step:                      0.09765625
+  perf:
+    env_steps_per_second:                       346809.9718289074
+    ppo_updates_per_second:                     42.221006864000486
+    eval-batch-orig-duration:                   0.49187234696000814
+    eval-batch-shift-duration:                  0.42986082192510366
+    eval-batch-tree-duration:                   0.45585884246975183
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.477783
+      mouse-cheese_dist_finite_wavg:            18.152029
+      mouse-cheese_dist_solvable_avg:           28.477783
+      mouse-cheese_dist_solvable_wavg:          18.152029
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.4547587
+    visit_patterns:
+      avg_first_visit_time:                     1506.9226
+      avg_last_visit_time:                      1757.2654
+      num_replay_batches:                       1767
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010682278
+    avg_actor_approxkl3:                        0.0008313843
+    avg_actor_clipfrac:                         0.020166015
+    avg_actor_loss:                             0.00028832976
+    avg_advantage:                              0.011848119
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0017612366
+    avg_entropy:                                0.051079262
+    avg_grad_norm_pre_clip:                     1.353494
+    avg_loss:                                   0.0011178687
+    max:
+      max_actor_approxkl1:                      0.0026697626
+      max_actor_approxkl3:                      0.0022788225
+      max_actor_clipfrac:                       0.034057617
+      max_actor_loss:                           0.002152212
+      max_advantage:                            0.15267634
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.002686888
+      max_entropy:                              0.063888624
+      max_grad_norm_pre_clip:                   2.4277332
+      max_loss:                                 0.0033575909
+    std:
+      std_actor_approxkl1:                      0.000763802
+      std_actor_approxkl3:                      0.0005661664
+      std_actor_clipfrac:                       0.0076693385
+      std_actor_loss:                           0.00072633964
+      std_advantage:                            0.059266046
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00047921445
+      std_entropy:                              0.0076513477
+      std_grad_norm_pre_clip:                   0.43539038
+      std_loss:                                 0.0008871688
+  eval-batch-orig:
+    avg_avg_episode_length:                     18.689133
+    avg_avg_return:                             0.9349897
+    avg_benchmark_regret:                       0.04983686
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09269714
+    proxy_corner:
+      avg_avg_return:                           0.9349897
+      avg_reward_per_step:                      0.09269714
+  eval-batch-shift:
+    avg_avg_episode_length:                     69.27243
+    avg_avg_return:                             0.22840543
+    avg_benchmark_regret:                       0.7584851
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.035980225
+    proxy_corner:
+      avg_avg_return:                           0.6303804
+      avg_reward_per_step:                      0.0051651
+  eval-batch-tree:
+    avg_avg_episode_length:                     19.155529
+    avg_avg_return:                             0.9289429
+    avg_benchmark_regret:                       0.055132635
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10946655
+    proxy_corner:
+      avg_avg_return:                           0.9289429
+      avg_reward_per_step:                      0.10946655
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        114294784
+    env-step-replay-before:                     58359808
+    env-step-used-before:                       114294784
+    ppo-update-before:                          69760
+    env-step-generate-before:                   55934976
+    env-step-all-after:                         114327552
+    env-step-replay-after:                      58392576
+    env-step-used-after:                        114327552
+    ppo-update-after:                           69780
+    env-step-generate-after:                    55934976
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.959312
+    avg_avg_return:                             0.53123796
+    avg_reward_per_step:                        0.014892578
+    proxy_corner:
+      avg_avg_return:                           0.53123796
+      avg_reward_per_step:                      0.014892578
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.959312
+    avg_avg_return:                             0.53123796
+    avg_reward_per_step:                        0.014892578
+    proxy_corner:
+      avg_avg_return:                           0.53123796
+      avg_reward_per_step:                      0.014892578
+  perf:
+    env_steps_per_second:                       297516.77665047004
+    ppo_updates_per_second:                     42.15746943967047
+    eval-batch-orig-duration:                   0.5372146284207702
+    eval-batch-shift-duration:                  0.47558029647916555
+    eval-batch-tree-duration:                   0.4421251295134425
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.474121
+      mouse-cheese_dist_finite_wavg:            16.352613
+      mouse-cheese_dist_solvable_avg:           28.474121
+      mouse-cheese_dist_solvable_wavg:          16.352615
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.45045477
+    visit_patterns:
+      avg_first_visit_time:                     1521.1404
+      avg_last_visit_time:                      1772.3044
+      num_replay_batches:                       1782
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013603262
+    avg_actor_approxkl3:                        0.0014611733
+    avg_actor_clipfrac:                         0.050097656
+    avg_actor_loss:                             0.00028093476
+    avg_advantage:                              -0.02315545
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0062168515
+    avg_entropy:                                0.1553155
+    avg_grad_norm_pre_clip:                     2.2449522
+    avg_loss:                                   0.0032340449
+    max:
+      max_actor_approxkl1:                      0.0034948408
+      max_actor_approxkl3:                      0.0029936521
+      max_actor_clipfrac:                       0.08178711
+      max_actor_loss:                           0.001782164
+      max_advantage:                            0.14123589
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.008792098
+      max_entropy:                              0.16908315
+      max_grad_norm_pre_clip:                   3.536542
+      max_loss:                                 0.004891815
+    std:
+      std_actor_approxkl1:                      0.0008951832
+      std_actor_approxkl3:                      0.00087460695
+      std_actor_clipfrac:                       0.019988073
+      std_actor_loss:                           0.00097114616
+      std_advantage:                            0.11131253
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0015833734
+      std_entropy:                              0.0077428385
+      std_grad_norm_pre_clip:                   0.7446788
+      std_loss:                                 0.001079689
+  eval-batch-orig:
+    avg_avg_episode_length:                     18.267086
+    avg_avg_return:                             0.9357539
+    avg_benchmark_regret:                       0.04907264
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.0923996
+    proxy_corner:
+      avg_avg_return:                           0.9357539
+      avg_reward_per_step:                      0.0923996
+  eval-batch-shift:
+    avg_avg_episode_length:                     67.827255
+    avg_avg_return:                             0.25782505
+    avg_benchmark_regret:                       0.7290654
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.036834717
+    proxy_corner:
+      avg_avg_return:                           0.63261473
+      avg_reward_per_step:                      0.005317688
+  eval-batch-tree:
+    avg_avg_episode_length:                     18.840302
+    avg_avg_return:                             0.93388575
+    avg_benchmark_regret:                       0.05018986
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10945892
+    proxy_corner:
+      avg_avg_return:                           0.93388575
+      avg_reward_per_step:                      0.10945892
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        115343360
+    env-step-replay-before:                     58982400
+    env-step-used-before:                       115343360
+    ppo-update-before:                          70400
+    env-step-generate-before:                   56360960
+    env-step-all-after:                         115376128
+    env-step-replay-after:                      58982400
+    env-step-used-after:                        115376128
+    ppo-update-after:                           70420
+    env-step-generate-after:                    56393728
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.242584
+    avg_avg_return:                             0.84980357
+    avg_reward_per_step:                        0.107940674
+    proxy_corner:
+      avg_avg_return:                           0.84980357
+      avg_reward_per_step:                      0.107940674
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.242584
+    avg_avg_return:                             0.84980357
+    avg_reward_per_step:                        0.107940674
+    proxy_corner:
+      avg_avg_return:                           0.84980357
+      avg_reward_per_step:                      0.107940674
+  perf:
+    env_steps_per_second:                       373634.3091898891
+    ppo_updates_per_second:                     42.181904591587625
+    eval-batch-orig-duration:                   0.4651999194175005
+    eval-batch-shift-duration:                  0.4480428332462907
+    eval-batch-tree-duration:                   0.4273696532472968
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.509521
+      mouse-cheese_dist_finite_wavg:            16.346872
+      mouse-cheese_dist_solvable_avg:           28.509521
+      mouse-cheese_dist_solvable_wavg:          16.346872
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000008
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.4456789
+    visit_patterns:
+      avg_first_visit_time:                     1533.2424
+      avg_last_visit_time:                      1790.1387
+      num_replay_batches:                       1800
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00066047517
+    avg_actor_approxkl3:                        0.00083074433
+    avg_actor_clipfrac:                         0.017053222
+    avg_actor_loss:                             -0.00034936445
+    avg_advantage:                              0.0019442049
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0012029603
+    avg_entropy:                                0.0447277
+    avg_grad_norm_pre_clip:                     1.1231481
+    avg_loss:                                   0.00020738787
+    max:
+      max_actor_approxkl1:                      0.0025193142
+      max_actor_approxkl3:                      0.0017034251
+      max_actor_clipfrac:                       0.03491211
+      max_actor_loss:                           0.0015159361
+      max_advantage:                            0.1431824
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0032658633
+      max_entropy:                              0.058587797
+      max_grad_norm_pre_clip:                   2.8675787
+      max_loss:                                 0.00248086
+    std:
+      std_actor_approxkl1:                      0.000824542
+      std_actor_approxkl3:                      0.0004753798
+      std_actor_clipfrac:                       0.007470374
+      std_actor_loss:                           0.00079421117
+      std_advantage:                            0.048845463
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00073902064
+      std_entropy:                              0.0067892782
+      std_grad_norm_pre_clip:                   0.63909554
+      std_loss:                                 0.0008271188
+  eval-batch-orig:
+    avg_avg_episode_length:                     18.053528
+    avg_avg_return:                             0.9403615
+    avg_benchmark_regret:                       0.04446502
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.0925293
+    proxy_corner:
+      avg_avg_return:                           0.9403615
+      avg_reward_per_step:                      0.0925293
+  eval-batch-shift:
+    avg_avg_episode_length:                     68.00987
+    avg_avg_return:                             0.2553383
+    avg_benchmark_regret:                       0.7315521
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.036239624
+    proxy_corner:
+      avg_avg_return:                           0.6167356
+      avg_reward_per_step:                      0.0051651
+  eval-batch-tree:
+    avg_avg_episode_length:                     18.826643
+    avg_avg_return:                             0.93213254
+    avg_benchmark_regret:                       0.051943056
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10921478
+    proxy_corner:
+      avg_avg_return:                           0.93213254
+      avg_reward_per_step:                      0.10921478
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        116391936
+    env-step-replay-before:                     59506688
+    env-step-used-before:                       116391936
+    ppo-update-before:                          71040
+    env-step-generate-before:                   56885248
+    env-step-all-after:                         116424704
+    env-step-replay-after:                      59539456
+    env-step-used-after:                        116424704
+    ppo-update-after:                           71060
+    env-step-generate-after:                    56885248
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     61.13305
+    avg_avg_return:                             0.49834424
+    avg_reward_per_step:                        0.013580322
+    proxy_corner:
+      avg_avg_return:                           0.49834424
+      avg_reward_per_step:                      0.013580322
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     61.13305
+    avg_avg_return:                             0.49834424
+    avg_reward_per_step:                        0.013580322
+    proxy_corner:
+      avg_avg_return:                           0.49834424
+      avg_reward_per_step:                      0.013580322
+  perf:
+    env_steps_per_second:                       323808.1460781905
+    ppo_updates_per_second:                     42.00751924475449
+    eval-batch-orig-duration:                   0.5366594549268484
+    eval-batch-shift-duration:                  0.499756938777864
+    eval-batch-tree-duration:                   0.48177865613251925
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.524414
+      mouse-cheese_dist_finite_wavg:            14.5434265
+      mouse-cheese_dist_solvable_avg:           28.524414
+      mouse-cheese_dist_solvable_wavg:          14.543428
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.44651958
+    visit_patterns:
+      avg_first_visit_time:                     1548.2109
+      avg_last_visit_time:                      1807.209
+      num_replay_batches:                       1817
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013733424
+    avg_actor_approxkl3:                        0.001162278
+    avg_actor_clipfrac:                         0.046545412
+    avg_actor_loss:                             -0.00071437884
+    avg_advantage:                              -0.020290883
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.007231393
+    avg_entropy:                                0.17453633
+    avg_grad_norm_pre_clip:                     1.8970867
+    avg_loss:                                   0.002726781
+    max:
+      max_actor_approxkl1:                      0.0031739622
+      max_actor_approxkl3:                      0.0022200067
+      max_actor_clipfrac:                       0.06726074
+      max_actor_loss:                           0.00075757504
+      max_advantage:                            0.1484943
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0112280175
+      max_entropy:                              0.19046071
+      max_grad_norm_pre_clip:                   3.5194707
+      max_loss:                                 0.0054321596
+    std:
+      std_actor_approxkl1:                      0.0008388015
+      std_actor_approxkl3:                      0.0005299756
+      std_actor_clipfrac:                       0.015779028
+      std_actor_loss:                           0.00065254595
+      std_advantage:                            0.120289326
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0015127453
+      std_entropy:                              0.007609577
+      std_grad_norm_pre_clip:                   0.7214165
+      std_loss:                                 0.0010568588
+  eval-batch-orig:
+    avg_avg_episode_length:                     18.310585
+    avg_avg_return:                             0.9337568
+    avg_benchmark_regret:                       0.051069707
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09198761
+    proxy_corner:
+      avg_avg_return:                           0.9337568
+      avg_reward_per_step:                      0.09198761
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.17351
+    avg_avg_return:                             0.2875706
+    avg_benchmark_regret:                       0.69931984
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.037712097
+    proxy_corner:
+      avg_avg_return:                           0.6191608
+      avg_reward_per_step:                      0.0055236816
+  eval-batch-tree:
+    avg_avg_episode_length:                     18.788956
+    avg_avg_return:                             0.9321825
+    avg_benchmark_regret:                       0.051893156
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10959625
+    proxy_corner:
+      avg_avg_return:                           0.9321825
+      avg_reward_per_step:                      0.10959625
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        117440512
+    env-step-replay-before:                     60030976
+    env-step-used-before:                       117440512
+    ppo-update-before:                          71680
+    env-step-generate-before:                   57409536
+    env-step-all-after:                         117473280
+    env-step-replay-after:                      60030976
+    env-step-used-after:                        117473280
+    ppo-update-after:                           71700
+    env-step-generate-after:                    57442304
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     20.344961
+    avg_avg_return:                             0.84342074
+    avg_reward_per_step:                        0.09841919
+    proxy_corner:
+      avg_avg_return:                           0.84342074
+      avg_reward_per_step:                      0.09841919
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     20.344961
+    avg_avg_return:                             0.84342074
+    avg_reward_per_step:                        0.09841919
+    proxy_corner:
+      avg_avg_return:                           0.84342074
+      avg_reward_per_step:                      0.09841919
+  perf:
+    env_steps_per_second:                       348115.59778082214
+    ppo_updates_per_second:                     44.61037259407002
+    eval-batch-orig-duration:                   0.5231676269322634
+    eval-batch-shift-duration:                  0.4664329206570983
+    eval-batch-tree-duration:                   0.48563808109611273
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.529053
+      mouse-cheese_dist_finite_wavg:            15.461628
+      mouse-cheese_dist_solvable_avg:           28.529053
+      mouse-cheese_dist_solvable_wavg:          15.461626
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.44348064
+    visit_patterns:
+      avg_first_visit_time:                     1562.7
+      avg_last_visit_time:                      1822.3057
+      num_replay_batches:                       1832
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.000624773
+    avg_actor_approxkl3:                        0.00098668
+    avg_actor_clipfrac:                         0.019049073
+    avg_actor_loss:                             7.489962e-05
+    avg_advantage:                              -0.0014932689
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.000871683
+    avg_entropy:                                0.041909877
+    avg_grad_norm_pre_clip:                     1.0854782
+    avg_loss:                                   0.00046883128
+    max:
+      max_actor_approxkl1:                      0.0018806318
+      max_actor_approxkl3:                      0.0022500765
+      max_actor_clipfrac:                       0.032714844
+      max_actor_loss:                           0.00087736454
+      max_advantage:                            0.1342203
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0019401798
+      max_entropy:                              0.054676633
+      max_grad_norm_pre_clip:                   1.938399
+      max_loss:                                 0.0015692367
+    std:
+      std_actor_approxkl1:                      0.0006052818
+      std_actor_approxkl3:                      0.0006224145
+      std_actor_clipfrac:                       0.0085773375
+      std_actor_loss:                           0.000563383
+      std_advantage:                            0.041476488
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0005436578
+      std_entropy:                              0.006708663
+      std_grad_norm_pre_clip:                   0.3725226
+      std_loss:                                 0.00059656677
+  eval-batch-orig:
+    avg_avg_episode_length:                     18.244358
+    avg_avg_return:                             0.9333822
+    avg_benchmark_regret:                       0.05144431
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09347534
+    proxy_corner:
+      avg_avg_return:                           0.9333822
+      avg_reward_per_step:                      0.09347534
+  eval-batch-shift:
+    avg_avg_episode_length:                     67.93256
+    avg_avg_return:                             0.25206685
+    avg_benchmark_regret:                       0.7348236
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.035980225
+    proxy_corner:
+      avg_avg_return:                           0.6385932
+      avg_reward_per_step:                      0.005393982
+  eval-batch-tree:
+    avg_avg_episode_length:                     18.400047
+    avg_avg_return:                             0.9342897
+    avg_benchmark_regret:                       0.049785923
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11066437
+    proxy_corner:
+      avg_avg_return:                           0.9342897
+      avg_reward_per_step:                      0.11066437
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        118489088
+    env-step-replay-before:                     60424192
+    env-step-used-before:                       118489088
+    ppo-update-before:                          72320
+    env-step-generate-before:                   58064896
+    env-step-all-after:                         118521856
+    env-step-replay-after:                      60424192
+    env-step-used-after:                        118521856
+    ppo-update-after:                           72340
+    env-step-generate-after:                    58097664
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.864727
+    avg_avg_return:                             0.8586438
+    avg_reward_per_step:                        0.09674072
+    proxy_corner:
+      avg_avg_return:                           0.8586438
+      avg_reward_per_step:                      0.09674072
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.864727
+    avg_avg_return:                             0.8586438
+    avg_reward_per_step:                        0.09674072
+    proxy_corner:
+      avg_avg_return:                           0.8586438
+      avg_reward_per_step:                      0.09674072
+  perf:
+    env_steps_per_second:                       350120.30271504755
+    ppo_updates_per_second:                     40.84527691292489
+    eval-batch-orig-duration:                   0.5205764714628458
+    eval-batch-shift-duration:                  0.4889101544395089
+    eval-batch-tree-duration:                   0.45924175437539816
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.63794
+      mouse-cheese_dist_finite_wavg:            17.26496
+      mouse-cheese_dist_solvable_avg:           28.63794
+      mouse-cheese_dist_solvable_wavg:          17.264961
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.46123457
+    visit_patterns:
+      avg_first_visit_time:                     1577.8005
+      avg_last_visit_time:                      1834.2229
+      num_replay_batches:                       1844
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00040201974
+    avg_actor_approxkl3:                        0.00051308144
+    avg_actor_clipfrac:                         0.014538574
+    avg_actor_loss:                             -0.00033711462
+    avg_advantage:                              0.0016930951
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00045380587
+    avg_entropy:                                0.04216952
+    avg_grad_norm_pre_clip:                     0.9868639
+    avg_loss:                                   -0.00015238118
+    max:
+      max_actor_approxkl1:                      0.0017099818
+      max_actor_approxkl3:                      0.0012299247
+      max_actor_clipfrac:                       0.024414062
+      max_actor_loss:                           0.000492638
+      max_advantage:                            0.12389481
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00090633176
+      max_entropy:                              0.049389627
+      max_grad_norm_pre_clip:                   2.0442455
+      max_loss:                                 0.0006187525
+    std:
+      std_actor_approxkl1:                      0.00048745793
+      std_actor_approxkl3:                      0.00030507168
+      std_actor_clipfrac:                       0.005570193
+      std_actor_loss:                           0.0005279461
+      std_advantage:                            0.030579265
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0002517277
+      std_entropy:                              0.004287012
+      std_grad_norm_pre_clip:                   0.4021936
+      std_loss:                                 0.0004853615
+  eval-batch-orig:
+    avg_avg_episode_length:                     18.180496
+    avg_avg_return:                             0.93813896
+    avg_benchmark_regret:                       0.046687596
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.0936203
+    proxy_corner:
+      avg_avg_return:                           0.93813896
+      avg_reward_per_step:                      0.0936203
+  eval-batch-shift:
+    avg_avg_episode_length:                     69.27924
+    avg_avg_return:                             0.23721802
+    avg_benchmark_regret:                       0.7496725
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.034118652
+    proxy_corner:
+      avg_avg_return:                           0.62014186
+      avg_reward_per_step:                      0.005180359
+  eval-batch-tree:
+    avg_avg_episode_length:                     18.459824
+    avg_avg_return:                             0.9333791
+    avg_benchmark_regret:                       0.05069649
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11022186
+    proxy_corner:
+      avg_avg_return:                           0.9333791
+      avg_reward_per_step:                      0.11022186
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        119537664
+    env-step-replay-before:                     60882944
+    env-step-used-before:                       119537664
+    ppo-update-before:                          72960
+    env-step-generate-before:                   58654720
+    env-step-all-after:                         119570432
+    env-step-replay-after:                      60882944
+    env-step-used-after:                        119570432
+    ppo-update-after:                           72980
+    env-step-generate-after:                    58687488
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     20.723324
+    avg_avg_return:                             0.84667265
+    avg_reward_per_step:                        0.08380127
+    proxy_corner:
+      avg_avg_return:                           0.84667265
+      avg_reward_per_step:                      0.08380127
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     20.723324
+    avg_avg_return:                             0.84667265
+    avg_reward_per_step:                        0.08380127
+    proxy_corner:
+      avg_avg_return:                           0.84667265
+      avg_reward_per_step:                      0.08380127
+  perf:
+    env_steps_per_second:                       289169.60168296506
+    ppo_updates_per_second:                     42.19020646280024
+    eval-batch-orig-duration:                   0.5428095739334822
+    eval-batch-shift-duration:                  0.48909036722034216
+    eval-batch-tree-duration:                   0.4875685600563884
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.642578
+      mouse-cheese_dist_finite_wavg:            16.365828
+      mouse-cheese_dist_solvable_avg:           28.642578
+      mouse-cheese_dist_solvable_wavg:          16.36583
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.46525234
+    visit_patterns:
+      avg_first_visit_time:                     1592.6804
+      avg_last_visit_time:                      1848.2502
+      num_replay_batches:                       1858
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0005287928
+    avg_actor_approxkl3:                        0.0005262223
+    avg_actor_clipfrac:                         0.013287353
+    avg_actor_loss:                             -0.0006462961
+    avg_advantage:                              0.016208827
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0006674835
+    avg_entropy:                                0.041755967
+    avg_grad_norm_pre_clip:                     0.8737524
+    avg_loss:                                   -0.00035431032
+    max:
+      max_actor_approxkl1:                      0.0014665117
+      max_actor_approxkl3:                      0.0015908601
+      max_actor_clipfrac:                       0.032958984
+      max_actor_loss:                           0.00043261796
+      max_advantage:                            0.14624774
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0015827055
+      max_entropy:                              0.057513278
+      max_grad_norm_pre_clip:                   1.8697463
+      max_loss:                                 0.0007439974
+    std:
+      std_actor_approxkl1:                      0.0003998005
+      std_actor_approxkl3:                      0.00039152816
+      std_actor_clipfrac:                       0.0070776865
+      std_actor_loss:                           0.0006573041
+      std_advantage:                            0.035388995
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00034914582
+      std_entropy:                              0.0065656104
+      std_grad_norm_pre_clip:                   0.38090923
+      std_loss:                                 0.00065680523
+  eval-batch-orig:
+    avg_avg_episode_length:                     18.2655
+    avg_avg_return:                             0.9382621
+    avg_benchmark_regret:                       0.046564374
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09314728
+    proxy_corner:
+      avg_avg_return:                           0.9382621
+      avg_reward_per_step:                      0.09314728
+  eval-batch-shift:
+    avg_avg_episode_length:                     68.20125
+    avg_avg_return:                             0.2447719
+    avg_benchmark_regret:                       0.7421186
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.036727905
+    proxy_corner:
+      avg_avg_return:                           0.64529693
+      avg_reward_per_step:                      0.005378723
+  eval-batch-tree:
+    avg_avg_episode_length:                     18.619997
+    avg_avg_return:                             0.93497086
+    avg_benchmark_regret:                       0.049104717
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11000824
+    proxy_corner:
+      avg_avg_return:                           0.93497086
+      avg_reward_per_step:                      0.11000824
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        120586240
+    env-step-replay-before:                     61407232
+    env-step-used-before:                       120586240
+    ppo-update-before:                          73600
+    env-step-generate-before:                   59179008
+    env-step-all-after:                         120619008
+    env-step-replay-after:                      61440000
+    env-step-used-after:                        120619008
+    ppo-update-after:                           73620
+    env-step-generate-after:                    59179008
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     60.872597
+    avg_avg_return:                             0.5278524
+    avg_reward_per_step:                        0.013946533
+    proxy_corner:
+      avg_avg_return:                           0.5278524
+      avg_reward_per_step:                      0.013946533
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     60.872597
+    avg_avg_return:                             0.5278524
+    avg_reward_per_step:                        0.013946533
+    proxy_corner:
+      avg_avg_return:                           0.5278524
+      avg_reward_per_step:                      0.013946533
+  perf:
+    env_steps_per_second:                       322596.0027109734
+    ppo_updates_per_second:                     42.33465832787471
+    eval-batch-orig-duration:                   0.5376579742878675
+    eval-batch-shift-duration:                  0.4223861116915941
+    eval-batch-tree-duration:                   0.4814814180135727
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.740967
+      mouse-cheese_dist_finite_wavg:            16.36572
+      mouse-cheese_dist_solvable_avg:           28.740967
+      mouse-cheese_dist_solvable_wavg:          16.365725
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.45088336
+    visit_patterns:
+      avg_first_visit_time:                     1605.3494
+      avg_last_visit_time:                      1865.2202
+      num_replay_batches:                       1875
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012457863
+    avg_actor_approxkl3:                        0.0012083432
+    avg_actor_clipfrac:                         0.044970702
+    avg_actor_loss:                             -0.00022144504
+    avg_advantage:                              -0.014940668
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.005546086
+    avg_entropy:                                0.15574399
+    avg_grad_norm_pre_clip:                     1.7345084
+    avg_loss:                                   0.002395854
+    max:
+      max_actor_approxkl1:                      0.0022524623
+      max_actor_approxkl3:                      0.0021943985
+      max_actor_clipfrac:                       0.06896973
+      max_actor_loss:                           0.0016710013
+      max_advantage:                            0.1289565
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00879178
+      max_entropy:                              0.1677764
+      max_grad_norm_pre_clip:                   2.9086123
+      max_loss:                                 0.00535229
+    std:
+      std_actor_approxkl1:                      0.00062383764
+      std_actor_approxkl3:                      0.0005002559
+      std_actor_clipfrac:                       0.014262462
+      std_actor_loss:                           0.0009243599
+      std_advantage:                            0.10505912
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0015624292
+      std_entropy:                              0.006129984
+      std_grad_norm_pre_clip:                   0.60684526
+      std_loss:                                 0.0012242916
+  eval-batch-orig:
+    avg_avg_episode_length:                     17.455416
+    avg_avg_return:                             0.944461
+    avg_benchmark_regret:                       0.040365547
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09402466
+    proxy_corner:
+      avg_avg_return:                           0.944461
+      avg_reward_per_step:                      0.09402466
+  eval-batch-shift:
+    avg_avg_episode_length:                     67.899956
+    avg_avg_return:                             0.2545619
+    avg_benchmark_regret:                       0.73232853
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.035568237
+    proxy_corner:
+      avg_avg_return:                           0.6444535
+      avg_reward_per_step:                      0.0055160522
+  eval-batch-tree:
+    avg_avg_episode_length:                     18.274563
+    avg_avg_return:                             0.93958926
+    avg_benchmark_regret:                       0.04448634
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11051178
+    proxy_corner:
+      avg_avg_return:                           0.93958926
+      avg_reward_per_step:                      0.11051178
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        121634816
+    env-step-replay-before:                     61964288
+    env-step-used-before:                       121634816
+    ppo-update-before:                          74240
+    env-step-generate-before:                   59670528
+    env-step-all-after:                         121667584
+    env-step-replay-after:                      61997056
+    env-step-used-after:                        121667584
+    ppo-update-after:                           74260
+    env-step-generate-after:                    59670528
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.626526
+    avg_avg_return:                             0.53443515
+    avg_reward_per_step:                        0.01473999
+    proxy_corner:
+      avg_avg_return:                           0.53443515
+      avg_reward_per_step:                      0.01473999
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.626526
+    avg_avg_return:                             0.53443515
+    avg_reward_per_step:                        0.01473999
+    proxy_corner:
+      avg_avg_return:                           0.53443515
+      avg_reward_per_step:                      0.01473999
+  perf:
+    env_steps_per_second:                       714481.3545230808
+    ppo_updates_per_second:                     42.19064502955957
+    eval-batch-orig-duration:                   0.3085832893848419
+    eval-batch-shift-duration:                  0.2689884742721915
+    eval-batch-tree-duration:                   0.24195881467312574
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.828125
+      mouse-cheese_dist_finite_wavg:            15.477119
+      mouse-cheese_dist_solvable_avg:           28.828125
+      mouse-cheese_dist_solvable_wavg:          15.477119
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.44923103
+    visit_patterns:
+      avg_first_visit_time:                     1614.197
+      avg_last_visit_time:                      1882.2751
+      num_replay_batches:                       1892
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0016799351
+    avg_actor_approxkl3:                        0.0015630372
+    avg_actor_clipfrac:                         0.054248046
+    avg_actor_loss:                             0.0002959881
+    avg_advantage:                              -0.005988463
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0061290967
+    avg_entropy:                                0.16274832
+    avg_grad_norm_pre_clip:                     1.9940283
+    avg_loss:                                   0.0031977878
+    max:
+      max_actor_approxkl1:                      0.0033344496
+      max_actor_approxkl3:                      0.0037709253
+      max_actor_clipfrac:                       0.099121094
+      max_actor_loss:                           0.004405074
+      max_advantage:                            0.18282312
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.008761856
+      max_entropy:                              0.17952242
+      max_grad_norm_pre_clip:                   3.8444717
+      max_loss:                                 0.008613047
+    std:
+      std_actor_approxkl1:                      0.0008714515
+      std_actor_approxkl3:                      0.00095231493
+      std_actor_clipfrac:                       0.021923449
+      std_actor_loss:                           0.0018411116
+      std_advantage:                            0.11053977
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010519515
+      std_entropy:                              0.007893682
+      std_grad_norm_pre_clip:                   0.8701091
+      std_loss:                                 0.0019756933
+  eval-batch-orig:
+    avg_avg_episode_length:                     17.499125
+    avg_avg_return:                             0.94428015
+    avg_benchmark_regret:                       0.04054638
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09366608
+    proxy_corner:
+      avg_avg_return:                           0.94428015
+      avg_reward_per_step:                      0.09366608
+  eval-batch-shift:
+    avg_avg_episode_length:                     67.51565
+    avg_avg_return:                             0.2631171
+    avg_benchmark_regret:                       0.72377336
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.036010742
+    proxy_corner:
+      avg_avg_return:                           0.6201144
+      avg_reward_per_step:                      0.0052871704
+  eval-batch-tree:
+    avg_avg_episode_length:                     18.325518
+    avg_avg_return:                             0.93371296
+    avg_benchmark_regret:                       0.050362613
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11035919
+    proxy_corner:
+      avg_avg_return:                           0.93371296
+      avg_reward_per_step:                      0.11035919
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        122683392
+    env-step-replay-before:                     62390272
+    env-step-used-before:                       122683392
+    ppo-update-before:                          74880
+    env-step-generate-before:                   60293120
+    env-step-all-after:                         122716160
+    env-step-replay-after:                      62390272
+    env-step-used-after:                        122716160
+    ppo-update-after:                           74900
+    env-step-generate-after:                    60325888
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.772915
+    avg_avg_return:                             0.84906614
+    avg_reward_per_step:                        0.10397339
+    proxy_corner:
+      avg_avg_return:                           0.84906614
+      avg_reward_per_step:                      0.10397339
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.772915
+    avg_avg_return:                             0.84906614
+    avg_reward_per_step:                        0.10397339
+    proxy_corner:
+      avg_avg_return:                           0.84906614
+      avg_reward_per_step:                      0.10397339
+  perf:
+    env_steps_per_second:                       295424.74405300256
+    ppo_updates_per_second:                     42.29140320907602
+    eval-batch-orig-duration:                   0.5231009796261787
+    eval-batch-shift-duration:                  0.4886764455586672
+    eval-batch-tree-duration:                   0.47918738331645727
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.796387
+      mouse-cheese_dist_finite_wavg:            16.376143
+      mouse-cheese_dist_solvable_avg:           28.796387
+      mouse-cheese_dist_solvable_wavg:          16.376143
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.4666936
+    visit_patterns:
+      avg_first_visit_time:                     1630.124
+      avg_last_visit_time:                      1894.2058
+      num_replay_batches:                       1904
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009264687
+    avg_actor_approxkl3:                        0.0007034758
+    avg_actor_clipfrac:                         0.018109132
+    avg_actor_loss:                             -0.00049708097
+    avg_advantage:                              0.010651752
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0010109805
+    avg_entropy:                                0.047988746
+    avg_grad_norm_pre_clip:                     1.1509637
+    avg_loss:                                   -3.957946e-05
+    max:
+      max_actor_approxkl1:                      0.0031200098
+      max_actor_approxkl3:                      0.0021148385
+      max_actor_clipfrac:                       0.032226562
+      max_actor_loss:                           0.0008556321
+      max_advantage:                            0.13266248
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0019567446
+      max_entropy:                              0.06417097
+      max_grad_norm_pre_clip:                   1.9234068
+      max_loss:                                 0.0009643172
+    std:
+      std_actor_approxkl1:                      0.0008714579
+      std_actor_approxkl3:                      0.00049224956
+      std_actor_clipfrac:                       0.008152183
+      std_actor_loss:                           0.0008186464
+      std_advantage:                            0.044594713
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0005570339
+      std_entropy:                              0.0052465615
+      std_grad_norm_pre_clip:                   0.31538454
+      std_loss:                                 0.0007507181
+  eval-batch-orig:
+    avg_avg_episode_length:                     17.806885
+    avg_avg_return:                             0.94102895
+    avg_benchmark_regret:                       0.0437976
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09277344
+    proxy_corner:
+      avg_avg_return:                           0.94102895
+      avg_reward_per_step:                      0.09277344
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.86244
+    avg_avg_return:                             0.27026862
+    avg_benchmark_regret:                       0.71662176
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03731537
+    proxy_corner:
+      avg_avg_return:                           0.6337687
+      avg_reward_per_step:                      0.005393982
+  eval-batch-tree:
+    avg_avg_episode_length:                     18.726158
+    avg_avg_return:                             0.9326918
+    avg_benchmark_regret:                       0.051383734
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10973358
+    proxy_corner:
+      avg_avg_return:                           0.9326918
+      avg_reward_per_step:                      0.10973358
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        123731968
+    env-step-replay-before:                     62717952
+    env-step-used-before:                       123731968
+    ppo-update-before:                          75520
+    env-step-generate-before:                   61014016
+    env-step-all-after:                         123764736
+    env-step-replay-after:                      62717952
+    env-step-used-after:                        123764736
+    ppo-update-after:                           75540
+    env-step-generate-after:                    61046784
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.458677
+    avg_avg_return:                             0.8554637
+    avg_reward_per_step:                        0.114868164
+    proxy_corner:
+      avg_avg_return:                           0.8554637
+      avg_reward_per_step:                      0.114868164
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.458677
+    avg_avg_return:                             0.8554637
+    avg_reward_per_step:                        0.114868164
+    proxy_corner:
+      avg_avg_return:                           0.8554637
+      avg_reward_per_step:                      0.114868164
+  perf:
+    env_steps_per_second:                       347828.4380949334
+    ppo_updates_per_second:                     42.23556941062358
+    eval-batch-orig-duration:                   0.5226623229682446
+    eval-batch-shift-duration:                  0.4878177037462592
+    eval-batch-tree-duration:                   0.49560865201056004
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.745605
+      mouse-cheese_dist_finite_wavg:            16.378056
+      mouse-cheese_dist_solvable_avg:           28.745605
+      mouse-cheese_dist_solvable_wavg:          16.378056
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000008
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.48122206
+    visit_patterns:
+      avg_first_visit_time:                     1645.5776
+      avg_last_visit_time:                      1904.206
+      num_replay_batches:                       1914
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0008279389
+    avg_actor_approxkl3:                        0.0005326143
+    avg_actor_clipfrac:                         0.015045166
+    avg_actor_loss:                             -0.00045272717
+    avg_advantage:                              0.004330357
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0010064746
+    avg_entropy:                                0.038688548
+    avg_grad_norm_pre_clip:                     0.8631412
+    avg_loss:                                   1.1821532e-05
+    max:
+      max_actor_approxkl1:                      0.0015151503
+      max_actor_approxkl3:                      0.0013627084
+      max_actor_clipfrac:                       0.028076172
+      max_actor_loss:                           0.0008592447
+      max_advantage:                            0.15434557
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.001930911
+      max_entropy:                              0.045700904
+      max_grad_norm_pre_clip:                   1.6879497
+      max_loss:                                 0.000996895
+    std:
+      std_actor_approxkl1:                      0.0004527439
+      std_actor_approxkl3:                      0.00029158033
+      std_actor_clipfrac:                       0.005815325
+      std_actor_loss:                           0.00085823936
+      std_advantage:                            0.0446534
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00044585805
+      std_entropy:                              0.0038005768
+      std_grad_norm_pre_clip:                   0.4116108
+      std_loss:                                 0.00076630374
+  eval-batch-orig:
+    avg_avg_episode_length:                     17.772598
+    avg_avg_return:                             0.94032484
+    avg_benchmark_regret:                       0.044501685
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09365082
+    proxy_corner:
+      avg_avg_return:                           0.94032484
+      avg_reward_per_step:                      0.09365082
+  eval-batch-shift:
+    avg_avg_episode_length:                     67.7072
+    avg_avg_return:                             0.25749728
+    avg_benchmark_regret:                       0.7293932
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.036972046
+    proxy_corner:
+      avg_avg_return:                           0.6305456
+      avg_reward_per_step:                      0.005355835
+  eval-batch-tree:
+    avg_avg_episode_length:                     18.377752
+    avg_avg_return:                             0.9333217
+    avg_benchmark_regret:                       0.050753877
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11061859
+    proxy_corner:
+      avg_avg_return:                           0.9333217
+      avg_reward_per_step:                      0.11061859
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        124780544
+    env-step-replay-before:                     63176704
+    env-step-used-before:                       124780544
+    ppo-update-before:                          76160
+    env-step-generate-before:                   61603840
+    env-step-all-after:                         124813312
+    env-step-replay-after:                      63176704
+    env-step-used-after:                        124813312
+    ppo-update-after:                           76180
+    env-step-generate-after:                    61636608
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     20.043072
+    avg_avg_return:                             0.849692
+    avg_reward_per_step:                        0.120513916
+    proxy_corner:
+      avg_avg_return:                           0.849692
+      avg_reward_per_step:                      0.120513916
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     20.043072
+    avg_avg_return:                             0.849692
+    avg_reward_per_step:                        0.120513916
+    proxy_corner:
+      avg_avg_return:                           0.849692
+      avg_reward_per_step:                      0.120513916
+  perf:
+    env_steps_per_second:                       290565.77323535166
+    ppo_updates_per_second:                     42.085350461815324
+    eval-batch-orig-duration:                   0.5450357776135206
+    eval-batch-shift-duration:                  0.49420648999512196
+    eval-batch-tree-duration:                   0.4820200949907303
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.773193
+      mouse-cheese_dist_finite_wavg:            16.375679
+      mouse-cheese_dist_solvable_avg:           28.773193
+      mouse-cheese_dist_solvable_wavg:          16.37568
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.4784425
+    visit_patterns:
+      avg_first_visit_time:                     1659.428
+      avg_last_visit_time:                      1918.3059
+      num_replay_batches:                       1928
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001050617
+    avg_actor_approxkl3:                        0.0009616466
+    avg_actor_clipfrac:                         0.021459961
+    avg_actor_loss:                             0.00034782663
+    avg_advantage:                              0.015906077
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.000935147
+    avg_entropy:                                0.05268497
+    avg_grad_norm_pre_clip:                     1.6116323
+    avg_loss:                                   0.0007627151
+    max:
+      max_actor_approxkl1:                      0.0031858857
+      max_actor_approxkl3:                      0.0034145513
+      max_actor_clipfrac:                       0.04296875
+      max_actor_loss:                           0.003277935
+      max_advantage:                            0.16834396
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.001556709
+      max_entropy:                              0.07294351
+      max_grad_norm_pre_clip:                   3.376874
+      max_loss:                                 0.0036758252
+    std:
+      std_actor_approxkl1:                      0.0009318171
+      std_actor_approxkl3:                      0.0009237131
+      std_actor_clipfrac:                       0.010702761
+      std_actor_loss:                           0.0010692647
+      std_advantage:                            0.04236807
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00046677925
+      std_entropy:                              0.009655387
+      std_grad_norm_pre_clip:                   0.7084237
+      std_loss:                                 0.0010398455
+  eval-batch-orig:
+    avg_avg_episode_length:                     18.071632
+    avg_avg_return:                             0.9375606
+    avg_benchmark_regret:                       0.047265887
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09327698
+    proxy_corner:
+      avg_avg_return:                           0.9375606
+      avg_reward_per_step:                      0.09327698
+  eval-batch-shift:
+    avg_avg_episode_length:                     67.73747
+    avg_avg_return:                             0.25534725
+    avg_benchmark_regret:                       0.7315432
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.037704468
+    proxy_corner:
+      avg_avg_return:                           0.6290499
+      avg_reward_per_step:                      0.005279541
+  eval-batch-tree:
+    avg_avg_episode_length:                     18.76712
+    avg_avg_return:                             0.92983186
+    avg_benchmark_regret:                       0.054243807
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11029053
+    proxy_corner:
+      avg_avg_return:                           0.92983186
+      avg_reward_per_step:                      0.11029053
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        125829120
+    env-step-replay-before:                     63799296
+    env-step-used-before:                       125829120
+    ppo-update-before:                          76800
+    env-step-generate-before:                   62029824
+    env-step-all-after:                         125861888
+    env-step-replay-after:                      63832064
+    env-step-used-after:                        125861888
+    ppo-update-after:                           76820
+    env-step-generate-after:                    62029824
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     62.056168
+    avg_avg_return:                             0.51068956
+    avg_reward_per_step:                        0.013793945
+    proxy_corner:
+      avg_avg_return:                           0.51068956
+      avg_reward_per_step:                      0.013793945
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     62.056168
+    avg_avg_return:                             0.51068956
+    avg_reward_per_step:                        0.013793945
+    proxy_corner:
+      avg_avg_return:                           0.51068956
+      avg_reward_per_step:                      0.013793945
+  perf:
+    env_steps_per_second:                       319675.51541891077
+    ppo_updates_per_second:                     42.31316240491719
+    eval-batch-orig-duration:                   0.5278612524271011
+    eval-batch-shift-duration:                  0.4869763543829322
+    eval-batch-tree-duration:                   0.48665353190153837
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.844482
+      mouse-cheese_dist_finite_wavg:            18.17618
+      mouse-cheese_dist_solvable_avg:           28.844482
+      mouse-cheese_dist_solvable_wavg:          18.176182
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000008
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201682
+    scoring:
+      avg_scores:                               0.4598906
+    visit_patterns:
+      avg_first_visit_time:                     1669.2253
+      avg_last_visit_time:                      1938.2339
+      num_replay_batches:                       1948
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010690114
+    avg_actor_approxkl3:                        0.0011161237
+    avg_actor_clipfrac:                         0.042907715
+    avg_actor_loss:                             -0.00033950954
+    avg_advantage:                              -0.001553777
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0063522155
+    avg_entropy:                                0.165655
+    avg_grad_norm_pre_clip:                     1.8109131
+    avg_loss:                                   0.0026709433
+    max:
+      max_actor_approxkl1:                      0.0020576096
+      max_actor_approxkl3:                      0.0020082097
+      max_actor_clipfrac:                       0.0625
+      max_actor_loss:                           0.0011806265
+      max_advantage:                            0.18764484
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.009983355
+      max_entropy:                              0.18397173
+      max_grad_norm_pre_clip:                   3.2982364
+      max_loss:                                 0.0048185056
+    std:
+      std_actor_approxkl1:                      0.0006260297
+      std_actor_approxkl3:                      0.00047390757
+      std_actor_clipfrac:                       0.014271502
+      std_actor_loss:                           0.0008745095
+      std_advantage:                            0.1125505
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011847186
+      std_entropy:                              0.010053537
+      std_grad_norm_pre_clip:                   0.64042085
+      std_loss:                                 0.0010049045
+  eval-batch-orig:
+    avg_avg_episode_length:                     17.600828
+    avg_avg_return:                             0.936995
+    avg_benchmark_regret:                       0.047831498
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09408569
+    proxy_corner:
+      avg_avg_return:                           0.936995
+      avg_reward_per_step:                      0.09408569
+  eval-batch-shift:
+    avg_avg_episode_length:                     67.7939
+    avg_avg_return:                             0.2632702
+    avg_benchmark_regret:                       0.72362024
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.035713196
+    proxy_corner:
+      avg_avg_return:                           0.5765302
+      avg_reward_per_step:                      0.005004883
+  eval-batch-tree:
+    avg_avg_episode_length:                     18.264975
+    avg_avg_return:                             0.93606484
+    avg_benchmark_regret:                       0.04801073
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11035919
+    proxy_corner:
+      avg_avg_return:                           0.93606484
+      avg_reward_per_step:                      0.11035919
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        126877696
+    env-step-replay-before:                     64356352
+    env-step-used-before:                       126877696
+    ppo-update-before:                          77440
+    env-step-generate-before:                   62521344
+    env-step-all-after:                         126910464
+    env-step-replay-after:                      64356352
+    env-step-used-after:                        126910464
+    ppo-update-after:                           77460
+    env-step-generate-after:                    62554112
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.703281
+    avg_avg_return:                             0.8461441
+    avg_reward_per_step:                        0.1015625
+    proxy_corner:
+      avg_avg_return:                           0.8461441
+      avg_reward_per_step:                      0.1015625
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.703281
+    avg_avg_return:                             0.8461441
+    avg_reward_per_step:                        0.1015625
+    proxy_corner:
+      avg_avg_return:                           0.8461441
+      avg_reward_per_step:                      0.1015625
+  perf:
+    env_steps_per_second:                       316755.73186709196
+    ppo_updates_per_second:                     42.259007046591066
+    eval-batch-orig-duration:                   0.5455142101272941
+    eval-batch-shift-duration:                  0.4876402886584401
+    eval-batch-tree-duration:                   0.488218205049634
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.855713
+      mouse-cheese_dist_finite_wavg:            18.184309
+      mouse-cheese_dist_solvable_avg:           28.855713
+      mouse-cheese_dist_solvable_wavg:          18.18431
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.44063988
+    visit_patterns:
+      avg_first_visit_time:                     1680.9724
+      avg_last_visit_time:                      1954.2588
+      num_replay_batches:                       1964
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00087437424
+    avg_actor_approxkl3:                        0.0006488983
+    avg_actor_clipfrac:                         0.01616211
+    avg_actor_loss:                             -0.0003987778
+    avg_advantage:                              0.011294997
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.001045222
+    avg_entropy:                                0.038915094
+    avg_grad_norm_pre_clip:                     0.99797153
+    avg_loss:                                   8.49182e-05
+    max:
+      max_actor_approxkl1:                      0.002574463
+      max_actor_approxkl3:                      0.0014228497
+      max_actor_clipfrac:                       0.03149414
+      max_actor_loss:                           0.0006543398
+      max_advantage:                            0.13137627
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0026745559
+      max_entropy:                              0.057803497
+      max_grad_norm_pre_clip:                   1.5681012
+      max_loss:                                 0.0016290287
+    std:
+      std_actor_approxkl1:                      0.0007206437
+      std_actor_approxkl3:                      0.000373168
+      std_actor_clipfrac:                       0.0072020865
+      std_actor_loss:                           0.00094778405
+      std_advantage:                            0.045355443
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00065194716
+      std_entropy:                              0.007424684
+      std_grad_norm_pre_clip:                   0.2952673
+      std_loss:                                 0.00087404554
+  eval-batch-orig:
+    avg_avg_episode_length:                     17.707062
+    avg_avg_return:                             0.936144
+    avg_benchmark_regret:                       0.048682455
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09398651
+    proxy_corner:
+      avg_avg_return:                           0.936144
+      avg_reward_per_step:                      0.09398651
+  eval-batch-shift:
+    avg_avg_episode_length:                     67.84863
+    avg_avg_return:                             0.25289166
+    avg_benchmark_regret:                       0.7339988
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03655243
+    proxy_corner:
+      avg_avg_return:                           0.58104694
+      avg_reward_per_step:                      0.004890442
+  eval-batch-tree:
+    avg_avg_episode_length:                     18.081493
+    avg_avg_return:                             0.9395617
+    avg_benchmark_regret:                       0.044513844
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11086273
+    proxy_corner:
+      avg_avg_return:                           0.9395617
+      avg_reward_per_step:                      0.11086273
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        127926272
+    env-step-replay-before:                     64946176
+    env-step-used-before:                       127926272
+    ppo-update-before:                          78080
+    env-step-generate-before:                   62980096
+    env-step-all-after:                         127959040
+    env-step-replay-after:                      64946176
+    env-step-used-after:                        127959040
+    ppo-update-after:                           78100
+    env-step-generate-after:                    63012864
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     21.937298
+    avg_avg_return:                             0.83079857
+    avg_reward_per_step:                        0.08111572
+    proxy_corner:
+      avg_avg_return:                           0.83079857
+      avg_reward_per_step:                      0.08111572
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     21.937298
+    avg_avg_return:                             0.83079857
+    avg_reward_per_step:                        0.08111572
+    proxy_corner:
+      avg_avg_return:                           0.83079857
+      avg_reward_per_step:                      0.08111572
+  perf:
+    env_steps_per_second:                       238331.51878535663
+    ppo_updates_per_second:                     42.22255554428323
+    eval-batch-orig-duration:                   0.523468972183764
+    eval-batch-shift-duration:                  0.4917528163641691
+    eval-batch-tree-duration:                   0.4909461997449398
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.895752
+      mouse-cheese_dist_finite_wavg:            16.394777
+      mouse-cheese_dist_solvable_avg:           28.895752
+      mouse-cheese_dist_solvable_wavg:          16.394777
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.4371683
+    visit_patterns:
+      avg_first_visit_time:                     1691.4192
+      avg_last_visit_time:                      1972.2642
+      num_replay_batches:                       1982
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011645716
+    avg_actor_approxkl3:                        0.0009858416
+    avg_actor_clipfrac:                         0.02034912
+    avg_actor_loss:                             -0.0009055907
+    avg_advantage:                              0.01704381
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0013876575
+    avg_entropy:                                0.046222042
+    avg_grad_norm_pre_clip:                     0.83421814
+    avg_loss:                                   -0.000257984
+    max:
+      max_actor_approxkl1:                      0.0069112736
+      max_actor_approxkl3:                      0.0042925426
+      max_actor_clipfrac:                       0.05407715
+      max_actor_loss:                           0.00032500364
+      max_advantage:                            0.14173466
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0027871109
+      max_entropy:                              0.062542535
+      max_grad_norm_pre_clip:                   1.3916599
+      max_loss:                                 0.0009795165
+    std:
+      std_actor_approxkl1:                      0.001685468
+      std_actor_approxkl3:                      0.0009276258
+      std_actor_clipfrac:                       0.012311146
+      std_actor_loss:                           0.0012109578
+      std_advantage:                            0.051856056
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0006348111
+      std_entropy:                              0.007053051
+      std_grad_norm_pre_clip:                   0.24733275
+      std_loss:                                 0.0012239966
+  eval-batch-orig:
+    avg_avg_episode_length:                     17.50257
+    avg_avg_return:                             0.938226
+    avg_benchmark_regret:                       0.04660049
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.0936203
+    proxy_corner:
+      avg_avg_return:                           0.938226
+      avg_reward_per_step:                      0.0936203
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.758934
+    avg_avg_return:                             0.27802795
+    avg_benchmark_regret:                       0.70886254
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.036956787
+    proxy_corner:
+      avg_avg_return:                           0.6120274
+      avg_reward_per_step:                      0.005393982
+  eval-batch-tree:
+    avg_avg_episode_length:                     18.240898
+    avg_avg_return:                             0.9364401
+    avg_benchmark_regret:                       0.04763551
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11048126
+    proxy_corner:
+      avg_avg_return:                           0.9364401
+      avg_reward_per_step:                      0.11048126
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        128974848
+    env-step-replay-before:                     65601536
+    env-step-used-before:                       128974848
+    ppo-update-before:                          78720
+    env-step-generate-before:                   63373312
+    env-step-all-after:                         129007616
+    env-step-replay-after:                      65601536
+    env-step-used-after:                        129007616
+    ppo-update-after:                           78740
+    env-step-generate-after:                    63406080
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.985527
+    avg_avg_return:                             0.85095716
+    avg_reward_per_step:                        0.097961426
+    proxy_corner:
+      avg_avg_return:                           0.85095716
+      avg_reward_per_step:                      0.097961426
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.985527
+    avg_avg_return:                             0.85095716
+    avg_reward_per_step:                        0.097961426
+    proxy_corner:
+      avg_avg_return:                           0.85095716
+      avg_reward_per_step:                      0.097961426
+  perf:
+    env_steps_per_second:                       346620.27125145285
+    ppo_updates_per_second:                     42.1800163113673
+    eval-batch-orig-duration:                   0.4788103671744466
+    eval-batch-shift-duration:                  0.4229250494390726
+    eval-batch-tree-duration:                   0.4207283901050687
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.90039
+      mouse-cheese_dist_finite_wavg:            16.402372
+      mouse-cheese_dist_solvable_avg:           28.90039
+      mouse-cheese_dist_solvable_wavg:          16.402374
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.43313104
+    visit_patterns:
+      avg_first_visit_time:                     1701.7839
+      avg_last_visit_time:                      1992.2993
+      num_replay_batches:                       2002
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0005201569
+    avg_actor_approxkl3:                        0.0006170238
+    avg_actor_clipfrac:                         0.013769532
+    avg_actor_loss:                             -0.000117841366
+    avg_advantage:                              0.008242486
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0009202817
+    avg_entropy:                                0.038205806
+    avg_grad_norm_pre_clip:                     1.0568068
+    avg_loss:                                   0.00030409367
+    max:
+      max_actor_approxkl1:                      0.002329697
+      max_actor_approxkl3:                      0.0017226713
+      max_actor_clipfrac:                       0.024536133
+      max_actor_loss:                           0.0014384314
+      max_advantage:                            0.12731981
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0020987422
+      max_entropy:                              0.050452523
+      max_grad_norm_pre_clip:                   1.7857734
+      max_loss:                                 0.0015457696
+    std:
+      std_actor_approxkl1:                      0.0006315969
+      std_actor_approxkl3:                      0.0003952302
+      std_actor_clipfrac:                       0.005158045
+      std_actor_loss:                           0.00077861187
+      std_advantage:                            0.041609786
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0005427464
+      std_entropy:                              0.006056296
+      std_grad_norm_pre_clip:                   0.40439063
+      std_loss:                                 0.0006738491
+  eval-batch-orig:
+    avg_avg_episode_length:                     17.148949
+    avg_avg_return:                             0.94335496
+    avg_benchmark_regret:                       0.04147156
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09435272
+    proxy_corner:
+      avg_avg_return:                           0.94335496
+      avg_reward_per_step:                      0.09435272
+  eval-batch-shift:
+    avg_avg_episode_length:                     67.76907
+    avg_avg_return:                             0.25416076
+    avg_benchmark_regret:                       0.7327297
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03605652
+    proxy_corner:
+      avg_avg_return:                           0.6242987
+      avg_reward_per_step:                      0.0053634644
+  eval-batch-tree:
+    avg_avg_episode_length:                     18.141674
+    avg_avg_return:                             0.9367007
+    avg_benchmark_regret:                       0.047374893
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11063385
+    proxy_corner:
+      avg_avg_return:                           0.9367007
+      avg_reward_per_step:                      0.11063385
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        130023424
+    env-step-replay-before:                     66158592
+    env-step-used-before:                       130023424
+    ppo-update-before:                          79360
+    env-step-generate-before:                   63864832
+    env-step-all-after:                         130056192
+    env-step-replay-after:                      66158592
+    env-step-used-after:                        130056192
+    ppo-update-after:                           79380
+    env-step-generate-after:                    63897600
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.472736
+    avg_avg_return:                             0.85289955
+    avg_reward_per_step:                        0.08480835
+    proxy_corner:
+      avg_avg_return:                           0.85289955
+      avg_reward_per_step:                      0.08480835
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.472736
+    avg_avg_return:                             0.85289955
+    avg_reward_per_step:                        0.08480835
+    proxy_corner:
+      avg_avg_return:                           0.85289955
+      avg_reward_per_step:                      0.08480835
+  perf:
+    env_steps_per_second:                       319757.3207996284
+    ppo_updates_per_second:                     44.541206800785346
+    eval-batch-orig-duration:                   0.5177464950829744
+    eval-batch-shift-duration:                  0.48920847196131945
+    eval-batch-tree-duration:                   0.4917361969128251
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.882568
+      mouse-cheese_dist_finite_wavg:            17.29226
+      mouse-cheese_dist_solvable_avg:           28.882568
+      mouse-cheese_dist_solvable_wavg:          17.292257
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999985
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.420168
+    scoring:
+      avg_scores:                               0.41941926
+    visit_patterns:
+      avg_first_visit_time:                     1715.1147
+      avg_last_visit_time:                      2009.2478
+      num_replay_batches:                       2019
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0003186647
+    avg_actor_approxkl3:                        0.00049988955
+    avg_actor_clipfrac:                         0.013568115
+    avg_actor_loss:                             -0.0002956867
+    avg_advantage:                              -0.0012107284
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0005552155
+    avg_entropy:                                0.038380846
+    avg_grad_norm_pre_clip:                     0.8526334
+    avg_loss:                                   -5.6459783e-05
+    max:
+      max_actor_approxkl1:                      0.0011951013
+      max_actor_approxkl3:                      0.0012156244
+      max_actor_clipfrac:                       0.024047852
+      max_actor_loss:                           0.001038488
+      max_advantage:                            0.11107379
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0013962919
+      max_entropy:                              0.04736958
+      max_grad_norm_pre_clip:                   1.5265908
+      max_loss:                                 0.001148424
+    std:
+      std_actor_approxkl1:                      0.00044494474
+      std_actor_approxkl3:                      0.000270687
+      std_actor_clipfrac:                       0.0054226257
+      std_actor_loss:                           0.00065672194
+      std_advantage:                            0.033691883
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00033128244
+      std_entropy:                              0.0046457364
+      std_grad_norm_pre_clip:                   0.3133619
+      std_loss:                                 0.00064231677
+  eval-batch-orig:
+    avg_avg_episode_length:                     17.320255
+    avg_avg_return:                             0.9418965
+    avg_benchmark_regret:                       0.042930022
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.094078064
+    proxy_corner:
+      avg_avg_return:                           0.9418965
+      avg_reward_per_step:                      0.094078064
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.187744
+    avg_avg_return:                             0.28403175
+    avg_benchmark_regret:                       0.7028587
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03791809
+    proxy_corner:
+      avg_avg_return:                           0.61159396
+      avg_reward_per_step:                      0.0054092407
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.994919
+    avg_avg_return:                             0.94035274
+    avg_benchmark_regret:                       0.043722898
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11084747
+    proxy_corner:
+      avg_avg_return:                           0.94035274
+      avg_reward_per_step:                      0.11084747
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        131072000
+    env-step-replay-before:                     66551808
+    env-step-used-before:                       131072000
+    ppo-update-before:                          80000
+    env-step-generate-before:                   64520192
+    env-step-all-after:                         131104768
+    env-step-replay-after:                      66551808
+    env-step-used-after:                        131104768
+    ppo-update-after:                           80020
+    env-step-generate-after:                    64552960
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     20.8897
+    avg_avg_return:                             0.842402
+    avg_reward_per_step:                        0.090911865
+    proxy_corner:
+      avg_avg_return:                           0.842402
+      avg_reward_per_step:                      0.090911865
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     20.8897
+    avg_avg_return:                             0.842402
+    avg_reward_per_step:                        0.090911865
+    proxy_corner:
+      avg_avg_return:                           0.842402
+      avg_reward_per_step:                      0.090911865
+  perf:
+    env_steps_per_second:                       291087.32523957966
+    ppo_updates_per_second:                     42.10522595868531
+    eval-batch-orig-duration:                   0.5566479926928878
+    eval-batch-shift-duration:                  0.48825670778751373
+    eval-batch-tree-duration:                   0.4882897138595581
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             28.961426
+      mouse-cheese_dist_finite_wavg:            13.711678
+      mouse-cheese_dist_solvable_avg:           28.961426
+      mouse-cheese_dist_solvable_wavg:          13.711678
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016816
+    scoring:
+      avg_scores:                               0.43364963
+    visit_patterns:
+      avg_first_visit_time:                     1732.083
+      avg_last_visit_time:                      2021.2803
+      num_replay_batches:                       2031
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00086834835
+    avg_actor_approxkl3:                        0.0007284729
+    avg_actor_clipfrac:                         0.016992187
+    avg_actor_loss:                             -0.000102626116
+    avg_advantage:                              0.0059346487
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0010009018
+    avg_entropy:                                0.038591538
+    avg_grad_norm_pre_clip:                     1.0991464
+    avg_loss:                                   0.0003592332
+    max:
+      max_actor_approxkl1:                      0.0022006421
+      max_actor_approxkl3:                      0.0018454092
+      max_actor_clipfrac:                       0.03149414
+      max_actor_loss:                           0.0023295283
+      max_advantage:                            0.111373186
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0025281953
+      max_entropy:                              0.051177926
+      max_grad_norm_pre_clip:                   2.1642883
+      max_loss:                                 0.002726278
+    std:
+      std_actor_approxkl1:                      0.0006915891
+      std_actor_approxkl3:                      0.0004212678
+      std_actor_clipfrac:                       0.0071607106
+      std_actor_loss:                           0.0009712731
+      std_advantage:                            0.04471994
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0006108475
+      std_entropy:                              0.006234987
+      std_grad_norm_pre_clip:                   0.42979735
+      std_loss:                                 0.00089107605
+  eval-batch-orig:
+    avg_avg_episode_length:                     17.389622
+    avg_avg_return:                             0.94156456
+    avg_benchmark_regret:                       0.043261986
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.094680786
+    proxy_corner:
+      avg_avg_return:                           0.94156456
+      avg_reward_per_step:                      0.094680786
+  eval-batch-shift:
+    avg_avg_episode_length:                     65.90358
+    avg_avg_return:                             0.28444552
+    avg_benchmark_regret:                       0.7024449
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.039497375
+    proxy_corner:
+      avg_avg_return:                           0.60571766
+      avg_reward_per_step:                      0.0053253174
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.855473
+    avg_avg_return:                             0.94082236
+    avg_benchmark_regret:                       0.04325322
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11118317
+    proxy_corner:
+      avg_avg_return:                           0.94082236
+      avg_reward_per_step:                      0.11118317
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        132120576
+    env-step-replay-before:                     67076096
+    env-step-used-before:                       132120576
+    ppo-update-before:                          80640
+    env-step-generate-before:                   65044480
+    env-step-all-after:                         132153344
+    env-step-replay-after:                      67076096
+    env-step-used-after:                        132153344
+    ppo-update-after:                           80660
+    env-step-generate-after:                    65077248
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.518261
+    avg_avg_return:                             0.84894073
+    avg_reward_per_step:                        0.092803955
+    proxy_corner:
+      avg_avg_return:                           0.84894073
+      avg_reward_per_step:                      0.092803955
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.518261
+    avg_avg_return:                             0.84894073
+    avg_reward_per_step:                        0.092803955
+    proxy_corner:
+      avg_avg_return:                           0.84894073
+      avg_reward_per_step:                      0.092803955
+  perf:
+    env_steps_per_second:                       325102.3791749513
+    ppo_updates_per_second:                     40.953107566250935
+    eval-batch-orig-duration:                   0.5338502544909716
+    eval-batch-shift-duration:                  0.49040045589208603
+    eval-batch-tree-duration:                   0.47345935367047787
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.019287
+      mouse-cheese_dist_finite_wavg:            16.390594
+      mouse-cheese_dist_solvable_avg:           29.019287
+      mouse-cheese_dist_solvable_wavg:          16.390596
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.4269248
+    visit_patterns:
+      avg_first_visit_time:                     1744.7063
+      avg_last_visit_time:                      2037.2466
+      num_replay_batches:                       2047
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009519504
+    avg_actor_approxkl3:                        0.00087545736
+    avg_actor_clipfrac:                         0.015905762
+    avg_actor_loss:                             0.00030347586
+    avg_advantage:                              0.004441929
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0009471401
+    avg_entropy:                                0.03721068
+    avg_grad_norm_pre_clip:                     1.1814841
+    avg_loss:                                   0.0007398352
+    max:
+      max_actor_approxkl1:                      0.003911976
+      max_actor_approxkl3:                      0.002428752
+      max_actor_clipfrac:                       0.030761719
+      max_actor_loss:                           0.0016535651
+      max_advantage:                            0.13320959
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.002396616
+      max_entropy:                              0.052422605
+      max_grad_norm_pre_clip:                   2.9492235
+      max_loss:                                 0.0022584975
+    std:
+      std_actor_approxkl1:                      0.0009995563
+      std_actor_approxkl3:                      0.00068110967
+      std_actor_clipfrac:                       0.0074547995
+      std_actor_loss:                           0.0006533124
+      std_advantage:                            0.043383345
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00061599445
+      std_entropy:                              0.007838875
+      std_grad_norm_pre_clip:                   0.53162795
+      std_loss:                                 0.0006485177
+  eval-batch-orig:
+    avg_avg_episode_length:                     17.447119
+    avg_avg_return:                             0.94173247
+    avg_benchmark_regret:                       0.043094076
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09439087
+    proxy_corner:
+      avg_avg_return:                           0.94173247
+      avg_reward_per_step:                      0.09439087
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.793274
+    avg_avg_return:                             0.26782474
+    avg_benchmark_regret:                       0.71906567
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.038024902
+    proxy_corner:
+      avg_avg_return:                           0.608762
+      avg_reward_per_step:                      0.0051651
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.690844
+    avg_avg_return:                             0.94141865
+    avg_benchmark_regret:                       0.04265698
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.111595154
+    proxy_corner:
+      avg_avg_return:                           0.94141865
+      avg_reward_per_step:                      0.111595154
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        133169152
+    env-step-replay-before:                     67600384
+    env-step-used-before:                       133169152
+    ppo-update-before:                          81280
+    env-step-generate-before:                   65568768
+    env-step-all-after:                         133201920
+    env-step-replay-after:                      67633152
+    env-step-used-after:                        133201920
+    ppo-update-after:                           81300
+    env-step-generate-after:                    65568768
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.104607
+    avg_avg_return:                             0.5373986
+    avg_reward_per_step:                        0.014984131
+    proxy_corner:
+      avg_avg_return:                           0.5373986
+      avg_reward_per_step:                      0.014984131
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.104607
+    avg_avg_return:                             0.5373986
+    avg_reward_per_step:                        0.014984131
+    proxy_corner:
+      avg_avg_return:                           0.5373986
+      avg_reward_per_step:                      0.014984131
+  perf:
+    env_steps_per_second:                       261450.67452603576
+    ppo_updates_per_second:                     42.21064045592698
+    eval-batch-orig-duration:                   0.6055479720234871
+    eval-batch-shift-duration:                  0.5897335428744555
+    eval-batch-tree-duration:                   0.5617926055565476
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.20044
+      mouse-cheese_dist_finite_wavg:            19.119526
+      mouse-cheese_dist_solvable_avg:           29.20044
+      mouse-cheese_dist_solvable_wavg:          19.119528
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.44533676
+    visit_patterns:
+      avg_first_visit_time:                     1755.769
+      avg_last_visit_time:                      2054.2944
+      num_replay_batches:                       2064
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001393605
+    avg_actor_approxkl3:                        0.0013592228
+    avg_actor_clipfrac:                         0.049401857
+    avg_actor_loss:                             -9.6849726e-05
+    avg_advantage:                              -0.0058541168
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0057098893
+    avg_entropy:                                0.15351412
+    avg_grad_norm_pre_clip:                     1.8952173
+    avg_loss:                                   0.0026045807
+    max:
+      max_actor_approxkl1:                      0.0031804417
+      max_actor_approxkl3:                      0.0025895857
+      max_actor_clipfrac:                       0.07458496
+      max_actor_loss:                           0.001784198
+      max_advantage:                            0.14995098
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.008009587
+      max_entropy:                              0.16605005
+      max_grad_norm_pre_clip:                   2.7330663
+      max_loss:                                 0.0047052237
+    std:
+      std_actor_approxkl1:                      0.0008023713
+      std_actor_approxkl3:                      0.0005711737
+      std_actor_clipfrac:                       0.015987813
+      std_actor_loss:                           0.0007416389
+      std_advantage:                            0.106959864
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010785332
+      std_entropy:                              0.008516223
+      std_grad_norm_pre_clip:                   0.4567672
+      std_loss:                                 0.00073846284
+  eval-batch-orig:
+    avg_avg_episode_length:                     17.63675
+    avg_avg_return:                             0.9383099
+    avg_benchmark_regret:                       0.04651662
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.0940094
+    proxy_corner:
+      avg_avg_return:                           0.9383099
+      avg_reward_per_step:                      0.0940094
+  eval-batch-shift:
+    avg_avg_episode_length:                     65.37657
+    avg_avg_return:                             0.29189932
+    avg_benchmark_regret:                       0.6949911
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03768921
+    proxy_corner:
+      avg_avg_return:                           0.5986594
+      avg_reward_per_step:                      0.0055160522
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.919163
+    avg_avg_return:                             0.93822443
+    avg_benchmark_regret:                       0.045851156
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11014557
+    proxy_corner:
+      avg_avg_return:                           0.93822443
+      avg_reward_per_step:                      0.11014557
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        134217728
+    env-step-replay-before:                     68157440
+    env-step-used-before:                       134217728
+    ppo-update-before:                          81920
+    env-step-generate-before:                   66060288
+    env-step-all-after:                         134250496
+    env-step-replay-after:                      68157440
+    env-step-used-after:                        134250496
+    ppo-update-after:                           81940
+    env-step-generate-after:                    66093056
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.891207
+    avg_avg_return:                             0.85540617
+    avg_reward_per_step:                        0.10787964
+    proxy_corner:
+      avg_avg_return:                           0.85540617
+      avg_reward_per_step:                      0.10787964
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.891207
+    avg_avg_return:                             0.85540617
+    avg_reward_per_step:                        0.10787964
+    proxy_corner:
+      avg_avg_return:                           0.85540617
+      avg_reward_per_step:                      0.10787964
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  perf:
+    env_steps_per_second:                       320514.05823643506
+    ppo_updates_per_second:                     42.06485925730365
+    eval-batch-orig-duration:                   0.643802291713655
+    eval-batch-shift-duration:                  0.5898267943412066
+    eval-batch-tree-duration:                   0.6003334950655699
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.332031
+      mouse-cheese_dist_finite_wavg:            15.525724
+      mouse-cheese_dist_solvable_avg:           29.332031
+      mouse-cheese_dist_solvable_wavg:          15.525724
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.41958696
+    visit_patterns:
+      avg_first_visit_time:                     1769.1362
+      avg_last_visit_time:                      2070.2922
+      num_replay_batches:                       2080
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00081801397
+    avg_actor_approxkl3:                        0.0006148223
+    avg_actor_clipfrac:                         0.017102052
+    avg_actor_loss:                             -0.00032832669
+    avg_advantage:                              0.01655811
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0009095411
+    avg_entropy:                                0.042794816
+    avg_grad_norm_pre_clip:                     1.02432
+    avg_loss:                                   8.3648985e-05
+    max:
+      max_actor_approxkl1:                      0.0017362805
+      max_actor_approxkl3:                      0.0011463964
+      max_actor_clipfrac:                       0.032592773
+      max_actor_loss:                           0.0010866467
+      max_advantage:                            0.13729918
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.002496335
+      max_entropy:                              0.056100257
+      max_grad_norm_pre_clip:                   1.9355612
+      max_loss:                                 0.001828637
+    std:
+      std_actor_approxkl1:                      0.0005211153
+      std_actor_approxkl3:                      0.00025894132
+      std_actor_clipfrac:                       0.0066427905
+      std_actor_loss:                           0.000767346
+      std_advantage:                            0.040774465
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0006558492
+      std_entropy:                              0.006788917
+      std_grad_norm_pre_clip:                   0.37978032
+      std_loss:                                 0.0007706852
+  eval-batch-orig:
+    avg_avg_episode_length:                     17.643297
+    avg_avg_return:                             0.9377283
+    avg_benchmark_regret:                       0.047098227
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09436798
+    proxy_corner:
+      avg_avg_return:                           0.9377283
+      avg_reward_per_step:                      0.09436798
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.3911
+    avg_avg_return:                             0.27577338
+    avg_benchmark_regret:                       0.7111171
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.038330078
+    proxy_corner:
+      avg_avg_return:                           0.6256528
+      avg_reward_per_step:                      0.005393982
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-tree:
+    avg_avg_episode_length:                     18.052505
+    avg_avg_return:                             0.9370699
+    avg_benchmark_regret:                       0.047005713
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11130524
+    proxy_corner:
+      avg_avg_return:                           0.9370699
+      avg_reward_per_step:                      0.11130524
+    rollouts_gif:                               float32(132, 225, 225, 3)
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        135266304
+    env-step-replay-before:                     68780032
+    env-step-used-before:                       135266304
+    ppo-update-before:                          82560
+    env-step-generate-before:                   66486272
+    env-step-all-after:                         135299072
+    env-step-replay-after:                      68780032
+    env-step-used-after:                        135299072
+    ppo-update-after:                           82580
+    env-step-generate-after:                    66519040
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.159477
+    avg_avg_return:                             0.8570327
+    avg_reward_per_step:                        0.10501099
+    proxy_corner:
+      avg_avg_return:                           0.8570327
+      avg_reward_per_step:                      0.10501099
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.159477
+    avg_avg_return:                             0.8570327
+    avg_reward_per_step:                        0.10501099
+    proxy_corner:
+      avg_avg_return:                           0.8570327
+      avg_reward_per_step:                      0.10501099
+  perf:
+    env_steps_per_second:                       313137.33196119446
+    ppo_updates_per_second:                     42.17950555775931
+    eval-batch-orig-duration:                   0.5418596472591162
+    eval-batch-shift-duration:                  0.49508462473750114
+    eval-batch-tree-duration:                   0.4878753572702408
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.466309
+      mouse-cheese_dist_finite_wavg:            13.755022
+      mouse-cheese_dist_solvable_avg:           29.466309
+      mouse-cheese_dist_solvable_wavg:          13.755022
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016795
+    scoring:
+      avg_scores:                               0.4121626
+    visit_patterns:
+      avg_first_visit_time:                     1781.9011
+      avg_last_visit_time:                      2089.2537
+      num_replay_batches:                       2099
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0002841679
+    avg_actor_approxkl3:                        0.00040333864
+    avg_actor_clipfrac:                         0.011383057
+    avg_actor_loss:                             2.6120759e-05
+    avg_advantage:                              -0.008098826
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0008647816
+    avg_entropy:                                0.03315064
+    avg_grad_norm_pre_clip:                     0.87245476
+    avg_loss:                                   0.00042536095
+    max:
+      max_actor_approxkl1:                      0.0010391871
+      max_actor_approxkl3:                      0.00093823986
+      max_actor_clipfrac:                       0.020507812
+      max_actor_loss:                           0.0025459006
+      max_advantage:                            0.116350174
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0020087906
+      max_entropy:                              0.045437127
+      max_grad_norm_pre_clip:                   2.0656195
+      max_loss:                                 0.002928988
+    std:
+      std_actor_approxkl1:                      0.00032202926
+      std_actor_approxkl3:                      0.00022406554
+      std_actor_clipfrac:                       0.004616756
+      std_actor_loss:                           0.0008280746
+      std_advantage:                            0.041229486
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00045301323
+      std_entropy:                              0.0046305303
+      std_grad_norm_pre_clip:                   0.48167896
+      std_loss:                                 0.0008057235
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.91567
+    avg_avg_return:                             0.9475241
+    avg_benchmark_regret:                       0.0373024
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.0946579
+    proxy_corner:
+      avg_avg_return:                           0.9475241
+      avg_reward_per_step:                      0.0946579
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.659485
+    avg_avg_return:                             0.27237555
+    avg_benchmark_regret:                       0.71451485
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.0365448
+    proxy_corner:
+      avg_avg_return:                           0.62816226
+      avg_reward_per_step:                      0.0055770874
+  eval-batch-tree:
+    avg_avg_episode_length:                     18.002325
+    avg_avg_return:                             0.937691
+    avg_benchmark_regret:                       0.046384636
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11117554
+    proxy_corner:
+      avg_avg_return:                           0.937691
+      avg_reward_per_step:                      0.11117554
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        136314880
+    env-step-replay-before:                     69107712
+    env-step-used-before:                       136314880
+    ppo-update-before:                          83200
+    env-step-generate-before:                   67207168
+    env-step-all-after:                         136347648
+    env-step-replay-after:                      69107712
+    env-step-used-after:                        136347648
+    ppo-update-after:                           83220
+    env-step-generate-after:                    67239936
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.759735
+    avg_avg_return:                             0.85613894
+    avg_reward_per_step:                        0.13137817
+    proxy_corner:
+      avg_avg_return:                           0.85613894
+      avg_reward_per_step:                      0.13137817
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.759735
+    avg_avg_return:                             0.85613894
+    avg_reward_per_step:                        0.13137817
+    proxy_corner:
+      avg_avg_return:                           0.85613894
+      avg_reward_per_step:                      0.13137817
+  perf:
+    env_steps_per_second:                       297242.1283378485
+    ppo_updates_per_second:                     42.15939327799076
+    eval-batch-orig-duration:                   0.5227320538833737
+    eval-batch-shift-duration:                  0.48429184779524803
+    eval-batch-tree-duration:                   0.45784584898501635
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.37378
+      mouse-cheese_dist_finite_wavg:            15.544075
+      mouse-cheese_dist_solvable_avg:           29.37378
+      mouse-cheese_dist_solvable_wavg:          15.544077
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.99999
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.41762576
+    visit_patterns:
+      avg_first_visit_time:                     1795.6726
+      avg_last_visit_time:                      2099.3555
+      num_replay_batches:                       2109
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00083286146
+    avg_actor_approxkl3:                        0.00091085647
+    avg_actor_clipfrac:                         0.014544678
+    avg_actor_loss:                             0.0003774578
+    avg_advantage:                              0.002890515
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0009177071
+    avg_entropy:                                0.035515532
+    avg_grad_norm_pre_clip:                     1.2073299
+    avg_loss:                                   0.0008007958
+    max:
+      max_actor_approxkl1:                      0.0030327812
+      max_actor_approxkl3:                      0.0036274348
+      max_actor_clipfrac:                       0.0357666
+      max_actor_loss:                           0.0029189251
+      max_advantage:                            0.107626975
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0019529857
+      max_entropy:                              0.052907977
+      max_grad_norm_pre_clip:                   2.2634218
+      max_loss:                                 0.0032599769
+    std:
+      std_actor_approxkl1:                      0.0009164759
+      std_actor_approxkl3:                      0.0009941673
+      std_actor_clipfrac:                       0.008394255
+      std_actor_loss:                           0.00093505863
+      std_advantage:                            0.043010168
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00049587665
+      std_entropy:                              0.006996691
+      std_grad_norm_pre_clip:                   0.5857988
+      std_loss:                                 0.0009410895
+  eval-batch-orig:
+    avg_avg_episode_length:                     17.536352
+    avg_avg_return:                             0.94038075
+    avg_benchmark_regret:                       0.0444458
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09436035
+    proxy_corner:
+      avg_avg_return:                           0.94038075
+      avg_reward_per_step:                      0.09436035
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.93765
+    avg_avg_return:                             0.26722658
+    avg_benchmark_regret:                       0.71966386
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.037094116
+    proxy_corner:
+      avg_avg_return:                           0.6397116
+      avg_reward_per_step:                      0.0055770874
+  eval-batch-tree:
+    avg_avg_episode_length:                     18.224724
+    avg_avg_return:                             0.9379404
+    avg_benchmark_regret:                       0.046135183
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11016846
+    proxy_corner:
+      avg_avg_return:                           0.9379404
+      avg_reward_per_step:                      0.11016846
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        137363456
+    env-step-replay-before:                     69664768
+    env-step-used-before:                       137363456
+    ppo-update-before:                          83840
+    env-step-generate-before:                   67698688
+    env-step-all-after:                         137396224
+    env-step-replay-after:                      69697536
+    env-step-used-after:                        137396224
+    ppo-update-after:                           83860
+    env-step-generate-after:                    67698688
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.165535
+    avg_avg_return:                             0.5198468
+    avg_reward_per_step:                        0.014984131
+    proxy_corner:
+      avg_avg_return:                           0.5198468
+      avg_reward_per_step:                      0.014984131
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.165535
+    avg_avg_return:                             0.5198468
+    avg_reward_per_step:                        0.014984131
+    proxy_corner:
+      avg_avg_return:                           0.5198468
+      avg_reward_per_step:                      0.014984131
+  perf:
+    env_steps_per_second:                       291180.9051635791
+    ppo_updates_per_second:                     42.20583736666163
+    eval-batch-orig-duration:                   0.5329427821561694
+    eval-batch-shift-duration:                  0.4891232792288065
+    eval-batch-tree-duration:                   0.48636909294873476
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.428467
+      mouse-cheese_dist_finite_wavg:            19.145197
+      mouse-cheese_dist_solvable_avg:           29.428467
+      mouse-cheese_dist_solvable_wavg:          19.145199
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.42489576
+    visit_patterns:
+      avg_first_visit_time:                     1808.0554
+      avg_last_visit_time:                      2117.3088
+      num_replay_batches:                       2127
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010808659
+    avg_actor_approxkl3:                        0.0008589709
+    avg_actor_clipfrac:                         0.035803225
+    avg_actor_loss:                             -0.0010591313
+    avg_advantage:                              -0.019936798
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.006847396
+    avg_entropy:                                0.15006784
+    avg_grad_norm_pre_clip:                     1.2509533
+    avg_loss:                                   0.0022144993
+    max:
+      max_actor_approxkl1:                      0.0030191112
+      max_actor_approxkl3:                      0.00198866
+      max_actor_clipfrac:                       0.068359375
+      max_actor_loss:                           0.00034295022
+      max_advantage:                            0.12790102
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.008314305
+      max_entropy:                              0.17208627
+      max_grad_norm_pre_clip:                   2.0460854
+      max_loss:                                 0.004039248
+    std:
+      std_actor_approxkl1:                      0.00071617233
+      std_actor_approxkl3:                      0.00046251438
+      std_actor_clipfrac:                       0.015421792
+      std_actor_loss:                           0.0008084132
+      std_advantage:                            0.116979994
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0008774406
+      std_entropy:                              0.01070172
+      std_grad_norm_pre_clip:                   0.40254635
+      std_loss:                                 0.00095176214
+  eval-batch-orig:
+    avg_avg_episode_length:                     17.18987
+    avg_avg_return:                             0.94240963
+    avg_benchmark_regret:                       0.04241692
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09461212
+    proxy_corner:
+      avg_avg_return:                           0.94240963
+      avg_reward_per_step:                      0.09461212
+  eval-batch-shift:
+    avg_avg_episode_length:                     68.26726
+    avg_avg_return:                             0.25312215
+    avg_benchmark_regret:                       0.7337683
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.035057068
+    proxy_corner:
+      avg_avg_return:                           0.6392314
+      avg_reward_per_step:                      0.0054626465
+  eval-batch-tree:
+    avg_avg_episode_length:                     18.174223
+    avg_avg_return:                             0.9356873
+    avg_benchmark_regret:                       0.048388258
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11122131
+    proxy_corner:
+      avg_avg_return:                           0.9356873
+      avg_reward_per_step:                      0.11122131
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        138412032
+    env-step-replay-before:                     70090752
+    env-step-used-before:                       138412032
+    ppo-update-before:                          84480
+    env-step-generate-before:                   68321280
+    env-step-all-after:                         138444800
+    env-step-replay-after:                      70123520
+    env-step-used-after:                        138444800
+    ppo-update-after:                           84500
+    env-step-generate-after:                    68321280
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     56.511986
+    avg_avg_return:                             0.5545671
+    avg_reward_per_step:                        0.015106201
+    proxy_corner:
+      avg_avg_return:                           0.5545671
+      avg_reward_per_step:                      0.015106201
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     56.511986
+    avg_avg_return:                             0.5545671
+    avg_reward_per_step:                        0.015106201
+    proxy_corner:
+      avg_avg_return:                           0.5545671
+      avg_reward_per_step:                      0.015106201
+  perf:
+    env_steps_per_second:                       331583.0764331338
+    ppo_updates_per_second:                     42.17529548065162
+    eval-batch-orig-duration:                   0.5303544299677014
+    eval-batch-shift-duration:                  0.48446526750922203
+    eval-batch-tree-duration:                   0.47485340759158134
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.44629
+      mouse-cheese_dist_finite_wavg:            13.739262
+      mouse-cheese_dist_solvable_avg:           29.44629
+      mouse-cheese_dist_solvable_wavg:          13.739264
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.44068372
+    visit_patterns:
+      avg_first_visit_time:                     1824.0063
+      avg_last_visit_time:                      2130.393
+      num_replay_batches:                       2140
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0015465978
+    avg_actor_approxkl3:                        0.0014774596
+    avg_actor_clipfrac:                         0.05087891
+    avg_actor_loss:                             -0.00018752702
+    avg_advantage:                              -0.0037077002
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.005241984
+    avg_entropy:                                0.14907756
+    avg_grad_norm_pre_clip:                     1.807003
+    avg_loss:                                   0.0022843876
+    max:
+      max_actor_approxkl1:                      0.0037680748
+      max_actor_approxkl3:                      0.002716883
+      max_actor_clipfrac:                       0.08093262
+      max_actor_loss:                           0.002238214
+      max_advantage:                            0.15747565
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0072046355
+      max_entropy:                              0.1670945
+      max_grad_norm_pre_clip:                   2.7299924
+      max_loss:                                 0.0044849855
+    std:
+      std_actor_approxkl1:                      0.0008283532
+      std_actor_approxkl3:                      0.00057318044
+      std_actor_clipfrac:                       0.017100766
+      std_actor_loss:                           0.0011821168
+      std_advantage:                            0.1021012
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.000951767
+      std_entropy:                              0.010473741
+      std_grad_norm_pre_clip:                   0.5535796
+      std_loss:                                 0.001132683
+  eval-batch-orig:
+    avg_avg_episode_length:                     17.009487
+    avg_avg_return:                             0.9451058
+    avg_benchmark_regret:                       0.039720714
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09455109
+    proxy_corner:
+      avg_avg_return:                           0.9451058
+      avg_reward_per_step:                      0.09455109
+  eval-batch-shift:
+    avg_avg_episode_length:                     64.353455
+    avg_avg_return:                             0.30393517
+    avg_benchmark_regret:                       0.6829553
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03942108
+    proxy_corner:
+      avg_avg_return:                           0.59719265
+      avg_reward_per_step:                      0.00579834
+  eval-batch-tree:
+    avg_avg_episode_length:                     18.489805
+    avg_avg_return:                             0.935424
+    avg_benchmark_regret:                       0.048651624
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.10863495
+    proxy_corner:
+      avg_avg_return:                           0.935424
+      avg_reward_per_step:                      0.10863495
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        139460608
+    env-step-replay-before:                     70713344
+    env-step-used-before:                       139460608
+    ppo-update-before:                          85120
+    env-step-generate-before:                   68747264
+    env-step-all-after:                         139493376
+    env-step-replay-after:                      70713344
+    env-step-used-after:                        139493376
+    ppo-update-after:                           85140
+    env-step-generate-after:                    68780032
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.6198
+    avg_avg_return:                             0.851063
+    avg_reward_per_step:                        0.11224365
+    proxy_corner:
+      avg_avg_return:                           0.851063
+      avg_reward_per_step:                      0.11224365
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.6198
+    avg_avg_return:                             0.851063
+    avg_reward_per_step:                        0.11224365
+    proxy_corner:
+      avg_avg_return:                           0.851063
+      avg_reward_per_step:                      0.11224365
+  perf:
+    env_steps_per_second:                       355236.26272093115
+    ppo_updates_per_second:                     40.99002511834421
+    eval-batch-orig-duration:                   0.5240622041746974
+    eval-batch-shift-duration:                  0.48278920352458954
+    eval-batch-tree-duration:                   0.46166471391916275
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.599121
+      mouse-cheese_dist_finite_wavg:            18.247488
+      mouse-cheese_dist_solvable_avg:           29.599121
+      mouse-cheese_dist_solvable_wavg:          18.24749
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999985
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.43725398
+    visit_patterns:
+      avg_first_visit_time:                     1840.1042
+      avg_last_visit_time:                      2148.1462
+      num_replay_batches:                       2158
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00030759076
+    avg_actor_approxkl3:                        0.00052819843
+    avg_actor_clipfrac:                         0.013378906
+    avg_actor_loss:                             -0.00039894876
+    avg_advantage:                              0.007855998
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00089951017
+    avg_entropy:                                0.038861874
+    avg_grad_norm_pre_clip:                     0.9116816
+    avg_loss:                                   1.1944457e-05
+    max:
+      max_actor_approxkl1:                      0.0013331163
+      max_actor_approxkl3:                      0.0015373987
+      max_actor_clipfrac:                       0.025634766
+      max_actor_loss:                           0.0005235858
+      max_advantage:                            0.13021672
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0020524345
+      max_entropy:                              0.049009226
+      max_grad_norm_pre_clip:                   1.4684774
+      max_loss:                                 0.0007123543
+    std:
+      std_actor_approxkl1:                      0.00039221707
+      std_actor_approxkl3:                      0.00039057407
+      std_actor_clipfrac:                       0.006491157
+      std_actor_loss:                           0.0005332682
+      std_advantage:                            0.04350975
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0004825875
+      std_entropy:                              0.0043824282
+      std_grad_norm_pre_clip:                   0.25491214
+      std_loss:                                 0.00045927623
+  eval-batch-orig:
+    avg_avg_episode_length:                     17.141327
+    avg_avg_return:                             0.9454954
+    avg_benchmark_regret:                       0.039331086
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09466553
+    proxy_corner:
+      avg_avg_return:                           0.9454954
+      avg_reward_per_step:                      0.09466553
+  eval-batch-shift:
+    avg_avg_episode_length:                     67.710625
+    avg_avg_return:                             0.25733888
+    avg_benchmark_regret:                       0.7295516
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.036766052
+    proxy_corner:
+      avg_avg_return:                           0.6062362
+      avg_reward_per_step:                      0.0051879883
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.717705
+    avg_avg_return:                             0.9396845
+    avg_benchmark_regret:                       0.044391133
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11139679
+    proxy_corner:
+      avg_avg_return:                           0.9396845
+      avg_reward_per_step:                      0.11139679
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        140509184
+    env-step-replay-before:                     71237632
+    env-step-used-before:                       140509184
+    ppo-update-before:                          85760
+    env-step-generate-before:                   69271552
+    env-step-all-after:                         140541952
+    env-step-replay-after:                      71270400
+    env-step-used-after:                        140541952
+    ppo-update-after:                           85780
+    env-step-generate-after:                    69271552
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     57.027866
+    avg_avg_return:                             0.5482336
+    avg_reward_per_step:                        0.015533447
+    proxy_corner:
+      avg_avg_return:                           0.5482336
+      avg_reward_per_step:                      0.015533447
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     57.027866
+    avg_avg_return:                             0.5482336
+    avg_reward_per_step:                        0.015533447
+    proxy_corner:
+      avg_avg_return:                           0.5482336
+      avg_reward_per_step:                      0.015533447
+  perf:
+    env_steps_per_second:                       316800.30973090336
+    ppo_updates_per_second:                     41.6100232808642
+    eval-batch-orig-duration:                   0.5009297039359808
+    eval-batch-shift-duration:                  0.43084709998220205
+    eval-batch-tree-duration:                   0.42041300144046545
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.607422
+      mouse-cheese_dist_finite_wavg:            17.356934
+      mouse-cheese_dist_solvable_avg:           29.607422
+      mouse-cheese_dist_solvable_wavg:          17.356934
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.42561626
+    visit_patterns:
+      avg_first_visit_time:                     1851.5764
+      avg_last_visit_time:                      2165.1506
+      num_replay_batches:                       2175
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014477986
+    avg_actor_approxkl3:                        0.0015022494
+    avg_actor_clipfrac:                         0.049536135
+    avg_actor_loss:                             0.0005948868
+    avg_advantage:                              -0.019140884
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0053552673
+    avg_entropy:                                0.13398091
+    avg_grad_norm_pre_clip:                     1.9871377
+    avg_loss:                                   0.0031385396
+    max:
+      max_actor_approxkl1:                      0.0027530137
+      max_actor_approxkl3:                      0.0032639652
+      max_actor_clipfrac:                       0.08093262
+      max_actor_loss:                           0.003340356
+      max_advantage:                            0.112576306
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0073091444
+      max_entropy:                              0.15063691
+      max_grad_norm_pre_clip:                   2.9706962
+      max_loss:                                 0.006594179
+    std:
+      std_actor_approxkl1:                      0.00083715963
+      std_actor_approxkl3:                      0.00071839895
+      std_actor_clipfrac:                       0.016555076
+      std_actor_loss:                           0.0010904921
+      std_advantage:                            0.10320509
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010394909
+      std_entropy:                              0.008894428
+      std_grad_norm_pre_clip:                   0.56082153
+      std_loss:                                 0.0013038667
+  eval-batch-orig:
+    avg_avg_episode_length:                     17.399212
+    avg_avg_return:                             0.9442055
+    avg_benchmark_regret:                       0.040621012
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.094429016
+    proxy_corner:
+      avg_avg_return:                           0.9442055
+      avg_reward_per_step:                      0.094429016
+  eval-batch-shift:
+    avg_avg_episode_length:                     67.29701
+    avg_avg_return:                             0.25971928
+    avg_benchmark_regret:                       0.7271712
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03601837
+    proxy_corner:
+      avg_avg_return:                           0.58332396
+      avg_reward_per_step:                      0.005088806
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.820183
+    avg_avg_return:                             0.9408467
+    avg_benchmark_regret:                       0.043228928
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.109436035
+    proxy_corner:
+      avg_avg_return:                           0.9408467
+      avg_reward_per_step:                      0.109436035
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        141557760
+    env-step-replay-before:                     71729152
+    env-step-used-before:                       141557760
+    ppo-update-before:                          86400
+    env-step-generate-before:                   69828608
+    env-step-all-after:                         141590528
+    env-step-replay-after:                      71761920
+    env-step-used-after:                        141590528
+    ppo-update-after:                           86420
+    env-step-generate-after:                    69828608
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     56.774918
+    avg_avg_return:                             0.5519454
+    avg_reward_per_step:                        0.01550293
+    proxy_corner:
+      avg_avg_return:                           0.5519454
+      avg_reward_per_step:                      0.01550293
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     56.774918
+    avg_avg_return:                             0.5519454
+    avg_reward_per_step:                        0.01550293
+    proxy_corner:
+      avg_avg_return:                           0.5519454
+      avg_reward_per_step:                      0.01550293
+  perf:
+    env_steps_per_second:                       316355.05225276196
+    ppo_updates_per_second:                     42.03097067877
+    eval-batch-orig-duration:                   0.5351526895537972
+    eval-batch-shift-duration:                  0.48347754031419754
+    eval-batch-tree-duration:                   0.48411733005195856
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.621826
+      mouse-cheese_dist_finite_wavg:            20.070127
+      mouse-cheese_dist_solvable_avg:           29.621826
+      mouse-cheese_dist_solvable_wavg:          20.070127
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.41997615
+    visit_patterns:
+      avg_first_visit_time:                     1868.0513
+      avg_last_visit_time:                      2180.27
+      num_replay_batches:                       2190
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0021024172
+    avg_actor_approxkl3:                        0.0019978092
+    avg_actor_clipfrac:                         0.05015259
+    avg_actor_loss:                             -4.0079096e-05
+    avg_advantage:                              -0.021561738
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0054500937
+    avg_entropy:                                0.12589651
+    avg_grad_norm_pre_clip:                     1.7028662
+    avg_loss:                                   0.002559071
+    max:
+      max_actor_approxkl1:                      0.0044378173
+      max_actor_approxkl3:                      0.0034343381
+      max_actor_clipfrac:                       0.07092285
+      max_actor_loss:                           0.002425693
+      max_advantage:                            0.114531934
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0078874305
+      max_entropy:                              0.13889307
+      max_grad_norm_pre_clip:                   2.8714588
+      max_loss:                                 0.005544785
+    std:
+      std_actor_approxkl1:                      0.0012661751
+      std_actor_approxkl3:                      0.0009217689
+      std_actor_clipfrac:                       0.015656263
+      std_actor_loss:                           0.0010474012
+      std_advantage:                            0.10442539
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0013749278
+      std_entropy:                              0.008034785
+      std_grad_norm_pre_clip:                   0.4770806
+      std_loss:                                 0.0012792965
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.838495
+    avg_avg_return:                             0.94502974
+    avg_benchmark_regret:                       0.039796773
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.095214844
+    proxy_corner:
+      avg_avg_return:                           0.94502974
+      avg_reward_per_step:                      0.095214844
+  eval-batch-shift:
+    avg_avg_episode_length:                     67.336426
+    avg_avg_return:                             0.26014555
+    avg_benchmark_regret:                       0.7267449
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.036819458
+    proxy_corner:
+      avg_avg_return:                           0.6356752
+      avg_reward_per_step:                      0.0057296753
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.76223
+    avg_avg_return:                             0.9429878
+    avg_benchmark_regret:                       0.04108781
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11087036
+    proxy_corner:
+      avg_avg_return:                           0.9429878
+      avg_reward_per_step:                      0.11087036
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        142606336
+    env-step-replay-before:                     72286208
+    env-step-used-before:                       142606336
+    ppo-update-before:                          87040
+    env-step-generate-before:                   70320128
+    env-step-all-after:                         142639104
+    env-step-replay-after:                      72318976
+    env-step-used-after:                        142639104
+    ppo-update-after:                           87060
+    env-step-generate-after:                    70320128
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     60.088997
+    avg_avg_return:                             0.5103273
+    avg_reward_per_step:                        0.014129639
+    proxy_corner:
+      avg_avg_return:                           0.5103273
+      avg_reward_per_step:                      0.014129639
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     60.088997
+    avg_avg_return:                             0.5103273
+    avg_reward_per_step:                        0.014129639
+    proxy_corner:
+      avg_avg_return:                           0.5103273
+      avg_reward_per_step:                      0.014129639
+  perf:
+    env_steps_per_second:                       310595.4494137004
+    ppo_updates_per_second:                     42.065717270053746
+    eval-batch-orig-duration:                   0.5421494077891111
+    eval-batch-shift-duration:                  0.511360670439899
+    eval-batch-tree-duration:                   0.4971959125250578
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.631104
+      mouse-cheese_dist_finite_wavg:            20.0642
+      mouse-cheese_dist_solvable_avg:           29.631104
+      mouse-cheese_dist_solvable_wavg:          20.064194
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000002
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.42772233
+    visit_patterns:
+      avg_first_visit_time:                     1881.7134
+      avg_last_visit_time:                      2197.3452
+      num_replay_batches:                       2207
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00079514337
+    avg_actor_approxkl3:                        0.00087632297
+    avg_actor_clipfrac:                         0.034954835
+    avg_actor_loss:                             -0.0006652918
+    avg_advantage:                              -0.026133677
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0069354842
+    avg_entropy:                                0.13934861
+    avg_grad_norm_pre_clip:                     1.5333412
+    avg_loss:                                   0.0026631018
+    max:
+      max_actor_approxkl1:                      0.0025788755
+      max_actor_approxkl3:                      0.0019446367
+      max_actor_clipfrac:                       0.06628418
+      max_actor_loss:                           0.00080566853
+      max_advantage:                            0.116283655
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0092934845
+      max_entropy:                              0.1495716
+      max_grad_norm_pre_clip:                   2.4603062
+      max_loss:                                 0.004114517
+    std:
+      std_actor_approxkl1:                      0.00060695136
+      std_actor_approxkl3:                      0.00050069706
+      std_actor_clipfrac:                       0.016239218
+      std_actor_loss:                           0.0008250401
+      std_advantage:                            0.11718555
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0013156383
+      std_entropy:                              0.005660906
+      std_grad_norm_pre_clip:                   0.44503006
+      std_loss:                                 0.0009789098
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.958683
+    avg_avg_return:                             0.9425236
+    avg_benchmark_regret:                       0.04230293
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09477997
+    proxy_corner:
+      avg_avg_return:                           0.9425236
+      avg_reward_per_step:                      0.09477997
+  eval-batch-shift:
+    avg_avg_episode_length:                     67.60256
+    avg_avg_return:                             0.25541666
+    avg_benchmark_regret:                       0.7314738
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.036209106
+    proxy_corner:
+      avg_avg_return:                           0.6495944
+      avg_reward_per_step:                      0.005554199
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.799116
+    avg_avg_return:                             0.9388906
+    avg_benchmark_regret:                       0.045185044
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11141205
+    proxy_corner:
+      avg_avg_return:                           0.9388906
+      avg_reward_per_step:                      0.11141205
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        143654912
+    env-step-replay-before:                     72941568
+    env-step-used-before:                       143654912
+    ppo-update-before:                          87680
+    env-step-generate-before:                   70713344
+    env-step-all-after:                         143687680
+    env-step-replay-after:                      72974336
+    env-step-used-after:                        143687680
+    ppo-update-after:                           87700
+    env-step-generate-after:                    70713344
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     57.627136
+    avg_avg_return:                             0.55184495
+    avg_reward_per_step:                        0.015167236
+    proxy_corner:
+      avg_avg_return:                           0.55184495
+      avg_reward_per_step:                      0.015167236
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     57.627136
+    avg_avg_return:                             0.55184495
+    avg_reward_per_step:                        0.015167236
+    proxy_corner:
+      avg_avg_return:                           0.55184495
+      avg_reward_per_step:                      0.015167236
+  perf:
+    env_steps_per_second:                       319157.33710532
+    ppo_updates_per_second:                     42.052355508812695
+    eval-batch-orig-duration:                   0.546773599460721
+    eval-batch-shift-duration:                  0.4858526336029172
+    eval-batch-tree-duration:                   0.49155355896800756
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.702148
+      mouse-cheese_dist_finite_wavg:            20.063948
+      mouse-cheese_dist_solvable_avg:           29.702148
+      mouse-cheese_dist_solvable_wavg:          20.063948
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.4220627
+    visit_patterns:
+      avg_first_visit_time:                     1892.4402
+      avg_last_visit_time:                      2217.2246
+      num_replay_batches:                       2227
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013423291
+    avg_actor_approxkl3:                        0.0012586903
+    avg_actor_clipfrac:                         0.045336913
+    avg_actor_loss:                             -0.00038849376
+    avg_advantage:                              -0.01543811
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.005043593
+    avg_entropy:                                0.14339224
+    avg_grad_norm_pre_clip:                     1.2929789
+    avg_loss:                                   0.0019899101
+    max:
+      max_actor_approxkl1:                      0.0028585445
+      max_actor_approxkl3:                      0.0017168198
+      max_actor_clipfrac:                       0.06335449
+      max_actor_loss:                           0.0021870583
+      max_advantage:                            0.1591959
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00688744
+      max_entropy:                              0.15892294
+      max_grad_norm_pre_clip:                   2.2239933
+      max_loss:                                 0.004155893
+    std:
+      std_actor_approxkl1:                      0.000657199
+      std_actor_approxkl3:                      0.00035850043
+      std_actor_clipfrac:                       0.012037668
+      std_actor_loss:                           0.00096103677
+      std_advantage:                            0.100402996
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00093727396
+      std_entropy:                              0.0071749664
+      std_grad_norm_pre_clip:                   0.3918
+      std_loss:                                 0.0010008287
+  eval-batch-orig:
+    avg_avg_episode_length:                     17.112772
+    avg_avg_return:                             0.945187
+    avg_benchmark_regret:                       0.039639547
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09475708
+    proxy_corner:
+      avg_avg_return:                           0.945187
+      avg_reward_per_step:                      0.09475708
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.9832
+    avg_avg_return:                             0.27705115
+    avg_benchmark_regret:                       0.7098393
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.037498474
+    proxy_corner:
+      avg_avg_return:                           0.6121893
+      avg_reward_per_step:                      0.0052719116
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.534931
+    avg_avg_return:                             0.9433888
+    avg_benchmark_regret:                       0.04068678
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.111328125
+    proxy_corner:
+      avg_avg_return:                           0.9433888
+      avg_reward_per_step:                      0.111328125
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        144703488
+    env-step-replay-before:                     73531392
+    env-step-used-before:                       144703488
+    ppo-update-before:                          88320
+    env-step-generate-before:                   71172096
+    env-step-all-after:                         144736256
+    env-step-replay-after:                      73564160
+    env-step-used-after:                        144736256
+    ppo-update-after:                           88340
+    env-step-generate-after:                    71172096
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     57.01483
+    avg_avg_return:                             0.55184096
+    avg_reward_per_step:                        0.015625
+    proxy_corner:
+      avg_avg_return:                           0.55184096
+      avg_reward_per_step:                      0.015625
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     57.01483
+    avg_avg_return:                             0.55184096
+    avg_reward_per_step:                        0.015625
+    proxy_corner:
+      avg_avg_return:                           0.55184096
+      avg_reward_per_step:                      0.015625
+  perf:
+    env_steps_per_second:                       348898.45619154884
+    ppo_updates_per_second:                     44.480795529727
+    eval-batch-orig-duration:                   0.5195826618000865
+    eval-batch-shift-duration:                  0.4930170411244035
+    eval-batch-tree-duration:                   0.49182954151183367
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.700195
+      mouse-cheese_dist_finite_wavg:            15.580082
+      mouse-cheese_dist_solvable_avg:           29.700195
+      mouse-cheese_dist_solvable_wavg:          15.580084
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999985
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.40842432
+    visit_patterns:
+      avg_first_visit_time:                     1904.2402
+      avg_last_visit_time:                      2235.2024
+      num_replay_batches:                       2245
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010816497
+    avg_actor_approxkl3:                        0.000979294
+    avg_actor_clipfrac:                         0.036480714
+    avg_actor_loss:                             -0.0006725308
+    avg_advantage:                              -0.017534459
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0054738782
+    avg_entropy:                                0.13108335
+    avg_grad_norm_pre_clip:                     1.5965588
+    avg_loss:                                   0.001933325
+    max:
+      max_actor_approxkl1:                      0.0029626663
+      max_actor_approxkl3:                      0.002088244
+      max_actor_clipfrac:                       0.06652832
+      max_actor_loss:                           0.00047756732
+      max_advantage:                            0.11611432
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00722911
+      max_entropy:                              0.14318913
+      max_grad_norm_pre_clip:                   2.6296954
+      max_loss:                                 0.0033520688
+    std:
+      std_actor_approxkl1:                      0.0008571099
+      std_actor_approxkl3:                      0.0005184597
+      std_actor_clipfrac:                       0.016648632
+      std_actor_loss:                           0.0007410882
+      std_advantage:                            0.10414241
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0008586407
+      std_entropy:                              0.0066205873
+      std_grad_norm_pre_clip:                   0.58251286
+      std_loss:                                 0.00084480323
+  eval-batch-orig:
+    avg_avg_episode_length:                     17.028692
+    avg_avg_return:                             0.94517064
+    avg_benchmark_regret:                       0.03965591
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.094825745
+    proxy_corner:
+      avg_avg_return:                           0.94517064
+      avg_reward_per_step:                      0.094825745
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.222664
+    avg_avg_return:                             0.27498442
+    avg_benchmark_regret:                       0.7119061
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03793335
+    proxy_corner:
+      avg_avg_return:                           0.6303029
+      avg_reward_per_step:                      0.005455017
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.733227
+    avg_avg_return:                             0.9412194
+    avg_benchmark_regret:                       0.042856216
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11084747
+    proxy_corner:
+      avg_avg_return:                           0.9412194
+      avg_reward_per_step:                      0.11084747
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        145752064
+    env-step-replay-before:                     74088448
+    env-step-used-before:                       145752064
+    ppo-update-before:                          88960
+    env-step-generate-before:                   71663616
+    env-step-all-after:                         145784832
+    env-step-replay-after:                      74121216
+    env-step-used-after:                        145784832
+    ppo-update-after:                           88980
+    env-step-generate-after:                    71663616
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     55.664383
+    avg_avg_return:                             0.5688275
+    avg_reward_per_step:                        0.016174316
+    proxy_corner:
+      avg_avg_return:                           0.5688275
+      avg_reward_per_step:                      0.016174316
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     55.664383
+    avg_avg_return:                             0.5688275
+    avg_reward_per_step:                        0.016174316
+    proxy_corner:
+      avg_avg_return:                           0.5688275
+      avg_reward_per_step:                      0.016174316
+  perf:
+    env_steps_per_second:                       289474.43601258873
+    ppo_updates_per_second:                     42.163722370002965
+    eval-batch-orig-duration:                   0.5273985518142581
+    eval-batch-shift-duration:                  0.4911916581913829
+    eval-batch-tree-duration:                   0.49031895864754915
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.71289
+      mouse-cheese_dist_finite_wavg:            15.575124
+      mouse-cheese_dist_solvable_avg:           29.71289
+      mouse-cheese_dist_solvable_wavg:          15.575126
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.41888812
+    visit_patterns:
+      avg_first_visit_time:                     1919.072
+      avg_last_visit_time:                      2252.3604
+      num_replay_batches:                       2262
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014305502
+    avg_actor_approxkl3:                        0.0010980854
+    avg_actor_clipfrac:                         0.037420657
+    avg_actor_loss:                             -0.00035808104
+    avg_advantage:                              0.0058327224
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.004647793
+    avg_entropy:                                0.12623721
+    avg_grad_norm_pre_clip:                     1.5533435
+    avg_loss:                                   0.0018395784
+    max:
+      max_actor_approxkl1:                      0.0031431192
+      max_actor_approxkl3:                      0.0017920303
+      max_actor_clipfrac:                       0.05493164
+      max_actor_loss:                           0.0020397902
+      max_advantage:                            0.15877873
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.006676055
+      max_entropy:                              0.13477422
+      max_grad_norm_pre_clip:                   2.5653763
+      max_loss:                                 0.0038191336
+    std:
+      std_actor_approxkl1:                      0.0008768015
+      std_actor_approxkl3:                      0.0004499816
+      std_actor_clipfrac:                       0.013766884
+      std_actor_loss:                           0.0009779716
+      std_advantage:                            0.09653104
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00084275566
+      std_entropy:                              0.005785764
+      std_grad_norm_pre_clip:                   0.47296327
+      std_loss:                                 0.00094735273
+  eval-batch-orig:
+    avg_avg_episode_length:                     17.167324
+    avg_avg_return:                             0.94250035
+    avg_benchmark_regret:                       0.042326108
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.094573975
+    proxy_corner:
+      avg_avg_return:                           0.94250035
+      avg_reward_per_step:                      0.094573975
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.828995
+    avg_avg_return:                             0.26850343
+    avg_benchmark_regret:                       0.718387
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03691864
+    proxy_corner:
+      avg_avg_return:                           0.6169485
+      avg_reward_per_step:                      0.005340576
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.859398
+    avg_avg_return:                             0.9399942
+    avg_benchmark_regret:                       0.044081368
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.110321045
+    proxy_corner:
+      avg_avg_return:                           0.9399942
+      avg_reward_per_step:                      0.110321045
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        146800640
+    env-step-replay-before:                     74711040
+    env-step-used-before:                       146800640
+    ppo-update-before:                          89600
+    env-step-generate-before:                   72089600
+    env-step-all-after:                         146833408
+    env-step-replay-after:                      74743808
+    env-step-used-after:                        146833408
+    ppo-update-after:                           89620
+    env-step-generate-after:                    72089600
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     55.652973
+    avg_avg_return:                             0.5727002
+    avg_reward_per_step:                        0.016357422
+    proxy_corner:
+      avg_avg_return:                           0.5727002
+      avg_reward_per_step:                      0.016357422
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     55.652973
+    avg_avg_return:                             0.5727002
+    avg_reward_per_step:                        0.016357422
+    proxy_corner:
+      avg_avg_return:                           0.5727002
+      avg_reward_per_step:                      0.016357422
+  perf:
+    env_steps_per_second:                       347743.728439215
+    ppo_updates_per_second:                     41.13241048778074
+    eval-batch-orig-duration:                   0.5216446090489626
+    eval-batch-shift-duration:                  0.4829403841868043
+    eval-batch-tree-duration:                   0.4881261810660362
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.701416
+      mouse-cheese_dist_finite_wavg:            19.170244
+      mouse-cheese_dist_solvable_avg:           29.701416
+      mouse-cheese_dist_solvable_wavg:          19.170242
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.39776942
+    visit_patterns:
+      avg_first_visit_time:                     1932.6692
+      avg_last_visit_time:                      2271.278
+      num_replay_batches:                       2281
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0018101953
+    avg_actor_approxkl3:                        0.0016189892
+    avg_actor_clipfrac:                         0.046917725
+    avg_actor_loss:                             -0.0006094139
+    avg_advantage:                              -0.008812159
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0042468747
+    avg_entropy:                                0.12908162
+    avg_grad_norm_pre_clip:                     1.442077
+    avg_loss:                                   0.0013849416
+    max:
+      max_actor_approxkl1:                      0.0039886395
+      max_actor_approxkl3:                      0.003570208
+      max_actor_clipfrac:                       0.07788086
+      max_actor_loss:                           0.00085414946
+      max_advantage:                            0.14177388
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0068481085
+      max_entropy:                              0.14655654
+      max_grad_norm_pre_clip:                   2.9034529
+      max_loss:                                 0.003078202
+    std:
+      std_actor_approxkl1:                      0.001213176
+      std_actor_approxkl3:                      0.0010097305
+      std_actor_clipfrac:                       0.020678304
+      std_actor_loss:                           0.0005514608
+      std_advantage:                            0.09177818
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011827241
+      std_entropy:                              0.010127964
+      std_grad_norm_pre_clip:                   0.64165133
+      std_loss:                                 0.0007928743
+  eval-batch-orig:
+    avg_avg_episode_length:                     17.471394
+    avg_avg_return:                             0.9417801
+    avg_benchmark_regret:                       0.043046378
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09350586
+    proxy_corner:
+      avg_avg_return:                           0.9417801
+      avg_reward_per_step:                      0.09350586
+  eval-batch-shift:
+    avg_avg_episode_length:                     64.50455
+    avg_avg_return:                             0.30354926
+    avg_benchmark_regret:                       0.68334115
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03891754
+    proxy_corner:
+      avg_avg_return:                           0.5825753
+      avg_reward_per_step:                      0.005355835
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.93141
+    avg_avg_return:                             0.9408418
+    avg_benchmark_regret:                       0.04323379
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11049652
+    proxy_corner:
+      avg_avg_return:                           0.9408418
+      avg_reward_per_step:                      0.11049652
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        147849216
+    env-step-replay-before:                     75235328
+    env-step-used-before:                       147849216
+    ppo-update-before:                          90240
+    env-step-generate-before:                   72613888
+    env-step-all-after:                         147881984
+    env-step-replay-after:                      75235328
+    env-step-used-after:                        147881984
+    ppo-update-after:                           90260
+    env-step-generate-after:                    72646656
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.354607
+    avg_avg_return:                             0.8615882
+    avg_reward_per_step:                        0.095825195
+    proxy_corner:
+      avg_avg_return:                           0.8615882
+      avg_reward_per_step:                      0.095825195
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.354607
+    avg_avg_return:                             0.8615882
+    avg_reward_per_step:                        0.095825195
+    proxy_corner:
+      avg_avg_return:                           0.8615882
+      avg_reward_per_step:                      0.095825195
+  perf:
+    env_steps_per_second:                       314158.95048690686
+    ppo_updates_per_second:                     40.90045302258088
+    eval-batch-orig-duration:                   0.530718095600605
+    eval-batch-shift-duration:                  0.4927814556285739
+    eval-batch-tree-duration:                   0.49317288771271706
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.694336
+      mouse-cheese_dist_finite_wavg:            16.473818
+      mouse-cheese_dist_solvable_avg:           29.694336
+      mouse-cheese_dist_solvable_wavg:          16.47382
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016816
+    scoring:
+      avg_scores:                               0.39801472
+    visit_patterns:
+      avg_first_visit_time:                     1948.085
+      avg_last_visit_time:                      2286.307
+      num_replay_batches:                       2296
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00032242705
+    avg_actor_approxkl3:                        0.0004155534
+    avg_actor_clipfrac:                         0.010437012
+    avg_actor_loss:                             -0.0006724314
+    avg_advantage:                              0.011490155
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00038086998
+    avg_entropy:                                0.0277454
+    avg_grad_norm_pre_clip:                     0.71313345
+    avg_loss:                                   -0.00050974183
+    max:
+      max_actor_approxkl1:                      0.0008950641
+      max_actor_approxkl3:                      0.00091390044
+      max_actor_clipfrac:                       0.017456055
+      max_actor_loss:                           3.7066638e-07
+      max_advantage:                            0.11276823
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00091203634
+      max_entropy:                              0.03574615
+      max_grad_norm_pre_clip:                   1.9405674
+      max_loss:                                 0.0001322353
+    std:
+      std_actor_approxkl1:                      0.00031742445
+      std_actor_approxkl3:                      0.00024674318
+      std_actor_clipfrac:                       0.0040676165
+      std_actor_loss:                           0.00037781548
+      std_advantage:                            0.02703997
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0002501371
+      std_entropy:                              0.004468489
+      std_grad_norm_pre_clip:                   0.36448887
+      std_loss:                                 0.00037901392
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.840628
+    avg_avg_return:                             0.9457661
+    avg_benchmark_regret:                       0.03906038
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09458923
+    proxy_corner:
+      avg_avg_return:                           0.9457661
+      avg_reward_per_step:                      0.09458923
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.3555
+    avg_avg_return:                             0.27874044
+    avg_benchmark_regret:                       0.70815
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.037635803
+    proxy_corner:
+      avg_avg_return:                           0.6115547
+      avg_reward_per_step:                      0.0054244995
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.232853
+    avg_avg_return:                             0.9442027
+    avg_benchmark_regret:                       0.039872877
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11166382
+    proxy_corner:
+      avg_avg_return:                           0.9442027
+      avg_reward_per_step:                      0.11166382
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        148897792
+    env-step-replay-before:                     75694080
+    env-step-used-before:                       148897792
+    ppo-update-before:                          90880
+    env-step-generate-before:                   73203712
+    env-step-all-after:                         148930560
+    env-step-replay-after:                      75726848
+    env-step-used-after:                        148930560
+    ppo-update-after:                           90900
+    env-step-generate-after:                    73203712
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     57.39381
+    avg_avg_return:                             0.5559515
+    avg_reward_per_step:                        0.015380859
+    proxy_corner:
+      avg_avg_return:                           0.5559515
+      avg_reward_per_step:                      0.015380859
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     57.39381
+    avg_avg_return:                             0.5559515
+    avg_reward_per_step:                        0.015380859
+    proxy_corner:
+      avg_avg_return:                           0.5559515
+      avg_reward_per_step:                      0.015380859
+  perf:
+    env_steps_per_second:                       310452.7890587373
+    ppo_updates_per_second:                     42.18533758351047
+    eval-batch-orig-duration:                   0.5315074482932687
+    eval-batch-shift-duration:                  0.4827496884390712
+    eval-batch-tree-duration:                   0.4386714603751898
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.7854
+      mouse-cheese_dist_finite_wavg:            17.378109
+      mouse-cheese_dist_solvable_avg:           29.7854
+      mouse-cheese_dist_solvable_wavg:          17.37811
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.99999
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.40299314
+    visit_patterns:
+      avg_first_visit_time:                     1964.7756
+      avg_last_visit_time:                      2301.3582
+      num_replay_batches:                       2311
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00097444403
+    avg_actor_approxkl3:                        0.0010832237
+    avg_actor_clipfrac:                         0.038256835
+    avg_actor_loss:                             -0.00037278255
+    avg_advantage:                              -0.018671058
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.004769181
+    avg_entropy:                                0.12688401
+    avg_grad_norm_pre_clip:                     1.3287926
+    avg_loss:                                   0.0018849237
+    max:
+      max_actor_approxkl1:                      0.002076554
+      max_actor_approxkl3:                      0.0016078337
+      max_actor_clipfrac:                       0.055786133
+      max_actor_loss:                           0.0010082722
+      max_advantage:                            0.11822122
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0067699235
+      max_entropy:                              0.1461253
+      max_grad_norm_pre_clip:                   2.027345
+      max_loss:                                 0.0032236015
+    std:
+      std_actor_approxkl1:                      0.00067572464
+      std_actor_approxkl3:                      0.0004037865
+      std_actor_clipfrac:                       0.012845509
+      std_actor_loss:                           0.00068889104
+      std_advantage:                            0.09756724
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0012567687
+      std_entropy:                              0.007823037
+      std_grad_norm_pre_clip:                   0.29591963
+      std_loss:                                 0.00085706846
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.796173
+    avg_avg_return:                             0.94697
+    avg_benchmark_regret:                       0.037856538
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09492493
+    proxy_corner:
+      avg_avg_return:                           0.94697
+      avg_reward_per_step:                      0.09492493
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.295654
+    avg_avg_return:                             0.27526265
+    avg_benchmark_regret:                       0.7116278
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.037902832
+    proxy_corner:
+      avg_avg_return:                           0.6355717
+      avg_reward_per_step:                      0.005607605
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.505436
+    avg_avg_return:                             0.9418165
+    avg_benchmark_regret:                       0.04225909
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11186218
+    proxy_corner:
+      avg_avg_return:                           0.9418165
+      avg_reward_per_step:                      0.11186218
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        149946368
+    env-step-replay-before:                     76152832
+    env-step-used-before:                       149946368
+    ppo-update-before:                          91520
+    env-step-generate-before:                   73793536
+    env-step-all-after:                         149979136
+    env-step-replay-after:                      76152832
+    env-step-used-after:                        149979136
+    ppo-update-after:                           91540
+    env-step-generate-after:                    73826304
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.692005
+    avg_avg_return:                             0.8598902
+    avg_reward_per_step:                        0.114746094
+    proxy_corner:
+      avg_avg_return:                           0.8598902
+      avg_reward_per_step:                      0.114746094
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.692005
+    avg_avg_return:                             0.8598902
+    avg_reward_per_step:                        0.114746094
+    proxy_corner:
+      avg_avg_return:                           0.8598902
+      avg_reward_per_step:                      0.114746094
+  perf:
+    env_steps_per_second:                       238033.07422432626
+    ppo_updates_per_second:                     41.861261327906824
+    eval-batch-orig-duration:                   0.36803953535854816
+    eval-batch-shift-duration:                  0.3019962627440691
+    eval-batch-tree-duration:                   0.3379078069701791
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.783691
+      mouse-cheese_dist_finite_wavg:            20.971664
+      mouse-cheese_dist_solvable_avg:           29.783691
+      mouse-cheese_dist_solvable_wavg:          20.971666
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.41662663
+    visit_patterns:
+      avg_first_visit_time:                     1980.0061
+      avg_last_visit_time:                      2314.2375
+      num_replay_batches:                       2324
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0007103311
+    avg_actor_approxkl3:                        0.0007549187
+    avg_actor_clipfrac:                         0.01552124
+    avg_actor_loss:                             0.00034101112
+    avg_advantage:                              0.00018117254
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00064087566
+    avg_entropy:                                0.031367596
+    avg_grad_norm_pre_clip:                     1.102816
+    avg_loss:                                   0.00063008134
+    max:
+      max_actor_approxkl1:                      0.0021409767
+      max_actor_approxkl3:                      0.001702759
+      max_actor_clipfrac:                       0.029785156
+      max_actor_loss:                           0.0026912112
+      max_advantage:                            0.14942604
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0019435826
+      max_entropy:                              0.04416025
+      max_grad_norm_pre_clip:                   2.1090457
+      max_loss:                                 0.002770714
+    std:
+      std_actor_approxkl1:                      0.0006718609
+      std_actor_approxkl3:                      0.0004969139
+      std_actor_clipfrac:                       0.0068405434
+      std_actor_loss:                           0.0011258834
+      std_advantage:                            0.035577618
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00053361047
+      std_entropy:                              0.0057899053
+      std_grad_norm_pre_clip:                   0.32941785
+      std_loss:                                 0.0009781414
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.916775
+    avg_avg_return:                             0.9468577
+    avg_benchmark_regret:                       0.03796885
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09523773
+    proxy_corner:
+      avg_avg_return:                           0.9468577
+      avg_reward_per_step:                      0.09523773
+  eval-batch-shift:
+    avg_avg_episode_length:                     67.42873
+    avg_avg_return:                             0.25718027
+    avg_benchmark_regret:                       0.72971016
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03882599
+    proxy_corner:
+      avg_avg_return:                           0.64021623
+      avg_reward_per_step:                      0.0054397583
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.563423
+    avg_avg_return:                             0.9421783
+    avg_benchmark_regret:                       0.041897304
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11200714
+    proxy_corner:
+      avg_avg_return:                           0.9421783
+      avg_reward_per_step:                      0.11200714
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        150994944
+    env-step-replay-before:                     76709888
+    env-step-used-before:                       150994944
+    ppo-update-before:                          92160
+    env-step-generate-before:                   74285056
+    env-step-all-after:                         151027712
+    env-step-replay-after:                      76742656
+    env-step-used-after:                        151027712
+    ppo-update-after:                           92180
+    env-step-generate-after:                    74285056
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     56.02391
+    avg_avg_return:                             0.55029076
+    avg_reward_per_step:                        0.016204834
+    proxy_corner:
+      avg_avg_return:                           0.55029076
+      avg_reward_per_step:                      0.016204834
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     56.02391
+    avg_avg_return:                             0.55029076
+    avg_reward_per_step:                        0.016204834
+    proxy_corner:
+      avg_avg_return:                           0.55029076
+      avg_reward_per_step:                      0.016204834
+  perf:
+    env_steps_per_second:                       296582.2026516229
+    ppo_updates_per_second:                     44.44717454166004
+    eval-batch-orig-duration:                   0.5521979909390211
+    eval-batch-shift-duration:                  0.45890011731535196
+    eval-batch-tree-duration:                   0.4779365696012974
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.831055
+      mouse-cheese_dist_finite_wavg:            19.184828
+      mouse-cheese_dist_solvable_avg:           29.831055
+      mouse-cheese_dist_solvable_wavg:          19.184828
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.40690142
+    visit_patterns:
+      avg_first_visit_time:                     1992.7356
+      avg_last_visit_time:                      2332.229
+      num_replay_batches:                       2342
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001265327
+    avg_actor_approxkl3:                        0.0011645157
+    avg_actor_clipfrac:                         0.04128418
+    avg_actor_loss:                             -0.0006479863
+    avg_advantage:                              -0.020321975
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0057816855
+    avg_entropy:                                0.12211146
+    avg_grad_norm_pre_clip:                     1.4341831
+    avg_loss:                                   0.0021207451
+    max:
+      max_actor_approxkl1:                      0.0033026228
+      max_actor_approxkl3:                      0.0020780019
+      max_actor_clipfrac:                       0.063964844
+      max_actor_loss:                           0.00042461604
+      max_advantage:                            0.108508706
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.006970652
+      max_entropy:                              0.13502872
+      max_grad_norm_pre_clip:                   2.6640153
+      max_loss:                                 0.003719922
+    std:
+      std_actor_approxkl1:                      0.0009352865
+      std_actor_approxkl3:                      0.00048829097
+      std_actor_clipfrac:                       0.012883687
+      std_actor_loss:                           0.0007035487
+      std_advantage:                            0.107325785
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0009445988
+      std_entropy:                              0.0071333586
+      std_grad_norm_pre_clip:                   0.55374295
+      std_loss:                                 0.0008356104
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.920387
+    avg_avg_return:                             0.942016
+    avg_benchmark_regret:                       0.042810485
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09489441
+    proxy_corner:
+      avg_avg_return:                           0.942016
+      avg_reward_per_step:                      0.09489441
+  eval-batch-shift:
+    avg_avg_episode_length:                     67.2832
+    avg_avg_return:                             0.2561577
+    avg_benchmark_regret:                       0.73073274
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03794098
+    proxy_corner:
+      avg_avg_return:                           0.6310938
+      avg_reward_per_step:                      0.0054626465
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.810516
+    avg_avg_return:                             0.94357145
+    avg_benchmark_regret:                       0.040504158
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11073303
+    proxy_corner:
+      avg_avg_return:                           0.94357145
+      avg_reward_per_step:                      0.11073303
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        152043520
+    env-step-replay-before:                     77299712
+    env-step-used-before:                       152043520
+    ppo-update-before:                          92800
+    env-step-generate-before:                   74743808
+    env-step-all-after:                         152076288
+    env-step-replay-after:                      77299712
+    env-step-used-after:                        152076288
+    ppo-update-after:                           92820
+    env-step-generate-after:                    74776576
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     20.71745
+    avg_avg_return:                             0.8415349
+    avg_reward_per_step:                        0.095062256
+    proxy_corner:
+      avg_avg_return:                           0.8415349
+      avg_reward_per_step:                      0.095062256
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     20.71745
+    avg_avg_return:                             0.8415349
+    avg_reward_per_step:                        0.095062256
+    proxy_corner:
+      avg_avg_return:                           0.8415349
+      avg_reward_per_step:                      0.095062256
+  perf:
+    env_steps_per_second:                       261550.66047074294
+    ppo_updates_per_second:                     41.54559736965672
+    eval-batch-orig-duration:                   0.5390041507780552
+    eval-batch-shift-duration:                  0.4846999254077673
+    eval-batch-tree-duration:                   0.48444890324026346
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.860352
+      mouse-cheese_dist_finite_wavg:            19.191198
+      mouse-cheese_dist_solvable_avg:           29.860352
+      mouse-cheese_dist_solvable_wavg:          19.1912
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.4102865
+    visit_patterns:
+      avg_first_visit_time:                     2007.3792
+      avg_last_visit_time:                      2349.2583
+      num_replay_batches:                       2359
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00084844057
+    avg_actor_approxkl3:                        0.0006507111
+    avg_actor_clipfrac:                         0.015454101
+    avg_actor_loss:                             -0.000263861
+    avg_advantage:                              0.012466645
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0009808507
+    avg_entropy:                                0.038943447
+    avg_grad_norm_pre_clip:                     1.1270152
+    avg_loss:                                   0.00018762088
+    max:
+      max_actor_approxkl1:                      0.0020179073
+      max_actor_approxkl3:                      0.0012667663
+      max_actor_clipfrac:                       0.025878906
+      max_actor_loss:                           0.0012377389
+      max_advantage:                            0.12123263
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0025561133
+      max_entropy:                              0.052050784
+      max_grad_norm_pre_clip:                   1.6894895
+      max_loss:                                 0.0019169917
+    std:
+      std_actor_approxkl1:                      0.000568485
+      std_actor_approxkl3:                      0.00038938614
+      std_actor_clipfrac:                       0.0064432314
+      std_actor_loss:                           0.000789322
+      std_advantage:                            0.0438838
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0006025186
+      std_entropy:                              0.007471653
+      std_grad_norm_pre_clip:                   0.34898835
+      std_loss:                                 0.00068868115
+  eval-batch-orig:
+    avg_avg_episode_length:                     17.023397
+    avg_avg_return:                             0.9429668
+    avg_benchmark_regret:                       0.04185966
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09452057
+    proxy_corner:
+      avg_avg_return:                           0.9429668
+      avg_reward_per_step:                      0.09452057
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.224884
+    avg_avg_return:                             0.2730159
+    avg_benchmark_regret:                       0.7138746
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.039909363
+    proxy_corner:
+      avg_avg_return:                           0.6127506
+      avg_reward_per_step:                      0.0054092407
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.66749
+    avg_avg_return:                             0.9412339
+    avg_benchmark_regret:                       0.04284169
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.109375
+    proxy_corner:
+      avg_avg_return:                           0.9412339
+      avg_reward_per_step:                      0.109375
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        153092096
+    env-step-replay-before:                     77889536
+    env-step-used-before:                       153092096
+    ppo-update-before:                          93440
+    env-step-generate-before:                   75202560
+    env-step-all-after:                         153124864
+    env-step-replay-after:                      77889536
+    env-step-used-after:                        153124864
+    ppo-update-after:                           93460
+    env-step-generate-after:                    75235328
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     17.79361
+    avg_avg_return:                             0.8702459
+    avg_reward_per_step:                        0.111450195
+    proxy_corner:
+      avg_avg_return:                           0.8702459
+      avg_reward_per_step:                      0.111450195
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     17.79361
+    avg_avg_return:                             0.8702459
+    avg_reward_per_step:                        0.111450195
+    proxy_corner:
+      avg_avg_return:                           0.8702459
+      avg_reward_per_step:                      0.111450195
+  perf:
+    env_steps_per_second:                       308484.7082477982
+    ppo_updates_per_second:                     42.24425457140707
+    eval-batch-orig-duration:                   0.5295645100995898
+    eval-batch-shift-duration:                  0.4873897656798363
+    eval-batch-tree-duration:                   0.48679540399461985
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.824219
+      mouse-cheese_dist_finite_wavg:            19.187407
+      mouse-cheese_dist_solvable_avg:           29.824219
+      mouse-cheese_dist_solvable_wavg:          19.187407
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.40975726
+    visit_patterns:
+      avg_first_visit_time:                     2019.3367
+      avg_last_visit_time:                      2367.2705
+      num_replay_batches:                       2377
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.000524905
+    avg_actor_approxkl3:                        0.00052152196
+    avg_actor_clipfrac:                         0.009710694
+    avg_actor_loss:                             -0.00028198873
+    avg_advantage:                              0.008398887
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00036041604
+    avg_entropy:                                0.027233297
+    avg_grad_norm_pre_clip:                     0.87603086
+    avg_loss:                                   -0.00012901402
+    max:
+      max_actor_approxkl1:                      0.0021890993
+      max_actor_approxkl3:                      0.0016946031
+      max_actor_clipfrac:                       0.016479492
+      max_actor_loss:                           0.00045317598
+      max_advantage:                            0.104157865
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0008530802
+      max_entropy:                              0.032778583
+      max_grad_norm_pre_clip:                   1.7848127
+      max_loss:                                 0.0005447126
+    std:
+      std_actor_approxkl1:                      0.000636861
+      std_actor_approxkl3:                      0.00047006385
+      std_actor_clipfrac:                       0.0047017634
+      std_actor_loss:                           0.00034375308
+      std_advantage:                            0.02670465
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00025530046
+      std_entropy:                              0.0034391212
+      std_grad_norm_pre_clip:                   0.32253283
+      std_loss:                                 0.00035757027
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.954014
+    avg_avg_return:                             0.9450778
+    avg_benchmark_regret:                       0.03974874
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09461975
+    proxy_corner:
+      avg_avg_return:                           0.9450778
+      avg_reward_per_step:                      0.09461975
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.17526
+    avg_avg_return:                             0.2766754
+    avg_benchmark_regret:                       0.71021503
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03869629
+    proxy_corner:
+      avg_avg_return:                           0.6134722
+      avg_reward_per_step:                      0.0053710938
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.54905
+    avg_avg_return:                             0.94042933
+    avg_benchmark_regret:                       0.04364624
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11066437
+    proxy_corner:
+      avg_avg_return:                           0.94042933
+      avg_reward_per_step:                      0.11066437
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        154140672
+    env-step-replay-before:                     78544896
+    env-step-used-before:                       154140672
+    ppo-update-before:                          94080
+    env-step-generate-before:                   75595776
+    env-step-all-after:                         154173440
+    env-step-replay-after:                      78577664
+    env-step-used-after:                        154173440
+    ppo-update-after:                           94100
+    env-step-generate-after:                    75595776
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     56.070877
+    avg_avg_return:                             0.575364
+    avg_reward_per_step:                        0.015808105
+    proxy_corner:
+      avg_avg_return:                           0.575364
+      avg_reward_per_step:                      0.015808105
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     56.070877
+    avg_avg_return:                             0.575364
+    avg_reward_per_step:                        0.015808105
+    proxy_corner:
+      avg_avg_return:                           0.575364
+      avg_reward_per_step:                      0.015808105
+  perf:
+    env_steps_per_second:                       346220.6781481731
+    ppo_updates_per_second:                     41.2193634700225
+    eval-batch-orig-duration:                   0.5433168029412627
+    eval-batch-shift-duration:                  0.48185546044260263
+    eval-batch-tree-duration:                   0.4887466998770833
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.890137
+      mouse-cheese_dist_finite_wavg:            20.084644
+      mouse-cheese_dist_solvable_avg:           29.890137
+      mouse-cheese_dist_solvable_wavg:          20.084646
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.38970006
+    visit_patterns:
+      avg_first_visit_time:                     2031.1992
+      avg_last_visit_time:                      2388.2378
+      num_replay_batches:                       2398
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0019757326
+    avg_actor_approxkl3:                        0.0018652969
+    avg_actor_clipfrac:                         0.050683595
+    avg_actor_loss:                             6.500389e-05
+    avg_advantage:                              -0.013855424
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.003929876
+    avg_entropy:                                0.13356458
+    avg_grad_norm_pre_clip:                     1.7281865
+    avg_loss:                                   0.0018963771
+    max:
+      max_actor_approxkl1:                      0.0043477966
+      max_actor_approxkl3:                      0.0034450605
+      max_actor_clipfrac:                       0.07763672
+      max_actor_loss:                           0.0016960129
+      max_advantage:                            0.117111325
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0067153783
+      max_entropy:                              0.1477994
+      max_grad_norm_pre_clip:                   2.8162599
+      max_loss:                                 0.0049084197
+    std:
+      std_actor_approxkl1:                      0.0012314534
+      std_actor_approxkl3:                      0.0009738307
+      std_actor_clipfrac:                       0.021622494
+      std_actor_loss:                           0.0009328819
+      std_advantage:                            0.08854944
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011082762
+      std_entropy:                              0.008261745
+      std_grad_norm_pre_clip:                   0.59679615
+      std_loss:                                 0.0010998502
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.789013
+    avg_avg_return:                             0.9466146
+    avg_benchmark_regret:                       0.038211893
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09485626
+    proxy_corner:
+      avg_avg_return:                           0.9466146
+      avg_reward_per_step:                      0.09485626
+  eval-batch-shift:
+    avg_avg_episode_length:                     65.68344
+    avg_avg_return:                             0.28228617
+    avg_benchmark_regret:                       0.70460427
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03929138
+    proxy_corner:
+      avg_avg_return:                           0.61250436
+      avg_reward_per_step:                      0.005264282
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.223999
+    avg_avg_return:                             0.9439637
+    avg_benchmark_regret:                       0.04011187
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.111839294
+    proxy_corner:
+      avg_avg_return:                           0.9439637
+      avg_reward_per_step:                      0.111839294
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        155189248
+    env-step-replay-before:                     79134720
+    env-step-used-before:                       155189248
+    ppo-update-before:                          94720
+    env-step-generate-before:                   76054528
+    env-step-all-after:                         155222016
+    env-step-replay-after:                      79134720
+    env-step-used-after:                        155222016
+    ppo-update-after:                           94740
+    env-step-generate-after:                    76087296
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.699991
+    avg_avg_return:                             0.8541384
+    avg_reward_per_step:                        0.10110474
+    proxy_corner:
+      avg_avg_return:                           0.8541384
+      avg_reward_per_step:                      0.10110474
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.699991
+    avg_avg_return:                             0.8541384
+    avg_reward_per_step:                        0.10110474
+    proxy_corner:
+      avg_avg_return:                           0.8541384
+      avg_reward_per_step:                      0.10110474
+  perf:
+    env_steps_per_second:                       318031.5796923762
+    ppo_updates_per_second:                     44.45156188194588
+    eval-batch-orig-duration:                   0.5136224096640944
+    eval-batch-shift-duration:                  0.49339236319065094
+    eval-batch-tree-duration:                   0.48850630782544613
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.838623
+      mouse-cheese_dist_finite_wavg:            16.48082
+      mouse-cheese_dist_solvable_avg:           29.838623
+      mouse-cheese_dist_solvable_wavg:          16.480822
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.39445636
+    visit_patterns:
+      avg_first_visit_time:                     2046.8386
+      avg_last_visit_time:                      2405.2637
+      num_replay_batches:                       2415
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00074001524
+    avg_actor_approxkl3:                        0.00053919695
+    avg_actor_clipfrac:                         0.010943604
+    avg_actor_loss:                             -0.000115733455
+    avg_advantage:                              0.0055157403
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00078587944
+    avg_entropy:                                0.026595926
+    avg_grad_norm_pre_clip:                     0.66770625
+    avg_loss:                                   0.00025061032
+    max:
+      max_actor_approxkl1:                      0.0026148858
+      max_actor_approxkl3:                      0.0016551284
+      max_actor_clipfrac:                       0.022460938
+      max_actor_loss:                           0.0004919842
+      max_advantage:                            0.107131004
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0013666097
+      max_entropy:                              0.034373492
+      max_grad_norm_pre_clip:                   1.4666367
+      max_loss:                                 0.00068534596
+    std:
+      std_actor_approxkl1:                      0.0006255069
+      std_actor_approxkl3:                      0.00034998293
+      std_actor_clipfrac:                       0.004502769
+      std_actor_loss:                           0.00039247953
+      std_advantage:                            0.03962741
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00043713543
+      std_entropy:                              0.0038768288
+      std_grad_norm_pre_clip:                   0.27178395
+      std_loss:                                 0.00037411306
+  eval-batch-orig:
+    avg_avg_episode_length:                     17.141367
+    avg_avg_return:                             0.9438763
+    avg_benchmark_regret:                       0.040950205
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.0946579
+    proxy_corner:
+      avg_avg_return:                           0.9438763
+      avg_reward_per_step:                      0.0946579
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.32049
+    avg_avg_return:                             0.27040866
+    avg_benchmark_regret:                       0.7164818
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.038742065
+    proxy_corner:
+      avg_avg_return:                           0.606288
+      avg_reward_per_step:                      0.0051574707
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.673162
+    avg_avg_return:                             0.9417589
+    avg_benchmark_regret:                       0.042316753
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11154938
+    proxy_corner:
+      avg_avg_return:                           0.9417589
+      avg_reward_per_step:                      0.11154938
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        156237824
+    env-step-replay-before:                     79462400
+    env-step-used-before:                       156237824
+    ppo-update-before:                          95360
+    env-step-generate-before:                   76775424
+    env-step-all-after:                         156270592
+    env-step-replay-after:                      79495168
+    env-step-used-after:                        156270592
+    ppo-update-after:                           95380
+    env-step-generate-after:                    76775424
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     53.902058
+    avg_avg_return:                             0.5755948
+    avg_reward_per_step:                        0.01675415
+    proxy_corner:
+      avg_avg_return:                           0.5755948
+      avg_reward_per_step:                      0.01675415
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     53.902058
+    avg_avg_return:                             0.5755948
+    avg_reward_per_step:                        0.01675415
+    proxy_corner:
+      avg_avg_return:                           0.5755948
+      avg_reward_per_step:                      0.01675415
+  perf:
+    env_steps_per_second:                       374534.20530020545
+    ppo_updates_per_second:                     42.21428251305253
+    eval-batch-orig-duration:                   0.4985189735889435
+    eval-batch-shift-duration:                  0.4237236836925149
+    eval-batch-tree-duration:                   0.450400710105896
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.883057
+      mouse-cheese_dist_finite_wavg:            20.080746
+      mouse-cheese_dist_solvable_avg:           29.883057
+      mouse-cheese_dist_solvable_wavg:          20.080746
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.41437337
+    visit_patterns:
+      avg_first_visit_time:                     2064.2126
+      avg_last_visit_time:                      2416.249
+      num_replay_batches:                       2426
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014151314
+    avg_actor_approxkl3:                        0.0013784264
+    avg_actor_clipfrac:                         0.039483644
+    avg_actor_loss:                             -0.00019302257
+    avg_advantage:                              -0.007024645
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.004713226
+    avg_entropy:                                0.11879795
+    avg_grad_norm_pre_clip:                     1.6409638
+    avg_loss:                                   0.0020447925
+    max:
+      max_actor_approxkl1:                      0.0038619405
+      max_actor_approxkl3:                      0.0027542466
+      max_actor_clipfrac:                       0.0769043
+      max_actor_loss:                           0.0021079257
+      max_advantage:                            0.1476177
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.007855747
+      max_entropy:                              0.14189275
+      max_grad_norm_pre_clip:                   2.769129
+      max_loss:                                 0.0038369785
+    std:
+      std_actor_approxkl1:                      0.0012211169
+      std_actor_approxkl3:                      0.0008440373
+      std_actor_clipfrac:                       0.018890454
+      std_actor_loss:                           0.0008987352
+      std_advantage:                            0.09696734
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0015537016
+      std_entropy:                              0.012127938
+      std_grad_norm_pre_clip:                   0.6040718
+      std_loss:                                 0.0010437428
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.626148
+    avg_avg_return:                             0.94718814
+    avg_benchmark_regret:                       0.03763838
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09525299
+    proxy_corner:
+      avg_avg_return:                           0.94718814
+      avg_reward_per_step:                      0.09525299
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.172554
+    avg_avg_return:                             0.27497467
+    avg_benchmark_regret:                       0.7119158
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03907013
+    proxy_corner:
+      avg_avg_return:                           0.60681105
+      avg_reward_per_step:                      0.0054473877
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.647432
+    avg_avg_return:                             0.93903136
+    avg_benchmark_regret:                       0.045044266
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11161041
+    proxy_corner:
+      avg_avg_return:                           0.93903136
+      avg_reward_per_step:                      0.11161041
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        157286400
+    env-step-replay-before:                     80084992
+    env-step-used-before:                       157286400
+    ppo-update-before:                          96000
+    env-step-generate-before:                   77201408
+    env-step-all-after:                         157319168
+    env-step-replay-after:                      80117760
+    env-step-used-after:                        157319168
+    ppo-update-after:                           96020
+    env-step-generate-after:                    77201408
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.43206
+    avg_avg_return:                             0.548391
+    avg_reward_per_step:                        0.014678955
+    proxy_corner:
+      avg_avg_return:                           0.548391
+      avg_reward_per_step:                      0.014678955
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     58.43206
+    avg_avg_return:                             0.548391
+    avg_reward_per_step:                        0.014678955
+    proxy_corner:
+      avg_avg_return:                           0.548391
+      avg_reward_per_step:                      0.014678955
+  perf:
+    env_steps_per_second:                       725901.6585479216
+    ppo_updates_per_second:                     42.1501254598469
+    eval-batch-orig-duration:                   0.3115091500803828
+    eval-batch-shift-duration:                  0.3623953191563487
+    eval-batch-tree-duration:                   0.4209781205281615
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.991455
+      mouse-cheese_dist_finite_wavg:            19.201214
+      mouse-cheese_dist_solvable_avg:           29.991455
+      mouse-cheese_dist_solvable_wavg:          19.201216
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.420168
+    scoring:
+      avg_scores:                               0.40897137
+    visit_patterns:
+      avg_first_visit_time:                     2076.0308
+      avg_last_visit_time:                      2435.2344
+      num_replay_batches:                       2445
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0008593394
+    avg_actor_approxkl3:                        0.001031508
+    avg_actor_clipfrac:                         0.039093018
+    avg_actor_loss:                             -0.0007875361
+    avg_advantage:                              -0.017777165
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0051555964
+    avg_entropy:                                0.13367437
+    avg_grad_norm_pre_clip:                     1.3043798
+    avg_loss:                                   0.0016565878
+    max:
+      max_actor_approxkl1:                      0.002115993
+      max_actor_approxkl3:                      0.0019106478
+      max_actor_clipfrac:                       0.059326172
+      max_actor_loss:                           0.0004087314
+      max_advantage:                            0.1090824
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.010887438
+      max_entropy:                              0.16302264
+      max_grad_norm_pre_clip:                   1.9925079
+      max_loss:                                 0.0052810246
+    std:
+      std_actor_approxkl1:                      0.0006822132
+      std_actor_approxkl3:                      0.0005295667
+      std_actor_clipfrac:                       0.015785718
+      std_actor_loss:                           0.0007833668
+      std_advantage:                            0.10074977
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0017296607
+      std_entropy:                              0.009736135
+      std_grad_norm_pre_clip:                   0.40046427
+      std_loss:                                 0.0010409873
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.589275
+    avg_avg_return:                             0.9481485
+    avg_benchmark_regret:                       0.036677986
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09526825
+    proxy_corner:
+      avg_avg_return:                           0.9481485
+      avg_reward_per_step:                      0.09526825
+  eval-batch-shift:
+    avg_avg_episode_length:                     65.41009
+    avg_avg_return:                             0.28927064
+    avg_benchmark_regret:                       0.6976198
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03956604
+    proxy_corner:
+      avg_avg_return:                           0.5741686
+      avg_reward_per_step:                      0.00504303
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.411888
+    avg_avg_return:                             0.9408597
+    avg_benchmark_regret:                       0.043215945
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11180878
+    proxy_corner:
+      avg_avg_return:                           0.9408597
+      avg_reward_per_step:                      0.11180878
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        158334976
+    env-step-replay-before:                     80642048
+    env-step-used-before:                       158334976
+    ppo-update-before:                          96640
+    env-step-generate-before:                   77692928
+    env-step-all-after:                         158367744
+    env-step-replay-after:                      80642048
+    env-step-used-after:                        158367744
+    ppo-update-after:                           96660
+    env-step-generate-after:                    77725696
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     20.426647
+    avg_avg_return:                             0.8435416
+    avg_reward_per_step:                        0.09118652
+    proxy_corner:
+      avg_avg_return:                           0.8435416
+      avg_reward_per_step:                      0.09118652
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     20.426647
+    avg_avg_return:                             0.8435416
+    avg_reward_per_step:                        0.09118652
+    proxy_corner:
+      avg_avg_return:                           0.8435416
+      avg_reward_per_step:                      0.09118652
+  perf:
+    env_steps_per_second:                       241968.90187017963
+    ppo_updates_per_second:                     42.36279640157587
+    eval-batch-orig-duration:                   0.6760507570579648
+    eval-batch-shift-duration:                  0.6032064631581306
+    eval-batch-tree-duration:                   0.6180612966418266
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             29.988525
+      mouse-cheese_dist_finite_wavg:            16.502792
+      mouse-cheese_dist_solvable_avg:           29.988525
+      mouse-cheese_dist_solvable_wavg:          16.502792
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.41392034
+    visit_patterns:
+      avg_first_visit_time:                     2090.3918
+      avg_last_visit_time:                      2451.1855
+      num_replay_batches:                       2461
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00084408105
+    avg_actor_approxkl3:                        0.0008962145
+    avg_actor_clipfrac:                         0.015264893
+    avg_actor_loss:                             -0.00010922824
+    avg_advantage:                              -0.004326296
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00091473165
+    avg_entropy:                                0.035003196
+    avg_grad_norm_pre_clip:                     1.0931048
+    avg_loss:                                   0.0003131343
+    max:
+      max_actor_approxkl1:                      0.003049587
+      max_actor_approxkl3:                      0.0021576043
+      max_actor_clipfrac:                       0.028442383
+      max_actor_loss:                           0.0018439572
+      max_advantage:                            0.11910796
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0019493969
+      max_entropy:                              0.04953433
+      max_grad_norm_pre_clip:                   2.2976522
+      max_loss:                                 0.0021920057
+    std:
+      std_actor_approxkl1:                      0.0009309033
+      std_actor_approxkl3:                      0.00061001367
+      std_actor_clipfrac:                       0.007267181
+      std_actor_loss:                           0.00095412764
+      std_advantage:                            0.042904805
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00047800393
+      std_entropy:                              0.0062942477
+      std_grad_norm_pre_clip:                   0.5686226
+      std_loss:                                 0.00095397636
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.771238
+    avg_avg_return:                             0.944632
+    avg_benchmark_regret:                       0.040194526
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09513092
+    proxy_corner:
+      avg_avg_return:                           0.944632
+      avg_reward_per_step:                      0.09513092
+  eval-batch-shift:
+    avg_avg_episode_length:                     65.65823
+    avg_avg_return:                             0.28329867
+    avg_benchmark_regret:                       0.70359176
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.04018402
+    proxy_corner:
+      avg_avg_return:                           0.5819992
+      avg_reward_per_step:                      0.0050811768
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.686586
+    avg_avg_return:                             0.9392824
+    avg_benchmark_regret:                       0.04479322
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.111328125
+    proxy_corner:
+      avg_avg_return:                           0.9392824
+      avg_reward_per_step:                      0.111328125
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        159383552
+    env-step-replay-before:                     81133568
+    env-step-used-before:                       159383552
+    ppo-update-before:                          97280
+    env-step-generate-before:                   78249984
+    env-step-all-after:                         159416320
+    env-step-replay-after:                      81133568
+    env-step-used-after:                        159416320
+    ppo-update-after:                           97300
+    env-step-generate-after:                    78282752
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.04907
+    avg_avg_return:                             0.86084497
+    avg_reward_per_step:                        0.112579346
+    proxy_corner:
+      avg_avg_return:                           0.86084497
+      avg_reward_per_step:                      0.112579346
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.04907
+    avg_avg_return:                             0.86084497
+    avg_reward_per_step:                        0.112579346
+    proxy_corner:
+      avg_avg_return:                           0.86084497
+      avg_reward_per_step:                      0.112579346
+  perf:
+    env_steps_per_second:                       294342.6704646709
+    ppo_updates_per_second:                     42.26045040167245
+    eval-batch-orig-duration:                   0.32916282024234533
+    eval-batch-shift-duration:                  0.2366055240854621
+    eval-batch-tree-duration:                   0.2652669595554471
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.008545
+      mouse-cheese_dist_finite_wavg:            14.694126
+      mouse-cheese_dist_solvable_avg:           30.008545
+      mouse-cheese_dist_solvable_wavg:          14.694126
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.43200815
+    visit_patterns:
+      avg_first_visit_time:                     2107.6072
+      avg_last_visit_time:                      2466.3186
+      num_replay_batches:                       2476
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00034839025
+    avg_actor_approxkl3:                        0.00045325095
+    avg_actor_clipfrac:                         0.012103272
+    avg_actor_loss:                             -7.906077e-05
+    avg_advantage:                              0.0059498264
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00034689813
+    avg_entropy:                                0.031060671
+    avg_grad_norm_pre_clip:                     1.016847
+    avg_loss:                                   6.332763e-05
+    max:
+      max_actor_approxkl1:                      0.0015676315
+      max_actor_approxkl3:                      0.0012102211
+      max_actor_clipfrac:                       0.024414062
+      max_actor_loss:                           0.00094493106
+      max_advantage:                            0.11052817
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0007985306
+      max_entropy:                              0.037711002
+      max_grad_norm_pre_clip:                   1.7261932
+      max_loss:                                 0.0010171017
+    std:
+      std_actor_approxkl1:                      0.000401629
+      std_actor_approxkl3:                      0.00027979873
+      std_actor_clipfrac:                       0.005106316
+      std_actor_loss:                           0.00046565736
+      std_advantage:                            0.026136955
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0002538663
+      std_entropy:                              0.0041870903
+      std_grad_norm_pre_clip:                   0.35507345
+      std_loss:                                 0.00048969075
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.55635
+    avg_avg_return:                             0.9491801
+    avg_benchmark_regret:                       0.03564643
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09547424
+    proxy_corner:
+      avg_avg_return:                           0.9491801
+      avg_reward_per_step:                      0.09547424
+  eval-batch-shift:
+    avg_avg_episode_length:                     65.853096
+    avg_avg_return:                             0.272549
+    avg_benchmark_regret:                       0.71434146
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.041000366
+    proxy_corner:
+      avg_avg_return:                           0.5779629
+      avg_reward_per_step:                      0.0051193237
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.706835
+    avg_avg_return:                             0.93975794
+    avg_benchmark_regret:                       0.044317715
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.111564636
+    proxy_corner:
+      avg_avg_return:                           0.93975794
+      avg_reward_per_step:                      0.111564636
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        160432128
+    env-step-replay-before:                     81625088
+    env-step-used-before:                       160432128
+    ppo-update-before:                          97920
+    env-step-generate-before:                   78807040
+    env-step-all-after:                         160464896
+    env-step-replay-after:                      81657856
+    env-step-used-after:                        160464896
+    ppo-update-after:                           97940
+    env-step-generate-after:                    78807040
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     55.967552
+    avg_avg_return:                             0.5678366
+    avg_reward_per_step:                        0.016082764
+    proxy_corner:
+      avg_avg_return:                           0.5678366
+      avg_reward_per_step:                      0.016082764
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     55.967552
+    avg_avg_return:                             0.5678366
+    avg_reward_per_step:                        0.016082764
+    proxy_corner:
+      avg_avg_return:                           0.5678366
+      avg_reward_per_step:                      0.016082764
+  perf:
+    env_steps_per_second:                       321628.6516983192
+    ppo_updates_per_second:                     40.899535560382915
+    eval-batch-orig-duration:                   0.5257293153554201
+    eval-batch-shift-duration:                  0.4815281508490443
+    eval-batch-tree-duration:                   0.48852200247347355
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.074951
+      mouse-cheese_dist_finite_wavg:            16.5238
+      mouse-cheese_dist_solvable_avg:           30.074951
+      mouse-cheese_dist_solvable_wavg:          16.523802
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.4279917
+    visit_patterns:
+      avg_first_visit_time:                     2122.8062
+      avg_last_visit_time:                      2482.213
+      num_replay_batches:                       2492
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0025227205
+    avg_actor_approxkl3:                        0.0022391228
+    avg_actor_clipfrac:                         0.061602782
+    avg_actor_loss:                             0.0005380629
+    avg_advantage:                              -0.0072745536
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.004436941
+    avg_entropy:                                0.13504145
+    avg_grad_norm_pre_clip:                     2.0845017
+    avg_loss:                                   0.002621492
+    max:
+      max_actor_approxkl1:                      0.0044953413
+      max_actor_approxkl3:                      0.005554951
+      max_actor_clipfrac:                       0.08666992
+      max_actor_loss:                           0.006888345
+      max_advantage:                            0.12201685
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0067919246
+      max_entropy:                              0.15124896
+      max_grad_norm_pre_clip:                   4.5914197
+      max_loss:                                 0.009133562
+    std:
+      std_actor_approxkl1:                      0.0011955018
+      std_actor_approxkl3:                      0.0012232284
+      std_actor_clipfrac:                       0.019211799
+      std_actor_loss:                           0.0019880948
+      std_advantage:                            0.0940371
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0009923653
+      std_entropy:                              0.010150597
+      std_grad_norm_pre_clip:                   0.87522894
+      std_loss:                                 0.0020295621
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.402699
+    avg_avg_return:                             0.9459368
+    avg_benchmark_regret:                       0.038889736
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09529877
+    proxy_corner:
+      avg_avg_return:                           0.9459368
+      avg_reward_per_step:                      0.09529877
+  eval-batch-shift:
+    avg_avg_episode_length:                     65.0304
+    avg_avg_return:                             0.28988972
+    avg_benchmark_regret:                       0.69700074
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.040359497
+    proxy_corner:
+      avg_avg_return:                           0.57108927
+      avg_reward_per_step:                      0.0052261353
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.550741
+    avg_avg_return:                             0.94219077
+    avg_benchmark_regret:                       0.041884784
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11110687
+    proxy_corner:
+      avg_avg_return:                           0.94219077
+      avg_reward_per_step:                      0.11110687
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        161480704
+    env-step-replay-before:                     82149376
+    env-step-used-before:                       161480704
+    ppo-update-before:                          98560
+    env-step-generate-before:                   79331328
+    env-step-all-after:                         161513472
+    env-step-replay-after:                      82149376
+    env-step-used-after:                        161513472
+    ppo-update-after:                           98580
+    env-step-generate-after:                    79364096
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     17.065104
+    avg_avg_return:                             0.8762467
+    avg_reward_per_step:                        0.11477661
+    proxy_corner:
+      avg_avg_return:                           0.8762467
+      avg_reward_per_step:                      0.11477661
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     17.065104
+    avg_avg_return:                             0.8762467
+    avg_reward_per_step:                        0.11477661
+    proxy_corner:
+      avg_avg_return:                           0.8762467
+      avg_reward_per_step:                      0.11477661
+  perf:
+    env_steps_per_second:                       317845.70758520404
+    ppo_updates_per_second:                     42.07995213771184
+    eval-batch-orig-duration:                   0.4992723762989044
+    eval-batch-shift-duration:                  0.42200432904064655
+    eval-batch-tree-duration:                   0.4266335591673851
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.177246
+      mouse-cheese_dist_finite_wavg:            16.50997
+      mouse-cheese_dist_solvable_avg:           30.177246
+      mouse-cheese_dist_solvable_wavg:          16.509972
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.43205315
+    visit_patterns:
+      avg_first_visit_time:                     2138.1587
+      avg_last_visit_time:                      2497.3083
+      num_replay_batches:                       2507
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00065737934
+    avg_actor_approxkl3:                        0.0005508389
+    avg_actor_clipfrac:                         0.0108154295
+    avg_actor_loss:                             6.5341876e-05
+    avg_advantage:                              0.020837335
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00038641726
+    avg_entropy:                                0.024044475
+    avg_grad_norm_pre_clip:                     0.7589363
+    avg_loss:                                   0.00023450598
+    max:
+      max_actor_approxkl1:                      0.0017198296
+      max_actor_approxkl3:                      0.0013804743
+      max_actor_clipfrac:                       0.02331543
+      max_actor_loss:                           0.001448499
+      max_advantage:                            0.12163335
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0010022579
+      max_entropy:                              0.039017856
+      max_grad_norm_pre_clip:                   1.640656
+      max_loss:                                 0.0015349091
+    std:
+      std_actor_approxkl1:                      0.0006545993
+      std_actor_approxkl3:                      0.00040064266
+      std_actor_clipfrac:                       0.0061245244
+      std_actor_loss:                           0.0005897753
+      std_advantage:                            0.026507022
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0002877844
+      std_entropy:                              0.0066838167
+      std_grad_norm_pre_clip:                   0.39696816
+      std_loss:                                 0.00056345464
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.400164
+    avg_avg_return:                             0.9481746
+    avg_benchmark_regret:                       0.036651947
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09513092
+    proxy_corner:
+      avg_avg_return:                           0.9481746
+      avg_reward_per_step:                      0.09513092
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.646576
+    avg_avg_return:                             0.26601475
+    avg_benchmark_regret:                       0.7208757
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.039146423
+    proxy_corner:
+      avg_avg_return:                           0.5688294
+      avg_reward_per_step:                      0.0048065186
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.072315
+    avg_avg_return:                             0.94420344
+    avg_benchmark_regret:                       0.039872155
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.111946106
+    proxy_corner:
+      avg_avg_return:                           0.94420344
+      avg_reward_per_step:                      0.111946106
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        162529280
+    env-step-replay-before:                     82542592
+    env-step-used-before:                       162529280
+    ppo-update-before:                          99200
+    env-step-generate-before:                   79986688
+    env-step-all-after:                         162562048
+    env-step-replay-after:                      82542592
+    env-step-used-after:                        162562048
+    ppo-update-after:                           99220
+    env-step-generate-after:                    80019456
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     21.015398
+    avg_avg_return:                             0.8455558
+    avg_reward_per_step:                        0.09429932
+    proxy_corner:
+      avg_avg_return:                           0.8455558
+      avg_reward_per_step:                      0.09429932
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     21.015398
+    avg_avg_return:                             0.8455558
+    avg_reward_per_step:                        0.09429932
+    proxy_corner:
+      avg_avg_return:                           0.8455558
+      avg_reward_per_step:                      0.09429932
+  perf:
+    env_steps_per_second:                       322643.3491774387
+    ppo_updates_per_second:                     41.06940236804056
+    eval-batch-orig-duration:                   0.5307673262432218
+    eval-batch-shift-duration:                  0.4820273211225867
+    eval-batch-tree-duration:                   0.4537642765790224
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.107178
+      mouse-cheese_dist_finite_wavg:            16.524488
+      mouse-cheese_dist_solvable_avg:           30.107178
+      mouse-cheese_dist_solvable_wavg:          16.524487
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.4257987
+    visit_patterns:
+      avg_first_visit_time:                     2151.9812
+      avg_last_visit_time:                      2509.2878
+      num_replay_batches:                       2519
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00092805363
+    avg_actor_approxkl3:                        0.0011344322
+    avg_actor_clipfrac:                         0.016656494
+    avg_actor_loss:                             0.00034549678
+    avg_advantage:                              -4.6218163e-05
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00049164717
+    avg_entropy:                                0.02821139
+    avg_grad_norm_pre_clip:                     1.3143477
+    avg_loss:                                   0.000563109
+    max:
+      max_actor_approxkl1:                      0.004218138
+      max_actor_approxkl3:                      0.0032595624
+      max_actor_clipfrac:                       0.028076172
+      max_actor_loss:                           0.0026717074
+      max_advantage:                            0.09218979
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0008111137
+      max_entropy:                              0.04138811
+      max_grad_norm_pre_clip:                   2.1838212
+      max_loss:                                 0.0030323116
+    std:
+      std_actor_approxkl1:                      0.0011300498
+      std_actor_approxkl3:                      0.0008814446
+      std_actor_clipfrac:                       0.0073489444
+      std_actor_loss:                           0.00087450224
+      std_advantage:                            0.031273667
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00029168976
+      std_entropy:                              0.0048798285
+      std_grad_norm_pre_clip:                   0.5950124
+      std_loss:                                 0.0008996107
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.777552
+    avg_avg_return:                             0.9432013
+    avg_benchmark_regret:                       0.04162518
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09513855
+    proxy_corner:
+      avg_avg_return:                           0.9432013
+      avg_reward_per_step:                      0.09513855
+  eval-batch-shift:
+    avg_avg_episode_length:                     65.50519
+    avg_avg_return:                             0.2792675
+    avg_benchmark_regret:                       0.707623
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.04147339
+    proxy_corner:
+      avg_avg_return:                           0.5483159
+      avg_reward_per_step:                      0.004951477
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.817947
+    avg_avg_return:                             0.9412632
+    avg_benchmark_regret:                       0.04281237
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.1118927
+    proxy_corner:
+      avg_avg_return:                           0.9412632
+      avg_reward_per_step:                      0.1118927
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        163577856
+    env-step-replay-before:                     83132416
+    env-step-used-before:                       163577856
+    ppo-update-before:                          99840
+    env-step-generate-before:                   80445440
+    env-step-all-after:                         163610624
+    env-step-replay-after:                      83165184
+    env-step-used-after:                        163610624
+    ppo-update-after:                           99860
+    env-step-generate-after:                    80445440
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     53.24778
+    avg_avg_return:                             0.600669
+    avg_reward_per_step:                        0.017242432
+    proxy_corner:
+      avg_avg_return:                           0.600669
+      avg_reward_per_step:                      0.017242432
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     53.24778
+    avg_avg_return:                             0.600669
+    avg_reward_per_step:                        0.017242432
+    proxy_corner:
+      avg_avg_return:                           0.600669
+      avg_reward_per_step:                      0.017242432
+  perf:
+    env_steps_per_second:                       320906.7476034834
+    ppo_updates_per_second:                     41.42281011868023
+    eval-batch-orig-duration:                   0.6633250070735812
+    eval-batch-shift-duration:                  0.6460183439776301
+    eval-batch-tree-duration:                   0.6111833034083247
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.102783
+      mouse-cheese_dist_finite_wavg:            16.522861
+      mouse-cheese_dist_solvable_avg:           30.102783
+      mouse-cheese_dist_solvable_wavg:          16.522865
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.41353184
+    visit_patterns:
+      avg_first_visit_time:                     2163.7139
+      avg_last_visit_time:                      2528.2507
+      num_replay_batches:                       2538
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00078769325
+    avg_actor_approxkl3:                        0.0012772143
+    avg_actor_clipfrac:                         0.039221194
+    avg_actor_loss:                             -0.0005131877
+    avg_advantage:                              -0.013706729
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0031902082
+    avg_entropy:                                0.11173129
+    avg_grad_norm_pre_clip:                     1.1669666
+    avg_loss:                                   0.00097018527
+    max:
+      max_actor_approxkl1:                      0.001860044
+      max_actor_approxkl3:                      0.0025439332
+      max_actor_clipfrac:                       0.056884766
+      max_actor_loss:                           0.0006342977
+      max_advantage:                            0.13084185
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0049435827
+      max_entropy:                              0.12382636
+      max_grad_norm_pre_clip:                   1.7418009
+      max_loss:                                 0.0022872845
+    std:
+      std_actor_approxkl1:                      0.00056632684
+      std_actor_approxkl3:                      0.00067860604
+      std_actor_clipfrac:                       0.014159898
+      std_actor_loss:                           0.0006312748
+      std_advantage:                            0.07945868
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010114542
+      std_entropy:                              0.006445576
+      std_grad_norm_pre_clip:                   0.3211407
+      std_loss:                                 0.0005919701
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.734552
+    avg_avg_return:                             0.9460778
+    avg_benchmark_regret:                       0.038748726
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09477234
+    proxy_corner:
+      avg_avg_return:                           0.9460778
+      avg_reward_per_step:                      0.09477234
+  eval-batch-shift:
+    avg_avg_episode_length:                     65.72796
+    avg_avg_return:                             0.28607628
+    avg_benchmark_regret:                       0.7008142
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.039855957
+    proxy_corner:
+      avg_avg_return:                           0.62564766
+      avg_reward_per_step:                      0.0054626465
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.175404
+    avg_avg_return:                             0.9454135
+    avg_benchmark_regret:                       0.038662154
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.111846924
+    proxy_corner:
+      avg_avg_return:                           0.9454135
+      avg_reward_per_step:                      0.111846924
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        164626432
+    env-step-replay-before:                     83722240
+    env-step-used-before:                       164626432
+    ppo-update-before:                          100480
+    env-step-generate-before:                   80904192
+    env-step-all-after:                         164659200
+    env-step-replay-after:                      83755008
+    env-step-used-after:                        164659200
+    ppo-update-after:                           100500
+    env-step-generate-after:                    80904192
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     52.674644
+    avg_avg_return:                             0.6034601
+    avg_reward_per_step:                        0.017700195
+    proxy_corner:
+      avg_avg_return:                           0.6034601
+      avg_reward_per_step:                      0.017700195
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     52.674644
+    avg_avg_return:                             0.6034601
+    avg_reward_per_step:                        0.017700195
+    proxy_corner:
+      avg_avg_return:                           0.6034601
+      avg_reward_per_step:                      0.017700195
+  perf:
+    env_steps_per_second:                       254444.6792670798
+    ppo_updates_per_second:                     42.37349604243036
+    eval-batch-orig-duration:                   0.6352279391139746
+    eval-batch-shift-duration:                  0.6432391405105591
+    eval-batch-tree-duration:                   0.6168215125799179
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.192139
+      mouse-cheese_dist_finite_wavg:            20.105587
+      mouse-cheese_dist_solvable_avg:           30.192139
+      mouse-cheese_dist_solvable_wavg:          20.105587
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016816
+    scoring:
+      avg_scores:                               0.390047
+    visit_patterns:
+      avg_first_visit_time:                     2177.1582
+      avg_last_visit_time:                      2546.1912
+      num_replay_batches:                       2556
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0018751431
+    avg_actor_approxkl3:                        0.0018679006
+    avg_actor_clipfrac:                         0.04863281
+    avg_actor_loss:                             9.256657e-05
+    avg_advantage:                              -0.008575521
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.003242839
+    avg_entropy:                                0.116544046
+    avg_grad_norm_pre_clip:                     1.531829
+    avg_loss:                                   0.0015974421
+    max:
+      max_actor_approxkl1:                      0.0051399325
+      max_actor_approxkl3:                      0.00416499
+      max_actor_clipfrac:                       0.075805664
+      max_actor_loss:                           0.0025706217
+      max_advantage:                            0.12341529
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0057473914
+      max_entropy:                              0.14140691
+      max_grad_norm_pre_clip:                   2.6742864
+      max_loss:                                 0.004016194
+    std:
+      std_actor_approxkl1:                      0.00126453
+      std_actor_approxkl3:                      0.0009074363
+      std_actor_clipfrac:                       0.015907565
+      std_actor_loss:                           0.0010815463
+      std_advantage:                            0.08017053
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0012976591
+      std_entropy:                              0.009070681
+      std_grad_norm_pre_clip:                   0.53973395
+      std_loss:                                 0.001221668
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.564404
+    avg_avg_return:                             0.94795984
+    avg_benchmark_regret:                       0.036866643
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09526825
+    proxy_corner:
+      avg_avg_return:                           0.94795984
+      avg_reward_per_step:                      0.09526825
+  eval-batch-shift:
+    avg_avg_episode_length:                     65.55931
+    avg_avg_return:                             0.288397
+    avg_benchmark_regret:                       0.6984934
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03932953
+    proxy_corner:
+      avg_avg_return:                           0.62379247
+      avg_reward_per_step:                      0.0055389404
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.234846
+    avg_avg_return:                             0.946911
+    avg_benchmark_regret:                       0.037164636
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.111968994
+    proxy_corner:
+      avg_avg_return:                           0.946911
+      avg_reward_per_step:                      0.111968994
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        165675008
+    env-step-replay-before:                     84344832
+    env-step-used-before:                       165675008
+    ppo-update-before:                          101120
+    env-step-generate-before:                   81330176
+    env-step-all-after:                         165707776
+    env-step-replay-after:                      84377600
+    env-step-used-after:                        165707776
+    ppo-update-after:                           101140
+    env-step-generate-after:                    81330176
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     54.61252
+    avg_avg_return:                             0.57300824
+    avg_reward_per_step:                        0.01638794
+    proxy_corner:
+      avg_avg_return:                           0.57300824
+      avg_reward_per_step:                      0.01638794
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     54.61252
+    avg_avg_return:                             0.57300824
+    avg_reward_per_step:                        0.01638794
+    proxy_corner:
+      avg_avg_return:                           0.57300824
+      avg_reward_per_step:                      0.01638794
+  perf:
+    env_steps_per_second:                       298502.0342263442
+    ppo_updates_per_second:                     42.1158679754936
+    eval-batch-orig-duration:                   0.5287964446470141
+    eval-batch-shift-duration:                  0.4752955948933959
+    eval-batch-tree-duration:                   0.4420120567083359
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.273438
+      mouse-cheese_dist_finite_wavg:            16.519482
+      mouse-cheese_dist_solvable_avg:           30.273438
+      mouse-cheese_dist_solvable_wavg:          16.519484
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.38932955
+    visit_patterns:
+      avg_first_visit_time:                     2188.729
+      avg_last_visit_time:                      2565.253
+      num_replay_batches:                       2575
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0018069219
+    avg_actor_approxkl3:                        0.0015019615
+    avg_actor_clipfrac:                         0.044226076
+    avg_actor_loss:                             4.591979e-06
+    avg_advantage:                              -0.010394843
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.004503387
+    avg_entropy:                                0.11881133
+    avg_grad_norm_pre_clip:                     1.6739355
+    avg_loss:                                   0.0021374736
+    max:
+      max_actor_approxkl1:                      0.004082244
+      max_actor_approxkl3:                      0.0033786471
+      max_actor_clipfrac:                       0.08093262
+      max_actor_loss:                           0.0014436468
+      max_advantage:                            0.106356084
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00669313
+      max_entropy:                              0.14052227
+      max_grad_norm_pre_clip:                   2.6063454
+      max_loss:                                 0.00440409
+    std:
+      std_actor_approxkl1:                      0.0010047224
+      std_actor_approxkl3:                      0.0007591216
+      std_actor_clipfrac:                       0.016155493
+      std_actor_loss:                           0.0009513342
+      std_advantage:                            0.095032535
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0012525215
+      std_entropy:                              0.011124257
+      std_grad_norm_pre_clip:                   0.4776586
+      std_loss:                                 0.0011653836
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.459587
+    avg_avg_return:                             0.9460111
+    avg_benchmark_regret:                       0.03881538
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.095451355
+    proxy_corner:
+      avg_avg_return:                           0.9460111
+      avg_reward_per_step:                      0.095451355
+  eval-batch-shift:
+    avg_avg_episode_length:                     63.924644
+    avg_avg_return:                             0.29489005
+    avg_benchmark_regret:                       0.69200045
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.04270935
+    proxy_corner:
+      avg_avg_return:                           0.5922649
+      avg_reward_per_step:                      0.0054016113
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.480627
+    avg_avg_return:                             0.94416976
+    avg_benchmark_regret:                       0.03990587
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11166382
+    proxy_corner:
+      avg_avg_return:                           0.94416976
+      avg_reward_per_step:                      0.11166382
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        166723584
+    env-step-replay-before:                     84934656
+    env-step-used-before:                       166723584
+    ppo-update-before:                          101760
+    env-step-generate-before:                   81788928
+    env-step-all-after:                         166756352
+    env-step-replay-after:                      84967424
+    env-step-used-after:                        166756352
+    ppo-update-after:                           101780
+    env-step-generate-after:                    81788928
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     53.505524
+    avg_avg_return:                             0.5964327
+    avg_reward_per_step:                        0.016937256
+    proxy_corner:
+      avg_avg_return:                           0.5964327
+      avg_reward_per_step:                      0.016937256
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     53.505524
+    avg_avg_return:                             0.5964327
+    avg_reward_per_step:                        0.016937256
+    proxy_corner:
+      avg_avg_return:                           0.5964327
+      avg_reward_per_step:                      0.016937256
+  perf:
+    env_steps_per_second:                       293218.4556133923
+    ppo_updates_per_second:                     44.41579647306397
+    eval-batch-orig-duration:                   0.5134204188361764
+    eval-batch-shift-duration:                  0.4890401288866997
+    eval-batch-tree-duration:                   0.4895932571962476
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.380127
+      mouse-cheese_dist_finite_wavg:            19.235987
+      mouse-cheese_dist_solvable_avg:           30.380127
+      mouse-cheese_dist_solvable_wavg:          19.235989
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.39454132
+    visit_patterns:
+      avg_first_visit_time:                     2203.723
+      avg_last_visit_time:                      2583.301
+      num_replay_batches:                       2593
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0016839044
+    avg_actor_approxkl3:                        0.0019450333
+    avg_actor_clipfrac:                         0.047912598
+    avg_actor_loss:                             0.0005826216
+    avg_advantage:                              -0.012387532
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0035686872
+    avg_entropy:                                0.1199064
+    avg_grad_norm_pre_clip:                     1.7564589
+    avg_loss:                                   0.0022470588
+    max:
+      max_actor_approxkl1:                      0.006467876
+      max_actor_approxkl3:                      0.005010618
+      max_actor_clipfrac:                       0.08642578
+      max_actor_loss:                           0.003384143
+      max_advantage:                            0.10644406
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.006843236
+      max_entropy:                              0.13250402
+      max_grad_norm_pre_clip:                   3.593223
+      max_loss:                                 0.0065193023
+    std:
+      std_actor_approxkl1:                      0.0014570904
+      std_actor_approxkl3:                      0.0012153658
+      std_actor_clipfrac:                       0.017962782
+      std_actor_loss:                           0.001090731
+      std_advantage:                            0.08370006
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0013634282
+      std_entropy:                              0.008082108
+      std_grad_norm_pre_clip:                   0.6803316
+      std_loss:                                 0.0012046112
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.505432
+    avg_avg_return:                             0.9472016
+    avg_benchmark_regret:                       0.03762487
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09577179
+    proxy_corner:
+      avg_avg_return:                           0.9472016
+      avg_reward_per_step:                      0.09577179
+  eval-batch-shift:
+    avg_avg_episode_length:                     63.417194
+    avg_avg_return:                             0.30580884
+    avg_benchmark_regret:                       0.68108165
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.04168701
+    proxy_corner:
+      avg_avg_return:                           0.589551
+      avg_reward_per_step:                      0.005378723
+  eval-batch-tree:
+    avg_avg_episode_length:                     16.946846
+    avg_avg_return:                             0.94518876
+    avg_benchmark_regret:                       0.038886875
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11226654
+    proxy_corner:
+      avg_avg_return:                           0.94518876
+      avg_reward_per_step:                      0.11226654
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        167772160
+    env-step-replay-before:                     85458944
+    env-step-used-before:                       167772160
+    ppo-update-before:                          102400
+    env-step-generate-before:                   82313216
+    env-step-all-after:                         167804928
+    env-step-replay-after:                      85491712
+    env-step-used-after:                        167804928
+    ppo-update-after:                           102420
+    env-step-generate-after:                    82313216
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     52.58859
+    avg_avg_return:                             0.5925534
+    avg_reward_per_step:                        0.017364502
+    proxy_corner:
+      avg_avg_return:                           0.5925534
+      avg_reward_per_step:                      0.017364502
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     52.58859
+    avg_avg_return:                             0.5925534
+    avg_reward_per_step:                        0.017364502
+    proxy_corner:
+      avg_avg_return:                           0.5925534
+      avg_reward_per_step:                      0.017364502
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  perf:
+    env_steps_per_second:                       319076.3534532014
+    ppo_updates_per_second:                     42.22810054530078
+    eval-batch-orig-duration:                   0.6595016606152058
+    eval-batch-shift-duration:                  0.6079602437093854
+    eval-batch-tree-duration:                   0.5991362044587731
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.370605
+      mouse-cheese_dist_finite_wavg:            12.94874
+      mouse-cheese_dist_solvable_avg:           30.370605
+      mouse-cheese_dist_solvable_wavg:          12.948742
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.39591363
+    visit_patterns:
+      avg_first_visit_time:                     2216.5803
+      avg_last_visit_time:                      2599.2292
+      num_replay_batches:                       2609
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0015689528
+    avg_actor_approxkl3:                        0.001172939
+    avg_actor_clipfrac:                         0.03647461
+    avg_actor_loss:                             -0.00053406414
+    avg_advantage:                              -4.9151015e-05
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0040339525
+    avg_entropy:                                0.10928476
+    avg_grad_norm_pre_clip:                     1.0873574
+    avg_loss:                                   0.0013736275
+    max:
+      max_actor_approxkl1:                      0.003266673
+      max_actor_approxkl3:                      0.0028178922
+      max_actor_clipfrac:                       0.052856445
+      max_actor_loss:                           0.0012527555
+      max_advantage:                            0.13688743
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.007873405
+      max_entropy:                              0.11897846
+      max_grad_norm_pre_clip:                   1.9983832
+      max_loss:                                 0.0037672885
+    std:
+      std_actor_approxkl1:                      0.0008209341
+      std_actor_approxkl3:                      0.000608197
+      std_actor_clipfrac:                       0.011849917
+      std_actor_loss:                           0.0008464606
+      std_advantage:                            0.09007013
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0013246117
+      std_entropy:                              0.0053274427
+      std_grad_norm_pre_clip:                   0.37341216
+      std_loss:                                 0.0010203795
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.491806
+    avg_avg_return:                             0.94837105
+    avg_benchmark_regret:                       0.03645543
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.095336914
+    proxy_corner:
+      avg_avg_return:                           0.94837105
+      avg_reward_per_step:                      0.095336914
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-shift:
+    avg_avg_episode_length:                     64.5515
+    avg_avg_return:                             0.29315963
+    avg_benchmark_regret:                       0.69373083
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.04108429
+    proxy_corner:
+      avg_avg_return:                           0.5511373
+      avg_reward_per_step:                      0.00504303
+    rollouts_gif:                               float32(132, 225, 225, 3)
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.3523
+    avg_avg_return:                             0.9434284
+    avg_benchmark_regret:                       0.04064723
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11180878
+    proxy_corner:
+      avg_avg_return:                           0.9434284
+      avg_reward_per_step:                      0.11180878
+    rollouts_gif:                               float32(132, 225, 225, 3)
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        168820736
+    env-step-replay-before:                     85983232
+    env-step-used-before:                       168820736
+    ppo-update-before:                          103040
+    env-step-generate-before:                   82837504
+    env-step-all-after:                         168853504
+    env-step-replay-after:                      85983232
+    env-step-used-after:                        168853504
+    ppo-update-after:                           103060
+    env-step-generate-after:                    82870272
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     17.322533
+    avg_avg_return:                             0.8746191
+    avg_reward_per_step:                        0.12261963
+    proxy_corner:
+      avg_avg_return:                           0.8746191
+      avg_reward_per_step:                      0.12261963
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     17.322533
+    avg_avg_return:                             0.8746191
+    avg_reward_per_step:                        0.12261963
+    proxy_corner:
+      avg_avg_return:                           0.8746191
+      avg_reward_per_step:                      0.12261963
+  perf:
+    env_steps_per_second:                       316095.1429924773
+    ppo_updates_per_second:                     42.289429579013095
+    eval-batch-orig-duration:                   0.5334589621052146
+    eval-batch-shift-duration:                  0.4861545516178012
+    eval-batch-tree-duration:                   0.4933358896523714
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.390137
+      mouse-cheese_dist_finite_wavg:            19.243095
+      mouse-cheese_dist_solvable_avg:           30.390137
+      mouse-cheese_dist_solvable_wavg:          19.243103
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999997
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.38652235
+    visit_patterns:
+      avg_first_visit_time:                     2235.4214
+      avg_last_visit_time:                      2614.2349
+      num_replay_batches:                       2624
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00059604365
+    avg_actor_approxkl3:                        0.00061848795
+    avg_actor_clipfrac:                         0.012310791
+    avg_actor_loss:                             0.0002528366
+    avg_advantage:                              0.016657127
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00032462648
+    avg_entropy:                                0.025598465
+    avg_grad_norm_pre_clip:                     1.0831354
+    avg_loss:                                   0.0003895514
+    max:
+      max_actor_approxkl1:                      0.0018684361
+      max_actor_approxkl3:                      0.0013049424
+      max_actor_clipfrac:                       0.021850586
+      max_actor_loss:                           0.0025638826
+      max_advantage:                            0.11501169
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00043185038
+      max_entropy:                              0.03333508
+      max_grad_norm_pre_clip:                   1.8133081
+      max_loss:                                 0.0026466642
+    std:
+      std_actor_approxkl1:                      0.00053459645
+      std_actor_approxkl3:                      0.00033664165
+      std_actor_clipfrac:                       0.0047854483
+      std_actor_loss:                           0.0010286067
+      std_advantage:                            0.020490449
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          8.644394e-05
+      std_entropy:                              0.0046331193
+      std_grad_norm_pre_clip:                   0.4600868
+      std_loss:                                 0.0010148931
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.372738
+    avg_avg_return:                             0.948351
+    avg_benchmark_regret:                       0.036475457
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09562683
+    proxy_corner:
+      avg_avg_return:                           0.948351
+      avg_reward_per_step:                      0.09562683
+  eval-batch-shift:
+    avg_avg_episode_length:                     65.49248
+    avg_avg_return:                             0.2827195
+    avg_benchmark_regret:                       0.70417094
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.04232025
+    proxy_corner:
+      avg_avg_return:                           0.62753636
+      avg_reward_per_step:                      0.0054626465
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.220585
+    avg_avg_return:                             0.9427393
+    avg_benchmark_regret:                       0.04133631
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.111824036
+    proxy_corner:
+      avg_avg_return:                           0.9427393
+      avg_reward_per_step:                      0.111824036
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        169869312
+    env-step-replay-before:                     86474752
+    env-step-used-before:                       169869312
+    ppo-update-before:                          103680
+    env-step-generate-before:                   83394560
+    env-step-all-after:                         169902080
+    env-step-replay-after:                      86474752
+    env-step-used-after:                        169902080
+    ppo-update-after:                           103700
+    env-step-generate-after:                    83427328
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     16.001963
+    avg_avg_return:                             0.8786726
+    avg_reward_per_step:                        0.113861084
+    proxy_corner:
+      avg_avg_return:                           0.8786726
+      avg_reward_per_step:                      0.113861084
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     16.001963
+    avg_avg_return:                             0.8786726
+    avg_reward_per_step:                        0.113861084
+    proxy_corner:
+      avg_avg_return:                           0.8786726
+      avg_reward_per_step:                      0.113861084
+  perf:
+    env_steps_per_second:                       287966.23321292637
+    ppo_updates_per_second:                     42.21639594607695
+    eval-batch-orig-duration:                   0.5254424903541803
+    eval-batch-shift-duration:                  0.47771461214870214
+    eval-batch-tree-duration:                   0.45613478403538465
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.41504
+      mouse-cheese_dist_finite_wavg:            16.542715
+      mouse-cheese_dist_solvable_avg:           30.41504
+      mouse-cheese_dist_solvable_wavg:          16.542717
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.3937838
+    visit_patterns:
+      avg_first_visit_time:                     2249.0674
+      avg_last_visit_time:                      2629.2832
+      num_replay_batches:                       2639
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.000527267
+    avg_actor_approxkl3:                        0.00049503235
+    avg_actor_clipfrac:                         0.009283448
+    avg_actor_loss:                             -0.00036859047
+    avg_advantage:                              0.03528118
+    avg_critic_clipfrac:                        0.0029357912
+    avg_critic_loss:                            0.00039965851
+    avg_entropy:                                0.020669948
+    avg_grad_norm_pre_clip:                     0.69478077
+    avg_loss:                                   -0.00018943116
+    max:
+      max_actor_approxkl1:                      0.0024854275
+      max_actor_approxkl3:                      0.0014395664
+      max_actor_clipfrac:                       0.016723633
+      max_actor_loss:                           0.0008893497
+      max_advantage:                            0.13378334
+      max_critic_clipfrac:                      0.015380859
+      max_critic_loss:                          0.0009913971
+      max_entropy:                              0.026554478
+      max_grad_norm_pre_clip:                   1.2503364
+      max_loss:                                 0.0010864284
+    std:
+      std_actor_approxkl1:                      0.00061105203
+      std_actor_approxkl3:                      0.0003815605
+      std_actor_clipfrac:                       0.003889137
+      std_actor_loss:                           0.0005041459
+      std_advantage:                            0.025996411
+      std_critic_clipfrac:                      0.0048370464
+      std_critic_loss:                          0.00017329954
+      std_entropy:                              0.003050929
+      std_grad_norm_pre_clip:                   0.25794387
+      std_loss:                                 0.0005364841
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.53187
+    avg_avg_return:                             0.9479563
+    avg_benchmark_regret:                       0.036870185
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.095718384
+    proxy_corner:
+      avg_avg_return:                           0.9479563
+      avg_reward_per_step:                      0.095718384
+  eval-batch-shift:
+    avg_avg_episode_length:                     65.71168
+    avg_avg_return:                             0.27938396
+    avg_benchmark_regret:                       0.70750654
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.040512085
+    proxy_corner:
+      avg_avg_return:                           0.6143639
+      avg_reward_per_step:                      0.0053253174
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.308773
+    avg_avg_return:                             0.94285434
+    avg_benchmark_regret:                       0.041221283
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.112083435
+    proxy_corner:
+      avg_avg_return:                           0.94285434
+      avg_reward_per_step:                      0.112083435
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        170917888
+    env-step-replay-before:                     86900736
+    env-step-used-before:                       170917888
+    ppo-update-before:                          104320
+    env-step-generate-before:                   84017152
+    env-step-all-after:                         170950656
+    env-step-replay-after:                      86900736
+    env-step-used-after:                        170950656
+    ppo-update-after:                           104340
+    env-step-generate-after:                    84049920
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     17.727137
+    avg_avg_return:                             0.86754966
+    avg_reward_per_step:                        0.10205078
+    proxy_corner:
+      avg_avg_return:                           0.86754966
+      avg_reward_per_step:                      0.10205078
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     17.727137
+    avg_avg_return:                             0.86754966
+    avg_reward_per_step:                        0.10205078
+    proxy_corner:
+      avg_avg_return:                           0.86754966
+      avg_reward_per_step:                      0.10205078
+  perf:
+    env_steps_per_second:                       344267.64656434866
+    ppo_updates_per_second:                     40.64234193665495
+    eval-batch-orig-duration:                   0.5200296929106116
+    eval-batch-shift-duration:                  0.4868559939786792
+    eval-batch-tree-duration:                   0.48696436546742916
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.43628
+      mouse-cheese_dist_finite_wavg:            18.351019
+      mouse-cheese_dist_solvable_avg:           30.43628
+      mouse-cheese_dist_solvable_wavg:          18.351019
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.40309072
+    visit_patterns:
+      avg_first_visit_time:                     2262.2983
+      avg_last_visit_time:                      2642.2378
+      num_replay_batches:                       2652
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00039257813
+    avg_actor_approxkl3:                        0.0005398908
+    avg_actor_clipfrac:                         0.010675049
+    avg_actor_loss:                             -3.574997e-05
+    avg_advantage:                              -4.913454e-05
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00045508504
+    avg_entropy:                                0.024999944
+    avg_grad_norm_pre_clip:                     0.80632395
+    avg_loss:                                   0.00016679264
+    max:
+      max_actor_approxkl1:                      0.0014074577
+      max_actor_approxkl3:                      0.001468715
+      max_actor_clipfrac:                       0.019042969
+      max_actor_loss:                           0.0010304013
+      max_advantage:                            0.09359455
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0013422839
+      max_entropy:                              0.03428783
+      max_grad_norm_pre_clip:                   1.4698027
+      max_loss:                                 0.0010866128
+    std:
+      std_actor_approxkl1:                      0.00044275873
+      std_actor_approxkl3:                      0.0003891552
+      std_actor_clipfrac:                       0.0050656204
+      std_actor_loss:                           0.0005369065
+      std_advantage:                            0.030061994
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0004352415
+      std_entropy:                              0.0047539687
+      std_grad_norm_pre_clip:                   0.33028907
+      std_loss:                                 0.00044922406
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.38591
+    avg_avg_return:                             0.94697565
+    avg_benchmark_regret:                       0.037850883
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09599304
+    proxy_corner:
+      avg_avg_return:                           0.94697565
+      avg_reward_per_step:                      0.09599304
+  eval-batch-shift:
+    avg_avg_episode_length:                     65.80536
+    avg_avg_return:                             0.27587408
+    avg_benchmark_regret:                       0.71101636
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.040245056
+    proxy_corner:
+      avg_avg_return:                           0.60178185
+      avg_reward_per_step:                      0.0051651
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.19107
+    avg_avg_return:                             0.942361
+    avg_benchmark_regret:                       0.041714594
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11231995
+    proxy_corner:
+      avg_avg_return:                           0.942361
+      avg_reward_per_step:                      0.11231995
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        171966464
+    env-step-replay-before:                     87457792
+    env-step-used-before:                       171966464
+    ppo-update-before:                          104960
+    env-step-generate-before:                   84508672
+    env-step-all-after:                         171999232
+    env-step-replay-after:                      87457792
+    env-step-used-after:                        171999232
+    ppo-update-after:                           104980
+    env-step-generate-after:                    84541440
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     17.83236
+    avg_avg_return:                             0.86567235
+    avg_reward_per_step:                        0.10977173
+    proxy_corner:
+      avg_avg_return:                           0.86567235
+      avg_reward_per_step:                      0.10977173
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     17.83236
+    avg_avg_return:                             0.86567235
+    avg_reward_per_step:                        0.10977173
+    proxy_corner:
+      avg_avg_return:                           0.86567235
+      avg_reward_per_step:                      0.10977173
+  perf:
+    env_steps_per_second:                       353871.3542059304
+    ppo_updates_per_second:                     40.34053668395435
+    eval-batch-orig-duration:                   0.5181794874370098
+    eval-batch-shift-duration:                  0.4534779479727149
+    eval-batch-tree-duration:                   0.4723821682855487
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.474854
+      mouse-cheese_dist_finite_wavg:            19.254885
+      mouse-cheese_dist_solvable_avg:           30.474854
+      mouse-cheese_dist_solvable_wavg:          19.254883
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0000001
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.99999
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.39537957
+    visit_patterns:
+      avg_first_visit_time:                     2275.5508
+      avg_last_visit_time:                      2659.1191
+      num_replay_batches:                       2669
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0008170558
+    avg_actor_approxkl3:                        0.00076247327
+    avg_actor_clipfrac:                         0.012097168
+    avg_actor_loss:                             0.00025887505
+    avg_advantage:                              0.015760845
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0005091128
+    avg_entropy:                                0.026803536
+    avg_grad_norm_pre_clip:                     1.0002489
+    avg_loss:                                   0.0004866279
+    max:
+      max_actor_approxkl1:                      0.002708333
+      max_actor_approxkl3:                      0.0017381244
+      max_actor_clipfrac:                       0.020996094
+      max_actor_loss:                           0.0020700633
+      max_advantage:                            0.114628196
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0013411904
+      max_entropy:                              0.032861378
+      max_grad_norm_pre_clip:                   1.5399706
+      max_loss:                                 0.0021372517
+    std:
+      std_actor_approxkl1:                      0.0008022981
+      std_actor_approxkl3:                      0.0005512201
+      std_actor_clipfrac:                       0.005096476
+      std_actor_loss:                           0.0007634639
+      std_advantage:                            0.03132438
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00036541675
+      std_entropy:                              0.0039038917
+      std_grad_norm_pre_clip:                   0.25746945
+      std_loss:                                 0.0007096531
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.47755
+    avg_avg_return:                             0.94632995
+    avg_benchmark_regret:                       0.03849653
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.095588684
+    proxy_corner:
+      avg_avg_return:                           0.94632995
+      avg_reward_per_step:                      0.095588684
+  eval-batch-shift:
+    avg_avg_episode_length:                     65.54843
+    avg_avg_return:                             0.28463775
+    avg_benchmark_regret:                       0.7022527
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.040527344
+    proxy_corner:
+      avg_avg_return:                           0.58604395
+      avg_reward_per_step:                      0.0052948
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.247488
+    avg_avg_return:                             0.9427146
+    avg_benchmark_regret:                       0.04136108
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11214447
+    proxy_corner:
+      avg_avg_return:                           0.9427146
+      avg_reward_per_step:                      0.11214447
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        173015040
+    env-step-replay-before:                     88014848
+    env-step-used-before:                       173015040
+    ppo-update-before:                          105600
+    env-step-generate-before:                   85000192
+    env-step-all-after:                         173047808
+    env-step-replay-after:                      88047616
+    env-step-used-after:                        173047808
+    ppo-update-after:                           105620
+    env-step-generate-after:                    85000192
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     55.40882
+    avg_avg_return:                             0.5659761
+    avg_reward_per_step:                        0.01586914
+    proxy_corner:
+      avg_avg_return:                           0.5659761
+      avg_reward_per_step:                      0.01586914
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     55.40882
+    avg_avg_return:                             0.5659761
+    avg_reward_per_step:                        0.01586914
+    proxy_corner:
+      avg_avg_return:                           0.5659761
+      avg_reward_per_step:                      0.01586914
+  perf:
+    env_steps_per_second:                       349351.1007842246
+    ppo_updates_per_second:                     44.17752779371895
+    eval-batch-orig-duration:                   0.5445992331951857
+    eval-batch-shift-duration:                  0.5061872461810708
+    eval-batch-tree-duration:                   0.46304839570075274
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.49707
+      mouse-cheese_dist_finite_wavg:            18.344313
+      mouse-cheese_dist_solvable_avg:           30.49707
+      mouse-cheese_dist_solvable_wavg:          18.344315
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.40588146
+    visit_patterns:
+      avg_first_visit_time:                     2289.147
+      avg_last_visit_time:                      2677.2043
+      num_replay_batches:                       2687
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0014337747
+    avg_actor_approxkl3:                        0.0010666491
+    avg_actor_clipfrac:                         0.037103273
+    avg_actor_loss:                             -0.0005495228
+    avg_advantage:                              -0.0032767055
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.004802647
+    avg_entropy:                                0.119216375
+    avg_grad_norm_pre_clip:                     1.3492477
+    avg_loss:                                   0.0017325844
+    max:
+      max_actor_approxkl1:                      0.003695061
+      max_actor_approxkl3:                      0.0020537786
+      max_actor_clipfrac:                       0.053466797
+      max_actor_loss:                           0.0006823018
+      max_advantage:                            0.13130158
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0072091273
+      max_entropy:                              0.1375533
+      max_grad_norm_pre_clip:                   2.1381218
+      max_loss:                                 0.0034501057
+    std:
+      std_actor_approxkl1:                      0.00091023423
+      std_actor_approxkl3:                      0.00048107922
+      std_actor_clipfrac:                       0.01227535
+      std_actor_loss:                           0.00088860776
+      std_advantage:                            0.098064184
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00093207374
+      std_entropy:                              0.010528365
+      std_grad_norm_pre_clip:                   0.42322007
+      std_loss:                                 0.0010000718
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.455345
+    avg_avg_return:                             0.948422
+    avg_benchmark_regret:                       0.036404498
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09555817
+    proxy_corner:
+      avg_avg_return:                           0.948422
+      avg_reward_per_step:                      0.09555817
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.293625
+    avg_avg_return:                             0.2693944
+    avg_benchmark_regret:                       0.71749604
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.041267395
+    proxy_corner:
+      avg_avg_return:                           0.613571
+      avg_reward_per_step:                      0.0051116943
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.125467
+    avg_avg_return:                             0.94625944
+    avg_benchmark_regret:                       0.037816145
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11225128
+    proxy_corner:
+      avg_avg_return:                           0.94625944
+      avg_reward_per_step:                      0.11225128
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        174063616
+    env-step-replay-before:                     88440832
+    env-step-used-before:                       174063616
+    ppo-update-before:                          106240
+    env-step-generate-before:                   85622784
+    env-step-all-after:                         174096384
+    env-step-replay-after:                      88440832
+    env-step-used-after:                        174096384
+    ppo-update-after:                           106260
+    env-step-generate-after:                    85655552
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.736511
+    avg_avg_return:                             0.85868996
+    avg_reward_per_step:                        0.09899902
+    proxy_corner:
+      avg_avg_return:                           0.85868996
+      avg_reward_per_step:                      0.09899902
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.736511
+    avg_avg_return:                             0.85868996
+    avg_reward_per_step:                        0.09899902
+    proxy_corner:
+      avg_avg_return:                           0.85868996
+      avg_reward_per_step:                      0.09899902
+  perf:
+    env_steps_per_second:                       339981.0674835763
+    ppo_updates_per_second:                     41.06901201356636
+    eval-batch-orig-duration:                   0.47501559369266033
+    eval-batch-shift-duration:                  0.43759302981197834
+    eval-batch-tree-duration:                   0.4281890541315079
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.450928
+      mouse-cheese_dist_finite_wavg:            21.052586
+      mouse-cheese_dist_solvable_avg:           30.450928
+      mouse-cheese_dist_solvable_wavg:          21.052588
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.40102464
+    visit_patterns:
+      avg_first_visit_time:                     2302.9255
+      avg_last_visit_time:                      2689.3523
+      num_replay_batches:                       2699
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00022203194
+    avg_actor_approxkl3:                        0.00041863872
+    avg_actor_clipfrac:                         0.011553955
+    avg_actor_loss:                             -5.0639803e-05
+    avg_advantage:                              0.001315692
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00045752004
+    avg_entropy:                                0.027164344
+    avg_grad_norm_pre_clip:                     0.8575036
+    avg_loss:                                   0.00015095585
+    max:
+      max_actor_approxkl1:                      0.0010355376
+      max_actor_approxkl3:                      0.0008434445
+      max_actor_clipfrac:                       0.01977539
+      max_actor_loss:                           0.0016393885
+      max_advantage:                            0.09163666
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0014262928
+      max_entropy:                              0.038225837
+      max_grad_norm_pre_clip:                   1.7596741
+      max_loss:                                 0.0018663179
+    std:
+      std_actor_approxkl1:                      0.0004008861
+      std_actor_approxkl3:                      0.00021511703
+      std_actor_clipfrac:                       0.004246622
+      std_actor_loss:                           0.00070900575
+      std_advantage:                            0.029795896
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0004124457
+      std_entropy:                              0.0042862515
+      std_grad_norm_pre_clip:                   0.46203965
+      std_loss:                                 0.0007267525
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.17909
+    avg_avg_return:                             0.9483355
+    avg_benchmark_regret:                       0.036491044
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09561157
+    proxy_corner:
+      avg_avg_return:                           0.9483355
+      avg_reward_per_step:                      0.09561157
+  eval-batch-shift:
+    avg_avg_episode_length:                     64.56406
+    avg_avg_return:                             0.29906452
+    avg_benchmark_regret:                       0.6878259
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.041427612
+    proxy_corner:
+      avg_avg_return:                           0.5559029
+      avg_reward_per_step:                      0.0049209595
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.076813
+    avg_avg_return:                             0.9440565
+    avg_benchmark_regret:                       0.04001911
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11199188
+    proxy_corner:
+      avg_avg_return:                           0.9440565
+      avg_reward_per_step:                      0.11199188
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        175112192
+    env-step-replay-before:                     88932352
+    env-step-used-before:                       175112192
+    ppo-update-before:                          106880
+    env-step-generate-before:                   86179840
+    env-step-all-after:                         175144960
+    env-step-replay-after:                      88965120
+    env-step-used-after:                        175144960
+    ppo-update-after:                           106900
+    env-step-generate-after:                    86179840
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     54.167633
+    avg_avg_return:                             0.5792239
+    avg_reward_per_step:                        0.016448975
+    proxy_corner:
+      avg_avg_return:                           0.5792239
+      avg_reward_per_step:                      0.016448975
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     54.167633
+    avg_avg_return:                             0.5792239
+    avg_reward_per_step:                        0.016448975
+    proxy_corner:
+      avg_avg_return:                           0.5792239
+      avg_reward_per_step:                      0.016448975
+  perf:
+    env_steps_per_second:                       339908.260411143
+    ppo_updates_per_second:                     42.073539711778714
+    eval-batch-orig-duration:                   0.460856637917459
+    eval-batch-shift-duration:                  0.4574068821966648
+    eval-batch-tree-duration:                   0.43014528043568134
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.527344
+      mouse-cheese_dist_finite_wavg:            20.148222
+      mouse-cheese_dist_solvable_avg:           30.527344
+      mouse-cheese_dist_solvable_wavg:          20.148224
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000008
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.40595472
+    visit_patterns:
+      avg_first_visit_time:                     2316.4614
+      avg_last_visit_time:                      2705.1973
+      num_replay_batches:                       2715
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.002208169
+    avg_actor_approxkl3:                        0.002070261
+    avg_actor_clipfrac:                         0.05475464
+    avg_actor_loss:                             -0.00017760023
+    avg_advantage:                              -0.008743087
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0042992407
+    avg_entropy:                                0.119036786
+    avg_grad_norm_pre_clip:                     1.8269894
+    avg_loss:                                   0.0018529835
+    max:
+      max_actor_approxkl1:                      0.0058681127
+      max_actor_approxkl3:                      0.0037946778
+      max_actor_clipfrac:                       0.08691406
+      max_actor_loss:                           0.001686193
+      max_advantage:                            0.13410062
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.006237969
+      max_entropy:                              0.1339911
+      max_grad_norm_pre_clip:                   3.324033
+      max_loss:                                 0.0041715717
+    std:
+      std_actor_approxkl1:                      0.0015547851
+      std_actor_approxkl3:                      0.0008287329
+      std_actor_clipfrac:                       0.018105317
+      std_actor_loss:                           0.0010258663
+      std_advantage:                            0.09208009
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0009560297
+      std_entropy:                              0.009430405
+      std_grad_norm_pre_clip:                   0.66747844
+      std_loss:                                 0.0011133958
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.428923
+    avg_avg_return:                             0.94492567
+    avg_benchmark_regret:                       0.03990084
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09609985
+    proxy_corner:
+      avg_avg_return:                           0.94492567
+      avg_reward_per_step:                      0.09609985
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.51831
+    avg_avg_return:                             0.27007225
+    avg_benchmark_regret:                       0.7168182
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.038650513
+    proxy_corner:
+      avg_avg_return:                           0.63688385
+      avg_reward_per_step:                      0.0053710938
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.215897
+    avg_avg_return:                             0.94542
+    avg_benchmark_regret:                       0.0386556
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11199951
+    proxy_corner:
+      avg_avg_return:                           0.94542
+      avg_reward_per_step:                      0.11199951
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        176160768
+    env-step-replay-before:                     89358336
+    env-step-used-before:                       176160768
+    ppo-update-before:                          107520
+    env-step-generate-before:                   86802432
+    env-step-all-after:                         176193536
+    env-step-replay-after:                      89391104
+    env-step-used-after:                        176193536
+    ppo-update-after:                           107540
+    env-step-generate-after:                    86802432
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     53.92022
+    avg_avg_return:                             0.58152306
+    avg_reward_per_step:                        0.016662598
+    proxy_corner:
+      avg_avg_return:                           0.58152306
+      avg_reward_per_step:                      0.016662598
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     53.92022
+    avg_avg_return:                             0.58152306
+    avg_reward_per_step:                        0.016662598
+    proxy_corner:
+      avg_avg_return:                           0.58152306
+      avg_reward_per_step:                      0.016662598
+  perf:
+    env_steps_per_second:                       290738.551229456
+    ppo_updates_per_second:                     42.09549252598199
+    eval-batch-orig-duration:                   0.5354593405500054
+    eval-batch-shift-duration:                  0.49099415354430676
+    eval-batch-tree-duration:                   0.4797924580052495
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.516357
+      mouse-cheese_dist_finite_wavg:            14.756151
+      mouse-cheese_dist_solvable_avg:           30.516357
+      mouse-cheese_dist_solvable_wavg:          14.756155
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.39743173
+    visit_patterns:
+      avg_first_visit_time:                     2329.8555
+      avg_last_visit_time:                      2718.21
+      num_replay_batches:                       2728
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0015980698
+    avg_actor_approxkl3:                        0.0012720996
+    avg_actor_clipfrac:                         0.041082766
+    avg_actor_loss:                             -0.00017810744
+    avg_advantage:                              -0.017080167
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0041529117
+    avg_entropy:                                0.10367109
+    avg_grad_norm_pre_clip:                     1.7830597
+    avg_loss:                                   0.0017946772
+    max:
+      max_actor_approxkl1:                      0.0033074445
+      max_actor_approxkl3:                      0.0020248627
+      max_actor_clipfrac:                       0.059936523
+      max_actor_loss:                           0.0025856271
+      max_advantage:                            0.103369534
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0062808977
+      max_entropy:                              0.12441749
+      max_grad_norm_pre_clip:                   3.4850194
+      max_loss:                                 0.00428646
+    std:
+      std_actor_approxkl1:                      0.00078137824
+      std_actor_approxkl3:                      0.00050140405
+      std_actor_clipfrac:                       0.013256716
+      std_actor_loss:                           0.0010642584
+      std_advantage:                            0.09107661
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010579539
+      std_entropy:                              0.009099671
+      std_grad_norm_pre_clip:                   0.72822756
+      std_loss:                                 0.001100256
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.594467
+    avg_avg_return:                             0.94431967
+    avg_benchmark_regret:                       0.040506877
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.095718384
+    proxy_corner:
+      avg_avg_return:                           0.94431967
+      avg_reward_per_step:                      0.095718384
+  eval-batch-shift:
+    avg_avg_episode_length:                     64.29695
+    avg_avg_return:                             0.29573846
+    avg_benchmark_regret:                       0.691152
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.044548035
+    proxy_corner:
+      avg_avg_return:                           0.58403254
+      avg_reward_per_step:                      0.005279541
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.618109
+    avg_avg_return:                             0.93715924
+    avg_benchmark_regret:                       0.046916373
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11187744
+    proxy_corner:
+      avg_avg_return:                           0.93715924
+      avg_reward_per_step:                      0.11187744
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        177209344
+    env-step-replay-before:                     89784320
+    env-step-used-before:                       177209344
+    ppo-update-before:                          108160
+    env-step-generate-before:                   87425024
+    env-step-all-after:                         177242112
+    env-step-replay-after:                      89784320
+    env-step-used-after:                        177242112
+    ppo-update-after:                           108180
+    env-step-generate-after:                    87457792
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     20.135735
+    avg_avg_return:                             0.8508652
+    avg_reward_per_step:                        0.08792114
+    proxy_corner:
+      avg_avg_return:                           0.8508652
+      avg_reward_per_step:                      0.08792114
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     20.135735
+    avg_avg_return:                             0.8508652
+    avg_reward_per_step:                        0.08792114
+    proxy_corner:
+      avg_avg_return:                           0.8508652
+      avg_reward_per_step:                      0.08792114
+  perf:
+    env_steps_per_second:                       329788.5076297279
+    ppo_updates_per_second:                     41.654877334956716
+    eval-batch-orig-duration:                   0.5048258388414979
+    eval-batch-shift-duration:                  0.4175720792263746
+    eval-batch-tree-duration:                   0.42448180355131626
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.513428
+      mouse-cheese_dist_finite_wavg:            19.26067
+      mouse-cheese_dist_solvable_avg:           30.513428
+      mouse-cheese_dist_solvable_wavg:          19.260672
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016813
+    scoring:
+      avg_scores:                               0.40031293
+    visit_patterns:
+      avg_first_visit_time:                     2348.6367
+      avg_last_visit_time:                      2730.2217
+      num_replay_batches:                       2740
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009956927
+    avg_actor_approxkl3:                        0.0012088042
+    avg_actor_clipfrac:                         0.014184571
+    avg_actor_loss:                             -0.00010003792
+    avg_advantage:                              0.003255522
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0006205134
+    avg_entropy:                                0.02569295
+    avg_grad_norm_pre_clip:                     0.92038155
+    avg_loss:                                   0.00018452585
+    max:
+      max_actor_approxkl1:                      0.002678744
+      max_actor_approxkl3:                      0.003330788
+      max_actor_clipfrac:                       0.025878906
+      max_actor_loss:                           0.0014905576
+      max_advantage:                            0.11247599
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0013755359
+      max_entropy:                              0.03787122
+      max_grad_norm_pre_clip:                   1.502249
+      max_loss:                                 0.0015530406
+    std:
+      std_actor_approxkl1:                      0.000886552
+      std_actor_approxkl3:                      0.0009832358
+      std_actor_clipfrac:                       0.0061479826
+      std_actor_loss:                           0.0006707291
+      std_advantage:                            0.035400428
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0003633148
+      std_entropy:                              0.00513444
+      std_grad_norm_pre_clip:                   0.27888483
+      std_loss:                                 0.00064401503
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.523138
+    avg_avg_return:                             0.9456661
+    avg_benchmark_regret:                       0.039160408
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09550476
+    proxy_corner:
+      avg_avg_return:                           0.9456661
+      avg_reward_per_step:                      0.09550476
+  eval-batch-shift:
+    avg_avg_episode_length:                     64.48856
+    avg_avg_return:                             0.29721695
+    avg_benchmark_regret:                       0.68967354
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.04310608
+    proxy_corner:
+      avg_avg_return:                           0.6172701
+      avg_reward_per_step:                      0.0054855347
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.400764
+    avg_avg_return:                             0.9394659
+    avg_benchmark_regret:                       0.04460972
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11213684
+    proxy_corner:
+      avg_avg_return:                           0.9394659
+      avg_reward_per_step:                      0.11213684
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        178257920
+    env-step-replay-before:                     90439680
+    env-step-used-before:                       178257920
+    ppo-update-before:                          108800
+    env-step-generate-before:                   87818240
+    env-step-all-after:                         178290688
+    env-step-replay-after:                      90472448
+    env-step-used-after:                        178290688
+    ppo-update-after:                           108820
+    env-step-generate-after:                    87818240
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     55.667538
+    avg_avg_return:                             0.5866883
+    avg_reward_per_step:                        0.016082764
+    proxy_corner:
+      avg_avg_return:                           0.5866883
+      avg_reward_per_step:                      0.016082764
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     55.667538
+    avg_avg_return:                             0.5866883
+    avg_reward_per_step:                        0.016082764
+    proxy_corner:
+      avg_avg_return:                           0.5866883
+      avg_reward_per_step:                      0.016082764
+  perf:
+    env_steps_per_second:                       291007.3291884857
+    ppo_updates_per_second:                     42.11360809132452
+    eval-batch-orig-duration:                   0.55945582408458
+    eval-batch-shift-duration:                  0.49195549730211496
+    eval-batch-tree-duration:                   0.4889896083623171
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.55664
+      mouse-cheese_dist_finite_wavg:            19.25012
+      mouse-cheese_dist_solvable_avg:           30.55664
+      mouse-cheese_dist_solvable_wavg:          19.250122
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.39223412
+    visit_patterns:
+      avg_first_visit_time:                     2358.7773
+      avg_last_visit_time:                      2751.2942
+      num_replay_batches:                       2761
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0010271486
+    avg_actor_approxkl3:                        0.0009724513
+    avg_actor_clipfrac:                         0.03134766
+    avg_actor_loss:                             -0.0006802004
+    avg_advantage:                              -0.008904426
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0032234497
+    avg_entropy:                                0.110556796
+    avg_grad_norm_pre_clip:                     1.0671957
+    avg_loss:                                   0.00082096783
+    max:
+      max_actor_approxkl1:                      0.0035353517
+      max_actor_approxkl3:                      0.0016614433
+      max_actor_clipfrac:                       0.049926758
+      max_actor_loss:                           0.00061546266
+      max_advantage:                            0.1176402
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0046064607
+      max_entropy:                              0.12895706
+      max_grad_norm_pre_clip:                   1.7443448
+      max_loss:                                 0.0023242282
+    std:
+      std_actor_approxkl1:                      0.0009190346
+      std_actor_approxkl3:                      0.00045436004
+      std_actor_clipfrac:                       0.012363233
+      std_actor_loss:                           0.0007677772
+      std_advantage:                            0.0802326
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00080205116
+      std_entropy:                              0.009495128
+      std_grad_norm_pre_clip:                   0.3954068
+      std_loss:                                 0.00078520295
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.428831
+    avg_avg_return:                             0.94826984
+    avg_benchmark_regret:                       0.036556676
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.095062256
+    proxy_corner:
+      avg_avg_return:                           0.94826984
+      avg_reward_per_step:                      0.095062256
+  eval-batch-shift:
+    avg_avg_episode_length:                     64.85343
+    avg_avg_return:                             0.2913985
+    avg_benchmark_regret:                       0.69549197
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.04197693
+    proxy_corner:
+      avg_avg_return:                           0.56363696
+      avg_reward_per_step:                      0.0047912598
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.569164
+    avg_avg_return:                             0.9447675
+    avg_benchmark_regret:                       0.03930812
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.110191345
+    proxy_corner:
+      avg_avg_return:                           0.9447675
+      avg_reward_per_step:                      0.110191345
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        179306496
+    env-step-replay-before:                     90931200
+    env-step-used-before:                       179306496
+    ppo-update-before:                          109440
+    env-step-generate-before:                   88375296
+    env-step-all-after:                         179339264
+    env-step-replay-after:                      90931200
+    env-step-used-after:                        179339264
+    ppo-update-after:                           109460
+    env-step-generate-after:                    88408064
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.73907
+    avg_avg_return:                             0.8586517
+    avg_reward_per_step:                        0.091796875
+    proxy_corner:
+      avg_avg_return:                           0.8586517
+      avg_reward_per_step:                      0.091796875
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     18.73907
+    avg_avg_return:                             0.8586517
+    avg_reward_per_step:                        0.091796875
+    proxy_corner:
+      avg_avg_return:                           0.8586517
+      avg_reward_per_step:                      0.091796875
+  perf:
+    env_steps_per_second:                       318494.12177384977
+    ppo_updates_per_second:                     42.24170377881856
+    eval-batch-orig-duration:                   0.5349185336381197
+    eval-batch-shift-duration:                  0.4839913733303547
+    eval-batch-tree-duration:                   0.48572636116296053
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.519775
+      mouse-cheese_dist_finite_wavg:            13.86044
+      mouse-cheese_dist_solvable_avg:           30.519775
+      mouse-cheese_dist_solvable_wavg:          13.860442
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.38577542
+    visit_patterns:
+      avg_first_visit_time:                     2372.8372
+      avg_last_visit_time:                      2765.3193
+      num_replay_batches:                       2775
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.000770193
+    avg_actor_approxkl3:                        0.00071362127
+    avg_actor_clipfrac:                         0.013500976
+    avg_actor_loss:                             -0.00025392798
+    avg_advantage:                              0.01223692
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00048261235
+    avg_entropy:                                0.02715232
+    avg_grad_norm_pre_clip:                     1.0617083
+    avg_loss:                                   -3.977414e-05
+    max:
+      max_actor_approxkl1:                      0.0026558663
+      max_actor_approxkl3:                      0.0014319476
+      max_actor_clipfrac:                       0.022705078
+      max_actor_loss:                           0.0019280836
+      max_advantage:                            0.105210245
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0008504043
+      max_entropy:                              0.03567274
+      max_grad_norm_pre_clip:                   3.2446613
+      max_loss:                                 0.0022485526
+    std:
+      std_actor_approxkl1:                      0.00067284907
+      std_actor_approxkl3:                      0.00038412062
+      std_actor_clipfrac:                       0.0056725796
+      std_actor_loss:                           0.0009397076
+      std_advantage:                            0.03236834
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00029015262
+      std_entropy:                              0.005158225
+      std_grad_norm_pre_clip:                   0.78493744
+      std_loss:                                 0.00093321607
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.212322
+    avg_avg_return:                             0.9471849
+    avg_benchmark_regret:                       0.03764161
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09598541
+    proxy_corner:
+      avg_avg_return:                           0.9471849
+      avg_reward_per_step:                      0.09598541
+  eval-batch-shift:
+    avg_avg_episode_length:                     64.95186
+    avg_avg_return:                             0.28733763
+    avg_benchmark_regret:                       0.6995528
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.041275024
+    proxy_corner:
+      avg_avg_return:                           0.60308206
+      avg_reward_per_step:                      0.005218506
+  eval-batch-tree:
+    avg_avg_episode_length:                     16.963531
+    avg_avg_return:                             0.94421375
+    avg_benchmark_regret:                       0.03986186
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.112350464
+    proxy_corner:
+      avg_avg_return:                           0.94421375
+      avg_reward_per_step:                      0.112350464
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        180355072
+    env-step-replay-before:                     91291648
+    env-step-used-before:                       180355072
+    ppo-update-before:                          110080
+    env-step-generate-before:                   89063424
+    env-step-all-after:                         180387840
+    env-step-replay-after:                      91324416
+    env-step-used-after:                        180387840
+    ppo-update-after:                           110100
+    env-step-generate-after:                    89063424
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     54.2371
+    avg_avg_return:                             0.5858749
+    avg_reward_per_step:                        0.01663208
+    proxy_corner:
+      avg_avg_return:                           0.5858749
+      avg_reward_per_step:                      0.01663208
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     54.2371
+    avg_avg_return:                             0.5858749
+    avg_reward_per_step:                        0.01663208
+    proxy_corner:
+      avg_avg_return:                           0.5858749
+      avg_reward_per_step:                      0.01663208
+  perf:
+    env_steps_per_second:                       348719.21762158367
+    ppo_updates_per_second:                     41.61347606188023
+    eval-batch-orig-duration:                   0.46965689957141876
+    eval-batch-shift-duration:                  0.4600937245413661
+    eval-batch-tree-duration:                   0.43193299416452646
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.473633
+      mouse-cheese_dist_finite_wavg:            20.162281
+      mouse-cheese_dist_solvable_avg:           30.473633
+      mouse-cheese_dist_solvable_wavg:          20.162281
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.392237
+    visit_patterns:
+      avg_first_visit_time:                     2389.6487
+      avg_last_visit_time:                      2777.2307
+      num_replay_batches:                       2787
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00093605986
+    avg_actor_approxkl3:                        0.0010662795
+    avg_actor_clipfrac:                         0.034344483
+    avg_actor_loss:                             -5.269684e-05
+    avg_advantage:                              -0.016521301
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0039285826
+    avg_entropy:                                0.099040024
+    avg_grad_norm_pre_clip:                     1.3904852
+    avg_loss:                                   0.0018125543
+    max:
+      max_actor_approxkl1:                      0.0019742297
+      max_actor_approxkl3:                      0.0016743696
+      max_actor_clipfrac:                       0.05041504
+      max_actor_loss:                           0.0022854768
+      max_advantage:                            0.121182084
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0058316146
+      max_entropy:                              0.10907829
+      max_grad_norm_pre_clip:                   2.5391085
+      max_loss:                                 0.003779076
+    std:
+      std_actor_approxkl1:                      0.0006040184
+      std_actor_approxkl3:                      0.00043599415
+      std_actor_clipfrac:                       0.010461737
+      std_actor_loss:                           0.0009714672
+      std_advantage:                            0.08818811
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0008896084
+      std_entropy:                              0.0064653466
+      std_grad_norm_pre_clip:                   0.48981273
+      std_loss:                                 0.0010717203
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.306274
+    avg_avg_return:                             0.9477629
+    avg_benchmark_regret:                       0.037063602
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.095962524
+    proxy_corner:
+      avg_avg_return:                           0.9477629
+      avg_reward_per_step:                      0.095962524
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.30152
+    avg_avg_return:                             0.26742214
+    avg_benchmark_regret:                       0.71946836
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.039497375
+    proxy_corner:
+      avg_avg_return:                           0.5913946
+      avg_reward_per_step:                      0.005088806
+  eval-batch-tree:
+    avg_avg_episode_length:                     16.910297
+    avg_avg_return:                             0.94536567
+    avg_benchmark_regret:                       0.03870993
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.1124115
+    proxy_corner:
+      avg_avg_return:                           0.94536567
+      avg_reward_per_step:                      0.1124115
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        181403648
+    env-step-replay-before:                     91947008
+    env-step-used-before:                       181403648
+    ppo-update-before:                          110720
+    env-step-generate-before:                   89456640
+    env-step-all-after:                         181436416
+    env-step-replay-after:                      91979776
+    env-step-used-after:                        181436416
+    ppo-update-after:                           110740
+    env-step-generate-after:                    89456640
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     55.096786
+    avg_avg_return:                             0.55799735
+    avg_reward_per_step:                        0.016326904
+    proxy_corner:
+      avg_avg_return:                           0.55799735
+      avg_reward_per_step:                      0.016326904
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     55.096786
+    avg_avg_return:                             0.55799735
+    avg_reward_per_step:                        0.016326904
+    proxy_corner:
+      avg_avg_return:                           0.55799735
+      avg_reward_per_step:                      0.016326904
+  perf:
+    env_steps_per_second:                       300794.12681426294
+    ppo_updates_per_second:                     41.03401428303355
+    eval-batch-orig-duration:                   0.5201874077320099
+    eval-batch-shift-duration:                  0.47489820793271065
+    eval-batch-tree-duration:                   0.4521664772182703
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.548584
+      mouse-cheese_dist_finite_wavg:            12.969852
+      mouse-cheese_dist_solvable_avg:           30.548584
+      mouse-cheese_dist_solvable_wavg:          12.969854
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016798
+    scoring:
+      avg_scores:                               0.38195738
+    visit_patterns:
+      avg_first_visit_time:                     2399.9358
+      avg_last_visit_time:                      2797.3423
+      num_replay_batches:                       2807
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012330114
+    avg_actor_approxkl3:                        0.0012130046
+    avg_actor_clipfrac:                         0.04003296
+    avg_actor_loss:                             -0.00083353743
+    avg_advantage:                              -0.024772495
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.005812269
+    avg_entropy:                                0.10973493
+    avg_grad_norm_pre_clip:                     1.4106313
+    avg_loss:                                   0.0019628627
+    max:
+      max_actor_approxkl1:                      0.0032961548
+      max_actor_approxkl3:                      0.0021403714
+      max_actor_clipfrac:                       0.055419922
+      max_actor_loss:                           0.0012841895
+      max_advantage:                            0.094798386
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.007815425
+      max_entropy:                              0.122865245
+      max_grad_norm_pre_clip:                   2.6769853
+      max_loss:                                 0.0046566385
+    std:
+      std_actor_approxkl1:                      0.0010731866
+      std_actor_approxkl3:                      0.0005287219
+      std_actor_clipfrac:                       0.013646864
+      std_actor_loss:                           0.0013456622
+      std_advantage:                            0.10752522
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010825836
+      std_entropy:                              0.007782019
+      std_grad_norm_pre_clip:                   0.58741283
+      std_loss:                                 0.0014321116
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.343256
+    avg_avg_return:                             0.9490628
+    avg_benchmark_regret:                       0.03576367
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09577179
+    proxy_corner:
+      avg_avg_return:                           0.9490628
+      avg_reward_per_step:                      0.09577179
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.84079
+    avg_avg_return:                             0.25997797
+    avg_benchmark_regret:                       0.7269125
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.03905487
+    proxy_corner:
+      avg_avg_return:                           0.5491787
+      avg_reward_per_step:                      0.004600525
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.229628
+    avg_avg_return:                             0.94042385
+    avg_benchmark_regret:                       0.043651793
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.1121521
+    proxy_corner:
+      avg_avg_return:                           0.94042385
+      avg_reward_per_step:                      0.1121521
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        182452224
+    env-step-replay-before:                     92471296
+    env-step-used-before:                       182452224
+    ppo-update-before:                          111360
+    env-step-generate-before:                   89980928
+    env-step-all-after:                         182484992
+    env-step-replay-after:                      92504064
+    env-step-used-after:                        182484992
+    ppo-update-after:                           111380
+    env-step-generate-after:                    89980928
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     50.333412
+    avg_avg_return:                             0.6164291
+    avg_reward_per_step:                        0.018554688
+    proxy_corner:
+      avg_avg_return:                           0.6164291
+      avg_reward_per_step:                      0.018554688
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     50.333412
+    avg_avg_return:                             0.6164291
+    avg_reward_per_step:                        0.018554688
+    proxy_corner:
+      avg_avg_return:                           0.6164291
+      avg_reward_per_step:                      0.018554688
+  perf:
+    env_steps_per_second:                       290658.19687663217
+    ppo_updates_per_second:                     42.13896520100474
+    eval-batch-orig-duration:                   0.6807702034711838
+    eval-batch-shift-duration:                  0.609906205907464
+    eval-batch-tree-duration:                   0.6223650788888335
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.581543
+      mouse-cheese_dist_finite_wavg:            21.072208
+      mouse-cheese_dist_solvable_avg:           30.581543
+      mouse-cheese_dist_solvable_wavg:          21.072214
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.38383943
+    visit_patterns:
+      avg_first_visit_time:                     2414.6243
+      avg_last_visit_time:                      2813.1816
+      num_replay_batches:                       2823
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0018796086
+    avg_actor_approxkl3:                        0.0019705272
+    avg_actor_clipfrac:                         0.043701172
+    avg_actor_loss:                             0.0007045053
+    avg_advantage:                              -0.014100839
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0033418355
+    avg_entropy:                                0.08957243
+    avg_grad_norm_pre_clip:                     1.7719319
+    avg_loss:                                   0.002285851
+    max:
+      max_actor_approxkl1:                      0.003309711
+      max_actor_approxkl3:                      0.0035153874
+      max_actor_clipfrac:                       0.061035156
+      max_actor_loss:                           0.0026862249
+      max_advantage:                            0.10503036
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.004836002
+      max_entropy:                              0.1033328
+      max_grad_norm_pre_clip:                   3.5376656
+      max_loss:                                 0.0045182845
+    std:
+      std_actor_approxkl1:                      0.00097158673
+      std_actor_approxkl3:                      0.0008709836
+      std_actor_clipfrac:                       0.013213569
+      std_actor_loss:                           0.0009280708
+      std_advantage:                            0.08129452
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011705778
+      std_entropy:                              0.007267786
+      std_grad_norm_pre_clip:                   0.74873817
+      std_loss:                                 0.0012384172
+  eval-batch-orig:
+    avg_avg_episode_length:                     15.995438
+    avg_avg_return:                             0.9502795
+    avg_benchmark_regret:                       0.03454703
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09611511
+    proxy_corner:
+      avg_avg_return:                           0.9502795
+      avg_reward_per_step:                      0.09611511
+  eval-batch-shift:
+    avg_avg_episode_length:                     66.169395
+    avg_avg_return:                             0.26903218
+    avg_benchmark_regret:                       0.7178583
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.04045868
+    proxy_corner:
+      avg_avg_return:                           0.5980245
+      avg_reward_per_step:                      0.005203247
+  eval-batch-tree:
+    avg_avg_episode_length:                     16.740435
+    avg_avg_return:                             0.9493249
+    avg_benchmark_regret:                       0.034750674
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11263275
+    proxy_corner:
+      avg_avg_return:                           0.9493249
+      avg_reward_per_step:                      0.11263275
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        183500800
+    env-step-replay-before:                     93028352
+    env-step-used-before:                       183500800
+    ppo-update-before:                          112000
+    env-step-generate-before:                   90472448
+    env-step-all-after:                         183533568
+    env-step-replay-after:                      93061120
+    env-step-used-after:                        183533568
+    ppo-update-after:                           112020
+    env-step-generate-after:                    90472448
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     53.760468
+    avg_avg_return:                             0.60283095
+    avg_reward_per_step:                        0.01675415
+    proxy_corner:
+      avg_avg_return:                           0.60283095
+      avg_reward_per_step:                      0.01675415
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     53.760468
+    avg_avg_return:                             0.60283095
+    avg_reward_per_step:                        0.01675415
+    proxy_corner:
+      avg_avg_return:                           0.60283095
+      avg_reward_per_step:                      0.01675415
+  perf:
+    env_steps_per_second:                       294217.69083259016
+    ppo_updates_per_second:                     42.19514270929354
+    eval-batch-orig-duration:                   0.5277775628492236
+    eval-batch-shift-duration:                  0.4707852015271783
+    eval-batch-tree-duration:                   0.446445575915277
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.610352
+      mouse-cheese_dist_finite_wavg:            12.964068
+      mouse-cheese_dist_solvable_avg:           30.610352
+      mouse-cheese_dist_solvable_wavg:          12.96407
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.000004
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.37441456
+    visit_patterns:
+      avg_first_visit_time:                     2425.7876
+      avg_last_visit_time:                      2830.3354
+      num_replay_batches:                       2840
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0012025206
+    avg_actor_approxkl3:                        0.0011319689
+    avg_actor_clipfrac:                         0.035107423
+    avg_actor_loss:                             -0.00027307495
+    avg_advantage:                              -0.011893249
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0029621369
+    avg_entropy:                                0.09758306
+    avg_grad_norm_pre_clip:                     1.0674131
+    avg_loss:                                   0.0011104103
+    max:
+      max_actor_approxkl1:                      0.0028957655
+      max_actor_approxkl3:                      0.0024574269
+      max_actor_clipfrac:                       0.056884766
+      max_actor_loss:                           0.001286611
+      max_advantage:                            0.09639692
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.004707347
+      max_entropy:                              0.108754486
+      max_grad_norm_pre_clip:                   1.6474115
+      max_loss:                                 0.0028231766
+    std:
+      std_actor_approxkl1:                      0.0007380202
+      std_actor_approxkl3:                      0.0004585417
+      std_actor_clipfrac:                       0.0101895
+      std_actor_loss:                           0.0007891635
+      std_advantage:                            0.07662257
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00086551
+      std_entropy:                              0.006383349
+      std_grad_norm_pre_clip:                   0.35500574
+      std_loss:                                 0.0008931782
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.14458
+    avg_avg_return:                             0.9485222
+    avg_benchmark_regret:                       0.03630431
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09613037
+    proxy_corner:
+      avg_avg_return:                           0.9485222
+      avg_reward_per_step:                      0.09613037
+  eval-batch-shift:
+    avg_avg_episode_length:                     65.134186
+    avg_avg_return:                             0.28188992
+    avg_benchmark_regret:                       0.70500064
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.040870667
+    proxy_corner:
+      avg_avg_return:                           0.56790996
+      avg_reward_per_step:                      0.005104065
+  eval-batch-tree:
+    avg_avg_episode_length:                     16.685162
+    avg_avg_return:                             0.94788367
+    avg_benchmark_regret:                       0.036191925
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11267853
+    proxy_corner:
+      avg_avg_return:                           0.94788367
+      avg_reward_per_step:                      0.11267853
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        184549376
+    env-step-replay-before:                     93618176
+    env-step-used-before:                       184549376
+    ppo-update-before:                          112640
+    env-step-generate-before:                   90931200
+    env-step-all-after:                         184582144
+    env-step-replay-after:                      93650944
+    env-step-used-after:                        184582144
+    ppo-update-after:                           112660
+    env-step-generate-after:                    90931200
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     52.241585
+    avg_avg_return:                             0.6001681
+    avg_reward_per_step:                        0.017181396
+    proxy_corner:
+      avg_avg_return:                           0.6001681
+      avg_reward_per_step:                      0.017181396
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     52.241585
+    avg_avg_return:                             0.6001681
+    avg_reward_per_step:                        0.017181396
+    proxy_corner:
+      avg_avg_return:                           0.6001681
+      avg_reward_per_step:                      0.017181396
+  perf:
+    env_steps_per_second:                       294991.7957853359
+    ppo_updates_per_second:                     42.13632483595566
+    eval-batch-orig-duration:                   0.5288869980722666
+    eval-batch-shift-duration:                  0.47405337914824486
+    eval-batch-tree-duration:                   0.45455129351466894
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.748047
+      mouse-cheese_dist_finite_wavg:            21.057777
+      mouse-cheese_dist_solvable_avg:           30.748047
+      mouse-cheese_dist_solvable_wavg:          21.057777
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.37722263
+    visit_patterns:
+      avg_first_visit_time:                     2437.0857
+      avg_last_visit_time:                      2848.2917
+      num_replay_batches:                       2858
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0008223949
+    avg_actor_approxkl3:                        0.0010929462
+    avg_actor_clipfrac:                         0.032043457
+    avg_actor_loss:                             -0.00045939928
+    avg_advantage:                              -0.01170326
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0035621275
+    avg_entropy:                                0.098175816
+    avg_grad_norm_pre_clip:                     1.1621481
+    avg_loss:                                   0.0012234885
+    max:
+      max_actor_approxkl1:                      0.0027247206
+      max_actor_approxkl3:                      0.002071428
+      max_actor_clipfrac:                       0.047973633
+      max_actor_loss:                           0.0009737015
+      max_advantage:                            0.102167964
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.005149715
+      max_entropy:                              0.10948822
+      max_grad_norm_pre_clip:                   2.572701
+      max_loss:                                 0.0026232894
+    std:
+      std_actor_approxkl1:                      0.0007805354
+      std_actor_approxkl3:                      0.0005685542
+      std_actor_clipfrac:                       0.012926578
+      std_actor_loss:                           0.0006664689
+      std_advantage:                            0.08433946
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010987943
+      std_entropy:                              0.00598811
+      std_grad_norm_pre_clip:                   0.4894114
+      std_loss:                                 0.00080135366
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.494194
+    avg_avg_return:                             0.944419
+    avg_benchmark_regret:                       0.04040749
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.096069336
+    proxy_corner:
+      avg_avg_return:                           0.944419
+      avg_reward_per_step:                      0.096069336
+  eval-batch-shift:
+    avg_avg_episode_length:                     65.9131
+    avg_avg_return:                             0.28072155
+    avg_benchmark_regret:                       0.70616895
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.039733887
+    proxy_corner:
+      avg_avg_return:                           0.5913491
+      avg_reward_per_step:                      0.005355835
+  eval-batch-tree:
+    avg_avg_episode_length:                     16.990173
+    avg_avg_return:                             0.94618404
+    avg_benchmark_regret:                       0.037891522
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11263275
+    proxy_corner:
+      avg_avg_return:                           0.94618404
+      avg_reward_per_step:                      0.11263275
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        185597952
+    env-step-replay-before:                     94142464
+    env-step-used-before:                       185597952
+    ppo-update-before:                          113280
+    env-step-generate-before:                   91455488
+    env-step-all-after:                         185630720
+    env-step-replay-after:                      94175232
+    env-step-used-after:                        185630720
+    ppo-update-after:                           113300
+    env-step-generate-after:                    91455488
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     55.03293
+    avg_avg_return:                             0.56743056
+    avg_reward_per_step:                        0.016082764
+    proxy_corner:
+      avg_avg_return:                           0.56743056
+      avg_reward_per_step:                      0.016082764
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     55.03293
+    avg_avg_return:                             0.56743056
+    avg_reward_per_step:                        0.016082764
+    proxy_corner:
+      avg_avg_return:                           0.56743056
+      avg_reward_per_step:                      0.016082764
+  perf:
+    env_steps_per_second:                       320273.83568330924
+    ppo_updates_per_second:                     41.877443156681
+    eval-batch-orig-duration:                   0.5382732367143035
+    eval-batch-shift-duration:                  0.4775325199589133
+    eval-batch-tree-duration:                   0.4785813745111227
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.798584
+      mouse-cheese_dist_finite_wavg:            20.184465
+      mouse-cheese_dist_solvable_avg:           30.798584
+      mouse-cheese_dist_solvable_wavg:          20.184467
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.39504236
+    visit_patterns:
+      avg_first_visit_time:                     2451.9663
+      avg_last_visit_time:                      2864.2878
+      num_replay_batches:                       2874
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0016202334
+    avg_actor_approxkl3:                        0.0012682684
+    avg_actor_clipfrac:                         0.037542727
+    avg_actor_loss:                             -0.0003869843
+    avg_advantage:                              -0.013198272
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0051140417
+    avg_entropy:                                0.11196195
+    avg_grad_norm_pre_clip:                     1.4829166
+    avg_loss:                                   0.0020580746
+    max:
+      max_actor_approxkl1:                      0.0037692292
+      max_actor_approxkl3:                      0.0019781287
+      max_actor_clipfrac:                       0.05053711
+      max_actor_loss:                           0.0008393377
+      max_advantage:                            0.11899191
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.007414224
+      max_entropy:                              0.1277859
+      max_grad_norm_pre_clip:                   2.6652968
+      max_loss:                                 0.0034888647
+    std:
+      std_actor_approxkl1:                      0.0010077134
+      std_actor_approxkl3:                      0.0005614677
+      std_actor_clipfrac:                       0.012009726
+      std_actor_loss:                           0.00079701334
+      std_advantage:                            0.1013228
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010953924
+      std_entropy:                              0.007536052
+      std_grad_norm_pre_clip:                   0.46330783
+      std_loss:                                 0.00097198767
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.799694
+    avg_avg_return:                             0.9399959
+    avg_benchmark_regret:                       0.044830635
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09584808
+    proxy_corner:
+      avg_avg_return:                           0.9399959
+      avg_reward_per_step:                      0.09584808
+  eval-batch-shift:
+    avg_avg_episode_length:                     65.55925
+    avg_avg_return:                             0.2761594
+    avg_benchmark_regret:                       0.710731
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.04232025
+    proxy_corner:
+      avg_avg_return:                           0.58882475
+      avg_reward_per_step:                      0.0051193237
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.565823
+    avg_avg_return:                             0.9442743
+    avg_benchmark_regret:                       0.03980134
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11175537
+    proxy_corner:
+      avg_avg_return:                           0.9442743
+      avg_reward_per_step:                      0.11175537
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        186646528
+    env-step-replay-before:                     94765056
+    env-step-used-before:                       186646528
+    ppo-update-before:                          113920
+    env-step-generate-before:                   91881472
+    env-step-all-after:                         186679296
+    env-step-replay-after:                      94797824
+    env-step-used-after:                        186679296
+    ppo-update-after:                           113940
+    env-step-generate-after:                    91881472
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     51.606697
+    avg_avg_return:                             0.60746783
+    avg_reward_per_step:                        0.017944336
+    proxy_corner:
+      avg_avg_return:                           0.60746783
+      avg_reward_per_step:                      0.017944336
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     51.606697
+    avg_avg_return:                             0.60746783
+    avg_reward_per_step:                        0.017944336
+    proxy_corner:
+      avg_avg_return:                           0.60746783
+      avg_reward_per_step:                      0.017944336
+  perf:
+    env_steps_per_second:                       321512.4956855881
+    ppo_updates_per_second:                     42.249229656896595
+    eval-batch-orig-duration:                   0.5031898654997349
+    eval-batch-shift-duration:                  0.4642208619043231
+    eval-batch-tree-duration:                   0.4907359257340431
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.832031
+      mouse-cheese_dist_finite_wavg:            12.098805
+      mouse-cheese_dist_solvable_avg:           30.832031
+      mouse-cheese_dist_solvable_wavg:          12.098805
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016816
+    scoring:
+      avg_scores:                               0.379977
+    visit_patterns:
+      avg_first_visit_time:                     2463.28
+      avg_last_visit_time:                      2883.214
+      num_replay_batches:                       2893
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011402306
+    avg_actor_approxkl3:                        0.00089017855
+    avg_actor_clipfrac:                         0.02956543
+    avg_actor_loss:                             -0.00096234813
+    avg_advantage:                              -0.010946755
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00345217
+    avg_entropy:                                0.09301867
+    avg_grad_norm_pre_clip:                     0.8491768
+    avg_loss:                                   0.00067071826
+    max:
+      max_actor_approxkl1:                      0.002247156
+      max_actor_approxkl3:                      0.0012604336
+      max_actor_clipfrac:                       0.041015625
+      max_actor_loss:                           0.00050289184
+      max_advantage:                            0.09399766
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.005206723
+      max_entropy:                              0.10142289
+      max_grad_norm_pre_clip:                   1.397612
+      max_loss:                                 0.0018380309
+    std:
+      std_actor_approxkl1:                      0.00064026733
+      std_actor_approxkl3:                      0.00034051918
+      std_actor_clipfrac:                       0.009873791
+      std_actor_loss:                           0.0008239781
+      std_advantage:                            0.08309114
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.000976649
+      std_entropy:                              0.004949187
+      std_grad_norm_pre_clip:                   0.26067296
+      std_loss:                                 0.0008313947
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.040562
+    avg_avg_return:                             0.9483304
+    avg_benchmark_regret:                       0.03649609
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.095832825
+    proxy_corner:
+      avg_avg_return:                           0.9483304
+      avg_reward_per_step:                      0.095832825
+  eval-batch-shift:
+    avg_avg_episode_length:                     64.89519
+    avg_avg_return:                             0.287274
+    avg_benchmark_regret:                       0.69961643
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.042015076
+    proxy_corner:
+      avg_avg_return:                           0.58463633
+      avg_reward_per_step:                      0.005004883
+  eval-batch-tree:
+    avg_avg_episode_length:                     16.973078
+    avg_avg_return:                             0.9481392
+    avg_benchmark_regret:                       0.03593643
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11223602
+    proxy_corner:
+      avg_avg_return:                           0.9481392
+      avg_reward_per_step:                      0.11223602
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        187695104
+    env-step-replay-before:                     95420416
+    env-step-used-before:                       187695104
+    ppo-update-before:                          114560
+    env-step-generate-before:                   92274688
+    env-step-all-after:                         187727872
+    env-step-replay-after:                      95420416
+    env-step-used-after:                        187727872
+    ppo-update-after:                           114580
+    env-step-generate-after:                    92307456
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     16.81037
+    avg_avg_return:                             0.8763721
+    avg_reward_per_step:                        0.1253357
+    proxy_corner:
+      avg_avg_return:                           0.8763721
+      avg_reward_per_step:                      0.1253357
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     16.81037
+    avg_avg_return:                             0.8763721
+    avg_reward_per_step:                        0.1253357
+    proxy_corner:
+      avg_avg_return:                           0.8763721
+      avg_reward_per_step:                      0.1253357
+  perf:
+    env_steps_per_second:                       727887.427740228
+    ppo_updates_per_second:                     42.146508608748505
+    eval-batch-orig-duration:                   0.30643358267843723
+    eval-batch-shift-duration:                  0.25878654420375824
+    eval-batch-tree-duration:                   0.24124284274876118
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.930664
+      mouse-cheese_dist_finite_wavg:            17.50513
+      mouse-cheese_dist_solvable_avg:           30.930664
+      mouse-cheese_dist_solvable_wavg:          17.505135
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999976
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.36175948
+    visit_patterns:
+      avg_first_visit_time:                     2474.4182
+      avg_last_visit_time:                      2902.1367
+      num_replay_batches:                       2912
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0009168557
+    avg_actor_approxkl3:                        0.00074234564
+    avg_actor_clipfrac:                         0.010510254
+    avg_actor_loss:                             -0.0001135628
+    avg_advantage:                              0.008027777
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00013661238
+    avg_entropy:                                0.024635961
+    avg_grad_norm_pre_clip:                     0.8526398
+    avg_loss:                                   -6.989258e-05
+    max:
+      max_actor_approxkl1:                      0.002584604
+      max_actor_approxkl3:                      0.002084481
+      max_actor_clipfrac:                       0.01977539
+      max_actor_loss:                           0.00077655166
+      max_advantage:                            0.09493321
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00016667685
+      max_entropy:                              0.035584867
+      max_grad_norm_pre_clip:                   1.6739855
+      max_loss:                                 0.00082423975
+    std:
+      std_actor_approxkl1:                      0.0006973574
+      std_actor_approxkl3:                      0.00055898976
+      std_actor_clipfrac:                       0.0043637194
+      std_actor_loss:                           0.00045957317
+      std_advantage:                            0.015677504
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          1.5474032e-05
+      std_entropy:                              0.004268926
+      std_grad_norm_pre_clip:                   0.30169475
+      std_loss:                                 0.00045880213
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.13702
+    avg_avg_return:                             0.9497452
+    avg_benchmark_regret:                       0.03508132
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09614563
+    proxy_corner:
+      avg_avg_return:                           0.9497452
+      avg_reward_per_step:                      0.09614563
+  eval-batch-shift:
+    avg_avg_episode_length:                     65.44377
+    avg_avg_return:                             0.28600103
+    avg_benchmark_regret:                       0.70088947
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.040672302
+    proxy_corner:
+      avg_avg_return:                           0.57670295
+      avg_reward_per_step:                      0.0049362183
+  eval-batch-tree:
+    avg_avg_episode_length:                     16.76511
+    avg_avg_return:                             0.9467788
+    avg_benchmark_regret:                       0.037296876
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11265564
+    proxy_corner:
+      avg_avg_return:                           0.9467788
+      avg_reward_per_step:                      0.11265564
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        188743680
+    env-step-replay-before:                     95911936
+    env-step-used-before:                       188743680
+    ppo-update-before:                          115200
+    env-step-generate-before:                   92831744
+    env-step-all-after:                         188776448
+    env-step-replay-after:                      95911936
+    env-step-used-after:                        188776448
+    ppo-update-after:                           115220
+    env-step-generate-after:                    92864512
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     17.618465
+    avg_avg_return:                             0.86729014
+    avg_reward_per_step:                        0.09176636
+    proxy_corner:
+      avg_avg_return:                           0.86729014
+      avg_reward_per_step:                      0.09176636
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     17.618465
+    avg_avg_return:                             0.86729014
+    avg_reward_per_step:                        0.09176636
+    proxy_corner:
+      avg_avg_return:                           0.86729014
+      avg_reward_per_step:                      0.09176636
+  perf:
+    env_steps_per_second:                       320775.1817965
+    ppo_updates_per_second:                     41.984395203280634
+    eval-batch-orig-duration:                   0.541510327719152
+    eval-batch-shift-duration:                  0.37135152984410524
+    eval-batch-tree-duration:                   0.23697076179087162
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.929688
+      mouse-cheese_dist_finite_wavg:            21.083384
+      mouse-cheese_dist_solvable_avg:           30.929688
+      mouse-cheese_dist_solvable_wavg:          21.083384
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.3679307
+    visit_patterns:
+      avg_first_visit_time:                     2490.0042
+      avg_last_visit_time:                      2917.3333
+      num_replay_batches:                       2927
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00067740347
+    avg_actor_approxkl3:                        0.00073188805
+    avg_actor_clipfrac:                         0.012396241
+    avg_actor_loss:                             -1.33147e-05
+    avg_advantage:                              0.011460254
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00044357777
+    avg_entropy:                                0.02298814
+    avg_grad_norm_pre_clip:                     0.7552398
+    avg_loss:                                   0.00018548605
+    max:
+      max_actor_approxkl1:                      0.001689157
+      max_actor_approxkl3:                      0.002263615
+      max_actor_clipfrac:                       0.022338867
+      max_actor_loss:                           0.0017980356
+      max_advantage:                            0.10263473
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0013627451
+      max_entropy:                              0.03084954
+      max_grad_norm_pre_clip:                   1.5059692
+      max_loss:                                 0.001844328
+    std:
+      std_actor_approxkl1:                      0.0005469513
+      std_actor_approxkl3:                      0.0005005797
+      std_actor_clipfrac:                       0.0049937763
+      std_actor_loss:                           0.0006846114
+      std_advantage:                            0.03002781
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00043525195
+      std_entropy:                              0.004151092
+      std_grad_norm_pre_clip:                   0.32198983
+      std_loss:                                 0.00071686186
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.223774
+    avg_avg_return:                             0.947561
+    avg_benchmark_regret:                       0.037265547
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09585571
+    proxy_corner:
+      avg_avg_return:                           0.947561
+      avg_reward_per_step:                      0.09585571
+  eval-batch-shift:
+    avg_avg_episode_length:                     64.371826
+    avg_avg_return:                             0.29697478
+    avg_benchmark_regret:                       0.68991566
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.046188354
+    proxy_corner:
+      avg_avg_return:                           0.55789244
+      avg_reward_per_step:                      0.005180359
+  eval-batch-tree:
+    avg_avg_episode_length:                     16.857708
+    avg_avg_return:                             0.94764197
+    avg_benchmark_regret:                       0.03643363
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11222839
+    proxy_corner:
+      avg_avg_return:                           0.94764197
+      avg_reward_per_step:                      0.11222839
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        189792256
+    env-step-replay-before:                     96501760
+    env-step-used-before:                       189792256
+    ppo-update-before:                          115840
+    env-step-generate-before:                   93290496
+    env-step-all-after:                         189825024
+    env-step-replay-after:                      96534528
+    env-step-used-after:                        189825024
+    ppo-update-after:                           115860
+    env-step-generate-after:                    93290496
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     52.54648
+    avg_avg_return:                             0.5979912
+    avg_reward_per_step:                        0.017120361
+    proxy_corner:
+      avg_avg_return:                           0.5979912
+      avg_reward_per_step:                      0.017120361
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     52.54648
+    avg_avg_return:                             0.5979912
+    avg_reward_per_step:                        0.017120361
+    proxy_corner:
+      avg_avg_return:                           0.5979912
+      avg_reward_per_step:                      0.017120361
+  perf:
+    env_steps_per_second:                       319649.16811449686
+    ppo_updates_per_second:                     42.22834085661944
+    eval-batch-orig-duration:                   0.5271545741707087
+    eval-batch-shift-duration:                  0.48621581215411425
+    eval-batch-tree-duration:                   0.4861429752781987
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             30.981201
+      mouse-cheese_dist_finite_wavg:            21.082045
+      mouse-cheese_dist_solvable_avg:           30.981201
+      mouse-cheese_dist_solvable_wavg:          21.082045
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.36476737
+    visit_patterns:
+      avg_first_visit_time:                     2502.4624
+      avg_last_visit_time:                      2936.1836
+      num_replay_batches:                       2946
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0017841522
+    avg_actor_approxkl3:                        0.0013771327
+    avg_actor_clipfrac:                         0.04001465
+    avg_actor_loss:                             -0.000785023
+    avg_advantage:                              -0.0051793214
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.003693001
+    avg_entropy:                                0.10711384
+    avg_grad_norm_pre_clip:                     1.2663692
+    avg_loss:                                   0.0009543636
+    max:
+      max_actor_approxkl1:                      0.0059836395
+      max_actor_approxkl3:                      0.0029868698
+      max_actor_clipfrac:                       0.06225586
+      max_actor_loss:                           0.00078918785
+      max_advantage:                            0.09935355
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.005746897
+      max_entropy:                              0.12127735
+      max_grad_norm_pre_clip:                   1.9390708
+      max_loss:                                 0.0032290893
+    std:
+      std_actor_approxkl1:                      0.0014554206
+      std_actor_approxkl3:                      0.000773899
+      std_actor_clipfrac:                       0.016724328
+      std_actor_loss:                           0.0010414361
+      std_advantage:                            0.08596529
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0011169771
+      std_entropy:                              0.0074736276
+      std_grad_norm_pre_clip:                   0.35508066
+      std_loss:                                 0.0011372801
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.341488
+    avg_avg_return:                             0.9482724
+    avg_benchmark_regret:                       0.036554143
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09539032
+    proxy_corner:
+      avg_avg_return:                           0.9482724
+      avg_reward_per_step:                      0.09539032
+  eval-batch-shift:
+    avg_avg_episode_length:                     63.867973
+    avg_avg_return:                             0.30197126
+    avg_benchmark_regret:                       0.6849192
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.046463013
+    proxy_corner:
+      avg_avg_return:                           0.59577286
+      avg_reward_per_step:                      0.0056533813
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.057693
+    avg_avg_return:                             0.946978
+    avg_benchmark_regret:                       0.037097592
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.112083435
+    proxy_corner:
+      avg_avg_return:                           0.946978
+      avg_reward_per_step:                      0.112083435
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        190840832
+    env-step-replay-before:                     96894976
+    env-step-used-before:                       190840832
+    ppo-update-before:                          116480
+    env-step-generate-before:                   93945856
+    env-step-all-after:                         190873600
+    env-step-replay-after:                      96894976
+    env-step-used-after:                        190873600
+    ppo-update-after:                           116500
+    env-step-generate-after:                    93978624
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.64379
+    avg_avg_return:                             0.84825313
+    avg_reward_per_step:                        0.097961426
+    proxy_corner:
+      avg_avg_return:                           0.84825313
+      avg_reward_per_step:                      0.097961426
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.64379
+    avg_avg_return:                             0.84825313
+    avg_reward_per_step:                        0.097961426
+    proxy_corner:
+      avg_avg_return:                           0.84825313
+      avg_reward_per_step:                      0.097961426
+  perf:
+    env_steps_per_second:                       298327.5299008162
+    ppo_updates_per_second:                     42.10492595681124
+    eval-batch-orig-duration:                   0.5193675206974149
+    eval-batch-shift-duration:                  0.4761991687119007
+    eval-batch-tree-duration:                   0.48172834794968367
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             31.014404
+      mouse-cheese_dist_finite_wavg:            22.906038
+      mouse-cheese_dist_solvable_avg:           31.014404
+      mouse-cheese_dist_solvable_wavg:          22.90604
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.99998
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.36757535
+    visit_patterns:
+      avg_first_visit_time:                     2519.5164
+      avg_last_visit_time:                      2947.3396
+      num_replay_batches:                       2957
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0006104896
+    avg_actor_approxkl3:                        0.0006467774
+    avg_actor_clipfrac:                         0.010412598
+    avg_actor_loss:                             -0.00041540992
+    avg_advantage:                              0.003919253
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00086718315
+    avg_entropy:                                0.023607498
+    avg_grad_norm_pre_clip:                     0.6910308
+    avg_loss:                                   -5.4258926e-06
+    max:
+      max_actor_approxkl1:                      0.0020178012
+      max_actor_approxkl3:                      0.0016766282
+      max_actor_clipfrac:                       0.020996094
+      max_actor_loss:                           0.00021142513
+      max_advantage:                            0.0934577
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0018755549
+      max_entropy:                              0.03130734
+      max_grad_norm_pre_clip:                   2.4613683
+      max_loss:                                 0.0011226247
+    std:
+      std_actor_approxkl1:                      0.0005390673
+      std_actor_approxkl3:                      0.00045235324
+      std_actor_clipfrac:                       0.006068761
+      std_actor_loss:                           0.00045923068
+      std_advantage:                            0.041261952
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00054428
+      std_entropy:                              0.0046032164
+      std_grad_norm_pre_clip:                   0.49878943
+      std_loss:                                 0.0005380359
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.202463
+    avg_avg_return:                             0.94883853
+    avg_benchmark_regret:                       0.035987973
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.096076965
+    proxy_corner:
+      avg_avg_return:                           0.94883853
+      avg_reward_per_step:                      0.096076965
+  eval-batch-shift:
+    avg_avg_episode_length:                     64.46735
+    avg_avg_return:                             0.29848868
+    avg_benchmark_regret:                       0.68840176
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.04321289
+    proxy_corner:
+      avg_avg_return:                           0.5508919
+      avg_reward_per_step:                      0.0054244995
+  eval-batch-tree:
+    avg_avg_episode_length:                     16.747948
+    avg_avg_return:                             0.94773793
+    avg_benchmark_regret:                       0.036337662
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.112724304
+    proxy_corner:
+      avg_avg_return:                           0.94773793
+      avg_reward_per_step:                      0.112724304
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        191889408
+    env-step-replay-before:                     97484800
+    env-step-used-before:                       191889408
+    ppo-update-before:                          117120
+    env-step-generate-before:                   94404608
+    env-step-all-after:                         191922176
+    env-step-replay-after:                      97517568
+    env-step-used-after:                        191922176
+    ppo-update-after:                           117140
+    env-step-generate-after:                    94404608
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     52.732933
+    avg_avg_return:                             0.5858835
+    avg_reward_per_step:                        0.016693115
+    proxy_corner:
+      avg_avg_return:                           0.5858835
+      avg_reward_per_step:                      0.016693115
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     52.732933
+    avg_avg_return:                             0.5858835
+    avg_reward_per_step:                        0.016693115
+    proxy_corner:
+      avg_avg_return:                           0.5858835
+      avg_reward_per_step:                      0.016693115
+  perf:
+    env_steps_per_second:                       292884.6375110796
+    ppo_updates_per_second:                     42.22064677226832
+    eval-batch-orig-duration:                   0.5231673773378134
+    eval-batch-shift-duration:                  0.47790035139769316
+    eval-batch-tree-duration:                   0.4718796443194151
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             31.08545
+      mouse-cheese_dist_finite_wavg:            21.107498
+      mouse-cheese_dist_solvable_avg:           31.08545
+      mouse-cheese_dist_solvable_wavg:          21.107504
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999998
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999992
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.36340874
+    visit_patterns:
+      avg_first_visit_time:                     2532.9417
+      avg_last_visit_time:                      2966.3071
+      num_replay_batches:                       2976
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0025104987
+    avg_actor_approxkl3:                        0.0017704692
+    avg_actor_clipfrac:                         0.039923098
+    avg_actor_loss:                             -0.0008869266
+    avg_advantage:                              -0.015952546
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0041305474
+    avg_entropy:                                0.10100396
+    avg_grad_norm_pre_clip:                     1.385719
+    avg_loss:                                   0.0010773431
+    max:
+      max_actor_approxkl1:                      0.0096212365
+      max_actor_approxkl3:                      0.0060534943
+      max_actor_clipfrac:                       0.076049805
+      max_actor_loss:                           0.0007707104
+      max_advantage:                            0.10271752
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0063080294
+      max_entropy:                              0.11807591
+      max_grad_norm_pre_clip:                   2.1658812
+      max_loss:                                 0.0029585874
+    std:
+      std_actor_approxkl1:                      0.0022868803
+      std_actor_approxkl3:                      0.0012987125
+      std_actor_clipfrac:                       0.017273078
+      std_actor_loss:                           0.00091080694
+      std_advantage:                            0.0907435
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00095704943
+      std_entropy:                              0.008664227
+      std_grad_norm_pre_clip:                   0.4085796
+      std_loss:                                 0.0010046208
+  eval-batch-orig:
+    avg_avg_episode_length:                     15.86306
+    avg_avg_return:                             0.951009
+    avg_benchmark_regret:                       0.033817537
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.096588135
+    proxy_corner:
+      avg_avg_return:                           0.951009
+      avg_reward_per_step:                      0.096588135
+  eval-batch-shift:
+    avg_avg_episode_length:                     63.87294
+    avg_avg_return:                             0.29683277
+    avg_benchmark_regret:                       0.6900577
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.04423523
+    proxy_corner:
+      avg_avg_return:                           0.59482455
+      avg_reward_per_step:                      0.005432129
+  eval-batch-tree:
+    avg_avg_episode_length:                     16.711464
+    avg_avg_return:                             0.94655406
+    avg_benchmark_regret:                       0.037521567
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11287689
+    proxy_corner:
+      avg_avg_return:                           0.94655406
+      avg_reward_per_step:                      0.11287689
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        192937984
+    env-step-replay-before:                     97910784
+    env-step-used-before:                       192937984
+    ppo-update-before:                          117760
+    env-step-generate-before:                   95027200
+    env-step-all-after:                         192970752
+    env-step-replay-after:                      97943552
+    env-step-used-after:                        192970752
+    ppo-update-after:                           117780
+    env-step-generate-after:                    95027200
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     53.122425
+    avg_avg_return:                             0.585171
+    avg_reward_per_step:                        0.017120361
+    proxy_corner:
+      avg_avg_return:                           0.585171
+      avg_reward_per_step:                      0.017120361
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     53.122425
+    avg_avg_return:                             0.585171
+    avg_reward_per_step:                        0.017120361
+    proxy_corner:
+      avg_avg_return:                           0.585171
+      avg_reward_per_step:                      0.017120361
+  perf:
+    env_steps_per_second:                       242165.9739387044
+    ppo_updates_per_second:                     42.31686495344531
+    eval-batch-orig-duration:                   0.6772112604230642
+    eval-batch-shift-duration:                  0.6319915698841214
+    eval-batch-tree-duration:                   0.646842266432941
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             31.081787
+      mouse-cheese_dist_finite_wavg:            17.50572
+      mouse-cheese_dist_solvable_avg:           31.081787
+      mouse-cheese_dist_solvable_wavg:          17.505722
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016804
+    scoring:
+      avg_scores:                               0.37398404
+    visit_patterns:
+      avg_first_visit_time:                     2546.8
+      avg_last_visit_time:                      2979.22
+      num_replay_batches:                       2989
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0020411897
+    avg_actor_approxkl3:                        0.0018219225
+    avg_actor_clipfrac:                         0.04041748
+    avg_actor_loss:                             -0.00022227262
+    avg_advantage:                              -0.0014964946
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.004096111
+    avg_entropy:                                0.09820848
+    avg_grad_norm_pre_clip:                     1.3649281
+    avg_loss:                                   0.0017275742
+    max:
+      max_actor_approxkl1:                      0.006421663
+      max_actor_approxkl3:                      0.005356709
+      max_actor_clipfrac:                       0.06274414
+      max_actor_loss:                           0.00094887614
+      max_advantage:                            0.10426974
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.005774971
+      max_entropy:                              0.10912472
+      max_grad_norm_pre_clip:                   2.337207
+      max_loss:                                 0.0029234227
+    std:
+      std_actor_approxkl1:                      0.001575678
+      std_actor_approxkl3:                      0.0012253086
+      std_actor_clipfrac:                       0.014534771
+      std_actor_loss:                           0.00090205163
+      std_advantage:                            0.09056238
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0009162798
+      std_entropy:                              0.005417057
+      std_grad_norm_pre_clip:                   0.5643121
+      std_loss:                                 0.0009217814
+  eval-batch-orig:
+    avg_avg_episode_length:                     15.961
+    avg_avg_return:                             0.95145583
+    avg_benchmark_regret:                       0.03337066
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09587097
+    proxy_corner:
+      avg_avg_return:                           0.95145583
+      avg_reward_per_step:                      0.09587097
+  eval-batch-shift:
+    avg_avg_episode_length:                     63.384773
+    avg_avg_return:                             0.31454095
+    avg_benchmark_regret:                       0.6723495
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.046676636
+    proxy_corner:
+      avg_avg_return:                           0.53242844
+      avg_reward_per_step:                      0.0051651
+  eval-batch-tree:
+    avg_avg_episode_length:                     16.628029
+    avg_avg_return:                             0.94955504
+    avg_benchmark_regret:                       0.03452059
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11264801
+    proxy_corner:
+      avg_avg_return:                           0.94955504
+      avg_reward_per_step:                      0.11264801
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        193986560
+    env-step-replay-before:                     98402304
+    env-step-used-before:                       193986560
+    ppo-update-before:                          118400
+    env-step-generate-before:                   95584256
+    env-step-all-after:                         194019328
+    env-step-replay-after:                      98435072
+    env-step-used-after:                        194019328
+    ppo-update-after:                           118420
+    env-step-generate-after:                    95584256
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     51.129898
+    avg_avg_return:                             0.60591364
+    avg_reward_per_step:                        0.017730713
+    proxy_corner:
+      avg_avg_return:                           0.60591364
+      avg_reward_per_step:                      0.017730713
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     51.129898
+    avg_avg_return:                             0.60591364
+    avg_reward_per_step:                        0.017730713
+    proxy_corner:
+      avg_avg_return:                           0.60591364
+      avg_reward_per_step:                      0.017730713
+  perf:
+    env_steps_per_second:                       348260.6860933379
+    ppo_updates_per_second:                     42.27393618050045
+    eval-batch-orig-duration:                   0.5227370467036963
+    eval-batch-shift-duration:                  0.4868638841435313
+    eval-batch-tree-duration:                   0.48504523653537035
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             31.22754
+      mouse-cheese_dist_finite_wavg:            22.02811
+      mouse-cheese_dist_solvable_avg:           31.22754
+      mouse-cheese_dist_solvable_wavg:          22.028112
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.420168
+    scoring:
+      avg_scores:                               0.3748492
+    visit_patterns:
+      avg_first_visit_time:                     2561.0002
+      avg_last_visit_time:                      2994.261
+      num_replay_batches:                       3004
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0030449233
+    avg_actor_approxkl3:                        0.0027619568
+    avg_actor_clipfrac:                         0.05150757
+    avg_actor_loss:                             0.0012970796
+    avg_advantage:                              -0.010702797
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0036033448
+    avg_entropy:                                0.096520275
+    avg_grad_norm_pre_clip:                     1.8468729
+    avg_loss:                                   0.0030022317
+    max:
+      max_actor_approxkl1:                      0.0069315033
+      max_actor_approxkl3:                      0.0062694517
+      max_actor_clipfrac:                       0.08178711
+      max_actor_loss:                           0.005605951
+      max_advantage:                            0.10059911
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0063947802
+      max_entropy:                              0.11042008
+      max_grad_norm_pre_clip:                   4.180737
+      max_loss:                                 0.008163814
+    std:
+      std_actor_approxkl1:                      0.0016085033
+      std_actor_approxkl3:                      0.0013998561
+      std_actor_clipfrac:                       0.01659054
+      std_actor_loss:                           0.0015171778
+      std_advantage:                            0.0845725
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.001223279
+      std_entropy:                              0.0064335093
+      std_grad_norm_pre_clip:                   0.8067137
+      std_loss:                                 0.0016896017
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.172493
+    avg_avg_return:                             0.9483638
+    avg_benchmark_regret:                       0.03646275
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09592438
+    proxy_corner:
+      avg_avg_return:                           0.9483638
+      avg_reward_per_step:                      0.09592438
+  eval-batch-shift:
+    avg_avg_episode_length:                     63.975033
+    avg_avg_return:                             0.30328333
+    avg_benchmark_regret:                       0.6836071
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.047065735
+    proxy_corner:
+      avg_avg_return:                           0.5352529
+      avg_reward_per_step:                      0.004966736
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.014568
+    avg_avg_return:                             0.9454719
+    avg_benchmark_regret:                       0.0386037
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11164093
+    proxy_corner:
+      avg_avg_return:                           0.9454719
+      avg_reward_per_step:                      0.11164093
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        195035136
+    env-step-replay-before:                     98762752
+    env-step-used-before:                       195035136
+    ppo-update-before:                          119040
+    env-step-generate-before:                   96272384
+    env-step-all-after:                         195067904
+    env-step-replay-after:                      98795520
+    env-step-used-after:                        195067904
+    ppo-update-after:                           119060
+    env-step-generate-after:                    96272384
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     54.622673
+    avg_avg_return:                             0.5558595
+    avg_reward_per_step:                        0.016235352
+    proxy_corner:
+      avg_avg_return:                           0.5558595
+      avg_reward_per_step:                      0.016235352
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     54.622673
+    avg_avg_return:                             0.5558595
+    avg_reward_per_step:                        0.016235352
+    proxy_corner:
+      avg_avg_return:                           0.5558595
+      avg_reward_per_step:                      0.016235352
+  perf:
+    env_steps_per_second:                       325923.7120631553
+    ppo_updates_per_second:                     41.91894143917549
+    eval-batch-orig-duration:                   0.5338329607620835
+    eval-batch-shift-duration:                  0.48590102791786194
+    eval-batch-tree-duration:                   0.4871233105659485
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             31.108398
+      mouse-cheese_dist_finite_wavg:            17.518686
+      mouse-cheese_dist_solvable_avg:           31.108398
+      mouse-cheese_dist_solvable_wavg:          17.518688
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999985
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.38767117
+    visit_patterns:
+      avg_first_visit_time:                     2578.2803
+      avg_last_visit_time:                      3005.3418
+      num_replay_batches:                       3015
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0016869009
+    avg_actor_approxkl3:                        0.0017141401
+    avg_actor_clipfrac:                         0.039733887
+    avg_actor_loss:                             -0.0005614754
+    avg_advantage:                              -0.023408018
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0061167865
+    avg_entropy:                                0.11035686
+    avg_grad_norm_pre_clip:                     1.374466
+    avg_loss:                                   0.0023865607
+    max:
+      max_actor_approxkl1:                      0.006572554
+      max_actor_approxkl3:                      0.0055127842
+      max_actor_clipfrac:                       0.064208984
+      max_actor_loss:                           0.00038477778
+      max_advantage:                            0.087804794
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.007221766
+      max_entropy:                              0.12479092
+      max_grad_norm_pre_clip:                   2.2313375
+      max_loss:                                 0.0037653109
+    std:
+      std_actor_approxkl1:                      0.0017879347
+      std_actor_approxkl3:                      0.0014909285
+      std_actor_clipfrac:                       0.017965313
+      std_actor_loss:                           0.0006129344
+      std_advantage:                            0.110021956
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00082516926
+      std_entropy:                              0.007127335
+      std_grad_norm_pre_clip:                   0.45952362
+      std_loss:                                 0.0008802602
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.286512
+    avg_avg_return:                             0.948622
+    avg_benchmark_regret:                       0.036204483
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09593201
+    proxy_corner:
+      avg_avg_return:                           0.948622
+      avg_reward_per_step:                      0.09593201
+  eval-batch-shift:
+    avg_avg_episode_length:                     63.62268
+    avg_avg_return:                             0.3040546
+    avg_benchmark_regret:                       0.6828359
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.047042847
+    proxy_corner:
+      avg_avg_return:                           0.5736685
+      avg_reward_per_step:                      0.0052719116
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.34946
+    avg_avg_return:                             0.94123447
+    avg_benchmark_regret:                       0.042841144
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11190033
+    proxy_corner:
+      avg_avg_return:                           0.94123447
+      avg_reward_per_step:                      0.11190033
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        196083712
+    env-step-replay-before:                     99319808
+    env-step-used-before:                       196083712
+    ppo-update-before:                          119680
+    env-step-generate-before:                   96763904
+    env-step-all-after:                         196116480
+    env-step-replay-after:                      99319808
+    env-step-used-after:                        196116480
+    ppo-update-after:                           119700
+    env-step-generate-after:                    96796672
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     17.763176
+    avg_avg_return:                             0.8684447
+    avg_reward_per_step:                        0.10321045
+    proxy_corner:
+      avg_avg_return:                           0.8684447
+      avg_reward_per_step:                      0.10321045
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     17.763176
+    avg_avg_return:                             0.8684447
+    avg_reward_per_step:                        0.10321045
+    proxy_corner:
+      avg_avg_return:                           0.8684447
+      avg_reward_per_step:                      0.10321045
+  perf:
+    env_steps_per_second:                       353907.5789429108
+    ppo_updates_per_second:                     42.26111198623055
+    eval-batch-orig-duration:                   0.5101613821461797
+    eval-batch-shift-duration:                  0.4852147288620472
+    eval-batch-tree-duration:                   0.4863515254110098
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             31.20459
+      mouse-cheese_dist_finite_wavg:            20.22907
+      mouse-cheese_dist_solvable_avg:           31.20459
+      mouse-cheese_dist_solvable_wavg:          20.229073
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.4201681
+    scoring:
+      avg_scores:                               0.3763311
+    visit_patterns:
+      avg_first_visit_time:                     2592.5664
+      avg_last_visit_time:                      3021.2979
+      num_replay_batches:                       3031
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.001141671
+    avg_actor_approxkl3:                        0.0009293627
+    avg_actor_clipfrac:                         0.012860107
+    avg_actor_loss:                             -0.00050955726
+    avg_advantage:                              6.145117e-05
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.00037738038
+    avg_entropy:                                0.02290019
+    avg_grad_norm_pre_clip:                     0.9160988
+    avg_loss:                                   -0.00034376723
+    max:
+      max_actor_approxkl1:                      0.004523997
+      max_actor_approxkl3:                      0.0025378894
+      max_actor_clipfrac:                       0.030273438
+      max_actor_loss:                           0.0010679876
+      max_advantage:                            0.0963586
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.00067741075
+      max_entropy:                              0.031019421
+      max_grad_norm_pre_clip:                   2.067464
+      max_loss:                                 0.001101643
+    std:
+      std_actor_approxkl1:                      0.0013490433
+      std_actor_approxkl3:                      0.00074819033
+      std_actor_clipfrac:                       0.00678334
+      std_actor_loss:                           0.0016641174
+      std_advantage:                            0.027827462
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00027147823
+      std_entropy:                              0.004963957
+      std_grad_norm_pre_clip:                   0.5327347
+      std_loss:                                 0.001602786
+  eval-batch-orig:
+    avg_avg_episode_length:                     15.95879
+    avg_avg_return:                             0.949604
+    avg_benchmark_regret:                       0.035222523
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09616852
+    proxy_corner:
+      avg_avg_return:                           0.949604
+      avg_reward_per_step:                      0.09616852
+  eval-batch-shift:
+    avg_avg_episode_length:                     63.77974
+    avg_avg_return:                             0.30065107
+    avg_benchmark_regret:                       0.6862394
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.04762268
+    proxy_corner:
+      avg_avg_return:                           0.55843836
+      avg_reward_per_step:                      0.0049972534
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.016438
+    avg_avg_return:                             0.94493216
+    avg_benchmark_regret:                       0.039143458
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.1124115
+    proxy_corner:
+      avg_avg_return:                           0.94493216
+      avg_reward_per_step:                      0.1124115
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        197132288
+    env-step-replay-before:                     99876864
+    env-step-used-before:                       197132288
+    ppo-update-before:                          120320
+    env-step-generate-before:                   97255424
+    env-step-all-after:                         197165056
+    env-step-replay-after:                      99876864
+    env-step-used-after:                        197165056
+    ppo-update-after:                           120340
+    env-step-generate-after:                    97288192
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     17.347084
+    avg_avg_return:                             0.87210006
+    avg_reward_per_step:                        0.09164429
+    proxy_corner:
+      avg_avg_return:                           0.87210006
+      avg_reward_per_step:                      0.09164429
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     17.347084
+    avg_avg_return:                             0.87210006
+    avg_reward_per_step:                        0.09164429
+    proxy_corner:
+      avg_avg_return:                           0.87210006
+      avg_reward_per_step:                      0.09164429
+  perf:
+    env_steps_per_second:                       292471.4395634827
+    ppo_updates_per_second:                     42.181335131900234
+    eval-batch-orig-duration:                   0.5277245845645666
+    eval-batch-shift-duration:                  0.4922567494213581
+    eval-batch-tree-duration:                   0.49309490341693163
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             31.193604
+      mouse-cheese_dist_finite_wavg:            20.22331
+      mouse-cheese_dist_solvable_avg:           31.193604
+      mouse-cheese_dist_solvable_wavg:          20.22331
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            1.0
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           50.0
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.3685605
+    visit_patterns:
+      avg_first_visit_time:                     2605.8806
+      avg_last_visit_time:                      3038.1926
+      num_replay_batches:                       3048
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.00073121715
+    avg_actor_approxkl3:                        0.00079736457
+    avg_actor_clipfrac:                         0.011987305
+    avg_actor_loss:                             -0.00027428105
+    avg_advantage:                              0.00026481674
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0002511073
+    avg_entropy:                                0.022746222
+    avg_grad_norm_pre_clip:                     0.6291094
+    avg_loss:                                   -0.00017147361
+    max:
+      max_actor_approxkl1:                      0.0019622874
+      max_actor_approxkl3:                      0.0018771751
+      max_actor_clipfrac:                       0.020263672
+      max_actor_loss:                           0.0012797185
+      max_advantage:                            0.084827244
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0007067586
+      max_entropy:                              0.031296358
+      max_grad_norm_pre_clip:                   1.0483285
+      max_loss:                                 0.0013105795
+    std:
+      std_actor_approxkl1:                      0.0005695214
+      std_actor_approxkl3:                      0.0004315222
+      std_actor_clipfrac:                       0.005337642
+      std_actor_loss:                           0.00053747563
+      std_advantage:                            0.022462184
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.00025472514
+      std_entropy:                              0.004457134
+      std_grad_norm_pre_clip:                   0.18403536
+      std_loss:                                 0.00048943417
+  eval-batch-orig:
+    avg_avg_episode_length:                     15.934
+    avg_avg_return:                             0.9507342
+    avg_benchmark_regret:                       0.034092326
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.09628296
+    proxy_corner:
+      avg_avg_return:                           0.9507342
+      avg_reward_per_step:                      0.09628296
+  eval-batch-shift:
+    avg_avg_episode_length:                     63.79801
+    avg_avg_return:                             0.29509073
+    avg_benchmark_regret:                       0.69179976
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.04688263
+    proxy_corner:
+      avg_avg_return:                           0.6021082
+      avg_reward_per_step:                      0.005279541
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.080658
+    avg_avg_return:                             0.94435185
+    avg_benchmark_regret:                       0.039723717
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11229706
+    proxy_corner:
+      avg_avg_return:                           0.94435185
+      avg_reward_per_step:                      0.11229706
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+===========================================================
+  step:
+    env-step-all-before:                        198180864
+    env-step-replay-before:                     100335616
+    env-step-used-before:                       198180864
+    ppo-update-before:                          120960
+    env-step-generate-before:                   97845248
+    env-step-all-after:                         198213632
+    env-step-replay-after:                      100368384
+    env-step-used-after:                        198213632
+    ppo-update-after:                           120980
+    env-step-generate-after:                    97845248
+  train-replay:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     51.97634
+    avg_avg_return:                             0.61189604
+    avg_reward_per_step:                        0.017333984
+    proxy_corner:
+      avg_avg_return:                           0.61189604
+      avg_reward_per_step:                      0.017333984
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     51.97634
+    avg_avg_return:                             0.61189604
+    avg_reward_per_step:                        0.017333984
+    proxy_corner:
+      avg_avg_return:                           0.61189604
+      avg_reward_per_step:                      0.017333984
+  perf:
+    env_steps_per_second:                       291235.96681812475
+    ppo_updates_per_second:                     42.075614838546535
+    eval-batch-orig-duration:                   0.5383859304711223
+    eval-batch-shift-duration:                  0.4873632974922657
+    eval-batch-tree-duration:                   0.47974641527980566
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             31.307129
+      mouse-cheese_dist_finite_wavg:            15.743446
+      mouse-cheese_dist_solvable_avg:           31.307129
+      mouse-cheese_dist_solvable_wavg:          15.743448
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.9999999
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.37901768
+    visit_patterns:
+      avg_first_visit_time:                     2617.4976
+      avg_last_visit_time:                      3053.36
+      num_replay_batches:                       3063
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0013903839
+    avg_actor_approxkl3:                        0.0012617492
+    avg_actor_clipfrac:                         0.034259032
+    avg_actor_loss:                             -0.0002883846
+    avg_advantage:                              -0.011638111
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0026050108
+    avg_entropy:                                0.092170075
+    avg_grad_norm_pre_clip:                     1.253348
+    avg_loss:                                   0.0009219507
+    max:
+      max_actor_approxkl1:                      0.0038410057
+      max_actor_approxkl3:                      0.0032699327
+      max_actor_clipfrac:                       0.060913086
+      max_actor_loss:                           0.0012355596
+      max_advantage:                            0.08110243
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.0052985414
+      max_entropy:                              0.10173008
+      max_grad_norm_pre_clip:                   1.7977028
+      max_loss:                                 0.0032022307
+    std:
+      std_actor_approxkl1:                      0.0009733396
+      std_actor_approxkl3:                      0.00078071974
+      std_actor_clipfrac:                       0.013999888
+      std_actor_loss:                           0.00063266023
+      std_advantage:                            0.07209195
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0010828566
+      std_entropy:                              0.0051860684
+      std_grad_norm_pre_clip:                   0.29643932
+      std_loss:                                 0.0007799953
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.250378
+    avg_avg_return:                             0.948915
+    avg_benchmark_regret:                       0.035911534
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.096199036
+    proxy_corner:
+      avg_avg_return:                           0.948915
+      avg_reward_per_step:                      0.096199036
+  eval-batch-shift:
+    avg_avg_episode_length:                     64.36116
+    avg_avg_return:                             0.29093844
+    avg_benchmark_regret:                       0.69595206
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.04550171
+    proxy_corner:
+      avg_avg_return:                           0.62523454
+      avg_reward_per_step:                      0.005821228
+  eval-batch-tree:
+    avg_avg_episode_length:                     16.683182
+    avg_avg_return:                             0.95031464
+    avg_benchmark_regret:                       0.03376094
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11292267
+    proxy_corner:
+      avg_avg_return:                           0.95031464
+      avg_reward_per_step:                      0.11292267
+===========================================================
+===========================================================
+  step:
+    env-step-all-before:                        199229440
+    env-step-replay-before:                     100728832
+    env-step-used-before:                       199229440
+    ppo-update-before:                          121600
+    env-step-generate-before:                   98500608
+    env-step-all-after:                         199262208
+    env-step-replay-after:                      100728832
+    env-step-used-after:                        199262208
+    ppo-update-after:                           121620
+    env-step-generate-after:                    98533376
+  train-generate:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.059921
+    avg_avg_return:                             0.8555519
+    avg_reward_per_step:                        0.09152222
+    proxy_corner:
+      avg_avg_return:                           0.8555519
+      avg_reward_per_step:                      0.09152222
+  train-all:
+    shift_proportion:                           0.0
+    avg_avg_episode_length:                     19.059921
+    avg_avg_return:                             0.8555519
+    avg_reward_per_step:                        0.09152222
+    proxy_corner:
+      avg_avg_return:                           0.8555519
+      avg_reward_per_step:                      0.09152222
+  perf:
+    env_steps_per_second:                       257972.94774023688
+    ppo_updates_per_second:                     41.56587450210187
+    eval-batch-orig-duration:                   0.5406419923529029
+    eval-batch-shift-duration:                  0.4874941222369671
+    eval-batch-tree-duration:                   0.4682203261181712
+  ued:
+    distances:
+      cheese-corne_dist_avg:                    0.0
+      cheese-corne_dist_wavg:                   0.0
+      levels_cheese_is_in_corner_avg:           1.0
+      mouse-cheese_dist_finite_avg:             31.291992
+      mouse-cheese_dist_finite_wavg:            19.335518
+      mouse-cheese_dist_solvable_avg:           31.291992
+      mouse-cheese_dist_solvable_wavg:          19.33552
+      solvable_avg:                             1.0
+      solvable_num:                             4096
+      solvable_wavg:                            0.99999994
+    layout:
+      levels16_img:                             float32(55, 55, 3)
+      num_walls_avg:                            50.0
+      num_walls_wavg:                           49.999996
+      prop_walls_avg:                           0.42016804
+      prop_walls_wavg:                          0.42016807
+    scoring:
+      avg_scores:                               0.39383623
+    visit_patterns:
+      avg_first_visit_time:                     2631.4824
+      avg_last_visit_time:                      3064.3135
+      num_replay_batches:                       3074
+    eta:                                        0.5
+  ppo:
+    avg_actor_approxkl1:                        0.0011322027
+    avg_actor_approxkl3:                        0.00077913067
+    avg_actor_clipfrac:                         0.012353516
+    avg_actor_loss:                             -0.0009438569
+    avg_advantage:                              0.0013640247
+    avg_critic_clipfrac:                        0.0
+    avg_critic_loss:                            0.0005268305
+    avg_entropy:                                0.022980323
+    avg_grad_norm_pre_clip:                     0.5301755
+    avg_loss:                                   -0.000703422
+    max:
+      max_actor_approxkl1:                      0.0042620897
+      max_actor_approxkl3:                      0.0018897604
+      max_actor_clipfrac:                       0.028564453
+      max_actor_loss:                           0.00065124454
+      max_advantage:                            0.08962959
+      max_critic_clipfrac:                      0.0
+      max_critic_loss:                          0.001278682
+      max_entropy:                              0.033428643
+      max_grad_norm_pre_clip:                   1.0177543
+      max_loss:                                 0.0006858248
+    std:
+      std_actor_approxkl1:                      0.0010181564
+      std_actor_approxkl3:                      0.00050063164
+      std_actor_clipfrac:                       0.006450766
+      std_actor_loss:                           0.0015252737
+      std_advantage:                            0.03242016
+      std_critic_clipfrac:                      0.0
+      std_critic_loss:                          0.0004009114
+      std_entropy:                              0.004186481
+      std_grad_norm_pre_clip:                   0.19887613
+      std_loss:                                 0.0014261352
+  eval-batch-orig:
+    avg_avg_episode_length:                     16.4351
+    avg_avg_return:                             0.9451646
+    avg_benchmark_regret:                       0.039661862
+    avg_benchmark_return:                       0.9848265
+    avg_reward_per_step:                        0.096321106
+    proxy_corner:
+      avg_avg_return:                           0.9451646
+      avg_reward_per_step:                      0.096321106
+  eval-batch-shift:
+    avg_avg_episode_length:                     64.47746
+    avg_avg_return:                             0.29074204
+    avg_benchmark_regret:                       0.69614846
+    avg_benchmark_return:                       0.98689044
+    avg_reward_per_step:                        0.045822144
+    proxy_corner:
+      avg_avg_return:                           0.594556
+      avg_reward_per_step:                      0.005355835
+  eval-batch-tree:
+    avg_avg_episode_length:                     17.059929
+    avg_avg_return:                             0.9445468
+    avg_benchmark_regret:                       0.039528765
+    avg_benchmark_return:                       0.9840756
+    avg_reward_per_step:                        0.11241913
+    proxy_corner:
+      avg_avg_return:                           0.9445468
+      avg_reward_per_step:                      0.11241913
+===========================================================
+saving checkpoint (wandb will sync at end of run)...
+training run complete.
+finishing checkpoints...
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/requirements.txt b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/requirements.txt
new file mode 100644
index 0000000000000000000000000000000000000000..9a34816762c35b9560dad283b25c61a2fd113170
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/requirements.txt
@@ -0,0 +1,254 @@
+kiwisolver==1.4.8
+aulis==0.1.0
+tokenizers==0.20.3
+xmltodict==0.14.2
+hf-xet==1.1.4
+timaeus-cli==0.1.0
+jsonlines==4.0.0
+DataProperty==1.1.0
+timaeus-aether==0.0.0
+nvidia-nvtx-cu12==12.6.77
+cachetools==5.5.2
+devinterp==1.3.2
+wcwidth==0.2.13
+pydantic_core==2.33.2
+tqdm-multiprocess==0.0.11
+protobuf==5.29.5
+decorator==4.4.2
+pyasn1==0.6.1
+zipp==3.23.0
+pre_commit==4.2.0
+typepy==1.3.4
+etils==1.13.0
+shellingham==1.5.4
+aiohttp==3.12.13
+prettytable==3.16.0
+readchar==4.2.1
+sentencepiece==0.2.0
+seaborn==0.13.2
+paramiko==3.5.1
+cftime==1.6.4.post1
+tabledata==1.3.4
+colorama==0.4.6
+GitPython==3.1.44
+cycler==0.12.1
+tcolorpy==0.1.7
+cfgv==3.4.0
+psutil==7.0.0
+aioitertools==0.12.0
+aiosignal==1.3.2
+humanize==4.12.3
+termcolor==3.1.0
+zope.interface==7.2
+pathvalidate==3.3.1
+nodeenv==1.9.1
+mbstrdecoder==1.1.4
+chardet==5.2.0
+zope.event==5.0
+cloudpickle==3.1.1
+opt_einsum==3.4.0
+MarkupSafe==3.0.2
+requests_ntlm==1.3.0
+pybind11==2.13.6
+fonttools==4.58.4
+gast==0.6.0
+packaging==25.0
+proglog==0.1.12
+Jinja2==3.1.6
+wadler_lindig==0.1.6
+fancy-einsum==0.0.3
+iniconfig==2.1.0
+pluggy==1.6.0
+idna==3.10
+nest-asyncio==1.6.0
+better-abc==0.0.3
+matplotlib==3.10.3
+portalocker==3.2.0
+tqdm==4.67.1
+smmap==5.0.2
+rich==14.0.0
+distro==1.9.0
+xarray==2025.6.1
+bcrypt==4.3.0
+regex==2024.11.6
+aiobotocore==2.22.0
+jmespath==1.0.1
+frozenlist==1.7.0
+typer==0.16.0
+cryptography==45.0.4
+python-dotenv==1.1.0
+pytest-xdist==3.7.0
+boto3==1.37.3
+grpc-google-iam-v1==0.14.2
+threadpoolctl==3.6.0
+aiohappyeyeballs==2.6.1
+pywinrm==0.5.0
+platformdirs==4.3.8
+nvidia-cuda-runtime-cu12==12.6.77
+aiofiles==24.1.0
+google-api-core==2.25.1
+proto-plus==1.26.1
+nvidia-cufile-cu12==1.11.1.6
+mpmath==1.3.0
+PyYAML==6.0.2
+filelock==3.18.0
+typing-inspection==0.4.1
+syrupy==4.9.1
+distlib==0.3.9
+annotated-types==0.7.0
+more-itertools==10.7.0
+multiprocess==0.70.16
+grpcio-status==1.71.2
+six==1.17.0
+requests==2.32.4
+mdurl==0.1.2
+pyinfra==3.3.1
+typing_extensions==4.14.0
+greenlet==3.2.3
+lxml==5.4.0
+einops==0.8.1
+ssh-python==1.1.1
+networkx==3.5
+attrs==25.3.0
+pyasn1_modules==0.4.2
+toolz==1.0.0
+joblib==1.5.1
+diskcache==5.6.3
+multidict==6.5.0
+gitdb==4.0.12
+transformer-lens==2.15.4
+execnet==2.1.1
+safetensors==0.5.3
+rsa==4.9.1
+zstandard==0.23.0
+google-cloud-compute==1.31.0
+rouge_score==0.1.2
+importlib_resources==6.5.2
+cffi==1.17.1
+pyspnego==0.11.2
+aioboto3==14.3.0
+Pygments==2.19.1
+tabulate==0.9.0
+typeguard==4.4.3
+click==8.2.1
+nvidia-cuda-cupti-cu12==12.6.80
+word2number==1.1
+pycparser==2.22
+gevent==25.5.1
+transformers-stream-generator==0.0.5
+sqlitedict==2.1.0
+datasets==3.6.0
+netCDF4==1.7.2
+msgpack==1.1.1
+urllib3==2.4.0
+pillow==11.2.1
+PyNaCl==1.5.0
+parallel-ssh==2.14.0
+nltk==3.9.1
+python-dateutil==2.9.0.post0
+yarl==1.20.1
+pyparsing==3.2.3
+ml_dtypes==0.5.3
+evaluate==0.4.3
+numexpr==2.11.0
+pydantic==2.11.7
+ssh2-python==1.1.2.post1
+moviepy==2.2.1
+markdown-it-py==3.0.0
+googleapis-common-protos==1.70.0
+xxhash==3.5.0
+dill==0.3.8
+sacrebleu==2.5.1
+identify==2.6.12
+fsspec==2025.3.0
+dm-tree==0.1.9
+chex==0.1.89
+huggingface-hub==0.33.0
+certifi==2025.6.15
+imageio-ffmpeg==0.6.0
+scipy==1.15.3
+pytz==2025.2
+pytablewriter==1.2.1
+optax==0.2.5
+google-cloud-tpu==1.23.2
+nvidia-curand-cu12==10.3.7.77
+charset-normalizer==3.4.2
+wandb-workspaces==0.1.16
+s3transfer==0.11.3
+absl-py==2.3.0
+accelerate==1.7.0
+wrapt==1.17.2
+narwhals==1.43.0
+virtualenv==20.31.2
+peft==0.15.2
+triton==3.3.1
+sympy==1.14.0
+nvidia-cusolver-cu12==11.7.1.2
+contourpy==1.3.2
+nvidia-cusparselt-cu12==0.6.3
+nvidia-cufft-cu12==11.3.0.4
+ruff==0.12.0
+nvidia-nccl-cu12==2.26.2
+nvidia-cusparse-cu12==12.5.4.2
+google-cloud-secret-manager==2.24.0
+pytest==7.4.3
+google-auth==2.40.3
+beartype==0.14.1
+sentry-sdk==2.30.0
+distrax==0.1.5
+grpcio==1.73.0
+pyarrow==20.0.0
+tensorflow-probability==0.25.0
+imageio==2.37.0
+wandb==0.21.0
+propcache==0.3.2
+tensorstore==0.1.76
+numpy==2.3.1
+nvidia-cuda-nvrtc-cu12==12.6.77
+nvidia-nvjitlink-cu12==12.6.85
+setuptools==80.9.0
+tzdata==2025.2
+transformers==4.45.2
+pandas==2.3.0
+scikit-learn==1.7.0
+lm_eval==0.4.5
+botocore==1.37.3
+torch==2.7.1
+plotly==6.1.2
+ipython_pygments_lexers==1.1.1
+comm==0.2.3
+matplotlib-inline==0.1.7
+ptyprocess==0.7.0
+jupyter_core==5.8.1
+executing==2.2.0
+pexpect==4.9.0
+pure_eval==0.2.3
+asttokens==3.0.0
+stack-data==0.6.3
+traitlets==5.14.3
+jupyter_client==8.6.3
+parso==0.8.4
+ipykernel==6.30.1
+pyzmq==27.0.1
+tornado==6.5.1
+ipython==9.4.0
+prompt_toolkit==3.0.51
+debugpy==1.8.16
+jedi==0.19.2
+jaxtyping==0.3.2
+jax-cuda12-plugin==0.6.2
+jax-cuda12-pjrt==0.6.2
+jaxlib==0.6.2
+jax==0.6.2
+nvidia-cublas-cu12==12.6.4.1
+nvidia-cudnn-cu12==9.5.1.17
+treescope==0.1.9
+flax==0.11.0
+simplejson==3.20.1
+orbax-checkpoint==0.11.20
+pip==24.0
+jaxgmg==0.0.0
+aulis==0.1.0
+devinterp==1.3.2
+timaeus-aether==0.0.0
+timaeus-cli==0.1.0
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/wandb-metadata.json b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/wandb-metadata.json
new file mode 100644
index 0000000000000000000000000000000000000000..08013aa40058b71f8fbba1845308364c284be2df
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/wandb-metadata.json
@@ -0,0 +1,60 @@
+{
+  "os":  "Linux-5.4.0-216-generic-x86_64-with-glibc2.39",
+  "python":  "CPython 3.11.11",
+  "startedAt":  "2025-08-07T18:02:11.433582Z",
+  "args":  [
+    "train",
+    "corner",
+    "--wall-prob",
+    "1.0",
+    "--env-size",
+    "13",
+    "--log-gifs",
+    "--env-layout",
+    "tree",
+    "--wandb-project",
+    "jaxgmg-wall-prob-tree",
+    "--wandb-name",
+    "wp:1.0_alpha:0",
+    "--num-total-env-steps",
+    "200_000_000",
+    "--prob-shift",
+    "0",
+    "--keep-all-checkpoints",
+    "--num-cycles-per-checkpoint",
+    "64"
+  ],
+  "program":  "-m jaxgmg.__main__",
+  "git":  {
+    "remote":  "git@github.com:timaeus-research/timaeus.git",
+    "commit":  "7415666d6a2e5d63453ea9fadaf6cecf63256a59"
+  },
+  "email":  "david@timaeus.co",
+  "root":  "/workspace/timaeus/projects/rl",
+  "host":  "6e66e15ca303",
+  "executable":  "/workspace/timaeus/.venv/bin/python",
+  "cpu_count":  24,
+  "cpu_count_logical":  48,
+  "gpu":  "NVIDIA GeForce RTX 3090",
+  "gpu_count":  1,
+  "disk":  {
+    "/":  {
+      "total":  "102005473280",
+      "used":  "43426013184"
+    }
+  },
+  "memory":  {
+    "total":  "135066595328"
+  },
+  "gpu_nvidia":  [
+    {
+      "name":  "NVIDIA GeForce RTX 3090",
+      "memoryTotal":  "25769803776",
+      "cudaCores":  10496,
+      "architecture":  "Ampere",
+      "uuid":  "GPU-43bd7524-bf94-73a9-1460-2b36dd09a208"
+    }
+  ],
+  "cudaVersion":  "12.8",
+  "writerId":  "2p3j9mzw6ml6zq9wucq3537c29k5b8sm"
+}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/wandb-summary.json b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/wandb-summary.json
new file mode 100644
index 0000000000000000000000000000000000000000..29a5f2f995d652614b71b1de449078654ad5e016
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/files/wandb-summary.json
@@ -0,0 +1 @@
+{"ppo/avg_grad_norm_pre_clip":0.530175507068634,"ppo/avg_entropy":0.022980323061347008,"ppo/max/max_actor_clipfrac":0.028564453125,"train-all/proxy_corner/avg_reward_per_step":0.091522216796875,"eval-batch-tree/proxy_corner/avg_reward_per_step":0.11241912841796875,"train-all/avg_reward_per_step":0.091522216796875,"ued/distances/cheese-corne_dist_avg":0,"ppo/std/std_advantage":0.03242015838623047,"step/env-step-generate-before":98500608,"ued/distances/solvable_wavg":0.9999999403953552,"eval-batch-orig/avg_avg_episode_length":16.435100555419922,"ppo/avg_actor_approxkl3":0.0007791306707076728,"train-replay/avg_avg_return":0.6118960380554199,"ppo/max/max_loss":0.0006858247797936201,"ued/layout/num_walls_wavg":49.999996185302734,"eval-batch-tree/proxy_corner/avg_avg_return":0.9445468187332153,"_runtime":3397.391956085,"perf/env_steps_per_second":257972.94774023688,"ppo/std/std_actor_clipfrac":0.006450766231864691,"ued/distances/cheese-corne_dist_wavg":0,"ued/distances/mouse-cheese_dist_finite_wavg":19.33551788330078,"train-generate/avg_avg_episode_length":19.059921264648438,"ppo/std/std_actor_loss":0.0015252736629918218,"train-replay/proxy_corner/avg_avg_return":0.6118960380554199,"eval-batch-shift/avg_benchmark_regret":0.6961484551429749,"eval-batch-shift/avg_benchmark_return":0.986890435218811,"ued/distances/mouse-cheese_dist_solvable_avg":31.2919921875,"ued/eta":0.5,"ppo/avg_critic_clipfrac":0,"step/env-step-replay-after":100728832,"train-all/avg_avg_return":0.8555518984794617,"eval-batch-shift/proxy_corner/avg_avg_return":0.5945559740066528,"train-replay/avg_reward_per_step":0.017333984375,"_wandb":{"runtime":3397},"train-generate/avg_avg_return":0.8555518984794617,"ppo/avg_actor_clipfrac":0.012353516183793545,"train-replay/rollouts_gif":{"height":225,"_type":"video-file","sha256":"b3d811352e422cf93cc6e256c9e504c0347fed15b61813fd2c39ee4497c9eda9","size":164928,"path":"media/videos/train-replay/rollouts_gif_5088_b3d811352e422cf93cc6.gif","width":225},"train-generate/proxy_corner/avg_reward_per_step":0.091522216796875,"step/env-step-all-after":199262208,"eval-batch-tree/avg_reward_per_step":0.11241912841796875,"ppo/std/std_actor_approxkl3":0.0005006316350772977,"eval-batch-shift/proxy_corner/avg_reward_per_step":0.0053558349609375,"eval-batch-shift/avg_reward_per_step":0.0458221435546875,"perf/eval-batch-orig-duration":0.5406419923529029,"ued/distances/mouse-cheese_dist_finite_avg":31.2919921875,"train-generate/rollouts_gif":{"sha256":"14950c5dc884e09b130d0db1ec62f29d6b65619f8f1f5a0aa30383515c54dc6a","size":295898,"path":"media/videos/train-generate/rollouts_gif_4064_14950c5dc884e09b130d.gif","width":225,"height":225,"_type":"video-file"},"ppo/max/max_grad_norm_pre_clip":1.017754316329956,"eval-batch-tree/avg_avg_return":0.9445468187332153,"ppo/std/std_grad_norm_pre_clip":0.19887612760066986,"step/ppo-update-after":121620,"ppo/std/std_actor_approxkl1":0.0010181564139202237,"step/env-step-generate-after":98533376,"ued/layout/levels16_img":{"_type":"image-file","sha256":"d51a82bd747fdf1842049c1660c6674d33ce75fbd106bc3689368c5e159d5835","size":936,"path":"media/images/ued/layout/levels16_img_6048_d51a82bd747fdf184204.png","format":"png","width":55,"height":55},"ued/distances/solvable_avg":1,"eval-batch-tree/rollouts_gif":{"sha256":"2401e16bcdff1cb057e2f15ca7b274142848bf0e182705c26b605eb0b31a5ebe","size":292025,"path":"media/videos/eval-batch-tree/rollouts_gif_5088_2401e16bcdff1cb057e2.gif","width":225,"height":225,"_type":"video-file"},"ued/visit_patterns/avg_last_visit_time":3064.3134765625,"step/ppo-update-before":121600,"ppo/avg_advantage":0.0013640247052535415,"step/env-step-used-before":199229440,"ppo/avg_actor_approxkl1":0.0011322026839479804,"eval-batch-orig/avg_reward_per_step":0.09632110595703125,"eval-batch-orig/avg_benchmark_regret":0.03966186195611954,"eval-batch-tree/avg_benchmark_regret":0.03952876478433609,"eval-batch-shift/avg_avg_episode_length":64.47746276855469,"train-all/proxy_corner/avg_avg_return":0.8555518984794617,"ued/distances/mouse-cheese_dist_solvable_wavg":19.335519790649414,"ppo/avg_loss":-0.0007034220034256577,"train-replay/proxy_corner/avg_reward_per_step":0.017333984375,"ued/layout/num_walls_avg":50,"_step":6080,"perf/eval-batch-tree-duration":0.4682203261181712,"train-generate/shift_proportion":0,"ppo/avg_actor_loss":-0.0009438568959012628,"eval-batch-orig/avg_avg_return":0.9451646208763123,"train-generate/avg_reward_per_step":0.091522216796875,"ppo/max/max_critic_clipfrac":0,"train-generate/proxy_corner/avg_avg_return":0.8555518984794617,"ued/visit_patterns/num_replay_batches":3074,"train-all/shift_proportion":0,"ppo/std/std_entropy":0.004186480771750212,"ued/distances/solvable_num":4096,"ppo/max/max_actor_loss":0.0006512445397675037,"eval-batch-orig/proxy_corner/avg_avg_return":0.9451646208763123,"ued/layout/prop_walls_avg":0.42016804218292236,"_timestamp":1.7545931192385986e+09,"ppo/std/std_critic_clipfrac":0,"perf/ppo_updates_per_second":41.56587450210187,"ppo/std/std_critic_loss":0.00040091140544973314,"step/env-step-replay-before":100728832,"eval-batch-orig/rollouts_gif":{"size":288474,"path":"media/videos/eval-batch-orig/rollouts_gif_5088_949a784c4acec1cebdc6.gif","width":225,"height":225,"_type":"video-file","sha256":"949a784c4acec1cebdc62fbcaa8a7d4498de4ce575e470b08aa74219f4720192"},"train-replay/avg_avg_episode_length":51.976341247558594,"eval-batch-orig/proxy_corner/avg_reward_per_step":0.09632110595703125,"eval-batch-shift/avg_avg_return":0.29074203968048096,"step/env-step-all-before":199229440,"ppo/max/max_advantage":0.08962959051132202,"eval-batch-shift/rollouts_gif":{"width":225,"height":225,"_type":"video-file","sha256":"6995668c199776ef24f5466ddb31e966d234723f24d6e6ab4247221bdc141d0c","size":185136,"path":"media/videos/eval-batch-shift/rollouts_gif_5088_6995668c199776ef24f5.gif"},"ppo/max/max_actor_approxkl1":0.004262089729309082,"ppo/max/max_entropy":0.03342864289879799,"train-all/avg_avg_episode_length":19.059921264648438,"ppo/std/std_loss":0.0014261351898312569,"ppo/avg_critic_loss":0.000526830495800823,"eval-batch-orig/avg_benchmark_return":0.9848265051841736,"ppo/max/max_critic_loss":0.0012786820298060775,"ued/scoring/avg_scores":0.39383623003959656,"eval-batch-tree/avg_avg_episode_length":17.05992889404297,"ued/visit_patterns/avg_first_visit_time":2631.482421875,"eval-batch-tree/avg_benchmark_return":0.9840756058692932,"ppo/max/max_actor_approxkl3":0.001889760373160243,"ued/distances/levels_cheese_is_in_corner_avg":1,"train-replay/shift_proportion":0,"train-all/rollouts_gif":{"sha256":"b3d811352e422cf93cc6e256c9e504c0347fed15b61813fd2c39ee4497c9eda9","size":164928,"path":"media/videos/train-all/rollouts_gif_5088_b3d811352e422cf93cc6.gif","width":225,"height":225,"_type":"video-file"},"step/env-step-used-after":199262208,"perf/eval-batch-shift-duration":0.4874941222369671,"ued/layout/prop_walls_wavg":0.42016807198524475}
\ No newline at end of file
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/logs/debug-core.log b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/logs/debug-core.log
new file mode 100644
index 0000000000000000000000000000000000000000..21bcb405d68608dfd603209cc320ae6523440aa8
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/logs/debug-core.log
@@ -0,0 +1,16 @@
+{"time":"2025-08-07T18:02:11.485193675Z","level":"INFO","msg":"main: starting server","port-filename":"/tmp/tmpy_8epspi/port-3869025.txt","pid":3869025,"log-level":0,"disable-analytics":false,"shutdown-on-parent-exit":false,"enable-dcgm-profiling":false}
+{"time":"2025-08-07T18:02:11.486270205Z","level":"INFO","msg":"server: will exit if parent process dies","ppid":3869025}
+{"time":"2025-08-07T18:02:11.486186414Z","level":"INFO","msg":"server: accepting connections","addr":{"Name":"/tmp/wandb-3869025-3869327-3238057737/socket","Net":"unix"}}
+{"time":"2025-08-07T18:02:11.649809848Z","level":"INFO","msg":"connection: ManageConnectionData: new connection created","id":"1(@)"}
+{"time":"2025-08-07T18:02:11.662311956Z","level":"INFO","msg":"handleInformInit: received","streamId":"wp_1.0_alpha_0","id":"1(@)"}
+{"time":"2025-08-07T18:02:12.019803774Z","level":"INFO","msg":"handleInformInit: stream started","streamId":"wp_1.0_alpha_0","id":"1(@)"}
+{"time":"2025-08-07T18:58:52.084257818Z","level":"INFO","msg":"handleInformFinish: finish message received","streamId":"wp_1.0_alpha_0","id":"1(@)"}
+{"time":"2025-08-07T18:58:52.093285687Z","level":"INFO","msg":"handleInformFinish: stream closed","streamId":"wp_1.0_alpha_0","id":"1(@)"}
+{"time":"2025-08-07T18:58:53.063461323Z","level":"INFO","msg":"handleInformTeardown: server teardown initiated","id":"1(@)"}
+{"time":"2025-08-07T18:58:53.06350975Z","level":"INFO","msg":"handleInformTeardown: server shutdown complete","id":"1(@)"}
+{"time":"2025-08-07T18:58:53.063526978Z","level":"INFO","msg":"server is shutting down"}
+{"time":"2025-08-07T18:58:53.063538235Z","level":"INFO","msg":"connection: closing","id":"1(@)"}
+{"time":"2025-08-07T18:58:53.063686605Z","level":"INFO","msg":"connection: closed successfully","id":"1(@)"}
+{"time":"2025-08-07T18:58:53.06373218Z","level":"INFO","msg":"connection: ManageConnectionData: connection closed","id":"1(@)"}
+{"time":"2025-08-07T18:58:53.063657164Z","level":"INFO","msg":"server: listener closed","addr":{"Name":"/tmp/wandb-3869025-3869327-3238057737/socket","Net":"unix"}}
+{"time":"2025-08-07T18:58:53.063783123Z","level":"INFO","msg":"server is closed"}
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/logs/debug-internal.log b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/logs/debug-internal.log
new file mode 100644
index 0000000000000000000000000000000000000000..f08029548b0f038a9cb5e4bd7e8536b90024409d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/logs/debug-internal.log
@@ -0,0 +1,1167 @@
+{"time":"2025-08-07T18:02:11.662606973Z","level":"INFO","msg":"stream: starting","core version":"0.21.0"}
+{"time":"2025-08-07T18:02:12.019688846Z","level":"INFO","msg":"stream: created new stream","id":"wp_1.0_alpha_0"}
+{"time":"2025-08-07T18:02:12.019782955Z","level":"INFO","msg":"stream: started","id":"wp_1.0_alpha_0"}
+{"time":"2025-08-07T18:02:12.019879857Z","level":"INFO","msg":"writer: Do: started","stream_id":"wp_1.0_alpha_0"}
+{"time":"2025-08-07T18:02:12.019886042Z","level":"INFO","msg":"sender: started","stream_id":"wp_1.0_alpha_0"}
+{"time":"2025-08-07T18:02:12.019948055Z","level":"INFO","msg":"handler: started","stream_id":"wp_1.0_alpha_0"}
+{"time":"2025-08-07T18:58:50.262815671Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0, Name: checkpoints/1920/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1920/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=119fcb55d98110d4a2000064f38c13badb037e72676efb1f4974b6c091cf613b2e42e2ed1a8acb75eab518cc0d56b2709d3577bb8558a562c6b3e732bbf061c1b6c9baa341a9b098e832e6577c734ef66dc6c101622a42825c5a754ae0a55de0ea131ce8e8b3f525db9ca55cd1df6b7298f74fecc850c3abd13ccf999c045b958c180daf17ccdf72b930ae7e8934136f489fc2490c1358c30c3e81f0b0272540e2cba7e1db1edb3f0c15aa99fb8ca1075450703072a5dc4979c0eef11794727102fa6f64c9127d95b52f70c7544d1d751714fc45ca4b07ce1725c10106b053dcf1e3b51b639b2577859abd9b2aadebb37d23da7b41364346b5a40e77e736ecf6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.262913168Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/array_metadatas, Name: checkpoints/1024/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1024/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=67519dc32bf9e1708d7d36b5fee60e300c4de9d75d40bfa83bf954e81d8ebf1c7f7085fa152456e7b47ead4620099b80c3b4da4b6f2e39b0b259446b9bfeccb30e4438668e07cb0bfd85a044eb117682277cf225dd00e8460c79e81779c641688c86e22849f04bb1ee0c16a78026ced16b36de5be4d542b7fcedb06b5b4da6710b572d94be0b871af47b8890ae182f8df81c1dbac3aa095ecf1809f08aba0312d6226b6023c4997a451da814e4cbf8e1325d604e544662fcd8147d8e5175429f3d931be9fd1364638fa7eeaed52e702c2178308eb9fba0870dfaeb0452b64bbf9092ea3b94ca385351b6d18cb1109ae7dc4d6f3099353ecd72eca45f6375ad83&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.262974561Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/array_metadatas, Name: checkpoints/4416/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4416/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=07d5f714800fd6ee96d9e0631a8b0d6be760cb8e1c06b01bb4f7ef253c3c9d89c32ee7a1435a1596ca09df0f12eb03430d9f33f53b776a6911bc9ac5367238a367fa636131055e891d34dab64405c57f0eeafd658cc527be71cf2a2f81c8a4b6d8af551bdd58225e8e6ef218bd1345d6395188013670c1ba5abd0fd3b5aa30b3af08fdcf5a76dd0a7cfa16ebbb244f066e53f2a4701ebc4d03d46bde820acb43c589f402f000b909b148eace171553151142717deae48b436533fc959ad784242c44fe1f06bbefd203b5e2c5d1fd02fc259a3a3220af09ffd136b64e014c88385cddaae287abe9dc42d6891df20ae748a6219f9267e3bdfe6340fa5014085897&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263025934Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d, Name: checkpoints/2880/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2880/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0d60cb41a3878377cd4ba90fb6cd74394cb000f8e24287670cfb4d5bfd8346f4a2fab1ccd9e577662c1da03ef54525f1934d304334fd98953968edd1c23d478149c02ea8182c154e806ae3136fc8146d0f4306e75aa676b0b8a3583aae1b3d084178921dbff627238a36c5eacd610e9a78cce9f8b15ad067628cde4c713f4e859f9d5b72c88ae24268ea830261de4e80eeb1fc39a72bb6d5302e58cb2a10e87c888235055c08ed0161373c5f9b795b2e776df2ac20dc58bf2f26cd68bce9cb9400b24d186f5c110482490757b3b6004bb164227fda6f91007afccb56fa717a140935f027b7ada7ffebc4f4529b9d215d28c4eb8d44d5f1ad34a3840d5f0d0d90&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263049073Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224, Name: checkpoints/4224, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4224?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b8f7530ef87ae4faf61ccbdd9920008b5b8677c947e21790ec42e81e88df114005eb44af489b3a5430f8f733ec7b8feffb830c8b7a5f9a899fecbb7532fc8704952ae3b110727ce0a08f811237fbec0a26cb45bf1e1c8871c10f4a257038f7ca3299fb533fb2f5e83b2c9f34da268c97822e0922ca6559eee2839463b2cf65790a389732fac1b7889470da620f37c51f75c8fba6542a3e450b236bb9243f9c05fd852b0193b47e48af9ac51aad3c963b8269a88701ae990e9e772967834b29060f14c81b2fe120baa8fd6d9118ae859b1ea47f19f9992e37e316d473e16f9e0311afbfff0737a78fd1b17f1b17a2e7ce0c2cec5aec035b8c5091731ad21ca201&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263066941Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/array_metadatas, Name: checkpoints/4672/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4672/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a2db75ada3f33505396dc35ff427c1f5c835fe56e58f0d5f6150c91bad734c0664c9cae681fd157c4db520d4368a7192e507005823e18102c2c8e157a29763da6c0cee04f548002389c25f38a21c06a535203fe3119d70dd7b4d752fd23fc32e0456f70ba8ba1a1b8f66e764dbdd566bebadf8ae03cec872d510da11c05888d18462ebf906ed1c25b621da18306a09b5963c64ae419b77f5752d5671e8c3fee0db3dd31308ce797b67cd7aa19acc706287c989a24b53cab116ef48cf320668e9e539d20524b63e31180c0213df37bbfc841f9a4aed69d5ee4885024101cb5cb9181ac9b5123c527bff2b92e80174b7ba8ce677cb7cd49a6cc67cd0a717044258&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263108838Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/array_metadatas, Name: checkpoints/384/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/384/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=70b406f72d1257ba03ba053f811b2f977606d5962a7814c3e9e6779a1232d1c8c6123c79355e0fcd271d62fcb4ffabb004237b842a7b1f8e9ee50505827418c2b8c5a25dba3b45bc59b53f2cb3d3cd7b8b0f67491159522c58378b7d66ae8d57257cd59b9dbee835145502268a40d4f3d4f84b2e7171ccc910294c977fe3c5a0f80ce800aaf59095b593831becc716a288714bb006862bc8f15cc8540d7eb883aff25c7578ac49f2cbb8abab6f1a1a51a95e5b3cc967af223d2dc6c845b64dbcb81a45c02f4b62d0c153cf0bc47b4cb42b5c8ef2cdf790289ce9d97359aacaf4b8dcf13f11d6c5ed9bb143a130c3d2f2461ba9f4a91434a9f7c99d9c2b086b9e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263144365Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0, Name: checkpoints/2624/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2624/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7f7c7ea4eaa6a997d2598c0b7ed1047053b35f827b4e554a746be712c75e511a26ff3cc480a89914ddc8b62d805edc7de1b65f0482209b833f818b3298472498c5d1a5333ee2b4c4e462d2561d69653272493ea74c8454449198b7b5e64f1fa26c985f9426a2ae99f8991f362081c66ec51e4b4b865ec91a9f53a6d35eb99ccb3dd1d46d08ef97a2984cfeb4f2187489955141c1d75689647420e95b37622e851413f22b9839920c2d3b6a5bfdaf282eaa50ca193ad0c97c8b66e7c13de36106b49d8611217cf3449d2e7219764ed0fcd8956feeffff1f965320412a2ee63570e8c641eb97d1e7de4bb61bb7251bae2bd17d0fb1db2d8823ca5004c83117f384&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263164804Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640, Name: checkpoints/640, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/640?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=413a64c23345e874fdb78a8bcf878d616c25d42d5d2ac070bc481c8fab6be103767f7e8b3c72e28c834d2e548c8f83d524fbcea9ffe580c7936cd40fc02e9ecb6ebfb5b9ffd725a50c69e43d267f4d7808bdd16247c4187b0b435edc494f8e48821ca7700d871c3c14206307414eadc76612a210644316ebf3a9745eafe5a749564a343b33e21bcf23ef1ed85ce7f357cdd99384ef3518acbffe300816b6d5c135451f326fc53c19706c8c47c02e951cb8536c797a3a2d811a74dc4ad7d667c485c609607fb490f695d5c576870314f20e72f644a3959933b524dbe22d72259bb2803108920c0948bb0beec8f0e2d65b35ec4a8ab8862796dc72039b0640cebe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263219037Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default, Name: checkpoints/1600/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1600/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=437c69df83f5581b62ae20a552b8c376e14c5304453037a35487e5d35f58a4d0aece2acbac2eb1154e23cb03a82a5f15a7667393ab951f62660216f53a809838aac1c3695ddcd540d84e43291a886c7002e2b1c9567281377e3979c7c0e1a328d3530e215cdb81d37db85ba67a692eeaa1a67c02363ad55e6564049ee4f2845a3f1ab73d7016a582a12ed79876a64a01de561ed44e21d04fe787bc68e2d3ac00d4e68b3067194189890fc062c32ac800b0ccf4c081d84d4edcb977b792e9b1e144f1f638f7a963eb669b82a2b4e8481620231e05644fa464f01d75725450d23a187ebe0d091b659cfbc5be6f3333a5666fa349a8119d2cde6ab90f9b698f19e2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263271509Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0, Name: checkpoints/192/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/192/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=330f233f5804a3f1f69dc7507117040d01b10455bed76d052c691e8d27c8460a4f090b86455fa3e9a4c256a3cc7bf5b53ac1fb04679ed433f3bddee1e214a6cb8ea88a4e62f6ed0d67a06c713753c5961f3cf57d133b3024c684154910da0fe08148666c3d136859d75b3dcfe8fd568ecd1903d37b35833c25b7724551635a671a2980c239068bc8cff17381d3f361f0f101a11347420f14d84592b18b5f5b4efb2efcb06939f57ea8923fc2a9f5479005b9430c252ede985832770c41d53c1ffaebda0fe7abb01b05a39b534a89120c7ba225eb46684454594146048418d61dbbea3e520c3ac0e647ae8167741880a2603103688497eccc419e9411766ea7fa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263348993Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/d, Name: checkpoints/6016/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6016/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2dddb7560fa5d4169ef788683afe43cdf8df1a3f3c49f474e474a0b6561b3b718ea7e4a79fbce88fecac074cb22d1ecfc6e6ad650ddff5ceb09057148c3c0c4472350342e9349ab69ac9e6283474b410cde5c5d69ab90c71dbe8698442fcf7f985068bf6507a4341c0d82235be344ebca0bb41b98e77de3d97fb0b0bb2540395d8af61ec8816003552c67424f771089cbdf340635315c80d60f974937c5452199d31b000a37a1aad80c8ccd441792cf974950059f0d62b4bbd58a8ac5328261d58b535be869606b32400ae5c74f251e8f489ae9ee6c81e91581e341fbab809febd5a0373c2075cfd345fab2a0582ec367b6043d263f5d062e4314689e00b2b4b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263232879Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default, Name: checkpoints/1984/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1984/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4bdc361490792f603c3fc6b636d9fdd233811e76dc85bf7e45c0361659e44f77496d81e7b2451cd3cbd8183b1c5c5e3c54eb5e58155543f74be6af85304fac9b396f8ad997c5b48134302107235d8d0c1c841bab331c41628d72e5e158733acf01e36919f87a5f9ee0a300855d4c5c7cf5cf59681c6d52ca006f61e664102fb19f58d8987d7231f44bd913b46e6a590d299268e9f5de554a46b74c28c1a26ddac123cbc67109ac874bb24f33ad4cf558b25c78dbcbc2df2f25389e70b3e8617bcfd34e38f942be8a0686c20d73ae844f7906c8f261acc0614c8e829e709abd9f3a4372718d7914052da0b9c3d3d8e41af43750a00aa117a7c495b062da5b6f8a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.26298745Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default, Name: checkpoints/5312/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5312/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=403a77b68396130a3e7a0a2155f7d06de5385d1e79bfefe0fdd50ce7ddf64f478c08f1e562c356850629a3c514c8cbc16eb2d85247057770bce9511f07198aff6a80717c3751061959c7db1f882d1159eeddd5af65bf313e3cf0d6a46c713172f75ab56b1dbb2828af0c9e20981916dcc1928b65e1cebd4f2568a6fde3fb48aa0b685112f5ac9690d472a8487d4dc92b70d9cbfbe742486bcabb2bd0935898538c75bf497b82b9f012fae6ddf06c62c37ee13ef8de88be5d7623e17f88511aa7d640884374cc6f192ed45a76412b6159bbb9c619c3db8aaa02c407522b81c2aa57199ef8601f18b8dbb9b8138905051149c098680868dc965c90b47696ccbb08&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263253495Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/d, Name: checkpoints/2304/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2304/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a5006bdcd2da0add1c596fd2676bddbf6ff60c645c8a89d02a9744641026c725e89ca310af5a80508dff265491305c872a5f755ca33608b5be13bcdf7f09c8cf274724bcfa40eefbd994f63a795a7a96959cbfdc347044c53caba5470770eaeb525aed5f7746126c4fbe2faefced6614a8c3a999dd0bc1e357e881346605c1da408ff4165853db3e40119176c95b05bbce7ff57ce89336ab6ae0027b8645baf8023bafe4cce52a249de74bd2b0d511081095fb6d8de368f288e37b7314eefc55e24fd86f731f6738f6e311d7c3ab2267dbbbd8d201491fc540fc2585988cd6fe3aae244c92c3b078b4f1594508525892547a6d7ca2d519f39189cbee35c1d434&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263019769Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default, Name: checkpoints/5120/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5120/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b94ec295aa5279fc71f9aab943df38e344b94abcf9f8f3babb689399fa4c7d7d3db8872d9e3a9bf1d7c0cb8e01f1c710fa8e7645647c12390fb8710eed5a584f2a49aca9b1abe0577c86200670763600c42ec493f740ff134040abcb800724f5d17b7f7169ea2c04b2a55f1897a0315c9e6a2cf6467d05dfe9c12af6262f1dabac8edcf5473a6bc7931976d6146b9959b6e7f34f94f24046e88a57baa587de144a56ad978ede59e9c0258559f4f9c3da7c1d054c12444732392efae747059aa0b39517fd017b949033692aec9d965754e296df391cdc29ec8db02cdd497f0cd98733e2caee4ef996a5aeba2ec3ddcab743454361793ff6ea180a62bacfcdf7fb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263525591Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0, Name: checkpoints/2048/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2048/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2bf9bd04beee3deef9eb3c0ccc0a0961736981c1e51c2a08fa81f5aaf606201cdd86dac27358f1a9c9bda7e281253550b472cc2d8cd89707196cb93646369e49fc26db66c847a5d5373ca5a7789cb3fa367acea3eb449c55592b7ce8f54ba3b7de2f99ad9a3eef86e40e9370066fdeacea79ca877e2337d871d93e387c70faf6659107e1d9f70af9013165e326598841c2c3610822110249a8d7da180bbe1e2dd3abea95682280467e467141f7673c1e6aaeb26afc05f6a71962fa83246af6dacaa809d1ca8593c776840bb79de2287c80b017f8177e350d676c10a2bda5b794338ac8c3dba83c73fa9ddf1df51b418784bb106472cfcd9eac14cfe518dd27a7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264242626Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0, Name: checkpoints/4480/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4480/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=32e9785417d25b540fafcc4f795cb1d9fae3ba3119693ae3a9e2b27d4247bed9c5d5260d8cdd536ca192e3ee4933eaa7556655657bb1ef20e311377cab1c8d52079b49502c765766c1a09fe48bf719f57afd3def2573528cb065cd44a37d851ff64759f8b1c79e99bf2bb25f7ce5e234d663d9167ae0e5265164266f90a75d069caa88968b190f7c233b4cdfaa844d4ed19e0c0bc01071caf73b0166958a6700ca90e307519ec7c999bc814e06b52674eea27349e51a08b340bb43412798d1ea6568472aefcc5afda8df04f7f9f838c721df2b1be82be7634b04dc7431ea123cdacb74607558dc41674e4a706b72a487c161f9ca8f536d1d8ab74db9621dc09a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264031477Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default, Name: checkpoints/5504/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5504/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a7f5835fa83b55cdf2ecb37716f0655047463c40ab87789d8ea5d8da732ee7bbc238340330c5774ff9a942507da978eb9d02aec54e02e043066b75ea7eb26ecca884038bc7a313c946b2aaba4688ce6740dfd7933b54c0e227d54382b33a1e8c02535e047fe7099f1f1dd635803a5324c881350a96ca2571bda99b722d1ab376d283892efa556e1aaaddb28b9b3eeabaf420717f7b640031ece1fe9d1a39e84a4dc4e95fdd514c25210ea379042a1cd943c415f3fd96b94eaf5840596ee00f60c99d9e6685f8d713b521892bb5ba86b746a880c7437a0d92f4167557c71b9d325c277c91fb592a9361a12034acf5a1c53b99be50c4dbccdec90bf346e11c709c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263104433Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/array_metadatas, Name: checkpoints/5120/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5120/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4225a990863d4d9ff203364cee1b90e6ece2661fc5c44e40a43f32501f5ff7e1c83fb80f5b789f31aeb2f84ff65ad44a390fed704c168184ed020da3748079e1675a4078f811db1a70ad083ad024a8c3b612c7e67c145106f872840c8d82c8a284132855de9c29debc543a8911247ef97a1eb7d0386cb62bd7221bf4f606e980efce2313d37304b4fb56258fc45de4d5f92c22150789aeb52525b0604d1375598cde14721c6671d6cbd11642576b2635a57ca29548a0d58468fe706f4649050f31af30ec1f6262c61ac1d6b0ef91e1e110ad42b39707ae340ab3c0ed425ef909c95b132c40ad68cf8c636708f0dc278b98599a2d78c9e61ac52245fea6eb7efc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.262942267Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d, Name: checkpoints/5632/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5632/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7536ee84f76023fbfe12885f64b1eca45fcbde0584c65f8bd463552ebf75b92f9a290a25eee45e26004233096e57a9505086a4355eccffc29f5561d0e4b669c9543897a991cb1d27e101a97de4db86b61f836eb10596810ba0fa0c85e095c71cfc46f128ddd851a41ab150406131515b783f5a38bf13893892109e3f9e6ec1697adb3e925b0a055b67791af3003b933c3d859db2dd2e0da557a184bb3500b9dbac37b9cd305f33600cc73cd9226777bcc95275e57bffdb75faf49308b9ea1e9e2859b84b0f5c7426d623606affc8ca0d7f21f25085c5f0f5a4d4cd6d500f3aeca0861a38163c2e000bb8d2dbe45a3fe32eeded07ea7754f3acbe0eb91a15e9a4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.262880144Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/array_metadatas, Name: checkpoints/4224/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4224/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=52943b33c4b9bba292632c10a7cc56707827a61308249b3201ead44aa1befdce63bef882335ebcbe86114e1838657aa383aeb52a436c6e249ea58cb4cc0f16ba5d1ec7c2f0c86e1d45d02c6843968475a82e7df07151a45e3a61cbd5880767f3ff062774fa4f28476b12d5c95d60cef6026ec33c63e5d34f047c30181182b4ef42d35f8479f02af5598ce01459737f99a953a07d30f78157867d059b0d002be19c6a0a7172adcb5a78a7f37e72527d170c846cb36d584e84a077520913d98451a90d4c4ef82444e6cbddb672ce34fc3a9d44291e9acbd64ea2741dacc0a8a3984a223f35a6f21fb8231447857266eeaf95ec1ec05af348aacbab833acdb37c25&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263527655Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/array_metadatas, Name: checkpoints/64/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/64/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3d761e76b5d584fb27b36fe87bc0a5f3ab1d232b878017d08123fcb6ddd0e1380b0b54ff3fe3ae42ee6de547dcadc6081751833e840d0aa64a2086457955ecc2c378988cadcbca152dd801e818ba31e929f416a54fcab485af8a302634677b380c390e7ea8d91de948813067dfb2e2a76c92261f0190e518b9be7db15e81248673447189c6cab563478e62d35d876e1b8e9e85960a67c0b632e318887801defce6546837fcb4890226e2fc1f698614406d452d9696313d03dd9b21864ee52dc09d4201c98f11b060cd998563da1e67c408a40369a70e1b8c5ed5a45d3f1f39f10d49c631f80a6fec158027fe48d44f7db2ca43bd1a47fc58c69902ce7e852c6d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264399372Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/d, Name: checkpoints/1728/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1728/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=162555472d845311893529c31c80b7c6d837552275251b94be44956d29fdb123f45ebb87c9fab01852d2b5b91d4bc79f3bfad1b50335396370ce12012786344df15a09fc7678cc8d810153a2c412fbc0b9627f90840de9538aa821476c30b414799170ff34c0eb8875cf765dc9c73a4dbc7d649a367bb2aab32169eeb282d896e8b858d76ed48e555177763a53409ca195e5f0880c4b36b5914cdaeaee34fcc66fcbbe48f5224dd68d268b052910ff8194807d20817653cdad4dfd706850e8472569f87e35a7a8a31117b9ff017c5464a5e52e5c06748622012c4ab13e771bfbab0605fb9e807e190f04257cc168c0bc84f433d9a2a4bd2ceb6c5d0dcc7320bb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.26295168Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928, Name: checkpoints/4928, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4928?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=261a18b071c73a60f6471b233ddd5970a2e72d40d286b9a72e3a07698003a36329756474da5be488cd2ea91ae517676be875ebd6e6d4cc8e1203e2523b314bd345e73944ea53156f8d7f804220f5b8ebb12570433895b8d354a246975ead219466f7f5ee8a5e820939b65a06b537c0bf8c974e8e27c2d88deab2c2d840d68f6c8c9e96ae847b3f12b9a604956d4214c0e20fbf1ff9db29caec3438ebbb3b040ecaed7bf7aa93e1f6f087a837a9d7cb717fcb2dae054051f39bc45ce5ad5ffe50fd1170b90ea8bbdc505244fc6e8c0096e47285fc3274dac8adb3c7bf7f89b339563e9933fb263d211988e4314c172d51282f4d93c897af845ff557505f5084a4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.262965575Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d, Name: checkpoints/448/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/448/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=26a5de71a18f14e55c33bd8175739b0133745f965e667d027fb16ec6ec47fae605d6e7fc7ed895ba67b1c7967dd6e7738d93cb6f134d264aff21eaeb313b48fb272542cbba215758044cdc2503f687e28609662fc368d62f740109930159afa99dde4838d7e0868a190012bd98880b9e311385d3fb33a41c25700daac1e35c274d882bdff575f478d0ae88be99309a5b2cbeff5cdab0d14f7a84bfbf1e4b41b7b21c2ab01c071bd80330d6791f8833fab689797f23c01fa8e936cb89deb87a2110fd26f93a2d337407dd601a701b5ed8b1ea98b935d56db64b2827786a9571340da13d5998a2e2e6fb86c97d56933ef3f96546784ace24a9d91b43f2dc9fb8fa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.262996598Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d, Name: checkpoints/5120/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5120/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=105b5c414aa7530a679354ef6a10b80f29978489f31ec453740988ad791850da9e0f131e720ed22faa9befa04923304b80b278dde4eb87e847e9e22f69e82aa7402311a68856035e06c4ec0e3cc85419cb8919c3550b65a7276c1ef2d28d734f44f0bf756cbf126501614583c24b03f03efc7b6d2097015ea2f8d80be9326835e7dcc8b9560234b3d02f6b4c284417a4a63e5b392ece9a92db67c19dc3c70d8d32058e53d8d9a6a7e10bb2002ff451fd741a622da8a4cefee110c329a49a36c3c0fb78a6866e830bef3439bbc7f3ba17b6bf9e67daa11ca922f9827169cf8604f9a537cb96f69f0292e86061528775673392090b5d4e2965a4abfd4360ebbca0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263110885Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/array_metadatas, Name: checkpoints/2880/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2880/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=07a62995ae81bfccaec5795379f94849356b6b484976eb8ccda2e6993900b9d9fa83fe57c6af9937d71681f98bd5d52f2ef039495322d889fc9576932e3f98340664f889a014a5ab1d96b187f5a1ad5678a2e7a4b4ce8034b56a3c403c3e8ce94edef0fa5b4be1248a958d9d5ca504230b08e94f078bd8687dd3cc8583c84875aaeeb0bfc77f1bdcff22854cc08430d98d29bcf8e1121ef419f4b49497322198e5533e38afdec59f6f8fdcd47ec3556da6a25fe7b1fd93e74369ce7d5f2ef4d1d0919ec6db7166e3b47f9b44611bcd6da3a2499ee0a41e157daf53c7477f1bb829f89b8f06cf774dd13407e97d1878a2f370a32cacc5870527c9c82e84876fbf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263121072Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default, Name: checkpoints/832/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/832/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8746c02ccfd9075413c13f46183a4140334b3e72a874a5a2cc40b3abdf19cea0677a527d0cb50cda7105fde146e1c557b40e823f92c3a4838b81466d8a16a96471a9b2132ea639bb8fd99c5ffa202295fc81dfebd61ef17183ea03f9c413b2aeff98194449f8a23b24cb4f0eec0438a94b52cdf49764612438e9b7e37fc6757a9a970a6fc9c913ecebab0cc777e37d1b6676bb64dea458ce9e169b826a61e60d8c761373fd6c57263c5927f1ee5c2404de31cf6cc22559abdcde358b52869e4b48b6b8f9086c494440df1097e3dfd6b70b77d1dc2ec9ac65665ab0300e0d9942e7934eb39dc0de8baed9cd722f8cb5408ca058e04b25c7aefed5545d8f5fcdaf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263203413Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/array_metadatas, Name: checkpoints/4800/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4800/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5cb560943c4dc130db38d1bb9ffff9b264c57fddea53b7df92451ed478155bc2a84f009804b5f6500ed7f4a846729e43a3da3f73b8793caa84923ecac30a389461bf5743be2997120eec565a6ace4bffeadf6d73bff2f95a4a595819cf6c022ae4b5a3441f9173ca02130fda06eb5ea11c0ba6264a27cbcf2dbd2e678931b83c41166c371c737f4019a7ee3515f750cd383b8c5c0e770e60b17c1fbe5df848d440742f65c1293e4c1779a34b91fbcdd453a0524cb5729277d9d2f305a6c582a9194ece30fd77b6b13d67e7850d9cfd580d8777b36325b654976bca2f616e0677685b2f1301620566814b3750a0d0d82aba0c8e73acabbc94bd4ce92189291bc1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263283542Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0, Name: checkpoints/1472/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1472/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=05df0e7d5d500020785d7b66a7b84dcde8523adb54c8658432a03c6a81801ec97456a88743730a686314533e97d43a13bb2b1f2b98d732f0c9d29ff45b36801d2863462b7c6fd8ec48db23e117fbdd63fe64844cafdc7620b51c96057ff7e27361a0cae06f73f9fdcfe8c9f46482c09fec78e6a4cf5ca5a3fe32d63b5001b110152d2a1c7bcf9252844e6f6c41e9ee5b80ad00664a82b01153eaf63679ca7e30e6a1f799b24e3ac50f3c81436c3974fa05c962d27e77ba3f0ef77f8ebf81f2d5395318cad28e53177761d27ad10775f1954f177702199b06c8d796d115889d87377e77d7c9183738d63050c9af7d7fc70be6eec132d6e9dd6bed452ea587ff97&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263272058Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/array_metadatas, Name: checkpoints/3136/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3136/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=01f18f11c4e7f4a0babdb0cd87305306720ca9cacefa8579b78c82d368e125db21b7a115c8bc96eee6132f2fbcdd68da68e7511c450fae0e104e5da75cb1f77f54b6ceefc650a24e30f1e27173d15abf1557632b6e74eab8f1663f8e970dd7d0a50b0b0e0aa7e8af8081718d367e5c020ee4b9d5785e8af5cb4e9fd91979385e58c729cd66a0b6f8b3a16414656056f82aad9608068febd6f229e72c8d522ea302959b71ae8e36fe540c73fce65df734d154ef031c4ca9b53aa5db8c7795a003d07ea334ccc9288a895110df1238c7ad2e789c16af644be84a84bf79ea60bf6cb3d3a1943dccdf3f9e97f6ac5ed604d7bc46d539c6f6031ab5e3089f9627c5e7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263285118Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d, Name: checkpoints/3968/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3968/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=aff02b488190ff36df37a27cf794c5494d8789e7017b202dfb6ecb8a5c9aec7a96cfbf19329959d1e7aa5a69a3de3e8a07548a945975572d7394adf47ea53aaa914c4b336301537c302b3bb0dce8b637d3d406c42058d0ff1be3bca503897acde96c8722a3d850357126b952f55ec30293379a8f5305a31b6c6a8189e47bce036e8266bad1b2942612b31b6d38365965e8831e4d1e3446ca94c07e406226801216a17318e54d02772e39378d0bd7d355733dcf7cbe6e47c8d9fa1ad14a9a9f7a62d1d5f02424ffc58987c5b1e55ec30d1923dd4cd88ef47e401f137257baf93f5fc592131aa8a6478cc3b674c65ef2c38ad7cbcfd736cd852c09e26e618b7bb7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263368175Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0, Name: checkpoints/4416/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4416/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=bd456fc3e6c9d2bd7d60eb889753598400b1f65125a1772c8bd2e03b930100d0ebb86b9141a6d0efabbe11126bebc30ffee4ce7d0fdcec5781ac52f48ec14bc6500f14cd1a737728fd2ae57ec35dd09e9635465654123d30687a474add8bc1a868024e64fe49a8137e759073108c50f31e978bcbebccdd40b0145648c64fe739bbf48886ec8c85ba6b0aee7e4911ab102547c1873e867356a375a3ad6d81b4c7b4d06305e498599bf87644286c407e669c6496abd84a5cf5369e9f102f9ca129dc66d1999bc4ddfe01920f35060de9aa66359152b3d9a97018a9a95be6a0733822ef1647dae548df7f357d891070b84b98263817a1c0ad101651672b17daabd4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263386798Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d, Name: checkpoints/5824/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5824/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4e67ba7c8ac8a53fbfd5326e9ebab571a2ae69ed35793da0064fece7d6f026e2438e2224a0d2aa6d5e7d6f96d7a064c3093ef9ab13ab2b0e426f33476e3c7dd71d2771c73d17eaa8036043ecfa62162e59323ee14f4db171adb15236563519d4f9f3aeabedf538d1f83d4a780a02ac02a26f818703dff87518062782cd0f07382342be60e2889c60ec5c030e337067924ec3087be954482a9d3d8c47e8d4322e7fc4ab37825829be99d9e9219af09d854287581598319aa5244ad7791ee9b5340143b29955aadc5b0ea7cfd773bb40e91fbcd2988615f3cbace137a0353a8c3310474b8cf4181150b926382c68550a2c9b48eace4a941579a80036fe8c4b4313&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264359699Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d, Name: checkpoints/4032/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4032/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6f7c34ee0d386e3b780af2564b2dd09f267b580f5ff64a2b477fc027cef70df60ebe0a8bc23e4aa53980fe2fcb5acdbb25a3791d1b84fc905f7aa812a14a865216bb96e127a99024f287a510d069b24722b016da16ab847f976cb138991a27417615d6e30e48fa738d36ee6cfb1c31d6d7a4d8101f738bce26a19c8f2fd85e0a7ab7f8d2536a54ebf0c5cbcf55449741862bfbd521bc75f544b6e579bf5685b121c7fe169ce0f38b1334616c56f001782bf188bc4411c45e7b00eb0d4e54c7bb712514fe212f75b3a56dafc90bdc7adc1472f5ba7d447d9a9718c7cdb85bda56918ab7bda23e6536b47e08108753a2fe7ebc66d4f99ef1c0ddb568d4ab980ec4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263202164Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/d, Name: checkpoints/3712/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3712/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=82be31787381b4831010bb271785bed785b119f937981077b20c7bb64e56ec09df784ad5f04cc652547e9d7ec7b947f6b0a595dae14ebd8c2b7a6362e6a6f28fb22b60e1fefac911cf01ee86f4c9a472b95e7d4c2fd1a5c13a0526ec987569a9c8874797da84a133c5c1a38a67f47e9c0a95eb081e96a8424eb2e36dd1343dae1cc170cf4053c43aaa62aaa3e355bdf915dabe088fb64ead7cc4a413b22ec97ac353623c4230bf61026907dd9cca436c5d19ede07f97523d31ff36b1278895bd2182938d4f3d71c0b8b32d8f4f96f89b6bc945424fb96d2af7c0ccb026171b5ff0046a343aa55f66e87254a72018f5c767b1174d79df2e391b8c47da243a5665&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264612749Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default, Name: checkpoints/3520/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3520/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7f4341682fec51d723fdb2c43bd2f44617bf623a093fbba0c33f096b19ecc38ba3844a7a1ea35b8ada91780fc566139d22a91dd7ea91db0baadcea2750694b12fb6e571a0dd3c00fc0318b69ffd48a4fa18208c420534f0743739918ec7be8b517cfc12d67419d6b2be006e647a790fc15533f1d2656c4aa8ac963d781a2866cfdc0e30f4af03199f0536f5ce1c77bf5c5fac65e137102e746d034db6d22e30b28e188defdd838ee8c3dd1e7d89ed1880918642b96141b3eda37e61e2fb766b4a2238cd3385034640599821f5429e8ef224fde74e602f8d91e188bbd8587e0c9113f32f34d086e6de9cd49eedc173b99381ff4a85bc78471272a152b8faac919&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264629128Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0, Name: checkpoints/256/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/256/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6435e458066d921218f0057a1f10e6841cdfa23d8af2359812fc9c2ba7a5cda8991b78adaa853b68f828473df20978221da8541a1340b5272c7a7168cf7427082ec1bb8b06335284656ca162c88390215dd3823c782d50aabd6454d3974c1386a13e9c56d1259d7808d73ca7966756f75f2b4928d8da256b604ce78901514badf7b3066577581199397be039bb9647f80f00559cddb17fa93be3c20be84261011b334bbe98afd9c54f53371f036355aaba50f531c0d7edc4fb65f3bb13dae546a65b684bf91315ab10519e3bee56f2b3abc75dbc617ea8bd74795decbd1c8de0e2dc7f9bdf1cb5f94d18076afc0e3d13d07922060ff539f7fb091f84591fa1f0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264677976Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default, Name: checkpoints/128/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/128/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1afd627f998e30a0fcdb00c1b8a304ad6e8968d87f75dff563066189a8f2b65e20a6ec1a08d3e6604b3601ad5c8edaa98689f8451ef9a371d09b38a2ccc0f9d3e6f598e884bb7a815942fe9ccba6a67cad53204e0a4c7172d2140c2b60d6a79f9b38240384cb253bbc949d22b5a70f283d16d9e3a2cbf9993dadb65222a5bea2247f81692db01d4bb06b107cb2eff2736a3fdc086a145a0d6cc2a443bdd0fa39a8a811cdaa3cbd845792acb9e124286de87e62cd12d0b70ee099ef1313ca92a2d1fd679f1befb9aca0a780020fc8f59ee9b90648615a41e0337c166b5ab34dca61b52fb6ba3ded5201fc64a74526c888c4d3967ca75ae217f59e72c4fe7609fb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.26353414Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default, Name: checkpoints/5440/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5440/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=24c7fa6aa51fe7f566c9e0414889180029779ad8b5a0838a3feeda1f40610a0343a3660b969845d9535b86d263144d145b2bd7250de84268859bb4dbc16bef1537a23e28582f009bfba4f413070360532a249f44f806660ff46eec86d0fc2388ebce59441c7ba2e529fc5fbb4fd2a8d64d1afe8725078145b9e34c3c7c63605dba9ebc329f24c59e9187c33a74d5a0908ddbc705bf5442d9ee18c0694b3996c33f3f29d5b7e638c0b66b51139ab1d33171636832a83bd0f12c7f4f39aaaa00cae24e31e01c4968076dbab499becd623534d55e197071d8d364737ea3984fa0d9962baba19d6c63a7142eea8c4758f4343e64556dbbbb5a7819f780b7944f835c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264452635Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/d, Name: checkpoints/4736/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4736/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=844e0ff1e4eb6b4bcccec243bd6943d6cdae2375a760e313040992b79295d0b2fbdd4b29e35f646222164af849b0124dfbd8236ce0d75e59f07d3676e08709eaed18f081a07ebf9d9a0e212527c9a12f0f4cd34bb0b2d2628ddc890e2f15b1c5d99f21db1507d449315bfe44a505961f532f4a943aad0cdd55d29b0ee4c87afd5802fc1609af1a04b24f0a81af98b99a6e50e8d17ecd7dd4a35c28a29c806f5947413a82d56f793ac6a39bf3dd0a1f3db7a307bfda85923d53dff36463f4e4b76ee78e54e3137f79f6ea4ee163ab768fc8bd79767393058a9647328fa32cbb49abcd847ae6e2d263c5981dd1d299938e971c0071779b3756edcc7400d46b46af&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264424338Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d, Name: checkpoints/2688/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2688/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=371e7585a5d9f00d6a85b08235fec1bdffc285d4e2d7079afc69c66709666ca14e9959fcd29eeeed317bd11694042d1a6031beaafdaba385b13674c67a82cc41720b7ce6495dec87a8ff86f543df69e04859261c9718b3fccdecb0406340256c4ee767a2e6a256d5149d45db4fcd8e0657b6d80cc453c1dc26770519ee13b04b03a22f46ee04733b3cc5df8126c0d54c61199c5114fb8b670d26fc02d9f48bdf332238ff39eea352d10281652679b5b4645e8747542bfdefcb2e29fb502fe2aff2f1047901afb75d1313db4e1eb0350cffb1ccc0ad1819bc817fb4d7801ee1fa7f704da520be41b9e87e0efcd234a1e085f78d274143963136d387f3db351c73&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264808293Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0, Name: checkpoints/2240/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2240/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1604bc69985428c742c242e3f972d73379cb5368bb0ab71638d57acb0cd0f2eccf42f9739ab4fe1455b62051d005560d0d85faa51ee13e474bafebe4ed08fbe279c256e0815e200d3a41a9c4b8549a48e5172a33f794d6e3566fe721432b1742f74efbc11bd9a4bdd98fa2bc23568a882d78eef06c730e3dacdbb68ab997b503a964161b728dc82936fd11a8be3179c80d8179ab973f20122b6f3f157deec567354f888cfa7f5f969aa0350f817974e7183d1ccb8acad7f188980b4839185f5c8756682f931512831b085689791192338088c6cb0ab869a5de1c9914e5c872b6241edb4b9f7dd3421a7e80fd4fb1db85d401a4d28f22af36f0ae917f6fb31fae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264815454Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/array_metadatas, Name: checkpoints/2432/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2432/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9e4ac22895d68de091f28953476430b3ce3ecc4c1cff2efa653daeffa25d234dfb0a8a396bcda0e0f89d7f5263e4886af50912b93989d0f124ab532d92968583e1eeebb9ec438a999edee74b79a44d4f8242844a78c545a441e8ee0215c81aeb2f220026e277bac82a91200dc8eb0595bd934ec563bb435b133edfec8d66b0ccaf626aec09dd9ac3d2550d89f6860538a5a5fb89375527d4dd6d4139a0dbf8231f14dacc07d6261f93700156077e6236fb5c2588e0e20cdd4589b5acd0cc2dd605b49139eb57a338f586b95ed3ee134afb5101ae6197c2b5a3ff01693ad73eee5e981be6f2701f9ddec51909abd090f6d8b219cb0d46d50946a4929d2f64743f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.26485181Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0, Name: checkpoints/4032/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4032/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=459737702d77f5bcfcbbf9176ef847ce353121a26cfc690873ce9d672123f56e2b312d84fb356a88c158ed40bfb2f46940a6931e5fe63c578131e984ccf1dce585ca8e2e066bcfe23e5370bb435fd26f9e738cb25810426f29e08a79e3e4c608295d511c444df08b785a33b82926a5bd29dfa697a741c29714b9db36080f9b87315d113dcd419d49adeb78205cb28d07bab9ce598b3465aed39c2d636896e2c2cfdb9bd72a4ae0ed4abca90373dd2bf41da3585443239fc71ac200afdc90e109b4480c0b86156a8831a6f7dbf92d4fdd780a8d69f9ef4603608374af1be8a6b0f8a217b35166eac72bc52af023e8990e553253815bbb96e8fb91661a0ff97cb3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.26488158Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default, Name: checkpoints/5952/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5952/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=84c342b4914e69a72aa393f0a3d15f9b85d326caa1a2d2a8c96ab80e802cca308ab54598087bb78859ac2b72d149ad59348431a8225061174d58ad3bed5e8b64f9f59dc5492b145e4746843b7a7f15b187e59210d2a92b84bbe59687f76e788b5756236f539905b0266effa3435212c328bfbf39daa04d8f0d3c75bd98f9e368237237a73b2f13796c18c94cb8a9c846e8c6f07746abffa5de06a8a66f688b5a9abd0e12472bd93e594c4bb232f4b3c2c46c0e95c427775a08c2152b8db736d9d7774e94da4f7489417eda0a4e78b7e4e01d2ffac5ea61a04ca3454ae00d1ae1f2c87f1bcc20733cdd85c5bd3b12c32a3eb400527be0148a6e8a308a0aa56d24&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264951944Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d, Name: checkpoints/2496/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2496/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=36949477d54c64f296fadc880823ed1cb2c54bb844bf99419af0bad0d99de290004e854d1b2dface7c7929b6a9482427d2895fccba153f3a15a5dc0fb1d37fd279966f7551b19e8f246cb71e4b10d76407a744863921cec7e86d298cb0c537ac1ed4e5be6871b2349d4f01691b5dbb8e0df647dc82aa0b7702c92ad15218ec8848b339ec48f9c8693b90a72f364b18d9da1828f3aaf85da59e7de6c0192306a317e557dc37164103b56ac15fade73f416876f1b24f4e2000464dce650515bac7d94a0b92a0e2851a15cde8425b4c10e95264f7c4d357136e9159bc5e2208cb63d614dd10b2b5fab4fd43324c9f4bd788d26e7f58fb75b16889a52690d01eb9b7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264430305Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0, Name: checkpoints/4288/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4288/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5d54f593b79019d104723613fa3ebb3b2bb17a60b375b63c3187903da6d12a04fadbbb729f305b0782cfa07a7f841e8b220de1470a374c15d83c633f82a6881acf5783fdb3b2fa1c398eb567fd2a223e8eb70489d5e696dcc622520c683440637af830945d23ca15faa165194a43628a8e95816d2a0d1d69a7aa17e7d1c771d7bd75444037b629ebaf8b0b9d5c6fd41adc91001638ee7c582c81be54c74bc06b01a6b7f6d9b781c7d34b5501fe5a141cefc809898adac0a2291635f5a1cd8e3693a12ea8f0ee09f6e4fda7ba36dc79525934ad8fd924180da32713ce42d06cb00df054034f8621a2197aaf1ca4f6dd489561b99b6ba22b8b13a4f2b512494773&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264912019Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0, Name: checkpoints/5120/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5120/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=42105ede82e926108fd60abfb7e991417f6881bc669a7acbf0623f1f949e9acfdc2a9d1df8e9fd6ef821d019eea90f0e9fe845ebee0d1467b26f31f6fe67c4433d7b2611a52fc679afe90ff2262fa4e69ec1e0e81f9e84b6a8926e8bfc8529b9ce430ac0885732b88bcb10d2f777f897c9c7b1d24796ae3b4484ec4d2da33d973bbe0a43111363ca6ed695fe2d3228f50bcfd7fa698a674aacdbb58e3279109ea20d5648d6db7f27061ab1b76d64df75fd5ce5de900b51031f79694e0de16c72116ceeb18dc4e364ef87aef52e7ffbd701389aaeaec85d8c2e0c5b2ab1d6fc32840bd9dfd9dbf6a70c25b46df6ffdafb57e15c2f982ac65e35fc12498e652de6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.2649816Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0, Name: checkpoints/4672/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4672/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=96e4d58cb1dd7072a848a7b06acad1a37c6cc09318cb180cc2b7e1e14c539fd5e2b9726d5b8900253dc2b8fbdbb8418b2f8e81197c1e6689e97a99f9e1fbb6a2367cd1cdc69a96c35e5b4b228e86ef114b7d555efbf13243ca5733cb81f3f0b754618287cddb3c601893295b36384885830d1cd702469de48c3e04b1ac2bce199a368b32bd17581bcc8dfb7a44e576c4510a91582760e9a1248cb8dec41f062f0731943a92d7bc91b9bb1770327efae303091a31abfbd5b941680aff498b1aca2f38beaddd0fdaba05c665a943b265fca61a92706b4f321e8658c00414b37cf570590dac36e725dbc1da160405d44b6d1710e8074f6409fb703d5c520199ed8e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264478827Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default, Name: checkpoints/192/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/192/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1d601c90179bb0c09758314675ad9493e05c31a4fb427f93d56d31c6ba2fa9e426ac30ef0f784f3b22720a4a1e03a47a21c39c969626a16823e75213d55e000a0ee4dbeb5256a254b9f092fe6a8252cac8b2d62b1242454f985d48aa0737de603dd2c0d9c901b078d062b6cb86f4ee52654f1b6662c951b9abf6eac49ddd24e643e5386d45c8e8efa41a90c1e73218e17fede8085b4ce0e0669b24287c42fe427266940560171636169bb7809dd9671afc50f252e73d310c411b854023d238c2592a128f7448127e491627a7453eb68f9e7e9c68dacae578a6d29e9777f4fb90f5efc729f26aa5aac4b5fff9be4d1528f9092d409f4f53d96cbcc99a063098fd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263434826Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default, Name: checkpoints/960/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/960/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1be081c5e5d9f699570fc5781fde6ef35241228401716e533a56d5fccc51e42a5fa6fab9cebec0a40d80d77beb86c94785b723ef6285c5d1d713ac49089bee36f3fce1a1406b3d7280fbc50c9940d7d4822aa22239cef490c40b26d79c4eeebcac866a55b39ba011d5f9ecfd78164b755c5490101710fdc4bf754712cd4e48fb97884f4a49834480c3de623f1aa04bc6e8260cc536154ab7fdaad0285cb71b035e1b8814942bbd3832b9922418c523517eb5f21a3e60d4fdec59ca6171ea1bcac7b1056196aa0789f2f5f340103c5ab2acbff1962da449caf7bde974972d26c063f077df53ad5bae7edac56a0d749dafbbf8275d2a34acfc098ab00f04eb2fde&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.26504276Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/d, Name: checkpoints/4992/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4992/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6ea10f4e050c0021f9d3ae0236d3274da4f4df7d23739e3fc3ab2dbf7c1951d57245cafddf58ed42f36795bd6eba59a157ee543b813c27914857a3a46169ed3be55d0b623675ff8416a91dab6bfc5a96b89492b84690836425d9f1bbf7e927166d3712c469a6be41abc3563bc4c7d98e5d6410e49a2fac3bb771883b9598ccb1d8d762c09af8d5aa5d2ba1d9404a3747cd3a4ea871d427a23e73f78c04b4608c6c8406b61067556ae9267f357aa162e5cd0c29efff7d6e906d3b43a8bc294371cb93201415303f3140698803c1359ec984d855b30435f5936297682ddbda4e33d35dca38d782cd6a948ade2598f8dfd98413645f324204abcff37db690a34c34&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265066086Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/d, Name: checkpoints/3840/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3840/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=073194bc776248b4e42b723442126e21eea5e5d91c981a22269ce2c49adfe709faf38fa84b0a6845e679a0d46c5364d3fda2ddc7689563eeb3e0df52302c9c6737d68f980c25c688fe020d08375f78e43dfe3246986ffab357485b77d1cd89e00c06cb6a33729caa47de71ca9d524e143549c36ac8b8ad55e261fe7fdc920202b19d6b0245c350917a40d89aceccc6f1c76609b20af32e4a5200a281dd2c2970adad2cd41bac2b302cb0d202499cd877fa406b39868edcea0b77ce75942f3be61a30d979e44710197c8f4aab1625437ecdae12cc8b7b574db9c66270b76b2b8ca145a5dd4c57fe202b20f59713d390966d30759326b1f362ab47ca59bfa40446&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265108649Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0, Name: checkpoints/2880/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2880/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=789f0fb2dd96be5823309f2cba55fdf40f77aa5f52e7b5eac7cce82f2922d19803ff2191c77691230be0db06ef6f927a39248eeae69bca83e124e515d3af980af5f3adaea9489e9046cf96f5871d4b4f4afa848358ccc470132ecee5a847f1d91e523f9f249c5523b5f7ef2f50381bcc1959557393fceb09853231889e418603c4c50ad640f643959def2b74eb41783c34d8cfca2d08505ac8ac89d76708d9d07c392002af84c8917304c4d6494390813e1d13c6c360d1593ed80d1b92cceb4b76c237578da15c0729d15892708def42c7159d12eb71b80e436dbfda4e98a8e3eb1cde019c8bbceb28aa77c03b4811a7209c4efadea0f2002cc4ff919c74162f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265048281Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d, Name: checkpoints/2624/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2624/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=64e4a9d4e7b3b83f3e1f178f3b95be1c77aa1f81285e68193f71d97bbf8a6392440220046a1bab434693bb0e6bc4debddaec5ab1918542da7ec4da680275abf07de4bccf161c17bc8b9c60dd14a30225ba324e0b2dfd1e9f57affa4a3f089da290bbeda7576b7564a793ad3cde469f0d9f1a82666b4465773c590f5a76ec7ec9ea8cfb7df7d3d8338e7116c316554417ba6b9b4f4e1ae2f36b1bdb1b0dd68f2633ee7ee5b070370ab0dfaff496415f0e96c0c0d753a6e87e152cf4c71fc51d3621d49c39983382297a7745dab118548d456d48884876a939ef6999a7fa995d44976f04b93508834a905c044dbe82559291a4ec04fc0bf0e97ffe1f0f8e51f8bc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265080432Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default, Name: checkpoints/448/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/448/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=03c5271e68795593141dafab2e078808075ee85a143f7bcdc18a292b9337c0a3a6a4d7a9a9074eba838a775f683f726e26e6f5371132b51603ec54e42c406c258d94067063afe3dba06f41b6591de13ef57868e569e47324b7ac17ddb53666d91e1d19cd6b30f5c2cc0a7282a0376f8015d864de5faf7df8511eb25fd854b06a5da3a0672f8f2852e44b73ed09368bdbb45f8dbdb8b640b08e68741ffcea66d194d3719067ec21f33f8908836f87250977a4fda2c4e7be98acef3ab2c32ade274de69dfab7c161d0281022f9eefa5abd60abcf19a781e2de57596cf85104a802892c9e4fa3b0af357b615d45e62dd0eabda7ac8ac8bb385ac3ccc9cb3c53102b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265162544Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0, Name: checkpoints/2496/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2496/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=67ceab3028e12b1474fe8c3a1d3f35156064351407c2eda21ae64c874e167b04a6878c0b6a7d80dd01e21db0f7d862377024364f2cf868c6302fdeb151f6fe3a7131d8aeb3498310eb97a946848966067b5b28b3d8e40435625dd1134455805b5effb014d330ac32f498878f33c6cb2882f68999fc4a0a9c9195cc17f1ec80434338d96ad3fe3336082d83ad587313b5347572ee8fd0c0a519916b498dface62d7f09b07c7929973c3912f836eb3005ca3b0971f211675135456d2fe40679e45f6709430730ba960a6a7cfb079dc6511ab2ec6d1c03c30692bb13367bffe4c27be0abd7c2fc3be3c07a26f34800ade7d293f362cfd836e0be133dc5fddc10da6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263527067Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/d, Name: checkpoints/896/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/896/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=18ad7607e3c93c4b81ef8660f7c795e306862071ab974bb65bc47c721ad9a653c02e0c53858f94d22228ee7bc0e12709e3c46c5b625e4fbab8ee22a7992919c97239a62adf0476c9a3aa1b846b77ee96374d19a47bbbf08e7c38ad863d6721b9c66554f43eaec1741f883bd2b3b5c243bf95077a8f8509ffffd9ac821f4e142c90ce5c7781f4ce082be16a5efaae0f011b9682f4e8c8df455eda16fdded40f884547ae2eb6daf82284a0213361e669b5fd9129c0a8e7bc747d7ccd0acadc966013bc1ce58ac32526326f69047d88c3ff27943e99f1e56219a4c4162da9fb5a0bf37b7a02ac1ecdd62764447ff6502c25229aabd95869cba3975c107d3c829933&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.26472177Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default, Name: checkpoints/2240/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2240/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=42b5f4b22cc9d50e3eef788b175d7968d827e6b729399810860058c413d5c7a5599223c4e8cfb2c6808bb68c67454c216a0cbc55a974c9495b974cad88f04bebf645dba4d318d96aa2d115fd0be0a2a2b88fe22c9b4f659dd67d292bc78646c0ce046b079f00973e42132ac3c0697dceb159c06898e085879fe2e58c7cbcd9b7672c53b3e7a86185cd8c2e6fc82bc2af52345ee3140c83ac8e888d327217e2380a59652a69ced00428d91d508cd4f7fae073a1b66e36f70ffb9f09a6c7acf32bfd07b8f28450e84a6759ee735dbb7f44bb860e08c3ff33d83aca2c8544fec4bc731a9ed382095f8777ea7a90a52478525c2bca2f82775b1d4e8373f62b369909&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265316808Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default, Name: checkpoints/3840/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3840/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=913692e75b793f0480252fba111772689e6fb4ca7975ef347d50baae6b6709769e41d9f21e6ac4b8ac1d03f3d7331dfd712cef8b8fb072299f601181c669b30e36777fa0bd02446f8d5692fa25f3c1515f4ac9faa995604620cd5b5a1bf01eabf18d1042df8902be8b88b58868c17cf596edd859a6641d59c330076f83d9a376ae1be9566e6e7cdab46ca2b9a3e12386b719398cf8d14c06283919724c7a865af1ea961efde9d94d047069520159045485b6402e6121e0a887733431bf140c36b99f698b8bb0957adf52a3aefbe66005760a1f4339d401cecce739109aa7fd13ca6055a87618c39e1e9a2c499bccb1306daf45b3a9c64a4bada67ddd67ebdb84&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265332779Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/array_metadatas, Name: checkpoints/4288/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4288/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0853cb966d62d9996a69a92818723d53533813fcbffaf6b925fa8340bdb9db9be54b7375d24e25cb92e1a15a4455441457c82624faf92982bb6b641bef29071690fafe57ccf13d75eca8969c0352b483b3bab39b47bde6f426a4a2d797cd2ff259eeed5045b3505ef6240b0073a9a6185db239efa8b8e3e8c4b6ccfcd31133bde8948494f4d4403f66f6e09c5a8bb54d7b40714a6b61d16a87fab820ef66eb116a5bd5b15b6ec816a35c827eda03aca16fb676649e56ba0f02edc141a26478a169860992e12ca1dda82c90521ad3a074ecbc68b1fae0170fcd0c227a1021b36e95df60fe091c31cdee4fc0ef3f6211b188fc0c729d6b514fdd36d8c7fb7aab5c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264792888Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0, Name: checkpoints/1024/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1024/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c20c30a04489b475806c00976d3bb5168f735ac3fb23e3d51d64ff5910fe8235ce85fe6d5c7183aac5bf5dcd0ab74705edc5b6b256acf62d9df68837fd09c12d1f58f8f4cc17fc8dae54fa491a946c1606d2790774d6601e6e92d426cc02b6e9bae83d58349afd43dfaa60fadfdca7c645a1954339946a4205da2f9ea2a89c27b7cc5f8d7f1170b55443d79d91f9155df582af88d56132567529802a46a15342399b47952f06da33169474fe16ea122657cbc72a3ebadbf4e3dea4cc4f4570bde80087276b1654814877e09c15b36778eda766d50e487766c10fb13971646f8d51cfc23fe30f550147f09f8276c9a91a4ad80dd3f717a12b5ac3d2ba95218418&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265452567Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0, Name: checkpoints/4224/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4224/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=013f8340f97b7a7ba009c3d0760a13d36f4206f216fdf59cb340f2c549e84b7fb7d2aee7a3f6d44accfb3f16371ef25fe70f9da9282fe785b458abdbf19190d2e91cfdd4ea595a3d127560414d9fac71dc03ebbc2e006f739a1fe667aec5afdfe9a178b79f9fcb4b773eddd4b69467fe291a1fc0f64332c40e10d3b2556ebb02354bb8993944f8965371e871e7566138c50c7c8d5f1017a22e31928ee859b109d7c7603faec822d28984688e5c347946c298d34bf61257fa7a3be258f3a3bd4b7ec2db438200103ef60250dfe07b95f88f0eb6ce6759cab750fede5952215e42756ade9d50d2e4a298b239ccf4cc9238a9e3c007aa68bc462f02b95c9a0b137b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265468827Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/d, Name: checkpoints/320/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/320/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7b7dc911d4b0cf0d985f2ff108f2b352d0ae553685c6dbc7b4b744546d4950dd8ec383f8951a645c31e7fa4655777ae3b72ac41dd33ba1c5dae1774dee09004933a18b0ddac2726ad691fbf7f057cfc68af44d976db53d8d9948b5e9507d4bc01bb87dec69429a418c35020ded1656204f534a42a6cda322157f9eb69e3d2006659b0053c0464ca47816c0dcd0861498d45c26ca3116d0c1a7e74fe7748e1dc0c5af4dd7d2f5f28adaa10c98957c045083d540e513886121dbebfbc88ff8d61c2b3125c5bf9d5819c8bb194240e339d93714c155827d22bbddee529c9029bdb8b51786d2ea9c25daf53abce52dd4fec7741a6a8b8315da62a20abc5da9451dc6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265505273Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/d, Name: checkpoints/1536/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1536/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=729025fa1e94a0852b6e7a9ed778f9b5067352e7580ebe0e9a8cd62e5dd7491dad03b821a0f1856726849794db96e9dfbbb825edaf011575bdc230249df2de61217f99e093398c1d1aae1680fdbcd907ccdfbedb4e56d91faae41b668ebe020ecb642836689533f72c00231ca20add32612fe3a9f034c241132195175f190e5862d17415f23dbbf8383d1ea31df890d4c11c534c05ab3d44efdbe91c2523584fc57e0233ad3c12322c545cdaf47d2f22d76e03d0cea0e4b5147ab94910de58a75998f7d6beb0c93c5f28807fb8a5660e0c1dbcad1c5fefa772648c0b89f0b400eba40c11976f19a96e552e65883370a85cf468df776d96dd4f6fd1d2d1b7e33a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265542427Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/d, Name: checkpoints/4480/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4480/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1e97007efe7c152d98dc966f4ec36067d42901278189f2a59549b5d275c600701d9fbf84f3312e8fa162a06122ae2a27373af0be7efa69c9800c2dfcab5aa0a06f437a99868cc64a174f96cbabdb9086c10a54db3894db94aa0f9a28cef59f419488639fac1f957c231d31f0a2a9ec8d3022728093a1161f6a5ca7da0dca4027a45e9be3ba294abb8bfcbcb32d473c2ef20b3a8e46ea9a156a84ec282e61437d8e9f174c858363e421ae72c96f101676c014501e09f53678c07508908d5877086ae6d29c13449e556de8ce28182e10e9a7da851343621db25eca2dc4a5f04282f17ff625087ff41d97ba54f19c0c167592dad8b2caf01d9548dce6c9de118488&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264845858Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032, Name: checkpoints/4032, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4032?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c06c467efb1db031b1609ec8bf69195e8c9649ac894b798faa3ba7fe9c7b9e5dc6fa0fd8aaa81574a38201d01dd3d38d2864710c7f1a85047b608a35ed6382c893f6f092302e3a0efdfe443a00673598d75abdf2d9e663f6c2f9c66b10ce5d93f2c7a4d53d050c2095663e91f7e0d746a2ab7680dc5bf2d605e21f591280dc3740a8f11dbdd291e77935ec7fdcf13b973de7fa6b8201e85104fe2bce174cbc7424eaf7de333235639fec983035a742a3a9c6533665a6015e7e0818e6eebc1a7707e2c7e9205250d6c224f9555f6791135a10b47065a21c5d923a6eede04e9d4c82b513ee59683ab6a6017511b8bfeb66bdd2bc0a8c83be5672c28bc5ff469367&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265635858Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64, Name: checkpoints/64, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/64?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=49db45559a062d1757d9e4af9b5e009d9a3f790d431d26573ad7610937cb37d546d7b590dbd9fc88c2e3e86dd6308e59c4ee1a10abe616a8329051351adbef91b9f811872c9d42fe5f93d4d66a67ed85ebd63b3b9718086bd05fd34d681e150967be291edc02834a28b3bfdada94a36bb8ffa3e93b4df79889943a1dc28a50b453fea266b3c9fb23bae7d16022bd833eb4e7f3504ea8ab2046d336736afaa5798d4afc1047562c67af1aebb8eafde3f2390dc9d551660f8be4c33f1e5033c623f7fb4e37cbcf2259ffb81c4d8d64c48d59fd1935caec83cbc500baba007cc2eb5e3995ba784f8bf3483c50c0ed9dccb5e4c915f6dd05861ce60b48d55956d967&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265616929Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/d, Name: checkpoints/3008/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3008/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5a67cd68f0f5afa98e477d291c4a97e192657a6b07e0920038da0d0dc65d4560bb3aa204da351729de4554585f2fd320f5bb2017123fbf14ccfde3e40f75f589b8e77ee9d47642da5e4c9259783de415bb69bf3a99aba4cdaf94221f3bca39430cff47946696afcba04d0483e002e6baaae17dffe449a6a3adc90be5fbd2c75676ba55e1fdca2993f0c520d888c33554c6b035836065fb0775a0336a1edf3498e5a85cf0c5da69b92a6ec67ea69621b401e495f49089de774b9f9a6289daba5a557c8333479f6198ecd12037d7f3684a495239828fce1eedb80654659b1d809e1ec502f4be186e0ed8fee1c65327df0bee72078f4d740d54aaf59a5bb96ae939&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265652106Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536, Name: checkpoints/1536, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1536?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=006f17fdcd5d4a1c80946b7472ab08344bdabd7ec6895435a87642f52a6c1aa9a2414c09a27f52bc6601409783ecf5f0a17c44c7e8cb1dd91e9cdab6bb02fcd35358d788aac17521584685bac00973422cdb02e3cfc5d4d1547dd40472c178645f15122dcfc815ee4086a0740331a72769db740a95cd063e2cf976757ef1171d97890892b932dda83a8c4a1a6b32f48472be574010a98f6c134c3d2b70a49f05617149f1e00e00cc4942d654d1f05d642361076b76461698cce487fe5b8c440e537e066d7b9cf6f3c0b8f1672f7c0cfcf111d2083d4130d0e9f7501455bce8f021e26607dbb615308909eb5376d1d1722e2a24e6d1571e142f81b0934c18d6cd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263364433Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0, Name: checkpoints/512/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/512/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b01376bf4a1a60c1dcc7539f9db959018e0ec6257e4340957e03095e385791592928c3c3c1127408ab72af83fd75818dd650a504515b601731cdec21a5d3245c4fa9eec5797b969ff7ba4924b835e26953923effe2f04930d0f9d3e5789b460a4c626d38dd805b1ee95715881e010516dfed891b3cbbf540494224ee1f33118464e033058f03f617c2a28f28d2b20b82519a107e2c50a90a0db8848d407b7d2aa27a5f7ce1b3d4504ac97d923a7c7ea5ec7c78f5021c3ef4eca84f8cc43646c95d23bdb089e8b50669d83f14737462a3c31b03aee8927e72e04d2ebfbe14d55a65528a54e248449a3436a0545353e7667973500ba1524953e26944c765ce0dd5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263402917Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/d, Name: checkpoints/4160/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4160/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7d57cee07eccc8597b4e24cc1c2e0562e9d0d5bfac85b931c3954d60ea3448dc611523174c5b3123aee2be70236a71d5682f62e009790c978ecb4bdb699dd7f1c4d3dd185b1a765892beebff23e679c3d59a5074d12130a2533df9748f4ac88d54fd3389ab52abe1912b671ba81dc1b902049ef5958a6130b8ce9921566a736c8371f11080735f3cd2ddaaa00aea9d194daf83b56cadf06c415b2e46258fc29b588f7584cda1a1d20404840262038dc97506deacca8e23f840137bccbcee79f8606bd4e52723736a809d2ae652c5d7b04b3e8889387bc4a3828cf1dc76ea0d85cbb57bb64eb7f509db6c1abe7b6bc174f570b1e32be01aee84efe45ca66c143e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264817559Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/array_metadatas, Name: checkpoints/2816/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2816/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=81eea785c2b15221ae26c6956a6a12f9fb8fa3701d71d4301c159171f4241a6bcaa5638548ed8fb6fd9b3cd3aeb20029d923113d3882633190755268eb439e84ee00f115b03d67f80e5c45c69243cd2ffc8091ea853bb5cd2f5a2f53a08f9ca053ed3fdf9da6a587d2378407bfa1cb4393954e4de5cfbfc1e5c9a501544714e6f6796b349553327325abb39be6553ed418d56ea5ea097033934d6806573258f71197f37a929db1704a942434c04c63ac5de301cebcf1210f4dcda4cdb531a7c89069999000c0fa88ce90008ec5876afd8f293bd1ee4d5173c36cbd2ca5f5863e898986086efb1a362d2f706ba3982a357f50bc60999fdf94452776e09b78539c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265723117Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/d, Name: checkpoints/2368/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2368/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=079510b7af2dea2c3b33930ad2bf3710d7515a4b831d0833fee794e2190fcfcf2523c284a5a7400cd0b95723698c01f329b7cf1075a4b333b6312b82be3bfd90bf4ff15b0b89f39d3482ba3433e1133e24734d9d5b7263fc2097870aaed05f0a5d3e12a10013cde4e42e581bebfa448ab4a1451b46f516e69f04263adecca5d6dc5eaf269e9f1e6fc788179a6787647584aaacb856037c17e71ad3ad1d3430843c79402045a9b578ce7f2f041c2cdf5a9e268f5996ce57b8cdfc527c978113f165716d701b38b7ec1d3037054af4d49c36eb061c6e4141bb2602242f5c47edbbb9f53a3804945b582b9f40a72aadfeed17f25810bcba5baa6fa938c6aceac013&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263158408Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/d, Name: checkpoints/3584/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3584/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=67137c5e95e496c8b8984f30ee7daf9df0f9731b7e7f2c05a288a8e2dcae549918dacfa79dd8af33a299cfb8645109dd888ecf6eb9e7e27bbced12f2fa7b786a02016889ce08c5f9af27c5fbe4ced253b55cfae50c77c36aaa154f8dbd4050a5bdba71d7c074b62649df0397b8c12f63cccac78e1ff7d4ae12ceddf1eca19f2b979d184cf5509eea5e9deec64ae2102e95112da5be303b791fc81a67521820c2063bd6de4fd28d32d38c5f2a2a1e881dd1603750ae093bda3a7c792793895d63497af0cc2ecf4beea8ac8cc19463fbda5ed8dd2628b841995923d7896e709ed8a3d5c198f61d6b21fdb54b68c6dc533bce19b97a014120c438c32569b49a9aef&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264959755Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0, Name: checkpoints/6080/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6080/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4b6fd4aab497de33f512e48a39a7dd2ecb406cd717ad796cbc450dbd2ba46c7972ae3686699f22f5a6cba3a3718443792410e078a78b19fc1f42d29d0b7d615bf790540526c49d54c89db568bd0079b880d7957d7f7ce8e4ce69d4fe0c5266b50a53c249c15c74e6c255e4402fd86b1f382edc399493cc2e1a6cf343af688d779df755db776c66ebdc2b1a0e8416a85cbaadf24f6d006ae9bc54bf01356dd1e6989683788a8553d9c5925d067ba039a83e469667608bee0118a36fec5b78437745a4abd7e02e45f659aef72f520c4a1b255582c7ecc5b11cf05d39144db4bc8a98c8a89c2b5a1a278d6cbb74e1bb0ef265b3d90eebe7bb63fc304bbebeecf2d3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265471924Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0, Name: checkpoints/2176/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2176/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8506c68880863562aacaf53a94f4a4c4c15d456c8dab942c75d2e64a0ef71ec3d6cde58980bb6d7416ff46fc05f679472fab668e71c21f428c011e4ddc412e383d2a800427be144eb3681369c5293a979d308159501c7895d56f4c6ff26310e24988cead6abb7f62d662c1c2a55ba7bba3e1f06e44f1cdcec47ffb0d3974053ff04a530369adc8074f676466ae1ef6654abc5affe9dab10ae7b8b9610375e53a12d7a25958f31b472cc9b003ccb354c872b4c82c67492b3e367b41ab0e457f7fb95affb5c182c052e9d36f0753aa84469aea2e5ce66f271e74191f8be7c752c721c541483daffdb0805e1931be71695c25a424687f31fe688526693afbf3d8a2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.262809039Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/d, Name: checkpoints/1088/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1088/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6f3ff8e5842bf723043dd1014f989bd74d6dec7a1690a9f16881c8d7cc160136891c0337d13505381d9f438c0315460e5dc07f0b1f9ce845dbee44643f5c547cc2fb099bf740b1e2f8b07bebf4c9df20898f9adf09dc28b19b5727d8d2a9f57b68223649fafc7e6baf2e55fd4050be43ca82bc3b17387d2dcb13bbb649473e81036e8d1461004747622e21d3fa9faf92c6aaf67580da0dcc2dc28b4f74abbaca40bd9cea48a510efae926944b3f82da9b0f732492c94645863679308d8f63f86de9df5c983716c7624ec20cc465799404f05271d9af68321d4e0b73ec128b6529c1958069130d4a2d26367feb57d3bab8f434b3570b29a6e0075f3bc7de495bd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265135821Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d, Name: checkpoints/6016/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6016/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=331be26adacb9d017ad09cc60ded9d845fc984218100a80faed05088a8de113c320aaa14c6c945ab27b654073fd5e5990d67880ef7b07ea5c2272c1c91849127a9bcbb04960f5c4090b3948341011f56aeed46c6906dc8a31ab616b61e5b467899de40ef0d979d33e712b9d572120b5149346e7933928422476dccfad0a611a9bb738bf4040cb59195edd4a0f8f088014ca9710e8e0d8ac84d6a87511cbbb0fc80de332cf5de70a0cdb8859c6766b5bba3b78759ba75e911e511bc40b68a1c5fdf2c946c9022ddaefe67c2d5bbb6c03d00419431853a468ceb807c0a1b5c2417e626851cf132959350ec5cbe4dfa6ccb35b5f6977680f688820cc531bb68bff3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265371279Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d, Name: checkpoints/5504/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5504/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1f9f903a85db1b21b29fd0dc418acff5a4ef9f4fda8e6ae2ff2d2d35d0a1e535382cd81ea8395fde0e8a461321ab74d692fbcee24596ce07f4929a3137947c1ca888da1ecce3f00e0f6ccc7a308d033ec0d75071aea82667e5d7ebb943788c8ea0b061b88c54883a9190b8c37df57d594d341cd18ca0d03ffd9bbb23137d67a14014de26bdac5b4d20996a5d71bc662162ad6595edf742d5de67f73f12d824230f4ba857ea135ba402fca2b15abba1ff6cf4ec0a1211d5ab2a2a2ff38060ee88fa7c487f85f36b53dca5ff7195ee61c98b4a322f3e7a38352a6a29728a80f14505139b1637e8bbf751a39f135ea0bb3538214ef56abdcc528fde5a2801d26eaa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265195419Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d, Name: checkpoints/1856/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1856/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=ae46f70698bea4a18d69e6883f7ef9d184328d88c05262eb5183643cb211e3109ae54cd1bac8601a02dcb2620c9ba36a7e1404f9f37cf4e366400cb44042d2ac99dae7fe5b5a8f740a4a6c1ffd8d4a20c26d9b9ecbaef5b7b8586c2e9492a5310614be93b3ae0490fa0c78ec21f87405ccc02b6cebfeadee937e746c31bf83b98eceb10894e3126bad86785795028adbdde1d89823b5b37be3ca9b1b88015facb5e99833b1e6bbf578c3e721c545c1264c5a3cf138211dbfb3f2ad6abbd73c77dbddb099c1fae5d5a85c8c3fdea820afb9b5d5fa9c62faeb6081960750a4b1782037b71937b0a364767fd208f5b10b7d62b6902b7b7b23d3d5edefe56671967a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.263446529Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0, Name: checkpoints/3840/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3840/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3a8936201759001f7c5a8ced8574dc5caeb3372c7c91a75b9aad42c788ac4544adfd68543730fdf79f918343155ac741a657ca1408ee184fd056dc7961399f008140648d94c3f5aaf3e24fe59657409e42725d7265fd36b38979992bc65aa6ed51701bb6a4c3211b7dad0feaece834de11785ae1341ef61b6f1c71842530f2b07019a96db4174a6c0e007b90bfe25f0ec36ac1d73b64cd834022f3fe3f45c1eadeda836f0673a51432430f141b2a89bca634f00d23df9d6fb91eb5650176a8c7d34ca3f8637e8d6d7c29b08d0eed71c18ca85377474c916945339d0e1436089693be8ccd60e887722771635a159287acd9b0e057d064fa1a4f505bbab219c52c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.266041756Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/array_metadatas, Name: checkpoints/1920/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1920/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=37471f74e51f169879c5eb824e4b1b2b4523c3e2d2264e302dd013e52c8c971e0a96996769fe9a6268a5d9f4ea8092203485f8331c29f94e124265037abfa3d1564499088e1064a2d4385a227389b6f190ee65829810f51c059d1c96af7cbe58d2d82d07b0452d2b318cbcfe420a7d8acbcf48878f730082a8c10737e47cc70eb6833e67fbc7ad477f16b5ca1813537c83afdf424a981eb57b3dff2e678a0afee275365341e2f8c6c2e5971a2633307e2f2199648034812250e6b7cfea8e8ac0efef9d4f4228a0c33f352054eae7bc3b9c2118a0ea83fc66fef097b32424e47db59a9dd196faa43766c00bd34904b31c76c28edadf58f142add39c9c3dd288b8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265485223Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0, Name: checkpoints/1600/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1600/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c2248996c6e1088e6b94a50af3fe90bf567e3efde0da712ce7f827790978af80403fd3108cf19ced692eb084010886d2a93044c2e00bf74116d549eb9fdf71f3937e287131a330b9fc38ba1e0d39a17f575b71153968540b3ac3c243c57b2ee98e986ea07ec26f60bf06ec6a3ff6428e536a3255fd64a99cec0f0431af2efb9e9df2e70ee08e2a9b1c22a7c2f1a959e94b362a9303f4f0b6d430c839dcbd1036d610876585b3a0125d9129ff41b578ecb4b1617880cb59b704a212154a3833599742b03403a4bc5504ccd842fea17980dc23c423d5074a21cbd997182e378fb304f8e722b6c140c209d340fa1a55196d690fa8cedf1eb4c23fd98c2573c65d6d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.26609021Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0, Name: checkpoints/2304/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2304/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=73caf2c3a144d663c8a3f7ad32c5ea9ee70036016994e2bd72346d3a2c5cc808a40ad5599757a76425c88de22c8f9d75f47496dada14a6c2d527f5aa0b9c9df4ab957cfe3698559edbaa6309e59675b636a56602ccce1183930d28609fab779ae3d3ba03e0227db19f6e72ae8210ce88de9c83ac693abac53b5e2fb6ea5cbaf79ca7170834836ba444a3e6dd3a6718fb16a07d33550c5c812ce8771cb09ae5d4faf393169fdaa758396547289c9330085525f31ebdf4083cff4083547e24bc2bfcbbc709c27d37e6117611a0c8341e9497f094706763cbd59ba39f2ce3ccf06b10d9f2f960789f7e97230a4eb3b8e4773dfc750e02b58aaf757b0397f7a06b18&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265053383Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default, Name: checkpoints/1408/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1408/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1c84d92dbfc594ec6b80898678a472ccc602403da6b4c753c719e0c131d55714e3a19008f03979d9e7d179e434cb4964eff9dde88f5b84b15bec35aae18eb831b2ddda4f645a32c0fc7fc3c39cbe4587c330ba0ccc9cfdea5b3eeddefe1aee57d61e1ad67917f6f74c68f101604c583939479825b168e218672d8ebd2e9e9210e30d21251ef8e8f9ee322450c719f9866e44fd062aa77923b57ac1cf28fca50bced00055e33724f998edba64f43861cd024e0eafe743637b6d9cda90e75e8a8751e4ab44d33feff634a3c988c805bfa1d9337d43d73deb1cc65c729ac1f1e30bd2b4b0f5d2b4b711928350e7565992c913ce39d91d1363eff29cf2c153517ffd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265628607Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/array_metadatas, Name: checkpoints/4160/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4160/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=86bb09d9a35e75527266d171a7f52db95cb449f1a8fde57bfff8df6c5b86ef932c1b84fb0e541748415ad639da93dfa93f0bccea9d2bae4f62231e34991ba0bca92063b23e5e43aee0a9c667a7155ab5bac5d3e534991a039b8a8038a2beb4b5c8aa3b9cdf36ed41bc11d41f104afaea1916d92cc990d910a0a1132f7d8687c3736da7e5c22dfbe96f34775eca36704d35a681172d578f61a91b6c52fa23bea05c092921da186f322bc9437e6b70c41bfa0a505acb0957f3c2f28c7a6672ddef329156b7b2239c2ab0c387d852585918ba0c86da8de87618499e6042567a44337813f545e803093c46bda5ba599d0a38e1a8d1eebe396900f3d99286bb21ca8f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265657824Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d, Name: checkpoints/704/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/704/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8e5d18c7ba679be60faca853b816ec3bcc93fbb4b0a2d3ea0f197111ac15a8ba43aa97df8ede92ad74bbe4045dd563b7b46211ec82447b29190780ef0f887e83887a60bee2a61bc0d18b946cda73fe10dbe91bcf3022dad1dd3add2029760aca25b18a077920df4e81639e84b5fe7e8419efc470f0822dd6265396537a04f2a613c1f613af68b8ed1537322a9311cb4cdb7c4c757593db75f929954822446334ddc10bdf2ccea02ffb06d9c5d6f6dbe9f1a862f92fc410070ebd98b8bec7162534f5205297526568ec4e285b4a8cf7a67bfcdd0191309eb9457dfec32cfcc105bbcf8919992b9b9e34ec44e2f93386e2c2abf190d000221698922c574a444253&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265460492Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/array_metadatas, Name: checkpoints/1472/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1472/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=603d268386a4f0303a9abdd5d656bba3e592335b8a0211b60ca5385f4126740f593ec55f29d616c699f0f08b5b95876e6b217dec384088d84b900694d64ac99128a30f1b1fab6656f752a67c5edc058b406931cd11d42832389c5741df6d0a3370266ed087a09cd48bc5ae14a6313e29344afccf3c943fd615c0011907b952cf1dc74c3f5cfec0d036bce191e4afaa9cc85dc3cf1130aecc7c698bca2e1391f5659fc529aa8545de8d4fb50647dc79cbbdc28131a7613a29a618b2f2dffac860e05412b1d16b7cfc200d0b9cf312b7cd8c117f86dc7ff819d385c6e773cc41d4e5bc01f002441624e5a5bdc8316a39d32c9540326f23fb5f3c83231c2fd153ca&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265670892Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d, Name: checkpoints/5376/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5376/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4066775666ebee4308a3667476fd435e405ad9c64e5672ebbf63fbad99905c265fa37f6cb038192e12b2226a668e5972a4d02b214e69a482bdbbd6c8a2cc02bc853fe602fcb38923f4023227314727e45e1a0e854f2ead83200a0f9b6027206ab6d5b7197496a923e5ced98a9172fb227e13622a8f35db69e6b8e3e6bf9d26cb7fa2f46c8db5346d2b226cb580834360a21d0b653fc146a28b6bfee4715f07e105b61cbd9d8b4c19453720f8c37ab12cb62e6dc3cf20560f87fc7c7c76ebb08165a8b404cd06bc3627d4ace6b95f63c5a02ad1392f242ba84ee00bfbdde76fd53b71ca53c5d6f421b2788579d0e81d710547322c89cbe9262b034356d847ca7d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265468546Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664, Name: checkpoints/1664, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1664?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=67a0c0720ce7c82222f46a002835d59d4b74b86adfddeeeb5e0ece30135ae4615fb62c20d89714b86ac7cb9726b4c4a89bb613de0dc9d8677abcf661ad324ffa0ac12cbf7efd0b908978babae755dfa0e8c87f3daee5f1a87e5bd3e8e266fd0810a7c8e66896cca28e3250dd3cdc4ee4dd0f9b3c615fdee6bf7247ba7b54c0ba1f6b78e94050b243b51ad5cbad8fbe16185a95473554d1045afdcc34473cfc52699b0e2f83522f0069129f29585c0c25013a3fe89541e768108ccab6d7e5236887fc1e2e88c1928d2da496e4017a485788525aca0b3d62890003bd6ec307f9930b8b7e1f90c0d1ac27961d06149525e06870a9c750d86823a2f2988767e9cd74&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.264885697Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/array_metadatas, Name: checkpoints/2624/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2624/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a7bf54c16cd48a48943cce4221576e334f44c649802b941bddd1ef41d447bcdfc43ccc472a20abdbbe3a373c313f4086f15c85371d12343509b18db2b81c5c52ca9d57666ac0945419cfb128ae649a058a35a55ce2c83191b6a50458be98ddc771470ca5122e1f9c4e58c5d8a540dc69e05d4943b1a8ba0eab5cf4e505ce9a4abcb92b9578ba40aceef029833b012a51cc2842e073759727aaae59be03810f526c9398bcc38e50e32c1423366b9727f2e3f6a61644474be4e628a8963dce971e739227ae6a274ca2cde6525137c7a1cd5a339abaceb2b4d89e4246cdf9ab950b79dca6bda848de653eba6d1125cea2e05db5571695e0ee6cc1095cb1df90c350&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265682935Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088, Name: checkpoints/1088, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1088?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=affda7720c181126fe058999277de049c1815d3bd1868eddf292937236adb0f5e1c0e3ffd29eb9feca457f05a521a70e6d85722ee2f0f3aa38fc2baa049635c98f71487dd041b67866c86694b987f12d017d0eb277f294de365c6a3c3a16f60703820cce8c099a85f184ff568fe6a147356fa3c4265f708921510e7eb92f8b0d55343a843a71729582e1578e2c88b24d04f3bd1d529623a4154504eabcb0a33f10579d55fbab93cb1b477ffb95f062b88461334a84db93d56b2565edff7ad79159d6af754132ac03a367fec9bde59e1e9b520ec190b5512190d6a5d9c78866fd526a7102e4868441e552b71a9deaa7178e28a07d2f6ccff068c99e19a7fe2651&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268025861Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768, Name: checkpoints/768, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/768?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8d5a57c99e06f3d31a45d48e9ed25e63701ae61e8887ae80dfa8272f06d2aa7b2b26d5d0f943c4bdafcfa5d612ea6031d1962e6f0d2a4aa453cc57ad1cca8cb32b247d4ae18c825203cce11fd2826fb35b7978c8fb49a66e800954f457341f60a91139829bd2fc359ddf56773e97272fae553228ceaea72c417b825011acfa5c0350ba80ed437f17891bbdbe5c073ac3844be56a7ce4f3ee87b14ae99cb981e139c1d89a7155860da68761845bd0b5e5488caef36b9a5480f13e97f3958d6bd6f50293d359faf44d0ac0319213e02eed9b019308799822fdea02d5aafcfed70f3089d41cd9eb6981f3a500b724f55cdf2c2a1d263258774d6b836a58f11cadf1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268077127Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default, Name: checkpoints/4800/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4800/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5351d8aebd5ec243fae30d365f2c4378da51da06577a270694eae587694f7ba02e3189018e1a6f49488698f58d06de03eb5096714a0a1f89391fb0bc38877ff427f42e8c31e41ef2ee599f8c39652733b72986a4164837bff041e9392e2d3acaa57e25b6df5655d7512da5005b79a55993c194b0b7d1c8eef05b0fd5e002a176a29a2f6ffc56e6830ab524745272120c530ac67a4662efb655db6305c3bef4d18f13ee5630079fba37cccbc61eb41e301e41a9979a88054d1ab2a4b145cfd2b6863c61876d8fbecd1e8fd13c544c75874483417b2433efd53fe8e1cd952212bc9341e327453a577e3af74c02208bbefebacdd0d96c5d73af237ae674a4d15025&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268180586Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264, Name: checkpoints/3264, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3264?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b7af4176cdbf6506369f10b9ea674cfbebbfd28fdfa8e565cf72872c945c8b2b753ff76fbff3cacd2ed9aa69e9f56fe26d0827af63454c864ae851b76b4cbaaef5b86499d1662df0c463ef8eadd12627dbdb472f79d917d390d96212d8f3740c1896ddcf1688d41c2e717ba126133964291e718d3f8828aa93ae5f86200cf7508de17f15a23fa26798248fa4be67bde6d28a728ea7428d66d72124ae04ac09e0173c065c69472115f8451b8a7c2292a04e2353232dd824208ead887a4e5bc2086d7704ad83707cbab56fed2b49b47c12e3aeb539bec8b9166b5a19f8e161399caf5c758243abd7cd85f47e93650b10f64e47b5b04a90958d86bef98a295ed9cc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268192382Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default, Name: checkpoints/2176/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2176/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6766d3b3766938ee5752007a9ce630cc010059549da475e47aa44bb599da49e399d19bcd3d2927ade376a3bc432bc7f3b932577f72c494e273da7c4caf0cf7f231b7b42b27602caf3dff280f09f9706853cf57c666e2fefd83f853d4d37f2095d0b98ca2e5c0f6667290f2c5560266b4a87fc9a7a8ef43eae57110307e98e8c7b8464a343fc8a1ff04118b37f24d5b4ddd921e4062fadacee04916c9827e8aab849854fc94430cb18fc0b3b7ea35375f9c0e5b4f28e0a3e973a710e560b87c714426699bf44f6cc699abc4f8eb39171b2e27286ba4e1b8c649fe59bdeb845c107c0b36efc4b85a9c843f54cb24558a987afbdc398144bab2f94aedbda19d7939&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265890682Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d, Name: checkpoints/4544/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4544/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=37d0141a96f7165a563469dc19c620f8192e217ac5fd7c52d1e671f363a250803702da0f7c45ea54a3ba9e2ed47068c602741db78ec2747b7f07f0adfea8461e630ae0ca9cc953799e24fc4ef4ffabb498f64657b80f80d5a947a4bc158fe373c4cbd8464bdb6783adeea8ade5ff3d9b794ed8a6eafa734fc782352c73f881e0da9a7358184572a4bb39d8b77a9074ac65710bc6de4018308e0a8b693803b886260a1b13bd55d0e869d005440654b97fea93d8325b85f1418d6f1459ec23bb264adab424275179c2fbf20944e51304349327110eb48abe002058821c0091c4b8c0e39950a6022135d5be4da018de987599906d6cdead853f14c4068ef6e82445&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268275348Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624, Name: checkpoints/2624, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2624?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=655b894637e302a59d688251d8b4a23ce4a8c698e62101a59cfe731a1f6d0d9d44f902b7b46e55b38b438463230009c91d6e0e38e1f401e6049d3f2b1419063c87998c2e859fdb12a4926baff8a4eb4d57231c36bb6cedcb17956eaddce69807356c4f2792a766259046cf1570a00d75a1fdb4e89a348f392c4e07649970d83972e533ddb028fa92c455569824df24297ff9e1bf820ec8eb04fa16cd80f3ca7bcd350842c74daad328046bf4b3c46350d515a7411ccca71dcb3171ca3372c4b22d8264fe42a8225bf4893f8f3a755f5c95b93e81006171846304d3936a7b64d1747e0d7648693ac68c5bbb85b6fb913a8afca88693d5d06b7f2da1a1ed932c7b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268332061Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default, Name: checkpoints/640/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/640/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0b5a773b46399db99bb1c91f5b20e2a18a58fd5a8dd2781bf9b8a64c712411bbf68e9787b33db6ee4b6951c6677f5a538138bf589b00cd0e5b6f96b80ccf548ed055a2fe41aed763fea981da5453c836d1025c1f0ef7221dca9f96b1963e5aef31583c69e22df60277c4ab62556c225ef0687be3191f48d2eea793deb1ab16b776884ff2eeb25a47cb1d40643c460a4b5b2fd7024cc3dda9375e4e72509975246ef0197c8ecd2bebae669324218265564c72b2ccb440573114565ecd3278f782ac6ef30b11b422ab9d659ffce18337fb20b73f6992269d20c31bf0167b7107e182a3e3482f3a7fda08a4c4955ab61a63e112c6dc91b15d8268ca49d1b32668fd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268364341Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776, Name: checkpoints/3776, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3776?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7465d9e43951e65b8002c725adb1c3d423bf4eff1a3a23bfc2d3f35bbdd0bd700c4cb6ff12d7d9e0f07ad8c1a8fbe260bbd29b9cdd525df0726477ccf11494c7828ff9f50be1450758df589dc23ef2f8b7dedfde5d4ed92e51dad41a49db5403f234c14a18125410c84022c51e31a53f1d1b008d5d350f4368759355ec989539c21834c447ab4d1052abcea2cde823f355a48bb952d8b530e55c9bf6b0a38761796d171068ab7f1e6fef2e954aef7cdae584b950a1f3222f7945c972528821488b0f8d16b093c29dfe4378380ee87575fdfc94ff2f2ce83a97321ae028af9a251e61153c827a4fba64f2be3f001e478a45f3d89c662ad97a4b83d15d471c75c3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268420567Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/d, Name: checkpoints/5376/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5376/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=81971af18190c9b5d789debc70baf722fcbc992ccc39920845e40dce86c1b179a4c763770c58d3e374843a958f57223258f67d224422f272ddac9bf36c95c662ed3869f76c33beb713d7d08bc9a479bd90a5474999b7cbbc0934583e9726ee216f233e2d90cfd164c8fef2fd340c1a3f63c9c139fe0e98ead69e4a5c592950d24e688e7fcfd79d9555ede2276dc88ee4d8742a7769c6e379f6c898b6a078d8bc280a6be659d622c4cc1ce17dce80af08b616c1ac48defa937195c356432723d4b588b20d52032312e8d744bc62a2806ec7c90f30c9c3459964aefbdf5ed15aac0c0d2106d98a82404a585f1b2091400a4c976b726d8fc9fbf638d89485c51a2f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268491511Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d, Name: checkpoints/5888/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5888/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6606d59dc95824d8b38c1c0b80f22d988af8e0ebafb56ce3843b5e34d7141ecb161ed1b19c45b0c6d399634f71687834600bd877d323ce6aa6bf31d760767cb7d5498c6a78cb89edfa86646b5eaea5f5ded0b3e630e9df8b296287b06e4be5cfd96ac358fbf9ebdfbc037311b123856a7fa2c947cd4fb24ffee5fb212fc55dcc2d1f8a284e241e2ae3444538e57d131ac59dd3de60a8670d13ddccf5c11dd0a5e96e3d8b8a6e68e755f2205373edbbf9ecadd766eacbdcd3c3b9fbd79a2407c84f58bfa6c48b28758d1be156dc41112b1608183050aa03ae010da7833a4b541db954bd941222098a463f0f2c5973c19db10e60f71af2cad0e046d98f2babd765&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.26826792Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default, Name: checkpoints/2880/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2880/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4edc378ef069b015fe6411eaa0c13ef1e073a83d822e611f2714c12ff8820df12133557ef69bfc57755082a799e006a09fb389a97470eea5c7dd15caed1c84f6ed4833500f24df5cfb6839c0b462ce9164b9863d6fba03d66183c457d24c41cbafb40f6ba4d65d3c142a27e19149936eec30434e7d218c4eb5ebba397107b7529db0ba027dc46d13d4eca43d7a325dc3628d54f8beb78f286bad203de388c03c9d228a841ed150726cfba41d2df594eafe2fdf630069541ef0058ca6999ea6ef0871f1116e39fce6bb84b78546d928d191c4f065b6ce36caecc80965d5238d7d55101d03de20cd8d724c70476ddbcb863b2b7e841e4796316f9a25887619c90a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268852729Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d, Name: checkpoints/2752/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2752/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6fa0973a5a29d0fb3f92d53946341b3a2d9cb5d76f0938c989cc249a904cd6b507dec16f74852d921424744d6039a4e5f0fd367d27181c0268812f467505c256012ba0f07b88be551a269b324469f74d7e7db6d4f1de423a369af20908c6f9a2935de803bcf991376318189cd242dc5ed00ed1d40737b03ba0ef8de3315403597d5ec2c4359081eee600a9d21ffe58ae7fd71efd37971a077679a1cd2b4ea881f9e7ce80b8b7b78bfa5fcd0130d551518ef7d711b8d81cd203b571632b4b8e73bf01dac55c5f181df6f8fca27cf083ed296c04d2004985b73912dee8a3b4afa42b447be97e1e729006aacf720bb5baad8956fdd0951fc89557b27f34dc8e9091&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.269081458Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/d, Name: checkpoints/1664/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1664/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=abb3eb902dbb5339906bf43143c91703596fb5bb173797a30fe49312485e8b202cc945602a46ef150d187d6fbe99c6e2468395958a523ffb5b8faecfc74941aea65b36266539ba4562dcfe80ac19bc6edbd20db97b80d5775a8fa1dca1d80f6166b3a281f9876d2d4d4e4944dd2e14d68dfa0e9b351642bf67164d8e2a1bb383fe89123dfeecc0675850967c6eb9f73464dc79ced0e43b201ead1db8d144ec3b075a4f01ba0f060a410eb6079fa069290c91bac4cb6f748189203e42bfb8a21dfd8c66ad7e846e520d18400fec5da2e98e9ebc2c4640177424d1c244463e68a4ad6998f7ba03c703f182581024857d5ed1a4f74fe4017ae4a099c95961850297&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.269124425Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default, Name: checkpoints/2304/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2304/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=15fc6ada76ef9c6b07305bf9fe1537489fdbaff7cede89c6f05707b4c74eb8cdf7c2e1c0d26658d52826a7affc2d017c21a4efda876f9626635d5f514743a66e5704297dd6fb56f63a0c0fd5ea442b659175a6ffa62004bbe279d92fbba7a7e6f353da6164a50525931fd7e54065e595b380490096cd14965636457dee5c3d78a05ad6ff5a961fa653430e01f8029febc39b897ba2d2cdb8bfcbbc04554e0be4f5ddcc35ae4665eda1d4aeb6e93151008c1c5f4e8bd1795fd22f22e7d6785da680e584c955ad6e8ef20470cedac563c7a476a6166af2da0f04c2dd22762cccbb8ed0809a4aff35f87ae38b7004437c819682e0214dbb9d185674efde9fab5771&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.269155728Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0, Name: checkpoints/0/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/0/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1d6112b441cdcc8f151e2214e8766e35a01d80826cc7c2bfc6a645e99dbfed196eddeb8f110a44d3c08f76e858e78dca76a63ce218b9c420c8acccb8a3413eaad9b36cda74f21b4045d8cba8e543a1d79bcaf2100aa020127f144fa04b9f84dde31c77436cfe3cfdc3652b106ff8fbd9699bbfe133d14f028c085867256b632fc1ff524e4f6d66f9540acfea78ebc640723541321978c9c69c5ad978bbf7afa33aecda447c204c5b222e9577737444183faf4c7617ad185af76224034c68522003588f187ef719859f3ec7aac9552fd4a7be539b72db5c3856e98085b13fdd4abc70a68088cf656a391466dea6b9b8e2791ec6c3918eb25bcae72f1c4004c9ce&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.269201777Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/array_metadatas, Name: checkpoints/1664/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1664/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=76a34bf893ac6b84f069e68d616c860c6c1b841921abf54981531fd9e5c013e655af6856fd5aa274d2b5267b6b3b47757cbd9f17e53bb5348bbe7556104ad19edf7c59d035c78b232ed33521d0d427673984fd316ff1c9fa8eaf5d5f11b2bd70263baa42f39e7729c5f4f2e97e0d6e86e4eda7675c085eb7ac477980f3df3b2f94f5f79cdcd432d0d9aa16447277fd66db137505a377f2efd5ccb06533e8b316bae550c038b4edf4cbab76d0eca31dd5870f0c40160653a996f551f489e33a8582b6d7daa755d3d8912a9a3bcd87b2eb90bdcf2f8df210359c96db76384a962ef650dc64357c2ddbe20b462013be45f54405e192480630522613bf5c8e763039&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.269261715Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176, Name: checkpoints/2176, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2176?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=047feaec07b3afe590cee5e5500e51412975d0bffd3b73e070b25a1b5f56061059f78be4dcba882500ae4f572081404ee75afc6bf0e5395a786b7470d3bd17c6996921f12239b9f377a14232020e36a03b5b004ca5d3e125a0de224fc9ddbc8f4bf4e8f848c90ae3bbd7ab875ffb8dbc8b3f05abdfa673987a9e8eb7f1926fd92d423f850569f3340534eaa7d89c4c19f6fc58cd1845514cb34793eb50e84a5e03aa79888b3b0ddb363e322f5bb0ef7347323f1e3c067464f35046f303afbee5837cf72bf430e6255e7260f7ff5ff51166e098b473e91124d05b9679455e110eb2330205d58490d2d643bc39e94aac1eb20cd1f305be413129b6a7e49fc5eb5f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268386898Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default, Name: checkpoints/4992/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4992/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=813ec7932436623ff77e86465d9a015451ee1d6fda6ecb21706c3b8dc2154dd1e2663b52b7eb01c7981760cc9e567807952e9f2473ed3bf628527c53d596563ef57c7933c298572dbc0e355dc77ca076cf79926c46facb06553add2935410145d740e39f423e903e4505a1f7110095da3baf82e8e776481e51925c661ed0383587c27b1220655a13c812aae535834f7c120d4b157c89cf12f60b88e718edfc36a4465f75c596a2bdc557c66bc8d090912ec69cb105d62a80a73ad80b25a8408130eb28929ce0fd293edc526838fab1a808f0d143633ac31a9f091386845eb9888e9fa53f3b453e5775708e70fe380a9b85ffdd945259a70ecfda01ea3fc31583&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.269336137Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/d, Name: checkpoints/5824/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5824/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1a3e6f822a7fc3edc5d90a0334f0257e900c7e248983f23b97c128e31c0ebc59d2d73e08cb0f9b03993d3aae2416bb4925f4338843a6e09765edd4ff70557a22068083b51b6b27e5835efed14e7f57a198d57f49b268f071fdcc4ae50ba0659c25687d0389f4f331958875bb766d2bf86239e53d905153c19bdbf61e8a472d16e06149b8e16b9cc1a3e0ef2ace892b024af2a2fd65a30e3d9ef4eb7fddc3dffe083378ac051c20500f023096c328d1b1071bbf17912278c142471e8369b8770bb5b62179851faab2272f90f469a69b7877fe4a796b2ccadec72310828c54b06c862ae8e7585fa48919822b79350085ac3fe98433b3831e7f83472eedeaa916ce&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.265806744Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888, Name: checkpoints/5888, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5888?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3c7c1ec60ad406fff3c4569e4ef25920affbd2135fb4cebe36466d484b49c8aca1f946e890d1dc9a50716ab99ab813578c7f2598473b2f286e94b3706a8db961ed781311180008cd5c835101b3019307e6a49d75e81e7ffad17537fd08941a376f7605e52e5976a94a457d580725e7f1233f427f279903a830e738b3a35bb0c0f98f700b13c44a26f54f98a04263fd318dc4b6e04f924252ea79a680a475df072c63be83a815c539b9462f6dce25331a35de45d49c9e493793729708c7d3d0c1485d9c2c3cbb20226d00df56d1fe7203b412f864ca26914aed19cff2daacf1e9d5acf85280e0a152dba6973457f583064d83f5f274c789f3a5c3fd53835c672d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268368297Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096, Name: checkpoints/4096, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4096?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c4ccb0f6533b1f2742bfb8a0db7b8449d772964090b9860c51b7b00026e729e87826adea020e83579ca4bb61d56c04c39637b819cf64c5e2e4ec287b11967aa256bda3e248c339738fb377b0f650b34c09f38d29d08c218b745520375a4acae02bb44015ef8919e35df80acab34c24030752480511c6fe137989bc0d06a6b0314ca138c48a012ca97c368e80f797c3d67113325321ef2a4dc319e3fd87dca221997c2ff394cb0d09d0ccbf27b6a084e3f446c6a7fbdc93d7163860bf9b099cb9e8d271e2b93f5fd689c9c3bd0736f8c21672b6f74a0aa17e4d9db679fd9943a4d8b600ffb3b3cdb5413726587d477251207277c02db628765c3e2806c0e9ee6f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.268173691Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/d, Name: checkpoints/5632/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5632/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2717c6c98278f889af2c601918fc73c2e1512429e2a7120b3f1352b90e84f2f13f09a4d8e1b72e00732e0c78ac7cb7306bacd4a04aea02551c96508b73f517e1d84d5131559e4fd42a16306dd66f6dc8ae3926874d92153bbdbe99299abdf0592988b73cf876b99b7d339073df0cb695545137170a62bf9612bb377e9bc8d781260f7b26d1441d5233bbe1be747ec20521ca144826e62f722d6d04e41a1ddc68e996a3a3ff076ad79f4a6dd16b7e0d217a866370d671b6e593b8d2fbfd02fcfd1d7e7b03e40dce8682c96560507d33507138680bbbb6ad55d89a85bc4963da2d2c511252bc5066f457a9f5e29f3710a83e8f961d3a690c9ec09cf468a51f16b5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.269033838Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0, Name: checkpoints/4992/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4992/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b9ccf202b5e9485144200b203a683e6f622faf8cacfd147aec909b5fb201e22114781af8b83d458631ca2a5d5cd9554305b859e8bc6d369673341c230a09c0d15dc1e62c9153a2b5f8a504a04d0ce24a48887288d05f403895ad46b245f60f32f35330a774b308ad9fcf2964c005671c41fc51388119dd15fc2d27c4498e32958e995c415e2f9c5187b62bdc52c17d67760952ecec91b6808c4eef61a33ce5a5fce1f7756b550f4a65efde1d0a2ebc19ba2ff33dcd4dc2a1b50a83373704659d2b0a4675b610296b392827aa555e57378c355915e0a9ad740b493f41fa6ef1cd9e6967e01989cfeda016c09f8712db51b5b466d84d6022da4de2c972231313c9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.2690886Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d, Name: checkpoints/4672/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4672/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8a394fd30c320a0d02eea3683997e3bb152de0f1256bc46789029c722007292568bb4f18914e420d3810cc0be15fd7ab19b2aa407fb63fe6cf810929c0812831bac8bfa8bec4e9caf754423aab6931f840c0191600abef01f40267a8eb32790f599537b7eb94fb4e11b35747b0db44f6219ecd1a1700d01a419c7f347eda027f659b5dac087da4a409b01d476331a209901772ded6c0ca122f09a95d9e90ea82a135a37697ca51fbaae9b66026a05154f10b087332958b12c729d306e4637adeeedd3ef7cd34dc765e78a114ae46e2f9431e8b5d997e4b88882d59de41bea13d7e738100beb1638fe4a3761b5096249cabbde22e1f17bbb2a70bb69e40b2aaa6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.269355096Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d, Name: checkpoints/5312/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5312/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=896093065e11a4f907ae281a9698366763b2f002dc88bb5de939bba089a4b25e0533eda9300b18a1cf2e96f923d726a3b63aa4088df32ce7c20cc9612e297265bec0ba44c06f35de2831cddefaf407875c30e5c5c62812fd4ba2aad437d88bfae9cb93d26bae6b480c211f12e976b3a8c1411e4817de9116f7b33da51011d6dc32dda4f8c38810ac9bc1f5390d0c1bb5b84000eebc605da52fea8702a664f7d967016d7a8d902b2cf7633427095c038d4ed45784c8e5354ab426011e1986a26d677ce899b97f1c17db070ff0676a3c0a3694e175c31beff7a5073b737523da9eaf8c15dbfdf35a8f30ad45d4894cb37c791c24ef62fe9a5562a4fa183c547e6c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.26940621Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/array_metadatas, Name: checkpoints/5952/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5952/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b4ddc31c4701dcb92f689c6eb3f0c422a80f55d61a6cd75f69c710219abf8d462b479a6667c03dc3aadba3ca279940c14ca51a71d40783c3917e551dbf79bdb218fabe26b7ac93bd93b0457bd9730931c88b61a3659b93e90dd1cd85b2f9917650ab13b816a32be7c37c6e8afce5ba9e6bad5335ca12059206dd4a1c4d65d20bab515d24f99a0e9f0201187b5bd8b5c1bad8e1633e3746c7629ad59b18c1b0d6da4d4899996006a71221edae8fcebcc22b569401ab49991283d36da49f83791e74501f374f63114d5acbc6257fcdd9e4527b521de71e5655a079f91b88fadfd7a9b3cf841f95f8a196b332f3d8f9c8ef3c277b2d33884270f553fa58f9bb6390&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.270058121Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/array_metadatas, Name: checkpoints/4352/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4352/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=507caff6ea512fc174a3871a44a369b6d8953e71b1480f88cec63dea13f006a743bc0a58000eca44becca26b0e81d5fc0a2b000c8ea94713e68f48e5b62d84df4fa0188028b2e71047081e3294d023f5f0f045bd1d4da6e8cacc4f7d933d77bfa6bc4002df0a9e1902f49307cd14157ff598552c7b8bb237c6ffc31ca20814223ca919cddbba429d524b9c4de06a6824fee7bb9c1da7f9b3cf0251490fe8dd40a9f9b0583409f31579562a554ac40df1b8c607957d354bcf2a13a01b89ab90d443e60c8f1429cd6ac3b868f6c641927c0f2d252eee69d50c91669f9f65fc4412a2514243747a9890c98ec168e5a0ac5e16964fe0a69b4910dd27d103b3c60c09&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.270145161Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0, Name: checkpoints/3072/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3072/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=57521c02a13f01914ee3061d39037b47fb79f2545d2df79ff4cedf59daa9ca2bf1bbfb8dad6f7c09a857b6b1ed88b3e12435d64971ee2b773a618e95583b6847b485c1e0639ba7fdf432834f95d54b9c3d44351ca356d091f2e3c93306e634ff0917c8b3f978efcf4e940643351bc3a68b80bcafdaadd397e51ff09d4990385e1d2efc000f52143d852b5c24938a59709437ae28e6ff6a880317831fc307b2629656f1177ac1f12845e43e42931e9575fbc9eb1fbec190487a35b65f883c9cd145e622fb7cddd1e84d8a2be8e76c2470c2c52e7352b166750602823779dae750403dadfb37c1755529e95a807c2abc8ffe30b13c4d05bb40d0ee5982ce4811a0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.270291559Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/array_metadatas, Name: checkpoints/0/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/0/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=85530a682f3574f067ac94c7903283e9a5eb1278475c31891e1b182af061b5e2a12bd72609d225a39b859eabd2715ba67f0a2534f2f80244e000a0b58e55d90dad52219e6f732736dfb019184b7a260887edc25eb08492d3b4bf9a58c461d9c35b8b87e7c55b69b6374a917535083c1c62f9cf419f073e9f744ba1f945324520e756c57b7f374427069f4b928df307f33b411a32f23b5df87561759a2d471a25912625f1ca030a5bd4e2ae82eea4ca213b955689b5ddd52033a783bab980a05b235738ff1b39b1cfe98a0afd212cd453f2a748dff2a55e950441342892abeb949bb39b20501c84512ff9656c8aaa18f88babf5bfe19da07cecd04bd35aaf0e96&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.272419897Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/d, Name: checkpoints/4800/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4800/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8a97c7e80de6d74519d77bab21c764584870cb2dfcd35a1c7fbff4cc563325345482254c0c93069bca94494aae9796ad698fcf01c867ab59038c6892fae594413cc7046525151acbb5bab1dd1d7e67330931107306349a06731f4635b90bee40539ccb03aa90095289b0b3aada18348c22adc56f2338a62a6b53bc87c8448b1e1c702723ea92d8131ca290f4f4f3ed6d301366e4f631808e87f6aeeece86a79b80ad06b3e5a4fb02ccbd5cb7ecf418b9c910ac9623642c67ff3bbd1d109a6d628bde69b104f505199bbad3be5b5ac75285579348251d9c8b745ebe98cbb42f44e660fca2570a86316f49a696cb2e37606d87847d862bbe1f0644bcd27153e350&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274056463Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/d, Name: checkpoints/4096/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4096/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7037be5a00f45ae1b8316eec97c249b8d71fdb7553274b2c45b7aef6b6dd6aa29dc9f31b740efa263de864372e112afc687a339ae4c3d1ee4e8593b5efa7b2ce023eb62c06947e3b5cfe08fc77d7b2caa80d0f8bbc576ed881650d5ac199cb84100ede0526d5e8cf98c82ea6557b44be32d89404f86eb1e08b1daf0d480ec703f84c938ffd355c3420a1589da93602065654b76fe0fb1bfde4098094ec1d4a4cc176fbb93e1a65da6b4acdf4411a0ccafb85e3b170bf7f9e7a3df42a5877a84407d163477059475c357f3087bcac8175825982ed8b02aecc55baa7b07e2c5072f213dbe236192c49399256510c8510a126c75efd9585181e37fd630b3e86b136&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274073525Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default, Name: checkpoints/3968/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3968/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3c5770f02b6489cfac947e102c469e0a130e6f8b7b2eb5182be276eaf403556888f475be43732c3d70400b4160c7e3c3e4c7fd0e0b5beb9bcd7235fdbac5ac04d8c0560cbd22f32e40f629efbed3f1d7cd2fe6f2b18ff45f62196dc30b317fed1cff9e8ccbb50f267da7c940044bf757d7e3a10e6bb93e088e9b0e92ebac0036e5d66da38cf0ce979f5f356a17a5af1c1c72c55add67ee920d5530b3335cd59a1d70295299c3636fa8b2e44d9ae679511f67c9836bb2da59c38e8a68b1b5c98dd37896950ecc364476a1893aa32c2ce57c674f25cfb9989ac3357373397b3e28dc3ba6b214c16d9cdaab0c730641bb28ddcd1ace70495256ff4b73a1c61a0312&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274079127Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/array_metadatas, Name: checkpoints/5888/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5888/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5e070ce453261a343c59365c5827ba3d0cc66ffd7720177b10bdea895b7dbeb2c394ec10ebaa27bf441607a6562f80a73916e749f100950bed87e44597c76df72a0cbcd9e360086c4725767ec06fad6cfc9d06784a27640f0c919ae5e3524448e352d47ecdf618cda4e52210a8bb4fe2dff532199d8274574cd4f145351d614695dc16a877371af1fdca10044c1b2a267529bc2d2711f7f87a2087e4831554f847eca7b868e14af880c9cd5ac661e4cab33914dfe2c4d632c46ba66f41818b59fba95cb8eb69614b92e71bd52c3542808fbd909b75b939591016508b4d28fdf67ecd228ca3a0f2cb03f32eefaa791d9f28878305587f75b7b9577dace6ba27af&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274071552Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016, Name: checkpoints/6016, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6016?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=302c230b26b66a76117a36264112258ffc233a1e6216d008d1ce6546f63febd658677dd70a1466ee890645b6aa2507ebff69f2b175d046fbc0bb869fce10d541a0db3a5f419c83f63473e913ec8797030f96aef2d08cc415107995d63c3027f561543696afaf97076a6a9e11542be8577ee21d7a0a1d58e3d4bebb78f0874d96e98c77e4caa61f84395532cd4cfef93790e78cd0a0e4727403de9e81c33740644421ab32f0eb15e4ff8aabaefa0d8a73f8249f4726a7e9d936f2e357b4eb38aa70f13d67523285d0599069443245b0ef0623d6a53ad4695ce4d6756264ff2ce6154f940e553305ad467a121b82e8081c35abac35015a3a05df7a8c32efe27474&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274070465Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d, Name: checkpoints/3840/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3840/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1cfc69b8245821209823e6e0932cd18d55ee6678290579bc2b5aea67444df74f2d9f3f0d915a8ce7bca9cfc1e5b7738e980661a178658c1a2284816e32f4d8aab2c740f09b2640cc8f5e501f946f45899d1df6300a655e56d7867a34153c5b32fa62fb73af232fb31340bac23ba34439cb25e6034ee0c926c2f3b896360c299b8f8675d2d984d297915655b3e530571b4b36049bd13ca4ac8059dc70a480bce68c6eb3ff70fe8beef67043895556d731a89a9534d3a4665f90d3e287c5b978f99bbb5878dcaaf10aa8314333b077bcba72c47a27e5b774289b0bab6c271e31ae45a373db5fa2edf930f81cb9114060947123d4127d21da438df275347fdf1f43&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274168738Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384, Name: checkpoints/384, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/384?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=705e767e4b9d0ee72007e4609e4c2eda14d9afd8db9270ea717f278c3ad39c30d7ef884bc90968411327f93cb164e54ecc5e561d7ba61bc0adc60724e33963cd7a265189b6096ba6c8a3ae00bafe80236f2e5ecf698adc179f264f3cec1636d5cc63ea636efdbb4ed7d04d9cb509d4f3eeacb352fda7e97f7376b734bddfb010577bdb4df44d8b67f79029a9697876fc6e3ae9510c055dbea1cc0652534dcf5c20c7e833cf73cb4d2c51bdf5b7075d797b3ff2ea1f01d653a368697aba0fc65a00aeabed83207e6fd4bca070775990a88a56ba37f39a9bfc0f0f536399dcf0a13a88efe1033207d45a5af7767a3f401581444295e617bb1d92b51df9194ebd35&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274058669Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/array_metadatas, Name: checkpoints/4928/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4928/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1be6da55ae781c34721c6b1bdf2941b669fc6ae2910224d2503ca938cdf05a94173fbbc39f5df77709419b6dbaea971149f18f46a03086f76f4b8481770f4eca2eb7f8fe910ea530279a8462923567503245cebfda4f71123622d7e1b59045f709f8734d74dc0a46af81d7ca9793833b2aa6348b613fa51ea8591b8d40da79408dec52f79775461a83c1c09fbee4b16f46500ccfb986288b50c219ed4a5b708597c35513e7a79c18bf621f7f1cb0cbabb0f2d7b11cda9c3ffa661b7263febc320762ab2ff62f33a49cf36a221dd94a180fdf5e1516f80cdb30fc1bed643cc3c013381602c39d84d4c4c3bec34236b980f67e99a83fccb7be9c22ba73921375da&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274200356Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default, Name: checkpoints/4544/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4544/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=052395f154b121961d6e6dc794d894bff84381ebcf2af1b517e84632e1519d7a5fcf8e927d4c7eabd3b2bcc909ba7cecef72ed2ec373635e5c042aa1875138248b557d1c497bf5b99a5b15bfc6d8bd508464dc68e304cef2618786dcd15c61fbbd11d32f120823e6f44992cd00622c97196428e8de6a09d59c397d7fe303820113a320266e1a5c1fa4448407afcc5dcebdcf57b93227e2e49bedd3819a80d43fb7b81815c4f42f170f49b56f3b524c3dfa0e2d79a2c169914a541419279f09542fcc31220e2a05bc6335b3027337d44610606f3d0c08fce1ef238ea2a8c2fbe2db9c3603187ab00ae65386e4ea14249427b8839c9494382e985389b163a992c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274248464Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d, Name: checkpoints/4096/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4096/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0acece29107f77bbf68ad5cfbd88edcfc25f25db081a8bae3c9fccca54d4b5e2a1e9d282da0662ab3106ec7fe88959931add5671cacf8b6c350fc038af11b2f7f25e6cd27eea3fcb6e9956f22ee989259a09764aefc1ff8e62230481964686bc5a7ef94f113e9e81f7b529b2580ed4371640447f08de39902a7e17bbbe74aeb42c147b4984f5a84dc3e83f7160bbeea347f5d83a7b99735e38fdaaf2f07bea829066f99528e4735acfb4dd6451b5dfd64547a632395272e18370a9cddd15b8e670272ff974533b274238d3c7c97bfa8b499efb3d154bcf3596eeab8ee77853c8326bfbe654b2f4ccc836128070ff5edbbd4eeed1574bb30fb14431ff6a80adbd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274205156Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d, Name: checkpoints/2816/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2816/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=507f398126db769da92514497a5dff35ada1f1f491c68cb10b569a42b965cabdd7e6437e55c4599a8ffbba54fa07790b6db0e58351e329aa3c51dd4f313ad2b532a4c4fc56757fd7724bdc8575fec877b644157476ffac55a60270fcbbbcd19e162b015e95fc1fd4622ac8d70879ac07f2ed152117a7a6a932608c4db840f04f5be2d060ed39328f687a5c7c06f20c2fcd2335ce80a7daf0bcde543e6aa36d8f0cca2c699ffe54f67bdb5a5810ac524f2667844c2c10697a0e27ba22bacf513a14d6f341b192917362217603c09e9c3ce98cf5fcb1dddc060cc28945962eb83ae993d6a75da244aa1f4031ac5769ba48337dcb2acb997da7343562be940319e7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274257009Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/array_metadatas, Name: checkpoints/256/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/256/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4e6e1d0a24493fed07c51c72bf68f4d46b8b8a1a8e668153477701e19b6b3736c9ddc2cc323a8773c432eb464c7dd1d7c190df426fddd376e20ae8b75d9f71243fd85cf13cfdb1dc9c687974f835f914f154df829b4a7ecf77bc82a533dbaba3a5fec28679660ad3c12eb69a11861296d0dd87cc44cc6f8bf784b8a2f7e6d6add06214e2a8968b24553652015b5856565dcdd209376ccb7752574d68868532408cf42f4b8996cfb5bdda4fe402fee1b1b39dbed91cbcb87abd566e64b93d9f5fad88448a4abb856d4cc0b97ad2bffff17c2adaf4d95dc42519f2904392a44b9705a9b0886af44014e3410aea1c5d1bebec3edd0da1c74af3961d9f31f0d52bda&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27426698Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d, Name: checkpoints/2112/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2112/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4151809913b0fd21524710548ac55463d7ebd156c6c723a002ca02c575a01a9e8e265010a27b2a3baebf28e78267024cf076552dfc67ab81657c2bb34bb3cea7be5e279ee2c7c6382056293572a6373abd58d858cdb2fce0ec72ab2717b663dcd934525c78f76dac3ecb6c855da4e43f5634edd29648362e7b2dd0d9305f31ed7a6bb48e308052860fb984832f55c36b3d9a1eada2256275886601da426febc830da55b958ba5320fa6bf3f835e31adc952cb69aee7b021966ed5f03d3757c0599e290453440474dfe0e4ac6f58090b7dcaf1ff23af2aabd569d9bcbe9d1ac92cc2b056781c2a1e2ac3928d586dc1444b0002175d9e05b81b987a0d549f3de1d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274349093Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default, Name: checkpoints/3264/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3264/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3a3a223f748a0fb130b55970640767f60bcce17e4cc071e7ce7f2f86cb804af82a2743ed25cec59dab38ebbd730d41044e929b6f80a36b47c3c040f62a3f3b6a1a268d8a3881fd9c0a0ec6f48694f6b61c17e900da521ee8a435ebc50fcd635b113243f1ef7ec8b9bd5c4a4d136b50e1a50dcf15dbe0c15a878637a849b10647316404063b6aba0ef9ed366b3142118f6f58d3aacd18b94aba987f3a0996ec18df5362c2e823ffd9da0cc4347361a7289b05af86d4432ac4d8fa2a3accbec6b4159519ec5518492b71b380dbce3c92b49c099e1c58c2f322a7aff5582fc5a3e698d6640ca8d4f0916094fc91a5fefd5b4af09e8a8ad2ac82c0a1243a2bb739fb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274831702Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d, Name: checkpoints/5248/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5248/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=51966ce2c015903a98014da593ecf2c2bbb4ede833d2863066a805d82edabceed53810e38badd1ca139b202080ba8eb0c3b0ae19ffcebbfe5831c3d0a4236fc8878b3c4f5baafd77edb21834303c9ad9b38610b19f7597f711212e8a03e22970ca4ca064679b068efedbcb95dcbd26c259d1c4bebd0db9de6ed2382af1832bbc2efdc1fc03be089a7aed0255d33d7ae81624a230acbf9c13edf75231838c5e7bcd69d0ceabc8b01484aea853029b3f0a4d3c38cf7d773c867f9aaad0e0eea27d54cfa0ce72ba10bbaf809f7e1747a634c420ecda7bda684f42a998e5e7e270c51baf946ae8688e5e1feecc5f2e5f0b30307519172b1b297a8a4d4ea604c66848&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274862565Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/array_metadatas, Name: checkpoints/5248/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5248/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=bd3905b65aa60ca8001b01cf81b9096d4152950fa85337ad0492ab1e048106970cf58597062b6785e77714015876ed4a38a17ab5129a0e834d20acd0e237ca4c2a42e167ed5701503a13f0652ea4c0e956a2c4d0eb007cfde1d02d1776910d61d6174f03f225fb6f8d95cd9b9dabe51b4eb521484b98ad647a80a59ab3ad970f3fd268b1cb008d69b9016559feef2c7143066abb086f95cceb6699c010ad7dbe19b3954f2b89781dab10aba70884054980f2bfc2df8daf87c90b21be4bc7ba2cc5187f6ad6687b43191420be7000440b086be8df0585c832cd0fa87ec6e5095d87040ce4aa221840d7c970f8d5a179f4616ca658629d8647e489d0b6929fcf46&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274910256Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d, Name: checkpoints/3200/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3200/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=89f8f7c98aa9d3702bdd193c9054091ab6682edf6cf4a5bc942f865dc974151269accaece7a788ded75819fd5c4a7ac88a00a84c86dc210834641f6f40ab231542562371cd1a5cd44f61d3fcdab8496f5b66ae318741d812199909e6ac611b63d0d81e0bb1129f588e45e145251894004856f135ba73f8fe2180a93a310f3872a02d4fd6ebee27afef3d668f309bea823d626dd6f48c47a51b1b9884208ed512d534d09ef40b36d08168a074f578e282cb263e1643459ab0a61af06d12a51dba81e9f52d62746a5e4abdf36cfa60d1b522c4d3a00dab402d94615c7d7b538d9c82a949a411603b4ecf7664a69e0bbb7b240229398914862b54c6129acf5c925d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274964489Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200, Name: checkpoints/3200, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3200?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7738e1792fe3f54402fc7c553e5d85f909ad347de91886029b2d2f358b0f097ccd6d086e2c4d4fc5b3bbaedf7918a51197866d3e466c51e1f464e60868dfb199e08c45bdecea0c0bbd0fadb42bab4e84f144cbec61c8ec2c32a778eee4014cfff379b28831b879c803de53132e1804abd422c89af2c2f208b70244623aacf6bfeb83c39310b2fbb47633f6711fb4519a51a68ba85a5730aab936c9571faf528acfd7010db24a94d2822090087f9fdfcb0af3ce49dbea7583472dacefc9ff3c38cf8a27d7034ad493cb22353b213bd57b525196bbba919e0367dfb609dc0f1b149e93fdb0dd1dc625106fd3edec3381b8b6861bf75203a6d8e86881aa3764d02e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274985094Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0, Name: checkpoints/2944/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2944/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7ff54fb3316e07e7dfb781836496b17ff70c84ac725fb4f98a0528aa88e4f72c3639f36b399a8a0218d882ad16027cf0b13281f6833a4fefe0d1711087e6ec40daa47c670bef6ca4ff08da9e0b9f3e6dfffa48c9a8cb0dc19701b37eb5b2a0bfe3e4352cac1ab8747da556a53b866c8a6062a7553cfb1132bef8fce1b912351b6ce046c722f06d390a180c74937aa79463bb303649b8e7bf7392e8f05989156e377dcd6cab92817d61148d384da1ff97d33388dd508e13454254e5d2e2b404fe1ad9720d62d1c88e7cefe9cac77808fec8abb0c07783266d980f8349e6a78d0c0d10111c0c3b18e900525c1b38928a484bce52f1637e80dfb05da116bbf5fa95&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275015278Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/d, Name: checkpoints/2752/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2752/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a51d8820a0c9a59637db5ca47ce8cd447fdacd5fe06288eb8e8e04710e86cdd69774e28d902149cd54e7f8c24702d8540be7bff2bedc069489260eb5e4d86185d75b21a16e61df7713c45de5a567bef02080eb106ed34b98aa34fb3d00a80bd2bcb232b2fecd3e3cb12511dddced73ecf76178ec276fc06875b367e1cf575994ba89d0d63fbbc5669a2b9f7308a2fc2455dfd8ce92a7fceb487f00af5e763abd3ae430ea7fe9926bbebf9f667db66d98f004b8b1455e97627398170da28be908b32a6caa216652cced88b076588d4f8e8f840e539c5e2bd22c8614474ab0a2e858d2c2dcbf66510ab2a039770bd1d078fb498b044c2e664224236f47eed2b9e6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275036899Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default, Name: checkpoints/3072/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3072/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=880249221879bf5af60d0845f8c480e76f751c6acaf7e0a6ede83b1c5718449f1422f1d2b61a607ed9be241be28307d999b44f0d64aa586741724b18f4f78442fba53ca482ec4dbc46f700941c82eabd8039a34dde1a1def3e5fc24fc84521d8cfe265576086dbe757414d6029c72ba827f8fe61832a56a6aa70668f907a3a9176e92427a28e569f0b4c5a7177817aae226d83eea29935cbecf5abc6294df539f8e14acd9c887897734515587ed8f6123204731c0c6797ab8ff70f5f542827eacd01353ccc259a631040ef8c53b7d4fac5638a4885ddfc784b6292a790ff29229c58c4455b43212104716c546ae4c4c753d6a8121901885f746fe8560263734f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275039762Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0, Name: checkpoints/5376/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5376/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8ed6edabafa296469c39d641d07b0c759ace372ab5b8bc7190a622bfd667cc97aa369f6118f18d5b0ee59418704b201ba7ca7c9fca6e521f861ecd2c68eccff13672660c02bbef55342c6c2ff6412f4caaf2652e840e910657ad5066dadadb706f850a3c6633b6a4840a86ce73fcbe582224a7ed0e7a6d5304d32daaf189f8543870dd19dc66415a54780d2cc0c579b0a8784ab3e7fcb5732591e0e83b0e1f6d15fb1228177e1b4725c9f31256260b2d5200386a3c02bfc8651be17a7524f9bb3c05e103ecd3f38abf4f086da3ce42fa498357de443df565fbdc1c1a32c13bad4580bf8dc3c09ec74a23b7136a49c3ab4eebf92cea375bb028b12351a62a1738&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275071221Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default, Name: checkpoints/1088/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1088/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2706ff5f2adf8e3fc77902cb3e5ab5dea3c40e8b460bdbd37a24e55117edd6fb8f3889afca221c18b39e6cdbb15741f7b9082d16ccc2a6fdbb6b2d7c5ad84577ee6270c7b3607a4b6eca40ab25463cd5a48a45f9782fad217d1ef9497bb00c4c0f4f636e30559834f028d6141483e2da28bfb40e05ad563c63bd7bad207888e39fcb7c3a636bb28c3131a3e2787749fb5d1df4f08bba84596129e1e585cda1c7d714af95046d50e19999326ea17ea710e423674dd0087ed232c3d400d40b96376c72794365bc0c01ba5c43084ec3a53e2556213a26c728839199b10010a048f80b00192206a184255eff2a6440f0cfc9d0ad149099b6bb9a75cd94c4c26f13b6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275092411Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0, Name: checkpoints/5504/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5504/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=019b9396d90ef3738be21bbd86c16e69d35c64b74bcf2b88866bdb3d27c7952cd9de09b22742740a1c60826061936caa01ce842cf105abb37cfeddc0a121b1c4053b309b8ce99101819e8cfa83453d1fbb38403629f44ff9ede9be82eb79e3304c0ea0165fec5fa8cc3266f4d7dd581722145884ce9839634129bb62db75d943f862f1e41d3aecd4e59669e506c28f07c0692961acd354704eafe5679ac71990d8fe21c8f173c80cf0e1db589db9470a651daf418f45e3b9700d613d386a25d7918842f90276a1bd8832a0a01ce4fbc020c71c051ddf9477565d2008b4220f615eb535d76d448560aaba184e8c6e5b056354caea0f2820c29db8fee0a800f54f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275108191Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints, Name: checkpoints, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5d231a7f4d4e53e28922913af8892dbbb19dce8c26c4f3e147641da16904c5171451623c894a297dc709e98fee1e78e8cd20c06745e7c4f65ccbb54ef66e3c959a8245ef8df693fca9fca8f87791c6a2510e6dd454ac73668e6958ec8e2a0df38893b9e4a09fda3e9f999d60ef5cce042b5bbb5412b0dcee7fa16da720e5c279e07c6fca5efd9177ba8e98db38eca3af0ceec7ca9dfa7fea2585841db1090f0f0acaf461e907e683e76a228c3516c864636f130c7a056d3371416bba3a2ead12d6eda38cc32baf19d050f37969bd53df21d97ca75ac56f8d49065a8668a361cb959dce4aaa172fb937a78f16ecfd2e5b41392114f75c11f5a4915e089703503c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275140319Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d, Name: checkpoints/1280/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1280/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1fc5bbae754fbf8c568f0d6bfe202657466e3cf393aaa04bca68d30bec01a08f27217df55ad73e47fde24e1b49f37709bddc558148c349f40574aaf58e25da28dfb3e2c8678e14d845f87665ec5a1a9eb2e68e706fef77899a333227de122580874e896b9914be47eeffb846c6e8c59c6b6b74a9a0ebc93e704271513932cb5f4655386a981317d4d949c98c856ee478cf57b082be89b367811140fd263ee564d818f06d2e7c6e1a06bb66e16cd8a4ec582bab2f8e06afc241ab8131d6c666032c504f22870541283b7fa9af97bd7d5822fcb91ec2cd461b3a6102fc6d1d13a8a2794a93f447696c432e4f34bdc10c23ad938c6dfe91505fdfa2bfd92ef7e8c7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275148989Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default, Name: checkpoints/6016/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6016/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=244764c5bfa533d1a360e6d4db221854c67b4359c0df7f887c9a7e9d73a69d0e5e23d8b5e04b97e8c7f9a1ce2591ee73de1c258d8c686713483be59499d87e88e6ef6db15418b758049327c5514053b3a5a20e39ccd1d249bdc9e7a4fa69586b661cbefb9d97bd98243c35d9395046d3cd76b6b13fd70b3d633cd7085ac93f912e43d7d07a8be79ad30894a7e2b3b2d0c0599049b6b76c0d4f24a80077a2c381dc85653b9b9a058e1318ad612d6342cb18369dce6f27feab27ed9fbee06e0a28ba65edcb38fc27ca0b21e241ae32a9213d6beaac6cb13425b94bca903fbd31e9472334341d8d8e345fc0a1697a893dd0160870f579e174d3f417d52adfc7b8ff&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275164615Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d, Name: checkpoints/3776/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3776/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=175d9f043ded4f8d9970877522c75b1410f57a28817dd02f768d97ac646b9a129a7693cccf0e4cfb4b039ff700e47fed44449961947dca308aeea7ab465befe46066dcbe543c227291d91bd498df12052269f36ca619c1a4f4a023128bc0572c5e66ebfe5155b668ece242af0ade2878b90a5be97f357b79aec41dbe049535d7b00f2e0560572f57364eb1f8e5b4666b61f194283bb31452e1d49545d5f29062df1eb93e0fd9ecbc018ea9946d7fdde56f277400d3f4785652514fbd9299ed02fac2fe37a588993e905d4ba8381dfd17b33f9029afcbce71d661025cb2a6a226d9159687d1dcae778e20f77f5f5bdce74b5e8f2350cabc1174afde1f28cb8674&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275168747Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136, Name: checkpoints/3136, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3136?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=39d2199705862478bb5ff9283ea8f4479141917265f96c2fe41624c4bd94abcfffedf68b526525e585a93435f2441542233e14886a5abf39aa9c5b5f2df83beccaab5405d15592ce3c75747f11f5cb20b7f04695e0a91ac2c414b08589f6db940c9e5ef0722cdceec47dc359e740400189c090b14c37ae4ea35353656b0584113f58509713f5a678429e151a0c3ae453bf7636b3abbfb2c01af5a27fc5ee8d2fc5d47fffd86cb6928722dc3c1505eb84950978f3dc694f4c4370be9cea8caafb95c5eb0347eabca10d3ce8a7a00859d0c11dbb116dd55487c4bacd1e9508403c0c28d74371b9116ae2d11f34acbe12fe56598a02c6c47d3a95570a00691f2e53&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275207946Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248, Name: checkpoints/5248, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5248?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=622cffaac1833f58cbba7d1bf730347fb24d82e95956b905446ff20a41343196dce3461553b72631801208e55e5c3bbd5184101337246713e1b5f4a0ef856f4d7676c0fd02c0f2906944e178abc6650e770ea84a9dcb98b093e53735c1caa1356075b38e2b9678fc9d06368fa12e76182848ce35f4e5f9aa78df98742beb223b5edce6a99a7da1feaad71c05fbfcf6616fb38c95dbe8f122ec8d4885745eb91d4d24fca4596d24bfbef732e5eb522d0a6644b03e75543143e080edab01fc7ffaec882ce13f7e5023fbd176a8b4d92f58d1c140efd7a0834cc704689c2ab7de5ce33a3b35789aac7b4b4b31c5c6232d243d6f4c530ce9ecb5b81ab7b90ff5cdd9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275223088Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/array_metadatas, Name: checkpoints/4608/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4608/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=214eb9097f5cfa77f5925934c41385d9fe2f0ab7da8af5a5844e7c8113805bf4cbed9f00998e9f0b6447174f6caa947cb9fd1085846a63ce341fe1092e21aa92ea39c141cd585d151e19ca50157349b4f33ecedb05ff20c893ca3025f966d7d1457a85e25af3fa0dcd86628642c95bf9ecf271699e57a3921c1d3fc6a462e96cdd80ac8444e22f72866da1c7e726555dcc569d6084bb24640929911fa1181eb41ebedefaba6ec5a9e5cb00ac03e3c3a94ef1ac1513e22af41d65ac2565d965c866c781bd704bb1283eeef157edf8a83c0e3c19a85de2e8aa3ed2ae8e01f30a02d27abbf8f5fb37650758919b36c6baded52eb17484e2a2878580c95c028a3ab6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275237075Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d, Name: checkpoints/4480/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4480/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=77814e02f9709766bcba9069e4776d8a0a42a76df75b1a895d8cf2656a752924324364413267562949b17baae9827288fe7859b8d80af13844a65c6988337c3cff745ac8533b0dbfa68f3954e4a70641abe90e173cbdbcce14e5ce8ce2630813ec3ccef5796c82925e72539a52c6cb0d46c8926a01c765f56b0b81837807b39360adbbdfd34ec8bb2e91d340b666a4bd273c6c1d72cbe007eac9493a7615e84b7e60ba948e9280984601230189311452549065700a4858d2080d3e5ff3ae2f28845f182f53c4c6558d0f9a830c545e3bcc49ec5067d948e8c9320c1e4b53be648e660673df06a41c20d9ebef1a622f8bfa024ff1e4839dcf425429f49352fc12&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275280504Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/array_metadatas, Name: checkpoints/5568/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5568/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7cbe8209693f89423ed17f7f9a8976365da3ee183409a1054495196a1faeb1c15e829f59befa6367fd09274911bdac9c69d2c84ec26026eb29e6e43f96c064b6f2eb4a8fc8f4e1d7ee89de7cad5c0102e2a82dd7eddd1b5586ae76dab57cd489498817a58d0610b613bf1d9ba8f57bac6f45efc19e3b751627ded257365113095b365642b15e0868f36245845b52ef6040f43d9006de25136024bf0607bca24d49c13369f2808a8959358dcf6c8fa9241459f1ec492353aee88f54c73a4c356636b75126265dcd954f0586730f5623c5e9ce5327f2a7d077fa379c394a708ee7b92c9946c3ef6bacd26dc3825a0574b60aa76426a9072877a0e9dc656a59669d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275292318Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d, Name: checkpoints/896/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/896/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=674d05f61b8f256697ef99467d6bbff99b33c729763f2dda2d0b0b0c2e2095624d09e20a47b591921dbf9d1200c8f4f3a3c2441465db0ac48d5df549f53685048f6e4e57a376a16a783dd1bec895cbeedd2f9450ff8bcdfce90786feeb29b4cc959f428a85983214a6fa52b05963c03e77221cc858e742e406ab8839e53c6ab662b307d9d3f3b3b50c32771aa1cfcb42ccfff072286b53701784809f2bbf3c5ba3025b4526271745fe3f5600deeecc7eea775ff722daece16a4e8793253e04fd99711be9a46b42e3ab7b17c1019f97aea5a005434411d11fa1eba2f8562abd63f9bbe7ec185d8a01d404edeb4ea9c9b79251b8cb534007fd6f6cb61b66a35b94&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275362296Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default, Name: checkpoints/1728/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1728/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a620975d967458a9f572bc4f7c52cdfeadbb96a0b148a2ada75e5f60e234c6fae1757739416b5d5cae3b078e8467bb85fd0cdc52b67b152827d75d524ec5213674e934e87d4b7ef572687efedf65cb0a6a594b5b041486fca6e9d49e1e837aa8a5b51e813d1d33f613cdbfa1b44fc7b2292aef431817a5119d2a744b141f7ab553d563297b1010358fd44c03de891c5e8fa89126d19e9a4d2543ea19288f2cff0f8c16855dc12d28becfb5edc7a161f78baf343aa1857934d3482fab519e6523b34acad5d3e7546b7f1cb36f149b6269da5ece64c045d8e793b7c0bf32a97ce3c6af2a776cfcc3e5ba30f10c22193d273d7db4ea5790db00ea7a467cbedae7f9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275370585Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default, Name: checkpoints/0/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/0/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=70c364e67b2a6966a2fe323149084d179e8ef8106e5720c0c7e055e1aa69263f4a6cfdcad2f324dfc53f0655c823a29969c73811cd7cecb7cfd6fcc1b580519ed9165a3d3a654262320652f199fd10afe738559fc836a816f41b543262652b514bf22a9aff60e0fdc0c17b88d1b2583f947e449619e6ebe44811e94fa37a7d146d7bbb66d76e8c9f705f165305b26b35d2143386d94ae5d067bb93450cf1923aaa5639baf189b0f7b5b6bc883e6ad37573abf332d1fc241afbee430ca413a58a46fede0e1c4d93840da58b57f092f689626600a24daf4f07fc0a1cc2b73be124eca6f8dc00cbee2f3648c51064663603945e40ab136f868bf218cb84fe33f1a0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275386434Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/array_metadatas, Name: checkpoints/2496/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2496/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=702a13846f6edf768ca0058969c40e5930b99c11dcd542456bebc94e432b123f2c044cf0a42e9292f1b64d634bc046e8c9a6630de364b7f2f80a0985824fa8663f6addbae1beac33ca07b47af17e751adf86c604e9d718ea5c7808a1894798ba023b8e8cae5b5db198815c3e9b739a7651436fc3cef525bb63a5aef7a539c08fbcc57f667a461c7236c05405123b978a9f8697d052579ea01f5844337078980d88444bf89e18fc368d341596a04282ed67d7f12152c0c9d67d338c5827a5d1bb384168d41388e6bb4d6eb482f9b5137c8579d9a9f1279123e7bcec0344ce5347b9754660fb177478ef1553597e8f49e6286b2ba6c36d1d9d9d8b45de75ca4b2d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275454668Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368, Name: checkpoints/2368, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2368?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8a79e76c3e88e6dff27f48a27aafccc1796483bab2b77c9be6b2992fee84d6575958f3f2ae8c0cbb074728fc60bb1eb3d26d35f06bcd18ca1c919c38c3c42077cec17005ba05e1aa2de7f0e782bb6fc35e8ba54a0dbfa617490fd970a44ba07e755aef781d4e532bb4cd7a697c75156bd4f70576e2f089bd63bd9fc9b5d69cedb5f05ed58e034fea159650a0b07c545e5a21dc2b9e66c4c84cc6f5b05fc8c998d4041cca0783ea066501eaadfb5f738bf96977e80158ef9d6b41a4d2ffb4c09fc1e28f160405705d467d2c43855f185fb7321b9e382ee860be94de83ab07d5f0183505614687a65be9d8a2638f70d66f9a1bcf65688e22aac3ad223d8dda8ff3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275445697Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/d, Name: checkpoints/1024/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1024/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=78bfdd46f73253aba37554615e88e223f2e2ac083dd324b2665cb1840b67e69b4fbfe2d3ee6b49356d00bcd4e6adc3fc32789864f17ac003b6958d1acbd7f2bd0e246f0ac5f6941b2860ac00a113eb742d9b9df7305bc41e3fddfc160e5ba34455f01ab3c268b4f5bb9a97ecee9b13a1d55eb8d66430b581869394ac386c5648ffa4257d530c751f847eb4fded2fed8efd904f1985025c7c3d729073ba4f1d46dabba3db2637c718928c2623d48c29e22d4b89b5b40deff05cb9b84851ef040a0e80805c4cdd892846c07ad3a39b59222674c64e49700490732e8edd69101bd84ea9f39a8fda531190bf162182151efdf0988970d243c339d89fb4eb8fc8fdc3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275501928Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d, Name: checkpoints/2304/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2304/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a403b354fc64c935ebbd3d4c31a0a88871ab24dff81c4505b745d9b714bae64633a1b52f71271aa36453d7789870364e9e8b600f5acd1b5f3cd54c00b7de8bd9efea42a90f26650d450d67ea283bec760ed7cdc065de46f311611506c3a9a7fbb29e58aa0de4239a67cc1470a4039efa265d01ec93faac5876df9b8edff12109b9697949f034b42891259e61b5af094d2739d2371352351c9dd5ba984f79cf7c09fd6ff82d558828c0bb5ec9b347c45aca2ab83433ac688b1f7facfb4cde6bcb0d4f29e7bbc1cf897de331a59b3e3ed380d1360e420a2158edf1efb6632679e6154b2a182b96ddacf35ac7aa99b38b2409719d047eb2fb1592468486b707ec28&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275536461Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d, Name: checkpoints/576/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/576/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2f69b8975ef523b66ba8bfd850baf1a73c72bed763ad4f957e24fa0479169e5f25fda130094b00f161fea385f89bea283f76f517f933bab94a4ae4b4fc4454c22895350c9f9640b4e7df7276b681e9c3c9ea545ac9fb70666a332e6cf4607764aee5af03064cf52bf314a70f41f1dfe9dbdad86fd3e3fd7a364eb03f4dca4c166508ba69cc9d807bec548d39c6df6394f3859a4a8165afeaebb60028c9b323658327444b1ab1b4ddfaa4594de4b604218914d26194031c58cea02688d51444a1f5d78acf4577be01f588481a882ebdac24e3d8b82001fbf71b77a438280eeee9871c27e0d6b676f223507061068daa19e7ca4c0fdfff590c3b99bb91eb1a0ca4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275539196Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/d, Name: checkpoints/1408/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1408/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=785542b21d971f978fb68e03ef6d003fb14f7c50fde7f7eb22a4535f8bd4259e1d6ef6225de8c919131d7e5b7fcbdc90fed31da7625cab005f5691aa4d7d35a59140e249096697fe068a3db403d04b10603994e356f70cbce2917dba80dc6bce8b254fd2ae512fe8296501aa089a4fc93a05903dc4278730165507619c3bb08972bf7ce69e52167ed887180ac755abd003f0a25b41dd84d65315e60b375cdba85f6f24217724461ae6cb369db6e50aaefd36907e16bc85b0cbcbb72543d0e6848b4b21af357c2ae6ba0f28a7d4710e7d89a1a2ff053216bf417b459c91f4863517ff6d275c6b603269f86e51a20b7ef53bccebf4f7899c5cd2620dda6fc0c1ae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275535279Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/array_metadatas, Name: checkpoints/4992/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4992/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4c00298c0d7928c003eb417cb23b94a7cef0721e36c8abc26c67f5bfb672f3dbd7f945903ced474a89d3fcfb119a00c448caa14e2f544aa95b01dd452ca85eb44515877d8b1a2e716893f94d0f56ff6d18939ed98ee52770636883a74804f445861ee0e06a129dad71a42b9def0067e28807a0d06e58a6942fe87ee97e56db41b3d1e4ea42087eaef060ea8bf75e5449a2e4cddcb995ca4b55e4cbedb3baba4700859bf2617cbb030f68d2138660cd547f143ec843871b715e98bc87d1b3a8da58e5339360dbac626ad8f43484874b9f11564e58163b6699d7c547f755d9a999850593b9365cf41b92c2ff9a965b75cbbd8d055589156774bd5915049994fd21&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275402335Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/array_metadatas, Name: checkpoints/3072/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3072/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a1a503f28b2ee6a706ad3c821a57b47fa7c81fe60ef5d8a8f8b480c0ecc47eddceaab99f5a1cc9604cadaa1c107e481fa5d3f62261525beebe1005a7dece155246979fe14f7689ed052f9c9d4bd0e531753b7e72a7e94672565503b95af0baf074c71d4a06a7d4c13b9d650c1220c3c9aa3faf2da00b06359cdb3420357fb0a88848e8897bc7df2b13ed99c013815ee2fec58bad3e36f2af98ce5a610c8fb30a9c9239832ee04b62c4d2b3b0c7e0992c973427b830d2649c3024c9fb63620a114a365d05349a5e2773be99a3b27a15a3dccb2bb4e61205dc8fb116dcce44c3390d39e432e1331813df9d5ca2360ccba1ef96dc702162279f01acf8a42c5d9723&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275224196Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d, Name: checkpoints/832/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/832/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3e2f1820c3deddd284116ac791ab85bdda5e65f3f8e4ec79ba62de385c05940ea069eceb12a4a38c38bf19ab8803d8a761aa013357697398cc29540816635baad9c5c7edfcce35dc1cff28710cc937bed023371d2b143bbd1f93771345aad316848211fc4f27c8b03a452c7cff1d89219537750db44b17b5c9da47e5595093aa0fc2470e1aff09c0861b54952629fad6f9ef2bc0c39f4d89fbd11d5348ee7393efac508dd65050a1a92af64a4db31c45a2c70a5a15f5fe292499ceab5a13da3be3eaacd48955b6d980c9b9b1686ca80d63d942082840f96fb95704104cd5635729b8a1ff48f45ded020731df8c1b74bd1a146131b95b153ea4de234b4c6e5e84&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275687825Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0, Name: checkpoints/4928/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4928/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=97bf34b3ab436d5ab9017b783da951d29cee80f8ab312076efa7f0e3efae2035f0d7e6a8542d71641e8b999a4f78df7fbb904f727188fb91aae523b2072fd95a9de877515982887386f6b8f36b32ab83645e7aa7e1bf47441ee23e4f0fee8e07ba3cb71fdf2046d98bad573d9860e5e55a0c82791538ff6016544895f26ea455dd1414e41cf83b9bc315671a9559356680013806f19c5f159f5a007f4f4db2f1efe610b13ab4a90098304c0e494ff0c0749db620a91b19b6bf1cffee9cc420866db6c0dd71fddd2325363fadea2b204d5d146e1b0e4b6ca7ee3b2ce1a8bd1c550f161db1a48b9acdf07887cddd491ec8f5c68d557aa9ee08e9367bcfe4217bea&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275067942Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/d, Name: checkpoints/1984/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1984/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=243ac2180075eb0e706138208295d6281dcbbd82455784ac2fee0e1d3b9775780230364ae11ba32380f5bd8ed10092d99432608eb61f126a979cac3fdcd24dbb71132c26c8f9933940b447d409d643e65abd5bcff2f214cbb1358d3ee269a048923b08f01ece336910ab0bdd0985a0dad22a1177018e9b733a8d818e4e5815d3d28675def51015977c991c14124b3e49230c26aaebd3c8725c2c039892ef9fe7316361b86b91aa25c5be9a6f02163198d53b127bde95959a8d5a30b1e7dfc55bc5c1be96edc515e48a011e96b2f422f34641a92e098e79e38e928262613f28f1125586709ff8efb0f44fa4caa61ad2a446d2e0a39fc204ee61d9d79d7c843ec6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.274970436Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/array_metadatas, Name: checkpoints/1984/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1984/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=401e06ff75837d877c9efe835d2816071ff360a1f7277b52729af5eaefffa89acad59308bde32ba4093b686a81ba5eca77e0c3554270406940061e84f7c1a3b26cbc68b66625a4538e7e48c057a63c98df7e28e70f1ff6738b5caa7c4c97ff9b5fba58a9537eb9a9783bf633afcbf56a8764fb04db0cf61c0a8160b5476e512d5c034f7bd5f74b6cfe62e3d8ad5922b5b9c18a630c608b15f8c0c03312c64037fb9a0d46e924be11011a107e7c7bb306c4a073bda80d4f3d749e39e2d10fb456327cdb1ab66fe8f6368091d62b975695dd729fda3466567c01c0d6d0bf5dcb8c26cf7b65e70a0728dde2258d55c2ce88b25587fc2decc625d47b4e4f19d9374c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275112609Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/array_metadatas, Name: checkpoints/1408/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1408/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=056644fdb6aaa7f7aff823188527196c27ff906858e9a67124fa20e4a0e50a70788ed95f893ec9158be540f8f3b9e319ca6e383e2395b42f9e17e4f53c912335cb721cd38a2f950c47f072e594c37b7836babf5a9fb3e5a9ae885abbd3d5758023e535d34bd4278ee4930a6c0e7921daf291d2b3e4081c6a05bdaa9306794e4c2409bcabe146debc56412b0f44eeb3628cebdefa184118868766b1be906c44b4efacf82fa16089154311fed0905c13f52e27ebef2f8b65109ce904141c3a31792da78664ff69dc3b324538d47c9d377479b880b555f9156b2be040943ea01a0b35c77a3fcacc2dd7aa49663dbfb64f64a4502e5cb1bf61e20f982c2bf9a3b2eb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276599448Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default, Name: checkpoints/4928/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4928/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9166a33ecc90c74996b1f6a9d85db5a4bc5bbc8d4487013ac8bcb8235e60c69fdaf902489291b4f7092728490175c6057055f7c4c992e7e3a71c7c7ee67aea313f81e771394b27edf8dc428eba5640e0b12c82e9fc3b7cd0088c543172b5abc0c0a2bd3d4cecb17a378a96897eae2c91748da813e81e0f195b64712302838e907251a6fa19a7eef0dbbe04af1b3970bc60994bbe0932f7dfd003b1c755ae3915b0ab1a032053d5276bf05b895ac749e38cf2e898fb548e802e8c0b3fe95852a1153f5a31a1ef265216f37a02285a7bf64b34b6ac90f207214064bf927dceea3d4e8059fbfedf2641ee49ef443ff00ff7733c3a4f5d1152c3ca01e89688c4c055&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275078345Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0, Name: checkpoints/5824/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5824/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=97022a88f331e364b4388f487924138ba05e2cc0e5007549b8c3893cdc586450e6ef66d5ddd09bcf97c71efb0f9c03e688de82aee894f7db0430f34c076aec59d098b18df2bc9f57bd3679c2af549d4eb4d54c36477b232b1d2f52a33d96b2cb6dfbae3d1a8f39d9ff2bd0f7ca1b0b03e71fd832baa4ef17a3d6215f331fc325ef7c429b43c565cdf656435b288143cb956d9f89ab643612c57cfd04bbd527e5d093655c3d59fa264f304f9268d54cc7baa34a4226a96632d30619fd2d44f5c56a165b70f3fa2f5b6d6bd53ed0f46b7f1e473400a559ec56dd45982fddc01527e4472ebca4f64a24d74bc22a939ba145384742cd2b830619abcac0802ada2f3d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275713863Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/d, Name: checkpoints/2560/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2560/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0dae24d7dcf30dc06ff02e5c970ad9c21de925207c9fd2bccff0847de6d776da27a5aa1c61d352ec988125fca03a3fb59e6b150335d83555f9f8bef258ec63ef33908f7627b6741d19e0c62ce510fb61724153fe2d99cd553a45f40a9a520153f00b82c21f3698bbe7a14bfb69404cf35e90e40d6b2f301eae0c699177018124a6f0b747e700401ae350e608a5801e36bcb21db85da3fc175633a54c99d08522a37b23e85c173d9204ee9e29333e6ece0512da32725e475c47fab99e508649b250e26721f78e0872d07c610a9a7b10bf38d6b0e4640956eb20dd39aa15396d817438524a5b677c2ccfeab15c4ba2d86456370a3a8824e1fb3864457d1e3d9372&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276602742Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/array_metadatas, Name: checkpoints/1600/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1600/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=541c48aee4e3d06245b9138271b28b3232e3ff8caa4748b7a9c926c72c4451a60c2321fe2939f54a6e0453db0c7feb6ade330fc809a3db37f217621e828d9640206dcd1326d2c7e8deea1eae5325be23f39679a04950b2b389046245ab5960ad191c274d3f4fcc85e10bbc3187a8ba71cfbb7f08f48b6f30c5eb7723be9fa13b215077d9c3b8fe78c8796a977e5d8fde702b9b2fd3439a00a19e6a6e21a4fad726d514e3fa026c7d5ff67dfd8989844752199566359dea820c34faf60b3eb85d945a4d5ee173dbcf7932caa642359b61eb0b487e057ef56d1c67505c567aa429fdb8012503b06f7e360dae21d55af98f92b35919c501cb3df426f2d5a3dd11fd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275209453Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0, Name: checkpoints/128/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/128/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3d7249060bf496e948978f728a208fa06821df592ed5b6233ce5c991baa44fc5ad807a586839ff5847ba2738a6a30eee649438e56160a1f28c61076884c36fbfa42298b0da8153a4d79bb074b4fe9d7ce0ecac9b636777c3942f6b5fff035f8bc225f579f5df17118538d9d53fed983d1dec053fe575fcb90efaad30bce0984924df037b24036214d053ea41cf2f7b1d0d99b7bdeddf667df2ecbd595383b4e5b353fe00f2f2cdc99b8ea02389ef6451585d882c36deef277c47110749578950e894db0b34f34536b161af3b14d57294480da5c5bc101078da824421b19e90e0018ab4a471d1925482322c5457eb25abdc62067a64d6cf9e079a3dc44a77659e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275533681Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408, Name: checkpoints/1408, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1408?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9c5e3aee5d6c4a95b875e36a4d71569312f5e274db06a4a7b39070c4037937ef99931ef091e1340f988bf3d3247706e61472da39f4d12f7c7fdb24a4b6db4598a9060dbcc43e144a473495586185958d16b140d365c44b7d69f5733e2db98f1f49361d7900b553944aee953610f08203fc220010286311980ec73349aaa5269a4d20d58629ad4b2c3d987a08f905edfc336d05f6c77ff0ead936808ccddc578b991e6a55788275aa25dd2e66beadb387141a81e7f84712686ed6243d4967240f22c6ea9544767f455438f903a976c2ee70f4825d974679d90d346665d35bb8ff00ad31df013528c9408841442d404faa049d4406119a901a023c80a60383cae2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276697269Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/d, Name: checkpoints/2624/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2624/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=086fdb8b9294c7b1532a9691d88b6a7a90029f79299107a25e26bb81ad61160b7c71e8e621eebe86949b15eacc75a674e0bd2d20f0bd1bfdc34d800a9ea572ec2bdef2df53583235e55c8535fa47534bef88d989c298ebb0f3f16094673056737f8673586d19e9a51aca9759056184fe6771248911f39e981a8fada150af32a644feba828b5a0c0302169442036a20e93d021de333cf08ddcf02c019df2cd403bd647634964dd2d4b77ca79e6c3a78166202a3fdc7b0e131ae8b0058da10940bcbb8c02d949c302ef2c8f736abab62c8588bc9222310ede4fcb29b0dd57a3047b0d7256ce8f434cecc2340ee444c31a3b7317a70b67b10cd5747960544c18221&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276708981Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/d, Name: checkpoints/768/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/768/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a951525086203db2b4a6d7001bf8c11418f699268c03b94358c3e5a46377b5635671fd546dcfd3a59a624ff690425b9148815524cfec941577bde6d169d453266df03f6021ea639f7876aa434724e98daf3b260a2e2b6a9e3272fe150c501a4be2248a7f5b1f3506a237deac53eeb2ca1aa4044dfb7551ea7cbec1914189cdf4eb8ddb862592a3ca7d9ab195be6a815620b4d9318d866396439415d21e33d98f2192b376be17ec9cf23ab97454c53b342dcae75b273a8ab29a54bd243b8e7f6217454f3dbbe0b04f1f82b41566927f8dbf325548b7ccc2f6ff149a8267c3e08c8ccf56f5117377b80532dc795b37ddefa24004599c51af8135935455835e741b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276690952Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0, Name: checkpoints/5248/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5248/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a3527cd6748516df34acc9ccddd91b3602338e549b18eb3fc7f634426ac7fc96d01742f37b57cff3380469fed141cd7cf43d48267006df027532cc4a4c064e72815eba5b7dd439cc77dadf321081572311cc6dadb3cbc4bba7fdc22a2d5b991c58c9468c95f61d2903534b546b3b018280153702597cc80d4b17e1b41add9cde2024793a62938ca647f3149a6b3eb2be671fb8100781ddc41d55f140cb7a71dcdc91eee95d650496b21914d93165b86c07856a1111cac3cb9c7f367f0e9e9a42c321924a250b816f2f91760ccccf0ba04838f36b65fce57e9fee605a90e66f94f3b45c6c14d5071b9eb6a44b7c80882bc875745ba8004b098626c99394c957ce&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276754854Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default, Name: checkpoints/2368/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2368/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=984db9b99c7c7c75546f5b037abdf4ccc19497fbb422f0d93aabeb95a807c519037da77bb5d342b582824e08e0c1af1cfa1aa254a2c9836764b8e218653b8aa013336292aea6ca8fce856d32d4facf41e96ca8170738e23f101725a9661e836792eef39f4b7d8aaa7b66559c25fd59be05f62958f9d07d2241052af169a2122803ca4d0481cb28a76065da984ca7e0c0f16dceccd28ce80b341508936f1c782de22543e2922f7b0ba6d6eb262e33224df192242d0bc23ac88bc2a2d9237d9cccbe2f83595bff8ea00df4e65f47075275cb03a8032506b25c3e219b78804e4f13dd0b75c22ad05788a7a41941e4e7a5d7b139c57d1781dd480480b3851b5cb40a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27503091Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/d, Name: checkpoints/3456/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3456/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=08019263496789041313d79db6a29c1d085948c0bedbdbcc5183245f8e7a425104e7c770ea011b9937668a6e5d2c92ba3e2cddf94a5615c7814214e63d917fc02308f4a14509ca8fde1a404ca2e0a6d4f15cb613459c7e107137a330a7a5d283e6e6f434805cb315186dd9a2a87920d6e20b1c3811de45ee25d38d41124bfb6981bb5f1748392a3964381b9ddf61681b81c5bd66302e42cc10929423622950e3f3cb4e1fd4bcef4438a8868e8129873c63e083b144348a7527984d68ec5a7553095357773939b3d0268e6e7b215c0f93309c81498687e9eec3c317f977bd0e2ade31a004677dea1c0c4e61809ed15804af6d715f03c155e6175c47b96a31eaa7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276809833Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/d, Name: checkpoints/384/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/384/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=ba57372d428e0c06776196256ee06c344913d2b64879dd8c65eedc48b3320c015a6bc74e7adcf336f4ead594296a9af7a41aad4bc1e84b5defb5b2231115d899128a19b3d7f12ccc3721c4140adfd762211d94359983af8674cc60b09d5b4147121391f44556297122fc9d0c201c3c2c07dc648ae47536e369d7c6433c6c98a55cfcb9b9fa336ba9191b25fdedd443bfbcb474f2217a445501f4c470d9e932e98d07f78b9aa69cac0b1ca7bf48644a793b64fbd09c8afb9d9b56fa6420a48fd0f2dc70edf3a0ad3da376708213e7f67b390d7131b3a3cc72d793abebaae9aa926d55be2428fa443a1d23c7b0eaa2852cd410621724c3df8ca2e04297d3a12de3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276840457Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/array_metadatas, Name: checkpoints/5504/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5504/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a9454fe463378c90588f03601a447bc631337e3a1c910aed023d979dc64c6e2fee28af91df3ab289b9ce286920b72eef6a3b1805da7bfa1fef93b86362d258245ac0b76f8fa0a0d6d71762425e36a35c0490b5905cca8ef63df889c07e6901f7a64d845d161ceb7bc379f4aef3e95eb3a1731ca549af9ee56218e95707e7b9ffce95bcc07c9b54fe2bc41a4f8a7f524976ff91b9fdd06aaa6f7afb85d4587e687a8d83dd866af70098022bd25eb8a61999a9c420c772d759f0e2840653cafd97c8c4fe4f65bed9096752fee1027ca2543f286c88caff5371913799de4c6074cd8627be33b84321143fbe978544cc1df97b78ee8c51d6387fa67eb39f081aeb7f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276738246Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256, Name: checkpoints/256, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/256?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9a753eccd55e914ee4961744c124c6d6cd47b87a5c4079461e083bd5925d9b97332b4a4d7be8d6356098f33e851f2bf51470680642c66cc45c82adf0ba7694f2bd52686246ed1cd63666664ee4bda69d02277a1189cc7dd0ea5d28ce8ce8c845d450b4fe5ccb949a1479099bc5765e7098710e5c75d122c22107b673a54191bb5af2bafe3243856aedb4f2efd9573a07aa107becbf4a4f6a67139d25f27dc4a8663842ea17a755ba0aba9098bd0f60762d1a2185f03d4601bc999c1b0c90ea12fc8f366ad36acf510248249477211fa95e564bca014cef5bebbef74dd676521398aa153cffe2856d82eb0b30a603281883bb97cd30d0d95f97392b621dccab4c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276890954Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0, Name: checkpoints/5888/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5888/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0a6c808a86fa9262815337bd6aa579019f07d46b5f784b6ba61cf04c9fd71561d06752efd0a486b11723faf1995049782297b0bdcf28dce8f388a17646c076b4544de90106c9df8e40f9ae5f4d1dbca1d1b4d1a93bf13b1ebd5407716a2280e3f92c25267b190cbaf065b9e3f94f85da2a5c2bba779f001cbc179f6d647e9293cc6092740c71475ea15e9e848b6c298fecd689b3144ca2d732db282a8b58a70bfadea1ec0c32239c6cf4e704da2bc7d1cd364f46b52f050efec0bae53ef09dfa73d72375af6aa9312b686475fb35d47b69b959581643e880de19f936cb0c4852efca0058431da3a6adfbe33def6c94baec9e9e2aa461a171c86e9dc092f93c8c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276908372Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0, Name: checkpoints/384/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/384/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0b17021f00248773708b9c4e76c411a38297e21fd008609dce452741ea0d12a90e8ed8744d87039213053a410b3c9e286afd16d6df4c953124d98924d25ee8b35d85aaf3339b48095d36a832434ac6b144b8b122d969d50c17140c701a1ecf183137f5eab2c3605864810a0ff7c7843a811e957e8d55afa99e99bef9d33315393586e0121df48df2717525de3a9003ccbbc1308a2ee61a85242b0a6309148be014e41ce9413863388cf8d2a7e95f51aa8386e4b6ccacd2c41b50e1b812283701064c5f4fbc15481145fdf36da2672db9455b5408c0b2c11be15113fa2608ea8cf21107f672e0071fcb593ab956b5ea64fedecb45f7fee2645e22963ace27e505&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276920072Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968, Name: checkpoints/3968, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3968?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=62973a1063f2d04d44928770801bb71c5af554792c54fd52a4572f1765c5db588ff5f6e13447ea34c7d645b01e9f88bd9aad15499458cc2d45669473fe6eba82514c4e0dce8e42dec57b489e93278d577d0cadad6702e6080363176151b37ce05c026806355465829e0f99044fda456b2425fc6af8300a00b07af5f7d723563164d7177b9105613ae8c9710e5519c75a2132a1fb5e6d022649cd63959149d340240868e50e938fd539476e77ad1880eca1136d2170c8268be23886dccdc92f5639a9ab465889a38e6b03399ef262431b048f9e696db60bb57ba42cf83434350a27ae64fc59d2e9a4789646a7e5713a338863c407a67130fef1396c1145313d1b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27692435Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480, Name: checkpoints/4480, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4480?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a3cf4be2de63555378abdd465845250dd903ba2cfa5a34aaac263e2c627d491668768b1f47af8a5bc5a1acc574a55cba69419696c84126c5394dbf4ae5afd5cd2a9c193b289a3976aa57bb49bfb553974feb3a06c708938e99f414c35c51350ad5d744fc325a8ad23b10524622d97299ecd67430cf9719f01e98b827d32bb333d65c217df5a1204a5b203b10668c8efc8375c35c4a4a95d6a4b800bb4eeddc70095ab956da2a1f9e49b0de7f9da3c73ebbcb63c283785a01fed97c8a268d11d664e8d967eb41fdf9a708c009a9922a9d4e8db8e4a4173553259ffc7eaa4983bcc905199379ddc9866f5034dda70c4011c0cf6d8a521bf28ddff8d18c215605e4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276928121Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d, Name: checkpoints/1920/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1920/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=31a34d9ac2624f2775aa0098255c1e06e14f65c4b4679b8f3dda37e4aad78e968a6c72df4bcfb3b7e6c028131c0a921571f85b3e8e2e5eb804d81f0300d4c8c8daefbf1e4715a6fca917eb1ba44bee1e69e46f2397702bd1e2d82a5a467e649b29c1d1daba9a429182b1eb426564362bd767309176dd588c6813772400bd93b0537b32053101d2a78becb221e48a194241280e3c092d02e576429d96aac48043cd690515b736fa11196d275b25927f080ac99cba11a9e1e498dd45fe8bcdc72d6b90969636821d22be058722e46160bcf898707d1f9e88b6a280c858c76553735985983f00980683134d2eec97abf4b546d36bd357b24c4e7ac69eeb410a8ed1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276812899Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default, Name: checkpoints/5632/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5632/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7a61f39562dfe8dd44efea3ba429a4ec3ba6cd1ab5cdb37a7a76df5b5a4ec34229be092e3bdef8560dd427bae50d80a173e99a3f78d2aa73c84336bf9398d8c3f6d225cfe4d12a184da35ae6222a22e7ed0015ca77d103077d55b30ce7c407bef93e6ced8e8e8908fec606194745308b7b624d8e6e1a66e17971869e57eadbdfcf0fe5add161e63f54f9e0d1d764495c111e82519fedb4db23fa345c5ce7c5729782214ad388bede6d7b3e1109085087ac7015ffb500c4185dc70cfaa5349da2e531bb57aa49bc19b1ef7ad803b33b92a79613423be148e7fe0f85ccba6b8f5c3ad2ce6104863456415f31a85f60c9dbe9123fea4a25299d69bef3302e2f9167&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277003773Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024, Name: checkpoints/1024, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1024?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=245893cf0c02e448bf5c066d798a2ab66bd3ec17736286d10008e3e37117967d9f9e7fa93030c755f66c0ce352383d3b483b4f23641d200226cf881a3374f7f1a605c38d08346276259f9f012937bcd01ac7928b9c83b43be945f6157dc54b3f5de80d5a85a8d30ebae668f059a406d21154d2d170eef28fff17963d22f58a3c0dc1b6a5354b4099d02b31b3a21ded572039494bc6097b19d0535ce5a12b9c3f01e21f308f793aba431857a8a3f3186332814e89dad5987904e1058d1cf8575d6933acb321960b85dfc759455bb636557f4cf543060dc2657c726ef6c3924d984f540cb894fb1c4889e6021caddb6a0fd15963d6297c936e125c888047d5dd45&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276842896Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560, Name: checkpoints/2560, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2560?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=87ddab72ce9073d466d0bae0e069c63ed294cd101cbdcdbebd642850ab8def0b3593c48fae7b4a930c11524a32932fa6ff70bb6ac9987ccd4d15721c510812d4a8e548b5c810d33a23da05ae25dc27b77966061d89f4906fca1a6e9200778f5cadf001710b1f9f0043eea06c30343010bdbd12480770cf2f8ed5479ef869b276af541e49a2c8ac4dc88c31e9b92ab909aa88c7eda5d6fa8be20090d76b98f2d01849c2192ddec05554bedb7b0ac5f120f7d09daae8ae14ce25d1333b821bfd85f5ea88e08bfa36e79d7fa38a4b5d4ab8d1abb2ebec1d8c31a4e59dcb1c0de32cd12a8d3d2ad69146469b6422a6cab10189581111511a1a1abf15e1120d63f583&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277032424Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216, Name: checkpoints/1216, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1216?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=771220f6ace9b20d761846387c01090cd82be1b54bcbea013166e7a1f898a18cfad1e55c8b3649d61cac345222768d7121559a8dc296d05a83a613f70f5a4d18144fa9af66b0b981f5c79abefb6690e946a37a24dc9b7b7b48ce5c49c70764c549db6310a760ed168b28425a054aa1c55a3d0b963d498d8e60b344b8720e1c636392762f231a1dc883d541b07c403dcd1500a032ba219acf2b122e2aeb3f044752bd8f72a999d8a9ef14bf35f77e4b4554b672dfdd68359b3b76991d0bbf7df5658e2004735490b99c03a98c374a0a7dddeb44c0f0f94b96ea74102cf7c94bfe72c125810fab0051039c8082dce58487f90b29ae8f83f90dffff41fe494e6d9b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277049765Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d, Name: checkpoints/2432/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2432/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3a5d24c6d1cf42c7744a5697f32c1b46304f239f83731216c91153804df8a304be5e62875fd8688dcc0f62988ba69e4de46e92deca18e901b83d34b675831e560cbc833fd1fca66539e49bb8d0c5c362b37eac5720d6a01eb27cb872815233b548f133ac98bcc92a73a592449330bd0d2ca325f750d054e98f5f268433e62c643113233cad0f1f189f2382dd2851b9ec5018b7ac78a939dda75ec679565e10868ab6b22f4b4f5c76cb8a4042dd5acbbd3fb04ffe7d55ac4d16ac5ace0c37764a8ec287d0227605cacb992772019837776e30b0e580f5e37ebbfd18bf06c5ea685b538938ad0f4fb65508388706ed4620974fba41512603d84e39fbed7d98057a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277053038Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d, Name: checkpoints/3072/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3072/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1a7ce69eefee70256097f58c0817c8dea8049311c2ffa9518e6181d263f63861631654fe4c1c17acfff93119d0627ed41d4b9ed8beee220e6332308600a856ffa7898ab033da9c550f2f5ce7da872c3a1b26b4398ab601df09769dccd9a5936d0f88ae8ff59f9b306517bdde161534a1842270b62a03f78faa09ff034d0008ceb54d76d1be3316dc803db3721e7ed02e77fa2cc13d85dae0cde486d4dd08b3e6fdb4ddac5341bf8996e46d72d0432c081e3671a16ffe978dbcee594c4eb9ab53e4a29ef6d2d53df66294c595eb59982b106f819c94a489412bfdcbfe6c56385ac2150648639a5e3190c0c20fe5d7940526c7c7a73c49f54ab92d3636fad383d8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275136834Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/array_metadatas, Name: checkpoints/704/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/704/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=54034b4d83016e23bc0698978b660e21ab14fc51334c4b7d9f53f78e33bde22b7de456283fdac6aad8258b4a31c54e5041cdb0bebd02e9e2882c0b7f2946d8a79f628373a3e0c74a4c40e7d1baa4533465418d81ca7eea5e86f08da23a84f46cf3c495f72c70df364ea6a5a9fa94ca5a8fbb1d51290fddbf96fca23768b2b6f0cbdc54456be9a634f8af9b359ea15dcdf5fd4d9472050831ce7b924a9bd1a740c275aa6b471c8d55916a7a8973ccdf95e1cb4b6b73fc33b2a25fc805589338925a0b111dc8ca682dfe153578df31d6feda34c1c7a85e7feafcb925e6265f15dabe182da15ece38a0c199c09b5c9709b4447e473e66b5e5c3fe68ae557b55cec0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277052755Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d, Name: checkpoints/1344/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1344/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8d446286df1cfc3e13d5ff2e6a8368eea054452ec2d0891f46f986e9838bb03d69a8920dba9da4da1fac068101a596f6a71ce1220bcc8d660fca3625d90855fb9173ee10f95f3f9c0968dbc4c5eda171a366ef6ec10f0e6364440780b1290519b3b6859b518f93163e4666b56eeffc23a9beb5acb5444e817fba50a763fc07293f1ea813241b7f537d9121b231ecbb6b9263f9fc500b094c6eaf95de7944e0bf6dc8fc26fe4f01fe9b6da32b547b7b0c9135a9f59af8c3de1baea0170b9bf5a2a055a950d6755c624bb6eba94982eaf40bbdc6d04f9205b1cd2dc513648c44d4114d9f9a07b7aeec5c6da6f0e4d967447c21c7deee91d683c9a1606aa72a0ec4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277139527Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default, Name: checkpoints/896/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/896/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=14eed89e99b936d6de8e10ed038073a9f35b1567698079994a373c7a108c73b975d527c07ca9ffdfb39ade94611303cd70513a985aac06bf0833e50bff8100fb993cd78822c094b6ec0872f198e1b97909ae4eaa28039e3168151de27205faa123e44d72bff368a4a19b869edfc7663369c9e845e2eaccb59f2298374e83a8d6d7148c3649377346422248b8f22200e417d18a2d762baa2dbcb13bf3c7afba81e1966d08e536decc2e9c6c9414f7d8b949f718c7ae1c47f5302d03a0bd1a6bf18c9d3cbbb3e82acb6d8a69406e1f0b6c2a80d9cc45341d02c81813b5f9f85da670e4e2aedcfc2564903be35b2d97ce92a50668dbd2ec1bfca2612d9b9e2f56ea&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277020467Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/d, Name: checkpoints/704/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/704/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=33003ff6e9f1fccbc3c3ab0dd3622908caff4b819abe8d6a184e18cc116b7bc5ea4550193b96f1efd87b6af1d72ec510d174d2e966f05a3c3fa895ae8501519b363ee593171d921d38e73b89896fdf283feab09091226c2ac11a55f717618c432635f03e5ffa1ee9b06e645ab467104b86121736ec9aa07c8f27f0ed8423ac0b9fecddf40bb276b8636edf79d7084447eb5ad33cfb26a878ee2ebf24e1a321a6b23ac03d3884a54dfb8320c49698a3ce3859ccea2b1673dfc7156a161740d980b9f66b7a30ebdeb8c0b0bdbf5a443bba28685fc02f67f1f9ad78d84c8d734066c1c760adddf28ee542f24558c9704d20f4ef6179e6dbab45010cc6d4b8e58cce&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277147851Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/array_metadatas, Name: checkpoints/512/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/512/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=07a0c69374dcbec1b434ce10d484922e4305263970637b3f2af0e11230ea1d508360ae34abee58c71098d9c5f592f3e9a7ac5b0a6bc1d63b0f281d4ddb3eb829d8bdd02c935d1824154a7615e23ea3c6328f009cdde8eb6acec80ac8c5f5bcc8ff1f39f3c9d5ea67bccf6c22e24f3c495ab072b5e83901a69d0b6c911bf4f766e756431070833965ac7a0f0cba50aa1be97db4fbf8e660af3b01e06cf06646b6aa7dea0806a3994f2a81c23e859e35ca70b5f513a9fdff0c5b499b7534e4fe11ce1b92487f2dbbe6790d72b8a7067b5b8b7f854c28508bb9dd2e2421c24914ceca5dbedbe2b9209430c6d8535fdedfb3f6980d316a22eb9b44c94f4aecd74644&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277182817Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512, Name: checkpoints/512, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/512?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=222694d515c6fad904d0d934b2e3c321dc0446082bc6c0e7c4adbbdc74a2fdecfb430af96436e84e342e4b36c30c899c0dc14b95022180c1e4284332f17f72bd7b27046cdc386a9ddb57e4abdce226eb1491cc04cece0b862dd977b09ca691bd9f598dc230a1e9a66b47976f2c9dbdfc6202ac11e3dacd23b2a354b77f5012090c9a17a53d0e793c35cadbbd286cb882e6fc7b42ba6988e9a479d9306d791da05ee3ab385392c9aab8739c01ef35d8534245719a7e05b2dfebd14277b39cd306a372fa65c2f034d478e436cd8b341ce4af02407f5cd2a60aa8a1009abce8701a67e4831cd8afba46730fcbd834b0edc0f8baab601ec5e381db08008a6bb66ae5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276664853Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0, Name: checkpoints/3776/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3776/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=794afa0fb46b6ececce91f068e25ed0528b75ee6c2f5218c64967e60794cd7ded3d6496d2d6c8fbf5d79604f611392ea14c5d697ad4b9f78a9b53e2ca785c82e980337544d3610a40ae17f466357901f06e79838644befb2f6144f60fd734c46165788ca154072f3752e5ca3ea38abf86b19f28e8886e5ba215aada313766b1c4b82fe2e5a0d42608088ab4570193c60f8b9b9b68a34c456c8ee9b192f12f48088b3ffaae5edcd9b9d6276954b7ac4c6900ff3af51e5ae85d80601463c0c34dfba0e2ba82027cfb3e25619b45fe3ac0c02143ce9f87666bb68b0fd74e02036b4bd16fb8bb2e96b0614ada3bd081ed8cec4637ffa3461edfdeb74ee9f41eb1bf7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275118696Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default, Name: checkpoints/1536/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1536/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2baed1202144af81ccfe7da24cd689102e349ec506d3724b6ca98e18fb71278be28c2b2fbcb2aaa2b575cd36a939b71eea56a0a47c7b2bef38273020c6bd82964bf296619269dd5c5a290da93b5ec277a5abcfedf1399365905b31d78fe0831357b89d2d7a6e102aadf13ceda145d194466be0a7ca13614d266567dfb700a4b497714a88bd29948cf5bbfa6ddaeae8beaf6bf5a5ff2d67263e8440f6ab2eabd2d446d87eabb23ed641200f00cb35adcb8297b4f6ecc880547eaca7dc94e91efa4345f9746f2108360cb8628c8cc970933eb29f9f0d5e719b1466cefbb44e4c8a32013b30e7de385115e190b6e932c7c466caf692f4a9bef402b7fdb36abfd0e3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275112242Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default, Name: checkpoints/3328/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3328/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=11a0c2fe4007b1b999743153ea3858f9d3c2160e97dbd819b1907ae200674ce2c2390ea3ef8f7d89c108f47a45d90b3da423b0c8e100912c90d37f4d715f6c7074e3ddeba43508a92588e25b7d109d175eabd8856e2ef1b0c3f43fab3387358d3a2d622dfa4c0f093a5698fd1775ca6da0da8469fa043ca40fdc2a8dc008150c673a4986eb26bd60283b447c57fcb00ba30b9026dc9af1bfea92887908946cca2ff2161f7340fb90ff5c9e146e29032c760a42a0f0831ea1a561991398379cea8b98cdf1a03cdeee03af1827f1ec6af9f9f1766b63079336cc37bed3b691a60f2ac7262506098989cbd6bf29489b6d9d87f93dea3ac4c3df5b066e35edf14729&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275468473Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0, Name: checkpoints/5952/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5952/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=91e689f2958b63450ff217fe5ab22003bd7eebb0bf7bd7aa726eced3297bea1e9a0979eac7381c567f36f258aa952960d2cac554298c8fed9ddef4049ad109384beebb73a82f774e7bd399b4870a9514cd5efe5db2095d509eb2053ddafbc2d57f9c116d823cbc5c002305d68698848b4c9c344b7eff5e8b3fe2c09cbe5cb4eef3a96fb96933ea1368e28485352678b4f7ca18d26d5f966b05c35faff757e882366ad2f0d8f781fdecca2713ac27df41e5673e806ba81a9665bcab31eed38edba33c7351aba729224404199ee0e50a6ca5e1e6610e89c8660f5f93a27ce45e95f007904fcbc31199f03ff141b855ab38145127e92675162a3f6c25b895106f36&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277213894Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default, Name: checkpoints/4736/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4736/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0c8529ffcf9baeb30745fa710d261412991426ef9001814861a5951be0fc8ce52238026d5188b5da90f06339b578a4e4429924c94e168f66b340004fce90854f0d99f00a766956ba0f621d1f58d1d45efafc692c13c935c16465a2e16221e35106160ac14aee2ea909149079810689a7877db23e0c2fc62a884feb51f79444aacdb1fee06af38d0fc2c81a86cbf03aadeb85985739414f415af2154f594166bc4c0626780b32d3d05a687610a5769e8b91ee70df1a89a477552ac603c13c8419ef468175c0a572c796c002ad9b174ee386293faf0989eb83d556bc25ba8de91e759431f8510175275fe40528d9df81cad4c9dfeeb9ff906509e919de91d474ae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27515004Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0, Name: checkpoints/640/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/640/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a2ed1b2787571ae2af1e409b569ed04493c2da26425c69729371ff0f8aede3c5468c07c9cdfa884ec720e669178e42be71a79376e8b9dd89092688edb478c20be11f76f0cdbe72597bdf7852b2f6a0acdc92b5c87f6cf71f615f5887ccfe8cf3622b75e5eb13483fbf8b667f97c25d451405ebc75b892567ed2b1d930395a93804b1071691c269da8cd1c807839245f610c0a5ce7802ce87631495b43e74284e7cae8bec2538145ee2e28ddad2728153daf23cab828f22a822afcf6a9418af97d5c130bd0414f895e7a1bbbd501af47edd168f00f7bcd9d9b3b0df79207e0575b333f30746c2811ae43033ea17ba53a49fe8a59148ee632ba2136a8142775a74&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275012718Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default, Name: checkpoints/3712/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3712/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3334b1f16c8c6443044d498f1ff9d96276ed4fcd94e5907cce320807fe4b2514513a3bb89c01cd5d83d126401a7e9fbaa6847c39eb4bb16fdeb906bbd17efba9ea82c49ccd23f99e2433f329c7307496646e766f2d4cea6d767bd5ea057c1e13cf1cd18eac54c1dd798dac25b2da4190a71a60489ede8f3c51a6d1f5fd541c730d8ffd4315d8d57bdc32539fb63d4da246566617afa5167d6f6021bfd2189b4a61cb9403ee8c36f2aae7651b12c2c3e4edb9dda5db7988842ad13f14fa071dba1060b50046c1319f347905da5d0ed496fd97a5b81fe4a40b43f965170e799ddd533bf05daff0ed7e46686ae57c776d74ab3f0fc21c4b863627865c7f7b0c07e6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275209926Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/array_metadatas, Name: checkpoints/3776/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3776/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=79112074e2292173da61f8fbf94ceb584b3cca2e3fa363ca05a170bc227be0b53e9538d09a15dd58f006a9e4d5bb842696819b317c08c239f3aec65069680ed12f2c75930776a204283513b28fbfc3ddd711d781b3ed63d59845ca9901f219c60a188782405e1fd336521a1dd8b4861a47f39326baad9404ceaff60e397722529ac77df30f6f5d3bb30bcb58e081356e74f97e9a3c0e759b01f086cb125995d0d2207377b34fd01500535f4c49cee9c6f93fe046772eb9af8cc743eb0278931b8a476b32d274c29d80abdb8b97a660a2bc25b0165c210fd084080f775c7700688caf59d402ff8c0427f18e59610448f482d6d518660b8f5e49d1fa883904d4c4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275293768Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/array_metadatas, Name: checkpoints/2176/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2176/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3409be322ad241e7df613daac97113a8dcf262407004bb00878aefd76c0980a2e1d8197a21e63d5566dd917babf4da032391409a8ca1451a19a9ffe692295f8c61548f4c1829735dcc7409e33931bf03f93242edf61791f24f3e37f0cd2c32886108498efa1e0a9094b4b803710d19aa227333544ffd0b43894f93b3dced1160a3aecd386b596346d2bab1bbec4599d82ee99277b50c20f5cf6ca9d328669355ac44c29edea4883d126eb1a7ad6c6522377dad3ea2a2b0f6db583e721bccaadab0526fc283607f73b451fc0ab3903c57290086b66986ecbce16d815470df277e11037f356eb901869157b12bc4d3d2183604b66a4161ffc48b08e78dd4406229&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275319126Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default, Name: checkpoints/6080/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6080/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6651d7417a9d57801581e869c25bb005eb070b6999f7502f759a232c27e6cbe50cbfa90a94c77072e3381d17c3ed42dcf2f3a0d762f238d74cf3f722d82dc77933fc8cd16c77793c2c3bfb37ee6517148143214a623a98e656afa640ab26a5a6c748e408383ece01c090472bd64fb567bf6787ae9360316b37a59cc44e7aace62442bb34a5a76d7399b61d770c35439f30144502c0a492f92b0552fe5afea7cf7fbc49e86f495f168130827c5b38b5efbeff00f8d08b21b5b693041e8dde52528ae9bfff3529d450e7698d6af48c51ad075d167b04ed1047e19b2bf13beb4f31f42b00fd49b3b815e43700214f4962baea542ef50febb388bd0503c44eb49951&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275401628Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/d, Name: checkpoints/5248/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5248/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=02ab19b4efbd6560d73d307121df81a661334c20104c922f2aa79a646cf5d141fb75e5ed80eb75e3f795d599f61e47bf0f8a9d83ecf2aa695fda24df74403232d1e3df32a705fcca5c8d05a1af016851aad52343294ce95bf1a0b3fc4c2ac51d5b2e079914f298ddd968308508fa6df1d03f741331adab3efed946d27cb45795171a522d9de3b7b28882b043c752dfbae3c1d3c199a5453eede7407aaf29b0fa047a99fdc3f567d3a1ce477120aa47705a15351321c851f90302cf4e86990c4ff1cde8b77bd2077e7b07dd836438aadc357384e4c23cacd1e4cc2f78a34fc8d3ef60aa1f564fc086ba0f95036c225d85c327360434c7d286a14563a571e2e18e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27567293Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/d, Name: checkpoints/2176/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2176/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7847bc2cd9c123931ac01e04bdfed5cc1d51c61e2163b60eb1b7ee8dd4e21b8afbb37873ff0e62ec8a6b7caf7097fe9a011969c568c85b362b03d4957d56f05e6d3ff2502c3df3af0498a75c3bf701982993acf0ed317a55037dcfa31a5de9a48b096fa53c88fde92f14d5f5f4fa255437e1d02687ad4b8ca8ff9c99f843ef336f55c5fa62cd470d4a104a79b48c0bca889526fdf5183527441183f2d347f0014a4511b55655187066283813b8a279caf43fb0f4eab2ec45e4015468a41fcde96caa343046df2a4206ea9033872096b7b2d4b82ae204c9dcfa68f0c78f16659ddb640dc3cebc9a8490234e517c87e20bd2f408d0c5cbb615c587a96d01c40dce&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276631813Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/array_metadatas, Name: checkpoints/3456/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3456/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3446611bab83e5517700a3979d558ffc41001aa59f145bf41c67f964b7e6fe487f9e66592eb1049d2bcfe5988d637bc6cc0ed15772b1ed44915c531f810f085a38a5fffb21a421f39cb323e6c5d37373d53c478f58ffebef8264cedc57753de906829cb3049e47ecab7f26844c310b3c219551767c9ec7d17564c49814f41114b64cbe3553247da42efefa439d975f938005f523cbd5540743788734816053967e63d37d41a9ce4c633fc29597cf227f4fbc27f95d1a69d4fd43a14050fc1c9df89555e3de6947b512da8522ab6162a1357e4551ca5070487f5ca4714a057526b3668380f6a39fdf93575825da81d020c19eda25e71df5697455863b7b29dc02&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276626431Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d, Name: checkpoints/2240/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2240/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1621b842bf6036b171e21de82421291ee9afd553ba85aa4a59f5f77e5b5019c9dadc020b48881e7de60eb38057d6ea4060e2c8b53e7487a781b8d22a1655c337362509e64d2ae20fcc3ca8b85268e0861da7de68e67a921fbbf6ab0dbe274e4d68f84ad85429e943e9fb7ebbc05c2e6ad7bd7e73839af2258d0d659ccbc542fd4de5228cc96cc8d49381e4aaefff42484b4c2ee4e689e11bb6c7efef1ea26e7baf5c737d92292655f571cd69bbf4d9d30e8f7969296c05c4e976f391266d0b879a412f3c45913ad6a18be979efaa6f32d046d2d0eff69d9b88e224b5b78a8c680a779bb3e4dfb602d0785eb7594c8679c3415281fa77f7cc23f27334d36a08be&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277361517Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/array_metadatas, Name: checkpoints/1536/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1536/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3977291699776975e7f6beafe46fa05dafa108f601148de5bae2c790f3e75f681e241fccff8ca1739bbfaf661f72f8f44b828eb10c5e2f142f15e05f61329aae17f132d8bf65dfaa1fa9f0818700e80d16742d8c8dbb3d7af12784a3a4baa72ff53f7abef18ccd55ea0534406a04428370a48412036bd4b3b21334433de4831a956a0c99c34bfe570bd2be65a4b671b263da3b4018ea7b21caed164c8d42e04a8dba8cf1f3784fd55cf358dce0e23e11751544dd4148384480cd4d94259139edee3b31422bc77ab61a4e2014f18715aabcf46e66ebbd689850ed023b74f12ed92e4b254609e18f3a2faca7cb775f84c5704701d4bb4f7641093c1234568af127&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27660575Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d, Name: checkpoints/1088/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1088/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=74225ad9e3901973b93c9e2b85d6716523048153493a64f1b03c54d8f323be12be0852e1958ac0ad77a86906ea8ffd38515b2b51e00a8eb46a028f71e78da25c16076801b97c948e65bdcc4c46909047ef271f2a918e7c32e8c2c354b0a0482b02e9c58a1216bb13f0c0309fe5c8924fce9fc777b7ab2c5267fbdf11de16089f806e9287423ac805544ca9906f83e73339743a14c289e32b257f6727ced18004a77394e169fcbcee1779db56c63976489775b09ef5c8021fccf5a18fe5f5dd3ebc45cd0846bb0f5867fedd4e3e0651e7a5ed159b92461c1d21c7ecc760e9948eb177156aeab304a1d994e8114680aeee7219093bd220e19fceba7163bd283ed1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277357515Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/array_metadatas, Name: checkpoints/1856/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1856/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=329c796a34904c7b8c3c9a3104dc39a97fbb5ef2f03d706a1bc6d5713eb7fecccaaf240f287cfaa4aeb62577d32fdade680434f1af0fb61172b1dd4671508ba00d24a65b2d9417568f252c6c0004bbdda06588d15bfbabb78a7683d53cae67ae71fb12cf8f9a62984d6922284d78d5f20e554ef367473761e8213bd0155d5210e2ff493c39bbb806669ddc5b106f5e46408439129c15de220f6ae63fa561a105e3019ff5905203cf5e739e6dab23e584dc1258aaab105366c451fb8c92f6149441067d79f5837161c11a62aacb85be021603672e6b9675f2af9382172a27f90dd01f273d299c805b3d176743d073230546c5c1b2ea9408d7afb66165e52684d6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275005626Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0, Name: checkpoints/3456/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3456/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1711f32326b771042b572520e4a1a4de8a8bfaa8839d67d53baf77b8dc545abc4f878c3a1ab2cd20a91d7a3dc90f97b82c6d3e43eddd3d685dd30923614ce16ed35449dbb2584bfb96d67827d24a92dbb5b84c9df76cac5e30911a2fdb5462f21c837c315011bdd4a168d4f6c08cf293e11511574e57bb083540c514b5dda7fada2fa514addd28c4fec0fa895adbe3e141e47916f88531f4ff0869b5c584907d41ce395d4db1a1e9776973ac20d255fadd89da71791578c5165365c02ef2d155dc0b2984af53aa30aa4bb7f26cba917d798919709daf891d76a3c6a82982a6b6eb2a923c28273b23635a0d443469850e621e94efecbbc359a84eca29c9d2b7ff&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27665742Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688, Name: checkpoints/2688, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2688?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1341bdc1cfb34c55213352c0dfed8993dd723845ee5a265983f69198085fa7bea47ae20064b978e8c630524277e400e90e9e56e98472d11eca666f07ef78b7723277812c9c565ba1d5e7da1b035ad4d9a2bc80dfcb5dda3b16b424c1d9c6da8eb11c33467f4776e375b5b6434d73a638a06060b77bf8afe9d93d41431ecec747f64ba208e7d787d04387117695ecaa147fc84d67f36e5d46c9f86a10a84e299933c70c821ce24655b8e38d1a810ac7b1c57f6fea0d65269763952732be4adbdd93a309f5288d9a4501cf0292f999e818e6ba7ea1e7b56981b10587d372b046d90dcd65cb819f85e2cd6287201f1571b0f42dd7ebe69ff656ca9a8c366a089ca8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277399578Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/array_metadatas, Name: checkpoints/1344/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1344/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6e9ed9a41cb6462ea68c787777a1f4ac7c0bcc7e50be82b8ce61139821d4e4dcf5316db2e87c546d24a716b41313ce2bb20216eb73abaa963339910df48ceb3d0057e1f38f8071eddb178d370c9c8df34127608787fdbf10434a6ccf734b924594c73acf91a4dabe9714b57b447d2d301b3ae8d676d004004faa04034027b0476eb5e4648cb048d951fd4eb84ce32058a8da3789c93b034dbc35bdf7fc349c854578d22a1704874d8d6148340eb82911f6a1eaafaa6f85791bd1411c1203d19f6a076a79b74f39f21229283d5b5a540d42e46dfb15ebed457b5e68a068fc40e60917ffe24f2ad03d42214738c298fedf011ddd17ac049f99bb8095299a3e8702&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275312328Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0, Name: checkpoints/1664/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1664/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=48dd1b96a52b77db1c56cb17aa95c6c021de3378e238125f81f076acae557efd97b89e7f6075df4db145a60cc7f84629df73ed77a037ad4ee3d238570e5d50a440fe4cf8623bb4128153f1c3e1726e8f8720c6519d12e01443478a434619710a0621a4621bfe8c69eaf6183fd95b6ac073df7bd3c8bfab0000e779b984d7e1922162373b40f13e6d3854160dc27a0a13ed67ab464b2d8528111b519d655926a23df22cd0e245ef0aaef45d530ffaffa332e4e4919f241857747d2cca03fa2c0490b0e164461ef94516919eeb0537172e1de9747511b0928fb76dd976a1bbfa95b02b95f1c01eb2da1996acb46a5c301a884ab3dcadaa693946399d9140fca4a3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276656689Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240, Name: checkpoints/2240, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2240?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=658d73cdef2eccbb75de1e26fabec16e001d3f8743519cf21eba19668c996ad56c67753f05fb94c298d688ad4317b11c0de86368796075191cacf85c3352c6fc752ecc7fe2fee9ea2ab9a67798a282c74b0b4d56312ee79fa337a12d074b3c52ae859db6779a3085750a5b52aec33a703fe872512f0615251ba6ce8f527f22dc43bde1a9e9432d010b2ca9cac9fdb76a8430ac632f48f9c94390e898b834130e4ecf2541b97c97975a04fcf78d08fb2157c08676967eba89c05a348f324b04c30f9dd4321416a9c87c1cc861830f8a9bb69261e8b1efc4c36ababf2acf036bc39155cfa456927f18ac9cc463aec2322764478e37afde1e9197b180867015bb11&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275076717Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default, Name: checkpoints/2816/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2816/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3e33e1c49da095843dcd75b4bb267f5ac48adcee7dda19038bafbc908cf952854be60afb20d8bb011afd09076b11e74736ff39ef1b4a07c6d557f83433af0fea9148086ac07732d9ab4dc9434f9061b43bd561f6424fbfad766f337d3d22518dd3798a0ebc3eb44794d0bea37c443539b7546b1804d7b704d6d3ff0acecc94493941f1a23bd083de68e0279d6c1c55561186335f15034a61e8eeb6dd57d5280117ed858824fc9905a5ddcb6e12e8a1ff3386c12e170b935d6cccaca1ce250f0c399dff97aecdff9dc8173e8fe913979a9fdf76e9b0582ed2ce5a2b07440e3b5085e5fa9bf0d4d705f83037369ba1afbec33d1ea9418c4abf67902fb1dd7812ad&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277502254Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/d, Name: checkpoints/3904/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3904/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=442577d6b4c415cf6fa55d704939520f2afba771b2dc00113da9fa881893a790a8010fb19a909fe9227b96324ec7ec1eb80aac6f84669e92c97a4554a4a948465ceba6bd6b99d659136af8280f7b028067a17f5c57e7b317623094fff6e299cb05565085c8b4151e4632d435708c91026c88801eda8cc09a75f21380122562de5e71d04711e0dd622043f52147ea7ec1ef4f363f2b00e7e59cf98298b26e3e504fb246b9c28e9c04d51e2c7e4e5c8181e3bd918f64ce143293c4c31fe385979bba22bee62cfd230ca3929c739891ff8ed4889b61ab7bf2a8bf23eaf7c34eb139b56e8c77f6b91dd5d7069aeaff24be4563660597f5bc24a9ef23796a52858290&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27529099Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128, Name: checkpoints/128, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/128?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4f0d0d15e8e726237f0ef698f2fa74cb6ec0273d4feaa34184d7b06c124818d65eb9ee67698895b98c7435284585f750c66e5d1d8c51fd02cf000878daa91bcef6357d31d51e6a4d3c2e5279d3ca49805fdd7737f2e9795385c1a494897498c870a41cd0070000b8dc47477b11d4311b04d774d8030c6c117f6565dc0786899d1d3fbc9f375561fe11a4eb347e39beac21de3823e25b142fd1024f8049afbc9d9d4afaa8d843acb8a42f694f85dc9030629d26349064941ebaa182837746949d01c8a1ff9299cc2b7ef868256673a960fe391ba7b2bac1147875cd5971a89b9041c4ad2c1ebbd68235010a7fa205bc85b296f65886790f1fae623b0fb512e970&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27665992Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/array_metadatas, Name: checkpoints/2944/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2944/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7e812d12f7a1abe090167ee7414a652090cf98c7744a2e7833cfbb5e5509572f43d18e881ea75e387fd8e0b0a2b4d8c757c12aade39878bd3f6d35cdd5407b1f7b4bddac0a8b4f1a4ce919779c3ca8d79d201ef6c651be2ba35af3a1b33f3a9a72fda86bc82f5be1b663e82389127546c8fecfa7bbe1906290facd4914680c9e78d5187ed0aac6747659efff11d1cebbd7eaf5ff381d9e34d82533af44c2a150bfc54c6674f23b4824d320d98c9249f3d7447e83f25dfe8eef56bc400220c6d41307e265051c8f379f4d42ea338ee9848956bf821a946564b5db7a60c8692850aa40a915acc8d63fdd79cfaeaf85cd191cb5adb17c63b2034d1c85c4afaf70c8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275605258Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d, Name: checkpoints/5440/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5440/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=86ab8f53038f38cd135fe0bf8c160078a6f0d3430a156979a9d17c8df3f9b3a3b9bf1af352b091241d1e0ffffb7eda3d3d8ab16e8547f037ad7eac9cc1f531da18cb1500534ebae994e34c993d62af178af5e6312481a51d5a8a60e3e1b2c548f2dc4f9c0591894a574174ee0487b74e2a51937fbbc0827aa839e43a277fcee4dadbf76a7f3806b91e0ef151561a67fdb74b254cd6155edc2f26196f809bd02d3ae320c7cea647901bc22b82460fb5faf1c92ea1fbeb671fab65fc59795fa628c633fad5ffe6655f11d788dc35f83b4325cb97090fca1a6964b09a37e231affe4a56567f1a3ba588ae075dd46e589fb9a3e6816f7793d9815390c91dfb2b9839&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276732016Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d, Name: checkpoints/3584/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3584/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=baae405f778fff7ad56f795fdb2b87a0ed176c11ce1384eb138914d408828a040ea1b14e33226c12104d5d4296b8bb67a9471578cebb49f86d33a61ae7813d39fca5afbe9228c480a74e016d1c66dd248e25ff7baffffc393877509f11d09614c971166a03fd0a5a5f279144209148ec7ca802b81c8be3aa74e9954fa7ac858dadaefe53792e9e9e9008befab8c18c9ee8d43fd862dbc11a4403f60b79ccca91f95b52c22fc14b05ed903c7db8ee41d0a513784e57d1e88a01cbd3bb3106d0c08a7d75987832fc8e293523d2c21a33ef55a24d5c3b1eac451f2ce87e88ae38807d733a492511b1a64978301bbba6c34b1b3e8e6629d090aab98a0c2f3f8deb43&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277624414Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/array_metadatas, Name: checkpoints/3200/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3200/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=916170304f1fa5c28138a3996d2aa9ece1884162315cf3b2efddebd78b19586fadb6ac42aad897221499de8c0baeaab726c56f393eb2f70b3711aef1fba9e0b7cca3fd30f3090143ff0c6e4c9dcbe546cea0f4cbe88c89680ad2aa615be5b98daad2579015aa0bbc1b9482cda83c1e046641f682a12b2b039ec9868d6c9f1a57e064fffa38e59c53331119c4916dcdb5f5fba8cefbeec50c178fd98e9c6186babd18757f69861c3dccac818d7f325925c7c3becbc8a1d5a12f4aa67660ca7403ef73fdff16926fb8e4f317e475fad6ac87b8c5cae48a3eef3b91855ee1f559712d25c60d5ee6f2b86e6374fe763d103179484028d0ac8dc92cbcb198f7353420&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276944876Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/array_metadatas, Name: checkpoints/5824/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5824/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=494d84d8088ab8b2763f2f9a2f982556d2535bd4b7a46f48cf04969254f7dc0b1c86ef1dad486450770c0c30e1827e6d97bcc1b2d07e6eb03e6fb4544a2c50d61400bb0a217595cbab5c53446bac0df035c0af90ad8098b9c9bf789e2ee2596ded3700215ddc91f4164bbe4ad869a4c8b966deda00371fc4daedea7d90c4b90fa8bd9552366c90d26c2063e859cfc0c68de6f545100e6b7f16cb99125b7479f3b7978a4a52af0ae45816c4a54c3fbc4eb0759744095b550d28385e78a5c562f4908c5d6344e2e45226d16543e61d95e8a99611c1e1551dbdfcc41e7e0e234237c733e71a6dbfc5d00f8b60bd8b7e6877d5f4e0a1c4d2141bb8e730f8aeaabdd0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27694067Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/d, Name: checkpoints/4672/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4672/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1fc4d74eb4d209b67b46324857570ef1be26a79ab03e9086debdc3d78dc5c8dbc312ea0d4b97f82950df3e52793d3f05953ac9f1385c86c76733729b8b2474ebcfa11bfec593a71fbe27b75898230896631bcf33c797da8e0c2f9b292d4ff3966315c08896bae0b3feb63011a3593c1f765aed2f2e66f94d4b5fb9cfc7bf19bee12fde48ec58a18ddcb2409e6672786e431cce46f2be935911f01a0c8e3987283fce77bcede0c5f9911ab4e42f326ec5ef620bb670423a1d9ce1336efc77e5fd4b5ae2e8d3663a114bf7d7bd82f85af2fb794607ded4cd20a04f259034df78b1a0e26aa1a1a6d80a5df3f4b0e7587906de55d8f078731e8b1eded35d5f1c9758&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276925824Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default, Name: checkpoints/3136/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3136/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=aa40f06a9503354e062d047bb653fbaa970040607e00b172bd62708bed0dbfce4d06b67649b1438127ece7e003b917f4fb49470aeb3100cf4a1f576c2e1519d13627b90c4e7036be3ec818e5c204b1d39efb5ab01c912aa0963fe86859d0a88dcb0ac19273b487910b000583f876d456fcbcbd801401fdc07da7fdcba11404a4d2115885c9d940a4cc2141387e2ab8caff4bc7f0425d8b51c09d554fe6bdf04cc632fe8944bea57a4e66ae54453c22236207b7965a17bb9e99ff62b533b9f1831dfafdce7c751287c52d4817b7047164519a0b1b24be55a99c56025c90849baccec87c18a849598b222985107a8e3bfc567df166333fd8b53ee04feec78af2c2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277687843Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/array_metadatas, Name: checkpoints/3904/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3904/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5d78749a139224f5ddc9cbfde0f571d11aad9d84b7383ab3de06eb98a718d37b432da98744e33457eb902a2a2ca4817a9e1eecbc101d5aa87458fa0d8bb1ad346aee2a53d15591fa9a880579b8c1da5f1be30c260f06dbfde6c560190e58882ccb196b6f31edf2e3465e430d0fe795baf9abbea544ad4f90249105a224f079ad351cdc838e2eb0c14b982397ff7364a2230e7eef9ca34e568b54efe762860a60dd343ca9cc3b29385e54d8cdfaf57658f6158ac40027df040403dddb8ee9aebfbef21876391ebdfec8ef771574be1af70a780d193e9af22e6f0cdd3f80d4648d9b6474d5d5af3f157611bad7845539912b5cfc1a16101a9f3e26cebad9c54cd7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277697588Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/array_metadatas, Name: checkpoints/1216/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1216/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9710612ca98c1dcf51eacd12abec57e54b9dc42557952e6da64dfc138241f149652e6e10cf42cdf50537254c6d2396f2c4f069d2c8398ea098f976f52fc852faf502bca77b3cf506ac0726ba77bffc062b657b4bf94b98ea0718d8da84caa4742d47f2b825db8932b521352c5feff0d92f8eca240638899669d66adca0e9f334a5c3ffdcd35d1b787361652029e84cbb83bb70391d89e84bdcb47adfc48276328e0e3b6400ef27d53d8d03352ec146f9a09e83d776f1809add6209b944f94ae870f240f790cdc925e07e70a8df59206e1675ade3996314cecbf6cb26aff1664b0ac9ec49d54bd40c94472bbd8c613aadef6d2a150b3ddef6607f6b449344e85b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277730394Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704, Name: checkpoints/704, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/704?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2bdbee9391c7ad22a0b29d3c6aee61fc9ce5a72437e0a6d87fcd2a8be1f3fd84e80fc18d0fd300dd0c828a11d6e8460e70d0d3701e40d7a41a9c3c3e517139a51542995ee234648ceba68258bd832b5ba638927c9017c17d91ed92886f1404b000330fe638aa221583d028555f8377a90a42c21cf5a893c8a8c25d7352baadc2fa99c382333ae57d09fa2fbf3b0a60f19c7fac5367d0866627f932d18939bf7d8f36cd491ce672a4b6a9e7e49229c673917a6a4c8d17bb4492fdb3572f55c6e3accbfdeb31484a092293fbe01e50a615648ec7358af1502ea56a4995b406f704ef77f66fefd47c64376a933e607bf24a14f6c9098ef02000d7d4ffe2612d838a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277743365Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952, Name: checkpoints/5952, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5952?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=31ca06210739ebe120b5e96e20a998ea4a291dd19f6c46b5f87e16e5f303b3e8b4acad57e03f27e23ee41cfa08f8d1638a3bb12553902a3dfef083cc93aa3dd6a03393095e018ee5d3cda433e36395ae031b4c75224241d46f7ce3972cb39eeed81563629602be6ccc668b0b7cc1ae4663ffc844a1506f87c6af44cb1d9a31b8c2a579dc30d50591540c69ef1127f7fe430623921a899f9d14793c7b16b415ad352b799157d2c0c3b5ec3b79fb36530b66179ba466cfcbf0eb0d3b8714a5aff019444c31f66fc14de2d4b3bb46f1160ff3178bb92f78647ead494ea4880c356c438f4282615688045de47c67d7b073c0f0dea97717f3905eb6f5d7e3061f0196&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276846103Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d, Name: checkpoints/5952/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5952/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0c881bd07199c16ed49bc8913ef822fc22b61eef990fe4f7a0585cdf820baa5fa53761d139fa0fda8c2483f4b385ad16a244bb69b313d3d609f56af5b8c7b97b00251cd2978b13f04c1d7f28df71a6b0273b593fb0913253782d7f976bcaf494aee064b8a86ed3dfc6ddcf24cbb0d8c559c7f516c6993ac56d2b00cea6907ad024267326b65efe956ca075bcf61436ebe8f211136d98a837ec69b2ebbcfe23c1080ff37bb590aca5c6964648c4056da09c97325c399d9085148a71eb1ec9055d15eb2045d3a37fadab243ee68d751375abdb944a8ddaf677a61f00312ed04098546f9a8b326d3513ee02d3d6f2b6e94e5c69665ae07acd1ff1613e6b5a8133dc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27697767Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960, Name: checkpoints/960, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/960?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5959937cb2e2f19eca4097cfa8d3ce0ca270b21242be466d9dd955b6a407d263f666294703444ac75fb46e2257ef7242d272e88b7eafa22d6c438d2c6371853d7e9d760b1e1689520494f0f43f27f4d9a1096a7a51c57612684b78dc6e463c8922c26c495dc3ac7e4ec87b6df50dfa4269b7544a3309aafefabba07621fcc91e70afcd473dae5c0a6724e82a9b99a79382c189668628a575788f303508464fdc5bb443c7958f77f851c1f3d6c9b2fb32a1aa90effbc52b2850207fd60462e8574510e8035f1e0bf693b54cc6cbbc4478e6c58772e89362662323f43b34190327b16da737568852696c7a9674c03503d57146b8ad19389fff6e0bd18362004782&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276956494Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default, Name: checkpoints/2944/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2944/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c11a0d930c28707292956b6082d960599041a7a95a8475d34ff531de20b30ccf0014c29004d4aa352b9b0d2ec96124219fe7451fbbbb4a05f9aa9fbcdeabf1f29dfd0ea3a266e9f9ba7eb1be6a1d1ab2b8eddcb3a8d2291ad0998db824097d2a68fa7e1efaad556280fd2e14c8b44d9366822f72868b01b7ab5e41d3dca6102f299bfeaec0d46163cd3437540dbe7a5bf340e730e89ec8e7d114a64da900d06924e2168734b07ce0f4a34eb1622ff40def841c742cef7f0491d21bce9395a1b3c082ac06d4009c66ea2b16969b00bd8328b0b8fa51510beba1960af26b3faa9a490d6072df9460fb70a1b12f0d8fc1b80c7219cc9f3b896d114cee8f408a7216&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277021083Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/array_metadatas, Name: checkpoints/5376/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5376/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=795d12806ac06bbd6f8ecad82d3ab3e8019e7ff2c9629eb4a7f4a6e7149088235dd25a0e4b0396a83b20133cd3f0396effc1e14cc76e3c5611828071693d39ad8057702c70e0b1cf75fa5dd73be24a714f15e0e23ed6d0878d988034c43c8098813b5272139673a7e8046459c9b00bc1dc547557b6463de5f093e79696549b5362c3f6e90a02b49de7b1819660e22edb77ef2da91140918c88ad2d081cc4504b7326c5354c69a98076fda2194283cbdb0a7ab6d61cb40e0e826636d3dc65fce9366271e3cc49aa149aad28be10f748264632895c16c8e89fafdcf955cadbb2d1da44f9df744f9bba4251bc48bc5db84476e8965c3a265fbcf6525e8acb35f3a5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277052797Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/d, Name: checkpoints/1856/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1856/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6abb85c76c02f175e67f503dbdc9f4437bf5ef4d0df42308363c631b2a1d935ff5a2a1ee2efad1acddcc2bfe02d981e49241d4ce59f7df7ca09fb4c603b2068f7234c92ff8f838cc67c7e26314a7f5ac58caea273e93e39b2772eb1b2fb918f4a75a06016582d682d14f3ae05b916765a573bd662ee904190cff9103e5c2a38ce7e1b98c80203767d1631a3b9f6b3028d35415a70d4c8f63ae558b8aebedc4426839227e64d8db919ab3192dbb0390773436fa50343014f7c27844c5f83367d163da134b57edd87f8621bc212b0d480c247529ecb21547c81f8b735847245783c025b2ef28e41f4d85ef03aaa842194bddc6416c0d040a3275ea1aa93f0e5ee5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277794674Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0, Name: checkpoints/448/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/448/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=940b1640fea8176d024fda41ef250f856c69ee92df4592a0def037e81a28671f6ca093895645e8d36692108c9446e1c51c64172b01e7ff41a2e0c9a712d5f034fd9ad03b0c8202fbd8823a5b318c60ed8100e10bb966537927d9c82e802a6f29ec3d3291d2801cdeff81b0e9f59e4cda64e433eb4905fb3b6b6c0b20d33fa648b7aa034c411ddf81bbf7213e14d08618edb1a80ba488c94282f3781db9d623cf63af0cbb4e9bb81b3e5457cb2a4080337eec43b4cb179995934a27b872de5f5bc25a0389c91041ecc53728c98e5af19ab3b01f97d306eb5bbf7837d69d1508a1062d209886488f8e4affd7e9412ba3d73dcc1ee63dce623910a742075f31119a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277036395Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608, Name: checkpoints/4608, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4608?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1693920b420c8fda83b765522e9826fcc979356d99a089ee5d11498e3ea97211d044c95140152e46a6d06f59aaf718d1c3abfb92d958cc6a45ae5a3fe08d6466297433266ac49cf72f87330ff6362dfd9e0cfbf02d6e11a29798cc783cf26abf3ec44eb92059151ec759feb7db92f51af228516d859efb79fa9f67d00e9653b921797d678bc665247710b5fcb5778d741ea19dc43fe149fe07c491f1dfea441e6ef80c91b19e0514bf83bd904f2d41fee442203cca389bd95b166e4770918cf21bb90c5331f569cf8c2154c47eea6a720accd9f4a64a41ff39f4150b3626f946a57e3c433c447651608791a2b8fdd68375cba7c69bfd78b621537d37826e1f55&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277034774Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d, Name: checkpoints/1408/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1408/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=402f85640722e6b2f7eccad4f419b8a674fcaed009332c96a038317842a789bf5ccd8bf7389a1e7bffd4e15aaadda4b3d3041b3cd3a05014c03b112354233e51a6b22e6cbeb967d921fcc657eeada4da05e007670798fcfb506c9526e784e4a8fa632690ef8f913675fb0ccb89679780c736454477f64fa6b806b18d2e2f1731743818bd6d168c88cd4ceb41c4025b8c923fc67abebcc368841930ec449433cf9f3ac1c38c7589e150689245af382942e1ddddf34f1d178f8156a907850f17c65583c8f738ff652795f2e423bddfa57d5cd2b2ae3904b68b04eb0f151c7f1b4cc28a869c1468a88e687df566a44e91daa5dd46ae104fb0a56ab3ddefebe1409e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277831525Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/array_metadatas, Name: checkpoints/3968/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3968/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=03c8e9f8cf445dad8a43798cee0e5047a2ecac47a8239c0a708a8e536f9fa9454cb902727b9a419bce6b66bb246bae5b0104e7cd000ca83449ad0df95ac4502c3d6ff050fc86a25af67db1016bb90565ca8eee685fc1408de1eda787d3b43d052a767b9fff13c638e1bf9d904633158fd186e6bce1e33fd22b95c7d2a35cc85dfec8e918ca84780f53741fc3a4370a07dbb0d86efcbb0ff86e6917512772d887e6bb01f7b4d28442d29e89fd3aa901de3e709c00432927c83f4d0c7f42ff134fc562a5a93a70fc412a00029001a33bf2bd27e17ecccff080e76dfad68201f4318327d0506a80350e7cf15b3d3e8d7d90043856857fa32d0cfa20cfd6c3211d48&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277062921Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/d, Name: checkpoints/5888/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5888/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4b917fc9e1e0f32eea1e4c2c267f63bfd0b19bc5d9e20b5e73fba4fd0ccf49f75725d867c0cf7da0d6e5bcc6543f45117b3cdd8ff986fe1b193c6a59fc930f908050d534cd04a56d097447ff63828f2b0f2e1a3795ba93c7fa52a1e4866a6d51f2c77bea7649d176b6b84cc16fcba7dc6c841660e9a4efddc1d0718258c531a336562177761ce816912d876a694052529dac93879a31254feb6adfcd26370b83efd2bfbf082d787a94e7ce09608427d668bee89bdc72e906f5632b1feb05554894bd6498d43301092a688a105867410b12965f53129b97aeeccfff65129201d76a24851bc4f2418ba219880456665e494638e817c1c3df6f1d214dcd43170bbd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277860236Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d, Name: checkpoints/3520/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3520/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=007deadb4725609014469ac68d56916d5554dc2672e3f23ceb4b40ee62bc0aff37f072e757ecfc1702fc3f2944644db3107fcdddf74a4dfe0904771775b297f7cfbb4f2572a67a791dd90b7d2f6b937b99955897dcd8a639e76a9131a8a8dd2255efe8ff682987ddc790bb57f70ab9fa7e6f94c928dab326578060a0e9bb113ad2485bfcc51d002a75b78d1c8916ae6cae7b4aa2fe5c8989e5098ad1dfe2d29aebb4c3d535d43e074a0976c508f0d99ac7a28324979b27488452d8b5c960d846dc4a343b9f0d11b0ddbfc3be2dbe7815e8b2deb6e5ea10d94ee48da9e18860b5440ec297df52aa398fea21b993e0048e07d98c9e0c67b2a6dcd21eb442049b38&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277887121Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/d, Name: checkpoints/5568/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5568/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0881e15932fcc8ab09f382f3439f876bea3c9d62df33ddba1358326a4c27f75c2696edc91a62806b9d2697d9582c736d3234186f8ca15372da49257fd4212488f8d6911f69abe6c720f6ca9cd04fe654220c1dfba53125dbcf63c57152f5163ccea39ca45bfb688299f9bebcd3d0c4a74e505f81325a7a8e910a771797fe48d7cd0019096cbf369dc8dd34534c7eff8629a7c2a056359ef940f84c07ac6652eec598ec64572aaa5c88236c21eb78c7cf10b2d09c4f06a0499a771cbe3a9a06d6d340e726a7771372caa0eb6c0ba8730b0e9734ce616a47857f504ce13b6f59f774dc930b65521c40bb02bb4ba5316046993e5013e40677c1493c12c5c273a92e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277983711Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d, Name: checkpoints/1664/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1664/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=034e4bb8dc96886a7342f64e24d87e20a8a970b43e1dac0158b3b0c1baef30ab8556e2b050954c36982b7a081deabf4285b1f27ee97b6cffbf7376506976f965238f83be2c355122861d00bd0d7640708175c1376b35d5ec132e84b74d81591115e4fed874535bbd41d566b9acd8a0bf6e88760fc764999b21dde0fd395d44088d04194b8d96e5dcab55c4f1235640b077f865cfd0393d0a14add6ce136869a466d17f8b559f2ec6ad395059d0056d2e68a717dc7d0a3ef86cba6671dd117bb205d43144046f5409bc040166be5df6396496cfd95663b7cc88fa63e831a1bd41375bb27b4ebc8c72eff8508c87cd07362477b759ceb55e9a51f482352581f6a2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276553626Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0, Name: checkpoints/1984/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1984/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5d7fe2c084c8fde8c12012e6fb48e211614a6eaf7356756849129f44a49421ba27e3a1e5aa39ee1cad26b60694836dae76f29d51ff70e3c2c9fc815de1888e5c83b0ea577395d478a1f238f1b2dda4032ceff490d0343597f1a44849847908c5228600077c6872123ae8d076acc120613ef22cca53e7f8a40e27b89ae974b7244218c25c3995096f1ad67c2b963768b05d6bde764e013dd92be6f509c1402d1d06c757bad9e9e1041bcf6e09decaf239ae7c1e21b69323a0e23a08ea26a27991d5e2adbf67da3d3f183ca44e30f5d3fe6e5d25fc322fb2668068798c91c366ac67c81f670fe9e7bd8c86abba6e11fdf0245b238a393fadf6cc3f6063bc5ab747&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277119662Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default, Name: checkpoints/3776/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3776/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=34420e64926682234d5beac20df608e339e1bafb39e7a80fd6db573cbd46863a3fed01f054e578b42df6ce92c1d6c20b009864f0924132396ff79735ccc689ec08a33671f013ce872ddce1ef6b3ea72699cf0bac52fa80bb8610370c9d222898ccf8c5b971bbcbb7307db8d41d9a88216dcd45d0e20070369569a90d3383c2c8a6b8ce76bc98a946b2fb8a2315c66c9c5b65cacee9e4a74158012393830ec01e2de9cba5704576a7e006f7a088331d4a11c8fc24bb05b6c2eecbaa7969bc075eb23d84d3b133068d8ff706e5dd2413af6ccae73bf95fd9bd16be0eedaddceb89c47ddc3e3b8e206b4b996fa174197b45de5f7eb2f8107bdead6e7f03ee0d55fa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277108779Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d, Name: checkpoints/4224/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4224/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=206ccc462a76354674a3b9b8286031636f3f144fc13e4da532a027fcc4f1752879ea3ba4863039d0dbe229eaaaa3565866a50e2856b4f0f91f4a0a708bf4bf86ad94a97f2ca87649985678e8ba04f8d63971143ff01658da2e0cdce7745eddf206acda9050c099c8af20397478537caac1c300bb00cf73a96fb7d9afc921d7dadf51dd441f7921b374728e8e042b0f09ef2f9c3454d5691903ce60bf1e8df79d700818180e7dd405dfa9053febb49493b8a0a9dc5e243c76279063bcd55fb1a4cae55635db4a7502f3e26694247413b6676f36dfd864f65ee2c1221c14e581df1a38c89b8bda8f4d63fb8ade91069c09ec738659b738684aa1842deb101b757b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277147113Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d, Name: checkpoints/4736/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4736/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4df5cf3a1980fb521a2559d74ccfb94dc564632c6e5c61fd0e4959c02057eab04c868f1ba02f64e4db7c943e263046ddb13580bc987fff546ff9ea15b28eec4e9c74f96c0aef7f8576fa0f5b9f6b838f114aaef46cdd4e956fa8cb7bc9f8d18686ebbb8c7aab4661029b2469b272c430b173b6f6eecf4165547b5b4a00708652b5d61b1779397730e975f371d5f89a7317e52f4043055f77bee6fdf7c494d6a5ffea991e111548bad00aeb0c201afa9fa5f349e36aa1e04630e077b3adba236d7a5c060f00e70d3ca7173d2f587eab63b7ccca05830990126d88f189fca113807c20179d00fcbd4777db2eda57f56cc359e6ff9705cd905973d166e52f271a48&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277142595Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d, Name: checkpoints/4608/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4608/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6302f17f6592ad0aad72445c4bfbe6c92c0c4c729b42021d104a0445e65b69cc47728949d7b888e799a87fd3651a1fad18c4f358659f41e96d31442e5a9fb062f04eed44d90e8a5f3ee6557dc353e4d5733e5591e043db398cef3b02e8789d76741bdbed40f454aed8e58dbd0371beabc05ccad4067e9c57f2560afa3d3146f740ac184ec29902d17ee6961143804e97163180e77f8cdc254d8c9c0ef442b75e80c2de05b2f7d5b85621bcf81e4b2d6ef7b3341c4a0c6b561ebcfcdaba1045fb3f1591c289685b73f8a9273f308552c39f5c751fd81ad45e0b9c0ce2b1ba15933f3c53de0eca51c6201977b41abe09468cb2dbc71afffc184848f21112789013&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277174689Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0, Name: checkpoints/4608/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4608/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0338acb819faba4a8e5be0afafc33681584bbf21ae020ba2b9cc23ca4674ba1c40f4a3c25164053457322a5a1cfadd91adeff59976a66fb47724202e9104d95a19084bd70cbb96dd65d8ca2d36ad8b982a2de07169b2696814cb740fdc2924327f733a1e64a625823a470dc4442a4515ceacca3923838414377e4497285aabaf0d891efe6991499110310b9348cf8a3f61ae1aaec134db83fbd92f963340095b722144bec2505b0de34be6d0f8269234cba696298cf911ecac86233e0b420a35b52cbf579747b9274e3a0d4840e78a44a39042c8fb880da7f5516b850067955f51d6ffe8ebbbf381b70a2911ce2b03dc49bb1966af49807d61cce692549f9d26&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277180413Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default, Name: checkpoints/1024/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1024/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=77fc8a406bdb8473d945d81bcea64e8ae2b9db669aa6f0fd274a0a4596402d3038eb7f642c29cdddeb84cf83ad36fe7f8da1aec66402d131c48ebbe17bac98c6e9734ec140c1bc53c74413ae9654c08885d12ba61e924e4adc869fa626a5de968b055bd37f374b6e8d0ba8f4da9945d6558dbe658f075da703be570ee549c016b786703ad4b427fa286ed74d529facd467627df1a9171dea9327c58c65cc4954d276cde214af822f3b51cbeb70e806bd7f750d247b49ef1053d104511ca6aa8715743f0ff9117eb929cf9fdfa290fc55f30253b91a8df105d0722eb2737f2200d02c7cc1c1f96420be8db39e18a18ea84e2577743b903882d42ece491ffc1e8b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27722837Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0, Name: checkpoints/4544/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4544/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1db502f94ed86b05cbe62dbe0a4cf9fe12da6c6c6d77aeeda787a67836289be168826976e201a223f8231eb1b99dac34053319f4e4ec51e28abc89cc2d3cb2fd5ec04feb70b05975807b89ff253a2708d9a60ccbd07fe10a5f8351229b71fbff107985e137ebc7353c96e3e6c6374c712afb477978dfc43ece33ea3f8fc06bfd62e2c0beab060289a9f0551d876b6591a563ce160b5d7a1d0a397d53d787915355068a6ebcafdf72040be1c5214d7282217e00310fa87d5b0662798a6ed7374178b9286a255a1650cbdced0ad8892ee3b5bbc576bb43d6fdf2ea478be7c00131267c776d90989435c7562bbf163a1c7c4555ede6429261cdbf18837382ad2249&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277187203Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/array_metadatas, Name: checkpoints/192/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/192/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=99a635336ceac6b3b6426c683a1db1f4b392847af004bce6c6ce3ffd0c71aadaa3fa61cc2c66ca0e9cc7cbba8b02fd45c55eb5ca8f446e5ae0aa006ba6d309a47bcf447e9feb9f2b560c0b4c8e63830fc8d43184822bf511fe9c2870d4a87b5f4ebb57e1e38efc74e94500bdc788086e3e4202da88d880886a99a6514ce62c1c4f171fe092a1055001930330c216484d16d733df4ea35b712e8bc77d51c32ee25af2b225b752e74081b62d1f94c57bce8cdb07877d75c3f65c38ec8574e889659a5552872d3951eae10e899b3d32362c244b36471225e7fd22800351963d0675200ff413e24cdb033da5050f7c33feac3988425055023ea6038fb3df6bf2adef&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277245761Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0, Name: checkpoints/1728/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1728/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=337424009198d556e0762821d1bb1bcdc400a54a5b0c159f64b3e2465d64288d15498510c1cae82294c81976ff9f6a221a048a29adc9695e1d8ad2030a22c844af58995082ac2544928a44fb380228e026b6e132bebaa12b1e61500c5e44e3c3e25ddf090a1d75825109d0e70bb879c23a7c0a1ff37eaa759136e76a98c1e483c37a13be587c3bc252a0f12e1f5519b54545f5f6d9a0df568100b6e2722a4c7054856b2e6ea96e2c7817585485a8bf896c919109832c82024e534bfd7aeafa1dda51c9e0e9ebc3c2ebeebac146b10deaf6b87f4b6709278dffaaf353684e01dc5dac057fe80efd0e938485843d3c5b888ee0b547431e86fde38971cb70015318&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277218407Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d, Name: checkpoints/4352/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4352/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=52a49bb03be65c469c290f53380f0779bf9c4d47cf849b7c63bacb82c9137b532d08509581644dfbe7c444046d1c1cc76fb591b5dd1d488745b6444cb33dda5b3dc52b1211e052e3bf2aa2746cce6f3c0f0f91dd5582c5ce7ea6bf0e3875d5140be8cd36e46025aab448ed9a4b8d668866d900bc699958e93ec6ef3b7249507818a467dfa318b7c6a9e5787179401361fdfe0d5e0b640e918329c7e6a31ea46d38d6893bb3fc065eea5855e3ac4f51240806560bb0ea7365121b4e21e4da28f43cdc69bccbe2a9e7255532cd683c0268806ef1d244c165e57b29a6021246439f5851af85d32f30c7a141b46ea8bc413b30c0e68fbd2c7b269146bc235e22f963&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.275131461Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default, Name: checkpoints/2624/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2624/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=54a13cfc56e0e3a67aae7d05eb6d5ea17502ab0dd0d48f6a94aa1640bd29081986bad39806e4861db908afa79de366aa2c872dd3bed437b74020f6114f267d7f4bca42bcdd048bf82c7d68988e15f52525a8bdf1876e8cd0065271918d13dfa863f9c59ec40fa877a63c36aa00d96f0dfbbe2bb3ed3579418777d145f11b5ee9003b3f70cd8d569f134fb1d5f3866db12459238fcbcee005961e85bc3808b49b6dddaf5d04b80025e37b8878050b702d73a4a6fd560d81f7bef76c70e3ca520e55fe3bfa0ab0cff7eab0ff2ad760a3437193264978374b4ea2a9060dee542482f168fd9de2c50ca2c82f0918ed2394f444da07072b7fe9708465655696fd2f67&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277290872Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default, Name: checkpoints/384/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/384/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1832c39c94b4bec9c5688aa773eadc84d81efa6fc40dc1b916c2cc0f5e0e65b3653d9d6a584613f1f4d9755934eacec8ef0e775afbf94d3867198465e5ce60753cbd14be3aecfa4a026030bfcb9089093e79f532b8c7007adfb6f7f317a47361b14f780715bfedc868dee4222f51c3cf834e100a79b510ca044929688a478608905502b901f547e0f16fc65b21a846557a0f9c9112e4b0517004fc5372c7c6defae0906aced637245741170274ec32752240e89e6abd119879d69a752ae4b75a65444919eb80d947994155aa98b04c58e1f7d08c75b5da5b1c89643caa740d584f15274a991587e828b0f38ba34098a1defe16c719ea114dad4be9a8264b1475&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.276556338Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d, Name: checkpoints/384/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/384/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=290543e4e4703451b5dac9f5899d3111de14e158c1fe1c3ba366b13f91dcd77ec52ad87abd9758f17eb41f13f3624f7984253c6f81cb9cf8093cd9b4a25de6d0146f957aaa21f1c8516bc7f0f8ce84d45edc02a37e287189ce2c7eb043fc86cbc93a446693887d2ca09ccb77ccb04a1d061c02a87a632cdb35dc5a745e263e0a933763911e4f3c60a4250f713816f078f9638a02fd7114d6789d6ef38a90c22eb0debc17b00c82949b482185f8d9fdc37ea5122339fdb5abcb97590f5e9304cd971e6e08445b96921f1f689237974797a675845d26257563cca47876481151d7b427a25ff9f36727cb58afe8b0184acd8fafea0206bd9d2bd16393a5ca1b71e0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277313305Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992, Name: checkpoints/4992, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4992?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=bef3183c8847e5dfce9d06e33852caf040833b54825f6bbe146afe2ddf310e8352e999e1b0a7ed9402e7716201fa72aec618529b46acdc4a4bde92d7d25a861c9934f7c8547ca37d0716a78986872b0e12dcf1defebece23f8493656c9e78ac89d403609438c1e21db49e5718e9c51185fc2644130855eaa3f186b564b436a236c35a6e735601b225a7f63c828a2ae498bfe652e7a397f827881eb84d714120d6df3b3cbc3ca9eb65d9a4985f4319f86e8308d52faa927fd019ad68cab57fa29e6ff4196d124710591bfcb52e47c742a9a39df3ee5b251d8ac510053567abac71ee4f900ef4608b814c989c932fa9626b962fce454f285f56616bcbfafc70e66&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277283102Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/d, Name: checkpoints/448/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/448/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2a2716be0bb2bd60749cd34ae2de5a2f69a1a0a745d644c47714590c2a157b0ab3bc2c0816cfb51b64160148184301820bf94947fb7db863547cf94fa807a328d94f013aa9acefae1faea7d4cfe6972601f70d58a363a350dd9e141e6e0e1fc55662761c9108030be56c100958ab44bf508da108187cf66c58d69b2a7193cf7b90caf81832a4d05c9febff3d607793e37bfb439d7594bd2b076d4cb80e6f8f5fdb09a757b35d4478ad43eb034d0029b93d04fb2c664864f8bdd5cd043e385b2494d086beaacae8a1bc3680038761084a374ebbba680c60c3e59483a945eea7a41e8c5058db845e6a35da4bf445513092ab60b067735abba26986030b0ea97563&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277299499Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d, Name: checkpoints/2048/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2048/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9aa3bf5739fc19fd64d28b36bab7702da140bc3cc5e38db05b3cdc2482b12d8775751182721c1c1ebd38b63796977eca1290889ca611aad655c6d7b6ad8bd9f055e9c9ca5aec5124c8fba602e8b5918f529c11e16fc135156461a545ca228c9ed09a5ca63a77b579f643f2b7080afff2e56f777ab857e04a39907edb1eb208e4097c99f5e08004baa956def2eaea4f2a93ec3b517ebf9b5844c527d4ee539d91bbd16c04213a431ce5a90075fb649382bfd0f8e394e9bcbbf24d7b11cf4cda302160dcc257aa2148bee8d1459dcc5a47e7fc7b113807c38622a4a7b03bc45e3a69ce8522357f5223889e86460d3821ea7ff7fd41ae450f0fd82daf5b963f4cc5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277358907Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0, Name: checkpoints/3712/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3712/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7e5562ba233e4ac1b7059788de64d8dabbfaf6ff6d23336636a2a1f7197154bca962e2003a1e9c83bbd65d3e554a5aa903e96ee437c188d7f1d3879be323fb6217a025ff656685f9bf6171729261e624d907344dfd41a4e4bb44d6289e7616912edc0e97560e51e801a800edfdc205d4e39d4e131dad9b8a0d40346d2b610fef2c68732e9f2608e94f2a08cebc50a59bcc608fa6e3c759b0e6e45c6a6d64c07c53184e50e3f1c0b81a461dc11b91f0ceffce77a7020c48f57afc252f94986ea5dec978525ee51204072f19b2cf5d25096c77cea052721f6227ca758467c5cf2353641268b1f9ab7d52943693d8bca974833eaa4c5c6c0f80f34956082b51b7b0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27752023Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072, Name: checkpoints/3072, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3072?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b65b3f3d45b938591147d41db36c7d640a5bab266b3418aebe17b1bf1130a6ba3d919f3764c5aff81ad4921dcaa427a9ce31d0833220ba383f2d806a7120e9f3293d411fdc63bad5d2c960347fc7bf73f110f0b8aaca13b62b93cb79882798f3ca14b3c1479d1726445aa6cb05b2cddb5683de3ffaa21a97b538c3ea183a261e8938a2405ce00e9f9ab8801463578f79f34bf6b3704593232d5595d7d196c234acb4b6e7a5d0c34530ebaa138a651f79f0f12a273c0ed8b48c9ae2746b9fcd6fc0cc950a7c3917df0f6ecc864804372628161838b37d2a06571fa1508a2c690c8eebb5cf5292e89b7bf1beb4b6aff030b1099c1ed694106d0ac798791fd3be2c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277532542Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default, Name: checkpoints/4864/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4864/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=14b14dfcc9a7369518180f3d1b9d7bdacd7d5f5a591d9aab593d55c71bd4db4a453759ec69166341897c1695c3bc3bcefab1532319e2fdf353b291cc3e4fd27135865b51194f420207c40937cec40937cd95c1366906c5a1132b3aa10078d6981316cdf0d9cbc0f14cd9c75b5e84e709d3b12cdbfed7e6e279af5e854d60083e875453be027711a86df3b310c66a3e4c80f0272c8aafca795bc281ea5652f8d2d03d961e3475ef891293902e9a11b5a0fffb465758d1144347bb383b47eae8dca4270ab1b14564a6e8f7bc277c1f76178e5cd6fc240edada005cc19fb8a103511ff5d138da77c9e797ddf7d5bb5aef527a4b2f3b28df42ad27e76dd1698bffd8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277593458Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0, Name: checkpoints/3584/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3584/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6e1b6779a1890b1171e1fff701025e96a880da95b6ac2cdebf312579fbd4ec15feb55e78f7873e02d1a78a913b5c25ad692748c1617636b27873a28b51ed9d04e763fc6cf5033c4b41b23e4b504bd846d2bee0718133e5b25565adda90983215cb6dd946ab783ea0ba2060115c857273f039f800fdf0460d0d0be61ed00e227ed0f560992a1245033d556246d57dcd4e1e6c047c6f08bffad437f9633c5a7e4366c39a4ae75fcb708f5bdbce248ff7065c4ad658aab218e54466e4c379f818ca327d7de164990ac23c6d213322558eeeaf483ef08cd19d51c0cf0fe1ce54267abba323ab807b7411ba5b9047785a7ce5ad3a9806aa981fe7a3a5b8b3bea2b3ba&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277566372Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d, Name: checkpoints/1216/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1216/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=65b195ac3491e5d76dcc012df1802dd5acb08f871e7b6596d284eb9fbfd3cb486792e043f4c1254a22d644b31ae866c54d24066ac1022d0fa32dd07952317efd21dca7abb5313344f82846f9f36a79a6b5d07f2b39953a98ce04d8c239deea471efd4af8b24673cf4c54f97d843b7128569c5b1ad5f1e84fc01862a36ad4e1db72729b6060588588b3ab878f2cb7f7374e156df8545c595526f0641b40b24f9982916d03c08148f7665e379c4b00984212beac856e78ed1a990a30b00e63c5143208a31da4eea71a5a85aed02030b8b0584bc7cded8e7f10d3ec58dd1c81e7425723c3e5d4b462c214819ce723429500c4ed11b8e0e2a263c15b6f9c25a3e685&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277628868Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0, Name: checkpoints/2688/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2688/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c30be1409e64718a5143014b6ddb7e30a0865b1c639602c89371057115545e4ff770f3108fe32198cbc5f47c16083ef0b87d9ccd98af74c9552364ba355d83d4b4e6400ee8d984941878ca674b24c42468302a244d9f94af91296eacdf9939fac7f297ceb2f45e1cf4e0677c15c9c92ef13e594138060c6104fbb83280b7c3341f0938971d7e40afaeb49f335c4d27dba833d5c7086f3ae8df6592d9522931c801cfd02d9006a560d5b84946a0b1e67be1129859edebb1062ff4d5bd451bc8d3edab465a0f79ebd03e38be059d50fc0a26549b076cef5d695390fc22bbabac53fefef6a1cb7a5684009191f98e0ee5a358007405c41616866111fc0d60a63079&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277672454Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600, Name: checkpoints/1600, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1600?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5f3cf2d3cfbc0c3f0d8b507f5e8a4cce7b29f5b288d5ab26f0143180a283d501090d1a03e5eba0732a42a5e713a1842cb3a7a5c9a08065842daa926a9e0adf9828224b3213a7203744fd755f7f3cead64a91c11647749c658ff5172287dbea7bf181835f0ef29f2505bee1dd82f6c795997dda19333a21c47a49b2bc7b49b3fa26a53e4a65f17a631f224bb5aa62d9a841df42b0c571e1e96feb91c2f6645f0f7f96539b0c953b925b75a76b0e53534a8ff63165d1b580d7d87f651039f20b1ee288370916bdf487dd17372e4ad124cbf073dee001a709cb667cb0d31e4e87664cd1c8cf2922f1deb62c8b2343193e17cda51e26929e1a726a431a9c9ea4ac9f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277766971Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default, Name: checkpoints/3456/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3456/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8ee15074aca09a1d06ac648af6b24c2ac100d6c604b053571d7229eed5840d025e8efe17ad797ea3895e88e3c194412692debabe770cd793a68680d920d31cc32ae9fb6a54389171648c9746dd6fea424a70ef92f6e34daced3a4761f65c605224a1cb5a1e334fc3644a964fa4fc4f46193c62191da35fc4e4bc1aa5a79c8bcd06e102f06e1e38609d566637952043d6ad8262ad2f842955c3e8de423b8c7ecf2bf303d6df5aedc75613bdb6e57bd320ff76e8fe11a8142c7a220a5f52780050a249d091e153d854937cbbe8a8ce02665d98c5556bb880affc77a4118448efb0e4444f4cdf650d759d54fca655667738f35387239552e04ecaf997ba94eedbcb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277031131Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904, Name: checkpoints/3904, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3904?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=bf5ba3bb247359bfdbc67b920d2bb10c64a358709fe1e6bc81ef53a16351f0a20a7e5ad1fd5d770bf323fb312b8a46eb93830553495f4d675c43663943b1fae68fdbfd67dc33e9a9a32b9f4e1d66878569b5ebbd1dcabf100f5bcd0a1dfbdaa616acc0aad326e9339fc0f1582a6631ae8366cc536cf612da8f9f992df57bcfcbf5d67be7fba128f85b58cdd842233ce60ca8f054de305bcde94406cfc1e7466f162ad9673e5c83d23ae51fb8b16b5e1553b4188b49933403348663d73a670ac46a4b4b2865c3d9ccdd34e09036cb6fecc549bae4f75c820b6831e0e760ab8d2dcb80e4f9b0c8c0936d3d639b12a100461057f9e3c05266d96da55764e7c8f7b0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27778348Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584, Name: checkpoints/3584, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3584?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=731df541b997669a2a534c893ba3a58eb658cd3bc864f0d4615037c06a288dd28cc462750cf90d47dad983b2b93b575f7c6b9bf9f7360a43c284011190bbdf4a087174a8ef8f2df0d94ddb137f5d89377041761249d6e8cf3474574632af8e939d53affd40780fc3e43f72af9c20144aac174ab8ac5a23203cac47861150f723591d798beb70314d8eb33bc99473de87501964ca7ae5c64f3dccf80460e8e157a93af3819b547f44d20d0063021524a5ba74987116892ae5eaff38bcb141f3563bb1a5cf1aab851921cf4a03bbe8651653f4cfd42e3e4eab385eefbddcd35ecfd28f6a6926b4707d0ff5e43211c923ec9d262ae6d6ff0802382b23eeb3e94542&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277844752Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default, Name: checkpoints/1792/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1792/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1844681299b30d4c7fbbd349da3bf8ece91e193becc83d8719be114f32eb3c96b28d2396972131487e8c5b4b8d37a251da219f66423fb93f6ffdd0745d4343f255e5bd3905a5257c2fe3731495d5ac8db7506421721cfb8a48eaa6fdbbeed0d67b41063928fb7b51a38f1dd0bdb5dc7b8f1b8bf3e388a761c070616948d674bd8f5f724e839d003f1b23f7b13d490fb9ded47ead1ebb4dc21b65e086ded5b3b54c2d0377ea20fdddb67eb6679762e3545caaf225afc92858713f449522bb8bc807899d17edd461c7cd57a947ad2993870a1475fb5e0b60743b8dc01e3240b403d1c9bbd8c25356c1ea57ba8f92b3157c29c66939e9193d0fa65af376d2ea92bc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277098738Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d, Name: checkpoints/4800/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4800/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7feef6cc65ecdd1460024344543cd09078adda9fcc16bf85067c434f3e1232e69cee6417d96bb56ec4d12f079dadb674d73c95dbe2f55c9b028ac94eaeb49978b29f8836d7e58b20d39801f7ffe4c8dfeeea1871db9fb7ce3accd3dcc138a4b3b91cd02ffb61fc6a25f5f8ea7e601c24db6c3ef20fda7b5aad06f008bc129d85af2a3ea8062d8b705a7b1d5af6c0382c420015b14d208506c07d8cde3481d2baf85a9919945742d1f544371a1520c4d7009945055388460d3cc4722364ebdc0bae473809a6b1c24a938cac08b7b87a5bf9bc2a64a332310506ed761391e088b139f282a611d93b338ac6d2858a7e4b51c8a3ed7ea3baa06d2acf010ad2395211&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278457124Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/d, Name: checkpoints/832/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/832/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=78bdc60e5a9ab3d6a30d9b5e55c8cfbcdb2f628eaf206c20ef02846ba9281792f41e1644fb5433303bdc20fbf497d30faf6f24620a34c54bdc2bcd2b732b561c83133bd0b73f3aeab2ad5f24fb53db579ad88115312af6472931a88da48e3589e03865ffbf6150f559161ece4eb0487500b95e12765f497d2e0dffc2ff1b453f3f74bcaacf1d748fd3ef0367f058d8132a271ab996bf58cefebfc055b4b21ef30af051955c7f360233bd5ac2c9324d5a1ea3ba196e20886ede3d6dfcb069726d729e297b6ee729475c3ed1e611393e83ef418ededdc79441e40f6ad137ac40b02258fe5a4c085599b8629819cd5c46f617022803208ea8d2e9770bf3c4ed48e9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27846881Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d, Name: checkpoints/4992/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4992/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=bd07efd29c7049ae224fa43fc6629d9af1efc7e11ca52de0700ce0ea57aa7b00bbea7e65e7ae03b6f622b88270d3aaedf88191c6a5ab4350d8dccb0a28a9582b3c0b33ee9d6d49f60536572c6275aea80df6c3eda97091738f7af8cad7102f04155eae57148ba77808a17c95011b06330f396be1d466b01e63d8a8a8a4fd872e23e479c60bce662ca0f33b0113d81c88eb958c1f0bd1b35dc673bfff5770743f9a5f8ceebc300851392f9a093f856e7b468ed92c32fb87a5b8efb542f8d9b17b5b73152308d3f5032297fb8168f6372fc566968c9ab4f250a4596baaef0ad215d07273a3b98c4603007c22de6c34d776cd5855a8b30f41d1200f231639f285b8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278482874Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default, Name: checkpoints/2752/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2752/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=819747b74a1c1448502bce85f2434624cba80bda06651445811e78a89d6db5b29c0b2daf0d2282c26fb2975af379ddb5dbc664bc5b3f84b26da1fa4ef46a0d8ef25acdf50d4baf693a26703a3389212c06a78ecdb0293560d5b437b96e009843f081beec6fefb25e291cf8a0d08e7dab815a4946fe5bf3500d656df20c79d3aff474a8861a4665d0231ccff64438f72339bba226b9325f151a808cafaaf4b6313fc878a848e6ac861dfedfb20af61faf473b64fe3110a698543f6919a5755069b1d50d4664eaec7bd9f1f9c4f2c7a88cc897c26df0246c4cc76df2c3d1ded180b144fa41ea3c33ac5880901b5315459bdadc3b3beb577f79a58729d21c646775&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278516326Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default, Name: checkpoints/5568/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5568/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=84d7b2d3abd92b43e6a2348c869568c396a54e80445ab6b29b05c790dbca1b9555a907d171a0afe7f649cddcb5a12c08656ed06ff6ff06a25ea09054bdf69cd6c292ab73d13b7760f98f55f7cf96b19048c0e3638d68cc8379b8416b48e6b9f9764246dd6b756cee51387cd6c3c49f0a99dbae6cf56422b3e5253f56e66751fe996cc286d978ec293a79bf430e1f3856e2d943d4255a6321c7ca6a5a1aabd172e14977d88f56890d30168d90a5172a4a2bbb4f71f70379182750fb9971d037ce5e3a425225b6dc9a3665d3ae146646af8c637affdfa9c1d8fe89956594e2daee84dff84d3bbb1c5b2342d233f8236755db1bdb4466c0d851d7cacee9f894dfcc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278488653Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/array_metadatas, Name: checkpoints/3712/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3712/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=533f72eb156d6dc92edc1e5ff2606ab60ad45c3769cb7c497b12c6571aad04572bb1ca1107204c7849c77b0f8450c96c1ff86250afa640d4a3eb48684328077d790c961ba175450f07a03b69544a2f3bb963a4f1aae677b7e8841d151e07d3b73c35c2368ee2eece2d10ecca59732e6e8d7a3f273be2b8b542e4ede3c53c95a051ad36e7496cbdffa3294b2ad723302e66f612b0dddf33efc6d0e54eab754657613df528d71b3937cc39663bcb9097ee7ec14420c1e4389cf242762657f94caa9181e9fff7f2099c4701c05bd683890545b4d8c9a8de0e704b89aa11cc3a9dca9a71d1de56779c635520f87ba0888d6480c22e4d44ada9aa58433fc59daedbd5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278641691Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default, Name: checkpoints/1920/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1920/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=50ece88151b4591c31a0128ba1789e7ef3b312b8a45444cfc1aefb0e8d3e4a06e02832cb3170770f604a0832a2cd22fba7703113d55afd8300326c1beed0381da93cc6524e52b4f8cd5dd159c8f8874afa525e6ca9f700c1dd9b7df048efb155efed969d634ca78986d512842980f4eb780ccc0580b95ba2353a4aed4ca4c14f8c8aa2adcec6ea3ba6a34410789db1fad46835037b0d0ef1800d8f8e733655ccb30b6e554a7f6a4e0366966e021f9e8f5ad589ab183e22fe141ec1259e4b56454d2e6a6ff091844e0dbb1fcf7fc779a1352d4c70e2473ec6818f9a47bbd7d0bb995ccd66f80a6164062c63eee579897554eb6c8f14fa2116e283b18312d1975b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278643564Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/array_metadatas, Name: checkpoints/1728/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1728/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=20bff5db31d6f9472ad665e0d585dce1cdfe07c237260004d79de731e6ac7bc3016afc8d8d2aca6adb198572080d7b41eef581bab720c8f1e4d7e21f034cfae95e40bae426fa206c92b83586e3e9b6ae50bc22be8ff6124e21004b89a785763f8d1628eb118808860518ec2762271eaa3aeeccee62d27d3ac6df533eb2c33b293af78a7570676a4d7b1fa9dd475536b0b2725536d429781ac9d0b8dbf226496b8e933e08a448464fd0c2251fdeccdab73420235c5c304e43876b36b5e09461e4b05c83492ffca94991a0bc37264ae2455e03ddc97ab4ecb0e7fe10d93bfaf88c4fc9841d1abb20bff516ccffb7edc97ac015981f6500153a1bdc51d830773039&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278684297Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d, Name: checkpoints/128/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/128/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=20b48674b2effc92ee40356b0afbdeba8814c5eaed438c0bf35c4cab371015e607405d70ec083a4b1f12f3ddacca1f4e6f2d61fff3431701535473dfad3f9d65cc41dab0b95eefffbf3580085bd813159a023d82cfa859b16bb1cf9e6800df970e89f0f21edaea3a50e6f09f5c8b78331e35a782e09b6a8a5948fc13a9cca48166c7723beef75a0e58843b62534064c73d9730a237ea3c77eac6aab024b058de233c4b5a6689d615caf11b500ab03a3650711b1a5585ceb76bb2620632d4af8cc7c8558f8d1163adeb6fed571faf5c094f6d0ba7121d250e23842eb95382c205eb414f8bfb72a624f6f8543b01d804bb020812ecf7b2e0bf627a8b6bd473f52b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278715722Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d, Name: checkpoints/512/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/512/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=264cf3bce619c615ba3245c49bb73d6d848349eb44f17bbb4f9ebdbdd069d9b683ac4768dc5b3e9bddb9ace35a72c3995c7f48bbc947d00a6f10a04a8950f9ae4929d3cc48178903aa2183a4ded9e140d6a0c572227218647784374389fc41c71c5b763da169011e3a608dd95a4668733205cb52262792c75515247e01bc0048253a69fc2df8375e83132152faae509075f66d6875ba208a6f86e7ac455cbf807e6ae74bf141549301b797ce12eb81174784245ade25976c1ed6a400ef8176197c3ea286e908a036c85fdf959e5bab8f4ff0181080f3273ec526b64148d4db451ffce57c6f52ff12ff02529c5d33d8fd10038ace35e9d42e148e497ea5eb351d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.27877198Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/array_metadatas, Name: checkpoints/4544/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4544/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5f9879c87d9bfc4962736dca71efa2f86a4dede0c1dd55d01e26eb109253836f89a208ae154d4f2f27c6d7273eb25f958593e91af1d0bb132ea9f0856f669ffd26f36735161c425562d7b14c25661096f5c503ba606ebe104efe0829a08786481979457031663576eff769ca40231b156dc12d1bd549bf6b3494f51c54b78a738ef9261d73c952388cab7baa6f2fb711c7f430772a392e783bf1b9de681334ab151282b9647b49c3e945b5bd62cc195fc5ac79b974c81c32e0e2f1435404fb8518c324f5aa6b8a3cad1cc3cef705866e2d8dd304c5c433c50bf3e299efaacd2c87213b15a39bbfa775368fc01cdd2cd1cc10626659671f5e74240bdf2f37d53d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278772707Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0, Name: checkpoints/1088/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1088/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=29ea2c82391de40b7886ab38403633d0d20a7f430a22f967068a936bd65ede2206ca8c95ed6fa023711f4796e95189a8675d941d6ced5f10a37d230beb18356511c138eb39f78f8bd3d7463d09f5aa5e3ec561538e480d78c161206b44fb334eab9df8dadd74defa8c69279c6efd1876797ea338fa2a2c3ef4828d09dede7edf654b9c776e6564479d1b93d3e99b9901162f2d33b433cf6f01c31058b4830d43106cf72d21e141ac68fe63e172cddc9fdde1d65b8c3db8d6cde6dcb5f954a29778f44a0dbf6f73b39c3e0848d80e6caa10f00de4239a57cf5fd9394e043d374f343d451ac057d5ce69b6c9637361e74569f9fcc50ce214756e5ddb57a8ce1c65&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278771976Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/array_metadatas, Name: checkpoints/320/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/320/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3158f2b4f687ec8506efde373181b16ee7f98b0c36bfa8102fdf780805bfcd5d75801c904284ea5793fe6f2aaa59c4654cf69e5f4c9de9103562fa112f59a51acf22cb61ab428df657ee9e3fcbde74a7510304e165ccfcf5167638b6c880a3b02b23028eb2efa53c6082a616ce2d61aac282c008d4a55d296215c3de84437c700995edd685694e005c17eea14304bbb89dc3104c88fcb749e75bfeddbf7582815f3a16be0f7fa3952c08837bdb3fc303ccb0adc07b4032da95d0fefe6506f693d1c1c439c1f1713f8ec80c65bf1ecd3e4f4d1372d14806f9bf64535a5a8b1c84e1f26e387920b494904a387e95c452f75c310ae9658a585b7e06f876c58f07b8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278840905Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/d, Name: checkpoints/192/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/192/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4793df6f31889dccbc4ef76bd2775af0286c785dba8a0876629866070eb88bb2183ef6012845ed123ca4945a7ba1089518fcb15ebef27bd841252002460f9488695378fee2b016e1f9a89e83cd8e4263adccd4c66aa1c723b951fdafe5d8d1bbe364ff1194547f0adfd2387a7dfb05e45d85077249d3bceafe2aa5cb43f7f7b15d3caeba96b8a78005ffa373767f3fea27bca7a34f2473df5e7f6d8b0fb4d671ba782a9ca40f538e43aef322d68c9eaecca271349c8358b0add08fbace100bcda74a5ea0a52982f8902c4623c1a5ffbe973a28cabbe339ff8c232b49d4e39c32f42aacfb7ca063c78d9e5f0ede18684abe21fc41fe5680adf2aa3363e56c1d2b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278873979Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d, Name: checkpoints/4160/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4160/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=214f79141ba28b195b4f750521d16bd7a3353f2a336528f4cc38e4884addd273b3d7f15ef7a813ed7de29593b4f707fc0972446a2720115db4df27e59c52f3f6e0b199b141e3dd1b745fd8329c606a937e887ac735215a1d03dafa242fed73a5380392fa4905a02aea0855b1978979521f92dbfb46b038b6116c66d0fb85ff382b69a21ed06ce3741dc28ff57c0eb6e042c8c88c5fbe3265532b0332c93c3ffe12084ef4ad952e937959d1c1a5aa9407398ab059124fd83088726820a50d119ed3d6528b8e0f5c42d03f42fadb6dfddf8f3de908ada3288cb9a77451a69f1949dcab450933b1e11bb80271f7eab0c765456c59c966a643c5b7815c3c52a10c5c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278873496Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760, Name: checkpoints/5760, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5760?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=aed115c675af71d9dd51f5151fc91768b3ebc11d9275d581f97f8b1aa4bd95bf346b3106557206c8fd9949a8bbf0872377d717e0b3477de6eea7e7a50c27260cef7e349a4627470ccf85b9865b9aad6abddb08e166087b934a692f8e19a6ae32ff6c356f55e26568afb3983f2136dda7b5364ae603d91a884eafbbfc03921203379961b2dc5cd1728bc2e53a69150eceacb733b0f2b2fa137e63537db28c61015fda8c5ee2492fafe172f0abbc3d3c26b432e3f75bae78b9fb6c271bba734a310318400eb020941af6a053e5b6301b79fb91703d11f4b26f14acf45dd2f28a2e671ecf3995f3449eda0729fb4126dcffe6f8fc6e749754fdad26b0f1a17c07ee&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.278925267Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0, Name: checkpoints/1216/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1216/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6251ae8921760f1c384afeb482870c78c7a453c01b4585815ced79b257fc34a9e809f181ce0fb70b88df74754dca9b1066a80a660541fd4bf52d5faf5311c8f7cd7db2a3f14c9e539a7fae1f1035d06cbade8f8dd40c67ade5948061b30e6f8c5ab310812ac23de6401f49bfa078e553ec352d1dda9d9fa58f0f262cae08e33690f09a13221d9b2149bfdbe6eaab395b5b7f78bc35db63303470a91d14541e367877c2b7219cee26eedba643267663bf3a6806356d1aa3bb8d87aa76cc4bb62d88aa54e5fc7cfc67bad19bb679eb4bd6ee7d4591a153eb66446dc11fa5c025f7bbe5e2999ae52de4b42372eafff65c757b4152cf4c56ce56af621cff462d8d52&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.277853115Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default, Name: checkpoints/1664/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1664/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8d23609b196ff0af9d930ede7d9933c8deb33ae2ff9e05b5edbfea95d77a77beb4f1971a38cc1de2bef6bc73f0e1edec883c50610128d60a563076126c711c98dd8ab95659854f7c61ca0c54d4ba5ab590db62d7bc460ff6169b53b38992c6dfcfc45c510a0dde02624bce30fa05fd3a56bd611630096e93ee7d1825f9b0813bd9dcfd8e4ee46b7c3861cd006beac3b34ed9f35be13c283db9f33188ccae86db3ec9d99b9636ad78ca65ad078c66376c916353128559947e83834dbbb34bd72c9ebba098a5ffbaa68552298be7d48d6aaff2228a834707c83873a442a882b5869734a10cd6424d47d71af6fde04ad1fb0de735f578ca5f3451787726a387750f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.316002736Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192, Name: checkpoints/192, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/192?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=be72e358c2231a427755add729c544d1c745993983302f614ec4bba212000b473976b80a84939be3cca61f10946c0a041af45783bf4fc257f6dcc4368a76651ee17bccb6cf7ba4b6e74274562c71faac2d9cf5908ac0450875c7b802a3b0f8b093b61c958467f5a69f27bf107e6c78c7d140c9ad4d19890e99a700ca48a036f0603e3dd77b00f7b8711605789a14f3e8ccfd080bf74868ebf149d2083881ee3d7a411c192cc1392bfb81000eb70e735fe0a79b0dcce0833ee9f12847646de1c8f881fcd5bfdd1f1014f5b2fa996c4e38c6b75c7ab19ccab58d601ac956f7b50ee2c5f43beab4c38471a222c4e4471e2cf934003ec7ab95889c4f0925c9566993&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.316045381Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0, Name: checkpoints/5568/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5568/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4f74f6b6934843b9faa8efe5c5c0fff6495d808e647a20aa98d91cb3df6b27036ae21d80d93252a683b6cfb6d3d23e20a4890ad6d8f0f16763f808273633e48f85f1e983f9ae238ee72bdbfd6368fe0dd196d66e693a680108920629400f29e3a8ce646f76f1870fbe62f0eae35d6961d1824ade3397c70eb4dd75cc5f27a53de09fddaa01bf4813437d0c23b48d6daab7712960bc4520227dcde13aae4b682b9d6d3ddfdd2daf14dd07cd3a7da8457413f5b5303695ca5883aec7c9d350323be4803ac447767a53e3559a0cd74db9d3c7bbae477ae2cb17df79d8d5d80f53a576dc8b447ccdcecfd0b44a2d5c0c800c099d237aa444d33ff3c25d42692ce352&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.316082901Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0, Name: checkpoints/1152/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1152/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b7f3b21d2513b2b4c473128cb5a93fc8a34b2fd5a83abd1be766071d461d58faa74f272c04795f672d5281092fe220fded6cf4343d61ec4f2df21775e009cd710d0c4ff47e78dfe6d0523fdc203f1cdcdda67cf785f53ae9703756ea9280839a7b35bc5d41734e975668b2df163ba0d051a1c9e1002be32e3c069240d20c616dfd090a032682fb01020abe2fcde16d2205c34a9af74a3d2c3e11226a3f45394a038de8ab1b62073f419fe87eeb613fa10cd42692ab4ba86bbd000c511a405363e0c5449dd86f50a95c1898563e5d2dd4002c4a0539a90557b680d5fd2426452b0d9a5ec7d0af0ec6896117f91fb1314f8b461c0d145d2fb2a155a654fcee1a37&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.316141594Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/d, Name: checkpoints/3072/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3072/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7f867b5136694c79de577dbb16c3ee8a0225f41ffbc9b98f13d75d142372187029e95d46d2acb7359afb012c5fed87d6baa2bf3fffd915a7e1d95b59c8617a852bb221ae0c1d3d2b10a67a8be64dce26771a8c1f92b6164502debf300af03994e0240be06b9683626ae84d6f6c920afd74cb08c5f20eac6b5a8696a0cf2ee6b1d9f8efa7a0e597118a239c8f7f770e02b52feaaf86d7376cf28197e6f2c107248192635fd0412c0a39e9fcdf7e7437a52ba404eb5b093ca499d1f5f10894da5da4233cb8bcf9305d8ef780cc8dbf62f8e5a39332149eb10224118e9c7ade20fb4a6fa34b75902184d4529d0ef23bcd4c9b7535928407d31b13245821fc8d44f3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.315979373Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/d, Name: checkpoints/3648/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3648/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=95fa5345191421aa7235fd20dfc90790b40478ded3464c5606a10184a0d6d17e246197be173d34b5c74b06529ca071c40ddcb94b829c8e530d41bc816be6f94fd46bfcaa7085cc8b4d36c9d95dad1cdf1edfae1c62bdec10569782c88000aa781e9205549c78be8247c22f7c99ce50931ae9fdc5d2da3abf8ac5209a46448b3164c5bf87bfbfde2b12e014981df94d4fe2b3164794f89cbb283133126b04ab4f5b512903b8a14667a5469aa11905d569a7bedc67f3672bb617ab73207be76d03877fbb01891772285447773d4a3851c5cd7b7346b212e733015a7061b6d33be9d42ff73182b1055016f3c4b56dace0f598dcb975d59e8bdf78aec86014591c61&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.316017194Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0, Name: checkpoints/2368/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2368/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=008e74947d7e5d10cc57a22e0d6d2ab131a0877db4785e6f7c6b304e70eedc3f27d535baba98549c602215be1bbbebf6e79ea12de219c5c17e301ef4e2088e142f8efda329e2ddd4530d9b5497407a64f7edeef94e7fdbaa40ce7fd172e11a70f5ae747154f23bcbe4b1f9c41cc34e507845321ebf7347c6a2279cb3382a37f1f8d97def4932343559d94c2211566dc29cb5decf761887f778c2a2cfaa0af55a22281bd331080c45f47935d42e67506b9006a3390172d992c2a0f8b4b6cc08eaa52bdecfa64cf6485713bb45c3bb5537b0a5547d743bc5f50f2fa57bbb0821b6efaa4b936f21a7e82aace27b5ecf571628792bd3e037a8b336eee792259b742c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.31616046Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672, Name: checkpoints/4672, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4672?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4061faa2f8ded8689744d7257879affff1787318f5e708f374786f1081711a3cf56b044310fad2d23ff3b72f8c096098e9eee4fcd04c29759bf8e262a6a87ee6935d617af40a8cee5c045854f17c939fa9ee69652ab74e9e5124a685bf0961ead6c2c46a67c0e245f6b7b498575a5dac44891a792703158bb504ba93f43d2e279b8ac022b5bd04b15a349ac79207133065d8242ddf4a8741ebb0340fa323396c86f628836f57b128b8ef452961baa8bac704c19a1b3929f743d6b3990214d484df5b427d1c9338d723dc8e58391b4b8a28beb40a85479adc405bd3cf7aea917d223cce5140c63fd84558b867f9919b718aa9dc64270f4fc037ac510cbbbb8339&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.316163539Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d, Name: checkpoints/4928/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4928/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7971857ccfbd9d69cad91b44dc978517c4beee942180a0a9cecdb60ef8a2188287be388f54d75077ea5f9195490a0ff7373efbfea027476741225727b097947453d31b92ec17c13dcaefcc1ba19419b10537c5b97506506d2fc0bd777c1e120758d78a6a6135e207dad0417c3d67e7b8ffdc2710640ac6e119ddcf212495fa7edd90452cdcb03687a609083aa2f69943d63f112d2b147580d7acb55a3891f24f2d40f73aa5608ea5f4a31b16e728940e95c722b713057fdcb2e50fb0710036a10ba1dd164610b9f37ea82f47ac3af9ebc13ef3769234d0c55bc2ab153219fe9dcc13cdc69b2e78afb38302bf3a54959b4ce027e31ef33a478a76fe3649af40ba&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.316555032Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/array_metadatas, Name: checkpoints/576/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/576/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5e7d4eda1c38e3b04f9d3052b35eba4b32c71a85b5cf18b8eb60682421c3f141c0a822d848f089f434ef18c58a196d06eec73aa3c41f9ede28b1d460ecbe2193043e6dc5b8f0f56bb5693fd8f32df3b545fdb5bc35b5e483badfa4b7c52d1138b69829bf3449329139006d59f5dabec1b9a0c0a3d30f0fae674e05a49e2a94b2964ee4868b10a80eed8bae870c06ad7116ea8d7b065ea232ebd26d86795c3906b4b793893e1e980898d883fbfbb95f250ce404bfa25679060f84d7227d0dc5ece068bd2877b73c975f63b56c6192b01dd81baa786cd519da2cde348c7697973366b15a00aa3add2ed325a1760de61e15b61cd290b63afaf05d84b0bdb147cb19&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.31615544Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0, Name: checkpoints/832/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/832/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=bfbdfe4897d4bc582c9afa1f2f17e5711b46584490eed3f6e9d8e67410525ea22247ae96ef53edd76bbcbe3dc5b197c3aa44db55a5735bea379cb7110b2305b245bd56abc846403b4b7fe6e012febe2f1e758798436bb9555ac7b615098e82f0295a71b60606280eed6207f08bf00525808478b9257d065eea626e0e49de41442b4186ad1329f42e7414b56d53967d17c8cf63a432562eec6e062376d3e59cae50fae7dff4e3e03834966f3fcfe3a6ff2fc7ce825768088f41d82f106a34e0fcca253d8aa87ba3eb4a8094be367b9a7fe047c970f8c7aa840148344cabbeb3b09dd25153c79c258f5ffc6fa9fc91dc5e567cf4a88f39486e9f35b9fdcf1e87dc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.316138103Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d, Name: checkpoints/5696/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5696/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=55e15d05aa24b648dd8995d81a7e7a12be7c9c9b0e24ebab00fd3e2f3d15eaf4b141957071686a4f93b22f543d4ee4047f52f50f96637d8d81b25830a216e56836e2cd82b9243f802fd24091c22eaba0bb21f633161f931f85f3bb93fd76cf13866ea3a0659279d5526ee97d91a1d28daecf024e19da588a7a55b2cf4766eaea7aedab58c52ab89f2007a1a4567a8409e272ff37a4dc58024d8186527c2e56e6ae69178842ca431b8d866270cbb8bd231699d4f7b8e40e68149df0b92f967b75e2c35195d06af6c650016a727d4987c8565e7a391bbca5e87dac0a46572b74e2fe80fe6b7d8b7b02b7174fedeeac2d37f9717921e1eae3a546bd849dff7d195f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.315992036Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0, Name: checkpoints/3008/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3008/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b68c6e807d391d84a8695d248ad745fde6011cd2a6f719b585bc7b10ea8409c58af0c7978e9f0f2f4246fca2b74fd5f7b0d3f5e170d5c2de01f1fa0ab5de7548efa25bbd84597dbab0a5a3d9ea87ff487a36663b687c231ab2c7281a3aae9492ef91990d264d464f9b35ec77abcc13a30860ee662c965c25e531f826003d93a95d260fd8c2346544fd88e4eced25ac3a1db2676b5d614c878c09d746765414df45ca827b9c7408722ce28d63837011eafa2c5db686708adf82b1a9d1c10f48e591d5af35c595d7bf610343d19ff948b03783ed4f03d05261f7e8d3705da5f47de3e3e3d7177ca6e903b5534e9b7181a6ad8f23a945839cd277333673df3415d5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.31832123Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/d, Name: checkpoints/5440/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5440/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b6b8c342083c629c8394ea38c848126ed7573c32b3cac1e1a62fcda5a171c9d6da632bff117e0fbe636be506c621844ce030776034e69592395e29c340e7559ad1d2c32814cbcbba50df6c34439300653938669930686712c59b31b7e49da46257a42a72e6e30c579042592ffcd3e52bc1bf0820a10d052bfe307f02211860e11429f5e3b61f8c78a6220bd4a6efe1ebc772f2beada38a991a6641afe795b9de8b5b0ab2246af594d46944b374f51c38b5b068b2bb7be0a006271cef09c3377927112f935529b851d78212272e806a57b9d48ae19259d2d2f3d7c74a5cc913a51159e121fcf6a4ae1cfae92321774d4e23e01c4abe9cbf5a5f117ff15b05df68&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.319430563Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520, Name: checkpoints/3520, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3520?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=428f39c8118caefeee90a8522ef0176dc4d150be7e8e63eca7e5b352c85f8d13ec179965a0987bf58c54ece81fc86d82c9f45511fb457efa58cce1fc5ec3a741202b263527a6369be1caf225fb440275a1821c2002ca5e8b0b7771a1e293e14aeb2b1dceedb27989d3c6ea2d10c7c6bb698167c6dae4494da4170bace3f9c57b154b389a91ec3b6637465e66b62f0fb2a1459707d45b15e1d1381b288e1159fb6eb4ae41ff7dae0cbadf3b7030cee74d30a9a7fbe645026956bf246c735f16dc406838640deb5bd5eb8f5ad6e0ae1e98413e706150beb904b35aea2894e1be23feaf24aaf23042df88597aaa41f1fa6da1beb3f3e39b29d071f76f8461cc64f7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.319528499Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default, Name: checkpoints/256/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/256/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2a4c9e6845688c93bc96597a3c8c967f8474b6cbfdea77f138e3cca33460695d0aa1e20d89bd473f0619460aab12ce3a35e14796d26379fc84fee1f37714e1e59217d0734fabecd41e66fe99cab52740edbfbce70e12b00368b84ee9ab622328bb6f927bfa0ae773cee561b3a7852a511f6b31fd603d184d630d8617175cbd922fc93499d503dd1f51f8345d7938490e95e01e9fd2df94ff3d167b00a32c37db6e567dbb7736ba5152c750a86a434407bef11361ccb809a1fd701e00956440a63b1fcb0c86f586ba0441071481442b83c99d713ab6ad16f05d5cd7cd1dbb4e2db86ac7c7aebeb511dc7ba610f28865c980946c579484af0dd9a75bc373e7698c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.319551566Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default, Name: checkpoints/4672/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4672/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0c5195f8c05fd2ad7dd63cc6072e15dd02d08984f78758a15a9637c8cb8dc5225d9e676de2abd3d8eb89468e70aad3352d546b0d2e964cbcf0f5b19a29c3620c1a7ce1f21a932bbda532bf6a4e79384083df8f0185542f20e8011fb971b4c5fb61a6fd11d8b24cd9e18563733a057b75b1545d82534289fad416121d119bcffe7230cd466afdf891834127e726f1934e8e63f5064fa45cc4120234b97dffd8a054b8b07a8e5542ab57b42fe4d87c41fafab80f27fc7b767f861fdfa39aed92373003154fcd5d649fdff48b57deceadb934e2beaf42ebb9dedf38fb1303885c24677c4ab7e574bfa85920275c7fd09032624547128fb754b2611865a1089b229f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.319557868Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0, Name: checkpoints/960/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/960/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=69c435767f0737305ef5642694cf1cb332add2a0ae8e411aac50e31b6752e77f797ebe63c8c5d9eb0476a4fdfd40819ff4158fce84267ee3830b29922aad624c2fd875a4991565e4ab6dad5a55bf3efeb66af08c93cb6233bdae979891dbdedfbf2d2f22baaee7df6bb7caf9903700e109243cab4dd0d17d4e5601b5e86ed9a9aae0b6cf51133bd9dec8578a5b95e79c3970c85136ab2b710230e5fd5e88340d61be5dce88c1ef3f7da6c107deab3d00506122487fec69e8dbf6cb70aa9a7ba08c209a5b1ef4bb9cdcb9b50dff820a9b483e071617208ad52ebf15f5562ea44e5a758e26c5204ba239bb1c93e874f4ce3948319120687074c2a0c6e55ee56030&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.319581067Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/array_metadatas, Name: checkpoints/1088/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1088/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=ae0b557f140362fac39cf28c097e234374a9e70a356b97acd14e44cbb9cdd534ed94386ce6c72bb70bd39a578efd40d878199099a1b24888cace73934f5b74ff73d4160d91c5efb37fb30b1da2c962233f31eb75fef36a45d2cd4fa225a8cf2123176c48ff63b3cc7e44906b631e5ce26c0ab15e8e01fdcbd49ab793687db9272df10ec351f60776bdaaf70d3fb623325b642bb392a5a037a424483b9894797992c751de2cdafe20c768ff36464101e3f1941096fd7c50b7e35b28dcff873d282ef82bcd70823d2af6e48e4faee3392274344f5e4657ff87aaace57a5e6bbcf1f1abc9ca09c072a32a5e4b63eb85ac4646885fdcf07cd74b13475a2c0d0e657a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.319649838Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0, Name: checkpoints/1280/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1280/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4bb4f050ba40877ff82d371591630c790c911a28cbf57a94df73e854b0634309d5a84e59eccdbdde7f7d850f664b8dd79c794bb29599a0cb8f5f1d06cbf26fe0e79498a8309714935e4a53e44b47aa60e6d949c1e6fb350735814a4ba61289f925adaacaab84937394cf7f0e4f2bc78c10134ab7b1f2285cfba0dd3aa4e6cb450e779cb07b97c14afd946c0313cf67be94476d8bf3fecff3f40bffaa4638abfcf63042b515680b41ed3ad3d1bc57781be144af3c707b223c4e7498f9442eca9f1ded0c39a28b2d339b7af78b9432c99c7cc9666cea6fa22e892bb61d443bf7e3d233bfcb10e4558ce99425e80bb3196d0ce2db66226b5986d5d4f2bc98f26f69&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.319729901Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d, Name: checkpoints/6080/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6080/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b61828ed7c4f7ee7974d8d4cf1c597b3b6175501ffa542cfeef7069aacafc8029c8b3e5148b9d24bca0ed391f2ca40fec34f7c59bc7c3c4d85f83cb69f818478a7d5cccee48bd5d593a31e66a5bdaf210376e773285fb0fc84cb68da33b05b241c0caa50a0061c7848d85230f02215468f2b6e3985d7bb3199c503904867c7c935a2c624c457babf61881779f820fcf966451c558b8c2bdb280aedfb40a55e238129d608a67f13bb05dde9ccb331f0b38f402e621bf03b5ecf82cc356db01c97c055e145e6b62f49866d99420481d71823ffdb71b49b4c664d54b0617ab28343647e6091fe0847871c2bafefec2628d2a92bb3ece5fe02ceeca755150c074cf6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.319782421Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0, Name: checkpoints/4352/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4352/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=79bd5942c15e58d3bfa9a969588c2c3ace723046f6b68804a5a1343531e42dc562acc7940de25b675b91dc312c622ddbb430c2a08783368c455384012ec35d2188a283617ce691ccb1f0b6367cb41949f6537f907fc3da7f40b2c9e8d0f7b53e30f59408a2482997dcfbba03d27d5b50b72f0519f5786ec61e6ac106acf63cf8152c4b3cba537c505cf8196c85ef1edd06bbfcb57d2e13f3e0871cbb49f3b61a95a3feef45fffb7b548f60d0428c74543c639b872a9fa569cfab2f502cccdd2f4fe2d489f0f7558b3611751ed6a55f168d3310d11c77223c0d199ad7eb7a7321c112902175469709d832a22534724207abb115c89e5d80e1c78f3f00d2ebc60f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.319799131Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default, Name: checkpoints/320/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/320/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1533aa84aec195ffbf24b6f25e531ed6daf2e0d21fc963d7f1a6319249a529500f0236a3daf176012825e5a7c7a633c046c413d0064409af6cee6097121ef2be16d972465e306abb659f868f54e555148a9249c1a2390780e2cefabf1fb81c0fe7d3473015df369cce3a5af0ec95d083f506c4f898b713749c7a6e2408f588cc0a7113c4ff3f10159ab3a8223e04e505c1cf6823555261a91cf7da66af536451e5c1090eafb994ebbe8fad7f55f9e13b7ebd36e0642910981399828d77e56b156753ca0ed30bed91ff46f1b2e9fc29c70c642fa12163941109b95910c690b54ab153e821bdd358149f247d0962399d971962f2227e254c538ec7dacf70ce7282&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.31983686Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default, Name: checkpoints/2496/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2496/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2e7dea6c1db9d7af85f2a084865dea6ee3e513393c8a2c21f88d4f01d2f24b9591da86751ac23408e58d99be9eb15edc2a5d771a25c9aa29f8bb8c2450c7bc89ee713c090faffd9f6c4d3ac219fc2796eaec0460264e04abe6c7da8f577c3f90db7c011d6a848e18b767da18110822fada05e7adf21c9f8f76593a26212d2cc93367644d74729bf8ef0e956c030941de8df2eb6c7e2e319cf29cd06a4e64c8087ed53c77885b89bf9756694c28cb129c270dc20ce89495cd65285e968f24ed0855d8371b12893ec43cccd3033d0b05a13929c10d8112bed04ae8f56b7ff53f7dffdd37e8b15656d664c4a5a0c74fd3700b426d47cc86f01868021526a831734f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.319897026Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/d, Name: checkpoints/4928/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4928/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9090f039b53794cd5f3a08f182ecd4da9708028e87025c4ab33955b72d40a75ec4fdcdd346f3a8bea280f04df6bf785cca41673e7d445117ceb3913bdc19fc59916185afbbe4e594c6702725cec3abbc35922dc3d3f6569e4d6b18905b7ffc4635c2c5f1c1965320d2c040a5cc2c1d261cdd91a80a9981dded69d6659374fa895964b78185fc29612ecb901c9b2a73b9050dc15d1f98029477bbad349aba04a40d82bb2947a88e03c45c625246ef6c8d950cb1bbdc8ca8ee86de1e6059e775ec1aef369d360ca159593fc0253dbb50a3f4c1fd17e7ea054f2f0d9f3f8a04b69c56327dd803ae925cfc932af8c78d0a08264828293a2096d10a839c460f2eaae4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.319969392Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/d, Name: checkpoints/64/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/64/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=78834d364204b5f938d03284ff2774ade8644a537d4cc3733e15ca50ce50518538111842b4ca97cdc75710b3ea8e91114a2f6b740cfb2952e170f19f567c371cf807522ebc7de1c6ece50503bc183a410a9c7ef635f77ffb9b5cfad8b2f34dcd0e0ac24d8ca23294a4cd3115b64190f69b0c29100db44c0b619c2f0039bd113a52d654c01c865af613d376d8bec4800d47e95193a155d41341f8d568f80815a63ef65adac05eb7588f268c5b259c3e150f426ac31689fd98b5f3cabc35d86f2bd537f95494b3bea46f712f22a690a5b16810d6dba9c231cd72d2d88fc93e50d126150cce6144756558c5a37e9eada76571f621ba5f8704693f05d19fec375263&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.320108023Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/d, Name: checkpoints/1792/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1792/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=87a7f33a6cffb2234589068ae1e7a3fbac02acbd61296303b563ad0cb5b76b0f7fa427e65496a908c363ae7c5f5541d6a5bf8d2b51686dac14967a62637a49592c95db7460658ab0bf3ae87e2ff048ee9c9441c4b44f85a09408d0800ef7fbc197e9835ac66e0868738a3114c5f3e980eed364689c61c2716ba8120269bfdb4df103058833bbc2ce6442c8ec473f7179ba7fd93dd2f3ccdee5948f45dcce0dc93682a66adcf71c27a89e8fca91b5071fba3b80412ff29662cc2393344aa3b22c5c3ea77a2d96279d188110bb3f2e8cc0f707fc0c92c0b0f058fd6b1648703918b5012e5bb06c703d00568e2b349257c765e2eb89ad2fa78fcb7c08eb5834590c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.320178047Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default, Name: checkpoints/704/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/704/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=58bde679d46e8a3e90982102dea7ab7077533e6a5ec574f9dd266227790f52c876f8916f0f05cb712aa049ea58a2310462f9b3ff820371533bfe2f39db102ccfc0b9ddec924103014bf8bc2dd3bad7b8e17af032fe5a1064d0e966a7ae3f9825460bcbd1ee5e52c37c566ca7f38a0f713dd93502de0f062b344654217ce120923cb109e957db0cbc849ed6df24f967ed76807a48e30ce80efbefd981210c12444eb174587bce5a834b1edab8da6ad0ea9ac544f2b427ad9c3e4053450b1efe217bf6ed3d33ce3f39efc1a8fa5228ba2502b57b9cd306ee42f4524d2c338258bad7da8118439cb0028680363b50aa5a6699f360e7111c19fd6afdf3663baa0701&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.320256316Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792, Name: checkpoints/1792, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1792?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=906580b34625802e4b51a92f833bae3c0dd864e6c7d819ff68ba6a31799b3205b2a3c42a1779c38b6a0e80a3fe7acbedb068ea506ac6b82c04bac94437133201df810997a6dbf1b9bf0bc14d12d7a9063b2a2702b2f16936652a8e577bb868ade10cfd96408ea70a7d277ce54a781aa710fd59ba590b38173958d48e1aceb26b8b1c67e3a15a09339d6edb612c9c0ad2b51c1a8e1c5b4daaf6d5557ec84b15b3217a1f4dda78bf99669c86bde5d7ef251b28f56e4fd27fb104beea8afc3dc1cf7d68a0760594b4ddb50605bf9275f97be31fae67ced138faff6107387c42e5a8145d93d7fcf68b4d3bc03b47c5ceaebe2d98680f0ec37874920ea6a4c5477d57&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.320389892Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/array_metadatas, Name: checkpoints/5184/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5184/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b9157e641777a9a5afa9bc5443a4c9c9f191fe23b15a1ea705c403cf12fe9f496a643bccef30d06a2b89816539a5ec69786aa18073f306d8388dbd3f4173a099cf35e5aa4b1a126f80d79bcf07319ed34da94497c85f40262eacd3663ddd5c8eda778724295f268a6bb5d4346820fc72ff41b3c9df06f312c77a0687fb9c306b15c3478e024844533208923ba623623f4f4c1e5d58a2864c25f54679414150ba91b197d77211f12c0255c97ed282893610d09645ed33c1fe78e91f1e5f7f9bcd36fca7577f99b6719b4627b089c7ee23dc84de4726b79b85b745366439f9dafa65d9b920a2db216f0cef3f4800c7aa76c3d58529eec691dae485df0e2f48e8a2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.320440181Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0, Name: checkpoints/1408/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1408/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6aa4c19811a21242a941a9ac6c063b55112f79c07f9737d7213341e5e86daf93a4d9eb012a8c3ff8e3c60442b642a2ae251880d49659b66cac62a9dec0ba46b638f663f844430eb50f98469d715988045ae0d5123d3195af089bc5787a7e865a803c0b40f1c7c59e8229b7cdb37e6923b236e086f80a7d8dff6a9ae586866c4bac50891b538ed40325dd75c9031a0227512e8858a48274386d295c7285df01d3e563dc710a169b140ca9aac3a93be557db7aae74ed142403ff5f2cb853211fd5e32fbd91e73b9d90c5c2d23d6f0b07350c938b557bf297243b1606c506685068088679359038a20c7e8ae933fccb5e591e5992873836772945be9c2e54e5389c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.320681229Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376, Name: checkpoints/5376, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5376?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8a47ef81590765d64fd1133ce3a531ca03af071b63e517d01ea2bf9791b7510142a65d6dfecc4740d2b4402a3f7da92de836924825f0eb423ad0ff703fe52b21e3db1e281b92ea33ad39e312d2ab3bb588455a62134c1b006a118a32303d021de9900ec574b11cdfcba88de59f4818e03ee9141afce7d2d86c3edb897528bf7b7baee695232c9bcf1127b7bfbfe339e34bd14197a7bc373475137eb70a13064a68e477635b7b38717f6a4b056d32811e815e3c1575a65f5c5781e8c17ad54125c41552eb991940acc1384db71a5d796c4db16dc74baa2b211eecd80e8ce8731c81a108225eb4885ad4fe2832e1299f716a02e2fc50bd9b0da4ca78aef94af724&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.320685095Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280, Name: checkpoints/1280, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1280?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=ba9769666558dae26088f88349b17791b57049b4330c250251e110f5174526bf270c5284893b6c5357e519f2ef316ad04b60ddafd6438cc70d51320a37574e83b6b62d7e257c55296a40d50bf7793af4817609cf0dd294d80fc41182ee519920bf865d4552cc3f291dd1bf50b1b8a9950f9ad87b531ec426104bbfa2259fb12056fd1fb14a4436e5a81b40f17105e509e8b64093241daca376305eebcf2008583eddb2fe56bf67db4e5d48aa7e646d306c5ff7dc253bf6846d70c671fc56cd3d01601aeb433d115c7f926182d0d5d17548e9d65ae6cea4eec02572e829d907fa738f75c25429e0a280c5c0f615d7a5b14e071b49dae9c88c2c9f4ce849b6c473&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.320710816Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696, Name: checkpoints/5696, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5696?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4f57c276cc1243edbd4ddd36137d08019f5347d2757e1ae00688949920aa2bab293e73ec711c9351a74a518e80d175b3506fd1caae06e5963ef57063ed3211f94ff995494246d34e363bd43ac8ab885b8fbe0272623119b4844f6f75ad672c6e02677c4d98f4c2ec3baff783260b9a971ed33759dabfd52a8ff3874d72d0f11cb368f153f5eab63218b03fb323a4b5644f52ea80e934a51fc69a9fdcd8f12f8755d4f2411675759daebaad8886b3d01258498cbf47279c249e6c8f4b4585a4d40f6224ce548e5bd0f8c423509b0a5e921cc8fdbc2fdf0d8f9985918b8aedb1bbc2299b98b1a8ef174bdcc9bd60647b338ffa759325ddcae4932cd4adb9c379c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32069223Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default, Name: checkpoints/5888/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5888/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6c0d14c871cb578218fe716239b7e8b6f41e56095edc7fb2563f6feb88e8180892ebdf65f0e2b0a2d97d3d38e0cb3e16c0f44d96d6bd059c9e55d56a53efe1eb8416456b52a8717efeb09b8224cd9ea161b995eff1433114213c84d64c6e39876408a0a4ade62fd7234426c8adfd29200baa696d66b78b36c226dce4ed76864e6011f59cf9a41e14ce3dd58b9551ec30abace52e7dcf3c694e1ffa2aa08d773710408a881f2ca32af9b1c92ad319fddeb9740d744eca5df6fe0b25639b27e4e6b63d1cde9e98664be11ce21b79b42f72c21cd7ad49867d16b69e6f9dc3d96e8c1b8cb647984a88898ecd854b09bce30e83b79faa13e6013b9e9d999ef1dc79cc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.320754448Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/d, Name: checkpoints/3776/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3776/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8bba2bdf7aba2ac850160ca3beed4b0c3b74b7a01e4e258ea26788a989cc05e25784c7629b851ed12c6627c2a383cae3e2b144249224320a971ee8ea3c2203a2c0f2446e492657f2567957cff899670fba6e50b3aceb1a799f2f838309ab26b334bf0a8203df91eabfdd8b0530949b61bae6b439df17934f59a656b8405cbc26801449f7f93aaa7b061f491bd8993a7c28058ec30cc82d9f5f450d043525b87d09a8ca3d4239ffb881f93e0a0f1d84e27627c8ef8361e0c98f1da6aaab6fb33658460235d2d8f9ba2f7c5706ab50b152dc5a4f4423856b2f06efd40f37d2e94b8a525b3e8688bfc03179cba30ae499dd96e8c4535053777c8d985067d57735a6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.320753607Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default, Name: checkpoints/2560/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2560/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=78dcd3670effdc997541c0bed7742c6b24747d5687bfe17c6e94dd780d54fbf661acf0ec963798f30cc0c1a7493f8647afa1d1f8b91d982ce38aaec3fb135c46ed9086b7eb84f26aa4d98c291e19c0ddfc2c1b06fd3d6f01abc3fcfc72d666b86f434b431473b71b019a878bd0068b83e61ba407ae2da169c9754809a19ed53e7f2fb23cc7f20f54a37023bc77bcbb7ca81889f770fa43414ce55a9bf53b61a271430261b5d7eed9578f2d315e3396a6d57b67367fe5a7c6550875ca99c2d5f598e5aad13f49fc98fed53995e703485f1bd0638d02102dc52089d4d4c5d618d5d7f237da1c1fe3b1f62643d8d82d1a83381a893fad76393d4078fcbb32cc051a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.320970934Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d, Name: checkpoints/64/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/64/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=79533ccf414730bd00e07182b324251d131f9d90efcae92e8f7d1814e5c36493d143412c688cbe3343f971e1126ffa14a1ee7a699e3e3b7140dd25d774da4444d83e403e3e7aca186256e29cea9c735eff17ad8de60ee028463b226659c316ef35d45834bfdca7ad9377873d605f75555f3dfd80ff2c0cdecae08020a9d0ff28e2733b8d9a6924adb66bc99d749ff2bd047b8f95b34656f068ed4fe5e16aad929e924f20d4d1e40070fd0380c4252c4d150b0dd4e49609b49c8121d5f4c0f40cd858f97e0e89e66f183d2668e68df1679da25bc362c317217c93d9c9ef1cbe36ec79a1986dd2b2db2c0d01c25d68836ee2910ee1bb50cdf0ed749f7a52e8ee41&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.321200107Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392, Name: checkpoints/3392, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3392?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=01d5adeda194e9802d289b6c8ecfc18ac50412f7513a10df1d1176fdb2b26c6065fdcc6b60b9c4d6c92934d983e93094307c77cf1a5b68ec44b41cb1cdc3f5581026271278105d9c30852f16004d52d507ec384af7343e055fc5b305a63d7db06a812cdfe56a28809ea9f4e68ae6a0fd19ef0ed72e20e7f642a6a457b3a3e849474317eb4ce28645342ff0689c5de063fb313e0bf029c04b57fddae3f23a2a2dbe97204fee4c8c72a7331270eb83ce75fd46794cdf1c4cb4ce983898219d3033c773e8333eae31d094c001cde2e10944050d8efb2ae26ff2e20662aae95a1412fa64a31e79719fbfbf2913cfe3b8c0a682a90387e81f289c0970d3082bedebfc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.321480941Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0, Name: checkpoints/768/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/768/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5dea9480583a7e9ba949ae74e499a1d3151262118e690cbfa4e7ae2e1f0ae39177ab0499e07973f0fe862ff41b46bbab566ae46f1e5312dbaf78943dce59600da64a4ef8d7ec6c7fa5479df4686ccf850026749be36daedc544b85526544f3260a6a5660c242324bc25f6dbeeef3c089ee53dc444eca255948be32603e4cda7fea7bc23a141fe4fc79f488b997ca5dd26db7693d2220fad76ba993d66b3e9a741b8fc72392579bc6a3f900cd8e4c57c7066dda8d0a04a24816f8f4c954359fd8c6db735a5f2fff9c42bd8c5911d8c45096913e493edbd508a8f399f9230f9336ffbfcf90099e4c02135591acff627002fe6398d4da741a39ac95e149d31a5349&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.321515082Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default, Name: checkpoints/3904/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3904/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4101a0826f5e14454acae7548dbdfeaf08bef130990338211d6321c8c1514fcc95f54b60c217acf6702a66d83672f3116057aeace0de149ef1bc5cc037819890c34b67b0d079543ffebb0d1cdbe1378d2f42507286772d37027cff9162cb61bccdd291ec71b2d4065a9fd2504a054aa117e4b68399f72345edf842cb78d2fa1ac1ab8fd6524d9d54db6687c315c0967935024bd655e58eb6e0ce98e2bc8693e06e0ef50bd07a066c3122be5b0cf7c66204c6352fed415f24f4bdb3b881a997c03503b974ccece3b50491ce7ac39b9d544fc035e11032524b090177f642ac6eeea3c48cbaef9fb8d3cb156e9f7ca5e49db4ab9a35703ea27536e308853df46d15&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.321522468Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896, Name: checkpoints/896, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/896?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9f89a1f5075b4aaf4d2170163ccf5780bb95462811d405fbc7a026a2a87fe4bfb36c99a065edab6c861a1ae39f6f20c02333cd5b3cb036c90bc15d2cd351d49b3398bd49bf9205fc7c87f0af5ce1116b30f046f841b2236916015c1e018ae31c9ffcad2a04a5d3899a4ada09bd623314179bf46224be6d8bbcedac81cf4505f98ba0e20a3f2961b94ad664f94260e349f8f88fcff3a2e5af751a7673ac7436b192f2b51cc43a3169e1539962f2379d499e88a2bdc4e483eb31c7cd3119e3b77bc0f6d26a45d3e084cea8dcde1a679fd5691ab629e444271135c4ed82165bac078191d1eb34144e0f014b36513ee56c2b6892645a211be698ff4cf25af3b09708&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.321605933Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d, Name: checkpoints/640/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/640/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3f7a5e2880e2f6b17972b7e8d6e18c3afe8230d90450082dfb8030258c9bff024b0fb22be7348df64131a01d503cb3d5f13c0299e6d214ff415c36c6fe076a038cf283897e945a30c9bcd7844d3c7163e15e8d888ebe5fc905254adde335087e22f6b55ef8c099092cdf0072d0508b38286cd8d45c6da65faeca94ea28c71d393f13845b2e65de8dd79f1b5444bba78dd58506f27b2480466d3c3945058328d054a55a17c306178f4328f7a5797266cc1640c10b31f283cf2f74bef874407b1a0e5a30f5a291d96710538741d763076376faf08e8d93e95eb0ab17a06f782c84ff5946152e0a9377640532842dfbcbdf78b448a92a1799ddd55856d2abe47363&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.321653217Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d, Name: checkpoints/768/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/768/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8b35e6f020ae0114f192a254d4c7695f79ac07b16ca980ecc7b5c2f71e89dbff73a631ad919b6588d53924ffc06b0da495dde61b32fe4c60a4f77612dcdc1b08e20301ad9815c41a9398163c02ddd7aeea8bfe494c18051744d7bfded3d88f96d685c28bef51f2022a928d1016540e942fb4faac80fe0039ff40443e18297c6d67f5b479e067757caa77da1e5013b2f0fe8c03f272aaa918e5d6d99e8b50a62a308511ce7a371e2937aab04b7be43158c4a85da3ffb988574fa488712b639970ef1cce779dcea4a63f484d205aa4afacd2986290544807b50a200b5912e33f6ff0c919e547cbaa2795ecedd45e7071c01922a598bbebd16524186792eb515d5a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.321675372Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0, Name: checkpoints/1856/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1856/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9b84d731fbdc2bdfca34be9194981ccb1ec981280b7f2126fe33985f7fc2a640d767fdf592d2a11a60933bdbb3ad53b4dbaf51f15d8d26fccfe0d9b4442f46799896f8581d303d6b7a21fddc368e9b561080a7b40fe91d699fcdad5b76f9d820cb4d2eea6fc55e26b8bd5e721149619b4d6446bdaf3ea80f28ecf0acebc80a60b13e09953634b773bc482a32b709b62789d89e3f33f2ad0f7c7241a4d68295888a9cd866d31d29c5daeb9575b2477cb5c22eed7815f6f56c0e06643069a0bffc6063628fd518d0d9e5ff0cc0e956a8b0a461d9526b0791561f103d8f2e28aadc749947bf3753b0c09042e7bdc825ad67f900d72c1f203ea4f31d31ca54eb5040&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.321847029Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416, Name: checkpoints/4416, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4416?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=38c33365d560969c6d75456e8df67d3f8227e73c5646af2820db500cebc56f0d370e6d527d2cffcfae87aeda2ca1053e35734b7e8087f927e98dd8093f9d967d353c8e35b34dc2bdd39a5c31e97abddafcdd4d0d724c7f3630f8577d4b6fd38e95e217aaa2ebbcb0ed96043981afb871efe440ec32a8e03f582b7ad2a6ed050553295655769a87f45a52aff96e691a3f3da01f4b739048a06a1f0b9325ced7590fd1a0684ab3668330dce288d7411677bbebb35aa3117268f93c46bccbda4c3c2e75ef7474fdf1b89bf2a762f0a11a922540d519288a5faafb15fb64854695a0c2566626b013d75fb6161dab080eb9dd71d68a6f0ef33030d4d864ca0e8b27f9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.321684142Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default, Name: checkpoints/1280/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1280/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0f9abb1e0113ad20e035e3e0711b6c5d1ee935e779ee412d9a8b72fdfef5e2bba23412fdce9d315a7ccd1b3138a679b514fcdc550a79f2c561860b58414866be40290ea6e7a92a53f3cb2224ab02aba72b470f9560fb6a85e823ec7fb1a4e13e688fcf916cc6e67bb3b54af14e604c1fb23daddc9fd1038774ef3a9a93fa60372ce5de889ddd7458fe564f84cf7b7be73ae609ecace8d1daddcfb5764c40421a2b337fe24a8ab1f29c32995b12b6dc6a2dd5a945ee8e4eb5ffb1e4562ee520833a1c78b6eb1269866cc1367c00300490d27e0e11fddae341eebf3d14824bb07de56c2100da92f60d7dc45e76b9a67af74185709718cf503710570f6030f84017&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.322094989Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0, Name: checkpoints/1536/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1536/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=23de6ce038b7ceb1711c56b22d34eb49901a8e7847a7a303226321cd2409f3b3e833f195440f8992f097853bc8e836bfb7c4ebd94c25ef2500b7211050fb7807ed1935dfbd9874ac29e63df536178b488d277875be0a6d93d3ea70cde8e1597ef62ea6aedf23a66d4ea2daf5400487929c7fc9c457d84c0fcff44e5edd92fd6a6624fdf870ebe6b95d2c2acf3496add2abd7699b76341cbc8e9108b87fad17876e292f2d27964a4a8ba352880da218968b1bfadcdea203cd9fff028834724cd7155afafa9cea4fa112daebb159d3ee7a2ebc187e47f8517ac11edff69548ebbfacf40f62a151d5ea775507f9b77adda9b6871147502dc58ecc90f2191f072fb9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.322266825Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0, Name: checkpoints/1344/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1344/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=09e06c529fc984096d7e28b592f01b11c073b4b02eb3df50ba432ac90d775e9ef8f3d80c6bddd9e25d99101f16844c7930aa698300c005a50d375c99daae78e83c83f2e572d190f802a87d4502df04a19956976f726dfcf5c9a2b06a6a1af07bcabc3353756e3b5decd2a1118a388d928c6842aa51c80216c7cfdae01b2aec5c7029a0fd2f03b6b9675f6aead8e4e09dba0001a9bf574267697f4ebfe622d6cb880ef6a922837851f23c098f76eae6187438d39d3de312157c294aa7e4004adce50df2a2cac79db58001a5db81c6062021afcef0c8a58164ab718275d03efa48927f3b7383c3ba4ac2c640952eec841a0530a5d9da7bea250c21e6e7e0373871&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.322376597Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default, Name: checkpoints/1216/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1216/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=96e9ca14dfe417d5700a8ea2075f2b45903f815a2c7b4dba4e884824501c89ec5ca1d60aa764b2a05ff53af8f0466ba4d41489d5f171597b400c8f5336d1ec8c33f1f77a87c92d77840bb5a1690203cf5567bf7b26c05081f34da0a83644dbd24e7fff041b3aeaf1a52ba86a68146494ca9cd4ab1bfa97e6dfaf03eb67b2e8e448e146e511a909b22461770cdf439b6dc020af1ed671476fa7ae596319659e0f345b986a299ad8112b2b92ca291288aa876e6d24e0d47afb0753e3f315d50b28f0c6fc0abf9bf541ead7958df07899a72e59a1f28a70aea3f40ba7455af3997519e3c5f10125debd94209b58770ce73af88659cc5ad987eb135e20294f234459&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.322410209Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default, Name: checkpoints/5824/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5824/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6bf56dbf8d0b764dd8b7f7861da01f3b72ee215176149e485196680f5162d258e47f07175d3d7ecd138eac752ba827bf4deca6c8d59f27f368cdc90d17bf31081919bba63542fbcd658ac52b75a87211b9ffc90c9b9dc4c5db4befd721dfb08aa8edf16fe1f94ed0cbc7860d8efa6bb1a0504d22a6dd5ff2a5fcc7aed6ce72c4918d0dd21996e211806d1163588bc1e590615fb5777ba45b92f5140a6e9d8d098c650f8e84b4e56da9906f294d32695d3245e8c1d8a1a06a8d1e98f8e31b6cf3d07e2b537ff2e3664619bd4ea8a843ab9c92a3b5e01e49751e2a0d15c6dadc2ae66aa0ac4ab139ec1f418413c026b2f97ce8731ea92e0abb7951bfd11a60d30f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.322302703Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/d, Name: checkpoints/2880/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2880/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=550f0cbaefebce759dbab700c3d91813d6675c44018868d82a96cde0f6da903db34d74f73828109f9d47f31fb18fd49b21c9fec423b4517a152fc8cb32819c17a2e85b386c9d902187c4ff455b302040efbe2123cac265c1a9da8724766a84f60707585a3d8a8434bed570acf9c4fed3cc4ab8903c448d65f6bc9ef9dad43c26b8de9fc792be73d271af272bd64cf4dcb1f8a2f53543bb5482da1f612db051e8635458700eae8da3d4238dbd60f7442595064e30a590853fe2985fb18b6bcd1f70a7541979b501ecd6e2b2fed23db6ca267c471d992d95773b9dab38107d86a288796d8acdb040b6f8d01475c507b54ab3ffbe4b60efcde88ae13ea3b79e6293&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.322572467Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0, Name: checkpoints/3968/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3968/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8509bbd825014a5c4af4764d719ecbbf16747cfe44b0e621fdedd9007a387c0cee657a3fb5513f60c5bce6bfaaeb8bd2dd297580a962cd7ae14ac03c9a50a29ec27dab889509fd70f70856ef998d089957d26d73ec451f661746f7771c8fc2959ee944634ede0f8ff30106066d2c89bb946b5ba6537d4b67e2b22a66e9986f77a3deebb0c81acc1e34a1dbefcd192187e857128b330c40fe7b446bc35de6a69225a2c112be7cb1409633dd6508fbf9eec07acca2c577fa7116815447ef54aa07938c74c1f7ba425303aba8ef4c0c4efbfc3a0fce28197fa5e2bbdea5cbd380267062282feeb247da4556fef2540533b3c7d2396499a25727922971360f855888&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.322635623Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320, Name: checkpoints/320, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/320?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=82db80d458a53ce9248e60309884eb65c32bf94a4ed1dfc337fb37936be87a9332c5af562d21aca9e5867ce969da0b7df189c1b5472735b9a6ad4abb7fdcbe638c4591a144165fd7b5271f9d8cc0e35f09ed22134440b277fd732c7328767dba0924930433e3d72c054647fb13d3a78ac3211f5f7bb66b88e8623bd238da3c0c45872911bb5ee0530ac6fb01453316bf0a21bb40945d30b2d2c189240f0c3cb472e7489c6ad50243e9f48c496012f558b45deb846a3aacfb80164d07bfafc73b392322663019fbce17a66bbfc36c43ad1bcaf8c896dc6fa2450e5d68354b64dfd7a8291878b7dd741e15204464ce45d3292ec37bcc389f7ebc8ebde4fa8424fc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.322909501Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/array_metadatas, Name: checkpoints/1152/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1152/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=73e0354dd6a1083add7d33857bb935ef8434e3d03e0a64994ae1ade4212afa1f2760f75eebe77034e61b2f27519ec8424b4910d0478738d5d79f278bfc0246c02ff12fe5f401e45b6300863cac616de8de648b8b930c16ebbc634852ceca765ba0bdbfdbe694783f294ae2300242a4311d9db36ddb3d55a5e2131dd6fd8905982c267e8b53e9f7b28566709d28b0020e0c1eb76fda24e4f69b9eb9bb283391a6fd64911b5495f41ba75bccb85c8b846a7d75d24a17901ae972d2086594040c6af07874740a69226f5bb2d4ead9461fd742ff83b58b9446dfc1925cdcadad198ebb9c08db1ededb4bc4155389ff9113800395c6ecebe7cb7d898c92c28b7d3d5a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.323044826Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/array_metadatas, Name: checkpoints/2368/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2368/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3bd59ea94107027a8aa0ce8e34445b9b6b39fc7ae752ed7630de74ac4c12beb06e120476509c7eeeccb764af96761f01999dd69fe39f3c493c35ecab8356cd95fb500dbe6979ac7a28ea79ceca155ed5ff828ce58dbfeef6f206c9bf7423b3f0588599da592e62c6af5c52cd1fe496cd74c7f8c67519a04a4958f93dfe06dcfd3da8386aaecfd345145550519c791b84708bd36fc06716ffc1d2ea5372f2d9c3ac8389cc5a89207a44aa6050224739d4787541f358b49448779116977bf490efdc894d1ee003490d362a1292bab22a97c884cf8bbef78cec8f340a1e1fd4f3194f83ec2099121e8c38f79f297b2d25a88327048f91b02e0cf071dfdac98057e2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.323067043Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/d, Name: checkpoints/2048/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2048/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6ac4a5546bd71ad3f9071322fa4ac245f047cf6ac94556639caa5d8d4bf0f344023eb09a65931af0d0f4a44910c3b899e87e806731c90fe96aca770aa41e3e609583a81a933101385217dce1fbf267bcbbf39fce232e96a909a7188af85469a56eaf641fdb94fb2fe69f701f94dc6ea6c0bbe2ff7522c55818bc6e7cda9843faca20585026410012816ef471d3d2962a4579843cd2ad9b6b6213d1c152c7b844d8296c2aef090e8087bee8d7bad8cec992658ffa64b2a278bce4edc1a1f72e72467749552101107ec2f72b746eb1fde0a95293eb3f602d6c42cc73ffa75ceec99841e5206b88778dd701ed0bb314ac463a637806770919044d8e1d3d81986872&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.322953731Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0, Name: checkpoints/3200/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3200/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=34f542511dc1736d6af711e0ab5e18e83883c9a6e601f5e2639b3dc7077c7e62a50d7034bc8ad24b6ccae46877ac72e34017a8a757065b3566dbc8dfa0c04d76d15be85c403c64fd84baae13d08b34707a12aabd083fd61a711f41d4780a1b6ddb3ecce9522c6d1303fdae156283e8a59a581235a14292d153e7005b27bc71a4a4358063cb953f69f0419d88f9d7997bdad3474bc95930f3566c62764d16aa58658e9f6a55167e9bf4952599aeadef711e35fab087ae701619f5102d1f9ab9441ef713b1dbce338b7253eb5509ca53147523354a961a84165fdd8a0bb95d646ba4eb82b33ff00340e25db88b6068c23152ebb4f8753dd899562a5608a94fb0f1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.323237509Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/d, Name: checkpoints/4352/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4352/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2d2812c5101ae2b4811f0d979bfd254230db77ad2d4f9c252ea11565e9ce0bd6ae24e05e7bde1a56feb1d6d89c018d252749f6f8c96ee600247aba32b28709c67c149fe1bc4fe7afd9034f122b2a02c87b3b7141e8dcf43093aadf377668864f2e6e770f670d0ae87c0c90382ec830c3e82753b42240d7bbb0ccd29f4ea921854567ee455e00d7aae6efa9b405d9857c8ddb381098c1b5483b0471c7891a07ea777378d6ea810da38403a03002247461783f49f0f2a5f4d60f7bd36359a991e6fafa508ff70d6b4dfd652e43d2bde751ba8c9daaacb59e1b313baef1fc9bd5901ad49d90847ed2d5aba113fe2bfcaefe62c0eb100822042f4f3485349bec318d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.323280674Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856, Name: checkpoints/1856, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1856?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=190ce3b6616f181f603e7e443e3de78ee37b67f4762c02a2a4cb0c93774cec2a9203e564eff2bbcaa5a426b972545cb6fb15869bb419ed92585ed3d8bf0b07544ea10092ee68610a175ce430dc30b8e8e4d737cf1adb8f0d61674a5c38275baa7776f70e6cdb83e45955d8cc054394e8f217d7e36128ea3a03b2dbb4f6826d51731ddda0ad9d28a657482c8917fc002c44a06fee948371ac018fe8445c2f3c42132fd09c758aec1c4de604c234843670e55e6821f678a7284675944fe7fb69f3e5f16aba999a076244082fcdd6d3de1f2a700169d77fd87073376f9c67831d677486a4c0676a8417a6514963c3113c1c84c3bfab6307e86751a73d474d81c242&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.323249876Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056, Name: checkpoints/5056, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5056?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=85645a917db3ef0460d1cadce36cfcadaa18920eff6517d8d6e942067b3e704a26bc30d5ea1b8fcdd1e76fe61990f967fddd59119d09005219ff8bb73a31fe984b639d85c1deccb633ede8e57a9caa753d6024df692310a827ac5999c50e526cc2a7436db10a9e1168f7bea74b2294b9d4dc2e78e2e86e30446cf310c1d5a48b59a0c442556d290c3594c74fb2a26db3b1f981564aa707983804e2e2310b4c9947e41dbbc44898fd9911fb039b4fdec412ce48e7f56a2dcfed628bb7a5661bed0c37520953683a3cf6e796f5f3c51666cf0fc47d2164a4f1b2f45f4f8793a8873437389858a38b3f3d3c83f4d14149174f14a136f92245766c21ef205e657cc3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.323414976Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/d, Name: checkpoints/2496/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2496/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=99a85d39a4b43f8a66f87d100ab0f72218ac1d0a87fe40772d47e2e3e50120fae347aaafdffbd3b4854072b42272101d3132bd3855a087d590b7e8306e653f96474281ddf3c8bd8c99abd1b00ce6382a3c69f8fafd97edec86c2f0df6fbfe544b09b7a36a558112cd1fb54e4842364f0d3f912a5331dabf9adb39eef17200e8e5a96a73f3f5c4c88a1107645b5bdd05d3f9cec0fb6c0e8cde13d086d5af80c47a51dc1a556ffe6a70a7e0d2cf7f2b8682b8624c4aa965408808c0bed16891d1b74621bf06de22ad82aa33bd0492043414fd9c8514618e16bb8de493c48455e1768b88d09fb7392275d57127f120b56482d21f1bb6cd8c36d455618b8ca538aeb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.323298437Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/array_metadatas, Name: checkpoints/3008/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3008/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9c5cf9109fedab92c96611fd4187cfc79634ce3a183e729e48764362269a2415a390f69a16d63b6eef9172e19256913552e8dbfa9c5402f13bc774ced6b449b5c867946948bb699a8b7beef6ac7834eb31c9bbab887b8d506b02d9a10c2e64d3d421327f2a62d721f67cf50b6dfba008344f4cb601286c5eac381087e0ffd7962270aa9f39448d6cee08135c2cc57ab8ba193f6a639f5a768ad790800c545dfea01e468534100d4a1c87b8dfd8b035a92f37f6aaf6d8d18782295298af60d46b2e7812f4d72fef5b21c44b6b3e3bc5eb742b43b11514bae00e5adc33ea8d9083857e8f25f75bf34d741b902c073ea556b9d591ad2557840f45ffd011f2c4bd56&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.323341113Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0, Name: checkpoints/3904/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3904/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1d3960b7c9edbb063cabc33a814214906e87d548cd025da1882d87d881fc9c9fab17e91139f196cef95630e275ec9c9fdb847bab7567f85ecd32e65b2fa2c6e1ac95bc557b53049b0cb1f8630e7c64e84fac0f7d1f713cd9f319cedcb32c0c094d7b8b858db30e53088bca5280ef38838de35388c637acf5ef300f415799a12f00c672c74f9c5a79d74ad4948d0516fb4c659c52aaa9450f8b448ec4deecb1986b09f89be942ed16175726cced90d52f9e7f67a3adb302fd85fc2640c30fcb768012b4481ada80c3f32c7ede3c4e5f618720779a7c95639ad846496a4b72138ceb5cc2f3244f97e9f1893999eedec4e8bfa1b5f1fa3f87a7770743f48fd5a0d5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.323737616Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/array_metadatas, Name: checkpoints/3584/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3584/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=64501640fb8e8cb525c6d356fd0a10dd0c68de7ac51eaf54b39be1deb86630a136f23f48a31ce0fa9ca591a59c559e601d20a2bd8049ba8f8887d860d2e2bd2506ba3e762b67b96fcd980ba1b68ed727a5c1605f404e1922354256863985544604f9bebbf32468412481f9688da3d895b9a777a2d647bef88a7a689ba5ae2fc65cf0132404269a686ff072f19dc64c703c61138ffa3f96260e7e80142d54cd2f49e20818c7366abaea9037b150f953e4035a3792db9b2aabadfe1edbe043a38922a9dcf124de98da639000ee86097ed87823c4f8229abe32379345af4a46b60483b74a9d1529d0b682c5e6be1fdc7f3c3add43d71af0f7097066764ec31c3a41&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.323893581Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/d, Name: checkpoints/960/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/960/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=09abb006b681f2692ccac04c16e359ced815c9d11679960859903f690f0caa8f711e860ef229e41ede9d32f95fac54cfa33a12ba0019eda1be1f38d69c9fcc84e4b74daed449e143678ee8bbcfea7a55a36bffd0c6339d9626bb020c07e3f11f54dee5fff3d54d01c960f52a39d341293131d65696bf77e81de54eed900c8c4e15080f5244d678ed52513932856c378619710f64d05296188d135d5b66bac4f8aaeef0bd5f300728ab48a82ac3dbcd0bbfa2432f397b1e723d8ffe3903583b1dafe72d75a924679336aeba0a0a004820eb6540820deb36121ec8574564af3157c647c3e76a99b5d034db91f130246e3e04c3998a6e589c20b65e533e653a4a29&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.323937584Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/array_metadatas, Name: checkpoints/2688/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2688/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c142bd36fe2768a6c6c6b478ca804b1e776b262ffa387809b86bb091e28a34b0990152b4ea454ff193e1c1c6d3fad2fb648dda61b9e4cda42830a7a11aeecf0ff59c70eed3983ba5b43583cea149ec3d77209508ab016465f9aa84a1eaa41bfa6478800f51949f2c13cbc5942ab07d51c686890534a574632b9756a660594eccef3e7202e8a29b96fb54b9f380108d33566e2114c9ca9569415c08dbccea20a63a5ab53385888cfe7b16e483dd4f9354faaf538014a186ac238ec93b9132f1aabb822dd3918f02283545eac0c2f212048cb2a9ab2ae57038469d25d0fe1dc7cd1c3368866c240dcb489d475e1cea3c295e481c78b9390120a8e30f7de0906556&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324040852Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/d, Name: checkpoints/2944/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2944/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0a8e8c5faf169443ebc90569dc0dc72bf04584d0e8a2ea78dc0c461689eab768d391e275599e9f712a7fe8cd7e26c9c5769c6ff7f348fb2574a43872644fec227c1dbec0948bd13c8c23e5e4dcc83b91cfa1c88d8eb2b77b67050ee9d7ca96acc51be06e7acf7f94f42ae09db73775823352ea21c7da083ea8415fabeb2d2efc8a0d4e07131cb6a46912de14dfbe27cbe213dfb26cdf60ba1c2f1cc6329d28e816a93c78d08c1d35ccfb74765f4bc5dd63c12b833a759443ff0e487bb8ba9936917871c7b36af5c272fa6c2528e3fd41c816bb601c543b28851baacf3a2ac933fbfb774f0ed9945e56ec12bdfbf4aef5fe7f88fb41e8d57c7812b4ad72357e37&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324084264Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default, Name: checkpoints/4416/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4416/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=41fe165715fddab59cd802cfa057e5c6c46c3408248bd235c4968c136c2482b42811ea71bdffdb8df1c48eec82ad198bfec1fbc08e8b6762e9bfce394c79d5126f1ef6cb1afe171f35c7b06f79d88822185042b205d9989725157fae5982a894e1574072b51970a8b0b4688eac62985c532515ccf3461225a9a5bfdddd7635a25af7d5ca31e2ab35931b3306295fc601006a4f84ccb5a2933dd05735a3b148317ee94fb67bfcc7166d0b095ffc535d652879f218c5673968c0e44e54bcb49e2338a6e8eee8b119f523692c13626326d43b60e602f8050a1f82e98afbc14b4ce015c35917e5a5a384bd385c40d80a70567f0553105d155f82f010f8da0ca69c43&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324107874Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/d, Name: checkpoints/5760/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5760/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=023e4cc0724588b912675017170d8f3b3a9743e1095f59a5d3a7608f24b785cb3ed9ad760d768e5a89caf420f409d96a11c0990dd452b7dc00ef0507429fdae29cde446f92f3d3b2d44fb321e7ce8faffb942cd200fae4d06eeddab406e4eabe9f2046ddb092e2f89a3dc376d3254f5c8febf1ba9a824878312734f912bc5de25a90388195ed80ac7cd3f6fd874b224e99b46be79a954261a29ca517ab1801bdc85f89a3bab16fd339d283b55558568bb5491a3ec45893670d186bf0fa6fadb514dc755eefef114befd8ddab5eeb94fa222d21b8b48b04f22abadc4868108b655a78a0dead2a481e5dcad6236cbb051e89338bfd12ee55e9db60ad79c5ab31c7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324152646Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824, Name: checkpoints/5824, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5824?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=34ed2dc286d7ba919a195e86e7b384ce8b154da5d9dd7fc05d9ebfff223f164665c02c4041b359963f34994cf9721eb5e69a0d9d38f3eb7f546f3035e4cdc05e79815618a8d550bbe008d9917c17eab0b4e03b7ceee34c3e3ae3a5690882a6ffa1604bb28b24131107d41143e8da924685fc07fa45909fe5756ffacc1ac77342093a0bacd73462bd22959513c169c342c8036392eaae643f6801e5c06c0e669b4e91153b3dd7ff85ab214559018abbef678ba492e79346d6143ac38d1107f1fa30d328ca92cb6edbe0fee6dad8a0c060d932e5caace257671a51fadb14a8549d89487b12ddc0f95f7586a9e4b4183eb6ef3ab74bea376117469b9f0466e9ae43&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324298981Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default, Name: checkpoints/2432/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2432/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=36cdc2802512f2cac35b588500f2297b1740b763bff7627cc6f129b55ed758a94ae77bb1ce8d8d97271a6d3477817bd96c052fc1e7078c33dc8bf376bbef66c0bf777d8ba4f4d776efb09d44c3ad82a8fa1be3c0d3a4c65ce0e7bec70ff13acfe849ffa4ab427ae660447e25efd6020d63061bbc5c43736d9fd60185d114c7a78763c6b5a27c2485c9df6725c3c21b738d6492510370e1878f948cc00accfc533770de7b9b38bee7faa37fc30ddbc3f441419eedca00ce88af31a530bed96302de14b9ef5b47067754e3643baee5dd3d35162bfe1547ce118882e427d82702762607c7cc826c96e2623b11e0e46eb892529283fa6c828655b9f47a15b8602b7e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324623955Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0, Name: checkpoints/64/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/64/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7551223621822c8f476c02eb47444f28be42379d4bed2a7631ff7952f352b7de52f556b1cdcfd27629c6a621ced6e462e3bcd4b2ef739409bb3452fcc22e34f65b6d653602dcf6c90e1e772fac6f98f1a5c8446dbbbcd0e54dc876b98369a7534520659bf1326dea42e12d7db433291bb9249b9221c700b0c5d705dad568a7b78fc3b9f2b47830b423329949f9212e49cd1cd118aaead7c720ac559b3f4a2eb681f04c783e0b05953eafe26bd22cd1f12a4d7e66e3a241a20bb2e1b5d461e0b98f35e81865f98adbadbc163bee09dec526cb77621119ed5e88e75538503120f06e9cedb889a8ec288bf818a636fe146fd2eefa5931232e2dac4628cf2dd0f9fc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32465732Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0, Name: checkpoints/4096/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4096/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1ace0d927b4f34847c4a9f56565a11ee22bdf78261c87e6085836a452290457171595d919f6cd314dd51c2af95b2e15960be6dce6232127c7c2cb3f25469da9d58513bc99d3274304271e547e945dc326caac8976c4adaf9a500280a9eea68a23a2a79805d1b9e968af7ad837270165494e70197885d359b9154a2e336fe55d3cd213fb5480bc89f085d98d4d449ed7c6cb38ff0cc33c81866ab59e3cbf34bf2dcfe8089e7f8afc648ffac6a82a58ef0ac31b394cf5d6b1e52739353b440f04286ab68e686c388fd7bf06f213f95f339a94995e62526d73752561ec3e96cde2ecbab86c0edf50b99ca050a588584c8fed9932dbf8d1a17c711db5511c98b6363&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324693203Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304, Name: checkpoints/2304, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2304?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a7b9471607ad0b5dac647026c924bc0721fd6ba849969607ac9e4616e80e94d96d938ba559b1403f9a45a10d8724972889d9270e4e3efd88b719b9cdee948958b7f904658dd5666df75d38baad2da974f27b55aeddfaeeb82774bc010fd4a950945f4adbcf1faebaf82ea2f739b3657fc170254fabe5b80882ef5d07fd5b99e0449f5cd8dbd9fbdb6486c792f0fe6cd056ec4850182d8a8c5c05ea65234554bf20d6a46d42aea721b22045284318e2a8d9feb6a32b936cca93203a322064f97e78dd5aa89ba81b7633eda3d99d7cd4d0252a965e5105c5554f35d85417c52eda00cee8d3bd03a0016a13372cbe65366b627a5f5dc7592fa982d345b49931a1ae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324712972Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/d, Name: checkpoints/5696/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5696/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=36cd9949404bae941025d57dc65bde64d1555288d9885b7eeabd2842d59818d43b81fd66c59b15709d180d3f3382bfeca3be591de5456245435c6cce66798dc1311ce2af51f5f9e16cbbca04326d9643b8552318df06886570bdafb3726382c007b0f155b3cd19ae3c5e883b125229ba2cff2852aace57841f95344c1a0adf525eff9463159354301bb250869c32f4dd6d5586b50c6143d8643cf36364d3dab7d6236dd044ca94d56e04e668e6edf58c2f21ed807f6cffcb03682398c51d078bf529dda7c037202bcbb94db11ac729e44e812a1757925f72945d2022c7b252fc06b8b2d4e097a285a666098496753329376e6f482acbe93711c4932d74d5af3e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32473087Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d, Name: checkpoints/3136/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3136/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c3d0e2bd6bf63c821acefcc63ab620c24f6626b0ad870d24ca7b37391b4fc5ca6f26e248c27c1e6a37586c87a22c59d5661e972579c6f2b9309194ebd01631696bb44158baa2f0cdb98abdf7b80419d8a511f7a16b5139ca15cc846c0ad18066933be9eb8a25f7041db2db940d0a38adc6d9bfc2cdc91e9277dc78b033a301f022695ec9090750c72b8c302bf31da9229ef8a6fe80e659820dc8541b50dbbabc73482ac8e3cd203d97f696f56c535054e1235b5b0bce0777a6001ecf3ce1942979f13aaf38cf4ca1ef9f8fa3604609571582ef3a53219d389e57bf5062d9ccd3d69e2c25fb50f1efed6be7aaa1e01bd446efb94a629e9d4261b7dbc52ec395f1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324758954Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/array_metadatas, Name: checkpoints/6016/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6016/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6e4686d3cac14db457c11fbca9f4b4a95b93dab05591762cf96d0410b59a3332946721a9b1b00e944cb8aa9b71932cceac4172153d1dd8d0d057bc31f4e8a2da6aa0496cfe3baf7ac169702d3f6164d19138124f9fefb1a65d366712641fbd41b8c8bc334df7bbe45b49d20d64cfcef5b13fec1d741919c671950b64b1e448681804ccac761f1c789412f882e711ee17c71edb99e7e11534a7f63a759e1b856882e1f1e79823bc58e73a23691cdf7c0c27b044d8aed759ca80b346e0eec077cc8aa33fe72c109b2763de160f52d66203bd40c71d4f9c25d717e79c88f9db7f4cb10edeb84b632f055cbb2e1d32f49191fa9fd9bdca76b1db5de4ef6f1f43fb75&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324780828Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/d, Name: checkpoints/128/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/128/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a5337d86d53b0443d06eb5e330a107b1ddab351ae3227c7af2069ec442ae5900f244cce12f71b551dccfeed30b54332938b532132f9baf945cde58ab39a81a39832b44adc25d0c46b2d0d6413fca234f7d62194e166ee52c7afc6a0a9410678263df43d7d6eb632d4f1cfe8c2c7c4720fba1f3277f4b2aa0134bc3ddced5aa878e750552d2b12eea1d82dafd28c825b96d8076a77c3ee7e9c1809fced281c38a69c71ee0a6ee874bddfe96ccda3f24c0d94542adfad9a061e4167cb17b6e0c6e74701cf7bd1a533f5b38f1d8ef685a6af19c799317d328d87612e317cdf4eafdfa23cef4d8f2f6b957c6f4fdfa5ef4db8dd56ecec16e3d9a0c53dd9c1a24d3fb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324813683Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/array_metadatas, Name: checkpoints/5696/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5696/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=202685df0c6255d5b8db4fb8570faa52b01bf9b88c3812b25250b9cc6753acb38f8fb267355d82bfeea10f03e7d4872e976d6c364a606e747fd61638f730d5c4faf507377902b749375e2e4a05ac3a7ddbb1ba18647266a3c35c5d93631f47a5ab96d3846db5c019b20e7e7219ec4c8cca410ba0fec0d61ef869192ed9e2743d37e1b908561067184462043dc9e5e0016f09b7b63b507481fd78f653d9b6a9f43a602886c61afc127b71662577a894f2fec5b1e9675812d7ed713fc4eb323fb55ffd2d7d7d2f20b0115c8eefb1221f886f4c87ceedc89b8f721ec1dfc8cd56a228f2e4c64330c7d96dcd677f7b6a660de07edb2d5e27c9cc159bedcd5878bdd6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324845932Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default, Name: checkpoints/3648/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3648/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=57e0a2bf4646d79ac3c42d760b58eeef417bb83d99745d2321d17009dd32208db39bf207ff65ecb778fa843aff5330cfe9dc0256d21b7f9c0fffcc07a17766acabf8333eaff661a426de0fd7300ff4e7a0875de07f95791c9011d64b82777ceff31bca1b31a7357984e42c9b3d373b01382f79b5eb6cf983deec2adc5d73e7aef24d2b57c61d2ee05b4973d9daf01205cb168a5d92ab8c09edc22ac4253aed87fcea918ec1fe579cbd3b1081c44ba2c0ddbc04d93e821589141d21d89f7b39c4d01fe77cbbe52f6879e0cacb2250aa533be7502d73bbed5b5a763a887e6bc678c7fb1297f54333a4679623f547e2073715368bea6996b8609fac71fcc93def39&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324866275Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/d, Name: checkpoints/3392/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3392/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=83f5e2febaa23e84a55952da2c36a3b4c704bd19ab64cb2ad3852059bc1d2cc16840ec8f88a3a0b31b3e5ee6ec77ae16723702ad411eb3ea0af6d233c24843ca8a15b2698cafd1763771845b70a34171d14291e4509db68e3f31ba0ada4637c9312c2eedff604b0f760fe386891218b5a4a1e5e829835cddc05b87df36d47d87547e188988f2cad259514563602cacdf8804683942864030e988ea88c15ff60d774113bba40e0a25f07a257d749c43040ddae8a147581b32cc97efd28ff70d873cb1852e3b6036992581f36dd04b736ebc8c62536fa424a947c2efc86dfac2fcce1cc8cbb62901dcab70d2663ee2b0948733130ee93680e2deb4692f472769fa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324889117Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504, Name: checkpoints/5504, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5504?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c12312bb8cc2beba088fa89848383f9507d55beb729ad4afe75e491409fcfb087fbe6bf1853809afb6dfcbb3bdb21502c69f0d1937cd898398495b3b68b06d3770fe10e81d148dc0f2b3b0cb6739b1a3a89d4655af73376e1500f063ff23a4e8d426ca89d38fb4fed65ebb849596f1b71f4d376c7fe0c9c4611c9ac0f6c98a23d3d2fe0292c28037be13a75b85ad0a57672bdde39f6d542453a296472becf3cc27e540cbe69982cd8b966ca74a90df307770f66d1057b9964c2831d5d0c13ee9f09e895d18fb7e251d28ef0bd0270bdb093feb65004b2a854b16e5f8583acb689b2d03de64db5a41db281ae7b6ad1500aaa1cafb37900de88ab7794b6ab82561&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324951466Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/array_metadatas, Name: checkpoints/1280/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1280/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7bea07ac44627cba20376092c2ed061d8cb73ae722372a9054d4224d55575be83c7656464d9cff43a5720ea0930e717a43aed70780730ff07298a7e14ef140e87e9e8f6a14b921c474409ea66a2e966b621a61de6d392372affcb0db574ba2dd89bdf11e3d514ce1818bb8bc8c6d043607cb656d81d5f39e02ba73fefa73db95c964171bf2eaa4ca87d2a3cec39424860738d56aaee7224d962dfc726be41baef36a2bf1e22008c253ca7a4e639cd967028ca385530cdc5bd9eb32cdf35dc5dd38c507c3caa57ae0e8e2c19bc2fabc1d0fae7f0d89fc6f8b089a148e84d53555cb62b1b6bfeab8a9e8ba6af7ec6e93db215b78994df900360a4e5619cee6d4c5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324964014Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/d, Name: checkpoints/4608/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4608/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2e98ce6d431fd7e8d5cc8ce5974e40478e58bd4006913b2de5a87632a08e486f68f66429f819609eb18007bec053e69bc6a88adac23dec794ab26e43da14935233c375902410504608f041d0972347f9625b3c8ede3bb7656d5245bd427c48b4b81cfb325743f9a2d5e24d0cda48a79a275010d6fa86fbd65ace5c233c7543ec43d77ddeaa74c29868ca8620393f8212492b56a5f30d7f97c317e3a72646c1cd184e70fdcb718dda2338b611d79aadad2042727fdb5a3fc5c3febcca0596612f7fd2392dad6b1a2e0564f2d31e6db4adb3cd236954ad386b7d8d4760d3025d12f1a43d7444c3aa0a944b43b74f6a93eb160d495a00a5b915c174b50c3fbadf81&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32501531Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d, Name: checkpoints/1792/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1792/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=36c9827548c619d79d2346cc21f8f6ef101a38cec98ce4f0f44f6c0fc320598d7d80be6a189cce886627b04aa058ac46c06f0e17083ba77045a65ce7219460cf351d471a4fed965b75407d94fca5184b81bc2d58e6750b79b7c3f0898669459bea1ed99992387c86cdb8d355cb69b27bcc7ea264823e861d3397fca3a0a291d934f635026b7cb86b22c74d2acaf1f019acaa0b07b9f159c1805a62d2a040c36c33523856a35b08584d74feab0d6745d6f858979a8d1eabbcb8886018511ef522c62ddc4e2a7e98190445044113707e9d029f7d5f86e78d665979413ed4f16989ed1afc3d6bbb47676efc2a6bc44991f0bb7dae51543541d0778318af0bbb22e1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.325032469Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0, Name: checkpoints/576/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/576/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=92d6e3b7fb9d8d4fc22a5459f05bd9a4f8642ffb7a64e0cba9d81983c5ef1a8e5024fbef6451c91853fd2ecc4e40c71b2fc431989812a624fdb9ee9c167e6c86c7c03737dcaee368d8921f8f62e499dedae2a4aeebd786de8211dee08f3cbbea25a16a8850db4bf1afd3e1d0ee66ccc00351b62ff05773b7f198d36c1f236ddae142a8f9f0aaa43286045ae7011f349be041f11d1dcfe1a383fcf8952dd1248481f5e7d232b9a03baaa7d194a0dc6eb9c020684f4cc9811cd21deb131a8eda5fa296bc9534f20df4ad3cd7a59319a46d2073091dc15010d134154d88c67391e260599da63a1cbe6d8c3697d34020d4c2f5a178c4cd940b8c98c77b6864027b9e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.325061805Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/d, Name: checkpoints/3520/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3520/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8dcb05d3ae42fc77c914932568b2f278bcb8f8384a8576425217f57f285ce8fb7da64bffaeeeabc6c34699e51d1e80d6cb714d27dbfb553da9e13426db40061986ea8ca50c816638c203dd8983939522911c2298c92909c117cc1302d52e4087af634981d2a2f83fc5101f4a151551c532c9a5f12b03486cb1d60b9cf91a60c463b0051bdba2e370d27f093da1b6d27465ed5e3d9d1bef6dd439cf56ec4624c025ddef908cd3d22a4113b80c9c6ca572f16d58c11f2f02efe4b3ef8a5e0af5c704440d852f5905c12570fc52e27ad77a74b7ec180a86dded60feede45f3c39d5b2b0ea467a65d927977db0947111638bc06a1a1c97553fdf4c4cb53e82774f7e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32510078Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/array_metadatas, Name: checkpoints/448/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/448/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8e757fcf3123087778c28529527f99becd338271597baeda92d5d607a329e1eb42c2ffd1a013109ad678fa52e1a0d68718d4ed3062f7bc996c77157be8936659cf70a747a2b1d421f1aa9f47d4e4eab42cfd44335f8049518f0a1e6a51681dcec28d5159ccbfae5ec02e3761d59be2980c9d4b5e4a115b208975ad044c4c3943b0f0ff8b0d9e7bcd382b4e3a0d6d0adfcba4ae54ba46d5519deb3e999e9b32fc2999ff4edc1f91cfb9f8c851bc9326f78b3f7a02d533855b32ca92128586b47308e141a74fbb746a63ce6ab63fc6b04b108e6bb0e264ea4c98b5b9f24d7eb148ec37f1e64219c9b2285081d271335ae102e7348504940ec45a1eac9ba686a955&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.325140658Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d, Name: checkpoints/3392/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3392/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c1e218f77df8dc2d583b370349e3a972b39b92eab9ef3f32a41661eb2b46ed58bd473ea2c0fc059cc2767da8160aaf03d23273119279aad0b6511d6a61856a31748830fdc98aaa111fe381b415f2251563c13549c88b9def72dc58514688d9718b77da9744d136e3c2b94543670acb872ab910f76149b06d9cb31de8f2b98b9995abcc71541730ca221f11704ea9647fd5c86dc710aff131357a948b2c41ba28317895e53e7110afe8a0f3f66cd59c678f99f6a52e7051d211d4910ab00d9482c15c00985e316b978b3efdc54f7ea4d382674e4d34efdc132744dbc248de99c20e3df2f01ed8147992f159c65cde65b513bce6ac08dae4f3435c4c229d61c63b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.325185022Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0, Name: checkpoints/0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a06b3770b651e7f859d69d3b5815c9072d1029d62d881a60b3f780d5f1cad43202b431a4a3eb560259492047732cc8b902bacfee735f2685ce4b21172ef213d4bb6e4e08c6ceade96cf4c3999b4ace32e89fe0a9e9881cc3794f7209fabecf990801c951d8e420ea4ffbf9ca066272c71f731c01e91230fe1eaf275b6d9e671d642ceb6f24f6b53544c8b8ad1486adff358ca8289a2488557e6ac51dfa896ab28cf0cb0e87b673878d7d462f9e24afb1bd5c501e67bcb206793bc0a9fa2b70361117479a63401dc5f8de36684e1716d9e8cf1b1ddfbc51836bcaf3e4c65a0accb85d554c5cd7599bc14422524c142d138d6f705d45c3224faf4f86dc8325871f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324729821Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/d, Name: checkpoints/4032/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4032/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7faf1a1d65c5b516d7adceb19fa52c3190addf628edbaa1de0bb6707d36818a34269e2799033af8f36858e65b328fccd5c3305e83fa8c505548afe1bdd1b12197c48e99beb156b3fc0438cffded2495bec5ae7dfd6bcdb2ec60041c9a7c171716d670e815eee7e935b0e92c8fff9a78eb39966ce090c9c80767476070e4c0da7e007af969201a81669012cdb236b1e8f3b1a5d53a375d6a55b20282f8d1541c9bf6b0b733d9db2dfc56f4c1fc889c1d1d69afd706c5b01922ae9902f6db28e2538f79eec45785d5f9bde9ca21c6643ebbda526cb6a65585630863184558e218bd66d0737e972a635545dbc5ea514eca20d2196607fd63edca59b82a75632f193&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.325049651Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default, Name: checkpoints/512/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/512/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9f643d432c091859d06cdfa46a9deb8601506e2bb38ea93383d265d33e66de8b79893b6448d4c1fd3651d82af300829777b860e42594d89aecf806e070ce46e1e3bb7e5d543cd3870d5f5b7426d0205451d54709d4b5b3267e7170dbb2bc9f38512ef5f572cd82c594274fbb09008cf5c68a4fb6292e50448ef90c576b02daf3c2e3e150e510e66f8e9cae20195381b7bee9abe6f6021346af3b0b16bd5c2951ea037e95fe377b211b31f4ae7915dedd3892a3012a7c7747bf19c10e673b2f219b65367a9660eb351cfefa36689129c1dcb27c596abb1a2c1799888c32c415100580864a6a38b08b40966d85e6ff87a62a9cdaa0d3b0274ba1a0b72ee5e12927&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324875077Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default, Name: checkpoints/1344/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1344/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=83658d7ae528d8cea8672b97c617ec29c7790d586ce0cfb266cb39b54e9abc7321b12eafb2ce4364b05fa10391ee7ead4a207777ead98d9f5ba05160fb69bee4208414698e45382cf6ed8eef5252972e8fded42af798000415c1eed891843cbf3bea16eceb41a5b59eb7f43bd33fae4d5dc25f99b9b416973a16439dc58f1f02f9d35e826dddac59f53bdcc3b28a2d052f2127d19eaf1b64897f1d42fa718ecee9a3bc4cb0b612721caecb1a1663c5e1c9ae9f0c512227430af221e698dba388aff728ca071df6527c41e5215cf550f3cc060bf42e9b5d2b7c5f5fbf7d9bb95d16cf01132dc26dcd54856d34b3ce89757f740db11db50c50b6117c94db4e8a81&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324979471Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048, Name: checkpoints/2048, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2048?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=32aaeeda41042c5fd058eb4067bed7ffced8e056a6f22778708bec624fd01329ee7d1035c53b55cb2ea13b34db4510298547551f0194d7b5e6c98b004b50f124f8a639e99ce20e68f61cde2bef2e4fd866ae555f0ddcab01b051e8924c5e3fd2265594dd28013a28d14cf9615111bb8ab4768e7bc3f59f2f5758d4bc783ea902f4c8c3c05925d1ad27e33df99127b584190a042cd2bf7bc60f55cd19a02a82a19987c8f74ad1fd7af2af081ee945023b22a367dd8f1361c03ac2361f5977cbfd68006f5d13ed95529628393e2763f8dc66dc99bf7258f0d704a665d4cd9841ff7e77cf0810ca32f0963f1781076e03931da51df85c2afb7702c270a4b81cadf0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.325314177Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840, Name: checkpoints/3840, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3840?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=28c2530238d76f718308c82128bd0bd0908e9bfb93c4f68d41601b3fa3c1e964533a94acddfa87713b4b1ff0dcf7252c8666630c25db658ca98f1b6873610093b7d3fb8eb6478acfa07d90ad4d5c49fa4e4e3bb1e3c4b94f2f7cbf4fb27ad36da839f448de6fd2ffc2aefc7639b5a37fecfc6d3af1f996ff38ada10e72b79ff75eea09ace192a687edbb34b417943bcd93dd9df707b72f1844e4c199d95acf85fde8e147b5aabf7e1144999781e1af714177d6f7b0aa209e8d2606a3e35d40b7cbbfb0d76cb012f36047ba43540b93ca4d06042be34f8aeecdba49a877002044a793fab899ce59733585fede765c684b926c83b4230928d36bbc512434c80578&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32596923Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/d, Name: checkpoints/1152/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1152/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0fa9692524b6158d5a17b3e398420673d7301aed22b1d8e1fb0930db9eb80ced967ec1abf1e01f2047039b8244459542f40461a8e44271db7ced6a54a05e20ee6f9e4528d62e8183c6d2a1c8965bdf57057a9dbe7bebe240336490c9c320280a2775bfc0d75e5822883e08e07f10587932b45cbcd342483baf0901201d4cb73ce5cc7378e720ad479835cb0b0aa06363ce007523f65b37c9e79440e1fc838d4cf629bda0a738368424396a3df38444849d9f48ec23cc11ec3ae26c5b9beaf979116e36ac95e645d02d87e8c9226265ae2fcb3e0f92a02851aa71add6c2f48dff42ad891976b129ac033dd848813dbc9524ad62de5164829dfc3271a64bdd9ccb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324664371Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d, Name: checkpoints/1536/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1536/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=78ae6faf074fa497ac7f902eafc55b95476caf57bcfd272b5c90247976854e1a8022c032bb95d5f71fea1fddc6e91a9d929dde29f5fc385dfdfa8f047e14bf775b7e70431d3ad4a602e4fb1d44399918a0b505b764f14fee681ae40069e11782f73315c08ea0e8670a5b22e79b229fa9b715365584513c43be32fba178532a0eefe27bc0d60a46e76432d803d8d263a52469821287fd64e4eb5e5bb16d362ac00e0377c121a66ad857ae8c14f4cf13b707eada1238794323eb45b4a0c93b196728f999211aecc9a8b8b6a13aaf91508c86385fbe118dcb185f218706f9ce5b390eb4c2180b2eb834139715be574aee9ec10185b5446fbbb6cc844f3bcf67c666&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324759355Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0, Name: checkpoints/5760/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5760/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5a6f6429963a526fb236b06d0f426667ad4aee77f197054f6c015055b6717bff7821366142667b878ac66500190edc776dba6b32f1fa8e46d5ceeb31e016782ea88a270c8b1ae37777674e879890fcc0b2a75bc466e21fe10cfbe301dd752efe7d73d78ce1b99e05ead22f4097317fdd5d533b636b5df1a39558e0043f05268e38bd38f927c6eec2aeafc17f2d094dc2276dfd1f01212ecf6218842285a1bb4ec223825e4bead014f1909111f77f70714a9574703f91ddfc7e49bf0b8999e693381fc04c05cf6e08f23e0aea80d88c5b2d84c42ef12ed2853aa1bbcae11716ddee3ceaa91e951e995d399d6b3349e87ebd0855277c53fca824c8f7a557a1852a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324747263Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184, Name: checkpoints/5184, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5184?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=64afe75c363bab7ef65cb58b343bc0966876c79ff852e319e85096145a081760dd6f2bb57e13f5e940c979bce269e38b60d2f0d030c2f25e9e4715fed55dc0b104d705e3d37b462beabe806b7a75ddd1a0d1ee20fa8ef8196a13db9f7dacd013b94781bde485cd49973fab0651aa82c26d5f39eba2d2ad480858d6d9998174ddf7224b62ad261d2e452e9ccd149f3e8edcd2d9d6840f1d683e5d8423e560ad65eb4aad3bf4470e299c1b46bfaaa9479bb1d6ed6f2087a4be5ba26fcbd3d07d5a7dd4f83d6828bd7b5b05eec5e65b7c5505357eae8131780f4b899c0da22312dfb8eea921b6a90c0080d25e29770dd8e73b33c484dc922d0ceb848fdb51cf46c5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324785763Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632, Name: checkpoints/5632, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5632?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4a9a86a6d4f948228740fbc5d25e06c129e2760befdeb21915986d890fc0906404b2b15ea7e5b3e2d891086b4ced3edc28a73e34a7013d2d90a77db91c04f2cc5f2e46fc643bc071cef8ac27abfb79ffdd895e0a0aa804393c005f164f94b202e2eeab22c6bf72efd3433bda01b8787ecec835bd8cd0a325196dc0d453cf194c6ff449f3c6ade156646d4417def22bd9e4232337a582a824924f4da145e04c50a741cb65d0e629f6626927864ed6c89941b9e4aa85d47003260837f90d77bd325027991fb97d54331f6454855f9bb0fcec66db625b167be49ae574190abceed503981a1fb70bc83dd254aa1145c00ddc4d69552629db9c68d482b6d4ccc633b0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324781173Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120, Name: checkpoints/5120, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5120?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3a9ca849cf1c06761c0fa63c2af71c20f489964336cf38ca2b5ae53895c860e07380189467f188a6077a46788a324e14bd37574d27df7a68417a63107fe93789523344e962caadf104fdd1e930b309511288fc32fce4570ada2be90860005fcc78496cbd7954b26ad4cc9da2efbf50357779998bcc54afc5c3710ae205464742ca8b4410b785778859c266e75ea23980a962459700476ab1f5bd773c508d047aadfdcde3551149866884a5f4dc917b0749bc531fed8d6605d7820c6546c829dc937e9c0266c9e01fb1dfa9d46dd38a9cbc1499990071850c9822acd19126c66f56a7afd807725aa2f7995a15af03da7f78d8306db1741225b0e77173e5ea0081&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324802469Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0, Name: checkpoints/3648/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3648/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6f14ca5d49a9404dae8092c4e5dc1ce78007aa33153f4b44378332349954073ff90f726f182a40a4dc0cf304e5c6a6f09548c640ddce243ce32bc9110ac7fbfb163f813a26e951c92a23ef5ef0a0d613ed5f52c45c56fb8d02b6a8e1391936d85f4ced5fb463ce573e2a336430d08ad7e335882650d3d9143e3fc721a38b8b806b004db08a4345c64785465512b5d3585110cc3beded7cd1f5cb45b6d2413e66c74c448b201b500adf3191a754c7a041795d9eb7e1435e9a7e7339440ff2fb285a1ee6119fb0cd09fc9eb69545cab590643f65505fb29e0bf82e79fe453f19aa5aaa11b2b15d041fd73e0e0b77698405f9cef7e5396ffa51fe8dacadbb9135e0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324838137Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/d, Name: checkpoints/2688/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2688/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b3c9f8b1e37f505bd82a2c6058df6a8de99aea731483d55ae1801af6832044e5c96243390d599343f29a01e45877665cd71fbcf0ded5c1d9da691c46063046c3318a16fd36bab9872f9f8b6fe1f4ba0a784195a73e4f134296f573b5468c1ee1f68ce76205b3fe8b7c49262c34d671ef603faab1466b42aa778ea1635b82e0830bd8d7b5ab63bff9d355018751dce6ea9a07a551d2822f5a77c38097d46e5defa7214f1c4f4544097c6a80c4385684c14cf5e7af8ff885af679641316c161dc2f4927a06a5695e391bf8e205056621cc7b3973fc65c0c17e4082013f34cc6735c2bb96ae2e82386779ee807cfde867898c87848b030c6bd4dcc9f1882cf7b1b8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324866368Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544, Name: checkpoints/4544, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4544?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0cc0609f91a6584ef1dd5c3f7ad02ccb494c1068dbdacc8f19c42e81f315ad43c6d5516f280ec1ef27b0e01ef4c54ec7e02892e17239fd1493604ca28e703fd57defb4f88d6bbbefa651a5cc4268a5ee9da48730c6ae3e37ef1593f814d55c3d6d2002d981a1a860211e04978d1fc8626c636bd13a9be1179786e5b413d2ded057e3f22d33ba01e47737b9516f2b3775fcc0d7fc26c6909bbbe864738fc71569fa53ab4bccd8e00b15a3413ac70fece31cf8b26a9ef1c5a1ab898cb5e0990b2173e42d4c793e593e4a50c937e5b63399ffe46dbb69023d91bb9fdedb73b6bd6e7feb87c7abfc5092b619ef4c1cb1af6cf3c5dd07890c0b752a157e4e7a0e468f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324856597Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112, Name: checkpoints/2112, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2112?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2c1439aba59cffd14e1f405ca3150c055f2390eaaab58802b73fc2d2d038555596c853a88e413c7ebccab7014f3ac071805bc882cce4985c5a31ff0ae1bb797e052fc683720981fcb70976409e2bbd922330650009103aeb32695e946f2510de08aa44f67b78a79f6dd0e3d2f85b9667763eee816faac71b6a6ad40bf073de20675cb0f03d7963ef08221dcc388c9263aa438051938b2c20e1bd61440f0ab69859516a3c2ba4e241111e11286a8195dd15f897c5b890f2ea9dd78673365fc7fd589a7fc4cde0beb979afb860d0129a51f2e1313b7a33f550f66fcb1cdc5a2606758a9faf0b028886419e4908776770691b39afdffad6cc755229f1d2de905e5f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324887625Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/d, Name: checkpoints/2816/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2816/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=82698ce5fe30ca475e1c263d1b64d485a4e5c47a9c5b9cf4f4a00724857fc4ddb5b3879e18237d6ea3cf414ef094324f412af0172a514cfa616d371ef7388e6b975ba00979582860d18579f9ac792fbc11770dd913ff93f07c5290b9dbcb87fac6a7c65463af1c3fbcfaad370b09e2bcbf7ef108520711e33d738c05f7374f209f88a6e7f74fa1c0dae347389937c5c369f43018f45c9ab5bb0788f0f37646edd2a5740752d83a09405d5f98dc361d72665c1dbd3d57accf7c00de06f5b893b0bdf559742f7c300ad8f96aaf7fdae160ac0cea9b5934147a3f1f22559c158990199358f7297e943881277c59b95010b2238627ba2cb444efdf5a1ee00606eb4a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324928222Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/d, Name: checkpoints/3328/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3328/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0112f4a8ef41a6be6a1606c5330b3609bba026be24eb0151e02414e717ceca9f81eacf5cfc7b96e7823900b1970f33fd008b84cd7af6389a45215fc1b3c152b67aa1c24ce6e3ca2cca534da2cf34e079b0a9cdc1db0df86ee7e2089c370204a9b3d1aa9e6a9d6a59301071e39280d3d49635df5fa875e1c04f03637f5434b3dcdd2f5f6b3a9abb173d7f0c9962871a7103e2f79121c60f9110754e287da93bfdae48308ea6c4291eae16471511b4078ee173eab1a3598f71188419f67a2894ce9c80f9ee1cf514e747f0e0374fb5b5925a10b8fa4eaad1f58f8b27929937dda0844c03a7e72d9af5bf5dd9b24471361fc7ae0cb3939a6f642c53416c236431d8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.325188597Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344, Name: checkpoints/1344, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1344?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a0cd6a172a0cb37ec26dcfd9e45ff21483fe520a44a5bdeceb31c5814a5534955c07fb35ac1b76685d99648ba4766bffd7fb58c110d3ca4142aea7c039426cc040feedff4aa77eda372edb105d51a6b119579b72e9eb0bc9b89276ad8553993de9d51ee4558ca8c3bfb86d69884767e559b601f1fde0ae7a4b4ddb02b9e14022a8af6fe762e26ab1e42024b6528a109540c7dbc47778ec6be51d8a21e8bef9683db105a3624326dfedd2e60462977bf7e86922115bd1b235d21f5e37830a3f01190d7ab2f732dacc819d160629031968843420c8206b07feb0c3b2b7038db271a902db9479964be727e3cb72395242b5fcb6f9d4436c8e1f84b999c593c1dfc5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324909098Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/array_metadatas, Name: checkpoints/960/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/960/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9c0d452739858beefeb011b45d2f356dfe8d1a96560ec7497f8a5705072f061ff83a1b73a453e8e04185d70878f70cbff563730752164a6a7828aef50a4c97dc877263ce9afcaeaae72c79c942321f0e0b53ad731f68f59078029e0256aa306c2aec056a4f106999c833c2ce313fadcee60a0cb06bb15d8c3ea53527ba5a0358c71270372ce2bc16ef12c628c8bdaec01a674bf73506d8ae4bd24624fd51dd630dcb32be53538992a88875d752a4baca9608b8af4379d161d25fb5a53df708247ea30fa4a8c9c469e7242e9cd2f1435f97064e3b1cd711f651aaf41dd80aec44650d7f69c992e06c6193e883a71f7ddef8dc595213035bc9fddc6720d6f173b7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324966121Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/array_metadatas, Name: checkpoints/4864/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4864/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a46198dfc970a8ee13eb03ab69d329dd8faa1da0ec0356bfe9ad43c393ada99ce99c3b695a46ff35f129d01b87a0bba5536ee32223f2da63e78eda1714e00ff7fe24699d52095e0cc04aea820aab48459a7874e88754875711cbc2acc166858e28fa6eeffcb06731145e97e1b4100b7e48812345d6b2879866f01750f357a040b8fed55e924d3e7c6f17aa0ea5845d8e759520e04c28c8d83b9aeb835412939b799d1767d86164eaadfbbbae1d24fcf894694431153caac2ed5f1e329ceef7ca57755da8a56b986fbb72cd09c3b503ea33d53ff699acfda270c0efde8c30ee10e65478639fe6974000497d419555971469f0d44a979896292929e73e991cd393&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326236788Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864, Name: checkpoints/4864, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4864?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=699cab229d7181f2ac5cd29ae3381c39ac86057d51dad32ee0a5d9da4fe20c6cd184259c91346315c3e7fdd6181c410e9bf4b6bed52a8e50b66e609b637fdd978094b388ce5bd5898c6bad1db7efddddb911a47916394de5097cdc699ccf2b92cf2f5de992930d8737e94e8594c9fc4f21b18efe254a10e1a1a536e8382d9ea304e1e17bac1b514d0380440dea95debf5705305079c3924a96e3d6b595b1048c33ec618b5e7eefa354ff41f6335d1bca2ddc3b0d0ee3ea6765a3b6ff5c423ee044f40c82e0c5924fe394052d3f6ce5cc5289224ebffef4b50ecb8a53ecced0deb3571e0a57b805753ce4738dae303c68ff4ad8db5f43e916fb48ea8f05d04188&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.325131559Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d, Name: checkpoints/1152/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1152/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0b3672b48cccd2ac38792e8b73f52c72632b55c4c2d2572446da0b29c22cdbeb6fea739d00220afaaca0185fbba4ce7acf2e49924372d8ce031801cafa52237da88648ff83dc0dab2657b4405376270d55ffb11fd2e519b25c8716574ffff40d5a1b53d78d6c0ca0c3022e114a7f881ed1b5e4c927b12ea1865a897aec398153c8e40676a116330d9c889813ad42bc83f2e2bc4e400d13a9405b24741c354ed444e73c7b3129a126b2e3de8d5392303f2ab299d1a6474a63d754568093a28bc06f6eb4b709b273633dfbe92cfefaf331dfe214560010e7fa2552960f9fc5031fbc162a1338e0583ffeaa20d2a8f857855bfd21b3608dbe50dd16611f944a9a3f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326323707Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default, Name: checkpoints/4288/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4288/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b03f4594b58f238935ab020d7eb7e781f3bb0b2581431bd713ff82b90f7c62128df1486989389d0ce001ab71e3bd4d48ba29c8774845be658533aeaf1a186842dafddac5aab0dd60a1a7ad882ff1112a4fa93e08fa415db73d9e5026e53e4d110db16492d781a394fcb06e238ab0dfbc6df067a9392ecd0de45234cc7cb8b39fc60ac9e36f21971c6398d28cf42287ca6480652e6c691f6c55fbc55222cb2e7b7e00fbe06fc1c7624676142371af05920225f8f792f74206d279f6553ff8950c05b9a95502418cb7ab0655dfcbdb6713038f25a76808c399ae05181d1ae25ed46e43d108ca40bed58e75519d5c3f8b5111c6b7937b9da60856bd20a5f345f853&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326335578Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0, Name: checkpoints/3264/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3264/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9ce38d025a11faf702299bb968b5c1712d48a1ca2047f6f6f22494655c841e7c3afc209526a9b77998753baa75d50bb650a71382163b6f9cf5a35ac0aa5766706d5fe8a988c6dfde3864c16fa39a2840c87f24b60016f4d51a9cdfadf04986f6e8837f6eba4ed7a3b94e61b4c950a2ebfc90e181ffb69b7fc8cb0935e08f4ee4099d203e08e70392a96749c94b888a481cd699689dd53262b7019175666d2dfb2466964ef054b73b7c2ba2492e5cfbd96251cef8fc3327329c9b048b139e77f8b690caed96bb838b9de55fa7bce87696e7a4a7b83713e5b2a1a2701ff41f6412a7b5b95203e92353b45f5daed4611f70e8708f5afd86c5259e04ead6c23c45b7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326430823Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default, Name: checkpoints/2048/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2048/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=ae79df3f6b3068bd76ee4537b9347230ee9153794ef855c290bf71d30613e075bd301aac4d1bce04278c8b6de3b8bccc55a8fa07f0f9d2f6e1efe5a4d6f34154fba9b8dc3c490db91d54b7a4e9a6350f9f9732b84b4101d1d258febd3a8e4b8198397a347a208055690329470863ac04dd813962bd3dc70ee0c6599fd65bee0c6f01e7686d82595b0b3646e0bfe69bbddba12ebde88e95a35dc8867cff41fa41ce5f158c31351780c603f1933738d32378743a363e1865424b5ccb6454fa52471f3b4c0a10cb457f81200743381958851ab714d13ea198449966c0b5bcb02e8659de603195d4c5d2b41ecdf83b589f5fa243fd00a0154af76efafdbcea24e515&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32529695Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d, Name: checkpoints/0/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/0/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5943b68ebee66a7340dec3606813836a7e21bb5b12654386f0ee23af9ed44d381a6069e29f14d3760a49c396aa33efa77e89bcf649c71983547ed8b717ef108d06ff4ba7ebc9bf041fbfb5b5381aa5f48bdcb71bdfa67aa9069f53265c8891d677f3d455d9f39516fc7c1cebba452c5d4be61d3e0e0b3e55e021496f53db499b381910bf9daa94dc4ae8b2604cfdacb1c1568a877a97e38eda3d2ac312b65fb3ace61458e7b1cafdbc6cfc5c8706a591ca88d8fdc937a14b1021539c3e381f622c2895b2e8b1dd219a6e71b3605afb7e204c28bb07640058dc9232c6463bf0626afbd387741cadcb4c08adb4c1fdfb9ce4850e7d40d421875d7065ff23e285cc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326456664Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default, Name: checkpoints/4352/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4352/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=38c5a592313253ca1face7e890bed513da9cf3cc14b0f41009aa39981d29c6a07a8bf71f1d7b6920df76ba15191bd5d0b00dfb51dab1151895eda3da6623fe89d2ecca1e3c2ec5508c07570356cb46d03752d96e7bc9ee8685f425442cc6e8ff76a25bbea9ffbcce4e14025e74c9c4646403076db2c204553d21434765617c7b9b8095e202315a457048885b89a7847762de26f0ce831988c9cf0662b21fb3e20202c56099287cfd522335b52b9d6d7fba665851ed0a64741edec464bed03a17a659d2332f3f9eb414ef2f01c0a722818a68716013da350a2e5b4f8ef94e28f3871f7391e4cf169b34276ebcb4161b19cdac453c93e82350a8a7b0deecbc7357&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326450289Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/array_metadatas, Name: checkpoints/5312/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5312/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8db784fdfa6c378ba84a9828d58b43c27f03ef69c29a5a16bd37b3f9dd12d07807d0cb58dec149bd4f7c5eb3dcd5e25d657a9af3fc3684ce355373593b51da59266beab96765a7f423abe378b0a9e57771c90e2addab72006d200e88f36e722821ab2091cccd6f01f3a12cc1db61c812cd291220275a0225614b81c227cae879d7d3e6033b828cab9ba33df3d88222d8d3c3c8e7eb80af2192fe13557d7979b0bd78751e812ece8276e3a5f6179aaabac4661c0ba425824a47672e74998cf9aca6bda2990bf7010a727fb5040aaec1b9de554985668efb16c1c36ada4fec7627e2307e5610144189866296278aa5132b7ec610277d094d1de9dfd3d8b504d5a4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326524307Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d, Name: checkpoints/3456/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3456/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6f1875c79cf3535886e0567ff46390633e12fc3793b05393d3a7b604dbb7960cf92629db74b0230371893844d5d54f695815b62fc81600b38a1766bafac1967747a50289c73c370ad7fa8f585900b1bf8bbd761ab14cff3c958c87372d1b4dcd1a33617a64aaf4f28a7be008bfaf0456b8775a5c60b8c9123c7ab1f4c0b2f915fdd7e319d677aa2adfa25d240c6d25307eaa821a8e07366d3fb1b32bb9b748db5bf330db4c6fdb50ab2d5f007d3a3d67cf11b7c13f1dd1dffe02a1c8a56e92da0626fac55de8fdf4d383423da428b9c18e57e6d7e3ec30be5b07211267eea44d04a531a3327e0891544eb0dd275e52335a5b6354ba35b40842135f0f6f889c2f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326609292Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/d, Name: checkpoints/3968/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3968/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=41c69823ba004ef47d23471ae7962f6eec1b2bfb8b020ee24cd22175b08e4266f42559f5fc771d0e884b881415daaf8c994545443092004ac842afe8258a150c3bd2c76ab2c45501e5ed1cd00ded674fa9f7848824785f2825494bf542a5a28ce33758224570748c1c01d4b52e4b33c68b5f151df6946169c85eada7543e33fac73048790903bdc14a3037e1db5e8e6a2a9ee04f14d5258bea8857fbecc61fbcf0dc4904f877e73aa5244101128c15e4f4f5e45235de3cf352e7e0f4607f30da135dd50906da89b53244ddfc4b3b0c02d1d77e40cf208710c0161a9a5ced4649367f54f0510f39dab843f64a8cff7788a0321f9bd51e8062d6b423a0c8f426b4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326661342Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default, Name: checkpoints/64/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/64/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=33be788730c18ca0bbe0ef72bd7d6e60cedf231ab937e9db008fcbf3ac6b825f44553a7a3602d68137d5509eb790359a0dba5d13ecdaa47e698c65200543c6e30d0f070d0c17f107a74035c59d25db0f17951933eb7f2599bee54ac8fc402e716782457aae11eee9515f462e9e44da937999700ef5136d44c878cfcdea4dbbb9cd3c0c84c62bcc7ec8faa8b34d065e638c45ba049c6241ddf3b65b4a1b3ca397c569cdad65f9a377bdad2c17d9fb17a40efaaa35f4352a8f47e506f2eb376b510a579e3b517edbc201820b885788ccaf22f45f9c3282df6632148d6fcc3fdc3b0daf0629f813e21ff888181017e7c04d2af865c4b11d98168d9dca4bfafe9b78&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326737033Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default, Name: checkpoints/5184/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5184/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=69772fad4acd599df7a3c96f121ac89d2cca26b7605f4e319a622107c93a4c20f1c82e0d137656b46d45b0f1c03a719b9ccb187085b22681b9ae785241dda4f65301c07eb88e717151ce3485e14329f5fb9074a5da64ebda019063b61ffedddb6134e3dcb1ad11229eba49378dabc05265947f31d6da6d8dc150355c7c9dbb63b853744d6ff195a1c97562ddaf4e6c4d86815ba1a4e95397f17367aa3ec0e423982c803c027ee6982bde856da9a4e21eb339aba63346b7b0cc205a33cf4d3ff06a1ecfb68a2b29523d4af41cb2a5165455ac9c52b109e84a375316682d8cc967a94ad9aa77b431123d002075333b972ea2eadf67928c605f3889ae5ffbaedd84&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326775667Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d, Name: checkpoints/5184/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5184/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=18de6ebb435333b1e59d293c928b55719ee994caa23e2f401021d0af4d72647d720160afd05af084f86f7a1ca903b443b3597befab7913b8b74970d156226fae109e7299ba4fd3bb8bd456715b99d9b367d2593a40474671f27869368d19838c758a71146eefcbed6109686d3cd356e0b956116b2caed37aa92ba45ab9f5b0864c01da8286d91a74786171840c24acd61ec15d4ff208e41acb45897dee7e0af4588a15cfd2436acf1f4681f7f7b37b8bd339412de5ebdd950ac18125994baf98f420f34a0c61af9713f4d239b38d620d65c1ade91924c8beb3ba955537a1a1855e58501b505d374ef59b34137e1c6ed37755dfc12c0f62cfe3f541dec43f9c10&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326881186Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default, Name: checkpoints/4224/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4224/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1d9aaa736c254e5bb4653208e0399debe53bf295bb3528aba43aa1a173503e5c729a7f84b7b1f0345688f723d0d593f6301cec097f4f6bce456140d85106c216e15f27896cd33dede7b4142739fd8350b00c185453a9f62a59ecda78bb4ebeb6c20da070ffd568b4636c6ab8e8a2d06419f4dc4d2081979a23466b7ead7274e50f07e4206a43eb67d1ed4a4c6268b1cb9634aa788123f5a7737ad6098edae531d458bfae81bfcb63828734c1c481bffcf0386c7f7b9233a7b99dfc97d293b69b0610a973667edbbc8fb006f438c8ee91a04b05adf20b3a3d9514466be1fc42fc6b43b5f4c46bacf6bbce3fe600abce7b4364fcf88d4628cd0916f0d18089f8f5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326905661Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/array_metadatas, Name: checkpoints/4736/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4736/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7086994998ecb0bf9fb816cceb72e5d09cb62767e9797ed141f0b49920a6b3068dc42c83ccf8f0cbc4afc1ad470e8cd39a11ef780b7c9573ff9d1469afcb5d9eb849d224d09990b24134ec6f20ad9b1c69c8914320d69f9ada77296fb07d4451a399272f33e60cd45daf87632654c08f6685ed9126da5a1f7b7970b1a3f9164cecb3c75d1e3bda2f41e774f69a2617e67e59ac5d08e60d7c8cad8f36bbe417fbda7e4b42fcfc7efcd4970e78fc8dd9e9c1ddb9b33ea0458c598f65521955b357810556985994cca58f1a84688f68dd93dc455b90f2ea38420af1177e978097f46595e069085aa20de4eda115cabc5ddd387594fd5e2c93bc60a5a92f9e607575&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.327030576Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d, Name: checkpoints/5760/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5760/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2c6321dfcaa190f9b5df19a12cdfc48c8f41d8fc0ad2fe97e18a016fd64843ff3926af7c281ba402881678512153b0a56e73a720cb04b6fcf9a02d1decafd6ac17d68b4a5d1d4cefaa7d148cf387dd45e53a5c2cd2631c1cd1e9a11a6178fa0723aef494400106b0cbfa061385f18d5ecf8d50be0ce636fa003872a08d3dcee4495b549496a61fbe930eba319decd9b95a01dd9f5efc318629fdaea48e3eafe889364c632b805d99e8a4a627c0a6cda4067e77e1189d2932b9766a51890e9196753a9348c522e664153420830bb2c25e9ea92812f9fcbf5422778b9072d409cbfbc42a920bd98514f9fba272666f4ffc00aada7899a5930e5bc6b3e70fdbe911&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326195337Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/array_metadatas, Name: checkpoints/3328/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3328/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3441451a56127a9401de026bdd502581d193dc1c1bc70aa2e65a859219156c8b470bdbf1174dc4fd3d9b271bffc7173e798ff5c37311d91e30b8ad18360457e19ea28b3c5811ff7e0288dd365a88fb69fe271aa35b0db6cce39f2f6384f6093030f52387521a559a40644530414b04bd0b1746b15ba4be85fb2df4ae48d6bb51b7112a887f0672c13b1a5e7547dd66e08bccdf623e76ee5a0f2d497c1499efbcf9f8ad69a50042c626b4b5cbd2dc853c4e885f47b0acde294dd7fa44a07fc1eb70ad4843b3a62caff0ee3d09ee9816775be1c8969d29b6b7b6dffe05050c07f9db1aaf380c103410627f4af1deffcdf1a439b80ff09f5862352952dcfbcf9e1d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.327228646Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008, Name: checkpoints/3008, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3008?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3cc01cde5293088e72f1f7808f904fb6cb140b2a927a56e8e0a89a9b42a33549ca48720420510e8a1ae6893c5e6a4940f31e1571754108b16918746b9e253feb74da233d3fbd22a0eb10a33b117676504368ae2154319a878276a386cf9b6ce8c532f60f8406ed354128263b49b1c4b9808e6e1eb8baa7b16f2330782e7262a041a83295422c1e4b9de61329a0f1069db922ade2183bb73f250e6f70c4b1f2e530192e724c331adeb02c759b959700c8720533f1066824c4f6e9e5ae8398abc00c4965ff1ad6422e0ce53b0f10ee4c8f3c0782828e06cb0128c9cd23fef1ff51d9e3da9e97bbda802b66e629aea789f69cdf260223a13c16e933f0571e64cde0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.3246444Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/array_metadatas, Name: checkpoints/5632/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5632/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=96fa898e03892024d3c14446a1a8df2df1c93dc582277d9bd329a29afa9fd711cfe8965032041b2232af8daa6e3e765767b025f7bd63932adf8f118858c9089ccd9df8fa646b1279daf2cdf01a9eb191445c79a54b15990728dc66d46d60cd162391db03957a35274bbd6c2c11a22a70a802d9ca83ac89622e4bc3b0bc3a9abbd525954a9ca99b1aadf9d08f8ebc0af982fc1e9ed5626482d77d307e6e549d5dbd2d41337c1f22b72f61bf8fb33b16a1b78227cfd0b59d304b86459c16c9f9b155bcc67c52d892997c226a684fd546cd996676c6cbe860c7242fccf50ed7604fcd8b742cf526ec2f891e79bc368ed000b55fad3135b26341bab0312e2ce96209&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324779826Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d, Name: checkpoints/2560/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2560/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1eb1d4319516cb443bd26fde14b251ae24af33a5d32623de0fbbd5b7d04a0431bbe5d049c43ed3f4114259c520ad99e40daa39e3dc40f5ef6ae138dd1e6c1958bdd3b5da1e793553bbfd92ccb1ca3a1f8217e65797c762ef8bdac7ad628bb7c94584405b32927a44608aee4a2761b6c68c85525703c9d8118e564f62699230442e1065747ef399b8250cbec5f542b38d608aafb1d7752b0ffcc7357c412f7c0c393ffe0180345d15d635741c61abf52b87c56343eb3cdd19b03c6e66efd3bec938eb93adc564fb4150c3390feebf40d6a68912830b3987e1543fffcc4efb1b809f37fe6c6c817481eb232642d5a20834bfdd2d35584b4f55e4934261167e4341&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326082113Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default, Name: checkpoints/1856/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1856/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=93928661b8d66eaec308efa99ca84ac8c4a9b00471c78d5b4a262592e212ae34e7ec11f833ac8e6d2a2d3b05ab76012a5b12b1294c9e60565fa59d12c7438ff091f1ff16f87c6bcfbcdab899056d36809dc31e232912f5769639727a6dc7220fa22670cc469b31cda5f73ae5a9fa549da9864d58e95812e1b9c08b646814e2e8445f6da041efb02976991480e158bc0d04ac5d08d5889505bf88f4a034a5bae83f1a54205e38d5306d2058319783b2b46d8a02895614a4ea9c56879022e5e9dfa0fa319461e89e1ecf6afe1f75ff9e9732d166773b758db0c66fe36a93ced7f65a20d8f8855ea6a3a778ea8fe3c4506443008e2699c67f521fd74d39b5cc5643&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.327636Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0, Name: checkpoints/3392/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3392/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8272698e173d4df43f7625e3dc2edd6fae2e358f76cabd0ed68e4e6e7789848d7e70f3c13dc478114583a61774758ed3c06ba6442e1074a8ab62ea104a52d90e3e4b37906884f22608a286f34be340f102944877ba8cda3ad7a283b4f1a1ffb51d131c1d2a786ad468c59a2190ead25e06ce4742a85c0db0fe3082e90a00681cfa5d1ae92f9fed2b6fb84199b363a67f128f279c566ad70c0673e6f8978370c2b6274204e9be2ba57800daafa307c975403706e16c9ee6aa161e17273420504ee95ebdd7f3f19fb287f5204c825a7b9fa61fe5b1575a97840d51117d478b3f0c909a1dab861a0ce61c12634808221ae63f96f4a9fab1cc92b61d0559606f43be&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32680964Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d, Name: checkpoints/1600/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1600/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6bed78f59b89813c9d55edd430f226f950ae94ae5a396ee918de8ad48425f6bc3406c9a13b17db71154d9e66e993c4280767857d4bbce8c4ef760a121b0813f9ca86b4bfb7dec17c1657f3ce738289d3a16a3cf3a66d918801bbb4e838fef5df8c13ba00810009c4f0855b2d3abe416d5f88c8ec17f34e1a6d116a3eb556d1957688a1a303ecf239cf4ccad17b5adcd47b89138830c288767a6e089faa52a50c6910e9343aa4cd829fb40bcd9e0501ae81bdacacf5bf25647be42a8d4ad6af0489ec1b4b149def666550908ed9afa0f3e623b5bfde297407a52361a20da625fa013661b8ab158c0e2fc0ec3f4454422e655b2ba934cdd5930b4a24e27584b100&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.327747995Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312, Name: checkpoints/5312, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5312?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=56789a4912e6a59f713494cfc7cee0e617e9ab0c26a1d2bc6d14906dd43231860307ee5914e3dc89014dbc46e807bb9b92ece7671fa33fbdb4585e68222eade355c43fc505125a0feb266c7e090aa0f72e8498ee011d0fc07621a5d32eba83b10b9da686adac604005bee7597f8294671b1a948dc0fe5ae88c338dabac355b8d3c538185ddddb48e98a9fde13dce9fa06997bf602b32c4b5de56e999be2ed16484f514801c87c101b4c91efc5de3dc0723c70f4e1d3eada8bb2a858c23abc4ee58b5e453400f84297af7b5db960d10914f95b85b927a00d33615e218fe839439eb3a6c39fb174cda64aff66a1bffbabf57cecf1571b8a218727bd1256a07e3e3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.327635303Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/array_metadatas, Name: checkpoints/768/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/768/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7e9a2c9fdb4df27f7bac6d3441f625f56e58b292a46a229749d3a240a00e59273eb23f91944d05e4e67eb4c5a90b897231887a90ead453cb566df01d5bd1629fa79c456651e5d809a6aafe29775138e64b274e1836e2c8c232f1a3ea0ef2ceafdac4d6635b6b433ef9da254d78c153db66462a9c6497473cf98f1727188e1380030471683f2749c3446fbc6b292f95d14c483d15f5191a6098ef02bbeada7f707887ac6c1e8cd329bc2d17b6fd944297b6a1a1ebf8eb9e5a133848db3dfe53688e7809ceca3321aea01601f64ca5a715eaf38f2a28a6bbb7d0f016e5a9a85a7f264c6260984734843d22efaa040d50c7e653485d12a5075950774678feee86cb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328297656Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default, Name: checkpoints/1472/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1472/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6fe809fa468fd6eb54508866da8c77303fc4f601e70aa73f45ed36405527521749cacabeba3900500595ecb814319c4ad6da920f4aa34f609ab20f2d24b0611e500b15fd6959f00ac112203bd392cc14a2316fa8d4ba6a5cc1a9e62ca40e739d7e3dab2b27d41c12b93d5f644e72c2136b01d2c260ac4915553f274747b1fa9753c395290ae2ffda5404ab0814923004cc5604b2a1cfa99030b0535a7cf0f362b39b90753cd40414f7de1fc1e696b11bce9757ac01186865f0db0e2a5e6353738eb6ce8bd505b55566e0b364b96adfd81f9efe70cf28c3c58a86459d5438c42943374278811d5f6ad3814709bf47776ab0ae0f8b59fc5672d8191cd88d9d8600&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326641084Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352, Name: checkpoints/4352, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4352?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=291f0b9e79f729c3b6fd404826c27fa4485fe0aa3a71d420f3bbf2c7ad21d54cc26f3820b906f85e9fc43f8ff7891f6a008981bf6c152138f5b79c6fb4f251c79c64ac9f39fb355ca810912b12b3581b9739bf3e2e68d2bcc0dc0c6c329b4a13a9be055ba1a863928a2e8ed6aaa8152dcf1f46a54f65265575e47f5946209af46d11f2db367a9f49ce2acb6eaef24da3e8c8c9a42352bd7b2f2fc697326576bf9b49bc97bdad87d3078d6b655d594a072f13c8e2b4da20ab59ccfbc6af3ce7b338d8bd759a9122c923d68a531cd0b1a3f76940ff45416d9a2ea0ed201903a25c5f410c3fc92fbaff435be5728a47d3749424093b1a826eb5e04a84b0c7b356a0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32844769Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/array_metadatas, Name: checkpoints/640/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/640/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3d121fedd5200300d083f7dfe29099810e4c0090b72e23a0de27548f829c895088c93293717b98b9546ba8f981f73b5456536fdeeee39deb60b9ab5c7b01114b5967966fdb1789cc998decd6e06872d7bceb3358f2d30a3741733ffc4e2b0515034867f4728b55f2bce05ff0ef858dc2506ea2469ae80cff869e07650d52962d7c8a92ec0519fb1b76e2885f81a52704246ea6d779e4c09700a8f5344eadbc3dfee7824b303e0668b6e4435f909f41291021e7336cad2d6064c6f5ac0c1151b25f8e34f61a31bfa43ba8429fe5d6947d6025ca9c7f3249cc34b01971ada96d906abec56ea23536c589464a20c4667a7b9ca4eddc55be16877d567f2c5a250c4d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32824404Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0, Name: checkpoints/2752/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2752/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=632587c802ac1f7c4c1a9fa392332db0fe405202004d7814d98018e377e19acd5bc3380c1e0533547713224b9728f23ffe2fbcd19a41d1765a7c1fc0870b7e1985e045d03aef1f481f1e0fbd38e5f580703097aa6816007144f20f4e18d0e2b6e54913ba4eb3cb46a346d15c4ea8ca9705479bb09e364d8d7932cfc80642d5555e2c050410da4e8beb6afe2eb17532d133946335c07d34de550c8ff4fe94ee216b7463ee1b7f8d71440e819dfb5d31774f056cc5f7871f72b6633b5603c2d2ba39ebfbe8512c8093e35a8585d96f864400c853a4ea6a6b3be1521a8c961f5ac2b0886b35ff1eaa0a9f1bc72c934088eb35ffd9a08497ec9b6a4f9cdd03b62091&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324841152Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/array_metadatas, Name: checkpoints/5056/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5056/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=227eeac17e81dbb11cdaec7f2ed747def41e9fa46b6213029382525511f33ef7293416a6999f2710988120bba2fdf1bae60ba37ac1f7bcdc9b8887e21e61e4ad836071e732ebde38363f9a18def0ecac5f3f9dcd16911a31aad39219b09659f8d85f28366f8d3bf5314f1eb90689f3852376c373ec6adf0750eb3687077dd69a462b021af6c32a1fce828ae8a2299196b0462e2b2f5ec3572db12d0b2b35b766a59e7d3f34da05e9c509a0aee14acd39141044d9ef1d60c374dc8499cae6c307b057d84789798fd82315e01675a926014b80d5ed427b291ffb11dd5815c2da606000a22bc62efe22715976e13558af09194d2bea348a3d72793dc44a12c94e2d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324954506Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/array_metadatas, Name: checkpoints/1792/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1792/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9db66ea88f2ae0270395288ed5a0644496a035fb1fbac833d7d814022eafba6521b56a1afb62fffd7ad4e9548c3afe635a6bcd0ca06f8a20b26204f93ba1152128983d7e50250c4970c07246fec6fb240aaed79ddb75d4f5217afbd5ab1b3f5b0bc325a0a1c836cf6f2fbe387e5fc2a91f8b455b4fe26022cfc337d8f4f17eafe0b6f4dea142bf45f39ae22333ef20e97cd9b1fbcbfa7ddb13203ecef2d9e04c203ecdf2fc485be2a12b1e5a92eedf79331b7e6d78016bcf8e5bd013bc3144752d57fa5e414405fa56dfae4444a0e42c1e3f46f7f0836cf649d8555fb29de80c81512cf7021a5cde18159c54e23b696117f525120ad9eab247065376dc804f1e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328733245Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816, Name: checkpoints/2816, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2816?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1e49a10806de53bee3077bf11b67076dfd183f3b8e5944302adec8771167ecb888c1899cece2304e6bf454428f7775905baaf96a55d3455f8373bf4b22fee244bb62b84f2b91054dcd8584c807603f9db7a2e4adb48269d326336b1c4384c9ccf90bcbe9483b94bca3b52df5f34a6a6969217ac76a5742b6a3b0868cccc5eb6ade09742e49d4124df1bd51276526177729596a5bc71b9f78e49dc65c257850674da47cec19e3512cb1e0ff5cc632ac670b25260293877d6967f0247a2786dd2ff1e7bcbb9935f4f4de96db0f3d0214aa0096ff70717f65c83ce53c8c1a94f6acfd18caade3f7b3070be97e419f349d1eb6bf81949a8d65a79d8acf292a5f3716&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326920764Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d, Name: checkpoints/1984/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1984/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=125270df3f57184965e3a7ebe37923116526f9513062bc806b93ee468d2baca7bb7f36de90f1ac82ad08b786e26d67fed423377fd58a30b58e13396f314ac78bfce2942e1ccb9c5251e966e4ccdf1aa5b7a93b919bbe842607784eb8b97096380a1e03a22289d43f3af662764994d1506fdf68bbc49dca4dc446976b2cf069a4dec95e7c956c427c55b9cd3f168f3cddba50f4b99e524e997614a6195d8140c2bb10d94e2b5b62c78450ea36deb52868e274f2ee02228edb64bf0554276c2b407396fa062891655550587a7adbfa678730169908d14849286af7a063cc16d1cd0e9c0d35e814a3551df87aaabb4ff6af3065637b5f7ad9ce0fc4c8b6bfd760e6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328746087Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576, Name: checkpoints/576, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/576?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=45e8253959c3e4d574d5b61290751488cdf5409e449c66837d5feb10281140947544bca5be971366de84f777d6a12d7673137a7dd747aca1579a1d9765a9971eef289a713551e78127a0c86093efab1f4686963e8ee17442d324d756be899dfde873bc4050bcf22f59bab873b568c0c47eff6a80e7d456b8687e9ebae1b9c346aa911b3f80decae1e1e0885abc317587ff606aea62b8972e0057250da4634ac63494ad3217a9f7833d35211bac3e010bf597dc3b2a4ecf1d116e6c4f7fafcadaec565cb2c98076f1dbd5d947e5017f6547d95beef1ea88cf16ca064007d2068d500b1c0344e52c7336f42e806286faed9ead7bb875ef2429d29fc46392f97b4c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32876813Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472, Name: checkpoints/1472, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1472?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3fe1bd7375724bee6577d2e562089306249e7387cea60a4296926f3507936f55698e08ac84c2d66820222a4ce8ddc9c1fc69d36887b24878d94db246bac66924b3102b823bea0aa0b71e88dad51ddd43ba871eae26d89e8daf0553d277ddf148008fc9654287f2a46420f4f1f4f1b5faded132f9c0c98613ee75bb516cb5192913837418fa2e124bf62c0e6dafe4f9d1722d9bf52662dd1e066acb9e740c9ff13505718998c8b87dcfc96f43261686a29f444e482ad00e5183081c17ef3d2fa5a8c637964036d3f9b4900ec62af5ec5c49145dd4aa0b931d525f9248a55a4f6f1f2e7102672474169d6d2451f4fbca04d2f9670f1798a2d880d305fc4e0b23de&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328799844Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default, Name: checkpoints/4608/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4608/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=77852c3594b43bce5a0ca39b0a7e8b69f30c4655e717c74fd32e1c441906e5f92c585038d84d00c1ae789efeeeb386511b115d6147a4279ff4cf7b6166643158060dfb2a84bad7d38764a1561854238c8db9750021f27bb73274c0e2f90da5f23d6b863cfdc0c53e3da5b86c6cc17e8329aeda48f533338d031cf4f7abf8020c265409f2e9ff099f333a8821f3338ba4ace14f69850f90a18c76be9590283e7ae349fe920100f9125dda2341d05ae9756f2efce8037ca966a12c73fd9d460eb36132edc269699bfbcf5a7e6c2492066bc7f5c2508f50ffc5803cc7c81823fa963ace7daa1846a07223ef05db1b2da7fd8c773aa687cf883b89e06a2566162ecb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328871862Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0, Name: checkpoints/5632/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5632/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5e1a8f16c17c107650a86b45376ea6e89cfd5e1bbed4e46ae9a5b121a3ab899f25d2b86eda25d35582bf266cfa537f8b0ae4c24107c3d71ac88468e4c474bac3fa72cc5c8683261d3d393e228b162b81451361fedbf6aa5d63de46250ec636abdff080aad72beacf5299c0a9e3aa5e05896288da12e6f6f2373c0d2a53c23fc6991a14e8901067adda84d1b400af7d6bebbbccb19b23d5685ae3985af50100d1b55fc3fa4154cdfd7e4fb31c56ef7cb7bbdde805e92b6eaf75449ebb6c2f0757be53b5f35cc6a19689b9e1f717204711f7574e409298724e37d88c6f29a3c3062434c3d3382a57f16cfd6f79faa94019c57f55c86c35ce99051594979f440554&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328920107Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d, Name: checkpoints/1024/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1024/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1bc2e23bcf0112d0694825cd9f39d187206d71675f8e87e1fe0cc86334f130563a726ebb63c3297649886b13ee1e0621da5e950bb15a695a9ed282664a657b0f01734b76b891a5844620f6dd095f9b429687136655bee38ea9226dfd530c9db0f9303c7b74fe84a3037902a0eab103cda65be6c2bcc878fffefcd13fc9882e47df5021de259b8d8575bd73e337995d36a0d6325687ad728017e1fb765e1bff42ba2c78e32b1d59013939985c30140b40615c64117f6b14cad0bc384f12a8001201b1c0636df7bd86e2794470d8f2a0e9ab5c0da8a3eab1a448553c0b70a882d9ae9fe6a2141ca2063ce7ff94466730425edf82e61fdbabd32acbd3fd899430a0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328330072Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648, Name: checkpoints/3648, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3648?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7b42f1962a86ae5dd1b67b0d7ab475ea837e8ef76e124c34ebb35dde1236467563c7f685f68675fafbabe8c367f8364bffabbf410812c1e6b32bbaa41039791ad61d9a7bb601eb6d765afb203fe1cb8bc2f6d642fd021f5342da6a5e9fbcd726f11a50687185086fcbb8e68ad133864afd37e3f2ee2694354d7e9f5c46e2d16eda8d8b1c2b6e92cedee4aa72d218c86ccce578f740fa0eecfd21c4e1ee6aaf92032b37f608816745f82defa1c7b259a722a753bab4c8eb0607d5bafeca5b2e7a4a157c9efed618af071f5b306e191da17cf0fb7e667c51541c78e94751c4b9155086d6e90af37ec4d58a77da7b62e379c11a27f19182f9e4462c052ae62a7d30&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.325087387Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712, Name: checkpoints/3712, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3712?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2100b80ac44e8fafc7e344535c1af6b08ce25322383a37bae97b1aae2b9a400b7543f0bd62b50d881d222466e6ed71a7980e2ebd9449be14deced7b7f18a8a5058117468b4c1eeeef70c06df031fdbaccb4418559c48c724535d2f833a350d7390280b762a9fa1685d489c0afeacbe560341f0774d6b74890cf8ffcf87276971d3987b1e936e6a341ed3161d09a77d676dcac08775a847d1388fd48b6021231ce967bc9a6c519438d43dcd6d6f017bce4d7e93cf6967a74c3975d9b7e1f5e1d0cacc90c8f1bac9e2d8c0f085ffc7b27d9dc42138872bcd76ef9e11965f47c8678bba7292de5acd1670d4af02860d61e55dc79c8a7cd74f2ce0ae62c7c70a708d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328970223Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456, Name: checkpoints/3456, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3456?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8bcacb09aa498ac5992a9ea1248fab51c31e22bbd905aeae78fa11e28862b917b4bdc5f3d0619bd7a8cc6a8fbb0e1424436faef8bf61d9e66d4597bfdb6ec48104287f1dcb6978339e5eabbad4e26ccd2857378d9ab679e36f241f806450b30709bb4b541a2165ae8f02b51c8bf75b1a478cfa281c9858ecd4beec96b5625ab793dfcbf851cc8c1d0e9cadf6d78698ecf7e8dee92ecf88924dcd579eb2bddaa6a035510183dd5953c301f58148336aeb8fb0f0abc86de2476234b7bba4f3bdfd9243bee891ecc9e704cd9385bc8c0270bf104cfc7595195b005cd096552b69973b8284c8878e5f66eb34571852cfaa901cbf379a5ee8cdb33ad2636eac359067&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32876716Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d, Name: checkpoints/3008/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3008/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=870f3b1e46771b62d6b1273f975fca46b5908a9a6a4b4348faec2043b3930812d663b958150d707fc189fb58eab8e6239b78eb544c10d65782a58b8cf2a15982dec8c0b84f90fc8c8eb982d59fe007db394b98160c5239686cf90cee906ef182f98b7c210474b287c0d03dd9399387ff235cb7e1f1271eb5ea2effe0db8ce205c8c81dfa08d37af80b2c144cb561a182eee24aa9f505190046e910a05bd994b982f25c7bd6e39eb864dbbae5c3145560861f7ef7f5b4a19d90ca17f1ade07d73e8aa3ef698b774e1362ddfc2f4ecf9ca70c720e7663344c99400663a0b74e0e8f8da80028714155155a2d998f1334448b8f93170e584e29b01e2cb1549d6e1b5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32902559Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/array_metadatas, Name: checkpoints/4096/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4096/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7c711dff97765f22836f9e7e42192df98f2fdc48aafb8a020b36265fa39a16647ff70d9621d142ac5d703ec0832e85b467286f53717888ddc8dfb1d3c9c1fd9b8a18020d1afc321998d002c1a9962123b769eff8129aaf981707491bb4a3a266fe798563947b690e2d9027deedbfacbb53ecd02857e32e8f3407576fc5a06bfe886878f44221f27a26f2fc6b05828534a320f125b9f98fcb5c002ddf3d68cd562c4cc31068b9a09270455e6d2cd12e41669ef4908720cc61e9f68d6b637a844c97d24ae562cbace276e0021f76d381b57d99f17a82838e0af2a40d22ada40ed3e5c94e9decf26268bb636686641db685c6d70724ba87edcc5a08f4fdc8e5d7cc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328893375Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0, Name: checkpoints/6016/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6016/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=137e661ee2bd39a2cdb29f22ea7c54655554376491c901e5f6e39c852f9afe9b678c556f7837557ef2667ff5fb9728353f3b3d263985a1763a719925584aff53e2175374d7dbaae4993a8415496db6ffe09ec71972063e175c7d42574ea0a472e91141df71249cfc88335d5505e217d78d617ea1b7a8a826da2faea4b9e4b5efe4ae5d8d31e8d3deb073b1abda7cd3c36879618e5fc98f0079b84f78d6966ebb02277778303dabb484485527dab2580279679150988ea3333759fba62e7b20406071ab2d7e0d10431ea230095eda06ef285cdad7bfe527c3f85d66d652df2bb97071640b2119640ef180669da92f5f8d748221bd8be6773150447f5dfd423c91&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324712245Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/d, Name: checkpoints/5120/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5120/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7221dd7fb98f9aa9aa790fc585b0cb5d5d0ef8def6de0c794aa0787f81bd36c893f87c469b35905c2c92dc44f3cafd401dae9edd51694647c64ba1b1ea9fc91907dc4cd6f95f728d23bd043dabd97666669a276badc90fc9175153840e9937342fe8e3f8f38f5bdaccf638d43ae51a79544fbdaecc3c883e1cbffe133b78d9c725d104e64d33f11d2f5e3a1fc10912e95dbfdd98d397438fbcbb9d5cb40aca780a998cd67b8c46165ed96785b10591a73ba6973a91613c39d421de4c6a1350910525bb39a8a29066eaf95d097158cb15f9a0a3fc0d6cecfffa678d1987c25c8b9bbd3014e37258d615eded0778ffa0d2700332230d26644ade3033b2703b9ba9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.329054719Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0, Name: checkpoints/5696/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5696/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3166e074ae0517f1ba563a7b13118d2ed8dd0a8d3b8157abd3515ab8b499776d37a34ee344ba9ddba07847d7db5a821577aa97a3d9beea54a031b9a840a2e118143a9e9e35727766b46d0c3ef99438fa870a2faf73e1122dce04683667ab519c9eb6ab903be1bb16694fc835b0f1cb3fe47f177e6bdae825ed2ce0df1c38942a5653cf4531efac90d749617e989dccc2fcb39d77429feebb345e5dbef3686c651e0d8f45b6b66fff1e5f8cee9bde1e17465840b4b15b7726bbe53dd0b1de246189a6b64b39d33b7af6a85b53446b1dc89d3a317f7f4a244432bf8ecd64009008c07159d9c7094317b18e533ecc5952c2a19a3ac219d90ec0abf0961de626d52e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326228443Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/array_metadatas, Name: checkpoints/2560/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2560/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=09b4fdf3f1b91fe4e8eeba4948e70064c8c825cfa97a36a2a3c8601557a41af4c2847a297b73ad1e780269eb2c04a4ff1cede51817de2673f0782ba31ad436ca2a6d5ccd985c39593d31073f1122cf8bd0cc109e0f06f8bc39bbf3e5f4739c402ee62a49828a9dd4c720ae7c855d72d99882cc7879c13126dfc8c92ebe81fbed92431bc4122ee94ab211e267decf2bf24153265f8bca187b1e8bb79176cf9b4987894e4439dffcd61d6427b6f2cafdbd86e1d6c76644285bc19898258700e37a2e39087f21ab964660fb094e0a545a4c3bfde9254962c4cae127181661e7e335409685cae99c31647e14be521272a4498a9ec9cd139060860cfc0b316017ef93&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326242406Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/d, Name: checkpoints/2432/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2432/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9e89268b7bcb22cb338bc470482e1e513f646f911b8243441cd0a4589ef01f6a055e50fc269bab6c7d8e41f2e118141f39b310331c08e89d7915c94c78976a2b94e722db85e436dd1cf0232d9f2595be72338a7b7c0e72723fa0340675a8b25489416c4760d606b22a2ca89e105468b5530c013c45fb2f29451815961feecf4a8bbf33527b859aabf7fb1e7810cfe0efeb3b37ed7f86fae7c94c18f9c0b424bfbeec0d8b3c6f2c8a9098c47f7dd418f6f90e1bc9022b5d68302315260a394c5871b04a4cebf5974acbb865442c8c8458237593ab15bbb96703788583a55ea6afc16058b4eae5bef12aeb9714973d0b611dcb05ddab8432ddec6641b8e1a9cd72&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.325099046Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/d, Name: checkpoints/5184/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5184/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=771a8415a8cfb148101c2ea96dc43bf86b6fb3ebb5747b3efd089e1cfe855efb000a28c42ee640027a963c7633938dbb70558931cb111b12f1a7020e8bcf1e714cc03414f3952cdc27a8ba8c2dc128bedd9d4a649ca29b7791642b27f89d03bf3b16329f2460d75875ee2f8b04e3355c51dc2c9e3e51d5ed699108bede2bc1e73412b310c309bdb2e3abaec34cfb0d2e5e217c32e2cd416410423727462e5587b7f9c4d71b70b92bcb8934fe44b645e57febae4287ffb2e5195a8fa1e4b13354e1793e6a4b5fe09652bb9247c81daa2fb8ac437f011b74dfad1d3a3af8bd5acc1bdba9ee4e850ddd67dcdad37c4e70a432a02fa0effd7e612ebd0899e9c30710&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.324710135Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752, Name: checkpoints/2752, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2752?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7a20430b766ab01075d3894e211c93f3bd70e41f17c8250f0ff85e98de78ca3655289d8de0ed8500a692c1787392f124f4afe824fd83127e4a3265f006c726837ac8cce57a6c869e849c631d8dd7cd481a92904b01a1c77d4eff35e4f71d0a7f6534e9fd4a7d248502903e5a27cdbf5c68c946405ca445a41526968ed4a19f490ae6c8c521145c2c0e6ffc90a597a31c30eb7c2232a92d5c387b531bbf6ede6ba33dd80fc053b5927d3873421659c3b44cd0d2dcabd6ecbfc707d19bb056afc7aa359ad47d6ab15fcc7094375a466d1d7344857937b1b82cca6357008832a9ee30d49a0cd007da0841fe71aa9669dc9b798190d05c7a2f620654d06f4edc65ae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326217104Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/d, Name: checkpoints/1472/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1472/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0858f096674ed609f095af82a550cb30a14ca4d83f6c5986701cbb39955d61cd54ff92a2c9a6ada731ac134d44eef2035ee1ce3ac8a85f6f6f0f4dd0e44bef23bc3ed5c97bd99ef77fedfad789583a0c624bfaa20d850ee1516644ea27bfb816d2e094862950f22d913c7380fd6d6ec2194a409f4cd4abc8f71656a9f7206a8a04210a990bf1d1066a67d2ec4adbe0d83c48b85add97227f6fefc1efa896ce9c3778cc3cc4b5e30f0c2dc34950829d2e76fbf666b8f1729813e1496d892c39d0e5e764883849f5ba73ca3a038b44d9bfa783381022f893de0c19524adbabaad83f44778b9e82012501f92d9dbd6a8a23fcd9f40a47bbeb30c7e6ba079db8577d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326109482Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d, Name: checkpoints/960/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/960/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c1eb26730e095ed822c2936054e08224f361b4d875e4779e8202ad5c64f8c061f368fd6f26842af572b3f6d8baefd4c4a8fa4b8b72f44bb5de9ce4ea197d869d1885e343ff777f09525f6588c61fa9f54e851d6340afa3de056cbe961099686abe07a9de383dfe317a72cfec281e6c2961532222c515f1f11404e066b35ffea746b5c2bc177c52842fb385c17a14ad5c056517c19bd870180535bd1085a30376fc52f36d98a39204897cf9ae309c6594365e1d5c1f2bdf0747be0fa09aecbe70e8bd4a94d5aec33a7647bdeb8071a4063f195a0b5eea1a4e1abd2e5931dbbde57bac41b176f4e864bf40ff50e4b8e4a92b57c5c6c8e8e2a7a36ef42691be0fea&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.326242183Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/d, Name: checkpoints/3264/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3264/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1b2a0ccb957633512223bd60ada691c85cedf2bbdb73c9835233d2f50bb6ec356e0d6f6b68117c2eb1978b601e5dc88f7d467c1d5ef357ea50e68477b2a2bb4f50666077ca7884863808401e038b8097db207dec6dd44e81ae9aa270d4c3e6ddf008036f8faf63fb63b4c181985f5d48f4d3cb7cda232d7b91091138551b8050357c05cb3b37724a6da7b871857d4fc15d3f65ba5ccfed98951127614ed56c741d2e25551cf7b56561abd2a9c934b707802e428a1222f68dcbf34020df1023596f8252f2b672cb35b484522116396a330ee5d8401f5d35bb3e4a1b54f5eb3c120363c20eed1151a06e450a658fcf4d2c80737dd970a7e9b22733ef84e9d0e175&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32499398Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/d, Name: checkpoints/6080/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6080/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=58262a302619b98f9aa0c4fdadb0d54de90abf7f612177ef4f3ceff82f1307b0904f6b114f1e4dd36e3f3d05a189e0b94b332c983ebcf3fa8cca426dfc5e45100cc29947aadd1cba6ee53d7bfdf3147bcbda06c41361516292682926ceb90e19e9e1cd76ef44a3e99732bf888b694128907f7d42fed7d6fc3afbcbc3b50aaeecb6209c9c5442a5c1f5d7654b0ea7b0837c40d7b07ed3b85361249db180ce51ca1b9ed899376affa213399bd048b4bebb98647d528fcbb01b320cd5cb761ab9bdf4ef1a7bc0ffa625a91cdc24beff36f81179ded4e25b19edeb2707b45221ff6a6c9b99555e11458e45920ce5de3ece3bae78f1ed8c841d740590452ee87064f8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32868801Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default, Name: checkpoints/1152/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1152/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=29b313f24456ed04e7889ed9a1a804a6aa97060b6747f429b71affb05f950fcaeb64d7569f09125e20a7b2337e5285c7d10d5e99e5a0205290bd38e2c115ddb1833e9d60efd0cfe442b505c885545b46f583823ce9d25327817502141e2215323c88ab626a40124cac71910ee742831bbc586eaa02bd445d099d141aa973acf9ae6b71c9ab47fd7275ea29e6daa1db455188b13269d20c3e4b5506090d2005f3d01b43a3a0da4f92f0896dd52a4e3fb87ca5758bd2ea9a39a6a518b31e0c8626bd5e20c54ba0f40cb290e955f1c7619b50f634100ad538048d3bb98ad96c0d44b3bfa812d4d7c0a526f6d8aaf8175d6debe581e84f3524c4efaf0826758af4f2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328700222Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d, Name: checkpoints/1472/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1472/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a8832f6f19f7b24f6e11e5add58d7fdc11c935950cfa154be7e110d92ccd14e7969c327c39748e0e83c8b4f7163942468089a4dc18f992239c9a2ea6447e5f34deac5d1dde426f9873c98e812bdf8b83b3c9cbdfc5a09a66b7c340dc97ea54bbc0fd4632d0dd2d2a1e03ad8631046c8f82d3af5bac7d8b05db132282ed34ace00c57ce2a53a7837528827ce06ae583a47854b6de04795b7b571836774a8bc9018f8733e809554f46b20ce496cb7cabf9dd808bc1246341145bef2ce3ca21f8553ee6920e3b2612cc9545f8a0b237319c58e98acaf2216f2d5d2b7c9a0eeff591d4e7bd86e721d4de7884f06bf1b166c1af90c2d929c36baff9197e414dafa1a1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328768835Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/d, Name: checkpoints/512/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/512/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3b24d5e40494d6f7f6e77a18acfb9c7925800976ec27eb014c501c61ba31e117381e93e09f042a5d0b1776e404b19e7c000e6f460114e3f8ebea199e1463ce65e468bf9a0702c252b53086c323331ac85c35726b3feb6fa75eadfd427a8d968c7518d7a8891084c110a780d7d0c1e6651c0727bcd7f2f0dfff5889d2cd0fee143d8ef5da50b5a56f9c7840b0ab10cc5bd645819cb8ab5f939553e3c9b9207eb48d42fb7d397456f857bc46796cb020e1eaa5cc4cc1d4181ea05672d4b3960c119159862706aa6804aec593216d81cc8b12e6d6bd7ddd1b46e582145777f19f98c42b49aaa260d4588cca7c404fcea1c6b0b541a12ea839126d7f9e50f79d549b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.329499644Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800, Name: checkpoints/4800, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4800?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9e4deedac55b1ce7754d1ce9192fb22ae1d5299c3c7470e664754707a0b8cb82643dc76333ae4d62f754abcf4bfa564c8d97225eac19e0f4b589d8cb4b2c4c7058d8e686531c97c9d64cf80623077e2578fead7c75b6df52223775079ff63677ac1d1baaa64816e736dc5ea6aa162615c5321ef640e94d2eb25e5179c72f4ec5ae7c99f447d45c85906b829986cc08d1a6ec8d4762e732087fd4602484a5275d749612f408246b356da3e30719950e1fc5684ce3263a9e3bd24ff8f9bc75d10ef3f8d92ed49a13f97db101de33555a030516508ba165b55eae04f639f283b75ee3e9a540c12920556a3b958976950e6b194b30694b7e817ce39dc4101e83f5a7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328708791Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default, Name: checkpoints/3200/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3200/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=40a746f5d7f783656299ae760ffdbf9c2af9e88852723ad0ed66d17e8833a374366bdb88a2edd977afa29ba27408922734fe2b80756f07ee724cd8ec1758e60d9e647ac184040bacf75e619167b6b791b4fc449371709df1b0064ae5af5a84586af41ce56348901393de3a29c4c6bfa0655f6e0f78d959b0326102953ee2414a61d1f126634309c0da304b1d671722a88af1a898d53f9c70b496d3ca4094b3191e63f602a62b2900c0f8316da19368e7923193487353c7cedb6c4265dc9500df9ebc8ecc94a4d479e773d546d300e3ceb516c65c5d7126cd0734972d5f7ccd0a74e4e53cc7a15a6334c4aa9fb989c4b2f50f56dd87389df381916a1b84328eb2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32898167Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d, Name: checkpoints/3904/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3904/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7e2d9be553f29f354648b1a6ad35a8f1f4e329108427b53b4ffdd7e49acd5923db102fb2663fcba94785431016ef359e694929d5c20f78b29384d50d5ff6d1bbe62ee6953b4b4c548eb18b09a2947a538639b80bf60604ef4df3a22c3fb17e8834c0f31970ff02fb96a0abfd56af0bad6ccb923e3ea1b0b573de5556f22b507b96617f66d3f0c3cb183e74f8edb6db88518e9892476ea99e9f93316921e75c55fcfe6b1e886cd191d83c8baa6e1a48e8b150f773bbdb0bcee75e7af64f3ec460b306e02f33f9c992eb89833008187b39f17aa3d0d3c456e019c4c38f50633c2ba48c89f50334d54e0701c1f2c97e001dacb7811176539801f203fb0164ea6944&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.32896746Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832, Name: checkpoints/832, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/832?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=69c3a6e7a57ad2ee0ebb7ae2b3c05b2b1bc5a21118cac5d2bdfbbc174071d9e964c09ca2be9c423f28026c4e59fa8f4942a8cb587b7484271de873c0154d7369b298856b7742e437675e50136b03ad34fd5ec94dac0eb7ad32a0a0a3dbe455c9a388b2f56bdb2c4c1e1fac20f8ec0b681c95f1ab2588d0f6246e46bc690b19c28f9dda7ef09ae2d0e9c15c96382b186381e2ed8348fbe900d84067fb6f8ef10e9119dad5af0bd6281bd89f2c1761cb50540f20f4d230272902e64107630911b073798216ee06039a085fdd0360442f635ec42da84e4fbb69597c28181dfdadd15c5b07ab6128b888f3751dc536cf01e5d3040c248e898a37cb9bc6ec16dd540f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328925541Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default, Name: checkpoints/3008/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3008/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b68c10865830fef3a5b9c16c77502e1a2e57e03af246973da079fabc04cdac1fdd788791a458a040c049db9360ae5d1b3c817be7302a02d85a11200c7fa0270c8c2c44b3bfd589562082151ecbd3bc5ae13053673bf40452731b75a82dd4b8059c8b2ff67b68302187da19ece37f29a521790b58c5288aad01e72eed8713d438cc36357b7cd17b5965d4e4b744763c76d184fb978824cfec0dd0b206b95a01e9bdf4f8463fbf6004c7bbc10db0b72177ad3c11ee50a5278fddc795898c84de15237c19f207785e2a9cb201c06c83b935f81bf770dec71238c12e46803cbe70b37cb721f8356c822b3506c7e22cdb082626f726d644835f37ff3cdcf01734158a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.328971579Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/array_metadatas, Name: checkpoints/5760/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5760/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5a5d7dc8b89c4b456e447cd0ab7c770107e588c522427034ffb38a31ded6bd99ca19647496453172fe1893142ae0482380854c8fb28ae12d0ca948a94ebe989c409104c447694fc7a178429f438031ccda097f7d1ec5dac6c8bc317aa60d8c0a011c4c1ef1213aaba6f8893cacd4e0dd5a8767dc08c50dd52c861fc6de58ed0f70b6b13e63b06990f4a8aaefabaee4914b796b5bf8ffc0900f0c95d8d1bb6571f7d671a5c9728e842dfe33624b8bd2bfc10a8ce5b567227934eec1d91aae394ea563c4884277ec2fb3a13ca6ff69edd3fce1c592514dd39b55ff83515104c848102ceb6af9daab1e61fec78d9278efdba307f6ebd3b9896efb291ca0c915a6d0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.3290257Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d, Name: checkpoints/5568/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5568/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a7ec784d67a15c9ffce2d80efd00f4235f84db9772308eef6c08c3b5af72a15e455aece3ac5d31ad79b404e032b12c7d1e13575a7f9f06f8086f8b12e510982579db1d28925768414539a57c623da7049c9456185a80891b7b5732a91b9cd0bae58699f2bc70791c80c07f0e268e4e4bb345bcbf4d9d1b9db9fe3fec9182abb6961a0618b56c41aba123cf2b82b094eee79ecd84e3e5a5c26f9472e4611d4d3da394b17de847957bdd4b57616e7fbc84034ff8ac2bd8c8aa9ec88d450ef2b5f1942d7bc11a08e5f47b3751644621168aa19d8c7ff27a6c368772ff19fca20b447e1adbc158768e5c6af9896f6c20b24aa3614dda6a96c9ce0979fde0789a5a49&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.329189988Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944, Name: checkpoints/2944, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2944?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8f5899870d161931414300d5c3de4e9d274bcd429e07343a8614f8ae784dd6588ea87f7ae4bd24d35defef70cd12d5da8364e2d5c113fd9c210608d5b59b360584371e67f7ee47f61f18d9903beba1bc1c3cdb7319e7564af7a1c955516f75bb9d6564d331be05a0496177c3b8ad11d1d46b842daf48635580b785b7a1b10b79b78b11da5d42174101107a32fdbd2a845173498167113615f031ab534715897d1794df8da6128ae9f16b7a9452a44f281aa149aad5c879a0fd03e5706c022277e431a1e446e6c94c7cb267f37c617f7ed07d2afd93a38835963947df58ae6700e5bb126bd505aa8748d506a6132e64076adc96f02dd37bbd90352be418328a50&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331540473Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432, Name: checkpoints/2432, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2432?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=07bd6e09689e1c56e51fce18563084a43579b4735180d5763067b7082271a089a57e37c8158cc0c875375e16b4a09f0360aa025890d65392d496fc52361a6c458bc36d0a66098572c69eb56be8ebcd911800e8bed5dd773bb37e640c7c04b8fd58fa924b0d5df83b18efd8c4feb1ccf3370b3766dcd586ad2f2408700cd85100f6d39d3e6a9896f8e861076540c7945766d02ecb24792501ae0b3f0486efe28242d377342bedcc2bf75e19faa849d7fa48e6bef8dcd39b226c546656be239fff4a361433774e5774d3e4aa0d90f701be6386fd1e0384108808fb390cd8f3e240127d1fd1c6f78037327e9aa99acb8b5ba4e3913891cb6e371720f817c5be58f5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331580099Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/array_metadatas, Name: checkpoints/2048/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2048/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=597ab9011095a667c6207ffbaed68a8ab7c4d3cb5ad310bb1719b25475f3c0e84a1811f976dd0c8c3330af32769be3916b4dd30a8132f87bf891fcd1dfdc4fdd97a7a46445218474717643247e4dab4822474a43718886f92c64fd5012358916d5d82e844a1ffcadcf176f9118f29ce3e009aca30220834057f7957def8d3c9321fb0a585ab3707ca596e208261be796438afe82d3c9df83ffbaa3a333baa777240b2b6f5389a984bb8f76f271c2e49f35e0d2b37838defff9ae2a7907e5cfec1496b73e8e9e14369ad4c67c999f495511d3774762e7e2acd0e6411bce182d67dded87a25d7820048be24a566a083a768300ad8f6c4afc4db8d954e69599ac8f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331511094Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/d, Name: checkpoints/5504/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5504/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=210584c5765f905b1ca46a285b4583f064f2688d7db74a1b660836748b0fecf8f9a8f3f80c9436b25cf9a9e817c47f0ef24b2c4d84c659d1082c2e064946f43886d55b59fd4fd6d11edc97556203ba0d3037ea6e1c6325c9ec975e44f2ca36f9b249929467b9ec8772e3ee60c30e2a72ec0caea4a832c1813a82d4ae487917ebad9305b28636e5a0502ab5bfb2c1d34631290508a5872c341af75b5806ca6417d497181bde01514f2c61639939576be5e0cf8e728b0cdb96ce425f92e3dcf183f564ce636d5f6c63bc60b0ee2bdda9d1619244fb372bdf81cbb96e3a823a76ee4702379a8b3b31ed657877b34f844fd736a038154f2fd1460974d23bf81de033&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331381187Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/d, Name: checkpoints/1280/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1280/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b3bc09223d1afd43a928216bd1a3acf341ca0180eea6254027e99d3a480159b4e3303cff3baecbeeee451300d600fbef919e3a87752444ad6fd5487ddc8938d0fed5019d088d7ca80aa77ec0e16292bea98c36b1ac8c751681293ebe0057faf7d1461e4f8fd4ee6d57d369de6bb90a48ae6360afd16357ef516929b6bbf44119edf832e6a07ce374cd8a98a89dff48f5bf773a7f13a12b3ff0b90645b99238ea8810717971f4c0f60b27fde1e6bb5e52f42b7f01c60b98df4add05f17d0fd7b29f3dc432994965efbdce7e0fbe7f113a294638c070cc5ad1ca64c92547d6ae208a6ae57475ad11f50ac41e6a46db9e0dc2bd28be3ed5c0e7922e7989b291d2d4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331624501Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/array_metadatas, Name: checkpoints/896/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/896/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=35453be86cc0d9e2b835ce03a5e1a6925669ab592dbc0d1cdc53f90206f36f1d605569a0dfafff0c291b7850f0ac9d6cb35336bf1fa41600f80f9f5ccc9f9b0f4f2e706cf2022237b06e196a17510801cf53f1b5cc558e51e6d1fd96dc2467f5b6efbaff344d21ce9684b5447457b562e0ae444eacd2f0f5c47f6a398aeb5af2c8000992adb49cd8e4672d418808f570fc67cbc38f1bac20203cf7b069857189464281c19ca8452064102a6efae325a05931540a2f9bbd480fedd33fd64dce73803a3b114408cac295820262ffd11b53d23a294f9b5b643f7bd8312ba9fa5b6d90440e3f621a929d0b96f5db3d3a2bd46c4860108614ecbbb05ecbb09e5bee10&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.33163045Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default, Name: checkpoints/3584/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3584/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=87443a955bf9602ad7a110c653e15d847fb01cbf7a85c19225faa1fa04a607490844798d3461df1e594a65bb630333ef7e4040c1fe9d13bda3f40e537efa3efbe7c957323273a0684d402062c5e83bf38b240a0176b1e821c30b7846ae2a02608fc528594377a366ffd72c1371b847d63a265cee2cb45c899b1512dc235de3b3a79b81ce4dfac4fd0c0bc19bb88ee1dd76e6d7ade9bcbc6cc7151d05a3d1311d26d044aa4dd6d7b7d1ec5617b6af2e51900f2722a785c7cc62bfadd42b99d745a43d28ba2dd781738a15288c538ddee792498e69256963e9269aa9b41f23492a5b01702a444faab58c1fe4d35f87b19d6debea69f9f89312eb4d1cccbe1e7674&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331678168Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default, Name: checkpoints/4032/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4032/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=060495e7c5eef41db487d9e599bd0b445cce33b9ec006f08d08f3bd9ba64c90bcaefcd1cce8f3da8bce1556cacc3efdd7b431f2ca643cd8739b05082c814d7892e1534ca3be55b77e8108cc6a9e4e284a49a705dc203e2731b698c24a670057beb6dfa682e3af05e88eaa8a850e38be176eca560bb8b2c30d97a26354bd7941ee4524d770a153bf0d0ecd06035a63eb4b25ed6aec63a0e53abbf6aa6bee24620e80437a8a2443126ded95d71c6f67db4411a5a216cd92bb809ad78bc0a55304508e29f42535fed7d2d404f2362d6563264fe1ee6d2cbe141f1e2d20e34310b127585858f899bf1d9ca541f33335251284a14a98f652b92332e8aca5184297e92&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331689835Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d, Name: checkpoints/3648/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3648/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=904ee75e93991580b237d37ddf2d2fc3948107ce5e1848a4290b6fdb374661a812551152ddbb4732274e24702058144e0998697e164536ac7daf2539e4367936f4c046d6e9c3ce9b954f3bbbe5e122912551112e913fd58ccee446723232b47bf11f8a09103df07f35b2fb367ed9d8755c88ea9689765cba374c8f0892b1e0a0986b84af2a21cf59b149f91ebcecf8024cde494c0f5ed0c5e3dd62882d8f91e6199255c1993e5b612642561e5cba0e37f9cd933444621a8d72bd8295038da35f847ee6d97ae5f06dc7dca710bba7f442a1fde5fc5be4b40c26c5a0c771fc47ef2228e7914cd53120317c72213f86f7cd6fe8ad405e5c62d1271102dbca37c17c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331701028Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default, Name: checkpoints/4096/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4096/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6b0353bb70fcf89bf980b50bfbb03431dff29a2cea9c4856e194e66b8a7cb49ccf314630c00ed74cabc5c9e585fc2ea674c614d4c5f4982db4262538bbd625ce24df5971c0c86d34737cebc0dfce5f150f05cb9bf52a59c264cf9d60f011552acad18f1d4ec230ee4c9e7f77d4e6167ea96e7efc6d2eaefb653233b81afaa7a6d1f0567d077d7211586ce880b40b684fde17baa5f34b4fee84cb3ddc68a401284655fa194bdb2efbf6b049186ec09f9c62a0f5c0e098ac49764bfc2f395727e3a8525deb78b3281d504645785322bac73b69ce62874efd317ec9896c6ecd7668ea1a43c2e8f1e0db1e68942cd265ec1d3a7ee41bad469ab9a2a558f98acc1b14&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331810827Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/d, Name: checkpoints/2112/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2112/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2531a9d22af6a409871828d25ad524378b6943a17ed23e362b7a3a507fcf0c7f9dd14155e5290616594cbc24af772843f261bc3d32b6b90f425d82e579d2bac23b4e88cbb5bff030fe44b922646b16f462be5b0ec6db5873a274358a48f97574bcd6a65f19033b81168516242063c8c07e7db141f05a627bc7eba3ae1213c9c6f25ca19b75dda339ce1e6a6486b98ba2f387228180b7b8efe8c42db188505cc5227e05b3321f75462ed2db32881d422ce0a0c004d1cf5d2043507580a60404d7024376e4554780ea488e4c87a40b9fd5c663406c8b23f9f8949ed9e9ad18bbf517207b20626bd797b350f257ba91b4991fd6e535542e611f8c2212cf12c5aaac&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331833535Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/d, Name: checkpoints/1216/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1216/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7eebd1de5c1f7341928204e9789f3c7c95fb27cfa7b78742248cd1d2205a0badf88c944cc96626108b871bf85e9dec934d89028cb1f33d1604b0fadb56831ee46950ecabc033cdbd968a058d8dfa98c6460c1cc16e3fcdb796d5ba24fbea56d7644a66f9f7069ad96725a81f3fd3496cac9c8c47d5e61a1a66fbad5480cb8782811958a48522add7e3588d70781bf38b1590685a81cddd0f2e8c70be7ae48cc1d3f319037316fd55a22b12efaab1ce01a5c7070982e13097e2aa979778cb5d607b544e8139d471baf4df5c56ea0e70646b6b3f555ead9400b262304e25f3c29dc0cd6320b88e5a8de463cb8499498c7d5383d750ca44f9ef1ceab703c9b8e135&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.33181634Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d, Name: checkpoints/256/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/256/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=306462fc3048c4e96f85a11f374c7a51ea940813c0de9e09d751c7d31b8f8fa5ca433fd03ada09f4345b7a8bc235c25f40fd5849c8d5657c3960d78b4a98943205109e70b55689077cccd8d29def3cdcd50764ed84f0cd17a98b958c1f09fdec90a0ff666052e7f1b2bdbc927a20cd5e1f3e274ed2e5dc9c1fe9bdb8386edf4a3cd4b84e6bcdd8d12ed72c67870cd6dc4b4c09ed694f306bfbaa18da67941375a9011021822dcba387dfbc78e738ff44b79257259fc8fd5fb08a1aa5635cacc7b40ccde4da2f37ab87f5405b654baf42a37a2b242bdab60b4a5747163dc8a63364e823bc9e4423a0d69a01b1c83079a4301d31cb214f559f05fd358d451b3c3c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.33186299Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/array_metadatas, Name: checkpoints/2240/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2240/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=18ee1cb0b9f60908e3f65ca49e129a08ec7091e19ef6055681801c5d9e207c4badff3ff3cff8e6576b862eb4de625b51bc2ee40a36090c567c71bddfecda6935de772eecf5359241bc43685a5873c81c9296df7cad18a4c89390af274b6709921c7dbc63af57380911805d86d614fc365ccf851680eaae161f3aa4a5c57eb75efaf12fee0e232aaee05498fab4448e0af9f9d46800058ecb3715f37323b1da525c1901cadddaad63304508c8e885a42aa73428c20047fa76dba177ecf4402cbd81874484f60b0723bf4fbb3370071d11353a1704ed181c38e6fc2be1cc092be1b8436313c6351ba4568974603319ed9d685f22f4b88414fda267df085ddd1013&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.33189555Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/array_metadatas, Name: checkpoints/3264/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3264/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=39d64e4d759b1dc29d30846f1ba7038f6dd2719c8b453d844b70e0940320f046d37bc045a2e8847cfcec235a221c687c16111d31aed14c123b122635794c69dce2cc435bd9f81b3cb7cf902b16cb218f7a96a2c5384e6fe71a514464744aac87cef91fec9e4dc3f47fd6282d5b7473be61de780470f1986a75b7f9da2f1b721bd08523739ba01394312232f24a01c86d445b2dc54cd992b83d4a359ff2c8a51dbe028d97b2bc25471dc99119ccc19ad738ed421a58e5a0aefa9d3b0f0f532d55c4377679276b71fd2190dacfa94699ace1148c5a8df36d005a233f5efb4f494f5e0cc15155c45762aafad396223b87c14e8add21021f0c8d2055a5d00c98e177&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331904054Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0, Name: checkpoints/320/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/320/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0e575db05e5440cb108744bed5ee90c11572c4bba4432998af68ccf306437cf8587609ace641be2c5397df3acbf513b05cba5ea2b07b16c67b5e33123df47bd1c696ffb9ac6128e5a1ace3d58e933b31dca92dd3ef7402be16905448b63d7aaa68efc711f079f2305eeb7326a8fe8d0efb6e101984cb77129a76f9937612e47ceba027b4bd533355b6ce359ead8a9cc325021412f7e2d69d52b9cb0ee6a70ab86b4be589234f02d72cc94c2c412cb22c88dd16d0e630cd2c289044cc95aa9fe9b2cd62d64cd978bd524551b1c0a7645bce05174a00a21cb8a27b577872b1a8524b4dcd846ab36c3cd5ce5bc118bcff5a99ace9da50685ed8b16d2471c202d9b9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331914823Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0, Name: checkpoints/2816/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2816/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=466974256784f38c84707e3ecc0f89bde5e9e231588e6d9d627619cfd4827bbb4dddcc045d7f4a1b3d84728f2a3826b7adc505bf5a108cb20ea68cf5c51051f5745a1a88e41061f83f3b0d9df0ef217a29232e987561130a5d6451a152cdc8539f55bb9c655039038c46e2d8148242c15f7b2a7f4658b12cf4252802ffe2ba923bd1c78a88a463ab3631f04e070dfc747fecb69afba2bbe952930a3be48738ba37550e223133176a9aef9d52cf102ade2f68145810dcf423fdfabe12e1a6796b886b894fe6d1923e4f46c32a3997c6568607929f0b15bdcd83af4223b8c30742b5ed60d306332cfdf76864c3c7142b17ba489ab5421f7f68b71ea4957c8bb583&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331933817Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/d, Name: checkpoints/640/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/640/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=911e58887ed036f656c34264bc21597f2f60a0e5c3df24446a693e20be452297728914c8282d616bfba8c99f20cdeed54a00615f1e29f244a15c2a90a54d5ddc40107d62ac2a3c36183fd7fc2cdf3d5fa1d9c6339a57f0f176c5772cbd149f172771797dce611204b39d4a87266ebc523fc0a4a2c279d85dfb8a9584131779bffa8fe4d057062f1953632ae515650bdd1096771e53c04189e51ef668bed8f43c268e13c2076e5b4c5f50277e6acb459bd7f5d5a5638e4124e01c0ef205f16e266c93d30676f033cefbde6d0f14cdc6b2b727fbc1a742644dc4befe85e1f9f203ea8a34b05707e0b9d64dc72e29827bb04cb67349634483fde649bf7725a52f5a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331955999Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/d, Name: checkpoints/1344/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1344/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2c0a7b21d3efafce50715d74050422f2e5a3aacff1658912ac24fab0b50b79a6f6fe7e86bf34da366a23c62ce3baae3c866c4a707046143830a4f102b4ac90214f93c54a045d3d98c5aa3e14a50bd9e61023a522d64c7b2dfd5ed8cafc30587dba43f9cdb7d758fcb1463913b2cb34100c6bd66558b972c8967201e42aa8680951025714a4ef0bc011e204d89dcf2fd650e0e0f2e4ea5dcb99a6cc5a024124dbb1eba36b67348fc96034dd2c045908744fb9221669e7ca79c94d7dbd59a17a0121731f25fb63f6d3d02a215139bd7b72659293d06f2831d1242ce0ecdf2214f070e9c7dfa3c98dfd759de04983b928b41c538ab87bb2c6c8cf9af54a29601292&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331967394Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880, Name: checkpoints/2880, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2880?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=abe325f6fab0cc94524c2b1d5fab0463e8dd59ee2892b9af6f5da4290645196168c930d56e7d553831b8af340e78183994f8ea6a6373090c86f930f48f2e006e55d68054c7bc5a84f8f4fc78754b970331ef461bdc547159014db27d246591003faf653af9b02dfe3730a0c44c136eac4688724048d14d53540a073455ec562657531f5c8b26932eb1308698c652136e712418346edba6240c713785a52221d5d57e915ccb4a03f5f0d5c3ad0e17b13fd679219f4d129d35279f4974a15378ca9fe29980b9645280927d37cc17a1cc9b8a54d1eacd3b661919c0997c1ca0a5959834e6d214470e1658a4f27517a36c08d6cde12e0fc040505537e13c43ca6d66&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.332000833Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/array_metadatas, Name: checkpoints/3392/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3392/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1e1fafa07fb1bf90ba5c8745359a4ec095b154880c8740c49375ad7bcd3527965deae2db45220e0b73f3402289d2c63aac45f1ba8ff6c7166ae830eb1fb977f33bcef196d8d3fc0ae0d7aefb15b339a6fec13a60370a09a899a5e16f61245aebe9415fde29697f302575fa9dc588e83ce106b706853bae638ca3316424ba429566d044a972d91fa29eda6fb37426cc8ab7f6907ae9941a6396852c518f5eab063ef5c57e393deefa489437ed9f559165c647806b08dc9ace9cbbc32c3b2815822cc3d9131c2818987eea56da349775e920a1f41ab72a907c797c4044015ae6a39f8e4b1fce2ca0be8c3a8aa35ae5be2c0eb78ab215d8594281948a50598cd1c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331979638Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d, Name: checkpoints/4864/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4864/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b8e030fb8be096e837187550bc71f064ec39e0186e7226d5cbd36bd915aa5de578320a596dd2132aa124dfc2c782ac5bba277276db66d24c268aee66abdb0f520a754b7c4bb324100f350e67d9f7a0d1c10d6ec58da35788d48448b253c704539af78891c3122c5cbee1370ffaabca387cf339a403d39331ca2cddb559b6f1cc21ce47ceea42201f256d3ad049ad588d0978a8602624ed83e5f7907830642672cc568a1066a3834353183b73c98b69eba4a5b05e0ccfa796655843a27ffcd1652b2d5c747e633921cc8cfd0c75cda994fecc4db5cf41cc1f16a1e1175bac37beb52df03e4edc410540adb23cdb56f17555b39328ddacd8288fa884d04d414fb8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331972272Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328, Name: checkpoints/3328, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3328?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=65a9a86b86eb094284e1aa992f286491c1fab8b146fd9f5078ed38b1a5439a4efccc6ad599c56861d52f933ea7ef768b30632cd9f92e243acc7b4422f5e1db8947555915ef46f26fc8e7f767b41af26204063c42435200f3f02553699a4dfbd6fe8c35dce99cc6249618d6b3d246000569461375f7e243a6ae3af5ea3f35f1e5f2de525bde1e823509c48cfddaeb071ce9f22f092ac6d1f33829a0720003721288c8a17191c1cd1cf5d71707562e761955eae7429e0b10983ff5d343a8edf33e76dfb433998ef1252b1618aef7692d7fe025dff9dfac8c9d0ee2b1476ddefbc2d3ea2f21c1ffadb820638d90b34801a4ba76210eec93f74ecb797da913fe5794&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331962649Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d, Name: checkpoints/3264/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3264/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=56e59c5528a2c1859f488a7ca4185f218ef4c3482b8e15da2d47b9d17d7f032b0fe37eb4687449f0ab5a4cbaa09014bec363efc8e5a87e9a4d9e20a7ded3c83c83cbe130b1e9f34ac2130b0f902ae97a07e5e219361c157549563b8e055800b4b5e15fa1efbd7682424c83434d17ebd4d41378b8fd94bb41f33c5410fe2a0420d698548d5bfb75eec3f5040d203b9d2c7c5736812cb8d114d6d0ff89cb391844226bdc2c2f65f73d3258d6cab759280950f15ec1257eefeeba13c4eb293cbdb82d276c89845b6332838f444aa247acf96c36c0cd257ff0dd1f3ef304af5fdf27d49c0cfa80755eef350bf4a0058795762b19764eddbee5498540bff036a65dfa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331912066Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0, Name: checkpoints/3328/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3328/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=badd62fc77463ada7313c59ca8112f4e903e1113562714bc094e45e9f14890bd44593b991f13954df0655bd11b5e878be1177d3fdd464e1c2ad131dadcc9574724444bf4b1df8556fc64bcd83367c7deedc446b532b0389936954940302e4f513ee0cebe2d95cf55fce8e77a637c3d19ac31810dfd59bb4a6d08ef893428ff033f2a41abe84cc1db323ac32f54947a21e241598bbe749e3a1887000a25cc52a101a83b10d2536e37dc7efbbd66ce63089323bea9bc40b91fb2c8d88731994237db39c02904cf43852aa3504214d87e14f7ad19b0db4827d7a1a7ba2ee06e9bdba5d9ef99c5e00f896ad32d854f58294c80e615c4e63dbe1224f5bb92694cc882&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331982254Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/array_metadatas, Name: checkpoints/832/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/832/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4b6b5a786dd9126a087ac8781d0b9a4a2cbf0176a2a9ef680d8ec52f3203fda507af2686384d01c2c6de07458e5403995f6ad72c335d13aeef0bd613097c6427b9537a81dd25fcf2916104deb8d9b3e20bd4c63468ed37fcf3157e34b8ecc972e4412b577fee8683bb59e959758eb3c5b488b735b985d9403a1223f237c5633c40fc4d6ddbba6756db20a96eba3dfdda9da056843e6692a1d456c05c5f63088cb591759e6d2612c5a0c0724bfae48787f4be5d21af41cbcfe0267649f0c7099c578f4293ac11290aaf9372403d91c380d0ee6a16cf75cc5ec4313a69ca752dac4b6a69a715fc5cc05c2114725076f1ac089b931c0a346d5e93daf74abb52f9ab&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.3322219Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920, Name: checkpoints/1920, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1920?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=becc5fe079bcaf22c00eefcf46eee648ac18594e5ec6d2d3d02c0e00c13be7549fd7af2e8813d2b09335a46f7a0042adf78f8b5cc54ef6d16ec07e683e5d87af2c0ee07da53a5a4bee86b0c27cecfc37d423c72c5fb276a61a4bd204bc5e1c0326163757a887c7552c62336cef3e98795422855df11ec2c6989a458a7d37c877e501994302c49c442268e814505932c632effe86b4fcd9cd647b0e5d70821d8c4f645e4f88312ddfb7dcbdabeb18a2fd9b4245cc80d76d6b3294a50fd23a1d7cb37adcb106247f740481e85cc5a69f4bc26701527e91d886e2db9c8666dc83d6a944c74b7ca203d9ea18002c78217cfc7f6514032fd5f646bdf4b259650472a2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.332254496Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d, Name: checkpoints/1728/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1728/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4b26826583cb50cd9e6b752f27702fe84c9c6c1c6a3cb793f4af6f8c096926e83dcce1a8f57f2fddb831ad7b5f298adbf064bf20e6b559b8da7d41fe77fef804ea8e6749e1d77147135e5bea35fbfdd4d6730a4c70e8f40171adab3bd6512968dce49fac11e5627b8b6b16ef67923808a50a72562f986f0a7312d98528ab8b9910aa425b6692bcc79f23881ad28ecb56e2000994aef2751919cf177496bd0a9777ab228f1a9a317379b5f5f9667399252ea3b09d8036b8485136a10362fd9615e962684682a28bf445423f0d2fedb8d13d6c838beb92eeb09a36a7e4d59e8dc53f7a76be36c68bb6d290060ed8e6c317edc3528fd80a340e9233d5fb83775276&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331954845Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0, Name: checkpoints/3136/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3136/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4fcf6540bd1d59803f8a391529d091ab08dfb22297288f0a7586ae8a2becfebe0cc0cc4a481388dbdcc9d7b8194e54d28a502d20096662a35b2bd8af209932faba5ed15998db1e4ed3b2cecbbe8ca302c681eb17acfe41a05244c8d881d7595b07932c5ec0ab89a910cd09083007ad8d5b9c5850623316541436924f106315677b5dded3cddcaa50f2031fc5679013460e8960b77b3820db5edd2525ac3d28179b938eacd2055ff70f5eb2cecaeb48af5f130f18cc066559d3f8175f965322fa5c54f52cd40bfab2d4073e7425e675125ce4091701be1d5d04e686707ff18012e4da22138e3a7bfd8fb6dc68e70f53573eaa05de21bf2ff6103dceb6cad7341e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.332296474Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288, Name: checkpoints/4288, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4288?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=bbe53118447cf38f0aed906c633cb6d942bbcaa366110a8866158f348573ee2aad45304826863b1798e48b2e7f1d5e77e422fbcb10b520bc78c5c08217b73f289c11bfe2445c957e3c1437e85c7de3e7a6ad7f85151a6e38cbec6cebd65471b7b6d736e4ea637721725050fa149386d1ecfaa0ace6427e41bc6e0d366ad325182005dae6a6bf9004d346f31a934b39f67929c7297236c5a85d944dd960da1461d8f3d6bac915409e87f8d75486196ccb137553add4eddf196462ed6f7ac262fde8cd07f7e3c8a948195b470735f398b1b2996df5fd506de65906761f995d6d0d6b15f5ad5183f3b4d5ea89cd588601c0bd7eed4695e4349542442e3e70102e1f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331919Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984, Name: checkpoints/1984, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1984?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9a968f329457ed2552ad0312aa32cdf612e663166d7f02e7447809395ece40f7c49fa7ce958cb78e0bfeb07bfaa65dff7856c338473cbd2656293d213be0383b7093ce00fc4c4aa54d6e57f641494b7e0be11e6e06f68d6c93ee236ff2ea20b1823ab335a19ff1de0e514e5efad0785b71cfed4bf43f32a0801c0a2e56fa3ecea815cb450ae42c806392c5f96a817dc5b3baba2d1e58189d73b1926fcde313063bdd4682cca531e3a5383b54ee5b0cbc8bb53068d09c26679617a2e20eb5b50933a69bd185390f0290f6f2d62c763f95e48f68b3f70c078f398160ca7922af9f9fcccf3a32af6fcd6b593cda7ee35cf8d652348c7bfe5f628050103d4306f07c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.332444218Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440, Name: checkpoints/5440, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5440?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=372dc1a0dd090578ad310be080d5bb19063bee6955bb0a11f432a59576c28a02be629134f788201a5418b93459a57c2b33e47b1020f6459e3f21065300dcec25a814f984e7a6473331511f3b0952eaeb8e4396419b8c9424e06e01608d9d68b3b09d0a825ca73b9bb0d9300b72aa5548f17911969c6ba3aa5195d63b3f376b9c4a770a1119ef5ad3c6e151ba702f026f4595cf374fd3dc1e47b554cca1150d8a5e31d3dd74ce576dcf9cfc312b26117d65c49057c237de3765a70db8c9cc9d8875af07736e705c3e98e8adc8ecdc55ec43e034aa07a8c84ce15ade606238ac37fc01aff7395e95f161ae02a7af2b0f111cbecbe9034754b56b164654848f98d4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.332517546Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default, Name: checkpoints/5376/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5376/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=81ec02d9f827f4affd952bd17df04f535d4755be5271fc7607001115bb8f224debe198c36861df1d347d482f87b013e76eb0f9b115a47e7bdece328cb838f2914a419ea89b1bf7c0a2b1ace402cc19f4f5ad309ff65889fbb36ba8e1a4154af18e12a498b31767dd9a579fb8bcd4896c105805a8217e8fa4fcd361e926573bd5ea06e376ce4e333c69cd79e655e12db4ccf0cd1ddf4aa900765ce107f348d4a79d58ac6bb7c1910f33dedac93b6ae3878b4df8029123f9e93e4387d825be0853813dbab9c3603b4643055726d68810da6e3d51f96a466f68cf912fed43f0a5f73036e39e471dccab93784dc1f26c39882cdce6e48d2bcf9acac9cfc18cc35147&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.332529419Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/d, Name: checkpoints/576/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/576/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=890898fc828ec49952c96c2b22f0b20436c82673f30ecfa00d2b4f5e8d122650cf21d241c02d1087d4cd4087c96e8ca31399942476d046d38b82ca2a09bafda99a0b8f514ab0b3e208bf375f705c4738e0d9fa7514b3ecd2d1fcb0b6bb55214b7e4821af07146ee82035087d5a57333d2dee20a59933fcb6f7eb93419835837e3b935c926a110763f193ab81f1e762135a4b0b8f3bc37a76c0bb77b6424446720a9a3912ed37de6f3ae2368c324e49bc26c0940a6180679c4436a6d2f53cfe6a86fa4fed2cfa0914e8f85e66cf464129fb574adc536a55ca4431149da062509b1b34f5dd0d0b416ce3804988f89139f22d69c4ff055a3cd441858d1a62e9bc97&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.332583709Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0, Name: checkpoints/5184/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5184/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=85fafe39648e858f79cd832c52b24f973a90d18f39f9548f765b705a0e0037a9fbd7543ac7d61382248a732db74485e59eeeefa731bacb41e7b39fb5307cc6cb4102430e89894d82192f4a9d554d4fad44a096ffcf47f73da5564381a8017c837418f4202e66cca2ea6ba495ec2bb4915d3cb41f25999a87e2d661ce87414125bd5903f9a937c7e80ee17770f9e2e14e83e3004b06efaa0e6c278872b7b3112ec1bff022b2a943ce102b204a27b299e86666ce5eb384f00ce5ad1e2208f1ca42597fd3c4518281f256a83ddaf22af19554f60ba3f1b6d0eda72a30d42b9f6dfcfee7317619ba76f6e9e689b85dc11778e28bada5d843fec9bcb2ff054a4e9c54&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331894457Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0, Name: checkpoints/2112/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2112/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=256dc350eb206eebe417c0301f1a6e1dac2f797392208892b4d219daf7a851276d4e1cbe8bcef0908e7c1f9970e3e8c03b17c2f8d25dba951cc7d1df03ff8c0931916dc1952492241ac073dbb172952e75d3cb7d73dbabbc5f2a9a3b0bf6de9d1ad2b59fe1d5127a8e1c4629927432e42c4271b7ac680dc1213bf16a13cb1c938ce18e38fc2d98068a6faeb9f458019d7afbedf09db08a16e5526317b58e27f03ca83f00670d102a0d41bc0fb0cf6296ebb731ca34daa5d3e5fac8b301e51cecf0c21e7ddddd2bcbb60f2a89f584201343c8f5ea3391a78e7f4cda02df0e484a0c8094ee0b163ebfa13afb3b5194aee96fec0233aea3ec8352bcb78cd891cb6b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.33276125Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d, Name: checkpoints/2176/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2176/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2ea8003e206e34cd5d8688da5c87211983b36edef29685dc8319725449fcb6886b473b6476d6a8dfc642d1f2bc6625d9e320cc70da16990cfe5747e8db07f963c00d8fc4cce882a9582f28f4b95b419f0b7fe7bc551f8a3c162cf341cbe890d917fa1e593361640a7585cd4d76ca34562af5031ed06a1c6594baefeab422bf052d2e34b6fecd0f0e76e9e563baae37af68033674bc19d332a5b89f824ce079667097611a12518b7fb014b50399537c182ccb870bfe2a397739f6131d2ef57c89e7f56b52b78ece029e168954f9e37a0ca7838a25c966223a0f7acbeab1fec458abcaaf832887aa5c6e872641e068319fa9037a365a03ab1f403f32536b540481&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331720614Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/array_metadatas, Name: checkpoints/2112/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2112/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=97757db709b9e9d9e3c31b9486c0ccf019dc10b5a1ee6316bf7793acba8e3d053e114dc3b4dfe089d7d4fea5963f36f982779a173538258bc4df3764f2b82376b1b2c1ce939fbf989df23b8d42cfd67b406b5d53d475e8a31c64abdd576b2fed8a374846e7d6099592010c680ad0f50643e3f18201e28268071712be7a8659914a4ce4857b5af13cd98aef623309b033aab0c6ce40b55eb603cae5ce514a1e39c170aee84a997d72e06a6181bdf922566690292f28826fe2c000f14ce4a4fe024cabd1b574f21ee96743ea4a1167e4ab60886536a579b2e0e6e82e30819e877573f56e40599ad4797a1113e0b9c818d040f40b50ce55cf9d16c42d86e49fc938&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331717797Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default, Name: checkpoints/3392/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3392/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0aee75e9d56a13983cd917933e556762878d6aa1216b903ce6b17eac84433aeef6a6ea929081ff2848d45cd17dde6a24e907735ee959717aca4a477c47a40c469e045ab7b4016daa62cddc82700d229cba97d0d02f0ed17d16acc1b7972ee067c788400922f7d128ea83e470370cff1680d5ea358f0f0004f3e50a2cfdb2606eed3780301aaa26b96de6549a67d9b6b5f98200a68965633ffcdf61bf911bd8ff21d2801c7918eceb6d37f7aa8e5a2b8a008b251ae0f3879c3ff7f8df2f96162a41f65dc74113794fca206799b9b8dfaac5b00e943d990ca5d4b9dca8e634156f982eec13675e6c2a00cff35f1c589e7cf50bb73675ca3491522555318fe5933c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.333010121Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d, Name: checkpoints/320/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/320/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=74a5a048b79b41faf470b322821e8a7c2c9a9b7e774bb50eb549647432ec03b398d7428107a1782c8242cc0d2d52cb57c8ddb8b75a5678600f34c9246edeecd2bfbafead20e05953477308c23c75d3304fdd0ba41aec8240ab08542ec5f0400814129fc76919012f821410b7b5449b9bac26da4390f32c1758cc01eede599d3d9110cf99254f090a046771643fe8c93e03e530a8ee090050eee5c469bf9b745b5913d6e1b6177b7ca095d2ef791082c4fffd50603e683890c41ff34209f43a34d76b68b37f8645531e3a4f0fe99469b5bad4c6bc42476b98593f53da10c69c4208a7e60e3f129c9be7d22188b9831a389763a2081d922a6f06704348470fa498&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.333050376Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default, Name: checkpoints/5760/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5760/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=48c9faea056648f3bbf1f3f1f0b320585f18cc56d001443c987fcdc1a13fd940d75a44b451ce9ee779fdf9cb9ad5c1e4e8ee750a91036887aba358362154d0021ba16f168c9f302d3d6705a9b748841c350b4ccd42ac1c411b2c46b6edcb7f4373333b62692526ccff6bb404f6ed8a5e085c80104a4e4b151caac48e1baf4b22e79849a1973b5303c892d3c93ebc78e8f5507dd42757bb613f011efee59758d93651114770ebda57819c62ccd06090a9f966e39b0f6a53925356448fd6abebd5499e17287f9a4171858f3c0b4cbc41ff243ab03a21b8e5bfff1bffe839436ae3c7e58301ef2caa9f38efe9149c91c7b8d315f360c571952b0810535fe98b087f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331740257Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/array_metadatas, Name: checkpoints/4480/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4480/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a7b4caadcd2c1b77fc0f06549cdc76287ccf62381a791ff4e86a9a9e081384265031eb510435a279fc61a226146ce440cf9445dbcb34335ec89537238a7e34fd315959a38775569383954672954fe62c10f5ea07e56e6ade29ce5a3f21d94e8b658b320d6e715c2a6862a728dbedf2cce0633ef7e9f7eece808e2f04bc6cfe32465a638fc9bc98d34e2d6ca4a7f3e3b63b334e080256aa1bce541704cfbdf199e75442256d61e2f8ce3709714f63ce112a5e2c10c30de6de79deddea276f253ae8acb011a4918db09420405325abe97a1bc0363e3246481adfb133c3efc8e6cfae5c81abbbc7d8218508e8e51eb422dd8b4058ed5e811c692fd9022a2c08bc66&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331797939Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d, Name: checkpoints/3712/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3712/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b4f5db4450dfdbdda1807b8137127b568ed5a1f1e4155c4644f03a0ae49654b926fd989fc0510c005b5982572fdd68f553ebf55887d2c725eb68f3bc580566fcc11c93382cbf75fa735688e458219e1b36c7e3d57e677c7f8812b7c0672ea992bf776944527efd57d10344a681c7235e3f74778ceaec8ba2ada0908f7e60b78eadf3dc5f6a1af6b80a3323c67c7d7e56159830ee5cb135b807321eb05536a69623818d885a441477727dd717cfcf8ea96472ca7538ba7f5f90540e691ec7e628435bd591f55f60aa21c7aba776b2dab729e3870a5eda3274eab2c2aac39f8da8d0334df5ca7f39788c09a3514aa4d5cc8794db9e0c5d9e21095b24d7feb91e7d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331783001Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0, Name: checkpoints/4864/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4864/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=bd44db02f7f2332f0eef1a7bcbf034acbaa879be77d93a53e4984447056eb7f1bdf10ebd26996e4d5a03c03258ba304bb87af6bc92dc5a747b1390c7f19f31a03659a59b6f78f9773018a1de4780467bba7717d9de39b6a9cee3b16adc77aa16004e730c6ddc78396170221db831916e2232079edf1b29ebca7ee295a44dcc59e86298ad862226eab7292699809033bf0559f2c7be0ac2694ecfb2a0cd456a7b53e95714871a2f4db1686689174edca739fa8caa0deb3ecb1e18532acc6c134b8d3741c9bb1bc93a4a0feceaaae9240ecfbffeec4021e41a86b228c62cf84f2f614a8bf4437046702dc3181aca646651402b5fe28d8f297ff3ca436b92f19fa4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331813437Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/d, Name: checkpoints/5952/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5952/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=24943a0e08e3c3d5fc2a10b700514822b2b23aa014854caa947b5d36a3b899a5b5bab06d38b35b439f983216f3ac66faf8e6d4411d9e7e158ea9766ae510dfd4b566364e239513935a7da557ed7f3ea09f284d71c569163555c159a770c327ba3dc3eeefbab91e8f79627b0a331080e45bc6c9fb22105a2513904081bb66f3d4f10f40286915008079c0c98a3e0d833f2e728bb129fba0e860b89202e911fbeb5cd3acfc5040677f97db71b6ae332e5dd3348113a2eace7aed1ea451d7284aec48affd1c34309b5ff55dca541df6d71dbc613fb3872afe5e751f31beecc2a2f11bf9a75afa0a03008315a8d1cdf7247da79fa96ebd5849c744b783f61c0688dc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331680481Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0, Name: checkpoints/3520/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3520/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8058ebf85e48bdb133ff9a2df94478702b1185d9acd4abc0936d85b2d6381686e90da15806409594370734f48343b12a0e329f68f6b6154afbe87ec449e389f938ad0f7f2650551049a44d765508785469bb2c9f2a7bac03a495463f294edd8a2b44063240ee80b6b5da1ce41b5113439e7b4d52e9985085c884cd595ebb0aca59b3a83094d4e51acbe7bea91ffef5401535960e405d0da46cf289a50db4ae6c2122b73716a836e63c6dab1f5d31e713c7592fca21115cc28bdb68c54fd4a827427d06f733de0c1b1b9f125bb934857dd53d5ab2b193452f6607654b12c2baf821a976511722ce02980f37ecae498997c6b5812ca47eea3562df954092851468&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331806604Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/array_metadatas, Name: checkpoints/4032/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4032/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=840aa6bebe71300307e73dc768147b6bfd73037071c2ce081d02b9f780e243ac447d49408288391d27861ee2dbc85e675754fd87a81dec63fb1aaf223decc98a6e7a2846bd2d26d41471dbd16cf99dc0b3f03f6cc9112055a991065b4ca276e04f46591ca89eb853265384156bf5f6aa4518b7a8c70b61cff9f59c8b9e622bfdf7b4d10716d7efc5cc1457a68838c132e40d1037d64bb2b098f35441d7887b5f401d6b547f31361280e93944224432d63495d180ab5fcd40b4f56c49c4be3bdbb5045cb47e553c4179c341df0867399bdf377390d0433fddf5af65cfc1f0e19130c1605bf1504764fa3fe3cfc238945d8f4bdaf0b92a413657d33a4ada58c8c7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331831996Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default, Name: checkpoints/576/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/576/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b4d6684a68f81a7e3a180c35df86262cc8cdda8b54e802c2fcae1aad0f1a96cb0820ea1a94c059235aaf20f0d2ad4571b34f5ae145a74c3bf2400723b353ca468e14a4b8207def027a5318ec1126aee01b5eb24470c5319beeac5840e7b9489c63f213fa1ce994a029e29d040a463f6a9ea2a43eaf04f8a429137a5a6295fdab88f642170cee188dd28538d338e5dc393311f632c9820d4c77165d3bf5f31579caa704dbd628bc57c41715a29c99e2a201d1b95f5d08e814d7d7ae1afbbeb9ad9c9687e81ba8c7fdb7848e162393a407681e6b483c6c3fa4df78887a8b9e1b3bf19480a59d763527a22688e6cfc4da0205ac077bfc031ac0f603ed13d57332e3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331815185Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0, Name: checkpoints/2560/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2560/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=56707fad2c9a44bd7fd1af2bf358b6f8e12c56976fb8ef1b788eda41249ae8dc87b0da9ef31d02ea8650e5a8182bc6fb852574d6252ab9343fd1a8eca88b279f53ee3782b84f4ff93891979a13307ba9974544387c42784d58262d8dc99561a7da70552c03371547262f2277e40fb8852b8ff682792aee85f3fd73dbff9fd2f40d50f7f649d03a61fe7eac96128a8afe045d58f382457a9b35468d48644a1a2e69f69c6e057a49156bc7dd6330815333b5e0e2b2926738f86699f31d0d315cebae1ce6eb6090af18adbff8816df8018deaf265d5eb1eb7399a27b5be5de0cc86d90d3e653c3135bd1b0b40017d3282b41fdaad8ed6978a30fbeb2f38f95f9a55&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331836659Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/array_metadatas, Name: checkpoints/3648/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3648/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=320556db64a10c606650af39217fe34036bc794a011c48f14357cf20d1735149bb3726ffcddd92a9fbf2fa952767ad128310078947ae203b2edbb6960e6a04a65990c64a75b10d0889f1b45771212d9cbeeead8df6cfab63758c6343a1afc02bd139c983c5418d6c0fe88c0770adf892f0bf9b1ae752ac8de3ce567ec293399e7a96d1e0664fc75a86dd5ec4e4924324309637c4876dfed2da1e5cc8d8acccdd278356c1a5631a799ef087402b1a213cab60ba649a712d24e6e2f384ba0f0fe2f85094f8178c50832b4aeb96096322db831d7cfea18140770d75c2345461fbe3f0f59c8da4c97d022e14f8fe6e67208ee1d86d49997d2cae1b73c2eb8e01cff3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331850145Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default, Name: checkpoints/4160/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4160/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c234802c3567c877720c345adea419aed7bb5dea8d47031709c39b056c106416284cc90880620804181c2008186cef0662d29c231a211812411988344147e5935b5eff9ebe6e3e3ed502f92e49219c27056e018099c0fb74e74666682fc6eac2da9175a6fcab7257f5d1630fa7c43b5bfbf5240288bd0f885665191ee6a14d1977fef3e812316f7a465f3d388c12ab453becd5ab5d59316f1a47a17f83fa7e9c748e16d30909c6588169baa81a9e21b0bab024c2fbd73cb60fbacf90eb5032c04b679da92fce1d539c31b949b58f65979478ccd751ea09a6cdab4f80a0af9349bb4564a3a643823b55fd875f3e925b02a34e21dac6afffb4665fe6b761080a88&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331688399Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448, Name: checkpoints/448, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/448?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=18739c59906efb57353876647ce4dba7024937398584492b8060451f9b5a2f571c178241c445a6f6e4a7ea636514ffdb5fa5028796578d5a2be6c212666d66922a18f9ffbdb27a489704da0a19b0ef67d58c604a5e683e290bc7b0ad88970aac73e332c6d8660616b0ff57dca55259b27e0435ecbb5aecc771bc4867d86338ca46aff52dad59464ab9d91b7b9b06def89d6e8506c4b43ca940150b34cddda56dabf1f877ed772eed2037b5bbd1f5c6b5a4e6e5387b53cebe60df2a5fd524b50da244a4efc60013ab09db17bd7bcb586d7e5b7578256c90fb401211b2eeebfe4bb8b1c4210099613f4f924bdd75228c8aebf01296c31d7066edb07d20cdf5f966&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331855299Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/array_metadatas, Name: checkpoints/5440/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5440/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=54da16a699b92e6cc8ec72fa4e1b988024ee6fbda8056381511880d257a2e342619fc644829ee0d96879997c021b482c9a36849abcc16acb2725800950f841af458f9b43418edf886eed7c0c7c7be2717ef1e00cfb6e47f61e5c43dd6b0f3c36c17affabd61d5e686fd2314645c903cb37768393b2c414a5a38fce08f51483dc22b52195347b4698f52261a890ad151b4592f70f4b77e281083af6de90216a4f53f256dfd28a7255b5dc5b356991991854032628d026331c0ecd58bcc91537afdc919c92dac6939ea2dcccb9782d6235edde0101a36eda3ab781d5e3b9ce5a452e5f0c2b4585da3aa5f541109306df7b64efb546483321d28b24636ae379e574&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331824266Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/array_metadatas, Name: checkpoints/2304/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2304/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a0ab3ad6ed3996db32820bb9aa3cfe1eafe95f5dd36432c65c1a0251f3f5d83d32d4e20c9e411ccb834f49fc8b66023eadf7773944ff44aa65e4227412fb485317daac2edfa91635c419016036bcdb1874d924cbf78885bca4aa38bceca932517d248642de7d2cfab6714bafe8460d44c24c4cd49b58c610b9afafb754e15afb26bc19f28a0a7ec6c4b151d1154ff7658e817f4586046eb48c1c995bdd0ffa5f769e6a0c7b7c4e21f197cd73d88bf04ed38bf2be70f9e463f37516e262613f3272c0209e3e027cfcef69fd6615b935ecd258b5ea435a357f406fef6839f513e60dde26831ee086d0da820a129cefa9c5c7ee6d9c3d10ce1cdea4459c543b54f1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.3318994Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/d, Name: checkpoints/3200/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3200/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=217de988f4010436f94d784052538d3194a39e33dc36902e3a35a9794d47c5a52f7aa942212e686291f291f3e119794fc642168d47566db5fa7e7f237fc319819b08ecce6b8fc1f06d834639cc57f16f0165486fd06a817c2bce77ce503c22af0317b18590cd3b40044ce5ceff5a47ca7dbeb9b23adb5d50cac4a906657efca1fa487e25dbdce80a88ca26ee49556f1c965dd0a4180f10aea25e1b15eae8b57730f2a6632fb1170d48c2662371d63f1191ad407f3d1c27540cb57d35b810f3b429ae660c4d2dd8009630b333f2612fd3a65714711db1bbdf11856bebd91a9aa4386bf4d098ca8f4c0bc1ba0dc270a3e1584e773ac4a88befa5b08a68a3882925&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331701987Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0, Name: checkpoints/1792/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1792/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7a3efd46e4a4204256979df7049c75b642164a380a2e3109d3d457229bf0271439c2fe78fa8327c1568fd5b5efcf75687a082a9f754c5b623aae92f7471a7a35abb6d3a37a10784fd6df4f3ca3d74605abaa14904a8b2a152d67e2abc47b3768273f952f84be2e3f7a22103bae3663ee516087a25de785c26b2b02e961aede859db7a64f679bd20c5f69800e503eeaefbdef46dcd43615ddc8c871a308e1f7f38d534bd3a29210485b2e229278724aab5fc3323518d96fd4ccf1a3524c7807a58745b809e878f503055a6e3e4b399abbd0d54390f3ce1282349e9798ffc4fc407be2445a6c213bc9a16ec5ec6c3aab03b23cabd10802579f12bc592c195ff4ec&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331956622Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default, Name: checkpoints/5056/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5056/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=49f35915e90d4162b52ddc1775f46837299922b7b718bca14ddf1c95fd6f6e34868a8dc006ee550b64fa385e8258e4977afcac67b4666d6a40f9860805c937e5f1e789caeace347493739a28b4fe9c8a7bfbb05eefea51badc66498ebb4c8e3ead63a9e51ff626d9f469dfbe91675588f56b90477bd9cda6e25a44b65110ab447ceb19054aaaf4fe9e4f17b426286a6dbe0a72fe793a1c8f892c2b42aa378e1fb978195364de3d714c3ea635a45800f63c075589518a6bfc01433d2f426448c78bbaef62ab1e3b37c1bfa737c85746b0f411d025d2cb9d52ec02f8ca28dd224960f3f252b81ea39fb8a3312f2f35287363ba0da335d283608e7dfe12a9879a13&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.332013793Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d, Name: checkpoints/192/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/192/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1d6d59ac3d3fda7ed57627bd6d3af09383a1af97b95702c03db11c3d07c088e0cbf986d68cfdfe4aeac055127356a87d10eb5958f357c33894871ec50bbe5f15bdc017ece4073976efd3d4a63883bf673a43c4386b62e49c032640c1ff58cd1ad95625f0a9d18307d6f502b7a7c4af9c0829daa7c00abe5d059b35ca12fe28392aed5e9c6e039719d410865d6f26d6b1d655724e35d98946fbb15c4a0c9e06af877afd524cbbaaa0bf7a1c8a9a8b22099ba0ac4eb8bc5c18bcbc031f7af26b0c048f303d4fcb0a3710b20551fdd27d4bdd81a8d9d31f2e1a25586e71dfacf693bb718b2b7ff74247603876676bf2df03817c091dc2116a760ea2d4415168ca02&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331950928Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0, Name: checkpoints/4736/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4736/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=77dfbd611210757af6642f32e7c46575082c6ec6aa48a4f94be301cdc9a90b9f4c77987d0b12f3c3ea122dd357bc178f0324584c9e80d18440cc26c33a3996af52402f63acc35159b2b859411ab1ac3556ed6f297221b7beab925f7ee46715ae8d023371f930001e149de19f6d69a605c244906d1539b922981e4d45dc575036f6851755668ab1e41d94fb08882977c81fbd94d68950240f2c4492028d516d1d0d00221f867ed854376abfe6ad5e1dab06d743ccb3dae29b1ab98da64043965e61ba6a042f5be53bea315ba7b87561cff1eacda78dbcc0a467d7580f59d44e632223ed17621feda99c8dddaf1f9781d20cb2666eba1cdcd79be114d89dd180d4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.33193547Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/d, Name: checkpoints/4224/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4224/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=2ada1746b4dc9ef962a2c8f59b52a57b12bf3fcdfa2b17b98a834e11318889e173771ac4724481d6f7da4484ff4e6a1ca290b55acf6f9a968f3820ca9e02899599832ed0e2543e70fa45d8cc8f8b8f764b76323953446c6609849c4beb86e1867e858311527262ef2a220f780ba80c2319659b2d6c806b81bf99a6e9cb4c40ef49feae2e2005ded8eef9971164d14e669dc33135200f13fd4a81ccdbe7ff98b0673a99df9a1e2affdf7b522c9e7d9b6ebf7b24c8c77c149b30979af80b58e190fdac85fd2fef174421860295a3a8be6065894f11d468a71d9f1865bfe3811b272dae9b51aa763c1ccee6c0084e3d34e23aea524643492e1a171a2878b4eedf50&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.332066178Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d, Name: checkpoints/2944/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2944/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5ed526c7123108bf5a8c7be88c8c3d7fe59de724db8c515a9a4dac1eb2a3d7144bebc44d944026c7fbbc99d9e4cc4db1c49517e5cb258f07b51abb498715b1d29a0967d3249e665cdf0b3bd8f0897c9c25e556ace3a7832bcb257e81b28ca3df6650e0e18912c1580c0818e49ecfe844950113764738d274dd73283a5c3b5c9bd2540efbc6bc072b46395e78b304b34a58eea7ca8bdd23ddf29465a93eb4ff35531eb192cdd43663ffe2e16a8ac6dc4b34a73d153c4120dec6279a508e2db2f99631845bbf6f4b696b73fb2ac26951363fc7415be4fda4ca1d19b29a9a344c83ff8b614705e237b7ebf1c23a468e47f419f5309a79c9848ae7c2268a7e8461b9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331966604Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/d, Name: checkpoints/5312/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5312/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6c66235f15b2d200b41d971ddf7ad9e9841138b534b65315485e8b7b0d274f61d083f219319e25b7b9b36319509177334c433a704b4d172089d0621fa349e078730c3adb8fb055b10b7303c23e7af2eb7e33b36baf8f0d66cd8c4ecf6a626e87a003311d0c590245880a2220113aa6a5e78559fe615417dabde7a5d778bfba7bd5c08c8c92ec6a3ee93ff60ac8d83d1771c125f5cc35c2179049f2efe6a9da257208fa9334df318907500bf3ed39ad8ec5fe2feab7fbe395d4aa4f79d2589ba561396bebdc4198f78de7be7f90c43bc589d18da1f797a5844f7d9a8baaac3fa9a2e0005a4d688c5d2297fa7cf95a52617c26c3f0d4c41e47aba9ad43451428aa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.332066361Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default, Name: checkpoints/5696/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5696/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=184e3a1f0213578de7b2286283eb6e6db561dbaf001103680436b1c7dddba7e562a2a2a87ddc102fe9ea5172d78c2e55ba751370a56606fafdeea18dc288fd1c6d4efafa4e233d4bd8620a2fe641580a7fbf7d69271965496b6b214b68e9b6b7c11f39e591158fa76ae75f961cf71ba25af7102f597239bce1558b7791e10294bd54a90a471841f35d0a30fa4dccf7b095d79c733cb50ddde590db8d3d754e11dedc8e95c7a498eb34793205d5ea78f007ce72a2e1ccecf011b48ff897384863995e16cd5b3394e0f564c19d240433bd8201d67e48b751ee6e41ffb627322c224c375a23b1141162af3e38d4e64b2cbadceae99ca970fc1c1384cf5618e75062&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.331705599Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/d, Name: checkpoints/1920/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1920/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=28bb03a0a1ffe71ff7edf02b0222f3654f0365950d2b94f6ce260d084f6460359d212a5d51d1975996f6fd3dccc6a3bd16c0d6e95fafd7c698754e1c561cddf1266df952ce60ad663087310898fc83ac31a005199b0fb690b6b9678416b7d9a479a5c7d901ede815ef372cc2af419c95ff04b06551ed8ddaf8026ba0ece42065fbcee2532ca1541ce7113e74c202a95e9751345d2e88fa99613e2ec80a1110b3e6424096e8408cfb7ceac237a410aba17f802171159721773c26e6f7adc14af3d641199613b5eb11c9e2517a14a4a362a427514e32ea934952916008b56fe91094cf2106a1bcad4663a2cfd095ea115c0c7ee3fbd40c992929160e70f448436d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.334246374Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728, Name: checkpoints/1728, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1728?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=1e851dad19a32762bd482f7f8f22e094b5b73b8a72b3e22d18038cdf25ca3ceaadbd8e67b3f6983e46c140663167492bc3d65efaed523b0a1509e671f175e3688449792c54b9eefe3b3805d861ebc4387b2ebb7882fbf8b353e85b46a841d5e5ce41baf7c6b51cf5f22d43cd25e76df803a670a01d33540d74747d13a317920e412cd9e6275d85fb63acb6fa9b7712c9f55663e2f1e4dfbc012e7a31198d37e08ecacc3539a68441b8771a0c394e12ef4057732e4bab5e4944c6aa83a6eae924d1a7dace76b2c62b8e42aa3e1bbb6c03fd34d8a83b94af201a9d32a5af7fe755ca20613009b08b58a9b68dc5a63ac81688dc6aea5d28a2b357f2fb761766b82c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.334328972Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/array_metadatas, Name: checkpoints/3520/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3520/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4ea213980db6236991b444589cd589f6f98f85da20639713ea131d92aab1e563c2b7dd0d496bee93bf5e6ad7e259214a58df234938500c0c70f56dd4f2fade98a5aeef23f0cecb4c122e76f1ae8c067f9c6118f73e5c9d8dd5ffedf8da5971e6cb2f1e7e4abe1531af1e8fa4e56e6e45d370a52c677c8c4f1459f285e10f40a8f3f584f80adca552028e3306b58e198862b79dcc65898b3df6b6d244994f7e9967c9389e7540f6cb21b8770bedbdfc4cbdff0111eb84b1cb0d48bf9b48564f4edf18a17af9d8c18e70593c5e22987ad92c5e13a864d2e2935898a74457fed825e33cbb1cf556d86e451e22bb70ef3335d0f8079ad904770b015bd8f7ecb74c0a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.334417326Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160, Name: checkpoints/4160, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4160?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8b9dcec27fbfd21d4ae0a14797eb7ae444de2f4c01dd3f1aebf2b80f78ad79f76a073cc9a8be2b7c9504b802372a3e1dd13b791cae07707612847456915bc805a6073cb6f7710aa04b38343d591eb19b6f9c73504af47ef2727380f8a0fa6a417333955aeab1811a5b0ca24c8731ccb89c7b2a5ecc42e4e0b22ceedc1398f35021d5d3fce523de8c68fb6f79b9cc0d1602ced0ff5489db61da190cc69866fa2cbc08657537a25d60de8f208d71dd99ca96e6a1962cd3b304a5fd62efe244e0fae137f3d6746b944ba12339ac69e1707abd9e0fe5cd5839baf53789d036ff2a2999988fe6f8a10230889460d57e4fc459efc1d758932e33153bef139717f6d6da&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.334603779Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0, Name: checkpoints/5312/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5312/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8d98b5ffb6cfcc4b7f8d29d7c97489255ea7b371d03c36c32aaac55599cfc1579787e678e65f1b755952369d323941a8c27afb03873afe25585362106edc338de427e76b914bfdcf624e4e8bac938486cabaafc317433e9afca8e578de92259eb0865def28d3fb321c426fe4de7efe591b73faebb3018e6e505ba8740d9954a7daf469e3b18e409d41ad01f2667b59c88fb317f819574dddb09687fe79ad338b5e09829960dd2b142b423e0dbe13d879df63bebbafa1311d4607fba5b811ca1fad3dd66f7a607040707be999609f8c6a79c8413ab5ae1ab4b8f14dcf1228f29a20fad6d86604a3785b0015b4fee789cb7c82620ba7d51f3ffa4da9270a5b0a6d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.334624939Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/array_metadatas, Name: checkpoints/2752/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2752/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=779d0553ef691000569528d1b092c5afdba3a9a4651a238032d15e839643325c74cb572a60e7ad860f8f1cc73afb355b62135066adf547199e342d85484ffc5bcad3f288b72d2a73c1873a0202b955864da97d0a0254774d72839bfe75bf18dfc1bfad8361c00763882595cac4a9f9a220216b8c781805fb1258a0fc03940e7763523fac9f2fa1e3272b17289b2fd53e82527220191f150c029e0e35d06b3270c0e5fd12f59634c2527c339e72730febf3e05f615326e5218751ca92914416cde001a73f71f534ca3e7c7c6653482b0de5bc74156e55a98e222e87d315c2aca07c980ed963224dbd271fd81f37f156a8bb8a0efa439d6be62672a8677eea414a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.335434126Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/d, Name: checkpoints/4416/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4416/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=25b8cf5f85b2414e4cc2ecf0b6c4e087ca051465bd93c635f72dbf2ecd8a006f8eea4d21667122d3b1be1ee1bcce2b7868be766d578e3b610f8236935edb1b5328320b718c3d0989afc97df22008e2a38f21a0ff8fda5659155539311e96e3e6705b6e47e15091445a574583c599cd089d60978fd9524bc2f79da4a40a34bcbc7ee31ac37b816413a0128e4f8c5ca80d5b84f7ffceb3b4349b7670dd228461195d7eca8f1f8158ca03cc4d190b005c8e9f040974313814ada3ce234657d629c417e60bd624139feea4dc9d77fb8fd2e3888edc932ac8025c1f7d2797509219d80537978a319aab13a5eb6fc57b00bd8225b4d916dde6f1ac6bb4209813bc6134&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.335598795Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0, Name: checkpoints/5056/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5056/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=95347907bb75ef77d00a958e9a3ee8b35cf6c2bf7ac230b989192db297c5056d658b367f49d4fcbad63897580b4638ff00d541a230cae05672708712222ef88d281a30371d59fed61e9507cb26ff23daf8014e016c8900eefceab090d49c5521e5a39799d0bb4dfff8d54587101c9b8825237e63e5d4ad5b11f2a99537c99bbd2f64ec1773ddc035ff1c454b0ea0d72173ccf7665a3b6bb42c3007746750f69d9b5b9229c2865dc7c7fcd8dda2be10c3e5ba7b98884aaebc4f85cbf0ea3f33bb96139b9751347375230c35480a6a2ac0752447ea6671e5655d491624473d61dedd62ddb5d07724f253ae73d939276f22390f72fbff8aa0156a65c0d75154832a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.336447846Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d, Name: checkpoints/3328/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3328/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=685fd0c07a211f6453541bbe7afd7449f019dbf5094cef261fc8b8d109b295d31849840837d5f7749e5a394b71e6a8ecc9766a3fa2207214c427b7c610c2bef39e3f17e6517af91963b9b16bf612095b7fdbe60bb3246f334e582ddcc369f331c3fe6f6302e90c08342553c86be1a5dceb909cf181e3c5944e079cf3787b4f857f0f0b61ecfd7d3ed4f21defc866c6c7bec86334cee31491b63fd58e89dcb1617f89a1b7fb6a74bf3645151faa0834b704466fff5620ef7cc6435c4c115cc5d7b79e9b4fb46988b15cbd7eae7e50213027aafd98efffb4bf22261711ba3c2add79b68695efe31fb86a3eccc589e3f98c7ddf41a19175036adb3093f82486d181&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.33657108Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568, Name: checkpoints/5568, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5568?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=253d59544d08e626fcebe29ec02c916902fd808d94fc5c6ea0cf787bc7389eff885fbfef8a217e9f77e88bf8246960cd7fff0ea8db9e02de940ff47fc224740a5caf579867c279f1c64c807edb046671a722cfae14d0ad029f59268a6b66b24a80a68d8661d7a83d73f01559f68997c557debf3e0a6593a1ce396d92dc85d4c93299c3ab069105d472c80f56aa7bd4f0f6130a1921e536d2701829703dd676e93f966ac8da3a3c563d0bfee005d26c2c3dd98547940589465caf408d67e2bea5a96c7e6913b9fb3c4a17c9db696d2747716ff89fff5713ebf66f14f6a071be7f0b40dc0a09b39e9792b2db0c0e0476f2aacc6d67a68afcd48e013da74626db63&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.336810161Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0, Name: checkpoints/4800/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4800/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0cccd4891a49bb284d607a37ef06c09cf4d343ec49cc109c7c2cbd3722a7a9be16bd41969c15bfa3776b4e146722b74367edd7729afa08989678f8644e8bb62152c710fb9cf9704fc591e6260d13ca66f7aa12a8dffcbf80e0a3d82a3002258fd0b4462f93ed57863061ac8131f1229234ee976b6b1a6b9c9270e535ff33863536d4f2d61b4583c28e15851d6eaa17d863bfb69eda00e332e280925bfa66d1b539fd417a6896b195af5ecf0dfb19b2bc36feb653ae931faa34db92d17fec7953a92b32f1d189b0995c4b9bc32eeb73239b85a0e6995db1fd66fa0b539c3e860605dccf161683e510e9ca371283d33fba39f6928514ba38a645d2b8febda8e1d0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.336858092Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/d, Name: checkpoints/3136/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3136/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=502f63d91b9d61a5aaee1a0c67997c5d8d93b7c336507cb38e1dcaad26a679e1044169e5987c8aa5765b61e3a298b1eccb012fedb52e694ec04f4f3012ed46e16cf8ee9f9d5e8048fd6dcf1df0f017856d9b7edcbec2b1b6fe33a1169f3ec33b4ab7f21931ff0143bddfaaa0b8f203dd034a78143918fa7c0f466f53e2cd02d5081adeaf13253ec8e83b3eb5d242d74b6684ee061b75b4c0a6276b34c051f4e2272458175652ce7ba733aed06c90cab0a07a03ea1d7220233b79db2d1b9f61595aec7b58f969a3f6b361c3cf94cdce04b9429dda4fb2286474ad50c6042b63d47667157222c482830cd19e86c33ed0491bffc11862cfab2dfae561782a6b7164&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.336956061Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/array_metadatas, Name: checkpoints/128/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/128/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0ac8b2521f7324649f56a2d8a970448a8a1d5bb563371f8056d414449a4186b4bb5f0eff3f87803766ee4b4777168c29793b0936035ae92e407cf83617c080b9444882889c1cffe8d8340506548a63afc6867d4c10068657e5c4e43c2bd8734498b313ebacffe721efc6585fefa6562286cf8bce05c2a437ce71a77b6ca4ad108b9c1204f6577ec93f52f6acb3c7673460a82fcf9c782302ad99c7e8f6c335376c164e233b51a58f269f091bf6685e8b816b77f22e569a72c61c9f12ede64f128158f0b5147dbef2f4bbc70944082a447b413efa6ae747c1f455d72838e37ca8073c886952df8a1a2867e25ac89760686f2d37cc7836e7e2a841caececfb50b4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.336462513Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d, Name: checkpoints/4288/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4288/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=247b96763ce724263ccded0c82cb14cb63e9717707d176784ff0997f279c70ea1a913ec1acafbc1abb07797b8ae4abe3bfa989521aab218b679a1f55b40f7a5784c2e0461d2c3c81d0f600cee5b45c9708a6c4404f164da8ce6a62fc4041f982f19f93f90a799b923c6934be630454e8de9d26cb713d98a3d0fb6d6a88e9e298b65ea19b83a008d76e26158465d3c78bac8663680fe7ba97576865c09633140373b23f9a8c0ff9ca99fa048b1392489f67248fd3ddfbffaddf9276192ccdbfdb51e21e88293859ce49a39c0faf75a30618291a973d5f08c02cad149231ee87adad39ac913dd6efcd59f214d9ede3f8b833fa983b9e2800497de25f60aad8d83f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.337659913Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/d, Name: checkpoints/0/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/0/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=27d532a568257b02cd63cfbe5df58e4f096ebcf20d71300605bae9753ed295e32b0789f396f8d1d24d92b48abe84ebea44b6788d0f35b05499f8de0905bb04670a380e2af954914c6be22aee00bb2ab7849f50f536e2c75980761bf15d684dbd1a9ded9ef2e2ebdc4996fb5e3d8f2fbc8a0b5f47635c3a6ea5b05cb0e766758046395f30412a4d910f044d986aea6b909ee8de52cb16ec14d4df101d4fcaef51ea8c54a814db86b9c045295e02a013b24babae27d3cec1ad7454e83eacca7e511ec9cc55612065e085ea42eac7f86c674009d5f3c0fb8ead7fa412f4c591291a70efd0d55de601bee9a29095c518c486c08ce351813dd1fdfe88610f21d402bf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.337891719Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152, Name: checkpoints/1152, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1152?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=93029a67ad5e911cd671c6770153ef0cc91b587e83b6b1f7192bd79cd5dd356872054ca0cab9b978dc04db0b3aa41eb2f87e935d2aaf78609c1087c4599e4ace4216ed5b35afd464badce052d0d9e81922af54b214c9ec87df54e87bb81954a53aef0232a823b7d26905ac7e61e3ea91746475e9eca737c5af7c5aeb907581b3009d64cda358fc12972aa7660b95639896fd546211ef0528fd6b3c4f91aaaad8d4a82d218c37acdde68aca0751ac66e3ee02ae679e95a411b33e7a651d5dbc8172aca79955b1a4ccfac23c621fa9582e01eaf8f250fa328a5a8d8a180b54ad696679fe4b798af5a7269960d8c154cb4917150a2aff2944914cf1090a5e2bf9e0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.339517065Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496, Name: checkpoints/2496, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2496?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a6be33abd389a34268d2aa9a731beffc557face4b8c1aedf2f9b909d7d46356ebe1447a1f04704f218e7d724b542ac3f61e80ef984edcee36ef5841cfd7ae17c6809b2756d8df98185178aefe9fe38ff229996e28f1bc096245804c5e3f6ad7da58fa4633290fa9c6320b66bc0e3766856d0cc09b55c44822ee900797888842155b6616ee9f23d1b70d9ed2958c599ed491954ccafe9567ff1f2c384ada05a042c7ef3fc47547e7ec0cd1e02be6f40e4c82e928e69358e9f60413635557da5da1a41e46a338328194f5b15170f6e976804885796aea2bfe75a668a337d6c8fb62628a860665ef0cfb61768173824371f657a0e5342f7b795cae3fee7f60aea4d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.340833756Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/array_metadatas, Name: checkpoints/6080/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6080/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a80ad10dd642e762c2f19df18b3cd4e384022aeea8190bacce41e5c11be8f2cd97cca9144ef9bf3a8294e3bc74a965bfac0004684b43c0fa442d10631e9cdf3e609ff5c31ec7077a3d5689f4c7f08503267f7094a34e944cc793ee074c3074d89e6d815410e1f2a604f1e69792ec5519b9a668ae87ad21c09836eb827d41f63de2ea3599bc19bf1aadf6094392bae1937aa087ef61c1d710c0260bccfabcd0ce22d7cf828ace6e136e3348388ee52b5dd2c8535934af4f89ced26af733c451424aa54a9434c0dfcddc1e05e5894d10d65243122b41ad1d3d23ede411d4ad9462376e1ec479847223baf1bc22152e51e734fa1800f57976e60c5a9d74b879dbc7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.340844013Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/d, Name: checkpoints/2240/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2240/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7bcdf221cc3e1761665acb1bc572523d5bf141166a81fc7ba882814e36ecc8ee8b94f544c1a8e91f397992823f3c216f902b8d8b6bbf1241ccdefc77623a858f739a5c9fd2c89e9a6789bfd975a773e2e0a98efb968aab8d83fa96165049d25024f65084e5a05709d8844fdd51f8aaa8e231d000cd3d1b0b49bab06aae62a9ba6b69a518df33887e9a14ffc602df4aea12f35615368d9ba0b441a465a59fb5934915143dc9f8532fea2fcb7acf186eb22bf574f30057ed6e014c6df9e7469e8ddf041c5db53e9e3afd2ecc529959e70cdac3f57c808f38e2d7acd74b5af16df2b8aafa4095f5b0d1025b22f487e2be8e2c8fbee6e1363a273e16c7814f6f53ed&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.341100354Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0, Name: checkpoints/2432/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2432/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=90f58d987e1fc1790f70f7a8720adb32922a46d5063f8d8aaf1d25321b5b69c531856a1cf750850fe4c4b44813809b7b9da095acd3646d90dcd60aa23fb7996dc6176d9e9f6dfb9939a85dd427222b1c8bf1ab82e4b124f36da0c65172d2a66881773cd394ab528d3866b675630a7e3e38729d093813a0e2e73fca9101750fdf31b5269d83c14691d6ffcda2a3aa0a8346caa798cc62003a7295ec6cff507a0be409f01f3f7f5dab6becca95cc91b4baa2299af086c3a9d0b4daac5315d847280e6df3dafab2fbe9d26028980bb135624740d0ee417f67aa80720e576df9e61b7ec4d680f2df1f55f650c8d97871c399b6c54df50f50bcab8aaa03eb9eb1b3f6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.341226162Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0, Name: checkpoints/4160/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4160/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7c4d19cde93ac20ec72015b894f8ef734c5a11dcced21e1602c220ac0fd921aae4c760a4e82e6cc7d362c170ff3634023e88efbacb93cb02c0c4d85b277382ac34f312ac4920823efc28bd5a61ec1f4553b025a652bb7241c8e0bc2920ae3d1c40afc0ead944c8ab082b94157e2e18eae6cf289c8a01a3a383cbf7a8fc9a2707d3ca56aa18dc0b49fa0a33cdd49d9233664c1bad5e75edd712fd4bf9f68c69edd4f582a26d43c9a6fb8ae979d0f26b24dc280d6ca9d65e454044552eda4eb2139195b3f1babeab84623cb2653ecfa38031d11a1a8006b8e884b533a91d68eb29502a7dee26eb5ea9b6054d80f981a89056f7d167644b0c4b390f4307778d3ddd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.341251041Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default, Name: checkpoints/2688/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2688/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9d0790efbc1f5b3f355a6968f8f36d39554827259b8889614ba5a2c8fe8fd6b5cd514f3a64e5974125c29e7eebc1b5ee0042bc91a8a815b92a6244b62f23afa53fc84c32655a0d8ffe66aaadb0e65bc782b5c97a3737f68324d7e1af6e5347b37a7c90f291b5efe83522869babaa10befdb3e39a8cf17b12013b801a64fcf6b2f703322bdcc2868247d5941eb5ddc266fa78df234707baf90b573f46182e4662a2e80569924f8fa40693b96195e26ad07abfc601ba79c0792302ba7cb3ef13eb14bab5adcc19d863ed7f2e34b7926883c5cf606dd651fb0a4ee9eca6a27d99de804aacb5d2d76ec03ce31eb0701645c4b2c050c9df9f3a82297b7cb026a1c0a4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.341234624Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/d, Name: checkpoints/4864/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4864/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6b3cb7eda719d201af09db9e06fb67309d60760ac607f37b7a3ba17856f1f3b4f8da5e5e2681e0848c562c9d3f298f0be74b8b4ac0d2338799541191366254e663685380ca38160b8a99c9a3d1f75cc00eeb31206d257e8c1eb9e5e95cf24844807de6aebe208da168a4d4723c65ab13ef7114d3cb9ce74afc77079f56113890cee833d2419f3a9f2b8af6ead03c9ac7af8bc8e7c2a3d7109be8bbe01f00e9579d6341355e66e556a4bd09e671f3e783e03283926d436fd97543ea33e8ae1d504b69e5458f5437547f70efb084d0c0cdb0c1c01045039bf9a9240254205f3aec1763f13b1bbcb078319dc0a275ac45d3a5fa5c80a5e196a7c2303b6898d4b164&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.341315529Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0, Name: checkpoints/704/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/704/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=0c849a04b78a29119e822b6a87e2c36e962309bb00403f873b37476f50a4ea9fe5f43ec1039bff87ae04f88b32f9b0a63959df5a00c1670e4e179b0ec71c6f7fb580820bbebbc136d59635938f07ad9b5286601e491969bef2eb2660ded544144ef4900a1d06b01711fd224da0420ce2d5444f5c1c9da14e71f8bb37195b3d4b097b4c6e0b3c4217b123ca8ac04c5dbdc96c6723f8e253cb3cc5d2385480396d19799074d10c56f624a9b1e7241a50135c713669686baf8a3193338bf9889302217e6a0cad1c73181fb95a688677d8f237fbca6257788faa414ff72933023ca17f1805c2fa55e3d795c0979472ad55bc146b4aa9b8820ccc05849aa7107e66e5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.341379343Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default, Name: checkpoints/768/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/768/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=8fba896ca1f0f5c720a8e187cf6f98f67d90d1634e3b1571e995c851cc3418c1b2fd30cc0df6cb7b864d3bf7340ce37897d98b9449cbd0974e67e08791a39dd6d6e668d18a863dda7367f819b1c047960016a8c3f2c0381361b1d8db0c89929276f11b80d4dbc120318a635cf61411e7c5e6f7e6fd1d3099d4c16572dcf6a3bb33a9ce44c1a52b04202671e347ac68e1f395ac9942ff75a767bb857861cfb3b188175e22fd59697289cefdf354095890c01dd4bfe2c5fa5b17f3288d6e70526c9fa7d68787a394ea932ae7f7044dfc104c48f12834c33f869708690c48cc15ee44c4ba5a0deadfe9044e9bcb899515b1cf49743f57998898bf776797b310f9bc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.341254437Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/d, Name: checkpoints/256/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/256/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=4a22ede53bfda16a5f41119274d9952c4517c05088ddab9a45ba40b5391f7ae4f1f392e8b730e24f07a3024241858a139fc9ca8f5c212b71e3af1aa0c210875ce0f064de897c66cdcf6c25200dec263109e24dc6fe8a8fa7ee479a881ee77f6a53cb28243a996be36dad924569f58509b3937be1f030d78a08eb98d45a45fa056cada6f323d199df3676a0786694fefb3736627968f8d051fc2cd525f6d51299ffbf3068fd64b426f0b7d2be1629b770be220a11bf35cce27971ec9e2bd478b7027513b53b0171e249fa68cf2ddc68a73f896c8b789b90cd0c462102226bd4fa615c31f31d0bac758e1837a9efa4e3218289dc1b76feecc1846bd3a3e6283883&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.341494433Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080, Name: checkpoints/6080, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6080?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a8014d8cdd3f329e0120b766e98cb7cf6a5b15c372d4586adc03c3daaea0980986a5ec675b3989f359059d1fe722cd87266e67d98e64d4376bcd8e0d2854681837aa5f6dd8b2258e3306f9764fd0c6edf69ac60740f67a2c385d179ac5a2d7cff4c83a069a75392c67858788a9528f61ae9cd4eafb7ba37ecf39cc8375f8c96552dd00ef6c01bbff65ad5ad6b5d991599a77f5c2e8db4b5bca8876f82b7eb336187f833bc6fafae3a7cc92135443fa5d80179d763d63c104be306580fdeed9f26563b86f0e7c11e343b1b0e8150db63cd563bfc747587f6f3f873629c8a4e16ef39985d50e5b56b4382eb8c67d21adfb137e4bc24169238c5d952ad53c818b02&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.341717348Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d, Name: checkpoints/4416/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4416/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=b90cc9a41b96e471717776209e02efc099e8d68c28e5b84bc50b98d574380f88db73b05f226d0cb6b919ca15c752d99d826ea4e6f230c0086cb3438799dff5fae9c2c3b1e877e0d1ffffa4f16da3f27eff905826f5d7883bb026b2675944ff15c3e67f38b2c3da1512c7de5fbd519c1732ba61178b47fd4ba79e42f57d06acb1221d4b7acd56dc3cc9712ccdb8d5fc4477951f4b490153836395399f528cb29129f5bf576b0aea6ee177990b24e2757775b80d971d6cdac20a2a7cf8ef8e7e1ae9775c12edf16066569def9cfebe0afd2a7f203d6fdb617ce7bf6f41b8775b46d636cbf434f985e8a0a2976233ee66f1159602e93ba6823d95888feb86ce23df&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.341954949Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d, Name: checkpoints/2368/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2368/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=3740f6977bb83e4b194a0ea9367d9b2333b9ec5a553e96caff9d0981fafe5e00b4f5104d3305bf97f24b88e07d6cf1f8f40c68f089393c9c72adf5734027740602f81da1b149ce87d8e3fbbddd6a82b579c5dec5bbb3fe020c0b3aedabcf466490297c4341c09db83aed14fe180036b864ae93c655b45d6c4de83079e0c36865a97de1227a8544403b12149df176c0c9a905fff3077310e6034712957732aefd106191a3112584b30f58a0fd0e2e9a871b56476753c02d5b4f4eae00bdd20174eca9f405bd378e6255322a08a85c2fc19ecad397814b2634f3fc00822236aa36a8db751f097385954fc6b82bce055018d8b116ea2a20c18419887f310c3bd2b2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.342177528Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/d, Name: checkpoints/5056/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5056/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=7629b2d864c71bac359cc77fbb9df72efac00b6f372c564aa6eea7f59e5aea836394b8ee23c069887413024031e17a1708b71436546708709ee967fe1433de461552ee3bd7ec8926fbc86039bf90578b420a82bc0f63ba10c5c3dc3a8910f5ca6d45d847677fa70b7f68900774dabdcb3244a209fb670deebd97068183e56b7cae57035adc903abae5d91241cf6d2fe192b33f6129adfd7afc9080a70c0ff38b146d9761384b399b79b3eab22107d57109cf4e7c08bb5f8c88211dbf3d219bfde279b83c745a0d5c0f48722df7aea42c2de1bf7429d53431abdfa24cf49a00bf82f91e2bc12eac976ffb01c020c92b35d7021a66dc22b17b310cba68f21ccc63&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.342197316Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736, Name: checkpoints/4736, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4736?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=ad82e2ae8bc6ea5133a3edbd7c57d87c4332fae8328537f29626f33b8af5804f2be4d07c4bfb78a276b566fe8bbf359b1a661e6a52457f6975f29690176f4b74813682671120cdb1a05866806096755d44ecd4bf39bf634a91a3c1ec67555406d231b906cdb0e3b0b7c747ee175414bb00d41e26c67929258c16c7ba31341c390c6ad713bb8ae7f707cf836052ad9c646540f2e1ca0ffd61d81f49b84dffbde113ba57698e31879222b7fb741a90081c40824fdc865ffa10d3564c8583821ca05e2b5b65b5129c58461b92f5a7ce0fec3954ccc40350fbf7f26d5efdd1a8cf96858f37f6e759373abd8e55cc099a1d3fb061f155f4352d3bb67d965ed7d57753&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.342275305Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default, Name: checkpoints/5248/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5248/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=11158c9fa6268073b9aad939bdb7de709aa7006ec4570de89d2536e50b096a48f03d179441f37f3d3fbac8fe1a64605a7e6b006bbd3678fa6505765268fa488bbafdb540b6acf7ce359f20ae549908070eacafbcd9be3d972e9f2c66577541969ce4efc5442d1284a69d16a4f883dad1cf14b7af198523e1f2b2e7c681ade88e3bf8181bf3aa84783262ec05c11b88164eb8639bc70edf51213975c7fa6cc38c45f4c3bfed0f5500cec62c9152a03e2659fbb588180567f034ce9dfb2e83a2df7e7c182a5a30ca5b8cf4722fdd1cb583682ec7e74179091974f31bd8678e52658cbd4ddc73ec1af7cd8ce2d49cb321a394c89a839b2ac5152fd81f8e45a0943f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.342188754Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0, Name: checkpoints/896/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/896/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=be4b00f4975a02867a99f81c68a3df3dcf2b36caf028916d041cddd3d44b9b042da2aaa27a90e3e8444925df35c06499d44edb2058e8191fff00297eee2612c76de9d1d36868dd5e26c2c1e11d94bb3d1ce91d05f5141e8b61c5f4ad0231f36c2552262e8d65c4d9f37f9b5eb794982894acba9d8f4e10ee54dfcbf2bf25c7389d5bd41c3a040ad25d323d5f5ff0fe52a01345bd309dc553788c688f7f3d6fa3aa25dc90477cfcbb697677522af3ae9dc4fdcf4c9c10da531aa569a0ed75eb3c0fc45cc74633ee948024faaa7fadb8677e0058e052a4a6f837628f0d82cce622ecfa6196f7a15ecce9fbdbc731dea4d1bd780b50d58b4369d12b3d2ee528ad61&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.342370689Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default, Name: checkpoints/4480/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4480/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=5ee1ebddac1fb5440813c426c76ff44764b6ac03c42b6b50159334cb10d9b50d171f4cad99245ad03fc956729faba75463fb5535f1b3acf1ed1a1c27cd4b56d7c4e1c77026a5711e3ec572d19e24ee9c03f63d3025e125b7cf71ebb932c115b1f191a018862eaecd277c81f9d82e8c1c383eecc0d7f419852c6a4f5a072dc598ac915a3b1381c49b9101d256e7069b78f4996f582a77f4f73bdeeadbb3cbfb1750a935929cab7f3b67727405bdf7d53d3e2c412f94a211b6feec2350d703d60934f8cd47c6935eae1b9cd76257c099eea247bb3141011fdbb5db742b17eac530fd11abe4f86006913f80a5bb2d20dd5fcc6be6a013027bdbbc7670c31bf74dda&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.3424235Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/d, Name: checkpoints/4544/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4544/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=a9e0c6a0fe66ee5eb3ad39bf73dd68b65b6efc14092e06241eea95b9527cca9dca243f43a448b3c9f53b1a42fd9a2c40dd1bd484297016a94d3255e73f757953cdf975ac9637fe59201ece887f386cf73d6e8b438cf716c30be9c4b55b45fbce4aa2cac52a4c5652933adb4387579c0eb1cb2168d45666c5e2790930ff4cb6c3bda8b8a2fc827446887b8a73905addf9f3462b65e5b2261e89d9de9336696e56c568f0fb002445e48ddda8a009886faa38551248894e034ee52abad1d326bbc9e12f4ffabaa3f415180d38d7a748c2d5648ab3439c44c22a20ccf577dffa91faa91c709c0b18f08242698ed3f13bbb643c5d061d502de06287745d1275a8f0a0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.342289832Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default, Name: checkpoints/2112/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2112/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=99895f8d3e449be070f6dc262cfba61ac2026f58e0e13c1bdc32795b053d63a0e760da4f187c67187cf735cc4e04519a09d4d06fd9d9d3e6c3e3a728851637b8c231766f9e17bf1b1fd490806af44994c644175b37cc2beb332049d796933c1ef0c3f8551d37371c1d6e124b4773d23369286a855ebf4b8dd1801e0e6b4d8a9276e46759a81ecd0450f1cfea7c9c79bcb829919286537dbe13983f930e616174f7e47cf90dc6f17551691c2c0d4f7fd9e138f9209f1cc3ca70d26eb151aad8ef96baaedbfd4d9798b000fda3277b79b53a237fa1327a3617e56ef2b633442f91fd77167430c8f5bc55191ac3982030e9cc04e56cf3bff51103713b4c72d93746&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.342512618Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d, Name: checkpoints/5056/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5056/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=79fbdf7ce87abc8d3631b920c2636ec3d36d560cd59713d0dd626efda487f16a5b3bf67edd60527cd1d4cce6d71499a0eb5e6f558b4c61cec0556f3fbded18c18cc65c447492fdfea39da383f7572e8c967fb058095641e0fb7f1ca6e1757e09c3ce03bee2abc0cfd6dd6c15407f57793a6253f86fd2537275430c4ad5f723d33ecccbe9ef39f960914fa9fe0f72294b2a84b29d8cc5856a4ebb45101ca3bc7be21c297031c39a87dbafe6661e7f51fa3679da87246710ec040d3738038818f9f63846a1d3efec75f493419c436375cc7e3d3f851dd4c7a0a3823133378daad896ae035590f1f26ac704098fa8e2dc8e4b92dcef659c7a4729999eda482f9b6f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.343526785Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0, Name: checkpoints/5440/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5440/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=59c3b8b57ee98911807dd9ec44755abd98ed56c6ea6f5f2ca2a211ad8f7f1281ebff6b250294556e1eb2a4a72653f5035b1d26c91c4a7698723dc9e8b828b5cad3f061aab9072ba311e6e19232e4fc10612023cab8ebc04869ac96d11eae17d32e60de9e375d03ca6fac283790b4ad22ae69dc1ddfe5595c45fb72711c6632d52c93d0da8db7442c7d0e212ddb38ce8542a8433f2ebf9b51a5e512c8ffb867f4ebc7f9eb2f5372d654e5d946d9127995f70640102cc9afbaa2027ed5893aa2ef612f539f2435f3ed332b0438db1bfeb4ebae10644df699c0dc7a1298e851914d0fc54d1d96b319cb07bdb7e112bc41a0c1fe916610e0bb0109e21e81f6eea81c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.345051191Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/d, Name: checkpoints/1600/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1600/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=9248c8cdd1a32c3a47040d051e122a3cb52883bfdc3f222839eebb0c09af52c0588d4c33ba257b8e93cc77ea5d7d4c44cc443db2aba9557968684e0d96adee0307523eb44eff8f123b00c5362099a1940c9cab2908f5edc4f6466c2481c180025a98a9c1c815e9bde7d9f619b993dfe9162840d8c1672d7451a465488da298b3c03367285591a2238149b6d425892efd200d800222e97e004eb492b6cf8eda8222fb3a171a039dce210887a69fb3049823ed6308cf1ae50d7ce40c499f1b5f2ea9ef4194041efd5906bf0d92c4ebac39d6d0491faee97af62158c22058a0689f2a12672f44a60788fdea89dffa1058be3f9931c14685f3d22cc85a9d95d27748&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.345417272Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/array_metadatas, Name: checkpoints/3840/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3840/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=c01c37f591d031428d2f710a96461fa97a9c7436d2aaf2dd624a35f21bd74816b8dec58c8a73b12def9dd1d095f7ca6a972266fc3d5e6e46051f2cf9bb81578107c63bff574f93d54e8f2e03978c607840a9bbf4c61ed0521cd5984effd983c42408a124bda5ad46598c37e94144c844f1134aa0a28ba4a2f8227502ea8f912d04f273b383c305aabaf5dabd7c2af8c1c2b1e56c7a3e210850b6a89299bde240381965b2e96622c353bbd77f4b0ea8f2c369be69bf553d003b06972b9f87d2832f475090f8e4169029e7e5025940d2fb97aa96eaa644df6fab71f5e96f25acb538356be6d4d046020d0ac0f932780f3dda20602cb8f7b34c6ebaafbf980d9a62&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.345428935Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/d, Name: checkpoints/4288/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4288/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185849Z&X-Goog-Expires=86399&X-Goog-Signature=6ac31db83cd29773f9a6edf7cbe8cdb3ea4f4aec3f34de79d09f97ebe032c3b686f6b179d88a072b8a373799d4d0d85b7517351432accb55b5fd30f1f3a8eb98e6a47f6e788ff260c7f7ac9745cdab199b582c5fa3a3051df12927b6afe9fb2f8d6ab26f1d8aad816c883b940e1b261cfb3ebc11bd14efd9c9c75fb6c116f126a6dd8cfd55625bd80735e8bd3709409314c95d616736edd1a13cee88c094ac24c29c5c1b331fbefb8a4f5f7e98aeefb122bbbd5834c8ea0db0d1ee0e94f82f3ab28693be23173bfd7e5b50e4186f321b2914812a65c7c5f27f49f2f9b7f9c810566e61f5e8433cdd405006c7e0a90518195f3a6617709e3a23a148bca6efb6bb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.481604646Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default, Name: checkpoints/64/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/64/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5be05365330b57ea8f4807f6d0adc54e622cf36440416a28ad6d8d755bf29bcd7dc44a2414b0eedcf605868e6d99a533116b383424ee27d7321e91556de4359401f7556b2199c9851e0f6983655df591b7ae8ae9a5efd39e89185c0a1993694a785f508dca8f421c87559e50ee0baad0200fae6436205df0124940fd3bd56800b430f5fedb2c5c2db85fe712e95b6f3de4b729b0e5e0f03d996f3fbdd25c55dfe64aeb74d1883a316f874e742497ffd7625dcb74afc3121edb9e603490a6dd90ab3ac2e108f16bd71660b94ebebd594acf0ab49b042acf55bfa608242c712d6183ddf658707428eb8c6251a53e8c3018d71e2045863732d4ab59b93f7ae3dda8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.481672396Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default, Name: checkpoints/960/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/960/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6d97f5872ffe26e3633020931fef0cfe3347b88a83797a740d710f42c779c0a1b82e9e8d205c977be3a0f02ac7980b594ac00a5bd2a78d5ec08e0dd658cd0f34d110d83b1d1370a64ada642a2330e6570935a8a28f26acaaa1ab3aacdb73ffe50782d5635b14c2d6fbaf35c8f6c8c8a42bea2b936133ffa44e359cc7666b069b390a6554e1fdd0626095234423d6386b6550cf2a7e60c5ff7892a49d93812e4e3e287098dd890838702fcbaa825f36d6cc97f76a1d5d21741a124f29dc293a14a78dc72dbbfa02d65cc658a2eeb47f6001a420aefd909e34df55f37d85828535e3c5c75a0fb6639a87aa79e87e6f5b628767634141a10ab16b86309b1f2e7b23&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48173268Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128, Name: checkpoints/128, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/128?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c4b5c03bf5ca62de80eb7acef9df6fcef3fa39be22b681ccdcac82fa31e8956ecd91bd83ce3b5b42b692c109b8217bb9dc0117b3ec8abf2196ea833c73e84109bc7df19a38ed057cbad8b8a7a0caa90af19f1b74a094b2ba1918ee960f74cda34982e20999ac2dfc783d01dbaccd5cbee849b010963b4a5901c94ab02d03338f5542639a134d68e14917165c8b179766b175dad557479c6308a34b383d23a0304d8d132f65ca5f2477d434a130fea88d86066bb91994e3a8a291671086c630c6d4412f4c4c0feb832eb2914b5fad81d274ea8bef67b0c756dafe1689aec09046e0151870a3acbf1b9a9ba5dc7816fc25703fefc0dba2f0ce2e445b7839da6273&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.481763479Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/d, Name: checkpoints/5440/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5440/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=007592fafea18e86306b279720106244a70074b35798168000adc31f91e08d308b6ad7ba7ae56b604ce1a4a56280baf2571bf26a8bff2973b1ab786dfbab75e6ff5f3935d91f8cb001bcfb06d2c6fc71adfb78d35aff169011af47bd4bd1973f252f6437d94a5926a2c842802e7e826c1b7c8ec6d247f0824b973c23bee98d9018f0a7e50530f72fab101118c13582975124935c5d82c7968372531c59633b0445b28cbe9f2699850fd66d82abe78eb5a8681c960617912ccddb5f64620f845869304ca857f4af7cb676c1576b916b41cddd163a812e2db9ac1def6bcbb979efa1422f4b67862dcb9770f16fe38ecea99d2b8cee3c176b5ed6378d5c9e87144a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.481847213Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default, Name: checkpoints/3520/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3520/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3db7a10d3fb91c5e6667508de1d74a57a5cfdc301fcd6c23bf22c6d1df2b3800e019afad444893036b9ff49eff83ecffda416cf4d08ee8b42c978d3f0b9e562a046322cdfdf9e8e31ff8e9702908094eb8dc544a1b910e4f1383ddf0b52dd25b1209d4041cfbb57d17bc40ac2798696c6953683213ed1434152e8c5bc8c0f52004c7e48dfb8c37c37901fb66a84cead464ab45caf01d569e398da654268284990d43ae9d4e9be4ed87fb2aa5a16a5878d2bac01a9b3ce5935f9403b30717b8e9d4df56641ff2bba4b46b7e8eda2c91a8620992e56c5ab293de088e142742fd22789b6ea7a5299c4365de52248a7db5117a9aa1fbe45c0761fa77209575194273&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.481947869Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752, Name: checkpoints/2752, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2752?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a42fb064cbcd43ba4c3dbcc914f564a39bc851e18234f966819f730fe4099af24d66c0f7668608bbf62a41e9da22033851a4ec4be97bf1bc69523f989277f37d48c94354ff4d94aee2b18c9bf85da3698689056d6b632c0fcd9afaa8f3641e60f7fa9842c1e854b511e7ca3634c9ec4988c2091dc8306a38a64352924c7cc0ed2fd571e507aca40bb3fdb3949c5dcd1e4d11b38afe8aefce2537f3cd1276b15aa4c818eb3079afe27fef390d090af62b9df04adaeb6b24c2ab07ad8cecfff8e10125d6912b1481fc9b531b3113d64cfe9607f1abd7b2a6c124363c703a7488049e98db1803a6149ce5b5ac884e0aefac9b99ec9ec1cace7cdb7f4606967c9248&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48199567Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0/d, Name: checkpoints/2752/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2752/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6318a4a59156905bc76ec750e4de2f96cbc1e5f4349afc582436397cd646031040a8f6861bb62eff805ca7e80eebad7adf25819e1256ad63c2612f4613a3ebfe1ed7f582abd2b0b637a1b645ab1493c096e0e8aa00b6c49124d990a7359b0e76fd477998ab50e94837a6a5cbbe6d23426fd6a94cec71afb67b8895d00aa96af087d081e618ba51d16cbe0e0361100dcb7f7440697574f9bc543633f478f34085260e93c2ce5a999b2da9542d77b1db62f4559ec5ac48f21429ddedf72d2e8b7e5cae5ead8fe2831625e7d2629fd39340e6901190c75f74f2bbeb0c1a2cbe7e4f3bc1d2812463453c67394057e17dd1060b8d51cfc611eb8cab614e6e709154ab&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.481855921Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0, Name: checkpoints/5696/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5696/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4c504a2c1a476a0a6152bf6d8a0da166ca0cd9a0ee33e8f6d39b649f75997e20d308b5011de3e76a3686286f098325594784814e36cc932bfe3b6a7ee86b970b1d362506d6d49a0fe4562272e7c7de7f37407f4616d3ec42526caee066716ea70fba5c0ec5033e1a270e9af57125a6c94aba47fc6364543af24ab4ada795037555086f710b6f32b26f9d6c03521b0d28277106ad5d8d26923da34702c4ebd61f7e3653c19f52262c797771198368b5aa2c1fc6092f5ce586384a277a9e0abadb18a4d9a534a4c73d7098d5b65e76938a7d01ba025d2e3a1ce480d52b70e12003e3e1d52d3440df6de7e4fcc96dc877aa9e24ebb87d5ec3c140784b5b2f791b39&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.481854494Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0, Name: checkpoints/896/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/896/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=50d7fb6010e7eef69b67cfe47091f116c4f03c23d5e9c92864ed90f85c29b282cf62c774f89e6db2c7372f6e4cb28ea002bfcf4de08a7f51a97bd3c849834ec252c2bbaa93523a39900cace2da1081bb4a1bacec60e3600ac55152f9e5b0bc847706aa6218d8ecfba6486f37ae0b1edeee07add9debaa052dd4b71872900457766a2ea792fc21bb01264677f26f0eb11dc9f38da5117834d5c155dc2fdaba24ab7346bad98c519c7a2882a9184f0e5430bec97f10862c6d4afd45f2c4cf4159737b64b99f0f920505f652971ccd77ea0099000a72bdfe265398a23e997a3e03803151208c6335b9cf0bb0907782d2eab22268fbef51767284a6f03f7bff59dd4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482066905Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/array_metadatas, Name: checkpoints/1984/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1984/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=702bf8281cd007c598f26424d3da9619656d82b59db9b917e13b239b165d13b0c9630619a494f233a0a85b2bfb348f12a18d38d824313e21be8ac9127aec5b959c8835e1ec47053fdfa11cd5306df61c01024178d8818aafb772de6c826da685600f63b753383ae4da8f0cdde60fd3c20e727157f5f97fcaef347e5ef258179a5056d6c2d8213ae3c2455ef9a7c2ed4dfd05cd1b5b5a39431cdea28d72da87a322e4fa7ec7fdc8f82840a356a9b60f04dfcfab22a0d73d5a48a6d7a3fb5a223a0238d6773011aedd35df81abe3b9081a7414c1dd3b0ded5dcd3ed30b868137d4d1b72b9ffa070a2a184a3764bffe85f4b4811a7533fad5ee7b024cbe2377e272&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482165186Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0, Name: checkpoints/5504/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5504/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=810dcadde89ecdd070c7718244c9ab52d61f17f4c570205a093326dc2abd1ba463083cc26eb906d7085ba17ead319ffcbd1039c6d385d4c5115584c26c46b5f9b884791c55dcee1e48113d38d4d028e700aa264e286aaaed758a5d32d95ab398e4b42137bf27c2bf50f80a82dee1d501db8539979c0a59b2bb60baf5e8dcc5e1bb910449edb7bfef76897055eba6f91cfe60f969b65bb91c7b03240db1579ca9bdf9581de4b21e5ca26b2bbd612f2b8ce86139d5a2f7b42d54c67a603809cf6218c5df4c8e7309d2727955266013733604105682274005dc566a803905a8a2dbda5e5a2f67ee2688543bd54c64de06470c29edbb353722b4205700fbff2ec9a9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482116179Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default, Name: checkpoints/320/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/320/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=16e68f85653a059eebfca3b7fc186c03911a1a4b1bef79c487c2b06ecf847b2a257be7e6c71f7e3f22a08ca17ddbc4c05f8c7d780d2566b0aec526cf4bdfdd5ebac5e512b633bd0d79fe48556afa16237d2f4f1be51272b04749f6c9466ac9ab6b5a59dd504c6fbf1a1e6393c03b851a76675f8dfcadabc359806bfc7ccaa7a6a477a8ca621da9b3e504cdb7635159808f5a53074fe277df60346cb149d8401311de6acebac1fa8c0c01babcecb696bb0220e9bb4d56bd64541e408307f6037345e556b84c717a6d1809a85a188a9f3b0c7b930e83b6c076299b1574a9f23aab9a2fd2a74cc2a54a75a078546118515fda657a5f54f9660e493cc20bb595e282&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482176174Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608, Name: checkpoints/4608, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4608?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=be982a54425f67affe281bbf3052eb3071c7d172373d6b2672eec5b23bb27254308b2c65a9cdf7130e1fd2178b889933df4da8dcdf405d6a0debc94012efcbabbf91892b6d5b630c019f7bf0373aa678e9b6cbc651e5907fa70d8589d12c1f3ec6a2c767d0df6cea240eb794c732eab6da7acf98a1bcef0ff5019535db05ec9f91637a58d4b232fb02e82d1413bd0f216a83ce4ce63f259cae5b25f176b35bfafd0f199a4e947c7554fbb15cfe218bd11e23c3cf699c2c7bdedd25956aa522311f56c4e2a5585db9eca1645105402556ac19418e502a16df505c731749e1194c0fdb8c9e930930ce81e76f9d1c81b05f8b52e4897ff9a5bc9627c4259ae79fe6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482198904Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/array_metadatas, Name: checkpoints/3776/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3776/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=235bf0058f3a5dcbb48db05a1620420ddd62729249abf4be3604bed17628a7d80f2dab80a3f8a86539ff926b5c62d2bfa35d61b3aa7cedd0e30c2d7572eac409caf0d2114d2849b6773dd5d5bbfad18ce9669a7282e24239f34c2a86d5fde6c1ce6b9ffbed3fd8fa9c2dd4721866adf1b8249f9f8b63d26e8fe853bea2ec3bb88826920f43bd6777b5385c4a2c680b7955d8cee627b7249131086f2d31d5e721709c7538b5972770d6d60f948e4297ac632ab6f98c48f80664bd62bee30d6a9a6946a8d4c669ca99e54ee5b9e07498e7037083af4f5610140cedd9110e81d56d1c226ee256e58468ab31d5eb9aa4fae08a2a3db99f289c2f76935d1a2e2e1d8c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482280474Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0/d, Name: checkpoints/1024/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1024/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=601320704c649b6d547f6f295fd898ad16f360778ba50c2fd2e50031c8990e4470d2a6d75a53b70b037c334e2ddb61f753c2c178025de8ee638c0f63e5fff70e83ce3a0dc76972a512ff8323bd19131dee91404cc53e279f923a53fdf30a4502a31f2a7defdcd0f5290ad7c7c65496ee269a95aafde7e3a666bff0f7b5292e56dd76f9199f23cc723dcaddccf40433f662aed91c393bb2d0dbbaa5aba0c093b556db1f0d48ff0c17c0b9ce4c60f114b7626e759300a01bd6b45e3b2d2e634739a448ad958e7ea320265faa6d091ec5ec25ece65b605577c4afcf778b62c97668940348a2d1d218cc12d4678f3e2d7b5de99c29994dd17a8dc69d94be0f2b3ab7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482321753Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/array_metadatas, Name: checkpoints/3520/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3520/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=81989e4ac35abfc7c47747275fd464012c6c0fb8eb8a933588321c9a679ad260714806cd0531661b065980f8af681afaf560da32e046ff673192800e23bb768279bfc0e3993ae42a31e6cab10c5b2bbea126db36e2ce67b61223fe56bc88140ac0616eef7d1d6f18e7f0dca7755751852949d76b6da953a004e4b5774cc02b5aa7538a1c30fea8d7adb808821bbf61c663cf7770a2008bf5e9b9c1485a021042a3948155d5f7157a1aae216dec74a2cba6d4b0615d8a2f207abb73078c3a142d6667bad9082f0f193a1257d84dc4383b65204f4ad8da6b927ebbb19b5e43b192020a7f755ea1e836fdd3486472d37c4c6fc48883137aa9f36e078993617e47b3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482028081Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default, Name: checkpoints/1600/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1600/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3b381b98e5d7fb66cbf7f926bc1378491d045a962a9594bf63a02de7096a826e60e5b116e23d756a9302fba411d055dcb09651464c881549e38f274dee241599671327195d33f93b247a42190a2bccdc538d7d0198e2242da40dd4cfa0de7783028b0017259d3f556542459d535260eba19c6a32d9db7e00022f41e127d7339f7faedda5e7cb2107d5b437fd0381b6004d3cccafd11c2aa9ec092a182e979acecafec4b25ad50209ea3b7225eb8dd8e98c8ad131145fc1c051cf8fe36f730a1eb9e18770949e7e3f4a12723a370b7417f6c58219a2eb409e2289d9dcd842c153c6faec687b28fb0eb24c30c81f1116f0439a7921edb041ced73404b3a2c4dcbc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482176162Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/array_metadatas, Name: checkpoints/5248/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5248/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=63b00a6aa009049ad5a78744b95fefacfd1cd4abcb7b36d300e5c1320a9c3aa975a67ec45740ac33379d79d1b6c8ae3e41c9ca6c13e4f92a2f15d3f58dc41ec1e49327c008da123e124c5b089ddfbea0cfdbd70d94f649482bbe7720038423ed5f80f812d2321a9bcdc4a959ffabcb047e8ddd0337ae96f28cff9442c6e8a2af25ff54dd7a1ca9c29d14fbea5ce8ea86460da1a592fee0080467c4e7198f4c3386bd893a7f2bb3fc115bf41778b2278f4c63c2e4d064e1aee86586c38da69d98f879bd2b9149a44ca81f6b7a513e83362ec982cf8ec135f43404a3068d8dae053203d46eca5d01efe4f4d1118237f4cbb800e9a08d49319d89adf5da91852a90&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482307966Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0/d, Name: checkpoints/2688/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2688/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a57a957a73079c479656acca981f5514cf3c8b5b7574eb807d8883ae8d9c4ff776f0fe484d5037b403a8f829e829aeceb3786c3559c6248ecffd80aba23e6a1850dc7af8f4aecc6c2d1b4a8b1efccc71f52cfd10a746769d7cff9f720365b4bafb9f4bc80c61be8fb4f52cc94f52e412c8427ab608f35f8257619934bf6b2cb6750a4edc83cc81d802cc8c8e9c897c752453a5ad11b05e2fd48d25a4561b3695bc603147248cd26a3c5cd169117413737c350ed693d8ce7b62278444af0eaaa419585aa03432ffbd0f73c3074a0b772c048b9512632688a9670c90d16c93b344e8ed98129c57a28d1eb8c15fdfe58e9f59fe00e1e2a47be636e8aeae27d77367&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482381986Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0/d, Name: checkpoints/3136/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3136/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=310ef023322389b44d680e0fddecd06e908a9b916c66d9a5543411219b228999b8a7dffe8c0eb3dd88baf696c87a7269de2f3d2f30013da20fc62b33d7c1109820545fc4004cec83cff2c5b49a2a71a7a33a0464e8852a8271648048c989f0b817a108ce3edf79895ca04a42973a4fe43657f1933fc365f9352252083e4fc0a7677fa8daef972da864fc9e27310f34c622286225d2f8cbbed81b3a1efb3d37a24b46ffe382d0d1db1c87622875d14acb1205cd39f8a24c2649252972ab66248b442bd3d7584c0a281dbe28751b07783c9ba149a251d2d8606faabcd93adbf9a6878ade2a08a6bef2eb7009b686a239ffcd8c3dfa1def0a1a3b9635f9a4a47cc0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482411025Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/d, Name: checkpoints/4032/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4032/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=66d7b46b743054a82bcbe199ca297e206d8166c0e045a1338fab2d85168d1498ff830c8bf72308b633472da4e105983f72ff8c917dd584ee232c0b9b135552bab785bee1dbadeae6dbe9e5b33f4ff14e1bfe90b5ee980830fa46a7dd7cb8fad0ed8039dc781c17001f6db39c7fe0756ce1719ace0ecddb291b728431aa95cb26598433df62b304bd5ed5aa22c617d8b38fb663dc3b5d1a5e5d24d7ec9c4ea19638d6d2c554178abab1b0f255bd5d65977d1d61982354a206f1f030f5132c5a9b41ca08a6550a519a75585c1d9d301c74950fbbee6bf46a532dcbdd193e3ee94cb7497c922074dbf6f999e72450411e828555dadac603d9bd6a2ec41afddd3ee9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482479744Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/ocdbt.process_0/d, Name: checkpoints/5504/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5504/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b5ac2437bf9e8e6b8cb8079ef78b2efb79628912b1507a5270bb4a3e1df2f6f75fd7d500499a4d5bcf8f5f114bf1ef51bae70231e6923801f07cde2bebe8b59d606842784e8399c24e8d4013834cd905dc46710a8433069e28269e93371f72c5ed223b49c36fb01e1537e98fb9876eaf30bdb5185d2ff4918ecc2b14fd97df3c5b82d2377e47e1b7612b6bc77dfd1f25b85281a9dd2ce7d3546b097c4f13403b637b5558c58a2fb676f724e0fa3260d4b287f47eea007c39b9b0a748dd64e76e8b17877715483b1cd2eb8679d3b75bb6afeb455ebc4b133dc55380560d365f19331e29934ff38585076325cba31f50ab5ddff37f074255b1ec7becf3b643798b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482545576Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184, Name: checkpoints/5184, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5184?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7d232e9372564048e5ed26645157187f5e675aad48390a503f6007d5580fb59c08a4cb329fec945564cfcfae3811aa2f95ec975e102cfeeb750cefb4d0721e7e20f984dc7aeaf5d8405fb3cc9111317f2b9d0c3a32e364201c9fd59cb56aea87d7a4886fbb4834496a380af7680f1fabd759e4c98a6b4820ecb7d0fdf786c0cf34cec0ad84861f3b3ff2377df2d36b585c7fb9b931d631d37d1f876fb5676afb407e9fe3167f56fc341004d064b2bd42f81ae0924142991c43be34a69e4dca3a11295754ae4c55268923ba39231daf6f1b856922fe3e3f68e17707c2b31a7a190c0d4c37b1ed2a539e535fa79a1ce4577f06378f5ddce3e5972ee3ce3a43eda0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48225971Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0, Name: checkpoints/3008/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3008/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bf8a34997e49d004e9c38f6a841a88de400c9cce496a0b7fc67f9a877d497eda7408dee5e8d01ad6cdad125098f7134683d761db5e19739057bef9aafdbeecbb8b94bff76c5d4716d7b40646c100dc682bc44d5d16056a121e5c81918933de597b9764fa2ebc6d6e2c4e1a1d94ebcf6e4dd9f090a22c44f904331429aa798824ba1b58af44f0d9ebc21b106dfbc564623e08842c1ca6e43e9ee71cfb4861e87b18876a25dbba2b5b43523f758b00e9d626f5b95cf42415b5a57466079e62d877daf5b38bbfc587f20a3e3f7038f16c0c6fde2d49db101dd7641bdccc1dfaf4130fe3ba41a6c9b577870c33f86cde88e86caf7f720a612dbdbbce95af959acdfc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482591491Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0, Name: checkpoints/3264/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3264/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=95c81b27afe62e2d21e97b93fa0e7141816e4237d3c61225620a866ac7778e78a75a16bb806727849d2b53edd3d0800d753c315ac80d9c0ba26dd28a356f082604ee97f6cd86b5c297294d37cb69c4dc5a296419153e785f9c9840ae7631163799598072a18a1fac9f4c6f9bb8a4df0e2d2227d846440b2bc0c3d3c341d8302c7ba8e289e2c8e89f0cc5646ce80b9fd20d62c8196bc00207dd5f39af98b864a050c0c0eab4db45600dce5eded087b94fba6857d2c3e231c56d0f65ea2d14d1bf1d1051265bdbf6707a2b2908cd44fdd28eebd450eedbd5e34ad0e7dae9e6e32314cfdf5ed3983c378dcf818aa18c5283e86248349d2290f50eb1ec5a02302547&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482623017Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0, Name: checkpoints/3584/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3584/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8c9b07672d57c6ee9f67f64f2e37c89d15d81454c72d79ae41e055d764469b382485c27b8603e729f18df0e7f3cf1f5396b6ab074608c1c93e74e7cb54cbc4fc202791ca636257065e29849b517fa0422f24d4d2a23300700cd8f4876eda1c8e6a9bec9eb79daaacf7797099d6c71c3b8ef7af3b41f100b114ba5a043489bd7c413acc410d842914b2e679ff00285b86636f4273d02f8e6edabb0c54927fe8366d094161581ea7e12dd4d1e079867c75cfd38ed41846c267d2c393be92fdb9a594d0b66d9251f4aea04191efb619d6279f5e50af5f3c7848f7acb924a1f70710260f232b8ec1865dc2633817c2e475fb8cb0e86182824476d3363f5f2f401198&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482635555Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072, Name: checkpoints/3072, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3072?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6e6e16bc4c2282d4ce399e29510522ab9f568ba888af37e73b7fafea7f42db3e7d00c3e7ba90e268bd5409e29775e9681d520da9b6777ad2a983f6d2540504a78c5c74e6902c0315e92a0c6e9393ba1ab779b27a937fa4aa2189bf931b150e209b9b83989f875efb060de735b2269ce718d1a771ff1d70ea4def990d70d18e7f4341a19b43677f69ca884651619b1a08ed73440712684e0aa5cd83de55be6bb4316991da1a54ec43a0b2d53cca7d4764d2524d20d3e6d3c5f0da474a0c17a23d955ce15f0811cb7f15368ca4b66c8b4029b921bb36e305c8e9267b552279600cf75861a25a3ef18ed6fe2222596b4b33a76a76d4100de9d56eff35478c1bdc9b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482478466Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0/d, Name: checkpoints/4352/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4352/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8815229af51d1b1fdb05234c49b4afb3cbaeec8198829dfb5a21499414a9118ac52e35899a9e8b2d6784796a8fb92fbc3a693846d32fd4462b97fbee1f868ba59952a6ee2373d020e190cf7b3953d7df2b3b0d8aaf7c8d4752c7826ce1815ce8c7357d13dd18fa08feccbca944ea38ca547ae491b74927a12131cda79a7e74882f5e46229faa297ca723097ada5c264ce0d09356db140506fa79de58b453e06fdd4ad72baa302774500f0ef40980c05b9ebef9454faaaa9dc6a64b222a08e20f8cb64e658b59e4892effc28010dfd6f6dec7d4eddea0bf5bfd8df9fb7ced03343a1205ba8863fd0e739fcddd0a6b54a0d0df7663cee67a38814ae4421efe25cf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482682655Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default, Name: checkpoints/5248/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5248/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=49abf39e614fde5c131a496ebf1cb607f3b803956cb0950dbf76c4f59fee335d2eea4be662e7b314847fae9d76c8ae996fe2d1e3b90825733bbc4cfc10099df7fd43bdfead5fcc6f3388a0068618dc2d9a63b01aa534eddd6ce29bf08927a8d5ced7273ee136d70412d869bf0504c0e364617fbb9b22f27e4f63b8f33e21060e3a04b2067eb4b900dbd1d2fe2bf11c9924ecc1c1364db0816b548d1a11b896856fa29102e90310de3522e6a855e8b7cf3eeeb67b25621d98029c14b21040079757b4c9111c1c5689c5ce4069caa89f3b4f6a20fb3abe64eb28cfdace1a5cc536044eaa4e4c4391ae588d60fc55a30aff82c31858faa19e84bc5ae7010bb2d1e6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48268036Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/d, Name: checkpoints/3712/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3712/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=50b2a7a994efbaca5fa81afa85a9502db49fce680e6392c0639d37b3aa1cc6ca97dc424679bf5f07fd3a87975649813d12f2e654fabae63ccecfb5c93a716414cb2c8630e61889edf5b2405d7077a604fb0c64c91d50af68bb810422cf1075d757bfde2b74e2177154d324b386882ad02315da41c9dbfacf9fbd09f390fe2b4c65b5cfd6bb9f5a271a42b842c67ce7c4765be28f72f67384be621e8a74f17e67e486e1353b4b967c55f54a84650b3b64ce446a711bbc47567d1b5e3026f056abe2e7d19198fca63aa61f70cf314bf72ff9a6b9fc72732775cf184b2d7d7a633578acbbc637d7dbdd1b5ab22ce15b60d837d4c86586bbf2093e11579f80ea2400&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482705994Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0, Name: checkpoints/4224/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4224/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c3573b59d113aad393e8a01c077f3dc7595c9d1bebc272186c92d332f044fcbed84d584c3044709b4aaedf44dad040985c77368be5790c306c2329288c2a4e7d6bc124868aa313f4432948a4a86d2ffb93c3182d5d3d0302dac3cd1936e9fc90e8aa5c08816d41deb979eb53f30ffffa0fc69ac2b2914aa22fd48b9c7ae85b9382ef4f06911c0802431dfb0e213a9d7589737c08a7875e5728041478358bab07267ed0369a0e31f64a1f163ad5e9eecf3b8654f054f64172c9b0379b46a955b24df6983087e1192cabed458a14ea9c4b8b8dc1947d167f6e04654f509b4848b9e4765b7f44432c50403f3c495a60aea8ecea1b1365a699709deba07a6c612c07&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482766793Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0/d, Name: checkpoints/4864/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4864/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=040dd851456304354cb7695f685ce83bcd6737f567232f156e45c1e6ce7636922187d2674c289b847389d86fbd44c4ba22672abfa36a0f164b762f3a0933c66ef03caf256f9b58ee889cc51f47415e292ebd996c06a698a698d5d6b6f221f9f6df8d1b936688eb80012c18a3fe128866a4547e83a753eec50c6515b0e9dbcb120fcc24d014abec0e4af934c7e1d198f964bbdbc8b858da5598783537de9f53371ebb8abe1f760528437931f10159f9a412f95ed8639882dc6a3b9e491c3c8d5f36c8769ed296593b0f0e5f2815ccce2c312c6b0d823901adf8eb18fd8503164c65fd61674986c3f9e64b6838d80fe40c1e274dab7fdc5d0e97b1657113ae79b8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482791817Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0, Name: checkpoints/768/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/768/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=415e9ca0db5c8b4997e93dd924a9bdc0ccb7bc559a6a56018553c166430080e8e6f9b8aeeb3b18733becc175a6f149f2a18b3fd435e9ca275bad6077272a80eda900fb91bbfe8d53a826dc84a82120b311a4c0c94e1003085e610eea501be919b941b8fdf28aafa36c88a1c088fff609c2ff9d0239e08b467467c55b2be63cf2140c9503f1f902cac208f958f7d020467104fe0f04f2f1f24b67e13b7f06d20bcea40b28a14929a8daae8fab84c5f593825d050314d79c3bd30586e3d6f3031e841dadc2809e7be961ea92251af276a100a5f374b0bc8b170681876d867f3f7ba2678309100ff31b4a00e57406ccc2e0bc1bfe9d3f2c87eef23becbf9bfe2d32&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482750676Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0, Name: checkpoints/4928/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4928/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1fed027fc10835058fe41e7b2d243b87fe76128efcc7bd525663fd47ffcbebe22f52f21894178c1903b25cc220ace53035e2f5ac954f3a645f703b4f6a1a4b4e111bfce4efdcec30dab448d89d12eb4b0eed0bcbf919ee81fe3611e88612e5cc2ee6f334ffbb7c7b9cb62576dc1d74fe58f433d19bcfedee17a0701fd1ff66d9b59b15f6ef73c37fd9f1aa4190eb17d7416ff4b32fcfdc10607495e8061063380d97a81889138532a5e6a818a7e3c497308dd3a0cd3ad5e9df741eccb4dc65c4504e2a690750bdd82e8a95bf65bbffc3f1dc7d64b18e3faaf5c4082c18bbf5a255ad31e69763cf733da6542b5bdb7f0ba0fc3e5747f10a2c1e4a302d8c9a5b83&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482819712Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default, Name: checkpoints/896/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/896/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0477a806a86c7c32ba1205f92a8d25849c67dd7cc170828e16f6a0b30610d05391f5620d4370b28bcc5104a8208ac61398b26363b13a868029a4461b5dc1d0bb1be055bf0e06845033b98b98e06d7d9a844a25eccc202fd1a3dd09227d436c72a16feac585c7941c13ec92fde72a28beaab5e6c8b73cc120ecd161c6cca3e4cc7e6d6912f434a7dde5d3b91ba56ee88debac5ec3be074db836f07d6173eab9c05ed6147f4935234ca0c9addb3ff612b1f3ad928e6da6125f8a4a702575daef0b9115037a326a5ba9c6512fe5d030f33673ba21d094513f79c6b0429a2ef053437be908f951a14019161a802aef9a5046ff7c6765824bc811c4742532fac799ee&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.481642758Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688, Name: checkpoints/2688, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2688?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b573c967a236875a962855ea5317c1ce64f66f4aaef4b512d4c5be26b9866251add123dcb44ac694f963762d0ff8635ab488b847d4600853147e1ec2c7f4d24994606b769a30cfef2fef0fbe0dfc637f5cbebbbe48d16ab65c46b74a36206c58078cd4a55d66177cb3518f735c1df5fc778e7ce41ac9172b8bc05077fc715031de908bdd2439fffd74012b0ae6daacb655ca6aca386d23d431de0a8a8e3c3f62bbd8b27fbcfcf266a30ffe86d28b6bc1c179d3ae73926592527a0cec55c127aeb1df605d6984995f387c5844b8a3789c29b33953b7d725362c87df75a32579013b7f0ad4bd495b83ac1649a8c2b649e19a025f0129d8467091e4350f70623513&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48285509Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/array_metadatas, Name: checkpoints/1920/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1920/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=68e5abfb81378ea6dccdc00c18c65b840edae3c0d39b1e0a54bccb5a4009b1cdf202faebe8d7dea51b258a94731a9338821497a36616e740c0edf8cbb12af6c34fe016314f88928d0be996fa2a8c3a33e145787948905cb32be98177c44447c25662ee3a3f9390fd7fb8cc23fea17ac370162b02b3fd759d852ef029bc7ef7deb028b2f61b8442bf647df1da086bbc3105434b8bc255fee38245564d6b710a6070ecc0f50beb3d73a73ff4246150543ac4a1c5c8ce666886773060f71bd47a1f761010cb9af43026717aee3da0393a1a2fe2fd0a7c6a7814a2f7ebec4b99bbb7defc95603d8ca0bb672803d3ea2d9671fdd759d2dc1ac81496d69b8f6d21fabb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482745609Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0/d, Name: checkpoints/1152/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1152/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6bfba25803ffcb5482a33fd4269307a5cd24021ad2e9aa5c51c730dcb6a795d65a9b6fef908a22e8e439934715514529147de784057087812619b2905cc74180f19316337f8da7c90e573dd137789945cb10f72b8892623f4ab9bb5dc16bb7e82e1cb3263b4cfac52a50319edacc55a7d35c2a4a5f61924be25697106e2604084cda45bf6c35ee88c60b501b58c2b37ca0ea281cbf9705868584d47778698367b0b28b91e54ef5e56c74f02b77f7e5d3faa963304dd13c23849ba6267aa9b937fa727bb1eb779843c1c3ba84197a61437afd2c812b94cb1b905fd09746d4920e56192af70e85d47fa31c3b3c6d8e6ee2497db0dde22473a0f558882fa49b3c70&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482957163Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/d, Name: checkpoints/768/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/768/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a504a18b697139afcb5a3bd28d1234b965ac7fb85be5de23603d18f1b0700299f5f0c26bc1a9335ea90b779a2f96efb1f7da749033866198ac4aada03bf5459a240c5bf65be65313d2ae73ce8ab578ee8077381f2c98cf5ea6c9ee1cf4332170c6846adf867c0465ea5aeb0184d6ca7481d38b8f09c95d7d94a21e3eb762f7b491c442e7ca457876e7fd985215bf572dd655d3f3fde97a959df25f12d5086d7e72496d8b9c92b0d40615e4dd15bebfd416be82c1eeeedde13972530c51712e2d20f98d255d838e8cf6eb746e16a2acfabd717711c3fbc31df8c21e56eb922fb27ba76a8d6ea8d778ecd7bf75d6938199f40325ec74746cd1690cb957c98951c5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48295985Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0/d, Name: checkpoints/3328/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3328/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c34728af47674fc1361e8eed11f9efc19fef461ec31d92d1dd5a846803b30efb13863c201e0f8d981b00f1b5d83cbafac329f9ea982afbd91a565c2b0e07eebbe7642b2939bc21efa49fb79328e2681d1902f8e0b58977908b09063f86404dd093b311542f85615dc13cf6852162a1d8fb32396df365c2a77e9d3ac67caea67943630dd5b5462310cec27ec10e20e3aadeaf0b685c9be12a18d7f96d6a17afcff7fb177e6ec553e038a2c276707eb6eb36f1ee94cb8b028da2d1ef34550b5dfef16f66eb639b0b0da78e330aa0b198c4b4f08358bcf9be8fe6b9841a5929b89ccd88b14c97a6b110591c71ed90f9a1df4b3c2e5d16c93f65678bcb53c726a8c5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483014657Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default, Name: checkpoints/3456/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3456/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0542c9fa519e8915d840d6d36091a78bf1ab22dc1d38e941d6ed9e8c0fcb33d204b7585bb4ac10248b2e030c74d9164bb1fae07b23f2d21188661d9c02c7aff55d0aee065707d94a970b857bae57cb5a11f1f827f9abbfc9fc2f2f68a912c3b9989e46b934eb80e4df41c54bb36e9c4a8fdeaf718b15f1766f6332dcb899258bbcf60a1f52fd1fbd565ac717521531bc9566dec70a5d3a6b3ce66a990b9ac2bae9a258dae074b2ea949601dea1023e48a4e30482a38a7849bc8801c549f61afb43707df69069dda9a5c083b80862902224757f008d7963aea037edba9629755721de710f075f62631a8272416bf391c36e13c13cf0280c26698e977a3249e757&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482893014Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default, Name: checkpoints/2752/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2752/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=647e97eb05bb715492c6c8d2c2b45f67ce137007a2b5f14b02b3b5366226220d3cef583f7e966ba987d169e6aee980b5ca15cadb6aabd074420a6e069147156c9092a5e3abc2ecd22d2680881d14291093e4460f185d56bb9481b99cbdfb5e21494eff4209ef340a7918941c01d4ff1b6664946a5f8ab993c6cf125667dfe65ae993483826aeaca0dcd7101f1000b1c0d37b3e74b9392a3cecc3cee4f39097022acb6a7f7806f767ab6e02bdeb01111bbeacfefc515f90dd6801d17da570b11d9f33ec9b6653b56b22649e6e13f7a4045dbcd569fefb36e522ad52ac0930cc90020a466d3b851b3610fb5c0b890b13dd40c22d220b9a787d5feba4bbbbe72e74&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483062222Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0, Name: checkpoints/4416/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4416/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=70aa7ec0bfcf60aeb7b11238c658fbda034923cd14c88b5a72778bf7b65947995451d84a7d1db6fdbe6ae5d9d2d8f46ce7a6995f47d9177eda1b5e7c4776d10a2ac62d4315f1fcfdf874a8c271503abfb0f1b5a751e423f3d3bf10ce7781fcc6345d812b6ae6604ddb77c078d8dac3cee57abb823c5f3de11b201ea32914acefb0e90482f0051e03366e25b058072b18aba5d710f3213e362057d49e55809e32bd9c1e8462a21c7277cb8e20e255d76bc8dd59a9f9c7fe23cc70c9dfbc67111c23ab841d7f70cd66896d1fa988234053caf95357d60d7a6f8f60362803d62086112590b902f6bd0d432eacca8a544bafea56458a26b140c28cba8d5f362e76a2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483101068Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/d, Name: checkpoints/2240/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2240/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a7f76f98b9f7a61068a66a5cbc59ba9c035246e2ed9c8b97f5cd295cbdcdb76a8866748d9e4dfe5119d1bcc8bfc3f509c62819b7e36ada5dbf52677533cff9ce7e174e174302ea3e172d471109d6ccbeb68d5a499623a20bcf7d38b9e2c532d63575c154a09fd7c5bb54112e15f92dcc1b50249b7b0edf654d2a9dcd50afcbef4fd610e194981676dc4e98375d7557d3882c2a12a741795a5c6e682e63d1610182b9f86fcc2eabb7a367b1d62c813f1ff1627c660e817522e096805396c0f8dfbb59c2ea067a804e81e5a8e8b90633e95dbae0abefcd0bb211a9345c5bb24871ac9b2dc20bb799dfc7f721ba1d5c3dec2e7b3980cf128e5dd9dd68da340a852f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483375514Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/d, Name: checkpoints/3904/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3904/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=46c75289d791d5a48710e7d2e9b1142853f36b71d94707d69b8ca05275ecc676113f6cd4afa2738c78509a6ac84dff1bfd8ea93bf933bedd4db151fde2a574fa80755c4e422237c06e4a7f34fd3e5b40a78f5a00d5d0664c54f4bc27559bab103d62b1fd8b52893b939d17b29e9fa976c7d67c9ed51ce7f8254c52127b17441ca9843cf3dca4c3c3822b412e75b3f603bf27b7b23f36e32f60263a08a7f459473b39e0efd2d1eb10ef466f78cd5490e9664d4b13e2cce4ac7c021dabeccd9b96b1f100cdec433a463d5dedadd3249f5f83513c0970266d65966eb29eea617dcd74e067ed285a848d00b154e4d53d35c8d596a636f4d4d8a39af7410a02940bdd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483456175Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0, Name: checkpoints/1408/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1408/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3a65d6fc14c57c15e5c3c0ff17a38fdb3bc66d81e10d39e3998eb9ff74e7f5dfb3bccf918178d9cb79cb60b86de4f94ebd97d2e7a3fa662210eac1e7860282a37396beeab227f2bd703ba8aa3c5dc1ddedd4aab66e2c9758f3c3df9560ca3237d2f3483ed590aff9f1e0a5f3ee5caa08df1e300301172c5d211d84e564c859354ce2943c44931ff6db47959896578171aefaeb33cb0a7f83ecd215c16b8159494fcda003f9e13b06b52b808e1b7f3f47ce4cccff385929121961984c6a94578ebb9c8dff9ac48cff175e14e40841234b51c14e0ddab59db2580b77524fc9254ac04bff4ee016c1cadd2aad29fc3f39dd15a9f1fccc1fe24cfe7bb8a4b109bce1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483438768Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/d, Name: checkpoints/2048/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2048/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3c64bfbbfc6bc0be40113de1152516c21d4b8ea1b61fb31df1f79d36061c6204be4f5201c74eb1fb211b2313b0f46d90c99286ef799204d2594278b7d3a6e672df071bbd70165739be274639a6e4752c995278907aba4f2651db5ec5e49c5199c9841990c0c3245c0fbbf35009303cad2419ae4a3cb1f9bc22c7e5ce12bb112c1780742f0776da21e15fe2f8710927f666d09506e472a269e1290e7fee681bca2dc11d232675c6d5ccfcd1e6bfd1970bd6dfc747adb2432495c094952c764f82867abbaf52c1194c68369c09d3ccc5a364d7db8272c5d64603b03f12337d86b12ecba181458eb7316c9a8f842b0bc564c2af17c7bca115fa23d23b4cda561804&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483518754Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default, Name: checkpoints/5824/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5824/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0c5772b4cf6f47d3fa1c657f463adbf86ae697d8b76c2ec031a1202a2026b11f11ad17aa82feb26869b93112963baac3ff2c599fdebd30581afcc1a27e9d723273d2496bb60935db30cf760121fff4e084465c0d58ca498aded167f48f57ecd7baf276668f6c25382d2105b19680a1f4324ae2a2a865ea3e8d828892cdf1af8812b10b472a9eabf53d0188bf737aaa20a3c239205a5e28944dcb117e56141653abd9a44c8af352ea3b7474f5f3efd488e074fafa85187f600a8fdf5bfe41e662e55ac13d95a7c34e42f67a0dfb1a614f9c3bc0e7755e00aee598aa88f89eb2f08f624acd24590f63077d82f015f0efa76e899335556b64d0204a1aee96d17c11&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482307099Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0, Name: checkpoints/3648/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3648/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=44cbe7fdd4e58b7278eb7bffe65a1f1d9cd5a9c3b3872da1195993d33d00dd3e30e6de8d443538ea4ddbe3d7d51e8c08ba6c3e48b5c0ffec3380b8f261dbd31ef9c4b510da2308e76aba8a0f97492e6a7c7f005ec845dadea4191c3e74d6a8e7cecd93c39bc918098e08d3bed6dee1b0075697b333039b16f8d87a39b23cbb10e7de175c1fa9991c4e530045a4fcab0cf9d50a9400d3939dd89298d7b62b66f49a807ed66f04986d182c4997ce667cb154ae3e33082f5e896b01b46561977b67ef4da7c3a5a4bc866aa790a0e1daf90e27e4d33189c941341f1ee0ea1811311039dc3cd860ebb7ba0cc7622a966208f9da2b4cc76f57ac726ba92af241a686ee&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483533902Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/d, Name: checkpoints/832/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/832/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=ae992135c502a80744cf9b2fa17995411d04afa5b322a24f04e9267619abe48d96075a09c2c8f0c8d99123d62ec6b3a19ced9a76b5467e9a75549325e7fc0873323f81ce16f6ae46187fab368bd88ccc829e2436816ab3c28110ce98f8dcb4e651c48a0b84b13dc5d61b622ee34160628e34e2a18bc30f1af94665811946653052813db52b7fcf3bad594383cdb531335f1eaf192adb15ca0720347e773d72348d141a2affed050f6ac2b4a3bf594b9b53419944ed23ec25b2f2917042671e1318834a4f2bc6a3cd4706387835c49b9a6095395848bcdf399b8b7ed3340490a9ab14a3e2dd2ccf15f3fd2b29bd352be7cb791c865dda7b8aed16cfad2f65be90&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483480596Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0, Name: checkpoints/5312/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5312/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c3f02231380b1aa20952878b7477dc0b63bc3bf4cd190be132e4dd56c1ce88c976abdb78a561a4c1d31758bf8ab511135862dbb0e2227dea9fac9cc563a342ed5d595a170e13bd27d4eb7d2bdf0c84c4197278af68ee4895e3ca6d3a85ed47e51a287e943fc164c91367868a8a7875a54834eb787f37d6d3c7a082acb29d5a777e34f161369a685fb2b8fc98c993552b889e3490995539ac67c99aae5b89f501bad41736fd39433b2eaa47bc4eee09e7ba7451c4e419378fa86ed6f9487ac7909f613d2389cdd91742e915288724ddb0ff0792a9da9ae373f730fd11f2422ad23afe2a7a059d708b117a8705cc2742c4c48c800570de3072bd3258d45375a454&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483357967Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/d, Name: checkpoints/1344/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1344/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1274f57cb61806446db838fc5c2886bdd25af4e64aea01a1c5f221d97c88da896ad888159705ed60e92d8debed14e66ea3641db39e4c288bd6a615df6ec296a478b7b599dc851cdebeb950eb9b21c2391382582fdfd3e5a7ba5b2a09c15582b20993d0bc98e889e5af8476fc8a324343f983624540d1e3e29bef3f9f1d2d1a28e9e3d560f4614fac86bbc0d094bdbfd59344d0d1c1de0c5687b2d1c4ae8a25f73d978f5730499695cbb1615c64b1711b8c7f87851c09d0f62b59ae305dc206dc362dc17b26c8f84c89f751537a9b7c05c40c94736c964c8b1baebbe68130d9cdd13e9b097b5cd8c82e3e9d3cd899f569c79896f670440f1d04593679962bc7a1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483624088Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0, Name: checkpoints/1856/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1856/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=621ceb26d82995cc888e25aff7f854cae85732963b5c58584340df2ad9746688da56bb39b8bc79b121c39b07ab02aaf71f1a741169b2ae01c6ed1888acc633116acf2c2e510d1aecaa76fe7ba367359adef3aa2a07d373d7fe23076462af51622e2d60ef6ef42e6a6a8c37a4bf2df0d3b8989b419f3856e651d8af15c9c818d7324d1700b4aa274b9a090b0419ae53c3062d73579fad4a9187b0be680ccd0155b37b6121186e1ce295b1611df20d8aeb37ba6cb12db01252ad1b8fd32bd0074f71e1f28135f4a0403f5ca9144e9b3419016600929e5af8f75a15c5827add79bae84e75db26715c37337ca5b98e35e65134951fd11d01b277df0c90bb92ebf536&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482316154Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/array_metadatas, Name: checkpoints/2304/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2304/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9b25a0d6b1dcbc62eac23986be31167f89510a030ab8ae3b6f0c34cc0e5b506bdc091e5ab39738ae173a440cf14980b173f307206d9aa1e7c9e5043dab79159435504ac314d2f9bcfcf190154f7c015f7986ca91d57bd2a169a3855d7a4424d9aa39aae31af00770bdce826a42d0d5c7215af9566da13486d976ef42bc1db78da0ccc12d6c14c6fd93a39cc76aab7988e4fbbc5d4883f844dec4e8a0afd971812b2aef2999a95338e57e16eccc3660a1140fffa5a1836d2a6e67eecf86c2557bc9a542508f35e584067c818c61b0c637a444429b3ecf11fe3ab8dc21e707058e8ff2be610dcce2497fb2e5596fe2f90139ed795bf391cef8a2414986220bff96&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483458083Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/array_metadatas, Name: checkpoints/4544/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4544/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=77004d39c91a2caaabadb76dd68887bfd8c37f80e55f4f0eafeb17aad1c7bc498ff0a8c1027f992e77190414ad946ef4b55fb146f247f5d59929e78b26eb517e63e0b5d817ac5abbdd252700e2c7e2b8cdadf28aa3e04400c911dd73f386a492476d306aa57b6ad2fe3c40b992083793edb76d280e110ded3b112d2f0c77c595af561471e92c9d402d07def3244cd89c141a855e9f0a7af9ce3162114821124b673684f31c0b6815b5ad3bb4ed6addad587d601e1643f92eeca6fe60637bdb9982d782f4309769db08d617d4c222c97a3ca95e8fe26b48e379dda2fa5031dcb00baf6f602d1915243c25a6fee70e812a71b17d5f49c7b8da2757a287e76131a5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482739544Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0/d, Name: checkpoints/1792/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1792/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=107a590e9607b0588874336ec75ff15fd4e80a8766c18e70d2225c8f37687534780be0ab37f150996d0c10b0c7e3f3a028a46c74bc98ce4244a7b9703b28a41afd0ff5f494ca057e0d44d8d48c7dff281c7c7ef920728f72401abeb18627b0c25a9824be45b1957ca3a3ce4d552fcd4b487433d5869b83ff5899a335df4012e471b0b48d9d215b0fdd7a1b9a29eb1e2aa1dc4749ea308b156679e4e3e4676f8bc6cf681f6361c52fa6582ba9b8ac9b4e2cede5bcab5223afd4cb21307e25773b96c79eff8f4fbc3ef54d3f9b05ba2f02f80510adfc2e927ddf9b68b97dc5f04c8d53c453f403bac52e1c3fa6bff015250457a9633e59515b5e69e7f962733c57&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482392423Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704, Name: checkpoints/704, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/704?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6eb9adfa61a51a0abe53adefb5c9d74b1e69d9df935ffe444ce09f587ca8eb9f0ed21fba307ff8e2b3e2b6bd93b5b159e03a2e906b891e12eba2ae5e6e1eb7ce6d8b582647ce747b862e5073cc3bcb1ad0ce5ad87abee7b6d4a3f07046dbffa7a796cf1beb4ff7986be6164246a220bae50662b17cbd72bbd3d5108681a54a576e0b5943e7841b32fe4fa259e164f98d904b10e72aa7fad1e6e7c24310dd049f5d01204219219ac831af5a8802a1d15de6862d80ad261a84d204d43961e081f5ffb9822531a1269a8c66fadec4628de4da1ca1c301585c6e0a7b80e516faf9158b598a8f04c8407addd76cec8bc92857dbeb0ee24522ae0e89b5468f3fd850ac&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483511273Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0/d, Name: checkpoints/4288/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4288/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6706f3a8f9dcd23edde962e8c893ab427745f76823a7c2e6237539ac2b742f6580dc108e5b0bf316069c26504ec2dd3511979e0db7c7dbb740a7b39a6a4d7670e711fa84f095fd5cc0caed0342713355be9668b3804571333b155351a174cdad798cfaffec0b9c74f6d2beb6b996986d559015110e467df43bd5b2af0500815fb9a6b7296d4518d9675b455024b392f03eec3a8666d2cc6ed84358f7214507085ae8fa088f4914b2c0a388de0867888d12baa70fa408e3c7b4a746a602ba7d3baa12a3c37f2d3bcb2b5dec1d2e57e20a94672766e0872828e6ed631dc8de58c0d0a746f767896e15fb8bf78ff27a7adf7a2a1e97cbb3273ac34bdd6140a469a7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48248263Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/ocdbt.process_0, Name: checkpoints/1024/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1024/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=831c716a3d6b4484e261b3fc71563c4b8bd689fc727c7618174c72d4898095a2e37e69876f3bd0248fcd1f2275ef28253f4900656179223876ef0d817391669f16b8b7a5830f9d28d2553f29ed90391cf485045deda395a71e0d960dfa56b64d439d50e6333238d8152a88d12f074473b6a39d397636ce8aa942f9607fa473c8889c66508ab607ff8a3102c7880928bafbf9249dfcd6b39d6aec564cd1dd1268989ba01bc5aaa1989263208d4553e62b3bb58ab690cff20cf97e778df95312ee77c57a45a79a50dd00ef3c2b28364740cc169f3ca8a91927f2798cb767003f0655dc58e9adf6c408362aae7f7b72c184fb5d4d8b539d53db08db855418f9365f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483512204Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/array_metadatas, Name: checkpoints/4864/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4864/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7920bf4d83508ab5d16aefe5ce6b4689580fa20bef7eaf5f63b1aba8b34fa15a0508a4d78c80bc24b0a822a476339692c92fe9a22c535aa12133ceddfb65a8f57c86a19a0326899947b3cf406114aee63bb2dbe958812c1a2fe5c279c27ceffc8d7897849e83cfd547a9235df81c74ae66964965c38bc9482184d042907f67221eec8918819241cc75f4e1fb84d7bd01599c17639937c48def9e8a7bfc0b6969eb3d0568ed647e40d07d7da7863e15351412d70c63b9c466c171a18edd8d0a4b439e41bc1f25d8f1681702b01aaddc121bfabe789b50eb5c804f21ce30e2a409669be7608d1dadec980e28af49daf7e7d065932d205934a90a73ca23a6475518&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482958664Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/array_metadatas, Name: checkpoints/3840/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3840/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c072a9b62c2931316a55cdcd961ce65f60e71c2e58c7bb79abe22a9b9e0c24c15997183e6a75226d6ab15ba5753458f1c7e8436e64539aa38528bbb7f7b8b05a29ca0266a8c2990617902ddea5745e6ad0f66a135590960c9f31f5bcf5658858c556c9ee26ccddee5ec3f421a94b8eec39960fbe80105fa20b1fecc0a202f7cd0da4826fe76c233d994c4126b9e4ab4e359c98211a837ba35a08b6c441f2f59a5654c3b2edcb8b221822626c232ccbabb7c1e21110ff14832738274a0e9b3e4fcf051ecfe39258993e57ded8c2bbb64d85a42e59a9d8e14b694e3b7b925faca248f1bc9432edc24b5b77e694b1935402d949f5ba89fa72f10475a0ae08589846&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483032331Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/array_metadatas, Name: checkpoints/3072/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3072/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8777e4da708525808af7a644f64d7d81f475f82e64772f8a8936bb451d7532b26fbe9bf4c64788ddbb97a244ca14104291ec50cd7d463625d9b993dbc92d343e0b2859d3e18a33298529ab89bf4d90958dde09c22cf51872d5a82fd738fad72fd45dcda42c17749e41c02ec6ff44415924debd972d2512da03552c1caa74efc1f71b721c7fbd6f884e29e099e168cd696e2fdc41e76e505e411746a0a4c5ab916529284ee0dc5c015385b618d846830ae8f9558757c24c5fefc1dd51111318b5e6f17c2b4a9bfb7cab6a2aac4862efee14948be41f410de2a545f368fdf3b02f618a301d80bfe2e1616cc061886cf1bd29c9e7ee50e9c3aff4f2f1ec1e7108c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483950743Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0, Name: checkpoints/4032/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4032/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8e1a1a2480537d7e20597f44d52dae2dd52fef2eb9458b707db091201fe226eb5d396682135bee12b6a94641efab4c70745d31a5c169b2fcf48f519c80611e5ae201ba77258e8bcdbc2931d9ea388ee609c52402bd3c0e8a383d7893221e87e34b909f527bf30e4151915316967565d169298e8a3613d210881066d442b9c775dc8f8f265b93eb7aaefe7b7953c92c027831b5462f7c547dca9a8a8e3cee4a4c8137b70a997b1c522c6642fa1dbe961c42a499a595c8d656ae1257e48c2343acf5ca243510975a037a1bd06e695a455fa1c0bfa093d9fc22644bfe06ac1194e1d7a757b9bdd9e1bf9c8eb7c57067eaf8eed7db72a68374765c87a75e6fef2ef0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48344972Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/array_metadatas, Name: checkpoints/896/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/896/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b9823271e10ca387ad289815a76c0307aa07f6729185c511a2497da0b5d720f329f908c1211f4dcb71563fab466535780aeb4557d0cb56fb1406c8f3fe3ca7216f87fe496a198068b7585be4008bb1102fe0550814cfbad14b95959861d60d70c44d14ce5fb2e685543033f900e0834dfb98b3a02de57d86e4eee60f6de8c2cb19663510048fbbd79e5215a0b18842fb4549d5a7d47833f08c8f186f96e547f2931353938b207373671f02f6c7237be3b3c6930e70cc3a12fb9b9b46e3ee7e94ff0b26942a500e48f790a14b4cb92cc5b506ce7943a957afd2da4e55c28a8103ac9b089bf18fd46149a9be0af674a2a5dda984f35e5d2d1757b90463ed7ce554&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.484015049Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/d, Name: checkpoints/2368/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2368/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6838207b65e31b0a1b5d2ba5e14b3d9ee7f0d25b31910d9f1acd67cb07abe6b640183b4f8179a2e8d9f98fc0228f3721b25da2c0a362a4e689e58db17b1840ff0f5e0c2e635e533f26f8b40dd283f71b22662931ef3f6dac614389583ebb3df489ed2613943a7ed874e86646df4139b26fce144d88fe2c22fa0499d16f52df72c6f40361c5898c4a1e49f7b9258a6987e9ff5fadc90f059b4e8e7c4fa9f0fb0eadc5fdf156f21ed7effadb65d713b39f87ec42bf2f7d23c06497ad6e67687e70375481f15295438979798cb134713c14f6dfb29dca56d89d6af06264e4a4aaa2a96768dfb345103855fb8ee99de3b559942e3e7b73293a149387bf6eab87c821&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48213258Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/array_metadatas, Name: checkpoints/1856/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1856/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b9bb52eacdc0ee9f57231c1b0f1fd8a6718a53e82e928f9bee62d4a6a55d24e0ca366b7637eb87450672543ee296a327b7b6d39d9261020f37160d5d7ded0ee0ce72196c4788ff1983692847e28c7d9e676fa83e7ab698025f307f5bf72dfa53058827cd1c8a55256873c6bbcdb7120975dae12f97f1366aaf5ee6c858368814b678252f9f49268c691014fa3a9d8c392769e408b692889fe7ceb6267c228b5a91884691c858c4ff95e5d8579569cc60f46f061226ab33fe7db65eed5d105506fa32384df07731a97d01f2794fb14bd5a985b93556168e433e7bcfbc690fc4cf441602644aae271aba2d51df96cffe59c47c0e6ed2a5641981646df54a7115fa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.484856776Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0/d, Name: checkpoints/3520/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3520/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9552a636abf8a141a0df42aa68e3bdf5c814cb6ed5bf8ecb5d8264277f94ca7c13705f5a2255fc7406817a7eea5897ddfb9c7f33a5563feee84707f00b29e01fdbc82adf21fea236a9dbe5446433ad0f1f6af9f4f356e0e66a75c683370a19ac5d1c1d1cbd3b2fd75e4085d24c9ddab0a9efc55b93d0add8d7b56ac68d56e6c98e9c8cfe1f083e67f1c9400ca4b60c5c9e3b875553ca81df4f3ab23f935e454a81be3311ec4bd991ba6c5da5d6d11999af530f516fca27c83322a4783f4104fd5a50c89e754cb8ec3114e9c656edb9619190e44a56ea69397b128886ef38be884dafc9a9e3d840f7f7a2798d7e141f2cad95e4552bdae7538de2e12a45d88460&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483492281Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0/d, Name: checkpoints/2560/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2560/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6e1f02853d7a61f1e0b90c0cbb21a5f0f78f958f4dcc37a9b55e3b7c71fbc5dbcb46c1c7e70456f2c1e9ba6431e5d9309c40542b4b4967c6e0921570f0da4d68e9233ea4898367bdc2a4ca259ee9beb65c193c9666928c8236fff3a394e3755d37618583f6f5c24de4c6e2c6feb36dd56d1e39525772e35be2224b57abba5b552c8a90818355c22d879d0fe3b3de806ed8c7e89ae3d97b26dae6ee4420c33633befa7a066da64dcf486a3f3262cca533f01b5956b563c024a33b2b02a47954922d93e8fbde78242808b89c8958564113035bd00ace0491915358dbd7086939f682b34dff25b15ad642e28ac69920c8c7e96ef3f308172449253f1d468d869b40&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.484013555Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384, Name: checkpoints/384, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/384?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a00d620f97c1e2e11bdf2bb19053b3d9896831549de1b9d185a0e2ee813d93e287c69bcbd8c8c14bffd35e31eab59a2c779ab3ade41458e4d15faeb8aa28dff8f747c7abd6654e88f7429b1cdbb36b16dd40005c7742411db4dc046591632cec9af9293056852adcb24394bb004ab30dbd11482a3470b4c495759fd828bb254c72fa7498eb991505b4483b9f730a082c1d6bd680bb72b3a65472c802be224497092a1eb97134e7e42cad42082a0d332d3728f6545c36c93979980c44e2be1b16e8f624a79f92f02daa89bdb6180eb193b20dce68d881e875f8b76e6e7d13f0799976df4e8e310b4bcf4dca2b35fcbe632693f1825d70137b984e70525d77e3f8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482816132Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/array_metadatas, Name: checkpoints/4288/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4288/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=465d5b2ce73dca61e851b62a2adc205f550e9fe57f77d9187618ff36fc9f7ad3bbbc12e967dd8f90d804ed8f00b51c3a8c31f54178e2643411e69b60a24d022d9763e18a2c647f5752b555e5aa197806c743d754e3fd4d22967331e26117521353903b4a82ea31cff4509a94612feb81ac8bdfdc460b4bbf024ae7711180963b1ec11663c3e57aa6af4de90a2df84f679087a0556994ce5cefac82ba1cf2eb11d5c82d2d2fbb4ee38a5e79a0cd9fb1fb85781bb75fb7c60f651a2703cdba47513e7b68b9fbeb481f4e160ebfeb931d9d9125c29265a487f9e37303cc8e575703eef7d7e590306da83c131c317a9617d716ae35843bd4546b66f8a97259ae5ec0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485127025Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568, Name: checkpoints/5568, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5568?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=ba8f393182018f87e104030e1bad89c948305cc0beacec5e61fe1a55f8741527f9768196c7db9962ae2b8f4e66e4839ee936eabd8c2935d96cd87469ee6f8606bdaff043011173838d9aebb71c2af6870357097d1b7eb5ff33cb726e5c0c84a9b844af62ff7e1ad4dea7aed9c9fb15790ce349b0f385c77487d25da0f4fe2bf56df949f2067a3e0bf10cd4955e8b66530a795e4eb99f3057b3595b29f7990f29b64c3e7e0b6ee0d380ffeac116cbf4aa20ee81377a8e3fe131a0ea93cff0fad6dc8d6e8929465d7489b09cad2b785492a009e0be595cc1c8019a10f18cf12f1f528b1c257c262b9f0f68626c4bff2ba97dfa5e9f7716e2510478e0e8583081d6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482534399Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/array_metadatas, Name: checkpoints/320/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/320/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3d832725741a2f0ff829d8431c5d2dbeedd63daff8d9d0e5501d9c0223eebc33d0c26827709cc88d7c9e5b115be240d11bf66cf3fd05cad40e1ecc9299c048ad8e82d4034d4caf1faf243d9f674b925555540cd57865bb5c7fa4c05e17a58e6d1670f424ac4d6f078873fec1111ecbe0fea87c300d03f431ba9f4dfbe048bc56b4bdfd50592894b4a4fdd93bb1a2239a7c5d4061c81cd226076fe26bcd8bbf4d5fd76f294e83352eae14dc93926c599b1baf39ff3ea60ccffc624c565a650db04e1233d7470d9cb850835c19e6636da21e146923e05c1025cbbcfd7de8e041a0af29b5ffc2001ea092bead7b8d11ce6d6e8833cc04983e2f4d16255d777a0732&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482702216Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default, Name: checkpoints/5632/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5632/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=52218e3d439986ea961b48cfa222703e086a8e4ba22429a88ac10ff8bccc9d896e5379e1ec43967b8c6e8ffb086dfdf84ce5b3745cc774bde3fca60899130fbed13ece81ba0aac59ca6b35a1d083f7442424d0faec9b82b643d352922e854facb923dfafa51a93832a5223d9d9cd872ae44cedad240c994918a51f81ff7dcc3774144e077fee3c6babc435298faabad0f9c406a95b917e6cce3e8edc5a6e34327e596151ef0c0e3d20193b45a3253e99b292481efe40d2e7f4cf82eb921080ded0d9b0ac9beca2d3b47b890c7b362389c6548ba412c210ff803091092d8d4db38d490d3c2e7546a6ad5f21a040f70f1eb315cfee9f3358ee428acb7c75493370&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482642889Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/array_metadatas, Name: checkpoints/3904/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3904/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1b86ac488844ced758f7cce00911ba42e4db0639a98340c4536a87ce6c62a58fd5226b101e896c81ef2139b5e5faad5a991afb93670685b8c0fe7c82c6ac98ed2419c86c00bf0aa931d83d2425a2abeea88490fa32359c9c315884b82ef7bdc9856bf5904716a1de57502c1f25dacfee5df4ca6f160246af20d6b62478308fac98d3fcbf548f3b4527c9a14b0b826fcfc77c16d8799bdfcf21c553ccb27ff74c415fa7885dbfd6af37f105cdeb43fa73dba286e45960046ce4a264f942b1143f1446f43cd1857fc9038249bd1ba219460b91533aba2bea83c3d8bfc2ca7cf973e2c863b65c1800822c5abcde5f12776dd99a17f4a652107c7728f776782fd65e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482217406Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432, Name: checkpoints/2432, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2432?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b9bd72284fd084fa1fd2bd1de538cb6e7d78cbefd04d0b57dfa89e71f0b86c72fad56a4df5a807845830e1bbf7db64a8e10dd6fab320e5c827fc7bab0e38d41a82bf0e6aaa8c3889ccf07c1d2eb5889b88a102dd89a9578c070de5ac9f465780539c1619b4ab7c5897710740d5c6d0e350080b549a65c456450c1b9830aeb51820d9d65d2188801505ba0153a72369d75ff2a5f7da003d0b45a41315ab15deda0046277b34d262949a5ba5fc2a3e80f7d55d98930d7181618b2d0801233a97a5cc5566c350317612c85406f60240c5c690096544c44b6fd6f3928e25e76db7b4b4734495abed8eec025cd16e6705da90c341431098040625d6d2bddf7172163f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48215218Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280, Name: checkpoints/1280, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1280?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2db2c1a251ae9b2b2740385f7ff246e1ac0b6aaa0f5c9f156719a46ccbe4216d3ab1569ba0cbb32b935f7915727a7f7fa7a5f41f96976ddbcf01cc2e5d4da3a408aea30f72e23f0e7509f7778596e913fd474500dd2180e6037f879913429f7289cca367e102814a6e4152685f492b7d8fc410d18fe08854b8520e7e261be79a4eac6298b12fd7feb34a19bd85c64097dd11d38e016bd74d639ed0e4de355f70007c47c35ae545d39d9a7e1e3f6edd2374e0aa218362e129b9f8f6a402f824b47b68ad8b267b1ea0986b1426fb64f29793eeeb5c40e0dea81cb1500938631425297fd87eb481e956df6fd7533ba5fcf48f2e8ca04b211d79d073263c6be8121a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482090447Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/array_metadatas, Name: checkpoints/1344/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1344/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c24af34cf4db77c2307614532634560cb92bb5f7a9aafa3119574018896ea01e46edb995d9aaf1c5da1adcc7b8da322bd4cd3cce2973024f01470c5cecc533fe3f8a6bcecf027973b968898d3ea7806df8a99f1df23933158ebe07dba9756bf06fc11bff198484e530f5af27e12a5dec1331ab33de0f3751512423384739ddaabfa7cf1fe756dcf6751f18719536e8d2b8a0574c2dd4d0ac7fb1644b26a94ba38b90b2cfc5a9d3cc172d763ce80d26ce48ad714b4e8b549530a16cd419ad1898a48ced96c3b6fc88126c31967a2159c8c3715c467041ff29792727b63408b00838c977dce6fd79c0728ece4d0e092b4ec369ea106e8d7ae9ec089d09a140896c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482603566Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824, Name: checkpoints/5824, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5824?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b98f25a9407ad233276d6970865e643ecc0f0bd08c7325596fe7d45428c2d3ba97a1b994be75f788c1eae813ac972c6d48db859af83da0d44becd3335ce2282f76dde5ab0ad65599a78670142dca0606ca5b3e3c80f0b6d4f780fc568b1e64bdded430ce203ce13abd650e130fac2fdca0093207aaad8c87300cae8f2c3cb8885f34eca7af539ac17d3dc7fd89bfc22ced56ddb64756a974ad369b4e4bcafb1bd164ed14cfd242ed8e7b3efe064b93ef2167f1f0b64f226babf7b978af344382de6efafac1d1f08cc57aee729888c683232b232251143a2ea4fd84c8fd2b94cd3edcc964d37d4af50dd0988fd5f24a1b3c2a6b77cdf6b4176ac1d1f8cdd888f7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48209881Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/ocdbt.process_0, Name: checkpoints/1152/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1152/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b7533cb144084785870c3ffcb01c05fb32cc8df31269a3a1a410261b650501bfdcd2ac934f6c573dd3f368d6c8cb630177cf934e33fb5ec0ba0f739a50e117d7e79e9e9e2c597a863c689bbed87ff9b6c0c0e85172455dae044abe56e346ff42e0ec3b393a09e826dfa8b8d2cba7170f1c2e60d3b4f6963bd15b73168ff804aeb2b1ee19a3602d411aa50c441c014056a49364af2a34e0e8905f7d27a2d0f0a3df4e0cd3313756549d3a5927636cd22c4992035f74c309a99df4cab7f982607a1c91d49753eaa60b27138550103db6cf564f7897cd0c63c401bd68f2aa4137e744cec52884b6694aeb206893a583fa9ce67757fd793aaba5386d78b398415f16&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482257068Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default, Name: checkpoints/3264/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3264/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=156612bf9fefefdda6b25f11f89540728522acf09f39cb0fd394efde2fbb2ee3c7ffd09903156cd7102112fe83733b9c5f6fece099637a903f8432a3cf9b5b82c0a05672117c94ee33ce0d4b6fb7da75757a0b4e5f1952d28bca2ab6c60b7c257d6c45fa875a75051b07ef9e889ef54d24fd13ab93ff2d86495dbbe3b57a8aaec41f916992e624b9d7e7fe9bb7a3e6d09454d7df1f02fe23f5723315bfb6c48a43d55ff2a58938d24f40455635a6f1e5d800f05ebcc065e3bdefbd17efce89475a448f1812c363587161f9d993ec52b4ec106172a4395011439a49239e23306a6c0c4565fba6c97930fcadc7a4c42cc6ac00e3b655e86512583e682593202ec5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482309538Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/d, Name: checkpoints/4864/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4864/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=23110d9e3fced21ed675a210805029a001956e797cceaaa74c7722033379486f430c071c1dac85f3c0926f266d062753722f274439f7ad4e01fe368737a28e232d5db02689ee2931b3176441f48f51c263d8e7d64938f0d61286a52c4587dc1b1fca85d20745eeeefc690a68cd374009c437004ae26be750a10aea2e49e2fa4fff0ad4533d4195a94bdc7286c5c8d12cafc964fc8a42c5321aed06bfcc9235fae95559d4f130d7f7cb457cbb0d3b8dd71ef3aa5cba65cb04d25f3c26c173716b624a17c9dc10eaae2bd0be655f1cb86bee5908c1290dfd29c02b97cd542d2b06777ae25544ff0dd9298350331e9d32c303ec596d2c4a6590dce4775b0b8720d8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48242772Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/ocdbt.process_0/d, Name: checkpoints/4224/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4224/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b0a9b2aeafee09de6967653a7cfb4f20bc8ff8a3789a295bd184fbb90955bcd3509406766c6cc80a701033dd2456ece18761cd085f30f2c9be96e89bc9f175d3ed3c1a254b10cc416e805e8a19692d61aee824968b3b65c2fa86ca2c2aa7dce9df5264146831c3fc0d56be35c604761a55f2e897b5aacab873115d5537f4cc0759eb0ab6284de1508b91b2873e134cd8f2239fc69f46c32e159fc7684834ba44d0c2695cc6ba251a4890d8ecae90f8cca2efed07c91e09481e5e753b5a337d709ce4a3ddce7d1d09d1215734814aa3f1287ea1c49cbeedd312ec4bbf98e36b456ec0546100e39534df05f5f4ae6a71d7a49d77ac65744cdb1593418b2a827100&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482170209Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/array_metadatas, Name: checkpoints/2496/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2496/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b6dc086a97cf941587d202a2c8232375957d7e0e78802194b33d40a5cf2e35ab11490bcefeb38a4032b0c5056ec260b1f40a8edfa42f6fbf3169b3cc2e3db7200a60ea90ff44a8fba77776f2a34e2e4dc160bc1fccc614481dc5cf9e85cc790929b529a8d626d8b911eaafcbc4f22c8560cf72e68b6e1dfd355c52e5ca7d17b03e092412caae5282a82ebb8cb2892f6d7602be0bcd1d16b2b90a56bf086806d2cb01895a32d33d4642e2ab88b5925df353baf184dc13a447ccfb5ea779815c5e6da5ffd4d2444e71927a0043ba78e4e46d17ce6b44f35d3333afa11612da571cff9cf22113761adab21248715c2c46d9df3abf4407966a17198ce07200da830a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482426324Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0, Name: checkpoints/5760/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5760/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=118d8d13e129b330b6159f2551f3f607b9d5fe6c7f663cd5bba65e33115fd98cee1844501d687b547c9e132920811228036967022ed75deb022f59dc66f4b25a3eee560a741e62e9d0ccb3fc9b1ae6e56d2466310b6b3b1e00955e8dcdc91169c4fd5519db15c31a3d342007052342021e1a25e6942f8b195fb69317f49a3cdc2e7cf371002e3ac9986a3fab575d94e4a2bcde3f34f90998967d3ce51570b36f8d48d1a4411c6898e48b2d4922939811e62298440264241df7f0bd9eda008a45a463a5ad7fabc59b297c453fcb5a663cbfb7e44a8255725c04c511bdc9373370da2ad0f10081b061729548dd5cc011deaa7109a4a9bbf6fe23ccdd51cc135e32&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482422632Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0, Name: checkpoints/192/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/192/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=334d6dfa59f6111ad4c25e3c3d13d1dd67bdd1a48cc42668fcffc3ca0f6718ad027c6d344658ec6f15d161cf4bb4ced24fbeff7672c991fdad2ea4d02ff425d40ebb3b643aeb53736e9cdd4aa94111c6f75b1f8df2debbbb3545b5a9c2ef8fbb95136397198a87ea871f3cfa3d07c3d143a6aaec4d8da1aede9a767c8d20acbc62daae3d80345747eb8902e63d119f3aa0242610ff08bf4796ccb35ae37232a8be55496d443eefe411825c70767f3acb5fef0c9c0255d8b9028d5acfd0a2c1e524f5fc2371ac7ea642f6b41b3070f7463d4a10c0c7c0b77e83160999befa9d9baf87d967ac9a51d82300290be812c347c51faff2bf1f1d45a3a6ae0422ac9857&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482409317Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0/d, Name: checkpoints/1728/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1728/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9a23276861fef55482b781095f96d1ed8241ff71ed7f24eca13c26af4e0f043c0a46d7cdda91ca37db20c22d30032c3b51b3c827a712a0d0a7ebc4cae8525afd5f4c0b933495dfddbd890d9819516ed2794da5b1a7b2e44bd0eee9a0ede86f5726b1c999d844a57287c635aad6cec44e1f343b45a766340e5bf0dd3a0f116253a3cd4d3678149336c72fd2e7fe40eed62dd80e3a625287043d1a0d0413c93017a081dd9c0e1fb4c92f484d096cc9ef74477b382305bc5422fcc9e5c319d0ee37a7290d867411cb6396eed4d1c5e68812b54b0bc9c7dabc8aa56b891bd83597fe3deef47af20b68112c032232914625257c0c67fa5d31d8c82a5dbe23103707ee&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482129046Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0/d, Name: checkpoints/3200/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3200/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=33c9bd796e298ba3d825f5fbec40b708ade190d8783caf6e212fdc1b2731fe01bde21a77a40cc972b9a0ff13038f856eeeb291010c3a9a461a14fe8622e714d042dfdc192f0d6bee428dfc72a22341db78c0abf8fab1a5ce5ffb34d3e9594cafabc9514839443a04146c249f6adf19615b70b0b16bbd779bdbc0a1fc3b141cc6a933cbe8ec4f2c2c7322defc48ba33ebdac95c399ffb3a4d82ab4265e19fa761428c0c7b83449ed356613d6232b071c6774011c5fead46ccea938ae9036fe6acf937d52e330562b465d202a9d2e5ef3af81942b49f596e745f164ca6be84aab2d5eba8c6ee65cb0165be824438facb5676ee9306ec3809d6df9c008efe5d7282&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482555208Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320, Name: checkpoints/320, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/320?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a7aadc4d8dde7cb6e61f2c513d6e69d372dd5cb1e59abe5ddc214e182337264312c16647e82b9f3887244cdca7a8bf5c8b83bf046d1455747c2862026d52e50f960b963e7d25f545df63eed387963641afe5d50950a7f4b07bbe75ea59e4df17a0acc024c4729ad4ec686200b60701a557bb99f39cce67587fdd0ea94fa39da2cbb81568a4003dd518f9607543452f5a82f89ab500c16bcb21e3288bdd11749066aceda338c0e9144032572e08beb67a9244b40c383942f74e2fcedc9f6762b6fbcefc6924f05a4592dd4e9aaaf44f87a0d714391fce024de944af2c74aa1690ce4310c6b138eab44b7097a63f521b75fa6fcd35a4c45b115416a7fedae9727d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482570895Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/array_metadatas, Name: checkpoints/5632/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5632/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b228221234d7fa1464ea0746b9961cd51ea837726d02b75cd1f81b02cde18bba086ce6c37ff86d61fb2ed7d2e32e702705142d117667b089f9ac7bf579ebf24df1242fff93e863aca8a98e56491c3cad66a88004a5d7f41cb246726ca74662ae6a75423a75b5f6bf82ac00bbc6effa5cf560abce092c11fc3c7f72e257551d3f909d8092aef038788378363866b55da6847620b33794b130a2b19658ead6a5344077590762afd4a50adf29e38d53a3b3358e15855d6f568a96a641e07d5a281fa98abecdce89cbd39b383bae6b81895093c0951168c838a80507881c1cf8800dd7c69702520060377ba45f93d3e4d16074258dff9db2411e2ad27331742d6c81&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485379709Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/d, Name: checkpoints/1152/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1152/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=72b8b8b8653dc3e0b936a1dc5b0315a59a4cf7bc2cbc9e02ec951f71fe5d49c5eaee67beeedee7306444bccb4d58db3b3747f4ba74e759fba81a08b971fa45b1223eb9ef445211a243e99f1a3b32c92701143a98ff913892a26885ea0e5834f672cf2316c20a5b214a5e885158d7e2e90d6f90d8552083031d9ea1d2d975774d73ba905357af6f8cfd37a3b31df5652b6a52ed345d5a53806a43882c3a26de66feda9801cdd128d84b788516ce8e0cd1842b575495637df65ad9c020bdbf14ceecdd626e078fc47f771ec0fb4c23cf63970118eaee7f58819fa16ace04d3021136a2f2e0dec9e0af3267a63bbd997c29941c64b25ed23475e5313eef30d8b9a6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485395489Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/array_metadatas, Name: checkpoints/2880/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2880/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bcf5a94243aad1f0f448792eb998eaa278459cdbc8c9f47aa98872f63546a75b1e3ca1a720415fdcc5958403010c63474a167db306c67c4bd6a59d2ad26de1fb807bd6646cd5f6a0b41ada113ea1876470b6fc4edf124b9737cc658f6a25f7d5776ff75d87f3072292a3da1c36d046366471725a27c5e0e700821038d9760796174cfbd81b21100b096d710e304743dc616e3db49b3ac682c37d30743f0d6282c41d544e9d784f1a662df360e2d7355042048087144cf3a84b8edcd51dfb718e328e021144e5d25dab74c6d17c05892f89f7cc27b4e242ee44b7ed70b645a9520e5bf9b4f1db11b84d90fc4d829f5767de4632bff8f70d9ee06d9a7061e667dc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485405058Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0/d, Name: checkpoints/5888/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5888/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6dd26ddb5014ad2eeb48b3839a129595f63d80e21cb6ebcb52def8ba351d50fe3f14335d3dd64cd0ec4a521f90aa52cf7154ff1e5e604618e1a8cb76f556da5d8cc57384cc5392d5ad69f11ad69bbe935c7149cecb0de6b79758c13cc669ca2da05b6c4bafd04addb277e91dcb07f84a74e0caa532b16277fa91fd99de5dfc41a023d5c711a066e4fa142c97eddb0f2637e138b9e8aab2e504fd2108a0b75357ac3cace38d94e4d01f530211f2e28b39b0e12db1a55f2791fae87d86aa70618108736294a20d1df204f2021d97a902ee1406451d686cc6f93c3570d37db80a2fe19b396e8f45396a6f62ee201966b8e22b69f7a8d9d50d89f46b073bfc2554bc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485450081Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default, Name: checkpoints/768/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/768/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7b1582cb512ee196abc3eb24a3dcc5b27eb5f0405f7d8372c66722cdc706fa3400594595ad7f1dea58bc6fb25cec0d60a8a0ab78d1d06569252a6bcaf97ff3b868699f1b6f4534643411fd82dd941a4d25634d4c3f8a4ad685bbb27d1819e042c7cff7eedf6ca130ff0fc3ee72ff9f937b4a6fc1d30ff045357091d2d04a03abcbe28b44592e41effaa66a219f0276791dc31764a59857868094a716a4287ebc71e54391222864a85ca96298f5ebd03652e129e7108793511229dbb3afb43aa66bc3f37d2a0c0f0943c34d99d469b9ced05ddff2ea8ede1c1e453891761225db7a7ba8633262a6b5012e9abee2e88de3aec735d4fa5524b2f85fece54d7c88c6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485464949Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/d, Name: checkpoints/1280/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1280/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=39dd0f3caea529697d37454e3d60556720d08c8d1eae7e13bde98cf353a66d57b36db2eb34af22d31fbee131c2c385b7ef4ea2342ca0b70fa8358a4f8c064631d9e35fb5961352b89328e5902e50469aee2752bb3c243c76be4e4f3fa0c912d7bbfbe84a17c42f73720864b64b5b00570429c4a3b2eb78ded3e0f9d2638851c67520d739866ead82bec2e0edc66904ae9116670f1df760560ca9dbed410ce294f7b806e298ab8d9e4474801e109430394fcb7999fa459beaddeda0deb0265171decc133fc67592ca88d210c36714bad857657044edea762eb3fded49631e04525d104ce5cdb8b679f06b2f463f55fe76f287e293001f886f90cf60a72aed4c12&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485502285Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/array_metadatas, Name: checkpoints/2688/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2688/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=302713d527d37b387dd4000917ab73eb24d67a2e2f2e4f94d5902283719dbb2248523408b1a6a005e10969b4a9152edcf5bc5cde1c565ee33620fa69d1b49b44cd67149a4da3f408cb6a600cf8b29b38eacd0b3084e6091177870880c14328b90b7b35a08f0acc332ff160fd3a3f21b6255aeaee0217da21c8122ee166a86b2411b687010778cef36d3d2271b4e66617312a074f120b4c67e42cc6536f3d09e8336f4b0eb303132931d6685d459661e815929f0212ba98f368bb534de22131435a8cc639ef6135278f24fe5654d304e132d134e585f9254871f2b911d91c98ef17459b9703646227d6dba909bed28a2bd8918d94836c9b53b947c444927d2118&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485544576Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/array_metadatas, Name: checkpoints/5120/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5120/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4d62bfdb66934cd52b8c5204b82b276ae273f0c891218358702ce980f708c86595c951b7a7a879504f36470e053368c08fdc09d17af951810f75d86c3caaaf331645ec9728cc45572e2429722ca7694cce8e257a8ee710eb8ed314ffa73074a440dcfdfeb459383a964c279c20129c4d47ad169213c5298ffa5ab19d7ae701bda8cf542a3ab553889b5f778e07a220439a285617a765db5b859654fea4579126237917b0def97dc5bcf6289d7f03c7c3133549eaf7da55f040c5074f35df00de2ba89240832576d7ab0b1b80c62f91a2ccf45635de80f5819a58b2c34fb5231f3c98f70d31114bd79603c29ac30ca28a808e52942e079daa1bfcb8df8aac049d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482584844Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/d, Name: checkpoints/2880/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2880/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bd50130b512b6b20873bdf607ec2a446d7f518234cf4df230371df7ce1dd450cae802536594f93e60f5b8e14eb435b2e75e92807f611bdc82bb6a352efa2288e53a69c4b3481bff3466e03b622cd75ee68ac6425c44bd59836dc3ab6b655e34130e61834bc7ea533dfb31f89801cf808b151f6ac2ed96f6adaa95e6f96180175b418713678577cfc0be7f8ca76f83b2eada1452a263d5d2808071dd383345d3fb881f649d4e69cf9f2d3481dd312fd3d336c9578799b5f1ab559d20b9f2f7ee82958e83ad7dd5f739f8ddd901832772d4cc3b667e94b7ebfd590e248b1924046fce84947fdfa6186aa0674f9dcfc57fecfcc1f2a99167f948c8aa0c9a2c72d23&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482663744Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0/d, Name: checkpoints/5568/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5568/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=ba6072fbedded4bb50f236c00a9c6d5e2fa7d7c28cdd17f81559fa4fba2ca28f6eee5a831dd2d40754ece50fd44238fa6e6486d9f9c18cf1bc35ee84c35b17786b5b3da7b24445548d5ae865f6042ee88e317190fdc13570e5039bc53ce050627ab8a63ba75f5a77bc8ec376806e262f035d08812aa95754d1c80d060ad87aceac44ab7e44d294151bcd525c30e34624ff51f6f85b39e63003378755fe46e95175fb01b7d62d3a8a79dfe4c7311f9b362abd517e77969bda8d74814b0af99925939930763ce4838c05786fe86245e78bd46f30f4724fd83e8c4c7b8c83d1bbd4849e18eb71cbcde6c75af3daa3874110f9a137ac52cb93330d21b35a30b4ce9c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482630483Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/array_metadatas, Name: checkpoints/4160/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4160/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=599a11fbdcd0d18901013d98ef93efe3c2de800a2b14f9dbdf04e695e0a41463eb57bc6253cac9dd597a4c49d55bad21040df15813b65711c7904a17ec836b00295269ed956dee55bfffeb81205435f709080a71421f38a286286e13b66ee24ee01507ebcecb825d4c98032d23177cd97cd3214e7204ac94dd4181e8acf35c33e82b4ffc4e6d444181ac4998476c239ee1b086fa27c8bb4ed331f1543d92e194bf85ca9b55720c484019ac2ded8ea3f0de450b46a46d712eb700dcfd286de674690274067697c75b39a9eb4ad74e76cb0bd4236990f275d3e6c388636b5a18d53bcfc75d9ee1f3ba95c31654e0927f0e295ee7d7ea52a1fb0b178883b23d706a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482703519Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456, Name: checkpoints/3456, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3456?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a037dc42c3fc78c45cabdb121bae90be86819dabd84a3068a26c05d49fd3e002b4e1cd694a4af6720f2e25acd523a213b0c3b2c362f3b12072aa7af430260bcbe89fb049757f55cc2a3c074a7f18aa3e709f4cf1a0bac7e3829628aef33a4ee4cefe9f65e2ee3159ff0e2ad41da5de2733f90abd32e239b132b2a6385edaeafae1fcdcd908f054f616bce4c0f24ed8274a550b8bb319ce88de2bec51bc90127a7b6036b8c5be8a44f572f002e50d81044cca83f87cc6e49eb4f14d947a7d603174fc0354d71663d2ee6da428489586238df05aaf477b6cedbb17051100ef0805dca24e4376e10ec4387abbc4febfc3dc35e785f53490f0ed9b9066244382604f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482732076Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/ocdbt.process_0/d, Name: checkpoints/3648/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3648/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b84ec846d8737493dfa14b59ee7eeee98e78bdb57ef8c970b877441fcf36a511f4bea2404696071c66add7bc78177d6e23ba7541254bb483cddccd28b8d2c48c19fa3c14b9cffdcfb9bed2907ee6d7335ae75e04164f81e21d51cebc26b3c8c8fbf06f3a685d1ed2743361a8448660988fc0076d59b76ffe424b8bca9917d8c75da1a3f298a15268d34c2f07e61df1bb6ed32c5ef00c1596ce4269ce9983d6f7d10712283d41413817ca011d757f90175e96fb7943856c2e17313cd3d228a0e246252732ba4a638d05b17c78701ff0ed5082489f7d0788e87d6339a266b203d7db0e2c8e1eb8112f4f0e33118e8873175f861266396cafb67cc7f8654c3ba438&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482740058Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/array_metadatas, Name: checkpoints/4032/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4032/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4c9431c160ef70b044e07940e10fb50944c3852d74857a15dbbff6811e81f5699e518d6fb5751918f53eb6bd0ac03197762a7388ced5daa203506f9a643b2750f1add92a228183f8da8f982325e1145b000dfc73bcf171fdfd852456cc5e5f74aa5814b704c3a922ca19478dbe943b31e1663ba921087caa4dfc020bceae75c1c0d773cca7ec861a727e420a281435bee387330e54d55b549f6ca72a90872d9137f488c89e981d0bd3b693bedc9d808ab0ae4ab2c0259bd6fd07102e4d32ce4e8dd633f6a5baa4d591fe093158dd004a4fc76995853d018417ddb1630cb9429573fb6cac37ce304bcd89d17189729fca1a1d76c0e4896c9c7d0d55dbf343b4e0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482691099Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default, Name: checkpoints/2560/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2560/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=50d36a0dae8cc24b2d82c09820c09eb2eb28a2bd4e4ef0273aec6a6c2058360a9f11b55018274bfc8c10eb8c98da4992f4d5c720b161364dcbfdfb031995fbb2657ff5a6000ca05ecbf7ca17ff7b0b211b871853836ff0980fd9f0a169b38dde5d6e3b35ca51c842256047d44b68b0fc672716ff3b8a19a4dccecf7942267fadba8032b25fd3e2c1321be531118ad82cbdd0cc12d683f69d5801d3c4e2f13c03c37ccf1184ab8b8b34dac3ae843cc9d15c31f49d7275e6c6a8a46dc12cfd58a62f3713bdf959156327473b7ebc6a6b14ac6dc6e16a2002e6ab9905b4d171f14c3d54fa3b725a54583f530090c39a8fc8bae72945c3836073b2a02986645f4ed2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48257025Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472, Name: checkpoints/1472, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1472?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a5000fe95c8770141970b78175d54a6d3e6a91468c56fab8fcb22ac80999080076ff8c5194639a5c69677ef3c24b65cf4891df0a7e47479b2c60a800388d286e14cf6da34b524a58a080fd74271eab027b73f33c9e953830cb5765df635f70d0237b086adc178788003280fc6f576eef820e38666c55e6d57acf9e4e79c9cf20e56ec6d0ccdc64ab1ab7fae3b419d4fe5560eaff7df0b385e99bcf5567ac019f166762104f13794cf0dce2d0fc746c5f2df9c1ce466a580251d65a5c34209dfc5354ebc25f9a9630d2c44a113d65635e78cb51eb4b9759973b6680e8ffc1416a2c8692bcd8b7b836a8682341fa6b8147b6ab7678d8bafdd7b01789a45f568733&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485377239Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0/d, Name: checkpoints/2240/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2240/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=385fd4d981971f57a1e598ae1a4f9d03a249b4017e459f84e930b2141bb792b33c6dde801444aee1ca955053672bae383dde27dfddbdc7f160ffb93ca1ead7cc0c95bc4a0ad1a58472af902041d20f8feee217775bbec17068c00a656ff5736f7ef4e5e20536b5d71819241fddfe335f5d47144b79f9d06910cebb6cf40bc7dd2bd19ba72d3c52b82dc32c431d3af3b6eedc2f758ebd175470c98cb38aab0f4e4a700d177644842a760e23b2000c8a7369bddf40deb85678a6aca6ba75d166bc1ed73f796ce41c8b0e7510e19118d72e970d0b5eb4c36b2a1d70df578e772d45be5f0522e6979408d50094f7ecf369e60f3a62c47b93e6a4982341dcc7e7ec05&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485644305Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/d, Name: checkpoints/1920/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1920/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=633b36010e5c3d182991beb037697ea179ff4b8ffb5c8027898a9f1fb00d6fc6190316ddae6b56a4678bc765b47f9fbdcd4631f3e8dadff8f79fc102c5095efab3c9995b89249ee2c636b7ea67224c995ec96bcd575d8534804ee9b2d96effbb0a12a76930cebeef6b71c0032c58fdf0c9356e895a2576cbde7654cba1e7c7966725ca60af95672d3f45542aa96de5fe3d55ba1347749461a88b326810cec78f7e359a6851a06fd14f2ce02a07fc54ad7ee45d7b24f25f5cbddbb6850f7120e7d2ad147bddc08e2bb56055af76a6a599d1a97e9ca65304b031aa6f1cadca22137bcf96f531aee265520183378646b9951e6214b99afe7da8623eeb9b2b77bc3b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482658032Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/d, Name: checkpoints/64/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/64/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a666418fc627dbe53057da02034a53e650bea66926449e7edbfd33dee1853477bd088b2ee67326cac6b814cd391056ad26bb70fca349b9ae30725e5914bd63e8cc2cba985a32e6dc39d195b57f8f460739adaa56cd2c7e57eece74e8ae2a7b83fe3f2b1b25af2a4c6ed4fe6b1f0ddbacb7e41272bb3fe23c10790fd874b9b61fa9aaa57ca0b0cfbbe3ccacaaf29826efa23caf34344f6e6b94112292a4a5d7d78d45fb3c6365d5a3d6582cb52fd31d0019894424415f988c31d166821d14065b742714f48015dbfd8a3bf8c083f4954a01b53443dbe7c936c1d01bbbe25a5df1cdc9f310e8655b7f0ed6fab01d4ab0f13ed046f006951e5a1ed89910c578d8f3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482786367Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0/d, Name: checkpoints/5440/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5440/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=97e696d671bcf650f7b2c05861c7dd3193d053652dd597249d90c69168248340a44236984b3178787968d678d6a093f6c16d884b881c3caf14bf2348b2b678943becd7430b2979819b9a4cf72624af52508635aeccdc8c9f17f028ce5c1d7683cc71511e160205e85cf379428633ef30be25e56edecb076b72904a4c533db139bd78528004a271f212d8a716d56dcd8da5188c29ac8765322ccbffec7c889ee5fcd147f26529aa147edcb15b2f439d2db67619816339ccf83a98c833bd62cdd79c677e20e439519aefc12d7ac28ad54e5c4bb79d4084a4f04b06fb081bf94eaed415b96419aaae4914f19bc21743b3d99e9ff9913feec06bf9ce10f07565bc33&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48279962Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/d, Name: checkpoints/5888/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5888/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6778762b60bedf384559dd9998f10d1df3548f871c5b4aa541367d99390558e97db6054f7a1f0a84103a7856c7e1a6496bed6d0862e4f9d3a60c4c714f2b45357a2c378dc28a26301d4d583bd401cf5e2a19d7fd2ee3da169c0fe768d50d62fdfc06301cc8dab919055fe496c38e40f9d8693e9418a734d139502dfc8bbc5e916d75bca29f9e7f959f52bf3972c960f90afed37fd1b2cf03a34181a1e5929be6c9dd1623401e6576341f9bb64fdb3cc62ea66d1b81f64b4699d6f83de733a2456148052d87a8f20fa59382b94e72337dc0b72418a939125dd19e429f1d701765c2383bb9bbfbef61d8007ffa93c42e2c701bccdab991518fd437125e58295219&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482495401Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0/d, Name: checkpoints/5632/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5632/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=19db1543bd8f8ab0765c428f62566d03fbe57a39427811ff7ee0c1df261826cbe2d9bdd7a462d743425951fdce5035e6d7baf5799d016f79da233d59d3df7bd660cf851afeee549ddfe78bbd586c98497c42c09e201514fc142074149862f8a64d7db715829ae92dc5332ef522e163c8ee60b78dd8c49b92653ce8d2e10532b68adc4eb2a8ca1f86897e15a5673b7f74791ae69af16cd62495731f5862f75e0fa3d65b0a98afb4c414e7c368fbcb412d887b7745026de0c8d332710a4c8e71c6f33c73874029c524c46c32fdca2446d2181f74adb63d47e0df4c08c9ac50c39a6bf3998427a3ea57642dafa0574728932268f0a381933099adb45f233c4ca806&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482837754Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default, Name: checkpoints/4032/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4032/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=73b82cbc09f91b6b23c3f6b08453b513d45499c88a4a7c278d2e7bf624f8c51c7d8584f49720c9d87f984ad0fb1613981d6329adcaf416b008edc196bd3f153a779365d262d451818d57ebea00e34bf0bee2dcde4dedc0a393c2180b36c4de6ede92e286f511c44ebfec89f35b87181ebff15cf79bd5b2dac069914c2cf4a214a801222235462747aa88cf1e81d2582147fe5828fff8d087d4e197b969a99cb2d6c06f81b8c5f98796cdba74704fdc7795e79d83fafafb685998f391b9bf08a5352013ece4e90f1c820345ad3034cf5c0e7f206a8f0abf67fdd268e872ea8a44807b77dba441bee5ae7d5b35d29218cd5c0836b4648387061911880209ea007d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482825229Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default, Name: checkpoints/1152/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1152/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=186e63efa2054b7d34d71bcb97b671089799f48a355c588749fa261124e1793de4c3da775b131f73ddf3bf620e0dcf5974b478543a19db0b1d631dc885272424379d2dbc007e61667f30e2e9f0db8f7d40a7ef0b4439d880df5fadb64e0f9030917bd4485e42af4a01b1bf55ef16025de822ae61888594682ebf973ace8c6603b8e436dcbeff88df9f578b8ded5ff5c220997b553e319161379d4395eda750579702c4edd1f77ddff5434ac0cec4a4a36224607cf85f1d26964b3ab97eda85e58d2799ba54b67d6d0f2e73ae2165569e60f7b8058822bb26d04a7ae69189198d2df916f26c6cee81f0b29919590d11342e2340dc9ec7ef4465e1380d132315fc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482759918Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default, Name: checkpoints/2176/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2176/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6d4cbc9820a6bbf9217240a9471df132fd22130d6ebeee2eb9c22ff541f732f072fba95e61c3702ddd2f85e4d26f5c3a14dcd226c8c2ea71db0417330209ebc4b72b639b34903d7fa397ad97c4e0ff4d992eef7a306cae2a918968f8b05538d6003dbdb66ba98537e6b5fa6bf4c01de164b84f714ffdb37a32cbd2b8d37162f13267fe017353830a51cabd401da37be97bdede12f88c8b1853305b108b46b6fbfd802294fd00e9509b004b7f898b30d2eb565ec36e8c0b0a4bda939c1202118f111b6b08035f99ff697321b0fc815b5577e05c21ffbef30444d6bc2587a37bd37da4067bbb2e2960fd5fc33553284d129db2c38d53ccd84eeb3a2c85268c2db5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485751743Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/array_metadatas, Name: checkpoints/6016/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6016/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5eda6c061b79841c3c6103127a6c493ee61c373790911a69406acde0b3604de6971a02773501e94383d503a35f1990b635049f99e2245d1cb615eb105d752ef8e011dbfd798b0119c7a99f2c404e662d88ff707d85eddf7e536ebc30a91db9cd08a69e7ace82470cd6b7a50a0fafb3b0878fc417c0808d252af67ac37e098ba10dc0885c9fcb81d6b903cca7f4bcafe63e2c1a29a8a42e51eb7b07923b20ef5258b2bf30cdce778c9285532d8429b241d0ce938697400874bfbc6f3b08402a0ffc3ff8fd5c2c914a5f35bcecae43d942a6f556d9f5dc0397e339849ca7a82cf3b79b2af29f11072cf822255b9765324de070890afeb417ecfce9459111b552f2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485777762Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0, Name: checkpoints/0/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/0/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7a490ef312c28ce8309d36ecb06aac5e0f4c146d9c347325dc105748ed829a7fdf80e2519e2764949bfe3176a535aaca8699c6402ce039cdb83d4cef76b7421ca5dc74424b790a7d0fc63c7f5b7f4f024b68ff0df69b30a96fddf7f5f14d47dfae4a174470d27dae86d16e902daca05dfa12ea6abfd4630afeac00af8e2e0d0592c8b71bb2a8180df450c5f484f8d0868ebe2cd2fa6fc8106a123cc48e2174d33f510b92f53f0666c33bf45b59072c3f5c4b1099c49a263c7d14bd8145cce4b9ef59382a2c69c1626a15bf69fccd3b4865e2fcecd41772c87b1bdec29360bd14b6bf439d65a30fcbb40aef3cd0282c17ed9f6fa0ebe50a97c8882384ad7b5255&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485784604Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/d, Name: checkpoints/4352/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4352/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8b048b18c8aba31c413eafe122d408f2e41883b977dc433b9c568a44479cbca7234cd6fcfe78bff893b60835cf4f29321e9c85d5d41d259c0f3b231c52651147835631133647194291dbd9bbbc8a9b821c86e50db311fe1711a9ecd6a92b27325cd15d426de05ded97a60959e32c54c307653a3454978b2deb8c9d527aaaacac1042a53859b03b999adc7cbf9123149b81c20eca6c8733e2d3aa438d3f271b4eb80f5eedba78b0a43d7e17260a0ac1e278bf160b63ca28ba2df8e5b04678b1f9376acacac293bb86fcd0bce634f660b57d34f8b45931b13fcb30956e7b7a2e9cf2900f3afee3e576129c90bbe8cbc1be0e23b4b914b61b4256edb9a4567f614e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485801273Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216, Name: checkpoints/1216, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1216?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=63e364acb22efaf0a1915c6e645e3881e32af8cf724cfaa5faf021e87d3f1c0d39b23a123e73f37dfd06f0fe858ab0b013821f9eaf582062034f1c68b2b6fcba25db4ec15a87105fb94784bc12bf663fb253452ee3b9827086c1db8855a3e5250299f3f27a396ea88dbbbf603b9668cd463f775a55d29a921513e14fb39647aff6f6c698e18a3adb90ae2a6b00372f99d450a8f543e6fbf5d52df9847ff0ddb8baaacb8cf67c3dce42cb7963c9885fd7b6d6faed95ecac879f0f721d4c0b10fb91ebe1e54904085041029bbdfe4e8234a408cf3be9dde78598f3beb7e6de37065ae9e8d7f22af0327c1b3e155f7c5ba1fc18acaaa954f92190f7be1a778db638&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482945714Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0/d, Name: checkpoints/5824/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5824/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=621aff808282347e2cbd6ce9954eca19c9e35678e7563f6b0ff4c3f76bc86f330ca2bd653fdb2847a002c2447da29c7df287f2c0e22f541daba70ec068f8bd8ce1bf490ca1409a26e836ab56579d24e7198ebeed45eb79f6d483da660161f9e456d6c960916e22fc588ee5e5a9129d786dfb2c64b2e244504cee2971adf1d5530a6038b1309b904bd1ffb9a9a98966116691d2bb8f8b175b9c00880acae4b6f1cb6bf054ab82ea1797d738a6be01fecda8c87a294876aaf0425f59f47414c7674ea30aee827ea68fc619bf224e74e7ac71f0c541a35dfd8b75bda56a3d8e5c67e50934bb9ff270b1d08ac1b57d7ad8f1b5b0bb9bcad64b63397dc404337281f1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48587814Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default, Name: checkpoints/1856/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1856/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=91b708ba89d82127191c762445769fc80fc7af4d39a505e66bb9a3bfe3b4df158a16c9ada856d0ad24ff1af1cc6ada6fab47fb07db6d9068c9632af4836b051d9534f4a3a9cc6425f014502b747c43f3967c9043c743dd74c6228e686f83ff3ca71196dbd4f0e46ba42f0ec3d34a85efd80f14e15e8312c995f02f92b6327c14f40597452c69054bfe9732b59bbaa53562baf51813fb1c6653bfa82d958daad561ce0b3744aa956ac45d829915991b74f0bdcfeb6cda4f2e63bd60906b4a1a34698bd968bf686f8fff5de014b68cc779a74a76c1e88b37702141a8762842e5c652bad07046d75885d1c0d01e6974c8e8807159a28067f6b68998c51b21b82201&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485850499Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256, Name: checkpoints/256, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/256?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3ba6505e17267433eb1b2780d60f31288cbebfad0366cb51971b3c717d1978f030d9830f2127620542ab9f3eff69d13237b309657a8f3394058aaba2e402c663afd0cb7f3afb27151f6340b250bca50794a25714c9ba946f1ec9dca82d81719b57f208bc864510d2f1f151feefc3a3f2b0a9c533a153485611164af04340b18cb14fa77738086545d89fd38dd4fed3947ec35b907469e8cedae2a0b3b79057e73c15328b1fbeca3824cf8caeb220d86a6e99b220257f24f389233909158bd38611111d0e7ce5aac20ef2a7759f78a6ea8ec950812042f5f3db84e0e93fcf9c6425326fb69afdb5e51cbc92c213c17b03af0dccca10e7b046b506578993ad9f88&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485936167Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0/d, Name: checkpoints/5376/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5376/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=99c21eae9d2ad0e4e028d90d9518923ab072dc7ba2cc60ca36a2ece22c90c2df8609faf529363249c335c1da786f2fb9a769f7460f9df0ce92b1ae31893736b3d94deb1bbf8ad94cadb8cf1f4188cfb7e800b354ecd7d86b934d41e279ce1df99a0096b940d892dfe6b4d9dad620fe5df24cec985591702e66070d3d124baee670229fcf33106bce38dbcbda89070cecdc0c4965d0885963864d99e51a2c286b076c4e4dcc37df2559dceaa186df1a2c99d791fb14102dc4e1ff8166cc6d328691d3a44700ce9eedf4665b02cbef9b3dca106e00ab384c08fbed07c0de71804aa3d58cb88bddfb90ebba91a5a6142fb1a4296bbc44079c0ee8e8f2443d750b72&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485942032Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096, Name: checkpoints/4096, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4096?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5ae73d2cd67cab0525f82a48f8c53d8550273e4fb8a4121c79b07e26ac71342d50ca5de9197f75d10cef4fe10ab7790ff8c450e177b20b66588a276001c051d6d0350e4a34551eb29603c85b844501823d17de8efe18f6a8bed180a63af5b540341e578e6f5a584e528d30a2b96dfc15139e7fbe1fdade13b108eb36a0aca886d670d04e98879842aa694dbcf19e4709013909593be945535a2c987550fbb3f8d754f8ce3619e5ef67f72a93beae510142a756cdb3d3a835c2da2314baed87772c0d71445737ab0165a12a69cd5382ef5d1bca0e8401158233df14fd2668cac99391572e8fa0359e3a6c786ee29f1bcbe8c477b9463f7ea06afc7aa9dceff8d5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485957302Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default, Name: checkpoints/4160/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4160/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bf5e271f7721abbc6e3cd8aabc8abba2acff81c2e449956159d2a6c1489cc21c21598d053aa60046aaa6c65bb92db683c89a873bb421b4237a1b13081be72a6765f4ac1e5dc2f34fd7a237a3931b454137bc7a65b5966a18aa9c3121b31e3a9176f88891b6307c66243c6e1c5c6052cc6bc8fbb4528a29330281e5f2fdce215f362be2e60683412732e924e62e7868a6075c98de6f49552070cadfd081477b4841f2731c4fe324b92f6a28ee0e2578822371bcbfe81cc2b50d44d20fb7bddd1817d1c521c0b6673f732771a781bcf8296804e8d04217dacc408cfb2fccb21646f42a284d9d630a311749637c8ed25a25f7bd93b9f7184d9be7d11ef0fea0b246&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482946371Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0, Name: checkpoints/1216/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1216/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b333681c56d9cd772cbd2da37a9b4e029a1fbb014e3982368610f2c57acdb28993f481625bfa651bb153b5280690e5f22e0927b3b49cefe7e3775806dfcc3e5e512fb486a9c195e5879dfca7e7f779c7e3096b76f8ed56c04e2030a7397d330675a11c02f3b789825ab4dfc4ad6924b4b62c0f06189b29481681c891f59c443196d127072b227461369adfcee2bbfe299c8c27dc3d8e62cb08e830eb8f487b2f1ec87f23d58faa634b283916b2e6342ef7995a8fec87a413ebd0436d223d8632b1272fed681b967c013595025b34e960d6c43167e6f23379a40efbe2e73f074110be0d380c9abf1f56dd6f5bcd5613227434b258c17177b0b9f431b0a71a5063&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486024935Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0, Name: checkpoints/5184/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5184/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=49dd5a7df46e32df408fd8ac8ccbbfb49518ff40d4bc906b711ab07f2081213c775213bb7605206f54f2530671f6456a4098f83d0ef555d9f1adc891874d1e9b4dcd8b5aec2afc79a6674eaca7af730f153425e8fa787f6eaece4fb61ea3125617cb3d77981b9e06b2f46330129041400ff0e6e915cfe80a4f5ef04f6806a62c698643324eada89de131b837376dae1852f1ef5737964c8cb0d6c91a51e24ec3dff62784254f57ec4d4eab05260e4d17fb7927d1d740be3831642ea8dd6b696ec6587659977d23fa2894c9a9e388016783f2618f8dfecf5fab8a71ebdce4c75a7a3a8a830b09ec2e419ac72d15f3eddf375d1e97f2a0edd39f8f598ddcf8d88f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486041371Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0/d, Name: checkpoints/5248/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5248/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c0c802805dfd84fd1e0c7369fa1e169a38208771b14fa122e9badd0ffe5abc95d239048fa21b25e65ba5213c68b39be01ae6f3ff3d91203923765456933526cce4a42e64ce608ee6b7d31d36e89c66d35b24e3349d938e63f8bce82218976efe7770fc04f1d1bc953fb000e1b95fa591674eb6ebdd49214c733f66f4d47f76505c6c174710664781343fef4f565620d44a52bc3db554d1909ece8323cb3bfc825d636659eb40598182249dd20bf56213a1c33f2181ecc1d8c8cd1a047c6f131bebf00144a9d34e20748ddd152320ab252471b1b73369e373da92b6e0201b9b0835d08b0ca44d1f18cac1b370020a5df865f374056574f2696744cc4793197fe3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482948944Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/d, Name: checkpoints/384/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/384/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=731c66a3613dc4aa4c5887397ad2398096c78703c470fcda148e1673ebb5efc1bea7895e726a5fac317dca8542c9c74f57dd12eac9a4e7d6d5f3473ee1276b864a5a8b3a80c60992750e1ec156c67d5d83ec2a9a10e0a4f22f205a27debe90d37642a0f13e681ef6d9e7d6daee685bc62f1af8fbf322f1bc31f1e0af098f149c87e56b418579aa5f2d338d65843d7de30918e10a03af1ba98315b6c33e69929a9df88b4d973806f9183f84d8b371eb6085c701fff3e487daaa502cf125ab474a4ebc09d3863c606181005ee0b6e83c7c4eab172fb1c572403091d0077360fcd41d1a3157a0fff379ca2628d867091440410d94966f9302573874a553027a2639&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486109136Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504, Name: checkpoints/5504, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5504?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b41733a329307d90fede2642db409e8e15bef0892874b23bc687703d97800b5aba4ef620e8e891c6565212e89257714690dc144991f4baeeef9d3e5f685aa2cd131c7148f09cfd870d1e8666af94715a628869b99858c18217e6efe1891d38251ce9a9aa0dbb2ab0a365663c5d87bf0db922dc818430926050862f1aafe44ce2c86e39afa6e365976ce1c7aef658a020c52a8b2fc93fe6a1f40cfc83def8b4e7e5b624fc48baab631b8be76d326fbbfd5770922bc14bc651f40c6eecc810a752c8bad3a37e1c439e9aaa8566952e335a41cfe2ea661ba0c62fa57a8aa40226f3f7bed61223132e0c412169fb1e1d61c5d1e95efb952c37b194d4edf149db7003&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486124947Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0/d, Name: checkpoints/704/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/704/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5ccce54655e0a20a1192756244a616419434855c20167f51ed324d4b5e9e3a38218fe6520842d3b5436f2e8068b5e64e1a969d0867a38bdd3b671895c4a150e866ea4e00d8e7081044a4907f9f5db6a6eb847ad4fd89e2518fbdaf867bbe3a83274dc23e16f1675315271ec3c6b1d5dd47f455398fbb6eaf0965f04660dcdc348237940b067c269b00c65a885ea9daddafb523aed9e2293d6d057d170113af3d8b3998ec41414142df7975a3ea6b36b0833a250cf147b981f4ff673509956941253adacf0f7de4191e8e4e262ef57c7a8b5822090248076a1ae73fe298e20da51e912381baff7001c76522c2600ec9e35464ab3f9b83b52fdbb6840890a652f7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486154761Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/ocdbt.process_0, Name: checkpoints/3136/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3136/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a71ff76cf6f6eae008ee90804bdadb16583bd117660542f5be2b4dbf1733f758142c9e34afaf0e9eb07e7be3ab8a6731b87376f2eedbb87b3519567d68f84b32733d22916de031d71f195a277de54d80c64c4a59cfda54f225921059f2246843e2514b8bedd77e95c44cf33c03ec1e0b9dd0767b127061082167f7d6d9fbd439905c3b7e98563971802fa9c6e3692b78d118a3a2eac534ada345b5ff7d54397586b55893b6c2ae23067f353fea79bb7263c082a3560875e42f90bc90b9c421862734f218b73291f3508f0d4e00506bbc980a2f0bb081daed89bede375931041f04148efe04db3c700abcba9147f77231b858d1768a79c6d7b3887a9d16a69322&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486225099Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default, Name: checkpoints/2048/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2048/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8020e4adf1d43905623353cde02fdc33dad9c65fc248e017275229d018a5e57cf5b4a07fca210f9a470516b148f88c6d1cd4aac5852f5fcedeb97302a8cf7771d4ee970d6a5e66d3a2b7ceb236b5aa786f97490472c124f9e11571a552be320d2ef55823767a75e7093663384f018a1da901c4801166780dd8345e63b7459acc44b43c8f8ff517b801c577a67be371e48be98e783f4be5c71809fd6d583f803f49f1dee1cbb14af2b4ecd82d83b319f9d7154d7e280a25c77fba8526246bb05fcbd72e2be14afa4d74f100f93a27e97165451cdaf231f6b4b0951c56941a4f370fe68a95f57f72fdac34ae6a48bbb278c8a2600bd379d4c0e2a91bc35067afc1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486219791Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0, Name: checkpoints/4800/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4800/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5256f8ba8e1a26907915f59b33aef4b7f79c4c3ab01684f13e791cf357a26bdc1786fabfc2f949c0f785b83aa56f2fb19570b0a38a7d4485141445d6db72182919971e23f201fd4ad497d6f5b2441048dc5052a35add47b5dada669f7987e0c6202f9283f84f33851f8fc0ac54f56a79b9ed051ad009cf35c635b9044d8102262f9f9b70d82678b55e69c54a937e8f5978d3bf24c912ff3a28b4e70b7a5c5726aa20c2237e647bb2af7662a4295180d4895c48f9f28d3bacfbd7326c63d3111a5aca063f806153ca1e82cccfd6c3f08cdcc74be93b7cab20846a18e69aac39e82ddf9483e21652bf2a4edc9506f1b168fe3af832418484b4d620365b1efd94c8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486260968Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520, Name: checkpoints/3520, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3520?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4c31cce9a7f040fab5faeb59cad3881e11d0bddf9de0a7a2c36cfdfadc38780dafc06c41aace2fe58b610e01e709f11875d4fef9e3a0d96608c11b108ff3883e08b73ceeb32b64e37f311a05e009ec8daef7c80c07a9be0363477cec8874d6beb5a31288587ac677de084e6f5e0b201fff583c150238d50e78a69ecebe775d785e6d145ca1b2c4c755e7b1b6618870d0de8d7cf468cd397b75530ae69a354e90bda7f1760404fea3cf8c9abf250053004a3e72e2075e6f1f67d5a47d62a5730d15c9fa7926f1227e6c68be5d7a82114d45f60cde3d561e95b0ae65e751e2e2c4d3971d4d0ce0d6122104417bc85b60ce19d18b1075e93c8d65fdde8a470882d3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486276005Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/d, Name: checkpoints/5184/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5184/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c144c15751330c111b0409a1e8c69262bccb867cdbe8b73fed4d9ba2fd4dcdde3d8f0d2ac48b009f6ecd546ffd0a094a23a04a93977a8d094d29c4ab34b4cc4aac68c72e70828a69c20b7ffa9bd4fbec77f8ae93325a01b2b4e451c6572074d807c5d695adbc1479b3df183e556bb38d77cb65852f551d7dd173e0e2999f688afd3f860808e293488a67ddf0b21252fe5ede16061febd3e217500509c50403c1b17087ceb70380a3bf57738dd3a9a73d5480df38b79e6dd80a6f2cda4c665e46cb6919b61727902bbaae915427a3844f418e6a3c4a046ba5ab3c5a451c967d503f79db875494a4515a3b3f644e1ebae0b93ce01a61e233dc3f6771d828e31651&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486299419Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default, Name: checkpoints/3904/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3904/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=69fab2c198927ee0c74eb4f0faf39c73701a40af4cacc9f48254cb90b6f84105fcd0f8ca1d88d571e0df1cf66604229e72fa764d3f9e3f93f6f1fb51b2367ee23f28c492b18b91c54f87d355e2000596ef6cf96cf6bf333ec1d80e8cd828996adf3719056187746ccb82a32b7f353b277e464d1832e492fc3daf1e597f2ddbd02e453b5dc9db48c2cd239d55cd7f165d227298a1daa162effceac3a86f0e5d40e9cb97309e5a121bd8e19470801d6ea6232f4c51d5720c77ea5dd43bc2ed066ad5657f96ee84eb58c5f3e2e5ed5818048eb1cbd846193847b6ef26024cba4b7b71888a1bdc7cb8c324ab83f55a810000f51e3d8d077602267489740b5e2f9a47&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48632261Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800, Name: checkpoints/4800, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4800?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=97d1b6a446819233130a48aa9e307f6c282df972178c9995c3a1c089eb4f35bd1056d526da1e2f87506c03a8de0116d2cd24501e18e3c2e8caeaeeb9557aa561a6f31cce0c763f97cc3973e7ab8096bbfc2e113a630baa82d200b50d08d14f8bda85b2419a067354f5d769b28149e3b73629cbf0ceea5c64d5be45800e631068b99a206d2629018f2cb3dbf2734487bb6c2938ac80275de12dd19c1f332862f2e31ee48bac3618f5149f05322bc57e950056a7575557bd793f2d6861d3d1e3e0115282f16a3961f40e206b06f8736734dfacc6e5567c3121455441b0510c0f18143d16ce4ad76d606089c82e46e219ae82da95c6eee99a3cce46265006382921&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486370566Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0, Name: checkpoints/1472/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1472/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bab4559bc39e5b9dea606e3b3a90acb4d93dbaa5ed36a0db1c670aed4a3a16f875ffb096cfbd4f1b5cf263db430ecf8e71448a2bd494d3661680e85ae657687488b6561deed85f6d11576034531835ca53a99444d9b4754a6ce9f22297e56df71f0b0ad64158a8e960ea8499cf17532dd8be8d3aca4f09892d2b280bf0833052415c86e7a1586ae012354cf4dfe494969eca70d7f07573dd221448af0bbf8d43eb58f41fb351253034caa274cdae4a26c1616bd2c39e1e6b46e980000a32d146d71524963f9153201ff2da184b1aee9dc1d296b57f4bd103128ea9d52e77de89eff378138d2a52573f16d05e93554ae51be85ada3d54e06187330f9554932e23&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486396177Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768, Name: checkpoints/768, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/768?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=66c415c8a561da28eb9036e5cfe33ee79704d1c13ea0c8e8bddcd9925a219223c1484c7196ee9e5095ee1429623bad83886d2b0b8ee5a46cd992d6b48a841b52ac13eeb392ab22ccbad6df681e176c488fd10fbb6e6ac33f6daefa6cbc786f4fc6daf904eeb90a894537df9db4a26ffb7651e5f331d87417bce4eb3abc14491f47619e29c6051dbb9536faa74788924135bc0f6d46f8527abf60d984706da2b86297e099fd1bdfb8cc12e263432a622f6eeae71236e4e921d558f8f854fa99c2efe53e7bbb3972c21fd76544cf88f82f32cc929fcff74651d50fcee704197059e2a7820521ba6c377865ddba7406e8471e94d37e2739d4699436f435a3bc5699&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486391336Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/d, Name: checkpoints/2496/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2496/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=45be1d98c3d37f621f03f47a99692d62e9a2e3e8167db9aad510b5893a4d23be181d5a89816d86a09f360170a0008a29d15e8bfbeb284c6e942c3aa87b47704f83428176d5f920dcaf40476339f138549e0ece4ed87ca09df12766f1c5ed81550191a7b1e1625fb136982127d629ef6023974ba2eca10ee0b7f700a57098dd74d443ceda6a3f1ab8aa6f796728fc76de90cab07f68e5a4cdfa10fa6b69c0b0e4df9d27bd28d53e49f2aa99530a4f92ced4368340c7ad8bb6eef1b7c4ca633bd06b50fef28298b3e11038d36137abc2bd3842eb57559d6a8f8127eb282effdc751a898f2bf60e26e030cbab0bebb597c5535ed0d6d10cb74dfeb9ff66341feff1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486434822Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default, Name: checkpoints/448/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/448/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b9c165d3e9b88405c9d5f69ec5f9c9aeeca0be5778310de41ccc90abb2d656cf8d331337ed9ee059d0c5bb47aa6b5fdbf5cff823b4d8a5203270f589764ce0ccd6d0961a254bf7ea08c8656f99aa7701e4aba8e51a95cb8ffe4da0d87198255dc37681feba1358cfddd5088d02fda74df1fcb79216b57ac50a11ec197bd03b68c1cc597d9b3bcd9a878e90610dfdf09365a79f92b654de3e3edc5c5993467e6eaed8d8052139be63271723c7846e20015690e3b72ddcc22d5922c0c8597adf13767e69ab9fb984e38479b1725ec45e649ef751de425005b15b584a7d4c7d06c84624856052733227c02c2f6d542604863aea2935490b68b49e5578596fea1a49&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486482052Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default, Name: checkpoints/192/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/192/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=15031f6b49479a8a384ac87d9395d0bb4dab5517880a5d57afa2b6dbc78d84f82292d341cab29ce81dc66fbe4a7b85e52e28ed7374065809caf1eb4a79663eca549934986f7318caca647e78003b3444a40080eece3bfa39314cc80fb221bbafb9c4949439a3bab3d95c6c2a5e16669dcc3b50450af51ffabbf991445a01432578ed42de12f526ba01d14a03827ce869017092fc824d078091a33fad443eff66b8dea615f0d4ab5294955dabe9ee40ea52a26994ef0d68230ec60596023cc646c30844c1269ee6cb5159097a8519fbaa0b1f6fba64ca0d9e389564cbe8a4175c0b192ce538f89c78909e9b421f0bfbbc318841fe5294d29575ffd6f4b8fb2e15&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486526079Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/array_metadatas, Name: checkpoints/2432/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2432/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b74286ed5c2ee0dfd019976e40804b27947479eb6ee2de6c8b3f6a542a2f2aa0e0d35759787334b30f02c636ec96b73a130f0ae5a5b3a64ab8f63e8cacd97289c450bf75200f37bbc39cb4670ea5b91744f2a6c8ba69532e533019806251ef412b122dd45b1b3b4df62aa944892d4a083abde3cd534fcd6dc236c6523c9f69eb873dc99e64653dda98b35434091b7482515fd94fb121d7651027da54fd7a088abfbf13fc39eb28febb90f33e73bdab1502543ed7a6538a7d623b02d178c24c42ddc481811d43d4afecaa472420ac984deb61797cc68647abcf0df0568b49d688b0e8be707494df5cb6f51c6a23021da328f53b053e2d9685dff9dbe8ff8647a0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486542146Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0, Name: checkpoints/4992/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4992/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=924ef8a69ab99fa1579bccf70e44fc3443bfbf03ad59b71d85324dfee6c423a24dfb5e95c5f8a985f6d3d30fa90d63f891db4042590e0f1e76b540af6fcd4532369183c8065189c16405e5338935976b2b7bddef670085f3d9dd9de255ac6a776e5544a440eb5c4f8d33ea32d76851a107ad3eb52442ae6079a671c6d3429b6c3cef720e09ca46564dc8902fa5c63cf44bc2b9d5313d531f03cec5e761100b8a73e5fde82efaa97d981da1a5af617fb41bbfe50c7926d4a576238894dab270f3cd5a6302a94c8744940ccefdb26ebd96c15abc52ae4b227fbdf1cddb4daa8e7aad708ecf7a3809c6999a19c2046f70d248e486924fba06de6976dd4cd1ec7ec0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483000205Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/array_metadatas, Name: checkpoints/256/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/256/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6a61f0f85260c6617e8ab6c402d5771811b9eb0e8017cba46101ba778cfe7b22c3e1363cdcbbace364cad798d04b18030f3aa3e6da6f09696dd4cce4cbfc21dc8afb7ebc4d67f70c84690bd94611cb017e47a823e01aaff8e4301a53deb87c9bfbdb9bccb26d6b7db86c2efb3f731814d7f5daa0ba4140a7530c66bc427cf330c164842f8de9e09720cac6d28aeaf79b532093146c1c67e878b0fdb38eb20618011bab807ce0bd0a54198ba3c5429f8902859c8e54fd52b1d4e282c0da3ed701890b734f86a93527f0c5b43a8808a58a8bed2771cc13df0cf6402698b4ed4c9da6ce83c7f305046a5eac6f9f3c5b7fc93c0ac6e768b78dd475a0bebc1d08bdb1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486370782Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/array_metadatas, Name: checkpoints/2624/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2624/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9de26c81b2248aaa328a81b49eae47365d1e11bd1fd7947e81c1ddd16b5eb9cd1375d2d4778966adf3553e5f72720c4481a9208a31b78193186690dc940ccd640a762f865fab6dee728add00c235cafe7d95f8ebd88760a2dcc2b0fe286acde8248d1bb1a7107fd3f5cb75a0c9c0adf4ee96c2cb023c554a7da17678b7a2d535de76815e663c87e269b11f49b8dc9e025ef5484a3273e10b3e07896a0b016f41048314be4c83ad1271c15bfc4e2e76aa24b217ac5583653b65ca3c33271da1b26bc839fe2119be04a951f623ef3b99f52eecb35b2c61764e4bd77fa012007ec8984b89495fafaf6ace2ba8f059ca32b669c5df6c84362d0df30d9eee473dbccd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48655264Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default, Name: checkpoints/2944/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2944/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8eaf6cb1a9ef5e57ba1ac0294e48bb33530d0a984a526c020348fc0cc73d446b9a478c284269d21d61fe1ccf93a1ddae6b99f4963840ea4f4771907daa0736760b9f79d7049fe42e4683edfb8bd5ec43b2edb1c432c982541cd419689a71063acb179a4641ef3b28008e48a4fb948d578fe266f0eb79dcec01ce64a7502be04503577965eab3301ad87c0a9dbbbb94491f25d4679bc05501ed304ab744cd0e0fab644e4d744fe448341a2957ba846d1d712fe58608500ee90452260f60728c0505a5091a262e19667613bb4fe0bc7af815fcb1d8ed87894660567cdca8fec92b91346e431d6ac0d96ab826e34c5465a27b585d987d0ae4c5a2af799d5d2a20a0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486552637Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/ocdbt.process_0, Name: checkpoints/4352/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4352/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2d17e5bd965107d1e27c45cf81a89f5a501ae13bf7972ce78f7fffbe506e34d86e18650c077ea4a41c6689a0800881ee1c0b82e2d51e9e658ea56017039166f067b5dc0efdd3e1efe9fd2baf493136978493e9ab88437c6717546529322183bf16b99385a8dce1edfacca5397cbef3eca32ecf6d31aacd9ea76fafcba202485f702eabc76003ba97cc63d29738651c58b1d915d67a970b1fa8c7b6d95311ad37c40616fde620bbc0cc2115a8d0f757dacc6373a77b17aa40e9b61a89c9f5d5c15b46f8f9296dc72a35d1038474579e559d511221c6722b47dd587e4612353fabc30f12c16cd24aa85a3171c726b9bd8a18483a021d405c722f4297086ab77d5d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486700615Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/array_metadatas, Name: checkpoints/2240/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2240/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=10ee6aa16a557cf092b93484dabc6b428a30716c3729f75cfcdd673c17241720bff76b0230a0f6967a8fc04cbf272942cd1d31914728fcebbaf26c6f35388ea5e832170fab2f3ac5de61fb4cabe34128da393efcb2f13c160a3e190b5760b1be3399195d7f20c6a89e4c05130994296c3144343dcee5a035e60efbd5cdb3020033ea4e7ae3cc982e199a33cd983c828b16c0ddcf5e71cd52b3ba4d597301071ba50a72fbbfbe268d50c19b100619a27a37bbe24130f240c0ca48bb793dfc89b5bd43de5540152f00471978516f4e60fb61af5e19a1ba7f04f60680e69fa234ecf46ca99a2831f9d3beb2780e45ef27208d570a1db314dde072ab68924bc4d762&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48671644Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/array_metadatas, Name: checkpoints/4736/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4736/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8eb72a7f01ed04f9e0e26355fe27d41bc009d6369da7dc62b110b393f828222c6c3819711207d44020e83d15d08b74bd85546fcf7f8098e1ab71d501b6ba9c76b9ab95a452917130a4a23a3e8084c140c7497ede48c64645c75df60f18229f7a185c74f2627fb9c91d795f097deb1e5f4c482921ca8f10d15e2e61c0d4600dd8cc5ba6befe25daf45e254beaec5b9e590879fd33f318b9dc72c09bde0466507bb1d4b4b0e3aa5294497d701c5ea4e7de525b99233fbc702947b1383150c3341d8979dd9a9ecaadddbb120c3865b245110ea57058a7694009266b3416fd2d29517725371a3005e11924fd23bd9493d3e39c60b60a61f3ebd56c2c66cac5e3486c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486728324Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/array_metadatas, Name: checkpoints/832/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/832/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3b4221edc117fe629cb651908d8a10a2addadd0ecc37c0b11b030e014e99d50e65bdd4f779e44ade36bf9cf3cb075c558b47780c881230746e6af924ae07f4ffb865a4aad46762db5d0fa235fe26632ac4d9ee8f64e8a72cb5d8913db687e21c35be67e7c5109583b92254472d460f387a67cba79434311c81fac55990b06fd7a25b92fa9e3b0f7fcf8068f069461f5cfa2873cb1252e6477d55de868cd833d3138182aeb0250ea8e295f594f98147223367722b25ba10b11c4a6fa73f4da8d361d826395eb7623427baca0a59ee873240c45b6102f183bd7d168bce03c71a7d33bc497d9ab34145fb3115f0f07f9024c4a15c49aa53641fe26808c81037f57f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486770726Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0, Name: checkpoints/64/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/64/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=51b0c94fd01ea34f0eb884937bc835a7ba060b799469decedd5feb1162f29c80b3d79183e83fede2eee2956952913126f6b0b56973bb382410c15b21e37f71d347cdfb4a03cdd45a52a7e0a2dc02efa9690005f4488e3935e0d3f517158db1caf08e1aabb4c445388d7dff4b866c9528c77b4c7fe97dade898c187dd96a40498397ddc4a0a661a232d440071338ff3793d69a9ff2976b2dbf8cf412b153453d965317537933992598638605238ccfe01916b6826b0fe4322b22d422de54fa69650462003e3b7335137db46187eee0dea524a8634e28337cad6915d481b3cf073da5a1023e64468740b4cf38b263c4dc6dafc7d6362282ae38357accbae9c20be&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51590501Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376, Name: checkpoints/5376, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5376?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c30275ffb845251b3adbd1cc6c22ddc81d867d2bfa0f639930a52f7794037afa084e316aa065b4e8e4744119ce91764932edff40041341182d044f2334da59d4c0558193edf98f1c2e9a7b99ebe02a1be43f0c76e2b33b58b69994f8105de0cbcd40816507b0b05f9a0667a5e2045b3a21ece08ff61f01b495aa6b5ddc618c4b43aaeee862ec4a0f6c475d7993391ec1579f5e61c8976404fc5746884dc555e45b2434301fc8da6910077934e38972a94657de4dfea3d39983274dc2d01273150a8c3da446c1944f211573e592d8d77947c75e69f930f8a29566c67d4d3ff27ee7ebca8d11525f1369db3d91430b335b4d5ec0af781393181a5f9b84a93073fc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.515928364Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0/d, Name: checkpoints/6016/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6016/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=94ba89947e051320a155cc3086fe945057ed28ed49e31cbaa5bb1dea1a8a93a8e80106e4b160e514b877882a0927d857d6aea1654ddff92c3df2323d30356570749c854b329db05e8a0d053639d6c3b6459e059602daf63a75574b133798c9d8cd13674440ccfd45117bc98773ad59c65b607daa4cbb09e869e8518977c720634c8d9288e2974eb91f0463f25dc3e6c86e3f6e35c41551d3e72dc78f6203d79572f08bb0d81c1e5964bd70efc40cfe13a41af73fe3b20ed5093153d95c4786d99b5dc4a3073d58e47197b499ba4d7ea120a372db1fdcfaed3f41130b8cb88d1463b724ceddbfec565cd26aece8ec6bc6cbb4535dff3c3e687073c7d8b9e2519e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486717068Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0, Name: checkpoints/3968/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3968/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2f19bd402d46a6c3c85736e1da0c5a6b799a2d19b2a35733f6bc65314885903abc3e5ab51da88bce35e202d0559ea558dd90a328bdcf6143752ab5de312304116df7133d506f8639952f5a209c49ad9ef7cbd1e3f95c5cdb2daea13168e53d2efb1a3b24acbc70cca60203dd044d53d9b1608f9e04dad8846c222e93bde522cbcaba16c51fb081d2637e287fa78055063fab565ddf2cc46a3bce3ed4ac58e50f4493aa465806051c7490d91bda47f67e4f9c5ea528563e3945409ddef60a6505f4370252e8740bce167a1f1fba98bcb5d54f181daa3311fe734cf7121178857bcb404d775291d60c0ae1e32b0e1a5ccdb464ed6a00437cd10881cde61f937a16&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516114933Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default, Name: checkpoints/3968/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3968/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1e07b76cb07ac8dfc8ad7fcca72645ef7625055358a7e1ec6cffb745f5f6bb2153da30a63523e8e64305242d53d33436ef965f63027b0d586eba076075068fd44bfeb0b4cce3416583527ed5d7682c9386cd0d5276a5fd49d6b41ae894eef8a42a4562f6c1f8584913b99ce956af5b65416645dc52c3fbb4fc6b8fc139379722d4338c6e79bc82eb0d87c742aefa15b0db4dc0195419f5e3fe93bda2a320407d658e8dbbdf0bc3a6570648649447a52da4d42b7ce4ca6349eb68e9b490068571b242e78f5826c1eb2c93b607415aaaa9c9a13ae5b7386a6e744cef8c321c31236d2b01c29ee941f6b0322e815cbe91ddca3c9266b4aadb4e75faf6ae742a91d2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486760921Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728, Name: checkpoints/1728, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1728?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b9243b9cf5205a1e9b2af4110859aa3c8aad94243dce4356d868db468642efb3ff9bb823592b9a5573a03fdc1d0261088b1d9af82fba2d01d0546790c90c541136b08a64506e8c2faceb8663a5a922f9b28a1e4bbb17ebed5cb40fa5ed3ecd0b6e333aa7a4a70db8c323c94ed3e78b9aee4ebbd403e5c48fd1f5699d4c81b06b28042a0ba3c0f292d6debc6d8094f84fc777250e4ab2dad30ad624b274bd763e49f0da851e6a41d43966052e29d22f2567e2198b6730d9b45fcd0127d18f10e82a02083e1b2f2fc65f6bd19f1b4833cddcf00c5d3f3d6cc92927bc2c3396df34b3fb4a11cf7beaa74220965ff5dfa4c48c88d81000d4158a752c50354d24f560&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.484041045Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0/d, Name: checkpoints/1088/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1088/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4df33941bc6a458d28475dde6913841dd2193c7de652c5819b3580e2b6fc40688dba9ab0e835ee5d744acf31029f1a03e706dc1cac25fda60e47357c096a29d3406f1c5dfd58b9df2678132862bc49915a4c445bdbfb3661b51c4fa6298425a51513e48828a8e2e3ef62b084311d4b7535b20062e8b6cc670e0ae29f0fa6e532eeeda8c2a2d9da197a020027e38d906699bf87eccec7e8eb02ea1d0f5f2959e672e1f59607573f1b6b13f08dd1f263ec2d31bacfaf1cd098a7715edebef18531c8603b24ba3f7cb3ccddfec735a75e7155976af8c01a1ad1dca01ead807b20a085d05e3f5ea9dcb7cd8e3441d6066716efe2d7974295bfc5b3994e5fda773c17&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486300587Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/ocdbt.process_0, Name: checkpoints/5568/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5568/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=17959f04218def508f3406ec2a14859ad999aa6ca4fa63feafbf479dcd003701f500647568978301bc896aaf1ca2968502717a6d8cc373d4bbecb59e9e54c543ecd22e668752e62cfdb26323a8abbd195b9627f2dcabe81f0c584322b0f1f8fc41234ca48eb6ef151aa744fb7e18a390d2e10c4a3b52c007dd4b5e06da400b7bfc5d4853ab538cdf18c405aed31699421c25a4a6df03c5bd1090eb3c4651e090a86e3f32ff4c1d29c82f4bbda96a21516f9a7b12115698b572d1dea81ed5befaed9d735f8d07b17844ceb8f01fc002364533935f14ea003a3cb6204c4d908ad253ce097b56ae979aadaeec6b9de6e5057807f5d9093856eff3439185dbc021f4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486696422Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/array_metadatas, Name: checkpoints/1280/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1280/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9513af39cc0a22d446dedce010d1b165623e05f4364e818c020ed99de578e6e4d9107c4029c6e0b87259c3ff658b432b7cf61b7ca86252ce44b2562652df6f9cfefcfdb15f983258a591bda6a186a0f4967465d33889dfd84ba6ccec035b1b628b4a6c159e2e3f6a575787d31ba26d29b2f834791067364ef7e99a0224ee36ac8fbcb149f2655e3724d81647d6775a86eb150b91e92524651fbdc250ff4b8e8a0fdce06482a589e7c46255a3673628f907f8ce73d0544b0f4884390583293bab24e1abbbe2cef5620faa95e2c7dba19b980a65a9d06119da66b8e214f9eadf3b5576ebc1f973d5f07822b1b914e92efb887bc1f5c9d552e5844d6f41accface4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483021963Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default, Name: checkpoints/4800/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4800/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2f1e695460c29b2dcb3c48e024d4a05394e305ba1855323171b856f0136e9377891cec9f3f27ed68a8a8976488e32839f2c6234785e85ed7a590be3acb5ea2e17ef962d8ed7d5a2b80ba43f09054dc6807ad14301bc2516ab5a37c4f271254faa418cbf1fa036e67d1a004b62d6c25567833ce8bcc711236aa462f7259e172655bf81bfcc512c1f8d38722e36684a002be6a96402c71a8c74562a807c13430050140df306ef2af57bc7d6474fcfbf84467508743725c1686f80a677b4de519185f81bffe30449cd871766d1a5fcbe3135f6e9e75ddfffc6a252d760da1c0586155667f97be8c1b6440c92e0cd5fc2dcf098d713304ebfe0920c075e73433feaa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482875339Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240, Name: checkpoints/2240, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2240?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b4ad006d7c2d8ad2179ad28997af0b6dddbbbdb749a85917c4ac88274dc24ce95591d5f576c761ce4d62eb10a03ea6fcc1237dc96a9294e9d089713a9fd293926b6a4b407fdb208dbf22f94553db7737c381031c98195d07d6c628401861d28f4e28e70eb5e0ab0d48bb59a10b83dfd60d41195e5943d0159af7baaaa647d2b83f73958a8ebcaa5a2e19f2e88bb308cc2c9d5c972c50c96296a244deba9474e776fff1bb489e759fd794522afea4102ac2e06964e7a0e32972a1ab0c40a0a7d624224f1451d657f8cf0a614aad674203dc8d8897e45a2ff39de628089cbe5b0574a7d0afe537133acbc55774e07716963b7ec7be9c8224315fca8684fce69ff2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.4830336Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/d, Name: checkpoints/1216/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1216/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=39fc44a8dbf48711fc447dde21ab82e8736d90e61edb5d630cfe4f52d9342883478c23131eaade4a8685609237ac0edc045e5137e2fbeae773c0433b41e2fc636b15affc1510d322dac567bb70e7a47702b7a13ede0b0e10b8d55d45a76f7477e0e0c6ea913aa53b4ff79d72086d7880b845ccea3cc6939152ef538654406186c34fa511fc3a99b2b5da5214890da0e527a2acf1ae7e35073c37cc58fde0ca3d2e12954fdce0fce7ca4386c07e8477f6ccb32e4f42d345ed72637a1594b3a1dcbd403ba462053f4c929612f9d1f2e60b36d16b3d948acafba2a471485853f5ed0fe9e0894e5d56a376fade07dd3e3c6af408c8eb1681edfde74b881e8424adf5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483013447Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0/d, Name: checkpoints/4544/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4544/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a5e63ff0a0805af7cd3a557749ca1fb5844ceb4d9a4df31d2afc0822f85c1b34fe9277845f4ff60c54fd0570d0cf98b061c3464de41adb2c21bb559000102c445e78de0568db08ba4675db9ef69febb4921fe2be3baa8a264bc30cee4869e364fd1bcfc11bf0fe8726613196b013eb06931ed66c7e41c5915df37a70172eec5f42521a745d5d52a24ca3097009ef82e826e52e2b7cbce60c1dfb8f537f72299c0f355e1591ec24b3df01dcdaf0abbf756404e23fa9251c24e7cf0ee1d0de70ab35a3cb430decf086559e1cbe1f61d45c7428ad8292a53dcdb4221522ca3f9c84388a7a86b8302169ce2074ed2409eab1b8a2828bdc218163e15b3000b5dabf8c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48303618Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/array_metadatas, Name: checkpoints/576/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/576/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5189e8580cc722d8715d3db63ab0d4d321f5fb1c89f26d3d67569ea69ac45498107f573bb8f56097f63381f5f9d85e6b5a9a50c637348de521226daf5d5fee805d7e12a7fc82613bd1c5e32619f937ba81f6d8e1d8ce88e7e9c8e4a54b020c66e9497709f68b6c0162b5cec28fe54ff0a85629035cb706f5f79856bcfffc07e5d4a5a56a43373dc0180af756425a36c88f9cca9b8fdead3e1b541406afa05c18657a51533610a0f7b80bb0173497e137aee2d8a547a0c56467791eb2b1d777b8db48115017e9c2888c7b9617b650c1636b7f03ace2e842235524cb944cadbb5323e2faf86f352bb40cf43771a4be99b164ceaae31a245ab074147bb1b2b48002&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51652616Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0/d, Name: checkpoints/3392/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3392/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b64071164fc2b4c1141c58d058c93dbc8c8677ec436c5675a70e3b18e246bb173bf8d373d9e781f889c680f3b0b6e9e8070278d4b4795bc25caab7b134a5b67252593c50702d52fd9cf44518e03c007dd0f1facec9e11e727cfb8698c5ade3878e91c086f8ad9f47c2fb397a17266059d8d9a0af11f5b0100cefe126ba1427010de3dabe223ff613aaa1b46d042b6a8da88369dc66614859b224944556b2a8a8ebd6cf3315b19617dfcfef291f6fb3c1fb69fc11e9c9b318bb5f99848d4bd41fa81ea2a04b3da0e5ed4d9e966507e85ebe41961ad58cb85c9f9c708935d6c28765c90fd924ed3d4f404b72c4c2deb742b5cb5396a737e383e02ace382421e88d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516566537Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/ocdbt.process_0, Name: checkpoints/5440/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5440/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5b956654e9dca43dead6f7d4752bc2f526b986bb6cc3dbc75aa7546946a4892b8d7d2658ebbcc2699fd2f0dde864dfd0c0b43c6c8a95b1bd60a6a5aa34aa9dca635ad12cc73fff5293d772aaf68f88b921f679ec38abdda6f29e821702df195d9cff89476efa7ebb06a6e37798ed425af8aa980f1453bdbb73042deca289457eca959d0a1707b097721915234c9c1e19057fbac3885c8767740a6d9795d37ed7687dac436cadcc210969bab232800d9e6b5621a1c1c1007e01e40e3930f713a02d86bf15290912958fcf58280e30b664b228a4d55d08ba9c08d9b6f4e99275158a29a8f3b7247a38d8434d1cc8213b968f4fca331dd0e5f8ef8e027b8f6a398e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516583188Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/array_metadatas, Name: checkpoints/5312/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5312/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=390cef52fbd826e4624d2bb31ed11e2020340ebf0e00e333813efef48af5288a0eefa8d7e86feece970c7eff7395b331e6406f510375240a9234e6aeabdcf9547e529a2a36f8ab42b037dbd5e802f77ec4d0d54afd9fa1edc508511439ae0035b8c7f6a39a13a979a20155e1b99f66055db64f59b1c5322e192a17f771e35f0088a78c9d111e29990c64565539209fbe555cbc20ecf64cdce2c1072ae7bb4af99bb46cec7284f898827380042e1bb33c76ae8a295bf28f8127f15dbe8d2e58660ae10afdbec0131a7a95a2e56f49bcdcd6cf7478038bbf334fb37bea8fa9eacfc45c65ce451537bcee598695d27f4ed2eec180490dd9912fb7c999d0f0dba9fd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516546159Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0/d, Name: checkpoints/2176/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2176/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3ddbf06519fd149ef5fb7c3a123b720ef5514f01ff20b3ed403fcf15006fcfceee94dae4073e21e84b6d9ef8a286236059b3b0e9e5a1ae48e339447de577a0674bb821c59e200de8a3f41c2171305cc3892b219fc00f59ee1d8686665e9d84d7b9b805478f8b5da7bb9bfc79d0de0f197d5a8f1b12d4bd7f64968c313cb297e9d77ef6f061494823011bd71a8d72ae424377c18e50ae3e8877545510d6987931be73e13566463ad16d6d74879763da0087318538f488848c4fb8645989a49f2f9a480b49d66a00286ece45ef40268912f242fce6224a830136874469679df97cfb9b633ab867f48e431a28c1752e5da4baa770130225d9f464f2a48ae05a0f85&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516652001Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0/d, Name: checkpoints/1664/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1664/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=21994318be51283315ab958b78c40afd725240105e3bb3f97dae81d072f211da772561d054fddaed130b1d231a6d55b79a940e1e6a13cb847d6843be5b0c468e420de944d37451330866db6374b95d027f7943ee59ca142e8d550a9e026446083c97a38328e66b4b2b2e171f973732669a6505510c7f7a970ceee4ce934e702a58a89bd5b19620de487524bcf6f1657b8429a3b3b07a4680fcc8233a949a5b7d2ee8c1ad46a63ce8988b150a80258bc70935e6efb6728a970ae54c27a5188cb7a003b8a534797089b42203ee3d67e87fc33256f676ceb4ea923c033b4952f9c2b10f346263a46af11fa21517936b4bbf85ab6878ef4fb0e07a98447709f438e1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51667696Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984, Name: checkpoints/1984, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1984?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=230eddc18f62e7ab562e7a061e72a98293c57ccae5add8a528f7ad05cc38854922661ca1355664faecb112faba58a64f9136fb116c3a0b5ed0894d07afbb35e8873bb2bf73f50cdc2ad815ccd7bc07bbbe3daacc97bff0ab3444601ced250b02b7cc828d7753f4f4fe1434e9f1bb74df7dc8dea7f5a0950f010f04404e1f87d6120c8f198cb7980444908182541edf366c77bc2913ede47ceba7a52f5b275bf0d0b442849cb37b3fcdc3a40e64058a5f647a539eb0fc85779f8b658eb8923725a6cfd4c54094307befaaa5684ff8cac4e69c929ae2cbf362a3e88c9eb6bf1c9254504b4c0474b689b4a1626534930c3d52858fc3ad09f75f45254b344514f0a8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516752555Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/d, Name: checkpoints/5952/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5952/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1b76a53e7bacb7a796ece60c1c25365d86dd0328f7a23068d2028f8352e55a382164c7c93bc3d6e3162057e197fbb6b290dde1e0697c994ce96e0554570cc2d8983d0bcf2f0a2406beb1cedd641d62898be32a19a9f8e17966e0199292d8f6b030977e260600f799bdbfc685f0b34e6d513bc105dbc0320bb53b6e6e1fb66b57a0a0e99fc99d346b9bf00fd4a4f36bc967b9fa4ef180668a6fdc8998d352a83cb05559f879e1cc6fb826930f381b154378504a1ef84a2ab26b103d21878f88a0dc9ed0533fe41ea1b7622f4890c40a53b9601f710af0a554b506e715ef92fffb15162aca00589b105277976bb48d9163fe2b9f29e2973ecaf50aa1f966ef4d20&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516908476Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/d, Name: checkpoints/4288/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4288/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4a467f9e9ecd0f347c51934729f1837a75c64ba5c4bc47694b91558dec96af9cbec4a5a26407d3aaa853df71449397ca542caac23bfc2fdfb004c3169434d3dd33c464083ce2534b6b41a87a83fd226d6495f9457ce02168be8b5aac01a8284c31fd4db14ee229c171f718d409f7d47b97ee15e9c339dc71adbd963422362c01f3baffe8d6bfa2d22c12e548e99e0ed64b81fc89075a8c01707a2f4451884ad6c67899742e9aaa12d11b832350d15dd07ac89a24a2ae101408b1f972a553a429a08ba44cdbb7dfcebc83cc2958a4ed06d58a6cb7956622174e6ee12230941f69c4b23ed2cfa960681923dd53b7212321a6bf7aee26cc3d860f5339a84e4e9592&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517210967Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/array_metadatas, Name: checkpoints/448/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/448/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=196d94e86e7afec98e88dbe077ca59e5ffce5f7b4c1bfa503d40b6689c1364f9639e62b26f7987c4866576c785ee72b3a3b926fdd2a42b4046bf72ae3189360f2b885fd34b4a80bd3f69563e4ed11319496274b2316073a6d5604bde94abc0a45053e50aa72b1120281cac26284be0f71f0f75b2cbbcf4b74cb829a04f53f50d4d87298c8620ce409e925f660c23190fb2e6f79d6f3d25c8f3e499e149f4faff2599d64de44b70ea2d0195febc72ca305b7050b1b3d43ebce923f4572f25ac57df04aafeb3b6fd0c4d10cba0bec1c46f62ee28bc82e51766eed878a8f062f5cd531e2271aef957db802160b76fa945414442173e9a257fda96583865a0a075fd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517209778Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default, Name: checkpoints/704/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/704/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=648253e663a8e3973c461b1a921fb23b69443da5c4feae22834cbd51d69a80f9a151487411edbd116c899689bc66ad646daf43c08aed545a42ea35f89b215e3f3f798b787b728bd6b15d9a84fab694770721e571327f2dd3333abf378a437db8a095789b6e6d91a4e3ca2745a44709851f43549955ea529dd7963a990fed0e075051691730b31b99c566443bb1be18bd7b941b1960b9174cf7bf5df7ca3878a3ac859cebf2273564d110afea4e78d66aed9ca644b1dc01b4b6140f78a6f16f243595253a21408ca19b6982383de16c94722cab90854289ebfce4bbf208c07333927bdf5c9fd3634cda879ce923fcb6bf3d399e1a5537b6a294d051e38540e562&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517216379Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0/d, Name: checkpoints/4160/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4160/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=63b6ec8d3b5c4c6bba77c0760af9e4ca87255c935c12a8119b77902b0a83846fbde86e8e6f9eafe2e0cb18485feeb61427b34f4856d3b8dfdf7b753ee7f834c5c661d6971808f549f98cfb6b9b1e98970ac5cf3c03991c8ac2a307dceb7b581e812d626cadbacc24b6f6c95cf2124ee8d8637adc0062793d2d49553eb7b87e912ef608573d85b65abb8d00c81b35f12d746f44ccb78716a652d7cfdbd4f031ea9faa1f8af8dbb40789d601aad33987da8156168609341ed6f8862906cb5f62b587d662ab45dadcb24b15148b22f4c7294ed26498aab3d4bc393cc6ed6826a944cac7f5ae5d302a1fe9032238e1e4abfc906295c293843fe778e1658ee777b183&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517251643Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008, Name: checkpoints/3008, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3008?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=68726ae305f10a02b8cb9fd08dc3060b16ac51a12fe48bfeb1a20b7843e516c933f7c9283f074ba3b403d4631770a49457bd40304eb26efe61c18ea99c924221dba3f80c4619676d7f8986eeffb6ae9e9a22ad1481c95ebde4951c53fbe1866b9ebe91fbaa07521780dd568ed60aaa737defdc7dce62a8648cd5570b7b0d332994c62be224c5f72d0a8016117615a34c30a03bb7004a7318a429d81cb79838b9ea8b33cd5bb4113870e4a317ba6cbe5da9675a107511fedd13fcdcb054ca94fc886b6d85ec154fecaf6b621f01b3a944b23c85f81b04358e50fd980f9f99eea253312fade9b3f2cb78c204750a27fcd3fb28f8753fa11a33124980787d69fd1b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517238425Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default, Name: checkpoints/4608/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4608/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2249bf81011a484da9e6fb322b343e0a38712d6b76c219193c3926b0fd7916a58974ef864d6a6dcedf790dc9747761dbcbe0b34b3a097c7c8d0dff3a4d8d0d5c5a5a615903f903f06be39ea366971c3ed15bfae34289a433d7943a761679167dc952bce57a9b361226ba47d503cefefedff6669fa67e2d6754e186c8bec596a790f1abcbdd8203c858a88281107fe7ea8bac0a4f2da07efd85510c0b195a33cbcb11e6a84045f3c6c717da2275af3cfec2647a383047e364d7736255966b8beedb0ee907cb31ff0dc6ec09f0498841a03860dbbca0c82bb03739d6928faf6c9446562ea31bdf696b79e7bd4dc8cf11a39f992cc4485f030d3031213f0f0a4132&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517347438Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0, Name: checkpoints/3072/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3072/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bc255dbde6ed3d781bac2b4eb2eb995474a032ae43b9541f8d3bc3336fb82764fd651ab9ffef11690a5bc146087e585e0e21e615b9af6f6c30949ed647b164be3b9e6f52aa43d22662308abfe6399075a6c5b1b4148b9e366f4c73278bb61023b592b02058fa05e378f50f323dbd8091352df1dff99e1929a0347d4a01fcf99ae191d416f85db900e34b5f6aaf8d04ed2738adec990b763f08a122bf8f5267b9c6ce94c5e5355d723b430bdfa443591ea9d59166f8afa0c6f7f65500cde8d3dc3373a672133df8d8585ac47ced12349a4c3902310c2b820f61891652d18b31ca3891f9a19011f8e3000bc85be6b039bc27cbb0f749c77d3ac3f332b5dd7169e7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517355033Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default, Name: checkpoints/4352/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4352/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4d3549df679c1508ceeb6b06b69d5b6e1ddfa155ba8c59260e4888e849a05c0240605d36917eb9c2ff178981801536fb9b3aaffde39c0f4521668955bb6df8b7e8f7f6f6e7ac9b033896b185e8c42a802887ce964d87fd3a1d2077fb3ac998c97d3b4db6eafae193b0b969dce48c6caf8f650dc275d8a27b1e676d877cccb9d056b7387f4538b52a7049059f28550720b0a0886fa11b987a029be95ef4cf631f6bba7f5bc3c08eb5a59803fbe5670c133cbcc126bf49e78dba2872dbaa21e4543db0672f47370b0c34be2b144301defbe8e15dfa6ad1d159be079b872ef8e8d888371b7a5e466e9d73ce7722f7aef1a52d175c0b1653276824b5a1bf9c6937a5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517291838Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0/d, Name: checkpoints/3840/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3840/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1eea23b53fffd0204c599e950dbf74791724e65fde6f32b03f408d7ea4010ff15f23f463580877ce0fbbee8b3085f40dd6e155efbd996f422edd4a21a7aaadccca7e7baf34c76a3784a3be4c91ea1eff57341c35533d8b75ba3bb263d55cb661f8c65cb66e403cdb889938e0d6b6debbb2ecf6f8742be2ac62050e7124317f583e251ecf90e918ae8cac26c32114db0cc2b049ab16a2ef25193835b961d6cc17b381d2e31aadf7c78c275e937635078460db8f94c458df2c785a583d797c7cb0b371f503f6bb50ba349e4830d1d3e19afad1c97d704ffa1118641d48c59d3de4097343e5d78b0dc4034fd7bdc74405a0defe401abc921189654fc0858f286580&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517421161Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/ocdbt.process_0/d, Name: checkpoints/0/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/0/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bd82c7936341b75563bb1f1b04bbad074ad1b685af41cf13d8b4f199d34565795f735ff37c1ca703b6e980de55356f2e103c2a621d70a00e59b083d6186be728c1a980cd28bcfa9b3596f8649e72ef2560015db9d2b09d65103e7a41b4ca3758529edf6137bab35a395c3bd02408a483659fe1b69a7cfa0be7de52e859fc168c867835ec764a82e6483ad2808fa0f0f62684444b1c56dd8776c9f6733ffb06bf43e45d13481904b7a1fbead49ebd70aabe474c08134a5b521384d482493026268872fba15539168980b6806294a04e481adb6bec1fdf52db2ed14802d7273269b88612f07d0a99e9faec08a0b7ffd9021b6e03fe415376d6fe8bec0a3c520f24&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517299726Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/ocdbt.process_0, Name: checkpoints/3200/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3200/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2be87a8c4d47524eee12ae3fa9700af46fe3f6dc6b240f610d41efa57c8eed540dacd44a2fc330aae07a4295a4b2860d36b1534dad5030e89e92a7d1ce6e7b2ca5381fa05728a1be25a22d6208d574a5ab5de9ba7614c64563ad26db1024922de4ca8370c780ee0fa88ab9f4e9af6d7eca4f914170e35030df4bfe0c010a7629a6c65135d1b8d17e1500cd00d6604423189c13a0ffb81126a1feb2563234688c71197ae80424782dfa12e0fda52c771c9cc5c94dab73eb0737dd4dc390c6ac307763208691ed172dff69468f6f8de595c921bcd97b13fae950c9b137338aa0d14d1b08ed4db53cf009e9f099848fabef2e1850ab0ab950e46f51591bf83426f9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517441595Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default/ocdbt.process_0, Name: checkpoints/4288/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4288/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2cb2351d0cf1b6a011766ca3b06adb69086ccb626c2b376892332d9f9596090bc19b9b0cd2c43bcec3c5bc0551d59b1e47e2102f483b8411e1d1a2a3e7fa0eb4f27f3de2d5e83579eaf5f2ee1dbc28cfdee562709e1d68af68b33db80e6a16f58488e827eafd4860d4126845148d8fa1baa8147fccde854c9f13cfe0029e3fefdbd3844f96dd8ca52d5409dd43fcacf46f284b4c66a779167aa3a10dd694d902fa076273cc1158bbb0e17cdb57b0a7127211603e54955e6bd8e35e389bc0d7535a68d76b22994bd1242ed443946c383241d09ba89eced710e45b00f76fad2bf5a3439029a804b2bba48d4ed5991fa355bd20c2afa98f54e5c9bfb3ca9503af0f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51746464Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/d, Name: checkpoints/192/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/192/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=07a8aefcae0b783a5370d3864eccae2fe4e7d6919c18a10c7632c3ff0f553e01b4069a91bf25625fd554b72be964f3fd036d2158e6f942b300347033703905bfe8748f0b6995dc39eb5ec8e8c3f5e43ebe688da19da3c8ba6001528bdd51fcb83c3c26362234806c30ad03c6e6c828a572469e6875be550fa0a92df74fb188dbcb9a277aa8b7a354d1a80517c9df0d3e052549db0928d0d617f7864664469d87afd6e240799fb89316cc02fcbf9f2a1dc2407e943039fa3ad05fb6787003a976b945ccf45d56c0fe807d12266608ebcdd056ae3ab76f8bf028eee45fc2e31c2f1dae24bd35bb9996727aeba1ac474dfa5c72817ff094007c4f5598f128acf7b6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517479151Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default, Name: checkpoints/2240/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2240/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=109aa3a935322231170382e1c3446058e097ecdc8861771c06cadd8ca8afdd4981455ad24fef5dea308d573a77ded784577dcf6fc17d4278cd1aaaf134cf564bbaa4e727ce0cf72d7ab0624f9b61f758c69542ece4f474ec8b6dd43d259292e3aa7821e9dc780ec921468a288eaa8c83373f73dfa8d39a8fd448eea9f57045bf64e6b861d280da18a425675da6363122151c0a32549f1fa35f610440a941009b45a0e4af62e5bee9566598ca44380d1dc528388ff275fbe6a1d54bfbe3a887179e3e75a109cd2239e002602851a2adcb695e26be2f4f3346da96b1c63aa85274aa62546101a145796692f7a2fd40e8ec3d73b6a37d1142359f0445d4c32b91cb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517540889Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/d, Name: checkpoints/4928/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4928/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4616c4a7e267facf37d10efeadd3320efe9f05c2f1dc775ae015157bf86f9ad849adfc94506b3110899912c31f7c8a768064269c89eb19f255f3152d4ceed01876f056466fbb72d7cb8031e49ee86db14882ba07a91849cf95bb4cd9b32fbeb0c4c9cf0be0c35f2fbe603ffca47db40c805a588aba3d340461ced4674b7b85292d64f6cc284fd7a4099685754f8d149e7978fa78d3e6d0e39da393049502c2fc9e4f035aa261a2960acbdc3f68a0781909ed1908d7dadc9dc7a93a4617ed724c5c6a06a73fba3af3ef827f3fbb5e35c92637ffe6c5932162683953fe7f234f6e7f453cbcc4b4a61619e25a81773a23fef7dc8bebf6b1d3e77c9707f7372898e8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51718866Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0/d, Name: checkpoints/2048/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2048/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2110118428f5b72df8cb87b967b8de093ab1effc5deb002b7a68ed8edcfc2facf324a8b5c081f3957ceeb4f7a09b6386506f9b9608d796dfa967963dafd16a0df79a42f7e14628429571f223c7a1c1213f0d9b94c586784889fc5429a1efb4915edfc6a445d2f78244966fcf83c565cf23d501c157acfdeb30021ca8d57e6d01adab4a912a8b557ad2d1e84d36e84d7ee6b2ea45e20ad02cfeb0bb84429f2bde06159f160c88232b0b12f65eae3a319142311c0687ed80a8bf0ad8c0205815718776829df9f140975ad33243b8ea64a5b655f79cc6f168f146591ee4ccbef0473678fe85f607b9749f0b755a2a4d6cef0e84c1aef2bb5cf87437b56599802481&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517466336Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/ocdbt.process_0, Name: checkpoints/5888/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5888/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=754a95a49316451888edd1594dbf8e6b202a70c0f14c195408490f8439d0afb12f9a2623c4d4aa79e46a551bbacc60fd7f6d954c87b1d3af90576b26b223523b653ae30ce97cb53e53aead1ae07cd366d8ad6aea79d3007f768792c7fd8c138923ac87c49374d6663755be61bfff874a4b688fae0deff511b66ad514dad13a356b0f5377280c41aca6ed04eb1cfdd88f26d0f89a11070509c4ad2d129c1324e3914309990fc853f03d769d41f2e79d9dff0f770ca0253748357a71dcdc278568eafdc918aaa34c30c787cfaa9c254fbe22cb978b0153867a31a792ac0720bc91bf2510ec10d17c7d3c5740efa90dd4834c672a3ff38d9f49cab1b7085f81daf4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517595999Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880, Name: checkpoints/2880, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2880?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bc93cd965dc49f51207e9ba42ce214b5d467d49f17cb87fbbdd2def59424e029a792d0b26e6679b5301793c6a198d2abbdfb4b01c45c6e6dcab9063b1203534900e14cc26c3e6f76c7ed377080c25c4be50e75d498ad38eeea9afcc5d160b24df905d4caf3c80019337a407a8b1d651723901e3d62f108212eda7481f8828dddd2dbadf695dccb94b2ed5869d54b7b459345d70538e6c5356eaa3c5dab5d01c839f1e39ecf5902005fa66d87705634f0e5888dd6a2bb49a503e9c93e9423baca0b38c167de41307078e909622608433fba0d368b4cd9aee0ae7c3f97bd862fc46c1c6ca2fc4fca72b9d1ea513ecb92a878fa979bd60546cd1a941805144554d2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517541397Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576, Name: checkpoints/576, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/576?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b5cf30d08477bc74383e3870f7b844b039eabc06999e8aba5ba7b3cb85d687a72c2d64b0f5a83ca5bae62d920c770deeb52f64e6edf0ff97f9566005a34abe7a341db9509924060b7829a08b836006c062919453551a3fecd12ce222fc319fd250a3ad4f2cda831cabe3e5fd73fad056a5dcabf72c17543bfa58b7f3a9783e7678f55c588483b8ea3ee50672742e64fe8dd45a3fe4e65c19579d60530bb82c3608b8395c127652761379f2e1560c12dba57bdca00ba3d161fecb79b38b47a5a2f443d2a37a205e933f2647ce16ef962ac91c7b2fbfb0a2f61ee0efb3d9ed2cca7bb75a6c41c6d13e45c61ed2d8a189dd3f38cd4e85f274bb1b8eb2314e77a162&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517698006Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0, Name: checkpoints/1920/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1920/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=35c2c41d6d57444f17c694af644798f10212737ff87769753cfc85bb5e22530ca6a5ead32804e05c4a1cf841ea2b09567bde28e33935559f812d89b76443850945eab01a0e12357dc9bc051354931f9768f5470c1cd431bb1924572acd4ee546cf8fba58ee95cf68ef4feb08f6fb0454bd8b55ed63dd66ee0ac17aa67edf5c0bbd08d7b16ef8330dbd2049f08a8aca138d7d2312e3fc52b7055c4e9ee9a94c21e71d3a08c21e3176de7942511a077caa4623112dcc553acb73aedfac09862beef750e845a8e0a29c27527816c2f2ab14d310205ac74b5d4a321b138a2d6c338cbe3ebc99e8307775d9f0453dfd595a3f9dcea56ae3c58ee193de2f0548afedb9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517748258Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/d, Name: checkpoints/3072/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3072/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7d2cbb5128dad500bc006d455c3c312544a798e3651901c6e1c7d5fcc2f337bec8534fd051026351627c976dfcb517ef539a505cfe7df7fb5f170e26d4b472718cea745481cf997ce6d05c2c9f38b4809cd56ff0aa2b3a6f0d1ede8fbee9049cc4ec1eb5dcd366e540a6776cbf98e25417336af3655417d8988a27620b6f07315f107a2f1ae4c1ba24b692c23d5253b8a6fb0e5715f676064bfe29640a756db06e2aee0bd6150d079a0be249552715f8a4596c64a906b8850c93dd26a70cb0f77d6188f21c04e9bb22ffd8ea210a75801fdce507a31bd1a87a6dde4e2a5a4bd06bbe1d1ec1db19a24a8c7f3887df4cd1e2dbdf5a9cbe49bb908165c44fcfdb1d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519186074Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/array_metadatas, Name: checkpoints/2112/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2112/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=14a5d20a4acce812555a39f9404a7b655aa097f54c302faafa550beb7dc16e651ed56337c798a915f7e696f080dcffc465125fd0b744f428a0f57eaf6977598f63a377295e5009eb11d71bd259f8ce3630bb6019ac8344643f490e5153a23c2dbaf8a25ec9f0dec805596903914211da9e157ae32a8578d58543d837bcf0ab2669f224bfc863e097d64f8b6136b7f35537b4aeba1ab6457e2c3e05768b8c8f9a2684437b4831d99cd5918bf564c9858849e5b92a067bb2e5d9ba871d7ee22e7cda7604d540f46bb2b6d99bd1e74da3d87c989740a83cc6ff93fffd382cf4511a33eb7f3918ff59e87917e6be763656b2b71950fd0d7299dc914ce5fa91dbd453&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519227723Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/array_metadatas, Name: checkpoints/4480/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4480/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8e2889c2a4da9085f3ad55bf61af8261b536447199ce351c89e6c55cc85f032fe79d405a1ed937b66db68968c9763bc6af8ac64ad43732ab2e3f828e2470482e532dc49e5c616e441b1901970f339892935dd383ec5617ab6088d8e4a0de72cbdf74065d1b5ee908120afdce1d40d90f90e364ef1a0e9562c11e671a4feb826930601acf54b857687b64f10cfe1c93a56d5265191545adcecaac130ecf4b0e20b98e6b534b65041dc89d4371b67ac86f75792705b9f121683a1bd1e89eecbf26c1a8bb35070070109dea00fdea5e13e2c539410433f58d66f57ddf4b95845c94ed9a75c0a3d262a7db713bcd07c438899dfaef39b28829000e5f9a2074c7e961&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51927176Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248, Name: checkpoints/5248, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5248?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=181ac7cc3b3b9513f589dbd080a4c4c86f7ea46e0a60fae813fac4dc215f91d6a0f144dfe5e8942557c08f9db1851de8a19ddd64fa2869d6ddf33239c667806ccfb11ff01091900407d811012729bf9369036dfc3b5dbaf6312d407f283660ad58c7dae440f502ef47f29e1d6c67e043e3dc90f6487c60671d540b953688328379ec78b66236dcbd77e53c6c93cb8eba416af968f1caf62f4aa384e256ee390b33f2c41dbb2c6d9864feebf084786484a570974649c2eea3d5e84b20c7c3a5403d58169dd1f42e4d17d897407bdfa8c1807c059084bba40a36bb539f1dd89ffe8c9c7b7ef9cf712ef30497ceed84ec7073138a17614624bae471ac7f132528f2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517673181Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/ocdbt.process_0, Name: checkpoints/3840/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3840/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=415a94931b3065e4c725a2d8f58d6f20ac0dc4c52ce3123f115cffd0ca6db195e7483a09b2b445d4e8b518232612e3a4af6a58ed18e9da5567b3d2ce12797c424a1207098266f4a9f5711bcad6ffe79190b981e226de83527a7f8d903046fdcdb0ab3837366820eb5555635642094511ed67b13ef6bad3214c078b59f828b1085cb4f8e2835d7d27b9dbbdaf430fcb5ea63c94f63d41f9a61f1fed1383af8653918b46af76d5852f7d2140b7b4a3e1fd713fd31d5035ddd6a0336ad9192aa16e14286f2b43157faf49e4db2229901707d466fa67e3f02bb6fabe059c8c72bac393c899ed7cae8c1a1ec63d9dcec58ede813cb4189c9b8cc0e399cfb353f364a6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519192321Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0/d, Name: checkpoints/2368/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2368/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8de1d22a041feec15efae24f961a5009caf2f5ca9db7d0aaf6a62729d2175009456f64077d46a05cd62cef3a4e9790c2ab351249e0f08ba97e7689f9bb9db03c45c15c98524355e343d20ffd1a902857561bb941b8edfd77b68ce75f9a71a03a5174a086669db1774a81c8d5edb8b6ab13e30d8838f7c675cadb23c793375888e29611b970faafb7acab1cf028649fb6571e9bb95ab579cfba72d7626d8f8dc18c07187ded89565ef625b84d6b35f8e5b5b392958d3beb33607a6c09030f6088f55b13efeba3f49036f856aad089dce95209d7bad3f1ba0f8ed6db9b9ffe37f930a943185d51b71db898665f0946679d67a21efc03705a4e6098890c7a3dfb01&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519338904Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/d, Name: checkpoints/5376/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5376/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3650f2e52508ed685a47ced77ca2fd0c5f22377c300d7baabfd71687baa29a50d1c87c614c9206c47274719649fb69a28a3b7f90ffcfb23b15ce54adf7ab7b922586eb8cd6114a44eda60ad40600f027fda1cc76bea9afed795a378a3867cad36ff61aadbd8f26f09bd92b1d2b65b03b2fdc17e226a57cc4d994b99a762c4b6f9c6aeb773d9d942661da65f9ae71bd2970a139242aea067c56e7974a60e787d74790e87d4b40cbb56c8f197752bab8db5ade41b5c64114ed09b200d55af2b0fc96604d0213c7e18bef3cad9bfdff779dd80559e561d2d090b13db1fd55f338197e8467fa4ea8c71fbe1e0d3793a50bdfcce2cdf7d0ca3b23487786aacc3da7b8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519345159Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/d, Name: checkpoints/5760/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5760/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0ddb1ef435ef192a2939d55b26c4f711b7223d6d88a3e171f505a9e79707fa8e73a45e3b9853a213a962b92f651fd39827ec09ab1186383dd3c0e4ad4d8ec5850d9c00eefebcc0fc6da333de556de1e99ae9c8ffcbbd6e06f312bd68116468a08b56756531d9cf69a864edb12b64f3a73fc035de039c41ff656cc99b246f68c2ad890f88c6bca58da1d6c09fed0f6d114ce9f69e477ff8198b81d6ea6dc58abed398c75729458d1b7034f3309f5f269f300176ed20086b0969b29c903d2fc36100af6f1458e3c1089bca2401cb8a387c65538b007beb9801236436eb57a2cb004892a888de74bbe981dc16e2d26f91a22720db136baee95d648d2493dac392d6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519394573Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0, Name: checkpoints/832/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/832/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=68aca98e00fde03cd43fcba884f03c805faed9499ac25877e72360847d4e07f434c1caa5df14d7fe439cd21135da7de4913bebc83a36a53eef42d04b8eab757136e59da991b67020821698eadac0baa4d35aed42dea7972b2ce2a8baec5eafe398b6b5b7073100e622d033e74d44acfb5f989fca925420a10cafea92e402b7e1ec24cacb4ab0a32c196601a09c2bf193aec8d8e83b46b591b618f6523953625f28512c4b9270029f386b0bdaed733b31c370a62d14dede1e77b503adccdc9ea01bf31ae0b96f692a1414d447774671f778d11b81416d8ff417584c8a1a890ce3c24249590f224fbf10d26ae28fb9ce7d143dffc76a258718562ba4385ca3f04d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519407434Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/d, Name: checkpoints/5248/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5248/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c25ec3179f974b9b07d8efbb6f38377935f8b28340dbf1578083d9b665e4c1764157eb9d44d2358b22a0017ad0bc8a769bdc2f9abbe9908541c6b3e11ccc7f5811c4a555448f34e9cb11f33c0690616f248f4c04c59ce80938c31288c6e7df853996a9bd59b7f4930263fea50a51262d6da2279d993dd3438cc0165b847ee757bc481d9aaf9f8d18d7644bc5373e8b31e3e71dee767936bd865fc99ffd476f64ca7d39020dc3201baa1189503634176861d681fe87541659c5459c2f4552a09ee00847dd7e5e852d3c2421b389747ad9f5cd6f16b01d50aa368750e295a0b037ee7be3951b19c8c68945b53aa0ca6ceee2354ef5276862483b503fc9df145248&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519441552Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default, Name: checkpoints/6016/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6016/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=16a5cbd839a50723142a7f977b99a312edfbb1953e337534baa6dbe5f5fe1298470e3190a863c33b93b11a66e468c158df3ef3c3111ddbb5d1be655f7c378ddb0e59444b963468cf395254742bfd37e568636998f16aec43e616a07177a5c2cea7daef342f1fa91b0d833d90b9d4a75cf494203826275166a531df0000b8fc6690f5755c95e76cf8dc82da81238e0b855485217dac47f0f3b165a859e389efaa50de9c55cd32ffebd4bd8642d6d59753baaf0e059912d9493e29d166d5f8b050ae8a4ba3c2f5a5e65fe558d876ef77817273dbae0ad0cfbea0137df07e2a88325372dfab5301b054eb0aaf6fdf4229a128fc7da1a447380260034d272f38a29b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519426098Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0, Name: checkpoints/0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=90236595e586a23b6d831a73143b6b83623525230f05f1ad42fe46967931421b636763eed38fb9ad76376fb6f8b5cff2d5528efe642c421eb9d42ee49385a2dd00990c44485d5859c5fcdcba5d203e80555b97b0e439927bb6dcf163a852284846c121a8159eb74bd9928916bfc5c6506e484cdadfab79f8a6a8f2d40fc6997f0fe01471201f94d3fd2a066cda61bea49fdc534e46ae7cdd7aa85340b5556515ade6e131edfd56e78f6037b740dbf44817aa6bfe65d172d016edf8ed378c349172dea49299b4789be4c33a10c0203208d3198b7c2e867ab943dfacba27e773bfd981713ab34b4739ea1e8d543fa1d86a932015170f81e36565d9b1982a3b97a5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519231341Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/array_metadatas, Name: checkpoints/4928/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4928/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=46c6d12c978b7ca618f42e1c6294bdd5fb2d3d1798162f7401e71bd97ed4ee663f877bae51fbdbaf1162857634915fc0b3eb3ffe17703fc94194c1c805a37c106d26106db36542e555213840014d630a9b36693875576b8b6dea39205cc98c920efa53280b4285f63447e651f674018faab74a24400a8b28875172869da2cdf0bea4c8220a663802e126392752a80f981c0494f86f6b09f8c08b9d9c3f293bd316cf226c9068451390ab496ef860bfd3cd66fa7d29c5ad3e8f1b963ba2dd939221d286e6987c690e453f94fc0c36976bf452ec1e9ac2cd10f46cdaf707942096a80fec750ee7a062a09101cb73dece5a40941f4d67440799e8ec3c1bef38c93c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519477864Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/array_metadatas, Name: checkpoints/3008/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3008/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=80b0a396645c178188e689d5c398939aa43cd049accb9c37012509373712223dd88c648891d658fa857e37edbcf05579d379e70e2084daab5c2606cb93fa29e0f5b91a03f3222c85a1283ecb95f51f8345f299143b2113b228aab25a32b6c7fb6e7d7ea25a8fb35acb51af49e018618818196de1049177cbe8b6be3297269cbefce9562f38627f657667ffcf755d1b9141ed5ba10b49d07858d0e43e60fb803733493cbb17a45168146a4d504ec3c9186bd57ab93ac0191d6e7cbebd2ba631d9e738f736028347c383dcae0ef505c2f508f18b54f343e8a47b3c670b9fac4d957ee25ab16299850398f7f51aa3997fd7967d23a7d321bb5e40d2a46932ba022f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519515061Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0, Name: checkpoints/1344/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1344/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2a38f04f47b69c9de04595fe7c72dc61ae9e3700d4d61897cada4ef7341181e85b2154d782a80f71e9c29657f39d838b9430df4bc4201eda565af7b06848def582b1ae52389f1cd90065676a1d5017189ca41a3fb294a066427287d5599fdeecded2652fbc2ff995f4fdb0591551d38cbddb38ccd727f8bd1edf09ec349fcd3e4344d2bf2599ff35324fdd53b9f678a4dc9676863b0c048896cbadf4f633cd35f71f19c26890e675ae9f16522248eae4ceca6dee5e8e735e3a3c213beda239c6bf7f8b5897b98654a77de4c0b6e1ec1083e995ec3646d0db8e08f1fdcb60a07a25b29ce08049664787faae9df9587bc4e1bb3703e8fadac0d399e2531cf3b31b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519537336Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/ocdbt.process_0/d, Name: checkpoints/4800/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4800/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5052318c5c59b16df2d521e7646857418fc65d2f748929fc73b4f7e2f817092866ef8fc45e7b8024038b868ee7d389cfb6011c5bd7f4dfeeedd3c84f5f2442af25d242154c490c374e9de04785e1131464040b8b4f33d13814e7198aa1c6a67b04db999104d4c9447ce195435d92fcd66e037325da75b6629c5074687c0aa346901d39b487138432c000f15d5387a7242ca99c23813c10fca33d57f354a72b7d5758fc88b2fb2f36e54ba49df3674a0fe4b1fdb014f26549064d69d5d0afffdfc37c0ba824c876f8008db30e25e7f037ae30b44b04a6c4bbc68d662a2b190f750a5152829e8b44fed56b27138a6837ac664e913b61c3f6354bcdb7721ef027de&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519418939Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864, Name: checkpoints/4864, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4864?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=96550e8dc2ee1ec574821eab792f150c031a12667381ece7e1504a9f73936d1e3865071a102c5276e415b5e4134036dfa832f2967220a693328b7cd0039a5a0e21c1cddb19e1e459bbbb0bb078b1042aca3a023000158de4157a661202289566af58def3cdb4d4fe87d7625b82ce611b39a7c7bf5accec5f6bdbd11a1d0a9ad337a39f66fefda70e1c5ff3e911d677ac33082112bb4f03ea893305be865b9b3bf4dda080a9d53346dd472265dd256534ca61b937f2bfe19a39d2aede10bb43573a718f393c57d5300616b10ddecd212b8378ffa0449a06c12338497abd5c941021a3b62b4207fb5e7f5259fd39833918cd5b92bdd1cf786412150c64473a245a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519578337Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5248/default/ocdbt.process_0, Name: checkpoints/5248/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5248/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1aaf435a550dcf3d5eeccc39e9023908d1234a0f6f1883844ae4c3e0faee7dae160fcf664ddd375ce2acceb6754951dfdc5e5882de4a88ce88e8d2e9b96a51fc137f923a54dc7ea92061fe9155ab3e2f7b6dcc11d61e067a2614451fb590ae920683ca46616b8601a6b690ad89e117e686920aaffd0e06ca7d7a1f38bdf78d65263d340def913507ec97ad3d2d3e0cd0c16494e0b710e2e99ca8edf97784a665c4c3ff8021d172dccb7af08c79e4566c068925711b7dba30891f5dc9599cf1c7d3dd18bed4db547ea30829bf4b77f556c77abe7ba67943c2d4a11caa903714ed501bb62aeab35ece04133a3edcf104f268001380ee4c729ec85f655cf5e90a19&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519616198Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/ocdbt.process_0/d, Name: checkpoints/896/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/896/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=14669fde866c735e7f28c9026149fa1dd975bbf633b436adb660b1341a4eff6f975da76d5aec7e3c0a86dfca1fd16f62dd6cec9c12a6db2d4acc788516a5ebea8770ab8c9f0da29fd70ad637e8a9b78638c551a964905e6055b920fe99fd2085357cdea3343f39423b51578df429874aba8cce24710306efa30d4986e25ddaeef6b6548931edaefeaeb4ede68d13b4a5fb299b65f1ff7d730af7f87c24277ae24c38990f61b8e267434bff5a3b49a240d849734c300c118e9459f2e83da2b133c8b7ae9e99c3a65f01247128edbf112dbf885a02772972574e169919ab9f9aa73d253b208e7b2016b579f48f3abcb39ab7ad65887fa7665d180dce6889ecdf21&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517274383Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0, Name: checkpoints/3456/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3456/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=53aaae1bd026d22ae22468ea04bd40cd21d558e9faa2fa1d19593689ece4553a3e2dd276de1e014d27aa9e0e2a114ebff696f51a3e30234f0570a1e50a98bc087a0fb7c4839a9feec759819609a3eb9b6e8a889b84e0918504c824d220d74242bf7030974c8fab60b5c800340310f288e1c8298b74e5ab2036007915d490240919aa5277236d1c8576ddf55e05266e5b21cc2ebb4d2da0da36d3fcd38ceeed9682a34785cd1b35b7b87c94c33b4777095071a7ba644cad4fd72053975447e057fdc6b3a0ffa3a895ab8bd5ddf9e6963b074f4365f2f5d859e9b71360410ce27054fb50429ca377908935ea0f05c803225549f7c65f69e9bb01df9deea5b78ab7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51958904Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0/d, Name: checkpoints/2880/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2880/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=12cf4d0662e06bc9f180646e54ca5bcb0c640e2abbd32093c256d14d609f09b68e478ff9cb977bcca36e712f276e8e66227fbcbcf0fad85eec36e9593466ad2586b9b321399e8de37af11b43e8535605b5ffee958bcd0f76a2de9f0fd1c85d5beed570e0694c79230e3b3cf71e0a977f80554ad90afc431616bc9cb926aa9a5e4e8ff8ba7d193e8cb8fb8ccaa7eab1a45f91ffd2c2e1b063e8c80dcdc571bac72eadc0dcccb3118a35a5cd237037fa16bd887bb01d7ffa956f6b36ef853a5b91e743df71979db0c338f2782dfaa66eee2a41f3548586b1817632cd980151bc5c305e1537801799bcb14d9cbbbedda771f9185aa3289177615de3699198a52757&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519664624Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0/d, Name: checkpoints/448/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/448/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b0b5a2a144caf4dce24cb9b233f91c021aa5be2d7ac0fbd00b96dbde7bfe16bee2ef1de3e1b296a644cc18c410f39aa05fd913bca5387d03931e6a810b39f3b98c6abd30aa4c38f78e1589f14ab677feac9403459b0dfe1c1ad360a249a4e587925fcd000c7cb8a2997894eddfb3cecad8c1795efddd7219a00b2286ada8ca746ab110ab126644bc012a5d1539ae52a47ea7c0213cc1c2ec2c12987f47cee9c1acb667777e1f039e2012b207c06969849f0f955476363da968e4737c6e5f44603aa9daa672a1c8a00e4d1e1bbb144b26e1b1f1e1aae163a4fbf5256a9e455d5328a730b134b9817fc36e0062edb6ecbace7f255cbec7362b726843d75f657e8c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519707029Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default, Name: checkpoints/2432/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2432/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=67dc895917429aefe0fac0b7811858d68647a9f7e26efc12dfbe81eebb2f97e9bcc8acc454b591dacf926dea6ab93605ba84b33694192edaac80b47a871cb4bfe0358eed66d426cb208970ef5f06e6000cff7797b6247cea4412c01d075d75f6baea76de6d1019786901a619fb0b5295464386cbb230ed9cc590b3f022c89c3b3242a5a3b6e10e3aaf240645593ac162cd5fef30c3f79ac45a6f3b5c342bc00fa4490ebba1fdc62ba842291adcbc608383cc0a4a0519b2ff2e2211a1fedda068a90f8298cc929eac64f1fb7f4fddca1473cf4cd43a67469699527d7ca66912864b8037d0271ca2a3619d8c05c326f6b7f86161267664f854ea866995620de115&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519725848Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/array_metadatas, Name: checkpoints/3648/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3648/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8dc9dc5dd5e5b697137a1710c333e4ae76a6db60d13e193fcfebe5695923d7d55e4dd2cfc58d4b809d440e5226bc4893b13dbc049bde2828a798bd551af2e52bb610b3289d1d2fd26b829f3714ec29d709131617c7d409b176b08be57dc301dc689b1bfbfd49b0f62b10235eaaa10f825909d8212cfc515ac3623c886fa0adf4d48aef2469b248211f4dc5b818bebb909d0268f9d9b555947e6698c283eb178478f2addead86658159054caea19d3d09099d787f04a125d025e474b36240462b3a5b98c41366e013faa3f7c6e828c8f9591aff2aa8bd2eccacb4a85326ad147ef6fae2eda360afc0197062a03d8cbbbaa1a5a107d54a88a53b16331b51ebbe9f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519742043Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0/d, Name: checkpoints/4672/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4672/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=12b6c94ab0567c028ea30030f52470ee7a096ae5da30a7fb43f2355e4fd2e80c5587f97b87dcfe31ab80a4e41adcd65b7ea6b71f242fb213f162ff5ab069ca8b86d21dd268ad247b5d78f733baac21a7711cff799be0db8a0cf68c38d02e1be2fc4d1266c6936d2f3ade50853b412a561f189234db0f3d046d9909b2990cc38ee29b359ad90ad42e58222cf7744bd94a61a6d9b5c752c1a7e979dc02c3fea99faa7c675cc42300b4ea623d8fe4824b9e30683ddfea74f66bf402be3cf54d6c7df4635dfe4f16da3cf677e08430eb2edf97d8e9d901e92ee1779b4c2dd8aa46ae5d315ef19ebe6bccb2fce2068106d6f63320765b8701fb3b1ed333ab240dc646&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519814413Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/d, Name: checkpoints/1472/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1472/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5eebaa670ceb42f9dbced56ab5b5974f682bd835ca80ee75b4eb13470092da31c9d4b4f5a6a681ae6b0b8b0fd1efc9120303345217bfb55844160c5ce902301deaccb5ca3918ddcf369286531b8da9302ef81f4d2ebce9f325aeedcbe0e0b989567369766ecd8b3f11cf7b147f11b963dc0e5f31bfb1eafa2b6a59c30219405f5c613fc56a0c7a5633af6fa78ee354e165d925f9eee8d6a09f1309f057b80d3c83eca38d8de14516775d1f68b0fb0c7dcc0c5b95f0361ad5424c3a715d6ac474315f84381a7ebccf5ef9e231f25c3d6709213778f3a32a4e2f02f0e40faa9e4980b50b1878fc0b8bde231a41214fec91417fb4af392ebe5794f2589acdcdce1b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51983182Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304, Name: checkpoints/2304, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2304?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4676fc5ffe163021b84ff0926887b1ebc6bff68b4527fb43c85292c5965b8c85ad29ff015829c4ef9629b6d77315f6dd36e45b7b561f665f03b017ae9aadedcad73da18c3a47991e43b3a4a73e806efd6e4230979064ef355fd25bdfbc5f63e6540001bdf27885179e29a4765b16fef23147553becc1dbdf97df46eccd1664713d13ff8dee621f15c26adaca22954980955d2113b1092c5c6763a54fe5c8cab664ee1bd75f88aa5f078392fbab32241404e6f5676d323b3af587a50b8db81fb86975f8318c0aa21d5cf2ffaafd7c8ae81208b83d57e640fe7e2eae812395318b3fa45398818dc3274e8215f00e1dc744894c27c81564d37ac1bf2e471f6d99cd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519831849Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/d, Name: checkpoints/4480/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4480/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=44b9d343eb09e4b2371118ef13737b31dbaa14d29e0532c93458f13dcc816df0f27c1e8506e5748810313b4c0939369c24d180ad8d7ba1ae50b4407f123c65193ccb61ab77908a177952498512d509777b9c27b9eed77157e0d494f4075fe40b9624701c8fcc63c3c94e4d9735bff4f89ab15b6bb4c8f77a77813477bfe2fce2ebe301114ca1038161abc31c9bc2b4ce93bf33525da6bc3b48e5124bc03dda74dc188913baa1391873b7cf6644c738d2b62b6383f73b25003b952e9a50808784a49702f9ce79e2f4e8d829427049e352b8d7e86ea022f81884bb40860132f077b0526a4a974e0fda9b1ddf296181a511424ad6004d6bc241b5285b678d23e795&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519895853Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/array_metadatas, Name: checkpoints/640/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/640/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4353ef946ae08d47134d7c1eff49110c881c829541855a7148a3fcb7fa59baee01a40818ba7806f0fa940bc8bda3099501be5e5f0bd78949af460428037de3c02236036b8e9f337619c6da327faf283d65fc43198bdafca16f79b06387a3c88e10255244e677c03cdb52ec01b9198d66ccfc56a2634f60d96cfda1065cfa5bdd605b2964198f6d397e2fb893d2565f9c480d50f27ea8938dedae56a58a4ca80abe2dd196c4c9719401c1cbe65d0451f16293b63854b4377337dd53d026432bda1e479d834d7fffd8ab4b0885b33cf4e3d81687032018bf4ecfa99217dba5ec49c8dd2ace53b2be9b0d729144605ea10b5732da03b0c9b75c141e0eb556af40d8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519937796Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088, Name: checkpoints/1088, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1088?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0f960bc6c38e69810beaa19d5684da73cc634284a84c430d127c91d431daa571f3a999be8552b7c4d15775e7c41f0f7753d113877697fef2ae0cdbed37babe54e7d31104c2f35e89399abafe53384e84df349e4410f514c784b96ccc2764398dd865ade4ecda4a4593cfb068662e28815c81e6c6c1f2147a22c2d9ddb01e5609d75495171970fefe6d711115adce19b59a74d4c5dfcc6c2bce36dfe27b579df75187b2d2c9c1ee1868b337201ca04bf499b1730d52a20a846e436d3ff3dd72ea17d9377dfa7b53d20dec986e9325d68276512f1dafceddc2391fd5eb780ef7495b10fec53427ae2595f6594ce524beeb9401b9f555e644797d21ca4bf19f939f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519960743Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default, Name: checkpoints/5120/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5120/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=305ae533e454147fabfe58f54cbf96ad25623c4bd592e307b3201588cdeb333cbd7865f979598d6b711a94562254466ff8ee38a3fb7a803f003a2e9237792c5a6b6f323922d3fb0114182f9d88b58ceb56a8f1388dab7d04040199bd4ca8633af70a0ba010d31b1e8fd2385da0a29be9a15570d2d60d5a2f3d76f6b814f3463cbe4c0246416bba50ece1693c87641bfaf008bb2de0a7ee891c6280755e5ab7cc14ca7fd4cbce15a535f34fc76fa0d7cffc138c9a89a4f3e9bbbcec1ea5bfc0a0c04730e5775f98cbb68cd835b31ec8d664863cb7d20a91c30524bb217bf58fb25cb2ef7743c4f46d977824527da34005d10de30495fe4ae9570d88a4c4cf4fc8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519830403Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/array_metadatas, Name: checkpoints/3264/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3264/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0c615005413f77b4436c86d4557c74400fb8db6e24f8ec886055a93262fd91cb0491a223382a838f8ad75ed542d257109bea9a6aba8628f050b2241545def8147a55e4aa6c1330e94121b8950777803069aea87985d77b2af7f7c6b4c7ea9c08265bb561f8945bdecdaa0d5330061835df4ebfbdd523d3dcf55c4f0bfff220458540d616c9306141e8b13edd088d11433acda91e1a83a133958fb5604e9485a4b23cf7489332ff4e58390488c5f7b0f438316edaf1f79d72c734cef642114c26e40896705fea5de163bcfd6318562b2f65f0b8333a3817722cb7fc8c449a64a7b1a0aa8727ccc9c707728a101abeb96059bf46837202f9e36581b45f30f74637&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520018573Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264, Name: checkpoints/3264, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3264?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2ff4a970c1a5dc9f727f81a8cc5f66cee251d6b02d5763b56d6c8daf42ce226544e5d4477babe74c3d6b28a962fcd053f851e90b06034c30b8baae6ca2171c3a122feedaf39ed31c7eabbe310e7c7241eebca260d36333ce5af28223b9aad5cade158335ba558c014cb5581fba3e51be8f311d3c2c82368d4e31131076947551e58b69db68a623b5799330866f13ae77d0f3acad61315d14cc8144a147ff549eba589b2455d4b818fba693269969278adfdd1277b9b2d044355c597724e623a130aba5c6e3d038cbb1428358c2ea70e8a6e877ecede91bff4f81c1df2ee3c025082116fe819d27f8145e132a04de00c5b5bc18d5c3384398edd17d619614221f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520055106Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default, Name: checkpoints/4544/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4544/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=15193fd79781faf32f37ef05cd42268e10c43a2e385e853387f1c580da017138ac9fa32de967910235c074c7c4eee890de2e2e7fc3226af28508c6ab9125d22b3d0ddbc6938e20f77ee304ae35aab0bbcae14681e73fe9f40a81fad7c3006b7dcff044d7f16d39e6f7169aacc9798e33a529dafad187a890dc39a5932e1869d0a9dcf2a7e6fb2a6fab627047e33a45661b476d4bf55e6f4d8f7b00e77613727511e3bcbaab08f968561b9cf0460d0a73271cad21378ccb63abbac8d825d4f0a2b379c59aef705c1e14858bfdb28442073e87183badaa195a4431e1291797930286781ad48ff2662c7b3e2a42f4f0f6578a04aa9cee806326616da9b5fea53cc1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520083536Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/d, Name: checkpoints/3008/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3008/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9e8f47dcf159cd756a24bbee80c2ca0138fc92aa0124248b38e38aa4f71a2ec3a9f5ebb1e77914671d688b0e3f7f722205342c44234f6e9d72b5f058ab561d31775f6c950978e397bd3a6038cdcd99ca29ae97fa3f50ba6353c4c1c844e5e73d61c9c1f3283d14cb67f3fafe2485d9ed3e3c8201ba5055fe71e972e85921228df1edeb52e7dcabb8f2ad2204f3d9fc05b09c2a169a80e775c5eb5e9036f83c94315c40bdd9c6b34a3dc05ab394c49ecb5ffd4978a685461c2514e0ab4fbf55894068cab3f44247d4f803ed714783977078c5763f07c9702adb6d6f9354417b9751ffc0cb877a494ac7154009b2b6bbf84cfe4572e73bd38f0e9ddf65b40baa4c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520083503Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584, Name: checkpoints/3584, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3584?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0c512688c9aa66d30d930195a070e6b5e0f72bdab1e8270961d110d52a696aa4cfe1473ff2ff384a1b06653c523faa5c67ac65b5136fcfce6cfe06592f882d7cef39e73a65f5638dd53019d846451a0b62f0ead886d6e811626f323ac52cbb787a4a8f688890dc7f6a25f2ec4d70164af88d3479f3f7470adf469979d14508a3e48af65b4f0d8746b99562c8e6ac59de98fb0757dbd80130b40dc47a2769763dbde8c7e964dff4297c55f9c568e2d71e0b0e36e974f83f1e712a462014fde03875d4287707e2a1b7ac10806f253b572fb4b8f004868465b37d1965cbd1dc55f3a32c3167c9643371e00f36044e163b18dda6569ef391a00df770cfc3b0731d9e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520083566Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0, Name: checkpoints/4736/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4736/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c4057306c83db7f0120921ef4f212dca2e3fe4769621bd17f9f0521c27cb3d0b66a7fde78e921b37b23321bd2282212368d756f42fc3db47e4d5f089d80c933bbd9d769da79a7a16ad6a82842d56f94bb37bd1d353e53b8f4831fded37f1a2dfb7db50d15e0e43e14ddf7560ed314c877aed6b72b5d4a7c8fdf7885b1983f66bb291716205aaeeb7d6f2b1cdcbc89fa4880a500e97b17d564592063af8b2aea7833b8ff3adbd9837dea47bb1c7afe112b9bf362d331830e8606949ec6e95a12fab332df033cda66a25aa895dc3069ecd22616e4bb4ee2008d5186f3fe48bb1a10c4d97619a12ee1d4bcd2006ca288dd33c2c0bc5c4c31d318786c12ab9253b7d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519877121Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0, Name: checkpoints/2496/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2496/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=097fb898d990abc1a282ae9e673ee1bd80e09329b6144d0a6cf833dcd8af6a5a023279419b14cd03865e0a852f63c7f913b924cec4211493ea2c6227000ee042634bb57e830dc0200238a5569707a03bf892118000ddbc211d6a6133d728d1ee09de491e058c1a7b483d0a3640bd5698f01d35a9f390bbdc9842180d44c2e94651dc2f5054e3aecaf6a14d2906d11fc46dce317c09d290a2232514eea8a6abf11f8c3f81a2193ac41925c1f619c595ba4b528a55b566195443c5a131402013deda81d143f8011a9ee0b8343f7d82cc98ace66704b6442ffe3a8cfe66ec2abbac30dc293485658bbcb73c85c3194dc058ebddee0da4395107af3b88833be4eee8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519768763Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default, Name: checkpoints/576/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/576/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2660fb7a8c9bce0d47c6dc0199036465e578ec9db9a48a6ea8eca006fb53c9d599d049ed923cc16bbbacda24d2db6344bd4c435229b08b9deefd53c59cd9b293a3845ca65a09b2fbf57544f7b09a3f9a4de481616157d0b9c9b0076c4cc1e634eb2f52debe2efc449e389ac6a2f47f28799ba0528b9884b0d9e441697c9a13e6f4aa0b9f205de1b6387a13310e21addf56bd31a6ab1ddbab9b8350c356c018dd333bcf48162a5babb0d8db6cec1182a66f2394f0c17a08a5a7c07a27090aa21b21f780d184debb49f8326e1004fd9ff8e96ab8586d93a245ab02f75fdc94d9c99b242a0b6b462152d651e589ce82018fb5fbe4def2c459d74378d02cbf132e9d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519969293Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default, Name: checkpoints/1280/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1280/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=ab11e2b26e8693f91a4d75765b17a73bd53c429221daebe354fc93eb05726356cd93269fae70faabf45e3035dd69577f907b01b68fee3f1d875ffd0083e2cb8a96b696d0b74d5fc542b45b8365c1c78a65ffba549b0ed459ed2fedaecb0b981bc144f7f9a0e649f27e8c5f65ff30adba3af99730b1d79b8e2384669b9cb51197faacf283503480440eba4b6a898f0f12a8220dbb7c35dfef79266f1b711ee84b1b60647da23589f5288bbf5f93bdbfc3c97cef174ef8537f7ab0cad1d996693412b6853077a126d19c0e13460f52e4b2aad771f413edb4417a11f27e702d48300b4f1c48ba16d64926f8783aff3c3a29e81cb02857ed26aa85ac24003f08622f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520219283Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0, Name: checkpoints/4608/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4608/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=30cb1aba890851694eb2cabd8acdfe255bb38f2b2998b19cb8cff9a5625d39b8e78dbfa804f170562ab41011f38424582fedc223c3656ded1448f32ea5f7ec6a78d8f6f8c6c99de1b8814ec13cb1d524a327e3b308aff9719e9f15faef1eb5eb687a9ab3e56123da61c24143e864dca8a2e6acdd42f0f79812d0fdb07c2d01762a4f9adfc1da1a7c3152d1f7912acbc0cf3d102ed62ed0d7c90b37022e2dd6c0f72613a0c38f714c11cf6eecfa937b6a03ad154869879561c3e591875e821487278c074223fa07c158ea237a975bace671fd3370c7c3e153b81f323d310e89e1dcd0bd5016a1d45452f292ba2a83b1acbf7d1b886fc600af349fca2c37247366&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520228027Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/ocdbt.process_0, Name: checkpoints/3392/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3392/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6c10e853a2889affd4410d0f9ec2336248d06cf179388404476696a2f5c2c4a30dbc992343d571bb9e9af32fb9ad28a09aae66db4eb21c53936727fd8e304cc0be9abad4742f31b66963738df7d139dd0a044d9a8533c1ab91735b443410b5670f4d38d9b8ab642ef63369f494184d2bdd4391d50e428f553bd9c5d45546b68961441f178fff5e9bc5e950491de111156bcfef0565e893f4d034a0d2516b994e5c481929e74addd032c4120832c6af8673524bfe75bd406965f7a2e36ec8487322cbec0380b131883e02e1352e8dc31ce36c12c8cb0f3009dbb5fa06f1a54977f981dcef68db2d562d7e52cdd7d54eaada0b9fa7f1ed98bf467dec3f0105cfee&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520117176Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default, Name: checkpoints/1728/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1728/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=ac082f43966cd5b00aa06637a5ec814ff70a4a71e4dcee241ab5413f08cf6412c89d6e4d94ff78d126842f6000c352695131338e719fdb95e71688ac99e74353099054eaec0412438fd60b46b4f4f9dc3d92bef00c3e068536ea68c4227502270733462c1695178ed912a371eb812e0742e4c993bc7204313793a12a889f9baade2f8e0597ffe271cabe3cd91272a9ea7dd964cb65818491250d694b34f0803cb704ebe392b5d286c0726d8de84fb98fbb9736d28f634533a2b5efc7a95d1e6a8def0d0f8dc8aeb7e92b8cba655e0b5f6f9cdd38929c5284de2c3bc2751b1fc1f40422d5740d74122d3c3595a568009d2c1beab8f982218e494474bdcfa0cad0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519345223Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default, Name: checkpoints/4992/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4992/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=45e446c298d986c964da8b9ef521afd62942351802540972a70e6f72b490b73436dd5531161fcb918250d8576218097876a9a603d65b0df43698a52ce3fac0fea63f2196d8b1dc7e276259290feb58d9b13450bddd2a55acaebd19fd1863b5411f47d5efce806230324e498ec67ea258dfb8092959911ea9673873da91dc512254bf95430eeaff740a055e8bff0cdab46ff75f2a868ef2220ed71abe530102b4fa2f3bdf63d12058916230fef0833df2aeea6e96947f28e3b6eb33a3918f7660cb7f60a6dc51761ba7ddebdf67aca1f4136caa3be42a9d6e2fc01682860426b7f0efb6ab57ee4f60ce5b1e81f92bf961f0721c744d83cb24d326dac9016d2e13&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520300994Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default, Name: checkpoints/4736/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4736/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=78597cdb4f4b0a08df54cdb82240ba1bdb2d79c410c7b7d6d74908af59eeac2f2831ad29985c37805f7256ae4251403b1753b666377a66e93dafa25b83fe35d9ac2bf572b82abf58b83e0146e1230b4b7a4e6ab25b54dfdf9d9fb5044873622bfbd8adb5f5f19c4d67830c15c9ba72764e96f305608231889c24883d2edc7f4b85a6301f511e2c53355973ad8c29514a3e94edfd96d143433bd80a4af5fbf7ed05c1737ecf54da8bcbe2be46caacacfa0006b614d5936aa40d4d4bee4dcbd354c5b2956f40c818a63854d634d30fde70bf261ec0ef13fdb710f19080ab95b3e2d369a4719efa551491282fab0411dba34e216a08c3aea7de1d62df4f9011113b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51954855Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0, Name: checkpoints/3776/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3776/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8d7307a95fe572b2d029c0c90a48fcad3b5e80f00da72ec692aedacf529fed480105292da0da6952e67e63fae47d6814e95b9f245b5df34cd021e9a2f66a8f363b2145cd1107c6f37c16aa00fc70e02f1677b89fb9bda6bf3349614a5efc843f87e332c4238f9b3e56982b70a828d7c35c9968a24cf2b6bb384e9de764249d3ccfcf898aea443feb06603f38912433f5a592ab3800f5ba10c3d7a8a8b4ba3fb878ca7ac398d4a1ee2d1ac2c64e43010c07cb3b61472d34d63d79a9a6886cd949efdce48a4a8bd95e5f5cfd023d9a101ba36f4d1663ad2be2ab422d4ed13f0da9a7562aaaff3cf16a04b54895647ed966dbc6371c1c4ecb88829947720b044b1f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520336359Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640, Name: checkpoints/640, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/640?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b3fea5e286e3d5525767395d5bc902ae4b08e3e2f83e04bf90ca64d9fd32d981e5790656bdd7a636091f91fcef6ae7b0b3cfc039dc5e0e1e920d05243376217df352e3d48648c3265c10c05723ef2c2558e368dc769c12eebc63ee37ea171d29f2ff584e21b8ff4a573466622fa2cac68f96e8f8cce701f52b8e38361739fce832f8cb6ac170b429dcc87320250d591326bf7692caeba5a0a5e9760b7914aefbe90b6abcc98b6e3bd0eee9acb142d0d3b308dbb5f951981a8e6efbd6deed15844365321bf5374b2066f86dbedb1cd012af01e60f43eeb5ebf92a015bf1d41941345ec55149e98a6d2696df8d034d792779c3bac03ff2432f57403272ac6a7d03&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520090212Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default, Name: checkpoints/3776/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3776/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7da4f809ffc255c92d0cb1d3c50209ba6d18c24930883b6dafcc28b5ffcf9c6b18ab897a3d285647ea0bc56b4eaf73d0a13d0a684b7efc2c52dc239a6cfefa6ffda190651c73b67e06d0dba47b58df7325731e9713fde3b36f4466146236d2873cdd82786fb295bbae31dd3b0854d1a4dd5b0f2226fa33aa3438d975a88ff64930baaec87f48232b086075273a9075b0d672bda7200ca38cc99f8080c6ca2c0bb41edaec3149d7cb8a10b4244b1de8b095251f35d80eb26856693af26ab377581f02c091948dc9c5e2b3fce00230604fb1eaa9e16d60c7e0dcdf1b5677f17db8819c90b2cb87850332931a3ad07d8d929d416e786e03cdd27b7ef009c174c042&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520415967Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368, Name: checkpoints/2368, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2368?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6ac0ec5cb1631936d465e2f3267ffb2f613a85303ae73ff15062bb9bb22aceed621ddd9da6c29dde93ae188bad935c675ba3c131c10e63c5e20a557f0b6f6687f8025487c268e6bbf6af8a46dbac8114c2c5d3023e4b27d0b37b29ef9dbce6547cf36c58bab5154dd83b342689df54aa195469ad847bba8c9fe63c9657d86028f4dc67e4ede009888580996a98c284d05a36410b0c9f0226beae269eba3555e7931b38706aa784c7385fe037d8c1c789a63bd54e1a6a7f06606ca6e24fc22510cf1d7dee2453cec2ff33641b748ff1d7152e3326e30bd0988f745f24abe9a0b1e553cb14dbfc803dcd46ddf865404bba00fe74b3d2585a74530566b60dbde84c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520226116Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/ocdbt.process_0/d, Name: checkpoints/5760/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5760/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5f39894508feb26051af0f3df48c3f446a6019d6a150f465d82994571dc8fa9bc69df98e3ecd5ecdfc2957320448357001fa3edb9a8374711dfaea8f9d96767203670a37c65aa2e4e940ad1387b42a783146d004c224f7dac680b81b2ebaa27d99343c857021e2de289dc44b45a8d61d3890cddbbaa7d7dca7788f1d179ce097a9b9ec26520caa3bd7d9c73e7a528401c3875544b472ddbde6e67ccacc887ec68e14670a4ae59f5f5e937e7724e4a0c7f874c374ec948e13be4985865c524362c5e43ca4663fcd489c6066e8c2158f5296f47333cb1acd55ce24a3f2977f54a2b4b952f9bdad0d1fef2041009ab91290bced48041e15454b0ce3b95e01d908e6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520485066Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default, Name: checkpoints/2368/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2368/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=04ec5df1dca082e169b7b929f6c381c5da719336494c6c466275c5f024fa962efe491928e994a3fed627fdc536adad018096791b4c8d90342ad9addc367258a97cf410c365c42f096779b54d090cb5e6d68dc9a86e2ce48192e3b2a937df6be89f55527c1ec9d7106b317c490072718681fa6be75216d4b76e6924868640b68312d151478a2902035cf577a39ab549fd6e9d2edf9434c92acbe464152734cd63aa1a6150d105930d0c043d380646d222cac31f4eabdf56f52241fdee7be347014f7390334f7281ae565bc809cee4d131aabaf9970b99a8ad84175fe3bb0f8b27abd7c8e65359b94450c54a266cff67eac2bac85b5548129bc8af1f1eb4f2d30e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520495131Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0/d, Name: checkpoints/1984/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1984/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0ce957a1e7ef1cbaa80abaa5e497ff06eb1e4544f094484fe848e1c6fa4ce0e4ae527ab667f70d7721bf5542ecb9cb9ad21b137987562662219054eb7fc183c2814ceba244b2b98108d6fce0dac981963b22d9f861d455da50d122e6e982efa81212e6943ea1fb80db0f09fa19baddf656159af339e04a7bb00cb57dd77aa70fef1b6bee63c1fb50fdbe8218fa036f437e2ce05ba5f93dd9e554d98b9c7c441488e198de24fb43953d5139051d854df965466b7e78891926c5bacf4e45a0e9df58a3703abcef337c15087f8b20ac46f5670be1536298e4df651e194931c77fd4c469495dc8c6f908046f983bf4b010f10d12e11cf0e838dcf22463fb051eee90&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520542095Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/ocdbt.process_0/d, Name: checkpoints/3456/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3456/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c47f59159ffdafb09f3d6a8efa0a6d8b63d398d841d166b55d4cffac8cb997b52d7d107578b16b2e100ae361314daccabcf873a9402446c715ae7275d7d68e3438c44e1cdc2ec06f007bf43b88ac45465a31107f8c377f6d8cf1e4170d484f36b9769b96dcca58ccf488d3358cbbe76d59628316fb698d1aa141eba3ea9921b31e7b4c99b75a8a9fc2a93f9a84780214b630c1888ec19712d8d7e28d2592c11a56ed59510844f7fe88bb8c6ba51b699f0bdcf76ff5fc06bd1f3c87f2559fb20c76ea26cdabaff45f7494f2164791fe379af14f8cd7e143489de735275b8134f5625ff20ad9be2d1dfcf659f8601566071d60573677749ec2abd71e9ed46abc05&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520562009Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/array_metadatas, Name: checkpoints/768/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/768/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=07f0580ddb832a5b4cf61ec5e75958acda465a052dfc235da4b0cc7e740140fbe383f14c91074f28b40657e77a70b9ab1955b394665a14a923551d7192927bd606aed3510501c38bce87124a9c880c06afb5905b5c28ec607577e001236b819a2fa229d8222781f12cb9aa4237f8df574f118ba30185b697ff9ef712dbf3ef47ed135ebee3c65ebb306c1071213cd9084576a77ec361082b64ed47bcb31ab022696f5a0f688d95797ade33f5fb80e63a6ae65d22939558d89472e614dc399d989ce71285538242aeaca0b61229d619eb74322c328162e7794110f0490523ec5138420633e5a98c0e5508109cc6993d10dc480d2af3f9d5f1d07a418d301945c2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520573021Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856, Name: checkpoints/1856, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1856?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3624ef657129a12df116bb73fa93dbaf90737c0ed64456ebbfc620021709e64026af1837a7e853b9a1e47c70139fabf881ece71d463c417399d491dca727f9c5d4fd51a40f2ac3b2a46a0a5b7f2bf3ec6d5ba12bae9a157d10aa016dbb04adca1507a4daed9b10642d6aa355c0b30e43856afb962d3cca3563617c27a661f7610f3f2a51fae17d81ddcb9768bdd9fa56056b04b00a94186e7d04110e5bd83a6dfe11cf6f30641f7006b597aefe120d9b919f4dc1dee85c5f9e392fa084d136c8bcff0fff64efc3a989278f9df4e06fb6c5a6efd1593437acb68e7c24dbbd2516e331cde6359c4969bcc3c6a9b2c03b227a02e3a58ed1edaab6184d6167f5dbcb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520630714Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0, Name: checkpoints/2624/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2624/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=be7ff124391e49feecd3db640f3b35385f9709117dca6719d9c55aa5d94fbadf4acad28f1574054968b596648c0b8d885ada572dcb2e3a7a17103738fa0035aa1c47e552fb6b000b7bb854ea46fcf8cf0c5545990b21674cf0ab3b679e49c7e3ab653dd434a5d97a2191f1993378c97987b471fa9d7b08d01a3939dccf46f79538357bc5d28ac72cabab9f3e211e1af147340e8ce51d921be0c8d5aed24e4a8b10c2e30ad7354d1a121d6cb9aa5c81757c84106973f5a6aaefd2e793a74edc9ba5e8cc656fa24ec91e99b59939d216a2c873ca56730438d1ea10f1c7459c47931b2400901e7bf61b96b05e6725d99b922eb584aeb1a8275a8964f0fb77d47ccc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520317375Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0, Name: checkpoints/384/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/384/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4630ab26e5bbd6ddd1a6d5747e9fe5ba631b429734c1d96920ff8f3032de9055fc5be5792ceda8984350ff15ec7be773b5a27cb378e42de36e05f71e06e649bc18a081b132a624c8cd8fb825ef544cdfc26842f5b011f4ee83f1098ad8a56260c7318f98ea9daec7130c24e4461ab26d51e7e3a3ff9ae13c7cc2b34fda00a06ae2bfeceead848594e2231aa752b377e832b0fbbe7db35657520b8206a801ceaeda01c8744f3be8cf8972e4b026c9891d53739bece981ddc8846a062977747020d13482ad62d0a4555d47b28d28372f38c0385dbc0d4f9ffd1528e8070d21689e1bd168dfa0fb00faea55a262f9a337ce657323a52182114125f5fda4d0d18d64&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520651531Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/array_metadatas, Name: checkpoints/5568/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5568/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=973ecc69e055f2d49f9f34ed6a1cfcd0df787c9e22e99642e733b3760fb21c75eebd3b8afeccda4c96ee5aab126ed780daf1552589bcd2ed2a0579eca21f2f7222f95e32ca2b3abbf7dfbfe23c1af0d5f692877e456530d57d3a7d65e9450a264d42ad6a1fd8f1fb8d40ca4627a52d3edade30d1f41287df5276386c04a9acbf6e2a00a0b78d67fc03d89df3c4597fd24b9ab1f030ea9ecff7b9c71b03d591a2dee1f12da8cf89a052154ef84c633f568e48c167d497af405ac90b301a40d42de71109fa205c2f10ce9a4ee75a3fff3c0ececd1475a9da9cacac91d07c77230f5cbade66aaf748de59f5bb5a9dead894b8d5de1e150826a87a64f2214447292e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520678244Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/d, Name: checkpoints/4224/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4224/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=10cfebf668a7949e8e424dedc755e8b505cfba69d605e0cff7b3cb1bcd9fea532d5c7169da498a5ef208f263c555d70652248aabb46b036e6ad2f161eeb87d08be8810e549554c4094bf8f940f0f400491c7d2f0d9b00c641922fa9f8baad97d4fc40a30e83e77a63ff0f7ed7848c5d4b638b813b8ee890c342a355cf115d5d5c1c0b81453946af4e0e1183b67ee28303c85a49d3f94e17c6ddf464c36f0ab4bfce09ad8829a50a232926a48dd6c2bd1752e5facb64ac806abfdef73d2c0da1340469bba417614dbbbda20e6fb3de1f7e7f1bde314924a8b3f0634025b457465050e572e74b8540543167a629bfb0e7abacc70fe117f0d23d47e16d98ee63190&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520561855Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default, Name: checkpoints/4928/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4928/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=83530307168c1292fb1e1cdd720c1afdede3031e8e55e05c9926e5a3e0158ed5f5f7cd9827b8fb6aa919588ebf6c56edbe9fba6cd1ac7689bbb4b18447e945e7d3e7cd62b7e81d4bda94191d10e300c6388d3e62e789e62d5e45466028d8f22270f509fb6c78379bf85dc7035a94e39d44534d30327a3a98409d6b5090b26996e330b0cb7c565c425709e1647bf5c75f89de191a1e2f656624a2e4082251a1d46d86ab302b6d58c77af36390121b434155cbc444a70f3b53f2c58b3dc97bc837ae889c71ce9483e1e14e39da031e092c2a5aa1f058f563603b5ec4e8ffd4961171497ede830549e66746dad67cdc8ddf7d6f557564e143b57135d119f58c0e72&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520729404Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default/ocdbt.process_0, Name: checkpoints/2880/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2880/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3bdaabe435dbba6746014cbd0b598b6629da801d9df4c3f81e4a5c3ec650b498ec3ec749f5bdf559786df3037aa4d1d1a974cd3ab1174a02aacd9aee0d68310aa6477ac026942b8a6d8f115f2d49e9b753e6ffc688a0fa542d7c709ebb913ef67f1ef615fccf71aa05f705d586b2016bd6cc66171670df7d5cb11af292c5714d2c7218d9e4a375f2c48de1540967d42d2a985ed5e02824ec296a209160c8d7c8d9e771c32663a42ed4366b7fbc2b55559922dd8d212e618bce945be3bb0e64e471e7fceb7a1f8a08bba9a3a1afd70c21ab4819749fc99dc0b5dcb80301384fbac77ce349b096492ebc78b80c839f119ab73c297d5c1d466da69db196358a0fab&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520743816Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/array_metadatas, Name: checkpoints/1472/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1472/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7c7b08bf5a45473dca887cedd5607bea6231c3b275ad797d777b07ca98765aabfc55e590d1481d81ec2c9b9d973ec4cad47132be8cf337eaf2ea6cc1b21df32bd88b983aa344ea214da87c0b33f393fb90eb74e23f3d5c954943dbc5f42b1ced5c42cb1e9ebfeb83d9649584d37ca044a451bb6993bcf0caf9727e1058aea8e89434c59486cdcea8c52bdefe50df1245d8b7376582ac64ef327b422fdad7f0dc056a9fe6e51dbeb3d4e0271f059010445feaac5e1109d7b786eec8d574bfc65530ea1b398642c63f33956e4a4ffa725432db547bcbd67fc1bf1fcd306afdcd9a9c0c255d2f328b62d371583b0c7b5556668abd054f56513654ff3ec9e1c04c58&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520752815Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/ocdbt.process_0/d, Name: checkpoints/2624/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2624/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2d0909b6e345ab1b442e14d951db480c36acde647041c6fbb4754a0f213380f1aea30ed7db952e00ba7c83a12936ea8be9494152b1d69b2765dcc7cdcf0cf78a28058ce2623d017b534ac7a5f980bd22508dd6126b6430016b979ea22fe744bd0a664766b182ef7addd9b1aff0344dbcedbb786cb1abe9bc8fe2cfbcb394c570f580dfc0e8c01952ba82aa6fb0b134685b432a198c255f0b00f455d5112e8888a0d913f77b54f7e83fe798572f40b53995e46c37585f1e5943c2b58018a03dd17d29f0cfd27071893dad884cbdb4c928fb61416168617e1466b5231d59ab3e03b9da0130f4a95a9f1e5f2e4a7a9d4a1589f77db6787e6b04421e57fd725b7a32&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.52078932Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120, Name: checkpoints/5120, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5120?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0f8d5f8cb26dfa7e57ed7fb2b05071481f2d205f16f19275a8a8a32a8822eb68a246047ce52dda9a87994ba3c1cb3f9bde0edbd5df0c64c6f20e8023cbfa877b7929126ac84303c02127a0d6a8129cbf87e4d6976521eb7394ee518049758e51c2e5839e559963605fab0c79f03c7421e115ae83cc28350533061a3ee1f5201683467fc5eb886dd27b68667c0bb9ac07774a9db829ae8f8cca10b757dd35ba1c6f9d0dfc5025f1d3173d781bf75b4d325eebcb118e112be33cda07b7663a2f8fce9cb11ac1e6e23bbe67b99171f8ce0149787ff167e5869c80729587fc5277dbd45bf5b01ce78c4e842fd09100f0ca5b62437751f3da8b5f0f83aa89e281b464&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520758813Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default, Name: checkpoints/5696/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5696/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1afe1937520203a84ddacce4241b6ed9de4e8aab5f8c18e4cdf778e2535b087e8b8bd84adc1529462293cd57e5fabbb20c23308a3a918ee5c41770f7a83c5323bc3af67214eab7edc1a68e78021d044191cded335c3d0b0e161fd5061d86f1684a43fb79c578b27b6b8bde434e956dd6b6d2afd50704a3f5d703b05c06302551020e7ac0c7bcf57c0546c35d0214e569aac506be757864d1df5f5ddb6200893e11bc42378b010b7a1a31ac64e342308cb96b17779931cea6ef281b65a3d17213dd918061eec6a260cb2dc85a1aeed44c214a0ad9626c080daed155c05a51aa6af040d882b08366f7694b45a82eac6ee812cc8c5a63d3d695d2ebd858601974bf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.52083672Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/d, Name: checkpoints/448/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/448/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=abdb68a0c4f323bcd12e6d6b66bd2de94febe9507842d213cf4c575a58ffb951f135395ed46d515b9b8ff90bdbe7a2a3ff63525935fe249a891d3dd373c51546c6729009920f986bbb78ed942516cd60185061a5ce97de453a41cb3114f9dd69262d1f1530059dabbd424ae3501e2ab8432594009864bb8b33ca4ce0d5ff4db1f645ebe61129e68d99f54c72324aeec22e08a970ebe7ce819574094c012d6cc7d5e6756c4d433152f607dcb3b59d0847460b8a4e14ee122c94f327a9aee03076b19c1748c83e407873dacb6f19b5a346595ebd3720c93d04b7f3a017e9763abd2804de8e03e414303f7d44682e582de4bbe940f01b7927c619647d96cae71c6d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519602765Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/d, Name: checkpoints/6080/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6080/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=79d650e06d1587e373f9123e852b964ec97d982a0078a9150f6aa7f236bef5c16deac153e7e580c92eeacd266a22dc274339d1350d58b0e560dcb7e93badb6afa640ef8e3a219de504f6f1eb3ba402beb4a4cddf1d77c55f4d679c3b3abb9252d6461862e1af766bb78a54fc984e8877047189c90f3f52417828f9600a6b9a2fc58d31b0f2d829c91844bd52a698ddfc6c7cccaca2c7769a1dac4b2f174d914c8d751cffde62b3590cdad2a591678e5b75ede3791d39ff3b232720bd7ac3da550bd0c2cc69935ef8ab4192cc57e764ab53623a574c6a769ddba43987eeabd06bf4415191b9dfd5564be1c3881df874b588cdb85d5f2064188175b48563e7962e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520754416Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/ocdbt.process_0, Name: checkpoints/1664/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1664/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4056412d6285ff9640496d084b91866ec5dc0063ef2e0b0f733d0ace927c730c40ee65019434ad4cf70c65b4506aeb7f6769dfc053aaa37887b260a8bdab3d84316519c64bd6b77f48ddc1b7a8b55541ee586c911d6679011f8664ad5de28a25e71bacd6f1bb2abd493da4e4e6e16dc2a0e949ab8722dc60fa240156794657af1e8dfc81bf3eb275603a78fb0adb09c9ee20fe5861888b6da52f1bb1ae41b65f0a64e9fdcb2828f41af1d54d6e7ae3f4658d534a59f4ea94ba43aea6296bd9c2ea9dc7209794897e9f15f90d63a7aec482912fcb5c4a7ee65863157a4786dd2e64ba53cdce577684778080dfde019afc033715451172d4ebd264b452243b2c80&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520863413Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/ocdbt.process_0/d, Name: checkpoints/384/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/384/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=30f76d08c6278e93cbf0dc37714b711832eb1fb201c55f16917107f0c2168e94af40b8310d2ae576aeddad419859f3fe5c0d4ce3d104ca7f56da0681c7dc58a4df603de7555c560e0cd32ecb24aef0efaadc20bb0c0250564b4cc76e1b4d1c0a5d218e7ab1f73f5f3442789ad19ecfcd7ba1283346ac8d8bc8970f8810c27551bf27c9e59a2663af158960dab38a7d4b479a14e1011a588b7378ff665d96520e82f77b0c9f18448087c6c519a99a27e78c0d4813c00ca5f07f5b1377d9c448be4338a1b712847daa7f6a1a55e53dfc649bdf2bba1ae2f6eb670f403713434a13312b50150c397754cf9ddb82f508d1f6afe613274165a6283a3be6fad5f60349&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519712692Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192, Name: checkpoints/192, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/192?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=344fbf3980c7ffba6e758d5397f3af2eb14c0cd7c39fed21000615de658df6daebed65c01d82bc18dcf57a90ae1263498797ca35be1cec6daf736606f7d5fb06fb2db92b01aa630ce9f78ad2ce59736e63a37ecdee4ad8c93bc3dab5b58c30905974e639881dbabed11e7d665face94cac07e96e9e437f473c1be3c6346f1b4d138728e7d5e26aa63c7e2789b82714adffd17c67f12c1c6cf7f084e55cc0e5e0d57a366b96ea35649f8b77f259261fbfffea15f0487bfa0e91971c545eeb2f1a7aa37bff7fc07fe3348bf072812651ef2b7f3e915b058aa4472bce0066cb416382d7676bbae1ab184b87f90bb536bf2e07c8fb5b82c668b12b285885db7df567&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520649895Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/d, Name: checkpoints/2688/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2688/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=43e189213f8a66641fc8ccf5a518c36c8b49b27126feed7c17e4df3c82fa309398c1a0b2965fffa971b2218f90fe077cd0d6def901e5edeb5746f65b2f3129ab95bdf552be735e3c7f0de9f5467523f77b1f93d28b4207fc865ab4272473432af78b3e6e5d7eed7e49976af55a4ffa9c1485fde5cf9c4f9cea92ad326f4093e34d0d7c8e8ed6aa37457d51f36848a3887eb05c9ac327d51c048f79e5b9f739ee40c63ce75dc3a5991963d59d0d65d1290144bb26feb49469ca5a4bacf08f64dc4874c894c3036fec3e776a55e4456594ed8d60298b260a3316d257c9c81d45c5757bc90a1a55c36aa2317f708bfb1dc521e52eec8e642e084c8ecf5d098fd5de&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520924202Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/ocdbt.process_0/d, Name: checkpoints/3776/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3776/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a9e48a2b24aae0882dde0a384125241adc3a864b71236dfee6372723f48019c3599271fee840f96772365fb793462e01fadf2e9280f54366ca05d40f40f21aec7fef6eebf47474ef0674b2c2e2f494b0bd8c0148860bef39a25ac7f2e7b13b2d3ca2854f930f07b386641631e37f1f3f7f5220f3d3f724c607e85744b9f61b13777310b1d964fa9d13489f03a6d2c08dffd1a67f970b6db17dc72ca2a68157aa15aa5d73d0f65a79a4b8760be156b4c33272407664d28f49ccd6cd73050db4e90982b42351917abe17a95b87d372f41e2cc866e850cdbbdd7d7b574d1d17ec42f8a93ad660d3375a10d4e18461f053a69e407711405694dd81f813b2801ba6a9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520944672Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/d, Name: checkpoints/704/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/704/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a45b34e8fc13afefcd56a3516e912538b5de310c9b02bf140dc2762214bf3503cc44c3cea2029737e525c44c12acbc82b485808db2604f412c2f5c1e6df2ee9ae0c27722dd18c2d81867523f26cde97f3c281420de1617e50e72472086cabd7d3efd46ee497bf52c114e906fc15f7376f5bcb068c6d4257b618319b416a09f323c39956d9511064bb9d6aa90d16e648e709842f72313b6af6a4f2284f53dc17a70f96f00d4b39fe9be666df5e5bbf0e8674e9ee4673b4ad385864ec67568029f9f4ed0fbeb3dead530d8581eab7b233829f073398ff627cd38683b8336ce6aec2242673bd88f77f2d834d49b9b3768dd17db234fd38d727525ea772d6175c0ae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520970189Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/array_metadatas, Name: checkpoints/2560/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2560/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4c7cc6e86b0915f09a5d4cdb310626d2c6d8633c2fcfd36630e4ae1879a67d6e748f1fe778d68b4fe0443ad82372b920c1a8905734c056378fa49323674ae95a80b137d5c04130848ae467522c2d5d3028d5182d2a15c0023e3d864401fd97f9f3bf43d5e735368a52aee60103ddc75297adf5d0e31b7dd3b7cae2d81d2564c79d5c4ec34c28430ec72372897bddeb5820abd56fb6eb10a5853cdbb01b26e34607f353cbb83916e1e42ee2a549cbedeaabea9739704301afbdbf7e23ad2782c74c76f92e3f54a524e6072c902663f009fd99c19b4cfded8a7ee6b941c55c3c6b8c94afdebff75d37ea51a62b1a48c7f2ac25bc063037351993ffcdb6da65b78e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521041524Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/d, Name: checkpoints/1728/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1728/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bc66a37c88673e1eb50e25763ef9fec9077ad3eac436358f473f3d72f9d09ffacfedb6dac44ed2f4b16f66552e6c27176ae86c1f3094238a1091f6b4b9a3decc032cbfe6a6ab617d87169b52ce66a95cdd12a588660c5161a926e28c4cc2a40711a2458e6df46dece120be6e96c0e380283f87728b3e2e4faf17a2c7ee1cb65580cef5b6a6ddbe9947409b8ca6fdb912186a1eaa1aeaf97d8d60dd7c644006f5e3e13802428a1c9842fffb5f7427e0c1be1bfec53be9d16b84b63a4c525cbe109f58b9eee1676a544f5f48679d534e3233af97e0cce8c03c802fbb726588d289f29041fbc9d60c5bca032b9e7cd7f2b19406d00c9de2fd9ff1909f66cb10c106&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519505432Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/ocdbt.process_0, Name: checkpoints/3520/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3520/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=aa4d6d7cb2434d3cf97528a59c4569b05f46e81f20bff6d10fd3660141f82533a6adbefa2fc826ea12e975b2396367ed8d37b224f7c9fbcd537f54cfd9f236bfd9797d51a85ee5c990f1a89e4d51cad467bbe9dfb4cb3142cfde6f5e5962c1d0ff5193d5c0783f24588007537046c9d1d165e0a2bbe6b56037497cc2ce47b4a84e53fb566ad2d31fb7d79dc75005616ec85a286eb9c4f07481c0b26bf323fa9a73945c4310845f4dd7094a65b8525584975936d1cb7e698630dfb92683a3eddb88e34a01151497a7c31be080c73edd433ce56fbaea8bc2648932dc8b58a535792d851ece7bcb3fba1968b1a3925ef08936520a20456bc19d01cd527c605488a5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.52108671Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672, Name: checkpoints/4672, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4672?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3320b0b7d2224d859eba33ae87885c28229a52a27d87401546ca1b7994229a9d7f6499384bb0535e275bb174d2c7b16d4f27bc3548a2c84b26ec08493dbf4ecf0bb54cb2bd9144764b51484ea8ccfdbc1d09dbf2952a9c1ad729d211d268e6843e7e215a0f027ff0f4b14e9d3e4dd62f1b4eaf588a9d0e198b8eadf99f87f33bbc4e3881e3bca98898e9a7e253807903b72319a6c351b4484173119e42e65d8c4b4ad874a7e58b3a27f1063fe781189160c5d43b9beda92e8bdd5e91198b229540f76d5e1daedbb1eaab64e0bd1e0494aa4f78bac76be82afda8c67fe310ef4dd2841a65f13ede066594b6cb4dd6a887252a6a2fc7ab54866cb962d5b5c3e7a9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517712838Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/array_metadatas, Name: checkpoints/3584/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3584/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=000662c4e009700e3c06b08bed8646682accc94f95259be92bde40a94076bcb2b9c09a5c5a86f0810e71c0ea30a1cbcfaa98dfb2f6861c3f826350a313c8e2ebf0d4031ba3b8f43ab32da683c5b6d3123a99ec004e8ce8f1e35d114235fd4963511b4ba5646e285cd2153b13db1435f617f6afb8b1bdd224b5ca82e22d5725182e4dab10ff4fe5ff16dbd87fac0853719779735009b86947ff970655ce7f2ef4380532e66a6e2f2a2f5e0dacaf767e2f2067b4ed7f1dfda67ef6adc34a61e34a1523beccb87e91f419b84657fb003fb9cf04670f10daf360ae03bc84933c470c4d39939a57033f6d407c1864be882ed5f94c3957c51c18b90ab9b57806e7ea14&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521066964Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/d, Name: checkpoints/256/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/256/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a8ff70c3a2a59557343f88569cb7a9ecb7b5dce53549dc36b04254af8779f9f4c0c362fb8c6890a72560b87ba4ae968bbda6adfdd6573a124168e1f521ce3e33ee3c0485b496f974e1b376b3b3d474aaa377fa5e4e61d3193e67fff8775db5c289a56d07a6fd54a9d5a8df737a881aa927ce5d9b2ec09b73982d4dbda8cae1bb7756e315b98be684e495d7aebf0eb269e03ebfc59fa2a6aaf4f7f94fb203d14d29c1f16d1d03ac537699eeccd6efb3f0b31c701d3d5c179d09934206cd7029ec51f798d69c2a2d098f06e63f635c1f8dd5b6d5a540df59e083e288d4ad41f7a7a2db0aac1ebee43d24612e1f6d53fd48c038932489af25fe701d36c6c53a85eb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521131374Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/array_metadatas, Name: checkpoints/5952/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5952/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=ac51d4e5f60bb1ce6c29806200cda3a94ef68175b50b42b4bd01d3266409ff92428071ec66917fc7d1ce221139707ec3fcdcf05ea320b76f4d731956c71198772b3cf42b1757e81f7b9e515db717bf1ea41ecf1fdeff76ad4329bd82b10acb609c709de1f3f631dc2681cf5eb8ced5ce93ba0a63d2f4f68fd33920f6b6c8edaf4191157f4245e902f98b3d82570594117f69655ccc71b21ef16d59b4c1b59c7fc5ed453eb8fbc2bf6d41f19e28fdd855ecdfa7273dfbdcd208c605a49e6f2c7059bfdd83f39e53803528fb702776477f35c3bf7c6af1dd0c25a1cda1c6f01dfa698d0ceb2a3f49b6205d917d72623897201f7660b056f0d36ba7eed2c562ccc7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520962454Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/ocdbt.process_0/d, Name: checkpoints/1408/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1408/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3b4640eded7a64d314555875d7a978e5b636d2ffebd15340fa66a38863ee69a95aede65f781240b41b38e0a40f26decbacda832abaf11fc3bbac3d2dd75e9a02b96982bdb7e682ad15ecd01508b411a8cf69c7b2926663370e4435247233f5bd5ac19f0c129dc6f8b2c24c6ab1436d82a9771280ffe5a7ba96700824dbd3e790544b70fbcf9fedacd3588e1e6460a0e9333c4aabe80e0dc6c4882d8e71411f4465760e2d5a4c5409ff927327d69c7a67aa7cce6fe202caea9b4458b1f6557e738919e82228e5c27412556f86be6172a97e8d531c5da7e3cc772f04bd2885bb11da0a7b75d8ebcab411de161d108734a3e70978e4750ea84110c0fddb985a8367&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521197679Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392, Name: checkpoints/3392, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3392?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7cf847444ab3dde1aaa780808cc2a68a459c74ed69e001a632ac886e608e2829ac4c4753d120121a6cf5c429333721116e577e68e605750b66e632c0665ae7c0d2ad04e37b4d07f6ad35304a9ba27fd9b292598c5b659ae1fc391d04f7b9bca8991117001986a72cbaaf7254e5c6516d675c9ddd2ee85039555155b0b5fb9ac30d1a52cbbaadae75973fd167974ff8df3368972ee2ae05c168f341d995532ea7b7d80a47645154a760a2dd43322c32d36ad5d8209159c8c55f77b1121ad5dbe9a5a179f83e5749845d78cd0a62cdb9353d507abe39be0db44ac3f7d18fd55cd119af21d2fb827995c2cdfdc2dc3938f9b6cd6357849ce6985a2ae55a1bc5a265&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.5172366Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632, Name: checkpoints/5632, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5632?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7633198d2be3cc7fb6d62e44406bb5e02803c9034f2becc03d41a45312caaddc1833a05a6aee58f8e2c6048f3bfa8d595eaba18bda125a3dbe1fe21098e0f8c068137800d3905f493183b0d45abd33bdb5c499ad3cd670f52fef1cbb29ff70179aa1b7be617d2618c693129628e6c95f0c00cd6d4770a91e9b83551a3f0265fa437d4dbc2a442c67e6d2e3108a264cc17a25582d849f94636304a31f88a5b2695a8c0a0f4c56c2a343f8c716fcfb6174ec62fef90b5d726cf37fc291216f0da7c7e47ff9ff4608aacb3f4f01c5f15c6f7a5f1ea3a735029b94417480a7b1030a6444ce0a559cf6fbda93942ead08aca3986c1855c50c8e261c64974a429b5d1d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51729018Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288/default, Name: checkpoints/4288/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4288/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=855cd6f016cd580c1074eaf5668a7cfcf4c9e9932faf067a1bd4b839062a03000bd00d8a8b6254a176de1dbe9cbea5f818613d98f4686dd4963d689996a744be765685fbaaf380a86c73501bc0ff2a17edcb3902c02a2d1329ad03ecfa739444df4570eed7294a76284d29ea4499ec3182a4924f7be5e678b7d9fe116c0174f50c867c505e6e7216fed0dded57a4709eb51d41c4ba96f5ce3da7d8322e30dee062fa974fb7dde0af3cd9c5146f6e1b299c11759e115ed3febcdf76d73ffc3cad051c062eebbe81b680cd705a39ab8b4076627aad9238239ea199aa1456694d3c1f8c437b43b15eb5f3e3f8130b6d6d544bf1dc74b5fa564e5dd809083d951718&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516725941Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/d, Name: checkpoints/3328/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3328/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9378a071420e1a01dcb7f6c9a4782620559faaa2ea74e1edb0c58dff3a873030c52ee05d2ddc2e4883f2e93e187366f28be0d1a4abdf16bd93b801e353f836593fd90757b43f889407cc2accfd6d7e428ed31f490e9ae21870f407d490df2f9b508a838f69a1ab24ae263f4eb408150d19f7d4ef78dbbdfde9f7e89a077d8f00dff48a88bc1e8aa750480377e144e3f10ed8678f86cbb7eed0ddf4f39adc5d8d0412ef8cc87b9b00aab902c257bbb42c93462243da934ded1f753fa321b63199e3eb823149fb09347569ddb722fd26aa79caee25caac22ee5b41948b8b1edacee4446e370fd50008d563a2683453434f00704f388f5a261c74919d63039ddf0d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516933191Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344, Name: checkpoints/1344, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1344?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=21b9dc8c12373c3c76da01dea2a1c69bf461bd307fbc8842470eb79051df4ff8e0aaaa4aaebd53f4514257d2a9e21a46b3973636276b47157958f07fbf6d9fbf529841146d09fc16bb8577808b55233b449f01335651988c857e1e29d20f520f06602313ebf73958435742e12f5df7290e88a22d44707b232c77b8d0ddf0fc8d95fab2507076ff4d950941f7213429773e1206fc29e79642b05c89d5dcb97cb63725f2309a99fa1130412c9a63ff0edd447f04b9efdc913f3deab1e5c44d6f528f57c7a515907d53437f566d281dc92ef08f0b972c6dab7776f589237659c1c27755bd2b97c0b67e39a199357825c50b43017ebd595c090b113db5b0afd0c39f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516582846Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0, Name: checkpoints/1536/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1536/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=ba416798ca084a521b523c85a684cce12b79d1f28179f0ee51e0310eac0e36e0fb93fec515fb9c7a5ed12e15f97b95c2d9604e2163d7f85bce66de5750f5519bf118306fcb7cb22ececd2760097b9a4130eeed704aa8a62db488a16ff3c83df3c0ed2bca689259a02429abffb382a168f918f1dae92092b42466e8780b9677283dc4f82372468d8797a8a17c7d9c391f6d48b0c6ae108d40db9dc35158a32426a90041481579ae9685c8ce9b5359bce35cb5dd667c6720672e83bc765b40c26439b4176666b82587f186c486c339ab171b8d9e92911178b8e8750ff478921b3601adfe2bcd834ddc3ffc911bca4af160b5aef3ab3842418f1980435306cfd9e7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521144881Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0/d, Name: checkpoints/2304/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2304/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4afd74564f24ea63d10f249e4d241171a3f6315c29b5976874bae83202e9bf0e2716739d62767f4dc4424df29c145e04c1ed78e45d8789fcca28f601dc941f1b72e6c1c9b8c787c4ef61bc24805ef6d314e820c7fc81f9231db0866a342bd39040519313db75838feb54bf2a15f58207098fd977d19fe723d886e557fae6f9e0d69bd793e332a91464bd17bbefd56f003b02ec8528d2cc7351724f0b27ae824678b44c6c215385d484896612ee6073aee2096460ecd549387d166ff6267fff9e8655d26b496d2a815896fa5373b4f0b61223d4e77a59eabc136cdab7196b73970819ca77e1dde579240e2781e9f3245864c56975e1b474b192b22f54c17d7aa4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521257784Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/ocdbt.process_0/d, Name: checkpoints/3968/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3968/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0c2ee8876aad2db9c3bf3267265150b3a7c763e4421286a151b45dca4cfed31dc65f5234a691ae3d73ec8f0033e2e248fe93d465a7623fc03f12cf8eb260a57459b1136a0196925844457bafed6ec3a14f3cf10da2a2fa7565cd012cc6f639a1debf689b0180fb8bdbea9bceb886b5aa0b03d41c858bef29836010a9b246bf1be1168e390f06c869b0a868e68583430fa8831b95eab1536c3f18bead3d748fe6b78c6e57347dc5d9f3f6b39ae9531a68aad5c174fd240cd797ba43ddbea38d010992f4229c9e7fe4f6514c120765747999a9135a801f260e5d7d0a7956a83001cc5201cff848e24021d787d1a8bd8a8c1a3c893fbd9fa5ba7d7223eb420a0c67&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517638185Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0, Name: checkpoints/512/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/512/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=49efc3b1cb8de0a13cb1bdcdfbe9b289f620b73cd575244106c57938a834aa25e039b2479874ff01d615e6f7fdc3590fc87a95f38a7bb5ea81cab4f4c8d34a7b68dbca34c26551e8cc190722df21c24b2fdc78c41349f44d8de83b333f65e0412592bb655804329650a1f91f0e8f2aec32322dfb6df32dab193cd0bfd5adcc328efe0cd987c6f883b36ad8f005c80f614e04d21548eafa8ae6ba100efde1e3809c66d83d755cd357b7f3bac1ec9a0a871e3ef70224eb4d5626d4bb9689a58713389e8b03d9e73b4c5cda27d4b3cf5efa3a01a5a5be0d677c9b5cf1508934d81ef2b92a31809dda788b6110408a73e05b103c59990376638a8c2b416e51e5a231&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483105408Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default/array_metadatas, Name: checkpoints/5888/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5888/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=70bde3099cd389efd197a3886ab7c26f3db732d568f8d8f0fc057cc80d4e267c2312c067043e6f646b5c0ac40ee867b5d2a34715e8a432eb9d83a9006dc81be6207dd14b48c8b800dc2f1dd45e7db3d10388004c0945cbf4bdad30e62f99ae371c801bc105c4de54111f8a059682cf06edb41f8f0a62db5e3fb77f80ce9329b5954b24d6fae8e69c49d0dda813923f00957f3aeabe595f79ac5b09fedcc3cee07017b00493ab1f06ca299c2476903afeebedf799edd653284d87f616f1f17449a9ba0943d95b68b774ccebf75cf45e4d975822bf6b11632ea109768638746640b1288988468b95d2737b17e4205be772ef293a17e01d092a3fd0e130dbd8d154&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483103665Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792, Name: checkpoints/1792, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1792?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3f665273230e81fdaefd719bb89ce3512dc3223b9b028f87d16ac98b21e8b9438c539156ae169664252553382e50172a8064f7bbed129f0639022497f5a9f86bbddaeeec4b0591b74358cbc64e9922d198a4eaf99bbda3e8d78db9ad35dbd57e287199a2a0e7f61d0a4856181e6a95b5ed75ba5c107d1f58e363b375dfcebe84f16ead35039ccaaea05d84377082b898c17fac46d60fecee33d00f5f779b15b1141ca1af4f40c6616b6aa865928be38bc6008649bd151c020f9115dc514d052953bf2cd98df0c0decbbefa4711f7d7888327c5c2141904d7e06974139f0c772c4401b3a9c06dc6af534c9737752d5d135ef946c4f4c026f5c6080fa1fe5af661&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483101761Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/array_metadatas, Name: checkpoints/1536/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1536/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c257ada61e2769f05263b14bdac2b656c9bad4c216ee9dbf13895e89d9a9d97217a56cf8e9e1732503770a9b69e539882539809f4220e403f84dc59718184f6eaef45bfcadde657e571a55de0e4d6b0b45c53f84b61894b74f95761540c65d27908e0c659639bb5d22e5920eb6eb6a26ec606823d3ad239d6d2c9851243465a084022a62d5cfc3d354f538e91ab12f1b94ac36566146d6c5243a9f830d4a5f049a22630bfc8b770bae1cc8616e52f5c7998a18cadb1c1253f3fce97e6d9f5b885cc7d099136081f4a18afbd9360e9da6cb53d9164af59f4c232b122ac3538239b29fe83f6fb1a7b9e9f1c68e2c88c313a248cf6820c76a51ddbb307073e09998&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483090633Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default/d, Name: checkpoints/2624/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2624/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8e4fb8431bb7e79a3d1aa90a007b29d02d7c5bef755403830e6682c8e51c86f99c0c96c74c4090c4da2cccc9b75b481361289bd17cb8f8aba89b739e46639e5ad2575e99efc0d9acacb61840206b252ac734aafa7458a4d630a5c8511e62a3a923d8ea67af6d49973ffae7b79df76ac3e198d39b4a1f7730df9673444261a79ffa9056ff5a5a026254e53857814ee22a0bd577bbe5ff18db0fb1a0f5bff44784ed467caa2c6c8a0e83d26d0cbd348fd04f21f844e4b01c90304831b87271088ba119b9290bc7803c23387bf8752dcc1e63c8b9289b9a7e1d36d00f139d37286323fa2dca65c236c76bc045f4caddd1085cc73d93d5a2053e3824030da927b08b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483117274Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/ocdbt.process_0, Name: checkpoints/5376/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5376/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b73ffd9cd33744e4e9c6efff5a573d5a695e53ed9379adaa1a8969d7f4b22a26999c2e507fb0ee2141c60b3f91e989ffab33b4fa46c3377a2e466a6335e953895e5c93c5796fe472973e82658f861bdc720de6abde04961c05e0767a46880b21378ac094a57cfa149cc45847d9b7d3231602d8a5295adba9c8be44f8bbc4df9025faec8f93108c669ae6028ceedad3791439343fa43f46a13f3f0956f6a73c7bb6eb124e237a076bc566a85a4ab9f15a92e77a78922e9c24d706b3873969e8c766fae19a0ccaedabcfdfd406023d3211aca5de6cfc9927491fae159e602e7da00f05e42964efc13a855d73b111df88f7330d8fd92bb9a97a26b6f50e64d8f891&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483141424Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default, Name: checkpoints/5568/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5568/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5e92789f86b74f5ae333813f536278ee70d375788ce5a72a45cbb9b84d863c307bc35c9d269b21e1b860c69f231d664360d08177611431b36febb4fb24fb2919755e155db0e068be9a7236b00d9fa304be3d802fbb2024d99ff770e7cfcafb15da3827cf071c08501ed9456b2abb518ef93865d0d4ecaab265b1c3e888dadf99b96c6b256da0cc0631f5d65928c6fe3e8514af72e871bff241381f510940d557d5995e09ad44bc95934cce29e80847d6ac0a41c0c6efec7a3dc2799faf6d6b5b14df6166f8e77123e2e6da1d8c34566047b3ea9dec4b445e44b3a800e2a342586587c47baae6838da16740f6844d722a52213dce125c84ed93f0c20bcfec6544&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483111977Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/d, Name: checkpoints/5312/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5312/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=96e1aaa568d2aadb925b40484c64612c0a6519d85daca8a63297d9ca040487885b377d653a8d561ef343b70eacfc0ade02b90daa73092cf67d413f459522b124f93229e36104bb4fb6cef2f1113756d361d77c4443c23f0587425bb7f9e9e5782a1d728039cc201a3328981c814bd08ab9233709c9bde758f562713502464db038f7ec4d3f33f9fa1f901f5a4ba4ce352d74fb472e05dcb4d841d7ac80c957174465da0ac81d8d7bc5d654bb8e1f681af239cd2361f7111c0b84cef6afcbbd6909a601c006ebc53d9ef348ede4fabff0f5342ecba7ad292177cacbcd7b3354be7bad1db39b80a42a5cf6e4a00e4ea868f136a6b0501b1c43f5de3deb30768869&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483086609Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/d, Name: checkpoints/3968/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3968/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=167f5679df8300efd566e667b6e713425d866023dc44eada30c9e6f5ea16ee76723f9b4a659a00b1df4fbe0b86e16fc5a872953aac96edf9bfd266607b329eedeb950e003fed4c49caedee3bb6a0fdc5734f1f34bb50eb8590bd5c839cba15e7a15c683c56ea80a812432ab0cf52f06f16e2929f4de70e71564c0cd62405d7c4b3e62b3c7945752bd4d9c39a428d43599c5dc3e0dc38cc055af2979203f3b118f1566233257e8766263a5ed4f50fd4cae3021abc84227dbf4bda6e0a16d2654c84ff5977b341cbc0e952c5d4fa5c5c1d64091969d959986f21850da514960cb15843bd7e6cb8a0658191488ef1fb89f17e25f149ca67629e85e031f5dd9b4cbe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48311376Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/ocdbt.process_0/d, Name: checkpoints/3584/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3584/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=29df74273d4a4f689e7e7216615eb0a16dfdad41b0bea97f3ea8277fb28cc3ecad52b250d304a7d221c0e6ef2816733522a0076509cbd8a04bfac3b4964dcbce219edca7c67e7747d76bb6be72526e6d06b26446f448708b362481bbe9a658bc899a960d03877ec267ee9303b25f43bdeeeb219644fa186adef7ea3b1226b4fa92036cb8abaaf99aa21c418a8974700927e119b448687d0aebb7478338ee8bd01e91261eddd88e6eafde771ec3e45681af783a9533cf0ac25ef8e5e2e2dbd24e8a2576ffbb2066d64d3001eba6df8f26b3ed2233fe09af1501915b59d40fc7797cb86a8f93a1649bd85f31be6d085d4c7055778f3fe9697ecc6bc29bc4fedef7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483150917Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/array_metadatas, Name: checkpoints/4672/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4672/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=94c7df99dd578ba2f20c9f4f79db83cb7c305a257c40b9e8827507b80403f3208ef1ffa0e37bc76582d1d0471c47d1a3c9eb0171996d489a1e3bbbbda17dcda4a8a83d8c497b7fa3dd156479866dbe3580ef757ad0c0f4ad89920adba87bf01fb68206b1f7fdfe633caceee6c14da28083a8d4f7dbe6340de43c1d933444ecb1db80ece9a12f198ddd8de6350a9e4d8abadb2601e753cdf91ff1833aeef3d7a91ef0b2d5fd8ad571a61d28f98f76d399b31a1ed285d6e0819a25069ff7821621fdac9e21b66086f5c1e382074054ab92cb47c582377e1655a7ecdaeebe13f45bff0ace9ad522a9196cbd9a274a6828d7dfd100716131681ebb6fcc5769f102c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483164485Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/array_metadatas, Name: checkpoints/3456/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3456/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=93c3e9c1a2a92632485388735f50eeb592c883fdf498e91ab57800365ac48d151974f3b04cb0488a784ba03f8181b06c72f61e766580f2ab6441ab00283e3de9019e15fb447cacd661f530fda5a208808b2a4a0783558440dcb5d4415756e646001e7f27cd27cfe80448f2c9317b26160ddef7c3b4fa33c153ebf46f8453121c10b6b90891149611ae3e4ee0ecdf9e4460fba88d6f62faa54c416dc0ec485754bf7c41ce484deeccf8b84bf832fd677b2e822b040e22029a1d0e2cf1173cc13ee953269c54e6e4c9a308ebe380b5144c66713c12aa714c95a5e4b19fc9e2a4e2f551a0668f02f07a91fc9d7145aee72d05d04168c3a15a73be54bb1f6a7802c1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483141549Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0/d, Name: checkpoints/5120/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5120/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2228682ef1d9beed775c6c15e9d053632cfd89ab8095d603d9e96d5a921db16fd8db1f2fde01427773dc891cb0d84f4fd3859072e9a810a02efae5da515fccca75a4802576b6d5839fc1db5bc5419cfbe4f9fef2116db3c55a0bbbe13c82957663cbcb262ed389776d995495e2ac38563749d78387e316cb40148b9dc60f67a14333342cda44cb4499dcf57ef8235069cecb96fbc6fe4472811d9d0847a37afa76f284c3fa7347a1c1b7dd8a0e730d84611696e88500ed2a6c5169d1e25e62c98147a660483062d6539889ae2ce00bb9c19d9a36de1c586026512e281397e0d7011f0fbbb355effa0cab4f971f63b8c97d328db9ce84dbc7f26e36f9afb564bc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483148649Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0, Name: checkpoints/960/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/960/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a38b72da5c7b6f90d72306471d34ad42069eb05ae6dd475be7f3dedfa631319e08bf06eb50f7cc00c740119e7ba8c35ce7c8c74551789c008c435fdafbef6a10b8deb6d9053184a3da310e049c608ce9ca4bc56eac8734ef2ff1b739a9449064c8805df9490ae1413c600db9b0218f77db1449b262bd7124665ff9c67dae635bebbec3fa6ceeea2d62d74fdbdff3d158cad0e52100db3219e21964d0ba8b732fc647b5e88c6b14b847f5f6525f7a72deda9c06bf159b2195635973339c8804636192455b57e1341b9812f8d7947fe8bc6a796f9c2042dd3c0e3633b21e1115eb79d741e425e7b5e54ee5bb30d16b01da250baf8017bee2b7967d274ad7c39bd6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483169381Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2880/default, Name: checkpoints/2880/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2880/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=26e0d5f98141a7fc49426e0fd8fed0401b5148842ef34c4e2c18edd7d440636c602e0374d8b25ee4f4f8163d65ed8dce2a85cfca9066b8f26eeb8463a4af8b31b8678efb6d96d525b87d77f678a04a86356e9d4196b2b4ce9711d0e1b7757fe3cae3c502c7f02b115d3d816f23445b90ca0984af731e30b12e38c423df476034872c742bf48edf6a083438603193bf68ccb7c50096fcabc011ae9051392c0b41f5ba184110d1c812abc50c1879c457deeb02e8d98ddb3a3889c45e388a312cb51a434d263a6ab7b9bada9f9cec0d3b028f36236d03297e7d349eeb8d2c754cbb9d22cf250d3350f478437cc3d1ff686e1e28a1245400bd38b8cbac0999554c1d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483148861Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0/d, Name: checkpoints/2112/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2112/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5bb62ad8fd8a853621c24104b78eca5018961dda07b9a4332eebd095b6d6f4d621d3784db3e2ac2a538ef35f03342faec752d4dc55418fbe96115ab2b0cd76a37dafbdb236427747d8f8dacd518453e90e2c620192d4926f9a1380f09e7a2b6eb1ff519f34090724b27b106cb429497abeec5fa2addd260db0270728232aac361011daa370064ff9de8f14de77f0c6d9ea874e8518e487e7198ba0ce3dee18b9322440526337177013ef51a8108cd3ffae58b87dad75b5d3449770f790f16e9d633044d192f3f8dd557ffa028b70472932b39187cc4b865cef788af0fc2e7a0f52f2a2978f39332d39372fb0f8d3117b8364f506616ae16028002bb856db3872&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483199537Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/d, Name: checkpoints/1600/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1600/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=614f44f6cc4c4e37a3906669d56c57a2c34c234d788bbad24f9743bb12eff6f66abf661cb5674fbb09d84c5a3d91dbb48b827e115b26e9a69379cfa63bcc57da0cd36002fe482dda0988e81a10506c6830ec6e595c43f87220cdce2fa141b1878a5b0143fa7ce49de40b171e53413b2ed253ed61771d00f8c9e07ebf1418f6e5fc4f248fc09c180975b2dbb73093a4616f052c452130bcef106c28d004f5d4a2e62485e6fd022f2fba8c94aef35b8855694659e6e87a5d7470f7985739c7d85de0a81fa6966c6d4ff74e86e60255c178fd7b4ad1049efdd84583ba5208d2067aaa88f026b697d465139d0e50c789e618e09e5f00063bc5ad01ff0b38a90a45d2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483218763Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/array_metadatas, Name: checkpoints/128/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/128/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b23c0fcfe620759937374180f1566796ebecfc2474272f58e79e8062362cf3aac7fd32dd29ad49aa78ee148f28e2340e155f7e9e3ab14d1204b3e95534a5d8f2b5b915494cd0e69a870dfa142be31cdbfab31c5d2b9444608b566898c615f05911dec82d4821df08021ad80754c261d7bd83a7683a5730e34b322a9f2562e4a2268bf47d182b0f60f034d14d52f44396175f34f2974eb33a4c351584100e3d5939f70e458e853b061bee188a4f38fdd545dcdf9979551386d03a954405348b55aa5597dd768b11b60ca864aee0c2a88752ee2257cbc558e6e38355783d13ace826f9f183c9303538ec51b824e2b70d38f8fb2c306150e4a63935bf4774ad5006&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521468127Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default, Name: checkpoints/4672/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4672/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9359315263890c372ef4efc7781b6621fd201eac32250511a4f051b04dd83a6ce33468af5169fb069893aa8f8ccda0923bfdc827878bab38eed36f24d876c9d4aa44e333919d14223436b14928c1cd57a6b3be6f8ec1b44569698a6e4e533680ca79144b1b1ea4139c63c54e2901fa0cb33632b4d2a24b2bd8590798d5f50afc621fa4d1a06585ac2caeae27440cc56abd566a37ffe3780b2203d4c2450afe35617612087f6390e8fc7aa2a37ac1b29fac767efec0e5f1bab80a280c856047b9f7a8dbad48c9b4f58b1741e08efacb5136b168379e408dfd3dbc80f4387d8c1311893e0e00dffa9cc69080da296a38717b28fe6fdf2c9a5cf4332da64e2b12e3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483178373Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/d, Name: checkpoints/960/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/960/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=01012804b9edffb1f9efc0b2256d170c44860e8916d8647d267fdf115216f98cbd45e50d8fc3ab73a924238e2f18a1221a1d8c2a6732c91b536a4bbb0c8f5425df7eae0b895cd0ad68d7c867260fa510065abb5e839946b3789a76e9e3284579b83b885afd88b766bca468cd68460fe60daa6b094fae6c742f59f2000604d986e829dac1fdf0e1f9aaf4774a61ea24c2c3d417df32fdf3ff1fc760dd2c14455d74f3b5f9d3a97fb57e41aa4f028d7ee716a60fea6a919f2d02f56a148f5e824f06cf4549e6b630dea822b1f011eaf3d29e7b7721d6a4b71bfea16f9c3dced4de93d89977cf3a43929a52ba9676a498e1a786de66b4f38356869276c10f082988&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483226322Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/d, Name: checkpoints/3392/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3392/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=892e20a60784148154643a9f37ac278aa7b43ecebc89b37f5398bd64341eb0447c24e3b42bc218991de6bcb1c31b0a8939cd4e25f420bd1dfb400ccdcd29c7cd264280806c4743a39b39a26af67490e61d81c49dd05895750dd74c8d31fbb3091a72becbb0dc33d0bae1683f3037d797c15027313f0cc52efb040b8f30ea089c72388059caab13d19e1c3b3ab0f1edaa0f2f918e28bfe93f32a333e13e473c46a6ca131e337e001f8cc97fa59aa7c7f7e16641dad20c6d16cc3a2ee201c8fbc4dab863443f7e52ca8c1978c8626c9c45dc98875c688e225d9fd38f80c30fe71eec30289e2cfd974f8570072e4f120277889b7889d3210ef760c2fd71de9e458d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483268635Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/d, Name: checkpoints/1792/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1792/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9b9c6e52f514657f81005d2a0adedb27bacaaf7463bad132992bd81db8d5e0ead58ea6c96133dbb4abeb92401a86a52b98554d32bf017e16a5cee06734029a1ead5fc43e6a4f03f12d59549722020e26efa1700ffb09ff19a3068bc7646f36fc9078f9b35e9249e1928b13724f815c948f2bb035c84f244de076e398615dbd3f72240fb46c24350c742a560f461cac4360a8b5d476e001340f356d769a8956b3aab858647e340b67c1d8095735c4dd294525405c57513a988588241f7c75e44cf6f8776f19f4b0815ba52ee629f12f46fbcbe52a297cf8d3284b9968d722237fd8c8fdb10875d1d2a7551bc04f5a9d130ab08c58ae53c9dcae9aed0b5a39a9b5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483265625Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/d, Name: checkpoints/1536/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1536/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0706f9472c36a911c3f21a6cb4aec504b62753db9dd1a1bca370663d94cb895918e55f0f03a319cd128208391ca83051600b5f8ec6e71730d98392a1f8f6d2f5b141b1e938f1b4a963b0bb57d2be4fde57175f9c069d2b9ef9207a82dae9fe2f31b9a6fcf00dcf6b82ed188ade2050976547e2cde2117e97a5b72abf00f8d6dd34a15625874988daa3106b62de51213a7117b3914aad283483557b84d34340846ab22eb8918ca3a7b4ffc1e9d5cf545081a5437118f4bbcfa0202bfd23d75b00304523c696714cf9773ba62e5af69ad9752ecf708ba7d8959860676df9b3a84e69810d9dd717ad3ed3443d4a18ea8d9f143fe6eca13a81cbd7ab1e76c8462c53&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483257847Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/d, Name: checkpoints/3136/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3136/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a56b08e382dd2a371dbf2983ede3fd768747a76cd18ae4706b95cf84a0300e3097f15feac0bd7b22f2cd2b69d1cfe83650519c259c05a901a65fde76255c018358f6f4f70097a1113390ecd26ef2f25703f3aa981a3f25f6a9bb63189aea37e6d7fca0ca87978f9fab5295bdca3476c1e925995edd7a83673a13cdd82e62b4c5117e0d3755e0460c5265fe870b766ba5ade85ca11557cd7168a3ff7f5ffac3c1d8a6d3a9a7395fe288e2e32f7e1f1989f288765ed6993eb338ab539da6e3fe3b1e42c82acc433552c2fcb82bc2077d5292369a3062620a76e6b8ac2a987d4a57ad1557e7d1236159b8cdf8e119ced0d0ed6b5cbc76f899cbff68e767ab8e5c8e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483222291Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/d, Name: checkpoints/4544/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4544/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7aabaaad8ac3c95540055cfe980225709f0a93cc224ee0157dbee7221bcf0710d122a227180d257760865ca08bb6a6f19d263f2c52fc804a4e09a4969c3c00f101d4a8efdee8c07e4e95a4fc6240db7b6cc6c119ebd7b6c214ede0b073a85cdc8c35f10fc9848c4438786f0f6760e18474f6ddc69434083d41eade0eae995fc148faad0a0623c4febb379951abf08f191b5c1f7a644675c2378abd37d819018d80d68d7ab1ed99b2d78b94e31eaab359f44190b6252641e031dcdc3d522de1d8aeeb0b51b57700742d480273427f157b1453d98dca17bcedbe802e38a2337157e0999c50cc25e381c21ac36a8599fbb5df9e06e57964cbc758bcf47f6363e754&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483275544Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default/d, Name: checkpoints/3648/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3648/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=786fbc51480845d99d3dc32a8765381a0f5188767819c7a2c2f7a261e4af574069594fc41dcc3721984d55eca6011e085d473343bc5e978bb3f5e92631c3c52cfc7c813698f1c4a3eaf41782307acf9d96938e4f0036df87ae50c74c3363dfeccfe8b1c077ade6cb5f7c9822db948c3457b5fa5532b25884b8466a1ab1b3ba19a305dc65c38d023a33332f1b03a50ca37e84fa5a7fba066dbc80781e1fb436f856f9b2ca0cd5ba134b7f2d81d2ede39067cc17b81e79a9cc5762141b4f0d1574ce8ebde3967894e00344feabea9a25257da82d583fec4c918c8beb474bd349de06d8b3e59fdfe276d3043e7d72819f5dbd36de1882fca095e2511180840d195c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483277118Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/d, Name: checkpoints/4672/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4672/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=af43ab7c202287786a0a322d451cca62d4659b7218ad1850876033e6d1b18cba3c7faecc3fed1a05f4d50ca1c435ec4d9e99b7f2b8ef025d6ce6d9cf59051a909b947d4853466f3714518ddc92866bfff1f7aeaec39e1dfc87b70fb4bad72f667fb750190851ebb346c7d39c7a85b20acb9a95532eb3c5c1aaf7976b6ce5de468866f98f79b39c93e37609cd6de73df60c85f8965b14d663f86ef5a2010d83cbe35edbe6037426f370bb0e279a5347e0abe6779c38a9dae6d014c7ebd1471a32e6ff12dffdbaa86fb02f0944bef7e3377e079c63b013328d4a144d651fc369cf8980eafe2e8bb858fce9c8f49a7bfaacc2f3eaf1a7a85ddfb1cd4e27766c0087&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483283942Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440, Name: checkpoints/5440, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5440?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bedacf337bc7f90d46c5ed1520df55290e5cc9df41ecfccd25b148be8f75a21a36788475c124021e8d82221ec322347eace0ae9d64e42a6aeb952aa7974b25d21d674fd09539e83206b296df8150eba3dae8bf1964df4627333a3dffbee01c1b4d082093ab2e3961c6eb81c93d90522f2df6dc61a4db31000af2dddd6f92fb0577b7446c738380e7e6e526730b5bf68d845ee2ae82da471696045ded41e82693909d6d1532f7481142d25d2060d929454bcebf952f0a6e8faa9c27976e966a8d2d46fc8dc42cca03c8dfe45216d8164c2c63fb1fb6f6f626782915254095a7a09099f43c0162b8765e09b33c911b036b9b205cfce92f0e972913e45da5418cb7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482879189Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0/d, Name: checkpoints/6080/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6080/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2e6e6c9f26d26f0a4d1ad8bc37d0bd9c43664c90c093028040aa5fcfead3fbde5a882ec0e47db4cca657731e22db13d89036eb261fe394523750b69b1a9db713e743b755cb2b427fd017109b8f425d44674c8d319c5b9512561e73f1f1f7f6762100386fb1bd689ffeb09337bcf2af68fb60b250f9fac1bfe301f7a7c423c26d5a1169a2f1aa10e547692afdf437f8a12d0435065aa96446161705642f3538eb92f5ddff42dfa7b5cd1d87079c152c86221af1cf61fe6024f12e926583f5539c6df65a9fe11eb59c583a37605caffc7db201785457dcbc752897f9f4179813bfc1a49f5a90c567dd56a2fedc83e056eae80ca81c5712e51a89f94dc945892ccc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483274895Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/array_metadatas, Name: checkpoints/3200/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3200/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a37dc4d4fdd5f341493a3df62b43ca5e1aaf698d83299caa7476032062d1597e34589dfef8714dc700e03055720fca956dde4c816f98dc774218822d909b67a38985ad01b447c957637c3dd1c1c14a2bc9c305ac746d88eb40a881ea652df4e4cbb5f891ba8bcb6ee8dde3ae0896727f6de916992f5b8e2cd1375e123d70e435edfd3395f4285d3aad71f52e9a102765e8160697f508c9a8d8cace198e568f440d3ab7e885c6fe5571c49e97c7a3b14aa1e3fb6c22642952c8253b3545facd26e9a6ce7783dc35c5e84b7716d9c93fe2c117b0b88259443f64cfbf45ff0381d1cfd9ca4dc06435e51e0d2171e49a8e10eb13a2805aa1314669e8e5b84f182fd0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483304334Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/array_metadatas, Name: checkpoints/1728/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1728/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1b3c5629a4dfe8f7bf3526a42a5bcc99e1dccbc54d2bf2885607186e6f47aab2549e128049ffbfd834b3bf509ad6ea8550067dc9cab04d029e491c156d92bdfb06152fe769723f8da726da7ce172b3c3e6c41db913aa9248a443730a23de5472aa7deff1e20e8d83022bfbf41dbc5b3e89aebed1f056a266e6729b6d0c6fb815950fc289db5968df6b1602a95cc854a10cb14731ae959e67394c1b183fe6b224904fc5d2dd1d892360dd7ed9a5bccc3139d3ac6d6576f1084aa913a2e323b0bcd85aa681821d68b3684bfa22213cc7e700ee5279bfeded89b099b47ff45a44604ee9cb81c731adcd2b5cdd0e93e761120d6e6ee9aa4815465cca639ce57154aa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483320386Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600, Name: checkpoints/1600, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1600?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=196a9edb7a5d5088ab446602efd9c73dd5634aa4c322e4c86c6ad3fe7dea442df1b2a81ee1f8be722587e78c4861108365bf78904b0584cb6d86555fc62240dfc1f43e25ab2647ac1c668711c15e84ff2ef64d1d060adc5a32beac13390043958a36e13b0c6c1c894a545433d9b2643d9e40daaf96923d8e7e7d8146b74d1f4a821fcfa47716b42db402eb371648666f7c68962f60e7a5a87b570ce5f690e0bf2c840a632e630e7f7bfe8cf7f999ca110a7d3f3980bc01bc6830a2369599b29a3f3d2658e59b1fcd6a3a15bf872c41e3d9fe6d62cac4f0afe86be0710e715dd6545e461877cd158497477087b76769ff0c42f5228a117767a94dabcba695b096&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483315449Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default, Name: checkpoints/5376/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5376/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=432c68932473501256f6812a91dc495a05c92b0d0775b1ae587f1245bd6f109141a715aa22af3468153e8b4876d70b58d2f8c8d1bdc751a1370de711218f706816f9ea76590fecc81d7977ffa46a381b2393f9036941686410f7e3aa77c3b7dfe3689937b3b166af55b752bb539eb635ff3aa77c81b7603700997330b94d6f0ec5b88784f182cc0d4bfb7dce58293b3af3531208b3ecd7c8b59d4a2d15dc33622ee4983cd2e1df66b06f1f2da047a6e90d39d6e49aef963ec340b34a0bfa911ba1c8ba58fe11db007f8fa1f521be74b544324bac2d037908e130b808673be136694db14ec80bbe8c3ec0e0b64701f5e5be8e472fc475a8817116f103746a510b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483280495Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default, Name: checkpoints/128/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/128/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=48b26fa95a46d68ea1d7ed084485b31baa16c7562ad2959246cc4f06ea1d42c5d3a8653a89eaa11d1a1e7244a6976e65262346e80f96fedc948f50e540af5dce46054e7e75edf87a4620fbccc705032f3893742cf49289e5517a956b894ef2db09458d49f09e9f5537ebe05299f0fdb00cec7a6c13e19692c43ca081ba80727d53dc30d6e41a11316ff1d895bc7f443d18c999eead0e656318869f2c56c689a0e6fe28379fb7196aba57fa335c23b926daa96e286b74adf8f2513a818e39d172b06661d5546740a44046ffa2fc2e5739fe46dd1de6cdecd3f910bb8effed0b75b170eca0e658687f82ea81a094e691be87a388d5016927bc4c7228d627d86921&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483308831Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0, Name: checkpoints/320/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/320/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=00d69675a5a4ade19d677ea7dc55215c60b70d58d6fab19d1071373a88f96d20bf380d164e1ce2d429f6d036bd4b58c3bfefca2119946bac9b65136582da831c2b3192c9836f2965bd16dd8fd861a7c293bf17d27d2e980f239f8caffb7ec587868864f39dd03f998593ce26ab8fee683d69fa53b02ef48c1721a143884e75f2b2adfe04d40a46367d80ec2598febf0b77f4996193ea360d5dc1cc5d28e4fb2b2daaa6e58dd669ae41ce9b4258ebceb0b0dd9d2ef6fcabf08fca87adc94a5c9f942ef65f34c42fc1076da5473392532e05309ff9bff075ef7f35c50f8d62742f2bf441e8f6ffac6a51fa3a02c626dd9cd7c842c9ddebee4812b5e57604c94634&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48328299Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default/array_metadatas, Name: checkpoints/384/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/384/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8f19d10f372ef6df22751c281f1fef67dd2d8639349af7edcf7d8e3664459735d121683c3e53c2733c20d9a33d0d34a551358181d27d0471881a3264102bd524cea800f52e76f5269af723d9c754355b78f412ea5922d4c0a2d5151918d246ce103cc42117593d92714e1f7886056c40c5c7bb9e9a31743b4ac8bd661f41d7d92e772bbc8aa51729d8241f15285cb5f84666ec7fe7fdcef29d601054dfdfbe880a7fbbfa4ae768f36abc9aaf656925f9b8dee9c7909ad4a32e723a173fec2e587bdd49ebf31056240de0cc57322aa901f9fb29ad84b445fa5715d99297ffac9b8ba9f05ba80792b76fa6b90dbe7c3c1d2b65ad91e287cc33b8af797ad2af9184&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483351825Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default/d, Name: checkpoints/3584/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3584/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=20d6328fe9499cf84a32cfefe09898ecdbe34f8744fc864606d3d75b07fbeb459521915c8a513c8396933150c32fd44e011119d0546403ce5667e952a45d2a502d1a56bffcd9fbee272fa7fcf5c3c049d49dc51e4f299bd1d31cd0be4c9951d851404d781dbba30f143756872f8a2673c8cc9a51ef511590d0b5ceab9a64a11077e59721195d3d4ebe681715ef1174b7f23c3f531803cbcd693bbb0e4112e2be9e1ed152f15eb9a5bb8502e00d4b19b82c5e13e56bfa9e24a527202155dc7e3a88393c75a56b18f1e1bf562b008b220ae42f8f63b49d38676808c42cfc50a67c216a21245f77dc0068a75f932bb169a7acfebadfa9e3990f770bb5dfe307ba99&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483236769Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default, Name: checkpoints/4096/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4096/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9a5d2b1595f472f5d0c9812c382a65bd37388d0bc2192c893ad8d8522dd071a816ed5b5a5dcc89824f657a73269d042793538193f14219e02fa613cb7847f3a8430f704dd9aa2855a2ec376b2f87bd42ca5aa2dc9e1fc2ee5b80ee979e38afc04dcad95c35123372b1acf0dc76e9f6909b39cb2d87cf59623e2935cee1e7a1914511e1141cf92df3cd7ca3ab7c09bf00d4deebe7eb814e65b497762755d0c088d83152a5e8f505747e830769235979d37585dbbb8d63c8c4b8e2f2e29bac6f5a36f315219cdd9135381f822ea3e33d469999939aec6c174f58ab883ad64a5fe0972991b977714610478c2e1ddde47ba7a3b0d9d3e3979a64db73d435b5bfc4c5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483307905Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352, Name: checkpoints/4352, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4352?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=00f11d793d0b255d1ae20b9b6198bc5021bfdf47b046bdf9c591142d6caee18ab2be7c5b2d7eabab02b7a656585a1adeaad00ad1324e30397bc84f1baaaede6a2768f4764095d7cc578a1fb11e21b517edc5f469ea32e375cdff93deb7aa909221dd6b1b4e3b3324e7521ac7f67865436fff0272f1a1b561046affa3dac16a3ade1bf6a0b287f8393e2fcb7fc8e953581687aad98bf926b9886fdd52838e8fa8bbab0d40cd83252365626d66379b9ee716ccacd39255f986d1a30228509311f1522acb56d611cb4c2d331f8992302efb740d32d776361253bd26e065c8e6b26a8038a619d2f918970afff92c76dad81076291570d4e04318e8ee236de7b7abe5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483383076Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840, Name: checkpoints/3840, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3840?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=aa0ca352ad727779306f8d48f4793afa998599fc62218e20bd98c80db5530b243217dfaa1bc52d732c1078acff32e03d1e55e8b00d922348a6a452bf370ecc029796360a43b33a99850287209203a180aacde9c47624c0ce456e01baedc7f371b054ae7d631cd7311da3d73b3c3a2b947c0014aeecbb1c4b75ce6971247cef47f108643f3289f25cd518debce25af92b4c11668eb9feadf2c4c894711b8c250785d6e9769db3c85c8921bd66e1b18fb1674e26fdb40bb93f7d9ca03d79a55debdc66f27413589028564aab57e714d756f4714119af6df2ed3c0eb4644153988d8fa75bb1cc1736236565fa43b1fc50bb27fe4f092cbbad277c53e101298cab6a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483327989Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default, Name: checkpoints/1536/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1536/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8eca7b170ed5402a0cd687808d6d3fe26801796dc93a6bb640cb99f2791f343d5b195d1a4995f8a82754ac83e8c95d187761b010aa082938be02352578f818d70fb83a64d99b861ff902bb19310311ee4c9a50c289454bc5f8377b57a3c5f2ecd9840c2479bdfa8afefa3747a49f5416f89acc27ad2e497e35be857a11da9e767b20ea3bd2fd08f7fa12d70aa70577e7e4017b302b48f3089b85ebe00f4699332cb942d3ab7130befeb20b6b481d25a4666e2bf21da339f43e4f3ca3f88246d84c461e29ee9a7a5f2422b0bd3cf0926333068e4e0cb43d0e974b23a1f984bdf6f93ce004e1c1a49f2dcf7381ad1a9bd2d6236c9941fee7a0e919b1e97d532a73&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483386717Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/array_metadatas, Name: checkpoints/4096/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4096/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=99f46a59b501576e1a495c1863556718455c9ff46c8f06371aa15bc7987a80f3736fa10d6e01ad26a09cc786ebaa944dd48563c57736f1ffc1f9c6cd7b6cc9519aa6979831ce91ab9a1c04b0f0fdd09d8ae8f34aef1b70e055141e9ce7c97271a0e791b72e46b31b338298fd4c90ebcb97882a110c0446564fcdfea8b0c91722b888257bfdb4519668735ce1fbb26e0078adaa770f7e65e892970fe84360ec0e5521e4d76fce51cc0b7b123935c56adcfb8136326ad4b297be78704b2198cfeb08705830ac15223dace417370fedf30f3629e9b9eaaacb74ffb6120c9ed4266d61894765fc49179df86186780836b9712e4635ec3af0efafb9edfc3262857077&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483344625Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/array_metadatas, Name: checkpoints/6080/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6080/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=924ab0332f9910840260dad3159b273146257d6b5f23d1f95df5603300df31a96a7633b82efc57eed8d3be74cb2d805df929a6f94f0cb5dec2ebe0bca1ffa7b6974418ac3106faba87d3b62a6aca0b18a71f10fbb29bd23a1cef53227cb5c22911a2ba15f59c06af5555a08f74dd713de17a0a8dc4b4c7d1db69ebac0028f69f21aebf9e75d8b47c0e30e73a7dd3f2a9191898be9226bebc40c0a4a165ca7a978b82c04d078ee382b1cd9525ca29a8918d65142f76d4030d5542da44090f1525e2910a433ca3f2a9a129e11b64f4562dd2bd1018ea67a1c3f7580fa9c34ae8d47a7a4ce5d83decf80e486d364ecf25d3745e77b7a7d017be1b8512d2455fe0a4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483375007Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default/array_metadatas, Name: checkpoints/5440/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5440/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=77b4d8ea9b5bcd237df37619ae80e17357b51f3da368e0b4271b07ad136880356c8fc808477ab23f6abe1655049c542cdc154c973bc8778b9e5b1ea61eee2458f8bf7cba857fc6387b5911e3fd1690a256e9995b85812318c6c63283ef2fc6b63322c4eecd4e0c362149afc98eb60b2258be8ecc913efd28e72fc351f9629af3bc0151030a9c215646f3b2cf753c3f54a6ab1e5074d7e6fa58d5c32933750802e2bc71711bbd9e3aba6af869ba822ba55a3cccb39cd17dd121298be25d1564fc9c87b1c713815c11572a762ce44441a1fd909ce081f83a767e5b95cf54f70e8555860735a0144b3ac390d8fb4023fdac6388e2d89c3fb5a8f8fbb0987d3dd0f2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483362691Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/ocdbt.process_0, Name: checkpoints/2112/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2112/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=25419c5ac634b314a6af57426735a7281a6b29e573a78fcbb28c2bd280d1f1e2fe52dea0e566740cd20db536280b835beda8468460a622701194754a0df9873c9eb97389c66c56f5427422b8a2f88bab200d5c2eea4f9a6f173499a5c62927e983f6de3744c1e53dd5dbc738789a34b169849759f8e8fb30ad29aed326deb8f99e878c32677917e3523c2986e55fabccf6f85d8a219e074f4d4b77e692d7a59ac43256277f0f59c4103c6b4dc351af959d273f6930683e74b2d4e56e3fc2707cbc44d7209f012eff52caa342d15f1d83f44fbd33bb7fadfe7c364c94aed8a4f97c782573513216061983a3ce817002ff8a01da2d4254678e11f95f29a94049ed&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48335492Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default, Name: checkpoints/3136/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3136/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2b9b47dbc1b22f46e05fbed0071500b2f5e830d1129a70c49229c07a9ce81ecd7d3f8ff06b43d63995ab11c0aa9166f3a367a1f7b6f6928d5d567616c8c56a8c8832b8e6f408b527a3dd25d68d187a53c45b99fc65c980d1b5331f00082ced08dfc1523a9b2e3c5fa9e15f38850f5d1c5dfa95e4f597513facf1f00015fd2b9118378599aae2168bc329d95ddf310fe469b7e435fc75a264ceef09aec2ecbd3d206b21c1a5564dd29c91ba7bee3a82b01ca7804bb7b1df4a4488eb2373ae4155fc4d9e70b057e9086c5bb84e5abcb869089f8542bc749f897b5cc4dece2503f2ff10024f740f4a4171ec09c1618db7372a47f1990e419faf6d22fce4158cf055&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483431704Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/d, Name: checkpoints/5056/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5056/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=be3713480598647f98464c7731b13ba4fccb04cf90eaf0023950101711eb656f5550a3f8d01ab6405fec3fb3413371d6bc3d97506cf36bc05ff64bb9d7d54e65757b6933cdf392dcae99b69dd10a6c953e6f5caa22c47a15cd7f8bbae3e6e5fbc02a7b4ec9e9ee1f95e917d00f00b822197a9c2b9cb916564c1e648336fbe3ff595f3ab9d9e4f5938f0293507023567b955e1f55cb4cded6b44b8892a934165bbc8ceae63ba1573f1d9f0beaa6af47b0ede8eba5fb1794438c6673180791df66996ae14512923b14b95a17814039137e30b7ede33afb6b391803f8fd4401280626c1433ff0a41519ed15587128a70937adae799827b5aae58f879e8d25477b92&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483419996Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0/d, Name: checkpoints/5056/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5056/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3afd2281fb05e0c4fcc1bc59d24c841662d46b3e8cc20d50cb5ba894f61ace468713989ca9c60cfea0f620033b5a51c6e8d47fe51e0d1d31e4c34c5e3746c81ec97be4a0995da85e22e349fb32d70df8b47b0f42907969ac57cdcb99e3a1f044c2a1c76ba9eecc3a9692b65b2793687edfe750e2fd6b24fdef42ce89cca46d28f698ccf43b3b56e5c0ee52e2ec596cc43b55d2852d0b163f582da034ddbf143ef8e9b231cce96af7f993a88795a1abe2086e2f2b932403dc37c099149bf7031e0ffcd4ddc067c54a9bd77acad648f0e601364e306acf3544e923eac8ba50e8307a93a8e43197648ef598e23a68832547ae63b3532492a22497b1d169e5bb419d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483424968Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048, Name: checkpoints/2048, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2048?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=40cd7aa7372f8c97f795d92d5b5658e7e8fb72f2c61f6ccb169bedd237e2587a26a906cb33fa071aae9dbc0593b8383e94e5a7982c1bf4c41b077db6a53e66ade7b46de4842b8b0f25da341b2949652014169d0c787111ae731a799cda935cdf05a44b2ca9ce9b74261f7477922adc3d0e963540be6bb46b7fa96b7a8bacfd67e57e6f46c18e25f8b23702ddf030d015e1c4fc3ba6753f80831078da903aa44211b53effbf142218aa4edca1f0673779caac11e8dcc7aee1548e0b9cd96e284252cf1f2b279d5e64b4a8a318d1a5e29a2e9fc6d7946f632326770ed78c5c9d8f97a29124d65a89a12bedd7167e674b3a3286004af5ca94cf61065f2aeb29166a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483476652Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968, Name: checkpoints/3968, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3968?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0a60e178e79ba0ce84b2f4cc2b0ff8dd2e4987e2ecc450e38645179823761d461ebd4c7394808f5d978e28eb770c5fd5e446fa55c202c4a04e1ab0d864c4290c71372a28684c4e86cc6c0c67ccc5940f60ad93a4cb28deda55e1d7562a58b80a79ff6e58ddf4362ddb150826dd4c662252a9ff059a046148448fed9135f15d7c4632327fe545ddfbf356819e1e880518adba434b6893716c6108a5995100383eb32b1e655202183a51437a0dbaf8f4390e74dc95a73ea30ba0de7f8a823487ff1d311ca2ae7f5355f72e835f1da13bcf50ca357abae15947fb5921f25a6902763af5d34587a31be5a22215427340857af12f15f55c2e03391ebf7f6457c30253&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483870122Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960, Name: checkpoints/960, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/960?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=574f978a23f683f22b620bd1e8fae62c7003aeeed1b9f8caf4505d5e436268fa900817cdf9786a95c1ca38400c0ede9be294b602911fbab317f17c008a49fb69cc62e53a9e2ed752fe97b7831f641074cbd0024d4d5f32c6977ceaebe80aac429e1433af6afa7812256387fc98915c9e4aef9954c622a8600a974ce42f8e03ffbf4a3f1a4a2cf37a23a957c352db52169b8a88c0c9f8ea6387eea29af4ba2dd437e28641bf4ea34c2b083d9b9b9a40810d07a2bb8488708c9904f7f5063ba3b1da95c759141d765d834511238950fc95c400b5cc07373cd69f78b018feb044e06741f09e72f5fc745c0fe14c5b93664890158ff94c7e66a1a58976a3035a92b0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483442698Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/ocdbt.process_0, Name: checkpoints/2048/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2048/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4943aeca4554e256f95b05daac1a3b72dd0bcd6fe13fc75aecf2bbbb10c878bd12c3c995b37906340d8ccc40260baf00959e6dba6a4ee66a3d16f93f9cded88e9c1c90aed79358ddb9aa313a1d0e1ded4b1c33d9fb7c1349b1899b77cbfeddd314da397b994c746b7571e9a77dc479f12627282673f6497e9e07ec35f4fee3698c13807da65fd616a31dc21794b7e07db16d55fa0e9d1a30e29874ab3599a6c540b67b2267f55999cebfa5828a607455ce8867e710af5da8478edef51355ef879c520405878c763c33b468cbc64d14928f68479b31a1bd773096f25d2f9a1a189a36cc9e1d36a5a638bf4dbd6cffef60fd79d54c7d41bed3aa6515f80a6f3e29&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483976115Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/d, Name: checkpoints/4992/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4992/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4ba8005acd8a9f4a9cfef2f44e281daacc22f35868e3ca41fbc0061d785f01d769633f7414da9742462245abccd3d9eb16326b3e38c4c6096238629f86cad8421dbb532727d21fdcd6f0b898e3220b9e586ba0feb4aef1b10db031373af2ccacf6ecd48d5e03b36af141744fbde5bb1e19a13685f9fac5bf80cd7a9b2ae26b98ccebd24e5f45e5d87ffd64a2aeb65756c3b8b829451e927ff49ad6ffec5774e0d6dfbdeacdc11de992dc3ff0b3bb45e5938756be1694a321236c5999bd9d85d359bbd2b995a402c8d46dcf53b97d8940701d1c999afb362f41448c3e2e18876af03c63debc84a57d641b5cf3f0fce589a17fe9c6e1d1ce53ddadebe0c9862402&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48330434Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0, Name: checkpoints/1600/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1600/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4c20c96f5d1544038b3c957d80e6762e1cc5089bac79e13116ec008e5137ba3df927e3803962d8f1ea5e7e13eb75bebf7051de025d1225d529b6be4ffba84dc4a68dc980e7f42f14da381fdd9200df08143df857fb9816e0fe14d20c937d86e0e5471825536813a7a85799ec7bec6e0005a65df3518bcb76f76eed37d1b3e9aef3ff80cad946385d9245fdfc2cee94fcf93c714bab9428f41a8dd068e3151ce87e005203b09a7b294774d6de762c796c5da3da0dabecd17d5b86107461a6d9326054e02c4ca86187b0d26c3e699331052d34fe06edb94ff9edb0a00a038871cba88b849d10db4d1280ab0162389ab8c9cbf8ed0640120a6f0c97d24f01110201&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48402179Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default, Name: checkpoints/6080/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6080/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=796a24b2544fbb1b2f6dbdbc3c11ad1f025da62bd0d1703147877a1dd3d236d3140d98c028d71e2fe3726542cae24f767fb223c536686d1b6bbf0d35d24db4e3a903a23599ef87982f62f122a9bb5863eee55b6dedf5f4817e1f64416101a8c0dad67313af7f5cc01b9e8ed3cfdd44badeb26ea46ab39f185f552cc67c6686ea9165851775dc6fda61d07cf34fbd78fdf444133243a8e3f805a362f68834649c1111e17fd5264cf5b753deb652cd188322707e07f2432dc1017c3ca70d51197f0553c1d4460c4b3c1fad2af8b97c66819023ad99e8454f024496cd58aa1d00f4d74af8a0f363a685a60340e407cd1640aabc8f5ca52397d408162c8896c62efa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48400449Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024, Name: checkpoints/1024, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1024?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=76aeb05ebc141a97d0250664bed751a5a2f2a338dad1663fbf7493f4039d3514f1d633db588c72f330ae72cd5d9409f3da5715f02d2aa8c29e37b82eab8997fec9d64af35316e3f03781b8bc99beb204343113a50ab56e66941c55fd73d4175496dade3e5d9a4a8daa4bfc8a6787a46a197d5b809cf666aabe1ad7d181f85d787608edda75eb74ff5c22f37f28ab41ef1a54f764b73de9758d931480c16233cefc6e466572306fe5e38fb81527e60512331b62833102ebc43215655ab41b0b1bacac1544d7a55001de2403b79669f67ae60ac0e8ac3bf1a0722bb927e69f6f02b02bd05ffc71776ed5688fb19d5314eb26e6ce8415185cd7a64176f6558bcef6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.484042452Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928/default/ocdbt.process_0/d, Name: checkpoints/4928/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4928/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7926d34ab66d9ec49ca0327be76f0e20f5882eec229936fe8ebd5e5e22d59a59a245360063576d5c982faf3f5c2baeb8abf8b2724ee1deb47e28b676a3297316d90036a8d97614cccf7380d12125bcd46da8d2bffd922be5e4e37e6ea548c8e0222265b114373752a96fd6d532ce74bfded8b9ad34232a190ea65a5e1327e919c4efb86ca8cf8f7b575572ce85a9f07052aae75aac8aff4ce54d8c6d8ddd35546ccc5b58cc209ce86a3ea7832bcb535915961887fefe073fc124d13f2ad166095c6daf355b82fe5bbb591136641ab034faf6876070a7c0c15ab00fdffece854631b1f0d644705c4564c91804310d36b3b738e709ee81c34c5b282c805ebb9e74&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482968001Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0, Name: checkpoints/4480/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4480/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=14553e4468d3bf51980b284b8355fcb04840e7a744132500dc4f1dd21bae2524ce2db7e302c28c18f7fcba9c63a6db462a37fdf3e85d8bd7daacdaf001538f1a4525925e73be97d0b34c59fee07c9e19c721d611854c523b9f0fe337b4181cdd9b17d51307a8bfa35aacc4711a673fc70905023920e2b41f3ec3e98340c0d497d50382b8562d65359026faaacadbdbe0692c8ea5c5ab3195e0dc26a41f645ff493ea8f3487c20c147f127311d02e6b319d4542b55ba996f9a1dab0575969a64016263faa0d84d66ac31486522908318801c638c3e5e41911add8be2156970751d7b6131087fef24eb66581e7bc63b27cb6533d02cfa3196ce94a1548a2ab030f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48335293Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5440/default, Name: checkpoints/5440/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5440/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=782ad297871d3ca8b6cd93a9c58160674002802b1264b2fe5f603b7362ea06d72d1291aca91ff7aade5517c0107678700cd17d4baca1baba4b61a674491457ff2a3b597f84f621e406894883cf7188c0d6e94a7193a042824591a9e58d65cf1558ce06b37722c3b9f98490a018821797f6c89995c617bdce25578aee18e05eabee2b9725cdc602034a4fecd4e58615b3090a2f607eb3be44f09aecd1248d96b365abd52e5a1bad9d006cbd7f29f28334fc89f46d77a3866533e2c5eaea70ef90b544167d8b7db5f4a771d8c73bc07884871d900daebba4195418f5662e7891baa713e721c7973ae0d5ce8806c47dcf4fc69ee98a1a09975af367c02314a4f673&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483300664Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/array_metadatas, Name: checkpoints/2368/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2368/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b10fa4b7d4b985606cc1640639c0c0f70949f5b8379f729798a68b0da9c282c6e391955f77ade7e2aebd3c28daab3d9e3eac90429ff6d2f3f762ba3c97f142184db239e71aeffd4a45079af4474db3e021bed6f81246606095cffa07080d84e73a098659a6553e08986efcde9ac57c54ed13213ce5e08ac37a0b217ba562d3e3f8bfed9a1e370a0845ebf6d75a2dd4afb32b6fd4b3baf3a0201a5f7e8ddba4ba1b29794d7f9a0f7b734a3b33d10d08471812fd6117d62c248481db16fcebc73a86e247076d1f30c7a7abb4efe9d44285f98c3adb5ea9f88bedca812778a91dceef2d6e6ce0a904d65d70b2720e72e84530f25ea90e45acd8b6245f489f9d062f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482865658Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default, Name: checkpoints/3328/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3328/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5f4d80bfa463b9915dcc7eb74782f8b659ab4b741856bd03a86b180bed4e1e274503a55b9efe70b0d78f340f77fe26ec23222a93518012d6acd7995bed70da59d81719d243601e35432ba8296638b985efe9396250ba5f1d1ffb41b18b0446d513864e11c5d3ca7d194114457067c8707c549c938bfd58a4dc44e2038a0e8745adce16ede1296b4e2a47b7963ec60c11e195d79910e7fa060eed7f37569988900d2226b450d67383fe65113f64ed8c7e00fafcc52b8430568252a5c0f3e5ca491248160c4c58f65f5e0172199611009755e60f5c4bffa1f4e54a8f528479c753cf28a8137504cbabb0bd8af049c05648b047820270cecc972280abed6b410d9c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483213281Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default/ocdbt.process_0/d, Name: checkpoints/4480/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4480/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=ac5f0a4d9026b735c72111491584f9d06cf6ea18faa448e8f675534abe74bdf7e153b8d262b23fe4b684e8d1a3613b2f71a2bfdfeb2a042049bdcff6e987789dc0669bf94d63a8f14afeb279188482addf9393157aaae77312ec580041a486ff6420f14b4d1f4789fd0b1e13888a8448ec630bf86bbc668618c390f5e60d092afbb3a8d44814d495c0511b293c6267d833b29d57d872ba2748580472343f0288065a06430f7270b00ec3db55ebee1f1b440764faa576ac521ae9b7a1b3bc8b516225db56e8f5eb298d0e79fa63cb6670f27b836b41ced403b1a921016a8ac83f0140ae8b23ea4260cf1569e92862c36fa150fc0a4a26bd42c9234c2cd8ec83d0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.484021939Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/d, Name: checkpoints/0/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/0/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=71120fde60a8bb609c44709a5db275a40b338ba23f0a1fac86453d9d545bfd97f1c5258416a623c96337b0fc8d1f2d0b5db16968f18be562f68c6e37f0cb753198e45ba77369fb6bbc0b85ad61c31c6e1b6f3ca6a3337f139b8d985a4992963e7b15e1b252fd0c03a96b117820fdbfdd2052da604bd884e9ba738ce0562faa90601f4fab27a42889fed66277f6ef9e5df7451f0d3edcfe6be567d1adf2a22d65549829d2843048bf2779d3d9a7f69254602ee6bdba88a37966c096f2e521ac3d1ae45c9ee3dadd06f198cb6b3364a3f787e46240de790ca2152c7175fa7c317505a7492a155d632bcc4690e7038405b37472877e5d85942ccd95eed00e82557e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483396655Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448, Name: checkpoints/448, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/448?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0675da3b19d3e736f748f6aed3dfd65079a75d8071839fd8783cc52b8145dab9e5a8f4551d2a1766ed308ea60be3277300318aad9e4bf42add024b9ea0fd67f8b5df33e5a651780a0ad3bc01ba2632b29471ac78051f308a7835422333a7c6416004bdd6cb793623acdc8232b83faed80a217f6c21529fb5f5964d389e6e103a4e0ce2439c5e4176ec87f5b99f77ef7fd1a04f26400e4a1ccfb2838d56c23bda4e44e8896d026442d8283b4b1f078dbeae01133c365fdb7ffb7bff0c3ffe8129a9770712c4f9b92992288e22e04dbe78d1d2ca59a91a87080f24320d9851bb6e6d4a4f430f6fff52817a1bd6bf1f863c7eaeab6cd196392787a22840d2606df9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48523231Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/array_metadatas, Name: checkpoints/704/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/704/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0ad57020463b42f41a930960a9adeac6eabde2330a6086806fa2a11942e3ff3527a32a1c419cf185bdcc0359faa350202c42ebe34276b8c31679dcb1ce057ec31f9a90b09f9123533e22a9f79aa1305483f6fa6c62eee93e080d841522783d41ce6e6198df72606ea4f11bc0e0a4fb1a1cbe514281361dca684314a14c9a7bd9569c1393d67701d970e504b4ab10265ed263dd5b78514decfe56a91ad5d7c0b7817680fee37d75c2f33df72da607640bbb376c5590e178572180cfbd9cb81b3f5936b8818365d555fab736bdc4b5f891442d5b3e2e7f217a5e6adb737f34f52204d46061dd6344aea622c5b20a6f5fc65bd70a2363e46124749bd9a6283bd683&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485247459Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default, Name: checkpoints/1472/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1472/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=22871ad0e02d82a0da9f5997f6e171e6ff9af3578cd053748dcbd91295e23c3a0bd598ff4e1f4dcd1cb5ba56497340022888bb9b2e75b726f4f70a35a4713e11d86012da9f2781b23af6de0007f3b1d088e1962b24c19ff38ad09349d9b9163b206612c71b1bf926b9cb0285c8ebb6f367bd33dc511ddcfa384d7b46413a9435be7ed38e73429ac088cd7482058ff6ea707c98a71e7fcd012a37a70ebd303a66c488b71768d29042c6aec9ec112094ff25bbfe4bf977db019d1f4efca0bf4a7271ba573b28ca47b9902cf56ba742137d3eea1121e76d08c419a2789827b122cf143fbf33625b4c4b29c4463ac661dd10505d1daa7c2de6daab607c42e57e40ea&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485259543Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default, Name: checkpoints/4416/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4416/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3ea04d0b983355e353820b6dfdf87c5df8458f2dfb288fee1dc6c096606ac6a275f1a7caaed05c3e2679a2aca0134bc348cb8ffbdd21c6e9bec0d45ee51dcaf4e6396f55e2deb553b1cf2ba13bd9b5a8e49dca0a2085bd0b3e2154ea2aad5db79928e5b99d044371ac228f83efdcaaec22a9e9096f19d38710db89c87a64b6841b492e8430eac4dc26f77e49967d10dfbe0a21806e98e25b53e34299980c771c7ae4bdc0fbb39c37dee0acb86c5733cac450cb4cab4879d98ef09b2afa453992552db4b08fc05ee1094389a064223a74dc19d3193fefde79521c3d357590f9709f810ddea1e8f75c4ea8939d6a7754d412b9cb1f0f6d72dba9008f47ff2b80d3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485275225Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/array_metadatas, Name: checkpoints/5824/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5824/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2138cc58fdd74d8daf6568e5e269e2c2dee56857bf2895518ffe28a35339771e3a29c126dc81e650436b85d667027a9a1f8edb9b1773cc504cd09b538aac8a5746ae5b355eb6d2be5be387a35f3dfbe08aae1105dcb37e02636ccf6d291a7be3e08788b69b86b60a462fdc95ba13c73c98d29ec9f9a665f06b6ed47b5e8d25c3d03b42c022c25e3a149cb36270ffa95a1efb129a8858523e033ebe669f4731400b03f766120811eac576316a111531f4daa759c771a50b01b2572b8a0c1a6195feece8ce5d7af957206aabd99fd5797eaa7a56268e246d786e240dcb14012c4f827e9898e312229a7351515fe827f38847af4f267335d048f5ff0443744dad36&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485319175Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/768/default/ocdbt.process_0/d, Name: checkpoints/768/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/768/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0d957da01b9157444953990c0a210cdb6130f86e200f00b4a27eb533b7ed6c1b3697c6884797cc04196e102ddcf391ac16e13c5e419a277eb8b82aae7684b127141d7f57165ff87c12ab77ebfdb5dfa61179a121dcef391057a3cb7998073859248dc6e408be7f0885e6ae655903915c95e30a15de705fea639dcc5812595bdab20a68c0e985407a56f4dcb9894c3de9558d33c36170cbaa49e220cb2007c7379fe70ac88c89082390dccdf8b1f1253d9db50b1299bd2c0501f1540ef9ea2eb17e499ad679a554937300d5baf30b358292f6fee67a13715aabd0442e80f301edd45286860db4b2758dcc8ab7dae9d10df19ab03a5d6a8cc1f7447834d0574e81&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485320411Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712, Name: checkpoints/3712, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3712?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5bc1deea65794ecda5ecd5a135f18b0e569ecf15d63036f20f994bc877b8889a237a75680f92a9b2f77c8ea1e3b35ad364699b6381e3a4fd214c40a9b993131f36b6a26b67e63fae29e38d2720d11aa4bd211b617d890ff343078bc3296fd33c437a68c4300d914135e95265edd4bb534d595b05d512baf5a798022b033187928c747c17eb485fc183138655cb0cb213eef292a90974658a8be6e04e7972f56018c6754de161fd4988fc373e2b22d9fe61b878a272d306daafb4d58fa51bcc2bae5379571e6d2bff7849bc98aa41ca85584c9fa24867e33ddcf6dbcf71ba64fc35969e70399a393c5106a108aec7716eaf43272c069b2c3537b414ff9dda44f1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485318741Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4288, Name: checkpoints/4288, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4288?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4b37a28fb4ddee4ed57dd5238796bc63e825f6fab0fe93f660238197eceb864470aa7c56754d346e721b53fd3d475b49e90ff2de5ddc4dbc55107698023603a8a15a13d0174ca0a8ae5f2750aab6e0281bd39df33c8d308350452617e7ece70473d456e44f1b393e34578951cbd4445c98accb04d518da8afbe09e5de39e586ef8b8a09b325b36071357238fb4e38fcaa2c6535b47e401a59f3a9db1c5429afa859b81bf7954faa6bd8178414edebb9a8c739221aca6052d4629ccc9af08cb051e5703fad7d29e4937eca76d531d06b6eae5c0844713f88c06368bd94a94d1c205a116c3278635480c02fb9a4cd717afe4a116943f3b2f9e5fc31ad8b84360e3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483486877Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default, Name: checkpoints/1984/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1984/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=beb6f27e8e2726c50133bf18a41a1165241bdc70de868372c0ae5051deeeacf4cece0d7e3a2977fe597664c76a6efb4c60f31ab52e5f1238b2f52f8842be5d97b98d653f69c82e3e86fece58b91adba3b189ea728619667dde0ce0b10a953f9364d3e74101b480a73f2747a9e44b68c933e64109281ebdceb13db03dfdb4613c74de851a64d97a9872be52d8dd00f5a2b68a3a164731537430e6b78212994f4bb938fbad3213cbc9c828aa9c62cff6c80c8b72e24172a6ae6b92b366735268254f8eb4567a776bce42afb2f78183f2317d9d39e5008024c6ce763ac5cecbb7202c56bcb3e771580c4a9ccf8b9927c771a4f1d727b8595104b57fd83ac77a14e8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485348387Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/array_metadatas, Name: checkpoints/1664/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1664/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bfea166bb13d651e5db1571a10e042821853b090254a3ebf42709d911b8564ca9172d278002163c6cfc40655f49f3a940c197e46d8db7f5a8a88796d90836a31f96f52645d95ab6890c39ecdc4691639bdb127cc40dec8243631d87daac073b71c66c075397f1d0151fcf0ccfbdf08fa4da6d84c89d3522722316218cc31708c432327721800b8a8d7f98064caf0d3358e9707089614a43b682c5ac35c35fb5e741eae7e5474a9c3508d1d2c400910de866f64a70856a54ab250f1bc74e389724bc9f4e1f74fbe4a62071141ab0bc5c3f18a44f4d786003d4d8afadf80837d073b784a59b58f8d7bbfff363e96d3428229dff321964733469b8c14957d9febf2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485378728Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152, Name: checkpoints/1152, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1152?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=930d82dad8053e34dbf70c3116890be7255637ec61ae010e7a578ce32391b81fbc2b861342a83c784331e7882e050d026408f14986160ccbb503d2f8cde27e1ef87487f5e397773b22c0da94c2abea0561303c6669f9109849db336a9b4592faa53b54b1f25e2f982acef3c1a4cb62e53e897b9a93380c250b025b2cb799273e58cec793424b20fed723beb6f214970da3bbb4542162dc0cb75293bfc323d62bf7a5111889359f30031624876a2ec03a92efd1bbaade16fee0d24e584edc004222e896f6e493203f634784033489db4b980537e133aae5bf8b3b75a1127c84a2401aa21fc93670c461943a14c478f772083bab8c2384e0847369c9b2e6cd27e2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482565118Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648/default, Name: checkpoints/3648/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3648/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4abb5808c94cbbdc08b0d2ce99c93cffee154b1a5e33aac84eea81ca2f1eb129b69d7cff635ea0973d547c2312f82b5fdd788ea86f7c975ac81a45b098ee26fd47c8c2a4790d6d955617dd3c5a0f85961b014311971b1734eb6fa8c387185e62171af05252e7bf758fe11bb4074b7e9fd2bf7031b4e0e86504630248e10436cf98df3ce3b0a3542ccd8550729bc2037afd16e081592d5432c3ee38793041f7e56f3d7420b7f4c17b611e33e602514da2138df5de2325f44a703880b0d520874ea4e7f254162372091211d51dc37aad2cbdcebdc4b61a4a606ff896482adff4b94102913cd54049f30a91842ff35cc394e44f5cc5d61611bcaca3979b88178036&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485425757Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/d, Name: checkpoints/2560/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2560/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=06b692a5df9a8e2ebe713eaabb536d39bb9e5390f56948e8679a72864912584c36ae6aad8ae3f96b4f76da24544702e80d1866869810557c31456984b2ce4cd085b1c4294a276b32cafa4fb2366a90d85084133cdefe338be1e7dac57163700410fa7c27afa4965411c087cdc845a6bca3ab0fa9c898cc686d4bc8778bbd7d0dc694b2c6f1968c5d471b57a975b3b0ef98dfa8e9791133dc355d909fb63cab7ba49ff3d278572dc71b40ab6e0ac5c4f52f5b97dd2c9a9d5ea69eeee123525010fbdc0f86010d666c8e6fe97c0f00fb71148252bd9a143260530a2bcd3c9ecf2e7f6ce034069e2235c441b6d0d684e219b715ccec1fa3858709f91fad965d32ff&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485466917Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/array_metadatas, Name: checkpoints/1792/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1792/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=be8bd0c2bd2bb1dba24bce573d0296abfd9dfe2a72c69cc01ef4ce15fba2ea32966be1efdc9302ebc8199f168bed3a51b2fd1c7ef8d12c7060b01b993381850e76160d021f77cb0c6a7eabf96944fb59acd34235a243306f9c15997c443f092a0efe9d375c1e8febf60dbf2773dd12caefea3f22a5c959ddd8550dc4e62b5aa5c1471a9e0631e3161b437482c973a2f7e1e2fed9241fc4db7d78aaa498d0ec5b5772e43738a8cfc9e825b970bbb7c7fa8488cb1169d5850dcfccc0fffc73630f63a192be9d55318bbb32225adce083bba8b75f164873c2ffb308a083f29a3565727626caf2f176134ffca515b95cdd971d6dd72a779751aa5daa36c1af2654a3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485556635Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544, Name: checkpoints/4544, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4544?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7b3619888752c0beb82c6b8c4da881f1c2c85d6da779b3ba745aead52539b53a7c01b9b3f448d322cd4c5fb8f71d20e1c4ca39fa710e9b7d934aab33c204e5eef93a7015601ff271ac9fdfa5d1e80605a5f75ca7e16536d3ea66f91b86c1a148669be8c56ed2d082b01e4f682aa4938b0977291da213131b39bf37434bfee349bffe25b02ba6c583f78877ddea5078ace2b93fd9dac5ef588a152ae73682cd489700bf31b41187dea322fc83eecae821ca347c0ee768b06a4d5ac9ec53566f7b901f493a8531cf6a687a98a398caa9c233801215e84211f7f3af48461e0fabbf700f656563a661b045edb05306424af66123ae1ba23f91a8d3d0dc6f2715ab9f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485567528Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/ocdbt.process_0/d, Name: checkpoints/4992/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4992/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a1a8828a98abab611d9c6aab15f0dd1342f3db1f9b509c6167516d42ab2b5cb768ddc2cac4263edcf7452f32eb9035c38098aca3a5455a87bce55ee6b78dcf7fb30aab7180bd330759195978bb5789f7ca6da87048164e3aec5e5aef2a010747dfa2defc4f905a6d66f0d0128c8adbbdfe64808ac6a2948f77b742c46bf78aa2eb5db8cfe7ad2fb77de1e463d097a3a0c826d7cb0070eda624a4e2a642d933ebbfb5f42de1655756b5ef86b9aed12bcd3a536f28a30ec5aa7eb18b048cca0c6805a6abdff751ba9493bfd9ebd6a598e94702f98c97f26eb114cc0f8f98863a3858389be170fc1ac6ca5da4ece7c15d7234098551b91ff0322c163eb43bd0a0c6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485575964Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/d, Name: checkpoints/2944/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2944/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=83b16213bbfea83310c270b2c0b05d07e845c0bf0e9bfd3478c285f7f121bafd4560d0175bcb8342166d2169b873400346e3c36ba0fac8ee62d3131456fcdf501eb0d8c6ed6e20e17cab9e422d328ed0929141edf13ed3d2a40bd57a3431f14b0989e1634c140f42862327357f762039b83cb21c8ba025899af0962b6032156ab3f83dfe11d1567444189cac56c6851717d9298f6a1a8b52e2af63410b7529036b72a7e49d9e95c960d5b0b7b837e36572588054e9a85c570dca2e72218dee5935575aa989aecc0e4c016b737961d8423d8deb290b2ed9f55cc3d16c5bc1b8036e2af596175d6884b41771e25d0699e8c3e302b23c3bdc5e9ad39e295f7ec2fe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485625753Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0/d, Name: checkpoints/5952/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5952/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7a6f160afc1a9efec258449bbd77b4b101b4ccade8e58b81541d84ec91e8614b1d198cb52f47438ad4cbf210d2ff14772d1cf9eaf3f8a33e08492dd87be8b86d80d1ca8a473b3ef0a5a2007a073014fa421b4715c55f52b67be88b38b449685681b268662b0faafce984969474cdca27e9dd7fb7d057f4f794adc447d964fe4363d658b34668e1c80f6335fdc8285cc80233e165bf98254f11807c06bcb73b266438ccea81d1111d59318ed231e9ac504c4c2173b6c544dde7a54f6bf2e18701fd59b0fc7baf7a5a974190ece92737e283e14b28d5033facd442ed98c39ad649fb04dd34b5b889e127cdfa689e1512a30b10f6024cce915872e958ed08c625a2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485656875Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560/default/ocdbt.process_0, Name: checkpoints/2560/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2560/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1735ef89a4486598d4284eed795632fa6a2d8117ab24852ce284836eea3a9f19f35e736046ffd369a72211f605684a04de17b535385dac4aa49545839fe18ce14c986b3e03fb83cfeb7627ca9e371d5288751c3f7b09f0d3487d03e7f7c33b797183e51a78d6eefa3dc1e26bad6e637dd79e019b96b531920b3a12a225ac6ada01ec91c668ca3808d65f29a9ba5ba3eb168106481e2a5100016d3b0a8127c057c8d1d4d7f67a448e3ed3c70dd728efcd2971a933ac6e52c26b90458e7747fd338888592ec00921f5029aa2f89aca314ea77fbda96997db0b9423fe2e474a2d627cdf2a455d77f1393fe48c5c0a37edd061572f753f3ac5467fa7cdb8a80dfe0d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485655523Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176, Name: checkpoints/2176, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2176?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8e33fb24f2b8cab60b85e193b40e9bd8799e53459f16769e1d276618fc186aaf5005b65073cb53c905a988f935f8dc69a8861e99287f85f1f559122821d0d15e1f2b660ab4b122f27e5f28c692ad1ae2892bf0a6ee0eecd5a24643e3900000777263f025f821559f0652f92c09cda5ebd623835cfe659d235153c0282979f7673675a29a9d7525e57b63c42c0a6aec6a9b66b7a7f02307b38c2363c9e2839127e25b2cbdaa47e8a19bf47023a312c348ec53b8994a61c0b751f6b24dc688aae139f89741be6156c3dfa3e0dff27885d79214f45eecee494857383da7bfb7a04fe2d53584f6b1a366df27fe9d3286de79d07870cb54a55ecd0bc3dd37e408903d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485666881Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0/d, Name: checkpoints/128/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/128/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=80adefe975c070ee83790ea9709fe0490629e220d7ffbdea7b66b5d7514584308b62c09b9f85fb494bec368c698a904b703545530f54276a9a944314865502bdd3b66ca8229ce74f38434c680ef8921fd9c13633b53f52e78c79c6974980626c18e320ef3726ff5f04c632cefe7e816a3723e8242661ea42619626c3e83479d1a6bb87af924ae3b7005594409cd6f89d0c15880bd7123e4ffd8c7bdcbe0e83a501bf79c654afac4738d732a418ad44e255ae2c23dc7832f852a38ab437037bb9bec10b519a0215b969571ca63b68473a389a491f60b82a50e4c8898751bbb452837d59f18168c37b48b6ec3293c71a9d0f36827bbb0e334c44860410d013cff9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485667149Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/448/default/ocdbt.process_0, Name: checkpoints/448/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/448/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7217a4dcbc7dfca9a04b65414c5598ee05a82e09cfeabac1b4696bbece751c1e9393dbf8c5fca7b1517c143036a06af2d9698dc719ed3192046faa11769f0748bba6a13ac3c1d39a7ce46f5a2eeaa03b10a86157bd998a548d0012e1199ed7856f13a9f7546717db225f135f97606fb53f215ede4bd44a033c046e81640b4add5d2433e7aeb49c0e4c1ee846db206ac39564462242a1cd96290ace4100cb4433fa2a1223db781bec9fdbb93b89e4b17bcaa9c9daa564dc512330e9071549d833d9072048883e4011b850ed587923b5509f3d9b72db4fd3ede8cc1f311ba79674e050f59f65229d1a87a18c247e7cecdc71841d5f7695a5f2a8cb1472a4b06db4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485697916Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/ocdbt.process_0/d, Name: checkpoints/64/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/64/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=49e25c7be8380e10d1d3f87a1950cfa1f4ea3fba32ba8e6f4895ad6df2f608d09b1f36f1da999790de750f8cbb2ca3ee04d3680bb88e430b58e970c57a79a3c3e587e72ce6ee5daa29af46f5a35a36c6351661f4ab78f51318a69464aa9bba915103d87366596bbf46a1a6b8e5b7b416ba18f11aa30216ce5d156858b5d66375c84e5f54b28684206625b75af989ff8b905c916e959a72bbd4e0ccfd8569a00deec0e282f900ca760771048c5918b8755b9082a94408b4eb6f14f4eaf4e51c5b2eb3d5eb49b36d45c3d99ee51fb8e930aeb5e1358cd34e834c46110691ada5fbc32b9c11520ba98e336574ce794d7f9470d61c0a4489a91abe11d03fa105ede8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485724901Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default/ocdbt.process_0, Name: checkpoints/5952/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5952/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2742a5e075d810eddd1b2d492503a0c091d917ed6e6c5d50ea0e60d1f09f6a9050eab3705882179908ca1ab7c5bba6aa5b1de469591a30898cf4efd30e5255f33504eaeba1623d9c0dc503fc8cc0d57cc9fd0818f8019925a425cd526a710433d0192e033ffd4441d9f6211302abf447bd089808ab1b79cdd936396fb9c895a8e02e2916f682a323dbe73618b77ca6d91c60a59265e90e889d1ade3cd9b5f2cbf56460cc178e3923a0cae737c6b9f922a51a54bdac2c7c7e0c85b0c4c68c46ac84b115009213fe55dc634d8932c24eee56fb3a1be5c72f8f7f07d152ec5d1b0eecbf3485f54936633cc0cec60e7fadfbfc2eea7a7e5aefbbfb813db8a2d61403&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485744556Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/array_metadatas, Name: checkpoints/2176/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2176/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=53811c96aff89d43f793f52f2571c0415553bdeb01eb1371629c480c4a8dce732dc493aa8df5cba8d5771b2b14f5e7fdb326c34c53593c16c0c4d3fd0d9ee6636b78327a19452b9d35ec9c563f4f99cc00e0f4844def01b9dac09dc0b4aaacc946a273ddc7f09122298469e450aa2a82a5d9ba9f7917a7846b2ac8cf20a0312e7dd7b7df0050ce719729ba624329e1bea519978b0438b008c1c4b0d3e1e4af442ea16183f8bcaa46c3ae01dcc1d3fd850728669ce211723a53457d92e9a199690aa17303de4121a158f2f635a26824e87920c699fec9feef01a0c3f22df698d585e13fd8be54085356f4455afd6dc934e25307dd1ced975943a74bc296f8f748&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485767742Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/ocdbt.process_0, Name: checkpoints/1088/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1088/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9289fbfdda8024049407f583fcfd436fc0bb03527353eea5f9e02fe93e38f732bd338601e007206b94daed48eeb4f4af63532c7edf90b0fc1c24193adddaeabc88ff1b378d0290a96eba3e474f60b1ce286638c2ee2c96ab18cae9e19ec88a89cf4ec80aa9e8bbb996025efcbc76acbdc2764e6b7018f882e6e1f04665dcc1be95b2ccf0d6a206e0554cc3fb7fc18c514f31b2096b6e59141d0af735b4ea9ed3c74fdbac4bd48d6f3ffd4e4fe3dea26a20ec5cbaeffc2189625276af5f2a2eef707e9f0ee082cf86b4e753df4715a2d90895a51efc0bb74d1a407d446b4fbe2558a28e0f69aa4f1483908604c089c67e441f30d4261220931c622933405bc903&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483928516Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0, Name: checkpoints/3712/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3712/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6559d6b6f0895efe58f985534910c8ed828dd84a433ff3a191f1cf49b41291e66955f3d25e7f184592097b1d00bd7e23dd7907b25eb3e5cb05cf49ef3e3e1b9e7220e4a3dc4013694a9d196c57303d860ba452a91ff664406f2dbb73774be71fba8093b241836ecc62329e11d62f8a56e6133f04c36d4b0257b282f5b03d5ce27a7c3dca8f0dda61cbd26797c576df7b25c7d6f3fa5d5435d4df3367e4406aa7e429152d0a566c61132b897a5b741a055454751eaac40261ff310f41d4ea6c58908a1dfaf6125ef36b92c21c3525484b0bd17d85e34b6ce8b55c9b52742aa4cd5868272596576a0d366f1dbaef589254add2b179cb1f7d1aea1bb3fcb595d985&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482939839Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/d, Name: checkpoints/4096/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4096/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=113390b08d447051b6f1b05c7267b4c6bb4a7b0fe8459d97003ac5d86e975add59a3adee0f6339ffa92c52f6dc3ad00307ad5088570fc760c57f4db9e83a60290c651a21558957d45815b349e41e1475352a7d1b1b13b431aabef5dc29ed0898af8cec95e002b84e0a45fae707b96547a29052ea0728cbf1088b3981c03f27fc28b9c9076cbff6b4913784170f614a2578d7a47a7156530eee58c6533d6978832d11e2bd76ea9e8fe29db1786e29b318b9312b7e40c82678b0974be2ca31e813381f9aa0b58f32eab6ec4ac788285a3da9fafa7661b71850ffd3059e6061710daeb165b932903458fceddd8949431a7023bb02bdda799e711f524d11062c08d2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485821594Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/ocdbt.process_0/d, Name: checkpoints/960/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/960/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5d5fb2327f362bf7cbc9a09d75b45b680ec2a280e667257d6287974bcfc8cb5067657e623a8528568716fa520b558ef34814a5ea0de0b36fb232c06162640a6842b9c57338fbd3a991fbd5ef06d0557bc94cd41a2a4d04688ddd2e952fb8f1ce82d99bdbd9730a0e6465c3e8692285a2d03aeffb6b49749afead8fca5312fb0079405e57e2d9edf2ccb5ad426136f0282d329c3f8ab521099e4f5a0ce499a37d44296e9d8323b047f72bdf5de6541035a0ab839232a66acabc5cc20c026e3a713910d94d25d07bf13cee0d89700c12aabe6d922dbd1a7bb226bf8d95c893d0f286581d4a42083d7f4bd432ad93657423044b46ecc4389c66dedb3e94cb562c8b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485971458Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0, Name: checkpoints/2816/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2816/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4ebf9ff81785350ccb5811a6320a16bf27d674af05b26a05a24d35a417b6e14692bf04f6955931fbba41945fb3584a49b2281332bd22aeca921130894166f326474682f9c0bbf1d64ca8ac51420eb0641a89ecbdb0e4ef056af4a05a25951122682bcb9b6f007974fddd5cb03f8df7bbfab8d797d6187749bcb19b83719f4e2965ca50835a58ce6e02cb78f911f1e4d03a2172c4356a63c1a524f9ef7f2e8aabc876bffa84d671069b37a222788666b38b394107da908c00da6550563f554af90e245bbe8aaf92820be5cb90043d213c4df9aa5632cde4a8b64bfc1ec55ec820291bb4445244b347acdbc99a7864d6f716c2c3d2471f51496b9c9ad5a4bb08de&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485977556Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896, Name: checkpoints/896, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/896?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8a69cbe80abce82d43268de09b4ecbef54fa20c65927ef48bbc97df4e1a56cbbaac2a49fd4f4c9b08eb6a80fa45c5410c10dcb5b6a6ae33eed485d8bb059a5bfd883ee2908fb3640a3516b4614ad9999d7438a09cb25c88a2a186cefad082c46a0818d0cbe1b34e816f66c7dc9dc2a21577525d02bfdd5c258c65140c6dd32f4f5c844b1a2facd02e30bb1bfec67a146bc186964af8d046955687daf60f65a02a6e4eb56a681f7ce18e9d5ce61bc4fb9008e7137eb5c80e2d956d6a851be21441e1a21cbb8df57df9b69582cac699173195cd2d1a8c43ec03a5d7befbea60db396d6345f42a565c1e8609fe478ca2c824df6388a5eaf49acb48c64558c678ea6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485979661Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default/ocdbt.process_0/d, Name: checkpoints/832/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/832/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b045dcd91d12e9c314d4ee5e22d9da03f1c1cacb1b3b3a1e17122385f64ba5fc41d67d07f4a9709edd8c4f1fe1ab50a7f1687f001d81daadd948905acae5f606b595a3356b726e207f66d866f1a0303266058140781ca728e6037588663ef9bda89acd2a97467255c833146eee4a63fd7b58e8573e9ab34b4a83c7bda8c5eb209c57132ad0a39dc257e88657e85b05063e1fec27693a177c1a50653ce8989617277173e4ff44b02a4925c76cfdd89d351a088d6fdc323bffea3d6e6269f2e59d74c0da4a70c73db5bc79da66d7886bbb899756ae995c81452f34856b7e4e2a4e6edceabc94a823e67f804233329ed3c2cf551dd4fd158abf158842570f86010d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.485985186Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64, Name: checkpoints/64, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/64?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9d3c16de689401cdd01fb559fee6812a0f31cbdcb7445b1729437eee2fc11ae36cfd638d1868bf4643080fa4c45789f2f525469df5de719bd51e324512f0b6912e49b2bf557e50b23d26198c8aabbcf0445bca4efdf6d5bc3e4c6bbd879788b26ca72086ed8fd3073c6a5bd659ffe81669db3b47f264696cd7abd69dce32d08881cdef39e401516870f4efd29af3c19b87d06c7a67021c8d37f5f1dd6ac00db227e2be48641df11e9835d5f3a5ae3635bb0478bb94c272175b4ce91285d850ee57ae0b4fd754a63879c995e2f6a07aacdc4712e95e9541ab2c61485aa1cffe4d58329903f22fd1eb0f7eadc93b4b482239a370e7b338e37ef550c04e465d97df&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48599076Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/array_metadatas, Name: checkpoints/512/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/512/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bd62c1217167fd6e66e39092de80f2ab9752ec07850380ab8da4b38e82fe31ce2d620aa47eb6f10830b158c7dcd832de6bb72e893751879b8bd17396986175eff78c852474d283e2b22dc0654802ec327c248ef6a644a3f6af7da00c9f2286d00d90b9c0c8f117bd645af12ead78dd74912c3aded3086282883e1b7a445d9a5b7754209383aade0d9e078079c908bf1ef1b6986bafa2278b6dde6839dc716a7d471347490f245bc7b9692180b1ac381e6f2179a99e1dc84004780da23ba3c67db0214efbffedbcaf356095b5cb2b77836406876bf3858f807ab16277f131a0da82578d299029a592f37f694b3f86ec01b06c9d9e3147f9bd84442fd31e19ec34&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486019104Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776, Name: checkpoints/3776, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3776?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=80a93c587a519ca3187a0d36720fb519c4cef75cbf5d0490b259806212f5ba479b62674f1dadddc2663a8d4d87951c14b2c127ec3798caf2b2336532788e52c38be8771d556ea9bfea56d4271058960b73af58664d9eeb7064adfff5e78f90a36eb1a88659916460720cb2a58d4c6c57479a09ac4a7e8bc9042d9ddf1b900f57fb8f65b3bf241c7a378d5209875417f7195a9c72317ce9e77c428efbfbe6a33c12ac85f2bbb9a90d461e1d476f7d63abe15754b040c59cc7fc7e856c145de3c023b242beb14b6027b8702ee90a228e9a2676b3b2c2a55a651de12b5b943d1df84e134bce943ab5fa3b972ba4a79acc49dfa59e01e3e497d20f046a295ef47ade&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486067277Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default, Name: checkpoints/3840/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3840/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=765238539f914ca94cc4694295e7620495940ddb4040f2d87eb7aaf1ed8a33e4466430039b31ac73e9e71a171b8bce1032a8afe8637d84b939091d649b20627fb378f2d3ec32dde8c3a1669c429acffdf302f611d2339175b5bf3b95a910553fd32438469603a63743c76543295cedd002ef870e7c6b9b0344d0355451921b3f467d53d86234342777801ca29829d9986508825d6e271ce1ddc143fcf5d9895cb6d633c8f7e938b4c8a135b5b259475f4acd47cb945093197bd269d6fdfc94aed28e1e409cad913d0d296f22169d01087cf5c5c8ae158306f53c3f51fa3eaa74e442e473d5b0da472d14b6fedebf869b716bf0c5c7d4e879d95aed09c1c34b7d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486073149Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832/default, Name: checkpoints/832/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/832/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0358494cf2a39c87208d042f02430ce5b5b969b99bb92791ed74203a6738c3e4fc212e70917146e4c4d6eb26f6dd8570baa5641f409ed827c9063f7f8bf4e545628ab35d29f9b8ca70e170f4a48f6e877573a5cab3a5769f69104bcadfc0ccf6f9166dcf885c073f0a027654726b9bfd30fed72882b8e1673e0a0a1bece7f1c1f74b664c995a2231eac6f3ff97db86e8db47163423b93af574c0a6efecd7c998ee8c4f5a3257964cd23549663af504436332d582067605fa1e2536d5d0445c51dcd39e6d3ca5e2115c580b040c757b3070d5753f52eca44fa5e07a9c99fb4baf4330720a71d11a87310c552d5a372972f127673d60cb8a8d1e13e1a8bcc1a91e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482962525Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0/d, Name: checkpoints/2944/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2944/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=276a8500e6a81443bac80bd68ab6a1da6ee28d85707a70a06c9900bc25cd9e34344e4db137ab1891770b36dcc6ae2377e34a1ebf5971579d365488a7b8956fcfc68f096d2ecd58f6eaac8ff37d93c93d269e956da416e25f5556516af067cd6898b132438863cd7bd16e655c5a8c7503993dfebc5b725e0fd48b47f8632530f1639f4526a6ec0f1ca4b2562e35cccce70d6eb1e40c1ffde8d96a08184bfe8fd13dc521bff7da0d911a6f1ee333b6378ec66524860caaf1384a3033331538aabdfa05cf1734d7cee72738711e3cec23315416a83dff4871c8aac61587dc5bdf97f1186178e01dab30c3018bbce7ae9fd1ca2edbee16300a0056df0caabd7a274b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486169943Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/ocdbt.process_0, Name: checkpoints/5056/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5056/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=78e32bc1276d066d6bc28a49d5396ee39baa0475c70564604d4b86c120674793bd1560d5b1c28441e1bde775e7d0773d54d7537cc3b65da5fb98c257b8a17cebc24175ada11ca53e28edf8c03d9ca648f4a3335250b7fe48ef2e001637ecae7eb3fb0572663445b87bf2ad669420826983216143b321313cf3388053425181c9aa79e1b7c56cfb78322bdbc67c4d73d5abc9a9c56221295d82e7179a1e8e9f8d0a39881483e434d9de4a76fc92bb48f7c93aaff69c2e51236de694f63e256c5efa1bd84cfd85c0bab18394d8d0306b6174fd3380de7e3960f7bbd204c3de065cdc01b2e3c043b75ce6828b8658b07c8b4892805a232db1a707eb96536607870b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486182629Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/array_metadatas, Name: checkpoints/2752/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2752/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=52d8e00fd91304c3b8ebcc28b36894f1ee2d8c83d018d350f2dad41e9340a11b43029c3387f5bd17d5a006d614f16a41383acfea06d477e8d5d003c7c4f1375d86a05643554cc32f7bfd71cef69b519619b79fc04b1f3ba1134c70607fe86ddcec3fe27db23ae9b23c7bb0328641537d3c97e07fda23971c03bf9615d9f87de888f1cea98bc8c453e6ba1aaa27d8876b531fc8e686f560b701b50d1b73351bd5c6bbbe5c4dd0915e384870a5afb148045daf000043d85415202fa3736a149470e8abe74a27ed92a166b3120088f08f554935af2ceb3e1f21d5a32755661bb9432a09dc876a740335139e5de8dc06021f435ce20e0a49340041c236ec5e17c57c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.482991094Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default, Name: checkpoints/0/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/0/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3f7ffb60f7f68af26204c6a2a45ac7bcef611754bda5c2d717ed2134e0bf96216b89a538eea7cfa5191feb9520bcbb512db328a4bcaafd50165cb51b78846912a78a787f1b2f86f4c20f49afa2ad4a9c811b1f9b4af3f711550930e564aebaf0f55e6f51fe6840e4a7771a8ba0885a3f0292f93fbca985c4f927d39139d62ff68c3cddf016feb688e84920b7570783483e21412ac0941b1ccd9eb9eeb0fb257cece5fa7256834e0c46dc4c654d76659f3daf6468942ad53935d23a88a772b67f4bca6481d5315400383156cea85de423715ea313a5014aebd14292960d24a7b6aece6a6a203540b86bac8a55672ec2b5ced373599b2a58d0608cc8cd0cccc97e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486300713Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/d, Name: checkpoints/3264/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3264/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=97e800b24cc99a63876fe67e1d90093f43138092c80a5acee7161f47c4f14a4ea9d0e57af950d76958be994c70cd8f39ac9246942b0e59cfbc7ac9f52b8d60054066815e7043518a6d394b6b296b48ff2ba106d6255a08ae19724ec1f89d90a77f2bc5418bacf8075252286ae75d7bb2a1e71f5b1c047952301ff292c4538162691efa93501900e8ceb24581548a0c1f68b1ab6b858eb70e5e14736f1bd9a754c7da2df48bebb37589efc043258433e0d25ebbd9e90daade7cb9489ceba5f0b77d56fffa01c84f6b9e9a1fd5adb89bd4706449698746d84e0c5e32cbf57669faa1842e587919a4131edc4755d27b2ef0be31dc5a1215efece652e83605ac3450&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486321329Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408, Name: checkpoints/1408, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1408?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9a333747c66a177acf81c6a0c3cc7412767dab4efbb2b7fda6987f6ac68ab8d22c5215154f9c81d9955c227043ca29ac917a71fdfe097b393ce6f9c53384465fd386aad2be8153fb7760b60b9f8798c84520066c1eba1ff8f660d18300f8782d69d7e03c0fef734677f4df60dc35dab927b2f8a4c5cb41783d1209a67d417245dd862dad9c0b5f99ac8c94b6c85f1b892d3ab58d86db6b5c00a797c15343743f13e2150f09f952686ef5454894cf2a59d2fc4bfeed3419187b2540aae5a934485d4c3207967aaddcd787a780d86e1b6b2d8019c6c227fda8cc42465005ffcaadc20c46f3fe7bf04abced80e7b50df09d41b4932e989b051949246898b1f650b8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486307401Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/d, Name: checkpoints/4800/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4800/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6fe49838bb4a6bfcf81b05d846c5b67e1fd313881db17940929a4e435d3741dabc51655f98b944969090697ae2b0bf2ad95639f8e558893c56b7d5e1cddc0c103bee615b19c1de45842ee6888e4aa79d46895812c3d1689be5f122b5b6784ed73a35812b54e11c7a5936f5d2977e054f97352303913fb6403030a3b30bb1e2a1d9a3cae42791ae542d6d18938d821deb37aad96a2d0747345fa5f2fce574ba1ecf54a7fe1bcc021ccdb1c4e992d8463d482a4e2d202ee3652718c84d2cfe341dc677d72f45911da07a4f5adeb30779e91c37ea5dc94db56330b1b5fc0b70a7a64d0e7be7beeba199f0e05b3772e184b07c28fc428264325a1f083667a32a8b50&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48639596Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200, Name: checkpoints/3200, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3200?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=465cc9601f1ceabae5e9569f5025456c7a303ff7a3a4a8952254c73b5e7494474a019f10fc620156ac7c55196827d8e50a18f403ecc7617f6fde5d6bf609f60cb87a47757f5b0451cabd3d23b67ec1009cb7fd6a3297f4ce2adef3a109ddce1348245783837c2aef8a216afad530a2400d15c3975534790b7ea7372d3fddc63fc31005d39f6330145c945cbe56fee1678c75dd259d6ae8d49a02b622c32e12e75ebfee691419cc3c27d36b8e1d495969757a0aec1338511fb89334e8bfdc3c458483ad47541b00f9b34450ca48bdaa6ed19596536f4147a9bc829d9a406150026a2b858201b02946f41255cf4e85d6b1461d836d01b24b278d0632b1093d65b9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486463812Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/d, Name: checkpoints/1088/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1088/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3715f5eeb5178fa67e954b5cac5fd8f3a5de755880854706b4dfd4693aa49694e9014495cae5f233c37899a2abb4dbe01e0a450b24e2c18ad4358ebb115e74a4fde652fefa70067e5999bd5055230abe24136f9cbdf93f5d66d586d4e1d6c50ea15ab6b159d2d9eb2d67c572340e7dc438be41764139d5e93a3b19c57a4458283e34a2c4dbcaea9d5d9d757068ea7d4d29c3fc50fb593685dca82605198905dfee0436a18b2c384f5c03f94ae365289daf83e36c1396117875cf45f927181101563307098efc7a7e704017d08b2f1a596abffdb1f419ced8a4216bfc0d0621719bdc37f8a7ec4f51dd5aec89c29c18b032f1ab59f5c4def160f6f780eb507598&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486603606Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/ocdbt.process_0/d, Name: checkpoints/1216/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1216/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=84231a2150e0eda503c236c9a1ceec9f6519ee3bb5e2b00c0b26b69e48902992b7b0c212214c999bc79f14f5376e966d06e4fa4fea4fc1aa604b434e926b050f216b787335edca5a8620859a855635c156421b5b7216a99b507cfab04885f28c51a5c82800b30313613c5e713639ed6515a9ccd8ae46ecdfdea447f9f5af4e1945f1032724ab1da921d1f514cb1b04185a79a6fe825e03c2b7bb458e63845a67f8fcdf8591dd08d59c70903ebaf2f9c23f9c000b05603a7e66f8690f06120fdc61b67ffa29eaf1f59270398c8b06c98d8e4e2a783c41dedd33bd18d285793a65d7db2ac35b0bbfacc46e077e6454a8a61576cde4994f0de4267a46d7ea7b9f34&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486648451Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default, Name: checkpoints/1664/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1664/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6303b4c158f2c51ea7218e1e497b34578d4b72f4b2c503db5a03fc888652e64de50fd70150060bf6af6a6273b74427f7281b8c8e3e15837c2f6acfbcb9123826430f5244a3262b120d3c88481716bfb9c4c9a6f08218ea00b2e29b30e7b8e85dd4e4f90d03a356d89c544bc21d6120bc7c9d7cfda7e6f9ffb5f143af22a6a809a69b3cff039bed2f711de6050c1e5ba6f2542633c93d4f3a75a9c42ca70de4401183e5fec3b2e6811b035e0367163c2b054ea5799889035854d7c4a13ab1ba1e7e5739a20d0875960f21b1bd1212dfb42719a79701c3e61b5044876362746163a4c820934455a9968f2314d8c6132e3d5c946bc17b6c5a9cf8dd5922a145af3c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487120623Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/d, Name: checkpoints/128/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/128/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0be6a2192b22c40f44c754b2c3349b178484acc1c0d74f57e3a2f0e07bbea5f0524f0987e6634bf127e5a50df6cd09745adf37d904429e4e64b15f874d145dd1904b4f93d815408ec93c2ef116c779ae4132c8d622a81a72a8b52163ee4d5f5a9d0cc63df0be2a3d2e73276da66e07262da4ad6a788175a33a916ebdc49b5105c6be6f0e2b48c029224f76a164599eafdff17c2693db174827174c96275e6c618d4cd966fd93bcd17ae104617cf1955a2ca5443861d2164edcc72f122bd59499b95e317e648ec6e503c906e365fb44c1160286a250b90c61437cd0ba509268ddc6f1d294d225ffdb8c3f16d0e071fb853a92b6c4d95f7daa8e947f8096b85ca4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48713864Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736, Name: checkpoints/4736, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4736?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6b6491c05e932af0e213ad4f5bcef1e6d25db51237a3ba7365ac3d1b3c8ae8cee0d8ec6cd6af33e5f6cae5c2b4299dc214bf27b642e39dbc3fc7c570fa50db7d7918b61d35cdc348579f7d8374b9402b68bae625a5dc4fd94c5adbeede4369f9c0fc60c99e347f15828b87512c27242b1d50776660ed8fd2635d0e3d10dc07f8f271542c23dfa8226fd7cb317f75ef7a6d2aea872d34005350c0c217cdfbb4b2548ee077402d469b93b74987e1f2f65147cdfb69e8d175d689d125ec544efc85c577ae05c775a69cf2ee11c17c4ebc7aa4af6e0e2f671c843172c8f97ceca451df5b450b8402c9d9bfd63d4344cfd99044bcddc75f908b1b21a5f2289685f616&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487143426Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080, Name: checkpoints/6080, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6080?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6ee8fa58e3d3582aa1b144e8fc5d00729e0933e22ccf6ee2e022b5a3c0cdd69c8f9c315d788331bf9bd566ceaef5ecbfa70eb5fd8926e5b4633773a19ab34ebbd42a5de280792eb1cbfc656adc9be47f901e5426f6a83a17383b5a196707942f165d2c37d4874533183373e0682d4e8ea234106ec01cf9a649b2af760d6896baa0c44dfcc341bf3068a223181554e1246ab4a919d11740443d4dad9c71d2f6f0c5d36622ed4f96063a34c998239fc7e44f89666ff0f49690d34ee761d7483d8c103c92ee96589dd9f0db4fc81895e7a3b39463d674fd5f09f840d0ee04ccebd1589694ae8e9436f0a5e38ebd339a64f237b5c032b3e746106f4903e5759d0028&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487179262Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default, Name: checkpoints/1920/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1920/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=90a1711099990656147823c9569e0a91093af9954ea03a907557500869fe1c03ec35559dc9f3b6192582ca333adb151ac4549c10987ab122e44264a5d4ff08cd07d5b994a740588e758189d75c489d71f884da1f9f33d476e4f8b0b1e2914501a299e7a54c4749c5fd6ab26e84e32c748e64b9002e21099534fa675f7d7f099ad65c6141dc7efa3090a00baeaba9b1f5332d370645727a2d4037b8e8015e04ee5237e8af134356f8a05d4a981ff132e755243ebc4468ad6abbdd110276382138e2a095061afe52767c7ece8d0d9f46ab6d1c6e2b70cda93441219b3edf80ea66e662ddef4da8ff3f28d87ec184377d4e2a8c54fc96ec110d2bd960b480224166&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487303265Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default, Name: checkpoints/2496/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2496/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=20fa7e8345a9201b46f931096c5d49c22e41b79ff853b1c6d12e5206e382aa3a07478a90d3cd17b2adbab5050280790d5de159eb3241b5fbcd57dd80cf44038e30aaa66be82f7dbe13e8f4be7963d2621b585b197b7f21553939bdf49bd0b2e9b1f3ec17e8afe65e1c65e553c9b4466edc378b520098927033a2a8e5463b612161f94f267bf1d58787a372a1b4d79297c34a898157184e330c6eeb61da7e18dbc3602ca21560b4a2732713cb28220feb998a744812836d07cc07923e43db6be3ae5309f64d101baa462522cc4b901348c9730be70e13b2b4c24155bc864dadd050608d373f6403a11fcf9e9957facd2e46ee19a3685313d5b055a1ab0c43ff9f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487324741Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/array_metadatas, Name: checkpoints/4608/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4608/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9c1e0c755618630df49a5f7fad08ae06122e10ed53e952d1be3aa9e27434c84cb80f82cca6d75635f8c60a96b7d731d86a678671905e3ffd88598178ebdda5d3316b64042c3500d34e2eab0cd9c411d0fd181762b9cfc5e1d9003aa78c38413ee1d0c3810fb63760e0edab04caab90412cf478107c452898137bce30a9ed762afb7433f4a83f13b35fdc89b80d58f693b2c0dc5c9bd5fb22cb0e7f53e699b7cfd262ae26b81f36bcd2a72114d2aa154e98502dedee63fe5ea81212719026d303afa117d7588d0d4cef1176d32b514573448128a772d65fc68d1eb17c037eef4619284a565fa4c8a95354d512df67065552ceeb9ed156928d8e82132a80d10034&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487219341Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default/array_metadatas, Name: checkpoints/3392/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3392/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a638a0be14615a9490df87fd196af54bdd3703e1145592b17b60281503a80a8a91cdc3eba732d0652fdd95e89984306a2c63d3b4bc39629115936d131883c01a3d5a290caf179592c0090d96337871669c6886f7d6b71b60a6f034d48f45e91032cdbdd0009c38051111123bdf3e67b6a7e938f2fe185441c6023f5860975e2d0a5a2fd59e0aa5fe031a005a60a6f2d2b7681d5d9ca3042ab24a79f7d5a0aeb3ccbe315e03555ac2bc93f127bc5eb0d591f3cdbd7c8e8d20e0f01c904a8629d1df2b762e5b5b86e11645c1d23d876f3bc8d418d37c76928d05eacd7769f1391593a44f41df41d9d96adab0eab076e19eb452221320e08eeff70332e42580782c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487441868Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1472/default/ocdbt.process_0/d, Name: checkpoints/1472/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1472/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=750cece16eed143424bfe4195fc076fb1721b4811e53907bacf15b51e87761dc3dc2ab2b1bb0017898133ad9318414dcb5cb3029e234cbac3e44475a319603c70b1cc8d544a17ff1f38c7a5be478759d836101d819044bcaf7035d137c0408204ddff2cc4deb2657b1076c500c09244ffe42edc35ded3cf4385970176f1443ec259908025b23ce3c8ec3b5fb415afaa85e96b0b77d6e757812d0d16b13c725a2c41c87399962791b2d153fce78b8ae6439c373c6d4528deee3ff71052153e97e784729768a43d4dd276e57079376a8d8f1412021f20967b2f2e95811181f048538c00b743063d4b235f90f605656a3018bd6d348d0adcf528c935346b7d849fe&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48748667Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1728/default/ocdbt.process_0, Name: checkpoints/1728/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1728/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=160bdd6e52aae114643acc7bd11e251c16e42fc6ed5f4bf49f4c25bf7e2daa2c970bfce71d495be894c377ea658fece0fc30a65f5f3d48e296ae4150f732e1959ecb741384eb61d4fb8cb0d4f9415543a3d0d4bfc870b1f2eb0dc888a7a70b78813dd3bc05adb1b8d9ce97cdb476df8b8e88f8ceb7401a4dc893fe40d37f22dc1af1c7e99f2436b7cc53eaa42869f6438b9e07844d65ba2b4f564423690170ea53ab127d58650b4c0f2f9321852edb9ca4b01d4bd3d236199aef8f0633f319eb09ef5bc07db7181dcf451a914294f3a19cfd1af2bfbbd8784f9c6b5890e25a304a512252efaa871442aa588b5cb8e591733bfd118c43e24925aa2f49cfddd710&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487562828Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/ocdbt.process_0/d, Name: checkpoints/5696/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5696/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a3c305c0ff91706027277de23b452f5d52d00e6c226a75ad31fce73d4ebdf9db5135697dfec180b8b7dd9971d52e1fca8ff6804e901e870a7162795dec853f08e87b497d7ccc0b2a92925750c9254900843c0c967dd70cbcc810b1690f333e76798f9a8c5ebbdd64ae034821cb42e40277b9df2914c0ad938dd1902c1cacd0122dbfd5a0b898e3faf9cc05e297e502dd09ebc3a00504cb7942303b0a584a6b40fc7195e0051eda782fa650fa3980ae7bdd27b9d9f1b45ebd5f8e1a2cb9992eda2b0c78d408f932c3b5ae7937ad91195ff641d57c29574c508ecc06232351db5740bec09871e4c9793750a052268b24f7f1fa7b15029e91306a2a85e22da70538&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487634324Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/d, Name: checkpoints/640/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/640/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8896350142d5280074dac14f8f1d2ef62c7fcbfdd1819bb95d8e8ceebd05fd7536374aa597dada0100ff3d01e93ce002573eb64ae3499f0d51a205c3aede83ccff4ce404d834a189f3d877258f6d78b5e4bb7ce6e4c21ff7b4f60f43d7cb88f3ee7ed73e3992ea423a9e2cd64c5e9715b12436ab1369e4dd15bf98db98ad573ca86b74388e646fbab8e31397727e59abea571188f17dad3ca3211c8de941b938ca596233d82226b0daaf16b97b2f86f6e9c04431bea13fc54c2f5fe4349c6b1374d4ccf10b3ed229f721f39edd7320658c31db21b7c9ada6ea56d3e2d65617ddfa1a0a565a41391ab447e346272d98c3ac777e65e8cab63dab9bb937108a5c41&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487687792Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/array_metadatas, Name: checkpoints/1600/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1600/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=19f7c9f8d146ed99cce2f526cae684c6126e3c2d517cb5058dadcdbd2baf4bea80fee8e50ddff3905635c69b58643b5c9e0af3df749e83f5417182de22a675958d0e1df878018af44ed0651e3bc6ac0d54f9b17ac097f22a8ea8753ebe87f60cb81a6106a74cab1a2beb9b7156e3d532b9e405e4d79db5affa9c87afb5a470b79ec6f9f38eb6a0ede3e7f86ad29270de3fa320aceae1db41683b613dc741f9e6f1811b70616b47337585e7d913f2cbcf386ab9c2a1f0d0486b629d99fd6e32e72dedd282f8d8859a7ba485154afa40afa6455aae9642b05c7e1d0cc03b39d90e128bee115899dbadcc10665bcc82dad82741f97199e3c0a6f5018786c463680a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487820382Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default, Name: checkpoints/3072/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3072/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b5a7427b2426a10288e8b8db87c2c069e69e7b2dafbd559e0f90730d2910da9979b8c1f5bdfb3b78e8a40e334b631d2d5a8bac4d85c72b6d1d99d01f8204bd2df2629296f83eea2faa921a33dcd68ccf6286d4cc0d7c38699514e647437b9d0cb829585f2348ee91cf0b1905ea72851c4c82af053f083703765482e1400cf798d0eb2e8a84039f971aa6254d44bca5a2b368be932ec33fb563aa0d67171301bfeaeb6750f7b6b4ccd4de17af99ea43f154b6bb28a70292600f3cc36396422e6fdec0aae5d3aca6b22d45e55c5b74a598144f3b8e13339b4955ffd49697f07b519061e47e5e7a98e90d8a1cc343a378ba2486c094e057aa5c0cb79f4769451eb4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487724672Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496/default/ocdbt.process_0/d, Name: checkpoints/2496/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2496/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c51b8bce4c4f33145d37b242f330daaa2d747d1c6fbb59ecbd4b701f8a4a3d165400408c2ae45d9f54839c81fd527fbb0c9732447a0f653b80d5f9ab82e94ccac786697335e1a50189ae970b447fc96e627925e8ed9612d9d1c524d63cc38ec3c0a4997159da037c7e2ff49fee11a69be5ffe5fe2a03c590b5d8ba5ca7f3ca8099428a67f9f12bc192271ff3791dcf273ecb28985f210f94bdb7e2eea230b76bc4a8b10466dfeaa507723cd304b876333f2d8c0e01dbda87ec4d041b1944c811ace31b7241c103a432fdf60db2a3db9fdfc584f7901f2db76a28c928a25ffae3039cfb4fed0963838a97802164d4a7a856ecffbbe4535626961ada24c942cbba&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.487943304Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default, Name: checkpoints/3712/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3712/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=697efe4d7c16246f23e14ea878a5e7b98b972fb75d8ac3501980b97b3bf8dfd247457d940e28a54ffb8b8658f02b650e83c783c6c13993866839f481f13e7c5205d5c2e66582cabd45544963006f2428bad75c4e4746de9941010a545b087f6b4cb2c7b3826ca3b3e9b167c54948d4f84788a508b7c5938f984439938b8f30164e58937aa61539a98ac0abd745c5474fe15d6c1dab2c48259b856dd749d19f6d2e632a54552468331bc34a38584d3665bd4586ed59475442b600d22ffb9731107787cd1b972f5d81cea8e3684196ee4af380ffbc1fa5d1f4d3c37948608fee22a2c40fdf3e7a1a3bc7d4466854cbee5b731254a788fd3bfcebd8416d027f9755&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.488027667Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952, Name: checkpoints/5952, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5952?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6153ce726fda4471657c46d4e1df4af2c6ddcade60fd20b41f52ac5ca0528905e10f1cbd4c09a1c54b19fd0129cc5f2e5594a343ce9515ac64997ff259ec323d47600d619499ac3df0a27ddbd737553684812aca44644b9876a93ec532c4d3453a4ca5592835d2a47ec381d8401b0c6ce5560e0390c3092374218f6d7d2693f07ac8b1218b0a5b0896013ccae94f09a268e77df5f945eaff42a442770f382b3a3d8c62a72018769ae30d1ee1e2264b1753ad676d68ee8feb5027efa88ee5ffc4b5dc4fea96d8b3165e70f0048db063c12ce1bebeb12ff1de5448b8eb5bf26794b50abb41e6e673e1fd3ad7bce18f9902c8859f82c5f68beffbe0fdcfa2521551&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.488118968Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224, Name: checkpoints/4224, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4224?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5cef0666599d004719e11cce57b2066d8d9520b2d7f3343de09ae8d36b0a54af5f53b895dea8cf50689dbe48a91990f0abbc4a8b47d7fdd1abe1f0b36dc729ab20efba4ec116c6c4b3cea041ab1448acd218030a661e062e055f32357292f9aa7f0a88124d0da0bc3430e1bb1d6ea371908e51082c2aefb6a7d2d5e9e7755f4167c6273b66815e29f25af75703b733fa79a93d354ec3b16b3d35e4f9718c5fd1d72369f4389207e2d3a2d59c150e2eae554c862a2751ef5065ae51b65c9e8c1c2ee2090646b126014c467688a9fb219fb82aad58be958abb33c81c4b8e84169b8a3d7dc616591e6fff327877e7288dbabca5c18a1df4741a2421a6cc9a5de25d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.488211173Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3520/default/d, Name: checkpoints/3520/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3520/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0711d127e75443bebf7d43b72cd456930738cc73942df1deab50eb92b73176e55b54268bcb27d464a960f9efaccdfb95fbb9ce346d30235b231890e7c81c479e69827e89de58377c0e9a9b9f21ecc0dde1deea253e8ca0a6607759f7e89a8dccd805952656c0e3e8cfb4b381ffc20a33e968f995c104d0438739d96080d502728172a7064d33a23c90d3b8aa92ffbd72ef8fdc7dd7a4b180ce20c7d62a6ff5d422f4f1695e995715f6fc6fd83e38ca1685bf509d85e59f599237c59a8d7b49fe21c3a70c58418ee75e8ea1e6e9f75e431a7fac2062f62011ac865bdab9bce71a9074e3c2a55e7ec0c7158c9a82a097a6bee3e037dfd79c7322aa6627fd1bc49a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.488304457Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/ocdbt.process_0, Name: checkpoints/3328/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3328/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1973bc6b354e24182e4a43bc03864de9dcced6a8b4c05ec598297103b462b28a605a23249526efbf9b92311aec5be4266519d73777f9e39f90dfb242bbe386e800fb8648af96342d8b30b45bb6f3dec7f0915e3a2ee269e460d31a9b9424aaefdf5afd6469f44ca81c9745ef751b5a2972d89df6539b1b5a830a2adf9b42c62f2510b82f96e1e99e623a06cb7fd872ecf93af9f7804734a27f8d10c4028b0b8cf041f8d6784cf9d4d23b0d8cdf306253bb49f722806444f91eea5f2010669a07d51bbc9fde09d76a78e7a7a9c0b8decd8f2d096915cfd5af17380466cdc7282a2b56ed77956e3b61c386dc0d28f46ae5126d5dd354c3a05e7b02a013c194333e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.488382126Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0/d, Name: checkpoints/2432/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2432/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bbac2608dce3c4a9a227efa7afb1973a6137cedd5513f478d96181a843005aaa3de76fdb603d7f447da48831f6c4a7f33ee5cb6a8a405e4e8d9a790fefdabae908709339875708ea0811a309c606799f19d41e334b428171a563294acd66dc394650a1039832b6a89b043a3bdf979265fe45a9c16f3682275804e33803da1bee881349e82b078bcb14df362a4a1896e496040aa47a7ef1b48aa28c52f07549cb810b60f82508527a6bd43a749f93d3e7f56eee8a50d20f986ba6ee73a9d63054393f17b9dc00f10aaad8fb4ae995912d18beafb605874d07f70b15ee61b228d132ff9fd7bfeae3ec3ed9b354ec258aa96a91e06471a8c8a7267d0f890a37c4df&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.488501227Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/d, Name: checkpoints/2176/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2176/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a5eadba7c26fccbd11753b3f117a0f7eac3a5994cb3077645933887262ae039bd965b247ad08565e3d3bbaab8c621f0820c4f85d68e6a1d6216d289cfa784b61809d9735c830d484d101735801a8691d6397705477f4f6ac91f9d11b0a6c57905ce8c94fbf2f3d26145bd10280f455c8952acc1d3fffc025f321515eb38027f9f41c176bb922a71450ced3106dac45e2f800819142c26c349f5c3c2dd8d107d5fc0dc21856311a19f2fa580dfae8d125aa95f1488fe16502c88b8ac248cf633a9faf440b4e97d8fbd74be2c1a4a5389d02311545b564e021f9cca722a5530f4cfacaa1a894dac4cd255e291b4c24aa7aab6ad49b9f17b311b354b8b2d9150b0f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.488623882Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default, Name: checkpoints/2688/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2688/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0cb255014e1b4f8fc3b1edbc2ec2fc3f1053b1b9a08f5132a281c2945a8e3de18d71d55b0349941c4cfa2aed655fa650be539e452380ceb7299efcb0537911746563bd7835d898d34227d2f37052698504d1b1ff970aabe70a3316dd0ea89121e5dcb130fb89c65a096de8bed6bf909182b2f4ff7e9c35d611eb3715701c579ea8e598936cb8bf7e3492404c9c12e8bded827bfb9c105ae2380dd899bfbb30efab0070831f370477a302bbdbc142ddf1e61313f14038d63f8119f4eacfcc1b3d42fd29d6f981f0ca65c6b87bf598fb3528baab8c6b47963d9e2785b9019bbb91b648a4d6427214647085d8ef7544a7aaa40be9272bb7515cd80f2a70ce7825f9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.488677567Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3264/default/ocdbt.process_0/d, Name: checkpoints/3264/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3264/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a12dc44afd7cabedcc557c493c93430bf4fbaa2bb16739e828cf0f3f35e988887e438273447e129b929d4b71fa8a1ed2830c3f65fdef4c3f0f7696c2f3d03159bdde2b73032d11fa27681b572ba703598a94903be7b63845c226d05dab6fb273805b180fbfe777cd3f60ee7e9e8e4dbba2293ab3fbd4e2cd2c82a9c35c4889dde7548ace77b3502c3ff0fddab97d8174200b6a3065bba06644bead1a1b4f790c739f95f7f5c9aeb0f276d747b21ccce1f0e933bbb0749fa43dd3ba4bad46556fdd9aded3d051617e4cfcfcbc546828008ac7ef48ce5edca46264efa050516415045f1019cf9872f6f6a7aee772701f7ae1c8b260bffb43668f88922d7511e1b6&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.486559256Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056, Name: checkpoints/5056, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5056?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1a1f026adc5c07229b26344b7b51e40c389e6abb1755aa42d8f5a894b588b7a2ca1d6b3d97056af4243f5ce665ca67a7f304b108f50a0fcc5a2baf91e76c3242a6aed18b41702659c4e1a7e49982ebf0c12ce0836ce3f348ab83f4e76ab9e4ffad60e729f445a1a45564eb8a384074561eecefcd66ed33718ebfde4c6ccb8f938cbcccb0c1482d1b4143964e5b24cd9368886e1f7e1727fca89a8228c7864a878d8df1e39c1918258636416938b8a0f5bda0fbae3fc19962a724c9c97fb69424430272135f47e7d87e5b88fdf2cf3db94baa3054c28b865c9d400092bfaed2edf00ff3d6b3c8eb207d7e3fb965e7a57a0edb8b81eb70e7af4087f61568c3145c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.515895756Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/array_metadatas, Name: checkpoints/2944/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2944/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=290423198a3747254c03d86ea6f8b89a724bc6582b04bd9f46725a340c44c3ab65e25eaeb7ca6ee54cd73eaac988dccd7c8c3de6eda40e46b7ad0de623d13ca7460aaff514bfc93437cd76ca126f0379f24ce77e87ec5fd1e1be20d46a409c0916cf4ffe7e39509f4b8a5c9c2e8d58e28977f314fa6188ff67b51862019fda6b898b958a8e57e08dac5cc78f1f774408744b6b191cbb34265d834edc1b39d289428e58af759f30fecd79422f0595df06653f5cea53cfce53b76406e3dfa06f0f8d3a5d98342300cfd3589599386890926ccc24351833ab90ff5ff8131fd834422194abeb7936f0742b2a5035a54998f69a32c59e6ccfea2c3eb8047af74bee3f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.515924621Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888, Name: checkpoints/5888, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5888?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=96c1ea32e4bfc3b274bbfd6f5573fb41a1460167948e7094589594f71188d2191232e583daaebdc686a6cc37a9046283315cd5d2e96576fd8a6e7c51fa634b610a96c79963eaad67e62c1e3bd973e3412a137aa37e71924a37eda8b5259beb55d43cd6c7c9df1b34287383064adcd6f0658c440878627cee856b55aba6c61c2fc576cae7fcb4c93f16815dfc8c65871ee671325c21b8bb6eae4d4dac18b2a3172b864ba5abf78cd5448a997a8c0a7cb1ec2575d16fb3fc58abf4e429737b3b7e07dd0e41316b4ae7915db4559cd96010f3a94ef2270ae00e676ba7244a2992eb5f34c0f341f0ddcccc6538c56024d33786c19fa54db801316a41adb2a718926f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.515984406Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/d, Name: checkpoints/5120/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5120/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2f83cffd470256fab0ae61da8e1ffa685554a28443b35fad0222f9cbc2532a9c2a19799c005445a17c472e2369298952efb25aac36d458a06d64c4a079ef6f5ffac7a3e09de82b3b903b171e17cc3b2da468ef17e247c8b5a912fa18134b263de9a7f2d2088a4819c2856cd1023412f28415d216e7feb946fc7acfb96cea9c76ff233e89445160bd9b7d0cf9580f8cba0165f1ae5cb30bb5de1002e4848e72901ccafc3917918a74cfa047296ae8e49ee3ca3ab5a944050002cf2099f8c09cdd05f7732288c141e408d233eab9eb65aae50c3a3dc58fc03039eb93fbe0841f8417c01f9281d5b681e2c228c78fa56c1ee63a456fe81f18ceeb40c7f9b9917e0b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516061577Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0, Name: checkpoints/640/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/640/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=64f0fdda50558a704c56f62f5108e18c729e96a5a2438a54ec04f5113355010d1f049baa0c10a36e3af4da9bb651f7a0e9b299a4d3db11722a5a9820b995ebde23b1e32dc1c80311b5fbfedad91a53d3e3d5f03fac1ce256ef2cc2e7cb22ef9d0ecedfc3bab1f8c02e159f209ad479eb1d44a4f4c706bc8d02301f63d79bfd90f4329ee7b332180854c9bb04ec25d03995246c98522c2072f369431f5c4dfaadd5e93c7447701b7e8e5b231bb2efcb0c53670d255fb24a160207c2b1ba2715e793e469c1e552c0ea38aeb724009a1b497813c78b95c1d4fd945006523f46045eb7dff5f42ff3124cf5a2a9b24b67151423acc5ed8a472202c00e867069930e99&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516166979Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4928, Name: checkpoints/4928, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4928?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a72d3ac941153a451819f3ab37d45160f19c40e51af50e348075fed56c9ed983e93ac3f9f4f150dfff3db633404a4d681183cb30647af0f089fb84d8947e16c46b4055ea64539cc23de8e537fad643d75902ac7811049a2374328a39e3bf2bc6eb0e5b4b7b9f31ae94c5d32bfd6a451baad1eeb2a8e58c04cecf89bd2810a35e3bd942c103cd43722a1d75ff72d9a4086d2e5f49f15acd94277563ef6cb3bcc8246eb7ed8c21e954117e56fcc7a46b413f636b00343797869e985982be819572cda3c6dc7144bc811533ff5c21eca3ca73e00dcc8aa4062db3edf36c9b8c070e03fe9c459dcecb1934acb437716bc6b6dd473872ccfc04c0bb6ef94a0f4ecff2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516276299Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/ocdbt.process_0/d, Name: checkpoints/4416/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4416/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=15467994ed3b236bdcffdf942525c44d0f4ca0b40a54aa37eef25f5cf716660c5c9595b4e16d282f761ed3c141a3d17b33e2aee72fbf9bb543eb109182331080a9dc82167e4d1eae984383faf59ad4c1cc1ecea53f8326dcd14856fbcdccb8289beaf515eafcd9a65277392b2ccedf90917273c01987c7b62c3033303f3990335fdb5e7fba2cbf9ce2ee531c9714e5b53fff7a05eec2a4cf3288b3035668814d2c74373d7cd979a89bc833aeff00f403692d18829fcd80042821cb7ed923a309b6b1b2b133b7f3cd383345ac8674ad5dc44b797ac447805868ae329af3d2f1ff9eb68321086863e38975a0a6ac6d7cdb72cb2b4423f6d5cca31aed463e9756f3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51649456Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default, Name: checkpoints/5184/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5184/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=185fc34c34a451f46aaa231b36674c7edb064f2f88f6b6e7f96d845c4f627c25859d9a4baecc1a2e6d5dc4f2adfd25c5bb77657d6d65c40f2afef4aae96b891e6aa0b623b5f615008a7f09cb02fc81a1e28fca674da8adf2e5786f990b0bcefe9728c53f3bcf1a9bd7924928391be00f9788188757cf95687027ee594ec25c382a6d3fbfcf3c3128f64f8a641b42991edc83b1e04dc49094b709340ffa07f5a5bf3372716418f4b856ae518cd38631dfced1ec49d43ceeb0b8bb71e65e2f745e61319c2230039a017346974eaa7e6a657815c86f8944ce8a6bae8d4a45f68dca3596164ef97895a1e001b09d3e917c343e59082ec185dd7d9cf1507df0cf4994&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516511897Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/64/default/array_metadatas, Name: checkpoints/64/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/64/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3f2f5a4ef4ebdba4986ad6e8b4e127d670c6fad0c1df174dae61e4f16fdb76993fe1f408dada4cc8cd767567d2499a54f03310be5ac31fadb41027eeaa76def269ccefd7558df985933fa0b129dcd89bbd11a560385fa2ca0f53ab87037f46ac202662589a3a9e1a2cf3070eb05cbede29da6725667d1398d57caa9804e67c79e0c6390c54e80e7c678fdc15f8a2715b2b639a042364be6fc377fabbd14be7eb636e109a1fedf75f6016863a629851ee8957279c7a9a6487130ca84863b7e9ac67c1050d89f1b38aec8b08687422046e17da400e5dd98bf9bc3b30c1f37326a5514f05376bcc2d6c0ba26c6258a3fe9af9b845fa44478b46fbe2784e703eea47&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516550498Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5952/default, Name: checkpoints/5952/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5952/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=41f0c667d71cc8723986d34c8116a593ef19a3842b27c135259ae845a3f04110cb7e2a2bfe485542a3dc8015fae4bc6c4cd7fc4f50f4b8ffa105b62f5940d987623e1887d4c374efb7c22815442bea51fb5d732a820f2b029f8001952890f8bb0f194d64a7a63fee7b85bcf92953b9c6902d25ad10746657de34909da3dd4f0b613fcd61131569ee4d260e8d9a13bfaa25cfe24502d85e936b5a4c4829b1f08d81e49adc28f6c15a5bda1629f5fd6b4ec956146347a57ae055d4663ce028ae36bba37e175db7b7891184d4c27252935d0c71dc36eb2f340d217a86d1387cfc53eb73dc89deb3be982df0c84099236074cba6d0e6218b4fd0c674bce4a323e1aa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.483083138Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3840/default/d, Name: checkpoints/3840/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3840/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=450c4e9c33e97a42aefe8775c8395cae7d81e00eb534ec2cffaae0a9debc6ebe06175b3afead29df40c967035ccae40febc0bb16893069b82b83f8d7cbb0fdab791dcd1e2416b85f5fcf30525661207e545410e602c7c98a33ecf6ab030986f8b06d903e8527bc82e6627fe80846ad7930d4a48f985a0ac01e9a86d40c13cebaf26527132b665788adc5238397dfb5f03ceaf5e6ee6b8c0bc601070d7b2ed0e1132a7ca0857f1301bf540a4a262253d51947326ffe1f63f19ad6bc2d713524ce8df15a776022ff7b870a47d25ac2376e1de8e560ef9b3556fa1a37fc458852d6a2e0395c77f01e15ff7c3310ff8e3a2bbfecd85f21d04cd7fb9c5e4cebb22c40&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516554349Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default, Name: checkpoints/2112/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2112/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=524bfc7930f8b525daf76e18406261b4c4195e703ec81961b368cc45c678e56be2b7c6061908d295b508af5b046f30e0e70d479bc4fe321d015a0889607fd6ce17b6f0f482a021de5582d7f0631848bda9033b288d3e735790a81ba4776ebeda979396e631f68401cfb548ebb542ab809d25d76551080b2a38632ffc95ad3a79322122c7bb75edd513972b73ef757b32f33ec13950ded8e4c4adfc4560e716d844d8266b1db31dea2afd80d16c703f0069a6d47fa9bf1acfd7133f05cf92ca204219dc69fe54d71e5973bc3d39b0106657901bc05b759e135264e06116b14c07f32017862f652f4ee92c081e3a9453c404fa5ebc78e20c2a9d4172dea700433b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516550277Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/ocdbt.process_0/d, Name: checkpoints/2816/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2816/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2a68cae55990f7a65dd63c0594e41e796e91482989b9c3187f4b0addc97894b11302401f36f4eaf717bdd52b06ef8ba15d94017336d69180dc1cc183ab4f581ba178527cb55553e65d482ba5eff3b8471d0905d9b83f0a774ddabf8f494821297affc77cf0369da6df6d6c90795dc569c95355f349ac30ef3c7180dcc7462d3e29ca1a0da3469e394a3f8bdf36f20140e4070f43bf3b91d3ba75b8ae695b71acf2515ca692e76f4ceca7abc6e8163b244170514cd71f62b6ecf64ba3940cd46039f359e117b0b4319fd413fc6eb30678e627e6769d427b180910417335afb1d6232bc943ebead8c4a0ed26d264251cff6d7b6bf834eca9ce7c5fd70e68dd16e2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516580545Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5888/default, Name: checkpoints/5888/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5888/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=704d9d10e7ee65f067b7234328214ebde1ce4426dedff57543b236b7da3c4835a3809b146cef6fcd6548668ddd84bee50a2049b16e35168164ce2168476978a6c8b14a04a504cc74060682d1c9582677e0e3507a2902983bf4da7b8290a440bad48ee3300be4a8abd44adee7f9656c31ce24daf116e66a08b35a7f1a9850e4d2315ddd3fbbcb65ccc36044c86170accb987aa63c8c24481d287a2873c474dbd7cd02f03c53857395999e9e615e2fac33f33fc3d3581618ee34842d7b698d08011175a0d73e42d12f63cd1808817159ec390f7e6407eb86e2584168c3b242ec9e07efdbdd7fd8e1e83b6274453dd20d17687adf7994c3832a018c174e5915ea13&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516616925Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0, Name: checkpoints/4096/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4096/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=08dd0eae94603a0cacd4d22912d0b3c7c1cb052d7d9ab5d4e9e4361d64a49c102698bbba446a9d31f198bd75bdc7b4d77803eeb5f7f7d1f19a0c74e67155529f41c2e0a277ae343ce27bd629562c59b692ee8a6b72c9fd0ae7298ef20d7ac4daae8d2fc06291dc3fe10973d15ae649bbbd88a6f0b3b83d6333144ec58d95c33002cdd7130f7615f30ad80302da620eaaf8889b2511dcd837a0811915cf856f5f104eac63f8d036941cbb8ce80ba42f0aa03f68b5065f37ebced24f8c0901e7a86dca455a229cb2d7405a6c91096bdf400a4bd37884f46b7cf2a550dbe17213fbb122ef1115537721c894ab07a804d408bc57791306870a1522d10ea7024d3adf&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516573064Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/704/default/ocdbt.process_0, Name: checkpoints/704/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/704/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0b55c6b5354b071a622a4b810a0e70b2c3e92d362cac3d1ee42326dce42d49caee632884eb65190e70e13f082539ea401669b1a24d3e223a4ed22bcd14a1d1f65402b15e8d72999b983e521363eacbdbfc3d2a042773f858fb4da63642194e2b3c267ad44245daa2b634b8a20c47c8e33aee339ba9810c02dee834cd11a54bdc8d3f3d3c9b428c103ce3d66784f27cb49e7aaa4bf16ccffb938196be29019abafa15c649b4ce4ee3f02072861a6b1bd11048f131557f05bbb14234764c424cfa69f80c3c9701412bbdcfa112e89e555fdcbad156e41282d1d7a87ff2f54feec8161663064b78643e08aac59f6af1bfec0df50bd28e962b53fdb673cb12d18cd2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516616052Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/ocdbt.process_0/d, Name: checkpoints/5184/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5184/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=113fb9fcba8e12e0180dca861b284186f95e3f5a14971f83940edc4aac8d885c7ceb061faef53371c845506f37161c2d2cb7999e7e58f2128546c54629346fb39757ff204892706e6e213e612afd9dc25ebbb9fb0f69d928dcfaeffb419f38ea988163a36a70d05cac82c48670856331de6ec6331173e49b3e42b6f9d658ada43d875b9e86a03a1d91506ab17b3c8272307b268f4c1d1c0e8832fea04dae18eb049bd31e7b3714aaada1c96317f6fd657a6d3fbcca22912f7c0ad45e6dec0e1b5d93a91d309700331bdf18468e02f0a0d223c55aa9d0e0d5f7dd9e0b61c1ce08a9c778faf307fba00fd285fb1d83fafb2a36b5692698b6cb7489cd934ad197a1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516612567Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/896/default/d, Name: checkpoints/896/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/896/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0418561c774508f2809b5883a92f6b19e81ea6e6ea8021fdd82e6bf7d3762a9a2e0b249c25fa2c50a042c5b1b115079af12a6ceec0722ad036b5e4ab6aceca7e492429dfe2d3aed2e4ba79e63890b5fcf97a346bd56170125d7657efc2b2a28f3eea50dbed9ba242d5d82c2a9a79f2766b97ff6ff61110ccd87c2e017d612c9315f48337e18888d72a7fab6b06e50d018d5ebea38bb6db311296e9868da736ba8945c759c758c89593e5619f3d3a8bf94749e4ebe21e44033e7c416f48fe21aa7295320b169ae2ab70b11b6743e9e340f13d815d46bb04250fabba4c782d73566221f63c7ff07103ac789a47176b6ae20693f7401892694f4c659ac77e880d33&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51664635Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2368/default/ocdbt.process_0, Name: checkpoints/2368/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2368/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=00868dc8e3e3646b133d077f8a45cf32e433e1938551f98f7406412b511df6bbf14fb19e3afadf5e314fcff42327f5f0d0fbec95e38e279ad884c20b032a4436fce41c8fb03feec9f60f9dd96fc9458188c877ac51cf073a89ea4f1ecfd251277c4895080768ac22f6783e10947da499217a3b9ce4acee2582e0e2985bd94161a3f06d798795ed6b7428f196b1de8859eb943d8f0f09eafbca2d5bc7785b2b978da770a82630f68f5d0e02dc0090e376fcef1cc193913b78dffc9be3e3efcab8bdc94f3b2414aee88c310db2cd44ca7d623cb5e90c9d3244f12cf5385e13111c54e4fc54d7191756eb8b827ecd94eaeba2f189fcc8eb647d343d3d91ae4b999e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51653014Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536/default/ocdbt.process_0/d, Name: checkpoints/1536/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1536/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c485455ae6cca8f37d5c6c4c4913ddeda3765fda0433e057763e901603d3f59e96814a33b98589f78d59d3407461fa7c0d8f01497a3d2c080ec695668b1daa7a20e2bb092671fa911aa9726538635ab42e59f88418ab1d5ea6555c6c0fbe03166b1197c303b081b1246147cd876359d37a96601764aaa62086f1bf84a3d3a17a6d489ef49bb2b3a1b80fff297c00fe09e0ec78a7efcb30e5c73d469db238267a268199bd827f557033b0541b1db2699590a85f61d3e4a4f0b9eb191f727e2fda5efc3ec5b9f25174792864e20d1ef3d6c876d841590caa025b68a0ea9a380ad1571ffe5638ca5575f0d3161e1a1d080b7f1c205c715238733460d3a4ed0bef81&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516622043Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2560, Name: checkpoints/2560, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2560?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7c6a168eee8bf5e953f32d715c5722fb60ae9e5f12218697724b21d401dd3f260c4ccdcc3c34e3b8ea99dd86a73e828d964d4012b12cc0e12b7894d8ac3b25bf34dbadd632ea56345bd641b37ea3a5bf62f655c711fa7eef42c0e152457e433f490d598eb68d0105b9ebc51657ef3a2f2e85473b4fb0c56d2d70db96ecf51f3d17391d299ea75a54f0d2a9c6f1351c9a3192185be7dd7d457e70e582014b7f0569950da7c43ebcfd046b53f0e966efb9f9e2a5f1342fcef3af3bb5b9c3b4cdeb61c58ce4ecabd0241741f88b124a0d5fdfabd40a8c766fae91160837e07077594b20f08fbe5e0707460401d81baa8380d8cb2224ea931254caf7f5f299b30b4b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516642481Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4672/default/ocdbt.process_0, Name: checkpoints/4672/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4672/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9c5cd4e18b5ea508ccbaa01a573a8dcac013f96cc1555f47847c7e3c58bb9ddda054741c872487c55435da37baf8c2322f470d7e34f7d89c92e22e708ede63a2595f892eedf995293d6f2159d16f61a4d6163503b594952db7ebc01b57ebf73d4bb0d513bb0685f91345d8166ae71c339584f4a547d004007bd1d2352cc9f131d679f9911ae738050087d7a6ace815fd1328810ad901e2d37b452be14e6adc53ea5e670c55e31618b162bade769f82b16652a867e5c6b768ec1c6b13295075edac5ddd66f387cbb1aa6fb9d1b1fc2ccd00e2817f018027c7ab45917f3cc000f8c630faf6948912fc172dd6a01dec1480cdb536489790e0994080ad891cca5922&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516658207Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3392/default, Name: checkpoints/3392/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3392/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4f4e35b33dd8dfa6112ad7cce1649aa0dd9f6ab841fa1d7c04cddd993e1b15a3aa37d490cf4ed21921e6e319eb8702f7e43ee82c5a32a6b7313d78b690111010c8212022aecf06b679fed77f7d85cee0fd7375f55d571e3564a4982a378cdef0036d950bffe63571d3e7ce2ac9a94da877ad8981480ca056389679efa3ef1a577a6818332a9676187aa8d80b2c73aa9d016d3e24a90efd86a3fbbc4e2efc855292ddec6edc8357da35e9a506fd46e1b0a449c03c2fdd0e8fa6a4696b1372f0ca540696085c63070ef6348efcb3e28f40e1f1e27255d7d0e294eb624f8512ec2b88d738ae69bec67b7b30fa0cdf70d155cf0859866889c41b9f6a71123444c0a2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516693355Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/array_metadatas, Name: checkpoints/4416/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4416/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bc5603d3288c77411a50cf356c797f6ffeeac4b26270f0c701e3e20aab097cf8269af0b8f7c71f2d379df8d2e252bae6c6f2c1c5e7677190b01d50442928687400757d3aa21e19fb4f5da3f1c21f3ce82482b2728c3c049ce169b9d982d3e81286adc3b238fbcb4d1b7776796660928c344ec330ae8f3e1c5bb7eac66a696cf1c5ee2f574725a2d119aceb91b2362e5d4b7d9c0038648d8bed25f3d845cee2e1d03daa76a6633d1664f6eb2e9983f140d44106296c5bc529eede0db6c7bf433d140d98256cc37b030a3237a7c8d522f986b39c86e8e85a0f43ed9dd53fe19f4c553879e0245762826c0c6baf45758a5696d5758d0906762580a6881dd756878f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516764546Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312, Name: checkpoints/5312, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5312?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c2f1899eeeef52582cb50858de7f169c20766d612290347bbcac703afcb14765a1d5f448fc26c9dc4d3456c10635e6ac1bd736ac6350d43a4d1a69de61d916eae06015b77f550084f36dc4de27ec1871db83f256cefcc9561b9254d9771b12b7daadbb3be0df23c7d2335106114501b359ecc4e036e95def3e90bafec17b3f521112434735e70d85ba9e85d0d616f8ad3170136e4a2f50d0cac09cccb3441fef480f0234a4bb136e8970b1634bef00ef636030007e30a72522c6dd5f416c6dfce96ca83f8f7a70d287b0af42b430b2b1b511ea84713ea7405f52084dd53ab6783b59a3c21da196e4a0bf19d799e6ff2e488903045b1c9c6dd0691b13023dea90&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516949911Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/d, Name: checkpoints/1408/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1408/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2fbbb024c2266313f36c3509d684a8607981be64d6cc15f8e813f449126f0f43ab611e5c244940a3f2e93976c5137d664a584e31219afc30a3edec957f233eb994216f3f72d7644cbb9b4ee162ee95ac3cba433d71ee7c65cdcfa637b4a802917b1fa4883279569fbe2e3a954b6ce4372d2a4231e1f0c2bbb69fa7dc26fd5725fd5d0113a8875dc32e27065246a4e6a981b18154e089d42a65dcc4948794efae2b70f71ef90ad2fd0b5ffb77b73e12b2b172b73b5589adcb08bbdf10466789656d2fd59a8152ba65fff1f026b0b714794c0d3cac6973036e42e4b09655e95e7efc5fb244f8c680556b1bb43823ac222f4c67a870f6d83b02d72e6da4d11aad02&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517139628Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/d, Name: checkpoints/2752/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2752/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=381785b55cd90182dc4c7457bcdf3d6ec815d574fd97b633b556ae4f385846c9e5cac0433ffb7efc1fddb493ff8d63169d95d4dcb81aeff1ae081863cd605c763b4c439f52380fe2419377039d96cdf8759975ce1f78aa83c04c661dbdbf9f70ecbfcd67feee80d2232286ca9e13c2370dac54491b14851feb83b58a2c400a36f984c5d395f7c580ffe519df45274c9bfbc08ebf075f9ac6fb5f6684278ad57105032a631d95092a98fdd3135c7b68d806b5db019e4cf07574a65fa53a9ed5e62386ba92f078e6da524ed1dc561b94662213772dfcced8c0d8a8d9a3dd4271f5cfe036226d2c9448ddaedfd6ea096047b967ef00d25431d90ea73239bb10611c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517158081Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/ocdbt.process_0/d, Name: checkpoints/192/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/192/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=86c37dcd2e3b4bf8e4bf4679b56814c0f0a41a077bf37c4e6edcc20c772f2eaa5b576465b68bf30f800c1e18327f9e143d6453748623516584b81fa1eda6454616e4ca4886d958b930443aef4f2ff6d45a36c27af96490cfd70a2d6e698e63eaef732864170061137898db7c704074d421e7f0ee85df1ffb0221055c731fae6fd73c9bd065fe54b2949f7450e83fb9e49d9e44f0c995b926c467df76fc83b0ada425a552904fa9651b1043cb80c28c7172366955426139dc4421d6cf0d3a7aceb5b631a3902de34c0bb28820bbea94aa8eaa854d70b5a9d93f420e994ef0f917e1b9d81e6eab4d63dfadb84758b5314533038ef0cb9d4acc7742d955aba8c2f3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.516958032Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3776/default/d, Name: checkpoints/3776/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3776/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=392c5b2ca11f40b67b40b4f475bdc3c63ecc56996a8d1f1127cad01c31b8608cf5b85ebcb48e0de986b05da54121a59ef7a33055ab61558aabe51081f41b31b91fe92720f5b08240913017715083dc7fbf81951325a52fcfcc69720e89b52b907bd6393fd35a287020013ba287e1cb8f269e8da91c829d84c6646a726549331511a65e041ccd1555057a13f7555b4adb973cdeb3b0ade12bf52d0c4a643148fe8e659b9240e77f5fff1b5ba40b4c074b5748ff06633136876a20c2fbbf071777cc7282ba46f81f361181b29493ed7b537a445b754c10636242b4879ef95aeb32b5e64c9f67db8fd1badd6b84a9ee312b83ff751fca722915e3319d6d9c28ffae&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517196622Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5376/default/array_metadatas, Name: checkpoints/5376/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5376/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c1c5f0f7caf61e3f084196dcde204588aeeea54c7c74e40d6f3c67d11e240ff83ced4b68ef9ad262c07316311afc37a3c8e18254854ae4f202783ef823cf92ce3927c00773bb6c68578ba1a087177c12b4ae144a2044263659c593bec811391071c93ba757d57c80b10e243098d13254d85ea3a5bdfb050aebb50022c577ae4ec8fb2ab24e747420180d1df3951f3ae10b27b17eb91419260c98baf3632087fdfc98a2edcd4fb69382a8dda882fa22cc71204d81984302a28bee59df23629018a2e0bf066de714186af12e8c112e61c741a062788c812cef42ec41a83acf622199e1129dfd389a30d1cefdddf5886ae863d35e58cfa01cddd5f016632bf84bd0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517238194Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944, Name: checkpoints/2944, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2944?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=be5e95145e30e51d1149f35f2706dd7a8fd33f9be179d97576e9b41f2e408896a621e3745827cccf709187d54bbd12216a9f4553a740c58c85f7f6d7665e68934c2824f017d023f15d96339de4525f7886c574546085ea8795559c4180e5103af95fa9ec0b974f0498e31f2d391d0592516b8e6da29a5d45c9f83792c7d783b486017151f0cb7b4c915724b076286095980b076054b3adfdb772ce88adf535bb57c579ca6f0ca5910073e43d576a00671323716b88d807977e4343d2725fb389886c9d5c69a412f4503c9226ea72d43fbf2735e2f1bc944bc6d4464288d884defbe504716b5f0a98848c13644388356609d1d704f5f31f6109404822c28c521a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517377187Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920/default/ocdbt.process_0/d, Name: checkpoints/1920/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1920/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=61a69c55a7fb772a7d5e88266553e3870479033110fdc67abe537b430b4c21333c6c2fc69372f18574e468169ef140c8f5038a005c9f5f4dff187b2687cb6c048a830997661080c7678051672586977550668d1e65adb7aba546ef76a5b0f736bb6f27f6956f16ca10e70c650d4c85c12c829a59664cf00f0b137f0575d59bc42fa2628f6a6c9336ab57c1265670295e8b1480dcf79ed6768a257dfc2b58df63e4a5526210f020066d54b36886f015a4da7fdcd7dbac60b1130e3094baad6dae09de4eecc5a3a4744e238b77c3b3743e72a16dea4852a3c059b7e1b84aa988f7e72f0801247a13b95ef21b22c963dee9e255cb2867bf3b99901b25d5bf91a44e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517407975Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default/array_metadatas, Name: checkpoints/1216/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1216/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1c955be9801daa043b5dff0c289ae2835711649c43aacbb6b11a2ff6761d6d54744ee5424177c5ce0f5f0e068a88e74ebd4cf6eba3cac1182092179b6c0d500299e42444ad2b89264acee0e337da156c0a71eb0db2f64cfd24522cee51f3dfc5f09c61faf5ff3b0e3babd76aac8087299cf3a656444b2424e76121b9548a92fe610d2f0c3e2788889b5c3e87c2f618dbf9c2eb661b459d91636163397b4f3efed7e10a1c9273ff94f7c9ad33b687cc58a341b257f21d2259b9aafe1d02526101f3db81e6484372c4301de4d24ec2a89834a6dfb44f7d69563fc1a70c8eda5c4c6329fc132ef828e8ba85749ecae947911b1a185b9d21a58fc2feca1db9dab6b7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517409194Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760, Name: checkpoints/5760, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5760?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=843215770273c9b216e7bf2efc036d98a4d585a27a0010312016e88b777bb0ba8e99d53e4dcc5a77793a6a5159d291adcc8598672d37bdefcb744582d36954b8fce2296db4e448885885e35348c11b42708835c2e91fb450f36e5dac78aef7e25a4c97410f3e90ff3de9cad123d9e6c3999556af856e1fa2f4407b70db233c14ba0ddccf89f3c4f04152744d1c6a5b7bc77635449f0f8f0f1cbe6540c0ac31a59e6d601939dc5f07d00a0958aaf0d0f0001762319f494bbb31319819a7580877925a0b92094f1b18a5c9865307c587c7e358d8ff534e4d898ca8fcc3e0354c9c776651a40d05e3071b65aafd764a28c4cfa3ec36ac2d39b7e645db505827972a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517546308Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6080/default/ocdbt.process_0, Name: checkpoints/6080/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6080/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5985a282b87bc531237a5c3500de4fe5517a111ab145959aec2c530b277780b4619094034725e2803c8f27dc3ef0d9e9a81ac2b730b229bffbbc97f7f8d42662d6f2cb16a571387850316c9a7b051ec8bab2dd25541a1a0a3a56f99b3339e7dca704a02b41cd514b4336e0d1f514740fbc89a3e45b32e4918e3b6b0196c47bd3e1c68461810c54859465305867bac40122ae46f6dcaa99e9933f8602464c80a788bef3ca6f211e4e3413e43b6e41b04d1f528158d97f119a94181c6729ed076fc7648f4bfcf7d763872ff8f725f1f815eb931c506a224380d0e07daaf387403ae71412872efea2f026bb3e8ea4b3ab43ffed66da2681cd571d0bf8a39e021f30&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517571035Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664, Name: checkpoints/1664, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1664?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2b964d8c58733f2cf22a08fbf7086eaf2ff49ad2b90cff378eb45cae541c0149c0abfc540f7ce3478d979356c6c505062d48f198774a8e08694d51f31a31e6499f7cc48a7b87c52edcad4c53ea490710f976b0132f72adffe9d5805454087bdefbc791ce57f1a2066a5a04c629a9242366f40d2febb1a20a49e4be75f68c50146e26858fe214fed06ee453977bd8ce18782e721040c8bed04b89383d1c3203a149303dd9cd16d1218668feeab2301f3d656c46dff09d3041ed46d05a5bcffd3c033614d423fb6ed3173aefcc92658e6070c6c3fea34c296b060efa91a15694ada69ed504dc16a351ad321ab21f0b6abb3c0cf2a286db230b3fddbb0f75873f6d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517600799Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112/default/d, Name: checkpoints/2112/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2112/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b01aac19e0993672426350e8e124bff9a451b2e7760787e66481daabe71bbcf8b0d80c1eea4ce6df274b971c37bbd503303fac5921c7b07c0845b007c537fca6de11a80000393415e89fbb33964bf16e4d074f3d20ca0ec88c8885530d348424402055bce6b63b948e1fa132f03766eb279e9729c0bd28266c460b16d104b9c1289ae6c82ef9a2d1eccec9ea4ee57f166d6d0faf8f47d504a0a2d2fc8fcf893c0267c45fe2dc9f36c53312a09b41092a4d416553d76ea060f95ab65cd3e91ae74e53d32451d46451dce33f9ce3ec286a26593d12afe3dc7b0f891e45bac254f5b2d4cde5291e42a0e6c6db7ea89c803a7d8500c59e5f5fe36dcf8feb1fedfc36&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517641785Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2176/default/ocdbt.process_0, Name: checkpoints/2176/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2176/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=938a976706ecb6cbf96de4c4d3fefc980d3093d292e6d0d1f8eb001401dd96fa669d2df72b0beeae7940001c362414782f3def9cb2e604c088c831f053649b1b4693e9f114cb6f265a6c538349c4d27602cf6dd89e54029353859b6dd104f4138bd5c68835316b4e7ed19acd1e230d89c5962d76132ea8f832b6f54b30672d6b1e337e03e51e820c4bf238dc488e98c2e36d3458c33a31179d6516906412c8665c54a33d5b1c89cef56dfccd359b77e8768d9c7edc191decd73a5c35e1f57fb9c2171fe8ec6f044009399947952b8606df41ca767155a3bc431f8d7f0b2372f38c64d54c7eafbc738a6b6f4c77bf37ec98132f30468e760a26b31f15636ae460&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517709905Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/ocdbt.process_0/d, Name: checkpoints/1856/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1856/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b491c6e2ca62bcfab4230de8dd3d2c44a24f8084145b4075a1bc3160b044daf40a2791a9c6f996749a7bf177ee3e7dbbdb9df8f833e7c71a7fc22b029637b179859063bf7e44cbde3e6da4c48dc68ce1b5d62392e8850ac3b3e6ed85e95862beaebecc4cbd2921bd925d70d86f8561a3e79ba6a596eba8bc6ee8efc3a7d08db50240139d028cae7ed72aa9814a26488d212702b47155fc2af7dee55020bc7ceaa4edd86b11ea0825d3f5a895211e207f390d0e7b24b43f139b75ccf5777ae855ba211b629e6399d387f93a68eae0a6d8099038dfc2a2329a5ddd5c7f62638d9888d88cbdb559e586202dc6657c2991628951efdf1f89ac9909f5869095591869&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.517742304Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/ocdbt.process_0, Name: checkpoints/4160/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4160/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=095b9c6d41d0687a64c3e2c985258df640f5ed988399df5a2d22e8ce03f11e438ae72012aa900c5cb79a077e6d11df85931f463c0f20b590c4c4d56d218913b3c562ebaf82b6c9017c1a0b4878feb90e9efe388f1f6206499350fac656abbac5fa5e8b24948f3990e1168599117b940311434347c20b06549bab963f3990ee0eb29892fdfe880e01aa03c6e8189d6904a878b1e92d7e1553f57131f00d8df24055c55e6a2de399353b3b8ec02b57187326a2d2385879cb9c8978cec7e01f93f3a7f9e75d9250285fbb26250aed8a723999ce81984981cc35367cca29ca27f60aaeffdc1f20a81b4b01b3f9d3cf572030700a61b23fbd148db6e2305dbde21667&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519210522Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/array_metadatas, Name: checkpoints/1024/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1024/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=92df37ee08454d099089ed04390ae255b12f910bf9c92d2a2a2b25fa28130914ca063914cba0c3479dcfdcdb61f7c62dbd2727159da8a4f81ef6f9bd799d2ae204c075f3a1f90b12d122376caa0120d95fda54caa8b0fb17120d29c105083f43e15ed5c654ea2e081914d2b85d4333617a8d4156c09fe3ff9d3fdbc5d9d9d8e07ccd54f3646275aab95640a86daaea535d16a1199aedd37ce50916ee583dbf051380af534c6f71518fbde284cb6c7920a3e7b3a699e0308b4d5a4cea1bb6c75c5ac745dba43ddfec387fbc70ae617d0111482d7ae71f552f405887b2bb323d6eb7ea77d6952425c6c6a54658b2e2f75ac4c33bc5d75e5ec22d8a373945e9d1b8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519234516Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2496, Name: checkpoints/2496, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2496?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bd78d45640fbb9459f883785e835d684ffc135ddf14089803795c7fac384607551135e7f52715d3d37460eb9359c2b921e7b00b7d75c72c7b232f2ac602dd22ce074df9091d9dbfcf35115d00c208c8862bb9e8c0ca285f11485e84c7adf3873b457476c255fb532bf59f4526907f763a4517df83bc38450e99e51182eb415a6f9747c8d781a63172e9a878e8fa517b38e829ef575246cba5c71b5ed0904fe0063806003b214236869ccc7edc5ada269710871fc103aaea59efd806e7b03ca0007d82c0dfec400850b1b9c12904d21f6d9b1e047837d96702e54b3704e23233b30a443f39488bd980e0008e854acc8121ca32e08110a3805d852b91015f4d625&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519288913Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/d, Name: checkpoints/4608/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4608/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bd6bc6f85f1461e92f61f4510367e317e92abdc996e64a801fae012a74828c9a35685d7b33dc899747cbb1a312972c7f708115bf014452df6165c29fa8a0cf77d3033ce60febb834bf3b83eab84f5c5d58c1fdd81856290df6b6ee5a9dc1e137237af56ad7a19d51c1a1bfd5889a0452ac48825c3735730e240b8033467c6513ecba12e831caf64bfbdf97a810497c711dbe9e6eb100a09c6a096e86991e73bafd185461cb21faa36ce9a005109f5e1e0b384284f6f3873acd96143a8d6e13fdbf1f46563eb9b9e8bf1112e25f671fbc77d7a264c7693d5afeb3dbdb5bbcabd7869a2519ed3fdcf43eaaaad1b1ae596ca28cf5ef2da54e2df3df6ab96e1dc639&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519293588Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3072/default/ocdbt.process_0/d, Name: checkpoints/3072/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3072/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=94d232cf5064cfac1a7a254a6d24f06188e2955e51c4c2ce5098aa46adbde5e79476cb1e54a0fdaacd216afe3ade7628410d33af92f5f6d33569f9929ac572eff47cdb5ecd45e24adb0ec61888ab2e48976df995a6d455d200090d9b7b0cfc261ec65040b877f8dbab835bb91e2c9b8bd5e590634df60d209aea52a6644a3eaa6f1e83044a9c4ce848181963a118e3c8c32a9d03de51df80f4e215479c8a21b2bf1147e2c2ae63b7a69de0eef763df614902a37133e2c8a74b5bf590fc596a787666bb581c752266d815e4af40be8fe77d60e686164b9d87f2bb85a24cf65b8fdc3f3ba25b6b3ae2e2880edeed4e8dbe1aa415a28c0ecee22ae3648d0c757bba&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519298062Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/ocdbt.process_0, Name: checkpoints/5632/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5632/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3bf97277e3abcee9ea8aafd3d4849e688f9698bd9e6226c09b7b5430e210c44d3115c656ca3fabf43eeed0b3c2bed90949e64d28c737421868c539c2063adba5e28d2206b6f89e0ec8afd356cc406335563e290b25567d222d96c8b1d78f50b57f6c1258bf556e98ed98a666242af095e315744bb1955687f2c77425ad6764daebb8c746d8bff82e64c8952a8991fdde0e8ea70a7577ddf40316b593d95c00d26e649a3b2e45133951e66f3906993d01feab29dca391ae5fe6fb31b80f9bc99a8d2d9eef1e9506938b48ffe3b09e5d979e11b203ef9fa5271c26a942365b9a95da21ddf08a8a6ae9650cc5c2fb34e74b4976d807424e2ad7ad42dd5ffce93c3c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519362649Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1856/default/d, Name: checkpoints/1856/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1856/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=58fea280c9473759327c8dfd20985f61286e69ae96ab3dc3ec98d21bd093876be00d4ebdd3e8f0d8cc3ebda5d927e41d63a267c237cda87d93c907f34ccd185c797051762fbbee41dee1a82da209e76a56c3b174a098ab753474909e77e63612ce919eedddee01b06f1aa1640a24c4aac2a4bb3c079d137d454d6ea9af4d6fb7cc1b92135605911355932e7b569f308cd1e8638b50d966571f56364efb5c767b1da37d826e41839b2be23054eb77fa48710927b79008bcc0274bb7b968054c4a9f2383bf13a1115932cbb736ab2b62120cbd866aec1ece4e0adda43f7980b43a55315887ce4dc2b58379deef28d3562da3ccc92e40bb62f1f3bfee52da139193&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519355691Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/array_metadatas, Name: checkpoints/5504/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5504/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b40abc17ee3b230e8660f533dee49aa2c82f7ee051879b27975d092d49a3ae315111320633bac0eca280f7fe24373abca7642418418ad7611b253f89000a85bcc989e93fc7b3049d569529c5eec4b8d0a90f2bf922372d3a5fb13dd0bc7a767c8965584dd1c1cebc3168ac04d3a0879cc173afd12ab29e2afc7c29bf31f514b2cf18f35aab05d3a8eed93e3e528f8051a2877145e4c48a3a2ce3867947e679bf9bd3d2d936056ffd1f4dcc9142ba542a0ebc713ac7923b74ce73734a0a53ee96d99f06301cedd3436c12224fb57a3a2d018407a3e827c2e41d5a3672719bcead2963e80ea51e3143d4eab6532e9db55a679d7b8c1b06a0eb3ef5d16fa94b9497&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519345441Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0/d, Name: checkpoints/576/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/576/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8a17e8e2a351f21bc5b24ac91f0a78ceaa0c1ce382550de35eb20ed57f40d3031cd3bf7fbaaac0e2a393c59b78d609c05dd283027d8f0b9a39ee7f5773575d7bd32339c28f518db36a249739e4e51f9df3e186d7312185ca687e21d9c804bad4376b1c992f6468a2f76f7b870be5c52ba3e848a5626db2c89caf348620b423806398227de5a7454a65ec61af6d22f3df7a268c5a7c45bdf191cb44c1c4b6e2f89e8617a28fd60250dbdaecd9e892f659ef59c41944da44cd9ab0360960266161a9f01294b79f0098656b82698e9a67998fa7828de978c6264e4a399ec264360e7b4edb7a2ae912d64dd311716d119d29af88a13351f632a29844e1d3cefa2bc0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519363611Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1664/default/d, Name: checkpoints/1664/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1664/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=adc2bed409a40c606e1b4f21db8b8f8743c4548061fe621b7b1561d86def21ffeb84ed1b792b8d3460b4f8c51bbcdf03c5e97e917c1d58c7208242c7b04b73aaa6f2dbc0a133406404ad32f1114053435c76348da688903cc14a9351219d68a7b002ea84468835fb56ed557681b7d4dad8d07bab04aadcee93c4c7d154c957ca836c84e94d03a7b481e791a49d0a9afefa799da5cc8ac48b707629513ef2b7d4e280feee5243b9602f93b9c852928be26e70624b7b1d51f27391b7f57d637c40ab8376a16cdabbd51e2b985afcc8c9b898d9346fa89d2349bdcadb5bb8e5bd60134e7d7f0348dd6e605c0fbe2a293cca0397c624345b1ae025f2a61e72ba3e6f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519367957Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/384/default, Name: checkpoints/384/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/384/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5e8951b1a64be51573652a907375b0fc92f4dc1c8e4e636d9898614ccf589dd0f9e19047c1fd5c1b1658e08d44ab377c2ba760b4f0c311e86de3a2bcbb8abeb1715dd0d90f247ce1dcbfd51817c8ae6a6bf07dec3bf30b374c3bb2f566202b2a1f234fdec0744eda892cd60ed611dcac36b7fdf39f94454e98a8ebd69b4bfd9a05ed3a6f1eb5390115cc3363f97e426299f0934ee80c86ba6d63a6d0dd94d65986a8fb25762965915d1d11ba4522fce729dfabcb06c09f39a2e304f7ce2bca1924c9121f041f7785115cddb06f88156850d16506f2582b4f33842c840a0c504d59746669666e5c5961f4b157ad36a4b5e69221858dcfc36b8957f8b63b07d088&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519402783Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016, Name: checkpoints/6016, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6016?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5b3b75aeef2ab0502a5d653f0f49dae4da0cae0b183acf2fe4e937c78407a92bc39ecc576433a4d8df571c62ca3b8bd7956b7d1769dd14fd5e04492b182b15431111598e8885f4ab001386791636e608ca3dd989cccee2b0968bfd7f44da6b319345b8b81ad69232681150f5bf5b6f5e05ab964fc7bf1244bacb756b288329b0072518a04813a0f546a4f4e1e670e1f60e6986e0c8503b2c69913b316de4975ebe9cb4da447f80c85efa522dbcbbacea4677f5d4f662d19fd7a3ff1c41171d75a7d734b91319d07356afa9e0646b10d51b9a26effb39077771e43dcf16fd5d089806c3faad892f9fdf197ba37a54a27e6532520b1db6976cead2ee9c1e3f11e1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519420454Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default/ocdbt.process_0/d, Name: checkpoints/640/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/640/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3486b2c5fcbece5de3645ad3c3935de1fa87b5795a03eb3618681c8b30a810e864b93c004567ae0a941ffb462b4f9ea4b5a036c49eae54288861c6995108c6174284521ec3881498b7e22da63b49fbf64d1805fe09b313551a965f818c4d2280b321f174f8f7e62e36230c66b25c505fef80ae535f6edc95c0f09e3efbec9c82998c6c2f631657d4455384a9622f23171e0d1b1d8c95097695ef18a0e8ec67a726a4faa127a001f02292c3e9d813dab51ea1de0dcea9137068862113676ef9d61ae19f1e4490c83718b3445c716ebd4941afb4fe992ea0f94778af500316239b79bcf9cf5fb33bbb434c09881e37c03bf1d3b3652cb4c1f4d709ff91bbf4290c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519443291Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1152/default/array_metadatas, Name: checkpoints/1152/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1152/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=59e744f47bc0c3659f97bf58ef4169645338d899e911aca2344d2deb834609eb98dd57b7d5b2e22eda8041a8e3a5e769b28ab0e3b0f07f0fd11a2eb941ee96e78950ebd3628a94aea21a53be43d3352aa744d37197ad9b467244f22a93c8495c4a6c1e792c123337d2131c1efc55d477984f67b946ede273d056c304b7d5d5371452a70cc36e40f5fc10a5f5ab2af70dc681139dd064057aed2629c3a67ae80a2bd7e4e57eab095a49b6ddaf10bb86e5bd4535377cd536b1ccdb819d456d3c7b3232b323ff9fd22670ee066f831201f4cc5732636ad8dd68b8c20be464011c95e86fd31334ae917ef4822f181ef48c6a926f40289096fee627e7e517686d309e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519464762Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/ocdbt.process_0, Name: checkpoints/2304/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2304/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5aa1c1a7027bb7bff9a2c6e41ba9512b37cdb67f1605202f49d5dec5fb32480ce6c6489a44e1a10603cb4530cdb7cd05bc1928749c63f47254fffb581436ad1595706c60181a4da2eef08a8ab505e5c5b09557a8a62ef8542febbded5edd5b32896a5fe971ca5e603fc16e93069526b03767c54304ae49236d9768fb3b633d3d2a52e4b96235c9358074a653bd39df07c74529b0ed70440f5bb3cda15ec20d783a9572d73c463b91e89ba9eb7be33c0395999d3b6eb08a3a2f64998c1435333258b11e668c6b52adaab6011cfc468efa37bc853077b0614e6f2211f385ccb3c8d7fd062f1fa22bb5adeb4564a5a8f992bc271eef9fb13a97251d9c48a99ca4db&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51948447Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992, Name: checkpoints/4992, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4992?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2f92624fdd73453ab173676b07e776823927f6ec6f3262f493eb0fd955653254ed03637fd3f941bccf301f5248c162c96cbb11721f5ee0f4cc5955819d08f28614176bed5eceed39b8de6d4248dee61b977a6f0026dc0ab66c76b3423790b4310b80620ba6a030d09d997cc93d4ebe7a784b212cee00fe714db6497022d910734b42021bed62f352c805566ddde4931d380879177f316539412966ebf1e1e15682f603d6c4bb46cf51e0974ebf304aab9badac4f11c3291dd38ba003c715a5b5fbf33279d5767fd7b1f49af493c547423460b65d281bffba4c3ff9856ea30f22f528b2967c654634164cda9b54a09add067182336e9b0c2c28c4bed983a5a1a8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519470606Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3584/default, Name: checkpoints/3584/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3584/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=15e8ca757c894e9496518940f53885a85597441831a330c3547506dbbb2082fb5ed9c4517891ec1cab40b918598bf47d23dce7f31ff0e106d1a5436e95e1941ce98be336091c6585f46343fb15b8b4ae8c756ff6cb6d3635799d2209f9e4104f8f0577c76160c079468a7db07b41994a31fd02068352371d218073fd5335dc079082cca212527b67e97de0828d819ebeef4b281c814214f931ca8089c2d45fd85b95ef81913d8ec9f92eb244efb4273df4a6709a708b35fea47e339328c3853d789dc69ea524d8ed90e7291279b3dcfbdbcf741556b507ab8a6208602da2993217e541ed537b7fdacb00a22bcc20b04bc3aeb15347405e3b46e21a6cbb2a1960&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519473376Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/array_metadatas, Name: checkpoints/3712/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3712/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b9b602af34404de1915a5afe936760f6b6d84368a35c2cf2785340f4ac7b452eb54abe18340a078eef429e1c76df4b48ab0dc9a0d1edd0be70d0a7cf5e24784977c7b85cf1cb65801b1332ee6536b443e83e122d469a440bb242b3587335d16ad9e5bae5d269bab46b2c17dbcda4a0a232dab6048266a850b25edaa2ac54bb07a6b0267b78c3ac0e6844b146b81f26c634623b902c0ccf7deebd64b317af5e7d7c2d96b294622343351abe5351c8cf4a1db56b5cf6b7574f044b4c48e5ccb9b774a88878983e5786907b181dae9deed7381e8861886473bbebfb0d480951dbc1004b47a1e4517de41e09311d5eee13b279f7ebee726930955d4316aa67fc34fb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519479276Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816, Name: checkpoints/2816, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2816?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3f28e04ce01bb38e2a2852308e9f517758950c6b1dd832fc77257711dabfabb60c688122576f3edeaa62f7ae7352552d8a6940c6496fdc8ecd2f58c3277ae0b4eee24de7612396bd6fe737baadba28766f13b4fd7c6e7417782bdd5e848c7e66616fc55171e6f925f6bbb4b3cf72f18c1ade02ca37df1d40827036a5ced5d039e1e1b06797ad3d19da88bd683cf6c023a3a1b58da4a0a99f54bab2d9972ace395433154f09e7175e6062ae26b435531101b2366f6be6569b84a820ad4c5b0705338a2a282f85ef28312b6a8f56797e9cc8a533601c9cb68193f24a3bc37bfb4f22e875b42c575d20f54716d2b2334ef445b5998cbe3d061f745acfbb849bf69d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519524013Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/ocdbt.process_0, Name: checkpoints/1984/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1984/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=1b7b3001162ab4d123d208506f9546c87b80bf71cea78de134608974e9ca9be17e9ccaa66a5cd2851b7a639b1cc07210e32a63735eef446abcbb4161cf240bdec0fd9d8fd4c477624b47d08c756fbc3b13164da1e00227ee10f99d199cc1e366ac8c691523181ef5f7963449e1d3f14635aeb6dba7a92c9f95c0446ee9df56ab0dcaa9052f13155a6646c6385e998df6ffca0b337e9b364468bb180cf02ecaa66ba6931eef0210fe017786ec9b1475b0a51f999ff9e81ae71ddb54a111d31f7da2ad497f68c4db8ee1afe2bfa830a15b8a1db7720434ea4ae638d660fe65047fda973e0705f2709f987d7fa8700e3a410e2e39eb6f6aae8cb792a3d1ef3a62d2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519521638Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3712/default/ocdbt.process_0/d, Name: checkpoints/3712/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3712/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a1e45edce35f90ee7cdd20934cee6c0757ca8b55c59e4665fff3f9343c11655213b0f1213403a4f58cf01a4699269d772ec4e314bca9e17fbf8f0e5a42a07901d527306073f027a73481feb13ff95060dc898939f7987d6c3786b65b79a5a55f9d6553fb136bb6ec2f941855e979cd6106598670c91550d2bb2e9bf160beaac7497e66c9de3ea5117e31fa6f19c908d6ceea8f9b69d5b0aed77ca3b64e347e21742b6527a450a6bb64dfbaf6e0e080162e399ecf23d92ec1777f6f170b58e4a1a2c504b64b6eb26cb85b90da755f80583e1d7d44f24080a5fbf01a30df4a07a0fdf6e70abc0029e3f8594ddab53b47e1addaf898fc2fd29a689f6788e83c58dd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51951561Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default, Name: checkpoints/1344/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1344/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=84a990b43acd0f128a44d05ca9a958ee3fff13563ef321c44a738ee50c26a8b8abe91ec8f191aa81caede8daf790f01a951f78d5b09b75fd7d436aed32984da8f0341ed4f3b364cf0ceeb8ac8c026245530a16118210d02007d7e789edf60d975070c99b917999307b7d832648eece4483117fe73cb719855c3faa8038b038d6ac81bae928ec702e2fa3d463906f3b4f69132f55dde14e70944d7322af1c6a6c1b85b45b5413814185e2f5b441d0abb2c2400ba4e1bf27efd78301e25a37407b043283d5a4f8e9d1ca535c693dd5e1a0e8ff5649bea09c7aae9d9a63aeed9d71fb980f6fd3043fe536c0947cab20cec6059fd4ab560f37f29ad2d736a95103cd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519538242Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default, Name: checkpoints/5312/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5312/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7934008b8f0a142ff227f482d17d5aa800831c20b316d95b9f3aa9a2f722bccb7c175f979d60885ad9ba1fa66e71a3aafbc544a18ed16f0396d5fd85980dcd78df56a3fc0bcaca3568ae2f7e5523323f8e6d3e27c83bf8d618c3aeba43e43aa6bcfef30b9f6b0320aec3c2b1e6b1610607e06918bd645a550160233b04c877dd7b7262669008f61e69f22017bf69c0cafaa6b1cceebd91c8f88d5efa1c580a6c503bc79a629f58f4e5a5721b05ccf199a58ac7928eac7f7c2832baf578cc11c81c0efc6e97377b93263874d7f99a617375e705f05283499d0be4476738018c07b67cae14573085022b61e22b13f5fa044a4ca6ad1580d26135a383a5b1d5dfa9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519541543Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default/d, Name: checkpoints/3200/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3200/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=bf1dcc9b133ab40d09770bf7e94aeb20d183552de4a7a904c1b621afd0640e3b5142a583e06353cadc2752c2511a08006af6293470bfb56aae81118950478b012e4e9c858ece631e28416d3d21c31560efec1d40785df436d7b49b43092d1d67139ca587c00ddd88dd4860f73f05f1d89260e1daa527eff55145cccb65157b7ca7d1d7c8091c7257d31137aa4098a2365b3cda9d14de82803724438919f1c1be373129037e8aa8db60a276243547e2f8721e1fe9d043a34650a1c77e7db9605dd1bcb4ddf852b1cbe5506c04ff4291b1fd2ddaf7346cf8b324e4bc6df5218c07549de8daa70f270ad7578f8a52b1a8cd5baa3a666099fa42584c21843c05ffb4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519537875Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default/array_metadatas, Name: checkpoints/4224/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4224/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=31fe3f2da634b3d548768c854487e4aa026b5085cc445835de00f6c79065619ff4406ce53aaac057987ef7d996087db79b864bf1851ef3fa1b78ef432222d9d11fad6b6ab121a87f945905115671914ca7f3bd2d023be8ed614104d8cac69c53ae67e8f62d305de0309ec208761a82c4cb320f0704621e83a59d360fbfd19b8c407280ea5b022679f7825274b40625e06f9ad88d8d2e612335899d02c3a296829a8d2be63823d2ccd81a921363b7baa607255672b4cadca4126614262d6264b4b21534187f3b71ea733cd9b9beca978cc73833cb84109950b7c2de4acb4341cc4cbfcc1ba3735c356647eaeca7b16f56ce857206bd7fe2578adad8b5a438fc18&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519588502Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/d, Name: checkpoints/512/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/512/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=591ee0527b7481008b8557dfff2d8d360ada70cda909c096e6501eba5566a06f50e8099ef973bfb5a82a94b9067d6d0f09a039820ce7627e8f3116b254cd13315dca3efcaca46a68d00166244e1971213edbb4fdda085063e00857eeb3d5f854b4e08a6cc49e9a54a43d718819c86f04581e00e17fa9d6bd693487e798b14aa8cbb00bf0c884d926fcb6f14425dd476dcdcb1a41d5e0c8ad6c78c9a87c78d82bab715bdf3c87b5714fb66013f6af4ea37addedc9813e8b5aaff732329e8385983e470294a56d637c9ac9181176e3aa73ae12ecdf7cb0e74b1536ef0db5dc1a0730cc2aa9aabb27108d06737bca883d51e2234d60cff35a232f515134b3292ad1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519606345Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default, Name: checkpoints/3008/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3008/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6168c81f466ff11adbb3bfe702fbdd06ef3e6ad43245d60fc4856dd17a61c2803681044e542bac84f68a825d6aed78754bc3728efe71306279003891cb90e52ce422f5082ad6608228d7eb2d59388af1e22a5b96d7315bc84848dfa3b58fc7cb49094a803387b4a4c53c17d2a1979f14320521eb3cb98166838b3592564907417f23170e7c742c96e8e459d6fa593ad7d624417083fb4e2c3ae9314fa3207332cb6854f7881d1616e0dfffd965b5128dcb3d8e437860e2bca38376c20d04b26dac02865b825e31d5c4b6dc5290a1952ee518718dd21072cf29463f4ebeb7eddf487c88283b620b039a70f315a51b4eb1db0c5fdabe86ec64b7c9bb1dcfa7a4a7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519629491Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default/array_metadatas, Name: checkpoints/5760/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5760/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c164855580eeb445e7d2ff5ed5e80dd132736518aa789b3386da6766597d8d499d9b1288ec506174316d8a0395b0efc529ec2780051341b6bacaaaac77cdc65c55820eb77f1dac155e6c26e85cbe0367aa37974378e6c06428ac1317653ed7cdef272146908f3c448fc6fe27df50d566646d911863c01c0e92e844f50d6c59814a06c4c36281312b73dbc1d79f56972ff16bb7e5a058cc50896f737246a8976bbbf72c358335361b8d1ccecc8c7021f5b8bfc11bf5aa983dfb57a62fa211ade6ef947bf2d98aa5a39063e5e1e188b4d17e077f2df2429d2acafd18b0282c5f67fd5f7d1633ad237b5819d23a24ac29d0bf91ca321c23fac40cb02fcb28b69f2f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519713479Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/array_metadatas, Name: checkpoints/5696/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5696/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=398c7a60f3f8e4c418512a85935488c3a01ad9395c9de5ab794906f00da290b4836d21c0e74c4186abd699860ca0522d7d9cbcecf9b1b09c2ccfc5414db59019f55fa5f36b86131192ce8d0e73081bd5d49b0e326b94e2c2bdada4ac3a6a7b30fa3f3cf7bb0c9dceaa6a4c15ba8122d2ccbb25b38cffa1955f34c6f56ca9957ce4fbb8a6191d4657f064d4c267fbf8a0c5fe3aa309f7c2850e49ac8fd2d0e0d463e3af3f42cd556bc51666f93be653257d1ab21ec8ee90e42955ea4220d962a640c25806e1999caba510f81d64621a8b9cb5143d75a0ef1c6b4945e24d31d0e215b262e662745e7d3716a67250415a8bf73165e452b2983f802e87663dfd1be9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519716149Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512, Name: checkpoints/512, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/512?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8bec80a5ace7ae38aa29f03195e1f9b85aef7a3d62ca1b6c09d751fe7ffa7250ebc895d0af0fdbc5ae31fece480eed1fa22c5708983460714d5f4a70d964b9a395192b84a1e778c695167630558b8849f6bec1bd8a06385463dc05af299bdf965cd513b7167d244eb62e9cf9b9641df3d15a5834636c6297aa26ca403f85db45ddd3e1342da9d5961250c9ce3f6fe6e404ca73aeb80804ba8f1723e36a023b1d197d4603bed3263abf9914f86db583d5a32a539b9c5f9d3da43e99523e4f63f8c73e1a0bbb890c764926dc01568296074262c7aa324f3d269ab9379b96305c7abdc436966c8b46f95cd16e117926431ff0911007fc3bc93bd71372c3b93ddc4b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519796874Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/0/default/array_metadatas, Name: checkpoints/0/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/0/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=28dbe58fd5c270ee35713b548551bdbaaa532cd26e70f76145de179446edaafba89b51de151d937739557ee91a3b3e3494e8af4b32365dc2f38ac24cd7e62b603c705045ca2c6c4ef6b0434c494d9f4d4d4fae1f72617219ff3588146e68bc75a6a5ff8afb66f4333fd1a88f1d579dc223e95a4b85b80586e3aadb7d2ee68e5b6a33f88464339ae459ec8f99b292b517125e327bfa18cdf60fdb0b70293f0288ec970da077f6ae4c1b07960b8936dcf6350f05fef0b340b941577b15e701fd9e6340b6a47230938a3f4caf5bb76729581d467979d8c3ef01a2d65d97abd87c10e0ad04ad48c04d29902fe5dd131cbb81b2ebd036d22ca0d11f3efe01cec89c1f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519796982Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4096/default/ocdbt.process_0/d, Name: checkpoints/4096/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4096/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=baeec32a0adfb8ca3f96fd3190491f5bc804fac843d684d6459f776e246f7f0b6867fc99ba847ca0128caf1fa28391d995095747d15aa7bd01edd7e559eb18bdca401af5050baacb596f13d8f28122d277c805f1a201396c4db8cfe562b495a800a31c0405afba43eed4650f2d44a025572ebe4dfc3fd20f3dcf8c255facf46746ac67fa7bfc20f1e3a02dee77a2c4b71a09b27b2ccd91c2c2163c7d923827cdd12c99b7c0350ac1d39c55f69c26a9392a9d77ab5cd16561f9ac4c9c9c7fdb4bbef82bed71ebafc5740d55fef9bd901a84a71b6ed4c6d52b98d44cae5244f13c36922f8ea88b76afd911f820ec168b97a3097969fb30ee3647a51f8022be3d06&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519804787Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/ocdbt.process_0/d, Name: checkpoints/4736/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4736/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=68d250b2d5ac4d91ad92526504b22ca57f874c48afb2cb9392e661ea852730757899e97dc0c9d7a2c9c45e3d0783c859e68250c8122b2f8c08a8bf1c7e5bf6ef164897c441024064f63ead3bd4ae9355dfb2fc59ef5744e1759cc6e81533d419864bbab443fda2cc4dcff44a047a847a86796c024355c2cfe6aeba5cd9e71c64c988f705e04bed4c48cfe88d650f467e2aab3e525d6be0ddd2571a29f4992a4c387793e0e5f2896dd418691fa3d74672546fb23b41d67396170df1c159bc9f4245862a2c61886c794dcae15fec00b15fa61ef34de168b31780423bb6bd7e0f154ee3dc7aed145c592a0b9d71afb379aa60882fea6f76988aaf28e35fdc004f21&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.51982243Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136, Name: checkpoints/3136, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3136?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5231d81ef1baaa81a5a0fa89d66af913504870fa17e5604feb7051841b42ecc3d9676d2548f9ca8605e5597dd24f992cb50ec171e8449dab417a3a3f1eef67f2399fe9564d38f5dc2449c65e507502a7cfe7acb909d4e7f5ce394a7caea15008271e20b7054c3d693f0fb68763cfbdf7b305141beea4589123c2a2ad81a18158b05f1b606407187fafadd82d158b10abe73153c24e657b4adc87601e8252ff838e6390128dee97deba44f2f8e213844b8395601f4ed044829ccd513c0be3dea44ffc160fd6737ef57b134298356f6941b1139def68e2267d23ae819c4635d115b26acfa9afc8ab20ce124571e8a7b88d72dbf269ab35e505c0221d689517cb96&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519852611Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0, Name: checkpoints/3904/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3904/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5d925c734a5a39dda769b46fdae9327e1c073c817c67836ce83eccee6ee9824e0cbd8e63727cab8c978158ea23267299009e436093ab6bf8c74ffa1947a8e6e768c8ae7c7934895336612782e5c484388dac0a5f3fce575e47c30fccc2b14721469826d00602af6bd50ae9057c215c3f678d93d14162d30089a21ba488e9a9d431c45d3f54a53ed5455764a81a8a041476cf50f085d5aeafb23ba1dfac497f0d156f7b1f6e863e68509acfca3fc780377c9a50386897c0ed3cb10e51f42f3a2b2d612caee71d32294f42b9d5e0f7d36db16e74b24fe29cbaf59ab103024a702bd5cb12d2caa581f52f7e65ec33ebf024c30d7c3bfd3d4df3d8b55bd9d19c5659&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519885473Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/832, Name: checkpoints/832, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/832?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=15237e6e30e2212e82b2deae7a1fda0d2d82c561c35add0b5c3021f50e749494b6abddb1b6dc6146f8b341fee46ac3febe3e062979a9a1e4d96610e3aa35edc71f09af8ee66064325aabe1c9352a2d04a8d59863dfb2f57132bf2de09fab21495f3700f2f84e6e12c07355a8241601a5194e4b91ac71997e648c880b5e347c340bb8e2faa00a62c92b8eddfb9df7d84c36a11fb605b83f4ceb63bedd7f05484e2a05f8581444b0089db3637c5780f20de38c0c4324927a143fe2bbde7a9215e5d453d181c033370222e58336890b77a5dc2cf65f7f24c1c62dc55957d87c32e26827be9aee122d0d3f1c266cca76b7aaf530cb48b5eb17e26a7b00b931418b96&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519958314Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4224/default, Name: checkpoints/4224/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4224/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=875c66456f023f0da00ff09cbb1f83d5e181a6ab9ab27d0d3d3351d26bd7b38f32fefddcbcc7ba2377cae9e0c01ed129c8eb6a77f2fc9dc41f7c8b4e6e20c3fb1fa4181ffbb79a9375ea03fa0f674e736a90967d4d73d8f9caa35ae440620fffc1e5ccc0173ecba174871a163f20bbf51f8d408fd3e47c6174670a0b5c200cbf4a6d062a4e5d834477e03f55266a55e9ec8683730233e4863df9871ee275f84547f04bbc8065cc0253521bbb01d5d683c8fff1dc3551d3f87e7f214558855f0d9ce02b6783cb3024cf14924bef84ab932feef92d170e475d429e128dfea960122d8b60a15e45e42dd3dd077b9ef07cb6db9a3a0b57b577aeb93edda94cb23448&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.519881975Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2240/default/ocdbt.process_0, Name: checkpoints/2240/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2240/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=877ff4af1fdf495519c9215890d392043cd06e5aaaa304b7f83d9ced35ecbafacc5f1ce65a80f9d3629993a8cf0eb91ff81f9b98674a6e4e15cd67cb548be489c9866ba9cbf8e734a8ad80c3f5363f9891409e5d8d02d72e8a4509c6fcf6123625867f7a755959a1410236dc70eec05a31e78bca3e1472259b686c15a9a9e336f840a6cf600d3104249a7e7ae47a3dc9cb248a0d1403fea9bdf5cc4b47aa9aae8825d8534bb10d46166cb207f7d80c7c32e95ced1fa83763d9f1bd7e8448bc35ac66d6100ba6a3c81e8ebf8374fa2efd968fa24ea8e11b960010438793cabff4fae3db6a2a126a3cb70098eccd207a1b9c50632a9050aea358fa5b94a074f1dd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.52005056Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2944/default/ocdbt.process_0, Name: checkpoints/2944/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2944/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4ecb66866266623f7ac65e8351e92ea892817bb6566eb7f16f5c4fefa59fe067781d6485a8cc8057f940feb489b0b0c7ad343df33bf887126b20055771c61e3c074618a0aadb7985e89248490dc4645e81ca45c1eb573179c408eed9d57cfae8307eea819ed0e70340b04dd2edea68c402db59c833aa251c83251a2d5e5ca0a7ae1f30bc41ef359ed844fd5565f2b5592492d6deb72de654283de5c1a90c3db6de5e2c1a6e91760c6373eb610b38ff952e27246e7dfa98b2ec367caa21fa9dc7e23ff43d48ff1056f7dad13cccf3ae44ef81c62d1c846957cb46af40513cda3b78787aae786d7b92c29f43ba747a1e01a2cc34967bef02a920f86fc061d5e284&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520133278Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3968/default/array_metadatas, Name: checkpoints/3968/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3968/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9d3a6de0cf11d2b4328de1069071b2330ecf159232a137afdc32a0d9aa080db66d49b49a9eeeccab0bdc45a55a307d8d53e76218698f992e7b33d7ea8010d46224e4a44fa98b477ec68760ebf27b04983d9aff9102a0a8e57f74066a9614d3b7f4a3ffd537b8871a002e5edf41e3458ee89006eaac4dda41dc0a93fca289056acffa43977a09eed2724a690e0d3dc51e93765dfb08babbe1e86c6aa5f04f5d9c7d830915eee06c67cb26dd346b619d88a9e8d015befe9b7f6b016c0aa518cb9ec0731931bbee754499da879f7ccd672f42408680c5e77a48cc50c47d5ce0d22d92f95b347a6a43745c9648f4558a022f21d947405f904eddf504e62797ed94f0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520119114Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/d, Name: checkpoints/2432/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2432/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4a9b557fc7469ef574e720e50245ad4cec29bf5f4f3729fa7dcfebc45e886908bf19c4e9020abf2ddc55ba0e2abca7b1f5beaf11d9435a709e955c54a4b65cdc05d364d7671e1f65efeccd5e9d360316271b2d0f0f4e90f6d0c7725a2d9240b039b82e588e3f6c62486c6754534585743bccc3b48e54ba63b60811ef05fe5e7a982c28bce50ff2fd52c66e17f4ad8c15eeeb2829a06f1633ad9d01a852d741df78dfdacf2076a3668412fbaf5cc078410d41131f93769e52a05729545816c120a141848e8c54b8d17a61fe3b1a2aff02b00a6682db062195e1be9c9d30ecfdd4ab17e9db56a49cbf4eff11893324e79e2ba550e40cdf71b9a967cab36b9490a0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520163876Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0/d, Name: checkpoints/256/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/256/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=93c84ff06c20eb546a0826766664c8c3e7c5071e83dd677015760d5c8610337e621ee3cee5f065818141c988010c30f47d7b530acf4f4b7427f2ad7f42aa41a8f48862f0f2f008e12f5519340b6a6635082c2207b2f666b2009a061b58458fb26bc7efb2f63e263685bde130ed6055372a04ef293600b112cfa0fb79135ed121c88cb434945b869543f2b4483239b0d1b094f0dc63389fddff1818c221d57c599cecc58bb3aa9cff515fc45694023263ff275b83596bd172fcd3a3cfe7c38504dfc84a7a4c584dc944660f25e9e8e8810089f97c706d7edc5ab75f9a75690d6b84d5e4d95b20b65f122eb02eeccbe700a4f9cdaa1d242178a94c704e887e0695&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520158299Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3200/default, Name: checkpoints/3200/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3200/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2901466392ae120ef533543b1e1178cc30c524fd23a19f5668a4cb7c91ea0c3a4e5fc6b2a74c6113230f722eb27db371b99046426c761eeb01c9ceba1d184e24f88c998b45402a34dbf73c57fc3c72d6f239f2ed797a38807bb38772740ad9e4820e5c14b259bdf485e65e7ccb343a56a1b3690e11eb6b2c5a4b614453dbf44be0111c8a44def4a137a01c04081c6bdb8dcfe8a4577fb9d4e233a984c6c60dc651949cb8521de07a23f0fa9ea7a002dc743a706df0544d1f000285dab5e28b35252c20d60870f70f778c257f698c93f592b5512f38b71c86c8df0637257eb82d26286bc1770fc3462af6cf205345436b778cac1ddd727bf6dadcdb2722e8ea2c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520050896Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/d, Name: checkpoints/6016/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6016/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3beca96c1d1f4539331c3d2983f00f025e5eba9e6db1414ddf785823d78fb09ecb73711444445945e7fd6f937f5b10a705153dc661992132ecd4fcf850eaff12b140f1cfa2c915ecd85806afe082bdd40d9e14f6069b7620fd8e6735cf9c1a4ef843cc2c05ca7c5e66b4e48161b98d2810a3ba63e490bbda476cc29e27f41ec2bca2f79dbe80b6c1fbfb57c36677a5c8fa5a18e1902add884bf2aaa79a5deb6deba1734e5fb9feb5ac7e19b70b089569b9bfd683d3ec67a3465c5b8893eb2d811d58058aeb67700e09c45f6aecc55afcd8c9765ea1ed217ab972865a968b2bc0ce3cb4094fdd3cff62103fdfd4a9077f1553866ab7e7428a74443403b6cfdd22&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520159558Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2432/default/ocdbt.process_0, Name: checkpoints/2432/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2432/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0debaa329b2d282bd410dce04b7a40956011973c2473aeee64cce7d5d8ad240b3b6c6acf127fe36ff8d8e24719077fca07e07c1552fdfd2201d823b72b3f94803aeaf88cccbeee5a174c07d992255971f100de2346a60847ec86c5d75e25471695b1f5ac56f5b2bf9023e06c6df5bb118d8dcce09035d594b9607a9335f7188142b651f27a78402f6e4b2ebfd13c1eebc44dbeed0e176d72073b9eeb28ff2c47a501f6b5f16b87bd3aa81933c9778588c0889b3707b490b283f8eee1d789a95c2ef5cb0140c553ee42c15e06b713ed1561c78126bd2b3e07f8f2bd8fcebb7a1f52641be6f159febd8faf9df2dcb90f8bb70e71562678bd815037124ee67e48f2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520147381Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/640/default, Name: checkpoints/640/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/640/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c25402294b84e8dc255183975194c166ee426bf7f5f7274c2c79fb2515b355a03aa1f3b899487efd1d91e30b0e99b77cbeaa63de8c44555e600ed59ce89374cdc22628a311d703d062ce885cfcb8d17fdc98456d4142cacdd5b75e6f88af273f9a21a2491c29b698f3bc190e9b9a58ee7ab365e71cb022dff6f36894ef5f88acd5643ceaace962ddfdf92b252e16495d317e373336f204b1861de08683a39efb41323ee5621328796b632ca9315ceb901fd6c84f7cd67f22bafc353198be83ccdea0ae8cb460f55b86e7c5ce68e736760d774529525a5d047440806f7ab87d8690e7c085454d84f29d94f820e2d75022e412aafc1626fbf0cf9f4fd30d493976&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520150304Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4352/default/array_metadatas, Name: checkpoints/4352/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4352/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0057d5e4a2e67071483ed7a0318f0f76f2200c6361853ea57378f5c91ff6917f8690d916bb1f6f70040fcd1657aae61d7f526a28e8836c67e9f6437f1b779811211b09d3ab9a05173d4d70a3b8f9784bc55746f14870b0f7d285f88cbb5d81397f60c174db2aca8fdecead94b25dd46aa2f98218ee65b4d292a5a9dcdc8e7d4ad8943b4a42cca7345242f5b546fdd1f5aa9cdf087d5f87284e7920cc78905eb4907628485607387b445322dc6e25e7c7ea9124ee7283bfed54569692576eea17f09d89638c78a1e054105a346913b32be687b44e0ba1bf0dc634cd3994510ba0f154a0871e84f01ba05a5ef7b235a281f4104405485311c90de1622ee4b32f56&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520191257Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0, Name: checkpoints/1280/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1280/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=822b4cd72c294a66f582d7bb88ae7fdc04f7ccab8c15f9aa64094d2a18f6b6d41ce856113122ed2da4ec319d96ca7d981b80da26e767b695ca6bc4dd19df02c2a11be7b2c8b46e479f22ad547c015240bc0737b14991aec6773b238ba18772cc1d43e7cdb3f98fcd2e8dfbe1f3160c87db24a87a5f4f988ed72cd4d1067d6d6e27d0c1bd7535afd64c669b8ac116973216c39fdd1ba2be4ddd7d946e921be480f3045e12e802287c683e30e5ebff571f2acf7c9f89344adc480aca2280a05bc895142b305a73d9e2e9b5f4ca8e0678d12fe37dc02bfaad62a954e8964276d2bc50f0e4ce6d744ef3cf8b39fc1b02b82be4224b266600e1fe3b53a980d97e06b8&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.5202377Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default/d, Name: checkpoints/1024/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1024/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=66e7533a18c7fa0d1eef6bd2b2fc7fbee81c000be8cf197a4884b938ba01c2040030a3784ece2f4e1bff9df85f21a38cb482fab4e2f905aeae17bc3c2ed5c6ff993713930d56294bb84ba8dd54f82887d6fc1dadec6e1a870ae28574c75431e1dbf4ac31540ae28fcd198e187607a500bf1d2038bf159c16d8a45a062c6acb858d0c782de121c58e1ec60fc7561bdb9e9662c0376b9196c1828a7e718d84e3949baf0a1186e6c73314a7674914d191a58c386d370b8b69b82303409b52a52312fc76d7d2e8abcde4ea747ea5f04619ad1d3cfbcde4ef34a42824e3babc517b1fe6fd003ff5472ebb5c027c0a4a99dcc450048272a742290e8e5a386da75fd78e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520245284Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/192/default/array_metadatas, Name: checkpoints/192/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/192/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3cb00fe35a19ea5984b33dd4d68ee3cbddb243c3afd352aa50c51d552d89e38c2483070d5cf1966887afa0b973cb201cdede646814396cf928154d1ad0f4662a9fe4bf7f73e0941673abb2bb047b0ad99d46f695ac1b62f11d0e9171f79a00a1e6ec8cde12dc6d369f486604dc0fb04acbdf36696b7ef4995c9c94876fcc28d755b15765fd68d0b1fb6de79bb4c02693c90d501e552da39d63bd7b5129253481b5fb0ed571a51b4724a51963f38732fa04b179c10416f665677a7eec858db4ef07a8789fdc27f9e1f0f642f258085c56a72325930da731a2f0d4d21e72aec5ab940e214a880099483342de060d442c42fe0553da1575209fe7fe2dceca710c46&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520242361Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1024/default, Name: checkpoints/1024/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1024/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=60474308343f24297faa896dc6e2c9cfaa34961c3ab635ec6290005bd960d4723a6704e38330ff3dbf25e656c13578a745ac3f12434ae046d05c0828a304601d42e295a785e31892c822c341fdb536bb94a75c960a80b63e2be21de7187bb85779ad6250f998194405ccc43e5914b3f29030ed3796732e42d2d56b874b3072d38f1084d5fc9277f2a083aaf6a75b8263cabd56da5deea5abe1421c292a2ebad297d1a1d6a0f9fc4c115df89af401427b16e44969cfbaf0ee48d4d6d875b65891ef8c99fe7081507119b736d3ae7f7ac89986bc5e70e344d3efc2121b87ebce7629455113b14a1471b3fa0264178c12b19bc03af52bdf9e0017433b4054ed7ed9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520244369Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default, Name: checkpoints/2816/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2816/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0ed3fd666e51b75640c3921390b1d42b386f041c2846cd9bd6c86d6a2656be797f1428a6703e155cb2766dbf23215260a7d2aa35f35a7231c06ece9e001476c9b2828a6407d0f0394bcbfe7db4e61da3ca11ede5094186cffa125d9b98ef856e28a36325fb80cef10e79345af908366c7736919a2604795c2c39720bbb1b1eb3c5798c628f8e200b27be4dba2e17b141c897438f1e2ebe938aaaacccce56f2e59f4b0cc0b802cb213c3a23f760ea5a4efc6d917de0a7c38e58b25ce3f636c40a7d2645b6ce8af65aef590bae29bd24a4106ed6e433f889c808e8552243d37d5662378c980a6873b836927fe6c7f9a2d6710ade1b48167b5a21f4c612755687f5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.52025733Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/d, Name: checkpoints/5824/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5824/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=32dddde2b9daa763342fa72f30398adecb89d22ec18f8d8d8d31de8dbab900b2fe043539e40a94a39032cfe0ebd6ba5d8623aae9aef86ff789d61eefa682bcbaf342b84da99e6fef9d3f2b5f440a82e312c889b9f9f8e3e1d2e525749b294d775dad4380ec553d54022fc1a15d07f664a5b2ff8339e4a85534afe9c2aa64a54f72485d654045d0b4e573e2e77216aa97aa4d43890909595353230a5698c350bd63f9e149d8a9e9dd13cd7edc258ab2a624d8e3b34507c76dbf0add39361e0c9408ad9955fa26c9da45bc6508a48116b694e98d740c47af2970e2a1e3e0b28e36be29dd584806e94f65872fc558fbf82181bbab6a6bc43e0d93b95d7987265bf1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520244467Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1600/default/ocdbt.process_0/d, Name: checkpoints/1600/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1600/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8583625501433c6133bca3d5eba7d813ccc9becded114f0e7ce939b3c6ed455c833ac1c6aba49450a02c6a18d695cf20b91c65f2c470faf5a6593eef0164db0dbed77ba03e1a036574541d4fe9a02b1babc4ca7ace375be2d93c0daae8b145dd9210c2d6c378f1d2bf0e39bd2711d771ee28165d597ecb09c5931226be5452b772c1a291b1f4d2162306c7502802f9918933af20122c5f88003fbaefea6010a879980a1672b6433dfea7ae287f40794722727d57bd464fcd527b7dd1d67d3e5c57a24c807cb97ba5caf05ebe61aa98f8a03eb068fdc3ef8aab1647f2be766719ef4d9a9f964cd2c7aed8f3ad5e4cd8f27aa1f924f52766a16f72bbe0759476db&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520326369Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696/default/d, Name: checkpoints/5696/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5696/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2560ed2d3e598c1990b9fc175eca6a05cc2231d019c395668aa646f9ccd2eefb5a46164dc39f0baa8c40bd0deca6d108c7cbd6b74fe4c2dfe541a63b608c0b322f9051931817064140d50c3918ea0cb044565f0a33e407bffcfa96a01ec5770ae019181e0ab861a81ac16ad92b31bbda08c4000fd4aaaade42db6bf93267e3b1e9e3b23bae02c5e773ef186e9c196345a3040dbc0c79b04591b09f68756e97b897d27dbf7815aeee6d95e0a286b2d9cda8742dcfa5f4415899f15a1338feaecb920cddd38d9e856d166da40e37bf3c440db893fd7fe7e3e13030da9ba8b57b11b6103e0b996f358410739033b2b334f336f6b672a66e767ac2b0cca19edd3ffa&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520320332Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default/ocdbt.process_0/d, Name: checkpoints/512/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/512/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3bec6b3355d329cad43391552d72e4c188a51a72873da4c79d5a8b46903b095edba035906e87269618608982b7a2dd17dd3820dc114de07829465b01c591d5023f5ac05f5808c8fc56c1b7b8966474810cf788cea1b8c89884c3a2d31798170ca1fc8b3388af774aa791a927ef5980b8f6df5f92de46bd79ec0ccb52df7553d1043913185ab17f6125e3c8f8ff2089d628bddedac343239e3f753a00099584f58bd0698760b6a68138942e20fcf207a71ec3dd9dcbe5d6249c463532e4e79e6a5c2a9771d7970493e5b1017f0df94db8e12f9d58dc950c4becbc22b53d7b0f0824adadca62217f5652af9410420d364a9803ecf00e9dc14cdd2cdd2871bacecd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520332634Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default/array_metadatas, Name: checkpoints/5056/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5056/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3be6931bce055edec50b34ec98a0ffa4ac63aa606fd0844001259d52ef08f7f940aaf143e936b4f8356f715ccce7a6c704ff12aefb9fbe064aae141870476cc492c02b0ad042d25404614bf93c9e50aa9d271675a2f988545b8b1d0609788071cc559d836bc8bc4d3dc15046f5dda2c119294336d303fc079f0012a1791959f7da220ef3e39132494af446c61185b2ebd5055097ce12f8760d463b7a9180791e2ffc405ee0c21b5d16bb62db3fd12c801b0408382e662c95c241a626491badd0ac856c18079b49f2fae83ddb7657dd3b70bd01442a012b623906463b22e052122cca76adfbf594ee8cbb66f1ba29c52b87b55b3fd26fb3f4931f107a23007356&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520338753Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5568/default/d, Name: checkpoints/5568/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5568/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=be5d751cda836fac681af13b35026041003ea8e3bda9abfb1c617472eacdaa3fdd019cdca9701719f7d2b86a99503c07ba8e202a12ec1dd9d6ed2757ca9dc67c25634b98244a135ed7dd0bf49e5e6aa2b1d41d752ecd454a92148a9da796f4f11c32987106ff5e294291c96cd031e55993a7a61ce401953be6ad082a92c2bc47a9aae1f1b269116e10699228bf6ce3e101d9d15a5b0e1606c1af9a1aecc775ba7f4cb16d9c37a415a07f6050e14adf69b6d65ad1237a47e308127b52ed701b4d00f3b20514b61ed58d10db38ca4d293317edff0e1496483af6b8624c9574cf32fecd0d4e125525cd4db747eaf09179e79ed52107a8e6939e3341a7f35565f422&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520309967Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160, Name: checkpoints/4160, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4160?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=58b3d145c1245f5b6ef5e5a3396ecdf15f6affeb5ec6121261d11fcbdfbc903fd7fb22c849ebaaafab6c1c0bb589b56aa159aff612740f8198b5c5279c2daf1c22978f6ceaf16bda04897fd76aa3d1a831a58cd1652881f190efa6548415ca48045a440b536950b4064d6a7a58101fce44e470ac25c05de418c976fbe61d58df00dc1d7813e581bd2aba0f43913434ae050bf3aafa2f5cb58ac6b1f2964ead1079d9e74bac98be485b43ae32e302c0bff95f13326af03187c810bd214717c9fe4a2a20212395691af902d1b1ea65cbfe6d586379c0a808a8e2cad4f7faf08bcceb401051a6daa0589b679db3bb93432c1c5efac5b3669013d480dc026d420ccd&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520343824Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3136/default/array_metadatas, Name: checkpoints/3136/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3136/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5a0af2e6d32f145ebbf3bb8ae56a73f386c15a7eb73c987af7d9ffb5bcf36127bf1c5614690cb760c0d824b1e5ed2d8d676b4ed400ceb95968d8b9fcf791c6aa511af2bd2e668b6b3003bb9289ae00494995ee3da0c8ae371f349c8acf235c24c94660d7e2aeb5e0c5bada1853a4ca5e7e36b62652fccf5cced91e811eb6436b8c2bfe178bd011f41598b369b616f2cd19c6723d88fd1d61303fa79ae22e81225c782110598edf4310c1e9af34c1527c7c13d6f365ef820bd63e359fd472129638aca1e60c81b8b442da4e7f45bf1fa9dcf55887b97e6e7129f014637e4a7e3e22c413e799733db6fb1b8bb6349766c08de70c6c42ec153b02025051e6d5c456&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520365167Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4800/default/array_metadatas, Name: checkpoints/4800/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4800/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9248a928722f2240b16b48e38f33fc0e3d1e2bd10115d5b1c8608aa08c8141016ae6fdcdc03dbbf03d2836806a84f1e80539a262c38775d40adb0ed030953bc7f2155f8e638e812bb113bcb0b9d95f519677debed81a1bce2566f3b40dce936276b80ac8355c4bf9b7bd7d695a13841274d3968582b9d3089436838f4dcb45091da1c5469a387f4b37c03138b1f6aa196b76c24decbdff4888e364a833f6b6d69be89c7ad391536075772b2acf3a207fbcf9e4a7e4d46faf98fe3414553f6174f4feea8b05489c391d204cf543d7f0a9ae2f5db2bb4ad7d89a475109d45f1dfff132ccb03660ecf2b2fd56955f62f52183e60d02dfbf43ac3b7cf75b500fef1d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520383838Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default, Name: checkpoints/1088/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1088/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=2329f77c6ce9bcbc8a7be4ec9f25fb2d82605c5c72e3ef083d38dc5d036315b19583fc208a9fc42b2465d6e4bd298ff5c9e4a8ff62e4148fcb9b184708913882a438fac9d5e48ec6bb212b34f1dfe897a76bce9f3007ca8c39133769c16973b0abcc307211419f601a0522ecf652dbb8b55ddb32e666549ffc715dd6f9322b37336349c3d31fa8e97d3933225c2cd21ef6e559c665d314426d994693c864fae458dc0f1a0540fc9bd4ce8074580f31199469f86aad7d05f9ba3579b2136fc8944960c422b32b979d4e01c14e76afc57fcbfcf6b83d8939dec6f5a72414c59e26a86566f0f890c22fe9c4d19f4181860b5f089eb44e00e52f7c439e9f365d6bb4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520395539Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default, Name: checkpoints/256/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/256/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=72e6f79367279f213a0e338ee4124ca2ceb61801c3b9f93812b6527ac46b8caaaae731c8423b1d678cb6bd20c29dee5dd2d3dfaaf5d768a6e295865e5371d5e46946f84613a5fc6a50eee880066e166e3ec1b670fa4dc91b3b84e19b309eaad029500c33c2b0f319a4a21b597adca63739f9fa78f466e92201568875f7358ac36099fd94be163ef4a8a3ac21a8571a90c5942a1a97745a5c6830336441d9996a5726c0668cc840522d6f4f9a3294de34ca6727d1c968f7bf78b8a714a3da09cdbb4dcb310bb49588df5c15293358e74267a54dcf0e1e3ebc4625bc01c4ea84eda70bd9594d594f1e89dc1576e5c57c90194ac4d41313d04476d0fe19133e5548&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.5204106Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2752/default/ocdbt.process_0, Name: checkpoints/2752/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2752/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a228235ce1d16ba74c0d84a46f8e4bc4d9517daf7ee81809f61f1f2ec889e55573828bcfb48d18017be84ed16756e9ed7524ec38b720891f1f3aa8bd9961a762fd769cfda5322d2266ddd4d0ca040b66ab3eaac06d5f8bd310ba2668664bb12c04282fe4d623fd34fcce825d33337877573374fc6c19eab4c16faf87ac087b788beb391891b68353bc1fbda11b2fede5afd85a35ffc910510f79c5a06316ecdb38d574b7315eb19d3f1ed200da0a60884e6694e3a05e9889345bd3d43a11dc27bcbe4b8d1a8560b1daede88c01d1e5da143be5b62b59c57d77e334a8f65d370ff7a08703638b9456d3639151ab8c0b2d1c98e4fa066aeb05e06b67b1ac541d88&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520475745Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5184/default/array_metadatas, Name: checkpoints/5184/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5184/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6bd59baafdabe4db37580684c9f27c512f0811affc9697a5f1fc36af8aa8ec01dc0ecd218c647396621442064d8bb25d14d8937986c8e84b31bd671dea9c76688a6d91b9c0ee8b37f385b0ebd6e8efabd1fefe3a3bd7ef9c8ff5beccfd8c2dc34597f5e0708d3559ac70409a73938deb8b5ec87c7dc3af638851e64e57f098b1ae5e4f97c95264ba9ca9711c051ac2c3b9b4a371bd05f2987c77a4aec655610ce5932e569530d4d8350534ef85829c1b80301f5b8607c7b6af07a756d93062d886bcb9252e04957cab6cbcf07dd350e010cd29fa6fed8d2d04765e30f7535dc01ed9079a27b13a1e404d20b6d46162dcd7a3d014e18887ee5faf24523291942b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520441638Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/d, Name: checkpoints/576/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/576/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7e01f479897ce5a83f8bdca723c5aed9c98c11c4a44c01b74a5a5cba2dbcf5eeb3a6fd145c16e38f9a5b9b8ba69a4548421725fad34beafc70d46bff8b4042227246510bb1f6a4da085aa2f157008a34c91c315b6553414baf60b3a473f6a3ea02036d37bb872b8f10c1c3f122eaaae935641e990cc0aafd029f3caefcfd84658ad17f19e5e3ae2ff9f81458b18d782204106b9ca2ee23ac3c7f85f5986587896ebd7de2e15fe7927fc2906837dc9906164b74e9b15a5c83ee59b8c3780c20e836e1c8a3d95d14551b5a8e4fc1fa8457d3339953c0094a755ed2f34c790a41d17482941f90f864fc99c2d0743d4c941f710b21b5774eebd14b974179075509a0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520484469Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624/default, Name: checkpoints/2624/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2624/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7ca77243e07f479ebbf5379ae72a7b5c5efb7848d392f8131b69d6ef065bd886a112972d13f695050c4f44386f9d6b0546ce1c063895eb7ffbf89aa0bdb05551bf03220ca8f88630912355931b0bbfcc6e1addc20379310fa0689d53a992dbdbd3e8bcdd6ec117368aa4068ce4508085a3dddc4e7e8df3860b26fbf9b7177165c75f76f97b74ce95ea14d352d79f6c7bd939de201fcfdfb67228fa6ae5c9d4b1e92363ffc390625094706d8978a8676aafe5a94ca2190f1f318a961924968f6bf23ed48ea761bc1d9d4f8199aaf4ffcc0359ddecf3ee25e7aca15eb2f14cb40fb37c520bfd5a99accce83680af89c88064a46dcc0b02df10f63699effcf31ec0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520475037Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default, Name: checkpoints/5504/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5504/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=058408dcb9d24b0d38cc5619398fc280fe2c2bfc8203c3e573d4a60816b0e3ad1b21c8664eebabf913f7b5b7ee4a10b9318c905fa26d505bc6cdfcc83cb929c46bff15be6a5c2dd53ad6c1bb6dfafaf3614bf7d8d0dca7f3cbac0b1a3d7472569ee5f2a320771447a9ed9e04332de8e245a26ab42e0265c6330a1ab9d00ab1552f6568163f602134f424fc3bc6ecaef50a012b6b30d346d3c63a40ffca6756b47671cbd7ac6aa89667a4b58ebdaa1bca63557f6c752a07eba31641e5e21b140c5208029758a94f229ca71c0abc46a83e5de813c6493834947f05c2ea8df6c136feb1afcd1e760a765a313dbcbf3bfce449df9a67a2f5cc7f126702418ca36981&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520524913Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1920, Name: checkpoints/1920, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1920?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=27b801cf6285bb307b518f804688882af6e55b99647d6f6763b03842371fb4de786da82ddadb2efbb3d9be527ac4374b5f4c59de384a542a668ae69821d76bde1c257c4cbdce2124d1a287b6f7d4d5f624843c1ef8863e29908226918e1552d76649a5f930e57196a285ccd0cfbe3faba7918413d1041ec79a4dbe1bc3d35855430ea26c20053279458d7c36f5077b130e95be7946ab0fa654c38d60e0a751159327c8ea4bf9b4b306885669fa626d7b66f034b400716da46902f73ac2ca7804a2de32961023acc41493c8042cee2af37c115684e8b93bde4b69b6bb1ad9f71433b91f87bb6ca9208de5df7df12687fd032923d8e5f46c2391cb9f7a0fda1884&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520566402Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1216/default, Name: checkpoints/1216/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1216/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=579a5ef8c99f385b49efd559fbe91d466a53822036add7525ff12a0eaecb95dbc45bf430f174fc012a7586ea141c58455a818b9f02948c5357533059882f87c72e2735cecd27e66b5a01e1b801d9e0ab51e6c7609b8f8ed129b22db45ad5a488f728bd8cc210a76bf5bdcac0bcc614e6bf1bcb3804b67f17d722a21b76d72c66e3fe171cf70b6b302af890b4aa7a62b550abe57752fb197463c16abce634adb059881fde34c002e90cc1bc84292df6264a209756dc53cc3d963a7a4c2e79d84a6ca7946758f36948b508ea952f44deac00572924d36d651c4c3550815ab1a44222ab5ca38b204f398023f59aec83b4c9aabbe30ce508c9ecdf72fb75b11cf4c3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520602184Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4736/default/d, Name: checkpoints/4736/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4736/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=262aeaace4c8072ac724817c1eff75deab76eea02328d21e3afa9ceaec718a1772330774542ab1b98112d418a9c26eb289538375ee640d62fbdccd1c600d54ab89a8b740bd0c865d260d072fd4f88c878b6ae5d20b7cd7f86ae69f1af99be3cbee73ea6fe8dad28c5e1a3f4b04e6674b2ab285e9d5c88b69c2c36d17d8068401a4097670f6efe98e729ab6eff257dc8408f123814bb2fc615f4581c3d96ef97b22158636f0ac51a075be4a64c066148cec7aed38a395900b8f2f1559a218e252ac388afe7b4c4fb9d47fbc67d0498830935fa8de9fa1daea784ad7902c2cd14cec651b84c5c16906e135f163f2aa958caf205be3195fa17352f4b7142c0c72af&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520614145Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1280/default/ocdbt.process_0/d, Name: checkpoints/1280/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1280/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7a13b169ac1326b59c600d5acb3531c94bd047986e698470366366c650b42f9b563427045ef27e3fa2bd8b9e54b34312e96525e75a79254791c6965a12b24c58ad2857f08c76cb1633dcad740f9abdceabef380d7905281936e097e8d74f9b8156a6f0a3cdae8ea0d022d52963ef57e69bf188f04b02ca297351d1b2d407ef4cb08fbf5012fbb364e58d601b0428d2c662d78a7a2d534598c1619eaf0f87734e2ff62c3072815959f813094e965817b064dab78f471e5b7e0f52fabcaad2b74c855358c51eff358a7276ce91f7f6b8aa25d20b3e9f4b15c14d04ea021f6b079d6751b92bd6d576c7cd6004e755f9fc709678dfdbca3dfcf743145d22806a0e7b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520625559Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5760/default, Name: checkpoints/5760/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5760/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=972a95744988ae3cf8d1d41c3f1fce77f9431d67f7d34f118a2d53fa9a553bbc26419d2c71537c39f100277103cca1f97a78851d947778b57eef94eac4ed1399b7a97037f5a1f5d5253ed22f24230e2d1f0ed697f5764df90e432e753f574f4a4316a3f67ef494928b45202a96831cbcda41a6912af1cd18ae7c34c591f554267e20b7c76df84f2aef7f7970898d06f190343797ccfdac79774614c333f1baae357fa5c0eda4b1704ca4d16d401afbe2ebdd75a3d808db1aa06d8ca1eabb62f503acbc06d7880bdde77483613891e40500cea19c593ee0e3e0806100b5cdd1c7278bb569ba882f9c72f4c29d714ba22afd942d5da959bf15d12539ebdb059980&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520590507Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4992/default/array_metadatas, Name: checkpoints/4992/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4992/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a37714c234e8f410880ed81f72a972346fd301a51523dd0f76e84b47045fc4db70a63aa8eef9938f2a62dc28a4569d2875f1fd6fbbe26c00970a5728621b73cf9bd26fc61db0915eb1b71db61d853137bffbb66d238fbca9656b09658c51ac0ff37f7f27c0db56e7ef637cbcb001ab54a76437a2b81833906c1981036ca5e9e2f695e900bd5081ff34ce7147760c9a66fd40d79312763f2c531c90403b1e31fb682b7398d2c420c0e09777d25298aa47a7292584473940cbc64d31264c60d8bc252c783927078bb5722ab0cb2e5ca34671f88716a092f16dc00b4f6b751387532cbcf7df7231dd012c9df94c25b8769ecaf1a21b8e45d1805d4e1c4167d98fef&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520614964Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/512/default, Name: checkpoints/512/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/512/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=ada14c2696840b850e236c37ef46939bf09001b01602b7e2c726548cf4cc3fafdf9e2c967b9e1a9c9c2a6d76d1fb907b5405cdcecfce314806e6fd7411ad5f8fb90e97979703c9b1281c559185442f51a53bcd83b15a9cafe3a043b321339cc58dd74f0def504ce38b100143002d174ab4c7a49f5f663a518dfe838af46b98e25200caf78cccd6d47258ba7cc09cdc7b140c2a906a153b90cc92760658c3f15d1e2cea6184ad91bb72b8e17518414fda32dfd9491991ac8795a9d41123d4ff18118637dcbf4b3c73dc1fc8ac3d7e500b52da2967e050fcc3b0a776b874536e4bd6bf5b759c9e35c759b006da2f3f8a121c56f04ed56e327a7dea3395e76d0e51&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520682223Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5504/default/d, Name: checkpoints/5504/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5504/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=33adba019eeead41e5a704d0e79187591759f9f71a5a054b0a5aa22c93846d4dfb2ed49db1a7c1bcd602fab97ca945d362d14067a7646db312677f286d7ef81b53a77a53764b5df2a3392c6936d8fd2c0cfb12b3a8212ba09732d896f95a8771ec37905cd3835bd61563a1decd8c0908aec40a39e4c2cfb848db989f05c04871f22976f1beb06846dac6770852d9f25d4f7bd3ead7a87faf599c333ea0d744aa5c50515475410d3299ba60d266dfda86c99099425ba6b58b37ceea2fdd445b6915603fdd65d741a9cf3f85a620abaf52a2d4ad22396852a7de4950ef8a200af686e6318dbd6706cc8bbeb884eff1399073c6f0077b45ced518944ad3278e7794&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520678614Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default/d, Name: checkpoints/2304/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2304/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8db0647c4d11000ef810969f5ca7876c63f3031092fed4bb20d36d047d3ebb5ffb013a199636215427bd6ea771710b544f1a38a4c9cbf536737e8c5eefa763d141312a5a038bc50555a81b4dde6847e2c7faefbcd15a964e2d7cc9f98d28c2304daab477ca90f544121c1f6e9888c065098e0353c700fdd6db70b479d9e0041cdad57e5d43655e6d091c312e4968b7fb57287db8024bd45d2a0053ca1c73c7ca8d709ecf5612084a1a2e27203747e5d9b13d410d8d12a9eeeb278dc9501825fd8172f714ab69f8014e1bf5255d86781ca7e56a4abb56166dce41e1fd06580e44af249d915ad48b0d3aa89cc8cab10826ff1a1518d55b8251b9acc67c70a65844&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520720366Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032, Name: checkpoints/4032, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4032?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c2435fab7036b125b6e7fe0e0e1d701fe4e1f3b94b2b4251e06f89a063757bf53aa1b1cc21678f13281b40a732c9bc01e36e0c559b5f2ec65d9ad04ff04975360001f85507200c8028f132da22ea2ca84fec37aeafe38f62afca78b20eabaa39cd30bdbb723355586bffda85777408a995b6570bbe11681e3e1a6e6d77d72212fabeda84c86dde919baa405737cbf2ca8db07f499dd6ce200d1644c72ede862b228072fe249aaeb4a8a6bcc54320263a05eb578e62ca9bff3f62bba911b6d1944bf8a40656c745653665349881896e8a07d884960e683a4e7b46087818226f829899ae3a17eac6ea44f8f9181e28dda22f04458cfc7f5940474a904c5d073976&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520731442Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416, Name: checkpoints/4416, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4416?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3b564b4e91cc3ecf7631f765201bfe5c9c97cca7867b7fc5104d6eb6bfee8ed1a0f2d538489ef08a858c24161e8ad9958f577cdc9dbe5fce782251a64313c3f1a6af14eaa47050f8b8abe2f522daf284d0c2b2bf9d17b55a48a948fbc2366f6f529562925e725c97f8d076af5d0b575e27e30a0ca6007e9629d1b099f25959372698b1049111ce697585afc43e33df76a20039ecaa50aa5b1af8a51008a4cb8f47e7a005cfc4118c2d1caaa09755d8e1b038d98e7b71d8ad5fa7f7d87b39c3c23f9210a10cd841ca9255490240c93ea3611caf44077e5fc8e520cffcbce7b011e06a90e8cd62fedb3f8f088ddf4b2499b33d341f310818d1921e5a858159a8e1&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520737098Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/d, Name: checkpoints/2816/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2816/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8bc6978147274f9de72d29016d72de5de9420f65bfe5cdad5a00ac65d0292ae5eb3da6d05fc01b4590bd3f663ae3d11e8eaac0851ae375540fab302ba2251a369ec44f0c92195a6869be70a2bde1c1c7c50c64e04c6257a3d331ccd0f100c495e5acd6d69e09fb092d2e87fdac105c0ba59cc92cf8f069cfa14349578f005fb76f5a3bc0fb350bb016faa65557899b438dbcc5ecb743646936edc849feec45806dae20ab71e53b9b222383779dd994f2d864d2ce66c18faf8715cfd060db281813512588f21035aa6010fcd6807c0ceb65b0d34f9726866600e7c8eb452a8ddeb9ad9bb97d6d7b9c23177678d95b6ca834cb89648dae490aa03cc8bc0910b3bb&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.52077998Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2816/default/array_metadatas, Name: checkpoints/2816/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2816/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=ad703a022ad0e8b5bb591dad8a97e9d9682de395d1e975ce519d8df8b79735f889c386db94f9a24dd69189c44afde4c28c2c6425f55c065bff1de79c58f5421f57086fd1ad41005a2bf84b7d160bf43d76500c9f0fe860ebb756d3f3168cef639ad4a45a972218d4690a9aeec81ca9ce5f3e8a4f842b8bdf1b47217d70339cf9cc57b4387e8ca9e1f796ac0fc792734fd99c99a5554bdcf6dab149f01bbd6f2d82b0edf8fa6ea27909410e68f0840e325759b575f32c8ae63989cb9bc71f4cb295220413b58efafb5847acb8e174c5e3137bc4c7a1a3a9699417c5fc3e312cf5a2d5d642c32c2f73fe38295f5269e91b7cfedb04876f53c8db375c139a5bbb38&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520795958Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2304/default, Name: checkpoints/2304/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2304/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b29bcd992712c5f8cbcbc347ccfcf9ef8c78b3d0936a1b41dc06c98dc3615369928637da9bf014258fc96711085e00b1654e9040a4e11b5cfb11697b04d02f26900b5b7890f5a9f753c6bb0f05430c4ea53fb4a3c4ef4f1de19e859dcac6f956b8bd71b32e338e64999770811b9b5ac6a9cf0599b8a3c224313e61bd2f625744f3f905d157e662db60ea60d20cac8f132d99871ed99fcdd2b15f6a445d5f2eb81a45e23a3250bb0866f8826f46ffa8221ff951b48ca850906973977b37abbead008776b204a5119d4b4d59dd77707dd82c388d9acb18d51a0c164f5e18cd1fffc62a0e3a3179bd528b6414ce38c72ef618c6e20f2a9ecf31daff8b7c9650804d&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520747226Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3648, Name: checkpoints/3648, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3648?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=7f7368919955a9b714d77f70480daadbac2e1e4215856669553b8278ae10ba797b8163de298ebb6e35c9cb241a73d297f559b05bb2ebb76add88f13950918840c542754eb08a65ddf6e4e00f8416492f0babf9f79b5632dcccad70af0401e2f4380ee32486382ae96279c0bd3e1ee85bbee27160dc204d612cf1c5f92f43312cbb90da42819c9c666897c9812c03110abf40d57f56ba48b8bb483b68969dcdfb608d66b9f8ca63aed44e26df4402839e7e78b66c361de611cc4e33bac87085a8a6b6f1c8ed81d708f20cee15ad42e11752aa7639d94cf9a94afff074e5622a55bff13a88ffb678f9ee10d07ede77249ea6cd018f013ff9f17b9f7bff9a0a3a16&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520808134Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4032/default/ocdbt.process_0/d, Name: checkpoints/4032/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4032/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4d5615e40d2490d9df8e8cf33c4e31b8eb2efc8d853d77e486b9fefa4bc55a0f9d645d043e35d6700ab83a27129bef04023923daa4e391697f37138304f877349408d6ce68c7018c6eb1b7fa5e84f6a04d91479b823aeb2d2a25547afb1bd1d43f18516a5dbc9d1d3e0afb86c21b823d3adfd2175aa3363d4014fe8df2aee020cdd566ece060cafac8570a8ccf1ea46c85cb243b694496f250f70b76cae293fb1865cecf158028b1000098d13de3f15b4c9e5f3e1f756222aa9a8e4eb30425a1122a1ec8b1d3e2c9995ca0b6976f252baea2340a2bae0d9b60ea13d5b3d6b04d8bd7eb349d772ce299c61666469dc126c37cc678bfbceb69f2fe99276a9a7f8e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520845718Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480, Name: checkpoints/4480, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4480?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8adbd5ae61a81090584326fdd0f778da4487829224943f9daa734aabd53d59de7fd7f674e46abac6a076a4b1c2aa0ea3ed0811c8a64b6c5bcc91bb9fed5a5d5285ecdb50db9a375e417789a8db8269843383c73e778dfda077a940014d5a4390438c33f834d821b170b8821dcd09eb8bc3347a24131c57b733f96e5c94fa989a8c47b8d7df9e1f325fe037e99efc8c0e19f5000c3a93e27bb02f7298be7c1aeedadeb154fcc8e954965847096ecc547fa17eee051f91c1a737ca9dc409df385895cf7464133adc2b9d7e2991081588f12c20ff0fc17f79841754cc4749c2bd4beefb9068bbdd6295eacdc01c3ca21d5c0acf1b20608f2d25c41ba1f1e32f414e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520841812Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/576/default/ocdbt.process_0, Name: checkpoints/576/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/576/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8cd065f0675b901c4dad8092c2fe9aaa3b9fdf0d6cd80ba7d4d075fae4f70c6c480cee3af96730ab1dc8cabe6102155acd1dd57a954a97d1af0d9218eb8ff43e4494b2528a67371757082ac66300d722838d02e05daf3a78dcdf7599403a6f31423f767faec56fddb1fe76f58812c4c87447f423e87d47702e07ef094165d5714c3729b2ea04993a95415e4835ef615d54e1b3aa82eca5ff4934246c51df412fb345f34f758c1f2532e5485d75c637189400958fd9fda328b6811a894460c4216e970166ebd070a9767b8ecb3dd9c9750c604e2f27efb5e47c5c6fd0604d07b874c42857d0c79a32de1f31ff3614e1a11d539b1cb5e81c8e2920af51256920a7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520841799Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904/default/ocdbt.process_0/d, Name: checkpoints/3904/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3904/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=5daaf01dc34c41933ce73aced3ef86d87c8317109218879162fe8b34cbc26405b7dae0f9ccf706067577e646196e35b7e38dafdd730c0a13bf99515fe8febcea4b4b4024d85fc185a184ae6d7f87b66e834401d6eca597f716761ee2650dbad7f7086ec7b68471ddfe98db506557f34500b92d9c35a202e0ec3836074de4ed0c6d4604152fbc9c83f6d242fa01cd8aaf304fdc4393a498c7563f6750fa9ec9669220928c12b8286a3e046d2822b86e31d166fc21a14acc798f9be7ed97706b0a9bf4ac0dbf966d3dcbc51e2a7a9a7e24eec70a70581706faf1cb49a03d2ed68c35f220322e7102ab727ae46740d70a0adaa3cd58acdea63d0683784ac9b38adc&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520864924Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5824/default/ocdbt.process_0, Name: checkpoints/5824/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5824/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0c122c380c4e323532014f04457aa0c7d9aa9642dfa7112de61d7516793dcf039e336948faf64011ea0468b4493d65f41379a229968278f66eace366f80aaa9cd61136e24a23d2653eb9115728e4245a01d25662d9102420b5c1a81ee6540f79e705175d636f562c9e2989e2f9e54ca1d98e5a72558d8c39f45e98911a245a14b5251a09a6e9a04b86d901259c3e18630de0a4cd0389b678fe3e99912d356b343d59968eb96863a75f18e479ce5a49a262bcc6517f271992eb09110caf62b8c58f2be49dd9b1232454e6ea49bb36729f4533e0dba34dfa4e35ec026b52b9cff15d1573d9f6459045b2951959c47cebaeade09e0d8a844a1511669b82e02ba235&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520855613Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2624, Name: checkpoints/2624, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2624?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=932dd37a26ffb9f817200770bf0817f89b85123331255278b8f736b4b67d5ffd7dcd5f50aac176bfd510f49f403cdec463df913082aee9c0d76fcddc9d5e1e5e5e0415daf054ea9c09613a7796d2321e52fefc75fb5cdeef38b859d5e027859e1562d8d6488322a69a395f7673b320751e7654a1c0c37ceae689b30ae98afa21e11f66f8b42992870835afc9791b71587a0d11678af60100617c3b6d2ee24dd4483d4f04e5c8aad3e18d7139ed5f69b0780a374fb950afc3dbfcf60d080c3bc6dff9444fd91c31b8d7c1081a73f353c290445497c77d47e431c62f4abec5071f6be7ab8d1037c6a79504fb19d655e94764b69a79066b074917e9afb16fa46df9&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.520889589Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3456/default/d, Name: checkpoints/3456/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3456/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=184b430b83ef989dd7ee624fd0419fe65ccb99deed9a34e1bc773b63e3a624299e23486393e69abdc2fa8ac31b90ffb350481fe9247d68a6fe880d043e96c2c409a546c4171a30a2dd7ea49462cc925aeb32e5bc7e3764238228e48efac9504e18e9b0db6a9ea55dff80f512744707cfaab6b4ed9acabc41a45bf7168798998ce145dd45741f9cd09cb45a493c6ac6c2d4c185db7fbb69637cb9a824cda18308b47dce9fe240962c8146a70c74ec730d98a408a9e41a31ac7e74ecdbcb1430616544f2e3ae6e8ad8e5d176448184a92c5b68129d28e96c4c06b02bb9a38bd9df1c49ce415d26f0757c6f70cc0a8cfe7716f8a4cc52e36dd6e7aacf3f89aeb34e&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.52091135Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/d, Name: checkpoints/320/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/320/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4b77194bd8adc5fa622df90ecacbd7fe1650e6636381e9e2e8d1412c6d1cd2083eed86d1fb52c8406428863c203acd02f19d82dc58341f2df2ce45efb69dafb06bb3f45a82e449c3051ac014da6680659ba6e97ddc0fa4ddb6e1aa7c004677fdf9d6cbbb7cd6a2be39014e6806d319d3e6296059f35b100bb2dc6509d10983a49d13a1ae00abe9d50e4884df34a4666bb5e0fc6dddbfb49567c82626a44e7fa30b09144b9f918606d9d45ffbfd65ba1f4a8087d58d37a59e5dab20b8735d637b3b1b98dd01a444f46dabfd30f7b588ea14a3c1929754305fd988a42cd13cefd9c6214c0f328da9b5ebdcd0aa354ade211bb2724bc9f391e95022676907247ec3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.52090865Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/320/default/ocdbt.process_0/d, Name: checkpoints/320/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/320/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=c192a3bf2ad10f504fe8ab60c4f0845e2853ef8c1e0318123496d0b21ba511e6bb439e96d1cc3c5e52d94bac07469530b19e2c195817c910a1bc49d17faeea961c711e5a24a10534f2b39d89f37330685b9e57e7ec5ee1ae58fa0bc54badf4bfd0ade47fa5562dbf14e01409b8b7ee37a9217dab6197f3c069f684ceb9408be0a4ab244f6772dade2a8c6c2f6e7edcaedfa5db6b007fd6ccef46c20a249957a8f7dacfb6123ca726dfe25d6c569bb29525e40aff27b4cc2846f43a20bcee69d8cb6e8309ca0a9567052e469e07d16926b5d2296733e6e2ca38d796e43ddfc0598f2365495c50c3107b90afba99ece465b92068097d9a36ca05f6c9e72a5364da&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521040385Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1088/default/array_metadatas, Name: checkpoints/1088/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1088/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8e02cd47a20fa64eed94d6179f9bac39033e0b0294a93d42f3de273bab7c3e9d58b6fc131c55f336139f89d7e8132f622e2cd5d541d6f0aac713955200fcf88eddc5d8c1867f81e52a7a2121d16074090b0a63d2081700b31bda7ac6b8917f88e5397d1625326b3c7a482042c15cca8612497b60d9540eb666184215c2ca599e0a1c1e83d8aeea4e6c3400510599c31bce2d8a777990ef8a2392f68b5f31ff66c3c849fdf782ee5493d04bc5fd01a26742b0fbd9e805da8b86fa27e3ae87569156f6036e4a5582a577663579c0e1e10c427510cbc92f4689a88a7fcca8f1dcacd34687e7812a2b71e934487b0e5eb18d3419f900dc84d91dd317ba6acd495146&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.52105705Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1984/default/d, Name: checkpoints/1984/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1984/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=64911880bf91f48120e117bd10a0b713f64785a58f4856ee8b8d37df1e5648df9f9528b168e61f64103dc977f49ad1a5dddad1e517740553c5dd167e13a0012a254d2f920edb7e55d9a60bd9717f2491a7d716ce9755a0070382d11a0e8d909197c3fd5d1089dc752d03eab668c9cc0d25edde1009e7355612ff3aa6d01bf3f216bea41e417e9b8524e525bc96310f745e6e672b83ddad2c25b01525fbd9a8421e8563b82c7293d53723753de21e0395e0d65dfab1f02738331560ebed3a7e772109ac6d100e39abe50660cb66dcb7b1da7aee9641a279d3ee8f37cdced62be53c7fcf260141a400a8bbfd9db1e3f9fd6a9e9957bca4aeeb0f4c702cbd55531f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521057157Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5056/default, Name: checkpoints/5056/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5056/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9000967cc1f75287e09d02a6a68bb9fee3145a3666a6bc49b4fa3e34c6be87c202367ea0219f3747fe7bc02602a0f005e0121a78107d05b100c9c108e06b8edcb025f77ec32d9cf8b14d0f8c1ee456ef5cc8f9e2c33b0efa33dc885bd96d270c40c0f71a65141c66c8dddeb3ff1b0308bf0927c5210101def84eab47eeddadd915d6cbb2d9a375762447440e48f50075b4c70d1efe40b6ccf78592fd37300d5be6f14794470b30e41990923e29e2707a40358eb3844b376bd2afaec12247ae403ec8795e45de61bfb016e5f271dbee7eb151c1a56c1309523121c6b647ba089c37d8edf5910a032317a75e34d2f28f5d9592a1e7c6e0790506afabb144c6a163&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.52106235Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints, Name: checkpoints, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=342a68c7f47f8c89058706a38bbac5c4adcfa27bff728c659df2a3c812915e14e392dbc348f9db06d52da274099230d5577cb988b6e38b5bb2c89fd82c576cef91772e87d3c702fec773540b55119f08750c775798e2ab64c2708d77e58202655515c906270d5debd5287561fe571a46d8278635fca96224e005b761dc17fc6c8b84bea60bb5add0b6fc5bccde9837aba98d6f83ff8b76086fe53a0c28fad3e291e933b37fea356885b78509bfbac191a72a023f7eab02b567b6dfac8df813a29b3fb1c1d484c09c745e04b2de02536c3c06cea15d41bb97bac71b055cb16cc2b85faff3abb191b8f522f0d3662c5aca2c1e661654873ea65f07ed8cdac6b2f4&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521099695Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/256/default/ocdbt.process_0, Name: checkpoints/256/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/256/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b212ae5c219d033c1c0c34332d88f36308dd0b96da56e73ac6b929f7c6d27706af4516cd8c79540d9db2f386e7031691d58dc9e39bb767c60a06d121f8ff9be3503526e08304fdd2969f48ae222f08d8b30f4ecb0d7bc7d8502a8be7dc81e879db99ac65b1a6e4de463d3677317c5201b0cd7ceb81ccf369799f44ed2a862ad04ea45158e94bcef7471229154891806a0c17b4abf968969558405a07d16ab37e110da5a96314d881cbd4510fa81a1f4764935ea95eeeb29af6a0b4adadbf4d47b4e2cdacef067e7de0aae4614cc749b395e132054461baf3e2fd897a771b6367ca314357dff816cfa7866514fb54e6c57b7dde1117f9574cda02ad817cb593a2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.52109653Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default/ocdbt.process_0, Name: checkpoints/1792/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1792/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=15ddf6cb77e8f06e684132114305c3432ce62879b2fc95b94a8ceea810cff52bb0356ccd047f4595500ad9fb29ffdc8d572d37975bdd778c306de974db9d68c2043c460905e759c077f39f914827b695c99db7c21180f219330c4101a6ebb331b5fcafd7c8f6749482485f9066641e0079a52f8086d1fd41bb9289f62da659c2469585870db51a10ceda0e6ac694023140ecb6605d8abb9daf0f5139fc515c90b2dcbea813fc3fbb733da2b0a9fc29102c81eb92dd015324eacd4ea400b055901a12bc5bc68feb9fc4810cecb4caf02dcde464e8e2eab46deca66a31e93562bbe4ffc819fbe582e9dfc9a6cad3b422f4b234a6d8b52aa964d4f8c2ecb3918e19&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521153963Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1536, Name: checkpoints/1536, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1536?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6fcd8097b9d34d562ff292c23387e3c8c122de839d171484024e2bb6a8aa782bbae8c02b62d3c833c2bbb740eba3bac219fb648c1d389081bee3ca44939bf3ac6293c22e87cb61dd2c951ff6e57b3e296ece96482d6ec6a3e4cbef7b720c018422d6d48a9120058b849916fc79b2390ddd837022970940b195eb67225abffdd64471a664f70a8a38f6ef1e446ff7e2f238080cdd4615a6239bdee35c5578d6d69a3cfcc526c4aaaacce2f01e3c9a13fe01af5269b7bf1dc944716440e3556dda62cd415b2dafea3891060f99783caffb465b11da30b4e6097de181f3a3762c7bf5ca3b3330a8f35306e59ecc1b221ae793b6c2a5f2a68b8032562de22647c58f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521160568Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5120/default/ocdbt.process_0, Name: checkpoints/5120/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5120/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b82c25ff053fe564b0857936767fc358bd34a5729b4a762123962d57bbdf49e55557de3c69135a37074480c9e844ac7d15a3ea2833032250fa1cbb0bda9bb60bfa6f088ba8fd7f5f2de76ead08e0315fe5ea1a7f587d11f44c7f4e8e8e3df0cff059969579277e7933df4aff37b7722e2a14daeb1dfb60a3a55fd0300797848b43940197a6442e85494546c06eff19f98179d38fdce6b51bba3cfa15c71262d685364ed11c1a64c67d8fdabc1b60829b01a2aeaa0d282b932ebdba50e88ba567191c69414ce1c9c94a79cb1eef7688fa564704e4e3988d1ef5d744af69e1df446cfb43097ea07514cebaba86474a797a03ae7c6f1ab0900395516183c89c06d5&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521202302Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/6016/default/ocdbt.process_0, Name: checkpoints/6016/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/6016/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=08b5cbc9df8a475db556376b7e164ff430ece8afcb90172cfbf760e34d24b9181494cefc0b1937c964c01dbb1a9ac81d61f19f0613ad0880703aa43e6d7b35128d873644bb1ea664ba79a4f54122c8413d8f5c1ca997eedb05516c98a5dd13e2fc2b4128870f9ecfc3152e91cbcb0c8529f98a312a1752204b472817bf946e26b013f3f9a5835bb1c2e4aa54ddf7c5c3236553aed2a036b5f35dfd21516fe824a82f219ed25ecdf8f71962c2cf242e5f3c8da9c212eb20a59f16fe0f2e44779e553b64a23ed54c64f6eebe9f08657ee877cf69f8fa2ff2b7e8f38bb7fac48567111a367000620c3309df206deb3626aff2b5b355e33e956d3880e0f366e91c68&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521223126Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default, Name: checkpoints/4864/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4864/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=392a7f86eba30841d06bc5ebde146323f9a6b0a819a8ff0a3510f30646161e133bdb0274f2cf8c7311b370851bb341b9bc5c2b25702aa5d5f9ff022b96343a88fcc53e5d8f963521fab9f39c44f008657b244a92f8da0dd2868dd0b93dc6c19528371850713ea197dd11a61e20a2ceb6fe925e2fa587a39beefbc15757d6343a1a453f654577979336c4536c0c1a94d1e20674980564df6ddf7b5b004db5cc4d035ba432c249e15ff645b4c9dcc14ae5e04ca0320a03ad94fe42522809b71d4160da88367793d9f9d486039c33f7027e2df8be6ebbb2772f7ebb13287fdab501fef453abcf15cccb043b86ffb5eb3b0d77f7baccdb29329e06a98ced3e0e17f3&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521232665Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2688/default/ocdbt.process_0, Name: checkpoints/2688/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2688/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9b4ad49d1add7fe22b73f2178ba3b9f54530f30fa0de084ce308fae5b8efea3727db21c8026d32cbe9a8651135fb029dbeca17032b88f64a0fbc207687c859ea6e50fa14aa4b9a7f29c9e79ed92d5d09fecb94f828a5cb55fdec4f65dd7c4f69e58878fb3fbdf586ee0dcd80bc05d2cbe6281da065f97935c4ce295566377348e5ae355d825799a4457d3679582546b997aabd88e75760693de81c269e356f7ef1c5872139815f1b3c99f1b42ff011a3ebdfbf5a845433e2456774646086777ff0e3db97d8fd021116aa0e84fc8dfc2bb4eb153c4e2c34301804b9920b88ba81235c246d84e116f1d4b3c715ac2bf660da03b2d5a76e1316b748c70019fb8619&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521241516Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4864/default/ocdbt.process_0, Name: checkpoints/4864/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4864/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=37ef823afaf29730d0a2d327e4036a78440eb3852736d03446e136899214543858c394799137cba1d8f5b229fb274be18412548374720287eee12c21386bc94017f1c8b3509a4bea9c72ff4d2e1bef1e29aea85ba912b254f2449bd2ae7e007c05c054a8c547ca453af152eb94f7011902ddcd0b74f39269250c587982bfe75e61f6718a2fcae54ab1712c4f81ccb5650076fb6918557bfa828227966a7269026acc195cd36f2804625690f5faa95222dbcb97496f747d28a60a0920e95ae8e4462622d4c9155d83323fb171201dd26c6f12e866fef8eb9ee154f9eef393750e4aa36a6f2ee3607dcce70f5db155daf6cd9478977b7424827f89f142456da030&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521239124Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4608/default/ocdbt.process_0/d, Name: checkpoints/4608/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4608/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0348a41918a261b167e3d22d99826b7736d5a686d81b062a97f52b6cb1fd70199f8b9ae27b85a23ebf9fe326c9370438d156f7399a856d72d4db3af86ea9df95df54f71f5a1684735606cc2c7f63a86ac0c8554dc12e50ca1889effa68424dcdc4dd5a502aebb641c126c76e6ca4fb73905f546309b17eeb7abe3ddd8fb2570bff8b2069d27af3794d914ba1c3656b3d610532d6514a3ac336f6dc8db0b376a0a0055166831614b23a5af9c1f5eb764498db12dd1de972dc7adde74b7703e32c213b3e6c8de0570a79f53a4a89638def91c74e606f06784cfb7d87602dedb9c0aefc8e29cf55438ec68b72808bcced263585bc8c5b2c0d828cba757fee711ba2&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521279125Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5632/default/d, Name: checkpoints/5632/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5632/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=a383513219a0e3ef9d76c37b71ff2f4d06b2446b4b1af424efe614dd88e6dfd2c9bf2cb92cfc972b2663a044f7ba696657d83b378371a561b4f1f712558d6c45b388355b49fa0c02153837e370162a86273bd7ce10d8eb14a0af09972f4a04a2ee0d93689335abd0b5777df085e2209c72cfd9b6e0494e58576257961eb6f45a32c979a91e090f9bc92d054c31513ff36c2b372e9d6d9002b53fab7bd4f0f9ed39b6cf735392a2d3e4d88dbcf1d93975914c784004df0e82938c9cda4d048d553784ec9bbb059eb18192f93ce6e2767c6642cad2711f264c63492d21c64a3e6b53559fd8ff187f1db3959a498ffd23dacbcba97b4fba2d08a74a60c36990b59c&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521289639Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default/array_metadatas, Name: checkpoints/1408/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1408/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6b1827b797b487e327860c3f984d23e85803d5dbc86d185f38f47387b7f2e53f9b6a1d69eb1bf6aaa8889736e050bf867adf684fe58b06b565b3ff390914dcbad6066888b0098b52fbd0dbf338d616aa1163b235aaae65b44ca7fc3735d0c76d05b84ee2af978a7d66be65da00219fdfef0e1469adf84ce63cc36c566dd870617babc0f2e569429062fe04f5196d3ac49943fd0fef5f562fc71c00af05e70000626908d61a55dcea6278bd5b40eaecbd70c72dd9ccae0eed74bda9f30d9eab2e35b6e76f81b35ea6bf4979192ef76551edb4e861e0b2bc942f78f26c579b5a5bc1d3c81bba3581890e9a04c96c9c4e97bd098f2b4ffef06959a4925fc9e2a57a&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521299212Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2112, Name: checkpoints/2112, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2112?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=accdace88deccc966dae7fbb8ba6b47c34666c62e2d6c8aacca06c32cb7baa98f13c6b5bc7afda514ca6e16c48ee3a191a30cdb5208a9ae5d194b81a11620811d0f83f43dae223415afbd319ee4382b4f839f94e30ebe2dcf156bc7886c21ef8b925f25f228d07470332f5c5cf9d7c75991ed26d4b588a5aad7817365c781f602aa846276a1cb4b208d509d369d2505c793b964836782c38c60646fe1d0461dd2d2ab611e697590c6b15c776c68c9090e15cf5ef67356be8645791691b6f906173584af11c1536b604f2ebd4fb9586a0e7675f056e0496763dc4e53ba75dc239942777bb6c81d9740c6a3f818fab4ebb84fc320993de90558397320505bce59b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521315633Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4416/default/d, Name: checkpoints/4416/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4416/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=232020313867d8f6676c851bf0b9824b5ac4a2f59ddd0b36f0d0b8e8a3cd30d8a02be5d41de9e666161de6351801fa70e5ef8ae129202753a590cd649d98922ea916e1c5afb64657e0b33af9c129fde4cc7e5bf4244a974d7b753992d199b482f86b95c051cd6e075e1f3129e118b66643ba36d9ca336f3649860487b59c08a6888e16fa0c1019dafa606076f282a7aeb7672e702e5b901ffec1ebe1610fc843543d35ea4a5ea655914020dc7f68cb3800b5c26271a2daa36b38dfb99e8dff05840d22e6d5fb896ed9deb005f4bf05869d0f0c59a7bd4a2c332ce1fa41a25259dd6f1c7e98b6e1845d7619c1bd444a279f7e746ef10a8ba4fd37dd41feb747b0&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521333232Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3008/default/ocdbt.process_0/d, Name: checkpoints/3008/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3008/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=8e124e5fafef4ad49a87fdf46fb048dfe8c3fe8c8d60a27b2cac843f21d61864df1c9ea19d1cef75afdfc43dafd4b96d4825802575755fd0daf52fcda4f968edfbfe661638ff6ebcdd4cb82ff78dd9fa551b6136c058a46dde02a354f107e1ea5dbdc79c917cf458266d4b3485671cad695d4e54364047213e62cac134dd73a50ef717150248336872c12803007f5e5592ebdd852095943683c59dad5ebcc011b4361c38f77e9614149388bc19eacfcb4ab7bfb4e969c63ddd53236bd57d45b09a1994709536e02e240c0777bec38669a0ad21aa74b055f4db58f500f17dcd49a4b1e5d79dc0d2193bc9dad1ca32818591db3eeda6535192aa0c6d6ec8dbcdec&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521320125Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1792/default, Name: checkpoints/1792/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1792/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=52424820a3ff8185aa35ddd5af5ab483840fcc8d7ef690820bdeffa12525a6d1a174b1475b2b7671b2e6ff372b9009a531d7c662cf692f67135726734bcd527c00df49626f165f934fb5e521e18b080f29c0919b487879df6a77d8fb1b6cbc7b7cd4f733aaefb3e864383b0aa6ea937a767b6f687b17cac6132412414316cbd7dfceafbbcf92a8dc652c2cc91d8aa8c9b47bcccd4d7f6cbe0a9f6515c7bedc28cb269e2400f98813efc9c5b141969799ecacedcf96d3f0931bc65d146854c22cdc8e1d77041ecfb0d5910f0047c067bfd2307c6a66f2943150d0df22559b5873b5342d99149a6e8129dff799aaf94018a3cf389aad66105563a68b425b86d11f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521342586Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328, Name: checkpoints/3328, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3328?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=4bb5a54112c0cc49049ddbf04af36de0596fac9f45325a629e3d001ca0759c8fe91f9ecb2dd9ddd26c4be9b68f46fbabeeeabb27ac8a50c70fbc00b97e2dc737b5834354aba2b9ccf0e468e966db15bfc59457031bbed990affc30c0ebf8b0dea8c52dae6f743d4e19764840cf7cffd8054b05729229c74a104f946a34bf863051ff5cdaf4020d3ac0a7dbce7aba53d41fc2d500795ba59f6ed11d200fdb1d9f8ce9b785ea7bb9ccc3a60637735ccb10cb87f23f2454e7e8166054103c04ad9508994c0a461a3d8623975ee1ceac1f9be6504e21d4e4144e10f12a8af93ec77c03c2ce234bca775f6fcd728f3d8fdd553ffd701b9232836322624e586894d402&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521346013Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4160/default/d, Name: checkpoints/4160/default/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4160/default/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=36e1b54dca53b1e0d2d00191d2674d7001dd178f66fadad50073159575ff2ffee0b2976b19f74564405cd2278e66ae05b06d17294a6fd92434186ee4a42dfaf2294053ec293c19b417daa9c69a84cb42dddf489ecd62d0f4084c4fa6cdc274a0524b64af30b1e1549c72efb47f77865d09aa18ea680a8051e757962575038fe053322c5a9f1407558fe42811e43277816ac0e68f80793e2b5a53d76548dff85ddc55b1ab2dfdb965bdf85ce5499ae43487849dc98de5d62b6e8bff3c0a294774425141c0d1816fd1a3b4443578220a850d080060d7d85e45face0ddc7bf9eba68fca1ea6ea3e2c0deecbb7d5d7480fe47205264240acb2f7e5dc69e9cdc3b280&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.52135632Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/2048/default/array_metadatas, Name: checkpoints/2048/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/2048/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b305aaff5142b0b581f6db11af809b6fe3c18191d0c7ee0f3c380be5c1d88a8ffd55625301adc2a47379c6cb998cd7688d856579c520d703e629925d597f9c78185a78690d4f64d50867d7132748139ee3acaac136584a1766425c174e0a83a7f744db014e6b01d091deb7d5b23f3aa1a4c974c6207dbbbc7c08c35254211bf93ce8c54e528c67bb25621ffe3a9be127cea737ac6e26fb776cf9a78e5c44d2e2155b5361f82475afc909981e3aae53ee1533f8e84ed23642b431692f57b6bdc7903c131f2203e0dac603a34a6b67ab8b6d239fd7843a834a3b242d45d1632b7d1656cbfc85dc99c48625705078f5ed55acde9bb82edb649f1a40aa1576c8f183&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521382895Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/128/default/ocdbt.process_0, Name: checkpoints/128/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/128/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0701a33e706da0b0d49dc733759f0f32a177d6d50d807fffd48248f70daec4e98a3670341023909ccedab89dbad7776c5e56ebb7c6f08656d7e108db0b81831caca73f837e5f5b0075925d6e53cc1b5b2f3e796c8df52b67cde6cdc759d7e5524f8360f6665ec4c943ceff5417dd3aa8c3d9a568e17e5b8e871f7fae4e155ea9ba87a28274462ae6a98e604ea683b6285f3b5dcdcc46e29586d839096a23a7e7ae9648d18606974ecf8be2e38b33f3cdfee78bc6d0d9cc023d8b424da223701ce33be5a47bcbb2e0bdacc005302a2e371f8cd53de5d8406aaaf85ef09b3b6412bbee6432198b07aa064212901ed0ec93e52da58f5073c493763fe4a155a6f26b&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521400992Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1408/default, Name: checkpoints/1408/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1408/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=9cb2ecd40792b957f2aa2daac37c1064e2874a31f8248694bb057a7196b7528c1e356faf7a09148dbf808f675963413f2ad1875add787622c3ae05ce966975be86e93e9bc2534692583fcc30c0eb5c1e3626acda091ea8dcec594426e8937f96c20ac37db33db24b89b2b6a30da1a078cda13d5ce4280f99b37779808b82cab58814f8b40d720e0c35dec18909808f849856c9a02e58ee4c1e24ba1edca81f184d47325f3e010a3b9ff444b1b9762b6a308a2828490f055d649f1c810f865c172008b12de72d9572ec795c1bdfd4db4d4e9c80d32a4c8e65943a98303a097d2b1450b278dc2eecab786ccd3699c02eebe87b81bd457b013e5614741563520944&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521437432Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4544/default/ocdbt.process_0, Name: checkpoints/4544/default/ocdbt.process_0, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4544/default/ocdbt.process_0?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=0bcf5eb4546c367259b4062345ad1f854f4c778293ff3da98bfcc2ec503380997d1b4e32fb7e86bb005a2263d699392f07cbb467fab4d0d89353bdd976195b383d86f72ede92020a67540628faf876114dee6921d3d20f95bac1322830790dfe499e2563e17c6def88b0b1f431f861a6202c962ac0c6002089425400fae96497d16d2b3affabafdcb77b0c8aa4cf4b8465280e5b2ab1351b6da6ad9949aa8d6c555b754fdca918cd1d573dcccbae6604015c6a40d9d3b555838dd450c8fb1389ea9322de77c47d545cafb359666b5bdd9e3c7ad3afbaf5215b0ab17c0d8ce319362dfae74a1403982a7819c71d2201938d93a677739793b135234e2046dbb28f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521428721Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/960/default/array_metadatas, Name: checkpoints/960/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/960/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=6b2650df8dfdb0e5c451c9697ecc7702c97b31f936d047aaccfcbc7c3e112a9d836f689d3a206009d8e02c441d443ca928d37a9061e5418b36f5d7ebf9d1ad87937b826dda13771b7a7ddbd6889a7452a4560e7ceb7eaaf4039929b3cbdc887e701a951cf23a1f42d1a95524c0655582eeb9944ab5c1cf72dbfdc486666be8e95404966019c10895d803935580d72e1ac269250b845aa60a9ba445bd47ea96847ef7284bb6f04cef3d2d7c058ce53512a3e0a1502027d627a609dfb3103eb671bd791732100c3233d221d8a63d8092bd49866da0ef43b8cf6b93d9ac3a608fa4fcc1949b41db4047b9a52d3566b0ffeb6b63bb0b23e3612faab61928afa5c200&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521440993Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3904, Name: checkpoints/3904, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3904?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=94057bad465db9c4ebe13a2f49ee5d4878f0af7047f1dd65db9c42d42f3327510bc2963e0d1834c4b1145ab45dad449fb5ec3feb0b33009da093f88faf13f49b9c0da03e1a848913119345826c412dd137ab1eae458083ecc7a9a6f19355e76ea8cd683590bc51d572af0bf4dfabe8896d082889c5d0132a9f49d8464cd8b87afdec32252ec79c469d5f8851aca5cf97e12d9ddad0a2f7a4b432fa0af5678348640dbaab45fd9c683b988968acee343b914c7b11dee76e92abd5e98a6ab625dfe2c2a583774e3d04cd303b7c0e62c7ad007f88a6106e96ca150fb9a2dcb5140cffb653788b83112c470644b2702012cbd4d29c0e75fb8efc08169a8b14271cc7&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521484504Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5696, Name: checkpoints/5696, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5696?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=92573f90c5034b6638f58b74d8f158df6d5650dfa4da9cc7102c7a008566db83bb46fa6500781a3e73aa5ab41a08831acffc9b61f044131f00ad42380b6cadf512ddb6e5ef356d8d8a14db250969112a4098080fd7b1abf389922d14c7a941c64de1a3bcd09a5dc5c4c1aa0233bb616c91a36d7320bb7f031e1f7a6512f0dd4009aa34d210b98cf5ec9e0c43cc2e01b03c721e2252b47205533984777b0f48d5294f4e829433f8084f60805d38db39cc08d83e0dc7cefcac0b3524e79abe3b022b0b7d0ca213eef59e151c2e8b444b0ac6e8d26f76bf52fef9dcaa33de491c41d8919aeb9954fa16306c4caa1676c8a2f5b104eda1059fbedbe4d2cd26dc040f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521507978Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/4480/default, Name: checkpoints/4480/default, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/4480/default?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=30e330effeca8481ac413664710bed3b284b4497ec1503bb3e033399488c086a53d11b511f213d110d482736218218e31f10fef01d499d3ac2a25e623fe9ca532d4b9e5810467a9fe0388ea850cd6a04f3f388849fb62a5321ad989aa2e90e1b96cd636eb8a178655d0d1bb361de10748d1a97e06614079c923ac0fdfe7f502c800513f1bcf11eef7b2148a2b5c9465e762aba0bca3e3e813a799e8d9e40b02aa94853a4e735f305e04bff2fefec39fb19f8a473708ff581f7cc23c5f16ff14760599c960ea04cea6f0b1688fa59eba7b047247d9d46ed89392cc2a278495118831092f3b9134b7ac35d8a3ce28ff518abe1a43b7f6c1b9153e8d1eba58d9074&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521521046Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/5312/default/ocdbt.process_0/d, Name: checkpoints/5312/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/5312/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=3dbf864a499c11ec3ba766b7d0d74f84922c397c9f85113ff4150ad29c88a1577fd3dee5e8b76a71445b599afb5b38afbac27feda74fc2b62d7ef354295f5eaedff4c7cf1dd3eb0fce3be7a0194f32a4ad7abfc789198379b1c5d254293e9b196c507afb1fcc51fe16c408758a7e20d5d7ef1f7975124d5b9f29acc09db0e024ef4a719eca54e2b955f33823f938ee256a2a6fb285e4476e777682e4759c03facfff81910a7dab3cc5dff38efafcc8bb7ff0c00866439c4524cefadfa1f43a47ae99376c10d0171b9c10ee05f84ad6cd59ad6b96f9cd4926f591e4e38ae4011e3041ba223f956ad10854681af85cf47ad08d10eb96e31341fbdefdf3f19b3b80&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.48323033Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/array_metadatas","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/3328/default/array_metadatas, Name: checkpoints/3328/default/array_metadatas, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/3328/default/array_metadatas?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=75739b942e7038d900743c95257b4d1426629339e228972f94bfb105944646e3fa6a32b16ec3cbcf9d07c36fe6fefa83280d48ee138edda480d3bc4c69226f6055606bb96393f058a1d7c84a2ffadeb36b6aea499721876d62bfbc3569cd541bbf529226261d3fd6a5b438fc4136825ac573cbf88743e6cb7eec6f96dfcff5935e4204f4d39cb790506dc7c77731d555836e3fc03d9fa9f36b7711e97825dd3f98cafe3303c8bc15cfa10b164bd953f9e27aeabe3663b16cdf608002981c73e7f5b25c115179da33093b0e2a6a05f1645fd420f1a3402ee129fe5ef28f7a7b536247ac3a55752a23be8e11cbff56cd5e09449a304d033d276efef8068198f876&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.521528507Z","level":"ERROR","msg":"file transfer: upload: cannot upload directory /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d","task":"DefaultUploadTask{FileKind: 0, Path: /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/files/checkpoints/1344/default/ocdbt.process_0/d, Name: checkpoints/1344/default/ocdbt.process_0/d, Url: https://storage.googleapis.com/wandb-production.appspot.com/devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0/checkpoints/1344/default/ocdbt.process_0/d?X-Goog-Algorithm=GOOG4-RSA-SHA256&X-Goog-Credential=gorilla-files-url-signer-man%40wandb-production.iam.gserviceaccount.com%2F20250807%2Fauto%2Fstorage%2Fgoog4_request&X-Goog-Date=20250807T185850Z&X-Goog-Expires=86399&X-Goog-Signature=b8525db92537d74ca7531fe9b989419a44c44690a6190d26d385ab4f7405325db67a6b3044a2193c4556388e094a25e2d87122e8ee32cfbe218d8cf21415867a322cce1d52436e8520df53d1f6edf0393740a2dc298313ef508b5746f13eaffebf73d06d64207e53359ae67d56e32054b47c74fb22b3f490252ba9284ab82c1559f8711807036476e22a08cc7304e174b54e704f1f0d279dd7722007ab5734572cefdd05d0a0e154dccf1e2eff42d735d2583bf7446f1c2c83195987322e9cea0708e78b9b94f36457af6f7a4961fbbb1dffbfcb652af1c6004fdb95d916824c68c458d972e9d0859dd98c2652b7f2266c2317ea15f0b40eb46eeefa945bb46f&X-Goog-SignedHeaders=host&X-User=davidtimaeus, Size: 0}"}
+{"time":"2025-08-07T18:58:50.63463426Z","level":"INFO","msg":"handler: operation stats","stats":{"operations":[{"desc":"uploading wandb-summary.json","runtime_seconds":0.371770495,"progress":"6.7KB/6.7KB"},{"desc":"uploading output.log","runtime_seconds":0.356208437,"progress":"1.0MB/1.1MB"}],"total_operations":2}}
+{"time":"2025-08-07T18:58:51.79900197Z","level":"INFO","msg":"fileTransfer: Close: file transfer manager closed"}
+{"time":"2025-08-07T18:58:52.084306801Z","level":"INFO","msg":"stream: closing","id":"wp_1.0_alpha_0"}
+{"time":"2025-08-07T18:58:52.084338051Z","level":"INFO","msg":"handler: closed","stream_id":"wp_1.0_alpha_0"}
+{"time":"2025-08-07T18:58:52.084356454Z","level":"INFO","msg":"writer: Close: closed","stream_id":"wp_1.0_alpha_0"}
+{"time":"2025-08-07T18:58:52.084481708Z","level":"INFO","msg":"sender: closed","stream_id":"wp_1.0_alpha_0"}
+{"time":"2025-08-07T18:58:52.084769871Z","level":"INFO","msg":"stream: closed","id":"wp_1.0_alpha_0"}
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/logs/debug.log b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/logs/debug.log
new file mode 100644
index 0000000000000000000000000000000000000000..74f255b27bfd4816631b5212ce70288ef3c4c859
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/logs/debug.log
@@ -0,0 +1,28 @@
+2025-08-07 18:02:11,436 INFO    MainThread:3869025 [wandb_setup.py:_flush():80] Current SDK version is 0.21.0
+2025-08-07 18:02:11,436 INFO    MainThread:3869025 [wandb_setup.py:_flush():80] Configure stats pid to 3869025
+2025-08-07 18:02:11,436 INFO    MainThread:3869025 [wandb_setup.py:_flush():80] Loading settings from /root/.config/wandb/settings
+2025-08-07 18:02:11,436 INFO    MainThread:3869025 [wandb_setup.py:_flush():80] Loading settings from /workspace/timaeus/projects/rl/wandb/settings
+2025-08-07 18:02:11,436 INFO    MainThread:3869025 [wandb_setup.py:_flush():80] Loading settings from environment variables
+2025-08-07 18:02:11,437 INFO    MainThread:3869025 [wandb_init.py:setup_run_log_directory():703] Logging user logs to /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/logs/debug.log
+2025-08-07 18:02:11,437 INFO    MainThread:3869025 [wandb_init.py:setup_run_log_directory():704] Logging internal logs to /workspace/timaeus/projects/rl/wandb/run-20250807_180211-wp_1.0_alpha_0/logs/debug-internal.log
+2025-08-07 18:02:11,437 INFO    MainThread:3869025 [wandb_init.py:init():830] calling init triggers
+2025-08-07 18:02:11,437 INFO    MainThread:3869025 [wandb_init.py:init():835] wandb.init called with sweep_config: {}
+config: {'env_size': 13, 'env_layout': 'tree', 'wall_prob': 1.0, 'env_corner_size': 1, 'env_terminate_after_corner': False, 'obs_level_of_detail': 0, 'img_level_of_detail': 1, 'env_penalize_time': False, 'net_cnn_type': 'large', 'net_rnn_type': 'ff', 'net_width': 256, 'ued': 'plr', 'prob_shift': 0.0, 'num_train_levels': 2048, 'plr_buffer_size': 4096, 'plr_temperature': 0.1, 'plr_staleness_coeff': 0.1, 'plr_prob_replay': 0.5, 'plr_regret_estimator': 'maxmc-actor', 'plr_robust': False, 'num_mutate_steps': 12, 'prob_mutate_shift': 0.0, 'chain_mutate': True, 'mutate_cheese': True, 'train_proxy_critic': False, 'plr_proxy_shaping': False, 'proxy_name': 'proxy_corner', 'plr_proxy_shaping_coeff': 0.5, 'clipping': False, 'ppo_lr': 5e-05, 'ppo_gamma': 0.999, 'ppo_clip_eps': 0.1, 'ppo_gae_lambda': 0.95, 'ppo_entropy_coeff': 0.001, 'ppo_critic_coeff': 0.5, 'ppo_proxy_critic_coeff': 0.5, 'ppo_max_grad_norm': 0.5, 'ppo_lr_annealing': False, 'num_minibatches_per_epoch': 4, 'num_epochs_per_cycle': 5, 'num_total_env_steps': 200000000, 'num_env_steps_per_cycle': 128, 'num_parallel_envs': 256, 'console_log': True, 'log_gifs': True, 'log_imgs': True, 'log_hists': False, 'num_cycles_per_log': 32, 'num_cycles_per_eval': 32, 'num_cycles_per_gifs': 1024, 'num_cycles_per_big_eval': 1024, 'use_fixed_eval_levels': False, 'evals_num_env_steps': 512, 'evals_num_levels': 256, 'gif_grid_width': 16, 'level_splayer': 'mouse', 'checkpointing': True, 'keep_all_checkpoints': True, 'max_num_checkpoints': 1, 'num_cycles_per_checkpoint': 64, 'seed': 42, '_wandb': {}}
+2025-08-07 18:02:11,437 INFO    MainThread:3869025 [wandb_init.py:init():871] starting backend
+2025-08-07 18:02:11,650 INFO    MainThread:3869025 [wandb_init.py:init():874] sending inform_init request
+2025-08-07 18:02:11,656 INFO    MainThread:3869025 [wandb_init.py:init():882] backend started and connected
+2025-08-07 18:02:11,660 INFO    MainThread:3869025 [wandb_init.py:init():953] updated telemetry
+2025-08-07 18:02:11,669 INFO    MainThread:3869025 [wandb_init.py:init():977] communicating run to backend with 90.0 second timeout
+2025-08-07 18:02:12,236 INFO    MainThread:3869025 [wandb_init.py:init():1029] starting run threads in backend
+2025-08-07 18:02:12,575 INFO    MainThread:3869025 [wandb_run.py:_console_start():2458] atexit reg
+2025-08-07 18:02:12,575 INFO    MainThread:3869025 [wandb_run.py:_redirect():2306] redirect: wrap_raw
+2025-08-07 18:02:12,575 INFO    MainThread:3869025 [wandb_run.py:_redirect():2375] Wrapping output streams.
+2025-08-07 18:02:12,575 INFO    MainThread:3869025 [wandb_run.py:_redirect():2398] Redirects installed.
+2025-08-07 18:02:12,578 INFO    MainThread:3869025 [wandb_init.py:init():1075] run started, returning control to user process
+2025-08-07 18:58:49,629 INFO    MainThread:3869025 [wandb_run.py:_finish():2224] finishing run devinterp/jaxgmg-wall-prob-tree/wp_1.0_alpha_0
+2025-08-07 18:58:49,630 INFO    MainThread:3869025 [wandb_run.py:_atexit_cleanup():2423] got exitcode: 0
+2025-08-07 18:58:49,631 INFO    MainThread:3869025 [wandb_run.py:_restore():2405] restore
+2025-08-07 18:58:49,631 INFO    MainThread:3869025 [wandb_run.py:_restore():2411] restore done
+2025-08-07 18:58:52,061 INFO    MainThread:3869025 [wandb_run.py:_footer_history_summary_info():3903] rendering history
+2025-08-07 18:58:52,076 INFO    MainThread:3869025 [wandb_run.py:_footer_history_summary_info():3935] rendering summary
+2025-08-07 18:58:52,083 INFO    MainThread:3869025 [wandb_run.py:_footer_sync_info():3864] logging synced files
diff --git a/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/run-wp_1.0_alpha_0.wandb b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/run-wp_1.0_alpha_0.wandb
new file mode 100644
index 0000000000000000000000000000000000000000..43c1fcdc61eb1255f6d12a3479dcb4c6ef88e64d
--- /dev/null
+++ b/wall-prob-tree-13x13/run-20250807_180211-wp_1.0_alpha_0/run-wp_1.0_alpha_0.wandb
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:30840bd3709603c418f135d6146e0ab88db7cf8f8acc226e8183fdac8a23c86d
+size 6251918